[deliverable/linux.git] / net / mac80211 / vht.c

/*
 * VHT handling
 *
 * 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;

	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;

	/* don't support VHT for TDLS peers for now */
	if (test_sta_flag(sta, WLAN_STA_TDLS_PEER))
		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);
	}

	/* 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);
}

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;
}

static 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;

	bw = ieee80211_chan_width_to_rx_bw(sdata->vif.bss_conf.chandef.width);
	bw = min(bw, ieee80211_sta_cap_rx_bw(sta));
	bw = min(bw, sta->cur_max_bandwidth);

	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 ieee80211_band band, bool nss_only)
{
	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;
	}

	if (nss_only)
		return 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_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
				 struct sta_info *sta, u8 opmode,
				 enum ieee80211_band band, bool nss_only)
{
	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, nss_only);

	if (changed > 0)
		rate_control_rate_update(local, sband, sta, changed);
}
Commit	Line	Data
818255ea MP	1	/*
	2	* VHT handling
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of the GNU General Public License version 2 as
	6	* published by the Free Software Foundation.
	7	*/
	8
	9	#include <linux/ieee80211.h>
	10	#include <linux/export.h>
	11	#include <net/mac80211.h>
	12	#include "ieee80211_i.h"
0af83d3d	13	#include "rate.h"
818255ea MP	14
818255ea MP	15
dd5ecfea JB	16	static void __check_vhtcap_disable(struct ieee80211_sub_if_data *sdata,
	17	struct ieee80211_sta_vht_cap *vht_cap,
	18	u32 flag)
	19	{
	20	__le32 le_flag = cpu_to_le32(flag);
	21
	22	if (sdata->u.mgd.vht_capa_mask.vht_cap_info & le_flag &&
	23	!(sdata->u.mgd.vht_capa.vht_cap_info & le_flag))
	24	vht_cap->cap &= ~flag;
	25	}
	26
	27	void ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data *sdata,
	28	struct ieee80211_sta_vht_cap *vht_cap)
	29	{
	30	int i;
	31	u16 rxmcs_mask, rxmcs_cap, rxmcs_n, txmcs_mask, txmcs_cap, txmcs_n;
	32
	33	if (!vht_cap->vht_supported)
	34	return;
	35
	36	if (sdata->vif.type != NL80211_IFTYPE_STATION)
	37	return;
	38
	39	__check_vhtcap_disable(sdata, vht_cap,
	40	IEEE80211_VHT_CAP_RXLDPC);
	41	__check_vhtcap_disable(sdata, vht_cap,
	42	IEEE80211_VHT_CAP_SHORT_GI_80);
	43	__check_vhtcap_disable(sdata, vht_cap,
	44	IEEE80211_VHT_CAP_SHORT_GI_160);
	45	__check_vhtcap_disable(sdata, vht_cap,
	46	IEEE80211_VHT_CAP_TXSTBC);
	47	__check_vhtcap_disable(sdata, vht_cap,
	48	IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
	49	__check_vhtcap_disable(sdata, vht_cap,
	50	IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
	51	__check_vhtcap_disable(sdata, vht_cap,
	52	IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN);
	53	__check_vhtcap_disable(sdata, vht_cap,
	54	IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN);
	55
	56	/* Allow user to decrease AMPDU length exponent */
	57	if (sdata->u.mgd.vht_capa_mask.vht_cap_info &
	58	cpu_to_le32(IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK)) {
	59	u32 cap, n;
	60
	61	n = le32_to_cpu(sdata->u.mgd.vht_capa.vht_cap_info) &
	62	IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	63	n >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	64	cap = vht_cap->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	65	cap >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	66
	67	if (n < cap) {
	68	vht_cap->cap &=
	69	~IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
	70	vht_cap->cap \|=
	71	n << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
	72	}
	73	}
	74
	75	/* Allow the user to decrease MCSes */
	76	rxmcs_mask =
	77	le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.rx_mcs_map);
	78	rxmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.rx_mcs_map);
	79	rxmcs_n &= rxmcs_mask;
80	rxmcs_cap = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
81
82	txmcs_mask =
83	le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.tx_mcs_map);
84	txmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.tx_mcs_map);
85	txmcs_n &= txmcs_mask;
86	txmcs_cap = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
87	for (i = 0; i < 8; i++) {
88	u8 m, n, c;
89
90	m = (rxmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
91	n = (rxmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
92	c = (rxmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
93
94	if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) \|\|
95	n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
96	rxmcs_cap &= ~(3 << 2*i);
97	rxmcs_cap \|= (rxmcs_n & (3 << 2*i));
98	}
99
100	m = (txmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
101	n = (txmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
102	c = (txmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
103
104	if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) \|\|
105	n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
106	txmcs_cap &= ~(3 << 2*i);
107	txmcs_cap \|= (txmcs_n & (3 << 2*i));
108	}
109	}
110	vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(rxmcs_cap);
111	vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(txmcs_cap);
112	}
113
4a3cb702 JB	114	void
	115	ieee80211_vht_cap_ie_to_sta_vht_cap(struct ieee80211_sub_if_data *sdata,
	116	struct ieee80211_supported_band *sband,
	117	const struct ieee80211_vht_cap *vht_cap_ie,
	118	struct sta_info *sta)
818255ea	119	{
4a34215e	120	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
55d942f4 JB	121	struct ieee80211_sta_vht_cap own_cap;
55d942f4 JB	122	u32 cap_info, i;
818255ea MP	123
	124	memset(vht_cap, 0, sizeof(*vht_cap));
	125
4a34215e JB	126	if (!sta->sta.ht_cap.ht_supported)
	127	return;
	128
818255ea MP	129	if (!vht_cap_ie \|\| !sband->vht_cap.vht_supported)
	130	return;
	131
bed766bd AN	132	/* don't support VHT for TDLS peers for now */
	133	if (test_sta_flag(sta, WLAN_STA_TDLS_PEER))
	134	return;
	135
4a817aa7 JB	136	/*
	137	* A VHT STA must support 40 MHz, but if we verify that here
	138	* then we break a few things - some APs (e.g. Netgear R6300v2
	139	* and others based on the BCM4360 chipset) will unset this
	140	* capability bit when operating in 20 MHz.
	141	*/
e1a0c6b3	142
818255ea MP	143	vht_cap->vht_supported = true;
818255ea MP	144
55d942f4 JB	145	own_cap = sband->vht_cap;
	146	/*
	147	* If user has specified capability overrides, take care
	148	* of that if the station we're setting up is the AP that
	149	* we advertised a restricted capability set to. Override
	150	* our own capabilities and then use those below.
	151	*/
	152	if (sdata->vif.type == NL80211_IFTYPE_STATION &&
	153	!test_sta_flag(sta, WLAN_STA_TDLS_PEER))
	154	ieee80211_apply_vhtcap_overrides(sdata, &own_cap);
	155
	156	/* take some capabilities as-is */
	157	cap_info = le32_to_cpu(vht_cap_ie->vht_cap_info);
	158	vht_cap->cap = cap_info;
	159	vht_cap->cap &= IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895 \|
	160	IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991 \|
	161	IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454 \|
	162	IEEE80211_VHT_CAP_RXLDPC \|
	163	IEEE80211_VHT_CAP_VHT_TXOP_PS \|
	164	IEEE80211_VHT_CAP_HTC_VHT \|
	165	IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK \|
	166	IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB \|
	167	IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB \|
	168	IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN \|
	169	IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;
	170
	171	/* and some based on our own capabilities */
	172	switch (own_cap.cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
	173	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
	174	vht_cap->cap \|= cap_info &
	175	IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
	176	break;
	177	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
	178	vht_cap->cap \|= cap_info &
	179	IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
	180	break;
	181	default:
	182	/* nothing */
	183	break;
	184	}
	185
	186	/* symmetric capabilities */
	187	vht_cap->cap \|= cap_info & own_cap.cap &
	188	(IEEE80211_VHT_CAP_SHORT_GI_80 \|
	189	IEEE80211_VHT_CAP_SHORT_GI_160);
	190
	191	/* remaining ones */
fbdd90ea	192	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
55d942f4 JB	193	vht_cap->cap \|= cap_info &
55d942f4 JB	194	(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE \|
fbdd90ea	195	IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
55d942f4 JB	196
	197	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
	198	vht_cap->cap \|= cap_info &
5eb7906b	199	(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE \|
fbdd90ea	200	IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
55d942f4 JB	201
	202	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)
	203	vht_cap->cap \|= cap_info &
	204	IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
	205
	206	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE)
	207	vht_cap->cap \|= cap_info &
	208	IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE;
	209
	210	if (own_cap.cap & IEEE80211_VHT_CAP_TXSTBC)
	211	vht_cap->cap \|= cap_info & IEEE80211_VHT_CAP_RXSTBC_MASK;
	212
	213	if (own_cap.cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
	214	vht_cap->cap \|= cap_info & IEEE80211_VHT_CAP_TXSTBC;
818255ea MP	215
	216	/* Copy peer MCS info, the driver might need them. */
	217	memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
	218	sizeof(struct ieee80211_vht_mcs_info));
e1a0c6b3	219
55d942f4 JB	220	/* but also restrict MCSes */
	221	for (i = 0; i < 8; i++) {
	222	u16 own_rx, own_tx, peer_rx, peer_tx;
	223
	224	own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
	225	own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	226
	227	own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
	228	own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	229
	230	peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
	231	peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	232
	233	peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
	234	peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
	235
	236	if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	237	if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
	238	peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
	239	else if (own_rx < peer_tx)
	240	peer_tx = own_rx;
	241	}
	242
	243	if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	244	if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
	245	peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
	246	else if (own_tx < peer_rx)
	247	peer_rx = own_tx;
	248	}
	249
	250	vht_cap->vht_mcs.rx_mcs_map &=
	251	~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
	252	vht_cap->vht_mcs.rx_mcs_map \|= cpu_to_le16(peer_rx << i * 2);
	253
	254	vht_cap->vht_mcs.tx_mcs_map &=
	255	~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
	256	vht_cap->vht_mcs.tx_mcs_map \|= cpu_to_le16(peer_tx << i * 2);
	257	}
	258
	259	/* finally set up the bandwidth */
