wireless/scan.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * cfg80211 scan result handling
   4  *
   5  * Copyright 2008 Johannes Berg <johannes@sipsolutions.net>
   6  * Copyright 2013-2014  Intel Mobile Communications GmbH
   7  * Copyright 2016       Intel Deutschland GmbH
   8  * Copyright (C) 2018-2021 Intel Corporation
   9  */
  10 #include <linux/kernel.h>
  11 #include <linux/slab.h>
  12 #include <linux/module.h>
  13 #include <linux/netdevice.h>
  14 #include <linux/wireless.h>
  15 #include <linux/nl80211.h>
  16 #include <linux/etherdevice.h>
  17 #include <linux/crc32.h>
  18 #include <linux/bitfield.h>
  19 #include <net/arp.h>
  20 #include <net/cfg80211.h>
  21 #include <net/cfg80211-wext.h>
  22 #include <net/iw_handler.h>
  23 #include "core.h"
  24 #include "nl80211.h"
  25 #include "wext-compat.h"
  26 #include "rdev-ops.h"
  27
  28 /**
  29  * DOC: BSS tree/list structure
  30  *
  31  * At the top level, the BSS list is kept in both a list in each
  32  * registered device (@bss_list) as well as an RB-tree for faster
  33  * lookup. In the RB-tree, entries can be looked up using their
  34  * channel, MESHID, MESHCONF (for MBSSes) or channel, BSSID, SSID
  35  * for other BSSes.
  36  *
  37  * Due to the possibility of hidden SSIDs, there's a second level
  38  * structure, the "hidden_list" and "hidden_beacon_bss" pointer.
  39  * The hidden_list connects all BSSes belonging to a single AP
  40  * that has a hidden SSID, and connects beacon and probe response
  41  * entries. For a probe response entry for a hidden SSID, the
  42  * hidden_beacon_bss pointer points to the BSS struct holding the
  43  * beacon's information.
  44  *
  45  * Reference counting is done for all these references except for
  46  * the hidden_list, so that a beacon BSS struct that is otherwise
  47  * not referenced has one reference for being on the bss_list and
  48  * one for each probe response entry that points to it using the
  49  * hidden_beacon_bss pointer. When a BSS struct that has such a
  50  * pointer is get/put, the refcount update is also propagated to
  51  * the referenced struct, this ensure that it cannot get removed
  52  * while somebody is using the probe response version.
  53  *
  54  * Note that the hidden_beacon_bss pointer never changes, due to
  55  * the reference counting. Therefore, no locking is needed for
  56  * it.
  57  *
  58  * Also note that the hidden_beacon_bss pointer is only relevant
  59  * if the driver uses something other than the IEs, e.g. private
  60  * data stored in the BSS struct, since the beacon IEs are
  61  * also linked into the probe response struct.
  62  */
  63
  64 /*
  65  * Limit the number of BSS entries stored in mac80211. Each one is
  66  * a bit over 4k at most, so this limits to roughly 4-5M of memory.
  67  * If somebody wants to really attack this though, they'd likely
  68  * use small beacons, and only one type of frame, limiting each of
  69  * the entries to a much smaller size (in order to generate more
  70  * entries in total, so overhead is bigger.)
  71  */
  72 static int bss_entries_limit = 1000;
  73 module_param(bss_entries_limit, int, 0644);
  74 MODULE_PARM_DESC(bss_entries_limit,
  75                  "limit to number of scan BSS entries (per wiphy, default 1000)");
  76
  77 #define IEEE80211_SCAN_RESULT_EXPIRE    (30 * HZ)
  78
  79 /**
  80  * struct cfg80211_colocated_ap - colocated AP information
  81  *
  82  * @list: linked list to all colocated aPS
  83  * @bssid: BSSID of the reported AP
  84  * @ssid: SSID of the reported AP
  85  * @ssid_len: length of the ssid
  86  * @center_freq: frequency the reported AP is on
  87  * @unsolicited_probe: the reported AP is part of an ESS, where all the APs
  88  *      that operate in the same channel as the reported AP and that might be
  89  *      detected by a STA receiving this frame, are transmitting unsolicited
  90  *      Probe Response frames every 20 TUs
  91  * @oct_recommended: OCT is recommended to exchange MMPDUs with the reported AP
  92  * @same_ssid: the reported AP has the same SSID as the reporting AP
  93  * @multi_bss: the reported AP is part of a multiple BSSID set
  94  * @transmitted_bssid: the reported AP is the transmitting BSSID
  95  * @colocated_ess: all the APs that share the same ESS as the reported AP are
  96  *      colocated and can be discovered via legacy bands.
  97  * @short_ssid_valid: short_ssid is valid and can be used
  98  * @short_ssid: the short SSID for this SSID
  99  */
 100 struct cfg80211_colocated_ap {
 101         struct list_head list;
 102         u8 bssid[ETH_ALEN];
 103         u8 ssid[IEEE80211_MAX_SSID_LEN];
 104         size_t ssid_len;
 105         u32 short_ssid;
 106         u32 center_freq;
 107         u8 unsolicited_probe:1,
 108            oct_recommended:1,
 109            same_ssid:1,
 110            multi_bss:1,
 111            transmitted_bssid:1,
 112            colocated_ess:1,
 113            short_ssid_valid:1;
 114 };
 115
 116 static void bss_free(struct cfg80211_internal_bss *bss)
 117 {
 118         struct cfg80211_bss_ies *ies;
 119
 120         if (WARN_ON(atomic_read(&bss->hold)))
 121                 return;
 122
 123         ies = (void *)rcu_access_pointer(bss->pub.beacon_ies);
 124         if (ies && !bss->pub.hidden_beacon_bss)
 125                 kfree_rcu(ies, rcu_head);
 126         ies = (void *)rcu_access_pointer(bss->pub.proberesp_ies);
 127         if (ies)
 128                 kfree_rcu(ies, rcu_head);
 129
 130         /*
 131          * This happens when the module is removed, it doesn't
 132          * really matter any more save for completeness
 133          */
 134         if (!list_empty(&bss->hidden_list))
 135                 list_del(&bss->hidden_list);
 136
 137         kfree(bss);
 138 }
 139
 140 static inline void bss_ref_get(struct cfg80211_registered_device *rdev,
 141                                struct cfg80211_internal_bss *bss)
 142 {
 143         lockdep_assert_held(&rdev->bss_lock);
 144
 145         bss->refcount++;
 146
 147         if (bss->pub.hidden_beacon_bss)
 148                 bss_from_pub(bss->pub.hidden_beacon_bss)->refcount++;
 149
 150         if (bss->pub.transmitted_bss)
 151                 bss_from_pub(bss->pub.transmitted_bss)->refcount++;
 152 }
 153
 154 static inline void bss_ref_put(struct cfg80211_registered_device *rdev,
 155                                struct cfg80211_internal_bss *bss)
 156 {
 157         lockdep_assert_held(&rdev->bss_lock);
 158
 159         if (bss->pub.hidden_beacon_bss) {
 160                 struct cfg80211_internal_bss *hbss;
 161                 hbss = container_of(bss->pub.hidden_beacon_bss,
 162                                     struct cfg80211_internal_bss,
 163                                     pub);
 164                 hbss->refcount--;
 165                 if (hbss->refcount == 0)
 166                         bss_free(hbss);
 167         }
 168
 169         if (bss->pub.transmitted_bss) {
 170                 struct cfg80211_internal_bss *tbss;
 171
 172                 tbss = container_of(bss->pub.transmitted_bss,
 173                                     struct cfg80211_internal_bss,
 174                                     pub);
 175                 tbss->refcount--;
 176                 if (tbss->refcount == 0)
 177                         bss_free(tbss);
 178         }
 179
 180         bss->refcount--;
 181         if (bss->refcount == 0)
 182                 bss_free(bss);
 183 }
 184
 185 static bool __cfg80211_unlink_bss(struct cfg80211_registered_device *rdev,
 186                                   struct cfg80211_internal_bss *bss)
 187 {
 188         lockdep_assert_held(&rdev->bss_lock);
 189
 190         if (!list_empty(&bss->hidden_list)) {
 191                 /*
 192                  * don't remove the beacon entry if it has
 193                  * probe responses associated with it
 194                  */
 195                 if (!bss->pub.hidden_beacon_bss)
 196                         return false;
 197                 /*
 198                  * if it's a probe response entry break its
 199                  * link to the other entries in the group
 200                  */
 201                 list_del_init(&bss->hidden_list);
 202         }
 203
 204         list_del_init(&bss->list);
 205         list_del_init(&bss->pub.nontrans_list);
 206         rb_erase(&bss->rbn, &rdev->bss_tree);
 207         rdev->bss_entries--;
 208         WARN_ONCE((rdev->bss_entries == 0) ^ list_empty(&rdev->bss_list),
 209                   "rdev bss entries[%d]/list[empty:%d] corruption\n",
 210                   rdev->bss_entries, list_empty(&rdev->bss_list));
 211         bss_ref_put(rdev, bss);
 212         return true;
 213 }
 214
 215 bool cfg80211_is_element_inherited(const struct element *elem,
 216                                    const struct element *non_inherit_elem)
 217 {
 218         u8 id_len, ext_id_len, i, loop_len, id;
 219         const u8 *list;
 220
 221         if (elem->id == WLAN_EID_MULTIPLE_BSSID)
 222                 return false;
 223
 224         if (!non_inherit_elem || non_inherit_elem->datalen < 2)
 225                 return true;
 226
 227         /*
 228          * non inheritance element format is:
 229          * ext ID (56) | IDs list len | list | extension IDs list len | list
 230          * Both lists are optional. Both lengths are mandatory.
 231          * This means valid length is:
 232          * elem_len = 1 (extension ID) + 2 (list len fields) + list lengths
 233          */
 234         id_len = non_inherit_elem->data[1];
 235         if (non_inherit_elem->datalen < 3 + id_len)
 236                 return true;
 237
 238         ext_id_len = non_inherit_elem->data[2 + id_len];
 239         if (non_inherit_elem->datalen < 3 + id_len + ext_id_len)
 240                 return true;
 241
 242         if (elem->id == WLAN_EID_EXTENSION) {
 243                 if (!ext_id_len)
 244                         return true;
 245                 loop_len = ext_id_len;
 246                 list = &non_inherit_elem->data[3 + id_len];
 247                 id = elem->data[0];
 248         } else {
 249                 if (!id_len)
 250                         return true;
 251                 loop_len = id_len;
 252                 list = &non_inherit_elem->data[2];
 253                 id = elem->id;
 254         }
 255
 256         for (i = 0; i < loop_len; i++) {
 257                 if (list[i] == id)
 258                         return false;
 259         }
 260
 261         return true;
 262 }
 263 EXPORT_SYMBOL(cfg80211_is_element_inherited);
 264
 265 static size_t cfg80211_copy_elem_with_frags(const struct element *elem,
 266                                             const u8 *ie, size_t ie_len,
 267                                             u8 **pos, u8 *buf, size_t buf_len)
 268 {
 269         if (WARN_ON((u8 *)elem < ie || elem->data > ie + ie_len ||
 270                     elem->data + elem->datalen > ie + ie_len))
 271                 return 0;
 272
 273         if (elem->datalen + 2 > buf + buf_len - *pos)
 274                 return 0;
 275
 276         memcpy(*pos, elem, elem->datalen + 2);
 277         *pos += elem->datalen + 2;
 278
 279         /* Finish if it is not fragmented  */
 280         if (elem->datalen != 255)
 281                 return *pos - buf;
 282
 283         ie_len = ie + ie_len - elem->data - elem->datalen;
 284         ie = (const u8 *)elem->data + elem->datalen;
 285
 286         for_each_element(elem, ie, ie_len) {
 287                 if (elem->id != WLAN_EID_FRAGMENT)
 288                         break;
 289
 290                 if (elem->datalen + 2 > buf + buf_len - *pos)
 291                         return 0;
 292
 293                 memcpy(*pos, elem, elem->datalen + 2);
 294                 *pos += elem->datalen + 2;
 295
 296                 if (elem->datalen != 255)
 297                         break;
 298         }
 299
 300         return *pos - buf;
 301 }
 302
 303 static size_t cfg80211_gen_new_ie(const u8 *ie, size_t ielen,
 304                                   const u8 *subie, size_t subie_len,
 305                                   u8 *new_ie, size_t new_ie_len)
 306 {
 307         const struct element *non_inherit_elem, *parent, *sub;
 308         u8 *pos = new_ie;
 309         u8 id, ext_id;
 310         unsigned int match_len;
 311
 312         non_inherit_elem = cfg80211_find_ext_elem(WLAN_EID_EXT_NON_INHERITANCE,
 313                                                   subie, subie_len);
 314
 315         /* We copy the elements one by one from the parent to the generated
 316          * elements.
 317          * If they are not inherited (included in subie or in the non
 318          * inheritance element), then we copy all occurrences the first time
 319          * we see this element type.
 320          */
 321         for_each_element(parent, ie, ielen) {
 322                 if (parent->id == WLAN_EID_FRAGMENT)
 323                         continue;
 324
 325                 if (parent->id == WLAN_EID_EXTENSION) {
 326                         if (parent->datalen < 1)
 327                                 continue;
 328
 329                         id = WLAN_EID_EXTENSION;
 330                         ext_id = parent->data[0];
 331                         match_len = 1;
 332                 } else {
 333                         id = parent->id;
 334                         match_len = 0;
 335                 }
 336
 337                 /* Find first occurrence in subie */
 338                 sub = cfg80211_find_elem_match(id, subie, subie_len,
 339                                                &ext_id, match_len, 0);
 340
 341                 /* Copy from parent if not in subie and inherited */
 342                 if (!sub &&
 343                     cfg80211_is_element_inherited(parent, non_inherit_elem)) {
 344                         if (!cfg80211_copy_elem_with_frags(parent,
 345                                                            ie, ielen,
 346                                                            &pos, new_ie,
 347                                                            new_ie_len))
 348                                 return 0;
 349
 350                         continue;
 351                 }
 352
 353                 /* Already copied if an earlier element had the same type */
 354                 if (cfg80211_find_elem_match(id, ie, (u8 *)parent - ie,
 355                                              &ext_id, match_len, 0))
 356                         continue;
 357
 358                 /* Not inheriting, copy all similar elements from subie */
 359                 while (sub) {
 360                         if (!cfg80211_copy_elem_with_frags(sub,
 361                                                            subie, subie_len,
 362                                                            &pos, new_ie,
 363                                                            new_ie_len))
 364                                 return 0;
 365
 366                         sub = cfg80211_find_elem_match(id,
 367                                                        sub->data + sub->datalen,
 368                                                        subie_len + subie -
 369                                                        (sub->data +
 370                                                         sub->datalen),
 371                                                        &ext_id, match_len, 0);
 372                 }
 373         }
 374
 375         /* The above misses elements that are included in subie but not in the
 376          * parent, so do a pass over subie and append those.
 377          * Skip the non-tx BSSID caps and non-inheritance element.
 378          */
 379         for_each_element(sub, subie, subie_len) {
 380                 if (sub->id == WLAN_EID_NON_TX_BSSID_CAP)
 381                         continue;
 382
 383                 if (sub->id == WLAN_EID_FRAGMENT)
 384                         continue;
 385
 386                 if (sub->id == WLAN_EID_EXTENSION) {
 387                         if (sub->datalen < 1)
 388                                 continue;
 389
 390                         id = WLAN_EID_EXTENSION;
 391                         ext_id = sub->data[0];
 392                         match_len = 1;
 393
 394                         if (ext_id == WLAN_EID_EXT_NON_INHERITANCE)
 395                                 continue;
 396                 } else {
 397                         id = sub->id;
 398                         match_len = 0;
 399                 }
 400
 401                 /* Processed if one was included in the parent */
 402                 if (cfg80211_find_elem_match(id, ie, ielen,
 403                                              &ext_id, match_len, 0))
 404                         continue;
 405
 406                 if (!cfg80211_copy_elem_with_frags(sub, subie, subie_len,
 407                                                    &pos, new_ie, new_ie_len))
 408                         return 0;
 409         }
 410
 411         return pos - new_ie;
 412 }
 413
 414 static bool is_bss(struct cfg80211_bss *a, const u8 *bssid,
 415                    const u8 *ssid, size_t ssid_len)
 416 {
 417         const struct cfg80211_bss_ies *ies;
 418         const u8 *ssidie;
 419
 420         if (bssid && !ether_addr_equal(a->bssid, bssid))
 421                 return false;
 422
 423         if (!ssid)
 424                 return true;
 425
 426         ies = rcu_access_pointer(a->ies);
 427         if (!ies)
 428                 return false;
 429         ssidie = cfg80211_find_ie(WLAN_EID_SSID, ies->data, ies->len);
 430         if (!ssidie)
 431                 return false;
 432         if (ssidie[1] != ssid_len)
 433                 return false;
 434         return memcmp(ssidie + 2, ssid, ssid_len) == 0;
 435 }
 436
 437 static int
 438 cfg80211_add_nontrans_list(struct cfg80211_bss *trans_bss,
 439                            struct cfg80211_bss *nontrans_bss)
 440 {
 441         const u8 *ssid;
 442         size_t ssid_len;
 443         struct cfg80211_bss *bss = NULL;
 444
 445         rcu_read_lock();
 446         ssid = ieee80211_bss_get_ie(nontrans_bss, WLAN_EID_SSID);
 447         if (!ssid) {
 448                 rcu_read_unlock();
 449                 return -EINVAL;
 450         }
 451         ssid_len = ssid[1];
 452         ssid = ssid + 2;
 453
 454         /* check if nontrans_bss is in the list */
 455         list_for_each_entry(bss, &trans_bss->nontrans_list, nontrans_list) {
 456                 if (is_bss(bss, nontrans_bss->bssid, ssid, ssid_len)) {
 457                         rcu_read_unlock();
 458                         return 0;
 459                 }
 460         }
 461
 462         rcu_read_unlock();
 463
 464         /*
 465          * This is a bit weird - it's not on the list, but already on another
 466          * one! The only way that could happen is if there's some BSSID/SSID
 467          * shared by multiple APs in their multi-BSSID profiles, potentially
 468          * with hidden SSID mixed in ... ignore it.
 469          */
 470         if (!list_empty(&nontrans_bss->nontrans_list))
 471                 return -EINVAL;
 472
 473         /* add to the list */
 474         list_add_tail(&nontrans_bss->nontrans_list, &trans_bss->nontrans_list);
 475         return 0;
 476 }
 477
 478 static void __cfg80211_bss_expire(struct cfg80211_registered_device *rdev,
 479                                   unsigned long expire_time)
 480 {
 481         struct cfg80211_internal_bss *bss, *tmp;
 482         bool expired = false;
 483
 484         lockdep_assert_held(&rdev->bss_lock);
 485
 486         list_for_each_entry_safe(bss, tmp, &rdev->bss_list, list) {
 487                 if (atomic_read(&bss->hold))
 488                         continue;
 489                 if (!time_after(expire_time, bss->ts))
 490                         continue;
 491
 492                 if (__cfg80211_unlink_bss(rdev, bss))
 493                         expired = true;
 494         }
 495
 496         if (expired)
 497                 rdev->bss_generation++;
 498 }
 499
 500 static bool cfg80211_bss_expire_oldest(struct cfg80211_registered_device *rdev)
 501 {
 502         struct cfg80211_internal_bss *bss, *oldest = NULL;
 503         bool ret;
 504
 505         lockdep_assert_held(&rdev->bss_lock);
 506
 507         list_for_each_entry(bss, &rdev->bss_list, list) {
 508                 if (atomic_read(&bss->hold))
 509                         continue;
 510
 511                 if (!list_empty(&bss->hidden_list) &&
 512                     !bss->pub.hidden_beacon_bss)
 513                         continue;
 514
 515                 if (oldest && time_before(oldest->ts, bss->ts))
 516                         continue;
 517                 oldest = bss;
 518         }
 519
 520         if (WARN_ON(!oldest))
 521                 return false;
 522
 523         /*
 524          * The callers make sure to increase rdev->bss_generation if anything
 525          * gets removed (and a new entry added), so there's no need to also do
 526          * it here.
 527          */
 528
 529         ret = __cfg80211_unlink_bss(rdev, oldest);
 530         WARN_ON(!ret);
 531         return ret;
 532 }
 533
 534 static u8 cfg80211_parse_bss_param(u8 data,
 535                                    struct cfg80211_colocated_ap *coloc_ap)
 536 {
 537         coloc_ap->oct_recommended =
 538                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_OCT_RECOMMENDED);
 539         coloc_ap->same_ssid =
 540                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_SAME_SSID);
 541         coloc_ap->multi_bss =
 542                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_MULTI_BSSID);
 543         coloc_ap->transmitted_bssid =
 544                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_TRANSMITTED_BSSID);
 545         coloc_ap->unsolicited_probe =
 546                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_PROBE_ACTIVE);
 547         coloc_ap->colocated_ess =
 548                 u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_COLOC_ESS);
 549
 550         return u8_get_bits(data, IEEE80211_RNR_TBTT_PARAMS_COLOC_AP);
 551 }
 552
 553 static int cfg80211_calc_short_ssid(const struct cfg80211_bss_ies *ies,
 554                                     const struct element **elem, u32 *s_ssid)
 555 {
 556
 557         *elem = cfg80211_find_elem(WLAN_EID_SSID, ies->data, ies->len);
 558         if (!*elem || (*elem)->datalen > IEEE80211_MAX_SSID_LEN)
 559                 return -EINVAL;
 560
 561         *s_ssid = ~crc32_le(~0, (*elem)->data, (*elem)->datalen);
 562         return 0;
 563 }
 564
 565 static void cfg80211_free_coloc_ap_list(struct list_head *coloc_ap_list)
 566 {
 567         struct cfg80211_colocated_ap *ap, *tmp_ap;
 568
 569         list_for_each_entry_safe(ap, tmp_ap, coloc_ap_list, list) {
 570                 list_del(&ap->list);
 571                 kfree(ap);
 572         }
 573 }
 574
 575 static int cfg80211_parse_ap_info(struct cfg80211_colocated_ap *entry,
 576                                   const u8 *pos, u8 length,
 577                                   const struct element *ssid_elem,
 578                                   int s_ssid_tmp)
 579 {
 580         /* skip the TBTT offset */
 581         pos++;
 582
 583         memcpy(entry->bssid, pos, ETH_ALEN);
 584         pos += ETH_ALEN;
 585
 586         if (length == IEEE80211_TBTT_INFO_OFFSET_BSSID_SSSID_BSS_PARAM) {
 587                 memcpy(&entry->short_ssid, pos,
 588                        sizeof(entry->short_ssid));
 589                 entry->short_ssid_valid = true;
 590                 pos += 4;
 591         }
 592
 593         /* skip non colocated APs */
 594         if (!cfg80211_parse_bss_param(*pos, entry))
 595                 return -EINVAL;
 596         pos++;
 597
 598         if (length == IEEE80211_TBTT_INFO_OFFSET_BSSID_BSS_PARAM) {
 599                 /*
 600                  * no information about the short ssid. Consider the entry valid
 601                  * for now. It would later be dropped in case there are explicit
 602                  * SSIDs that need to be matched
 603                  */
 604                 if (!entry->same_ssid)
 605                         return 0;
 606         }
 607
 608         if (entry->same_ssid) {
 609                 entry->short_ssid = s_ssid_tmp;
 610                 entry->short_ssid_valid = true;
 611
 612                 /*
 613                  * This is safe because we validate datalen in
 614                  * cfg80211_parse_colocated_ap(), before calling this
 615                  * function.
 616                  */
 617                 memcpy(&entry->ssid, &ssid_elem->data,
 618                        ssid_elem->datalen);
 619                 entry->ssid_len = ssid_elem->datalen;
 620         }
 621         return 0;
 622 }
 623
 624 static int cfg80211_parse_colocated_ap(const struct cfg80211_bss_ies *ies,
 625                                        struct list_head *list)
 626 {
 627         struct ieee80211_neighbor_ap_info *ap_info;
 628         const struct element *elem, *ssid_elem;
 629         const u8 *pos, *end;
 630         u32 s_ssid_tmp;
 631         int n_coloc = 0, ret;
 632         LIST_HEAD(ap_list);
 633
 634         elem = cfg80211_find_elem(WLAN_EID_REDUCED_NEIGHBOR_REPORT, ies->data,
 635                                   ies->len);
 636         if (!elem)
 637                 return 0;
 638
 639         pos = elem->data;
 640         end = pos + elem->datalen;
 641
 642         ret = cfg80211_calc_short_ssid(ies, &ssid_elem, &s_ssid_tmp);
 643         if (ret)
 644                 return 0;
 645
 646         /* RNR IE may contain more than one NEIGHBOR_AP_INFO */
 647         while (pos + sizeof(*ap_info) <= end) {
 648                 enum nl80211_band band;
 649                 int freq;
 650                 u8 length, i, count;
 651
 652                 ap_info = (void *)pos;
 653                 count = u8_get_bits(ap_info->tbtt_info_hdr,
 654                                     IEEE80211_AP_INFO_TBTT_HDR_COUNT) + 1;
 655                 length = ap_info->tbtt_info_len;
 656
 657                 pos += sizeof(*ap_info);
 658
 659                 if (!ieee80211_operating_class_to_band(ap_info->op_class,
 660                                                        &band))
 661                         break;
 662
 663                 freq = ieee80211_channel_to_frequency(ap_info->channel, band);
 664
 665                 if (end - pos < count * length)
 666                         break;
 667
 668                 /*
 669                  * TBTT info must include bss param + BSSID +
 670                  * (short SSID or same_ssid bit to be set).
 671                  * ignore other options, and move to the
 672                  * next AP info
 673                  */
 674                 if (band != NL80211_BAND_6GHZ ||
 675                     (length != IEEE80211_TBTT_INFO_OFFSET_BSSID_BSS_PARAM &&
 676                      length < IEEE80211_TBTT_INFO_OFFSET_BSSID_SSSID_BSS_PARAM)) {
 677                         pos += count * length;
 678                         continue;
 679                 }
 680
 681                 for (i = 0; i < count; i++) {
 682                         struct cfg80211_colocated_ap *entry;
 683
 684                         entry = kzalloc(sizeof(*entry) + IEEE80211_MAX_SSID_LEN,
 685                                         GFP_ATOMIC);
 686
 687                         if (!entry)
 688                                 break;
 689
 690                         entry->center_freq = freq;
 691
 692                         if (!cfg80211_parse_ap_info(entry, pos, length,
 693                                                     ssid_elem, s_ssid_tmp)) {
 694                                 n_coloc++;
 695                                 list_add_tail(&entry->list, &ap_list);
 696                         } else {
 697                                 kfree(entry);
 698                         }
 699
 700                         pos += length;
 701                 }
 702         }
 703
 704         if (pos != end) {
 705                 cfg80211_free_coloc_ap_list(&ap_list);
 706                 return 0;
 707         }
 708
 709         list_splice_tail(&ap_list, list);
 710         return n_coloc;
 711 }
 712
 713 static  void cfg80211_scan_req_add_chan(struct cfg80211_scan_request *request,
 714                                         struct ieee80211_channel *chan,
 715                                         bool add_to_6ghz)
 716 {
 717         int i;
 718         u32 n_channels = request->n_channels;
 719         struct cfg80211_scan_6ghz_params *params =
 720                 &request->scan_6ghz_params[request->n_6ghz_params];
 721
 722         for (i = 0; i < n_channels; i++) {
 723                 if (request->channels[i] == chan) {
 724                         if (add_to_6ghz)
 725                                 params->channel_idx = i;
 726                         return;
 727                 }
 728         }
 729
 730         request->channels[n_channels] = chan;
 731         if (add_to_6ghz)
 732                 request->scan_6ghz_params[request->n_6ghz_params].channel_idx =
 733                         n_channels;
 734
 735         request->n_channels++;
 736 }
 737
 738 static bool cfg80211_find_ssid_match(struct cfg80211_colocated_ap *ap,
 739                                      struct cfg80211_scan_request *request)
 740 {
 741         int i;
 742         u32 s_ssid;
 743
 744         for (i = 0; i < request->n_ssids; i++) {
 745                 /* wildcard ssid in the scan request */
 746                 if (!request->ssids[i].ssid_len) {
 747                         if (ap->multi_bss && !ap->transmitted_bssid)
 748                                 continue;
 749
 750                         return true;
 751                 }
 752
 753                 if (ap->ssid_len &&
 754                     ap->ssid_len == request->ssids[i].ssid_len) {
 755                         if (!memcmp(request->ssids[i].ssid, ap->ssid,
 756                                     ap->ssid_len))
 757                                 return true;
 758                 } else if (ap->short_ssid_valid) {
 759                         s_ssid = ~crc32_le(~0, request->ssids[i].ssid,
 760                                            request->ssids[i].ssid_len);
 761
 762                         if (ap->short_ssid == s_ssid)
 763                                 return true;
 764                 }
 765         }
 766
 767         return false;
 768 }
 769
 770 static int cfg80211_scan_6ghz(struct cfg80211_registered_device *rdev)
 771 {
 772         u8 i;
 773         struct cfg80211_colocated_ap *ap;
 774         int n_channels, count = 0, err;
 775         struct cfg80211_scan_request *request, *rdev_req = rdev->scan_req;
 776         LIST_HEAD(coloc_ap_list);
 777         bool need_scan_psc = true;
 778         const struct ieee80211_sband_iftype_data *iftd;
 779
 780         rdev_req->scan_6ghz = true;
 781
 782         if (!rdev->wiphy.bands[NL80211_BAND_6GHZ])
 783                 return -EOPNOTSUPP;
 784
 785         iftd = ieee80211_get_sband_iftype_data(rdev->wiphy.bands[NL80211_BAND_6GHZ],
 786                                                rdev_req->wdev->iftype);
 787         if (!iftd || !iftd->he_cap.has_he)
 788                 return -EOPNOTSUPP;
 789
 790         n_channels = rdev->wiphy.bands[NL80211_BAND_6GHZ]->n_channels;
 791
 792         if (rdev_req->flags & NL80211_SCAN_FLAG_COLOCATED_6GHZ) {
 793                 struct cfg80211_internal_bss *intbss;
 794
 795                 spin_lock_bh(&rdev->bss_lock);
 796                 list_for_each_entry(intbss, &rdev->bss_list, list) {
 797                         struct cfg80211_bss *res = &intbss->pub;
 798                         const struct cfg80211_bss_ies *ies;
 799
 800                         ies = rcu_access_pointer(res->ies);
 801                         count += cfg80211_parse_colocated_ap(ies,
 802                                                              &coloc_ap_list);
 803                 }
 804                 spin_unlock_bh(&rdev->bss_lock);
 805         }
 806
 807         request = kzalloc(struct_size(request, channels, n_channels) +
 808                           sizeof(*request->scan_6ghz_params) * count +
 809                           sizeof(*request->ssids) * rdev_req->n_ssids,
 810                           GFP_KERNEL);
 811         if (!request) {
 812                 cfg80211_free_coloc_ap_list(&coloc_ap_list);
 813                 return -ENOMEM;
 814         }
 815
 816         *request = *rdev_req;
 817         request->n_channels = 0;
 818         request->scan_6ghz_params =
 819                 (void *)&request->channels[n_channels];
 820
 821         /*
 822          * PSC channels should not be scanned in case of direct scan with 1 SSID
 823          * and at least one of the reported co-located APs with same SSID
 824          * indicating that all APs in the same ESS are co-located
 825          */
 826         if (count && request->n_ssids == 1 && request->ssids[0].ssid_len) {
 827                 list_for_each_entry(ap, &coloc_ap_list, list) {
 828                         if (ap->colocated_ess &&
 829                             cfg80211_find_ssid_match(ap, request)) {
 830                                 need_scan_psc = false;
 831                                 break;
 832                         }
 833                 }
 834         }
 835
 836         /*
 837          * add to the scan request the channels that need to be scanned
 838          * regardless of the collocated APs (PSC channels or all channels
 839          * in case that NL80211_SCAN_FLAG_COLOCATED_6GHZ is not set)
 840          */
 841         for (i = 0; i < rdev_req->n_channels; i++) {
 842                 if (rdev_req->channels[i]->band == NL80211_BAND_6GHZ &&
 843                     ((need_scan_psc &&
 844                       cfg80211_channel_is_psc(rdev_req->channels[i])) ||
 845                      !(rdev_req->flags & NL80211_SCAN_FLAG_COLOCATED_6GHZ))) {
 846                         cfg80211_scan_req_add_chan(request,
 847                                                    rdev_req->channels[i],
 848                                                    false);
 849                 }
 850         }
 851
 852         if (!(rdev_req->flags & NL80211_SCAN_FLAG_COLOCATED_6GHZ))
 853                 goto skip;
 854
 855         list_for_each_entry(ap, &coloc_ap_list, list) {
 856                 bool found = false;
 857                 struct cfg80211_scan_6ghz_params *scan_6ghz_params =
 858                         &request->scan_6ghz_params[request->n_6ghz_params];
 859                 struct ieee80211_channel *chan =
 860                         ieee80211_get_channel(&rdev->wiphy, ap->center_freq);
 861
 862                 if (!chan || chan->flags & IEEE80211_CHAN_DISABLED)
 863                         continue;
 864
 865                 for (i = 0; i < rdev_req->n_channels; i++) {
 866                         if (rdev_req->channels[i] == chan)
 867                                 found = true;
 868                 }
 869
 870                 if (!found)
 871                         continue;
 872
 873                 if (request->n_ssids > 0 &&
 874                     !cfg80211_find_ssid_match(ap, request))
 875                         continue;
 876
 877                 if (!is_broadcast_ether_addr(request->bssid) &&
 878                     !ether_addr_equal(request->bssid, ap->bssid))
 879                         continue;
 880
 881                 if (!request->n_ssids && ap->multi_bss && !ap->transmitted_bssid)
 882                         continue;
 883
 884                 cfg80211_scan_req_add_chan(request, chan, true);
 885                 memcpy(scan_6ghz_params->bssid, ap->bssid, ETH_ALEN);
 886                 scan_6ghz_params->short_ssid = ap->short_ssid;
 887                 scan_6ghz_params->short_ssid_valid = ap->short_ssid_valid;
 888                 scan_6ghz_params->unsolicited_probe = ap->unsolicited_probe;
 889
 890                 /*
 891                  * If a PSC channel is added to the scan and 'need_scan_psc' is
 892                  * set to false, then all the APs that the scan logic is
 893                  * interested with on the channel are collocated and thus there
 894                  * is no need to perform the initial PSC channel listen.
 895                  */
 896                 if (cfg80211_channel_is_psc(chan) && !need_scan_psc)
 897                         scan_6ghz_params->psc_no_listen = true;
 898
 899                 request->n_6ghz_params++;
 900         }
 901
 902 skip:
 903         cfg80211_free_coloc_ap_list(&coloc_ap_list);
 904
 905         if (request->n_channels) {
 906                 struct cfg80211_scan_request *old = rdev->int_scan_req;
 907                 rdev->int_scan_req = request;
 908
 909                 /*
 910                  * Add the ssids from the parent scan request to the new scan
 911                  * request, so the driver would be able to use them in its
 912                  * probe requests to discover hidden APs on PSC channels.
 913                  */
 914                 request->ssids = (void *)&request->channels[request->n_channels];
 915                 request->n_ssids = rdev_req->n_ssids;
 916                 memcpy(request->ssids, rdev_req->ssids, sizeof(*request->ssids) *
 917                        request->n_ssids);
 918
 919                 /*
 920                  * If this scan follows a previous scan, save the scan start
 921                  * info from the first part of the scan
 922                  */
 923                 if (old)
 924                         rdev->int_scan_req->info = old->info;
 925
 926                 err = rdev_scan(rdev, request);
 927                 if (err) {
 928                         rdev->int_scan_req = old;
 929                         kfree(request);
 930                 } else {
 931                         kfree(old);
 932                 }
 933
 934                 return err;
 935         }
 936
 937         kfree(request);
 938         return -EINVAL;
 939 }
 940
 941 int cfg80211_scan(struct cfg80211_registered_device *rdev)
 942 {
 943         struct cfg80211_scan_request *request;
 944         struct cfg80211_scan_request *rdev_req = rdev->scan_req;
 945         u32 n_channels = 0, idx, i;
 946
 947         if (!(rdev->wiphy.flags & WIPHY_FLAG_SPLIT_SCAN_6GHZ))
 948                 return rdev_scan(rdev, rdev_req);
 949
 950         for (i = 0; i < rdev_req->n_channels; i++) {
 951                 if (rdev_req->channels[i]->band != NL80211_BAND_6GHZ)
 952                         n_channels++;
 953         }
 954
 955         if (!n_channels)
 956                 return cfg80211_scan_6ghz(rdev);
 957
 958         request = kzalloc(struct_size(request, channels, n_channels),
 959                           GFP_KERNEL);
 960         if (!request)
 961                 return -ENOMEM;
 962
 963         *request = *rdev_req;
 964         request->n_channels = n_channels;
 965
 966         for (i = idx = 0; i < rdev_req->n_channels; i++) {
 967                 if (rdev_req->channels[i]->band != NL80211_BAND_6GHZ)
 968                         request->channels[idx++] = rdev_req->channels[i];
 969         }
 970
 971         rdev_req->scan_6ghz = false;
 972         rdev->int_scan_req = request;
 973         return rdev_scan(rdev, request);
 974 }
 975
 976 void ___cfg80211_scan_done(struct cfg80211_registered_device *rdev,
 977                            bool send_message)
 978 {
 979         struct cfg80211_scan_request *request, *rdev_req;
 980         struct wireless_dev *wdev;
 981         struct sk_buff *msg;
 982 #ifdef CONFIG_CFG80211_WEXT
 983         union iwreq_data wrqu;
 984 #endif
 985
 986         lockdep_assert_held(&rdev->wiphy.mtx);
 987
 988         if (rdev->scan_msg) {
 989                 nl80211_send_scan_msg(rdev, rdev->scan_msg);
 990                 rdev->scan_msg = NULL;
 991                 return;
 992         }
 993
 994         rdev_req = rdev->scan_req;
 995         if (!rdev_req)
 996                 return;
 997
 998         wdev = rdev_req->wdev;
 999         request = rdev->int_scan_req ? rdev->int_scan_req : rdev_req;
1000
1001         if (wdev_running(wdev) &&
1002             (rdev->wiphy.flags & WIPHY_FLAG_SPLIT_SCAN_6GHZ) &&
1003             !rdev_req->scan_6ghz && !request->info.aborted &&
1004             !cfg80211_scan_6ghz(rdev))
1005                 return;
1006
1007         /*
1008          * This must be before sending the other events!
1009          * Otherwise, wpa_supplicant gets completely confused with
1010          * wext events.
1011          */
1012         if (wdev->netdev)
1013                 cfg80211_sme_scan_done(wdev->netdev);
1014
1015         if (!request->info.aborted &&
1016             request->flags & NL80211_SCAN_FLAG_FLUSH) {
1017                 /* flush entries from previous scans */
1018                 spin_lock_bh(&rdev->bss_lock);
1019                 __cfg80211_bss_expire(rdev, request->scan_start);
1020                 spin_unlock_bh(&rdev->bss_lock);
1021         }
1022
1023         msg = nl80211_build_scan_msg(rdev, wdev, request->info.aborted);
1024
1025 #ifdef CONFIG_CFG80211_WEXT
1026         if (wdev->netdev && !request->info.aborted) {
1027                 memset(&wrqu, 0, sizeof(wrqu));
1028
1029                 wireless_send_event(wdev->netdev, SIOCGIWSCAN, &wrqu, NULL);
1030         }
1031 #endif
1032
1033         dev_put(wdev->netdev);
1034
1035         kfree(rdev->int_scan_req);
1036         rdev->int_scan_req = NULL;
1037
1038         kfree(rdev->scan_req);
1039         rdev->scan_req = NULL;
1040
1041         if (!send_message)
1042                 rdev->scan_msg = msg;
1043         else
1044                 nl80211_send_scan_msg(rdev, msg);
1045 }
1046
1047 void __cfg80211_scan_done(struct work_struct *wk)
1048 {
1049         struct cfg80211_registered_device *rdev;
1050
1051         rdev = container_of(wk, struct cfg80211_registered_device,
1052                             scan_done_wk);
1053
1054         wiphy_lock(&rdev->wiphy);
1055         ___cfg80211_scan_done(rdev, true);
1056         wiphy_unlock(&rdev->wiphy);
1057 }
1058
1059 void cfg80211_scan_done(struct cfg80211_scan_request *request,
1060                         struct cfg80211_scan_info *info)
1061 {
1062         struct cfg80211_scan_info old_info = request->info;
1063
1064         trace_cfg80211_scan_done(request, info);
1065         WARN_ON(request != wiphy_to_rdev(request->wiphy)->scan_req &&
1066                 request != wiphy_to_rdev(request->wiphy)->int_scan_req);
1067
1068         request->info = *info;
1069
1070         /*
1071          * In case the scan is split, the scan_start_tsf and tsf_bssid should
1072          * be of the first part. In such a case old_info.scan_start_tsf should
1073          * be non zero.
1074          */
1075         if (request->scan_6ghz && old_info.scan_start_tsf) {
1076                 request->info.scan_start_tsf = old_info.scan_start_tsf;
1077                 memcpy(request->info.tsf_bssid, old_info.tsf_bssid,
1078                        sizeof(request->info.tsf_bssid));
1079         }
1080
1081         request->notified = true;
1082         queue_work(cfg80211_wq, &wiphy_to_rdev(request->wiphy)->scan_done_wk);
1083 }
1084 EXPORT_SYMBOL(cfg80211_scan_done);
1085
1086 void cfg80211_add_sched_scan_req(struct cfg80211_registered_device *rdev,
1087                                  struct cfg80211_sched_scan_request *req)
1088 {
1089         lockdep_assert_held(&rdev->wiphy.mtx);
1090
1091         list_add_rcu(&req->list, &rdev->sched_scan_req_list);
1092 }
1093
1094 static void cfg80211_del_sched_scan_req(struct cfg80211_registered_device *rdev,
1095                                         struct cfg80211_sched_scan_request *req)
1096 {
1097         lockdep_assert_held(&rdev->wiphy.mtx);
1098
1099         list_del_rcu(&req->list);
1100         kfree_rcu(req, rcu_head);
1101 }
1102
1103 static struct cfg80211_sched_scan_request *
1104 cfg80211_find_sched_scan_req(struct cfg80211_registered_device *rdev, u64 reqid)
1105 {
1106         struct cfg80211_sched_scan_request *pos;
1107
1108         list_for_each_entry_rcu(pos, &rdev->sched_scan_req_list, list,
1109                                 lockdep_is_held(&rdev->wiphy.mtx)) {
1110                 if (pos->reqid == reqid)
1111                         return pos;
1112         }
1113         return NULL;
1114 }
1115
1116 /*
1117  * Determines if a scheduled scan request can be handled. When a legacy
1118  * scheduled scan is running no other scheduled scan is allowed regardless
1119  * whether the request is for legacy or multi-support scan. When a multi-support
1120  * scheduled scan is running a request for legacy scan is not allowed. In this
1121  * case a request for multi-support scan can be handled if resources are
1122  * available, ie. struct wiphy::max_sched_scan_reqs limit is not yet reached.
1123  */
1124 int cfg80211_sched_scan_req_possible(struct cfg80211_registered_device *rdev,
1125                                      bool want_multi)
1126 {
1127         struct cfg80211_sched_scan_request *pos;
1128         int i = 0;
1129
1130         list_for_each_entry(pos, &rdev->sched_scan_req_list, list) {
1131                 /* request id zero means legacy in progress */
1132                 if (!i && !pos->reqid)
1133                         return -EINPROGRESS;
1134                 i++;
1135         }
1136
1137         if (i) {
1138                 /* no legacy allowed when multi request(s) are active */
1139                 if (!want_multi)
1140                         return -EINPROGRESS;
1141
1142                 /* resource limit reached */
1143                 if (i == rdev->wiphy.max_sched_scan_reqs)
1144                         return -ENOSPC;
1145         }
1146         return 0;
1147 }
1148
1149 void cfg80211_sched_scan_results_wk(struct work_struct *work)
1150 {
1151         struct cfg80211_registered_device *rdev;
1152         struct cfg80211_sched_scan_request *req, *tmp;
1153
1154         rdev = container_of(work, struct cfg80211_registered_device,
1155                            sched_scan_res_wk);
1156
1157         wiphy_lock(&rdev->wiphy);
1158         list_for_each_entry_safe(req, tmp, &rdev->sched_scan_req_list, list) {
1159                 if (req->report_results) {
1160                         req->report_results = false;
1161                         if (req->flags & NL80211_SCAN_FLAG_FLUSH) {
1162                                 /* flush entries from previous scans */
1163                                 spin_lock_bh(&rdev->bss_lock);
1164                                 __cfg80211_bss_expire(rdev, req->scan_start);
1165                                 spin_unlock_bh(&rdev->bss_lock);
1166                                 req->scan_start = jiffies;
1167                         }
1168                         nl80211_send_sched_scan(req,
1169                                                 NL80211_CMD_SCHED_SCAN_RESULTS);
1170                 }
1171         }
1172         wiphy_unlock(&rdev->wiphy);
1173 }
1174
1175 void cfg80211_sched_scan_results(struct wiphy *wiphy, u64 reqid)
1176 {
1177         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1178         struct cfg80211_sched_scan_request *request;
1179
1180         trace_cfg80211_sched_scan_results(wiphy, reqid);
1181         /* ignore if we're not scanning */
1182
1183         rcu_read_lock();
1184         request = cfg80211_find_sched_scan_req(rdev, reqid);
1185         if (request) {
1186                 request->report_results = true;
1187                 queue_work(cfg80211_wq, &rdev->sched_scan_res_wk);
1188         }
1189         rcu_read_unlock();
1190 }
1191 EXPORT_SYMBOL(cfg80211_sched_scan_results);
1192
1193 void cfg80211_sched_scan_stopped_locked(struct wiphy *wiphy, u64 reqid)
1194 {
1195         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1196
1197         lockdep_assert_held(&wiphy->mtx);
1198
1199         trace_cfg80211_sched_scan_stopped(wiphy, reqid);
1200
1201         __cfg80211_stop_sched_scan(rdev, reqid, true);
1202 }
1203 EXPORT_SYMBOL(cfg80211_sched_scan_stopped_locked);
1204
1205 void cfg80211_sched_scan_stopped(struct wiphy *wiphy, u64 reqid)
1206 {
1207         wiphy_lock(wiphy);
1208         cfg80211_sched_scan_stopped_locked(wiphy, reqid);
1209         wiphy_unlock(wiphy);
1210 }
1211 EXPORT_SYMBOL(cfg80211_sched_scan_stopped);
1212
1213 int cfg80211_stop_sched_scan_req(struct cfg80211_registered_device *rdev,
1214                                  struct cfg80211_sched_scan_request *req,
1215                                  bool driver_initiated)
1216 {
1217         lockdep_assert_held(&rdev->wiphy.mtx);
1218
1219         if (!driver_initiated) {
1220                 int err = rdev_sched_scan_stop(rdev, req->dev, req->reqid);
1221                 if (err)
1222                         return err;
1223         }
1224
1225         nl80211_send_sched_scan(req, NL80211_CMD_SCHED_SCAN_STOPPED);
1226
1227         cfg80211_del_sched_scan_req(rdev, req);
1228
1229         return 0;
1230 }
1231
1232 int __cfg80211_stop_sched_scan(struct cfg80211_registered_device *rdev,
1233                                u64 reqid, bool driver_initiated)
1234 {
1235         struct cfg80211_sched_scan_request *sched_scan_req;
1236
1237         lockdep_assert_held(&rdev->wiphy.mtx);
1238
1239         sched_scan_req = cfg80211_find_sched_scan_req(rdev, reqid);
1240         if (!sched_scan_req)
1241                 return -ENOENT;
1242
1243         return cfg80211_stop_sched_scan_req(rdev, sched_scan_req,
1244                                             driver_initiated);
1245 }
1246
1247 void cfg80211_bss_age(struct cfg80211_registered_device *rdev,
1248                       unsigned long age_secs)
1249 {
1250         struct cfg80211_internal_bss *bss;
1251         unsigned long age_jiffies = msecs_to_jiffies(age_secs * MSEC_PER_SEC);
1252
1253         spin_lock_bh(&rdev->bss_lock);
1254         list_for_each_entry(bss, &rdev->bss_list, list)
1255                 bss->ts -= age_jiffies;
1256         spin_unlock_bh(&rdev->bss_lock);
1257 }
1258
1259 void cfg80211_bss_expire(struct cfg80211_registered_device *rdev)
1260 {
1261         __cfg80211_bss_expire(rdev, jiffies - IEEE80211_SCAN_RESULT_EXPIRE);
1262 }
1263
1264 void cfg80211_bss_flush(struct wiphy *wiphy)
1265 {
1266         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1267
1268         spin_lock_bh(&rdev->bss_lock);
1269         __cfg80211_bss_expire(rdev, jiffies);
1270         spin_unlock_bh(&rdev->bss_lock);
1271 }
1272 EXPORT_SYMBOL(cfg80211_bss_flush);
1273
1274 const struct element *
1275 cfg80211_find_elem_match(u8 eid, const u8 *ies, unsigned int len,
1276                          const u8 *match, unsigned int match_len,
1277                          unsigned int match_offset)
1278 {
1279         const struct element *elem;
1280
1281         for_each_element_id(elem, eid, ies, len) {
1282                 if (elem->datalen >= match_offset + match_len &&
1283                     !memcmp(elem->data + match_offset, match, match_len))
1284                         return elem;
1285         }
1286
1287         return NULL;
1288 }
1289 EXPORT_SYMBOL(cfg80211_find_elem_match);
1290
1291 const struct element *cfg80211_find_vendor_elem(unsigned int oui, int oui_type,
1292                                                 const u8 *ies,
1293                                                 unsigned int len)
1294 {
1295         const struct element *elem;
1296         u8 match[] = { oui >> 16, oui >> 8, oui, oui_type };
1297         int match_len = (oui_type < 0) ? 3 : sizeof(match);
1298
1299         if (WARN_ON(oui_type > 0xff))
1300                 return NULL;
1301
1302         elem = cfg80211_find_elem_match(WLAN_EID_VENDOR_SPECIFIC, ies, len,
1303                                         match, match_len, 0);
1304
1305         if (!elem || elem->datalen < 4)
1306                 return NULL;
1307
1308         return elem;
1309 }
1310 EXPORT_SYMBOL(cfg80211_find_vendor_elem);
1311
1312 /**
1313  * enum bss_compare_mode - BSS compare mode
1314  * @BSS_CMP_REGULAR: regular compare mode (for insertion and normal find)
1315  * @BSS_CMP_HIDE_ZLEN: find hidden SSID with zero-length mode
1316  * @BSS_CMP_HIDE_NUL: find hidden SSID with NUL-ed out mode
1317  */
1318 enum bss_compare_mode {
1319         BSS_CMP_REGULAR,
1320         BSS_CMP_HIDE_ZLEN,
1321         BSS_CMP_HIDE_NUL,
1322 };
1323
1324 static int cmp_bss(struct cfg80211_bss *a,
1325                    struct cfg80211_bss *b,
1326                    enum bss_compare_mode mode)
1327 {
1328         const struct cfg80211_bss_ies *a_ies, *b_ies;
1329         const u8 *ie1 = NULL;
1330         const u8 *ie2 = NULL;
1331         int i, r;
1332
1333         if (a->channel != b->channel)
1334                 return b->channel->center_freq - a->channel->center_freq;
1335
1336         a_ies = rcu_access_pointer(a->ies);
1337         if (!a_ies)
1338                 return -1;
1339         b_ies = rcu_access_pointer(b->ies);
1340         if (!b_ies)
1341                 return 1;
1342
1343         if (WLAN_CAPABILITY_IS_STA_BSS(a->capability))
1344                 ie1 = cfg80211_find_ie(WLAN_EID_MESH_ID,
1345                                        a_ies->data, a_ies->len);
1346         if (WLAN_CAPABILITY_IS_STA_BSS(b->capability))
1347                 ie2 = cfg80211_find_ie(WLAN_EID_MESH_ID,
1348                                        b_ies->data, b_ies->len);
1349         if (ie1 && ie2) {
1350                 int mesh_id_cmp;
1351
1352                 if (ie1[1] == ie2[1])
1353                         mesh_id_cmp = memcmp(ie1 + 2, ie2 + 2, ie1[1]);
1354                 else
1355                         mesh_id_cmp = ie2[1] - ie1[1];
1356
1357                 ie1 = cfg80211_find_ie(WLAN_EID_MESH_CONFIG,
1358                                        a_ies->data, a_ies->len);
1359                 ie2 = cfg80211_find_ie(WLAN_EID_MESH_CONFIG,
1360                                        b_ies->data, b_ies->len);
1361                 if (ie1 && ie2) {
1362                         if (mesh_id_cmp)
1363                                 return mesh_id_cmp;
1364                         if (ie1[1] != ie2[1])
1365                                 return ie2[1] - ie1[1];
1366                         return memcmp(ie1 + 2, ie2 + 2, ie1[1]);
1367                 }
1368         }
1369
1370         r = memcmp(a->bssid, b->bssid, sizeof(a->bssid));
1371         if (r)
1372                 return r;
1373
1374         ie1 = cfg80211_find_ie(WLAN_EID_SSID, a_ies->data, a_ies->len);
1375         ie2 = cfg80211_find_ie(WLAN_EID_SSID, b_ies->data, b_ies->len);
1376
1377         if (!ie1 && !ie2)
1378                 return 0;
1379
1380         /*
1381          * Note that with "hide_ssid", the function returns a match if
1382          * the already-present BSS ("b") is a hidden SSID beacon for
1383          * the new BSS ("a").
1384          */
1385
1386         /* sort missing IE before (left of) present IE */
1387         if (!ie1)
1388                 return -1;
1389         if (!ie2)
1390                 return 1;
1391
1392         switch (mode) {
1393         case BSS_CMP_HIDE_ZLEN:
1394                 /*
1395                  * In ZLEN mode we assume the BSS entry we're
1396                  * looking for has a zero-length SSID. So if
1397                  * the one we're looking at right now has that,
1398                  * return 0. Otherwise, return the difference
1399                  * in length, but since we're looking for the
1400                  * 0-length it's really equivalent to returning
1401                  * the length of the one we're looking at.
1402                  *
1403                  * No content comparison is needed as we assume
1404                  * the content length is zero.
1405                  */
1406                 return ie2[1];
1407         case BSS_CMP_REGULAR:
1408         default:
1409                 /* sort by length first, then by contents */
1410                 if (ie1[1] != ie2[1])
1411                         return ie2[1] - ie1[1];
1412                 return memcmp(ie1 + 2, ie2 + 2, ie1[1]);
1413         case BSS_CMP_HIDE_NUL:
1414                 if (ie1[1] != ie2[1])
1415                         return ie2[1] - ie1[1];
1416                 /* this is equivalent to memcmp(zeroes, ie2 + 2, len) */
1417                 for (i = 0; i < ie2[1]; i++)
1418                         if (ie2[i + 2])
1419                                 return -1;
1420                 return 0;
1421         }
1422 }
1423
1424 static bool cfg80211_bss_type_match(u16 capability,
1425                                     enum nl80211_band band,
1426                                     enum ieee80211_bss_type bss_type)
1427 {
1428         bool ret = true;
1429         u16 mask, val;
1430
1431         if (bss_type == IEEE80211_BSS_TYPE_ANY)
1432                 return ret;
1433
1434         if (band == NL80211_BAND_60GHZ) {
1435                 mask = WLAN_CAPABILITY_DMG_TYPE_MASK;
1436                 switch (bss_type) {
1437                 case IEEE80211_BSS_TYPE_ESS:
1438                         val = WLAN_CAPABILITY_DMG_TYPE_AP;
1439                         break;
1440                 case IEEE80211_BSS_TYPE_PBSS:
1441                         val = WLAN_CAPABILITY_DMG_TYPE_PBSS;
1442                         break;
1443                 case IEEE80211_BSS_TYPE_IBSS:
1444                         val = WLAN_CAPABILITY_DMG_TYPE_IBSS;
1445                         break;
1446                 default:
1447                         return false;
1448                 }
1449         } else {
1450                 mask = WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_IBSS;
1451                 switch (bss_type) {
1452                 case IEEE80211_BSS_TYPE_ESS:
1453                         val = WLAN_CAPABILITY_ESS;
1454                         break;
1455                 case IEEE80211_BSS_TYPE_IBSS:
1456                         val = WLAN_CAPABILITY_IBSS;
1457                         break;
1458                 case IEEE80211_BSS_TYPE_MBSS:
1459                         val = 0;
1460                         break;
1461                 default:
1462                         return false;
1463                 }
1464         }
1465
1466         ret = ((capability & mask) == val);
1467         return ret;
1468 }
1469
1470 /* Returned bss is reference counted and must be cleaned up appropriately. */
1471 struct cfg80211_bss *cfg80211_get_bss(struct wiphy *wiphy,
1472                                       struct ieee80211_channel *channel,
1473                                       const u8 *bssid,
1474                                       const u8 *ssid, size_t ssid_len,
1475                                       enum ieee80211_bss_type bss_type,
1476                                       enum ieee80211_privacy privacy)
1477 {
1478         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1479         struct cfg80211_internal_bss *bss, *res = NULL;
1480         unsigned long now = jiffies;
1481         int bss_privacy;
1482
1483         trace_cfg80211_get_bss(wiphy, channel, bssid, ssid, ssid_len, bss_type,
1484                                privacy);
1485
1486         spin_lock_bh(&rdev->bss_lock);
1487
1488         list_for_each_entry(bss, &rdev->bss_list, list) {
1489                 if (!cfg80211_bss_type_match(bss->pub.capability,
1490                                              bss->pub.channel->band, bss_type))
1491                         continue;
1492
1493                 bss_privacy = (bss->pub.capability & WLAN_CAPABILITY_PRIVACY);
1494                 if ((privacy == IEEE80211_PRIVACY_ON && !bss_privacy) ||
1495                     (privacy == IEEE80211_PRIVACY_OFF && bss_privacy))
1496                         continue;
1497                 if (channel && bss->pub.channel != channel)
1498                         continue;
1499                 if (!is_valid_ether_addr(bss->pub.bssid))
1500                         continue;
1501                 /* Don't get expired BSS structs */
1502                 if (time_after(now, bss->ts + IEEE80211_SCAN_RESULT_EXPIRE) &&
1503                     !atomic_read(&bss->hold))
1504                         continue;
1505                 if (is_bss(&bss->pub, bssid, ssid, ssid_len)) {
1506                         res = bss;
1507                         bss_ref_get(rdev, res);
1508                         break;
1509                 }
1510         }
1511
1512         spin_unlock_bh(&rdev->bss_lock);
1513         if (!res)
1514                 return NULL;
1515         trace_cfg80211_return_bss(&res->pub);
1516         return &res->pub;
1517 }
1518 EXPORT_SYMBOL(cfg80211_get_bss);
1519
1520 static void rb_insert_bss(struct cfg80211_registered_device *rdev,
1521                           struct cfg80211_internal_bss *bss)
1522 {
1523         struct rb_node **p = &rdev->bss_tree.rb_node;
1524         struct rb_node *parent = NULL;
1525         struct cfg80211_internal_bss *tbss;
1526         int cmp;
1527
1528         while (*p) {
1529                 parent = *p;
1530                 tbss = rb_entry(parent, struct cfg80211_internal_bss, rbn);
1531
1532                 cmp = cmp_bss(&bss->pub, &tbss->pub, BSS_CMP_REGULAR);
1533
1534                 if (WARN_ON(!cmp)) {
1535                         /* will sort of leak this BSS */
1536                         return;
1537                 }
1538
1539                 if (cmp < 0)
1540                         p = &(*p)->rb_left;
1541                 else
1542                         p = &(*p)->rb_right;
1543         }
1544
1545         rb_link_node(&bss->rbn, parent, p);
1546         rb_insert_color(&bss->rbn, &rdev->bss_tree);
1547 }
1548
1549 static struct cfg80211_internal_bss *
1550 rb_find_bss(struct cfg80211_registered_device *rdev,
1551             struct cfg80211_internal_bss *res,
1552             enum bss_compare_mode mode)
1553 {
1554         struct rb_node *n = rdev->bss_tree.rb_node;
1555         struct cfg80211_internal_bss *bss;
1556         int r;
1557
1558         while (n) {
1559                 bss = rb_entry(n, struct cfg80211_internal_bss, rbn);
1560                 r = cmp_bss(&res->pub, &bss->pub, mode);
1561
1562                 if (r == 0)
1563                         return bss;
1564                 else if (r < 0)
1565                         n = n->rb_left;
1566                 else
1567                         n = n->rb_right;
1568         }
1569
1570         return NULL;
1571 }
1572
1573 static bool cfg80211_combine_bsses(struct cfg80211_registered_device *rdev,
1574                                    struct cfg80211_internal_bss *new)
1575 {
1576         const struct cfg80211_bss_ies *ies;
1577         struct cfg80211_internal_bss *bss;
1578         const u8 *ie;
1579         int i, ssidlen;
1580         u8 fold = 0;
1581         u32 n_entries = 0;
1582
1583         ies = rcu_access_pointer(new->pub.beacon_ies);
1584         if (WARN_ON(!ies))
1585                 return false;
1586
1587         ie = cfg80211_find_ie(WLAN_EID_SSID, ies->data, ies->len);
1588         if (!ie) {
1589                 /* nothing to do */
1590                 return true;
1591         }
1592
1593         ssidlen = ie[1];
1594         for (i = 0; i < ssidlen; i++)
1595                 fold |= ie[2 + i];
1596
1597         if (fold) {
1598                 /* not a hidden SSID */
1599                 return true;
1600         }
1601
1602         /* This is the bad part ... */
1603
1604         list_for_each_entry(bss, &rdev->bss_list, list) {
1605                 /*
1606                  * we're iterating all the entries anyway, so take the
1607                  * opportunity to validate the list length accounting
1608                  */
1609                 n_entries++;
1610
1611                 if (!ether_addr_equal(bss->pub.bssid, new->pub.bssid))
1612                         continue;
1613                 if (bss->pub.channel != new->pub.channel)
1614                         continue;
1615                 if (bss->pub.scan_width != new->pub.scan_width)
1616                         continue;
1617                 if (rcu_access_pointer(bss->pub.beacon_ies))
1618                         continue;
1619                 ies = rcu_access_pointer(bss->pub.ies);
1620                 if (!ies)
1621                         continue;
1622                 ie = cfg80211_find_ie(WLAN_EID_SSID, ies->data, ies->len);
1623                 if (!ie)
1624                         continue;
1625                 if (ssidlen && ie[1] != ssidlen)
1626                         continue;
1627                 if (WARN_ON_ONCE(bss->pub.hidden_beacon_bss))
1628                         continue;
1629                 if (WARN_ON_ONCE(!list_empty(&bss->hidden_list)))
1630                         list_del(&bss->hidden_list);
1631                 /* combine them */
1632                 list_add(&bss->hidden_list, &new->hidden_list);
1633                 bss->pub.hidden_beacon_bss = &new->pub;
1634                 new->refcount += bss->refcount;
1635                 rcu_assign_pointer(bss->pub.beacon_ies,
1636                                    new->pub.beacon_ies);
1637         }
1638
1639         WARN_ONCE(n_entries != rdev->bss_entries,
1640                   "rdev bss entries[%d]/list[len:%d] corruption\n",
1641                   rdev->bss_entries, n_entries);
1642
1643         return true;
1644 }
1645
1646 struct cfg80211_non_tx_bss {
1647         struct cfg80211_bss *tx_bss;
1648         u8 max_bssid_indicator;
1649         u8 bssid_index;
1650 };
1651
1652 static void cfg80211_update_hidden_bsses(struct cfg80211_internal_bss *known,
1653                                          const struct cfg80211_bss_ies *new_ies,
1654                                          const struct cfg80211_bss_ies *old_ies)
1655 {
1656         struct cfg80211_internal_bss *bss;
1657
1658         /* Assign beacon IEs to all sub entries */
1659         list_for_each_entry(bss, &known->hidden_list, hidden_list) {
1660                 const struct cfg80211_bss_ies *ies;
1661
1662                 ies = rcu_access_pointer(bss->pub.beacon_ies);
1663                 WARN_ON(ies != old_ies);
1664
1665                 rcu_assign_pointer(bss->pub.beacon_ies, new_ies);
1666         }
1667 }
1668
1669 static bool
1670 cfg80211_update_known_bss(struct cfg80211_registered_device *rdev,
1671                           struct cfg80211_internal_bss *known,
1672                           struct cfg80211_internal_bss *new,
1673                           bool signal_valid)
1674 {
1675         lockdep_assert_held(&rdev->bss_lock);
1676
1677         /* Update IEs */
1678         if (rcu_access_pointer(new->pub.proberesp_ies)) {
1679                 const struct cfg80211_bss_ies *old;
1680
1681                 old = rcu_access_pointer(known->pub.proberesp_ies);
1682
1683                 rcu_assign_pointer(known->pub.proberesp_ies,
1684                                    new->pub.proberesp_ies);
1685                 /* Override possible earlier Beacon frame IEs */
1686                 rcu_assign_pointer(known->pub.ies,
1687                                    new->pub.proberesp_ies);
1688                 if (old)
1689                         kfree_rcu((struct cfg80211_bss_ies *)old, rcu_head);
1690         } else if (rcu_access_pointer(new->pub.beacon_ies)) {
1691                 const struct cfg80211_bss_ies *old;
1692
1693                 if (known->pub.hidden_beacon_bss &&
1694                     !list_empty(&known->hidden_list)) {
1695                         const struct cfg80211_bss_ies *f;
1696
1697                         /* The known BSS struct is one of the probe
1698                          * response members of a group, but we're
1699                          * receiving a beacon (beacon_ies in the new
1700                          * bss is used). This can only mean that the
1701                          * AP changed its beacon from not having an
1702                          * SSID to showing it, which is confusing so
1703                          * drop this information.
1704                          */
1705
1706                         f = rcu_access_pointer(new->pub.beacon_ies);
1707                         kfree_rcu((struct cfg80211_bss_ies *)f, rcu_head);
1708                         return false;
1709                 }
1710
1711                 old = rcu_access_pointer(known->pub.beacon_ies);
1712
1713                 rcu_assign_pointer(known->pub.beacon_ies, new->pub.beacon_ies);
1714
1715                 /* Override IEs if they were from a beacon before */
1716                 if (old == rcu_access_pointer(known->pub.ies))
1717                         rcu_assign_pointer(known->pub.ies, new->pub.beacon_ies);
1718
1719                 cfg80211_update_hidden_bsses(known,
1720                                              rcu_access_pointer(new->pub.beacon_ies),
1721                                              old);
1722
1723                 if (old)
1724                         kfree_rcu((struct cfg80211_bss_ies *)old, rcu_head);
1725         }
1726
1727         known->pub.beacon_interval = new->pub.beacon_interval;
1728
1729         /* don't update the signal if beacon was heard on
1730          * adjacent channel.
1731          */
1732         if (signal_valid)
1733                 known->pub.signal = new->pub.signal;
1734         known->pub.capability = new->pub.capability;
1735         known->ts = new->ts;
1736         known->ts_boottime = new->ts_boottime;
1737         known->parent_tsf = new->parent_tsf;
1738         known->pub.chains = new->pub.chains;
1739         memcpy(known->pub.chain_signal, new->pub.chain_signal,
1740                IEEE80211_MAX_CHAINS);
1741         ether_addr_copy(known->parent_bssid, new->parent_bssid);
1742         known->pub.max_bssid_indicator = new->pub.max_bssid_indicator;
1743         known->pub.bssid_index = new->pub.bssid_index;
1744
1745         return true;
1746 }
1747
1748 /* Returned bss is reference counted and must be cleaned up appropriately. */
1749 struct cfg80211_internal_bss *
1750 cfg80211_bss_update(struct cfg80211_registered_device *rdev,
1751                     struct cfg80211_internal_bss *tmp,
1752                     bool signal_valid, unsigned long ts)
1753 {
1754         struct cfg80211_internal_bss *found = NULL;
1755
1756         if (WARN_ON(!tmp->pub.channel))
1757                 return NULL;
1758
1759         tmp->ts = ts;
1760
1761         spin_lock_bh(&rdev->bss_lock);
1762
1763         if (WARN_ON(!rcu_access_pointer(tmp->pub.ies))) {
1764                 spin_unlock_bh(&rdev->bss_lock);
1765                 return NULL;
1766         }
1767
1768         found = rb_find_bss(rdev, tmp, BSS_CMP_REGULAR);
1769
1770         if (found) {
1771                 if (!cfg80211_update_known_bss(rdev, found, tmp, signal_valid))
1772                         goto drop;
1773         } else {
1774                 struct cfg80211_internal_bss *new;
1775                 struct cfg80211_internal_bss *hidden;
1776                 struct cfg80211_bss_ies *ies;
1777
1778                 /*
1779                  * create a copy -- the "res" variable that is passed in
1780                  * is allocated on the stack since it's not needed in the
1781                  * more common case of an update
1782                  */
1783                 new = kzalloc(sizeof(*new) + rdev->wiphy.bss_priv_size,
1784                               GFP_ATOMIC);
1785                 if (!new) {
1786                         ies = (void *)rcu_dereference(tmp->pub.beacon_ies);
1787                         if (ies)
1788                                 kfree_rcu(ies, rcu_head);
1789                         ies = (void *)rcu_dereference(tmp->pub.proberesp_ies);
1790                         if (ies)
1791                                 kfree_rcu(ies, rcu_head);
1792                         goto drop;
1793                 }
1794                 memcpy(new, tmp, sizeof(*new));
1795                 new->refcount = 1;
1796                 INIT_LIST_HEAD(&new->hidden_list);
1797                 INIT_LIST_HEAD(&new->pub.nontrans_list);
1798                 /* we'll set this later if it was non-NULL */
1799                 new->pub.transmitted_bss = NULL;
1800
1801                 if (rcu_access_pointer(tmp->pub.proberesp_ies)) {
1802                         hidden = rb_find_bss(rdev, tmp, BSS_CMP_HIDE_ZLEN);
1803                         if (!hidden)
1804                                 hidden = rb_find_bss(rdev, tmp,
1805                                                      BSS_CMP_HIDE_NUL);
1806                         if (hidden) {
1807                                 new->pub.hidden_beacon_bss = &hidden->pub;
1808                                 list_add(&new->hidden_list,
1809                                          &hidden->hidden_list);
1810                                 hidden->refcount++;
1811                                 rcu_assign_pointer(new->pub.beacon_ies,
1812                                                    hidden->pub.beacon_ies);
1813                         }
1814                 } else {
1815                         /*
1816                          * Ok so we found a beacon, and don't have an entry. If
1817                          * it's a beacon with hidden SSID, we might be in for an
1818                          * expensive search for any probe responses that should
1819                          * be grouped with this beacon for updates ...
1820                          */
1821                         if (!cfg80211_combine_bsses(rdev, new)) {
1822                                 bss_ref_put(rdev, new);
1823                                 goto drop;
1824                         }
1825                 }
1826
1827                 if (rdev->bss_entries >= bss_entries_limit &&
1828                     !cfg80211_bss_expire_oldest(rdev)) {
1829                         bss_ref_put(rdev, new);
1830                         goto drop;
1831                 }
1832
1833                 /* This must be before the call to bss_ref_get */
1834                 if (tmp->pub.transmitted_bss) {
1835                         struct cfg80211_internal_bss *pbss =
1836                                 container_of(tmp->pub.transmitted_bss,
1837                                              struct cfg80211_internal_bss,
1838                                              pub);
1839
1840                         new->pub.transmitted_bss = tmp->pub.transmitted_bss;
1841                         bss_ref_get(rdev, pbss);
1842                 }
1843
1844                 list_add_tail(&new->list, &rdev->bss_list);
1845                 rdev->bss_entries++;
1846                 rb_insert_bss(rdev, new);
1847                 found = new;
1848         }
1849
1850         rdev->bss_generation++;
1851         bss_ref_get(rdev, found);
1852         spin_unlock_bh(&rdev->bss_lock);
1853
1854         return found;
1855  drop:
1856         spin_unlock_bh(&rdev->bss_lock);
1857         return NULL;
1858 }
1859
1860 /*
1861  * Update RX channel information based on the available frame payload
1862  * information. This is mainly for the 2.4 GHz band where frames can be received
1863  * from neighboring channels and the Beacon frames use the DSSS Parameter Set
1864  * element to indicate the current (transmitting) channel, but this might also
1865  * be needed on other bands if RX frequency does not match with the actual
1866  * operating channel of a BSS.
1867  */
1868 static struct ieee80211_channel *
1869 cfg80211_get_bss_channel(struct wiphy *wiphy, const u8 *ie, size_t ielen,
1870                          struct ieee80211_channel *channel,
1871                          enum nl80211_bss_scan_width scan_width)
1872 {
1873         const u8 *tmp;
1874         u32 freq;
1875         int channel_number = -1;
1876         struct ieee80211_channel *alt_channel;
1877
1878         if (channel->band == NL80211_BAND_S1GHZ) {
1879                 tmp = cfg80211_find_ie(WLAN_EID_S1G_OPERATION, ie, ielen);
1880                 if (tmp && tmp[1] >= sizeof(struct ieee80211_s1g_oper_ie)) {
1881                         struct ieee80211_s1g_oper_ie *s1gop = (void *)(tmp + 2);
1882
1883                         channel_number = s1gop->primary_ch;
1884                 }
1885         } else {
1886                 tmp = cfg80211_find_ie(WLAN_EID_DS_PARAMS, ie, ielen);
1887                 if (tmp && tmp[1] == 1) {
1888                         channel_number = tmp[2];
1889                 } else {
1890                         tmp = cfg80211_find_ie(WLAN_EID_HT_OPERATION, ie, ielen);
1891                         if (tmp && tmp[1] >= sizeof(struct ieee80211_ht_operation)) {
1892                                 struct ieee80211_ht_operation *htop = (void *)(tmp + 2);
1893
1894                                 channel_number = htop->primary_chan;
1895                         }
1896                 }
1897         }
1898
1899         if (channel_number < 0) {
1900                 /* No channel information in frame payload */
1901                 return channel;
1902         }
1903
1904         freq = ieee80211_channel_to_freq_khz(channel_number, channel->band);
1905         alt_channel = ieee80211_get_channel_khz(wiphy, freq);
1906         if (!alt_channel) {
1907                 if (channel->band == NL80211_BAND_2GHZ) {
1908                         /*
1909                          * Better not allow unexpected channels when that could
1910                          * be going beyond the 1-11 range (e.g., discovering
1911                          * BSS on channel 12 when radio is configured for
1912                          * channel 11.
1913                          */
1914                         return NULL;
1915                 }
1916
1917                 /* No match for the payload channel number - ignore it */
1918                 return channel;
1919         }
1920
1921         if (scan_width == NL80211_BSS_CHAN_WIDTH_10 ||
1922             scan_width == NL80211_BSS_CHAN_WIDTH_5) {
1923                 /*
1924                  * Ignore channel number in 5 and 10 MHz channels where there
1925                  * may not be an n:1 or 1:n mapping between frequencies and
1926                  * channel numbers.
1927                  */
1928                 return channel;
1929         }
1930
1931         /*
1932          * Use the channel determined through the payload channel number
1933          * instead of the RX channel reported by the driver.
1934          */
1935         if (alt_channel->flags & IEEE80211_CHAN_DISABLED)
1936                 return NULL;
1937         return alt_channel;
1938 }
1939
1940 /* Returned bss is reference counted and must be cleaned up appropriately. */
1941 static struct cfg80211_bss *
1942 cfg80211_inform_single_bss_data(struct wiphy *wiphy,
1943                                 struct cfg80211_inform_bss *data,
1944                                 enum cfg80211_bss_frame_type ftype,
1945                                 const u8 *bssid, u64 tsf, u16 capability,
1946                                 u16 beacon_interval, const u8 *ie, size_t ielen,
1947                                 struct cfg80211_non_tx_bss *non_tx_data,
1948                                 gfp_t gfp)
1949 {
1950         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
1951         struct cfg80211_bss_ies *ies;
1952         struct ieee80211_channel *channel;
1953         struct cfg80211_internal_bss tmp = {}, *res;
1954         int bss_type;
1955         bool signal_valid;
1956         unsigned long ts;
1957
1958         if (WARN_ON(!wiphy))
1959                 return NULL;
1960
1961         if (WARN_ON(wiphy->signal_type == CFG80211_SIGNAL_TYPE_UNSPEC &&
1962                     (data->signal < 0 || data->signal > 100)))
1963                 return NULL;
1964
1965         channel = cfg80211_get_bss_channel(wiphy, ie, ielen, data->chan,
1966                                            data->scan_width);
1967         if (!channel)
1968                 return NULL;
1969
1970         memcpy(tmp.pub.bssid, bssid, ETH_ALEN);
1971         tmp.pub.channel = channel;
1972         tmp.pub.scan_width = data->scan_width;
1973         tmp.pub.signal = data->signal;
1974         tmp.pub.beacon_interval = beacon_interval;
1975         tmp.pub.capability = capability;
1976         tmp.ts_boottime = data->boottime_ns;
1977         tmp.parent_tsf = data->parent_tsf;
1978         ether_addr_copy(tmp.parent_bssid, data->parent_bssid);
1979
1980         if (non_tx_data) {
1981                 tmp.pub.transmitted_bss = non_tx_data->tx_bss;
1982                 ts = bss_from_pub(non_tx_data->tx_bss)->ts;
1983                 tmp.pub.bssid_index = non_tx_data->bssid_index;
1984                 tmp.pub.max_bssid_indicator = non_tx_data->max_bssid_indicator;
1985         } else {
1986                 ts = jiffies;
1987         }
1988
1989         /*
1990          * If we do not know here whether the IEs are from a Beacon or Probe
1991          * Response frame, we need to pick one of the options and only use it
1992          * with the driver that does not provide the full Beacon/Probe Response
1993          * frame. Use Beacon frame pointer to avoid indicating that this should
1994          * override the IEs pointer should we have received an earlier
1995          * indication of Probe Response data.
1996          */
1997         ies = kzalloc(sizeof(*ies) + ielen, gfp);
1998         if (!ies)
1999                 return NULL;
2000         ies->len = ielen;
2001         ies->tsf = tsf;
2002         ies->from_beacon = false;
2003         memcpy(ies->data, ie, ielen);
2004
2005         switch (ftype) {
2006         case CFG80211_BSS_FTYPE_BEACON:
2007                 ies->from_beacon = true;
2008                 fallthrough;
2009         case CFG80211_BSS_FTYPE_UNKNOWN:
2010                 rcu_assign_pointer(tmp.pub.beacon_ies, ies);
2011                 break;
2012         case CFG80211_BSS_FTYPE_PRESP:
2013                 rcu_assign_pointer(tmp.pub.proberesp_ies, ies);
2014                 break;
2015         }
2016         rcu_assign_pointer(tmp.pub.ies, ies);
2017
2018         signal_valid = data->chan == channel;
2019         res = cfg80211_bss_update(wiphy_to_rdev(wiphy), &tmp, signal_valid, ts);
2020         if (!res)
2021                 return NULL;
2022
2023         if (channel->band == NL80211_BAND_60GHZ) {
2024                 bss_type = res->pub.capability & WLAN_CAPABILITY_DMG_TYPE_MASK;
2025                 if (bss_type == WLAN_CAPABILITY_DMG_TYPE_AP ||
2026                     bss_type == WLAN_CAPABILITY_DMG_TYPE_PBSS)
2027                         regulatory_hint_found_beacon(wiphy, channel, gfp);
2028         } else {
2029                 if (res->pub.capability & WLAN_CAPABILITY_ESS)
2030                         regulatory_hint_found_beacon(wiphy, channel, gfp);
2031         }
2032
2033         if (non_tx_data) {
2034                 /* this is a nontransmitting bss, we need to add it to
2035                  * transmitting bss' list if it is not there
2036                  */
2037                 spin_lock_bh(&rdev->bss_lock);
2038                 if (cfg80211_add_nontrans_list(non_tx_data->tx_bss,
2039                                                &res->pub)) {
2040                         if (__cfg80211_unlink_bss(rdev, res)) {
2041                                 rdev->bss_generation++;
2042                                 res = NULL;
2043                         }
2044                 }
2045                 spin_unlock_bh(&rdev->bss_lock);
2046
2047                 if (!res)
2048                         return NULL;
2049         }
2050
2051         trace_cfg80211_return_bss(&res->pub);
2052         /* cfg80211_bss_update gives us a referenced result */
2053         return &res->pub;
2054 }
2055
2056 static const struct element
2057 *cfg80211_get_profile_continuation(const u8 *ie, size_t ielen,
2058                                    const struct element *mbssid_elem,
2059                                    const struct element *sub_elem)
2060 {
2061         const u8 *mbssid_end = mbssid_elem->data + mbssid_elem->datalen;
2062         const struct element *next_mbssid;
2063         const struct element *next_sub;
2064
2065         next_mbssid = cfg80211_find_elem(WLAN_EID_MULTIPLE_BSSID,
2066                                          mbssid_end,
2067                                          ielen - (mbssid_end - ie));
2068
2069         /*
2070          * If it is not the last subelement in current MBSSID IE or there isn't
2071          * a next MBSSID IE - profile is complete.
2072         */
2073         if ((sub_elem->data + sub_elem->datalen < mbssid_end - 1) ||
2074             !next_mbssid)
2075                 return NULL;
2076
2077         /* For any length error, just return NULL */
2078
2079         if (next_mbssid->datalen < 4)
2080                 return NULL;
2081
2082         next_sub = (void *)&next_mbssid->data[1];
2083
2084         if (next_mbssid->data + next_mbssid->datalen <
2085             next_sub->data + next_sub->datalen)
2086                 return NULL;
2087
2088         if (next_sub->id != 0 || next_sub->datalen < 2)
2089                 return NULL;
2090
2091         /*
2092          * Check if the first element in the next sub element is a start
2093          * of a new profile
2094          */
2095         return next_sub->data[0] == WLAN_EID_NON_TX_BSSID_CAP ?
2096                NULL : next_mbssid;
2097 }
2098
2099 size_t cfg80211_merge_profile(const u8 *ie, size_t ielen,
2100                               const struct element *mbssid_elem,
2101                               const struct element *sub_elem,
2102                               u8 *merged_ie, size_t max_copy_len)
2103 {
2104         size_t copied_len = sub_elem->datalen;
2105         const struct element *next_mbssid;
2106
2107         if (sub_elem->datalen > max_copy_len)
2108                 return 0;
2109
2110         memcpy(merged_ie, sub_elem->data, sub_elem->datalen);
2111
2112         while ((next_mbssid = cfg80211_get_profile_continuation(ie, ielen,
2113                                                                 mbssid_elem,
2114                                                                 sub_elem))) {
2115                 const struct element *next_sub = (void *)&next_mbssid->data[1];
2116
2117                 if (copied_len + next_sub->datalen > max_copy_len)
2118                         break;
2119                 memcpy(merged_ie + copied_len, next_sub->data,
2120                        next_sub->datalen);
2121                 copied_len += next_sub->datalen;
2122         }
2123
2124         return copied_len;
2125 }
2126 EXPORT_SYMBOL(cfg80211_merge_profile);
2127
2128 static void cfg80211_parse_mbssid_data(struct wiphy *wiphy,
2129                                        struct cfg80211_inform_bss *data,
2130                                        enum cfg80211_bss_frame_type ftype,
2131                                        const u8 *bssid, u64 tsf,
2132                                        u16 beacon_interval, const u8 *ie,
2133                                        size_t ielen,
2134                                        struct cfg80211_non_tx_bss *non_tx_data,
2135                                        gfp_t gfp)
2136 {
2137         const u8 *mbssid_index_ie;
2138         const struct element *elem, *sub;
2139         size_t new_ie_len;
2140         u8 new_bssid[ETH_ALEN];
2141         u8 *new_ie, *profile;
2142         u64 seen_indices = 0;
2143         u16 capability;
2144         struct cfg80211_bss *bss;
2145
2146         if (!non_tx_data)
2147                 return;
2148         if (!cfg80211_find_ie(WLAN_EID_MULTIPLE_BSSID, ie, ielen))
2149                 return;
2150         if (!wiphy->support_mbssid)
2151                 return;
2152         if (wiphy->support_only_he_mbssid &&
2153             !cfg80211_find_ext_ie(WLAN_EID_EXT_HE_CAPABILITY, ie, ielen))
2154                 return;
2155
2156         new_ie = kmalloc(IEEE80211_MAX_DATA_LEN, gfp);
2157         if (!new_ie)
2158                 return;
2159
2160         profile = kmalloc(ielen, gfp);
2161         if (!profile)
2162                 goto out;
2163
2164         for_each_element_id(elem, WLAN_EID_MULTIPLE_BSSID, ie, ielen) {
2165                 if (elem->datalen < 4)
2166                         continue;
2167                 if (elem->data[0] < 1 || (int)elem->data[0] > 8)
2168                         continue;
2169                 for_each_element(sub, elem->data + 1, elem->datalen - 1) {
2170                         u8 profile_len;
2171
2172                         if (sub->id != 0 || sub->datalen < 4) {
2173                                 /* not a valid BSS profile */
2174                                 continue;
2175                         }
2176
2177                         if (sub->data[0] != WLAN_EID_NON_TX_BSSID_CAP ||
2178                             sub->data[1] != 2) {
2179                                 /* The first element within the Nontransmitted
2180                                  * BSSID Profile is not the Nontransmitted
2181                                  * BSSID Capability element.
2182                                  */
2183                                 continue;
2184                         }
2185
2186                         memset(profile, 0, ielen);
2187                         profile_len = cfg80211_merge_profile(ie, ielen,
2188                                                              elem,
2189                                                              sub,
2190                                                              profile,
2191                                                              ielen);
2192
2193                         /* found a Nontransmitted BSSID Profile */
2194                         mbssid_index_ie = cfg80211_find_ie
2195                                 (WLAN_EID_MULTI_BSSID_IDX,
2196                                  profile, profile_len);
2197                         if (!mbssid_index_ie || mbssid_index_ie[1] < 1 ||
2198                             mbssid_index_ie[2] == 0 ||
2199                             mbssid_index_ie[2] > 46) {
2200                                 /* No valid Multiple BSSID-Index element */
2201                                 continue;
2202                         }
2203
2204                         if (seen_indices & BIT_ULL(mbssid_index_ie[2]))
2205                                 /* We don't support legacy split of a profile */
2206                                 net_dbg_ratelimited("Partial info for BSSID index %d\n",
2207                                                     mbssid_index_ie[2]);
2208
2209                         seen_indices |= BIT_ULL(mbssid_index_ie[2]);
2210
2211                         non_tx_data->bssid_index = mbssid_index_ie[2];
2212                         non_tx_data->max_bssid_indicator = elem->data[0];
2213
2214                         cfg80211_gen_new_bssid(bssid,
2215                                                non_tx_data->max_bssid_indicator,
2216                                                non_tx_data->bssid_index,
2217                                                new_bssid);
2218                         memset(new_ie, 0, IEEE80211_MAX_DATA_LEN);
2219                         new_ie_len = cfg80211_gen_new_ie(ie, ielen,
2220                                                          profile,
2221                                                          profile_len, new_ie,
2222                                                          IEEE80211_MAX_DATA_LEN);
2223                         if (!new_ie_len)
2224                                 continue;
2225
2226                         capability = get_unaligned_le16(profile + 2);
2227                         bss = cfg80211_inform_single_bss_data(wiphy, data,
2228                                                               ftype,
2229                                                               new_bssid, tsf,
2230                                                               capability,
2231                                                               beacon_interval,
2232                                                               new_ie,
2233                                                               new_ie_len,
2234                                                               non_tx_data,
2235                                                               gfp);
2236                         if (!bss)
2237                                 break;
2238                         cfg80211_put_bss(wiphy, bss);
2239                 }
2240         }
2241
2242 out:
2243         kfree(new_ie);
2244         kfree(profile);
2245 }
2246
2247 struct cfg80211_bss *
2248 cfg80211_inform_bss_data(struct wiphy *wiphy,
2249                          struct cfg80211_inform_bss *data,
2250                          enum cfg80211_bss_frame_type ftype,
2251                          const u8 *bssid, u64 tsf, u16 capability,
2252                          u16 beacon_interval, const u8 *ie, size_t ielen,
2253                          gfp_t gfp)
2254 {
2255         struct cfg80211_bss *res;
2256         struct cfg80211_non_tx_bss non_tx_data;
2257
2258         res = cfg80211_inform_single_bss_data(wiphy, data, ftype, bssid, tsf,
2259                                               capability, beacon_interval, ie,
2260                                               ielen, NULL, gfp);
2261         if (!res)
2262                 return NULL;
2263         non_tx_data.tx_bss = res;
2264         cfg80211_parse_mbssid_data(wiphy, data, ftype, bssid, tsf,
2265                                    beacon_interval, ie, ielen, &non_tx_data,
2266                                    gfp);
2267         return res;
2268 }
2269 EXPORT_SYMBOL(cfg80211_inform_bss_data);
2270
2271 static void
2272 cfg80211_parse_mbssid_frame_data(struct wiphy *wiphy,
2273                                  struct cfg80211_inform_bss *data,
2274                                  struct ieee80211_mgmt *mgmt, size_t len,
2275                                  struct cfg80211_non_tx_bss *non_tx_data,
2276                                  gfp_t gfp)
2277 {
2278         enum cfg80211_bss_frame_type ftype;
2279         const u8 *ie = mgmt->u.probe_resp.variable;
2280         size_t ielen = len - offsetof(struct ieee80211_mgmt,
2281                                       u.probe_resp.variable);
2282
2283         ftype = ieee80211_is_beacon(mgmt->frame_control) ?
2284                 CFG80211_BSS_FTYPE_BEACON : CFG80211_BSS_FTYPE_PRESP;
2285
2286         cfg80211_parse_mbssid_data(wiphy, data, ftype, mgmt->bssid,
2287                                    le64_to_cpu(mgmt->u.probe_resp.timestamp),
2288                                    le16_to_cpu(mgmt->u.probe_resp.beacon_int),
2289                                    ie, ielen, non_tx_data, gfp);
2290 }
2291
2292 static void
2293 cfg80211_update_notlisted_nontrans(struct wiphy *wiphy,
2294                                    struct cfg80211_bss *nontrans_bss,
2295                                    struct ieee80211_mgmt *mgmt, size_t len)
2296 {
2297         u8 *ie, *new_ie, *pos;
2298         const u8 *nontrans_ssid, *trans_ssid, *mbssid;
2299         size_t ielen = len - offsetof(struct ieee80211_mgmt,
2300                                       u.probe_resp.variable);
2301         size_t new_ie_len;
2302         struct cfg80211_bss_ies *new_ies;
2303         const struct cfg80211_bss_ies *old;
2304         size_t cpy_len;
2305
2306         lockdep_assert_held(&wiphy_to_rdev(wiphy)->bss_lock);
2307
2308         ie = mgmt->u.probe_resp.variable;
2309
2310         new_ie_len = ielen;
2311         trans_ssid = cfg80211_find_ie(WLAN_EID_SSID, ie, ielen);
2312         if (!trans_ssid)
2313                 return;
2314         new_ie_len -= trans_ssid[1];
2315         mbssid = cfg80211_find_ie(WLAN_EID_MULTIPLE_BSSID, ie, ielen);
2316         /*
2317          * It's not valid to have the MBSSID element before SSID
2318          * ignore if that happens - the code below assumes it is
2319          * after (while copying things inbetween).
2320          */
2321         if (!mbssid || mbssid < trans_ssid)
2322                 return;
2323         new_ie_len -= mbssid[1];
2324
2325         nontrans_ssid = ieee80211_bss_get_ie(nontrans_bss, WLAN_EID_SSID);
2326         if (!nontrans_ssid)
2327                 return;
2328
2329         new_ie_len += nontrans_ssid[1];
2330
2331         /* generate new ie for nontrans BSS
2332          * 1. replace SSID with nontrans BSS' SSID
2333          * 2. skip MBSSID IE
2334          */
2335         new_ie = kzalloc(new_ie_len, GFP_ATOMIC);
2336         if (!new_ie)
2337                 return;
2338
2339         new_ies = kzalloc(sizeof(*new_ies) + new_ie_len, GFP_ATOMIC);
2340         if (!new_ies)
2341                 goto out_free;
2342
2343         pos = new_ie;
2344
2345         /* copy the nontransmitted SSID */
2346         cpy_len = nontrans_ssid[1] + 2;
2347         memcpy(pos, nontrans_ssid, cpy_len);
2348         pos += cpy_len;
2349         /* copy the IEs between SSID and MBSSID */
2350         cpy_len = trans_ssid[1] + 2;
2351         memcpy(pos, (trans_ssid + cpy_len), (mbssid - (trans_ssid + cpy_len)));
2352         pos += (mbssid - (trans_ssid + cpy_len));
2353         /* copy the IEs after MBSSID */
2354         cpy_len = mbssid[1] + 2;
2355         memcpy(pos, mbssid + cpy_len, ((ie + ielen) - (mbssid + cpy_len)));
2356
2357         /* update ie */
2358         new_ies->len = new_ie_len;
2359         new_ies->tsf = le64_to_cpu(mgmt->u.probe_resp.timestamp);
2360         new_ies->from_beacon = ieee80211_is_beacon(mgmt->frame_control);
2361         memcpy(new_ies->data, new_ie, new_ie_len);
2362         if (ieee80211_is_probe_resp(mgmt->frame_control)) {
2363                 old = rcu_access_pointer(nontrans_bss->proberesp_ies);
2364                 rcu_assign_pointer(nontrans_bss->proberesp_ies, new_ies);
2365                 rcu_assign_pointer(nontrans_bss->ies, new_ies);
2366                 if (old)
2367                         kfree_rcu((struct cfg80211_bss_ies *)old, rcu_head);
2368         } else {
2369                 old = rcu_access_pointer(nontrans_bss->beacon_ies);
2370                 rcu_assign_pointer(nontrans_bss->beacon_ies, new_ies);
2371                 cfg80211_update_hidden_bsses(bss_from_pub(nontrans_bss),
2372                                              new_ies, old);
2373                 rcu_assign_pointer(nontrans_bss->ies, new_ies);
2374                 if (old)
2375                         kfree_rcu((struct cfg80211_bss_ies *)old, rcu_head);
2376         }
2377
2378 out_free:
2379         kfree(new_ie);
2380 }
2381
2382 /* cfg80211_inform_bss_width_frame helper */
2383 static struct cfg80211_bss *
2384 cfg80211_inform_single_bss_frame_data(struct wiphy *wiphy,
2385                                       struct cfg80211_inform_bss *data,
2386                                       struct ieee80211_mgmt *mgmt, size_t len,
2387                                       gfp_t gfp)
2388 {
2389         struct cfg80211_internal_bss tmp = {}, *res;
2390         struct cfg80211_bss_ies *ies;
2391         struct ieee80211_channel *channel;
2392         bool signal_valid;
2393         struct ieee80211_ext *ext = NULL;
2394         u8 *bssid, *variable;
2395         u16 capability, beacon_int;
2396         size_t ielen, min_hdr_len = offsetof(struct ieee80211_mgmt,
2397                                              u.probe_resp.variable);
2398         int bss_type;
2399
2400         BUILD_BUG_ON(offsetof(struct ieee80211_mgmt, u.probe_resp.variable) !=
2401                         offsetof(struct ieee80211_mgmt, u.beacon.variable));
2402
2403         trace_cfg80211_inform_bss_frame(wiphy, data, mgmt, len);
2404
2405         if (WARN_ON(!mgmt))
2406                 return NULL;
2407
2408         if (WARN_ON(!wiphy))
2409                 return NULL;
2410
2411         if (WARN_ON(wiphy->signal_type == CFG80211_SIGNAL_TYPE_UNSPEC &&
2412                     (data->signal < 0 || data->signal > 100)))
2413                 return NULL;
2414
2415         if (ieee80211_is_s1g_beacon(mgmt->frame_control)) {
2416                 ext = (void *) mgmt;
2417                 min_hdr_len = offsetof(struct ieee80211_ext, u.s1g_beacon);
2418                 if (ieee80211_is_s1g_short_beacon(mgmt->frame_control))
2419                         min_hdr_len = offsetof(struct ieee80211_ext,
2420                                                u.s1g_short_beacon.variable);
2421         }
2422
2423         if (WARN_ON(len < min_hdr_len))
2424                 return NULL;
2425
2426         ielen = len - min_hdr_len;
2427         variable = mgmt->u.probe_resp.variable;
2428         if (ext) {
2429                 if (ieee80211_is_s1g_short_beacon(mgmt->frame_control))
2430                         variable = ext->u.s1g_short_beacon.variable;
2431                 else
2432                         variable = ext->u.s1g_beacon.variable;
2433         }
2434
2435         channel = cfg80211_get_bss_channel(wiphy, variable,
2436                                            ielen, data->chan, data->scan_width);
2437         if (!channel)
2438                 return NULL;
2439
2440         if (ext) {
2441                 const struct ieee80211_s1g_bcn_compat_ie *compat;
2442                 const struct element *elem;
2443
2444                 elem = cfg80211_find_elem(WLAN_EID_S1G_BCN_COMPAT,
2445                                           variable, ielen);
2446                 if (!elem)
2447                         return NULL;
2448                 if (elem->datalen < sizeof(*compat))
2449                         return NULL;
2450                 compat = (void *)elem->data;
2451                 bssid = ext->u.s1g_beacon.sa;
2452                 capability = le16_to_cpu(compat->compat_info);
2453                 beacon_int = le16_to_cpu(compat->beacon_int);
2454         } else {
2455                 bssid = mgmt->bssid;
2456                 beacon_int = le16_to_cpu(mgmt->u.probe_resp.beacon_int);
2457                 capability = le16_to_cpu(mgmt->u.probe_resp.capab_info);
2458         }
2459
2460         ies = kzalloc(sizeof(*ies) + ielen, gfp);
2461         if (!ies)
2462                 return NULL;
2463         ies->len = ielen;
2464         ies->tsf = le64_to_cpu(mgmt->u.probe_resp.timestamp);
2465         ies->from_beacon = ieee80211_is_beacon(mgmt->frame_control) ||
2466                            ieee80211_is_s1g_beacon(mgmt->frame_control);
2467         memcpy(ies->data, variable, ielen);
2468
2469         if (ieee80211_is_probe_resp(mgmt->frame_control))
2470                 rcu_assign_pointer(tmp.pub.proberesp_ies, ies);
2471         else
2472                 rcu_assign_pointer(tmp.pub.beacon_ies, ies);
2473         rcu_assign_pointer(tmp.pub.ies, ies);
2474
2475         memcpy(tmp.pub.bssid, bssid, ETH_ALEN);
2476         tmp.pub.beacon_interval = beacon_int;
2477         tmp.pub.capability = capability;
2478         tmp.pub.channel = channel;
2479         tmp.pub.scan_width = data->scan_width;
2480         tmp.pub.signal = data->signal;
2481         tmp.ts_boottime = data->boottime_ns;
2482         tmp.parent_tsf = data->parent_tsf;
2483         tmp.pub.chains = data->chains;
2484         memcpy(tmp.pub.chain_signal, data->chain_signal, IEEE80211_MAX_CHAINS);
2485         ether_addr_copy(tmp.parent_bssid, data->parent_bssid);
2486
2487         signal_valid = data->chan == channel;
2488         res = cfg80211_bss_update(wiphy_to_rdev(wiphy), &tmp, signal_valid,
2489                                   jiffies);
2490         if (!res)
2491                 return NULL;
2492
2493         if (channel->band == NL80211_BAND_60GHZ) {
2494                 bss_type = res->pub.capability & WLAN_CAPABILITY_DMG_TYPE_MASK;
2495                 if (bss_type == WLAN_CAPABILITY_DMG_TYPE_AP ||
2496                     bss_type == WLAN_CAPABILITY_DMG_TYPE_PBSS)
2497                         regulatory_hint_found_beacon(wiphy, channel, gfp);
2498         } else {
2499                 if (res->pub.capability & WLAN_CAPABILITY_ESS)
2500                         regulatory_hint_found_beacon(wiphy, channel, gfp);
2501         }
2502
2503         trace_cfg80211_return_bss(&res->pub);
2504         /* cfg80211_bss_update gives us a referenced result */
2505         return &res->pub;
2506 }
2507
2508 struct cfg80211_bss *
2509 cfg80211_inform_bss_frame_data(struct wiphy *wiphy,
2510                                struct cfg80211_inform_bss *data,
2511                                struct ieee80211_mgmt *mgmt, size_t len,
2512                                gfp_t gfp)
2513 {
2514         struct cfg80211_bss *res, *tmp_bss;
2515         const u8 *ie = mgmt->u.probe_resp.variable;
2516         const struct cfg80211_bss_ies *ies1, *ies2;
2517         size_t ielen = len - offsetof(struct ieee80211_mgmt,
2518                                       u.probe_resp.variable);
2519         struct cfg80211_non_tx_bss non_tx_data = {};
2520
2521         res = cfg80211_inform_single_bss_frame_data(wiphy, data, mgmt,
2522                                                     len, gfp);
2523
2524         /* don't do any further MBSSID handling for S1G */
2525         if (ieee80211_is_s1g_beacon(mgmt->frame_control))
2526                 return res;
2527
2528         if (!res || !wiphy->support_mbssid ||
2529             !cfg80211_find_ie(WLAN_EID_MULTIPLE_BSSID, ie, ielen))
2530                 return res;
2531         if (wiphy->support_only_he_mbssid &&
2532             !cfg80211_find_ext_ie(WLAN_EID_EXT_HE_CAPABILITY, ie, ielen))
2533                 return res;
2534
2535         non_tx_data.tx_bss = res;
2536         /* process each non-transmitting bss */
2537         cfg80211_parse_mbssid_frame_data(wiphy, data, mgmt, len,
2538                                          &non_tx_data, gfp);
2539
2540         spin_lock_bh(&wiphy_to_rdev(wiphy)->bss_lock);
2541
2542         /* check if the res has other nontransmitting bss which is not
2543          * in MBSSID IE
2544          */
2545         ies1 = rcu_access_pointer(res->ies);
2546
2547         /* go through nontrans_list, if the timestamp of the BSS is
2548          * earlier than the timestamp of the transmitting BSS then
2549          * update it
2550          */
2551         list_for_each_entry(tmp_bss, &res->nontrans_list,
2552                             nontrans_list) {
2553                 ies2 = rcu_access_pointer(tmp_bss->ies);
2554                 if (ies2->tsf < ies1->tsf)
2555                         cfg80211_update_notlisted_nontrans(wiphy, tmp_bss,
2556                                                            mgmt, len);
2557         }
2558         spin_unlock_bh(&wiphy_to_rdev(wiphy)->bss_lock);
2559
2560         return res;
2561 }
2562 EXPORT_SYMBOL(cfg80211_inform_bss_frame_data);
2563
2564 void cfg80211_ref_bss(struct wiphy *wiphy, struct cfg80211_bss *pub)
2565 {
2566         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
2567         struct cfg80211_internal_bss *bss;
2568
2569         if (!pub)
2570                 return;
2571
2572         bss = container_of(pub, struct cfg80211_internal_bss, pub);
2573
2574         spin_lock_bh(&rdev->bss_lock);
2575         bss_ref_get(rdev, bss);
2576         spin_unlock_bh(&rdev->bss_lock);
2577 }
2578 EXPORT_SYMBOL(cfg80211_ref_bss);
2579
2580 void cfg80211_put_bss(struct wiphy *wiphy, struct cfg80211_bss *pub)
2581 {
2582         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
2583         struct cfg80211_internal_bss *bss;
2584
2585         if (!pub)
2586                 return;
2587
2588         bss = container_of(pub, struct cfg80211_internal_bss, pub);
2589
2590         spin_lock_bh(&rdev->bss_lock);
2591         bss_ref_put(rdev, bss);
2592         spin_unlock_bh(&rdev->bss_lock);
2593 }
2594 EXPORT_SYMBOL(cfg80211_put_bss);
2595
2596 void cfg80211_unlink_bss(struct wiphy *wiphy, struct cfg80211_bss *pub)
2597 {
2598         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
2599         struct cfg80211_internal_bss *bss, *tmp1;
2600         struct cfg80211_bss *nontrans_bss, *tmp;
2601
2602         if (WARN_ON(!pub))
2603                 return;
2604
2605         bss = container_of(pub, struct cfg80211_internal_bss, pub);
2606
2607         spin_lock_bh(&rdev->bss_lock);
2608         if (list_empty(&bss->list))
2609                 goto out;
2610
2611         list_for_each_entry_safe(nontrans_bss, tmp,
2612                                  &pub->nontrans_list,
2613                                  nontrans_list) {
2614                 tmp1 = container_of(nontrans_bss,
2615                                     struct cfg80211_internal_bss, pub);
2616                 if (__cfg80211_unlink_bss(rdev, tmp1))
2617                         rdev->bss_generation++;
2618         }
2619
2620         if (__cfg80211_unlink_bss(rdev, bss))
2621                 rdev->bss_generation++;
2622 out:
2623         spin_unlock_bh(&rdev->bss_lock);
2624 }
2625 EXPORT_SYMBOL(cfg80211_unlink_bss);
2626
2627 void cfg80211_bss_iter(struct wiphy *wiphy,
2628                        struct cfg80211_chan_def *chandef,
2629                        void (*iter)(struct wiphy *wiphy,
2630                                     struct cfg80211_bss *bss,
2631                                     void *data),
2632                        void *iter_data)
2633 {
2634         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
2635         struct cfg80211_internal_bss *bss;
2636
2637         spin_lock_bh(&rdev->bss_lock);
2638
2639         list_for_each_entry(bss, &rdev->bss_list, list) {
2640                 if (!chandef || cfg80211_is_sub_chan(chandef, bss->pub.channel))
2641                         iter(wiphy, &bss->pub, iter_data);
2642         }
2643
2644         spin_unlock_bh(&rdev->bss_lock);
2645 }
2646 EXPORT_SYMBOL(cfg80211_bss_iter);
2647
2648 void cfg80211_update_assoc_bss_entry(struct wireless_dev *wdev,
2649                                      struct ieee80211_channel *chan)
2650 {
2651         struct wiphy *wiphy = wdev->wiphy;
2652         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
2653         struct cfg80211_internal_bss *cbss = wdev->current_bss;
2654         struct cfg80211_internal_bss *new = NULL;
2655         struct cfg80211_internal_bss *bss;
2656         struct cfg80211_bss *nontrans_bss;
2657         struct cfg80211_bss *tmp;
2658
2659         spin_lock_bh(&rdev->bss_lock);
2660
2661         /*
2662          * Some APs use CSA also for bandwidth changes, i.e., without actually
2663          * changing the control channel, so no need to update in such a case.
2664          */
2665         if (cbss->pub.channel == chan)
2666                 goto done;
2667
2668         /* use transmitting bss */
2669         if (cbss->pub.transmitted_bss)
2670                 cbss = container_of(cbss->pub.transmitted_bss,
2671                                     struct cfg80211_internal_bss,
2672                                     pub);
2673
2674         cbss->pub.channel = chan;
2675
2676         list_for_each_entry(bss, &rdev->bss_list, list) {
2677                 if (!cfg80211_bss_type_match(bss->pub.capability,
2678                                              bss->pub.channel->band,
2679                                              wdev->conn_bss_type))
2680                         continue;
2681
2682                 if (bss == cbss)
2683                         continue;
2684
2685                 if (!cmp_bss(&bss->pub, &cbss->pub, BSS_CMP_REGULAR)) {
2686                         new = bss;
2687                         break;
2688                 }
2689         }
2690
2691         if (new) {
2692                 /* to save time, update IEs for transmitting bss only */
2693                 if (cfg80211_update_known_bss(rdev, cbss, new, false)) {
2694                         new->pub.proberesp_ies = NULL;
2695                         new->pub.beacon_ies = NULL;
2696                 }
2697
2698                 list_for_each_entry_safe(nontrans_bss, tmp,
2699                                          &new->pub.nontrans_list,
2700                                          nontrans_list) {
2701                         bss = container_of(nontrans_bss,
2702                                            struct cfg80211_internal_bss, pub);
2703                         if (__cfg80211_unlink_bss(rdev, bss))
2704                                 rdev->bss_generation++;
2705                 }
2706
2707                 WARN_ON(atomic_read(&new->hold));
2708                 if (!WARN_ON(!__cfg80211_unlink_bss(rdev, new)))
2709                         rdev->bss_generation++;
2710         }
2711
2712         rb_erase(&cbss->rbn, &rdev->bss_tree);
2713         rb_insert_bss(rdev, cbss);
2714         rdev->bss_generation++;
2715
2716         list_for_each_entry_safe(nontrans_bss, tmp,
2717                                  &cbss->pub.nontrans_list,
2718                                  nontrans_list) {
2719                 bss = container_of(nontrans_bss,
2720                                    struct cfg80211_internal_bss, pub);
2721                 bss->pub.channel = chan;
2722                 rb_erase(&bss->rbn, &rdev->bss_tree);
2723                 rb_insert_bss(rdev, bss);
2724                 rdev->bss_generation++;
2725         }
2726
2727 done:
2728         spin_unlock_bh(&rdev->bss_lock);
2729 }
2730
2731 #ifdef CONFIG_CFG80211_WEXT
2732 static struct cfg80211_registered_device *
2733 cfg80211_get_dev_from_ifindex(struct net *net, int ifindex)
2734 {
2735         struct cfg80211_registered_device *rdev;
2736         struct net_device *dev;
2737
2738         ASSERT_RTNL();
2739
2740         dev = dev_get_by_index(net, ifindex);
2741         if (!dev)
2742                 return ERR_PTR(-ENODEV);
2743         if (dev->ieee80211_ptr)
2744                 rdev = wiphy_to_rdev(dev->ieee80211_ptr->wiphy);
2745         else
2746                 rdev = ERR_PTR(-ENODEV);
2747         dev_put(dev);
2748         return rdev;
2749 }
2750
2751 int cfg80211_wext_siwscan(struct net_device *dev,
2752                           struct iw_request_info *info,
2753                           union iwreq_data *wrqu, char *extra)
2754 {
2755         struct cfg80211_registered_device *rdev;
2756         struct wiphy *wiphy;
2757         struct iw_scan_req *wreq = NULL;
2758         struct cfg80211_scan_request *creq = NULL;
2759         int i, err, n_channels = 0;
2760         enum nl80211_band band;
2761
2762         if (!netif_running(dev))
2763                 return -ENETDOWN;
2764
2765         if (wrqu->data.length == sizeof(struct iw_scan_req))
2766                 wreq = (struct iw_scan_req *)extra;
2767
2768         rdev = cfg80211_get_dev_from_ifindex(dev_net(dev), dev->ifindex);
2769
2770         if (IS_ERR(rdev))
2771                 return PTR_ERR(rdev);
2772
2773         if (rdev->scan_req || rdev->scan_msg) {
2774                 err = -EBUSY;
2775                 goto out;
2776         }
2777
2778         wiphy = &rdev->wiphy;
2779
2780         /* Determine number of channels, needed to allocate creq */
2781         if (wreq && wreq->num_channels)
2782                 n_channels = wreq->num_channels;
2783         else
2784                 n_channels = ieee80211_get_num_supported_channels(wiphy);
2785
2786         creq = kzalloc(sizeof(*creq) + sizeof(struct cfg80211_ssid) +
2787                        n_channels * sizeof(void *),
2788                        GFP_ATOMIC);
2789         if (!creq) {
2790                 err = -ENOMEM;
2791                 goto out;
2792         }
2793
2794         creq->wiphy = wiphy;
2795         creq->wdev = dev->ieee80211_ptr;
2796         /* SSIDs come after channels */
2797         creq->ssids = (void *)&creq->channels[n_channels];
2798         creq->n_channels = n_channels;
2799         creq->n_ssids = 1;
2800         creq->scan_start = jiffies;
2801
2802         /* translate "Scan on frequencies" request */
2803         i = 0;
2804         for (band = 0; band < NUM_NL80211_BANDS; band++) {
2805                 int j;
2806
2807                 if (!wiphy->bands[band])
2808                         continue;
2809
2810                 for (j = 0; j < wiphy->bands[band]->n_channels; j++) {
2811                         /* ignore disabled channels */
2812                         if (wiphy->bands[band]->channels[j].flags &
2813                                                 IEEE80211_CHAN_DISABLED)
2814                                 continue;
2815
2816                         /* If we have a wireless request structure and the
2817                          * wireless request specifies frequencies, then search
2818                          * for the matching hardware channel.
2819                          */
2820                         if (wreq && wreq->num_channels) {
2821                                 int k;
2822                                 int wiphy_freq = wiphy->bands[band]->channels[j].center_freq;
2823                                 for (k = 0; k < wreq->num_channels; k++) {
2824                                         struct iw_freq *freq =
2825                                                 &wreq->channel_list[k];
2826                                         int wext_freq =
2827                                                 cfg80211_wext_freq(freq);
2828
2829                                         if (wext_freq == wiphy_freq)
2830                                                 goto wext_freq_found;
2831                                 }
2832                                 goto wext_freq_not_found;
2833                         }
2834
2835                 wext_freq_found:
2836                         creq->channels[i] = &wiphy->bands[band]->channels[j];
2837                         i++;
2838                 wext_freq_not_found: ;
2839                 }
2840         }
2841         /* No channels found? */
2842         if (!i) {
2843                 err = -EINVAL;
2844                 goto out;
2845         }
2846
2847         /* Set real number of channels specified in creq->channels[] */
2848         creq->n_channels = i;
2849
2850         /* translate "Scan for SSID" request */
2851         if (wreq) {
2852                 if (wrqu->data.flags & IW_SCAN_THIS_ESSID) {
2853                         if (wreq->essid_len > IEEE80211_MAX_SSID_LEN) {
2854                                 err = -EINVAL;
2855                                 goto out;
2856                         }
2857                         memcpy(creq->ssids[0].ssid, wreq->essid, wreq->essid_len);
2858                         creq->ssids[0].ssid_len = wreq->essid_len;
2859                 }
2860                 if (wreq->scan_type == IW_SCAN_TYPE_PASSIVE)
2861                         creq->n_ssids = 0;
2862         }
2863
2864         for (i = 0; i < NUM_NL80211_BANDS; i++)
2865                 if (wiphy->bands[i])
2866                         creq->rates[i] = (1 << wiphy->bands[i]->n_bitrates) - 1;
2867
2868         eth_broadcast_addr(creq->bssid);
2869
2870         wiphy_lock(&rdev->wiphy);
2871
2872         rdev->scan_req = creq;
2873         err = rdev_scan(rdev, creq);
2874         if (err) {
2875                 rdev->scan_req = NULL;
2876                 /* creq will be freed below */
2877         } else {
2878                 nl80211_send_scan_start(rdev, dev->ieee80211_ptr);
2879                 /* creq now owned by driver */
2880                 creq = NULL;
2881                 dev_hold(dev);
2882         }
2883         wiphy_unlock(&rdev->wiphy);
2884  out:
2885         kfree(creq);
2886         return err;
2887 }
2888 EXPORT_WEXT_HANDLER(cfg80211_wext_siwscan);
2889
2890 static char *ieee80211_scan_add_ies(struct iw_request_info *info,
2891                                     const struct cfg80211_bss_ies *ies,
2892                                     char *current_ev, char *end_buf)
2893 {
2894         const u8 *pos, *end, *next;
2895         struct iw_event iwe;
2896
2897         if (!ies)
2898                 return current_ev;
2899
2900         /*
2901          * If needed, fragment the IEs buffer (at IE boundaries) into short
2902          * enough fragments to fit into IW_GENERIC_IE_MAX octet messages.
2903          */
2904         pos = ies->data;
2905         end = pos + ies->len;
2906
2907         while (end - pos > IW_GENERIC_IE_MAX) {
2908                 next = pos + 2 + pos[1];
2909                 while (next + 2 + next[1] - pos < IW_GENERIC_IE_MAX)
2910                         next = next + 2 + next[1];
2911
2912                 memset(&iwe, 0, sizeof(iwe));
2913                 iwe.cmd = IWEVGENIE;
2914                 iwe.u.data.length = next - pos;
2915                 current_ev = iwe_stream_add_point_check(info, current_ev,
2916                                                         end_buf, &iwe,
2917                                                         (void *)pos);
2918                 if (IS_ERR(current_ev))
2919                         return current_ev;
2920                 pos = next;
2921         }
2922
2923         if (end > pos) {
2924                 memset(&iwe, 0, sizeof(iwe));
2925                 iwe.cmd = IWEVGENIE;
2926                 iwe.u.data.length = end - pos;
2927                 current_ev = iwe_stream_add_point_check(info, current_ev,
2928                                                         end_buf, &iwe,
2929                                                         (void *)pos);
2930                 if (IS_ERR(current_ev))
2931                         return current_ev;
2932         }
2933
2934         return current_ev;
2935 }
2936
2937 static char *
2938 ieee80211_bss(struct wiphy *wiphy, struct iw_request_info *info,
2939               struct cfg80211_internal_bss *bss, char *current_ev,
2940               char *end_buf)
2941 {
2942         const struct cfg80211_bss_ies *ies;
2943         struct iw_event iwe;
2944         const u8 *ie;
2945         u8 buf[50];
2946         u8 *cfg, *p, *tmp;
2947         int rem, i, sig;
2948         bool ismesh = false;
2949
2950         memset(&iwe, 0, sizeof(iwe));
2951         iwe.cmd = SIOCGIWAP;
2952         iwe.u.ap_addr.sa_family = ARPHRD_ETHER;
2953         memcpy(iwe.u.ap_addr.sa_data, bss->pub.bssid, ETH_ALEN);
2954         current_ev = iwe_stream_add_event_check(info, current_ev, end_buf, &iwe,
2955                                                 IW_EV_ADDR_LEN);
2956         if (IS_ERR(current_ev))
2957                 return current_ev;
2958
2959         memset(&iwe, 0, sizeof(iwe));
2960         iwe.cmd = SIOCGIWFREQ;
2961         iwe.u.freq.m = ieee80211_frequency_to_channel(bss->pub.channel->center_freq);
2962         iwe.u.freq.e = 0;
2963         current_ev = iwe_stream_add_event_check(info, current_ev, end_buf, &iwe,
2964                                                 IW_EV_FREQ_LEN);
2965         if (IS_ERR(current_ev))
2966                 return current_ev;
2967
2968         memset(&iwe, 0, sizeof(iwe));
2969         iwe.cmd = SIOCGIWFREQ;
2970         iwe.u.freq.m = bss->pub.channel->center_freq;
2971         iwe.u.freq.e = 6;
2972         current_ev = iwe_stream_add_event_check(info, current_ev, end_buf, &iwe,
2973                                                 IW_EV_FREQ_LEN);
2974         if (IS_ERR(current_ev))
2975                 return current_ev;
2976
2977         if (wiphy->signal_type != CFG80211_SIGNAL_TYPE_NONE) {
2978                 memset(&iwe, 0, sizeof(iwe));
2979                 iwe.cmd = IWEVQUAL;
2980                 iwe.u.qual.updated = IW_QUAL_LEVEL_UPDATED |
2981                                      IW_QUAL_NOISE_INVALID |
2982                                      IW_QUAL_QUAL_UPDATED;
2983                 switch (wiphy->signal_type) {
2984                 case CFG80211_SIGNAL_TYPE_MBM:
2985                         sig = bss->pub.signal / 100;
2986                         iwe.u.qual.level = sig;
2987                         iwe.u.qual.updated |= IW_QUAL_DBM;
2988                         if (sig < -110)         /* rather bad */
2989                                 sig = -110;
2990                         else if (sig > -40)     /* perfect */
2991                                 sig = -40;
2992                         /* will give a range of 0 .. 70 */
2993                         iwe.u.qual.qual = sig + 110;
2994                         break;
2995                 case CFG80211_SIGNAL_TYPE_UNSPEC:
2996                         iwe.u.qual.level = bss->pub.signal;
2997                         /* will give range 0 .. 100 */
2998                         iwe.u.qual.qual = bss->pub.signal;
2999                         break;
3000                 default:
3001                         /* not reached */
3002                         break;
3003                 }
3004                 current_ev = iwe_stream_add_event_check(info, current_ev,
3005                                                         end_buf, &iwe,
3006                                                         IW_EV_QUAL_LEN);
3007                 if (IS_ERR(current_ev))
3008                         return current_ev;
3009         }
3010
3011         memset(&iwe, 0, sizeof(iwe));
3012         iwe.cmd = SIOCGIWENCODE;
3013         if (bss->pub.capability & WLAN_CAPABILITY_PRIVACY)
3014                 iwe.u.data.flags = IW_ENCODE_ENABLED | IW_ENCODE_NOKEY;
3015         else
3016                 iwe.u.data.flags = IW_ENCODE_DISABLED;
3017         iwe.u.data.length = 0;
3018         current_ev = iwe_stream_add_point_check(info, current_ev, end_buf,
3019                                                 &iwe, "");
3020         if (IS_ERR(current_ev))
3021                 return current_ev;
3022
3023         rcu_read_lock();
3024         ies = rcu_dereference(bss->pub.ies);
3025         rem = ies->len;
3026         ie = ies->data;
3027
3028         while (rem >= 2) {
3029                 /* invalid data */
3030                 if (ie[1] > rem - 2)
3031                         break;
3032
3033                 switch (ie[0]) {
3034                 case WLAN_EID_SSID:
3035                         memset(&iwe, 0, sizeof(iwe));
3036                         iwe.cmd = SIOCGIWESSID;
3037                         iwe.u.data.length = ie[1];
3038                         iwe.u.data.flags = 1;
3039                         current_ev = iwe_stream_add_point_check(info,
3040                                                                 current_ev,
3041                                                                 end_buf, &iwe,
3042                                                                 (u8 *)ie + 2);
3043                         if (IS_ERR(current_ev))
3044                                 goto unlock;
3045                         break;
3046                 case WLAN_EID_MESH_ID:
3047                         memset(&iwe, 0, sizeof(iwe));
3048                         iwe.cmd = SIOCGIWESSID;
3049                         iwe.u.data.length = ie[1];
3050                         iwe.u.data.flags = 1;
3051                         current_ev = iwe_stream_add_point_check(info,
3052                                                                 current_ev,
3053                                                                 end_buf, &iwe,
3054                                                                 (u8 *)ie + 2);
3055                         if (IS_ERR(current_ev))
3056                                 goto unlock;
3057                         break;
3058                 case WLAN_EID_MESH_CONFIG:
3059                         ismesh = true;
3060                         if (ie[1] != sizeof(struct ieee80211_meshconf_ie))
3061                                 break;
3062                         cfg = (u8 *)ie + 2;
3063                         memset(&iwe, 0, sizeof(iwe));
3064                         iwe.cmd = IWEVCUSTOM;
3065                         sprintf(buf, "Mesh Network Path Selection Protocol ID: "
3066                                 "0x%02X", cfg[0]);
3067                         iwe.u.data.length = strlen(buf);
3068                         current_ev = iwe_stream_add_point_check(info,
3069                                                                 current_ev,
3070                                                                 end_buf,
3071                                                                 &iwe, buf);
3072                         if (IS_ERR(current_ev))
3073                                 goto unlock;
3074                         sprintf(buf, "Path Selection Metric ID: 0x%02X",
3075                                 cfg[1]);
3076                         iwe.u.data.length = strlen(buf);
3077                         current_ev = iwe_stream_add_point_check(info,
3078                                                                 current_ev,
3079                                                                 end_buf,
3080                                                                 &iwe, buf);
3081                         if (IS_ERR(current_ev))
3082                                 goto unlock;
3083                         sprintf(buf, "Congestion Control Mode ID: 0x%02X",
3084                                 cfg[2]);
3085                         iwe.u.data.length = strlen(buf);
3086                         current_ev = iwe_stream_add_point_check(info,
3087                                                                 current_ev,
3088                                                                 end_buf,
3089                                                                 &iwe, buf);
3090                         if (IS_ERR(current_ev))
3091                                 goto unlock;
3092                         sprintf(buf, "Synchronization ID: 0x%02X", cfg[3]);
3093                         iwe.u.data.length = strlen(buf);
3094                         current_ev = iwe_stream_add_point_check(info,
3095                                                                 current_ev,
3096                                                                 end_buf,
3097                                                                 &iwe, buf);
3098                         if (IS_ERR(current_ev))
3099                                 goto unlock;
3100                         sprintf(buf, "Authentication ID: 0x%02X", cfg[4]);
3101                         iwe.u.data.length = strlen(buf);
3102                         current_ev = iwe_stream_add_point_check(info,
3103                                                                 current_ev,
3104                                                                 end_buf,
3105                                                                 &iwe, buf);
3106                         if (IS_ERR(current_ev))
3107                                 goto unlock;
3108                         sprintf(buf, "Formation Info: 0x%02X", cfg[5]);
3109                         iwe.u.data.length = strlen(buf);
3110                         current_ev = iwe_stream_add_point_check(info,
3111                                                                 current_ev,
3112                                                                 end_buf,
3113                                                                 &iwe, buf);
3114                         if (IS_ERR(current_ev))
3115                                 goto unlock;
3116                         sprintf(buf, "Capabilities: 0x%02X", cfg[6]);
3117                         iwe.u.data.length = strlen(buf);
3118                         current_ev = iwe_stream_add_point_check(info,
3119                                                                 current_ev,
3120                                                                 end_buf,
3121                                                                 &iwe, buf);
3122                         if (IS_ERR(current_ev))
3123                                 goto unlock;
3124                         break;
3125                 case WLAN_EID_SUPP_RATES:
3126                 case WLAN_EID_EXT_SUPP_RATES:
3127                         /* display all supported rates in readable format */
3128                         p = current_ev + iwe_stream_lcp_len(info);
3129
3130                         memset(&iwe, 0, sizeof(iwe));
3131                         iwe.cmd = SIOCGIWRATE;
3132                         /* Those two flags are ignored... */
3133                         iwe.u.bitrate.fixed = iwe.u.bitrate.disabled = 0;
3134
3135                         for (i = 0; i < ie[1]; i++) {
3136                                 iwe.u.bitrate.value =
3137                                         ((ie[i + 2] & 0x7f) * 500000);
3138                                 tmp = p;
3139                                 p = iwe_stream_add_value(info, current_ev, p,
3140                                                          end_buf, &iwe,
3141                                                          IW_EV_PARAM_LEN);
3142                                 if (p == tmp) {
3143                                         current_ev = ERR_PTR(-E2BIG);
3144                                         goto unlock;
3145                                 }
3146                         }
3147                         current_ev = p;
3148                         break;
3149                 }
3150                 rem -= ie[1] + 2;
3151                 ie += ie[1] + 2;
3152         }
3153
3154         if (bss->pub.capability & (WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_IBSS) ||
3155             ismesh) {
3156                 memset(&iwe, 0, sizeof(iwe));
3157                 iwe.cmd = SIOCGIWMODE;
3158                 if (ismesh)
3159                         iwe.u.mode = IW_MODE_MESH;
3160                 else if (bss->pub.capability & WLAN_CAPABILITY_ESS)
3161                         iwe.u.mode = IW_MODE_MASTER;
3162                 else
3163                         iwe.u.mode = IW_MODE_ADHOC;
3164                 current_ev = iwe_stream_add_event_check(info, current_ev,
3165                                                         end_buf, &iwe,
3166                                                         IW_EV_UINT_LEN);
3167                 if (IS_ERR(current_ev))
3168                         goto unlock;
3169         }
3170
3171         memset(&iwe, 0, sizeof(iwe));
3172         iwe.cmd = IWEVCUSTOM;
3173         sprintf(buf, "tsf=%016llx", (unsigned long long)(ies->tsf));
3174         iwe.u.data.length = strlen(buf);
3175         current_ev = iwe_stream_add_point_check(info, current_ev, end_buf,
3176                                                 &iwe, buf);
3177         if (IS_ERR(current_ev))
3178                 goto unlock;
3179         memset(&iwe, 0, sizeof(iwe));
3180         iwe.cmd = IWEVCUSTOM;
3181         sprintf(buf, " Last beacon: %ums ago",
3182                 elapsed_jiffies_msecs(bss->ts));
3183         iwe.u.data.length = strlen(buf);
3184         current_ev = iwe_stream_add_point_check(info, current_ev,
3185                                                 end_buf, &iwe, buf);
3186         if (IS_ERR(current_ev))
3187                 goto unlock;
3188
3189         current_ev = ieee80211_scan_add_ies(info, ies, current_ev, end_buf);
3190
3191  unlock:
3192         rcu_read_unlock();
3193         return current_ev;
3194 }
3195
3196
3197 static int ieee80211_scan_results(struct cfg80211_registered_device *rdev,
3198                                   struct iw_request_info *info,
3199                                   char *buf, size_t len)
3200 {
3201         char *current_ev = buf;
3202         char *end_buf = buf + len;
3203         struct cfg80211_internal_bss *bss;
3204         int err = 0;
3205
3206         spin_lock_bh(&rdev->bss_lock);
3207         cfg80211_bss_expire(rdev);
3208
3209         list_for_each_entry(bss, &rdev->bss_list, list) {
3210                 if (buf + len - current_ev <= IW_EV_ADDR_LEN) {
3211                         err = -E2BIG;
3212                         break;
3213                 }
3214                 current_ev = ieee80211_bss(&rdev->wiphy, info, bss,
3215                                            current_ev, end_buf);
3216                 if (IS_ERR(current_ev)) {
3217                         err = PTR_ERR(current_ev);
3218                         break;
3219                 }
3220         }
3221         spin_unlock_bh(&rdev->bss_lock);
3222
3223         if (err)
3224                 return err;
3225         return current_ev - buf;
3226 }
3227
3228
3229 int cfg80211_wext_giwscan(struct net_device *dev,
3230                           struct iw_request_info *info,
3231                           struct iw_point *data, char *extra)
3232 {
3233         struct cfg80211_registered_device *rdev;
3234         int res;
3235
3236         if (!netif_running(dev))
3237                 return -ENETDOWN;
3238
3239         rdev = cfg80211_get_dev_from_ifindex(dev_net(dev), dev->ifindex);
3240
3241         if (IS_ERR(rdev))
3242                 return PTR_ERR(rdev);
3243
3244         if (rdev->scan_req || rdev->scan_msg)
3245                 return -EAGAIN;
3246
3247         res = ieee80211_scan_results(rdev, info, extra, data->length);
3248         data->length = 0;
3249         if (res >= 0) {
3250                 data->length = res;
3251                 res = 0;
3252         }
3253
3254         return res;
3255 }
3256 EXPORT_WEXT_HANDLER(cfg80211_wext_giwscan);
3257 #endif