aegis-node 0.1.0__py3-none-any.whl

aegis_node/__init__.py

@@ -0,0 +1,4 @@
+"""Aegis WiFi Node Agent — connects a local USB WiFi adapter to the Aegis portal."""
+__version__ = '0.1.0'
+
+DEFAULT_SERVER = 'https://portal.aegis.divoltbtc.com'

aegis_node/__main__.py

@@ -0,0 +1,4 @@
+"""Allow `python -m aegis_node`."""
+from .agent import main
+
+main()

aegis_node/adapters.py

@@ -0,0 +1,128 @@
+"""USB WiFi adapter detection and capability probing."""
+from __future__ import annotations
+
+import logging
+import re
+import shutil
+import subprocess
+
+log = logging.getLogger('aegis.wifi_node.adapters')
+
+# VID:PID → (chipset, driver, monitor_default, inject_default)
+KNOWN_ADAPTERS: dict[str, tuple[str, str, bool, bool]] = {
+    # Proven pentest adapters
+    '0e8d:7612': ('MT7612U', 'mt76x2u', True, True),
+    '148f:761a': ('MT7612U', 'mt76x2u', True, True),
+    '0b05:17d1': ('MT7612U', 'mt76x2u', True, True),  # ASUS USB-AC54
+    '0cf3:9271': ('AR9271', 'ath9k_htc', True, True),  # Alfa AWUS036NHA
+    '0bda:8812': ('RTL8812AU', 'rtl8812au', True, True),
+    '0bda:881a': ('RTL8812AU', 'rtl8812au', True, True),
+    '0846:9052': ('RTL8812AU', 'rtl8812au', True, True),  # Netgear A6100
+    '0bda:8187': ('RTL8187L', 'rtl8187', True, True),  # WIFISKY WS-980WG
+    # Managed-mode only
+    '0bda:c832': ('RTL8832CU', 'rtl8832cu', False, False),  # NEWFAST NF-U353
+    '0bda:b832': ('RTL8832CU', 'rtl8832cu', False, False),
+    '0bda:c82c': ('RTL8822CU', 'rtl8822cu', False, False),
+    '0bda:8179': ('RTL8188EUS', 'rtl8188eu', False, False),
+    '2357:011e': ('RTL8812AU', 'rtl8812au', True, True),  # TP-Link T4U
+}
+
+
+def _run(cmd: list[str], timeout: int = 10) -> str:
+    try:
+        r = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout)
+        return r.stdout + r.stderr
+    except (subprocess.TimeoutExpired, FileNotFoundError, OSError):
+        return ''
+
+
+def detect_adapters() -> list[dict]:
+    """Detect USB WiFi adapters via lsusb + iw dev."""
+    adapters = []
+    lsusb = _run(['lsusb'])
+    for line in lsusb.splitlines():
+        m = re.match(r'Bus \d+ Device \d+: ID (\w{4}:\w{4})\s+(.*)', line)
+        if not m:
+            continue
+        vid_pid = m.group(1).lower()
+        desc = m.group(2).strip()
+        if vid_pid not in KNOWN_ADAPTERS:
+            continue
+        chipset, driver, mon_default, inj_default = KNOWN_ADAPTERS[vid_pid]
+        adapters.append({
+            'vid_pid': vid_pid,
+            'chipset': chipset,
+            'driver': driver,
+            'description': desc,
+            'cap_monitor_mode': mon_default,
+            'cap_injection': inj_default,
+        })
+
+    # Enrich with iw interface info
+    iw_out = _run(['iw', 'dev'])
+    ifaces = _parse_iw_dev(iw_out)
+    for a in adapters:
+        for iface in ifaces:
+            if iface.get('type') == 'managed':
+                a.setdefault('interface', iface['name'])
+                a.setdefault('mac', iface.get('addr', ''))
+                break
+
+    return adapters
+
+
+def _parse_iw_dev(output: str) -> list[dict]:
+    """Parse `iw dev` output into a list of interface dicts."""
+    ifaces = []
+    current: dict | None = None
+    for line in output.splitlines():
+        line = line.strip()
+        if line.startswith('Interface'):
+            if current:
+                ifaces.append(current)
+            current = {'name': line.split()[-1]}
+        elif current:
+            if line.startswith('addr'):
+                current['addr'] = line.split()[-1]
+            elif line.startswith('type'):
+                current['type'] = line.split()[-1]
+    if current:
+        ifaces.append(current)
+    return ifaces
+
+
+def probe_monitor_mode(interface: str) -> bool:
+    """Test if an interface supports monitor mode via iw."""
+    if not interface:
+        return False
+    out = _run(['iw', 'phy'])
+    return 'monitor' in out.lower()
+
+
+def probe_injection(interface: str) -> bool:
+    """Test packet injection via aireplay-ng (requires monitor mode active)."""
+    if not shutil.which('aireplay-ng'):
+        return False
+    out = _run(['aireplay-ng', '--test', interface], timeout=15)
+    return 'Injection is working!' in out
+
+
+def detect_host_tools() -> dict[str, bool]:
+    """Check which security tools are available on the host."""
+    tools = {}
+    for tool in ('airmon-ng', 'airodump-ng', 'aireplay-ng', 'aircrack-ng',
+                 'nmap', 'hashcat', 'hcxdumptool', 'hcxpcapngtool',
+                 'hostapd', 'dnsmasq', 'tcpdump', 'tshark', 'iw', 'iwconfig',
+                 'reaver', 'arpspoof', 'macchanger'):
+        tools[tool] = shutil.which(tool) is not None
+    return tools
+
+
+def get_host_info() -> dict:
+    """Collect host OS and hostname."""
+    import platform
+    return {
+        'os': f'{platform.system()} {platform.release()}',
+        'hostname': platform.node(),
+        'tools': detect_host_tools(),
+    }