[mirror_ubuntu-zesty-kernel.git] / net / mac80211 / vht.c

/*
 * VHT handling
 *
 * Portions of this file
 * Copyright(c) 2015 - 2016 Intel Deutschland GmbH
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/ieee80211.h>
#include <linux/export.h>
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "rate.h"


static void __check_vhtcap_disable(struct ieee80211_sub_if_data *sdata,
				   struct ieee80211_sta_vht_cap *vht_cap,
				   u32 flag)
{
	__le32 le_flag = cpu_to_le32(flag);

	if (sdata->u.mgd.vht_capa_mask.vht_cap_info & le_flag &&
	    !(sdata->u.mgd.vht_capa.vht_cap_info & le_flag))
		vht_cap->cap &= ~flag;
}

void ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data *sdata,
				      struct ieee80211_sta_vht_cap *vht_cap)
{
	int i;
	u16 rxmcs_mask, rxmcs_cap, rxmcs_n, txmcs_mask, txmcs_cap, txmcs_n;

	if (!vht_cap->vht_supported)
		return;

	if (sdata->vif.type != NL80211_IFTYPE_STATION)
		return;

	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_RXLDPC);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_SHORT_GI_80);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_SHORT_GI_160);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_TXSTBC);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN);
	__check_vhtcap_disable(sdata, vht_cap,
			       IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN);

	/* Allow user to decrease AMPDU length exponent */
	if (sdata->u.mgd.vht_capa_mask.vht_cap_info &
	    cpu_to_le32(IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK)) {
		u32 cap, n;

		n = le32_to_cpu(sdata->u.mgd.vht_capa.vht_cap_info) &
			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
		n >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
		cap = vht_cap->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
		cap >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;

		if (n < cap) {
			vht_cap->cap &=
				~IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
			vht_cap->cap |=
				n << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
		}
	}

	/* Allow the user to decrease MCSes */
	rxmcs_mask =
		le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.rx_mcs_map);
	rxmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.rx_mcs_map);
	rxmcs_n &= rxmcs_mask;
	rxmcs_cap = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);

	txmcs_mask =
		le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.tx_mcs_map);
	txmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.tx_mcs_map);
	txmcs_n &= txmcs_mask;
	txmcs_cap = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
	for (i = 0; i < 8; i++) {
		u8 m, n, c;

		m = (rxmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
		n = (rxmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
		c = (rxmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
			  n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
			rxmcs_cap &= ~(3 << 2*i);
			rxmcs_cap |= (rxmcs_n & (3 << 2*i));
		}

		m = (txmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
		n = (txmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
		c = (txmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
			  n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
			txmcs_cap &= ~(3 << 2*i);
			txmcs_cap |= (txmcs_n & (3 << 2*i));
		}
	}
	vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(rxmcs_cap);
	vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(txmcs_cap);
}

void
ieee80211_vht_cap_ie_to_sta_vht_cap(struct ieee80211_sub_if_data *sdata,
				    struct ieee80211_supported_band *sband,
				    const struct ieee80211_vht_cap *vht_cap_ie,
				    struct sta_info *sta)
{
	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
	struct ieee80211_sta_vht_cap own_cap;
	u32 cap_info, i;
	bool have_80mhz;

	memset(vht_cap, 0, sizeof(*vht_cap));

	if (!sta->sta.ht_cap.ht_supported)
		return;

	if (!vht_cap_ie || !sband->vht_cap.vht_supported)
		return;

	/* Allow VHT if at least one channel on the sband supports 80 MHz */
	have_80mhz = false;
	for (i = 0; i < sband->n_channels; i++) {
		if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
						IEEE80211_CHAN_NO_80MHZ))
			continue;

		have_80mhz = true;
		break;
	}

	if (!have_80mhz)
		return;

	/*
	 * A VHT STA must support 40 MHz, but if we verify that here
	 * then we break a few things - some APs (e.g. Netgear R6300v2
	 * and others based on the BCM4360 chipset) will unset this
	 * capability bit when operating in 20 MHz.
	 */

	vht_cap->vht_supported = true;

	own_cap = sband->vht_cap;
	/*
	 * If user has specified capability overrides, take care
	 * of that if the station we're setting up is the AP that
	 * we advertised a restricted capability set to. Override
	 * our own capabilities and then use those below.
	 */
	if (sdata->vif.type == NL80211_IFTYPE_STATION &&
	    !test_sta_flag(sta, WLAN_STA_TDLS_PEER))
		ieee80211_apply_vhtcap_overrides(sdata, &own_cap);

	/* take some capabilities as-is */
	cap_info = le32_to_cpu(vht_cap_ie->vht_cap_info);
	vht_cap->cap = cap_info;
	vht_cap->cap &= IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895 |
			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991 |
			IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 |
			IEEE80211_VHT_CAP_RXLDPC |
			IEEE80211_VHT_CAP_VHT_TXOP_PS |
			IEEE80211_VHT_CAP_HTC_VHT |
			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
			IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB |
			IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB |
			IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
			IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;

	/* and some based on our own capabilities */
	switch (own_cap.cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
		vht_cap->cap |= cap_info &
				IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
		break;
	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
		vht_cap->cap |= cap_info &
				IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
		break;
	default:
		/* nothing */
		break;
	}

	/* symmetric capabilities */
	vht_cap->cap |= cap_info & own_cap.cap &
			(IEEE80211_VHT_CAP_SHORT_GI_80 |
			 IEEE80211_VHT_CAP_SHORT_GI_160);

	/* remaining ones */
	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
		vht_cap->cap |= cap_info &
				(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
				 IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);

	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
		vht_cap->cap |= cap_info &
				(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
				 IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);

	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)
		vht_cap->cap |= cap_info &
				IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;

	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE)
		vht_cap->cap |= cap_info &
				IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE;

	if (own_cap.cap & IEEE80211_VHT_CAP_TXSTBC)
		vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_RXSTBC_MASK;

	if (own_cap.cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
		vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_TXSTBC;

	/* Copy peer MCS info, the driver might need them. */
	memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
	       sizeof(struct ieee80211_vht_mcs_info));

	/* but also restrict MCSes */
	for (i = 0; i < 8; i++) {
		u16 own_rx, own_tx, peer_rx, peer_tx;

		own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
		own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
		own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
		peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
		peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;

		if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
			if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
				peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
			else if (own_rx < peer_tx)
				peer_tx = own_rx;
		}

		if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
			if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
				peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
			else if (own_tx < peer_rx)
				peer_rx = own_tx;
		}

		vht_cap->vht_mcs.rx_mcs_map &=
			~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
		vht_cap->vht_mcs.rx_mcs_map |= cpu_to_le16(peer_rx << i * 2);

		vht_cap->vht_mcs.tx_mcs_map &=
			~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
		vht_cap->vht_mcs.tx_mcs_map |= cpu_to_le16(peer_tx << i * 2);
	}

	/*
	 * This is a workaround for VHT-enabled STAs which break the spec
	 * and have the VHT-MCS Rx map filled in with value 3 for all eight
	 * spacial streams, an example is AR9462.
	 *
	 * As per spec, in section 22.1.1 Introduction to the VHT PHY
	 * A VHT STA shall support at least single spactial stream VHT-MCSs
	 * 0 to 7 (transmit and receive) in all supported channel widths.
	 */
	if (vht_cap->vht_mcs.rx_mcs_map == cpu_to_le16(0xFFFF)) {
		vht_cap->vht_supported = false;
		sdata_info(sdata, "Ignoring VHT IE from %pM due to invalid rx_mcs_map\n",
			   sta->addr);
		return;
	}

	/* finally set up the bandwidth */
	switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
		break;
	default:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
	}

	sta->sta.bandwidth = ieee80211_sta_cur_vht_bw(sta);

	/* If HT IE reported 3839 bytes only, stay with that size. */
	if (sta->sta.max_amsdu_len == IEEE80211_MAX_MPDU_LEN_HT_3839)
		return;

	switch (vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK) {
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
		break;
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
		break;
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
	default:
		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
		break;
	}
}

enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
{
	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
	u32 cap_width;

	if (!vht_cap->vht_supported)
		return sta->sta.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
				IEEE80211_STA_RX_BW_40 :
				IEEE80211_STA_RX_BW_20;

	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;

	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
	    cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
		return IEEE80211_STA_RX_BW_160;

	return IEEE80211_STA_RX_BW_80;
}

enum nl80211_chan_width ieee80211_sta_cap_chan_bw(struct sta_info *sta)
{
	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
	u32 cap_width;

	if (!vht_cap->vht_supported) {
		if (!sta->sta.ht_cap.ht_supported)
			return NL80211_CHAN_WIDTH_20_NOHT;

		return sta->sta.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
				NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
	}

	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;

	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
		return NL80211_CHAN_WIDTH_160;
	else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
		return NL80211_CHAN_WIDTH_80P80;

	return NL80211_CHAN_WIDTH_80;
}

enum ieee80211_sta_rx_bandwidth
ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
{
	switch (width) {
	case NL80211_CHAN_WIDTH_20_NOHT:
	case NL80211_CHAN_WIDTH_20:
		return IEEE80211_STA_RX_BW_20;
	case NL80211_CHAN_WIDTH_40:
		return IEEE80211_STA_RX_BW_40;
	case NL80211_CHAN_WIDTH_80:
		return IEEE80211_STA_RX_BW_80;
	case NL80211_CHAN_WIDTH_160:
	case NL80211_CHAN_WIDTH_80P80:
		return IEEE80211_STA_RX_BW_160;
	default:
		WARN_ON_ONCE(1);
		return IEEE80211_STA_RX_BW_20;
	}
}

enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
{
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	enum ieee80211_sta_rx_bandwidth bw;
	enum nl80211_chan_width bss_width = sdata->vif.bss_conf.chandef.width;

	bw = ieee80211_sta_cap_rx_bw(sta);
	bw = min(bw, sta->cur_max_bandwidth);
	bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));

	return bw;
}

void ieee80211_sta_set_rx_nss(struct sta_info *sta)
{
	u8 ht_rx_nss = 0, vht_rx_nss = 0;

	/* if we received a notification already don't overwrite it */
	if (sta->sta.rx_nss)
		return;

	if (sta->sta.ht_cap.ht_supported) {
		if (sta->sta.ht_cap.mcs.rx_mask[0])
			ht_rx_nss++;
		if (sta->sta.ht_cap.mcs.rx_mask[1])
			ht_rx_nss++;
		if (sta->sta.ht_cap.mcs.rx_mask[2])
			ht_rx_nss++;
		if (sta->sta.ht_cap.mcs.rx_mask[3])
			ht_rx_nss++;
		/* FIXME: consider rx_highest? */
	}

	if (sta->sta.vht_cap.vht_supported) {
		int i;
		u16 rx_mcs_map;

		rx_mcs_map = le16_to_cpu(sta->sta.vht_cap.vht_mcs.rx_mcs_map);

		for (i = 7; i >= 0; i--) {
			u8 mcs = (rx_mcs_map >> (2 * i)) & 3;

			if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
				vht_rx_nss = i + 1;
				break;
			}
		}
		/* FIXME: consider rx_highest? */
	}

	ht_rx_nss = max(ht_rx_nss, vht_rx_nss);
	sta->sta.rx_nss = max_t(u8, 1, ht_rx_nss);
}

u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
				  struct sta_info *sta, u8 opmode,
				  enum nl80211_band band)
{
	struct ieee80211_local *local = sdata->local;
	struct ieee80211_supported_band *sband;
	enum ieee80211_sta_rx_bandwidth new_bw;
	u32 changed = 0;
	u8 nss;

	sband = local->hw.wiphy->bands[band];

	/* ignore - no support for BF yet */
	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
		return 0;

	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
	nss += 1;

