/* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ #include "nsWifiMonitor.h" #include "nsWifiAccessPoint.h" #include "nsServiceManagerUtils.h" #include "nsComponentManagerUtils.h" #include "nsIMutableArray.h" #include "plstr.h" #include <glib.h> #define DLADM_STRSIZE 256 #define DLADM_SECTIONS 3 using namespace mozilla; struct val_strength_t { const char *strength_name; int signal_value; }; static val_strength_t strength_vals[] = { { "very weak", -112 }, { "weak", -88 }, { "good", -68 }, { "very good", -40 }, { "excellent", -16 } }; static nsWifiAccessPoint * do_parse_str(char *bssid_str, char *essid_str, char *strength) { unsigned char mac_as_int[6] = { 0 }; sscanf(bssid_str, "%x:%x:%x:%x:%x:%x", &mac_as_int[0], &mac_as_int[1], &mac_as_int[2], &mac_as_int[3], &mac_as_int[4], &mac_as_int[5]); int signal = 0; uint32_t strength_vals_count = sizeof(strength_vals) / sizeof (val_strength_t); for (uint32_t i = 0; i < strength_vals_count; i++) { if (!strncasecmp(strength, strength_vals[i].strength_name, DLADM_STRSIZE)) { signal = strength_vals[i].signal_value; break; } } nsWifiAccessPoint *ap = new nsWifiAccessPoint(); if (ap) { ap->setMac(mac_as_int); ap->setSignal(signal); ap->setSSID(essid_str, PL_strnlen(essid_str, DLADM_STRSIZE)); } return ap; } static void do_dladm(nsCOMArray<nsWifiAccessPoint> &accessPoints) { GError *err = nullptr; char *sout = nullptr; char *serr = nullptr; int exit_status = 0; char * dladm_args[] = { "/usr/bin/pfexec", "/usr/sbin/dladm", "scan-wifi", "-p", "-o", "BSSID,ESSID,STRENGTH" }; gboolean rv = g_spawn_sync("/", dladm_args, nullptr, (GSpawnFlags)0, nullptr, nullptr, &sout, &serr, &exit_status, &err); if (rv && !exit_status) { char wlan[DLADM_SECTIONS][DLADM_STRSIZE+1]; uint32_t section = 0; uint32_t sout_scan = 0; uint32_t wlan_put = 0; bool escape = false; nsWifiAccessPoint* ap; char sout_char; do { sout_char = sout[sout_scan++]; if (escape) { escape = false; if (sout_char != '\0') { wlan[section][wlan_put++] = sout_char; continue; } } if (sout_char =='\\') { escape = true; continue; } if (sout_char == ':') { wlan[section][wlan_put] = '\0'; section++; wlan_put = 0; continue; } if ((sout_char == '\0') || (sout_char == '\n')) { wlan[section][wlan_put] = '\0'; if (section == DLADM_SECTIONS - 1) { ap = do_parse_str(wlan[0], wlan[1], wlan[2]); if (ap) { accessPoints.AppendObject(ap); } } section = 0; wlan_put = 0; continue; } wlan[section][wlan_put++] = sout_char; } while ((wlan_put <= DLADM_STRSIZE) && (section < DLADM_SECTIONS) && (sout_char != '\0')); } g_free(sout); g_free(serr); } nsresult nsWifiMonitor::DoScan() { // Regularly get the access point data. nsCOMArray<nsWifiAccessPoint> lastAccessPoints; nsCOMArray<nsWifiAccessPoint> accessPoints; while (mKeepGoing) { accessPoints.Clear(); do_dladm(accessPoints); bool accessPointsChanged = !AccessPointsEqual(accessPoints, lastAccessPoints); ReplaceArray(lastAccessPoints, accessPoints); nsresult rv = CallWifiListeners(lastAccessPoints, accessPointsChanged); NS_ENSURE_SUCCESS(rv, rv); LOG(("waiting on monitor\n")); ReentrantMonitorAutoEnter mon(mReentrantMonitor); mon.Wait(PR_SecondsToInterval(kDefaultWifiScanInterval)); } return NS_OK; }