tristan/ip-reporter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixed redundency reports

Browse source
This commit is contained in:
Tristan Smith 2024-06-10 10:20:08 -04:00
parent 42457dcb0b
commit 8a57c7d020
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
qt.py
View file

@ -47,6 +47,9 @@ class IPReporter(QMainWindow):
self.listening = False self.listening = False
# Set to keep track of unique IP addresses
self.unique_ips = set()
def extract_packet_info(self, packet): def extract_packet_info(self, packet):
# Check if the packet is an IP packet with UDP layer # Check if the packet is an IP packet with UDP layer
if IP in packet and UDP in packet: if IP in packet and UDP in packet:
@ -61,8 +64,13 @@ class IPReporter(QMainWindow):
udp_destination_port == destination_port): udp_destination_port == destination_port):
# Extract the MAC address from the Ethernet layer # Extract the MAC address from the Ethernet layer
source_mac = packet[Ether].src source_mac = packet[Ether].src
# Display the information in the tree view
self.tree.addTopLevelItem(QTreeWidgetItem([source_ip, source_mac])) # 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]))
def listen_for_packets(self): def listen_for_packets(self):
sniff(prn=self.extract_packet_info, filter="udp and ip", store=0) sniff(prn=self.extract_packet_info, filter="udp and ip", store=0)