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 - 2022 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                                     struct sta_info *sta)
 120 {
 121         struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
 122         struct ieee80211_sta_vht_cap own_cap;
 123         u32 cap_info, i;
 124         bool have_80mhz;
 125
 126         memset(vht_cap, 0, sizeof(*vht_cap));
 127
 128         if (!sta->sta.deflink.ht_cap.ht_supported)
 129                 return;
 130
 131         if (!vht_cap_ie || !sband->vht_cap.vht_supported)
 132                 return;
 133
 134         /* Allow VHT if at least one channel on the sband supports 80 MHz */
 135         have_80mhz = false;
 136         for (i = 0; i < sband->n_channels; i++) {
 137                 if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
 138                                                 IEEE80211_CHAN_NO_80MHZ))
 139                         continue;
 140
 141                 have_80mhz = true;
 142                 break;
 143         }
 144
 145         if (!have_80mhz)
 146                 return;
 147
 148         /*
 149          * A VHT STA must support 40 MHz, but if we verify that here
 150          * then we break a few things - some APs (e.g. Netgear R6300v2
 151          * and others based on the BCM4360 chipset) will unset this
 152          * capability bit when operating in 20 MHz.
 153          */
 154
 155         vht_cap->vht_supported = true;
 156
 157         own_cap = sband->vht_cap;
 158         /*
 159          * If user has specified capability overrides, take care
 160          * of that if the station we're setting up is the AP that
 161          * we advertised a restricted capability set to. Override
 162          * our own capabilities and then use those below.
 163          */
 164         if (sdata->vif.type == NL80211_IFTYPE_STATION &&
 165             !test_sta_flag(sta, WLAN_STA_TDLS_PEER))
 166                 ieee80211_apply_vhtcap_overrides(sdata, &own_cap);
 167
 168         /* take some capabilities as-is */
 169         cap_info = le32_to_cpu(vht_cap_ie->vht_cap_info);
 170         vht_cap->cap = cap_info;
 171         vht_cap->cap &= IEEE80211_VHT_CAP_RXLDPC |
 172                         IEEE80211_VHT_CAP_VHT_TXOP_PS |
 173                         IEEE80211_VHT_CAP_HTC_VHT |
 174                         IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
 175                         IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB |
 176                         IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB |
 177                         IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
 178                         IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;
 179
 180         vht_cap->cap |= min_t(u32, cap_info & IEEE80211_VHT_CAP_MAX_MPDU_MASK,
 181                               own_cap.cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK);
 182
 183         /* and some based on our own capabilities */
 184         switch (own_cap.cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
 185         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
 186                 vht_cap->cap |= cap_info &
 187                                 IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
 188                 break;
 189         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
 190                 vht_cap->cap |= cap_info &
 191                                 IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 192                 break;
 193         default:
 194                 /* nothing */
 195                 break;
 196         }
 197
 198         /* symmetric capabilities */
 199         vht_cap->cap |= cap_info & own_cap.cap &
 200                         (IEEE80211_VHT_CAP_SHORT_GI_80 |
 201                          IEEE80211_VHT_CAP_SHORT_GI_160);
 202
 203         /* remaining ones */
 204         if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
 205                 vht_cap->cap |= cap_info &
 206                                 (IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
 207                                  IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
 208
 209         if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
 210                 vht_cap->cap |= cap_info &
 211                                 (IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
 212                                  IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
 213
 214         if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)
 215                 vht_cap->cap |= cap_info &
 216                                 IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
 217
 218         if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE)
 219                 vht_cap->cap |= cap_info &
 220                                 IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE;
 221
 222         if (own_cap.cap & IEEE80211_VHT_CAP_TXSTBC)
 223                 vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_RXSTBC_MASK;
 224
 225         if (own_cap.cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
 226                 vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_TXSTBC;
 227
 228         /* Copy peer MCS info, the driver might need them. */
 229         memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
 230                sizeof(struct ieee80211_vht_mcs_info));
 231
 232         /* copy EXT_NSS_BW Support value or remove the capability */
 233         if (ieee80211_hw_check(&sdata->local->hw, SUPPORTS_VHT_EXT_NSS_BW))
 234                 vht_cap->cap |= (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK);
 235         else
 236                 vht_cap->vht_mcs.tx_highest &=
 237                         ~cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE);
 238
 239         /* but also restrict MCSes */
 240         for (i = 0; i < 8; i++) {
 241                 u16 own_rx, own_tx, peer_rx, peer_tx;
 242
 243                 own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
 244                 own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 245
 246                 own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
 247                 own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 248
 249                 peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
 250                 peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 251
 252                 peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
 253                 peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 254
 255                 if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 256                         if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
 257                                 peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
 258                         else if (own_rx < peer_tx)
 259                                 peer_tx = own_rx;
 260                 }
 261
 262                 if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 263                         if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
 264                                 peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
 265                         else if (own_tx < peer_rx)
 266                                 peer_rx = own_tx;
 267                 }
 268
 269                 vht_cap->vht_mcs.rx_mcs_map &=
 270                         ~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
 271                 vht_cap->vht_mcs.rx_mcs_map |= cpu_to_le16(peer_rx << i * 2);
 272
 273                 vht_cap->vht_mcs.tx_mcs_map &=
 274                         ~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
 275                 vht_cap->vht_mcs.tx_mcs_map |= cpu_to_le16(peer_tx << i * 2);
 276         }
 277
 278         /*
 279          * This is a workaround for VHT-enabled STAs which break the spec
 280          * and have the VHT-MCS Rx map filled in with value 3 for all eight
 281          * spacial streams, an example is AR9462.
 282          *
 283          * As per spec, in section 22.1.1 Introduction to the VHT PHY
 284          * A VHT STA shall support at least single spactial stream VHT-MCSs
 285          * 0 to 7 (transmit and receive) in all supported channel widths.
 286          */
 287         if (vht_cap->vht_mcs.rx_mcs_map == cpu_to_le16(0xFFFF)) {
 288                 vht_cap->vht_supported = false;
 289                 sdata_info(sdata, "Ignoring VHT IE from %pM due to invalid rx_mcs_map\n",
 290                            sta->addr);
 291                 return;
 292         }
 293
 294         /* finally set up the bandwidth */
 295         switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
 296         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
 297         case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
 298                 sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 299                 break;
 300         default:
 301                 sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
 302
 303                 if (!(vht_cap->vht_mcs.tx_highest &
 304                                 cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE)))
 305                         break;
 306
 307                 /*
 308                  * If this is non-zero, then it does support 160 MHz after all,
 309                  * in one form or the other. We don't distinguish here (or even
 310                  * above) between 160 and 80+80 yet.
 311                  */
 312                 if (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
 313                         sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 314         }
 315
 316         sta->sta.deflink.bandwidth = ieee80211_sta_cur_vht_bw(sta, 0);
 317
 318         switch (vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK) {
 319         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
 320                 sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
 321                 break;
 322         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
 323                 sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
 324                 break;
 325         case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
 326         default:
 327                 sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
 328                 break;
 329         }
 330 }
 331
 332 /* FIXME: move this to some better location - parses HE/EHT now */
 333 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta,
 334                                                         unsigned int link_id)
 335 {
 336         struct ieee80211_bss_conf *link_conf = sta->sdata->vif.link_conf[link_id];
 337         struct ieee80211_link_sta *link_sta = sta->sta.link[link_id];
 338         struct ieee80211_sta_vht_cap *vht_cap = &link_sta->vht_cap;
 339         struct ieee80211_sta_he_cap *he_cap = &link_sta->he_cap;
 340         struct ieee80211_sta_eht_cap *eht_cap = &link_sta->eht_cap;
 341         u32 cap_width;
 342
 343         if (he_cap->has_he) {
 344                 u8 info;
 345
 346                 if (eht_cap->has_eht &&
 347                     link_conf->chandef.chan->band == NL80211_BAND_6GHZ) {
 348                         info = eht_cap->eht_cap_elem.phy_cap_info[0];
 349
 350                         if (info & IEEE80211_EHT_PHY_CAP0_320MHZ_IN_6GHZ)
 351                                 return IEEE80211_STA_RX_BW_320;
 352                 }
 353
 354                 info = he_cap->he_cap_elem.phy_cap_info[0];
 355
 356                 if (link_conf->chandef.chan->band == NL80211_BAND_2GHZ) {
 357                         if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_IN_2G)
 358                                 return IEEE80211_STA_RX_BW_40;
 359                         else
 360                                 return IEEE80211_STA_RX_BW_20;
 361                 }
 362
 363                 if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G ||
 364                     info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G)
 365                         return IEEE80211_STA_RX_BW_160;
 366                 else if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_80MHZ_IN_5G)
 367                         return IEEE80211_STA_RX_BW_80;
 368
 369                 return IEEE80211_STA_RX_BW_20;
 370         }
 371
 372         if (!vht_cap->vht_supported)
 373                 return link_sta->ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
 374                                 IEEE80211_STA_RX_BW_40 :
 375                                 IEEE80211_STA_RX_BW_20;
 376
 377         cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 378
 379         if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
 380             cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
 381                 return IEEE80211_STA_RX_BW_160;
 382
 383         /*
 384          * If this is non-zero, then it does support 160 MHz after all,
 385          * in one form or the other. We don't distinguish here (or even
 386          * above) between 160 and 80+80 yet.
 387          */
 388         if (vht_cap->cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
 389                 return IEEE80211_STA_RX_BW_160;
 390
 391         return IEEE80211_STA_RX_BW_80;
 392 }
 393
 394 enum nl80211_chan_width ieee80211_sta_cap_chan_bw(struct sta_info *sta)
 395 {
 396         struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
 397         u32 cap_width;
 398
 399         if (!vht_cap->vht_supported) {
 400                 if (!sta->sta.deflink.ht_cap.ht_supported)
 401                         return NL80211_CHAN_WIDTH_20_NOHT;
 402
 403                 return sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
 404                                 NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
 405         }
 406
 407         cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 408
 409         if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
 410                 return NL80211_CHAN_WIDTH_160;
 411         else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
 412                 return NL80211_CHAN_WIDTH_80P80;
 413
 414         return NL80211_CHAN_WIDTH_80;
 415 }
 416
 417 enum nl80211_chan_width
 418 ieee80211_sta_rx_bw_to_chan_width(struct sta_info *sta)
 419 {
 420         enum ieee80211_sta_rx_bandwidth cur_bw = sta->sta.deflink.bandwidth;
 421         struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
 422         u32 cap_width;
 423
 424         switch (cur_bw) {
 425         case IEEE80211_STA_RX_BW_20:
 426                 if (!sta->sta.deflink.ht_cap.ht_supported)
 427                         return NL80211_CHAN_WIDTH_20_NOHT;
 428                 else
 429                         return NL80211_CHAN_WIDTH_20;
 430         case IEEE80211_STA_RX_BW_40:
 431                 return NL80211_CHAN_WIDTH_40;
 432         case IEEE80211_STA_RX_BW_80:
 433                 return NL80211_CHAN_WIDTH_80;
 434         case IEEE80211_STA_RX_BW_160:
 435                 cap_width =
 436                         vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
 437
 438                 if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
 439                         return NL80211_CHAN_WIDTH_160;
 440
 441                 return NL80211_CHAN_WIDTH_80P80;
 442         default:
 443                 return NL80211_CHAN_WIDTH_20;
 444         }
 445 }
 446
 447 enum ieee80211_sta_rx_bandwidth
 448 ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
 449 {
 450         switch (width) {
 451         case NL80211_CHAN_WIDTH_20_NOHT:
 452         case NL80211_CHAN_WIDTH_20:
 453                 return IEEE80211_STA_RX_BW_20;
 454         case NL80211_CHAN_WIDTH_40:
 455                 return IEEE80211_STA_RX_BW_40;
 456         case NL80211_CHAN_WIDTH_80:
 457                 return IEEE80211_STA_RX_BW_80;
 458         case NL80211_CHAN_WIDTH_160:
 459         case NL80211_CHAN_WIDTH_80P80:
 460                 return IEEE80211_STA_RX_BW_160;
 461         case NL80211_CHAN_WIDTH_320:
 462                 return IEEE80211_STA_RX_BW_320;
 463         default:
 464                 WARN_ON_ONCE(1);
 465                 return IEEE80211_STA_RX_BW_20;
 466         }
 467 }
 468
 469 /* FIXME: rename/move - this deals with everything not just VHT */
 470 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta,
 471                                                          unsigned int link_id)
 472 {
 473         struct ieee80211_bss_conf *link_conf = sta->sdata->vif.link_conf[link_id];
 474         enum nl80211_chan_width bss_width = link_conf->chandef.width;
 475         enum ieee80211_sta_rx_bandwidth bw;
 476
 477         bw = ieee80211_sta_cap_rx_bw(sta, link_id);
 478         bw = min(bw, sta->link[link_id]->cur_max_bandwidth);
 479
 480         /* Don't consider AP's bandwidth for TDLS peers, section 11.23.1 of
 481          * IEEE80211-2016 specification makes higher bandwidth operation
 482          * possible on the TDLS link if the peers have wider bandwidth
 483          * capability.
 484          *
 485          * However, in this case, and only if the TDLS peer is authorized,
 486          * limit to the tdls_chandef so that the configuration here isn't
 487          * wider than what's actually requested on the channel context.
 488          */
 489         if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
 490             test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW) &&
 491             test_sta_flag(sta, WLAN_STA_AUTHORIZED) &&
 492             sta->tdls_chandef.chan)
 493                 bw = min(bw, ieee80211_chan_width_to_rx_bw(sta->tdls_chandef.width));
 494         else
 495                 bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
 496
 497         return bw;
 498 }
 499
 500 void ieee80211_sta_set_rx_nss(struct sta_info *sta)
 501 {
 502         u8 ht_rx_nss = 0, vht_rx_nss = 0, he_rx_nss = 0, eht_rx_nss = 0, rx_nss;
 503         bool support_160;
 504
 505         /* if we received a notification already don't overwrite it */
 506         if (sta->sta.deflink.rx_nss)
 507                 return;
 508
 509         if (sta->sta.deflink.eht_cap.has_eht) {
 510                 int i;
 511                 const u8 *rx_nss_mcs = (void *)&sta->sta.deflink.eht_cap.eht_mcs_nss_supp;
 512
 513                 /* get the max nss for EHT over all possible bandwidths and mcs */
 514                 for (i = 0; i < sizeof(struct ieee80211_eht_mcs_nss_supp); i++)
 515                         eht_rx_nss = max_t(u8, eht_rx_nss,
 516                                            u8_get_bits(rx_nss_mcs[i],
 517                                                        IEEE80211_EHT_MCS_NSS_RX));
 518         }
 519
 520         if (sta->sta.deflink.he_cap.has_he) {
 521                 int i;
 522                 u8 rx_mcs_80 = 0, rx_mcs_160 = 0;
 523                 const struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
 524                 u16 mcs_160_map =
 525                         le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_160);
 526                 u16 mcs_80_map = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_80);
 527
 528                 for (i = 7; i >= 0; i--) {
 529                         u8 mcs_160 = (mcs_160_map >> (2 * i)) & 3;
 530
 531                         if (mcs_160 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
 532                                 rx_mcs_160 = i + 1;
 533                                 break;
 534                         }
 535                 }
 536                 for (i = 7; i >= 0; i--) {
 537                         u8 mcs_80 = (mcs_80_map >> (2 * i)) & 3;
 538
 539                         if (mcs_80 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
 540                                 rx_mcs_80 = i + 1;
 541                                 break;
 542                         }
 543                 }
 544
 545                 support_160 = he_cap->he_cap_elem.phy_cap_info[0] &
 546                               IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
 547
 548                 if (support_160)
 549                         he_rx_nss = min(rx_mcs_80, rx_mcs_160);
 550                 else
 551                         he_rx_nss = rx_mcs_80;
 552         }
 553
 554         if (sta->sta.deflink.ht_cap.ht_supported) {
 555                 if (sta->sta.deflink.ht_cap.mcs.rx_mask[0])
 556                         ht_rx_nss++;
 557                 if (sta->sta.deflink.ht_cap.mcs.rx_mask[1])
 558                         ht_rx_nss++;
 559                 if (sta->sta.deflink.ht_cap.mcs.rx_mask[2])
 560                         ht_rx_nss++;
 561                 if (sta->sta.deflink.ht_cap.mcs.rx_mask[3])
 562                         ht_rx_nss++;
 563                 /* FIXME: consider rx_highest? */
 564         }
 565
 566         if (sta->sta.deflink.vht_cap.vht_supported) {
 567                 int i;
 568                 u16 rx_mcs_map;
 569
 570                 rx_mcs_map = le16_to_cpu(sta->sta.deflink.vht_cap.vht_mcs.rx_mcs_map);
 571
 572                 for (i = 7; i >= 0; i--) {
 573                         u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
 574
 575                         if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
 576                                 vht_rx_nss = i + 1;
 577                                 break;
 578                         }
 579                 }
 580                 /* FIXME: consider rx_highest? */
 581         }
 582
 583         rx_nss = max(vht_rx_nss, ht_rx_nss);
 584         rx_nss = max(he_rx_nss, rx_nss);
 585         rx_nss = max(eht_rx_nss, rx_nss);
 586         sta->sta.deflink.rx_nss = max_t(u8, 1, rx_nss);
 587 }
 588
 589 u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
 590                                   struct sta_info *sta, u8 opmode,
 591                                   enum nl80211_band band)
 592 {
 593         enum ieee80211_sta_rx_bandwidth new_bw;
 594         struct sta_opmode_info sta_opmode = {};
 595         u32 changed = 0;
 596         u8 nss;
 597
 598         /* ignore - no support for BF yet */
 599         if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
 600                 return 0;
 601
 602         nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
 603         nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
 604         nss += 1;
 605
 606         if (sta->sta.deflink.rx_nss != nss) {
 607                 sta->sta.deflink.rx_nss = nss;
 608                 sta_opmode.rx_nss = nss;
 609                 changed |= IEEE80211_RC_NSS_CHANGED;
 610                 sta_opmode.changed |= STA_OPMODE_N_SS_CHANGED;
 611         }
 612
 613         switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
 614         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
 615                 /* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
 616                 sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
 617                 break;
 618         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
 619                 /* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
 620                 sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
 621                 break;
 622         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
 623                 if (opmode & IEEE80211_OPMODE_NOTIF_BW_160_80P80)
 624                         sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 625                 else
 626                         sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
 627                 break;
 628         case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
 629                 /* legacy only, no longer used by newer spec */
 630                 sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
 631                 break;
 632         }
 633
 634         new_bw = ieee80211_sta_cur_vht_bw(sta, 0);
 635         if (new_bw != sta->sta.deflink.bandwidth) {
 636                 sta->sta.deflink.bandwidth = new_bw;
 637                 sta_opmode.bw = ieee80211_sta_rx_bw_to_chan_width(sta);
 638                 changed |= IEEE80211_RC_BW_CHANGED;
 639                 sta_opmode.changed |= STA_OPMODE_MAX_BW_CHANGED;
 640         }
 641
 642         if (sta_opmode.changed)
 643                 cfg80211_sta_opmode_change_notify(sdata->dev, sta->addr,
 644                                                   &sta_opmode, GFP_KERNEL);
 645
 646         return changed;
 647 }
 648
 649 void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
 650                                  struct ieee80211_mgmt *mgmt)
 651 {
 652         struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
 653
 654         if (!sdata->vif.bss_conf.mu_mimo_owner)
 655                 return;
 656
 657         if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
 658                     bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
 659             !memcmp(mgmt->u.action.u.vht_group_notif.membership,
 660                     bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
 661                 return;
 662
 663         memcpy(bss_conf->mu_group.membership,
 664                mgmt->u.action.u.vht_group_notif.membership,
 665                WLAN_MEMBERSHIP_LEN);
 666         memcpy(bss_conf->mu_group.position,
 667                mgmt->u.action.u.vht_group_notif.position,
 668                WLAN_USER_POSITION_LEN);
 669
 670         ieee80211_link_info_change_notify(sdata, 0, BSS_CHANGED_MU_GROUPS);
 671 }
 672
 673 void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
 674                                 const u8 *membership, const u8 *position)
 675 {
 676         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
 677
 678         if (WARN_ON_ONCE(!vif->bss_conf.mu_mimo_owner))
 679                 return;
 680
 681         memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
 682         memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
 683 }
 684 EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
 685
 686 void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
 687                                  struct sta_info *sta, u8 opmode,
 688                                  enum nl80211_band band)
 689 {
 690         struct ieee80211_local *local = sdata->local;
 691         struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
 692
 693         u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);
 694
 695         if (changed > 0) {
 696                 ieee80211_recalc_min_chandef(sdata);
 697                 rate_control_rate_update(local, sband, sta, 0, changed);
 698         }
 699 }
 700
 701 void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
 702                                      u16 vht_mask[NL80211_VHT_NSS_MAX])
 703 {
 704         int i;
 705         u16 mask, cap = le16_to_cpu(vht_cap);
 706
 707         for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
 708                 mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
 709                 switch (mask) {
 710                 case IEEE80211_VHT_MCS_SUPPORT_0_7:
 711                         vht_mask[i] = 0x00FF;
 712                         break;
 713                 case IEEE80211_VHT_MCS_SUPPORT_0_8:
 714                         vht_mask[i] = 0x01FF;
 715                         break;
 716                 case IEEE80211_VHT_MCS_SUPPORT_0_9:
 717                         vht_mask[i] = 0x03FF;
 718                         break;
 719                 case IEEE80211_VHT_MCS_NOT_SUPPORTED:
 720                 default:
 721                         vht_mask[i] = 0;
 722                         break;
 723                 }
 724         }
 725 }