xref: /linux/drivers/net/wireless/intel/iwlwifi/mvm/rxmq.c (revision af2d6148d2a159e1a0862bce5a2c88c1618a2b27)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /*
3  * Copyright (C) 2012-2014, 2018-2025 Intel Corporation
4  * Copyright (C) 2013-2015 Intel Mobile Communications GmbH
5  * Copyright (C) 2015-2017 Intel Deutschland GmbH
6  */
7 #include <linux/etherdevice.h>
8 #include <linux/skbuff.h>
9 #include "iwl-trans.h"
10 #include "mvm.h"
11 #include "fw-api.h"
12 #include "time-sync.h"
13 
14 static inline int iwl_mvm_check_pn(struct iwl_mvm *mvm, struct sk_buff *skb,
15 				   int queue, struct ieee80211_sta *sta)
16 {
17 	struct iwl_mvm_sta *mvmsta;
18 	struct ieee80211_hdr *hdr = (void *)skb_mac_header(skb);
19 	struct ieee80211_rx_status *stats = IEEE80211_SKB_RXCB(skb);
20 	struct iwl_mvm_key_pn *ptk_pn;
21 	int res;
22 	u8 tid, keyidx;
23 	u8 pn[IEEE80211_CCMP_PN_LEN];
24 	u8 *extiv;
25 
26 	/* do PN checking */
27 
28 	/* multicast and non-data only arrives on default queue */
29 	if (!ieee80211_is_data(hdr->frame_control) ||
30 	    is_multicast_ether_addr(hdr->addr1))
31 		return 0;
32 
33 	/* do not check PN for open AP */
34 	if (!(stats->flag & RX_FLAG_DECRYPTED))
35 		return 0;
36 
37 	/*
38 	 * avoid checking for default queue - we don't want to replicate
39 	 * all the logic that's necessary for checking the PN on fragmented
40 	 * frames, leave that to mac80211
41 	 */
42 	if (queue == 0)
43 		return 0;
44 
45 	/* if we are here - this for sure is either CCMP or GCMP */
46 	if (IS_ERR_OR_NULL(sta)) {
47 		IWL_DEBUG_DROP(mvm,
48 			       "expected hw-decrypted unicast frame for station\n");
49 		return -1;
50 	}
51 
52 	mvmsta = iwl_mvm_sta_from_mac80211(sta);
53 
54 	extiv = (u8 *)hdr + ieee80211_hdrlen(hdr->frame_control);
55 	keyidx = extiv[3] >> 6;
56 
57 	ptk_pn = rcu_dereference(mvmsta->ptk_pn[keyidx]);
58 	if (!ptk_pn)
59 		return -1;
60 
61 	if (ieee80211_is_data_qos(hdr->frame_control))
62 		tid = ieee80211_get_tid(hdr);
63 	else
64 		tid = 0;
65 
66 	/* we don't use HCCA/802.11 QoS TSPECs, so drop such frames */
67 	if (tid >= IWL_MAX_TID_COUNT)
68 		return -1;
69 
70 	/* load pn */
71 	pn[0] = extiv[7];
72 	pn[1] = extiv[6];
73 	pn[2] = extiv[5];
74 	pn[3] = extiv[4];
75 	pn[4] = extiv[1];
76 	pn[5] = extiv[0];
77 
78 	res = memcmp(pn, ptk_pn->q[queue].pn[tid], IEEE80211_CCMP_PN_LEN);
79 	if (res < 0)
80 		return -1;
81 	if (!res && !(stats->flag & RX_FLAG_ALLOW_SAME_PN))
82 		return -1;
83 
84 	memcpy(ptk_pn->q[queue].pn[tid], pn, IEEE80211_CCMP_PN_LEN);
85 	stats->flag |= RX_FLAG_PN_VALIDATED;
86 
87 	return 0;
88 }
89 
90 /* iwl_mvm_create_skb Adds the rxb to a new skb */
91 static int iwl_mvm_create_skb(struct iwl_mvm *mvm, struct sk_buff *skb,
92 			      struct ieee80211_hdr *hdr, u16 len, u8 crypt_len,
93 			      struct iwl_rx_cmd_buffer *rxb)
94 {
95 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
96 	struct iwl_rx_mpdu_desc *desc = (void *)pkt->data;
97 	unsigned int headlen, fraglen, pad_len = 0;
98 	unsigned int hdrlen = ieee80211_hdrlen(hdr->frame_control);
99 	u8 mic_crc_len = u8_get_bits(desc->mac_flags1,
100 				     IWL_RX_MPDU_MFLG1_MIC_CRC_LEN_MASK) << 1;
101 
102 	if (desc->mac_flags2 & IWL_RX_MPDU_MFLG2_PAD) {
103 		len -= 2;
104 		pad_len = 2;
105 	}
106 
107 	/*
108 	 * For non monitor interface strip the bytes the RADA might not have
109 	 * removed (it might be disabled, e.g. for mgmt frames). As a monitor
110 	 * interface cannot exist with other interfaces, this removal is safe
111 	 * and sufficient, in monitor mode there's no decryption being done.
112 	 */
113 	if (len > mic_crc_len && !ieee80211_hw_check(mvm->hw, RX_INCLUDES_FCS))
114 		len -= mic_crc_len;
115 
116 	/* If frame is small enough to fit in skb->head, pull it completely.
117 	 * If not, only pull ieee80211_hdr (including crypto if present, and
118 	 * an additional 8 bytes for SNAP/ethertype, see below) so that
119 	 * splice() or TCP coalesce are more efficient.
120 	 *
121 	 * Since, in addition, ieee80211_data_to_8023() always pull in at
122 	 * least 8 bytes (possibly more for mesh) we can do the same here
123 	 * to save the cost of doing it later. That still doesn't pull in
124 	 * the actual IP header since the typical case has a SNAP header.
125 	 * If the latter changes (there are efforts in the standards group
126 	 * to do so) we should revisit this and ieee80211_data_to_8023().
127 	 */
128 	headlen = (len <= skb_tailroom(skb)) ? len :
129 					       hdrlen + crypt_len + 8;
130 
131 	/* The firmware may align the packet to DWORD.
132 	 * The padding is inserted after the IV.
133 	 * After copying the header + IV skip the padding if
134 	 * present before copying packet data.
135 	 */
136 	hdrlen += crypt_len;
137 
138 	if (unlikely(headlen < hdrlen))
139 		return -EINVAL;
140 
141 	/* Since data doesn't move data while putting data on skb and that is
142 	 * the only way we use, data + len is the next place that hdr would be put
143 	 */
144 	skb_set_mac_header(skb, skb->len);
145 	skb_put_data(skb, hdr, hdrlen);
146 	skb_put_data(skb, (u8 *)hdr + hdrlen + pad_len, headlen - hdrlen);
147 
148 	/*
149 	 * If we did CHECKSUM_COMPLETE, the hardware only does it right for
150 	 * certain cases and starts the checksum after the SNAP. Check if
151 	 * this is the case - it's easier to just bail out to CHECKSUM_NONE
152 	 * in the cases the hardware didn't handle, since it's rare to see
153 	 * such packets, even though the hardware did calculate the checksum
154 	 * in this case, just starting after the MAC header instead.
155 	 *
156 	 * Starting from Bz hardware, it calculates starting directly after
157 	 * the MAC header, so that matches mac80211's expectation.
158 	 */
159 	if (skb->ip_summed == CHECKSUM_COMPLETE) {
160 		struct {
161 			u8 hdr[6];
162 			__be16 type;
163 		} __packed *shdr = (void *)((u8 *)hdr + hdrlen + pad_len);
164 
165 		if (unlikely(headlen - hdrlen < sizeof(*shdr) ||
166 			     !ether_addr_equal(shdr->hdr, rfc1042_header) ||
167 			     (shdr->type != htons(ETH_P_IP) &&
168 			      shdr->type != htons(ETH_P_ARP) &&
169 			      shdr->type != htons(ETH_P_IPV6) &&
170 			      shdr->type != htons(ETH_P_8021Q) &&
171 			      shdr->type != htons(ETH_P_PAE) &&
172 			      shdr->type != htons(ETH_P_TDLS))))
173 			skb->ip_summed = CHECKSUM_NONE;
174 		else if (mvm->trans->mac_cfg->device_family < IWL_DEVICE_FAMILY_BZ)
175 			/* mac80211 assumes full CSUM including SNAP header */
176 			skb_postpush_rcsum(skb, shdr, sizeof(*shdr));
177 	}
178 
179 	fraglen = len - headlen;
180 
181 	if (fraglen) {
182 		int offset = (u8 *)hdr + headlen + pad_len -
183 			     (u8 *)rxb_addr(rxb) + rxb_offset(rxb);
184 
185 		skb_add_rx_frag(skb, 0, rxb_steal_page(rxb), offset,
186 				fraglen, rxb->truesize);
187 	}
188 
189 	return 0;
190 }
191 
192 /* put a TLV on the skb and return data pointer
193  *
194  * Also pad to 4 the len and zero out all data part
195  */
196 static void *
197 iwl_mvm_radiotap_put_tlv(struct sk_buff *skb, u16 type, u16 len)
198 {
199 	struct ieee80211_radiotap_tlv *tlv;
200 
201 	tlv = skb_put(skb, sizeof(*tlv));
202 	tlv->type = cpu_to_le16(type);
203 	tlv->len = cpu_to_le16(len);
204 	return skb_put_zero(skb, ALIGN(len, 4));
205 }
206 
207 static void iwl_mvm_add_rtap_sniffer_config(struct iwl_mvm *mvm,
208 					    struct sk_buff *skb)
209 {
210 	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
211 	struct ieee80211_radiotap_vendor_content *radiotap;
212 	const u16 vendor_data_len = sizeof(mvm->cur_aid);
213 
214 	if (!mvm->cur_aid)
215 		return;
216 
217 	radiotap = iwl_mvm_radiotap_put_tlv(skb,
218 					    IEEE80211_RADIOTAP_VENDOR_NAMESPACE,
219 					    sizeof(*radiotap) + vendor_data_len);
220 
221 	/* Intel OUI */
222 	radiotap->oui[0] = 0xf6;
223 	radiotap->oui[1] = 0x54;
224 	radiotap->oui[2] = 0x25;
225 	/* radiotap sniffer config sub-namespace */
226 	radiotap->oui_subtype = 1;
227 	radiotap->vendor_type = 0;
228 
229 	/* fill the data now */
230 	memcpy(radiotap->data, &mvm->cur_aid, sizeof(mvm->cur_aid));
231 
232 	rx_status->flag |= RX_FLAG_RADIOTAP_TLV_AT_END;
233 }
234 
235 /* iwl_mvm_pass_packet_to_mac80211 - passes the packet for mac80211 */
236 static void iwl_mvm_pass_packet_to_mac80211(struct iwl_mvm *mvm,
237 					    struct napi_struct *napi,
238 					    struct sk_buff *skb, int queue,
239 					    struct ieee80211_sta *sta)
240 {
241 	if (unlikely(iwl_mvm_check_pn(mvm, skb, queue, sta))) {
242 		kfree_skb(skb);
243 		return;
244 	}
245 
246 	ieee80211_rx_napi(mvm->hw, sta, skb, napi);
247 }
248 
249 static bool iwl_mvm_used_average_energy(struct iwl_mvm *mvm,
250 					struct iwl_rx_mpdu_desc *desc,
251 					struct ieee80211_hdr *hdr,
252 					struct ieee80211_rx_status *rx_status)
253 {
254 	struct iwl_mvm_vif *mvm_vif;
255 	struct ieee80211_vif *vif;
256 	u32 id;
257 
258 	if (unlikely(!hdr || !desc))
259 		return false;
260 
261 	if (likely(!ieee80211_is_beacon(hdr->frame_control)))
262 		return false;
263 
264 	/* for the link conf lookup */
265 	guard(rcu)();
266 
267 	/* MAC or link ID depending on FW, but driver has them equal */
268 	id = u8_get_bits(desc->mac_phy_band,
269 			 IWL_RX_MPDU_MAC_PHY_BAND_MAC_MASK);
270 
271 	/* >= means AUX MAC/link ID, no energy correction needed then */
272 	if (id >= ARRAY_SIZE(mvm->vif_id_to_mac))
273 		return false;
274 
275 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, true);
276 	if (!vif)
277 		return false;
278 
279 	mvm_vif = iwl_mvm_vif_from_mac80211(vif);
280 
281 	/*
282 	 * If we know the MAC by MAC or link ID then the frame was
283 	 * received for the link, so by filtering it means it was
284 	 * from the AP the link is connected to.
285 	 */
286 
287 	/* skip also in case we don't have it (yet) */
288 	if (!mvm_vif->deflink.average_beacon_energy)
289 		return false;
290 
291 	IWL_DEBUG_STATS(mvm, "energy override by average %d\n",
292 			mvm_vif->deflink.average_beacon_energy);
293 	rx_status->signal = -mvm_vif->deflink.average_beacon_energy;
294 	return true;
295 }
296 
297 static void iwl_mvm_get_signal_strength(struct iwl_mvm *mvm,
298 					struct iwl_rx_mpdu_desc *desc,
299 					struct ieee80211_hdr *hdr,
300 					struct ieee80211_rx_status *rx_status,
301 					u32 rate_n_flags, int energy_a,
302 					int energy_b)
303 {
304 	int max_energy;
305 
306 	energy_a = energy_a ? -energy_a : S8_MIN;
307 	energy_b = energy_b ? -energy_b : S8_MIN;
308 	max_energy = max(energy_a, energy_b);
309 
310 	IWL_DEBUG_STATS(mvm, "energy In A %d B %d, and max %d\n",
311 			energy_a, energy_b, max_energy);
312 
313 	if (iwl_mvm_used_average_energy(mvm, desc, hdr, rx_status))
314 		return;
315 
316 	rx_status->signal = max_energy;
317 	rx_status->chains = u32_get_bits(rate_n_flags, RATE_MCS_ANT_AB_MSK);
318 	rx_status->chain_signal[0] = energy_a;
319 	rx_status->chain_signal[1] = energy_b;
320 }
321 
322 static int iwl_mvm_rx_mgmt_prot(struct ieee80211_sta *sta,
323 				struct ieee80211_hdr *hdr,
324 				struct iwl_rx_mpdu_desc *desc,
325 				u32 status,
326 				struct ieee80211_rx_status *stats)
327 {
328 	struct wireless_dev *wdev;
329 	struct iwl_mvm_sta *mvmsta;
330 	struct iwl_mvm_vif *mvmvif;
331 	u8 keyid;
332 	struct ieee80211_key_conf *key;
333 	u32 len = le16_to_cpu(desc->mpdu_len);
334 	const u8 *frame = (void *)hdr;
335 
336 	if ((status & IWL_RX_MPDU_STATUS_SEC_MASK) == IWL_RX_MPDU_STATUS_SEC_NONE)
337 		return 0;
338 
339 	/*
340 	 * For non-beacon, we don't really care. But beacons may
341 	 * be filtered out, and we thus need the firmware's replay
342 	 * detection, otherwise beacons the firmware previously
343 	 * filtered could be replayed, or something like that, and
344 	 * it can filter a lot - though usually only if nothing has
345 	 * changed.
346 	 */
347 	if (!ieee80211_is_beacon(hdr->frame_control))
348 		return 0;
349 
350 	if (!sta)
351 		return -1;
352 
353 	mvmsta = iwl_mvm_sta_from_mac80211(sta);
354 	mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
355 
356 	/* key mismatch - will also report !MIC_OK but we shouldn't count it */
357 	if (!(status & IWL_RX_MPDU_STATUS_KEY_VALID))
358 		goto report;
359 
360 	/* good cases */
361 	if (likely(status & IWL_RX_MPDU_STATUS_MIC_OK &&
362 		   !(status & IWL_RX_MPDU_STATUS_REPLAY_ERROR))) {
363 		stats->flag |= RX_FLAG_DECRYPTED;
364 		return 0;
365 	}
366 
367 	/*
368 	 * both keys will have the same cipher and MIC length, use
369 	 * whichever one is available
370 	 */
371 	key = rcu_dereference(mvmvif->bcn_prot.keys[0]);
372 	if (!key) {
373 		key = rcu_dereference(mvmvif->bcn_prot.keys[1]);
374 		if (!key)
375 			goto report;
376 	}
377 
378 	if (len < key->icv_len + IEEE80211_GMAC_PN_LEN + 2)
379 		goto report;
380 
381 	/* get the real key ID */
382 	keyid = frame[len - key->icv_len - IEEE80211_GMAC_PN_LEN - 2];
383 	/* and if that's the other key, look it up */
384 	if (keyid != key->keyidx) {
385 		/*
386 		 * shouldn't happen since firmware checked, but be safe
387 		 * in case the MIC length is wrong too, for example
388 		 */
389 		if (keyid != 6 && keyid != 7)
390 			return -1;
391 		key = rcu_dereference(mvmvif->bcn_prot.keys[keyid - 6]);
392 		if (!key)
393 			goto report;
394 	}
395 
396 	/* Report status to mac80211 */
397 	if (!(status & IWL_RX_MPDU_STATUS_MIC_OK))
398 		ieee80211_key_mic_failure(key);
399 	else if (status & IWL_RX_MPDU_STATUS_REPLAY_ERROR)
400 		ieee80211_key_replay(key);
401 report:
402 	wdev = ieee80211_vif_to_wdev(mvmsta->vif);
403 	if (wdev->netdev)
404 		cfg80211_rx_unprot_mlme_mgmt(wdev->netdev, (void *)hdr, len);
405 
406 	return -1;
407 }
408 
409 static int iwl_mvm_rx_crypto(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
410 			     struct ieee80211_hdr *hdr,
411 			     struct ieee80211_rx_status *stats, u16 phy_info,
412 			     struct iwl_rx_mpdu_desc *desc,
413 			     u32 pkt_flags, int queue, u8 *crypt_len)
414 {
415 	u32 status = le32_to_cpu(desc->status);
416 
417 	/*
418 	 * Drop UNKNOWN frames in aggregation, unless in monitor mode
419 	 * (where we don't have the keys).
420 	 * We limit this to aggregation because in TKIP this is a valid
421 	 * scenario, since we may not have the (correct) TTAK (phase 1
422 	 * key) in the firmware.
423 	 */
424 	if (phy_info & IWL_RX_MPDU_PHY_AMPDU &&
425 	    (status & IWL_RX_MPDU_STATUS_SEC_MASK) ==
426 	    IWL_RX_MPDU_STATUS_SEC_UNKNOWN && !mvm->monitor_on) {
427 		IWL_DEBUG_DROP(mvm, "Dropping packets, bad enc status\n");
428 		return -1;
429 	}
430 
431 	if (unlikely(ieee80211_is_mgmt(hdr->frame_control) &&
432 		     !ieee80211_has_protected(hdr->frame_control)))
433 		return iwl_mvm_rx_mgmt_prot(sta, hdr, desc, status, stats);
434 
435 	if (!ieee80211_has_protected(hdr->frame_control) ||
436 	    (status & IWL_RX_MPDU_STATUS_SEC_MASK) ==
437 	    IWL_RX_MPDU_STATUS_SEC_NONE)
438 		return 0;
439 
440 	/* TODO: handle packets encrypted with unknown alg */
441 
442 	switch (status & IWL_RX_MPDU_STATUS_SEC_MASK) {
443 	case IWL_RX_MPDU_STATUS_SEC_CCM:
444 	case IWL_RX_MPDU_STATUS_SEC_GCM:
445 		BUILD_BUG_ON(IEEE80211_CCMP_PN_LEN != IEEE80211_GCMP_PN_LEN);
446 		/* alg is CCM: check MIC only */
447 		if (!(status & IWL_RX_MPDU_STATUS_MIC_OK)) {
448 			IWL_DEBUG_DROP(mvm,
449 				       "Dropping packet, bad MIC (CCM/GCM)\n");
450 			return -1;
451 		}
452 
453 		stats->flag |= RX_FLAG_DECRYPTED | RX_FLAG_MIC_STRIPPED;
454 		*crypt_len = IEEE80211_CCMP_HDR_LEN;
455 		return 0;
456 	case IWL_RX_MPDU_STATUS_SEC_TKIP:
457 		/* Don't drop the frame and decrypt it in SW */
458 		if (!fw_has_api(&mvm->fw->ucode_capa,
459 				IWL_UCODE_TLV_API_DEPRECATE_TTAK) &&
460 		    !(status & IWL_RX_MPDU_RES_STATUS_TTAK_OK))
461 			return 0;
462 
463 		if (mvm->trans->mac_cfg->gen2 &&
464 		    !(status & RX_MPDU_RES_STATUS_MIC_OK))
465 			stats->flag |= RX_FLAG_MMIC_ERROR;
466 
467 		*crypt_len = IEEE80211_TKIP_IV_LEN;
468 		fallthrough;
469 	case IWL_RX_MPDU_STATUS_SEC_WEP:
470 		if (!(status & IWL_RX_MPDU_STATUS_ICV_OK))
471 			return -1;
472 
473 		stats->flag |= RX_FLAG_DECRYPTED;
474 		if ((status & IWL_RX_MPDU_STATUS_SEC_MASK) ==
475 				IWL_RX_MPDU_STATUS_SEC_WEP)
476 			*crypt_len = IEEE80211_WEP_IV_LEN;
477 
478 		if (pkt_flags & FH_RSCSR_RADA_EN) {
479 			stats->flag |= RX_FLAG_ICV_STRIPPED;
480 			if (mvm->trans->mac_cfg->gen2)
481 				stats->flag |= RX_FLAG_MMIC_STRIPPED;
482 		}
483 
484 		return 0;
485 	case IWL_RX_MPDU_STATUS_SEC_EXT_ENC:
486 		if (!(status & IWL_RX_MPDU_STATUS_MIC_OK))
487 			return -1;
488 		stats->flag |= RX_FLAG_DECRYPTED;
489 		return 0;
490 	case RX_MPDU_RES_STATUS_SEC_CMAC_GMAC_ENC:
491 		break;
492 	default:
493 		/*
494 		 * Sometimes we can get frames that were not decrypted
495 		 * because the firmware didn't have the keys yet. This can
496 		 * happen after connection where we can get multicast frames
497 		 * before the GTK is installed.
498 		 * Silently drop those frames.
499 		 * Also drop un-decrypted frames in monitor mode.
500 		 */
501 		if (!is_multicast_ether_addr(hdr->addr1) &&
502 		    !mvm->monitor_on && net_ratelimit())
503 			IWL_WARN(mvm, "Unhandled alg: 0x%x\n", status);
504 	}
505 
506 	return 0;
507 }
508 
509 static void iwl_mvm_rx_csum(struct iwl_mvm *mvm,
510 			    struct ieee80211_sta *sta,
511 			    struct sk_buff *skb,
512 			    struct iwl_rx_packet *pkt)
513 {
514 	struct iwl_rx_mpdu_desc *desc = (void *)pkt->data;
515 
516 	if (mvm->trans->mac_cfg->device_family >= IWL_DEVICE_FAMILY_AX210) {
517 		if (pkt->len_n_flags & cpu_to_le32(FH_RSCSR_RPA_EN)) {
518 			u16 hwsum = be16_to_cpu(desc->v3.raw_xsum);
519 
520 			skb->ip_summed = CHECKSUM_COMPLETE;
521 			skb->csum = csum_unfold(~(__force __sum16)hwsum);
522 		}
523 	} else {
524 		struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
525 		struct iwl_mvm_vif *mvmvif;
526 		u16 flags = le16_to_cpu(desc->l3l4_flags);
527 		u8 l3_prot = (u8)((flags & IWL_RX_L3L4_L3_PROTO_MASK) >>
528 				  IWL_RX_L3_PROTO_POS);
529 
530 		mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
531 
532 		if (mvmvif->features & NETIF_F_RXCSUM &&
533 		    flags & IWL_RX_L3L4_TCP_UDP_CSUM_OK &&
534 		    (flags & IWL_RX_L3L4_IP_HDR_CSUM_OK ||
535 		     l3_prot == IWL_RX_L3_TYPE_IPV6 ||
536 		     l3_prot == IWL_RX_L3_TYPE_IPV6_FRAG))
537 			skb->ip_summed = CHECKSUM_UNNECESSARY;
538 	}
539 }
540 
541 /*
542  * returns true if a packet is a duplicate or invalid tid and should be dropped.
543  * Updates AMSDU PN tracking info
544  */
545 static bool iwl_mvm_is_dup(struct ieee80211_sta *sta, int queue,
546 			   struct ieee80211_rx_status *rx_status,
547 			   struct ieee80211_hdr *hdr,
548 			   struct iwl_rx_mpdu_desc *desc)
549 {
550 	struct iwl_mvm_sta *mvm_sta;
551 	struct iwl_mvm_rxq_dup_data *dup_data;
552 	u8 tid, sub_frame_idx;
553 
554 	if (WARN_ON(IS_ERR_OR_NULL(sta)))
555 		return false;
556 
557 	mvm_sta = iwl_mvm_sta_from_mac80211(sta);
558 
559 	if (WARN_ON_ONCE(!mvm_sta->dup_data))
560 		return false;
561 
562 	dup_data = &mvm_sta->dup_data[queue];
563 
564 	/*
565 	 * Drop duplicate 802.11 retransmissions
566 	 * (IEEE 802.11-2012: 9.3.2.10 "Duplicate detection and recovery")
567 	 */
568 	if (ieee80211_is_ctl(hdr->frame_control) ||
569 	    ieee80211_is_any_nullfunc(hdr->frame_control) ||
570 	    is_multicast_ether_addr(hdr->addr1))
571 		return false;
572 
573 	if (ieee80211_is_data_qos(hdr->frame_control)) {
574 		/* frame has qos control */
575 		tid = ieee80211_get_tid(hdr);
576 		if (tid >= IWL_MAX_TID_COUNT)
577 			return true;
578 	} else {
579 		tid = IWL_MAX_TID_COUNT;
580 	}
581 
582 	/* If this wasn't a part of an A-MSDU the sub-frame index will be 0 */
583 	sub_frame_idx = desc->amsdu_info &
584 		IWL_RX_MPDU_AMSDU_SUBFRAME_IDX_MASK;
585 
586 	if (unlikely(ieee80211_has_retry(hdr->frame_control) &&
587 		     dup_data->last_seq[tid] == hdr->seq_ctrl &&
588 		     dup_data->last_sub_frame[tid] >= sub_frame_idx))
589 		return true;
590 
591 	/* Allow same PN as the first subframe for following sub frames */
592 	if (dup_data->last_seq[tid] == hdr->seq_ctrl &&
593 	    sub_frame_idx > dup_data->last_sub_frame[tid] &&
594 	    desc->mac_flags2 & IWL_RX_MPDU_MFLG2_AMSDU)
595 		rx_status->flag |= RX_FLAG_ALLOW_SAME_PN;
596 
597 	dup_data->last_seq[tid] = hdr->seq_ctrl;
598 	dup_data->last_sub_frame[tid] = sub_frame_idx;
599 
600 	rx_status->flag |= RX_FLAG_DUP_VALIDATED;
601 
602 	return false;
603 }
604 
605 static void iwl_mvm_release_frames(struct iwl_mvm *mvm,
606 				   struct ieee80211_sta *sta,
607 				   struct napi_struct *napi,
608 				   struct iwl_mvm_baid_data *baid_data,
609 				   struct iwl_mvm_reorder_buffer *reorder_buf,
610 				   u16 nssn)
611 {
612 	struct iwl_mvm_reorder_buf_entry *entries =
613 		&baid_data->entries[reorder_buf->queue *
614 				    baid_data->entries_per_queue];
615 	u16 ssn = reorder_buf->head_sn;
616 
617 	lockdep_assert_held(&reorder_buf->lock);
618 
619 	while (ieee80211_sn_less(ssn, nssn)) {
620 		int index = ssn % baid_data->buf_size;
621 		struct sk_buff_head *skb_list = &entries[index].frames;
622 		struct sk_buff *skb;
623 
624 		ssn = ieee80211_sn_inc(ssn);
625 
626 		/*
627 		 * Empty the list. Will have more than one frame for A-MSDU.
628 		 * Empty list is valid as well since nssn indicates frames were
629 		 * received.
630 		 */
631 		while ((skb = __skb_dequeue(skb_list))) {
632 			iwl_mvm_pass_packet_to_mac80211(mvm, napi, skb,
633 							reorder_buf->queue,
634 							sta);
635 			reorder_buf->num_stored--;
636 		}
637 	}
638 	reorder_buf->head_sn = nssn;
639 }
640 
641 static void iwl_mvm_del_ba(struct iwl_mvm *mvm, int queue,
642 			   struct iwl_mvm_delba_data *data)
643 {
644 	struct iwl_mvm_baid_data *ba_data;
645 	struct ieee80211_sta *sta;
646 	struct iwl_mvm_reorder_buffer *reorder_buf;
647 	u8 baid = data->baid;
648 	u32 sta_id;
649 
650 	if (WARN_ONCE(baid >= IWL_MAX_BAID, "invalid BAID: %x\n", baid))
651 		return;
652 
653 	rcu_read_lock();
654 
655 	ba_data = rcu_dereference(mvm->baid_map[baid]);
656 	if (WARN_ON_ONCE(!ba_data))
657 		goto out;
658 
659 	/* pick any STA ID to find the pointer */
660 	sta_id = ffs(ba_data->sta_mask) - 1;
661 	sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
662 	if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
663 		goto out;
664 
665 	reorder_buf = &ba_data->reorder_buf[queue];
666 
667 	/* release all frames that are in the reorder buffer to the stack */
668 	spin_lock_bh(&reorder_buf->lock);
669 	iwl_mvm_release_frames(mvm, sta, NULL, ba_data, reorder_buf,
670 			       ieee80211_sn_add(reorder_buf->head_sn,
671 						ba_data->buf_size));
672 	spin_unlock_bh(&reorder_buf->lock);
673 
674 out:
675 	rcu_read_unlock();
676 }
677 
678 static void iwl_mvm_release_frames_from_notif(struct iwl_mvm *mvm,
679 					      struct napi_struct *napi,
680 					      u8 baid, u16 nssn, int queue)
681 {
682 	struct ieee80211_sta *sta;
683 	struct iwl_mvm_reorder_buffer *reorder_buf;
684 	struct iwl_mvm_baid_data *ba_data;
685 	u32 sta_id;
686 
687 	IWL_DEBUG_HT(mvm, "Frame release notification for BAID %u, NSSN %d\n",
688 		     baid, nssn);
689 
690 	if (IWL_FW_CHECK(mvm,
691 			 baid == IWL_RX_REORDER_DATA_INVALID_BAID ||
692 			 baid >= ARRAY_SIZE(mvm->baid_map),
693 			 "invalid BAID from FW: %d\n", baid))
694 		return;
695 
696 	rcu_read_lock();
697 
698 	ba_data = rcu_dereference(mvm->baid_map[baid]);
699 	if (!ba_data) {
700 		IWL_DEBUG_RX(mvm,
701 			     "Got valid BAID %d but not allocated, invalid frame release!\n",
702 			     baid);
703 		goto out;
704 	}
705 
706 	/* pick any STA ID to find the pointer */
707 	sta_id = ffs(ba_data->sta_mask) - 1;
708 	sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
709 	if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
710 		goto out;
711 
712 	reorder_buf = &ba_data->reorder_buf[queue];
713 
714 	spin_lock_bh(&reorder_buf->lock);
715 	iwl_mvm_release_frames(mvm, sta, napi, ba_data,
716 			       reorder_buf, nssn);
717 	spin_unlock_bh(&reorder_buf->lock);
718 
719 out:
720 	rcu_read_unlock();
721 }
722 
723 void iwl_mvm_rx_queue_notif(struct iwl_mvm *mvm, struct napi_struct *napi,
724 			    struct iwl_rx_cmd_buffer *rxb, int queue)
725 {
726 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
727 	struct iwl_rxq_sync_notification *notif;
728 	struct iwl_mvm_internal_rxq_notif *internal_notif;
729 	u32 len = iwl_rx_packet_payload_len(pkt);
730 
731 	notif = (void *)pkt->data;
732 	internal_notif = (void *)notif->payload;
733 
734 	if (WARN_ONCE(len < sizeof(*notif) + sizeof(*internal_notif),
735 		      "invalid notification size %d (%d)",
736 		      len, (int)(sizeof(*notif) + sizeof(*internal_notif))))
737 		return;
738 	len -= sizeof(*notif) + sizeof(*internal_notif);
739 
740 	if (WARN_ONCE(internal_notif->sync &&
741 		      mvm->queue_sync_cookie != internal_notif->cookie,
742 		      "Received expired RX queue sync message (cookie %d but wanted %d, queue %d)\n",
743 		      internal_notif->cookie, mvm->queue_sync_cookie, queue))
744 		return;
745 
746 	switch (internal_notif->type) {
747 	case IWL_MVM_RXQ_EMPTY:
748 		WARN_ONCE(len, "invalid empty notification size %d", len);
749 		break;
750 	case IWL_MVM_RXQ_NOTIF_DEL_BA:
751 		if (WARN_ONCE(len != sizeof(struct iwl_mvm_delba_data),
752 			      "invalid delba notification size %d (%d)",
753 			      len, (int)sizeof(struct iwl_mvm_delba_data)))
754 			break;
755 		iwl_mvm_del_ba(mvm, queue, (void *)internal_notif->data);
756 		break;
757 	default:
758 		WARN_ONCE(1, "Invalid identifier %d", internal_notif->type);
759 	}
760 
761 	if (internal_notif->sync) {
762 		WARN_ONCE(!test_and_clear_bit(queue, &mvm->queue_sync_state),
763 			  "queue sync: queue %d responded a second time!\n",
764 			  queue);
765 		if (READ_ONCE(mvm->queue_sync_state) == 0)
766 			wake_up(&mvm->rx_sync_waitq);
767 	}
768 }
769 
770 /*
771  * Returns true if the MPDU was buffered\dropped, false if it should be passed
772  * to upper layer.
773  */
774 static bool iwl_mvm_reorder(struct iwl_mvm *mvm,
775 			    struct napi_struct *napi,
776 			    int queue,
777 			    struct ieee80211_sta *sta,
778 			    struct sk_buff *skb,
779 			    struct iwl_rx_mpdu_desc *desc)
780 {
781 	struct ieee80211_hdr *hdr = (void *)skb_mac_header(skb);
782 	struct iwl_mvm_baid_data *baid_data;
783 	struct iwl_mvm_reorder_buffer *buffer;
784 	u32 reorder = le32_to_cpu(desc->reorder_data);
785 	bool amsdu = desc->mac_flags2 & IWL_RX_MPDU_MFLG2_AMSDU;
786 	bool last_subframe =
787 		desc->amsdu_info & IWL_RX_MPDU_AMSDU_LAST_SUBFRAME;
788 	u8 tid = ieee80211_get_tid(hdr);
789 	struct iwl_mvm_reorder_buf_entry *entries;
790 	u32 sta_mask;
791 	int index;
792 	u16 nssn, sn;
793 	u8 baid;
794 
795 	baid = (reorder & IWL_RX_MPDU_REORDER_BAID_MASK) >>
796 		IWL_RX_MPDU_REORDER_BAID_SHIFT;
797 
798 	if (mvm->trans->mac_cfg->device_family == IWL_DEVICE_FAMILY_9000)
799 		return false;
800 
801 	/*
802 	 * This also covers the case of receiving a Block Ack Request
803 	 * outside a BA session; we'll pass it to mac80211 and that
804 	 * then sends a delBA action frame.
805 	 * This also covers pure monitor mode, in which case we won't
806 	 * have any BA sessions.
807 	 */
808 	if (baid == IWL_RX_REORDER_DATA_INVALID_BAID)
809 		return false;
810 
811 	/* no sta yet */
812 	if (WARN_ONCE(IS_ERR_OR_NULL(sta),
813 		      "Got valid BAID without a valid station assigned\n"))
814 		return false;
815 
816 	/* not a data packet or a bar */
817 	if (!ieee80211_is_back_req(hdr->frame_control) &&
818 	    (!ieee80211_is_data_qos(hdr->frame_control) ||
819 	     is_multicast_ether_addr(hdr->addr1)))
820 		return false;
821 
822 	if (unlikely(!ieee80211_is_data_present(hdr->frame_control)))
823 		return false;
824 
825 	baid_data = rcu_dereference(mvm->baid_map[baid]);
826 	if (!baid_data) {
827 		IWL_DEBUG_RX(mvm,
828 			     "Got valid BAID but no baid allocated, bypass the re-ordering buffer. Baid %d reorder 0x%x\n",
829 			      baid, reorder);
830 		return false;
831 	}
832 
833 	sta_mask = iwl_mvm_sta_fw_id_mask(mvm, sta, -1);
834 
835 	if (IWL_FW_CHECK(mvm,
836 			 tid != baid_data->tid ||
837 			 !(sta_mask & baid_data->sta_mask),
838 			 "baid 0x%x is mapped to sta_mask:0x%x tid:%d, but was received for sta_mask:0x%x tid:%d\n",
839 			 baid, baid_data->sta_mask, baid_data->tid,
840 			 sta_mask, tid))
841 		return false;
842 
843 	nssn = reorder & IWL_RX_MPDU_REORDER_NSSN_MASK;
844 	sn = (reorder & IWL_RX_MPDU_REORDER_SN_MASK) >>
845 		IWL_RX_MPDU_REORDER_SN_SHIFT;
846 
847 	buffer = &baid_data->reorder_buf[queue];
848 	entries = &baid_data->entries[queue * baid_data->entries_per_queue];
849 
850 	spin_lock_bh(&buffer->lock);
851 
852 	if (!buffer->valid) {
853 		if (reorder & IWL_RX_MPDU_REORDER_BA_OLD_SN) {
854 			spin_unlock_bh(&buffer->lock);
855 			return false;
856 		}
857 		buffer->valid = true;
858 	}
859 
860 	/* drop any duplicated packets */
861 	if (desc->status & cpu_to_le32(IWL_RX_MPDU_STATUS_DUPLICATE))
862 		goto drop;
863 
864 	/* drop any oudated packets */
865 	if (reorder & IWL_RX_MPDU_REORDER_BA_OLD_SN)
866 		goto drop;
867 
868 	/* release immediately if allowed by nssn and no stored frames */
869 	if (!buffer->num_stored && ieee80211_sn_less(sn, nssn)) {
870 		if (!amsdu || last_subframe)
871 			buffer->head_sn = nssn;
872 
873 		spin_unlock_bh(&buffer->lock);
874 		return false;
875 	}
876 
877 	/*
878 	 * release immediately if there are no stored frames, and the sn is
879 	 * equal to the head.
880 	 * This can happen due to reorder timer, where NSSN is behind head_sn.
881 	 * When we released everything, and we got the next frame in the
882 	 * sequence, according to the NSSN we can't release immediately,
883 	 * while technically there is no hole and we can move forward.
884 	 */
885 	if (!buffer->num_stored && sn == buffer->head_sn) {
886 		if (!amsdu || last_subframe)
887 			buffer->head_sn = ieee80211_sn_inc(buffer->head_sn);
888 
889 		spin_unlock_bh(&buffer->lock);
890 		return false;
891 	}
892 
893 	/* put in reorder buffer */
894 	index = sn % baid_data->buf_size;
895 	__skb_queue_tail(&entries[index].frames, skb);
896 	buffer->num_stored++;
897 
898 	/*
899 	 * We cannot trust NSSN for AMSDU sub-frames that are not the last.
900 	 * The reason is that NSSN advances on the first sub-frame, and may
901 	 * cause the reorder buffer to advance before all the sub-frames arrive.
902 	 * Example: reorder buffer contains SN 0 & 2, and we receive AMSDU with
903 	 * SN 1. NSSN for first sub frame will be 3 with the result of driver
904 	 * releasing SN 0,1, 2. When sub-frame 1 arrives - reorder buffer is
905 	 * already ahead and it will be dropped.
906 	 * If the last sub-frame is not on this queue - we will get frame
907 	 * release notification with up to date NSSN.
908 	 */
909 	if (!amsdu || last_subframe)
910 		iwl_mvm_release_frames(mvm, sta, napi, baid_data,
911 				       buffer, nssn);
912 
913 	spin_unlock_bh(&buffer->lock);
914 	return true;
915 
916 drop:
917 	kfree_skb(skb);
918 	spin_unlock_bh(&buffer->lock);
919 	return true;
920 }
921 
922 static void iwl_mvm_agg_rx_received(struct iwl_mvm *mvm,
923 				    u32 reorder_data, u8 baid)
924 {
925 	unsigned long now = jiffies;
926 	unsigned long timeout;
927 	struct iwl_mvm_baid_data *data;
928 
929 	rcu_read_lock();
930 
931 	data = rcu_dereference(mvm->baid_map[baid]);
932 	if (!data) {
933 		IWL_DEBUG_RX(mvm,
934 			     "Got valid BAID but no baid allocated, bypass the re-ordering buffer. Baid %d reorder 0x%x\n",
935 			      baid, reorder_data);
936 		goto out;
937 	}
938 
939 	if (!data->timeout)
940 		goto out;
941 
942 	timeout = data->timeout;
943 	/*
944 	 * Do not update last rx all the time to avoid cache bouncing
945 	 * between the rx queues.
946 	 * Update it every timeout. Worst case is the session will
947 	 * expire after ~ 2 * timeout, which doesn't matter that much.
948 	 */
949 	if (time_before(data->last_rx + TU_TO_JIFFIES(timeout), now))
950 		/* Update is atomic */
951 		data->last_rx = now;
952 
953 out:
954 	rcu_read_unlock();
955 }
956 
957 static void iwl_mvm_flip_address(u8 *addr)
958 {
959 	int i;
960 	u8 mac_addr[ETH_ALEN];
961 
962 	for (i = 0; i < ETH_ALEN; i++)
963 		mac_addr[i] = addr[ETH_ALEN - i - 1];
964 	ether_addr_copy(addr, mac_addr);
965 }
966 
967 struct iwl_mvm_rx_phy_data {
968 	enum iwl_rx_phy_info_type info_type;
969 	__le32 d0, d1, d2, d3, eht_d4, d5;
970 	__le16 d4;
971 	bool with_data;
972 	bool first_subframe;
973 	__le32 rx_vec[4];
974 
975 	u32 rate_n_flags;
976 	u32 gp2_on_air_rise;
977 	u16 phy_info;
978 	u8 energy_a, energy_b;
979 	u8 channel;
980 };
981 
982 static void iwl_mvm_decode_he_mu_ext(struct iwl_mvm *mvm,
983 				     struct iwl_mvm_rx_phy_data *phy_data,
984 				     struct ieee80211_radiotap_he_mu *he_mu)
985 {
986 	u32 phy_data2 = le32_to_cpu(phy_data->d2);
987 	u32 phy_data3 = le32_to_cpu(phy_data->d3);
988 	u16 phy_data4 = le16_to_cpu(phy_data->d4);
989 	u32 rate_n_flags = phy_data->rate_n_flags;
990 
991 	if (FIELD_GET(IWL_RX_PHY_DATA4_HE_MU_EXT_CH1_CRC_OK, phy_data4)) {
992 		he_mu->flags1 |=
993 			cpu_to_le16(IEEE80211_RADIOTAP_HE_MU_FLAGS1_CH1_RU_KNOWN |
994 				    IEEE80211_RADIOTAP_HE_MU_FLAGS1_CH1_CTR_26T_RU_KNOWN);
995 
996 		he_mu->flags1 |=
997 			le16_encode_bits(FIELD_GET(IWL_RX_PHY_DATA4_HE_MU_EXT_CH1_CTR_RU,
998 						   phy_data4),
999 					 IEEE80211_RADIOTAP_HE_MU_FLAGS1_CH1_CTR_26T_RU);
1000 
1001 		he_mu->ru_ch1[0] = FIELD_GET(IWL_RX_PHY_DATA2_HE_MU_EXT_CH1_RU0,
1002 					     phy_data2);
1003 		he_mu->ru_ch1[1] = FIELD_GET(IWL_RX_PHY_DATA3_HE_MU_EXT_CH1_RU1,
1004 					     phy_data3);
1005 		he_mu->ru_ch1[2] = FIELD_GET(IWL_RX_PHY_DATA2_HE_MU_EXT_CH1_RU2,
1006 					     phy_data2);
1007 		he_mu->ru_ch1[3] = FIELD_GET(IWL_RX_PHY_DATA3_HE_MU_EXT_CH1_RU3,
1008 					     phy_data3);
1009 	}
1010 
1011 	if (FIELD_GET(IWL_RX_PHY_DATA4_HE_MU_EXT_CH2_CRC_OK, phy_data4) &&
1012 	    (rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK_V1) != RATE_MCS_CHAN_WIDTH_20) {
1013 		he_mu->flags1 |=
1014 			cpu_to_le16(IEEE80211_RADIOTAP_HE_MU_FLAGS1_CH2_RU_KNOWN |
1015 				    IEEE80211_RADIOTAP_HE_MU_FLAGS1_CH2_CTR_26T_RU_KNOWN);
1016 
1017 		he_mu->flags2 |=
1018 			le16_encode_bits(FIELD_GET(IWL_RX_PHY_DATA4_HE_MU_EXT_CH2_CTR_RU,
1019 						   phy_data4),
1020 					 IEEE80211_RADIOTAP_HE_MU_FLAGS2_CH2_CTR_26T_RU);
1021 
1022 		he_mu->ru_ch2[0] = FIELD_GET(IWL_RX_PHY_DATA2_HE_MU_EXT_CH2_RU0,
1023 					     phy_data2);
1024 		he_mu->ru_ch2[1] = FIELD_GET(IWL_RX_PHY_DATA3_HE_MU_EXT_CH2_RU1,
1025 					     phy_data3);
1026 		he_mu->ru_ch2[2] = FIELD_GET(IWL_RX_PHY_DATA2_HE_MU_EXT_CH2_RU2,
1027 					     phy_data2);
1028 		he_mu->ru_ch2[3] = FIELD_GET(IWL_RX_PHY_DATA3_HE_MU_EXT_CH2_RU3,
1029 					     phy_data3);
1030 	}
1031 }
1032 
1033 static void
1034 iwl_mvm_decode_he_phy_ru_alloc(struct iwl_mvm_rx_phy_data *phy_data,
1035 			       struct ieee80211_radiotap_he *he,
1036 			       struct ieee80211_radiotap_he_mu *he_mu,
1037 			       struct ieee80211_rx_status *rx_status)
1038 {
1039 	/*
1040 	 * Unfortunately, we have to leave the mac80211 data
1041 	 * incorrect for the case that we receive an HE-MU
1042 	 * transmission and *don't* have the HE phy data (due
1043 	 * to the bits being used for TSF). This shouldn't
1044 	 * happen though as management frames where we need
1045 	 * the TSF/timers are not be transmitted in HE-MU.
1046 	 */
1047 	u8 ru = le32_get_bits(phy_data->d1, IWL_RX_PHY_DATA1_HE_RU_ALLOC_MASK);
1048 	u32 rate_n_flags = phy_data->rate_n_flags;
1049 	u32 he_type = rate_n_flags & RATE_MCS_HE_TYPE_MSK;
1050 	u8 offs = 0;
1051 
1052 	rx_status->bw = RATE_INFO_BW_HE_RU;
1053 
1054 	he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_BW_RU_ALLOC_KNOWN);
1055 
1056 	switch (ru) {
1057 	case 0 ... 36:
1058 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_26;
1059 		offs = ru;
1060 		break;
1061 	case 37 ... 52:
1062 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_52;
1063 		offs = ru - 37;
1064 		break;
1065 	case 53 ... 60:
1066 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_106;
1067 		offs = ru - 53;
1068 		break;
1069 	case 61 ... 64:
1070 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_242;
1071 		offs = ru - 61;
1072 		break;
1073 	case 65 ... 66:
1074 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_484;
1075 		offs = ru - 65;
1076 		break;
1077 	case 67:
1078 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_996;
1079 		break;
1080 	case 68:
1081 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_2x996;
1082 		break;
1083 	}
1084 	he->data2 |= le16_encode_bits(offs,
1085 				      IEEE80211_RADIOTAP_HE_DATA2_RU_OFFSET);
1086 	he->data2 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA2_PRISEC_80_KNOWN |
1087 				 IEEE80211_RADIOTAP_HE_DATA2_RU_OFFSET_KNOWN);
1088 	if (phy_data->d1 & cpu_to_le32(IWL_RX_PHY_DATA1_HE_RU_ALLOC_SEC80))
1089 		he->data2 |=
1090 			cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA2_PRISEC_80_SEC);
1091 
1092 #define CHECK_BW(bw) \
1093 	BUILD_BUG_ON(IEEE80211_RADIOTAP_HE_MU_FLAGS2_BW_FROM_SIG_A_BW_ ## bw ## MHZ != \
1094 		     RATE_MCS_CHAN_WIDTH_##bw >> RATE_MCS_CHAN_WIDTH_POS); \
1095 	BUILD_BUG_ON(IEEE80211_RADIOTAP_HE_DATA6_TB_PPDU_BW_ ## bw ## MHZ != \
1096 		     RATE_MCS_CHAN_WIDTH_##bw >> RATE_MCS_CHAN_WIDTH_POS)
1097 	CHECK_BW(20);
1098 	CHECK_BW(40);
1099 	CHECK_BW(80);
1100 	CHECK_BW(160);
1101 
1102 	if (he_mu)
1103 		he_mu->flags2 |=
1104 			le16_encode_bits(FIELD_GET(RATE_MCS_CHAN_WIDTH_MSK,
1105 						   rate_n_flags),
1106 					 IEEE80211_RADIOTAP_HE_MU_FLAGS2_BW_FROM_SIG_A_BW);
1107 	else if (he_type == RATE_MCS_HE_TYPE_TRIG)
1108 		he->data6 |=
1109 			cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA6_TB_PPDU_BW_KNOWN) |
1110 			le16_encode_bits(FIELD_GET(RATE_MCS_CHAN_WIDTH_MSK,
1111 						   rate_n_flags),
1112 					 IEEE80211_RADIOTAP_HE_DATA6_TB_PPDU_BW);
1113 }
1114 
1115 static void iwl_mvm_decode_he_phy_data(struct iwl_mvm *mvm,
1116 				       struct iwl_mvm_rx_phy_data *phy_data,
1117 				       struct ieee80211_radiotap_he *he,
1118 				       struct ieee80211_radiotap_he_mu *he_mu,
1119 				       struct ieee80211_rx_status *rx_status,
1120 				       int queue)
1121 {
1122 	switch (phy_data->info_type) {
1123 	case IWL_RX_PHY_INFO_TYPE_NONE:
1124 	case IWL_RX_PHY_INFO_TYPE_CCK:
1125 	case IWL_RX_PHY_INFO_TYPE_OFDM_LGCY:
1126 	case IWL_RX_PHY_INFO_TYPE_HT:
1127 	case IWL_RX_PHY_INFO_TYPE_VHT_SU:
1128 	case IWL_RX_PHY_INFO_TYPE_VHT_MU:
1129 	case IWL_RX_PHY_INFO_TYPE_EHT_MU:
1130 	case IWL_RX_PHY_INFO_TYPE_EHT_TB:
1131 	case IWL_RX_PHY_INFO_TYPE_EHT_MU_EXT:
1132 	case IWL_RX_PHY_INFO_TYPE_EHT_TB_EXT:
1133 		return;
1134 	case IWL_RX_PHY_INFO_TYPE_HE_TB_EXT:
1135 		he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_SPTL_REUSE_KNOWN |
1136 					 IEEE80211_RADIOTAP_HE_DATA1_SPTL_REUSE2_KNOWN |
1137 					 IEEE80211_RADIOTAP_HE_DATA1_SPTL_REUSE3_KNOWN |
1138 					 IEEE80211_RADIOTAP_HE_DATA1_SPTL_REUSE4_KNOWN);
1139 		he->data4 |= le16_encode_bits(le32_get_bits(phy_data->d2,
1140 							    IWL_RX_PHY_DATA2_HE_TB_EXT_SPTL_REUSE1),
1141 					      IEEE80211_RADIOTAP_HE_DATA4_TB_SPTL_REUSE1);
1142 		he->data4 |= le16_encode_bits(le32_get_bits(phy_data->d2,
1143 							    IWL_RX_PHY_DATA2_HE_TB_EXT_SPTL_REUSE2),
1144 					      IEEE80211_RADIOTAP_HE_DATA4_TB_SPTL_REUSE2);
1145 		he->data4 |= le16_encode_bits(le32_get_bits(phy_data->d2,
1146 							    IWL_RX_PHY_DATA2_HE_TB_EXT_SPTL_REUSE3),
1147 					      IEEE80211_RADIOTAP_HE_DATA4_TB_SPTL_REUSE3);
1148 		he->data4 |= le16_encode_bits(le32_get_bits(phy_data->d2,
1149 							    IWL_RX_PHY_DATA2_HE_TB_EXT_SPTL_REUSE4),
1150 					      IEEE80211_RADIOTAP_HE_DATA4_TB_SPTL_REUSE4);
1151 		fallthrough;
1152 	case IWL_RX_PHY_INFO_TYPE_HE_SU:
1153 	case IWL_RX_PHY_INFO_TYPE_HE_MU:
1154 	case IWL_RX_PHY_INFO_TYPE_HE_MU_EXT:
1155 	case IWL_RX_PHY_INFO_TYPE_HE_TB:
1156 		/* HE common */
1157 		he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_LDPC_XSYMSEG_KNOWN |
1158 					 IEEE80211_RADIOTAP_HE_DATA1_DOPPLER_KNOWN |
1159 					 IEEE80211_RADIOTAP_HE_DATA1_BSS_COLOR_KNOWN);
1160 		he->data2 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA2_PRE_FEC_PAD_KNOWN |
1161 					 IEEE80211_RADIOTAP_HE_DATA2_PE_DISAMBIG_KNOWN |
1162 					 IEEE80211_RADIOTAP_HE_DATA2_TXOP_KNOWN |
1163 					 IEEE80211_RADIOTAP_HE_DATA2_NUM_LTF_SYMS_KNOWN);
1164 		he->data3 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1165 							    IWL_RX_PHY_DATA0_HE_BSS_COLOR_MASK),
1166 					      IEEE80211_RADIOTAP_HE_DATA3_BSS_COLOR);
1167 		if (phy_data->info_type != IWL_RX_PHY_INFO_TYPE_HE_TB &&
1168 		    phy_data->info_type != IWL_RX_PHY_INFO_TYPE_HE_TB_EXT) {
1169 			he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_UL_DL_KNOWN);
1170 			he->data3 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1171 							    IWL_RX_PHY_DATA0_HE_UPLINK),
1172 						      IEEE80211_RADIOTAP_HE_DATA3_UL_DL);
1173 		}
1174 		he->data3 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1175 							    IWL_RX_PHY_DATA0_HE_LDPC_EXT_SYM),
1176 					      IEEE80211_RADIOTAP_HE_DATA3_LDPC_XSYMSEG);
1177 		he->data5 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1178 							    IWL_RX_PHY_DATA0_HE_PRE_FEC_PAD_MASK),
1179 					      IEEE80211_RADIOTAP_HE_DATA5_PRE_FEC_PAD);
1180 		he->data5 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1181 							    IWL_RX_PHY_DATA0_HE_PE_DISAMBIG),
1182 					      IEEE80211_RADIOTAP_HE_DATA5_PE_DISAMBIG);
1183 		he->data5 |= le16_encode_bits(le32_get_bits(phy_data->d1,
1184 							    IWL_RX_PHY_DATA1_HE_LTF_NUM_MASK),
1185 					      IEEE80211_RADIOTAP_HE_DATA5_NUM_LTF_SYMS);
1186 		he->data6 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1187 							    IWL_RX_PHY_DATA0_HE_TXOP_DUR_MASK),
1188 					      IEEE80211_RADIOTAP_HE_DATA6_TXOP);
1189 		he->data6 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1190 							    IWL_RX_PHY_DATA0_HE_DOPPLER),
1191 					      IEEE80211_RADIOTAP_HE_DATA6_DOPPLER);
1192 		break;
1193 	}
1194 
1195 	switch (phy_data->info_type) {
1196 	case IWL_RX_PHY_INFO_TYPE_HE_MU_EXT:
1197 	case IWL_RX_PHY_INFO_TYPE_HE_MU:
1198 	case IWL_RX_PHY_INFO_TYPE_HE_SU:
1199 		he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_SPTL_REUSE_KNOWN);
1200 		he->data4 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1201 							    IWL_RX_PHY_DATA0_HE_SPATIAL_REUSE_MASK),
1202 					      IEEE80211_RADIOTAP_HE_DATA4_SU_MU_SPTL_REUSE);
1203 		break;
1204 	default:
1205 		/* nothing here */
1206 		break;
1207 	}
1208 
1209 	switch (phy_data->info_type) {
1210 	case IWL_RX_PHY_INFO_TYPE_HE_MU_EXT:
1211 		he_mu->flags1 |=
1212 			le16_encode_bits(le16_get_bits(phy_data->d4,
1213 						       IWL_RX_PHY_DATA4_HE_MU_EXT_SIGB_DCM),
1214 					 IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_DCM);
1215 		he_mu->flags1 |=
1216 			le16_encode_bits(le16_get_bits(phy_data->d4,
1217 						       IWL_RX_PHY_DATA4_HE_MU_EXT_SIGB_MCS_MASK),
1218 					 IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_MCS);
1219 		he_mu->flags2 |=
1220 			le16_encode_bits(le16_get_bits(phy_data->d4,
1221 						       IWL_RX_PHY_DATA4_HE_MU_EXT_PREAMBLE_PUNC_TYPE_MASK),
1222 					 IEEE80211_RADIOTAP_HE_MU_FLAGS2_PUNC_FROM_SIG_A_BW);
1223 		iwl_mvm_decode_he_mu_ext(mvm, phy_data, he_mu);
1224 		fallthrough;
1225 	case IWL_RX_PHY_INFO_TYPE_HE_MU:
1226 		he_mu->flags2 |=
1227 			le16_encode_bits(le32_get_bits(phy_data->d1,
1228 						       IWL_RX_PHY_DATA1_HE_MU_SIBG_SYM_OR_USER_NUM_MASK),
1229 					 IEEE80211_RADIOTAP_HE_MU_FLAGS2_SIG_B_SYMS_USERS);
1230 		he_mu->flags2 |=
1231 			le16_encode_bits(le32_get_bits(phy_data->d1,
1232 						       IWL_RX_PHY_DATA1_HE_MU_SIGB_COMPRESSION),
1233 					 IEEE80211_RADIOTAP_HE_MU_FLAGS2_SIG_B_COMP);
1234 		fallthrough;
1235 	case IWL_RX_PHY_INFO_TYPE_HE_TB:
1236 	case IWL_RX_PHY_INFO_TYPE_HE_TB_EXT:
1237 		iwl_mvm_decode_he_phy_ru_alloc(phy_data, he, he_mu, rx_status);
1238 		break;
1239 	case IWL_RX_PHY_INFO_TYPE_HE_SU:
1240 		he->data1 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_BEAM_CHANGE_KNOWN);
1241 		he->data3 |= le16_encode_bits(le32_get_bits(phy_data->d0,
1242 							    IWL_RX_PHY_DATA0_HE_BEAM_CHNG),
1243 					      IEEE80211_RADIOTAP_HE_DATA3_BEAM_CHANGE);
1244 		break;
1245 	default:
1246 		/* nothing */
1247 		break;
1248 	}
1249 }
1250 
1251 #define LE32_DEC_ENC(value, dec_bits, enc_bits) \
1252 	le32_encode_bits(le32_get_bits(value, dec_bits), enc_bits)
1253 
1254 #define IWL_MVM_ENC_USIG_VALUE_MASK(usig, in_value, dec_bits, enc_bits) do { \
1255 	typeof(enc_bits) _enc_bits = enc_bits; \
1256 	typeof(usig) _usig = usig; \
1257 	(_usig)->mask |= cpu_to_le32(_enc_bits); \
1258 	(_usig)->value |= LE32_DEC_ENC(in_value, dec_bits, _enc_bits); \
1259 } while (0)
1260 
1261 #define __IWL_MVM_ENC_EHT_RU(rt_data, rt_ru, fw_data, fw_ru) \
1262 	eht->data[(rt_data)] |= \
1263 		(cpu_to_le32 \
1264 		 (IEEE80211_RADIOTAP_EHT_DATA ## rt_data ## _RU_ALLOC_CC_ ## rt_ru ## _KNOWN) | \
1265 		 LE32_DEC_ENC(data ## fw_data, \
1266 			      IWL_RX_PHY_DATA ## fw_data ## _EHT_MU_EXT_RU_ALLOC_ ## fw_ru, \
1267 			      IEEE80211_RADIOTAP_EHT_DATA ## rt_data ## _RU_ALLOC_CC_ ## rt_ru))
1268 
1269 #define _IWL_MVM_ENC_EHT_RU(rt_data, rt_ru, fw_data, fw_ru)	\
1270 	__IWL_MVM_ENC_EHT_RU(rt_data, rt_ru, fw_data, fw_ru)
1271 
1272 #define IEEE80211_RADIOTAP_RU_DATA_1_1_1	1
1273 #define IEEE80211_RADIOTAP_RU_DATA_2_1_1	2
1274 #define IEEE80211_RADIOTAP_RU_DATA_1_1_2	2
1275 #define IEEE80211_RADIOTAP_RU_DATA_2_1_2	2
1276 #define IEEE80211_RADIOTAP_RU_DATA_1_2_1	3
1277 #define IEEE80211_RADIOTAP_RU_DATA_2_2_1	3
1278 #define IEEE80211_RADIOTAP_RU_DATA_1_2_2	3
1279 #define IEEE80211_RADIOTAP_RU_DATA_2_2_2	4
1280 
1281 #define IWL_RX_RU_DATA_A1			2
1282 #define IWL_RX_RU_DATA_A2			2
1283 #define IWL_RX_RU_DATA_B1			2
1284 #define IWL_RX_RU_DATA_B2			4
1285 #define IWL_RX_RU_DATA_C1			3
1286 #define IWL_RX_RU_DATA_C2			3
1287 #define IWL_RX_RU_DATA_D1			4
1288 #define IWL_RX_RU_DATA_D2			4
1289 
1290 #define IWL_MVM_ENC_EHT_RU(rt_ru, fw_ru)				\
1291 	_IWL_MVM_ENC_EHT_RU(IEEE80211_RADIOTAP_RU_DATA_ ## rt_ru,	\
1292 			    rt_ru,					\
1293 			    IWL_RX_RU_DATA_ ## fw_ru,			\
1294 			    fw_ru)
1295 
1296 static void iwl_mvm_decode_eht_ext_mu(struct iwl_mvm *mvm,
1297 				      struct iwl_mvm_rx_phy_data *phy_data,
1298 				      struct ieee80211_rx_status *rx_status,
1299 				      struct ieee80211_radiotap_eht *eht,
1300 				      struct ieee80211_radiotap_eht_usig *usig)
1301 {
1302 	if (phy_data->with_data) {
1303 		__le32 data1 = phy_data->d1;
1304 		__le32 data2 = phy_data->d2;
1305 		__le32 data3 = phy_data->d3;
1306 		__le32 data4 = phy_data->eht_d4;
1307 		__le32 data5 = phy_data->d5;
1308 		u32 phy_bw = phy_data->rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK;
1309 
1310 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data5,
1311 					    IWL_RX_PHY_DATA5_EHT_TYPE_AND_COMP,
1312 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B0_B1_PPDU_TYPE);
1313 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data5,
1314 					    IWL_RX_PHY_DATA5_EHT_MU_PUNC_CH_CODE,
1315 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B3_B7_PUNCTURED_INFO);
1316 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data4,
1317 					    IWL_RX_PHY_DATA4_EHT_MU_EXT_SIGB_MCS,
1318 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B9_B10_SIG_MCS);
1319 		IWL_MVM_ENC_USIG_VALUE_MASK
1320 			(usig, data1, IWL_RX_PHY_DATA1_EHT_MU_NUM_SIG_SYM_USIGA2,
1321 			 IEEE80211_RADIOTAP_EHT_USIG2_MU_B11_B15_EHT_SIG_SYMBOLS);
1322 
1323 		eht->user_info[0] |=
1324 			cpu_to_le32(IEEE80211_RADIOTAP_EHT_USER_INFO_STA_ID_KNOWN) |
1325 			LE32_DEC_ENC(data5, IWL_RX_PHY_DATA5_EHT_MU_STA_ID_USR,
1326 				     IEEE80211_RADIOTAP_EHT_USER_INFO_STA_ID);
1327 
1328 		eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_NR_NON_OFDMA_USERS_M);
1329 		eht->data[7] |= LE32_DEC_ENC
1330 			(data5, IWL_RX_PHY_DATA5_EHT_MU_NUM_USR_NON_OFDMA,
1331 			 IEEE80211_RADIOTAP_EHT_DATA7_NUM_OF_NON_OFDMA_USERS);
1332 
1333 		/*
1334 		 * Hardware labels the content channels/RU allocation values
1335 		 * as follows:
1336 		 *           Content Channel 1		Content Channel 2
1337 		 *   20 MHz: A1
1338 		 *   40 MHz: A1				B1
1339 		 *   80 MHz: A1 C1			B1 D1
1340 		 *  160 MHz: A1 C1 A2 C2		B1 D1 B2 D2
1341 		 *  320 MHz: A1 C1 A2 C2 A3 C3 A4 C4	B1 D1 B2 D2 B3 D3 B4 D4
1342 		 *
1343 		 * However firmware can only give us A1-D2, so the higher
1344 		 * frequencies are missing.
1345 		 */
1346 
1347 		switch (phy_bw) {
1348 		case RATE_MCS_CHAN_WIDTH_320:
1349 			/* additional values are missing in RX metadata */
1350 		case RATE_MCS_CHAN_WIDTH_160:
1351 			/* content channel 1 */
1352 			IWL_MVM_ENC_EHT_RU(1_2_1, A2);
1353 			IWL_MVM_ENC_EHT_RU(1_2_2, C2);
1354 			/* content channel 2 */
1355 			IWL_MVM_ENC_EHT_RU(2_2_1, B2);
1356 			IWL_MVM_ENC_EHT_RU(2_2_2, D2);
1357 			fallthrough;
1358 		case RATE_MCS_CHAN_WIDTH_80:
1359 			/* content channel 1 */
1360 			IWL_MVM_ENC_EHT_RU(1_1_2, C1);
1361 			/* content channel 2 */
1362 			IWL_MVM_ENC_EHT_RU(2_1_2, D1);
1363 			fallthrough;
1364 		case RATE_MCS_CHAN_WIDTH_40:
1365 			/* content channel 2 */
1366 			IWL_MVM_ENC_EHT_RU(2_1_1, B1);
1367 			fallthrough;
1368 		case RATE_MCS_CHAN_WIDTH_20:
1369 			IWL_MVM_ENC_EHT_RU(1_1_1, A1);
1370 			break;
1371 		}
1372 	} else {
1373 		__le32 usig_a1 = phy_data->rx_vec[0];
1374 		__le32 usig_a2 = phy_data->rx_vec[1];
1375 
1376 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a1,
1377 					    IWL_RX_USIG_A1_DISREGARD,
1378 					    IEEE80211_RADIOTAP_EHT_USIG1_MU_B20_B24_DISREGARD);
1379 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a1,
1380 					    IWL_RX_USIG_A1_VALIDATE,
1381 					    IEEE80211_RADIOTAP_EHT_USIG1_MU_B25_VALIDATE);
1382 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1383 					    IWL_RX_USIG_A2_EHT_PPDU_TYPE,
1384 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B0_B1_PPDU_TYPE);
1385 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1386 					    IWL_RX_USIG_A2_EHT_USIG2_VALIDATE_B2,
1387 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B2_VALIDATE);
1388 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1389 					    IWL_RX_USIG_A2_EHT_PUNC_CHANNEL,
1390 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B3_B7_PUNCTURED_INFO);
1391 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1392 					    IWL_RX_USIG_A2_EHT_USIG2_VALIDATE_B8,
1393 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B8_VALIDATE);
1394 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1395 					    IWL_RX_USIG_A2_EHT_SIG_MCS,
1396 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B9_B10_SIG_MCS);
1397 		IWL_MVM_ENC_USIG_VALUE_MASK
1398 			(usig, usig_a2, IWL_RX_USIG_A2_EHT_SIG_SYM_NUM,
1399 			 IEEE80211_RADIOTAP_EHT_USIG2_MU_B11_B15_EHT_SIG_SYMBOLS);
1400 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1401 					    IWL_RX_USIG_A2_EHT_CRC_OK,
1402 					    IEEE80211_RADIOTAP_EHT_USIG2_MU_B16_B19_CRC);
1403 	}
1404 }
1405 
1406 static void iwl_mvm_decode_eht_ext_tb(struct iwl_mvm *mvm,
1407 				      struct iwl_mvm_rx_phy_data *phy_data,
1408 				      struct ieee80211_rx_status *rx_status,
1409 				      struct ieee80211_radiotap_eht *eht,
1410 				      struct ieee80211_radiotap_eht_usig *usig)
1411 {
1412 	if (phy_data->with_data) {
1413 		__le32 data5 = phy_data->d5;
1414 
1415 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data5,
1416 					    IWL_RX_PHY_DATA5_EHT_TYPE_AND_COMP,
1417 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B0_B1_PPDU_TYPE);
1418 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data5,
1419 					    IWL_RX_PHY_DATA5_EHT_TB_SPATIAL_REUSE1,
1420 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B3_B6_SPATIAL_REUSE_1);
1421 
1422 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, data5,
1423 					    IWL_RX_PHY_DATA5_EHT_TB_SPATIAL_REUSE2,
1424 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B7_B10_SPATIAL_REUSE_2);
1425 	} else {
1426 		__le32 usig_a1 = phy_data->rx_vec[0];
1427 		__le32 usig_a2 = phy_data->rx_vec[1];
1428 
1429 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a1,
1430 					    IWL_RX_USIG_A1_DISREGARD,
1431 					    IEEE80211_RADIOTAP_EHT_USIG1_TB_B20_B25_DISREGARD);
1432 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1433 					    IWL_RX_USIG_A2_EHT_PPDU_TYPE,
1434 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B0_B1_PPDU_TYPE);
1435 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1436 					    IWL_RX_USIG_A2_EHT_USIG2_VALIDATE_B2,
1437 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B2_VALIDATE);
1438 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1439 					    IWL_RX_USIG_A2_EHT_TRIG_SPATIAL_REUSE_1,
1440 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B3_B6_SPATIAL_REUSE_1);
1441 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1442 					    IWL_RX_USIG_A2_EHT_TRIG_SPATIAL_REUSE_2,
1443 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B7_B10_SPATIAL_REUSE_2);
1444 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1445 					    IWL_RX_USIG_A2_EHT_TRIG_USIG2_DISREGARD,
1446 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B11_B15_DISREGARD);
1447 		IWL_MVM_ENC_USIG_VALUE_MASK(usig, usig_a2,
1448 					    IWL_RX_USIG_A2_EHT_CRC_OK,
1449 					    IEEE80211_RADIOTAP_EHT_USIG2_TB_B16_B19_CRC);
1450 	}
1451 }
1452 
1453 static void iwl_mvm_decode_eht_ru(struct iwl_mvm *mvm,
1454 				  struct ieee80211_rx_status *rx_status,
1455 				  struct ieee80211_radiotap_eht *eht)
1456 {
1457 	u32 ru = le32_get_bits(eht->data[8],
1458 			       IEEE80211_RADIOTAP_EHT_DATA8_RU_ALLOC_TB_FMT_B7_B1);
1459 	enum nl80211_eht_ru_alloc nl_ru;
1460 
1461 	/* Using D1.5 Table 9-53a - Encoding of PS160 and RU Allocation subfields
1462 	 * in an EHT variant User Info field
1463 	 */
1464 
1465 	switch (ru) {
1466 	case 0 ... 36:
1467 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_26;
1468 		break;
1469 	case 37 ... 52:
1470 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_52;
1471 		break;
1472 	case 53 ... 60:
1473 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_106;
1474 		break;
1475 	case 61 ... 64:
1476 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_242;
1477 		break;
1478 	case 65 ... 66:
1479 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_484;
1480 		break;
1481 	case 67:
1482 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_996;
1483 		break;
1484 	case 68:
1485 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_2x996;
1486 		break;
1487 	case 69:
1488 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_4x996;
1489 		break;
1490 	case 70 ... 81:
1491 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_52P26;
1492 		break;
1493 	case 82 ... 89:
1494 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_106P26;
1495 		break;
1496 	case 90 ... 93:
1497 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_484P242;
1498 		break;
1499 	case 94 ... 95:
1500 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_996P484;
1501 		break;
1502 	case 96 ... 99:
1503 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_996P484P242;
1504 		break;
1505 	case 100 ... 103:
1506 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_2x996P484;
1507 		break;
1508 	case 104:
1509 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_3x996;
1510 		break;
1511 	case 105 ... 106:
1512 		nl_ru = NL80211_RATE_INFO_EHT_RU_ALLOC_3x996P484;
1513 		break;
1514 	default:
1515 		return;
1516 	}
1517 
1518 	rx_status->bw = RATE_INFO_BW_EHT_RU;
1519 	rx_status->eht.ru = nl_ru;
1520 }
1521 
1522 static void iwl_mvm_decode_eht_phy_data(struct iwl_mvm *mvm,
1523 					struct iwl_mvm_rx_phy_data *phy_data,
1524 					struct ieee80211_rx_status *rx_status,
1525 					struct ieee80211_radiotap_eht *eht,
1526 					struct ieee80211_radiotap_eht_usig *usig)
1527 
1528 {
1529 	__le32 data0 = phy_data->d0;
1530 	__le32 data1 = phy_data->d1;
1531 	__le32 usig_a1 = phy_data->rx_vec[0];
1532 	u8 info_type = phy_data->info_type;
1533 
1534 	/* Not in EHT range */
1535 	if (info_type < IWL_RX_PHY_INFO_TYPE_EHT_MU ||
1536 	    info_type > IWL_RX_PHY_INFO_TYPE_EHT_TB_EXT)
1537 		return;
1538 
1539 	usig->common |= cpu_to_le32
1540 		(IEEE80211_RADIOTAP_EHT_USIG_COMMON_UL_DL_KNOWN |
1541 		 IEEE80211_RADIOTAP_EHT_USIG_COMMON_BSS_COLOR_KNOWN);
1542 	if (phy_data->with_data) {
1543 		usig->common |= LE32_DEC_ENC(data0,
1544 					     IWL_RX_PHY_DATA0_EHT_UPLINK,
1545 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_UL_DL);
1546 		usig->common |= LE32_DEC_ENC(data0,
1547 					     IWL_RX_PHY_DATA0_EHT_BSS_COLOR_MASK,
1548 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_BSS_COLOR);
1549 	} else {
1550 		usig->common |= LE32_DEC_ENC(usig_a1,
1551 					     IWL_RX_USIG_A1_UL_FLAG,
1552 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_UL_DL);
1553 		usig->common |= LE32_DEC_ENC(usig_a1,
1554 					     IWL_RX_USIG_A1_BSS_COLOR,
1555 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_BSS_COLOR);
1556 	}
1557 
1558 	if (fw_has_capa(&mvm->fw->ucode_capa,
1559 			IWL_UCODE_TLV_CAPA_SNIFF_VALIDATE_SUPPORT)) {
1560 		usig->common |=
1561 			cpu_to_le32(IEEE80211_RADIOTAP_EHT_USIG_COMMON_VALIDATE_BITS_CHECKED);
1562 		usig->common |=
1563 			LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_VALIDATE,
1564 				     IEEE80211_RADIOTAP_EHT_USIG_COMMON_VALIDATE_BITS_OK);
1565 	}
1566 
1567 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_SPATIAL_REUSE);
1568 	eht->data[0] |= LE32_DEC_ENC(data0,
1569 				     IWL_RX_PHY_DATA0_ETH_SPATIAL_REUSE_MASK,
1570 				     IEEE80211_RADIOTAP_EHT_DATA0_SPATIAL_REUSE);
1571 
1572 	/* All RU allocating size/index is in TB format */
1573 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_RU_ALLOC_TB_FMT);
1574 	eht->data[8] |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_PS160,
1575 				     IEEE80211_RADIOTAP_EHT_DATA8_RU_ALLOC_TB_FMT_PS_160);
1576 	eht->data[8] |= LE32_DEC_ENC(data1, IWL_RX_PHY_DATA1_EHT_RU_ALLOC_B0,
1577 				     IEEE80211_RADIOTAP_EHT_DATA8_RU_ALLOC_TB_FMT_B0);
1578 	eht->data[8] |= LE32_DEC_ENC(data1, IWL_RX_PHY_DATA1_EHT_RU_ALLOC_B1_B7,
1579 				     IEEE80211_RADIOTAP_EHT_DATA8_RU_ALLOC_TB_FMT_B7_B1);
1580 
1581 	iwl_mvm_decode_eht_ru(mvm, rx_status, eht);
1582 
1583 	/* We only get here in case of IWL_RX_MPDU_PHY_TSF_OVERLOAD is set
1584 	 * which is on only in case of monitor mode so no need to check monitor
1585 	 * mode
1586 	 */
1587 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_PRIMARY_80);
1588 	eht->data[1] |=
1589 		le32_encode_bits(mvm->monitor_p80,
1590 				 IEEE80211_RADIOTAP_EHT_DATA1_PRIMARY_80);
1591 
1592 	usig->common |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_USIG_COMMON_TXOP_KNOWN);
1593 	if (phy_data->with_data)
1594 		usig->common |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_TXOP_DUR_MASK,
1595 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_TXOP);
1596 	else
1597 		usig->common |= LE32_DEC_ENC(usig_a1, IWL_RX_USIG_A1_TXOP_DURATION,
1598 					     IEEE80211_RADIOTAP_EHT_USIG_COMMON_TXOP);
1599 
1600 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_LDPC_EXTRA_SYM_OM);
1601 	eht->data[0] |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_LDPC_EXT_SYM,
1602 				     IEEE80211_RADIOTAP_EHT_DATA0_LDPC_EXTRA_SYM_OM);
1603 
1604 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_PRE_PADD_FACOR_OM);
1605 	eht->data[0] |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_PRE_FEC_PAD_MASK,
1606 				    IEEE80211_RADIOTAP_EHT_DATA0_PRE_PADD_FACOR_OM);
1607 
1608 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_PE_DISAMBIGUITY_OM);
1609 	eht->data[0] |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_PE_DISAMBIG,
1610 				     IEEE80211_RADIOTAP_EHT_DATA0_PE_DISAMBIGUITY_OM);
1611 
1612 	/* TODO: what about IWL_RX_PHY_DATA0_EHT_BW320_SLOT */
1613 
1614 	if (!le32_get_bits(data0, IWL_RX_PHY_DATA0_EHT_SIGA_CRC_OK))
1615 		usig->common |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_USIG_COMMON_BAD_USIG_CRC);
1616 
1617 	usig->common |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_USIG_COMMON_PHY_VER_KNOWN);
1618 	usig->common |= LE32_DEC_ENC(data0, IWL_RX_PHY_DATA0_EHT_PHY_VER,
1619 				     IEEE80211_RADIOTAP_EHT_USIG_COMMON_PHY_VER);
1620 
1621 	/*
1622 	 * TODO: what about TB - IWL_RX_PHY_DATA1_EHT_TB_PILOT_TYPE,
1623 	 *			 IWL_RX_PHY_DATA1_EHT_TB_LOW_SS
1624 	 */
1625 
1626 	eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_EHT_LTF);
1627 	eht->data[0] |= LE32_DEC_ENC(data1, IWL_RX_PHY_DATA1_EHT_SIG_LTF_NUM,
1628 				     IEEE80211_RADIOTAP_EHT_DATA0_EHT_LTF);
1629 
1630 	if (info_type == IWL_RX_PHY_INFO_TYPE_EHT_TB_EXT ||
1631 	    info_type == IWL_RX_PHY_INFO_TYPE_EHT_TB)
1632 		iwl_mvm_decode_eht_ext_tb(mvm, phy_data, rx_status, eht, usig);
1633 
1634 	if (info_type == IWL_RX_PHY_INFO_TYPE_EHT_MU_EXT ||
1635 	    info_type == IWL_RX_PHY_INFO_TYPE_EHT_MU)
1636 		iwl_mvm_decode_eht_ext_mu(mvm, phy_data, rx_status, eht, usig);
1637 }
1638 
1639 static void iwl_mvm_rx_eht(struct iwl_mvm *mvm, struct sk_buff *skb,
1640 			   struct iwl_mvm_rx_phy_data *phy_data,
1641 			   int queue)
1642 {
1643 	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1644 
1645 	struct ieee80211_radiotap_eht *eht;
1646 	struct ieee80211_radiotap_eht_usig *usig;
1647 	size_t eht_len = sizeof(*eht);
1648 
1649 	u32 rate_n_flags = phy_data->rate_n_flags;
1650 	u32 he_type = rate_n_flags & RATE_MCS_HE_TYPE_MSK;
1651 	/* EHT and HE have the same valus for LTF */
1652 	u8 ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_UNKNOWN;
1653 	u16 phy_info = phy_data->phy_info;
1654 	u32 bw;
1655 
1656 	/* u32 for 1 user_info */
1657 	if (phy_data->with_data)
1658 		eht_len += sizeof(u32);
1659 
1660 	eht = iwl_mvm_radiotap_put_tlv(skb, IEEE80211_RADIOTAP_EHT, eht_len);
1661 
1662 	usig = iwl_mvm_radiotap_put_tlv(skb, IEEE80211_RADIOTAP_EHT_USIG,
1663 					sizeof(*usig));
1664 	rx_status->flag |= RX_FLAG_RADIOTAP_TLV_AT_END;
1665 	usig->common |=
1666 		cpu_to_le32(IEEE80211_RADIOTAP_EHT_USIG_COMMON_BW_KNOWN);
1667 
1668 	/* specific handling for 320MHz */
1669 	bw = FIELD_GET(RATE_MCS_CHAN_WIDTH_MSK, rate_n_flags);
1670 	if (bw == RATE_MCS_CHAN_WIDTH_320_VAL)
1671 		bw += FIELD_GET(IWL_RX_PHY_DATA0_EHT_BW320_SLOT,
1672 				le32_to_cpu(phy_data->d0));
1673 
1674 	usig->common |= cpu_to_le32
1675 		(FIELD_PREP(IEEE80211_RADIOTAP_EHT_USIG_COMMON_BW, bw));
1676 
1677 	/* report the AMPDU-EOF bit on single frames */
1678 	if (!queue && !(phy_info & IWL_RX_MPDU_PHY_AMPDU)) {
1679 		rx_status->flag |= RX_FLAG_AMPDU_DETAILS;
1680 		rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT_KNOWN;
1681 		if (phy_data->d0 & cpu_to_le32(IWL_RX_PHY_DATA0_EHT_DELIM_EOF))
1682 			rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT;
1683 	}
1684 
1685 	/* update aggregation data for monitor sake on default queue */
1686 	if (!queue && (phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD) &&
1687 	    (phy_info & IWL_RX_MPDU_PHY_AMPDU) && phy_data->first_subframe) {
1688 		rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT_KNOWN;
1689 		if (phy_data->d0 & cpu_to_le32(IWL_RX_PHY_DATA0_EHT_DELIM_EOF))
1690 			rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT;
1691 	}
1692 
1693 	if (phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD)
1694 		iwl_mvm_decode_eht_phy_data(mvm, phy_data, rx_status, eht, usig);
1695 
1696 #define CHECK_TYPE(F)							\
1697 	BUILD_BUG_ON(IEEE80211_RADIOTAP_HE_DATA1_FORMAT_ ## F !=	\
1698 		     (RATE_MCS_HE_TYPE_ ## F >> RATE_MCS_HE_TYPE_POS))
1699 
1700 	CHECK_TYPE(SU);
1701 	CHECK_TYPE(EXT_SU);
1702 	CHECK_TYPE(MU);
1703 	CHECK_TYPE(TRIG);
1704 
1705 	switch (FIELD_GET(RATE_MCS_HE_GI_LTF_MSK, rate_n_flags)) {
1706 	case 0:
1707 		if (he_type == RATE_MCS_HE_TYPE_TRIG) {
1708 			rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_1_6;
1709 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_1X;
1710 		} else {
1711 			rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_0_8;
1712 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_2X;
1713 		}
1714 		break;
1715 	case 1:
1716 		rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_1_6;
1717 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_2X;
1718 		break;
1719 	case 2:
1720 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1721 		if (he_type == RATE_MCS_HE_TYPE_TRIG)
1722 			rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_3_2;
1723 		else
1724 			rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_0_8;
1725 		break;
1726 	case 3:
1727 		if (he_type != RATE_MCS_HE_TYPE_TRIG) {
1728 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1729 			rx_status->eht.gi = NL80211_RATE_INFO_EHT_GI_3_2;
1730 		}
1731 		break;
1732 	default:
1733 		/* nothing here */
1734 		break;
1735 	}
1736 
1737 	if (ltf != IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_UNKNOWN) {
1738 		eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_GI);
1739 		eht->data[0] |= cpu_to_le32
1740 			(FIELD_PREP(IEEE80211_RADIOTAP_EHT_DATA0_LTF,
1741 				    ltf) |
1742 			 FIELD_PREP(IEEE80211_RADIOTAP_EHT_DATA0_GI,
1743 				    rx_status->eht.gi));
1744 	}
1745 
1746 
1747 	if (!phy_data->with_data) {
1748 		eht->known |= cpu_to_le32(IEEE80211_RADIOTAP_EHT_KNOWN_NSS_S |
1749 					  IEEE80211_RADIOTAP_EHT_KNOWN_BEAMFORMED_S);
1750 		eht->data[7] |=
1751 			le32_encode_bits(le32_get_bits(phy_data->rx_vec[2],
1752 						       RX_NO_DATA_RX_VEC2_EHT_NSTS_MSK),
1753 					 IEEE80211_RADIOTAP_EHT_DATA7_NSS_S);
1754 		if (rate_n_flags & RATE_MCS_BF_MSK)
1755 			eht->data[7] |=
1756 				cpu_to_le32(IEEE80211_RADIOTAP_EHT_DATA7_BEAMFORMED_S);
1757 	} else {
1758 		eht->user_info[0] |=
1759 			cpu_to_le32(IEEE80211_RADIOTAP_EHT_USER_INFO_MCS_KNOWN |
1760 				    IEEE80211_RADIOTAP_EHT_USER_INFO_CODING_KNOWN |
1761 				    IEEE80211_RADIOTAP_EHT_USER_INFO_NSS_KNOWN_O |
1762 				    IEEE80211_RADIOTAP_EHT_USER_INFO_BEAMFORMING_KNOWN_O |
1763 				    IEEE80211_RADIOTAP_EHT_USER_INFO_DATA_FOR_USER);
1764 
1765 		if (rate_n_flags & RATE_MCS_BF_MSK)
1766 			eht->user_info[0] |=
1767 				cpu_to_le32(IEEE80211_RADIOTAP_EHT_USER_INFO_BEAMFORMING_O);
1768 
1769 		if (rate_n_flags & RATE_MCS_LDPC_MSK)
1770 			eht->user_info[0] |=
1771 				cpu_to_le32(IEEE80211_RADIOTAP_EHT_USER_INFO_CODING);
1772 
1773 		eht->user_info[0] |= cpu_to_le32
1774 			(FIELD_PREP(IEEE80211_RADIOTAP_EHT_USER_INFO_MCS,
1775 				    FIELD_GET(RATE_VHT_MCS_RATE_CODE_MSK,
1776 					      rate_n_flags)) |
1777 			 FIELD_PREP(IEEE80211_RADIOTAP_EHT_USER_INFO_NSS_O,
1778 				    FIELD_GET(RATE_MCS_NSS_MSK, rate_n_flags)));
1779 	}
1780 }
1781 
1782 static void iwl_mvm_rx_he(struct iwl_mvm *mvm, struct sk_buff *skb,
1783 			  struct iwl_mvm_rx_phy_data *phy_data,
1784 			  int queue)
1785 {
1786 	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1787 	struct ieee80211_radiotap_he *he = NULL;
1788 	struct ieee80211_radiotap_he_mu *he_mu = NULL;
1789 	u32 rate_n_flags = phy_data->rate_n_flags;
1790 	u32 he_type = rate_n_flags & RATE_MCS_HE_TYPE_MSK;
1791 	u8 ltf;
1792 	static const struct ieee80211_radiotap_he known = {
1793 		.data1 = cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_DATA_MCS_KNOWN |
1794 				     IEEE80211_RADIOTAP_HE_DATA1_DATA_DCM_KNOWN |
1795 				     IEEE80211_RADIOTAP_HE_DATA1_STBC_KNOWN |
1796 				     IEEE80211_RADIOTAP_HE_DATA1_CODING_KNOWN),
1797 		.data2 = cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA2_GI_KNOWN |
1798 				     IEEE80211_RADIOTAP_HE_DATA2_TXBF_KNOWN),
1799 	};
1800 	static const struct ieee80211_radiotap_he_mu mu_known = {
1801 		.flags1 = cpu_to_le16(IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_MCS_KNOWN |
1802 				      IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_DCM_KNOWN |
1803 				      IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_SYMS_USERS_KNOWN |
1804 				      IEEE80211_RADIOTAP_HE_MU_FLAGS1_SIG_B_COMP_KNOWN),
1805 		.flags2 = cpu_to_le16(IEEE80211_RADIOTAP_HE_MU_FLAGS2_PUNC_FROM_SIG_A_BW_KNOWN |
1806 				      IEEE80211_RADIOTAP_HE_MU_FLAGS2_BW_FROM_SIG_A_BW_KNOWN),
1807 	};
1808 	u16 phy_info = phy_data->phy_info;
1809 
1810 	he = skb_put_data(skb, &known, sizeof(known));
1811 	rx_status->flag |= RX_FLAG_RADIOTAP_HE;
1812 
1813 	if (phy_data->info_type == IWL_RX_PHY_INFO_TYPE_HE_MU ||
1814 	    phy_data->info_type == IWL_RX_PHY_INFO_TYPE_HE_MU_EXT) {
1815 		he_mu = skb_put_data(skb, &mu_known, sizeof(mu_known));
1816 		rx_status->flag |= RX_FLAG_RADIOTAP_HE_MU;
1817 	}
1818 
1819 	/* report the AMPDU-EOF bit on single frames */
1820 	if (!queue && !(phy_info & IWL_RX_MPDU_PHY_AMPDU)) {
1821 		rx_status->flag |= RX_FLAG_AMPDU_DETAILS;
1822 		rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT_KNOWN;
1823 		if (phy_data->d0 & cpu_to_le32(IWL_RX_PHY_DATA0_HE_DELIM_EOF))
1824 			rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT;
1825 	}
1826 
1827 	if (phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD)
1828 		iwl_mvm_decode_he_phy_data(mvm, phy_data, he, he_mu, rx_status,
1829 					   queue);
1830 
1831 	/* update aggregation data for monitor sake on default queue */
1832 	if (!queue && (phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD) &&
1833 	    (phy_info & IWL_RX_MPDU_PHY_AMPDU) && phy_data->first_subframe) {
1834 		rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT_KNOWN;
1835 		if (phy_data->d0 & cpu_to_le32(IWL_RX_PHY_DATA0_EHT_DELIM_EOF))
1836 			rx_status->flag |= RX_FLAG_AMPDU_EOF_BIT;
1837 	}
1838 
1839 	if (he_type == RATE_MCS_HE_TYPE_EXT_SU &&
1840 	    rate_n_flags & RATE_MCS_HE_106T_MSK) {
1841 		rx_status->bw = RATE_INFO_BW_HE_RU;
1842 		rx_status->he_ru = NL80211_RATE_INFO_HE_RU_ALLOC_106;
1843 	}
1844 
1845 	/* actually data is filled in mac80211 */
1846 	if (he_type == RATE_MCS_HE_TYPE_SU ||
1847 	    he_type == RATE_MCS_HE_TYPE_EXT_SU)
1848 		he->data1 |=
1849 			cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA1_BW_RU_ALLOC_KNOWN);
1850 
1851 #define CHECK_TYPE(F)							\
1852 	BUILD_BUG_ON(IEEE80211_RADIOTAP_HE_DATA1_FORMAT_ ## F !=	\
1853 		     (RATE_MCS_HE_TYPE_ ## F >> RATE_MCS_HE_TYPE_POS))
1854 
1855 	CHECK_TYPE(SU);
1856 	CHECK_TYPE(EXT_SU);
1857 	CHECK_TYPE(MU);
1858 	CHECK_TYPE(TRIG);
1859 
1860 	he->data1 |= cpu_to_le16(he_type >> RATE_MCS_HE_TYPE_POS);
1861 
1862 	if (rate_n_flags & RATE_MCS_BF_MSK)
1863 		he->data5 |= cpu_to_le16(IEEE80211_RADIOTAP_HE_DATA5_TXBF);
1864 
1865 	switch ((rate_n_flags & RATE_MCS_HE_GI_LTF_MSK) >>
1866 		RATE_MCS_HE_GI_LTF_POS) {
1867 	case 0:
1868 		if (he_type == RATE_MCS_HE_TYPE_TRIG)
1869 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
1870 		else
1871 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
1872 		if (he_type == RATE_MCS_HE_TYPE_MU)
1873 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1874 		else
1875 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_1X;
1876 		break;
1877 	case 1:
1878 		if (he_type == RATE_MCS_HE_TYPE_TRIG)
1879 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
1880 		else
1881 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
1882 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_2X;
1883 		break;
1884 	case 2:
1885 		if (he_type == RATE_MCS_HE_TYPE_TRIG) {
1886 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_3_2;
1887 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1888 		} else {
1889 			rx_status->he_gi = NL80211_RATE_INFO_HE_GI_1_6;
1890 			ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_2X;
1891 		}
1892 		break;
1893 	case 3:
1894 		rx_status->he_gi = NL80211_RATE_INFO_HE_GI_3_2;
1895 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1896 		break;
1897 	case 4:
1898 		rx_status->he_gi = NL80211_RATE_INFO_HE_GI_0_8;
1899 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_4X;
1900 		break;
1901 	default:
1902 		ltf = IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE_UNKNOWN;
1903 	}
1904 
1905 	he->data5 |= le16_encode_bits(ltf,
1906 				      IEEE80211_RADIOTAP_HE_DATA5_LTF_SIZE);
1907 }
1908 
1909 static void iwl_mvm_decode_lsig(struct sk_buff *skb,
1910 				struct iwl_mvm_rx_phy_data *phy_data)
1911 {
1912 	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1913 	struct ieee80211_radiotap_lsig *lsig;
1914 
1915 	switch (phy_data->info_type) {
1916 	case IWL_RX_PHY_INFO_TYPE_HT:
1917 	case IWL_RX_PHY_INFO_TYPE_VHT_SU:
1918 	case IWL_RX_PHY_INFO_TYPE_VHT_MU:
1919 	case IWL_RX_PHY_INFO_TYPE_HE_TB_EXT:
1920 	case IWL_RX_PHY_INFO_TYPE_HE_SU:
1921 	case IWL_RX_PHY_INFO_TYPE_HE_MU:
1922 	case IWL_RX_PHY_INFO_TYPE_HE_MU_EXT:
1923 	case IWL_RX_PHY_INFO_TYPE_HE_TB:
1924 	case IWL_RX_PHY_INFO_TYPE_EHT_MU:
1925 	case IWL_RX_PHY_INFO_TYPE_EHT_TB:
1926 	case IWL_RX_PHY_INFO_TYPE_EHT_MU_EXT:
1927 	case IWL_RX_PHY_INFO_TYPE_EHT_TB_EXT:
1928 		lsig = skb_put(skb, sizeof(*lsig));
1929 		lsig->data1 = cpu_to_le16(IEEE80211_RADIOTAP_LSIG_DATA1_LENGTH_KNOWN);
1930 		lsig->data2 = le16_encode_bits(le32_get_bits(phy_data->d1,
1931 							     IWL_RX_PHY_DATA1_LSIG_LEN_MASK),
1932 					       IEEE80211_RADIOTAP_LSIG_DATA2_LENGTH);
1933 		rx_status->flag |= RX_FLAG_RADIOTAP_LSIG;
1934 		break;
1935 	default:
1936 		break;
1937 	}
1938 }
1939 
1940 struct iwl_rx_sta_csa {
1941 	bool all_sta_unblocked;
1942 	struct ieee80211_vif *vif;
1943 };
1944 
1945 static void iwl_mvm_rx_get_sta_block_tx(void *data, struct ieee80211_sta *sta)
1946 {
1947 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1948 	struct iwl_rx_sta_csa *rx_sta_csa = data;
1949 
1950 	if (mvmsta->vif != rx_sta_csa->vif)
1951 		return;
1952 
1953 	if (mvmsta->disable_tx)
1954 		rx_sta_csa->all_sta_unblocked = false;
1955 }
1956 
1957 /*
1958  * Note: requires also rx_status->band to be prefilled, as well
1959  * as phy_data (apart from phy_data->info_type)
1960  * Note: desc/hdr may be NULL
1961  */
1962 static void iwl_mvm_rx_fill_status(struct iwl_mvm *mvm,
1963 				   struct iwl_rx_mpdu_desc *desc,
1964 				   struct ieee80211_hdr *hdr,
1965 				   struct sk_buff *skb,
1966 				   struct iwl_mvm_rx_phy_data *phy_data,
1967 				   int queue)
1968 {
1969 	struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1970 	u32 rate_n_flags = phy_data->rate_n_flags;
1971 	u8 stbc = u32_get_bits(rate_n_flags, RATE_MCS_STBC_MSK);
1972 	u32 format = rate_n_flags & RATE_MCS_MOD_TYPE_MSK;
1973 	bool is_sgi;
1974 
1975 	phy_data->info_type = IWL_RX_PHY_INFO_TYPE_NONE;
1976 
1977 	if (phy_data->phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD)
1978 		phy_data->info_type =
1979 			le32_get_bits(phy_data->d1,
1980 				      IWL_RX_PHY_DATA1_INFO_TYPE_MASK);
1981 
1982 	/* This may be overridden by iwl_mvm_rx_he() to HE_RU */
1983 	switch (rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK) {
1984 	case RATE_MCS_CHAN_WIDTH_20:
1985 		break;
1986 	case RATE_MCS_CHAN_WIDTH_40:
1987 		rx_status->bw = RATE_INFO_BW_40;
1988 		break;
1989 	case RATE_MCS_CHAN_WIDTH_80:
1990 		rx_status->bw = RATE_INFO_BW_80;
1991 		break;
1992 	case RATE_MCS_CHAN_WIDTH_160:
1993 		rx_status->bw = RATE_INFO_BW_160;
1994 		break;
1995 	case RATE_MCS_CHAN_WIDTH_320:
1996 		rx_status->bw = RATE_INFO_BW_320;
1997 		break;
1998 	}
1999 
2000 	/* must be before L-SIG data */
2001 	if (format == RATE_MCS_MOD_TYPE_HE)
2002 		iwl_mvm_rx_he(mvm, skb, phy_data, queue);
2003 
2004 	iwl_mvm_decode_lsig(skb, phy_data);
2005 
2006 	rx_status->device_timestamp = phy_data->gp2_on_air_rise;
2007 
2008 	if (mvm->rx_ts_ptp && mvm->monitor_on) {
2009 		u64 adj_time =
2010 			iwl_mvm_ptp_get_adj_time(mvm, phy_data->gp2_on_air_rise * NSEC_PER_USEC);
2011 
2012 		rx_status->mactime = div64_u64(adj_time, NSEC_PER_USEC);
2013 		rx_status->flag |= RX_FLAG_MACTIME_IS_RTAP_TS64;
2014 		rx_status->flag &= ~RX_FLAG_MACTIME;
2015 	}
2016 
2017 	rx_status->freq = ieee80211_channel_to_frequency(phy_data->channel,
2018 							 rx_status->band);
2019 	iwl_mvm_get_signal_strength(mvm, desc, hdr, rx_status, rate_n_flags,
2020 				    phy_data->energy_a, phy_data->energy_b);
2021 
2022 	/* using TLV format and must be after all fixed len fields */
2023 	if (format == RATE_MCS_MOD_TYPE_EHT)
2024 		iwl_mvm_rx_eht(mvm, skb, phy_data, queue);
2025 
2026 	if (unlikely(mvm->monitor_on))
2027 		iwl_mvm_add_rtap_sniffer_config(mvm, skb);
2028 
2029 	is_sgi = format == RATE_MCS_MOD_TYPE_HE ?
2030 		iwl_he_is_sgi(rate_n_flags) :
2031 		rate_n_flags & RATE_MCS_SGI_MSK;
2032 
2033 	if (!(format == RATE_MCS_MOD_TYPE_CCK) && is_sgi)
2034 		rx_status->enc_flags |= RX_ENC_FLAG_SHORT_GI;
2035 
2036 	if (rate_n_flags & RATE_MCS_LDPC_MSK)
2037 		rx_status->enc_flags |= RX_ENC_FLAG_LDPC;
2038 
2039 	switch (format) {
2040 	case RATE_MCS_MOD_TYPE_VHT:
2041 		rx_status->encoding = RX_ENC_VHT;
2042 		break;
2043 	case RATE_MCS_MOD_TYPE_HE:
2044 		rx_status->encoding = RX_ENC_HE;
2045 		rx_status->he_dcm =
2046 			!!(rate_n_flags & RATE_HE_DUAL_CARRIER_MODE_MSK);
2047 		break;
2048 	case RATE_MCS_MOD_TYPE_EHT:
2049 		rx_status->encoding = RX_ENC_EHT;
2050 		break;
2051 	}
2052 
2053 	switch (format) {
2054 	case RATE_MCS_MOD_TYPE_HT:
2055 		rx_status->encoding = RX_ENC_HT;
2056 		rx_status->rate_idx = RATE_HT_MCS_INDEX(rate_n_flags);
2057 		rx_status->enc_flags |= stbc << RX_ENC_FLAG_STBC_SHIFT;
2058 		break;
2059 	case RATE_MCS_MOD_TYPE_VHT:
2060 	case RATE_MCS_MOD_TYPE_HE:
2061 	case RATE_MCS_MOD_TYPE_EHT:
2062 		rx_status->nss =
2063 			u32_get_bits(rate_n_flags, RATE_MCS_NSS_MSK) + 1;
2064 		rx_status->rate_idx = rate_n_flags & RATE_MCS_CODE_MSK;
2065 		rx_status->enc_flags |= stbc << RX_ENC_FLAG_STBC_SHIFT;
2066 		break;
2067 	default: {
2068 		int rate = iwl_mvm_legacy_hw_idx_to_mac80211_idx(rate_n_flags,
2069 								 rx_status->band);
2070 
2071 		rx_status->rate_idx = rate;
2072 
2073 		if ((rate < 0 || rate > 0xFF)) {
2074 			rx_status->rate_idx = 0;
2075 			if (net_ratelimit())
2076 				IWL_ERR(mvm, "Invalid rate flags 0x%x, band %d,\n",
2077 					rate_n_flags, rx_status->band);
2078 		}
2079 
2080 		break;
2081 		}
2082 	}
2083 }
2084 
2085 void iwl_mvm_rx_mpdu_mq(struct iwl_mvm *mvm, struct napi_struct *napi,
2086 			struct iwl_rx_cmd_buffer *rxb, int queue)
2087 {
2088 	struct ieee80211_rx_status *rx_status;
2089 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
2090 	struct iwl_rx_mpdu_desc *desc = (void *)pkt->data;
2091 	struct ieee80211_hdr *hdr;
2092 	u32 len;
2093 	u32 pkt_len = iwl_rx_packet_payload_len(pkt);
2094 	struct ieee80211_sta *sta = NULL;
2095 	struct sk_buff *skb;
2096 	u8 crypt_len = 0;
2097 	u8 sta_id = le32_get_bits(desc->status, IWL_RX_MPDU_STATUS_STA_ID);
2098 	size_t desc_size;
2099 	struct iwl_mvm_rx_phy_data phy_data = {};
2100 	u32 format;
2101 
2102 	if (unlikely(test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)))
2103 		return;
2104 
2105 	if (mvm->trans->mac_cfg->device_family >= IWL_DEVICE_FAMILY_AX210)
2106 		desc_size = sizeof(*desc);
2107 	else
2108 		desc_size = IWL_RX_DESC_SIZE_V1;
2109 
2110 	if (unlikely(pkt_len < desc_size)) {
2111 		IWL_DEBUG_DROP(mvm, "Bad REPLY_RX_MPDU_CMD size\n");
2112 		return;
2113 	}
2114 
2115 	if (mvm->trans->mac_cfg->device_family >= IWL_DEVICE_FAMILY_AX210) {
2116 		phy_data.rate_n_flags =
2117 			iwl_mvm_v3_rate_from_fw(desc->v3.rate_n_flags,
2118 						mvm->fw_rates_ver);
2119 		phy_data.channel = desc->v3.channel;
2120 		phy_data.gp2_on_air_rise = le32_to_cpu(desc->v3.gp2_on_air_rise);
2121 		phy_data.energy_a = desc->v3.energy_a;
2122 		phy_data.energy_b = desc->v3.energy_b;
2123 
2124 		phy_data.d0 = desc->v3.phy_data0;
2125 		phy_data.d1 = desc->v3.phy_data1;
2126 		phy_data.d2 = desc->v3.phy_data2;
2127 		phy_data.d3 = desc->v3.phy_data3;
2128 		phy_data.eht_d4 = desc->phy_eht_data4;
2129 		phy_data.d5 = desc->v3.phy_data5;
2130 	} else {
2131 		phy_data.rate_n_flags =
2132 			iwl_mvm_v3_rate_from_fw(desc->v1.rate_n_flags,
2133 						mvm->fw_rates_ver);
2134 		phy_data.channel = desc->v1.channel;
2135 		phy_data.gp2_on_air_rise = le32_to_cpu(desc->v1.gp2_on_air_rise);
2136 		phy_data.energy_a = desc->v1.energy_a;
2137 		phy_data.energy_b = desc->v1.energy_b;
2138 
2139 		phy_data.d0 = desc->v1.phy_data0;
2140 		phy_data.d1 = desc->v1.phy_data1;
2141 		phy_data.d2 = desc->v1.phy_data2;
2142 		phy_data.d3 = desc->v1.phy_data3;
2143 	}
2144 
2145 	format = phy_data.rate_n_flags & RATE_MCS_MOD_TYPE_MSK;
2146 
2147 	len = le16_to_cpu(desc->mpdu_len);
2148 
2149 	if (unlikely(len + desc_size > pkt_len)) {
2150 		IWL_DEBUG_DROP(mvm, "FW lied about packet len\n");
2151 		return;
2152 	}
2153 
2154 	phy_data.with_data = true;
2155 	phy_data.phy_info = le16_to_cpu(desc->phy_info);
2156 	phy_data.d4 = desc->phy_data4;
2157 
2158 	hdr = (void *)(pkt->data + desc_size);
2159 	/* Dont use dev_alloc_skb(), we'll have enough headroom once
2160 	 * ieee80211_hdr pulled.
2161 	 */
2162 	skb = alloc_skb(128, GFP_ATOMIC);
2163 	if (!skb) {
2164 		IWL_ERR(mvm, "alloc_skb failed\n");
2165 		return;
2166 	}
2167 
2168 	if (desc->mac_flags2 & IWL_RX_MPDU_MFLG2_PAD) {
2169 		/*
2170 		 * If the device inserted padding it means that (it thought)
2171 		 * the 802.11 header wasn't a multiple of 4 bytes long. In
2172 		 * this case, reserve two bytes at the start of the SKB to
2173 		 * align the payload properly in case we end up copying it.
2174 		 */
2175 		skb_reserve(skb, 2);
2176 	}
2177 
2178 	rx_status = IEEE80211_SKB_RXCB(skb);
2179 
2180 	/*
2181 	 * Keep packets with CRC errors (and with overrun) for monitor mode
2182 	 * (otherwise the firmware discards them) but mark them as bad.
2183 	 */
2184 	if (!(desc->status & cpu_to_le32(IWL_RX_MPDU_STATUS_CRC_OK)) ||
2185 	    !(desc->status & cpu_to_le32(IWL_RX_MPDU_STATUS_OVERRUN_OK))) {
2186 		IWL_DEBUG_RX(mvm, "Bad CRC or FIFO: 0x%08X.\n",
2187 			     le32_to_cpu(desc->status));
2188 		rx_status->flag |= RX_FLAG_FAILED_FCS_CRC;
2189 	}
2190 
2191 	/* set the preamble flag if appropriate */
2192 	if (format == RATE_MCS_MOD_TYPE_CCK &&
2193 	    phy_data.phy_info & IWL_RX_MPDU_PHY_SHORT_PREAMBLE)
2194 		rx_status->enc_flags |= RX_ENC_FLAG_SHORTPRE;
2195 
2196 	if (likely(!(phy_data.phy_info & IWL_RX_MPDU_PHY_TSF_OVERLOAD))) {
2197 		u64 tsf_on_air_rise;
2198 
2199 		if (mvm->trans->mac_cfg->device_family >=
2200 		    IWL_DEVICE_FAMILY_AX210)
2201 			tsf_on_air_rise = le64_to_cpu(desc->v3.tsf_on_air_rise);
2202 		else
2203 			tsf_on_air_rise = le64_to_cpu(desc->v1.tsf_on_air_rise);
2204 
2205 		rx_status->mactime = tsf_on_air_rise;
2206 		/* TSF as indicated by the firmware is at INA time */
2207 		rx_status->flag |= RX_FLAG_MACTIME_PLCP_START;
2208 	}
2209 
2210 	if (iwl_mvm_is_band_in_rx_supported(mvm)) {
2211 		u8 band = u8_get_bits(desc->mac_phy_band,
2212 				      IWL_RX_MPDU_MAC_PHY_BAND_BAND_MASK);
2213 
2214 		rx_status->band = iwl_mvm_nl80211_band_from_phy(band);
2215 	} else {
2216 		rx_status->band = phy_data.channel > 14 ? NL80211_BAND_5GHZ :
2217 			NL80211_BAND_2GHZ;
2218 	}
2219 
2220 	/* update aggregation data for monitor sake on default queue */
2221 	if (!queue && (phy_data.phy_info & IWL_RX_MPDU_PHY_AMPDU)) {
2222 		bool toggle_bit;
2223 
2224 		toggle_bit = phy_data.phy_info & IWL_RX_MPDU_PHY_AMPDU_TOGGLE;
2225 		rx_status->flag |= RX_FLAG_AMPDU_DETAILS;
2226 		/*
2227 		 * Toggle is switched whenever new aggregation starts. Make
2228 		 * sure ampdu_reference is never 0 so we can later use it to
2229 		 * see if the frame was really part of an A-MPDU or not.
2230 		 */
2231 		if (toggle_bit != mvm->ampdu_toggle) {
2232 			mvm->ampdu_ref++;
2233 			if (mvm->ampdu_ref == 0)
2234 				mvm->ampdu_ref++;
2235 			mvm->ampdu_toggle = toggle_bit;
2236 			phy_data.first_subframe = true;
2237 		}
2238 		rx_status->ampdu_reference = mvm->ampdu_ref;
2239 	}
2240 
2241 	rcu_read_lock();
2242 
2243 	if (desc->status & cpu_to_le32(IWL_RX_MPDU_STATUS_SRC_STA_FOUND)) {
2244 		if (!WARN_ON_ONCE(sta_id >= mvm->fw->ucode_capa.num_stations)) {
2245 			struct ieee80211_link_sta *link_sta;
2246 
2247 			sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
2248 			if (IS_ERR(sta))
2249 				sta = NULL;
2250 			link_sta = rcu_dereference(mvm->fw_id_to_link_sta[sta_id]);
2251 
2252 			if (sta && sta->valid_links && link_sta) {
2253 				rx_status->link_valid = 1;
2254 				rx_status->link_id = link_sta->link_id;
2255 			}
2256 		}
2257 	} else if (!is_multicast_ether_addr(hdr->addr2)) {
2258 		/*
2259 		 * This is fine since we prevent two stations with the same
2260 		 * address from being added.
2261 		 */
2262 		sta = ieee80211_find_sta_by_ifaddr(mvm->hw, hdr->addr2, NULL);
2263 	}
2264 
2265 	if (iwl_mvm_rx_crypto(mvm, sta, hdr, rx_status, phy_data.phy_info, desc,
2266 			      le32_to_cpu(pkt->len_n_flags), queue,
2267 			      &crypt_len)) {
2268 		kfree_skb(skb);
2269 		goto out;
2270 	}
2271 
2272 	iwl_mvm_rx_fill_status(mvm, desc, hdr, skb, &phy_data, queue);
2273 
2274 	if (sta) {
2275 		struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
2276 		struct ieee80211_vif *tx_blocked_vif =
2277 			rcu_dereference(mvm->csa_tx_blocked_vif);
2278 		u8 baid = (u8)((le32_to_cpu(desc->reorder_data) &
2279 			       IWL_RX_MPDU_REORDER_BAID_MASK) >>
2280 			       IWL_RX_MPDU_REORDER_BAID_SHIFT);
2281 		struct iwl_fw_dbg_trigger_tlv *trig;
2282 		struct ieee80211_vif *vif = mvmsta->vif;
2283 
2284 		if (!mvm->tcm.paused && len >= sizeof(*hdr) &&
2285 		    !is_multicast_ether_addr(hdr->addr1) &&
2286 		    ieee80211_is_data(hdr->frame_control) &&
2287 		    time_after(jiffies, mvm->tcm.ts + MVM_TCM_PERIOD))
2288 			schedule_delayed_work(&mvm->tcm.work, 0);
2289 
2290 		/*
2291 		 * We have tx blocked stations (with CS bit). If we heard
2292 		 * frames from a blocked station on a new channel we can
2293 		 * TX to it again.
2294 		 */
2295 		if (unlikely(tx_blocked_vif) && tx_blocked_vif == vif) {
2296 			struct iwl_mvm_vif *mvmvif =
2297 				iwl_mvm_vif_from_mac80211(tx_blocked_vif);
2298 			struct iwl_rx_sta_csa rx_sta_csa = {
2299 				.all_sta_unblocked = true,
2300 				.vif = tx_blocked_vif,
2301 			};
2302 
2303 			if (mvmvif->csa_target_freq == rx_status->freq)
2304 				iwl_mvm_sta_modify_disable_tx_ap(mvm, sta,
2305 								 false);
2306 			ieee80211_iterate_stations_atomic(mvm->hw,
2307 							  iwl_mvm_rx_get_sta_block_tx,
2308 							  &rx_sta_csa);
2309 
2310 			if (rx_sta_csa.all_sta_unblocked) {
2311 				RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
2312 				/* Unblock BCAST / MCAST station */
2313 				iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, false);
2314 				cancel_delayed_work(&mvm->cs_tx_unblock_dwork);
2315 			}
2316 		}
2317 
2318 		rs_update_last_rssi(mvm, mvmsta, rx_status);
2319 
2320 		trig = iwl_fw_dbg_trigger_on(&mvm->fwrt,
2321 					     ieee80211_vif_to_wdev(vif),
2322 					     FW_DBG_TRIGGER_RSSI);
2323 
2324 		if (trig && ieee80211_is_beacon(hdr->frame_control)) {
2325 			struct iwl_fw_dbg_trigger_low_rssi *rssi_trig;
2326 			s32 rssi;
2327 
2328 			rssi_trig = (void *)trig->data;
2329 			rssi = le32_to_cpu(rssi_trig->rssi);
2330 
2331 			if (rx_status->signal < rssi)
2332 				iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
2333 							NULL);
2334 		}
2335 
2336 		if (ieee80211_is_data(hdr->frame_control))
2337 			iwl_mvm_rx_csum(mvm, sta, skb, pkt);
2338 
2339 		if (iwl_mvm_is_dup(sta, queue, rx_status, hdr, desc)) {
2340 			IWL_DEBUG_DROP(mvm, "Dropping duplicate packet 0x%x\n",
2341 				       le16_to_cpu(hdr->seq_ctrl));
2342 			kfree_skb(skb);
2343 			goto out;
2344 		}
2345 
2346 		/*
2347 		 * Our hardware de-aggregates AMSDUs but copies the mac header
2348 		 * as it to the de-aggregated MPDUs. We need to turn off the
2349 		 * AMSDU bit in the QoS control ourselves.
2350 		 * In addition, HW reverses addr3 and addr4 - reverse it back.
2351 		 */
2352 		if ((desc->mac_flags2 & IWL_RX_MPDU_MFLG2_AMSDU) &&
2353 		    !WARN_ON(!ieee80211_is_data_qos(hdr->frame_control))) {
2354 			u8 *qc = ieee80211_get_qos_ctl(hdr);
2355 
2356 			*qc &= ~IEEE80211_QOS_CTL_A_MSDU_PRESENT;
2357 
2358 			if (mvm->trans->mac_cfg->device_family ==
2359 			    IWL_DEVICE_FAMILY_9000) {
2360 				iwl_mvm_flip_address(hdr->addr3);
2361 
2362 				if (ieee80211_has_a4(hdr->frame_control))
2363 					iwl_mvm_flip_address(hdr->addr4);
2364 			}
2365 		}
2366 		if (baid != IWL_RX_REORDER_DATA_INVALID_BAID) {
2367 			u32 reorder_data = le32_to_cpu(desc->reorder_data);
2368 
2369 			iwl_mvm_agg_rx_received(mvm, reorder_data, baid);
2370 		}
2371 
2372 		if (ieee80211_is_data(hdr->frame_control)) {
2373 			u8 sub_frame_idx = desc->amsdu_info &
2374 				IWL_RX_MPDU_AMSDU_SUBFRAME_IDX_MASK;
2375 
2376 			/* 0 means not an A-MSDU, and 1 means a new A-MSDU */
2377 			if (!sub_frame_idx || sub_frame_idx == 1)
2378 				iwl_mvm_count_mpdu(mvmsta, sta_id, 1, false,
2379 						   queue);
2380 		}
2381 	}
2382 
2383 	/* management stuff on default queue */
2384 	if (!queue) {
2385 		if (unlikely((ieee80211_is_beacon(hdr->frame_control) ||
2386 			      ieee80211_is_probe_resp(hdr->frame_control)) &&
2387 			     mvm->sched_scan_pass_all ==
2388 			     SCHED_SCAN_PASS_ALL_ENABLED))
2389 			mvm->sched_scan_pass_all = SCHED_SCAN_PASS_ALL_FOUND;
2390 
2391 		if (unlikely(ieee80211_is_beacon(hdr->frame_control) ||
2392 			     ieee80211_is_probe_resp(hdr->frame_control)))
2393 			rx_status->boottime_ns = ktime_get_boottime_ns();
2394 	}
2395 
2396 	if (iwl_mvm_create_skb(mvm, skb, hdr, len, crypt_len, rxb)) {
2397 		kfree_skb(skb);
2398 		goto out;
2399 	}
2400 
2401 	if (!iwl_mvm_reorder(mvm, napi, queue, sta, skb, desc) &&
2402 	    likely(!iwl_mvm_time_sync_frame(mvm, skb, hdr->addr2)) &&
2403 	    likely(!iwl_mvm_mei_filter_scan(mvm, skb))) {
2404 		if (mvm->trans->mac_cfg->device_family == IWL_DEVICE_FAMILY_9000 &&
2405 		    (desc->mac_flags2 & IWL_RX_MPDU_MFLG2_AMSDU) &&
2406 		    !(desc->amsdu_info & IWL_RX_MPDU_AMSDU_LAST_SUBFRAME))
2407 			rx_status->flag |= RX_FLAG_AMSDU_MORE;
2408 
2409 		iwl_mvm_pass_packet_to_mac80211(mvm, napi, skb, queue, sta);
2410 	}
2411 out:
2412 	rcu_read_unlock();
2413 }
2414 
2415 void iwl_mvm_rx_monitor_no_data(struct iwl_mvm *mvm, struct napi_struct *napi,
2416 				struct iwl_rx_cmd_buffer *rxb, int queue)
2417 {
2418 	struct ieee80211_rx_status *rx_status;
2419 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
2420 	struct iwl_rx_no_data_ver_3 *desc = (void *)pkt->data;
2421 	u32 rssi;
2422 	struct ieee80211_sta *sta = NULL;
2423 	struct sk_buff *skb;
2424 	struct iwl_mvm_rx_phy_data phy_data;
2425 	u32 format;
2426 
2427 	if (unlikely(test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)))
2428 		return;
2429 
2430 	if (unlikely(iwl_rx_packet_payload_len(pkt) < sizeof(struct iwl_rx_no_data)))
2431 		return;
2432 
2433 	rssi = le32_to_cpu(desc->rssi);
2434 	phy_data.d0 = desc->phy_info[0];
2435 	phy_data.d1 = desc->phy_info[1];
2436 	phy_data.phy_info = IWL_RX_MPDU_PHY_TSF_OVERLOAD;
2437 	phy_data.gp2_on_air_rise = le32_to_cpu(desc->on_air_rise_time);
2438 	phy_data.energy_a = u32_get_bits(rssi, RX_NO_DATA_CHAIN_A_MSK);
2439 	phy_data.energy_b = u32_get_bits(rssi, RX_NO_DATA_CHAIN_B_MSK);
2440 	phy_data.channel = u32_get_bits(rssi, RX_NO_DATA_CHANNEL_MSK);
2441 	phy_data.with_data = false;
2442 	phy_data.rx_vec[0] = desc->rx_vec[0];
2443 	phy_data.rx_vec[1] = desc->rx_vec[1];
2444 
2445 	phy_data.rate_n_flags = iwl_mvm_v3_rate_from_fw(desc->rate,
2446 							mvm->fw_rates_ver);
2447 
2448 	format = phy_data.rate_n_flags & RATE_MCS_MOD_TYPE_MSK;
2449 
2450 	if (iwl_fw_lookup_notif_ver(mvm->fw, DATA_PATH_GROUP,
2451 				    RX_NO_DATA_NOTIF, 0) >= 3) {
2452 		if (unlikely(iwl_rx_packet_payload_len(pkt) <
2453 		    sizeof(struct iwl_rx_no_data_ver_3)))
2454 		/* invalid len for ver 3 */
2455 			return;
2456 		phy_data.rx_vec[2] = desc->rx_vec[2];
2457 		phy_data.rx_vec[3] = desc->rx_vec[3];
2458 	} else {
2459 		if (format == RATE_MCS_MOD_TYPE_EHT)
2460 			/* no support for EHT before version 3 API */
2461 			return;
2462 	}
2463 
2464 	/* Dont use dev_alloc_skb(), we'll have enough headroom once
2465 	 * ieee80211_hdr pulled.
2466 	 */
2467 	skb = alloc_skb(128, GFP_ATOMIC);
2468 	if (!skb) {
2469 		IWL_ERR(mvm, "alloc_skb failed\n");
2470 		return;
2471 	}
2472 
2473 	rx_status = IEEE80211_SKB_RXCB(skb);
2474 
2475 	/* 0-length PSDU */
2476 	rx_status->flag |= RX_FLAG_NO_PSDU;
2477 
2478 	/* mark as failed PLCP on any errors to skip checks in mac80211 */
2479 	if (le32_get_bits(desc->info, RX_NO_DATA_INFO_ERR_MSK) !=
2480 	    RX_NO_DATA_INFO_ERR_NONE)
2481 		rx_status->flag |= RX_FLAG_FAILED_PLCP_CRC;
2482 
2483 	switch (le32_get_bits(desc->info, RX_NO_DATA_INFO_TYPE_MSK)) {
2484 	case RX_NO_DATA_INFO_TYPE_NDP:
2485 		rx_status->zero_length_psdu_type =
2486 			IEEE80211_RADIOTAP_ZERO_LEN_PSDU_SOUNDING;
2487 		break;
2488 	case RX_NO_DATA_INFO_TYPE_MU_UNMATCHED:
2489 	case RX_NO_DATA_INFO_TYPE_TB_UNMATCHED:
2490 		rx_status->zero_length_psdu_type =
2491 			IEEE80211_RADIOTAP_ZERO_LEN_PSDU_NOT_CAPTURED;
2492 		break;
2493 	default:
2494 		rx_status->zero_length_psdu_type =
2495 			IEEE80211_RADIOTAP_ZERO_LEN_PSDU_VENDOR;
2496 		break;
2497 	}
2498 
2499 	rx_status->band = phy_data.channel > 14 ? NL80211_BAND_5GHZ :
2500 		NL80211_BAND_2GHZ;
2501 
2502 	iwl_mvm_rx_fill_status(mvm, NULL, NULL, skb, &phy_data, queue);
2503 
2504 	/* no more radio tap info should be put after this point.
2505 	 *
2506 	 * We mark it as mac header, for upper layers to know where
2507 	 * all radio tap header ends.
2508 	 *
2509 	 * Since data doesn't move data while putting data on skb and that is
2510 	 * the only way we use, data + len is the next place that hdr would be put
2511 	 */
2512 	skb_set_mac_header(skb, skb->len);
2513 
2514 	/*
2515 	 * Override the nss from the rx_vec since the rate_n_flags has
2516 	 * only 2 bits for the nss which gives a max of 4 ss but there
2517 	 * may be up to 8 spatial streams.
2518 	 */
2519 	switch (format) {
2520 	case RATE_MCS_MOD_TYPE_VHT:
2521 		rx_status->nss =
2522 			le32_get_bits(desc->rx_vec[0],
2523 				      RX_NO_DATA_RX_VEC0_VHT_NSTS_MSK) + 1;
2524 		break;
2525 	case RATE_MCS_MOD_TYPE_HE:
2526 		rx_status->nss =
2527 			le32_get_bits(desc->rx_vec[0],
2528 				      RX_NO_DATA_RX_VEC0_HE_NSTS_MSK) + 1;
2529 		break;
2530 	case RATE_MCS_MOD_TYPE_EHT:
2531 		rx_status->nss =
2532 			le32_get_bits(desc->rx_vec[2],
2533 				      RX_NO_DATA_RX_VEC2_EHT_NSTS_MSK) + 1;
2534 	}
2535 
2536 	rcu_read_lock();
2537 	ieee80211_rx_napi(mvm->hw, sta, skb, napi);
2538 	rcu_read_unlock();
2539 }
2540 
2541 void iwl_mvm_rx_frame_release(struct iwl_mvm *mvm, struct napi_struct *napi,
2542 			      struct iwl_rx_cmd_buffer *rxb, int queue)
2543 {
2544 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
2545 	struct iwl_frame_release *release = (void *)pkt->data;
2546 
2547 	if (unlikely(iwl_rx_packet_payload_len(pkt) < sizeof(*release)))
2548 		return;
2549 
2550 	iwl_mvm_release_frames_from_notif(mvm, napi, release->baid,
2551 					  le16_to_cpu(release->nssn),
2552 					  queue);
2553 }
2554 
2555 void iwl_mvm_rx_bar_frame_release(struct iwl_mvm *mvm, struct napi_struct *napi,
2556 				  struct iwl_rx_cmd_buffer *rxb, int queue)
2557 {
2558 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
2559 	struct iwl_bar_frame_release *release = (void *)pkt->data;
2560 	struct iwl_mvm_baid_data *baid_data;
2561 	u32 pkt_len = iwl_rx_packet_payload_len(pkt);
2562 	unsigned int baid, nssn, sta_id, tid;
2563 
2564 	if (IWL_FW_CHECK(mvm, pkt_len < sizeof(*release),
2565 			 "Unexpected frame release notif size %d (expected %zu)\n",
2566 			 pkt_len, sizeof(*release)))
2567 		return;
2568 
2569 	baid = le32_get_bits(release->ba_info,
2570 			     IWL_BAR_FRAME_RELEASE_BAID_MASK);
2571 	nssn = le32_get_bits(release->ba_info,
2572 			     IWL_BAR_FRAME_RELEASE_NSSN_MASK);
2573 	sta_id = le32_get_bits(release->sta_tid,
2574 			       IWL_BAR_FRAME_RELEASE_STA_MASK);
2575 	tid = le32_get_bits(release->sta_tid,
2576 			    IWL_BAR_FRAME_RELEASE_TID_MASK);
2577 
2578 	if (WARN_ON_ONCE(baid == IWL_RX_REORDER_DATA_INVALID_BAID ||
2579 			 baid >= ARRAY_SIZE(mvm->baid_map)))
2580 		return;
2581 
2582 	rcu_read_lock();
2583 	baid_data = rcu_dereference(mvm->baid_map[baid]);
2584 	if (!baid_data) {
2585 		IWL_DEBUG_RX(mvm,
2586 			     "Got valid BAID %d but not allocated, invalid BAR release!\n",
2587 			      baid);
2588 		goto out;
2589 	}
2590 
2591 	if (WARN(tid != baid_data->tid || sta_id > IWL_STATION_COUNT_MAX ||
2592 		 !(baid_data->sta_mask & BIT(sta_id)),
2593 		 "baid 0x%x is mapped to sta_mask:0x%x tid:%d, but BAR release received for sta:%d tid:%d\n",
2594 		 baid, baid_data->sta_mask, baid_data->tid, sta_id,
2595 		 tid))
2596 		goto out;
2597 
2598 	IWL_DEBUG_DROP(mvm, "Received a BAR, expect packet loss: nssn %d\n",
2599 		       nssn);
2600 
2601 	iwl_mvm_release_frames_from_notif(mvm, napi, baid, nssn, queue);
2602 out:
2603 	rcu_read_unlock();
2604 }
2605 
2606 void iwl_mvm_rx_beacon_filter_notif(struct iwl_mvm *mvm,
2607 				    struct iwl_rx_cmd_buffer *rxb)
2608 {
2609 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
2610 	/* MAC or link ID in v1/v2, but driver has the IDs equal */
2611 	struct iwl_beacon_filter_notif *notif = (void *)pkt->data;
2612 	u32 id = le32_to_cpu(notif->link_id);
2613 	struct iwl_mvm_vif *mvm_vif;
2614 	struct ieee80211_vif *vif;
2615 
2616 	/* >= means AUX MAC/link ID, no energy correction needed then */
2617 	if (IWL_FW_CHECK(mvm, id >= ARRAY_SIZE(mvm->vif_id_to_mac),
2618 			 "invalid link ID %d\n", id))
2619 		return;
2620 
2621 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, id, false);
2622 	if (!vif)
2623 		return;
2624 
2625 	mvm_vif = iwl_mvm_vif_from_mac80211(vif);
2626 
2627 	mvm_vif->deflink.average_beacon_energy =
2628 		le32_to_cpu(notif->average_energy);
2629 }
2630