Refactor enrichment module by adding utility functions, enhancing CSV handling, and implementing SSID metrics extraction. Update run_test script to improve logging on speed test failures.

enrichment/metrics_signals.py

@@ -0,0 +1,126 @@
+# enrichment/metrics_signals.py
+
+from statistics import mean
+from enrichment.utils import get_channel_from_freq
+from enrichment.metrics_clients import get_clients_on_channel  # in case you want to consolidate later
+
+def get_aps(capture, ap_channel):
+    try:
+        ap_channel = int(ap_channel)
+    except ValueError:
+        print(f"[!] Could not parse channel number: {ap_channel}")
+        return 0
+    
+    aps = set()
+
+    for packet in capture:
+        try:
+            if 'radiotap' not in packet or 'wlan' not in packet:
+                continue
+
+            radio = packet.radiotap
+            wlan = packet.wlan
+
+            if not hasattr(radio, 'channel') or not hasattr(radio.channel, 'freq'):
+                continue
+
+            packet_freq = int(radio.channel.freq)
+            packet_channel = get_channel_from_freq(packet_freq)
+
+            if packet_channel != ap_channel:
+                continue
+
+            ts_hex = getattr(wlan, 'type_subtype', None)
+            if ts_hex is None:
+                continue
+
+            ts = int(ts_hex, 16)
+            if ts not in (5, 8):  # Beacon/Probe
+                continue
+
+            bssid = getattr(wlan, 'bssid', '').lower()
+            if bssid and bssid != 'ff:ff:ff:ff:ff:ff':
+                aps.add(bssid)
+
+        except Exception:
+            continue
+    
+    return aps
+
+
+def get_aps_on_channel(capture, ap_channel):
+    return len(get_aps(capture, ap_channel))
+
+
+def calculate_signal_strength_stats(capture, ap_channel):
+    try:
+        ap_channel = int(ap_channel)
+    except ValueError:
+        print(f"[!] Could not parse channel number: {ap_channel}")
+        return 0, 0
+
+    ap_signals = []
+    for packet in capture:
+        try:
+            if 'radiotap' not in packet or 'wlan' not in packet:
+                continue
+
+            radio = packet.radiotap
+            wlan = packet.wlan
+
+            if not hasattr(radio, 'channel') or not hasattr(radio.channel, 'freq'):
+                continue
+
+            packet_freq = int(radio.channel.freq)
+            packet_channel = get_channel_from_freq(packet_freq)
+
+            if packet_channel != ap_channel:
+                continue
+
+            ts_hex = getattr(wlan, 'type_subtype', None)
+            if ts_hex is None:
+                continue
+
+            ts = int(ts_hex, 16)
+            if ts not in (5, 8):  # Beacon or Probe
+                continue
+
+            signal_strength = getattr(radio, 'dbm_antsignal', None)
+            if signal_strength is not None:
+                ap_signals.append(int(signal_strength))
+
+        except Exception:
+            continue
+
+    return (mean(ap_signals), max(ap_signals)) if ap_signals else (0, 0)
+
+
+def get_unlinked_devices(capture, ap_channel):
+    aps = get_aps(capture, ap_channel)
+    ghost_clients = set()
+
+    for packet in capture:
+        try:
+            if 'radiotap' not in packet or 'wlan' not in packet:
+                continue
+
+            radio = packet.radiotap
+            wlan = packet.wlan
+
+            if not hasattr(radio, 'channel') or not hasattr(radio.channel, 'freq'):
+                continue
+
+            packet_freq = int(radio.channel.freq)
+            packet_channel = get_channel_from_freq(packet_freq)
+            if packet_channel != ap_channel:
+                continue
+
+            for mac in (getattr(wlan, 'sa', ''), getattr(wlan, 'da', '')):
+                mac = mac.lower()
+                if mac and mac != 'ff:ff:ff:ff:ff:ff' and mac not in aps:
+                    ghost_clients.add(mac)
+
+        except Exception:
+            continue
+
+    return len(ghost_clients)