From 7fceebc82e5a7102f5508186a5ab31d54845eaee Mon Sep 17 00:00:00 2001
From: Tristan Smith <tristan.smith@gmail.com>
Date: Fri, 31 May 2024 09:28:48 -0400
Subject: [PATCH] Added a simple gui interface. Shows the IP and mac address,
 allows for exporting as a text file.

---
 gui.py | 90 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 gui.py

diff --git a/gui.py b/gui.py
new file mode 100644
index 0000000..48ed7a3
--- /dev/null
+++ b/gui.py
@@ -0,0 +1,90 @@
+import tkinter as tk
+from tkinter import ttk, filedialog
+from scapy.all import IP, UDP, Ether, sniff
+
+# Define the destination IP address and UDP ports to filter
+destination_ip = '255.255.255.255'  # Destination IP address
+source_port = 14236  # Source port
+destination_port = 14235  # Destination port
+
+# Function to extract packet information and display it in the tree view
+def extract_packet_info(packet):
+    # Check if the packet is an IP packet with UDP layer
+    if IP in packet and UDP in packet:
+        # Extract the source IP address, source port, and destination port
+        source_ip = packet[IP].src
+        udp_source_port = packet[UDP].sport
+        udp_destination_port = packet[UDP].dport
+        
+        # Check if the packet matches the specified destination IP address and UDP ports
+        if (packet[IP].dst == destination_ip and
+            udp_source_port == source_port and
+            udp_destination_port == destination_port):
+            # Extract the MAC address from the Ethernet layer
+            source_mac = packet[Ether].src
+            # Display the information in the tree view
+            tree.insert("", tk.END, values=(source_ip, source_mac))
+
+# Function to start listening for packets
+def listen_for_packets():
+    sniff(prn=extract_packet_info, filter="udp and ip", store=0)
+
+# Function to start/stop the packet listening
+def toggle_listening():
+    global listening
+    if not listening:
+        # Start listening for packets in a separate thread
+        import threading
+        threading.Thread(target=listen_for_packets, daemon=True).start()
+        start_button.config(text="Stop")
+        status_label.config(text="Listening...")
+        listening = True
+    else:
+        # Stop listening for packets (terminate the program)
+        start_button.config(text="Start")
+        status_label.config(text="Stopped")
+        listening = False
+
+# Function to export the data to a file
+def export_data():
+    # Get the file path from the user
+    file_path = filedialog.asksaveasfilename(defaultextension=".txt", 
+                                             filetypes=[("Text files", "*.txt"), 
+                                                        ("All files", "*.*")])
+    if file_path:
+        with open(file_path, "w") as file:
+            for row_id in tree.get_children():
+                row = tree.item(row_id)['values']
+                file.write(f"IP Address: {row[0]}, MAC Address: {row[1]}\n")
+        status_label.config(text=f"Data exported to {file_path}")
+
+# Create the main window
+root = tk.Tk()
+root.title("IP Reporter")
+
+# Create a frame for the tree view
+frame = tk.Frame(root)
+frame.pack(padx=10, pady=10)
+
+# Create a tree view with columns for IP and MAC
+columns = ("IP Address", "MAC Address")
+tree = ttk.Treeview(frame, columns=columns, show="headings")
+tree.heading("IP Address", text="IP Address")
+tree.heading("MAC Address", text="MAC Address")
+tree.pack(fill=tk.BOTH, expand=True)
+
+# Create a start button
+start_button = tk.Button(root, text="Start", command=toggle_listening)
+start_button.pack(pady=5)
+
+# Create an export button
+export_button = tk.Button(root, text="Export", command=export_data)
+export_button.pack(pady=5)
+
+# Create a status label
+status_label = tk.Label(root, text="Stopped")
+status_label.pack(pady=5)
+
+# Start the Tkinter event loop
+listening = False
+root.mainloop()