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