Removing all the debug stuff since we found the bug.

This commit is contained in:
Yaro Kasear 2025-04-11 08:21:56 -05:00
parent 299d6e27fa
commit 13749d3d1c

View file

@ -25,28 +25,22 @@ def get_clients_on_ap(capture, ap_bssid):
for packet in capture: for packet in capture:
try: try:
if not hasattr(packet, 'wlan'): if not hasattr(packet, 'wlan'):
print("No WLAN layer found in packet")
continue continue
sa = getattr(packet.wlan, 'sa', '').lower() sa = getattr(packet.wlan, 'sa', '').lower()
da = getattr(packet.wlan, 'da', '').lower() da = getattr(packet.wlan, 'da', '').lower()
bssid = getattr(packet.wlan, 'bssid', '').lower() bssid = getattr(packet.wlan, 'bssid', '').lower()
print(f"Processing packet: SA={sa}, DA={da}, BSSID={bssid}")
# Count any frame *to or from* a client, if AP is involved # Count any frame *to or from* a client, if AP is involved
if bssid == ap_bssid or sa == ap_bssid or da == ap_bssid: if bssid == ap_bssid or sa == ap_bssid or da == ap_bssid:
print("Found packet related to AP")
# If it's the AP sending, add the destination (client) # If it's the AP sending, add the destination (client)
if sa == ap_bssid and da and da != ap_bssid: if sa == ap_bssid and da and da != ap_bssid:
print(f"Adding client: {da}")
clients.add(da) clients.add(da)
# If it's the client sending, add the source # If it's the client sending, add the source
elif sa and sa != ap_bssid: elif sa and sa != ap_bssid:
print(f"Adding client: {sa}")
clients.add(sa) clients.add(sa)
except AttributeError: except AttributeError:
print("Attribute error in packet processing")
continue continue
return len(clients) return len(clients)
@ -61,9 +55,6 @@ def analyze_pcap(pcapng_path, start_ts, end_ts, ap_bssid, ap_channel):
include_raw=False include_raw=False
) )
# Show for debug purposes the packets in the filtered capture as well as the filter applied
print(f"[+] Filter applied: time >= {start_ts} && time <= {end_ts}")
clients_on_ap = get_clients_on_ap(cap, ap_bssid.lower()) clients_on_ap = get_clients_on_ap(cap, ap_bssid.lower())
# Placeholder: Logic will be added for: # Placeholder: Logic will be added for:
@ -86,7 +77,6 @@ def main():
include_raw=False, include_raw=False,
keep_packets=False keep_packets=False
) )
cap.set_debug(True)
# Checking if the pcapng file is valid # Checking if the pcapng file is valid
count = 0 count = 0