Merge branch 'batman-adv/next' of git://git.open-mesh.org/ecsv/linux-merge
[pandora-kernel.git] / net / mac80211 / work.c
1 /*
2  * mac80211 work implementation
3  *
4  * Copyright 2003-2008, Jouni Malinen <j@w1.fi>
5  * Copyright 2004, Instant802 Networks, Inc.
6  * Copyright 2005, Devicescape Software, Inc.
7  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
8  * Copyright 2007, Michael Wu <flamingice@sourmilk.net>
9  * Copyright 2009, Johannes Berg <johannes@sipsolutions.net>
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License version 2 as
13  * published by the Free Software Foundation.
14  */
15
16 #include <linux/delay.h>
17 #include <linux/if_ether.h>
18 #include <linux/skbuff.h>
19 #include <linux/if_arp.h>
20 #include <linux/etherdevice.h>
21 #include <linux/crc32.h>
22 #include <linux/slab.h>
23 #include <net/mac80211.h>
24 #include <asm/unaligned.h>
25
26 #include "ieee80211_i.h"
27 #include "rate.h"
28
29 #define IEEE80211_AUTH_TIMEOUT (HZ / 5)
30 #define IEEE80211_AUTH_MAX_TRIES 3
31 #define IEEE80211_ASSOC_TIMEOUT (HZ / 5)
32 #define IEEE80211_ASSOC_MAX_TRIES 3
33 #define IEEE80211_MAX_PROBE_TRIES 5
34
35 enum work_action {
36         WORK_ACT_MISMATCH,
37         WORK_ACT_NONE,
38         WORK_ACT_TIMEOUT,
39         WORK_ACT_DONE,
40 };
41
42
43 /* utils */
44 static inline void ASSERT_WORK_MTX(struct ieee80211_local *local)
45 {
46         lockdep_assert_held(&local->mtx);
47 }
48
49 /*
50  * We can have multiple work items (and connection probing)
51  * scheduling this timer, but we need to take care to only
52  * reschedule it when it should fire _earlier_ than it was
53  * asked for before, or if it's not pending right now. This
54  * function ensures that. Note that it then is required to
55  * run this function for all timeouts after the first one
56  * has happened -- the work that runs from this timer will
57  * do that.
58  */
59 static void run_again(struct ieee80211_local *local,
60                       unsigned long timeout)
61 {
62         ASSERT_WORK_MTX(local);
63
64         if (!timer_pending(&local->work_timer) ||
65             time_before(timeout, local->work_timer.expires))
66                 mod_timer(&local->work_timer, timeout);
67 }
68
69 static void work_free_rcu(struct rcu_head *head)
70 {
71         struct ieee80211_work *wk =
72                 container_of(head, struct ieee80211_work, rcu_head);
73
74         kfree(wk);
75 }
76
77 void free_work(struct ieee80211_work *wk)
78 {
79         call_rcu(&wk->rcu_head, work_free_rcu);
80 }
81
82 static int ieee80211_compatible_rates(const u8 *supp_rates, int supp_rates_len,
83                                       struct ieee80211_supported_band *sband,
84                                       u32 *rates)
85 {
86         int i, j, count;
87         *rates = 0;
88         count = 0;
89         for (i = 0; i < supp_rates_len; i++) {
90                 int rate = (supp_rates[i] & 0x7F) * 5;
91
92                 for (j = 0; j < sband->n_bitrates; j++)
93                         if (sband->bitrates[j].bitrate == rate) {
94                                 *rates |= BIT(j);
95                                 count++;
96                                 break;
97                         }
98         }
99
100         return count;
101 }
102
103 /* frame sending functions */
104
105 static void ieee80211_add_ht_ie(struct sk_buff *skb, const u8 *ht_info_ie,
106                                 struct ieee80211_supported_band *sband,
107                                 struct ieee80211_channel *channel,
108                                 enum ieee80211_smps_mode smps)
109 {
110         struct ieee80211_ht_info *ht_info;
111         u8 *pos;
112         u32 flags = channel->flags;
113         u16 cap = sband->ht_cap.cap;
114         __le16 tmp;
115
116         if (!sband->ht_cap.ht_supported)
117                 return;
118
119         if (!ht_info_ie)
120                 return;
121
122         if (ht_info_ie[1] < sizeof(struct ieee80211_ht_info))
123                 return;
124
125         ht_info = (struct ieee80211_ht_info *)(ht_info_ie + 2);
126
127         /* determine capability flags */
128
129         if (ieee80211_disable_40mhz_24ghz &&
130             sband->band == IEEE80211_BAND_2GHZ) {
131                 cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
132                 cap &= ~IEEE80211_HT_CAP_SGI_40;
133         }
134
135         switch (ht_info->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
136         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
137                 if (flags & IEEE80211_CHAN_NO_HT40PLUS) {
138                         cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
139                         cap &= ~IEEE80211_HT_CAP_SGI_40;
140                 }
141                 break;
142         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
143                 if (flags & IEEE80211_CHAN_NO_HT40MINUS) {
144                         cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
145                         cap &= ~IEEE80211_HT_CAP_SGI_40;
146                 }
147                 break;
148         }
149
150         /* set SM PS mode properly */
151         cap &= ~IEEE80211_HT_CAP_SM_PS;
152         switch (smps) {
153         case IEEE80211_SMPS_AUTOMATIC:
154         case IEEE80211_SMPS_NUM_MODES:
155                 WARN_ON(1);
156         case IEEE80211_SMPS_OFF:
157                 cap |= WLAN_HT_CAP_SM_PS_DISABLED <<
158                         IEEE80211_HT_CAP_SM_PS_SHIFT;
159                 break;
160         case IEEE80211_SMPS_STATIC:
161                 cap |= WLAN_HT_CAP_SM_PS_STATIC <<
162                         IEEE80211_HT_CAP_SM_PS_SHIFT;
163                 break;
164         case IEEE80211_SMPS_DYNAMIC:
165                 cap |= WLAN_HT_CAP_SM_PS_DYNAMIC <<
166                         IEEE80211_HT_CAP_SM_PS_SHIFT;
167                 break;
168         }
169
170         /* reserve and fill IE */
171
172         pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
173         *pos++ = WLAN_EID_HT_CAPABILITY;
174         *pos++ = sizeof(struct ieee80211_ht_cap);
175         memset(pos, 0, sizeof(struct ieee80211_ht_cap));
176
177         /* capability flags */
178         tmp = cpu_to_le16(cap);
179         memcpy(pos, &tmp, sizeof(u16));
180         pos += sizeof(u16);
181
182         /* AMPDU parameters */
183         *pos++ = sband->ht_cap.ampdu_factor |
184                  (sband->ht_cap.ampdu_density <<
185                         IEEE80211_HT_AMPDU_PARM_DENSITY_SHIFT);
186
187         /* MCS set */
188         memcpy(pos, &sband->ht_cap.mcs, sizeof(sband->ht_cap.mcs));
189         pos += sizeof(sband->ht_cap.mcs);
190
191         /* extended capabilities */
192         pos += sizeof(__le16);
193
194         /* BF capabilities */
195         pos += sizeof(__le32);
196
197         /* antenna selection */
198         pos += sizeof(u8);
199 }
200
201 static void ieee80211_send_assoc(struct ieee80211_sub_if_data *sdata,
202                                  struct ieee80211_work *wk)
203 {
204         struct ieee80211_local *local = sdata->local;
205         struct sk_buff *skb;
206         struct ieee80211_mgmt *mgmt;
207         u8 *pos, qos_info;
208         const u8 *ies;
209         size_t offset = 0, noffset;
210         int i, len, count, rates_len, supp_rates_len;
211         u16 capab;
212         struct ieee80211_supported_band *sband;
213         u32 rates = 0;
214
215         sband = local->hw.wiphy->bands[wk->chan->band];
216
217         if (wk->assoc.supp_rates_len) {
218                 /*
219                  * Get all rates supported by the device and the AP as
220                  * some APs don't like getting a superset of their rates
221                  * in the association request (e.g. D-Link DAP 1353 in
222                  * b-only mode)...
223                  */
224                 rates_len = ieee80211_compatible_rates(wk->assoc.supp_rates,
225                                                        wk->assoc.supp_rates_len,
226                                                        sband, &rates);
227         } else {
228                 /*
229                  * In case AP not provide any supported rates information
230                  * before association, we send information element(s) with
231                  * all rates that we support.
232                  */
233                 rates = ~0;
234                 rates_len = sband->n_bitrates;
235         }
236
237         skb = alloc_skb(local->hw.extra_tx_headroom +
238                         sizeof(*mgmt) + /* bit too much but doesn't matter */
239                         2 + wk->assoc.ssid_len + /* SSID */
240                         4 + rates_len + /* (extended) rates */
241                         4 + /* power capability */
242                         2 + 2 * sband->n_channels + /* supported channels */
243                         2 + sizeof(struct ieee80211_ht_cap) + /* HT */
244                         wk->ie_len + /* extra IEs */
245                         9, /* WMM */
246                         GFP_KERNEL);
247         if (!skb) {
248                 printk(KERN_DEBUG "%s: failed to allocate buffer for assoc "
249                        "frame\n", sdata->name);
250                 return;
251         }
252         skb_reserve(skb, local->hw.extra_tx_headroom);
253
254         capab = WLAN_CAPABILITY_ESS;
255
256         if (sband->band == IEEE80211_BAND_2GHZ) {
257                 if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_SLOT_INCAPABLE))
258                         capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
259                 if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_PREAMBLE_INCAPABLE))
260                         capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
261         }
262
263         if (wk->assoc.capability & WLAN_CAPABILITY_PRIVACY)
264                 capab |= WLAN_CAPABILITY_PRIVACY;
265
266         if ((wk->assoc.capability & WLAN_CAPABILITY_SPECTRUM_MGMT) &&
267             (local->hw.flags & IEEE80211_HW_SPECTRUM_MGMT))
268                 capab |= WLAN_CAPABILITY_SPECTRUM_MGMT;
269
270         mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
271         memset(mgmt, 0, 24);
272         memcpy(mgmt->da, wk->filter_ta, ETH_ALEN);
273         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
274         memcpy(mgmt->bssid, wk->filter_ta, ETH_ALEN);
275
276         if (!is_zero_ether_addr(wk->assoc.prev_bssid)) {
277                 skb_put(skb, 10);
278                 mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
279                                                   IEEE80211_STYPE_REASSOC_REQ);
280                 mgmt->u.reassoc_req.capab_info = cpu_to_le16(capab);
281                 mgmt->u.reassoc_req.listen_interval =
282                                 cpu_to_le16(local->hw.conf.listen_interval);
283                 memcpy(mgmt->u.reassoc_req.current_ap, wk->assoc.prev_bssid,
284                        ETH_ALEN);
285         } else {
286                 skb_put(skb, 4);
287                 mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
288                                                   IEEE80211_STYPE_ASSOC_REQ);
289                 mgmt->u.assoc_req.capab_info = cpu_to_le16(capab);
290                 mgmt->u.assoc_req.listen_interval =
291                                 cpu_to_le16(local->hw.conf.listen_interval);
292         }
293
294         /* SSID */
295         ies = pos = skb_put(skb, 2 + wk->assoc.ssid_len);
296         *pos++ = WLAN_EID_SSID;
297         *pos++ = wk->assoc.ssid_len;
298         memcpy(pos, wk->assoc.ssid, wk->assoc.ssid_len);
299
300         /* add all rates which were marked to be used above */
301         supp_rates_len = rates_len;
302         if (supp_rates_len > 8)
303                 supp_rates_len = 8;
304
305         len = sband->n_bitrates;
306         pos = skb_put(skb, supp_rates_len + 2);
307         *pos++ = WLAN_EID_SUPP_RATES;
308         *pos++ = supp_rates_len;
309
310         count = 0;
311         for (i = 0; i < sband->n_bitrates; i++) {
312                 if (BIT(i) & rates) {
313                         int rate = sband->bitrates[i].bitrate;
314                         *pos++ = (u8) (rate / 5);
315                         if (++count == 8)
316                                 break;
317                 }
318         }
319
320         if (rates_len > count) {
321                 pos = skb_put(skb, rates_len - count + 2);
322                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
323                 *pos++ = rates_len - count;
324
325                 for (i++; i < sband->n_bitrates; i++) {
326                         if (BIT(i) & rates) {
327                                 int rate = sband->bitrates[i].bitrate;
328                                 *pos++ = (u8) (rate / 5);
329                         }
330                 }
331         }
332
333         if (capab & WLAN_CAPABILITY_SPECTRUM_MGMT) {
334                 /* 1. power capabilities */
335                 pos = skb_put(skb, 4);
336                 *pos++ = WLAN_EID_PWR_CAPABILITY;
337                 *pos++ = 2;
338                 *pos++ = 0; /* min tx power */
339                 *pos++ = wk->chan->max_power; /* max tx power */
340
341                 /* 2. supported channels */
342                 /* TODO: get this in reg domain format */
343                 pos = skb_put(skb, 2 * sband->n_channels + 2);
344                 *pos++ = WLAN_EID_SUPPORTED_CHANNELS;
345                 *pos++ = 2 * sband->n_channels;
346                 for (i = 0; i < sband->n_channels; i++) {
347                         *pos++ = ieee80211_frequency_to_channel(
348                                         sband->channels[i].center_freq);
349                         *pos++ = 1; /* one channel in the subband*/
350                 }
351         }
352
353         /* if present, add any custom IEs that go before HT */
354         if (wk->ie_len && wk->ie) {
355                 static const u8 before_ht[] = {
356                         WLAN_EID_SSID,
357                         WLAN_EID_SUPP_RATES,
358                         WLAN_EID_EXT_SUPP_RATES,
359                         WLAN_EID_PWR_CAPABILITY,
360                         WLAN_EID_SUPPORTED_CHANNELS,
361                         WLAN_EID_RSN,
362                         WLAN_EID_QOS_CAPA,
363                         WLAN_EID_RRM_ENABLED_CAPABILITIES,
364                         WLAN_EID_MOBILITY_DOMAIN,
365                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
366                 };
367                 noffset = ieee80211_ie_split(wk->ie, wk->ie_len,
368                                              before_ht, ARRAY_SIZE(before_ht),
369                                              offset);
370                 pos = skb_put(skb, noffset - offset);
371                 memcpy(pos, wk->ie + offset, noffset - offset);
372                 offset = noffset;
373         }
374
375         if (wk->assoc.use_11n && wk->assoc.wmm_used &&
376             local->hw.queues >= 4)
377                 ieee80211_add_ht_ie(skb, wk->assoc.ht_information_ie,
378                                     sband, wk->chan, wk->assoc.smps);
379
380         /* if present, add any custom non-vendor IEs that go after HT */
381         if (wk->ie_len && wk->ie) {
382                 noffset = ieee80211_ie_split_vendor(wk->ie, wk->ie_len,
383                                                     offset);
384                 pos = skb_put(skb, noffset - offset);
385                 memcpy(pos, wk->ie + offset, noffset - offset);
386                 offset = noffset;
387         }
388
389         if (wk->assoc.wmm_used && local->hw.queues >= 4) {
390                 if (wk->assoc.uapsd_used) {
391                         qos_info = local->uapsd_queues;
392                         qos_info |= (local->uapsd_max_sp_len <<
393                                      IEEE80211_WMM_IE_STA_QOSINFO_SP_SHIFT);
394                 } else {
395                         qos_info = 0;
396                 }
397
398                 pos = skb_put(skb, 9);
399                 *pos++ = WLAN_EID_VENDOR_SPECIFIC;
400                 *pos++ = 7; /* len */
401                 *pos++ = 0x00; /* Microsoft OUI 00:50:F2 */
402                 *pos++ = 0x50;
403                 *pos++ = 0xf2;
404                 *pos++ = 2; /* WME */
405                 *pos++ = 0; /* WME info */
406                 *pos++ = 1; /* WME ver */
407                 *pos++ = qos_info;
408         }
409
410         /* add any remaining custom (i.e. vendor specific here) IEs */
411         if (wk->ie_len && wk->ie) {
412                 noffset = wk->ie_len;
413                 pos = skb_put(skb, noffset - offset);
414                 memcpy(pos, wk->ie + offset, noffset - offset);
415         }
416
417         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
418         ieee80211_tx_skb(sdata, skb);
419 }
420
421 static void ieee80211_remove_auth_bss(struct ieee80211_local *local,
422                                       struct ieee80211_work *wk)
423 {
424         struct cfg80211_bss *cbss;
425         u16 capa_val = WLAN_CAPABILITY_ESS;
426
427         if (wk->probe_auth.privacy)
428                 capa_val |= WLAN_CAPABILITY_PRIVACY;
429
430         cbss = cfg80211_get_bss(local->hw.wiphy, wk->chan, wk->filter_ta,
431                                 wk->probe_auth.ssid, wk->probe_auth.ssid_len,
432                                 WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
433                                 capa_val);
434         if (!cbss)
435                 return;
436
437         cfg80211_unlink_bss(local->hw.wiphy, cbss);
438         cfg80211_put_bss(cbss);
439 }
440
441 static enum work_action __must_check
442 ieee80211_direct_probe(struct ieee80211_work *wk)
443 {
444         struct ieee80211_sub_if_data *sdata = wk->sdata;
445         struct ieee80211_local *local = sdata->local;
446
447         wk->probe_auth.tries++;
448         if (wk->probe_auth.tries > IEEE80211_AUTH_MAX_TRIES) {
449                 printk(KERN_DEBUG "%s: direct probe to %pM timed out\n",
450                        sdata->name, wk->filter_ta);
451
452                 /*
453                  * Most likely AP is not in the range so remove the
454                  * bss struct for that AP.
455                  */
456                 ieee80211_remove_auth_bss(local, wk);
457
458                 return WORK_ACT_TIMEOUT;
459         }
460
461         printk(KERN_DEBUG "%s: direct probe to %pM (try %d/%i)\n",
462                sdata->name, wk->filter_ta, wk->probe_auth.tries,
463                IEEE80211_AUTH_MAX_TRIES);
464
465         /*
466          * Direct probe is sent to broadcast address as some APs
467          * will not answer to direct packet in unassociated state.
468          */
469         ieee80211_send_probe_req(sdata, NULL, wk->probe_auth.ssid,
470                                  wk->probe_auth.ssid_len, NULL, 0);
471
472         wk->timeout = jiffies + IEEE80211_AUTH_TIMEOUT;
473         run_again(local, wk->timeout);
474
475         return WORK_ACT_NONE;
476 }
477
478
479 static enum work_action __must_check
480 ieee80211_authenticate(struct ieee80211_work *wk)
481 {
482         struct ieee80211_sub_if_data *sdata = wk->sdata;
483         struct ieee80211_local *local = sdata->local;
484
485         wk->probe_auth.tries++;
486         if (wk->probe_auth.tries > IEEE80211_AUTH_MAX_TRIES) {
487                 printk(KERN_DEBUG "%s: authentication with %pM"
488                        " timed out\n", sdata->name, wk->filter_ta);
489
490                 /*
491                  * Most likely AP is not in the range so remove the
492                  * bss struct for that AP.
493                  */
494                 ieee80211_remove_auth_bss(local, wk);
495
496                 return WORK_ACT_TIMEOUT;
497         }
498
499         printk(KERN_DEBUG "%s: authenticate with %pM (try %d)\n",
500                sdata->name, wk->filter_ta, wk->probe_auth.tries);
501
502         ieee80211_send_auth(sdata, 1, wk->probe_auth.algorithm, wk->ie,
503                             wk->ie_len, wk->filter_ta, NULL, 0, 0);
504         wk->probe_auth.transaction = 2;
505
506         wk->timeout = jiffies + IEEE80211_AUTH_TIMEOUT;
507         run_again(local, wk->timeout);
508
509         return WORK_ACT_NONE;
510 }
511
512 static enum work_action __must_check
513 ieee80211_associate(struct ieee80211_work *wk)
514 {
515         struct ieee80211_sub_if_data *sdata = wk->sdata;
516         struct ieee80211_local *local = sdata->local;
517
518         wk->assoc.tries++;
519         if (wk->assoc.tries > IEEE80211_ASSOC_MAX_TRIES) {
520                 printk(KERN_DEBUG "%s: association with %pM"
521                        " timed out\n",
522                        sdata->name, wk->filter_ta);
523
524                 /*
525                  * Most likely AP is not in the range so remove the
526                  * bss struct for that AP.
527                  */
528                 if (wk->assoc.bss)
529                         cfg80211_unlink_bss(local->hw.wiphy, wk->assoc.bss);
530
531                 return WORK_ACT_TIMEOUT;
532         }
533
534         printk(KERN_DEBUG "%s: associate with %pM (try %d)\n",
535                sdata->name, wk->filter_ta, wk->assoc.tries);
536         ieee80211_send_assoc(sdata, wk);
537
538         wk->timeout = jiffies + IEEE80211_ASSOC_TIMEOUT;
539         run_again(local, wk->timeout);
540
541         return WORK_ACT_NONE;
542 }
543
544 static enum work_action __must_check
545 ieee80211_remain_on_channel_timeout(struct ieee80211_work *wk)
546 {
547         /*
548          * First time we run, do nothing -- the generic code will
549          * have switched to the right channel etc.
550          */
551         if (!wk->started) {
552                 wk->timeout = jiffies + msecs_to_jiffies(wk->remain.duration);
553
554                 cfg80211_ready_on_channel(wk->sdata->dev, (unsigned long) wk,
555                                           wk->chan, wk->chan_type,
556                                           wk->remain.duration, GFP_KERNEL);
557
558                 return WORK_ACT_NONE;
559         }
560
561         return WORK_ACT_TIMEOUT;
562 }
563
564 static enum work_action __must_check
565 ieee80211_offchannel_tx(struct ieee80211_work *wk)
566 {
567         if (!wk->started) {
568                 wk->timeout = jiffies + msecs_to_jiffies(wk->offchan_tx.wait);
569
570                 /*
571                  * After this, offchan_tx.frame remains but now is no
572                  * longer a valid pointer -- we still need it as the
573                  * cookie for canceling this work.
574                  */
575                 ieee80211_tx_skb(wk->sdata, wk->offchan_tx.frame);
576
577                 return WORK_ACT_NONE;
578         }
579
580         return WORK_ACT_TIMEOUT;
581 }
582
583 static enum work_action __must_check
584 ieee80211_assoc_beacon_wait(struct ieee80211_work *wk)
585 {
586         if (wk->started)
587                 return WORK_ACT_TIMEOUT;
588
589         /*
590          * Wait up to one beacon interval ...
591          * should this be more if we miss one?
592          */
593         printk(KERN_DEBUG "%s: waiting for beacon from %pM\n",
594                wk->sdata->name, wk->filter_ta);
595         wk->timeout = TU_TO_EXP_TIME(wk->assoc.bss->beacon_interval);
596         return WORK_ACT_NONE;
597 }
598
599 static void ieee80211_auth_challenge(struct ieee80211_work *wk,
600                                      struct ieee80211_mgmt *mgmt,
601                                      size_t len)
602 {
603         struct ieee80211_sub_if_data *sdata = wk->sdata;
604         u8 *pos;
605         struct ieee802_11_elems elems;
606
607         pos = mgmt->u.auth.variable;
608         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems);
609         if (!elems.challenge)
610                 return;
611         ieee80211_send_auth(sdata, 3, wk->probe_auth.algorithm,
612                             elems.challenge - 2, elems.challenge_len + 2,
613                             wk->filter_ta, wk->probe_auth.key,
614                             wk->probe_auth.key_len, wk->probe_auth.key_idx);
615         wk->probe_auth.transaction = 4;
616 }
617
618 static enum work_action __must_check
619 ieee80211_rx_mgmt_auth(struct ieee80211_work *wk,
620                        struct ieee80211_mgmt *mgmt, size_t len)
621 {
622         u16 auth_alg, auth_transaction, status_code;
623
624         if (wk->type != IEEE80211_WORK_AUTH)
625                 return WORK_ACT_MISMATCH;
626
627         if (len < 24 + 6)
628                 return WORK_ACT_NONE;
629
630         auth_alg = le16_to_cpu(mgmt->u.auth.auth_alg);
631         auth_transaction = le16_to_cpu(mgmt->u.auth.auth_transaction);
632         status_code = le16_to_cpu(mgmt->u.auth.status_code);
633
634         if (auth_alg != wk->probe_auth.algorithm ||
635             auth_transaction != wk->probe_auth.transaction)
636                 return WORK_ACT_NONE;
637
638         if (status_code != WLAN_STATUS_SUCCESS) {
639                 printk(KERN_DEBUG "%s: %pM denied authentication (status %d)\n",
640                        wk->sdata->name, mgmt->sa, status_code);
641                 return WORK_ACT_DONE;
642         }
643
644         switch (wk->probe_auth.algorithm) {
645         case WLAN_AUTH_OPEN:
646         case WLAN_AUTH_LEAP:
647         case WLAN_AUTH_FT:
648                 break;
649         case WLAN_AUTH_SHARED_KEY:
650                 if (wk->probe_auth.transaction != 4) {
651                         ieee80211_auth_challenge(wk, mgmt, len);
652                         /* need another frame */
653                         return WORK_ACT_NONE;
654                 }
655                 break;
656         default:
657                 WARN_ON(1);
658                 return WORK_ACT_NONE;
659         }
660
661         printk(KERN_DEBUG "%s: authenticated\n", wk->sdata->name);
662         return WORK_ACT_DONE;
663 }
664
665 static enum work_action __must_check
666 ieee80211_rx_mgmt_assoc_resp(struct ieee80211_work *wk,
667                              struct ieee80211_mgmt *mgmt, size_t len,
668                              bool reassoc)
669 {
670         struct ieee80211_sub_if_data *sdata = wk->sdata;
671         struct ieee80211_local *local = sdata->local;
672         u16 capab_info, status_code, aid;
673         struct ieee802_11_elems elems;
674         u8 *pos;
675
676         if (wk->type != IEEE80211_WORK_ASSOC)
677                 return WORK_ACT_MISMATCH;
678
679         /*
680          * AssocResp and ReassocResp have identical structure, so process both
681          * of them in this function.
682          */
683
684         if (len < 24 + 6)
685                 return WORK_ACT_NONE;
686
687         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
688         status_code = le16_to_cpu(mgmt->u.assoc_resp.status_code);
689         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
690
691         printk(KERN_DEBUG "%s: RX %sssocResp from %pM (capab=0x%x "
692                "status=%d aid=%d)\n",
693                sdata->name, reassoc ? "Rea" : "A", mgmt->sa,
694                capab_info, status_code, (u16)(aid & ~(BIT(15) | BIT(14))));
695
696         pos = mgmt->u.assoc_resp.variable;
697         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems);
698
699         if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY &&
700             elems.timeout_int && elems.timeout_int_len == 5 &&
701             elems.timeout_int[0] == WLAN_TIMEOUT_ASSOC_COMEBACK) {
702                 u32 tu, ms;
703                 tu = get_unaligned_le32(elems.timeout_int + 1);
704                 ms = tu * 1024 / 1000;
705                 printk(KERN_DEBUG "%s: %pM rejected association temporarily; "
706                        "comeback duration %u TU (%u ms)\n",
707                        sdata->name, mgmt->sa, tu, ms);
708                 wk->timeout = jiffies + msecs_to_jiffies(ms);
709                 if (ms > IEEE80211_ASSOC_TIMEOUT)
710                         run_again(local, wk->timeout);
711                 return WORK_ACT_NONE;
712         }
713
714         if (status_code != WLAN_STATUS_SUCCESS)
715                 printk(KERN_DEBUG "%s: %pM denied association (code=%d)\n",
716                        sdata->name, mgmt->sa, status_code);
717         else
718                 printk(KERN_DEBUG "%s: associated\n", sdata->name);
719
720         return WORK_ACT_DONE;
721 }
722
723 static enum work_action __must_check
724 ieee80211_rx_mgmt_probe_resp(struct ieee80211_work *wk,
725                              struct ieee80211_mgmt *mgmt, size_t len,
726                              struct ieee80211_rx_status *rx_status)
727 {
728         struct ieee80211_sub_if_data *sdata = wk->sdata;
729         struct ieee80211_local *local = sdata->local;
730         size_t baselen;
731
732         ASSERT_WORK_MTX(local);
733
734         if (wk->type != IEEE80211_WORK_DIRECT_PROBE)
735                 return WORK_ACT_MISMATCH;
736
737         if (len < 24 + 12)
738                 return WORK_ACT_NONE;
739
740         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
741         if (baselen > len)
742                 return WORK_ACT_NONE;
743
744         printk(KERN_DEBUG "%s: direct probe responded\n", sdata->name);
745         return WORK_ACT_DONE;
746 }
747
748 static enum work_action __must_check
749 ieee80211_rx_mgmt_beacon(struct ieee80211_work *wk,
750                          struct ieee80211_mgmt *mgmt, size_t len)
751 {
752         struct ieee80211_sub_if_data *sdata = wk->sdata;
753         struct ieee80211_local *local = sdata->local;
754
755         ASSERT_WORK_MTX(local);
756
757         if (wk->type != IEEE80211_WORK_ASSOC_BEACON_WAIT)
758                 return WORK_ACT_MISMATCH;
759
760         if (len < 24 + 12)
761                 return WORK_ACT_NONE;
762
763         printk(KERN_DEBUG "%s: beacon received\n", sdata->name);
764         return WORK_ACT_DONE;
765 }
766
767 static void ieee80211_work_rx_queued_mgmt(struct ieee80211_local *local,
768                                           struct sk_buff *skb)
769 {
770         struct ieee80211_rx_status *rx_status;
771         struct ieee80211_mgmt *mgmt;
772         struct ieee80211_work *wk;
773         enum work_action rma = WORK_ACT_NONE;
774         u16 fc;
775
776         rx_status = (struct ieee80211_rx_status *) skb->cb;
777         mgmt = (struct ieee80211_mgmt *) skb->data;
778         fc = le16_to_cpu(mgmt->frame_control);
779
780         mutex_lock(&local->mtx);
781
782         list_for_each_entry(wk, &local->work_list, list) {
783                 const u8 *bssid = NULL;
784
785                 switch (wk->type) {
786                 case IEEE80211_WORK_DIRECT_PROBE:
787                 case IEEE80211_WORK_AUTH:
788                 case IEEE80211_WORK_ASSOC:
789                 case IEEE80211_WORK_ASSOC_BEACON_WAIT:
790                         bssid = wk->filter_ta;
791                         break;
792                 default:
793                         continue;
794                 }
795
796                 /*
797                  * Before queuing, we already verified mgmt->sa,
798                  * so this is needed just for matching.
799                  */
800                 if (compare_ether_addr(bssid, mgmt->bssid))
801                         continue;
802
803                 switch (fc & IEEE80211_FCTL_STYPE) {
804                 case IEEE80211_STYPE_BEACON:
805                         rma = ieee80211_rx_mgmt_beacon(wk, mgmt, skb->len);
806                         break;
807                 case IEEE80211_STYPE_PROBE_RESP:
808                         rma = ieee80211_rx_mgmt_probe_resp(wk, mgmt, skb->len,
809                                                            rx_status);
810                         break;
811                 case IEEE80211_STYPE_AUTH:
812                         rma = ieee80211_rx_mgmt_auth(wk, mgmt, skb->len);
813                         break;
814                 case IEEE80211_STYPE_ASSOC_RESP:
815                         rma = ieee80211_rx_mgmt_assoc_resp(wk, mgmt,
816                                                            skb->len, false);
817                         break;
818                 case IEEE80211_STYPE_REASSOC_RESP:
819                         rma = ieee80211_rx_mgmt_assoc_resp(wk, mgmt,
820                                                            skb->len, true);
821                         break;
822                 default:
823                         WARN_ON(1);
824                         rma = WORK_ACT_NONE;
825                 }
826
827                 /*
828                  * We've either received an unexpected frame, or we have
829                  * multiple work items and need to match the frame to the
830                  * right one.
831                  */
832                 if (rma == WORK_ACT_MISMATCH)
833                         continue;
834
835                 /*
836                  * We've processed this frame for that work, so it can't
837                  * belong to another work struct.
838                  * NB: this is also required for correctness for 'rma'!
839                  */
840                 break;
841         }
842
843         switch (rma) {
844         case WORK_ACT_MISMATCH:
845                 /* ignore this unmatched frame */
846                 break;
847         case WORK_ACT_NONE:
848                 break;
849         case WORK_ACT_DONE:
850                 list_del_rcu(&wk->list);
851                 break;
852         default:
853                 WARN(1, "unexpected: %d", rma);
854         }
855
856         mutex_unlock(&local->mtx);
857
858         if (rma != WORK_ACT_DONE)
859                 goto out;
860
861         switch (wk->done(wk, skb)) {
862         case WORK_DONE_DESTROY:
863                 free_work(wk);
864                 break;
865         case WORK_DONE_REQUEUE:
866                 synchronize_rcu();
867                 wk->started = false; /* restart */
868                 mutex_lock(&local->mtx);
869                 list_add_tail(&wk->list, &local->work_list);
870                 mutex_unlock(&local->mtx);
871         }
872
873  out:
874         kfree_skb(skb);
875 }
876
877 static void ieee80211_work_timer(unsigned long data)
878 {
879         struct ieee80211_local *local = (void *) data;
880
881         if (local->quiescing)
882                 return;
883
884         ieee80211_queue_work(&local->hw, &local->work_work);
885 }
886
887 static void ieee80211_work_work(struct work_struct *work)
888 {
889         struct ieee80211_local *local =
890                 container_of(work, struct ieee80211_local, work_work);
891         struct sk_buff *skb;
892         struct ieee80211_work *wk, *tmp;
893         LIST_HEAD(free_work);
894         enum work_action rma;
895         bool remain_off_channel = false;
896
897         if (local->scanning)
898                 return;
899
900         /*
901          * ieee80211_queue_work() should have picked up most cases,
902          * here we'll pick the rest.
903          */
904         if (WARN(local->suspended, "work scheduled while going to suspend\n"))
905                 return;
906
907         /* first process frames to avoid timing out while a frame is pending */
908         while ((skb = skb_dequeue(&local->work_skb_queue)))
909                 ieee80211_work_rx_queued_mgmt(local, skb);
910
911         mutex_lock(&local->mtx);
912
913         ieee80211_recalc_idle(local);
914
915         list_for_each_entry_safe(wk, tmp, &local->work_list, list) {
916                 bool started = wk->started;
917
918                 /* mark work as started if it's on the current off-channel */
919                 if (!started && local->tmp_channel &&
920                     wk->chan == local->tmp_channel &&
921                     wk->chan_type == local->tmp_channel_type) {
922                         started = true;
923                         wk->timeout = jiffies;
924                 }
925
926                 if (!started && !local->tmp_channel) {
927                         bool on_oper_chan;
928                         bool tmp_chan_changed = false;
929                         bool on_oper_chan2;
930                         on_oper_chan = ieee80211_cfg_on_oper_channel(local);
931                         if (local->tmp_channel)
932                                 if ((local->tmp_channel != wk->chan) ||
933                                     (local->tmp_channel_type != wk->chan_type))
934                                         tmp_chan_changed = true;
935
936                         local->tmp_channel = wk->chan;
937                         local->tmp_channel_type = wk->chan_type;
938                         /*
939                          * Leave the station vifs in awake mode if they
940                          * happen to be on the same channel as
941                          * the requested channel.
942                          */
943                         on_oper_chan2 = ieee80211_cfg_on_oper_channel(local);
944                         if (on_oper_chan != on_oper_chan2) {
945                                 if (on_oper_chan2) {
946                                         /* going off oper channel, PS too */
947                                         ieee80211_offchannel_stop_vifs(local,
948                                                                        true);
949                                         ieee80211_hw_config(local, 0);
950                                 } else {
951                                         /* going on channel, but leave PS
952                                          * off-channel. */
953                                         ieee80211_hw_config(local, 0);
954                                         ieee80211_offchannel_return(local,
955                                                                     true,
956                                                                     false);
957                                 }
958                         } else if (tmp_chan_changed)
959                                 /* Still off-channel, but on some other
960                                  * channel, so update hardware.
961                                  * PS should already be off-channel.
962                                  */
963                                 ieee80211_hw_config(local, 0);
964
965                         started = true;
966                         wk->timeout = jiffies;
967                 }
968
969                 /* don't try to work with items that aren't started */
970                 if (!started)
971                         continue;
972
973                 if (time_is_after_jiffies(wk->timeout)) {
974                         /*
975                          * This work item isn't supposed to be worked on
976                          * right now, but take care to adjust the timer
977                          * properly.
978                          */
979                         run_again(local, wk->timeout);
980                         continue;
981                 }
982
983                 switch (wk->type) {
984                 default:
985                         WARN_ON(1);
986                         /* nothing */
987                         rma = WORK_ACT_NONE;
988                         break;
989                 case IEEE80211_WORK_ABORT:
990                         rma = WORK_ACT_TIMEOUT;
991                         break;
992                 case IEEE80211_WORK_DIRECT_PROBE:
993                         rma = ieee80211_direct_probe(wk);
994                         break;
995                 case IEEE80211_WORK_AUTH:
996                         rma = ieee80211_authenticate(wk);
997                         break;
998                 case IEEE80211_WORK_ASSOC:
999                         rma = ieee80211_associate(wk);
1000                         break;
1001                 case IEEE80211_WORK_REMAIN_ON_CHANNEL:
1002                         rma = ieee80211_remain_on_channel_timeout(wk);
1003                         break;
1004                 case IEEE80211_WORK_OFFCHANNEL_TX:
1005                         rma = ieee80211_offchannel_tx(wk);
1006                         break;
1007                 case IEEE80211_WORK_ASSOC_BEACON_WAIT:
1008                         rma = ieee80211_assoc_beacon_wait(wk);
1009                         break;
1010                 }
1011
1012                 wk->started = started;
1013
1014                 switch (rma) {
1015                 case WORK_ACT_NONE:
1016                         /* might have changed the timeout */
1017                         run_again(local, wk->timeout);
1018                         break;
1019                 case WORK_ACT_TIMEOUT:
1020                         list_del_rcu(&wk->list);
1021                         synchronize_rcu();
1022                         list_add(&wk->list, &free_work);
1023                         break;
1024                 default:
1025                         WARN(1, "unexpected: %d", rma);
1026                 }
1027         }
1028
1029         list_for_each_entry(wk, &local->work_list, list) {
1030                 if (!wk->started)
1031                         continue;
1032                 if (wk->chan != local->tmp_channel)
1033                         continue;
1034                 if (wk->chan_type != local->tmp_channel_type)
1035                         continue;
1036                 remain_off_channel = true;
1037         }
1038
1039         if (!remain_off_channel && local->tmp_channel) {
1040                 bool on_oper_chan = ieee80211_cfg_on_oper_channel(local);
1041                 local->tmp_channel = NULL;
1042                 /* If tmp_channel wasn't operating channel, then
1043                  * we need to go back on-channel.
1044                  * NOTE:  If we can ever be here while scannning,
1045                  * or if the hw_config() channel config logic changes,
1046                  * then we may need to do a more thorough check to see if
1047                  * we still need to do a hardware config.  Currently,
1048                  * we cannot be here while scanning, however.
1049                  */
1050                 if (ieee80211_cfg_on_oper_channel(local) && !on_oper_chan)
1051                         ieee80211_hw_config(local, 0);
1052
1053                 /* At the least, we need to disable offchannel_ps,
1054                  * so just go ahead and run the entire offchannel
1055                  * return logic here.  We *could* skip enabling
1056                  * beaconing if we were already on-oper-channel
1057                  * as a future optimization.
1058                  */
1059                 ieee80211_offchannel_return(local, true, true);
1060
1061                 /* give connection some time to breathe */
1062                 run_again(local, jiffies + HZ/2);
1063         }
1064
1065         if (list_empty(&local->work_list) && local->scan_req &&
1066             !local->scanning)
1067                 ieee80211_queue_delayed_work(&local->hw,
1068                                              &local->scan_work,
1069                                              round_jiffies_relative(0));
1070
1071         ieee80211_recalc_idle(local);
1072
1073         mutex_unlock(&local->mtx);
1074
1075         list_for_each_entry_safe(wk, tmp, &free_work, list) {
1076                 wk->done(wk, NULL);
1077                 list_del(&wk->list);
1078                 kfree(wk);
1079         }
1080 }
1081
1082 void ieee80211_add_work(struct ieee80211_work *wk)
1083 {
1084         struct ieee80211_local *local;
1085
1086         if (WARN_ON(!wk->chan))
1087                 return;
1088
1089         if (WARN_ON(!wk->sdata))
1090                 return;
1091
1092         if (WARN_ON(!wk->done))
1093                 return;
1094
1095         if (WARN_ON(!ieee80211_sdata_running(wk->sdata)))
1096                 return;
1097
1098         wk->started = false;
1099
1100         local = wk->sdata->local;
1101         mutex_lock(&local->mtx);
1102         list_add_tail(&wk->list, &local->work_list);
1103         mutex_unlock(&local->mtx);
1104
1105         ieee80211_queue_work(&local->hw, &local->work_work);
1106 }
1107
1108 void ieee80211_work_init(struct ieee80211_local *local)
1109 {
1110         INIT_LIST_HEAD(&local->work_list);
1111         setup_timer(&local->work_timer, ieee80211_work_timer,
1112                     (unsigned long)local);
1113         INIT_WORK(&local->work_work, ieee80211_work_work);
1114         skb_queue_head_init(&local->work_skb_queue);
1115 }
1116
1117 void ieee80211_work_purge(struct ieee80211_sub_if_data *sdata)
1118 {
1119         struct ieee80211_local *local = sdata->local;
1120         struct ieee80211_work *wk;
1121         bool cleanup = false;
1122
1123         mutex_lock(&local->mtx);
1124         list_for_each_entry(wk, &local->work_list, list) {
1125                 if (wk->sdata != sdata)
1126                         continue;
1127                 cleanup = true;
1128                 wk->type = IEEE80211_WORK_ABORT;
1129                 wk->started = true;
1130                 wk->timeout = jiffies;
1131         }
1132         mutex_unlock(&local->mtx);
1133
1134         /* run cleanups etc. */
1135         if (cleanup)
1136                 ieee80211_work_work(&local->work_work);
1137
1138         mutex_lock(&local->mtx);
1139         list_for_each_entry(wk, &local->work_list, list) {
1140                 if (wk->sdata != sdata)
1141                         continue;
1142                 WARN_ON(1);
1143                 break;
1144         }
1145         mutex_unlock(&local->mtx);
1146 }
1147
1148 ieee80211_rx_result ieee80211_work_rx_mgmt(struct ieee80211_sub_if_data *sdata,
1149                                            struct sk_buff *skb)
1150 {
1151         struct ieee80211_local *local = sdata->local;
1152         struct ieee80211_mgmt *mgmt;
1153         struct ieee80211_work *wk;
1154         u16 fc;
1155
1156         if (skb->len < 24)
1157                 return RX_DROP_MONITOR;
1158
1159         mgmt = (struct ieee80211_mgmt *) skb->data;
1160         fc = le16_to_cpu(mgmt->frame_control);
1161
1162         list_for_each_entry_rcu(wk, &local->work_list, list) {
1163                 if (sdata != wk->sdata)
1164                         continue;
1165                 if (compare_ether_addr(wk->filter_ta, mgmt->sa))
1166                         continue;
1167                 if (compare_ether_addr(wk->filter_ta, mgmt->bssid))
1168                         continue;
1169
1170                 switch (fc & IEEE80211_FCTL_STYPE) {
1171                 case IEEE80211_STYPE_AUTH:
1172                 case IEEE80211_STYPE_PROBE_RESP:
1173                 case IEEE80211_STYPE_ASSOC_RESP:
1174                 case IEEE80211_STYPE_REASSOC_RESP:
1175                 case IEEE80211_STYPE_BEACON:
1176                         skb_queue_tail(&local->work_skb_queue, skb);
1177                         ieee80211_queue_work(&local->hw, &local->work_work);
1178                         return RX_QUEUED;
1179                 }
1180         }
1181
1182         return RX_CONTINUE;
1183 }
1184
1185 static enum work_done_result ieee80211_remain_done(struct ieee80211_work *wk,
1186                                                    struct sk_buff *skb)
1187 {
1188         /*
1189          * We are done serving the remain-on-channel command.
1190          */
1191         cfg80211_remain_on_channel_expired(wk->sdata->dev, (unsigned long) wk,
1192                                            wk->chan, wk->chan_type,
1193                                            GFP_KERNEL);
1194
1195         return WORK_DONE_DESTROY;
1196 }
1197
1198 int ieee80211_wk_remain_on_channel(struct ieee80211_sub_if_data *sdata,
1199                                    struct ieee80211_channel *chan,
1200                                    enum nl80211_channel_type channel_type,
1201                                    unsigned int duration, u64 *cookie)
1202 {
1203         struct ieee80211_work *wk;
1204
1205         wk = kzalloc(sizeof(*wk), GFP_KERNEL);
1206         if (!wk)
1207                 return -ENOMEM;
1208
1209         wk->type = IEEE80211_WORK_REMAIN_ON_CHANNEL;
1210         wk->chan = chan;
1211         wk->chan_type = channel_type;
1212         wk->sdata = sdata;
1213         wk->done = ieee80211_remain_done;
1214
1215         wk->remain.duration = duration;
1216
1217         *cookie = (unsigned long) wk;
1218
1219         ieee80211_add_work(wk);
1220
1221         return 0;
1222 }
1223
1224 int ieee80211_wk_cancel_remain_on_channel(struct ieee80211_sub_if_data *sdata,
1225                                           u64 cookie)
1226 {
1227         struct ieee80211_local *local = sdata->local;
1228         struct ieee80211_work *wk, *tmp;
1229         bool found = false;
1230
1231         mutex_lock(&local->mtx);
1232         list_for_each_entry_safe(wk, tmp, &local->work_list, list) {
1233                 if ((unsigned long) wk == cookie) {
1234                         wk->timeout = jiffies;
1235                         found = true;
1236                         break;
1237                 }
1238         }
1239         mutex_unlock(&local->mtx);
1240
1241         if (!found)
1242                 return -ENOENT;
1243
1244         ieee80211_queue_work(&local->hw, &local->work_work);
1245
1246         return 0;
1247 }