	if (sta->sta.rx_nss != nss) {
		sta->sta.rx_nss = nss;
		changed |= IEEE80211_RC_NSS_CHANGED;
	}

	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
		break;
	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
		break;
	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
		break;
	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
		break;
	}

	new_bw = ieee80211_sta_cur_vht_bw(sta);
	if (new_bw != sta->sta.bandwidth) {
		sta->sta.bandwidth = new_bw;
		changed |= IEEE80211_RC_BW_CHANGED;
	}

	return changed;
}

void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
				 struct ieee80211_mgmt *mgmt)
{
	struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;

	if (!sdata->vif.mu_mimo_owner)
		return;

	if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
		    bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
	    !memcmp(mgmt->u.action.u.vht_group_notif.membership,
		    bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
		return;

	memcpy(bss_conf->mu_group.membership,
	       mgmt->u.action.u.vht_group_notif.membership,
	       WLAN_MEMBERSHIP_LEN);
	memcpy(bss_conf->mu_group.position,
	       mgmt->u.action.u.vht_group_notif.position,
	       WLAN_USER_POSITION_LEN);

	ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_MU_GROUPS);
}

void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
				const u8 *membership, const u8 *position)
{
	struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;

	if (WARN_ON_ONCE(!vif->mu_mimo_owner))
		return;

	memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
	memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
}
EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);

void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
				 struct sta_info *sta, u8 opmode,
				 enum nl80211_band band)
{
	struct ieee80211_local *local = sdata->local;
	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];

	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);

	if (changed > 0) {
		ieee80211_recalc_min_chandef(sdata);
		rate_control_rate_update(local, sband, sta, changed);
	}
}

void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
				     u16 vht_mask[NL80211_VHT_NSS_MAX])
{
	int i;
	u16 mask, cap = le16_to_cpu(vht_cap);

	for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
		mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
		switch (mask) {
		case IEEE80211_VHT_MCS_SUPPORT_0_7:
			vht_mask[i] = 0x00FF;
			break;
		case IEEE80211_VHT_MCS_SUPPORT_0_8:
			vht_mask[i] = 0x01FF;
			break;
		case IEEE80211_VHT_MCS_SUPPORT_0_9:
			vht_mask[i] = 0x03FF;
			break;
		case IEEE80211_VHT_MCS_NOT_SUPPORTED:
		default:
			vht_mask[i] = 0;
			break;
		}
	}
}
Commit	Line	Data
818255ea MP	1	/*
	2	* VHT handling
	3	*
23a1f8d4	4	* Portions of this file
65554d07	5	* Copyright(c) 2015 - 2016 Intel Deutschland GmbH
23a1f8d4	6	*
818255ea MP	7	* This program is free software; you can redistribute it and/or modify
	8	* it under the terms of the GNU General Public License version 2 as
	9	* published by the Free Software Foundation.
	10	*/
	11
	12	#include <linux/ieee80211.h>
	13	#include <linux/export.h>
	14	#include <net/mac80211.h>
	15	#include "ieee80211_i.h"
