Line data Source code
1 : /*
2 : * WPA Supplicant - background scan and roaming module: learn
3 : * Copyright (c) 2009-2010, Jouni Malinen <j@w1.fi>
4 : *
5 : * This software may be distributed under the terms of the BSD license.
6 : * See README for more details.
7 : */
8 :
9 : #include "includes.h"
10 :
11 : #include "common.h"
12 : #include "eloop.h"
13 : #include "list.h"
14 : #include "common/ieee802_11_defs.h"
15 : #include "drivers/driver.h"
16 : #include "config_ssid.h"
17 : #include "wpa_supplicant_i.h"
18 : #include "driver_i.h"
19 : #include "scan.h"
20 : #include "bgscan.h"
21 :
22 : struct bgscan_learn_bss {
23 : struct dl_list list;
24 : u8 bssid[ETH_ALEN];
25 : int freq;
26 : u8 *neigh; /* num_neigh * ETH_ALEN buffer */
27 : size_t num_neigh;
28 : };
29 :
30 : struct bgscan_learn_data {
31 : struct wpa_supplicant *wpa_s;
32 : const struct wpa_ssid *ssid;
33 : int scan_interval;
34 : int signal_threshold;
35 : int short_interval; /* use if signal < threshold */
36 : int long_interval; /* use if signal > threshold */
37 : struct os_reltime last_bgscan;
38 : char *fname;
39 : struct dl_list bss;
40 : int *supp_freqs;
41 : int probe_idx;
42 : };
43 :
44 :
45 10 : static void bss_free(struct bgscan_learn_bss *bss)
46 : {
47 10 : os_free(bss->neigh);
48 10 : os_free(bss);
49 10 : }
50 :
51 :
52 20 : static int bssid_in_array(u8 *array, size_t array_len, const u8 *bssid)
53 : {
54 : size_t i;
55 :
56 20 : if (array == NULL || array_len == 0)
57 10 : return 0;
58 :
59 10 : for (i = 0; i < array_len; i++) {
60 10 : if (os_memcmp(array + i * ETH_ALEN, bssid, ETH_ALEN) == 0)
61 10 : return 1;
62 : }
63 :
64 0 : return 0;
65 : }
66 :
67 :
68 38 : static void bgscan_learn_add_neighbor(struct bgscan_learn_bss *bss,
69 : const u8 *bssid)
70 : {
71 : u8 *n;
72 :
73 38 : if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
74 18 : return;
75 20 : if (bssid_in_array(bss->neigh, bss->num_neigh, bssid))
76 10 : return;
77 :
78 10 : n = os_realloc_array(bss->neigh, bss->num_neigh + 1, ETH_ALEN);
79 10 : if (n == NULL)
80 0 : return;
81 :
82 10 : os_memcpy(n + bss->num_neigh * ETH_ALEN, bssid, ETH_ALEN);
83 10 : bss->neigh = n;
84 10 : bss->num_neigh++;
85 : }
86 :
87 :
88 20 : static struct bgscan_learn_bss * bgscan_learn_get_bss(
89 : struct bgscan_learn_data *data, const u8 *bssid)
90 : {
91 : struct bgscan_learn_bss *bss;
92 :
93 30 : dl_list_for_each(bss, &data->bss, struct bgscan_learn_bss, list) {
94 22 : if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
95 12 : return bss;
96 : }
97 8 : return NULL;
98 : }
99 :
100 :
101 5 : static int bgscan_learn_load(struct bgscan_learn_data *data)
102 : {
103 : FILE *f;
104 : char buf[128];
105 : struct bgscan_learn_bss *bss;
106 :
107 5 : if (data->fname == NULL)
108 3 : return 0;
109 :
110 2 : f = fopen(data->fname, "r");
111 2 : if (f == NULL)
112 1 : return 0;
113 :
114 1 : wpa_printf(MSG_DEBUG, "bgscan learn: Loading data from %s",
115 : data->fname);
116 :
117 2 : if (fgets(buf, sizeof(buf), f) == NULL ||
118 1 : os_strncmp(buf, "wpa_supplicant-bgscan-learn\n", 28) != 0) {
119 0 : wpa_printf(MSG_INFO, "bgscan learn: Invalid data file %s",
120 : data->fname);
121 0 : fclose(f);
122 0 : return -1;
123 : }
124 :
125 6 : while (fgets(buf, sizeof(buf), f)) {
126 4 : if (os_strncmp(buf, "BSS ", 4) == 0) {
127 2 : bss = os_zalloc(sizeof(*bss));
128 2 : if (!bss)
129 0 : continue;
130 2 : if (hwaddr_aton(buf + 4, bss->bssid) < 0) {
131 0 : bss_free(bss);
132 0 : continue;
133 : }
134 2 : bss->freq = atoi(buf + 4 + 18);
135 2 : dl_list_add(&data->bss, &bss->list);
136 14 : wpa_printf(MSG_DEBUG, "bgscan learn: Loaded BSS "
137 : "entry: " MACSTR " freq=%d",
138 12 : MAC2STR(bss->bssid), bss->freq);
139 : }
140 :
141 4 : if (os_strncmp(buf, "NEIGHBOR ", 9) == 0) {
142 : u8 addr[ETH_ALEN];
143 :
144 2 : if (hwaddr_aton(buf + 9, addr) < 0)
145 0 : continue;
146 2 : bss = bgscan_learn_get_bss(data, addr);
147 2 : if (bss == NULL)
148 0 : continue;
149 2 : if (hwaddr_aton(buf + 9 + 18, addr) < 0)
150 0 : continue;
151 :
152 2 : bgscan_learn_add_neighbor(bss, addr);
153 : }
154 : }
155 :
156 1 : fclose(f);
157 1 : return 0;
158 : }
159 :
160 :
161 5 : static void bgscan_learn_save(struct bgscan_learn_data *data)
162 : {
163 : FILE *f;
164 : struct bgscan_learn_bss *bss;
165 :
166 5 : if (data->fname == NULL)
167 3 : return;
168 :
169 2 : wpa_printf(MSG_DEBUG, "bgscan learn: Saving data to %s",
170 : data->fname);
171 :
172 2 : f = fopen(data->fname, "w");
173 2 : if (f == NULL)
174 0 : return;
175 2 : fprintf(f, "wpa_supplicant-bgscan-learn\n");
176 :
177 6 : dl_list_for_each(bss, &data->bss, struct bgscan_learn_bss, list) {
178 28 : fprintf(f, "BSS " MACSTR " %d\n",
179 24 : MAC2STR(bss->bssid), bss->freq);
180 : }
181 :
182 6 : dl_list_for_each(bss, &data->bss, struct bgscan_learn_bss, list) {
183 : size_t i;
184 8 : for (i = 0; i < bss->num_neigh; i++) {
185 48 : fprintf(f, "NEIGHBOR " MACSTR " " MACSTR "\n",
186 24 : MAC2STR(bss->bssid),
187 24 : MAC2STR(bss->neigh + i * ETH_ALEN));
188 : }
189 : }
190 :
191 2 : fclose(f);
192 : }
193 :
194 :
195 264 : static int in_array(int *array, int val)
196 : {
197 : int i;
198 :
199 264 : if (array == NULL)
200 6 : return 0;
201 :
202 4229 : for (i = 0; array[i]; i++) {
203 4043 : if (array[i] == val)
204 72 : return 1;
205 : }
206 :
207 186 : return 0;
208 : }
209 :
210 :
211 1 : static int * bgscan_learn_get_freqs(struct bgscan_learn_data *data,
212 : size_t *count)
213 : {
214 : struct bgscan_learn_bss *bss;
215 1 : int *freqs = NULL, *n;
216 :
217 1 : *count = 0;
218 :
219 3 : dl_list_for_each(bss, &data->bss, struct bgscan_learn_bss, list) {
220 2 : if (in_array(freqs, bss->freq))
221 1 : continue;
222 1 : n = os_realloc_array(freqs, *count + 2, sizeof(int));
223 1 : if (n == NULL)
224 0 : return freqs;
225 1 : freqs = n;
226 1 : freqs[*count] = bss->freq;
227 1 : (*count)++;
228 1 : freqs[*count] = 0;
229 : }
230 :
231 1 : return freqs;
232 : }
233 :
234 :
235 1 : static int * bgscan_learn_get_probe_freq(struct bgscan_learn_data *data,
236 : int *freqs, size_t count)
237 : {
238 : int idx, *n;
239 :
240 1 : if (data->supp_freqs == NULL)
241 0 : return freqs;
242 :
243 1 : idx = data->probe_idx;
244 : do {
245 2 : if (!in_array(freqs, data->supp_freqs[idx])) {
246 1 : wpa_printf(MSG_DEBUG, "bgscan learn: Probe new freq "
247 1 : "%u", data->supp_freqs[idx]);
248 1 : data->probe_idx = idx + 1;
249 1 : if (data->supp_freqs[data->probe_idx] == 0)
250 0 : data->probe_idx = 0;
251 1 : n = os_realloc_array(freqs, count + 2, sizeof(int));
252 1 : if (n == NULL)
253 0 : return freqs;
254 1 : freqs = n;
255 1 : freqs[count] = data->supp_freqs[idx];
256 1 : count++;
257 1 : freqs[count] = 0;
258 1 : break;
259 : }
260 :
261 1 : idx++;
262 1 : if (data->supp_freqs[idx] == 0)
263 0 : idx = 0;
264 1 : } while (idx != data->probe_idx);
265 :
266 1 : return freqs;
267 : }
268 :
269 :
270 4 : static void bgscan_learn_timeout(void *eloop_ctx, void *timeout_ctx)
271 : {
272 4 : struct bgscan_learn_data *data = eloop_ctx;
273 4 : struct wpa_supplicant *wpa_s = data->wpa_s;
274 : struct wpa_driver_scan_params params;
275 4 : int *freqs = NULL;
276 : size_t count, i;
277 : char msg[100], *pos;
278 :
279 4 : os_memset(¶ms, 0, sizeof(params));
280 4 : params.num_ssids = 1;
281 4 : params.ssids[0].ssid = data->ssid->ssid;
282 4 : params.ssids[0].ssid_len = data->ssid->ssid_len;
283 4 : if (data->ssid->scan_freq)
284 3 : params.freqs = data->ssid->scan_freq;
285 : else {
286 1 : freqs = bgscan_learn_get_freqs(data, &count);
287 1 : wpa_printf(MSG_DEBUG, "bgscan learn: BSSes in this ESS have "
288 : "been seen on %u channels", (unsigned int) count);
289 1 : freqs = bgscan_learn_get_probe_freq(data, freqs, count);
290 :
291 1 : msg[0] = '\0';
292 1 : pos = msg;
293 3 : for (i = 0; freqs && freqs[i]; i++) {
294 : int ret;
295 2 : ret = os_snprintf(pos, msg + sizeof(msg) - pos, " %d",
296 2 : freqs[i]);
297 2 : if (os_snprintf_error(msg + sizeof(msg) - pos, ret))
298 0 : break;
299 2 : pos += ret;
300 : }
301 1 : pos[0] = '\0';
302 1 : wpa_printf(MSG_DEBUG, "bgscan learn: Scanning frequencies:%s",
303 : msg);
304 1 : params.freqs = freqs;
305 : }
306 :
307 4 : wpa_printf(MSG_DEBUG, "bgscan learn: Request a background scan");
308 4 : if (wpa_supplicant_trigger_scan(wpa_s, ¶ms)) {
309 0 : wpa_printf(MSG_DEBUG, "bgscan learn: Failed to trigger scan");
310 0 : eloop_register_timeout(data->scan_interval, 0,
311 : bgscan_learn_timeout, data, NULL);
312 : } else
313 4 : os_get_reltime(&data->last_bgscan);
314 4 : os_free(freqs);
315 4 : }
316 :
317 :
318 7 : static int bgscan_learn_get_params(struct bgscan_learn_data *data,
319 : const char *params)
320 : {
321 : const char *pos;
322 :
323 7 : if (params == NULL)
324 0 : return 0;
325 :
326 7 : data->short_interval = atoi(params);
327 :
328 7 : pos = os_strchr(params, ':');
329 7 : if (pos == NULL)
330 2 : return 0;
331 5 : pos++;
332 5 : data->signal_threshold = atoi(pos);
333 5 : pos = os_strchr(pos, ':');
334 5 : if (pos == NULL) {
335 2 : wpa_printf(MSG_ERROR, "bgscan learn: Missing scan interval "
336 : "for high signal");
337 2 : return -1;
338 : }
339 3 : pos++;
340 3 : data->long_interval = atoi(pos);
341 3 : pos = os_strchr(pos, ':');
342 3 : if (pos) {
343 2 : pos++;
344 2 : data->fname = os_strdup(pos);
345 : }
346 :
347 3 : return 0;
348 : }
349 :
350 :
351 5 : static int * bgscan_learn_get_supp_freqs(struct wpa_supplicant *wpa_s)
352 : {
353 : struct hostapd_hw_modes *modes;
354 5 : int i, j, *freqs = NULL, *n;
355 5 : size_t count = 0;
356 :
357 5 : modes = wpa_s->hw.modes;
358 5 : if (modes == NULL)
359 0 : return NULL;
360 :
361 20 : for (i = 0; i < wpa_s->hw.num_modes; i++) {
362 275 : for (j = 0; j < modes[i].num_channels; j++) {
363 260 : if (modes[i].channels[j].flag & HOSTAPD_CHAN_DISABLED)
364 0 : continue;
365 : /* some hw modes (e.g. 11b & 11g) contain same freqs */
366 260 : if (in_array(freqs, modes[i].channels[j].freq))
367 70 : continue;
368 190 : n = os_realloc_array(freqs, count + 2, sizeof(int));
369 190 : if (n == NULL)
370 0 : continue;
371 :
372 190 : freqs = n;
373 190 : freqs[count] = modes[i].channels[j].freq;
374 190 : count++;
375 190 : freqs[count] = 0;
376 : }
377 : }
378 :
379 5 : return freqs;
380 : }
381 :
382 :
383 7 : static void * bgscan_learn_init(struct wpa_supplicant *wpa_s,
384 : const char *params,
385 : const struct wpa_ssid *ssid)
386 : {
387 : struct bgscan_learn_data *data;
388 :
389 7 : data = os_zalloc(sizeof(*data));
390 7 : if (data == NULL)
391 0 : return NULL;
392 7 : dl_list_init(&data->bss);
393 7 : data->wpa_s = wpa_s;
394 7 : data->ssid = ssid;
395 7 : if (bgscan_learn_get_params(data, params) < 0) {
396 2 : os_free(data->fname);
397 2 : os_free(data);
398 2 : return NULL;
399 : }
400 5 : if (data->short_interval <= 0)
401 1 : data->short_interval = 30;
402 5 : if (data->long_interval <= 0)
403 2 : data->long_interval = 30;
404 :
405 5 : if (bgscan_learn_load(data) < 0) {
406 0 : os_free(data->fname);
407 0 : os_free(data);
408 0 : return NULL;
409 : }
410 :
411 5 : wpa_printf(MSG_DEBUG, "bgscan learn: Signal strength threshold %d "
412 : "Short bgscan interval %d Long bgscan interval %d",
413 : data->signal_threshold, data->short_interval,
414 : data->long_interval);
415 :
416 8 : if (data->signal_threshold &&
417 3 : wpa_drv_signal_monitor(wpa_s, data->signal_threshold, 4) < 0) {
418 0 : wpa_printf(MSG_ERROR, "bgscan learn: Failed to enable "
419 : "signal strength monitoring");
420 : }
421 :
422 5 : data->supp_freqs = bgscan_learn_get_supp_freqs(wpa_s);
423 5 : data->scan_interval = data->short_interval;
424 5 : if (data->signal_threshold) {
425 : /* Poll for signal info to set initial scan interval */
426 : struct wpa_signal_info siginfo;
427 6 : if (wpa_drv_signal_poll(wpa_s, &siginfo) == 0 &&
428 3 : siginfo.current_signal >= data->signal_threshold)
429 2 : data->scan_interval = data->long_interval;
430 : }
431 :
432 5 : eloop_register_timeout(data->scan_interval, 0, bgscan_learn_timeout,
433 : data, NULL);
434 :
435 : /*
436 : * This function is called immediately after an association, so it is
437 : * reasonable to assume that a scan was completed recently. This makes
438 : * us skip an immediate new scan in cases where the current signal
439 : * level is below the bgscan threshold.
440 : */
441 5 : os_get_reltime(&data->last_bgscan);
442 :
443 5 : return data;
444 : }
445 :
446 :
447 5 : static void bgscan_learn_deinit(void *priv)
448 : {
449 5 : struct bgscan_learn_data *data = priv;
450 : struct bgscan_learn_bss *bss, *n;
451 :
452 5 : bgscan_learn_save(data);
453 5 : eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
454 5 : if (data->signal_threshold)
455 3 : wpa_drv_signal_monitor(data->wpa_s, 0, 0);
456 5 : os_free(data->fname);
457 15 : dl_list_for_each_safe(bss, n, &data->bss, struct bgscan_learn_bss,
458 : list) {
459 10 : dl_list_del(&bss->list);
460 10 : bss_free(bss);
461 : }
462 5 : os_free(data->supp_freqs);
463 5 : os_free(data);
464 5 : }
465 :
466 :
467 36 : static int bgscan_learn_bss_match(struct bgscan_learn_data *data,
468 : struct wpa_scan_res *bss)
469 : {
470 : const u8 *ie;
471 :
472 36 : ie = wpa_scan_get_ie(bss, WLAN_EID_SSID);
473 36 : if (ie == NULL)
474 0 : return 0;
475 :
476 72 : if (data->ssid->ssid_len != ie[1] ||
477 36 : os_memcmp(data->ssid->ssid, ie + 2, ie[1]) != 0)
478 0 : return 0; /* SSID mismatch */
479 :
480 36 : return 1;
481 : }
482 :
483 :
484 9 : static int bgscan_learn_notify_scan(void *priv,
485 : struct wpa_scan_results *scan_res)
486 : {
487 9 : struct bgscan_learn_data *data = priv;
488 : size_t i, j;
489 : #define MAX_BSS 50
490 : u8 bssid[MAX_BSS * ETH_ALEN];
491 9 : size_t num_bssid = 0;
492 :
493 9 : wpa_printf(MSG_DEBUG, "bgscan learn: scan result notification");
494 :
495 9 : eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
496 9 : eloop_register_timeout(data->scan_interval, 0, bgscan_learn_timeout,
497 : data, NULL);
498 :
499 27 : for (i = 0; i < scan_res->num; i++) {
500 18 : struct wpa_scan_res *res = scan_res->res[i];
501 18 : if (!bgscan_learn_bss_match(data, res))
502 0 : continue;
503 :
504 18 : if (num_bssid < MAX_BSS) {
505 18 : os_memcpy(bssid + num_bssid * ETH_ALEN, res->bssid,
506 : ETH_ALEN);
507 18 : num_bssid++;
508 : }
509 : }
510 9 : wpa_printf(MSG_DEBUG, "bgscan learn: %u matching BSSes in scan "
511 : "results", (unsigned int) num_bssid);
512 :
513 27 : for (i = 0; i < scan_res->num; i++) {
514 18 : struct wpa_scan_res *res = scan_res->res[i];
515 : struct bgscan_learn_bss *bss;
516 :
517 18 : if (!bgscan_learn_bss_match(data, res))
518 0 : continue;
519 :
520 18 : bss = bgscan_learn_get_bss(data, res->bssid);
521 18 : if (bss && bss->freq != res->freq) {
522 0 : wpa_printf(MSG_DEBUG, "bgscan learn: Update BSS "
523 : MACSTR " freq %d -> %d",
524 0 : MAC2STR(res->bssid), bss->freq, res->freq);
525 0 : bss->freq = res->freq;
526 18 : } else if (!bss) {
527 56 : wpa_printf(MSG_DEBUG, "bgscan learn: Add BSS " MACSTR
528 48 : " freq=%d", MAC2STR(res->bssid), res->freq);
529 8 : bss = os_zalloc(sizeof(*bss));
530 8 : if (!bss)
531 0 : continue;
532 8 : os_memcpy(bss->bssid, res->bssid, ETH_ALEN);
533 8 : bss->freq = res->freq;
534 8 : dl_list_add(&data->bss, &bss->list);
535 : }
536 :
537 54 : for (j = 0; j < num_bssid; j++) {
538 36 : u8 *addr = bssid + j * ETH_ALEN;
539 36 : bgscan_learn_add_neighbor(bss, addr);
540 : }
541 : }
542 :
543 : /*
544 : * A more advanced bgscan could process scan results internally, select
545 : * the BSS and request roam if needed. This sample uses the existing
546 : * BSS/ESS selection routine. Change this to return 1 if selection is
547 : * done inside the bgscan module.
548 : */
549 :
550 9 : return 0;
551 : }
552 :
553 :
554 0 : static void bgscan_learn_notify_beacon_loss(void *priv)
555 : {
556 0 : wpa_printf(MSG_DEBUG, "bgscan learn: beacon loss");
557 : /* TODO: speed up background scanning */
558 0 : }
559 :
560 :
561 3 : static void bgscan_learn_notify_signal_change(void *priv, int above,
562 : int current_signal,
563 : int current_noise,
564 : int current_txrate)
565 : {
566 3 : struct bgscan_learn_data *data = priv;
567 3 : int scan = 0;
568 : struct os_reltime now;
569 :
570 6 : if (data->short_interval == data->long_interval ||
571 3 : data->signal_threshold == 0)
572 3 : return;
573 :
574 3 : wpa_printf(MSG_DEBUG, "bgscan learn: signal level changed "
575 : "(above=%d current_signal=%d current_noise=%d "
576 : "current_txrate=%d)", above, current_signal,
577 : current_noise, current_txrate);
578 3 : if (data->scan_interval == data->long_interval && !above) {
579 0 : wpa_printf(MSG_DEBUG, "bgscan learn: Start using short bgscan "
580 : "interval");
581 0 : data->scan_interval = data->short_interval;
582 0 : os_get_reltime(&now);
583 0 : if (now.sec > data->last_bgscan.sec + 1)
584 0 : scan = 1;
585 3 : } else if (data->scan_interval == data->short_interval && above) {
586 0 : wpa_printf(MSG_DEBUG, "bgscan learn: Start using long bgscan "
587 : "interval");
588 0 : data->scan_interval = data->long_interval;
589 0 : eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
590 0 : eloop_register_timeout(data->scan_interval, 0,
591 : bgscan_learn_timeout, data, NULL);
592 3 : } else if (!above) {
593 : /*
594 : * Signal dropped further 4 dB. Request a new scan if we have
595 : * not yet scanned in a while.
596 : */
597 1 : os_get_reltime(&now);
598 1 : if (now.sec > data->last_bgscan.sec + 10)
599 0 : scan = 1;
600 : }
601 :
602 3 : if (scan) {
603 0 : wpa_printf(MSG_DEBUG, "bgscan learn: Trigger immediate scan");
604 0 : eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
605 0 : eloop_register_timeout(0, 0, bgscan_learn_timeout, data, NULL);
606 : }
607 : }
608 :
609 :
610 : const struct bgscan_ops bgscan_learn_ops = {
611 : .name = "learn",
612 : .init = bgscan_learn_init,
613 : .deinit = bgscan_learn_deinit,
614 : .notify_scan = bgscan_learn_notify_scan,
615 : .notify_beacon_loss = bgscan_learn_notify_beacon_loss,
616 : .notify_signal_change = bgscan_learn_notify_signal_change,
617 : };
|