0af83d3d JB	260	switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
	261	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
	262	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
	263	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
	264	break;
	265	default:
	266	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
	267	}
	268
e1a0c6b3 JB	269	sta->sta.bandwidth = ieee80211_sta_cur_vht_bw(sta);
	270	}
	271
1c45c5ce	272	enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
e1a0c6b3	273	{
1c45c5ce EP	274	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
1c45c5ce EP	275	u32 cap_width;
e1a0c6b3	276
1c45c5ce EP	277	if (!vht_cap->vht_supported)
	278	return sta->sta.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
	279	IEEE80211_STA_RX_BW_40 :
	280	IEEE80211_STA_RX_BW_20;
e1a0c6b3	281
1c45c5ce EP	282	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
	283
	284	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ \|\|
	285	cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
	286	return IEEE80211_STA_RX_BW_160;
	287
	288	return IEEE80211_STA_RX_BW_80;
	289	}
	290
	291	static enum ieee80211_sta_rx_bandwidth
	292	ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
	293	{
	294	switch (width) {
e1a0c6b3 JB	295	case NL80211_CHAN_WIDTH_20_NOHT:
e1a0c6b3 JB	296	case NL80211_CHAN_WIDTH_20:
1c45c5ce	297	return IEEE80211_STA_RX_BW_20;
e1a0c6b3	298	case NL80211_CHAN_WIDTH_40:
1c45c5ce EP	299	return IEEE80211_STA_RX_BW_40;
	300	case NL80211_CHAN_WIDTH_80:
	301	return IEEE80211_STA_RX_BW_80;
e1a0c6b3	302	case NL80211_CHAN_WIDTH_160:
e1a0c6b3	303	case NL80211_CHAN_WIDTH_80P80:
1c45c5ce EP	304	return IEEE80211_STA_RX_BW_160;
	305	default:
	306	WARN_ON_ONCE(1);
	307	return IEEE80211_STA_RX_BW_20;
e1a0c6b3	308	}
1c45c5ce EP	309	}
	310
	311	enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
	312	{
	313	struct ieee80211_sub_if_data *sdata = sta->sdata;
	314	enum ieee80211_sta_rx_bandwidth bw;
	315
	316	bw = ieee80211_chan_width_to_rx_bw(sdata->vif.bss_conf.chandef.width);
	317	bw = min(bw, ieee80211_sta_cap_rx_bw(sta));
	318	bw = min(bw, sta->cur_max_bandwidth);
0af83d3d	319
0af83d3d	320	return bw;
818255ea	321	}
8921d04e JB	322
	323	void ieee80211_sta_set_rx_nss(struct sta_info *sta)
	324	{
	325	u8 ht_rx_nss = 0, vht_rx_nss = 0;
	326
	327	/* if we received a notification already don't overwrite it */
	328	if (sta->sta.rx_nss)
	329	return;
	330
	331	if (sta->sta.ht_cap.ht_supported) {
	332	if (sta->sta.ht_cap.mcs.rx_mask[0])
	333	ht_rx_nss++;
	334	if (sta->sta.ht_cap.mcs.rx_mask[1])
	335	ht_rx_nss++;
	336	if (sta->sta.ht_cap.mcs.rx_mask[2])
	337	ht_rx_nss++;
	338	if (sta->sta.ht_cap.mcs.rx_mask[3])
	339	ht_rx_nss++;
	340	/* FIXME: consider rx_highest? */
	341	}
	342
	343	if (sta->sta.vht_cap.vht_supported) {
	344	int i;
	345	u16 rx_mcs_map;
	346
	347	rx_mcs_map = le16_to_cpu(sta->sta.vht_cap.vht_mcs.rx_mcs_map);
	348
	349	for (i = 7; i >= 0; i--) {
	350	u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
	351
	352	if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
	353	vht_rx_nss = i + 1;
	354	break;
	355	}
	356	}
	357	/* FIXME: consider rx_highest? */
	358	}
	359
	360	ht_rx_nss = max(ht_rx_nss, vht_rx_nss);
	361	sta->sta.rx_nss = max_t(u8, 1, ht_rx_nss);
	362	}
