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  * Copyright 2013-2014  Intel Mobile Communications GmbH
   7  * Copyright (C) 2015-2017      Intel Deutschland GmbH
   8  * Copyright (C) 2018 Intel Corporation
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License version 2 as
  12  * published by the Free Software Foundation.
  13  *
  14  * utilities for mac80211
  15  */
  16
  17 #include <net/mac80211.h>
  18 #include <linux/netdevice.h>
  19 #include <linux/export.h>
  20 #include <linux/types.h>
  21 #include <linux/slab.h>
  22 #include <linux/skbuff.h>
  23 #include <linux/etherdevice.h>
  24 #include <linux/if_arp.h>
  25 #include <linux/bitmap.h>
  26 #include <linux/crc32.h>
  27 #include <net/net_namespace.h>
  28 #include <net/cfg80211.h>
  29 #include <net/rtnetlink.h>
  30
  31 #include "ieee80211_i.h"
  32 #include "driver-ops.h"
  33 #include "rate.h"
  34 #include "mesh.h"
  35 #include "wme.h"
  36 #include "led.h"
  37 #include "wep.h"
  38
  39 /* privid for wiphys to determine whether they belong to us or not */
  40 const void *const mac80211_wiphy_privid = &mac80211_wiphy_privid;
  41
  42 struct ieee80211_hw *wiphy_to_ieee80211_hw(struct wiphy *wiphy)
  43 {
  44         struct ieee80211_local *local;
  45         BUG_ON(!wiphy);
  46
  47         local = wiphy_priv(wiphy);
  48         return &local->hw;
  49 }
  50 EXPORT_SYMBOL(wiphy_to_ieee80211_hw);
  51
  52 void ieee80211_tx_set_protected(struct ieee80211_tx_data *tx)
  53 {
  54         struct sk_buff *skb;
  55         struct ieee80211_hdr *hdr;
  56
  57         skb_queue_walk(&tx->skbs, skb) {
  58                 hdr = (struct ieee80211_hdr *) skb->data;
  59                 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
  60         }
  61 }
  62
  63 int ieee80211_frame_duration(enum nl80211_band band, size_t len,
  64                              int rate, int erp, int short_preamble,
  65                              int shift)
  66 {
  67         int dur;
  68
  69         /* calculate duration (in microseconds, rounded up to next higher
  70          * integer if it includes a fractional microsecond) to send frame of
  71          * len bytes (does not include FCS) at the given rate. Duration will
  72          * also include SIFS.
  73          *
  74          * rate is in 100 kbps, so divident is multiplied by 10 in the
  75          * DIV_ROUND_UP() operations.
  76          *
  77          * shift may be 2 for 5 MHz channels or 1 for 10 MHz channels, and
  78          * is assumed to be 0 otherwise.
  79          */
  80
  81         if (band == NL80211_BAND_5GHZ || erp) {
  82                 /*
  83                  * OFDM:
  84                  *
  85                  * N_DBPS = DATARATE x 4
  86                  * N_SYM = Ceiling((16+8xLENGTH+6) / N_DBPS)
  87                  *      (16 = SIGNAL time, 6 = tail bits)
  88                  * TXTIME = T_PREAMBLE + T_SIGNAL + T_SYM x N_SYM + Signal Ext
  89                  *
  90                  * T_SYM = 4 usec
  91                  * 802.11a - 18.5.2: aSIFSTime = 16 usec
  92                  * 802.11g - 19.8.4: aSIFSTime = 10 usec +
  93                  *      signal ext = 6 usec
  94                  */
  95                 dur = 16; /* SIFS + signal ext */
  96                 dur += 16; /* IEEE 802.11-2012 18.3.2.4: T_PREAMBLE = 16 usec */
  97                 dur += 4; /* IEEE 802.11-2012 18.3.2.4: T_SIGNAL = 4 usec */
  98
  99                 /* IEEE 802.11-2012 18.3.2.4: all values above are:
 100                  *  * times 4 for 5 MHz
 101                  *  * times 2 for 10 MHz
 102                  */
 103                 dur *= 1 << shift;
 104
 105                 /* rates should already consider the channel bandwidth,
 106                  * don't apply divisor again.
 107                  */
 108                 dur += 4 * DIV_ROUND_UP((16 + 8 * (len + 4) + 6) * 10,
 109                                         4 * rate); /* T_SYM x N_SYM */
 110         } else {
 111                 /*
 112                  * 802.11b or 802.11g with 802.11b compatibility:
 113                  * 18.3.4: TXTIME = PreambleLength + PLCPHeaderTime +
 114                  * Ceiling(((LENGTH+PBCC)x8)/DATARATE). PBCC=0.
 115                  *
 116                  * 802.11 (DS): 15.3.3, 802.11b: 18.3.4
 117                  * aSIFSTime = 10 usec
 118                  * aPreambleLength = 144 usec or 72 usec with short preamble
 119                  * aPLCPHeaderLength = 48 usec or 24 usec with short preamble
 120                  */
 121                 dur = 10; /* aSIFSTime = 10 usec */
 122                 dur += short_preamble ? (72 + 24) : (144 + 48);
 123
 124                 dur += DIV_ROUND_UP(8 * (len + 4) * 10, rate);
 125         }
 126
 127         return dur;
 128 }
 129
 130 /* Exported duration function for driver use */
 131 __le16 ieee80211_generic_frame_duration(struct ieee80211_hw *hw,
 132                                         struct ieee80211_vif *vif,
 133                                         enum nl80211_band band,
 134                                         size_t frame_len,
 135                                         struct ieee80211_rate *rate)
 136 {
 137         struct ieee80211_sub_if_data *sdata;
 138         u16 dur;
 139         int erp, shift = 0;
 140         bool short_preamble = false;
 141
 142         erp = 0;
 143         if (vif) {
 144                 sdata = vif_to_sdata(vif);
 145                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 146                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 147                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 148                 shift = ieee80211_vif_get_shift(vif);
 149         }
 150
 151         dur = ieee80211_frame_duration(band, frame_len, rate->bitrate, erp,
 152                                        short_preamble, shift);
 153
 154         return cpu_to_le16(dur);
 155 }
 156 EXPORT_SYMBOL(ieee80211_generic_frame_duration);
 157
 158 __le16 ieee80211_rts_duration(struct ieee80211_hw *hw,
 159                               struct ieee80211_vif *vif, size_t frame_len,
 160                               const struct ieee80211_tx_info *frame_txctl)
 161 {
 162         struct ieee80211_local *local = hw_to_local(hw);
 163         struct ieee80211_rate *rate;
 164         struct ieee80211_sub_if_data *sdata;
 165         bool short_preamble;
 166         int erp, shift = 0, bitrate;
 167         u16 dur;
 168         struct ieee80211_supported_band *sband;
 169
 170         sband = local->hw.wiphy->bands[frame_txctl->band];
 171
 172         short_preamble = false;
 173
 174         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
 175
 176         erp = 0;
 177         if (vif) {
 178                 sdata = vif_to_sdata(vif);
 179                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 180                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 181                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 182                 shift = ieee80211_vif_get_shift(vif);
 183         }
 184
 185         bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
 186
 187         /* CTS duration */
 188         dur = ieee80211_frame_duration(sband->band, 10, bitrate,
 189                                        erp, short_preamble, shift);
 190         /* Data frame duration */
 191         dur += ieee80211_frame_duration(sband->band, frame_len, bitrate,
 192                                         erp, short_preamble, shift);
 193         /* ACK duration */
 194         dur += ieee80211_frame_duration(sband->band, 10, bitrate,
 195                                         erp, short_preamble, shift);
 196
 197         return cpu_to_le16(dur);
 198 }
 199 EXPORT_SYMBOL(ieee80211_rts_duration);
 200
 201 __le16 ieee80211_ctstoself_duration(struct ieee80211_hw *hw,
 202                                     struct ieee80211_vif *vif,
 203                                     size_t frame_len,
 204                                     const struct ieee80211_tx_info *frame_txctl)
 205 {
 206         struct ieee80211_local *local = hw_to_local(hw);
 207         struct ieee80211_rate *rate;
 208         struct ieee80211_sub_if_data *sdata;
 209         bool short_preamble;
 210         int erp, shift = 0, bitrate;
 211         u16 dur;
 212         struct ieee80211_supported_band *sband;
 213
 214         sband = local->hw.wiphy->bands[frame_txctl->band];
 215
 216         short_preamble = false;
 217
 218         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
 219         erp = 0;
 220         if (vif) {
 221                 sdata = vif_to_sdata(vif);
 222                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
 223                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
 224                         erp = rate->flags & IEEE80211_RATE_ERP_G;
 225                 shift = ieee80211_vif_get_shift(vif);
 226         }
 227
 228         bitrate = DIV_ROUND_UP(rate->bitrate, 1 << shift);
 229
 230         /* Data frame duration */
 231         dur = ieee80211_frame_duration(sband->band, frame_len, bitrate,
 232                                        erp, short_preamble, shift);
 233         if (!(frame_txctl->flags & IEEE80211_TX_CTL_NO_ACK)) {
 234                 /* ACK duration */
 235                 dur += ieee80211_frame_duration(sband->band, 10, bitrate,
 236                                                 erp, short_preamble, shift);
 237         }
 238
 239         return cpu_to_le16(dur);
 240 }
 241 EXPORT_SYMBOL(ieee80211_ctstoself_duration);
 242
 243 static void __ieee80211_wake_txqs(struct ieee80211_sub_if_data *sdata, int ac)
 244 {
 245         struct ieee80211_local *local = sdata->local;
 246         struct ieee80211_vif *vif = &sdata->vif;
 247         struct fq *fq = &local->fq;
 248         struct ps_data *ps = NULL;
 249         struct txq_info *txqi;
 250         struct sta_info *sta;
 251         int i;
 252
 253         spin_lock_bh(&fq->lock);
 254
 255         if (sdata->vif.type == NL80211_IFTYPE_AP)
 256                 ps = &sdata->bss->ps;
 257
 258         sdata->vif.txqs_stopped[ac] = false;
 259
 260         list_for_each_entry_rcu(sta, &local->sta_list, list) {
 261                 if (sdata != sta->sdata)
 262                         continue;
 263
 264                 for (i = 0; i < ARRAY_SIZE(sta->sta.txq); i++) {
 265                         struct ieee80211_txq *txq = sta->sta.txq[i];
 266
 267                         if (!txq)
 268                                 continue;
 269
 270                         txqi = to_txq_info(txq);
 271
 272                         if (ac != txq->ac)
 273                                 continue;
 274
 275                         if (!test_and_clear_bit(IEEE80211_TXQ_STOP_NETIF_TX,
 276                                                 &txqi->flags))
 277                                 continue;
 278
 279                         spin_unlock_bh(&fq->lock);
 280                         drv_wake_tx_queue(local, txqi);
 281                         spin_lock_bh(&fq->lock);
 282                 }
 283         }
 284
 285         if (!vif->txq)
 286                 goto out;
 287
 288         txqi = to_txq_info(vif->txq);
 289
 290         if (!test_and_clear_bit(IEEE80211_TXQ_STOP_NETIF_TX, &txqi->flags) ||
 291             (ps && atomic_read(&ps->num_sta_ps)) || ac != vif->txq->ac)
 292                 goto out;
 293
 294         spin_unlock_bh(&fq->lock);
 295
 296         drv_wake_tx_queue(local, txqi);
 297         return;
 298 out:
 299         spin_unlock_bh(&fq->lock);
 300 }
 301
 302 static void
 303 __releases(&local->queue_stop_reason_lock)
 304 __acquires(&local->queue_stop_reason_lock)
 305 _ieee80211_wake_txqs(struct ieee80211_local *local, unsigned long *flags)
 306 {
 307         struct ieee80211_sub_if_data *sdata;
 308         int n_acs = IEEE80211_NUM_ACS;
 309         int i;
 310
 311         rcu_read_lock();
 312
 313         if (local->hw.queues < IEEE80211_NUM_ACS)
 314                 n_acs = 1;
 315
 316         for (i = 0; i < local->hw.queues; i++) {
 317                 if (local->queue_stop_reasons[i])
 318                         continue;
 319
 320                 spin_unlock_irqrestore(&local->queue_stop_reason_lock, *flags);
 321                 list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 322                         int ac;
 323
 324                         for (ac = 0; ac < n_acs; ac++) {
 325                                 int ac_queue = sdata->vif.hw_queue[ac];
 326
 327                                 if (ac_queue == i ||
 328                                     sdata->vif.cab_queue == i)
 329                                         __ieee80211_wake_txqs(sdata, ac);
 330                         }
 331                 }
 332                 spin_lock_irqsave(&local->queue_stop_reason_lock, *flags);
 333         }
 334
 335         rcu_read_unlock();
 336 }
 337
 338 void ieee80211_wake_txqs(unsigned long data)
 339 {
 340         struct ieee80211_local *local = (struct ieee80211_local *)data;
 341         unsigned long flags;
 342
 343         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 344         _ieee80211_wake_txqs(local, &flags);
 345         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 346 }
 347
 348 void ieee80211_propagate_queue_wake(struct ieee80211_local *local, int queue)
 349 {
 350         struct ieee80211_sub_if_data *sdata;
 351         int n_acs = IEEE80211_NUM_ACS;
 352
 353         if (local->ops->wake_tx_queue)
 354                 return;
 355
 356         if (local->hw.queues < IEEE80211_NUM_ACS)
 357                 n_acs = 1;
 358
 359         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 360                 int ac;
 361
 362                 if (!sdata->dev)
 363                         continue;
 364
 365                 if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE &&
 366                     local->queue_stop_reasons[sdata->vif.cab_queue] != 0)
 367                         continue;
 368
 369                 for (ac = 0; ac < n_acs; ac++) {
 370                         int ac_queue = sdata->vif.hw_queue[ac];
 371
 372                         if (ac_queue == queue ||
 373                             (sdata->vif.cab_queue == queue &&
 374                              local->queue_stop_reasons[ac_queue] == 0 &&
 375                              skb_queue_empty(&local->pending[ac_queue])))
 376                                 netif_wake_subqueue(sdata->dev, ac);
 377                 }
 378         }
 379 }
 380
 381 static void __ieee80211_wake_queue(struct ieee80211_hw *hw, int queue,
 382                                    enum queue_stop_reason reason,
 383                                    bool refcounted,
 384                                    unsigned long *flags)
 385 {
 386         struct ieee80211_local *local = hw_to_local(hw);
 387
 388         trace_wake_queue(local, queue, reason);
 389
 390         if (WARN_ON(queue >= hw->queues))
 391                 return;
 392
 393         if (!test_bit(reason, &local->queue_stop_reasons[queue]))
 394                 return;
 395
 396         if (!refcounted) {
 397                 local->q_stop_reasons[queue][reason] = 0;
 398         } else {
 399                 local->q_stop_reasons[queue][reason]--;
 400                 if (WARN_ON(local->q_stop_reasons[queue][reason] < 0))
 401                         local->q_stop_reasons[queue][reason] = 0;
 402         }
 403
 404         if (local->q_stop_reasons[queue][reason] == 0)
 405                 __clear_bit(reason, &local->queue_stop_reasons[queue]);
 406
 407         if (local->queue_stop_reasons[queue] != 0)
 408                 /* someone still has this queue stopped */
 409                 return;
 410
 411         if (skb_queue_empty(&local->pending[queue])) {
 412                 rcu_read_lock();
 413                 ieee80211_propagate_queue_wake(local, queue);
 414                 rcu_read_unlock();
 415         } else
 416                 tasklet_schedule(&local->tx_pending_tasklet);
 417
 418         /*
 419          * Calling _ieee80211_wake_txqs here can be a problem because it may
 420          * release queue_stop_reason_lock which has been taken by
 421          * __ieee80211_wake_queue's caller. It is certainly not very nice to
 422          * release someone's lock, but it is fine because all the callers of
 423          * __ieee80211_wake_queue call it right before releasing the lock.
 424          */
 425         if (local->ops->wake_tx_queue) {
 426                 if (reason == IEEE80211_QUEUE_STOP_REASON_DRIVER)
 427                         tasklet_schedule(&local->wake_txqs_tasklet);
 428                 else
 429                         _ieee80211_wake_txqs(local, flags);
 430         }
 431 }
 432
 433 void ieee80211_wake_queue_by_reason(struct ieee80211_hw *hw, int queue,
 434                                     enum queue_stop_reason reason,
 435                                     bool refcounted)
 436 {
 437         struct ieee80211_local *local = hw_to_local(hw);
 438         unsigned long flags;
 439
 440         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 441         __ieee80211_wake_queue(hw, queue, reason, refcounted, &flags);
 442         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 443 }
 444
 445 void ieee80211_wake_queue(struct ieee80211_hw *hw, int queue)
 446 {
 447         ieee80211_wake_queue_by_reason(hw, queue,
 448                                        IEEE80211_QUEUE_STOP_REASON_DRIVER,
 449                                        false);
 450 }
 451 EXPORT_SYMBOL(ieee80211_wake_queue);
 452
 453 static void __ieee80211_stop_queue(struct ieee80211_hw *hw, int queue,
 454                                    enum queue_stop_reason reason,
 455                                    bool refcounted)
 456 {
 457         struct ieee80211_local *local = hw_to_local(hw);
 458         struct ieee80211_sub_if_data *sdata;
 459         int n_acs = IEEE80211_NUM_ACS;
 460
 461         trace_stop_queue(local, queue, reason);
 462
 463         if (WARN_ON(queue >= hw->queues))
 464                 return;
 465
 466         if (!refcounted)
 467                 local->q_stop_reasons[queue][reason] = 1;
 468         else
 469                 local->q_stop_reasons[queue][reason]++;
 470
 471         if (__test_and_set_bit(reason, &local->queue_stop_reasons[queue]))
 472                 return;
 473
 474         if (local->hw.queues < IEEE80211_NUM_ACS)
 475                 n_acs = 1;
 476
 477         rcu_read_lock();
 478         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 479                 int ac;
 480
 481                 if (!sdata->dev)
 482                         continue;
 483
 484                 for (ac = 0; ac < n_acs; ac++) {
 485                         if (sdata->vif.hw_queue[ac] == queue ||
 486                             sdata->vif.cab_queue == queue) {
 487                                 if (!local->ops->wake_tx_queue) {
 488                                         netif_stop_subqueue(sdata->dev, ac);
 489                                         continue;
 490                                 }
 491                                 spin_lock(&local->fq.lock);
 492                                 sdata->vif.txqs_stopped[ac] = true;
 493                                 spin_unlock(&local->fq.lock);
 494                         }
 495                 }
 496         }
 497         rcu_read_unlock();
 498 }
 499
 500 void ieee80211_stop_queue_by_reason(struct ieee80211_hw *hw, int queue,
 501                                     enum queue_stop_reason reason,
 502                                     bool refcounted)
 503 {
 504         struct ieee80211_local *local = hw_to_local(hw);
 505         unsigned long flags;
 506
 507         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 508         __ieee80211_stop_queue(hw, queue, reason, refcounted);
 509         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 510 }
 511
 512 void ieee80211_stop_queue(struct ieee80211_hw *hw, int queue)
 513 {
 514         ieee80211_stop_queue_by_reason(hw, queue,
 515                                        IEEE80211_QUEUE_STOP_REASON_DRIVER,
 516                                        false);
 517 }
 518 EXPORT_SYMBOL(ieee80211_stop_queue);
 519
 520 void ieee80211_add_pending_skb(struct ieee80211_local *local,
 521                                struct sk_buff *skb)
 522 {
 523         struct ieee80211_hw *hw = &local->hw;
 524         unsigned long flags;
 525         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 526         int queue = info->hw_queue;
 527
 528         if (WARN_ON(!info->control.vif)) {
 529                 ieee80211_free_txskb(&local->hw, skb);
 530                 return;
 531         }
 532
 533         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 534         __ieee80211_stop_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
 535                                false);
 536         __skb_queue_tail(&local->pending[queue], skb);
 537         __ieee80211_wake_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
 538                                false, &flags);
 539         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 540 }
 541
 542 void ieee80211_add_pending_skbs(struct ieee80211_local *local,
 543                                 struct sk_buff_head *skbs)
 544 {
 545         struct ieee80211_hw *hw = &local->hw;
 546         struct sk_buff *skb;
 547         unsigned long flags;
 548         int queue, i;
 549
 550         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 551         while ((skb = skb_dequeue(skbs))) {
 552                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 553
 554                 if (WARN_ON(!info->control.vif)) {
 555                         ieee80211_free_txskb(&local->hw, skb);
 556                         continue;
 557                 }
 558
 559                 queue = info->hw_queue;
 560
 561                 __ieee80211_stop_queue(hw, queue,
 562                                 IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
 563                                 false);
 564
 565                 __skb_queue_tail(&local->pending[queue], skb);
 566         }
 567
 568         for (i = 0; i < hw->queues; i++)
 569                 __ieee80211_wake_queue(hw, i,
 570                         IEEE80211_QUEUE_STOP_REASON_SKB_ADD,
 571                         false, &flags);
 572         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 573 }
 574
 575 void ieee80211_stop_queues_by_reason(struct ieee80211_hw *hw,
 576                                      unsigned long queues,
 577                                      enum queue_stop_reason reason,
 578                                      bool refcounted)
 579 {
 580         struct ieee80211_local *local = hw_to_local(hw);
 581         unsigned long flags;
 582         int i;
 583
 584         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 585
 586         for_each_set_bit(i, &queues, hw->queues)
 587                 __ieee80211_stop_queue(hw, i, reason, refcounted);
 588
 589         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 590 }
 591
 592 void ieee80211_stop_queues(struct ieee80211_hw *hw)
 593 {
 594         ieee80211_stop_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
 595                                         IEEE80211_QUEUE_STOP_REASON_DRIVER,
 596                                         false);
 597 }
 598 EXPORT_SYMBOL(ieee80211_stop_queues);
 599
 600 int ieee80211_queue_stopped(struct ieee80211_hw *hw, int queue)
 601 {
 602         struct ieee80211_local *local = hw_to_local(hw);
 603         unsigned long flags;
 604         int ret;
 605
 606         if (WARN_ON(queue >= hw->queues))
 607                 return true;
 608
 609         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 610         ret = test_bit(IEEE80211_QUEUE_STOP_REASON_DRIVER,
 611                        &local->queue_stop_reasons[queue]);
 612         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 613         return ret;
 614 }
 615 EXPORT_SYMBOL(ieee80211_queue_stopped);
 616
 617 void ieee80211_wake_queues_by_reason(struct ieee80211_hw *hw,
 618                                      unsigned long queues,
 619                                      enum queue_stop_reason reason,
 620                                      bool refcounted)
 621 {
 622         struct ieee80211_local *local = hw_to_local(hw);
 623         unsigned long flags;
 624         int i;
 625
 626         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
 627
 628         for_each_set_bit(i, &queues, hw->queues)
 629                 __ieee80211_wake_queue(hw, i, reason, refcounted, &flags);
 630
 631         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
 632 }
 633
 634 void ieee80211_wake_queues(struct ieee80211_hw *hw)
 635 {
 636         ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
 637                                         IEEE80211_QUEUE_STOP_REASON_DRIVER,
 638                                         false);
 639 }
 640 EXPORT_SYMBOL(ieee80211_wake_queues);
 641
 642 static unsigned int
 643 ieee80211_get_vif_queues(struct ieee80211_local *local,
 644                          struct ieee80211_sub_if_data *sdata)
 645 {
 646         unsigned int queues;
 647
 648         if (sdata && ieee80211_hw_check(&local->hw, QUEUE_CONTROL)) {
 649                 int ac;
 650
 651                 queues = 0;
 652
 653                 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
 654                         queues |= BIT(sdata->vif.hw_queue[ac]);
 655                 if (sdata->vif.cab_queue != IEEE80211_INVAL_HW_QUEUE)
 656                         queues |= BIT(sdata->vif.cab_queue);
 657         } else {
 658                 /* all queues */
 659                 queues = BIT(local->hw.queues) - 1;
 660         }
 661
 662         return queues;
 663 }
 664
 665 void __ieee80211_flush_queues(struct ieee80211_local *local,
 666                               struct ieee80211_sub_if_data *sdata,
 667                               unsigned int queues, bool drop)
 668 {
 669         if (!local->ops->flush)
 670                 return;
 671
 672         /*
 673          * If no queue was set, or if the HW doesn't support
 674          * IEEE80211_HW_QUEUE_CONTROL - flush all queues
 675          */
 676         if (!queues || !ieee80211_hw_check(&local->hw, QUEUE_CONTROL))
 677                 queues = ieee80211_get_vif_queues(local, sdata);
 678
 679         ieee80211_stop_queues_by_reason(&local->hw, queues,
 680                                         IEEE80211_QUEUE_STOP_REASON_FLUSH,
 681                                         false);
 682
 683         drv_flush(local, sdata, queues, drop);
 684
 685         ieee80211_wake_queues_by_reason(&local->hw, queues,
 686                                         IEEE80211_QUEUE_STOP_REASON_FLUSH,
 687                                         false);
 688 }
 689
 690 void ieee80211_flush_queues(struct ieee80211_local *local,
 691                             struct ieee80211_sub_if_data *sdata, bool drop)
 692 {
 693         __ieee80211_flush_queues(local, sdata, 0, drop);
 694 }
 695
 696 void ieee80211_stop_vif_queues(struct ieee80211_local *local,
 697                                struct ieee80211_sub_if_data *sdata,
 698                                enum queue_stop_reason reason)
 699 {
 700         ieee80211_stop_queues_by_reason(&local->hw,
 701                                         ieee80211_get_vif_queues(local, sdata),
 702                                         reason, true);
 703 }
 704
 705 void ieee80211_wake_vif_queues(struct ieee80211_local *local,
 706                                struct ieee80211_sub_if_data *sdata,
 707                                enum queue_stop_reason reason)
 708 {
 709         ieee80211_wake_queues_by_reason(&local->hw,
 710                                         ieee80211_get_vif_queues(local, sdata),
 711                                         reason, true);
 712 }
 713
 714 static void __iterate_interfaces(struct ieee80211_local *local,
 715                                  u32 iter_flags,
 716                                  void (*iterator)(void *data, u8 *mac,
 717                                                   struct ieee80211_vif *vif),
 718                                  void *data)
 719 {
 720         struct ieee80211_sub_if_data *sdata;
 721         bool active_only = iter_flags & IEEE80211_IFACE_ITER_ACTIVE;
 722
 723         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
 724                 switch (sdata->vif.type) {
 725                 case NL80211_IFTYPE_MONITOR:
 726                         if (!(sdata->u.mntr.flags & MONITOR_FLAG_ACTIVE))
 727                                 continue;
 728                         break;
 729                 case NL80211_IFTYPE_AP_VLAN:
 730                         continue;
 731                 default:
 732                         break;
 733                 }
 734                 if (!(iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL) &&
 735                     active_only && !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 736                         continue;
 737                 if (ieee80211_sdata_running(sdata) || !active_only)
 738                         iterator(data, sdata->vif.addr,
 739                                  &sdata->vif);
 740         }
 741
 742         sdata = rcu_dereference_check(local->monitor_sdata,
 743                                       lockdep_is_held(&local->iflist_mtx) ||
 744                                       lockdep_rtnl_is_held());
 745         if (sdata &&
 746             (iter_flags & IEEE80211_IFACE_ITER_RESUME_ALL || !active_only ||
 747              sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 748                 iterator(data, sdata->vif.addr, &sdata->vif);
 749 }
 750
 751 void ieee80211_iterate_interfaces(
 752         struct ieee80211_hw *hw, u32 iter_flags,
 753         void (*iterator)(void *data, u8 *mac,
 754                          struct ieee80211_vif *vif),
 755         void *data)
 756 {
 757         struct ieee80211_local *local = hw_to_local(hw);
 758
 759         mutex_lock(&local->iflist_mtx);
 760         __iterate_interfaces(local, iter_flags, iterator, data);
 761         mutex_unlock(&local->iflist_mtx);
 762 }
 763 EXPORT_SYMBOL_GPL(ieee80211_iterate_interfaces);
 764
 765 void ieee80211_iterate_active_interfaces_atomic(
 766         struct ieee80211_hw *hw, u32 iter_flags,
 767         void (*iterator)(void *data, u8 *mac,
 768                          struct ieee80211_vif *vif),
 769         void *data)
 770 {
 771         struct ieee80211_local *local = hw_to_local(hw);
 772
 773         rcu_read_lock();
 774         __iterate_interfaces(local, iter_flags | IEEE80211_IFACE_ITER_ACTIVE,
 775                              iterator, data);
 776         rcu_read_unlock();
 777 }
 778 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_atomic);
 779
 780 void ieee80211_iterate_active_interfaces_rtnl(
 781         struct ieee80211_hw *hw, u32 iter_flags,
 782         void (*iterator)(void *data, u8 *mac,
 783                          struct ieee80211_vif *vif),
 784         void *data)
 785 {
 786         struct ieee80211_local *local = hw_to_local(hw);
 787
 788         ASSERT_RTNL();
 789
 790         __iterate_interfaces(local, iter_flags | IEEE80211_IFACE_ITER_ACTIVE,
 791                              iterator, data);
 792 }
 793 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_rtnl);
 794
 795 static void __iterate_stations(struct ieee80211_local *local,
 796                                void (*iterator)(void *data,
 797                                                 struct ieee80211_sta *sta),
 798                                void *data)
 799 {
 800         struct sta_info *sta;
 801
 802         list_for_each_entry_rcu(sta, &local->sta_list, list) {
 803                 if (!sta->uploaded)
 804                         continue;
 805
 806                 iterator(data, &sta->sta);
 807         }
 808 }
 809
 810 void ieee80211_iterate_stations_atomic(struct ieee80211_hw *hw,
 811                         void (*iterator)(void *data,
 812                                          struct ieee80211_sta *sta),
 813                         void *data)
 814 {
 815         struct ieee80211_local *local = hw_to_local(hw);
 816
 817         rcu_read_lock();
 818         __iterate_stations(local, iterator, data);
 819         rcu_read_unlock();
 820 }
 821 EXPORT_SYMBOL_GPL(ieee80211_iterate_stations_atomic);
 822
 823 struct ieee80211_vif *wdev_to_ieee80211_vif(struct wireless_dev *wdev)
 824 {
 825         struct ieee80211_sub_if_data *sdata = IEEE80211_WDEV_TO_SUB_IF(wdev);
 826
 827         if (!ieee80211_sdata_running(sdata) ||
 828             !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 829                 return NULL;
 830         return &sdata->vif;
 831 }
 832 EXPORT_SYMBOL_GPL(wdev_to_ieee80211_vif);
 833
 834 struct wireless_dev *ieee80211_vif_to_wdev(struct ieee80211_vif *vif)
 835 {
 836         struct ieee80211_sub_if_data *sdata;
 837
 838         if (!vif)
 839                 return NULL;
 840
 841         sdata = vif_to_sdata(vif);
 842
 843         if (!ieee80211_sdata_running(sdata) ||
 844             !(sdata->flags & IEEE80211_SDATA_IN_DRIVER))
 845                 return NULL;
 846
 847         return &sdata->wdev;
 848 }
 849 EXPORT_SYMBOL_GPL(ieee80211_vif_to_wdev);
 850
 851 /*
 852  * Nothing should have been stuffed into the workqueue during
 853  * the suspend->resume cycle. Since we can't check each caller
 854  * of this function if we are already quiescing / suspended,
 855  * check here and don't WARN since this can actually happen when
 856  * the rx path (for example) is racing against __ieee80211_suspend
 857  * and suspending / quiescing was set after the rx path checked
 858  * them.
 859  */
 860 static bool ieee80211_can_queue_work(struct ieee80211_local *local)
 861 {
 862         if (local->quiescing || (local->suspended && !local->resuming)) {
 863                 pr_warn("queueing ieee80211 work while going to suspend\n");
 864                 return false;
 865         }
 866
 867         return true;
 868 }
 869
 870 void ieee80211_queue_work(struct ieee80211_hw *hw, struct work_struct *work)
 871 {
 872         struct ieee80211_local *local = hw_to_local(hw);
 873
 874         if (!ieee80211_can_queue_work(local))
 875                 return;
 876
 877         queue_work(local->workqueue, work);
 878 }
 879 EXPORT_SYMBOL(ieee80211_queue_work);
 880
 881 void ieee80211_queue_delayed_work(struct ieee80211_hw *hw,
 882                                   struct delayed_work *dwork,
 883                                   unsigned long delay)
 884 {
 885         struct ieee80211_local *local = hw_to_local(hw);
 886
 887         if (!ieee80211_can_queue_work(local))
 888                 return;
 889
 890         queue_delayed_work(local->workqueue, dwork, delay);
 891 }
 892 EXPORT_SYMBOL(ieee80211_queue_delayed_work);
 893
 894 u32 ieee802_11_parse_elems_crc(const u8 *start, size_t len, bool action,
 895                                struct ieee802_11_elems *elems,
 896                                u64 filter, u32 crc)
 897 {
 898         size_t left = len;
 899         const u8 *pos = start;
 900         bool calc_crc = filter != 0;
 901         DECLARE_BITMAP(seen_elems, 256);
 902         const u8 *ie;
 903
 904         bitmap_zero(seen_elems, 256);
 905         memset(elems, 0, sizeof(*elems));
 906         elems->ie_start = start;
 907         elems->total_len = len;
 908
 909         while (left >= 2) {
 910                 u8 id, elen;
 911                 bool elem_parse_failed;
 912
 913                 id = *pos++;
 914                 elen = *pos++;
 915                 left -= 2;
 916
 917                 if (elen > left) {
 918                         elems->parse_error = true;
 919                         break;
 920                 }
 921
 922                 switch (id) {
 923                 case WLAN_EID_SSID:
 924                 case WLAN_EID_SUPP_RATES:
 925                 case WLAN_EID_FH_PARAMS:
 926                 case WLAN_EID_DS_PARAMS:
 927                 case WLAN_EID_CF_PARAMS:
 928                 case WLAN_EID_TIM:
 929                 case WLAN_EID_IBSS_PARAMS:
 930                 case WLAN_EID_CHALLENGE:
 931                 case WLAN_EID_RSN:
 932                 case WLAN_EID_ERP_INFO:
 933                 case WLAN_EID_EXT_SUPP_RATES:
 934                 case WLAN_EID_HT_CAPABILITY:
 935                 case WLAN_EID_HT_OPERATION:
 936                 case WLAN_EID_VHT_CAPABILITY:
 937                 case WLAN_EID_VHT_OPERATION:
 938                 case WLAN_EID_MESH_ID:
 939                 case WLAN_EID_MESH_CONFIG:
 940                 case WLAN_EID_PEER_MGMT:
 941                 case WLAN_EID_PREQ:
 942                 case WLAN_EID_PREP:
 943                 case WLAN_EID_PERR:
 944                 case WLAN_EID_RANN:
 945                 case WLAN_EID_CHANNEL_SWITCH:
 946                 case WLAN_EID_EXT_CHANSWITCH_ANN:
 947                 case WLAN_EID_COUNTRY:
 948                 case WLAN_EID_PWR_CONSTRAINT:
 949                 case WLAN_EID_TIMEOUT_INTERVAL:
 950                 case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
 951                 case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
 952                 case WLAN_EID_CHAN_SWITCH_PARAM:
 953                 case WLAN_EID_EXT_CAPABILITY:
 954                 case WLAN_EID_CHAN_SWITCH_TIMING:
 955                 case WLAN_EID_LINK_ID:
 956                 case WLAN_EID_BSS_MAX_IDLE_PERIOD:
 957                 /*
 958                  * not listing WLAN_EID_CHANNEL_SWITCH_WRAPPER -- it seems possible
 959                  * that if the content gets bigger it might be needed more than once
 960                  */
 961                         if (test_bit(id, seen_elems)) {
 962                                 elems->parse_error = true;
 963                                 left -= elen;
 964                                 pos += elen;
 965                                 continue;
 966                         }
 967                         break;
 968                 }
 969
 970                 if (calc_crc && id < 64 && (filter & (1ULL << id)))
 971                         crc = crc32_be(crc, pos - 2, elen + 2);
 972
 973                 elem_parse_failed = false;
 974
 975                 switch (id) {
 976                 case WLAN_EID_LINK_ID:
 977                         if (elen + 2 != sizeof(struct ieee80211_tdls_lnkie)) {
 978                                 elem_parse_failed = true;
 979                                 break;
 980                         }
 981                         elems->lnk_id = (void *)(pos - 2);
 982                         break;
 983                 case WLAN_EID_CHAN_SWITCH_TIMING:
 984                         if (elen != sizeof(struct ieee80211_ch_switch_timing)) {
 985                                 elem_parse_failed = true;
 986                                 break;
 987                         }
 988                         elems->ch_sw_timing = (void *)pos;
 989                         break;
 990                 case WLAN_EID_EXT_CAPABILITY:
 991                         elems->ext_capab = pos;
 992                         elems->ext_capab_len = elen;
 993                         break;
 994                 case WLAN_EID_SSID:
 995                         elems->ssid = pos;
 996                         elems->ssid_len = elen;
 997                         break;
 998                 case WLAN_EID_SUPP_RATES:
 999                         elems->supp_rates = pos;
1000                         elems->supp_rates_len = elen;
1001                         break;
1002                 case WLAN_EID_DS_PARAMS:
1003                         if (elen >= 1)
1004                                 elems->ds_params = pos;
1005                         else
1006                                 elem_parse_failed = true;
1007                         break;
1008                 case WLAN_EID_TIM:
1009                         if (elen >= sizeof(struct ieee80211_tim_ie)) {
1010                                 elems->tim = (void *)pos;
1011                                 elems->tim_len = elen;
1012                         } else
1013                                 elem_parse_failed = true;
1014                         break;
1015                 case WLAN_EID_CHALLENGE:
1016                         elems->challenge = pos;
1017                         elems->challenge_len = elen;
1018                         break;
1019                 case WLAN_EID_VENDOR_SPECIFIC:
1020                         if (elen >= 4 && pos[0] == 0x00 && pos[1] == 0x50 &&
1021                             pos[2] == 0xf2) {
1022                                 /* Microsoft OUI (00:50:F2) */
1023
1024                                 if (calc_crc)
1025                                         crc = crc32_be(crc, pos - 2, elen + 2);
1026
1027                                 if (elen >= 5 && pos[3] == 2) {
1028                                         /* OUI Type 2 - WMM IE */
1029                                         if (pos[4] == 0) {
1030                                                 elems->wmm_info = pos;
1031                                                 elems->wmm_info_len = elen;
1032                                         } else if (pos[4] == 1) {
1033                                                 elems->wmm_param = pos;
1034                                                 elems->wmm_param_len = elen;
1035                                         }
1036                                 }
1037                         }
1038                         break;
1039                 case WLAN_EID_RSN:
1040                         elems->rsn = pos;
1041                         elems->rsn_len = elen;
1042                         break;
1043                 case WLAN_EID_ERP_INFO:
1044                         if (elen >= 1)
1045                                 elems->erp_info = pos;
1046                         else
1047                                 elem_parse_failed = true;
1048                         break;
1049                 case WLAN_EID_EXT_SUPP_RATES:
1050                         elems->ext_supp_rates = pos;
1051                         elems->ext_supp_rates_len = elen;
1052                         break;
1053                 case WLAN_EID_HT_CAPABILITY:
1054                         if (elen >= sizeof(struct ieee80211_ht_cap))
1055                                 elems->ht_cap_elem = (void *)pos;
1056                         else
1057                                 elem_parse_failed = true;
1058                         break;
1059                 case WLAN_EID_HT_OPERATION:
1060                         if (elen >= sizeof(struct ieee80211_ht_operation))
1061                                 elems->ht_operation = (void *)pos;
1062                         else
1063                                 elem_parse_failed = true;
1064                         break;
1065                 case WLAN_EID_VHT_CAPABILITY:
1066                         if (elen >= sizeof(struct ieee80211_vht_cap))
1067                                 elems->vht_cap_elem = (void *)pos;
1068                         else
1069                                 elem_parse_failed = true;
1070                         break;
1071                 case WLAN_EID_VHT_OPERATION:
1072                         if (elen >= sizeof(struct ieee80211_vht_operation))
1073                                 elems->vht_operation = (void *)pos;
1074                         else
1075                                 elem_parse_failed = true;
1076                         break;
1077                 case WLAN_EID_OPMODE_NOTIF:
1078                         if (elen > 0)
1079                                 elems->opmode_notif = pos;
1080                         else
1081                                 elem_parse_failed = true;
1082                         break;
1083                 case WLAN_EID_MESH_ID:
1084                         elems->mesh_id = pos;
1085                         elems->mesh_id_len = elen;
1086                         break;
1087                 case WLAN_EID_MESH_CONFIG:
1088                         if (elen >= sizeof(struct ieee80211_meshconf_ie))
1089                                 elems->mesh_config = (void *)pos;
1090                         else
1091                                 elem_parse_failed = true;
1092                         break;
1093                 case WLAN_EID_PEER_MGMT:
1094                         elems->peering = pos;
1095                         elems->peering_len = elen;
1096                         break;
1097                 case WLAN_EID_MESH_AWAKE_WINDOW:
1098                         if (elen >= 2)
1099                                 elems->awake_window = (void *)pos;
1100                         break;
1101                 case WLAN_EID_PREQ:
1102                         elems->preq = pos;
1103                         elems->preq_len = elen;
1104                         break;
1105                 case WLAN_EID_PREP:
1106                         elems->prep = pos;
1107                         elems->prep_len = elen;
1108                         break;
1109                 case WLAN_EID_PERR:
1110                         elems->perr = pos;
1111                         elems->perr_len = elen;
1112                         break;
1113                 case WLAN_EID_RANN:
1114                         if (elen >= sizeof(struct ieee80211_rann_ie))
1115                                 elems->rann = (void *)pos;
1116                         else
1117                                 elem_parse_failed = true;
1118                         break;
1119                 case WLAN_EID_CHANNEL_SWITCH:
1120                         if (elen != sizeof(struct ieee80211_channel_sw_ie)) {
1121                                 elem_parse_failed = true;
1122                                 break;
1123                         }
1124                         elems->ch_switch_ie = (void *)pos;
1125                         break;
1126                 case WLAN_EID_EXT_CHANSWITCH_ANN:
1127                         if (elen != sizeof(struct ieee80211_ext_chansw_ie)) {
1128                                 elem_parse_failed = true;
1129                                 break;
1130                         }
1131                         elems->ext_chansw_ie = (void *)pos;
1132                         break;
1133                 case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
1134                         if (elen != sizeof(struct ieee80211_sec_chan_offs_ie)) {
1135                                 elem_parse_failed = true;
1136                                 break;
1137                         }
1138                         elems->sec_chan_offs = (void *)pos;
1139                         break;
1140                 case WLAN_EID_CHAN_SWITCH_PARAM:
1141                         if (elen !=
1142                             sizeof(*elems->mesh_chansw_params_ie)) {
1143                                 elem_parse_failed = true;
1144                                 break;
1145                         }
1146                         elems->mesh_chansw_params_ie = (void *)pos;
1147                         break;
1148                 case WLAN_EID_WIDE_BW_CHANNEL_SWITCH:
1149                         if (!action ||
1150                             elen != sizeof(*elems->wide_bw_chansw_ie)) {
1151                                 elem_parse_failed = true;
1152                                 break;
1153                         }
1154                         elems->wide_bw_chansw_ie = (void *)pos;
1155                         break;
1156                 case WLAN_EID_CHANNEL_SWITCH_WRAPPER:
1157                         if (action) {
1158                                 elem_parse_failed = true;
1159                                 break;
1160                         }
1161                         /*
1162                          * This is a bit tricky, but as we only care about
1163                          * the wide bandwidth channel switch element, so
1164                          * just parse it out manually.
1165                          */
1166                         ie = cfg80211_find_ie(WLAN_EID_WIDE_BW_CHANNEL_SWITCH,
1167                                               pos, elen);
1168                         if (ie) {
1169                                 if (ie[1] == sizeof(*elems->wide_bw_chansw_ie))
1170                                         elems->wide_bw_chansw_ie =
1171                                                 (void *)(ie + 2);
1172                                 else
1173                                         elem_parse_failed = true;
1174                         }
1175                         break;
1176                 case WLAN_EID_COUNTRY:
1177                         elems->country_elem = pos;
1178                         elems->country_elem_len = elen;
1179                         break;
1180                 case WLAN_EID_PWR_CONSTRAINT:
1181                         if (elen != 1) {
1182                                 elem_parse_failed = true;
1183                                 break;
1184                         }
1185                         elems->pwr_constr_elem = pos;
1186                         break;
1187                 case WLAN_EID_CISCO_VENDOR_SPECIFIC:
1188                         /* Lots of different options exist, but we only care
1189                          * about the Dynamic Transmit Power Control element.
1190                          * First check for the Cisco OUI, then for the DTPC
1191                          * tag (0x00).
1192                          */
1193                         if (elen < 4) {
1194                                 elem_parse_failed = true;
1195                                 break;
1196                         }
1197
1198                         if (pos[0] != 0x00 || pos[1] != 0x40 ||
1199                             pos[2] != 0x96 || pos[3] != 0x00)
1200                                 break;
1201
1202                         if (elen != 6) {
1203                                 elem_parse_failed = true;
1204                                 break;
1205                         }
1206
1207                         if (calc_crc)
1208                                 crc = crc32_be(crc, pos - 2, elen + 2);
1209
1210                         elems->cisco_dtpc_elem = pos;
1211                         break;
1212                 case WLAN_EID_TIMEOUT_INTERVAL:
1213                         if (elen >= sizeof(struct ieee80211_timeout_interval_ie))
1214                                 elems->timeout_int = (void *)pos;
1215                         else
1216                                 elem_parse_failed = true;
1217                         break;
1218                 case WLAN_EID_BSS_MAX_IDLE_PERIOD:
1219                         if (elen >= sizeof(*elems->max_idle_period_ie))
1220                                 elems->max_idle_period_ie = (void *)pos;
1221                         break;
1222                 case WLAN_EID_EXTENSION:
1223                         if (pos[0] == WLAN_EID_EXT_HE_MU_EDCA &&
1224                             elen >= (sizeof(*elems->mu_edca_param_set) + 1)) {
1225                                 elems->mu_edca_param_set = (void *)&pos[1];
1226                                 if (calc_crc)
1227                                         crc = crc32_be(crc, pos - 2, elen + 2);
1228                         } else if (pos[0] == WLAN_EID_EXT_HE_CAPABILITY) {
1229                                 elems->he_cap = (void *)&pos[1];
1230                                 elems->he_cap_len = elen - 1;
1231                         } else if (pos[0] == WLAN_EID_EXT_HE_OPERATION &&
1232                                    elen >= sizeof(*elems->he_operation) &&
1233                                    elen >= ieee80211_he_oper_size(&pos[1])) {
1234                                 elems->he_operation = (void *)&pos[1];
1235                         } else if (pos[0] == WLAN_EID_EXT_UORA && elen >= 1) {
1236                                 elems->uora_element = (void *)&pos[1];
1237                         }
1238                         break;
1239                 default:
1240                         break;
1241                 }
1242
1243                 if (elem_parse_failed)
1244                         elems->parse_error = true;
1245                 else
1246                         __set_bit(id, seen_elems);
1247
1248                 left -= elen;
1249                 pos += elen;
1250         }
1251
1252         if (left != 0)
1253                 elems->parse_error = true;
1254
1255         return crc;
1256 }
1257
1258 void ieee80211_regulatory_limit_wmm_params(struct ieee80211_sub_if_data *sdata,
1259                                            struct ieee80211_tx_queue_params
1260                                            *qparam, int ac)
1261 {
1262         struct ieee80211_chanctx_conf *chanctx_conf;
1263         const struct ieee80211_reg_rule *rrule;
1264         const struct ieee80211_wmm_ac *wmm_ac;
1265         u16 center_freq = 0;
1266
1267         if (sdata->vif.type != NL80211_IFTYPE_AP &&
1268             sdata->vif.type != NL80211_IFTYPE_STATION)
1269                 return;
1270
1271         rcu_read_lock();
1272         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1273         if (chanctx_conf)
1274                 center_freq = chanctx_conf->def.chan->center_freq;
1275
1276         if (!center_freq) {
1277                 rcu_read_unlock();
1278                 return;
1279         }
1280
1281         rrule = freq_reg_info(sdata->wdev.wiphy, MHZ_TO_KHZ(center_freq));
1282
1283         if (IS_ERR_OR_NULL(rrule) || !rrule->has_wmm) {
1284                 rcu_read_unlock();
1285                 return;
1286         }
1287
1288         if (sdata->vif.type == NL80211_IFTYPE_AP)
1289                 wmm_ac = &rrule->wmm_rule.ap[ac];
1290         else
1291                 wmm_ac = &rrule->wmm_rule.client[ac];
1292         qparam->cw_min = max_t(u16, qparam->cw_min, wmm_ac->cw_min);
1293         qparam->cw_max = max_t(u16, qparam->cw_max, wmm_ac->cw_max);
1294         qparam->aifs = max_t(u8, qparam->aifs, wmm_ac->aifsn);
1295         qparam->txop = min_t(u16, qparam->txop, wmm_ac->cot / 32);
1296         rcu_read_unlock();
1297 }
1298
1299 void ieee80211_set_wmm_default(struct ieee80211_sub_if_data *sdata,
1300                                bool bss_notify, bool enable_qos)
1301 {
1302         struct ieee80211_local *local = sdata->local;
1303         struct ieee80211_tx_queue_params qparam;
1304         struct ieee80211_chanctx_conf *chanctx_conf;
1305         int ac;
1306         bool use_11b;
1307         bool is_ocb; /* Use another EDCA parameters if dot11OCBActivated=true */
1308         int aCWmin, aCWmax;
1309
1310         if (!local->ops->conf_tx)
1311                 return;
1312
1313         if (local->hw.queues < IEEE80211_NUM_ACS)
1314                 return;
1315
1316         memset(&qparam, 0, sizeof(qparam));
1317
1318         rcu_read_lock();
1319         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
1320         use_11b = (chanctx_conf &&
1321                    chanctx_conf->def.chan->band == NL80211_BAND_2GHZ) &&
1322                  !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE);
1323         rcu_read_unlock();
1324
1325         is_ocb = (sdata->vif.type == NL80211_IFTYPE_OCB);
1326
1327         /* Set defaults according to 802.11-2007 Table 7-37 */
1328         aCWmax = 1023;
1329         if (use_11b)
1330                 aCWmin = 31;
1331         else
1332                 aCWmin = 15;
1333
1334         /* Confiure old 802.11b/g medium access rules. */
1335         qparam.cw_max = aCWmax;
1336         qparam.cw_min = aCWmin;
1337         qparam.txop = 0;
1338         qparam.aifs = 2;
1339
1340         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1341                 /* Update if QoS is enabled. */
1342                 if (enable_qos) {
1343                         switch (ac) {
1344                         case IEEE80211_AC_BK:
1345                                 qparam.cw_max = aCWmax;
1346                                 qparam.cw_min = aCWmin;
1347                                 qparam.txop = 0;
1348                                 if (is_ocb)
1349                                         qparam.aifs = 9;
1350                                 else
1351                                         qparam.aifs = 7;
1352                                 break;
1353                         /* never happens but let's not leave undefined */
1354                         default:
1355                         case IEEE80211_AC_BE:
1356                                 qparam.cw_max = aCWmax;
1357                                 qparam.cw_min = aCWmin;
1358                                 qparam.txop = 0;
1359                                 if (is_ocb)
1360                                         qparam.aifs = 6;
1361                                 else
1362                                         qparam.aifs = 3;
1363                                 break;
1364                         case IEEE80211_AC_VI:
1365                                 qparam.cw_max = aCWmin;
1366                                 qparam.cw_min = (aCWmin + 1) / 2 - 1;
1367                                 if (is_ocb)
1368                                         qparam.txop = 0;
1369                                 else if (use_11b)
1370                                         qparam.txop = 6016/32;
1371                                 else
1372                                         qparam.txop = 3008/32;
1373
1374                                 if (is_ocb)
1375                                         qparam.aifs = 3;
1376                                 else
1377                                         qparam.aifs = 2;
1378                                 break;
1379                         case IEEE80211_AC_VO:
1380                                 qparam.cw_max = (aCWmin + 1) / 2 - 1;
1381                                 qparam.cw_min = (aCWmin + 1) / 4 - 1;
1382                                 if (is_ocb)
1383                                         qparam.txop = 0;
1384                                 else if (use_11b)
1385                                         qparam.txop = 3264/32;
1386                                 else
1387                                         qparam.txop = 1504/32;
1388                                 qparam.aifs = 2;
1389                                 break;
1390                         }
1391                 }
1392                 ieee80211_regulatory_limit_wmm_params(sdata, &qparam, ac);
1393
1394                 qparam.uapsd = false;
1395
1396                 sdata->tx_conf[ac] = qparam;
1397                 drv_conf_tx(local, sdata, ac, &qparam);
1398         }
1399
1400         if (sdata->vif.type != NL80211_IFTYPE_MONITOR &&
1401             sdata->vif.type != NL80211_IFTYPE_P2P_DEVICE &&
1402             sdata->vif.type != NL80211_IFTYPE_NAN) {
1403                 sdata->vif.bss_conf.qos = enable_qos;
1404                 if (bss_notify)
1405                         ieee80211_bss_info_change_notify(sdata,
1406                                                          BSS_CHANGED_QOS);
1407         }
1408 }
1409
1410 void ieee80211_send_auth(struct ieee80211_sub_if_data *sdata,
1411                          u16 transaction, u16 auth_alg, u16 status,
1412                          const u8 *extra, size_t extra_len, const u8 *da,
1413                          const u8 *bssid, const u8 *key, u8 key_len, u8 key_idx,
1414                          u32 tx_flags)
1415 {
1416         struct ieee80211_local *local = sdata->local;
1417         struct sk_buff *skb;
1418         struct ieee80211_mgmt *mgmt;
1419         int err;
1420
1421         /* 24 + 6 = header + auth_algo + auth_transaction + status_code */
1422         skb = dev_alloc_skb(local->hw.extra_tx_headroom + IEEE80211_WEP_IV_LEN +
1423                             24 + 6 + extra_len + IEEE80211_WEP_ICV_LEN);
1424         if (!skb)
1425                 return;
1426
1427         skb_reserve(skb, local->hw.extra_tx_headroom + IEEE80211_WEP_IV_LEN);
1428
1429         mgmt = skb_put_zero(skb, 24 + 6);
1430         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1431                                           IEEE80211_STYPE_AUTH);
1432         memcpy(mgmt->da, da, ETH_ALEN);
1433         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1434         memcpy(mgmt->bssid, bssid, ETH_ALEN);
1435         mgmt->u.auth.auth_alg = cpu_to_le16(auth_alg);
1436         mgmt->u.auth.auth_transaction = cpu_to_le16(transaction);
1437         mgmt->u.auth.status_code = cpu_to_le16(status);
1438         if (extra)
1439                 skb_put_data(skb, extra, extra_len);
1440
1441         if (auth_alg == WLAN_AUTH_SHARED_KEY && transaction == 3) {
1442                 mgmt->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
1443                 err = ieee80211_wep_encrypt(local, skb, key, key_len, key_idx);
1444                 WARN_ON(err);
1445         }
1446
1447         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1448                                         tx_flags;
1449         ieee80211_tx_skb(sdata, skb);
1450 }
1451
1452 void ieee80211_send_deauth_disassoc(struct ieee80211_sub_if_data *sdata,
1453                                     const u8 *bssid, u16 stype, u16 reason,
1454                                     bool send_frame, u8 *frame_buf)
1455 {
1456         struct ieee80211_local *local = sdata->local;
1457         struct sk_buff *skb;
1458         struct ieee80211_mgmt *mgmt = (void *)frame_buf;
1459
1460         /* build frame */
1461         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT | stype);
1462         mgmt->duration = 0; /* initialize only */
1463         mgmt->seq_ctrl = 0; /* initialize only */
1464         memcpy(mgmt->da, bssid, ETH_ALEN);
1465         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
1466         memcpy(mgmt->bssid, bssid, ETH_ALEN);
1467         /* u.deauth.reason_code == u.disassoc.reason_code */
1468         mgmt->u.deauth.reason_code = cpu_to_le16(reason);
1469
1470         if (send_frame) {
1471                 skb = dev_alloc_skb(local->hw.extra_tx_headroom +
1472                                     IEEE80211_DEAUTH_FRAME_LEN);
1473                 if (!skb)
1474                         return;
1475
1476                 skb_reserve(skb, local->hw.extra_tx_headroom);
1477
1478                 /* copy in frame */
1479                 skb_put_data(skb, mgmt, IEEE80211_DEAUTH_FRAME_LEN);
1480
1481                 if (sdata->vif.type != NL80211_IFTYPE_STATION ||
1482                     !(sdata->u.mgd.flags & IEEE80211_STA_MFP_ENABLED))
1483                         IEEE80211_SKB_CB(skb)->flags |=
1484                                 IEEE80211_TX_INTFL_DONT_ENCRYPT;
1485
1486                 ieee80211_tx_skb(sdata, skb);
1487         }
1488 }
1489
1490 static int ieee80211_build_preq_ies_band(struct ieee80211_local *local,
1491                                          u8 *buffer, size_t buffer_len,
1492                                          const u8 *ie, size_t ie_len,
1493                                          enum nl80211_band band,
1494                                          u32 rate_mask,
1495                                          struct cfg80211_chan_def *chandef,
1496                                          size_t *offset, u32 flags)
1497 {
1498         struct ieee80211_supported_band *sband;
1499         const struct ieee80211_sta_he_cap *he_cap;
1500         u8 *pos = buffer, *end = buffer + buffer_len;
1501         size_t noffset;
1502         int supp_rates_len, i;
1503         u8 rates[32];
1504         int num_rates;
1505         int ext_rates_len;
1506         int shift;
1507         u32 rate_flags;
1508         bool have_80mhz = false;
1509
1510         *offset = 0;
1511
1512         sband = local->hw.wiphy->bands[band];
1513         if (WARN_ON_ONCE(!sband))
1514                 return 0;
1515
1516         rate_flags = ieee80211_chandef_rate_flags(chandef);
1517         shift = ieee80211_chandef_get_shift(chandef);
1518
1519         num_rates = 0;
1520         for (i = 0; i < sband->n_bitrates; i++) {
1521                 if ((BIT(i) & rate_mask) == 0)
1522                         continue; /* skip rate */
1523                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
1524                         continue;
1525
1526                 rates[num_rates++] =
1527                         (u8) DIV_ROUND_UP(sband->bitrates[i].bitrate,
1528                                           (1 << shift) * 5);
1529         }
1530
1531         supp_rates_len = min_t(int, num_rates, 8);
1532
1533         if (end - pos < 2 + supp_rates_len)
1534                 goto out_err;
1535         *pos++ = WLAN_EID_SUPP_RATES;
1536         *pos++ = supp_rates_len;
1537         memcpy(pos, rates, supp_rates_len);
1538         pos += supp_rates_len;
1539
1540         /* insert "request information" if in custom IEs */
1541         if (ie && ie_len) {
1542                 static const u8 before_extrates[] = {
1543                         WLAN_EID_SSID,
1544                         WLAN_EID_SUPP_RATES,
1545                         WLAN_EID_REQUEST,
1546                 };
1547                 noffset = ieee80211_ie_split(ie, ie_len,
1548                                              before_extrates,
1549                                              ARRAY_SIZE(before_extrates),
1550                                              *offset);
1551                 if (end - pos < noffset - *offset)
1552                         goto out_err;
1553                 memcpy(pos, ie + *offset, noffset - *offset);
1554                 pos += noffset - *offset;
1555                 *offset = noffset;
1556         }
1557
1558         ext_rates_len = num_rates - supp_rates_len;
1559         if (ext_rates_len > 0) {
1560                 if (end - pos < 2 + ext_rates_len)
1561                         goto out_err;
1562                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
1563                 *pos++ = ext_rates_len;
1564                 memcpy(pos, rates + supp_rates_len, ext_rates_len);
1565                 pos += ext_rates_len;
1566         }
1567
1568         if (chandef->chan && sband->band == NL80211_BAND_2GHZ) {
1569                 if (end - pos < 3)
1570                         goto out_err;
1571                 *pos++ = WLAN_EID_DS_PARAMS;
1572                 *pos++ = 1;
1573                 *pos++ = ieee80211_frequency_to_channel(
1574                                 chandef->chan->center_freq);
1575         }
1576
1577         if (flags & IEEE80211_PROBE_FLAG_MIN_CONTENT)
1578                 goto done;
1579
1580         /* insert custom IEs that go before HT */
1581         if (ie && ie_len) {
1582                 static const u8 before_ht[] = {
1583                         /*
1584                          * no need to list the ones split off already
1585                          * (or generated here)
1586                          */
1587                         WLAN_EID_DS_PARAMS,
1588                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
1589                 };
1590                 noffset = ieee80211_ie_split(ie, ie_len,
1591                                              before_ht, ARRAY_SIZE(before_ht),
1592                                              *offset);
1593                 if (end - pos < noffset - *offset)
1594                         goto out_err;
1595                 memcpy(pos, ie + *offset, noffset - *offset);
1596                 pos += noffset - *offset;
1597                 *offset = noffset;
1598         }
1599
1600         if (sband->ht_cap.ht_supported) {
1601                 if (end - pos < 2 + sizeof(struct ieee80211_ht_cap))
1602                         goto out_err;
1603                 pos = ieee80211_ie_build_ht_cap(pos, &sband->ht_cap,
1604                                                 sband->ht_cap.cap);
1605         }
1606
1607         /* insert custom IEs that go before VHT */
1608         if (ie && ie_len) {
1609                 static const u8 before_vht[] = {
1610                         /*
1611                          * no need to list the ones split off already
1612                          * (or generated here)
1613                          */
1614                         WLAN_EID_BSS_COEX_2040,
1615                         WLAN_EID_EXT_CAPABILITY,
1616                         WLAN_EID_SSID_LIST,
1617                         WLAN_EID_CHANNEL_USAGE,
1618                         WLAN_EID_INTERWORKING,
1619                         WLAN_EID_MESH_ID,
1620                         /* 60 GHz (Multi-band, DMG, MMS) can't happen */
1621                 };
1622                 noffset = ieee80211_ie_split(ie, ie_len,
1623                                              before_vht, ARRAY_SIZE(before_vht),
1624                                              *offset);
1625                 if (end - pos < noffset - *offset)
1626                         goto out_err;
1627                 memcpy(pos, ie + *offset, noffset - *offset);
1628                 pos += noffset - *offset;
1629                 *offset = noffset;
1630         }
1631
1632         /* Check if any channel in this sband supports at least 80 MHz */
1633         for (i = 0; i < sband->n_channels; i++) {
1634                 if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
1635                                                 IEEE80211_CHAN_NO_80MHZ))
1636                         continue;
1637
1638                 have_80mhz = true;
1639                 break;
1640         }
1641
1642         if (sband->vht_cap.vht_supported && have_80mhz) {
1643                 if (end - pos < 2 + sizeof(struct ieee80211_vht_cap))
1644                         goto out_err;
1645                 pos = ieee80211_ie_build_vht_cap(pos, &sband->vht_cap,
1646                                                  sband->vht_cap.cap);
1647         }
1648
1649         /* insert custom IEs that go before HE */
1650         if (ie && ie_len) {
1651                 static const u8 before_he[] = {
1652                         /*
1653                          * no need to list the ones split off before VHT
1654                          * or generated here
1655                          */
1656                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_REQ_PARAMS,
1657                         WLAN_EID_AP_CSN,
1658                         /* TODO: add 11ah/11aj/11ak elements */
1659                 };
1660                 noffset = ieee80211_ie_split(ie, ie_len,
1661                                              before_he, ARRAY_SIZE(before_he),
1662                                              *offset);
1663                 if (end - pos < noffset - *offset)
1664                         goto out_err;
1665                 memcpy(pos, ie + *offset, noffset - *offset);
1666                 pos += noffset - *offset;
1667                 *offset = noffset;
1668         }
1669
1670         he_cap = ieee80211_get_he_sta_cap(sband);
1671         if (he_cap) {
1672                 pos = ieee80211_ie_build_he_cap(pos, he_cap, end);
1673                 if (!pos)
1674                         goto out_err;
1675         }
1676
1677         /*
1678          * If adding more here, adjust code in main.c
1679          * that calculates local->scan_ies_len.
1680          */
1681
1682         return pos - buffer;
1683  out_err:
1684         WARN_ONCE(1, "not enough space for preq IEs\n");
1685  done:
1686         return pos - buffer;
1687 }
1688
1689 int ieee80211_build_preq_ies(struct ieee80211_local *local, u8 *buffer,
1690                              size_t buffer_len,
1691                              struct ieee80211_scan_ies *ie_desc,
1692                              const u8 *ie, size_t ie_len,
1693                              u8 bands_used, u32 *rate_masks,
1694                              struct cfg80211_chan_def *chandef,
1695                              u32 flags)
1696 {
1697         size_t pos = 0, old_pos = 0, custom_ie_offset = 0;
1698         int i;
1699
1700         memset(ie_desc, 0, sizeof(*ie_desc));
1701
1702         for (i = 0; i < NUM_NL80211_BANDS; i++) {
1703                 if (bands_used & BIT(i)) {
1704                         pos += ieee80211_build_preq_ies_band(local,
1705                                                              buffer + pos,
1706                                                              buffer_len - pos,
1707                                                              ie, ie_len, i,
1708                                                              rate_masks[i],
1709                                                              chandef,
1710                                                              &custom_ie_offset,
1711                                                              flags);
1712                         ie_desc->ies[i] = buffer + old_pos;
1713                         ie_desc->len[i] = pos - old_pos;
1714                         old_pos = pos;
1715                 }
1716         }
1717
1718         /* add any remaining custom IEs */
1719         if (ie && ie_len) {
1720                 if (WARN_ONCE(buffer_len - pos < ie_len - custom_ie_offset,
1721                               "not enough space for preq custom IEs\n"))
1722                         return pos;
1723                 memcpy(buffer + pos, ie + custom_ie_offset,
1724                        ie_len - custom_ie_offset);
1725                 ie_desc->common_ies = buffer + pos;
1726                 ie_desc->common_ie_len = ie_len - custom_ie_offset;
1727                 pos += ie_len - custom_ie_offset;
1728         }
1729
1730         return pos;
1731 };
1732
1733 struct sk_buff *ieee80211_build_probe_req(struct ieee80211_sub_if_data *sdata,
1734                                           const u8 *src, const u8 *dst,
1735                                           u32 ratemask,
1736                                           struct ieee80211_channel *chan,
1737                                           const u8 *ssid, size_t ssid_len,
1738                                           const u8 *ie, size_t ie_len,
1739                                           u32 flags)
1740 {
1741         struct ieee80211_local *local = sdata->local;
1742         struct cfg80211_chan_def chandef;
1743         struct sk_buff *skb;
1744         struct ieee80211_mgmt *mgmt;
1745         int ies_len;
1746         u32 rate_masks[NUM_NL80211_BANDS] = {};
1747         struct ieee80211_scan_ies dummy_ie_desc;
1748
1749         /*
1750          * Do not send DS Channel parameter for directed probe requests
1751          * in order to maximize the chance that we get a response.  Some
1752          * badly-behaved APs don't respond when this parameter is included.
1753          */
1754         chandef.width = sdata->vif.bss_conf.chandef.width;
1755         if (flags & IEEE80211_PROBE_FLAG_DIRECTED)
1756                 chandef.chan = NULL;
1757         else
1758                 chandef.chan = chan;
1759
1760         skb = ieee80211_probereq_get(&local->hw, src, ssid, ssid_len,
1761                                      100 + ie_len);
1762         if (!skb)
1763                 return NULL;
1764
1765         rate_masks[chan->band] = ratemask;
1766         ies_len = ieee80211_build_preq_ies(local, skb_tail_pointer(skb),
1767                                            skb_tailroom(skb), &dummy_ie_desc,
1768                                            ie, ie_len, BIT(chan->band),
1769                                            rate_masks, &chandef, flags);
1770         skb_put(skb, ies_len);
1771
1772         if (dst) {
1773                 mgmt = (struct ieee80211_mgmt *) skb->data;
1774                 memcpy(mgmt->da, dst, ETH_ALEN);
1775                 memcpy(mgmt->bssid, dst, ETH_ALEN);
1776         }
1777
1778         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1779
1780         return skb;
1781 }
1782
1783 u32 ieee80211_sta_get_rates(struct ieee80211_sub_if_data *sdata,
1784                             struct ieee802_11_elems *elems,
1785                             enum nl80211_band band, u32 *basic_rates)
1786 {
1787         struct ieee80211_supported_band *sband;
1788         size_t num_rates;
1789         u32 supp_rates, rate_flags;
1790         int i, j, shift;
1791
1792         sband = sdata->local->hw.wiphy->bands[band];
1793         if (WARN_ON(!sband))
1794                 return 1;
1795
1796         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
1797         shift = ieee80211_vif_get_shift(&sdata->vif);
1798
1799         num_rates = sband->n_bitrates;
1800         supp_rates = 0;
1801         for (i = 0; i < elems->supp_rates_len +
1802                      elems->ext_supp_rates_len; i++) {
1803                 u8 rate = 0;
1804                 int own_rate;
1805                 bool is_basic;
1806                 if (i < elems->supp_rates_len)
1807                         rate = elems->supp_rates[i];
1808                 else if (elems->ext_supp_rates)
1809                         rate = elems->ext_supp_rates
1810                                 [i - elems->supp_rates_len];
1811                 own_rate = 5 * (rate & 0x7f);
1812                 is_basic = !!(rate & 0x80);
1813
1814                 if (is_basic && (rate & 0x7f) == BSS_MEMBERSHIP_SELECTOR_HT_PHY)
1815                         continue;
1816
1817                 for (j = 0; j < num_rates; j++) {
1818                         int brate;
1819                         if ((rate_flags & sband->bitrates[j].flags)
1820                             != rate_flags)
1821                                 continue;
1822
1823                         brate = DIV_ROUND_UP(sband->bitrates[j].bitrate,
1824                                              1 << shift);
1825
1826                         if (brate == own_rate) {
1827                                 supp_rates |= BIT(j);
1828                                 if (basic_rates && is_basic)
1829                                         *basic_rates |= BIT(j);
1830                         }
1831                 }
1832         }
1833         return supp_rates;
1834 }
1835
1836 void ieee80211_stop_device(struct ieee80211_local *local)
1837 {
1838         ieee80211_led_radio(local, false);
1839         ieee80211_mod_tpt_led_trig(local, 0, IEEE80211_TPT_LEDTRIG_FL_RADIO);
1840
1841         cancel_work_sync(&local->reconfig_filter);
1842
1843         flush_workqueue(local->workqueue);
1844         drv_stop(local);
1845 }
1846
1847 static void ieee80211_flush_completed_scan(struct ieee80211_local *local,
1848                                            bool aborted)
1849 {
1850         /* It's possible that we don't handle the scan completion in
1851          * time during suspend, so if it's still marked as completed
1852          * here, queue the work and flush it to clean things up.
1853          * Instead of calling the worker function directly here, we
1854          * really queue it to avoid potential races with other flows
1855          * scheduling the same work.
1856          */
1857         if (test_bit(SCAN_COMPLETED, &local->scanning)) {
1858                 /* If coming from reconfiguration failure, abort the scan so
1859                  * we don't attempt to continue a partial HW scan - which is
1860                  * possible otherwise if (e.g.) the 2.4 GHz portion was the
1861                  * completed scan, and a 5 GHz portion is still pending.
1862                  */
1863                 if (aborted)
1864                         set_bit(SCAN_ABORTED, &local->scanning);
1865                 ieee80211_queue_delayed_work(&local->hw, &local->scan_work, 0);
1866                 flush_delayed_work(&local->scan_work);
1867         }
1868 }
1869
1870 static void ieee80211_handle_reconfig_failure(struct ieee80211_local *local)
1871 {
1872         struct ieee80211_sub_if_data *sdata;
1873         struct ieee80211_chanctx *ctx;
1874
1875         /*
1876          * We get here if during resume the device can't be restarted properly.
1877          * We might also get here if this happens during HW reset, which is a
1878          * slightly different situation and we need to drop all connections in
1879          * the latter case.
1880          *
1881          * Ask cfg80211 to turn off all interfaces, this will result in more
1882          * warnings but at least we'll then get into a clean stopped state.
1883          */
1884
1885         local->resuming = false;
1886         local->suspended = false;
1887         local->in_reconfig = false;
1888
1889         ieee80211_flush_completed_scan(local, true);
1890
1891         /* scheduled scan clearly can't be running any more, but tell
1892          * cfg80211 and clear local state
1893          */
1894         ieee80211_sched_scan_end(local);
1895
1896         list_for_each_entry(sdata, &local->interfaces, list)
1897                 sdata->flags &= ~IEEE80211_SDATA_IN_DRIVER;
1898
1899         /* Mark channel contexts as not being in the driver any more to avoid
1900          * removing them from the driver during the shutdown process...
1901          */
1902         mutex_lock(&local->chanctx_mtx);
1903         list_for_each_entry(ctx, &local->chanctx_list, list)
1904                 ctx->driver_present = false;
1905         mutex_unlock(&local->chanctx_mtx);
1906
1907         cfg80211_shutdown_all_interfaces(local->hw.wiphy);
1908 }
1909
1910 static void ieee80211_assign_chanctx(struct ieee80211_local *local,
1911                                      struct ieee80211_sub_if_data *sdata)
1912 {
1913         struct ieee80211_chanctx_conf *conf;
1914         struct ieee80211_chanctx *ctx;
1915
1916         if (!local->use_chanctx)
1917                 return;
1918
1919         mutex_lock(&local->chanctx_mtx);
1920         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1921                                          lockdep_is_held(&local->chanctx_mtx));
1922         if (conf) {
1923                 ctx = container_of(conf, struct ieee80211_chanctx, conf);
1924                 drv_assign_vif_chanctx(local, sdata, ctx);
1925         }
1926         mutex_unlock(&local->chanctx_mtx);
1927 }
1928
1929 static void ieee80211_reconfig_stations(struct ieee80211_sub_if_data *sdata)
1930 {
1931         struct ieee80211_local *local = sdata->local;
1932         struct sta_info *sta;
1933
1934         /* add STAs back */
1935         mutex_lock(&local->sta_mtx);
1936         list_for_each_entry(sta, &local->sta_list, list) {
1937                 enum ieee80211_sta_state state;
1938
1939                 if (!sta->uploaded || sta->sdata != sdata)
1940                         continue;
1941
1942                 for (state = IEEE80211_STA_NOTEXIST;
1943                      state < sta->sta_state; state++)
1944                         WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
1945                                               state + 1));
1946         }
1947         mutex_unlock(&local->sta_mtx);
1948 }
1949
1950 static int ieee80211_reconfig_nan(struct ieee80211_sub_if_data *sdata)
1951 {
1952         struct cfg80211_nan_func *func, **funcs;
1953         int res, id, i = 0;
1954
1955         res = drv_start_nan(sdata->local, sdata,
1956                             &sdata->u.nan.conf);
1957         if (WARN_ON(res))
1958                 return res;
1959
1960         funcs = kcalloc(sdata->local->hw.max_nan_de_entries + 1,
1961                         sizeof(*funcs),
1962                         GFP_KERNEL);
1963         if (!funcs)
1964                 return -ENOMEM;
1965
1966         /* Add all the functions:
1967          * This is a little bit ugly. We need to call a potentially sleeping
1968          * callback for each NAN function, so we can't hold the spinlock.
1969          */
1970         spin_lock_bh(&sdata->u.nan.func_lock);
1971
1972         idr_for_each_entry(&sdata->u.nan.function_inst_ids, func, id)
1973                 funcs[i++] = func;
1974
1975         spin_unlock_bh(&sdata->u.nan.func_lock);
1976
1977         for (i = 0; funcs[i]; i++) {
1978                 res = drv_add_nan_func(sdata->local, sdata, funcs[i]);
1979                 if (WARN_ON(res))
1980                         ieee80211_nan_func_terminated(&sdata->vif,
1981                                                       funcs[i]->instance_id,
1982                                                       NL80211_NAN_FUNC_TERM_REASON_ERROR,
1983                                                       GFP_KERNEL);
1984         }
1985
1986         kfree(funcs);
1987
1988         return 0;
1989 }
1990
1991 int ieee80211_reconfig(struct ieee80211_local *local)
1992 {
1993         struct ieee80211_hw *hw = &local->hw;
1994         struct ieee80211_sub_if_data *sdata;
1995         struct ieee80211_chanctx *ctx;
1996         struct sta_info *sta;
1997         int res, i;
1998         bool reconfig_due_to_wowlan = false;
1999         struct ieee80211_sub_if_data *sched_scan_sdata;
2000         struct cfg80211_sched_scan_request *sched_scan_req;
2001         bool sched_scan_stopped = false;
2002         bool suspended = local->suspended;
2003
2004         /* nothing to do if HW shouldn't run */
2005         if (!local->open_count)
2006                 goto wake_up;
2007
2008 #ifdef CONFIG_PM
2009         if (suspended)
2010                 local->resuming = true;
2011
2012         if (local->wowlan) {
2013                 /*
2014                  * In the wowlan case, both mac80211 and the device
2015                  * are functional when the resume op is called, so
2016                  * clear local->suspended so the device could operate
2017                  * normally (e.g. pass rx frames).
2018                  */
2019                 local->suspended = false;
2020                 res = drv_resume(local);
2021                 local->wowlan = false;
2022                 if (res < 0) {
2023                         local->resuming = false;
2024                         return res;
2025                 }
2026                 if (res == 0)
2027                         goto wake_up;
2028                 WARN_ON(res > 1);
2029                 /*
2030                  * res is 1, which means the driver requested
2031                  * to go through a regular reset on wakeup.
2032                  * restore local->suspended in this case.
2033                  */
2034                 reconfig_due_to_wowlan = true;
2035                 local->suspended = true;
2036         }
2037 #endif
2038
2039         /*
2040          * In case of hw_restart during suspend (without wowlan),
2041          * cancel restart work, as we are reconfiguring the device
2042          * anyway.
2043          * Note that restart_work is scheduled on a frozen workqueue,
2044          * so we can't deadlock in this case.
2045          */
2046         if (suspended && local->in_reconfig && !reconfig_due_to_wowlan)
2047                 cancel_work_sync(&local->restart_work);
2048
2049         local->started = false;
2050
2051         /*
2052          * Upon resume hardware can sometimes be goofy due to
2053          * various platform / driver / bus issues, so restarting
2054          * the device may at times not work immediately. Propagate
2055          * the error.
2056          */
2057         res = drv_start(local);
2058         if (res) {
2059                 if (suspended)
2060                         WARN(1, "Hardware became unavailable upon resume. This could be a software issue prior to suspend or a hardware issue.\n");
2061                 else
2062                         WARN(1, "Hardware became unavailable during restart.\n");
2063                 ieee80211_handle_reconfig_failure(local);
2064                 return res;
2065         }
2066
2067         /* setup fragmentation threshold */
2068         drv_set_frag_threshold(local, hw->wiphy->frag_threshold);
2069
2070         /* setup RTS threshold */
2071         drv_set_rts_threshold(local, hw->wiphy->rts_threshold);
2072
2073         /* reset coverage class */
2074         drv_set_coverage_class(local, hw->wiphy->coverage_class);
2075
2076         ieee80211_led_radio(local, true);
2077         ieee80211_mod_tpt_led_trig(local,
2078                                    IEEE80211_TPT_LEDTRIG_FL_RADIO, 0);
2079
2080         /* add interfaces */
2081         sdata = rtnl_dereference(local->monitor_sdata);
2082         if (sdata) {
2083                 /* in HW restart it exists already */
2084                 WARN_ON(local->resuming);
2085                 res = drv_add_interface(local, sdata);
2086                 if (WARN_ON(res)) {
2087                         RCU_INIT_POINTER(local->monitor_sdata, NULL);
2088                         synchronize_net();
2089                         kfree(sdata);
2090                 }
2091         }
2092
2093         list_for_each_entry(sdata, &local->interfaces, list) {
2094                 if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
2095                     sdata->vif.type != NL80211_IFTYPE_MONITOR &&
2096                     ieee80211_sdata_running(sdata)) {
2097                         res = drv_add_interface(local, sdata);
2098                         if (WARN_ON(res))
2099                                 break;
2100                 }
2101         }
2102
2103         /* If adding any of the interfaces failed above, roll back and
2104          * report failure.
2105          */
2106         if (res) {
2107                 list_for_each_entry_continue_reverse(sdata, &local->interfaces,
2108                                                      list)
2109                         if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
2110                             sdata->vif.type != NL80211_IFTYPE_MONITOR &&
2111                             ieee80211_sdata_running(sdata))
2112                                 drv_remove_interface(local, sdata);
2113                 ieee80211_handle_reconfig_failure(local);
2114                 return res;
2115         }
2116
2117         /* add channel contexts */
2118         if (local->use_chanctx) {
2119                 mutex_lock(&local->chanctx_mtx);
2120                 list_for_each_entry(ctx, &local->chanctx_list, list)
2121                         if (ctx->replace_state !=
2122                             IEEE80211_CHANCTX_REPLACES_OTHER)
2123                                 WARN_ON(drv_add_chanctx(local, ctx));
2124                 mutex_unlock(&local->chanctx_mtx);
2125
2126                 sdata = rtnl_dereference(local->monitor_sdata);
2127                 if (sdata && ieee80211_sdata_running(sdata))
2128                         ieee80211_assign_chanctx(local, sdata);
2129         }
2130
2131         /* reconfigure hardware */
2132         ieee80211_hw_config(local, ~0);
2133
2134         ieee80211_configure_filter(local);
2135
2136         /* Finally also reconfigure all the BSS information */
2137         list_for_each_entry(sdata, &local->interfaces, list) {
2138                 u32 changed;
2139
2140                 if (!ieee80211_sdata_running(sdata))
2141                         continue;
2142
2143                 ieee80211_assign_chanctx(local, sdata);
2144
2145                 switch (sdata->vif.type) {
2146                 case NL80211_IFTYPE_AP_VLAN:
2147                 case NL80211_IFTYPE_MONITOR:
2148                         break;
2149                 default:
2150                         ieee80211_reconfig_stations(sdata);
2151                         /* fall through */
2152                 case NL80211_IFTYPE_AP: /* AP stations are handled later */
2153                         for (i = 0; i < IEEE80211_NUM_ACS; i++)
2154                                 drv_conf_tx(local, sdata, i,
2155                                             &sdata->tx_conf[i]);
2156                         break;
2157                 }
2158
2159                 /* common change flags for all interface types */
2160                 changed = BSS_CHANGED_ERP_CTS_PROT |
2161                           BSS_CHANGED_ERP_PREAMBLE |
2162                           BSS_CHANGED_ERP_SLOT |
2163                           BSS_CHANGED_HT |
2164                           BSS_CHANGED_BASIC_RATES |
2165                           BSS_CHANGED_BEACON_INT |
2166                           BSS_CHANGED_BSSID |
2167                           BSS_CHANGED_CQM |
2168                           BSS_CHANGED_QOS |
2169                           BSS_CHANGED_IDLE |
2170                           BSS_CHANGED_TXPOWER |
2171                           BSS_CHANGED_MCAST_RATE;
2172
2173                 if (sdata->vif.mu_mimo_owner)
2174                         changed |= BSS_CHANGED_MU_GROUPS;
2175
2176                 switch (sdata->vif.type) {
2177                 case NL80211_IFTYPE_STATION:
2178                         changed |= BSS_CHANGED_ASSOC |
2179                                    BSS_CHANGED_ARP_FILTER |
2180                                    BSS_CHANGED_PS;
2181
2182                         /* Re-send beacon info report to the driver */
2183                         if (sdata->u.mgd.have_beacon)
2184                                 changed |= BSS_CHANGED_BEACON_INFO;
2185
2186                         if (sdata->vif.bss_conf.max_idle_period ||
2187                             sdata->vif.bss_conf.protected_keep_alive)
2188                                 changed |= BSS_CHANGED_KEEP_ALIVE;
2189
2190                         sdata_lock(sdata);
2191                         ieee80211_bss_info_change_notify(sdata, changed);
2192                         sdata_unlock(sdata);
2193                         break;
2194                 case NL80211_IFTYPE_OCB:
2195                         changed |= BSS_CHANGED_OCB;
2196                         ieee80211_bss_info_change_notify(sdata, changed);
2197                         break;
2198                 case NL80211_IFTYPE_ADHOC:
2199                         changed |= BSS_CHANGED_IBSS;
2200                         /* fall through */
2201                 case NL80211_IFTYPE_AP:
2202                         changed |= BSS_CHANGED_SSID | BSS_CHANGED_P2P_PS;
2203
2204                         if (sdata->vif.bss_conf.ftm_responder == 1 &&
2205                             wiphy_ext_feature_isset(sdata->local->hw.wiphy,
2206                                         NL80211_EXT_FEATURE_ENABLE_FTM_RESPONDER))
2207                                 changed |= BSS_CHANGED_FTM_RESPONDER;
2208
2209                         if (sdata->vif.type == NL80211_IFTYPE_AP) {
2210                                 changed |= BSS_CHANGED_AP_PROBE_RESP;
2211
2212                                 if (rcu_access_pointer(sdata->u.ap.beacon))
2213                                         drv_start_ap(local, sdata);
2214                         }
2215
2216                         /* fall through */
2217                 case NL80211_IFTYPE_MESH_POINT:
2218                         if (sdata->vif.bss_conf.enable_beacon) {
2219                                 changed |= BSS_CHANGED_BEACON |
2220                                            BSS_CHANGED_BEACON_ENABLED;
2221                                 ieee80211_bss_info_change_notify(sdata, changed);
2222                         }
2223                         break;
2224                 case NL80211_IFTYPE_NAN:
2225                         res = ieee80211_reconfig_nan(sdata);
2226                         if (res < 0) {
2227                                 ieee80211_handle_reconfig_failure(local);
2228                                 return res;
2229                         }
2230                         break;
2231                 case NL80211_IFTYPE_WDS:
2232                 case NL80211_IFTYPE_AP_VLAN:
2233                 case NL80211_IFTYPE_MONITOR:
2234                 case NL80211_IFTYPE_P2P_DEVICE:
2235                         /* nothing to do */
2236                         break;
2237                 case NL80211_IFTYPE_UNSPECIFIED:
2238                 case NUM_NL80211_IFTYPES:
2239                 case NL80211_IFTYPE_P2P_CLIENT:
2240                 case NL80211_IFTYPE_P2P_GO:
2241                         WARN_ON(1);
2242                         break;
2243                 }
2244         }
2245
2246         ieee80211_recalc_ps(local);
2247
2248         /*
2249          * The sta might be in psm against the ap (e.g. because
2250          * this was the state before a hw restart), so we
2251          * explicitly send a null packet in order to make sure
2252          * it'll sync against the ap (and get out of psm).
2253          */
2254         if (!(local->hw.conf.flags & IEEE80211_CONF_PS)) {
2255                 list_for_each_entry(sdata, &local->interfaces, list) {
2256                         if (sdata->vif.type != NL80211_IFTYPE_STATION)
2257                                 continue;
2258                         if (!sdata->u.mgd.associated)
2259                                 continue;
2260
2261                         ieee80211_send_nullfunc(local, sdata, false);
2262                 }
2263         }
2264
2265         /* APs are now beaconing, add back stations */
2266         mutex_lock(&local->sta_mtx);
2267         list_for_each_entry(sta, &local->sta_list, list) {
2268                 enum ieee80211_sta_state state;
2269
2270                 if (!sta->uploaded)
2271                         continue;
2272
2273                 if (sta->sdata->vif.type != NL80211_IFTYPE_AP &&
2274                     sta->sdata->vif.type != NL80211_IFTYPE_AP_VLAN)
2275                         continue;
2276
2277                 for (state = IEEE80211_STA_NOTEXIST;
2278                      state < sta->sta_state; state++)
2279                         WARN_ON(drv_sta_state(local, sta->sdata, sta, state,
2280                                               state + 1));
2281         }
2282         mutex_unlock(&local->sta_mtx);
2283
2284         /* add back keys */
2285         list_for_each_entry(sdata, &local->interfaces, list)
2286                 ieee80211_reset_crypto_tx_tailroom(sdata);
2287
2288         list_for_each_entry(sdata, &local->interfaces, list)
2289                 if (ieee80211_sdata_running(sdata))
2290                         ieee80211_enable_keys(sdata);
2291
2292         /* Reconfigure sched scan if it was interrupted by FW restart */
2293         mutex_lock(&local->mtx);
2294         sched_scan_sdata = rcu_dereference_protected(local->sched_scan_sdata,
2295                                                 lockdep_is_held(&local->mtx));
2296         sched_scan_req = rcu_dereference_protected(local->sched_scan_req,
2297                                                 lockdep_is_held(&local->mtx));
2298         if (sched_scan_sdata && sched_scan_req)
2299                 /*
2300                  * Sched scan stopped, but we don't want to report it. Instead,
2301                  * we're trying to reschedule. However, if more than one scan
2302                  * plan was set, we cannot reschedule since we don't know which
2303                  * scan plan was currently running (and some scan plans may have
2304                  * already finished).
2305                  */
2306                 if (sched_scan_req->n_scan_plans > 1 ||
2307                     __ieee80211_request_sched_scan_start(sched_scan_sdata,
2308                                                          sched_scan_req)) {
2309                         RCU_INIT_POINTER(local->sched_scan_sdata, NULL);
2310                         RCU_INIT_POINTER(local->sched_scan_req, NULL);
2311                         sched_scan_stopped = true;
2312                 }
2313         mutex_unlock(&local->mtx);
2314
2315         if (sched_scan_stopped)
2316                 cfg80211_sched_scan_stopped_rtnl(local->hw.wiphy, 0);
2317
2318  wake_up:
2319
2320         if (local->monitors == local->open_count && local->monitors > 0)
2321                 ieee80211_add_virtual_monitor(local);
2322
2323         /*
2324          * Clear the WLAN_STA_BLOCK_BA flag so new aggregation
2325          * sessions can be established after a resume.
2326          *
2327          * Also tear down aggregation sessions since reconfiguring
2328          * them in a hardware restart scenario is not easily done
2329          * right now, and the hardware will have lost information
2330          * about the sessions, but we and the AP still think they
2331          * are active. This is really a workaround though.
2332          */
2333         if (ieee80211_hw_check(hw, AMPDU_AGGREGATION)) {
2334                 mutex_lock(&local->sta_mtx);
2335
2336                 list_for_each_entry(sta, &local->sta_list, list) {
2337                         if (!local->resuming)
2338                                 ieee80211_sta_tear_down_BA_sessions(
2339                                                 sta, AGG_STOP_LOCAL_REQUEST);
2340                         clear_sta_flag(sta, WLAN_STA_BLOCK_BA);
2341                 }
2342
2343                 mutex_unlock(&local->sta_mtx);
2344         }
2345
2346         if (local->in_reconfig) {
2347                 local->in_reconfig = false;
2348                 barrier();
2349
2350                 /* Restart deferred ROCs */
2351                 mutex_lock(&local->mtx);
2352                 ieee80211_start_next_roc(local);
2353                 mutex_unlock(&local->mtx);
2354         }
2355
2356         ieee80211_wake_queues_by_reason(hw, IEEE80211_MAX_QUEUE_MAP,
2357                                         IEEE80211_QUEUE_STOP_REASON_SUSPEND,
2358                                         false);
2359
2360         /*
2361          * If this is for hw restart things are still running.
2362          * We may want to change that later, however.
2363          */
2364         if (local->open_count && (!suspended || reconfig_due_to_wowlan))
2365                 drv_reconfig_complete(local, IEEE80211_RECONFIG_TYPE_RESTART);
2366
2367         if (!suspended)
2368                 return 0;
2369
2370 #ifdef CONFIG_PM
2371         /* first set suspended false, then resuming */
2372         local->suspended = false;
2373         mb();
2374         local->resuming = false;
2375
2376         ieee80211_flush_completed_scan(local, false);
2377
2378         if (local->open_count && !reconfig_due_to_wowlan)
2379                 drv_reconfig_complete(local, IEEE80211_RECONFIG_TYPE_SUSPEND);
2380
2381         list_for_each_entry(sdata, &local->interfaces, list) {
2382                 if (!ieee80211_sdata_running(sdata))
2383                         continue;
2384                 if (sdata->vif.type == NL80211_IFTYPE_STATION)
2385                         ieee80211_sta_restart(sdata);
2386         }
2387
2388         mod_timer(&local->sta_cleanup, jiffies + 1);
2389 #else
2390         WARN_ON(1);
2391 #endif
2392
2393         return 0;
2394 }
2395
2396 void ieee80211_resume_disconnect(struct ieee80211_vif *vif)
2397 {
2398         struct ieee80211_sub_if_data *sdata;
2399         struct ieee80211_local *local;
2400         struct ieee80211_key *key;
2401
2402         if (WARN_ON(!vif))
2403                 return;
2404
2405         sdata = vif_to_sdata(vif);
2406         local = sdata->local;
2407
2408         if (WARN_ON(!local->resuming))
2409                 return;
2410
2411         if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
2412                 return;
2413
2414         sdata->flags |= IEEE80211_SDATA_DISCONNECT_RESUME;
2415
2416         mutex_lock(&local->key_mtx);
2417         list_for_each_entry(key, &sdata->key_list, list)
2418                 key->flags |= KEY_FLAG_TAINTED;
2419         mutex_unlock(&local->key_mtx);
2420 }
2421 EXPORT_SYMBOL_GPL(ieee80211_resume_disconnect);
2422
2423 void ieee80211_recalc_smps(struct ieee80211_sub_if_data *sdata)
2424 {
2425         struct ieee80211_local *local = sdata->local;
2426         struct ieee80211_chanctx_conf *chanctx_conf;
2427         struct ieee80211_chanctx *chanctx;
2428
2429         mutex_lock(&local->chanctx_mtx);
2430
2431         chanctx_conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
2432                                         lockdep_is_held(&local->chanctx_mtx));
2433
2434         /*
2435          * This function can be called from a work, thus it may be possible
2436          * that the chanctx_conf is removed (due to a disconnection, for
2437          * example).
2438          * So nothing should be done in such case.
2439          */
2440         if (!chanctx_conf)
2441                 goto unlock;
2442
2443         chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
2444         ieee80211_recalc_smps_chanctx(local, chanctx);
2445  unlock:
2446         mutex_unlock(&local->chanctx_mtx);
2447 }
2448
2449 void ieee80211_recalc_min_chandef(struct ieee80211_sub_if_data *sdata)
2450 {
2451         struct ieee80211_local *local = sdata->local;
2452         struct ieee80211_chanctx_conf *chanctx_conf;
2453         struct ieee80211_chanctx *chanctx;
2454
2455         mutex_lock(&local->chanctx_mtx);
2456
2457         chanctx_conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
2458                                         lockdep_is_held(&local->chanctx_mtx));
2459
2460         if (WARN_ON_ONCE(!chanctx_conf))
2461                 goto unlock;
2462
2463         chanctx = container_of(chanctx_conf, struct ieee80211_chanctx, conf);
2464         ieee80211_recalc_chanctx_min_def(local, chanctx);
2465  unlock:
2466         mutex_unlock(&local->chanctx_mtx);
2467 }
2468
2469 size_t ieee80211_ie_split_vendor(const u8 *ies, size_t ielen, size_t offset)
2470 {
2471         size_t pos = offset;
2472
2473         while (pos < ielen && ies[pos] != WLAN_EID_VENDOR_SPECIFIC)
2474                 pos += 2 + ies[pos + 1];
2475
2476         return pos;
2477 }
2478
2479 static void _ieee80211_enable_rssi_reports(struct ieee80211_sub_if_data *sdata,
2480                                             int rssi_min_thold,
2481                                             int rssi_max_thold)
2482 {
2483         trace_api_enable_rssi_reports(sdata, rssi_min_thold, rssi_max_thold);
2484
2485         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
2486                 return;
2487
2488         /*
2489          * Scale up threshold values before storing it, as the RSSI averaging
2490          * algorithm uses a scaled up value as well. Change this scaling
2491          * factor if the RSSI averaging algorithm changes.
2492          */
2493         sdata->u.mgd.rssi_min_thold = rssi_min_thold*16;
2494         sdata->u.mgd.rssi_max_thold = rssi_max_thold*16;
2495 }
2496
2497 void ieee80211_enable_rssi_reports(struct ieee80211_vif *vif,
2498                                     int rssi_min_thold,
2499                                     int rssi_max_thold)
2500 {
2501         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2502
2503         WARN_ON(rssi_min_thold == rssi_max_thold ||
2504                 rssi_min_thold > rssi_max_thold);
2505
2506         _ieee80211_enable_rssi_reports(sdata, rssi_min_thold,
2507                                        rssi_max_thold);
2508 }
2509 EXPORT_SYMBOL(ieee80211_enable_rssi_reports);
2510
2511 void ieee80211_disable_rssi_reports(struct ieee80211_vif *vif)
2512 {
2513         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2514
2515         _ieee80211_enable_rssi_reports(sdata, 0, 0);
2516 }
2517 EXPORT_SYMBOL(ieee80211_disable_rssi_reports);
2518
2519 u8 *ieee80211_ie_build_ht_cap(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
2520                               u16 cap)
2521 {
2522         __le16 tmp;
2523
2524         *pos++ = WLAN_EID_HT_CAPABILITY;
2525         *pos++ = sizeof(struct ieee80211_ht_cap);
2526         memset(pos, 0, sizeof(struct ieee80211_ht_cap));
2527
2528         /* capability flags */
2529         tmp = cpu_to_le16(cap);
2530         memcpy(pos, &tmp, sizeof(u16));
2531         pos += sizeof(u16);
2532
2533         /* AMPDU parameters */
2534         *pos++ = ht_cap->ampdu_factor |
2535                  (ht_cap->ampdu_density <<
2536                         IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT);
2537
2538         /* MCS set */
2539         memcpy(pos, &ht_cap->mcs, sizeof(ht_cap->mcs));
2540         pos += sizeof(ht_cap->mcs);
2541
2542         /* extended capabilities */
2543         pos += sizeof(__le16);
2544
2545         /* BF capabilities */
2546         pos += sizeof(__le32);
2547
2548         /* antenna selection */
2549         pos += sizeof(u8);
2550
2551         return pos;
2552 }
2553
2554 u8 *ieee80211_ie_build_vht_cap(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
2555                                u32 cap)
2556 {
2557         __le32 tmp;
2558
2559         *pos++ = WLAN_EID_VHT_CAPABILITY;
2560         *pos++ = sizeof(struct ieee80211_vht_cap);
2561         memset(pos, 0, sizeof(struct ieee80211_vht_cap));
2562
2563         /* capability flags */
2564         tmp = cpu_to_le32(cap);
2565         memcpy(pos, &tmp, sizeof(u32));
2566         pos += sizeof(u32);
2567
2568         /* VHT MCS set */
2569         memcpy(pos, &vht_cap->vht_mcs, sizeof(vht_cap->vht_mcs));
2570         pos += sizeof(vht_cap->vht_mcs);
2571
2572         return pos;
2573 }
2574
2575 u8 *ieee80211_ie_build_he_cap(u8 *pos,
2576                               const struct ieee80211_sta_he_cap *he_cap,
2577                               u8 *end)
2578 {
2579         u8 n;
2580         u8 ie_len;
2581         u8 *orig_pos = pos;
2582
2583         /* Make sure we have place for the IE */
2584         /*
2585          * TODO: the 1 added is because this temporarily is under the EXTENSION
2586          * IE. Get rid of it when it moves.
2587          */
2588         if (!he_cap)
2589                 return orig_pos;
2590
2591         n = ieee80211_he_mcs_nss_size(&he_cap->he_cap_elem);
2592         ie_len = 2 + 1 +
2593                  sizeof(he_cap->he_cap_elem) + n +
2594                  ieee80211_he_ppe_size(he_cap->ppe_thres[0],
2595                                        he_cap->he_cap_elem.phy_cap_info);
2596
2597         if ((end - pos) < ie_len)
2598                 return orig_pos;
2599
2600         *pos++ = WLAN_EID_EXTENSION;
2601         pos++; /* We'll set the size later below */
2602         *pos++ = WLAN_EID_EXT_HE_CAPABILITY;
2603
2604         /* Fixed data */
2605         memcpy(pos, &he_cap->he_cap_elem, sizeof(he_cap->he_cap_elem));
2606         pos += sizeof(he_cap->he_cap_elem);
2607
2608         memcpy(pos, &he_cap->he_mcs_nss_supp, n);
2609         pos += n;
2610
2611         /* Check if PPE Threshold should be present */
2612         if ((he_cap->he_cap_elem.phy_cap_info[6] &
2613              IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT) == 0)
2614                 goto end;
2615
2616         /*
2617          * Calculate how many PPET16/PPET8 pairs are to come. Algorithm:
2618          * (NSS_M1 + 1) x (num of 1 bits in RU_INDEX_BITMASK)
2619          */
2620         n = hweight8(he_cap->ppe_thres[0] &
2621                      IEEE80211_PPE_THRES_RU_INDEX_BITMASK_MASK);
2622         n *= (1 + ((he_cap->ppe_thres[0] & IEEE80211_PPE_THRES_NSS_MASK) >>
2623                    IEEE80211_PPE_THRES_NSS_POS));
2624
2625         /*
2626          * Each pair is 6 bits, and we need to add the 7 "header" bits to the
2627          * total size.
2628          */
2629         n = (n * IEEE80211_PPE_THRES_INFO_PPET_SIZE * 2) + 7;
2630         n = DIV_ROUND_UP(n, 8);
2631
2632         /* Copy PPE Thresholds */
2633         memcpy(pos, &he_cap->ppe_thres, n);
2634         pos += n;
2635
2636 end:
2637         orig_pos[1] = (pos - orig_pos) - 2;
2638         return pos;
2639 }
2640
2641 u8 *ieee80211_ie_build_ht_oper(u8 *pos, struct ieee80211_sta_ht_cap *ht_cap,
2642                                const struct cfg80211_chan_def *chandef,
2643                                u16 prot_mode, bool rifs_mode)
2644 {
2645         struct ieee80211_ht_operation *ht_oper;
2646         /* Build HT Information */
2647         *pos++ = WLAN_EID_HT_OPERATION;
2648         *pos++ = sizeof(struct ieee80211_ht_operation);
2649         ht_oper = (struct ieee80211_ht_operation *)pos;
2650         ht_oper->primary_chan = ieee80211_frequency_to_channel(
2651                                         chandef->chan->center_freq);
2652         switch (chandef->width) {
2653         case NL80211_CHAN_WIDTH_160:
2654         case NL80211_CHAN_WIDTH_80P80:
2655         case NL80211_CHAN_WIDTH_80:
2656         case NL80211_CHAN_WIDTH_40:
2657                 if (chandef->center_freq1 > chandef->chan->center_freq)
2658                         ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
2659                 else
2660                         ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
2661                 break;
2662         default:
2663                 ht_oper->ht_param = IEEE80211_HT_PARAM_CHA_SEC_NONE;
2664                 break;
2665         }
2666         if (ht_cap->cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 &&
2667             chandef->width != NL80211_CHAN_WIDTH_20_NOHT &&
2668             chandef->width != NL80211_CHAN_WIDTH_20)
2669                 ht_oper->ht_param |= IEEE80211_HT_PARAM_CHAN_WIDTH_ANY;
2670
2671         if (rifs_mode)
2672                 ht_oper->ht_param |= IEEE80211_HT_PARAM_RIFS_MODE;
2673
2674         ht_oper->operation_mode = cpu_to_le16(prot_mode);
2675         ht_oper->stbc_param = 0x0000;
2676
2677         /* It seems that Basic MCS set and Supported MCS set
2678            are identical for the first 10 bytes */
2679         memset(&ht_oper->basic_set, 0, 16);
2680         memcpy(&ht_oper->basic_set, &ht_cap->mcs, 10);
2681
2682         return pos + sizeof(struct ieee80211_ht_operation);
2683 }
2684
2685 void ieee80211_ie_build_wide_bw_cs(u8 *pos,
2686                                    const struct cfg80211_chan_def *chandef)
2687 {
2688         *pos++ = WLAN_EID_WIDE_BW_CHANNEL_SWITCH;       /* EID */
2689         *pos++ = 3;                                     /* IE length */
2690         /* New channel width */
2691         switch (chandef->width) {
2692         case NL80211_CHAN_WIDTH_80:
2693                 *pos++ = IEEE80211_VHT_CHANWIDTH_80MHZ;
2694                 break;
2695         case NL80211_CHAN_WIDTH_160:
2696                 *pos++ = IEEE80211_VHT_CHANWIDTH_160MHZ;
2697                 break;
2698         case NL80211_CHAN_WIDTH_80P80:
2699                 *pos++ = IEEE80211_VHT_CHANWIDTH_80P80MHZ;
2700                 break;
2701         default:
2702                 *pos++ = IEEE80211_VHT_CHANWIDTH_USE_HT;
2703         }
2704
2705         /* new center frequency segment 0 */
2706         *pos++ = ieee80211_frequency_to_channel(chandef->center_freq1);
2707         /* new center frequency segment 1 */
2708         if (chandef->center_freq2)
2709                 *pos++ = ieee80211_frequency_to_channel(chandef->center_freq2);
2710         else
2711                 *pos++ = 0;
2712 }
2713
2714 u8 *ieee80211_ie_build_vht_oper(u8 *pos, struct ieee80211_sta_vht_cap *vht_cap,
2715                                 const struct cfg80211_chan_def *chandef)
2716 {
2717         struct ieee80211_vht_operation *vht_oper;
2718
2719         *pos++ = WLAN_EID_VHT_OPERATION;
2720         *pos++ = sizeof(struct ieee80211_vht_operation);
2721         vht_oper = (struct ieee80211_vht_operation *)pos;
2722         vht_oper->center_freq_seg0_idx = ieee80211_frequency_to_channel(
2723                                                         chandef->center_freq1);
2724         if (chandef->center_freq2)
2725                 vht_oper->center_freq_seg1_idx =
2726                         ieee80211_frequency_to_channel(chandef->center_freq2);
2727         else
2728                 vht_oper->center_freq_seg1_idx = 0x00;
2729
2730         switch (chandef->width) {
2731         case NL80211_CHAN_WIDTH_160:
2732                 /*
2733                  * Convert 160 MHz channel width to new style as interop
2734                  * workaround.
2735                  */
2736                 vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2737                 vht_oper->center_freq_seg1_idx = vht_oper->center_freq_seg0_idx;
2738                 if (chandef->chan->center_freq < chandef->center_freq1)
2739                         vht_oper->center_freq_seg0_idx -= 8;
2740                 else
2741                         vht_oper->center_freq_seg0_idx += 8;
2742                 break;
2743         case NL80211_CHAN_WIDTH_80P80:
2744                 /*
2745                  * Convert 80+80 MHz channel width to new style as interop
2746                  * workaround.
2747                  */
2748                 vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2749                 break;
2750         case NL80211_CHAN_WIDTH_80:
2751                 vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_80MHZ;
2752                 break;
2753         default:
2754                 vht_oper->chan_width = IEEE80211_VHT_CHANWIDTH_USE_HT;
2755                 break;
2756         }
2757
2758         /* don't require special VHT peer rates */
2759         vht_oper->basic_mcs_set = cpu_to_le16(0xffff);
2760
2761         return pos + sizeof(struct ieee80211_vht_operation);
2762 }
2763
2764 bool ieee80211_chandef_ht_oper(const struct ieee80211_ht_operation *ht_oper,
2765                                struct cfg80211_chan_def *chandef)
2766 {
2767         enum nl80211_channel_type channel_type;
2768
2769         if (!ht_oper)
2770                 return false;
2771
2772         switch (ht_oper->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
2773         case IEEE80211_HT_PARAM_CHA_SEC_NONE:
2774                 channel_type = NL80211_CHAN_HT20;
2775                 break;
2776         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
2777                 channel_type = NL80211_CHAN_HT40PLUS;
2778                 break;
2779         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
2780                 channel_type = NL80211_CHAN_HT40MINUS;
2781                 break;
2782         default:
2783                 channel_type = NL80211_CHAN_NO_HT;
2784                 return false;
2785         }
2786
2787         cfg80211_chandef_create(chandef, chandef->chan, channel_type);
2788         return true;
2789 }
2790
2791 bool ieee80211_chandef_vht_oper(struct ieee80211_hw *hw,
2792                                 const struct ieee80211_vht_operation *oper,
2793                                 const struct ieee80211_ht_operation *htop,
2794                                 struct cfg80211_chan_def *chandef)
2795 {
2796         struct cfg80211_chan_def new = *chandef;
2797         int cf0, cf1;
2798         int ccfs0, ccfs1, ccfs2;
2799         int ccf0, ccf1;
2800
2801         if (!oper || !htop)
2802                 return false;
2803
2804         ccfs0 = oper->center_freq_seg0_idx;
2805         ccfs1 = oper->center_freq_seg1_idx;
2806         ccfs2 = (le16_to_cpu(htop->operation_mode) &
2807                                 IEEE80211_HT_OP_MODE_CCFS2_MASK)
2808                         >> IEEE80211_HT_OP_MODE_CCFS2_SHIFT;
2809
2810         /* when parsing (and we know how to) CCFS1 and CCFS2 are equivalent */
2811         ccf0 = ccfs0;
2812         ccf1 = ccfs1;
2813         if (!ccfs1 && ieee80211_hw_check(hw, SUPPORTS_VHT_EXT_NSS_BW))
2814                 ccf1 = ccfs2;
2815
2816         cf0 = ieee80211_channel_to_frequency(ccf0, chandef->chan->band);
2817         cf1 = ieee80211_channel_to_frequency(ccf1, chandef->chan->band);
2818
2819         switch (oper->chan_width) {
2820         case IEEE80211_VHT_CHANWIDTH_USE_HT:
2821                 /* just use HT information directly */
2822                 break;
2823         case IEEE80211_VHT_CHANWIDTH_80MHZ:
2824                 new.width = NL80211_CHAN_WIDTH_80;
2825                 new.center_freq1 = cf0;
2826                 /* If needed, adjust based on the newer interop workaround. */
2827                 if (ccf1) {
2828                         unsigned int diff;
2829
2830                         diff = abs(ccf1 - ccf0);
2831                         if (diff == 8) {
2832                                 new.width = NL80211_CHAN_WIDTH_160;
2833                                 new.center_freq1 = cf1;
2834                         } else if (diff > 8) {
2835                                 new.width = NL80211_CHAN_WIDTH_80P80;
2836                                 new.center_freq2 = cf1;
2837                         }
2838                 }
2839                 break;
2840         case IEEE80211_VHT_CHANWIDTH_160MHZ:
2841                 /* deprecated encoding */
2842                 new.width = NL80211_CHAN_WIDTH_160;
2843                 new.center_freq1 = cf0;
2844                 break;
2845         case IEEE80211_VHT_CHANWIDTH_80P80MHZ:
2846                 /* deprecated encoding */
2847                 new.width = NL80211_CHAN_WIDTH_80P80;
2848                 new.center_freq1 = cf0;
2849                 new.center_freq2 = cf1;
2850                 break;
2851         default:
2852                 return false;
2853         }
2854
2855         if (!cfg80211_chandef_valid(&new))
2856                 return false;
2857
2858         *chandef = new;
2859         return true;
2860 }
2861
2862 int ieee80211_parse_bitrates(struct cfg80211_chan_def *chandef,
2863                              const struct ieee80211_supported_band *sband,
2864                              const u8 *srates, int srates_len, u32 *rates)
2865 {
2866         u32 rate_flags = ieee80211_chandef_rate_flags(chandef);
2867         int shift = ieee80211_chandef_get_shift(chandef);
2868         struct ieee80211_rate *br;
2869         int brate, rate, i, j, count = 0;
2870
2871         *rates = 0;
2872
2873         for (i = 0; i < srates_len; i++) {
2874                 rate = srates[i] & 0x7f;
2875
2876                 for (j = 0; j < sband->n_bitrates; j++) {
2877                         br = &sband->bitrates[j];
2878                         if ((rate_flags & br->flags) != rate_flags)
2879                                 continue;
2880
2881                         brate = DIV_ROUND_UP(br->bitrate, (1 << shift) * 5);
2882                         if (brate == rate) {
2883                                 *rates |= BIT(j);
2884                                 count++;
2885                                 break;
2886                         }
2887                 }
2888         }
2889         return count;
2890 }
2891
2892 int ieee80211_add_srates_ie(struct ieee80211_sub_if_data *sdata,
2893                             struct sk_buff *skb, bool need_basic,
2894                             enum nl80211_band band)
2895 {
2896         struct ieee80211_local *local = sdata->local;
2897         struct ieee80211_supported_band *sband;
2898         int rate, shift;
2899         u8 i, rates, *pos;
2900         u32 basic_rates = sdata->vif.bss_conf.basic_rates;
2901         u32 rate_flags;
2902
2903         shift = ieee80211_vif_get_shift(&sdata->vif);
2904         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
2905         sband = local->hw.wiphy->bands[band];
2906         rates = 0;
2907         for (i = 0; i < sband->n_bitrates; i++) {
2908                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2909                         continue;
2910                 rates++;
2911         }
2912         if (rates > 8)
2913                 rates = 8;
2914
2915         if (skb_tailroom(skb) < rates + 2)
2916                 return -ENOMEM;
2917
2918         pos = skb_put(skb, rates + 2);
2919         *pos++ = WLAN_EID_SUPP_RATES;
2920         *pos++ = rates;
2921         for (i = 0; i < rates; i++) {
2922                 u8 basic = 0;
2923                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2924                         continue;
2925
2926                 if (need_basic && basic_rates & BIT(i))
2927                         basic = 0x80;
2928                 rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
2929                                     5 * (1 << shift));
2930                 *pos++ = basic | (u8) rate;
2931         }
2932
2933         return 0;
2934 }
2935
2936 int ieee80211_add_ext_srates_ie(struct ieee80211_sub_if_data *sdata,
2937                                 struct sk_buff *skb, bool need_basic,
2938                                 enum nl80211_band band)
2939 {
2940         struct ieee80211_local *local = sdata->local;
2941         struct ieee80211_supported_band *sband;
2942         int rate, shift;
2943         u8 i, exrates, *pos;
2944         u32 basic_rates = sdata->vif.bss_conf.basic_rates;
2945         u32 rate_flags;
2946
2947         rate_flags = ieee80211_chandef_rate_flags(&sdata->vif.bss_conf.chandef);
2948         shift = ieee80211_vif_get_shift(&sdata->vif);
2949
2950         sband = local->hw.wiphy->bands[band];
2951         exrates = 0;
2952         for (i = 0; i < sband->n_bitrates; i++) {
2953                 if ((rate_flags & sband->bitrates[i].flags) != rate_flags)
2954                         continue;
2955                 exrates++;
2956         }
2957
2958         if (exrates > 8)
2959                 exrates -= 8;
2960         else
2961                 exrates = 0;
2962
2963         if (skb_tailroom(skb) < exrates + 2)
2964                 return -ENOMEM;
2965
2966         if (exrates) {
2967                 pos = skb_put(skb, exrates + 2);
2968                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
2969                 *pos++ = exrates;
2970                 for (i = 8; i < sband->n_bitrates; i++) {
2971                         u8 basic = 0;
2972                         if ((rate_flags & sband->bitrates[i].flags)
2973                             != rate_flags)
2974                                 continue;
2975                         if (need_basic && basic_rates & BIT(i))
2976                                 basic = 0x80;
2977                         rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
2978                                             5 * (1 << shift));
2979                         *pos++ = basic | (u8) rate;
2980                 }
2981         }
2982         return 0;
2983 }
2984
2985 int ieee80211_ave_rssi(struct ieee80211_vif *vif)
2986 {
2987         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2988         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2989
2990         if (WARN_ON_ONCE(sdata->vif.type != NL80211_IFTYPE_STATION)) {
2991                 /* non-managed type inferfaces */
2992                 return 0;
2993         }
2994         return -ewma_beacon_signal_read(&ifmgd->ave_beacon_signal);
2995 }
2996 EXPORT_SYMBOL_GPL(ieee80211_ave_rssi);
2997
2998 u8 ieee80211_mcs_to_chains(const struct ieee80211_mcs_info *mcs)
2999 {
3000         if (!mcs)
3001                 return 1;
3002
3003         /* TODO: consider rx_highest */
3004
3005         if (mcs->rx_mask[3])
3006                 return 4;
3007         if (mcs->rx_mask[2])
3008                 return 3;
3009         if (mcs->rx_mask[1])
3010                 return 2;
3011         return 1;
3012 }
3013
3014 /**
3015  * ieee80211_calculate_rx_timestamp - calculate timestamp in frame
3016  * @local: mac80211 hw info struct
3017  * @status: RX status
3018  * @mpdu_len: total MPDU length (including FCS)
3019  * @mpdu_offset: offset into MPDU to calculate timestamp at
3020  *
3021  * This function calculates the RX timestamp at the given MPDU offset, taking
3022  * into account what the RX timestamp was. An offset of 0 will just normalize
3023  * the timestamp to TSF at beginning of MPDU reception.
3024  */
3025 u64 ieee80211_calculate_rx_timestamp(struct ieee80211_local *local,
3026                                      struct ieee80211_rx_status *status,
3027                                      unsigned int mpdu_len,
3028                                      unsigned int mpdu_offset)
3029 {
3030         u64 ts = status->mactime;
3031         struct rate_info ri;
3032         u16 rate;
3033
3034         if (WARN_ON(!ieee80211_have_rx_timestamp(status)))
3035                 return 0;
3036
3037         memset(&ri, 0, sizeof(ri));
3038
3039         ri.bw = status->bw;
3040
3041         /* Fill cfg80211 rate info */
3042         switch (status->encoding) {
3043         case RX_ENC_HT:
3044                 ri.mcs = status->rate_idx;
3045                 ri.flags |= RATE_INFO_FLAGS_MCS;
3046                 if (status->enc_flags & RX_ENC_FLAG_SHORT_GI)
3047                         ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
3048                 break;
3049         case RX_ENC_VHT:
3050                 ri.flags |= RATE_INFO_FLAGS_VHT_MCS;
3051                 ri.mcs = status->rate_idx;
3052                 ri.nss = status->nss;
3053                 if (status->enc_flags & RX_ENC_FLAG_SHORT_GI)
3054                         ri.flags |= RATE_INFO_FLAGS_SHORT_GI;
3055                 break;
3056         default:
3057                 WARN_ON(1);
3058                 /* fall through */
3059         case RX_ENC_LEGACY: {
3060                 struct ieee80211_supported_band *sband;
3061                 int shift = 0;
3062                 int bitrate;
3063
3064                 switch (status->bw) {
3065                 case RATE_INFO_BW_10:
3066                         shift = 1;
3067                         break;
3068                 case RATE_INFO_BW_5:
3069                         shift = 2;
3070                         break;
3071                 }
3072
3073                 sband = local->hw.wiphy->bands[status->band];
3074                 bitrate = sband->bitrates[status->rate_idx].bitrate;
3075                 ri.legacy = DIV_ROUND_UP(bitrate, (1 << shift));
3076
3077                 if (status->flag & RX_FLAG_MACTIME_PLCP_START) {
3078                         /* TODO: handle HT/VHT preambles */
3079                         if (status->band == NL80211_BAND_5GHZ) {
3080                                 ts += 20 << shift;
3081                                 mpdu_offset += 2;
3082                         } else if (status->enc_flags & RX_ENC_FLAG_SHORTPRE) {
3083                                 ts += 96;
3084                         } else {
3085                                 ts += 192;
3086                         }
3087                 }
3088                 break;
3089                 }
3090         }
3091
3092         rate = cfg80211_calculate_bitrate(&ri);
3093         if (WARN_ONCE(!rate,
3094                       "Invalid bitrate: flags=0x%llx, idx=%d, vht_nss=%d\n",
3095                       (unsigned long long)status->flag, status->rate_idx,
3096                       status->nss))
3097                 return 0;
3098
3099         /* rewind from end of MPDU */
3100         if (status->flag & RX_FLAG_MACTIME_END)
3101                 ts -= mpdu_len * 8 * 10 / rate;
3102
3103         ts += mpdu_offset * 8 * 10 / rate;
3104
3105         return ts;
3106 }
3107
3108 void ieee80211_dfs_cac_cancel(struct ieee80211_local *local)
3109 {
3110         struct ieee80211_sub_if_data *sdata;
3111         struct cfg80211_chan_def chandef;
3112
3113         /* for interface list, to avoid linking iflist_mtx and chanctx_mtx */
3114         ASSERT_RTNL();
3115
3116         mutex_lock(&local->mtx);
3117         list_for_each_entry(sdata, &local->interfaces, list) {
3118                 /* it might be waiting for the local->mtx, but then
3119                  * by the time it gets it, sdata->wdev.cac_started
3120                  * will no longer be true
3121                  */
3122                 cancel_delayed_work(&sdata->dfs_cac_timer_work);
3123
3124                 if (sdata->wdev.cac_started) {
3125                         chandef = sdata->vif.bss_conf.chandef;
3126                         ieee80211_vif_release_channel(sdata);
3127                         cfg80211_cac_event(sdata->dev,
3128                                            &chandef,
3129                                            NL80211_RADAR_CAC_ABORTED,
3130                                            GFP_KERNEL);
3131                 }
3132         }
3133         mutex_unlock(&local->mtx);
3134 }
3135
3136 void ieee80211_dfs_radar_detected_work(struct work_struct *work)
3137 {
3138         struct ieee80211_local *local =
3139                 container_of(work, struct ieee80211_local, radar_detected_work);
3140         struct cfg80211_chan_def chandef = local->hw.conf.chandef;
3141         struct ieee80211_chanctx *ctx;
3142         int num_chanctx = 0;
3143
3144         mutex_lock(&local->chanctx_mtx);
3145         list_for_each_entry(ctx, &local->chanctx_list, list) {
3146                 if (ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER)
3147                         continue;
3148
3149                 num_chanctx++;
3150                 chandef = ctx->conf.def;
3151         }
3152         mutex_unlock(&local->chanctx_mtx);
3153
3154         rtnl_lock();
3155         ieee80211_dfs_cac_cancel(local);
3156         rtnl_unlock();
3157
3158         if (num_chanctx > 1)
3159                 /* XXX: multi-channel is not supported yet */
3160                 WARN_ON(1);
3161         else
3162                 cfg80211_radar_event(local->hw.wiphy, &chandef, GFP_KERNEL);
3163 }
3164
3165 void ieee80211_radar_detected(struct ieee80211_hw *hw)
3166 {
3167         struct ieee80211_local *local = hw_to_local(hw);
3168
3169         trace_api_radar_detected(local);
3170
3171         schedule_work(&local->radar_detected_work);
3172 }
3173 EXPORT_SYMBOL(ieee80211_radar_detected);
3174
3175 u32 ieee80211_chandef_downgrade(struct cfg80211_chan_def *c)
3176 {
3177         u32 ret;
3178         int tmp;
3179
3180         switch (c->width) {
3181         case NL80211_CHAN_WIDTH_20:
3182                 c->width = NL80211_CHAN_WIDTH_20_NOHT;
3183                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3184                 break;
3185         case NL80211_CHAN_WIDTH_40:
3186                 c->width = NL80211_CHAN_WIDTH_20;
3187                 c->center_freq1 = c->chan->center_freq;
3188                 ret = IEEE80211_STA_DISABLE_40MHZ |
3189                       IEEE80211_STA_DISABLE_VHT;
3190                 break;
3191         case NL80211_CHAN_WIDTH_80:
3192                 tmp = (30 + c->chan->center_freq - c->center_freq1)/20;
3193                 /* n_P40 */
3194                 tmp /= 2;
3195                 /* freq_P40 */
3196                 c->center_freq1 = c->center_freq1 - 20 + 40 * tmp;
3197                 c->width = NL80211_CHAN_WIDTH_40;
3198                 ret = IEEE80211_STA_DISABLE_VHT;
3199                 break;
3200         case NL80211_CHAN_WIDTH_80P80:
3201                 c->center_freq2 = 0;
3202                 c->width = NL80211_CHAN_WIDTH_80;
3203                 ret = IEEE80211_STA_DISABLE_80P80MHZ |
3204                       IEEE80211_STA_DISABLE_160MHZ;
3205                 break;
3206         case NL80211_CHAN_WIDTH_160:
3207                 /* n_P20 */
3208                 tmp = (70 + c->chan->center_freq - c->center_freq1)/20;
3209                 /* n_P80 */
3210                 tmp /= 4;
3211                 c->center_freq1 = c->center_freq1 - 40 + 80 * tmp;
3212                 c->width = NL80211_CHAN_WIDTH_80;
3213                 ret = IEEE80211_STA_DISABLE_80P80MHZ |
3214                       IEEE80211_STA_DISABLE_160MHZ;
3215                 break;
3216         default:
3217         case NL80211_CHAN_WIDTH_20_NOHT:
3218                 WARN_ON_ONCE(1);
3219                 c->width = NL80211_CHAN_WIDTH_20_NOHT;
3220                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3221                 break;
3222         case NL80211_CHAN_WIDTH_5:
3223         case NL80211_CHAN_WIDTH_10:
3224                 WARN_ON_ONCE(1);
3225                 /* keep c->width */
3226                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
3227                 break;
3228         }
3229
3230         WARN_ON_ONCE(!cfg80211_chandef_valid(c));
3231
3232         return ret;
3233 }
3234
3235 /*
3236  * Returns true if smps_mode_new is strictly more restrictive than
3237  * smps_mode_old.
3238  */
3239 bool ieee80211_smps_is_restrictive(enum ieee80211_smps_mode smps_mode_old,
3240                                    enum ieee80211_smps_mode smps_mode_new)
3241 {
3242         if (WARN_ON_ONCE(smps_mode_old == IEEE80211_SMPS_AUTOMATIC ||
3243                          smps_mode_new == IEEE80211_SMPS_AUTOMATIC))
3244                 return false;
3245
3246         switch (smps_mode_old) {
3247         case IEEE80211_SMPS_STATIC:
3248                 return false;
3249         case IEEE80211_SMPS_DYNAMIC:
3250                 return smps_mode_new == IEEE80211_SMPS_STATIC;
3251         case IEEE80211_SMPS_OFF:
3252                 return smps_mode_new != IEEE80211_SMPS_OFF;
3253         default:
3254                 WARN_ON(1);
3255         }
3256
3257         return false;
3258 }
3259
3260 int ieee80211_send_action_csa(struct ieee80211_sub_if_data *sdata,
3261                               struct cfg80211_csa_settings *csa_settings)
3262 {
3263         struct sk_buff *skb;
3264         struct ieee80211_mgmt *mgmt;
3265         struct ieee80211_local *local = sdata->local;
3266         int freq;
3267         int hdr_len = offsetofend(struct ieee80211_mgmt,
3268                                   u.action.u.chan_switch);
3269         u8 *pos;
3270
3271         if (sdata->vif.type != NL80211_IFTYPE_ADHOC &&
3272             sdata->vif.type != NL80211_IFTYPE_MESH_POINT)
3273                 return -EOPNOTSUPP;
3274
3275         skb = dev_alloc_skb(local->tx_headroom + hdr_len +
3276                             5 + /* channel switch announcement element */
3277                             3 + /* secondary channel offset element */
3278                             5 + /* wide bandwidth channel switch announcement */
3279                             8); /* mesh channel switch parameters element */
3280         if (!skb)
3281                 return -ENOMEM;
3282
3283         skb_reserve(skb, local->tx_headroom);
3284         mgmt = skb_put_zero(skb, hdr_len);
3285         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
3286                                           IEEE80211_STYPE_ACTION);
3287
3288         eth_broadcast_addr(mgmt->da);
3289         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
3290         if (ieee80211_vif_is_mesh(&sdata->vif)) {
3291                 memcpy(mgmt->bssid, sdata->vif.addr, ETH_ALEN);
3292         } else {
3293                 struct ieee80211_if_ibss *ifibss = &sdata->u.ibss;
3294                 memcpy(mgmt->bssid, ifibss->bssid, ETH_ALEN);
3295         }
3296         mgmt->u.action.category = WLAN_CATEGORY_SPECTRUM_MGMT;
3297         mgmt->u.action.u.chan_switch.action_code = WLAN_ACTION_SPCT_CHL_SWITCH;
3298         pos = skb_put(skb, 5);
3299         *pos++ = WLAN_EID_CHANNEL_SWITCH;                       /* EID */
3300         *pos++ = 3;                                             /* IE length */
3301         *pos++ = csa_settings->block_tx ? 1 : 0;                /* CSA mode */
3302         freq = csa_settings->chandef.chan->center_freq;
3303         *pos++ = ieee80211_frequency_to_channel(freq);          /* channel */
3304         *pos++ = csa_settings->count;                           /* count */
3305
3306         if (csa_settings->chandef.width == NL80211_CHAN_WIDTH_40) {
3307                 enum nl80211_channel_type ch_type;
3308
3309                 skb_put(skb, 3);
3310                 *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;     /* EID */
3311                 *pos++ = 1;                                     /* IE length */
3312                 ch_type = cfg80211_get_chandef_type(&csa_settings->chandef);
3313                 if (ch_type == NL80211_CHAN_HT40PLUS)
3314                         *pos++ = IEEE80211_HT_PARAM_CHA_SEC_ABOVE;
3315                 else
3316                         *pos++ = IEEE80211_HT_PARAM_CHA_SEC_BELOW;
3317         }
3318
3319         if (ieee80211_vif_is_mesh(&sdata->vif)) {
3320                 struct ieee80211_if_mesh *ifmsh = &sdata->u.mesh;
3321
3322                 skb_put(skb, 8);
3323                 *pos++ = WLAN_EID_CHAN_SWITCH_PARAM;            /* EID */
3324                 *pos++ = 6;                                     /* IE length */
3325                 *pos++ = sdata->u.mesh.mshcfg.dot11MeshTTL;     /* Mesh TTL */
3326                 *pos = 0x00;    /* Mesh Flag: Tx Restrict, Initiator, Reason */
3327                 *pos |= WLAN_EID_CHAN_SWITCH_PARAM_INITIATOR;
3328                 *pos++ |= csa_settings->block_tx ?
3329                           WLAN_EID_CHAN_SWITCH_PARAM_TX_RESTRICT : 0x00;
3330                 put_unaligned_le16(WLAN_REASON_MESH_CHAN, pos); /* Reason Cd */
3331                 pos += 2;
3332                 put_unaligned_le16(ifmsh->pre_value, pos);/* Precedence Value */
3333                 pos += 2;
3334         }
3335
3336         if (csa_settings->chandef.width == NL80211_CHAN_WIDTH_80 ||
3337             csa_settings->chandef.width == NL80211_CHAN_WIDTH_80P80 ||
3338             csa_settings->chandef.width == NL80211_CHAN_WIDTH_160) {
3339                 skb_put(skb, 5);
3340                 ieee80211_ie_build_wide_bw_cs(pos, &csa_settings->chandef);
3341         }
3342
3343         ieee80211_tx_skb(sdata, skb);
3344         return 0;
3345 }
3346
3347 bool ieee80211_cs_valid(const struct ieee80211_cipher_scheme *cs)
3348 {
3349         return !(cs == NULL || cs->cipher == 0 ||
3350                  cs->hdr_len < cs->pn_len + cs->pn_off ||
3351                  cs->hdr_len <= cs->key_idx_off ||
3352                  cs->key_idx_shift > 7 ||
3353                  cs->key_idx_mask == 0);
3354 }
3355
3356 bool ieee80211_cs_list_valid(const struct ieee80211_cipher_scheme *cs, int n)
3357 {
3358         int i;
3359
3360         /* Ensure we have enough iftype bitmap space for all iftype values */
3361         WARN_ON((NUM_NL80211_IFTYPES / 8 + 1) > sizeof(cs[0].iftype));
3362
3363         for (i = 0; i < n; i++)
3364                 if (!ieee80211_cs_valid(&cs[i]))
3365                         return false;
3366
3367         return true;
3368 }
3369
3370 const struct ieee80211_cipher_scheme *
3371 ieee80211_cs_get(struct ieee80211_local *local, u32 cipher,
3372                  enum nl80211_iftype iftype)
3373 {
3374         const struct ieee80211_cipher_scheme *l = local->hw.cipher_schemes;
3375         int n = local->hw.n_cipher_schemes;
3376         int i;
3377         const struct ieee80211_cipher_scheme *cs = NULL;
3378
3379         for (i = 0; i < n; i++) {
3380                 if (l[i].cipher == cipher) {
3381                         cs = &l[i];
3382                         break;
3383                 }
3384         }
3385
3386         if (!cs || !(cs->iftype & BIT(iftype)))
3387                 return NULL;
3388
3389         return cs;
3390 }
3391
3392 int ieee80211_cs_headroom(struct ieee80211_local *local,
3393                           struct cfg80211_crypto_settings *crypto,
3394                           enum nl80211_iftype iftype)
3395 {
3396         const struct ieee80211_cipher_scheme *cs;
3397         int headroom = IEEE80211_ENCRYPT_HEADROOM;
3398         int i;
3399
3400         for (i = 0; i < crypto->n_ciphers_pairwise; i++) {
3401                 cs = ieee80211_cs_get(local, crypto->ciphers_pairwise[i],
3402                                       iftype);
3403
3404                 if (cs && headroom < cs->hdr_len)
3405                         headroom = cs->hdr_len;
3406         }
3407
3408         cs = ieee80211_cs_get(local, crypto->cipher_group, iftype);
3409         if (cs && headroom < cs->hdr_len)
3410                 headroom = cs->hdr_len;
3411
3412         return headroom;
3413 }
3414
3415 static bool
3416 ieee80211_extend_noa_desc(struct ieee80211_noa_data *data, u32 tsf, int i)
3417 {
3418         s32 end = data->desc[i].start + data->desc[i].duration - (tsf + 1);
3419         int skip;
3420
3421         if (end > 0)
3422                 return false;
3423
3424         /* One shot NOA  */
3425         if (data->count[i] == 1)
3426                 return false;
3427
3428         if (data->desc[i].interval == 0)
3429                 return false;
3430
3431         /* End time is in the past, check for repetitions */
3432         skip = DIV_ROUND_UP(-end, data->desc[i].interval);
3433         if (data->count[i] < 255) {
3434                 if (data->count[i] <= skip) {
3435                         data->count[i] = 0;
3436                         return false;
3437                 }
3438
3439                 data->count[i] -= skip;
3440         }
3441
3442         data->desc[i].start += skip * data->desc[i].interval;
3443
3444         return true;
3445 }
3446
3447 static bool
3448 ieee80211_extend_absent_time(struct ieee80211_noa_data *data, u32 tsf,
3449                              s32 *offset)
3450 {
3451         bool ret = false;
3452         int i;
3453
3454         for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3455                 s32 cur;
3456
3457                 if (!data->count[i])
3458                         continue;
3459
3460                 if (ieee80211_extend_noa_desc(data, tsf + *offset, i))
3461                         ret = true;
3462
3463                 cur = data->desc[i].start - tsf;
3464                 if (cur > *offset)
3465                         continue;
3466
3467                 cur = data->desc[i].start + data->desc[i].duration - tsf;
3468                 if (cur > *offset)
3469                         *offset = cur;
3470         }
3471
3472         return ret;
3473 }
3474
3475 static u32
3476 ieee80211_get_noa_absent_time(struct ieee80211_noa_data *data, u32 tsf)
3477 {
3478         s32 offset = 0;
3479         int tries = 0;
3480         /*
3481          * arbitrary limit, used to avoid infinite loops when combined NoA
3482          * descriptors cover the full time period.
3483          */
3484         int max_tries = 5;
3485
3486         ieee80211_extend_absent_time(data, tsf, &offset);
3487         do {
3488                 if (!ieee80211_extend_absent_time(data, tsf, &offset))
3489                         break;
3490
3491                 tries++;
3492         } while (tries < max_tries);
3493
3494         return offset;
3495 }
3496
3497 void ieee80211_update_p2p_noa(struct ieee80211_noa_data *data, u32 tsf)
3498 {
3499         u32 next_offset = BIT(31) - 1;
3500         int i;
3501
3502         data->absent = 0;
3503         data->has_next_tsf = false;
3504         for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3505                 s32 start;
3506
3507                 if (!data->count[i])
3508                         continue;
3509
3510                 ieee80211_extend_noa_desc(data, tsf, i);
3511                 start = data->desc[i].start - tsf;
3512                 if (start <= 0)
3513                         data->absent |= BIT(i);
3514
3515                 if (next_offset > start)
3516                         next_offset = start;
3517
3518                 data->has_next_tsf = true;
3519         }
3520
3521         if (data->absent)
3522                 next_offset = ieee80211_get_noa_absent_time(data, tsf);
3523
3524         data->next_tsf = tsf + next_offset;
3525 }
3526 EXPORT_SYMBOL(ieee80211_update_p2p_noa);
3527
3528 int ieee80211_parse_p2p_noa(const struct ieee80211_p2p_noa_attr *attr,
3529                             struct ieee80211_noa_data *data, u32 tsf)
3530 {
3531         int ret = 0;
3532         int i;
3533
3534         memset(data, 0, sizeof(*data));
3535
3536         for (i = 0; i < IEEE80211_P2P_NOA_DESC_MAX; i++) {
3537                 const struct ieee80211_p2p_noa_desc *desc = &attr->desc[i];
3538
3539                 if (!desc->count || !desc->duration)
3540                         continue;
3541
3542                 data->count[i] = desc->count;
3543                 data->desc[i].start = le32_to_cpu(desc->start_time);
3544                 data->desc[i].duration = le32_to_cpu(desc->duration);
3545                 data->desc[i].interval = le32_to_cpu(desc->interval);
3546
3547                 if (data->count[i] > 1 &&
3548                     data->desc[i].interval < data->desc[i].duration)
3549                         continue;
3550
3551                 ieee80211_extend_noa_desc(data, tsf, i);
3552                 ret++;
3553         }
3554
3555         if (ret)
3556                 ieee80211_update_p2p_noa(data, tsf);
3557
3558         return ret;
3559 }
3560 EXPORT_SYMBOL(ieee80211_parse_p2p_noa);
3561
3562 void ieee80211_recalc_dtim(struct ieee80211_local *local,
3563                            struct ieee80211_sub_if_data *sdata)
3564 {
3565         u64 tsf = drv_get_tsf(local, sdata);
3566         u64 dtim_count = 0;
3567         u16 beacon_int = sdata->vif.bss_conf.beacon_int * 1024;
3568         u8 dtim_period = sdata->vif.bss_conf.dtim_period;
3569         struct ps_data *ps;
3570         u8 bcns_from_dtim;
3571
3572         if (tsf == -1ULL || !beacon_int || !dtim_period)
3573                 return;
3574
3575         if (sdata->vif.type == NL80211_IFTYPE_AP ||
3576             sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
3577                 if (!sdata->bss)
3578                         return;
3579
3580                 ps = &sdata->bss->ps;
3581         } else if (ieee80211_vif_is_mesh(&sdata->vif)) {
3582                 ps = &sdata->u.mesh.ps;
3583         } else {
3584                 return;
3585         }
3586
3587         /*
3588          * actually finds last dtim_count, mac80211 will update in
3589          * __beacon_add_tim().
3590          * dtim_count = dtim_period - (tsf / bcn_int) % dtim_period
3591          */
3592         do_div(tsf, beacon_int);
3593         bcns_from_dtim = do_div(tsf, dtim_period);
3594         /* just had a DTIM */
3595         if (!bcns_from_dtim)
3596                 dtim_count = 0;
3597         else
3598                 dtim_count = dtim_period - bcns_from_dtim;
3599
3600         ps->dtim_count = dtim_count;
3601 }
3602
3603 static u8 ieee80211_chanctx_radar_detect(struct ieee80211_local *local,
3604                                          struct ieee80211_chanctx *ctx)
3605 {
3606         struct ieee80211_sub_if_data *sdata;
3607         u8 radar_detect = 0;
3608
3609         lockdep_assert_held(&local->chanctx_mtx);
3610
3611         if (WARN_ON(ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED))
3612                 return 0;
3613
3614         list_for_each_entry(sdata, &ctx->reserved_vifs, reserved_chanctx_list)
3615                 if (sdata->reserved_radar_required)
3616                         radar_detect |= BIT(sdata->reserved_chandef.width);
3617
3618         /*
3619          * An in-place reservation context should not have any assigned vifs
3620          * until it replaces the other context.
3621          */
3622         WARN_ON(ctx->replace_state == IEEE80211_CHANCTX_REPLACES_OTHER &&
3623                 !list_empty(&ctx->assigned_vifs));
3624
3625         list_for_each_entry(sdata, &ctx->assigned_vifs, assigned_chanctx_list)
3626                 if (sdata->radar_required)
3627                         radar_detect |= BIT(sdata->vif.bss_conf.chandef.width);
3628
3629         return radar_detect;
3630 }
3631
3632 int ieee80211_check_combinations(struct ieee80211_sub_if_data *sdata,
3633                                  const struct cfg80211_chan_def *chandef,
3634                                  enum ieee80211_chanctx_mode chanmode,
3635                                  u8 radar_detect)
3636 {
3637         struct ieee80211_local *local = sdata->local;
3638         struct ieee80211_sub_if_data *sdata_iter;
3639         enum nl80211_iftype iftype = sdata->wdev.iftype;
3640         struct ieee80211_chanctx *ctx;
3641         int total = 1;
3642         struct iface_combination_params params = {
3643                 .radar_detect = radar_detect,
3644         };
3645
3646         lockdep_assert_held(&local->chanctx_mtx);
3647
3648         if (WARN_ON(hweight32(radar_detect) > 1))
3649                 return -EINVAL;
3650
3651         if (WARN_ON(chandef && chanmode == IEEE80211_CHANCTX_SHARED &&
3652                     !chandef->chan))
3653                 return -EINVAL;
3654
3655         if (WARN_ON(iftype >= NUM_NL80211_IFTYPES))
3656                 return -EINVAL;
3657
3658         if (sdata->vif.type == NL80211_IFTYPE_AP ||
3659             sdata->vif.type == NL80211_IFTYPE_MESH_POINT) {
3660                 /*
3661                  * always passing this is harmless, since it'll be the
3662                  * same value that cfg80211 finds if it finds the same
3663                  * interface ... and that's always allowed
3664                  */
3665                 params.new_beacon_int = sdata->vif.bss_conf.beacon_int;
3666         }
3667
3668         /* Always allow software iftypes */
3669         if (local->hw.wiphy->software_iftypes & BIT(iftype)) {
3670                 if (radar_detect)
3671                         return -EINVAL;
3672                 return 0;
3673         }
3674
3675         if (chandef)
3676                 params.num_different_channels = 1;
3677
3678         if (iftype != NL80211_IFTYPE_UNSPECIFIED)
3679                 params.iftype_num[iftype] = 1;
3680
3681         list_for_each_entry(ctx, &local->chanctx_list, list) {
3682                 if (ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED)
3683                         continue;
3684                 params.radar_detect |=
3685                         ieee80211_chanctx_radar_detect(local, ctx);
3686                 if (ctx->mode == IEEE80211_CHANCTX_EXCLUSIVE) {
3687                         params.num_different_channels++;
3688                         continue;
3689                 }
3690                 if (chandef && chanmode == IEEE80211_CHANCTX_SHARED &&
3691                     cfg80211_chandef_compatible(chandef,
3692                                                 &ctx->conf.def))
3693                         continue;
3694                 params.num_different_channels++;
3695         }
3696
3697         list_for_each_entry_rcu(sdata_iter, &local->interfaces, list) {
3698                 struct wireless_dev *wdev_iter;
3699
3700                 wdev_iter = &sdata_iter->wdev;
3701
3702                 if (sdata_iter == sdata ||
3703                     !ieee80211_sdata_running(sdata_iter) ||
3704                     local->hw.wiphy->software_iftypes & BIT(wdev_iter->iftype))
3705                         continue;
3706
3707                 params.iftype_num[wdev_iter->iftype]++;
3708                 total++;
3709         }
3710
3711         if (total == 1 && !params.radar_detect)
3712                 return 0;
3713
3714         return cfg80211_check_combinations(local->hw.wiphy, &params);
3715 }
3716
3717 static void
3718 ieee80211_iter_max_chans(const struct ieee80211_iface_combination *c,
3719                          void *data)
3720 {
3721         u32 *max_num_different_channels = data;
3722
3723         *max_num_different_channels = max(*max_num_different_channels,
3724                                           c->num_different_channels);
3725 }
3726
3727 int ieee80211_max_num_channels(struct ieee80211_local *local)
3728 {
3729         struct ieee80211_sub_if_data *sdata;
3730         struct ieee80211_chanctx *ctx;
3731         u32 max_num_different_channels = 1;
3732         int err;
3733         struct iface_combination_params params = {0};
3734
3735         lockdep_assert_held(&local->chanctx_mtx);
3736
3737         list_for_each_entry(ctx, &local->chanctx_list, list) {
3738                 if (ctx->replace_state == IEEE80211_CHANCTX_WILL_BE_REPLACED)
3739                         continue;
3740
3741                 params.num_different_channels++;
3742
3743                 params.radar_detect |=
3744                         ieee80211_chanctx_radar_detect(local, ctx);
3745         }
3746
3747         list_for_each_entry_rcu(sdata, &local->interfaces, list)
3748                 params.iftype_num[sdata->wdev.iftype]++;
3749
3750         err = cfg80211_iter_combinations(local->hw.wiphy, &params,
3751                                          ieee80211_iter_max_chans,
3752                                          &max_num_different_channels);
3753         if (err < 0)
3754                 return err;
3755
3756         return max_num_different_channels;
3757 }
3758
3759 u8 *ieee80211_add_wmm_info_ie(u8 *buf, u8 qosinfo)
3760 {
3761         *buf++ = WLAN_EID_VENDOR_SPECIFIC;
3762         *buf++ = 7; /* len */
3763         *buf++ = 0x00; /* Microsoft OUI 00:50:F2 */
3764         *buf++ = 0x50;
3765         *buf++ = 0xf2;
3766         *buf++ = 2; /* WME */
3767         *buf++ = 0; /* WME info */
3768         *buf++ = 1; /* WME ver */
3769         *buf++ = qosinfo; /* U-APSD no in use */
3770
3771         return buf;
3772 }
3773
3774 void ieee80211_txq_get_depth(struct ieee80211_txq *txq,
3775                              unsigned long *frame_cnt,
3776                              unsigned long *byte_cnt)
3777 {
3778         struct txq_info *txqi = to_txq_info(txq);
3779         u32 frag_cnt = 0, frag_bytes = 0;
3780         struct sk_buff *skb;
3781
3782         skb_queue_walk(&txqi->frags, skb) {
3783                 frag_cnt++;
3784                 frag_bytes += skb->len;
3785         }
3786
3787         if (frame_cnt)
3788                 *frame_cnt = txqi->tin.backlog_packets + frag_cnt;
3789
3790         if (byte_cnt)
3791                 *byte_cnt = txqi->tin.backlog_bytes + frag_bytes;
3792 }
3793 EXPORT_SYMBOL(ieee80211_txq_get_depth);
3794
3795 const u8 ieee80211_ac_to_qos_mask[IEEE80211_NUM_ACS] = {
3796         IEEE80211_WMM_IE_STA_QOSINFO_AC_VO,
3797         IEEE80211_WMM_IE_STA_QOSINFO_AC_VI,
3798         IEEE80211_WMM_IE_STA_QOSINFO_AC_BE,
3799         IEEE80211_WMM_IE_STA_QOSINFO_AC_BK
3800 };