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  * Copyright 2013-2014  Intel Mobile Communications GmbH
   9  * Copyright (C) 2015 - 2017 Intel Deutschland GmbH
  10  * Copyright (C) 2018        Intel Corporation
  11  *
  12  * This program is free software; you can redistribute it and/or modify
  13  * it under the terms of the GNU General Public License version 2 as
  14  * published by the Free Software Foundation.
  15  */
  16
  17 #include <linux/delay.h>
  18 #include <linux/if_ether.h>
  19 #include <linux/skbuff.h>
  20 #include <linux/if_arp.h>
  21 #include <linux/etherdevice.h>
  22 #include <linux/moduleparam.h>
  23 #include <linux/rtnetlink.h>
  24 #include <linux/crc32.h>
  25 #include <linux/slab.h>
  26 #include <linux/export.h>
  27 #include <net/mac80211.h>
  28 #include <asm/unaligned.h>
  29
  30 #include "ieee80211_i.h"
  31 #include "driver-ops.h"
  32 #include "rate.h"
  33 #include "led.h"
  34 #include "fils_aead.h"
  35
  36 #define IEEE80211_AUTH_TIMEOUT          (HZ / 5)
  37 #define IEEE80211_AUTH_TIMEOUT_LONG     (HZ / 2)
  38 #define IEEE80211_AUTH_TIMEOUT_SHORT    (HZ / 10)
  39 #define IEEE80211_AUTH_TIMEOUT_SAE      (HZ * 2)
  40 #define IEEE80211_AUTH_MAX_TRIES        3
  41 #define IEEE80211_AUTH_WAIT_ASSOC       (HZ * 5)
  42 #define IEEE80211_ASSOC_TIMEOUT         (HZ / 5)
  43 #define IEEE80211_ASSOC_TIMEOUT_LONG    (HZ / 2)
  44 #define IEEE80211_ASSOC_TIMEOUT_SHORT   (HZ / 10)
  45 #define IEEE80211_ASSOC_MAX_TRIES       3
  46
  47 static int max_nullfunc_tries = 2;
  48 module_param(max_nullfunc_tries, int, 0644);
  49 MODULE_PARM_DESC(max_nullfunc_tries,
  50                  "Maximum nullfunc tx tries before disconnecting (reason 4).");
  51
  52 static int max_probe_tries = 5;
  53 module_param(max_probe_tries, int, 0644);
  54 MODULE_PARM_DESC(max_probe_tries,
  55                  "Maximum probe tries before disconnecting (reason 4).");
  56
  57 /*
  58  * Beacon loss timeout is calculated as N frames times the
  59  * advertised beacon interval.  This may need to be somewhat
  60  * higher than what hardware might detect to account for
  61  * delays in the host processing frames. But since we also
  62  * probe on beacon miss before declaring the connection lost
  63  * default to what we want.
  64  */
  65 static int beacon_loss_count = 7;
  66 module_param(beacon_loss_count, int, 0644);
  67 MODULE_PARM_DESC(beacon_loss_count,
  68                  "Number of beacon intervals before we decide beacon was lost.");
  69
  70 /*
  71  * Time the connection can be idle before we probe
  72  * it to see if we can still talk to the AP.
  73  */
  74 #define IEEE80211_CONNECTION_IDLE_TIME  (30 * HZ)
  75 /*
  76  * Time we wait for a probe response after sending
  77  * a probe request because of beacon loss or for
  78  * checking the connection still works.
  79  */
  80 static int probe_wait_ms = 500;
  81 module_param(probe_wait_ms, int, 0644);
  82 MODULE_PARM_DESC(probe_wait_ms,
  83                  "Maximum time(ms) to wait for probe response"
  84                  " before disconnecting (reason 4).");
  85
  86 /*
  87  * How many Beacon frames need to have been used in average signal strength
  88  * before starting to indicate signal change events.
  89  */
  90 #define IEEE80211_SIGNAL_AVE_MIN_COUNT  4
  91
  92 /*
  93  * We can have multiple work items (and connection probing)
  94  * scheduling this timer, but we need to take care to only
  95  * reschedule it when it should fire _earlier_ than it was
  96  * asked for before, or if it's not pending right now. This
  97  * function ensures that. Note that it then is required to
  98  * run this function for all timeouts after the first one
  99  * has happened -- the work that runs from this timer will
 100  * do that.
 101  */
 102 static void run_again(struct ieee80211_sub_if_data *sdata,
 103                       unsigned long timeout)
 104 {
 105         sdata_assert_lock(sdata);
 106
 107         if (!timer_pending(&sdata->u.mgd.timer) ||
 108             time_before(timeout, sdata->u.mgd.timer.expires))
 109                 mod_timer(&sdata->u.mgd.timer, timeout);
 110 }
 111
 112 void ieee80211_sta_reset_beacon_monitor(struct ieee80211_sub_if_data *sdata)
 113 {
 114         if (sdata->vif.driver_flags & IEEE80211_VIF_BEACON_FILTER)
 115                 return;
 116
 117         if (ieee80211_hw_check(&sdata->local->hw, CONNECTION_MONITOR))
 118                 return;
 119
 120         mod_timer(&sdata->u.mgd.bcn_mon_timer,
 121                   round_jiffies_up(jiffies + sdata->u.mgd.beacon_timeout));
 122 }
 123
 124 void ieee80211_sta_reset_conn_monitor(struct ieee80211_sub_if_data *sdata)
 125 {
 126         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 127
 128         if (unlikely(!ifmgd->associated))
 129                 return;
 130
 131         if (ifmgd->probe_send_count)
 132                 ifmgd->probe_send_count = 0;
 133
 134         if (ieee80211_hw_check(&sdata->local->hw, CONNECTION_MONITOR))
 135                 return;
 136
 137         mod_timer(&ifmgd->conn_mon_timer,
 138                   round_jiffies_up(jiffies + IEEE80211_CONNECTION_IDLE_TIME));
 139 }
 140
 141 static int ecw2cw(int ecw)
 142 {
 143         return (1 << ecw) - 1;
 144 }
 145
 146 static u32
 147 ieee80211_determine_chantype(struct ieee80211_sub_if_data *sdata,
 148                              struct ieee80211_supported_band *sband,
 149                              struct ieee80211_channel *channel,
 150                              const struct ieee80211_ht_operation *ht_oper,
 151                              const struct ieee80211_vht_operation *vht_oper,
 152                              const struct ieee80211_he_operation *he_oper,
 153                              struct cfg80211_chan_def *chandef, bool tracking)
 154 {
 155         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 156         struct cfg80211_chan_def vht_chandef;
 157         struct ieee80211_sta_ht_cap sta_ht_cap;
 158         u32 ht_cfreq, ret;
 159
 160         memcpy(&sta_ht_cap, &sband->ht_cap, sizeof(sta_ht_cap));
 161         ieee80211_apply_htcap_overrides(sdata, &sta_ht_cap);
 162
 163         chandef->chan = channel;
 164         chandef->width = NL80211_CHAN_WIDTH_20_NOHT;
 165         chandef->center_freq1 = channel->center_freq;
 166         chandef->center_freq2 = 0;
 167
 168         if (!ht_oper || !sta_ht_cap.ht_supported) {
 169                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
 170                 goto out;
 171         }
 172
 173         chandef->width = NL80211_CHAN_WIDTH_20;
 174
 175         ht_cfreq = ieee80211_channel_to_frequency(ht_oper->primary_chan,
 176                                                   channel->band);
 177         /* check that channel matches the right operating channel */
 178         if (!tracking && channel->center_freq != ht_cfreq) {
 179                 /*
 180                  * It's possible that some APs are confused here;
 181                  * Netgear WNDR3700 sometimes reports 4 higher than
 182                  * the actual channel in association responses, but
 183                  * since we look at probe response/beacon data here
 184                  * it should be OK.
 185                  */
 186                 sdata_info(sdata,
 187                            "Wrong control channel: center-freq: %d ht-cfreq: %d ht->primary_chan: %d band: %d - Disabling HT\n",
 188                            channel->center_freq, ht_cfreq,
 189                            ht_oper->primary_chan, channel->band);
 190                 ret = IEEE80211_STA_DISABLE_HT | IEEE80211_STA_DISABLE_VHT;
 191                 goto out;
 192         }
 193
 194         /* check 40 MHz support, if we have it */
 195         if (sta_ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40) {
 196                 ieee80211_chandef_ht_oper(ht_oper, chandef);
 197         } else {
 198                 /* 40 MHz (and 80 MHz) must be supported for VHT */
 199                 ret = IEEE80211_STA_DISABLE_VHT;
 200                 /* also mark 40 MHz disabled */
 201                 ret |= IEEE80211_STA_DISABLE_40MHZ;
 202                 goto out;
 203         }
 204
 205         if (!vht_oper || !sband->vht_cap.vht_supported) {
 206                 ret = IEEE80211_STA_DISABLE_VHT;
 207                 goto out;
 208         }
 209
 210         vht_chandef = *chandef;
 211         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HE) && he_oper &&
 212             (le32_to_cpu(he_oper->he_oper_params) &
 213              IEEE80211_HE_OPERATION_VHT_OPER_INFO)) {
 214                 struct ieee80211_vht_operation he_oper_vht_cap;
 215
 216                 /*
 217                  * Set only first 3 bytes (other 2 aren't used in
 218                  * ieee80211_chandef_vht_oper() anyway)
 219                  */
 220                 memcpy(&he_oper_vht_cap, he_oper->optional, 3);
 221                 he_oper_vht_cap.basic_mcs_set = cpu_to_le16(0);
 222
 223                 if (!ieee80211_chandef_vht_oper(&he_oper_vht_cap,
 224                                                 &vht_chandef)) {
 225                         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HE))
 226                                 sdata_info(sdata,
 227                                            "HE AP VHT information is invalid, disable HE\n");
 228                         ret = IEEE80211_STA_DISABLE_HE;
 229                         goto out;
 230                 }
 231         } else if (!ieee80211_chandef_vht_oper(vht_oper, &vht_chandef)) {
 232                 if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT))
 233                         sdata_info(sdata,
 234                                    "AP VHT information is invalid, disable VHT\n");
 235                 ret = IEEE80211_STA_DISABLE_VHT;
 236                 goto out;
 237         }
 238
 239         if (!cfg80211_chandef_valid(&vht_chandef)) {
 240                 if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT))
 241                         sdata_info(sdata,
 242                                    "AP VHT information is invalid, disable VHT\n");
 243                 ret = IEEE80211_STA_DISABLE_VHT;
 244                 goto out;
 245         }
 246
 247         if (cfg80211_chandef_identical(chandef, &vht_chandef)) {
 248                 ret = 0;
 249                 goto out;
 250         }
 251
 252         if (!cfg80211_chandef_compatible(chandef, &vht_chandef)) {
 253                 if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT))
 254                         sdata_info(sdata,
 255                                    "AP VHT information doesn't match HT, disable VHT\n");
 256                 ret = IEEE80211_STA_DISABLE_VHT;
 257                 goto out;
 258         }
 259
 260         *chandef = vht_chandef;
 261
 262         ret = 0;
 263
 264 out:
 265         /*
 266          * When tracking the current AP, don't do any further checks if the
 267          * new chandef is identical to the one we're currently using for the
 268          * connection. This keeps us from playing ping-pong with regulatory,
 269          * without it the following can happen (for example):
 270          *  - connect to an AP with 80 MHz, world regdom allows 80 MHz
 271          *  - AP advertises regdom US
 272          *  - CRDA loads regdom US with 80 MHz prohibited (old database)
 273          *  - the code below detects an unsupported channel, downgrades, and
 274          *    we disconnect from the AP in the caller
 275          *  - disconnect causes CRDA to reload world regdomain and the game
 276          *    starts anew.
 277          * (see https://bugzilla.kernel.org/show_bug.cgi?id=70881)
 278          *
 279          * It seems possible that there are still scenarios with CSA or real
 280          * bandwidth changes where a this could happen, but those cases are
 281          * less common and wouldn't completely prevent using the AP.
 282          */
 283         if (tracking &&
 284             cfg80211_chandef_identical(chandef, &sdata->vif.bss_conf.chandef))
 285                 return ret;
 286
 287         /* don't print the message below for VHT mismatch if VHT is disabled */
 288         if (ret & IEEE80211_STA_DISABLE_VHT)
 289                 vht_chandef = *chandef;
 290
 291         /*
 292          * Ignore the DISABLED flag when we're already connected and only
 293          * tracking the APs beacon for bandwidth changes - otherwise we
 294          * might get disconnected here if we connect to an AP, update our
 295          * regulatory information based on the AP's country IE and the
 296          * information we have is wrong/outdated and disables the channel
 297          * that we're actually using for the connection to the AP.
 298          */
 299         while (!cfg80211_chandef_usable(sdata->local->hw.wiphy, chandef,
 300                                         tracking ? 0 :
 301                                                    IEEE80211_CHAN_DISABLED)) {
 302                 if (WARN_ON(chandef->width == NL80211_CHAN_WIDTH_20_NOHT)) {
 303                         ret = IEEE80211_STA_DISABLE_HT |
 304                               IEEE80211_STA_DISABLE_VHT;
 305                         break;
 306                 }
 307
 308                 ret |= ieee80211_chandef_downgrade(chandef);
 309         }
 310
 311         if (chandef->width != vht_chandef.width && !tracking)
 312                 sdata_info(sdata,
 313                            "capabilities/regulatory prevented using AP HT/VHT configuration, downgraded\n");
 314
 315         WARN_ON_ONCE(!cfg80211_chandef_valid(chandef));
 316         return ret;
 317 }
 318
 319 static int ieee80211_config_bw(struct ieee80211_sub_if_data *sdata,
 320                                struct sta_info *sta,
 321                                const struct ieee80211_ht_cap *ht_cap,
 322                                const struct ieee80211_ht_operation *ht_oper,
 323                                const struct ieee80211_vht_operation *vht_oper,
 324                                const struct ieee80211_he_operation *he_oper,
 325                                const u8 *bssid, u32 *changed)
 326 {
 327         struct ieee80211_local *local = sdata->local;
 328         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 329         struct ieee80211_channel *chan = sdata->vif.bss_conf.chandef.chan;
 330         struct ieee80211_supported_band *sband =
 331                 local->hw.wiphy->bands[chan->band];
 332         struct cfg80211_chan_def chandef;
 333         u16 ht_opmode;
 334         u32 flags;
 335         enum ieee80211_sta_rx_bandwidth new_sta_bw;
 336         int ret;
 337
 338         /* if HT was/is disabled, don't track any bandwidth changes */
 339         if (ifmgd->flags & IEEE80211_STA_DISABLE_HT || !ht_oper)
 340                 return 0;
 341
 342         /* don't check VHT if we associated as non-VHT station */
 343         if (ifmgd->flags & IEEE80211_STA_DISABLE_VHT)
 344                 vht_oper = NULL;
 345
 346         /* don't check HE if we associated as non-HE station */
 347         if (ifmgd->flags & IEEE80211_STA_DISABLE_HE ||
 348             !ieee80211_get_he_sta_cap(sband))
 349                 he_oper = NULL;
 350
 351         if (WARN_ON_ONCE(!sta))
 352                 return -EINVAL;
 353
 354         /*
 355          * if bss configuration changed store the new one -
 356          * this may be applicable even if channel is identical
 357          */
 358         ht_opmode = le16_to_cpu(ht_oper->operation_mode);
 359         if (sdata->vif.bss_conf.ht_operation_mode != ht_opmode) {
 360                 *changed |= BSS_CHANGED_HT;
 361                 sdata->vif.bss_conf.ht_operation_mode = ht_opmode;
 362         }
 363
 364         /* calculate new channel (type) based on HT/VHT/HE operation IEs */
 365         flags = ieee80211_determine_chantype(sdata, sband, chan,
 366                                              ht_oper, vht_oper, he_oper,
 367                                              &chandef, true);
 368
 369         /*
 370          * Downgrade the new channel if we associated with restricted
 371          * capabilities. For example, if we associated as a 20 MHz STA
 372          * to a 40 MHz AP (due to regulatory, capabilities or config
 373          * reasons) then switching to a 40 MHz channel now won't do us
 374          * any good -- we couldn't use it with the AP.
 375          */
 376         if (ifmgd->flags & IEEE80211_STA_DISABLE_80P80MHZ &&
 377             chandef.width == NL80211_CHAN_WIDTH_80P80)
 378                 flags |= ieee80211_chandef_downgrade(&chandef);
 379         if (ifmgd->flags & IEEE80211_STA_DISABLE_160MHZ &&
 380             chandef.width == NL80211_CHAN_WIDTH_160)
 381                 flags |= ieee80211_chandef_downgrade(&chandef);
 382         if (ifmgd->flags & IEEE80211_STA_DISABLE_40MHZ &&
 383             chandef.width > NL80211_CHAN_WIDTH_20)
 384                 flags |= ieee80211_chandef_downgrade(&chandef);
 385
 386         if (cfg80211_chandef_identical(&chandef, &sdata->vif.bss_conf.chandef))
 387                 return 0;
 388
 389         sdata_info(sdata,
 390                    "AP %pM changed bandwidth, new config is %d MHz, width %d (%d/%d MHz)\n",
 391                    ifmgd->bssid, chandef.chan->center_freq, chandef.width,
 392                    chandef.center_freq1, chandef.center_freq2);
 393
 394         if (flags != (ifmgd->flags & (IEEE80211_STA_DISABLE_HT |
 395                                       IEEE80211_STA_DISABLE_VHT |
 396                                       IEEE80211_STA_DISABLE_40MHZ |
 397                                       IEEE80211_STA_DISABLE_80P80MHZ |
 398                                       IEEE80211_STA_DISABLE_160MHZ)) ||
 399             !cfg80211_chandef_valid(&chandef)) {
 400                 sdata_info(sdata,
 401                            "AP %pM changed bandwidth in a way we can't support - disconnect\n",
 402                            ifmgd->bssid);
 403                 return -EINVAL;
 404         }
 405
 406         switch (chandef.width) {
 407         case NL80211_CHAN_WIDTH_20_NOHT:
 408         case NL80211_CHAN_WIDTH_20:
 409                 new_sta_bw = IEEE80211_STA_RX_BW_20;
 410                 break;
 411         case NL80211_CHAN_WIDTH_40:
 412                 new_sta_bw = IEEE80211_STA_RX_BW_40;
 413                 break;
 414         case NL80211_CHAN_WIDTH_80:
 415                 new_sta_bw = IEEE80211_STA_RX_BW_80;
 416                 break;
 417         case NL80211_CHAN_WIDTH_80P80:
 418         case NL80211_CHAN_WIDTH_160:
 419                 new_sta_bw = IEEE80211_STA_RX_BW_160;
 420                 break;
 421         default:
 422                 return -EINVAL;
 423         }
 424
 425         if (new_sta_bw > sta->cur_max_bandwidth)
 426                 new_sta_bw = sta->cur_max_bandwidth;
 427
 428         if (new_sta_bw < sta->sta.bandwidth) {
 429                 sta->sta.bandwidth = new_sta_bw;
 430                 rate_control_rate_update(local, sband, sta,
 431                                          IEEE80211_RC_BW_CHANGED);
 432         }
 433
 434         ret = ieee80211_vif_change_bandwidth(sdata, &chandef, changed);
 435         if (ret) {
 436                 sdata_info(sdata,
 437                            "AP %pM changed bandwidth to incompatible one - disconnect\n",
 438                            ifmgd->bssid);
 439                 return ret;
 440         }
 441
 442         if (new_sta_bw > sta->sta.bandwidth) {
 443                 sta->sta.bandwidth = new_sta_bw;
 444                 rate_control_rate_update(local, sband, sta,
 445                                          IEEE80211_RC_BW_CHANGED);
 446         }
 447
 448         return 0;
 449 }
 450
 451 /* frame sending functions */
 452
 453 static void ieee80211_add_ht_ie(struct ieee80211_sub_if_data *sdata,
 454                                 struct sk_buff *skb, u8 ap_ht_param,
 455                                 struct ieee80211_supported_band *sband,
 456                                 struct ieee80211_channel *channel,
 457                                 enum ieee80211_smps_mode smps)
 458 {
 459         u8 *pos;
 460         u32 flags = channel->flags;
 461         u16 cap;
 462         struct ieee80211_sta_ht_cap ht_cap;
 463
 464         BUILD_BUG_ON(sizeof(ht_cap) != sizeof(sband->ht_cap));
 465
 466         memcpy(&ht_cap, &sband->ht_cap, sizeof(ht_cap));
 467         ieee80211_apply_htcap_overrides(sdata, &ht_cap);
 468
 469         /* determine capability flags */
 470         cap = ht_cap.cap;
 471
 472         switch (ap_ht_param & IEEE80211_HT_PARAM_CHA_SEC_OFFSET) {
 473         case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
 474                 if (flags & IEEE80211_CHAN_NO_HT40PLUS) {
 475                         cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
 476                         cap &= ~IEEE80211_HT_CAP_SGI_40;
 477                 }
 478                 break;
 479         case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
 480                 if (flags & IEEE80211_CHAN_NO_HT40MINUS) {
 481                         cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
 482                         cap &= ~IEEE80211_HT_CAP_SGI_40;
 483                 }
 484                 break;
 485         }
 486
 487         /*
 488          * If 40 MHz was disabled associate as though we weren't
 489          * capable of 40 MHz -- some broken APs will never fall
 490          * back to trying to transmit in 20 MHz.
 491          */
 492         if (sdata->u.mgd.flags & IEEE80211_STA_DISABLE_40MHZ) {
 493                 cap &= ~IEEE80211_HT_CAP_SUP_WIDTH_20_40;
 494                 cap &= ~IEEE80211_HT_CAP_SGI_40;
 495         }
 496
 497         /* set SM PS mode properly */
 498         cap &= ~IEEE80211_HT_CAP_SM_PS;
 499         switch (smps) {
 500         case IEEE80211_SMPS_AUTOMATIC:
 501         case IEEE80211_SMPS_NUM_MODES:
 502                 WARN_ON(1);
 503                 /* fall through */
 504         case IEEE80211_SMPS_OFF:
 505                 cap |= WLAN_HT_CAP_SM_PS_DISABLED <<
 506                         IEEE80211_HT_CAP_SM_PS_SHIFT;
 507                 break;
 508         case IEEE80211_SMPS_STATIC:
 509                 cap |= WLAN_HT_CAP_SM_PS_STATIC <<
 510                         IEEE80211_HT_CAP_SM_PS_SHIFT;
 511                 break;
 512         case IEEE80211_SMPS_DYNAMIC:
 513                 cap |= WLAN_HT_CAP_SM_PS_DYNAMIC <<
 514                         IEEE80211_HT_CAP_SM_PS_SHIFT;
 515                 break;
 516         }
 517
 518         /* reserve and fill IE */
 519         pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
 520         ieee80211_ie_build_ht_cap(pos, &ht_cap, cap);
 521 }
 522
 523 /* This function determines vht capability flags for the association
 524  * and builds the IE.
 525  * Note - the function may set the owner of the MU-MIMO capability
 526  */
 527 static void ieee80211_add_vht_ie(struct ieee80211_sub_if_data *sdata,
 528                                  struct sk_buff *skb,
 529                                  struct ieee80211_supported_band *sband,
 530                                  struct ieee80211_vht_cap *ap_vht_cap)
 531 {
 532         struct ieee80211_local *local = sdata->local;
 533         u8 *pos;
 534         u32 cap;
 535         struct ieee80211_sta_vht_cap vht_cap;
 536         u32 mask, ap_bf_sts, our_bf_sts;
 537
 538         BUILD_BUG_ON(sizeof(vht_cap) != sizeof(sband->vht_cap));
 539
 540         memcpy(&vht_cap, &sband->vht_cap, sizeof(vht_cap));
 541         ieee80211_apply_vhtcap_overrides(sdata, &vht_cap);
 542
 543         /* determine capability flags */
 544         cap = vht_cap.cap;
 545
 546         if (sdata->u.mgd.flags & IEEE80211_STA_DISABLE_80P80MHZ) {
 547                 u32 bw = cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 548
 549                 cap &= ~IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 550                 if (bw == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
 551                     bw == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
 552                         cap |= IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
 553         }
 554
 555         if (sdata->u.mgd.flags & IEEE80211_STA_DISABLE_160MHZ) {
 556                 cap &= ~IEEE80211_VHT_CAP_SHORT_GI_160;
 557                 cap &= ~IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 558         }
 559
 560         /*
 561          * Some APs apparently get confused if our capabilities are better
 562          * than theirs, so restrict what we advertise in the assoc request.
 563          */
 564         if (!(ap_vht_cap->vht_cap_info &
 565                         cpu_to_le32(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)))
 566                 cap &= ~(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
 567                          IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE);
 568         else if (!(ap_vht_cap->vht_cap_info &
 569                         cpu_to_le32(IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)))
 570                 cap &= ~IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
 571
 572         /*
 573          * If some other vif is using the MU-MIMO capablity we cannot associate
 574          * using MU-MIMO - this will lead to contradictions in the group-id
 575          * mechanism.
 576          * Ownership is defined since association request, in order to avoid
 577          * simultaneous associations with MU-MIMO.
 578          */
 579         if (cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE) {
 580                 bool disable_mu_mimo = false;
 581                 struct ieee80211_sub_if_data *other;
 582
 583                 list_for_each_entry_rcu(other, &local->interfaces, list) {
 584                         if (other->vif.mu_mimo_owner) {
 585                                 disable_mu_mimo = true;
 586                                 break;
 587                         }
 588                 }
 589                 if (disable_mu_mimo)
 590                         cap &= ~IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
 591                 else
 592                         sdata->vif.mu_mimo_owner = true;
 593         }
 594
 595         mask = IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK;
 596
 597         ap_bf_sts = le32_to_cpu(ap_vht_cap->vht_cap_info) & mask;
 598         our_bf_sts = cap & mask;
 599
 600         if (ap_bf_sts < our_bf_sts) {
 601                 cap &= ~mask;
 602                 cap |= ap_bf_sts;
 603         }
 604
 605         /* reserve and fill IE */
 606         pos = skb_put(skb, sizeof(struct ieee80211_vht_cap) + 2);
 607         ieee80211_ie_build_vht_cap(pos, &vht_cap, cap);
 608 }
 609
 610 /* This function determines HE capability flags for the association
 611  * and builds the IE.
 612  */
 613 static void ieee80211_add_he_ie(struct ieee80211_sub_if_data *sdata,
 614                                 struct sk_buff *skb,
 615                                 struct ieee80211_supported_band *sband)
 616 {
 617         u8 *pos;
 618         const struct ieee80211_sta_he_cap *he_cap = NULL;
 619         u8 he_cap_size;
 620
 621         he_cap = ieee80211_get_he_sta_cap(sband);
 622         if (!he_cap)
 623                 return;
 624
 625         /*
 626          * TODO: the 1 added is because this temporarily is under the EXTENSION
 627          * IE. Get rid of it when it moves.
 628          */
 629         he_cap_size =
 630                 2 + 1 + sizeof(he_cap->he_cap_elem) +
 631                 ieee80211_he_mcs_nss_size(&he_cap->he_cap_elem) +
 632                 ieee80211_he_ppe_size(he_cap->ppe_thres[0],
 633                                       he_cap->he_cap_elem.phy_cap_info);
 634         pos = skb_put(skb, he_cap_size);
 635         ieee80211_ie_build_he_cap(pos, he_cap, pos + he_cap_size);
 636 }
 637
 638 static void ieee80211_send_assoc(struct ieee80211_sub_if_data *sdata)
 639 {
 640         struct ieee80211_local *local = sdata->local;
 641         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 642         struct ieee80211_mgd_assoc_data *assoc_data = ifmgd->assoc_data;
 643         struct sk_buff *skb;
 644         struct ieee80211_mgmt *mgmt;
 645         u8 *pos, qos_info;
 646         size_t offset = 0, noffset;
 647         int i, count, rates_len, supp_rates_len, shift;
 648         u16 capab;
 649         struct ieee80211_supported_band *sband;
 650         struct ieee80211_chanctx_conf *chanctx_conf;
 651         struct ieee80211_channel *chan;
 652         u32 rates = 0;
 653
 654         sdata_assert_lock(sdata);
 655
 656         rcu_read_lock();
 657         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
 658         if (WARN_ON(!chanctx_conf)) {
 659                 rcu_read_unlock();
 660                 return;
 661         }
 662         chan = chanctx_conf->def.chan;
 663         rcu_read_unlock();
 664         sband = local->hw.wiphy->bands[chan->band];
 665         shift = ieee80211_vif_get_shift(&sdata->vif);
 666
 667         if (assoc_data->supp_rates_len) {
 668                 /*
 669                  * Get all rates supported by the device and the AP as
 670                  * some APs don't like getting a superset of their rates
 671                  * in the association request (e.g. D-Link DAP 1353 in
 672                  * b-only mode)...
 673                  */
 674                 rates_len = ieee80211_parse_bitrates(&chanctx_conf->def, sband,
 675                                                      assoc_data->supp_rates,
 676                                                      assoc_data->supp_rates_len,
 677                                                      &rates);
 678         } else {
 679                 /*
 680                  * In case AP not provide any supported rates information
 681                  * before association, we send information element(s) with
 682                  * all rates that we support.
 683                  */
 684                 rates_len = 0;
 685                 for (i = 0; i < sband->n_bitrates; i++) {
 686                         rates |= BIT(i);
 687                         rates_len++;
 688                 }
 689         }
 690
 691         skb = alloc_skb(local->hw.extra_tx_headroom +
 692                         sizeof(*mgmt) + /* bit too much but doesn't matter */
 693                         2 + assoc_data->ssid_len + /* SSID */
 694                         4 + rates_len + /* (extended) rates */
 695                         4 + /* power capability */
 696                         2 + 2 * sband->n_channels + /* supported channels */
 697                         2 + sizeof(struct ieee80211_ht_cap) + /* HT */
 698                         2 + sizeof(struct ieee80211_vht_cap) + /* VHT */
 699                         2 + 1 + sizeof(struct ieee80211_he_cap_elem) + /* HE */
 700                                 sizeof(struct ieee80211_he_mcs_nss_supp) +
 701                                 IEEE80211_HE_PPE_THRES_MAX_LEN +
 702                         assoc_data->ie_len + /* extra IEs */
 703                         (assoc_data->fils_kek_len ? 16 /* AES-SIV */ : 0) +
 704                         9, /* WMM */
 705                         GFP_KERNEL);
 706         if (!skb)
 707                 return;
 708
 709         skb_reserve(skb, local->hw.extra_tx_headroom);
 710
 711         capab = WLAN_CAPABILITY_ESS;
 712
 713         if (sband->band == NL80211_BAND_2GHZ) {
 714                 capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
 715                 capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
 716         }
 717
 718         if (assoc_data->capability & WLAN_CAPABILITY_PRIVACY)
 719                 capab |= WLAN_CAPABILITY_PRIVACY;
 720
 721         if ((assoc_data->capability & WLAN_CAPABILITY_SPECTRUM_MGMT) &&
 722             ieee80211_hw_check(&local->hw, SPECTRUM_MGMT))
 723                 capab |= WLAN_CAPABILITY_SPECTRUM_MGMT;
 724
 725         if (ifmgd->flags & IEEE80211_STA_ENABLE_RRM)
 726                 capab |= WLAN_CAPABILITY_RADIO_MEASURE;
 727
 728         mgmt = skb_put_zero(skb, 24);
 729         memcpy(mgmt->da, assoc_data->bss->bssid, ETH_ALEN);
 730         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
 731         memcpy(mgmt->bssid, assoc_data->bss->bssid, ETH_ALEN);
 732
 733         if (!is_zero_ether_addr(assoc_data->prev_bssid)) {
 734                 skb_put(skb, 10);
 735                 mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
 736                                                   IEEE80211_STYPE_REASSOC_REQ);
 737                 mgmt->u.reassoc_req.capab_info = cpu_to_le16(capab);
 738                 mgmt->u.reassoc_req.listen_interval =
 739                                 cpu_to_le16(local->hw.conf.listen_interval);
 740                 memcpy(mgmt->u.reassoc_req.current_ap, assoc_data->prev_bssid,
 741                        ETH_ALEN);
 742         } else {
 743                 skb_put(skb, 4);
 744                 mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
 745                                                   IEEE80211_STYPE_ASSOC_REQ);
 746                 mgmt->u.assoc_req.capab_info = cpu_to_le16(capab);
 747                 mgmt->u.assoc_req.listen_interval =
 748                                 cpu_to_le16(local->hw.conf.listen_interval);
 749         }
 750
 751         /* SSID */
 752         pos = skb_put(skb, 2 + assoc_data->ssid_len);
 753         *pos++ = WLAN_EID_SSID;
 754         *pos++ = assoc_data->ssid_len;
 755         memcpy(pos, assoc_data->ssid, assoc_data->ssid_len);
 756
 757         /* add all rates which were marked to be used above */
 758         supp_rates_len = rates_len;
 759         if (supp_rates_len > 8)
 760                 supp_rates_len = 8;
 761
 762         pos = skb_put(skb, supp_rates_len + 2);
 763         *pos++ = WLAN_EID_SUPP_RATES;
 764         *pos++ = supp_rates_len;
 765
 766         count = 0;
 767         for (i = 0; i < sband->n_bitrates; i++) {
 768                 if (BIT(i) & rates) {
 769                         int rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
 770                                                 5 * (1 << shift));
 771                         *pos++ = (u8) rate;
 772                         if (++count == 8)
 773                                 break;
 774                 }
 775         }
 776
 777         if (rates_len > count) {
 778                 pos = skb_put(skb, rates_len - count + 2);
 779                 *pos++ = WLAN_EID_EXT_SUPP_RATES;
 780                 *pos++ = rates_len - count;
 781
 782                 for (i++; i < sband->n_bitrates; i++) {
 783                         if (BIT(i) & rates) {
 784                                 int rate;
 785                                 rate = DIV_ROUND_UP(sband->bitrates[i].bitrate,
 786                                                     5 * (1 << shift));
 787                                 *pos++ = (u8) rate;
 788                         }
 789                 }
 790         }
 791
 792         if (capab & WLAN_CAPABILITY_SPECTRUM_MGMT ||
 793             capab & WLAN_CAPABILITY_RADIO_MEASURE) {
 794                 pos = skb_put(skb, 4);
 795                 *pos++ = WLAN_EID_PWR_CAPABILITY;
 796                 *pos++ = 2;
 797                 *pos++ = 0; /* min tx power */
 798                  /* max tx power */
 799                 *pos++ = ieee80211_chandef_max_power(&chanctx_conf->def);
 800         }
 801
 802         if (capab & WLAN_CAPABILITY_SPECTRUM_MGMT) {
 803                 /* TODO: get this in reg domain format */
 804                 pos = skb_put(skb, 2 * sband->n_channels + 2);
 805                 *pos++ = WLAN_EID_SUPPORTED_CHANNELS;
 806                 *pos++ = 2 * sband->n_channels;
 807                 for (i = 0; i < sband->n_channels; i++) {
 808                         *pos++ = ieee80211_frequency_to_channel(
 809                                         sband->channels[i].center_freq);
 810                         *pos++ = 1; /* one channel in the subband*/
 811                 }
 812         }
 813
 814         /* if present, add any custom IEs that go before HT */
 815         if (assoc_data->ie_len) {
 816                 static const u8 before_ht[] = {
 817                         WLAN_EID_SSID,
 818                         WLAN_EID_SUPP_RATES,
 819                         WLAN_EID_EXT_SUPP_RATES,
 820                         WLAN_EID_PWR_CAPABILITY,
 821                         WLAN_EID_SUPPORTED_CHANNELS,
 822                         WLAN_EID_RSN,
 823                         WLAN_EID_QOS_CAPA,
 824                         WLAN_EID_RRM_ENABLED_CAPABILITIES,
 825                         WLAN_EID_MOBILITY_DOMAIN,
 826                         WLAN_EID_FAST_BSS_TRANSITION,   /* reassoc only */
 827                         WLAN_EID_RIC_DATA,              /* reassoc only */
 828                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
 829                 };
 830                 static const u8 after_ric[] = {
 831                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
 832                         WLAN_EID_HT_CAPABILITY,
 833                         WLAN_EID_BSS_COEX_2040,
 834                         /* luckily this is almost always there */
 835                         WLAN_EID_EXT_CAPABILITY,
 836                         WLAN_EID_QOS_TRAFFIC_CAPA,
 837                         WLAN_EID_TIM_BCAST_REQ,
 838                         WLAN_EID_INTERWORKING,
 839                         /* 60 GHz (Multi-band, DMG, MMS) can't happen */
 840                         WLAN_EID_VHT_CAPABILITY,
 841                         WLAN_EID_OPMODE_NOTIF,
 842                 };
 843
 844                 noffset = ieee80211_ie_split_ric(assoc_data->ie,
 845                                                  assoc_data->ie_len,
 846                                                  before_ht,
 847                                                  ARRAY_SIZE(before_ht),
 848                                                  after_ric,
 849                                                  ARRAY_SIZE(after_ric),
 850                                                  offset);
 851                 skb_put_data(skb, assoc_data->ie + offset, noffset - offset);
 852                 offset = noffset;
 853         }
 854
 855         if (WARN_ON_ONCE((ifmgd->flags & IEEE80211_STA_DISABLE_HT) &&
 856                          !(ifmgd->flags & IEEE80211_STA_DISABLE_VHT)))
 857                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
 858
 859         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT))
 860                 ieee80211_add_ht_ie(sdata, skb, assoc_data->ap_ht_param,
 861                                     sband, chan, sdata->smps_mode);
 862
 863         /* if present, add any custom IEs that go before VHT */
 864         if (assoc_data->ie_len) {
 865                 static const u8 before_vht[] = {
 866                         /*
 867                          * no need to list the ones split off before HT
 868                          * or generated here
 869                          */
 870                         WLAN_EID_BSS_COEX_2040,
 871                         WLAN_EID_EXT_CAPABILITY,
 872                         WLAN_EID_QOS_TRAFFIC_CAPA,
 873                         WLAN_EID_TIM_BCAST_REQ,
 874                         WLAN_EID_INTERWORKING,
 875                         /* 60 GHz (Multi-band, DMG, MMS) can't happen */
 876                 };
 877
 878                 /* RIC already taken above, so no need to handle here anymore */
 879                 noffset = ieee80211_ie_split(assoc_data->ie, assoc_data->ie_len,
 880                                              before_vht, ARRAY_SIZE(before_vht),
 881                                              offset);
 882                 skb_put_data(skb, assoc_data->ie + offset, noffset - offset);
 883                 offset = noffset;
 884         }
 885
 886         /* if present, add any custom IEs that go before HE */
 887         if (assoc_data->ie_len) {
 888                 static const u8 before_he[] = {
 889                         /*
 890                          * no need to list the ones split off before VHT
 891                          * or generated here
 892                          */
 893                         WLAN_EID_OPMODE_NOTIF,
 894                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FUTURE_CHAN_GUIDANCE,
 895                         /* 11ai elements */
 896                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_SESSION,
 897                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_PUBLIC_KEY,
 898                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_KEY_CONFIRM,
 899                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_HLP_CONTAINER,
 900                         WLAN_EID_EXTENSION, WLAN_EID_EXT_FILS_IP_ADDR_ASSIGN,
 901                         /* TODO: add 11ah/11aj/11ak elements */
 902                 };
 903
 904                 /* RIC already taken above, so no need to handle here anymore */
 905                 noffset = ieee80211_ie_split(assoc_data->ie, assoc_data->ie_len,
 906                                              before_he, ARRAY_SIZE(before_he),
 907                                              offset);
 908                 pos = skb_put(skb, noffset - offset);
 909                 memcpy(pos, assoc_data->ie + offset, noffset - offset);
 910                 offset = noffset;
 911         }
 912
 913         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT))
 914                 ieee80211_add_vht_ie(sdata, skb, sband,
 915                                      &assoc_data->ap_vht_cap);
 916
 917         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HE))
 918                 ieee80211_add_he_ie(sdata, skb, sband);
 919
 920         /* if present, add any custom non-vendor IEs that go after HE */
 921         if (assoc_data->ie_len) {
 922                 noffset = ieee80211_ie_split_vendor(assoc_data->ie,
 923                                                     assoc_data->ie_len,
 924                                                     offset);
 925                 skb_put_data(skb, assoc_data->ie + offset, noffset - offset);
 926                 offset = noffset;
 927         }
 928
 929         if (assoc_data->wmm) {
 930                 if (assoc_data->uapsd) {
 931                         qos_info = ifmgd->uapsd_queues;
 932                         qos_info |= (ifmgd->uapsd_max_sp_len <<
 933                                      IEEE80211_WMM_IE_STA_QOSINFO_SP_SHIFT);
 934                 } else {
 935                         qos_info = 0;
 936                 }
 937
 938                 pos = ieee80211_add_wmm_info_ie(skb_put(skb, 9), qos_info);
 939         }
 940
 941         /* add any remaining custom (i.e. vendor specific here) IEs */
 942         if (assoc_data->ie_len) {
 943                 noffset = assoc_data->ie_len;
 944                 skb_put_data(skb, assoc_data->ie + offset, noffset - offset);
 945         }
 946
 947         if (assoc_data->fils_kek_len &&
 948             fils_encrypt_assoc_req(skb, assoc_data) < 0) {
 949                 dev_kfree_skb(skb);
 950                 return;
 951         }
 952
 953         drv_mgd_prepare_tx(local, sdata, 0);
 954
 955         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 956         if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
 957                 IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS |
 958                                                 IEEE80211_TX_INTFL_MLME_CONN_TX;
 959         ieee80211_tx_skb(sdata, skb);
 960 }
 961
 962 void ieee80211_send_pspoll(struct ieee80211_local *local,
 963                            struct ieee80211_sub_if_data *sdata)
 964 {
 965         struct ieee80211_pspoll *pspoll;
 966         struct sk_buff *skb;
 967
 968         skb = ieee80211_pspoll_get(&local->hw, &sdata->vif);
 969         if (!skb)
 970                 return;
 971
 972         pspoll = (struct ieee80211_pspoll *) skb->data;
 973         pspoll->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
 974
 975         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
 976         ieee80211_tx_skb(sdata, skb);
 977 }
 978
 979 void ieee80211_send_nullfunc(struct ieee80211_local *local,
 980                              struct ieee80211_sub_if_data *sdata,
 981                              bool powersave)
 982 {
 983         struct sk_buff *skb;
 984         struct ieee80211_hdr_3addr *nullfunc;
 985         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 986
 987         /* Don't send NDPs when STA is connected HE */
 988         if (sdata->vif.type == NL80211_IFTYPE_STATION &&
 989             !(ifmgd->flags & IEEE80211_STA_DISABLE_HE))
 990                 return;
 991
 992         skb = ieee80211_nullfunc_get(&local->hw, &sdata->vif,
 993                 !ieee80211_hw_check(&local->hw, DOESNT_SUPPORT_QOS_NDP));
 994         if (!skb)
 995                 return;
 996
 997         nullfunc = (struct ieee80211_hdr_3addr *) skb->data;
 998         if (powersave)
 999                 nullfunc->frame_control |= cpu_to_le16(IEEE80211_FCTL_PM);
1000
1001         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT |
1002                                         IEEE80211_TX_INTFL_OFFCHAN_TX_OK;
1003
1004         if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
1005                 IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
1006
1007         if (ifmgd->flags & IEEE80211_STA_CONNECTION_POLL)
1008                 IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_CTL_USE_MINRATE;
1009
1010         ieee80211_tx_skb(sdata, skb);
1011 }
1012
1013 static void ieee80211_send_4addr_nullfunc(struct ieee80211_local *local,
1014                                           struct ieee80211_sub_if_data *sdata)
1015 {
1016         struct sk_buff *skb;
1017         struct ieee80211_hdr *nullfunc;
1018         __le16 fc;
1019
1020         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
1021                 return;
1022
1023         /* Don't send NDPs when connected HE */
1024         if (!(sdata->u.mgd.flags & IEEE80211_STA_DISABLE_HE))
1025                 return;
1026
1027         skb = dev_alloc_skb(local->hw.extra_tx_headroom + 30);
1028         if (!skb)
1029                 return;
1030
1031         skb_reserve(skb, local->hw.extra_tx_headroom);
1032
1033         nullfunc = skb_put_zero(skb, 30);
1034         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_NULLFUNC |
1035                          IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS);
1036         nullfunc->frame_control = fc;
1037         memcpy(nullfunc->addr1, sdata->u.mgd.bssid, ETH_ALEN);
1038         memcpy(nullfunc->addr2, sdata->vif.addr, ETH_ALEN);
1039         memcpy(nullfunc->addr3, sdata->u.mgd.bssid, ETH_ALEN);
1040         memcpy(nullfunc->addr4, sdata->vif.addr, ETH_ALEN);
1041
1042         IEEE80211_SKB_CB(skb)->flags |= IEEE80211_TX_INTFL_DONT_ENCRYPT;
1043         ieee80211_tx_skb(sdata, skb);
1044 }
1045
1046 /* spectrum management related things */
1047 static void ieee80211_chswitch_work(struct work_struct *work)
1048 {
1049         struct ieee80211_sub_if_data *sdata =
1050                 container_of(work, struct ieee80211_sub_if_data, u.mgd.chswitch_work);
1051         struct ieee80211_local *local = sdata->local;
1052         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1053         int ret;
1054
1055         if (!ieee80211_sdata_running(sdata))
1056                 return;
1057
1058         sdata_lock(sdata);
1059         mutex_lock(&local->mtx);
1060         mutex_lock(&local->chanctx_mtx);
1061
1062         if (!ifmgd->associated)
1063                 goto out;
1064
1065         if (!sdata->vif.csa_active)
1066                 goto out;
1067
1068         /*
1069          * using reservation isn't immediate as it may be deferred until later
1070          * with multi-vif. once reservation is complete it will re-schedule the
1071          * work with no reserved_chanctx so verify chandef to check if it
1072          * completed successfully
1073          */
1074
1075         if (sdata->reserved_chanctx) {
1076                 struct ieee80211_supported_band *sband = NULL;
1077                 struct sta_info *mgd_sta = NULL;
1078                 enum ieee80211_sta_rx_bandwidth bw = IEEE80211_STA_RX_BW_20;
1079
1080                 /*
1081                  * with multi-vif csa driver may call ieee80211_csa_finish()
1082                  * many times while waiting for other interfaces to use their
1083                  * reservations
1084                  */
1085                 if (sdata->reserved_ready)
1086                         goto out;
1087
1088                 if (sdata->vif.bss_conf.chandef.width !=
1089                     sdata->csa_chandef.width) {
1090                         /*
1091                          * For managed interface, we need to also update the AP
1092                          * station bandwidth and align the rate scale algorithm
1093                          * on the bandwidth change. Here we only consider the
1094                          * bandwidth of the new channel definition (as channel
1095                          * switch flow does not have the full HT/VHT/HE
1096                          * information), assuming that if additional changes are
1097                          * required they would be done as part of the processing
1098                          * of the next beacon from the AP.
1099                          */
1100                         switch (sdata->csa_chandef.width) {
1101                         case NL80211_CHAN_WIDTH_20_NOHT:
1102                         case NL80211_CHAN_WIDTH_20:
1103                         default:
1104                                 bw = IEEE80211_STA_RX_BW_20;
1105                                 break;
1106                         case NL80211_CHAN_WIDTH_40:
1107                                 bw = IEEE80211_STA_RX_BW_40;
1108                                 break;
1109                         case NL80211_CHAN_WIDTH_80:
1110                                 bw = IEEE80211_STA_RX_BW_80;
1111                                 break;
1112                         case NL80211_CHAN_WIDTH_80P80:
1113                         case NL80211_CHAN_WIDTH_160:
1114                                 bw = IEEE80211_STA_RX_BW_160;
1115                                 break;
1116                         }
1117
1118                         mgd_sta = sta_info_get(sdata, ifmgd->bssid);
1119                         sband =
1120                                 local->hw.wiphy->bands[sdata->csa_chandef.chan->band];
1121                 }
1122
1123                 if (sdata->vif.bss_conf.chandef.width >
1124                     sdata->csa_chandef.width) {
1125                         mgd_sta->sta.bandwidth = bw;
1126                         rate_control_rate_update(local, sband, mgd_sta,
1127                                                  IEEE80211_RC_BW_CHANGED);
1128                 }
1129
1130                 ret = ieee80211_vif_use_reserved_context(sdata);
1131                 if (ret) {
1132                         sdata_info(sdata,
1133                                    "failed to use reserved channel context, disconnecting (err=%d)\n",
1134                                    ret);
1135                         ieee80211_queue_work(&sdata->local->hw,
1136                                              &ifmgd->csa_connection_drop_work);
1137                         goto out;
1138                 }
1139
1140                 if (sdata->vif.bss_conf.chandef.width <
1141                     sdata->csa_chandef.width) {
1142                         mgd_sta->sta.bandwidth = bw;
1143                         rate_control_rate_update(local, sband, mgd_sta,
1144                                                  IEEE80211_RC_BW_CHANGED);
1145                 }
1146
1147                 goto out;
1148         }
1149
1150         if (!cfg80211_chandef_identical(&sdata->vif.bss_conf.chandef,
1151                                         &sdata->csa_chandef)) {
1152                 sdata_info(sdata,
1153                            "failed to finalize channel switch, disconnecting\n");
1154                 ieee80211_queue_work(&sdata->local->hw,
1155                                      &ifmgd->csa_connection_drop_work);
1156                 goto out;
1157         }
1158
1159         ifmgd->csa_waiting_bcn = true;
1160
1161         ieee80211_sta_reset_beacon_monitor(sdata);
1162         ieee80211_sta_reset_conn_monitor(sdata);
1163
1164 out:
1165         mutex_unlock(&local->chanctx_mtx);
1166         mutex_unlock(&local->mtx);
1167         sdata_unlock(sdata);
1168 }
1169
1170 static void ieee80211_chswitch_post_beacon(struct ieee80211_sub_if_data *sdata)
1171 {
1172         struct ieee80211_local *local = sdata->local;
1173         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1174         int ret;
1175
1176         sdata_assert_lock(sdata);
1177
1178         WARN_ON(!sdata->vif.csa_active);
1179
1180         if (sdata->csa_block_tx) {
1181                 ieee80211_wake_vif_queues(local, sdata,
1182                                           IEEE80211_QUEUE_STOP_REASON_CSA);
1183                 sdata->csa_block_tx = false;
1184         }
1185
1186         sdata->vif.csa_active = false;
1187         ifmgd->csa_waiting_bcn = false;
1188         /*
1189          * If the CSA IE is still present on the beacon after the switch,
1190          * we need to consider it as a new CSA (possibly to self).
1191          */
1192         ifmgd->beacon_crc_valid = false;
1193
1194         ret = drv_post_channel_switch(sdata);
1195         if (ret) {
1196                 sdata_info(sdata,
1197                            "driver post channel switch failed, disconnecting\n");
1198                 ieee80211_queue_work(&local->hw,
1199                                      &ifmgd->csa_connection_drop_work);
1200                 return;
1201         }
1202
1203         cfg80211_ch_switch_notify(sdata->dev, &sdata->reserved_chandef);
1204 }
1205
1206 void ieee80211_chswitch_done(struct ieee80211_vif *vif, bool success)
1207 {
1208         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1209         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1210
1211         trace_api_chswitch_done(sdata, success);
1212         if (!success) {
1213                 sdata_info(sdata,
1214                            "driver channel switch failed, disconnecting\n");
1215                 ieee80211_queue_work(&sdata->local->hw,
1216                                      &ifmgd->csa_connection_drop_work);
1217         } else {
1218                 ieee80211_queue_work(&sdata->local->hw, &ifmgd->chswitch_work);
1219         }
1220 }
1221 EXPORT_SYMBOL(ieee80211_chswitch_done);
1222
1223 static void ieee80211_chswitch_timer(struct timer_list *t)
1224 {
1225         struct ieee80211_sub_if_data *sdata =
1226                 from_timer(sdata, t, u.mgd.chswitch_timer);
1227
1228         ieee80211_queue_work(&sdata->local->hw, &sdata->u.mgd.chswitch_work);
1229 }
1230
1231 static void
1232 ieee80211_sta_process_chanswitch(struct ieee80211_sub_if_data *sdata,
1233                                  u64 timestamp, u32 device_timestamp,
1234                                  struct ieee802_11_elems *elems,
1235                                  bool beacon)
1236 {
1237         struct ieee80211_local *local = sdata->local;
1238         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1239         struct cfg80211_bss *cbss = ifmgd->associated;
1240         struct ieee80211_chanctx_conf *conf;
1241         struct ieee80211_chanctx *chanctx;
1242         enum nl80211_band current_band;
1243         struct ieee80211_csa_ie csa_ie;
1244         struct ieee80211_channel_switch ch_switch;
1245         int res;
1246
1247         sdata_assert_lock(sdata);
1248
1249         if (!cbss)
1250                 return;
1251
1252         if (local->scanning)
1253                 return;
1254
1255         /* disregard subsequent announcements if we are already processing */
1256         if (sdata->vif.csa_active)
1257                 return;
1258
1259         current_band = cbss->channel->band;
1260         res = ieee80211_parse_ch_switch_ie(sdata, elems, current_band,
1261                                            ifmgd->flags,
1262                                            ifmgd->associated->bssid, &csa_ie);
1263         if (res < 0)
1264                 ieee80211_queue_work(&local->hw,
1265                                      &ifmgd->csa_connection_drop_work);
1266         if (res)
1267                 return;
1268
1269         if (!cfg80211_chandef_usable(local->hw.wiphy, &csa_ie.chandef,
1270                                      IEEE80211_CHAN_DISABLED)) {
1271                 sdata_info(sdata,
1272                            "AP %pM switches to unsupported channel (%d MHz, width:%d, CF1/2: %d/%d MHz), disconnecting\n",
1273                            ifmgd->associated->bssid,
1274                            csa_ie.chandef.chan->center_freq,
1275                            csa_ie.chandef.width, csa_ie.chandef.center_freq1,
1276                            csa_ie.chandef.center_freq2);
1277                 ieee80211_queue_work(&local->hw,
1278                                      &ifmgd->csa_connection_drop_work);
1279                 return;
1280         }
1281
1282         if (cfg80211_chandef_identical(&csa_ie.chandef,
1283                                        &sdata->vif.bss_conf.chandef)) {
1284                 if (ifmgd->csa_ignored_same_chan)
1285                         return;
1286                 sdata_info(sdata,
1287                            "AP %pM tries to chanswitch to same channel, ignore\n",
1288                            ifmgd->associated->bssid);
1289                 ifmgd->csa_ignored_same_chan = true;
1290                 return;
1291         }
1292
1293         /*
1294          * Drop all TDLS peers - either we disconnect or move to a different
1295          * channel from this point on. There's no telling what our peer will do.
1296          * The TDLS WIDER_BW scenario is also problematic, as peers might now
1297          * have an incompatible wider chandef.
1298          */
1299         ieee80211_teardown_tdls_peers(sdata);
1300
1301         mutex_lock(&local->mtx);
1302         mutex_lock(&local->chanctx_mtx);
1303         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1304                                          lockdep_is_held(&local->chanctx_mtx));
1305         if (!conf) {
1306                 sdata_info(sdata,
1307                            "no channel context assigned to vif?, disconnecting\n");
1308                 goto drop_connection;
1309         }
1310
1311         chanctx = container_of(conf, struct ieee80211_chanctx, conf);
1312
1313         if (local->use_chanctx &&
1314             !ieee80211_hw_check(&local->hw, CHANCTX_STA_CSA)) {
1315                 sdata_info(sdata,
1316                            "driver doesn't support chan-switch with channel contexts\n");
1317                 goto drop_connection;
1318         }
1319
1320         ch_switch.timestamp = timestamp;
1321         ch_switch.device_timestamp = device_timestamp;
1322         ch_switch.block_tx = csa_ie.mode;
1323         ch_switch.chandef = csa_ie.chandef;
1324         ch_switch.count = csa_ie.count;
1325
1326         if (drv_pre_channel_switch(sdata, &ch_switch)) {
1327                 sdata_info(sdata,
1328                            "preparing for channel switch failed, disconnecting\n");
1329                 goto drop_connection;
1330         }
1331
1332         res = ieee80211_vif_reserve_chanctx(sdata, &csa_ie.chandef,
1333                                             chanctx->mode, false);
1334         if (res) {
1335                 sdata_info(sdata,
1336                            "failed to reserve channel context for channel switch, disconnecting (err=%d)\n",
1337                            res);
1338                 goto drop_connection;
1339         }
1340         mutex_unlock(&local->chanctx_mtx);
1341
1342         sdata->vif.csa_active = true;
1343         sdata->csa_chandef = csa_ie.chandef;
1344         sdata->csa_block_tx = csa_ie.mode;
1345         ifmgd->csa_ignored_same_chan = false;
1346
1347         if (sdata->csa_block_tx)
1348                 ieee80211_stop_vif_queues(local, sdata,
1349                                           IEEE80211_QUEUE_STOP_REASON_CSA);
1350         mutex_unlock(&local->mtx);
1351
1352         cfg80211_ch_switch_started_notify(sdata->dev, &csa_ie.chandef,
1353                                           csa_ie.count);
1354
1355         if (local->ops->channel_switch) {
1356                 /* use driver's channel switch callback */
1357                 drv_channel_switch(local, sdata, &ch_switch);
1358                 return;
1359         }
1360
1361         /* channel switch handled in software */
1362         if (csa_ie.count <= 1)
1363                 ieee80211_queue_work(&local->hw, &ifmgd->chswitch_work);
1364         else
1365                 mod_timer(&ifmgd->chswitch_timer,
1366                           TU_TO_EXP_TIME((csa_ie.count - 1) *
1367                                          cbss->beacon_interval));
1368         return;
1369  drop_connection:
1370         /*
1371          * This is just so that the disconnect flow will know that
1372          * we were trying to switch channel and failed. In case the
1373          * mode is 1 (we are not allowed to Tx), we will know not to
1374          * send a deauthentication frame. Those two fields will be
1375          * reset when the disconnection worker runs.
1376          */
1377         sdata->vif.csa_active = true;
1378         sdata->csa_block_tx = csa_ie.mode;
1379
1380         ieee80211_queue_work(&local->hw, &ifmgd->csa_connection_drop_work);
1381         mutex_unlock(&local->chanctx_mtx);
1382         mutex_unlock(&local->mtx);
1383 }
1384
1385 static bool
1386 ieee80211_find_80211h_pwr_constr(struct ieee80211_sub_if_data *sdata,
1387                                  struct ieee80211_channel *channel,
1388                                  const u8 *country_ie, u8 country_ie_len,
1389                                  const u8 *pwr_constr_elem,
1390                                  int *chan_pwr, int *pwr_reduction)
1391 {
1392         struct ieee80211_country_ie_triplet *triplet;
1393         int chan = ieee80211_frequency_to_channel(channel->center_freq);
1394         int i, chan_increment;
1395         bool have_chan_pwr = false;
1396
1397         /* Invalid IE */
1398         if (country_ie_len % 2 || country_ie_len < IEEE80211_COUNTRY_IE_MIN_LEN)
1399                 return false;
1400
1401         triplet = (void *)(country_ie + 3);
1402         country_ie_len -= 3;
1403
1404         switch (channel->band) {
1405         default:
1406                 WARN_ON_ONCE(1);
1407                 /* fall through */
1408         case NL80211_BAND_2GHZ:
1409         case NL80211_BAND_60GHZ:
1410                 chan_increment = 1;
1411                 break;
1412         case NL80211_BAND_5GHZ:
1413                 chan_increment = 4;
1414                 break;
1415         }
1416
1417         /* find channel */
1418         while (country_ie_len >= 3) {
1419                 u8 first_channel = triplet->chans.first_channel;
1420
1421                 if (first_channel >= IEEE80211_COUNTRY_EXTENSION_ID)
1422                         goto next;
1423
1424                 for (i = 0; i < triplet->chans.num_channels; i++) {
1425                         if (first_channel + i * chan_increment == chan) {
1426                                 have_chan_pwr = true;
1427                                 *chan_pwr = triplet->chans.max_power;
1428                                 break;
1429                         }
1430                 }
1431                 if (have_chan_pwr)
1432                         break;
1433
1434  next:
1435                 triplet++;
1436                 country_ie_len -= 3;
1437         }
1438
1439         if (have_chan_pwr && pwr_constr_elem)
1440                 *pwr_reduction = *pwr_constr_elem;
1441         else
1442                 *pwr_reduction = 0;
1443
1444         return have_chan_pwr;
1445 }
1446
1447 static void ieee80211_find_cisco_dtpc(struct ieee80211_sub_if_data *sdata,
1448                                       struct ieee80211_channel *channel,
1449                                       const u8 *cisco_dtpc_ie,
1450                                       int *pwr_level)
1451 {
1452         /* From practical testing, the first data byte of the DTPC element
1453          * seems to contain the requested dBm level, and the CLI on Cisco
1454          * APs clearly state the range is -127 to 127 dBm, which indicates
1455          * a signed byte, although it seemingly never actually goes negative.
1456          * The other byte seems to always be zero.
1457          */
1458         *pwr_level = (__s8)cisco_dtpc_ie[4];
1459 }
1460
1461 static u32 ieee80211_handle_pwr_constr(struct ieee80211_sub_if_data *sdata,
1462                                        struct ieee80211_channel *channel,
1463                                        struct ieee80211_mgmt *mgmt,
1464                                        const u8 *country_ie, u8 country_ie_len,
1465                                        const u8 *pwr_constr_ie,
1466                                        const u8 *cisco_dtpc_ie)
1467 {
1468         bool has_80211h_pwr = false, has_cisco_pwr = false;
1469         int chan_pwr = 0, pwr_reduction_80211h = 0;
1470         int pwr_level_cisco, pwr_level_80211h;
1471         int new_ap_level;
1472         __le16 capab = mgmt->u.probe_resp.capab_info;
1473
1474         if (country_ie &&
1475             (capab & cpu_to_le16(WLAN_CAPABILITY_SPECTRUM_MGMT) ||
1476              capab & cpu_to_le16(WLAN_CAPABILITY_RADIO_MEASURE))) {
1477                 has_80211h_pwr = ieee80211_find_80211h_pwr_constr(
1478                         sdata, channel, country_ie, country_ie_len,
1479                         pwr_constr_ie, &chan_pwr, &pwr_reduction_80211h);
1480                 pwr_level_80211h =
1481                         max_t(int, 0, chan_pwr - pwr_reduction_80211h);
1482         }
1483
1484         if (cisco_dtpc_ie) {
1485                 ieee80211_find_cisco_dtpc(
1486                         sdata, channel, cisco_dtpc_ie, &pwr_level_cisco);
1487                 has_cisco_pwr = true;
1488         }
1489
1490         if (!has_80211h_pwr && !has_cisco_pwr)
1491                 return 0;
1492
1493         /* If we have both 802.11h and Cisco DTPC, apply both limits
1494          * by picking the smallest of the two power levels advertised.
1495          */
1496         if (has_80211h_pwr &&
1497             (!has_cisco_pwr || pwr_level_80211h <= pwr_level_cisco)) {
1498                 new_ap_level = pwr_level_80211h;
1499
1500                 if (sdata->ap_power_level == new_ap_level)
1501                         return 0;
1502
1503                 sdata_dbg(sdata,
1504                           "Limiting TX power to %d (%d - %d) dBm as advertised by %pM\n",
1505                           pwr_level_80211h, chan_pwr, pwr_reduction_80211h,
1506                           sdata->u.mgd.bssid);
1507         } else {  /* has_cisco_pwr is always true here. */
1508                 new_ap_level = pwr_level_cisco;
1509
1510                 if (sdata->ap_power_level == new_ap_level)
1511                         return 0;
1512
1513                 sdata_dbg(sdata,
1514                           "Limiting TX power to %d dBm as advertised by %pM\n",
1515                           pwr_level_cisco, sdata->u.mgd.bssid);
1516         }
1517
1518         sdata->ap_power_level = new_ap_level;
1519         if (__ieee80211_recalc_txpower(sdata))
1520                 return BSS_CHANGED_TXPOWER;
1521         return 0;
1522 }
1523
1524 /* powersave */
1525 static void ieee80211_enable_ps(struct ieee80211_local *local,
1526                                 struct ieee80211_sub_if_data *sdata)
1527 {
1528         struct ieee80211_conf *conf = &local->hw.conf;
1529
1530         /*
1531          * If we are scanning right now then the parameters will
1532          * take effect when scan finishes.
1533          */
1534         if (local->scanning)
1535                 return;
1536
1537         if (conf->dynamic_ps_timeout > 0 &&
1538             !ieee80211_hw_check(&local->hw, SUPPORTS_DYNAMIC_PS)) {
1539                 mod_timer(&local->dynamic_ps_timer, jiffies +
1540                           msecs_to_jiffies(conf->dynamic_ps_timeout));
1541         } else {
1542                 if (ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK))
1543                         ieee80211_send_nullfunc(local, sdata, true);
1544
1545                 if (ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK) &&
1546                     ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
1547                         return;
1548
1549                 conf->flags |= IEEE80211_CONF_PS;
1550                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
1551         }
1552 }
1553
1554 static void ieee80211_change_ps(struct ieee80211_local *local)
1555 {
1556         struct ieee80211_conf *conf = &local->hw.conf;
1557
1558         if (local->ps_sdata) {
1559                 ieee80211_enable_ps(local, local->ps_sdata);
1560         } else if (conf->flags & IEEE80211_CONF_PS) {
1561                 conf->flags &= ~IEEE80211_CONF_PS;
1562                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
1563                 del_timer_sync(&local->dynamic_ps_timer);
1564                 cancel_work_sync(&local->dynamic_ps_enable_work);
1565         }
1566 }
1567
1568 static bool ieee80211_powersave_allowed(struct ieee80211_sub_if_data *sdata)
1569 {
1570         struct ieee80211_if_managed *mgd = &sdata->u.mgd;
1571         struct sta_info *sta = NULL;
1572         bool authorized = false;
1573
1574         if (!mgd->powersave)
1575                 return false;
1576
1577         if (mgd->broken_ap)
1578                 return false;
1579
1580         if (!mgd->associated)
1581                 return false;
1582
1583         if (mgd->flags & IEEE80211_STA_CONNECTION_POLL)
1584                 return false;
1585
1586         if (!mgd->have_beacon)
1587                 return false;
1588
1589         rcu_read_lock();
1590         sta = sta_info_get(sdata, mgd->bssid);
1591         if (sta)
1592                 authorized = test_sta_flag(sta, WLAN_STA_AUTHORIZED);
1593         rcu_read_unlock();
1594
1595         return authorized;
1596 }
1597
1598 /* need to hold RTNL or interface lock */
1599 void ieee80211_recalc_ps(struct ieee80211_local *local)
1600 {
1601         struct ieee80211_sub_if_data *sdata, *found = NULL;
1602         int count = 0;
1603         int timeout;
1604
1605         if (!ieee80211_hw_check(&local->hw, SUPPORTS_PS)) {
1606                 local->ps_sdata = NULL;
1607                 return;
1608         }
1609
1610         list_for_each_entry(sdata, &local->interfaces, list) {
1611                 if (!ieee80211_sdata_running(sdata))
1612                         continue;
1613                 if (sdata->vif.type == NL80211_IFTYPE_AP) {
1614                         /* If an AP vif is found, then disable PS
1615                          * by setting the count to zero thereby setting
1616                          * ps_sdata to NULL.
1617                          */
1618                         count = 0;
1619                         break;
1620                 }
1621                 if (sdata->vif.type != NL80211_IFTYPE_STATION)
1622                         continue;
1623                 found = sdata;
1624                 count++;
1625         }
1626
1627         if (count == 1 && ieee80211_powersave_allowed(found)) {
1628                 u8 dtimper = found->u.mgd.dtim_period;
1629
1630                 timeout = local->dynamic_ps_forced_timeout;
1631                 if (timeout < 0)
1632                         timeout = 100;
1633                 local->hw.conf.dynamic_ps_timeout = timeout;
1634
1635                 /* If the TIM IE is invalid, pretend the value is 1 */
1636                 if (!dtimper)
1637                         dtimper = 1;
1638
1639                 local->hw.conf.ps_dtim_period = dtimper;
1640                 local->ps_sdata = found;
1641         } else {
1642                 local->ps_sdata = NULL;
1643         }
1644
1645         ieee80211_change_ps(local);
1646 }
1647
1648 void ieee80211_recalc_ps_vif(struct ieee80211_sub_if_data *sdata)
1649 {
1650         bool ps_allowed = ieee80211_powersave_allowed(sdata);
1651
1652         if (sdata->vif.bss_conf.ps != ps_allowed) {
1653                 sdata->vif.bss_conf.ps = ps_allowed;
1654                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_PS);
1655         }
1656 }
1657
1658 void ieee80211_dynamic_ps_disable_work(struct work_struct *work)
1659 {
1660         struct ieee80211_local *local =
1661                 container_of(work, struct ieee80211_local,
1662                              dynamic_ps_disable_work);
1663
1664         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
1665                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
1666                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
1667         }
1668
1669         ieee80211_wake_queues_by_reason(&local->hw,
1670                                         IEEE80211_MAX_QUEUE_MAP,
1671                                         IEEE80211_QUEUE_STOP_REASON_PS,
1672                                         false);
1673 }
1674
1675 void ieee80211_dynamic_ps_enable_work(struct work_struct *work)
1676 {
1677         struct ieee80211_local *local =
1678                 container_of(work, struct ieee80211_local,
1679                              dynamic_ps_enable_work);
1680         struct ieee80211_sub_if_data *sdata = local->ps_sdata;
1681         struct ieee80211_if_managed *ifmgd;
1682         unsigned long flags;
1683         int q;
1684
1685         /* can only happen when PS was just disabled anyway */
1686         if (!sdata)
1687                 return;
1688
1689         ifmgd = &sdata->u.mgd;
1690
1691         if (local->hw.conf.flags & IEEE80211_CONF_PS)
1692                 return;
1693
1694         if (local->hw.conf.dynamic_ps_timeout > 0) {
1695                 /* don't enter PS if TX frames are pending */
1696                 if (drv_tx_frames_pending(local)) {
1697                         mod_timer(&local->dynamic_ps_timer, jiffies +
1698                                   msecs_to_jiffies(
1699                                   local->hw.conf.dynamic_ps_timeout));
1700                         return;
1701                 }
1702
1703                 /*
1704                  * transmission can be stopped by others which leads to
1705                  * dynamic_ps_timer expiry. Postpone the ps timer if it
1706                  * is not the actual idle state.
1707                  */
1708                 spin_lock_irqsave(&local->queue_stop_reason_lock, flags);
1709                 for (q = 0; q < local->hw.queues; q++) {
1710                         if (local->queue_stop_reasons[q]) {
1711                                 spin_unlock_irqrestore(&local->queue_stop_reason_lock,
1712                                                        flags);
1713                                 mod_timer(&local->dynamic_ps_timer, jiffies +
1714                                           msecs_to_jiffies(
1715                                           local->hw.conf.dynamic_ps_timeout));
1716                                 return;
1717                         }
1718                 }
1719                 spin_unlock_irqrestore(&local->queue_stop_reason_lock, flags);
1720         }
1721
1722         if (ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK) &&
1723             !(ifmgd->flags & IEEE80211_STA_NULLFUNC_ACKED)) {
1724                 if (drv_tx_frames_pending(local)) {
1725                         mod_timer(&local->dynamic_ps_timer, jiffies +
1726                                   msecs_to_jiffies(
1727                                   local->hw.conf.dynamic_ps_timeout));
1728                 } else {
1729                         ieee80211_send_nullfunc(local, sdata, true);
1730                         /* Flush to get the tx status of nullfunc frame */
1731                         ieee80211_flush_queues(local, sdata, false);
1732                 }
1733         }
1734
1735         if (!(ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS) &&
1736               ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK)) ||
1737             (ifmgd->flags & IEEE80211_STA_NULLFUNC_ACKED)) {
1738                 ifmgd->flags &= ~IEEE80211_STA_NULLFUNC_ACKED;
1739                 local->hw.conf.flags |= IEEE80211_CONF_PS;
1740                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
1741         }
1742 }
1743
1744 void ieee80211_dynamic_ps_timer(struct timer_list *t)
1745 {
1746         struct ieee80211_local *local = from_timer(local, t, dynamic_ps_timer);
1747
1748         ieee80211_queue_work(&local->hw, &local->dynamic_ps_enable_work);
1749 }
1750
1751 void ieee80211_dfs_cac_timer_work(struct work_struct *work)
1752 {
1753         struct delayed_work *delayed_work = to_delayed_work(work);
1754         struct ieee80211_sub_if_data *sdata =
1755                 container_of(delayed_work, struct ieee80211_sub_if_data,
1756                              dfs_cac_timer_work);
1757         struct cfg80211_chan_def chandef = sdata->vif.bss_conf.chandef;
1758
1759         mutex_lock(&sdata->local->mtx);
1760         if (sdata->wdev.cac_started) {
1761                 ieee80211_vif_release_channel(sdata);
1762                 cfg80211_cac_event(sdata->dev, &chandef,
1763                                    NL80211_RADAR_CAC_FINISHED,
1764                                    GFP_KERNEL);
1765         }
1766         mutex_unlock(&sdata->local->mtx);
1767 }
1768
1769 static bool
1770 __ieee80211_sta_handle_tspec_ac_params(struct ieee80211_sub_if_data *sdata)
1771 {
1772         struct ieee80211_local *local = sdata->local;
1773         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1774         bool ret = false;
1775         int ac;
1776
1777         if (local->hw.queues < IEEE80211_NUM_ACS)
1778                 return false;
1779
1780         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1781                 struct ieee80211_sta_tx_tspec *tx_tspec = &ifmgd->tx_tspec[ac];
1782                 int non_acm_ac;
1783                 unsigned long now = jiffies;
1784
1785                 if (tx_tspec->action == TX_TSPEC_ACTION_NONE &&
1786                     tx_tspec->admitted_time &&
1787                     time_after(now, tx_tspec->time_slice_start + HZ)) {
1788                         tx_tspec->consumed_tx_time = 0;
1789                         tx_tspec->time_slice_start = now;
1790
1791                         if (tx_tspec->downgraded)
1792                                 tx_tspec->action =
1793                                         TX_TSPEC_ACTION_STOP_DOWNGRADE;
1794                 }
1795
1796                 switch (tx_tspec->action) {
1797                 case TX_TSPEC_ACTION_STOP_DOWNGRADE:
1798                         /* take the original parameters */
1799                         if (drv_conf_tx(local, sdata, ac, &sdata->tx_conf[ac]))
1800                                 sdata_err(sdata,
1801                                           "failed to set TX queue parameters for queue %d\n",
1802                                           ac);
1803                         tx_tspec->action = TX_TSPEC_ACTION_NONE;
1804                         tx_tspec->downgraded = false;
1805                         ret = true;
1806                         break;
1807                 case TX_TSPEC_ACTION_DOWNGRADE:
1808                         if (time_after(now, tx_tspec->time_slice_start + HZ)) {
1809                                 tx_tspec->action = TX_TSPEC_ACTION_NONE;
1810                                 ret = true;
1811                                 break;
1812                         }
1813                         /* downgrade next lower non-ACM AC */
1814                         for (non_acm_ac = ac + 1;
1815                              non_acm_ac < IEEE80211_NUM_ACS;
1816                              non_acm_ac++)
1817                                 if (!(sdata->wmm_acm & BIT(7 - 2 * non_acm_ac)))
1818                                         break;
1819                         /* Usually the loop will result in using BK even if it
1820                          * requires admission control, but such a configuration
1821                          * makes no sense and we have to transmit somehow - the
1822                          * AC selection does the same thing.
1823                          * If we started out trying to downgrade from BK, then
1824                          * the extra condition here might be needed.
1825                          */
1826                         if (non_acm_ac >= IEEE80211_NUM_ACS)
1827                                 non_acm_ac = IEEE80211_AC_BK;
1828                         if (drv_conf_tx(local, sdata, ac,
1829                                         &sdata->tx_conf[non_acm_ac]))
1830                                 sdata_err(sdata,
1831                                           "failed to set TX queue parameters for queue %d\n",
1832                                           ac);
1833                         tx_tspec->action = TX_TSPEC_ACTION_NONE;
1834                         ret = true;
1835                         schedule_delayed_work(&ifmgd->tx_tspec_wk,
1836                                 tx_tspec->time_slice_start + HZ - now + 1);
1837                         break;
1838                 case TX_TSPEC_ACTION_NONE:
1839                         /* nothing now */
1840                         break;
1841                 }
1842         }
1843
1844         return ret;
1845 }
1846
1847 void ieee80211_sta_handle_tspec_ac_params(struct ieee80211_sub_if_data *sdata)
1848 {
1849         if (__ieee80211_sta_handle_tspec_ac_params(sdata))
1850                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_QOS);
1851 }
1852
1853 static void ieee80211_sta_handle_tspec_ac_params_wk(struct work_struct *work)
1854 {
1855         struct ieee80211_sub_if_data *sdata;
1856
1857         sdata = container_of(work, struct ieee80211_sub_if_data,
1858                              u.mgd.tx_tspec_wk.work);
1859         ieee80211_sta_handle_tspec_ac_params(sdata);
1860 }
1861
1862 /* MLME */
1863 static bool
1864 ieee80211_sta_wmm_params(struct ieee80211_local *local,
1865                          struct ieee80211_sub_if_data *sdata,
1866                          const u8 *wmm_param, size_t wmm_param_len,
1867                          const struct ieee80211_mu_edca_param_set *mu_edca)
1868 {
1869         struct ieee80211_tx_queue_params params[IEEE80211_NUM_ACS];
1870         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1871         size_t left;
1872         int count, ac;
1873         const u8 *pos;
1874         u8 uapsd_queues = 0;
1875
1876         if (!local->ops->conf_tx)
1877                 return false;
1878
1879         if (local->hw.queues < IEEE80211_NUM_ACS)
1880                 return false;
1881
1882         if (!wmm_param)
1883                 return false;
1884
1885         if (wmm_param_len < 8 || wmm_param[5] /* version */ != 1)
1886                 return false;
1887
1888         if (ifmgd->flags & IEEE80211_STA_UAPSD_ENABLED)
1889                 uapsd_queues = ifmgd->uapsd_queues;
1890
1891         count = wmm_param[6] & 0x0f;
1892         if (count == ifmgd->wmm_last_param_set)
1893                 return false;
1894         ifmgd->wmm_last_param_set = count;
1895
1896         pos = wmm_param + 8;
1897         left = wmm_param_len - 8;
1898
1899         memset(&params, 0, sizeof(params));
1900
1901         sdata->wmm_acm = 0;
1902         for (; left >= 4; left -= 4, pos += 4) {
1903                 int aci = (pos[0] >> 5) & 0x03;
1904                 int acm = (pos[0] >> 4) & 0x01;
1905                 bool uapsd = false;
1906
1907                 switch (aci) {
1908                 case 1: /* AC_BK */
1909                         ac = IEEE80211_AC_BK;
1910                         if (acm)
1911                                 sdata->wmm_acm |= BIT(1) | BIT(2); /* BK/- */
1912                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BK)
1913                                 uapsd = true;
1914                         params[ac].mu_edca = !!mu_edca;
1915                         if (mu_edca)
1916                                 params[ac].mu_edca_param_rec = mu_edca->ac_bk;
1917                         break;
1918                 case 2: /* AC_VI */
1919                         ac = IEEE80211_AC_VI;
1920                         if (acm)
1921                                 sdata->wmm_acm |= BIT(4) | BIT(5); /* CL/VI */
1922                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VI)
1923                                 uapsd = true;
1924                         params[ac].mu_edca = !!mu_edca;
1925                         if (mu_edca)
1926                                 params[ac].mu_edca_param_rec = mu_edca->ac_vi;
1927                         break;
1928                 case 3: /* AC_VO */
1929                         ac = IEEE80211_AC_VO;
1930                         if (acm)
1931                                 sdata->wmm_acm |= BIT(6) | BIT(7); /* VO/NC */
1932                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VO)
1933                                 uapsd = true;
1934                         params[ac].mu_edca = !!mu_edca;
1935                         if (mu_edca)
1936                                 params[ac].mu_edca_param_rec = mu_edca->ac_vo;
1937                         break;
1938                 case 0: /* AC_BE */
1939                 default:
1940                         ac = IEEE80211_AC_BE;
1941                         if (acm)
1942                                 sdata->wmm_acm |= BIT(0) | BIT(3); /* BE/EE */
1943                         if (uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BE)
1944                                 uapsd = true;
1945                         params[ac].mu_edca = !!mu_edca;
1946                         if (mu_edca)
1947                                 params[ac].mu_edca_param_rec = mu_edca->ac_be;
1948                         break;
1949                 }
1950
1951                 params[ac].aifs = pos[0] & 0x0f;
1952
1953                 if (params[ac].aifs < 2) {
1954                         sdata_info(sdata,
1955                                    "AP has invalid WMM params (AIFSN=%d for ACI %d), will use 2\n",
1956                                    params[ac].aifs, aci);
1957                         params[ac].aifs = 2;
1958                 }
1959                 params[ac].cw_max = ecw2cw((pos[1] & 0xf0) >> 4);
1960                 params[ac].cw_min = ecw2cw(pos[1] & 0x0f);
1961                 params[ac].txop = get_unaligned_le16(pos + 2);
1962                 params[ac].acm = acm;
1963                 params[ac].uapsd = uapsd;
1964
1965                 if (params[ac].cw_min == 0 ||
1966                     params[ac].cw_min > params[ac].cw_max) {
1967                         sdata_info(sdata,
1968                                    "AP has invalid WMM params (CWmin/max=%d/%d for ACI %d), using defaults\n",
1969                                    params[ac].cw_min, params[ac].cw_max, aci);
1970                         return false;
1971                 }
1972                 ieee80211_regulatory_limit_wmm_params(sdata, &params[ac], ac);
1973         }
1974
1975         /* WMM specification requires all 4 ACIs. */
1976         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1977                 if (params[ac].cw_min == 0) {
1978                         sdata_info(sdata,
1979                                    "AP has invalid WMM params (missing AC %d), using defaults\n",
1980                                    ac);
1981                         return false;
1982                 }
1983         }
1984
1985         for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
1986                 mlme_dbg(sdata,
1987                          "WMM AC=%d acm=%d aifs=%d cWmin=%d cWmax=%d txop=%d uapsd=%d, downgraded=%d\n",
1988                          ac, params[ac].acm,
1989                          params[ac].aifs, params[ac].cw_min, params[ac].cw_max,
1990                          params[ac].txop, params[ac].uapsd,
1991                          ifmgd->tx_tspec[ac].downgraded);
1992                 sdata->tx_conf[ac] = params[ac];
1993                 if (!ifmgd->tx_tspec[ac].downgraded &&
1994                     drv_conf_tx(local, sdata, ac, &params[ac]))
1995                         sdata_err(sdata,
1996                                   "failed to set TX queue parameters for AC %d\n",
1997                                   ac);
1998         }
1999
2000         /* enable WMM or activate new settings */
2001         sdata->vif.bss_conf.qos = true;
2002         return true;
2003 }
2004
2005 static void __ieee80211_stop_poll(struct ieee80211_sub_if_data *sdata)
2006 {
2007         lockdep_assert_held(&sdata->local->mtx);
2008
2009         sdata->u.mgd.flags &= ~IEEE80211_STA_CONNECTION_POLL;
2010         ieee80211_run_deferred_scan(sdata->local);
2011 }
2012
2013 static void ieee80211_stop_poll(struct ieee80211_sub_if_data *sdata)
2014 {
2015         mutex_lock(&sdata->local->mtx);
2016         __ieee80211_stop_poll(sdata);
2017         mutex_unlock(&sdata->local->mtx);
2018 }
2019
2020 static u32 ieee80211_handle_bss_capability(struct ieee80211_sub_if_data *sdata,
2021                                            u16 capab, bool erp_valid, u8 erp)
2022 {
2023         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
2024         struct ieee80211_supported_band *sband;
2025         u32 changed = 0;
2026         bool use_protection;
2027         bool use_short_preamble;
2028         bool use_short_slot;
2029
2030         sband = ieee80211_get_sband(sdata);
2031         if (!sband)
2032                 return changed;
2033
2034         if (erp_valid) {
2035                 use_protection = (erp & WLAN_ERP_USE_PROTECTION) != 0;
2036                 use_short_preamble = (erp & WLAN_ERP_BARKER_PREAMBLE) == 0;
2037         } else {
2038                 use_protection = false;
2039                 use_short_preamble = !!(capab & WLAN_CAPABILITY_SHORT_PREAMBLE);
2040         }
2041
2042         use_short_slot = !!(capab & WLAN_CAPABILITY_SHORT_SLOT_TIME);
2043         if (sband->band == NL80211_BAND_5GHZ)
2044                 use_short_slot = true;
2045
2046         if (use_protection != bss_conf->use_cts_prot) {
2047                 bss_conf->use_cts_prot = use_protection;
2048                 changed |= BSS_CHANGED_ERP_CTS_PROT;
2049         }
2050
2051         if (use_short_preamble != bss_conf->use_short_preamble) {
2052                 bss_conf->use_short_preamble = use_short_preamble;
2053                 changed |= BSS_CHANGED_ERP_PREAMBLE;
2054         }
2055
2056         if (use_short_slot != bss_conf->use_short_slot) {
2057                 bss_conf->use_short_slot = use_short_slot;
2058                 changed |= BSS_CHANGED_ERP_SLOT;
2059         }
2060
2061         return changed;
2062 }
2063
2064 static void ieee80211_set_associated(struct ieee80211_sub_if_data *sdata,
2065                                      struct cfg80211_bss *cbss,
2066                                      u32 bss_info_changed)
2067 {
2068         struct ieee80211_bss *bss = (void *)cbss->priv;
2069         struct ieee80211_local *local = sdata->local;
2070         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
2071
2072         bss_info_changed |= BSS_CHANGED_ASSOC;
2073         bss_info_changed |= ieee80211_handle_bss_capability(sdata,
2074                 bss_conf->assoc_capability, bss->has_erp_value, bss->erp_value);
2075
2076         sdata->u.mgd.beacon_timeout = usecs_to_jiffies(ieee80211_tu_to_usec(
2077                 beacon_loss_count * bss_conf->beacon_int));
2078
2079         sdata->u.mgd.associated = cbss;
2080         memcpy(sdata->u.mgd.bssid, cbss->bssid, ETH_ALEN);
2081
2082         ieee80211_check_rate_mask(sdata);
2083
2084         sdata->u.mgd.flags |= IEEE80211_STA_RESET_SIGNAL_AVE;
2085
2086         if (sdata->vif.p2p ||
2087             sdata->vif.driver_flags & IEEE80211_VIF_GET_NOA_UPDATE) {
2088                 const struct cfg80211_bss_ies *ies;
2089
2090                 rcu_read_lock();
2091                 ies = rcu_dereference(cbss->ies);
2092                 if (ies) {
2093                         int ret;
2094
2095                         ret = cfg80211_get_p2p_attr(
2096                                         ies->data, ies->len,
2097                                         IEEE80211_P2P_ATTR_ABSENCE_NOTICE,
2098                                         (u8 *) &bss_conf->p2p_noa_attr,
2099                                         sizeof(bss_conf->p2p_noa_attr));
2100                         if (ret >= 2) {
2101                                 sdata->u.mgd.p2p_noa_index =
2102                                         bss_conf->p2p_noa_attr.index;
2103                                 bss_info_changed |= BSS_CHANGED_P2P_PS;
2104                         }
2105                 }
2106                 rcu_read_unlock();
2107         }
2108
2109         /* just to be sure */
2110         ieee80211_stop_poll(sdata);
2111
2112         ieee80211_led_assoc(local, 1);
2113
2114         if (sdata->u.mgd.have_beacon) {
2115                 /*
2116                  * If the AP is buggy we may get here with no DTIM period
2117                  * known, so assume it's 1 which is the only safe assumption
2118                  * in that case, although if the TIM IE is broken powersave
2119                  * probably just won't work at all.
2120                  */
2121                 bss_conf->dtim_period = sdata->u.mgd.dtim_period ?: 1;
2122                 bss_conf->beacon_rate = bss->beacon_rate;
2123                 bss_info_changed |= BSS_CHANGED_BEACON_INFO;
2124         } else {
2125                 bss_conf->beacon_rate = NULL;
2126                 bss_conf->dtim_period = 0;
2127         }
2128
2129         bss_conf->assoc = 1;
2130
2131         /* Tell the driver to monitor connection quality (if supported) */
2132         if (sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI &&
2133             bss_conf->cqm_rssi_thold)
2134                 bss_info_changed |= BSS_CHANGED_CQM;
2135
2136         /* Enable ARP filtering */
2137         if (bss_conf->arp_addr_cnt)
2138                 bss_info_changed |= BSS_CHANGED_ARP_FILTER;
2139
2140         ieee80211_bss_info_change_notify(sdata, bss_info_changed);
2141
2142         mutex_lock(&local->iflist_mtx);
2143         ieee80211_recalc_ps(local);
2144         mutex_unlock(&local->iflist_mtx);
2145
2146         ieee80211_recalc_smps(sdata);
2147         ieee80211_recalc_ps_vif(sdata);
2148
2149         netif_carrier_on(sdata->dev);
2150 }
2151
2152 static void ieee80211_set_disassoc(struct ieee80211_sub_if_data *sdata,
2153                                    u16 stype, u16 reason, bool tx,
2154                                    u8 *frame_buf)
2155 {
2156         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2157         struct ieee80211_local *local = sdata->local;
2158         u32 changed = 0;
2159
2160         sdata_assert_lock(sdata);
2161
2162         if (WARN_ON_ONCE(tx && !frame_buf))
2163                 return;
2164
2165         if (WARN_ON(!ifmgd->associated))
2166                 return;
2167
2168         ieee80211_stop_poll(sdata);
2169
2170         ifmgd->associated = NULL;
2171         netif_carrier_off(sdata->dev);
2172
2173         /*
2174          * if we want to get out of ps before disassoc (why?) we have
2175          * to do it before sending disassoc, as otherwise the null-packet
2176          * won't be valid.
2177          */
2178         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
2179                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
2180                 ieee80211_hw_config(local, IEEE80211_CONF_CHANGE_PS);
2181         }
2182         local->ps_sdata = NULL;
2183
2184         /* disable per-vif ps */
2185         ieee80211_recalc_ps_vif(sdata);
2186
2187         /* make sure ongoing transmission finishes */
2188         synchronize_net();
2189
2190         /*
2191          * drop any frame before deauth/disassoc, this can be data or
2192          * management frame. Since we are disconnecting, we should not
2193          * insist sending these frames which can take time and delay
2194          * the disconnection and possible the roaming.
2195          */
2196         if (tx)
2197                 ieee80211_flush_queues(local, sdata, true);
2198
2199         /* deauthenticate/disassociate now */
2200         if (tx || frame_buf) {
2201                 /*
2202                  * In multi channel scenarios guarantee that the virtual
2203                  * interface is granted immediate airtime to transmit the
2204                  * deauthentication frame by calling mgd_prepare_tx, if the
2205                  * driver requested so.
2206                  */
2207                 if (ieee80211_hw_check(&local->hw, DEAUTH_NEED_MGD_TX_PREP) &&
2208                     !ifmgd->have_beacon)
2209                         drv_mgd_prepare_tx(sdata->local, sdata, 0);
2210
2211                 ieee80211_send_deauth_disassoc(sdata, ifmgd->bssid, stype,
2212                                                reason, tx, frame_buf);
2213         }
2214
2215         /* flush out frame - make sure the deauth was actually sent */
2216         if (tx)
2217                 ieee80211_flush_queues(local, sdata, false);
2218
2219         /* clear bssid only after building the needed mgmt frames */
2220         eth_zero_addr(ifmgd->bssid);
2221
2222         /* remove AP and TDLS peers */
2223         sta_info_flush(sdata);
2224
2225         /* finally reset all BSS / config parameters */
2226         changed |= ieee80211_reset_erp_info(sdata);
2227
2228         ieee80211_led_assoc(local, 0);
2229         changed |= BSS_CHANGED_ASSOC;
2230         sdata->vif.bss_conf.assoc = false;
2231
2232         ifmgd->p2p_noa_index = -1;
2233         memset(&sdata->vif.bss_conf.p2p_noa_attr, 0,
2234                sizeof(sdata->vif.bss_conf.p2p_noa_attr));
2235
2236         /* on the next assoc, re-program HT/VHT parameters */
2237         memset(&ifmgd->ht_capa, 0, sizeof(ifmgd->ht_capa));
2238         memset(&ifmgd->ht_capa_mask, 0, sizeof(ifmgd->ht_capa_mask));
2239         memset(&ifmgd->vht_capa, 0, sizeof(ifmgd->vht_capa));
2240         memset(&ifmgd->vht_capa_mask, 0, sizeof(ifmgd->vht_capa_mask));
2241
2242         /* reset MU-MIMO ownership and group data */
2243         memset(sdata->vif.bss_conf.mu_group.membership, 0,
2244                sizeof(sdata->vif.bss_conf.mu_group.membership));
2245         memset(sdata->vif.bss_conf.mu_group.position, 0,
2246                sizeof(sdata->vif.bss_conf.mu_group.position));
2247         changed |= BSS_CHANGED_MU_GROUPS;
2248         sdata->vif.mu_mimo_owner = false;
2249
2250         sdata->ap_power_level = IEEE80211_UNSET_POWER_LEVEL;
2251
2252         del_timer_sync(&local->dynamic_ps_timer);
2253         cancel_work_sync(&local->dynamic_ps_enable_work);
2254
2255         /* Disable ARP filtering */
2256         if (sdata->vif.bss_conf.arp_addr_cnt)
2257                 changed |= BSS_CHANGED_ARP_FILTER;
2258
2259         sdata->vif.bss_conf.qos = false;
2260         changed |= BSS_CHANGED_QOS;
2261
2262         /* The BSSID (not really interesting) and HT changed */
2263         changed |= BSS_CHANGED_BSSID | BSS_CHANGED_HT;
2264         ieee80211_bss_info_change_notify(sdata, changed);
2265
2266         /* disassociated - set to defaults now */
2267         ieee80211_set_wmm_default(sdata, false, false);
2268
2269         del_timer_sync(&sdata->u.mgd.conn_mon_timer);
2270         del_timer_sync(&sdata->u.mgd.bcn_mon_timer);
2271         del_timer_sync(&sdata->u.mgd.timer);
2272         del_timer_sync(&sdata->u.mgd.chswitch_timer);
2273
2274         sdata->vif.bss_conf.dtim_period = 0;
2275         sdata->vif.bss_conf.beacon_rate = NULL;
2276
2277         ifmgd->have_beacon = false;
2278
2279         ifmgd->flags = 0;
2280         mutex_lock(&local->mtx);
2281         ieee80211_vif_release_channel(sdata);
2282
2283         sdata->vif.csa_active = false;
2284         ifmgd->csa_waiting_bcn = false;
2285         ifmgd->csa_ignored_same_chan = false;
2286         if (sdata->csa_block_tx) {
2287                 ieee80211_wake_vif_queues(local, sdata,
2288                                           IEEE80211_QUEUE_STOP_REASON_CSA);
2289                 sdata->csa_block_tx = false;
2290         }
2291         mutex_unlock(&local->mtx);
2292
2293         /* existing TX TSPEC sessions no longer exist */
2294         memset(ifmgd->tx_tspec, 0, sizeof(ifmgd->tx_tspec));
2295         cancel_delayed_work_sync(&ifmgd->tx_tspec_wk);
2296
2297         sdata->encrypt_headroom = IEEE80211_ENCRYPT_HEADROOM;
2298 }
2299
2300 void ieee80211_sta_rx_notify(struct ieee80211_sub_if_data *sdata,
2301                              struct ieee80211_hdr *hdr)
2302 {
2303         /*
2304          * We can postpone the mgd.timer whenever receiving unicast frames
2305          * from AP because we know that the connection is working both ways
2306          * at that time. But multicast frames (and hence also beacons) must
2307          * be ignored here, because we need to trigger the timer during
2308          * data idle periods for sending the periodic probe request to the
2309          * AP we're connected to.
2310          */
2311         if (is_multicast_ether_addr(hdr->addr1))
2312                 return;
2313
2314         ieee80211_sta_reset_conn_monitor(sdata);
2315 }
2316
2317 static void ieee80211_reset_ap_probe(struct ieee80211_sub_if_data *sdata)
2318 {
2319         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2320         struct ieee80211_local *local = sdata->local;
2321
2322         mutex_lock(&local->mtx);
2323         if (!(ifmgd->flags & IEEE80211_STA_CONNECTION_POLL))
2324                 goto out;
2325
2326         __ieee80211_stop_poll(sdata);
2327
2328         mutex_lock(&local->iflist_mtx);
2329         ieee80211_recalc_ps(local);
2330         mutex_unlock(&local->iflist_mtx);
2331
2332         if (ieee80211_hw_check(&sdata->local->hw, CONNECTION_MONITOR))
2333                 goto out;
2334
2335         /*
2336          * We've received a probe response, but are not sure whether
2337          * we have or will be receiving any beacons or data, so let's
2338          * schedule the timers again, just in case.
2339          */
2340         ieee80211_sta_reset_beacon_monitor(sdata);
2341
2342         mod_timer(&ifmgd->conn_mon_timer,
2343                   round_jiffies_up(jiffies +
2344                                    IEEE80211_CONNECTION_IDLE_TIME));
2345 out:
2346         mutex_unlock(&local->mtx);
2347 }
2348
2349 static void ieee80211_sta_tx_wmm_ac_notify(struct ieee80211_sub_if_data *sdata,
2350                                            struct ieee80211_hdr *hdr,
2351                                            u16 tx_time)
2352 {
2353         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2354         u16 tid = ieee80211_get_tid(hdr);
2355         int ac = ieee80211_ac_from_tid(tid);
2356         struct ieee80211_sta_tx_tspec *tx_tspec = &ifmgd->tx_tspec[ac];
2357         unsigned long now = jiffies;
2358
2359         if (likely(!tx_tspec->admitted_time))
2360                 return;
2361
2362         if (time_after(now, tx_tspec->time_slice_start + HZ)) {
2363                 tx_tspec->consumed_tx_time = 0;
2364                 tx_tspec->time_slice_start = now;
2365
2366                 if (tx_tspec->downgraded) {
2367                         tx_tspec->action = TX_TSPEC_ACTION_STOP_DOWNGRADE;
2368                         schedule_delayed_work(&ifmgd->tx_tspec_wk, 0);
2369                 }
2370         }
2371
2372         if (tx_tspec->downgraded)
2373                 return;
2374
2375         tx_tspec->consumed_tx_time += tx_time;
2376
2377         if (tx_tspec->consumed_tx_time >= tx_tspec->admitted_time) {
2378                 tx_tspec->downgraded = true;
2379                 tx_tspec->action = TX_TSPEC_ACTION_DOWNGRADE;
2380                 schedule_delayed_work(&ifmgd->tx_tspec_wk, 0);
2381         }
2382 }
2383
2384 void ieee80211_sta_tx_notify(struct ieee80211_sub_if_data *sdata,
2385                              struct ieee80211_hdr *hdr, bool ack, u16 tx_time)
2386 {
2387         ieee80211_sta_tx_wmm_ac_notify(sdata, hdr, tx_time);
2388
2389         if (!ieee80211_is_data(hdr->frame_control))
2390             return;
2391
2392         if (ieee80211_is_any_nullfunc(hdr->frame_control) &&
2393             sdata->u.mgd.probe_send_count > 0) {
2394                 if (ack)
2395                         ieee80211_sta_reset_conn_monitor(sdata);
2396                 else
2397                         sdata->u.mgd.nullfunc_failed = true;
2398                 ieee80211_queue_work(&sdata->local->hw, &sdata->work);
2399                 return;
2400         }
2401
2402         if (ack)
2403                 ieee80211_sta_reset_conn_monitor(sdata);
2404 }
2405
2406 static void ieee80211_mlme_send_probe_req(struct ieee80211_sub_if_data *sdata,
2407                                           const u8 *src, const u8 *dst,
2408                                           const u8 *ssid, size_t ssid_len,
2409                                           struct ieee80211_channel *channel)
2410 {
2411         struct sk_buff *skb;
2412
2413         skb = ieee80211_build_probe_req(sdata, src, dst, (u32)-1, channel,
2414                                         ssid, ssid_len, NULL, 0,
2415                                         IEEE80211_PROBE_FLAG_DIRECTED);
2416         if (skb)
2417                 ieee80211_tx_skb(sdata, skb);
2418 }
2419
2420 static void ieee80211_mgd_probe_ap_send(struct ieee80211_sub_if_data *sdata)
2421 {
2422         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2423         const u8 *ssid;
2424         u8 *dst = ifmgd->associated->bssid;
2425         u8 unicast_limit = max(1, max_probe_tries - 3);
2426         struct sta_info *sta;
2427
2428         /*
2429          * Try sending broadcast probe requests for the last three
2430          * probe requests after the first ones failed since some
2431          * buggy APs only support broadcast probe requests.
2432          */
2433         if (ifmgd->probe_send_count >= unicast_limit)
2434                 dst = NULL;
2435
2436         /*
2437          * When the hardware reports an accurate Tx ACK status, it's
2438          * better to send a nullfunc frame instead of a probe request,
2439          * as it will kick us off the AP quickly if we aren't associated
2440          * anymore. The timeout will be reset if the frame is ACKed by
2441          * the AP.
2442          */
2443         ifmgd->probe_send_count++;
2444
2445         if (dst) {
2446                 mutex_lock(&sdata->local->sta_mtx);
2447                 sta = sta_info_get(sdata, dst);
2448                 if (!WARN_ON(!sta))
2449                         ieee80211_check_fast_rx(sta);
2450                 mutex_unlock(&sdata->local->sta_mtx);
2451         }
2452
2453         if (ieee80211_hw_check(&sdata->local->hw, REPORTS_TX_ACK_STATUS)) {
2454                 ifmgd->nullfunc_failed = false;
2455                 ieee80211_send_nullfunc(sdata->local, sdata, false);
2456         } else {
2457                 int ssid_len;
2458
2459                 rcu_read_lock();
2460                 ssid = ieee80211_bss_get_ie(ifmgd->associated, WLAN_EID_SSID);
2461                 if (WARN_ON_ONCE(ssid == NULL))
2462                         ssid_len = 0;
2463                 else
2464                         ssid_len = ssid[1];
2465
2466                 ieee80211_mlme_send_probe_req(sdata, sdata->vif.addr, dst,
2467                                               ssid + 2, ssid_len,
2468                                               ifmgd->associated->channel);
2469                 rcu_read_unlock();
2470         }
2471
2472         ifmgd->probe_timeout = jiffies + msecs_to_jiffies(probe_wait_ms);
2473         run_again(sdata, ifmgd->probe_timeout);
2474 }
2475
2476 static void ieee80211_mgd_probe_ap(struct ieee80211_sub_if_data *sdata,
2477                                    bool beacon)
2478 {
2479         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2480         bool already = false;
2481
2482         if (!ieee80211_sdata_running(sdata))
2483                 return;
2484
2485         sdata_lock(sdata);
2486
2487         if (!ifmgd->associated)
2488                 goto out;
2489
2490         mutex_lock(&sdata->local->mtx);
2491
2492         if (sdata->local->tmp_channel || sdata->local->scanning) {
2493                 mutex_unlock(&sdata->local->mtx);
2494                 goto out;
2495         }
2496
2497         if (beacon) {
2498                 mlme_dbg_ratelimited(sdata,
2499                                      "detected beacon loss from AP (missed %d beacons) - probing\n",
2500                                      beacon_loss_count);
2501
2502                 ieee80211_cqm_beacon_loss_notify(&sdata->vif, GFP_KERNEL);
2503         }
2504
2505         /*
2506          * The driver/our work has already reported this event or the
2507          * connection monitoring has kicked in and we have already sent
2508          * a probe request. Or maybe the AP died and the driver keeps
2509          * reporting until we disassociate...
2510          *
2511          * In either case we have to ignore the current call to this
2512          * function (except for setting the correct probe reason bit)
2513          * because otherwise we would reset the timer every time and
2514          * never check whether we received a probe response!
2515          */
2516         if (ifmgd->flags & IEEE80211_STA_CONNECTION_POLL)
2517                 already = true;
2518
2519         ifmgd->flags |= IEEE80211_STA_CONNECTION_POLL;
2520
2521         mutex_unlock(&sdata->local->mtx);
2522
2523         if (already)
2524                 goto out;
2525
2526         mutex_lock(&sdata->local->iflist_mtx);
2527         ieee80211_recalc_ps(sdata->local);
2528         mutex_unlock(&sdata->local->iflist_mtx);
2529
2530         ifmgd->probe_send_count = 0;
2531         ieee80211_mgd_probe_ap_send(sdata);
2532  out:
2533         sdata_unlock(sdata);
2534 }
2535
2536 struct sk_buff *ieee80211_ap_probereq_get(struct ieee80211_hw *hw,
2537                                           struct ieee80211_vif *vif)
2538 {
2539         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2540         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2541         struct cfg80211_bss *cbss;
2542         struct sk_buff *skb;
2543         const u8 *ssid;
2544         int ssid_len;
2545
2546         if (WARN_ON(sdata->vif.type != NL80211_IFTYPE_STATION))
2547                 return NULL;
2548
2549         sdata_assert_lock(sdata);
2550
2551         if (ifmgd->associated)
2552                 cbss = ifmgd->associated;
2553         else if (ifmgd->auth_data)
2554                 cbss = ifmgd->auth_data->bss;
2555         else if (ifmgd->assoc_data)
2556                 cbss = ifmgd->assoc_data->bss;
2557         else
2558                 return NULL;
2559
2560         rcu_read_lock();
2561         ssid = ieee80211_bss_get_ie(cbss, WLAN_EID_SSID);
2562         if (WARN_ONCE(!ssid || ssid[1] > IEEE80211_MAX_SSID_LEN,
2563                       "invalid SSID element (len=%d)", ssid ? ssid[1] : -1))
2564                 ssid_len = 0;
2565         else
2566                 ssid_len = ssid[1];
2567
2568         skb = ieee80211_build_probe_req(sdata, sdata->vif.addr, cbss->bssid,
2569                                         (u32) -1, cbss->channel,
2570                                         ssid + 2, ssid_len,
2571                                         NULL, 0, IEEE80211_PROBE_FLAG_DIRECTED);
2572         rcu_read_unlock();
2573
2574         return skb;
2575 }
2576 EXPORT_SYMBOL(ieee80211_ap_probereq_get);
2577
2578 static void ieee80211_report_disconnect(struct ieee80211_sub_if_data *sdata,
2579                                         const u8 *buf, size_t len, bool tx,
2580                                         u16 reason)
2581 {
2582         struct ieee80211_event event = {
2583                 .type = MLME_EVENT,
2584                 .u.mlme.data = tx ? DEAUTH_TX_EVENT : DEAUTH_RX_EVENT,
2585                 .u.mlme.reason = reason,
2586         };
2587
2588         if (tx)
2589                 cfg80211_tx_mlme_mgmt(sdata->dev, buf, len);
2590         else
2591                 cfg80211_rx_mlme_mgmt(sdata->dev, buf, len);
2592
2593         drv_event_callback(sdata->local, sdata, &event);
2594 }
2595
2596 static void __ieee80211_disconnect(struct ieee80211_sub_if_data *sdata)
2597 {
2598         struct ieee80211_local *local = sdata->local;
2599         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2600         u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
2601         bool tx;
2602
2603         sdata_lock(sdata);
2604         if (!ifmgd->associated) {
2605                 sdata_unlock(sdata);
2606                 return;
2607         }
2608
2609         tx = !sdata->csa_block_tx;
2610
2611         /* AP is probably out of range (or not reachable for another reason) so
2612          * remove the bss struct for that AP.
2613          */
2614         cfg80211_unlink_bss(local->hw.wiphy, ifmgd->associated);
2615
2616         ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH,
2617                                WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
2618                                tx, frame_buf);
2619         mutex_lock(&local->mtx);
2620         sdata->vif.csa_active = false;
2621         ifmgd->csa_waiting_bcn = false;
2622         if (sdata->csa_block_tx) {
2623                 ieee80211_wake_vif_queues(local, sdata,
2624                                           IEEE80211_QUEUE_STOP_REASON_CSA);
2625                 sdata->csa_block_tx = false;
2626         }
2627         mutex_unlock(&local->mtx);
2628
2629         ieee80211_report_disconnect(sdata, frame_buf, sizeof(frame_buf), tx,
2630                                     WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY);
2631
2632         sdata_unlock(sdata);
2633 }
2634
2635 static void ieee80211_beacon_connection_loss_work(struct work_struct *work)
2636 {
2637         struct ieee80211_sub_if_data *sdata =
2638                 container_of(work, struct ieee80211_sub_if_data,
2639                              u.mgd.beacon_connection_loss_work);
2640         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2641
2642         if (ifmgd->associated)
2643                 ifmgd->beacon_loss_count++;
2644
2645         if (ifmgd->connection_loss) {
2646                 sdata_info(sdata, "Connection to AP %pM lost\n",
2647                            ifmgd->bssid);
2648                 __ieee80211_disconnect(sdata);
2649         } else {
2650                 ieee80211_mgd_probe_ap(sdata, true);
2651         }
2652 }
2653
2654 static void ieee80211_csa_connection_drop_work(struct work_struct *work)
2655 {
2656         struct ieee80211_sub_if_data *sdata =
2657                 container_of(work, struct ieee80211_sub_if_data,
2658                              u.mgd.csa_connection_drop_work);
2659
2660         __ieee80211_disconnect(sdata);
2661 }
2662
2663 void ieee80211_beacon_loss(struct ieee80211_vif *vif)
2664 {
2665         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2666         struct ieee80211_hw *hw = &sdata->local->hw;
2667
2668         trace_api_beacon_loss(sdata);
2669
2670         sdata->u.mgd.connection_loss = false;
2671         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
2672 }
2673 EXPORT_SYMBOL(ieee80211_beacon_loss);
2674
2675 void ieee80211_connection_loss(struct ieee80211_vif *vif)
2676 {
2677         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
2678         struct ieee80211_hw *hw = &sdata->local->hw;
2679
2680         trace_api_connection_loss(sdata);
2681
2682         sdata->u.mgd.connection_loss = true;
2683         ieee80211_queue_work(hw, &sdata->u.mgd.beacon_connection_loss_work);
2684 }
2685 EXPORT_SYMBOL(ieee80211_connection_loss);
2686
2687
2688 static void ieee80211_destroy_auth_data(struct ieee80211_sub_if_data *sdata,
2689                                         bool assoc)
2690 {
2691         struct ieee80211_mgd_auth_data *auth_data = sdata->u.mgd.auth_data;
2692
2693         sdata_assert_lock(sdata);
2694
2695         if (!assoc) {
2696                 /*
2697                  * we are not authenticated yet, the only timer that could be
2698                  * running is the timeout for the authentication response which
2699                  * which is not relevant anymore.
2700                  */
2701                 del_timer_sync(&sdata->u.mgd.timer);
2702                 sta_info_destroy_addr(sdata, auth_data->bss->bssid);
2703
2704                 eth_zero_addr(sdata->u.mgd.bssid);
2705                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BSSID);
2706                 sdata->u.mgd.flags = 0;
2707                 mutex_lock(&sdata->local->mtx);
2708                 ieee80211_vif_release_channel(sdata);
2709                 mutex_unlock(&sdata->local->mtx);
2710         }
2711
2712         cfg80211_put_bss(sdata->local->hw.wiphy, auth_data->bss);
2713         kfree(auth_data);
2714         sdata->u.mgd.auth_data = NULL;
2715 }
2716
2717 static void ieee80211_destroy_assoc_data(struct ieee80211_sub_if_data *sdata,
2718                                          bool assoc, bool abandon)
2719 {
2720         struct ieee80211_mgd_assoc_data *assoc_data = sdata->u.mgd.assoc_data;
2721
2722         sdata_assert_lock(sdata);
2723
2724         if (!assoc) {
2725                 /*
2726                  * we are not associated yet, the only timer that could be
2727                  * running is the timeout for the association response which
2728                  * which is not relevant anymore.
2729                  */
2730                 del_timer_sync(&sdata->u.mgd.timer);
2731                 sta_info_destroy_addr(sdata, assoc_data->bss->bssid);
2732
2733                 eth_zero_addr(sdata->u.mgd.bssid);
2734                 ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BSSID);
2735                 sdata->u.mgd.flags = 0;
2736                 sdata->vif.mu_mimo_owner = false;
2737
2738                 mutex_lock(&sdata->local->mtx);
2739                 ieee80211_vif_release_channel(sdata);
2740                 mutex_unlock(&sdata->local->mtx);
2741
2742                 if (abandon)
2743                         cfg80211_abandon_assoc(sdata->dev, assoc_data->bss);
2744         }
2745
2746         kfree(assoc_data);
2747         sdata->u.mgd.assoc_data = NULL;
2748 }
2749
2750 static void ieee80211_auth_challenge(struct ieee80211_sub_if_data *sdata,
2751                                      struct ieee80211_mgmt *mgmt, size_t len)
2752 {
2753         struct ieee80211_local *local = sdata->local;
2754         struct ieee80211_mgd_auth_data *auth_data = sdata->u.mgd.auth_data;
2755         u8 *pos;
2756         struct ieee802_11_elems elems;
2757         u32 tx_flags = 0;
2758
2759         pos = mgmt->u.auth.variable;
2760         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), false, &elems);
2761         if (!elems.challenge)
2762                 return;
2763         auth_data->expected_transaction = 4;
2764         drv_mgd_prepare_tx(sdata->local, sdata, 0);
2765         if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
2766                 tx_flags = IEEE80211_TX_CTL_REQ_TX_STATUS |
2767                            IEEE80211_TX_INTFL_MLME_CONN_TX;
2768         ieee80211_send_auth(sdata, 3, auth_data->algorithm, 0,
2769                             elems.challenge - 2, elems.challenge_len + 2,
2770                             auth_data->bss->bssid, auth_data->bss->bssid,
2771                             auth_data->key, auth_data->key_len,
2772                             auth_data->key_idx, tx_flags);
2773 }
2774
2775 static void ieee80211_rx_mgmt_auth(struct ieee80211_sub_if_data *sdata,
2776                                    struct ieee80211_mgmt *mgmt, size_t len)
2777 {
2778         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2779         u8 bssid[ETH_ALEN];
2780         u16 auth_alg, auth_transaction, status_code;
2781         struct sta_info *sta;
2782         struct ieee80211_event event = {
2783                 .type = MLME_EVENT,
2784                 .u.mlme.data = AUTH_EVENT,
2785         };
2786
2787         sdata_assert_lock(sdata);
2788
2789         if (len < 24 + 6)
2790                 return;
2791
2792         if (!ifmgd->auth_data || ifmgd->auth_data->done)
2793                 return;
2794
2795         memcpy(bssid, ifmgd->auth_data->bss->bssid, ETH_ALEN);
2796
2797         if (!ether_addr_equal(bssid, mgmt->bssid))
2798                 return;
2799
2800         auth_alg = le16_to_cpu(mgmt->u.auth.auth_alg);
2801         auth_transaction = le16_to_cpu(mgmt->u.auth.auth_transaction);
2802         status_code = le16_to_cpu(mgmt->u.auth.status_code);
2803
2804         if (auth_alg != ifmgd->auth_data->algorithm ||
2805             auth_transaction != ifmgd->auth_data->expected_transaction) {
2806                 sdata_info(sdata, "%pM unexpected authentication state: alg %d (expected %d) transact %d (expected %d)\n",
2807                            mgmt->sa, auth_alg, ifmgd->auth_data->algorithm,
2808                            auth_transaction,
2809                            ifmgd->auth_data->expected_transaction);
2810                 return;
2811         }
2812
2813         if (status_code != WLAN_STATUS_SUCCESS) {
2814                 sdata_info(sdata, "%pM denied authentication (status %d)\n",
2815                            mgmt->sa, status_code);
2816                 ieee80211_destroy_auth_data(sdata, false);
2817                 cfg80211_rx_mlme_mgmt(sdata->dev, (u8 *)mgmt, len);
2818                 event.u.mlme.status = MLME_DENIED;
2819                 event.u.mlme.reason = status_code;
2820                 drv_event_callback(sdata->local, sdata, &event);
2821                 return;
2822         }
2823
2824         switch (ifmgd->auth_data->algorithm) {
2825         case WLAN_AUTH_OPEN:
2826         case WLAN_AUTH_LEAP:
2827         case WLAN_AUTH_FT:
2828         case WLAN_AUTH_SAE:
2829         case WLAN_AUTH_FILS_SK:
2830         case WLAN_AUTH_FILS_SK_PFS:
2831         case WLAN_AUTH_FILS_PK:
2832                 break;
2833         case WLAN_AUTH_SHARED_KEY:
2834                 if (ifmgd->auth_data->expected_transaction != 4) {
2835                         ieee80211_auth_challenge(sdata, mgmt, len);
2836                         /* need another frame */
2837                         return;
2838                 }
2839                 break;
2840         default:
2841                 WARN_ONCE(1, "invalid auth alg %d",
2842                           ifmgd->auth_data->algorithm);
2843                 return;
2844         }
2845
2846         event.u.mlme.status = MLME_SUCCESS;
2847         drv_event_callback(sdata->local, sdata, &event);
2848         sdata_info(sdata, "authenticated\n");
2849         ifmgd->auth_data->done = true;
2850         ifmgd->auth_data->timeout = jiffies + IEEE80211_AUTH_WAIT_ASSOC;
2851         ifmgd->auth_data->timeout_started = true;
2852         run_again(sdata, ifmgd->auth_data->timeout);
2853
2854         if (ifmgd->auth_data->algorithm == WLAN_AUTH_SAE &&
2855             ifmgd->auth_data->expected_transaction != 2) {
2856                 /*
2857                  * Report auth frame to user space for processing since another
2858                  * round of Authentication frames is still needed.
2859                  */
2860                 cfg80211_rx_mlme_mgmt(sdata->dev, (u8 *)mgmt, len);
2861                 return;
2862         }
2863
2864         /* move station state to auth */
2865         mutex_lock(&sdata->local->sta_mtx);
2866         sta = sta_info_get(sdata, bssid);
2867         if (!sta) {
2868                 WARN_ONCE(1, "%s: STA %pM not found", sdata->name, bssid);
2869                 goto out_err;
2870         }
2871         if (sta_info_move_state(sta, IEEE80211_STA_AUTH)) {
2872                 sdata_info(sdata, "failed moving %pM to auth\n", bssid);
2873                 goto out_err;
2874         }
2875         mutex_unlock(&sdata->local->sta_mtx);
2876
2877         cfg80211_rx_mlme_mgmt(sdata->dev, (u8 *)mgmt, len);
2878         return;
2879  out_err:
2880         mutex_unlock(&sdata->local->sta_mtx);
2881         /* ignore frame -- wait for timeout */
2882 }
2883
2884 #define case_WLAN(type) \
2885         case WLAN_REASON_##type: return #type
2886
2887 const char *ieee80211_get_reason_code_string(u16 reason_code)
2888 {
2889         switch (reason_code) {
2890         case_WLAN(UNSPECIFIED);
2891         case_WLAN(PREV_AUTH_NOT_VALID);
2892         case_WLAN(DEAUTH_LEAVING);
2893         case_WLAN(DISASSOC_DUE_TO_INACTIVITY);
2894         case_WLAN(DISASSOC_AP_BUSY);
2895         case_WLAN(CLASS2_FRAME_FROM_NONAUTH_STA);
2896         case_WLAN(CLASS3_FRAME_FROM_NONASSOC_STA);
2897         case_WLAN(DISASSOC_STA_HAS_LEFT);
2898         case_WLAN(STA_REQ_ASSOC_WITHOUT_AUTH);
2899         case_WLAN(DISASSOC_BAD_POWER);
2900         case_WLAN(DISASSOC_BAD_SUPP_CHAN);
2901         case_WLAN(INVALID_IE);
2902         case_WLAN(MIC_FAILURE);
2903         case_WLAN(4WAY_HANDSHAKE_TIMEOUT);
2904         case_WLAN(GROUP_KEY_HANDSHAKE_TIMEOUT);
2905         case_WLAN(IE_DIFFERENT);
2906         case_WLAN(INVALID_GROUP_CIPHER);
2907         case_WLAN(INVALID_PAIRWISE_CIPHER);
2908         case_WLAN(INVALID_AKMP);
2909         case_WLAN(UNSUPP_RSN_VERSION);
2910         case_WLAN(INVALID_RSN_IE_CAP);
2911         case_WLAN(IEEE8021X_FAILED);
2912         case_WLAN(CIPHER_SUITE_REJECTED);
2913         case_WLAN(DISASSOC_UNSPECIFIED_QOS);
2914         case_WLAN(DISASSOC_QAP_NO_BANDWIDTH);
2915         case_WLAN(DISASSOC_LOW_ACK);
2916         case_WLAN(DISASSOC_QAP_EXCEED_TXOP);
2917         case_WLAN(QSTA_LEAVE_QBSS);
2918         case_WLAN(QSTA_NOT_USE);
2919         case_WLAN(QSTA_REQUIRE_SETUP);
2920         case_WLAN(QSTA_TIMEOUT);
2921         case_WLAN(QSTA_CIPHER_NOT_SUPP);
2922         case_WLAN(MESH_PEER_CANCELED);
2923         case_WLAN(MESH_MAX_PEERS);
2924         case_WLAN(MESH_CONFIG);
2925         case_WLAN(MESH_CLOSE);
2926         case_WLAN(MESH_MAX_RETRIES);
2927         case_WLAN(MESH_CONFIRM_TIMEOUT);
2928         case_WLAN(MESH_INVALID_GTK);
2929         case_WLAN(MESH_INCONSISTENT_PARAM);
2930         case_WLAN(MESH_INVALID_SECURITY);
2931         case_WLAN(MESH_PATH_ERROR);
2932         case_WLAN(MESH_PATH_NOFORWARD);
2933         case_WLAN(MESH_PATH_DEST_UNREACHABLE);
2934         case_WLAN(MAC_EXISTS_IN_MBSS);
2935         case_WLAN(MESH_CHAN_REGULATORY);
2936         case_WLAN(MESH_CHAN);
2937         default: return "<unknown>";
2938         }
2939 }
2940
2941 static void ieee80211_rx_mgmt_deauth(struct ieee80211_sub_if_data *sdata,
2942                                      struct ieee80211_mgmt *mgmt, size_t len)
2943 {
2944         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2945         u16 reason_code = le16_to_cpu(mgmt->u.deauth.reason_code);
2946
2947         sdata_assert_lock(sdata);
2948
2949         if (len < 24 + 2)
2950                 return;
2951
2952         if (!ether_addr_equal(mgmt->bssid, mgmt->sa)) {
2953                 ieee80211_tdls_handle_disconnect(sdata, mgmt->sa, reason_code);
2954                 return;
2955         }
2956
2957         if (ifmgd->associated &&
2958             ether_addr_equal(mgmt->bssid, ifmgd->associated->bssid)) {
2959                 const u8 *bssid = ifmgd->associated->bssid;
2960
2961                 sdata_info(sdata, "deauthenticated from %pM (Reason: %u=%s)\n",
2962                            bssid, reason_code,
2963                            ieee80211_get_reason_code_string(reason_code));
2964
2965                 ieee80211_set_disassoc(sdata, 0, 0, false, NULL);
2966
2967                 ieee80211_report_disconnect(sdata, (u8 *)mgmt, len, false,
2968                                             reason_code);
2969                 return;
2970         }
2971
2972         if (ifmgd->assoc_data &&
2973             ether_addr_equal(mgmt->bssid, ifmgd->assoc_data->bss->bssid)) {
2974                 const u8 *bssid = ifmgd->assoc_data->bss->bssid;
2975
2976                 sdata_info(sdata,
2977                            "deauthenticated from %pM while associating (Reason: %u=%s)\n",
2978                            bssid, reason_code,
2979                            ieee80211_get_reason_code_string(reason_code));
2980
2981                 ieee80211_destroy_assoc_data(sdata, false, true);
2982
2983                 cfg80211_rx_mlme_mgmt(sdata->dev, (u8 *)mgmt, len);
2984                 return;
2985         }
2986 }
2987
2988
2989 static void ieee80211_rx_mgmt_disassoc(struct ieee80211_sub_if_data *sdata,
2990                                        struct ieee80211_mgmt *mgmt, size_t len)
2991 {
2992         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
2993         u16 reason_code;
2994
2995         sdata_assert_lock(sdata);
2996
2997         if (len < 24 + 2)
2998                 return;
2999
3000         if (!ifmgd->associated ||
3001             !ether_addr_equal(mgmt->bssid, ifmgd->associated->bssid))
3002                 return;
3003
3004         reason_code = le16_to_cpu(mgmt->u.disassoc.reason_code);
3005
3006         if (!ether_addr_equal(mgmt->bssid, mgmt->sa)) {
3007                 ieee80211_tdls_handle_disconnect(sdata, mgmt->sa, reason_code);
3008                 return;
3009         }
3010
3011         sdata_info(sdata, "disassociated from %pM (Reason: %u=%s)\n",
3012                    mgmt->sa, reason_code,
3013                    ieee80211_get_reason_code_string(reason_code));
3014
3015         ieee80211_set_disassoc(sdata, 0, 0, false, NULL);
3016
3017         ieee80211_report_disconnect(sdata, (u8 *)mgmt, len, false, reason_code);
3018 }
3019
3020 static void ieee80211_get_rates(struct ieee80211_supported_band *sband,
3021                                 u8 *supp_rates, unsigned int supp_rates_len,
3022                                 u32 *rates, u32 *basic_rates,
3023                                 bool *have_higher_than_11mbit,
3024                                 int *min_rate, int *min_rate_index,
3025                                 int shift)
3026 {
3027         int i, j;
3028
3029         for (i = 0; i < supp_rates_len; i++) {
3030                 int rate = supp_rates[i] & 0x7f;
3031                 bool is_basic = !!(supp_rates[i] & 0x80);
3032
3033                 if ((rate * 5 * (1 << shift)) > 110)
3034                         *have_higher_than_11mbit = true;
3035
3036                 /*
3037                  * Skip HT and VHT BSS membership selectors since they're not
3038                  * rates.
3039                  *
3040                  * Note: Even though the membership selector and the basic
3041                  *       rate flag share the same bit, they are not exactly
3042                  *       the same.
3043                  */
3044                 if (supp_rates[i] == (0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY) ||
3045                     supp_rates[i] == (0x80 | BSS_MEMBERSHIP_SELECTOR_VHT_PHY))
3046                         continue;
3047
3048                 for (j = 0; j < sband->n_bitrates; j++) {
3049                         struct ieee80211_rate *br;
3050                         int brate;
3051
3052                         br = &sband->bitrates[j];
3053
3054                         brate = DIV_ROUND_UP(br->bitrate, (1 << shift) * 5);
3055                         if (brate == rate) {
3056                                 *rates |= BIT(j);
3057                                 if (is_basic)
3058                                         *basic_rates |= BIT(j);
3059                                 if ((rate * 5) < *min_rate) {
3060                                         *min_rate = rate * 5;
3061                                         *min_rate_index = j;
3062                                 }
3063                                 break;
3064                         }
3065                 }
3066         }
3067 }
3068
3069 static bool ieee80211_assoc_success(struct ieee80211_sub_if_data *sdata,
3070                                     struct cfg80211_bss *cbss,
3071                                     struct ieee80211_mgmt *mgmt, size_t len)
3072 {
3073         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
3074         struct ieee80211_local *local = sdata->local;
3075         struct ieee80211_supported_band *sband;
3076         struct sta_info *sta;
3077         u8 *pos;
3078         u16 capab_info, aid;
3079         struct ieee802_11_elems elems;
3080         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
3081         const struct cfg80211_bss_ies *bss_ies = NULL;
3082         struct ieee80211_mgd_assoc_data *assoc_data = ifmgd->assoc_data;
3083         u32 changed = 0;
3084         int err;
3085         bool ret;
3086
3087         /* AssocResp and ReassocResp have identical structure */
3088
3089         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
3090         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
3091
3092         /*
3093          * The 5 MSB of the AID field are reserved
3094          * (802.11-2016 9.4.1.8 AID field)
3095          */
3096         aid &= 0x7ff;
3097
3098         ifmgd->broken_ap = false;
3099
3100         if (aid == 0 || aid > IEEE80211_MAX_AID) {
3101                 sdata_info(sdata, "invalid AID value %d (out of range), turn off PS\n",
3102                            aid);
3103                 aid = 0;
3104                 ifmgd->broken_ap = true;
3105         }
3106
3107         pos = mgmt->u.assoc_resp.variable;
3108         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), false, &elems);
3109
3110         if (!elems.supp_rates) {
3111                 sdata_info(sdata, "no SuppRates element in AssocResp\n");
3112                 return false;
3113         }
3114
3115         ifmgd->aid = aid;
3116         ifmgd->tdls_chan_switch_prohibited =
3117                 elems.ext_capab && elems.ext_capab_len >= 5 &&
3118                 (elems.ext_capab[4] & WLAN_EXT_CAPA5_TDLS_CH_SW_PROHIBITED);
3119
3120         /*
3121          * Some APs are erroneously not including some information in their
3122          * (re)association response frames. Try to recover by using the data
3123          * from the beacon or probe response. This seems to afflict mobile
3124          * 2G/3G/4G wifi routers, reported models include the "Onda PN51T",
3125          * "Vodafone PocketWiFi 2", "ZTE MF60" and a similar T-Mobile device.
3126          */
3127         if ((assoc_data->wmm && !elems.wmm_param) ||
3128             (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT) &&
3129              (!elems.ht_cap_elem || !elems.ht_operation)) ||
3130             (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT) &&
3131              (!elems.vht_cap_elem || !elems.vht_operation))) {
3132                 const struct cfg80211_bss_ies *ies;
3133                 struct ieee802_11_elems bss_elems;
3134
3135                 rcu_read_lock();
3136                 ies = rcu_dereference(cbss->ies);
3137                 if (ies)
3138                         bss_ies = kmemdup(ies, sizeof(*ies) + ies->len,
3139                                           GFP_ATOMIC);
3140                 rcu_read_unlock();
3141                 if (!bss_ies)
3142                         return false;
3143
3144                 ieee802_11_parse_elems(bss_ies->data, bss_ies->len,
3145                                        false, &bss_elems);
3146                 if (assoc_data->wmm &&
3147                     !elems.wmm_param && bss_elems.wmm_param) {
3148                         elems.wmm_param = bss_elems.wmm_param;
3149                         sdata_info(sdata,
3150                                    "AP bug: WMM param missing from AssocResp\n");
3151                 }
3152
3153                 /*
3154                  * Also check if we requested HT/VHT, otherwise the AP doesn't
3155                  * have to include the IEs in the (re)association response.
3156                  */
3157                 if (!elems.ht_cap_elem && bss_elems.ht_cap_elem &&
3158                     !(ifmgd->flags & IEEE80211_STA_DISABLE_HT)) {
3159                         elems.ht_cap_elem = bss_elems.ht_cap_elem;
3160                         sdata_info(sdata,
3161                                    "AP bug: HT capability missing from AssocResp\n");
3162                 }
3163                 if (!elems.ht_operation && bss_elems.ht_operation &&
3164                     !(ifmgd->flags & IEEE80211_STA_DISABLE_HT)) {
3165                         elems.ht_operation = bss_elems.ht_operation;
3166                         sdata_info(sdata,
3167                                    "AP bug: HT operation missing from AssocResp\n");
3168                 }
3169                 if (!elems.vht_cap_elem && bss_elems.vht_cap_elem &&
3170                     !(ifmgd->flags & IEEE80211_STA_DISABLE_VHT)) {
3171                         elems.vht_cap_elem = bss_elems.vht_cap_elem;
3172                         sdata_info(sdata,
3173                                    "AP bug: VHT capa missing from AssocResp\n");
3174                 }
3175                 if (!elems.vht_operation && bss_elems.vht_operation &&
3176                     !(ifmgd->flags & IEEE80211_STA_DISABLE_VHT)) {
3177                         elems.vht_operation = bss_elems.vht_operation;
3178                         sdata_info(sdata,
3179                                    "AP bug: VHT operation missing from AssocResp\n");
3180                 }
3181         }
3182
3183         /*
3184          * We previously checked these in the beacon/probe response, so
3185          * they should be present here. This is just a safety net.
3186          */
3187         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT) &&
3188             (!elems.wmm_param || !elems.ht_cap_elem || !elems.ht_operation)) {
3189                 sdata_info(sdata,
3190                            "HT AP is missing WMM params or HT capability/operation\n");
3191                 ret = false;
3192                 goto out;
3193         }
3194
3195         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT) &&
3196             (!elems.vht_cap_elem || !elems.vht_operation)) {
3197                 sdata_info(sdata,
3198                            "VHT AP is missing VHT capability/operation\n");
3199                 ret = false;
3200                 goto out;
3201         }
3202
3203         mutex_lock(&sdata->local->sta_mtx);
3204         /*
3205          * station info was already allocated and inserted before
3206          * the association and should be available to us
3207          */
3208         sta = sta_info_get(sdata, cbss->bssid);
3209         if (WARN_ON(!sta)) {
3210                 mutex_unlock(&sdata->local->sta_mtx);
3211                 ret = false;
3212                 goto out;
3213         }
3214
3215         sband = ieee80211_get_sband(sdata);
3216         if (!sband) {
3217                 mutex_unlock(&sdata->local->sta_mtx);
3218                 ret = false;
3219                 goto out;
3220         }
3221
3222         /*
3223          * If AP doesn't support HT, or it doesn't have HE mandatory IEs, mark
3224          * HE as disabled. If on the 5GHz band, make sure it supports VHT.
3225          */
3226         if (ifmgd->flags & IEEE80211_STA_DISABLE_HT ||
3227             (sband->band == NL80211_BAND_5GHZ &&
3228              ifmgd->flags & IEEE80211_STA_DISABLE_VHT) ||
3229             (!elems.he_cap && !elems.he_operation))
3230                 ifmgd->flags |= IEEE80211_STA_DISABLE_HE;
3231
3232         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HE) &&
3233             (!elems.he_cap || !elems.he_operation)) {
3234                 mutex_unlock(&sdata->local->sta_mtx);
3235                 sdata_info(sdata,
3236                            "HE AP is missing HE capability/operation\n");
3237                 ret = false;
3238                 goto out;
3239         }
3240
3241         /* Set up internal HT/VHT capabilities */
3242         if (elems.ht_cap_elem && !(ifmgd->flags & IEEE80211_STA_DISABLE_HT))
3243                 ieee80211_ht_cap_ie_to_sta_ht_cap(sdata, sband,
3244                                                   elems.ht_cap_elem, sta);
3245
3246         if (elems.vht_cap_elem && !(ifmgd->flags & IEEE80211_STA_DISABLE_VHT))
3247                 ieee80211_vht_cap_ie_to_sta_vht_cap(sdata, sband,
3248                                                     elems.vht_cap_elem, sta);
3249
3250         if (elems.he_operation && !(ifmgd->flags & IEEE80211_STA_DISABLE_HE) &&
3251             elems.he_cap) {
3252                 ieee80211_he_cap_ie_to_sta_he_cap(sdata, sband,
3253                                                   elems.he_cap,
3254                                                   elems.he_cap_len,
3255                                                   sta);
3256
3257                 bss_conf->he_support = sta->sta.he_cap.has_he;
3258         } else {
3259                 bss_conf->he_support = false;
3260         }
3261
3262         if (bss_conf->he_support) {
3263                 bss_conf->bss_color =
3264                         le32_get_bits(elems.he_operation->he_oper_params,
3265                                       IEEE80211_HE_OPERATION_BSS_COLOR_MASK);
3266
3267                 bss_conf->htc_trig_based_pkt_ext =
3268                         le32_get_bits(elems.he_operation->he_oper_params,
3269                               IEEE80211_HE_OPERATION_DFLT_PE_DURATION_MASK);
3270                 bss_conf->frame_time_rts_th =
3271                         le32_get_bits(elems.he_operation->he_oper_params,
3272                               IEEE80211_HE_OPERATION_RTS_THRESHOLD_MASK);
3273
3274                 bss_conf->multi_sta_back_32bit =
3275                         sta->sta.he_cap.he_cap_elem.mac_cap_info[2] &
3276                         IEEE80211_HE_MAC_CAP2_32BIT_BA_BITMAP;
3277
3278                 bss_conf->ack_enabled =
3279                         sta->sta.he_cap.he_cap_elem.mac_cap_info[2] &
3280                         IEEE80211_HE_MAC_CAP2_ACK_EN;
3281
3282                 bss_conf->uora_exists = !!elems.uora_element;
3283                 if (elems.uora_element)
3284                         bss_conf->uora_ocw_range = elems.uora_element[0];
3285
3286                 /* TODO: OPEN: what happens if BSS color disable is set? */
3287         }
3288
3289         /*
3290          * Some APs, e.g. Netgear WNDR3700, report invalid HT operation data
3291          * in their association response, so ignore that data for our own
3292          * configuration. If it changed since the last beacon, we'll get the
3293          * next beacon and update then.
3294          */
3295
3296         /*
3297          * If an operating mode notification IE is present, override the
3298          * NSS calculation (that would be done in rate_control_rate_init())
3299          * and use the # of streams from that element.
3300          */
3301         if (elems.opmode_notif &&
3302             !(*elems.opmode_notif & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)) {
3303                 u8 nss;
3304
3305                 nss = *elems.opmode_notif & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
3306                 nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
3307                 nss += 1;
3308                 sta->sta.rx_nss = nss;
3309         }
3310
3311         rate_control_rate_init(sta);
3312
3313         if (ifmgd->flags & IEEE80211_STA_MFP_ENABLED) {
3314                 set_sta_flag(sta, WLAN_STA_MFP);
3315                 sta->sta.mfp = true;
3316         } else {
3317                 sta->sta.mfp = false;
3318         }
3319
3320         sta->sta.wme = elems.wmm_param && local->hw.queues >= IEEE80211_NUM_ACS;
3321
3322         err = sta_info_move_state(sta, IEEE80211_STA_ASSOC);
3323         if (!err && !(ifmgd->flags & IEEE80211_STA_CONTROL_PORT))
3324                 err = sta_info_move_state(sta, IEEE80211_STA_AUTHORIZED);
3325         if (err) {
3326                 sdata_info(sdata,
3327                            "failed to move station %pM to desired state\n",
3328                            sta->sta.addr);
3329                 WARN_ON(__sta_info_destroy(sta));
3330                 mutex_unlock(&sdata->local->sta_mtx);
3331                 ret = false;
3332                 goto out;
3333         }
3334
3335         mutex_unlock(&sdata->local->sta_mtx);
3336
3337         /*
3338          * Always handle WMM once after association regardless
3339          * of the first value the AP uses. Setting -1 here has
3340          * that effect because the AP values is an unsigned
3341          * 4-bit value.
3342          */
3343         ifmgd->wmm_last_param_set = -1;
3344
3345         if (ifmgd->flags & IEEE80211_STA_DISABLE_WMM) {
3346                 ieee80211_set_wmm_default(sdata, false, false);
3347         } else if (!ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
3348                                              elems.wmm_param_len,
3349                                              elems.mu_edca_param_set)) {
3350                 /* still enable QoS since we might have HT/VHT */
3351                 ieee80211_set_wmm_default(sdata, false, true);
3352                 /* set the disable-WMM flag in this case to disable
3353                  * tracking WMM parameter changes in the beacon if
3354                  * the parameters weren't actually valid. Doing so
3355                  * avoids changing parameters very strangely when
3356                  * the AP is going back and forth between valid and
3357                  * invalid parameters.
3358                  */
3359                 ifmgd->flags |= IEEE80211_STA_DISABLE_WMM;
3360         }
3361         changed |= BSS_CHANGED_QOS;
3362
3363         if (elems.max_idle_period_ie) {
3364                 bss_conf->max_idle_period =
3365                         le16_to_cpu(elems.max_idle_period_ie->max_idle_period);
3366                 bss_conf->protected_keep_alive =
3367                         !!(elems.max_idle_period_ie->idle_options &
3368                            WLAN_IDLE_OPTIONS_PROTECTED_KEEP_ALIVE);
3369                 changed |= BSS_CHANGED_KEEP_ALIVE;
3370         } else {
3371                 bss_conf->max_idle_period = 0;
3372                 bss_conf->protected_keep_alive = false;
3373         }
3374
3375         /* set AID and assoc capability,
3376          * ieee80211_set_associated() will tell the driver */
3377         bss_conf->aid = aid;
3378         bss_conf->assoc_capability = capab_info;
3379         ieee80211_set_associated(sdata, cbss, changed);
3380
3381         /*
3382          * If we're using 4-addr mode, let the AP know that we're
3383          * doing so, so that it can create the STA VLAN on its side
3384          */
3385         if (ifmgd->use_4addr)
3386                 ieee80211_send_4addr_nullfunc(local, sdata);
3387
3388         /*
3389          * Start timer to probe the connection to the AP now.
3390          * Also start the timer that will detect beacon loss.
3391          */
3392         ieee80211_sta_rx_notify(sdata, (struct ieee80211_hdr *)mgmt);
3393         ieee80211_sta_reset_beacon_monitor(sdata);
3394
3395         ret = true;
3396  out:
3397         kfree(bss_ies);
3398         return ret;
3399 }
3400
3401 static void ieee80211_rx_mgmt_assoc_resp(struct ieee80211_sub_if_data *sdata,
3402                                          struct ieee80211_mgmt *mgmt,
3403                                          size_t len)
3404 {
3405         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
3406         struct ieee80211_mgd_assoc_data *assoc_data = ifmgd->assoc_data;
3407         u16 capab_info, status_code, aid;
3408         struct ieee802_11_elems elems;
3409         int ac, uapsd_queues = -1;
3410         u8 *pos;
3411         bool reassoc;
3412         struct cfg80211_bss *bss;
3413         struct ieee80211_event event = {
3414                 .type = MLME_EVENT,
3415                 .u.mlme.data = ASSOC_EVENT,
3416         };
3417
3418         sdata_assert_lock(sdata);
3419
3420         if (!assoc_data)
3421                 return;
3422         if (!ether_addr_equal(assoc_data->bss->bssid, mgmt->bssid))
3423                 return;
3424
3425         /*
3426          * AssocResp and ReassocResp have identical structure, so process both
3427          * of them in this function.
3428          */
3429
3430         if (len < 24 + 6)
3431                 return;
3432
3433         reassoc = ieee80211_is_reassoc_resp(mgmt->frame_control);
3434         capab_info = le16_to_cpu(mgmt->u.assoc_resp.capab_info);
3435         status_code = le16_to_cpu(mgmt->u.assoc_resp.status_code);
3436         aid = le16_to_cpu(mgmt->u.assoc_resp.aid);
3437
3438         sdata_info(sdata,
3439                    "RX %sssocResp from %pM (capab=0x%x status=%d aid=%d)\n",
3440                    reassoc ? "Rea" : "A", mgmt->sa,
3441                    capab_info, status_code, (u16)(aid & ~(BIT(15) | BIT(14))));
3442
3443         if (assoc_data->fils_kek_len &&
3444             fils_decrypt_assoc_resp(sdata, (u8 *)mgmt, &len, assoc_data) < 0)
3445                 return;
3446
3447         pos = mgmt->u.assoc_resp.variable;
3448         ieee802_11_parse_elems(pos, len - (pos - (u8 *) mgmt), false, &elems);
3449
3450         if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY &&
3451             elems.timeout_int &&
3452             elems.timeout_int->type == WLAN_TIMEOUT_ASSOC_COMEBACK) {
3453                 u32 tu, ms;
3454                 tu = le32_to_cpu(elems.timeout_int->value);
3455                 ms = tu * 1024 / 1000;
3456                 sdata_info(sdata,
3457                            "%pM rejected association temporarily; comeback duration %u TU (%u ms)\n",
3458                            mgmt->sa, tu, ms);
3459                 assoc_data->timeout = jiffies + msecs_to_jiffies(ms);
3460                 assoc_data->timeout_started = true;
3461                 if (ms > IEEE80211_ASSOC_TIMEOUT)
3462                         run_again(sdata, assoc_data->timeout);
3463                 return;
3464         }
3465
3466         bss = assoc_data->bss;
3467
3468         if (status_code != WLAN_STATUS_SUCCESS) {
3469                 sdata_info(sdata, "%pM denied association (code=%d)\n",
3470                            mgmt->sa, status_code);
3471                 ieee80211_destroy_assoc_data(sdata, false, false);
3472                 event.u.mlme.status = MLME_DENIED;
3473                 event.u.mlme.reason = status_code;
3474                 drv_event_callback(sdata->local, sdata, &event);
3475         } else {
3476                 if (!ieee80211_assoc_success(sdata, bss, mgmt, len)) {
3477                         /* oops -- internal error -- send timeout for now */
3478                         ieee80211_destroy_assoc_data(sdata, false, false);
3479                         cfg80211_assoc_timeout(sdata->dev, bss);
3480                         return;
3481                 }
3482                 event.u.mlme.status = MLME_SUCCESS;
3483                 drv_event_callback(sdata->local, sdata, &event);
3484                 sdata_info(sdata, "associated\n");
3485
3486                 /*
3487                  * destroy assoc_data afterwards, as otherwise an idle
3488                  * recalc after assoc_data is NULL but before associated
3489                  * is set can cause the interface to go idle
3490                  */
3491                 ieee80211_destroy_assoc_data(sdata, true, false);
3492
3493                 /* get uapsd queues configuration */
3494                 uapsd_queues = 0;
3495                 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
3496                         if (sdata->tx_conf[ac].uapsd)
3497                                 uapsd_queues |= ieee80211_ac_to_qos_mask[ac];
3498         }
3499
3500         cfg80211_rx_assoc_resp(sdata->dev, bss, (u8 *)mgmt, len, uapsd_queues);
3501 }
3502
3503 static void ieee80211_rx_bss_info(struct ieee80211_sub_if_data *sdata,
3504                                   struct ieee80211_mgmt *mgmt, size_t len,
3505                                   struct ieee80211_rx_status *rx_status,
3506                                   struct ieee802_11_elems *elems)
3507 {
3508         struct ieee80211_local *local = sdata->local;
3509         struct ieee80211_bss *bss;
3510         struct ieee80211_channel *channel;
3511
3512         sdata_assert_lock(sdata);
3513
3514         channel = ieee80211_get_channel(local->hw.wiphy, rx_status->freq);
3515         if (!channel)
3516                 return;
3517
3518         bss = ieee80211_bss_info_update(local, rx_status, mgmt, len, elems,
3519                                         channel);
3520         if (bss) {
3521                 sdata->vif.bss_conf.beacon_rate = bss->beacon_rate;
3522                 ieee80211_rx_bss_put(local, bss);
3523         }
3524 }
3525
3526
3527 static void ieee80211_rx_mgmt_probe_resp(struct ieee80211_sub_if_data *sdata,
3528                                          struct sk_buff *skb)
3529 {
3530         struct ieee80211_mgmt *mgmt = (void *)skb->data;
3531         struct ieee80211_if_managed *ifmgd;
3532         struct ieee80211_rx_status *rx_status = (void *) skb->cb;
3533         size_t baselen, len = skb->len;
3534         struct ieee802_11_elems elems;
3535
3536         ifmgd = &sdata->u.mgd;
3537
3538         sdata_assert_lock(sdata);
3539
3540         if (!ether_addr_equal(mgmt->da, sdata->vif.addr))
3541                 return; /* ignore ProbeResp to foreign address */
3542
3543         baselen = (u8 *) mgmt->u.probe_resp.variable - (u8 *) mgmt;
3544         if (baselen > len)
3545                 return;
3546
3547         ieee802_11_parse_elems(mgmt->u.probe_resp.variable, len - baselen,
3548                                false, &elems);
3549
3550         ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems);
3551
3552         if (ifmgd->associated &&
3553             ether_addr_equal(mgmt->bssid, ifmgd->associated->bssid))
3554                 ieee80211_reset_ap_probe(sdata);
3555 }
3556
3557 /*
3558  * This is the canonical list of information elements we care about,
3559  * the filter code also gives us all changes to the Microsoft OUI
3560  * (00:50:F2) vendor IE which is used for WMM which we need to track,
3561  * as well as the DTPC IE (part of the Cisco OUI) used for signaling
3562  * changes to requested client power.
3563  *
3564  * We implement beacon filtering in software since that means we can
3565  * avoid processing the frame here and in cfg80211, and userspace
3566  * will not be able to tell whether the hardware supports it or not.
3567  *
3568  * XXX: This list needs to be dynamic -- userspace needs to be able to
3569  *      add items it requires. It also needs to be able to tell us to
3570  *      look out for other vendor IEs.
3571  */
3572 static const u64 care_about_ies =
3573         (1ULL << WLAN_EID_COUNTRY) |
3574         (1ULL << WLAN_EID_ERP_INFO) |
3575         (1ULL << WLAN_EID_CHANNEL_SWITCH) |
3576         (1ULL << WLAN_EID_PWR_CONSTRAINT) |
3577         (1ULL << WLAN_EID_HT_CAPABILITY) |
3578         (1ULL << WLAN_EID_HT_OPERATION) |
3579         (1ULL << WLAN_EID_EXT_CHANSWITCH_ANN);
3580
3581 static void ieee80211_handle_beacon_sig(struct ieee80211_sub_if_data *sdata,
3582                                         struct ieee80211_if_managed *ifmgd,
3583                                         struct ieee80211_bss_conf *bss_conf,
3584                                         struct ieee80211_local *local,
3585                                         struct ieee80211_rx_status *rx_status)
3586 {
3587         /* Track average RSSI from the Beacon frames of the current AP */
3588
3589         if (ifmgd->flags & IEEE80211_STA_RESET_SIGNAL_AVE) {
3590                 ifmgd->flags &= ~IEEE80211_STA_RESET_SIGNAL_AVE;
3591                 ewma_beacon_signal_init(&ifmgd->ave_beacon_signal);
3592                 ifmgd->last_cqm_event_signal = 0;
3593                 ifmgd->count_beacon_signal = 1;
3594                 ifmgd->last_ave_beacon_signal = 0;
3595         } else {
3596                 ifmgd->count_beacon_signal++;
3597         }
3598
3599         ewma_beacon_signal_add(&ifmgd->ave_beacon_signal, -rx_status->signal);
3600
3601         if (ifmgd->rssi_min_thold != ifmgd->rssi_max_thold &&
3602             ifmgd->count_beacon_signal >= IEEE80211_SIGNAL_AVE_MIN_COUNT) {
3603                 int sig = -ewma_beacon_signal_read(&ifmgd->ave_beacon_signal);
3604                 int last_sig = ifmgd->last_ave_beacon_signal;
3605                 struct ieee80211_event event = {
3606                         .type = RSSI_EVENT,
3607                 };
3608
3609                 /*
3610                  * if signal crosses either of the boundaries, invoke callback
3611                  * with appropriate parameters
3612                  */
3613                 if (sig > ifmgd->rssi_max_thold &&
3614                     (last_sig <= ifmgd->rssi_min_thold || last_sig == 0)) {
3615                         ifmgd->last_ave_beacon_signal = sig;
3616                         event.u.rssi.data = RSSI_EVENT_HIGH;
3617                         drv_event_callback(local, sdata, &event);
3618                 } else if (sig < ifmgd->rssi_min_thold &&
3619                            (last_sig >= ifmgd->rssi_max_thold ||
3620                            last_sig == 0)) {
3621                         ifmgd->last_ave_beacon_signal = sig;
3622                         event.u.rssi.data = RSSI_EVENT_LOW;
3623                         drv_event_callback(local, sdata, &event);
3624                 }
3625         }
3626
3627         if (bss_conf->cqm_rssi_thold &&
3628             ifmgd->count_beacon_signal >= IEEE80211_SIGNAL_AVE_MIN_COUNT &&
3629             !(sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI)) {
3630                 int sig = -ewma_beacon_signal_read(&ifmgd->ave_beacon_signal);
3631                 int last_event = ifmgd->last_cqm_event_signal;
3632                 int thold = bss_conf->cqm_rssi_thold;
3633                 int hyst = bss_conf->cqm_rssi_hyst;
3634
3635                 if (sig < thold &&
3636                     (last_event == 0 || sig < last_event - hyst)) {
3637                         ifmgd->last_cqm_event_signal = sig;
3638                         ieee80211_cqm_rssi_notify(
3639                                 &sdata->vif,
3640                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW,
3641                                 sig, GFP_KERNEL);
3642                 } else if (sig > thold &&
3643                            (last_event == 0 || sig > last_event + hyst)) {
3644                         ifmgd->last_cqm_event_signal = sig;
3645                         ieee80211_cqm_rssi_notify(
3646                                 &sdata->vif,
3647                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH,
3648                                 sig, GFP_KERNEL);
3649                 }
3650         }
3651
3652         if (bss_conf->cqm_rssi_low &&
3653             ifmgd->count_beacon_signal >= IEEE80211_SIGNAL_AVE_MIN_COUNT) {
3654                 int sig = -ewma_beacon_signal_read(&ifmgd->ave_beacon_signal);
3655                 int last_event = ifmgd->last_cqm_event_signal;
3656                 int low = bss_conf->cqm_rssi_low;
3657                 int high = bss_conf->cqm_rssi_high;
3658
3659                 if (sig < low &&
3660                     (last_event == 0 || last_event >= low)) {
3661                         ifmgd->last_cqm_event_signal = sig;
3662                         ieee80211_cqm_rssi_notify(
3663                                 &sdata->vif,
3664                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW,
3665                                 sig, GFP_KERNEL);
3666                 } else if (sig > high &&
3667                            (last_event == 0 || last_event <= high)) {
3668                         ifmgd->last_cqm_event_signal = sig;
3669                         ieee80211_cqm_rssi_notify(
3670                                 &sdata->vif,
3671                                 NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH,
3672                                 sig, GFP_KERNEL);
3673                 }
3674         }
3675 }
3676
3677 static void ieee80211_rx_mgmt_beacon(struct ieee80211_sub_if_data *sdata,
3678                                      struct ieee80211_mgmt *mgmt, size_t len,
3679                                      struct ieee80211_rx_status *rx_status)
3680 {
3681         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
3682         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
3683         size_t baselen;
3684         struct ieee802_11_elems elems;
3685         struct ieee80211_local *local = sdata->local;
3686         struct ieee80211_chanctx_conf *chanctx_conf;
3687         struct ieee80211_channel *chan;
3688         struct sta_info *sta;
3689         u32 changed = 0;
3690         bool erp_valid;
3691         u8 erp_value = 0;
3692         u32 ncrc;
3693         u8 *bssid;
3694         u8 deauth_buf[IEEE80211_DEAUTH_FRAME_LEN];
3695
3696         sdata_assert_lock(sdata);
3697
3698         /* Process beacon from the current BSS */
3699         baselen = (u8 *) mgmt->u.beacon.variable - (u8 *) mgmt;
3700         if (baselen > len)
3701                 return;
3702
3703         rcu_read_lock();
3704         chanctx_conf = rcu_dereference(sdata->vif.chanctx_conf);
3705         if (!chanctx_conf) {
3706                 rcu_read_unlock();
3707                 return;
3708         }
3709
3710         if (rx_status->freq != chanctx_conf->def.chan->center_freq) {
3711                 rcu_read_unlock();
3712                 return;
3713         }
3714         chan = chanctx_conf->def.chan;
3715         rcu_read_unlock();
3716
3717         if (ifmgd->assoc_data && ifmgd->assoc_data->need_beacon &&
3718             ether_addr_equal(mgmt->bssid, ifmgd->assoc_data->bss->bssid)) {
3719                 ieee802_11_parse_elems(mgmt->u.beacon.variable,
3720                                        len - baselen, false, &elems);
3721
3722                 ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems);
3723                 if (elems.tim && !elems.parse_error) {
3724                         const struct ieee80211_tim_ie *tim_ie = elems.tim;
3725                         ifmgd->dtim_period = tim_ie->dtim_period;
3726                 }
3727                 ifmgd->have_beacon = true;
3728                 ifmgd->assoc_data->need_beacon = false;
3729                 if (ieee80211_hw_check(&local->hw, TIMING_BEACON_ONLY)) {
3730                         sdata->vif.bss_conf.sync_tsf =
3731                                 le64_to_cpu(mgmt->u.beacon.timestamp);
3732                         sdata->vif.bss_conf.sync_device_ts =
3733                                 rx_status->device_timestamp;
3734                         if (elems.tim)
3735                                 sdata->vif.bss_conf.sync_dtim_count =
3736                                         elems.tim->dtim_count;
3737                         else
3738                                 sdata->vif.bss_conf.sync_dtim_count = 0;
3739                 }
3740                 /* continue assoc process */
3741                 ifmgd->assoc_data->timeout = jiffies;
3742                 ifmgd->assoc_data->timeout_started = true;
3743                 run_again(sdata, ifmgd->assoc_data->timeout);
3744                 return;
3745         }
3746
3747         if (!ifmgd->associated ||
3748             !ether_addr_equal(mgmt->bssid, ifmgd->associated->bssid))
3749                 return;
3750         bssid = ifmgd->associated->bssid;
3751
3752         if (!(rx_status->flag & RX_FLAG_NO_SIGNAL_VAL))
3753                 ieee80211_handle_beacon_sig(sdata, ifmgd, bss_conf,
3754                                             local, rx_status);
3755
3756         if (ifmgd->flags & IEEE80211_STA_CONNECTION_POLL) {
3757                 mlme_dbg_ratelimited(sdata,
3758                                      "cancelling AP probe due to a received beacon\n");
3759                 ieee80211_reset_ap_probe(sdata);
3760         }
3761
3762         /*
3763          * Push the beacon loss detection into the future since
3764          * we are processing a beacon from the AP just now.
3765          */
3766         ieee80211_sta_reset_beacon_monitor(sdata);
3767
3768         ncrc = crc32_be(0, (void *)&mgmt->u.beacon.beacon_int, 4);
3769         ncrc = ieee802_11_parse_elems_crc(mgmt->u.beacon.variable,
3770                                           len - baselen, false, &elems,
3771                                           care_about_ies, ncrc);
3772
3773         if (ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK) &&
3774             ieee80211_check_tim(elems.tim, elems.tim_len, ifmgd->aid)) {
3775                 if (local->hw.conf.dynamic_ps_timeout > 0) {
3776                         if (local->hw.conf.flags & IEEE80211_CONF_PS) {
3777                                 local->hw.conf.flags &= ~IEEE80211_CONF_PS;
3778                                 ieee80211_hw_config(local,
3779                                                     IEEE80211_CONF_CHANGE_PS);
3780                         }
3781                         ieee80211_send_nullfunc(local, sdata, false);
3782                 } else if (!local->pspolling && sdata->u.mgd.powersave) {
3783                         local->pspolling = true;
3784
3785                         /*
3786                          * Here is assumed that the driver will be
3787                          * able to send ps-poll frame and receive a
3788                          * response even though power save mode is
3789                          * enabled, but some drivers might require
3790                          * to disable power save here. This needs
3791                          * to be investigated.
3792                          */
3793                         ieee80211_send_pspoll(local, sdata);
3794                 }
3795         }
3796
3797         if (sdata->vif.p2p ||
3798             sdata->vif.driver_flags & IEEE80211_VIF_GET_NOA_UPDATE) {
3799                 struct ieee80211_p2p_noa_attr noa = {};
3800                 int ret;
3801
3802                 ret = cfg80211_get_p2p_attr(mgmt->u.beacon.variable,
3803                                             len - baselen,
3804                                             IEEE80211_P2P_ATTR_ABSENCE_NOTICE,
3805                                             (u8 *) &noa, sizeof(noa));
3806                 if (ret >= 2) {
3807                         if (sdata->u.mgd.p2p_noa_index != noa.index) {
3808                                 /* valid noa_attr and index changed */
3809                                 sdata->u.mgd.p2p_noa_index = noa.index;
3810                                 memcpy(&bss_conf->p2p_noa_attr, &noa, sizeof(noa));
3811                                 changed |= BSS_CHANGED_P2P_PS;
3812                                 /*
3813                                  * make sure we update all information, the CRC
3814                                  * mechanism doesn't look at P2P attributes.
3815                                  */
3816                                 ifmgd->beacon_crc_valid = false;
3817                         }
3818                 } else if (sdata->u.mgd.p2p_noa_index != -1) {
3819                         /* noa_attr not found and we had valid noa_attr before */
3820                         sdata->u.mgd.p2p_noa_index = -1;
3821                         memset(&bss_conf->p2p_noa_attr, 0, sizeof(bss_conf->p2p_noa_attr));
3822                         changed |= BSS_CHANGED_P2P_PS;
3823                         ifmgd->beacon_crc_valid = false;
3824                 }
3825         }
3826
3827         if (ifmgd->csa_waiting_bcn)
3828                 ieee80211_chswitch_post_beacon(sdata);
3829
3830         /*
3831          * Update beacon timing and dtim count on every beacon appearance. This
3832          * will allow the driver to use the most updated values. Do it before
3833          * comparing this one with last received beacon.
3834          * IMPORTANT: These parameters would possibly be out of sync by the time
3835          * the driver will use them. The synchronized view is currently
3836          * guaranteed only in certain callbacks.
3837          */
3838         if (ieee80211_hw_check(&local->hw, TIMING_BEACON_ONLY)) {
3839                 sdata->vif.bss_conf.sync_tsf =
3840                         le64_to_cpu(mgmt->u.beacon.timestamp);
3841                 sdata->vif.bss_conf.sync_device_ts =
3842                         rx_status->device_timestamp;
3843                 if (elems.tim)
3844                         sdata->vif.bss_conf.sync_dtim_count =
3845                                 elems.tim->dtim_count;
3846                 else
3847                         sdata->vif.bss_conf.sync_dtim_count = 0;
3848         }
3849
3850         if (ncrc == ifmgd->beacon_crc && ifmgd->beacon_crc_valid)
3851                 return;
3852         ifmgd->beacon_crc = ncrc;
3853         ifmgd->beacon_crc_valid = true;
3854
3855         ieee80211_rx_bss_info(sdata, mgmt, len, rx_status, &elems);
3856
3857         ieee80211_sta_process_chanswitch(sdata, rx_status->mactime,
3858                                          rx_status->device_timestamp,
3859                                          &elems, true);
3860
3861         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_WMM) &&
3862             ieee80211_sta_wmm_params(local, sdata, elems.wmm_param,
3863                                      elems.wmm_param_len,
3864                                      elems.mu_edca_param_set))
3865                 changed |= BSS_CHANGED_QOS;
3866
3867         /*
3868          * If we haven't had a beacon before, tell the driver about the
3869          * DTIM period (and beacon timing if desired) now.
3870          */
3871         if (!ifmgd->have_beacon) {
3872                 /* a few bogus AP send dtim_period = 0 or no TIM IE */
3873                 if (elems.tim)
3874                         bss_conf->dtim_period = elems.tim->dtim_period ?: 1;
3875                 else
3876                         bss_conf->dtim_period = 1;
3877
3878                 changed |= BSS_CHANGED_BEACON_INFO;
3879                 ifmgd->have_beacon = true;
3880
3881                 mutex_lock(&local->iflist_mtx);
3882                 ieee80211_recalc_ps(local);
3883                 mutex_unlock(&local->iflist_mtx);
3884
3885                 ieee80211_recalc_ps_vif(sdata);
3886         }
3887
3888         if (elems.erp_info) {
3889                 erp_valid = true;
3890                 erp_value = elems.erp_info[0];
3891         } else {
3892                 erp_valid = false;
3893         }
3894         changed |= ieee80211_handle_bss_capability(sdata,
3895                         le16_to_cpu(mgmt->u.beacon.capab_info),
3896                         erp_valid, erp_value);
3897
3898         mutex_lock(&local->sta_mtx);
3899         sta = sta_info_get(sdata, bssid);
3900
3901         if (ieee80211_config_bw(sdata, sta,
3902                                 elems.ht_cap_elem, elems.ht_operation,
3903                                 elems.vht_operation, elems.he_operation,
3904                                 bssid, &changed)) {
3905                 mutex_unlock(&local->sta_mtx);
3906                 sdata_info(sdata,
3907                            "failed to follow AP %pM bandwidth change, disconnect\n",
3908                            bssid);
3909                 ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH,
3910                                        WLAN_REASON_DEAUTH_LEAVING,
3911                                        true, deauth_buf);
3912                 ieee80211_report_disconnect(sdata, deauth_buf,
3913                                             sizeof(deauth_buf), true,
3914                                             WLAN_REASON_DEAUTH_LEAVING);
3915                 return;
3916         }
3917
3918         if (sta && elems.opmode_notif)
3919                 ieee80211_vht_handle_opmode(sdata, sta, *elems.opmode_notif,
3920                                             rx_status->band);
3921         mutex_unlock(&local->sta_mtx);
3922
3923         changed |= ieee80211_handle_pwr_constr(sdata, chan, mgmt,
3924                                                elems.country_elem,
3925                                                elems.country_elem_len,
3926                                                elems.pwr_constr_elem,
3927                                                elems.cisco_dtpc_elem);
3928
3929         ieee80211_bss_info_change_notify(sdata, changed);
3930 }
3931
3932 void ieee80211_sta_rx_queued_mgmt(struct ieee80211_sub_if_data *sdata,
3933                                   struct sk_buff *skb)
3934 {
3935         struct ieee80211_rx_status *rx_status;
3936         struct ieee80211_mgmt *mgmt;
3937         u16 fc;
3938         struct ieee802_11_elems elems;
3939         int ies_len;
3940
3941         rx_status = (struct ieee80211_rx_status *) skb->cb;
3942         mgmt = (struct ieee80211_mgmt *) skb->data;
3943         fc = le16_to_cpu(mgmt->frame_control);
3944
3945         sdata_lock(sdata);
3946
3947         switch (fc & IEEE80211_FCTL_STYPE) {
3948         case IEEE80211_STYPE_BEACON:
3949                 ieee80211_rx_mgmt_beacon(sdata, mgmt, skb->len, rx_status);
3950                 break;
3951         case IEEE80211_STYPE_PROBE_RESP:
3952                 ieee80211_rx_mgmt_probe_resp(sdata, skb);
3953                 break;
3954         case IEEE80211_STYPE_AUTH:
3955                 ieee80211_rx_mgmt_auth(sdata, mgmt, skb->len);
3956                 break;
3957         case IEEE80211_STYPE_DEAUTH:
3958                 ieee80211_rx_mgmt_deauth(sdata, mgmt, skb->len);
3959                 break;
3960         case IEEE80211_STYPE_DISASSOC:
3961                 ieee80211_rx_mgmt_disassoc(sdata, mgmt, skb->len);
3962                 break;
3963         case IEEE80211_STYPE_ASSOC_RESP:
3964         case IEEE80211_STYPE_REASSOC_RESP:
3965                 ieee80211_rx_mgmt_assoc_resp(sdata, mgmt, skb->len);
3966                 break;
3967         case IEEE80211_STYPE_ACTION:
3968                 if (mgmt->u.action.category == WLAN_CATEGORY_SPECTRUM_MGMT) {
3969                         ies_len = skb->len -
3970                                   offsetof(struct ieee80211_mgmt,
3971                                            u.action.u.chan_switch.variable);
3972
3973                         if (ies_len < 0)
3974                                 break;
3975
3976                         ieee802_11_parse_elems(
3977                                 mgmt->u.action.u.chan_switch.variable,
3978                                 ies_len, true, &elems);
3979
3980                         if (elems.parse_error)
3981                                 break;
3982
3983                         ieee80211_sta_process_chanswitch(sdata,
3984                                                  rx_status->mactime,
3985                                                  rx_status->device_timestamp,
3986                                                  &elems, false);
3987                 } else if (mgmt->u.action.category == WLAN_CATEGORY_PUBLIC) {
3988                         ies_len = skb->len -
3989                                   offsetof(struct ieee80211_mgmt,
3990                                            u.action.u.ext_chan_switch.variable);
3991
3992                         if (ies_len < 0)
3993                                 break;
3994
3995                         ieee802_11_parse_elems(
3996                                 mgmt->u.action.u.ext_chan_switch.variable,
3997                                 ies_len, true, &elems);
3998
3999                         if (elems.parse_error)
4000                                 break;
4001
4002                         /* for the handling code pretend this was also an IE */
4003                         elems.ext_chansw_ie =
4004                                 &mgmt->u.action.u.ext_chan_switch.data;
4005
4006                         ieee80211_sta_process_chanswitch(sdata,
4007                                                  rx_status->mactime,
4008                                                  rx_status->device_timestamp,
4009                                                  &elems, false);
4010                 }
4011                 break;
4012         }
4013         sdata_unlock(sdata);
4014 }
4015
4016 static void ieee80211_sta_timer(struct timer_list *t)
4017 {
4018         struct ieee80211_sub_if_data *sdata =
4019                 from_timer(sdata, t, u.mgd.timer);
4020
4021         ieee80211_queue_work(&sdata->local->hw, &sdata->work);
4022 }
4023
4024 static void ieee80211_sta_connection_lost(struct ieee80211_sub_if_data *sdata,
4025                                           u8 *bssid, u8 reason, bool tx)
4026 {
4027         u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
4028
4029         ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH, reason,
4030                                tx, frame_buf);
4031
4032         ieee80211_report_disconnect(sdata, frame_buf, sizeof(frame_buf), true,
4033                                     reason);
4034 }
4035
4036 static int ieee80211_auth(struct ieee80211_sub_if_data *sdata)
4037 {
4038         struct ieee80211_local *local = sdata->local;
4039         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4040         struct ieee80211_mgd_auth_data *auth_data = ifmgd->auth_data;
4041         u32 tx_flags = 0;
4042         u16 trans = 1;
4043         u16 status = 0;
4044         u16 prepare_tx_duration = 0;
4045
4046         sdata_assert_lock(sdata);
4047
4048         if (WARN_ON_ONCE(!auth_data))
4049                 return -EINVAL;
4050
4051         auth_data->tries++;
4052
4053         if (auth_data->tries > IEEE80211_AUTH_MAX_TRIES) {
4054                 sdata_info(sdata, "authentication with %pM timed out\n",
4055                            auth_data->bss->bssid);
4056
4057                 /*
4058                  * Most likely AP is not in the range so remove the
4059                  * bss struct for that AP.
4060                  */
4061                 cfg80211_unlink_bss(local->hw.wiphy, auth_data->bss);
4062
4063                 return -ETIMEDOUT;
4064         }
4065
4066         if (auth_data->algorithm == WLAN_AUTH_SAE)
4067                 prepare_tx_duration =
4068                         jiffies_to_msecs(IEEE80211_AUTH_TIMEOUT_SAE);
4069
4070         drv_mgd_prepare_tx(local, sdata, prepare_tx_duration);
4071
4072         sdata_info(sdata, "send auth to %pM (try %d/%d)\n",
4073                    auth_data->bss->bssid, auth_data->tries,
4074                    IEEE80211_AUTH_MAX_TRIES);
4075
4076         auth_data->expected_transaction = 2;
4077
4078         if (auth_data->algorithm == WLAN_AUTH_SAE) {
4079                 trans = auth_data->sae_trans;
4080                 status = auth_data->sae_status;
4081                 auth_data->expected_transaction = trans;
4082         }
4083
4084         if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
4085                 tx_flags = IEEE80211_TX_CTL_REQ_TX_STATUS |
4086                            IEEE80211_TX_INTFL_MLME_CONN_TX;
4087
4088         ieee80211_send_auth(sdata, trans, auth_data->algorithm, status,
4089                             auth_data->data, auth_data->data_len,
4090                             auth_data->bss->bssid,
4091                             auth_data->bss->bssid, NULL, 0, 0,
4092                             tx_flags);
4093
4094         if (tx_flags == 0) {
4095                 if (auth_data->algorithm == WLAN_AUTH_SAE)
4096                         auth_data->timeout = jiffies +
4097                                 IEEE80211_AUTH_TIMEOUT_SAE;
4098                 else
4099                         auth_data->timeout = jiffies + IEEE80211_AUTH_TIMEOUT;
4100         } else {
4101                 auth_data->timeout =
4102                         round_jiffies_up(jiffies + IEEE80211_AUTH_TIMEOUT_LONG);
4103         }
4104
4105         auth_data->timeout_started = true;
4106         run_again(sdata, auth_data->timeout);
4107
4108         return 0;
4109 }
4110
4111 static int ieee80211_do_assoc(struct ieee80211_sub_if_data *sdata)
4112 {
4113         struct ieee80211_mgd_assoc_data *assoc_data = sdata->u.mgd.assoc_data;
4114         struct ieee80211_local *local = sdata->local;
4115
4116         sdata_assert_lock(sdata);
4117
4118         assoc_data->tries++;
4119         if (assoc_data->tries > IEEE80211_ASSOC_MAX_TRIES) {
4120                 sdata_info(sdata, "association with %pM timed out\n",
4121                            assoc_data->bss->bssid);
4122
4123                 /*
4124                  * Most likely AP is not in the range so remove the
4125                  * bss struct for that AP.
4126                  */
4127                 cfg80211_unlink_bss(local->hw.wiphy, assoc_data->bss);
4128
4129                 return -ETIMEDOUT;
4130         }
4131
4132         sdata_info(sdata, "associate with %pM (try %d/%d)\n",
4133                    assoc_data->bss->bssid, assoc_data->tries,
4134                    IEEE80211_ASSOC_MAX_TRIES);
4135         ieee80211_send_assoc(sdata);
4136
4137         if (!ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS)) {
4138                 assoc_data->timeout = jiffies + IEEE80211_ASSOC_TIMEOUT;
4139                 assoc_data->timeout_started = true;
4140                 run_again(sdata, assoc_data->timeout);
4141         } else {
4142                 assoc_data->timeout =
4143                         round_jiffies_up(jiffies +
4144                                          IEEE80211_ASSOC_TIMEOUT_LONG);
4145                 assoc_data->timeout_started = true;
4146                 run_again(sdata, assoc_data->timeout);
4147         }
4148
4149         return 0;
4150 }
4151
4152 void ieee80211_mgd_conn_tx_status(struct ieee80211_sub_if_data *sdata,
4153                                   __le16 fc, bool acked)
4154 {
4155         struct ieee80211_local *local = sdata->local;
4156
4157         sdata->u.mgd.status_fc = fc;
4158         sdata->u.mgd.status_acked = acked;
4159         sdata->u.mgd.status_received = true;
4160
4161         ieee80211_queue_work(&local->hw, &sdata->work);
4162 }
4163
4164 void ieee80211_sta_work(struct ieee80211_sub_if_data *sdata)
4165 {
4166         struct ieee80211_local *local = sdata->local;
4167         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4168
4169         sdata_lock(sdata);
4170
4171         if (ifmgd->status_received) {
4172                 __le16 fc = ifmgd->status_fc;
4173                 bool status_acked = ifmgd->status_acked;
4174
4175                 ifmgd->status_received = false;
4176                 if (ifmgd->auth_data && ieee80211_is_auth(fc)) {
4177                         if (status_acked) {
4178                                 if (ifmgd->auth_data->algorithm ==
4179                                     WLAN_AUTH_SAE)
4180                                         ifmgd->auth_data->timeout =
4181                                                 jiffies +
4182                                                 IEEE80211_AUTH_TIMEOUT_SAE;
4183                                 else
4184                                         ifmgd->auth_data->timeout =
4185                                                 jiffies +
4186                                                 IEEE80211_AUTH_TIMEOUT_SHORT;
4187                                 run_again(sdata, ifmgd->auth_data->timeout);
4188                         } else {
4189                                 ifmgd->auth_data->timeout = jiffies - 1;
4190                         }
4191                         ifmgd->auth_data->timeout_started = true;
4192                 } else if (ifmgd->assoc_data &&
4193                            (ieee80211_is_assoc_req(fc) ||
4194                             ieee80211_is_reassoc_req(fc))) {
4195                         if (status_acked) {
4196                                 ifmgd->assoc_data->timeout =
4197                                         jiffies + IEEE80211_ASSOC_TIMEOUT_SHORT;
4198                                 run_again(sdata, ifmgd->assoc_data->timeout);
4199                         } else {
4200                                 ifmgd->assoc_data->timeout = jiffies - 1;
4201                         }
4202                         ifmgd->assoc_data->timeout_started = true;
4203                 }
4204         }
4205
4206         if (ifmgd->auth_data && ifmgd->auth_data->timeout_started &&
4207             time_after(jiffies, ifmgd->auth_data->timeout)) {
4208                 if (ifmgd->auth_data->done) {
4209                         /*
4210                          * ok ... we waited for assoc but userspace didn't,
4211                          * so let's just kill the auth data
4212                          */
4213                         ieee80211_destroy_auth_data(sdata, false);
4214                 } else if (ieee80211_auth(sdata)) {
4215                         u8 bssid[ETH_ALEN];
4216                         struct ieee80211_event event = {
4217                                 .type = MLME_EVENT,
4218                                 .u.mlme.data = AUTH_EVENT,
4219                                 .u.mlme.status = MLME_TIMEOUT,
4220                         };
4221
4222                         memcpy(bssid, ifmgd->auth_data->bss->bssid, ETH_ALEN);
4223
4224                         ieee80211_destroy_auth_data(sdata, false);
4225
4226                         cfg80211_auth_timeout(sdata->dev, bssid);
4227                         drv_event_callback(sdata->local, sdata, &event);
4228                 }
4229         } else if (ifmgd->auth_data && ifmgd->auth_data->timeout_started)
4230                 run_again(sdata, ifmgd->auth_data->timeout);
4231
4232         if (ifmgd->assoc_data && ifmgd->assoc_data->timeout_started &&
4233             time_after(jiffies, ifmgd->assoc_data->timeout)) {
4234                 if ((ifmgd->assoc_data->need_beacon && !ifmgd->have_beacon) ||
4235                     ieee80211_do_assoc(sdata)) {
4236                         struct cfg80211_bss *bss = ifmgd->assoc_data->bss;
4237                         struct ieee80211_event event = {
4238                                 .type = MLME_EVENT,
4239                                 .u.mlme.data = ASSOC_EVENT,
4240                                 .u.mlme.status = MLME_TIMEOUT,
4241                         };
4242
4243                         ieee80211_destroy_assoc_data(sdata, false, false);
4244                         cfg80211_assoc_timeout(sdata->dev, bss);
4245                         drv_event_callback(sdata->local, sdata, &event);
4246                 }
4247         } else if (ifmgd->assoc_data && ifmgd->assoc_data->timeout_started)
4248                 run_again(sdata, ifmgd->assoc_data->timeout);
4249
4250         if (ifmgd->flags & IEEE80211_STA_CONNECTION_POLL &&
4251             ifmgd->associated) {
4252                 u8 bssid[ETH_ALEN];
4253                 int max_tries;
4254
4255                 memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
4256
4257                 if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS))
4258                         max_tries = max_nullfunc_tries;
4259                 else
4260                         max_tries = max_probe_tries;
4261
4262                 /* ACK received for nullfunc probing frame */
4263                 if (!ifmgd->probe_send_count)
4264                         ieee80211_reset_ap_probe(sdata);
4265                 else if (ifmgd->nullfunc_failed) {
4266                         if (ifmgd->probe_send_count < max_tries) {
4267                                 mlme_dbg(sdata,
4268                                          "No ack for nullfunc frame to AP %pM, try %d/%i\n",
4269                                          bssid, ifmgd->probe_send_count,
4270                                          max_tries);
4271                                 ieee80211_mgd_probe_ap_send(sdata);
4272                         } else {
4273                                 mlme_dbg(sdata,
4274                                          "No ack for nullfunc frame to AP %pM, disconnecting.\n",
4275                                          bssid);
4276                                 ieee80211_sta_connection_lost(sdata, bssid,
4277                                         WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY,
4278                                         false);
4279                         }
4280                 } else if (time_is_after_jiffies(ifmgd->probe_timeout))
4281                         run_again(sdata, ifmgd->probe_timeout);
4282                 else if (ieee80211_hw_check(&local->hw, REPORTS_TX_ACK_STATUS)) {
4283                         mlme_dbg(sdata,
4284                                  "Failed to send nullfunc to AP %pM after %dms, disconnecting\n",
4285                                  bssid, probe_wait_ms);
4286                         ieee80211_sta_connection_lost(sdata, bssid,
4287                                 WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY, false);
4288                 } else if (ifmgd->probe_send_count < max_tries) {
4289                         mlme_dbg(sdata,
4290                                  "No probe response from AP %pM after %dms, try %d/%i\n",
4291                                  bssid, probe_wait_ms,
4292                                  ifmgd->probe_send_count, max_tries);
4293                         ieee80211_mgd_probe_ap_send(sdata);
4294                 } else {
4295                         /*
4296                          * We actually lost the connection ... or did we?
4297                          * Let's make sure!
4298                          */
4299                         mlme_dbg(sdata,
4300                                  "No probe response from AP %pM after %dms, disconnecting.\n",
4301                                  bssid, probe_wait_ms);
4302
4303                         ieee80211_sta_connection_lost(sdata, bssid,
4304                                 WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY, false);
4305                 }
4306         }
4307
4308         sdata_unlock(sdata);
4309 }
4310
4311 static void ieee80211_sta_bcn_mon_timer(struct timer_list *t)
4312 {
4313         struct ieee80211_sub_if_data *sdata =
4314                 from_timer(sdata, t, u.mgd.bcn_mon_timer);
4315         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4316
4317         if (sdata->vif.csa_active && !ifmgd->csa_waiting_bcn)
4318                 return;
4319
4320         sdata->u.mgd.connection_loss = false;
4321         ieee80211_queue_work(&sdata->local->hw,
4322                              &sdata->u.mgd.beacon_connection_loss_work);
4323 }
4324
4325 static void ieee80211_sta_conn_mon_timer(struct timer_list *t)
4326 {
4327         struct ieee80211_sub_if_data *sdata =
4328                 from_timer(sdata, t, u.mgd.conn_mon_timer);
4329         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4330         struct ieee80211_local *local = sdata->local;
4331
4332         if (sdata->vif.csa_active && !ifmgd->csa_waiting_bcn)
4333                 return;
4334
4335         ieee80211_queue_work(&local->hw, &ifmgd->monitor_work);
4336 }
4337
4338 static void ieee80211_sta_monitor_work(struct work_struct *work)
4339 {
4340         struct ieee80211_sub_if_data *sdata =
4341                 container_of(work, struct ieee80211_sub_if_data,
4342                              u.mgd.monitor_work);
4343
4344         ieee80211_mgd_probe_ap(sdata, false);
4345 }
4346
4347 static void ieee80211_restart_sta_timer(struct ieee80211_sub_if_data *sdata)
4348 {
4349         if (sdata->vif.type == NL80211_IFTYPE_STATION) {
4350                 __ieee80211_stop_poll(sdata);
4351
4352                 /* let's probe the connection once */
4353                 if (!ieee80211_hw_check(&sdata->local->hw, CONNECTION_MONITOR))
4354                         ieee80211_queue_work(&sdata->local->hw,
4355                                              &sdata->u.mgd.monitor_work);
4356         }
4357 }
4358
4359 #ifdef CONFIG_PM
4360 void ieee80211_mgd_quiesce(struct ieee80211_sub_if_data *sdata)
4361 {
4362         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4363         u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
4364
4365         sdata_lock(sdata);
4366
4367         if (ifmgd->auth_data || ifmgd->assoc_data) {
4368                 const u8 *bssid = ifmgd->auth_data ?
4369                                 ifmgd->auth_data->bss->bssid :
4370                                 ifmgd->assoc_data->bss->bssid;
4371
4372                 /*
4373                  * If we are trying to authenticate / associate while suspending,
4374                  * cfg80211 won't know and won't actually abort those attempts,
4375                  * thus we need to do that ourselves.
4376                  */
4377                 ieee80211_send_deauth_disassoc(sdata, bssid,
4378                                                IEEE80211_STYPE_DEAUTH,
4379                                                WLAN_REASON_DEAUTH_LEAVING,
4380                                                false, frame_buf);
4381                 if (ifmgd->assoc_data)
4382                         ieee80211_destroy_assoc_data(sdata, false, true);
4383                 if (ifmgd->auth_data)
4384                         ieee80211_destroy_auth_data(sdata, false);
4385                 cfg80211_tx_mlme_mgmt(sdata->dev, frame_buf,
4386                                       IEEE80211_DEAUTH_FRAME_LEN);
4387         }
4388
4389         /* This is a bit of a hack - we should find a better and more generic
4390          * solution to this. Normally when suspending, cfg80211 will in fact
4391          * deauthenticate. However, it doesn't (and cannot) stop an ongoing
4392          * auth (not so important) or assoc (this is the problem) process.
4393          *
4394          * As a consequence, it can happen that we are in the process of both
4395          * associating and suspending, and receive an association response
4396          * after cfg80211 has checked if it needs to disconnect, but before
4397          * we actually set the flag to drop incoming frames. This will then
4398          * cause the workqueue flush to process the association response in
4399          * the suspend, resulting in a successful association just before it
4400          * tries to remove the interface from the driver, which now though
4401          * has a channel context assigned ... this results in issues.
4402          *
4403          * To work around this (for now) simply deauth here again if we're
4404          * now connected.
4405          */
4406         if (ifmgd->associated && !sdata->local->wowlan) {
4407                 u8 bssid[ETH_ALEN];
4408                 struct cfg80211_deauth_request req = {
4409                         .reason_code = WLAN_REASON_DEAUTH_LEAVING,
4410                         .bssid = bssid,
4411                 };
4412
4413                 memcpy(bssid, ifmgd->associated->bssid, ETH_ALEN);
4414                 ieee80211_mgd_deauth(sdata, &req);
4415         }
4416
4417         sdata_unlock(sdata);
4418 }
4419
4420 void ieee80211_sta_restart(struct ieee80211_sub_if_data *sdata)
4421 {
4422         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4423
4424         sdata_lock(sdata);
4425         if (!ifmgd->associated) {
4426                 sdata_unlock(sdata);
4427                 return;
4428         }
4429
4430         if (sdata->flags & IEEE80211_SDATA_DISCONNECT_RESUME) {
4431                 sdata->flags &= ~IEEE80211_SDATA_DISCONNECT_RESUME;
4432                 mlme_dbg(sdata, "driver requested disconnect after resume\n");
4433                 ieee80211_sta_connection_lost(sdata,
4434                                               ifmgd->associated->bssid,
4435                                               WLAN_REASON_UNSPECIFIED,
4436                                               true);
4437                 sdata_unlock(sdata);
4438                 return;
4439         }
4440         sdata_unlock(sdata);
4441 }
4442 #endif
4443
4444 /* interface setup */
4445 void ieee80211_sta_setup_sdata(struct ieee80211_sub_if_data *sdata)
4446 {
4447         struct ieee80211_if_managed *ifmgd;
4448
4449         ifmgd = &sdata->u.mgd;
4450         INIT_WORK(&ifmgd->monitor_work, ieee80211_sta_monitor_work);
4451         INIT_WORK(&ifmgd->chswitch_work, ieee80211_chswitch_work);
4452         INIT_WORK(&ifmgd->beacon_connection_loss_work,
4453                   ieee80211_beacon_connection_loss_work);
4454         INIT_WORK(&ifmgd->csa_connection_drop_work,
4455                   ieee80211_csa_connection_drop_work);
4456         INIT_WORK(&ifmgd->request_smps_work, ieee80211_request_smps_mgd_work);
4457         INIT_DELAYED_WORK(&ifmgd->tdls_peer_del_work,
4458                           ieee80211_tdls_peer_del_work);
4459         timer_setup(&ifmgd->timer, ieee80211_sta_timer, 0);
4460         timer_setup(&ifmgd->bcn_mon_timer, ieee80211_sta_bcn_mon_timer, 0);
4461         timer_setup(&ifmgd->conn_mon_timer, ieee80211_sta_conn_mon_timer, 0);
4462         timer_setup(&ifmgd->chswitch_timer, ieee80211_chswitch_timer, 0);
4463         INIT_DELAYED_WORK(&ifmgd->tx_tspec_wk,
4464                           ieee80211_sta_handle_tspec_ac_params_wk);
4465
4466         ifmgd->flags = 0;
4467         ifmgd->powersave = sdata->wdev.ps;
4468         ifmgd->uapsd_queues = sdata->local->hw.uapsd_queues;
4469         ifmgd->uapsd_max_sp_len = sdata->local->hw.uapsd_max_sp_len;
4470         ifmgd->p2p_noa_index = -1;
4471
4472         if (sdata->local->hw.wiphy->features & NL80211_FEATURE_DYNAMIC_SMPS)
4473                 ifmgd->req_smps = IEEE80211_SMPS_AUTOMATIC;
4474         else
4475                 ifmgd->req_smps = IEEE80211_SMPS_OFF;
4476
4477         /* Setup TDLS data */
4478         spin_lock_init(&ifmgd->teardown_lock);
4479         ifmgd->teardown_skb = NULL;
4480         ifmgd->orig_teardown_skb = NULL;
4481 }
4482
4483 /* scan finished notification */
4484 void ieee80211_mlme_notify_scan_completed(struct ieee80211_local *local)
4485 {
4486         struct ieee80211_sub_if_data *sdata;
4487
4488         /* Restart STA timers */
4489         rcu_read_lock();
4490         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
4491                 if (ieee80211_sdata_running(sdata))
4492                         ieee80211_restart_sta_timer(sdata);
4493         }
4494         rcu_read_unlock();
4495 }
4496
4497 static u8 ieee80211_ht_vht_rx_chains(struct ieee80211_sub_if_data *sdata,
4498                                      struct cfg80211_bss *cbss)
4499 {
4500         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4501         const u8 *ht_cap_ie, *vht_cap_ie;
4502         const struct ieee80211_ht_cap *ht_cap;
4503         const struct ieee80211_vht_cap *vht_cap;
4504         u8 chains = 1;
4505
4506         if (ifmgd->flags & IEEE80211_STA_DISABLE_HT)
4507                 return chains;
4508
4509         ht_cap_ie = ieee80211_bss_get_ie(cbss, WLAN_EID_HT_CAPABILITY);
4510         if (ht_cap_ie && ht_cap_ie[1] >= sizeof(*ht_cap)) {
4511                 ht_cap = (void *)(ht_cap_ie + 2);
4512                 chains = ieee80211_mcs_to_chains(&ht_cap->mcs);
4513                 /*
4514                  * TODO: use "Tx Maximum Number Spatial Streams Supported" and
4515                  *       "Tx Unequal Modulation Supported" fields.
4516                  */
4517         }
4518
4519         if (ifmgd->flags & IEEE80211_STA_DISABLE_VHT)
4520                 return chains;
4521
4522         vht_cap_ie = ieee80211_bss_get_ie(cbss, WLAN_EID_VHT_CAPABILITY);
4523         if (vht_cap_ie && vht_cap_ie[1] >= sizeof(*vht_cap)) {
4524                 u8 nss;
4525                 u16 tx_mcs_map;
4526
4527                 vht_cap = (void *)(vht_cap_ie + 2);
4528                 tx_mcs_map = le16_to_cpu(vht_cap->supp_mcs.tx_mcs_map);
4529                 for (nss = 8; nss > 0; nss--) {
4530                         if (((tx_mcs_map >> (2 * (nss - 1))) & 3) !=
4531                                         IEEE80211_VHT_MCS_NOT_SUPPORTED)
4532                                 break;
4533                 }
4534                 /* TODO: use "Tx Highest Supported Long GI Data Rate" field? */
4535                 chains = max(chains, nss);
4536         }
4537
4538         return chains;
4539 }
4540
4541 static bool
4542 ieee80211_verify_sta_he_mcs_support(struct ieee80211_supported_band *sband,
4543                                     const struct ieee80211_he_operation *he_op)
4544 {
4545         const struct ieee80211_sta_he_cap *sta_he_cap =
4546                 ieee80211_get_he_sta_cap(sband);
4547         u16 ap_min_req_set;
4548         int i;
4549
4550         if (!sta_he_cap || !he_op)
4551                 return false;
4552
4553         ap_min_req_set = le16_to_cpu(he_op->he_mcs_nss_set);
4554
4555         /* Need to go over for 80MHz, 160MHz and for 80+80 */
4556         for (i = 0; i < 3; i++) {
4557                 const struct ieee80211_he_mcs_nss_supp *sta_mcs_nss_supp =
4558                         &sta_he_cap->he_mcs_nss_supp;
4559                 u16 sta_mcs_map_rx =
4560                         le16_to_cpu(((__le16 *)sta_mcs_nss_supp)[2 * i]);
4561                 u16 sta_mcs_map_tx =
4562                         le16_to_cpu(((__le16 *)sta_mcs_nss_supp)[2 * i + 1]);
4563                 u8 nss;
4564                 bool verified = true;
4565
4566                 /*
4567                  * For each band there is a maximum of 8 spatial streams
4568                  * possible. Each of the sta_mcs_map_* is a 16-bit struct built
4569                  * of 2 bits per NSS (1-8), with the values defined in enum
4570                  * ieee80211_he_mcs_support. Need to make sure STA TX and RX
4571                  * capabilities aren't less than the AP's minimum requirements
4572                  * for this HE BSS per SS.
4573                  * It is enough to find one such band that meets the reqs.
4574                  */
4575                 for (nss = 8; nss > 0; nss--) {
4576                         u8 sta_rx_val = (sta_mcs_map_rx >> (2 * (nss - 1))) & 3;
4577                         u8 sta_tx_val = (sta_mcs_map_tx >> (2 * (nss - 1))) & 3;
4578                         u8 ap_val = (ap_min_req_set >> (2 * (nss - 1))) & 3;
4579
4580                         if (ap_val == IEEE80211_HE_MCS_NOT_SUPPORTED)
4581                                 continue;
4582
4583                         /*
4584                          * Make sure the HE AP doesn't require MCSs that aren't
4585                          * supported by the client
4586                          */
4587                         if (sta_rx_val == IEEE80211_HE_MCS_NOT_SUPPORTED ||
4588                             sta_tx_val == IEEE80211_HE_MCS_NOT_SUPPORTED ||
4589                             (ap_val > sta_rx_val) || (ap_val > sta_tx_val)) {
4590                                 verified = false;
4591                                 break;
4592                         }
4593                 }
4594
4595                 if (verified)
4596                         return true;
4597         }
4598
4599         /* If here, STA doesn't meet AP's HE min requirements */
4600         return false;
4601 }
4602
4603 static int ieee80211_prep_channel(struct ieee80211_sub_if_data *sdata,
4604                                   struct cfg80211_bss *cbss)
4605 {
4606         struct ieee80211_local *local = sdata->local;
4607         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4608         const struct ieee80211_ht_cap *ht_cap = NULL;
4609         const struct ieee80211_ht_operation *ht_oper = NULL;
4610         const struct ieee80211_vht_operation *vht_oper = NULL;
4611         const struct ieee80211_he_operation *he_oper = NULL;
4612         struct ieee80211_supported_band *sband;
4613         struct cfg80211_chan_def chandef;
4614         int ret;
4615         u32 i;
4616         bool have_80mhz;
4617
4618         sband = local->hw.wiphy->bands[cbss->channel->band];
4619
4620         ifmgd->flags &= ~(IEEE80211_STA_DISABLE_40MHZ |
4621                           IEEE80211_STA_DISABLE_80P80MHZ |
4622                           IEEE80211_STA_DISABLE_160MHZ);
4623
4624         rcu_read_lock();
4625
4626         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT) &&
4627             sband->ht_cap.ht_supported) {
4628                 const u8 *ht_oper_ie, *ht_cap_ie;
4629
4630                 ht_oper_ie = ieee80211_bss_get_ie(cbss, WLAN_EID_HT_OPERATION);
4631                 if (ht_oper_ie && ht_oper_ie[1] >= sizeof(*ht_oper))
4632                         ht_oper = (void *)(ht_oper_ie + 2);
4633
4634                 ht_cap_ie = ieee80211_bss_get_ie(cbss, WLAN_EID_HT_CAPABILITY);
4635                 if (ht_cap_ie && ht_cap_ie[1] >= sizeof(*ht_cap))
4636                         ht_cap = (void *)(ht_cap_ie + 2);
4637
4638                 if (!ht_cap) {
4639                         ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
4640                         ht_oper = NULL;
4641                 }
4642         }
4643
4644         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT) &&
4645             sband->vht_cap.vht_supported) {
4646                 const u8 *vht_oper_ie, *vht_cap;
4647
4648                 vht_oper_ie = ieee80211_bss_get_ie(cbss,
4649                                                    WLAN_EID_VHT_OPERATION);
4650                 if (vht_oper_ie && vht_oper_ie[1] >= sizeof(*vht_oper))
4651                         vht_oper = (void *)(vht_oper_ie + 2);
4652                 if (vht_oper && !ht_oper) {
4653                         vht_oper = NULL;
4654                         sdata_info(sdata,
4655                                    "AP advertised VHT without HT, disabling both\n");
4656                         ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
4657                         ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
4658                 }
4659
4660                 vht_cap = ieee80211_bss_get_ie(cbss, WLAN_EID_VHT_CAPABILITY);
4661                 if (!vht_cap || vht_cap[1] < sizeof(struct ieee80211_vht_cap)) {
4662                         ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
4663                         vht_oper = NULL;
4664                 }
4665         }
4666
4667         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HE) &&
4668             ieee80211_get_he_sta_cap(sband)) {
4669                 const struct cfg80211_bss_ies *ies;
4670                 const u8 *he_oper_ie;
4671
4672                 ies = rcu_dereference(cbss->ies);
4673                 he_oper_ie = cfg80211_find_ext_ie(WLAN_EID_EXT_HE_OPERATION,
4674                                                   ies->data, ies->len);
4675                 if (he_oper_ie &&
4676                     he_oper_ie[1] == ieee80211_he_oper_size(&he_oper_ie[3]))
4677                         he_oper = (void *)(he_oper_ie + 3);
4678                 else
4679                         he_oper = NULL;
4680
4681                 if (!ieee80211_verify_sta_he_mcs_support(sband, he_oper))
4682                         ifmgd->flags |= IEEE80211_STA_DISABLE_HE;
4683         }
4684
4685         /* Allow VHT if at least one channel on the sband supports 80 MHz */
4686         have_80mhz = false;
4687         for (i = 0; i < sband->n_channels; i++) {
4688                 if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
4689                                                 IEEE80211_CHAN_NO_80MHZ))
4690                         continue;
4691
4692                 have_80mhz = true;
4693                 break;
4694         }
4695
4696         if (!have_80mhz)
4697                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
4698
4699         ifmgd->flags |= ieee80211_determine_chantype(sdata, sband,
4700                                                      cbss->channel,
4701                                                      ht_oper, vht_oper, he_oper,
4702                                                      &chandef, false);
4703
4704         sdata->needed_rx_chains = min(ieee80211_ht_vht_rx_chains(sdata, cbss),
4705                                       local->rx_chains);
4706
4707         rcu_read_unlock();
4708
4709         /* will change later if needed */
4710         sdata->smps_mode = IEEE80211_SMPS_OFF;
4711
4712         mutex_lock(&local->mtx);
4713         /*
4714          * If this fails (possibly due to channel context sharing
4715          * on incompatible channels, e.g. 80+80 and 160 sharing the
4716          * same control channel) try to use a smaller bandwidth.
4717          */
4718         ret = ieee80211_vif_use_channel(sdata, &chandef,
4719                                         IEEE80211_CHANCTX_SHARED);
4720
4721         /* don't downgrade for 5 and 10 MHz channels, though. */
4722         if (chandef.width == NL80211_CHAN_WIDTH_5 ||
4723             chandef.width == NL80211_CHAN_WIDTH_10)
4724                 goto out;
4725
4726         while (ret && chandef.width != NL80211_CHAN_WIDTH_20_NOHT) {
4727                 ifmgd->flags |= ieee80211_chandef_downgrade(&chandef);
4728                 ret = ieee80211_vif_use_channel(sdata, &chandef,
4729                                                 IEEE80211_CHANCTX_SHARED);
4730         }
4731  out:
4732         mutex_unlock(&local->mtx);
4733         return ret;
4734 }
4735
4736 static int ieee80211_prep_connection(struct ieee80211_sub_if_data *sdata,
4737                                      struct cfg80211_bss *cbss, bool assoc,
4738                                      bool override)
4739 {
4740         struct ieee80211_local *local = sdata->local;
4741         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4742         struct ieee80211_bss *bss = (void *)cbss->priv;
4743         struct sta_info *new_sta = NULL;
4744         struct ieee80211_supported_band *sband;
4745         bool have_sta = false;
4746         int err;
4747
4748         sband = local->hw.wiphy->bands[cbss->channel->band];
4749
4750         if (WARN_ON(!ifmgd->auth_data && !ifmgd->assoc_data))
4751                 return -EINVAL;
4752
4753         /* If a reconfig is happening, bail out */
4754         if (local->in_reconfig)
4755                 return -EBUSY;
4756
4757         if (assoc) {
4758                 rcu_read_lock();
4759                 have_sta = sta_info_get(sdata, cbss->bssid);
4760                 rcu_read_unlock();
4761         }
4762
4763         if (!have_sta) {
4764                 new_sta = sta_info_alloc(sdata, cbss->bssid, GFP_KERNEL);
4765                 if (!new_sta)
4766                         return -ENOMEM;
4767         }
4768
4769         /*
4770          * Set up the information for the new channel before setting the
4771          * new channel. We can't - completely race-free - change the basic
4772          * rates bitmap and the channel (sband) that it refers to, but if
4773          * we set it up before we at least avoid calling into the driver's
4774          * bss_info_changed() method with invalid information (since we do
4775          * call that from changing the channel - only for IDLE and perhaps
4776          * some others, but ...).
4777          *
4778          * So to avoid that, just set up all the new information before the
4779          * channel, but tell the driver to apply it only afterwards, since
4780          * it might need the new channel for that.
4781          */
4782         if (new_sta) {
4783                 u32 rates = 0, basic_rates = 0;
4784                 bool have_higher_than_11mbit;
4785                 int min_rate = INT_MAX, min_rate_index = -1;
4786                 const struct cfg80211_bss_ies *ies;
4787                 int shift = ieee80211_vif_get_shift(&sdata->vif);
4788
4789                 ieee80211_get_rates(sband, bss->supp_rates,
4790                                     bss->supp_rates_len,
4791                                     &rates, &basic_rates,
4792                                     &have_higher_than_11mbit,
4793                                     &min_rate, &min_rate_index,
4794                                     shift);
4795
4796                 /*
4797                  * This used to be a workaround for basic rates missing
4798                  * in the association response frame. Now that we no
4799                  * longer use the basic rates from there, it probably
4800                  * doesn't happen any more, but keep the workaround so
4801                  * in case some *other* APs are buggy in different ways
4802                  * we can connect -- with a warning.
4803                  */
4804                 if (!basic_rates && min_rate_index >= 0) {
4805                         sdata_info(sdata,
4806                                    "No basic rates, using min rate instead\n");
4807                         basic_rates = BIT(min_rate_index);
4808                 }
4809
4810                 new_sta->sta.supp_rates[cbss->channel->band] = rates;
4811                 sdata->vif.bss_conf.basic_rates = basic_rates;
4812
4813                 /* cf. IEEE 802.11 9.2.12 */
4814                 if (cbss->channel->band == NL80211_BAND_2GHZ &&
4815                     have_higher_than_11mbit)
4816                         sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
4817                 else
4818                         sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;
4819
4820                 memcpy(ifmgd->bssid, cbss->bssid, ETH_ALEN);
4821
4822                 /* set timing information */
4823                 sdata->vif.bss_conf.beacon_int = cbss->beacon_interval;
4824                 rcu_read_lock();
4825                 ies = rcu_dereference(cbss->beacon_ies);
4826                 if (ies) {
4827                         const u8 *tim_ie;
4828
4829                         sdata->vif.bss_conf.sync_tsf = ies->tsf;
4830                         sdata->vif.bss_conf.sync_device_ts =
4831                                 bss->device_ts_beacon;
4832                         tim_ie = cfg80211_find_ie(WLAN_EID_TIM,
4833                                                   ies->data, ies->len);
4834                         if (tim_ie && tim_ie[1] >= 2)
4835                                 sdata->vif.bss_conf.sync_dtim_count = tim_ie[2];
4836                         else
4837                                 sdata->vif.bss_conf.sync_dtim_count = 0;
4838                 } else if (!ieee80211_hw_check(&sdata->local->hw,
4839                                                TIMING_BEACON_ONLY)) {
4840                         ies = rcu_dereference(cbss->proberesp_ies);
4841                         /* must be non-NULL since beacon IEs were NULL */
4842                         sdata->vif.bss_conf.sync_tsf = ies->tsf;
4843                         sdata->vif.bss_conf.sync_device_ts =
4844                                 bss->device_ts_presp;
4845                         sdata->vif.bss_conf.sync_dtim_count = 0;
4846                 } else {
4847                         sdata->vif.bss_conf.sync_tsf = 0;
4848                         sdata->vif.bss_conf.sync_device_ts = 0;
4849                         sdata->vif.bss_conf.sync_dtim_count = 0;
4850                 }
4851                 rcu_read_unlock();
4852         }
4853
4854         if (new_sta || override) {
4855                 err = ieee80211_prep_channel(sdata, cbss);
4856                 if (err) {
4857                         if (new_sta)
4858                                 sta_info_free(local, new_sta);
4859                         return -EINVAL;
4860                 }
4861         }
4862
4863         if (new_sta) {
4864                 /*
4865                  * tell driver about BSSID, basic rates and timing
4866                  * this was set up above, before setting the channel
4867                  */
4868                 ieee80211_bss_info_change_notify(sdata,
4869                         BSS_CHANGED_BSSID | BSS_CHANGED_BASIC_RATES |
4870                         BSS_CHANGED_BEACON_INT);
4871
4872                 if (assoc)
4873                         sta_info_pre_move_state(new_sta, IEEE80211_STA_AUTH);
4874
4875                 err = sta_info_insert(new_sta);
4876                 new_sta = NULL;
4877                 if (err) {
4878                         sdata_info(sdata,
4879                                    "failed to insert STA entry for the AP (error %d)\n",
4880                                    err);
4881                         return err;
4882                 }
4883         } else
4884                 WARN_ON_ONCE(!ether_addr_equal(ifmgd->bssid, cbss->bssid));
4885
4886         /* Cancel scan to ensure that nothing interferes with connection */
4887         if (local->scanning)
4888                 ieee80211_scan_cancel(local);
4889
4890         return 0;
4891 }
4892
4893 /* config hooks */
4894 int ieee80211_mgd_auth(struct ieee80211_sub_if_data *sdata,
4895                        struct cfg80211_auth_request *req)
4896 {
4897         struct ieee80211_local *local = sdata->local;
4898         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
4899         struct ieee80211_mgd_auth_data *auth_data;
4900         u16 auth_alg;
4901         int err;
4902
4903         /* prepare auth data structure */
4904
4905         switch (req->auth_type) {
4906         case NL80211_AUTHTYPE_OPEN_SYSTEM:
4907                 auth_alg = WLAN_AUTH_OPEN;
4908                 break;
4909         case NL80211_AUTHTYPE_SHARED_KEY:
4910                 if (IS_ERR(local->wep_tx_tfm))
4911                         return -EOPNOTSUPP;
4912                 auth_alg = WLAN_AUTH_SHARED_KEY;
4913                 break;
4914         case NL80211_AUTHTYPE_FT:
4915                 auth_alg = WLAN_AUTH_FT;
4916                 break;
4917         case NL80211_AUTHTYPE_NETWORK_EAP:
4918                 auth_alg = WLAN_AUTH_LEAP;
4919                 break;
4920         case NL80211_AUTHTYPE_SAE:
4921                 auth_alg = WLAN_AUTH_SAE;
4922                 break;
4923         case NL80211_AUTHTYPE_FILS_SK:
4924                 auth_alg = WLAN_AUTH_FILS_SK;
4925                 break;
4926         case NL80211_AUTHTYPE_FILS_SK_PFS:
4927                 auth_alg = WLAN_AUTH_FILS_SK_PFS;
4928                 break;
4929         case NL80211_AUTHTYPE_FILS_PK:
4930                 auth_alg = WLAN_AUTH_FILS_PK;
4931                 break;
4932         default:
4933                 return -EOPNOTSUPP;
4934         }
4935
4936         auth_data = kzalloc(sizeof(*auth_data) + req->auth_data_len +
4937                             req->ie_len, GFP_KERNEL);
4938         if (!auth_data)
4939                 return -ENOMEM;
4940
4941         auth_data->bss = req->bss;
4942
4943         if (req->auth_data_len >= 4) {
4944                 if (req->auth_type == NL80211_AUTHTYPE_SAE) {
4945                         __le16 *pos = (__le16 *) req->auth_data;
4946
4947                         auth_data->sae_trans = le16_to_cpu(pos[0]);
4948                         auth_data->sae_status = le16_to_cpu(pos[1]);
4949                 }
4950                 memcpy(auth_data->data, req->auth_data + 4,
4951                        req->auth_data_len - 4);
4952                 auth_data->data_len += req->auth_data_len - 4;
4953         }
4954
4955         if (req->ie && req->ie_len) {
4956                 memcpy(&auth_data->data[auth_data->data_len],
4957                        req->ie, req->ie_len);
4958                 auth_data->data_len += req->ie_len;
4959         }
4960
4961         if (req->key && req->key_len) {
4962                 auth_data->key_len = req->key_len;
4963                 auth_data->key_idx = req->key_idx;
4964                 memcpy(auth_data->key, req->key, req->key_len);
4965         }
4966
4967         auth_data->algorithm = auth_alg;
4968
4969         /* try to authenticate/probe */
4970
4971         if ((ifmgd->auth_data && !ifmgd->auth_data->done) ||
4972             ifmgd->assoc_data) {
4973                 err = -EBUSY;
4974                 goto err_free;
4975         }
4976
4977         if (ifmgd->auth_data)
4978                 ieee80211_destroy_auth_data(sdata, false);
4979
4980         /* prep auth_data so we don't go into idle on disassoc */
4981         ifmgd->auth_data = auth_data;
4982
4983         if (ifmgd->associated) {
4984                 u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
4985
4986                 sdata_info(sdata,
4987                            "disconnect from AP %pM for new auth to %pM\n",
4988                            ifmgd->associated->bssid, req->bss->bssid);
4989                 ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH,
4990                                        WLAN_REASON_UNSPECIFIED,
4991                                        false, frame_buf);
4992
4993                 ieee80211_report_disconnect(sdata, frame_buf,
4994                                             sizeof(frame_buf), true,
4995                                             WLAN_REASON_UNSPECIFIED);
4996         }
4997
4998         sdata_info(sdata, "authenticate with %pM\n", req->bss->bssid);
4999
5000         err = ieee80211_prep_connection(sdata, req->bss, false, false);
5001         if (err)
5002                 goto err_clear;
5003
5004         err = ieee80211_auth(sdata);
5005         if (err) {
5006                 sta_info_destroy_addr(sdata, req->bss->bssid);
5007                 goto err_clear;
5008         }
5009
5010         /* hold our own reference */
5011         cfg80211_ref_bss(local->hw.wiphy, auth_data->bss);
5012         return 0;
5013
5014  err_clear:
5015         eth_zero_addr(ifmgd->bssid);
5016         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BSSID);
5017         ifmgd->auth_data = NULL;
5018         mutex_lock(&sdata->local->mtx);
5019         ieee80211_vif_release_channel(sdata);
5020         mutex_unlock(&sdata->local->mtx);
5021  err_free:
5022         kfree(auth_data);
5023         return err;
5024 }
5025
5026 int ieee80211_mgd_assoc(struct ieee80211_sub_if_data *sdata,
5027                         struct cfg80211_assoc_request *req)
5028 {
5029         struct ieee80211_local *local = sdata->local;
5030         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
5031         struct ieee80211_bss *bss = (void *)req->bss->priv;
5032         struct ieee80211_mgd_assoc_data *assoc_data;
5033         const struct cfg80211_bss_ies *beacon_ies;
5034         struct ieee80211_supported_band *sband;
5035         const u8 *ssidie, *ht_ie, *vht_ie;
5036         int i, err;
5037         bool override = false;
5038
5039         assoc_data = kzalloc(sizeof(*assoc_data) + req->ie_len, GFP_KERNEL);
5040         if (!assoc_data)
5041                 return -ENOMEM;
5042
5043         rcu_read_lock();
5044         ssidie = ieee80211_bss_get_ie(req->bss, WLAN_EID_SSID);
5045         if (!ssidie || ssidie[1] > sizeof(assoc_data->ssid)) {
5046                 rcu_read_unlock();
5047                 kfree(assoc_data);
5048                 return -EINVAL;
5049         }
5050         memcpy(assoc_data->ssid, ssidie + 2, ssidie[1]);
5051         assoc_data->ssid_len = ssidie[1];
5052         rcu_read_unlock();
5053
5054         if (ifmgd->associated) {
5055                 u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
5056
5057                 sdata_info(sdata,
5058                            "disconnect from AP %pM for new assoc to %pM\n",
5059                            ifmgd->associated->bssid, req->bss->bssid);
5060                 ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH,
5061                                        WLAN_REASON_UNSPECIFIED,
5062                                        false, frame_buf);
5063
5064                 ieee80211_report_disconnect(sdata, frame_buf,
5065                                             sizeof(frame_buf), true,
5066                                             WLAN_REASON_UNSPECIFIED);
5067         }
5068
5069         if (ifmgd->auth_data && !ifmgd->auth_data->done) {
5070                 err = -EBUSY;
5071                 goto err_free;
5072         }
5073
5074         if (ifmgd->assoc_data) {
5075                 err = -EBUSY;
5076                 goto err_free;
5077         }
5078
5079         if (ifmgd->auth_data) {
5080                 bool match;
5081
5082                 /* keep sta info, bssid if matching */
5083                 match = ether_addr_equal(ifmgd->bssid, req->bss->bssid);
5084                 ieee80211_destroy_auth_data(sdata, match);
5085         }
5086
5087         /* prepare assoc data */
5088
5089         ifmgd->beacon_crc_valid = false;
5090
5091         assoc_data->wmm = bss->wmm_used &&
5092                           (local->hw.queues >= IEEE80211_NUM_ACS);
5093
5094         /*
5095          * IEEE802.11n does not allow TKIP/WEP as pairwise ciphers in HT mode.
5096          * We still associate in non-HT mode (11a/b/g) if any one of these
5097          * ciphers is configured as pairwise.
5098          * We can set this to true for non-11n hardware, that'll be checked
5099          * separately along with the peer capabilities.
5100          */
5101         for (i = 0; i < req->crypto.n_ciphers_pairwise; i++) {
5102                 if (req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP40 ||
5103                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_TKIP ||
5104                     req->crypto.ciphers_pairwise[i] == WLAN_CIPHER_SUITE_WEP104) {
5105                         ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
5106                         ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
5107                         ifmgd->flags |= IEEE80211_STA_DISABLE_HE;
5108                         netdev_info(sdata->dev,
5109                                     "disabling HE/HT/VHT due to WEP/TKIP use\n");
5110                 }
5111         }
5112
5113         /* Also disable HT if we don't support it or the AP doesn't use WMM */
5114         sband = local->hw.wiphy->bands[req->bss->channel->band];
5115         if (!sband->ht_cap.ht_supported ||
5116             local->hw.queues < IEEE80211_NUM_ACS || !bss->wmm_used ||
5117             ifmgd->flags & IEEE80211_STA_DISABLE_WMM) {
5118                 ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
5119                 if (!bss->wmm_used &&
5120                     !(ifmgd->flags & IEEE80211_STA_DISABLE_WMM))
5121                         netdev_info(sdata->dev,
5122                                     "disabling HT as WMM/QoS is not supported by the AP\n");
5123         }
5124
5125         /* disable VHT if we don't support it or the AP doesn't use WMM */
5126         if (!sband->vht_cap.vht_supported ||
5127             local->hw.queues < IEEE80211_NUM_ACS || !bss->wmm_used ||
5128             ifmgd->flags & IEEE80211_STA_DISABLE_WMM) {
5129                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
5130                 if (!bss->wmm_used &&
5131                     !(ifmgd->flags & IEEE80211_STA_DISABLE_WMM))
5132                         netdev_info(sdata->dev,
5133                                     "disabling VHT as WMM/QoS is not supported by the AP\n");
5134         }
5135
5136         memcpy(&ifmgd->ht_capa, &req->ht_capa, sizeof(ifmgd->ht_capa));
5137         memcpy(&ifmgd->ht_capa_mask, &req->ht_capa_mask,
5138                sizeof(ifmgd->ht_capa_mask));
5139
5140         memcpy(&ifmgd->vht_capa, &req->vht_capa, sizeof(ifmgd->vht_capa));
5141         memcpy(&ifmgd->vht_capa_mask, &req->vht_capa_mask,
5142                sizeof(ifmgd->vht_capa_mask));
5143
5144         if (req->ie && req->ie_len) {
5145                 memcpy(assoc_data->ie, req->ie, req->ie_len);
5146                 assoc_data->ie_len = req->ie_len;
5147         }
5148
5149         if (req->fils_kek) {
5150                 /* should already be checked in cfg80211 - so warn */
5151                 if (WARN_ON(req->fils_kek_len > FILS_MAX_KEK_LEN)) {
5152                         err = -EINVAL;
5153                         goto err_free;
5154                 }
5155                 memcpy(assoc_data->fils_kek, req->fils_kek,
5156                        req->fils_kek_len);
5157                 assoc_data->fils_kek_len = req->fils_kek_len;
5158         }
5159
5160         if (req->fils_nonces)
5161                 memcpy(assoc_data->fils_nonces, req->fils_nonces,
5162                        2 * FILS_NONCE_LEN);
5163
5164         assoc_data->bss = req->bss;
5165
5166         if (ifmgd->req_smps == IEEE80211_SMPS_AUTOMATIC) {
5167                 if (ifmgd->powersave)
5168                         sdata->smps_mode = IEEE80211_SMPS_DYNAMIC;
5169                 else
5170                         sdata->smps_mode = IEEE80211_SMPS_OFF;
5171         } else
5172                 sdata->smps_mode = ifmgd->req_smps;
5173
5174         assoc_data->capability = req->bss->capability;
5175         assoc_data->supp_rates = bss->supp_rates;
5176         assoc_data->supp_rates_len = bss->supp_rates_len;
5177
5178         rcu_read_lock();
5179         ht_ie = ieee80211_bss_get_ie(req->bss, WLAN_EID_HT_OPERATION);
5180         if (ht_ie && ht_ie[1] >= sizeof(struct ieee80211_ht_operation))
5181                 assoc_data->ap_ht_param =
5182                         ((struct ieee80211_ht_operation *)(ht_ie + 2))->ht_param;
5183         else
5184                 ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
5185         vht_ie = ieee80211_bss_get_ie(req->bss, WLAN_EID_VHT_CAPABILITY);
5186         if (vht_ie && vht_ie[1] >= sizeof(struct ieee80211_vht_cap))
5187                 memcpy(&assoc_data->ap_vht_cap, vht_ie + 2,
5188                        sizeof(struct ieee80211_vht_cap));
5189         else
5190                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
5191         rcu_read_unlock();
5192
5193         if (WARN((sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_UAPSD) &&
5194                  ieee80211_hw_check(&local->hw, PS_NULLFUNC_STACK),
5195              "U-APSD not supported with HW_PS_NULLFUNC_STACK\n"))
5196                 sdata->vif.driver_flags &= ~IEEE80211_VIF_SUPPORTS_UAPSD;
5197
5198         if (bss->wmm_used && bss->uapsd_supported &&
5199             (sdata->vif.driver_flags & IEEE80211_VIF_SUPPORTS_UAPSD)) {
5200                 assoc_data->uapsd = true;
5201                 ifmgd->flags |= IEEE80211_STA_UAPSD_ENABLED;
5202         } else {
5203                 assoc_data->uapsd = false;
5204                 ifmgd->flags &= ~IEEE80211_STA_UAPSD_ENABLED;
5205         }
5206
5207         if (req->prev_bssid)
5208                 memcpy(assoc_data->prev_bssid, req->prev_bssid, ETH_ALEN);
5209
5210         if (req->use_mfp) {
5211                 ifmgd->mfp = IEEE80211_MFP_REQUIRED;
5212                 ifmgd->flags |= IEEE80211_STA_MFP_ENABLED;
5213         } else {
5214                 ifmgd->mfp = IEEE80211_MFP_DISABLED;
5215                 ifmgd->flags &= ~IEEE80211_STA_MFP_ENABLED;
5216         }
5217
5218         if (req->flags & ASSOC_REQ_USE_RRM)
5219                 ifmgd->flags |= IEEE80211_STA_ENABLE_RRM;
5220         else
5221                 ifmgd->flags &= ~IEEE80211_STA_ENABLE_RRM;
5222
5223         if (req->crypto.control_port)
5224                 ifmgd->flags |= IEEE80211_STA_CONTROL_PORT;
5225         else
5226                 ifmgd->flags &= ~IEEE80211_STA_CONTROL_PORT;
5227
5228         sdata->control_port_protocol = req->crypto.control_port_ethertype;
5229         sdata->control_port_no_encrypt = req->crypto.control_port_no_encrypt;
5230         sdata->control_port_over_nl80211 =
5231                                         req->crypto.control_port_over_nl80211;
5232         sdata->encrypt_headroom = ieee80211_cs_headroom(local, &req->crypto,
5233                                                         sdata->vif.type);
5234
5235         /* kick off associate process */
5236
5237         ifmgd->assoc_data = assoc_data;
5238         ifmgd->dtim_period = 0;
5239         ifmgd->have_beacon = false;
5240
5241         /* override HT/VHT configuration only if the AP and we support it */
5242         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT)) {
5243                 struct ieee80211_sta_ht_cap sta_ht_cap;
5244
5245                 if (req->flags & ASSOC_REQ_DISABLE_HT)
5246                         override = true;
5247
5248                 memcpy(&sta_ht_cap, &sband->ht_cap, sizeof(sta_ht_cap));
5249                 ieee80211_apply_htcap_overrides(sdata, &sta_ht_cap);
5250
5251                 /* check for 40 MHz disable override */
5252                 if (!(ifmgd->flags & IEEE80211_STA_DISABLE_40MHZ) &&
5253                     sband->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 &&
5254                     !(sta_ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40))
5255                         override = true;
5256
5257                 if (!(ifmgd->flags & IEEE80211_STA_DISABLE_VHT) &&
5258                     req->flags & ASSOC_REQ_DISABLE_VHT)
5259                         override = true;
5260         }
5261
5262         if (req->flags & ASSOC_REQ_DISABLE_HT) {
5263                 ifmgd->flags |= IEEE80211_STA_DISABLE_HT;
5264                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
5265         }
5266
5267         if (req->flags & ASSOC_REQ_DISABLE_VHT)
5268                 ifmgd->flags |= IEEE80211_STA_DISABLE_VHT;
5269
5270         err = ieee80211_prep_connection(sdata, req->bss, true, override);
5271         if (err)
5272                 goto err_clear;
5273
5274         rcu_read_lock();
5275         beacon_ies = rcu_dereference(req->bss->beacon_ies);
5276
5277         if (ieee80211_hw_check(&sdata->local->hw, NEED_DTIM_BEFORE_ASSOC) &&
5278             !beacon_ies) {
5279                 /*
5280                  * Wait up to one beacon interval ...
5281                  * should this be more if we miss one?
5282                  */
5283                 sdata_info(sdata, "waiting for beacon from %pM\n",
5284                            ifmgd->bssid);
5285                 assoc_data->timeout = TU_TO_EXP_TIME(req->bss->beacon_interval);
5286                 assoc_data->timeout_started = true;
5287                 assoc_data->need_beacon = true;
5288         } else if (beacon_ies) {
5289                 const u8 *tim_ie = cfg80211_find_ie(WLAN_EID_TIM,
5290                                                     beacon_ies->data,
5291                                                     beacon_ies->len);
5292                 u8 dtim_count = 0;
5293
5294                 if (tim_ie && tim_ie[1] >= sizeof(struct ieee80211_tim_ie)) {
5295                         const struct ieee80211_tim_ie *tim;
5296                         tim = (void *)(tim_ie + 2);
5297                         ifmgd->dtim_period = tim->dtim_period;
5298                         dtim_count = tim->dtim_count;
5299                 }
5300                 ifmgd->have_beacon = true;
5301                 assoc_data->timeout = jiffies;
5302                 assoc_data->timeout_started = true;
5303
5304                 if (ieee80211_hw_check(&local->hw, TIMING_BEACON_ONLY)) {
5305                         sdata->vif.bss_conf.sync_tsf = beacon_ies->tsf;
5306                         sdata->vif.bss_conf.sync_device_ts =
5307                                 bss->device_ts_beacon;
5308                         sdata->vif.bss_conf.sync_dtim_count = dtim_count;
5309                 }
5310         } else {
5311                 assoc_data->timeout = jiffies;
5312                 assoc_data->timeout_started = true;
5313         }
5314         rcu_read_unlock();
5315
5316         run_again(sdata, assoc_data->timeout);
5317
5318         if (bss->corrupt_data) {
5319                 char *corrupt_type = "data";
5320                 if (bss->corrupt_data & IEEE80211_BSS_CORRUPT_BEACON) {
5321                         if (bss->corrupt_data &
5322                                         IEEE80211_BSS_CORRUPT_PROBE_RESP)
5323                                 corrupt_type = "beacon and probe response";
5324                         else
5325                                 corrupt_type = "beacon";
5326                 } else if (bss->corrupt_data & IEEE80211_BSS_CORRUPT_PROBE_RESP)
5327                         corrupt_type = "probe response";
5328                 sdata_info(sdata, "associating with AP with corrupt %s\n",
5329                            corrupt_type);
5330         }
5331
5332         return 0;
5333  err_clear:
5334         eth_zero_addr(ifmgd->bssid);
5335         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_BSSID);
5336         ifmgd->assoc_data = NULL;
5337  err_free:
5338         kfree(assoc_data);
5339         return err;
5340 }
5341
5342 int ieee80211_mgd_deauth(struct ieee80211_sub_if_data *sdata,
5343                          struct cfg80211_deauth_request *req)
5344 {
5345         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
5346         u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
5347         bool tx = !req->local_state_change;
5348
5349         if (ifmgd->auth_data &&
5350             ether_addr_equal(ifmgd->auth_data->bss->bssid, req->bssid)) {
5351                 sdata_info(sdata,
5352                            "aborting authentication with %pM by local choice (Reason: %u=%s)\n",
5353                            req->bssid, req->reason_code,
5354                            ieee80211_get_reason_code_string(req->reason_code));
5355
5356                 drv_mgd_prepare_tx(sdata->local, sdata, 0);
5357                 ieee80211_send_deauth_disassoc(sdata, req->bssid,
5358                                                IEEE80211_STYPE_DEAUTH,
5359                                                req->reason_code, tx,
5360                                                frame_buf);
5361                 ieee80211_destroy_auth_data(sdata, false);
5362                 ieee80211_report_disconnect(sdata, frame_buf,
5363                                             sizeof(frame_buf), true,
5364                                             req->reason_code);
5365
5366                 return 0;
5367         }
5368
5369         if (ifmgd->assoc_data &&
5370             ether_addr_equal(ifmgd->assoc_data->bss->bssid, req->bssid)) {
5371                 sdata_info(sdata,
5372                            "aborting association with %pM by local choice (Reason: %u=%s)\n",
5373                            req->bssid, req->reason_code,
5374                            ieee80211_get_reason_code_string(req->reason_code));
5375
5376                 drv_mgd_prepare_tx(sdata->local, sdata, 0);
5377                 ieee80211_send_deauth_disassoc(sdata, req->bssid,
5378                                                IEEE80211_STYPE_DEAUTH,
5379                                                req->reason_code, tx,
5380                                                frame_buf);
5381                 ieee80211_destroy_assoc_data(sdata, false, true);
5382                 ieee80211_report_disconnect(sdata, frame_buf,
5383                                             sizeof(frame_buf), true,
5384                                             req->reason_code);
5385                 return 0;
5386         }
5387
5388         if (ifmgd->associated &&
5389             ether_addr_equal(ifmgd->associated->bssid, req->bssid)) {
5390                 sdata_info(sdata,
5391                            "deauthenticating from %pM by local choice (Reason: %u=%s)\n",
5392                            req->bssid, req->reason_code,
5393                            ieee80211_get_reason_code_string(req->reason_code));
5394
5395                 ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DEAUTH,
5396                                        req->reason_code, tx, frame_buf);
5397                 ieee80211_report_disconnect(sdata, frame_buf,
5398                                             sizeof(frame_buf), true,
5399                                             req->reason_code);
5400                 return 0;
5401         }
5402
5403         return -ENOTCONN;
5404 }
5405
5406 int ieee80211_mgd_disassoc(struct ieee80211_sub_if_data *sdata,
5407                            struct cfg80211_disassoc_request *req)
5408 {
5409         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
5410         u8 bssid[ETH_ALEN];
5411         u8 frame_buf[IEEE80211_DEAUTH_FRAME_LEN];
5412
5413         /*
5414          * cfg80211 should catch this ... but it's racy since
5415          * we can receive a disassoc frame, process it, hand it
5416          * to cfg80211 while that's in a locked section already
5417          * trying to tell us that the user wants to disconnect.
5418          */
5419         if (ifmgd->associated != req->bss)
5420                 return -ENOLINK;
5421
5422         sdata_info(sdata,
5423                    "disassociating from %pM by local choice (Reason: %u=%s)\n",
5424                    req->bss->bssid, req->reason_code, ieee80211_get_reason_code_string(req->reason_code));
5425
5426         memcpy(bssid, req->bss->bssid, ETH_ALEN);
5427         ieee80211_set_disassoc(sdata, IEEE80211_STYPE_DISASSOC,
5428                                req->reason_code, !req->local_state_change,
5429                                frame_buf);
5430
5431         ieee80211_report_disconnect(sdata, frame_buf, sizeof(frame_buf), true,
5432                                     req->reason_code);
5433
5434         return 0;
5435 }
5436
5437 void ieee80211_mgd_stop(struct ieee80211_sub_if_data *sdata)
5438 {
5439         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
5440
5441         /*
5442          * Make sure some work items will not run after this,
5443          * they will not do anything but might not have been
5444          * cancelled when disconnecting.
5445          */
5446         cancel_work_sync(&ifmgd->monitor_work);
5447         cancel_work_sync(&ifmgd->beacon_connection_loss_work);
5448         cancel_work_sync(&ifmgd->request_smps_work);
5449         cancel_work_sync(&ifmgd->csa_connection_drop_work);
5450         cancel_work_sync(&ifmgd->chswitch_work);
5451         cancel_delayed_work_sync(&ifmgd->tdls_peer_del_work);
5452
5453         sdata_lock(sdata);
5454         if (ifmgd->assoc_data) {
5455                 struct cfg80211_bss *bss = ifmgd->assoc_data->bss;
5456                 ieee80211_destroy_assoc_data(sdata, false, false);
5457                 cfg80211_assoc_timeout(sdata->dev, bss);
5458         }
5459         if (ifmgd->auth_data)
5460                 ieee80211_destroy_auth_data(sdata, false);
5461         spin_lock_bh(&ifmgd->teardown_lock);
5462         if (ifmgd->teardown_skb) {
5463                 kfree_skb(ifmgd->teardown_skb);
5464                 ifmgd->teardown_skb = NULL;
5465                 ifmgd->orig_teardown_skb = NULL;
5466         }
5467         spin_unlock_bh(&ifmgd->teardown_lock);
5468         del_timer_sync(&ifmgd->timer);
5469         sdata_unlock(sdata);
5470 }
5471
5472 void ieee80211_cqm_rssi_notify(struct ieee80211_vif *vif,
5473                                enum nl80211_cqm_rssi_threshold_event rssi_event,
5474                                s32 rssi_level,
5475                                gfp_t gfp)
5476 {
5477         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
5478
5479         trace_api_cqm_rssi_notify(sdata, rssi_event, rssi_level);
5480
5481         cfg80211_cqm_rssi_notify(sdata->dev, rssi_event, rssi_level, gfp);
5482 }
5483 EXPORT_SYMBOL(ieee80211_cqm_rssi_notify);
5484
5485 void ieee80211_cqm_beacon_loss_notify(struct ieee80211_vif *vif, gfp_t gfp)
5486 {
5487         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
5488
5489         trace_api_cqm_beacon_loss_notify(sdata->local, sdata);
5490
5491         cfg80211_cqm_beacon_loss_notify(sdata->dev, gfp);
5492 }
5493 EXPORT_SYMBOL(ieee80211_cqm_beacon_loss_notify);