[mirror_ubuntu-jammy-kernel.git] / drivers / net / wireless / realtek / rtw88 / rx.c

// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/* Copyright(c) 2018-2019  Realtek Corporation
 */

#include "main.h"
#include "rx.h"
#include "ps.h"
#include "debug.h"

void rtw_rx_stats(struct rtw_dev *rtwdev, struct ieee80211_vif *vif,
		  struct sk_buff *skb)
{
	struct ieee80211_hdr *hdr;
	struct rtw_vif *rtwvif;

	hdr = (struct ieee80211_hdr *)skb->data;

	if (!ieee80211_is_data(hdr->frame_control))
		return;

	if (!is_broadcast_ether_addr(hdr->addr1) &&
	    !is_multicast_ether_addr(hdr->addr1)) {
		rtwdev->stats.rx_unicast += skb->len;
		rtwdev->stats.rx_cnt++;
		if (vif) {
			rtwvif = (struct rtw_vif *)vif->drv_priv;
			rtwvif->stats.rx_unicast += skb->len;
			rtwvif->stats.rx_cnt++;
		}
	}
}
EXPORT_SYMBOL(rtw_rx_stats);

struct rtw_rx_addr_match_data {
	struct rtw_dev *rtwdev;
	struct ieee80211_hdr *hdr;
	struct rtw_rx_pkt_stat *pkt_stat;
	u8 *bssid;
};

static void rtw_rx_phy_stat(struct rtw_dev *rtwdev,
			    struct rtw_rx_pkt_stat *pkt_stat,
			    struct ieee80211_hdr *hdr)
{
	struct rtw_dm_info *dm_info = &rtwdev->dm_info;
	struct rtw_pkt_count *cur_pkt_cnt = &dm_info->cur_pkt_count;
	u8 rate_ss, rate_ss_evm, evm_id;
	u8 i, idx;

	dm_info->curr_rx_rate = pkt_stat->rate;

	if (ieee80211_is_beacon(hdr->frame_control))
		cur_pkt_cnt->num_bcn_pkt++;

	switch (pkt_stat->rate) {
	case DESC_RATE1M...DESC_RATE11M:
		goto pkt_num;
	case DESC_RATE6M...DESC_RATE54M:
		rate_ss = 0;
		rate_ss_evm = 1;
		evm_id = RTW_EVM_OFDM;
		break;
	case DESC_RATEMCS0...DESC_RATEMCS7:
	case DESC_RATEVHT1SS_MCS0...DESC_RATEVHT1SS_MCS9:
		rate_ss = 1;
		rate_ss_evm = 1;
		evm_id = RTW_EVM_1SS;
		break;
	case DESC_RATEMCS8...DESC_RATEMCS15:
	case DESC_RATEVHT2SS_MCS0...DESC_RATEVHT2SS_MCS9:
		rate_ss = 2;
		rate_ss_evm = 2;
		evm_id = RTW_EVM_2SS_A;
		break;
	default:
		rtw_warn(rtwdev, "unknown pkt rate = %d\n", pkt_stat->rate);
		return;
	}

	for (i = 0; i < rate_ss_evm; i++) {
		idx = evm_id + i;
		ewma_evm_add(&dm_info->ewma_evm[idx],
			     dm_info->rx_evm_dbm[i]);
	}

	for (i = 0; i < rtwdev->hal.rf_path_num; i++) {
		idx = RTW_SNR_OFDM_A + 4 * rate_ss + i;
		ewma_snr_add(&dm_info->ewma_snr[idx],
			     dm_info->rx_snr[i]);
	}
pkt_num:
	cur_pkt_cnt->num_qry_pkt[pkt_stat->rate]++;
}

static void rtw_rx_addr_match_iter(void *data, u8 *mac,
				   struct ieee80211_vif *vif)
{
	struct rtw_rx_addr_match_data *iter_data = data;
	struct ieee80211_sta *sta;
	struct ieee80211_hdr *hdr = iter_data->hdr;
	struct rtw_dev *rtwdev = iter_data->rtwdev;
	struct rtw_sta_info *si;
	struct rtw_rx_pkt_stat *pkt_stat = iter_data->pkt_stat;
	u8 *bssid = iter_data->bssid;

	if (!ether_addr_equal(vif->bss_conf.bssid, bssid))
		return;

	if (!(ether_addr_equal(vif->addr, hdr->addr1) ||
	      ieee80211_is_beacon(hdr->frame_control)))
		return;

