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