0af83d3d	16	#include "rate.h"
818255ea MP	17
818255ea MP	18
dd5ecfea JB	19	static void __check_vhtcap_disable(struct ieee80211_sub_if_data *sdata,
	20	struct ieee80211_sta_vht_cap *vht_cap,
	21	u32 flag)
	22	{
	23	__le32 le_flag = cpu_to_le32(flag);
	24
	25	if (sdata->u.mgd.vht_capa_mask.vht_cap_info & le_flag &&
	26	!(sdata->u.mgd.vht_capa.vht_cap_info & le_flag))
	27	vht_cap->cap &= ~flag;
	28	}
	29
	30	void ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data *sdata,
	31	struct ieee80211_sta_vht_cap *vht_cap)
	32	{
	33	int i;
	34	u16 rxmcs_mask, rxmcs_cap, rxmcs_n, txmcs_mask, txmcs_cap, txmcs_n;
	35
	36	if (!vht_cap->vht_supported)
	37	return;
	38
	39	if (sdata->vif.type != NL80211_IFTYPE_STATION)
	40	return;
	41
	42	__check_vhtcap_disable(sdata, vht_cap,
	43	IEEE80211_VHT_CAP_RXLDPC);
	44	__check_vhtcap_disable(sdata, vht_cap,
	45	IEEE80211_VHT_CAP_SHORT_GI_80);
	46	__check_vhtcap_disable(sdata, vht_cap,
	47	IEEE80211_VHT_CAP_SHORT_GI_160);
	48	__check_vhtcap_disable(sdata, vht_cap,
	49	IEEE80211_VHT_CAP_TXSTBC);
	50	__check_vhtcap_disable(sdata, vht_cap,
	51	IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
	52	__check_vhtcap_disable(sdata, vht_cap,
	53	IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
	54	__check_vhtcap_disable(sdata, vht_cap,
	55	IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN);
	56	__check_vhtcap_disable(sdata, vht_cap,
	57	IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN);
	58
	59	/* Allow user to decrease AMPDU length exponent */
	60	if (sdata->u.mgd.vht_capa_mask.vht_cap_info &
	61	cpu_to_le32(IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK)) {
	62	u32 cap, n;
	63
	64	n = le32_to_cpu(sdata->u.mgd.vht_capa.vht_cap_info) &
	65	IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	66	n >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	67	cap = vht_cap->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	68	cap >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	69
	70	if (n < cap) {
	71	vht_cap->cap &=
	72	~IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	73	vht_cap->cap \|=
	74	n << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	75	}
	76	}
	77
	78	/* Allow the user to decrease MCSes */
	79	rxmcs_mask =
	80	le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.rx_mcs_map);
	81	rxmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.rx_mcs_map);
	82	rxmcs_n &= rxmcs_mask;
83	rxmcs_cap = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
84
85	txmcs_mask =
86	le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.tx_mcs_map);
87	txmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.tx_mcs_map);
88	txmcs_n &= txmcs_mask;
89	txmcs_cap = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
90	for (i = 0; i < 8; i++) {
91	u8 m, n, c;
92
93	m = (rxmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
94	n = (rxmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
95	c = (rxmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
96
97	if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) \|\|
98	n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
99	rxmcs_cap &= ~(3 << 2*i);
100	rxmcs_cap \|= (rxmcs_n & (3 << 2*i));
101	}
102
103	m = (txmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
104	n = (txmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
105	c = (txmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
106
107	if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) \|\|
108	n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
109	txmcs_cap &= ~(3 << 2*i);
110	txmcs_cap \|= (txmcs_n & (3 << 2*i));
111	}
112	}
113	vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(rxmcs_cap);
114	vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(txmcs_cap);
115	}
116
4a3cb702 JB	117	void
	118	ieee80211_vht_cap_ie_to_sta_vht_cap(struct ieee80211_sub_if_data *sdata,
	119	struct ieee80211_supported_band *sband,
	120	const struct ieee80211_vht_cap *vht_cap_ie,
	121	struct sta_info *sta)
818255ea	122	{
4a34215e	123	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
55d942f4 JB	124	struct ieee80211_sta_vht_cap own_cap;
55d942f4 JB	125	u32 cap_info, i;
52a45f38	126	bool have_80mhz;
818255ea MP	127
	128	memset(vht_cap, 0, sizeof(*vht_cap));
	129
4a34215e JB	130	if (!sta->sta.ht_cap.ht_supported)
	131	return;
	132
818255ea MP	133	if (!vht_cap_ie \|\| !sband->vht_cap.vht_supported)
	134	return;
	135
52a45f38 AN	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
4a817aa7 JB	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	*/
e1a0c6b3	156
818255ea MP	157	vht_cap->vht_supported = true;
818255ea MP	158
55d942f4 JB	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(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_MAX_MPDU_LENGTH_3895 \|
	174	IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991 \|
	175	IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 \|
	176	IEEE80211_VHT_CAP_RXLDPC \|
	177	IEEE80211_VHT_CAP_VHT_TXOP_PS \|
	178	IEEE80211_VHT_CAP_HTC_VHT \|
	179	IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK \|
	180	IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB \|
	181	IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB \|
	182	IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN \|
	183	IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;
	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 */
fbdd90ea	206	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
55d942f4 JB	207	vht_cap->cap \|= cap_info &
55d942f4 JB	208	(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE \|
fbdd90ea	209	IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
55d942f4 JB	210
	211	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
	212	vht_cap->cap \|= cap_info &
5eb7906b	213	(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE \|
fbdd90ea	214	IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
55d942f4 JB	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;
818255ea MP	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));
e1a0c6b3	233
55d942f4 JB	234	/* but also restrict MCSes */
	235	for (i = 0; i < 8; i++) {
	236	u16 own_rx, own_tx, peer_rx, peer_tx;
	237
	238	own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
	239	own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	240
	241	own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
	242	own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	243
	244	peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
	245	peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	246
	247	peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
	248	peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	249
	250	if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	251	if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
	252	peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
	253	else if (own_rx < peer_tx)
	254	peer_tx = own_rx;
	255	}
	256
	257	if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	258	if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
	259	peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
	260	else if (own_tx < peer_rx)
	261	peer_rx = own_tx;
	262	}
	263
	264	vht_cap->vht_mcs.rx_mcs_map &=
	265	~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
	266	vht_cap->vht_mcs.rx_mcs_map \|= cpu_to_le16(peer_rx << i * 2);
	267
	268	vht_cap->vht_mcs.tx_mcs_map &=
	269	~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
	270	vht_cap->vht_mcs.tx_mcs_map \|= cpu_to_le16(peer_tx << i * 2);
	271	}
	272
c8eaf347 FM	273	/*
	274	* This is a workaround for VHT-enabled STAs which break the spec
	275	* and have the VHT-MCS Rx map filled in with value 3 for all eight
	276	* spacial streams, an example is AR9462.
	277	*
	278	* As per spec, in section 22.1.1 Introduction to the VHT PHY
	279	* A VHT STA shall support at least single spactial stream VHT-MCSs
	280	* 0 to 7 (transmit and receive) in all supported channel widths.
	281	*/
	282	if (vht_cap->vht_mcs.rx_mcs_map == cpu_to_le16(0xFFFF)) {
	283	vht_cap->vht_supported = false;
	284	sdata_info(sdata, "Ignoring VHT IE from %pM due to invalid rx_mcs_map\n",
	285	sta->addr);
	286	return;
	287	}
	288
55d942f4	289	/* finally set up the bandwidth */
0af83d3d JB	290	switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
	291	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
	292	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
	293	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
	294	break;
	295	default:
	296	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
	297	}
	298