	rtw_rx_phy_stat(rtwdev, pkt_stat, hdr);
	sta = ieee80211_find_sta_by_ifaddr(rtwdev->hw, hdr->addr2,
					   vif->addr);
	if (!sta)
		return;

	si = (struct rtw_sta_info *)sta->drv_priv;
	ewma_rssi_add(&si->avg_rssi, pkt_stat->rssi);
}

static void rtw_rx_addr_match(struct rtw_dev *rtwdev,
			      struct rtw_rx_pkt_stat *pkt_stat,
			      struct ieee80211_hdr *hdr)
{
	struct rtw_rx_addr_match_data data = {};

	if (pkt_stat->crc_err || pkt_stat->icv_err || !pkt_stat->phy_status ||
	    ieee80211_is_ctl(hdr->frame_control))
		return;

	data.rtwdev = rtwdev;
	data.hdr = hdr;
	data.pkt_stat = pkt_stat;
	data.bssid = get_hdr_bssid(hdr);

	rtw_iterate_vifs_atomic(rtwdev, rtw_rx_addr_match_iter, &data);
}

void rtw_rx_fill_rx_status(struct rtw_dev *rtwdev,
			   struct rtw_rx_pkt_stat *pkt_stat,
			   struct ieee80211_hdr *hdr,
			   struct ieee80211_rx_status *rx_status,
			   u8 *phy_status)
{
	struct ieee80211_hw *hw = rtwdev->hw;
	u8 path;

	memset(rx_status, 0, sizeof(*rx_status));
	rx_status->freq = hw->conf.chandef.chan->center_freq;
	rx_status->band = hw->conf.chandef.chan->band;
	if (pkt_stat->crc_err)
		rx_status->flag |= RX_FLAG_FAILED_FCS_CRC;
	if (pkt_stat->decrypted)
		rx_status->flag |= RX_FLAG_DECRYPTED;

	if (pkt_stat->rate >= DESC_RATEVHT1SS_MCS0)
		rx_status->encoding = RX_ENC_VHT;
	else if (pkt_stat->rate >= DESC_RATEMCS0)
		rx_status->encoding = RX_ENC_HT;

	if (rx_status->band == NL80211_BAND_5GHZ &&
	    pkt_stat->rate >= DESC_RATE6M &&
	    pkt_stat->rate <= DESC_RATE54M) {
		rx_status->rate_idx = pkt_stat->rate - DESC_RATE6M;
	} else if (rx_status->band == NL80211_BAND_2GHZ &&
		   pkt_stat->rate >= DESC_RATE1M &&
		   pkt_stat->rate <= DESC_RATE54M) {
		rx_status->rate_idx = pkt_stat->rate - DESC_RATE1M;
	} else if (pkt_stat->rate >= DESC_RATEMCS0) {
		rtw_desc_to_mcsrate(pkt_stat->rate, &rx_status->rate_idx,
				    &rx_status->nss);
	}

	rx_status->flag |= RX_FLAG_MACTIME_START;
	rx_status->mactime = pkt_stat->tsf_low;

	if (pkt_stat->bw == RTW_CHANNEL_WIDTH_80)
		rx_status->bw = RATE_INFO_BW_80;
	else if (pkt_stat->bw == RTW_CHANNEL_WIDTH_40)
		rx_status->bw = RATE_INFO_BW_40;
	else
		rx_status->bw = RATE_INFO_BW_20;

	rx_status->signal = pkt_stat->signal_power;
	for (path = 0; path < rtwdev->hal.rf_path_num; path++) {
		rx_status->chains |= BIT(path);
		rx_status->chain_signal[path] = pkt_stat->rx_power[path];
	}

	rtw_rx_addr_match(rtwdev, pkt_stat, hdr);
}
Commit	Line	Data
e3037485 YHC	1	// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
	2	/* Copyright(c) 2018-2019 Realtek Corporation
	3	*/
	4
	5	#include "main.h"
	6	#include "rx.h"
	7	#include "ps.h"
