net/mac80211/util.c

   1 /*
   2  * Copyright 2002-2005, Instant802 Networks, Inc.
   3  * Copyright 2005-2006, Devicescape Software, Inc.
   4  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
   5  * Copyright 2007       Johannes Berg <johannes@sipsolutions.net>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License version 2 as
   9  * published by the Free Software Foundation.
  10  *
  11  * utilities for mac80211
  12  */
  13
  14 #include <net/mac80211.h>
  15 #include <linux/netdevice.h>
  16 #include <linux/export.h>
  17 #include <linux/types.h>
  18 #include <linux/slab.h>
  19 #include <linux/skbuff.h>
  20 #include <linux/etherdevice.h>
  21 #include <linux/if_arp.h>
  22 #include <linux/bitmap.h>
  23 #include <linux/crc32.h>
  24 #include <net/net_namespace.h>
  25 #include <net/cfg80211.h>
  26 #include <net/rtnetlink.h>
  27
  28 #include "ieee80211_i.h"
  29 #include "driver-ops.h"
  30 #include "rate.h"
  31 #include "mesh.h"
  32 #include "wme.h"
  33 #include "led.h"
  34 #include "wep.h"
  35
  36 /* privid for wiphys to determine whether they belong to us or not */
  37 void *mac80211_wiphy_privid = &mac80211_wiphy_privid;
  38
  39 struct ieee80211_hw *wiphy_to_ieee80211_hw(struct wiphy *wiphy)
  40 {
  41         struct ieee80211_local *local;
  42         BUG_ON(!wiphy);
  43
  44         local = wiphy_priv(wiphy);
  45         return &local->hw;
  46 }
  47 EXPORT_SYMBOL(wiphy_to_ieee80211_hw);
  48
  49 u8 *ieee80211_get_bssid(struct ieee80211_hdr *hdr, size_t len,
  50                         enum nl80211_iftype type)
  51 {
  52         __le16 fc = hdr->frame_control;
  53
  54          /* drop ACK/CTS frames and incorrect hdr len (ctrl) */
  55         if (len < 16)
  56                 return NULL;
  57
  58         if (ieee80211_is_data(fc)) {
  59                 if (len < 24) /* drop incorrect hdr len (data) */
  60                         return NULL;
  61
  62                 if (ieee80211_has_a4(fc))
  63                         return NULL;
  64                 if (ieee80211_has_tods(fc))
  65                         return hdr->addr1;
  66                 if (ieee80211_has_fromds(fc))
  67                         return hdr->addr2;
  68
  69                 return hdr->addr3;
  70         }
  71
  72         if (ieee80211_is_mgmt(fc)) {
  73                 if (len < 24) /* drop incorrect hdr len (mgmt) */
  74                         return NULL;
  75                 return hdr->addr3;
  76         }
  77
  78         if (ieee80211_is_ctl(fc)) {
  79                 if(ieee80211_is_pspoll(fc))
  80                         return hdr->addr1;
  81
  82                 if (ieee80211_is_back_req(fc)) {
  83                         switch (type) {
  84                         case NL80211_IFTYPE_STATION:
  85                                 return hdr->addr2;
  86                         case NL80211_IFTYPE_AP:
  87                         case NL80211_IFTYPE_AP_VLAN:
  88                                 return hdr->addr1;
  89                         default:
  90                                 break; /* fall through to the return */
  91                         }
  92                 }
  93         }
  94
  95         return NULL;
  96 }
  97
  98 void ieee80211_tx_set_protected(struct ieee80211_tx_data *tx)
  99 {
 100         struct sk_buff *skb;
 101         struct ieee80211_hdr *hdr;
 102
 103         skb_queue_walk(&tx->skbs, skb) {
 104                 hdr = (struct ieee80211_hdr *) skb->data;
 105                 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
 106         }
 107 }
 108
 109 int ieee80211_frame_duration(enum ieee80211_band band, size_t len,
 110                              int rate, int erp, int short_preamble,
 111                              int shift)
 112 {
 113         int dur;
 114
 115         /* calculate duration (in microseconds, rounded up to next higher
 116          * integer if it includes a fractional microsecond) to send frame of
 117          * len bytes (does not include FCS) at the given rate. Duration will
 118          * also include SIFS.
 119          *
 120          * rate is in 100 kbps, so divident is multiplied by 10 in the
 121          * DIV_ROUND_UP() operations.
 122          *
 123          * shift may be 2 for 5 MHz channels or 1 for 10 MHz channels, and
 124          * is assumed to be 0 otherwise.
 125          */
 126
 127         if (band == IEEE80211_BAND_5GHZ || erp) {
 128                 /*
 129                  * OFDM:
 130                  *
 131                  * N_DBPS = DATARATE x 4
 132                  * N_SYM = Ceiling((16+8xLENGTH+6) / N_DBPS)
 133                  *      (16 = SIGNAL time, 6 = tail bits)
 134                  * TXTIME = T_PREAMBLE + T_SIGNAL + T_SYM x N_SYM + Signal Ext
 135                  *
 136                  * T_SYM = 4 usec
 137                  * 802.11a - 18.5.2: aSIFSTime = 16 usec
 138                  * 802.11g - 19.8.4: aSIFSTime = 10 usec +
 139                  *      signal ext = 6 usec
 140                  */
 141                 dur = 16; /* SIFS + signal ext */
 142                 dur += 16; /* IEEE 802.11-2012 18.3.2.4: T_PREAMBLE = 16 usec */
 143                 dur += 4; /* IEEE 802.11-2012 18.3.2.4: T_SIGNAL = 4 usec */
 144
 145                 /* IEEE 802.11-2012 18.3.2.4: all values above are:
 146                  *  * times 4 for 5 MHz
 147                  *  * times 2 for 10 MHz
 148                  */
 149                 dur *= 1 << shift;
 150
 151                 /* rates should already consider the channel bandwidth,
 152                  * don't apply divisor again.
 153                  */
 154                 dur += 4 * DIV_ROUND_UP((16 + 8 * (len + 4) + 6) * 10,
 155                                         4 * rate); /* T_SYM x N_SYM */
 156         } else {
 157                 /*
 158                  * 802.11b or 802.11g with 802.11b compatibility:
 159                  * 18.3.4: TXTIME = PreambleLength + PLCPHeaderTime +
 160                  * Ceiling(((LENGTH+PBCC)x8)/DATARATE). PBCC=0.
 161                  *
 162                  * 802.11 (DS): 15.3.3, 802.11b: 18.3.4
 163                  * aSIFSTime = 10 usec
 164                  * aPreambleLength = 144 usec or 72 usec with short preamble
 165                  * aPLCPHeaderLength = 48 usec or 24 usec with short preamble
 166                  */
 167                 dur = 10; /* aSIFSTime = 10 usec */
 168                 dur += short_preamble ? (72 + 24) : (144 + 48);
 169
 170                 dur += DIV_ROUND_UP(8 * (len + 4) * 10, rate);
 171         }
 172
 173         return dur;
 174 }
 175
 176 /* Exported duration function for driver use */
 177 __le16 ieee80211_generic_frame_duration(struct ieee80211_hw *hw,
 178                                         struct ieee80211_vif *vif,
 179                                         enum ieee80211_band band,
 180                                         size_t frame_len,
 181                                         struct ieee80211_rate *rate)
 182 {
 183         struct ieee80211_sub_if_data *sdata;
 184         u16 dur;
 185         int erp, shift = 0;
 186         bool short_preamble = false;
 187
 188         erp = 0;
 189         if (vif) {
 190                 sdata = vif_to_sdata(vif);
 191                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 192                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 193                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 194                 shift = ieee80211_vif_get_shift(vif);
 195         }
 196
 197         dur = ieee80211_frame_duration(band, frame_len, rate->bitrate, erp,
 198                                        short_preamble, shift);
 199
 200         return cpu_to_le16(dur);
 201 }
 202 EXPORT_SYMBOL(ieee80211_generic_frame_duration);
 203
 204 __le16 ieee80211_rts_duration(struct ieee80211_hw *hw,
 205                               struct ieee80211_vif *vif, size_t frame_len,
 206                               const struct ieee80211_tx_info *frame_txctl)
 207 {
 208         struct ieee80211_local *local = hw_to_local(hw);
 209         struct ieee80211_rate *rate;
 210         struct ieee80211_sub_if_data *sdata;
 211         bool short_preamble;
 212         int erp, shift = 0, bitrate;
 213         u16 dur;
 214         struct ieee80211_supported_band *sband;
 215
 216         sband = local->hw.wiphy->bands[frame_txctl->band];
 217
 218         short_preamble = false;
 219
 220         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
 221
 222         erp = 0;
 223         if (vif) {
 224                 sdata = vif_to_sdata(vif);
 225                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 226                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 227                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 228                 shift = ieee80211_vif_get_shift(vif);
 229         }
 230
 231         bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
 232
 233         /* CTS duration */
 234         dur = ieee80211_frame_duration(sband->band, 10, bitrate,
 235                                        erp, short_preamble, shift);
 236         /* Data frame duration */
 237         dur += ieee80211_frame_duration(sband->band, frame_len, bitrate,
 238                                         erp, short_preamble, shift);
 239         /* ACK duration */
 240         dur += ieee80211_frame_duration(sband->band, 10, bitrate,
 241                                         erp, short_preamble, shift);
 242
 243         return cpu_to_le16(dur);
 244 }
 245 EXPORT_SYMBOL(ieee80211_rts_duration);
 246
 247 __le16 ieee80211_ctstoself_duration(struct ieee80211_hw *hw,
 248                                     struct ieee80211_vif *vif,
 249                                     size_t frame_len,
 250                                     const struct ieee80211_tx_info *frame_txctl)
 251 {
 252         struct ieee80211_local *local = hw_to_local(hw);
 253         struct ieee80211_rate *rate;
 254         struct ieee80211_sub_if_data *sdata;
 255         bool short_preamble;
 256         int erp, shift = 0, bitrate;
 257         u16 dur;
 258         struct ieee80211_supported_band *sband;
 259
 260         sband = local->hw.wiphy->bands[frame_txctl->band];
 261
 262         short_preamble = false;
 263
 264         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
 265         erp = 0;
 266         if (vif) {
 267                 sdata = vif_to_sdata(vif);
 268                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 269                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 270                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 271                 shift = ieee80211_vif_get_shift(vif);
 272         }
 273
 274         bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
 275
 276         /* Data frame duration */
 277         dur = ieee80211_frame_duration(sband->band, frame_len, bitrate,
 278                                        erp, short_preamble, shift);
 279         if (!(frame_txctl->flags & IEEE80211_TX_CTL_NO_ACK)) {
 280                 /* ACK duration */
 281                 dur += ieee80211_frame_duration(sband->band, 10, bitrate,
 282                                                 erp, short_preamble, shift);
 283         }
 284
 285         return cpu_to_le16(dur);
 286 }
 287 EXPORT_SYMBOL(ieee80211_ctstoself_duration);
 288
 289 void ieee80211_propagate_queue_wake(struct ieee80211_local *local, int queue)
 290 {
 291         struct ieee80211_sub_if_data *sdata;
 292         int n_acs = IEEE80211_NUM_ACS;
 293
 294         if (local->hw.queues < IEEE80211_NUM_ACS)
 295                 n_acs = 1;
 296
 297         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 298                 int ac;
 299
 300                 if (!sdata->dev)
 301                         continue;
 302
 303                 if (test_bit(SDATA_STATE_OFFCHANNEL, &sdata->state))
 304                         continue;
 305
 306                 if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE &&
 307                     local->queue_stop_reasons[sdata->vif.cab_queue] != 0)
 308                         continue;
 309
 310                 for (ac = 0; ac < n_acs; ac++) {
 311                         int ac_queue = sdata->vif.hw_queue[ac];
 312
 313                         if (ac_queue == queue ||
 314                             (sdata->vif.cab_queue == queue &&
 315                              local->queue_stop_reasons[ac_queue] == 0 &&
 316                              skb_queue_empty(&local->pending[ac_queue])))
 317                                 netif_wake_subqueue(sdata->dev, ac);
 318                 }
 319         }
 320 }
 321
 322 static void __ieee80211_wake_queue(struct ieee80211_hw *hw, int queue,
 323                                    enum queue_stop_reason reason)
 324 {
 325         struct ieee80211_local *local = hw_to_local(hw);
 326
 327         trace_wake_queue(local, queue, reason);
 328
 329         if (WARN_ON(queue >= hw->queues))
 330                 return;
 331
 332         if (!test_bit(reason, &local->queue_stop_reasons[queue]))
 333                 return;
 334
 335         __clear_bit(reason, &local->queue_stop_reasons[queue]);
 336
 337         if (local->queue_stop_reasons[queue] != 0)
 338                 /* someone still has this queue stopped */
 339                 return;
 340
 341         if (skb_queue_empty(&local->pending[queue])) {
 342                 rcu_read_lock();
 343                 ieee80211_propagate_queue_wake(local, queue);
 344                 rcu_read_unlock();
 345         } else
 346                 tasklet_schedule(&local->tx_pending_tasklet);
 347 }
 348
 349 void ieee80211_wake_queue_by_reason(struct ieee80211_hw *hw, int queue,
 350                                     enum queue_stop_reason reason)
 351 {
 352         struct ieee80211_local *local = hw_to_local(hw);
 353         unsigned long flags;
 354
 355         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 356         __ieee80211_wake_queue(hw, queue, reason);
 357         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 358 }
 359
 360 void ieee80211_wake_queue(struct ieee80211_hw *hw, int queue)
 361 {
 362         ieee80211_wake_queue_by_reason(hw, queue,
 363                                        IEEE80211_QUEUE_STOP_REASON_DRIVER);
 364 }
 365 EXPORT_SYMBOL(ieee80211_wake_queue);
 366
 367 static void __ieee80211_stop_queue(struct ieee80211_hw *hw, int queue,
 368                                    enum queue_stop_reason reason)
 369 {
 370         struct ieee80211_local *local = hw_to_local(hw);
 371         struct ieee80211_sub_if_data *sdata;
 372         int n_acs = IEEE80211_NUM_ACS;
 373
 374         trace_stop_queue(local, queue, reason);
 375
 376         if (WARN_ON(queue >= hw->queues))
 377                 return;
 378
 379         if (test_bit(reason, &local->queue_stop_reasons[queue]))
 380                 return;
 381
 382         __set_bit(reason, &local->queue_stop_reasons[queue]);
 383
 384         if (local->hw.queues < IEEE80211_NUM_ACS)
 385                 n_acs = 1;
 386
 387         rcu_read_lock();
 388         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 389                 int ac;
 390
 391                 if (!sdata->dev)
 392                         continue;
 393
 394                 for (ac = 0; ac < n_acs; ac++) {
 395                         if (sdata->vif.hw_queue[ac] == queue ||
 396                             sdata->vif.cab_queue == queue)
 397                                 netif_stop_subqueue(sdata->dev, ac);
 398                 }
 399         }
 400         rcu_read_unlock();
 401 }
 402
 403 void ieee80211_stop_queue_by_reason(struct ieee80211_hw *hw, int queue,
 404                                     enum queue_stop_reason reason)
 405 {
 406         struct ieee80211_local *local = hw_to_local(hw);
 407         unsigned long flags;
 408
 409         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 410         __ieee80211_stop_queue(hw, queue, reason);
 411         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 412 }
 413
 414 void ieee80211_stop_queue(struct ieee80211_hw *hw, int queue)
 415 {
 416         ieee80211_stop_queue_by_reason(hw, queue,
 417                                        IEEE80211_QUEUE_STOP_REASON_DRIVER);
 418 }
 419 EXPORT_SYMBOL(ieee80211_stop_queue);
 420
 421 void ieee80211_add_pending_skb(struct ieee80211_local *local,
 422                                struct sk_buff *skb)
 423 {
 424         struct ieee80211_hw *hw = &local->hw;
 425         unsigned long flags;
 426         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 427         int queue = info->hw_queue;
 428
 429         if (WARN_ON(!info->control.vif)) {
 430                 ieee80211_free_txskb(&local->hw, skb);
 431                 return;
 432         }
 433
 434         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 435         __ieee80211_stop_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
 436         __skb_queue_tail(&local->pending[queue], skb);
 437         __ieee80211_wake_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
 438         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 439 }
 440
 441 void ieee80211_add_pending_skbs_fn(struct ieee80211_local *local,
 442                                    struct sk_buff_head *skbs,
 443                                    void (*fn)(void *data), void *data)
 444 {
 445         struct ieee80211_hw *hw = &local->hw;
 446         struct sk_buff *skb;
 447         unsigned long flags;
 448         int queue, i;
 449
 450         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 451         while ((skb = skb_dequeue(skbs))) {
 452                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 453
 454                 if (WARN_ON(!info->control.vif)) {
 455                         ieee80211_free_txskb(&local->hw, skb);
 456                         continue;
 457                 }
 458
 459                 queue = info->hw_queue;
 460
 461                 __ieee80211_stop_queue(hw, queue,
 462                                 IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
 463
 464                 __skb_queue_tail(&local->pending[queue], skb);
 465         }
 466
 467         if (fn)
 468                 fn(data);
 469
 470         for (i = 0; i < hw->queues; i++)
 471                 __ieee80211_wake_queue(hw, i,
 472                         IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
 473         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 474 }
 475
 476 void ieee80211_stop_queues_by_reason(struct ieee80211_hw *hw,
 477                                      unsigned long queues,
 478                                      enum queue_stop_reason reason)
 479 {
 480         struct ieee80211_local *local = hw_to_local(hw);
 481         unsigned long flags;
 482         int i;
 483
 484         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 485
 486         for_each_set_bit(i, &queues, hw->queues)
 487                 __ieee80211_stop_queue(hw, i, reason);
 488
 489         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 490 }
 491
 492 void ieee80211_stop_queues(struct ieee80211_hw *hw)
 493 {
 494         ieee80211_stop_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
 495                                         IEEE80211_QUEUE_STOP_REASON_DRIVER);
 496 }
 497 EXPORT_SYMBOL(ieee80211_stop_queues);
 498
 499 int ieee80211_queue_stopped(struct ieee80211_hw *hw, int queue)
 500 {
 501         struct ieee80211_local *local = hw_to_local(hw);
 502         unsigned long flags;
 503         int ret;
 504
 505         if (WARN_ON(queue >= hw->queues))
 506                 return true;
 507
 508         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 509         ret = test_bit(IEEE80211_QUEUE_STOP_REASON_DRIVER,
 510                        &local->queue_stop_reasons[queue]);
 511         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 512         return ret;
 513 }
 514 EXPORT_SYMBOL(ieee80211_queue_stopped);
 515
 516 void ieee80211_wake_queues_by_reason(struct ieee80211_hw *hw,
 517                                      unsigned long queues,
 518                                      enum queue_stop_reason reason)
 519 {
 520         struct ieee80211_local *local = hw_to_local(hw);
 521         unsigned long flags;
 522         int i;
 523
 524         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 525
 526         for_each_set_bit(i, &queues, hw->queues)
 527                 __ieee80211_wake_queue(hw, i, reason);
 528
 529         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 530 }
 531
 532 void ieee80211_wake_queues(struct ieee80211_hw *hw)
 533 {
 534         ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
 535                                         IEEE80211_QUEUE_STOP_REASON_DRIVER);
 536 }
 537 EXPORT_SYMBOL(ieee80211_wake_queues);
 538
 539 void ieee80211_flush_queues(struct ieee80211_local *local,
 540                             struct ieee80211_sub_if_data *sdata)
 541 {
 542         u32 queues;
 543
 544         if (!local->ops->flush)
 545                 return;
 546
 547         if (sdata && local->hw.flags & IEEE80211_HW_QUEUE_CONTROL) {
 548                 int ac;
 549
 550                 queues = 0;
 551
 552                 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
 553                         queues |= BIT(sdata->vif.hw_queue[ac]);
 554                 if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE)
 555                         queues |= BIT(sdata->vif.cab_queue);
 556         } else {
 557                 /* all queues */
 558                 queues = BIT(local->hw.queues) - 1;
 559         }
 560
 561         ieee80211_stop_queues_by_reason(&local->hw, IEEE80211_MAX_QUEUE_MAP,
 562                                         IEEE80211_QUEUE_STOP_REASON_FLUSH);
 563
 564         drv_flush(local, queues, false);
 565
 566         ieee80211_wake_queues_by_reason(&local->hw, IEEE80211_MAX_QUEUE_MAP,
 567                                         IEEE80211_QUEUE_STOP_REASON_FLUSH);
 568 }
 569
 570 static void __iterate_active_interfaces(struct ieee80211_local *local,
 571                                         u32 iter_flags,
 572                                         void (*iterator)(void *data, u8 *mac,
 573                                                 struct ieee80211_vif *vif),
 574                                         void *data)
 575 {
 576         struct ieee80211_sub_if_data *sdata;
 577
 578         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 579                 switch (sdata->vif.type) {
 580                 case NL80211_IFTYPE_MONITOR:
 581                         if (!(sdata->u.mntr_flags & MONITOR_FLAG_ACTIVE))
 582                                 continue;
 583                         break;
 584                 case NL80211_IFTYPE_AP_VLAN:
 585                         continue;
 586                 default:
 587                         break;
 588                 }
 589                 if (!(iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL) &&
 590                     !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 591                         continue;
 592                 if (ieee80211_sdata_running(sdata))
 593                         iterator(data, sdata->vif.addr,
 594                                  &sdata->vif);
 595         }
 596
 597         sdata = rcu_dereference_check(local->monitor_sdata,
 598                                       lockdep_is_held(&local->iflist_mtx) ||
 599                                       lockdep_rtnl_is_held());
 600         if (sdata &&
 601             (iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL ||
 602              sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 603                 iterator(data, sdata->vif.addr, &sdata->vif);
 604 }
 605
 606 void ieee80211_iterate_active_interfaces(
 607         struct ieee80211_hw *hw, u32 iter_flags,
 608         void (*iterator)(void *data, u8 *mac,
 609                          struct ieee80211_vif *vif),
 610         void *data)
 611 {
 612         struct ieee80211_local *local = hw_to_local(hw);
 613
 614         mutex_lock(&local->iflist_mtx);
 615         __iterate_active_interfaces(local, iter_flags, iterator, data);
 616         mutex_unlock(&local->iflist_mtx);
 617 }
 618 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces);
 619
 620 void ieee80211_iterate_active_interfaces_atomic(
 621         struct ieee80211_hw *hw, u32 iter_flags,
 622         void (*iterator)(void *data, u8 *mac,
 623                          struct ieee80211_vif *vif),
 624         void *data)
 625 {
 626         struct ieee80211_local *local = hw_to_local(hw);
 627
 628         rcu_read_lock();
 629         __iterate_active_interfaces(local, iter_flags, iterator, data);
 630         rcu_read_unlock();
 631 }
 632 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_atomic);
 633
 634 void ieee80211_iterate_active_interfaces_rtnl(
 635         struct ieee80211_hw *hw, u32 iter_flags,
 636         void (*iterator)(void *data, u8 *mac,
 637                          struct ieee80211_vif *vif),
 638         void *data)
 639 {
 640         struct ieee80211_local *local = hw_to_local(hw);
 641
 642         ASSERT_RTNL();
 643
 644         __iterate_active_interfaces(local, iter_flags, iterator, data);
 645 }
 646 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_rtnl);
 647
 648 /*
 649  * Nothing should have been stuffed into the workqueue during
 650  * the suspend->resume cycle. If this WARN is seen then there
 651  * is a bug with either the driver suspend or something in
 652  * mac80211 stuffing into the workqueue which we haven't yet
 653  * cleared during mac80211's suspend cycle.
 654  */
 655 static bool ieee80211_can_queue_work(struct ieee80211_local *local)
 656 {
 657         if (WARN(local->suspended && !local->resuming,
 658                  "queueing ieee80211 work while going to suspend\n"))
 659                 return false;
 660
 661         return true;
 662 }
 663
 664 void ieee80211_queue_work(struct ieee80211_hw *hw, struct work_struct *work)
 665 {
 666         struct ieee80211_local *local = hw_to_local(hw);
 667
 668         if (!ieee80211_can_queue_work(local))
 669                 return;
 670
 671         queue_work(local->workqueue, work);
 672 }
 673 EXPORT_SYMBOL(ieee80211_queue_work);
 674
 675 void ieee80211_queue_delayed_work(struct ieee80211_hw *hw,
 676                                   struct delayed_work *dwork,
 677                                   unsigned long delay)
 678 {
 679         struct ieee80211_local *local = hw_to_local(hw);
 680
 681         if (!ieee80211_can_queue_work(local))
 682                 return;
 683
 684         queue_delayed_work(local->workqueue, dwork, delay);
 685 }
 686 EXPORT_SYMBOL(ieee80211_queue_delayed_work);
 687
 688 u32 ieee802_11_parse_elems_crc(const u8 *start, size_t len, bool action,
 689                                struct ieee802_11_elems *elems,
 690                                u64 filter, u32 crc)
 691 {
 692         size_t left = len;
 693         const u8 *pos = start;
 694         bool calc_crc = filter != 0;
 695         DECLARE_BITMAP(seen_elems, 256);
 696         const u8 *ie;
 697
 698         bitmap_zero(seen_elems, 256);
 699         memset(elems, 0, sizeof(*elems));
 700         elems->ie_start = start;
 701         elems->total_len = len;
 702
 703         while (left >= 2) {
 704                 u8 id, elen;
 705                 bool elem_parse_failed;
 706
 707                 id = *pos++;
 708                 elen = *pos++;
 709                 left -= 2;
 710
 711                 if (elen > left) {
 712                         elems->parse_error = true;
 713                         break;
 714                 }
 715
 716                 switch (id) {
 717                 case WLAN_EID_SSID:
 718                 case WLAN_EID_SUPP_RATES:
 719                 case WLAN_EID_FH_PARAMS:
 720                 case WLAN_EID_DS_PARAMS:
 721                 case WLAN_EID_CF_PARAMS:
 722                 case WLAN_EID_TIM:
 723                 case WLAN_EID_IBSS_PARAMS:
 724                 case WLAN_EID_CHALLENGE:
 725                 case WLAN_EID_RSN:
 726                 case WLAN_EID_ERP_INFO:
 727                 case WLAN_EID_EXT_SUPP_RATES:
 728                 case WLAN_EID_HT_CAPABILITY:
 729                 case WLAN_EID_HT_OPERATION:
 730                 case WLAN_EID_VHT_CAPABILITY:
 731                 case WLAN_EID_VHT_OPERATION:
 732                 case WLAN_EID_MESH_ID:
 733                 case WLAN_EID_MESH_CONFIG:
 734                 case WLAN_EID_PEER_MGMT:
 735                 case WLAN_EID_PREQ:
 736                 case WLAN_EID_PREP:
 737                 case WLAN_EID_PERR:
 738                 case WLAN_EID_RANN:
 739                 case WLAN_EID_CHANNEL_SWITCH:
 740                 case WLAN_EID_EXT_CHANSWITCH_ANN:
 741                 case WLAN_EID_COUNTRY:
 742                 case WLAN_EID_PWR_CONSTRAINT:
 743                 case WLAN_EID_TIMEOUT_INTERVAL:
 744                 case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
 745                 case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
 746                 /*
 747                  * not listing WLAN_EID_CHANNEL_SWITCH_WRAPPER -- it seems possible
 748                  * that if the content gets bigger it might be needed more than once
 749                  */
 750                         if (test_bit(id, seen_elems)) {
 751                                 elems->parse_error = true;
 752                                 left -= elen;
 753                                 pos += elen;
 754                                 continue;
 755                         }
 756                         break;
 757                 }
 758
 759                 if (calc_crc && id < 64 && (filter & (1ULL << id)))
 760                         crc = crc32_be(crc, pos - 2, elen + 2);
 761
 762                 elem_parse_failed = false;
 763
 764                 switch (id) {
 765                 case WLAN_EID_SSID:
 766                         elems->ssid = pos;
 767                         elems->ssid_len = elen;
 768                         break;
 769                 case WLAN_EID_SUPP_RATES:
 770                         elems->supp_rates = pos;
 771                         elems->supp_rates_len = elen;
 772                         break;
 773                 case WLAN_EID_DS_PARAMS:
 774                         if (elen >= 1)
 775                                 elems->ds_params = pos;
 776                         else
 777                                 elem_parse_failed = true;
 778                         break;
 779                 case WLAN_EID_TIM:
 780                         if (elen >= sizeof(struct ieee80211_tim_ie)) {
 781                                 elems->tim = (void *)pos;
 782                                 elems->tim_len = elen;
 783                         } else
 784                                 elem_parse_failed = true;
 785                         break;
 786                 case WLAN_EID_CHALLENGE:
 787                         elems->challenge = pos;
 788                         elems->challenge_len = elen;
 789                         break;
 790                 case WLAN_EID_VENDOR_SPECIFIC:
 791                         if (elen >= 4 && pos[0] == 0x00 && pos[1] == 0x50 &&
 792                             pos[2] == 0xf2) {
 793                                 /* Microsoft OUI (00:50:F2) */
 794
 795                                 if (calc_crc)
 796                                         crc = crc32_be(crc, pos - 2, elen + 2);
 797
 798                                 if (elen >= 5 && pos[3] == 2) {
 799                                         /* OUI Type 2 - WMM IE */
 800                                         if (pos[4] == 0) {
 801                                                 elems->wmm_info = pos;
 802                                                 elems->wmm_info_len = elen;
 803                                         } else if (pos[4] == 1) {
 804                                                 elems->wmm_param = pos;
 805                                                 elems->wmm_param_len = elen;
 806                                         }
 807                                 }
 808                         }
 809                         break;
 810                 case WLAN_EID_RSN:
 811                         elems->rsn = pos;
 812                         elems->rsn_len = elen;
 813                         break;
 814                 case WLAN_EID_ERP_INFO:
 815                         if (elen >= 1)
 816                                 elems->erp_info = pos;
 817                         else
 818                                 elem_parse_failed = true;
 819                         break;
 820                 case WLAN_EID_EXT_SUPP_RATES:
 821                         elems->ext_supp_rates = pos;
 822                         elems->ext_supp_rates_len = elen;
 823                         break;
 824                 case WLAN_EID_HT_CAPABILITY:
 825                         if (elen >= sizeof(struct ieee80211_ht_cap))
 826                                 elems->ht_cap_elem = (void *)pos;
 827                         else
 828                                 elem_parse_failed = true;
 829                         break;
 830                 case WLAN_EID_HT_OPERATION:
 831                         if (elen >= sizeof(struct ieee80211_ht_operation))
 832                                 elems->ht_operation = (void *)pos;
 833                         else
 834                                 elem_parse_failed = true;
 835                         break;
 836                 case WLAN_EID_VHT_CAPABILITY:
 837                         if (elen >= sizeof(struct ieee80211_vht_cap))
 838                                 elems->vht_cap_elem = (void *)pos;
 839                         else
 840                                 elem_parse_failed = true;
 841                         break;
 842                 case WLAN_EID_VHT_OPERATION:
 843                         if (elen >= sizeof(struct ieee80211_vht_operation))
 844                                 elems->vht_operation = (void *)pos;
 845                         else
 846                                 elem_parse_failed = true;
 847                         break;
 848                 case WLAN_EID_OPMODE_NOTIF:
 849                         if (elen > 0)
 850                                 elems->opmode_notif = pos;
 851                         else
 852                                 elem_parse_failed = true;
 853                         break;
 854                 case WLAN_EID_MESH_ID:
 855                         elems->mesh_id = pos;
 856                         elems->mesh_id_len = elen;
 857                         break;
 858                 case WLAN_EID_MESH_CONFIG:
 859                         if (elen >= sizeof(struct ieee80211_meshconf_ie))
 860                                 elems->mesh_config = (void *)pos;
 861                         else
 862                                 elem_parse_failed = true;
 863                         break;
 864                 case WLAN_EID_PEER_MGMT:
 865                         elems->peering = pos;
 866                         elems->peering_len = elen;
 867                         break;
 868                 case WLAN_EID_MESH_AWAKE_WINDOW:
 869                         if (elen >= 2)
 870                                 elems->awake_window = (void *)pos;
 871                         break;
 872                 case WLAN_EID_PREQ:
 873                         elems->preq = pos;
 874                         elems->preq_len = elen;
 875                         break;
 876                 case WLAN_EID_PREP:
 877                         elems->prep = pos;
 878                         elems->prep_len = elen;
 879                         break;
 880                 case WLAN_EID_PERR:
 881                         elems->perr = pos;
 882                         elems->perr_len = elen;
 883                         break;
 884                 case WLAN_EID_RANN:
 885                         if (elen >= sizeof(struct ieee80211_rann_ie))
 886                                 elems->rann = (void *)pos;
 887                         else
 888                                 elem_parse_failed = true;
 889                         break;
 890                 case WLAN_EID_CHANNEL_SWITCH:
 891                         if (elen != sizeof(struct ieee80211_channel_sw_ie)) {
 892                                 elem_parse_failed = true;
 893                                 break;
 894                         }
 895                         elems->ch_switch_ie = (void *)pos;
 896                         break;
 897                 case WLAN_EID_EXT_CHANSWITCH_ANN:
 898                         if (elen != sizeof(struct ieee80211_ext_chansw_ie)) {
 899                                 elem_parse_failed = true;
 900                                 break;
 901                         }
 902                         elems->ext_chansw_ie = (void *)pos;
 903                         break;
 904                 case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
 905                         if (elen != sizeof(struct ieee80211_sec_chan_offs_ie)) {
 906                                 elem_parse_failed = true;
 907                                 break;
 908                         }
 909                         elems->sec_chan_offs = (void *)pos;
 910                         break;
 911                 case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
 912                         if (!action ||
 913                             elen != sizeof(*elems->wide_bw_chansw_ie)) {
 914                                 elem_parse_failed = true;
 915                                 break;
 916                         }
 917                         elems->wide_bw_chansw_ie = (void *)pos;
 918                         break;
 919                 case WLAN_EID_CHANNEL_SWITCH_WRAPPER:
 920                         if (action) {
 921                                 elem_parse_failed = true;
 922                                 break;
 923                         }
 924                         /*
 925                          * This is a bit tricky, but as we only care about
 926                          * the wide bandwidth channel switch element, so
 927                          * just parse it out manually.
 928                          */
 929                         ie = cfg80211_find_ie(WLAN_EID_WIDE_BW_CHANNEL_SWITCH,
 930                                               pos, elen);
 931                         if (ie) {
 932                                 if (ie[1] == sizeof(*elems->wide_bw_chansw_ie))
 933                                         elems->wide_bw_chansw_ie =
 934                                                 (void *)(ie + 2);
 935                                 else
 936                                         elem_parse_failed = true;
 937                         }
 938                         break;
 939                 case WLAN_EID_COUNTRY:
 940                         elems->country_elem = pos;
 941                         elems->country_elem_len = elen;
 942                         break;
 943                 case WLAN_EID_PWR_CONSTRAINT:
 944                         if (elen != 1) {
 945                                 elem_parse_failed = true;
 946                                 break;
 947                         }
 948                         elems->pwr_constr_elem = pos;
 949                         break;
 950                 case WLAN_EID_TIMEOUT_INTERVAL:
 951                         if (elen >= sizeof(struct ieee80211_timeout_interval_ie))
 952                                 elems->timeout_int = (void *)pos;
 953                         else
 954                                 elem_parse_failed = true;
 955                         break;
 956                 default:
 957                         break;
 958                 }
 959
 960                 if (elem_parse_failed)
 961                         elems->parse_error = true;
 962                 else
 963                         __set_bit(id, seen_elems);
 964
 965                 left -= elen;
 966                 pos += elen;
 967         }
 968
 969         if (left != 0)
 970                 elems->parse_error = true;
 971
 972         return crc;
 973 }
 974
 975 void ieee80211_set_wmm_default(struct ieee80211_sub_if_data *sdata,
 976                                bool bss_notify)
 977 {
 978         struct ieee80211_local *local = sdata->local;
 979         struct ieee80211_tx_queue_params qparam;
 980         struct ieee80211_chanctx_conf *chanctx_conf;
 981         int ac;
 982         bool use_11b, enable_qos;
 983         int aCWmin, aCWmax;
 984
 985         if (!local->ops->conf_tx)
 986                 return;
 987
 988         if (local->hw.queues < IEEE80211_NUM_ACS)
 989                 return;
 990
 991         memset(&qparam, 0, sizeof(qparam));
 992
 993         rcu_read_lock();
 994         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
 995         use_11b = (chanctx_conf &&
 996                    chanctx_conf->def.chan->band == IEEE80211_BAND_2GHZ) &&
 997                  !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE);
 998         rcu_read_unlock();
 999
