drivers/net/wireless/iwlwifi/mvm/mac80211.c

   1 /******************************************************************************
   2  *
   3  * This file is provided under a dual BSD/GPLv2 license.  When using or
   4  * redistributing this file, you may do so under either license.
   5  *
   6  * GPL LICENSE SUMMARY
   7  *
   8  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
   9  * Copyright(c) 2013 - 2014 Intel Mobile Communications GmbH
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of version 2 of the GNU General Public License as
  13  * published by the Free Software Foundation.
  14  *
  15  * This program is distributed in the hope that it will be useful, but
  16  * WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
  23  * USA
  24  *
  25  * The full GNU General Public License is included in this distribution
  26  * in the file called COPYING.
  27  *
  28  * Contact Information:
  29  *  Intel Linux Wireless <ilw@linux.intel.com>
  30  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
  31  *
  32  * BSD LICENSE
  33  *
  34  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
  35  * Copyright(c) 2013 - 2014 Intel Mobile Communications GmbH
  36  * All rights reserved.
  37  *
  38  * Redistribution and use in source and binary forms, with or without
  39  * modification, are permitted provided that the following conditions
  40  * are met:
  41  *
  42  *  * Redistributions of source code must retain the above copyright
  43  *    notice, this list of conditions and the following disclaimer.
  44  *  * Redistributions in binary form must reproduce the above copyright
  45  *    notice, this list of conditions and the following disclaimer in
  46  *    the documentation and/or other materials provided with the
  47  *    distribution.
  48  *  * Neither the name Intel Corporation nor the names of its
  49  *    contributors may be used to endorse or promote products derived
  50  *    from this software without specific prior written permission.
  51  *
  52  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  53  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  54  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  55  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  56  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  57  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  58  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  59  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  60  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  61  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  62  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  63  *
  64  *****************************************************************************/
  65 #include <linux/kernel.h>
  66 #include <linux/slab.h>
  67 #include <linux/skbuff.h>
  68 #include <linux/netdevice.h>
  69 #include <linux/etherdevice.h>
  70 #include <linux/ip.h>
  71 #include <linux/if_arp.h>
  72 #include <net/mac80211.h>
  73 #include <net/ieee80211_radiotap.h>
  74 #include <net/tcp.h>
  75
  76 #include "iwl-op-mode.h"
  77 #include "iwl-io.h"
  78 #include "mvm.h"
  79 #include "sta.h"
  80 #include "time-event.h"
  81 #include "iwl-eeprom-parse.h"
  82 #include "fw-api-scan.h"
  83 #include "iwl-phy-db.h"
  84 #include "testmode.h"
  85 #include "iwl-fw-error-dump.h"
  86 #include "iwl-prph.h"
  87
  88 static const struct ieee80211_iface_limit iwl_mvm_limits[] = {
  89         {
  90                 .max = 1,
  91                 .types = BIT(NL80211_IFTYPE_STATION),
  92         },
  93         {
  94                 .max = 1,
  95                 .types = BIT(NL80211_IFTYPE_AP) |
  96                         BIT(NL80211_IFTYPE_P2P_CLIENT) |
  97                         BIT(NL80211_IFTYPE_P2P_GO),
  98         },
  99         {
 100                 .max = 1,
 101                 .types = BIT(NL80211_IFTYPE_P2P_DEVICE),
 102         },
 103 };
 104
 105 static const struct ieee80211_iface_combination iwl_mvm_iface_combinations[] = {
 106         {
 107                 .num_different_channels = 1,
 108                 .max_interfaces = 3,
 109                 .limits = iwl_mvm_limits,
 110                 .n_limits = ARRAY_SIZE(iwl_mvm_limits),
 111         },
 112 };
 113
 114 #ifdef CONFIG_PM_SLEEP
 115 static const struct nl80211_wowlan_tcp_data_token_feature
 116 iwl_mvm_wowlan_tcp_token_feature = {
 117         .min_len = 0,
 118         .max_len = 255,
 119         .bufsize = IWL_WOWLAN_REMOTE_WAKE_MAX_TOKENS,
 120 };
 121
 122 static const struct wiphy_wowlan_tcp_support iwl_mvm_wowlan_tcp_support = {
 123         .tok = &iwl_mvm_wowlan_tcp_token_feature,
 124         .data_payload_max = IWL_WOWLAN_TCP_MAX_PACKET_LEN -
 125                             sizeof(struct ethhdr) -
 126                             sizeof(struct iphdr) -
 127                             sizeof(struct tcphdr),
 128         .data_interval_max = 65535, /* __le16 in API */
 129         .wake_payload_max = IWL_WOWLAN_REMOTE_WAKE_MAX_PACKET_LEN -
 130                             sizeof(struct ethhdr) -
 131                             sizeof(struct iphdr) -
 132                             sizeof(struct tcphdr),
 133         .seq = true,
 134 };
 135 #endif
 136
 137 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
 138 /*
 139  * Use the reserved field to indicate magic values.
 140  * these values will only be used internally by the driver,
 141  * and won't make it to the fw (reserved will be 0).
 142  * BC_FILTER_MAGIC_IP - configure the val of this attribute to
 143  *      be the vif's ip address. in case there is not a single
 144  *      ip address (0, or more than 1), this attribute will
 145  *      be skipped.
 146  * BC_FILTER_MAGIC_MAC - set the val of this attribute to
 147  *      the LSB bytes of the vif's mac address
 148  */
 149 enum {
 150         BC_FILTER_MAGIC_NONE = 0,
 151         BC_FILTER_MAGIC_IP,
 152         BC_FILTER_MAGIC_MAC,
 153 };
 154
 155 static const struct iwl_fw_bcast_filter iwl_mvm_default_bcast_filters[] = {
 156         {
 157                 /* arp */
 158                 .discard = 0,
 159                 .frame_type = BCAST_FILTER_FRAME_TYPE_ALL,
 160                 .attrs = {
 161                         {
 162                                 /* frame type - arp, hw type - ethernet */
 163                                 .offset_type =
 164                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
 165                                 .offset = sizeof(rfc1042_header),
 166                                 .val = cpu_to_be32(0x08060001),
 167                                 .mask = cpu_to_be32(0xffffffff),
 168                         },
 169                         {
 170                                 /* arp dest ip */
 171                                 .offset_type =
 172                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
 173                                 .offset = sizeof(rfc1042_header) + 2 +
 174                                           sizeof(struct arphdr) +
 175                                           ETH_ALEN + sizeof(__be32) +
 176                                           ETH_ALEN,
 177                                 .mask = cpu_to_be32(0xffffffff),
 178                                 /* mark it as special field */
 179                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_IP),
 180                         },
 181                 },
 182         },
 183         {
 184                 /* dhcp offer bcast */
 185                 .discard = 0,
 186                 .frame_type = BCAST_FILTER_FRAME_TYPE_IPV4,
 187                 .attrs = {
 188                         {
 189                                 /* udp dest port - 68 (bootp client)*/
 190                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
 191                                 .offset = offsetof(struct udphdr, dest),
 192                                 .val = cpu_to_be32(0x00440000),
 193                                 .mask = cpu_to_be32(0xffff0000),
 194                         },
 195                         {
 196                                 /* dhcp - lsb bytes of client hw address */
 197                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
 198                                 .offset = 38,
 199                                 .mask = cpu_to_be32(0xffffffff),
 200                                 /* mark it as special field */
 201                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_MAC),
 202                         },
 203                 },
 204         },
 205         /* last filter must be empty */
 206         {},
 207 };
 208 #endif
 209
 210 void iwl_mvm_ref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
 211 {
 212         if (!iwl_mvm_is_d0i3_supported(mvm))
 213                 return;
 214
 215         IWL_DEBUG_RPM(mvm, "Take mvm reference - type %d\n", ref_type);
 216         spin_lock_bh(&mvm->refs_lock);
 217         mvm->refs[ref_type]++;
 218         spin_unlock_bh(&mvm->refs_lock);
 219         iwl_trans_ref(mvm->trans);
 220 }
 221
 222 void iwl_mvm_unref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
 223 {
 224         if (!iwl_mvm_is_d0i3_supported(mvm))
 225                 return;
 226
 227         IWL_DEBUG_RPM(mvm, "Leave mvm reference - type %d\n", ref_type);
 228         spin_lock_bh(&mvm->refs_lock);
 229         WARN_ON(!mvm->refs[ref_type]--);
 230         spin_unlock_bh(&mvm->refs_lock);
 231         iwl_trans_unref(mvm->trans);
 232 }
 233
 234 static void iwl_mvm_unref_all_except(struct iwl_mvm *mvm,
 235                                      enum iwl_mvm_ref_type except_ref)
 236 {
 237         int i, j;
 238
 239         if (!iwl_mvm_is_d0i3_supported(mvm))
 240                 return;
 241
 242         spin_lock_bh(&mvm->refs_lock);
 243         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
 244                 if (except_ref == i || !mvm->refs[i])
 245                         continue;
 246
 247                 IWL_DEBUG_RPM(mvm, "Cleanup: remove mvm ref type %d (%d)\n",
 248                               i, mvm->refs[i]);
 249                 for (j = 0; j < mvm->refs[i]; j++)
 250                         iwl_trans_unref(mvm->trans);
 251                 mvm->refs[i] = 0;
 252         }
 253         spin_unlock_bh(&mvm->refs_lock);
 254 }
 255
 256 int iwl_mvm_ref_sync(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
 257 {
 258         iwl_mvm_ref(mvm, ref_type);
 259
 260         if (!wait_event_timeout(mvm->d0i3_exit_waitq,
 261                                 !test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status),
 262                                 HZ)) {
 263                 WARN_ON_ONCE(1);
 264                 iwl_mvm_unref(mvm, ref_type);
 265                 return -EIO;
 266         }
 267
 268         return 0;
 269 }
 270
 271 static void iwl_mvm_reset_phy_ctxts(struct iwl_mvm *mvm)
 272 {
 273         int i;
 274
 275         memset(mvm->phy_ctxts, 0, sizeof(mvm->phy_ctxts));
 276         for (i = 0; i < NUM_PHY_CTX; i++) {
 277                 mvm->phy_ctxts[i].id = i;
 278                 mvm->phy_ctxts[i].ref = 0;
 279         }
 280 }
 281
 282 static int iwl_mvm_max_scan_ie_len(struct iwl_mvm *mvm)
 283 {
 284         /* we create the 802.11 header and SSID element */
 285         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_NO_BASIC_SSID)
 286                 return mvm->fw->ucode_capa.max_probe_length - 24 - 2;
 287         return mvm->fw->ucode_capa.max_probe_length - 24 - 34;
 288 }
 289
 290 int iwl_mvm_mac_setup_register(struct iwl_mvm *mvm)
 291 {
 292         struct ieee80211_hw *hw = mvm->hw;
 293         int num_mac, ret, i;
 294
 295         /* Tell mac80211 our characteristics */
 296         hw->flags = IEEE80211_HW_SIGNAL_DBM |
 297                     IEEE80211_HW_SPECTRUM_MGMT |
 298                     IEEE80211_HW_REPORTS_TX_ACK_STATUS |
 299                     IEEE80211_HW_QUEUE_CONTROL |
 300                     IEEE80211_HW_WANT_MONITOR_VIF |
 301                     IEEE80211_HW_SUPPORTS_PS |
 302                     IEEE80211_HW_SUPPORTS_DYNAMIC_PS |
 303                     IEEE80211_HW_AMPDU_AGGREGATION |
 304                     IEEE80211_HW_TIMING_BEACON_ONLY |
 305                     IEEE80211_HW_CONNECTION_MONITOR |
 306                     IEEE80211_HW_CHANCTX_STA_CSA |
 307                     IEEE80211_HW_SUPPORTS_DYNAMIC_SMPS |
 308                     IEEE80211_HW_SUPPORTS_STATIC_SMPS;
 309
 310         hw->queues = mvm->first_agg_queue;
 311         hw->offchannel_tx_hw_queue = IWL_MVM_OFFCHANNEL_QUEUE;
 312         hw->radiotap_mcs_details |= IEEE80211_RADIOTAP_MCS_HAVE_FEC |
 313                                     IEEE80211_RADIOTAP_MCS_HAVE_STBC;
 314         hw->radiotap_vht_details |= IEEE80211_RADIOTAP_VHT_KNOWN_STBC;
 315         hw->rate_control_algorithm = "iwl-mvm-rs";
 316
 317         /*
 318          * Enable 11w if advertised by firmware and software crypto
 319          * is not enabled (as the firmware will interpret some mgmt
 320          * packets, so enabling it with software crypto isn't safe)
 321          */
 322         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_MFP &&
 323             !iwlwifi_mod_params.sw_crypto)
 324                 hw->flags |= IEEE80211_HW_MFP_CAPABLE;
 325
 326         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_UAPSD_SUPPORT &&
 327             IWL_UCODE_API(mvm->fw->ucode_ver) >= 9 &&
 328             !iwlwifi_mod_params.uapsd_disable) {
 329                 hw->flags |= IEEE80211_HW_SUPPORTS_UAPSD;
 330                 hw->uapsd_queues = IWL_UAPSD_AC_INFO;
 331                 hw->uapsd_max_sp_len = IWL_UAPSD_MAX_SP;
 332         }
 333
 334         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
 335                 hw->flags |= IEEE80211_SINGLE_HW_SCAN_ON_ALL_BANDS;
 336
 337         hw->sta_data_size = sizeof(struct iwl_mvm_sta);
 338         hw->vif_data_size = sizeof(struct iwl_mvm_vif);
 339         hw->chanctx_data_size = sizeof(u16);
 340
 341         hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION) |
 342                 BIT(NL80211_IFTYPE_P2P_CLIENT) |
 343                 BIT(NL80211_IFTYPE_AP) |
 344                 BIT(NL80211_IFTYPE_P2P_GO) |
 345                 BIT(NL80211_IFTYPE_P2P_DEVICE) |
 346                 BIT(NL80211_IFTYPE_ADHOC);
 347
 348         hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
 349         hw->wiphy->regulatory_flags |= REGULATORY_CUSTOM_REG |
 350                                        REGULATORY_DISABLE_BEACON_HINTS;
 351
 352         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_GO_UAPSD)
 353                 hw->wiphy->flags |= WIPHY_FLAG_AP_UAPSD;
 354
 355         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_CSA_FLOW)
 356                 hw->wiphy->flags |= WIPHY_FLAG_HAS_CHANNEL_SWITCH;
 357
 358         hw->wiphy->iface_combinations = iwl_mvm_iface_combinations;
 359         hw->wiphy->n_iface_combinations =
 360                 ARRAY_SIZE(iwl_mvm_iface_combinations);
 361
 362         hw->wiphy->max_remain_on_channel_duration = 10000;
 363         hw->max_listen_interval = IWL_CONN_MAX_LISTEN_INTERVAL;
 364
 365         /* Extract MAC address */
 366         memcpy(mvm->addresses[0].addr, mvm->nvm_data->hw_addr, ETH_ALEN);
 367         hw->wiphy->addresses = mvm->addresses;
 368         hw->wiphy->n_addresses = 1;
 369
 370         /* Extract additional MAC addresses if available */
 371         num_mac = (mvm->nvm_data->n_hw_addrs > 1) ?
 372                 min(IWL_MVM_MAX_ADDRESSES, mvm->nvm_data->n_hw_addrs) : 1;
 373
 374         for (i = 1; i < num_mac; i++) {
 375                 memcpy(mvm->addresses[i].addr, mvm->addresses[i-1].addr,
 376                        ETH_ALEN);
 377                 mvm->addresses[i].addr[5]++;
 378                 hw->wiphy->n_addresses++;
 379         }
 380
 381         iwl_mvm_reset_phy_ctxts(mvm);
 382
 383         hw->wiphy->max_scan_ie_len = iwl_mvm_max_scan_ie_len(mvm);
 384
 385         hw->wiphy->max_scan_ssids = PROBE_OPTION_MAX;
 386
 387         if (mvm->nvm_data->bands[IEEE80211_BAND_2GHZ].n_channels)
 388                 hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
 389                         &mvm->nvm_data->bands[IEEE80211_BAND_2GHZ];
 390         if (mvm->nvm_data->bands[IEEE80211_BAND_5GHZ].n_channels)
 391                 hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
 392                         &mvm->nvm_data->bands[IEEE80211_BAND_5GHZ];
 393
 394         hw->wiphy->hw_version = mvm->trans->hw_id;
 395
 396         if (iwlmvm_mod_params.power_scheme != IWL_POWER_SCHEME_CAM)
 397                 hw->wiphy->flags |= WIPHY_FLAG_PS_ON_BY_DEFAULT;
 398         else
 399                 hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
 400
 401         /* TODO: enable that only for firmwares that don't crash */
 402         /* hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_SCHED_SCAN; */
 403         hw->wiphy->max_sched_scan_ssids = PROBE_OPTION_MAX;
 404         hw->wiphy->max_match_sets = IWL_SCAN_MAX_PROFILES;
 405         /* we create the 802.11 header and zero length SSID IE. */
 406         hw->wiphy->max_sched_scan_ie_len = SCAN_OFFLOAD_PROBE_REQ_SIZE - 24 - 2;
 407
 408         hw->wiphy->features |= NL80211_FEATURE_P2P_GO_CTWIN |
 409                                NL80211_FEATURE_LOW_PRIORITY_SCAN |
 410                                NL80211_FEATURE_P2P_GO_OPPPS;
 411
 412         mvm->rts_threshold = IEEE80211_MAX_RTS_THRESHOLD;
 413
 414         /* currently FW API supports only one optional cipher scheme */
 415         if (mvm->fw->cs[0].cipher) {
 416                 mvm->hw->n_cipher_schemes = 1;
 417                 mvm->hw->cipher_schemes = &mvm->fw->cs[0];
 418         }
 419
 420 #ifdef CONFIG_PM_SLEEP
 421         if (iwl_mvm_is_d0i3_supported(mvm) &&
 422             device_can_wakeup(mvm->trans->dev)) {
 423                 mvm->wowlan.flags = WIPHY_WOWLAN_ANY;
 424                 hw->wiphy->wowlan = &mvm->wowlan;
 425         } else if (mvm->fw->img[IWL_UCODE_WOWLAN].sec[0].len &&
 426             mvm->trans->ops->d3_suspend &&
 427             mvm->trans->ops->d3_resume &&
 428             device_can_wakeup(mvm->trans->dev)) {
 429                 mvm->wowlan.flags = WIPHY_WOWLAN_MAGIC_PKT |
 430                                     WIPHY_WOWLAN_DISCONNECT |
 431                                     WIPHY_WOWLAN_EAP_IDENTITY_REQ |
 432                                     WIPHY_WOWLAN_RFKILL_RELEASE;
 433                 if (!iwlwifi_mod_params.sw_crypto)
 434                         mvm->wowlan.flags |= WIPHY_WOWLAN_SUPPORTS_GTK_REKEY |
 435                                              WIPHY_WOWLAN_GTK_REKEY_FAILURE |
 436                                              WIPHY_WOWLAN_4WAY_HANDSHAKE;
 437
 438                 mvm->wowlan.n_patterns = IWL_WOWLAN_MAX_PATTERNS;
 439                 mvm->wowlan.pattern_min_len = IWL_WOWLAN_MIN_PATTERN_LEN;
 440                 mvm->wowlan.pattern_max_len = IWL_WOWLAN_MAX_PATTERN_LEN;
 441                 mvm->wowlan.tcp = &iwl_mvm_wowlan_tcp_support;
 442                 hw->wiphy->wowlan = &mvm->wowlan;
 443         }
 444 #endif
 445
 446 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
 447         /* assign default bcast filtering configuration */
 448         mvm->bcast_filters = iwl_mvm_default_bcast_filters;
 449 #endif
 450
 451         ret = iwl_mvm_leds_init(mvm);
 452         if (ret)
 453                 return ret;
 454
 455         ret = ieee80211_register_hw(mvm->hw);
 456         if (ret)
 457                 iwl_mvm_leds_exit(mvm);
 458
 459         return ret;
 460 }
 461
 462 static bool iwl_mvm_defer_tx(struct iwl_mvm *mvm,
 463                              struct ieee80211_sta *sta,
 464                              struct sk_buff *skb)
 465 {
 466         struct iwl_mvm_sta *mvmsta;
 467         bool defer = false;
 468
 469         /*
 470          * double check the IN_D0I3 flag both before and after
 471          * taking the spinlock, in order to prevent taking
 472          * the spinlock when not needed.
 473          */
 474         if (likely(!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status)))
 475                 return false;
 476
 477         spin_lock(&mvm->d0i3_tx_lock);
 478         /*
 479          * testing the flag again ensures the skb dequeue
 480          * loop (on d0i3 exit) hasn't run yet.
 481          */
 482         if (!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status))
 483                 goto out;
 484
 485         mvmsta = iwl_mvm_sta_from_mac80211(sta);
 486         if (mvmsta->sta_id == IWL_MVM_STATION_COUNT ||
 487             mvmsta->sta_id != mvm->d0i3_ap_sta_id)
 488                 goto out;
 489
 490         __skb_queue_tail(&mvm->d0i3_tx, skb);
 491         ieee80211_stop_queues(mvm->hw);
 492
 493         /* trigger wakeup */
 494         iwl_mvm_ref(mvm, IWL_MVM_REF_TX);
 495         iwl_mvm_unref(mvm, IWL_MVM_REF_TX);
 496
 497         defer = true;
 498 out:
 499         spin_unlock(&mvm->d0i3_tx_lock);
 500         return defer;
 501 }
 502
 503 static void iwl_mvm_mac_tx(struct ieee80211_hw *hw,
 504                            struct ieee80211_tx_control *control,
 505                            struct sk_buff *skb)
 506 {
 507         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 508         struct ieee80211_sta *sta = control->sta;
 509         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
 510         struct ieee80211_hdr *hdr = (void *)skb->data;
 511
 512         if (iwl_mvm_is_radio_killed(mvm)) {
 513                 IWL_DEBUG_DROP(mvm, "Dropping - RF/CT KILL\n");
 514                 goto drop;
 515         }
 516
 517         if (IEEE80211_SKB_CB(skb)->hw_queue == IWL_MVM_OFFCHANNEL_QUEUE &&
 518             !test_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status))
 519                 goto drop;
 520
 521         /* treat non-bufferable MMPDUs as broadcast if sta is sleeping */
 522         if (unlikely(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER &&
 523                      ieee80211_is_mgmt(hdr->frame_control) &&
 524                      !ieee80211_is_deauth(hdr->frame_control) &&
 525                      !ieee80211_is_disassoc(hdr->frame_control) &&
 526                      !ieee80211_is_action(hdr->frame_control)))
 527                 sta = NULL;
 528
 529         if (sta) {
 530                 if (iwl_mvm_defer_tx(mvm, sta, skb))
 531                         return;
 532                 if (iwl_mvm_tx_skb(mvm, skb, sta))
 533                         goto drop;
 534                 return;
 535         }
 536
 537         if (iwl_mvm_tx_skb_non_sta(mvm, skb))
 538                 goto drop;
 539         return;
 540  drop:
 541         ieee80211_free_txskb(hw, skb);
 542 }
 543
 544 static inline bool iwl_enable_rx_ampdu(const struct iwl_cfg *cfg)
 545 {
 546         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_RXAGG)
 547                 return false;
 548         return true;
 549 }
 550
 551 static inline bool iwl_enable_tx_ampdu(const struct iwl_cfg *cfg)
 552 {
 553         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_TXAGG)
 554                 return false;
 555         if (iwlwifi_mod_params.disable_11n & IWL_ENABLE_HT_TXAGG)
 556                 return true;
 557
 558         /* enabled by default */
 559         return true;
 560 }
 561
 562 static int iwl_mvm_mac_ampdu_action(struct ieee80211_hw *hw,
 563                                     struct ieee80211_vif *vif,
 564                                     enum ieee80211_ampdu_mlme_action action,
 565                                     struct ieee80211_sta *sta, u16 tid,
 566                                     u16 *ssn, u8 buf_size)
 567 {
 568         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 569         int ret;
 570         bool tx_agg_ref = false;
 571
 572         IWL_DEBUG_HT(mvm, "A-MPDU action on addr %pM tid %d: action %d\n",
 573                      sta->addr, tid, action);
 574
 575         if (!(mvm->nvm_data->sku_cap_11n_enable))
 576                 return -EACCES;
 577
 578         /* return from D0i3 before starting a new Tx aggregation */
 579         switch (action) {
 580         case IEEE80211_AMPDU_TX_START:
 581         case IEEE80211_AMPDU_TX_STOP_CONT:
 582         case IEEE80211_AMPDU_TX_STOP_FLUSH:
 583         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
 584         case IEEE80211_AMPDU_TX_OPERATIONAL:
 585                 /*
 586                  * for tx start, wait synchronously until D0i3 exit to
 587                  * get the correct sequence number for the tid.
 588                  * additionally, some other ampdu actions use direct
 589                  * target access, which is not handled automatically
 590                  * by the trans layer (unlike commands), so wait for
 591                  * d0i3 exit in these cases as well.
 592                  */
 593                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_TX_AGG);
 594                 if (ret)
 595                         return ret;
 596
 597                 tx_agg_ref = true;
 598                 break;
 599         default:
 600                 break;
 601         }
 602
 603         mutex_lock(&mvm->mutex);
 604
 605         switch (action) {
 606         case IEEE80211_AMPDU_RX_START:
 607                 if (!iwl_enable_rx_ampdu(mvm->cfg)) {
 608                         ret = -EINVAL;
 609                         break;
 610                 }
 611                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, *ssn, true);
 612                 break;
 613         case IEEE80211_AMPDU_RX_STOP:
 614                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, 0, false);
 615                 break;
 616         case IEEE80211_AMPDU_TX_START:
 617                 if (!iwl_enable_tx_ampdu(mvm->cfg)) {
 618                         ret = -EINVAL;
 619                         break;
 620                 }
 621                 ret = iwl_mvm_sta_tx_agg_start(mvm, vif, sta, tid, ssn);
 622                 break;
 623         case IEEE80211_AMPDU_TX_STOP_CONT:
 624                 ret = iwl_mvm_sta_tx_agg_stop(mvm, vif, sta, tid);
 625                 break;
 626         case IEEE80211_AMPDU_TX_STOP_FLUSH:
 627         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
 628                 ret = iwl_mvm_sta_tx_agg_flush(mvm, vif, sta, tid);
 629                 break;
 630         case IEEE80211_AMPDU_TX_OPERATIONAL:
 631                 ret = iwl_mvm_sta_tx_agg_oper(mvm, vif, sta, tid, buf_size);
 632                 break;
 633         default:
 634                 WARN_ON_ONCE(1);
 635                 ret = -EINVAL;
 636                 break;
 637         }
 638         mutex_unlock(&mvm->mutex);
 639
 640         /*
 641          * If the tid is marked as started, we won't use it for offloaded
 642          * traffic on the next D0i3 entry. It's safe to unref.
 643          */
 644         if (tx_agg_ref)
 645                 iwl_mvm_unref(mvm, IWL_MVM_REF_TX_AGG);
 646
 647         return ret;
 648 }
 649
 650 static void iwl_mvm_cleanup_iterator(void *data, u8 *mac,
 651                                      struct ieee80211_vif *vif)
 652 {
 653         struct iwl_mvm *mvm = data;
 654         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
 655
 656         mvmvif->uploaded = false;
 657         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
 658
 659         /* does this make sense at all? */
 660         mvmvif->color++;
 661
 662         spin_lock_bh(&mvm->time_event_lock);
 663         iwl_mvm_te_clear_data(mvm, &mvmvif->time_event_data);
 664         spin_unlock_bh(&mvm->time_event_lock);
 665
 666         mvmvif->phy_ctxt = NULL;
 667         memset(&mvmvif->bf_data, 0, sizeof(mvmvif->bf_data));
 668 }
 669
 670 #ifdef CONFIG_IWLWIFI_DEBUGFS
 671 static void iwl_mvm_fw_error_dump(struct iwl_mvm *mvm)
 672 {
 673         struct iwl_fw_error_dump_file *dump_file;
 674         struct iwl_fw_error_dump_data *dump_data;
 675         struct iwl_fw_error_dump_info *dump_info;
 676         struct iwl_mvm_dump_ptrs *fw_error_dump;
 677         const struct fw_img *img;
 678         u32 sram_len, sram_ofs;
 679         u32 file_len, rxf_len;
 680         unsigned long flags;
 681         int reg_val;
 682
 683         lockdep_assert_held(&mvm->mutex);
 684
 685         if (mvm->fw_error_dump)
 686                 return;
 687
 688         fw_error_dump = kzalloc(sizeof(*mvm->fw_error_dump), GFP_KERNEL);
 689         if (!fw_error_dump)
 690                 return;
 691
 692         img = &mvm->fw->img[mvm->cur_ucode];
 693         sram_ofs = img->sec[IWL_UCODE_SECTION_DATA].offset;
 694         sram_len = img->sec[IWL_UCODE_SECTION_DATA].len;
 695
 696         /* reading buffer size */
 697         reg_val = iwl_trans_read_prph(mvm->trans, RXF_SIZE_ADDR);
 698         rxf_len = (reg_val & RXF_SIZE_BYTE_CNT_MSK) >> RXF_SIZE_BYTE_CND_POS;
 699
 700         /* the register holds the value divided by 128 */
 701         rxf_len = rxf_len << 7;
 702
 703         file_len = sizeof(*dump_file) +
 704                    sizeof(*dump_data) * 3 +
 705                    sram_len +
 706                    rxf_len +
 707                    sizeof(*dump_info);
 708
 709         dump_file = vzalloc(file_len);
 710         if (!dump_file) {
 711                 kfree(fw_error_dump);
 712                 return;
 713         }
 714
 715         fw_error_dump->op_mode_ptr = dump_file;
 716
 717         dump_file->barker = cpu_to_le32(IWL_FW_ERROR_DUMP_BARKER);
 718         dump_data = (void *)dump_file->data;
 719
 720         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_DEV_FW_INFO);
 721         dump_data->len = cpu_to_le32(sizeof(*dump_info));
 722         dump_info = (void *) dump_data->data;
 723         dump_info->device_family =
 724                 mvm->cfg->device_family == IWL_DEVICE_FAMILY_7000 ?
 725                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_7) :
 726                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_8);
 727         memcpy(dump_info->fw_human_readable, mvm->fw->human_readable,
 728                sizeof(dump_info->fw_human_readable));
 729         strncpy(dump_info->dev_human_readable, mvm->cfg->name,
 730                 sizeof(dump_info->dev_human_readable));
 731         strncpy(dump_info->bus_human_readable, mvm->dev->bus->name,
 732                 sizeof(dump_info->bus_human_readable));
 733
 734         dump_data = iwl_fw_error_next_data(dump_data);
 735         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_RXF);
 736         dump_data->len = cpu_to_le32(rxf_len);
 737
 738         if (iwl_trans_grab_nic_access(mvm->trans, false, &flags)) {
 739                 u32 *rxf = (void *)dump_data->data;
 740                 int i;
 741
 742                 for (i = 0; i < (rxf_len / sizeof(u32)); i++) {
 743                         iwl_trans_write_prph(mvm->trans,
 744                                              RXF_LD_FENCE_OFFSET_ADDR,
 745                                              i * sizeof(u32));
 746                         rxf[i] = iwl_trans_read_prph(mvm->trans,
 747                                                      RXF_FIFO_RD_FENCE_ADDR);
 748                 }
 749                 iwl_trans_release_nic_access(mvm->trans, &flags);
 750         }
 751
 752         dump_data = iwl_fw_error_next_data(dump_data);
 753         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_SRAM);
 754         dump_data->len = cpu_to_le32(sram_len);
 755         iwl_trans_read_mem_bytes(mvm->trans, sram_ofs, dump_data->data,
 756                                  sram_len);
 757
 758         fw_error_dump->trans_ptr = iwl_trans_dump_data(mvm->trans);
 759         fw_error_dump->op_mode_len = file_len;
 760         if (fw_error_dump->trans_ptr)
 761                 file_len += fw_error_dump->trans_ptr->len;
 762         dump_file->file_len = cpu_to_le32(file_len);
 763         mvm->fw_error_dump = fw_error_dump;
 764 }
 765 #endif
 766
 767 static void iwl_mvm_restart_cleanup(struct iwl_mvm *mvm)
 768 {
 769 #ifdef CONFIG_IWLWIFI_DEBUGFS
 770         static char *env[] = { "DRIVER=iwlwifi", "EVENT=error_dump", NULL };
 771
 772         iwl_mvm_fw_error_dump(mvm);
 773
 774         /* notify the userspace about the error we had */
 775         kobject_uevent_env(&mvm->hw->wiphy->dev.kobj, KOBJ_CHANGE, env);
 776 #endif
 777
 778         iwl_trans_stop_device(mvm->trans);
 779
 780         mvm->scan_status = IWL_MVM_SCAN_NONE;
 781         mvm->ps_disabled = false;
 782
 783         /* just in case one was running */
 784         ieee80211_remain_on_channel_expired(mvm->hw);
 785
 786         ieee80211_iterate_active_interfaces_atomic(
 787                 mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
 788                 iwl_mvm_cleanup_iterator, mvm);
 789
 790         mvm->p2p_device_vif = NULL;
 791         mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
 792
 793         iwl_mvm_reset_phy_ctxts(mvm);
 794         memset(mvm->fw_key_table, 0, sizeof(mvm->fw_key_table));
 795         memset(mvm->sta_drained, 0, sizeof(mvm->sta_drained));
 796         memset(&mvm->last_bt_notif, 0, sizeof(mvm->last_bt_notif));
 797         memset(&mvm->last_bt_notif_old, 0, sizeof(mvm->last_bt_notif_old));
 798         memset(&mvm->last_bt_ci_cmd, 0, sizeof(mvm->last_bt_ci_cmd));
 799         memset(&mvm->last_bt_ci_cmd_old, 0, sizeof(mvm->last_bt_ci_cmd_old));
 800         memset(&mvm->bt_ack_kill_msk, 0, sizeof(mvm->bt_ack_kill_msk));
 801         memset(&mvm->bt_cts_kill_msk, 0, sizeof(mvm->bt_cts_kill_msk));
 802
 803         ieee80211_wake_queues(mvm->hw);
 804
 805         /* cleanup all stale references (scan, roc), but keep the
 806          * ucode_down ref until reconfig is complete */
 807         iwl_mvm_unref_all_except(mvm, IWL_MVM_REF_UCODE_DOWN);
 808
 809         /* clear any stale d0i3 state */
 810         clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
 811
 812         mvm->vif_count = 0;
 813         mvm->rx_ba_sessions = 0;
 814 }
 815
 816 static int iwl_mvm_mac_start(struct ieee80211_hw *hw)
 817 {
 818         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 819         int ret;
 820
 821         mutex_lock(&mvm->mutex);
 822
 823         /* Clean up some internal and mac80211 state on restart */
 824         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
 825                 iwl_mvm_restart_cleanup(mvm);
 826
 827         ret = iwl_mvm_up(mvm);
 828
 829         if (ret && test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
 830                 /* Something went wrong - we need to finish some cleanup
 831                  * that normally iwl_mvm_mac_restart_complete() below
 832                  * would do.
 833                  */
 834                 clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
 835                 iwl_mvm_d0i3_enable_tx(mvm, NULL);
 836         }
 837
 838         mutex_unlock(&mvm->mutex);
 839
 840         return ret;
 841 }
 842
 843 static void iwl_mvm_mac_restart_complete(struct ieee80211_hw *hw)
 844 {
 845         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 846         int ret;
 847
 848         mutex_lock(&mvm->mutex);
 849
 850         clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
 851         iwl_mvm_d0i3_enable_tx(mvm, NULL);
 852         ret = iwl_mvm_update_quotas(mvm, NULL);
 853         if (ret)
 854                 IWL_ERR(mvm, "Failed to update quotas after restart (%d)\n",
 855                         ret);
 856
 857         /* allow transport/FW low power modes */
 858         iwl_mvm_unref(mvm, IWL_MVM_REF_UCODE_DOWN);
 859
 860         mutex_unlock(&mvm->mutex);
 861 }
 862
 863 static void iwl_mvm_mac_stop(struct ieee80211_hw *hw)
 864 {
 865         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 866
 867         flush_work(&mvm->d0i3_exit_work);
 868         flush_work(&mvm->async_handlers_wk);
 869
 870         mutex_lock(&mvm->mutex);
 871
 872         /* disallow low power states when the FW is down */
 873         iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
 874
 875         /* async_handlers_wk is now blocked */
 876
 877         /*
 878          * The work item could be running or queued if the
 879          * ROC time event stops just as we get here.
 880          */
 881         cancel_work_sync(&mvm->roc_done_wk);
 882
 883         iwl_trans_stop_device(mvm->trans);
 884
 885         iwl_mvm_async_handlers_purge(mvm);
 886         /* async_handlers_list is empty and will stay empty: HW is stopped */
 887
 888         /* the fw is stopped, the aux sta is dead: clean up driver state */
 889         iwl_mvm_dealloc_int_sta(mvm, &mvm->aux_sta);
 890
 891         mutex_unlock(&mvm->mutex);
 892
 893         /*
 894          * The worker might have been waiting for the mutex, let it run and
 895          * discover that its list is now empty.
 896          */
 897         cancel_work_sync(&mvm->async_handlers_wk);
 898 }
 899
 900 static struct iwl_mvm_phy_ctxt *iwl_mvm_get_free_phy_ctxt(struct iwl_mvm *mvm)
 901 {
 902         u16 i;
 903
 904         lockdep_assert_held(&mvm->mutex);
 905
 906         for (i = 0; i < NUM_PHY_CTX; i++)
 907                 if (!mvm->phy_ctxts[i].ref)
 908                         return &mvm->phy_ctxts[i];
 909
 910         IWL_ERR(mvm, "No available PHY context\n");
 911         return NULL;
 912 }
 913
 914 static int iwl_mvm_set_tx_power(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
 915                                 s8 tx_power)
 916 {
 917         /* FW is in charge of regulatory enforcement */
 918         struct iwl_reduce_tx_power_cmd reduce_txpwr_cmd = {
 919                 .mac_context_id = iwl_mvm_vif_from_mac80211(vif)->id,
 920                 .pwr_restriction = cpu_to_le16(tx_power),
 921         };
 922
 923         return iwl_mvm_send_cmd_pdu(mvm, REDUCE_TX_POWER_CMD, 0,
 924                                     sizeof(reduce_txpwr_cmd),
 925                                     &reduce_txpwr_cmd);
 926 }
 927
 928 static int iwl_mvm_mac_add_interface(struct ieee80211_hw *hw,
 929                                      struct ieee80211_vif *vif)
 930 {
 931         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
 932         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
 933         int ret;
 934
 935         /*
 936          * make sure D0i3 exit is completed, otherwise a target access
 937          * during tx queue configuration could be done when still in
 938          * D0i3 state.
 939          */
 940         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_ADD_IF);
 941         if (ret)
 942                 return ret;
 943
 944         /*
 945          * Not much to do here. The stack will not allow interface
 946          * types or combinations that we didn't advertise, so we
 947          * don't really have to check the types.
 948          */
 949
 950         mutex_lock(&mvm->mutex);
 951
 952         /* Allocate resources for the MAC context, and add it to the fw  */
 953         ret = iwl_mvm_mac_ctxt_init(mvm, vif);
 954         if (ret)
 955                 goto out_unlock;
 956
 957         /* Counting number of interfaces is needed for legacy PM */
 958         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
 959                 mvm->vif_count++;
 960
 961         /*
 962          * The AP binding flow can be done only after the beacon
 963          * template is configured (which happens only in the mac80211
 964          * start_ap() flow), and adding the broadcast station can happen
 965          * only after the binding.
 966          * In addition, since modifying the MAC before adding a bcast
 967          * station is not allowed by the FW, delay the adding of MAC context to
 968          * the point where we can also add the bcast station.
 969          * In short: there's not much we can do at this point, other than
 970          * allocating resources :)
 971          */
 972         if (vif->type == NL80211_IFTYPE_AP ||
 973             vif->type == NL80211_IFTYPE_ADHOC) {
 974                 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
 975                 if (ret) {
 976                         IWL_ERR(mvm, "Failed to allocate bcast sta\n");
 977                         goto out_release;
 978                 }
 979
 980                 iwl_mvm_vif_dbgfs_register(mvm, vif);
 981                 goto out_unlock;
 982         }
 983
 984         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
 985         if (ret)
 986                 goto out_release;
 987
 988         ret = iwl_mvm_power_update_mac(mvm);
 989         if (ret)
 990                 goto out_release;
 991
 992         /* beacon filtering */
 993         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
 994         if (ret)
 995                 goto out_remove_mac;
 996
 997         if (!mvm->bf_allowed_vif &&
 998             vif->type == NL80211_IFTYPE_STATION && !vif->p2p) {
 999                 mvm->bf_allowed_vif = mvmvif;
1000                 vif->driver_flags |= IEEE80211_VIF_BEACON_FILTER |
1001                                      IEEE80211_VIF_SUPPORTS_CQM_RSSI;
1002         }
1003
1004         /*
1005          * P2P_DEVICE interface does not have a channel context assigned to it,
1006          * so a dedicated PHY context is allocated to it and the corresponding
1007          * MAC context is bound to it at this stage.
1008          */
1009         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1010
1011                 mvmvif->phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
1012                 if (!mvmvif->phy_ctxt) {
1013                         ret = -ENOSPC;
1014                         goto out_free_bf;
1015                 }
1016
1017                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
1018                 ret = iwl_mvm_binding_add_vif(mvm, vif);
1019                 if (ret)
1020                         goto out_unref_phy;
1021
1022                 ret = iwl_mvm_add_bcast_sta(mvm, vif);
1023                 if (ret)
1024                         goto out_unbind;
1025
1026                 /* Save a pointer to p2p device vif, so it can later be used to
1027                  * update the p2p device MAC when a GO is started/stopped */
1028                 mvm->p2p_device_vif = vif;
1029         }
1030
1031         iwl_mvm_vif_dbgfs_register(mvm, vif);
1032         goto out_unlock;
1033
1034  out_unbind:
1035         iwl_mvm_binding_remove_vif(mvm, vif);
1036  out_unref_phy:
1037         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1038  out_free_bf:
1039         if (mvm->bf_allowed_vif == mvmvif) {
1040                 mvm->bf_allowed_vif = NULL;
1041                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1042                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1043         }
1044  out_remove_mac:
1045         mvmvif->phy_ctxt = NULL;
1046         iwl_mvm_mac_ctxt_remove(mvm, vif);
1047  out_release:
1048         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1049                 mvm->vif_count--;
1050
1051         iwl_mvm_mac_ctxt_release(mvm, vif);
1052  out_unlock:
1053         mutex_unlock(&mvm->mutex);
1054
1055         iwl_mvm_unref(mvm, IWL_MVM_REF_ADD_IF);
1056
1057         return ret;
1058 }
1059
1060 static void iwl_mvm_prepare_mac_removal(struct iwl_mvm *mvm,
1061                                         struct ieee80211_vif *vif)
1062 {
1063         u32 tfd_msk = iwl_mvm_mac_get_queues_mask(mvm, vif);
1064
1065         if (tfd_msk) {
1066                 mutex_lock(&mvm->mutex);
1067                 iwl_mvm_flush_tx_path(mvm, tfd_msk, true);
1068                 mutex_unlock(&mvm->mutex);
1069         }
1070
1071         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1072                 /*
1073                  * Flush the ROC worker which will flush the OFFCHANNEL queue.
1074                  * We assume here that all the packets sent to the OFFCHANNEL
1075                  * queue are sent in ROC session.
1076                  */
1077                 flush_work(&mvm->roc_done_wk);
1078         } else {
1079                 /*
1080                  * By now, all the AC queues are empty. The AGG queues are
1081                  * empty too. We already got all the Tx responses for all the
1082                  * packets in the queues. The drain work can have been
1083                  * triggered. Flush it.
1084                  */
1085                 flush_work(&mvm->sta_drained_wk);
1086         }
1087 }
1088
1089 static void iwl_mvm_mac_remove_interface(struct ieee80211_hw *hw,
1090                                          struct ieee80211_vif *vif)
1091 {
1092         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1093         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1094
1095         iwl_mvm_prepare_mac_removal(mvm, vif);
1096
1097         mutex_lock(&mvm->mutex);
1098
1099         if (mvm->bf_allowed_vif == mvmvif) {
1100                 mvm->bf_allowed_vif = NULL;
1101                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1102                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1103         }
1104
1105         iwl_mvm_vif_dbgfs_clean(mvm, vif);
1106
1107         /*
1108          * For AP/GO interface, the tear down of the resources allocated to the
1109          * interface is be handled as part of the stop_ap flow.
1110          */
1111         if (vif->type == NL80211_IFTYPE_AP ||
1112             vif->type == NL80211_IFTYPE_ADHOC) {
1113 #ifdef CONFIG_NL80211_TESTMODE
1114                 if (vif == mvm->noa_vif) {
1115                         mvm->noa_vif = NULL;
1116                         mvm->noa_duration = 0;
1117                 }
1118 #endif
1119                 iwl_mvm_dealloc_bcast_sta(mvm, vif);
1120                 goto out_release;
1121         }
1122
1123         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1124                 mvm->p2p_device_vif = NULL;
1125                 iwl_mvm_rm_bcast_sta(mvm, vif);
1126                 iwl_mvm_binding_remove_vif(mvm, vif);
1127                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1128                 mvmvif->phy_ctxt = NULL;
1129         }
1130
1131         if (mvm->vif_count && vif->type != NL80211_IFTYPE_P2P_DEVICE)
1132                 mvm->vif_count--;
1133
1134         iwl_mvm_power_update_mac(mvm);
1135         iwl_mvm_mac_ctxt_remove(mvm, vif);
1136
1137 out_release:
1138         iwl_mvm_mac_ctxt_release(mvm, vif);
1139         mutex_unlock(&mvm->mutex);
1140 }
1141
1142 static int iwl_mvm_mac_config(struct ieee80211_hw *hw, u32 changed)
1143 {
1144         return 0;
1145 }
1146
1147 struct iwl_mvm_mc_iter_data {
1148         struct iwl_mvm *mvm;
1149         int port_id;
1150 };
1151
1152 static void iwl_mvm_mc_iface_iterator(void *_data, u8 *mac,
1153                                       struct ieee80211_vif *vif)
1154 {
1155         struct iwl_mvm_mc_iter_data *data = _data;
1156         struct iwl_mvm *mvm = data->mvm;
1157         struct iwl_mcast_filter_cmd *cmd = mvm->mcast_filter_cmd;
1158         int ret, len;
1159
1160         /* if we don't have free ports, mcast frames will be dropped */
1161         if (WARN_ON_ONCE(data->port_id >= MAX_PORT_ID_NUM))
1162                 return;
1163
1164         if (vif->type != NL80211_IFTYPE_STATION ||
1165             !vif->bss_conf.assoc)
1166                 return;
1167
1168         cmd->port_id = data->port_id++;
1169         memcpy(cmd->bssid, vif->bss_conf.bssid, ETH_ALEN);
1170         len = roundup(sizeof(*cmd) + cmd->count * ETH_ALEN, 4);
1171
1172         ret = iwl_mvm_send_cmd_pdu(mvm, MCAST_FILTER_CMD, CMD_ASYNC, len, cmd);
1173         if (ret)
1174                 IWL_ERR(mvm, "mcast filter cmd error. ret=%d\n", ret);
1175 }
1176
1177 static void iwl_mvm_recalc_multicast(struct iwl_mvm *mvm)
1178 {
1179         struct iwl_mvm_mc_iter_data iter_data = {
1180                 .mvm = mvm,
1181         };
1182
1183         lockdep_assert_held(&mvm->mutex);
1184
1185         if (WARN_ON_ONCE(!mvm->mcast_filter_cmd))
1186                 return;
1187
1188         ieee80211_iterate_active_interfaces_atomic(
1189                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1190                 iwl_mvm_mc_iface_iterator, &iter_data);
1191 }
1192
1193 static u64 iwl_mvm_prepare_multicast(struct ieee80211_hw *hw,
1194                                      struct netdev_hw_addr_list *mc_list)
1195 {
1196         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1197         struct iwl_mcast_filter_cmd *cmd;
1198         struct netdev_hw_addr *addr;
1199         int addr_count = netdev_hw_addr_list_count(mc_list);
1200         bool pass_all = false;
1201         int len;
1202
1203         if (addr_count > MAX_MCAST_FILTERING_ADDRESSES) {
1204                 pass_all = true;
1205                 addr_count = 0;
1206         }
1207
1208         len = roundup(sizeof(*cmd) + addr_count * ETH_ALEN, 4);
1209         cmd = kzalloc(len, GFP_ATOMIC);
1210         if (!cmd)
1211                 return 0;
1212
1213         if (pass_all) {
1214                 cmd->pass_all = 1;
1215                 return (u64)(unsigned long)cmd;
1216         }
1217
1218         netdev_hw_addr_list_for_each(addr, mc_list) {
1219                 IWL_DEBUG_MAC80211(mvm, "mcast addr (%d): %pM\n",
1220                                    cmd->count, addr->addr);
1221                 memcpy(&cmd->addr_list[cmd->count * ETH_ALEN],
1222                        addr->addr, ETH_ALEN);
1223                 cmd->count++;
1224         }
1225
1226         return (u64)(unsigned long)cmd;
1227 }
1228
1229 static void iwl_mvm_configure_filter(struct ieee80211_hw *hw,
1230                                      unsigned int changed_flags,
1231                                      unsigned int *total_flags,
1232                                      u64 multicast)
1233 {
1234         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1235         struct iwl_mcast_filter_cmd *cmd = (void *)(unsigned long)multicast;
1236
1237         mutex_lock(&mvm->mutex);
1238
1239         /* replace previous configuration */
1240         kfree(mvm->mcast_filter_cmd);
1241         mvm->mcast_filter_cmd = cmd;
1242
1243         if (!cmd)
1244                 goto out;
1245
1246         iwl_mvm_recalc_multicast(mvm);
1247 out:
1248         mutex_unlock(&mvm->mutex);
1249         *total_flags = 0;
1250 }
1251
1252 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
1253 struct iwl_bcast_iter_data {
1254         struct iwl_mvm *mvm;
1255         struct iwl_bcast_filter_cmd *cmd;
1256         u8 current_filter;
1257 };
1258
1259 static void
1260 iwl_mvm_set_bcast_filter(struct ieee80211_vif *vif,
1261                          const struct iwl_fw_bcast_filter *in_filter,
1262                          struct iwl_fw_bcast_filter *out_filter)
1263 {
1264         struct iwl_fw_bcast_filter_attr *attr;
1265         int i;
1266
1267         memcpy(out_filter, in_filter, sizeof(*out_filter));
1268
1269         for (i = 0; i < ARRAY_SIZE(out_filter->attrs); i++) {
1270                 attr = &out_filter->attrs[i];
1271
1272                 if (!attr->mask)
1273                         break;
1274
1275                 switch (attr->reserved1) {
1276                 case cpu_to_le16(BC_FILTER_MAGIC_IP):
1277                         if (vif->bss_conf.arp_addr_cnt != 1) {
1278                                 attr->mask = 0;
1279                                 continue;
1280                         }
1281
1282                         attr->val = vif->bss_conf.arp_addr_list[0];
1283                         break;
1284                 case cpu_to_le16(BC_FILTER_MAGIC_MAC):
1285                         attr->val = *(__be32 *)&vif->addr[2];
1286                         break;
1287                 default:
1288                         break;
1289                 }
1290                 attr->reserved1 = 0;
1291                 out_filter->num_attrs++;
1292         }
1293 }
1294
1295 static void iwl_mvm_bcast_filter_iterator(void *_data, u8 *mac,
1296                                           struct ieee80211_vif *vif)
1297 {
1298         struct iwl_bcast_iter_data *data = _data;
1299         struct iwl_mvm *mvm = data->mvm;
1300         struct iwl_bcast_filter_cmd *cmd = data->cmd;
1301         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1302         struct iwl_fw_bcast_mac *bcast_mac;
1303         int i;
1304
1305         if (WARN_ON(mvmvif->id >= ARRAY_SIZE(cmd->macs)))
1306                 return;
1307
1308         bcast_mac = &cmd->macs[mvmvif->id];
1309
1310         /*
1311          * enable filtering only for associated stations, but not for P2P
1312          * Clients
1313          */
1314         if (vif->type != NL80211_IFTYPE_STATION || vif->p2p ||
1315             !vif->bss_conf.assoc)
1316                 return;
1317
1318         bcast_mac->default_discard = 1;
1319
1320         /* copy all configured filters */
1321         for (i = 0; mvm->bcast_filters[i].attrs[0].mask; i++) {
1322                 /*
1323                  * Make sure we don't exceed our filters limit.
1324                  * if there is still a valid filter to be configured,
1325                  * be on the safe side and just allow bcast for this mac.
1326                  */
1327                 if (WARN_ON_ONCE(data->current_filter >=
1328                                  ARRAY_SIZE(cmd->filters))) {
1329                         bcast_mac->default_discard = 0;
1330                         bcast_mac->attached_filters = 0;
1331                         break;
1332                 }
1333
1334                 iwl_mvm_set_bcast_filter(vif,
1335                                          &mvm->bcast_filters[i],
1336                                          &cmd->filters[data->current_filter]);
1337
1338                 /* skip current filter if it contains no attributes */
1339                 if (!cmd->filters[data->current_filter].num_attrs)
1340                         continue;
1341
1342                 /* attach the filter to current mac */
1343                 bcast_mac->attached_filters |=
1344                                 cpu_to_le16(BIT(data->current_filter));
1345
1346                 data->current_filter++;
1347         }
1348 }
1349
1350 bool iwl_mvm_bcast_filter_build_cmd(struct iwl_mvm *mvm,
1351                                     struct iwl_bcast_filter_cmd *cmd)
1352 {
1353         struct iwl_bcast_iter_data iter_data = {
1354                 .mvm = mvm,
1355                 .cmd = cmd,
1356         };
1357
1358         memset(cmd, 0, sizeof(*cmd));
1359         cmd->max_bcast_filters = ARRAY_SIZE(cmd->filters);
1360         cmd->max_macs = ARRAY_SIZE(cmd->macs);
1361
1362 #ifdef CONFIG_IWLWIFI_DEBUGFS
1363         /* use debugfs filters/macs if override is configured */
1364         if (mvm->dbgfs_bcast_filtering.override) {
1365                 memcpy(cmd->filters, &mvm->dbgfs_bcast_filtering.cmd.filters,
1366                        sizeof(cmd->filters));
1367                 memcpy(cmd->macs, &mvm->dbgfs_bcast_filtering.cmd.macs,
1368                        sizeof(cmd->macs));
1369                 return true;
1370         }
1371 #endif
1372
1373         /* if no filters are configured, do nothing */
1374         if (!mvm->bcast_filters)
1375                 return false;
1376
1377         /* configure and attach these filters for each associated sta vif */
1378         ieee80211_iterate_active_interfaces(
1379                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1380                 iwl_mvm_bcast_filter_iterator, &iter_data);
1381
1382         return true;
1383 }
1384 static int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
1385                                           struct ieee80211_vif *vif)
1386 {
1387         struct iwl_bcast_filter_cmd cmd;
1388
1389         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_BCAST_FILTERING))
1390                 return 0;
1391
1392         if (!iwl_mvm_bcast_filter_build_cmd(mvm, &cmd))
1393                 return 0;
1394
1395         return iwl_mvm_send_cmd_pdu(mvm, BCAST_FILTER_CMD, 0,
1396                                     sizeof(cmd), &cmd);
1397 }
1398 #else
1399 static inline int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
1400                                                  struct ieee80211_vif *vif)
1401 {
1402         return 0;
1403 }
1404 #endif
1405
1406 static void iwl_mvm_teardown_tdls_peers(struct iwl_mvm *mvm)
1407 {
1408         struct ieee80211_sta *sta;
1409         struct iwl_mvm_sta *mvmsta;
1410         int i;
1411
1412         lockdep_assert_held(&mvm->mutex);
1413
1414         for (i = 0; i < IWL_MVM_STATION_COUNT; i++) {
1415                 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[i],
1416                                                 lockdep_is_held(&mvm->mutex));
1417                 if (!sta || IS_ERR(sta) || !sta->tdls)
1418                         continue;
1419
1420                 mvmsta = iwl_mvm_sta_from_mac80211(sta);
1421                 ieee80211_tdls_oper_request(mvmsta->vif, sta->addr,
1422                                 NL80211_TDLS_TEARDOWN,
1423                                 WLAN_REASON_TDLS_TEARDOWN_UNSPECIFIED,
1424                                 GFP_KERNEL);
1425         }
1426 }
1427
1428 static void iwl_mvm_bss_info_changed_station(struct iwl_mvm *mvm,
1429                                              struct ieee80211_vif *vif,
1430                                              struct ieee80211_bss_conf *bss_conf,
1431                                              u32 changes)
1432 {
1433         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1434         int ret;
1435
1436         /*
1437          * Re-calculate the tsf id, as the master-slave relations depend on the
1438          * beacon interval, which was not known when the station interface was
1439          * added.
1440          */
1441         if (changes & BSS_CHANGED_ASSOC && bss_conf->assoc)
1442                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
1443
1444         /*
1445          * If we're not associated yet, take the (new) BSSID before associating
1446          * so the firmware knows. If we're already associated, then use the old
1447          * BSSID here, and we'll send a cleared one later in the CHANGED_ASSOC
1448          * branch for disassociation below.
1449          */
1450         if (changes & BSS_CHANGED_BSSID && !mvmvif->associated)
1451                 memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1452
1453         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, mvmvif->bssid);
1454         if (ret)
1455                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
1456
1457         /* after sending it once, adopt mac80211 data */
1458         memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1459         mvmvif->associated = bss_conf->assoc;
1460
1461         if (changes & BSS_CHANGED_ASSOC) {
1462                 if (bss_conf->assoc) {
1463                         /* add quota for this interface */
1464                         ret = iwl_mvm_update_quotas(mvm, NULL);
1465                         if (ret) {
1466                                 IWL_ERR(mvm, "failed to update quotas\n");
1467                                 return;
1468                         }
1469
1470                         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART,
1471                                      &mvm->status)) {
1472                                 /*
1473                                  * If we're restarting then the firmware will
1474                                  * obviously have lost synchronisation with
1475                                  * the AP. It will attempt to synchronise by
1476                                  * itself, but we can make it more reliable by
1477                                  * scheduling a session protection time event.
1478                                  *
1479                                  * The firmware needs to receive a beacon to
1480                                  * catch up with synchronisation, use 110% of
1481                                  * the beacon interval.
1482                                  *
1483                                  * Set a large maximum delay to allow for more
1484                                  * than a single interface.
1485                                  */
1486                                 u32 dur = (11 * vif->bss_conf.beacon_int) / 10;
1487                                 iwl_mvm_protect_session(mvm, vif, dur, dur,
1488                                                         5 * dur, false);
1489                         }
1490
1491                         iwl_mvm_sf_update(mvm, vif, false);
1492                         iwl_mvm_power_vif_assoc(mvm, vif);
1493                         if (vif->p2p) {
1494                                 iwl_mvm_ref(mvm, IWL_MVM_REF_P2P_CLIENT);
1495                                 iwl_mvm_update_smps(mvm, vif,
1496                                                     IWL_MVM_SMPS_REQ_PROT,
1497                                                     IEEE80211_SMPS_DYNAMIC);
1498                         }
1499                 } else if (mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
1500                         /*
1501                          * If update fails - SF might be running in associated
1502                          * mode while disassociated - which is forbidden.
1503                          */
1504                         WARN_ONCE(iwl_mvm_sf_update(mvm, vif, false),
1505                                   "Failed to update SF upon disassociation\n");
1506
1507                         /* remove AP station now that the MAC is unassoc */
1508                         ret = iwl_mvm_rm_sta_id(mvm, vif, mvmvif->ap_sta_id);
1509                         if (ret)
1510                                 IWL_ERR(mvm, "failed to remove AP station\n");
1511
1512                         if (mvm->d0i3_ap_sta_id == mvmvif->ap_sta_id)
1513                                 mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1514                         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
1515                         /* remove quota for this interface */
1516                         ret = iwl_mvm_update_quotas(mvm, NULL);
1517                         if (ret)
1518                                 IWL_ERR(mvm, "failed to update quotas\n");
1519
1520                         if (vif->p2p)
1521                                 iwl_mvm_unref(mvm, IWL_MVM_REF_P2P_CLIENT);
1522
1523                         /* this will take the cleared BSSID from bss_conf */
1524                         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
1525                         if (ret)
1526                                 IWL_ERR(mvm,
1527                                         "failed to update MAC %pM (clear after unassoc)\n",
1528                                         vif->addr);
1529                 }
1530
1531                 iwl_mvm_recalc_multicast(mvm);
1532                 iwl_mvm_configure_bcast_filter(mvm, vif);
1533
1534                 /* reset rssi values */
1535                 mvmvif->bf_data.ave_beacon_signal = 0;
1536
1537                 iwl_mvm_bt_coex_vif_change(mvm);
1538                 iwl_mvm_update_smps(mvm, vif, IWL_MVM_SMPS_REQ_TT,
1539                                     IEEE80211_SMPS_AUTOMATIC);
1540         } else if (changes & BSS_CHANGED_BEACON_INFO) {
1541                 /*
1542                  * We received a beacon _after_ association so
1543                  * remove the session protection.
1544                  */
1545                 iwl_mvm_remove_time_event(mvm, mvmvif,
1546                                           &mvmvif->time_event_data);
1547         } else if (changes & (BSS_CHANGED_PS | BSS_CHANGED_P2P_PS |
1548                               BSS_CHANGED_QOS)) {
1549                 ret = iwl_mvm_power_update_mac(mvm);
1550                 if (ret)
1551                         IWL_ERR(mvm, "failed to update power mode\n");
1552         }
1553
1554         if (changes & BSS_CHANGED_BEACON_INFO) {
1555                 iwl_mvm_sf_update(mvm, vif, false);
1556                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
1557         }
1558
1559         if (changes & BSS_CHANGED_TXPOWER) {
1560                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
1561                                 bss_conf->txpower);
1562                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
1563         }
1564
1565         if (changes & BSS_CHANGED_CQM) {
1566                 IWL_DEBUG_MAC80211(mvm, "cqm info_changed\n");
1567                 /* reset cqm events tracking */
1568                 mvmvif->bf_data.last_cqm_event = 0;
1569                 if (mvmvif->bf_data.bf_enabled) {
1570                         ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
1571                         if (ret)
1572                                 IWL_ERR(mvm,
1573                                         "failed to update CQM thresholds\n");
1574                 }
1575         }
1576
1577         if (changes & BSS_CHANGED_ARP_FILTER) {
1578                 IWL_DEBUG_MAC80211(mvm, "arp filter changed\n");
1579                 iwl_mvm_configure_bcast_filter(mvm, vif);
1580         }
1581 }
1582
1583 static int iwl_mvm_start_ap_ibss(struct ieee80211_hw *hw,
1584                                  struct ieee80211_vif *vif)
1585 {
1586         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1587         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1588         int ret;
1589
1590         /*
1591          * iwl_mvm_mac_ctxt_add() might read directly from the device
1592          * (the system time), so make sure it is available.
1593          */
1594         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_START_AP);
1595         if (ret)
1596                 return ret;
1597
1598         mutex_lock(&mvm->mutex);
1599
1600         /* Send the beacon template */
1601         ret = iwl_mvm_mac_ctxt_beacon_changed(mvm, vif);
1602         if (ret)
1603                 goto out_unlock;
1604
1605         /*
1606          * Re-calculate the tsf id, as the master-slave relations depend on the
1607          * beacon interval, which was not known when the AP interface was added.
1608          */
1609         if (vif->type == NL80211_IFTYPE_AP)
1610                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
1611
1612         /* Add the mac context */
1613         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
1614         if (ret)
1615                 goto out_unlock;
1616
1617         /* Perform the binding */
1618         ret = iwl_mvm_binding_add_vif(mvm, vif);
1619         if (ret)
1620                 goto out_remove;
1621
1622         /* Send the bcast station. At this stage the TBTT and DTIM time events
1623          * are added and applied to the scheduler */
1624         ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
1625         if (ret)
1626                 goto out_unbind;
1627
1628         /* must be set before quota calculations */
1629         mvmvif->ap_ibss_active = true;
1630
1631         /* power updated needs to be done before quotas */
1632         iwl_mvm_power_update_mac(mvm);
1633
1634         ret = iwl_mvm_update_quotas(mvm, NULL);
1635         if (ret)
1636                 goto out_quota_failed;
1637
1638         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
1639         if (vif->p2p && mvm->p2p_device_vif)
1640                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
1641
1642         iwl_mvm_ref(mvm, IWL_MVM_REF_AP_IBSS);
1643
1644         iwl_mvm_bt_coex_vif_change(mvm);
1645
1646         /* we don't support TDLS during DCM */
1647         if (iwl_mvm_phy_ctx_count(mvm) > 1)
1648                 iwl_mvm_teardown_tdls_peers(mvm);
1649
1650         mutex_unlock(&mvm->mutex);
1651         return 0;
1652
1653 out_quota_failed:
1654         iwl_mvm_power_update_mac(mvm);
1655         mvmvif->ap_ibss_active = false;
1656         iwl_mvm_send_rm_bcast_sta(mvm, vif);
1657 out_unbind:
1658         iwl_mvm_binding_remove_vif(mvm, vif);
1659 out_remove:
1660         iwl_mvm_mac_ctxt_remove(mvm, vif);
1661 out_unlock:
1662         mutex_unlock(&mvm->mutex);
1663         iwl_mvm_unref(mvm, IWL_MVM_REF_START_AP);
1664         return ret;
1665 }
1666
1667 static void iwl_mvm_stop_ap_ibss(struct ieee80211_hw *hw,
1668                                  struct ieee80211_vif *vif)
1669 {
1670         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1671         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1672
1673         iwl_mvm_prepare_mac_removal(mvm, vif);
1674
1675         mutex_lock(&mvm->mutex);
1676
1677         /* Handle AP stop while in CSA */
1678         if (rcu_access_pointer(mvm->csa_vif) == vif) {
1679                 iwl_mvm_remove_time_event(mvm, mvmvif,
1680                                           &mvmvif->time_event_data);
1681                 RCU_INIT_POINTER(mvm->csa_vif, NULL);
1682         }
1683
1684         if (rcu_access_pointer(mvm->csa_tx_blocked_vif) == vif) {
1685                 RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
1686                 mvm->csa_tx_block_bcn_timeout = 0;
1687         }
1688
1689         mvmvif->ap_ibss_active = false;
1690         mvm->ap_last_beacon_gp2 = 0;
1691
1692         iwl_mvm_bt_coex_vif_change(mvm);
1693
1694         iwl_mvm_unref(mvm, IWL_MVM_REF_AP_IBSS);
1695
1696         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
1697         if (vif->p2p && mvm->p2p_device_vif)
1698                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
1699
1700         iwl_mvm_update_quotas(mvm, NULL);
1701         iwl_mvm_send_rm_bcast_sta(mvm, vif);
1702         iwl_mvm_binding_remove_vif(mvm, vif);
1703
1704         iwl_mvm_power_update_mac(mvm);
1705
1706         iwl_mvm_mac_ctxt_remove(mvm, vif);
1707
1708         mutex_unlock(&mvm->mutex);
1709 }
1710
1711 static void
1712 iwl_mvm_bss_info_changed_ap_ibss(struct iwl_mvm *mvm,
1713                                  struct ieee80211_vif *vif,
1714                                  struct ieee80211_bss_conf *bss_conf,
1715                                  u32 changes)
1716 {
1717         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1718
1719         /* Changes will be applied when the AP/IBSS is started */
1720         if (!mvmvif->ap_ibss_active)
1721                 return;
1722
1723         if (changes & (BSS_CHANGED_ERP_CTS_PROT | BSS_CHANGED_HT |
1724                        BSS_CHANGED_BANDWIDTH) &&
1725             iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL))
1726                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
1727
1728         /* Need to send a new beacon template to the FW */
1729         if (changes & BSS_CHANGED_BEACON &&
1730             iwl_mvm_mac_ctxt_beacon_changed(mvm, vif))
1731                 IWL_WARN(mvm, "Failed updating beacon data\n");
1732 }
1733
1734 static void iwl_mvm_bss_info_changed(struct ieee80211_hw *hw,
1735                                      struct ieee80211_vif *vif,
1736                                      struct ieee80211_bss_conf *bss_conf,
1737                                      u32 changes)
1738 {
1739         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1740
1741         /*
1742          * iwl_mvm_bss_info_changed_station() might call
1743          * iwl_mvm_protect_session(), which reads directly from
1744          * the device (the system time), so make sure it is available.
1745          */
1746         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_BSS_CHANGED))
1747                 return;
1748
1749         mutex_lock(&mvm->mutex);
1750
1751         if (changes & BSS_CHANGED_IDLE && !bss_conf->idle)
1752                 iwl_mvm_scan_offload_stop(mvm, true);
1753
1754         switch (vif->type) {
1755         case NL80211_IFTYPE_STATION:
1756                 iwl_mvm_bss_info_changed_station(mvm, vif, bss_conf, changes);
1757                 break;
1758         case NL80211_IFTYPE_AP:
1759         case NL80211_IFTYPE_ADHOC:
1760                 iwl_mvm_bss_info_changed_ap_ibss(mvm, vif, bss_conf, changes);
1761                 break;
1762         default:
1763                 /* shouldn't happen */
1764                 WARN_ON_ONCE(1);
1765         }
1766
1767         mutex_unlock(&mvm->mutex);
1768         iwl_mvm_unref(mvm, IWL_MVM_REF_BSS_CHANGED);
1769 }
1770
1771 static int iwl_mvm_cancel_scan_wait_notif(struct iwl_mvm *mvm,
1772                                           enum iwl_scan_status scan_type)
1773 {
1774         int ret;
1775         bool wait_for_handlers = false;
1776
1777         mutex_lock(&mvm->mutex);
1778
1779         if (mvm->scan_status != scan_type) {
1780                 ret = 0;
1781                 /* make sure there are no pending notifications */
1782                 wait_for_handlers = true;
1783                 goto out;
1784         }
1785
1786         switch (scan_type) {
1787         case IWL_MVM_SCAN_SCHED:
1788                 ret = iwl_mvm_scan_offload_stop(mvm, true);
1789                 break;
1790         case IWL_MVM_SCAN_OS:
1791                 ret = iwl_mvm_cancel_scan(mvm);
1792                 break;
1793         case IWL_MVM_SCAN_NONE:
1794         default:
1795                 WARN_ON_ONCE(1);
1796                 ret = -EINVAL;
1797                 break;
1798         }
1799         if (ret)
1800                 goto out;
1801
1802         wait_for_handlers = true;
1803 out:
1804         mutex_unlock(&mvm->mutex);
1805
1806         /* make sure we consume the completion notification */
1807         if (wait_for_handlers)
1808                 iwl_mvm_wait_for_async_handlers(mvm);
1809
1810         return ret;
1811 }
1812 static int iwl_mvm_mac_hw_scan(struct ieee80211_hw *hw,
1813                                struct ieee80211_vif *vif,
1814                                struct ieee80211_scan_request *hw_req)
1815 {
1816         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1817         struct cfg80211_scan_request *req = &hw_req->req;
1818         int ret;
1819
1820         if (req->n_channels == 0 ||
1821             req->n_channels > mvm->fw->ucode_capa.n_scan_channels)
1822                 return -EINVAL;
1823
1824         ret = iwl_mvm_cancel_scan_wait_notif(mvm, IWL_MVM_SCAN_SCHED);
1825         if (ret)
1826                 return ret;
1827
1828         mutex_lock(&mvm->mutex);
1829
1830         if (mvm->scan_status != IWL_MVM_SCAN_NONE) {
1831                 ret = -EBUSY;
1832                 goto out;
1833         }
1834
1835         iwl_mvm_ref(mvm, IWL_MVM_REF_SCAN);
1836
1837         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
1838                 ret = iwl_mvm_unified_scan_lmac(mvm, vif, hw_req);
1839         else
1840                 ret = iwl_mvm_scan_request(mvm, vif, req);
1841
1842         if (ret)
1843                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
1844 out:
1845         mutex_unlock(&mvm->mutex);
1846         return ret;
1847 }
1848
1849 static void iwl_mvm_mac_cancel_hw_scan(struct ieee80211_hw *hw,
1850                                        struct ieee80211_vif *vif)
1851 {
1852         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1853
1854         mutex_lock(&mvm->mutex);
1855
1856         iwl_mvm_cancel_scan(mvm);
1857
1858         mutex_unlock(&mvm->mutex);
1859 }
1860
1861 static void
1862 iwl_mvm_mac_allow_buffered_frames(struct ieee80211_hw *hw,
1863                                   struct ieee80211_sta *sta, u16 tids,
1864                                   int num_frames,
1865                                   enum ieee80211_frame_release_type reason,
1866                                   bool more_data)
1867 {
1868         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1869
1870         /* Called when we need to transmit (a) frame(s) from mac80211 */
1871
1872         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
1873                                           tids, more_data, false);
1874 }
1875
1876 static void
1877 iwl_mvm_mac_release_buffered_frames(struct ieee80211_hw *hw,
1878                                     struct ieee80211_sta *sta, u16 tids,
1879                                     int num_frames,
1880                                     enum ieee80211_frame_release_type reason,
1881                                     bool more_data)
1882 {
1883         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1884
1885         /* Called when we need to transmit (a) frame(s) from agg queue */
1886
1887         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
1888                                           tids, more_data, true);
1889 }
1890
1891 static void iwl_mvm_mac_sta_notify(struct ieee80211_hw *hw,
1892                                    struct ieee80211_vif *vif,
1893                                    enum sta_notify_cmd cmd,
1894                                    struct ieee80211_sta *sta)
1895 {
1896         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1897         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1898         int tid;
1899
1900         switch (cmd) {
1901         case STA_NOTIFY_SLEEP:
1902                 if (atomic_read(&mvm->pending_frames[mvmsta->sta_id]) > 0)
1903                         ieee80211_sta_block_awake(hw, sta, true);
1904                 spin_lock_bh(&mvmsta->lock);
1905                 for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
1906                         struct iwl_mvm_tid_data *tid_data;
1907
1908                         tid_data = &mvmsta->tid_data[tid];
1909                         if (tid_data->state != IWL_AGG_ON &&
1910                             tid_data->state != IWL_EMPTYING_HW_QUEUE_DELBA)
1911                                 continue;
1912                         if (iwl_mvm_tid_queued(tid_data) == 0)
1913                                 continue;
1914                         ieee80211_sta_set_buffered(sta, tid, true);
1915                 }
1916                 spin_unlock_bh(&mvmsta->lock);
1917                 /*
1918                  * The fw updates the STA to be asleep. Tx packets on the Tx
1919                  * queues to this station will not be transmitted. The fw will
1920                  * send a Tx response with TX_STATUS_FAIL_DEST_PS.
1921                  */
1922                 break;
1923         case STA_NOTIFY_AWAKE:
1924                 if (WARN_ON(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
1925                         break;
1926                 iwl_mvm_sta_modify_ps_wake(mvm, sta);
1927                 break;
1928         default:
1929                 break;
1930         }
1931 }
1932
1933 static void iwl_mvm_sta_pre_rcu_remove(struct ieee80211_hw *hw,
1934                                        struct ieee80211_vif *vif,
1935                                        struct ieee80211_sta *sta)
1936 {
1937         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1938         struct iwl_mvm_sta *mvm_sta = (void *)sta->drv_priv;
1939
1940         /*
1941          * This is called before mac80211 does RCU synchronisation,
1942          * so here we already invalidate our internal RCU-protected
1943          * station pointer. The rest of the code will thus no longer
1944          * be able to find the station this way, and we don't rely
1945          * on further RCU synchronisation after the sta_state()
1946          * callback deleted the station.
1947          */
1948         mutex_lock(&mvm->mutex);
1949         if (sta == rcu_access_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id]))
1950                 rcu_assign_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id],
1951                                    ERR_PTR(-ENOENT));
1952         mutex_unlock(&mvm->mutex);
1953 }
1954
1955 int iwl_mvm_tdls_sta_count(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1956 {
1957         struct ieee80211_sta *sta;
1958         struct iwl_mvm_sta *mvmsta;
1959         int count = 0;
1960         int i;
1961
1962         lockdep_assert_held(&mvm->mutex);
1963
1964         for (i = 0; i < IWL_MVM_STATION_COUNT; i++) {
1965                 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[i],
1966                                                 lockdep_is_held(&mvm->mutex));
1967                 if (!sta || IS_ERR(sta) || !sta->tdls)
1968                         continue;
1969
1970                 if (vif) {
1971                         mvmsta = iwl_mvm_sta_from_mac80211(sta);
1972                         if (mvmsta->vif != vif)
1973                                 continue;
1974                 }
1975
1976                 count++;
1977         }
1978
1979         return count;
1980 }
1981
1982 static void iwl_mvm_recalc_tdls_state(struct iwl_mvm *mvm,
1983                                       struct ieee80211_vif *vif,
1984                                       bool sta_added)
1985 {
1986         int tdls_sta_cnt = iwl_mvm_tdls_sta_count(mvm, vif);
1987
1988         /*
1989          * Disable ps when the first TDLS sta is added and re-enable it
1990          * when the last TDLS sta is removed
1991          */
1992         if ((tdls_sta_cnt == 1 && sta_added) ||
1993             (tdls_sta_cnt == 0 && !sta_added))
1994                 iwl_mvm_power_update_mac(mvm);
1995 }
1996
1997 static int iwl_mvm_mac_sta_state(struct ieee80211_hw *hw,
1998                                  struct ieee80211_vif *vif,
1999                                  struct ieee80211_sta *sta,
2000                                  enum ieee80211_sta_state old_state,
2001                                  enum ieee80211_sta_state new_state)
2002 {
2003         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2004         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2005         int ret;
2006
2007         IWL_DEBUG_MAC80211(mvm, "station %pM state change %d->%d\n",
2008                            sta->addr, old_state, new_state);
2009
2010         /* this would be a mac80211 bug ... but don't crash */
2011         if (WARN_ON_ONCE(!mvmvif->phy_ctxt))
2012                 return -EINVAL;
2013
2014         /* if a STA is being removed, reuse its ID */
2015         flush_work(&mvm->sta_drained_wk);
2016
2017         mutex_lock(&mvm->mutex);
2018         if (old_state == IEEE80211_STA_NOTEXIST &&
2019             new_state == IEEE80211_STA_NONE) {
2020                 /*
2021                  * Firmware bug - it'll crash if the beacon interval is less
2022                  * than 16. We can't avoid connecting at all, so refuse the
2023                  * station state change, this will cause mac80211 to abandon
2024                  * attempts to connect to this AP, and eventually wpa_s will
2025                  * blacklist the AP...
2026                  */
2027                 if (vif->type == NL80211_IFTYPE_STATION &&
2028                     vif->bss_conf.beacon_int < 16) {
2029                         IWL_ERR(mvm,
2030                                 "AP %pM beacon interval is %d, refusing due to firmware bug!\n",
2031                                 sta->addr, vif->bss_conf.beacon_int);
2032                         ret = -EINVAL;
2033                         goto out_unlock;
2034                 }
2035
2036                 if (sta->tdls &&
2037                     (vif->p2p ||
2038                      iwl_mvm_tdls_sta_count(mvm, NULL) ==
2039                                                 IWL_MVM_TDLS_STA_COUNT ||
2040                      iwl_mvm_phy_ctx_count(mvm) > 1)) {
2041                         IWL_DEBUG_MAC80211(mvm, "refusing TDLS sta\n");
2042                         ret = -EBUSY;
2043                         goto out_unlock;
2044                 }
2045
2046                 ret = iwl_mvm_add_sta(mvm, vif, sta);
2047                 if (sta->tdls && ret == 0)
2048                         iwl_mvm_recalc_tdls_state(mvm, vif, true);
2049         } else if (old_state == IEEE80211_STA_NONE &&
2050                    new_state == IEEE80211_STA_AUTH) {
2051                 /*
2052                  * EBS may be disabled due to previous failures reported by FW.
2053                  * Reset EBS status here assuming environment has been changed.
2054                  */
2055                 mvm->last_ebs_successful = true;
2056                 ret = 0;
2057         } else if (old_state == IEEE80211_STA_AUTH &&
2058                    new_state == IEEE80211_STA_ASSOC) {
2059                 ret = iwl_mvm_update_sta(mvm, vif, sta);
2060                 if (ret == 0)
2061                         iwl_mvm_rs_rate_init(mvm, sta,
2062                                              mvmvif->phy_ctxt->channel->band,
2063                                              true);
2064         } else if (old_state == IEEE80211_STA_ASSOC &&
2065                    new_state == IEEE80211_STA_AUTHORIZED) {
2066
2067                 /* we don't support TDLS during DCM */
2068                 if (iwl_mvm_phy_ctx_count(mvm) > 1)
2069                         iwl_mvm_teardown_tdls_peers(mvm);
2070
2071                 /* enable beacon filtering */
2072                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
2073                 ret = 0;
2074         } else if (old_state == IEEE80211_STA_AUTHORIZED &&
2075                    new_state == IEEE80211_STA_ASSOC) {
2076                 /* disable beacon filtering */
2077                 WARN_ON(iwl_mvm_disable_beacon_filter(mvm, vif, 0));
2078                 ret = 0;
2079         } else if (old_state == IEEE80211_STA_ASSOC &&
2080                    new_state == IEEE80211_STA_AUTH) {
2081                 ret = 0;
2082         } else if (old_state == IEEE80211_STA_AUTH &&
2083                    new_state == IEEE80211_STA_NONE) {
2084                 ret = 0;
2085         } else if (old_state == IEEE80211_STA_NONE &&
2086                    new_state == IEEE80211_STA_NOTEXIST) {
2087                 ret = iwl_mvm_rm_sta(mvm, vif, sta);
2088                 if (sta->tdls)
2089                         iwl_mvm_recalc_tdls_state(mvm, vif, false);
2090         } else {
2091                 ret = -EIO;
2092         }
2093  out_unlock:
2094         mutex_unlock(&mvm->mutex);
2095
2096         return ret;
2097 }
2098
2099 static int iwl_mvm_mac_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
2100 {
2101         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2102
2103         mvm->rts_threshold = value;
2104
2105         return 0;
2106 }
2107
2108 static void iwl_mvm_sta_rc_update(struct ieee80211_hw *hw,
2109                                   struct ieee80211_vif *vif,
2110                                   struct ieee80211_sta *sta, u32 changed)
2111 {
2112         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2113
2114         if (vif->type == NL80211_IFTYPE_STATION &&
2115             changed & IEEE80211_RC_NSS_CHANGED)
2116                 iwl_mvm_sf_update(mvm, vif, false);
2117 }
2118
2119 static int iwl_mvm_mac_conf_tx(struct ieee80211_hw *hw,
2120                                struct ieee80211_vif *vif, u16 ac,
2121                                const struct ieee80211_tx_queue_params *params)
2122 {
2123         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2124         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2125
2126         mvmvif->queue_params[ac] = *params;
2127
2128         /*
2129          * No need to update right away, we'll get BSS_CHANGED_QOS
2130          * The exception is P2P_DEVICE interface which needs immediate update.
2131          */
2132         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
2133                 int ret;
2134
2135                 mutex_lock(&mvm->mutex);
2136                 ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2137                 mutex_unlock(&mvm->mutex);
2138                 return ret;
2139         }
2140         return 0;
2141 }
2142
2143 static void iwl_mvm_mac_mgd_prepare_tx(struct ieee80211_hw *hw,
2144                                       struct ieee80211_vif *vif)
2145 {
2146         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2147         u32 duration = min(IWL_MVM_TE_SESSION_PROTECTION_MAX_TIME_MS,
2148                            200 + vif->bss_conf.beacon_int);
2149         u32 min_duration = min(IWL_MVM_TE_SESSION_PROTECTION_MIN_TIME_MS,
2150                                100 + vif->bss_conf.beacon_int);
2151
2152         if (WARN_ON_ONCE(vif->bss_conf.assoc))
2153                 return;
2154
2155         /*
2156          * iwl_mvm_protect_session() reads directly from the device
2157          * (the system time), so make sure it is available.
2158          */
2159         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PREPARE_TX))
2160                 return;
2161
2162         mutex_lock(&mvm->mutex);
2163         /* Try really hard to protect the session and hear a beacon */
2164         iwl_mvm_protect_session(mvm, vif, duration, min_duration, 500, false);
2165         mutex_unlock(&mvm->mutex);
2166
2167         iwl_mvm_unref(mvm, IWL_MVM_REF_PREPARE_TX);
2168 }
2169
2170 static void iwl_mvm_mac_mgd_protect_tdls_discover(struct ieee80211_hw *hw,
2171                                                   struct ieee80211_vif *vif)
2172 {
2173         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2174         u32 duration = 2 * vif->bss_conf.dtim_period * vif->bss_conf.beacon_int;
2175
2176         /*
2177          * iwl_mvm_protect_session() reads directly from the device
2178          * (the system time), so make sure it is available.
2179          */
2180         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PROTECT_TDLS))
2181                 return;
2182
2183         mutex_lock(&mvm->mutex);
2184         /* Protect the session to hear the TDLS setup response on the channel */
2185         iwl_mvm_protect_session(mvm, vif, duration, duration, 100, true);
2186         mutex_unlock(&mvm->mutex);
2187
2188         iwl_mvm_unref(mvm, IWL_MVM_REF_PROTECT_TDLS);
2189 }
2190
2191 static int iwl_mvm_mac_sched_scan_start(struct ieee80211_hw *hw,
2192                                         struct ieee80211_vif *vif,
2193                                         struct cfg80211_sched_scan_request *req,
2194                                         struct ieee80211_scan_ies *ies)
2195 {
2196         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2197         int ret;
2198
2199         ret = iwl_mvm_cancel_scan_wait_notif(mvm, IWL_MVM_SCAN_OS);
2200         if (ret)
2201                 return ret;
2202
2203         mutex_lock(&mvm->mutex);
2204
2205         if (!iwl_mvm_is_idle(mvm)) {
2206                 ret = -EBUSY;
2207                 goto out;
2208         }
2209
2210         if (mvm->scan_status != IWL_MVM_SCAN_NONE) {
2211                 ret = -EBUSY;
2212                 goto out;
2213         }
2214
2215         mvm->scan_status = IWL_MVM_SCAN_SCHED;
2216
2217         if (!(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)) {
2218                 ret = iwl_mvm_config_sched_scan(mvm, vif, req, ies);
2219                 if (ret)
2220                         goto err;
2221         }
2222
2223         ret = iwl_mvm_config_sched_scan_profiles(mvm, req);
2224         if (ret)
2225                 goto err;
2226
2227         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
2228                 ret = iwl_mvm_unified_sched_scan_lmac(mvm, vif, req, ies);
2229         else
2230                 ret = iwl_mvm_sched_scan_start(mvm, req);
2231
2232         if (!ret)
2233                 goto out;
2234 err:
2235         mvm->scan_status = IWL_MVM_SCAN_NONE;
2236 out:
2237         mutex_unlock(&mvm->mutex);
2238         return ret;
2239 }
2240
2241 static int iwl_mvm_mac_sched_scan_stop(struct ieee80211_hw *hw,
2242                                        struct ieee80211_vif *vif)
2243 {
2244         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2245         int ret;
2246
2247         mutex_lock(&mvm->mutex);
2248         ret = iwl_mvm_scan_offload_stop(mvm, false);
2249         mutex_unlock(&mvm->mutex);
2250         iwl_mvm_wait_for_async_handlers(mvm);
2251
2252         return ret;
2253 }
2254
2255 static int iwl_mvm_mac_set_key(struct ieee80211_hw *hw,
2256                                enum set_key_cmd cmd,
2257                                struct ieee80211_vif *vif,
2258                                struct ieee80211_sta *sta,
2259                                struct ieee80211_key_conf *key)
2260 {
2261         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2262         int ret;
2263
2264         if (iwlwifi_mod_params.sw_crypto) {
2265                 IWL_DEBUG_MAC80211(mvm, "leave - hwcrypto disabled\n");
2266                 return -EOPNOTSUPP;
2267         }
2268
2269         switch (key->cipher) {
2270         case WLAN_CIPHER_SUITE_TKIP:
2271                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
2272                 /* fall-through */
2273         case WLAN_CIPHER_SUITE_CCMP:
2274                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
2275                 break;
2276         case WLAN_CIPHER_SUITE_AES_CMAC:
2277                 WARN_ON_ONCE(!(hw->flags & IEEE80211_HW_MFP_CAPABLE));
2278                 break;
2279         case WLAN_CIPHER_SUITE_WEP40:
2280         case WLAN_CIPHER_SUITE_WEP104:
2281                 /*
2282                  * Support for TX only, at least for now, so accept
2283                  * the key and do nothing else. Then mac80211 will
2284                  * pass it for TX but we don't have to use it for RX.
2285                  */
2286                 return 0;
2287         default:
2288                 /* currently FW supports only one optional cipher scheme */
2289                 if (hw->n_cipher_schemes &&
2290                     hw->cipher_schemes->cipher == key->cipher)
2291                         key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2292                 else
2293                         return -EOPNOTSUPP;
2294         }
2295
2296         mutex_lock(&mvm->mutex);
2297
2298         switch (cmd) {
2299         case SET_KEY:
2300                 if ((vif->type == NL80211_IFTYPE_ADHOC ||
2301                      vif->type == NL80211_IFTYPE_AP) && !sta) {
2302                         /*
2303                          * GTK on AP interface is a TX-only key, return 0;
2304                          * on IBSS they're per-station and because we're lazy
2305                          * we don't support them for RX, so do the same.
2306                          */
2307                         ret = 0;
2308                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2309                         break;
2310                 }
2311
2312                 IWL_DEBUG_MAC80211(mvm, "set hwcrypto key\n");
2313                 ret = iwl_mvm_set_sta_key(mvm, vif, sta, key, false);
2314                 if (ret) {
2315                         IWL_WARN(mvm, "set key failed\n");
2316                         /*
2317                          * can't add key for RX, but we don't need it
2318                          * in the device for TX so still return 0
2319                          */
2320                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2321                         ret = 0;
2322                 }
2323
2324                 break;
2325         case DISABLE_KEY:
2326                 if (key->hw_key_idx == STA_KEY_IDX_INVALID) {
2327                         ret = 0;
2328                         break;
2329                 }
2330
2331                 IWL_DEBUG_MAC80211(mvm, "disable hwcrypto key\n");
2332                 ret = iwl_mvm_remove_sta_key(mvm, vif, sta, key);
2333                 break;
2334         default:
2335                 ret = -EINVAL;
2336         }
2337
2338         mutex_unlock(&mvm->mutex);
2339         return ret;
2340 }
2341
2342 static void iwl_mvm_mac_update_tkip_key(struct ieee80211_hw *hw,
2343                                         struct ieee80211_vif *vif,
2344                                         struct ieee80211_key_conf *keyconf,
2345                                         struct ieee80211_sta *sta,
2346                                         u32 iv32, u16 *phase1key)
2347 {
2348         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2349
2350         if (keyconf->hw_key_idx == STA_KEY_IDX_INVALID)
2351                 return;
2352
2353         iwl_mvm_update_tkip_key(mvm, vif, keyconf, sta, iv32, phase1key);
2354 }
2355
2356
2357 static bool iwl_mvm_rx_aux_roc(struct iwl_notif_wait_data *notif_wait,
2358                                struct iwl_rx_packet *pkt, void *data)
2359 {
2360         struct iwl_mvm *mvm =
2361                 container_of(notif_wait, struct iwl_mvm, notif_wait);
2362         struct iwl_hs20_roc_res *resp;
2363         int resp_len = iwl_rx_packet_payload_len(pkt);
2364         struct iwl_mvm_time_event_data *te_data = data;
2365
2366         if (WARN_ON(pkt->hdr.cmd != HOT_SPOT_CMD))
2367                 return true;
2368
2369         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
2370                 IWL_ERR(mvm, "Invalid HOT_SPOT_CMD response\n");
2371                 return true;
2372         }
2373
2374         resp = (void *)pkt->data;
2375
2376         IWL_DEBUG_TE(mvm,
2377                      "Aux ROC: Recieved response from ucode: status=%d uid=%d\n",
2378                      resp->status, resp->event_unique_id);
2379
2380         te_data->uid = le32_to_cpu(resp->event_unique_id);
2381         IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
2382                      te_data->uid);
2383
2384         spin_lock_bh(&mvm->time_event_lock);
2385         list_add_tail(&te_data->list, &mvm->aux_roc_te_list);
2386         spin_unlock_bh(&mvm->time_event_lock);
2387
2388         return true;
2389 }
2390
2391 #define AUX_ROC_MAX_DELAY_ON_CHANNEL 5000
2392 static int iwl_mvm_send_aux_roc_cmd(struct iwl_mvm *mvm,
2393                                     struct ieee80211_channel *channel,
2394                                     struct ieee80211_vif *vif,
2395                                     int duration)
2396 {
2397         int res, time_reg = DEVICE_SYSTEM_TIME_REG;
2398         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2399         struct iwl_mvm_time_event_data *te_data = &mvmvif->hs_time_event_data;
2400         static const u8 time_event_response[] = { HOT_SPOT_CMD };
2401         struct iwl_notification_wait wait_time_event;
2402         struct iwl_hs20_roc_req aux_roc_req = {
2403                 .action = cpu_to_le32(FW_CTXT_ACTION_ADD),
2404                 .id_and_color =
2405                         cpu_to_le32(FW_CMD_ID_AND_COLOR(MAC_INDEX_AUX, 0)),
2406                 .sta_id_and_color = cpu_to_le32(mvm->aux_sta.sta_id),
2407                 /* Set the channel info data */
2408                 .channel_info.band = (channel->band == IEEE80211_BAND_2GHZ) ?
2409                         PHY_BAND_24 : PHY_BAND_5,
2410                 .channel_info.channel = channel->hw_value,
2411                 .channel_info.width = PHY_VHT_CHANNEL_MODE20,
2412                 /* Set the time and duration */
2413                 .apply_time = cpu_to_le32(iwl_read_prph(mvm->trans, time_reg)),
2414                 .apply_time_max_delay =
2415                         cpu_to_le32(MSEC_TO_TU(AUX_ROC_MAX_DELAY_ON_CHANNEL)),
2416                 .duration = cpu_to_le32(MSEC_TO_TU(duration)),
2417          };
2418
2419         /* Set the node address */
2420         memcpy(aux_roc_req.node_addr, vif->addr, ETH_ALEN);
2421
2422         te_data->vif = vif;
2423         te_data->duration = duration;
2424         te_data->id = HOT_SPOT_CMD;
2425
2426         lockdep_assert_held(&mvm->mutex);
2427
2428         spin_lock_bh(&mvm->time_event_lock);
2429         list_add_tail(&te_data->list, &mvm->time_event_list);
2430         spin_unlock_bh(&mvm->time_event_lock);
2431
2432         /*
2433          * Use a notification wait, which really just processes the
2434          * command response and doesn't wait for anything, in order
2435          * to be able to process the response and get the UID inside
2436          * the RX path. Using CMD_WANT_SKB doesn't work because it
2437          * stores the buffer and then wakes up this thread, by which
2438          * time another notification (that the time event started)
2439          * might already be processed unsuccessfully.
2440          */
2441         iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
2442                                    time_event_response,
2443                                    ARRAY_SIZE(time_event_response),
2444                                    iwl_mvm_rx_aux_roc, te_data);
2445
2446         res = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0, sizeof(aux_roc_req),
2447                                    &aux_roc_req);
2448
2449         if (res) {
2450                 IWL_ERR(mvm, "Couldn't send HOT_SPOT_CMD: %d\n", res);
2451                 iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
2452                 goto out_clear_te;
2453         }
2454
2455         /* No need to wait for anything, so just pass 1 (0 isn't valid) */
2456         res = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
2457         /* should never fail */
2458         WARN_ON_ONCE(res);
2459
2460         if (res) {
2461  out_clear_te:
2462                 spin_lock_bh(&mvm->time_event_lock);
2463                 iwl_mvm_te_clear_data(mvm, te_data);
2464                 spin_unlock_bh(&mvm->time_event_lock);
2465         }
2466
2467         return res;
2468 }
2469
2470 static int iwl_mvm_roc(struct ieee80211_hw *hw,
2471                        struct ieee80211_vif *vif,
2472                        struct ieee80211_channel *channel,
2473                        int duration,
2474                        enum ieee80211_roc_type type)
2475 {
2476         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2477         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2478         struct cfg80211_chan_def chandef;
2479         struct iwl_mvm_phy_ctxt *phy_ctxt;
2480         int ret, i;
2481
2482         IWL_DEBUG_MAC80211(mvm, "enter (%d, %d, %d)\n", channel->hw_value,
2483                            duration, type);
2484
2485         switch (vif->type) {
2486         case NL80211_IFTYPE_STATION:
2487                 /* Use aux roc framework (HS20) */
2488                 ret = iwl_mvm_send_aux_roc_cmd(mvm, channel,
2489                                                vif, duration);
2490                 return ret;
2491         case NL80211_IFTYPE_P2P_DEVICE:
2492                 /* handle below */
2493                 break;
2494         default:
2495                 IWL_ERR(mvm, "vif isn't P2P_DEVICE: %d\n", vif->type);
2496                 return -EINVAL;
2497         }
2498
2499         mutex_lock(&mvm->mutex);
2500
2501         for (i = 0; i < NUM_PHY_CTX; i++) {
2502                 phy_ctxt = &mvm->phy_ctxts[i];
2503                 if (phy_ctxt->ref == 0 || mvmvif->phy_ctxt == phy_ctxt)
2504                         continue;
2505
2506                 if (phy_ctxt->ref && channel == phy_ctxt->channel) {
2507                         /*
2508                          * Unbind the P2P_DEVICE from the current PHY context,
2509                          * and if the PHY context is not used remove it.
2510                          */
2511                         ret = iwl_mvm_binding_remove_vif(mvm, vif);
2512                         if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
2513                                 goto out_unlock;
2514
2515                         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
2516
2517                         /* Bind the P2P_DEVICE to the current PHY Context */
2518                         mvmvif->phy_ctxt = phy_ctxt;
2519
2520                         ret = iwl_mvm_binding_add_vif(mvm, vif);
2521                         if (WARN(ret, "Failed binding P2P_DEVICE\n"))
2522                                 goto out_unlock;
2523
2524                         iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
2525                         goto schedule_time_event;
2526                 }
2527         }
2528
2529         /* Need to update the PHY context only if the ROC channel changed */
2530         if (channel == mvmvif->phy_ctxt->channel)
2531                 goto schedule_time_event;
2532
2533         cfg80211_chandef_create(&chandef, channel, NL80211_CHAN_NO_HT);
2534
2535         /*
2536          * Change the PHY context configuration as it is currently referenced
2537          * only by the P2P Device MAC
2538          */
2539         if (mvmvif->phy_ctxt->ref == 1) {
2540                 ret = iwl_mvm_phy_ctxt_changed(mvm, mvmvif->phy_ctxt,
2541                                                &chandef, 1, 1);
2542                 if (ret)
2543                         goto out_unlock;
2544         } else {
2545                 /*
2546                  * The PHY context is shared with other MACs. Need to remove the
2547                  * P2P Device from the binding, allocate an new PHY context and
2548                  * create a new binding
2549                  */
2550                 phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
2551                 if (!phy_ctxt) {
2552                         ret = -ENOSPC;
2553                         goto out_unlock;
2554                 }
2555
2556                 ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &chandef,
2557                                                1, 1);
2558                 if (ret) {
2559                         IWL_ERR(mvm, "Failed to change PHY context\n");
2560                         goto out_unlock;
2561                 }
2562
2563                 /* Unbind the P2P_DEVICE from the current PHY context */
2564                 ret = iwl_mvm_binding_remove_vif(mvm, vif);
2565                 if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
2566                         goto out_unlock;
2567
2568                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
2569
2570                 /* Bind the P2P_DEVICE to the new allocated PHY context */
2571                 mvmvif->phy_ctxt = phy_ctxt;
2572
2573                 ret = iwl_mvm_binding_add_vif(mvm, vif);
2574                 if (WARN(ret, "Failed binding P2P_DEVICE\n"))
2575                         goto out_unlock;
2576
2577                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
2578         }
2579
2580 schedule_time_event:
2581         /* Schedule the time events */
2582         ret = iwl_mvm_start_p2p_roc(mvm, vif, duration, type);
2583
2584 out_unlock:
2585         mutex_unlock(&mvm->mutex);
2586         IWL_DEBUG_MAC80211(mvm, "leave\n");
2587         return ret;
2588 }
2589
2590 static int iwl_mvm_cancel_roc(struct ieee80211_hw *hw)
2591 {
2592         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2593
2594         IWL_DEBUG_MAC80211(mvm, "enter\n");
2595
2596         mutex_lock(&mvm->mutex);
2597         iwl_mvm_stop_p2p_roc(mvm);
2598         mutex_unlock(&mvm->mutex);
2599
2600         IWL_DEBUG_MAC80211(mvm, "leave\n");
2601         return 0;
2602 }
2603
2604 static int __iwl_mvm_add_chanctx(struct iwl_mvm *mvm,
2605                                  struct ieee80211_chanctx_conf *ctx)
2606 {
2607         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2608         struct iwl_mvm_phy_ctxt *phy_ctxt;
2609         int ret;
2610
2611         lockdep_assert_held(&mvm->mutex);
2612
2613         IWL_DEBUG_MAC80211(mvm, "Add channel context\n");
2614
2615         phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
2616         if (!phy_ctxt) {
2617                 ret = -ENOSPC;
2618                 goto out;
2619         }
2620
2621         ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
2622                                        ctx->rx_chains_static,
2623                                        ctx->rx_chains_dynamic);
2624         if (ret) {
2625                 IWL_ERR(mvm, "Failed to add PHY context\n");
2626                 goto out;
2627         }
2628
2629         iwl_mvm_phy_ctxt_ref(mvm, phy_ctxt);
2630         *phy_ctxt_id = phy_ctxt->id;
2631 out:
2632         return ret;
2633 }
2634
2635 static int iwl_mvm_add_chanctx(struct ieee80211_hw *hw,
2636                                struct ieee80211_chanctx_conf *ctx)
2637 {
2638         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2639         int ret;
2640
2641         mutex_lock(&mvm->mutex);
2642         ret = __iwl_mvm_add_chanctx(mvm, ctx);
2643         mutex_unlock(&mvm->mutex);
2644
2645         return ret;
2646 }
2647
2648 static void __iwl_mvm_remove_chanctx(struct iwl_mvm *mvm,
2649                                      struct ieee80211_chanctx_conf *ctx)
2650 {
2651         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2652         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2653
2654         lockdep_assert_held(&mvm->mutex);
2655
2656         iwl_mvm_phy_ctxt_unref(mvm, phy_ctxt);
2657 }
2658
2659 static void iwl_mvm_remove_chanctx(struct ieee80211_hw *hw,
2660                                    struct ieee80211_chanctx_conf *ctx)
2661 {
2662         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2663
2664         mutex_lock(&mvm->mutex);
2665         __iwl_mvm_remove_chanctx(mvm, ctx);
2666         mutex_unlock(&mvm->mutex);
2667 }
2668
2669 static void iwl_mvm_change_chanctx(struct ieee80211_hw *hw,
2670                                    struct ieee80211_chanctx_conf *ctx,
2671                                    u32 changed)
2672 {
2673         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2674         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2675         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2676
2677         if (WARN_ONCE((phy_ctxt->ref > 1) &&
2678                       (changed & ~(IEEE80211_CHANCTX_CHANGE_WIDTH |
2679                                    IEEE80211_CHANCTX_CHANGE_RX_CHAINS |
2680                                    IEEE80211_CHANCTX_CHANGE_RADAR |
2681                                    IEEE80211_CHANCTX_CHANGE_MIN_WIDTH)),
2682                       "Cannot change PHY. Ref=%d, changed=0x%X\n",
2683                       phy_ctxt->ref, changed))
2684                 return;
2685
2686         mutex_lock(&mvm->mutex);
2687         iwl_mvm_bt_coex_vif_change(mvm);
2688         iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
2689                                  ctx->rx_chains_static,
2690                                  ctx->rx_chains_dynamic);
2691         mutex_unlock(&mvm->mutex);
2692 }
2693
2694 static int __iwl_mvm_assign_vif_chanctx(struct iwl_mvm *mvm,
2695                                         struct ieee80211_vif *vif,
2696                                         struct ieee80211_chanctx_conf *ctx,
2697                                         bool switching_chanctx)
2698 {
2699         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2700         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2701         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2702         int ret;
2703
2704         lockdep_assert_held(&mvm->mutex);
2705
2706         mvmvif->phy_ctxt = phy_ctxt;
2707
2708         switch (vif->type) {
2709         case NL80211_IFTYPE_AP:
2710                 /* Unless it's a CSA flow we have nothing to do here */
2711                 if (vif->csa_active) {
2712                         mvmvif->ap_ibss_active = true;
2713                         break;
2714                 }
2715         case NL80211_IFTYPE_ADHOC:
2716                 /*
2717                  * The AP binding flow is handled as part of the start_ap flow
2718                  * (in bss_info_changed), similarly for IBSS.
2719                  */
2720                 ret = 0;
2721                 goto out;
2722         case NL80211_IFTYPE_STATION:
2723                 break;
2724         case NL80211_IFTYPE_MONITOR:
2725                 /* always disable PS when a monitor interface is active */
2726                 mvmvif->ps_disabled = true;
2727                 break;
2728         default:
2729                 ret = -EINVAL;
2730                 goto out;
2731         }
2732
2733         ret = iwl_mvm_binding_add_vif(mvm, vif);
2734         if (ret)
2735                 goto out;
2736
2737         /*
2738          * Power state must be updated before quotas,
2739          * otherwise fw will complain.
2740          */
2741         iwl_mvm_power_update_mac(mvm);
2742
2743         /* Setting the quota at this stage is only required for monitor
2744          * interfaces. For the other types, the bss_info changed flow
2745          * will handle quota settings.
2746          */
2747         if (vif->type == NL80211_IFTYPE_MONITOR) {
2748                 mvmvif->monitor_active = true;
2749                 ret = iwl_mvm_update_quotas(mvm, NULL);
2750                 if (ret)
2751                         goto out_remove_binding;
2752         }
2753
2754         /* Handle binding during CSA */
2755         if ((vif->type == NL80211_IFTYPE_AP) ||
2756             (switching_chanctx && (vif->type == NL80211_IFTYPE_STATION))) {
2757                 iwl_mvm_update_quotas(mvm, NULL);
2758                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2759         }
2760
2761         if (vif->csa_active && vif->type == NL80211_IFTYPE_STATION) {
2762                 struct iwl_mvm_sta *mvmsta;
2763
2764                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
2765                                                           mvmvif->ap_sta_id);
2766
2767                 if (WARN_ON(!mvmsta))
2768                         goto out;
2769
2770                 /* TODO: only re-enable after the first beacon */
2771                 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, false);
2772         }
2773
2774         goto out;
2775
2776 out_remove_binding:
2777         iwl_mvm_binding_remove_vif(mvm, vif);
2778         iwl_mvm_power_update_mac(mvm);
2779 out:
2780         if (ret)
2781                 mvmvif->phy_ctxt = NULL;
2782         return ret;
2783 }
2784 static int iwl_mvm_assign_vif_chanctx(struct ieee80211_hw *hw,
2785                                       struct ieee80211_vif *vif,
2786                                       struct ieee80211_chanctx_conf *ctx)
2787 {
2788         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2789         int ret;
2790
2791         mutex_lock(&mvm->mutex);
2792         ret = __iwl_mvm_assign_vif_chanctx(mvm, vif, ctx, false);
2793         mutex_unlock(&mvm->mutex);
2794
2795         return ret;
2796 }
2797
2798 static void __iwl_mvm_unassign_vif_chanctx(struct iwl_mvm *mvm,
2799                                            struct ieee80211_vif *vif,
2800                                            struct ieee80211_chanctx_conf *ctx,
2801                                            bool switching_chanctx)
2802 {
2803         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2804         struct ieee80211_vif *disabled_vif = NULL;
2805         struct iwl_mvm_sta *mvmsta;
2806
2807         lockdep_assert_held(&mvm->mutex);
2808
2809         iwl_mvm_remove_time_event(mvm, mvmvif, &mvmvif->time_event_data);
2810
2811         switch (vif->type) {
2812         case NL80211_IFTYPE_ADHOC:
2813                 goto out;
2814         case NL80211_IFTYPE_MONITOR:
2815                 mvmvif->monitor_active = false;
2816                 mvmvif->ps_disabled = false;
2817                 break;
2818         case NL80211_IFTYPE_AP:
2819                 /* This part is triggered only during CSA */
2820                 if (!vif->csa_active || !mvmvif->ap_ibss_active)
2821                         goto out;
2822
2823                 /* Set CS bit on all the stations */
2824                 iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, true);
2825
2826                 /* Save blocked iface, the timeout is set on the next beacon */
2827                 rcu_assign_pointer(mvm->csa_tx_blocked_vif, vif);
2828
2829                 mvmvif->ap_ibss_active = false;
2830                 break;
2831         case NL80211_IFTYPE_STATION:
2832                 if (!switching_chanctx)
2833                         break;
2834
2835                 disabled_vif = vif;
2836
2837                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
2838                                                           mvmvif->ap_sta_id);
2839
2840                 if (!WARN_ON(!mvmsta))
2841                         iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, true);
2842
2843                 iwl_mvm_mac_ctxt_changed(mvm, vif, true, NULL);
2844                 break;
2845         default:
2846                 break;
2847         }
2848
2849         iwl_mvm_update_quotas(mvm, disabled_vif);
2850         iwl_mvm_binding_remove_vif(mvm, vif);
2851
2852 out:
2853         mvmvif->phy_ctxt = NULL;
2854         iwl_mvm_power_update_mac(mvm);
2855 }
2856
2857 static void iwl_mvm_unassign_vif_chanctx(struct ieee80211_hw *hw,
2858                                          struct ieee80211_vif *vif,
2859                                          struct ieee80211_chanctx_conf *ctx)
2860 {
2861         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2862
2863         mutex_lock(&mvm->mutex);
2864         __iwl_mvm_unassign_vif_chanctx(mvm, vif, ctx, false);
2865         mutex_unlock(&mvm->mutex);
2866 }
2867
2868 static int iwl_mvm_switch_vif_chanctx(struct ieee80211_hw *hw,
2869                                       struct ieee80211_vif_chanctx_switch *vifs,
2870                                       int n_vifs,
2871                                       enum ieee80211_chanctx_switch_mode mode)
2872 {
2873         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2874         int ret;
2875
2876         /* we only support SWAP_CONTEXTS and with a single-vif right now */
2877         if (mode != CHANCTX_SWMODE_SWAP_CONTEXTS || n_vifs > 1)
2878                 return -EOPNOTSUPP;
2879
2880         mutex_lock(&mvm->mutex);
2881         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
2882         __iwl_mvm_remove_chanctx(mvm, vifs[0].old_ctx);
2883
2884         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].new_ctx);
2885         if (ret) {
2886                 IWL_ERR(mvm, "failed to add new_ctx during channel switch\n");
2887                 goto out_reassign;
2888         }
2889
2890         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
2891                                            true);
2892         if (ret) {
2893                 IWL_ERR(mvm,
2894                         "failed to assign new_ctx during channel switch\n");
2895                 goto out_remove;
2896         }
2897
2898         /* we don't support TDLS during DCM - can be caused by channel switch */
2899         if (iwl_mvm_phy_ctx_count(mvm) > 1)
2900                 iwl_mvm_teardown_tdls_peers(mvm);
2901
2902         goto out;
2903
2904 out_remove:
2905         __iwl_mvm_remove_chanctx(mvm, vifs[0].new_ctx);
2906
2907 out_reassign:
2908         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].old_ctx);
2909         if (ret) {
2910                 IWL_ERR(mvm, "failed to add old_ctx back after failure.\n");
2911                 goto out_restart;
2912         }
2913
2914         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
2915                                            true);
2916         if (ret) {
2917                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
2918                 goto out_restart;
2919         }
2920
2921         goto out;
2922
2923 out_restart:
2924         /* things keep failing, better restart the hw */
2925         iwl_mvm_nic_restart(mvm, false);
2926
2927 out:
2928         mutex_unlock(&mvm->mutex);
2929         return ret;
2930 }
2931
2932 static int iwl_mvm_set_tim(struct ieee80211_hw *hw,
2933                            struct ieee80211_sta *sta,
2934                            bool set)
2935 {
2936         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2937         struct iwl_mvm_sta *mvm_sta = (void *)sta->drv_priv;
2938
2939         if (!mvm_sta || !mvm_sta->vif) {
2940                 IWL_ERR(mvm, "Station is not associated to a vif\n");
2941                 return -EINVAL;
2942         }
2943
2944         return iwl_mvm_mac_ctxt_beacon_changed(mvm, mvm_sta->vif);
2945 }
2946
2947 #ifdef CONFIG_NL80211_TESTMODE
2948 static const struct nla_policy iwl_mvm_tm_policy[IWL_MVM_TM_ATTR_MAX + 1] = {
2949         [IWL_MVM_TM_ATTR_CMD] = { .type = NLA_U32 },
2950         [IWL_MVM_TM_ATTR_NOA_DURATION] = { .type = NLA_U32 },
2951         [IWL_MVM_TM_ATTR_BEACON_FILTER_STATE] = { .type = NLA_U32 },
2952 };
2953
2954 static int __iwl_mvm_mac_testmode_cmd(struct iwl_mvm *mvm,
2955                                       struct ieee80211_vif *vif,
2956                                       void *data, int len)
2957 {
2958         struct nlattr *tb[IWL_MVM_TM_ATTR_MAX + 1];
2959         int err;
2960         u32 noa_duration;
2961
2962         err = nla_parse(tb, IWL_MVM_TM_ATTR_MAX, data, len, iwl_mvm_tm_policy);
2963         if (err)
2964                 return err;
2965
2966         if (!tb[IWL_MVM_TM_ATTR_CMD])
2967                 return -EINVAL;
2968
2969         switch (nla_get_u32(tb[IWL_MVM_TM_ATTR_CMD])) {
2970         case IWL_MVM_TM_CMD_SET_NOA:
2971                 if (!vif || vif->type != NL80211_IFTYPE_AP || !vif->p2p ||
2972                     !vif->bss_conf.enable_beacon ||
2973                     !tb[IWL_MVM_TM_ATTR_NOA_DURATION])
2974                         return -EINVAL;
2975
2976                 noa_duration = nla_get_u32(tb[IWL_MVM_TM_ATTR_NOA_DURATION]);
2977                 if (noa_duration >= vif->bss_conf.beacon_int)
2978                         return -EINVAL;
2979
2980                 mvm->noa_duration = noa_duration;
2981                 mvm->noa_vif = vif;
2982
2983                 return iwl_mvm_update_quotas(mvm, NULL);
2984         case IWL_MVM_TM_CMD_SET_BEACON_FILTER:
2985                 /* must be associated client vif - ignore authorized */
2986                 if (!vif || vif->type != NL80211_IFTYPE_STATION ||
2987                     !vif->bss_conf.assoc || !vif->bss_conf.dtim_period ||
2988                     !tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE])
2989                         return -EINVAL;
2990
2991                 if (nla_get_u32(tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE]))
2992                         return iwl_mvm_enable_beacon_filter(mvm, vif, 0);
2993                 return iwl_mvm_disable_beacon_filter(mvm, vif, 0);
2994         }
2995
2996         return -EOPNOTSUPP;
2997 }
2998
2999 static int iwl_mvm_mac_testmode_cmd(struct ieee80211_hw *hw,
3000                                     struct ieee80211_vif *vif,
3001                                     void *data, int len)
3002 {
3003         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3004         int err;
3005
3006         mutex_lock(&mvm->mutex);
3007         err = __iwl_mvm_mac_testmode_cmd(mvm, vif, data, len);
3008         mutex_unlock(&mvm->mutex);
3009
3010         return err;
3011 }
3012 #endif
3013
3014 static void iwl_mvm_channel_switch_beacon(struct ieee80211_hw *hw,
3015                                           struct ieee80211_vif *vif,
3016                                           struct cfg80211_chan_def *chandef)
3017 {
3018         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3019         struct ieee80211_vif *csa_vif;
3020
3021         mutex_lock(&mvm->mutex);
3022
3023         csa_vif = rcu_dereference_protected(mvm->csa_vif,
3024                                             lockdep_is_held(&mvm->mutex));
3025         if (WARN(csa_vif && csa_vif->csa_active,
3026                  "Another CSA is already in progress"))
3027                 goto out_unlock;
3028
3029         IWL_DEBUG_MAC80211(mvm, "CSA started to freq %d\n",
3030                            chandef->center_freq1);
3031         rcu_assign_pointer(mvm->csa_vif, vif);
3032
3033 out_unlock:
3034         mutex_unlock(&mvm->mutex);
3035 }
3036
3037 static void iwl_mvm_mac_flush(struct ieee80211_hw *hw,
3038                               struct ieee80211_vif *vif, u32 queues, bool drop)
3039 {
3040         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3041         struct iwl_mvm_vif *mvmvif;
3042         struct iwl_mvm_sta *mvmsta;
3043
3044         if (!vif || vif->type != NL80211_IFTYPE_STATION)
3045                 return;
3046
3047         mutex_lock(&mvm->mutex);
3048         mvmvif = iwl_mvm_vif_from_mac80211(vif);
3049         mvmsta = iwl_mvm_sta_from_staid_protected(mvm, mvmvif->ap_sta_id);
3050
3051         if (WARN_ON_ONCE(!mvmsta))
3052                 goto done;
3053
3054         if (drop) {
3055                 if (iwl_mvm_flush_tx_path(mvm, mvmsta->tfd_queue_msk, true))
3056                         IWL_ERR(mvm, "flush request fail\n");
3057         } else {
3058                 iwl_trans_wait_tx_queue_empty(mvm->trans,
3059                                               mvmsta->tfd_queue_msk);
3060         }
3061 done:
3062         mutex_unlock(&mvm->mutex);
3063 }
3064
3065 const struct ieee80211_ops iwl_mvm_hw_ops = {
3066         .tx = iwl_mvm_mac_tx,
3067         .ampdu_action = iwl_mvm_mac_ampdu_action,
3068         .start = iwl_mvm_mac_start,
3069         .restart_complete = iwl_mvm_mac_restart_complete,
3070         .stop = iwl_mvm_mac_stop,
3071         .add_interface = iwl_mvm_mac_add_interface,
3072         .remove_interface = iwl_mvm_mac_remove_interface,
3073         .config = iwl_mvm_mac_config,
3074         .prepare_multicast = iwl_mvm_prepare_multicast,
3075         .configure_filter = iwl_mvm_configure_filter,
3076         .bss_info_changed = iwl_mvm_bss_info_changed,
3077         .hw_scan = iwl_mvm_mac_hw_scan,
3078         .cancel_hw_scan = iwl_mvm_mac_cancel_hw_scan,
3079         .sta_pre_rcu_remove = iwl_mvm_sta_pre_rcu_remove,
3080         .sta_state = iwl_mvm_mac_sta_state,
3081         .sta_notify = iwl_mvm_mac_sta_notify,
3082         .allow_buffered_frames = iwl_mvm_mac_allow_buffered_frames,
3083         .release_buffered_frames = iwl_mvm_mac_release_buffered_frames,
3084         .set_rts_threshold = iwl_mvm_mac_set_rts_threshold,
3085         .sta_rc_update = iwl_mvm_sta_rc_update,
3086         .conf_tx = iwl_mvm_mac_conf_tx,
3087         .mgd_prepare_tx = iwl_mvm_mac_mgd_prepare_tx,
3088         .mgd_protect_tdls_discover = iwl_mvm_mac_mgd_protect_tdls_discover,
3089         .flush = iwl_mvm_mac_flush,
3090         .sched_scan_start = iwl_mvm_mac_sched_scan_start,
3091         .sched_scan_stop = iwl_mvm_mac_sched_scan_stop,
3092         .set_key = iwl_mvm_mac_set_key,
3093         .update_tkip_key = iwl_mvm_mac_update_tkip_key,
3094         .remain_on_channel = iwl_mvm_roc,
3095         .cancel_remain_on_channel = iwl_mvm_cancel_roc,
3096         .add_chanctx = iwl_mvm_add_chanctx,
3097         .remove_chanctx = iwl_mvm_remove_chanctx,
3098         .change_chanctx = iwl_mvm_change_chanctx,
3099         .assign_vif_chanctx = iwl_mvm_assign_vif_chanctx,
3100         .unassign_vif_chanctx = iwl_mvm_unassign_vif_chanctx,
3101         .switch_vif_chanctx = iwl_mvm_switch_vif_chanctx,
3102
3103         .start_ap = iwl_mvm_start_ap_ibss,
3104         .stop_ap = iwl_mvm_stop_ap_ibss,
3105         .join_ibss = iwl_mvm_start_ap_ibss,
3106         .leave_ibss = iwl_mvm_stop_ap_ibss,
3107
3108         .set_tim = iwl_mvm_set_tim,
3109
3110         .channel_switch_beacon = iwl_mvm_channel_switch_beacon,
3111
3112         CFG80211_TESTMODE_CMD(iwl_mvm_mac_testmode_cmd)
3113
3114 #ifdef CONFIG_PM_SLEEP
3115         /* look at d3.c */
3116         .suspend = iwl_mvm_suspend,
3117         .resume = iwl_mvm_resume,
3118         .set_wakeup = iwl_mvm_set_wakeup,
3119         .set_rekey_data = iwl_mvm_set_rekey_data,
3120 #if IS_ENABLED(CONFIG_IPV6)
3121         .ipv6_addr_change = iwl_mvm_ipv6_addr_change,
3122 #endif
3123         .set_default_unicast_key = iwl_mvm_set_default_unicast_key,
3124 #endif
3125 };