wpa_supplicant/bss.c

   1 /*
   2  * BSS table
   3  * Copyright (c) 2009-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 "drivers/driver.h"
  21 #include "wpa_supplicant_i.h"
  22 #include "notify.h"
  23 #include "bss.h"
  24
  25
  26 #ifndef WPA_BSS_MAX_COUNT
  27 #define WPA_BSS_MAX_COUNT 200
  28 #endif /* WPA_BSS_MAX_COUNT */
  29
  30 /**
  31  * WPA_BSS_EXPIRATION_PERIOD - Period of expiration run in seconds
  32  */
  33 #define WPA_BSS_EXPIRATION_PERIOD 10
  34
  35 /**
  36  * WPA_BSS_EXPIRATION_AGE - BSS entry age after which it can be expired
  37  *
  38  * This value control the time in seconds after which a BSS entry gets removed
  39  * if it has not been updated or is not in use.
  40  */
  41 #define WPA_BSS_EXPIRATION_AGE 180
  42
  43 /**
  44  * WPA_BSS_EXPIRATION_SCAN_COUNT - Expire BSS after number of scans
  45  *
  46  * If the BSS entry has not been seen in this many scans, it will be removed.
  47  * Value 1 means that the entry is removed after the first scan without the
  48  * BSSID being seen. Larger values can be used to avoid BSS entries
  49  * disappearing if they are not visible in every scan (e.g., low signal quality
  50  * or interference).
  51  */
  52 #define WPA_BSS_EXPIRATION_SCAN_COUNT 2
  53
  54
  55 static void wpa_bss_remove(struct wpa_supplicant *wpa_s, struct wpa_bss *bss)
  56 {
  57         dl_list_del(&bss->list);
  58         dl_list_del(&bss->list_id);
  59         wpa_s->num_bss--;
  60         wpa_printf(MSG_DEBUG, "BSS: Remove id %u BSSID " MACSTR " SSID '%s'",
  61                    bss->id, MAC2STR(bss->bssid),
  62                    wpa_ssid_txt(bss->ssid, bss->ssid_len));
  63         wpas_notify_bss_removed(wpa_s, bss->bssid, bss->id);
  64         os_free(bss);
  65 }
  66
  67
  68 struct wpa_bss * wpa_bss_get(struct wpa_supplicant *wpa_s, const u8 *bssid,
  69                              const u8 *ssid, size_t ssid_len)
  70 {
  71         struct wpa_bss *bss;
  72         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
  73                 if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0 &&
  74                     bss->ssid_len == ssid_len &&
  75                     os_memcmp(bss->ssid, ssid, ssid_len) == 0)
  76                         return bss;
  77         }
  78         return NULL;
  79 }
  80
  81
  82 static void wpa_bss_copy_res(struct wpa_bss *dst, struct wpa_scan_res *src)
  83 {
  84         os_time_t usec;
  85
  86         dst->flags = src->flags;
  87         os_memcpy(dst->bssid, src->bssid, ETH_ALEN);
  88         dst->freq = src->freq;
  89         dst->beacon_int = src->beacon_int;
  90         dst->caps = src->caps;
  91         dst->qual = src->qual;
  92         dst->noise = src->noise;
  93         dst->level = src->level;
  94         dst->tsf = src->tsf;
  95
  96         os_get_time(&dst->last_update);
  97         dst->last_update.sec -= src->age / 1000;
  98         usec = (src->age % 1000) * 1000;
  99         if (dst->last_update.usec < usec) {
 100                 dst->last_update.sec--;
 101                 dst->last_update.usec += 1000000;
 102         }
 103         dst->last_update.usec -= usec;
 104 }
 105
 106
 107 static void wpa_bss_add(struct wpa_supplicant *wpa_s,
 108                         const u8 *ssid, size_t ssid_len,
 109                         struct wpa_scan_res *res)
 110 {
 111         struct wpa_bss *bss;
 112
 113         bss = os_zalloc(sizeof(*bss) + res->ie_len);
 114         if (bss == NULL)
 115                 return;
 116         bss->id = wpa_s->bss_next_id++;
 117         bss->last_update_idx = wpa_s->bss_update_idx;
 118         wpa_bss_copy_res(bss, res);
 119         os_memcpy(bss->ssid, ssid, ssid_len);
 120         bss->ssid_len = ssid_len;
 121         bss->ie_len = res->ie_len;
 122         os_memcpy(bss + 1, res + 1, res->ie_len);
 123
 124         dl_list_add_tail(&wpa_s->bss, &bss->list);
 125         dl_list_add_tail(&wpa_s->bss_id, &bss->list_id);
 126         wpa_s->num_bss++;
 127         wpa_printf(MSG_DEBUG, "BSS: Add new id %u BSSID " MACSTR " SSID '%s'",
 128                    bss->id, MAC2STR(bss->bssid), wpa_ssid_txt(ssid, ssid_len));
 129         wpas_notify_bss_added(wpa_s, bss->bssid, bss->id);
 130         if (wpa_s->num_bss > WPA_BSS_MAX_COUNT) {
 131                 /* Remove the oldest entry */
 132                 wpa_bss_remove(wpa_s, dl_list_first(&wpa_s->bss,
 133                                                     struct wpa_bss, list));
 134         }
 135 }
 136
 137
 138 static void wpa_bss_update(struct wpa_supplicant *wpa_s, struct wpa_bss *bss,
 139                            struct wpa_scan_res *res)
 140 {
 141         bss->scan_miss_count = 0;
 142         bss->last_update_idx = wpa_s->bss_update_idx;
 143         wpa_bss_copy_res(bss, res);
 144         /* Move the entry to the end of the list */
 145         dl_list_del(&bss->list);
 146         if (bss->ie_len >= res->ie_len) {
 147                 os_memcpy(bss + 1, res + 1, res->ie_len);
 148                 bss->ie_len = res->ie_len;
 149         } else {
 150                 struct wpa_bss *nbss;
 151                 nbss = os_realloc(bss, sizeof(*bss) + res->ie_len);
 152                 if (nbss) {
 153                         bss = nbss;
 154                         os_memcpy(bss + 1, res + 1, res->ie_len);
 155                         bss->ie_len = res->ie_len;
 156                 }
 157         }
 158         dl_list_add_tail(&wpa_s->bss, &bss->list);
 159 }
 160
 161
 162 void wpa_bss_update_start(struct wpa_supplicant *wpa_s)
 163 {
 164         wpa_s->bss_update_idx++;
 165         wpa_printf(MSG_DEBUG, "BSS: Start scan result update %u",
 166                    wpa_s->bss_update_idx);
 167 }
 168
 169
 170 void wpa_bss_update_scan_res(struct wpa_supplicant *wpa_s,
 171                              struct wpa_scan_res *res)
 172 {
 173         const u8 *ssid;
 174         struct wpa_bss *bss;
 175
 176         ssid = wpa_scan_get_ie(res, WLAN_EID_SSID);
 177         if (ssid == NULL) {
 178                 wpa_printf(MSG_DEBUG, "BSS: No SSID IE included for " MACSTR,
 179                            MAC2STR(res->bssid));
 180                 return;
 181         }
 182         if (ssid[1] > 32) {
 183                 wpa_printf(MSG_DEBUG, "BSS: Too long SSID IE included for "
 184                            MACSTR, MAC2STR(res->bssid));
 185                 return;
 186         }
 187
 188         /* TODO: add option for ignoring BSSes we are not interested in
 189          * (to save memory) */
 190         bss = wpa_bss_get(wpa_s, res->bssid, ssid + 2, ssid[1]);
 191         if (bss == NULL)
 192                 wpa_bss_add(wpa_s, ssid + 2, ssid[1], res);
 193         else
 194                 wpa_bss_update(wpa_s, bss, res);
 195 }
 196
 197
 198 static int wpa_bss_included_in_scan(const struct wpa_bss *bss,
 199                                     const struct scan_info *info)
 200 {
 201         int found;
 202         size_t i;
 203
 204         if (info == NULL)
 205                 return 1;
 206
 207         if (info->num_freqs) {
 208                 found = 0;
 209                 for (i = 0; i < info->num_freqs; i++) {
 210                         if (bss->freq == info->freqs[i]) {
 211                                 found = 1;
 212                                 break;
 213                         }
 214                 }
 215                 if (!found)
 216                         return 0;
 217         }
 218
 219         if (info->num_ssids) {
 220                 found = 0;
 221                 for (i = 0; i < info->num_ssids; i++) {
 222                         const struct wpa_driver_scan_ssid *s = &info->ssids[i];
 223                         if ((s->ssid == NULL || s->ssid_len == 0) ||
 224                             (s->ssid_len == bss->ssid_len &&
 225                              os_memcmp(s->ssid, bss->ssid, bss->ssid_len) ==
 226                              0)) {
 227                                 found = 1;
 228                                 break;
 229                         }
 230                 }
 231                 if (!found)
 232                         return 0;
 233         }
 234
 235         return 1;
 236 }
 237
 238
 239 void wpa_bss_update_end(struct wpa_supplicant *wpa_s, struct scan_info *info,
 240                         int new_scan)
 241 {
 242         struct wpa_bss *bss, *n;
 243
 244         if (!new_scan)
 245                 return; /* do not expire entries without new scan */
 246
 247         dl_list_for_each_safe(bss, n, &wpa_s->bss, struct wpa_bss, list) {
 248                 if (!wpa_bss_included_in_scan(bss, info))
 249                         continue; /* expire only BSSes that were scanned */
 250                 if (bss->last_update_idx < wpa_s->bss_update_idx)
 251                         bss->scan_miss_count++;
 252                 if (bss->scan_miss_count >= WPA_BSS_EXPIRATION_SCAN_COUNT) {
 253                         wpa_printf(MSG_DEBUG, "BSS: Expire BSS %u due to no "
 254                                    "match in scan", bss->id);
 255                         wpa_bss_remove(wpa_s, bss);
 256                 }
 257         }
 258 }
 259
 260
 261 static void wpa_bss_timeout(void *eloop_ctx, void *timeout_ctx)
 262 {
 263         struct wpa_supplicant *wpa_s = eloop_ctx;
 264         struct wpa_bss *bss, *n;
 265         struct os_time t;
 266
 267         if (dl_list_empty(&wpa_s->bss))
 268                 return;
 269
 270         os_get_time(&t);
 271         t.sec -= WPA_BSS_EXPIRATION_AGE;
 272
 273         dl_list_for_each_safe(bss, n, &wpa_s->bss, struct wpa_bss, list) {
 274                 if (os_memcmp(bss->bssid, wpa_s->bssid, ETH_ALEN) == 0 ||
 275                     os_memcmp(bss->bssid, wpa_s->pending_bssid, ETH_ALEN) == 0)
 276                         continue; /* do not expire BSSes that are in use */
 277
 278                 if (os_time_before(&bss->last_update, &t)) {
 279                         wpa_printf(MSG_DEBUG, "BSS: Expire BSS %u due to age",
 280                                    bss->id);
 281                         wpa_bss_remove(wpa_s, bss);
 282                 } else
 283                         break;
 284         }
 285         eloop_register_timeout(WPA_BSS_EXPIRATION_PERIOD, 0,
 286                                wpa_bss_timeout, wpa_s, NULL);
 287 }
 288
 289
 290 int wpa_bss_init(struct wpa_supplicant *wpa_s)
 291 {
 292         dl_list_init(&wpa_s->bss);
 293         dl_list_init(&wpa_s->bss_id);
 294         eloop_register_timeout(WPA_BSS_EXPIRATION_PERIOD, 0,
 295                                wpa_bss_timeout, wpa_s, NULL);
 296         return 0;
 297 }
 298
 299
 300 void wpa_bss_deinit(struct wpa_supplicant *wpa_s)
 301 {
 302         struct wpa_bss *bss, *n;
 303         eloop_cancel_timeout(wpa_bss_timeout, wpa_s, NULL);
 304         if (wpa_s->bss.next == NULL)
 305                 return; /* BSS table not yet initialized */
 306         dl_list_for_each_safe(bss, n, &wpa_s->bss, struct wpa_bss, list)
 307                 wpa_bss_remove(wpa_s, bss);
 308 }
 309
 310
 311 struct wpa_bss * wpa_bss_get_bssid(struct wpa_supplicant *wpa_s,
 312                                    const u8 *bssid)
 313 {
 314         struct wpa_bss *bss;
 315         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
 316                 if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
 317                         return bss;
 318         }
 319         return NULL;
 320 }
 321
 322
 323 struct wpa_bss * wpa_bss_get_id(struct wpa_supplicant *wpa_s, unsigned int id)
 324 {
 325         struct wpa_bss *bss;
 326         dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
 327                 if (bss->id == id)
 328                         return bss;
 329         }
 330         return NULL;
 331 }
 332
 333
 334 const u8 * wpa_bss_get_ie(const struct wpa_bss *bss, u8 ie)
 335 {
 336         const u8 *end, *pos;
 337
 338         pos = (const u8 *) (bss + 1);
 339         end = pos + bss->ie_len;
 340
 341         while (pos + 1 < end) {
 342                 if (pos + 2 + pos[1] > end)
 343                         break;
 344                 if (pos[0] == ie)
 345                         return pos;
 346                 pos += 2 + pos[1];
 347         }
 348
 349         return NULL;
 350 }
 351
 352
 353 const u8 * wpa_bss_get_vendor_ie(const struct wpa_bss *bss, u32 vendor_type)
 354 {
 355         const u8 *end, *pos;
 356
 357         pos = (const u8 *) (bss + 1);
 358         end = pos + bss->ie_len;
 359
 360         while (pos + 1 < end) {
 361                 if (pos + 2 + pos[1] > end)
 362                         break;
 363                 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
 364                     vendor_type == WPA_GET_BE32(&pos[2]))
 365                         return pos;
 366                 pos += 2 + pos[1];
 367         }
 368
 369         return NULL;
 370 }
 371
 372
 373 struct wpabuf * wpa_bss_get_vendor_ie_multi(const struct wpa_bss *bss,
 374                                             u32 vendor_type)
 375 {
 376         struct wpabuf *buf;
 377         const u8 *end, *pos;
 378
 379         buf = wpabuf_alloc(bss->ie_len);
 380         if (buf == NULL)
 381                 return NULL;
 382
 383         pos = (const u8 *) (bss + 1);
 384         end = pos + bss->ie_len;
 385
 386         while (pos + 1 < end) {
 387                 if (pos + 2 + pos[1] > end)
 388                         break;
 389                 if (pos[0] == WLAN_EID_VENDOR_SPECIFIC && pos[1] >= 4 &&
 390                     vendor_type == WPA_GET_BE32(&pos[2]))
 391                         wpabuf_put_data(buf, pos + 2 + 4, pos[1] - 4);
 392                 pos += 2 + pos[1];
 393         }
 394
 395         if (wpabuf_len(buf) == 0) {
 396                 wpabuf_free(buf);
 397                 buf = NULL;
 398         }
 399
 400         return buf;
 401 }
 402
 403
 404 int wpa_bss_get_max_rate(const struct wpa_bss *bss)
 405 {
 406         int rate = 0;
 407         const u8 *ie;
 408         int i;
 409
 410         ie = wpa_bss_get_ie(bss, WLAN_EID_SUPP_RATES);
 411         for (i = 0; ie && i < ie[1]; i++) {
 412                 if ((ie[i + 2] & 0x7f) > rate)
 413                         rate = ie[i + 2] & 0x7f;
 414         }
 415
 416         ie = wpa_bss_get_ie(bss, WLAN_EID_EXT_SUPP_RATES);
 417         for (i = 0; ie && i < ie[1]; i++) {
 418                 if ((ie[i + 2] & 0x7f) > rate)
 419                         rate = ie[i + 2] & 0x7f;
 420         }
 421
 422         return rate;
 423 }