1 /*
2 * WPA Supplicant - Scanning
3 * Copyright (c) 2003-2010, Jouni Malinen <j@w1.fi>
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation.
8 *
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
11 *
12 * See README and COPYING for more details.
13 */
14
15 #include "utils/includes.h"
16
17 #include "utils/common.h"
18 #include "utils/eloop.h"
19 #include "common/ieee802_11_defs.h"
20 #include "config.h"
21 #include "wpa_supplicant_i.h"
22 #include "driver_i.h"
23 #include "mlme.h"
24 #include "wps_supplicant.h"
25 #include "p2p_supplicant.h"
26 #include "p2p/p2p.h"
27 #include "notify.h"
28 #include "bss.h"
29 #include "scan.h"
30
31
wpa_supplicant_gen_assoc_event(struct wpa_supplicant * wpa_s)32 static void wpa_supplicant_gen_assoc_event(struct wpa_supplicant *wpa_s)
33 {
34 struct wpa_ssid *ssid;
35 union wpa_event_data data;
36
37 ssid = wpa_supplicant_get_ssid(wpa_s);
38 if (ssid == NULL)
39 return;
40
41 if (wpa_s->current_ssid == NULL) {
42 wpa_s->current_ssid = ssid;
43 if (wpa_s->current_ssid != NULL)
44 wpas_notify_network_changed(wpa_s);
45 }
46 wpa_supplicant_initiate_eapol(wpa_s);
47 wpa_dbg(wpa_s, MSG_DEBUG, "Already associated with a configured "
48 "network - generating associated event");
49 os_memset(&data, 0, sizeof(data));
50 wpa_supplicant_event(wpa_s, EVENT_ASSOC, &data);
51 }
52
53
54 #ifdef CONFIG_WPS
wpas_wps_in_use(struct wpa_config * conf,enum wps_request_type * req_type)55 static int wpas_wps_in_use(struct wpa_config *conf,
56 enum wps_request_type *req_type)
57 {
58 struct wpa_ssid *ssid;
59 int wps = 0;
60
61 for (ssid = conf->ssid; ssid; ssid = ssid->next) {
62 if (!(ssid->key_mgmt & WPA_KEY_MGMT_WPS))
63 continue;
64
65 wps = 1;
66 *req_type = wpas_wps_get_req_type(ssid);
67 if (!ssid->eap.phase1)
68 continue;
69
70 if (os_strstr(ssid->eap.phase1, "pbc=1"))
71 return 2;
72 }
73
74 return wps;
75 }
76 #endif /* CONFIG_WPS */
77
78
wpa_supplicant_enabled_networks(struct wpa_config * conf)79 int wpa_supplicant_enabled_networks(struct wpa_config *conf)
80 {
81 struct wpa_ssid *ssid = conf->ssid;
82 int count = 0;
83 while (ssid) {
84 if (!ssid->disabled)
85 count++;
86 ssid = ssid->next;
87 }
88 return count;
89 }
90
91
wpa_supplicant_assoc_try(struct wpa_supplicant * wpa_s,struct wpa_ssid * ssid)92 static void wpa_supplicant_assoc_try(struct wpa_supplicant *wpa_s,
93 struct wpa_ssid *ssid)
94 {
95 while (ssid) {
96 if (!ssid->disabled)
97 break;
98 ssid = ssid->next;
99 }
100
101 /* ap_scan=2 mode - try to associate with each SSID. */
102 if (ssid == NULL) {
103 wpa_dbg(wpa_s, MSG_DEBUG, "wpa_supplicant_assoc_try: Reached "
104 "end of scan list - go back to beginning");
105 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
106 wpa_supplicant_req_scan(wpa_s, 0, 0);
107 return;
108 }
109 if (ssid->next) {
110 /* Continue from the next SSID on the next attempt. */
111 wpa_s->prev_scan_ssid = ssid;
112 } else {
113 /* Start from the beginning of the SSID list. */
114 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
115 }
116 wpa_supplicant_associate(wpa_s, NULL, ssid);
117 }
118
119
int_array_len(const int * a)120 static int int_array_len(const int *a)
121 {
122 int i;
123 for (i = 0; a && a[i]; i++)
124 ;
125 return i;
126 }
127
128
int_array_concat(int ** res,const int * a)129 static void int_array_concat(int **res, const int *a)
130 {
131 int reslen, alen, i;
132 int *n;
133
134 reslen = int_array_len(*res);
135 alen = int_array_len(a);
136
137 n = os_realloc(*res, (reslen + alen + 1) * sizeof(int));
138 if (n == NULL) {
139 os_free(*res);
140 *res = NULL;
141 return;
142 }
143 for (i = 0; i <= alen; i++)
144 n[reslen + i] = a[i];
145 *res = n;
146 }
147
148
freq_cmp(const void * a,const void * b)149 static int freq_cmp(const void *a, const void *b)
150 {
151 int _a = *(int *) a;
152 int _b = *(int *) b;
153
154 if (_a == 0)
155 return 1;
156 if (_b == 0)
157 return -1;
158 return _a - _b;
159 }
160
161
int_array_sort_unique(int * a)162 static void int_array_sort_unique(int *a)
163 {
164 int alen;
165 int i, j;
166
167 if (a == NULL)
168 return;
169
170 alen = int_array_len(a);
171 qsort(a, alen, sizeof(int), freq_cmp);
172
173 i = 0;
174 j = 1;
175 while (a[i] && a[j]) {
176 if (a[i] == a[j]) {
177 j++;
178 continue;
179 }
180 a[++i] = a[j++];
181 }
182 if (a[i])
183 i++;
184 a[i] = 0;
185 }
186
187
wpa_supplicant_trigger_scan(struct wpa_supplicant * wpa_s,struct wpa_driver_scan_params * params)188 int wpa_supplicant_trigger_scan(struct wpa_supplicant *wpa_s,
189 struct wpa_driver_scan_params *params)
190 {
191 int ret;
192
193 wpa_supplicant_notify_scanning(wpa_s, 1);
194
195 if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
196 ret = ieee80211_sta_req_scan(wpa_s, params);
197 else
198 ret = wpa_drv_scan(wpa_s, params);
199
200 if (ret) {
201 wpa_supplicant_notify_scanning(wpa_s, 0);
202 wpas_notify_scan_done(wpa_s, 0);
203 } else
204 wpa_s->scan_runs++;
205
206 return ret;
207 }
208
209
210 static struct wpa_driver_scan_filter *
wpa_supplicant_build_filter_ssids(struct wpa_config * conf,size_t * num_ssids)211 wpa_supplicant_build_filter_ssids(struct wpa_config *conf, size_t *num_ssids)
212 {
213 struct wpa_driver_scan_filter *ssids;
214 struct wpa_ssid *ssid;
215 size_t count;
216
217 *num_ssids = 0;
218 if (!conf->filter_ssids)
219 return NULL;
220
221 for (count = 0, ssid = conf->ssid; ssid; ssid = ssid->next) {
222 if (ssid->ssid && ssid->ssid_len)
223 count++;
224 }
225 if (count == 0)
226 return NULL;
227 ssids = os_zalloc(count * sizeof(struct wpa_driver_scan_filter));
228 if (ssids == NULL)
229 return NULL;
230
231 for (ssid = conf->ssid; ssid; ssid = ssid->next) {
232 if (!ssid->ssid || !ssid->ssid_len)
233 continue;
234 os_memcpy(ssids[*num_ssids].ssid, ssid->ssid, ssid->ssid_len);
235 ssids[*num_ssids].ssid_len = ssid->ssid_len;
236 (*num_ssids)++;
237 }
238
239 return ssids;
240 }
241
242
wpa_supplicant_scan(void * eloop_ctx,void * timeout_ctx)243 static void wpa_supplicant_scan(void *eloop_ctx, void *timeout_ctx)
244 {
245 struct wpa_supplicant *wpa_s = eloop_ctx;
246 struct wpa_ssid *ssid;
247 int scan_req = 0, ret;
248 struct wpabuf *wps_ie = NULL;
249 #ifdef CONFIG_WPS
250 int wps = 0;
251 enum wps_request_type req_type = WPS_REQ_ENROLLEE_INFO;
252 #endif /* CONFIG_WPS */
253 struct wpa_driver_scan_params params;
254 size_t max_ssids;
255 enum wpa_states prev_state;
256
257 if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED) {
258 wpa_dbg(wpa_s, MSG_DEBUG, "Skip scan - interface disabled");
259 return;
260 }
261
262 if (wpa_s->disconnected && !wpa_s->scan_req) {
263 wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
264 return;
265 }
266
267 if (!wpa_supplicant_enabled_networks(wpa_s->conf) &&
268 !wpa_s->scan_req) {
269 wpa_dbg(wpa_s, MSG_DEBUG, "No enabled networks - do not scan");
270 wpa_supplicant_set_state(wpa_s, WPA_INACTIVE);
271 return;
272 }
273
274 if (wpa_s->conf->ap_scan != 0 &&
275 (wpa_s->drv_flags & WPA_DRIVER_FLAGS_WIRED)) {
276 wpa_dbg(wpa_s, MSG_DEBUG, "Using wired authentication - "
277 "overriding ap_scan configuration");
278 wpa_s->conf->ap_scan = 0;
279 wpas_notify_ap_scan_changed(wpa_s);
280 }
281
282 if (wpa_s->conf->ap_scan == 0) {
283 wpa_supplicant_gen_assoc_event(wpa_s);
284 return;
285 }
286
287 if ((wpa_s->drv_flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME) ||
288 wpa_s->conf->ap_scan == 2)
289 max_ssids = 1;
290 else {
291 max_ssids = wpa_s->max_scan_ssids;
292 if (max_ssids > WPAS_MAX_SCAN_SSIDS)
293 max_ssids = WPAS_MAX_SCAN_SSIDS;
294 }
295
296 #ifdef CONFIG_WPS
297 wps = wpas_wps_in_use(wpa_s->conf, &req_type);
298 #endif /* CONFIG_WPS */
299
300 scan_req = wpa_s->scan_req;
301 wpa_s->scan_req = 0;
302
303 os_memset(¶ms, 0, sizeof(params));
304
305 prev_state = wpa_s->wpa_state;
306 if (wpa_s->wpa_state == WPA_DISCONNECTED ||
307 wpa_s->wpa_state == WPA_INACTIVE)
308 wpa_supplicant_set_state(wpa_s, WPA_SCANNING);
309
310 if (scan_req != 2 && wpa_s->connect_without_scan) {
311 for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
312 if (ssid == wpa_s->connect_without_scan)
313 break;
314 }
315 wpa_s->connect_without_scan = NULL;
316 if (ssid) {
317 wpa_printf(MSG_DEBUG, "Start a pre-selected network "
318 "without scan step");
319 wpa_supplicant_associate(wpa_s, NULL, ssid);
320 return;
321 }
322 }
323
324 /* Find the starting point from which to continue scanning */
325 ssid = wpa_s->conf->ssid;
326 if (wpa_s->prev_scan_ssid != WILDCARD_SSID_SCAN) {
327 while (ssid) {
328 if (ssid == wpa_s->prev_scan_ssid) {
329 ssid = ssid->next;
330 break;
331 }
332 ssid = ssid->next;
333 }
334 }
335
336 if (scan_req != 2 && wpa_s->conf->ap_scan == 2) {
337 wpa_s->connect_without_scan = NULL;
338 wpa_supplicant_assoc_try(wpa_s, ssid);
339 return;
340 #ifndef ANDROID
341 } else if (wpa_s->conf->ap_scan == 2) {
342 /*
343 * User-initiated scan request in ap_scan == 2; scan with
344 * wildcard SSID.
345 */
346 ssid = NULL;
347 #endif
348 } else {
349 struct wpa_ssid *start = ssid, *tssid;
350 int freqs_set = 0;
351 if (ssid == NULL && max_ssids > 1)
352 ssid = wpa_s->conf->ssid;
353 while (ssid) {
354 if (!ssid->disabled && ssid->scan_ssid) {
355 wpa_hexdump_ascii(MSG_DEBUG, "Scan SSID",
356 ssid->ssid, ssid->ssid_len);
357 params.ssids[params.num_ssids].ssid =
358 ssid->ssid;
359 params.ssids[params.num_ssids].ssid_len =
360 ssid->ssid_len;
361 params.num_ssids++;
362 if (params.num_ssids + 1 >= max_ssids)
363 break;
364 }
365 ssid = ssid->next;
366 if (ssid == start)
367 break;
368 if (ssid == NULL && max_ssids > 1 &&
369 start != wpa_s->conf->ssid)
370 ssid = wpa_s->conf->ssid;
371 }
372
373 for (tssid = wpa_s->conf->ssid; tssid; tssid = tssid->next) {
374 if (tssid->disabled)
375 continue;
376 if ((params.freqs || !freqs_set) && tssid->scan_freq) {
377 int_array_concat(¶ms.freqs,
378 tssid->scan_freq);
379 } else {
380 os_free(params.freqs);
381 params.freqs = NULL;
382 }
383 freqs_set = 1;
384 }
385 int_array_sort_unique(params.freqs);
386 }
387
388 if (ssid) {
389 wpa_s->prev_scan_ssid = ssid;
390 if (max_ssids > 1) {
391 wpa_dbg(wpa_s, MSG_DEBUG, "Include wildcard SSID in "
392 "the scan request");
393 params.num_ssids++;
394 }
395 wpa_dbg(wpa_s, MSG_DEBUG, "Starting AP scan for specific "
396 "SSID(s)");
397 } else {
398 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
399 params.num_ssids++;
400 wpa_dbg(wpa_s, MSG_DEBUG, "Starting AP scan for wildcard "
401 "SSID");
402 }
403
404 #ifdef CONFIG_P2P
405 wpa_s->wps->dev.p2p = 1;
406 if (!wps) {
407 wps = 1;
408 req_type = WPS_REQ_ENROLLEE_INFO;
409 }
410
411 if (params.freqs == NULL && wpa_s->p2p_in_provisioning &&
412 wpa_s->go_params) {
413 /* Optimize provisioning state scan based on GO information */
414 if (wpa_s->p2p_in_provisioning < 5 &&
415 wpa_s->go_params->freq > 0) {
416 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Scan only GO "
417 "preferred frequency %d MHz",
418 wpa_s->go_params->freq);
419 params.freqs = os_zalloc(2 * sizeof(int));
420 if (params.freqs)
421 params.freqs[0] = wpa_s->go_params->freq;
422 } else if (wpa_s->p2p_in_provisioning < 8 &&
423 wpa_s->go_params->freq_list[0]) {
424 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Scan only common "
425 "channels");
426 int_array_concat(¶ms.freqs,
427 wpa_s->go_params->freq_list);
428 if (params.freqs)
429 int_array_sort_unique(params.freqs);
430 }
431 wpa_s->p2p_in_provisioning++;
432 }
433 #endif /* CONFIG_P2P */
434
435 #ifdef CONFIG_WPS
436 if (params.freqs == NULL && wpa_s->after_wps && wpa_s->wps_freq) {
437 /*
438 * Optimize post-provisioning scan based on channel used
439 * during provisioning.
440 */
441 wpa_dbg(wpa_s, MSG_DEBUG, "WPS: Scan only frequency %u MHz "
442 "that was used during provisioning", wpa_s->wps_freq);
443 params.freqs = os_zalloc(2 * sizeof(int));
444 if (params.freqs)
445 params.freqs[0] = wpa_s->wps_freq;
446 wpa_s->after_wps--;
447 }
448
449 if (wps) {
450 wps_ie = wps_build_probe_req_ie(wps == 2, &wpa_s->wps->dev,
451 wpa_s->wps->uuid, req_type,
452 0, NULL);
453 if (wps_ie) {
454 params.extra_ies = wpabuf_head(wps_ie);
455 params.extra_ies_len = wpabuf_len(wps_ie);
456 }
457 }
458 #endif /* CONFIG_WPS */
459
460 #ifdef CONFIG_P2P
461 if (wps_ie) {
462 if (wpabuf_resize(&wps_ie, 100) == 0) {
463 wpas_p2p_scan_ie(wpa_s, wps_ie);
464 params.extra_ies = wpabuf_head(wps_ie);
465 params.extra_ies_len = wpabuf_len(wps_ie);
466 }
467 }
468 #endif /* CONFIG_P2P */
469
470 if (params.freqs == NULL && wpa_s->next_scan_freqs) {
471 wpa_dbg(wpa_s, MSG_DEBUG, "Optimize scan based on previously "
472 "generated frequency list");
473 params.freqs = wpa_s->next_scan_freqs;
474 } else
475 os_free(wpa_s->next_scan_freqs);
476 wpa_s->next_scan_freqs = NULL;
477
478 params.filter_ssids = wpa_supplicant_build_filter_ssids(
479 wpa_s->conf, ¶ms.num_filter_ssids);
480
481 ret = wpa_supplicant_trigger_scan(wpa_s, ¶ms);
482
483 wpabuf_free(wps_ie);
484 os_free(params.freqs);
485 os_free(params.filter_ssids);
486
487 if (ret) {
488 wpa_msg(wpa_s, MSG_WARNING, "Failed to initiate AP scan");
489 if (prev_state != wpa_s->wpa_state)
490 wpa_supplicant_set_state(wpa_s, prev_state);
491 wpa_supplicant_req_scan(wpa_s, 1, 0);
492 }
493 }
494
495
496 /**
497 * wpa_supplicant_req_scan - Schedule a scan for neighboring access points
498 * @wpa_s: Pointer to wpa_supplicant data
499 * @sec: Number of seconds after which to scan
500 * @usec: Number of microseconds after which to scan
501 *
502 * This function is used to schedule a scan for neighboring access points after
503 * the specified time.
504 */
wpa_supplicant_req_scan(struct wpa_supplicant * wpa_s,int sec,int usec)505 void wpa_supplicant_req_scan(struct wpa_supplicant *wpa_s, int sec, int usec)
506 {
507 /* If there's at least one network that should be specifically scanned
508 * then don't cancel the scan and reschedule. Some drivers do
509 * background scanning which generates frequent scan results, and that
510 * causes the specific SSID scan to get continually pushed back and
511 * never happen, which causes hidden APs to never get probe-scanned.
512 */
513 if (eloop_is_timeout_registered(wpa_supplicant_scan, wpa_s, NULL) &&
514 wpa_s->conf->ap_scan == 1) {
515 struct wpa_ssid *ssid = wpa_s->conf->ssid;
516
517 while (ssid) {
518 if (!ssid->disabled && ssid->scan_ssid)
519 break;
520 ssid = ssid->next;
521 }
522 if (ssid) {
523 wpa_dbg(wpa_s, MSG_DEBUG, "Not rescheduling scan to "
524 "ensure that specific SSID scans occur");
525 return;
526 }
527 }
528
529 wpa_dbg(wpa_s, MSG_DEBUG, "Setting scan request: %d sec %d usec",
530 sec, usec);
531 eloop_cancel_timeout(wpa_supplicant_scan, wpa_s, NULL);
532 eloop_register_timeout(sec, usec, wpa_supplicant_scan, wpa_s, NULL);
533 }
534
535
536 /**
537 * wpa_supplicant_cancel_scan - Cancel a scheduled scan request
538 * @wpa_s: Pointer to wpa_supplicant data
539 *
540 * This function is used to cancel a scan request scheduled with
541 * wpa_supplicant_req_scan().
542 */
wpa_supplicant_cancel_scan(struct wpa_supplicant * wpa_s)543 void wpa_supplicant_cancel_scan(struct wpa_supplicant *wpa_s)
544 {
545 wpa_dbg(wpa_s, MSG_DEBUG, "Cancelling scan request");
546 eloop_cancel_timeout(wpa_supplicant_scan, wpa_s, NULL);
547 wpa_supplicant_notify_scanning(wpa_s, 0);
548 }
549
550
wpa_supplicant_notify_scanning(struct wpa_supplicant * wpa_s,int scanning)551 void wpa_supplicant_notify_scanning(struct wpa_supplicant *wpa_s,
552 int scanning)
553 {
554 if (wpa_s->scanning != scanning) {
555 wpa_s->scanning = scanning;
556 wpas_notify_scanning(wpa_s);
557 }
558 }
559
560
wpa_scan_get_max_rate(const struct wpa_scan_res * res)561 static int wpa_scan_get_max_rate(const struct wpa_scan_res *res)
562 {
563 int rate = 0;
564 const u8 *ie;
565 int i;
566
567 ie = wpa_scan_get_ie(res, WLAN_EID_SUPP_RATES);
568 for (i = 0; ie && i < ie[1]; i++) {
569 if ((ie[i + 2] & 0x7f) > rate)
570 rate = ie[i + 2] & 0x7f;
571 }
572
573 ie = wpa_scan_get_ie(res, WLAN_EID_EXT_SUPP_RATES);
574 for (i = 0; ie && i < ie[1]; i++) {
575 if ((ie[i + 2] & 0x7f) > rate)
576 rate = ie[i + 2] & 0x7f;
577 }
578
579 return rate;
580 }
581
582
wpa_scan_get_ie(const struct wpa_scan_res * res,u8 ie)583 const u8 * wpa_scan_get_ie(const struct wpa_scan_res *res, u8 ie)
584 {
585 const u8 *end, *pos;
586
587 pos = (const u8 *) (res + 1);
588 end = pos + res->ie_len;
589
590 while (pos + 1 < end) {
591 if (pos + 2 + pos[1] > end)
592 break;
593 if (pos[0] == ie)
594 return pos;
595 pos += 2 + pos[1];
596 }
597
598 return NULL;
599 }
600
601
wpa_scan_get_vendor_ie(const struct wpa_scan_res * res,u32 vendor_type)602 const u8 * wpa_scan_get_vendor_ie(const struct wpa_scan_res *res,
603 u32 vendor_type)
604 {
605 const u8 *end, *pos;
606
607 pos = (const u8 *) (res + 1);
608 end = pos + res->ie_len;
609
610 while (pos + 1 < end) {
611 if (pos + 2 + pos[1] > end)
612 break;
613 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
614 vendor_type == WPA_GET_BE32(&pos[2]))
615 return pos;
616 pos += 2 + pos[1];
617 }
618
619 return NULL;
620 }
621
622
wpa_scan_get_vendor_ie_multi(const struct wpa_scan_res * res,u32 vendor_type)623 struct wpabuf * wpa_scan_get_vendor_ie_multi(const struct wpa_scan_res *res,
624 u32 vendor_type)
625 {
626 struct wpabuf *buf;
627 const u8 *end, *pos;
628
629 buf = wpabuf_alloc(res->ie_len);
630 if (buf == NULL)
631 return NULL;
632
633 pos = (const u8 *) (res + 1);
634 end = pos + res->ie_len;
635
636 while (pos + 1 < end) {
637 if (pos + 2 + pos[1] > end)
638 break;
639 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
640 vendor_type == WPA_GET_BE32(&pos[2]))
641 wpabuf_put_data(buf, pos + 2 + 4, pos[1] - 4);
642 pos += 2 + pos[1];
643 }
644
645 if (wpabuf_len(buf) == 0) {
646 wpabuf_free(buf);
647 buf = NULL;
648 }
649
650 return buf;
651 }
652
653
wpa_scan_get_vendor_ie_multi_beacon(const struct wpa_scan_res * res,u32 vendor_type)654 struct wpabuf * wpa_scan_get_vendor_ie_multi_beacon(
655 const struct wpa_scan_res *res, u32 vendor_type)
656 {
657 struct wpabuf *buf;
658 const u8 *end, *pos;
659
660 if (res->beacon_ie_len == 0)
661 return NULL;
662 buf = wpabuf_alloc(res->beacon_ie_len);
663 if (buf == NULL)
664 return NULL;
665
666 pos = (const u8 *) (res + 1);
667 pos += res->ie_len;
668 end = pos + res->beacon_ie_len;
669
670 while (pos + 1 < end) {
671 if (pos + 2 + pos[1] > end)
672 break;
673 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
674 vendor_type == WPA_GET_BE32(&pos[2]))
675 wpabuf_put_data(buf, pos + 2 + 4, pos[1] - 4);
676 pos += 2 + pos[1];
677 }
678
679 if (wpabuf_len(buf) == 0) {
680 wpabuf_free(buf);
681 buf = NULL;
682 }
683
684 return buf;
685 }
686
687
688 /* Compare function for sorting scan results. Return >0 if @b is considered
689 * better. */
wpa_scan_result_compar(const void * a,const void * b)690 static int wpa_scan_result_compar(const void *a, const void *b)
691 {
692 struct wpa_scan_res **_wa = (void *) a;
693 struct wpa_scan_res **_wb = (void *) b;
694 struct wpa_scan_res *wa = *_wa;
695 struct wpa_scan_res *wb = *_wb;
696 int wpa_a, wpa_b, maxrate_a, maxrate_b;
697
698 /* WPA/WPA2 support preferred */
699 wpa_a = wpa_scan_get_vendor_ie(wa, WPA_IE_VENDOR_TYPE) != NULL ||
700 wpa_scan_get_ie(wa, WLAN_EID_RSN) != NULL;
701 wpa_b = wpa_scan_get_vendor_ie(wb, WPA_IE_VENDOR_TYPE) != NULL ||
702 wpa_scan_get_ie(wb, WLAN_EID_RSN) != NULL;
703
704 if (wpa_b && !wpa_a)
705 return 1;
706 if (!wpa_b && wpa_a)
707 return -1;
708
709 /* privacy support preferred */
710 if ((wa->caps & IEEE80211_CAP_PRIVACY) == 0 &&
711 (wb->caps & IEEE80211_CAP_PRIVACY))
712 return 1;
713 if ((wa->caps & IEEE80211_CAP_PRIVACY) &&
714 (wb->caps & IEEE80211_CAP_PRIVACY) == 0)
715 return -1;
716
717 /* best/max rate preferred if signal level close enough XXX */
718 if ((wa->level && wb->level && abs(wb->level - wa->level) < 5) ||
719 (wa->qual && wb->qual && abs(wb->qual - wa->qual) < 10)) {
720 maxrate_a = wpa_scan_get_max_rate(wa);
721 maxrate_b = wpa_scan_get_max_rate(wb);
722 if (maxrate_a != maxrate_b)
723 return maxrate_b - maxrate_a;
724 }
725
726 /* use freq for channel preference */
727
728 /* all things being equal, use signal level; if signal levels are
729 * identical, use quality values since some drivers may only report
730 * that value and leave the signal level zero */
731 if (wb->level == wa->level)
732 return wb->qual - wa->qual;
733 return wb->level - wa->level;
734 }
735
736
737 #ifdef CONFIG_WPS
738 /* Compare function for sorting scan results when searching a WPS AP for
739 * provisioning. Return >0 if @b is considered better. */
wpa_scan_result_wps_compar(const void * a,const void * b)740 static int wpa_scan_result_wps_compar(const void *a, const void *b)
741 {
742 struct wpa_scan_res **_wa = (void *) a;
743 struct wpa_scan_res **_wb = (void *) b;
744 struct wpa_scan_res *wa = *_wa;
745 struct wpa_scan_res *wb = *_wb;
746 int uses_wps_a, uses_wps_b;
747 struct wpabuf *wps_a, *wps_b;
748 int res;
749
750 /* Optimization - check WPS IE existence before allocated memory and
751 * doing full reassembly. */
752 uses_wps_a = wpa_scan_get_vendor_ie(wa, WPS_IE_VENDOR_TYPE) != NULL;
753 uses_wps_b = wpa_scan_get_vendor_ie(wb, WPS_IE_VENDOR_TYPE) != NULL;
754 if (uses_wps_a && !uses_wps_b)
755 return -1;
756 if (!uses_wps_a && uses_wps_b)
757 return 1;
758
759 if (uses_wps_a && uses_wps_b) {
760 wps_a = wpa_scan_get_vendor_ie_multi(wa, WPS_IE_VENDOR_TYPE);
761 wps_b = wpa_scan_get_vendor_ie_multi(wb, WPS_IE_VENDOR_TYPE);
762 res = wps_ap_priority_compar(wps_a, wps_b);
763 wpabuf_free(wps_a);
764 wpabuf_free(wps_b);
765 if (res)
766 return res;
767 }
768
769 /*
770 * Do not use current AP security policy as a sorting criteria during
771 * WPS provisioning step since the AP may get reconfigured at the
772 * completion of provisioning.
773 */
774
775 /* all things being equal, use signal level; if signal levels are
776 * identical, use quality values since some drivers may only report
777 * that value and leave the signal level zero */
778 if (wb->level == wa->level)
779 return wb->qual - wa->qual;
780 return wb->level - wa->level;
781 }
782 #endif /* CONFIG_WPS */
783
784
785 /**
786 * wpa_supplicant_get_scan_results - Get scan results
787 * @wpa_s: Pointer to wpa_supplicant data
788 * @info: Information about what was scanned or %NULL if not available
789 * @new_scan: Whether a new scan was performed
790 * Returns: Scan results, %NULL on failure
791 *
792 * This function request the current scan results from the driver and updates
793 * the local BSS list wpa_s->bss. The caller is responsible for freeing the
794 * results with wpa_scan_results_free().
795 */
796 struct wpa_scan_results *
wpa_supplicant_get_scan_results(struct wpa_supplicant * wpa_s,struct scan_info * info,int new_scan)797 wpa_supplicant_get_scan_results(struct wpa_supplicant *wpa_s,
798 struct scan_info *info, int new_scan)
799 {
800 struct wpa_scan_results *scan_res;
801 size_t i;
802 int (*compar)(const void *, const void *) = wpa_scan_result_compar;
803
804 if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
805 scan_res = ieee80211_sta_get_scan_results(wpa_s);
806 else
807 scan_res = wpa_drv_get_scan_results2(wpa_s);
808 if (scan_res == NULL) {
809 wpa_dbg(wpa_s, MSG_DEBUG, "Failed to get scan results");
810 return NULL;
811 }
812
813 #ifdef CONFIG_WPS
814 if (wpas_wps_in_progress(wpa_s)) {
815 wpa_dbg(wpa_s, MSG_DEBUG, "WPS: Order scan results with WPS "
816 "provisioning rules");
817 compar = wpa_scan_result_wps_compar;
818 }
819 #endif /* CONFIG_WPS */
820
821 qsort(scan_res->res, scan_res->num, sizeof(struct wpa_scan_res *),
822 compar);
823
824 wpa_bss_update_start(wpa_s);
825 for (i = 0; i < scan_res->num; i++)
826 wpa_bss_update_scan_res(wpa_s, scan_res->res[i]);
827 wpa_bss_update_end(wpa_s, info, new_scan);
828
829 return scan_res;
830 }
831
832
wpa_supplicant_update_scan_results(struct wpa_supplicant * wpa_s)833 int wpa_supplicant_update_scan_results(struct wpa_supplicant *wpa_s)
834 {
835 struct wpa_scan_results *scan_res;
836 scan_res = wpa_supplicant_get_scan_results(wpa_s, NULL, 0);
837 if (scan_res == NULL)
838 return -1;
839 wpa_scan_results_free(scan_res);
840
841 return 0;
842 }
843
844
wpa_scan_results_free(struct wpa_scan_results * res)845 void wpa_scan_results_free(struct wpa_scan_results *res)
846 {
847 size_t i;
848
849 if (res == NULL)
850 return;
851
852 for (i = 0; i < res->num; i++)
853 os_free(res->res[i]);
854 os_free(res->res);
855 os_free(res);
856 }
857