net/mac80211/mlme.c

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