e1a0c6b3	299	sta->sta.bandwidth = ieee80211_sta_cur_vht_bw(sta);
506bcfa8 EG	300
	301	/* If HT IE reported 3839 bytes only, stay with that size. */
	302	if (sta->sta.max_amsdu_len == IEEE80211_MAX_MPDU_LEN_HT_3839)
	303	return;
	304
	305	switch (vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK) {
	306	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
	307	sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
	308	break;
	309	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
	310	sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
	311	break;
	312	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
	313	default:
	314	sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
	315	break;
	316	}
e1a0c6b3 JB	317	}
e1a0c6b3 JB	318
1c45c5ce	319	enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
e1a0c6b3	320	{
1c45c5ce EP	321	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
1c45c5ce EP	322	u32 cap_width;
e1a0c6b3	323
1c45c5ce EP	324	if (!vht_cap->vht_supported)
	325	return sta->sta.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
	326	IEEE80211_STA_RX_BW_40 :
	327	IEEE80211_STA_RX_BW_20;
e1a0c6b3	328
1c45c5ce EP	329	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
	330
	331	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ \|\|
	332	cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
	333	return IEEE80211_STA_RX_BW_160;
	334
	335	return IEEE80211_STA_RX_BW_80;
	336	}
	337
59021c67 AN	338	enum nl80211_chan_width ieee80211_sta_cap_chan_bw(struct sta_info *sta)
	339	{
	340	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
	341	u32 cap_width;
	342
	343	if (!vht_cap->vht_supported) {
	344	if (!sta->sta.ht_cap.ht_supported)
	345	return NL80211_CHAN_WIDTH_20_NOHT;
	346
	347	return sta->sta.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
	348	NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
	349	}
	350
	351	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
	352
	353	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
	354	return NL80211_CHAN_WIDTH_160;
	355	else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
	356	return NL80211_CHAN_WIDTH_80P80;
	357
	358	return NL80211_CHAN_WIDTH_80;
	359	}
	360
	361	enum ieee80211_sta_rx_bandwidth
