fixed redundency reports

This commit is contained in:
Tristan Smith 2024-06-10 10:20:08 -04:00
parent 42457dcb0b
commit 8a57c7d020

8
qt.py
View file

@ -47,6 +47,9 @@ class IPReporter(QMainWindow):
self.listening = False
# Set to keep track of unique IP addresses
self.unique_ips = set()
def extract_packet_info(self, packet):
# Check if the packet is an IP packet with UDP layer
if IP in packet and UDP in packet:
@ -61,6 +64,11 @@ class IPReporter(QMainWindow):
udp_destination_port == destination_port):
# Extract the MAC address from the Ethernet layer
source_mac = packet[Ether].src
# Check if the IP address is already in the set
if source_ip not in self.unique_ips:
# Add the IP address to the set
self.unique_ips.add(source_ip)
# Display the information in the tree view
self.tree.addTopLevelItem(QTreeWidgetItem([source_ip, source_mac]))