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
1099         ieee80211_hw_config(local, config_changed);
1100
1101         /* Disable ARP filtering */
1102         if (sdata->vif.bss_conf.arp_filter_enabled) {
1103                 sdata->vif.bss_conf.arp_filter_enabled = false;
1104                 changed |= BSS_CHANGED_ARP_FILTER;
1105         }
1106
1107         /* The BSSID (not really interesting) and HT changed */
1108         changed |= BSS_CHANGED_BSSID | BSS_CHANGED_HT;
1109         ieee80211_bss_info_change_notify(sdata, changed);
1110
1111         if (remove_sta)
1112                 sta_info_destroy_addr(sdata, bssid);
1113
1114         del_timer_sync(&sdata->u.mgd.conn_mon_timer);
1115         del_timer_sync(&sdata->u.mgd.bcn_mon_timer);
1116         del_timer_sync(&sdata->u.mgd.timer);
1117         del_timer_sync(&sdata->u.mgd.chswitch_timer);
1118 }
1119
1120 void ieee80211_sta_rx_notify(struct ieee80211_sub_if_data *sdata,
1121                              struct ieee80211_hdr *hdr)
1122 {
1123         /*
1124          * We can postpone the mgd.timer whenever receiving unicast frames
1125          * from AP because we know that the connection is working both ways
1126          * at that time. But multicast frames (and hence also beacons) must
1127          * be ignored here, because we need to trigger the timer during
1128          * data idle periods for sending the periodic probe request to the
1129          * AP we're connected to.
1130          */
1131         if (is_multicast_ether_addr(hdr->addr1))
1132                 return;
1133
1134         ieee80211_sta_reset_conn_monitor(sdata);
1135 }
1136
1137 static void ieee80211_reset_ap_probe(struct ieee80211_sub_if_data *sdata)
1138 {
1139         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1140
1141         if (!(ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1142                               IEEE80211_STA_CONNECTION_POLL)))
1143             return;
1144
1145         ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
1146                           IEEE80211_STA_BEACON_POLL);
1147         mutex_lock(&sdata->local->iflist_mtx);
1148         ieee80211_recalc_ps(sdata->local, -1);
1149         mutex_unlock(&sdata->local->iflist_mtx);
1150
1151         if (sdata->local->hw.flags & IEEE80211_HW_CONNECTION_MONITOR)
1152                 return;
1153
1154         /*
1155          * We've received a probe response, but are not sure whether
1156          * we have or will be receiving any beacons or data, so let's
1157          * schedule the timers again, just in case.
1158          */
1159         ieee80211_sta_reset_beacon_monitor(sdata);
1160
1161         mod_timer(&ifmgd->conn_mon_timer,
1162                   round_jiffies_up(jiffies +
1163                                    IEEE80211_CONNECTION_IDLE_TIME));
1164 }
1165
1166 void ieee80211_sta_tx_notify(struct ieee80211_sub_if_data *sdata,
1167                              struct ieee80211_hdr *hdr, bool ack)
1168 {
1169         if (!ieee80211_is_data(hdr->frame_control))
1170             return;
1171
1172         if (ack)
1173                 ieee80211_sta_reset_conn_monitor(sdata);
1174
1175         if (ieee80211_is_nullfunc(hdr->frame_control) &&
1176             sdata->u.mgd.probe_send_count > 0) {
1177                 if (ack)
1178                         sdata->u.mgd.probe_send_count = 0;
1179                 else
1180                         sdata->u.mgd.nullfunc_failed = true;
1181                 ieee80211_queue_work(&sdata->local->hw, &sdata->work);
1182         }
1183 }
1184
1185 static void ieee80211_mgd_probe_ap_send(struct ieee80211_sub_if_data *sdata)
1186 {
1187         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1188         const u8 *ssid;
1189         u8 *dst = ifmgd->associated->bssid;
1190         u8 unicast_limit = max(1, max_probe_tries - 3);
1191
1192         /*
1193          * Try sending broadcast probe requests for the last three
1194          * probe requests after the first ones failed since some
1195          * buggy APs only support broadcast probe requests.
1196          */
1197         if (ifmgd->probe_send_count >= unicast_limit)
1198                 dst = NULL;
1199
1200         /*
1201          * When the hardware reports an accurate Tx ACK status, it's
1202          * better to send a nullfunc frame instead of a probe request,
1203          * as it will kick us off the AP quickly if we aren't associated
1204          * anymore. The timeout will be reset if the frame is ACKed by
1205          * the AP.
1206          */
1207         if (sdata->local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS) {
1208                 ifmgd->nullfunc_failed = false;
1209                 ieee80211_send_nullfunc(sdata->local, sdata, 0);
1210         } else {
1211                 ssid = ieee80211_bss_get_ie(ifmgd->associated, WLAN_EID_SSID);
1212                 ieee80211_send_probe_req(sdata, dst, ssid + 2, ssid[1], NULL, 0);
1213         }
1214
1215         ifmgd->probe_send_count++;
1216         ifmgd->probe_timeout = jiffies + msecs_to_jiffies(probe_wait_ms);
1217         run_again(ifmgd, ifmgd->probe_timeout);
1218 }
1219
1220 static void ieee80211_mgd_probe_ap(struct ieee80211_sub_if_data *sdata,
1221                                    bool beacon)
1222 {
1223         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1224         bool already = false;
1225
1226         if (!ieee80211_sdata_running(sdata))
1227                 return;
1228
1229         if (sdata->local->scanning)
1230                 return;
1231
1232         if (sdata->local->tmp_channel)
1233                 return;
1234
1235         mutex_lock(&ifmgd->mtx);
1236
1237         if (!ifmgd->associated)
1238                 goto out;
1239
1240 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1241         if (beacon && net_ratelimit())
1242                 printk(KERN_DEBUG "%s: detected beacon loss from AP "
1243                        "- sending probe request\n", sdata->name);
1244 #endif
1245
1246         /*
1247          * The driver/our work has already reported this event or the
1248          * connection monitoring has kicked in and we have already sent
1249          * a probe request. Or maybe the AP died and the driver keeps
1250          * reporting until we disassociate...
1251          *
1252          * In either case we have to ignore the current call to this
1253          * function (except for setting the correct probe reason bit)
1254          * because otherwise we would reset the timer every time and
1255          * never check whether we received a probe response!
1256          */
1257         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1258                             IEEE80211_STA_CONNECTION_POLL))
1259                 already = true;
1260
1261         if (beacon)
1262                 ifmgd->flags |= IEEE80211_STA_BEACON_POLL;
1263         else
1264                 ifmgd->flags |= IEEE80211_STA_CONNECTION_POLL;
1265
1266         if (already)
1267                 goto out;
1268
1269         mutex_lock(&sdata->local->iflist_mtx);
1270         ieee80211_recalc_ps(sdata->local, -1);
1271         mutex_unlock(&sdata->local->iflist_mtx);
1272
1273         ifmgd->probe_send_count = 0;
1274         ieee80211_mgd_probe_ap_send(sdata);
1275  out:
1276         mutex_unlock(&ifmgd->mtx);
1277 }
1278
1279 struct sk_buff *ieee80211_ap_probereq_get(struct ieee80211_hw *hw,
1280                                           struct ieee80211_vif *vif)
1281 {
1282         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1283         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1284         struct sk_buff *skb;
1285         const u8 *ssid;
1286
1287         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
1288                 return NULL;
1289
1290         ASSERT_MGD_MTX(ifmgd);
1291
1292         if (!ifmgd->associated)
1293                 return NULL;
1294
1295         ssid = ieee80211_bss_get_ie(ifmgd->associated, WLAN_EID_SSID);
1296         skb = ieee80211_build_probe_req(sdata, ifmgd->associated->bssid,
1297                                         ssid + 2, ssid[1], NULL, 0);
1298
1299         return skb;
1300 }
1301 EXPORT_SYMBOL(ieee80211_ap_probereq_get);
1302
1303 static void __ieee80211_connection_loss(struct ieee80211_sub_if_data *sdata)
1304 {
1305         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1306         struct ieee80211_local *local = sdata->local;
1307         u8 bssid[ETH_ALEN];
1308
1309         mutex_lock(&ifmgd->mtx);
1310         if (!ifmgd->associated) {
1311                 mutex_unlock(&ifmgd->mtx);
1312                 return;
1313         }
1314
1315         memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
1316
1317         printk(KERN_DEBUG "%s: Connection to AP %pM lost.\n",
1318                sdata->name, bssid);
1319
1320         ieee80211_set_disassoc(sdata, true, true);
1321         mutex_unlock(&ifmgd->mtx);
1322
1323         mutex_lock(&local->mtx);
1324         ieee80211_recalc_idle(local);
1325         mutex_unlock(&local->mtx);
1326         /*
1327          * must be outside lock due to cfg80211,
1328          * but that's not a problem.
1329          */
1330         ieee80211_send_deauth_disassoc(sdata, bssid,
1331                                        IEEE80211_STYPE_DEAUTH,
1332                                        WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
1333                                        NULL, true);
1334 }
1335
1336 void ieee80211_beacon_connection_loss_work(struct work_struct *work)
1337 {
1338         struct ieee80211_sub_if_data *sdata =
1339                 container_of(work, struct ieee80211_sub_if_data,
1340                              u.mgd.beacon_connection_loss_work);
1341
1342         if (sdata->local->hw.flags & IEEE80211_HW_CONNECTION_MONITOR)
1343                 __ieee80211_connection_loss(sdata);
1344         else
1345                 ieee80211_mgd_probe_ap(sdata, true);
1346 }
1347
1348 void ieee80211_beacon_loss(struct ieee80211_vif *vif)
1349 {
1350         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1351         struct ieee80211_hw *hw = &sdata->local->hw;
1352
1353         trace_api_beacon_loss(sdata);
1354
1355         WARN_ON(hw->flags & IEEE80211_HW_CONNECTION_MONITOR);
1356         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
1357 }
1358 EXPORT_SYMBOL(ieee80211_beacon_loss);
1359
1360 void ieee80211_connection_loss(struct ieee80211_vif *vif)
1361 {
1362         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1363         struct ieee80211_hw *hw = &sdata->local->hw;
1364
1365         trace_api_connection_loss(sdata);
1366
1367         WARN_ON(!(hw->flags & IEEE80211_HW_CONNECTION_MONITOR));
1368         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
1369 }
1370 EXPORT_SYMBOL(ieee80211_connection_loss);
1371
1372
1373 static enum rx_mgmt_action __must_check
1374 ieee80211_rx_mgmt_deauth(struct ieee80211_sub_if_data *sdata,
1375                          struct ieee80211_mgmt *mgmt, size_t len)
1376 {
1377         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1378         const u8 *bssid = NULL;
1379         u16 reason_code;
1380
1381         if (len < 24 + 2)
1382                 return RX_MGMT_NONE;
1383
1384         ASSERT_MGD_MTX(ifmgd);
1385
1386         bssid = ifmgd->associated->bssid;
1387
1388         reason_code = le16_to_cpu(mgmt->u.deauth.reason_code);
1389
1390         printk(KERN_DEBUG "%s: deauthenticated from %pM (Reason: %u)\n",
1391                         sdata->name, bssid, reason_code);
1392
1393         ieee80211_set_disassoc(sdata, true, false);
1394         mutex_lock(&sdata->local->mtx);
1395         ieee80211_recalc_idle(sdata->local);
1396         mutex_unlock(&sdata->local->mtx);
1397
1398         return RX_MGMT_CFG80211_DEAUTH;
1399 }
1400
1401
1402 static enum rx_mgmt_action __must_check
1403 ieee80211_rx_mgmt_disassoc(struct ieee80211_sub_if_data *sdata,
1404                            struct ieee80211_mgmt *mgmt, size_t len)
1405 {
1406         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1407         u16 reason_code;
1408
1409         if (len < 24 + 2)
1410                 return RX_MGMT_NONE;
1411
1412         ASSERT_MGD_MTX(ifmgd);
1413
1414         if (WARN_ON(!ifmgd->associated))
1415                 return RX_MGMT_NONE;
1416
1417         if (WARN_ON(memcmp(ifmgd->associated->bssid, mgmt->sa, ETH_ALEN)))
1418                 return RX_MGMT_NONE;
1419
1420         reason_code = le16_to_cpu(mgmt->u.disassoc.reason_code);
1421
1422         printk(KERN_DEBUG "%s: disassociated from %pM (Reason: %u)\n",
1423                         sdata->name, mgmt->sa, reason_code);
1424
1425         ieee80211_set_disassoc(sdata, true, false);
1426         mutex_lock(&sdata->local->mtx);
1427         ieee80211_recalc_idle(sdata->local);
1428         mutex_unlock(&sdata->local->mtx);
1429         return RX_MGMT_CFG80211_DISASSOC;
1430 }
1431
1432
1433 static bool ieee80211_assoc_success(struct ieee80211_work *wk,
1434                                     struct ieee80211_mgmt *mgmt, size_t len)
1435 {
1436         struct ieee80211_sub_if_data *sdata = wk->sdata;
1437         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1438         struct ieee80211_local *local = sdata->local;
1439         struct ieee80211_supported_band *sband;
1440         struct sta_info *sta;
1441         struct cfg80211_bss *cbss = wk->assoc.bss;
1442         u8 *pos;
1443         u32 rates, basic_rates;
1444         u16 capab_info, aid;
1445         struct ieee802_11_elems elems;
1446         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
1447         u32 changed = 0;
1448         int i, j, err;
1449         bool have_higher_than_11mbit = false;
1450         u16 ap_ht_cap_flags;
1451
1452         /* AssocResp and ReassocResp have identical structure */
1453
1454         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
1455         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
1456
1457         if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14)))
1458                 printk(KERN_DEBUG "%s: invalid aid value %d; bits 15:14 not "
1459                        "set\n", sdata->name, aid);
1460         aid &= ~(BIT(15) | BIT(14));
1461
1462         pos = mgmt->u.assoc_resp.variable;
1463         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems);
1464
1465         if (!elems.supp_rates) {
1466                 printk(KERN_DEBUG "%s: no SuppRates element in AssocResp\n",
1467                        sdata->name);
1468                 return false;
1469         }
1470
1471         ifmgd->aid = aid;
1472
1473         sta = sta_info_alloc(sdata, cbss->bssid, GFP_KERNEL);
1474         if (!sta) {
1475                 printk(KERN_DEBUG "%s: failed to alloc STA entry for"
1476                        " the AP\n", sdata->name);
1477                 return false;
1478         }
1479
1480         set_sta_flags(sta, WLAN_STA_AUTH | WLAN_STA_ASSOC |
1481                            WLAN_STA_ASSOC_AP);
1482         if (!(ifmgd->flags & IEEE80211_STA_CONTROL_PORT))
1483                 set_sta_flags(sta, WLAN_STA_AUTHORIZED);
1484
1485         rates = 0;
1486         basic_rates = 0;
1487         sband = local->hw.wiphy->bands[wk->chan->band];
1488
1489         for (i = 0; i < elems.supp_rates_len; i++) {
1490                 int rate = (elems.supp_rates[i] & 0x7f) * 5;
1491                 bool is_basic = !!(elems.supp_rates[i] & 0x80);
1492
1493                 if (rate > 110)
1494                         have_higher_than_11mbit = true;
1495
1496                 for (j = 0; j < sband->n_bitrates; j++) {
1497                         if (sband->bitrates[j].bitrate == rate) {
1498                                 rates |= BIT(j);
1499                                 if (is_basic)
1500                                         basic_rates |= BIT(j);
1501                                 break;
1502                         }
1503                 }
1504         }
1505
1506         for (i = 0; i < elems.ext_supp_rates_len; i++) {
1507                 int rate = (elems.ext_supp_rates[i] & 0x7f) * 5;
1508                 bool is_basic = !!(elems.ext_supp_rates[i] & 0x80);
1509
1510                 if (rate > 110)
1511                         have_higher_than_11mbit = true;
1512
1513                 for (j = 0; j < sband->n_bitrates; j++) {
1514                         if (sband->bitrates[j].bitrate == rate) {
1515                                 rates |= BIT(j);
1516                                 if (is_basic)
1517                                         basic_rates |= BIT(j);
1518                                 break;
1519                         }
1520                 }
1521         }
1522
1523         sta->sta.supp_rates[wk->chan->band] = rates;
1524         sdata->vif.bss_conf.basic_rates = basic_rates;
1525
1526         /* cf. IEEE 802.11 9.2.12 */
1527         if (wk->chan->band == IEEE80211_BAND_2GHZ &&
1528             have_higher_than_11mbit)
1529                 sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
1530         else
1531                 sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;
1532
1533         if (elems.ht_cap_elem && !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1534                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1535                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1536
1537         ap_ht_cap_flags = sta->sta.ht_cap.cap;
1538
1539         rate_control_rate_init(sta);
1540
1541         if (ifmgd->flags & IEEE80211_STA_MFP_ENABLED)
1542                 set_sta_flags(sta, WLAN_STA_MFP);
1543
1544         if (elems.wmm_param)
1545                 set_sta_flags(sta, WLAN_STA_WME);
1546
1547         err = sta_info_insert(sta);
1548         sta = NULL;
1549         if (err) {
1550                 printk(KERN_DEBUG "%s: failed to insert STA entry for"
1551                        " the AP (error %d)\n", sdata->name, err);
1552                 return false;
1553         }
1554
1555         /*
1556          * Always handle WMM once after association regardless
1557          * of the first value the AP uses. Setting -1 here has
1558          * that effect because the AP values is an unsigned
1559          * 4-bit value.
1560          */
1561         ifmgd->wmm_last_param_set = -1;
1562
1563         if (elems.wmm_param)
1564                 ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
1565                                          elems.wmm_param_len);
1566         else
1567                 ieee80211_set_wmm_default(sdata);
1568
1569         local->oper_channel = wk->chan;
1570
1571         if (elems.ht_info_elem && elems.wmm_param &&
1572             (sdata->local->hw.queues >= 4) &&
1573             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1574                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1575                                                cbss->bssid, ap_ht_cap_flags);
1576
1577         /* set AID and assoc capability,
1578          * ieee80211_set_associated() will tell the driver */
1579         bss_conf->aid = aid;
1580         bss_conf->assoc_capability = capab_info;
1581         ieee80211_set_associated(sdata, cbss, changed);
1582
1583         /*
1584          * If we're using 4-addr mode, let the AP know that we're
1585          * doing so, so that it can create the STA VLAN on its side
1586          */
1587         if (ifmgd->use_4addr)
1588                 ieee80211_send_4addr_nullfunc(local, sdata);
1589
1590         /*
1591          * Start timer to probe the connection to the AP now.
1592          * Also start the timer that will detect beacon loss.
1593          */
1594         ieee80211_sta_rx_notify(sdata, (struct ieee80211_hdr *)mgmt);
1595         ieee80211_sta_reset_beacon_monitor(sdata);
1596
1597         return true;
1598 }
1599
1600
1601 static void ieee80211_rx_bss_info(struct ieee80211_sub_if_data *sdata,
1602                                   struct ieee80211_mgmt *mgmt,
1603                                   size_t len,
1604                                   struct ieee80211_rx_status *rx_status,
1605                                   struct ieee802_11_elems *elems,
1606                                   bool beacon)
1607 {
1608         struct ieee80211_local *local = sdata->local;
1609         int freq;
1610         struct ieee80211_bss *bss;
1611         struct ieee80211_channel *channel;
1612         bool need_ps = false;
1613
1614         if (sdata->u.mgd.associated) {
1615                 bss = (void *)sdata->u.mgd.associated->priv;
1616                 /* not previously set so we may need to recalc */
1617                 need_ps = !bss->dtim_period;
1618         }
1619
1620         if (elems->ds_params && elems->ds_params_len == 1)
1621                 freq = ieee80211_channel_to_frequency(elems->ds_params[0],
1622                                                       rx_status->band);
1623         else
1624                 freq = rx_status->freq;
1625
1626         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1627
1628         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1629                 return;
1630
1631         bss = ieee80211_bss_info_update(local, rx_status, mgmt, len, elems,
1632                                         channel, beacon);
1633         if (bss)
1634                 ieee80211_rx_bss_put(local, bss);
1635
1636         if (!sdata->u.mgd.associated)
1637                 return;
1638
1639         if (need_ps) {
1640                 mutex_lock(&local->iflist_mtx);
1641                 ieee80211_recalc_ps(local, -1);
1642                 mutex_unlock(&local->iflist_mtx);
1643         }
1644
1645         if (elems->ch_switch_elem && (elems->ch_switch_elem_len == 3) &&
1646             (memcmp(mgmt->bssid, sdata->u.mgd.associated->bssid,
1647                                                         ETH_ALEN) == 0)) {
1648                 struct ieee80211_channel_sw_ie *sw_elem =
1649                         (struct ieee80211_channel_sw_ie *)elems->ch_switch_elem;
1650                 ieee80211_sta_process_chanswitch(sdata, sw_elem,
1651                                                  bss, rx_status->mactime);
1652         }
1653 }
1654
1655
1656 static void ieee80211_rx_mgmt_probe_resp(struct ieee80211_sub_if_data *sdata,
1657                                          struct sk_buff *skb)
1658 {
1659         struct ieee80211_mgmt *mgmt = (void *)skb->data;
1660         struct ieee80211_if_managed *ifmgd;
1661         struct ieee80211_rx_status *rx_status = (void *) skb->cb;
1662         size_t baselen, len = skb->len;
1663         struct ieee802_11_elems elems;
1664
1665         ifmgd = &sdata->u.mgd;
1666
1667         ASSERT_MGD_MTX(ifmgd);
1668
1669         if (memcmp(mgmt->da, sdata->vif.addr, ETH_ALEN))
1670                 return; /* ignore ProbeResp to foreign address */
1671
1672         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1673         if (baselen > len)
1674                 return;
1675
1676         ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - baselen,
1677                                 &elems);
1678
1679         ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems, false);
1680
1681         if (ifmgd->associated &&
1682             memcmp(mgmt->bssid, ifmgd->associated->bssid, ETH_ALEN) == 0)
1683                 ieee80211_reset_ap_probe(sdata);
1684 }
1685
1686 /*
1687  * This is the canonical list of information elements we care about,
1688  * the filter code also gives us all changes to the Microsoft OUI
1689  * (00:50:F2) vendor IE which is used for WMM which we need to track.
1690  *
1691  * We implement beacon filtering in software since that means we can
1692  * avoid processing the frame here and in cfg80211, and userspace
1693  * will not be able to tell whether the hardware supports it or not.
1694  *
1695  * XXX: This list needs to be dynamic -- userspace needs to be able to
1696  *      add items it requires. It also needs to be able to tell us to
1697  *      look out for other vendor IEs.
1698  */
1699 static const u64 care_about_ies =
1700         (1ULL << WLAN_EID_COUNTRY) |
1701         (1ULL << WLAN_EID_ERP_INFO) |
1702         (1ULL << WLAN_EID_CHANNEL_SWITCH) |
1703         (1ULL << WLAN_EID_PWR_CONSTRAINT) |
1704         (1ULL << WLAN_EID_HT_CAPABILITY) |
1705         (1ULL << WLAN_EID_HT_INFORMATION);
1706
1707 static void ieee80211_rx_mgmt_beacon(struct ieee80211_sub_if_data *sdata,
1708                                      struct ieee80211_mgmt *mgmt,
1709                                      size_t len,
1710                                      struct ieee80211_rx_status *rx_status)
1711 {
1712         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1713         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
1714         size_t baselen;
1715         struct ieee802_11_elems elems;
1716         struct ieee80211_local *local = sdata->local;
1717         u32 changed = 0;
1718         bool erp_valid, directed_tim = false;
1719         u8 erp_value = 0;
1720         u32 ncrc;
1721         u8 *bssid;
1722
1723         ASSERT_MGD_MTX(ifmgd);
1724
1725         /* Process beacon from the current BSS */
1726         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
1727         if (baselen > len)
1728                 return;
1729
1730         if (rx_status->freq != local->hw.conf.channel->center_freq)
1731                 return;
1732
1733         /*
1734          * We might have received a number of frames, among them a
1735          * disassoc frame and a beacon...
1736          */
1737         if (!ifmgd->associated)
1738                 return;
1739
1740         bssid = ifmgd->associated->bssid;
1741
1742         /*
1743          * And in theory even frames from a different AP we were just
1744          * associated to a split-second ago!
1745          */
1746         if (memcmp(bssid, mgmt->bssid, ETH_ALEN) != 0)
1747                 return;
1748
1749         /* Track average RSSI from the Beacon frames of the current AP */
1750         ifmgd->last_beacon_signal = rx_status->signal;
1751         if (ifmgd->flags & IEEE80211_STA_RESET_SIGNAL_AVE) {
1752                 ifmgd->flags &= ~IEEE80211_STA_RESET_SIGNAL_AVE;
1753                 ifmgd->ave_beacon_signal = rx_status->signal * 16;
1754                 ifmgd->last_cqm_event_signal = 0;
1755                 ifmgd->count_beacon_signal = 1;
1756         } else {
1757                 ifmgd->ave_beacon_signal =
1758                         (IEEE80211_SIGNAL_AVE_WEIGHT * rx_status->signal * 16 +
1759                          (16 - IEEE80211_SIGNAL_AVE_WEIGHT) *
1760                          ifmgd->ave_beacon_signal) / 16;
1761                 ifmgd->count_beacon_signal++;
1762         }
1763         if (bss_conf->cqm_rssi_thold &&
1764             ifmgd->count_beacon_signal >= IEEE80211_SIGNAL_AVE_MIN_COUNT &&
1765             !(local->hw.flags & IEEE80211_HW_SUPPORTS_CQM_RSSI)) {
1766                 int sig = ifmgd->ave_beacon_signal / 16;
1767                 int last_event = ifmgd->last_cqm_event_signal;
1768                 int thold = bss_conf->cqm_rssi_thold;
1769                 int hyst = bss_conf->cqm_rssi_hyst;
1770                 if (sig < thold &&
1771                     (last_event == 0 || sig < last_event - hyst)) {
1772                         ifmgd->last_cqm_event_signal = sig;
1773                         ieee80211_cqm_rssi_notify(
1774                                 &sdata->vif,
1775                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW,
1776                                 GFP_KERNEL);
1777                 } else if (sig > thold &&
1778                            (last_event == 0 || sig > last_event + hyst)) {
1779                         ifmgd->last_cqm_event_signal = sig;
1780                         ieee80211_cqm_rssi_notify(
1781                                 &sdata->vif,
1782                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH,
1783                                 GFP_KERNEL);
1784                 }
1785         }
1786
1787         if (ifmgd->flags & IEEE80211_STA_BEACON_POLL) {
1788 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1789                 if (net_ratelimit()) {
1790                         printk(KERN_DEBUG "%s: cancelling probereq poll due "
1791                                "to a received beacon\n", sdata->name);
1792                 }
1793 #endif
1794                 ifmgd->flags &= ~IEEE80211_STA_BEACON_POLL;
1795                 mutex_lock(&local->iflist_mtx);
1796                 ieee80211_recalc_ps(local, -1);
1797                 mutex_unlock(&local->iflist_mtx);
1798         }
1799
1800         /*
1801          * Push the beacon loss detection into the future since
1802          * we are processing a beacon from the AP just now.
1803          */
1804         ieee80211_sta_reset_beacon_monitor(sdata);
1805
1806         ncrc = crc32_be(0, (void *)&mgmt->u.beacon.beacon_int, 4);
1807         ncrc = ieee802_11_parse_elems_crc(mgmt->u.beacon.variable,
1808                                           len - baselen, &elems,
1809                                           care_about_ies, ncrc);
1810
1811         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
1812                 directed_tim = ieee80211_check_tim(elems.tim, elems.tim_len,
1813                                                    ifmgd->aid);
1814
1815         if (ncrc != ifmgd->beacon_crc || !ifmgd->beacon_crc_valid) {
1816                 ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems,
1817                                       true);
1818
1819                 ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
1820                                          elems.wmm_param_len);
1821         }
1822
1823         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK) {
1824                 if (directed_tim) {
1825                         if (local->hw.conf.dynamic_ps_timeout > 0) {
1826                                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
1827                                 ieee80211_hw_config(local,
1828                                                     IEEE80211_CONF_CHANGE_PS);
1829                                 ieee80211_send_nullfunc(local, sdata, 0);
1830                         } else {
1831                                 local->pspolling = true;
1832
1833                                 /*
1834                                  * Here is assumed that the driver will be
1835                                  * able to send ps-poll frame and receive a
1836                                  * response even though power save mode is
1837                                  * enabled, but some drivers might require
1838                                  * to disable power save here. This needs
1839                                  * to be investigated.
1840                                  */
1841                                 ieee80211_send_pspoll(local, sdata);
1842                         }
1843                 }
1844         }
1845
1846         if (ncrc == ifmgd->beacon_crc && ifmgd->beacon_crc_valid)
1847                 return;
1848         ifmgd->beacon_crc = ncrc;
1849         ifmgd->beacon_crc_valid = true;
1850
1851         if (elems.erp_info && elems.erp_info_len >= 1) {
1852                 erp_valid = true;
1853                 erp_value = elems.erp_info[0];
1854         } else {
1855                 erp_valid = false;
1856         }
1857         changed |= ieee80211_handle_bss_capability(sdata,
1858                         le16_to_cpu(mgmt->u.beacon.capab_info),
1859                         erp_valid, erp_value);
1860
1861
1862         if (elems.ht_cap_elem && elems.ht_info_elem && elems.wmm_param &&
1863             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N)) {
1864                 struct sta_info *sta;
1865                 struct ieee80211_supported_band *sband;
1866                 u16 ap_ht_cap_flags;
1867
1868                 rcu_read_lock();
1869
1870                 sta = sta_info_get(sdata, bssid);
1871                 if (WARN_ON(!sta)) {
1872                         rcu_read_unlock();
1873                         return;
1874                 }
1875
1876                 sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
1877
1878                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1879                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1880
1881                 ap_ht_cap_flags = sta->sta.ht_cap.cap;
1882
1883                 rcu_read_unlock();
1884
1885                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1886                                                bssid, ap_ht_cap_flags);
1887         }
1888
1889         /* Note: country IE parsing is done for us by cfg80211 */
1890         if (elems.country_elem) {
1891                 /* TODO: IBSS also needs this */
1892                 if (elems.pwr_constr_elem)
1893                         ieee80211_handle_pwr_constr(sdata,
1894                                 le16_to_cpu(mgmt->u.probe_resp.capab_info),
1895                                 elems.pwr_constr_elem,
1896                                 elems.pwr_constr_elem_len);
1897         }
1898
1899         ieee80211_bss_info_change_notify(sdata, changed);
1900 }
1901
1902 void ieee80211_sta_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1903                                   struct sk_buff *skb)
1904 {
1905         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1906         struct ieee80211_rx_status *rx_status;
1907         struct ieee80211_mgmt *mgmt;
1908         enum rx_mgmt_action rma = RX_MGMT_NONE;
1909         u16 fc;
1910
1911         rx_status = (struct ieee80211_rx_status *) skb->cb;
1912         mgmt = (struct ieee80211_mgmt *) skb->data;
1913         fc = le16_to_cpu(mgmt->frame_control);
1914
1915         mutex_lock(&ifmgd->mtx);
1916
1917         if (ifmgd->associated &&
1918             memcmp(ifmgd->associated->bssid, mgmt->bssid, ETH_ALEN) == 0) {
1919                 switch (fc & IEEE80211_FCTL_STYPE) {
1920                 case IEEE80211_STYPE_BEACON:
1921                         ieee80211_rx_mgmt_beacon(sdata, mgmt, skb->len,
1922                                                  rx_status);
1923                         break;
1924                 case IEEE80211_STYPE_PROBE_RESP:
1925                         ieee80211_rx_mgmt_probe_resp(sdata, skb);
1926                         break;
1927                 case IEEE80211_STYPE_DEAUTH:
1928                         rma = ieee80211_rx_mgmt_deauth(sdata, mgmt, skb->len);
1929                         break;
1930                 case IEEE80211_STYPE_DISASSOC:
1931                         rma = ieee80211_rx_mgmt_disassoc(sdata, mgmt, skb->len);
1932                         break;
1933                 case IEEE80211_STYPE_ACTION:
1934                         switch (mgmt->u.action.category) {
1935                         case WLAN_CATEGORY_SPECTRUM_MGMT:
1936                                 ieee80211_sta_process_chanswitch(sdata,
1937                                                 &mgmt->u.action.u.chan_switch.sw_elem,
1938                                                 (void *)ifmgd->associated->priv,
1939                                                 rx_status->mactime);
1940                                 break;
1941                         }
1942                 }
1943                 mutex_unlock(&ifmgd->mtx);
1944
1945                 switch (rma) {
1946                 case RX_MGMT_NONE:
1947                         /* no action */
1948                         break;
1949                 case RX_MGMT_CFG80211_DEAUTH:
1950                         cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
1951                         break;
1952                 case RX_MGMT_CFG80211_DISASSOC:
1953                         cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
1954                         break;
1955                 default:
1956                         WARN(1, "unexpected: %d", rma);
1957                 }
1958                 return;
1959         }
1960
1961         mutex_unlock(&ifmgd->mtx);
1962
1963         if (skb->len >= 24 + 2 /* mgmt + deauth reason */ &&
1964             (fc & IEEE80211_FCTL_STYPE) == IEEE80211_STYPE_DEAUTH) {
1965                 struct ieee80211_local *local = sdata->local;
1966                 struct ieee80211_work *wk;
1967
1968                 mutex_lock(&local->mtx);
1969                 list_for_each_entry(wk, &local->work_list, list) {
1970                         if (wk->sdata != sdata)
1971                                 continue;
1972
1973                         if (wk->type != IEEE80211_WORK_ASSOC &&
1974                             wk->type != IEEE80211_WORK_ASSOC_BEACON_WAIT)
1975                                 continue;
1976
1977                         if (memcmp(mgmt->bssid, wk->filter_ta, ETH_ALEN))
1978                                 continue;
1979                         if (memcmp(mgmt->sa, wk->filter_ta, ETH_ALEN))
1980                                 continue;
1981
1982                         /*
1983                          * Printing the message only here means we can't
1984                          * spuriously print it, but it also means that it
1985                          * won't be printed when the frame comes in before
1986                          * we even tried to associate or in similar cases.
1987                          *
1988                          * Ultimately, I suspect cfg80211 should print the
1989                          * messages instead.
1990                          */
1991                         printk(KERN_DEBUG
1992                                "%s: deauthenticated from %pM (Reason: %u)\n",
1993                                sdata->name, mgmt->bssid,
1994                                le16_to_cpu(mgmt->u.deauth.reason_code));
1995
1996                         list_del_rcu(&wk->list);
1997                         free_work(wk);
1998                         break;
1999                 }
2000                 mutex_unlock(&local->mtx);
2001
2002                 cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
2003         }
2004 }
2005
2006 static void ieee80211_sta_timer(unsigned long data)
2007 {
2008         struct ieee80211_sub_if_data *sdata =
2009                 (struct ieee80211_sub_if_data *) data;
2010         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2011         struct ieee80211_local *local = sdata->local;
2012
2013         if (local->quiescing) {
2014                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
2015                 return;
2016         }
2017
2018         ieee80211_queue_work(&local->hw, &sdata->work);
2019 }
2020
2021 static void ieee80211_sta_connection_lost(struct ieee80211_sub_if_data *sdata,
2022                                           u8 *bssid)
2023 {
2024         struct ieee80211_local *local = sdata->local;
2025         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2026
2027         ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
2028                           IEEE80211_STA_BEACON_POLL);
2029
2030         ieee80211_set_disassoc(sdata, true, true);
2031         mutex_unlock(&ifmgd->mtx);
2032         mutex_lock(&local->mtx);
2033         ieee80211_recalc_idle(local);
2034         mutex_unlock(&local->mtx);
2035         /*
2036          * must be outside lock due to cfg80211,
2037          * but that's not a problem.
2038          */
2039         ieee80211_send_deauth_disassoc(sdata, bssid,
2040                         IEEE80211_STYPE_DEAUTH,
2041                         WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
2042                         NULL, true);
2043         mutex_lock(&ifmgd->mtx);
2044 }
2045
2046 void ieee80211_sta_work(struct ieee80211_sub_if_data *sdata)
2047 {
2048         struct ieee80211_local *local = sdata->local;
2049         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2050
2051         /* then process the rest of the work */
2052         mutex_lock(&ifmgd->mtx);
2053
2054         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
2055                             IEEE80211_STA_CONNECTION_POLL) &&
2056             ifmgd->associated) {
2057                 u8 bssid[ETH_ALEN];
2058                 int max_tries;
2059
2060                 memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
2061
2062                 if (local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS)
2063                         max_tries = max_nullfunc_tries;
2064                 else
2065                         max_tries = max_probe_tries;
2066
2067                 /* ACK received for nullfunc probing frame */
2068                 if (!ifmgd->probe_send_count)
2069                         ieee80211_reset_ap_probe(sdata);
2070                 else if (ifmgd->nullfunc_failed) {
2071                         if (ifmgd->probe_send_count < max_tries) {
2072 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2073                                 wiphy_debug(local->hw.wiphy,
2074                                             "%s: No ack for nullfunc frame to"
2075                                             " AP %pM, try %d/%i\n",
2076                                             sdata->name, bssid,
2077                                             ifmgd->probe_send_count, max_tries);
2078 #endif
2079                                 ieee80211_mgd_probe_ap_send(sdata);
2080                         } else {
2081 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2082                                 wiphy_debug(local->hw.wiphy,
2083                                             "%s: No ack for nullfunc frame to"
2084                                             " AP %pM, disconnecting.\n",
2085                                             sdata->name, bssid);
2086 #endif
2087                                 ieee80211_sta_connection_lost(sdata, bssid);
2088                         }
2089                 } else if (time_is_after_jiffies(ifmgd->probe_timeout))
2090                         run_again(ifmgd, ifmgd->probe_timeout);
2091                 else if (local->hw.flags & IEEE80211_HW_REPORTS_TX_ACK_STATUS) {
2092 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2093                         wiphy_debug(local->hw.wiphy,
2094                                     "%s: Failed to send nullfunc to AP %pM"
2095                                     " after %dms, disconnecting.\n",
2096                                     sdata->name,
2097                                     bssid, probe_wait_ms);
2098 #endif
2099                         ieee80211_sta_connection_lost(sdata, bssid);
2100                 } else if (ifmgd->probe_send_count < max_tries) {
2101 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
2102                         wiphy_debug(local->hw.wiphy,
2103                                     "%s: No probe response from AP %pM"
2104                                     " after %dms, try %d/%i\n",
2105                                     sdata->name,
2106                                     bssid, probe_wait_ms,
2107                                     ifmgd->probe_send_count, max_tries);
2108 #endif
2109                         ieee80211_mgd_probe_ap_send(sdata);
2110                 } else {
2111                         /*
2112                          * We actually lost the connection ... or did we?
2113                          * Let's make sure!
2114                          */
2115                         wiphy_debug(local->hw.wiphy,
2116                                     "%s: No probe response from AP %pM"
2117                                     " after %dms, disconnecting.\n",
2118                                     sdata->name,
2119                                     bssid, probe_wait_ms);
2120
2121                         ieee80211_sta_connection_lost(sdata, bssid);
2122                 }
2123         }
2124
2125         mutex_unlock(&ifmgd->mtx);
2126 }
2127
2128 static void ieee80211_sta_bcn_mon_timer(unsigned long data)
2129 {
2130         struct ieee80211_sub_if_data *sdata =
2131                 (struct ieee80211_sub_if_data *) data;
2132         struct ieee80211_local *local = sdata->local;
2133
2134         if (local->quiescing)
2135                 return;
2136
2137         ieee80211_queue_work(&sdata->local->hw,
2138                              &sdata->u.mgd.beacon_connection_loss_work);
2139 }
2140
2141 static void ieee80211_sta_conn_mon_timer(unsigned long data)
2142 {
2143         struct ieee80211_sub_if_data *sdata =
2144                 (struct ieee80211_sub_if_data *) data;
2145         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2146         struct ieee80211_local *local = sdata->local;
2147
2148         if (local->quiescing)
2149                 return;
2150
2151         ieee80211_queue_work(&local->hw, &ifmgd->monitor_work);
2152 }
2153
2154 static void ieee80211_sta_monitor_work(struct work_struct *work)
2155 {
2156         struct ieee80211_sub_if_data *sdata =
2157                 container_of(work, struct ieee80211_sub_if_data,
2158                              u.mgd.monitor_work);
2159
2160         ieee80211_mgd_probe_ap(sdata, false);
2161 }
2162
2163 static void ieee80211_restart_sta_timer(struct ieee80211_sub_if_data *sdata)
2164 {
2165         if (sdata->vif.type == NL80211_IFTYPE_STATION) {
2166                 sdata->u.mgd.flags &= ~(IEEE80211_STA_BEACON_POLL |
2167                                         IEEE80211_STA_CONNECTION_POLL);
2168
2169                 /* let's probe the connection once */
2170                 ieee80211_queue_work(&sdata->local->hw,
2171                            &sdata->u.mgd.monitor_work);
2172                 /* and do all the other regular work too */
2173                 ieee80211_queue_work(&sdata->local->hw, &sdata->work);
2174         }
2175 }
2176
2177 #ifdef CONFIG_PM
2178 void ieee80211_sta_quiesce(struct ieee80211_sub_if_data *sdata)
2179 {
2180         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2181
2182         /*
2183          * we need to use atomic bitops for the running bits
2184          * only because both timers might fire at the same
2185          * time -- the code here is properly synchronised.
2186          */
2187
2188         cancel_work_sync(&ifmgd->request_smps_work);
2189
2190         cancel_work_sync(&ifmgd->beacon_connection_loss_work);
2191         if (del_timer_sync(&ifmgd->timer))
2192                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
2193
2194         cancel_work_sync(&ifmgd->chswitch_work);
2195         if (del_timer_sync(&ifmgd->chswitch_timer))
2196                 set_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running);
2197
2198         cancel_work_sync(&ifmgd->monitor_work);
2199         /* these will just be re-established on connection */
2200         del_timer_sync(&ifmgd->conn_mon_timer);
2201         del_timer_sync(&ifmgd->bcn_mon_timer);
2202 }
2203
2204 void ieee80211_sta_restart(struct ieee80211_sub_if_data *sdata)
2205 {
2206         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2207
2208         if (test_and_clear_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running))
2209                 add_timer(&ifmgd->timer);
2210         if (test_and_clear_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running))
2211                 add_timer(&ifmgd->chswitch_timer);
2212         ieee80211_sta_reset_beacon_monitor(sdata);
2213         ieee80211_restart_sta_timer(sdata);
2214 }
2215 #endif
2216
2217 /* interface setup */
2218 void ieee80211_sta_setup_sdata(struct ieee80211_sub_if_data *sdata)
2219 {
2220         struct ieee80211_if_managed *ifmgd;
2221
2222         ifmgd = &sdata->u.mgd;
2223         INIT_WORK(&ifmgd->monitor_work, ieee80211_sta_monitor_work);
2224         INIT_WORK(&ifmgd->chswitch_work, ieee80211_chswitch_work);
2225         INIT_WORK(&ifmgd->beacon_connection_loss_work,
2226                   ieee80211_beacon_connection_loss_work);
2227         INIT_WORK(&ifmgd->request_smps_work, ieee80211_request_smps_work);
2228         setup_timer(&ifmgd->timer, ieee80211_sta_timer,
2229                     (unsigned long) sdata);
2230         setup_timer(&ifmgd->bcn_mon_timer, ieee80211_sta_bcn_mon_timer,
2231                     (unsigned long) sdata);
2232         setup_timer(&ifmgd->conn_mon_timer, ieee80211_sta_conn_mon_timer,
2233                     (unsigned long) sdata);
2234         setup_timer(&ifmgd->chswitch_timer, ieee80211_chswitch_timer,
2235                     (unsigned long) sdata);
2236
2237         ifmgd->flags = 0;
2238
2239         mutex_init(&ifmgd->mtx);
2240
2241         if (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_SMPS)
2242                 ifmgd->req_smps = IEEE80211_SMPS_AUTOMATIC;
2243         else
2244                 ifmgd->req_smps = IEEE80211_SMPS_OFF;
2245 }
2246
2247 /* scan finished notification */
2248 void ieee80211_mlme_notify_scan_completed(struct ieee80211_local *local)
2249 {
2250         struct ieee80211_sub_if_data *sdata = local->scan_sdata;
2251
2252         /* Restart STA timers */
2253         rcu_read_lock();
2254         list_for_each_entry_rcu(sdata, &local->interfaces, list)
2255                 ieee80211_restart_sta_timer(sdata);
2256         rcu_read_unlock();
2257 }
2258
2259 int ieee80211_max_network_latency(struct notifier_block *nb,
2260                                   unsigned long data, void *dummy)
2261 {
2262         s32 latency_usec = (s32) data;
2263         struct ieee80211_local *local =
2264                 container_of(nb, struct ieee80211_local,
2265                              network_latency_notifier);
2266
2267         mutex_lock(&local->iflist_mtx);
2268         ieee80211_recalc_ps(local, latency_usec);
2269         mutex_unlock(&local->iflist_mtx);
2270
2271         return 0;
2272 }
2273
2274 /* config hooks */
2275 static enum work_done_result
2276 ieee80211_probe_auth_done(struct ieee80211_work *wk,
2277                           struct sk_buff *skb)
2278 {
2279         if (!skb) {
2280                 cfg80211_send_auth_timeout(wk->sdata->dev, wk->filter_ta);
2281                 return WORK_DONE_DESTROY;
2282         }
2283
2284         if (wk->type == IEEE80211_WORK_AUTH) {
2285                 cfg80211_send_rx_auth(wk->sdata->dev, skb->data, skb->len);
2286                 return WORK_DONE_DESTROY;
2287         }
2288
2289         mutex_lock(&wk->sdata->u.mgd.mtx);
2290         ieee80211_rx_mgmt_probe_resp(wk->sdata, skb);
2291         mutex_unlock(&wk->sdata->u.mgd.mtx);
2292
2293         wk->type = IEEE80211_WORK_AUTH;
2294         wk->probe_auth.tries = 0;
2295         return WORK_DONE_REQUEUE;
2296 }
2297
2298 int ieee80211_mgd_auth(struct ieee80211_sub_if_data *sdata,
2299                        struct cfg80211_auth_request *req)
2300 {
2301         const u8 *ssid;
2302         struct ieee80211_work *wk;
2303         u16 auth_alg;
2304
2305         if (req->local_state_change)
2306                 return 0; /* no need to update mac80211 state */
2307
2308         switch (req->auth_type) {
2309         case NL80211_AUTHTYPE_OPEN_SYSTEM:
2310                 auth_alg = WLAN_AUTH_OPEN;
2311                 break;
2312         case NL80211_AUTHTYPE_SHARED_KEY:
2313                 if (IS_ERR(sdata->local->wep_tx_tfm))
2314                         return -EOPNOTSUPP;
2315                 auth_alg = WLAN_AUTH_SHARED_KEY;
2316                 break;
2317         case NL80211_AUTHTYPE_FT:
2318                 auth_alg = WLAN_AUTH_FT;
2319                 break;
2320         case NL80211_AUTHTYPE_NETWORK_EAP:
2321                 auth_alg = WLAN_AUTH_LEAP;
2322                 break;
2323         default:
2324                 return -EOPNOTSUPP;
2325         }
2326
2327         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
2328         if (!wk)
2329                 return -ENOMEM;
2330
2331         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);
2332
2333         if (req->ie && req->ie_len) {
2334                 memcpy(wk->ie, req->ie, req->ie_len);
2335                 wk->ie_len = req->ie_len;
2336         }
2337
2338         if (req->key && req->key_len) {
2339                 wk->probe_auth.key_len = req->key_len;
2340                 wk->probe_auth.key_idx = req->key_idx;
2341                 memcpy(wk->probe_auth.key, req->key, req->key_len);
2342         }
2343
2344         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
2345         memcpy(wk->probe_auth.ssid, ssid + 2, ssid[1]);
2346         wk->probe_auth.ssid_len = ssid[1];
2347
2348         wk->probe_auth.algorithm = auth_alg;
2349         wk->probe_auth.privacy = req->bss->capability & WLAN_CAPABILITY_PRIVACY;
2350
2351         /* if we already have a probe, don't probe again */
2352         if (req->bss->proberesp_ies)
2353                 wk->type = IEEE80211_WORK_AUTH;
2354         else
2355                 wk->type = IEEE80211_WORK_DIRECT_PROBE;
2356         wk->chan = req->bss->channel;
2357         wk->chan_type = NL80211_CHAN_NO_HT;
2358         wk->sdata = sdata;
2359         wk->done = ieee80211_probe_auth_done;
2360
2361         ieee80211_add_work(wk);
2362         return 0;
2363 }
2364
2365 static enum work_done_result ieee80211_assoc_done(struct ieee80211_work *wk,
2366                                                   struct sk_buff *skb)
2367 {
2368         struct ieee80211_mgmt *mgmt;
2369         struct ieee80211_rx_status *rx_status;
2370         struct ieee802_11_elems elems;
2371         u16 status;
2372
2373         if (!skb) {
2374                 cfg80211_send_assoc_timeout(wk->sdata->dev, wk->filter_ta);
2375                 return WORK_DONE_DESTROY;
2376         }
2377
2378         if (wk->type == IEEE80211_WORK_ASSOC_BEACON_WAIT) {
2379                 mutex_lock(&wk->sdata->u.mgd.mtx);
2380                 rx_status = (void *) skb->cb;
2381                 ieee802_11_parse_elems(skb->data + 24 + 12, skb->len - 24 - 12, &elems);
2382                 ieee80211_rx_bss_info(wk->sdata, (void *)skb->data, skb->len, rx_status,
2383                                       &elems, true);
2384                 mutex_unlock(&wk->sdata->u.mgd.mtx);
2385
2386                 wk->type = IEEE80211_WORK_ASSOC;
2387                 /* not really done yet */
2388                 return WORK_DONE_REQUEUE;
2389         }
2390
2391         mgmt = (void *)skb->data;
2392         status = le16_to_cpu(mgmt->u.assoc_resp.status_code);
2393
2394         if (status == WLAN_STATUS_SUCCESS) {
2395                 mutex_lock(&wk->sdata->u.mgd.mtx);
2396                 if (!ieee80211_assoc_success(wk, mgmt, skb->len)) {
2397                         mutex_unlock(&wk->sdata->u.mgd.mtx);
2398                         /* oops -- internal error -- send timeout for now */
2399                         cfg80211_send_assoc_timeout(wk->sdata->dev,
2400                                                     wk->filter_ta);
2401                         return WORK_DONE_DESTROY;
2402                 }
2403
2404                 mutex_unlock(&wk->sdata->u.mgd.mtx);
2405         }
2406
2407         cfg80211_send_rx_assoc(wk->sdata->dev, skb->data, skb->len);
2408         return WORK_DONE_DESTROY;
2409 }
2410
2411 int ieee80211_mgd_assoc(struct ieee80211_sub_if_data *sdata,
2412                         struct cfg80211_assoc_request *req)
2413 {
2414         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2415         struct ieee80211_bss *bss = (void *)req->bss->priv;
2416         struct ieee80211_work *wk;
2417         const u8 *ssid;
2418         int i;
2419
2420         mutex_lock(&ifmgd->mtx);
2421         if (ifmgd->associated) {
2422                 if (!req->prev_bssid ||
2423                     memcmp(req->prev_bssid, ifmgd->associated->bssid,
2424                            ETH_ALEN)) {
2425                         /*
2426                          * We are already associated and the request was not a
2427                          * reassociation request from the current BSS, so
2428                          * reject it.
2429                          */
2430                         mutex_unlock(&ifmgd->mtx);
2431                         return -EALREADY;
2432                 }
2433
2434                 /* Trying to reassociate - clear previous association state */
2435                 ieee80211_set_disassoc(sdata, true, false);
2436         }
2437         mutex_unlock(&ifmgd->mtx);
2438
2439         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
2440         if (!wk)
2441                 return -ENOMEM;
2442
2443         ifmgd->flags &= ~IEEE80211_STA_DISABLE_11N;
2444         ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
2445
2446         ifmgd->beacon_crc_valid = false;
2447
2448         for (i = 0; i < req->crypto.n_ciphers_pairwise; i++)
2449                 if (req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP40 ||
2450                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_TKIP ||
2451                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP104)
2452                         ifmgd->flags |= IEEE80211_STA_DISABLE_11N;
2453
2454
2455         if (req->ie && req->ie_len) {
2456                 memcpy(wk->ie, req->ie, req->ie_len);
2457                 wk->ie_len = req->ie_len;
2458         } else
2459                 wk->ie_len = 0;
2460
2461         wk->assoc.bss = req->bss;
2462
2463         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);
2464
2465         /* new association always uses requested smps mode */
2466         if (ifmgd->req_smps == IEEE80211_SMPS_AUTOMATIC) {
2467                 if (ifmgd->powersave)
2468                         ifmgd->ap_smps = IEEE80211_SMPS_DYNAMIC;
2469                 else
2470                         ifmgd->ap_smps = IEEE80211_SMPS_OFF;
2471         } else
2472                 ifmgd->ap_smps = ifmgd->req_smps;
2473
2474         wk->assoc.smps = ifmgd->ap_smps;
2475         /*
2476          * IEEE802.11n does not allow TKIP/WEP as pairwise ciphers in HT mode.
2477          * We still associate in non-HT mode (11a/b/g) if any one of these
2478          * ciphers is configured as pairwise.
2479          * We can set this to true for non-11n hardware, that'll be checked
2480          * separately along with the peer capabilities.
2481          */
2482         wk->assoc.use_11n = !(ifmgd->flags & IEEE80211_STA_DISABLE_11N);
2483         wk->assoc.capability = req->bss->capability;
2484         wk->assoc.wmm_used = bss->wmm_used;
2485         wk->assoc.supp_rates = bss->supp_rates;
2486         wk->assoc.supp_rates_len = bss->supp_rates_len;
2487         wk->assoc.ht_information_ie =
2488                 ieee80211_bss_get_ie(req->bss, WLAN_EID_HT_INFORMATION);
2489
2490         if (bss->wmm_used && bss->uapsd_supported &&
2491             (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_UAPSD)) {
2492                 wk->assoc.uapsd_used = true;
2493                 ifmgd->flags |= IEEE80211_STA_UAPSD_ENABLED;
2494         } else {
2495                 wk->assoc.uapsd_used = false;
2496                 ifmgd->flags &= ~IEEE80211_STA_UAPSD_ENABLED;
2497         }
2498
2499         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
2500         memcpy(wk->assoc.ssid, ssid + 2, ssid[1]);
2501         wk->assoc.ssid_len = ssid[1];
2502
2503         if (req->prev_bssid)
2504                 memcpy(wk->assoc.prev_bssid, req->prev_bssid, ETH_ALEN);
2505
2506         wk->chan = req->bss->channel;
2507         wk->chan_type = NL80211_CHAN_NO_HT;
2508         wk->sdata = sdata;
2509         wk->done = ieee80211_assoc_done;
2510         if (!bss->dtim_period &&
2511             sdata->local->hw.flags & IEEE80211_HW_NEED_DTIM_PERIOD)
2512                 wk->type = IEEE80211_WORK_ASSOC_BEACON_WAIT;
2513         else
2514                 wk->type = IEEE80211_WORK_ASSOC;
2515
2516         if (req->use_mfp) {
2517                 ifmgd->mfp = IEEE80211_MFP_REQUIRED;
2518                 ifmgd->flags |= IEEE80211_STA_MFP_ENABLED;
2519         } else {
2520                 ifmgd->mfp = IEEE80211_MFP_DISABLED;
2521                 ifmgd->flags &= ~IEEE80211_STA_MFP_ENABLED;
2522         }
2523
2524         if (req->crypto.control_port)
2525                 ifmgd->flags |= IEEE80211_STA_CONTROL_PORT;
2526         else
2527                 ifmgd->flags &= ~IEEE80211_STA_CONTROL_PORT;
2528
2529         sdata->control_port_protocol = req->crypto.control_port_ethertype;
2530         sdata->control_port_no_encrypt = req->crypto.control_port_no_encrypt;
2531
2532         ieee80211_add_work(wk);
2533         return 0;
2534 }
2535
2536 int ieee80211_mgd_deauth(struct ieee80211_sub_if_data *sdata,
2537                          struct cfg80211_deauth_request *req,
2538                          void *cookie)
2539 {
2540         struct ieee80211_local *local = sdata->local;
2541         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2542         struct ieee80211_work *wk;
2543         u8 bssid[ETH_ALEN];
2544         bool assoc_bss = false;
2545
2546         mutex_lock(&ifmgd->mtx);
2547
2548         memcpy(bssid, req->bss->bssid, ETH_ALEN);
2549         if (ifmgd->associated == req->bss) {
2550                 ieee80211_set_disassoc(sdata, false, true);
2551                 mutex_unlock(&ifmgd->mtx);
2552                 assoc_bss = true;
2553         } else {
2554                 bool not_auth_yet = false;
2555
2556                 mutex_unlock(&ifmgd->mtx);
2557
2558                 mutex_lock(&local->mtx);
2559                 list_for_each_entry(wk, &local->work_list, list) {
2560                         if (wk->sdata != sdata)
2561                                 continue;
2562
2563                         if (wk->type != IEEE80211_WORK_DIRECT_PROBE &&
2564                             wk->type != IEEE80211_WORK_AUTH &&
2565                             wk->type != IEEE80211_WORK_ASSOC &&
2566                             wk->type != IEEE80211_WORK_ASSOC_BEACON_WAIT)
2567                                 continue;
2568
2569                         if (memcmp(req->bss->bssid, wk->filter_ta, ETH_ALEN))
2570                                 continue;
2571
2572                         not_auth_yet = wk->type == IEEE80211_WORK_DIRECT_PROBE;
2573                         list_del_rcu(&wk->list);
2574                         free_work(wk);
2575                         break;
2576                 }
2577                 mutex_unlock(&local->mtx);
2578
2579                 /*
2580                  * If somebody requests authentication and we haven't
2581                  * sent out an auth frame yet there's no need to send
2582                  * out a deauth frame either. If the state was PROBE,
2583                  * then this is the case. If it's AUTH we have sent a
2584                  * frame, and if it's IDLE we have completed the auth
2585                  * process already.
2586                  */
2587                 if (not_auth_yet) {
2588                         __cfg80211_auth_canceled(sdata->dev, bssid);
2589                         return 0;
2590                 }
2591         }
2592
2593         printk(KERN_DEBUG "%s: deauthenticating from %pM by local choice (reason=%d)\n",
2594                sdata->name, bssid, req->reason_code);
2595
2596         ieee80211_send_deauth_disassoc(sdata, bssid, IEEE80211_STYPE_DEAUTH,
2597                                        req->reason_code, cookie,
2598                                        !req->local_state_change);
2599         if (assoc_bss)
2600                 sta_info_destroy_addr(sdata, bssid);
2601
2602         mutex_lock(&sdata->local->mtx);
2603         ieee80211_recalc_idle(sdata->local);
2604         mutex_unlock(&sdata->local->mtx);
2605
2606         return 0;
2607 }
2608
2609 int ieee80211_mgd_disassoc(struct ieee80211_sub_if_data *sdata,
2610                            struct cfg80211_disassoc_request *req,
2611                            void *cookie)
2612 {
2613         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2614         u8 bssid[ETH_ALEN];
2615
2616         mutex_lock(&ifmgd->mtx);
2617
2618         /*
2619          * cfg80211 should catch this ... but it's racy since
2620          * we can receive a disassoc frame, process it, hand it
2621          * to cfg80211 while that's in a locked section already
2622          * trying to tell us that the user wants to disconnect.
2623          */
2624         if (ifmgd->associated != req->bss) {
2625                 mutex_unlock(&ifmgd->mtx);
2626                 return -ENOLINK;
2627         }
2628
2629         printk(KERN_DEBUG "%s: disassociating from %pM by local choice (reason=%d)\n",
2630                sdata->name, req->bss->bssid, req->reason_code);
2631
2632         memcpy(bssid, req->bss->bssid, ETH_ALEN);
2633         ieee80211_set_disassoc(sdata, false, true);
2634
2635         mutex_unlock(&ifmgd->mtx);
2636
2637         ieee80211_send_deauth_disassoc(sdata, req->bss->bssid,
2638                         IEEE80211_STYPE_DISASSOC, req->reason_code,
2639                         cookie, !req->local_state_change);
2640         sta_info_destroy_addr(sdata, bssid);
2641
2642         mutex_lock(&sdata->local->mtx);
2643         ieee80211_recalc_idle(sdata->local);
2644         mutex_unlock(&sdata->local->mtx);
2645
2646         return 0;
2647 }
2648
2649 void ieee80211_cqm_rssi_notify(struct ieee80211_vif *vif,
2650                                enum nl80211_cqm_rssi_threshold_event rssi_event,
2651                                gfp_t gfp)
2652 {
2653         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2654
2655         trace_api_cqm_rssi_notify(sdata, rssi_event);
2656
2657         cfg80211_cqm_rssi_notify(sdata->dev, rssi_event, gfp);
2658 }
2659 EXPORT_SYMBOL(ieee80211_cqm_rssi_notify);