1c45c5ce EP	362	ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
	363	{
	364	switch (width) {
e1a0c6b3 JB	365	case NL80211_CHAN_WIDTH_20_NOHT:
e1a0c6b3 JB	366	case NL80211_CHAN_WIDTH_20:
1c45c5ce	367	return IEEE80211_STA_RX_BW_20;
e1a0c6b3	368	case NL80211_CHAN_WIDTH_40:
1c45c5ce EP	369	return IEEE80211_STA_RX_BW_40;
	370	case NL80211_CHAN_WIDTH_80:
	371	return IEEE80211_STA_RX_BW_80;
e1a0c6b3	372	case NL80211_CHAN_WIDTH_160:
e1a0c6b3	373	case NL80211_CHAN_WIDTH_80P80:
1c45c5ce EP	374	return IEEE80211_STA_RX_BW_160;
	375	default:
	376	WARN_ON_ONCE(1);
	377	return IEEE80211_STA_RX_BW_20;
e1a0c6b3	378	}
1c45c5ce EP	379	}
	380
	381	enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
	382	{
	383	struct ieee80211_sub_if_data *sdata = sta->sdata;
	384	enum ieee80211_sta_rx_bandwidth bw;
b98fb44f	385	enum nl80211_chan_width bss_width = sdata->vif.bss_conf.chandef.width;
1c45c5ce	386
b98fb44f	387	bw = ieee80211_sta_cap_rx_bw(sta);
1c45c5ce	388	bw = min(bw, sta->cur_max_bandwidth);
59021c67	389	bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
b98fb44f	390
0af83d3d	391	return bw;
818255ea	392	}
8921d04e JB	393
	394	void ieee80211_sta_set_rx_nss(struct sta_info *sta)
	395	{
	396	u8 ht_rx_nss = 0, vht_rx_nss = 0;
	397
	398	/* if we received a notification already don't overwrite it */
	399	if (sta->sta.rx_nss)
	400	return;
	401
	402	if (sta->sta.ht_cap.ht_supported) {
	403	if (sta->sta.ht_cap.mcs.rx_mask[0])
	404	ht_rx_nss++;
	405	if (sta->sta.ht_cap.mcs.rx_mask[1])
	406	ht_rx_nss++;
	407	if (sta->sta.ht_cap.mcs.rx_mask[2])
	408	ht_rx_nss++;
	409	if (sta->sta.ht_cap.mcs.rx_mask[3])
	410	ht_rx_nss++;
	411	/* FIXME: consider rx_highest? */
	412	}
	413
	414	if (sta->sta.vht_cap.vht_supported) {
	415	int i;
	416	u16 rx_mcs_map;
	417
	418	rx_mcs_map = le16_to_cpu(sta->sta.vht_cap.vht_mcs.rx_mcs_map);
	419
	420	for (i = 7; i >= 0; i--) {
	421	u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
	422
	423	if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	424	vht_rx_nss = i + 1;
	425	break;
	426	}
	427	}
	428	/* FIXME: consider rx_highest? */
	429	}
	430
	431	ht_rx_nss = max(ht_rx_nss, vht_rx_nss);
	432	sta->sta.rx_nss = max_t(u8, 1, ht_rx_nss);
	433	}
