Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net-2.6
[linux-2.6.git] / net / mac80211 / mlme.c
1 /*
2  * BSS client mode implementation
3  * Copyright 2003-2008, Jouni Malinen <j@w1.fi>
4  * Copyright 2004, Instant802 Networks, Inc.
5  * Copyright 2005, Devicescape Software, Inc.
6  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
7  * Copyright 2007, Michael Wu <flamingice@sourmilk.net>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License version 2 as
11  * published by the Free Software Foundation.
12  */
13
14 #include <linux/delay.h>
15 #include <linux/if_ether.h>
16 #include <linux/skbuff.h>
17 #include <linux/if_arp.h>
18 #include <linux/etherdevice.h>
19 #include <linux/rtnetlink.h>
20 #include <linux/pm_qos_params.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 "driver-ops.h"
28 #include "rate.h"
29 #include "led.h"
30
31 static int max_nullfunc_tries = 2;
32 module_param(max_nullfunc_tries, int, 0644);
33 MODULE_PARM_DESC(max_nullfunc_tries,
34                  "Maximum nullfunc tx tries before disconnecting (reason 4).");
35
36 static int max_probe_tries = 5;
37 module_param(max_probe_tries, int, 0644);
38 MODULE_PARM_DESC(max_probe_tries,
39                  "Maximum probe tries before disconnecting (reason 4).");
40
41 /*
42  * Beacon loss timeout is calculated as N frames times the
43  * advertised beacon interval.  This may need to be somewhat
44  * higher than what hardware might detect to account for
45  * delays in the host processing frames. But since we also
46  * probe on beacon miss before declaring the connection lost
47  * default to what we want.
48  */
49 #define IEEE80211_BEACON_LOSS_COUNT     7
50
51 /*
52  * Time the connection can be idle before we probe
53  * it to see if we can still talk to the AP.
54  */
55 #define IEEE80211_CONNECTION_IDLE_TIME  (30 * HZ)
56 /*
57  * Time we wait for a probe response after sending
58  * a probe request because of beacon loss or for
59  * checking the connection still works.
60  */
61 static int probe_wait_ms = 500;
62 module_param(probe_wait_ms, int, 0644);
63 MODULE_PARM_DESC(probe_wait_ms,
64                  "Maximum time(ms) to wait for probe response"
65                  " before disconnecting (reason 4).");
66
67 /*
68  * Weight given to the latest Beacon frame when calculating average signal
69  * strength for Beacon frames received in the current BSS. This must be
70  * between 1 and 15.
71  */
72 #define IEEE80211_SIGNAL_AVE_WEIGHT     3
73
74 /*
75  * How many Beacon frames need to have been used in average signal strength
76  * before starting to indicate signal change events.
77  */
78 #define IEEE80211_SIGNAL_AVE_MIN_COUNT  4
79
80 #define TMR_RUNNING_TIMER       0
81 #define TMR_RUNNING_CHANSW      1
82
83 /*
84  * All cfg80211 functions have to be called outside a locked
85  * section so that they can acquire a lock themselves... This
86  * is much simpler than queuing up things in cfg80211, but we
87  * do need some indirection for that here.
88  */
89 enum rx_mgmt_action {
90         /* no action required */
91         RX_MGMT_NONE,
92
93         /* caller must call cfg80211_send_deauth() */
94         RX_MGMT_CFG80211_DEAUTH,
95
96         /* caller must call cfg80211_send_disassoc() */
97         RX_MGMT_CFG80211_DISASSOC,
98 };
99
100 /* utils */
101 static inline void ASSERT_MGD_MTX(struct ieee80211_if_managed *ifmgd)
102 {
103         lockdep_assert_held(&ifmgd->mtx);
104 }
105
106 /*
107  * We can have multiple work items (and connection probing)
108  * scheduling this timer, but we need to take care to only
109  * reschedule it when it should fire _earlier_ than it was
110  * asked for before, or if it's not pending right now. This
111  * function ensures that. Note that it then is required to
112  * run this function for all timeouts after the first one
113  * has happened -- the work that runs from this timer will
114  * do that.
115  */
116 static void run_again(struct ieee80211_if_managed *ifmgd,
117                              unsigned long timeout)
118 {
119         ASSERT_MGD_MTX(ifmgd);
120
121         if (!timer_pending(&ifmgd->timer) ||
122             time_before(timeout, ifmgd->timer.expires))
123                 mod_timer(&ifmgd->timer, timeout);
124 }
125
126 void ieee80211_sta_reset_beacon_monitor(struct ieee80211_sub_if_data *sdata)
127 {
128         if (sdata->local->hw.flags & IEEE80211_HW_BEACON_FILTER)
129                 return;
130
131         mod_timer(&sdata->u.mgd.bcn_mon_timer,
132                   round_jiffies_up(jiffies + sdata->u.mgd.beacon_timeout));
133 }
134
135 void ieee80211_sta_reset_conn_monitor(struct ieee80211_sub_if_data *sdata)
136 {
137         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
138
139         if (unlikely(!sdata->u.mgd.associated))
140                 return;
141
142         if (sdata->local->hw.flags & IEEE80211_HW_CONNECTION_MONITOR)
143                 return;
144
145         mod_timer(&sdata->u.mgd.conn_mon_timer,
146                   round_jiffies_up(jiffies + IEEE80211_CONNECTION_IDLE_TIME));
147
148         ifmgd->probe_send_count = 0;
149 }
150
151 static int ecw2cw(int ecw)
152 {
153         return (1 << ecw) - 1;
154 }
155
156 /*
157  * ieee80211_enable_ht should be called only after the operating band
158  * has been determined as ht configuration depends on the hw's
159  * HT abilities for a specific band.
160  */
161 static u32 ieee80211_enable_ht(struct ieee80211_sub_if_data *sdata,
162                                struct ieee80211_ht_info *hti,
163                                const u8 *bssid, u16 ap_ht_cap_flags)
164 {
165         struct ieee80211_local *local = sdata->local;
166         struct ieee80211_supported_band *sband;
167         struct sta_info *sta;
168         u32 changed = 0;
169         int hti_cfreq;
170         u16 ht_opmode;
171         bool enable_ht = true;
172         enum nl80211_channel_type prev_chantype;
173         enum nl80211_channel_type channel_type = NL80211_CHAN_NO_HT;
174
175         sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
176
177         prev_chantype = sdata->vif.bss_conf.channel_type;
178
179         /* HT is not supported */
180         if (!sband->ht_cap.ht_supported)
181                 enable_ht = false;
182
183         if (enable_ht) {
184                 hti_cfreq = ieee80211_channel_to_frequency(hti->control_chan,
185                                                            sband->band);
186                 /* check that channel matches the right operating channel */
187                 if (local->hw.conf.channel->center_freq != hti_cfreq) {
188                         /* Some APs mess this up, evidently.
189                          * Netgear WNDR3700 sometimes reports 4 higher than
190                          * the actual channel, for instance.
191                          */
192                         printk(KERN_DEBUG
193                                "%s: Wrong control channel in association"
194                                " response: configured center-freq: %d"
195                                " hti-cfreq: %d  hti->control_chan: %d"
196                                " band: %d.  Disabling HT.\n",
197                                sdata->name,
198                                local->hw.conf.channel->center_freq,
199                                hti_cfreq, hti->control_chan,
200                                sband->band);
201                         enable_ht = false;
202                 }
203         }
204
205         if (enable_ht) {
206                 channel_type = NL80211_CHAN_HT20;
207
208                 if (!(ap_ht_cap_flags & IEEE80211_HT_CAP_40MHZ_INTOLERANT) &&
209                     (sband->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40) &&
210                     (hti->ht_param & IEEE80211_HT_PARAM_CHAN_WIDTH_ANY)) {
211                         switch(hti->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
212                         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
213                                 if (!(local->hw.conf.channel->flags &
214                                     IEEE80211_CHAN_NO_HT40PLUS))
215                                         channel_type = NL80211_CHAN_HT40PLUS;
216                                 break;
217                         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
218                                 if (!(local->hw.conf.channel->flags &
219                                     IEEE80211_CHAN_NO_HT40MINUS))
220                                         channel_type = NL80211_CHAN_HT40MINUS;
221                                 break;
222                         }
223                 }
224         }
225
226         if (local->tmp_channel)
227                 local->tmp_channel_type = channel_type;
228
229         if (!ieee80211_set_channel_type(local, sdata, channel_type)) {
230                 /* can only fail due to HT40+/- mismatch */
231                 channel_type = NL80211_CHAN_HT20;
232                 WARN_ON(!ieee80211_set_channel_type(local, sdata, channel_type));
233         }
234
235         ieee80211_stop_queues_by_reason(&sdata->local->hw,
236                                         IEEE80211_QUEUE_STOP_REASON_CSA);
237
238         /* channel_type change automatically detected */
239         ieee80211_hw_config(local, 0);
240
241         if (prev_chantype != channel_type) {
242                 rcu_read_lock();
243                 sta = sta_info_get(sdata, bssid);
244                 if (sta)
245                         rate_control_rate_update(local, sband, sta,
246                                                  IEEE80211_RC_HT_CHANGED,
247                                                  channel_type);
248                 rcu_read_unlock();
249         }
250
251         ieee80211_wake_queues_by_reason(&sdata->local->hw,
252                                         IEEE80211_QUEUE_STOP_REASON_CSA);
253
254         ht_opmode = le16_to_cpu(hti->operation_mode);
255
256         /* if bss configuration changed store the new one */
257         if (sdata->ht_opmode_valid != enable_ht ||
258             sdata->vif.bss_conf.ht_operation_mode != ht_opmode ||
259             prev_chantype != channel_type) {
260                 changed |= BSS_CHANGED_HT;
261                 sdata->vif.bss_conf.ht_operation_mode = ht_opmode;
262                 sdata->ht_opmode_valid = enable_ht;
263         }
264
265         return changed;
266 }
267
268 /* frame sending functions */
269
270 static void ieee80211_send_deauth_disassoc(struct ieee80211_sub_if_data *sdata,
271                                            const u8 *bssid, u16 stype, u16 reason,
272                                            void *cookie, bool send_frame)
273 {
274         struct ieee80211_local *local = sdata->local;
275         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
276         struct sk_buff *skb;
277         struct ieee80211_mgmt *mgmt;
278
279         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*mgmt));
280         if (!skb) {
281                 printk(KERN_DEBUG "%s: failed to allocate buffer for "
282                        "deauth/disassoc frame\n", sdata->name);
283                 return;
284         }
285         skb_reserve(skb, local->hw.extra_tx_headroom);
286
287         mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
288         memset(mgmt, 0, 24);
289         memcpy(mgmt->da, bssid, ETH_ALEN);
290         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
291         memcpy(mgmt->bssid, bssid, ETH_ALEN);
292         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT | stype);
293         skb_put(skb, 2);
294         /* u.deauth.reason_code == u.disassoc.reason_code */
295         mgmt->u.deauth.reason_code = cpu_to_le16(reason);
296
297         if (stype == IEEE80211_STYPE_DEAUTH)
298                 if (cookie)
299                         __cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
300                 else
301                         cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
302         else
303                 if (cookie)
304                         __cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
305                 else
306                         cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
307         if (!(ifmgd->flags & IEEE80211_STA_MFP_ENABLED))
308                 IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
309
310         if (send_frame)
311                 ieee80211_tx_skb(sdata, skb);
312         else
313                 kfree_skb(skb);
314 }
315
316 void ieee80211_send_pspoll(struct ieee80211_local *local,
317                            struct ieee80211_sub_if_data *sdata)
318 {
319         struct ieee80211_pspoll *pspoll;
320         struct sk_buff *skb;
321
322         skb = ieee80211_pspoll_get(&local->hw, &sdata->vif);
323         if (!skb)
324                 return;
325
326         pspoll = (struct ieee80211_pspoll *) skb->data;
327         pspoll->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
328
329         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
330         ieee80211_tx_skb(sdata, skb);
331 }
332
333 void ieee80211_send_nullfunc(struct ieee80211_local *local,
334                              struct ieee80211_sub_if_data *sdata,
335                              int powersave)
336 {
337         struct sk_buff *skb;
338         struct ieee80211_hdr_3addr *nullfunc;
339
340         skb = ieee80211_nullfunc_get(&local->hw, &sdata->vif);
341         if (!skb)
342                 return;
343
344         nullfunc = (struct ieee80211_hdr_3addr *) skb->data;
345         if (powersave)
346                 nullfunc->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
347
348         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
349         ieee80211_tx_skb(sdata, skb);
350 }
351
352 static void ieee80211_send_4addr_nullfunc(struct ieee80211_local *local,
353                                           struct ieee80211_sub_if_data *sdata)
354 {
355         struct sk_buff *skb;
356         struct ieee80211_hdr *nullfunc;
357         __le16 fc;
358
359         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
360                 return;
361
362         skb = dev_alloc_skb(local->hw.extra_tx_headroom + 30);
363         if (!skb) {
364                 printk(KERN_DEBUG "%s: failed to allocate buffer for 4addr "
365                        "nullfunc frame\n", sdata->name);
366                 return;
367         }
368         skb_reserve(skb, local->hw.extra_tx_headroom);
369
370         nullfunc = (struct ieee80211_hdr *) skb_put(skb, 30);
371         memset(nullfunc, 0, 30);
372         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_NULLFUNC |
373                          IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
374         nullfunc->frame_control = fc;
375         memcpy(nullfunc->addr1, sdata->u.mgd.bssid, ETH_ALEN);
376         memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
377         memcpy(nullfunc->addr3, sdata->u.mgd.bssid, ETH_ALEN);
378         memcpy(nullfunc->addr4, sdata->vif.addr, ETH_ALEN);
379
380         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
381         ieee80211_tx_skb(sdata, skb);
382 }
383
384 /* spectrum management related things */
385 static void ieee80211_chswitch_work(struct work_struct *work)
386 {
387         struct ieee80211_sub_if_data *sdata =
388                 container_of(work, struct ieee80211_sub_if_data, u.mgd.chswitch_work);
389         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
390
391         if (!ieee80211_sdata_running(sdata))
392                 return;
393
394         mutex_lock(&ifmgd->mtx);
395         if (!ifmgd->associated)
396                 goto out;
397
398         sdata->local->oper_channel = sdata->local->csa_channel;
399         if (!sdata->local->ops->channel_switch) {
400                 /* call "hw_config" only if doing sw channel switch */
401                 ieee80211_hw_config(sdata->local,
402                         IEEE80211_CONF_CHANGE_CHANNEL);
403         }
404
405         /* XXX: shouldn't really modify cfg80211-owned data! */
406         ifmgd->associated->channel = sdata->local->oper_channel;
407
408         ieee80211_wake_queues_by_reason(&sdata->local->hw,
409                                         IEEE80211_QUEUE_STOP_REASON_CSA);
410  out:
411         ifmgd->flags &= ~IEEE80211_STA_CSA_RECEIVED;
412         mutex_unlock(&ifmgd->mtx);
413 }
414
415 void ieee80211_chswitch_done(struct ieee80211_vif *vif, bool success)
416 {
417         struct ieee80211_sub_if_data *sdata;
418         struct ieee80211_if_managed *ifmgd;
419
420         sdata = vif_to_sdata(vif);
421         ifmgd = &sdata->u.mgd;
422
423         trace_api_chswitch_done(sdata, success);
424         if (!success) {
425                 /*
426                  * If the channel switch was not successful, stay
427                  * around on the old channel. We currently lack
428                  * good handling of this situation, possibly we
429                  * should just drop the association.
430                  */
431                 sdata->local->csa_channel = sdata->local->oper_channel;
432         }
433
434         ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
435 }
436 EXPORT_SYMBOL(ieee80211_chswitch_done);
437
438 static void ieee80211_chswitch_timer(unsigned long data)
439 {
440         struct ieee80211_sub_if_data *sdata =
441                 (struct ieee80211_sub_if_data *) data;
442         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
443
444         if (sdata->local->quiescing) {
445                 set_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running);
446                 return;
447         }
448
449         ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
450 }
451
452 void ieee80211_sta_process_chanswitch(struct ieee80211_sub_if_data *sdata,
453                                       struct ieee80211_channel_sw_ie *sw_elem,
454                                       struct ieee80211_bss *bss,
455                                       u64 timestamp)
456 {
457         struct cfg80211_bss *cbss =
458                 container_of((void *)bss, struct cfg80211_bss, priv);
459         struct ieee80211_channel *new_ch;
460         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
461         int new_freq = ieee80211_channel_to_frequency(sw_elem->new_ch_num,
462                                                       cbss->channel->band);
463
464         ASSERT_MGD_MTX(ifmgd);
465
466         if (!ifmgd->associated)
467                 return;
468
469         if (sdata->local->scanning)
470                 return;
471
472         /* Disregard subsequent beacons if we are already running a timer
473            processing a CSA */
474
475         if (ifmgd->flags & IEEE80211_STA_CSA_RECEIVED)
476                 return;
477
478         new_ch = ieee80211_get_channel(sdata->local->hw.wiphy, new_freq);
479         if (!new_ch || new_ch->flags & IEEE80211_CHAN_DISABLED)
480                 return;
481
482         sdata->local->csa_channel = new_ch;
483
484         if (sdata->local->ops->channel_switch) {
485                 /* use driver's channel switch callback */
486                 struct ieee80211_channel_switch ch_switch;
487                 memset(&ch_switch, 0, sizeof(ch_switch));
488                 ch_switch.timestamp = timestamp;
489                 if (sw_elem->mode) {
490                         ch_switch.block_tx = true;
491                         ieee80211_stop_queues_by_reason(&sdata->local->hw,
492                                         IEEE80211_QUEUE_STOP_REASON_CSA);
493                 }
494                 ch_switch.channel = new_ch;
495                 ch_switch.count = sw_elem->count;
496                 ifmgd->flags |= IEEE80211_STA_CSA_RECEIVED;
497                 drv_channel_switch(sdata->local, &ch_switch);
498                 return;
499         }
500
501         /* channel switch handled in software */
502         if (sw_elem->count <= 1) {
503                 ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
504         } else {
505                 if (sw_elem->mode)
506                         ieee80211_stop_queues_by_reason(&sdata->local->hw,
507                                         IEEE80211_QUEUE_STOP_REASON_CSA);
508                 ifmgd->flags |= IEEE80211_STA_CSA_RECEIVED;
509                 mod_timer(&ifmgd->chswitch_timer,
510                           jiffies +
511                           msecs_to_jiffies(sw_elem->count *
512                                            cbss->beacon_interval));
513         }
514 }
515
516 static void ieee80211_handle_pwr_constr(struct ieee80211_sub_if_data *sdata,
517                                         u16 capab_info, u8 *pwr_constr_elem,
518                                         u8 pwr_constr_elem_len)
519 {
520         struct ieee80211_conf *conf = &sdata->local->hw.conf;
521
522         if (!(capab_info & WLAN_CAPABILITY_SPECTRUM_MGMT))
523                 return;
524
525         /* Power constraint IE length should be 1 octet */
526         if (pwr_constr_elem_len != 1)
527                 return;
528
529         if ((*pwr_constr_elem <= conf->channel->max_power) &&
530             (*pwr_constr_elem != sdata->local->power_constr_level)) {
531                 sdata->local->power_constr_level = *pwr_constr_elem;
532                 ieee80211_hw_config(sdata->local, 0);
533         }
534 }
535
536 void ieee80211_enable_dyn_ps(struct ieee80211_vif *vif)
537 {
538         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
539         struct ieee80211_local *local = sdata->local;
540         struct ieee80211_conf *conf = &local->hw.conf;
541
542         WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION ||
543                 !(local->hw.flags & IEEE80211_HW_SUPPORTS_PS) ||
544                 (local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS));
545
546         local->disable_dynamic_ps = false;
547         conf->dynamic_ps_timeout = local->dynamic_ps_user_timeout;
548 }
549 EXPORT_SYMBOL(ieee80211_enable_dyn_ps);
550
551 void ieee80211_disable_dyn_ps(struct ieee80211_vif *vif)
552 {
553         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
554         struct ieee80211_local *local = sdata->local;
555         struct ieee80211_conf *conf = &local->hw.conf;
556
557         WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION ||
558                 !(local->hw.flags & IEEE80211_HW_SUPPORTS_PS) ||
559                 (local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS));
560
561         local->disable_dynamic_ps = true;
562         conf->dynamic_ps_timeout = 0;
563         del_timer_sync(&local->dynamic_ps_timer);
564         ieee80211_queue_work(&local->hw,
565                              &local->dynamic_ps_enable_work);
566 }
567 EXPORT_SYMBOL(ieee80211_disable_dyn_ps);
568
569 /* powersave */
570 static void ieee80211_enable_ps(struct ieee80211_local *local,
571                                 struct ieee80211_sub_if_data *sdata)
572 {
573         struct ieee80211_conf *conf = &local->hw.conf;
574
575         /*
576          * If we are scanning right now then the parameters will
577          * take effect when scan finishes.
578          */
579         if (local->scanning)
580                 return;
581
582         if (conf->dynamic_ps_timeout > 0 &&
583             !(local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS)) {
584                 mod_timer(&local->dynamic_ps_timer, jiffies +
585                           msecs_to_jiffies(conf->dynamic_ps_timeout));
586         } else {
587                 if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
588                         ieee80211_send_nullfunc(local, sdata, 1);
589
590                 if ((local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK) &&
591                     (local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS))
592                         return;
593
594                 conf->flags |= IEEE80211_CONF_PS;
595                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
596         }
597 }
598
599 static void ieee80211_change_ps(struct ieee80211_local *local)
600 {
601         struct ieee80211_conf *conf = &local->hw.conf;
602
603         if (local->ps_sdata) {
604                 ieee80211_enable_ps(local, local->ps_sdata);
605         } else if (conf->flags & IEEE80211_CONF_PS) {
606                 conf->flags &= ~IEEE80211_CONF_PS;
607                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
608                 del_timer_sync(&local->dynamic_ps_timer);
609                 cancel_work_sync(&local->dynamic_ps_enable_work);
610         }
611 }
612
613 static bool ieee80211_powersave_allowed(struct ieee80211_sub_if_data *sdata)
614 {
615         struct ieee80211_if_managed *mgd = &sdata->u.mgd;
616         struct sta_info *sta = NULL;
617         u32 sta_flags = 0;
618
619         if (!mgd->powersave)
620                 return false;
621
622         if (!mgd->associated)
623                 return false;
624
625         if (!mgd->associated->beacon_ies)
626                 return false;
627
628         if (mgd->flags & (IEEE80211_STA_BEACON_POLL |
629                           IEEE80211_STA_CONNECTION_POLL))
630                 return false;
631
632         rcu_read_lock();
633         sta = sta_info_get(sdata, mgd->bssid);
634         if (sta)
635                 sta_flags = get_sta_flags(sta);
636         rcu_read_unlock();
637
638         if (!(sta_flags & WLAN_STA_AUTHORIZED))
639                 return false;
640
641         return true;
642 }
643
644 /* need to hold RTNL or interface lock */
645 void ieee80211_recalc_ps(struct ieee80211_local *local, s32 latency)
646 {
647         struct ieee80211_sub_if_data *sdata, *found = NULL;
648         int count = 0;
649         int timeout;
650
651         if (!(local->hw.flags & IEEE80211_HW_SUPPORTS_PS)) {
652                 local->ps_sdata = NULL;
653                 return;
654         }
655
656         if (!list_empty(&local->work_list)) {
657                 local->ps_sdata = NULL;
658                 goto change;
659         }
660
661         list_for_each_entry(sdata, &local->interfaces, list) {
662                 if (!ieee80211_sdata_running(sdata))
663                         continue;
664                 if (sdata->vif.type == NL80211_IFTYPE_AP) {
665                         /* If an AP vif is found, then disable PS
666                          * by setting the count to zero thereby setting
667                          * ps_sdata to NULL.
668                          */
669                         count = 0;
670                         break;
671                 }
672                 if (sdata->vif.type != NL80211_IFTYPE_STATION)
673                         continue;
674                 found = sdata;
675                 count++;
676         }
677
678         if (count == 1 && ieee80211_powersave_allowed(found)) {
679                 struct ieee80211_conf *conf = &local->hw.conf;
680                 s32 beaconint_us;
681
682                 if (latency < 0)
683                         latency = pm_qos_request(PM_QOS_NETWORK_LATENCY);
684
685                 beaconint_us = ieee80211_tu_to_usec(
686                                         found->vif.bss_conf.beacon_int);
687
688                 timeout = local->dynamic_ps_forced_timeout;
689                 if (timeout < 0) {
690                         /*
691                          * Go to full PSM if the user configures a very low
692                          * latency requirement.
693                          * The 2000 second value is there for compatibility
694                          * until the PM_QOS_NETWORK_LATENCY is configured
695                          * with real values.
696                          */
697                         if (latency > (1900 * USEC_PER_MSEC) &&
698                             latency != (2000 * USEC_PER_SEC))
699                                 timeout = 0;
700                         else
701                                 timeout = 100;
702                 }
703                 local->dynamic_ps_user_timeout = timeout;
704                 if (!local->disable_dynamic_ps)
705                         conf->dynamic_ps_timeout =
706                                 local->dynamic_ps_user_timeout;
707
708                 if (beaconint_us > latency) {
709                         local->ps_sdata = NULL;
710                 } else {
711                         struct ieee80211_bss *bss;
712                         int maxslp = 1;
713                         u8 dtimper;
714
715                         bss = (void *)found->u.mgd.associated->priv;
716                         dtimper = bss->dtim_period;
717
718                         /* If the TIM IE is invalid, pretend the value is 1 */
719                         if (!dtimper)
720                                 dtimper = 1;
721                         else if (dtimper > 1)
722                                 maxslp = min_t(int, dtimper,
723                                                     latency / beaconint_us);
724
725                         local->hw.conf.max_sleep_period = maxslp;
726                         local->hw.conf.ps_dtim_period = dtimper;
727                         local->ps_sdata = found;
728                 }
729         } else {
730                 local->ps_sdata = NULL;
731         }
732
733  change:
734         ieee80211_change_ps(local);
735 }
736
737 void ieee80211_dynamic_ps_disable_work(struct work_struct *work)
738 {
739         struct ieee80211_local *local =
740                 container_of(work, struct ieee80211_local,
741                              dynamic_ps_disable_work);
742
743         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
744                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
745                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
746         }
747
748         ieee80211_wake_queues_by_reason(&local->hw,
749                                         IEEE80211_QUEUE_STOP_REASON_PS);
750 }
751
752 void ieee80211_dynamic_ps_enable_work(struct work_struct *work)
753 {
754         struct ieee80211_local *local =
755                 container_of(work, struct ieee80211_local,
756                              dynamic_ps_enable_work);
757         struct ieee80211_sub_if_data *sdata = local->ps_sdata;
758         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
759         unsigned long flags;
760         int q;
761
762         /* can only happen when PS was just disabled anyway */
763         if (!sdata)
764                 return;
765
766         if (local->hw.conf.flags & IEEE80211_CONF_PS)
767                 return;
768
769         /*
770          * transmission can be stopped by others which leads to
771          * dynamic_ps_timer expiry. Postpond the ps timer if it
772          * is not the actual idle state.
773          */
774         spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
775         for (q = 0; q < local->hw.queues; q++) {
776                 if (local->queue_stop_reasons[q]) {
777                         spin_unlock_irqrestore(&local->queue_stop_reason_lock,
778                                                flags);
779                         mod_timer(&local->dynamic_ps_timer, jiffies +
780                                   msecs_to_jiffies(
781                                   local->hw.conf.dynamic_ps_timeout));
782                         return;
783                 }
784         }
785         spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
786
787         if ((local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK) &&
788             (!(ifmgd->flags & IEEE80211_STA_NULLFUNC_ACKED))) {
789                 netif_tx_stop_all_queues(sdata->dev);
790
791                 if (drv_tx_frames_pending(local))
792                         mod_timer(&local->dynamic_ps_timer, jiffies +
793                                   msecs_to_jiffies(
794                                   local->hw.conf.dynamic_ps_timeout));
795                 else {
796                         ieee80211_send_nullfunc(local, sdata, 1);
797                         /* Flush to get the tx status of nullfunc frame */
798                         drv_flush(local, false);
799                 }
800         }
801
802         if (!((local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS) &&
803               (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)) ||
804             (ifmgd->flags & IEEE80211_STA_NULLFUNC_ACKED)) {
805                 ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
806                 local->hw.conf.flags |= IEEE80211_CONF_PS;
807                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
808         }
809
810         netif_tx_wake_all_queues(sdata->dev);
811 }
812
813 void ieee80211_dynamic_ps_timer(unsigned long data)
814 {
815         struct ieee80211_local *local = (void *) data;
816
817         if (local->quiescing || local->suspended)
818                 return;
819
820         ieee80211_queue_work(&local->hw, &local->dynamic_ps_enable_work);
821 }
822
823 /* MLME */
824 static void ieee80211_sta_wmm_params(struct ieee80211_local *local,
825                                      struct ieee80211_sub_if_data *sdata,
826                                      u8 *wmm_param, size_t wmm_param_len)
827 {
828         struct ieee80211_tx_queue_params params;
829         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
830         size_t left;
831         int count;
832         u8 *pos, uapsd_queues = 0;
833
834         if (!local->ops->conf_tx)
835                 return;
836
837         if (local->hw.queues < 4)
838                 return;
839
840         if (!wmm_param)
841                 return;
842
843         if (wmm_param_len < 8 || wmm_param[5] /* version */ != 1)
844                 return;
845
846         if (ifmgd->flags & IEEE80211_STA_UAPSD_ENABLED)
847                 uapsd_queues = local->uapsd_queues;
848
849         count = wmm_param[6] & 0x0f;
850         if (count == ifmgd->wmm_last_param_set)
851                 return;
852         ifmgd->wmm_last_param_set = count;
853
854         pos = wmm_param + 8;
855         left = wmm_param_len - 8;
856
857         memset(&params, 0, sizeof(params));
858
859         local->wmm_acm = 0;
860         for (; left >= 4; left -= 4, pos += 4) {
861                 int aci = (pos[0] >> 5) & 0x03;
862                 int acm = (pos[0] >> 4) & 0x01;
863                 bool uapsd = false;
864                 int queue;
865
866                 switch (aci) {
867                 case 1: /* AC_BK */
868                         queue = 3;
869                         if (acm)
870                                 local->wmm_acm |= BIT(1) | BIT(2); /* BK/- */
871                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BK)
872                                 uapsd = true;
873                         break;
874                 case 2: /* AC_VI */
875                         queue = 1;
876                         if (acm)
877                                 local->wmm_acm |= BIT(4) | BIT(5); /* CL/VI */
878                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VI)
879                                 uapsd = true;
880                         break;
881                 case 3: /* AC_VO */
882                         queue = 0;
883                         if (acm)
884                                 local->wmm_acm |= BIT(6) | BIT(7); /* VO/NC */
885                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VO)
886                                 uapsd = true;
887                         break;
888                 case 0: /* AC_BE */
889                 default:
890                         queue = 2;
891                         if (acm)
892                                 local->wmm_acm |= BIT(0) | BIT(3); /* BE/EE */
893                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BE)
894                                 uapsd = true;
895                         break;
896                 }
897
898                 params.aifs = pos[0] & 0x0f;
899                 params.cw_max = ecw2cw((pos[1] & 0xf0) >> 4);
900                 params.cw_min = ecw2cw(pos[1] & 0x0f);
901                 params.txop = get_unaligned_le16(pos + 2);
902                 params.uapsd = uapsd;
903
904 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
905                 wiphy_debug(local->hw.wiphy,
906                             "WMM queue=%d aci=%d acm=%d aifs=%d "
907                             "cWmin=%d cWmax=%d txop=%d uapsd=%d\n",
908                             queue, aci, acm,
909                             params.aifs, params.cw_min, params.cw_max,
910                             params.txop, params.uapsd);
911 #endif
912                 if (drv_conf_tx(local, queue, &params))
913                         wiphy_debug(local->hw.wiphy,
914                                     "failed to set TX queue parameters for queue %d\n",
915                                     queue);
916         }
917
918         /* enable WMM or activate new settings */
919         sdata->vif.bss_conf.qos = true;
920         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_QOS);
921 }
922
923 static u32 ieee80211_handle_bss_capability(struct ieee80211_sub_if_data *sdata,
924                                            u16 capab, bool erp_valid, u8 erp)
925 {
926         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
927         u32 changed = 0;
928         bool use_protection;
929         bool use_short_preamble;
930         bool use_short_slot;
931
932         if (erp_valid) {
933                 use_protection = (erp & WLAN_ERP_USE_PROTECTION) != 0;
934                 use_short_preamble = (erp & WLAN_ERP_BARKER_PREAMBLE) == 0;
935         } else {
936                 use_protection = false;
937                 use_short_preamble = !!(capab & WLAN_CAPABILITY_SHORT_PREAMBLE);
938         }
939
940         use_short_slot = !!(capab & WLAN_CAPABILITY_SHORT_SLOT_TIME);
941         if (sdata->local->hw.conf.channel->band == IEEE80211_BAND_5GHZ)
942                 use_short_slot = true;
943
944         if (use_protection != bss_conf->use_cts_prot) {
945                 bss_conf->use_cts_prot = use_protection;
946                 changed |= BSS_CHANGED_ERP_CTS_PROT;
947         }
948
949         if (use_short_preamble != bss_conf->use_short_preamble) {
950                 bss_conf->use_short_preamble = use_short_preamble;
951                 changed |= BSS_CHANGED_ERP_PREAMBLE;
952         }
953
954         if (use_short_slot != bss_conf->use_short_slot) {
955                 bss_conf->use_short_slot = use_short_slot;
956                 changed |= BSS_CHANGED_ERP_SLOT;
957         }
958
959         return changed;
960 }
961
962 static void ieee80211_set_associated(struct ieee80211_sub_if_data *sdata,
963                                      struct cfg80211_bss *cbss,
964                                      u32 bss_info_changed)
965 {
966         struct ieee80211_bss *bss = (void *)cbss->priv;
967         struct ieee80211_local *local = sdata->local;
968         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
969
970         bss_info_changed |= BSS_CHANGED_ASSOC;
971         /* set timing information */
972         bss_conf->beacon_int = cbss->beacon_interval;
973         bss_conf->timestamp = cbss->tsf;
974
975         bss_info_changed |= BSS_CHANGED_BEACON_INT;
976         bss_info_changed |= ieee80211_handle_bss_capability(sdata,
977                 cbss->capability, bss->has_erp_value, bss->erp_value);
978
979         sdata->u.mgd.beacon_timeout = usecs_to_jiffies(ieee80211_tu_to_usec(
980                 IEEE80211_BEACON_LOSS_COUNT * bss_conf->beacon_int));
981
982         sdata->u.mgd.associated = cbss;
983         memcpy(sdata->u.mgd.bssid, cbss->bssid, ETH_ALEN);
984
985         sdata->u.mgd.flags |= IEEE80211_STA_RESET_SIGNAL_AVE;
986
987         /* just to be sure */
988         sdata->u.mgd.flags &= ~(IEEE80211_STA_CONNECTION_POLL |
989                                 IEEE80211_STA_BEACON_POLL);
990
991         ieee80211_led_assoc(local, 1);
992
993         if (local->hw.flags & IEEE80211_HW_NEED_DTIM_PERIOD)
994                 bss_conf->dtim_period = bss->dtim_period;
995         else
996                 bss_conf->dtim_period = 0;
997
998         bss_conf->assoc = 1;
999         /*
1000          * For now just always ask the driver to update the basic rateset
1001          * when we have associated, we aren't checking whether it actually
1002          * changed or not.
1003          */
1004         bss_info_changed |= BSS_CHANGED_BASIC_RATES;
1005
1006         /* And the BSSID changed - we're associated now */
1007         bss_info_changed |= BSS_CHANGED_BSSID;
1008
1009         /* Tell the driver to monitor connection quality (if supported) */
1010         if ((local->hw.flags & IEEE80211_HW_SUPPORTS_CQM_RSSI) &&
1011             bss_conf->cqm_rssi_thold)
1012                 bss_info_changed |= BSS_CHANGED_CQM;
1013
1014         /* Enable ARP filtering */
1015         if (bss_conf->arp_filter_enabled != sdata->arp_filter_state) {
1016                 bss_conf->arp_filter_enabled = sdata->arp_filter_state;
1017                 bss_info_changed |= BSS_CHANGED_ARP_FILTER;
1018         }
1019
1020         ieee80211_bss_info_change_notify(sdata, bss_info_changed);
1021
1022         mutex_lock(&local->iflist_mtx);
1023         ieee80211_recalc_ps(local, -1);
1024         ieee80211_recalc_smps(local);
1025         mutex_unlock(&local->iflist_mtx);
1026
1027         netif_tx_start_all_queues(sdata->dev);
1028         netif_carrier_on(sdata->dev);
1029 }
1030
1031 static void ieee80211_set_disassoc(struct ieee80211_sub_if_data *sdata,
1032                                    bool remove_sta, bool tx)
1033 {
1034         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1035         struct ieee80211_local *local = sdata->local;
1036         struct sta_info *sta;
1037         u32 changed = 0, config_changed = 0;
1038         u8 bssid[ETH_ALEN];
1039
1040         ASSERT_MGD_MTX(ifmgd);
1041
1042         if (WARN_ON(!ifmgd->associated))
1043                 return;
1044
1045         memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
1046
1047         ifmgd->associated = NULL;
1048         memset(ifmgd->bssid, 0, ETH_ALEN);
1049
1050         /*
1051          * we need to commit the associated = NULL change because the
1052          * scan code uses that to determine whether this iface should
1053          * go to/wake up from powersave or not -- and could otherwise
1054          * wake the queues erroneously.
1055          */
1056         smp_mb();
1057
1058         /*
1059          * Thus, we can only afterwards stop the queues -- to account
1060          * for the case where another CPU is finishing a scan at this
1061          * time -- we don't want the scan code to enable queues.
1062          */
1063
1064         netif_tx_stop_all_queues(sdata->dev);
1065         netif_carrier_off(sdata->dev);
1066
1067         mutex_lock(&local->sta_mtx);
1068         sta = sta_info_get(sdata, bssid);
1069         if (sta) {
1070                 set_sta_flags(sta, WLAN_STA_BLOCK_BA);
1071                 ieee80211_sta_tear_down_BA_sessions(sta, tx);
1072         }
1073         mutex_unlock(&local->sta_mtx);
1074
1075         changed |= ieee80211_reset_erp_info(sdata);
1076
1077         ieee80211_led_assoc(local, 0);
1078         changed |= BSS_CHANGED_ASSOC;
1079         sdata->vif.bss_conf.assoc = false;
1080
1081         ieee80211_set_wmm_default(sdata);
1082
1083         /* channel(_type) changes are handled by ieee80211_hw_config */
1084         WARN_ON(!ieee80211_set_channel_type(local, sdata, NL80211_CHAN_NO_HT));
1085
1086         /* on the next assoc, re-program HT parameters */
1087         sdata->ht_opmode_valid = false;
1088
1089         local->power_constr_level = 0;
1090
1091         del_timer_sync(&local->dynamic_ps_timer);
1092         cancel_work_sync(&local->dynamic_ps_enable_work);
1093
1094         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
1095                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
1096                 config_changed |= IEEE80211_CONF_CHANGE_PS;
1097         }
1098         local->ps_sdata = NULL;
1099
1100         ieee80211_hw_config(local, config_changed);
1101
1102         /* Disable ARP filtering */
1103         if (sdata->vif.bss_conf.arp_filter_enabled) {
1104                 sdata->vif.bss_conf.arp_filter_enabled = false;
1105                 changed |= BSS_CHANGED_ARP_FILTER;
1106         }
1107
1108         /* The BSSID (not really interesting) and HT changed */
1109         changed |= BSS_CHANGED_BSSID | BSS_CHANGED_HT;
1110         ieee80211_bss_info_change_notify(sdata, changed);
1111
1112         if (remove_sta)
1113                 sta_info_destroy_addr(sdata, bssid);
1114
1115         del_timer_sync(&sdata->u.mgd.conn_mon_timer);
1116         del_timer_sync(&sdata->u.mgd.bcn_mon_timer);
1117         del_timer_sync(&sdata->u.mgd.timer);
1118         del_timer_sync(&sdata->u.mgd.chswitch_timer);
1119 }
1120
1121 void ieee80211_sta_rx_notify(struct ieee80211_sub_if_data *sdata,
1122                              struct ieee80211_hdr *hdr)
1123 {
1124         /*
1125          * We can postpone the mgd.timer whenever receiving unicast frames
1126          * from AP because we know that the connection is working both ways
1127          * at that time. But multicast frames (and hence also beacons) must
1128          * be ignored here, because we need to trigger the timer during
1129          * data idle periods for sending the periodic probe request to the
1130          * AP we're connected to.
1131          */
1132         if (is_multicast_ether_addr(hdr->addr1))
1133                 return;
1134
1135         ieee80211_sta_reset_conn_monitor(sdata);
1136 }
1137
1138 static void ieee80211_reset_ap_probe(struct ieee80211_sub_if_data *sdata)
1139 {
1140         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1141
1142         if (!(ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1143                               IEEE80211_STA_CONNECTION_POLL)))
1144             return;
1145
1146         ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
1147                           IEEE80211_STA_BEACON_POLL);
1148         mutex_lock(&sdata->local->iflist_mtx);
1149         ieee80211_recalc_ps(sdata->local, -1);
1150         mutex_unlock(&sdata->local->iflist_mtx);
1151
1152         if (sdata->local->hw.flags & IEEE80211_HW_CONNECTION_MONITOR)
1153                 return;
1154
1155         /*
1156          * We've received a probe response, but are not sure whether
1157          * we have or will be receiving any beacons or data, so let's
1158          * schedule the timers again, just in case.
1159          */
1160         ieee80211_sta_reset_beacon_monitor(sdata);
1161
1162         mod_timer(&ifmgd->conn_mon_timer,
1163                   round_jiffies_up(jiffies +
1164                                    IEEE80211_CONNECTION_IDLE_TIME));
1165 }
1166
1167 void ieee80211_sta_tx_notify(struct ieee80211_sub_if_data *sdata,
1168                              struct ieee80211_hdr *hdr, bool ack)
1169 {
1170         if (!ieee80211_is_data(hdr->frame_control))
1171             return;
1172
1173         if (ack)
1174                 ieee80211_sta_reset_conn_monitor(sdata);
1175
1176         if (ieee80211_is_nullfunc(hdr->frame_control) &&
1177             sdata->u.mgd.probe_send_count > 0) {
1178                 if (ack)
1179                         sdata->u.mgd.probe_send_count = 0;
1180                 else
1181                         sdata->u.mgd.nullfunc_failed = true;
1182                 ieee80211_queue_work(&sdata->local->hw, &sdata->work);
1183         }
1184 }
1185
1186 static void ieee80211_mgd_probe_ap_send(struct ieee80211_sub_if_data *sdata)
1187 {
1188         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1189         const u8 *ssid;
1190         u8 *dst = ifmgd->associated->bssid;
1191         u8 unicast_limit = max(1, max_probe_tries - 3);
1192
1193         /*
1194          * Try sending broadcast probe requests for the last three
1195          * probe requests after the first ones failed since some
1196          * buggy APs only support broadcast probe requests.
1197          */
1198         if (ifmgd->probe_send_count >= unicast_limit)
1199                 dst = NULL;
1200
1201         /*
1202          * When the hardware reports an accurate Tx ACK status, it's
1203          * better to send a nullfunc frame instead of a probe request,
1204          * as it will kick us off the AP quickly if we aren't associated
1205          * anymore. The timeout will be reset if the frame is ACKed by
1206          * the AP.
1207          */
1208         if (sdata->local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS) {
1209                 ifmgd->nullfunc_failed = false;
1210                 ieee80211_send_nullfunc(sdata->local, sdata, 0);
1211         } else {
1212                 ssid = ieee80211_bss_get_ie(ifmgd->associated, WLAN_EID_SSID);
1213                 ieee80211_send_probe_req(sdata, dst, ssid + 2, ssid[1], NULL, 0);
1214         }
1215
1216         ifmgd->probe_send_count++;
1217         ifmgd->probe_timeout = jiffies + msecs_to_jiffies(probe_wait_ms);
1218         run_again(ifmgd, ifmgd->probe_timeout);
1219 }
1220
1221 static void ieee80211_mgd_probe_ap(struct ieee80211_sub_if_data *sdata,
1222                                    bool beacon)
1223 {
1224         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1225         bool already = false;
1226
1227         if (!ieee80211_sdata_running(sdata))
1228                 return;
1229
1230         if (sdata->local->scanning)
1231                 return;
1232
1233         if (sdata->local->tmp_channel)
1234                 return;
1235
1236         mutex_lock(&ifmgd->mtx);
1237
1238         if (!ifmgd->associated)
1239                 goto out;
1240
1241 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1242         if (beacon && net_ratelimit())
1243                 printk(KERN_DEBUG "%s: detected beacon loss from AP "
1244                        "- sending probe request\n", sdata->name);
1245 #endif
1246
1247         /*
1248          * The driver/our work has already reported this event or the
1249          * connection monitoring has kicked in and we have already sent
1250          * a probe request. Or maybe the AP died and the driver keeps
1251          * reporting until we disassociate...
1252          *
1253          * In either case we have to ignore the current call to this
1254          * function (except for setting the correct probe reason bit)
1255          * because otherwise we would reset the timer every time and
1256          * never check whether we received a probe response!
1257          */
1258         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1259                             IEEE80211_STA_CONNECTION_POLL))
1260                 already = true;
1261
1262         if (beacon)
1263                 ifmgd->flags |= IEEE80211_STA_BEACON_POLL;
1264         else
1265                 ifmgd->flags |= IEEE80211_STA_CONNECTION_POLL;
1266
1267         if (already)
1268                 goto out;
1269
1270         mutex_lock(&sdata->local->iflist_mtx);
1271         ieee80211_recalc_ps(sdata->local, -1);
1272         mutex_unlock(&sdata->local->iflist_mtx);
1273
1274         ifmgd->probe_send_count = 0;
1275         ieee80211_mgd_probe_ap_send(sdata);
1276  out:
1277         mutex_unlock(&ifmgd->mtx);
1278 }
1279
1280 struct sk_buff *ieee80211_ap_probereq_get(struct ieee80211_hw *hw,
1281                                           struct ieee80211_vif *vif)
1282 {
1283         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1284         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1285         struct sk_buff *skb;
1286         const u8 *ssid;
1287
1288         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
1289                 return NULL;
1290
1291         ASSERT_MGD_MTX(ifmgd);
1292
1293         if (!ifmgd->associated)
1294                 return NULL;
1295
1296         ssid = ieee80211_bss_get_ie(ifmgd->associated, WLAN_EID_SSID);
1297         skb = ieee80211_build_probe_req(sdata, ifmgd->associated->bssid,
1298                                         ssid + 2, ssid[1], NULL, 0);
1299
1300         return skb;
1301 }
1302 EXPORT_SYMBOL(ieee80211_ap_probereq_get);
1303
1304 static void __ieee80211_connection_loss(struct ieee80211_sub_if_data *sdata)
1305 {
1306         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1307         struct ieee80211_local *local = sdata->local;
1308         u8 bssid[ETH_ALEN];
1309
1310         mutex_lock(&ifmgd->mtx);
1311         if (!ifmgd->associated) {
1312                 mutex_unlock(&ifmgd->mtx);
1313                 return;
1314         }
1315
1316         memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
1317
1318         printk(KERN_DEBUG "%s: Connection to AP %pM lost.\n",
1319                sdata->name, bssid);
1320
1321         ieee80211_set_disassoc(sdata, true, true);
1322         mutex_unlock(&ifmgd->mtx);
1323
1324         mutex_lock(&local->mtx);
1325         ieee80211_recalc_idle(local);
1326         mutex_unlock(&local->mtx);
1327         /*
1328          * must be outside lock due to cfg80211,
1329          * but that's not a problem.
1330          */
1331         ieee80211_send_deauth_disassoc(sdata, bssid,
1332                                        IEEE80211_STYPE_DEAUTH,
1333                                        WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
1334                                        NULL, true);
1335 }
1336
1337 void ieee80211_beacon_connection_loss_work(struct work_struct *work)
1338 {
1339         struct ieee80211_sub_if_data *sdata =
1340                 container_of(work, struct ieee80211_sub_if_data,
1341                              u.mgd.beacon_connection_loss_work);
1342
1343         if (sdata->local->hw.flags & IEEE80211_HW_CONNECTION_MONITOR)
1344                 __ieee80211_connection_loss(sdata);
1345         else
1346                 ieee80211_mgd_probe_ap(sdata, true);
1347 }
1348
1349 void ieee80211_beacon_loss(struct ieee80211_vif *vif)
1350 {
1351         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1352         struct ieee80211_hw *hw = &sdata->local->hw;
1353
1354         trace_api_beacon_loss(sdata);
1355
1356         WARN_ON(hw->flags & IEEE80211_HW_CONNECTION_MONITOR);
1357         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
1358 }
1359 EXPORT_SYMBOL(ieee80211_beacon_loss);
1360
1361 void ieee80211_connection_loss(struct ieee80211_vif *vif)
1362 {
1363         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1364         struct ieee80211_hw *hw = &sdata->local->hw;
1365
1366         trace_api_connection_loss(sdata);
1367
1368         WARN_ON(!(hw->flags & IEEE80211_HW_CONNECTION_MONITOR));
1369         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
1370 }
1371 EXPORT_SYMBOL(ieee80211_connection_loss);
1372
1373
1374 static enum rx_mgmt_action __must_check
1375 ieee80211_rx_mgmt_deauth(struct ieee80211_sub_if_data *sdata,
1376                          struct ieee80211_mgmt *mgmt, size_t len)
1377 {
1378         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1379         const u8 *bssid = NULL;
1380         u16 reason_code;
1381
1382         if (len < 24 + 2)
1383                 return RX_MGMT_NONE;
1384
1385         ASSERT_MGD_MTX(ifmgd);
1386
1387         bssid = ifmgd->associated->bssid;
1388
1389         reason_code = le16_to_cpu(mgmt->u.deauth.reason_code);
1390
1391         printk(KERN_DEBUG "%s: deauthenticated from %pM (Reason: %u)\n",
1392                         sdata->name, bssid, reason_code);
1393
1394         ieee80211_set_disassoc(sdata, true, false);
1395         mutex_lock(&sdata->local->mtx);
1396         ieee80211_recalc_idle(sdata->local);
1397         mutex_unlock(&sdata->local->mtx);
1398
1399         return RX_MGMT_CFG80211_DEAUTH;
1400 }
1401
1402
1403 static enum rx_mgmt_action __must_check
1404 ieee80211_rx_mgmt_disassoc(struct ieee80211_sub_if_data *sdata,
1405                            struct ieee80211_mgmt *mgmt, size_t len)
1406 {
1407         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1408         u16 reason_code;
1409
1410         if (len < 24 + 2)
1411                 return RX_MGMT_NONE;
1412
1413         ASSERT_MGD_MTX(ifmgd);
1414
1415         if (WARN_ON(!ifmgd->associated))
1416                 return RX_MGMT_NONE;
1417
1418         if (WARN_ON(memcmp(ifmgd->associated->bssid, mgmt->sa, ETH_ALEN)))
1419                 return RX_MGMT_NONE;
1420
1421         reason_code = le16_to_cpu(mgmt->u.disassoc.reason_code);
1422
1423         printk(KERN_DEBUG "%s: disassociated from %pM (Reason: %u)\n",
1424                         sdata->name, mgmt->sa, reason_code);
1425
1426         ieee80211_set_disassoc(sdata, true, false);
1427         mutex_lock(&sdata->local->mtx);
1428         ieee80211_recalc_idle(sdata->local);
1429         mutex_unlock(&sdata->local->mtx);
1430         return RX_MGMT_CFG80211_DISASSOC;
1431 }
1432
1433
1434 static bool ieee80211_assoc_success(struct ieee80211_work *wk,
1435                                     struct ieee80211_mgmt *mgmt, size_t len)
1436 {
1437         struct ieee80211_sub_if_data *sdata = wk->sdata;
1438         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1439         struct ieee80211_local *local = sdata->local;
1440         struct ieee80211_supported_band *sband;
1441         struct sta_info *sta;
1442         struct cfg80211_bss *cbss = wk->assoc.bss;
1443         u8 *pos;
1444         u32 rates, basic_rates;
1445         u16 capab_info, aid;
1446         struct ieee802_11_elems elems;
1447         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
1448         u32 changed = 0;
1449         int i, j, err;
1450         bool have_higher_than_11mbit = false;
1451         u16 ap_ht_cap_flags;
1452
1453         /* AssocResp and ReassocResp have identical structure */
1454
1455         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
1456         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
1457
1458         if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14)))
1459                 printk(KERN_DEBUG "%s: invalid aid value %d; bits 15:14 not "
1460                        "set\n", sdata->name, aid);
1461         aid &= ~(BIT(15) | BIT(14));
1462
1463         pos = mgmt->u.assoc_resp.variable;
1464         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems);
1465
1466         if (!elems.supp_rates) {
1467                 printk(KERN_DEBUG "%s: no SuppRates element in AssocResp\n",
1468                        sdata->name);
1469                 return false;
1470         }
1471
1472         ifmgd->aid = aid;
1473
1474         sta = sta_info_alloc(sdata, cbss->bssid, GFP_KERNEL);
1475         if (!sta) {
1476                 printk(KERN_DEBUG "%s: failed to alloc STA entry for"
1477                        " the AP\n", sdata->name);
1478                 return false;
1479         }
1480
1481         set_sta_flags(sta, WLAN_STA_AUTH | WLAN_STA_ASSOC |
1482                            WLAN_STA_ASSOC_AP);
1483         if (!(ifmgd->flags & IEEE80211_STA_CONTROL_PORT))
1484                 set_sta_flags(sta, WLAN_STA_AUTHORIZED);
1485
1486         rates = 0;
1487         basic_rates = 0;
1488         sband = local->hw.wiphy->bands[wk->chan->band];
1489
1490         for (i = 0; i < elems.supp_rates_len; i++) {
1491                 int rate = (elems.supp_rates[i] & 0x7f) * 5;
1492                 bool is_basic = !!(elems.supp_rates[i] & 0x80);
1493
1494                 if (rate > 110)
1495                         have_higher_than_11mbit = true;
1496
1497                 for (j = 0; j < sband->n_bitrates; j++) {
1498                         if (sband->bitrates[j].bitrate == rate) {
1499                                 rates |= BIT(j);
1500                                 if (is_basic)
1501                                         basic_rates |= BIT(j);
1502                                 break;
1503                         }
1504                 }
1505         }
1506
1507         for (i = 0; i < elems.ext_supp_rates_len; i++) {
1508                 int rate = (elems.ext_supp_rates[i] & 0x7f) * 5;
1509                 bool is_basic = !!(elems.ext_supp_rates[i] & 0x80);
1510
1511                 if (rate > 110)
1512                         have_higher_than_11mbit = true;
1513
1514                 for (j = 0; j < sband->n_bitrates; j++) {
1515                         if (sband->bitrates[j].bitrate == rate) {
1516                                 rates |= BIT(j);
1517                                 if (is_basic)
1518                                         basic_rates |= BIT(j);
1519                                 break;
1520                         }
1521                 }
1522         }
1523
1524         sta->sta.supp_rates[wk->chan->band] = rates;
1525         sdata->vif.bss_conf.basic_rates = basic_rates;
1526
1527         /* cf. IEEE 802.11 9.2.12 */
1528         if (wk->chan->band == IEEE80211_BAND_2GHZ &&
1529             have_higher_than_11mbit)
1530                 sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
1531         else
1532                 sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;
1533
1534         if (elems.ht_cap_elem && !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1535                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1536                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1537
1538         ap_ht_cap_flags = sta->sta.ht_cap.cap;
1539
1540         rate_control_rate_init(sta);
1541
1542         if (ifmgd->flags & IEEE80211_STA_MFP_ENABLED)
1543                 set_sta_flags(sta, WLAN_STA_MFP);
1544
1545         if (elems.wmm_param)
1546                 set_sta_flags(sta, WLAN_STA_WME);
1547
1548         err = sta_info_insert(sta);
1549         sta = NULL;
1550         if (err) {
1551                 printk(KERN_DEBUG "%s: failed to insert STA entry for"
1552                        " the AP (error %d)\n", sdata->name, err);
1553                 return false;
1554         }
1555
1556         /*
1557          * Always handle WMM once after association regardless
1558          * of the first value the AP uses. Setting -1 here has
1559          * that effect because the AP values is an unsigned
1560          * 4-bit value.
1561          */
1562         ifmgd->wmm_last_param_set = -1;
1563
1564         if (elems.wmm_param)
1565                 ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
1566                                          elems.wmm_param_len);
1567         else
1568                 ieee80211_set_wmm_default(sdata);
1569
1570         local->oper_channel = wk->chan;
1571
1572         if (elems.ht_info_elem && elems.wmm_param &&
1573             (sdata->local->hw.queues >= 4) &&
1574             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1575                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1576                                                cbss->bssid, ap_ht_cap_flags);
1577
1578         /* set AID and assoc capability,
1579          * ieee80211_set_associated() will tell the driver */
1580         bss_conf->aid = aid;
1581         bss_conf->assoc_capability = capab_info;
1582         ieee80211_set_associated(sdata, cbss, changed);
1583
1584         /*
1585          * If we're using 4-addr mode, let the AP know that we're
1586          * doing so, so that it can create the STA VLAN on its side
1587          */
1588         if (ifmgd->use_4addr)
1589                 ieee80211_send_4addr_nullfunc(local, sdata);
1590
1591         /*
1592          * Start timer to probe the connection to the AP now.
1593          * Also start the timer that will detect beacon loss.
1594          */
1595         ieee80211_sta_rx_notify(sdata, (struct ieee80211_hdr *)mgmt);
1596         ieee80211_sta_reset_beacon_monitor(sdata);
1597
1598         return true;
1599 }
1600
1601
1602 static void ieee80211_rx_bss_info(struct ieee80211_sub_if_data *sdata,
1603                                   struct ieee80211_mgmt *mgmt,
1604                                   size_t len,
1605                                   struct ieee80211_rx_status *rx_status,
1606                                   struct ieee802_11_elems *elems,
1607                                   bool beacon)
1608 {
1609         struct ieee80211_local *local = sdata->local;
1610         int freq;
1611         struct ieee80211_bss *bss;
1612         struct ieee80211_channel *channel;
1613         bool need_ps = false;
1614
1615         if (sdata->u.mgd.associated) {
1616                 bss = (void *)sdata->u.mgd.associated->priv;
1617                 /* not previously set so we may need to recalc */
1618                 need_ps = !bss->dtim_period;
1619         }
1620
1621         if (elems->ds_params && elems->ds_params_len == 1)
1622                 freq = ieee80211_channel_to_frequency(elems->ds_params[0],
1623                                                       rx_status->band);
1624         else
1625                 freq = rx_status->freq;
1626
1627         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1628
1629         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1630                 return;
1631
1632         bss = ieee80211_bss_info_update(local, rx_status, mgmt, len, elems,
1633                                         channel, beacon);
1634         if (bss)
1635                 ieee80211_rx_bss_put(local, bss);
1636
1637         if (!sdata->u.mgd.associated)
1638                 return;
1639
1640         if (need_ps) {
1641                 mutex_lock(&local->iflist_mtx);
1642                 ieee80211_recalc_ps(local, -1);
1643                 mutex_unlock(&local->iflist_mtx);
1644         }
1645
1646         if (elems->ch_switch_elem && (elems->ch_switch_elem_len == 3) &&
1647             (memcmp(mgmt->bssid, sdata->u.mgd.associated->bssid,
1648                                                         ETH_ALEN) == 0)) {
1649                 struct ieee80211_channel_sw_ie *sw_elem =
1650                         (struct ieee80211_channel_sw_ie *)elems->ch_switch_elem;
1651                 ieee80211_sta_process_chanswitch(sdata, sw_elem,
1652                                                  bss, rx_status->mactime);
1653         }
1654 }
1655
1656
1657 static void ieee80211_rx_mgmt_probe_resp(struct ieee80211_sub_if_data *sdata,
1658                                          struct sk_buff *skb)
1659 {
1660         struct ieee80211_mgmt *mgmt = (void *)skb->data;
1661         struct ieee80211_if_managed *ifmgd;
1662         struct ieee80211_rx_status *rx_status = (void *) skb->cb;
1663         size_t baselen, len = skb->len;
1664         struct ieee802_11_elems elems;
1665
1666         ifmgd = &sdata->u.mgd;
1667
1668         ASSERT_MGD_MTX(ifmgd);
1669
1670         if (memcmp(mgmt->da, sdata->vif.addr, ETH_ALEN))
1671                 return; /* ignore ProbeResp to foreign address */
1672
1673         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1674         if (baselen > len)
1675                 return;
1676
1677         ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - baselen,
1678                                 &elems);
1679
1680         ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems, false);
1681
1682         if (ifmgd->associated &&
1683             memcmp(mgmt->bssid, ifmgd->associated->bssid, ETH_ALEN) == 0)
1684                 ieee80211_reset_ap_probe(sdata);
1685 }
1686
1687 /*
1688  * This is the canonical list of information elements we care about,
1689  * the filter code also gives us all changes to the Microsoft OUI
1690  * (00:50:F2) vendor IE which is used for WMM which we need to track.
1691  *
1692  * We implement beacon filtering in software since that means we can
1693  * avoid processing the frame here and in cfg80211, and userspace
1694  * will not be able to tell whether the hardware supports it or not.
1695  *
1696  * XXX: This list needs to be dynamic -- userspace needs to be able to
1697  *      add items it requires. It also needs to be able to tell us to
1698  *      look out for other vendor IEs.
1699  */
1700 static const u64 care_about_ies =
1701         (1ULL << WLAN_EID_COUNTRY) |
1702         (1ULL << WLAN_EID_ERP_INFO) |
1703         (1ULL << WLAN_EID_CHANNEL_SWITCH) |
1704         (1ULL << WLAN_EID_PWR_CONSTRAINT) |
1705         (1ULL << WLAN_EID_HT_CAPABILITY) |
1706         (1ULL << WLAN_EID_HT_INFORMATION);
1707
1708 static void ieee80211_rx_mgmt_beacon(struct ieee80211_sub_if_data *sdata,
1709                                      struct ieee80211_mgmt *mgmt,
1710                                      size_t len,
1711                                      struct ieee80211_rx_status *rx_status)
1712 {
1713         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1714         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
1715         size_t baselen;
1716         struct ieee802_11_elems elems;
1717         struct ieee80211_local *local = sdata->local;
1718         u32 changed = 0;
1719         bool erp_valid, directed_tim = false;
1720         u8 erp_value = 0;
1721         u32 ncrc;
1722         u8 *bssid;
1723
1724         ASSERT_MGD_MTX(ifmgd);
1725
1726         /* Process beacon from the current BSS */
1727         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
1728         if (baselen > len)
1729                 return;
1730
1731         if (rx_status->freq != local->hw.conf.channel->center_freq)
1732                 return;
1733
1734         /*
1735          * We might have received a number of frames, among them a
1736          * disassoc frame and a beacon...
1737          */
1738         if (!ifmgd->associated)
1739                 return;
1740
1741         bssid = ifmgd->associated->bssid;
1742
1743         /*
1744          * And in theory even frames from a different AP we were just
1745          * associated to a split-second ago!
1746          */
1747         if (memcmp(bssid, mgmt->bssid, ETH_ALEN) != 0)
1748                 return;
1749
1750         /* Track average RSSI from the Beacon frames of the current AP */
1751         ifmgd->last_beacon_signal = rx_status->signal;
1752         if (ifmgd->flags & IEEE80211_STA_RESET_SIGNAL_AVE) {
1753                 ifmgd->flags &= ~IEEE80211_STA_RESET_SIGNAL_AVE;
1754                 ifmgd->ave_beacon_signal = rx_status->signal * 16;
1755                 ifmgd->last_cqm_event_signal = 0;
1756                 ifmgd->count_beacon_signal = 1;
1757         } else {
1758                 ifmgd->ave_beacon_signal =
1759                         (IEEE80211_SIGNAL_AVE_WEIGHT * rx_status->signal * 16 +
1760                          (16 - IEEE80211_SIGNAL_AVE_WEIGHT) *
1761                          ifmgd->ave_beacon_signal) / 16;
1762                 ifmgd->count_beacon_signal++;
1763         }
1764         if (bss_conf->cqm_rssi_thold &&
1765             ifmgd->count_beacon_signal >= IEEE80211_SIGNAL_AVE_MIN_COUNT &&
1766             !(local->hw.flags & IEEE80211_HW_SUPPORTS_CQM_RSSI)) {
1767                 int sig = ifmgd->ave_beacon_signal / 16;
1768                 int last_event = ifmgd->last_cqm_event_signal;
1769                 int thold = bss_conf->cqm_rssi_thold;
1770                 int hyst = bss_conf->cqm_rssi_hyst;
1771                 if (sig < thold &&
1772                     (last_event == 0 || sig < last_event - hyst)) {
1773                         ifmgd->last_cqm_event_signal = sig;
1774                         ieee80211_cqm_rssi_notify(
1775                                 &sdata->vif,
1776                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW,
1777                                 GFP_KERNEL);
1778                 } else if (sig > thold &&
1779                            (last_event == 0 || sig > last_event + hyst)) {
1780                         ifmgd->last_cqm_event_signal = sig;
1781                         ieee80211_cqm_rssi_notify(
1782                                 &sdata->vif,
1783                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH,
1784                                 GFP_KERNEL);
1785                 }
1786         }
1787
1788         if (ifmgd->flags & IEEE80211_STA_BEACON_POLL) {
1789 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1790                 if (net_ratelimit()) {
1791                         printk(KERN_DEBUG "%s: cancelling probereq poll due "
1792                                "to a received beacon\n", sdata->name);
1793                 }
1794 #endif
1795                 ifmgd->flags &= ~IEEE80211_STA_BEACON_POLL;
1796                 mutex_lock(&local->iflist_mtx);
1797                 ieee80211_recalc_ps(local, -1);
1798                 mutex_unlock(&local->iflist_mtx);
1799         }
1800
1801         /*
1802          * Push the beacon loss detection into the future since
1803          * we are processing a beacon from the AP just now.
1804          */
1805         ieee80211_sta_reset_beacon_monitor(sdata);
1806
1807         ncrc = crc32_be(0, (void *)&mgmt->u.beacon.beacon_int, 4);
1808         ncrc = ieee802_11_parse_elems_crc(mgmt->u.beacon.variable,
1809                                           len - baselen, &elems,
1810                                           care_about_ies, ncrc);
1811
1812         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
1813                 directed_tim = ieee80211_check_tim(elems.tim, elems.tim_len,
1814                                                    ifmgd->aid);
1815
1816         if (ncrc != ifmgd->beacon_crc || !ifmgd->beacon_crc_valid) {
1817                 ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems,
1818                                       true);
1819
1820                 ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
1821                                          elems.wmm_param_len);
1822         }
1823
1824         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK) {
1825                 if (directed_tim) {
1826                         if (local->hw.conf.dynamic_ps_timeout > 0) {
1827                                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
1828                                 ieee80211_hw_config(local,
1829                                                     IEEE80211_CONF_CHANGE_PS);
1830                                 ieee80211_send_nullfunc(local, sdata, 0);
1831                         } else {
1832                                 local->pspolling = true;
1833
1834                                 /*
1835                                  * Here is assumed that the driver will be
1836                                  * able to send ps-poll frame and receive a
1837                                  * response even though power save mode is
1838                                  * enabled, but some drivers might require
1839                                  * to disable power save here. This needs
1840                                  * to be investigated.
1841                                  */
1842                                 ieee80211_send_pspoll(local, sdata);
1843                         }
1844                 }
1845         }
1846
1847         if (ncrc == ifmgd->beacon_crc && ifmgd->beacon_crc_valid)
1848                 return;
1849         ifmgd->beacon_crc = ncrc;
1850         ifmgd->beacon_crc_valid = true;
1851
1852         if (elems.erp_info && elems.erp_info_len >= 1) {
1853                 erp_valid = true;
1854                 erp_value = elems.erp_info[0];
1855         } else {
1856                 erp_valid = false;
1857         }
1858         changed |= ieee80211_handle_bss_capability(sdata,
1859                         le16_to_cpu(mgmt->u.beacon.capab_info),
1860                         erp_valid, erp_value);
1861
1862
1863         if (elems.ht_cap_elem && elems.ht_info_elem && elems.wmm_param &&
1864             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N)) {
1865                 struct sta_info *sta;
1866                 struct ieee80211_supported_band *sband;
1867                 u16 ap_ht_cap_flags;
1868
1869                 rcu_read_lock();
1870
1871                 sta = sta_info_get(sdata, bssid);
1872                 if (WARN_ON(!sta)) {
1873                         rcu_read_unlock();
1874                         return;
1875                 }
1876
1877                 sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
1878
1879                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1880                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1881
1882                 ap_ht_cap_flags = sta->sta.ht_cap.cap;
1883
1884                 rcu_read_unlock();
1885
1886                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1887                                                bssid, ap_ht_cap_flags);
1888         }
1889
1890         /* Note: country IE parsing is done for us by cfg80211 */
1891         if (elems.country_elem) {
1892                 /* TODO: IBSS also needs this */
1893                 if (elems.pwr_constr_elem)
1894                         ieee80211_handle_pwr_constr(sdata,
1895                                 le16_to_cpu(mgmt->u.probe_resp.capab_info),
1896                                 elems.pwr_constr_elem,
1897                                 elems.pwr_constr_elem_len);
1898         }
1899
1900         ieee80211_bss_info_change_notify(sdata, changed);
1901 }
1902
1903 void ieee80211_sta_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1904                                   struct sk_buff *skb)
1905 {
1906         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1907         struct ieee80211_rx_status *rx_status;
1908         struct ieee80211_mgmt *mgmt;
1909         enum rx_mgmt_action rma = RX_MGMT_NONE;
1910         u16 fc;
1911
1912         rx_status = (struct ieee80211_rx_status *) skb->cb;
1913         mgmt = (struct ieee80211_mgmt *) skb->data;
1914         fc = le16_to_cpu(mgmt->frame_control);
1915
1916         mutex_lock(&ifmgd->mtx);
1917
1918         if (ifmgd->associated &&
1919             memcmp(ifmgd->associated->bssid, mgmt->bssid, ETH_ALEN) == 0) {
1920                 switch (fc & IEEE80211_FCTL_STYPE) {
1921                 case IEEE80211_STYPE_BEACON:
1922                         ieee80211_rx_mgmt_beacon(sdata, mgmt, skb->len,
1923                                                  rx_status);
1924                         break;
1925                 case IEEE80211_STYPE_PROBE_RESP:
1926                         ieee80211_rx_mgmt_probe_resp(sdata, skb);
1927                         break;
1928                 case IEEE80211_STYPE_DEAUTH:
1929                         rma = ieee80211_rx_mgmt_deauth(sdata, mgmt, skb->len);
1930                         break;
1931                 case IEEE80211_STYPE_DISASSOC:
1932                         rma = ieee80211_rx_mgmt_disassoc(sdata, mgmt, skb->len);
1933                         break;
1934                 case IEEE80211_STYPE_ACTION:
1935                         switch (mgmt->u.action.category) {
1936                         case WLAN_CATEGORY_SPECTRUM_MGMT:
1937                                 ieee80211_sta_process_chanswitch(sdata,
1938                                                 &mgmt->u.action.u.chan_switch.sw_elem,
1939                                                 (void *)ifmgd->associated->priv,
1940                                                 rx_status->mactime);
1941                                 break;
1942                         }
1943                 }
1944                 mutex_unlock(&ifmgd->mtx);
1945
1946                 switch (rma) {
1947                 case RX_MGMT_NONE:
1948                         /* no action */
1949                         break;
1950                 case RX_MGMT_CFG80211_DEAUTH:
1951                         cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
1952                         break;
1953                 case RX_MGMT_CFG80211_DISASSOC:
1954                         cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
1955                         break;
1956                 default:
1957                         WARN(1, "unexpected: %d", rma);
1958                 }
1959                 return;
1960         }
1961
1962         mutex_unlock(&ifmgd->mtx);
1963
1964         if (skb->len >= 24 + 2 /* mgmt + deauth reason */ &&
1965             (fc & IEEE80211_FCTL_STYPE) == IEEE80211_STYPE_DEAUTH) {
1966                 struct ieee80211_local *local = sdata->local;
1967                 struct ieee80211_work *wk;
1968
1969                 mutex_lock(&local->mtx);
1970                 list_for_each_entry(wk, &local->work_list, list) {
1971                         if (wk->sdata != sdata)
1972                                 continue;
1973
1974                         if (wk->type != IEEE80211_WORK_ASSOC &&
1975                             wk->type != IEEE80211_WORK_ASSOC_BEACON_WAIT)
1976                                 continue;
1977
1978                         if (memcmp(mgmt->bssid, wk->filter_ta, ETH_ALEN))
1979                                 continue;
1980                         if (memcmp(mgmt->sa, wk->filter_ta, ETH_ALEN))
1981                                 continue;
1982
1983                         /*
1984                          * Printing the message only here means we can't
1985                          * spuriously print it, but it also means that it
1986                          * won't be printed when the frame comes in before
1987                          * we even tried to associate or in similar cases.
1988                          *
1989                          * Ultimately, I suspect cfg80211 should print the
1990                          * messages instead.
1991                          */
1992                         printk(KERN_DEBUG
1993                                "%s: deauthenticated from %pM (Reason: %u)\n",
1994                                sdata->name, mgmt->bssid,
1995                                le16_to_cpu(mgmt->u.deauth.reason_code));
1996
1997                         list_del_rcu(&wk->list);
1998                         free_work(wk);
1999                         break;
2000                 }
2001                 mutex_unlock(&local->mtx);
2002
2003                 cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
2004         }
2005 }
2006
2007 static void ieee80211_sta_timer(unsigned long data)
2008 {
2009         struct ieee80211_sub_if_data *sdata =
2010                 (struct ieee80211_sub_if_data *) data;
2011         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2012         struct ieee80211_local *local = sdata->local;
2013
2014         if (local->quiescing) {
2015                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
2016                 return;
2017         }
2018
2019         ieee80211_queue_work(&local->hw, &sdata->work);
2020 }
2021
2022 static void ieee80211_sta_connection_lost(struct ieee80211_sub_if_data *sdata,
2023                                           u8 *bssid)
2024 {
2025         struct ieee80211_local *local = sdata->local;
2026         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2027
2028         ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
2029                           IEEE80211_STA_BEACON_POLL);
2030
2031         ieee80211_set_disassoc(sdata, true, true);
2032         mutex_unlock(&ifmgd->mtx);
2033         mutex_lock(&local->mtx);
2034         ieee80211_recalc_idle(local);
2035         mutex_unlock(&local->mtx);
2036         /*
2037          * must be outside lock due to cfg80211,
2038          * but that's not a problem.
2039          */
2040         ieee80211_send_deauth_disassoc(sdata, bssid,
2041                         IEEE80211_STYPE_DEAUTH,
2042                         WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
2043                         NULL, true);
2044         mutex_lock(&ifmgd->mtx);
2045 }
2046
2047 void ieee80211_sta_work(struct ieee80211_sub_if_data *sdata)
2048 {
2049         struct ieee80211_local *local = sdata->local;
2050         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2051
2052         /* then process the rest of the work */
2053         mutex_lock(&ifmgd->mtx);
2054
2055         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
2056                             IEEE80211_STA_CONNECTION_POLL) &&
2057             ifmgd->associated) {
2058                 u8 bssid[ETH_ALEN];
2059                 int max_tries;
2060
2061                 memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
2062
2063                 if (local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS)
2064                         max_tries = max_nullfunc_tries;
2065                 else
2066                         max_tries = max_probe_tries;
2067
2068                 /* ACK received for nullfunc probing frame */
2069                 if (!ifmgd->probe_send_count)
2070                         ieee80211_reset_ap_probe(sdata);
2071                 else if (ifmgd->nullfunc_failed) {
2072                         if (ifmgd->probe_send_count < max_tries) {
2073 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2074                                 wiphy_debug(local->hw.wiphy,
2075                                             "%s: No ack for nullfunc frame to"
2076                                             " AP %pM, try %d/%i\n",
2077                                             sdata->name, bssid,
2078                                             ifmgd->probe_send_count, max_tries);
2079 #endif
2080                                 ieee80211_mgd_probe_ap_send(sdata);
2081                         } else {
2082 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2083                                 wiphy_debug(local->hw.wiphy,
2084                                             "%s: No ack for nullfunc frame to"
2085                                             " AP %pM, disconnecting.\n",
2086                                             sdata->name, bssid);
2087 #endif
2088                                 ieee80211_sta_connection_lost(sdata, bssid);
2089                         }
2090                 } else if (time_is_after_jiffies(ifmgd->probe_timeout))
2091                         run_again(ifmgd, ifmgd->probe_timeout);
2092                 else if (local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS) {
2093 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2094                         wiphy_debug(local->hw.wiphy,
2095                                     "%s: Failed to send nullfunc to AP %pM"
2096                                     " after %dms, disconnecting.\n",
2097                                     sdata->name,
2098                                     bssid, probe_wait_ms);
2099 #endif
2100                         ieee80211_sta_connection_lost(sdata, bssid);
2101                 } else if (ifmgd->probe_send_count < max_tries) {
2102 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2103                         wiphy_debug(local->hw.wiphy,
2104                                     "%s: No probe response from AP %pM"
2105                                     " after %dms, try %d/%i\n",
2106                                     sdata->name,
2107                                     bssid, probe_wait_ms,
2108                                     ifmgd->probe_send_count, max_tries);
2109 #endif
2110                         ieee80211_mgd_probe_ap_send(sdata);
2111                 } else {
2112                         /*
2113                          * We actually lost the connection ... or did we?
2114                          * Let's make sure!
2115                          */
2116                         wiphy_debug(local->hw.wiphy,
2117                                     "%s: No probe response from AP %pM"
2118                                     " after %dms, disconnecting.\n",
2119                                     sdata->name,
2120                                     bssid, probe_wait_ms);
2121
2122                         ieee80211_sta_connection_lost(sdata, bssid);
2123                 }
2124         }
2125
2126         mutex_unlock(&ifmgd->mtx);
2127 }
2128
2129 static void ieee80211_sta_bcn_mon_timer(unsigned long data)
2130 {
2131         struct ieee80211_sub_if_data *sdata =
2132                 (struct ieee80211_sub_if_data *) data;
2133         struct ieee80211_local *local = sdata->local;
2134
2135         if (local->quiescing)
2136                 return;
2137
2138         ieee80211_queue_work(&sdata->local->hw,
2139                              &sdata->u.mgd.beacon_connection_loss_work);
2140 }
2141
2142 static void ieee80211_sta_conn_mon_timer(unsigned long data)
2143 {
2144         struct ieee80211_sub_if_data *sdata =
2145                 (struct ieee80211_sub_if_data *) data;
2146         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2147         struct ieee80211_local *local = sdata->local;
2148
2149         if (local->quiescing)
2150                 return;
2151
2152         ieee80211_queue_work(&local->hw, &ifmgd->monitor_work);
2153 }
2154
2155 static void ieee80211_sta_monitor_work(struct work_struct *work)
2156 {
2157         struct ieee80211_sub_if_data *sdata =
2158                 container_of(work, struct ieee80211_sub_if_data,
2159                              u.mgd.monitor_work);
2160
2161         ieee80211_mgd_probe_ap(sdata, false);
2162 }
2163
2164 static void ieee80211_restart_sta_timer(struct ieee80211_sub_if_data *sdata)
2165 {
2166         if (sdata->vif.type == NL80211_IFTYPE_STATION) {
2167                 sdata->u.mgd.flags &= ~(IEEE80211_STA_BEACON_POLL |
2168                                         IEEE80211_STA_CONNECTION_POLL);
2169
2170                 /* let's probe the connection once */
2171                 ieee80211_queue_work(&sdata->local->hw,
2172                            &sdata->u.mgd.monitor_work);
2173                 /* and do all the other regular work too */
2174                 ieee80211_queue_work(&sdata->local->hw, &sdata->work);
2175         }
2176 }
2177
2178 #ifdef CONFIG_PM
2179 void ieee80211_sta_quiesce(struct ieee80211_sub_if_data *sdata)
2180 {
2181         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2182
2183         /*
2184          * we need to use atomic bitops for the running bits
2185          * only because both timers might fire at the same
2186          * time -- the code here is properly synchronised.
2187          */
2188
2189         cancel_work_sync(&ifmgd->request_smps_work);
2190
2191         cancel_work_sync(&ifmgd->beacon_connection_loss_work);
2192         if (del_timer_sync(&ifmgd->timer))
2193                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
2194
2195         cancel_work_sync(&ifmgd->chswitch_work);
2196         if (del_timer_sync(&ifmgd->chswitch_timer))
2197                 set_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running);
2198
2199         cancel_work_sync(&ifmgd->monitor_work);
2200         /* these will just be re-established on connection */
2201         del_timer_sync(&ifmgd->conn_mon_timer);
2202         del_timer_sync(&ifmgd->bcn_mon_timer);
2203 }
2204
2205 void ieee80211_sta_restart(struct ieee80211_sub_if_data *sdata)
2206 {
2207         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2208
2209         if (test_and_clear_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running))
2210                 add_timer(&ifmgd->timer);
2211         if (test_and_clear_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running))
2212                 add_timer(&ifmgd->chswitch_timer);
2213         ieee80211_sta_reset_beacon_monitor(sdata);
2214         ieee80211_restart_sta_timer(sdata);
2215 }
2216 #endif
2217
2218 /* interface setup */
2219 void ieee80211_sta_setup_sdata(struct ieee80211_sub_if_data *sdata)
2220 {
2221         struct ieee80211_if_managed *ifmgd;
2222
2223         ifmgd = &sdata->u.mgd;
2224         INIT_WORK(&ifmgd->monitor_work, ieee80211_sta_monitor_work);
2225         INIT_WORK(&ifmgd->chswitch_work, ieee80211_chswitch_work);
2226         INIT_WORK(&ifmgd->beacon_connection_loss_work,
2227                   ieee80211_beacon_connection_loss_work);
2228         INIT_WORK(&ifmgd->request_smps_work, ieee80211_request_smps_work);
2229         setup_timer(&ifmgd->timer, ieee80211_sta_timer,
2230                     (unsigned long) sdata);
2231         setup_timer(&ifmgd->bcn_mon_timer, ieee80211_sta_bcn_mon_timer,
2232                     (unsigned long) sdata);
2233         setup_timer(&ifmgd->conn_mon_timer, ieee80211_sta_conn_mon_timer,
2234                     (unsigned long) sdata);
2235         setup_timer(&ifmgd->chswitch_timer, ieee80211_chswitch_timer,
2236                     (unsigned long) sdata);
2237
2238         ifmgd->flags = 0;
2239
2240         mutex_init(&ifmgd->mtx);
2241
2242         if (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_SMPS)
2243                 ifmgd->req_smps = IEEE80211_SMPS_AUTOMATIC;
2244         else
2245                 ifmgd->req_smps = IEEE80211_SMPS_OFF;
2246 }
2247
2248 /* scan finished notification */
2249 void ieee80211_mlme_notify_scan_completed(struct ieee80211_local *local)
2250 {
2251         struct ieee80211_sub_if_data *sdata = local->scan_sdata;
2252
2253         /* Restart STA timers */
2254         rcu_read_lock();
2255         list_for_each_entry_rcu(sdata, &local->interfaces, list)
2256                 ieee80211_restart_sta_timer(sdata);
2257         rcu_read_unlock();
2258 }
2259
2260 int ieee80211_max_network_latency(struct notifier_block *nb,
2261                                   unsigned long data, void *dummy)
2262 {
2263         s32 latency_usec = (s32) data;
2264         struct ieee80211_local *local =
2265                 container_of(nb, struct ieee80211_local,
2266                              network_latency_notifier);
2267
2268         mutex_lock(&local->iflist_mtx);
2269         ieee80211_recalc_ps(local, latency_usec);
2270         mutex_unlock(&local->iflist_mtx);
2271
2272         return 0;
2273 }
2274
2275 /* config hooks */
2276 static enum work_done_result
2277 ieee80211_probe_auth_done(struct ieee80211_work *wk,
2278                           struct sk_buff *skb)
2279 {
2280         if (!skb) {
2281                 cfg80211_send_auth_timeout(wk->sdata->dev, wk->filter_ta);
2282                 return WORK_DONE_DESTROY;
2283         }
2284
2285         if (wk->type == IEEE80211_WORK_AUTH) {
2286                 cfg80211_send_rx_auth(wk->sdata->dev, skb->data, skb->len);
2287                 return WORK_DONE_DESTROY;
2288         }
2289
2290         mutex_lock(&wk->sdata->u.mgd.mtx);
2291         ieee80211_rx_mgmt_probe_resp(wk->sdata, skb);
2292         mutex_unlock(&wk->sdata->u.mgd.mtx);
2293
2294         wk->type = IEEE80211_WORK_AUTH;
2295         wk->probe_auth.tries = 0;
2296         return WORK_DONE_REQUEUE;
2297 }
2298
2299 int ieee80211_mgd_auth(struct ieee80211_sub_if_data *sdata,
2300                        struct cfg80211_auth_request *req)
2301 {
2302         const u8 *ssid;
2303         struct ieee80211_work *wk;
2304         u16 auth_alg;
2305
2306         if (req->local_state_change)
2307                 return 0; /* no need to update mac80211 state */
2308
2309         switch (req->auth_type) {
2310         case NL80211_AUTHTYPE_OPEN_SYSTEM:
2311                 auth_alg = WLAN_AUTH_OPEN;
2312                 break;
2313         case NL80211_AUTHTYPE_SHARED_KEY:
2314                 if (IS_ERR(sdata->local->wep_tx_tfm))
2315                         return -EOPNOTSUPP;
2316                 auth_alg = WLAN_AUTH_SHARED_KEY;
2317                 break;
2318         case NL80211_AUTHTYPE_FT:
2319                 auth_alg = WLAN_AUTH_FT;
2320                 break;
2321         case NL80211_AUTHTYPE_NETWORK_EAP:
2322                 auth_alg = WLAN_AUTH_LEAP;
2323                 break;
2324         default:
2325                 return -EOPNOTSUPP;
2326         }
2327
2328         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
2329         if (!wk)
2330                 return -ENOMEM;
2331
2332         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);
2333
2334         if (req->ie && req->ie_len) {
2335                 memcpy(wk->ie, req->ie, req->ie_len);
2336                 wk->ie_len = req->ie_len;
2337         }
2338
2339         if (req->key && req->key_len) {
2340                 wk->probe_auth.key_len = req->key_len;
2341                 wk->probe_auth.key_idx = req->key_idx;
2342                 memcpy(wk->probe_auth.key, req->key, req->key_len);
2343         }
2344
2345         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
2346         memcpy(wk->probe_auth.ssid, ssid + 2, ssid[1]);
2347         wk->probe_auth.ssid_len = ssid[1];
2348
2349         wk->probe_auth.algorithm = auth_alg;
2350         wk->probe_auth.privacy = req->bss->capability & WLAN_CAPABILITY_PRIVACY;
2351
2352         /* if we already have a probe, don't probe again */
2353         if (req->bss->proberesp_ies)
2354                 wk->type = IEEE80211_WORK_AUTH;
2355         else
2356                 wk->type = IEEE80211_WORK_DIRECT_PROBE;
2357         wk->chan = req->bss->channel;
2358         wk->chan_type = NL80211_CHAN_NO_HT;
2359         wk->sdata = sdata;
2360         wk->done = ieee80211_probe_auth_done;
2361
2362         ieee80211_add_work(wk);
2363         return 0;
2364 }
2365
2366 static enum work_done_result ieee80211_assoc_done(struct ieee80211_work *wk,
2367                                                   struct sk_buff *skb)
2368 {
2369         struct ieee80211_mgmt *mgmt;
2370         struct ieee80211_rx_status *rx_status;
2371         struct ieee802_11_elems elems;
2372         u16 status;
2373
2374         if (!skb) {
2375                 cfg80211_send_assoc_timeout(wk->sdata->dev, wk->filter_ta);
2376                 return WORK_DONE_DESTROY;
2377         }
2378
2379         if (wk->type == IEEE80211_WORK_ASSOC_BEACON_WAIT) {
2380                 mutex_lock(&wk->sdata->u.mgd.mtx);
2381                 rx_status = (void *) skb->cb;
2382                 ieee802_11_parse_elems(skb->data + 24 + 12, skb->len - 24 - 12, &elems);
2383                 ieee80211_rx_bss_info(wk->sdata, (void *)skb->data, skb->len, rx_status,
2384                                       &elems, true);
2385                 mutex_unlock(&wk->sdata->u.mgd.mtx);
2386
2387                 wk->type = IEEE80211_WORK_ASSOC;
2388                 /* not really done yet */
2389                 return WORK_DONE_REQUEUE;
2390         }
2391
2392         mgmt = (void *)skb->data;
2393         status = le16_to_cpu(mgmt->u.assoc_resp.status_code);
2394
2395         if (status == WLAN_STATUS_SUCCESS) {
2396                 mutex_lock(&wk->sdata->u.mgd.mtx);
2397                 if (!ieee80211_assoc_success(wk, mgmt, skb->len)) {
2398                         mutex_unlock(&wk->sdata->u.mgd.mtx);
2399                         /* oops -- internal error -- send timeout for now */
2400                         cfg80211_send_assoc_timeout(wk->sdata->dev,
2401                                                     wk->filter_ta);
2402                         return WORK_DONE_DESTROY;
2403                 }
2404
2405                 mutex_unlock(&wk->sdata->u.mgd.mtx);
2406         }
2407
2408         cfg80211_send_rx_assoc(wk->sdata->dev, skb->data, skb->len);
2409         return WORK_DONE_DESTROY;
2410 }
2411
2412 int ieee80211_mgd_assoc(struct ieee80211_sub_if_data *sdata,
2413                         struct cfg80211_assoc_request *req)
2414 {
2415         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2416         struct ieee80211_bss *bss = (void *)req->bss->priv;
2417         struct ieee80211_work *wk;
2418         const u8 *ssid;
2419         int i;
2420
2421         mutex_lock(&ifmgd->mtx);
2422         if (ifmgd->associated) {
2423                 if (!req->prev_bssid ||
2424                     memcmp(req->prev_bssid, ifmgd->associated->bssid,
2425                            ETH_ALEN)) {
2426                         /*
2427                          * We are already associated and the request was not a
2428                          * reassociation request from the current BSS, so
2429                          * reject it.
2430                          */
2431                         mutex_unlock(&ifmgd->mtx);
2432                         return -EALREADY;
2433                 }
2434
2435                 /* Trying to reassociate - clear previous association state */
2436                 ieee80211_set_disassoc(sdata, true, false);
2437         }
2438         mutex_unlock(&ifmgd->mtx);
2439
2440         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
2441         if (!wk)
2442                 return -ENOMEM;
2443
2444         ifmgd->flags &= ~IEEE80211_STA_DISABLE_11N;
2445         ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
2446
2447         ifmgd->beacon_crc_valid = false;
2448
2449         for (i = 0; i < req->crypto.n_ciphers_pairwise; i++)
2450                 if (req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP40 ||
2451                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_TKIP ||
2452                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP104)
2453                         ifmgd->flags |= IEEE80211_STA_DISABLE_11N;
2454
2455
2456         if (req->ie && req->ie_len) {
2457                 memcpy(wk->ie, req->ie, req->ie_len);
2458                 wk->ie_len = req->ie_len;
2459         } else
2460                 wk->ie_len = 0;
2461
2462         wk->assoc.bss = req->bss;
2463
2464         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);
2465
2466         /* new association always uses requested smps mode */
2467         if (ifmgd->req_smps == IEEE80211_SMPS_AUTOMATIC) {
2468                 if (ifmgd->powersave)
2469                         ifmgd->ap_smps = IEEE80211_SMPS_DYNAMIC;
2470                 else
2471                         ifmgd->ap_smps = IEEE80211_SMPS_OFF;
2472         } else
2473                 ifmgd->ap_smps = ifmgd->req_smps;
2474
2475         wk->assoc.smps = ifmgd->ap_smps;
2476         /*
2477          * IEEE802.11n does not allow TKIP/WEP as pairwise ciphers in HT mode.
2478          * We still associate in non-HT mode (11a/b/g) if any one of these
2479          * ciphers is configured as pairwise.
2480          * We can set this to true for non-11n hardware, that'll be checked
2481          * separately along with the peer capabilities.
2482          */
2483         wk->assoc.use_11n = !(ifmgd->flags & IEEE80211_STA_DISABLE_11N);
2484         wk->assoc.capability = req->bss->capability;
2485         wk->assoc.wmm_used = bss->wmm_used;
2486         wk->assoc.supp_rates = bss->supp_rates;
2487         wk->assoc.supp_rates_len = bss->supp_rates_len;
2488         wk->assoc.ht_information_ie =
2489                 ieee80211_bss_get_ie(req->bss, WLAN_EID_HT_INFORMATION);
2490
2491         if (bss->wmm_used && bss->uapsd_supported &&
2492             (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_UAPSD)) {
2493                 wk->assoc.uapsd_used = true;
2494                 ifmgd->flags |= IEEE80211_STA_UAPSD_ENABLED;
2495         } else {
2496                 wk->assoc.uapsd_used = false;
2497                 ifmgd->flags &= ~IEEE80211_STA_UAPSD_ENABLED;
2498         }
2499
2500         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
2501         memcpy(wk->assoc.ssid, ssid + 2, ssid[1]);
2502         wk->assoc.ssid_len = ssid[1];
2503
2504         if (req->prev_bssid)
2505                 memcpy(wk->assoc.prev_bssid, req->prev_bssid, ETH_ALEN);
2506
2507         wk->chan = req->bss->channel;
2508         wk->chan_type = NL80211_CHAN_NO_HT;
2509         wk->sdata = sdata;
2510         wk->done = ieee80211_assoc_done;
2511         if (!bss->dtim_period &&
2512             sdata->local->hw.flags & IEEE80211_HW_NEED_DTIM_PERIOD)
2513                 wk->type = IEEE80211_WORK_ASSOC_BEACON_WAIT;
2514         else
2515                 wk->type = IEEE80211_WORK_ASSOC;
2516
2517         if (req->use_mfp) {
2518                 ifmgd->mfp = IEEE80211_MFP_REQUIRED;
2519                 ifmgd->flags |= IEEE80211_STA_MFP_ENABLED;
2520         } else {
2521                 ifmgd->mfp = IEEE80211_MFP_DISABLED;
2522                 ifmgd->flags &= ~IEEE80211_STA_MFP_ENABLED;
2523         }
2524
2525         if (req->crypto.control_port)
2526                 ifmgd->flags |= IEEE80211_STA_CONTROL_PORT;
2527         else
2528                 ifmgd->flags &= ~IEEE80211_STA_CONTROL_PORT;
2529
2530         sdata->control_port_protocol = req->crypto.control_port_ethertype;
2531         sdata->control_port_no_encrypt = req->crypto.control_port_no_encrypt;
2532
2533         ieee80211_add_work(wk);
2534         return 0;
2535 }
2536
2537 int ieee80211_mgd_deauth(struct ieee80211_sub_if_data *sdata,
2538                          struct cfg80211_deauth_request *req,
2539                          void *cookie)
2540 {
2541         struct ieee80211_local *local = sdata->local;
2542         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2543         struct ieee80211_work *wk;
2544         u8 bssid[ETH_ALEN];
2545         bool assoc_bss = false;
2546
2547         mutex_lock(&ifmgd->mtx);
2548
2549         memcpy(bssid, req->bss->bssid, ETH_ALEN);
2550         if (ifmgd->associated == req->bss) {
2551                 ieee80211_set_disassoc(sdata, false, true);
2552                 mutex_unlock(&ifmgd->mtx);
2553                 assoc_bss = true;
2554         } else {
2555                 bool not_auth_yet = false;
2556
2557                 mutex_unlock(&ifmgd->mtx);
2558
2559                 mutex_lock(&local->mtx);
2560                 list_for_each_entry(wk, &local->work_list, list) {
2561                         if (wk->sdata != sdata)
2562                                 continue;
2563
2564                         if (wk->type != IEEE80211_WORK_DIRECT_PROBE &&
2565                             wk->type != IEEE80211_WORK_AUTH &&
2566                             wk->type != IEEE80211_WORK_ASSOC &&
2567                             wk->type != IEEE80211_WORK_ASSOC_BEACON_WAIT)
2568                                 continue;
2569
2570                         if (memcmp(req->bss->bssid, wk->filter_ta, ETH_ALEN))
2571                                 continue;
2572
2573                         not_auth_yet = wk->type == IEEE80211_WORK_DIRECT_PROBE;
2574                         list_del_rcu(&wk->list);
2575                         free_work(wk);
2576                         break;
2577                 }
2578                 mutex_unlock(&local->mtx);
2579
2580                 /*
2581                  * If somebody requests authentication and we haven't
2582                  * sent out an auth frame yet there's no need to send
2583                  * out a deauth frame either. If the state was PROBE,
2584                  * then this is the case. If it's AUTH we have sent a
2585                  * frame, and if it's IDLE we have completed the auth
2586                  * process already.
2587                  */
2588                 if (not_auth_yet) {
2589                         __cfg80211_auth_canceled(sdata->dev, bssid);
2590                         return 0;
2591                 }
2592         }
2593
2594         printk(KERN_DEBUG "%s: deauthenticating from %pM by local choice (reason=%d)\n",
2595                sdata->name, bssid, req->reason_code);
2596
2597         ieee80211_send_deauth_disassoc(sdata, bssid, IEEE80211_STYPE_DEAUTH,
2598                                        req->reason_code, cookie,
2599                                        !req->local_state_change);
2600         if (assoc_bss)
2601                 sta_info_destroy_addr(sdata, bssid);
2602
2603         mutex_lock(&sdata->local->mtx);
2604         ieee80211_recalc_idle(sdata->local);
2605         mutex_unlock(&sdata->local->mtx);
2606
2607         return 0;
2608 }
2609
2610 int ieee80211_mgd_disassoc(struct ieee80211_sub_if_data *sdata,
2611                            struct cfg80211_disassoc_request *req,
2612                            void *cookie)
2613 {
2614         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2615         u8 bssid[ETH_ALEN];
2616
2617         mutex_lock(&ifmgd->mtx);
2618
2619         /*
2620          * cfg80211 should catch this ... but it's racy since
2621          * we can receive a disassoc frame, process it, hand it
2622          * to cfg80211 while that's in a locked section already
2623          * trying to tell us that the user wants to disconnect.
2624          */
2625         if (ifmgd->associated != req->bss) {
2626                 mutex_unlock(&ifmgd->mtx);
2627                 return -ENOLINK;
2628         }
2629
2630         printk(KERN_DEBUG "%s: disassociating from %pM by local choice (reason=%d)\n",
2631                sdata->name, req->bss->bssid, req->reason_code);
2632
2633         memcpy(bssid, req->bss->bssid, ETH_ALEN);
2634         ieee80211_set_disassoc(sdata, false, true);
2635
2636         mutex_unlock(&ifmgd->mtx);
2637
2638         ieee80211_send_deauth_disassoc(sdata, req->bss->bssid,
2639                         IEEE80211_STYPE_DISASSOC, req->reason_code,
2640                         cookie, !req->local_state_change);
2641         sta_info_destroy_addr(sdata, bssid);
2642
2643         mutex_lock(&sdata->local->mtx);
2644         ieee80211_recalc_idle(sdata->local);
2645         mutex_unlock(&sdata->local->mtx);
2646
2647         return 0;
2648 }
2649
2650 void ieee80211_cqm_rssi_notify(struct ieee80211_vif *vif,
2651                                enum nl80211_cqm_rssi_threshold_event rssi_event,
2652                                gfp_t gfp)
2653 {
2654         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2655
2656         trace_api_cqm_rssi_notify(sdata, rssi_event);
2657
2658         cfg80211_cqm_rssi_notify(sdata->dev, rssi_event, gfp);
2659 }
2660 EXPORT_SYMBOL(ieee80211_cqm_rssi_notify);