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