wpa_supplicant/scan.c

   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 "wps_supplicant.h"
  24 #include "p2p_supplicant.h"
  25 #include "p2p/p2p.h"
  26 #include "notify.h"
  27 #include "bss.h"
  28 #include "scan.h"
  29
  30
  31 static void wpa_supplicant_gen_assoc_event(struct wpa_supplicant *wpa_s)
  32 {
  33         struct wpa_ssid *ssid;
  34         union wpa_event_data data;
  35
  36         ssid = wpa_supplicant_get_ssid(wpa_s);
  37         if (ssid == NULL)
  38                 return;
  39
  40         if (wpa_s->current_ssid == NULL) {
  41                 wpa_s->current_ssid = ssid;
  42                 if (wpa_s->current_ssid != NULL)
  43                         wpas_notify_network_changed(wpa_s);
  44         }
  45         wpa_supplicant_initiate_eapol(wpa_s);
  46         wpa_dbg(wpa_s, MSG_DEBUG, "Already associated with a configured "
  47                 "network - generating associated event");
  48         os_memset(&data, 0, sizeof(data));
  49         wpa_supplicant_event(wpa_s, EVENT_ASSOC, &data);
  50 }
  51
  52
  53 #ifdef CONFIG_WPS
  54 static int wpas_wps_in_use(struct wpa_supplicant *wpa_s,
  55                            enum wps_request_type *req_type)
  56 {
  57         struct wpa_ssid *ssid;
  58         int wps = 0;
  59
  60         for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
  61                 if (!(ssid->key_mgmt & WPA_KEY_MGMT_WPS))
  62                         continue;
  63
  64                 wps = 1;
  65                 *req_type = wpas_wps_get_req_type(ssid);
  66                 if (!ssid->eap.phase1)
  67                         continue;
  68
  69                 if (os_strstr(ssid->eap.phase1, "pbc=1"))
  70                         return 2;
  71         }
  72
  73 #ifdef CONFIG_P2P
  74         wpa_s->wps->dev.p2p = 1;
  75         if (!wps) {
  76                 wps = 1;
  77                 *req_type = WPS_REQ_ENROLLEE_INFO;
  78         }
  79 #endif /* CONFIG_P2P */
  80
  81         return wps;
  82 }
  83 #endif /* CONFIG_WPS */
  84
  85
  86 int wpa_supplicant_enabled_networks(struct wpa_config *conf)
  87 {
  88         struct wpa_ssid *ssid = conf->ssid;
  89         int count = 0;
  90         while (ssid) {
  91                 if (!ssid->disabled)
  92                         count++;
  93                 ssid = ssid->next;
  94         }
  95         return count;
  96 }
  97
  98
  99 static void wpa_supplicant_assoc_try(struct wpa_supplicant *wpa_s,
 100                                      struct wpa_ssid *ssid)
 101 {
 102         while (ssid) {
 103                 if (!ssid->disabled)
 104                         break;
 105                 ssid = ssid->next;
 106         }
 107
 108         /* ap_scan=2 mode - try to associate with each SSID. */
 109         if (ssid == NULL) {
 110                 wpa_dbg(wpa_s, MSG_DEBUG, "wpa_supplicant_assoc_try: Reached "
 111                         "end of scan list - go back to beginning");
 112                 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
 113                 wpa_supplicant_req_scan(wpa_s, 0, 0);
 114                 return;
 115         }
 116         if (ssid->next) {
 117                 /* Continue from the next SSID on the next attempt. */
 118                 wpa_s->prev_scan_ssid = ssid;
 119         } else {
 120                 /* Start from the beginning of the SSID list. */
 121                 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
 122         }
 123         wpa_supplicant_associate(wpa_s, NULL, ssid);
 124 }
 125
 126
 127 static int int_array_len(const int *a)
 128 {
 129         int i;
 130         for (i = 0; a && a[i]; i++)
 131                 ;
 132         return i;
 133 }
 134
 135
 136 static void int_array_concat(int **res, const int *a)
 137 {
 138         int reslen, alen, i;
 139         int *n;
 140
 141         reslen = int_array_len(*res);
 142         alen = int_array_len(a);
 143
 144         n = os_realloc(*res, (reslen + alen + 1) * sizeof(int));
 145         if (n == NULL) {
 146                 os_free(*res);
 147                 *res = NULL;
 148                 return;
 149         }
 150         for (i = 0; i <= alen; i++)
 151                 n[reslen + i] = a[i];
 152         *res = n;
 153 }
 154
 155
 156 static int freq_cmp(const void *a, const void *b)
 157 {
 158         int _a = *(int *) a;
 159         int _b = *(int *) b;
 160
 161         if (_a == 0)
 162                 return 1;
 163         if (_b == 0)
 164                 return -1;
 165         return _a - _b;
 166 }
 167
 168
 169 static void int_array_sort_unique(int *a)
 170 {
 171         int alen;
 172         int i, j;
 173
 174         if (a == NULL)
 175                 return;
 176
 177         alen = int_array_len(a);
 178         qsort(a, alen, sizeof(int), freq_cmp);
 179
 180         i = 0;
 181         j = 1;
 182         while (a[i] && a[j]) {
 183                 if (a[i] == a[j]) {
 184                         j++;
 185                         continue;
 186                 }
 187                 a[++i] = a[j++];
 188         }
 189         if (a[i])
 190                 i++;
 191         a[i] = 0;
 192 }
 193
 194
 195 int wpa_supplicant_trigger_scan(struct wpa_supplicant *wpa_s,
 196                                 struct wpa_driver_scan_params *params)
 197 {
 198         int ret;
 199
 200         wpa_supplicant_notify_scanning(wpa_s, 1);
 201
 202         ret = wpa_drv_scan(wpa_s, params);
 203         if (ret) {
 204                 wpa_supplicant_notify_scanning(wpa_s, 0);
 205                 wpas_notify_scan_done(wpa_s, 0);
 206         } else
 207                 wpa_s->scan_runs++;
 208
 209         return ret;
 210 }
 211
 212
 213 static void
 214 wpa_supplicant_delayed_sched_scan_timeout(void *eloop_ctx, void *timeout_ctx)
 215 {
 216         struct wpa_supplicant *wpa_s = eloop_ctx;
 217
 218         wpa_dbg(wpa_s, MSG_DEBUG, "Starting delayed sched scan");
 219
 220         if (wpa_supplicant_req_sched_scan(wpa_s))
 221                 wpa_supplicant_req_scan(wpa_s, 0, 0);
 222 }
 223
 224
 225 static void
 226 wpa_supplicant_sched_scan_timeout(void *eloop_ctx, void *timeout_ctx)
 227 {
 228         struct wpa_supplicant *wpa_s = eloop_ctx;
 229
 230         wpa_dbg(wpa_s, MSG_DEBUG, "Sched scan timeout - stopping it");
 231
 232         wpa_s->sched_scan_timed_out = 1;
 233         wpa_supplicant_cancel_sched_scan(wpa_s);
 234 }
 235
 236
 237 static int
 238 wpa_supplicant_start_sched_scan(struct wpa_supplicant *wpa_s,
 239                                 struct wpa_driver_scan_params *params,
 240                                 int interval)
 241 {
 242         int ret;
 243
 244         wpa_supplicant_notify_scanning(wpa_s, 1);
 245         ret = wpa_drv_sched_scan(wpa_s, params, interval * 1000);
 246         if (ret)
 247                 wpa_supplicant_notify_scanning(wpa_s, 0);
 248         else
 249                 wpa_s->sched_scanning = 1;
 250
 251         return ret;
 252 }
 253
 254
 255 static int wpa_supplicant_stop_sched_scan(struct wpa_supplicant *wpa_s)
 256 {
 257         int ret;
 258
 259         ret = wpa_drv_stop_sched_scan(wpa_s);
 260         if (ret) {
 261                 wpa_dbg(wpa_s, MSG_DEBUG, "stopping sched_scan failed!");
 262                 /* TODO: what to do if stopping fails? */
 263                 return -1;
 264         }
 265
 266         return ret;
 267 }
 268
 269
 270 static struct wpa_driver_scan_filter *
 271 wpa_supplicant_build_filter_ssids(struct wpa_config *conf, size_t *num_ssids)
 272 {
 273         struct wpa_driver_scan_filter *ssids;
 274         struct wpa_ssid *ssid;
 275         size_t count;
 276
 277         *num_ssids = 0;
 278         if (!conf->filter_ssids)
 279                 return NULL;
 280
 281         for (count = 0, ssid = conf->ssid; ssid; ssid = ssid->next) {
 282                 if (ssid->ssid && ssid->ssid_len)
 283                         count++;
 284         }
 285         if (count == 0)
 286                 return NULL;
 287         ssids = os_zalloc(count * sizeof(struct wpa_driver_scan_filter));
 288         if (ssids == NULL)
 289                 return NULL;
 290
 291         for (ssid = conf->ssid; ssid; ssid = ssid->next) {
 292                 if (!ssid->ssid || !ssid->ssid_len)
 293                         continue;
 294                 os_memcpy(ssids[*num_ssids].ssid, ssid->ssid, ssid->ssid_len);
 295                 ssids[*num_ssids].ssid_len = ssid->ssid_len;
 296                 (*num_ssids)++;
 297         }
 298
 299         return ssids;
 300 }
 301
 302
 303 static void wpa_supplicant_optimize_freqs(
 304         struct wpa_supplicant *wpa_s, struct wpa_driver_scan_params *params)
 305 {
 306 #ifdef CONFIG_P2P
 307         if (params->freqs == NULL && wpa_s->p2p_in_provisioning &&
 308             wpa_s->go_params) {
 309                 /* Optimize provisioning state scan based on GO information */
 310                 if (wpa_s->p2p_in_provisioning < 5 &&
 311                     wpa_s->go_params->freq > 0) {
 312                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Scan only GO "
 313                                 "preferred frequency %d MHz",
 314                                 wpa_s->go_params->freq);
 315                         params->freqs = os_zalloc(2 * sizeof(int));
 316                         if (params->freqs)
 317                                 params->freqs[0] = wpa_s->go_params->freq;
 318                 } else if (wpa_s->p2p_in_provisioning < 8 &&
 319                            wpa_s->go_params->freq_list[0]) {
 320                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Scan only common "
 321                                 "channels");
 322                         int_array_concat(&params->freqs,
 323                                          wpa_s->go_params->freq_list);
 324                         if (params->freqs)
 325                                 int_array_sort_unique(params->freqs);
 326                 }
 327                 wpa_s->p2p_in_provisioning++;
 328         }
 329 #endif /* CONFIG_P2P */
 330
 331 #ifdef CONFIG_WPS
 332         if (params->freqs == NULL && wpa_s->after_wps && wpa_s->wps_freq) {
 333                 /*
 334                  * Optimize post-provisioning scan based on channel used
 335                  * during provisioning.
 336                  */
 337                 wpa_dbg(wpa_s, MSG_DEBUG, "WPS: Scan only frequency %u MHz "
 338                         "that was used during provisioning", wpa_s->wps_freq);
 339                 params->freqs = os_zalloc(2 * sizeof(int));
 340                 if (params->freqs)
 341                         params->freqs[0] = wpa_s->wps_freq;
 342                 wpa_s->after_wps--;
 343         }
 344
 345 #endif /* CONFIG_WPS */
 346 }
 347
 348
 349 #ifdef CONFIG_INTERWORKING
 350 static void wpas_add_interworking_elements(struct wpa_supplicant *wpa_s,
 351                                            struct wpabuf *buf)
 352 {
 353         if (wpa_s->conf->interworking == 0)
 354                 return;
 355
 356         wpabuf_put_u8(buf, WLAN_EID_EXT_CAPAB);
 357         wpabuf_put_u8(buf, 4);
 358         wpabuf_put_u8(buf, 0x00);
 359         wpabuf_put_u8(buf, 0x00);
 360         wpabuf_put_u8(buf, 0x00);
 361         wpabuf_put_u8(buf, 0x80); /* Bit 31 - Interworking */
 362
 363         wpabuf_put_u8(buf, WLAN_EID_INTERWORKING);
 364         wpabuf_put_u8(buf, is_zero_ether_addr(wpa_s->conf->hessid) ? 1 :
 365                       1 + ETH_ALEN);
 366         wpabuf_put_u8(buf, wpa_s->conf->access_network_type);
 367         /* No Venue Info */
 368         if (!is_zero_ether_addr(wpa_s->conf->hessid))
 369                 wpabuf_put_data(buf, wpa_s->conf->hessid, ETH_ALEN);
 370 }
 371 #endif /* CONFIG_INTERWORKING */
 372
 373
 374 static struct wpabuf *
 375 wpa_supplicant_extra_ies(struct wpa_supplicant *wpa_s,
 376                          struct wpa_driver_scan_params *params)
 377 {
 378         struct wpabuf *extra_ie = NULL;
 379 #ifdef CONFIG_WPS
 380         int wps = 0;
 381         enum wps_request_type req_type = WPS_REQ_ENROLLEE_INFO;
 382 #endif /* CONFIG_WPS */
 383
 384 #ifdef CONFIG_INTERWORKING
 385         if (wpa_s->conf->interworking &&
 386             wpabuf_resize(&extra_ie, 100) == 0)
 387                 wpas_add_interworking_elements(wpa_s, extra_ie);
 388 #endif /* CONFIG_INTERWORKING */
 389
 390 #ifdef CONFIG_WPS
 391         wps = wpas_wps_in_use(wpa_s, &req_type);
 392
 393         if (wps) {
 394                 struct wpabuf *wps_ie;
 395                 wps_ie = wps_build_probe_req_ie(wps == 2, &wpa_s->wps->dev,
 396                                                 wpa_s->wps->uuid, req_type,
 397                                                 0, NULL);
 398                 if (wps_ie) {
 399                         if (wpabuf_resize(&extra_ie, wpabuf_len(wps_ie)) == 0)
 400                                 wpabuf_put_buf(extra_ie, wps_ie);
 401                         wpabuf_free(wps_ie);
 402                 }
 403         }
 404
 405 #ifdef CONFIG_P2P
 406         if (wps) {
 407                 size_t ielen = p2p_scan_ie_buf_len(wpa_s->global->p2p);
 408                 if (wpabuf_resize(&extra_ie, ielen) == 0)
 409                         wpas_p2p_scan_ie(wpa_s, extra_ie);
 410         }
 411 #endif /* CONFIG_P2P */
 412
 413 #endif /* CONFIG_WPS */
 414
 415         return extra_ie;
 416 }
 417
 418
 419 static void wpa_supplicant_scan(void *eloop_ctx, void *timeout_ctx)
 420 {
 421         struct wpa_supplicant *wpa_s = eloop_ctx;
 422         struct wpa_ssid *ssid;
 423         int scan_req = 0, ret;
 424         struct wpabuf *extra_ie;
 425         struct wpa_driver_scan_params params;
 426         size_t max_ssids;
 427         enum wpa_states prev_state;
 428
 429         if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED) {
 430                 wpa_dbg(wpa_s, MSG_DEBUG, "Skip scan - interface disabled");
 431                 return;
 432         }
 433
 434         if (wpa_s->disconnected && !wpa_s->scan_req) {
 435                 wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
 436                 return;
 437         }
 438
 439         if (!wpa_supplicant_enabled_networks(wpa_s->conf) &&
 440             !wpa_s->scan_req) {
 441                 wpa_dbg(wpa_s, MSG_DEBUG, "No enabled networks - do not scan");
 442                 wpa_supplicant_set_state(wpa_s, WPA_INACTIVE);
 443                 return;
 444         }
 445
 446         if (wpa_s->conf->ap_scan != 0 &&
 447             (wpa_s->drv_flags & WPA_DRIVER_FLAGS_WIRED)) {
 448                 wpa_dbg(wpa_s, MSG_DEBUG, "Using wired authentication - "
 449                         "overriding ap_scan configuration");
 450                 wpa_s->conf->ap_scan = 0;
 451                 wpas_notify_ap_scan_changed(wpa_s);
 452         }
 453
 454         if (wpa_s->conf->ap_scan == 0) {
 455                 wpa_supplicant_gen_assoc_event(wpa_s);
 456                 return;
 457         }
 458
 459 #ifdef CONFIG_P2P
 460         if (wpas_p2p_in_progress(wpa_s)) {
 461                 if (wpa_s->wpa_state == WPA_SCANNING) {
 462                         wpa_dbg(wpa_s, MSG_DEBUG, "Delay station mode scan "
 463                                 "while P2P operation is in progress");
 464                         wpa_supplicant_req_scan(wpa_s, 5, 0);
 465                 } else {
 466                         wpa_dbg(wpa_s, MSG_DEBUG, "Do not request scan while "
 467                                 "P2P operation is in progress");
 468                 }
 469                 return;
 470         }
 471 #endif /* CONFIG_P2P */
 472
 473         if (wpa_s->conf->ap_scan == 2)
 474                 max_ssids = 1;
 475         else {
 476                 max_ssids = wpa_s->max_scan_ssids;
 477                 if (max_ssids > WPAS_MAX_SCAN_SSIDS)
 478                         max_ssids = WPAS_MAX_SCAN_SSIDS;
 479         }
 480
 481         scan_req = wpa_s->scan_req;
 482         wpa_s->scan_req = 0;
 483
 484         os_memset(&params, 0, sizeof(params));
 485
 486         prev_state = wpa_s->wpa_state;
 487         if (wpa_s->wpa_state == WPA_DISCONNECTED ||
 488             wpa_s->wpa_state == WPA_INACTIVE)
 489                 wpa_supplicant_set_state(wpa_s, WPA_SCANNING);
 490
 491         if (scan_req != 2 && wpa_s->connect_without_scan) {
 492                 for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
 493                         if (ssid == wpa_s->connect_without_scan)
 494                                 break;
 495                 }
 496                 wpa_s->connect_without_scan = NULL;
 497                 if (ssid) {
 498                         wpa_printf(MSG_DEBUG, "Start a pre-selected network "
 499                                    "without scan step");
 500                         wpa_supplicant_associate(wpa_s, NULL, ssid);
 501                         return;
 502                 }
 503         }
 504
 505         /* Find the starting point from which to continue scanning */
 506         ssid = wpa_s->conf->ssid;
 507         if (wpa_s->prev_scan_ssid != WILDCARD_SSID_SCAN) {
 508                 while (ssid) {
 509                         if (ssid == wpa_s->prev_scan_ssid) {
 510                                 ssid = ssid->next;
 511                                 break;
 512                         }
 513                         ssid = ssid->next;
 514                 }
 515         }
 516
 517         if (scan_req != 2 && wpa_s->conf->ap_scan == 2) {
 518                 wpa_s->connect_without_scan = NULL;
 519                 wpa_supplicant_assoc_try(wpa_s, ssid);
 520                 return;
 521         } else if (wpa_s->conf->ap_scan == 2) {
 522                 /*
 523                  * User-initiated scan request in ap_scan == 2; scan with
 524                  * wildcard SSID.
 525                  */
 526                 ssid = NULL;
 527         } else {
 528                 struct wpa_ssid *start = ssid, *tssid;
 529                 int freqs_set = 0;
 530                 if (ssid == NULL && max_ssids > 1)
 531                         ssid = wpa_s->conf->ssid;
 532                 while (ssid) {
 533                         if (!ssid->disabled && ssid->scan_ssid) {
 534                                 wpa_hexdump_ascii(MSG_DEBUG, "Scan SSID",
 535                                                   ssid->ssid, ssid->ssid_len);
 536                                 params.ssids[params.num_ssids].ssid =
 537                                         ssid->ssid;
 538                                 params.ssids[params.num_ssids].ssid_len =
 539                                         ssid->ssid_len;
 540                                 params.num_ssids++;
 541                                 if (params.num_ssids + 1 >= max_ssids)
 542                                         break;
 543                         }
 544                         ssid = ssid->next;
 545                         if (ssid == start)
 546                                 break;
 547                         if (ssid == NULL && max_ssids > 1 &&
 548                             start != wpa_s->conf->ssid)
 549                                 ssid = wpa_s->conf->ssid;
 550                 }
 551
 552                 for (tssid = wpa_s->conf->ssid; tssid; tssid = tssid->next) {
 553                         if (tssid->disabled)
 554                                 continue;
 555                         if ((params.freqs || !freqs_set) && tssid->scan_freq) {
 556                                 int_array_concat(&params.freqs,
 557                                                  tssid->scan_freq);
 558                         } else {
 559                                 os_free(params.freqs);
 560                                 params.freqs = NULL;
 561                         }
 562                         freqs_set = 1;
 563                 }
 564                 int_array_sort_unique(params.freqs);
 565         }
 566
 567         if (ssid) {
 568                 wpa_s->prev_scan_ssid = ssid;
 569                 if (max_ssids > 1) {
 570                         wpa_dbg(wpa_s, MSG_DEBUG, "Include wildcard SSID in "
 571                                 "the scan request");
 572                         params.num_ssids++;
 573                 }
 574                 wpa_dbg(wpa_s, MSG_DEBUG, "Starting AP scan for specific "
 575                         "SSID(s)");
 576         } else {
 577                 wpa_s->prev_scan_ssid = WILDCARD_SSID_SCAN;
 578                 params.num_ssids++;
 579                 wpa_dbg(wpa_s, MSG_DEBUG, "Starting AP scan for wildcard "
 580                         "SSID");
 581         }
 582
 583         wpa_supplicant_optimize_freqs(wpa_s, &params);
 584         extra_ie = wpa_supplicant_extra_ies(wpa_s, &params);
 585
 586         if (params.freqs == NULL && wpa_s->next_scan_freqs) {
 587                 wpa_dbg(wpa_s, MSG_DEBUG, "Optimize scan based on previously "
 588                         "generated frequency list");
 589                 params.freqs = wpa_s->next_scan_freqs;
 590         } else
 591                 os_free(wpa_s->next_scan_freqs);
 592         wpa_s->next_scan_freqs = NULL;
 593
 594         params.filter_ssids = wpa_supplicant_build_filter_ssids(
 595                 wpa_s->conf, &params.num_filter_ssids);
 596         if (extra_ie) {
 597                 params.extra_ies = wpabuf_head(extra_ie);
 598                 params.extra_ies_len = wpabuf_len(extra_ie);
 599         }
 600
 601 #ifdef CONFIG_P2P
 602         if (wpa_s->p2p_in_provisioning) {
 603                 /*
 604                  * The interface may not yet be in P2P mode, so we have to
 605                  * explicitly request P2P probe to disable CCK rates.
 606                  */
 607                 params.p2p_probe = 1;
 608         }
 609 #endif /* CONFIG_P2P */
 610
 611         ret = wpa_supplicant_trigger_scan(wpa_s, &params);
 612
 613         wpabuf_free(extra_ie);
 614         os_free(params.freqs);
 615         os_free(params.filter_ssids);
 616
 617         if (ret) {
 618                 wpa_msg(wpa_s, MSG_WARNING, "Failed to initiate AP scan");
 619                 if (prev_state != wpa_s->wpa_state)
 620                         wpa_supplicant_set_state(wpa_s, prev_state);
 621                 wpa_supplicant_req_scan(wpa_s, 1, 0);
 622         }
 623 }
 624
 625
 626 /**
 627  * wpa_supplicant_req_scan - Schedule a scan for neighboring access points
 628  * @wpa_s: Pointer to wpa_supplicant data
 629  * @sec: Number of seconds after which to scan
 630  * @usec: Number of microseconds after which to scan
 631  *
 632  * This function is used to schedule a scan for neighboring access points after
 633  * the specified time.
 634  */
 635 void wpa_supplicant_req_scan(struct wpa_supplicant *wpa_s, int sec, int usec)
 636 {
 637         /* If there's at least one network that should be specifically scanned
 638          * then don't cancel the scan and reschedule.  Some drivers do
 639          * background scanning which generates frequent scan results, and that
 640          * causes the specific SSID scan to get continually pushed back and
 641          * never happen, which causes hidden APs to never get probe-scanned.
 642          */
 643         if (eloop_is_timeout_registered(wpa_supplicant_scan, wpa_s, NULL) &&
 644             wpa_s->conf->ap_scan == 1) {
 645                 struct wpa_ssid *ssid = wpa_s->conf->ssid;
 646
 647                 while (ssid) {
 648                         if (!ssid->disabled && ssid->scan_ssid)
 649                                 break;
 650                         ssid = ssid->next;
 651                 }
 652                 if (ssid) {
 653                         wpa_dbg(wpa_s, MSG_DEBUG, "Not rescheduling scan to "
 654                                 "ensure that specific SSID scans occur");
 655                         return;
 656                 }
 657         }
 658
 659         wpa_dbg(wpa_s, MSG_DEBUG, "Setting scan request: %d sec %d usec",
 660                 sec, usec);
 661         eloop_cancel_timeout(wpa_supplicant_scan, wpa_s, NULL);
 662         eloop_register_timeout(sec, usec, wpa_supplicant_scan, wpa_s, NULL);
 663 }
 664
 665
 666 /**
 667  * wpa_supplicant_delayed_sched_scan - Request a delayed scheduled scan
 668  * @wpa_s: Pointer to wpa_supplicant data
 669  * @sec: Number of seconds after which to scan
 670  * @usec: Number of microseconds after which to scan
 671  *
 672  * This function is used to schedule periodic scans for neighboring
 673  * access points after the specified time.
 674  */
 675 int wpa_supplicant_delayed_sched_scan(struct wpa_supplicant *wpa_s,
 676                                       int sec, int usec)
 677 {
 678         if (!wpa_s->sched_scan_supported)
 679                 return -1;
 680
 681         eloop_register_timeout(sec, usec,
 682                                wpa_supplicant_delayed_sched_scan_timeout,
 683                                wpa_s, NULL);
 684
 685         return 0;
 686 }
 687
 688
 689 /**
 690  * wpa_supplicant_req_sched_scan - Start a periodic scheduled scan
 691  * @wpa_s: Pointer to wpa_supplicant data
 692  *
 693  * This function is used to schedule periodic scans for neighboring
 694  * access points repeating the scan continuously.
 695  */
 696 int wpa_supplicant_req_sched_scan(struct wpa_supplicant *wpa_s)
 697 {
 698         struct wpa_driver_scan_params params;
 699         enum wpa_states prev_state;
 700         struct wpa_ssid *ssid;
 701         struct wpabuf *wps_ie = NULL;
 702         int ret;
 703         unsigned int max_sched_scan_ssids;
 704
 705         if (!wpa_s->sched_scan_supported)
 706                 return -1;
 707
 708         if (wpa_s->max_sched_scan_ssids > WPAS_MAX_SCAN_SSIDS)
 709                 max_sched_scan_ssids = WPAS_MAX_SCAN_SSIDS;
 710         else
 711                 max_sched_scan_ssids = wpa_s->max_sched_scan_ssids;
 712
 713         if (wpa_s->sched_scanning)
 714                 return 0;
 715
 716         os_memset(&params, 0, sizeof(params));
 717
 718         /* If we can't allocate space for the filters, we just don't filter */
 719         params.filter_ssids = os_zalloc(wpa_s->max_match_sets *
 720                                         sizeof(struct wpa_driver_scan_filter));
 721
 722         prev_state = wpa_s->wpa_state;
 723         if (wpa_s->wpa_state == WPA_DISCONNECTED ||
 724             wpa_s->wpa_state == WPA_INACTIVE)
 725                 wpa_supplicant_set_state(wpa_s, WPA_SCANNING);
 726
 727         /* Find the starting point from which to continue scanning */
 728         ssid = wpa_s->conf->ssid;
 729         if (wpa_s->prev_sched_ssid) {
 730                 while (ssid) {
 731                         if (ssid == wpa_s->prev_sched_ssid) {
 732                                 ssid = ssid->next;
 733                                 break;
 734                         }
 735                         ssid = ssid->next;
 736                 }
 737         }
 738
 739         if (!ssid || !wpa_s->prev_sched_ssid) {
 740                 wpa_dbg(wpa_s, MSG_DEBUG, "Beginning of SSID list");
 741
 742                 wpa_s->sched_scan_interval = 2;
 743                 wpa_s->sched_scan_timeout = max_sched_scan_ssids * 2;
 744                 wpa_s->first_sched_scan = 1;
 745                 ssid = wpa_s->conf->ssid;
 746                 wpa_s->prev_sched_ssid = ssid;
 747         }
 748
 749         while (ssid) {
 750                 if (ssid->disabled) {
 751                         wpa_s->prev_sched_ssid = ssid;
 752                         ssid = ssid->next;
 753                         continue;
 754                 }
 755
 756                 if (params.filter_ssids && ssid->ssid && ssid->ssid_len) {
 757                         os_memcpy(params.filter_ssids[params.num_filter_ssids].ssid,
 758                                   ssid->ssid, ssid->ssid_len);
 759                         params.filter_ssids[params.num_filter_ssids].ssid_len =
 760                                 ssid->ssid_len;
 761                         params.num_filter_ssids++;
 762                 }
 763
 764                 if (ssid->scan_ssid) {
 765                         params.ssids[params.num_ssids].ssid =
 766                                 ssid->ssid;
 767                         params.ssids[params.num_ssids].ssid_len =
 768                                 ssid->ssid_len;
 769                         params.num_ssids++;
 770                         if (params.num_ssids >= max_sched_scan_ssids) {
 771                                 wpa_s->prev_sched_ssid = ssid;
 772                                 break;
 773                         }
 774                 }
 775
 776                 if (params.num_filter_ssids >= wpa_s->max_match_sets)
 777                         break;
 778                 wpa_s->prev_sched_ssid = ssid;
 779                 ssid = ssid->next;
 780         }
 781
 782         if (!params.num_ssids) {
 783                 os_free(params.filter_ssids);
 784                 return 0;
 785         }
 786
 787         if (wpa_s->wps)
 788                 wps_ie = wpa_supplicant_extra_ies(wpa_s, &params);
 789
 790         wpa_dbg(wpa_s, MSG_DEBUG,
 791                 "Starting sched scan: interval %d timeout %d",
 792                 wpa_s->sched_scan_interval, wpa_s->sched_scan_timeout);
 793
 794         ret = wpa_supplicant_start_sched_scan(wpa_s, &params,
 795                                               wpa_s->sched_scan_interval);
 796         wpabuf_free(wps_ie);
 797         os_free(params.filter_ssids);
 798         if (ret) {
 799                 wpa_msg(wpa_s, MSG_WARNING, "Failed to initiate sched scan");
 800                 if (prev_state != wpa_s->wpa_state)
 801                         wpa_supplicant_set_state(wpa_s, prev_state);
 802                 return ret;
 803         }
 804
 805         /* If we have more SSIDs to scan, add a timeout so we scan them too */
 806         if (ssid || !wpa_s->first_sched_scan) {
 807                 wpa_s->sched_scan_timed_out = 0;
 808                 eloop_register_timeout(wpa_s->sched_scan_timeout, 0,
 809                                        wpa_supplicant_sched_scan_timeout,
 810                                        wpa_s, NULL);
 811                 wpa_s->first_sched_scan = 0;
 812                 wpa_s->sched_scan_timeout /= 2;
 813                 wpa_s->sched_scan_interval *= 2;
 814         }
 815
 816         return 0;
 817 }
 818
 819
 820 /**
 821  * wpa_supplicant_cancel_scan - Cancel a scheduled scan request
 822  * @wpa_s: Pointer to wpa_supplicant data
 823  *
 824  * This function is used to cancel a scan request scheduled with
 825  * wpa_supplicant_req_scan().
 826  */
 827 void wpa_supplicant_cancel_scan(struct wpa_supplicant *wpa_s)
 828 {
 829         wpa_dbg(wpa_s, MSG_DEBUG, "Cancelling scan request");
 830         eloop_cancel_timeout(wpa_supplicant_scan, wpa_s, NULL);
 831 }
 832
 833
 834 /**
 835  * wpa_supplicant_cancel_sched_scan - Stop running scheduled scans
 836  * @wpa_s: Pointer to wpa_supplicant data
 837  *
 838  * This function is used to stop a periodic scheduled scan.
 839  */
 840 void wpa_supplicant_cancel_sched_scan(struct wpa_supplicant *wpa_s)
 841 {
 842         if (!wpa_s->sched_scanning)
 843                 return;
 844
 845         wpa_dbg(wpa_s, MSG_DEBUG, "Cancelling sched scan");
 846         eloop_cancel_timeout(wpa_supplicant_sched_scan_timeout, wpa_s, NULL);
 847         wpa_supplicant_stop_sched_scan(wpa_s);
 848 }
 849
 850
 851 void wpa_supplicant_notify_scanning(struct wpa_supplicant *wpa_s,
 852                                     int scanning)
 853 {
 854         if (wpa_s->scanning != scanning) {
 855                 wpa_s->scanning = scanning;
 856                 wpas_notify_scanning(wpa_s);
 857         }
 858 }
 859
 860
 861 static int wpa_scan_get_max_rate(const struct wpa_scan_res *res)
 862 {
 863         int rate = 0;
 864         const u8 *ie;
 865         int i;
 866
 867         ie = wpa_scan_get_ie(res, WLAN_EID_SUPP_RATES);
 868         for (i = 0; ie && i < ie[1]; i++) {
 869                 if ((ie[i + 2] & 0x7f) > rate)
 870                         rate = ie[i + 2] & 0x7f;
 871         }
 872
 873         ie = wpa_scan_get_ie(res, WLAN_EID_EXT_SUPP_RATES);
 874         for (i = 0; ie && i < ie[1]; i++) {
 875                 if ((ie[i + 2] & 0x7f) > rate)
 876                         rate = ie[i + 2] & 0x7f;
 877         }
 878
 879         return rate;
 880 }
 881
 882
 883 const u8 * wpa_scan_get_ie(const struct wpa_scan_res *res, u8 ie)
 884 {
 885         const u8 *end, *pos;
 886
 887         pos = (const u8 *) (res + 1);
 888         end = pos + res->ie_len;
 889
 890         while (pos + 1 < end) {
 891                 if (pos + 2 + pos[1] > end)
 892                         break;
 893                 if (pos[0] == ie)
 894                         return pos;
 895                 pos += 2 + pos[1];
 896         }
 897
 898         return NULL;
 899 }
 900
 901
 902 const u8 * wpa_scan_get_vendor_ie(const struct wpa_scan_res *res,
 903                                   u32 vendor_type)
 904 {
 905         const u8 *end, *pos;
 906
 907         pos = (const u8 *) (res + 1);
 908         end = pos + res->ie_len;
 909
 910         while (pos + 1 < end) {
 911                 if (pos + 2 + pos[1] > end)
 912                         break;
 913                 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
 914                     vendor_type == WPA_GET_BE32(&pos[2]))
 915                         return pos;
 916                 pos += 2 + pos[1];
 917         }
 918
 919         return NULL;
 920 }
 921
 922
 923 struct wpabuf * wpa_scan_get_vendor_ie_multi(const struct wpa_scan_res *res,
 924                                              u32 vendor_type)
 925 {
 926         struct wpabuf *buf;
 927         const u8 *end, *pos;
 928
 929         buf = wpabuf_alloc(res->ie_len);
 930         if (buf == NULL)
 931                 return NULL;
 932
 933         pos = (const u8 *) (res + 1);
 934         end = pos + res->ie_len;
 935
 936         while (pos + 1 < end) {
 937                 if (pos + 2 + pos[1] > end)
 938                         break;
 939                 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
 940                     vendor_type == WPA_GET_BE32(&pos[2]))
 941                         wpabuf_put_data(buf, pos + 2 + 4, pos[1] - 4);
 942                 pos += 2 + pos[1];
 943         }
 944
 945         if (wpabuf_len(buf) == 0) {
 946                 wpabuf_free(buf);
 947                 buf = NULL;
 948         }
 949
 950         return buf;
 951 }
 952
 953
 954 struct wpabuf * wpa_scan_get_vendor_ie_multi_beacon(
 955         const struct wpa_scan_res *res, u32 vendor_type)
 956 {
 957         struct wpabuf *buf;
 958         const u8 *end, *pos;
 959
 960         if (res->beacon_ie_len == 0)
 961                 return NULL;
 962         buf = wpabuf_alloc(res->beacon_ie_len);
 963         if (buf == NULL)
 964                 return NULL;
 965
 966         pos = (const u8 *) (res + 1);
 967         pos += res->ie_len;
 968         end = pos + res->beacon_ie_len;
 969
 970         while (pos + 1 < end) {
 971                 if (pos + 2 + pos[1] > end)
 972                         break;
 973                 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
 974                     vendor_type == WPA_GET_BE32(&pos[2]))
 975                         wpabuf_put_data(buf, pos + 2 + 4, pos[1] - 4);
 976                 pos += 2 + pos[1];
 977         }
 978
 979         if (wpabuf_len(buf) == 0) {
 980                 wpabuf_free(buf);
 981                 buf = NULL;
 982         }
 983
 984         return buf;
 985 }
 986
 987
 988 /* Compare function for sorting scan results. Return >0 if @b is considered
 989  * better. */
 990 static int wpa_scan_result_compar(const void *a, const void *b)
 991 {
 992         struct wpa_scan_res **_wa = (void *) a;
 993         struct wpa_scan_res **_wb = (void *) b;
 994         struct wpa_scan_res *wa = *_wa;
 995         struct wpa_scan_res *wb = *_wb;
 996         int wpa_a, wpa_b, maxrate_a, maxrate_b;
 997
 998         /* WPA/WPA2 support preferred */
 999         wpa_a = wpa_scan_get_vendor_ie(wa, WPA_IE_VENDOR_TYPE) != NULL ||
1000                 wpa_scan_get_ie(wa, WLAN_EID_RSN) != NULL;
1001         wpa_b = wpa_scan_get_vendor_ie(wb, WPA_IE_VENDOR_TYPE) != NULL ||
1002                 wpa_scan_get_ie(wb, WLAN_EID_RSN) != NULL;
1003
1004         if (wpa_b && !wpa_a)
1005                 return 1;
1006         if (!wpa_b && wpa_a)
1007                 return -1;
1008
1009         /* privacy support preferred */
1010         if ((wa->caps & IEEE80211_CAP_PRIVACY) == 0 &&
1011             (wb->caps & IEEE80211_CAP_PRIVACY))
1012                 return 1;
1013         if ((wa->caps & IEEE80211_CAP_PRIVACY) &&
1014             (wb->caps & IEEE80211_CAP_PRIVACY) == 0)
1015                 return -1;
1016
1017         /* best/max rate preferred if signal level close enough XXX */
1018         if ((wa->level && wb->level && abs(wb->level - wa->level) < 5) ||
1019             (wa->qual && wb->qual && abs(wb->qual - wa->qual) < 10)) {
1020                 maxrate_a = wpa_scan_get_max_rate(wa);
1021                 maxrate_b = wpa_scan_get_max_rate(wb);
1022                 if (maxrate_a != maxrate_b)
1023                         return maxrate_b - maxrate_a;
1024         }
1025
1026         /* use freq for channel preference */
1027
1028         /* all things being equal, use signal level; if signal levels are
1029          * identical, use quality values since some drivers may only report
1030          * that value and leave the signal level zero */
1031         if (wb->level == wa->level)
1032                 return wb->qual - wa->qual;
1033         return wb->level - wa->level;
1034 }
1035
1036
1037 #ifdef CONFIG_WPS
1038 /* Compare function for sorting scan results when searching a WPS AP for
1039  * provisioning. Return >0 if @b is considered better. */
1040 static int wpa_scan_result_wps_compar(const void *a, const void *b)
1041 {
1042         struct wpa_scan_res **_wa = (void *) a;
1043         struct wpa_scan_res **_wb = (void *) b;
1044         struct wpa_scan_res *wa = *_wa;
1045         struct wpa_scan_res *wb = *_wb;
1046         int uses_wps_a, uses_wps_b;
1047         struct wpabuf *wps_a, *wps_b;
1048         int res;
1049
1050         /* Optimization - check WPS IE existence before allocated memory and
1051          * doing full reassembly. */
1052         uses_wps_a = wpa_scan_get_vendor_ie(wa, WPS_IE_VENDOR_TYPE) != NULL;
1053         uses_wps_b = wpa_scan_get_vendor_ie(wb, WPS_IE_VENDOR_TYPE) != NULL;
1054         if (uses_wps_a && !uses_wps_b)
1055                 return -1;
1056         if (!uses_wps_a && uses_wps_b)
1057                 return 1;
1058
1059         if (uses_wps_a && uses_wps_b) {
1060                 wps_a = wpa_scan_get_vendor_ie_multi(wa, WPS_IE_VENDOR_TYPE);
1061                 wps_b = wpa_scan_get_vendor_ie_multi(wb, WPS_IE_VENDOR_TYPE);
1062                 res = wps_ap_priority_compar(wps_a, wps_b);
1063                 wpabuf_free(wps_a);
1064                 wpabuf_free(wps_b);
1065                 if (res)
1066                         return res;
1067         }
1068
1069         /*
1070          * Do not use current AP security policy as a sorting criteria during
1071          * WPS provisioning step since the AP may get reconfigured at the
1072          * completion of provisioning.
1073          */
1074
1075         /* all things being equal, use signal level; if signal levels are
1076          * identical, use quality values since some drivers may only report
1077          * that value and leave the signal level zero */
1078         if (wb->level == wa->level)
1079                 return wb->qual - wa->qual;
1080         return wb->level - wa->level;
1081 }
1082 #endif /* CONFIG_WPS */
1083
1084
1085 /**
1086  * wpa_supplicant_get_scan_results - Get scan results
1087  * @wpa_s: Pointer to wpa_supplicant data
1088  * @info: Information about what was scanned or %NULL if not available
1089  * @new_scan: Whether a new scan was performed
1090  * Returns: Scan results, %NULL on failure
1091  *
1092  * This function request the current scan results from the driver and updates
1093  * the local BSS list wpa_s->bss. The caller is responsible for freeing the
1094  * results with wpa_scan_results_free().
1095  */
1096 struct wpa_scan_results *
1097 wpa_supplicant_get_scan_results(struct wpa_supplicant *wpa_s,
1098                                 struct scan_info *info, int new_scan)
1099 {
1100         struct wpa_scan_results *scan_res;
1101         size_t i;
1102         int (*compar)(const void *, const void *) = wpa_scan_result_compar;
1103
1104         scan_res = wpa_drv_get_scan_results2(wpa_s);
1105         if (scan_res == NULL) {
1106                 wpa_dbg(wpa_s, MSG_DEBUG, "Failed to get scan results");
1107                 return NULL;
1108         }
1109
1110 #ifdef CONFIG_WPS
1111         if (wpas_wps_in_progress(wpa_s)) {
1112                 wpa_dbg(wpa_s, MSG_DEBUG, "WPS: Order scan results with WPS "
1113                         "provisioning rules");
1114                 compar = wpa_scan_result_wps_compar;
1115         }
1116 #endif /* CONFIG_WPS */
1117
1118         qsort(scan_res->res, scan_res->num, sizeof(struct wpa_scan_res *),
1119               compar);
1120
1121         wpa_bss_update_start(wpa_s);
1122         for (i = 0; i < scan_res->num; i++)
1123                 wpa_bss_update_scan_res(wpa_s, scan_res->res[i]);
1124         wpa_bss_update_end(wpa_s, info, new_scan);
1125
1126         return scan_res;
1127 }
1128
1129
1130 int wpa_supplicant_update_scan_results(struct wpa_supplicant *wpa_s)
1131 {
1132         struct wpa_scan_results *scan_res;
1133         scan_res = wpa_supplicant_get_scan_results(wpa_s, NULL, 0);
1134         if (scan_res == NULL)
1135                 return -1;
1136         wpa_scan_results_free(scan_res);
1137
1138         return 0;
1139 }