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 <net/mac80211.h>
  23 #include <asm/unaligned.h>
  24
  25 #include "ieee80211_i.h"
  26 #include "driver-ops.h"
  27 #include "rate.h"
  28 #include "led.h"
  29
  30 #define IEEE80211_AUTH_TIMEOUT (HZ / 5)
  31 #define IEEE80211_AUTH_MAX_TRIES 3
  32 #define IEEE80211_ASSOC_TIMEOUT (HZ / 5)
  33 #define IEEE80211_ASSOC_MAX_TRIES 3
  34 #define IEEE80211_MAX_PROBE_TRIES 5
  35
  36 /*
  37  * beacon loss detection timeout
  38  * XXX: should depend on beacon interval
  39  */
  40 #define IEEE80211_BEACON_LOSS_TIME      (2 * HZ)
  41 /*
  42  * Time the connection can be idle before we probe
  43  * it to see if we can still talk to the AP.
  44  */
  45 #define IEEE80211_CONNECTION_IDLE_TIME  (30 * HZ)
  46 /*
  47  * Time we wait for a probe response after sending
  48  * a probe request because of beacon loss or for
  49  * checking the connection still works.
  50  */
  51 #define IEEE80211_PROBE_WAIT            (HZ / 2)
  52
  53 #define TMR_RUNNING_TIMER       0
  54 #define TMR_RUNNING_CHANSW      1
  55
  56 /*
  57  * All cfg80211 functions have to be called outside a locked
  58  * section so that they can acquire a lock themselves... This
  59  * is much simpler than queuing up things in cfg80211, but we
  60  * do need some indirection for that here.
  61  */
  62 enum rx_mgmt_action {
  63         /* no action required */
  64         RX_MGMT_NONE,
  65
  66         /* caller must call cfg80211_send_rx_auth() */
  67         RX_MGMT_CFG80211_AUTH,
  68
  69         /* caller must call cfg80211_send_rx_assoc() */
  70         RX_MGMT_CFG80211_ASSOC,
  71
  72         /* caller must call cfg80211_send_deauth() */
  73         RX_MGMT_CFG80211_DEAUTH,
  74
  75         /* caller must call cfg80211_send_disassoc() */
  76         RX_MGMT_CFG80211_DISASSOC,
  77
  78         /* caller must tell cfg80211 about internal error */
  79         RX_MGMT_CFG80211_ASSOC_ERROR,
  80 };
  81
  82 /* utils */
  83 static inline void ASSERT_MGD_MTX(struct ieee80211_if_managed *ifmgd)
  84 {
  85         WARN_ON(!mutex_is_locked(&ifmgd->mtx));
  86 }
  87
  88 /*
  89  * We can have multiple work items (and connection probing)
  90  * scheduling this timer, but we need to take care to only
  91  * reschedule it when it should fire _earlier_ than it was
  92  * asked for before, or if it's not pending right now. This
  93  * function ensures that. Note that it then is required to
  94  * run this function for all timeouts after the first one
  95  * has happened -- the work that runs from this timer will
  96  * do that.
  97  */
  98 static void run_again(struct ieee80211_if_managed *ifmgd,
  99                              unsigned long timeout)
 100 {
 101         ASSERT_MGD_MTX(ifmgd);
 102
 103         if (!timer_pending(&ifmgd->timer) ||
 104             time_before(timeout, ifmgd->timer.expires))
 105                 mod_timer(&ifmgd->timer, timeout);
 106 }
 107
 108 static void mod_beacon_timer(struct ieee80211_sub_if_data *sdata)
 109 {
 110         if (sdata->local->hw.flags & IEEE80211_HW_BEACON_FILTER)
 111                 return;
 112
 113         mod_timer(&sdata->u.mgd.bcn_mon_timer,
 114                   round_jiffies_up(jiffies + IEEE80211_BEACON_LOSS_TIME));
 115 }
 116
 117 static int ecw2cw(int ecw)
 118 {
 119         return (1 << ecw) - 1;
 120 }
 121
 122 /*
 123  * ieee80211_enable_ht should be called only after the operating band
 124  * has been determined as ht configuration depends on the hw's
 125  * HT abilities for a specific band.
 126  */
 127 static u32 ieee80211_enable_ht(struct ieee80211_sub_if_data *sdata,
 128                                struct ieee80211_ht_info *hti,
 129                                const u8 *bssid, u16 ap_ht_cap_flags)
 130 {
 131         struct ieee80211_local *local = sdata->local;
 132         struct ieee80211_supported_band *sband;
 133         struct sta_info *sta;
 134         u32 changed = 0;
 135         u16 ht_opmode;
 136         bool enable_ht = true, ht_changed;
 137         enum nl80211_channel_type channel_type = NL80211_CHAN_NO_HT;
 138
 139         sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
 140
 141         /* HT is not supported */
 142         if (!sband->ht_cap.ht_supported)
 143                 enable_ht = false;
 144
 145         /* check that channel matches the right operating channel */
 146         if (local->hw.conf.channel->center_freq !=
 147             ieee80211_channel_to_frequency(hti->control_chan))
 148                 enable_ht = false;
 149
 150         if (enable_ht) {
 151                 channel_type = NL80211_CHAN_HT20;
 152
 153                 if (!(ap_ht_cap_flags & IEEE80211_HT_CAP_40MHZ_INTOLERANT) &&
 154                     (sband->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40) &&
 155                     (hti->ht_param & IEEE80211_HT_PARAM_CHAN_WIDTH_ANY)) {
 156                         switch(hti->ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
 157                         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
 158                                 if (!(local->hw.conf.channel->flags &
 159                                     IEEE80211_CHAN_NO_HT40PLUS))
 160                                         channel_type = NL80211_CHAN_HT40PLUS;
 161                                 break;
 162                         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
 163                                 if (!(local->hw.conf.channel->flags &
 164                                     IEEE80211_CHAN_NO_HT40MINUS))
 165                                         channel_type = NL80211_CHAN_HT40MINUS;
 166                                 break;
 167                         }
 168                 }
 169         }
 170
 171         ht_changed = conf_is_ht(&local->hw.conf) != enable_ht ||
 172                      channel_type != local->hw.conf.channel_type;
 173
 174         local->oper_channel_type = channel_type;
 175
 176         if (ht_changed) {
 177                 /* channel_type change automatically detected */
 178                 ieee80211_hw_config(local, 0);
 179
 180                 rcu_read_lock();
 181                 sta = sta_info_get(sdata, bssid);
 182                 if (sta)
 183                         rate_control_rate_update(local, sband, sta,
 184                                                  IEEE80211_RC_HT_CHANGED);
 185                 rcu_read_unlock();
 186         }
 187
 188         /* disable HT */
 189         if (!enable_ht)
 190                 return 0;
 191
 192         ht_opmode = le16_to_cpu(hti->operation_mode);
 193
 194         /* if bss configuration changed store the new one */
 195         if (!sdata->ht_opmode_valid ||
 196             sdata->vif.bss_conf.ht_operation_mode != ht_opmode) {
 197                 changed |= BSS_CHANGED_HT;
 198                 sdata->vif.bss_conf.ht_operation_mode = ht_opmode;
 199                 sdata->ht_opmode_valid = true;
 200         }
 201
 202         return changed;
 203 }
 204
 205 /* frame sending functions */
 206
 207 static void ieee80211_send_deauth_disassoc(struct ieee80211_sub_if_data *sdata,
 208                                            const u8 *bssid, u16 stype, u16 reason,
 209                                            void *cookie)
 210 {
 211         struct ieee80211_local *local = sdata->local;
 212         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 213         struct sk_buff *skb;
 214         struct ieee80211_mgmt *mgmt;
 215
 216         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*mgmt));
 217         if (!skb) {
 218                 printk(KERN_DEBUG "%s: failed to allocate buffer for "
 219                        "deauth/disassoc frame\n", sdata->name);
 220                 return;
 221         }
 222         skb_reserve(skb, local->hw.extra_tx_headroom);
 223
 224         mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
 225         memset(mgmt, 0, 24);
 226         memcpy(mgmt->da, bssid, ETH_ALEN);
 227         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
 228         memcpy(mgmt->bssid, bssid, ETH_ALEN);
 229         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT | stype);
 230         skb_put(skb, 2);
 231         /* u.deauth.reason_code == u.disassoc.reason_code */
 232         mgmt->u.deauth.reason_code = cpu_to_le16(reason);
 233
 234         if (stype == IEEE80211_STYPE_DEAUTH)
 235                 if (cookie)
 236                         __cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
 237                 else
 238                         cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
 239         else
 240                 if (cookie)
 241                         __cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
 242                 else
 243                         cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
 244         if (!(ifmgd->flags & IEEE80211_STA_MFP_ENABLED))
 245                 IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 246         ieee80211_tx_skb(sdata, skb);
 247 }
 248
 249 void ieee80211_send_pspoll(struct ieee80211_local *local,
 250                            struct ieee80211_sub_if_data *sdata)
 251 {
 252         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 253         struct ieee80211_pspoll *pspoll;
 254         struct sk_buff *skb;
 255         u16 fc;
 256
 257         skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*pspoll));
 258         if (!skb) {
 259                 printk(KERN_DEBUG "%s: failed to allocate buffer for "
 260                        "pspoll frame\n", sdata->name);
 261                 return;
 262         }
 263         skb_reserve(skb, local->hw.extra_tx_headroom);
 264
 265         pspoll = (struct ieee80211_pspoll *) skb_put(skb, sizeof(*pspoll));
 266         memset(pspoll, 0, sizeof(*pspoll));
 267         fc = IEEE80211_FTYPE_CTL | IEEE80211_STYPE_PSPOLL | IEEE80211_FCTL_PM;
 268         pspoll->frame_control = cpu_to_le16(fc);
 269         pspoll->aid = cpu_to_le16(ifmgd->aid);
 270
 271         /* aid in PS-Poll has its two MSBs each set to 1 */
 272         pspoll->aid |= cpu_to_le16(1 << 15 | 1 << 14);
 273
 274         memcpy(pspoll->bssid, ifmgd->bssid, ETH_ALEN);
 275         memcpy(pspoll->ta, sdata->vif.addr, ETH_ALEN);
 276
 277         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 278         ieee80211_tx_skb(sdata, skb);
 279 }
 280
 281 void ieee80211_send_nullfunc(struct ieee80211_local *local,
 282                              struct ieee80211_sub_if_data *sdata,
 283                              int powersave)
 284 {
 285         struct sk_buff *skb;
 286         struct ieee80211_hdr *nullfunc;
 287         __le16 fc;
 288
 289         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
 290                 return;
 291
 292         skb = dev_alloc_skb(local->hw.extra_tx_headroom + 24);
 293         if (!skb) {
 294                 printk(KERN_DEBUG "%s: failed to allocate buffer for nullfunc "
 295                        "frame\n", sdata->name);
 296                 return;
 297         }
 298         skb_reserve(skb, local->hw.extra_tx_headroom);
 299
 300         nullfunc = (struct ieee80211_hdr *) skb_put(skb, 24);
 301         memset(nullfunc, 0, 24);
 302         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_NULLFUNC |
 303                          IEEE80211_FCTL_TODS);
 304         if (powersave)
 305                 fc |= cpu_to_le16(IEEE80211_FCTL_PM);
 306         nullfunc->frame_control = fc;
 307         memcpy(nullfunc->addr1, sdata->u.mgd.bssid, ETH_ALEN);
 308         memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
 309         memcpy(nullfunc->addr3, sdata->u.mgd.bssid, ETH_ALEN);
 310
 311         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 312         ieee80211_tx_skb(sdata, skb);
 313 }
 314
 315 /* spectrum management related things */
 316 static void ieee80211_chswitch_work(struct work_struct *work)
 317 {
 318         struct ieee80211_sub_if_data *sdata =
 319                 container_of(work, struct ieee80211_sub_if_data, u.mgd.chswitch_work);
 320         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 321
 322         if (!ieee80211_sdata_running(sdata))
 323                 return;
 324
 325         mutex_lock(&ifmgd->mtx);
 326         if (!ifmgd->associated)
 327                 goto out;
 328
 329         sdata->local->oper_channel = sdata->local->csa_channel;
 330         ieee80211_hw_config(sdata->local, IEEE80211_CONF_CHANGE_CHANNEL);
 331
 332         /* XXX: shouldn't really modify cfg80211-owned data! */
 333         ifmgd->associated->cbss.channel = sdata->local->oper_channel;
 334
 335         ieee80211_wake_queues_by_reason(&sdata->local->hw,
 336                                         IEEE80211_QUEUE_STOP_REASON_CSA);
 337  out:
 338         ifmgd->flags &= ~IEEE80211_STA_CSA_RECEIVED;
 339         mutex_unlock(&ifmgd->mtx);
 340 }
 341
 342 static void ieee80211_chswitch_timer(unsigned long data)
 343 {
 344         struct ieee80211_sub_if_data *sdata =
 345                 (struct ieee80211_sub_if_data *) data;
 346         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 347
 348         if (sdata->local->quiescing) {
 349                 set_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running);
 350                 return;
 351         }
 352
 353         ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
 354 }
 355
 356 void ieee80211_sta_process_chanswitch(struct ieee80211_sub_if_data *sdata,
 357                                       struct ieee80211_channel_sw_ie *sw_elem,
 358                                       struct ieee80211_bss *bss)
 359 {
 360         struct ieee80211_channel *new_ch;
 361         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 362         int new_freq = ieee80211_channel_to_frequency(sw_elem->new_ch_num);
 363
 364         ASSERT_MGD_MTX(ifmgd);
 365
 366         if (!ifmgd->associated)
 367                 return;
 368
 369         if (sdata->local->scanning)
 370                 return;
 371
 372         /* Disregard subsequent beacons if we are already running a timer
 373            processing a CSA */
 374
 375         if (ifmgd->flags & IEEE80211_STA_CSA_RECEIVED)
 376                 return;
 377
 378         new_ch = ieee80211_get_channel(sdata->local->hw.wiphy, new_freq);
 379         if (!new_ch || new_ch->flags & IEEE80211_CHAN_DISABLED)
 380                 return;
 381
 382         sdata->local->csa_channel = new_ch;
 383
 384         if (sw_elem->count <= 1) {
 385                 ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
 386         } else {
 387                 ieee80211_stop_queues_by_reason(&sdata->local->hw,
 388                                         IEEE80211_QUEUE_STOP_REASON_CSA);
 389                 ifmgd->flags |= IEEE80211_STA_CSA_RECEIVED;
 390                 mod_timer(&ifmgd->chswitch_timer,
 391                           jiffies +
 392                           msecs_to_jiffies(sw_elem->count *
 393                                            bss->cbss.beacon_interval));
 394         }
 395 }
 396
 397 static void ieee80211_handle_pwr_constr(struct ieee80211_sub_if_data *sdata,
 398                                         u16 capab_info, u8 *pwr_constr_elem,
 399                                         u8 pwr_constr_elem_len)
 400 {
 401         struct ieee80211_conf *conf = &sdata->local->hw.conf;
 402
 403         if (!(capab_info & WLAN_CAPABILITY_SPECTRUM_MGMT))
 404                 return;
 405
 406         /* Power constraint IE length should be 1 octet */
 407         if (pwr_constr_elem_len != 1)
 408                 return;
 409
 410         if ((*pwr_constr_elem <= conf->channel->max_power) &&
 411             (*pwr_constr_elem != sdata->local->power_constr_level)) {
 412                 sdata->local->power_constr_level = *pwr_constr_elem;
 413                 ieee80211_hw_config(sdata->local, 0);
 414         }
 415 }
 416
 417 /* powersave */
 418 static void ieee80211_enable_ps(struct ieee80211_local *local,
 419                                 struct ieee80211_sub_if_data *sdata)
 420 {
 421         struct ieee80211_conf *conf = &local->hw.conf;
 422
 423         /*
 424          * If we are scanning right now then the parameters will
 425          * take effect when scan finishes.
 426          */
 427         if (local->scanning)
 428                 return;
 429
 430         if (conf->dynamic_ps_timeout > 0 &&
 431             !(local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_PS)) {
 432                 mod_timer(&local->dynamic_ps_timer, jiffies +
 433                           msecs_to_jiffies(conf->dynamic_ps_timeout));
 434         } else {
 435                 if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
 436                         ieee80211_send_nullfunc(local, sdata, 1);
 437                 conf->flags |= IEEE80211_CONF_PS;
 438                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
 439         }
 440 }
 441
 442 static void ieee80211_change_ps(struct ieee80211_local *local)
 443 {
 444         struct ieee80211_conf *conf = &local->hw.conf;
 445
 446         if (local->ps_sdata) {
 447                 ieee80211_enable_ps(local, local->ps_sdata);
 448         } else if (conf->flags & IEEE80211_CONF_PS) {
 449                 conf->flags &= ~IEEE80211_CONF_PS;
 450                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
 451                 del_timer_sync(&local->dynamic_ps_timer);
 452                 cancel_work_sync(&local->dynamic_ps_enable_work);
 453         }
 454 }
 455
 456 /* need to hold RTNL or interface lock */
 457 void ieee80211_recalc_ps(struct ieee80211_local *local, s32 latency)
 458 {
 459         struct ieee80211_sub_if_data *sdata, *found = NULL;
 460         int count = 0;
 461
 462         if (!(local->hw.flags & IEEE80211_HW_SUPPORTS_PS)) {
 463                 local->ps_sdata = NULL;
 464                 return;
 465         }
 466
 467         if (!list_empty(&local->work_list)) {
 468                 local->ps_sdata = NULL;
 469                 goto change;
 470         }
 471
 472         list_for_each_entry(sdata, &local->interfaces, list) {
 473                 if (!ieee80211_sdata_running(sdata))
 474                         continue;
 475                 if (sdata->vif.type != NL80211_IFTYPE_STATION)
 476                         continue;
 477                 found = sdata;
 478                 count++;
 479         }
 480
 481         if (count == 1 && found->u.mgd.powersave &&
 482             found->u.mgd.associated &&
 483             !(found->u.mgd.flags & (IEEE80211_STA_BEACON_POLL |
 484                                     IEEE80211_STA_CONNECTION_POLL))) {
 485                 s32 beaconint_us;
 486
 487                 if (latency < 0)
 488                         latency = pm_qos_requirement(PM_QOS_NETWORK_LATENCY);
 489
 490                 beaconint_us = ieee80211_tu_to_usec(
 491                                         found->vif.bss_conf.beacon_int);
 492
 493                 if (beaconint_us > latency) {
 494                         local->ps_sdata = NULL;
 495                 } else {
 496                         u8 dtimper = found->vif.bss_conf.dtim_period;
 497                         int maxslp = 1;
 498
 499                         if (dtimper > 1)
 500                                 maxslp = min_t(int, dtimper,
 501                                                     latency / beaconint_us);
 502
 503                         local->hw.conf.max_sleep_period = maxslp;
 504                         local->ps_sdata = found;
 505                 }
 506         } else {
 507                 local->ps_sdata = NULL;
 508         }
 509
 510  change:
 511         ieee80211_change_ps(local);
 512 }
 513
 514 void ieee80211_dynamic_ps_disable_work(struct work_struct *work)
 515 {
 516         struct ieee80211_local *local =
 517                 container_of(work, struct ieee80211_local,
 518                              dynamic_ps_disable_work);
 519
 520         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
 521                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
 522                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
 523         }
 524
 525         ieee80211_wake_queues_by_reason(&local->hw,
 526                                         IEEE80211_QUEUE_STOP_REASON_PS);
 527 }
 528
 529 void ieee80211_dynamic_ps_enable_work(struct work_struct *work)
 530 {
 531         struct ieee80211_local *local =
 532                 container_of(work, struct ieee80211_local,
 533                              dynamic_ps_enable_work);
 534         struct ieee80211_sub_if_data *sdata = local->ps_sdata;
 535
 536         /* can only happen when PS was just disabled anyway */
 537         if (!sdata)
 538                 return;
 539
 540         if (local->hw.conf.flags & IEEE80211_CONF_PS)
 541                 return;
 542
 543         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
 544                 ieee80211_send_nullfunc(local, sdata, 1);
 545
 546         local->hw.conf.flags |= IEEE80211_CONF_PS;
 547         ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
 548 }
 549
 550 void ieee80211_dynamic_ps_timer(unsigned long data)
 551 {
 552         struct ieee80211_local *local = (void *) data;
 553
 554         if (local->quiescing || local->suspended)
 555                 return;
 556
 557         ieee80211_queue_work(&local->hw, &local->dynamic_ps_enable_work);
 558 }
 559
 560 /* MLME */
 561 static void ieee80211_sta_wmm_params(struct ieee80211_local *local,
 562                                      struct ieee80211_if_managed *ifmgd,
 563                                      u8 *wmm_param, size_t wmm_param_len)
 564 {
 565         struct ieee80211_tx_queue_params params;
 566         size_t left;
 567         int count;
 568         u8 *pos;
 569
 570         if (local->hw.queues < 4)
 571                 return;
 572
 573         if (!wmm_param)
 574                 return;
 575
 576         if (wmm_param_len < 8 || wmm_param[5] /* version */ != 1)
 577                 return;
 578         count = wmm_param[6] & 0x0f;
 579         if (count == ifmgd->wmm_last_param_set)
 580                 return;
 581         ifmgd->wmm_last_param_set = count;
 582
 583         pos = wmm_param + 8;
 584         left = wmm_param_len - 8;
 585
 586         memset(&params, 0, sizeof(params));
 587
 588         local->wmm_acm = 0;
 589         for (; left >= 4; left -= 4, pos += 4) {
 590                 int aci = (pos[0] >> 5) & 0x03;
 591                 int acm = (pos[0] >> 4) & 0x01;
 592                 int queue;
 593
 594                 switch (aci) {
 595                 case 1: /* AC_BK */
 596                         queue = 3;
 597                         if (acm)
 598                                 local->wmm_acm |= BIT(1) | BIT(2); /* BK/- */
 599                         break;
 600                 case 2: /* AC_VI */
 601                         queue = 1;
 602                         if (acm)
 603                                 local->wmm_acm |= BIT(4) | BIT(5); /* CL/VI */
 604                         break;
 605                 case 3: /* AC_VO */
 606                         queue = 0;
 607                         if (acm)
 608                                 local->wmm_acm |= BIT(6) | BIT(7); /* VO/NC */
 609                         break;
 610                 case 0: /* AC_BE */
 611                 default:
 612                         queue = 2;
 613                         if (acm)
 614                                 local->wmm_acm |= BIT(0) | BIT(3); /* BE/EE */
 615                         break;
 616                 }
 617
 618                 params.aifs = pos[0] & 0x0f;
 619                 params.cw_max = ecw2cw((pos[1] & 0xf0) >> 4);
 620                 params.cw_min = ecw2cw(pos[1] & 0x0f);
 621                 params.txop = get_unaligned_le16(pos + 2);
 622 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
 623                 printk(KERN_DEBUG "%s: WMM queue=%d aci=%d acm=%d aifs=%d "
 624                        "cWmin=%d cWmax=%d txop=%d\n",
 625                        wiphy_name(local->hw.wiphy), queue, aci, acm,
 626                        params.aifs, params.cw_min, params.cw_max, params.txop);
 627 #endif
 628                 if (drv_conf_tx(local, queue, &params) && local->ops->conf_tx)
 629                         printk(KERN_DEBUG "%s: failed to set TX queue "
 630                                "parameters for queue %d\n",
 631                                wiphy_name(local->hw.wiphy), queue);
 632         }
 633 }
 634
 635 static u32 ieee80211_handle_bss_capability(struct ieee80211_sub_if_data *sdata,
 636                                            u16 capab, bool erp_valid, u8 erp)
 637 {
 638         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
 639         u32 changed = 0;
 640         bool use_protection;
 641         bool use_short_preamble;
 642         bool use_short_slot;
 643
 644         if (erp_valid) {
 645                 use_protection = (erp & WLAN_ERP_USE_PROTECTION) != 0;
 646                 use_short_preamble = (erp & WLAN_ERP_BARKER_PREAMBLE) == 0;
 647         } else {
 648                 use_protection = false;
 649                 use_short_preamble = !!(capab & WLAN_CAPABILITY_SHORT_PREAMBLE);
 650         }
 651
 652         use_short_slot = !!(capab & WLAN_CAPABILITY_SHORT_SLOT_TIME);
 653
 654         if (use_protection != bss_conf->use_cts_prot) {
 655                 bss_conf->use_cts_prot = use_protection;
 656                 changed |= BSS_CHANGED_ERP_CTS_PROT;
 657         }
 658
 659         if (use_short_preamble != bss_conf->use_short_preamble) {
 660                 bss_conf->use_short_preamble = use_short_preamble;
 661                 changed |= BSS_CHANGED_ERP_PREAMBLE;
 662         }
 663
 664         if (use_short_slot != bss_conf->use_short_slot) {
 665                 bss_conf->use_short_slot = use_short_slot;
 666                 changed |= BSS_CHANGED_ERP_SLOT;
 667         }
 668
 669         return changed;
 670 }
 671
 672 static void ieee80211_set_associated(struct ieee80211_sub_if_data *sdata,
 673                                      struct ieee80211_bss *bss,
 674                                      u32 bss_info_changed)
 675 {
 676         struct ieee80211_local *local = sdata->local;
 677
 678         bss_info_changed |= BSS_CHANGED_ASSOC;
 679         /* set timing information */
 680         sdata->vif.bss_conf.beacon_int = bss->cbss.beacon_interval;
 681         sdata->vif.bss_conf.timestamp = bss->cbss.tsf;
 682         sdata->vif.bss_conf.dtim_period = bss->dtim_period;
 683
 684         bss_info_changed |= BSS_CHANGED_BEACON_INT;
 685         bss_info_changed |= ieee80211_handle_bss_capability(sdata,
 686                 bss->cbss.capability, bss->has_erp_value, bss->erp_value);
 687
 688         sdata->u.mgd.associated = bss;
 689         memcpy(sdata->u.mgd.bssid, bss->cbss.bssid, ETH_ALEN);
 690
 691         /* just to be sure */
 692         sdata->u.mgd.flags &= ~(IEEE80211_STA_CONNECTION_POLL |
 693                                 IEEE80211_STA_BEACON_POLL);
 694
 695         /*
 696          * Always handle WMM once after association regardless
 697          * of the first value the AP uses. Setting -1 here has
 698          * that effect because the AP values is an unsigned
 699          * 4-bit value.
 700          */
 701         sdata->u.mgd.wmm_last_param_set = -1;
 702
 703         ieee80211_led_assoc(local, 1);
 704
 705         sdata->vif.bss_conf.assoc = 1;
 706         /*
 707          * For now just always ask the driver to update the basic rateset
 708          * when we have associated, we aren't checking whether it actually
 709          * changed or not.
 710          */
 711         bss_info_changed |= BSS_CHANGED_BASIC_RATES;
 712
 713         /* And the BSSID changed - we're associated now */
 714         bss_info_changed |= BSS_CHANGED_BSSID;
 715
 716         ieee80211_bss_info_change_notify(sdata, bss_info_changed);
 717
 718         mutex_lock(&local->iflist_mtx);
 719         ieee80211_recalc_ps(local, -1);
 720         ieee80211_recalc_smps(local, sdata);
 721         mutex_unlock(&local->iflist_mtx);
 722
 723         netif_start_queue(sdata->dev);
 724         netif_carrier_on(sdata->dev);
 725 }
 726
 727 static void ieee80211_set_disassoc(struct ieee80211_sub_if_data *sdata)
 728 {
 729         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 730         struct ieee80211_local *local = sdata->local;
 731         struct sta_info *sta;
 732         u32 changed = 0, config_changed = 0;
 733         u8 bssid[ETH_ALEN];
 734
 735         ASSERT_MGD_MTX(ifmgd);
 736
 737         if (WARN_ON(!ifmgd->associated))
 738                 return;
 739
 740         memcpy(bssid, ifmgd->associated->cbss.bssid, ETH_ALEN);
 741
 742         ifmgd->associated = NULL;
 743         memset(ifmgd->bssid, 0, ETH_ALEN);
 744
 745         /*
 746          * we need to commit the associated = NULL change because the
 747          * scan code uses that to determine whether this iface should
 748          * go to/wake up from powersave or not -- and could otherwise
 749          * wake the queues erroneously.
 750          */
 751         smp_mb();
 752
 753         /*
 754          * Thus, we can only afterwards stop the queues -- to account
 755          * for the case where another CPU is finishing a scan at this
 756          * time -- we don't want the scan code to enable queues.
 757          */
 758
 759         netif_stop_queue(sdata->dev);
 760         netif_carrier_off(sdata->dev);
 761
 762         rcu_read_lock();
 763         sta = sta_info_get(sdata, bssid);
 764         if (sta)
 765                 ieee80211_sta_tear_down_BA_sessions(sta);
 766         rcu_read_unlock();
 767
 768         changed |= ieee80211_reset_erp_info(sdata);
 769
 770         ieee80211_led_assoc(local, 0);
 771         changed |= BSS_CHANGED_ASSOC;
 772         sdata->vif.bss_conf.assoc = false;
 773
 774         ieee80211_set_wmm_default(sdata);
 775
 776         /* channel(_type) changes are handled by ieee80211_hw_config */
 777         local->oper_channel_type = NL80211_CHAN_NO_HT;
 778
 779         /* on the next assoc, re-program HT parameters */
 780         sdata->ht_opmode_valid = false;
 781
 782         local->power_constr_level = 0;
 783
 784         del_timer_sync(&local->dynamic_ps_timer);
 785         cancel_work_sync(&local->dynamic_ps_enable_work);
 786
 787         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
 788                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
 789                 config_changed |= IEEE80211_CONF_CHANGE_PS;
 790         }
 791
 792         ieee80211_hw_config(local, config_changed);
 793
 794         /* And the BSSID changed -- not very interesting here */
 795         changed |= BSS_CHANGED_BSSID;
 796         ieee80211_bss_info_change_notify(sdata, changed);
 797
 798         rcu_read_lock();
 799
 800         sta = sta_info_get(sdata, bssid);
 801         if (!sta) {
 802                 rcu_read_unlock();
 803                 return;
 804         }
 805
 806         sta_info_unlink(&sta);
 807
 808         rcu_read_unlock();
 809
 810         sta_info_destroy(sta);
 811 }
 812
 813 void ieee80211_sta_rx_notify(struct ieee80211_sub_if_data *sdata,
 814                              struct ieee80211_hdr *hdr)
 815 {
 816         /*
 817          * We can postpone the mgd.timer whenever receiving unicast frames
 818          * from AP because we know that the connection is working both ways
 819          * at that time. But multicast frames (and hence also beacons) must
 820          * be ignored here, because we need to trigger the timer during
 821          * data idle periods for sending the periodic probe request to the
 822          * AP we're connected to.
 823          */
 824         if (is_multicast_ether_addr(hdr->addr1))
 825                 return;
 826
 827         mod_timer(&sdata->u.mgd.conn_mon_timer,
 828                   round_jiffies_up(jiffies + IEEE80211_CONNECTION_IDLE_TIME));
 829 }
 830
 831 static void ieee80211_mgd_probe_ap_send(struct ieee80211_sub_if_data *sdata)
 832 {
 833         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 834         const u8 *ssid;
 835
 836         ssid = ieee80211_bss_get_ie(&ifmgd->associated->cbss, WLAN_EID_SSID);
 837         ieee80211_send_probe_req(sdata, ifmgd->associated->cbss.bssid,
 838                                  ssid + 2, ssid[1], NULL, 0);
 839
 840         ifmgd->probe_send_count++;
 841         ifmgd->probe_timeout = jiffies + IEEE80211_PROBE_WAIT;
 842         run_again(ifmgd, ifmgd->probe_timeout);
 843 }
 844
 845 static void ieee80211_mgd_probe_ap(struct ieee80211_sub_if_data *sdata,
 846                                    bool beacon)
 847 {
 848         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 849         bool already = false;
 850
 851         if (!ieee80211_sdata_running(sdata))
 852                 return;
 853
 854         if (sdata->local->scanning)
 855                 return;
 856
 857         mutex_lock(&ifmgd->mtx);
 858
 859         if (!ifmgd->associated)
 860                 goto out;
 861
 862 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
 863         if (beacon && net_ratelimit())
 864                 printk(KERN_DEBUG "%s: detected beacon loss from AP "
 865                        "- sending probe request\n", sdata->name);
 866 #endif
 867
 868         /*
 869          * The driver/our work has already reported this event or the
 870          * connection monitoring has kicked in and we have already sent
 871          * a probe request. Or maybe the AP died and the driver keeps
 872          * reporting until we disassociate...
 873          *
 874          * In either case we have to ignore the current call to this
 875          * function (except for setting the correct probe reason bit)
 876          * because otherwise we would reset the timer every time and
 877          * never check whether we received a probe response!
 878          */
 879         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
 880                             IEEE80211_STA_CONNECTION_POLL))
 881                 already = true;
 882
 883         if (beacon)
 884                 ifmgd->flags |= IEEE80211_STA_BEACON_POLL;
 885         else
 886                 ifmgd->flags |= IEEE80211_STA_CONNECTION_POLL;
 887
 888         if (already)
 889                 goto out;
 890
 891         mutex_lock(&sdata->local->iflist_mtx);
 892         ieee80211_recalc_ps(sdata->local, -1);
 893         mutex_unlock(&sdata->local->iflist_mtx);
 894
 895         ifmgd->probe_send_count = 0;
 896         ieee80211_mgd_probe_ap_send(sdata);
 897  out:
 898         mutex_unlock(&ifmgd->mtx);
 899 }
 900
 901 void ieee80211_beacon_loss_work(struct work_struct *work)
 902 {
 903         struct ieee80211_sub_if_data *sdata =
 904                 container_of(work, struct ieee80211_sub_if_data,
 905                              u.mgd.beacon_loss_work);
 906
 907         ieee80211_mgd_probe_ap(sdata, true);
 908 }
 909
 910 void ieee80211_beacon_loss(struct ieee80211_vif *vif)
 911 {
 912         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
 913
 914         ieee80211_queue_work(&sdata->local->hw, &sdata->u.mgd.beacon_loss_work);
 915 }
 916 EXPORT_SYMBOL(ieee80211_beacon_loss);
 917
 918 static enum rx_mgmt_action __must_check
 919 ieee80211_rx_mgmt_deauth(struct ieee80211_sub_if_data *sdata,
 920                          struct ieee80211_mgmt *mgmt, size_t len)
 921 {
 922         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 923         const u8 *bssid = NULL;
 924         u16 reason_code;
 925
 926         if (len < 24 + 2)
 927                 return RX_MGMT_NONE;
 928
 929         ASSERT_MGD_MTX(ifmgd);
 930
 931         bssid = ifmgd->associated->cbss.bssid;
 932
 933         reason_code = le16_to_cpu(mgmt->u.deauth.reason_code);
 934
 935         printk(KERN_DEBUG "%s: deauthenticated from %pM (Reason: %u)\n",
 936                         sdata->name, bssid, reason_code);
 937
 938         ieee80211_set_disassoc(sdata);
 939         ieee80211_recalc_idle(sdata->local);
 940
 941         return RX_MGMT_CFG80211_DEAUTH;
 942 }
 943
 944
 945 static enum rx_mgmt_action __must_check
 946 ieee80211_rx_mgmt_disassoc(struct ieee80211_sub_if_data *sdata,
 947                            struct ieee80211_mgmt *mgmt, size_t len)
 948 {
 949         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 950         u16 reason_code;
 951
 952         if (len < 24 + 2)
 953                 return RX_MGMT_NONE;
 954
 955         ASSERT_MGD_MTX(ifmgd);
 956
 957         if (WARN_ON(!ifmgd->associated))
 958                 return RX_MGMT_NONE;
 959
 960         if (WARN_ON(memcmp(ifmgd->associated->cbss.bssid, mgmt->sa, ETH_ALEN)))
 961                 return RX_MGMT_NONE;
 962
 963         reason_code = le16_to_cpu(mgmt->u.disassoc.reason_code);
 964
 965         printk(KERN_DEBUG "%s: disassociated from %pM (Reason: %u)\n",
 966                         sdata->name, mgmt->sa, reason_code);
 967
 968         ieee80211_set_disassoc(sdata);
 969         ieee80211_recalc_idle(sdata->local);
 970         return RX_MGMT_CFG80211_DISASSOC;
 971 }
 972
 973
 974 static bool ieee80211_assoc_success(struct ieee80211_work *wk,
 975                                     struct ieee80211_mgmt *mgmt, size_t len)
 976 {
 977         struct ieee80211_sub_if_data *sdata = wk->sdata;
 978         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 979         struct ieee80211_local *local = sdata->local;
 980         struct ieee80211_supported_band *sband;
 981         struct sta_info *sta;
 982         struct ieee80211_bss *bss = wk->assoc.bss;
 983         u8 *pos;
 984         u32 rates, basic_rates;
 985         u16 capab_info, aid;
 986         struct ieee802_11_elems elems;
 987         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
 988         u32 changed = 0;
 989         int i, j, err;
 990         bool have_higher_than_11mbit = false;
 991         u16 ap_ht_cap_flags;
 992
 993         /* AssocResp and ReassocResp have identical structure */
 994
 995         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
 996         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
 997
 998         if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14)))
 999                 printk(KERN_DEBUG "%s: invalid aid value %d; bits 15:14 not "
1000                        "set\n", sdata->name, aid);
1001         aid &= ~(BIT(15) | BIT(14));
1002
1003         pos = mgmt->u.assoc_resp.variable;
1004         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), &elems);
1005
1006         if (!elems.supp_rates) {
1007                 printk(KERN_DEBUG "%s: no SuppRates element in AssocResp\n",
1008                        sdata->name);
1009                 return false;
1010         }
1011
1012         ifmgd->aid = aid;
1013
1014         sta = sta_info_alloc(sdata, bss->cbss.bssid, GFP_KERNEL);
1015         if (!sta) {
1016                 printk(KERN_DEBUG "%s: failed to alloc STA entry for"
1017                        " the AP\n", sdata->name);
1018                 return false;
1019         }
1020
1021         set_sta_flags(sta, WLAN_STA_AUTH | WLAN_STA_ASSOC |
1022                            WLAN_STA_ASSOC_AP);
1023         if (!(ifmgd->flags & IEEE80211_STA_CONTROL_PORT))
1024                 set_sta_flags(sta, WLAN_STA_AUTHORIZED);
1025
1026         rates = 0;
1027         basic_rates = 0;
1028         sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
1029
1030         for (i = 0; i < elems.supp_rates_len; i++) {
1031                 int rate = (elems.supp_rates[i] & 0x7f) * 5;
1032                 bool is_basic = !!(elems.supp_rates[i] & 0x80);
1033
1034                 if (rate > 110)
1035                         have_higher_than_11mbit = true;
1036
1037                 for (j = 0; j < sband->n_bitrates; j++) {
1038                         if (sband->bitrates[j].bitrate == rate) {
1039                                 rates |= BIT(j);
1040                                 if (is_basic)
1041                                         basic_rates |= BIT(j);
1042                                 break;
1043                         }
1044                 }
1045         }
1046
1047         for (i = 0; i < elems.ext_supp_rates_len; i++) {
1048                 int rate = (elems.ext_supp_rates[i] & 0x7f) * 5;
1049                 bool is_basic = !!(elems.ext_supp_rates[i] & 0x80);
1050
1051                 if (rate > 110)
1052                         have_higher_than_11mbit = true;
1053
1054                 for (j = 0; j < sband->n_bitrates; j++) {
1055                         if (sband->bitrates[j].bitrate == rate) {
1056                                 rates |= BIT(j);
1057                                 if (is_basic)
1058                                         basic_rates |= BIT(j);
1059                                 break;
1060                         }
1061                 }
1062         }
1063
1064         sta->sta.supp_rates[local->hw.conf.channel->band] = rates;
1065         sdata->vif.bss_conf.basic_rates = basic_rates;
1066
1067         /* cf. IEEE 802.11 9.2.12 */
1068         if (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ &&
1069             have_higher_than_11mbit)
1070                 sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
1071         else
1072                 sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;
1073
1074         if (elems.ht_cap_elem && !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1075                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1076                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1077
1078         ap_ht_cap_flags = sta->sta.ht_cap.cap;
1079
1080         rate_control_rate_init(sta);
1081
1082         if (ifmgd->flags & IEEE80211_STA_MFP_ENABLED)
1083                 set_sta_flags(sta, WLAN_STA_MFP);
1084
1085         if (elems.wmm_param)
1086                 set_sta_flags(sta, WLAN_STA_WME);
1087
1088         err = sta_info_insert(sta);
1089         sta = NULL;
1090         if (err) {
1091                 printk(KERN_DEBUG "%s: failed to insert STA entry for"
1092                        " the AP (error %d)\n", sdata->name, err);
1093                 return RX_MGMT_CFG80211_ASSOC_ERROR;
1094         }
1095
1096         if (elems.wmm_param)
1097                 ieee80211_sta_wmm_params(local, ifmgd, elems.wmm_param,
1098                                          elems.wmm_param_len);
1099         else
1100                 ieee80211_set_wmm_default(sdata);
1101
1102         if (elems.ht_info_elem && elems.wmm_param &&
1103             (sdata->local->hw.queues >= 4) &&
1104             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N))
1105                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1106                                                bss->cbss.bssid,
1107                                                ap_ht_cap_flags);
1108
1109         /* set AID and assoc capability,
1110          * ieee80211_set_associated() will tell the driver */
1111         bss_conf->aid = aid;
1112         bss_conf->assoc_capability = capab_info;
1113         ieee80211_set_associated(sdata, bss, changed);
1114
1115         /*
1116          * Start timer to probe the connection to the AP now.
1117          * Also start the timer that will detect beacon loss.
1118          */
1119         ieee80211_sta_rx_notify(sdata, (struct ieee80211_hdr *)mgmt);
1120         mod_beacon_timer(sdata);
1121
1122         return true;
1123 }
1124
1125
1126 static void ieee80211_rx_bss_info(struct ieee80211_sub_if_data *sdata,
1127                                   struct ieee80211_mgmt *mgmt,
1128                                   size_t len,
1129                                   struct ieee80211_rx_status *rx_status,
1130                                   struct ieee802_11_elems *elems,
1131                                   bool beacon)
1132 {
1133         struct ieee80211_local *local = sdata->local;
1134         int freq;
1135         struct ieee80211_bss *bss;
1136         struct ieee80211_channel *channel;
1137
1138         if (elems->ds_params && elems->ds_params_len == 1)
1139                 freq = ieee80211_channel_to_frequency(elems->ds_params[0]);
1140         else
1141                 freq = rx_status->freq;
1142
1143         channel = ieee80211_get_channel(local->hw.wiphy, freq);
1144
1145         if (!channel || channel->flags & IEEE80211_CHAN_DISABLED)
1146                 return;
1147
1148         bss = ieee80211_bss_info_update(local, rx_status, mgmt, len, elems,
1149                                         channel, beacon);
1150         if (bss)
1151                 ieee80211_rx_bss_put(local, bss);
1152
1153         if (!sdata->u.mgd.associated)
1154                 return;
1155
1156         if (elems->ch_switch_elem && (elems->ch_switch_elem_len == 3) &&
1157             (memcmp(mgmt->bssid, sdata->u.mgd.associated->cbss.bssid,
1158                                                         ETH_ALEN) == 0)) {
1159                 struct ieee80211_channel_sw_ie *sw_elem =
1160                         (struct ieee80211_channel_sw_ie *)elems->ch_switch_elem;
1161                 ieee80211_sta_process_chanswitch(sdata, sw_elem, bss);
1162         }
1163 }
1164
1165
1166 static void ieee80211_rx_mgmt_probe_resp(struct ieee80211_sub_if_data *sdata,
1167                                          struct sk_buff *skb)
1168 {
1169         struct ieee80211_mgmt *mgmt = (void *)skb->data;
1170         struct ieee80211_if_managed *ifmgd;
1171         struct ieee80211_rx_status *rx_status = (void *) skb->cb;
1172         size_t baselen, len = skb->len;
1173         struct ieee802_11_elems elems;
1174
1175         ifmgd = &sdata->u.mgd;
1176
1177         ASSERT_MGD_MTX(ifmgd);
1178
1179         if (memcmp(mgmt->da, sdata->vif.addr, ETH_ALEN))
1180                 return; /* ignore ProbeResp to foreign address */
1181
1182         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
1183         if (baselen > len)
1184                 return;
1185
1186         ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - baselen,
1187                                 &elems);
1188
1189         ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems, false);
1190
1191         if (ifmgd->associated &&
1192             memcmp(mgmt->bssid, ifmgd->associated->cbss.bssid, ETH_ALEN) == 0 &&
1193             ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1194                             IEEE80211_STA_CONNECTION_POLL)) {
1195                 ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
1196                                   IEEE80211_STA_BEACON_POLL);
1197                 mutex_lock(&sdata->local->iflist_mtx);
1198                 ieee80211_recalc_ps(sdata->local, -1);
1199                 mutex_unlock(&sdata->local->iflist_mtx);
1200                 /*
1201                  * We've received a probe response, but are not sure whether
1202                  * we have or will be receiving any beacons or data, so let's
1203                  * schedule the timers again, just in case.
1204                  */
1205                 mod_beacon_timer(sdata);
1206                 mod_timer(&ifmgd->conn_mon_timer,
1207                           round_jiffies_up(jiffies +
1208                                            IEEE80211_CONNECTION_IDLE_TIME));
1209         }
1210 }
1211
1212 /*
1213  * This is the canonical list of information elements we care about,
1214  * the filter code also gives us all changes to the Microsoft OUI
1215  * (00:50:F2) vendor IE which is used for WMM which we need to track.
1216  *
1217  * We implement beacon filtering in software since that means we can
1218  * avoid processing the frame here and in cfg80211, and userspace
1219  * will not be able to tell whether the hardware supports it or not.
1220  *
1221  * XXX: This list needs to be dynamic -- userspace needs to be able to
1222  *      add items it requires. It also needs to be able to tell us to
1223  *      look out for other vendor IEs.
1224  */
1225 static const u64 care_about_ies =
1226         (1ULL << WLAN_EID_COUNTRY) |
1227         (1ULL << WLAN_EID_ERP_INFO) |
1228         (1ULL << WLAN_EID_CHANNEL_SWITCH) |
1229         (1ULL << WLAN_EID_PWR_CONSTRAINT) |
1230         (1ULL << WLAN_EID_HT_CAPABILITY) |
1231         (1ULL << WLAN_EID_HT_INFORMATION);
1232
1233 static void ieee80211_rx_mgmt_beacon(struct ieee80211_sub_if_data *sdata,
1234                                      struct ieee80211_mgmt *mgmt,
1235                                      size_t len,
1236                                      struct ieee80211_rx_status *rx_status)
1237 {
1238         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1239         size_t baselen;
1240         struct ieee802_11_elems elems;
1241         struct ieee80211_local *local = sdata->local;
1242         u32 changed = 0;
1243         bool erp_valid, directed_tim = false;
1244         u8 erp_value = 0;
1245         u32 ncrc;
1246         u8 *bssid;
1247
1248         ASSERT_MGD_MTX(ifmgd);
1249
1250         /* Process beacon from the current BSS */
1251         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
1252         if (baselen > len)
1253                 return;
1254
1255         if (rx_status->freq != local->hw.conf.channel->center_freq)
1256                 return;
1257
1258         /*
1259          * We might have received a number of frames, among them a
1260          * disassoc frame and a beacon...
1261          */
1262         if (!ifmgd->associated)
1263                 return;
1264
1265         bssid = ifmgd->associated->cbss.bssid;
1266
1267         /*
1268          * And in theory even frames from a different AP we were just
1269          * associated to a split-second ago!
1270          */
1271         if (memcmp(bssid, mgmt->bssid, ETH_ALEN) != 0)
1272                 return;
1273
1274         if (ifmgd->flags & IEEE80211_STA_BEACON_POLL) {
1275 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1276                 if (net_ratelimit()) {
1277                         printk(KERN_DEBUG "%s: cancelling probereq poll due "
1278                                "to a received beacon\n", sdata->name);
1279                 }
1280 #endif
1281                 ifmgd->flags &= ~IEEE80211_STA_BEACON_POLL;
1282                 mutex_lock(&local->iflist_mtx);
1283                 ieee80211_recalc_ps(local, -1);
1284                 mutex_unlock(&local->iflist_mtx);
1285         }
1286
1287         /*
1288          * Push the beacon loss detection into the future since
1289          * we are processing a beacon from the AP just now.
1290          */
1291         mod_beacon_timer(sdata);
1292
1293         ncrc = crc32_be(0, (void *)&mgmt->u.beacon.beacon_int, 4);
1294         ncrc = ieee802_11_parse_elems_crc(mgmt->u.beacon.variable,
1295                                           len - baselen, &elems,
1296                                           care_about_ies, ncrc);
1297
1298         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK)
1299                 directed_tim = ieee80211_check_tim(elems.tim, elems.tim_len,
1300                                                    ifmgd->aid);
1301
1302         if (ncrc != ifmgd->beacon_crc) {
1303                 ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems,
1304                                       true);
1305
1306                 ieee80211_sta_wmm_params(local, ifmgd, elems.wmm_param,
1307                                          elems.wmm_param_len);
1308         }
1309
1310         if (local->hw.flags & IEEE80211_HW_PS_NULLFUNC_STACK) {
1311                 if (directed_tim) {
1312                         if (local->hw.conf.dynamic_ps_timeout > 0) {
1313                                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
1314                                 ieee80211_hw_config(local,
1315                                                     IEEE80211_CONF_CHANGE_PS);
1316                                 ieee80211_send_nullfunc(local, sdata, 0);
1317                         } else {
1318                                 local->pspolling = true;
1319
1320                                 /*
1321                                  * Here is assumed that the driver will be
1322                                  * able to send ps-poll frame and receive a
1323                                  * response even though power save mode is
1324                                  * enabled, but some drivers might require
1325                                  * to disable power save here. This needs
1326                                  * to be investigated.
1327                                  */
1328                                 ieee80211_send_pspoll(local, sdata);
1329                         }
1330                 }
1331         }
1332
1333         if (ncrc == ifmgd->beacon_crc)
1334                 return;
1335         ifmgd->beacon_crc = ncrc;
1336
1337         if (elems.erp_info && elems.erp_info_len >= 1) {
1338                 erp_valid = true;
1339                 erp_value = elems.erp_info[0];
1340         } else {
1341                 erp_valid = false;
1342         }
1343         changed |= ieee80211_handle_bss_capability(sdata,
1344                         le16_to_cpu(mgmt->u.beacon.capab_info),
1345                         erp_valid, erp_value);
1346
1347
1348         if (elems.ht_cap_elem && elems.ht_info_elem && elems.wmm_param &&
1349             !(ifmgd->flags & IEEE80211_STA_DISABLE_11N)) {
1350                 struct sta_info *sta;
1351                 struct ieee80211_supported_band *sband;
1352                 u16 ap_ht_cap_flags;
1353
1354                 rcu_read_lock();
1355
1356                 sta = sta_info_get(sdata, bssid);
1357                 if (WARN_ON(!sta)) {
1358                         rcu_read_unlock();
1359                         return;
1360                 }
1361
1362                 sband = local->hw.wiphy->bands[local->hw.conf.channel->band];
1363
1364                 ieee80211_ht_cap_ie_to_sta_ht_cap(sband,
1365                                 elems.ht_cap_elem, &sta->sta.ht_cap);
1366
1367                 ap_ht_cap_flags = sta->sta.ht_cap.cap;
1368
1369                 rcu_read_unlock();
1370
1371                 changed |= ieee80211_enable_ht(sdata, elems.ht_info_elem,
1372                                                bssid, ap_ht_cap_flags);
1373         }
1374
1375         /* Note: country IE parsing is done for us by cfg80211 */
1376         if (elems.country_elem) {
1377                 /* TODO: IBSS also needs this */
1378                 if (elems.pwr_constr_elem)
1379                         ieee80211_handle_pwr_constr(sdata,
1380                                 le16_to_cpu(mgmt->u.probe_resp.capab_info),
1381                                 elems.pwr_constr_elem,
1382                                 elems.pwr_constr_elem_len);
1383         }
1384
1385         ieee80211_bss_info_change_notify(sdata, changed);
1386 }
1387
1388 ieee80211_rx_result ieee80211_sta_rx_mgmt(struct ieee80211_sub_if_data *sdata,
1389                                           struct sk_buff *skb)
1390 {
1391         struct ieee80211_local *local = sdata->local;
1392         struct ieee80211_mgmt *mgmt;
1393         u16 fc;
1394
1395         if (skb->len < 24)
1396                 return RX_DROP_MONITOR;
1397
1398         mgmt = (struct ieee80211_mgmt *) skb->data;
1399         fc = le16_to_cpu(mgmt->frame_control);
1400
1401         switch (fc & IEEE80211_FCTL_STYPE) {
1402         case IEEE80211_STYPE_PROBE_RESP:
1403         case IEEE80211_STYPE_BEACON:
1404         case IEEE80211_STYPE_DEAUTH:
1405         case IEEE80211_STYPE_DISASSOC:
1406         case IEEE80211_STYPE_ACTION:
1407                 skb_queue_tail(&sdata->u.mgd.skb_queue, skb);
1408                 ieee80211_queue_work(&local->hw, &sdata->u.mgd.work);
1409                 return RX_QUEUED;
1410         }
1411
1412         return RX_DROP_MONITOR;
1413 }
1414
1415 static void ieee80211_sta_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
1416                                          struct sk_buff *skb)
1417 {
1418         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1419         struct ieee80211_rx_status *rx_status;
1420         struct ieee80211_mgmt *mgmt;
1421         enum rx_mgmt_action rma = RX_MGMT_NONE;
1422         u16 fc;
1423
1424         rx_status = (struct ieee80211_rx_status *) skb->cb;
1425         mgmt = (struct ieee80211_mgmt *) skb->data;
1426         fc = le16_to_cpu(mgmt->frame_control);
1427
1428         mutex_lock(&ifmgd->mtx);
1429
1430         if (ifmgd->associated &&
1431             memcmp(ifmgd->associated->cbss.bssid, mgmt->bssid,
1432                                                         ETH_ALEN) == 0) {
1433                 switch (fc & IEEE80211_FCTL_STYPE) {
1434                 case IEEE80211_STYPE_BEACON:
1435                         ieee80211_rx_mgmt_beacon(sdata, mgmt, skb->len,
1436                                                  rx_status);
1437                         break;
1438                 case IEEE80211_STYPE_PROBE_RESP:
1439                         ieee80211_rx_mgmt_probe_resp(sdata, skb);
1440                         break;
1441                 case IEEE80211_STYPE_DEAUTH:
1442                         rma = ieee80211_rx_mgmt_deauth(sdata, mgmt, skb->len);
1443                         break;
1444                 case IEEE80211_STYPE_DISASSOC:
1445                         rma = ieee80211_rx_mgmt_disassoc(sdata, mgmt, skb->len);
1446                         break;
1447                 case IEEE80211_STYPE_ACTION:
1448                         /* XXX: differentiate, can only happen for CSA now! */
1449                         ieee80211_sta_process_chanswitch(sdata,
1450                                         &mgmt->u.action.u.chan_switch.sw_elem,
1451                                         ifmgd->associated);
1452                         break;
1453                 }
1454                 mutex_unlock(&ifmgd->mtx);
1455
1456                 switch (rma) {
1457                 case RX_MGMT_NONE:
1458                         /* no action */
1459                         break;
1460                 case RX_MGMT_CFG80211_DEAUTH:
1461                         cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
1462                         break;
1463                 case RX_MGMT_CFG80211_DISASSOC:
1464                         cfg80211_send_disassoc(sdata->dev, (u8 *)mgmt, skb->len);
1465                         break;
1466                 default:
1467                         WARN(1, "unexpected: %d", rma);
1468                 }
1469                 goto out;
1470         }
1471
1472         mutex_unlock(&ifmgd->mtx);
1473
1474         if (skb->len >= 24 + 2 /* mgmt + deauth reason */ &&
1475             (fc & IEEE80211_FCTL_STYPE) == IEEE80211_STYPE_DEAUTH)
1476                 cfg80211_send_deauth(sdata->dev, (u8 *)mgmt, skb->len);
1477
1478  out:
1479         kfree_skb(skb);
1480 }
1481
1482 static void ieee80211_sta_timer(unsigned long data)
1483 {
1484         struct ieee80211_sub_if_data *sdata =
1485                 (struct ieee80211_sub_if_data *) data;
1486         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1487         struct ieee80211_local *local = sdata->local;
1488
1489         if (local->quiescing) {
1490                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
1491                 return;
1492         }
1493
1494         ieee80211_queue_work(&local->hw, &ifmgd->work);
1495 }
1496
1497 static void ieee80211_sta_work(struct work_struct *work)
1498 {
1499         struct ieee80211_sub_if_data *sdata =
1500                 container_of(work, struct ieee80211_sub_if_data, u.mgd.work);
1501         struct ieee80211_local *local = sdata->local;
1502         struct ieee80211_if_managed *ifmgd;
1503         struct sk_buff *skb;
1504
1505         if (!ieee80211_sdata_running(sdata))
1506                 return;
1507
1508         if (local->scanning)
1509                 return;
1510
1511         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
1512                 return;
1513
1514         /*
1515          * ieee80211_queue_work() should have picked up most cases,
1516          * here we'll pick the the rest.
1517          */
1518         if (WARN(local->suspended, "STA MLME work scheduled while "
1519                  "going to suspend\n"))
1520                 return;
1521
1522         ifmgd = &sdata->u.mgd;
1523
1524         /* first process frames to avoid timing out while a frame is pending */
1525         while ((skb = skb_dequeue(&ifmgd->skb_queue)))
1526                 ieee80211_sta_rx_queued_mgmt(sdata, skb);
1527
1528         /* then process the rest of the work */
1529         mutex_lock(&ifmgd->mtx);
1530
1531         if (ifmgd->flags & (IEEE80211_STA_BEACON_POLL |
1532                             IEEE80211_STA_CONNECTION_POLL) &&
1533             ifmgd->associated) {
1534                 u8 bssid[ETH_ALEN];
1535
1536                 memcpy(bssid, ifmgd->associated->cbss.bssid, ETH_ALEN);
1537                 if (time_is_after_jiffies(ifmgd->probe_timeout))
1538                         run_again(ifmgd, ifmgd->probe_timeout);
1539
1540                 else if (ifmgd->probe_send_count < IEEE80211_MAX_PROBE_TRIES) {
1541 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
1542                         printk(KERN_DEBUG "No probe response from AP %pM"
1543                                 " after %dms, try %d\n", bssid,
1544                                 (1000 * IEEE80211_PROBE_WAIT)/HZ,
1545                                 ifmgd->probe_send_count);
1546 #endif
1547                         ieee80211_mgd_probe_ap_send(sdata);
1548                 } else {
1549                         /*
1550                          * We actually lost the connection ... or did we?
1551                          * Let's make sure!
1552                          */
1553                         ifmgd->flags &= ~(IEEE80211_STA_CONNECTION_POLL |
1554                                           IEEE80211_STA_BEACON_POLL);
1555                         printk(KERN_DEBUG "No probe response from AP %pM"
1556                                 " after %dms, disconnecting.\n",
1557                                 bssid, (1000 * IEEE80211_PROBE_WAIT)/HZ);
1558                         ieee80211_set_disassoc(sdata);
1559                         ieee80211_recalc_idle(local);
1560                         mutex_unlock(&ifmgd->mtx);
1561                         /*
1562                          * must be outside lock due to cfg80211,
1563                          * but that's not a problem.
1564                          */
1565                         ieee80211_send_deauth_disassoc(sdata, bssid,
1566                                         IEEE80211_STYPE_DEAUTH,
1567                                         WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
1568                                         NULL);
1569                         mutex_lock(&ifmgd->mtx);
1570                 }
1571         }
1572
1573         mutex_unlock(&ifmgd->mtx);
1574 }
1575
1576 static void ieee80211_sta_bcn_mon_timer(unsigned long data)
1577 {
1578         struct ieee80211_sub_if_data *sdata =
1579                 (struct ieee80211_sub_if_data *) data;
1580         struct ieee80211_local *local = sdata->local;
1581
1582         if (local->quiescing)
1583                 return;
1584
1585         ieee80211_queue_work(&sdata->local->hw, &sdata->u.mgd.beacon_loss_work);
1586 }
1587
1588 static void ieee80211_sta_conn_mon_timer(unsigned long data)
1589 {
1590         struct ieee80211_sub_if_data *sdata =
1591                 (struct ieee80211_sub_if_data *) data;
1592         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1593         struct ieee80211_local *local = sdata->local;
1594
1595         if (local->quiescing)
1596                 return;
1597
1598         ieee80211_queue_work(&local->hw, &ifmgd->monitor_work);
1599 }
1600
1601 static void ieee80211_sta_monitor_work(struct work_struct *work)
1602 {
1603         struct ieee80211_sub_if_data *sdata =
1604                 container_of(work, struct ieee80211_sub_if_data,
1605                              u.mgd.monitor_work);
1606
1607         ieee80211_mgd_probe_ap(sdata, false);
1608 }
1609
1610 static void ieee80211_restart_sta_timer(struct ieee80211_sub_if_data *sdata)
1611 {
1612         if (sdata->vif.type == NL80211_IFTYPE_STATION) {
1613                 sdata->u.mgd.flags &= ~(IEEE80211_STA_BEACON_POLL |
1614                                         IEEE80211_STA_CONNECTION_POLL);
1615
1616                 /* let's probe the connection once */
1617                 ieee80211_queue_work(&sdata->local->hw,
1618                            &sdata->u.mgd.monitor_work);
1619                 /* and do all the other regular work too */
1620                 ieee80211_queue_work(&sdata->local->hw,
1621                            &sdata->u.mgd.work);
1622         }
1623 }
1624
1625 #ifdef CONFIG_PM
1626 void ieee80211_sta_quiesce(struct ieee80211_sub_if_data *sdata)
1627 {
1628         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1629
1630         /*
1631          * we need to use atomic bitops for the running bits
1632          * only because both timers might fire at the same
1633          * time -- the code here is properly synchronised.
1634          */
1635
1636         cancel_work_sync(&ifmgd->work);
1637         cancel_work_sync(&ifmgd->beacon_loss_work);
1638         if (del_timer_sync(&ifmgd->timer))
1639                 set_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running);
1640
1641         cancel_work_sync(&ifmgd->chswitch_work);
1642         if (del_timer_sync(&ifmgd->chswitch_timer))
1643                 set_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running);
1644
1645         cancel_work_sync(&ifmgd->monitor_work);
1646         /* these will just be re-established on connection */
1647         del_timer_sync(&ifmgd->conn_mon_timer);
1648         del_timer_sync(&ifmgd->bcn_mon_timer);
1649 }
1650
1651 void ieee80211_sta_restart(struct ieee80211_sub_if_data *sdata)
1652 {
1653         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1654
1655         if (test_and_clear_bit(TMR_RUNNING_TIMER, &ifmgd->timers_running))
1656                 add_timer(&ifmgd->timer);
1657         if (test_and_clear_bit(TMR_RUNNING_CHANSW, &ifmgd->timers_running))
1658                 add_timer(&ifmgd->chswitch_timer);
1659 }
1660 #endif
1661
1662 /* interface setup */
1663 void ieee80211_sta_setup_sdata(struct ieee80211_sub_if_data *sdata)
1664 {
1665         struct ieee80211_if_managed *ifmgd;
1666
1667         ifmgd = &sdata->u.mgd;
1668         INIT_WORK(&ifmgd->work, ieee80211_sta_work);
1669         INIT_WORK(&ifmgd->monitor_work, ieee80211_sta_monitor_work);
1670         INIT_WORK(&ifmgd->chswitch_work, ieee80211_chswitch_work);
1671         INIT_WORK(&ifmgd->beacon_loss_work, ieee80211_beacon_loss_work);
1672         setup_timer(&ifmgd->timer, ieee80211_sta_timer,
1673                     (unsigned long) sdata);
1674         setup_timer(&ifmgd->bcn_mon_timer, ieee80211_sta_bcn_mon_timer,
1675                     (unsigned long) sdata);
1676         setup_timer(&ifmgd->conn_mon_timer, ieee80211_sta_conn_mon_timer,
1677                     (unsigned long) sdata);
1678         setup_timer(&ifmgd->chswitch_timer, ieee80211_chswitch_timer,
1679                     (unsigned long) sdata);
1680         skb_queue_head_init(&ifmgd->skb_queue);
1681
1682         ifmgd->flags = 0;
1683
1684         mutex_init(&ifmgd->mtx);
1685
1686         if (sdata->local->hw.flags & IEEE80211_HW_SUPPORTS_DYNAMIC_SMPS)
1687                 ifmgd->req_smps = IEEE80211_SMPS_AUTOMATIC;
1688         else
1689                 ifmgd->req_smps = IEEE80211_SMPS_OFF;
1690 }
1691
1692 /* scan finished notification */
1693 void ieee80211_mlme_notify_scan_completed(struct ieee80211_local *local)
1694 {
1695         struct ieee80211_sub_if_data *sdata = local->scan_sdata;
1696
1697         /* Restart STA timers */
1698         rcu_read_lock();
1699         list_for_each_entry_rcu(sdata, &local->interfaces, list)
1700                 ieee80211_restart_sta_timer(sdata);
1701         rcu_read_unlock();
1702 }
1703
1704 int ieee80211_max_network_latency(struct notifier_block *nb,
1705                                   unsigned long data, void *dummy)
1706 {
1707         s32 latency_usec = (s32) data;
1708         struct ieee80211_local *local =
1709                 container_of(nb, struct ieee80211_local,
1710                              network_latency_notifier);
1711
1712         mutex_lock(&local->iflist_mtx);
1713         ieee80211_recalc_ps(local, latency_usec);
1714         mutex_unlock(&local->iflist_mtx);
1715
1716         return 0;
1717 }
1718
1719 /* config hooks */
1720 static enum work_done_result
1721 ieee80211_probe_auth_done(struct ieee80211_work *wk,
1722                           struct sk_buff *skb)
1723 {
1724         if (!skb) {
1725                 cfg80211_send_auth_timeout(wk->sdata->dev, wk->filter_ta);
1726                 return WORK_DONE_DESTROY;
1727         }
1728
1729         if (wk->type == IEEE80211_WORK_AUTH) {
1730                 cfg80211_send_rx_auth(wk->sdata->dev, skb->data, skb->len);
1731                 return WORK_DONE_DESTROY;
1732         }
1733
1734         mutex_lock(&wk->sdata->u.mgd.mtx);
1735         ieee80211_rx_mgmt_probe_resp(wk->sdata, skb);
1736         mutex_unlock(&wk->sdata->u.mgd.mtx);
1737
1738         wk->type = IEEE80211_WORK_AUTH;
1739         wk->probe_auth.tries = 0;
1740         return WORK_DONE_REQUEUE;
1741 }
1742
1743 int ieee80211_mgd_auth(struct ieee80211_sub_if_data *sdata,
1744                        struct cfg80211_auth_request *req)
1745 {
1746         const u8 *ssid;
1747         struct ieee80211_work *wk;
1748         u16 auth_alg;
1749
1750         switch (req->auth_type) {
1751         case NL80211_AUTHTYPE_OPEN_SYSTEM:
1752                 auth_alg = WLAN_AUTH_OPEN;
1753                 break;
1754         case NL80211_AUTHTYPE_SHARED_KEY:
1755                 auth_alg = WLAN_AUTH_SHARED_KEY;
1756                 break;
1757         case NL80211_AUTHTYPE_FT:
1758                 auth_alg = WLAN_AUTH_FT;
1759                 break;
1760         case NL80211_AUTHTYPE_NETWORK_EAP:
1761                 auth_alg = WLAN_AUTH_LEAP;
1762                 break;
1763         default:
1764                 return -EOPNOTSUPP;
1765         }
1766
1767         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
1768         if (!wk)
1769                 return -ENOMEM;
1770
1771         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);;
1772
1773         if (req->ie && req->ie_len) {
1774                 memcpy(wk->ie, req->ie, req->ie_len);
1775                 wk->ie_len = req->ie_len;
1776         }
1777
1778         if (req->key && req->key_len) {
1779                 wk->probe_auth.key_len = req->key_len;
1780                 wk->probe_auth.key_idx = req->key_idx;
1781                 memcpy(wk->probe_auth.key, req->key, req->key_len);
1782         }
1783
1784         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
1785         memcpy(wk->probe_auth.ssid, ssid + 2, ssid[1]);
1786         wk->probe_auth.ssid_len = ssid[1];
1787
1788         wk->probe_auth.algorithm = auth_alg;
1789         wk->probe_auth.privacy = req->bss->capability & WLAN_CAPABILITY_PRIVACY;
1790
1791         wk->type = IEEE80211_WORK_DIRECT_PROBE;
1792         wk->chan = req->bss->channel;
1793         wk->sdata = sdata;
1794         wk->done = ieee80211_probe_auth_done;
1795
1796         /*
1797          * XXX: if still associated need to tell AP that we're going
1798          *      to sleep and then change channel etc.
1799          *      For now switch channel here, later will be handled
1800          *      by submitting this as an off-channel work item.
1801          */
1802         sdata->local->oper_channel = req->bss->channel;
1803         ieee80211_hw_config(sdata->local, 0);
1804
1805         ieee80211_add_work(wk);
1806         return 0;
1807 }
1808
1809 static enum work_done_result ieee80211_assoc_done(struct ieee80211_work *wk,
1810                                                   struct sk_buff *skb)
1811 {
1812         struct ieee80211_mgmt *mgmt;
1813         u16 status;
1814
1815         if (!skb) {
1816                 cfg80211_send_assoc_timeout(wk->sdata->dev, wk->filter_ta);
1817                 return WORK_DONE_DESTROY;
1818         }
1819
1820         mgmt = (void *)skb->data;
1821         status = le16_to_cpu(mgmt->u.assoc_resp.status_code);
1822
1823         if (status == WLAN_STATUS_SUCCESS) {
1824                 mutex_lock(&wk->sdata->u.mgd.mtx);
1825                 if (!ieee80211_assoc_success(wk, mgmt, skb->len)) {
1826                         mutex_unlock(&wk->sdata->u.mgd.mtx);
1827                         /* oops -- internal error -- send timeout for now */
1828                         cfg80211_send_assoc_timeout(wk->sdata->dev,
1829                                                     wk->filter_ta);
1830                         return WORK_DONE_DESTROY;
1831                 }
1832                 mutex_unlock(&wk->sdata->u.mgd.mtx);
1833         }
1834
1835         cfg80211_send_rx_assoc(wk->sdata->dev, skb->data, skb->len);
1836         return WORK_DONE_DESTROY;
1837 }
1838
1839 int ieee80211_mgd_assoc(struct ieee80211_sub_if_data *sdata,
1840                         struct cfg80211_assoc_request *req)
1841 {
1842         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1843         struct ieee80211_work *wk;
1844         const u8 *ssid;
1845         int i;
1846
1847         mutex_lock(&ifmgd->mtx);
1848         if (ifmgd->associated) {
1849                 mutex_unlock(&ifmgd->mtx);
1850                 return -EALREADY;
1851         }
1852         mutex_unlock(&ifmgd->mtx);
1853
1854         wk = kzalloc(sizeof(*wk) + req->ie_len, GFP_KERNEL);
1855         if (!wk)
1856                 return -ENOMEM;
1857
1858         ifmgd->flags &= ~IEEE80211_STA_DISABLE_11N;
1859
1860         for (i = 0; i < req->crypto.n_ciphers_pairwise; i++)
1861                 if (req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP40 ||
1862                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_TKIP ||
1863                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP104)
1864                         ifmgd->flags |= IEEE80211_STA_DISABLE_11N;
1865
1866
1867         if (req->ie && req->ie_len) {
1868                 memcpy(wk->ie, req->ie, req->ie_len);
1869                 wk->ie_len = req->ie_len;
1870         } else
1871                 wk->ie_len = 0;
1872
1873         wk->assoc.bss = (void *)req->bss;
1874
1875         memcpy(wk->filter_ta, req->bss->bssid, ETH_ALEN);
1876
1877         /* new association always uses requested smps mode */
1878         if (ifmgd->req_smps == IEEE80211_SMPS_AUTOMATIC) {
1879                 if (ifmgd->powersave)
1880                         ifmgd->ap_smps = IEEE80211_SMPS_DYNAMIC;
1881                 else
1882                         ifmgd->ap_smps = IEEE80211_SMPS_OFF;
1883         } else
1884                 ifmgd->ap_smps = ifmgd->req_smps;
1885
1886         wk->assoc.smps = ifmgd->ap_smps;
1887         wk->assoc.use_11n = !(ifmgd->flags & IEEE80211_STA_DISABLE_11N);
1888         wk->assoc.capability = req->bss->capability;
1889         wk->assoc.wmm_used = wk->assoc.bss->wmm_used;
1890         wk->assoc.supp_rates = wk->assoc.bss->supp_rates;
1891         wk->assoc.supp_rates_len = wk->assoc.bss->supp_rates_len;
1892         wk->assoc.ht_information_ie =
1893                 ieee80211_bss_get_ie(req->bss, WLAN_EID_HT_INFORMATION);
1894
1895         ssid = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
1896         memcpy(wk->assoc.ssid, ssid + 2, ssid[1]);
1897         wk->assoc.ssid_len = ssid[1];
1898
1899         if (req->prev_bssid)
1900                 memcpy(wk->assoc.prev_bssid, req->prev_bssid, ETH_ALEN);
1901
1902         wk->type = IEEE80211_WORK_ASSOC;
1903         wk->chan = req->bss->channel;
1904         wk->sdata = sdata;
1905         wk->done = ieee80211_assoc_done;
1906
1907         if (req->use_mfp) {
1908                 ifmgd->mfp = IEEE80211_MFP_REQUIRED;
1909                 ifmgd->flags |= IEEE80211_STA_MFP_ENABLED;
1910         } else {
1911                 ifmgd->mfp = IEEE80211_MFP_DISABLED;
1912                 ifmgd->flags &= ~IEEE80211_STA_MFP_ENABLED;
1913         }
1914
1915         if (req->crypto.control_port)
1916                 ifmgd->flags |= IEEE80211_STA_CONTROL_PORT;
1917         else
1918                 ifmgd->flags &= ~IEEE80211_STA_CONTROL_PORT;
1919
1920         sdata->local->oper_channel = req->bss->channel;
1921         ieee80211_hw_config(sdata->local, 0);
1922
1923         ieee80211_add_work(wk);
1924         return 0;
1925 }
1926
1927 int ieee80211_mgd_deauth(struct ieee80211_sub_if_data *sdata,
1928                          struct cfg80211_deauth_request *req,
1929                          void *cookie)
1930 {
1931         struct ieee80211_local *local = sdata->local;
1932         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1933         struct ieee80211_work *wk;
1934         const u8 *bssid = req->bss->bssid;
1935
1936         mutex_lock(&ifmgd->mtx);
1937
1938         if (ifmgd->associated && &ifmgd->associated->cbss == req->bss) {
1939                 bssid = req->bss->bssid;
1940                 ieee80211_set_disassoc(sdata);
1941                 mutex_unlock(&ifmgd->mtx);
1942         } else {
1943                 bool not_auth_yet = false;
1944
1945                 mutex_unlock(&ifmgd->mtx);
1946
1947                 mutex_lock(&local->work_mtx);
1948                 list_for_each_entry(wk, &local->work_list, list) {
1949                         if (wk->type != IEEE80211_WORK_DIRECT_PROBE)
1950                                 continue;
1951                         if (memcmp(req->bss->bssid, wk->filter_ta, ETH_ALEN))
1952                                 continue;
1953                         not_auth_yet = true;
1954                         list_del(&wk->list);
1955                         free_work(wk);
1956                         break;
1957                 }
1958                 mutex_unlock(&local->work_mtx);
1959
1960                 /*
1961                  * If somebody requests authentication and we haven't
1962                  * sent out an auth frame yet there's no need to send
1963                  * out a deauth frame either. If the state was PROBE,
1964                  * then this is the case. If it's AUTH we have sent a
1965                  * frame, and if it's IDLE we have completed the auth
1966                  * process already.
1967                  */
1968                 if (not_auth_yet) {
1969                         __cfg80211_auth_canceled(sdata->dev, bssid);
1970                         return 0;
1971                 }
1972         }
1973
1974         printk(KERN_DEBUG "%s: deauthenticating from %pM by local choice (reason=%d)\n",
1975                sdata->name, bssid, req->reason_code);
1976
1977         ieee80211_send_deauth_disassoc(sdata, bssid,
1978                         IEEE80211_STYPE_DEAUTH, req->reason_code,
1979                         cookie);
1980
1981         ieee80211_recalc_idle(sdata->local);
1982
1983         return 0;
1984 }
1985
1986 int ieee80211_mgd_disassoc(struct ieee80211_sub_if_data *sdata,
1987                            struct cfg80211_disassoc_request *req,
1988                            void *cookie)
1989 {
1990         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1991
1992         mutex_lock(&ifmgd->mtx);
1993
1994         /*
1995          * cfg80211 should catch this ... but it's racy since
1996          * we can receive a disassoc frame, process it, hand it
1997          * to cfg80211 while that's in a locked section already
1998          * trying to tell us that the user wants to disconnect.
1999          */
2000         if (&ifmgd->associated->cbss != req->bss) {
2001                 mutex_unlock(&ifmgd->mtx);
2002                 return -ENOLINK;
2003         }
2004
2005         printk(KERN_DEBUG "%s: disassociating from %pM by local choice (reason=%d)\n",
2006                sdata->name, req->bss->bssid, req->reason_code);
2007
2008         ieee80211_set_disassoc(sdata);
2009
2010         mutex_unlock(&ifmgd->mtx);
2011
2012         ieee80211_send_deauth_disassoc(sdata, req->bss->bssid,
2013                         IEEE80211_STYPE_DISASSOC, req->reason_code,
2014                         cookie);
2015
2016         ieee80211_recalc_idle(sdata->local);
2017
2018         return 0;
2019 }