1000         /*
1001          * By default disable QoS in STA mode for old access points, which do
1002          * not support 802.11e. New APs will provide proper queue parameters,
1003          * that we will configure later.
1004          */
1005         enable_qos = (sdata->vif.type != NL80211_IFTYPE_STATION);
1006
1007         /* Set defaults according to 802.11-2007 Table 7-37 */
1008         aCWmax = 1023;
1009         if (use_11b)
1010                 aCWmin = 31;
1011         else
1012                 aCWmin = 15;
1013
1014         /* Confiure old 802.11b/g medium access rules. */
1015         qparam.cw_max = aCWmax;
1016         qparam.cw_min = aCWmin;
1017         qparam.txop = 0;
1018         qparam.aifs = 2;
1019
1020         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1021                 /* Update if QoS is enabled. */
1022                 if (enable_qos) {
1023                         switch (ac) {
1024                         case IEEE80211_AC_BK:
1025                                 qparam.cw_max = aCWmax;
1026                                 qparam.cw_min = aCWmin;
1027                                 qparam.txop = 0;
1028                                 qparam.aifs = 7;
1029                                 break;
1030                         /* never happens but let's not leave undefined */
1031                         default:
1032                         case IEEE80211_AC_BE:
1033                                 qparam.cw_max = aCWmax;
1034                                 qparam.cw_min = aCWmin;
1035                                 qparam.txop = 0;
1036                                 qparam.aifs = 3;
1037                                 break;
1038                         case IEEE80211_AC_VI:
1039                                 qparam.cw_max = aCWmin;
1040                                 qparam.cw_min = (aCWmin + 1) / 2 - 1;
1041                                 if (use_11b)
1042                                         qparam.txop = 6016/32;
1043                                 else
1044                                         qparam.txop = 3008/32;
1045                                 qparam.aifs = 2;
1046                                 break;
1047                         case IEEE80211_AC_VO:
1048                                 qparam.cw_max = (aCWmin + 1) / 2 - 1;
1049                                 qparam.cw_min = (aCWmin + 1) / 4 - 1;
1050                                 if (use_11b)
1051                                         qparam.txop = 3264/32;
1052                                 else
1053                                         qparam.txop = 1504/32;
1054                                 qparam.aifs = 2;
1055                                 break;
1056                         }
1057                 }
1058
1059                 qparam.uapsd = false;
1060
1061                 sdata->tx_conf[ac] = qparam;
1062                 drv_conf_tx(local, sdata, ac, &qparam);
1063         }
1064
1065         if (sdata->vif.type != NL80211_IFTYPE_MONITOR &&
1066             sdata->vif.type != NL80211_IFTYPE_P2P_DEVICE) {
1067                 sdata->vif.bss_conf.qos = enable_qos;
1068                 if (bss_notify)
1069                         ieee80211_bss_info_change_notify(sdata,
1070                                                          BSS_CHANGED_QOS);
1071         }
1072 }
1073
1074 void ieee80211_send_auth(struct ieee80211_sub_if_data *sdata,
1075                          u16 transaction, u16 auth_alg, u16 status,
1076                          const u8 *extra, size_t extra_len, const u8 *da,
1077                          const u8 *bssid, const u8 *key, u8 key_len, u8 key_idx,
1078                          u32 tx_flags)
1079 {
1080         struct ieee80211_local *local = sdata->local;
1081         struct sk_buff *skb;
1082         struct ieee80211_mgmt *mgmt;
1083         int err;
1084
1085         /* 24 + 6 = header + auth_algo + auth_transaction + status_code */
1086         skb = dev_alloc_skb(local->hw.extra_tx_headroom + 24 + 6 + extra_len);
1087         if (!skb)
1088                 return;
1089
1090         skb_reserve(skb, local->hw.extra_tx_headroom);
1091
1092         mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24 + 6);
1093         memset(mgmt, 0, 24 + 6);
1094         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1095                                           IEEE80211_STYPE_AUTH);
1096         memcpy(mgmt->da, da, ETH_ALEN);
1097         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1098         memcpy(mgmt->bssid, bssid, ETH_ALEN);
1099         mgmt->u.auth.auth_alg = cpu_to_le16(auth_alg);
1100         mgmt->u.auth.auth_transaction = cpu_to_le16(transaction);
1101         mgmt->u.auth.status_code = cpu_to_le16(status);
1102         if (extra)
1103                 memcpy(skb_put(skb, extra_len), extra, extra_len);
1104
1105         if (auth_alg == WLAN_AUTH_SHARED_KEY && transaction == 3) {
1106                 mgmt->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
1107                 err = ieee80211_wep_encrypt(local, skb, key, key_len, key_idx);
1108                 WARN_ON(err);
1109         }
1110
1111         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1112                                         tx_flags;
1113         ieee80211_tx_skb(sdata, skb);
1114 }
1115
1116 void ieee80211_send_deauth_disassoc(struct ieee80211_sub_if_data *sdata,
1117                                     const u8 *bssid, u16 stype, u16 reason,
1118                                     bool send_frame, u8 *frame_buf)
1119 {
1120         struct ieee80211_local *local = sdata->local;
1121         struct sk_buff *skb;
1122         struct ieee80211_mgmt *mgmt = (void *)frame_buf;
1123
1124         /* build frame */
1125         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT | stype);
1126         mgmt->duration = 0; /* initialize only */
1127         mgmt->seq_ctrl = 0; /* initialize only */
1128         memcpy(mgmt->da, bssid, ETH_ALEN);
1129         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1130         memcpy(mgmt->bssid, bssid, ETH_ALEN);
1131         /* u.deauth.reason_code == u.disassoc.reason_code */
1132         mgmt->u.deauth.reason_code = cpu_to_le16(reason);
1133
1134         if (send_frame) {
1135                 skb = dev_alloc_skb(local->hw.extra_tx_headroom +
1136                                     IEEE80211_DEAUTH_FRAME_LEN);
1137                 if (!skb)
1138                         return;
1139
1140                 skb_reserve(skb, local->hw.extra_tx_headroom);
1141
1142                 /* copy in frame */
1143                 memcpy(skb_put(skb, IEEE80211_DEAUTH_FRAME_LEN),
1144                        mgmt, IEEE80211_DEAUTH_FRAME_LEN);
1145
1146                 if (sdata->vif.type != NL80211_IFTYPE_STATION ||
1147                     !(sdata->u.mgd.flags & IEEE80211_STA_MFP_ENABLED))
1148                         IEEE80211_SKB_CB(skb)->flags |=
1149                                 IEEE80211_TX_INTFL_DONT_ENCRYPT;
1150
1151                 ieee80211_tx_skb(sdata, skb);
1152         }
1153 }
1154
1155 int ieee80211_build_preq_ies(struct ieee80211_local *local, u8 *buffer,
1156                              size_t buffer_len, const u8 *ie, size_t ie_len,
1157                              enum ieee80211_band band, u32 rate_mask,
1158                              struct cfg80211_chan_def *chandef)
1159 {
1160         struct ieee80211_supported_band *sband;
1161         u8 *pos = buffer, *end = buffer + buffer_len;
1162         size_t offset = 0, noffset;
1163         int supp_rates_len, i;
1164         u8 rates[32];
1165         int num_rates;
1166         int ext_rates_len;
1167         int shift;
1168         u32 rate_flags;
1169
1170         sband = local->hw.wiphy->bands[band];
1171         if (WARN_ON_ONCE(!sband))
1172                 return 0;
1173
1174         rate_flags = ieee80211_chandef_rate_flags(chandef);
1175         shift = ieee80211_chandef_get_shift(chandef);
1176
1177         num_rates = 0;
1178         for (i = 0; i < sband->n_bitrates; i++) {
1179                 if ((BIT(i) & rate_mask) == 0)
1180                         continue; /* skip rate */
1181                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
1182                         continue;
1183
1184                 rates[num_rates++] =
1185                         (u8) DIV_ROUND_UP(sband->bitrates[i].bitrate,
1186                                           (1 << shift) * 5);
1187         }
1188
1189         supp_rates_len = min_t(int, num_rates, 8);
1190
1191         if (end - pos < 2 + supp_rates_len)
1192                 goto out_err;
1193         *pos++ = WLAN_EID_SUPP_RATES;
1194         *pos++ = supp_rates_len;
1195         memcpy(pos, rates, supp_rates_len);
1196         pos += supp_rates_len;
1197
1198         /* insert "request information" if in custom IEs */
1199         if (ie && ie_len) {
1200                 static const u8 before_extrates[] = {
1201                         WLAN_EID_SSID,
1202                         WLAN_EID_SUPP_RATES,
1203                         WLAN_EID_REQUEST,
1204                 };
1205                 noffset = ieee80211_ie_split(ie, ie_len,
1206                                              before_extrates,
1207                                              ARRAY_SIZE(before_extrates),
1208                                              offset);
1209                 if (end - pos < noffset - offset)
1210                         goto out_err;
1211                 memcpy(pos, ie + offset, noffset - offset);
1212                 pos += noffset - offset;
1213                 offset = noffset;
1214         }
1215
1216         ext_rates_len = num_rates - supp_rates_len;
1217         if (ext_rates_len > 0) {
1218                 if (end - pos < 2 + ext_rates_len)
1219                         goto out_err;
1220                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
1221                 *pos++ = ext_rates_len;
1222                 memcpy(pos, rates + supp_rates_len, ext_rates_len);
1223                 pos += ext_rates_len;
1224         }
1225
1226         if (chandef->chan && sband->band == IEEE80211_BAND_2GHZ) {
1227                 if (end - pos < 3)
1228                         goto out_err;
1229                 *pos++ = WLAN_EID_DS_PARAMS;
1230                 *pos++ = 1;
1231                 *pos++ = ieee80211_frequency_to_channel(
1232                                 chandef->chan->center_freq);
1233         }
1234
1235         /* insert custom IEs that go before HT */
1236         if (ie && ie_len) {
1237                 static const u8 before_ht[] = {
1238                         WLAN_EID_SSID,
1239                         WLAN_EID_SUPP_RATES,
1240                         WLAN_EID_REQUEST,
1241                         WLAN_EID_EXT_SUPP_RATES,
1242                         WLAN_EID_DS_PARAMS,
1243                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
1244                 };
1245                 noffset = ieee80211_ie_split(ie, ie_len,
1246                                              before_ht, ARRAY_SIZE(before_ht),
1247                                              offset);
1248                 if (end - pos < noffset - offset)
1249                         goto out_err;
1250                 memcpy(pos, ie + offset, noffset - offset);
1251                 pos += noffset - offset;
1252                 offset = noffset;
1253         }
1254
1255         if (sband->ht_cap.ht_supported) {
1256                 if (end - pos < 2 + sizeof(struct ieee80211_ht_cap))
1257                         goto out_err;
1258                 pos = ieee80211_ie_build_ht_cap(pos, &sband->ht_cap,
1259                                                 sband->ht_cap.cap);
1260         }
1261
1262         /*
1263          * If adding more here, adjust code in main.c
1264          * that calculates local->scan_ies_len.
1265          */
1266
1267         /* add any remaining custom IEs */
1268         if (ie && ie_len) {
1269                 noffset = ie_len;
1270                 if (end - pos < noffset - offset)
1271                         goto out_err;
1272                 memcpy(pos, ie + offset, noffset - offset);
1273                 pos += noffset - offset;
1274         }
1275
1276         if (sband->vht_cap.vht_supported) {
1277                 if (end - pos < 2 + sizeof(struct ieee80211_vht_cap))
1278                         goto out_err;
1279                 pos = ieee80211_ie_build_vht_cap(pos, &sband->vht_cap,
1280                                                  sband->vht_cap.cap);
1281         }
1282
1283         return pos - buffer;
1284  out_err:
1285         WARN_ONCE(1, "not enough space for preq IEs\n");
1286         return pos - buffer;
1287 }
1288
1289 struct sk_buff *ieee80211_build_probe_req(struct ieee80211_sub_if_data *sdata,
1290                                           u8 *dst, u32 ratemask,
1291                                           struct ieee80211_channel *chan,
1292                                           const u8 *ssid, size_t ssid_len,
1293                                           const u8 *ie, size_t ie_len,
1294                                           bool directed)
1295 {
1296         struct ieee80211_local *local = sdata->local;
1297         struct cfg80211_chan_def chandef;
1298         struct sk_buff *skb;
1299         struct ieee80211_mgmt *mgmt;
1300         int ies_len;
1301
1302         /*
1303          * Do not send DS Channel parameter for directed probe requests
1304          * in order to maximize the chance that we get a response.  Some
1305          * badly-behaved APs don't respond when this parameter is included.
1306          */
1307         chandef.width = sdata->vif.bss_conf.chandef.width;
1308         if (directed)
1309                 chandef.chan = NULL;
1310         else
1311                 chandef.chan = chan;
1312
1313         skb = ieee80211_probereq_get(&local->hw, &sdata->vif,
1314                                      ssid, ssid_len, 100 + ie_len);
1315         if (!skb)
1316                 return NULL;
1317
1318         ies_len = ieee80211_build_preq_ies(local, skb_tail_pointer(skb),
1319                                            skb_tailroom(skb),
1320                                            ie, ie_len, chan->band,
1321                                            ratemask, &chandef);
1322         skb_put(skb, ies_len);
1323
1324         if (dst) {
1325                 mgmt = (struct ieee80211_mgmt *) skb->data;
1326                 memcpy(mgmt->da, dst, ETH_ALEN);
1327                 memcpy(mgmt->bssid, dst, ETH_ALEN);
1328         }
1329
1330         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1331
1332         return skb;
1333 }
1334
1335 void ieee80211_send_probe_req(struct ieee80211_sub_if_data *sdata, u8 *dst,
1336                               const u8 *ssid, size_t ssid_len,
1337                               const u8 *ie, size_t ie_len,
1338                               u32 ratemask, bool directed, u32 tx_flags,
1339                               struct ieee80211_channel *channel, bool scan)
1340 {
1341         struct sk_buff *skb;
1342
1343         skb = ieee80211_build_probe_req(sdata, dst, ratemask, channel,
1344                                         ssid, ssid_len,
1345                                         ie, ie_len, directed);
1346         if (skb) {
1347                 IEEE80211_SKB_CB(skb)->flags |= tx_flags;
1348                 if (scan)
1349                         ieee80211_tx_skb_tid_band(sdata, skb, 7, channel->band);
1350                 else
1351                         ieee80211_tx_skb(sdata, skb);
1352         }
1353 }
1354
1355 u32 ieee80211_sta_get_rates(struct ieee80211_sub_if_data *sdata,
1356                             struct ieee802_11_elems *elems,
1357                             enum ieee80211_band band, u32 *basic_rates)
1358 {
1359         struct ieee80211_supported_band *sband;
1360         struct ieee80211_rate *bitrates;
1361         size_t num_rates;
1362         u32 supp_rates, rate_flags;
1363         int i, j, shift;
1364         sband = sdata->local->hw.wiphy->bands[band];
1365
1366         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
1367         shift = ieee80211_vif_get_shift(&sdata->vif);
1368
1369         if (WARN_ON(!sband))
1370                 return 1;
1371
1372         bitrates = sband->bitrates;
1373         num_rates = sband->n_bitrates;
1374         supp_rates = 0;
1375         for (i = 0; i < elems->supp_rates_len +
1376                      elems->ext_supp_rates_len; i++) {
1377                 u8 rate = 0;
1378                 int own_rate;
1379                 bool is_basic;
1380                 if (i < elems->supp_rates_len)
1381                         rate = elems->supp_rates[i];
1382                 else if (elems->ext_supp_rates)
1383                         rate = elems->ext_supp_rates
1384                                 [i - elems->supp_rates_len];
1385                 own_rate = 5 * (rate & 0x7f);
1386                 is_basic = !!(rate & 0x80);
1387
1388                 if (is_basic && (rate & 0x7f) == BSS_MEMBERSHIP_SELECTOR_HT_PHY)
1389                         continue;
1390
1391                 for (j = 0; j < num_rates; j++) {
1392                         int brate;
1393                         if ((rate_flags & sband->bitrates[j].flags)
1394                             != rate_flags)
1395                                 continue;
1396
1397                         brate = DIV_ROUND_UP(sband->bitrates[j].bitrate,
1398                                              1 << shift);
1399
1400                         if (brate == own_rate) {
1401                                 supp_rates |= BIT(j);
1402                                 if (basic_rates && is_basic)
1403                                         *basic_rates |= BIT(j);
1404                         }
1405                 }
1406         }
1407         return supp_rates;
1408 }
1409
1410 void ieee80211_stop_device(struct ieee80211_local *local)
1411 {
1412         ieee80211_led_radio(local, false);
1413         ieee80211_mod_tpt_led_trig(local, 0, IEEE80211_TPT_LEDTRIG_FL_RADIO);
1414
1415         cancel_work_sync(&local->reconfig_filter);
1416
1417         flush_workqueue(local->workqueue);
1418         drv_stop(local);
1419 }
1420
1421 static void ieee80211_assign_chanctx(struct ieee80211_local *local,
1422                                      struct ieee80211_sub_if_data *sdata)
1423 {
1424         struct ieee80211_chanctx_conf *conf;
1425         struct ieee80211_chanctx *ctx;
1426
1427         if (!local->use_chanctx)
1428                 return;
1429
1430         mutex_lock(&local->chanctx_mtx);
1431         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1432                                          lockdep_is_held(&local->chanctx_mtx));
1433         if (conf) {
1434                 ctx = container_of(conf, struct ieee80211_chanctx, conf);
1435                 drv_assign_vif_chanctx(local, sdata, ctx);
1436         }
1437         mutex_unlock(&local->chanctx_mtx);
1438 }
1439
1440 int ieee80211_reconfig(struct ieee80211_local *local)
1441 {
1442         struct ieee80211_hw *hw = &local->hw;
1443         struct ieee80211_sub_if_data *sdata;
1444         struct ieee80211_chanctx *ctx;
1445         struct sta_info *sta;
1446         int res, i;
1447         bool reconfig_due_to_wowlan = false;
1448
1449 #ifdef CONFIG_PM
1450         if (local->suspended)
1451                 local->resuming = true;
1452
1453         if (local->wowlan) {
1454                 res = drv_resume(local);
1455                 local->wowlan = false;
1456                 if (res < 0) {
1457                         local->resuming = false;
1458                         return res;
1459                 }
1460                 if (res == 0)
1461                         goto wake_up;
1462                 WARN_ON(res > 1);
1463                 /*
1464                  * res is 1, which means the driver requested
1465                  * to go through a regular reset on wakeup.
1466                  */
1467                 reconfig_due_to_wowlan = true;
1468         }
1469 #endif
1470         /* everything else happens only if HW was up & running */
1471         if (!local->open_count)
1472                 goto wake_up;
1473
1474         /*
1475          * Upon resume hardware can sometimes be goofy due to
1476          * various platform / driver / bus issues, so restarting
1477          * the device may at times not work immediately. Propagate
1478          * the error.
1479          */
1480         res = drv_start(local);
1481         if (res) {
1482                 WARN(local->suspended, "Hardware became unavailable "
1483                      "upon resume. This could be a software issue "
1484                      "prior to suspend or a hardware issue.\n");
1485                 return res;
1486         }
1487
1488         /* setup fragmentation threshold */
1489         drv_set_frag_threshold(local, hw->wiphy->frag_threshold);
1490
1491         /* setup RTS threshold */
1492         drv_set_rts_threshold(local, hw->wiphy->rts_threshold);
1493
1494         /* reset coverage class */
1495         drv_set_coverage_class(local, hw->wiphy->coverage_class);
1496
1497         ieee80211_led_radio(local, true);
1498         ieee80211_mod_tpt_led_trig(local,
1499                                    IEEE80211_TPT_LEDTRIG_FL_RADIO, 0);
1500
1501         /* add interfaces */
1502         sdata = rtnl_dereference(local->monitor_sdata);
1503         if (sdata) {
1504                 /* in HW restart it exists already */
1505                 WARN_ON(local->resuming);
1506                 res = drv_add_interface(local, sdata);
1507                 if (WARN_ON(res)) {
1508                         rcu_assign_pointer(local->monitor_sdata, NULL);
1509                         synchronize_net();
1510                         kfree(sdata);
1511                 }
1512         }
1513
1514         list_for_each_entry(sdata, &local->interfaces, list) {
1515                 if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
1516                     sdata->vif.type != NL80211_IFTYPE_MONITOR &&
1517                     ieee80211_sdata_running(sdata))
1518                         res = drv_add_interface(local, sdata);
1519         }
1520
1521         /* add channel contexts */
1522         if (local->use_chanctx) {
1523                 mutex_lock(&local->chanctx_mtx);
1524                 list_for_each_entry(ctx, &local->chanctx_list, list)
1525                         WARN_ON(drv_add_chanctx(local, ctx));
1526                 mutex_unlock(&local->chanctx_mtx);
1527         }
1528
1529         list_for_each_entry(sdata, &local->interfaces, list) {
1530                 if (!ieee80211_sdata_running(sdata))
1531                         continue;
1532                 ieee80211_assign_chanctx(local, sdata);
1533         }
1534
1535         sdata = rtnl_dereference(local->monitor_sdata);
1536         if (sdata && ieee80211_sdata_running(sdata))
1537                 ieee80211_assign_chanctx(local, sdata);
1538
1539         /* add STAs back */
1540         mutex_lock(&local->sta_mtx);
1541         list_for_each_entry(sta, &local->sta_list, list) {
1542                 enum ieee80211_sta_state state;
1543
1544                 if (!sta->uploaded)
1545                         continue;
1546
1547                 /* AP-mode stations will be added later */
1548                 if (sta->sdata->vif.type == NL80211_IFTYPE_AP)
1549                         continue;
1550
1551                 for (state = IEEE80211_STA_NOTEXIST;
1552                      state < sta->sta_state; state++)
1553                         WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
1554                                               state + 1));
1555         }
1556         mutex_unlock(&local->sta_mtx);
1557
1558         /* reconfigure tx conf */
1559         if (hw->queues >= IEEE80211_NUM_ACS) {
1560                 list_for_each_entry(sdata, &local->interfaces, list) {
1561                         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN ||
1562                             sdata->vif.type == NL80211_IFTYPE_MONITOR ||
1563                             !ieee80211_sdata_running(sdata))
1564                                 continue;
1565
1566                         for (i = 0; i < IEEE80211_NUM_ACS; i++)
1567                                 drv_conf_tx(local, sdata, i,
1568                                             &sdata->tx_conf[i]);
1569                 }
1570         }
1571
1572         /* reconfigure hardware */
1573         ieee80211_hw_config(local, ~0);
1574
1575         ieee80211_configure_filter(local);
1576
1577         /* Finally also reconfigure all the BSS information */
1578         list_for_each_entry(sdata, &local->interfaces, list) {
1579                 u32 changed;
1580
1581                 if (!ieee80211_sdata_running(sdata))
1582                         continue;
1583
1584                 /* common change flags for all interface types */
1585                 changed = BSS_CHANGED_ERP_CTS_PROT |
1586                           BSS_CHANGED_ERP_PREAMBLE |
1587                           BSS_CHANGED_ERP_SLOT |
1588                           BSS_CHANGED_HT |
1589                           BSS_CHANGED_BASIC_RATES |
1590                           BSS_CHANGED_BEACON_INT |
1591                           BSS_CHANGED_BSSID |
1592                           BSS_CHANGED_CQM |
1593                           BSS_CHANGED_QOS |
1594                           BSS_CHANGED_IDLE |
1595                           BSS_CHANGED_TXPOWER;
1596
1597                 switch (sdata->vif.type) {
1598                 case NL80211_IFTYPE_STATION:
1599                         changed |= BSS_CHANGED_ASSOC |
1600                                    BSS_CHANGED_ARP_FILTER |
1601                                    BSS_CHANGED_PS;
1602
1603                         /* Re-send beacon info report to the driver */
1604                         if (sdata->u.mgd.have_beacon)
1605                                 changed |= BSS_CHANGED_BEACON_INFO;
1606
1607                         sdata_lock(sdata);
1608                         ieee80211_bss_info_change_notify(sdata, changed);
1609                         sdata_unlock(sdata);
1610                         break;
1611                 case NL80211_IFTYPE_ADHOC:
1612                         changed |= BSS_CHANGED_IBSS;
1613                         /* fall through */
1614                 case NL80211_IFTYPE_AP:
1615                         changed |= BSS_CHANGED_SSID | BSS_CHANGED_P2P_PS;
1616
1617                         if (sdata->vif.type == NL80211_IFTYPE_AP) {
1618                                 changed |= BSS_CHANGED_AP_PROBE_RESP;
1619
1620                                 if (rcu_access_pointer(sdata->u.ap.beacon))
1621                                         drv_start_ap(local, sdata);
1622                         }
1623
1624                         /* fall through */
1625                 case NL80211_IFTYPE_MESH_POINT:
1626                         if (sdata->vif.bss_conf.enable_beacon) {
1627                                 changed |= BSS_CHANGED_BEACON |
1628                                            BSS_CHANGED_BEACON_ENABLED;
1629                                 ieee80211_bss_info_change_notify(sdata, changed);
1630                         }
1631                         break;
1632                 case NL80211_IFTYPE_WDS:
1633                         break;
1634                 case NL80211_IFTYPE_AP_VLAN:
1635                 case NL80211_IFTYPE_MONITOR:
1636                         /* ignore virtual */
1637                         break;
1638                 case NL80211_IFTYPE_P2P_DEVICE:
1639                         changed = BSS_CHANGED_IDLE;
1640                         break;
1641                 case NL80211_IFTYPE_UNSPECIFIED:
1642                 case NUM_NL80211_IFTYPES:
1643                 case NL80211_IFTYPE_P2P_CLIENT:
1644                 case NL80211_IFTYPE_P2P_GO:
1645                         WARN_ON(1);
1646                         break;
1647                 }
1648         }
1649
1650         ieee80211_recalc_ps(local, -1);
1651
1652         /*
1653          * The sta might be in psm against the ap (e.g. because
1654          * this was the state before a hw restart), so we
1655          * explicitly send a null packet in order to make sure
1656          * it'll sync against the ap (and get out of psm).
1657          */
1658         if (!(local->hw.conf.flags & IEEE80211_CONF_PS)) {
1659                 list_for_each_entry(sdata, &local->interfaces, list) {
1660                         if (sdata->vif.type != NL80211_IFTYPE_STATION)
1661                                 continue;
1662                         if (!sdata->u.mgd.associated)
1663                                 continue;
1664
1665                         ieee80211_send_nullfunc(local, sdata, 0);
1666                 }
1667         }
1668
1669         /* APs are now beaconing, add back stations */
1670         mutex_lock(&local->sta_mtx);
1671         list_for_each_entry(sta, &local->sta_list, list) {
1672                 enum ieee80211_sta_state state;
1673
1674                 if (!sta->uploaded)
1675                         continue;
1676
1677                 if (sta->sdata->vif.type != NL80211_IFTYPE_AP)
1678                         continue;
1679
1680                 for (state = IEEE80211_STA_NOTEXIST;
1681                      state < sta->sta_state; state++)
1682                         WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
1683                                               state + 1));
1684         }
1685         mutex_unlock(&local->sta_mtx);
1686
1687         /* add back keys */
1688         list_for_each_entry(sdata, &local->interfaces, list)
1689                 if (ieee80211_sdata_running(sdata))
1690                         ieee80211_enable_keys(sdata);
1691
1692  wake_up:
1693         local->in_reconfig = false;
1694         barrier();
1695
1696         if (local->monitors == local->open_count && local->monitors > 0)
1697                 ieee80211_add_virtual_monitor(local);
1698
1699         /*
1700          * Clear the WLAN_STA_BLOCK_BA flag so new aggregation
1701          * sessions can be established after a resume.
1702          *
1703          * Also tear down aggregation sessions since reconfiguring
1704          * them in a hardware restart scenario is not easily done
1705          * right now, and the hardware will have lost information
1706          * about the sessions, but we and the AP still think they
1707          * are active. This is really a workaround though.
1708          */
1709         if (hw->flags & IEEE80211_HW_AMPDU_AGGREGATION) {
1710                 mutex_lock(&local->sta_mtx);
1711
1712                 list_for_each_entry(sta, &local->sta_list, list) {
1713                         ieee80211_sta_tear_down_BA_sessions(
1714                                         sta, AGG_STOP_LOCAL_REQUEST);
1715                         clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
1716                 }
1717
1718                 mutex_unlock(&local->sta_mtx);
1719         }
1720
1721         ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
1722                                         IEEE80211_QUEUE_STOP_REASON_SUSPEND);
1723
1724         /*
1725          * If this is for hw restart things are still running.
1726          * We may want to change that later, however.
1727          */
1728         if (!local->suspended || reconfig_due_to_wowlan)
1729                 drv_restart_complete(local);
1730
1731         if (!local->suspended)
1732                 return 0;
1733
1734 #ifdef CONFIG_PM
1735         /* first set suspended false, then resuming */
1736         local->suspended = false;
1737         mb();
1738         local->resuming = false;
1739
1740         list_for_each_entry(sdata, &local->interfaces, list) {
1741                 if (!ieee80211_sdata_running(sdata))
1742                         continue;
1743                 if (sdata->vif.type == NL80211_IFTYPE_STATION)
1744                         ieee80211_sta_restart(sdata);
1745         }
1746
1747         mod_timer(&local->sta_cleanup, jiffies + 1);
1748 #else
1749         WARN_ON(1);
1750 #endif
1751         return 0;
1752 }
1753
1754 void ieee80211_resume_disconnect(struct ieee80211_vif *vif)
1755 {
1756         struct ieee80211_sub_if_data *sdata;
1757         struct ieee80211_local *local;
1758         struct ieee80211_key *key;
1759
1760         if (WARN_ON(!vif))
1761                 return;
1762
1763         sdata = vif_to_sdata(vif);
1764         local = sdata->local;
1765
1766         if (WARN_ON(!local->resuming))
1767                 return;
1768
1769         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
1770                 return;
1771
1772         sdata->flags |= IEEE80211_SDATA_DISCONNECT_RESUME;
1773
1774         mutex_lock(&local->key_mtx);
1775         list_for_each_entry(key, &sdata->key_list, list)
1776                 key->flags |= KEY_FLAG_TAINTED;
1777         mutex_unlock(&local->key_mtx);
1778 }
1779 EXPORT_SYMBOL_GPL(ieee80211_resume_disconnect);
1780
1781 void ieee80211_recalc_smps(struct ieee80211_sub_if_data *sdata)
1782 {
1783         struct ieee80211_local *local = sdata->local;
1784         struct ieee80211_chanctx_conf *chanctx_conf;
1785         struct ieee80211_chanctx *chanctx;
1786
1787         mutex_lock(&local->chanctx_mtx);
1788
1789         chanctx_conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1790                                         lockdep_is_held(&local->chanctx_mtx));
1791
1792         if (WARN_ON_ONCE(!chanctx_conf))
1793                 goto unlock;
1794
1795         chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
1796         ieee80211_recalc_smps_chanctx(local, chanctx);
1797  unlock:
1798         mutex_unlock(&local->chanctx_mtx);
1799 }
1800
1801 static bool ieee80211_id_in_list(const u8 *ids, int n_ids, u8 id)
1802 {
1803         int i;
1804
1805         for (i = 0; i < n_ids; i++)
1806                 if (ids[i] == id)
1807                         return true;
1808         return false;
1809 }
1810
1811 /**
1812  * ieee80211_ie_split - split an IE buffer according to ordering
1813  *
1814  * @ies: the IE buffer
1815  * @ielen: the length of the IE buffer
1816  * @ids: an array with element IDs that are allowed before
1817  *      the split
1818  * @n_ids: the size of the element ID array
1819  * @offset: offset where to start splitting in the buffer
1820  *
1821  * This function splits an IE buffer by updating the @offset
1822  * variable to point to the location where the buffer should be
1823  * split.
1824  *
1825  * It assumes that the given IE buffer is well-formed, this
1826  * has to be guaranteed by the caller!
1827  *
1828  * It also assumes that the IEs in the buffer are ordered
1829  * correctly, if not the result of using this function will not
1830  * be ordered correctly either, i.e. it does no reordering.
1831  *
1832  * The function returns the offset where the next part of the
1833  * buffer starts, which may be @ielen if the entire (remainder)
1834  * of the buffer should be used.
1835  */
1836 size_t ieee80211_ie_split(const u8 *ies, size_t ielen,
1837                           const u8 *ids, int n_ids, size_t offset)
1838 {
1839         size_t pos = offset;
1840
1841         while (pos < ielen && ieee80211_id_in_list(ids, n_ids, ies[pos]))
1842                 pos += 2 + ies[pos + 1];
1843
1844         return pos;
1845 }
1846
1847 size_t ieee80211_ie_split_vendor(const u8 *ies, size_t ielen, size_t offset)
1848 {
1849         size_t pos = offset;
1850
1851         while (pos < ielen && ies[pos] != WLAN_EID_VENDOR_SPECIFIC)
1852                 pos += 2 + ies[pos + 1];
1853
1854         return pos;
1855 }
1856
1857 static void _ieee80211_enable_rssi_reports(struct ieee80211_sub_if_data *sdata,
1858                                             int rssi_min_thold,
1859                                             int rssi_max_thold)
1860 {
1861         trace_api_enable_rssi_reports(sdata, rssi_min_thold, rssi_max_thold);
1862
1863         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
1864                 return;
1865
1866         /*
1867          * Scale up threshold values before storing it, as the RSSI averaging
1868          * algorithm uses a scaled up value as well. Change this scaling
1869          * factor if the RSSI averaging algorithm changes.
1870          */
1871         sdata->u.mgd.rssi_min_thold = rssi_min_thold*16;
1872         sdata->u.mgd.rssi_max_thold = rssi_max_thold*16;
1873 }
1874
1875 void ieee80211_enable_rssi_reports(struct ieee80211_vif *vif,
1876                                     int rssi_min_thold,
1877                                     int rssi_max_thold)
1878 {
1879         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1880
1881         WARN_ON(rssi_min_thold == rssi_max_thold ||
1882                 rssi_min_thold > rssi_max_thold);
1883
1884         _ieee80211_enable_rssi_reports(sdata, rssi_min_thold,
1885                                        rssi_max_thold);
1886 }
1887 EXPORT_SYMBOL(ieee80211_enable_rssi_reports);
1888
1889 void ieee80211_disable_rssi_reports(struct ieee80211_vif *vif)
1890 {
1891         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1892
1893         _ieee80211_enable_rssi_reports(sdata, 0, 0);
1894 }
1895 EXPORT_SYMBOL(ieee80211_disable_rssi_reports);
1896
1897 u8 *ieee80211_ie_build_ht_cap(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
1898                               u16 cap)
1899 {
1900         __le16 tmp;
1901
1902         *pos++ = WLAN_EID_HT_CAPABILITY;
1903         *pos++ = sizeof(struct ieee80211_ht_cap);
1904         memset(pos, 0, sizeof(struct ieee80211_ht_cap));
1905
1906         /* capability flags */
1907         tmp = cpu_to_le16(cap);
1908         memcpy(pos, &tmp, sizeof(u16));
1909         pos += sizeof(u16);
1910
1911         /* AMPDU parameters */
1912         *pos++ = ht_cap->ampdu_factor |
1913                  (ht_cap->ampdu_density <<
1914                         IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT);
1915
1916         /* MCS set */
1917         memcpy(pos, &ht_cap->mcs, sizeof(ht_cap->mcs));
1918         pos += sizeof(ht_cap->mcs);
1919
1920         /* extended capabilities */
1921         pos += sizeof(__le16);
1922
1923         /* BF capabilities */
1924         pos += sizeof(__le32);
1925
1926         /* antenna selection */
1927         pos += sizeof(u8);
1928
1929         return pos;
1930 }
1931
1932 u8 *ieee80211_ie_build_vht_cap(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
1933                                u32 cap)
1934 {
1935         __le32 tmp;
1936
1937         *pos++ = WLAN_EID_VHT_CAPABILITY;
1938         *pos++ = sizeof(struct ieee80211_vht_cap);
1939         memset(pos, 0, sizeof(struct ieee80211_vht_cap));
1940
1941         /* capability flags */
1942         tmp = cpu_to_le32(cap);
1943         memcpy(pos, &tmp, sizeof(u32));
1944         pos += sizeof(u32);
1945
1946         /* VHT MCS set */
1947         memcpy(pos, &vht_cap->vht_mcs, sizeof(vht_cap->vht_mcs));
1948         pos += sizeof(vht_cap->vht_mcs);
1949
1950         return pos;
1951 }
1952
1953 u8 *ieee80211_ie_build_ht_oper(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
1954                                const struct cfg80211_chan_def *chandef,
1955                                u16 prot_mode)
1956 {
1957         struct ieee80211_ht_operation *ht_oper;
1958         /* Build HT Information */
1959         *pos++ = WLAN_EID_HT_OPERATION;
1960         *pos++ = sizeof(struct ieee80211_ht_operation);
1961         ht_oper = (struct ieee80211_ht_operation *)pos;
1962         ht_oper->primary_chan = ieee80211_frequency_to_channel(
1963                                         chandef->chan->center_freq);
1964         switch (chandef->width) {
1965         case NL80211_CHAN_WIDTH_160:
1966         case NL80211_CHAN_WIDTH_80P80:
1967         case NL80211_CHAN_WIDTH_80:
1968         case NL80211_CHAN_WIDTH_40:
1969                 if (chandef->center_freq1 > chandef->chan->center_freq)
1970                         ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
1971                 else
1972                         ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
1973                 break;
1974         default:
1975                 ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_NONE;
1976                 break;
1977         }
1978         if (ht_cap->cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 &&
1979             chandef->width != NL80211_CHAN_WIDTH_20_NOHT &&
1980             chandef->width != NL80211_CHAN_WIDTH_20)
1981                 ht_oper->ht_param |= IEEE80211_HT_PARAM_CHAN_WIDTH_ANY;
1982
1983         ht_oper->operation_mode = cpu_to_le16(prot_mode);
1984         ht_oper->stbc_param = 0x0000;
1985
1986         /* It seems that Basic MCS set and Supported MCS set
1987            are identical for the first 10 bytes */
1988         memset(&ht_oper->basic_set, 0, 16);
1989         memcpy(&ht_oper->basic_set, &ht_cap->mcs, 10);
1990
1991         return pos + sizeof(struct ieee80211_ht_operation);
1992 }
1993
1994 void ieee80211_ht_oper_to_chandef(struct ieee80211_channel *control_chan,
1995                                   const struct ieee80211_ht_operation *ht_oper,
1996                                   struct cfg80211_chan_def *chandef)
1997 {
1998         enum nl80211_channel_type channel_type;
1999
2000         if (!ht_oper) {
2001                 cfg80211_chandef_create(chandef, control_chan,
2002                                         NL80211_CHAN_NO_HT);
2003                 return;
2004         }
2005
2006         switch (ht_oper->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
2007         case IEEE80211_HT_PARAM_CHA_SEC_NONE:
2008                 channel_type = NL80211_CHAN_HT20;
2009                 break;
2010         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
2011                 channel_type = NL80211_CHAN_HT40PLUS;
2012                 break;
2013         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
2014                 channel_type = NL80211_CHAN_HT40MINUS;
2015                 break;
2016         default:
2017                 channel_type = NL80211_CHAN_NO_HT;
2018         }
2019
2020         cfg80211_chandef_create(chandef, control_chan, channel_type);
2021 }
2022
2023 int ieee80211_parse_bitrates(struct cfg80211_chan_def *chandef,
2024                              const struct ieee80211_supported_band *sband,
2025                              const u8 *srates, int srates_len, u32 *rates)
2026 {
2027         u32 rate_flags = ieee80211_chandef_rate_flags(chandef);
2028         int shift = ieee80211_chandef_get_shift(chandef);
2029         struct ieee80211_rate *br;
2030         int brate, rate, i, j, count = 0;
2031
2032         *rates = 0;
2033
2034         for (i = 0; i < srates_len; i++) {
2035                 rate = srates[i] & 0x7f;
2036
2037                 for (j = 0; j < sband->n_bitrates; j++) {
2038                         br = &sband->bitrates[j];
2039                         if ((rate_flags & br->flags) != rate_flags)
2040                                 continue;
2041
2042                         brate = DIV_ROUND_UP(br->bitrate, (1 << shift) * 5);
2043                         if (brate == rate) {
2044                                 *rates |= BIT(j);
2045                                 count++;
2046                                 break;
2047                         }
2048                 }
2049         }
2050         return count;
2051 }
2052
2053 int ieee80211_add_srates_ie(struct ieee80211_sub_if_data *sdata,
2054                             struct sk_buff *skb, bool need_basic,
2055                             enum ieee80211_band band)
2056 {
2057         struct ieee80211_local *local = sdata->local;
2058         struct ieee80211_supported_band *sband;
2059         int rate, shift;
2060         u8 i, rates, *pos;
2061         u32 basic_rates = sdata->vif.bss_conf.basic_rates;
2062         u32 rate_flags;
2063
2064         shift = ieee80211_vif_get_shift(&sdata->vif);
2065         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
2066         sband = local->hw.wiphy->bands[band];
2067         rates = 0;
2068         for (i = 0; i < sband->n_bitrates; i++) {
2069                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2070                         continue;
2071                 rates++;
2072         }
2073         if (rates > 8)
2074                 rates = 8;
2075
2076         if (skb_tailroom(skb) < rates + 2)
2077                 return -ENOMEM;
2078
2079         pos = skb_put(skb, rates + 2);
2080         *pos++ = WLAN_EID_SUPP_RATES;
2081         *pos++ = rates;
2082         for (i = 0; i < rates; i++) {
2083                 u8 basic = 0;
2084                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2085                         continue;
2086
2087                 if (need_basic && basic_rates & BIT(i))
2088                         basic = 0x80;
2089                 rate = sband->bitrates[i].bitrate;
2090                 rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
2091                                     5 * (1 << shift));
2092                 *pos++ = basic | (u8) rate;
2093         }
2094
2095         return 0;
2096 }
2097
2098 int ieee80211_add_ext_srates_ie(struct ieee80211_sub_if_data *sdata,
2099                                 struct sk_buff *skb, bool need_basic,
2100                                 enum ieee80211_band band)
2101 {
2102         struct ieee80211_local *local = sdata->local;
2103         struct ieee80211_supported_band *sband;
2104         int rate, shift;
2105         u8 i, exrates, *pos;
2106         u32 basic_rates = sdata->vif.bss_conf.basic_rates;
2107         u32 rate_flags;
2108
2109         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
2110         shift = ieee80211_vif_get_shift(&sdata->vif);
2111
2112         sband = local->hw.wiphy->bands[band];
2113         exrates = 0;
2114         for (i = 0; i < sband->n_bitrates; i++) {
2115                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2116                         continue;
2117                 exrates++;
2118         }
2119
2120         if (exrates > 8)
2121                 exrates -= 8;
2122         else
2123                 exrates = 0;
2124
2125         if (skb_tailroom(skb) < exrates + 2)
2126                 return -ENOMEM;
2127
2128         if (exrates) {
2129                 pos = skb_put(skb, exrates + 2);
2130                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
2131                 *pos++ = exrates;
2132                 for (i = 8; i < sband->n_bitrates; i++) {
2133                         u8 basic = 0;
2134                         if ((rate_flags & sband->bitrates[i].flags)
2135                             != rate_flags)
2136                                 continue;
2137                         if (need_basic && basic_rates & BIT(i))
2138                                 basic = 0x80;
2139                         rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
2140                                             5 * (1 << shift));
2141                         *pos++ = basic | (u8) rate;
2142                 }
2143         }
2144         return 0;
2145 }
2146
2147 int ieee80211_ave_rssi(struct ieee80211_vif *vif)
2148 {
2149         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2150         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2151
2152         if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_STATION)) {
2153                 /* non-managed type inferfaces */
2154                 return 0;
2155         }
2156         return ifmgd->ave_beacon_signal / 16;
2157 }
2158 EXPORT_SYMBOL_GPL(ieee80211_ave_rssi);
2159
2160 u8 ieee80211_mcs_to_chains(const struct ieee80211_mcs_info *mcs)
2161 {
2162         if (!mcs)
2163                 return 1;
2164
2165         /* TODO: consider rx_highest */
2166
2167         if (mcs->rx_mask[3])
2168                 return 4;
2169         if (mcs->rx_mask[2])
2170                 return 3;
2171         if (mcs->rx_mask[1])
2172                 return 2;
2173         return 1;
2174 }
2175
2176 /**
2177  * ieee80211_calculate_rx_timestamp - calculate timestamp in frame
2178  * @local: mac80211 hw info struct
2179  * @status: RX status
2180  * @mpdu_len: total MPDU length (including FCS)
2181  * @mpdu_offset: offset into MPDU to calculate timestamp at
2182  *
2183  * This function calculates the RX timestamp at the given MPDU offset, taking
2184  * into account what the RX timestamp was. An offset of 0 will just normalize
2185  * the timestamp to TSF at beginning of MPDU reception.
2186  */
2187 u64 ieee80211_calculate_rx_timestamp(struct ieee80211_local *local,
2188                                      struct ieee80211_rx_status *status,
2189                                      unsigned int mpdu_len,
2190                                      unsigned int mpdu_offset)
2191 {
2192         u64 ts = status->mactime;
2193         struct rate_info ri;
2194         u16 rate;
2195
2196         if (WARN_ON(!ieee80211_have_rx_timestamp(status)))
2197                 return 0;
2198
2199         memset(&ri, 0, sizeof(ri));
2200
2201         /* Fill cfg80211 rate info */
2202         if (status->flag & RX_FLAG_HT) {
2203                 ri.mcs = status->rate_idx;
2204                 ri.flags |= RATE_INFO_FLAGS_MCS;
2205                 if (status->flag & RX_FLAG_40MHZ)
2206                         ri.flags |= RATE_INFO_FLAGS_40_MHZ_WIDTH;
2207                 if (status->flag & RX_FLAG_SHORT_GI)
2208                         ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
2209         } else if (status->flag & RX_FLAG_VHT) {
2210                 ri.flags |= RATE_INFO_FLAGS_VHT_MCS;
2211                 ri.mcs = status->rate_idx;
2212                 ri.nss = status->vht_nss;
2213                 if (status->flag & RX_FLAG_40MHZ)
2214                         ri.flags |= RATE_INFO_FLAGS_40_MHZ_WIDTH;
2215                 if (status->flag & RX_FLAG_80MHZ)
2216                         ri.flags |= RATE_INFO_FLAGS_80_MHZ_WIDTH;
2217                 if (status->flag & RX_FLAG_80P80MHZ)
2218                         ri.flags |= RATE_INFO_FLAGS_80P80_MHZ_WIDTH;
2219                 if (status->flag & RX_FLAG_160MHZ)
2220                         ri.flags |= RATE_INFO_FLAGS_160_MHZ_WIDTH;
2221                 if (status->flag & RX_FLAG_SHORT_GI)
2222                         ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
2223         } else {
2224                 struct ieee80211_supported_band *sband;
2225                 int shift = 0;
2226                 int bitrate;
2227
2228                 if (status->flag & RX_FLAG_10MHZ)
2229                         shift = 1;
2230                 if (status->flag & RX_FLAG_5MHZ)
2231                         shift = 2;
2232
2233                 sband = local->hw.wiphy->bands[status->band];
2234                 bitrate = sband->bitrates[status->rate_idx].bitrate;
2235                 ri.legacy = DIV_ROUND_UP(bitrate, (1 << shift));
2236         }
2237
2238         rate = cfg80211_calculate_bitrate(&ri);
2239         if (WARN_ONCE(!rate,
2240                       "Invalid bitrate: flags=0x%x, idx=%d, vht_nss=%d\n",
2241                       status->flag, status->rate_idx, status->vht_nss))
2242                 return 0;
2243
2244         /* rewind from end of MPDU */
2245         if (status->flag & RX_FLAG_MACTIME_END)
2246                 ts -= mpdu_len * 8 * 10 / rate;
2247
2248         ts += mpdu_offset * 8 * 10 / rate;
2249
2250         return ts;
2251 }
2252
2253 void ieee80211_dfs_cac_cancel(struct ieee80211_local *local)
2254 {
2255         struct ieee80211_sub_if_data *sdata;
2256
2257         mutex_lock(&local->iflist_mtx);
2258         list_for_each_entry(sdata, &local->interfaces, list) {
2259                 cancel_delayed_work_sync(&sdata->dfs_cac_timer_work);
2260
2261                 if (sdata->wdev.cac_started) {
2262                         ieee80211_vif_release_channel(sdata);
2263                         cfg80211_cac_event(sdata->dev,
2264                                            NL80211_RADAR_CAC_ABORTED,
2265                                            GFP_KERNEL);
2266                 }
2267         }
2268         mutex_unlock(&local->iflist_mtx);
2269 }
2270
2271 void ieee80211_dfs_radar_detected_work(struct work_struct *work)
2272 {
2273         struct ieee80211_local *local =
2274                 container_of(work, struct ieee80211_local, radar_detected_work);
2275         struct cfg80211_chan_def chandef;
2276
2277         ieee80211_dfs_cac_cancel(local);
2278
2279         if (local->use_chanctx)
2280                 /* currently not handled */
2281                 WARN_ON(1);
2282         else {
2283                 chandef = local->hw.conf.chandef;
2284                 cfg80211_radar_event(local->hw.wiphy, &chandef, GFP_KERNEL);
2285         }
2286 }
2287
2288 void ieee80211_radar_detected(struct ieee80211_hw *hw)
2289 {
2290         struct ieee80211_local *local = hw_to_local(hw);
2291
2292         trace_api_radar_detected(local);
2293
2294         ieee80211_queue_work(hw, &local->radar_detected_work);
2295 }
2296 EXPORT_SYMBOL(ieee80211_radar_detected);
2297
2298 u32 ieee80211_chandef_downgrade(struct cfg80211_chan_def *c)
2299 {
2300         u32 ret;
2301         int tmp;
2302
2303         switch (c->width) {
2304         case NL80211_CHAN_WIDTH_20:
2305                 c->width = NL80211_CHAN_WIDTH_20_NOHT;
2306                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
2307                 break;
2308         case NL80211_CHAN_WIDTH_40:
2309                 c->width = NL80211_CHAN_WIDTH_20;
2310                 c->center_freq1 = c->chan->center_freq;
2311                 ret = IEEE80211_STA_DISABLE_40MHZ |
2312                       IEEE80211_STA_DISABLE_VHT;
2313                 break;
2314         case NL80211_CHAN_WIDTH_80:
2315                 tmp = (30 + c->chan->center_freq - c->center_freq1)/20;
2316                 /* n_P40 */
2317                 tmp /= 2;
2318                 /* freq_P40 */
2319                 c->center_freq1 = c->center_freq1 - 20 + 40 * tmp;
2320                 c->width = NL80211_CHAN_WIDTH_40;
2321                 ret = IEEE80211_STA_DISABLE_VHT;
2322                 break;
2323         case NL80211_CHAN_WIDTH_80P80:
2324                 c->center_freq2 = 0;
2325                 c->width = NL80211_CHAN_WIDTH_80;
2326                 ret = IEEE80211_STA_DISABLE_80P80MHZ |
2327                       IEEE80211_STA_DISABLE_160MHZ;
2328                 break;
2329         case NL80211_CHAN_WIDTH_160:
2330                 /* n_P20 */
2331                 tmp = (70 + c->chan->center_freq - c->center_freq1)/20;
2332                 /* n_P80 */
2333                 tmp /= 4;
2334                 c->center_freq1 = c->center_freq1 - 40 + 80 * tmp;
2335                 c->width = NL80211_CHAN_WIDTH_80;
2336                 ret = IEEE80211_STA_DISABLE_80P80MHZ |
2337                       IEEE80211_STA_DISABLE_160MHZ;
2338                 break;
2339         default:
2340         case NL80211_CHAN_WIDTH_20_NOHT:
2341                 WARN_ON_ONCE(1);
2342                 c->width = NL80211_CHAN_WIDTH_20_NOHT;
2343                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
2344                 break;
2345         case NL80211_CHAN_WIDTH_5:
2346         case NL80211_CHAN_WIDTH_10:
2347                 WARN_ON_ONCE(1);
2348                 /* keep c->width */
2349                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
2350                 break;
2351         }
2352
2353         WARN_ON_ONCE(!cfg80211_chandef_valid(c));
2354
2355         return ret;
2356 }