net/mac80211/vht.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * VHT handling
   4  *
   5  * Portions of this file
   6  * Copyright(c) 2015 - 2016 Intel Deutschland GmbH
   7  * Copyright (C) 2018 - 2024 Intel Corporation
   8  */
   9
  10 #include <linux/ieee80211.h>
  11 #include <linux/export.h>
  12 #include <net/mac80211.h>
  13 #include "ieee80211_i.h"
  14 #include "rate.h"
  15
  16
  17 static void __check_vhtcap_disable(struct ieee80211_sub_if_data *sdata,
  18                                    struct ieee80211_sta_vht_cap *vht_cap,
  19                                    u32 flag)
  20 {
  21         __le32 le_flag = cpu_to_le32(flag);
  22
  23         if (sdata->u.mgd.vht_capa_mask.vht_cap_info & le_flag &&
  24             !(sdata->u.mgd.vht_capa.vht_cap_info & le_flag))
  25                 vht_cap->cap &= ~flag;
  26 }
  27
  28 void ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data *sdata,
  29                                       struct ieee80211_sta_vht_cap *vht_cap)
  30 {
  31         int i;
  32         u16 rxmcs_mask, rxmcs_cap, rxmcs_n, txmcs_mask, txmcs_cap, txmcs_n;
  33
  34         if (!vht_cap->vht_supported)
  35                 return;
  36
  37         if (sdata->vif.type != NL80211_IFTYPE_STATION)
  38                 return;
  39
  40         __check_vhtcap_disable(sdata, vht_cap,
  41                                IEEE80211_VHT_CAP_RXLDPC);
  42         __check_vhtcap_disable(sdata, vht_cap,
  43                                IEEE80211_VHT_CAP_SHORT_GI_80);
  44         __check_vhtcap_disable(sdata, vht_cap,
  45                                IEEE80211_VHT_CAP_SHORT_GI_160);
  46         __check_vhtcap_disable(sdata, vht_cap,
  47                                IEEE80211_VHT_CAP_TXSTBC);
  48         __check_vhtcap_disable(sdata, vht_cap,
  49                                IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
  50         __check_vhtcap_disable(sdata, vht_cap,
  51                                IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
  52         __check_vhtcap_disable(sdata, vht_cap,
  53                                IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN);
  54         __check_vhtcap_disable(sdata, vht_cap,
  55                                IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN);
  56
  57         /* Allow user to decrease AMPDU length exponent */
  58         if (sdata->u.mgd.vht_capa_mask.vht_cap_info &
  59             cpu_to_le32(IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK)) {
  60                 u32 cap, n;
  61
  62                 n = le32_to_cpu(sdata->u.mgd.vht_capa.vht_cap_info) &
  63                         IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
  64                 n >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
  65                 cap = vht_cap->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
  66                 cap >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
  67
  68                 if (n < cap) {
  69                         vht_cap->cap &=
  70                                 ~IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
  71                         vht_cap->cap |=
  72                                 n << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
  73                 }
  74         }
  75
  76         /* Allow the user to decrease MCSes */
  77         rxmcs_mask =
  78                 le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.rx_mcs_map);
  79         rxmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.rx_mcs_map);
  80         rxmcs_n &= rxmcs_mask;
  81         rxmcs_cap = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
  82
  83         txmcs_mask =
  84                 le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.tx_mcs_map);
  85         txmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.tx_mcs_map);
  86         txmcs_n &= txmcs_mask;
  87         txmcs_cap = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
  88         for (i = 0; i < 8; i++) {
  89                 u8 m, n, c;
  90
  91                 m = (rxmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
  92                 n = (rxmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
  93                 c = (rxmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
  94
  95                 if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
  96                           n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
  97                         rxmcs_cap &= ~(3 << 2*i);
  98                         rxmcs_cap |= (rxmcs_n & (3 << 2*i));
  99                 }
 100
 101                 m = (txmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 102                 n = (txmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 103                 c = (txmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 104
 105                 if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
 106                           n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
 107                         txmcs_cap &= ~(3 << 2*i);
 108                         txmcs_cap |= (txmcs_n & (3 << 2*i));
 109                 }
 110         }
 111         vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(rxmcs_cap);
 112         vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(txmcs_cap);
 113 }
 114
 115 void
 116 ieee80211_vht_cap_ie_to_sta_vht_cap(struct ieee80211_sub_if_data *sdata,
 117                                     struct ieee80211_supported_band *sband,
 118                                     const struct ieee80211_vht_cap *vht_cap_ie,
 119                                     const struct ieee80211_vht_cap *vht_cap_ie2,
 120                                     struct link_sta_info *link_sta)
 121 {
 122         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->pub->vht_cap;
 123         struct ieee80211_sta_vht_cap own_cap;
 124         u32 cap_info, i;
 125         bool have_80mhz;
 126         u32 mpdu_len;
 127
 128         memset(vht_cap, 0, sizeof(*vht_cap));
 129
 130         if (!link_sta->pub->ht_cap.ht_supported)
 131                 return;
 132
 133         if (!vht_cap_ie || !sband->vht_cap.vht_supported)
 134                 return;
 135
 136         /* Allow VHT if at least one channel on the sband supports 80 MHz */
 137         have_80mhz = false;
 138         for (i = 0; i < sband->n_channels; i++) {
 139                 if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
 140                                                 IEEE80211_CHAN_NO_80MHZ))
 141                         continue;
 142
 143                 have_80mhz = true;
 144                 break;
 145         }
 146
 147         if (!have_80mhz)
 148                 return;
 149
 150         /*
 151          * A VHT STA must support 40 MHz, but if we verify that here
 152          * then we break a few things - some APs (e.g. Netgear R6300v2
 153          * and others based on the BCM4360 chipset) will unset this
 154          * capability bit when operating in 20 MHz.
 155          */
 156
 157         vht_cap->vht_supported = true;
 158
 159         own_cap = sband->vht_cap;
 160         /*
 161          * If user has specified capability overrides, take care
 162          * of that if the station we're setting up is the AP that
 163          * we advertised a restricted capability set to. Override
 164          * our own capabilities and then use those below.
 165          */
 166         if (sdata->vif.type == NL80211_IFTYPE_STATION &&
 167             !test_sta_flag(link_sta->sta, WLAN_STA_TDLS_PEER))
 168                 ieee80211_apply_vhtcap_overrides(sdata, &own_cap);
 169
 170         /* take some capabilities as-is */
 171         cap_info = le32_to_cpu(vht_cap_ie->vht_cap_info);
 172         vht_cap->cap = cap_info;
 173         vht_cap->cap &= IEEE80211_VHT_CAP_RXLDPC |
 174                         IEEE80211_VHT_CAP_VHT_TXOP_PS |
 175                         IEEE80211_VHT_CAP_HTC_VHT |
 176                         IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
 177                         IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB |
 178                         IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB |
 179                         IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
 180                         IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;
 181
 182         vht_cap->cap |= min_t(u32, cap_info & IEEE80211_VHT_CAP_MAX_MPDU_MASK,
 183                               own_cap.cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK);
 184
 185         /* and some based on our own capabilities */
 186         switch (own_cap.cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
 187         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
 188                 vht_cap->cap |= cap_info &
 189                                 IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
 190                 break;
 191         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
 192                 vht_cap->cap |= cap_info &
 193                                 IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 194                 break;
 195         default:
 196                 /* nothing */
 197                 break;
 198         }
 199
 200         /* symmetric capabilities */
 201         vht_cap->cap |= cap_info & own_cap.cap &
 202                         (IEEE80211_VHT_CAP_SHORT_GI_80 |
 203                          IEEE80211_VHT_CAP_SHORT_GI_160);
 204
 205         /* remaining ones */
 206         if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
 207                 vht_cap->cap |= cap_info &
 208                                 (IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
 209                                  IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
 210
 211         if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
 212                 vht_cap->cap |= cap_info &
 213                                 (IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
 214                                  IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
 215
 216         if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)
 217                 vht_cap->cap |= cap_info &
 218                                 IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
 219
 220         if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE)
 221                 vht_cap->cap |= cap_info &
 222                                 IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE;
 223
 224         if (own_cap.cap & IEEE80211_VHT_CAP_TXSTBC)
 225                 vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_RXSTBC_MASK;
 226
 227         if (own_cap.cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
 228                 vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_TXSTBC;
 229
 230         /* Copy peer MCS info, the driver might need them. */
 231         memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
 232                sizeof(struct ieee80211_vht_mcs_info));
 233
 234         /* copy EXT_NSS_BW Support value or remove the capability */
 235         if (ieee80211_hw_check(&sdata->local->hw, SUPPORTS_VHT_EXT_NSS_BW))
 236                 vht_cap->cap |= (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK);
 237         else
 238                 vht_cap->vht_mcs.tx_highest &=
 239                         ~cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE);
 240
 241         /* but also restrict MCSes */
 242         for (i = 0; i < 8; i++) {
 243                 u16 own_rx, own_tx, peer_rx, peer_tx;
 244
 245                 own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
 246                 own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 247
 248                 own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
 249                 own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 250
 251                 peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
 252                 peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 253
 254                 peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
 255                 peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 256
 257                 if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 258                         if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
 259                                 peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
 260                         else if (own_rx < peer_tx)
 261                                 peer_tx = own_rx;
 262                 }
 263
 264                 if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 265                         if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
 266                                 peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
 267                         else if (own_tx < peer_rx)
 268                                 peer_rx = own_tx;
 269                 }
 270
 271                 vht_cap->vht_mcs.rx_mcs_map &=
 272                         ~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
 273                 vht_cap->vht_mcs.rx_mcs_map |= cpu_to_le16(peer_rx << i * 2);
 274
 275                 vht_cap->vht_mcs.tx_mcs_map &=
 276                         ~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
 277                 vht_cap->vht_mcs.tx_mcs_map |= cpu_to_le16(peer_tx << i * 2);
 278         }
 279
 280         /*
 281          * This is a workaround for VHT-enabled STAs which break the spec
 282          * and have the VHT-MCS Rx map filled in with value 3 for all eight
 283          * spacial streams, an example is AR9462.
 284          *
 285          * As per spec, in section 22.1.1 Introduction to the VHT PHY
 286          * A VHT STA shall support at least single spactial stream VHT-MCSs
 287          * 0 to 7 (transmit and receive) in all supported channel widths.
 288          */
 289         if (vht_cap->vht_mcs.rx_mcs_map == cpu_to_le16(0xFFFF)) {
 290                 vht_cap->vht_supported = false;
 291                 sdata_info(sdata,
 292                            "Ignoring VHT IE from %pM (link:%pM) due to invalid rx_mcs_map\n",
 293                            link_sta->sta->addr, link_sta->addr);
 294                 return;
 295         }
 296
 297         /* finally set up the bandwidth */
 298         switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
 299         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
 300         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
 301                 link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 302                 break;
 303         default:
 304                 link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
 305
 306                 if (!(vht_cap->vht_mcs.tx_highest &
 307                                 cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE)))
 308                         break;
 309
 310                 /*
 311                  * If this is non-zero, then it does support 160 MHz after all,
 312                  * in one form or the other. We don't distinguish here (or even
 313                  * above) between 160 and 80+80 yet.
 314                  */
 315                 if (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
 316                         link_sta->cur_max_bandwidth =
 317                                 IEEE80211_STA_RX_BW_160;
 318         }
 319
 320         link_sta->pub->bandwidth = ieee80211_sta_cur_vht_bw(link_sta);
 321
 322         /*
 323          * Work around the Cisco 9115 FW 17.3 bug by taking the min of
 324          * both reported MPDU lengths.
 325          */
 326         mpdu_len = vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK;
 327         if (vht_cap_ie2)
 328                 mpdu_len = min_t(u32, mpdu_len,
 329                                  le32_get_bits(vht_cap_ie2->vht_cap_info,
 330                                                IEEE80211_VHT_CAP_MAX_MPDU_MASK));
 331
 332         /*
 333          * FIXME - should the amsdu len be per link? store per link
 334          * and maintain a minimum?
 335          */
 336         switch (mpdu_len) {
 337         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
 338                 link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
 339                 break;
 340         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
 341                 link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
 342                 break;
 343         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
 344         default:
 345                 link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
 346                 break;
 347         }
 348
 349         ieee80211_sta_recalc_aggregates(&link_sta->sta->sta);
 350 }
 351
 352 /* FIXME: move this to some better location - parses HE/EHT now */
 353 enum ieee80211_sta_rx_bandwidth
 354 _ieee80211_sta_cap_rx_bw(struct link_sta_info *link_sta,
 355                          struct cfg80211_chan_def *chandef)
 356 {
 357         unsigned int link_id = link_sta->link_id;
 358         struct ieee80211_sub_if_data *sdata = link_sta->sta->sdata;
 359         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->pub->vht_cap;
 360         struct ieee80211_sta_he_cap *he_cap = &link_sta->pub->he_cap;
 361         struct ieee80211_sta_eht_cap *eht_cap = &link_sta->pub->eht_cap;
 362         u32 cap_width;
 363
 364         if (he_cap->has_he) {
 365                 enum nl80211_band band;
 366                 u8 info;
 367
 368                 if (chandef) {
 369                         band = chandef->chan->band;
 370                 } else {
 371                         struct ieee80211_bss_conf *link_conf;
 372
 373                         rcu_read_lock();
 374                         link_conf = rcu_dereference(sdata->vif.link_conf[link_id]);
 375                         band = link_conf->chanreq.oper.chan->band;
 376                         rcu_read_unlock();
 377                 }
 378
 379                 if (eht_cap->has_eht && band == NL80211_BAND_6GHZ) {
 380                         info = eht_cap->eht_cap_elem.phy_cap_info[0];
 381
 382                         if (info & IEEE80211_EHT_PHY_CAP0_320MHZ_IN_6GHZ)
 383                                 return IEEE80211_STA_RX_BW_320;
 384                 }
 385
 386                 info = he_cap->he_cap_elem.phy_cap_info[0];
 387
 388                 if (band == NL80211_BAND_2GHZ) {
 389                         if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_IN_2G)
 390                                 return IEEE80211_STA_RX_BW_40;
 391                         return IEEE80211_STA_RX_BW_20;
 392                 }
 393
 394                 if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G ||
 395                     info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G)
 396                         return IEEE80211_STA_RX_BW_160;
 397
 398                 if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_80MHZ_IN_5G)
 399                         return IEEE80211_STA_RX_BW_80;
 400
 401                 return IEEE80211_STA_RX_BW_20;
 402         }
 403
 404         if (!vht_cap->vht_supported)
 405                 return link_sta->pub->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
 406                                 IEEE80211_STA_RX_BW_40 :
 407                                 IEEE80211_STA_RX_BW_20;
 408
 409         cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 410
 411         if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
 412             cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
 413                 return IEEE80211_STA_RX_BW_160;
 414
 415         /*
 416          * If this is non-zero, then it does support 160 MHz after all,
 417          * in one form or the other. We don't distinguish here (or even
 418          * above) between 160 and 80+80 yet.
 419          */
 420         if (vht_cap->cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
 421                 return IEEE80211_STA_RX_BW_160;
 422
 423         return IEEE80211_STA_RX_BW_80;
 424 }
 425
 426 enum nl80211_chan_width
 427 ieee80211_sta_cap_chan_bw(struct link_sta_info *link_sta)
 428 {
 429         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->pub->vht_cap;
 430         u32 cap_width;
 431
 432         if (!vht_cap->vht_supported) {
 433                 if (!link_sta->pub->ht_cap.ht_supported)
 434                         return NL80211_CHAN_WIDTH_20_NOHT;
 435
 436                 return link_sta->pub->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
 437                                 NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
 438         }
 439
 440         cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 441
 442         if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
 443                 return NL80211_CHAN_WIDTH_160;
 444         else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
 445                 return NL80211_CHAN_WIDTH_80P80;
 446
 447         return NL80211_CHAN_WIDTH_80;
 448 }
 449
 450 enum nl80211_chan_width
 451 ieee80211_sta_rx_bw_to_chan_width(struct link_sta_info *link_sta)
 452 {
 453         enum ieee80211_sta_rx_bandwidth cur_bw =
 454                 link_sta->pub->bandwidth;
 455         struct ieee80211_sta_vht_cap *vht_cap =
 456                 &link_sta->pub->vht_cap;
 457         u32 cap_width;
 458
 459         switch (cur_bw) {
 460         case IEEE80211_STA_RX_BW_20:
 461                 if (!link_sta->pub->ht_cap.ht_supported)
 462                         return NL80211_CHAN_WIDTH_20_NOHT;
 463                 else
 464                         return NL80211_CHAN_WIDTH_20;
 465         case IEEE80211_STA_RX_BW_40:
 466                 return NL80211_CHAN_WIDTH_40;
 467         case IEEE80211_STA_RX_BW_80:
 468                 return NL80211_CHAN_WIDTH_80;
 469         case IEEE80211_STA_RX_BW_160:
 470                 cap_width =
 471                         vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 472
 473                 if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
 474                         return NL80211_CHAN_WIDTH_160;
 475
 476                 return NL80211_CHAN_WIDTH_80P80;
 477         default:
 478                 return NL80211_CHAN_WIDTH_20;
 479         }
 480 }
 481
 482 enum ieee80211_sta_rx_bandwidth
 483 ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
 484 {
 485         switch (width) {
 486         case NL80211_CHAN_WIDTH_20_NOHT:
 487         case NL80211_CHAN_WIDTH_20:
 488                 return IEEE80211_STA_RX_BW_20;
 489         case NL80211_CHAN_WIDTH_40:
 490                 return IEEE80211_STA_RX_BW_40;
 491         case NL80211_CHAN_WIDTH_80:
 492                 return IEEE80211_STA_RX_BW_80;
 493         case NL80211_CHAN_WIDTH_160:
 494         case NL80211_CHAN_WIDTH_80P80:
 495                 return IEEE80211_STA_RX_BW_160;
 496         case NL80211_CHAN_WIDTH_320:
 497                 return IEEE80211_STA_RX_BW_320;
 498         default:
 499                 WARN_ON_ONCE(1);
 500                 return IEEE80211_STA_RX_BW_20;
 501         }
 502 }
 503
 504 /* FIXME: rename/move - this deals with everything not just VHT */
 505 enum ieee80211_sta_rx_bandwidth
 506 _ieee80211_sta_cur_vht_bw(struct link_sta_info *link_sta,
 507                           struct cfg80211_chan_def *chandef)
 508 {
 509         struct sta_info *sta = link_sta->sta;
 510         enum nl80211_chan_width bss_width;
 511         enum ieee80211_sta_rx_bandwidth bw;
 512
 513         if (chandef) {
 514                 bss_width = chandef->width;
 515         } else {
 516                 struct ieee80211_bss_conf *link_conf;
 517
 518                 rcu_read_lock();
 519                 link_conf = rcu_dereference(sta->sdata->vif.link_conf[link_sta->link_id]);
 520                 if (WARN_ON_ONCE(!link_conf)) {
 521                         rcu_read_unlock();
 522                         return IEEE80211_STA_RX_BW_20;
 523                 }
 524                 bss_width = link_conf->chanreq.oper.width;
 525                 rcu_read_unlock();
 526         }
 527
 528         bw = _ieee80211_sta_cap_rx_bw(link_sta, chandef);
 529         bw = min(bw, link_sta->cur_max_bandwidth);
 530
 531         /* Don't consider AP's bandwidth for TDLS peers, section 11.23.1 of
 532          * IEEE80211-2016 specification makes higher bandwidth operation
 533          * possible on the TDLS link if the peers have wider bandwidth
 534          * capability.
 535          *
 536          * However, in this case, and only if the TDLS peer is authorized,
 537          * limit to the tdls_chandef so that the configuration here isn't
 538          * wider than what's actually requested on the channel context.
 539          */
 540         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
 541             test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW) &&
 542             test_sta_flag(sta, WLAN_STA_AUTHORIZED) &&
 543             sta->tdls_chandef.chan)
 544                 bw = min(bw, ieee80211_chan_width_to_rx_bw(sta->tdls_chandef.width));
 545         else
 546                 bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
 547
 548         return bw;
 549 }
 550
 551 void ieee80211_sta_init_nss(struct link_sta_info *link_sta)
 552 {
 553         u8 ht_rx_nss = 0, vht_rx_nss = 0, he_rx_nss = 0, eht_rx_nss = 0, rx_nss;
 554         bool support_160;
 555
 556         if (link_sta->pub->eht_cap.has_eht) {
 557                 int i;
 558                 const u8 *rx_nss_mcs = (void *)&link_sta->pub->eht_cap.eht_mcs_nss_supp;
 559
 560                 /* get the max nss for EHT over all possible bandwidths and mcs */
 561                 for (i = 0; i < sizeof(struct ieee80211_eht_mcs_nss_supp); i++)
 562                         eht_rx_nss = max_t(u8, eht_rx_nss,
 563                                            u8_get_bits(rx_nss_mcs[i],
 564                                                        IEEE80211_EHT_MCS_NSS_RX));
 565         }
 566
 567         if (link_sta->pub->he_cap.has_he) {
 568                 int i;
 569                 u8 rx_mcs_80 = 0, rx_mcs_160 = 0;
 570                 const struct ieee80211_sta_he_cap *he_cap = &link_sta->pub->he_cap;
 571                 u16 mcs_160_map =
 572                         le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_160);
 573                 u16 mcs_80_map = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_80);
 574
 575                 for (i = 7; i >= 0; i--) {
 576                         u8 mcs_160 = (mcs_160_map >> (2 * i)) & 3;
 577
 578                         if (mcs_160 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
 579                                 rx_mcs_160 = i + 1;
 580                                 break;
 581                         }
 582                 }
 583                 for (i = 7; i >= 0; i--) {
 584                         u8 mcs_80 = (mcs_80_map >> (2 * i)) & 3;
 585
 586                         if (mcs_80 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
 587                                 rx_mcs_80 = i + 1;
 588                                 break;
 589                         }
 590                 }
 591
 592                 support_160 = he_cap->he_cap_elem.phy_cap_info[0] &
 593                               IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
 594
 595                 if (support_160)
 596                         he_rx_nss = min(rx_mcs_80, rx_mcs_160);
 597                 else
 598                         he_rx_nss = rx_mcs_80;
 599         }
 600
 601         if (link_sta->pub->ht_cap.ht_supported) {
 602                 if (link_sta->pub->ht_cap.mcs.rx_mask[0])
 603                         ht_rx_nss++;
 604                 if (link_sta->pub->ht_cap.mcs.rx_mask[1])
 605                         ht_rx_nss++;
 606                 if (link_sta->pub->ht_cap.mcs.rx_mask[2])
 607                         ht_rx_nss++;
 608                 if (link_sta->pub->ht_cap.mcs.rx_mask[3])
 609                         ht_rx_nss++;
 610                 /* FIXME: consider rx_highest? */
 611         }
 612
 613         if (link_sta->pub->vht_cap.vht_supported) {
 614                 int i;
 615                 u16 rx_mcs_map;
 616
 617                 rx_mcs_map = le16_to_cpu(link_sta->pub->vht_cap.vht_mcs.rx_mcs_map);
 618
 619                 for (i = 7; i >= 0; i--) {
 620                         u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
 621
 622                         if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 623                                 vht_rx_nss = i + 1;
 624                                 break;
 625                         }
 626                 }
 627                 /* FIXME: consider rx_highest? */
 628         }
 629
 630         rx_nss = max(vht_rx_nss, ht_rx_nss);
 631         rx_nss = max(he_rx_nss, rx_nss);
 632         rx_nss = max(eht_rx_nss, rx_nss);
 633         rx_nss = max_t(u8, 1, rx_nss);
 634         link_sta->capa_nss = rx_nss;
 635
 636         /* that shouldn't be set yet, but we can handle it anyway */
 637         if (link_sta->op_mode_nss)
 638                 link_sta->pub->rx_nss =
 639                         min_t(u8, rx_nss, link_sta->op_mode_nss);
 640         else
 641                 link_sta->pub->rx_nss = rx_nss;
 642 }
 643
 644 u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
 645                                   struct link_sta_info *link_sta,
 646                                   u8 opmode, enum nl80211_band band)
 647 {
 648         enum ieee80211_sta_rx_bandwidth new_bw;
 649         struct sta_opmode_info sta_opmode = {};
 650         u32 changed = 0;
 651         u8 nss;
 652
 653         /* ignore - no support for BF yet */
 654         if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
 655                 return 0;
 656
 657         nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
 658         nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
 659         nss += 1;
 660
 661         if (link_sta->op_mode_nss != nss) {
 662                 if (nss <= link_sta->capa_nss) {
 663                         link_sta->op_mode_nss = nss;
 664
 665                         if (nss != link_sta->pub->rx_nss) {
 666                                 link_sta->pub->rx_nss = nss;
 667                                 changed |= IEEE80211_RC_NSS_CHANGED;
 668                                 sta_opmode.rx_nss = link_sta->pub->rx_nss;
 669                                 sta_opmode.changed |= STA_OPMODE_N_SS_CHANGED;
 670                         }
 671                 } else {
 672                         pr_warn_ratelimited("Ignoring NSS change in VHT Operating Mode Notification from %pM with invalid nss %d",
 673                                             link_sta->pub->addr, nss);
 674                 }
 675         }
 676
 677         switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
 678         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
 679                 /* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
 680                 link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
 681                 break;
 682         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
 683                 /* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
 684                 link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
 685                 break;
 686         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
 687                 if (opmode & IEEE80211_OPMODE_NOTIF_BW_160_80P80)
 688                         link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 689                 else
 690                         link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
 691                 break;
 692         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
 693                 /* legacy only, no longer used by newer spec */
 694                 link_sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 695                 break;
 696         }
 697
 698         new_bw = ieee80211_sta_cur_vht_bw(link_sta);
 699         if (new_bw != link_sta->pub->bandwidth) {
 700                 link_sta->pub->bandwidth = new_bw;
 701                 sta_opmode.bw = ieee80211_sta_rx_bw_to_chan_width(link_sta);
 702                 changed |= IEEE80211_RC_BW_CHANGED;
 703                 sta_opmode.changed |= STA_OPMODE_MAX_BW_CHANGED;
 704         }
 705
 706         if (sta_opmode.changed)
 707                 cfg80211_sta_opmode_change_notify(sdata->dev, link_sta->addr,
 708                                                   &sta_opmode, GFP_KERNEL);
 709
 710         return changed;
 711 }
 712
 713 void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
 714                                  struct ieee80211_link_data *link,
 715                                  struct ieee80211_mgmt *mgmt)
 716 {
 717         struct ieee80211_bss_conf *link_conf = link->conf;
 718
 719         if (!link_conf->mu_mimo_owner)
 720                 return;
 721
 722         if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
 723                     link_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
 724             !memcmp(mgmt->u.action.u.vht_group_notif.membership,
 725                     link_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
 726                 return;
 727
 728         memcpy(link_conf->mu_group.membership,
 729                mgmt->u.action.u.vht_group_notif.membership,
 730                WLAN_MEMBERSHIP_LEN);
 731         memcpy(link_conf->mu_group.position,
 732                mgmt->u.action.u.vht_group_notif.position,
 733                WLAN_USER_POSITION_LEN);
 734
 735         ieee80211_link_info_change_notify(sdata, link,
 736                                           BSS_CHANGED_MU_GROUPS);
 737 }
 738
 739 void ieee80211_update_mu_groups(struct ieee80211_vif *vif, unsigned int link_id,
 740                                 const u8 *membership, const u8 *position)
 741 {
 742         struct ieee80211_bss_conf *link_conf;
 743
 744         rcu_read_lock();
 745         link_conf = rcu_dereference(vif->link_conf[link_id]);
 746
 747         if (!WARN_ON_ONCE(!link_conf || !link_conf->mu_mimo_owner)) {
 748                 memcpy(link_conf->mu_group.membership, membership,
 749                        WLAN_MEMBERSHIP_LEN);
 750                 memcpy(link_conf->mu_group.position, position,
 751                        WLAN_USER_POSITION_LEN);
 752         }
 753         rcu_read_unlock();
 754 }
 755 EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
 756
 757 void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
 758                                  struct link_sta_info *link_sta,
 759                                  u8 opmode, enum nl80211_band band)
 760 {
 761         struct ieee80211_local *local = sdata->local;
 762         struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
 763
 764         u32 changed = __ieee80211_vht_handle_opmode(sdata, link_sta,
 765                                                     opmode, band);
 766
 767         if (changed > 0) {
 768                 ieee80211_recalc_min_chandef(sdata, link_sta->link_id);
 769                 rate_control_rate_update(local, sband, link_sta->sta,
 770                                          link_sta->link_id, changed);
 771         }
 772 }
 773
 774 void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
 775                                      u16 vht_mask[NL80211_VHT_NSS_MAX])
 776 {
 777         int i;
 778         u16 mask, cap = le16_to_cpu(vht_cap);
 779
 780         for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
 781                 mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 782                 switch (mask) {
 783                 case IEEE80211_VHT_MCS_SUPPORT_0_7:
 784                         vht_mask[i] = 0x00FF;
 785                         break;
 786                 case IEEE80211_VHT_MCS_SUPPORT_0_8:
 787                         vht_mask[i] = 0x01FF;
 788                         break;
 789                 case IEEE80211_VHT_MCS_SUPPORT_0_9:
 790                         vht_mask[i] = 0x03FF;
 791                         break;
 792                 case IEEE80211_VHT_MCS_NOT_SUPPORTED:
 793                 default:
 794                         vht_mask[i] = 0;
 795                         break;
 796                 }
 797         }
 798 }