yaro/wifi_test
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enhance client tracking and AP relationship mapping in handle_packet; update metrics to reflect active clients per AP.

Browse source
This commit is contained in:
Yaro Kasear 2025-05-01 09:06:18 -05:00
parent 50aad9ef15
commit de6f0b037b
1 changed files with 21 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

31
listener.py
View file

@ -17,6 +17,7 @@ aps = set()
ap_signals = defaultdict(list) # BSSID -> list of dBm ap_signals = defaultdict(list) # BSSID -> list of dBm
ssid_map = {} # BSSID -> SSID ssid_map = {} # BSSID -> SSID
ssid_signals = defaultdict(list) # SSID -> list of dBm ssid_signals = defaultdict(list) # SSID -> list of dBm
ap_clients = defaultdict(set)
# === Signal handling === # === Signal handling ===
def stop_sniff(signum, frame): def stop_sniff(signum, frame):
@ -65,17 +66,27 @@ def handle_packet(pkt):
dot11 = pkt[Dot11] dot11 = pkt[Dot11]
if dot11.type == 0 and dot11.subtype in (5, 8): # Probe Resp or Beacon # === Detect APs via beacons/probe responses ===
aps.add(dot11.addr2) if dot11.type == 0 and dot11.subtype in (5, 8): # Probe Response or Beacon
ssid = parse_ssid(pkt)
if ssid:
ssid_map[dot11.addr2] = ssid
else:
if dot11.addr1:
clients.add(dot11.addr1)
if dot11.addr2: if dot11.addr2:
clients.add(dot11.addr2) aps.add(dot11.addr2)
ssid = parse_ssid(pkt)
if ssid:
ssid_map[dot11.addr2] = ssid
# === Track all seen clients ===
if dot11.addr1:
clients.add(dot11.addr1)
if dot11.addr2:
clients.add(dot11.addr2)
# === Guess client <-> AP relationships ===
if dot11.addr1 in aps and dot11.addr2:
ap_clients[dot11.addr1].add(dot11.addr2)
elif dot11.addr2 in aps and dot11.addr1:
ap_clients[dot11.addr2].add(dot11.addr1)
# === Signal strength tracking ===
try: try:
signal_dbm = pkt[RadioTap].dBm_AntSignal signal_dbm = pkt[RadioTap].dBm_AntSignal
if dot11.addr2 in aps: if dot11.addr2 in aps:
@ -98,7 +109,7 @@ def write_csv(outfile):
"StrongestAPSignal": max([max(v) for v in ap_signals.values() if v], default=0), "StrongestAPSignal": max([max(v) for v in ap_signals.values() if v], default=0),
"AvgSSIDSignal": round(sum([sum(v)/len(v) for v in ssid_signals.values() if v]) / len(ssid_signals) if ssid_signals else 0, 2), "AvgSSIDSignal": round(sum([sum(v)/len(v) for v in ssid_signals.values() if v]) / len(ssid_signals) if ssid_signals else 0, 2),
"MaxSSIDSignal": max([max(v) for v in ssid_signals.values() if v], default=0), "MaxSSIDSignal": max([max(v) for v in ssid_signals.values() if v], default=0),
"ClientsOnAP": "N/A", "ClientsOnAP": sum(len(s) for s in ap_clients.values()),
"CiscoAvgReportedClients": "N/A", "CiscoAvgReportedClients": "N/A",
"CiscoMaxReportedClients": "N/A", "CiscoMaxReportedClients": "N/A",
"NumberofBSSIDsOnSSID": "N/A", "NumberofBSSIDsOnSSID": "N/A",