xref: /linux/drivers/net/wireless/intel/iwlwifi/mvm/mac-ctxt.c (revision f6e0a4984c2e7244689ea87b62b433bed9d07e94)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /*
3  * Copyright (C) 2012-2014, 2018-2024 Intel Corporation
4  * Copyright (C) 2013-2014 Intel Mobile Communications GmbH
5  * Copyright (C) 2015-2017 Intel Deutschland GmbH
6  */
7 #include <linux/etherdevice.h>
8 #include <linux/crc32.h>
9 #include <net/mac80211.h>
10 #include "iwl-io.h"
11 #include "iwl-prph.h"
12 #include "fw-api.h"
13 #include "mvm.h"
14 #include "time-event.h"
15 
16 const u8 iwl_mvm_ac_to_tx_fifo[] = {
17 	IWL_MVM_TX_FIFO_VO,
18 	IWL_MVM_TX_FIFO_VI,
19 	IWL_MVM_TX_FIFO_BE,
20 	IWL_MVM_TX_FIFO_BK,
21 };
22 
23 const u8 iwl_mvm_ac_to_gen2_tx_fifo[] = {
24 	IWL_GEN2_EDCA_TX_FIFO_VO,
25 	IWL_GEN2_EDCA_TX_FIFO_VI,
26 	IWL_GEN2_EDCA_TX_FIFO_BE,
27 	IWL_GEN2_EDCA_TX_FIFO_BK,
28 	IWL_GEN2_TRIG_TX_FIFO_VO,
29 	IWL_GEN2_TRIG_TX_FIFO_VI,
30 	IWL_GEN2_TRIG_TX_FIFO_BE,
31 	IWL_GEN2_TRIG_TX_FIFO_BK,
32 };
33 
34 const u8 iwl_mvm_ac_to_bz_tx_fifo[] = {
35 	IWL_BZ_EDCA_TX_FIFO_VO,
36 	IWL_BZ_EDCA_TX_FIFO_VI,
37 	IWL_BZ_EDCA_TX_FIFO_BE,
38 	IWL_BZ_EDCA_TX_FIFO_BK,
39 	IWL_BZ_TRIG_TX_FIFO_VO,
40 	IWL_BZ_TRIG_TX_FIFO_VI,
41 	IWL_BZ_TRIG_TX_FIFO_BE,
42 	IWL_BZ_TRIG_TX_FIFO_BK,
43 };
44 
45 struct iwl_mvm_mac_iface_iterator_data {
46 	struct iwl_mvm *mvm;
47 	struct ieee80211_vif *vif;
48 	unsigned long available_mac_ids[BITS_TO_LONGS(NUM_MAC_INDEX_DRIVER)];
49 	unsigned long available_tsf_ids[BITS_TO_LONGS(NUM_TSF_IDS)];
50 	enum iwl_tsf_id preferred_tsf;
51 	bool found_vif;
52 };
53 
54 static void iwl_mvm_mac_tsf_id_iter(void *_data, u8 *mac,
55 				    struct ieee80211_vif *vif)
56 {
57 	struct iwl_mvm_mac_iface_iterator_data *data = _data;
58 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
59 	u16 min_bi;
60 
61 	/* Skip the interface for which we are trying to assign a tsf_id  */
62 	if (vif == data->vif)
63 		return;
64 
65 	/*
66 	 * The TSF is a hardware/firmware resource, there are 4 and
67 	 * the driver should assign and free them as needed. However,
68 	 * there are cases where 2 MACs should share the same TSF ID
69 	 * for the purpose of clock sync, an optimization to avoid
70 	 * clock drift causing overlapping TBTTs/DTIMs for a GO and
71 	 * client in the system.
72 	 *
73 	 * The firmware will decide according to the MAC type which
74 	 * will be the leader and follower. Clients that need to sync
75 	 * with a remote station will be the leader, and an AP or GO
76 	 * will be the follower.
77 	 *
78 	 * Depending on the new interface type it can be following
79 	 * or become the leader of an existing interface.
80 	 */
81 	switch (data->vif->type) {
82 	case NL80211_IFTYPE_STATION:
83 		/*
84 		 * The new interface is a client, so if the one we're iterating
85 		 * is an AP, and the beacon interval of the AP is a multiple or
86 		 * divisor of the beacon interval of the client, the same TSF
87 		 * should be used to avoid drift between the new client and
88 		 * existing AP. The existing AP will get drift updates from the
89 		 * new client context in this case.
90 		 */
91 		if (vif->type != NL80211_IFTYPE_AP ||
92 		    data->preferred_tsf != NUM_TSF_IDS ||
93 		    !test_bit(mvmvif->tsf_id, data->available_tsf_ids))
94 			break;
95 
96 		min_bi = min(data->vif->bss_conf.beacon_int,
97 			     vif->bss_conf.beacon_int);
98 
99 		if (!min_bi)
100 			break;
101 
102 		if ((data->vif->bss_conf.beacon_int -
103 		     vif->bss_conf.beacon_int) % min_bi == 0) {
104 			data->preferred_tsf = mvmvif->tsf_id;
105 			return;
106 		}
107 		break;
108 
109 	case NL80211_IFTYPE_AP:
110 		/*
111 		 * The new interface is AP/GO, so if its beacon interval is a
112 		 * multiple or a divisor of the beacon interval of an existing
113 		 * interface, it should get drift updates from an existing
114 		 * client or use the same TSF as an existing GO. There's no
115 		 * drift between TSFs internally but if they used different
116 		 * TSFs then a new client MAC could update one of them and
117 		 * cause drift that way.
118 		 */
119 		if ((vif->type != NL80211_IFTYPE_AP &&
120 		     vif->type != NL80211_IFTYPE_STATION) ||
121 		    data->preferred_tsf != NUM_TSF_IDS ||
122 		    !test_bit(mvmvif->tsf_id, data->available_tsf_ids))
123 			break;
124 
125 		min_bi = min(data->vif->bss_conf.beacon_int,
126 			     vif->bss_conf.beacon_int);
127 
128 		if (!min_bi)
129 			break;
130 
131 		if ((data->vif->bss_conf.beacon_int -
132 		     vif->bss_conf.beacon_int) % min_bi == 0) {
133 			data->preferred_tsf = mvmvif->tsf_id;
134 			return;
135 		}
136 		break;
137 	default:
138 		/*
139 		 * For all other interface types there's no need to
140 		 * take drift into account. Either they're exclusive
141 		 * like IBSS and monitor, or we don't care much about
142 		 * their TSF (like P2P Device), but we won't be able
143 		 * to share the TSF resource.
144 		 */
145 		break;
146 	}
147 
148 	/*
149 	 * Unless we exited above, we can't share the TSF resource
150 	 * that the virtual interface we're iterating over is using
151 	 * with the new one, so clear the available bit and if this
152 	 * was the preferred one, reset that as well.
153 	 */
154 	__clear_bit(mvmvif->tsf_id, data->available_tsf_ids);
155 
156 	if (data->preferred_tsf == mvmvif->tsf_id)
157 		data->preferred_tsf = NUM_TSF_IDS;
158 }
159 
160 static void iwl_mvm_mac_iface_iterator(void *_data, u8 *mac,
161 				       struct ieee80211_vif *vif)
162 {
163 	struct iwl_mvm_mac_iface_iterator_data *data = _data;
164 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
165 
166 	/* Iterator may already find the interface being added -- skip it */
167 	if (vif == data->vif) {
168 		data->found_vif = true;
169 		return;
170 	}
171 
172 	/* Mark MAC IDs as used by clearing the available bit, and
173 	 * (below) mark TSFs as used if their existing use is not
174 	 * compatible with the new interface type.
175 	 * No locking or atomic bit operations are needed since the
176 	 * data is on the stack of the caller function.
177 	 */
178 	__clear_bit(mvmvif->id, data->available_mac_ids);
179 
180 	/* find a suitable tsf_id */
181 	iwl_mvm_mac_tsf_id_iter(_data, mac, vif);
182 }
183 
184 void iwl_mvm_mac_ctxt_recalc_tsf_id(struct iwl_mvm *mvm,
185 				    struct ieee80211_vif *vif)
186 {
187 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
188 	struct iwl_mvm_mac_iface_iterator_data data = {
189 		.mvm = mvm,
190 		.vif = vif,
191 		.available_tsf_ids = { (1 << NUM_TSF_IDS) - 1 },
192 		/* no preference yet */
193 		.preferred_tsf = NUM_TSF_IDS,
194 	};
195 
196 	ieee80211_iterate_active_interfaces_atomic(
197 		mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
198 		iwl_mvm_mac_tsf_id_iter, &data);
199 
200 	if (data.preferred_tsf != NUM_TSF_IDS)
201 		mvmvif->tsf_id = data.preferred_tsf;
202 	else if (!test_bit(mvmvif->tsf_id, data.available_tsf_ids))
203 		mvmvif->tsf_id = find_first_bit(data.available_tsf_ids,
204 						NUM_TSF_IDS);
205 }
206 
207 int iwl_mvm_mac_ctxt_init(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
208 {
209 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
210 	struct iwl_mvm_mac_iface_iterator_data data = {
211 		.mvm = mvm,
212 		.vif = vif,
213 		.available_mac_ids = { (1 << NUM_MAC_INDEX_DRIVER) - 1 },
214 		.available_tsf_ids = { (1 << NUM_TSF_IDS) - 1 },
215 		/* no preference yet */
216 		.preferred_tsf = NUM_TSF_IDS,
217 		.found_vif = false,
218 	};
219 	int ret, i;
220 
221 	lockdep_assert_held(&mvm->mutex);
222 
223 	/*
224 	 * Allocate a MAC ID and a TSF for this MAC, along with the queues
225 	 * and other resources.
226 	 */
227 
228 	/*
229 	 * Before the iterator, we start with all MAC IDs and TSFs available.
230 	 *
231 	 * During iteration, all MAC IDs are cleared that are in use by other
232 	 * virtual interfaces, and all TSF IDs are cleared that can't be used
233 	 * by this new virtual interface because they're used by an interface
234 	 * that can't share it with the new one.
235 	 * At the same time, we check if there's a preferred TSF in the case
236 	 * that we should share it with another interface.
237 	 */
238 
239 	/* MAC ID 0 should be used only for the managed/IBSS vif with non-MLO
240 	 * FW API
241 	 */
242 	if (!mvm->mld_api_is_used) {
243 		switch (vif->type) {
244 		case NL80211_IFTYPE_ADHOC:
245 			break;
246 		case NL80211_IFTYPE_STATION:
247 			if (!vif->p2p)
248 				break;
249 			fallthrough;
250 		default:
251 			__clear_bit(0, data.available_mac_ids);
252 		}
253 	}
254 
255 	ieee80211_iterate_active_interfaces_atomic(
256 		mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
257 		iwl_mvm_mac_iface_iterator, &data);
258 
259 	/*
260 	 * In the case we're getting here during resume, it's similar to
261 	 * firmware restart, and with RESUME_ALL the iterator will find
262 	 * the vif being added already.
263 	 * We don't want to reassign any IDs in either case since doing
264 	 * so would probably assign different IDs (as interfaces aren't
265 	 * necessarily added in the same order), but the old IDs were
266 	 * preserved anyway, so skip ID assignment for both resume and
267 	 * recovery.
268 	 */
269 	if (data.found_vif)
270 		return 0;
271 
272 	/* Therefore, in recovery, we can't get here */
273 	if (WARN_ON_ONCE(test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)))
274 		return -EBUSY;
275 
276 	mvmvif->id = find_first_bit(data.available_mac_ids,
277 				    NUM_MAC_INDEX_DRIVER);
278 	if (mvmvif->id == NUM_MAC_INDEX_DRIVER) {
279 		IWL_ERR(mvm, "Failed to init MAC context - no free ID!\n");
280 		ret = -EIO;
281 		goto exit_fail;
282 	}
283 
284 	if (data.preferred_tsf != NUM_TSF_IDS)
285 		mvmvif->tsf_id = data.preferred_tsf;
286 	else
287 		mvmvif->tsf_id = find_first_bit(data.available_tsf_ids,
288 						NUM_TSF_IDS);
289 	if (mvmvif->tsf_id == NUM_TSF_IDS) {
290 		IWL_ERR(mvm, "Failed to init MAC context - no free TSF!\n");
291 		ret = -EIO;
292 		goto exit_fail;
293 	}
294 
295 	mvmvif->color = 0;
296 
297 	INIT_LIST_HEAD(&mvmvif->time_event_data.list);
298 	mvmvif->time_event_data.id = TE_MAX;
299 
300 	mvmvif->deflink.bcast_sta.sta_id = IWL_MVM_INVALID_STA;
301 	mvmvif->deflink.mcast_sta.sta_id = IWL_MVM_INVALID_STA;
302 	mvmvif->deflink.ap_sta_id = IWL_MVM_INVALID_STA;
303 
304 	/* No need to allocate data queues to P2P Device MAC and NAN.*/
305 	if (vif->type == NL80211_IFTYPE_P2P_DEVICE)
306 		return 0;
307 
308 	/* Allocate the CAB queue for softAP and GO interfaces */
309 	if (vif->type == NL80211_IFTYPE_AP ||
310 	    vif->type == NL80211_IFTYPE_ADHOC) {
311 		/*
312 		 * For TVQM this will be overwritten later with the FW assigned
313 		 * queue value (when queue is enabled).
314 		 */
315 		mvmvif->deflink.cab_queue = IWL_MVM_DQA_GCAST_QUEUE;
316 	}
317 
318 	for (i = 0; i < NUM_IWL_MVM_SMPS_REQ; i++)
319 		mvmvif->deflink.smps_requests[i] = IEEE80211_SMPS_AUTOMATIC;
320 
321 	return 0;
322 
323 exit_fail:
324 	memset(mvmvif, 0, sizeof(struct iwl_mvm_vif));
325 	return ret;
326 }
327 
328 static void iwl_mvm_ack_rates(struct iwl_mvm *mvm,
329 			      struct ieee80211_vif *vif,
330 			      enum nl80211_band band,
331 			      u8 *cck_rates, u8 *ofdm_rates)
332 {
333 	struct ieee80211_supported_band *sband;
334 	unsigned long basic = vif->bss_conf.basic_rates;
335 	int lowest_present_ofdm = 100;
336 	int lowest_present_cck = 100;
337 	u8 cck = 0;
338 	u8 ofdm = 0;
339 	int i;
340 
341 	sband = mvm->hw->wiphy->bands[band];
342 
343 	for_each_set_bit(i, &basic, BITS_PER_LONG) {
344 		int hw = sband->bitrates[i].hw_value;
345 		if (hw >= IWL_FIRST_OFDM_RATE) {
346 			ofdm |= BIT(hw - IWL_FIRST_OFDM_RATE);
347 			if (lowest_present_ofdm > hw)
348 				lowest_present_ofdm = hw;
349 		} else {
350 			BUILD_BUG_ON(IWL_FIRST_CCK_RATE != 0);
351 
352 			cck |= BIT(hw);
353 			if (lowest_present_cck > hw)
354 				lowest_present_cck = hw;
355 		}
356 	}
357 
358 	/*
359 	 * Now we've got the basic rates as bitmaps in the ofdm and cck
360 	 * variables. This isn't sufficient though, as there might not
361 	 * be all the right rates in the bitmap. E.g. if the only basic
362 	 * rates are 5.5 Mbps and 11 Mbps, we still need to add 1 Mbps
363 	 * and 6 Mbps because the 802.11-2007 standard says in 9.6:
364 	 *
365 	 *    [...] a STA responding to a received frame shall transmit
366 	 *    its Control Response frame [...] at the highest rate in the
367 	 *    BSSBasicRateSet parameter that is less than or equal to the
368 	 *    rate of the immediately previous frame in the frame exchange
369 	 *    sequence ([...]) and that is of the same modulation class
370 	 *    ([...]) as the received frame. If no rate contained in the
371 	 *    BSSBasicRateSet parameter meets these conditions, then the
372 	 *    control frame sent in response to a received frame shall be
373 	 *    transmitted at the highest mandatory rate of the PHY that is
374 	 *    less than or equal to the rate of the received frame, and
375 	 *    that is of the same modulation class as the received frame.
376 	 *
377 	 * As a consequence, we need to add all mandatory rates that are
378 	 * lower than all of the basic rates to these bitmaps.
379 	 */
380 
381 	if (IWL_RATE_24M_INDEX < lowest_present_ofdm)
382 		ofdm |= IWL_RATE_BIT_MSK(24) >> IWL_FIRST_OFDM_RATE;
383 	if (IWL_RATE_12M_INDEX < lowest_present_ofdm)
384 		ofdm |= IWL_RATE_BIT_MSK(12) >> IWL_FIRST_OFDM_RATE;
385 	/* 6M already there or needed so always add */
386 	ofdm |= IWL_RATE_BIT_MSK(6) >> IWL_FIRST_OFDM_RATE;
387 
388 	/*
389 	 * CCK is a bit more complex with DSSS vs. HR/DSSS vs. ERP.
390 	 * Note, however:
391 	 *  - if no CCK rates are basic, it must be ERP since there must
392 	 *    be some basic rates at all, so they're OFDM => ERP PHY
393 	 *    (or we're in 5 GHz, and the cck bitmap will never be used)
394 	 *  - if 11M is a basic rate, it must be ERP as well, so add 5.5M
395 	 *  - if 5.5M is basic, 1M and 2M are mandatory
396 	 *  - if 2M is basic, 1M is mandatory
397 	 *  - if 1M is basic, that's the only valid ACK rate.
398 	 * As a consequence, it's not as complicated as it sounds, just add
399 	 * any lower rates to the ACK rate bitmap.
400 	 */
401 	if (IWL_RATE_11M_INDEX < lowest_present_cck)
402 		cck |= IWL_RATE_BIT_MSK(11) >> IWL_FIRST_CCK_RATE;
403 	if (IWL_RATE_5M_INDEX < lowest_present_cck)
404 		cck |= IWL_RATE_BIT_MSK(5) >> IWL_FIRST_CCK_RATE;
405 	if (IWL_RATE_2M_INDEX < lowest_present_cck)
406 		cck |= IWL_RATE_BIT_MSK(2) >> IWL_FIRST_CCK_RATE;
407 	/* 1M already there or needed so always add */
408 	cck |= IWL_RATE_BIT_MSK(1) >> IWL_FIRST_CCK_RATE;
409 
410 	*cck_rates = cck;
411 	*ofdm_rates = ofdm;
412 }
413 
414 void iwl_mvm_set_fw_basic_rates(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
415 				struct ieee80211_bss_conf *link_conf,
416 				__le32 *cck_rates, __le32 *ofdm_rates)
417 {
418 	struct ieee80211_chanctx_conf *chanctx;
419 	u8 cck_ack_rates = 0, ofdm_ack_rates = 0;
420 
421 	rcu_read_lock();
422 	chanctx = rcu_dereference(link_conf->chanctx_conf);
423 	iwl_mvm_ack_rates(mvm, vif, chanctx ? chanctx->def.chan->band
424 					    : NL80211_BAND_2GHZ,
425 			  &cck_ack_rates, &ofdm_ack_rates);
426 
427 	rcu_read_unlock();
428 
429 	*cck_rates = cpu_to_le32((u32)cck_ack_rates);
430 	*ofdm_rates = cpu_to_le32((u32)ofdm_ack_rates);
431 }
432 
433 void iwl_mvm_set_fw_protection_flags(struct iwl_mvm *mvm,
434 				     struct ieee80211_vif *vif,
435 				     struct ieee80211_bss_conf *link_conf,
436 				     __le32 *protection_flags, u32 ht_flag,
437 				     u32 tgg_flag)
438 {
439 	/* for both sta and ap, ht_operation_mode hold the protection_mode */
440 	u8 protection_mode = link_conf->ht_operation_mode &
441 				 IEEE80211_HT_OP_MODE_PROTECTION;
442 	bool ht_enabled = !!(link_conf->ht_operation_mode &
443 			     IEEE80211_HT_OP_MODE_PROTECTION);
444 
445 	if (link_conf->use_cts_prot)
446 		*protection_flags |= cpu_to_le32(tgg_flag);
447 
448 	IWL_DEBUG_RATE(mvm, "use_cts_prot %d, ht_operation_mode %d\n",
449 		       link_conf->use_cts_prot,
450 		       link_conf->ht_operation_mode);
451 
452 	if (!ht_enabled)
453 		return;
454 
455 	IWL_DEBUG_RATE(mvm, "protection mode set to %d\n", protection_mode);
456 	/*
457 	 * See section 9.23.3.1 of IEEE 80211-2012.
458 	 * Nongreenfield HT STAs Present is not supported.
459 	 */
460 	switch (protection_mode) {
461 	case IEEE80211_HT_OP_MODE_PROTECTION_NONE:
462 		break;
463 	case IEEE80211_HT_OP_MODE_PROTECTION_NONMEMBER:
464 	case IEEE80211_HT_OP_MODE_PROTECTION_NONHT_MIXED:
465 		*protection_flags |= cpu_to_le32(ht_flag);
466 		break;
467 	case IEEE80211_HT_OP_MODE_PROTECTION_20MHZ:
468 		/* Protect when channel wider than 20MHz */
469 		if (link_conf->chanreq.oper.width > NL80211_CHAN_WIDTH_20)
470 			*protection_flags |= cpu_to_le32(ht_flag);
471 		break;
472 	default:
473 		IWL_ERR(mvm, "Illegal protection mode %d\n",
474 			protection_mode);
475 		break;
476 	}
477 }
478 
479 void iwl_mvm_set_fw_qos_params(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
480 			       struct ieee80211_bss_conf *link_conf,
481 			       struct iwl_ac_qos *ac, __le32 *qos_flags)
482 {
483 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
484 	struct iwl_mvm_vif_link_info *mvm_link =
485 		mvmvif->link[link_conf->link_id];
486 	int i;
487 
488 	if (!mvm_link)
489 		return;
490 
491 	for (i = 0; i < IEEE80211_NUM_ACS; i++) {
492 		u8 txf = iwl_mvm_mac_ac_to_tx_fifo(mvm, i);
493 		u8 ucode_ac = iwl_mvm_mac80211_ac_to_ucode_ac(i);
494 
495 		ac[ucode_ac].cw_min =
496 			cpu_to_le16(mvm_link->queue_params[i].cw_min);
497 		ac[ucode_ac].cw_max =
498 			cpu_to_le16(mvm_link->queue_params[i].cw_max);
499 		ac[ucode_ac].edca_txop =
500 			cpu_to_le16(mvm_link->queue_params[i].txop * 32);
501 		ac[ucode_ac].aifsn = mvm_link->queue_params[i].aifs;
502 		ac[ucode_ac].fifos_mask = BIT(txf);
503 	}
504 
505 	if (link_conf->qos)
506 		*qos_flags |= cpu_to_le32(MAC_QOS_FLG_UPDATE_EDCA);
507 
508 	if (link_conf->chanreq.oper.width != NL80211_CHAN_WIDTH_20_NOHT)
509 		*qos_flags |= cpu_to_le32(MAC_QOS_FLG_TGN);
510 }
511 
512 int iwl_mvm_get_mac_type(struct ieee80211_vif *vif)
513 {
514 	u32 mac_type = FW_MAC_TYPE_BSS_STA;
515 
516 	switch (vif->type) {
517 	case NL80211_IFTYPE_STATION:
518 		if (vif->p2p)
519 			mac_type = FW_MAC_TYPE_P2P_STA;
520 		else
521 			mac_type = FW_MAC_TYPE_BSS_STA;
522 		break;
523 	case NL80211_IFTYPE_AP:
524 		mac_type = FW_MAC_TYPE_GO;
525 		break;
526 	case NL80211_IFTYPE_MONITOR:
527 		mac_type = FW_MAC_TYPE_LISTENER;
528 		break;
529 	case NL80211_IFTYPE_P2P_DEVICE:
530 		mac_type = FW_MAC_TYPE_P2P_DEVICE;
531 		break;
532 	case NL80211_IFTYPE_ADHOC:
533 		mac_type = FW_MAC_TYPE_IBSS;
534 		break;
535 	default:
536 		WARN_ON_ONCE(1);
537 	}
538 	return mac_type;
539 }
540 
541 static void iwl_mvm_mac_ctxt_cmd_common(struct iwl_mvm *mvm,
542 					struct ieee80211_vif *vif,
543 					struct iwl_mac_ctx_cmd *cmd,
544 					const u8 *bssid_override,
545 					u32 action)
546 {
547 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
548 	const u8 *bssid = bssid_override ?: vif->bss_conf.bssid;
549 	u32 ht_flag;
550 
551 	cmd->id_and_color = cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id,
552 							    mvmvif->color));
553 	cmd->action = cpu_to_le32(action);
554 	cmd->mac_type = cpu_to_le32(iwl_mvm_get_mac_type(vif));
555 
556 	cmd->tsf_id = cpu_to_le32(mvmvif->tsf_id);
557 
558 	memcpy(cmd->node_addr, vif->addr, ETH_ALEN);
559 
560 	if (bssid)
561 		memcpy(cmd->bssid_addr, bssid, ETH_ALEN);
562 	else
563 		eth_broadcast_addr(cmd->bssid_addr);
564 
565 	iwl_mvm_set_fw_basic_rates(mvm, vif, &vif->bss_conf, &cmd->cck_rates,
566 				   &cmd->ofdm_rates);
567 
568 	cmd->cck_short_preamble =
569 		cpu_to_le32(vif->bss_conf.use_short_preamble ?
570 			    MAC_FLG_SHORT_PREAMBLE : 0);
571 	cmd->short_slot =
572 		cpu_to_le32(vif->bss_conf.use_short_slot ?
573 			    MAC_FLG_SHORT_SLOT : 0);
574 
575 	cmd->filter_flags = 0;
576 
577 	iwl_mvm_set_fw_qos_params(mvm, vif, &vif->bss_conf, cmd->ac,
578 				  &cmd->qos_flags);
579 
580 	/* The fw does not distinguish between ht and fat */
581 	ht_flag = MAC_PROT_FLG_HT_PROT | MAC_PROT_FLG_FAT_PROT;
582 	iwl_mvm_set_fw_protection_flags(mvm, vif, &vif->bss_conf,
583 					&cmd->protection_flags,
584 					ht_flag, MAC_PROT_FLG_TGG_PROTECT);
585 }
586 
587 static int iwl_mvm_mac_ctxt_send_cmd(struct iwl_mvm *mvm,
588 				     struct iwl_mac_ctx_cmd *cmd)
589 {
590 	int ret = iwl_mvm_send_cmd_pdu(mvm, MAC_CONTEXT_CMD, 0,
591 				       sizeof(*cmd), cmd);
592 	if (ret)
593 		IWL_ERR(mvm, "Failed to send MAC_CONTEXT_CMD (action:%d): %d\n",
594 			le32_to_cpu(cmd->action), ret);
595 	return ret;
596 }
597 
598 void iwl_mvm_set_fw_dtim_tbtt(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
599 			      struct ieee80211_bss_conf *link_conf,
600 			      __le64 *dtim_tsf, __le32 *dtim_time,
601 			      __le32 *assoc_beacon_arrive_time)
602 {
603 	u32 dtim_offs;
604 
605 	/*
606 	 * The DTIM count counts down, so when it is N that means N
607 	 * more beacon intervals happen until the DTIM TBTT. Therefore
608 	 * add this to the current time. If that ends up being in the
609 	 * future, the firmware will handle it.
610 	 *
611 	 * Also note that the system_timestamp (which we get here as
612 	 * "sync_device_ts") and TSF timestamp aren't at exactly the
613 	 * same offset in the frame -- the TSF is at the first symbol
614 	 * of the TSF, the system timestamp is at signal acquisition
615 	 * time. This means there's an offset between them of at most
616 	 * a few hundred microseconds (24 * 8 bits + PLCP time gives
617 	 * 384us in the longest case), this is currently not relevant
618 	 * as the firmware wakes up around 2ms before the TBTT.
619 	 */
620 	dtim_offs = link_conf->sync_dtim_count *
621 			link_conf->beacon_int;
622 	/* convert TU to usecs */
623 	dtim_offs *= 1024;
624 
625 	*dtim_tsf =
626 		cpu_to_le64(link_conf->sync_tsf + dtim_offs);
627 	*dtim_time =
628 		cpu_to_le32(link_conf->sync_device_ts + dtim_offs);
629 	*assoc_beacon_arrive_time =
630 		cpu_to_le32(link_conf->sync_device_ts);
631 
632 	IWL_DEBUG_INFO(mvm, "DTIM TBTT is 0x%llx/0x%x, offset %d\n",
633 		       le64_to_cpu(*dtim_tsf),
634 		       le32_to_cpu(*dtim_time),
635 		       dtim_offs);
636 }
637 
638 __le32 iwl_mvm_mac_ctxt_cmd_p2p_sta_get_oppps_ctwin(struct iwl_mvm *mvm,
639 						    struct ieee80211_vif *vif)
640 {
641 	struct ieee80211_p2p_noa_attr *noa =
642 		&vif->bss_conf.p2p_noa_attr;
643 
644 	return cpu_to_le32(noa->oppps_ctwindow &
645 			IEEE80211_P2P_OPPPS_CTWINDOW_MASK);
646 }
647 
648 u32 iwl_mvm_mac_ctxt_cmd_sta_get_twt_policy(struct iwl_mvm *mvm,
649 					    struct ieee80211_vif *vif)
650 {
651 	u32 twt_policy = 0;
652 
653 	if (vif->bss_conf.twt_requester && IWL_MVM_USE_TWT)
654 		twt_policy |= TWT_SUPPORTED;
655 	if (vif->bss_conf.twt_protected)
656 		twt_policy |= PROTECTED_TWT_SUPPORTED;
657 	if (vif->bss_conf.twt_broadcast)
658 		twt_policy |= BROADCAST_TWT_SUPPORTED;
659 
660 	return twt_policy;
661 }
662 
663 static int iwl_mvm_mac_ctxt_cmd_sta(struct iwl_mvm *mvm,
664 				    struct ieee80211_vif *vif,
665 				    u32 action, bool force_assoc_off,
666 				    const u8 *bssid_override)
667 {
668 	struct iwl_mac_ctx_cmd cmd = {};
669 	struct iwl_mac_data_sta *ctxt_sta;
670 
671 	WARN_ON(vif->type != NL80211_IFTYPE_STATION);
672 
673 	/* Fill the common data for all mac context types */
674 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, bssid_override, action);
675 
676 	/*
677 	 * We always want to hear MCAST frames, if we're not authorized yet,
678 	 * we'll drop them.
679 	 */
680 	cmd.filter_flags |= cpu_to_le32(MAC_FILTER_ACCEPT_GRP);
681 
682 	if (vif->p2p) {
683 		cmd.p2p_sta.ctwin =
684 			iwl_mvm_mac_ctxt_cmd_p2p_sta_get_oppps_ctwin(mvm, vif);
685 
686 		ctxt_sta = &cmd.p2p_sta.sta;
687 	} else {
688 		ctxt_sta = &cmd.sta;
689 	}
690 
691 	/* We need the dtim_period to set the MAC as associated */
692 	if (vif->cfg.assoc && vif->bss_conf.dtim_period &&
693 	    !force_assoc_off) {
694 		struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
695 
696 		iwl_mvm_set_fw_dtim_tbtt(mvm, vif, &vif->bss_conf,
697 					 &ctxt_sta->dtim_tsf,
698 					 &ctxt_sta->dtim_time,
699 					 &ctxt_sta->assoc_beacon_arrive_time);
700 
701 		ctxt_sta->is_assoc = cpu_to_le32(1);
702 
703 		if (!mvmvif->authorized &&
704 		    fw_has_capa(&mvm->fw->ucode_capa,
705 				IWL_UCODE_TLV_CAPA_COEX_HIGH_PRIO))
706 			ctxt_sta->data_policy |=
707 				cpu_to_le32(COEX_HIGH_PRIORITY_ENABLE);
708 	} else {
709 		ctxt_sta->is_assoc = cpu_to_le32(0);
710 
711 		/* Allow beacons to pass through as long as we are not
712 		 * associated, or we do not have dtim period information.
713 		 */
714 		cmd.filter_flags |= cpu_to_le32(MAC_FILTER_IN_BEACON);
715 	}
716 
717 	ctxt_sta->bi = cpu_to_le32(vif->bss_conf.beacon_int);
718 	ctxt_sta->dtim_interval = cpu_to_le32(vif->bss_conf.beacon_int *
719 					      vif->bss_conf.dtim_period);
720 
721 	ctxt_sta->listen_interval = cpu_to_le32(mvm->hw->conf.listen_interval);
722 	ctxt_sta->assoc_id = cpu_to_le32(vif->cfg.aid);
723 
724 	if (vif->probe_req_reg && vif->cfg.assoc && vif->p2p)
725 		cmd.filter_flags |= cpu_to_le32(MAC_FILTER_IN_PROBE_REQUEST);
726 
727 	if (vif->bss_conf.he_support && !iwlwifi_mod_params.disable_11ax) {
728 		cmd.filter_flags |= cpu_to_le32(MAC_FILTER_IN_11AX);
729 		ctxt_sta->data_policy |=
730 			cpu_to_le32(iwl_mvm_mac_ctxt_cmd_sta_get_twt_policy(mvm, vif));
731 	}
732 
733 
734 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
735 }
736 
737 static int iwl_mvm_mac_ctxt_cmd_listener(struct iwl_mvm *mvm,
738 					 struct ieee80211_vif *vif,
739 					 u32 action)
740 {
741 	struct iwl_mac_ctx_cmd cmd = {};
742 	u32 tfd_queue_msk = 0;
743 	int ret;
744 
745 	WARN_ON(vif->type != NL80211_IFTYPE_MONITOR);
746 
747 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, NULL, action);
748 
749 	cmd.filter_flags = cpu_to_le32(MAC_FILTER_IN_PROMISC |
750 				       MAC_FILTER_IN_CONTROL_AND_MGMT |
751 				       MAC_FILTER_IN_BEACON |
752 				       MAC_FILTER_IN_PROBE_REQUEST |
753 				       MAC_FILTER_IN_CRC32 |
754 				       MAC_FILTER_ACCEPT_GRP);
755 	ieee80211_hw_set(mvm->hw, RX_INCLUDES_FCS);
756 
757 	/*
758 	 * the queue mask is only relevant for old TX API, and
759 	 * mvm->snif_queue isn't set here (it's still set to
760 	 * IWL_MVM_INVALID_QUEUE so the BIT() of it is UB)
761 	 */
762 	if (!iwl_mvm_has_new_tx_api(mvm))
763 		tfd_queue_msk = BIT(mvm->snif_queue);
764 
765 	/* Allocate sniffer station */
766 	ret = iwl_mvm_allocate_int_sta(mvm, &mvm->snif_sta, tfd_queue_msk,
767 				       vif->type, IWL_STA_GENERAL_PURPOSE);
768 	if (ret)
769 		return ret;
770 
771 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
772 }
773 
774 static int iwl_mvm_mac_ctxt_cmd_ibss(struct iwl_mvm *mvm,
775 				     struct ieee80211_vif *vif,
776 				     u32 action)
777 {
778 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
779 	struct iwl_mac_ctx_cmd cmd = {};
780 
781 	WARN_ON(vif->type != NL80211_IFTYPE_ADHOC);
782 
783 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, NULL, action);
784 
785 	cmd.filter_flags = cpu_to_le32(MAC_FILTER_IN_BEACON |
786 				       MAC_FILTER_IN_PROBE_REQUEST |
787 				       MAC_FILTER_ACCEPT_GRP);
788 
789 	/* cmd.ibss.beacon_time/cmd.ibss.beacon_tsf are curently ignored */
790 	cmd.ibss.bi = cpu_to_le32(vif->bss_conf.beacon_int);
791 
792 	/* TODO: Assumes that the beacon id == mac context id */
793 	cmd.ibss.beacon_template = cpu_to_le32(mvmvif->id);
794 
795 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
796 }
797 
798 struct iwl_mvm_go_iterator_data {
799 	bool go_active;
800 };
801 
802 static void iwl_mvm_go_iterator(void *_data, u8 *mac, struct ieee80211_vif *vif)
803 {
804 	struct iwl_mvm_go_iterator_data *data = _data;
805 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
806 
807 	if (vif->type == NL80211_IFTYPE_AP && vif->p2p &&
808 	    mvmvif->ap_ibss_active)
809 		data->go_active = true;
810 }
811 
812 __le32 iwl_mac_ctxt_p2p_dev_has_extended_disc(struct iwl_mvm *mvm,
813 					      struct ieee80211_vif *vif)
814 {
815 	struct iwl_mvm_go_iterator_data data = {};
816 
817 	/*
818 	 * This flag should be set to true when the P2P Device is
819 	 * discoverable and there is at least another active P2P GO. Settings
820 	 * this flag will allow the P2P Device to be discoverable on other
821 	 * channels in addition to its listen channel.
822 	 * Note that this flag should not be set in other cases as it opens the
823 	 * Rx filters on all MAC and increases the number of interrupts.
824 	 */
825 	ieee80211_iterate_active_interfaces_atomic(
826 		mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
827 		iwl_mvm_go_iterator, &data);
828 
829 	return cpu_to_le32(data.go_active ? 1 : 0);
830 }
831 
832 static int iwl_mvm_mac_ctxt_cmd_p2p_device(struct iwl_mvm *mvm,
833 					   struct ieee80211_vif *vif,
834 					   u32 action)
835 {
836 	struct iwl_mac_ctx_cmd cmd = {};
837 
838 	WARN_ON(vif->type != NL80211_IFTYPE_P2P_DEVICE);
839 
840 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, NULL, action);
841 
842 	cmd.p2p_dev.is_disc_extended =
843 		iwl_mac_ctxt_p2p_dev_has_extended_disc(mvm, vif);
844 
845 	/* Override the filter flags to accept only probe requests */
846 	cmd.filter_flags = cpu_to_le32(MAC_FILTER_IN_PROBE_REQUEST);
847 
848 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
849 }
850 
851 void iwl_mvm_mac_ctxt_set_tim(struct iwl_mvm *mvm,
852 			      __le32 *tim_index, __le32 *tim_size,
853 			      u8 *beacon, u32 frame_size)
854 {
855 	u32 tim_idx;
856 	struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)beacon;
857 
858 	/* The index is relative to frame start but we start looking at the
859 	 * variable-length part of the beacon. */
860 	tim_idx = mgmt->u.beacon.variable - beacon;
861 
862 	/* Parse variable-length elements of beacon to find WLAN_EID_TIM */
863 	while ((tim_idx < (frame_size - 2)) &&
864 			(beacon[tim_idx] != WLAN_EID_TIM))
865 		tim_idx += beacon[tim_idx+1] + 2;
866 
867 	/* If TIM field was found, set variables */
868 	if ((tim_idx < (frame_size - 1)) && (beacon[tim_idx] == WLAN_EID_TIM)) {
869 		*tim_index = cpu_to_le32(tim_idx);
870 		*tim_size = cpu_to_le32((u32)beacon[tim_idx + 1]);
871 	} else {
872 		IWL_WARN(mvm, "Unable to find TIM Element in beacon\n");
873 	}
874 }
875 
876 static u32 iwl_mvm_find_ie_offset(u8 *beacon, u8 eid, u32 frame_size)
877 {
878 	struct ieee80211_mgmt *mgmt = (void *)beacon;
879 	const u8 *ie;
880 
881 	if (WARN_ON_ONCE(frame_size <= (mgmt->u.beacon.variable - beacon)))
882 		return 0;
883 
884 	frame_size -= mgmt->u.beacon.variable - beacon;
885 
886 	ie = cfg80211_find_ie(eid, mgmt->u.beacon.variable, frame_size);
887 	if (!ie)
888 		return 0;
889 
890 	return ie - beacon;
891 }
892 
893 u8 iwl_mvm_mac_ctxt_get_lowest_rate(struct iwl_mvm *mvm,
894 				    struct ieee80211_tx_info *info,
895 				    struct ieee80211_vif *vif)
896 {
897 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
898 	struct ieee80211_supported_band *sband;
899 	unsigned long basic = vif->bss_conf.basic_rates;
900 	u16 lowest_cck = IWL_RATE_COUNT, lowest_ofdm = IWL_RATE_COUNT;
901 	u32 link_id = u32_get_bits(info->control.flags,
902 				   IEEE80211_TX_CTRL_MLO_LINK);
903 	u8 band = info->band;
904 	u8 rate;
905 	u32 i;
906 
907 	if (link_id == IEEE80211_LINK_UNSPECIFIED && ieee80211_vif_is_mld(vif)) {
908 		for (i = 0; i < ARRAY_SIZE(mvmvif->link); i++) {
909 			if (!mvmvif->link[i])
910 				continue;
911 			/* shouldn't do this when >1 link is active */
912 			WARN_ON_ONCE(link_id != IEEE80211_LINK_UNSPECIFIED);
913 			link_id = i;
914 		}
915 	}
916 
917 	if (link_id < IEEE80211_LINK_UNSPECIFIED) {
918 		struct ieee80211_bss_conf *link_conf;
919 
920 		rcu_read_lock();
921 		link_conf = rcu_dereference(vif->link_conf[link_id]);
922 		if (link_conf) {
923 			basic = link_conf->basic_rates;
924 			if (link_conf->chanreq.oper.chan)
925 				band = link_conf->chanreq.oper.chan->band;
926 		}
927 		rcu_read_unlock();
928 	}
929 
930 	sband = mvm->hw->wiphy->bands[band];
931 	for_each_set_bit(i, &basic, BITS_PER_LONG) {
932 		u16 hw = sband->bitrates[i].hw_value;
933 
934 		if (hw >= IWL_FIRST_OFDM_RATE) {
935 			if (lowest_ofdm > hw)
936 				lowest_ofdm = hw;
937 		} else if (lowest_cck > hw) {
938 			lowest_cck = hw;
939 		}
940 	}
941 
942 	if (band == NL80211_BAND_2GHZ && !vif->p2p &&
943 	    vif->type != NL80211_IFTYPE_P2P_DEVICE &&
944 	    !(info->flags & IEEE80211_TX_CTL_NO_CCK_RATE)) {
945 		if (lowest_cck != IWL_RATE_COUNT)
946 			rate = lowest_cck;
947 		else if (lowest_ofdm != IWL_RATE_COUNT)
948 			rate = lowest_ofdm;
949 		else
950 			rate = IWL_RATE_1M_INDEX;
951 	} else if (lowest_ofdm != IWL_RATE_COUNT) {
952 		rate = lowest_ofdm;
953 	} else {
954 		rate = IWL_RATE_6M_INDEX;
955 	}
956 
957 	return rate;
958 }
959 
960 u16 iwl_mvm_mac_ctxt_get_beacon_flags(const struct iwl_fw *fw, u8 rate_idx)
961 {
962 	u16 flags = iwl_mvm_mac80211_idx_to_hwrate(fw, rate_idx);
963 	bool is_new_rate = iwl_fw_lookup_cmd_ver(fw, BEACON_TEMPLATE_CMD, 0) > 10;
964 
965 	if (rate_idx <= IWL_FIRST_CCK_RATE)
966 		flags |= is_new_rate ? IWL_MAC_BEACON_CCK
967 			  : IWL_MAC_BEACON_CCK_V1;
968 
969 	return flags;
970 }
971 
972 u8 iwl_mvm_mac_ctxt_get_beacon_rate(struct iwl_mvm *mvm,
973 				    struct ieee80211_tx_info *info,
974 				    struct ieee80211_vif *vif)
975 {
976 	struct ieee80211_supported_band *sband =
977 		mvm->hw->wiphy->bands[info->band];
978 	u32 legacy = vif->bss_conf.beacon_tx_rate.control[info->band].legacy;
979 
980 	/* if beacon rate was configured try using it */
981 	if (hweight32(legacy) == 1) {
982 		u32 rate = ffs(legacy) - 1;
983 
984 		return sband->bitrates[rate].hw_value;
985 	}
986 
987 	return iwl_mvm_mac_ctxt_get_lowest_rate(mvm, info, vif);
988 }
989 
990 static void iwl_mvm_mac_ctxt_set_tx(struct iwl_mvm *mvm,
991 				    struct ieee80211_vif *vif,
992 				    struct sk_buff *beacon,
993 				    struct iwl_tx_cmd *tx)
994 {
995 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
996 	struct ieee80211_tx_info *info;
997 	u8 rate;
998 	u32 tx_flags;
999 
1000 	info = IEEE80211_SKB_CB(beacon);
1001 
1002 	/* Set up TX command fields */
1003 	tx->len = cpu_to_le16((u16)beacon->len);
1004 	tx->sta_id = mvmvif->deflink.bcast_sta.sta_id;
1005 	tx->life_time = cpu_to_le32(TX_CMD_LIFE_TIME_INFINITE);
1006 	tx_flags = TX_CMD_FLG_SEQ_CTL | TX_CMD_FLG_TSF;
1007 	tx_flags |=
1008 		iwl_mvm_bt_coex_tx_prio(mvm, (void *)beacon->data, info, 0) <<
1009 						TX_CMD_FLG_BT_PRIO_POS;
1010 	tx->tx_flags = cpu_to_le32(tx_flags);
1011 
1012 	if (!fw_has_capa(&mvm->fw->ucode_capa,
1013 			 IWL_UCODE_TLV_CAPA_BEACON_ANT_SELECTION))
1014 		iwl_mvm_toggle_tx_ant(mvm, &mvm->mgmt_last_antenna_idx);
1015 
1016 	tx->rate_n_flags =
1017 		cpu_to_le32(BIT(mvm->mgmt_last_antenna_idx) <<
1018 			    RATE_MCS_ANT_POS);
1019 
1020 	rate = iwl_mvm_mac_ctxt_get_beacon_rate(mvm, info, vif);
1021 
1022 	tx->rate_n_flags |=
1023 		cpu_to_le32(iwl_mvm_mac80211_idx_to_hwrate(mvm->fw, rate));
1024 	if (rate == IWL_FIRST_CCK_RATE)
1025 		tx->rate_n_flags |= cpu_to_le32(RATE_MCS_CCK_MSK_V1);
1026 
1027 }
1028 
1029 int iwl_mvm_mac_ctxt_send_beacon_cmd(struct iwl_mvm *mvm,
1030 				     struct sk_buff *beacon,
1031 				     void *data, int len)
1032 {
1033 	struct iwl_host_cmd cmd = {
1034 		.id = BEACON_TEMPLATE_CMD,
1035 		.flags = CMD_ASYNC,
1036 	};
1037 
1038 	cmd.len[0] = len;
1039 	cmd.data[0] = data;
1040 	cmd.dataflags[0] = 0;
1041 	cmd.len[1] = beacon->len;
1042 	cmd.data[1] = beacon->data;
1043 	cmd.dataflags[1] = IWL_HCMD_DFL_DUP;
1044 
1045 	return iwl_mvm_send_cmd(mvm, &cmd);
1046 }
1047 
1048 static int iwl_mvm_mac_ctxt_send_beacon_v6(struct iwl_mvm *mvm,
1049 					   struct ieee80211_vif *vif,
1050 					   struct sk_buff *beacon)
1051 {
1052 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1053 	struct iwl_mac_beacon_cmd_v6 beacon_cmd = {};
1054 
1055 	iwl_mvm_mac_ctxt_set_tx(mvm, vif, beacon, &beacon_cmd.tx);
1056 
1057 	beacon_cmd.template_id = cpu_to_le32((u32)mvmvif->id);
1058 
1059 	if (vif->type == NL80211_IFTYPE_AP)
1060 		iwl_mvm_mac_ctxt_set_tim(mvm, &beacon_cmd.tim_idx,
1061 					 &beacon_cmd.tim_size,
1062 					 beacon->data, beacon->len);
1063 
1064 	return iwl_mvm_mac_ctxt_send_beacon_cmd(mvm, beacon, &beacon_cmd,
1065 						sizeof(beacon_cmd));
1066 }
1067 
1068 static int iwl_mvm_mac_ctxt_send_beacon_v7(struct iwl_mvm *mvm,
1069 					   struct ieee80211_vif *vif,
1070 					   struct sk_buff *beacon)
1071 {
1072 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1073 	struct iwl_mac_beacon_cmd_v7 beacon_cmd = {};
1074 
1075 	iwl_mvm_mac_ctxt_set_tx(mvm, vif, beacon, &beacon_cmd.tx);
1076 
1077 	beacon_cmd.template_id = cpu_to_le32((u32)mvmvif->id);
1078 
1079 	if (vif->type == NL80211_IFTYPE_AP)
1080 		iwl_mvm_mac_ctxt_set_tim(mvm, &beacon_cmd.tim_idx,
1081 					 &beacon_cmd.tim_size,
1082 					 beacon->data, beacon->len);
1083 
1084 	beacon_cmd.csa_offset =
1085 		cpu_to_le32(iwl_mvm_find_ie_offset(beacon->data,
1086 						   WLAN_EID_CHANNEL_SWITCH,
1087 						   beacon->len));
1088 	beacon_cmd.ecsa_offset =
1089 		cpu_to_le32(iwl_mvm_find_ie_offset(beacon->data,
1090 						   WLAN_EID_EXT_CHANSWITCH_ANN,
1091 						   beacon->len));
1092 
1093 	return iwl_mvm_mac_ctxt_send_beacon_cmd(mvm, beacon, &beacon_cmd,
1094 						sizeof(beacon_cmd));
1095 }
1096 
1097 bool iwl_mvm_enable_fils(struct iwl_mvm *mvm,
1098 			 struct ieee80211_chanctx_conf *ctx)
1099 {
1100 	if (IWL_MVM_DISABLE_AP_FILS)
1101 		return false;
1102 
1103 	if (cfg80211_channel_is_psc(ctx->def.chan))
1104 		return true;
1105 
1106 	return (ctx->def.chan->band == NL80211_BAND_6GHZ &&
1107 		ctx->def.width >= NL80211_CHAN_WIDTH_80);
1108 }
1109 
1110 static int iwl_mvm_mac_ctxt_send_beacon_v9(struct iwl_mvm *mvm,
1111 					   struct ieee80211_vif *vif,
1112 					   struct sk_buff *beacon,
1113 					   struct ieee80211_bss_conf *link_conf)
1114 {
1115 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1116 	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(beacon);
1117 	struct iwl_mac_beacon_cmd beacon_cmd = {};
1118 	u8 rate = iwl_mvm_mac_ctxt_get_beacon_rate(mvm, info, vif);
1119 	u16 flags;
1120 	struct ieee80211_chanctx_conf *ctx;
1121 	int channel;
1122 	flags = iwl_mvm_mac_ctxt_get_beacon_flags(mvm->fw, rate);
1123 
1124 	/* Enable FILS on PSC channels only */
1125 	rcu_read_lock();
1126 	ctx = rcu_dereference(link_conf->chanctx_conf);
1127 	channel = ieee80211_frequency_to_channel(ctx->def.chan->center_freq);
1128 	WARN_ON(channel == 0);
1129 	if (iwl_mvm_enable_fils(mvm, ctx)) {
1130 		flags |= iwl_fw_lookup_cmd_ver(mvm->fw, BEACON_TEMPLATE_CMD,
1131 					       0) > 10 ?
1132 			IWL_MAC_BEACON_FILS :
1133 			IWL_MAC_BEACON_FILS_V1;
1134 		beacon_cmd.short_ssid =
1135 			cpu_to_le32(~crc32_le(~0, vif->cfg.ssid,
1136 					      vif->cfg.ssid_len));
1137 	}
1138 	rcu_read_unlock();
1139 
1140 	beacon_cmd.flags = cpu_to_le16(flags);
1141 	beacon_cmd.byte_cnt = cpu_to_le16((u16)beacon->len);
1142 
1143 	if (WARN_ON(!mvmvif->link[link_conf->link_id]))
1144 		return -EINVAL;
1145 
1146 	if (iwl_fw_lookup_cmd_ver(mvm->fw, BEACON_TEMPLATE_CMD, 0) > 12)
1147 		beacon_cmd.link_id =
1148 			cpu_to_le32(mvmvif->link[link_conf->link_id]->fw_link_id);
1149 	else
1150 		beacon_cmd.link_id = cpu_to_le32((u32)mvmvif->id);
1151 
1152 	if (vif->type == NL80211_IFTYPE_AP)
1153 		iwl_mvm_mac_ctxt_set_tim(mvm, &beacon_cmd.tim_idx,
1154 					 &beacon_cmd.tim_size,
1155 					 beacon->data, beacon->len);
1156 
1157 	beacon_cmd.csa_offset =
1158 		cpu_to_le32(iwl_mvm_find_ie_offset(beacon->data,
1159 						   WLAN_EID_CHANNEL_SWITCH,
1160 						   beacon->len));
1161 	beacon_cmd.ecsa_offset =
1162 		cpu_to_le32(iwl_mvm_find_ie_offset(beacon->data,
1163 						   WLAN_EID_EXT_CHANSWITCH_ANN,
1164 						   beacon->len));
1165 
1166 	return iwl_mvm_mac_ctxt_send_beacon_cmd(mvm, beacon, &beacon_cmd,
1167 						sizeof(beacon_cmd));
1168 }
1169 
1170 static int iwl_mvm_mac_ctxt_send_beacon(struct iwl_mvm *mvm,
1171 					struct ieee80211_vif *vif,
1172 					struct sk_buff *beacon,
1173 					struct ieee80211_bss_conf *link_conf)
1174 {
1175 	if (WARN_ON(!beacon))
1176 		return -EINVAL;
1177 
1178 	if (IWL_MVM_NON_TRANSMITTING_AP)
1179 		return 0;
1180 
1181 	if (!fw_has_capa(&mvm->fw->ucode_capa,
1182 			 IWL_UCODE_TLV_CAPA_CSA_AND_TBTT_OFFLOAD))
1183 		return iwl_mvm_mac_ctxt_send_beacon_v6(mvm, vif, beacon);
1184 
1185 	if (fw_has_api(&mvm->fw->ucode_capa,
1186 		       IWL_UCODE_TLV_API_NEW_BEACON_TEMPLATE))
1187 		return iwl_mvm_mac_ctxt_send_beacon_v9(mvm, vif, beacon,
1188 						       link_conf);
1189 
1190 	return iwl_mvm_mac_ctxt_send_beacon_v7(mvm, vif, beacon);
1191 }
1192 
1193 /* The beacon template for the AP/GO/IBSS has changed and needs update */
1194 int iwl_mvm_mac_ctxt_beacon_changed(struct iwl_mvm *mvm,
1195 				    struct ieee80211_vif *vif,
1196 				    struct ieee80211_bss_conf *link_conf)
1197 {
1198 	struct sk_buff *beacon;
1199 	int ret;
1200 
1201 	WARN_ON(vif->type != NL80211_IFTYPE_AP &&
1202 		vif->type != NL80211_IFTYPE_ADHOC);
1203 
1204 	beacon = ieee80211_beacon_get_template(mvm->hw, vif, NULL,
1205 					       link_conf->link_id);
1206 	if (!beacon)
1207 		return -ENOMEM;
1208 
1209 #ifdef CONFIG_IWLWIFI_DEBUGFS
1210 	if (mvm->beacon_inject_active) {
1211 		dev_kfree_skb(beacon);
1212 		return -EBUSY;
1213 	}
1214 #endif
1215 
1216 	ret = iwl_mvm_mac_ctxt_send_beacon(mvm, vif, beacon, link_conf);
1217 	dev_kfree_skb(beacon);
1218 	return ret;
1219 }
1220 
1221 struct iwl_mvm_mac_ap_iterator_data {
1222 	struct iwl_mvm *mvm;
1223 	struct ieee80211_vif *vif;
1224 	u32 beacon_device_ts;
1225 	u16 beacon_int;
1226 };
1227 
1228 /* Find the beacon_device_ts and beacon_int for a managed interface */
1229 static void iwl_mvm_mac_ap_iterator(void *_data, u8 *mac,
1230 				    struct ieee80211_vif *vif)
1231 {
1232 	struct iwl_mvm_mac_ap_iterator_data *data = _data;
1233 
1234 	if (vif->type != NL80211_IFTYPE_STATION || !vif->cfg.assoc)
1235 		return;
1236 
1237 	/* Station client has higher priority over P2P client*/
1238 	if (vif->p2p && data->beacon_device_ts)
1239 		return;
1240 
1241 	data->beacon_device_ts = vif->bss_conf.sync_device_ts;
1242 	data->beacon_int = vif->bss_conf.beacon_int;
1243 }
1244 
1245 /*
1246  * Fill the filter flags for mac context of type AP or P2P GO.
1247  */
1248 void iwl_mvm_mac_ctxt_cmd_ap_set_filter_flags(struct iwl_mvm *mvm,
1249 					      struct iwl_mvm_vif *mvmvif,
1250 					      __le32 *filter_flags,
1251 					      int accept_probe_req_flag,
1252 					      int accept_beacon_flag)
1253 {
1254 	/*
1255 	 * in AP mode, pass probe requests and beacons from other APs
1256 	 * (needed for ht protection); when there're no any associated
1257 	 * station don't ask FW to pass beacons to prevent unnecessary
1258 	 * wake-ups.
1259 	 */
1260 	*filter_flags |= cpu_to_le32(accept_probe_req_flag);
1261 	if (mvmvif->ap_assoc_sta_count || !mvm->drop_bcn_ap_mode) {
1262 		*filter_flags |= cpu_to_le32(accept_beacon_flag);
1263 		IWL_DEBUG_HC(mvm, "Asking FW to pass beacons\n");
1264 	} else {
1265 		IWL_DEBUG_HC(mvm, "No need to receive beacons\n");
1266 	}
1267 }
1268 
1269 /*
1270  * Fill the specific data for mac context of type AP of P2P GO
1271  */
1272 static void iwl_mvm_mac_ctxt_cmd_fill_ap(struct iwl_mvm *mvm,
1273 					 struct ieee80211_vif *vif,
1274 					 struct iwl_mac_ctx_cmd *cmd,
1275 					 struct iwl_mac_data_ap *ctxt_ap,
1276 					 bool add)
1277 {
1278 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1279 	struct iwl_mvm_mac_ap_iterator_data data = {
1280 		.mvm = mvm,
1281 		.vif = vif,
1282 		.beacon_device_ts = 0
1283 	};
1284 
1285 	/* in AP mode, the MCAST FIFO takes the EDCA params from VO */
1286 	cmd->ac[IWL_MVM_TX_FIFO_VO].fifos_mask |= BIT(IWL_MVM_TX_FIFO_MCAST);
1287 
1288 	iwl_mvm_mac_ctxt_cmd_ap_set_filter_flags(mvm, mvmvif,
1289 						 &cmd->filter_flags,
1290 						 MAC_FILTER_IN_PROBE_REQUEST,
1291 						 MAC_FILTER_IN_BEACON);
1292 
1293 	ctxt_ap->bi = cpu_to_le32(vif->bss_conf.beacon_int);
1294 	ctxt_ap->dtim_interval = cpu_to_le32(vif->bss_conf.beacon_int *
1295 					     vif->bss_conf.dtim_period);
1296 
1297 	if (!fw_has_api(&mvm->fw->ucode_capa,
1298 			IWL_UCODE_TLV_API_STA_TYPE))
1299 		ctxt_ap->mcast_qid = cpu_to_le32(mvmvif->deflink.cab_queue);
1300 
1301 	/*
1302 	 * Only set the beacon time when the MAC is being added, when we
1303 	 * just modify the MAC then we should keep the time -- the firmware
1304 	 * can otherwise have a "jumping" TBTT.
1305 	 */
1306 	if (add) {
1307 		/*
1308 		 * If there is a station/P2P client interface which is
1309 		 * associated, set the AP's TBTT far enough from the station's
1310 		 * TBTT. Otherwise, set it to the current system time
1311 		 */
1312 		ieee80211_iterate_active_interfaces_atomic(
1313 			mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
1314 			iwl_mvm_mac_ap_iterator, &data);
1315 
1316 		if (data.beacon_device_ts) {
1317 			u32 rand = get_random_u32_inclusive(36, 63);
1318 			mvmvif->ap_beacon_time = data.beacon_device_ts +
1319 				ieee80211_tu_to_usec(data.beacon_int * rand /
1320 						     100);
1321 		} else {
1322 			mvmvif->ap_beacon_time = iwl_mvm_get_systime(mvm);
1323 		}
1324 	}
1325 
1326 	ctxt_ap->beacon_time = cpu_to_le32(mvmvif->ap_beacon_time);
1327 	ctxt_ap->beacon_tsf = 0; /* unused */
1328 
1329 	/* TODO: Assume that the beacon id == mac context id */
1330 	ctxt_ap->beacon_template = cpu_to_le32(mvmvif->id);
1331 }
1332 
1333 static int iwl_mvm_mac_ctxt_cmd_ap(struct iwl_mvm *mvm,
1334 				   struct ieee80211_vif *vif,
1335 				   u32 action)
1336 {
1337 	struct iwl_mac_ctx_cmd cmd = {};
1338 
1339 	WARN_ON(vif->type != NL80211_IFTYPE_AP || vif->p2p);
1340 
1341 	/* Fill the common data for all mac context types */
1342 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, NULL, action);
1343 
1344 	/* Fill the data specific for ap mode */
1345 	iwl_mvm_mac_ctxt_cmd_fill_ap(mvm, vif, &cmd, &cmd.ap,
1346 				     action == FW_CTXT_ACTION_ADD);
1347 
1348 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
1349 }
1350 
1351 static int iwl_mvm_mac_ctxt_cmd_go(struct iwl_mvm *mvm,
1352 				   struct ieee80211_vif *vif,
1353 				   u32 action)
1354 {
1355 	struct iwl_mac_ctx_cmd cmd = {};
1356 	struct ieee80211_p2p_noa_attr *noa = &vif->bss_conf.p2p_noa_attr;
1357 
1358 	WARN_ON(vif->type != NL80211_IFTYPE_AP || !vif->p2p);
1359 
1360 	/* Fill the common data for all mac context types */
1361 	iwl_mvm_mac_ctxt_cmd_common(mvm, vif, &cmd, NULL, action);
1362 
1363 	/* Fill the data specific for GO mode */
1364 	iwl_mvm_mac_ctxt_cmd_fill_ap(mvm, vif, &cmd, &cmd.go.ap,
1365 				     action == FW_CTXT_ACTION_ADD);
1366 
1367 	cmd.go.ctwin = cpu_to_le32(noa->oppps_ctwindow &
1368 					IEEE80211_P2P_OPPPS_CTWINDOW_MASK);
1369 	cmd.go.opp_ps_enabled =
1370 			cpu_to_le32(!!(noa->oppps_ctwindow &
1371 					IEEE80211_P2P_OPPPS_ENABLE_BIT));
1372 
1373 	return iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
1374 }
1375 
1376 static int iwl_mvm_mac_ctx_send(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1377 				u32 action, bool force_assoc_off,
1378 				const u8 *bssid_override)
1379 {
1380 	switch (vif->type) {
1381 	case NL80211_IFTYPE_STATION:
1382 		return iwl_mvm_mac_ctxt_cmd_sta(mvm, vif, action,
1383 						force_assoc_off,
1384 						bssid_override);
1385 	case NL80211_IFTYPE_AP:
1386 		if (!vif->p2p)
1387 			return iwl_mvm_mac_ctxt_cmd_ap(mvm, vif, action);
1388 		else
1389 			return iwl_mvm_mac_ctxt_cmd_go(mvm, vif, action);
1390 	case NL80211_IFTYPE_MONITOR:
1391 		return iwl_mvm_mac_ctxt_cmd_listener(mvm, vif, action);
1392 	case NL80211_IFTYPE_P2P_DEVICE:
1393 		return iwl_mvm_mac_ctxt_cmd_p2p_device(mvm, vif, action);
1394 	case NL80211_IFTYPE_ADHOC:
1395 		return iwl_mvm_mac_ctxt_cmd_ibss(mvm, vif, action);
1396 	default:
1397 		break;
1398 	}
1399 
1400 	return -EOPNOTSUPP;
1401 }
1402 
1403 int iwl_mvm_mac_ctxt_add(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1404 {
1405 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1406 	int ret;
1407 
1408 	if (WARN_ONCE(mvmvif->uploaded, "Adding active MAC %pM/%d\n",
1409 		      vif->addr, ieee80211_vif_type_p2p(vif)))
1410 		return -EIO;
1411 
1412 	ret = iwl_mvm_mac_ctx_send(mvm, vif, FW_CTXT_ACTION_ADD,
1413 				   true, NULL);
1414 	if (ret)
1415 		return ret;
1416 
1417 	/* will only do anything at resume from D3 time */
1418 	iwl_mvm_set_last_nonqos_seq(mvm, vif);
1419 
1420 	mvmvif->uploaded = true;
1421 	return 0;
1422 }
1423 
1424 int iwl_mvm_mac_ctxt_changed(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1425 			     bool force_assoc_off, const u8 *bssid_override)
1426 {
1427 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1428 
1429 	if (WARN_ONCE(!mvmvif->uploaded, "Changing inactive MAC %pM/%d\n",
1430 		      vif->addr, ieee80211_vif_type_p2p(vif)))
1431 		return -EIO;
1432 
1433 	return iwl_mvm_mac_ctx_send(mvm, vif, FW_CTXT_ACTION_MODIFY,
1434 				    force_assoc_off, bssid_override);
1435 }
1436 
1437 int iwl_mvm_mac_ctxt_remove(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1438 {
1439 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1440 	struct iwl_mac_ctx_cmd cmd;
1441 	int ret;
1442 
1443 	if (WARN_ONCE(!mvmvif->uploaded, "Removing inactive MAC %pM/%d\n",
1444 		      vif->addr, ieee80211_vif_type_p2p(vif)))
1445 		return -EIO;
1446 
1447 	memset(&cmd, 0, sizeof(cmd));
1448 
1449 	cmd.id_and_color = cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id,
1450 							   mvmvif->color));
1451 	cmd.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE);
1452 
1453 	ret = iwl_mvm_mac_ctxt_send_cmd(mvm, &cmd);
1454 	if (ret)
1455 		return ret;
1456 
1457 	mvmvif->uploaded = false;
1458 
1459 	if (vif->type == NL80211_IFTYPE_MONITOR) {
1460 		__clear_bit(IEEE80211_HW_RX_INCLUDES_FCS, mvm->hw->flags);
1461 		iwl_mvm_dealloc_snif_sta(mvm);
1462 	}
1463 
1464 	return 0;
1465 }
1466 
1467 static void iwl_mvm_csa_count_down(struct iwl_mvm *mvm,
1468 				   struct ieee80211_vif *csa_vif, u32 gp2,
1469 				   bool tx_success)
1470 {
1471 	struct iwl_mvm_vif *mvmvif =
1472 			iwl_mvm_vif_from_mac80211(csa_vif);
1473 
1474 	/* Don't start to countdown from a failed beacon */
1475 	if (!tx_success && !mvmvif->csa_countdown)
1476 		return;
1477 
1478 	mvmvif->csa_countdown = true;
1479 
1480 	if (!ieee80211_beacon_cntdwn_is_complete(csa_vif, 0)) {
1481 		int c = ieee80211_beacon_update_cntdwn(csa_vif, 0);
1482 
1483 		iwl_mvm_mac_ctxt_beacon_changed(mvm, csa_vif,
1484 						&csa_vif->bss_conf);
1485 		if (csa_vif->p2p &&
1486 		    !iwl_mvm_te_scheduled(&mvmvif->time_event_data) && gp2 &&
1487 		    tx_success) {
1488 			u32 rel_time = (c + 1) *
1489 				       csa_vif->bss_conf.beacon_int -
1490 				       IWL_MVM_CHANNEL_SWITCH_TIME_GO;
1491 			u32 apply_time = gp2 + rel_time * 1024;
1492 
1493 			iwl_mvm_schedule_csa_period(mvm, csa_vif,
1494 					 IWL_MVM_CHANNEL_SWITCH_TIME_GO -
1495 					 IWL_MVM_CHANNEL_SWITCH_MARGIN,
1496 					 apply_time);
1497 		}
1498 	} else if (!iwl_mvm_te_scheduled(&mvmvif->time_event_data)) {
1499 		/* we don't have CSA NoA scheduled yet, switch now */
1500 		ieee80211_csa_finish(csa_vif, 0);
1501 		RCU_INIT_POINTER(mvm->csa_vif, NULL);
1502 	}
1503 }
1504 
1505 void iwl_mvm_rx_beacon_notif(struct iwl_mvm *mvm,
1506 			     struct iwl_rx_cmd_buffer *rxb)
1507 {
1508 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1509 	unsigned int pkt_len = iwl_rx_packet_payload_len(pkt);
1510 	struct iwl_extended_beacon_notif *beacon = (void *)pkt->data;
1511 	struct iwl_extended_beacon_notif_v5 *beacon_v5 = (void *)pkt->data;
1512 	struct ieee80211_vif *csa_vif;
1513 	struct ieee80211_vif *tx_blocked_vif;
1514 	struct agg_tx_status *agg_status;
1515 	u16 status;
1516 
1517 	lockdep_assert_held(&mvm->mutex);
1518 
1519 	mvm->ap_last_beacon_gp2 = le32_to_cpu(beacon->gp2);
1520 
1521 	if (!iwl_mvm_is_short_beacon_notif_supported(mvm)) {
1522 		struct iwl_mvm_tx_resp *beacon_notify_hdr =
1523 			&beacon_v5->beacon_notify_hdr;
1524 
1525 		if (unlikely(pkt_len < sizeof(*beacon_v5)))
1526 			return;
1527 
1528 		mvm->ibss_manager = beacon_v5->ibss_mgr_status != 0;
1529 		agg_status = iwl_mvm_get_agg_status(mvm, beacon_notify_hdr);
1530 		status = le16_to_cpu(agg_status->status) & TX_STATUS_MSK;
1531 		IWL_DEBUG_RX(mvm,
1532 			     "beacon status %#x retries:%d tsf:0x%016llX gp2:0x%X rate:%d\n",
1533 			     status, beacon_notify_hdr->failure_frame,
1534 			     le64_to_cpu(beacon->tsf),
1535 			     mvm->ap_last_beacon_gp2,
1536 			     le32_to_cpu(beacon_notify_hdr->initial_rate));
1537 	} else {
1538 		if (unlikely(pkt_len < sizeof(*beacon)))
1539 			return;
1540 
1541 		mvm->ibss_manager = beacon->ibss_mgr_status != 0;
1542 		status = le32_to_cpu(beacon->status) & TX_STATUS_MSK;
1543 		IWL_DEBUG_RX(mvm,
1544 			     "beacon status %#x tsf:0x%016llX gp2:0x%X\n",
1545 			     status, le64_to_cpu(beacon->tsf),
1546 			     mvm->ap_last_beacon_gp2);
1547 	}
1548 
1549 	csa_vif = rcu_dereference_protected(mvm->csa_vif,
1550 					    lockdep_is_held(&mvm->mutex));
1551 	if (unlikely(csa_vif && csa_vif->bss_conf.csa_active))
1552 		iwl_mvm_csa_count_down(mvm, csa_vif, mvm->ap_last_beacon_gp2,
1553 				       (status == TX_STATUS_SUCCESS));
1554 
1555 	tx_blocked_vif = rcu_dereference_protected(mvm->csa_tx_blocked_vif,
1556 						lockdep_is_held(&mvm->mutex));
1557 	if (unlikely(tx_blocked_vif)) {
1558 		struct iwl_mvm_vif *mvmvif =
1559 			iwl_mvm_vif_from_mac80211(tx_blocked_vif);
1560 
1561 		/*
1562 		 * The channel switch is started and we have blocked the
1563 		 * stations. If this is the first beacon (the timeout wasn't
1564 		 * set), set the unblock timeout, otherwise countdown
1565 		 */
1566 		if (!mvm->csa_tx_block_bcn_timeout)
1567 			mvm->csa_tx_block_bcn_timeout =
1568 				IWL_MVM_CS_UNBLOCK_TX_TIMEOUT;
1569 		else
1570 			mvm->csa_tx_block_bcn_timeout--;
1571 
1572 		/* Check if the timeout is expired, and unblock tx */
1573 		if (mvm->csa_tx_block_bcn_timeout == 0) {
1574 			iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, false);
1575 			RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
1576 		}
1577 	}
1578 }
1579 
1580 void iwl_mvm_rx_missed_beacons_notif(struct iwl_mvm *mvm,
1581 				     struct iwl_rx_cmd_buffer *rxb)
1582 {
1583 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1584 	struct iwl_missed_beacons_notif *mb = (void *)pkt->data;
1585 	struct iwl_fw_dbg_trigger_missed_bcon *bcon_trig;
1586 	struct iwl_fw_dbg_trigger_tlv *trigger;
1587 	u32 stop_trig_missed_bcon, stop_trig_missed_bcon_since_rx;
1588 	u32 rx_missed_bcon, rx_missed_bcon_since_rx;
1589 	struct ieee80211_vif *vif;
1590 	/* Id can be mac/link id depending on the notification version */
1591 	u32 id = le32_to_cpu(mb->link_id);
1592 	union iwl_dbg_tlv_tp_data tp_data = { .fw_pkt = pkt };
1593 	u32 mac_type;
1594 	u8 notif_ver = iwl_fw_lookup_notif_ver(mvm->fw, LEGACY_GROUP,
1595 					       MISSED_BEACONS_NOTIFICATION,
1596 					       0);
1597 
1598 	rcu_read_lock();
1599 
1600 	/* before version four the ID in the notification refers to mac ID */
1601 	if (notif_ver < 4) {
1602 		vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, true);
1603 	} else {
1604 		struct ieee80211_bss_conf *bss_conf =
1605 			iwl_mvm_rcu_fw_link_id_to_link_conf(mvm, id, true);
1606 
1607 		if (!bss_conf)
1608 			goto out;
1609 
1610 		vif = bss_conf->vif;
1611 	}
1612 
1613 	IWL_DEBUG_INFO(mvm,
1614 		       "missed bcn %s_id=%u, consecutive=%u (%u, %u, %u)\n",
1615 		       notif_ver < 4 ? "mac" : "link",
1616 		       id,
1617 		       le32_to_cpu(mb->consec_missed_beacons),
1618 		       le32_to_cpu(mb->consec_missed_beacons_since_last_rx),
1619 		       le32_to_cpu(mb->num_recvd_beacons),
1620 		       le32_to_cpu(mb->num_expected_beacons));
1621 
1622 	if (!vif)
1623 		goto out;
1624 
1625 	mac_type = iwl_mvm_get_mac_type(vif);
1626 
1627 	IWL_DEBUG_INFO(mvm, "missed beacon mac_type=%u,\n", mac_type);
1628 
1629 	mvm->trans->dbg.dump_file_name_ext_valid = true;
1630 	snprintf(mvm->trans->dbg.dump_file_name_ext, IWL_FW_INI_MAX_NAME,
1631 		 "MacId_%d_MacType_%d", id, mac_type);
1632 
1633 	rx_missed_bcon = le32_to_cpu(mb->consec_missed_beacons);
1634 	rx_missed_bcon_since_rx =
1635 		le32_to_cpu(mb->consec_missed_beacons_since_last_rx);
1636 	/*
1637 	 * TODO: the threshold should be adjusted based on latency conditions,
1638 	 * and/or in case of a CS flow on one of the other AP vifs.
1639 	 */
1640 	if (rx_missed_bcon >= IWL_MVM_MISSED_BEACONS_THRESHOLD_LONG) {
1641 		if (rx_missed_bcon_since_rx >= IWL_MVM_MISSED_BEACONS_SINCE_RX_THOLD) {
1642 			iwl_mvm_connection_loss(mvm, vif, "missed beacons");
1643 		} else {
1644 			IWL_WARN(mvm,
1645 				 "missed beacons exceeds threshold, but receiving data. Stay connected, Expect bugs.\n");
1646 			IWL_WARN(mvm,
1647 				 "missed_beacons:%d, missed_beacons_since_rx:%d\n",
1648 				 rx_missed_bcon, rx_missed_bcon_since_rx);
1649 		}
1650 	} else if (rx_missed_bcon_since_rx > IWL_MVM_MISSED_BEACONS_THRESHOLD) {
1651 		if (!iwl_mvm_has_new_tx_api(mvm))
1652 			ieee80211_beacon_loss(vif);
1653 		else
1654 			ieee80211_cqm_beacon_loss_notify(vif, GFP_ATOMIC);
1655 	}
1656 
1657 	iwl_dbg_tlv_time_point(&mvm->fwrt,
1658 			       IWL_FW_INI_TIME_POINT_MISSED_BEACONS, &tp_data);
1659 
1660 	trigger = iwl_fw_dbg_trigger_on(&mvm->fwrt, ieee80211_vif_to_wdev(vif),
1661 					FW_DBG_TRIGGER_MISSED_BEACONS);
1662 	if (!trigger)
1663 		goto out;
1664 
1665 	bcon_trig = (void *)trigger->data;
1666 	stop_trig_missed_bcon = le32_to_cpu(bcon_trig->stop_consec_missed_bcon);
1667 	stop_trig_missed_bcon_since_rx =
1668 		le32_to_cpu(bcon_trig->stop_consec_missed_bcon_since_rx);
1669 
1670 	/* TODO: implement start trigger */
1671 
1672 	if (rx_missed_bcon_since_rx >= stop_trig_missed_bcon_since_rx ||
1673 	    rx_missed_bcon >= stop_trig_missed_bcon)
1674 		iwl_fw_dbg_collect_trig(&mvm->fwrt, trigger, NULL);
1675 
1676 out:
1677 	rcu_read_unlock();
1678 }
1679 
1680 void iwl_mvm_rx_stored_beacon_notif(struct iwl_mvm *mvm,
1681 				    struct iwl_rx_cmd_buffer *rxb)
1682 {
1683 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1684 	unsigned int pkt_len = iwl_rx_packet_payload_len(pkt);
1685 	struct iwl_stored_beacon_notif_common *sb = (void *)pkt->data;
1686 	struct ieee80211_rx_status rx_status;
1687 	struct sk_buff *skb;
1688 	u8 *data;
1689 	u32 size = le32_to_cpu(sb->byte_count);
1690 	int ver = iwl_fw_lookup_cmd_ver(mvm->fw,
1691 					WIDE_ID(PROT_OFFLOAD_GROUP, STORED_BEACON_NTF),
1692 					0);
1693 
1694 	if (size == 0)
1695 		return;
1696 
1697 	/* handle per-version differences */
1698 	if (ver <= 2) {
1699 		struct iwl_stored_beacon_notif_v2 *sb_v2 = (void *)pkt->data;
1700 
1701 		if (pkt_len < struct_size(sb_v2, data, size))
1702 			return;
1703 
1704 		data = sb_v2->data;
1705 	} else {
1706 		struct iwl_stored_beacon_notif_v3 *sb_v3 = (void *)pkt->data;
1707 
1708 		if (pkt_len < struct_size(sb_v3, data, size))
1709 			return;
1710 
1711 		data = sb_v3->data;
1712 	}
1713 
1714 	skb = alloc_skb(size, GFP_ATOMIC);
1715 	if (!skb) {
1716 		IWL_ERR(mvm, "alloc_skb failed\n");
1717 		return;
1718 	}
1719 
1720 	/* update rx_status according to the notification's metadata */
1721 	memset(&rx_status, 0, sizeof(rx_status));
1722 	rx_status.mactime = le64_to_cpu(sb->tsf);
1723 	/* TSF as indicated by the firmware  is at INA time */
1724 	rx_status.flag |= RX_FLAG_MACTIME_PLCP_START;
1725 	rx_status.device_timestamp = le32_to_cpu(sb->system_time);
1726 	rx_status.band =
1727 		(sb->band & cpu_to_le16(RX_RES_PHY_FLAGS_BAND_24)) ?
1728 				NL80211_BAND_2GHZ : NL80211_BAND_5GHZ;
1729 	rx_status.freq =
1730 		ieee80211_channel_to_frequency(le16_to_cpu(sb->channel),
1731 					       rx_status.band);
1732 
1733 	/* copy the data */
1734 	skb_put_data(skb, data, size);
1735 	memcpy(IEEE80211_SKB_RXCB(skb), &rx_status, sizeof(rx_status));
1736 
1737 	/* pass it as regular rx to mac80211 */
1738 	ieee80211_rx_napi(mvm->hw, NULL, skb, NULL);
1739 }
1740 
1741 void iwl_mvm_probe_resp_data_notif(struct iwl_mvm *mvm,
1742 				   struct iwl_rx_cmd_buffer *rxb)
1743 {
1744 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1745 	struct iwl_probe_resp_data_notif *notif = (void *)pkt->data;
1746 	struct iwl_probe_resp_data *old_data, *new_data;
1747 	u32 id = le32_to_cpu(notif->mac_id);
1748 	struct ieee80211_vif *vif;
1749 	struct iwl_mvm_vif *mvmvif;
1750 
1751 	IWL_DEBUG_INFO(mvm, "Probe response data notif: noa %d, csa %d\n",
1752 		       notif->noa_active, notif->csa_counter);
1753 
1754 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, false);
1755 	if (!vif)
1756 		return;
1757 
1758 	mvmvif = iwl_mvm_vif_from_mac80211(vif);
1759 
1760 	new_data = kzalloc(sizeof(*new_data), GFP_KERNEL);
1761 	if (!new_data)
1762 		return;
1763 
1764 	memcpy(&new_data->notif, notif, sizeof(new_data->notif));
1765 
1766 	/* noa_attr contains 1 reserved byte, need to substruct it */
1767 	new_data->noa_len = sizeof(struct ieee80211_vendor_ie) +
1768 			    sizeof(new_data->notif.noa_attr) - 1;
1769 
1770 	/*
1771 	 * If it's a one time NoA, only one descriptor is needed,
1772 	 * adjust the length according to len_low.
1773 	 */
1774 	if (new_data->notif.noa_attr.len_low ==
1775 	    sizeof(struct ieee80211_p2p_noa_desc) + 2)
1776 		new_data->noa_len -= sizeof(struct ieee80211_p2p_noa_desc);
1777 
1778 	old_data = rcu_dereference_protected(mvmvif->deflink.probe_resp_data,
1779 					     lockdep_is_held(&mvmvif->mvm->mutex));
1780 	rcu_assign_pointer(mvmvif->deflink.probe_resp_data, new_data);
1781 
1782 	if (old_data)
1783 		kfree_rcu(old_data, rcu_head);
1784 
1785 	if (notif->csa_counter != IWL_PROBE_RESP_DATA_NO_CSA &&
1786 	    notif->csa_counter >= 1)
1787 		ieee80211_beacon_set_cntdwn(vif, notif->csa_counter);
1788 }
1789 
1790 void iwl_mvm_channel_switch_start_notif(struct iwl_mvm *mvm,
1791 					struct iwl_rx_cmd_buffer *rxb)
1792 {
1793 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1794 	struct ieee80211_vif *csa_vif, *vif;
1795 	struct iwl_mvm_vif *mvmvif, *csa_mvmvif;
1796 	u32 id_n_color, csa_id;
1797 	/* save mac_id or link_id to use later to cancel csa if needed */
1798 	u32 id;
1799 	u32 mac_link_id = 0;
1800 	u8 notif_ver = iwl_fw_lookup_notif_ver(mvm->fw, MAC_CONF_GROUP,
1801 					       CHANNEL_SWITCH_START_NOTIF, 0);
1802 	bool csa_active;
1803 
1804 	rcu_read_lock();
1805 
1806 	if (notif_ver < 3) {
1807 		struct iwl_channel_switch_start_notif_v1 *notif = (void *)pkt->data;
1808 		u32 mac_id;
1809 
1810 		id_n_color = le32_to_cpu(notif->id_and_color);
1811 		mac_id = id_n_color & FW_CTXT_ID_MSK;
1812 
1813 		vif = iwl_mvm_rcu_dereference_vif_id(mvm, mac_id, true);
1814 		if (!vif)
1815 			goto out_unlock;
1816 
1817 		id = mac_id;
1818 		csa_active = vif->bss_conf.csa_active;
1819 	} else {
1820 		struct iwl_channel_switch_start_notif *notif = (void *)pkt->data;
1821 		u32 link_id = le32_to_cpu(notif->link_id);
1822 		struct ieee80211_bss_conf *bss_conf =
1823 			iwl_mvm_rcu_fw_link_id_to_link_conf(mvm, link_id, true);
1824 
1825 		if (!bss_conf)
1826 			goto out_unlock;
1827 
1828 		id = link_id;
1829 		mac_link_id = bss_conf->link_id;
1830 		vif = bss_conf->vif;
1831 		csa_active = bss_conf->csa_active;
1832 	}
1833 
1834 	mvmvif = iwl_mvm_vif_from_mac80211(vif);
1835 	if (notif_ver >= 3)
1836 		id_n_color = FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color);
1837 
1838 	switch (vif->type) {
1839 	case NL80211_IFTYPE_AP:
1840 		csa_vif = rcu_dereference(mvm->csa_vif);
1841 		if (WARN_ON(!csa_vif || !csa_vif->bss_conf.csa_active ||
1842 			    csa_vif != vif))
1843 			goto out_unlock;
1844 
1845 		csa_mvmvif = iwl_mvm_vif_from_mac80211(csa_vif);
1846 		csa_id = FW_CMD_ID_AND_COLOR(csa_mvmvif->id, csa_mvmvif->color);
1847 		if (WARN(csa_id != id_n_color,
1848 			 "channel switch noa notification on unexpected vif (csa_vif=%d, notif=%d)",
1849 			 csa_id, id_n_color))
1850 			goto out_unlock;
1851 
1852 		IWL_DEBUG_INFO(mvm, "Channel Switch Started Notification\n");
1853 
1854 		schedule_delayed_work(&mvm->cs_tx_unblock_dwork,
1855 				      msecs_to_jiffies(IWL_MVM_CS_UNBLOCK_TX_TIMEOUT *
1856 						       csa_vif->bss_conf.beacon_int));
1857 
1858 		ieee80211_csa_finish(csa_vif, 0);
1859 
1860 		rcu_read_unlock();
1861 
1862 		RCU_INIT_POINTER(mvm->csa_vif, NULL);
1863 		return;
1864 	case NL80211_IFTYPE_STATION:
1865 		/*
1866 		 * if we don't know about an ongoing channel switch,
1867 		 * make sure FW cancels it
1868 		 */
1869 		if (iwl_fw_lookup_notif_ver(mvm->fw, MAC_CONF_GROUP,
1870 					    CHANNEL_SWITCH_ERROR_NOTIF,
1871 					    0) && !csa_active) {
1872 			IWL_DEBUG_INFO(mvm, "Channel Switch was canceled\n");
1873 			iwl_mvm_cancel_channel_switch(mvm, vif, id);
1874 			break;
1875 		}
1876 
1877 		iwl_mvm_csa_client_absent(mvm, vif);
1878 		cancel_delayed_work(&mvmvif->csa_work);
1879 		ieee80211_chswitch_done(vif, true, mac_link_id);
1880 		break;
1881 	default:
1882 		/* should never happen */
1883 		WARN_ON_ONCE(1);
1884 		break;
1885 	}
1886 out_unlock:
1887 	rcu_read_unlock();
1888 }
1889 
1890 void iwl_mvm_channel_switch_error_notif(struct iwl_mvm *mvm,
1891 					struct iwl_rx_cmd_buffer *rxb)
1892 {
1893 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1894 	struct iwl_channel_switch_error_notif *notif = (void *)pkt->data;
1895 	struct ieee80211_vif *vif;
1896 	u32 id = le32_to_cpu(notif->link_id);
1897 	u32 csa_err_mask = le32_to_cpu(notif->csa_err_mask);
1898 
1899 	rcu_read_lock();
1900 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, true);
1901 	if (!vif) {
1902 		rcu_read_unlock();
1903 		return;
1904 	}
1905 
1906 	IWL_DEBUG_INFO(mvm, "FW reports CSA error: id=%u, csa_err_mask=%u\n",
1907 		       id, csa_err_mask);
1908 	if (csa_err_mask & (CS_ERR_COUNT_ERROR |
1909 			    CS_ERR_LONG_DELAY_AFTER_CS |
1910 			    CS_ERR_TX_BLOCK_TIMER_EXPIRED))
1911 		ieee80211_channel_switch_disconnect(vif, true);
1912 	rcu_read_unlock();
1913 }
1914 
1915 void iwl_mvm_rx_missed_vap_notif(struct iwl_mvm *mvm,
1916 				 struct iwl_rx_cmd_buffer *rxb)
1917 {
1918 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1919 	struct iwl_missed_vap_notif *mb = (void *)pkt->data;
1920 	struct ieee80211_vif *vif;
1921 	u32 id = le32_to_cpu(mb->mac_id);
1922 
1923 	IWL_DEBUG_INFO(mvm,
1924 		       "missed_vap notify mac_id=%u, num_beacon_intervals_elapsed=%u, profile_periodicity=%u\n",
1925 		       le32_to_cpu(mb->mac_id),
1926 		       mb->num_beacon_intervals_elapsed,
1927 		       mb->profile_periodicity);
1928 
1929 	rcu_read_lock();
1930 
1931 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, true);
1932 	if (vif)
1933 		iwl_mvm_connection_loss(mvm, vif, "missed vap beacon");
1934 
1935 	rcu_read_unlock();
1936 }
1937