0af83d3d	363
b1bce14a MK	364	u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
	365	struct sta_info *sta, u8 opmode,
	366	enum ieee80211_band band, bool nss_only)
0af83d3d JB	367	{
	368	struct ieee80211_local *local = sdata->local;
	369	struct ieee80211_supported_band *sband;
	370	enum ieee80211_sta_rx_bandwidth new_bw;
	371	u32 changed = 0;
	372	u8 nss;
	373
	374	sband = local->hw.wiphy->bands[band];
	375
	376	/* ignore - no support for BF yet */
	377	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
b1bce14a	378	return 0;
0af83d3d JB	379
	380	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
	381	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
	382	nss += 1;
	383
	384	if (sta->sta.rx_nss != nss) {
	385	sta->sta.rx_nss = nss;
	386	changed \|= IEEE80211_RC_NSS_CHANGED;
	387	}
	388
bee7f586	389	if (nss_only)
b1bce14a	390	return changed;
bee7f586	391
0af83d3d JB	392	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
	393	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
	394	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
	395	break;
	396	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
	397	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
	398	break;
	399	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
	400	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
	401	break;
	402	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
	403	sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
	404	break;
	405	}
	406
	407	new_bw = ieee80211_sta_cur_vht_bw(sta);
	408	if (new_bw != sta->sta.bandwidth) {
	409	sta->sta.bandwidth = new_bw;
38745c74	410	changed \|= IEEE80211_RC_BW_CHANGED;
0af83d3d JB	411	}
0af83d3d JB	412
b1bce14a MK	413	return changed;
	414	}
	415
	416	void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
	417	struct sta_info *sta, u8 opmode,
	418	enum ieee80211_band band, bool nss_only)
	419	{
	420	struct ieee80211_local *local = sdata->local;
	421	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
	422
	423	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode,
	424	band, nss_only);
	425
	426	if (changed > 0)
0af83d3d JB	427	rate_control_rate_update(local, sband, sta, changed);
0af83d3d JB	428	}