xref: /linux/drivers/net/wireless/intel/iwlwifi/mvm/rx.c (revision 90d32e92011eaae8e70a9169b4e7acf4ca8f9d3a)
1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2 /*
3  * Copyright (C) 2012-2014, 2018-2024 Intel Corporation
4  * Copyright (C) 2013-2015 Intel Mobile Communications GmbH
5  * Copyright (C) 2016-2017 Intel Deutschland GmbH
6  */
7 #include <asm/unaligned.h>
8 #include <linux/etherdevice.h>
9 #include <linux/skbuff.h>
10 #include "iwl-trans.h"
11 #include "mvm.h"
12 #include "fw-api.h"
13 
14 /*
15  * iwl_mvm_rx_rx_phy_cmd - REPLY_RX_PHY_CMD handler
16  *
17  * Copies the phy information in mvm->last_phy_info, it will be used when the
18  * actual data will come from the fw in the next packet.
19  */
20 void iwl_mvm_rx_rx_phy_cmd(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
21 {
22 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
23 	unsigned int pkt_len = iwl_rx_packet_payload_len(pkt);
24 
25 	if (unlikely(pkt_len < sizeof(mvm->last_phy_info)))
26 		return;
27 
28 	memcpy(&mvm->last_phy_info, pkt->data, sizeof(mvm->last_phy_info));
29 	mvm->ampdu_ref++;
30 
31 #ifdef CONFIG_IWLWIFI_DEBUGFS
32 	if (mvm->last_phy_info.phy_flags & cpu_to_le16(RX_RES_PHY_FLAGS_AGG)) {
33 		spin_lock(&mvm->drv_stats_lock);
34 		mvm->drv_rx_stats.ampdu_count++;
35 		spin_unlock(&mvm->drv_stats_lock);
36 	}
37 #endif
38 }
39 
40 /*
41  * iwl_mvm_pass_packet_to_mac80211 - builds the packet for mac80211
42  *
43  * Adds the rxb to a new skb and give it to mac80211
44  */
45 static void iwl_mvm_pass_packet_to_mac80211(struct iwl_mvm *mvm,
46 					    struct ieee80211_sta *sta,
47 					    struct napi_struct *napi,
48 					    struct sk_buff *skb,
49 					    struct ieee80211_hdr *hdr, u16 len,
50 					    u8 crypt_len,
51 					    struct iwl_rx_cmd_buffer *rxb)
52 {
53 	unsigned int hdrlen = ieee80211_hdrlen(hdr->frame_control);
54 	unsigned int fraglen;
55 
56 	/*
57 	 * The 'hdrlen' (plus the 8 bytes for the SNAP and the crypt_len,
58 	 * but those are all multiples of 4 long) all goes away, but we
59 	 * want the *end* of it, which is going to be the start of the IP
60 	 * header, to be aligned when it gets pulled in.
61 	 * The beginning of the skb->data is aligned on at least a 4-byte
62 	 * boundary after allocation. Everything here is aligned at least
63 	 * on a 2-byte boundary so we can just take hdrlen & 3 and pad by
64 	 * the result.
65 	 */
66 	skb_reserve(skb, hdrlen & 3);
67 
68 	/* If frame is small enough to fit in skb->head, pull it completely.
69 	 * If not, only pull ieee80211_hdr (including crypto if present, and
70 	 * an additional 8 bytes for SNAP/ethertype, see below) so that
71 	 * splice() or TCP coalesce are more efficient.
72 	 *
73 	 * Since, in addition, ieee80211_data_to_8023() always pull in at
74 	 * least 8 bytes (possibly more for mesh) we can do the same here
75 	 * to save the cost of doing it later. That still doesn't pull in
76 	 * the actual IP header since the typical case has a SNAP header.
77 	 * If the latter changes (there are efforts in the standards group
78 	 * to do so) we should revisit this and ieee80211_data_to_8023().
79 	 */
80 	hdrlen = (len <= skb_tailroom(skb)) ? len : hdrlen + crypt_len + 8;
81 
82 	skb_put_data(skb, hdr, hdrlen);
83 	fraglen = len - hdrlen;
84 
85 	if (fraglen) {
86 		int offset = (u8 *)hdr + hdrlen -
87 			     (u8 *)rxb_addr(rxb) + rxb_offset(rxb);
88 
89 		skb_add_rx_frag(skb, 0, rxb_steal_page(rxb), offset,
90 				fraglen, rxb->truesize);
91 	}
92 
93 	ieee80211_rx_napi(mvm->hw, sta, skb, napi);
94 }
95 
96 /*
97  * iwl_mvm_get_signal_strength - use new rx PHY INFO API
98  * values are reported by the fw as positive values - need to negate
99  * to obtain their dBM.  Account for missing antennas by replacing 0
100  * values by -256dBm: practically 0 power and a non-feasible 8 bit value.
101  */
102 static void iwl_mvm_get_signal_strength(struct iwl_mvm *mvm,
103 					struct iwl_rx_phy_info *phy_info,
104 					struct ieee80211_rx_status *rx_status)
105 {
106 	int energy_a, energy_b, max_energy;
107 	u32 val;
108 
109 	val =
110 	    le32_to_cpu(phy_info->non_cfg_phy[IWL_RX_INFO_ENERGY_ANT_ABC_IDX]);
111 	energy_a = (val & IWL_RX_INFO_ENERGY_ANT_A_MSK) >>
112 						IWL_RX_INFO_ENERGY_ANT_A_POS;
113 	energy_a = energy_a ? -energy_a : S8_MIN;
114 	energy_b = (val & IWL_RX_INFO_ENERGY_ANT_B_MSK) >>
115 						IWL_RX_INFO_ENERGY_ANT_B_POS;
116 	energy_b = energy_b ? -energy_b : S8_MIN;
117 	max_energy = max(energy_a, energy_b);
118 
119 	IWL_DEBUG_STATS(mvm, "energy In A %d B %d  , and max %d\n",
120 			energy_a, energy_b, max_energy);
121 
122 	rx_status->signal = max_energy;
123 	rx_status->chains = (le16_to_cpu(phy_info->phy_flags) &
124 				RX_RES_PHY_FLAGS_ANTENNA)
125 					>> RX_RES_PHY_FLAGS_ANTENNA_POS;
126 	rx_status->chain_signal[0] = energy_a;
127 	rx_status->chain_signal[1] = energy_b;
128 }
129 
130 /*
131  * iwl_mvm_set_mac80211_rx_flag - translate fw status to mac80211 format
132  * @mvm: the mvm object
133  * @hdr: 80211 header
134  * @stats: status in mac80211's format
135  * @rx_pkt_status: status coming from fw
136  *
137  * returns non 0 value if the packet should be dropped
138  */
139 static u32 iwl_mvm_set_mac80211_rx_flag(struct iwl_mvm *mvm,
140 					struct ieee80211_hdr *hdr,
141 					struct ieee80211_rx_status *stats,
142 					u32 rx_pkt_status,
143 					u8 *crypt_len)
144 {
145 	if (!ieee80211_has_protected(hdr->frame_control) ||
146 	    (rx_pkt_status & RX_MPDU_RES_STATUS_SEC_ENC_MSK) ==
147 			     RX_MPDU_RES_STATUS_SEC_NO_ENC)
148 		return 0;
149 
150 	/* packet was encrypted with unknown alg */
151 	if ((rx_pkt_status & RX_MPDU_RES_STATUS_SEC_ENC_MSK) ==
152 					RX_MPDU_RES_STATUS_SEC_ENC_ERR)
153 		return 0;
154 
155 	switch (rx_pkt_status & RX_MPDU_RES_STATUS_SEC_ENC_MSK) {
156 	case RX_MPDU_RES_STATUS_SEC_CCM_ENC:
157 		/* alg is CCM: check MIC only */
158 		if (!(rx_pkt_status & RX_MPDU_RES_STATUS_MIC_OK))
159 			return -1;
160 
161 		stats->flag |= RX_FLAG_DECRYPTED;
162 		*crypt_len = IEEE80211_CCMP_HDR_LEN;
163 		return 0;
164 
165 	case RX_MPDU_RES_STATUS_SEC_TKIP_ENC:
166 		/* Don't drop the frame and decrypt it in SW */
167 		if (!fw_has_api(&mvm->fw->ucode_capa,
168 				IWL_UCODE_TLV_API_DEPRECATE_TTAK) &&
169 		    !(rx_pkt_status & RX_MPDU_RES_STATUS_TTAK_OK))
170 			return 0;
171 		*crypt_len = IEEE80211_TKIP_IV_LEN;
172 		fallthrough;
173 
174 	case RX_MPDU_RES_STATUS_SEC_WEP_ENC:
175 		if (!(rx_pkt_status & RX_MPDU_RES_STATUS_ICV_OK))
176 			return -1;
177 
178 		stats->flag |= RX_FLAG_DECRYPTED;
179 		if ((rx_pkt_status & RX_MPDU_RES_STATUS_SEC_ENC_MSK) ==
180 				RX_MPDU_RES_STATUS_SEC_WEP_ENC)
181 			*crypt_len = IEEE80211_WEP_IV_LEN;
182 		return 0;
183 
184 	case RX_MPDU_RES_STATUS_SEC_EXT_ENC:
185 		if (!(rx_pkt_status & RX_MPDU_RES_STATUS_MIC_OK))
186 			return -1;
187 		stats->flag |= RX_FLAG_DECRYPTED;
188 		return 0;
189 
190 	default:
191 		/* Expected in monitor (not having the keys) */
192 		if (!mvm->monitor_on)
193 			IWL_WARN(mvm, "Unhandled alg: 0x%x\n", rx_pkt_status);
194 	}
195 
196 	return 0;
197 }
198 
199 static void iwl_mvm_rx_handle_tcm(struct iwl_mvm *mvm,
200 				  struct ieee80211_sta *sta,
201 				  struct ieee80211_hdr *hdr, u32 len,
202 				  struct iwl_rx_phy_info *phy_info,
203 				  u32 rate_n_flags)
204 {
205 	struct iwl_mvm_sta *mvmsta;
206 	struct iwl_mvm_tcm_mac *mdata;
207 	int mac;
208 	int ac = IEEE80211_AC_BE; /* treat non-QoS as BE */
209 	struct iwl_mvm_vif *mvmvif;
210 	/* expected throughput in 100Kbps, single stream, 20 MHz */
211 	static const u8 thresh_tpt[] = {
212 		9, 18, 30, 42, 60, 78, 90, 96, 120, 135,
213 	};
214 	u16 thr;
215 
216 	if (ieee80211_is_data_qos(hdr->frame_control)) {
217 		u8 tid = ieee80211_get_tid(hdr);
218 
219 		if (tid < IWL_MAX_TID_COUNT)
220 			ac = tid_to_mac80211_ac[tid];
221 	}
222 
223 	mvmsta = iwl_mvm_sta_from_mac80211(sta);
224 	mac = mvmsta->mac_id_n_color & FW_CTXT_ID_MSK;
225 
226 	if (time_after(jiffies, mvm->tcm.ts + MVM_TCM_PERIOD))
227 		schedule_delayed_work(&mvm->tcm.work, 0);
228 	mdata = &mvm->tcm.data[mac];
229 	mdata->rx.pkts[ac]++;
230 
231 	/* count the airtime only once for each ampdu */
232 	if (mdata->rx.last_ampdu_ref != mvm->ampdu_ref) {
233 		mdata->rx.last_ampdu_ref = mvm->ampdu_ref;
234 		mdata->rx.airtime += le16_to_cpu(phy_info->frame_time);
235 	}
236 
237 	if (!(rate_n_flags & (RATE_MCS_HT_MSK_V1 | RATE_MCS_VHT_MSK_V1)))
238 		return;
239 
240 	mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
241 
242 	if (mdata->opened_rx_ba_sessions ||
243 	    mdata->uapsd_nonagg_detect.detected ||
244 	    (!mvmvif->deflink.queue_params[IEEE80211_AC_VO].uapsd &&
245 	     !mvmvif->deflink.queue_params[IEEE80211_AC_VI].uapsd &&
246 	     !mvmvif->deflink.queue_params[IEEE80211_AC_BE].uapsd &&
247 	     !mvmvif->deflink.queue_params[IEEE80211_AC_BK].uapsd) ||
248 	    mvmsta->deflink.sta_id != mvmvif->deflink.ap_sta_id)
249 		return;
250 
251 	if (rate_n_flags & RATE_MCS_HT_MSK_V1) {
252 		thr = thresh_tpt[rate_n_flags & RATE_HT_MCS_RATE_CODE_MSK_V1];
253 		thr *= 1 + ((rate_n_flags & RATE_HT_MCS_NSS_MSK_V1) >>
254 					RATE_HT_MCS_NSS_POS_V1);
255 	} else {
256 		if (WARN_ON((rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK) >=
257 				ARRAY_SIZE(thresh_tpt)))
258 			return;
259 		thr = thresh_tpt[rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK];
260 		thr *= 1 + FIELD_GET(RATE_MCS_NSS_MSK, rate_n_flags);
261 	}
262 
263 	thr <<= ((rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK_V1) >>
264 				RATE_MCS_CHAN_WIDTH_POS);
265 
266 	mdata->uapsd_nonagg_detect.rx_bytes += len;
267 	ewma_rate_add(&mdata->uapsd_nonagg_detect.rate, thr);
268 }
269 
270 static void iwl_mvm_rx_csum(struct ieee80211_sta *sta,
271 			    struct sk_buff *skb,
272 			    u32 status)
273 {
274 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
275 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(mvmsta->vif);
276 
277 	if (mvmvif->features & NETIF_F_RXCSUM &&
278 	    status & RX_MPDU_RES_STATUS_CSUM_DONE &&
279 	    status & RX_MPDU_RES_STATUS_CSUM_OK)
280 		skb->ip_summed = CHECKSUM_UNNECESSARY;
281 }
282 
283 /*
284  * iwl_mvm_rx_rx_mpdu - REPLY_RX_MPDU_CMD handler
285  *
286  * Handles the actual data of the Rx packet from the fw
287  */
288 void iwl_mvm_rx_rx_mpdu(struct iwl_mvm *mvm, struct napi_struct *napi,
289 			struct iwl_rx_cmd_buffer *rxb)
290 {
291 	struct ieee80211_hdr *hdr;
292 	struct ieee80211_rx_status *rx_status;
293 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
294 	struct iwl_rx_phy_info *phy_info;
295 	struct iwl_rx_mpdu_res_start *rx_res;
296 	struct ieee80211_sta *sta = NULL;
297 	struct sk_buff *skb;
298 	u32 len, pkt_len = iwl_rx_packet_payload_len(pkt);
299 	u32 rate_n_flags;
300 	u32 rx_pkt_status;
301 	u8 crypt_len = 0;
302 
303 	if (unlikely(pkt_len < sizeof(*rx_res))) {
304 		IWL_DEBUG_DROP(mvm, "Bad REPLY_RX_MPDU_CMD size\n");
305 		return;
306 	}
307 
308 	phy_info = &mvm->last_phy_info;
309 	rx_res = (struct iwl_rx_mpdu_res_start *)pkt->data;
310 	hdr = (struct ieee80211_hdr *)(pkt->data + sizeof(*rx_res));
311 	len = le16_to_cpu(rx_res->byte_count);
312 
313 	if (unlikely(len + sizeof(*rx_res) + sizeof(__le32) > pkt_len)) {
314 		IWL_DEBUG_DROP(mvm, "FW lied about packet len\n");
315 		return;
316 	}
317 
318 	rx_pkt_status = get_unaligned_le32((__le32 *)
319 		(pkt->data + sizeof(*rx_res) + len));
320 
321 	/* Dont use dev_alloc_skb(), we'll have enough headroom once
322 	 * ieee80211_hdr pulled.
323 	 */
324 	skb = alloc_skb(128, GFP_ATOMIC);
325 	if (!skb) {
326 		IWL_ERR(mvm, "alloc_skb failed\n");
327 		return;
328 	}
329 
330 	rx_status = IEEE80211_SKB_RXCB(skb);
331 
332 	/*
333 	 * Keep packets with CRC errors (and with overrun) for monitor mode
334 	 * (otherwise the firmware discards them) but mark them as bad.
335 	 */
336 	if (!(rx_pkt_status & RX_MPDU_RES_STATUS_CRC_OK) ||
337 	    !(rx_pkt_status & RX_MPDU_RES_STATUS_OVERRUN_OK)) {
338 		IWL_DEBUG_RX(mvm, "Bad CRC or FIFO: 0x%08X.\n", rx_pkt_status);
339 		rx_status->flag |= RX_FLAG_FAILED_FCS_CRC;
340 	}
341 
342 	/* This will be used in several places later */
343 	rate_n_flags = le32_to_cpu(phy_info->rate_n_flags);
344 
345 	/* rx_status carries information about the packet to mac80211 */
346 	rx_status->mactime = le64_to_cpu(phy_info->timestamp);
347 	rx_status->device_timestamp = le32_to_cpu(phy_info->system_timestamp);
348 	rx_status->band =
349 		(phy_info->phy_flags & cpu_to_le16(RX_RES_PHY_FLAGS_BAND_24)) ?
350 				NL80211_BAND_2GHZ : NL80211_BAND_5GHZ;
351 	rx_status->freq =
352 		ieee80211_channel_to_frequency(le16_to_cpu(phy_info->channel),
353 					       rx_status->band);
354 
355 	/* TSF as indicated by the firmware  is at INA time */
356 	rx_status->flag |= RX_FLAG_MACTIME_PLCP_START;
357 
358 	iwl_mvm_get_signal_strength(mvm, phy_info, rx_status);
359 
360 	IWL_DEBUG_STATS_LIMIT(mvm, "Rssi %d, TSF %llu\n", rx_status->signal,
361 			      (unsigned long long)rx_status->mactime);
362 
363 	rcu_read_lock();
364 	if (rx_pkt_status & RX_MPDU_RES_STATUS_SRC_STA_FOUND) {
365 		u32 id = rx_pkt_status & RX_MPDU_RES_STATUS_STA_ID_MSK;
366 
367 		id >>= RX_MDPU_RES_STATUS_STA_ID_SHIFT;
368 
369 		if (!WARN_ON_ONCE(id >= mvm->fw->ucode_capa.num_stations)) {
370 			sta = rcu_dereference(mvm->fw_id_to_mac_id[id]);
371 			if (IS_ERR(sta))
372 				sta = NULL;
373 		}
374 	} else if (!is_multicast_ether_addr(hdr->addr2)) {
375 		/* This is fine since we prevent two stations with the same
376 		 * address from being added.
377 		 */
378 		sta = ieee80211_find_sta_by_ifaddr(mvm->hw, hdr->addr2, NULL);
379 	}
380 
381 	if (sta) {
382 		struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
383 		struct ieee80211_vif *vif = mvmsta->vif;
384 		struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
385 
386 		/*
387 		 * Don't even try to decrypt a MCAST frame that was received
388 		 * before the managed vif is authorized, we'd fail anyway.
389 		 */
390 		if (is_multicast_ether_addr(hdr->addr1) &&
391 		    vif->type == NL80211_IFTYPE_STATION &&
392 		    !mvmvif->authorized &&
393 		    ieee80211_has_protected(hdr->frame_control)) {
394 			IWL_DEBUG_DROP(mvm, "MCAST before the vif is authorized\n");
395 			kfree_skb(skb);
396 			rcu_read_unlock();
397 			return;
398 		}
399 	}
400 
401 	/*
402 	 * drop the packet if it has failed being decrypted by HW
403 	 */
404 	if (iwl_mvm_set_mac80211_rx_flag(mvm, hdr, rx_status, rx_pkt_status,
405 					 &crypt_len)) {
406 		IWL_DEBUG_DROP(mvm, "Bad decryption results 0x%08x\n",
407 			       rx_pkt_status);
408 		kfree_skb(skb);
409 		rcu_read_unlock();
410 		return;
411 	}
412 
413 	if (sta) {
414 		struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
415 		struct ieee80211_vif *tx_blocked_vif =
416 			rcu_dereference(mvm->csa_tx_blocked_vif);
417 		struct iwl_fw_dbg_trigger_tlv *trig;
418 		struct ieee80211_vif *vif = mvmsta->vif;
419 
420 		/* We have tx blocked stations (with CS bit). If we heard
421 		 * frames from a blocked station on a new channel we can
422 		 * TX to it again.
423 		 */
424 		if (unlikely(tx_blocked_vif) && vif == tx_blocked_vif) {
425 			struct iwl_mvm_vif *mvmvif =
426 				iwl_mvm_vif_from_mac80211(tx_blocked_vif);
427 
428 			if (mvmvif->csa_target_freq == rx_status->freq)
429 				iwl_mvm_sta_modify_disable_tx_ap(mvm, sta,
430 								 false);
431 		}
432 
433 		rs_update_last_rssi(mvm, mvmsta, rx_status);
434 
435 		trig = iwl_fw_dbg_trigger_on(&mvm->fwrt,
436 					     ieee80211_vif_to_wdev(vif),
437 					     FW_DBG_TRIGGER_RSSI);
438 
439 		if (trig && ieee80211_is_beacon(hdr->frame_control)) {
440 			struct iwl_fw_dbg_trigger_low_rssi *rssi_trig;
441 			s32 rssi;
442 
443 			rssi_trig = (void *)trig->data;
444 			rssi = le32_to_cpu(rssi_trig->rssi);
445 
446 			if (rx_status->signal < rssi)
447 				iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
448 							NULL);
449 		}
450 
451 		if (!mvm->tcm.paused && len >= sizeof(*hdr) &&
452 		    !is_multicast_ether_addr(hdr->addr1) &&
453 		    ieee80211_is_data(hdr->frame_control))
454 			iwl_mvm_rx_handle_tcm(mvm, sta, hdr, len, phy_info,
455 					      rate_n_flags);
456 
457 		if (ieee80211_is_data(hdr->frame_control))
458 			iwl_mvm_rx_csum(sta, skb, rx_pkt_status);
459 	}
460 	rcu_read_unlock();
461 
462 	/* set the preamble flag if appropriate */
463 	if (phy_info->phy_flags & cpu_to_le16(RX_RES_PHY_FLAGS_SHORT_PREAMBLE))
464 		rx_status->enc_flags |= RX_ENC_FLAG_SHORTPRE;
465 
466 	if (phy_info->phy_flags & cpu_to_le16(RX_RES_PHY_FLAGS_AGG)) {
467 		/*
468 		 * We know which subframes of an A-MPDU belong
469 		 * together since we get a single PHY response
470 		 * from the firmware for all of them
471 		 */
472 		rx_status->flag |= RX_FLAG_AMPDU_DETAILS;
473 		rx_status->ampdu_reference = mvm->ampdu_ref;
474 	}
475 
476 	/* Set up the HT phy flags */
477 	switch (rate_n_flags & RATE_MCS_CHAN_WIDTH_MSK_V1) {
478 	case RATE_MCS_CHAN_WIDTH_20:
479 		break;
480 	case RATE_MCS_CHAN_WIDTH_40:
481 		rx_status->bw = RATE_INFO_BW_40;
482 		break;
483 	case RATE_MCS_CHAN_WIDTH_80:
484 		rx_status->bw = RATE_INFO_BW_80;
485 		break;
486 	case RATE_MCS_CHAN_WIDTH_160:
487 		rx_status->bw = RATE_INFO_BW_160;
488 		break;
489 	}
490 	if (!(rate_n_flags & RATE_MCS_CCK_MSK_V1) &&
491 	    rate_n_flags & RATE_MCS_SGI_MSK_V1)
492 		rx_status->enc_flags |= RX_ENC_FLAG_SHORT_GI;
493 	if (rate_n_flags & RATE_HT_MCS_GF_MSK)
494 		rx_status->enc_flags |= RX_ENC_FLAG_HT_GF;
495 	if (rate_n_flags & RATE_MCS_LDPC_MSK_V1)
496 		rx_status->enc_flags |= RX_ENC_FLAG_LDPC;
497 	if (rate_n_flags & RATE_MCS_HT_MSK_V1) {
498 		u8 stbc = (rate_n_flags & RATE_MCS_STBC_MSK) >>
499 				RATE_MCS_STBC_POS;
500 		rx_status->encoding = RX_ENC_HT;
501 		rx_status->rate_idx = rate_n_flags & RATE_HT_MCS_INDEX_MSK_V1;
502 		rx_status->enc_flags |= stbc << RX_ENC_FLAG_STBC_SHIFT;
503 	} else if (rate_n_flags & RATE_MCS_VHT_MSK_V1) {
504 		u8 stbc = (rate_n_flags & RATE_MCS_STBC_MSK) >>
505 				RATE_MCS_STBC_POS;
506 		rx_status->nss =
507 			FIELD_GET(RATE_MCS_NSS_MSK, rate_n_flags) + 1;
508 		rx_status->rate_idx = rate_n_flags & RATE_VHT_MCS_RATE_CODE_MSK;
509 		rx_status->encoding = RX_ENC_VHT;
510 		rx_status->enc_flags |= stbc << RX_ENC_FLAG_STBC_SHIFT;
511 		if (rate_n_flags & RATE_MCS_BF_MSK)
512 			rx_status->enc_flags |= RX_ENC_FLAG_BF;
513 	} else {
514 		int rate = iwl_mvm_legacy_rate_to_mac80211_idx(rate_n_flags,
515 							       rx_status->band);
516 
517 		if (WARN(rate < 0 || rate > 0xFF,
518 			 "Invalid rate flags 0x%x, band %d,\n",
519 			 rate_n_flags, rx_status->band)) {
520 			kfree_skb(skb);
521 			return;
522 		}
523 		rx_status->rate_idx = rate;
524 	}
525 
526 #ifdef CONFIG_IWLWIFI_DEBUGFS
527 	iwl_mvm_update_frame_stats(mvm, rate_n_flags,
528 				   rx_status->flag & RX_FLAG_AMPDU_DETAILS);
529 #endif
530 
531 	if (unlikely((ieee80211_is_beacon(hdr->frame_control) ||
532 		      ieee80211_is_probe_resp(hdr->frame_control)) &&
533 		     mvm->sched_scan_pass_all == SCHED_SCAN_PASS_ALL_ENABLED))
534 		mvm->sched_scan_pass_all = SCHED_SCAN_PASS_ALL_FOUND;
535 
536 	if (unlikely(ieee80211_is_beacon(hdr->frame_control) ||
537 		     ieee80211_is_probe_resp(hdr->frame_control)))
538 		rx_status->boottime_ns = ktime_get_boottime_ns();
539 
540 	iwl_mvm_pass_packet_to_mac80211(mvm, sta, napi, skb, hdr, len,
541 					crypt_len, rxb);
542 }
543 
544 struct iwl_mvm_stat_data {
545 	struct iwl_mvm *mvm;
546 	__le32 flags;
547 	__le32 mac_id;
548 	u8 beacon_filter_average_energy;
549 	__le32 *beacon_counter;
550 	u8 *beacon_average_energy;
551 };
552 
553 struct iwl_mvm_stat_data_all_macs {
554 	struct iwl_mvm *mvm;
555 	__le32 flags;
556 	struct iwl_stats_ntfy_per_mac *per_mac;
557 };
558 
559 static void iwl_mvm_update_link_sig(struct ieee80211_vif *vif, int sig,
560 				    struct iwl_mvm_vif_link_info *link_info)
561 {
562 	struct iwl_mvm *mvm = iwl_mvm_vif_from_mac80211(vif)->mvm;
563 	struct ieee80211_bss_conf *bss_conf =
564 		iwl_mvm_rcu_fw_link_id_to_link_conf(mvm, link_info->fw_link_id,
565 						    false);
566 	int thold = bss_conf->cqm_rssi_thold;
567 	int hyst = bss_conf->cqm_rssi_hyst;
568 	int last_event;
569 	s8 exit_esr_thresh;
570 
571 	if (sig == 0) {
572 		IWL_DEBUG_RX(mvm, "RSSI is 0 - skip signal based decision\n");
573 		return;
574 	}
575 
576 	link_info->bf_data.ave_beacon_signal = sig;
577 
578 	/* BT Coex */
579 	if (link_info->bf_data.bt_coex_min_thold !=
580 	    link_info->bf_data.bt_coex_max_thold) {
581 		last_event = link_info->bf_data.last_bt_coex_event;
582 		if (sig > link_info->bf_data.bt_coex_max_thold &&
583 		    (last_event <= link_info->bf_data.bt_coex_min_thold ||
584 		     last_event == 0)) {
585 			link_info->bf_data.last_bt_coex_event = sig;
586 			IWL_DEBUG_RX(mvm, "cqm_iterator bt coex high %d\n",
587 				     sig);
588 			iwl_mvm_bt_rssi_event(mvm, vif, RSSI_EVENT_HIGH);
589 		} else if (sig < link_info->bf_data.bt_coex_min_thold &&
590 			   (last_event >= link_info->bf_data.bt_coex_max_thold ||
591 			    last_event == 0)) {
592 			link_info->bf_data.last_bt_coex_event = sig;
593 			IWL_DEBUG_RX(mvm, "cqm_iterator bt coex low %d\n",
594 				     sig);
595 			iwl_mvm_bt_rssi_event(mvm, vif, RSSI_EVENT_LOW);
596 		}
597 	}
598 
599 	if (!(vif->driver_flags & IEEE80211_VIF_SUPPORTS_CQM_RSSI))
600 		return;
601 
602 	/* CQM Notification */
603 	last_event = link_info->bf_data.last_cqm_event;
604 	if (thold && sig < thold && (last_event == 0 ||
605 				     sig < last_event - hyst)) {
606 		link_info->bf_data.last_cqm_event = sig;
607 		IWL_DEBUG_RX(mvm, "cqm_iterator cqm low %d\n",
608 			     sig);
609 		ieee80211_cqm_rssi_notify(
610 			vif,
611 			NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW,
612 			sig,
613 			GFP_KERNEL);
614 	} else if (sig > thold &&
615 		   (last_event == 0 || sig > last_event + hyst)) {
616 		link_info->bf_data.last_cqm_event = sig;
617 		IWL_DEBUG_RX(mvm, "cqm_iterator cqm high %d\n",
618 			     sig);
619 		ieee80211_cqm_rssi_notify(
620 			vif,
621 			NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH,
622 			sig,
623 			GFP_KERNEL);
624 	}
625 
626 	/* ESR recalculation */
627 	if (!vif->cfg.assoc || !ieee80211_vif_is_mld(vif))
628 		return;
629 
630 	exit_esr_thresh =
631 		iwl_mvm_get_esr_rssi_thresh(mvm,
632 					    &bss_conf->chanreq.oper,
633 					    true);
634 
635 	if (sig < exit_esr_thresh)
636 		iwl_mvm_exit_esr(mvm, vif, IWL_MVM_ESR_EXIT_LOW_RSSI,
637 				 iwl_mvm_get_other_link(vif,
638 							bss_conf->link_id));
639 }
640 
641 static void iwl_mvm_stat_iterator(void *_data, u8 *mac,
642 				  struct ieee80211_vif *vif)
643 {
644 	struct iwl_mvm_stat_data *data = _data;
645 	int sig = -data->beacon_filter_average_energy;
646 	u16 id = le32_to_cpu(data->mac_id);
647 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
648 	u16 vif_id = mvmvif->id;
649 
650 	/* This doesn't need the MAC ID check since it's not taking the
651 	 * data copied into the "data" struct, but rather the data from
652 	 * the notification directly.
653 	 */
654 	mvmvif->deflink.beacon_stats.num_beacons =
655 		le32_to_cpu(data->beacon_counter[vif_id]);
656 	mvmvif->deflink.beacon_stats.avg_signal =
657 		-data->beacon_average_energy[vif_id];
658 
659 	if (mvmvif->id != id)
660 		return;
661 
662 	if (vif->type != NL80211_IFTYPE_STATION)
663 		return;
664 
665 	/* make sure that beacon statistics don't go backwards with TCM
666 	 * request to clear statistics
667 	 */
668 	if (le32_to_cpu(data->flags) & IWL_STATISTICS_REPLY_FLG_CLEAR)
669 		mvmvif->deflink.beacon_stats.accu_num_beacons +=
670 			mvmvif->deflink.beacon_stats.num_beacons;
671 
672 	/* This is used in pre-MLO API so use deflink */
673 	iwl_mvm_update_link_sig(vif, sig, &mvmvif->deflink);
674 }
675 
676 static void iwl_mvm_stat_iterator_all_macs(void *_data, u8 *mac,
677 					   struct ieee80211_vif *vif)
678 {
679 	struct iwl_mvm_stat_data_all_macs *data = _data;
680 	struct iwl_stats_ntfy_per_mac *mac_stats;
681 	int sig;
682 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
683 	u16 vif_id = mvmvif->id;
684 
685 	if (WARN_ONCE(vif_id >= MAC_INDEX_AUX, "invalid vif id: %d", vif_id))
686 		return;
687 
688 	if (vif->type != NL80211_IFTYPE_STATION)
689 		return;
690 
691 	mac_stats = &data->per_mac[vif_id];
692 
693 	mvmvif->deflink.beacon_stats.num_beacons =
694 		le32_to_cpu(mac_stats->beacon_counter);
695 	mvmvif->deflink.beacon_stats.avg_signal =
696 		-le32_to_cpu(mac_stats->beacon_average_energy);
697 
698 	/* make sure that beacon statistics don't go backwards with TCM
699 	 * request to clear statistics
700 	 */
701 	if (le32_to_cpu(data->flags) & IWL_STATISTICS_REPLY_FLG_CLEAR)
702 		mvmvif->deflink.beacon_stats.accu_num_beacons +=
703 			mvmvif->deflink.beacon_stats.num_beacons;
704 
705 	sig = -le32_to_cpu(mac_stats->beacon_filter_average_energy);
706 
707 	/* This is used in pre-MLO API so use deflink */
708 	iwl_mvm_update_link_sig(vif, sig, &mvmvif->deflink);
709 }
710 
711 static inline void
712 iwl_mvm_rx_stats_check_trigger(struct iwl_mvm *mvm, struct iwl_rx_packet *pkt)
713 {
714 	struct iwl_fw_dbg_trigger_tlv *trig;
715 	struct iwl_fw_dbg_trigger_stats *trig_stats;
716 	u32 trig_offset, trig_thold;
717 
718 	trig = iwl_fw_dbg_trigger_on(&mvm->fwrt, NULL, FW_DBG_TRIGGER_STATS);
719 	if (!trig)
720 		return;
721 
722 	trig_stats = (void *)trig->data;
723 
724 	trig_offset = le32_to_cpu(trig_stats->stop_offset);
725 	trig_thold = le32_to_cpu(trig_stats->stop_threshold);
726 
727 	if (WARN_ON_ONCE(trig_offset >= iwl_rx_packet_payload_len(pkt)))
728 		return;
729 
730 	if (le32_to_cpup((__le32 *) (pkt->data + trig_offset)) < trig_thold)
731 		return;
732 
733 	iwl_fw_dbg_collect_trig(&mvm->fwrt, trig, NULL);
734 }
735 
736 static void iwl_mvm_stats_energy_iter(void *_data,
737 				      struct ieee80211_sta *sta)
738 {
739 	struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
740 	u8 *energy = _data;
741 	u32 sta_id = mvmsta->deflink.sta_id;
742 
743 	if (WARN_ONCE(sta_id >= IWL_MVM_STATION_COUNT_MAX, "sta_id %d >= %d",
744 		      sta_id, IWL_MVM_STATION_COUNT_MAX))
745 		return;
746 
747 	if (energy[sta_id])
748 		mvmsta->deflink.avg_energy = energy[sta_id];
749 
750 }
751 
752 static void
753 iwl_mvm_update_tcm_from_stats(struct iwl_mvm *mvm, __le32 *air_time_le,
754 			      __le32 *rx_bytes_le)
755 {
756 	int i;
757 
758 	spin_lock(&mvm->tcm.lock);
759 	for (i = 0; i < NUM_MAC_INDEX_DRIVER; i++) {
760 		struct iwl_mvm_tcm_mac *mdata = &mvm->tcm.data[i];
761 		u32 rx_bytes = le32_to_cpu(rx_bytes_le[i]);
762 		u32 airtime = le32_to_cpu(air_time_le[i]);
763 
764 		mdata->rx.airtime += airtime;
765 		mdata->uapsd_nonagg_detect.rx_bytes += rx_bytes;
766 		if (airtime) {
767 			/* re-init every time to store rate from FW */
768 			ewma_rate_init(&mdata->uapsd_nonagg_detect.rate);
769 			ewma_rate_add(&mdata->uapsd_nonagg_detect.rate,
770 				      rx_bytes * 8 / airtime);
771 		}
772 	}
773 	spin_unlock(&mvm->tcm.lock);
774 }
775 
776 static void iwl_mvm_handle_per_phy_stats(struct iwl_mvm *mvm,
777 					 struct iwl_stats_ntfy_per_phy *per_phy)
778 {
779 	int i;
780 
781 	for (i = 0; i < NUM_PHY_CTX; i++) {
782 		if (!mvm->phy_ctxts[i].ref)
783 			continue;
784 		mvm->phy_ctxts[i].channel_load_by_us =
785 			le32_to_cpu(per_phy[i].channel_load_by_us);
786 	}
787 }
788 
789 static void
790 iwl_mvm_stats_ver_15(struct iwl_mvm *mvm,
791 		     struct iwl_statistics_operational_ntfy *stats)
792 {
793 	struct iwl_mvm_stat_data_all_macs data = {
794 		.mvm = mvm,
795 		.flags = stats->flags,
796 		.per_mac = stats->per_mac,
797 	};
798 
799 	ieee80211_iterate_active_interfaces(mvm->hw,
800 					    IEEE80211_IFACE_ITER_NORMAL,
801 					    iwl_mvm_stat_iterator_all_macs,
802 					    &data);
803 	iwl_mvm_handle_per_phy_stats(mvm, stats->per_phy);
804 }
805 
806 static void
807 iwl_mvm_stats_ver_14(struct iwl_mvm *mvm,
808 		     struct iwl_statistics_operational_ntfy_ver_14 *stats)
809 {
810 	struct iwl_mvm_stat_data data = {
811 		.mvm = mvm,
812 	};
813 
814 	u8 beacon_average_energy[MAC_INDEX_AUX];
815 	__le32 flags;
816 	int i;
817 
818 	flags = stats->flags;
819 
820 	data.mac_id = stats->mac_id;
821 	data.beacon_filter_average_energy =
822 		le32_to_cpu(stats->beacon_filter_average_energy);
823 	data.flags = flags;
824 	data.beacon_counter = stats->beacon_counter;
825 
826 	for (i = 0; i < ARRAY_SIZE(beacon_average_energy); i++)
827 		beacon_average_energy[i] =
828 			le32_to_cpu(stats->beacon_average_energy[i]);
829 
830 	data.beacon_average_energy = beacon_average_energy;
831 
832 	ieee80211_iterate_active_interfaces(mvm->hw,
833 					    IEEE80211_IFACE_ITER_NORMAL,
834 					    iwl_mvm_stat_iterator,
835 					    &data);
836 }
837 
838 static bool iwl_mvm_verify_stats_len(struct iwl_mvm *mvm,
839 				     struct iwl_rx_packet *pkt,
840 				     u32 expected_size)
841 {
842 	struct iwl_statistics_ntfy_hdr *hdr;
843 
844 	if (WARN_ONCE(iwl_rx_packet_payload_len(pkt) < expected_size,
845 		      "received invalid statistics size (%d)!, expected_size: %d\n",
846 		      iwl_rx_packet_payload_len(pkt), expected_size))
847 		return false;
848 
849 	hdr = (void *)&pkt->data;
850 
851 	if (WARN_ONCE((hdr->type & IWL_STATISTICS_TYPE_MSK) != FW_STATISTICS_OPERATIONAL ||
852 		      hdr->version !=
853 		      iwl_fw_lookup_notif_ver(mvm->fw, LEGACY_GROUP, STATISTICS_NOTIFICATION, 0),
854 		      "received unsupported hdr type %d, version %d\n",
855 		      hdr->type, hdr->version))
856 		return false;
857 
858 	if (WARN_ONCE(le16_to_cpu(hdr->size) != expected_size,
859 		      "received invalid statistics size in header (%d)!, expected_size: %d\n",
860 		      le16_to_cpu(hdr->size), expected_size))
861 		return false;
862 
863 	return true;
864 }
865 
866 static void
867 iwl_mvm_stat_iterator_all_links(struct iwl_mvm *mvm,
868 				struct iwl_stats_ntfy_per_link *per_link)
869 {
870 	u32 air_time[MAC_INDEX_AUX] = {};
871 	u32 rx_bytes[MAC_INDEX_AUX] = {};
872 	int fw_link_id;
873 
874 	for (fw_link_id = 0; fw_link_id < ARRAY_SIZE(mvm->link_id_to_link_conf);
875 	     fw_link_id++) {
876 		struct iwl_stats_ntfy_per_link *link_stats;
877 		struct ieee80211_bss_conf *bss_conf;
878 		struct iwl_mvm_vif *mvmvif;
879 		struct iwl_mvm_vif_link_info *link_info;
880 		int link_id;
881 		int sig;
882 
883 		bss_conf = iwl_mvm_rcu_fw_link_id_to_link_conf(mvm, fw_link_id,
884 							       false);
885 		if (!bss_conf)
886 			continue;
887 
888 		if (bss_conf->vif->type != NL80211_IFTYPE_STATION)
889 			continue;
890 
891 		link_id = bss_conf->link_id;
892 		if (link_id >= ARRAY_SIZE(mvmvif->link))
893 			continue;
894 
895 		mvmvif = iwl_mvm_vif_from_mac80211(bss_conf->vif);
896 		link_info = mvmvif->link[link_id];
897 		if (!link_info)
898 			continue;
899 
900 		link_stats = &per_link[fw_link_id];
901 
902 		link_info->beacon_stats.num_beacons =
903 			le32_to_cpu(link_stats->beacon_counter);
904 
905 		/* we basically just use the u8 to store 8 bits and then treat
906 		 * it as a s8 whenever we take it out to a different type.
907 		 */
908 		link_info->beacon_stats.avg_signal =
909 			-le32_to_cpu(link_stats->beacon_average_energy);
910 
911 		if (link_info->phy_ctxt &&
912 		    link_info->phy_ctxt->channel->band == NL80211_BAND_2GHZ)
913 			iwl_mvm_bt_coex_update_link_esr(mvm, bss_conf->vif,
914 							link_id);
915 
916 		/* make sure that beacon statistics don't go backwards with TCM
917 		 * request to clear statistics
918 		 */
919 		if (mvm->statistics_clear)
920 			mvmvif->link[link_id]->beacon_stats.accu_num_beacons +=
921 				mvmvif->link[link_id]->beacon_stats.num_beacons;
922 
923 		sig = -le32_to_cpu(link_stats->beacon_filter_average_energy);
924 		iwl_mvm_update_link_sig(bss_conf->vif, sig, link_info);
925 
926 		if (WARN_ONCE(mvmvif->id >= MAC_INDEX_AUX,
927 			      "invalid mvmvif id: %d", mvmvif->id))
928 			continue;
929 
930 		air_time[mvmvif->id] +=
931 			le32_to_cpu(per_link[fw_link_id].air_time);
932 		rx_bytes[mvmvif->id] +=
933 			le32_to_cpu(per_link[fw_link_id].rx_bytes);
934 	}
935 
936 	/* Don't update in case the statistics are not cleared, since
937 	 * we will end up counting twice the same airtime, once in TCM
938 	 * request and once in statistics notification.
939 	 */
940 	if (mvm->statistics_clear) {
941 		__le32 air_time_le[MAC_INDEX_AUX];
942 		__le32 rx_bytes_le[MAC_INDEX_AUX];
943 		int vif_id;
944 
945 		for (vif_id = 0; vif_id < ARRAY_SIZE(air_time_le); vif_id++) {
946 			air_time_le[vif_id] = cpu_to_le32(air_time[vif_id]);
947 			rx_bytes_le[vif_id] = cpu_to_le32(rx_bytes[vif_id]);
948 		}
949 
950 		iwl_mvm_update_tcm_from_stats(mvm, air_time_le, rx_bytes_le);
951 	}
952 }
953 
954 #define SEC_LINK_MIN_PERC 10
955 #define SEC_LINK_MIN_TX 3000
956 #define SEC_LINK_MIN_RX 400
957 
958 static void iwl_mvm_update_esr_mode_tpt(struct iwl_mvm *mvm)
959 {
960 	struct ieee80211_vif *bss_vif = iwl_mvm_get_bss_vif(mvm);
961 	struct iwl_mvm_vif *mvmvif;
962 	struct iwl_mvm_sta *mvmsta;
963 	unsigned long total_tx = 0, total_rx = 0;
964 	unsigned long sec_link_tx = 0, sec_link_rx = 0;
965 	u8 sec_link_tx_perc, sec_link_rx_perc;
966 	u8 sec_link;
967 
968 	lockdep_assert_held(&mvm->mutex);
969 
970 	if (!bss_vif)
971 		return;
972 
973 	mvmvif = iwl_mvm_vif_from_mac80211(bss_vif);
974 
975 	if (!mvmvif->esr_active || !mvmvif->ap_sta)
976 		return;
977 
978 	mvmsta = iwl_mvm_sta_from_mac80211(mvmvif->ap_sta);
979 	/* We only count for the AP sta in a MLO connection */
980 	if (!mvmsta->mpdu_counters)
981 		return;
982 
983 	/* Get the FW ID of the secondary link */
984 	sec_link = iwl_mvm_get_other_link(bss_vif,
985 					  iwl_mvm_get_primary_link(bss_vif));
986 	if (WARN_ON(!mvmvif->link[sec_link]))
987 		return;
988 	sec_link = mvmvif->link[sec_link]->fw_link_id;
989 
990 	/* Sum up RX and TX MPDUs from the different queues/links */
991 	for (int q = 0; q < mvm->trans->num_rx_queues; q++) {
992 		spin_lock_bh(&mvmsta->mpdu_counters[q].lock);
993 
994 		/* The link IDs that doesn't exist will contain 0 */
995 		for (int link = 0; link < IWL_MVM_FW_MAX_LINK_ID; link++) {
996 			total_tx += mvmsta->mpdu_counters[q].per_link[link].tx;
997 			total_rx += mvmsta->mpdu_counters[q].per_link[link].rx;
998 		}
999 
1000 		sec_link_tx += mvmsta->mpdu_counters[q].per_link[sec_link].tx;
1001 		sec_link_rx += mvmsta->mpdu_counters[q].per_link[sec_link].rx;
1002 
1003 		/*
1004 		 * In EMLSR we have statistics every 5 seconds, so we can reset
1005 		 * the counters upon every statistics notification.
1006 		 */
1007 		memset(mvmsta->mpdu_counters[q].per_link, 0,
1008 		       sizeof(mvmsta->mpdu_counters[q].per_link));
1009 
1010 		spin_unlock_bh(&mvmsta->mpdu_counters[q].lock);
1011 	}
1012 
1013 	/* If we don't have enough MPDUs - exit EMLSR */
1014 	if (total_tx < IWL_MVM_ENTER_ESR_TPT_THRESH &&
1015 	    total_rx < IWL_MVM_ENTER_ESR_TPT_THRESH) {
1016 		iwl_mvm_block_esr(mvm, bss_vif, IWL_MVM_ESR_BLOCKED_TPT,
1017 				  iwl_mvm_get_primary_link(bss_vif));
1018 		return;
1019 	}
1020 
1021 	/* Calculate the percentage of the secondary link TX/RX */
1022 	sec_link_tx_perc = total_tx ? sec_link_tx * 100 / total_tx : 0;
1023 	sec_link_rx_perc = total_rx ? sec_link_rx * 100 / total_rx : 0;
1024 
1025 	/*
1026 	 * The TX/RX percentage is checked only if it exceeds the required
1027 	 * minimum. In addition, RX is checked only if the TX check failed.
1028 	 */
1029 	if ((total_tx > SEC_LINK_MIN_TX &&
1030 	     sec_link_tx_perc < SEC_LINK_MIN_PERC) ||
1031 	    (total_rx > SEC_LINK_MIN_RX &&
1032 	     sec_link_rx_perc < SEC_LINK_MIN_PERC))
1033 		iwl_mvm_exit_esr(mvm, bss_vif, IWL_MVM_ESR_EXIT_LINK_USAGE,
1034 				 iwl_mvm_get_primary_link(bss_vif));
1035 }
1036 
1037 void iwl_mvm_handle_rx_system_oper_stats(struct iwl_mvm *mvm,
1038 					 struct iwl_rx_cmd_buffer *rxb)
1039 {
1040 	u8 average_energy[IWL_MVM_STATION_COUNT_MAX];
1041 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1042 	struct iwl_system_statistics_notif_oper *stats;
1043 	int i;
1044 	u32 notif_ver = iwl_fw_lookup_notif_ver(mvm->fw, STATISTICS_GROUP,
1045 						STATISTICS_OPER_NOTIF, 0);
1046 
1047 	if (notif_ver != 3) {
1048 		IWL_FW_CHECK_FAILED(mvm,
1049 				    "Oper stats notif ver %d is not supported\n",
1050 				    notif_ver);
1051 		return;
1052 	}
1053 
1054 	stats = (void *)&pkt->data;
1055 	iwl_mvm_stat_iterator_all_links(mvm, stats->per_link);
1056 
1057 	for (i = 0; i < ARRAY_SIZE(average_energy); i++)
1058 		average_energy[i] =
1059 			le32_to_cpu(stats->per_sta[i].average_energy);
1060 
1061 	ieee80211_iterate_stations_atomic(mvm->hw, iwl_mvm_stats_energy_iter,
1062 					  average_energy);
1063 	iwl_mvm_handle_per_phy_stats(mvm, stats->per_phy);
1064 
1065 	iwl_mvm_update_esr_mode_tpt(mvm);
1066 }
1067 
1068 void iwl_mvm_handle_rx_system_oper_part1_stats(struct iwl_mvm *mvm,
1069 					       struct iwl_rx_cmd_buffer *rxb)
1070 {
1071 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1072 	struct iwl_system_statistics_part1_notif_oper *part1_stats;
1073 	int i;
1074 	u32 notif_ver = iwl_fw_lookup_notif_ver(mvm->fw, STATISTICS_GROUP,
1075 						STATISTICS_OPER_PART1_NOTIF, 0);
1076 
1077 	if (notif_ver != 4) {
1078 		IWL_FW_CHECK_FAILED(mvm,
1079 				    "Part1 stats notif ver %d is not supported\n",
1080 				    notif_ver);
1081 		return;
1082 	}
1083 
1084 	part1_stats = (void *)&pkt->data;
1085 	mvm->radio_stats.rx_time = 0;
1086 	mvm->radio_stats.tx_time = 0;
1087 	for (i = 0; i < ARRAY_SIZE(part1_stats->per_link); i++) {
1088 		mvm->radio_stats.rx_time +=
1089 			le64_to_cpu(part1_stats->per_link[i].rx_time);
1090 		mvm->radio_stats.tx_time +=
1091 			le64_to_cpu(part1_stats->per_link[i].tx_time);
1092 	}
1093 }
1094 
1095 static void
1096 iwl_mvm_handle_rx_statistics_tlv(struct iwl_mvm *mvm,
1097 				 struct iwl_rx_packet *pkt)
1098 {
1099 	u8 average_energy[IWL_MVM_STATION_COUNT_MAX];
1100 	__le32 air_time[MAC_INDEX_AUX];
1101 	__le32 rx_bytes[MAC_INDEX_AUX];
1102 	__le32 flags = 0;
1103 	int i;
1104 	u32 notif_ver = iwl_fw_lookup_notif_ver(mvm->fw, LEGACY_GROUP,
1105 					      STATISTICS_NOTIFICATION, 0);
1106 
1107 	if (WARN_ONCE(notif_ver > 15,
1108 		      "invalid statistics version id: %d\n", notif_ver))
1109 		return;
1110 
1111 	if (notif_ver == 14) {
1112 		struct iwl_statistics_operational_ntfy_ver_14 *stats =
1113 			(void *)pkt->data;
1114 
1115 		if (!iwl_mvm_verify_stats_len(mvm, pkt, sizeof(*stats)))
1116 			return;
1117 
1118 		iwl_mvm_stats_ver_14(mvm, stats);
1119 
1120 		flags = stats->flags;
1121 		mvm->radio_stats.rx_time = le64_to_cpu(stats->rx_time);
1122 		mvm->radio_stats.tx_time = le64_to_cpu(stats->tx_time);
1123 		mvm->radio_stats.on_time_rf = le64_to_cpu(stats->on_time_rf);
1124 		mvm->radio_stats.on_time_scan =
1125 			le64_to_cpu(stats->on_time_scan);
1126 
1127 		for (i = 0; i < ARRAY_SIZE(average_energy); i++)
1128 			average_energy[i] = le32_to_cpu(stats->average_energy[i]);
1129 
1130 		for (i = 0; i < ARRAY_SIZE(air_time); i++) {
1131 			air_time[i] = stats->air_time[i];
1132 			rx_bytes[i] = stats->rx_bytes[i];
1133 		}
1134 	}
1135 
1136 	if (notif_ver == 15) {
1137 		struct iwl_statistics_operational_ntfy *stats =
1138 			(void *)pkt->data;
1139 
1140 		if (!iwl_mvm_verify_stats_len(mvm, pkt, sizeof(*stats)))
1141 			return;
1142 
1143 		iwl_mvm_stats_ver_15(mvm, stats);
1144 
1145 		flags = stats->flags;
1146 		mvm->radio_stats.rx_time = le64_to_cpu(stats->rx_time);
1147 		mvm->radio_stats.tx_time = le64_to_cpu(stats->tx_time);
1148 		mvm->radio_stats.on_time_rf = le64_to_cpu(stats->on_time_rf);
1149 		mvm->radio_stats.on_time_scan =
1150 			le64_to_cpu(stats->on_time_scan);
1151 
1152 		for (i = 0; i < ARRAY_SIZE(average_energy); i++)
1153 			average_energy[i] =
1154 				le32_to_cpu(stats->per_sta[i].average_energy);
1155 
1156 		for (i = 0; i < ARRAY_SIZE(air_time); i++) {
1157 			air_time[i] = stats->per_mac[i].air_time;
1158 			rx_bytes[i] = stats->per_mac[i].rx_bytes;
1159 		}
1160 	}
1161 
1162 	iwl_mvm_rx_stats_check_trigger(mvm, pkt);
1163 
1164 	ieee80211_iterate_stations_atomic(mvm->hw, iwl_mvm_stats_energy_iter,
1165 					  average_energy);
1166 	/*
1167 	 * Don't update in case the statistics are not cleared, since
1168 	 * we will end up counting twice the same airtime, once in TCM
1169 	 * request and once in statistics notification.
1170 	 */
1171 	if (le32_to_cpu(flags) & IWL_STATISTICS_REPLY_FLG_CLEAR)
1172 		iwl_mvm_update_tcm_from_stats(mvm, air_time, rx_bytes);
1173 }
1174 
1175 void iwl_mvm_handle_rx_statistics(struct iwl_mvm *mvm,
1176 				  struct iwl_rx_packet *pkt)
1177 {
1178 	struct iwl_mvm_stat_data data = {
1179 		.mvm = mvm,
1180 	};
1181 	__le32 *bytes, *air_time, flags;
1182 	int expected_size;
1183 	u8 *energy;
1184 	u8 cmd_ver = iwl_fw_lookup_cmd_ver(mvm->fw,
1185 					   WIDE_ID(SYSTEM_GROUP,
1186 						   SYSTEM_STATISTICS_CMD),
1187 					   IWL_FW_CMD_VER_UNKNOWN);
1188 
1189 	if (cmd_ver != IWL_FW_CMD_VER_UNKNOWN)
1190 		return;
1191 
1192 	/* From ver 14 and up we use TLV statistics format */
1193 	if (iwl_fw_lookup_notif_ver(mvm->fw, LEGACY_GROUP,
1194 				    STATISTICS_NOTIFICATION, 0) >= 14)
1195 		return iwl_mvm_handle_rx_statistics_tlv(mvm, pkt);
1196 
1197 	if (!iwl_mvm_has_new_rx_stats_api(mvm)) {
1198 		if (iwl_mvm_has_new_rx_api(mvm))
1199 			expected_size = sizeof(struct iwl_notif_statistics_v11);
1200 		else
1201 			expected_size = sizeof(struct iwl_notif_statistics_v10);
1202 	} else {
1203 		expected_size = sizeof(struct iwl_notif_statistics);
1204 	}
1205 
1206 	if (WARN_ONCE(iwl_rx_packet_payload_len(pkt) != expected_size,
1207 		      "received invalid statistics size (%d)!\n",
1208 		      iwl_rx_packet_payload_len(pkt)))
1209 		return;
1210 
1211 	if (!iwl_mvm_has_new_rx_stats_api(mvm)) {
1212 		struct iwl_notif_statistics_v11 *stats = (void *)&pkt->data;
1213 
1214 		data.mac_id = stats->rx.general.mac_id;
1215 		data.beacon_filter_average_energy =
1216 			stats->general.common.beacon_filter_average_energy;
1217 
1218 		mvm->rx_stats_v3 = stats->rx;
1219 
1220 		mvm->radio_stats.rx_time =
1221 			le64_to_cpu(stats->general.common.rx_time);
1222 		mvm->radio_stats.tx_time =
1223 			le64_to_cpu(stats->general.common.tx_time);
1224 		mvm->radio_stats.on_time_rf =
1225 			le64_to_cpu(stats->general.common.on_time_rf);
1226 		mvm->radio_stats.on_time_scan =
1227 			le64_to_cpu(stats->general.common.on_time_scan);
1228 
1229 		data.beacon_counter = stats->general.beacon_counter;
1230 		data.beacon_average_energy =
1231 			stats->general.beacon_average_energy;
1232 		flags = stats->flag;
1233 	} else {
1234 		struct iwl_notif_statistics *stats = (void *)&pkt->data;
1235 
1236 		data.mac_id = stats->rx.general.mac_id;
1237 		data.beacon_filter_average_energy =
1238 			stats->general.common.beacon_filter_average_energy;
1239 
1240 		mvm->rx_stats = stats->rx;
1241 
1242 		mvm->radio_stats.rx_time =
1243 			le64_to_cpu(stats->general.common.rx_time);
1244 		mvm->radio_stats.tx_time =
1245 			le64_to_cpu(stats->general.common.tx_time);
1246 		mvm->radio_stats.on_time_rf =
1247 			le64_to_cpu(stats->general.common.on_time_rf);
1248 		mvm->radio_stats.on_time_scan =
1249 			le64_to_cpu(stats->general.common.on_time_scan);
1250 
1251 		data.beacon_counter = stats->general.beacon_counter;
1252 		data.beacon_average_energy =
1253 			stats->general.beacon_average_energy;
1254 		flags = stats->flag;
1255 	}
1256 	data.flags = flags;
1257 
1258 	iwl_mvm_rx_stats_check_trigger(mvm, pkt);
1259 
1260 	ieee80211_iterate_active_interfaces(mvm->hw,
1261 					    IEEE80211_IFACE_ITER_NORMAL,
1262 					    iwl_mvm_stat_iterator,
1263 					    &data);
1264 
1265 	if (!iwl_mvm_has_new_rx_api(mvm))
1266 		return;
1267 
1268 	if (!iwl_mvm_has_new_rx_stats_api(mvm)) {
1269 		struct iwl_notif_statistics_v11 *v11 = (void *)&pkt->data;
1270 
1271 		energy = (void *)&v11->load_stats.avg_energy;
1272 		bytes = (void *)&v11->load_stats.byte_count;
1273 		air_time = (void *)&v11->load_stats.air_time;
1274 	} else {
1275 		struct iwl_notif_statistics *stats = (void *)&pkt->data;
1276 
1277 		energy = (void *)&stats->load_stats.avg_energy;
1278 		bytes = (void *)&stats->load_stats.byte_count;
1279 		air_time = (void *)&stats->load_stats.air_time;
1280 	}
1281 	ieee80211_iterate_stations_atomic(mvm->hw, iwl_mvm_stats_energy_iter,
1282 					  energy);
1283 
1284 	/*
1285 	 * Don't update in case the statistics are not cleared, since
1286 	 * we will end up counting twice the same airtime, once in TCM
1287 	 * request and once in statistics notification.
1288 	 */
1289 	if (le32_to_cpu(flags) & IWL_STATISTICS_REPLY_FLG_CLEAR)
1290 		iwl_mvm_update_tcm_from_stats(mvm, air_time, bytes);
1291 
1292 }
1293 
1294 void iwl_mvm_rx_statistics(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb)
1295 {
1296 	iwl_mvm_handle_rx_statistics(mvm, rxb_addr(rxb));
1297 }
1298 
1299 void iwl_mvm_window_status_notif(struct iwl_mvm *mvm,
1300 				 struct iwl_rx_cmd_buffer *rxb)
1301 {
1302 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
1303 	struct iwl_ba_window_status_notif *notif = (void *)pkt->data;
1304 	int i;
1305 
1306 	BUILD_BUG_ON(ARRAY_SIZE(notif->ra_tid) != BA_WINDOW_STREAMS_MAX);
1307 	BUILD_BUG_ON(ARRAY_SIZE(notif->mpdu_rx_count) != BA_WINDOW_STREAMS_MAX);
1308 	BUILD_BUG_ON(ARRAY_SIZE(notif->bitmap) != BA_WINDOW_STREAMS_MAX);
1309 	BUILD_BUG_ON(ARRAY_SIZE(notif->start_seq_num) != BA_WINDOW_STREAMS_MAX);
1310 
1311 	rcu_read_lock();
1312 	for (i = 0; i < BA_WINDOW_STREAMS_MAX; i++) {
1313 		struct ieee80211_sta *sta;
1314 		u8 sta_id, tid;
1315 		u64 bitmap;
1316 		u32 ssn;
1317 		u16 ratid;
1318 		u16 received_mpdu;
1319 
1320 		ratid = le16_to_cpu(notif->ra_tid[i]);
1321 		/* check that this TID is valid */
1322 		if (!(ratid & BA_WINDOW_STATUS_VALID_MSK))
1323 			continue;
1324 
1325 		received_mpdu = le16_to_cpu(notif->mpdu_rx_count[i]);
1326 		if (received_mpdu == 0)
1327 			continue;
1328 
1329 		tid = ratid & BA_WINDOW_STATUS_TID_MSK;
1330 		/* get the station */
1331 		sta_id = (ratid & BA_WINDOW_STATUS_STA_ID_MSK)
1332 			 >> BA_WINDOW_STATUS_STA_ID_POS;
1333 		sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1334 		if (IS_ERR_OR_NULL(sta))
1335 			continue;
1336 		bitmap = le64_to_cpu(notif->bitmap[i]);
1337 		ssn = le32_to_cpu(notif->start_seq_num[i]);
1338 
1339 		/* update mac80211 with the bitmap for the reordering buffer */
1340 		ieee80211_mark_rx_ba_filtered_frames(sta, tid, ssn, bitmap,
1341 						     received_mpdu);
1342 	}
1343 	rcu_read_unlock();
1344 }
1345