rtw88: add set_bitrate_mask support

Support setting bit rate from upper layer.
After configuring the original rate control result in the driver, the
result is then masked by the bit rate mask received from the ops
set_bitrate_mask. Lastly, the masked result will be sent to firmware.

Signed-off-by: Tzu-En Huang <tehuang@realtek.com>
Signed-off-by: Yan-Hsuan Chuang <yhchuang@realtek.com>
Reviewed-by: Chris Chiu <chiu@endlessm.com>
Signed-off-by: Kalle Valo <kvalo@codeaurora.org>

diff --git a/drivers/net/wireless/realtek/rtw88/mac80211.c b/drivers/net/wireless/realtek/rtw88/mac80211.c
index bc04cc280a968f57298f557b046da6e4632d1fa8..2247bd61e716d8791f6661b0715bf813287738c0 100644 (file)
--- a/drivers/net/wireless/realtek/rtw88/mac80211.c
+++ b/drivers/net/wireless/realtek/rtw88/mac80211.c
@@ -475,6 +475,8 @@ static int rtw_ops_sta_remove(struct ieee80211_hw *hw,
        for (i = 0; i < ARRAY_SIZE(sta->txq); i++)
                rtw_txq_cleanup(rtwdev, sta->txq[i]);
 
+       kfree(si->mask);
+
        rtwdev->sta_cnt--;
 
        rtw_info(rtwdev, "sta %pM with macid %d left\n",
@@ -684,6 +686,56 @@ static void rtw_ops_flush(struct ieee80211_hw *hw,
        mutex_unlock(&rtwdev->mutex);
 }
 
+struct rtw_iter_bitrate_mask_data {
+       struct rtw_dev *rtwdev;
+       struct ieee80211_vif *vif;
+       const struct cfg80211_bitrate_mask *mask;
+};
+
+static void rtw_ra_mask_info_update_iter(void *data, struct ieee80211_sta *sta)
+{
+       struct rtw_iter_bitrate_mask_data *br_data = data;
+       struct rtw_sta_info *si = (struct rtw_sta_info *)sta->drv_priv;
+
+       if (si->vif != br_data->vif)
+               return;
+
+       /* free previous mask setting */
+       kfree(si->mask);
+       si->mask = kmemdup(br_data->mask, sizeof(struct cfg80211_bitrate_mask),
+                          GFP_ATOMIC);
+       if (!si->mask) {
+               si->use_cfg_mask = false;
+               return;
+       }
+
+       si->use_cfg_mask = true;
+       rtw_update_sta_info(br_data->rtwdev, si);
+}
+
+static void rtw_ra_mask_info_update(struct rtw_dev *rtwdev,
+                                   struct ieee80211_vif *vif,
+                                   const struct cfg80211_bitrate_mask *mask)
+{
+       struct rtw_iter_bitrate_mask_data br_data;
+
+       br_data.rtwdev = rtwdev;
+       br_data.vif = vif;
+       br_data.mask = mask;
+       rtw_iterate_stas_atomic(rtwdev, rtw_ra_mask_info_update_iter, &br_data);
+}
+
+static int rtw_ops_set_bitrate_mask(struct ieee80211_hw *hw,
+                                   struct ieee80211_vif *vif,
+                                   const struct cfg80211_bitrate_mask *mask)
+{
+       struct rtw_dev *rtwdev = hw->priv;
+
+       rtw_ra_mask_info_update(rtwdev, vif, mask);
+
+       return 0;
+}
+
 const struct ieee80211_ops rtw_ops = {
        .tx                     = rtw_ops_tx,
        .wake_tx_queue          = rtw_ops_wake_tx_queue,
@@ -705,5 +757,6 @@ const struct ieee80211_ops rtw_ops = {
        .set_rts_threshold      = rtw_ops_set_rts_threshold,
        .sta_statistics         = rtw_ops_sta_statistics,
        .flush                  = rtw_ops_flush,
+       .set_bitrate_mask       = rtw_ops_set_bitrate_mask,
 };
 EXPORT_SYMBOL(rtw_ops);

diff --git a/drivers/net/wireless/realtek/rtw88/main.c b/drivers/net/wireless/realtek/rtw88/main.c
index 47e74f0aec06b8c1164c2f27992ecf77593ede80..e53143132a9b3ea1f179e9094b51ace1072d9976 100644 (file)
--- a/drivers/net/wireless/realtek/rtw88/main.c
+++ b/drivers/net/wireless/realtek/rtw88/main.c
@@ -612,12 +612,71 @@ static u8 get_rate_id(u8 wireless_set, enum rtw_bandwidth bw_mode, u8 tx_num)
 #define RA_MASK_OFDM_IN_HT_2G  0x00010
 #define RA_MASK_OFDM_IN_HT_5G  0x00030
 
+static u64 rtw_update_rate_mask(struct rtw_dev *rtwdev,
+                               struct rtw_sta_info *si,
+                               u64 ra_mask, bool is_vht_enable,
+                               u8 wireless_set)
+{
+       struct rtw_hal *hal = &rtwdev->hal;
+       const struct cfg80211_bitrate_mask *mask = si->mask;
+       u64 cfg_mask = GENMASK(63, 0);
+       u8 rssi_level, band;
+
+       if (wireless_set != WIRELESS_CCK) {
+               rssi_level = si->rssi_level;
+               if (rssi_level == 0)
+                       ra_mask &= 0xffffffffffffffffULL;
+               else if (rssi_level == 1)
+                       ra_mask &= 0xfffffffffffffff0ULL;
+               else if (rssi_level == 2)
+                       ra_mask &= 0xffffffffffffefe0ULL;
+               else if (rssi_level == 3)
+                       ra_mask &= 0xffffffffffffcfc0ULL;
+               else if (rssi_level == 4)
+                       ra_mask &= 0xffffffffffff8f80ULL;
+               else if (rssi_level >= 5)
+                       ra_mask &= 0xffffffffffff0f00ULL;
+       }
+
+       if (!si->use_cfg_mask)
+               return ra_mask;
+
+       band = hal->current_band_type;
+       if (band == RTW_BAND_2G) {
+               band = NL80211_BAND_2GHZ;
+               cfg_mask = mask->control[band].legacy;
+       } else if (band == RTW_BAND_5G) {
+               band = NL80211_BAND_5GHZ;
+               cfg_mask = u64_encode_bits(mask->control[band].legacy,
+                                          RA_MASK_OFDM_RATES);
+       }
+
+       if (!is_vht_enable) {
+               if (ra_mask & RA_MASK_HT_RATES_1SS)
+                       cfg_mask |= u64_encode_bits(mask->control[band].ht_mcs[0],
+                                                   RA_MASK_HT_RATES_1SS);
+               if (ra_mask & RA_MASK_HT_RATES_2SS)
+                       cfg_mask |= u64_encode_bits(mask->control[band].ht_mcs[1],
+                                                   RA_MASK_HT_RATES_2SS);
+       } else {
+               if (ra_mask & RA_MASK_VHT_RATES_1SS)
+                       cfg_mask |= u64_encode_bits(mask->control[band].vht_mcs[0],
+                                                   RA_MASK_VHT_RATES_1SS);
+               if (ra_mask & RA_MASK_VHT_RATES_2SS)
+                       cfg_mask |= u64_encode_bits(mask->control[band].vht_mcs[1],
+                                                   RA_MASK_VHT_RATES_2SS);
+       }
+
+       ra_mask &= cfg_mask;
+
+       return ra_mask;
+}
+
 void rtw_update_sta_info(struct rtw_dev *rtwdev, struct rtw_sta_info *si)
 {
        struct ieee80211_sta *sta = si->sta;
        struct rtw_efuse *efuse = &rtwdev->efuse;
        struct rtw_hal *hal = &rtwdev->hal;
-       u8 rssi_level;
        u8 wireless_set;
        u8 bw_mode;
        u8 rate_id;
@@ -710,21 +769,8 @@ void rtw_update_sta_info(struct rtw_dev *rtwdev, struct rtw_sta_info *si)
 
        rate_id = get_rate_id(wireless_set, bw_mode, tx_num);
 
-       if (wireless_set != WIRELESS_CCK) {
-               rssi_level = si->rssi_level;
-               if (rssi_level == 0)
-                       ra_mask &= 0xffffffffffffffffULL;
-               else if (rssi_level == 1)
-                       ra_mask &= 0xfffffffffffffff0ULL;
-               else if (rssi_level == 2)
-                       ra_mask &= 0xffffffffffffefe0ULL;
-               else if (rssi_level == 3)
-                       ra_mask &= 0xffffffffffffcfc0ULL;
-               else if (rssi_level == 4)
-                       ra_mask &= 0xffffffffffff8f80ULL;
-               else if (rssi_level >= 5)
-                       ra_mask &= 0xffffffffffff0f00ULL;
-       }
+       ra_mask = rtw_update_rate_mask(rtwdev, si, ra_mask, is_vht_enable,
+                                      wireless_set);
 
        si->bw_mode = bw_mode;
        si->stbc_en = stbc_en;

diff --git a/drivers/net/wireless/realtek/rtw88/main.h b/drivers/net/wireless/realtek/rtw88/main.h
index 14b35152fe9f326c2febf3dce906d93ae880a0b3..5e33b7e47a9df27d2400cd06542b5df31751c2a9 100644 (file)
--- a/drivers/net/wireless/realtek/rtw88/main.h
+++ b/drivers/net/wireless/realtek/rtw88/main.h
@@ -646,6 +646,9 @@ struct rtw_sta_info {
        DECLARE_BITMAP(tid_ba, IEEE80211_NUM_TIDS);
 
        struct rtw_ra_report ra_report;
+
+       bool use_cfg_mask;
+       struct cfg80211_bitrate_mask *mask;
 };
 
 enum rtw_bfee_role {