1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
/* 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 <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);
size_t len = essid_str ? strnlen(essid_str, DLADM_STRSIZE) : 0;
ap->setSSID(essid_str, len);
}
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;
}
|
