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