iwlwifi: mvm: Support CSA countdown offloading
[cascardo/linux.git] / drivers / net / wireless / intel / 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  * Copyright(c) 2016 Intel Deutschland GmbH
11  *
12  * This program is free software; you can redistribute it and/or modify
13  * it under the terms of version 2 of the GNU General Public License as
14  * published by the Free Software Foundation.
15  *
16  * This program is distributed in the hope that it will be useful, but
17  * WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
24  * USA
25  *
26  * The full GNU General Public License is included in this distribution
27  * in the file called COPYING.
28  *
29  * Contact Information:
30  *  Intel Linux Wireless <linuxwifi@intel.com>
31  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
32  *
33  * BSD LICENSE
34  *
35  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
36  * Copyright(c) 2013 - 2014 Intel Mobile Communications GmbH
37  * All rights reserved.
38  *
39  * Redistribution and use in source and binary forms, with or without
40  * modification, are permitted provided that the following conditions
41  * are met:
42  *
43  *  * Redistributions of source code must retain the above copyright
44  *    notice, this list of conditions and the following disclaimer.
45  *  * Redistributions in binary form must reproduce the above copyright
46  *    notice, this list of conditions and the following disclaimer in
47  *    the documentation and/or other materials provided with the
48  *    distribution.
49  *  * Neither the name Intel Corporation nor the names of its
50  *    contributors may be used to endorse or promote products derived
51  *    from this software without specific prior written permission.
52  *
53  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
54  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
55  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
56  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
57  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
58  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
59  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
60  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
61  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
62  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
63  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
64  *
65  *****************************************************************************/
66 #include <linux/kernel.h>
67 #include <linux/slab.h>
68 #include <linux/skbuff.h>
69 #include <linux/netdevice.h>
70 #include <linux/etherdevice.h>
71 #include <linux/ip.h>
72 #include <linux/if_arp.h>
73 #include <linux/time.h>
74 #include <net/mac80211.h>
75 #include <net/ieee80211_radiotap.h>
76 #include <net/tcp.h>
77
78 #include "iwl-op-mode.h"
79 #include "iwl-io.h"
80 #include "mvm.h"
81 #include "sta.h"
82 #include "time-event.h"
83 #include "iwl-eeprom-parse.h"
84 #include "iwl-phy-db.h"
85 #include "testmode.h"
86 #include "iwl-fw-error-dump.h"
87 #include "iwl-prph.h"
88 #include "iwl-nvm-parse.h"
89 #include "fw-dbg.h"
90
91 static const struct ieee80211_iface_limit iwl_mvm_limits[] = {
92         {
93                 .max = 1,
94                 .types = BIT(NL80211_IFTYPE_STATION),
95         },
96         {
97                 .max = 1,
98                 .types = BIT(NL80211_IFTYPE_AP) |
99                         BIT(NL80211_IFTYPE_P2P_CLIENT) |
100                         BIT(NL80211_IFTYPE_P2P_GO),
101         },
102         {
103                 .max = 1,
104                 .types = BIT(NL80211_IFTYPE_P2P_DEVICE),
105         },
106 };
107
108 static const struct ieee80211_iface_combination iwl_mvm_iface_combinations[] = {
109         {
110                 .num_different_channels = 2,
111                 .max_interfaces = 3,
112                 .limits = iwl_mvm_limits,
113                 .n_limits = ARRAY_SIZE(iwl_mvm_limits),
114         },
115 };
116
117 #ifdef CONFIG_PM_SLEEP
118 static const struct nl80211_wowlan_tcp_data_token_feature
119 iwl_mvm_wowlan_tcp_token_feature = {
120         .min_len = 0,
121         .max_len = 255,
122         .bufsize = IWL_WOWLAN_REMOTE_WAKE_MAX_TOKENS,
123 };
124
125 static const struct wiphy_wowlan_tcp_support iwl_mvm_wowlan_tcp_support = {
126         .tok = &iwl_mvm_wowlan_tcp_token_feature,
127         .data_payload_max = IWL_WOWLAN_TCP_MAX_PACKET_LEN -
128                             sizeof(struct ethhdr) -
129                             sizeof(struct iphdr) -
130                             sizeof(struct tcphdr),
131         .data_interval_max = 65535, /* __le16 in API */
132         .wake_payload_max = IWL_WOWLAN_REMOTE_WAKE_MAX_PACKET_LEN -
133                             sizeof(struct ethhdr) -
134                             sizeof(struct iphdr) -
135                             sizeof(struct tcphdr),
136         .seq = true,
137 };
138 #endif
139
140 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
141 /*
142  * Use the reserved field to indicate magic values.
143  * these values will only be used internally by the driver,
144  * and won't make it to the fw (reserved will be 0).
145  * BC_FILTER_MAGIC_IP - configure the val of this attribute to
146  *      be the vif's ip address. in case there is not a single
147  *      ip address (0, or more than 1), this attribute will
148  *      be skipped.
149  * BC_FILTER_MAGIC_MAC - set the val of this attribute to
150  *      the LSB bytes of the vif's mac address
151  */
152 enum {
153         BC_FILTER_MAGIC_NONE = 0,
154         BC_FILTER_MAGIC_IP,
155         BC_FILTER_MAGIC_MAC,
156 };
157
158 static const struct iwl_fw_bcast_filter iwl_mvm_default_bcast_filters[] = {
159         {
160                 /* arp */
161                 .discard = 0,
162                 .frame_type = BCAST_FILTER_FRAME_TYPE_ALL,
163                 .attrs = {
164                         {
165                                 /* frame type - arp, hw type - ethernet */
166                                 .offset_type =
167                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
168                                 .offset = sizeof(rfc1042_header),
169                                 .val = cpu_to_be32(0x08060001),
170                                 .mask = cpu_to_be32(0xffffffff),
171                         },
172                         {
173                                 /* arp dest ip */
174                                 .offset_type =
175                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
176                                 .offset = sizeof(rfc1042_header) + 2 +
177                                           sizeof(struct arphdr) +
178                                           ETH_ALEN + sizeof(__be32) +
179                                           ETH_ALEN,
180                                 .mask = cpu_to_be32(0xffffffff),
181                                 /* mark it as special field */
182                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_IP),
183                         },
184                 },
185         },
186         {
187                 /* dhcp offer bcast */
188                 .discard = 0,
189                 .frame_type = BCAST_FILTER_FRAME_TYPE_IPV4,
190                 .attrs = {
191                         {
192                                 /* udp dest port - 68 (bootp client)*/
193                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
194                                 .offset = offsetof(struct udphdr, dest),
195                                 .val = cpu_to_be32(0x00440000),
196                                 .mask = cpu_to_be32(0xffff0000),
197                         },
198                         {
199                                 /* dhcp - lsb bytes of client hw address */
200                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
201                                 .offset = 38,
202                                 .mask = cpu_to_be32(0xffffffff),
203                                 /* mark it as special field */
204                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_MAC),
205                         },
206                 },
207         },
208         /* last filter must be empty */
209         {},
210 };
211 #endif
212
213 void iwl_mvm_ref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
214 {
215         if (!iwl_mvm_is_d0i3_supported(mvm))
216                 return;
217
218         IWL_DEBUG_RPM(mvm, "Take mvm reference - type %d\n", ref_type);
219         spin_lock_bh(&mvm->refs_lock);
220         mvm->refs[ref_type]++;
221         spin_unlock_bh(&mvm->refs_lock);
222         iwl_trans_ref(mvm->trans);
223 }
224
225 void iwl_mvm_unref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
226 {
227         if (!iwl_mvm_is_d0i3_supported(mvm))
228                 return;
229
230         IWL_DEBUG_RPM(mvm, "Leave mvm reference - type %d\n", ref_type);
231         spin_lock_bh(&mvm->refs_lock);
232         if (WARN_ON(!mvm->refs[ref_type])) {
233                 spin_unlock_bh(&mvm->refs_lock);
234                 return;
235         }
236         mvm->refs[ref_type]--;
237         spin_unlock_bh(&mvm->refs_lock);
238         iwl_trans_unref(mvm->trans);
239 }
240
241 static void iwl_mvm_unref_all_except(struct iwl_mvm *mvm,
242                                      enum iwl_mvm_ref_type except_ref)
243 {
244         int i, j;
245
246         if (!iwl_mvm_is_d0i3_supported(mvm))
247                 return;
248
249         spin_lock_bh(&mvm->refs_lock);
250         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
251                 if (except_ref == i || !mvm->refs[i])
252                         continue;
253
254                 IWL_DEBUG_RPM(mvm, "Cleanup: remove mvm ref type %d (%d)\n",
255                               i, mvm->refs[i]);
256                 for (j = 0; j < mvm->refs[i]; j++)
257                         iwl_trans_unref(mvm->trans);
258                 mvm->refs[i] = 0;
259         }
260         spin_unlock_bh(&mvm->refs_lock);
261 }
262
263 bool iwl_mvm_ref_taken(struct iwl_mvm *mvm)
264 {
265         int i;
266         bool taken = false;
267
268         if (!iwl_mvm_is_d0i3_supported(mvm))
269                 return true;
270
271         spin_lock_bh(&mvm->refs_lock);
272         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
273                 if (mvm->refs[i]) {
274                         taken = true;
275                         break;
276                 }
277         }
278         spin_unlock_bh(&mvm->refs_lock);
279
280         return taken;
281 }
282
283 int iwl_mvm_ref_sync(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
284 {
285         iwl_mvm_ref(mvm, ref_type);
286
287         if (!wait_event_timeout(mvm->d0i3_exit_waitq,
288                                 !test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status),
289                                 HZ)) {
290                 WARN_ON_ONCE(1);
291                 iwl_mvm_unref(mvm, ref_type);
292                 return -EIO;
293         }
294
295         return 0;
296 }
297
298 static void iwl_mvm_reset_phy_ctxts(struct iwl_mvm *mvm)
299 {
300         int i;
301
302         memset(mvm->phy_ctxts, 0, sizeof(mvm->phy_ctxts));
303         for (i = 0; i < NUM_PHY_CTX; i++) {
304                 mvm->phy_ctxts[i].id = i;
305                 mvm->phy_ctxts[i].ref = 0;
306         }
307 }
308
309 struct ieee80211_regdomain *iwl_mvm_get_regdomain(struct wiphy *wiphy,
310                                                   const char *alpha2,
311                                                   enum iwl_mcc_source src_id,
312                                                   bool *changed)
313 {
314         struct ieee80211_regdomain *regd = NULL;
315         struct ieee80211_hw *hw = wiphy_to_ieee80211_hw(wiphy);
316         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
317         struct iwl_mcc_update_resp *resp;
318
319         IWL_DEBUG_LAR(mvm, "Getting regdomain data for %s from FW\n", alpha2);
320
321         lockdep_assert_held(&mvm->mutex);
322
323         resp = iwl_mvm_update_mcc(mvm, alpha2, src_id);
324         if (IS_ERR_OR_NULL(resp)) {
325                 IWL_DEBUG_LAR(mvm, "Could not get update from FW %d\n",
326                               PTR_ERR_OR_ZERO(resp));
327                 goto out;
328         }
329
330         if (changed)
331                 *changed = (resp->status == MCC_RESP_NEW_CHAN_PROFILE);
332
333         regd = iwl_parse_nvm_mcc_info(mvm->trans->dev, mvm->cfg,
334                                       __le32_to_cpu(resp->n_channels),
335                                       resp->channels,
336                                       __le16_to_cpu(resp->mcc));
337         /* Store the return source id */
338         src_id = resp->source_id;
339         kfree(resp);
340         if (IS_ERR_OR_NULL(regd)) {
341                 IWL_DEBUG_LAR(mvm, "Could not get parse update from FW %d\n",
342                               PTR_ERR_OR_ZERO(regd));
343                 goto out;
344         }
345
346         IWL_DEBUG_LAR(mvm, "setting alpha2 from FW to %s (0x%x, 0x%x) src=%d\n",
347                       regd->alpha2, regd->alpha2[0], regd->alpha2[1], src_id);
348         mvm->lar_regdom_set = true;
349         mvm->mcc_src = src_id;
350
351 out:
352         return regd;
353 }
354
355 void iwl_mvm_update_changed_regdom(struct iwl_mvm *mvm)
356 {
357         bool changed;
358         struct ieee80211_regdomain *regd;
359
360         if (!iwl_mvm_is_lar_supported(mvm))
361                 return;
362
363         regd = iwl_mvm_get_current_regdomain(mvm, &changed);
364         if (!IS_ERR_OR_NULL(regd)) {
365                 /* only update the regulatory core if changed */
366                 if (changed)
367                         regulatory_set_wiphy_regd(mvm->hw->wiphy, regd);
368
369                 kfree(regd);
370         }
371 }
372
373 struct ieee80211_regdomain *iwl_mvm_get_current_regdomain(struct iwl_mvm *mvm,
374                                                           bool *changed)
375 {
376         return iwl_mvm_get_regdomain(mvm->hw->wiphy, "ZZ",
377                                      iwl_mvm_is_wifi_mcc_supported(mvm) ?
378                                      MCC_SOURCE_GET_CURRENT :
379                                      MCC_SOURCE_OLD_FW, changed);
380 }
381
382 int iwl_mvm_init_fw_regd(struct iwl_mvm *mvm)
383 {
384         enum iwl_mcc_source used_src;
385         struct ieee80211_regdomain *regd;
386         int ret;
387         bool changed;
388         const struct ieee80211_regdomain *r =
389                         rtnl_dereference(mvm->hw->wiphy->regd);
390
391         if (!r)
392                 return -ENOENT;
393
394         /* save the last source in case we overwrite it below */
395         used_src = mvm->mcc_src;
396         if (iwl_mvm_is_wifi_mcc_supported(mvm)) {
397                 /* Notify the firmware we support wifi location updates */
398                 regd = iwl_mvm_get_current_regdomain(mvm, NULL);
399                 if (!IS_ERR_OR_NULL(regd))
400                         kfree(regd);
401         }
402
403         /* Now set our last stored MCC and source */
404         regd = iwl_mvm_get_regdomain(mvm->hw->wiphy, r->alpha2, used_src,
405                                      &changed);
406         if (IS_ERR_OR_NULL(regd))
407                 return -EIO;
408
409         /* update cfg80211 if the regdomain was changed */
410         if (changed)
411                 ret = regulatory_set_wiphy_regd_sync_rtnl(mvm->hw->wiphy, regd);
412         else
413                 ret = 0;
414
415         kfree(regd);
416         return ret;
417 }
418
419 int iwl_mvm_mac_setup_register(struct iwl_mvm *mvm)
420 {
421         struct ieee80211_hw *hw = mvm->hw;
422         int num_mac, ret, i;
423         static const u32 mvm_ciphers[] = {
424                 WLAN_CIPHER_SUITE_WEP40,
425                 WLAN_CIPHER_SUITE_WEP104,
426                 WLAN_CIPHER_SUITE_TKIP,
427                 WLAN_CIPHER_SUITE_CCMP,
428         };
429
430         /* Tell mac80211 our characteristics */
431         ieee80211_hw_set(hw, SIGNAL_DBM);
432         ieee80211_hw_set(hw, SPECTRUM_MGMT);
433         ieee80211_hw_set(hw, REPORTS_TX_ACK_STATUS);
434         ieee80211_hw_set(hw, QUEUE_CONTROL);
435         ieee80211_hw_set(hw, WANT_MONITOR_VIF);
436         ieee80211_hw_set(hw, SUPPORTS_PS);
437         ieee80211_hw_set(hw, SUPPORTS_DYNAMIC_PS);
438         ieee80211_hw_set(hw, AMPDU_AGGREGATION);
439         ieee80211_hw_set(hw, TIMING_BEACON_ONLY);
440         ieee80211_hw_set(hw, CONNECTION_MONITOR);
441         ieee80211_hw_set(hw, CHANCTX_STA_CSA);
442         ieee80211_hw_set(hw, SUPPORT_FAST_XMIT);
443         ieee80211_hw_set(hw, SUPPORTS_CLONED_SKBS);
444         ieee80211_hw_set(hw, SUPPORTS_AMSDU_IN_AMPDU);
445         ieee80211_hw_set(hw, NEEDS_UNIQUE_STA_ADDR);
446         if (iwl_mvm_has_new_rx_api(mvm))
447                 ieee80211_hw_set(hw, SUPPORTS_REORDERING_BUFFER);
448
449         if (mvm->trans->num_rx_queues > 1)
450                 ieee80211_hw_set(hw, USES_RSS);
451
452         if (mvm->trans->max_skb_frags)
453                 hw->netdev_features = NETIF_F_HIGHDMA | NETIF_F_SG;
454
455         if (!iwl_mvm_is_dqa_supported(mvm))
456                 hw->queues = mvm->first_agg_queue;
457         else
458                 hw->queues = IEEE80211_MAX_QUEUES;
459         hw->offchannel_tx_hw_queue = IWL_MVM_OFFCHANNEL_QUEUE;
460         hw->radiotap_mcs_details |= IEEE80211_RADIOTAP_MCS_HAVE_FEC |
461                                     IEEE80211_RADIOTAP_MCS_HAVE_STBC;
462         hw->radiotap_vht_details |= IEEE80211_RADIOTAP_VHT_KNOWN_STBC |
463                 IEEE80211_RADIOTAP_VHT_KNOWN_BEAMFORMED;
464         hw->rate_control_algorithm = "iwl-mvm-rs";
465         hw->uapsd_queues = IWL_MVM_UAPSD_QUEUES;
466         hw->uapsd_max_sp_len = IWL_UAPSD_MAX_SP;
467
468         BUILD_BUG_ON(ARRAY_SIZE(mvm->ciphers) < ARRAY_SIZE(mvm_ciphers) + 2);
469         memcpy(mvm->ciphers, mvm_ciphers, sizeof(mvm_ciphers));
470         hw->wiphy->n_cipher_suites = ARRAY_SIZE(mvm_ciphers);
471         hw->wiphy->cipher_suites = mvm->ciphers;
472
473         /*
474          * Enable 11w if advertised by firmware and software crypto
475          * is not enabled (as the firmware will interpret some mgmt
476          * packets, so enabling it with software crypto isn't safe)
477          */
478         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_MFP &&
479             !iwlwifi_mod_params.sw_crypto) {
480                 ieee80211_hw_set(hw, MFP_CAPABLE);
481                 mvm->ciphers[hw->wiphy->n_cipher_suites] =
482                         WLAN_CIPHER_SUITE_AES_CMAC;
483                 hw->wiphy->n_cipher_suites++;
484         }
485
486         /* currently FW API supports only one optional cipher scheme */
487         if (mvm->fw->cs[0].cipher) {
488                 mvm->hw->n_cipher_schemes = 1;
489                 mvm->hw->cipher_schemes = &mvm->fw->cs[0];
490                 mvm->ciphers[hw->wiphy->n_cipher_suites] =
491                         mvm->fw->cs[0].cipher;
492                 hw->wiphy->n_cipher_suites++;
493         }
494
495         ieee80211_hw_set(hw, SINGLE_SCAN_ON_ALL_BANDS);
496         hw->wiphy->features |=
497                 NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR |
498                 NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR |
499                 NL80211_FEATURE_ND_RANDOM_MAC_ADDR;
500
501         hw->sta_data_size = sizeof(struct iwl_mvm_sta);
502         hw->vif_data_size = sizeof(struct iwl_mvm_vif);
503         hw->chanctx_data_size = sizeof(u16);
504
505         hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION) |
506                 BIT(NL80211_IFTYPE_P2P_CLIENT) |
507                 BIT(NL80211_IFTYPE_AP) |
508                 BIT(NL80211_IFTYPE_P2P_GO) |
509                 BIT(NL80211_IFTYPE_P2P_DEVICE) |
510                 BIT(NL80211_IFTYPE_ADHOC);
511
512         hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
513         hw->wiphy->regulatory_flags |= REGULATORY_ENABLE_RELAX_NO_IR;
514         if (iwl_mvm_is_lar_supported(mvm))
515                 hw->wiphy->regulatory_flags |= REGULATORY_WIPHY_SELF_MANAGED;
516         else
517                 hw->wiphy->regulatory_flags |= REGULATORY_CUSTOM_REG |
518                                                REGULATORY_DISABLE_BEACON_HINTS;
519
520         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_GO_UAPSD)
521                 hw->wiphy->flags |= WIPHY_FLAG_AP_UAPSD;
522
523         hw->wiphy->flags |= WIPHY_FLAG_HAS_CHANNEL_SWITCH;
524
525         hw->wiphy->iface_combinations = iwl_mvm_iface_combinations;
526         hw->wiphy->n_iface_combinations =
527                 ARRAY_SIZE(iwl_mvm_iface_combinations);
528
529         hw->wiphy->max_remain_on_channel_duration = 10000;
530         hw->max_listen_interval = IWL_CONN_MAX_LISTEN_INTERVAL;
531         /* we can compensate an offset of up to 3 channels = 15 MHz */
532         hw->wiphy->max_adj_channel_rssi_comp = 3 * 5;
533
534         /* Extract MAC address */
535         memcpy(mvm->addresses[0].addr, mvm->nvm_data->hw_addr, ETH_ALEN);
536         hw->wiphy->addresses = mvm->addresses;
537         hw->wiphy->n_addresses = 1;
538
539         /* Extract additional MAC addresses if available */
540         num_mac = (mvm->nvm_data->n_hw_addrs > 1) ?
541                 min(IWL_MVM_MAX_ADDRESSES, mvm->nvm_data->n_hw_addrs) : 1;
542
543         for (i = 1; i < num_mac; i++) {
544                 memcpy(mvm->addresses[i].addr, mvm->addresses[i-1].addr,
545                        ETH_ALEN);
546                 mvm->addresses[i].addr[5]++;
547                 hw->wiphy->n_addresses++;
548         }
549
550         iwl_mvm_reset_phy_ctxts(mvm);
551
552         hw->wiphy->max_scan_ie_len = iwl_mvm_max_scan_ie_len(mvm);
553
554         hw->wiphy->max_scan_ssids = PROBE_OPTION_MAX;
555
556         BUILD_BUG_ON(IWL_MVM_SCAN_STOPPING_MASK & IWL_MVM_SCAN_MASK);
557         BUILD_BUG_ON(IWL_MVM_MAX_UMAC_SCANS > HWEIGHT32(IWL_MVM_SCAN_MASK) ||
558                      IWL_MVM_MAX_LMAC_SCANS > HWEIGHT32(IWL_MVM_SCAN_MASK));
559
560         if (fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_UMAC_SCAN))
561                 mvm->max_scans = IWL_MVM_MAX_UMAC_SCANS;
562         else
563                 mvm->max_scans = IWL_MVM_MAX_LMAC_SCANS;
564
565         if (mvm->nvm_data->bands[NL80211_BAND_2GHZ].n_channels)
566                 hw->wiphy->bands[NL80211_BAND_2GHZ] =
567                         &mvm->nvm_data->bands[NL80211_BAND_2GHZ];
568         if (mvm->nvm_data->bands[NL80211_BAND_5GHZ].n_channels) {
569                 hw->wiphy->bands[NL80211_BAND_5GHZ] =
570                         &mvm->nvm_data->bands[NL80211_BAND_5GHZ];
571
572                 if (fw_has_capa(&mvm->fw->ucode_capa,
573                                 IWL_UCODE_TLV_CAPA_BEAMFORMER) &&
574                     fw_has_api(&mvm->fw->ucode_capa,
575                                IWL_UCODE_TLV_API_LQ_SS_PARAMS))
576                         hw->wiphy->bands[NL80211_BAND_5GHZ]->vht_cap.cap |=
577                                 IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE;
578         }
579
580         hw->wiphy->hw_version = mvm->trans->hw_id;
581
582         if (iwlmvm_mod_params.power_scheme != IWL_POWER_SCHEME_CAM)
583                 hw->wiphy->flags |= WIPHY_FLAG_PS_ON_BY_DEFAULT;
584         else
585                 hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
586
587         hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_SCHED_SCAN;
588         hw->wiphy->max_sched_scan_ssids = PROBE_OPTION_MAX;
589         hw->wiphy->max_match_sets = IWL_SCAN_MAX_PROFILES;
590         /* we create the 802.11 header and zero length SSID IE. */
591         hw->wiphy->max_sched_scan_ie_len =
592                 SCAN_OFFLOAD_PROBE_REQ_SIZE - 24 - 2;
593         hw->wiphy->max_sched_scan_plans = IWL_MAX_SCHED_SCAN_PLANS;
594         hw->wiphy->max_sched_scan_plan_interval = U16_MAX;
595
596         /*
597          * the firmware uses u8 for num of iterations, but 0xff is saved for
598          * infinite loop, so the maximum number of iterations is actually 254.
599          */
600         hw->wiphy->max_sched_scan_plan_iterations = 254;
601
602         hw->wiphy->features |= NL80211_FEATURE_P2P_GO_CTWIN |
603                                NL80211_FEATURE_LOW_PRIORITY_SCAN |
604                                NL80211_FEATURE_P2P_GO_OPPPS |
605                                NL80211_FEATURE_DYNAMIC_SMPS |
606                                NL80211_FEATURE_STATIC_SMPS |
607                                NL80211_FEATURE_SUPPORTS_WMM_ADMISSION;
608
609         if (fw_has_capa(&mvm->fw->ucode_capa,
610                         IWL_UCODE_TLV_CAPA_TXPOWER_INSERTION_SUPPORT))
611                 hw->wiphy->features |= NL80211_FEATURE_TX_POWER_INSERTION;
612         if (fw_has_capa(&mvm->fw->ucode_capa,
613                         IWL_UCODE_TLV_CAPA_QUIET_PERIOD_SUPPORT))
614                 hw->wiphy->features |= NL80211_FEATURE_QUIET;
615
616         if (fw_has_capa(&mvm->fw->ucode_capa,
617                         IWL_UCODE_TLV_CAPA_DS_PARAM_SET_IE_SUPPORT))
618                 hw->wiphy->features |=
619                         NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES;
620
621         if (fw_has_capa(&mvm->fw->ucode_capa,
622                         IWL_UCODE_TLV_CAPA_WFA_TPC_REP_IE_SUPPORT))
623                 hw->wiphy->features |= NL80211_FEATURE_WFA_TPC_IE_IN_PROBES;
624
625         mvm->rts_threshold = IEEE80211_MAX_RTS_THRESHOLD;
626
627 #ifdef CONFIG_PM_SLEEP
628         if (iwl_mvm_is_d0i3_supported(mvm) &&
629             device_can_wakeup(mvm->trans->dev)) {
630                 mvm->wowlan.flags = WIPHY_WOWLAN_ANY;
631                 hw->wiphy->wowlan = &mvm->wowlan;
632         }
633
634         if (mvm->fw->img[IWL_UCODE_WOWLAN].sec[0].len &&
635             mvm->trans->ops->d3_suspend &&
636             mvm->trans->ops->d3_resume &&
637             device_can_wakeup(mvm->trans->dev)) {
638                 mvm->wowlan.flags |= WIPHY_WOWLAN_MAGIC_PKT |
639                                      WIPHY_WOWLAN_DISCONNECT |
640                                      WIPHY_WOWLAN_EAP_IDENTITY_REQ |
641                                      WIPHY_WOWLAN_RFKILL_RELEASE |
642                                      WIPHY_WOWLAN_NET_DETECT;
643                 if (!iwlwifi_mod_params.sw_crypto)
644                         mvm->wowlan.flags |= WIPHY_WOWLAN_SUPPORTS_GTK_REKEY |
645                                              WIPHY_WOWLAN_GTK_REKEY_FAILURE |
646                                              WIPHY_WOWLAN_4WAY_HANDSHAKE;
647
648                 mvm->wowlan.n_patterns = IWL_WOWLAN_MAX_PATTERNS;
649                 mvm->wowlan.pattern_min_len = IWL_WOWLAN_MIN_PATTERN_LEN;
650                 mvm->wowlan.pattern_max_len = IWL_WOWLAN_MAX_PATTERN_LEN;
651                 mvm->wowlan.max_nd_match_sets = IWL_SCAN_MAX_PROFILES;
652                 mvm->wowlan.tcp = &iwl_mvm_wowlan_tcp_support;
653                 hw->wiphy->wowlan = &mvm->wowlan;
654         }
655 #endif
656
657 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
658         /* assign default bcast filtering configuration */
659         mvm->bcast_filters = iwl_mvm_default_bcast_filters;
660 #endif
661
662         ret = iwl_mvm_leds_init(mvm);
663         if (ret)
664                 return ret;
665
666         if (fw_has_capa(&mvm->fw->ucode_capa,
667                         IWL_UCODE_TLV_CAPA_TDLS_SUPPORT)) {
668                 IWL_DEBUG_TDLS(mvm, "TDLS supported\n");
669                 hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_TDLS;
670                 ieee80211_hw_set(hw, TDLS_WIDER_BW);
671         }
672
673         if (fw_has_capa(&mvm->fw->ucode_capa,
674                         IWL_UCODE_TLV_CAPA_TDLS_CHANNEL_SWITCH)) {
675                 IWL_DEBUG_TDLS(mvm, "TDLS channel switch supported\n");
676                 hw->wiphy->features |= NL80211_FEATURE_TDLS_CHANNEL_SWITCH;
677         }
678
679         hw->netdev_features |= mvm->cfg->features;
680         if (!iwl_mvm_is_csum_supported(mvm)) {
681                 hw->netdev_features &= ~(IWL_TX_CSUM_NETIF_FLAGS |
682                                          NETIF_F_RXCSUM);
683                 /* We may support SW TX CSUM */
684                 if (IWL_MVM_SW_TX_CSUM_OFFLOAD)
685                         hw->netdev_features |= IWL_TX_CSUM_NETIF_FLAGS;
686         }
687
688         ret = ieee80211_register_hw(mvm->hw);
689         if (ret)
690                 iwl_mvm_leds_exit(mvm);
691
692         return ret;
693 }
694
695 static bool iwl_mvm_defer_tx(struct iwl_mvm *mvm,
696                              struct ieee80211_sta *sta,
697                              struct sk_buff *skb)
698 {
699         struct iwl_mvm_sta *mvmsta;
700         bool defer = false;
701
702         /*
703          * double check the IN_D0I3 flag both before and after
704          * taking the spinlock, in order to prevent taking
705          * the spinlock when not needed.
706          */
707         if (likely(!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status)))
708                 return false;
709
710         spin_lock(&mvm->d0i3_tx_lock);
711         /*
712          * testing the flag again ensures the skb dequeue
713          * loop (on d0i3 exit) hasn't run yet.
714          */
715         if (!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status))
716                 goto out;
717
718         mvmsta = iwl_mvm_sta_from_mac80211(sta);
719         if (mvmsta->sta_id == IWL_MVM_STATION_COUNT ||
720             mvmsta->sta_id != mvm->d0i3_ap_sta_id)
721                 goto out;
722
723         __skb_queue_tail(&mvm->d0i3_tx, skb);
724         ieee80211_stop_queues(mvm->hw);
725
726         /* trigger wakeup */
727         iwl_mvm_ref(mvm, IWL_MVM_REF_TX);
728         iwl_mvm_unref(mvm, IWL_MVM_REF_TX);
729
730         defer = true;
731 out:
732         spin_unlock(&mvm->d0i3_tx_lock);
733         return defer;
734 }
735
736 static void iwl_mvm_mac_tx(struct ieee80211_hw *hw,
737                            struct ieee80211_tx_control *control,
738                            struct sk_buff *skb)
739 {
740         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
741         struct ieee80211_sta *sta = control->sta;
742         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
743         struct ieee80211_hdr *hdr = (void *)skb->data;
744
745         if (iwl_mvm_is_radio_killed(mvm)) {
746                 IWL_DEBUG_DROP(mvm, "Dropping - RF/CT KILL\n");
747                 goto drop;
748         }
749
750         if (IEEE80211_SKB_CB(skb)->hw_queue == IWL_MVM_OFFCHANNEL_QUEUE &&
751             !test_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status) &&
752             !test_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status))
753                 goto drop;
754
755         /* treat non-bufferable MMPDUs as broadcast if sta is sleeping */
756         if (unlikely(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER &&
757                      ieee80211_is_mgmt(hdr->frame_control) &&
758                      !ieee80211_is_deauth(hdr->frame_control) &&
759                      !ieee80211_is_disassoc(hdr->frame_control) &&
760                      !ieee80211_is_action(hdr->frame_control)))
761                 sta = NULL;
762
763         if (sta) {
764                 if (iwl_mvm_defer_tx(mvm, sta, skb))
765                         return;
766                 if (iwl_mvm_tx_skb(mvm, skb, sta))
767                         goto drop;
768                 return;
769         }
770
771         if (iwl_mvm_tx_skb_non_sta(mvm, skb))
772                 goto drop;
773         return;
774  drop:
775         ieee80211_free_txskb(hw, skb);
776 }
777
778 static inline bool iwl_enable_rx_ampdu(const struct iwl_cfg *cfg)
779 {
780         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_RXAGG)
781                 return false;
782         return true;
783 }
784
785 static inline bool iwl_enable_tx_ampdu(const struct iwl_cfg *cfg)
786 {
787         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_TXAGG)
788                 return false;
789         if (iwlwifi_mod_params.disable_11n & IWL_ENABLE_HT_TXAGG)
790                 return true;
791
792         /* enabled by default */
793         return true;
794 }
795
796 #define CHECK_BA_TRIGGER(_mvm, _trig, _tid_bm, _tid, _fmt...)   \
797         do {                                                    \
798                 if (!(le16_to_cpu(_tid_bm) & BIT(_tid)))        \
799                         break;                                  \
800                 iwl_mvm_fw_dbg_collect_trig(_mvm, _trig, _fmt); \
801         } while (0)
802
803 static void
804 iwl_mvm_ampdu_check_trigger(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
805                             struct ieee80211_sta *sta, u16 tid, u16 rx_ba_ssn,
806                             enum ieee80211_ampdu_mlme_action action)
807 {
808         struct iwl_fw_dbg_trigger_tlv *trig;
809         struct iwl_fw_dbg_trigger_ba *ba_trig;
810
811         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
812                 return;
813
814         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
815         ba_trig = (void *)trig->data;
816
817         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
818                 return;
819
820         switch (action) {
821         case IEEE80211_AMPDU_TX_OPERATIONAL: {
822                 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
823                 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
824
825                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->tx_ba_start, tid,
826                                  "TX AGG START: MAC %pM tid %d ssn %d\n",
827                                  sta->addr, tid, tid_data->ssn);
828                 break;
829                 }
830         case IEEE80211_AMPDU_TX_STOP_CONT:
831                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->tx_ba_stop, tid,
832                                  "TX AGG STOP: MAC %pM tid %d\n",
833                                  sta->addr, tid);
834                 break;
835         case IEEE80211_AMPDU_RX_START:
836                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->rx_ba_start, tid,
837                                  "RX AGG START: MAC %pM tid %d ssn %d\n",
838                                  sta->addr, tid, rx_ba_ssn);
839                 break;
840         case IEEE80211_AMPDU_RX_STOP:
841                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->rx_ba_stop, tid,
842                                  "RX AGG STOP: MAC %pM tid %d\n",
843                                  sta->addr, tid);
844                 break;
845         default:
846                 break;
847         }
848 }
849
850 static int iwl_mvm_mac_ampdu_action(struct ieee80211_hw *hw,
851                                     struct ieee80211_vif *vif,
852                                     struct ieee80211_ampdu_params *params)
853 {
854         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
855         int ret;
856         bool tx_agg_ref = false;
857         struct ieee80211_sta *sta = params->sta;
858         enum ieee80211_ampdu_mlme_action action = params->action;
859         u16 tid = params->tid;
860         u16 *ssn = &params->ssn;
861         u8 buf_size = params->buf_size;
862         bool amsdu = params->amsdu;
863         u16 timeout = params->timeout;
864
865         IWL_DEBUG_HT(mvm, "A-MPDU action on addr %pM tid %d: action %d\n",
866                      sta->addr, tid, action);
867
868         if (!(mvm->nvm_data->sku_cap_11n_enable))
869                 return -EACCES;
870
871         /* return from D0i3 before starting a new Tx aggregation */
872         switch (action) {
873         case IEEE80211_AMPDU_TX_START:
874         case IEEE80211_AMPDU_TX_STOP_CONT:
875         case IEEE80211_AMPDU_TX_STOP_FLUSH:
876         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
877         case IEEE80211_AMPDU_TX_OPERATIONAL:
878                 /*
879                  * for tx start, wait synchronously until D0i3 exit to
880                  * get the correct sequence number for the tid.
881                  * additionally, some other ampdu actions use direct
882                  * target access, which is not handled automatically
883                  * by the trans layer (unlike commands), so wait for
884                  * d0i3 exit in these cases as well.
885                  */
886                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_TX_AGG);
887                 if (ret)
888                         return ret;
889
890                 tx_agg_ref = true;
891                 break;
892         default:
893                 break;
894         }
895
896         mutex_lock(&mvm->mutex);
897
898         switch (action) {
899         case IEEE80211_AMPDU_RX_START:
900                 if (!iwl_enable_rx_ampdu(mvm->cfg)) {
901                         ret = -EINVAL;
902                         break;
903                 }
904                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, *ssn, true, buf_size,
905                                          timeout);
906                 break;
907         case IEEE80211_AMPDU_RX_STOP:
908                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, 0, false, buf_size,
909                                          timeout);
910                 break;
911         case IEEE80211_AMPDU_TX_START:
912                 if (!iwl_enable_tx_ampdu(mvm->cfg)) {
913                         ret = -EINVAL;
914                         break;
915                 }
916                 ret = iwl_mvm_sta_tx_agg_start(mvm, vif, sta, tid, ssn);
917                 break;
918         case IEEE80211_AMPDU_TX_STOP_CONT:
919                 ret = iwl_mvm_sta_tx_agg_stop(mvm, vif, sta, tid);
920                 break;
921         case IEEE80211_AMPDU_TX_STOP_FLUSH:
922         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
923                 ret = iwl_mvm_sta_tx_agg_flush(mvm, vif, sta, tid);
924                 break;
925         case IEEE80211_AMPDU_TX_OPERATIONAL:
926                 ret = iwl_mvm_sta_tx_agg_oper(mvm, vif, sta, tid,
927                                               buf_size, amsdu);
928                 break;
929         default:
930                 WARN_ON_ONCE(1);
931                 ret = -EINVAL;
932                 break;
933         }
934
935         if (!ret) {
936                 u16 rx_ba_ssn = 0;
937
938                 if (action == IEEE80211_AMPDU_RX_START)
939                         rx_ba_ssn = *ssn;
940
941                 iwl_mvm_ampdu_check_trigger(mvm, vif, sta, tid,
942                                             rx_ba_ssn, action);
943         }
944         mutex_unlock(&mvm->mutex);
945
946         /*
947          * If the tid is marked as started, we won't use it for offloaded
948          * traffic on the next D0i3 entry. It's safe to unref.
949          */
950         if (tx_agg_ref)
951                 iwl_mvm_unref(mvm, IWL_MVM_REF_TX_AGG);
952
953         return ret;
954 }
955
956 static void iwl_mvm_cleanup_iterator(void *data, u8 *mac,
957                                      struct ieee80211_vif *vif)
958 {
959         struct iwl_mvm *mvm = data;
960         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
961
962         mvmvif->uploaded = false;
963         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
964
965         spin_lock_bh(&mvm->time_event_lock);
966         iwl_mvm_te_clear_data(mvm, &mvmvif->time_event_data);
967         spin_unlock_bh(&mvm->time_event_lock);
968
969         mvmvif->phy_ctxt = NULL;
970         memset(&mvmvif->bf_data, 0, sizeof(mvmvif->bf_data));
971 }
972
973 static void iwl_mvm_restart_cleanup(struct iwl_mvm *mvm)
974 {
975         /* clear the D3 reconfig, we only need it to avoid dumping a
976          * firmware coredump on reconfiguration, we shouldn't do that
977          * on D3->D0 transition
978          */
979         if (!test_and_clear_bit(IWL_MVM_STATUS_D3_RECONFIG, &mvm->status)) {
980                 mvm->fw_dump_desc = &iwl_mvm_dump_desc_assert;
981                 iwl_mvm_fw_error_dump(mvm);
982         }
983
984         /* cleanup all stale references (scan, roc), but keep the
985          * ucode_down ref until reconfig is complete
986          */
987         iwl_mvm_unref_all_except(mvm, IWL_MVM_REF_UCODE_DOWN);
988
989         iwl_mvm_stop_device(mvm);
990
991         mvm->scan_status = 0;
992         mvm->ps_disabled = false;
993         mvm->calibrating = false;
994
995         /* just in case one was running */
996         iwl_mvm_cleanup_roc_te(mvm);
997         ieee80211_remain_on_channel_expired(mvm->hw);
998
999         /*
1000          * cleanup all interfaces, even inactive ones, as some might have
1001          * gone down during the HW restart
1002          */
1003         ieee80211_iterate_interfaces(mvm->hw, 0, iwl_mvm_cleanup_iterator, mvm);
1004
1005         mvm->p2p_device_vif = NULL;
1006         mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1007
1008         iwl_mvm_reset_phy_ctxts(mvm);
1009         memset(mvm->fw_key_table, 0, sizeof(mvm->fw_key_table));
1010         memset(mvm->sta_drained, 0, sizeof(mvm->sta_drained));
1011         memset(mvm->sta_deferred_frames, 0, sizeof(mvm->sta_deferred_frames));
1012         memset(mvm->tfd_drained, 0, sizeof(mvm->tfd_drained));
1013         memset(&mvm->last_bt_notif, 0, sizeof(mvm->last_bt_notif));
1014         memset(&mvm->last_bt_notif_old, 0, sizeof(mvm->last_bt_notif_old));
1015         memset(&mvm->last_bt_ci_cmd, 0, sizeof(mvm->last_bt_ci_cmd));
1016         memset(&mvm->last_bt_ci_cmd_old, 0, sizeof(mvm->last_bt_ci_cmd_old));
1017         memset(&mvm->bt_ack_kill_msk, 0, sizeof(mvm->bt_ack_kill_msk));
1018         memset(&mvm->bt_cts_kill_msk, 0, sizeof(mvm->bt_cts_kill_msk));
1019
1020         ieee80211_wake_queues(mvm->hw);
1021
1022         /* clear any stale d0i3 state */
1023         clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
1024
1025         mvm->vif_count = 0;
1026         mvm->rx_ba_sessions = 0;
1027         mvm->fw_dbg_conf = FW_DBG_INVALID;
1028
1029         /* keep statistics ticking */
1030         iwl_mvm_accu_radio_stats(mvm);
1031 }
1032
1033 int __iwl_mvm_mac_start(struct iwl_mvm *mvm)
1034 {
1035         int ret;
1036
1037         lockdep_assert_held(&mvm->mutex);
1038
1039         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1040                 /* Clean up some internal and mac80211 state on restart */
1041                 iwl_mvm_restart_cleanup(mvm);
1042         } else {
1043                 /* Hold the reference to prevent runtime suspend while
1044                  * the start procedure runs.  It's a bit confusing
1045                  * that the UCODE_DOWN reference is taken, but it just
1046                  * means "UCODE is not UP yet". ( TODO: rename this
1047                  * reference).
1048                  */
1049                 iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
1050         }
1051         ret = iwl_mvm_up(mvm);
1052
1053         if (ret && test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1054                 /* Something went wrong - we need to finish some cleanup
1055                  * that normally iwl_mvm_mac_restart_complete() below
1056                  * would do.
1057                  */
1058                 clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
1059                 iwl_mvm_d0i3_enable_tx(mvm, NULL);
1060         }
1061
1062         return ret;
1063 }
1064
1065 static int iwl_mvm_mac_start(struct ieee80211_hw *hw)
1066 {
1067         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1068         int ret;
1069
1070         /* Some hw restart cleanups must not hold the mutex */
1071         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1072                 /*
1073                  * Make sure we are out of d0i3. This is needed
1074                  * to make sure the reference accounting is correct
1075                  * (and there is no stale d0i3_exit_work).
1076                  */
1077                 wait_event_timeout(mvm->d0i3_exit_waitq,
1078                                    !test_bit(IWL_MVM_STATUS_IN_D0I3,
1079                                              &mvm->status),
1080                                    HZ);
1081         }
1082
1083         mutex_lock(&mvm->mutex);
1084         ret = __iwl_mvm_mac_start(mvm);
1085         mutex_unlock(&mvm->mutex);
1086
1087         return ret;
1088 }
1089
1090 static void iwl_mvm_restart_complete(struct iwl_mvm *mvm)
1091 {
1092         int ret;
1093
1094         mutex_lock(&mvm->mutex);
1095
1096         clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
1097         iwl_mvm_d0i3_enable_tx(mvm, NULL);
1098         ret = iwl_mvm_update_quotas(mvm, true, NULL);
1099         if (ret)
1100                 IWL_ERR(mvm, "Failed to update quotas after restart (%d)\n",
1101                         ret);
1102
1103         /* allow transport/FW low power modes */
1104         iwl_mvm_unref(mvm, IWL_MVM_REF_UCODE_DOWN);
1105
1106         /*
1107          * If we have TDLS peers, remove them. We don't know the last seqno/PN
1108          * of packets the FW sent out, so we must reconnect.
1109          */
1110         iwl_mvm_teardown_tdls_peers(mvm);
1111
1112         mutex_unlock(&mvm->mutex);
1113 }
1114
1115 static void iwl_mvm_resume_complete(struct iwl_mvm *mvm)
1116 {
1117         if (iwl_mvm_is_d0i3_supported(mvm) &&
1118             iwl_mvm_enter_d0i3_on_suspend(mvm))
1119                 WARN_ONCE(!wait_event_timeout(mvm->d0i3_exit_waitq,
1120                                               !test_bit(IWL_MVM_STATUS_IN_D0I3,
1121                                                         &mvm->status),
1122                                               HZ),
1123                           "D0i3 exit on resume timed out\n");
1124 }
1125
1126 static void
1127 iwl_mvm_mac_reconfig_complete(struct ieee80211_hw *hw,
1128                               enum ieee80211_reconfig_type reconfig_type)
1129 {
1130         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1131
1132         switch (reconfig_type) {
1133         case IEEE80211_RECONFIG_TYPE_RESTART:
1134                 iwl_mvm_restart_complete(mvm);
1135                 break;
1136         case IEEE80211_RECONFIG_TYPE_SUSPEND:
1137                 iwl_mvm_resume_complete(mvm);
1138                 break;
1139         }
1140 }
1141
1142 void __iwl_mvm_mac_stop(struct iwl_mvm *mvm)
1143 {
1144         lockdep_assert_held(&mvm->mutex);
1145
1146         /* firmware counters are obviously reset now, but we shouldn't
1147          * partially track so also clear the fw_reset_accu counters.
1148          */
1149         memset(&mvm->accu_radio_stats, 0, sizeof(mvm->accu_radio_stats));
1150
1151         /* async_handlers_wk is now blocked */
1152
1153         /*
1154          * The work item could be running or queued if the
1155          * ROC time event stops just as we get here.
1156          */
1157         flush_work(&mvm->roc_done_wk);
1158
1159         iwl_mvm_stop_device(mvm);
1160
1161         iwl_mvm_async_handlers_purge(mvm);
1162         /* async_handlers_list is empty and will stay empty: HW is stopped */
1163
1164         /* the fw is stopped, the aux sta is dead: clean up driver state */
1165         iwl_mvm_del_aux_sta(mvm);
1166
1167         iwl_free_fw_paging(mvm);
1168
1169         /*
1170          * Clear IN_HW_RESTART flag when stopping the hw (as restart_complete()
1171          * won't be called in this case).
1172          * But make sure to cleanup interfaces that have gone down before/during
1173          * HW restart was requested.
1174          */
1175         if (test_and_clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1176                 ieee80211_iterate_interfaces(mvm->hw, 0,
1177                                              iwl_mvm_cleanup_iterator, mvm);
1178
1179         /* We shouldn't have any UIDs still set.  Loop over all the UIDs to
1180          * make sure there's nothing left there and warn if any is found.
1181          */
1182         if (fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_UMAC_SCAN)) {
1183                 int i;
1184
1185                 for (i = 0; i < mvm->max_scans; i++) {
1186                         if (WARN_ONCE(mvm->scan_uid_status[i],
1187                                       "UMAC scan UID %d status was not cleaned\n",
1188                                       i))
1189                                 mvm->scan_uid_status[i] = 0;
1190                 }
1191         }
1192 }
1193
1194 static void iwl_mvm_mac_stop(struct ieee80211_hw *hw)
1195 {
1196         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1197
1198         flush_work(&mvm->d0i3_exit_work);
1199         flush_work(&mvm->async_handlers_wk);
1200         flush_work(&mvm->add_stream_wk);
1201         cancel_delayed_work_sync(&mvm->fw_dump_wk);
1202         cancel_delayed_work_sync(&mvm->cs_tx_unblock_dwork);
1203         iwl_mvm_free_fw_dump_desc(mvm);
1204
1205         mutex_lock(&mvm->mutex);
1206         __iwl_mvm_mac_stop(mvm);
1207         mutex_unlock(&mvm->mutex);
1208
1209         /*
1210          * The worker might have been waiting for the mutex, let it run and
1211          * discover that its list is now empty.
1212          */
1213         cancel_work_sync(&mvm->async_handlers_wk);
1214 }
1215
1216 static struct iwl_mvm_phy_ctxt *iwl_mvm_get_free_phy_ctxt(struct iwl_mvm *mvm)
1217 {
1218         u16 i;
1219
1220         lockdep_assert_held(&mvm->mutex);
1221
1222         for (i = 0; i < NUM_PHY_CTX; i++)
1223                 if (!mvm->phy_ctxts[i].ref)
1224                         return &mvm->phy_ctxts[i];
1225
1226         IWL_ERR(mvm, "No available PHY context\n");
1227         return NULL;
1228 }
1229
1230 static int iwl_mvm_set_tx_power(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1231                                 s16 tx_power)
1232 {
1233         struct iwl_dev_tx_power_cmd cmd = {
1234                 .v2.set_mode = cpu_to_le32(IWL_TX_POWER_MODE_SET_MAC),
1235                 .v2.mac_context_id =
1236                         cpu_to_le32(iwl_mvm_vif_from_mac80211(vif)->id),
1237                 .v2.pwr_restriction = cpu_to_le16(8 * tx_power),
1238         };
1239         int len = sizeof(cmd);
1240
1241         if (tx_power == IWL_DEFAULT_MAX_TX_POWER)
1242                 cmd.v2.pwr_restriction = cpu_to_le16(IWL_DEV_MAX_TX_POWER);
1243
1244         if (!fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_TX_POWER_CHAIN))
1245                 len = sizeof(cmd.v2);
1246
1247         return iwl_mvm_send_cmd_pdu(mvm, REDUCE_TX_POWER_CMD, 0, len, &cmd);
1248 }
1249
1250 static int iwl_mvm_mac_add_interface(struct ieee80211_hw *hw,
1251                                      struct ieee80211_vif *vif)
1252 {
1253         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1254         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1255         int ret;
1256
1257         mvmvif->mvm = mvm;
1258
1259         /*
1260          * make sure D0i3 exit is completed, otherwise a target access
1261          * during tx queue configuration could be done when still in
1262          * D0i3 state.
1263          */
1264         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_ADD_IF);
1265         if (ret)
1266                 return ret;
1267
1268         /*
1269          * Not much to do here. The stack will not allow interface
1270          * types or combinations that we didn't advertise, so we
1271          * don't really have to check the types.
1272          */
1273
1274         mutex_lock(&mvm->mutex);
1275
1276         /* make sure that beacon statistics don't go backwards with FW reset */
1277         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1278                 mvmvif->beacon_stats.accu_num_beacons +=
1279                         mvmvif->beacon_stats.num_beacons;
1280
1281         /* Allocate resources for the MAC context, and add it to the fw  */
1282         ret = iwl_mvm_mac_ctxt_init(mvm, vif);
1283         if (ret)
1284                 goto out_unlock;
1285
1286         /* Counting number of interfaces is needed for legacy PM */
1287         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1288                 mvm->vif_count++;
1289
1290         /*
1291          * The AP binding flow can be done only after the beacon
1292          * template is configured (which happens only in the mac80211
1293          * start_ap() flow), and adding the broadcast station can happen
1294          * only after the binding.
1295          * In addition, since modifying the MAC before adding a bcast
1296          * station is not allowed by the FW, delay the adding of MAC context to
1297          * the point where we can also add the bcast station.
1298          * In short: there's not much we can do at this point, other than
1299          * allocating resources :)
1300          */
1301         if (vif->type == NL80211_IFTYPE_AP ||
1302             vif->type == NL80211_IFTYPE_ADHOC) {
1303                 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
1304                 if (ret) {
1305                         IWL_ERR(mvm, "Failed to allocate bcast sta\n");
1306                         goto out_release;
1307                 }
1308
1309                 iwl_mvm_vif_dbgfs_register(mvm, vif);
1310                 goto out_unlock;
1311         }
1312
1313         mvmvif->features |= hw->netdev_features;
1314
1315         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
1316         if (ret)
1317                 goto out_release;
1318
1319         ret = iwl_mvm_power_update_mac(mvm);
1320         if (ret)
1321                 goto out_remove_mac;
1322
1323         /* beacon filtering */
1324         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
1325         if (ret)
1326                 goto out_remove_mac;
1327
1328         if (!mvm->bf_allowed_vif &&
1329             vif->type == NL80211_IFTYPE_STATION && !vif->p2p) {
1330                 mvm->bf_allowed_vif = mvmvif;
1331                 vif->driver_flags |= IEEE80211_VIF_BEACON_FILTER |
1332                                      IEEE80211_VIF_SUPPORTS_CQM_RSSI;
1333         }
1334
1335         /*
1336          * P2P_DEVICE interface does not have a channel context assigned to it,
1337          * so a dedicated PHY context is allocated to it and the corresponding
1338          * MAC context is bound to it at this stage.
1339          */
1340         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1341
1342                 mvmvif->phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
1343                 if (!mvmvif->phy_ctxt) {
1344                         ret = -ENOSPC;
1345                         goto out_free_bf;
1346                 }
1347
1348                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
1349                 ret = iwl_mvm_binding_add_vif(mvm, vif);
1350                 if (ret)
1351                         goto out_unref_phy;
1352
1353                 ret = iwl_mvm_add_bcast_sta(mvm, vif);
1354                 if (ret)
1355                         goto out_unbind;
1356
1357                 /* Save a pointer to p2p device vif, so it can later be used to
1358                  * update the p2p device MAC when a GO is started/stopped */
1359                 mvm->p2p_device_vif = vif;
1360         }
1361
1362         iwl_mvm_vif_dbgfs_register(mvm, vif);
1363         goto out_unlock;
1364
1365  out_unbind:
1366         iwl_mvm_binding_remove_vif(mvm, vif);
1367  out_unref_phy:
1368         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1369  out_free_bf:
1370         if (mvm->bf_allowed_vif == mvmvif) {
1371                 mvm->bf_allowed_vif = NULL;
1372                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1373                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1374         }
1375  out_remove_mac:
1376         mvmvif->phy_ctxt = NULL;
1377         iwl_mvm_mac_ctxt_remove(mvm, vif);
1378  out_release:
1379         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1380                 mvm->vif_count--;
1381
1382         iwl_mvm_mac_ctxt_release(mvm, vif);
1383  out_unlock:
1384         mutex_unlock(&mvm->mutex);
1385
1386         iwl_mvm_unref(mvm, IWL_MVM_REF_ADD_IF);
1387
1388         return ret;
1389 }
1390
1391 static void iwl_mvm_prepare_mac_removal(struct iwl_mvm *mvm,
1392                                         struct ieee80211_vif *vif)
1393 {
1394         u32 tfd_msk = iwl_mvm_mac_get_queues_mask(vif);
1395
1396         if (tfd_msk) {
1397                 /*
1398                  * mac80211 first removes all the stations of the vif and
1399                  * then removes the vif. When it removes a station it also
1400                  * flushes the AMPDU session. So by now, all the AMPDU sessions
1401                  * of all the stations of this vif are closed, and the queues
1402                  * of these AMPDU sessions are properly closed.
1403                  * We still need to take care of the shared queues of the vif.
1404                  * Flush them here.
1405                  */
1406                 mutex_lock(&mvm->mutex);
1407                 iwl_mvm_flush_tx_path(mvm, tfd_msk, 0);
1408                 mutex_unlock(&mvm->mutex);
1409
1410                 /*
1411                  * There are transports that buffer a few frames in the host.
1412                  * For these, the flush above isn't enough since while we were
1413                  * flushing, the transport might have sent more frames to the
1414                  * device. To solve this, wait here until the transport is
1415                  * empty. Technically, this could have replaced the flush
1416                  * above, but flush is much faster than draining. So flush
1417                  * first, and drain to make sure we have no frames in the
1418                  * transport anymore.
1419                  * If a station still had frames on the shared queues, it is
1420                  * already marked as draining, so to complete the draining, we
1421                  * just need to wait until the transport is empty.
1422                  */
1423                 iwl_trans_wait_tx_queue_empty(mvm->trans, tfd_msk);
1424         }
1425
1426         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1427                 /*
1428                  * Flush the ROC worker which will flush the OFFCHANNEL queue.
1429                  * We assume here that all the packets sent to the OFFCHANNEL
1430                  * queue are sent in ROC session.
1431                  */
1432                 flush_work(&mvm->roc_done_wk);
1433         } else {
1434                 /*
1435                  * By now, all the AC queues are empty. The AGG queues are
1436                  * empty too. We already got all the Tx responses for all the
1437                  * packets in the queues. The drain work can have been
1438                  * triggered. Flush it.
1439                  */
1440                 flush_work(&mvm->sta_drained_wk);
1441         }
1442 }
1443
1444 static void iwl_mvm_mac_remove_interface(struct ieee80211_hw *hw,
1445                                          struct ieee80211_vif *vif)
1446 {
1447         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1448         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1449
1450         iwl_mvm_prepare_mac_removal(mvm, vif);
1451
1452         mutex_lock(&mvm->mutex);
1453
1454         if (mvm->bf_allowed_vif == mvmvif) {
1455                 mvm->bf_allowed_vif = NULL;
1456                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1457                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1458         }
1459
1460         iwl_mvm_vif_dbgfs_clean(mvm, vif);
1461
1462         /*
1463          * For AP/GO interface, the tear down of the resources allocated to the
1464          * interface is be handled as part of the stop_ap flow.
1465          */
1466         if (vif->type == NL80211_IFTYPE_AP ||
1467             vif->type == NL80211_IFTYPE_ADHOC) {
1468 #ifdef CONFIG_NL80211_TESTMODE
1469                 if (vif == mvm->noa_vif) {
1470                         mvm->noa_vif = NULL;
1471                         mvm->noa_duration = 0;
1472                 }
1473 #endif
1474                 iwl_mvm_dealloc_bcast_sta(mvm, vif);
1475                 goto out_release;
1476         }
1477
1478         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1479                 mvm->p2p_device_vif = NULL;
1480                 iwl_mvm_rm_bcast_sta(mvm, vif);
1481                 iwl_mvm_binding_remove_vif(mvm, vif);
1482                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1483                 mvmvif->phy_ctxt = NULL;
1484         }
1485
1486         if (mvm->vif_count && vif->type != NL80211_IFTYPE_P2P_DEVICE)
1487                 mvm->vif_count--;
1488
1489         iwl_mvm_power_update_mac(mvm);
1490         iwl_mvm_mac_ctxt_remove(mvm, vif);
1491
1492 out_release:
1493         iwl_mvm_mac_ctxt_release(mvm, vif);
1494         mutex_unlock(&mvm->mutex);
1495 }
1496
1497 static int iwl_mvm_mac_config(struct ieee80211_hw *hw, u32 changed)
1498 {
1499         return 0;
1500 }
1501
1502 struct iwl_mvm_mc_iter_data {
1503         struct iwl_mvm *mvm;
1504         int port_id;
1505 };
1506
1507 static void iwl_mvm_mc_iface_iterator(void *_data, u8 *mac,
1508                                       struct ieee80211_vif *vif)
1509 {
1510         struct iwl_mvm_mc_iter_data *data = _data;
1511         struct iwl_mvm *mvm = data->mvm;
1512         struct iwl_mcast_filter_cmd *cmd = mvm->mcast_filter_cmd;
1513         int ret, len;
1514
1515         /* if we don't have free ports, mcast frames will be dropped */
1516         if (WARN_ON_ONCE(data->port_id >= MAX_PORT_ID_NUM))
1517                 return;
1518
1519         if (vif->type != NL80211_IFTYPE_STATION ||
1520             !vif->bss_conf.assoc)
1521                 return;
1522
1523         cmd->port_id = data->port_id++;
1524         memcpy(cmd->bssid, vif->bss_conf.bssid, ETH_ALEN);
1525         len = roundup(sizeof(*cmd) + cmd->count * ETH_ALEN, 4);
1526
1527         ret = iwl_mvm_send_cmd_pdu(mvm, MCAST_FILTER_CMD, CMD_ASYNC, len, cmd);
1528         if (ret)
1529                 IWL_ERR(mvm, "mcast filter cmd error. ret=%d\n", ret);
1530 }
1531
1532 static void iwl_mvm_recalc_multicast(struct iwl_mvm *mvm)
1533 {
1534         struct iwl_mvm_mc_iter_data iter_data = {
1535                 .mvm = mvm,
1536         };
1537
1538         lockdep_assert_held(&mvm->mutex);
1539
1540         if (WARN_ON_ONCE(!mvm->mcast_filter_cmd))
1541                 return;
1542
1543         ieee80211_iterate_active_interfaces_atomic(
1544                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1545                 iwl_mvm_mc_iface_iterator, &iter_data);
1546 }
1547
1548 static u64 iwl_mvm_prepare_multicast(struct ieee80211_hw *hw,
1549                                      struct netdev_hw_addr_list *mc_list)
1550 {
1551         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1552         struct iwl_mcast_filter_cmd *cmd;
1553         struct netdev_hw_addr *addr;
1554         int addr_count;
1555         bool pass_all;
1556         int len;
1557
1558         addr_count = netdev_hw_addr_list_count(mc_list);
1559         pass_all = addr_count > MAX_MCAST_FILTERING_ADDRESSES ||
1560                    IWL_MVM_FW_MCAST_FILTER_PASS_ALL;
1561         if (pass_all)
1562                 addr_count = 0;
1563
1564         len = roundup(sizeof(*cmd) + addr_count * ETH_ALEN, 4);
1565         cmd = kzalloc(len, GFP_ATOMIC);
1566         if (!cmd)
1567                 return 0;
1568
1569         if (pass_all) {
1570                 cmd->pass_all = 1;
1571                 return (u64)(unsigned long)cmd;
1572         }
1573
1574         netdev_hw_addr_list_for_each(addr, mc_list) {
1575                 IWL_DEBUG_MAC80211(mvm, "mcast addr (%d): %pM\n",
1576                                    cmd->count, addr->addr);
1577                 memcpy(&cmd->addr_list[cmd->count * ETH_ALEN],
1578                        addr->addr, ETH_ALEN);
1579                 cmd->count++;
1580         }
1581
1582         return (u64)(unsigned long)cmd;
1583 }
1584
1585 static void iwl_mvm_configure_filter(struct ieee80211_hw *hw,
1586                                      unsigned int changed_flags,
1587                                      unsigned int *total_flags,
1588                                      u64 multicast)
1589 {
1590         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1591         struct iwl_mcast_filter_cmd *cmd = (void *)(unsigned long)multicast;
1592
1593         mutex_lock(&mvm->mutex);
1594
1595         /* replace previous configuration */
1596         kfree(mvm->mcast_filter_cmd);
1597         mvm->mcast_filter_cmd = cmd;
1598
1599         if (!cmd)
1600                 goto out;
1601
1602         iwl_mvm_recalc_multicast(mvm);
1603 out:
1604         mutex_unlock(&mvm->mutex);
1605         *total_flags = 0;
1606 }
1607
1608 static void iwl_mvm_config_iface_filter(struct ieee80211_hw *hw,
1609                                         struct ieee80211_vif *vif,
1610                                         unsigned int filter_flags,
1611                                         unsigned int changed_flags)
1612 {
1613         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1614
1615         /* We support only filter for probe requests */
1616         if (!(changed_flags & FIF_PROBE_REQ))
1617                 return;
1618
1619         /* Supported only for p2p client interfaces */
1620         if (vif->type != NL80211_IFTYPE_STATION || !vif->bss_conf.assoc ||
1621             !vif->p2p)
1622                 return;
1623
1624         mutex_lock(&mvm->mutex);
1625         iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
1626         mutex_unlock(&mvm->mutex);
1627 }
1628
1629 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
1630 struct iwl_bcast_iter_data {
1631         struct iwl_mvm *mvm;
1632         struct iwl_bcast_filter_cmd *cmd;
1633         u8 current_filter;
1634 };
1635
1636 static void
1637 iwl_mvm_set_bcast_filter(struct ieee80211_vif *vif,
1638                          const struct iwl_fw_bcast_filter *in_filter,
1639                          struct iwl_fw_bcast_filter *out_filter)
1640 {
1641         struct iwl_fw_bcast_filter_attr *attr;
1642         int i;
1643
1644         memcpy(out_filter, in_filter, sizeof(*out_filter));
1645
1646         for (i = 0; i < ARRAY_SIZE(out_filter->attrs); i++) {
1647                 attr = &out_filter->attrs[i];
1648
1649                 if (!attr->mask)
1650                         break;
1651
1652                 switch (attr->reserved1) {
1653                 case cpu_to_le16(BC_FILTER_MAGIC_IP):
1654                         if (vif->bss_conf.arp_addr_cnt != 1) {
1655                                 attr->mask = 0;
1656                                 continue;
1657                         }
1658
1659                         attr->val = vif->bss_conf.arp_addr_list[0];
1660                         break;
1661                 case cpu_to_le16(BC_FILTER_MAGIC_MAC):
1662                         attr->val = *(__be32 *)&vif->addr[2];
1663                         break;
1664                 default:
1665                         break;
1666                 }
1667                 attr->reserved1 = 0;
1668                 out_filter->num_attrs++;
1669         }
1670 }
1671
1672 static void iwl_mvm_bcast_filter_iterator(void *_data, u8 *mac,
1673                                           struct ieee80211_vif *vif)
1674 {
1675         struct iwl_bcast_iter_data *data = _data;
1676         struct iwl_mvm *mvm = data->mvm;
1677         struct iwl_bcast_filter_cmd *cmd = data->cmd;
1678         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1679         struct iwl_fw_bcast_mac *bcast_mac;
1680         int i;
1681
1682         if (WARN_ON(mvmvif->id >= ARRAY_SIZE(cmd->macs)))
1683                 return;
1684
1685         bcast_mac = &cmd->macs[mvmvif->id];
1686
1687         /*
1688          * enable filtering only for associated stations, but not for P2P
1689          * Clients
1690          */
1691         if (vif->type != NL80211_IFTYPE_STATION || vif->p2p ||
1692             !vif->bss_conf.assoc)
1693                 return;
1694
1695         bcast_mac->default_discard = 1;
1696
1697         /* copy all configured filters */
1698         for (i = 0; mvm->bcast_filters[i].attrs[0].mask; i++) {
1699                 /*
1700                  * Make sure we don't exceed our filters limit.
1701                  * if there is still a valid filter to be configured,
1702                  * be on the safe side and just allow bcast for this mac.
1703                  */
1704                 if (WARN_ON_ONCE(data->current_filter >=
1705                                  ARRAY_SIZE(cmd->filters))) {
1706                         bcast_mac->default_discard = 0;
1707                         bcast_mac->attached_filters = 0;
1708                         break;
1709                 }
1710
1711                 iwl_mvm_set_bcast_filter(vif,
1712                                          &mvm->bcast_filters[i],
1713                                          &cmd->filters[data->current_filter]);
1714
1715                 /* skip current filter if it contains no attributes */
1716                 if (!cmd->filters[data->current_filter].num_attrs)
1717                         continue;
1718
1719                 /* attach the filter to current mac */
1720                 bcast_mac->attached_filters |=
1721                                 cpu_to_le16(BIT(data->current_filter));
1722
1723                 data->current_filter++;
1724         }
1725 }
1726
1727 bool iwl_mvm_bcast_filter_build_cmd(struct iwl_mvm *mvm,
1728                                     struct iwl_bcast_filter_cmd *cmd)
1729 {
1730         struct iwl_bcast_iter_data iter_data = {
1731                 .mvm = mvm,
1732                 .cmd = cmd,
1733         };
1734
1735         if (IWL_MVM_FW_BCAST_FILTER_PASS_ALL)
1736                 return false;
1737
1738         memset(cmd, 0, sizeof(*cmd));
1739         cmd->max_bcast_filters = ARRAY_SIZE(cmd->filters);
1740         cmd->max_macs = ARRAY_SIZE(cmd->macs);
1741
1742 #ifdef CONFIG_IWLWIFI_DEBUGFS
1743         /* use debugfs filters/macs if override is configured */
1744         if (mvm->dbgfs_bcast_filtering.override) {
1745                 memcpy(cmd->filters, &mvm->dbgfs_bcast_filtering.cmd.filters,
1746                        sizeof(cmd->filters));
1747                 memcpy(cmd->macs, &mvm->dbgfs_bcast_filtering.cmd.macs,
1748                        sizeof(cmd->macs));
1749                 return true;
1750         }
1751 #endif
1752
1753         /* if no filters are configured, do nothing */
1754         if (!mvm->bcast_filters)
1755                 return false;
1756
1757         /* configure and attach these filters for each associated sta vif */
1758         ieee80211_iterate_active_interfaces(
1759                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1760                 iwl_mvm_bcast_filter_iterator, &iter_data);
1761
1762         return true;
1763 }
1764
1765 static int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm)
1766 {
1767         struct iwl_bcast_filter_cmd cmd;
1768
1769         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_BCAST_FILTERING))
1770                 return 0;
1771
1772         if (!iwl_mvm_bcast_filter_build_cmd(mvm, &cmd))
1773                 return 0;
1774
1775         return iwl_mvm_send_cmd_pdu(mvm, BCAST_FILTER_CMD, 0,
1776                                     sizeof(cmd), &cmd);
1777 }
1778 #else
1779 static inline int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm)
1780 {
1781         return 0;
1782 }
1783 #endif
1784
1785 static int iwl_mvm_update_mu_groups(struct iwl_mvm *mvm,
1786                                     struct ieee80211_vif *vif)
1787 {
1788         struct iwl_mu_group_mgmt_cmd cmd = {};
1789
1790         memcpy(cmd.membership_status, vif->bss_conf.mu_group.membership,
1791                WLAN_MEMBERSHIP_LEN);
1792         memcpy(cmd.user_position, vif->bss_conf.mu_group.position,
1793                WLAN_USER_POSITION_LEN);
1794
1795         return iwl_mvm_send_cmd_pdu(mvm,
1796                                     WIDE_ID(DATA_PATH_GROUP,
1797                                             UPDATE_MU_GROUPS_CMD),
1798                                     0, sizeof(cmd), &cmd);
1799 }
1800
1801 static void iwl_mvm_mu_mimo_iface_iterator(void *_data, u8 *mac,
1802                                            struct ieee80211_vif *vif)
1803 {
1804         if (vif->mu_mimo_owner) {
1805                 struct iwl_mu_group_mgmt_notif *notif = _data;
1806
1807                 /*
1808                  * MU-MIMO Group Id action frame is little endian. We treat
1809                  * the data received from firmware as if it came from the
1810                  * action frame, so no conversion is needed.
1811                  */
1812                 ieee80211_update_mu_groups(vif,
1813                                            (u8 *)&notif->membership_status,
1814                                            (u8 *)&notif->user_position);
1815         }
1816 }
1817
1818 void iwl_mvm_mu_mimo_grp_notif(struct iwl_mvm *mvm,
1819                                struct iwl_rx_cmd_buffer *rxb)
1820 {
1821         struct iwl_rx_packet *pkt = rxb_addr(rxb);
1822         struct iwl_mu_group_mgmt_notif *notif = (void *)pkt->data;
1823
1824         ieee80211_iterate_active_interfaces_atomic(
1825                         mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1826                         iwl_mvm_mu_mimo_iface_iterator, notif);
1827 }
1828
1829 static void iwl_mvm_bss_info_changed_station(struct iwl_mvm *mvm,
1830                                              struct ieee80211_vif *vif,
1831                                              struct ieee80211_bss_conf *bss_conf,
1832                                              u32 changes)
1833 {
1834         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1835         int ret;
1836
1837         /*
1838          * Re-calculate the tsf id, as the master-slave relations depend on the
1839          * beacon interval, which was not known when the station interface was
1840          * added.
1841          */
1842         if (changes & BSS_CHANGED_ASSOC && bss_conf->assoc)
1843                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
1844
1845         if (changes & BSS_CHANGED_ASSOC && !bss_conf->assoc &&
1846             mvmvif->lqm_active)
1847                 iwl_mvm_send_lqm_cmd(vif, LQM_CMD_OPERATION_STOP_MEASUREMENT,
1848                                      0, 0);
1849
1850         /*
1851          * If we're not associated yet, take the (new) BSSID before associating
1852          * so the firmware knows. If we're already associated, then use the old
1853          * BSSID here, and we'll send a cleared one later in the CHANGED_ASSOC
1854          * branch for disassociation below.
1855          */
1856         if (changes & BSS_CHANGED_BSSID && !mvmvif->associated)
1857                 memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1858
1859         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, mvmvif->bssid);
1860         if (ret)
1861                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
1862
1863         /* after sending it once, adopt mac80211 data */
1864         memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1865         mvmvif->associated = bss_conf->assoc;
1866
1867         if (changes & BSS_CHANGED_ASSOC) {
1868                 if (bss_conf->assoc) {
1869                         /* clear statistics to get clean beacon counter */
1870                         iwl_mvm_request_statistics(mvm, true);
1871                         memset(&mvmvif->beacon_stats, 0,
1872                                sizeof(mvmvif->beacon_stats));
1873
1874                         /* add quota for this interface */
1875                         ret = iwl_mvm_update_quotas(mvm, true, NULL);
1876                         if (ret) {
1877                                 IWL_ERR(mvm, "failed to update quotas\n");
1878                                 return;
1879                         }
1880
1881                         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART,
1882                                      &mvm->status)) {
1883                                 /*
1884                                  * If we're restarting then the firmware will
1885                                  * obviously have lost synchronisation with
1886                                  * the AP. It will attempt to synchronise by
1887                                  * itself, but we can make it more reliable by
1888                                  * scheduling a session protection time event.
1889                                  *
1890                                  * The firmware needs to receive a beacon to
1891                                  * catch up with synchronisation, use 110% of
1892                                  * the beacon interval.
1893                                  *
1894                                  * Set a large maximum delay to allow for more
1895                                  * than a single interface.
1896                                  */
1897                                 u32 dur = (11 * vif->bss_conf.beacon_int) / 10;
1898                                 iwl_mvm_protect_session(mvm, vif, dur, dur,
1899                                                         5 * dur, false);
1900                         }
1901
1902                         iwl_mvm_sf_update(mvm, vif, false);
1903                         iwl_mvm_power_vif_assoc(mvm, vif);
1904                         if (vif->p2p) {
1905                                 iwl_mvm_ref(mvm, IWL_MVM_REF_P2P_CLIENT);
1906                                 iwl_mvm_update_smps(mvm, vif,
1907                                                     IWL_MVM_SMPS_REQ_PROT,
1908                                                     IEEE80211_SMPS_DYNAMIC);
1909                         }
1910                 } else if (mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
1911                         /*
1912                          * If update fails - SF might be running in associated
1913                          * mode while disassociated - which is forbidden.
1914                          */
1915                         WARN_ONCE(iwl_mvm_sf_update(mvm, vif, false),
1916                                   "Failed to update SF upon disassociation\n");
1917
1918                         /* remove AP station now that the MAC is unassoc */
1919                         ret = iwl_mvm_rm_sta_id(mvm, vif, mvmvif->ap_sta_id);
1920                         if (ret)
1921                                 IWL_ERR(mvm, "failed to remove AP station\n");
1922
1923                         if (mvm->d0i3_ap_sta_id == mvmvif->ap_sta_id)
1924                                 mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1925                         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
1926                         /* remove quota for this interface */
1927                         ret = iwl_mvm_update_quotas(mvm, false, NULL);
1928                         if (ret)
1929                                 IWL_ERR(mvm, "failed to update quotas\n");
1930
1931                         if (vif->p2p)
1932                                 iwl_mvm_unref(mvm, IWL_MVM_REF_P2P_CLIENT);
1933
1934                         /* this will take the cleared BSSID from bss_conf */
1935                         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
1936                         if (ret)
1937                                 IWL_ERR(mvm,
1938                                         "failed to update MAC %pM (clear after unassoc)\n",
1939                                         vif->addr);
1940                 }
1941
1942                 /*
1943                  * The firmware tracks the MU-MIMO group on its own.
1944                  * However, on HW restart we should restore this data.
1945                  */
1946                 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status) &&
1947                     (changes & BSS_CHANGED_MU_GROUPS) && vif->mu_mimo_owner) {
1948                         ret = iwl_mvm_update_mu_groups(mvm, vif);
1949                         if (ret)
1950                                 IWL_ERR(mvm,
1951                                         "failed to update VHT MU_MIMO groups\n");
1952                 }
1953
1954                 iwl_mvm_recalc_multicast(mvm);
1955                 iwl_mvm_configure_bcast_filter(mvm);
1956
1957                 /* reset rssi values */
1958                 mvmvif->bf_data.ave_beacon_signal = 0;
1959
1960                 iwl_mvm_bt_coex_vif_change(mvm);
1961                 iwl_mvm_update_smps(mvm, vif, IWL_MVM_SMPS_REQ_TT,
1962                                     IEEE80211_SMPS_AUTOMATIC);
1963                 if (fw_has_capa(&mvm->fw->ucode_capa,
1964                                 IWL_UCODE_TLV_CAPA_UMAC_SCAN))
1965                         iwl_mvm_config_scan(mvm);
1966         } else if (changes & BSS_CHANGED_BEACON_INFO) {
1967                 /*
1968                  * We received a beacon _after_ association so
1969                  * remove the session protection.
1970                  */
1971                 iwl_mvm_remove_time_event(mvm, mvmvif,
1972                                           &mvmvif->time_event_data);
1973         }
1974
1975         if (changes & BSS_CHANGED_BEACON_INFO) {
1976                 iwl_mvm_sf_update(mvm, vif, false);
1977                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
1978         }
1979
1980         if (changes & (BSS_CHANGED_PS | BSS_CHANGED_P2P_PS | BSS_CHANGED_QOS |
1981                        /*
1982                         * Send power command on every beacon change,
1983                         * because we may have not enabled beacon abort yet.
1984                         */
1985                        BSS_CHANGED_BEACON_INFO)) {
1986                 ret = iwl_mvm_power_update_mac(mvm);
1987                 if (ret)
1988                         IWL_ERR(mvm, "failed to update power mode\n");
1989         }
1990
1991         if (changes & BSS_CHANGED_TXPOWER) {
1992                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
1993                                 bss_conf->txpower);
1994                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
1995         }
1996
1997         if (changes & BSS_CHANGED_CQM) {
1998                 IWL_DEBUG_MAC80211(mvm, "cqm info_changed\n");
1999                 /* reset cqm events tracking */
2000                 mvmvif->bf_data.last_cqm_event = 0;
2001                 if (mvmvif->bf_data.bf_enabled) {
2002                         ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
2003                         if (ret)
2004                                 IWL_ERR(mvm,
2005                                         "failed to update CQM thresholds\n");
2006                 }
2007         }
2008
2009         if (changes & BSS_CHANGED_ARP_FILTER) {
2010                 IWL_DEBUG_MAC80211(mvm, "arp filter changed\n");
2011                 iwl_mvm_configure_bcast_filter(mvm);
2012         }
2013 }
2014
2015 static int iwl_mvm_start_ap_ibss(struct ieee80211_hw *hw,
2016                                  struct ieee80211_vif *vif)
2017 {
2018         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2019         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2020         int ret;
2021
2022         /*
2023          * iwl_mvm_mac_ctxt_add() might read directly from the device
2024          * (the system time), so make sure it is available.
2025          */
2026         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_START_AP);
2027         if (ret)
2028                 return ret;
2029
2030         mutex_lock(&mvm->mutex);
2031
2032         /* Send the beacon template */
2033         ret = iwl_mvm_mac_ctxt_beacon_changed(mvm, vif);
2034         if (ret)
2035                 goto out_unlock;
2036
2037         /*
2038          * Re-calculate the tsf id, as the master-slave relations depend on the
2039          * beacon interval, which was not known when the AP interface was added.
2040          */
2041         if (vif->type == NL80211_IFTYPE_AP)
2042                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
2043
2044         mvmvif->ap_assoc_sta_count = 0;
2045
2046         /* Add the mac context */
2047         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
2048         if (ret)
2049                 goto out_unlock;
2050
2051         /* Perform the binding */
2052         ret = iwl_mvm_binding_add_vif(mvm, vif);
2053         if (ret)
2054                 goto out_remove;
2055
2056         /* Send the bcast station. At this stage the TBTT and DTIM time events
2057          * are added and applied to the scheduler */
2058         ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
2059         if (ret)
2060                 goto out_unbind;
2061
2062         /* must be set before quota calculations */
2063         mvmvif->ap_ibss_active = true;
2064
2065         /* power updated needs to be done before quotas */
2066         iwl_mvm_power_update_mac(mvm);
2067
2068         ret = iwl_mvm_update_quotas(mvm, false, NULL);
2069         if (ret)
2070                 goto out_quota_failed;
2071
2072         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
2073         if (vif->p2p && mvm->p2p_device_vif)
2074                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
2075
2076         iwl_mvm_ref(mvm, IWL_MVM_REF_AP_IBSS);
2077
2078         iwl_mvm_bt_coex_vif_change(mvm);
2079
2080         /* we don't support TDLS during DCM */
2081         if (iwl_mvm_phy_ctx_count(mvm) > 1)
2082                 iwl_mvm_teardown_tdls_peers(mvm);
2083
2084         goto out_unlock;
2085
2086 out_quota_failed:
2087         iwl_mvm_power_update_mac(mvm);
2088         mvmvif->ap_ibss_active = false;
2089         iwl_mvm_send_rm_bcast_sta(mvm, vif);
2090 out_unbind:
2091         iwl_mvm_binding_remove_vif(mvm, vif);
2092 out_remove:
2093         iwl_mvm_mac_ctxt_remove(mvm, vif);
2094 out_unlock:
2095         mutex_unlock(&mvm->mutex);
2096         iwl_mvm_unref(mvm, IWL_MVM_REF_START_AP);
2097         return ret;
2098 }
2099
2100 static void iwl_mvm_stop_ap_ibss(struct ieee80211_hw *hw,
2101                                  struct ieee80211_vif *vif)
2102 {
2103         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2104         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2105
2106         iwl_mvm_prepare_mac_removal(mvm, vif);
2107
2108         mutex_lock(&mvm->mutex);
2109
2110         /* Handle AP stop while in CSA */
2111         if (rcu_access_pointer(mvm->csa_vif) == vif) {
2112                 iwl_mvm_remove_time_event(mvm, mvmvif,
2113                                           &mvmvif->time_event_data);
2114                 RCU_INIT_POINTER(mvm->csa_vif, NULL);
2115                 mvmvif->csa_countdown = false;
2116         }
2117
2118         if (rcu_access_pointer(mvm->csa_tx_blocked_vif) == vif) {
2119                 RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
2120                 mvm->csa_tx_block_bcn_timeout = 0;
2121         }
2122
2123         mvmvif->ap_ibss_active = false;
2124         mvm->ap_last_beacon_gp2 = 0;
2125
2126         iwl_mvm_bt_coex_vif_change(mvm);
2127
2128         iwl_mvm_unref(mvm, IWL_MVM_REF_AP_IBSS);
2129
2130         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
2131         if (vif->p2p && mvm->p2p_device_vif)
2132                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
2133
2134         iwl_mvm_update_quotas(mvm, false, NULL);
2135         iwl_mvm_send_rm_bcast_sta(mvm, vif);
2136         iwl_mvm_binding_remove_vif(mvm, vif);
2137
2138         iwl_mvm_power_update_mac(mvm);
2139
2140         iwl_mvm_mac_ctxt_remove(mvm, vif);
2141
2142         mutex_unlock(&mvm->mutex);
2143 }
2144
2145 static void
2146 iwl_mvm_bss_info_changed_ap_ibss(struct iwl_mvm *mvm,
2147                                  struct ieee80211_vif *vif,
2148                                  struct ieee80211_bss_conf *bss_conf,
2149                                  u32 changes)
2150 {
2151         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2152
2153         /* Changes will be applied when the AP/IBSS is started */
2154         if (!mvmvif->ap_ibss_active)
2155                 return;
2156
2157         if (changes & (BSS_CHANGED_ERP_CTS_PROT | BSS_CHANGED_HT |
2158                        BSS_CHANGED_BANDWIDTH | BSS_CHANGED_QOS) &&
2159             iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL))
2160                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
2161
2162         /* Need to send a new beacon template to the FW */
2163         if (changes & BSS_CHANGED_BEACON &&
2164             iwl_mvm_mac_ctxt_beacon_changed(mvm, vif))
2165                 IWL_WARN(mvm, "Failed updating beacon data\n");
2166
2167         if (changes & BSS_CHANGED_TXPOWER) {
2168                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
2169                                 bss_conf->txpower);
2170                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
2171         }
2172 }
2173
2174 static void iwl_mvm_bss_info_changed(struct ieee80211_hw *hw,
2175                                      struct ieee80211_vif *vif,
2176                                      struct ieee80211_bss_conf *bss_conf,
2177                                      u32 changes)
2178 {
2179         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2180
2181         /*
2182          * iwl_mvm_bss_info_changed_station() might call
2183          * iwl_mvm_protect_session(), which reads directly from
2184          * the device (the system time), so make sure it is available.
2185          */
2186         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_BSS_CHANGED))
2187                 return;
2188
2189         mutex_lock(&mvm->mutex);
2190
2191         if (changes & BSS_CHANGED_IDLE && !bss_conf->idle)
2192                 iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_SCHED, true);
2193
2194         switch (vif->type) {
2195         case NL80211_IFTYPE_STATION:
2196                 iwl_mvm_bss_info_changed_station(mvm, vif, bss_conf, changes);
2197                 break;
2198         case NL80211_IFTYPE_AP:
2199         case NL80211_IFTYPE_ADHOC:
2200                 iwl_mvm_bss_info_changed_ap_ibss(mvm, vif, bss_conf, changes);
2201                 break;
2202         default:
2203                 /* shouldn't happen */
2204                 WARN_ON_ONCE(1);
2205         }
2206
2207         mutex_unlock(&mvm->mutex);
2208         iwl_mvm_unref(mvm, IWL_MVM_REF_BSS_CHANGED);
2209 }
2210
2211 static int iwl_mvm_mac_hw_scan(struct ieee80211_hw *hw,
2212                                struct ieee80211_vif *vif,
2213                                struct ieee80211_scan_request *hw_req)
2214 {
2215         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2216         int ret;
2217
2218         if (hw_req->req.n_channels == 0 ||
2219             hw_req->req.n_channels > mvm->fw->ucode_capa.n_scan_channels)
2220                 return -EINVAL;
2221
2222         mutex_lock(&mvm->mutex);
2223         ret = iwl_mvm_reg_scan_start(mvm, vif, &hw_req->req, &hw_req->ies);
2224         mutex_unlock(&mvm->mutex);
2225
2226         return ret;
2227 }
2228
2229 static void iwl_mvm_mac_cancel_hw_scan(struct ieee80211_hw *hw,
2230                                        struct ieee80211_vif *vif)
2231 {
2232         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2233
2234         mutex_lock(&mvm->mutex);
2235
2236         /* Due to a race condition, it's possible that mac80211 asks
2237          * us to stop a hw_scan when it's already stopped.  This can
2238          * happen, for instance, if we stopped the scan ourselves,
2239          * called ieee80211_scan_completed() and the userspace called
2240          * cancel scan scan before ieee80211_scan_work() could run.
2241          * To handle that, simply return if the scan is not running.
2242         */
2243         if (mvm->scan_status & IWL_MVM_SCAN_REGULAR)
2244                 iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_REGULAR, true);
2245
2246         mutex_unlock(&mvm->mutex);
2247 }
2248
2249 static void
2250 iwl_mvm_mac_allow_buffered_frames(struct ieee80211_hw *hw,
2251                                   struct ieee80211_sta *sta, u16 tids,
2252                                   int num_frames,
2253                                   enum ieee80211_frame_release_type reason,
2254                                   bool more_data)
2255 {
2256         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2257
2258         /* Called when we need to transmit (a) frame(s) from mac80211 */
2259
2260         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
2261                                           tids, more_data, false);
2262 }
2263
2264 static void
2265 iwl_mvm_mac_release_buffered_frames(struct ieee80211_hw *hw,
2266                                     struct ieee80211_sta *sta, u16 tids,
2267                                     int num_frames,
2268                                     enum ieee80211_frame_release_type reason,
2269                                     bool more_data)
2270 {
2271         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2272
2273         /* Called when we need to transmit (a) frame(s) from agg queue */
2274
2275         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
2276                                           tids, more_data, true);
2277 }
2278
2279 static void iwl_mvm_mac_sta_notify(struct ieee80211_hw *hw,
2280                                    struct ieee80211_vif *vif,
2281                                    enum sta_notify_cmd cmd,
2282                                    struct ieee80211_sta *sta)
2283 {
2284         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2285         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
2286         unsigned long txqs = 0, tids = 0;
2287         int tid;
2288
2289         spin_lock_bh(&mvmsta->lock);
2290         for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
2291                 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
2292
2293                 if (tid_data->state != IWL_AGG_ON &&
2294                     tid_data->state != IWL_EMPTYING_HW_QUEUE_DELBA)
2295                         continue;
2296
2297                 __set_bit(tid_data->txq_id, &txqs);
2298
2299                 if (iwl_mvm_tid_queued(tid_data) == 0)
2300                         continue;
2301
2302                 __set_bit(tid, &tids);
2303         }
2304
2305         switch (cmd) {
2306         case STA_NOTIFY_SLEEP:
2307                 if (atomic_read(&mvm->pending_frames[mvmsta->sta_id]) > 0)
2308                         ieee80211_sta_block_awake(hw, sta, true);
2309
2310                 for_each_set_bit(tid, &tids, IWL_MAX_TID_COUNT)
2311                         ieee80211_sta_set_buffered(sta, tid, true);
2312
2313                 if (txqs)
2314                         iwl_trans_freeze_txq_timer(mvm->trans, txqs, true);
2315                 /*
2316                  * The fw updates the STA to be asleep. Tx packets on the Tx
2317                  * queues to this station will not be transmitted. The fw will
2318                  * send a Tx response with TX_STATUS_FAIL_DEST_PS.
2319                  */
2320                 break;
2321         case STA_NOTIFY_AWAKE:
2322                 if (WARN_ON(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
2323                         break;
2324
2325                 if (txqs)
2326                         iwl_trans_freeze_txq_timer(mvm->trans, txqs, false);
2327                 iwl_mvm_sta_modify_ps_wake(mvm, sta);
2328                 break;
2329         default:
2330                 break;
2331         }
2332         spin_unlock_bh(&mvmsta->lock);
2333 }
2334
2335 static void iwl_mvm_sta_pre_rcu_remove(struct ieee80211_hw *hw,
2336                                        struct ieee80211_vif *vif,
2337                                        struct ieee80211_sta *sta)
2338 {
2339         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2340         struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2341
2342         /*
2343          * This is called before mac80211 does RCU synchronisation,
2344          * so here we already invalidate our internal RCU-protected
2345          * station pointer. The rest of the code will thus no longer
2346          * be able to find the station this way, and we don't rely
2347          * on further RCU synchronisation after the sta_state()
2348          * callback deleted the station.
2349          */
2350         mutex_lock(&mvm->mutex);
2351         if (sta == rcu_access_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id]))
2352                 rcu_assign_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id],
2353                                    ERR_PTR(-ENOENT));
2354
2355         mutex_unlock(&mvm->mutex);
2356 }
2357
2358 static void iwl_mvm_check_uapsd(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2359                                 const u8 *bssid)
2360 {
2361         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_UAPSD_SUPPORT))
2362                 return;
2363
2364         if (vif->p2p && !iwl_mvm_is_p2p_scm_uapsd_supported(mvm)) {
2365                 vif->driver_flags &= ~IEEE80211_VIF_SUPPORTS_UAPSD;
2366                 return;
2367         }
2368
2369         if (!vif->p2p &&
2370             (iwlwifi_mod_params.uapsd_disable & IWL_DISABLE_UAPSD_BSS)) {
2371                 vif->driver_flags &= ~IEEE80211_VIF_SUPPORTS_UAPSD;
2372                 return;
2373         }
2374
2375         vif->driver_flags |= IEEE80211_VIF_SUPPORTS_UAPSD;
2376 }
2377
2378 static void
2379 iwl_mvm_tdls_check_trigger(struct iwl_mvm *mvm,
2380                            struct ieee80211_vif *vif, u8 *peer_addr,
2381                            enum nl80211_tdls_operation action)
2382 {
2383         struct iwl_fw_dbg_trigger_tlv *trig;
2384         struct iwl_fw_dbg_trigger_tdls *tdls_trig;
2385
2386         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_TDLS))
2387                 return;
2388
2389         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_TDLS);
2390         tdls_trig = (void *)trig->data;
2391         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
2392                 return;
2393
2394         if (!(tdls_trig->action_bitmap & BIT(action)))
2395                 return;
2396
2397         if (tdls_trig->peer_mode &&
2398             memcmp(tdls_trig->peer, peer_addr, ETH_ALEN) != 0)
2399                 return;
2400
2401         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
2402                                     "TDLS event occurred, peer %pM, action %d",
2403                                     peer_addr, action);
2404 }
2405
2406 static void iwl_mvm_purge_deferred_tx_frames(struct iwl_mvm *mvm,
2407                                              struct iwl_mvm_sta *mvm_sta)
2408 {
2409         struct iwl_mvm_tid_data *tid_data;
2410         struct sk_buff *skb;
2411         int i;
2412
2413         spin_lock_bh(&mvm_sta->lock);
2414         for (i = 0; i <= IWL_MAX_TID_COUNT; i++) {
2415                 tid_data = &mvm_sta->tid_data[i];
2416                 while ((skb = __skb_dequeue(&tid_data->deferred_tx_frames)))
2417                         ieee80211_free_txskb(mvm->hw, skb);
2418         }
2419         spin_unlock_bh(&mvm_sta->lock);
2420 }
2421
2422 static int iwl_mvm_mac_sta_state(struct ieee80211_hw *hw,
2423                                  struct ieee80211_vif *vif,
2424                                  struct ieee80211_sta *sta,
2425                                  enum ieee80211_sta_state old_state,
2426                                  enum ieee80211_sta_state new_state)
2427 {
2428         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2429         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2430         int ret;
2431
2432         IWL_DEBUG_MAC80211(mvm, "station %pM state change %d->%d\n",
2433                            sta->addr, old_state, new_state);
2434
2435         /* this would be a mac80211 bug ... but don't crash */
2436         if (WARN_ON_ONCE(!mvmvif->phy_ctxt))
2437                 return -EINVAL;
2438
2439         /* if a STA is being removed, reuse its ID */
2440         flush_work(&mvm->sta_drained_wk);
2441
2442         /*
2443          * If we are in a STA removal flow and in DQA mode:
2444          *
2445          * This is after the sync_rcu part, so the queues have already been
2446          * flushed. No more TXs on their way in mac80211's path, and no more in
2447          * the queues.
2448          * Also, we won't be getting any new TX frames for this station.
2449          * What we might have are deferred TX frames that need to be taken care
2450          * of.
2451          *
2452          * Drop any still-queued deferred-frame before removing the STA, and
2453          * make sure the worker is no longer handling frames for this STA.
2454          */
2455         if (old_state == IEEE80211_STA_NONE &&
2456             new_state == IEEE80211_STA_NOTEXIST &&
2457             iwl_mvm_is_dqa_supported(mvm)) {
2458                 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2459
2460                 iwl_mvm_purge_deferred_tx_frames(mvm, mvm_sta);
2461                 flush_work(&mvm->add_stream_wk);
2462
2463                 /*
2464                  * No need to make sure deferred TX indication is off since the
2465                  * worker will already remove it if it was on
2466                  */
2467         }
2468
2469         mutex_lock(&mvm->mutex);
2470         if (old_state == IEEE80211_STA_NOTEXIST &&
2471             new_state == IEEE80211_STA_NONE) {
2472                 /*
2473                  * Firmware bug - it'll crash if the beacon interval is less
2474                  * than 16. We can't avoid connecting at all, so refuse the
2475                  * station state change, this will cause mac80211 to abandon
2476                  * attempts to connect to this AP, and eventually wpa_s will
2477                  * blacklist the AP...
2478                  */
2479                 if (vif->type == NL80211_IFTYPE_STATION &&
2480                     vif->bss_conf.beacon_int < 16) {
2481                         IWL_ERR(mvm,
2482                                 "AP %pM beacon interval is %d, refusing due to firmware bug!\n",
2483                                 sta->addr, vif->bss_conf.beacon_int);
2484                         ret = -EINVAL;
2485                         goto out_unlock;
2486                 }
2487
2488                 if (sta->tdls &&
2489                     (vif->p2p ||
2490                      iwl_mvm_tdls_sta_count(mvm, NULL) ==
2491                                                 IWL_MVM_TDLS_STA_COUNT ||
2492                      iwl_mvm_phy_ctx_count(mvm) > 1)) {
2493                         IWL_DEBUG_MAC80211(mvm, "refusing TDLS sta\n");
2494                         ret = -EBUSY;
2495                         goto out_unlock;
2496                 }
2497
2498                 ret = iwl_mvm_add_sta(mvm, vif, sta);
2499                 if (sta->tdls && ret == 0) {
2500                         iwl_mvm_recalc_tdls_state(mvm, vif, true);
2501                         iwl_mvm_tdls_check_trigger(mvm, vif, sta->addr,
2502                                                    NL80211_TDLS_SETUP);
2503                 }
2504         } else if (old_state == IEEE80211_STA_NONE &&
2505                    new_state == IEEE80211_STA_AUTH) {
2506                 /*
2507                  * EBS may be disabled due to previous failures reported by FW.
2508                  * Reset EBS status here assuming environment has been changed.
2509                  */
2510                 mvm->last_ebs_successful = true;
2511                 iwl_mvm_check_uapsd(mvm, vif, sta->addr);
2512                 ret = 0;
2513         } else if (old_state == IEEE80211_STA_AUTH &&
2514                    new_state == IEEE80211_STA_ASSOC) {
2515                 if (vif->type == NL80211_IFTYPE_AP) {
2516                         mvmvif->ap_assoc_sta_count++;
2517                         iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2518                 }
2519                 ret = iwl_mvm_update_sta(mvm, vif, sta);
2520                 if (ret == 0)
2521                         iwl_mvm_rs_rate_init(mvm, sta,
2522                                              mvmvif->phy_ctxt->channel->band,
2523                                              true);
2524         } else if (old_state == IEEE80211_STA_ASSOC &&
2525                    new_state == IEEE80211_STA_AUTHORIZED) {
2526
2527                 /* we don't support TDLS during DCM */
2528                 if (iwl_mvm_phy_ctx_count(mvm) > 1)
2529                         iwl_mvm_teardown_tdls_peers(mvm);
2530
2531                 if (sta->tdls)
2532                         iwl_mvm_tdls_check_trigger(mvm, vif, sta->addr,
2533                                                    NL80211_TDLS_ENABLE_LINK);
2534
2535                 /* enable beacon filtering */
2536                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
2537                 ret = 0;
2538         } else if (old_state == IEEE80211_STA_AUTHORIZED &&
2539                    new_state == IEEE80211_STA_ASSOC) {
2540                 /* disable beacon filtering */
2541                 WARN_ON(iwl_mvm_disable_beacon_filter(mvm, vif, 0));
2542                 ret = 0;
2543         } else if (old_state == IEEE80211_STA_ASSOC &&
2544                    new_state == IEEE80211_STA_AUTH) {
2545                 if (vif->type == NL80211_IFTYPE_AP) {
2546                         mvmvif->ap_assoc_sta_count--;
2547                         iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2548                 }
2549                 ret = 0;
2550         } else if (old_state == IEEE80211_STA_AUTH &&
2551                    new_state == IEEE80211_STA_NONE) {
2552                 ret = 0;
2553         } else if (old_state == IEEE80211_STA_NONE &&
2554                    new_state == IEEE80211_STA_NOTEXIST) {
2555                 ret = iwl_mvm_rm_sta(mvm, vif, sta);
2556                 if (sta->tdls) {
2557                         iwl_mvm_recalc_tdls_state(mvm, vif, false);
2558                         iwl_mvm_tdls_check_trigger(mvm, vif, sta->addr,
2559                                                    NL80211_TDLS_DISABLE_LINK);
2560                 }
2561         } else {
2562                 ret = -EIO;
2563         }
2564  out_unlock:
2565         mutex_unlock(&mvm->mutex);
2566
2567         if (sta->tdls && ret == 0) {
2568                 if (old_state == IEEE80211_STA_NOTEXIST &&
2569                     new_state == IEEE80211_STA_NONE)
2570                         ieee80211_reserve_tid(sta, IWL_MVM_TDLS_FW_TID);
2571                 else if (old_state == IEEE80211_STA_NONE &&
2572                          new_state == IEEE80211_STA_NOTEXIST)
2573                         ieee80211_unreserve_tid(sta, IWL_MVM_TDLS_FW_TID);
2574         }
2575
2576         return ret;
2577 }
2578
2579 static int iwl_mvm_mac_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
2580 {
2581         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2582
2583         mvm->rts_threshold = value;
2584
2585         return 0;
2586 }
2587
2588 static void iwl_mvm_sta_rc_update(struct ieee80211_hw *hw,
2589                                   struct ieee80211_vif *vif,
2590                                   struct ieee80211_sta *sta, u32 changed)
2591 {
2592         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2593
2594         if (vif->type == NL80211_IFTYPE_STATION &&
2595             changed & IEEE80211_RC_NSS_CHANGED)
2596                 iwl_mvm_sf_update(mvm, vif, false);
2597 }
2598
2599 static int iwl_mvm_mac_conf_tx(struct ieee80211_hw *hw,
2600                                struct ieee80211_vif *vif, u16 ac,
2601                                const struct ieee80211_tx_queue_params *params)
2602 {
2603         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2604         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2605
2606         mvmvif->queue_params[ac] = *params;
2607
2608         /*
2609          * No need to update right away, we'll get BSS_CHANGED_QOS
2610          * The exception is P2P_DEVICE interface which needs immediate update.
2611          */
2612         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
2613                 int ret;
2614
2615                 mutex_lock(&mvm->mutex);
2616                 ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2617                 mutex_unlock(&mvm->mutex);
2618                 return ret;
2619         }
2620         return 0;
2621 }
2622
2623 static void iwl_mvm_mac_mgd_prepare_tx(struct ieee80211_hw *hw,
2624                                       struct ieee80211_vif *vif)
2625 {
2626         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2627         u32 duration = IWL_MVM_TE_SESSION_PROTECTION_MAX_TIME_MS;
2628         u32 min_duration = IWL_MVM_TE_SESSION_PROTECTION_MIN_TIME_MS;
2629
2630         if (WARN_ON_ONCE(vif->bss_conf.assoc))
2631                 return;
2632
2633         /*
2634          * iwl_mvm_protect_session() reads directly from the device
2635          * (the system time), so make sure it is available.
2636          */
2637         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PREPARE_TX))
2638                 return;
2639
2640         mutex_lock(&mvm->mutex);
2641         /* Try really hard to protect the session and hear a beacon */
2642         iwl_mvm_protect_session(mvm, vif, duration, min_duration, 500, false);
2643         mutex_unlock(&mvm->mutex);
2644
2645         iwl_mvm_unref(mvm, IWL_MVM_REF_PREPARE_TX);
2646 }
2647
2648 static int iwl_mvm_mac_sched_scan_start(struct ieee80211_hw *hw,
2649                                         struct ieee80211_vif *vif,
2650                                         struct cfg80211_sched_scan_request *req,
2651                                         struct ieee80211_scan_ies *ies)
2652 {
2653         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2654
2655         int ret;
2656
2657         mutex_lock(&mvm->mutex);
2658
2659         if (!vif->bss_conf.idle) {
2660                 ret = -EBUSY;
2661                 goto out;
2662         }
2663
2664         ret = iwl_mvm_sched_scan_start(mvm, vif, req, ies, IWL_MVM_SCAN_SCHED);
2665
2666 out:
2667         mutex_unlock(&mvm->mutex);
2668         return ret;
2669 }
2670
2671 static int iwl_mvm_mac_sched_scan_stop(struct ieee80211_hw *hw,
2672                                        struct ieee80211_vif *vif)
2673 {
2674         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2675         int ret;
2676
2677         mutex_lock(&mvm->mutex);
2678
2679         /* Due to a race condition, it's possible that mac80211 asks
2680          * us to stop a sched_scan when it's already stopped.  This
2681          * can happen, for instance, if we stopped the scan ourselves,
2682          * called ieee80211_sched_scan_stopped() and the userspace called
2683          * stop sched scan scan before ieee80211_sched_scan_stopped_work()
2684          * could run.  To handle this, simply return if the scan is
2685          * not running.
2686         */
2687         if (!(mvm->scan_status & IWL_MVM_SCAN_SCHED)) {
2688                 mutex_unlock(&mvm->mutex);
2689                 return 0;
2690         }
2691
2692         ret = iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_SCHED, false);
2693         mutex_unlock(&mvm->mutex);
2694         iwl_mvm_wait_for_async_handlers(mvm);
2695
2696         return ret;
2697 }
2698
2699 static int iwl_mvm_mac_set_key(struct ieee80211_hw *hw,
2700                                enum set_key_cmd cmd,
2701                                struct ieee80211_vif *vif,
2702                                struct ieee80211_sta *sta,
2703                                struct ieee80211_key_conf *key)
2704 {
2705         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2706         struct iwl_mvm_sta *mvmsta;
2707         struct iwl_mvm_key_pn *ptk_pn;
2708         int keyidx = key->keyidx;
2709         int ret;
2710         u8 key_offset;
2711
2712         if (iwlwifi_mod_params.sw_crypto) {
2713                 IWL_DEBUG_MAC80211(mvm, "leave - hwcrypto disabled\n");
2714                 return -EOPNOTSUPP;
2715         }
2716
2717         switch (key->cipher) {
2718         case WLAN_CIPHER_SUITE_TKIP:
2719                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
2720                 key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2721                 break;
2722         case WLAN_CIPHER_SUITE_CCMP:
2723                 key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2724                 break;
2725         case WLAN_CIPHER_SUITE_AES_CMAC:
2726                 WARN_ON_ONCE(!ieee80211_hw_check(hw, MFP_CAPABLE));
2727                 break;
2728         case WLAN_CIPHER_SUITE_WEP40:
2729         case WLAN_CIPHER_SUITE_WEP104:
2730                 /* For non-client mode, only use WEP keys for TX as we probably
2731                  * don't have a station yet anyway and would then have to keep
2732                  * track of the keys, linking them to each of the clients/peers
2733                  * as they appear. For now, don't do that, for performance WEP
2734                  * offload doesn't really matter much, but we need it for some
2735                  * other offload features in client mode.
2736                  */
2737                 if (vif->type != NL80211_IFTYPE_STATION)
2738                         return 0;
2739                 break;
2740         default:
2741                 /* currently FW supports only one optional cipher scheme */
2742                 if (hw->n_cipher_schemes &&
2743                     hw->cipher_schemes->cipher == key->cipher)
2744                         key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2745                 else
2746                         return -EOPNOTSUPP;
2747         }
2748
2749         mutex_lock(&mvm->mutex);
2750
2751         switch (cmd) {
2752         case SET_KEY:
2753                 if ((vif->type == NL80211_IFTYPE_ADHOC ||
2754                      vif->type == NL80211_IFTYPE_AP) && !sta) {
2755                         /*
2756                          * GTK on AP interface is a TX-only key, return 0;
2757                          * on IBSS they're per-station and because we're lazy
2758                          * we don't support them for RX, so do the same.
2759                          * CMAC in AP/IBSS modes must be done in software.
2760                          */
2761                         if (key->cipher == WLAN_CIPHER_SUITE_AES_CMAC)
2762                                 ret = -EOPNOTSUPP;
2763                         else
2764                                 ret = 0;
2765                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2766                         break;
2767                 }
2768
2769                 /* During FW restart, in order to restore the state as it was,
2770                  * don't try to reprogram keys we previously failed for.
2771                  */
2772                 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status) &&
2773                     key->hw_key_idx == STA_KEY_IDX_INVALID) {
2774                         IWL_DEBUG_MAC80211(mvm,
2775                                            "skip invalid idx key programming during restart\n");
2776                         ret = 0;
2777                         break;
2778                 }
2779
2780                 if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status) &&
2781                     sta && iwl_mvm_has_new_rx_api(mvm) &&
2782                     key->flags & IEEE80211_KEY_FLAG_PAIRWISE &&
2783                     (key->cipher == WLAN_CIPHER_SUITE_CCMP ||
2784                      key->cipher == WLAN_CIPHER_SUITE_GCMP)) {
2785                         struct ieee80211_key_seq seq;
2786                         int tid, q;
2787
2788                         mvmsta = iwl_mvm_sta_from_mac80211(sta);
2789                         WARN_ON(rcu_access_pointer(mvmsta->ptk_pn[keyidx]));
2790                         ptk_pn = kzalloc(sizeof(*ptk_pn) +
2791                                          mvm->trans->num_rx_queues *
2792                                                 sizeof(ptk_pn->q[0]),
2793                                          GFP_KERNEL);
2794                         if (!ptk_pn) {
2795                                 ret = -ENOMEM;
2796                                 break;
2797                         }
2798
2799                         for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
2800                                 ieee80211_get_key_rx_seq(key, tid, &seq);
2801                                 for (q = 0; q < mvm->trans->num_rx_queues; q++)
2802                                         memcpy(ptk_pn->q[q].pn[tid],
2803                                                seq.ccmp.pn,
2804                                                IEEE80211_CCMP_PN_LEN);
2805                         }
2806
2807                         rcu_assign_pointer(mvmsta->ptk_pn[keyidx], ptk_pn);
2808                 }
2809
2810                 /* in HW restart reuse the index, otherwise request a new one */
2811                 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
2812                         key_offset = key->hw_key_idx;
2813                 else
2814                         key_offset = STA_KEY_IDX_INVALID;
2815
2816                 IWL_DEBUG_MAC80211(mvm, "set hwcrypto key\n");
2817                 ret = iwl_mvm_set_sta_key(mvm, vif, sta, key, key_offset);
2818                 if (ret) {
2819                         IWL_WARN(mvm, "set key failed\n");
2820                         /*
2821                          * can't add key for RX, but we don't need it
2822                          * in the device for TX so still return 0
2823                          */
2824                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2825                         ret = 0;
2826                 }
2827
2828                 break;
2829         case DISABLE_KEY:
2830                 if (key->hw_key_idx == STA_KEY_IDX_INVALID) {
2831                         ret = 0;
2832                         break;
2833                 }
2834
2835                 if (sta && iwl_mvm_has_new_rx_api(mvm) &&
2836                     key->flags & IEEE80211_KEY_FLAG_PAIRWISE &&
2837                     (key->cipher == WLAN_CIPHER_SUITE_CCMP ||
2838                      key->cipher == WLAN_CIPHER_SUITE_GCMP)) {
2839                         mvmsta = iwl_mvm_sta_from_mac80211(sta);
2840                         ptk_pn = rcu_dereference_protected(
2841                                                 mvmsta->ptk_pn[keyidx],
2842                                                 lockdep_is_held(&mvm->mutex));
2843                         RCU_INIT_POINTER(mvmsta->ptk_pn[keyidx], NULL);
2844                         if (ptk_pn)
2845                                 kfree_rcu(ptk_pn, rcu_head);
2846                 }
2847
2848                 IWL_DEBUG_MAC80211(mvm, "disable hwcrypto key\n");
2849                 ret = iwl_mvm_remove_sta_key(mvm, vif, sta, key);
2850                 break;
2851         default:
2852                 ret = -EINVAL;
2853         }
2854
2855         mutex_unlock(&mvm->mutex);
2856         return ret;
2857 }
2858
2859 static void iwl_mvm_mac_update_tkip_key(struct ieee80211_hw *hw,
2860                                         struct ieee80211_vif *vif,
2861                                         struct ieee80211_key_conf *keyconf,
2862                                         struct ieee80211_sta *sta,
2863                                         u32 iv32, u16 *phase1key)
2864 {
2865         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2866
2867         if (keyconf->hw_key_idx == STA_KEY_IDX_INVALID)
2868                 return;
2869
2870         iwl_mvm_update_tkip_key(mvm, vif, keyconf, sta, iv32, phase1key);
2871 }
2872
2873
2874 static bool iwl_mvm_rx_aux_roc(struct iwl_notif_wait_data *notif_wait,
2875                                struct iwl_rx_packet *pkt, void *data)
2876 {
2877         struct iwl_mvm *mvm =
2878                 container_of(notif_wait, struct iwl_mvm, notif_wait);
2879         struct iwl_hs20_roc_res *resp;
2880         int resp_len = iwl_rx_packet_payload_len(pkt);
2881         struct iwl_mvm_time_event_data *te_data = data;
2882
2883         if (WARN_ON(pkt->hdr.cmd != HOT_SPOT_CMD))
2884                 return true;
2885
2886         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
2887                 IWL_ERR(mvm, "Invalid HOT_SPOT_CMD response\n");
2888                 return true;
2889         }
2890
2891         resp = (void *)pkt->data;
2892
2893         IWL_DEBUG_TE(mvm,
2894                      "Aux ROC: Recieved response from ucode: status=%d uid=%d\n",
2895                      resp->status, resp->event_unique_id);
2896
2897         te_data->uid = le32_to_cpu(resp->event_unique_id);
2898         IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
2899                      te_data->uid);
2900
2901         spin_lock_bh(&mvm->time_event_lock);
2902         list_add_tail(&te_data->list, &mvm->aux_roc_te_list);
2903         spin_unlock_bh(&mvm->time_event_lock);
2904
2905         return true;
2906 }
2907
2908 #define AUX_ROC_MIN_DURATION MSEC_TO_TU(100)
2909 #define AUX_ROC_MIN_DELAY MSEC_TO_TU(200)
2910 #define AUX_ROC_MAX_DELAY MSEC_TO_TU(600)
2911 #define AUX_ROC_SAFETY_BUFFER MSEC_TO_TU(20)
2912 #define AUX_ROC_MIN_SAFETY_BUFFER MSEC_TO_TU(10)
2913 static int iwl_mvm_send_aux_roc_cmd(struct iwl_mvm *mvm,
2914                                     struct ieee80211_channel *channel,
2915                                     struct ieee80211_vif *vif,
2916                                     int duration)
2917 {
2918         int res, time_reg = DEVICE_SYSTEM_TIME_REG;
2919         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2920         struct iwl_mvm_time_event_data *te_data = &mvmvif->hs_time_event_data;
2921         static const u16 time_event_response[] = { HOT_SPOT_CMD };
2922         struct iwl_notification_wait wait_time_event;
2923         u32 dtim_interval = vif->bss_conf.dtim_period *
2924                 vif->bss_conf.beacon_int;
2925         u32 req_dur, delay;
2926         struct iwl_hs20_roc_req aux_roc_req = {
2927                 .action = cpu_to_le32(FW_CTXT_ACTION_ADD),
2928                 .id_and_color =
2929                         cpu_to_le32(FW_CMD_ID_AND_COLOR(MAC_INDEX_AUX, 0)),
2930                 .sta_id_and_color = cpu_to_le32(mvm->aux_sta.sta_id),
2931                 /* Set the channel info data */
2932                 .channel_info.band = (channel->band == NL80211_BAND_2GHZ) ?
2933                         PHY_BAND_24 : PHY_BAND_5,
2934                 .channel_info.channel = channel->hw_value,
2935                 .channel_info.width = PHY_VHT_CHANNEL_MODE20,
2936                 /* Set the time and duration */
2937                 .apply_time = cpu_to_le32(iwl_read_prph(mvm->trans, time_reg)),
2938          };
2939
2940         delay = AUX_ROC_MIN_DELAY;
2941         req_dur = MSEC_TO_TU(duration);
2942
2943         /*
2944          * If we are associated we want the delay time to be at least one
2945          * dtim interval so that the FW can wait until after the DTIM and
2946          * then start the time event, this will potentially allow us to
2947          * remain off-channel for the max duration.
2948          * Since we want to use almost a whole dtim interval we would also
2949          * like the delay to be for 2-3 dtim intervals, in case there are
2950          * other time events with higher priority.
2951          */
2952         if (vif->bss_conf.assoc) {
2953                 delay = min_t(u32, dtim_interval * 3, AUX_ROC_MAX_DELAY);
2954                 /* We cannot remain off-channel longer than the DTIM interval */
2955                 if (dtim_interval <= req_dur) {
2956                         req_dur = dtim_interval - AUX_ROC_SAFETY_BUFFER;
2957                         if (req_dur <= AUX_ROC_MIN_DURATION)
2958                                 req_dur = dtim_interval -
2959                                         AUX_ROC_MIN_SAFETY_BUFFER;
2960                 }
2961         }
2962
2963         aux_roc_req.duration = cpu_to_le32(req_dur);
2964         aux_roc_req.apply_time_max_delay = cpu_to_le32(delay);
2965
2966         IWL_DEBUG_TE(mvm,
2967                      "ROC: Requesting to remain on channel %u for %ums (requested = %ums, max_delay = %ums, dtim_interval = %ums)\n",
2968                      channel->hw_value, req_dur, duration, delay,
2969                      dtim_interval);
2970         /* Set the node address */
2971         memcpy(aux_roc_req.node_addr, vif->addr, ETH_ALEN);
2972
2973         lockdep_assert_held(&mvm->mutex);
2974
2975         spin_lock_bh(&mvm->time_event_lock);
2976
2977         if (WARN_ON(te_data->id == HOT_SPOT_CMD)) {
2978                 spin_unlock_bh(&mvm->time_event_lock);
2979                 return -EIO;
2980         }
2981
2982         te_data->vif = vif;
2983         te_data->duration = duration;
2984         te_data->id = HOT_SPOT_CMD;
2985
2986         spin_unlock_bh(&mvm->time_event_lock);
2987
2988         /*
2989          * Use a notification wait, which really just processes the
2990          * command response and doesn't wait for anything, in order
2991          * to be able to process the response and get the UID inside
2992          * the RX path. Using CMD_WANT_SKB doesn't work because it
2993          * stores the buffer and then wakes up this thread, by which
2994          * time another notification (that the time event started)
2995          * might already be processed unsuccessfully.
2996          */
2997         iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
2998                                    time_event_response,
2999                                    ARRAY_SIZE(time_event_response),
3000                                    iwl_mvm_rx_aux_roc, te_data);
3001
3002         res = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0, sizeof(aux_roc_req),
3003                                    &aux_roc_req);
3004
3005         if (res) {
3006                 IWL_ERR(mvm, "Couldn't send HOT_SPOT_CMD: %d\n", res);
3007                 iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
3008                 goto out_clear_te;
3009         }
3010
3011         /* No need to wait for anything, so just pass 1 (0 isn't valid) */
3012         res = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
3013         /* should never fail */
3014         WARN_ON_ONCE(res);
3015
3016         if (res) {
3017  out_clear_te:
3018                 spin_lock_bh(&mvm->time_event_lock);
3019                 iwl_mvm_te_clear_data(mvm, te_data);
3020                 spin_unlock_bh(&mvm->time_event_lock);
3021         }
3022
3023         return res;
3024 }
3025
3026 static int iwl_mvm_roc(struct ieee80211_hw *hw,
3027                        struct ieee80211_vif *vif,
3028                        struct ieee80211_channel *channel,
3029                        int duration,
3030                        enum ieee80211_roc_type type)
3031 {
3032         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3033         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3034         struct cfg80211_chan_def chandef;
3035         struct iwl_mvm_phy_ctxt *phy_ctxt;
3036         int ret, i;
3037
3038         IWL_DEBUG_MAC80211(mvm, "enter (%d, %d, %d)\n", channel->hw_value,
3039                            duration, type);
3040
3041         flush_work(&mvm->roc_done_wk);
3042
3043         mutex_lock(&mvm->mutex);
3044
3045         switch (vif->type) {
3046         case NL80211_IFTYPE_STATION:
3047                 if (fw_has_capa(&mvm->fw->ucode_capa,
3048                                 IWL_UCODE_TLV_CAPA_HOTSPOT_SUPPORT)) {
3049                         /* Use aux roc framework (HS20) */
3050                         ret = iwl_mvm_send_aux_roc_cmd(mvm, channel,
3051                                                        vif, duration);
3052                         goto out_unlock;
3053                 }
3054                 IWL_ERR(mvm, "hotspot not supported\n");
3055                 ret = -EINVAL;
3056                 goto out_unlock;
3057         case NL80211_IFTYPE_P2P_DEVICE:
3058                 /* handle below */
3059                 break;
3060         default:
3061                 IWL_ERR(mvm, "vif isn't P2P_DEVICE: %d\n", vif->type);
3062                 ret = -EINVAL;
3063                 goto out_unlock;
3064         }
3065
3066         for (i = 0; i < NUM_PHY_CTX; i++) {
3067                 phy_ctxt = &mvm->phy_ctxts[i];
3068                 if (phy_ctxt->ref == 0 || mvmvif->phy_ctxt == phy_ctxt)
3069                         continue;
3070
3071                 if (phy_ctxt->ref && channel == phy_ctxt->channel) {
3072                         /*
3073                          * Unbind the P2P_DEVICE from the current PHY context,
3074                          * and if the PHY context is not used remove it.
3075                          */
3076                         ret = iwl_mvm_binding_remove_vif(mvm, vif);
3077                         if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
3078                                 goto out_unlock;
3079
3080                         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
3081
3082                         /* Bind the P2P_DEVICE to the current PHY Context */
3083                         mvmvif->phy_ctxt = phy_ctxt;
3084
3085                         ret = iwl_mvm_binding_add_vif(mvm, vif);
3086                         if (WARN(ret, "Failed binding P2P_DEVICE\n"))
3087                                 goto out_unlock;
3088
3089                         iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
3090                         goto schedule_time_event;
3091                 }
3092         }
3093
3094         /* Need to update the PHY context only if the ROC channel changed */
3095         if (channel == mvmvif->phy_ctxt->channel)
3096                 goto schedule_time_event;
3097
3098         cfg80211_chandef_create(&chandef, channel, NL80211_CHAN_NO_HT);
3099
3100         /*
3101          * Change the PHY context configuration as it is currently referenced
3102          * only by the P2P Device MAC
3103          */
3104         if (mvmvif->phy_ctxt->ref == 1) {
3105                 ret = iwl_mvm_phy_ctxt_changed(mvm, mvmvif->phy_ctxt,
3106                                                &chandef, 1, 1);
3107                 if (ret)
3108                         goto out_unlock;
3109         } else {
3110                 /*
3111                  * The PHY context is shared with other MACs. Need to remove the
3112                  * P2P Device from the binding, allocate an new PHY context and
3113                  * create a new binding
3114                  */
3115                 phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
3116                 if (!phy_ctxt) {
3117                         ret = -ENOSPC;
3118                         goto out_unlock;
3119                 }
3120
3121                 ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &chandef,
3122                                                1, 1);
3123                 if (ret) {
3124                         IWL_ERR(mvm, "Failed to change PHY context\n");
3125                         goto out_unlock;
3126                 }
3127
3128                 /* Unbind the P2P_DEVICE from the current PHY context */
3129                 ret = iwl_mvm_binding_remove_vif(mvm, vif);
3130                 if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
3131                         goto out_unlock;
3132
3133                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
3134
3135                 /* Bind the P2P_DEVICE to the new allocated PHY context */
3136                 mvmvif->phy_ctxt = phy_ctxt;
3137
3138                 ret = iwl_mvm_binding_add_vif(mvm, vif);
3139                 if (WARN(ret, "Failed binding P2P_DEVICE\n"))
3140                         goto out_unlock;
3141
3142                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
3143         }
3144
3145 schedule_time_event:
3146         /* Schedule the time events */
3147         ret = iwl_mvm_start_p2p_roc(mvm, vif, duration, type);
3148
3149 out_unlock:
3150         mutex_unlock(&mvm->mutex);
3151         IWL_DEBUG_MAC80211(mvm, "leave\n");
3152         return ret;
3153 }
3154
3155 static int iwl_mvm_cancel_roc(struct ieee80211_hw *hw)
3156 {
3157         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3158
3159         IWL_DEBUG_MAC80211(mvm, "enter\n");
3160
3161         mutex_lock(&mvm->mutex);
3162         iwl_mvm_stop_roc(mvm);
3163         mutex_unlock(&mvm->mutex);
3164
3165         IWL_DEBUG_MAC80211(mvm, "leave\n");
3166         return 0;
3167 }
3168
3169 static int __iwl_mvm_add_chanctx(struct iwl_mvm *mvm,
3170                                  struct ieee80211_chanctx_conf *ctx)
3171 {
3172         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3173         struct iwl_mvm_phy_ctxt *phy_ctxt;
3174         int ret;
3175
3176         lockdep_assert_held(&mvm->mutex);
3177
3178         IWL_DEBUG_MAC80211(mvm, "Add channel context\n");
3179
3180         phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
3181         if (!phy_ctxt) {
3182                 ret = -ENOSPC;
3183                 goto out;
3184         }
3185
3186         ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
3187                                        ctx->rx_chains_static,
3188                                        ctx->rx_chains_dynamic);
3189         if (ret) {
3190                 IWL_ERR(mvm, "Failed to add PHY context\n");
3191                 goto out;
3192         }
3193
3194         iwl_mvm_phy_ctxt_ref(mvm, phy_ctxt);
3195         *phy_ctxt_id = phy_ctxt->id;
3196 out:
3197         return ret;
3198 }
3199
3200 static int iwl_mvm_add_chanctx(struct ieee80211_hw *hw,
3201                                struct ieee80211_chanctx_conf *ctx)
3202 {
3203         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3204         int ret;
3205
3206         mutex_lock(&mvm->mutex);
3207         ret = __iwl_mvm_add_chanctx(mvm, ctx);
3208         mutex_unlock(&mvm->mutex);
3209
3210         return ret;
3211 }
3212
3213 static void __iwl_mvm_remove_chanctx(struct iwl_mvm *mvm,
3214                                      struct ieee80211_chanctx_conf *ctx)
3215 {
3216         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3217         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3218
3219         lockdep_assert_held(&mvm->mutex);
3220
3221         iwl_mvm_phy_ctxt_unref(mvm, phy_ctxt);
3222 }
3223
3224 static void iwl_mvm_remove_chanctx(struct ieee80211_hw *hw,
3225                                    struct ieee80211_chanctx_conf *ctx)
3226 {
3227         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3228
3229         mutex_lock(&mvm->mutex);
3230         __iwl_mvm_remove_chanctx(mvm, ctx);
3231         mutex_unlock(&mvm->mutex);
3232 }
3233
3234 static void iwl_mvm_change_chanctx(struct ieee80211_hw *hw,
3235                                    struct ieee80211_chanctx_conf *ctx,
3236                                    u32 changed)
3237 {
3238         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3239         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3240         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3241
3242         if (WARN_ONCE((phy_ctxt->ref > 1) &&
3243                       (changed & ~(IEEE80211_CHANCTX_CHANGE_WIDTH |
3244                                    IEEE80211_CHANCTX_CHANGE_RX_CHAINS |
3245                                    IEEE80211_CHANCTX_CHANGE_RADAR |
3246                                    IEEE80211_CHANCTX_CHANGE_MIN_WIDTH)),
3247                       "Cannot change PHY. Ref=%d, changed=0x%X\n",
3248                       phy_ctxt->ref, changed))
3249                 return;
3250
3251         mutex_lock(&mvm->mutex);
3252         iwl_mvm_bt_coex_vif_change(mvm);
3253         iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
3254                                  ctx->rx_chains_static,
3255                                  ctx->rx_chains_dynamic);
3256         mutex_unlock(&mvm->mutex);
3257 }
3258
3259 static int __iwl_mvm_assign_vif_chanctx(struct iwl_mvm *mvm,
3260                                         struct ieee80211_vif *vif,
3261                                         struct ieee80211_chanctx_conf *ctx,
3262                                         bool switching_chanctx)
3263 {
3264         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3265         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3266         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3267         int ret;
3268
3269         lockdep_assert_held(&mvm->mutex);
3270
3271         mvmvif->phy_ctxt = phy_ctxt;
3272
3273         switch (vif->type) {
3274         case NL80211_IFTYPE_AP:
3275                 /* only needed if we're switching chanctx (i.e. during CSA) */
3276                 if (switching_chanctx) {
3277                         mvmvif->ap_ibss_active = true;
3278                         break;
3279                 }
3280         case NL80211_IFTYPE_ADHOC:
3281                 /*
3282                  * The AP binding flow is handled as part of the start_ap flow
3283                  * (in bss_info_changed), similarly for IBSS.
3284                  */
3285                 ret = 0;
3286                 goto out;
3287         case NL80211_IFTYPE_STATION:
3288                 break;
3289         case NL80211_IFTYPE_MONITOR:
3290                 /* always disable PS when a monitor interface is active */
3291                 mvmvif->ps_disabled = true;
3292                 break;
3293         default:
3294                 ret = -EINVAL;
3295                 goto out;
3296         }
3297
3298         ret = iwl_mvm_binding_add_vif(mvm, vif);
3299         if (ret)
3300                 goto out;
3301
3302         /*
3303          * Power state must be updated before quotas,
3304          * otherwise fw will complain.
3305          */
3306         iwl_mvm_power_update_mac(mvm);
3307
3308         /* Setting the quota at this stage is only required for monitor
3309          * interfaces. For the other types, the bss_info changed flow
3310          * will handle quota settings.
3311          */
3312         if (vif->type == NL80211_IFTYPE_MONITOR) {
3313                 mvmvif->monitor_active = true;
3314                 ret = iwl_mvm_update_quotas(mvm, false, NULL);
3315                 if (ret)
3316                         goto out_remove_binding;
3317
3318                 ret = iwl_mvm_add_snif_sta(mvm, vif);
3319                 if (ret)
3320                         goto out_remove_binding;
3321
3322         }
3323
3324         /* Handle binding during CSA */
3325         if (vif->type == NL80211_IFTYPE_AP) {
3326                 iwl_mvm_update_quotas(mvm, false, NULL);
3327                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
3328         }
3329
3330         if (switching_chanctx && vif->type == NL80211_IFTYPE_STATION) {
3331                 u32 duration = 2 * vif->bss_conf.beacon_int;
3332
3333                 /* iwl_mvm_protect_session() reads directly from the
3334                  * device (the system time), so make sure it is
3335                  * available.
3336                  */
3337                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PROTECT_CSA);
3338                 if (ret)
3339                         goto out_remove_binding;
3340
3341                 /* Protect the session to make sure we hear the first
3342                  * beacon on the new channel.
3343                  */
3344                 iwl_mvm_protect_session(mvm, vif, duration, duration,
3345                                         vif->bss_conf.beacon_int / 2,
3346                                         true);
3347
3348                 iwl_mvm_unref(mvm, IWL_MVM_REF_PROTECT_CSA);
3349
3350                 iwl_mvm_update_quotas(mvm, false, NULL);
3351         }
3352
3353         goto out;
3354
3355 out_remove_binding:
3356         iwl_mvm_binding_remove_vif(mvm, vif);
3357         iwl_mvm_power_update_mac(mvm);
3358 out:
3359         if (ret)
3360                 mvmvif->phy_ctxt = NULL;
3361         return ret;
3362 }
3363 static int iwl_mvm_assign_vif_chanctx(struct ieee80211_hw *hw,
3364                                       struct ieee80211_vif *vif,
3365                                       struct ieee80211_chanctx_conf *ctx)
3366 {
3367         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3368         int ret;
3369
3370         mutex_lock(&mvm->mutex);
3371         ret = __iwl_mvm_assign_vif_chanctx(mvm, vif, ctx, false);
3372         mutex_unlock(&mvm->mutex);
3373
3374         return ret;
3375 }
3376
3377 static void __iwl_mvm_unassign_vif_chanctx(struct iwl_mvm *mvm,
3378                                            struct ieee80211_vif *vif,
3379                                            struct ieee80211_chanctx_conf *ctx,
3380                                            bool switching_chanctx)
3381 {
3382         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3383         struct ieee80211_vif *disabled_vif = NULL;
3384
3385         lockdep_assert_held(&mvm->mutex);
3386
3387         iwl_mvm_remove_time_event(mvm, mvmvif, &mvmvif->time_event_data);
3388
3389         switch (vif->type) {
3390         case NL80211_IFTYPE_ADHOC:
3391                 goto out;
3392         case NL80211_IFTYPE_MONITOR:
3393                 mvmvif->monitor_active = false;
3394                 mvmvif->ps_disabled = false;
3395                 iwl_mvm_rm_snif_sta(mvm, vif);
3396                 break;
3397         case NL80211_IFTYPE_AP:
3398                 /* This part is triggered only during CSA */
3399                 if (!switching_chanctx || !mvmvif->ap_ibss_active)
3400                         goto out;
3401
3402                 mvmvif->csa_countdown = false;
3403
3404                 /* Set CS bit on all the stations */
3405                 iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, true);
3406
3407                 /* Save blocked iface, the timeout is set on the next beacon */
3408                 rcu_assign_pointer(mvm->csa_tx_blocked_vif, vif);
3409
3410                 mvmvif->ap_ibss_active = false;
3411                 break;
3412         case NL80211_IFTYPE_STATION:
3413                 if (!switching_chanctx)
3414                         break;
3415
3416                 disabled_vif = vif;
3417
3418                 iwl_mvm_mac_ctxt_changed(mvm, vif, true, NULL);
3419                 break;
3420         default:
3421                 break;
3422         }
3423
3424         iwl_mvm_update_quotas(mvm, false, disabled_vif);
3425         iwl_mvm_binding_remove_vif(mvm, vif);
3426
3427 out:
3428         mvmvif->phy_ctxt = NULL;
3429         iwl_mvm_power_update_mac(mvm);
3430 }
3431
3432 static void iwl_mvm_unassign_vif_chanctx(struct ieee80211_hw *hw,
3433                                          struct ieee80211_vif *vif,
3434                                          struct ieee80211_chanctx_conf *ctx)
3435 {
3436         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3437
3438         mutex_lock(&mvm->mutex);
3439         __iwl_mvm_unassign_vif_chanctx(mvm, vif, ctx, false);
3440         mutex_unlock(&mvm->mutex);
3441 }
3442
3443 static int
3444 iwl_mvm_switch_vif_chanctx_swap(struct iwl_mvm *mvm,
3445                                 struct ieee80211_vif_chanctx_switch *vifs)
3446 {
3447         int ret;
3448
3449         mutex_lock(&mvm->mutex);
3450         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
3451         __iwl_mvm_remove_chanctx(mvm, vifs[0].old_ctx);
3452
3453         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].new_ctx);
3454         if (ret) {
3455                 IWL_ERR(mvm, "failed to add new_ctx during channel switch\n");
3456                 goto out_reassign;
3457         }
3458
3459         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
3460                                            true);
3461         if (ret) {
3462                 IWL_ERR(mvm,
3463                         "failed to assign new_ctx during channel switch\n");
3464                 goto out_remove;
3465         }
3466
3467         /* we don't support TDLS during DCM - can be caused by channel switch */
3468         if (iwl_mvm_phy_ctx_count(mvm) > 1)
3469                 iwl_mvm_teardown_tdls_peers(mvm);
3470
3471         goto out;
3472
3473 out_remove:
3474         __iwl_mvm_remove_chanctx(mvm, vifs[0].new_ctx);
3475
3476 out_reassign:
3477         if (__iwl_mvm_add_chanctx(mvm, vifs[0].old_ctx)) {
3478                 IWL_ERR(mvm, "failed to add old_ctx back after failure.\n");
3479                 goto out_restart;
3480         }
3481
3482         if (__iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
3483                                          true)) {
3484                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
3485                 goto out_restart;
3486         }
3487
3488         goto out;
3489
3490 out_restart:
3491         /* things keep failing, better restart the hw */
3492         iwl_mvm_nic_restart(mvm, false);
3493
3494 out:
3495         mutex_unlock(&mvm->mutex);
3496
3497         return ret;
3498 }
3499
3500 static int
3501 iwl_mvm_switch_vif_chanctx_reassign(struct iwl_mvm *mvm,
3502                                     struct ieee80211_vif_chanctx_switch *vifs)
3503 {
3504         int ret;
3505
3506         mutex_lock(&mvm->mutex);
3507         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
3508
3509         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
3510                                            true);
3511         if (ret) {
3512                 IWL_ERR(mvm,
3513                         "failed to assign new_ctx during channel switch\n");
3514                 goto out_reassign;
3515         }
3516
3517         goto out;
3518
3519 out_reassign:
3520         if (__iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
3521                                          true)) {
3522                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
3523                 goto out_restart;
3524         }
3525
3526         goto out;
3527
3528 out_restart:
3529         /* things keep failing, better restart the hw */
3530         iwl_mvm_nic_restart(mvm, false);
3531
3532 out:
3533         mutex_unlock(&mvm->mutex);
3534
3535         return ret;
3536 }
3537
3538 static int iwl_mvm_switch_vif_chanctx(struct ieee80211_hw *hw,
3539                                       struct ieee80211_vif_chanctx_switch *vifs,
3540                                       int n_vifs,
3541                                       enum ieee80211_chanctx_switch_mode mode)
3542 {
3543         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3544         int ret;
3545
3546         /* we only support a single-vif right now */
3547         if (n_vifs > 1)
3548                 return -EOPNOTSUPP;
3549
3550         switch (mode) {
3551         case CHANCTX_SWMODE_SWAP_CONTEXTS:
3552                 ret = iwl_mvm_switch_vif_chanctx_swap(mvm, vifs);
3553                 break;
3554         case CHANCTX_SWMODE_REASSIGN_VIF:
3555                 ret = iwl_mvm_switch_vif_chanctx_reassign(mvm, vifs);
3556                 break;
3557         default:
3558                 ret = -EOPNOTSUPP;
3559                 break;
3560         }
3561
3562         return ret;
3563 }
3564
3565 static int iwl_mvm_set_tim(struct ieee80211_hw *hw,
3566                            struct ieee80211_sta *sta,
3567                            bool set)
3568 {
3569         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3570         struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
3571
3572         if (!mvm_sta || !mvm_sta->vif) {
3573                 IWL_ERR(mvm, "Station is not associated to a vif\n");
3574                 return -EINVAL;
3575         }
3576
3577         return iwl_mvm_mac_ctxt_beacon_changed(mvm, mvm_sta->vif);
3578 }
3579
3580 #ifdef CONFIG_NL80211_TESTMODE
3581 static const struct nla_policy iwl_mvm_tm_policy[IWL_MVM_TM_ATTR_MAX + 1] = {
3582         [IWL_MVM_TM_ATTR_CMD] = { .type = NLA_U32 },
3583         [IWL_MVM_TM_ATTR_NOA_DURATION] = { .type = NLA_U32 },
3584         [IWL_MVM_TM_ATTR_BEACON_FILTER_STATE] = { .type = NLA_U32 },
3585 };
3586
3587 static int __iwl_mvm_mac_testmode_cmd(struct iwl_mvm *mvm,
3588                                       struct ieee80211_vif *vif,
3589                                       void *data, int len)
3590 {
3591         struct nlattr *tb[IWL_MVM_TM_ATTR_MAX + 1];
3592         int err;
3593         u32 noa_duration;
3594
3595         err = nla_parse(tb, IWL_MVM_TM_ATTR_MAX, data, len, iwl_mvm_tm_policy);
3596         if (err)
3597                 return err;
3598
3599         if (!tb[IWL_MVM_TM_ATTR_CMD])
3600                 return -EINVAL;
3601
3602         switch (nla_get_u32(tb[IWL_MVM_TM_ATTR_CMD])) {
3603         case IWL_MVM_TM_CMD_SET_NOA:
3604                 if (!vif || vif->type != NL80211_IFTYPE_AP || !vif->p2p ||
3605                     !vif->bss_conf.enable_beacon ||
3606                     !tb[IWL_MVM_TM_ATTR_NOA_DURATION])
3607                         return -EINVAL;
3608
3609                 noa_duration = nla_get_u32(tb[IWL_MVM_TM_ATTR_NOA_DURATION]);
3610                 if (noa_duration >= vif->bss_conf.beacon_int)
3611                         return -EINVAL;
3612
3613                 mvm->noa_duration = noa_duration;
3614                 mvm->noa_vif = vif;
3615
3616                 return iwl_mvm_update_quotas(mvm, false, NULL);
3617         case IWL_MVM_TM_CMD_SET_BEACON_FILTER:
3618                 /* must be associated client vif - ignore authorized */
3619                 if (!vif || vif->type != NL80211_IFTYPE_STATION ||
3620                     !vif->bss_conf.assoc || !vif->bss_conf.dtim_period ||
3621                     !tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE])
3622                         return -EINVAL;
3623
3624                 if (nla_get_u32(tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE]))
3625                         return iwl_mvm_enable_beacon_filter(mvm, vif, 0);
3626                 return iwl_mvm_disable_beacon_filter(mvm, vif, 0);
3627         }
3628
3629         return -EOPNOTSUPP;
3630 }
3631
3632 static int iwl_mvm_mac_testmode_cmd(struct ieee80211_hw *hw,
3633                                     struct ieee80211_vif *vif,
3634                                     void *data, int len)
3635 {
3636         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3637         int err;
3638
3639         mutex_lock(&mvm->mutex);
3640         err = __iwl_mvm_mac_testmode_cmd(mvm, vif, data, len);
3641         mutex_unlock(&mvm->mutex);
3642
3643         return err;
3644 }
3645 #endif
3646
3647 static void iwl_mvm_channel_switch(struct ieee80211_hw *hw,
3648                                    struct ieee80211_vif *vif,
3649                                    struct ieee80211_channel_switch *chsw)
3650 {
3651         /* By implementing this operation, we prevent mac80211 from
3652          * starting its own channel switch timer, so that we can call
3653          * ieee80211_chswitch_done() ourselves at the right time
3654          * (which is when the absence time event starts).
3655          */
3656
3657         IWL_DEBUG_MAC80211(IWL_MAC80211_GET_MVM(hw),
3658                            "dummy channel switch op\n");
3659 }
3660
3661 static int iwl_mvm_pre_channel_switch(struct ieee80211_hw *hw,
3662                                       struct ieee80211_vif *vif,
3663                                       struct ieee80211_channel_switch *chsw)
3664 {
3665         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3666         struct ieee80211_vif *csa_vif;
3667         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3668         u32 apply_time;
3669         int ret;
3670
3671         mutex_lock(&mvm->mutex);
3672
3673         mvmvif->csa_failed = false;
3674
3675         IWL_DEBUG_MAC80211(mvm, "pre CSA to freq %d\n",
3676                            chsw->chandef.center_freq1);
3677
3678         iwl_fw_dbg_trigger_simple_stop(mvm, vif, FW_DBG_TRIGGER_CHANNEL_SWITCH);
3679
3680         switch (vif->type) {
3681         case NL80211_IFTYPE_AP:
3682                 csa_vif =
3683                         rcu_dereference_protected(mvm->csa_vif,
3684                                                   lockdep_is_held(&mvm->mutex));
3685                 if (WARN_ONCE(csa_vif && csa_vif->csa_active,
3686                               "Another CSA is already in progress")) {
3687                         ret = -EBUSY;
3688                         goto out_unlock;
3689                 }
3690
3691                 /* we still didn't unblock tx. prevent new CS meanwhile */
3692                 if (rcu_dereference_protected(mvm->csa_tx_blocked_vif,
3693                                               lockdep_is_held(&mvm->mutex))) {
3694                         ret = -EBUSY;
3695                         goto out_unlock;
3696                 }
3697
3698                 rcu_assign_pointer(mvm->csa_vif, vif);
3699
3700                 if (WARN_ONCE(mvmvif->csa_countdown,
3701                               "Previous CSA countdown didn't complete")) {
3702                         ret = -EBUSY;
3703                         goto out_unlock;
3704                 }
3705
3706                 mvmvif->csa_target_freq = chsw->chandef.chan->center_freq;
3707
3708                 break;
3709         case NL80211_IFTYPE_STATION:
3710                 if (mvmvif->lqm_active)
3711                         iwl_mvm_send_lqm_cmd(vif,
3712                                              LQM_CMD_OPERATION_STOP_MEASUREMENT,
3713                                              0, 0);
3714
3715                 /* Schedule the time event to a bit before beacon 1,
3716                  * to make sure we're in the new channel when the
3717                  * GO/AP arrives.
3718                  */
3719                 apply_time = chsw->device_timestamp +
3720                         ((vif->bss_conf.beacon_int * (chsw->count - 1) -
3721                           IWL_MVM_CHANNEL_SWITCH_TIME_CLIENT) * 1024);
3722
3723                 if (chsw->block_tx)
3724                         iwl_mvm_csa_client_absent(mvm, vif);
3725
3726                 iwl_mvm_schedule_csa_period(mvm, vif, vif->bss_conf.beacon_int,
3727                                             apply_time);
3728                 if (mvmvif->bf_data.bf_enabled) {
3729                         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
3730                         if (ret)
3731                                 goto out_unlock;
3732                 }
3733
3734                 break;
3735         default:
3736                 break;
3737         }
3738
3739         mvmvif->ps_disabled = true;
3740
3741         ret = iwl_mvm_power_update_ps(mvm);
3742         if (ret)
3743                 goto out_unlock;
3744
3745         /* we won't be on this channel any longer */
3746         iwl_mvm_teardown_tdls_peers(mvm);
3747
3748 out_unlock:
3749         mutex_unlock(&mvm->mutex);
3750
3751         return ret;
3752 }
3753
3754 static int iwl_mvm_post_channel_switch(struct ieee80211_hw *hw,
3755                                        struct ieee80211_vif *vif)
3756 {
3757         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3758         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3759         int ret;
3760
3761         mutex_lock(&mvm->mutex);
3762
3763         if (mvmvif->csa_failed) {
3764                 mvmvif->csa_failed = false;
3765                 ret = -EIO;
3766                 goto out_unlock;
3767         }
3768
3769         if (vif->type == NL80211_IFTYPE_STATION) {
3770                 struct iwl_mvm_sta *mvmsta;
3771
3772                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
3773                                                           mvmvif->ap_sta_id);
3774
3775                 if (WARN_ON(!mvmsta)) {
3776                         ret = -EIO;
3777                         goto out_unlock;
3778                 }
3779
3780                 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, false);
3781
3782                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
3783
3784                 ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
3785                 if (ret)
3786                         goto out_unlock;
3787
3788                 iwl_mvm_stop_session_protection(mvm, vif);
3789         }
3790
3791         mvmvif->ps_disabled = false;
3792
3793         ret = iwl_mvm_power_update_ps(mvm);
3794
3795 out_unlock:
3796         mutex_unlock(&mvm->mutex);
3797
3798         return ret;
3799 }
3800
3801 static void iwl_mvm_mac_flush(struct ieee80211_hw *hw,
3802                               struct ieee80211_vif *vif, u32 queues, bool drop)
3803 {
3804         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3805         struct iwl_mvm_vif *mvmvif;
3806         struct iwl_mvm_sta *mvmsta;
3807         struct ieee80211_sta *sta;
3808         int i;
3809         u32 msk = 0;
3810
3811         if (!vif || vif->type != NL80211_IFTYPE_STATION)
3812                 return;
3813
3814         /* Make sure we're done with the deferred traffic before flushing */
3815         if (iwl_mvm_is_dqa_supported(mvm))
3816                 flush_work(&mvm->add_stream_wk);
3817
3818         mutex_lock(&mvm->mutex);
3819         mvmvif = iwl_mvm_vif_from_mac80211(vif);
3820
3821         /* flush the AP-station and all TDLS peers */
3822         for (i = 0; i < IWL_MVM_STATION_COUNT; i++) {
3823                 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[i],
3824                                                 lockdep_is_held(&mvm->mutex));
3825                 if (IS_ERR_OR_NULL(sta))
3826                         continue;
3827
3828                 mvmsta = iwl_mvm_sta_from_mac80211(sta);
3829                 if (mvmsta->vif != vif)
3830                         continue;
3831
3832                 /* make sure only TDLS peers or the AP are flushed */
3833                 WARN_ON(i != mvmvif->ap_sta_id && !sta->tdls);
3834
3835                 msk |= mvmsta->tfd_queue_msk;
3836         }
3837
3838         if (drop) {
3839                 if (iwl_mvm_flush_tx_path(mvm, msk, 0))
3840                         IWL_ERR(mvm, "flush request fail\n");
3841                 mutex_unlock(&mvm->mutex);
3842         } else {
3843                 mutex_unlock(&mvm->mutex);
3844
3845                 /* this can take a while, and we may need/want other operations
3846                  * to succeed while doing this, so do it without the mutex held
3847                  */
3848                 iwl_trans_wait_tx_queue_empty(mvm->trans, msk);
3849         }
3850 }
3851
3852 static int iwl_mvm_mac_get_survey(struct ieee80211_hw *hw, int idx,
3853                                   struct survey_info *survey)
3854 {
3855         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3856         int ret;
3857
3858         memset(survey, 0, sizeof(*survey));
3859
3860         /* only support global statistics right now */
3861         if (idx != 0)
3862                 return -ENOENT;
3863
3864         if (fw_has_capa(&mvm->fw->ucode_capa,
3865                         IWL_UCODE_TLV_CAPA_RADIO_BEACON_STATS))
3866                 return -ENOENT;
3867
3868         mutex_lock(&mvm->mutex);
3869
3870         if (mvm->ucode_loaded) {
3871                 ret = iwl_mvm_request_statistics(mvm, false);
3872                 if (ret)
3873                         goto out;
3874         }
3875
3876         survey->filled = SURVEY_INFO_TIME |
3877                          SURVEY_INFO_TIME_RX |
3878                          SURVEY_INFO_TIME_TX |
3879                          SURVEY_INFO_TIME_SCAN;
3880         survey->time = mvm->accu_radio_stats.on_time_rf +
3881                        mvm->radio_stats.on_time_rf;
3882         do_div(survey->time, USEC_PER_MSEC);
3883
3884         survey->time_rx = mvm->accu_radio_stats.rx_time +
3885                           mvm->radio_stats.rx_time;
3886         do_div(survey->time_rx, USEC_PER_MSEC);
3887
3888         survey->time_tx = mvm->accu_radio_stats.tx_time +
3889                           mvm->radio_stats.tx_time;
3890         do_div(survey->time_tx, USEC_PER_MSEC);
3891
3892         survey->time_scan = mvm->accu_radio_stats.on_time_scan +
3893                             mvm->radio_stats.on_time_scan;
3894         do_div(survey->time_scan, USEC_PER_MSEC);
3895
3896         ret = 0;
3897  out:
3898         mutex_unlock(&mvm->mutex);
3899         return ret;
3900 }
3901
3902 static void iwl_mvm_mac_sta_statistics(struct ieee80211_hw *hw,
3903                                        struct ieee80211_vif *vif,
3904                                        struct ieee80211_sta *sta,
3905                                        struct station_info *sinfo)
3906 {
3907         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3908         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3909         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3910
3911         if (fw_has_capa(&mvm->fw->ucode_capa,
3912                         IWL_UCODE_TLV_CAPA_RADIO_BEACON_STATS))
3913                 return;
3914
3915         /* if beacon filtering isn't on mac80211 does it anyway */
3916         if (!(vif->driver_flags & IEEE80211_VIF_BEACON_FILTER))
3917                 return;
3918
3919         if (!vif->bss_conf.assoc)
3920                 return;
3921
3922         mutex_lock(&mvm->mutex);
3923
3924         if (mvmvif->ap_sta_id != mvmsta->sta_id)
3925                 goto unlock;
3926
3927         if (iwl_mvm_request_statistics(mvm, false))
3928                 goto unlock;
3929
3930         sinfo->rx_beacon = mvmvif->beacon_stats.num_beacons +
3931                            mvmvif->beacon_stats.accu_num_beacons;
3932         sinfo->filled |= BIT(NL80211_STA_INFO_BEACON_RX);
3933         if (mvmvif->beacon_stats.avg_signal) {
3934                 /* firmware only reports a value after RXing a few beacons */
3935                 sinfo->rx_beacon_signal_avg = mvmvif->beacon_stats.avg_signal;
3936                 sinfo->filled |= BIT(NL80211_STA_INFO_BEACON_SIGNAL_AVG);
3937         }
3938  unlock:
3939         mutex_unlock(&mvm->mutex);
3940 }
3941
3942 static void iwl_mvm_event_mlme_callback(struct iwl_mvm *mvm,
3943                                         struct ieee80211_vif *vif,
3944                                         const struct ieee80211_event *event)
3945 {
3946 #define CHECK_MLME_TRIGGER(_mvm, _trig, _buf, _cnt, _fmt...)    \
3947         do {                                                    \
3948                 if ((_cnt) && --(_cnt))                         \
3949                         break;                                  \
3950                 iwl_mvm_fw_dbg_collect_trig(_mvm, _trig, _fmt);\
3951         } while (0)
3952
3953         struct iwl_fw_dbg_trigger_tlv *trig;
3954         struct iwl_fw_dbg_trigger_mlme *trig_mlme;
3955
3956         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_MLME))
3957                 return;
3958
3959         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_MLME);
3960         trig_mlme = (void *)trig->data;
3961         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
3962                 return;
3963
3964         if (event->u.mlme.data == ASSOC_EVENT) {
3965                 if (event->u.mlme.status == MLME_DENIED)
3966                         CHECK_MLME_TRIGGER(mvm, trig, buf,
3967                                            trig_mlme->stop_assoc_denied,
3968                                            "DENIED ASSOC: reason %d",
3969                                             event->u.mlme.reason);
3970                 else if (event->u.mlme.status == MLME_TIMEOUT)
3971                         CHECK_MLME_TRIGGER(mvm, trig, buf,
3972                                            trig_mlme->stop_assoc_timeout,
3973                                            "ASSOC TIMEOUT");
3974         } else if (event->u.mlme.data == AUTH_EVENT) {
3975                 if (event->u.mlme.status == MLME_DENIED)
3976                         CHECK_MLME_TRIGGER(mvm, trig, buf,
3977                                            trig_mlme->stop_auth_denied,
3978                                            "DENIED AUTH: reason %d",
3979                                            event->u.mlme.reason);
3980                 else if (event->u.mlme.status == MLME_TIMEOUT)
3981                         CHECK_MLME_TRIGGER(mvm, trig, buf,
3982                                            trig_mlme->stop_auth_timeout,
3983                                            "AUTH TIMEOUT");
3984         } else if (event->u.mlme.data == DEAUTH_RX_EVENT) {
3985                 CHECK_MLME_TRIGGER(mvm, trig, buf,
3986                                    trig_mlme->stop_rx_deauth,
3987                                    "DEAUTH RX %d", event->u.mlme.reason);
3988         } else if (event->u.mlme.data == DEAUTH_TX_EVENT) {
3989                 CHECK_MLME_TRIGGER(mvm, trig, buf,
3990                                    trig_mlme->stop_tx_deauth,
3991                                    "DEAUTH TX %d", event->u.mlme.reason);
3992         }
3993 #undef CHECK_MLME_TRIGGER
3994 }
3995
3996 static void iwl_mvm_event_bar_rx_callback(struct iwl_mvm *mvm,
3997                                           struct ieee80211_vif *vif,
3998                                           const struct ieee80211_event *event)
3999 {
4000         struct iwl_fw_dbg_trigger_tlv *trig;
4001         struct iwl_fw_dbg_trigger_ba *ba_trig;
4002
4003         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
4004                 return;
4005
4006         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
4007         ba_trig = (void *)trig->data;
4008         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
4009                 return;
4010
4011         if (!(le16_to_cpu(ba_trig->rx_bar) & BIT(event->u.ba.tid)))
4012                 return;
4013
4014         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
4015                                     "BAR received from %pM, tid %d, ssn %d",
4016                                     event->u.ba.sta->addr, event->u.ba.tid,
4017                                     event->u.ba.ssn);
4018 }
4019
4020 static void
4021 iwl_mvm_event_frame_timeout_callback(struct iwl_mvm *mvm,
4022                                      struct ieee80211_vif *vif,
4023                                      const struct ieee80211_event *event)
4024 {
4025         struct iwl_fw_dbg_trigger_tlv *trig;
4026         struct iwl_fw_dbg_trigger_ba *ba_trig;
4027
4028         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
4029                 return;
4030
4031         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
4032         ba_trig = (void *)trig->data;
4033         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
4034                 return;
4035
4036         if (!(le16_to_cpu(ba_trig->frame_timeout) & BIT(event->u.ba.tid)))
4037                 return;
4038
4039         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
4040                                     "Frame from %pM timed out, tid %d",
4041                                     event->u.ba.sta->addr, event->u.ba.tid);
4042 }
4043
4044 static void iwl_mvm_mac_event_callback(struct ieee80211_hw *hw,
4045                                        struct ieee80211_vif *vif,
4046                                        const struct ieee80211_event *event)
4047 {
4048         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
4049
4050         switch (event->type) {
4051         case MLME_EVENT:
4052                 iwl_mvm_event_mlme_callback(mvm, vif, event);
4053                 break;
4054         case BAR_RX_EVENT:
4055                 iwl_mvm_event_bar_rx_callback(mvm, vif, event);
4056                 break;
4057         case BA_FRAME_TIMEOUT:
4058                 iwl_mvm_event_frame_timeout_callback(mvm, vif, event);
4059                 break;
4060         default:
4061                 break;
4062         }
4063 }
4064
4065 void iwl_mvm_sync_rx_queues_internal(struct iwl_mvm *mvm,
4066                                      struct iwl_mvm_internal_rxq_notif *notif,
4067                                      u32 size)
4068 {
4069         DECLARE_WAIT_QUEUE_HEAD_ONSTACK(notif_waitq);
4070         u32 qmask = BIT(mvm->trans->num_rx_queues) - 1;
4071         int ret;
4072
4073         lockdep_assert_held(&mvm->mutex);
4074
4075         if (!iwl_mvm_has_new_rx_api(mvm))
4076                 return;
4077
4078         notif->cookie = mvm->queue_sync_cookie;
4079
4080         if (notif->sync)
4081                 atomic_set(&mvm->queue_sync_counter,
4082                            mvm->trans->num_rx_queues);
4083
4084         ret = iwl_mvm_notify_rx_queue(mvm, qmask, (u8 *)notif, size);
4085         if (ret) {
4086                 IWL_ERR(mvm, "Failed to trigger RX queues sync (%d)\n", ret);
4087                 goto out;
4088         }
4089
4090         if (notif->sync)
4091                 ret = wait_event_timeout(notif_waitq,
4092                                          atomic_read(&mvm->queue_sync_counter) == 0,
4093                                          HZ);
4094         WARN_ON_ONCE(!ret);
4095
4096 out:
4097         atomic_set(&mvm->queue_sync_counter, 0);
4098         mvm->queue_sync_cookie++;
4099 }
4100
4101 static void iwl_mvm_sync_rx_queues(struct ieee80211_hw *hw)
4102 {
4103         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
4104         struct iwl_mvm_internal_rxq_notif data = {
4105                 .type = IWL_MVM_RXQ_EMPTY,
4106                 .sync = 1,
4107         };
4108
4109         mutex_lock(&mvm->mutex);
4110         iwl_mvm_sync_rx_queues_internal(mvm, &data, sizeof(data));
4111         mutex_unlock(&mvm->mutex);
4112 }
4113
4114 const struct ieee80211_ops iwl_mvm_hw_ops = {
4115         .tx = iwl_mvm_mac_tx,
4116         .ampdu_action = iwl_mvm_mac_ampdu_action,
4117         .start = iwl_mvm_mac_start,
4118         .reconfig_complete = iwl_mvm_mac_reconfig_complete,
4119         .stop = iwl_mvm_mac_stop,
4120         .add_interface = iwl_mvm_mac_add_interface,
4121         .remove_interface = iwl_mvm_mac_remove_interface,
4122         .config = iwl_mvm_mac_config,
4123         .prepare_multicast = iwl_mvm_prepare_multicast,
4124         .configure_filter = iwl_mvm_configure_filter,
4125         .config_iface_filter = iwl_mvm_config_iface_filter,
4126         .bss_info_changed = iwl_mvm_bss_info_changed,
4127         .hw_scan = iwl_mvm_mac_hw_scan,
4128         .cancel_hw_scan = iwl_mvm_mac_cancel_hw_scan,
4129         .sta_pre_rcu_remove = iwl_mvm_sta_pre_rcu_remove,
4130         .sta_state = iwl_mvm_mac_sta_state,
4131         .sta_notify = iwl_mvm_mac_sta_notify,
4132         .allow_buffered_frames = iwl_mvm_mac_allow_buffered_frames,
4133         .release_buffered_frames = iwl_mvm_mac_release_buffered_frames,
4134         .set_rts_threshold = iwl_mvm_mac_set_rts_threshold,
4135         .sta_rc_update = iwl_mvm_sta_rc_update,
4136         .conf_tx = iwl_mvm_mac_conf_tx,
4137         .mgd_prepare_tx = iwl_mvm_mac_mgd_prepare_tx,
4138         .mgd_protect_tdls_discover = iwl_mvm_mac_mgd_protect_tdls_discover,
4139         .flush = iwl_mvm_mac_flush,
4140         .sched_scan_start = iwl_mvm_mac_sched_scan_start,
4141         .sched_scan_stop = iwl_mvm_mac_sched_scan_stop,
4142         .set_key = iwl_mvm_mac_set_key,
4143         .update_tkip_key = iwl_mvm_mac_update_tkip_key,
4144         .remain_on_channel = iwl_mvm_roc,
4145         .cancel_remain_on_channel = iwl_mvm_cancel_roc,
4146         .add_chanctx = iwl_mvm_add_chanctx,
4147         .remove_chanctx = iwl_mvm_remove_chanctx,
4148         .change_chanctx = iwl_mvm_change_chanctx,
4149         .assign_vif_chanctx = iwl_mvm_assign_vif_chanctx,
4150         .unassign_vif_chanctx = iwl_mvm_unassign_vif_chanctx,
4151         .switch_vif_chanctx = iwl_mvm_switch_vif_chanctx,
4152
4153         .start_ap = iwl_mvm_start_ap_ibss,
4154         .stop_ap = iwl_mvm_stop_ap_ibss,
4155         .join_ibss = iwl_mvm_start_ap_ibss,
4156         .leave_ibss = iwl_mvm_stop_ap_ibss,
4157
4158         .set_tim = iwl_mvm_set_tim,
4159
4160         .channel_switch = iwl_mvm_channel_switch,
4161         .pre_channel_switch = iwl_mvm_pre_channel_switch,
4162         .post_channel_switch = iwl_mvm_post_channel_switch,
4163
4164         .tdls_channel_switch = iwl_mvm_tdls_channel_switch,
4165         .tdls_cancel_channel_switch = iwl_mvm_tdls_cancel_channel_switch,
4166         .tdls_recv_channel_switch = iwl_mvm_tdls_recv_channel_switch,
4167
4168         .event_callback = iwl_mvm_mac_event_callback,
4169
4170         .sync_rx_queues = iwl_mvm_sync_rx_queues,
4171
4172         CFG80211_TESTMODE_CMD(iwl_mvm_mac_testmode_cmd)
4173
4174 #ifdef CONFIG_PM_SLEEP
4175         /* look at d3.c */
4176         .suspend = iwl_mvm_suspend,
4177         .resume = iwl_mvm_resume,
4178         .set_wakeup = iwl_mvm_set_wakeup,
4179         .set_rekey_data = iwl_mvm_set_rekey_data,
4180 #if IS_ENABLED(CONFIG_IPV6)
4181         .ipv6_addr_change = iwl_mvm_ipv6_addr_change,
4182 #endif
4183         .set_default_unicast_key = iwl_mvm_set_default_unicast_key,
4184 #endif
4185         .get_survey = iwl_mvm_mac_get_survey,
4186         .sta_statistics = iwl_mvm_mac_sta_statistics,
4187 };