From dccbdba24212dc53e36bf401b49a18b4a3063fd4 Mon Sep 17 00:00:00 2001 From: Carter Li Date: Fri, 2 Aug 2024 15:10:04 +0800 Subject: [PATCH] Wifi (Linux): remove libnm dependency --- src/detection/wifi/wifi_linux.c | 429 +++++++++++--------------------- 1 file changed, 140 insertions(+), 289 deletions(-) diff --git a/src/detection/wifi/wifi_linux.c b/src/detection/wifi/wifi_linux.c index f85bb264a..137b81816 100644 --- a/src/detection/wifi/wifi_linux.c +++ b/src/detection/wifi/wifi_linux.c @@ -1,217 +1,67 @@ #include "wifi.h" +#include "common/processing.h" +#include "common/properties.h" #include "util/stringUtils.h" #include #include -#ifdef FF_HAVE_LIBNM - -#include "common/processing.h" -#include "common/properties.h" -#include "common/library.h" - -#include -#define NM_NO_INCLUDE_EXTRA_HEADERS 1 -#include - -//https://developer-old.gnome.org/libnm/stable/libnm-nm-dbus-interface.html -#ifndef NM_VERSION_1_26 - #define NM_802_11_AP_SEC_KEY_MGMT_OWE 0x00000800 - #define NM_802_11_AP_SEC_KEY_MGMT_OWE_TM 0x00001000 -#endif - -static const char* detectWifiWithLibnm(FFlist* result) +static const char* detectWifiWithIw(FFWifiResult* item, FFstrbuf* buffer) { - FF_LIBRARY_LOAD(nm, &instance.config.library.libnm, "dlopen libnm failed", "libnm" FF_LIBRARY_EXTENSION, 0); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_client_new); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_client_get_devices); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_device_get_iface); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_device_get_state); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_device_wifi_get_type); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_device_wifi_get_active_access_point); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_ssid); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_utils_ssid_to_utf8); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_bssid); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_strength); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_max_bitrate); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_flags); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_wpa_flags); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, nm_access_point_get_rsn_flags); + const char* error = NULL; + FF_STRBUF_AUTO_DESTROY output = ffStrbufCreate(); + if((error = ffProcessAppendStdOut(&output, (char* const[]){ + "iw", + "dev", + item->inf.description.chars, + "link", + NULL + }))) + return error; - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, g_type_check_instance_is_a); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, g_bytes_get_size); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, g_bytes_get_data); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, g_free); - FF_LIBRARY_LOAD_SYMBOL_MESSAGE(nm, g_object_unref); + if(output.length == 0) + return "iw command execution failed"; - NMClient* client = ffnm_client_new(NULL, NULL); - if(!client) - return "Could not create NMClient"; - - /* Get all devices managed by NetworkManager */ - const GPtrArray* devices = ffnm_client_get_devices(client); - - /* Go through the array and process Wi-Fi devices */ - for (guint i = 0; i < devices->len; i++) + if(!ffParsePropLines(output.chars, "Connected to ", &item->conn.bssid)) { - NMDevice *device = g_ptr_array_index(devices, i); - if (!({ - GTypeInstance *inst = (GTypeInstance*) device; - GType type = ffnm_device_wifi_get_type(); - gboolean return_value; - if (!inst) - return_value = FALSE; - else if (inst->g_class && inst->g_class->g_type == type) - return_value = TRUE; - else - return_value = ffg_type_check_instance_is_a (inst, type); - return_value; - }) //NM_IS_DEVICE_WIFI(device) - ) - continue; - - FFWifiResult* item = (FFWifiResult*) ffListAdd(result); - ffStrbufInit(&item->inf.description); - ffStrbufInit(&item->inf.status); - ffStrbufInit(&item->conn.status); - ffStrbufInit(&item->conn.ssid); - ffStrbufInit(&item->conn.bssid); - ffStrbufInit(&item->conn.protocol); - ffStrbufInit(&item->conn.security); - item->conn.signalQuality = 0.0/0.0; - item->conn.rxRate = 0.0/0.0; - item->conn.txRate = 0.0/0.0; - - ffStrbufAppendS(&item->inf.description, ffnm_device_get_iface(device)); - NMDeviceState state = ffnm_device_get_state(device); - switch(state) - { - case NM_DEVICE_STATE_UNKNOWN: - ffStrbufSetStatic(&item->inf.status, "Unknown"); - break; - case NM_DEVICE_STATE_UNMANAGED: - ffStrbufSetStatic(&item->inf.status, "Unmanaged"); - break; - case NM_DEVICE_STATE_UNAVAILABLE: - ffStrbufSetStatic(&item->inf.status, "Unavailable"); - break; - case NM_DEVICE_STATE_DISCONNECTED: - ffStrbufSetStatic(&item->inf.status, "Disconnected"); - break; - case NM_DEVICE_STATE_PREPARE: - ffStrbufSetStatic(&item->inf.status, "Prepare"); - break; - case NM_DEVICE_STATE_CONFIG: - ffStrbufSetStatic(&item->inf.status, "Config"); - break; - case NM_DEVICE_STATE_NEED_AUTH: - ffStrbufSetStatic(&item->inf.status, "Need auth"); - break; - case NM_DEVICE_STATE_IP_CONFIG: - ffStrbufSetStatic(&item->inf.status, "IP config"); - break; - case NM_DEVICE_STATE_IP_CHECK: - ffStrbufSetStatic(&item->inf.status, "IP check"); - break; - case NM_DEVICE_STATE_SECONDARIES: - ffStrbufSetStatic(&item->inf.status, "Secondaries"); - break; - case NM_DEVICE_STATE_ACTIVATED: - ffStrbufSetStatic(&item->inf.status, "Activated"); - break; - case NM_DEVICE_STATE_DEACTIVATING: - ffStrbufSetStatic(&item->inf.status, "Deactivating"); - break; - case NM_DEVICE_STATE_FAILED: - ffStrbufSetStatic(&item->inf.status, "Failed"); - break; - } - if(state != NM_DEVICE_STATE_ACTIVATED) - continue; - - NMAccessPoint* ap = ffnm_device_wifi_get_active_access_point((NMDeviceWifi *) device); - if (!ap) - continue; - - ffStrbufSetStatic(&item->conn.status, "Connected"); - - GBytes* activeSsid = ffnm_access_point_get_ssid(ap); - if (activeSsid) - { - char* ssid = ffnm_utils_ssid_to_utf8(ffg_bytes_get_data(activeSsid, NULL), ffg_bytes_get_size(activeSsid)); - ffStrbufAppendS(&item->conn.ssid, ssid); - ffg_free(ssid); - } - - ffStrbufAppendS(&item->conn.bssid, ffnm_access_point_get_bssid(ap)); - item->conn.signalQuality = ffnm_access_point_get_strength(ap); - item->conn.rxRate = ffnm_access_point_get_max_bitrate(ap); - - FF_STRBUF_AUTO_DESTROY output = ffStrbufCreate(); - if(!ffProcessAppendStdOut(&output, (char* const[]){ - "iw", - "dev", - item->inf.description.chars, - "link", - NULL - })) - { - if(ffParsePropLines(output.chars, "rx bitrate: ", &item->conn.protocol)) - { - sscanf(item->conn.protocol.chars, "%lf", &item->conn.rxRate); - } - if(ffParsePropLines(output.chars, "tx bitrate: ", &item->conn.protocol)) - { - if(ffStrbufContainS(&item->conn.protocol, " HE-MCS ")) - ffStrbufSetStatic(&item->conn.protocol, "802.11ax (Wi-Fi 6)"); - else if(ffStrbufContainS(&item->conn.protocol, " VHT-MCS ")) - ffStrbufSetStatic(&item->conn.protocol, "802.11ac (Wi-Fi 5)"); - else if(ffStrbufContainS(&item->conn.protocol, " MCS ")) - ffStrbufSetStatic(&item->conn.protocol, "802.11n (Wi-Fi 4)"); - else - ffStrbufClear(&item->conn.protocol); - - sscanf(item->conn.protocol.chars, "%lf", &item->conn.txRate); - } - } - - NM80211ApFlags flags = ffnm_access_point_get_flags(ap); - NM80211ApSecurityFlags wpaFlags = ffnm_access_point_get_wpa_flags(ap); - NM80211ApSecurityFlags rsnFlags = ffnm_access_point_get_rsn_flags(ap); - - if ((flags & NM_802_11_AP_FLAGS_PRIVACY) && (wpaFlags == NM_802_11_AP_SEC_NONE) - && (rsnFlags == NM_802_11_AP_SEC_NONE)) - ffStrbufAppendS(&item->conn.security, "WEP/"); - if (wpaFlags != NM_802_11_AP_SEC_NONE) - ffStrbufAppendS(&item->conn.security, "WPA/"); - if ((rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_PSK) - || (rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_802_1X)) { - ffStrbufAppendS(&item->conn.security, "WPA2/"); - } - if (rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_SAE) { - ffStrbufAppendS(&item->conn.security, "WPA3/"); - } - if ((rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_OWE) - || (rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_OWE_TM)) { - ffStrbufAppendS(&item->conn.security, "OWE/"); - } - if ((wpaFlags & NM_802_11_AP_SEC_KEY_MGMT_802_1X) - || (rsnFlags & NM_802_11_AP_SEC_KEY_MGMT_802_1X)) { - ffStrbufAppendS(&item->conn.security, "802.1X/"); - } - if (!item->conn.security.length) - ffStrbufAppendS(&item->conn.security, "Insecure"); - else - ffStrbufTrimRight(&item->conn.security, '/'); + ffStrbufAppendS(&item->conn.status, "disconnected"); + return NULL; } - ffg_object_unref(client); + + ffStrbufAppendS(&item->conn.status, "connected"); + ffStrbufSubstrBeforeFirstC(&item->conn.bssid, ' '); + + ffParsePropLines(output.chars, "SSID: ", &item->conn.ssid); + + ffStrbufClear(buffer); + if(ffParsePropLines(output.chars, "signal: ", buffer)) + { + int level = (int) ffStrbufToSInt(buffer, INT_MAX); + if (level != INT_MAX) + item->conn.signalQuality = level >= -50 ? 100 : level <= -100 ? 0 : (level + 100) * 2; + } + + ffStrbufClear(buffer); + if(ffParsePropLines(output.chars, "rx bitrate: ", buffer)) + item->conn.rxRate = ffStrbufToDouble(buffer); + + ffStrbufClear(buffer); + if(ffParsePropLines(output.chars, "tx bitrate: ", buffer)) + { + item->conn.txRate = ffStrbufToDouble(buffer); + + if(ffStrbufContainS(buffer, " HE-MCS ")) + ffStrbufSetStatic(&item->conn.protocol, "802.11ax (Wi-Fi 6)"); + else if(ffStrbufContainS(buffer, " VHT-MCS ")) + ffStrbufSetStatic(&item->conn.protocol, "802.11ac (Wi-Fi 5)"); + else if(ffStrbufContainS(buffer, " MCS ")) + ffStrbufSetStatic(&item->conn.protocol, "802.11n (Wi-Fi 4)"); + } return NULL; } -#endif - #if FF_HAVE_LINUX_WIRELESS #define FF_DETECT_WIFI_WITH_IOCTLS @@ -221,20 +71,97 @@ static const char* detectWifiWithLibnm(FFlist* result) #include #include #include -#include //TODO: Don't depend on kernel headers +#include -static const char* detectWifiWithIoctls(FFlist* result) +static const char* detectWifiWithIoctls(FFWifiResult* item) +{ + FF_AUTO_CLOSE_FD int sock = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0); + if(sock < 0) + return "socket() failed"; + + struct iwreq iwr; + strncpy(iwr.ifr_name, item->inf.description.chars, IFNAMSIZ); + ffStrbufEnsureFree(&item->conn.ssid, IW_ESSID_MAX_SIZE); + iwr.u.essid.pointer = (caddr_t) item->conn.ssid.chars; + iwr.u.essid.length = IW_ESSID_MAX_SIZE + 1; + iwr.u.essid.flags = 0; + if(ioctl(sock, SIOCGIWESSID, &iwr) >= 0) + ffStrbufRecalculateLength(&item->conn.ssid); + + if(ioctl(sock, SIOCGIWNAME, &iwr) >= 0 && !ffStrEqualsIgnCase(iwr.u.name, "IEEE 802.11")) + { + if(ffStrStartsWithIgnCase(iwr.u.name, "IEEE ")) + ffStrbufSetS(&item->conn.protocol, iwr.u.name + strlen("IEEE ")); + else + ffStrbufSetS(&item->conn.protocol, iwr.u.name); + } + + if(ioctl(sock, SIOCGIWAP, &iwr) >= 0) + { + for(int i = 0; i < 6; ++i) + ffStrbufAppendF(&item->conn.bssid, "%.2X-", (uint8_t) iwr.u.ap_addr.sa_data[i]); + ffStrbufTrimRight(&item->conn.bssid, '-'); + } + + struct iw_statistics stats; + iwr.u.data.pointer = &stats; + iwr.u.data.length = sizeof(stats); + iwr.u.data.flags = 0; + + if(ioctl(sock, SIOCGIWSTATS, &iwr) >= 0) + { + int8_t level = (int8_t) stats.qual.level; // https://stackoverflow.com/questions/18079771/wireless-h-how-do-i-print-out-the-signal-level + item->conn.signalQuality = level >= -50 ? 100 : level <= -100 ? 0 : (level + 100) * 2; + } + + //FIXME: doesn't work + struct iw_encode_ext iwe; + iwr.u.data.pointer = &iwe; + iwr.u.data.length = sizeof(iwe); + iwr.u.data.flags = 0; + if(ioctl(sock, SIOCGIWENCODEEXT, &iwr) >= 0) + { + switch(iwe.alg) + { + case IW_ENCODE_ALG_WEP: + ffStrbufAppendS(&item->conn.security, "WEP"); + break; + case IW_ENCODE_ALG_TKIP: + ffStrbufAppendS(&item->conn.security, "TKIP"); + break; + case IW_ENCODE_ALG_CCMP: + ffStrbufAppendS(&item->conn.security, "CCMP"); + break; + case IW_ENCODE_ALG_PMK: + ffStrbufAppendS(&item->conn.security, "PMK"); + break; + case IW_ENCODE_ALG_AES_CMAC: + ffStrbufAppendS(&item->conn.security, "CMAC"); + break; + default: + ffStrbufAppendF(&item->conn.security, "Unknown (%d)", (int) iwe.alg); + break; + } + } + + return NULL; +} + +#endif + +const char* ffDetectWifi(FF_MAYBE_UNUSED FFlist* result) { struct if_nameindex* infs = if_nameindex(); if(!infs) return "if_nameindex() failed"; - FF_STRBUF_AUTO_DESTROY path = ffStrbufCreate(); + FF_STRBUF_AUTO_DESTROY buffer = ffStrbufCreate(); + const char* error = NULL; for(struct if_nameindex* i = infs; !(i->if_index == 0 && i->if_name == NULL); ++i) { - ffStrbufSetF(&path, "/sys/class/net/%s/phy80211", i->if_name); - if(!ffPathExists(path.chars, FF_PATHTYPE_DIRECTORY)) + ffStrbufSetF(&buffer, "/sys/class/net/%s/phy80211", i->if_name); + if(!ffPathExists(buffer.chars, FF_PATHTYPE_DIRECTORY)) continue; FFWifiResult* item = (FFWifiResult*)ffListAdd(result); @@ -249,102 +176,26 @@ static const char* detectWifiWithIoctls(FFlist* result) item->conn.rxRate = 0.0/0.0; item->conn.txRate = 0.0/0.0; - ffStrbufSetF(&path, "/sys/class/net/%s/operstate", i->if_name); - if (!ffAppendFileBuffer(path.chars, &item->inf.status)) + ffStrbufSetF(&buffer, "/sys/class/net/%s/operstate", i->if_name); + if (!ffAppendFileBuffer(buffer.chars, &item->inf.status)) continue; ffStrbufTrimRightSpace(&item->inf.status); if (!ffStrbufEqualS(&item->inf.status, "up")) continue; - FF_AUTO_CLOSE_FD int sock = socket(AF_INET, SOCK_DGRAM | SOCK_CLOEXEC, 0); - if(sock < 0) + if (detectWifiWithIw(item, &buffer) == NULL) continue; - struct iwreq iwr; - strncpy(iwr.ifr_name, i->if_name, IFNAMSIZ); - ffStrbufEnsureFree(&item->conn.ssid, IW_ESSID_MAX_SIZE); - iwr.u.essid.pointer = (caddr_t) item->conn.ssid.chars; - iwr.u.essid.length = IW_ESSID_MAX_SIZE + 1; - iwr.u.essid.flags = 0; - if(ioctl(sock, SIOCGIWESSID, &iwr) >= 0) - ffStrbufRecalculateLength(&item->conn.ssid); + #ifdef FF_DETECT_WIFI_WITH_IOCTLS + detectWifiWithIoctls(item); + continue; + #endif - if(ioctl(sock, SIOCGIWNAME, &iwr) >= 0 && !ffStrEqualsIgnCase(iwr.u.name, "IEEE 802.11")) - { - if(ffStrStartsWithIgnCase(iwr.u.name, "IEEE ")) - ffStrbufSetS(&item->conn.protocol, iwr.u.name + strlen("IEEE ")); - else - ffStrbufSetS(&item->conn.protocol, iwr.u.name); - } - - if(ioctl(sock, SIOCGIWAP, &iwr) >= 0) - { - for(int i = 0; i < 6; ++i) - ffStrbufAppendF(&item->conn.bssid, "%.2X-", (uint8_t) iwr.u.ap_addr.sa_data[i]); - ffStrbufTrimRight(&item->conn.bssid, '-'); - } - - struct iw_statistics stats; - iwr.u.data.pointer = &stats; - iwr.u.data.length = sizeof(stats); - iwr.u.data.flags = 0; - - if(ioctl(sock, SIOCGIWSTATS, &iwr) >= 0) - { - int8_t level = (int8_t) stats.qual.level; // https://stackoverflow.com/questions/18079771/wireless-h-how-do-i-print-out-the-signal-level - item->conn.signalQuality = level >= -50 ? 100 : level <= -100 ? 0 : (level + 100) * 2; - } - - //FIXME: doesn't work - struct iw_encode_ext iwe; - iwr.u.data.pointer = &iwe; - iwr.u.data.length = sizeof(iwe); - iwr.u.data.flags = 0; - if(ioctl(sock, SIOCGIWENCODEEXT, &iwr) >= 0) - { - struct iw_encode_ext* iwe = iwr.u.data.pointer; - switch(iwe->alg) - { - case IW_ENCODE_ALG_WEP: - ffStrbufAppendS(&item->conn.security, "WEP"); - break; - case IW_ENCODE_ALG_TKIP: - ffStrbufAppendS(&item->conn.security, "TKIP"); - break; - case IW_ENCODE_ALG_CCMP: - ffStrbufAppendS(&item->conn.security, "CCMP"); - break; - case IW_ENCODE_ALG_PMK: - ffStrbufAppendS(&item->conn.security, "PMK"); - break; - case IW_ENCODE_ALG_AES_CMAC: - ffStrbufAppendS(&item->conn.security, "CMAC"); - break; - default: - ffStrbufAppendF(&item->conn.security, "Unknown (%d)", (int) iwe->alg); - break; - } - } + error = "`iw` failed and `linux/wireless.h` not found during compilation"; + break; } if_freenameindex(infs); - return NULL; -} - -#endif - -const char* ffDetectWifi(FF_MAYBE_UNUSED FFlist* result) -{ - #ifdef FF_HAVE_LIBNM - detectWifiWithLibnm(result); - if(result->length) // NetworkManager not enabled? #811 - return NULL; - #endif - - #ifdef FF_DETECT_WIFI_WITH_IOCTLS - return detectWifiWithIoctls(result); - #endif - - return "linux/wireless.h not found during compilation"; + return error; }