drivers/net/wireless/intel/iwlwifi/mvm/rs-fw.c

   1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
   2 /*
   3  * Copyright (C) 2017 Intel Deutschland GmbH
   4  * Copyright (C) 2018-2024 Intel Corporation
   5  */
   6 #include "rs.h"
   7 #include "fw-api.h"
   8 #include "sta.h"
   9 #include "iwl-op-mode.h"
  10 #include "mvm.h"
  11
  12 static u8 rs_fw_bw_from_sta_bw(const struct ieee80211_link_sta *link_sta)
  13 {
  14         switch (link_sta->bandwidth) {
  15         case IEEE80211_STA_RX_BW_320:
  16                 return IWL_TLC_MNG_CH_WIDTH_320MHZ;
  17         case IEEE80211_STA_RX_BW_160:
  18                 return IWL_TLC_MNG_CH_WIDTH_160MHZ;
  19         case IEEE80211_STA_RX_BW_80:
  20                 return IWL_TLC_MNG_CH_WIDTH_80MHZ;
  21         case IEEE80211_STA_RX_BW_40:
  22                 return IWL_TLC_MNG_CH_WIDTH_40MHZ;
  23         case IEEE80211_STA_RX_BW_20:
  24         default:
  25                 return IWL_TLC_MNG_CH_WIDTH_20MHZ;
  26         }
  27 }
  28
  29 static u8 rs_fw_set_active_chains(u8 chains)
  30 {
  31         u8 fw_chains = 0;
  32
  33         if (chains & ANT_A)
  34                 fw_chains |= IWL_TLC_MNG_CHAIN_A_MSK;
  35         if (chains & ANT_B)
  36                 fw_chains |= IWL_TLC_MNG_CHAIN_B_MSK;
  37
  38         return fw_chains;
  39 }
  40
  41 static u8 rs_fw_sgi_cw_support(struct ieee80211_link_sta *link_sta)
  42 {
  43         struct ieee80211_sta_ht_cap *ht_cap = &link_sta->ht_cap;
  44         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->vht_cap;
  45         struct ieee80211_sta_he_cap *he_cap = &link_sta->he_cap;
  46         u8 supp = 0;
  47
  48         if (he_cap->has_he)
  49                 return 0;
  50
  51         if (ht_cap->cap & IEEE80211_HT_CAP_SGI_20)
  52                 supp |= BIT(IWL_TLC_MNG_CH_WIDTH_20MHZ);
  53         if (ht_cap->cap & IEEE80211_HT_CAP_SGI_40)
  54                 supp |= BIT(IWL_TLC_MNG_CH_WIDTH_40MHZ);
  55         if (vht_cap->cap & IEEE80211_VHT_CAP_SHORT_GI_80)
  56                 supp |= BIT(IWL_TLC_MNG_CH_WIDTH_80MHZ);
  57         if (vht_cap->cap & IEEE80211_VHT_CAP_SHORT_GI_160)
  58                 supp |= BIT(IWL_TLC_MNG_CH_WIDTH_160MHZ);
  59
  60         return supp;
  61 }
  62
  63 static u16 rs_fw_get_config_flags(struct iwl_mvm *mvm,
  64                                   struct ieee80211_vif *vif,
  65                                   struct ieee80211_link_sta *link_sta,
  66                                   const struct ieee80211_sta_he_cap *sband_he_cap)
  67 {
  68         struct ieee80211_sta_ht_cap *ht_cap = &link_sta->ht_cap;
  69         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->vht_cap;
  70         struct ieee80211_sta_he_cap *he_cap = &link_sta->he_cap;
  71         bool vht_ena = vht_cap->vht_supported;
  72         u16 flags = 0;
  73
  74         /* get STBC flags */
  75         if (mvm->cfg->ht_params->stbc &&
  76             (num_of_ant(iwl_mvm_get_valid_tx_ant(mvm)) > 1)) {
  77                 if (he_cap->has_he && he_cap->he_cap_elem.phy_cap_info[2] &
  78                                       IEEE80211_HE_PHY_CAP2_STBC_RX_UNDER_80MHZ)
  79                         flags |= IWL_TLC_MNG_CFG_FLAGS_STBC_MSK;
  80                 else if (vht_cap->cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
  81                         flags |= IWL_TLC_MNG_CFG_FLAGS_STBC_MSK;
  82                 else if (ht_cap->cap & IEEE80211_HT_CAP_RX_STBC)
  83                         flags |= IWL_TLC_MNG_CFG_FLAGS_STBC_MSK;
  84         }
  85
  86         if (mvm->cfg->ht_params->ldpc &&
  87             ((ht_cap->cap & IEEE80211_HT_CAP_LDPC_CODING) ||
  88              (vht_ena && (vht_cap->cap & IEEE80211_VHT_CAP_RXLDPC))))
  89                 flags |= IWL_TLC_MNG_CFG_FLAGS_LDPC_MSK;
  90
  91         /* consider LDPC support in case of HE */
  92         if (he_cap->has_he && (he_cap->he_cap_elem.phy_cap_info[1] &
  93             IEEE80211_HE_PHY_CAP1_LDPC_CODING_IN_PAYLOAD))
  94                 flags |= IWL_TLC_MNG_CFG_FLAGS_LDPC_MSK;
  95
  96         if (sband_he_cap &&
  97             !(sband_he_cap->he_cap_elem.phy_cap_info[1] &
  98                         IEEE80211_HE_PHY_CAP1_LDPC_CODING_IN_PAYLOAD))
  99                 flags &= ~IWL_TLC_MNG_CFG_FLAGS_LDPC_MSK;
 100
 101         if (he_cap->has_he &&
 102             (he_cap->he_cap_elem.phy_cap_info[3] &
 103              IEEE80211_HE_PHY_CAP3_DCM_MAX_CONST_RX_MASK &&
 104              sband_he_cap &&
 105              sband_he_cap->he_cap_elem.phy_cap_info[3] &
 106                         IEEE80211_HE_PHY_CAP3_DCM_MAX_CONST_TX_MASK))
 107                 flags |= IWL_TLC_MNG_CFG_FLAGS_HE_DCM_NSS_1_MSK;
 108
 109         return flags;
 110 }
 111
 112 static
 113 int rs_fw_vht_highest_rx_mcs_index(const struct ieee80211_sta_vht_cap *vht_cap,
 114                                    int nss)
 115 {
 116         u16 rx_mcs = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map) &
 117                 (0x3 << (2 * (nss - 1)));
 118         rx_mcs >>= (2 * (nss - 1));
 119
 120         switch (rx_mcs) {
 121         case IEEE80211_VHT_MCS_SUPPORT_0_7:
 122                 return IWL_TLC_MNG_HT_RATE_MCS7;
 123         case IEEE80211_VHT_MCS_SUPPORT_0_8:
 124                 return IWL_TLC_MNG_HT_RATE_MCS8;
 125         case IEEE80211_VHT_MCS_SUPPORT_0_9:
 126                 return IWL_TLC_MNG_HT_RATE_MCS9;
 127         default:
 128                 WARN_ON_ONCE(1);
 129                 break;
 130         }
 131
 132         return 0;
 133 }
 134
 135 static void
 136 rs_fw_vht_set_enabled_rates(const struct ieee80211_link_sta *link_sta,
 137                             const struct ieee80211_sta_vht_cap *vht_cap,
 138                             struct iwl_tlc_config_cmd_v4 *cmd)
 139 {
 140         u16 supp;
 141         int i, highest_mcs;
 142         u8 max_nss = link_sta->rx_nss;
 143         struct ieee80211_vht_cap ieee_vht_cap = {
 144                 .vht_cap_info = cpu_to_le32(vht_cap->cap),
 145                 .supp_mcs = vht_cap->vht_mcs,
 146         };
 147
 148         /* the station support only a single receive chain */
 149         if (link_sta->smps_mode == IEEE80211_SMPS_STATIC)
 150                 max_nss = 1;
 151
 152         for (i = 0; i < max_nss && i < IWL_TLC_NSS_MAX; i++) {
 153                 int nss = i + 1;
 154
 155                 highest_mcs = rs_fw_vht_highest_rx_mcs_index(vht_cap, nss);
 156                 if (!highest_mcs)
 157                         continue;
 158
 159                 supp = BIT(highest_mcs + 1) - 1;
 160                 if (link_sta->bandwidth == IEEE80211_STA_RX_BW_20)
 161                         supp &= ~BIT(IWL_TLC_MNG_HT_RATE_MCS9);
 162
 163                 cmd->ht_rates[i][IWL_TLC_MCS_PER_BW_80] = cpu_to_le16(supp);
 164                 /*
 165                  * Check if VHT extended NSS indicates that the bandwidth/NSS
 166                  * configuration is supported - only for MCS 0 since we already
 167                  * decoded the MCS bits anyway ourselves.
 168                  */
 169                 if (link_sta->bandwidth == IEEE80211_STA_RX_BW_160 &&
 170                     ieee80211_get_vht_max_nss(&ieee_vht_cap,
 171                                               IEEE80211_VHT_CHANWIDTH_160MHZ,
 172                                               0, true, nss) >= nss)
 173                         cmd->ht_rates[i][IWL_TLC_MCS_PER_BW_160] =
 174                                 cmd->ht_rates[i][IWL_TLC_MCS_PER_BW_80];
 175         }
 176 }
 177
 178 static u16 rs_fw_he_ieee80211_mcs_to_rs_mcs(u16 mcs)
 179 {
 180         switch (mcs) {
 181         case IEEE80211_HE_MCS_SUPPORT_0_7:
 182                 return BIT(IWL_TLC_MNG_HT_RATE_MCS7 + 1) - 1;
 183         case IEEE80211_HE_MCS_SUPPORT_0_9:
 184                 return BIT(IWL_TLC_MNG_HT_RATE_MCS9 + 1) - 1;
 185         case IEEE80211_HE_MCS_SUPPORT_0_11:
 186                 return BIT(IWL_TLC_MNG_HT_RATE_MCS11 + 1) - 1;
 187         case IEEE80211_HE_MCS_NOT_SUPPORTED:
 188                 return 0;
 189         }
 190
 191         WARN(1, "invalid HE MCS %d\n", mcs);
 192         return 0;
 193 }
 194
 195 static void
 196 rs_fw_he_set_enabled_rates(const struct ieee80211_link_sta *link_sta,
 197                            const struct ieee80211_sta_he_cap *sband_he_cap,
 198                            struct iwl_tlc_config_cmd_v4 *cmd)
 199 {
 200         const struct ieee80211_sta_he_cap *he_cap = &link_sta->he_cap;
 201         u16 mcs_160 = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_160);
 202         u16 mcs_80 = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_80);
 203         u16 tx_mcs_80 = le16_to_cpu(sband_he_cap->he_mcs_nss_supp.tx_mcs_80);
 204         u16 tx_mcs_160 = le16_to_cpu(sband_he_cap->he_mcs_nss_supp.tx_mcs_160);
 205         int i;
 206         u8 nss = link_sta->rx_nss;
 207
 208         /* the station support only a single receive chain */
 209         if (link_sta->smps_mode == IEEE80211_SMPS_STATIC)
 210                 nss = 1;
 211
 212         for (i = 0; i < nss && i < IWL_TLC_NSS_MAX; i++) {
 213                 u16 _mcs_160 = (mcs_160 >> (2 * i)) & 0x3;
 214                 u16 _mcs_80 = (mcs_80 >> (2 * i)) & 0x3;
 215                 u16 _tx_mcs_160 = (tx_mcs_160 >> (2 * i)) & 0x3;
 216                 u16 _tx_mcs_80 = (tx_mcs_80 >> (2 * i)) & 0x3;
 217
 218                 /* If one side doesn't support - mark both as not supporting */
 219                 if (_mcs_80 == IEEE80211_HE_MCS_NOT_SUPPORTED ||
 220                     _tx_mcs_80 == IEEE80211_HE_MCS_NOT_SUPPORTED) {
 221                         _mcs_80 = IEEE80211_HE_MCS_NOT_SUPPORTED;
 222                         _tx_mcs_80 = IEEE80211_HE_MCS_NOT_SUPPORTED;
 223                 }
 224                 if (_mcs_80 > _tx_mcs_80)
 225                         _mcs_80 = _tx_mcs_80;
 226                 cmd->ht_rates[i][IWL_TLC_MCS_PER_BW_80] =
 227                         cpu_to_le16(rs_fw_he_ieee80211_mcs_to_rs_mcs(_mcs_80));
 228
 229                 /* If one side doesn't support - mark both as not supporting */
 230                 if (_mcs_160 == IEEE80211_HE_MCS_NOT_SUPPORTED ||
 231                     _tx_mcs_160 == IEEE80211_HE_MCS_NOT_SUPPORTED) {
 232                         _mcs_160 = IEEE80211_HE_MCS_NOT_SUPPORTED;
 233                         _tx_mcs_160 = IEEE80211_HE_MCS_NOT_SUPPORTED;
 234                 }
 235                 if (_mcs_160 > _tx_mcs_160)
 236                         _mcs_160 = _tx_mcs_160;
 237                 cmd->ht_rates[i][IWL_TLC_MCS_PER_BW_160] =
 238                         cpu_to_le16(rs_fw_he_ieee80211_mcs_to_rs_mcs(_mcs_160));
 239         }
 240 }
 241
 242 static u8 rs_fw_eht_max_nss(u8 rx_nss, u8 tx_nss)
 243 {
 244         u8 tx = u8_get_bits(tx_nss, IEEE80211_EHT_MCS_NSS_TX);
 245         u8 rx = u8_get_bits(rx_nss, IEEE80211_EHT_MCS_NSS_RX);
 246         /* the max nss that can be used,
 247          * is the min with our tx capa and the peer rx capa.
 248          */
 249         return min(tx, rx);
 250 }
 251
 252 #define MAX_NSS_MCS(mcs_num, rx, tx) \
 253         rs_fw_eht_max_nss((rx)->rx_tx_mcs ##mcs_num## _max_nss, \
 254                           (tx)->rx_tx_mcs ##mcs_num## _max_nss)
 255
 256 static void rs_fw_set_eht_mcs_nss(__le16 ht_rates[][3],
 257                                   enum IWL_TLC_MCS_PER_BW bw,
 258                                   u8 max_nss, u16 mcs_msk)
 259 {
 260         if (max_nss >= 2)
 261                 ht_rates[IWL_TLC_NSS_2][bw] |= cpu_to_le16(mcs_msk);
 262
 263         if (max_nss >= 1)
 264                 ht_rates[IWL_TLC_NSS_1][bw] |= cpu_to_le16(mcs_msk);
 265 }
 266
 267 static const
 268 struct ieee80211_eht_mcs_nss_supp_bw *
 269 rs_fw_rs_mcs2eht_mcs(enum IWL_TLC_MCS_PER_BW bw,
 270                      const struct ieee80211_eht_mcs_nss_supp *eht_mcs)
 271 {
 272         switch (bw) {
 273         case IWL_TLC_MCS_PER_BW_80:
 274                 return &eht_mcs->bw._80;
 275         case IWL_TLC_MCS_PER_BW_160:
 276                 return &eht_mcs->bw._160;
 277         case IWL_TLC_MCS_PER_BW_320:
 278                 return &eht_mcs->bw._320;
 279         default:
 280                 return NULL;
 281         }
 282 }
 283
 284 static void
 285 rs_fw_eht_set_enabled_rates(struct ieee80211_vif *vif,
 286                             const struct ieee80211_link_sta *link_sta,
 287                             const struct ieee80211_sta_he_cap *sband_he_cap,
 288                             const struct ieee80211_sta_eht_cap *sband_eht_cap,
 289                             struct iwl_tlc_config_cmd_v4 *cmd)
 290 {
 291         /* peer RX mcs capa */
 292         const struct ieee80211_eht_mcs_nss_supp *eht_rx_mcs =
 293                 &link_sta->eht_cap.eht_mcs_nss_supp;
 294         /* our TX mcs capa */
 295         const struct ieee80211_eht_mcs_nss_supp *eht_tx_mcs =
 296                 &sband_eht_cap->eht_mcs_nss_supp;
 297
 298         enum IWL_TLC_MCS_PER_BW bw;
 299         struct ieee80211_eht_mcs_nss_supp_20mhz_only mcs_rx_20;
 300         struct ieee80211_eht_mcs_nss_supp_20mhz_only mcs_tx_20;
 301
 302         /* peer is 20Mhz only */
 303         if (vif->type == NL80211_IFTYPE_AP &&
 304             !(link_sta->he_cap.he_cap_elem.phy_cap_info[0] &
 305               IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_MASK_ALL)) {
 306                 mcs_rx_20 = eht_rx_mcs->only_20mhz;
 307         } else {
 308                 mcs_rx_20.rx_tx_mcs7_max_nss = eht_rx_mcs->bw._80.rx_tx_mcs9_max_nss;
 309                 mcs_rx_20.rx_tx_mcs9_max_nss = eht_rx_mcs->bw._80.rx_tx_mcs9_max_nss;
 310                 mcs_rx_20.rx_tx_mcs11_max_nss = eht_rx_mcs->bw._80.rx_tx_mcs11_max_nss;
 311                 mcs_rx_20.rx_tx_mcs13_max_nss = eht_rx_mcs->bw._80.rx_tx_mcs13_max_nss;
 312         }
 313
 314         /* nic is 20Mhz only */
 315         if (!(sband_he_cap->he_cap_elem.phy_cap_info[0] &
 316               IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_MASK_ALL)) {
 317                 mcs_tx_20 = eht_tx_mcs->only_20mhz;
 318         } else {
 319                 mcs_tx_20.rx_tx_mcs7_max_nss = eht_tx_mcs->bw._80.rx_tx_mcs9_max_nss;
 320                 mcs_tx_20.rx_tx_mcs9_max_nss = eht_tx_mcs->bw._80.rx_tx_mcs9_max_nss;
 321                 mcs_tx_20.rx_tx_mcs11_max_nss = eht_tx_mcs->bw._80.rx_tx_mcs11_max_nss;
 322                 mcs_tx_20.rx_tx_mcs13_max_nss = eht_tx_mcs->bw._80.rx_tx_mcs13_max_nss;
 323         }
 324
 325         /* rates for 20/40/80 bw */
 326         bw = IWL_TLC_MCS_PER_BW_80;
 327         rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 328                               MAX_NSS_MCS(7, &mcs_rx_20, &mcs_tx_20), GENMASK(7, 0));
 329         rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 330                               MAX_NSS_MCS(9, &mcs_rx_20, &mcs_tx_20), GENMASK(9, 8));
 331         rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 332                               MAX_NSS_MCS(11, &mcs_rx_20, &mcs_tx_20), GENMASK(11, 10));
 333         rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 334                               MAX_NSS_MCS(13, &mcs_rx_20, &mcs_tx_20), GENMASK(13, 12));
 335
 336         /* rate for 160/320 bw */
 337         for (bw = IWL_TLC_MCS_PER_BW_160; bw <= IWL_TLC_MCS_PER_BW_320; bw++) {
 338                 const struct ieee80211_eht_mcs_nss_supp_bw *mcs_rx =
 339                         rs_fw_rs_mcs2eht_mcs(bw, eht_rx_mcs);
 340                 const struct ieee80211_eht_mcs_nss_supp_bw *mcs_tx =
 341                         rs_fw_rs_mcs2eht_mcs(bw, eht_tx_mcs);
 342
 343                 /* got unsupported index for bw */
 344                 if (!mcs_rx || !mcs_tx)
 345                         continue;
 346
 347                 /* break out if we don't support the bandwidth */
 348                 if (cmd->max_ch_width < (bw + IWL_TLC_MNG_CH_WIDTH_80MHZ))
 349                         break;
 350
 351                 rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 352                                       MAX_NSS_MCS(9, mcs_rx, mcs_tx), GENMASK(9, 0));
 353                 rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 354                                       MAX_NSS_MCS(11, mcs_rx, mcs_tx), GENMASK(11, 10));
 355                 rs_fw_set_eht_mcs_nss(cmd->ht_rates, bw,
 356                                       MAX_NSS_MCS(13, mcs_rx, mcs_tx), GENMASK(13, 12));
 357         }
 358
 359         /* the station support only a single receive chain */
 360         if (link_sta->smps_mode == IEEE80211_SMPS_STATIC ||
 361             link_sta->rx_nss < 2)
 362                 memset(cmd->ht_rates[IWL_TLC_NSS_2], 0,
 363                        sizeof(cmd->ht_rates[IWL_TLC_NSS_2]));
 364 }
 365
 366 static void rs_fw_set_supp_rates(struct ieee80211_vif *vif,
 367                                  struct ieee80211_link_sta *link_sta,
 368                                  struct ieee80211_supported_band *sband,
 369                                  const struct ieee80211_sta_he_cap *sband_he_cap,
 370                                  const struct ieee80211_sta_eht_cap *sband_eht_cap,
 371                                  struct iwl_tlc_config_cmd_v4 *cmd)
 372 {
 373         int i;
 374         u16 supp = 0;
 375         unsigned long tmp; /* must be unsigned long for for_each_set_bit */
 376         const struct ieee80211_sta_ht_cap *ht_cap = &link_sta->ht_cap;
 377         const struct ieee80211_sta_vht_cap *vht_cap = &link_sta->vht_cap;
 378         const struct ieee80211_sta_he_cap *he_cap = &link_sta->he_cap;
 379
 380         /* non HT rates */
 381         tmp = link_sta->supp_rates[sband->band];
 382         for_each_set_bit(i, &tmp, BITS_PER_LONG)
 383                 supp |= BIT(sband->bitrates[i].hw_value);
 384
 385         cmd->non_ht_rates = cpu_to_le16(supp);
 386         cmd->mode = IWL_TLC_MNG_MODE_NON_HT;
 387
 388         /* HT/VHT rates */
 389         if (link_sta->eht_cap.has_eht && sband_he_cap && sband_eht_cap) {
 390                 cmd->mode = IWL_TLC_MNG_MODE_EHT;
 391                 rs_fw_eht_set_enabled_rates(vif, link_sta, sband_he_cap,
 392                                             sband_eht_cap, cmd);
 393         } else if (he_cap->has_he && sband_he_cap) {
 394                 cmd->mode = IWL_TLC_MNG_MODE_HE;
 395                 rs_fw_he_set_enabled_rates(link_sta, sband_he_cap, cmd);
 396         } else if (vht_cap->vht_supported) {
 397                 cmd->mode = IWL_TLC_MNG_MODE_VHT;
 398                 rs_fw_vht_set_enabled_rates(link_sta, vht_cap, cmd);
 399         } else if (ht_cap->ht_supported) {
 400                 cmd->mode = IWL_TLC_MNG_MODE_HT;
 401                 cmd->ht_rates[IWL_TLC_NSS_1][IWL_TLC_MCS_PER_BW_80] =
 402                         cpu_to_le16(ht_cap->mcs.rx_mask[0]);
 403
 404                 /* the station support only a single receive chain */
 405                 if (link_sta->smps_mode == IEEE80211_SMPS_STATIC)
 406                         cmd->ht_rates[IWL_TLC_NSS_2][IWL_TLC_MCS_PER_BW_80] =
 407                                 0;
 408                 else
 409                         cmd->ht_rates[IWL_TLC_NSS_2][IWL_TLC_MCS_PER_BW_80] =
 410                                 cpu_to_le16(ht_cap->mcs.rx_mask[1]);
 411         }
 412 }
 413
 414 void iwl_mvm_tlc_update_notif(struct iwl_mvm *mvm,
 415                               struct iwl_rx_cmd_buffer *rxb)
 416 {
 417         struct iwl_rx_packet *pkt = rxb_addr(rxb);
 418         struct iwl_tlc_update_notif *notif;
 419         struct ieee80211_sta *sta;
 420         struct ieee80211_link_sta *link_sta;
 421         struct iwl_mvm_sta *mvmsta;
 422         struct iwl_mvm_link_sta *mvm_link_sta;
 423         struct iwl_lq_sta_rs_fw *lq_sta;
 424         u32 flags;
 425
 426         rcu_read_lock();
 427
 428         notif = (void *)pkt->data;
 429         link_sta = rcu_dereference(mvm->fw_id_to_link_sta[notif->sta_id]);
 430         sta = rcu_dereference(mvm->fw_id_to_mac_id[notif->sta_id]);
 431         if (IS_ERR_OR_NULL(sta) || !link_sta) {
 432                 /* can happen in remove station flow where mvm removed internally
 433                  * the station before removing from FW
 434                  */
 435                 IWL_DEBUG_RATE(mvm,
 436                                "Invalid mvm RCU pointer for sta id (%d) in TLC notification\n",
 437                                notif->sta_id);
 438                 goto out;
 439         }
 440
 441         mvmsta = iwl_mvm_sta_from_mac80211(sta);
 442
 443         if (!mvmsta) {
 444                 IWL_ERR(mvm, "Invalid sta id (%d) in FW TLC notification\n",
 445                         notif->sta_id);
 446                 goto out;
 447         }
 448
 449         flags = le32_to_cpu(notif->flags);
 450
 451         mvm_link_sta = rcu_dereference(mvmsta->link[link_sta->link_id]);
 452         if (!mvm_link_sta) {
 453                 IWL_DEBUG_RATE(mvm,
 454                                "Invalid mvmsta RCU pointer for link (%d) of  sta id (%d) in TLC notification\n",
 455                                link_sta->link_id, notif->sta_id);
 456                 goto out;
 457         }
 458         lq_sta = &mvm_link_sta->lq_sta.rs_fw;
 459
 460         if (flags & IWL_TLC_NOTIF_FLAG_RATE) {
 461                 char pretty_rate[100];
 462
 463                 if (iwl_fw_lookup_notif_ver(mvm->fw, DATA_PATH_GROUP,
 464                                             TLC_MNG_UPDATE_NOTIF, 0) < 3) {
 465                         rs_pretty_print_rate_v1(pretty_rate,
 466                                                 sizeof(pretty_rate),
 467                                                 le32_to_cpu(notif->rate));
 468                         IWL_DEBUG_RATE(mvm,
 469                                        "Got rate in old format. Rate: %s. Converting.\n",
 470                                        pretty_rate);
 471                         lq_sta->last_rate_n_flags =
 472                                 iwl_new_rate_from_v1(le32_to_cpu(notif->rate));
 473                 } else {
 474                         lq_sta->last_rate_n_flags = le32_to_cpu(notif->rate);
 475                 }
 476                 rs_pretty_print_rate(pretty_rate, sizeof(pretty_rate),
 477                                      lq_sta->last_rate_n_flags);
 478                 IWL_DEBUG_RATE(mvm, "new rate: %s\n", pretty_rate);
 479         }
 480
 481         if (flags & IWL_TLC_NOTIF_FLAG_AMSDU && !mvm_link_sta->orig_amsdu_len) {
 482                 u32 enabled = le32_to_cpu(notif->amsdu_enabled);
 483                 u16 size = le32_to_cpu(notif->amsdu_size);
 484                 int i;
 485
 486                 if (size < 2000) {
 487                         size = 0;
 488                         enabled = 0;
 489                 }
 490
 491                 if (link_sta->agg.max_amsdu_len < size) {
 492                         /*
 493                          * In debug link_sta->agg.max_amsdu_len < size
 494                          * so also check with orig_amsdu_len which holds the
 495                          * original data before debugfs changed the value
 496                          */
 497                         WARN_ON(mvm_link_sta->orig_amsdu_len < size);
 498                         goto out;
 499                 }
 500
 501                 mvmsta->amsdu_enabled = enabled;
 502                 mvmsta->max_amsdu_len = size;
 503                 link_sta->agg.max_rc_amsdu_len = mvmsta->max_amsdu_len;
 504
 505                 for (i = 0; i < IWL_MAX_TID_COUNT; i++) {
 506                         if (mvmsta->amsdu_enabled & BIT(i))
 507                                 link_sta->agg.max_tid_amsdu_len[i] =
 508                                         iwl_mvm_max_amsdu_size(mvm, sta, i);
 509                         else
 510                                 /*
 511                                  * Not so elegant, but this will effectively
 512                                  * prevent AMSDU on this TID
 513                                  */
 514                                 link_sta->agg.max_tid_amsdu_len[i] = 1;
 515                 }
 516
 517                 IWL_DEBUG_RATE(mvm,
 518                                "AMSDU update. AMSDU size: %d, AMSDU selected size: %d, AMSDU TID bitmap 0x%X\n",
 519                                le32_to_cpu(notif->amsdu_size), size,
 520                                mvmsta->amsdu_enabled);
 521         }
 522 out:
 523         rcu_read_unlock();
 524 }
 525
 526 u16 rs_fw_get_max_amsdu_len(struct ieee80211_sta *sta,
 527                             struct ieee80211_bss_conf *link_conf,
 528                             struct ieee80211_link_sta *link_sta)
 529 {
 530         const struct ieee80211_sta_vht_cap *vht_cap = &link_sta->vht_cap;
 531         const struct ieee80211_sta_ht_cap *ht_cap = &link_sta->ht_cap;
 532         const struct ieee80211_sta_eht_cap *eht_cap = &link_sta->eht_cap;
 533
 534         if (WARN_ON_ONCE(!link_conf->chanreq.oper.chan))
 535                 return IEEE80211_MAX_MPDU_LEN_VHT_3895;
 536
 537         if (link_conf->chanreq.oper.chan->band == NL80211_BAND_6GHZ) {
 538                 switch (le16_get_bits(link_sta->he_6ghz_capa.capa,
 539                                       IEEE80211_HE_6GHZ_CAP_MAX_MPDU_LEN)) {
 540                 case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
 541                         return IEEE80211_MAX_MPDU_LEN_VHT_11454;
 542                 case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
 543                         return IEEE80211_MAX_MPDU_LEN_VHT_7991;
 544                 default:
 545                         return IEEE80211_MAX_MPDU_LEN_VHT_3895;
 546                 }
 547         } else if (link_conf->chanreq.oper.chan->band == NL80211_BAND_2GHZ &&
 548                    eht_cap->has_eht) {
 549                 switch (u8_get_bits(eht_cap->eht_cap_elem.mac_cap_info[0],
 550                                     IEEE80211_EHT_MAC_CAP0_MAX_MPDU_LEN_MASK)) {
 551                 case IEEE80211_EHT_MAC_CAP0_MAX_MPDU_LEN_11454:
 552                         return IEEE80211_MAX_MPDU_LEN_VHT_11454;
 553                 case IEEE80211_EHT_MAC_CAP0_MAX_MPDU_LEN_7991:
 554                         return IEEE80211_MAX_MPDU_LEN_VHT_7991;
 555                 default:
 556                         return IEEE80211_MAX_MPDU_LEN_VHT_3895;
 557                 }
 558         } else if (vht_cap->vht_supported) {
 559                 switch (vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK) {
 560                 case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
 561                         return IEEE80211_MAX_MPDU_LEN_VHT_11454;
 562                 case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
 563                         return IEEE80211_MAX_MPDU_LEN_VHT_7991;
 564                 default:
 565                         return IEEE80211_MAX_MPDU_LEN_VHT_3895;
 566                 }
 567         } else if (ht_cap->ht_supported) {
 568                 if (ht_cap->cap & IEEE80211_HT_CAP_MAX_AMSDU)
 569                         /*
 570                          * agg is offloaded so we need to assume that agg
 571                          * are enabled and max mpdu in ampdu is 4095
 572                          * (spec 802.11-2016 9.3.2.1)
 573                          */
 574                         return IEEE80211_MAX_MPDU_LEN_HT_BA;
 575                 else
 576                         return IEEE80211_MAX_MPDU_LEN_HT_3839;
 577         }
 578
 579         /* in legacy mode no amsdu is enabled so return zero */
 580         return 0;
 581 }
 582
 583 void iwl_mvm_rs_fw_rate_init(struct iwl_mvm *mvm,
 584                              struct ieee80211_vif *vif,
 585                              struct ieee80211_sta *sta,
 586                              struct ieee80211_bss_conf *link_conf,
 587                              struct ieee80211_link_sta *link_sta,
 588                              enum nl80211_band band)
 589 {
 590         struct ieee80211_hw *hw = mvm->hw;
 591         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
 592         u32 cmd_id = WIDE_ID(DATA_PATH_GROUP, TLC_MNG_CONFIG_CMD);
 593         struct ieee80211_supported_band *sband = hw->wiphy->bands[band];
 594         u16 max_amsdu_len = rs_fw_get_max_amsdu_len(sta, link_conf, link_sta);
 595         const struct ieee80211_sta_he_cap *sband_he_cap =
 596                 ieee80211_get_he_iftype_cap_vif(sband, vif);
 597         const struct ieee80211_sta_eht_cap *sband_eht_cap =
 598                 ieee80211_get_eht_iftype_cap_vif(sband, vif);
 599         struct iwl_mvm_link_sta *mvm_link_sta;
 600         struct iwl_lq_sta_rs_fw *lq_sta;
 601         struct iwl_tlc_config_cmd_v4 cfg_cmd = {
 602                 .max_ch_width = mvmsta->authorized ?
 603                         rs_fw_bw_from_sta_bw(link_sta) : IWL_TLC_MNG_CH_WIDTH_20MHZ,
 604                 .flags = cpu_to_le16(rs_fw_get_config_flags(mvm, vif, link_sta,
 605                                                             sband_he_cap)),
 606                 .chains = rs_fw_set_active_chains(iwl_mvm_get_valid_tx_ant(mvm)),
 607                 .sgi_ch_width_supp = rs_fw_sgi_cw_support(link_sta),
 608                 .max_mpdu_len = iwl_mvm_is_csum_supported(mvm) ?
 609                                 cpu_to_le16(max_amsdu_len) : 0,
 610         };
 611         unsigned int link_id = link_conf->link_id;
 612         int cmd_ver;
 613         int ret;
 614
 615         /* Enable external EHT LTF only for GL device and if there's
 616          * mutual support by AP and client
 617          */
 618         if (CSR_HW_REV_TYPE(mvm->trans->hw_rev) == IWL_CFG_MAC_TYPE_GL &&
 619             sband_eht_cap &&
 620             sband_eht_cap->eht_cap_elem.phy_cap_info[5] &
 621                 IEEE80211_EHT_PHY_CAP5_SUPP_EXTRA_EHT_LTF &&
 622             link_sta->eht_cap.has_eht &&
 623             link_sta->eht_cap.eht_cap_elem.phy_cap_info[5] &
 624             IEEE80211_EHT_PHY_CAP5_SUPP_EXTRA_EHT_LTF) {
 625                 IWL_DEBUG_RATE(mvm, "Set support for Extra EHT LTF\n");
 626                 cfg_cmd.flags |=
 627                         cpu_to_le16(IWL_TLC_MNG_CFG_FLAGS_EHT_EXTRA_LTF_MSK);
 628         }
 629
 630         rcu_read_lock();
 631         mvm_link_sta = rcu_dereference(mvmsta->link[link_id]);
 632         if (WARN_ON_ONCE(!mvm_link_sta)) {
 633                 rcu_read_unlock();
 634                 return;
 635         }
 636
 637         cfg_cmd.sta_id = mvm_link_sta->sta_id;
 638
 639         lq_sta = &mvm_link_sta->lq_sta.rs_fw;
 640         memset(lq_sta, 0, offsetof(typeof(*lq_sta), pers));
 641
 642         rcu_read_unlock();
 643
 644 #ifdef CONFIG_IWLWIFI_DEBUGFS
 645         iwl_mvm_reset_frame_stats(mvm);
 646 #endif
 647         rs_fw_set_supp_rates(vif, link_sta, sband,
 648                              sband_he_cap, sband_eht_cap,
 649                              &cfg_cmd);
 650
 651         /*
 652          * since TLC offload works with one mode we can assume
 653          * that only vht/ht is used and also set it as station max amsdu
 654          */
 655         sta->deflink.agg.max_amsdu_len = max_amsdu_len;
 656
 657         cmd_ver = iwl_fw_lookup_cmd_ver(mvm->fw,
 658                                         WIDE_ID(DATA_PATH_GROUP,
 659                                                 TLC_MNG_CONFIG_CMD),
 660                                         0);
 661         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, sta_id=%d, max_ch_width=%d, mode=%d\n",
 662                        cfg_cmd.sta_id, cfg_cmd.max_ch_width, cfg_cmd.mode);
 663         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, chains=0x%X, ch_wid_supp=%d, flags=0x%X\n",
 664                        cfg_cmd.chains, cfg_cmd.sgi_ch_width_supp, cfg_cmd.flags);
 665         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, mpdu_len=%d, no_ht_rate=0x%X, tx_op=%d\n",
 666                        cfg_cmd.max_mpdu_len, cfg_cmd.non_ht_rates, cfg_cmd.max_tx_op);
 667         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, ht_rate[0][0]=0x%X, ht_rate[1][0]=0x%X\n",
 668                        cfg_cmd.ht_rates[0][0], cfg_cmd.ht_rates[1][0]);
 669         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, ht_rate[0][1]=0x%X, ht_rate[1][1]=0x%X\n",
 670                        cfg_cmd.ht_rates[0][1], cfg_cmd.ht_rates[1][1]);
 671         IWL_DEBUG_RATE(mvm, "TLC CONFIG CMD, ht_rate[0][2]=0x%X, ht_rate[1][2]=0x%X\n",
 672                        cfg_cmd.ht_rates[0][2], cfg_cmd.ht_rates[1][2]);
 673         if (cmd_ver == 4) {
 674                 ret = iwl_mvm_send_cmd_pdu(mvm, cmd_id, CMD_ASYNC,
 675                                            sizeof(cfg_cmd), &cfg_cmd);
 676         } else if (cmd_ver < 4) {
 677                 struct iwl_tlc_config_cmd_v3 cfg_cmd_v3 = {
 678                         .sta_id = cfg_cmd.sta_id,
 679                         .max_ch_width = cfg_cmd.max_ch_width,
 680                         .mode = cfg_cmd.mode,
 681                         .chains = cfg_cmd.chains,
 682                         .amsdu = !!cfg_cmd.max_mpdu_len,
 683                         .flags = cfg_cmd.flags,
 684                         .non_ht_rates = cfg_cmd.non_ht_rates,
 685                         .ht_rates[0][0] = cfg_cmd.ht_rates[0][0],
 686                         .ht_rates[0][1] = cfg_cmd.ht_rates[0][1],
 687                         .ht_rates[1][0] = cfg_cmd.ht_rates[1][0],
 688                         .ht_rates[1][1] = cfg_cmd.ht_rates[1][1],
 689                         .sgi_ch_width_supp = cfg_cmd.sgi_ch_width_supp,
 690                         .max_mpdu_len = cfg_cmd.max_mpdu_len,
 691                 };
 692
 693                 u16 cmd_size = sizeof(cfg_cmd_v3);
 694
 695                 /* In old versions of the API the struct is 4 bytes smaller */
 696                 if (iwl_fw_lookup_cmd_ver(mvm->fw,
 697                                           WIDE_ID(DATA_PATH_GROUP,
 698                                                   TLC_MNG_CONFIG_CMD), 0) < 3)
 699                         cmd_size -= 4;
 700
 701                 ret = iwl_mvm_send_cmd_pdu(mvm, cmd_id, CMD_ASYNC, cmd_size,
 702                                            &cfg_cmd_v3);
 703         } else {
 704                 ret = -EINVAL;
 705         }
 706
 707         if (ret)
 708                 IWL_ERR(mvm, "Failed to send rate scale config (%d)\n", ret);
 709 }
 710
 711 int rs_fw_tx_protection(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvmsta,
 712                         bool enable)
 713 {
 714         /* TODO: need to introduce a new FW cmd since LQ cmd is not relevant */
 715         IWL_DEBUG_RATE(mvm, "tx protection - not implemented yet.\n");
 716         return 0;
 717 }
 718
 719 void iwl_mvm_rs_add_sta_link(struct iwl_mvm *mvm,
 720                              struct iwl_mvm_link_sta *link_sta)
 721 {
 722         struct iwl_lq_sta_rs_fw *lq_sta;
 723
 724         lq_sta = &link_sta->lq_sta.rs_fw;
 725
 726         lq_sta->pers.drv = mvm;
 727         lq_sta->pers.sta_id = link_sta->sta_id;
 728         lq_sta->pers.chains = 0;
 729         memset(lq_sta->pers.chain_signal, 0,
 730                sizeof(lq_sta->pers.chain_signal));
 731         lq_sta->pers.last_rssi = S8_MIN;
 732         lq_sta->last_rate_n_flags = 0;
 733
 734 #ifdef CONFIG_MAC80211_DEBUGFS
 735         lq_sta->pers.dbg_fixed_rate = 0;
 736 #endif
 737 }
 738
 739 void iwl_mvm_rs_add_sta(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvmsta)
 740 {
 741         unsigned int link_id;
 742
 743         IWL_DEBUG_RATE(mvm, "create station rate scale window\n");
 744
 745         for (link_id = 0; link_id < ARRAY_SIZE(mvmsta->link); link_id++) {
 746                 struct iwl_mvm_link_sta *link =
 747                         rcu_dereference_protected(mvmsta->link[link_id],
 748                                                   lockdep_is_held(&mvm->mutex));
 749                 if (!link)
 750                         continue;
 751
 752                 iwl_mvm_rs_add_sta_link(mvm, link);
 753         }
 754 }