Shane Smiskol
5 days ago
parent
c6b7a71fc2
commit
3a14daccc0
6 changed files with 330 additions and 1061 deletions
File diff suppressed because it is too large
Load Diff
@ -1,390 +0,0 @@ |
|||||||
import atexit |
|
||||||
import copy |
|
||||||
import dbus |
|
||||||
import threading |
|
||||||
import time |
|
||||||
import uuid |
|
||||||
from collections.abc import Callable |
|
||||||
from dataclasses import dataclass |
|
||||||
from enum import IntEnum |
|
||||||
|
|
||||||
from openpilot.common.swaglog import cloudlog |
|
||||||
from openpilot.system.ui.lib.networkmanager import * |
|
||||||
|
|
||||||
try: |
|
||||||
from openpilot.common.params import Params |
|
||||||
except ImportError: |
|
||||||
# Params/Cythonized modules are not available in zipapp |
|
||||||
Params = None |
|
||||||
|
|
||||||
TETHERING_IP_ADDRESS = "192.168.43.1" |
|
||||||
DEFAULT_TETHERING_PASSWORD = "swagswagcomma" |
|
||||||
|
|
||||||
|
|
||||||
class SecurityType(IntEnum): |
|
||||||
OPEN = 0 |
|
||||||
WPA = 1 |
|
||||||
WPA2 = 2 |
|
||||||
WPA3 = 3 |
|
||||||
UNSUPPORTED = 4 |
|
||||||
|
|
||||||
|
|
||||||
def get_security_type(flags: int, wpa_flags: int, rsn_flags: int) -> SecurityType: |
|
||||||
wpa_props = wpa_flags | rsn_flags |
|
||||||
|
|
||||||
# obtained by looking at flags of networks in the office as reported by an Android phone |
|
||||||
supports_wpa = NM_802_11_AP_SEC_PAIR_WEP40 | NM_802_11_AP_SEC_PAIR_WEP104 | NM_802_11_AP_SEC_GROUP_WEP40 | NM_802_11_AP_SEC_GROUP_WEP104 | NM_802_11_AP_SEC_KEY_MGMT_PSK; |
|
||||||
|
|
||||||
if (flags == NM_802_11_AP_FLAGS_NONE) or ((flags & NM_802_11_AP_FLAGS_WPS) and not (wpa_props & supports_wpa)): |
|
||||||
return SecurityType.OPEN |
|
||||||
elif (flags & NM_802_11_AP_FLAGS_PRIVACY) and (wpa_props & supports_wpa) and not (wpa_props & NM_802_11_AP_SEC_KEY_MGMT_802_1X): |
|
||||||
return SecurityType.WPA |
|
||||||
else: |
|
||||||
cloudlog.warning(f"Unsupported network! flags: {flags}, wpa_flags: {wpa_flags}, rsn_flags: {rsn_flags}") |
|
||||||
return SecurityType.UNSUPPORTED |
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True) |
|
||||||
class Network: |
|
||||||
ssid: str |
|
||||||
strength: int |
|
||||||
is_connected: bool |
|
||||||
security_type: SecurityType |
|
||||||
is_saved: bool |
|
||||||
|
|
||||||
@classmethod |
|
||||||
def from_dbus(cls, ssid: str, aps: list["AccessPoint"], active_ap_path: dbus.ObjectPath, is_saved: bool) -> "Network": |
|
||||||
# we only want to show the strongest AP for each Network/SSID |
|
||||||
strongest_ap = max(aps, key=lambda ap: ap.strength) |
|
||||||
|
|
||||||
is_connected = any(ap.ap_path == active_ap_path for ap in aps) # TODO: just any is_connected aps! |
|
||||||
security_type = get_security_type(strongest_ap.flags, strongest_ap.wpa_flags, strongest_ap.rsn_flags) |
|
||||||
|
|
||||||
return cls( |
|
||||||
ssid=ssid, |
|
||||||
strength=strongest_ap.strength, |
|
||||||
is_connected=is_connected, |
|
||||||
security_type=security_type, |
|
||||||
is_saved=is_saved, |
|
||||||
) |
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True) |
|
||||||
class AccessPoint: |
|
||||||
ssid: str |
|
||||||
bssid: str |
|
||||||
strength: int |
|
||||||
is_connected: bool |
|
||||||
flags: int |
|
||||||
wpa_flags: int |
|
||||||
rsn_flags: int |
|
||||||
ap_path: dbus.ObjectPath |
|
||||||
|
|
||||||
@classmethod |
|
||||||
def from_dbus(cls, ap_props: dbus.Interface, ap_path: dbus.ObjectPath, active_ap_path: dbus.ObjectPath) -> "AccessPoint": |
|
||||||
ssid = bytes(ap_props.Get(NM_ACCESS_POINT_IFACE, "Ssid")).decode("utf-8", "replace") |
|
||||||
bssid = str(ap_props.Get(NM_ACCESS_POINT_IFACE, "HwAddress")) |
|
||||||
strength = int(ap_props.Get(NM_ACCESS_POINT_IFACE, "Strength")) |
|
||||||
flags = int(ap_props.Get(NM_ACCESS_POINT_IFACE, "Flags")) |
|
||||||
wpa_flags = int(ap_props.Get(NM_ACCESS_POINT_IFACE, "WpaFlags")) |
|
||||||
rsn_flags = int(ap_props.Get(NM_ACCESS_POINT_IFACE, "RsnFlags")) |
|
||||||
|
|
||||||
return cls( |
|
||||||
ssid=ssid, |
|
||||||
bssid=bssid, |
|
||||||
strength=strength, |
|
||||||
is_connected=ap_path == active_ap_path, |
|
||||||
flags=flags, |
|
||||||
wpa_flags=wpa_flags, |
|
||||||
rsn_flags=rsn_flags, |
|
||||||
ap_path=ap_path, |
|
||||||
) |
|
||||||
|
|
||||||
|
|
||||||
class WifiManager: |
|
||||||
def __init__(self): |
|
||||||
self._networks = [] # a network can be comprised of multiple APs |
|
||||||
self._active = True # used to not run this cycle when not in settings |
|
||||||
self._running = True |
|
||||||
|
|
||||||
# DBus and NetworkManager setup |
|
||||||
self._bus = dbus.SystemBus() |
|
||||||
self._nm = dbus.Interface(self._bus.get_object(NM, NM_PATH), NM_IFACE) |
|
||||||
self._props = dbus.Interface(self._bus.get_object(NM, NM_PATH), NM_PROPERTIES_IFACE) |
|
||||||
|
|
||||||
self._bus2 = dbus.SystemBus(private=True) |
|
||||||
|
|
||||||
# State |
|
||||||
self._connecting_to_ssid: str = "" |
|
||||||
|
|
||||||
# Callbacks |
|
||||||
# TODO: some of these are called from threads, either: |
|
||||||
# 1. make sure this is fine |
|
||||||
# 2. add callback event list that user can call from main thread to get callbacks safely |
|
||||||
self._need_auth: Callable[[str], None] | None = None |
|
||||||
self._activated: Callable[[], None] | None = None |
|
||||||
self._forgotten: Callable[[str], None] | None = None |
|
||||||
self._networks_updated: Callable[[list[Network]], None] | None = None |
|
||||||
self._disconnected: Callable[[], None] | None = None |
|
||||||
|
|
||||||
self._thread = threading.Thread(target=self._run, daemon=True) |
|
||||||
self._thread.start() |
|
||||||
|
|
||||||
self._state_thread = threading.Thread(target=self._monitor_state, daemon=True) |
|
||||||
self._state_thread.start() |
|
||||||
|
|
||||||
atexit.register(self.stop) |
|
||||||
|
|
||||||
def __del__(self): |
|
||||||
self.stop() |
|
||||||
|
|
||||||
def stop(self): |
|
||||||
self._running = False |
|
||||||
self._thread.join() |
|
||||||
self._state_thread.join() |
|
||||||
self._bus.close() |
|
||||||
self._bus2.close() |
|
||||||
|
|
||||||
def _monitor_state(self): |
|
||||||
prev_state = -1 |
|
||||||
|
|
||||||
device_path = self._get_wifi_device() |
|
||||||
props_dev = dbus.Interface(self._bus2.get_object(NM, device_path), NM_PROPERTIES_IFACE) |
|
||||||
_props = dbus.Interface(self._bus2.get_object(NM, NM_PATH), NM_PROPERTIES_IFACE) |
|
||||||
|
|
||||||
while self._running: |
|
||||||
if self._active: |
|
||||||
print('moritring state ACTivE!!1') |
|
||||||
dev_state = int(props_dev.Get(NM_DEVICE_IFACE, "State")) |
|
||||||
state_reason = props_dev.Get(NM_DEVICE_IFACE, "StateReason") # (u state, u reason) |
|
||||||
reason = int(state_reason[1]) if isinstance(state_reason, (list, tuple)) and len(state_reason) == 2 else 0 |
|
||||||
|
|
||||||
if dev_state != prev_state: |
|
||||||
print(f" WiFi device state change: {dev_state}, reason: {reason}") |
|
||||||
if dev_state == NMDeviceState.NEED_AUTH and reason == NM_DEVICE_STATE_REASON_SUPPLICANT_DISCONNECT and self._connecting_to_ssid: |
|
||||||
print('------ NEED AUTH - SUPPLICANT DISCONNECT') |
|
||||||
self.forget_connection(self._connecting_to_ssid, block=True) |
|
||||||
if self._need_auth is not None: |
|
||||||
self._need_auth(self._connecting_to_ssid) |
|
||||||
self._connecting_to_ssid = "" |
|
||||||
elif dev_state == NMDeviceState.ACTIVATED: |
|
||||||
print('------ ACTIVATED') |
|
||||||
if self._activated is not None: |
|
||||||
self._activated() |
|
||||||
self._connecting_to_ssid = "" |
|
||||||
elif dev_state == NMDeviceState.DISCONNECTED: |
|
||||||
print('------ DISCONNECTED') |
|
||||||
self._connecting_to_ssid = "" |
|
||||||
if self._disconnected is not None: |
|
||||||
self._disconnected() |
|
||||||
|
|
||||||
print() |
|
||||||
|
|
||||||
if self._connecting_to_ssid: |
|
||||||
print(' CONNECTING', self._connecting_to_ssid) |
|
||||||
|
|
||||||
prev_state = dev_state |
|
||||||
|
|
||||||
time.sleep(1 / 2.) |
|
||||||
|
|
||||||
def set_callbacks(self, need_auth: Callable[[str], None], |
|
||||||
activated: Callable[[], None] | None, |
|
||||||
forgotten: Callable[[str], None], |
|
||||||
networks_updated: Callable[[list[Network]], None], |
|
||||||
disconnected: Callable[[], None]): |
|
||||||
self._need_auth = need_auth |
|
||||||
self._activated = activated |
|
||||||
self._forgotten = forgotten |
|
||||||
self._networks_updated = networks_updated |
|
||||||
self._disconnected = disconnected |
|
||||||
|
|
||||||
def _run(self): |
|
||||||
i = 0 |
|
||||||
while self._running: |
|
||||||
if self._active: |
|
||||||
print('we;re acti!!!!!!!!!!!!') |
|
||||||
# Scan for networks every 5 seconds |
|
||||||
if i % 5 == 0: |
|
||||||
# TODO: should watch when scan is complete, but this is more than good enough for now |
|
||||||
self._update_networks() |
|
||||||
self._request_scan() |
|
||||||
|
|
||||||
i += 1 |
|
||||||
time.sleep(1) |
|
||||||
|
|
||||||
def set_active(self, active: bool): |
|
||||||
print('SETTING ACTIVE', active) |
|
||||||
self._active = active |
|
||||||
|
|
||||||
def _get_wifi_device(self) -> dbus.ObjectPath | None: |
|
||||||
# TODO: cache if slow |
|
||||||
t = time.monotonic() |
|
||||||
device_paths = self._nm.GetDevices() |
|
||||||
# print(f'DEVICE PATHS: {device_paths}') |
|
||||||
|
|
||||||
wifi_device = None |
|
||||||
for device_path in device_paths: |
|
||||||
dev_props = dbus.Interface(self._bus.get_object(NM, device_path), NM_PROPERTIES_IFACE) |
|
||||||
dev_type = dev_props.Get(NM_DEVICE_IFACE, "DeviceType") |
|
||||||
|
|
||||||
if dev_type == NM_DEVICE_TYPE_WIFI: |
|
||||||
wifi_device = device_path |
|
||||||
break |
|
||||||
|
|
||||||
# print(f"Got wifi device in {time.monotonic() - t}s: {wifi_device}") |
|
||||||
return wifi_device |
|
||||||
|
|
||||||
def connect_to_network(self, ssid: str, password: str | None): |
|
||||||
def worker(): |
|
||||||
t = time.monotonic() |
|
||||||
|
|
||||||
# Clear all connections that may already exist to the network we are connecting |
|
||||||
self._connecting_to_ssid = ssid |
|
||||||
self.forget_connection(ssid, block=True) |
|
||||||
|
|
||||||
is_hidden = False |
|
||||||
|
|
||||||
connection = { |
|
||||||
'connection': { |
|
||||||
'type': '802-11-wireless', |
|
||||||
'uuid': str(uuid.uuid4()), |
|
||||||
'id': f'openpilot connection {ssid}', |
|
||||||
'autoconnect-retries': 0, |
|
||||||
}, |
|
||||||
'802-11-wireless': { |
|
||||||
'ssid': dbus.ByteArray(ssid.encode("utf-8")), |
|
||||||
'hidden': is_hidden, |
|
||||||
'mode': 'infrastructure', |
|
||||||
}, |
|
||||||
'ipv4': { |
|
||||||
'method': 'auto', |
|
||||||
'dns-priority': 600, |
|
||||||
}, |
|
||||||
'ipv6': {'method': 'ignore'}, |
|
||||||
} |
|
||||||
|
|
||||||
if password is not None: |
|
||||||
connection['802-11-wireless-security'] = { |
|
||||||
'key-mgmt': 'wpa-psk', |
|
||||||
'auth-alg': 'open', |
|
||||||
'psk': password, |
|
||||||
} |
|
||||||
|
|
||||||
settings = dbus.Interface( |
|
||||||
self._bus.get_object(NM, NM_SETTINGS_PATH), |
|
||||||
NM_SETTINGS_IFACE |
|
||||||
) |
|
||||||
|
|
||||||
conn_path = settings.AddConnection(connection) |
|
||||||
|
|
||||||
print('Added connection', conn_path) |
|
||||||
|
|
||||||
print(f'Connecting to network took {time.monotonic() - t}s') |
|
||||||
|
|
||||||
self.activate_connection(ssid) |
|
||||||
|
|
||||||
threading.Thread(target=worker, daemon=True).start() |
|
||||||
|
|
||||||
def _get_connections(self) -> list[dbus.ObjectPath]: |
|
||||||
settings_iface = dbus.Interface(self._bus.get_object(NM, NM_SETTINGS_PATH), NM_SETTINGS_IFACE) |
|
||||||
return settings_iface.ListConnections() |
|
||||||
|
|
||||||
def _connection_by_ssid(self, ssid: str, known_connections: list[dbus.ObjectPath] | None = None) -> dbus.ObjectPath | None: |
|
||||||
for conn_path in known_connections or self._get_connections(): |
|
||||||
conn_props = dbus.Interface(self._bus.get_object(NM, conn_path), NM_CONNECTION_IFACE) |
|
||||||
settings = conn_props.GetSettings() |
|
||||||
if "802-11-wireless" in settings and bytes(settings["802-11-wireless"]["ssid"]).decode("utf-8", "replace") == ssid: |
|
||||||
return conn_path |
|
||||||
return None |
|
||||||
|
|
||||||
def forget_connection(self, ssid: str, block: bool = False): |
|
||||||
def worker(): |
|
||||||
t = time.monotonic() |
|
||||||
conn_path = self._connection_by_ssid(ssid) |
|
||||||
print(f'Finding connection by SSID took {time.monotonic() - t}s: {conn_path}') |
|
||||||
if conn_path is not None: |
|
||||||
conn_iface = dbus.Interface(self._bus.get_object(NM, conn_path), NM_CONNECTION_IFACE) |
|
||||||
conn_iface.Delete() |
|
||||||
print(f'Forgetting connection took {time.monotonic() - t}s') |
|
||||||
if self._forgotten is not None: |
|
||||||
self._forgotten(ssid) |
|
||||||
|
|
||||||
# TODO: make a helper when it makes sense |
|
||||||
if block: |
|
||||||
worker() |
|
||||||
else: |
|
||||||
threading.Thread(target=worker, daemon=True).start() |
|
||||||
|
|
||||||
def activate_connection(self, ssid: str): |
|
||||||
t = time.monotonic() |
|
||||||
conn_path = self._connection_by_ssid(ssid) |
|
||||||
if conn_path is not None: |
|
||||||
self._connecting_to_ssid = ssid |
|
||||||
device_path = self._get_wifi_device() |
|
||||||
if device_path is None: |
|
||||||
cloudlog.warning("No WiFi device found") |
|
||||||
return |
|
||||||
|
|
||||||
print(f'Activating connection to {ssid}') |
|
||||||
self._nm.ActivateConnection(conn_path, device_path, dbus.ObjectPath("/")) |
|
||||||
print(f"Activated connection in {time.monotonic() - t}s") |
|
||||||
# FIXME: deadlock issue with ui |
|
||||||
# if self._activated is not None: |
|
||||||
# self._activated() |
|
||||||
|
|
||||||
def _request_scan(self): |
|
||||||
device_path = self._get_wifi_device() |
|
||||||
if device_path is None: |
|
||||||
cloudlog.warning("No WiFi device found") |
|
||||||
return |
|
||||||
|
|
||||||
wifi_iface = dbus.Interface(self._bus.get_object(NM, device_path), NM_WIRELESS_IFACE) |
|
||||||
try: |
|
||||||
wifi_iface.RequestScan({}) |
|
||||||
print('Requested scan') |
|
||||||
except dbus.exceptions.DBusException as e: |
|
||||||
# TODO: copilot is wrong, this never happens |
|
||||||
if "org.freedesktop.NetworkManager.Device.Error.AlreadyScanning" in str(e): |
|
||||||
print('Already scanning, skipping') |
|
||||||
else: |
|
||||||
cloudlog.exception("Failed to request scan") |
|
||||||
|
|
||||||
def _update_networks(self): |
|
||||||
# TODO: only run this function on scan complete! |
|
||||||
print('UPDATING NETWORKS!!!!') |
|
||||||
|
|
||||||
device_path = self._get_wifi_device() |
|
||||||
if device_path is None: |
|
||||||
cloudlog.warning("No WiFi device found") |
|
||||||
return |
|
||||||
|
|
||||||
wifi_iface = dbus.Interface(self._bus.get_object(NM, device_path), NM_WIRELESS_IFACE) |
|
||||||
dev_props = dbus.Interface(self._bus.get_object(NM, device_path), NM_PROPERTIES_IFACE) |
|
||||||
active_ap_path = dev_props.Get(NM_WIRELESS_IFACE, "ActiveAccessPoint") |
|
||||||
|
|
||||||
aps: dict[str, list[AccessPoint]] = {} |
|
||||||
|
|
||||||
for ap_path in wifi_iface.GetAllAccessPoints(): |
|
||||||
ap_props = dbus.Interface(self._bus.get_object(NM, ap_path), NM_PROPERTIES_IFACE) |
|
||||||
|
|
||||||
try: |
|
||||||
ap = AccessPoint.from_dbus(ap_props, ap_path, active_ap_path) |
|
||||||
if ap.ssid == "": |
|
||||||
continue |
|
||||||
|
|
||||||
if ap.ssid not in aps: |
|
||||||
aps[ap.ssid] = [] |
|
||||||
|
|
||||||
aps[ap.ssid].append(ap) |
|
||||||
except dbus.exceptions.DBusException: |
|
||||||
# some APs have been seen dropping off during iteration |
|
||||||
cloudlog.exception(f"Failed to get AP properties for {ap_path}") |
|
||||||
|
|
||||||
known_connections = self._get_connections() |
|
||||||
self._networks = [Network.from_dbus(ssid, ap_list, active_ap_path, self._connection_by_ssid(ssid, known_connections) is not None) |
|
||||||
for ssid, ap_list in aps.items()] |
|
||||||
if self._networks_updated is not None: |
|
||||||
self._networks_updated(self._networks) |
|
||||||
|
|
||||||
def get_networks(self): |
|
||||||
return self._networks |
|
||||||
Loading…
Reference in new issue