cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

vht.c (22395B)


      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 - 2021 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
     17static 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
     28void 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
    115void
    116ieee80211_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);
    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 */
    333enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
    334{
    335	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
    336	struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
    337	struct ieee80211_sta_eht_cap *eht_cap = &sta->sta.deflink.eht_cap;
    338	u32 cap_width;
    339
    340	if (he_cap->has_he) {
    341		u8 info;
    342
    343		if (eht_cap->has_eht &&
    344		    sta->sdata->vif.bss_conf.chandef.chan->band ==
    345		    NL80211_BAND_6GHZ) {
    346			info = eht_cap->eht_cap_elem.phy_cap_info[0];
    347
    348			if (info & IEEE80211_EHT_PHY_CAP0_320MHZ_IN_6GHZ)
    349				return IEEE80211_STA_RX_BW_320;
    350		}
    351
    352		info = he_cap->he_cap_elem.phy_cap_info[0];
    353
    354		if (sta->sdata->vif.bss_conf.chandef.chan->band ==
    355				NL80211_BAND_2GHZ) {
    356			if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_IN_2G)
    357				return IEEE80211_STA_RX_BW_40;
    358			else
    359				return IEEE80211_STA_RX_BW_20;
    360		}
    361
    362		if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G ||
    363		    info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G)
    364			return IEEE80211_STA_RX_BW_160;
    365		else if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_80MHZ_IN_5G)
    366			return IEEE80211_STA_RX_BW_80;
    367
    368		return IEEE80211_STA_RX_BW_20;
    369	}
    370
    371	if (!vht_cap->vht_supported)
    372		return sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
    373				IEEE80211_STA_RX_BW_40 :
    374				IEEE80211_STA_RX_BW_20;
    375
    376	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
    377
    378	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
    379	    cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
    380		return IEEE80211_STA_RX_BW_160;
    381
    382	/*
    383	 * If this is non-zero, then it does support 160 MHz after all,
    384	 * in one form or the other. We don't distinguish here (or even
    385	 * above) between 160 and 80+80 yet.
    386	 */
    387	if (vht_cap->cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
    388		return IEEE80211_STA_RX_BW_160;
    389
    390	return IEEE80211_STA_RX_BW_80;
    391}
    392
    393enum nl80211_chan_width ieee80211_sta_cap_chan_bw(struct sta_info *sta)
    394{
    395	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
    396	u32 cap_width;
    397
    398	if (!vht_cap->vht_supported) {
    399		if (!sta->sta.deflink.ht_cap.ht_supported)
    400			return NL80211_CHAN_WIDTH_20_NOHT;
    401
    402		return sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
    403				NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
    404	}
    405
    406	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
    407
    408	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
    409		return NL80211_CHAN_WIDTH_160;
    410	else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
    411		return NL80211_CHAN_WIDTH_80P80;
    412
    413	return NL80211_CHAN_WIDTH_80;
    414}
    415
    416enum nl80211_chan_width
    417ieee80211_sta_rx_bw_to_chan_width(struct sta_info *sta)
    418{
    419	enum ieee80211_sta_rx_bandwidth cur_bw = sta->sta.deflink.bandwidth;
    420	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
    421	u32 cap_width;
    422
    423	switch (cur_bw) {
    424	case IEEE80211_STA_RX_BW_20:
    425		if (!sta->sta.deflink.ht_cap.ht_supported)
    426			return NL80211_CHAN_WIDTH_20_NOHT;
    427		else
    428			return NL80211_CHAN_WIDTH_20;
    429	case IEEE80211_STA_RX_BW_40:
    430		return NL80211_CHAN_WIDTH_40;
    431	case IEEE80211_STA_RX_BW_80:
    432		return NL80211_CHAN_WIDTH_80;
    433	case IEEE80211_STA_RX_BW_160:
    434		cap_width =
    435			vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
    436
    437		if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
    438			return NL80211_CHAN_WIDTH_160;
    439
    440		return NL80211_CHAN_WIDTH_80P80;
    441	default:
    442		return NL80211_CHAN_WIDTH_20;
    443	}
    444}
    445
    446enum ieee80211_sta_rx_bandwidth
    447ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
    448{
    449	switch (width) {
    450	case NL80211_CHAN_WIDTH_20_NOHT:
    451	case NL80211_CHAN_WIDTH_20:
    452		return IEEE80211_STA_RX_BW_20;
    453	case NL80211_CHAN_WIDTH_40:
    454		return IEEE80211_STA_RX_BW_40;
    455	case NL80211_CHAN_WIDTH_80:
    456		return IEEE80211_STA_RX_BW_80;
    457	case NL80211_CHAN_WIDTH_160:
    458	case NL80211_CHAN_WIDTH_80P80:
    459		return IEEE80211_STA_RX_BW_160;
    460	case NL80211_CHAN_WIDTH_320:
    461		return IEEE80211_STA_RX_BW_320;
    462	default:
    463		WARN_ON_ONCE(1);
    464		return IEEE80211_STA_RX_BW_20;
    465	}
    466}
    467
    468/* FIXME: rename/move - this deals with everything not just VHT */
    469enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
    470{
    471	struct ieee80211_sub_if_data *sdata = sta->sdata;
    472	enum ieee80211_sta_rx_bandwidth bw;
    473	enum nl80211_chan_width bss_width = sdata->vif.bss_conf.chandef.width;
    474
    475	bw = ieee80211_sta_cap_rx_bw(sta);
    476	bw = min(bw, sta->deflink.cur_max_bandwidth);
    477
    478	/* Don't consider AP's bandwidth for TDLS peers, section 11.23.1 of
    479	 * IEEE80211-2016 specification makes higher bandwidth operation
    480	 * possible on the TDLS link if the peers have wider bandwidth
    481	 * capability.
    482	 *
    483	 * However, in this case, and only if the TDLS peer is authorized,
    484	 * limit to the tdls_chandef so that the configuration here isn't
    485	 * wider than what's actually requested on the channel context.
    486	 */
    487	if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
    488	    test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW) &&
    489	    test_sta_flag(sta, WLAN_STA_AUTHORIZED) &&
    490	    sta->tdls_chandef.chan)
    491		bw = min(bw, ieee80211_chan_width_to_rx_bw(sta->tdls_chandef.width));
    492	else
    493		bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
    494
    495	return bw;
    496}
    497
    498void ieee80211_sta_set_rx_nss(struct sta_info *sta)
    499{
    500	u8 ht_rx_nss = 0, vht_rx_nss = 0, he_rx_nss = 0, eht_rx_nss = 0, rx_nss;
    501	bool support_160;
    502
    503	/* if we received a notification already don't overwrite it */
    504	if (sta->sta.deflink.rx_nss)
    505		return;
    506
    507	if (sta->sta.deflink.eht_cap.has_eht) {
    508		int i;
    509		const u8 *rx_nss_mcs = (void *)&sta->sta.deflink.eht_cap.eht_mcs_nss_supp;
    510
    511		/* get the max nss for EHT over all possible bandwidths and mcs */
    512		for (i = 0; i < sizeof(struct ieee80211_eht_mcs_nss_supp); i++)
    513			eht_rx_nss = max_t(u8, eht_rx_nss,
    514					   u8_get_bits(rx_nss_mcs[i],
    515						       IEEE80211_EHT_MCS_NSS_RX));
    516	}
    517
    518	if (sta->sta.deflink.he_cap.has_he) {
    519		int i;
    520		u8 rx_mcs_80 = 0, rx_mcs_160 = 0;
    521		const struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
    522		u16 mcs_160_map =
    523			le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_160);
    524		u16 mcs_80_map = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_80);
    525
    526		for (i = 7; i >= 0; i--) {
    527			u8 mcs_160 = (mcs_160_map >> (2 * i)) & 3;
    528
    529			if (mcs_160 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
    530				rx_mcs_160 = i + 1;
    531				break;
    532			}
    533		}
    534		for (i = 7; i >= 0; i--) {
    535			u8 mcs_80 = (mcs_80_map >> (2 * i)) & 3;
    536
    537			if (mcs_80 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
    538				rx_mcs_80 = i + 1;
    539				break;
    540			}
    541		}
    542
    543		support_160 = he_cap->he_cap_elem.phy_cap_info[0] &
    544			      IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
    545
    546		if (support_160)
    547			he_rx_nss = min(rx_mcs_80, rx_mcs_160);
    548		else
    549			he_rx_nss = rx_mcs_80;
    550	}
    551
    552	if (sta->sta.deflink.ht_cap.ht_supported) {
    553		if (sta->sta.deflink.ht_cap.mcs.rx_mask[0])
    554			ht_rx_nss++;
    555		if (sta->sta.deflink.ht_cap.mcs.rx_mask[1])
    556			ht_rx_nss++;
    557		if (sta->sta.deflink.ht_cap.mcs.rx_mask[2])
    558			ht_rx_nss++;
    559		if (sta->sta.deflink.ht_cap.mcs.rx_mask[3])
    560			ht_rx_nss++;
    561		/* FIXME: consider rx_highest? */
    562	}
    563
    564	if (sta->sta.deflink.vht_cap.vht_supported) {
    565		int i;
    566		u16 rx_mcs_map;
    567
    568		rx_mcs_map = le16_to_cpu(sta->sta.deflink.vht_cap.vht_mcs.rx_mcs_map);
    569
    570		for (i = 7; i >= 0; i--) {
    571			u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
    572
    573			if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
    574				vht_rx_nss = i + 1;
    575				break;
    576			}
    577		}
    578		/* FIXME: consider rx_highest? */
    579	}
    580
    581	rx_nss = max(vht_rx_nss, ht_rx_nss);
    582	rx_nss = max(he_rx_nss, rx_nss);
    583	rx_nss = max(eht_rx_nss, rx_nss);
    584	sta->sta.deflink.rx_nss = max_t(u8, 1, rx_nss);
    585}
    586
    587u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
    588				  struct sta_info *sta, u8 opmode,
    589				  enum nl80211_band band)
    590{
    591	enum ieee80211_sta_rx_bandwidth new_bw;
    592	struct sta_opmode_info sta_opmode = {};
    593	u32 changed = 0;
    594	u8 nss;
    595
    596	/* ignore - no support for BF yet */
    597	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
    598		return 0;
    599
    600	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
    601	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
    602	nss += 1;
    603
    604	if (sta->sta.deflink.rx_nss != nss) {
    605		sta->sta.deflink.rx_nss = nss;
    606		sta_opmode.rx_nss = nss;
    607		changed |= IEEE80211_RC_NSS_CHANGED;
    608		sta_opmode.changed |= STA_OPMODE_N_SS_CHANGED;
    609	}
    610
    611	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
    612	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
    613		/* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
    614		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
    615		break;
    616	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
    617		/* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
    618		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
    619		break;
    620	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
    621		if (opmode & IEEE80211_OPMODE_NOTIF_BW_160_80P80)
    622			sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
    623		else
    624			sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
    625		break;
    626	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
    627		/* legacy only, no longer used by newer spec */
    628		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
    629		break;
    630	}
    631
    632	new_bw = ieee80211_sta_cur_vht_bw(sta);
    633	if (new_bw != sta->sta.deflink.bandwidth) {
    634		sta->sta.deflink.bandwidth = new_bw;
    635		sta_opmode.bw = ieee80211_sta_rx_bw_to_chan_width(sta);
    636		changed |= IEEE80211_RC_BW_CHANGED;
    637		sta_opmode.changed |= STA_OPMODE_MAX_BW_CHANGED;
    638	}
    639
    640	if (sta_opmode.changed)
    641		cfg80211_sta_opmode_change_notify(sdata->dev, sta->addr,
    642						  &sta_opmode, GFP_KERNEL);
    643
    644	return changed;
    645}
    646
    647void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
    648				 struct ieee80211_mgmt *mgmt)
    649{
    650	struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
    651
    652	if (!sdata->vif.mu_mimo_owner)
    653		return;
    654
    655	if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
    656		    bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
    657	    !memcmp(mgmt->u.action.u.vht_group_notif.membership,
    658		    bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
    659		return;
    660
    661	memcpy(bss_conf->mu_group.membership,
    662	       mgmt->u.action.u.vht_group_notif.membership,
    663	       WLAN_MEMBERSHIP_LEN);
    664	memcpy(bss_conf->mu_group.position,
    665	       mgmt->u.action.u.vht_group_notif.position,
    666	       WLAN_USER_POSITION_LEN);
    667
    668	ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_MU_GROUPS);
    669}
    670
    671void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
    672				const u8 *membership, const u8 *position)
    673{
    674	struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
    675
    676	if (WARN_ON_ONCE(!vif->mu_mimo_owner))
    677		return;
    678
    679	memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
    680	memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
    681}
    682EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
    683
    684void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
    685				 struct sta_info *sta, u8 opmode,
    686				 enum nl80211_band band)
    687{
    688	struct ieee80211_local *local = sdata->local;
    689	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
    690
    691	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);
    692
    693	if (changed > 0) {
    694		ieee80211_recalc_min_chandef(sdata);
    695		rate_control_rate_update(local, sband, sta, changed);
    696	}
    697}
    698
    699void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
    700				     u16 vht_mask[NL80211_VHT_NSS_MAX])
    701{
    702	int i;
    703	u16 mask, cap = le16_to_cpu(vht_cap);
    704
    705	for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
    706		mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
    707		switch (mask) {
    708		case IEEE80211_VHT_MCS_SUPPORT_0_7:
    709			vht_mask[i] = 0x00FF;
    710			break;
    711		case IEEE80211_VHT_MCS_SUPPORT_0_8:
    712			vht_mask[i] = 0x01FF;
    713			break;
    714		case IEEE80211_VHT_MCS_SUPPORT_0_9:
    715			vht_mask[i] = 0x03FF;
    716			break;
    717		case IEEE80211_VHT_MCS_NOT_SUPPORTED:
    718		default:
    719			vht_mask[i] = 0;
    720			break;
    721		}
    722	}
    723}