0af83d3d	434
b1bce14a MK	435	u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
b1bce14a MK	436	struct sta_info *sta, u8 opmode,
57fbcce3	437	enum nl80211_band band)
0af83d3d JB	438	{
	439	struct ieee80211_local *local = sdata->local;
	440	struct ieee80211_supported_band *sband;
	441	enum ieee80211_sta_rx_bandwidth new_bw;
	442	u32 changed = 0;
	443	u8 nss;
	444
	445	sband = local->hw.wiphy->bands[band];
	446
	447	/* ignore - no support for BF yet */
	448	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
b1bce14a	449	return 0;
0af83d3d JB	450
	451	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
	452	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
	453	nss += 1;
	454
	455	if (sta->sta.rx_nss != nss) {
	456	sta->sta.rx_nss = nss;
	457	changed \|= IEEE80211_RC_NSS_CHANGED;
	458	}
	459
	460	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
	461	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
	462	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
	463	break;
	464	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
	465	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
	466	break;
	467	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
	468	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
	469	break;
	470	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
	471	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
	472	break;
	473	}
	474
	475	new_bw = ieee80211_sta_cur_vht_bw(sta);
	476	if (new_bw != sta->sta.bandwidth) {
	477	sta->sta.bandwidth = new_bw;
38745c74	478	changed \|= IEEE80211_RC_BW_CHANGED;
0af83d3d JB	479	}
0af83d3d JB	480
b1bce14a MK	481	return changed;
	482	}
	483
23a1f8d4 SS	484	void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
	485	struct ieee80211_mgmt *mgmt)
	486	{
	487	struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
	488
b5a33d52	489	if (!sdata->vif.mu_mimo_owner)
23a1f8d4 SS	490	return;
	491
	492	if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
	493	bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
	494	!memcmp(mgmt->u.action.u.vht_group_notif.membership,
	495	bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
	496	return;
	497
0241fa19 SS	498	memcpy(bss_conf->mu_group.membership,
	499	mgmt->u.action.u.vht_group_notif.membership,
	500	WLAN_MEMBERSHIP_LEN);
	501	memcpy(bss_conf->mu_group.position,
	502	mgmt->u.action.u.vht_group_notif.position,
	503	WLAN_USER_POSITION_LEN);
23a1f8d4 SS	504
	505	ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_MU_GROUPS);
	506	}
	507
65554d07 SS	508	void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
	509	const u8 membership, const u8 position)
	510	{
b5a33d52	511	struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
65554d07	512
b5a33d52	513	if (WARN_ON_ONCE(!vif->mu_mimo_owner))
65554d07 SS	514	return;
	515
	516	memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
	517	memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
	518	}
	519	EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
	520
b1bce14a MK	521	void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
b1bce14a MK	522	struct sta_info *sta, u8 opmode,
57fbcce3	523	enum nl80211_band band)
b1bce14a MK	524	{
	525	struct ieee80211_local *local = sdata->local;
	526	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
	527
cf1e05c6	528	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);
b1bce14a	529
d2941df8 JB	530	if (changed > 0) {
d2941df8 JB	531	ieee80211_recalc_min_chandef(sdata);
0af83d3d	532	rate_control_rate_update(local, sband, sta, changed);
d2941df8	533	}
0af83d3d	534	}
b119ad6e LB	535
	536	void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
	537	u16 vht_mask[NL80211_VHT_NSS_MAX])
	538	{
	539	int i;
	540	u16 mask, cap = le16_to_cpu(vht_cap);
	541
	542	for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
	543	mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	544	switch (mask) {
	545	case IEEE80211_VHT_MCS_SUPPORT_0_7:
	546	vht_mask[i] = 0x00FF;
	547	break;
	548	case IEEE80211_VHT_MCS_SUPPORT_0_8:
	549	vht_mask[i] = 0x01FF;
	550	break;
	551	case IEEE80211_VHT_MCS_SUPPORT_0_9:
	552	vht_mask[i] = 0x03FF;
	553	break;
	554	case IEEE80211_VHT_MCS_NOT_SUPPORTED:
	555	default:
	556	vht_mask[i] = 0;
	557	break;
	558	}
	559	}
	560	}