082a36dc	8	#include "debug.h"
e3037485 YHC	9
	10	void rtw_rx_stats(struct rtw_dev rtwdev, struct ieee80211_vif vif,
	11	struct sk_buff *skb)
	12	{
	13	struct ieee80211_hdr *hdr;
	14	struct rtw_vif *rtwvif;
	15
	16	hdr = (struct ieee80211_hdr *)skb->data;
	17
	18	if (!ieee80211_is_data(hdr->frame_control))
	19	return;
	20
	21	if (!is_broadcast_ether_addr(hdr->addr1) &&
	22	!is_multicast_ether_addr(hdr->addr1)) {
	23	rtwdev->stats.rx_unicast += skb->len;
	24	rtwdev->stats.rx_cnt++;
	25	if (vif) {
	26	rtwvif = (struct rtw_vif *)vif->drv_priv;
	27	rtwvif->stats.rx_unicast += skb->len;
	28	rtwvif->stats.rx_cnt++;
e3037485 YHC	29	}
	30	}
	31	}
	32	EXPORT_SYMBOL(rtw_rx_stats);
	33
	34	struct rtw_rx_addr_match_data {
	35	struct rtw_dev *rtwdev;
	36	struct ieee80211_hdr *hdr;
	37	struct rtw_rx_pkt_stat *pkt_stat;
	38	u8 *bssid;
	39	};
	40
082a36dc TSL	41	static void rtw_rx_phy_stat(struct rtw_dev *rtwdev,
	42	struct rtw_rx_pkt_stat *pkt_stat,
	43	struct ieee80211_hdr *hdr)
	44	{
	45	struct rtw_dm_info *dm_info = &rtwdev->dm_info;
	46	struct rtw_pkt_count *cur_pkt_cnt = &dm_info->cur_pkt_count;
	47	u8 rate_ss, rate_ss_evm, evm_id;
	48	u8 i, idx;
	49
	50	dm_info->curr_rx_rate = pkt_stat->rate;
	51
	52	if (ieee80211_is_beacon(hdr->frame_control))
	53	cur_pkt_cnt->num_bcn_pkt++;
	54
	55	switch (pkt_stat->rate) {
	56	case DESC_RATE1M...DESC_RATE11M:
	57	goto pkt_num;
	58	case DESC_RATE6M...DESC_RATE54M:
	59	rate_ss = 0;
	60	rate_ss_evm = 1;
	61	evm_id = RTW_EVM_OFDM;
	62	break;
	63	case DESC_RATEMCS0...DESC_RATEMCS7:
	64	case DESC_RATEVHT1SS_MCS0...DESC_RATEVHT1SS_MCS9:
	65	rate_ss = 1;
	66	rate_ss_evm = 1;
	67	evm_id = RTW_EVM_1SS;
	68	break;
	69	case DESC_RATEMCS8...DESC_RATEMCS15:
	70	case DESC_RATEVHT2SS_MCS0...DESC_RATEVHT2SS_MCS9:
	71	rate_ss = 2;
	72	rate_ss_evm = 2;
	73	evm_id = RTW_EVM_2SS_A;
	74	break;
	75	default:
	76	rtw_warn(rtwdev, "unknown pkt rate = %d\n", pkt_stat->rate);
	77	return;
	78	}
	79
	80	for (i = 0; i < rate_ss_evm; i++) {
	81	idx = evm_id + i;
	82	ewma_evm_add(&dm_info->ewma_evm[idx],
	83	dm_info->rx_evm_dbm[i]);
	84	}
	85
	86	for (i = 0; i < rtwdev->hal.rf_path_num; i++) {
	87	idx = RTW_SNR_OFDM_A + 4 * rate_ss + i;
	88	ewma_snr_add(&dm_info->ewma_snr[idx],
	89	dm_info->rx_snr[i]);
	90	}
	91	pkt_num:
	92	cur_pkt_cnt->num_qry_pkt[pkt_stat->rate]++;
	93	}
	94
