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