9f21a69f09171ca0a290007e0c87e834a0e5e6b7
[linux-2.6.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/types.h>
17 #include <linux/slab.h>
18 #include <linux/skbuff.h>
19 #include <linux/etherdevice.h>
20 #include <linux/if_arp.h>
21 #include <linux/bitmap.h>
22 #include <linux/crc32.h>
23 #include <net/net_namespace.h>
24 #include <net/cfg80211.h>
25 #include <net/rtnetlink.h>
26
27 #include "ieee80211_i.h"
28 #include "driver-ops.h"
29 #include "rate.h"
30 #include "mesh.h"
31 #include "wme.h"
32 #include "led.h"
33 #include "wep.h"
34
35 /* privid for wiphys to determine whether they belong to us or not */
36 void *mac80211_wiphy_privid = &mac80211_wiphy_privid;
37
38 struct ieee80211_hw *wiphy_to_ieee80211_hw(struct wiphy *wiphy)
39 {
40         struct ieee80211_local *local;
41         BUG_ON(!wiphy);
42
43         local = wiphy_priv(wiphy);
44         return &local->hw;
45 }
46 EXPORT_SYMBOL(wiphy_to_ieee80211_hw);
47
48 u8 *ieee80211_get_bssid(struct ieee80211_hdr *hdr, size_t len,
49                         enum nl80211_iftype type)
50 {
51         __le16 fc = hdr->frame_control;
52
53          /* drop ACK/CTS frames and incorrect hdr len (ctrl) */
54         if (len < 16)
55                 return NULL;
56
57         if (ieee80211_is_data(fc)) {
58                 if (len < 24) /* drop incorrect hdr len (data) */
59                         return NULL;
60
61                 if (ieee80211_has_a4(fc))
62                         return NULL;
63                 if (ieee80211_has_tods(fc))
64                         return hdr->addr1;
65                 if (ieee80211_has_fromds(fc))
66                         return hdr->addr2;
67
68                 return hdr->addr3;
69         }
70
71         if (ieee80211_is_mgmt(fc)) {
72                 if (len < 24) /* drop incorrect hdr len (mgmt) */
73                         return NULL;
74                 return hdr->addr3;
75         }
76
77         if (ieee80211_is_ctl(fc)) {
78                 if(ieee80211_is_pspoll(fc))
79                         return hdr->addr1;
80
81                 if (ieee80211_is_back_req(fc)) {
82                         switch (type) {
83                         case NL80211_IFTYPE_STATION:
84                                 return hdr->addr2;
85                         case NL80211_IFTYPE_AP:
86                         case NL80211_IFTYPE_AP_VLAN:
87                                 return hdr->addr1;
88                         default:
89                                 break; /* fall through to the return */
90                         }
91                 }
92         }
93
94         return NULL;
95 }
96
97 void ieee80211_tx_set_protected(struct ieee80211_tx_data *tx)
98 {
99         struct sk_buff *skb = tx->skb;
100         struct ieee80211_hdr *hdr;
101
102         do {
103                 hdr = (struct ieee80211_hdr *) skb->data;
104                 hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
105         } while ((skb = skb->next));
106 }
107
108 int ieee80211_frame_duration(struct ieee80211_local *local, size_t len,
109                              int rate, int erp, int short_preamble)
110 {
111         int dur;
112
113         /* calculate duration (in microseconds, rounded up to next higher
114          * integer if it includes a fractional microsecond) to send frame of
115          * len bytes (does not include FCS) at the given rate. Duration will
116          * also include SIFS.
117          *
118          * rate is in 100 kbps, so divident is multiplied by 10 in the
119          * DIV_ROUND_UP() operations.
120          */
121
122         if (local->hw.conf.channel->band == IEEE80211_BAND_5GHZ || erp) {
123                 /*
124                  * OFDM:
125                  *
126                  * N_DBPS = DATARATE x 4
127                  * N_SYM = Ceiling((16+8xLENGTH+6) / N_DBPS)
128                  *      (16 = SIGNAL time, 6 = tail bits)
129                  * TXTIME = T_PREAMBLE + T_SIGNAL + T_SYM x N_SYM + Signal Ext
130                  *
131                  * T_SYM = 4 usec
132                  * 802.11a - 17.5.2: aSIFSTime = 16 usec
133                  * 802.11g - 19.8.4: aSIFSTime = 10 usec +
134                  *      signal ext = 6 usec
135                  */
136                 dur = 16; /* SIFS + signal ext */
137                 dur += 16; /* 17.3.2.3: T_PREAMBLE = 16 usec */
138                 dur += 4; /* 17.3.2.3: T_SIGNAL = 4 usec */
139                 dur += 4 * DIV_ROUND_UP((16 + 8 * (len + 4) + 6) * 10,
140                                         4 * rate); /* T_SYM x N_SYM */
141         } else {
142                 /*
143                  * 802.11b or 802.11g with 802.11b compatibility:
144                  * 18.3.4: TXTIME = PreambleLength + PLCPHeaderTime +
145                  * Ceiling(((LENGTH+PBCC)x8)/DATARATE). PBCC=0.
146                  *
147                  * 802.11 (DS): 15.3.3, 802.11b: 18.3.4
148                  * aSIFSTime = 10 usec
149                  * aPreambleLength = 144 usec or 72 usec with short preamble
150                  * aPLCPHeaderLength = 48 usec or 24 usec with short preamble
151                  */
152                 dur = 10; /* aSIFSTime = 10 usec */
153                 dur += short_preamble ? (72 + 24) : (144 + 48);
154
155                 dur += DIV_ROUND_UP(8 * (len + 4) * 10, rate);
156         }
157
158         return dur;
159 }
160
161 /* Exported duration function for driver use */
162 __le16 ieee80211_generic_frame_duration(struct ieee80211_hw *hw,
163                                         struct ieee80211_vif *vif,
164                                         size_t frame_len,
165                                         struct ieee80211_rate *rate)
166 {
167         struct ieee80211_local *local = hw_to_local(hw);
168         struct ieee80211_sub_if_data *sdata;
169         u16 dur;
170         int erp;
171         bool short_preamble = false;
172
173         erp = 0;
174         if (vif) {
175                 sdata = vif_to_sdata(vif);
176                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
177                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
178                         erp = rate->flags & IEEE80211_RATE_ERP_G;
179         }
180
181         dur = ieee80211_frame_duration(local, frame_len, rate->bitrate, erp,
182                                        short_preamble);
183
184         return cpu_to_le16(dur);
185 }
186 EXPORT_SYMBOL(ieee80211_generic_frame_duration);
187
188 __le16 ieee80211_rts_duration(struct ieee80211_hw *hw,
189                               struct ieee80211_vif *vif, size_t frame_len,
190                               const struct ieee80211_tx_info *frame_txctl)
191 {
192         struct ieee80211_local *local = hw_to_local(hw);
193         struct ieee80211_rate *rate;
194         struct ieee80211_sub_if_data *sdata;
195         bool short_preamble;
196         int erp;
197         u16 dur;
198         struct ieee80211_supported_band *sband;
199
200         sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
201
202         short_preamble = false;
203
204         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
205
206         erp = 0;
207         if (vif) {
208                 sdata = vif_to_sdata(vif);
209                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
210                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
211                         erp = rate->flags & IEEE80211_RATE_ERP_G;
212         }
213
214         /* CTS duration */
215         dur = ieee80211_frame_duration(local, 10, rate->bitrate,
216                                        erp, short_preamble);
217         /* Data frame duration */
218         dur += ieee80211_frame_duration(local, frame_len, rate->bitrate,
219                                         erp, short_preamble);
220         /* ACK duration */
221         dur += ieee80211_frame_duration(local, 10, rate->bitrate,
222                                         erp, short_preamble);
223
224         return cpu_to_le16(dur);
225 }
226 EXPORT_SYMBOL(ieee80211_rts_duration);
227
228 __le16 ieee80211_ctstoself_duration(struct ieee80211_hw *hw,
229                                     struct ieee80211_vif *vif,
230                                     size_t frame_len,
231                                     const struct ieee80211_tx_info *frame_txctl)
232 {
233         struct ieee80211_local *local = hw_to_local(hw);
234         struct ieee80211_rate *rate;
235         struct ieee80211_sub_if_data *sdata;
236         bool short_preamble;
237         int erp;
238         u16 dur;
239         struct ieee80211_supported_band *sband;
240
241         sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
242
243         short_preamble = false;
244
245         rate = &sband->bitrates[frame_txctl->control.rts_cts_rate_idx];
246         erp = 0;
247         if (vif) {
248                 sdata = vif_to_sdata(vif);
249                 short_preamble = sdata->vif.bss_conf.use_short_preamble;
250                 if (sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
251                         erp = rate->flags & IEEE80211_RATE_ERP_G;
252         }
253
254         /* Data frame duration */
255         dur = ieee80211_frame_duration(local, frame_len, rate->bitrate,
256                                        erp, short_preamble);
257         if (!(frame_txctl->flags & IEEE80211_TX_CTL_NO_ACK)) {
258                 /* ACK duration */
259                 dur += ieee80211_frame_duration(local, 10, rate->bitrate,
260                                                 erp, short_preamble);
261         }
262
263         return cpu_to_le16(dur);
264 }
265 EXPORT_SYMBOL(ieee80211_ctstoself_duration);
266
267 static void __ieee80211_wake_queue(struct ieee80211_hw *hw, int queue,
268                                    enum queue_stop_reason reason)
269 {
270         struct ieee80211_local *local = hw_to_local(hw);
271         struct ieee80211_sub_if_data *sdata;
272
273         trace_wake_queue(local, queue, reason);
274
275         if (WARN_ON(queue >= hw->queues))
276                 return;
277
278         __clear_bit(reason, &local->queue_stop_reasons[queue]);
279
280         if (local->queue_stop_reasons[queue] != 0)
281                 /* someone still has this queue stopped */
282                 return;
283
284         if (skb_queue_empty(&local->pending[queue])) {
285                 rcu_read_lock();
286                 list_for_each_entry_rcu(sdata, &local->interfaces, list) {
287                         if (test_bit(SDATA_STATE_OFFCHANNEL, &sdata->state))
288                                 continue;
289                         netif_wake_subqueue(sdata->dev, queue);
290                 }
291                 rcu_read_unlock();
292         } else
293                 tasklet_schedule(&local->tx_pending_tasklet);
294 }
295
296 void ieee80211_wake_queue_by_reason(struct ieee80211_hw *hw, int queue,
297                                     enum queue_stop_reason reason)
298 {
299         struct ieee80211_local *local = hw_to_local(hw);
300         unsigned long flags;
301
302         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
303         __ieee80211_wake_queue(hw, queue, reason);
304         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
305 }
306
307 void ieee80211_wake_queue(struct ieee80211_hw *hw, int queue)
308 {
309         ieee80211_wake_queue_by_reason(hw, queue,
310                                        IEEE80211_QUEUE_STOP_REASON_DRIVER);
311 }
312 EXPORT_SYMBOL(ieee80211_wake_queue);
313
314 static void __ieee80211_stop_queue(struct ieee80211_hw *hw, int queue,
315                                    enum queue_stop_reason reason)
316 {
317         struct ieee80211_local *local = hw_to_local(hw);
318         struct ieee80211_sub_if_data *sdata;
319
320         trace_stop_queue(local, queue, reason);
321
322         if (WARN_ON(queue >= hw->queues))
323                 return;
324
325         __set_bit(reason, &local->queue_stop_reasons[queue]);
326
327         rcu_read_lock();
328         list_for_each_entry_rcu(sdata, &local->interfaces, list)
329                 netif_stop_subqueue(sdata->dev, queue);
330         rcu_read_unlock();
331 }
332
333 void ieee80211_stop_queue_by_reason(struct ieee80211_hw *hw, int queue,
334                                     enum queue_stop_reason reason)
335 {
336         struct ieee80211_local *local = hw_to_local(hw);
337         unsigned long flags;
338
339         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
340         __ieee80211_stop_queue(hw, queue, reason);
341         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
342 }
343
344 void ieee80211_stop_queue(struct ieee80211_hw *hw, int queue)
345 {
346         ieee80211_stop_queue_by_reason(hw, queue,
347                                        IEEE80211_QUEUE_STOP_REASON_DRIVER);
348 }
349 EXPORT_SYMBOL(ieee80211_stop_queue);
350
351 void ieee80211_add_pending_skb(struct ieee80211_local *local,
352                                struct sk_buff *skb)
353 {
354         struct ieee80211_hw *hw = &local->hw;
355         unsigned long flags;
356         int queue = skb_get_queue_mapping(skb);
357         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
358
359         if (WARN_ON(!info->control.vif)) {
360                 kfree_skb(skb);
361                 return;
362         }
363
364         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
365         __ieee80211_stop_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
366         __skb_queue_tail(&local->pending[queue], skb);
367         __ieee80211_wake_queue(hw, queue, IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
368         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
369 }
370
371 int ieee80211_add_pending_skbs(struct ieee80211_local *local,
372                                struct sk_buff_head *skbs)
373 {
374         struct ieee80211_hw *hw = &local->hw;
375         struct sk_buff *skb;
376         unsigned long flags;
377         int queue, ret = 0, i;
378
379         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
380         for (i = 0; i < hw->queues; i++)
381                 __ieee80211_stop_queue(hw, i,
382                         IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
383
384         while ((skb = skb_dequeue(skbs))) {
385                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
386
387                 if (WARN_ON(!info->control.vif)) {
388                         kfree_skb(skb);
389                         continue;
390                 }
391
392                 ret++;
393                 queue = skb_get_queue_mapping(skb);
394                 __skb_queue_tail(&local->pending[queue], skb);
395         }
396
397         for (i = 0; i < hw->queues; i++)
398                 __ieee80211_wake_queue(hw, i,
399                         IEEE80211_QUEUE_STOP_REASON_SKB_ADD);
400         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
401
402         return ret;
403 }
404
405 void ieee80211_stop_queues_by_reason(struct ieee80211_hw *hw,
406                                     enum queue_stop_reason reason)
407 {
408         struct ieee80211_local *local = hw_to_local(hw);
409         unsigned long flags;
410         int i;
411
412         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
413
414         for (i = 0; i < hw->queues; i++)
415                 __ieee80211_stop_queue(hw, i, reason);
416
417         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
418 }
419
420 void ieee80211_stop_queues(struct ieee80211_hw *hw)
421 {
422         ieee80211_stop_queues_by_reason(hw,
423                                         IEEE80211_QUEUE_STOP_REASON_DRIVER);
424 }
425 EXPORT_SYMBOL(ieee80211_stop_queues);
426
427 int ieee80211_queue_stopped(struct ieee80211_hw *hw, int queue)
428 {
429         struct ieee80211_local *local = hw_to_local(hw);
430         unsigned long flags;
431         int ret;
432
433         if (WARN_ON(queue >= hw->queues))
434                 return true;
435
436         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
437         ret = !!local->queue_stop_reasons[queue];
438         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
439         return ret;
440 }
441 EXPORT_SYMBOL(ieee80211_queue_stopped);
442
443 void ieee80211_wake_queues_by_reason(struct ieee80211_hw *hw,
444                                      enum queue_stop_reason reason)
445 {
446         struct ieee80211_local *local = hw_to_local(hw);
447         unsigned long flags;
448         int i;
449
450         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
451
452         for (i = 0; i < hw->queues; i++)
453                 __ieee80211_wake_queue(hw, i, reason);
454
455         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
456 }
457
458 void ieee80211_wake_queues(struct ieee80211_hw *hw)
459 {
460         ieee80211_wake_queues_by_reason(hw, IEEE80211_QUEUE_STOP_REASON_DRIVER);
461 }
462 EXPORT_SYMBOL(ieee80211_wake_queues);
463
464 void ieee80211_iterate_active_interfaces(
465         struct ieee80211_hw *hw,
466         void (*iterator)(void *data, u8 *mac,
467                          struct ieee80211_vif *vif),
468         void *data)
469 {
470         struct ieee80211_local *local = hw_to_local(hw);
471         struct ieee80211_sub_if_data *sdata;
472
473         mutex_lock(&local->iflist_mtx);
474
475         list_for_each_entry(sdata, &local->interfaces, list) {
476                 switch (sdata->vif.type) {
477                 case NUM_NL80211_IFTYPES:
478                 case NL80211_IFTYPE_UNSPECIFIED:
479                 case NL80211_IFTYPE_MONITOR:
480                 case NL80211_IFTYPE_AP_VLAN:
481                         continue;
482                 case NL80211_IFTYPE_AP:
483                 case NL80211_IFTYPE_STATION:
484                 case NL80211_IFTYPE_ADHOC:
485                 case NL80211_IFTYPE_WDS:
486                 case NL80211_IFTYPE_MESH_POINT:
487                         break;
488                 }
489                 if (ieee80211_sdata_running(sdata))
490                         iterator(data, sdata->vif.addr,
491                                  &sdata->vif);
492         }
493
494         mutex_unlock(&local->iflist_mtx);
495 }
496 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces);
497
498 void ieee80211_iterate_active_interfaces_atomic(
499         struct ieee80211_hw *hw,
500         void (*iterator)(void *data, u8 *mac,
501                          struct ieee80211_vif *vif),
502         void *data)
503 {
504         struct ieee80211_local *local = hw_to_local(hw);
505         struct ieee80211_sub_if_data *sdata;
506
507         rcu_read_lock();
508
509         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
510                 switch (sdata->vif.type) {
511                 case NUM_NL80211_IFTYPES:
512                 case NL80211_IFTYPE_UNSPECIFIED:
513                 case NL80211_IFTYPE_MONITOR:
514                 case NL80211_IFTYPE_AP_VLAN:
515                         continue;
516                 case NL80211_IFTYPE_AP:
517                 case NL80211_IFTYPE_STATION:
518                 case NL80211_IFTYPE_ADHOC:
519                 case NL80211_IFTYPE_WDS:
520                 case NL80211_IFTYPE_MESH_POINT:
521                         break;
522                 }
523                 if (ieee80211_sdata_running(sdata))
524                         iterator(data, sdata->vif.addr,
525                                  &sdata->vif);
526         }
527
528         rcu_read_unlock();
529 }
530 EXPORT_SYMBOL_GPL(ieee80211_iterate_active_interfaces_atomic);
531
532 /*
533  * Nothing should have been stuffed into the workqueue during
534  * the suspend->resume cycle. If this WARN is seen then there
535  * is a bug with either the driver suspend or something in
536  * mac80211 stuffing into the workqueue which we haven't yet
537  * cleared during mac80211's suspend cycle.
538  */
539 static bool ieee80211_can_queue_work(struct ieee80211_local *local)
540 {
541         if (WARN(local->suspended && !local->resuming,
542                  "queueing ieee80211 work while going to suspend\n"))
543                 return false;
544
545         return true;
546 }
547
548 void ieee80211_queue_work(struct ieee80211_hw *hw, struct work_struct *work)
549 {
550         struct ieee80211_local *local = hw_to_local(hw);
551
552         if (!ieee80211_can_queue_work(local))
553                 return;
554
555         queue_work(local->workqueue, work);
556 }
557 EXPORT_SYMBOL(ieee80211_queue_work);
558
559 void ieee80211_queue_delayed_work(struct ieee80211_hw *hw,
560                                   struct delayed_work *dwork,
561                                   unsigned long delay)
562 {
563         struct ieee80211_local *local = hw_to_local(hw);
564
565         if (!ieee80211_can_queue_work(local))
566                 return;
567
568         queue_delayed_work(local->workqueue, dwork, delay);
569 }
570 EXPORT_SYMBOL(ieee80211_queue_delayed_work);
571
572 void ieee802_11_parse_elems(u8 *start, size_t len,
573                             struct ieee802_11_elems *elems)
574 {
575         ieee802_11_parse_elems_crc(start, len, elems, 0, 0);
576 }
577
578 u32 ieee802_11_parse_elems_crc(u8 *start, size_t len,
579                                struct ieee802_11_elems *elems,
580                                u64 filter, u32 crc)
581 {
582         size_t left = len;
583         u8 *pos = start;
584         bool calc_crc = filter != 0;
585
586         memset(elems, 0, sizeof(*elems));
587         elems->ie_start = start;
588         elems->total_len = len;
589
590         while (left >= 2) {
591                 u8 id, elen;
592
593                 id = *pos++;
594                 elen = *pos++;
595                 left -= 2;
596
597                 if (elen > left)
598                         break;
599
600                 if (calc_crc && id < 64 && (filter & (1ULL << id)))
601                         crc = crc32_be(crc, pos - 2, elen + 2);
602
603                 switch (id) {
604                 case WLAN_EID_SSID:
605                         elems->ssid = pos;
606                         elems->ssid_len = elen;
607                         break;
608                 case WLAN_EID_SUPP_RATES:
609                         elems->supp_rates = pos;
610                         elems->supp_rates_len = elen;
611                         break;
612                 case WLAN_EID_FH_PARAMS:
613                         elems->fh_params = pos;
614                         elems->fh_params_len = elen;
615                         break;
616                 case WLAN_EID_DS_PARAMS:
617                         elems->ds_params = pos;
618                         elems->ds_params_len = elen;
619                         break;
620                 case WLAN_EID_CF_PARAMS:
621                         elems->cf_params = pos;
622                         elems->cf_params_len = elen;
623                         break;
624                 case WLAN_EID_TIM:
625                         if (elen >= sizeof(struct ieee80211_tim_ie)) {
626                                 elems->tim = (void *)pos;
627                                 elems->tim_len = elen;
628                         }
629                         break;
630                 case WLAN_EID_IBSS_PARAMS:
631                         elems->ibss_params = pos;
632                         elems->ibss_params_len = elen;
633                         break;
634                 case WLAN_EID_CHALLENGE:
635                         elems->challenge = pos;
636                         elems->challenge_len = elen;
637                         break;
638                 case WLAN_EID_VENDOR_SPECIFIC:
639                         if (elen >= 4 && pos[0] == 0x00 && pos[1] == 0x50 &&
640                             pos[2] == 0xf2) {
641                                 /* Microsoft OUI (00:50:F2) */
642
643                                 if (calc_crc)
644                                         crc = crc32_be(crc, pos - 2, elen + 2);
645
646                                 if (pos[3] == 1) {
647                                         /* OUI Type 1 - WPA IE */
648                                         elems->wpa = pos;
649                                         elems->wpa_len = elen;
650                                 } else if (elen >= 5 && pos[3] == 2) {
651                                         /* OUI Type 2 - WMM IE */
652                                         if (pos[4] == 0) {
653                                                 elems->wmm_info = pos;
654                                                 elems->wmm_info_len = elen;
655                                         } else if (pos[4] == 1) {
656                                                 elems->wmm_param = pos;
657                                                 elems->wmm_param_len = elen;
658                                         }
659                                 }
660                         }
661                         break;
662                 case WLAN_EID_RSN:
663                         elems->rsn = pos;
664                         elems->rsn_len = elen;
665                         break;
666                 case WLAN_EID_ERP_INFO:
667                         elems->erp_info = pos;
668                         elems->erp_info_len = elen;
669                         break;
670                 case WLAN_EID_EXT_SUPP_RATES:
671                         elems->ext_supp_rates = pos;
672                         elems->ext_supp_rates_len = elen;
673                         break;
674                 case WLAN_EID_HT_CAPABILITY:
675                         if (elen >= sizeof(struct ieee80211_ht_cap))
676                                 elems->ht_cap_elem = (void *)pos;
677                         break;
678                 case WLAN_EID_HT_INFORMATION:
679                         if (elen >= sizeof(struct ieee80211_ht_info))
680                                 elems->ht_info_elem = (void *)pos;
681                         break;
682                 case WLAN_EID_MESH_ID:
683                         elems->mesh_id = pos;
684                         elems->mesh_id_len = elen;
685                         break;
686                 case WLAN_EID_MESH_CONFIG:
687                         if (elen >= sizeof(struct ieee80211_meshconf_ie))
688                                 elems->mesh_config = (void *)pos;
689                         break;
690                 case WLAN_EID_PEER_LINK:
691                         elems->peer_link = pos;
692                         elems->peer_link_len = elen;
693                         break;
694                 case WLAN_EID_PREQ:
695                         elems->preq = pos;
696                         elems->preq_len = elen;
697                         break;
698                 case WLAN_EID_PREP:
699                         elems->prep = pos;
700                         elems->prep_len = elen;
701                         break;
702                 case WLAN_EID_PERR:
703                         elems->perr = pos;
704                         elems->perr_len = elen;
705                         break;
706                 case WLAN_EID_RANN:
707                         if (elen >= sizeof(struct ieee80211_rann_ie))
708                                 elems->rann = (void *)pos;
709                         break;
710                 case WLAN_EID_CHANNEL_SWITCH:
711                         elems->ch_switch_elem = pos;
712                         elems->ch_switch_elem_len = elen;
713                         break;
714                 case WLAN_EID_QUIET:
715                         if (!elems->quiet_elem) {
716                                 elems->quiet_elem = pos;
717                                 elems->quiet_elem_len = elen;
718                         }
719                         elems->num_of_quiet_elem++;
720                         break;
721                 case WLAN_EID_COUNTRY:
722                         elems->country_elem = pos;
723                         elems->country_elem_len = elen;
724                         break;
725                 case WLAN_EID_PWR_CONSTRAINT:
726                         elems->pwr_constr_elem = pos;
727                         elems->pwr_constr_elem_len = elen;
728                         break;
729                 case WLAN_EID_TIMEOUT_INTERVAL:
730                         elems->timeout_int = pos;
731                         elems->timeout_int_len = elen;
732                         break;
733                 default:
734                         break;
735                 }
736
737                 left -= elen;
738                 pos += elen;
739         }
740
741         return crc;
742 }
743
744 void ieee80211_set_wmm_default(struct ieee80211_sub_if_data *sdata)
745 {
746         struct ieee80211_local *local = sdata->local;
747         struct ieee80211_tx_queue_params qparam;
748         int queue;
749         bool use_11b;
750         int aCWmin, aCWmax;
751
752         if (!local->ops->conf_tx)
753                 return;
754
755         memset(&qparam, 0, sizeof(qparam));
756
757         use_11b = (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ) &&
758                  !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE);
759
760         for (queue = 0; queue < local_to_hw(local)->queues; queue++) {
761                 /* Set defaults according to 802.11-2007 Table 7-37 */
762                 aCWmax = 1023;
763                 if (use_11b)
764                         aCWmin = 31;
765                 else
766                         aCWmin = 15;
767
768                 switch (queue) {
769                 case 3: /* AC_BK */
770                         qparam.cw_max = aCWmax;
771                         qparam.cw_min = aCWmin;
772                         qparam.txop = 0;
773                         qparam.aifs = 7;
774                         break;
775                 default: /* never happens but let's not leave undefined */
776                 case 2: /* AC_BE */
777                         qparam.cw_max = aCWmax;
778                         qparam.cw_min = aCWmin;
779                         qparam.txop = 0;
780                         qparam.aifs = 3;
781                         break;
782                 case 1: /* AC_VI */
783                         qparam.cw_max = aCWmin;
784                         qparam.cw_min = (aCWmin + 1) / 2 - 1;
785                         if (use_11b)
786                                 qparam.txop = 6016/32;
787                         else
788                                 qparam.txop = 3008/32;
789                         qparam.aifs = 2;
790                         break;
791                 case 0: /* AC_VO */
792                         qparam.cw_max = (aCWmin + 1) / 2 - 1;
793                         qparam.cw_min = (aCWmin + 1) / 4 - 1;
794                         if (use_11b)
795                                 qparam.txop = 3264/32;
796                         else
797                                 qparam.txop = 1504/32;
798                         qparam.aifs = 2;
799                         break;
800                 }
801
802                 qparam.uapsd = false;
803
804                 drv_conf_tx(local, queue, &qparam);
805         }
806
807         /* after reinitialize QoS TX queues setting to default,
808          * disable QoS at all */
809
810         if (sdata->vif.type != NL80211_IFTYPE_MONITOR) {
811                 sdata->vif.bss_conf.qos =
812                         sdata->vif.type != NL80211_IFTYPE_STATION;
813                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_QOS);
814         }
815 }
816
817 void ieee80211_sta_def_wmm_params(struct ieee80211_sub_if_data *sdata,
818                                   const size_t supp_rates_len,
819                                   const u8 *supp_rates)
820 {
821         struct ieee80211_local *local = sdata->local;
822         int i, have_higher_than_11mbit = 0;
823
824         /* cf. IEEE 802.11 9.2.12 */
825         for (i = 0; i < supp_rates_len; i++)
826                 if ((supp_rates[i] & 0x7f) * 5 > 110)
827                         have_higher_than_11mbit = 1;
828
829         if (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ &&
830             have_higher_than_11mbit)
831                 sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
832         else
833                 sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;
834
835         ieee80211_set_wmm_default(sdata);
836 }
837
838 u32 ieee80211_mandatory_rates(struct ieee80211_local *local,
839                               enum ieee80211_band band)
840 {
841         struct ieee80211_supported_band *sband;
842         struct ieee80211_rate *bitrates;
843         u32 mandatory_rates;
844         enum ieee80211_rate_flags mandatory_flag;
845         int i;
846
847         sband = local->hw.wiphy->bands[band];
848         if (!sband) {
849                 WARN_ON(1);
850                 sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
851         }
852
853         if (band == IEEE80211_BAND_2GHZ)
854                 mandatory_flag = IEEE80211_RATE_MANDATORY_B;
855         else
856                 mandatory_flag = IEEE80211_RATE_MANDATORY_A;
857
858         bitrates = sband->bitrates;
859         mandatory_rates = 0;
860         for (i = 0; i < sband->n_bitrates; i++)
861                 if (bitrates[i].flags & mandatory_flag)
862                         mandatory_rates |= BIT(i);
863         return mandatory_rates;
864 }
865
866 void ieee80211_send_auth(struct ieee80211_sub_if_data *sdata,
867                          u16 transaction, u16 auth_alg,
868                          u8 *extra, size_t extra_len, const u8 *bssid,
869                          const u8 *key, u8 key_len, u8 key_idx)
870 {
871         struct ieee80211_local *local = sdata->local;
872         struct sk_buff *skb;
873         struct ieee80211_mgmt *mgmt;
874         int err;
875
876         skb = dev_alloc_skb(local->hw.extra_tx_headroom +
877                             sizeof(*mgmt) + 6 + extra_len);
878         if (!skb) {
879                 printk(KERN_DEBUG "%s: failed to allocate buffer for auth "
880                        "frame\n", sdata->name);
881                 return;
882         }
883         skb_reserve(skb, local->hw.extra_tx_headroom);
884
885         mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24 + 6);
886         memset(mgmt, 0, 24 + 6);
887         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
888                                           IEEE80211_STYPE_AUTH);
889         memcpy(mgmt->da, bssid, ETH_ALEN);
890         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
891         memcpy(mgmt->bssid, bssid, ETH_ALEN);
892         mgmt->u.auth.auth_alg = cpu_to_le16(auth_alg);
893         mgmt->u.auth.auth_transaction = cpu_to_le16(transaction);
894         mgmt->u.auth.status_code = cpu_to_le16(0);
895         if (extra)
896                 memcpy(skb_put(skb, extra_len), extra, extra_len);
897
898         if (auth_alg == WLAN_AUTH_SHARED_KEY && transaction == 3) {
899                 mgmt->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
900                 err = ieee80211_wep_encrypt(local, skb, key, key_len, key_idx);
901                 WARN_ON(err);
902         }
903
904         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
905         ieee80211_tx_skb(sdata, skb);
906 }
907
908 int ieee80211_build_preq_ies(struct ieee80211_local *local, u8 *buffer,
909                              const u8 *ie, size_t ie_len,
910                              enum ieee80211_band band)
911 {
912         struct ieee80211_supported_band *sband;
913         u8 *pos;
914         size_t offset = 0, noffset;
915         int supp_rates_len, i;
916
917         sband = local->hw.wiphy->bands[band];
918
919         pos = buffer;
920
921         supp_rates_len = min_t(int, sband->n_bitrates, 8);
922
923         *pos++ = WLAN_EID_SUPP_RATES;
924         *pos++ = supp_rates_len;
925
926         for (i = 0; i < supp_rates_len; i++) {
927                 int rate = sband->bitrates[i].bitrate;
928                 *pos++ = (u8) (rate / 5);
929         }
930
931         /* insert "request information" if in custom IEs */
932         if (ie && ie_len) {
933                 static const u8 before_extrates[] = {
934                         WLAN_EID_SSID,
935                         WLAN_EID_SUPP_RATES,
936                         WLAN_EID_REQUEST,
937                 };
938                 noffset = ieee80211_ie_split(ie, ie_len,
939                                              before_extrates,
940                                              ARRAY_SIZE(before_extrates),
941                                              offset);
942                 memcpy(pos, ie + offset, noffset - offset);
943                 pos += noffset - offset;
944                 offset = noffset;
945         }
946
947         if (sband->n_bitrates > i) {
948                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
949                 *pos++ = sband->n_bitrates - i;
950
951                 for (; i < sband->n_bitrates; i++) {
952                         int rate = sband->bitrates[i].bitrate;
953                         *pos++ = (u8) (rate / 5);
954                 }
955         }
956
957         /* insert custom IEs that go before HT */
958         if (ie && ie_len) {
959                 static const u8 before_ht[] = {
960                         WLAN_EID_SSID,
961                         WLAN_EID_SUPP_RATES,
962                         WLAN_EID_REQUEST,
963                         WLAN_EID_EXT_SUPP_RATES,
964                         WLAN_EID_DS_PARAMS,
965                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
966                 };
967                 noffset = ieee80211_ie_split(ie, ie_len,
968                                              before_ht, ARRAY_SIZE(before_ht),
969                                              offset);
970                 memcpy(pos, ie + offset, noffset - offset);
971                 pos += noffset - offset;
972                 offset = noffset;
973         }
974
975         if (sband->ht_cap.ht_supported) {
976                 u16 cap = sband->ht_cap.cap;
977                 __le16 tmp;
978
979                 if (ieee80211_disable_40mhz_24ghz &&
980                     sband->band == IEEE80211_BAND_2GHZ) {
981                         cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
982                         cap &= ~IEEE80211_HT_CAP_SGI_40;
983                 }
984
985                 *pos++ = WLAN_EID_HT_CAPABILITY;
986                 *pos++ = sizeof(struct ieee80211_ht_cap);
987                 memset(pos, 0, sizeof(struct ieee80211_ht_cap));
988                 tmp = cpu_to_le16(cap);
989                 memcpy(pos, &tmp, sizeof(u16));
990                 pos += sizeof(u16);
991                 *pos++ = sband->ht_cap.ampdu_factor |
992                          (sband->ht_cap.ampdu_density <<
993                                 IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT);
994                 memcpy(pos, &sband->ht_cap.mcs, sizeof(sband->ht_cap.mcs));
995                 pos += sizeof(sband->ht_cap.mcs);
996                 pos += 2 + 4 + 1; /* ext info, BF cap, antsel */
997         }
998
999         /*
1000          * If adding more here, adjust code in main.c
1001          * that calculates local->scan_ies_len.
1002          */
1003
1004         /* add any remaining custom IEs */
1005         if (ie && ie_len) {
1006                 noffset = ie_len;
1007                 memcpy(pos, ie + offset, noffset - offset);
1008                 pos += noffset - offset;
1009         }
1010
1011         return pos - buffer;
1012 }
1013
1014 void ieee80211_send_probe_req(struct ieee80211_sub_if_data *sdata, u8 *dst,
1015                               const u8 *ssid, size_t ssid_len,
1016                               const u8 *ie, size_t ie_len)
1017 {
1018         struct ieee80211_local *local = sdata->local;
1019         struct sk_buff *skb;
1020         struct ieee80211_mgmt *mgmt;
1021         size_t buf_len;
1022         u8 *buf;
1023
1024         /* FIXME: come up with a proper value */
1025         buf = kmalloc(200 + ie_len, GFP_KERNEL);
1026         if (!buf) {
1027                 printk(KERN_DEBUG "%s: failed to allocate temporary IE "
1028                        "buffer\n", sdata->name);
1029                 return;
1030         }
1031
1032         buf_len = ieee80211_build_preq_ies(local, buf, ie, ie_len,
1033                                            local->hw.conf.channel->band);
1034
1035         skb = ieee80211_probereq_get(&local->hw, &sdata->vif,
1036                                      ssid, ssid_len,
1037                                      buf, buf_len);
1038
1039         if (dst) {
1040                 mgmt = (struct ieee80211_mgmt *) skb->data;
1041                 memcpy(mgmt->da, dst, ETH_ALEN);
1042                 memcpy(mgmt->bssid, dst, ETH_ALEN);
1043         }
1044
1045         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1046         ieee80211_tx_skb(sdata, skb);
1047         kfree(buf);
1048 }
1049
1050 u32 ieee80211_sta_get_rates(struct ieee80211_local *local,
1051                             struct ieee802_11_elems *elems,
1052                             enum ieee80211_band band)
1053 {
1054         struct ieee80211_supported_band *sband;
1055         struct ieee80211_rate *bitrates;
1056         size_t num_rates;
1057         u32 supp_rates;
1058         int i, j;
1059         sband = local->hw.wiphy->bands[band];
1060
1061         if (!sband) {
1062                 WARN_ON(1);
1063                 sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
1064         }
1065
1066         bitrates = sband->bitrates;
1067         num_rates = sband->n_bitrates;
1068         supp_rates = 0;
1069         for (i = 0; i < elems->supp_rates_len +
1070                      elems->ext_supp_rates_len; i++) {
1071                 u8 rate = 0;
1072                 int own_rate;
1073                 if (i < elems->supp_rates_len)
1074                         rate = elems->supp_rates[i];
1075                 else if (elems->ext_supp_rates)
1076                         rate = elems->ext_supp_rates
1077                                 [i - elems->supp_rates_len];
1078                 own_rate = 5 * (rate & 0x7f);
1079                 for (j = 0; j < num_rates; j++)
1080                         if (bitrates[j].bitrate == own_rate)
1081                                 supp_rates |= BIT(j);
1082         }
1083         return supp_rates;
1084 }
1085
1086 void ieee80211_stop_device(struct ieee80211_local *local)
1087 {
1088         ieee80211_led_radio(local, false);
1089
1090         cancel_work_sync(&local->reconfig_filter);
1091
1092         flush_workqueue(local->workqueue);
1093         drv_stop(local);
1094 }
1095
1096 int ieee80211_reconfig(struct ieee80211_local *local)
1097 {
1098         struct ieee80211_hw *hw = &local->hw;
1099         struct ieee80211_sub_if_data *sdata;
1100         struct sta_info *sta;
1101         int res;
1102
1103         if (local->suspended)
1104                 local->resuming = true;
1105
1106         /* restart hardware */
1107         if (local->open_count) {
1108                 /*
1109                  * Upon resume hardware can sometimes be goofy due to
1110                  * various platform / driver / bus issues, so restarting
1111                  * the device may at times not work immediately. Propagate
1112                  * the error.
1113                  */
1114                 res = drv_start(local);
1115                 if (res) {
1116                         WARN(local->suspended, "Hardware became unavailable "
1117                              "upon resume. This could be a software issue "
1118                              "prior to suspend or a hardware issue.\n");
1119                         return res;
1120                 }
1121
1122                 ieee80211_led_radio(local, true);
1123         }
1124
1125         /* add interfaces */
1126         list_for_each_entry(sdata, &local->interfaces, list) {
1127                 if (sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
1128                     sdata->vif.type != NL80211_IFTYPE_MONITOR &&
1129                     ieee80211_sdata_running(sdata))
1130                         res = drv_add_interface(local, &sdata->vif);
1131         }
1132
1133         /* add STAs back */
1134         mutex_lock(&local->sta_mtx);
1135         list_for_each_entry(sta, &local->sta_list, list) {
1136                 if (sta->uploaded) {
1137                         sdata = sta->sdata;
1138                         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
1139                                 sdata = container_of(sdata->bss,
1140                                              struct ieee80211_sub_if_data,
1141                                              u.ap);
1142
1143                         WARN_ON(drv_sta_add(local, sdata, &sta->sta));
1144                 }
1145         }
1146         mutex_unlock(&local->sta_mtx);
1147
1148         /* setup RTS threshold */
1149         drv_set_rts_threshold(local, hw->wiphy->rts_threshold);
1150
1151         /* reconfigure hardware */
1152         ieee80211_hw_config(local, ~0);
1153
1154         ieee80211_configure_filter(local);
1155
1156         /* Finally also reconfigure all the BSS information */
1157         list_for_each_entry(sdata, &local->interfaces, list) {
1158                 u32 changed;
1159
1160                 if (!ieee80211_sdata_running(sdata))
1161                         continue;
1162
1163                 /* common change flags for all interface types */
1164                 changed = BSS_CHANGED_ERP_CTS_PROT |
1165                           BSS_CHANGED_ERP_PREAMBLE |
1166                           BSS_CHANGED_ERP_SLOT |
1167                           BSS_CHANGED_HT |
1168                           BSS_CHANGED_BASIC_RATES |
1169                           BSS_CHANGED_BEACON_INT |
1170                           BSS_CHANGED_BSSID |
1171                           BSS_CHANGED_CQM |
1172                           BSS_CHANGED_QOS;
1173
1174                 switch (sdata->vif.type) {
1175                 case NL80211_IFTYPE_STATION:
1176                         changed |= BSS_CHANGED_ASSOC;
1177                         ieee80211_bss_info_change_notify(sdata, changed);
1178                         break;
1179                 case NL80211_IFTYPE_ADHOC:
1180                         changed |= BSS_CHANGED_IBSS;
1181                         /* fall through */
1182                 case NL80211_IFTYPE_AP:
1183                 case NL80211_IFTYPE_MESH_POINT:
1184                         changed |= BSS_CHANGED_BEACON |
1185                                    BSS_CHANGED_BEACON_ENABLED;
1186                         ieee80211_bss_info_change_notify(sdata, changed);
1187                         break;
1188                 case NL80211_IFTYPE_WDS:
1189                         break;
1190                 case NL80211_IFTYPE_AP_VLAN:
1191                 case NL80211_IFTYPE_MONITOR:
1192                         /* ignore virtual */
1193                         break;
1194                 case NL80211_IFTYPE_UNSPECIFIED:
1195                 case NUM_NL80211_IFTYPES:
1196                         WARN_ON(1);
1197                         break;
1198                 }
1199         }
1200
1201         /*
1202          * Clear the WLAN_STA_BLOCK_BA flag so new aggregation
1203          * sessions can be established after a resume.
1204          *
1205          * Also tear down aggregation sessions since reconfiguring
1206          * them in a hardware restart scenario is not easily done
1207          * right now, and the hardware will have lost information
1208          * about the sessions, but we and the AP still think they
1209          * are active. This is really a workaround though.
1210          */
1211         if (hw->flags & IEEE80211_HW_AMPDU_AGGREGATION) {
1212                 mutex_lock(&local->sta_mtx);
1213
1214                 list_for_each_entry(sta, &local->sta_list, list) {
1215                         ieee80211_sta_tear_down_BA_sessions(sta);
1216                         clear_sta_flags(sta, WLAN_STA_BLOCK_BA);
1217                 }
1218
1219                 mutex_unlock(&local->sta_mtx);
1220         }
1221
1222         /* add back keys */
1223         list_for_each_entry(sdata, &local->interfaces, list)
1224                 if (ieee80211_sdata_running(sdata))
1225                         ieee80211_enable_keys(sdata);
1226
1227         ieee80211_wake_queues_by_reason(hw,
1228                         IEEE80211_QUEUE_STOP_REASON_SUSPEND);
1229
1230         /*
1231          * If this is for hw restart things are still running.
1232          * We may want to change that later, however.
1233          */
1234         if (!local->suspended)
1235                 return 0;
1236
1237 #ifdef CONFIG_PM
1238         /* first set suspended false, then resuming */
1239         local->suspended = false;
1240         mb();
1241         local->resuming = false;
1242
1243         list_for_each_entry(sdata, &local->interfaces, list) {
1244                 switch(sdata->vif.type) {
1245                 case NL80211_IFTYPE_STATION:
1246                         ieee80211_sta_restart(sdata);
1247                         break;
1248                 case NL80211_IFTYPE_ADHOC:
1249                         ieee80211_ibss_restart(sdata);
1250                         break;
1251                 case NL80211_IFTYPE_MESH_POINT:
1252                         ieee80211_mesh_restart(sdata);
1253                         break;
1254                 default:
1255                         break;
1256                 }
1257         }
1258
1259         add_timer(&local->sta_cleanup);
1260
1261         mutex_lock(&local->sta_mtx);
1262         list_for_each_entry(sta, &local->sta_list, list)
1263                 mesh_plink_restart(sta);
1264         mutex_unlock(&local->sta_mtx);
1265 #else
1266         WARN_ON(1);
1267 #endif
1268         return 0;
1269 }
1270
1271 static int check_mgd_smps(struct ieee80211_if_managed *ifmgd,
1272                           enum ieee80211_smps_mode *smps_mode)
1273 {
1274         if (ifmgd->associated) {
1275                 *smps_mode = ifmgd->ap_smps;
1276
1277                 if (*smps_mode == IEEE80211_SMPS_AUTOMATIC) {
1278                         if (ifmgd->powersave)
1279                                 *smps_mode = IEEE80211_SMPS_DYNAMIC;
1280                         else
1281                                 *smps_mode = IEEE80211_SMPS_OFF;
1282                 }
1283
1284                 return 1;
1285         }
1286
1287         return 0;
1288 }
1289
1290 /* must hold iflist_mtx */
1291 void ieee80211_recalc_smps(struct ieee80211_local *local,
1292                            struct ieee80211_sub_if_data *forsdata)
1293 {
1294         struct ieee80211_sub_if_data *sdata;
1295         enum ieee80211_smps_mode smps_mode = IEEE80211_SMPS_OFF;
1296         int count = 0;
1297
1298         if (forsdata)
1299                 lockdep_assert_held(&forsdata->u.mgd.mtx);
1300
1301         lockdep_assert_held(&local->iflist_mtx);
1302
1303         /*
1304          * This function could be improved to handle multiple
1305          * interfaces better, but right now it makes any
1306          * non-station interfaces force SM PS to be turned
1307          * off. If there are multiple station interfaces it
1308          * could also use the best possible mode, e.g. if
1309          * one is in static and the other in dynamic then
1310          * dynamic is ok.
1311          */
1312
1313         list_for_each_entry(sdata, &local->interfaces, list) {
1314                 if (!ieee80211_sdata_running(sdata))
1315                         continue;
1316                 if (sdata->vif.type != NL80211_IFTYPE_STATION)
1317                         goto set;
1318                 if (sdata != forsdata) {
1319                         /*
1320                          * This nested is ok -- we are holding the iflist_mtx
1321                          * so can't get here twice or so. But it's required
1322                          * since normally we acquire it first and then the
1323                          * iflist_mtx.
1324                          */
1325                         mutex_lock_nested(&sdata->u.mgd.mtx, SINGLE_DEPTH_NESTING);
1326                         count += check_mgd_smps(&sdata->u.mgd, &smps_mode);
1327                         mutex_unlock(&sdata->u.mgd.mtx);
1328                 } else
1329                         count += check_mgd_smps(&sdata->u.mgd, &smps_mode);
1330
1331                 if (count > 1) {
1332                         smps_mode = IEEE80211_SMPS_OFF;
1333                         break;
1334                 }
1335         }
1336
1337         if (smps_mode == local->smps_mode)
1338                 return;
1339
1340  set:
1341         local->smps_mode = smps_mode;
1342         /* changed flag is auto-detected for this */
1343         ieee80211_hw_config(local, 0);
1344 }
1345
1346 static bool ieee80211_id_in_list(const u8 *ids, int n_ids, u8 id)
1347 {
1348         int i;
1349
1350         for (i = 0; i < n_ids; i++)
1351                 if (ids[i] == id)
1352                         return true;
1353         return false;
1354 }
1355
1356 /**
1357  * ieee80211_ie_split - split an IE buffer according to ordering
1358  *
1359  * @ies: the IE buffer
1360  * @ielen: the length of the IE buffer
1361  * @ids: an array with element IDs that are allowed before
1362  *      the split
1363  * @n_ids: the size of the element ID array
1364  * @offset: offset where to start splitting in the buffer
1365  *
1366  * This function splits an IE buffer by updating the @offset
1367  * variable to point to the location where the buffer should be
1368  * split.
1369  *
1370  * It assumes that the given IE buffer is well-formed, this
1371  * has to be guaranteed by the caller!
1372  *
1373  * It also assumes that the IEs in the buffer are ordered
1374  * correctly, if not the result of using this function will not
1375  * be ordered correctly either, i.e. it does no reordering.
1376  *
1377  * The function returns the offset where the next part of the
1378  * buffer starts, which may be @ielen if the entire (remainder)
1379  * of the buffer should be used.
1380  */
1381 size_t ieee80211_ie_split(const u8 *ies, size_t ielen,
1382                           const u8 *ids, int n_ids, size_t offset)
1383 {
1384         size_t pos = offset;
1385
1386         while (pos < ielen && ieee80211_id_in_list(ids, n_ids, ies[pos]))
1387                 pos += 2 + ies[pos + 1];
1388
1389         return pos;
1390 }
1391
1392 size_t ieee80211_ie_split_vendor(const u8 *ies, size_t ielen, size_t offset)
1393 {
1394         size_t pos = offset;
1395
1396         while (pos < ielen && ies[pos] != WLAN_EID_VENDOR_SPECIFIC)
1397                 pos += 2 + ies[pos + 1];
1398
1399         return pos;
1400 }