drivers/net/wireless/realtek/rtw88/rx.c

   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
  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++;
  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
  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
  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
 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)))
 111                 return;
 112
 113         rtw_rx_phy_stat(rtwdev, pkt_stat, hdr);
 114         sta = ieee80211_find_sta_by_ifaddr(rtwdev->hw, hdr->addr2,
 115                                            vif->addr);
 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;
 148         u8 path;
 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
 163         if (rx_status->band == NL80211_BAND_5GHZ &&
 164             pkt_stat->rate >= DESC_RATE6M &&
 165             pkt_stat->rate <= DESC_RATE54M) {
 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;
 171         } else if (pkt_stat->rate >= DESC_RATEMCS0) {
 172                 rtw_desc_to_mcsrate(pkt_stat->rate, &rx_status->rate_idx,
 173                                     &rx_status->nss);
 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;
 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         }
 191
 192         rtw_rx_addr_match(rtwdev, pkt_stat, hdr);
 193 }