e3037485 YHC	95	static void rtw_rx_addr_match_iter(void data, u8 mac,
	96	struct ieee80211_vif *vif)
	97	{
	98	struct rtw_rx_addr_match_data *iter_data = data;
	99	struct ieee80211_sta *sta;
	100	struct ieee80211_hdr *hdr = iter_data->hdr;
	101	struct rtw_dev *rtwdev = iter_data->rtwdev;
	102	struct rtw_sta_info *si;
	103	struct rtw_rx_pkt_stat *pkt_stat = iter_data->pkt_stat;
	104	u8 *bssid = iter_data->bssid;
	105
082a36dc TSL	106	if (!ether_addr_equal(vif->bss_conf.bssid, bssid))
	107	return;
	108
	109	if (!(ether_addr_equal(vif->addr, hdr->addr1) \|\|
	110	ieee80211_is_beacon(hdr->frame_control)))
e3037485 YHC	111	return;
e3037485 YHC	112
082a36dc TSL	113	rtw_rx_phy_stat(rtwdev, pkt_stat, hdr);
	114	sta = ieee80211_find_sta_by_ifaddr(rtwdev->hw, hdr->addr2,
	115	vif->addr);
e3037485 YHC	116	if (!sta)
	117	return;
	118
	119	si = (struct rtw_sta_info *)sta->drv_priv;
	120	ewma_rssi_add(&si->avg_rssi, pkt_stat->rssi);
	121	}
	122
	123	static void rtw_rx_addr_match(struct rtw_dev *rtwdev,
	124	struct rtw_rx_pkt_stat *pkt_stat,
	125	struct ieee80211_hdr *hdr)
	126	{
	127	struct rtw_rx_addr_match_data data = {};
	128
	129	if (pkt_stat->crc_err \|\| pkt_stat->icv_err \|\| !pkt_stat->phy_status \|\|
	130	ieee80211_is_ctl(hdr->frame_control))
	131	return;
	132
	133	data.rtwdev = rtwdev;
	134	data.hdr = hdr;
	135	data.pkt_stat = pkt_stat;
	136	data.bssid = get_hdr_bssid(hdr);
	137
	138	rtw_iterate_vifs_atomic(rtwdev, rtw_rx_addr_match_iter, &data);
	139	}
	140
	141	void rtw_rx_fill_rx_status(struct rtw_dev *rtwdev,
	142	struct rtw_rx_pkt_stat *pkt_stat,
	143	struct ieee80211_hdr *hdr,
	144	struct ieee80211_rx_status *rx_status,
	145	u8 *phy_status)
	146	{
	147	struct ieee80211_hw *hw = rtwdev->hw;
98ab76ef	148	u8 path;
e3037485 YHC	149
	150	memset(rx_status, 0, sizeof(*rx_status));
	151	rx_status->freq = hw->conf.chandef.chan->center_freq;
	152	rx_status->band = hw->conf.chandef.chan->band;
	153	if (pkt_stat->crc_err)
	154	rx_status->flag \|= RX_FLAG_FAILED_FCS_CRC;
	155	if (pkt_stat->decrypted)
	156	rx_status->flag \|= RX_FLAG_DECRYPTED;
	157
	158	if (pkt_stat->rate >= DESC_RATEVHT1SS_MCS0)
	159	rx_status->encoding = RX_ENC_VHT;
	160	else if (pkt_stat->rate >= DESC_RATEMCS0)
	161	rx_status->encoding = RX_ENC_HT;
	162
ff0dfe5b YHC	163	if (rx_status->band == NL80211_BAND_5GHZ &&
	164	pkt_stat->rate >= DESC_RATE6M &&
	165	pkt_stat->rate <= DESC_RATE54M) {
e3037485 YHC	166	rx_status->rate_idx = pkt_stat->rate - DESC_RATE6M;
	167	} else if (rx_status->band == NL80211_BAND_2GHZ &&
	168	pkt_stat->rate >= DESC_RATE1M &&
	169	pkt_stat->rate <= DESC_RATE54M) {
	170	rx_status->rate_idx = pkt_stat->rate - DESC_RATE1M;
ff0dfe5b YHC	171	} else if (pkt_stat->rate >= DESC_RATEMCS0) {
	172	rtw_desc_to_mcsrate(pkt_stat->rate, &rx_status->rate_idx,
	173	&rx_status->nss);
e3037485 YHC	174	}
	175
	176	rx_status->flag \|= RX_FLAG_MACTIME_START;
	177	rx_status->mactime = pkt_stat->tsf_low;
	178
	179	if (pkt_stat->bw == RTW_CHANNEL_WIDTH_80)
	180	rx_status->bw = RATE_INFO_BW_80;
	181	else if (pkt_stat->bw == RTW_CHANNEL_WIDTH_40)
	182	rx_status->bw = RATE_INFO_BW_40;
	183	else
	184	rx_status->bw = RATE_INFO_BW_20;
	185
	186	rx_status->signal = pkt_stat->signal_power;
98ab76ef YHC	187	for (path = 0; path < rtwdev->hal.rf_path_num; path++) {
	188	rx_status->chains \|= BIT(path);
	189	rx_status->chain_signal[path] = pkt_stat->rx_power[path];
	190	}
e3037485 YHC	191
	192	rtw_rx_addr_match(rtwdev, pkt_stat, hdr);
	193	}