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"
8 #include "debug.h"
9 #include "fw.h"
10
rtw_rx_stats(struct rtw_dev * rtwdev,struct ieee80211_vif * vif,struct sk_buff * skb)11 void rtw_rx_stats(struct rtw_dev *rtwdev, struct ieee80211_vif *vif,
12 struct sk_buff *skb)
13 {
14 struct ieee80211_hdr *hdr;
15 struct rtw_vif *rtwvif;
16
17 hdr = (struct ieee80211_hdr *)skb->data;
18
19 if (!ieee80211_is_data(hdr->frame_control))
20 return;
21
22 if (!is_broadcast_ether_addr(hdr->addr1) &&
23 !is_multicast_ether_addr(hdr->addr1)) {
24 rtwdev->stats.rx_unicast += skb->len;
25 rtwdev->stats.rx_cnt++;
26 if (vif) {
27 rtwvif = (struct rtw_vif *)vif->drv_priv;
28 rtwvif->stats.rx_unicast += skb->len;
29 rtwvif->stats.rx_cnt++;
30 }
31 }
32 }
33 EXPORT_SYMBOL(rtw_rx_stats);
34
35 struct rtw_rx_addr_match_data {
36 struct rtw_dev *rtwdev;
37 struct ieee80211_hdr *hdr;
38 struct rtw_rx_pkt_stat *pkt_stat;
39 u8 *bssid;
40 };
41
rtw_rx_phy_stat(struct rtw_dev * rtwdev,struct rtw_rx_pkt_stat * pkt_stat,struct ieee80211_hdr * hdr)42 static void rtw_rx_phy_stat(struct rtw_dev *rtwdev,
43 struct rtw_rx_pkt_stat *pkt_stat,
44 struct ieee80211_hdr *hdr)
45 {
46 struct rtw_dm_info *dm_info = &rtwdev->dm_info;
47 struct rtw_pkt_count *cur_pkt_cnt = &dm_info->cur_pkt_count;
48 u8 rate_ss, rate_ss_evm, evm_id;
49 u8 i, idx;
50
51 dm_info->curr_rx_rate = pkt_stat->rate;
52
53 if (ieee80211_is_beacon(hdr->frame_control))
54 cur_pkt_cnt->num_bcn_pkt++;
55
56 switch (pkt_stat->rate) {
57 case DESC_RATE1M...DESC_RATE11M:
58 goto pkt_num;
59 case DESC_RATE6M...DESC_RATE54M:
60 rate_ss = 0;
61 rate_ss_evm = 1;
62 evm_id = RTW_EVM_OFDM;
63 break;
64 case DESC_RATEMCS0...DESC_RATEMCS7:
65 case DESC_RATEVHT1SS_MCS0...DESC_RATEVHT1SS_MCS9:
66 rate_ss = 1;
67 rate_ss_evm = 1;
68 evm_id = RTW_EVM_1SS;
69 break;
70 case DESC_RATEMCS8...DESC_RATEMCS15:
71 case DESC_RATEVHT2SS_MCS0...DESC_RATEVHT2SS_MCS9:
72 rate_ss = 2;
73 rate_ss_evm = 2;
74 evm_id = RTW_EVM_2SS_A;
75 break;
76 default:
77 rtw_warn(rtwdev, "unknown pkt rate = %d\n", pkt_stat->rate);
78 return;
79 }
80
81 for (i = 0; i < rate_ss_evm; i++) {
82 idx = evm_id + i;
83 ewma_evm_add(&dm_info->ewma_evm[idx],
84 dm_info->rx_evm_dbm[i]);
85 }
86
87 for (i = 0; i < rtwdev->hal.rf_path_num; i++) {
88 idx = RTW_SNR_OFDM_A + 4 * rate_ss + i;
89 ewma_snr_add(&dm_info->ewma_snr[idx],
90 dm_info->rx_snr[i]);
91 }
92 pkt_num:
93 cur_pkt_cnt->num_qry_pkt[pkt_stat->rate]++;
94 }
95
rtw_rx_addr_match_iter(void * data,u8 * mac,struct ieee80211_vif * vif)96 static void rtw_rx_addr_match_iter(void *data, u8 *mac,
97 struct ieee80211_vif *vif)
98 {
99 struct rtw_rx_addr_match_data *iter_data = data;
100 struct ieee80211_sta *sta;
101 struct ieee80211_hdr *hdr = iter_data->hdr;
102 struct rtw_dev *rtwdev = iter_data->rtwdev;
103 struct rtw_sta_info *si;
104 struct rtw_rx_pkt_stat *pkt_stat = iter_data->pkt_stat;
105 u8 *bssid = iter_data->bssid;
106
107 if (!ether_addr_equal(vif->bss_conf.bssid, bssid))
108 return;
109
110 if (!(ether_addr_equal(vif->addr, hdr->addr1) ||
111 ieee80211_is_beacon(hdr->frame_control)))
112 return;
113
114 rtw_rx_phy_stat(rtwdev, pkt_stat, hdr);
115 sta = ieee80211_find_sta_by_ifaddr(rtwdev->hw, hdr->addr2,
116 vif->addr);
117 if (!sta)
118 return;
119
120 si = (struct rtw_sta_info *)sta->drv_priv;
121 ewma_rssi_add(&si->avg_rssi, pkt_stat->rssi);
122 }
123
rtw_rx_addr_match(struct rtw_dev * rtwdev,struct rtw_rx_pkt_stat * pkt_stat,struct ieee80211_hdr * hdr)124 static void rtw_rx_addr_match(struct rtw_dev *rtwdev,
125 struct rtw_rx_pkt_stat *pkt_stat,
126 struct ieee80211_hdr *hdr)
127 {
128 struct rtw_rx_addr_match_data data = {};
129
130 if (pkt_stat->crc_err || pkt_stat->icv_err || !pkt_stat->phy_status ||
131 ieee80211_is_ctl(hdr->frame_control))
132 return;
133
134 data.rtwdev = rtwdev;
135 data.hdr = hdr;
136 data.pkt_stat = pkt_stat;
137 data.bssid = get_hdr_bssid(hdr);
138
139 rtw_iterate_vifs_atomic(rtwdev, rtw_rx_addr_match_iter, &data);
140 }
141
rtw_set_rx_freq_by_pktstat(struct rtw_rx_pkt_stat * pkt_stat,struct ieee80211_rx_status * rx_status)142 static void rtw_set_rx_freq_by_pktstat(struct rtw_rx_pkt_stat *pkt_stat,
143 struct ieee80211_rx_status *rx_status)
144 {
145 rx_status->freq = pkt_stat->freq;
146 rx_status->band = pkt_stat->band;
147 }
148
rtw_update_rx_freq_from_ie(struct rtw_dev * rtwdev,struct sk_buff * skb,struct ieee80211_rx_status * rx_status,struct rtw_rx_pkt_stat * pkt_stat)149 void rtw_update_rx_freq_from_ie(struct rtw_dev *rtwdev, struct sk_buff *skb,
150 struct ieee80211_rx_status *rx_status,
151 struct rtw_rx_pkt_stat *pkt_stat)
152 {
153 struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)skb->data;
154 int channel = rtwdev->hal.current_channel;
155 size_t hdr_len, ielen;
156 int channel_number;
157 u8 *variable;
158
159 if (!test_bit(RTW_FLAG_SCANNING, rtwdev->flags))
160 goto fill_rx_status;
161
162 if (ieee80211_is_beacon(mgmt->frame_control)) {
163 variable = mgmt->u.beacon.variable;
164 hdr_len = offsetof(struct ieee80211_mgmt,
165 u.beacon.variable);
166 } else if (ieee80211_is_probe_resp(mgmt->frame_control)) {
167 variable = mgmt->u.probe_resp.variable;
168 hdr_len = offsetof(struct ieee80211_mgmt,
169 u.probe_resp.variable);
170 } else {
171 goto fill_rx_status;
172 }
173
174 if (skb->len > hdr_len)
175 ielen = skb->len - hdr_len;
176 else
177 goto fill_rx_status;
178
179 channel_number = cfg80211_get_ies_channel_number(variable, ielen,
180 NL80211_BAND_2GHZ);
181 if (channel_number != -1)
182 channel = channel_number;
183
184 fill_rx_status:
185 rtw_set_rx_freq_band(pkt_stat, channel);
186 rtw_set_rx_freq_by_pktstat(pkt_stat, rx_status);
187 }
188 EXPORT_SYMBOL(rtw_update_rx_freq_from_ie);
189
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)190 void rtw_rx_fill_rx_status(struct rtw_dev *rtwdev,
191 struct rtw_rx_pkt_stat *pkt_stat,
192 struct ieee80211_hdr *hdr,
193 struct ieee80211_rx_status *rx_status,
194 u8 *phy_status)
195 {
196 struct ieee80211_hw *hw = rtwdev->hw;
197 u8 path;
198
199 memset(rx_status, 0, sizeof(*rx_status));
200 rx_status->freq = hw->conf.chandef.chan->center_freq;
201 rx_status->band = hw->conf.chandef.chan->band;
202 if (rtw_fw_feature_check(&rtwdev->fw, FW_FEATURE_SCAN_OFFLOAD) &&
203 test_bit(RTW_FLAG_SCANNING, rtwdev->flags))
204 rtw_set_rx_freq_by_pktstat(pkt_stat, rx_status);
205 if (pkt_stat->crc_err)
206 rx_status->flag |= RX_FLAG_FAILED_FCS_CRC;
207 if (pkt_stat->decrypted)
208 rx_status->flag |= RX_FLAG_DECRYPTED;
209
210 if (pkt_stat->rate >= DESC_RATEVHT1SS_MCS0)
211 rx_status->encoding = RX_ENC_VHT;
212 else if (pkt_stat->rate >= DESC_RATEMCS0)
213 rx_status->encoding = RX_ENC_HT;
214
215 if (rx_status->band == NL80211_BAND_5GHZ &&
216 pkt_stat->rate >= DESC_RATE6M &&
217 pkt_stat->rate <= DESC_RATE54M) {
218 rx_status->rate_idx = pkt_stat->rate - DESC_RATE6M;
219 } else if (rx_status->band == NL80211_BAND_2GHZ &&
220 pkt_stat->rate >= DESC_RATE1M &&
221 pkt_stat->rate <= DESC_RATE54M) {
222 rx_status->rate_idx = pkt_stat->rate - DESC_RATE1M;
223 } else if (pkt_stat->rate >= DESC_RATEMCS0) {
224 rtw_desc_to_mcsrate(pkt_stat->rate, &rx_status->rate_idx,
225 &rx_status->nss);
226 }
227
228 rx_status->flag |= RX_FLAG_MACTIME_START;
229 rx_status->mactime = pkt_stat->tsf_low;
230
231 if (pkt_stat->bw == RTW_CHANNEL_WIDTH_80)
232 rx_status->bw = RATE_INFO_BW_80;
233 else if (pkt_stat->bw == RTW_CHANNEL_WIDTH_40)
234 rx_status->bw = RATE_INFO_BW_40;
235 else
236 rx_status->bw = RATE_INFO_BW_20;
237
238 rx_status->signal = pkt_stat->signal_power;
239 for (path = 0; path < rtwdev->hal.rf_path_num; path++) {
240 rx_status->chains |= BIT(path);
241 rx_status->chain_signal[path] = pkt_stat->rx_power[path];
242 }
243
244 rtw_rx_addr_match(rtwdev, pkt_stat, hdr);
245 }
246 EXPORT_SYMBOL(rtw_rx_fill_rx_status);
247