net/mac80211/he.c - kernel/common - Git at Google

 // SPDX-License-Identifier: GPL-2.0-only
 /*
  * HE handling
  *
  * Copyright(c) 2017 Intel Deutschland GmbH
  * Copyright(c) 2019 - 2024 Intel Corporation
  */

 #include "ieee80211_i.h"
 #include "rate.h"

 static void
 ieee80211_update_from_he_6ghz_capa(const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
 				   struct link_sta_info *link_sta)
 {
 	struct sta_info *sta = link_sta->sta;
 	enum ieee80211_smps_mode smps_mode;

 	if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
 	    sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
 		switch (le16_get_bits(he_6ghz_capa->capa,
 				      IEEE80211_HE_6GHZ_CAP_SM_PS)) {
 		case WLAN_HT_CAP_SM_PS_INVALID:
 		case WLAN_HT_CAP_SM_PS_STATIC:
 			smps_mode = IEEE80211_SMPS_STATIC;
 			break;
 		case WLAN_HT_CAP_SM_PS_DYNAMIC:
 			smps_mode = IEEE80211_SMPS_DYNAMIC;
 			break;
 		case WLAN_HT_CAP_SM_PS_DISABLED:
 			smps_mode = IEEE80211_SMPS_OFF;
 			break;
 		}

 		link_sta->pub->smps_mode = smps_mode;
 	} else {
 		link_sta->pub->smps_mode = IEEE80211_SMPS_OFF;
 	}

 	switch (le16_get_bits(he_6ghz_capa->capa,
 			      IEEE80211_HE_6GHZ_CAP_MAX_MPDU_LEN)) {
 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
 		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
 		break;
 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
 		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
 		break;
 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
 	default:
 		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
 		break;
 	}

 	ieee80211_sta_recalc_aggregates(&sta->sta);

 	link_sta->pub->he_6ghz_capa = *he_6ghz_capa;
 }

 static void ieee80211_he_mcs_disable(__le16 *he_mcs)
 {
 	u32 i;

 	for (i = 0; i < 8; i++)
 		*he_mcs |= cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
 }

 static void ieee80211_he_mcs_intersection(__le16 *he_own_rx, __le16 *he_peer_rx,
 					  __le16 *he_own_tx, __le16 *he_peer_tx)
 {
 	u32 i;
 	u16 own_rx, own_tx, peer_rx, peer_tx;

 	for (i = 0; i < 8; i++) {
 		own_rx = le16_to_cpu(*he_own_rx);
 		own_rx = (own_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

 		own_tx = le16_to_cpu(*he_own_tx);
 		own_tx = (own_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

 		peer_rx = le16_to_cpu(*he_peer_rx);
 		peer_rx = (peer_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

 		peer_tx = le16_to_cpu(*he_peer_tx);
 		peer_tx = (peer_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

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

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

 		*he_peer_rx &=
 			~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
 		*he_peer_rx |= cpu_to_le16(peer_rx << i * 2);

 		*he_peer_tx &=
 			~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
 		*he_peer_tx |= cpu_to_le16(peer_tx << i * 2);
 	}
 }

 void
 ieee80211_he_cap_ie_to_sta_he_cap(struct ieee80211_sub_if_data *sdata,
 				  struct ieee80211_supported_band *sband,
 				  const u8 *he_cap_ie, u8 he_cap_len,
 				  const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
 				  struct link_sta_info *link_sta)
 {
 	struct ieee80211_sta_he_cap *he_cap = &link_sta->pub->he_cap;
 	const struct ieee80211_sta_he_cap *own_he_cap_ptr;
 	struct ieee80211_sta_he_cap own_he_cap;
 	struct ieee80211_he_cap_elem *he_cap_ie_elem = (void *)he_cap_ie;
 	u8 he_ppe_size;
 	u8 mcs_nss_size;
 	u8 he_total_size;
 	bool own_160, peer_160, own_80p80, peer_80p80;

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

 	if (!he_cap_ie)
 		return;

 	own_he_cap_ptr =
 		ieee80211_get_he_iftype_cap_vif(sband, &sdata->vif);
 	if (!own_he_cap_ptr)
 		return;

 	own_he_cap = *own_he_cap_ptr;

 	/* Make sure size is OK */
 	mcs_nss_size = ieee80211_he_mcs_nss_size(he_cap_ie_elem);
 	he_ppe_size =
 		ieee80211_he_ppe_size(he_cap_ie[sizeof(he_cap->he_cap_elem) +
 						mcs_nss_size],
 				      he_cap_ie_elem->phy_cap_info);
 	he_total_size = sizeof(he_cap->he_cap_elem) + mcs_nss_size +
 			he_ppe_size;
 	if (he_cap_len < he_total_size)
 		return;

 	memcpy(&he_cap->he_cap_elem, he_cap_ie, sizeof(he_cap->he_cap_elem));

 	/* HE Tx/Rx HE MCS NSS Support Field */
 	memcpy(&he_cap->he_mcs_nss_supp,
 	       &he_cap_ie[sizeof(he_cap->he_cap_elem)], mcs_nss_size);

 	/* Check if there are (optional) PPE Thresholds */
 	if (he_cap->he_cap_elem.phy_cap_info[6] &
 	    IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT)
 		memcpy(he_cap->ppe_thres,
 		       &he_cap_ie[sizeof(he_cap->he_cap_elem) + mcs_nss_size],
 		       he_ppe_size);

 	he_cap->has_he = true;

 	link_sta->cur_max_bandwidth = ieee80211_sta_cap_rx_bw(link_sta);
 	link_sta->pub->bandwidth = ieee80211_sta_cur_vht_bw(link_sta);

 	if (sband->band == NL80211_BAND_6GHZ && he_6ghz_capa)
 		ieee80211_update_from_he_6ghz_capa(he_6ghz_capa, link_sta);

 	ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80,
 				      &he_cap->he_mcs_nss_supp.rx_mcs_80,
 				      &own_he_cap.he_mcs_nss_supp.tx_mcs_80,
 				      &he_cap->he_mcs_nss_supp.tx_mcs_80);

 	own_160 = own_he_cap.he_cap_elem.phy_cap_info[0] &
 		  IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
 	peer_160 = he_cap->he_cap_elem.phy_cap_info[0] &
 		   IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;

 	if (peer_160 && own_160) {
 		ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_160,
 					      &he_cap->he_mcs_nss_supp.rx_mcs_160,
 					      &own_he_cap.he_mcs_nss_supp.tx_mcs_160,
 					      &he_cap->he_mcs_nss_supp.tx_mcs_160);
 	} else if (peer_160 && !own_160) {
 		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_160);
 		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_160);
 		he_cap->he_cap_elem.phy_cap_info[0] &=
 			~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
 	}

 	own_80p80 = own_he_cap.he_cap_elem.phy_cap_info[0] &
 		    IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
 	peer_80p80 = he_cap->he_cap_elem.phy_cap_info[0] &
 		     IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;

 	if (peer_80p80 && own_80p80) {
 		ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80p80,
 					      &he_cap->he_mcs_nss_supp.rx_mcs_80p80,
 					      &own_he_cap.he_mcs_nss_supp.tx_mcs_80p80,
 					      &he_cap->he_mcs_nss_supp.tx_mcs_80p80);
 	} else if (peer_80p80 && !own_80p80) {
 		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_80p80);
 		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_80p80);
 		he_cap->he_cap_elem.phy_cap_info[0] &=
 			~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
 	}
 }

 void
 ieee80211_he_op_ie_to_bss_conf(struct ieee80211_vif *vif,
 			const struct ieee80211_he_operation *he_op_ie)
 {
 	memset(&vif->bss_conf.he_oper, 0, sizeof(vif->bss_conf.he_oper));
 	if (!he_op_ie)
 		return;

 	vif->bss_conf.he_oper.params = __le32_to_cpu(he_op_ie->he_oper_params);
 	vif->bss_conf.he_oper.nss_set = __le16_to_cpu(he_op_ie->he_mcs_nss_set);
 }

 void
 ieee80211_he_spr_ie_to_bss_conf(struct ieee80211_vif *vif,
 				const struct ieee80211_he_spr *he_spr_ie_elem)
 {
 	struct ieee80211_he_obss_pd *he_obss_pd =
 					&vif->bss_conf.he_obss_pd;
 	const u8 *data;

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

 	if (!he_spr_ie_elem)
 		return;

 	he_obss_pd->sr_ctrl = he_spr_ie_elem->he_sr_control;
 	data = he_spr_ie_elem->optional;

 	if (he_spr_ie_elem->he_sr_control &
 	    IEEE80211_HE_SPR_NON_SRG_OFFSET_PRESENT)
 		he_obss_pd->non_srg_max_offset = *data++;

 	if (he_spr_ie_elem->he_sr_control &
 	    IEEE80211_HE_SPR_SRG_INFORMATION_PRESENT) {
 		he_obss_pd->min_offset = *data++;
 		he_obss_pd->max_offset = *data++;
 		memcpy(he_obss_pd->bss_color_bitmap, data, 8);
 		data += 8;
 		memcpy(he_obss_pd->partial_bssid_bitmap, data, 8);
 		he_obss_pd->enable = true;
 	}
 }

 static void ieee80211_link_sta_rc_update_omi(struct ieee80211_link_data *link,
 					     struct link_sta_info *link_sta)
 {
 	struct ieee80211_sub_if_data *sdata = link->sdata;
 	struct ieee80211_supported_band *sband;
 	enum ieee80211_sta_rx_bandwidth new_bw;
 	enum nl80211_band band;

 	band = link->conf->chanreq.oper.chan->band;
 	sband = sdata->local->hw.wiphy->bands[band];

 	new_bw = ieee80211_sta_cur_vht_bw(link_sta);
 	if (link_sta->pub->bandwidth == new_bw)
 		return;

 	link_sta->pub->bandwidth = new_bw;
 	rate_control_rate_update(sdata->local, sband, link_sta,
 				 IEEE80211_RC_BW_CHANGED);
 }

 bool ieee80211_prepare_rx_omi_bw(struct ieee80211_link_sta *pub_link_sta,
 				 enum ieee80211_sta_rx_bandwidth bw)
 {
 	struct sta_info *sta = container_of(pub_link_sta->sta,
 					    struct sta_info, sta);
 	struct ieee80211_local *local = sta->sdata->local;
 	struct link_sta_info *link_sta =
 		sdata_dereference(sta->link[pub_link_sta->link_id], sta->sdata);
 	struct ieee80211_link_data *link =
 		sdata_dereference(sta->sdata->link[pub_link_sta->link_id],
 				  sta->sdata);
 	struct ieee80211_chanctx_conf *conf;
 	struct ieee80211_chanctx *chanctx;
 	bool ret;

 	if (WARN_ON(!link || !link_sta || link_sta->pub != pub_link_sta))
 		return false;

 	conf = sdata_dereference(link->conf->chanctx_conf, sta->sdata);
 	if (WARN_ON(!conf))
 		return false;

 	trace_api_prepare_rx_omi_bw(local, sta->sdata, link_sta, bw);

 	chanctx = container_of(conf, typeof(*chanctx), conf);

 	if (link_sta->rx_omi_bw_staging == bw) {
 		ret = false;
 		goto trace;
 	}

 	/* must call this API in pairs */
 	if (WARN_ON(link_sta->rx_omi_bw_tx != link_sta->rx_omi_bw_staging ||
 		    link_sta->rx_omi_bw_rx != link_sta->rx_omi_bw_staging)) {
 		ret = false;
 		goto trace;
 	}

 	if (bw < link_sta->rx_omi_bw_staging) {
 		link_sta->rx_omi_bw_tx = bw;
 		ieee80211_link_sta_rc_update_omi(link, link_sta);
 	} else {
 		link_sta->rx_omi_bw_rx = bw;
 		ieee80211_recalc_chanctx_min_def(local, chanctx, NULL, false);
 	}

 	link_sta->rx_omi_bw_staging = bw;
 	ret = true;
 trace:
 	trace_api_return_bool(local, ret);
 	return ret;
 }
 EXPORT_SYMBOL_GPL(ieee80211_prepare_rx_omi_bw);

 void ieee80211_finalize_rx_omi_bw(struct ieee80211_link_sta *pub_link_sta)
 {
 	struct sta_info *sta = container_of(pub_link_sta->sta,
 					    struct sta_info, sta);
 	struct ieee80211_local *local = sta->sdata->local;
 	struct link_sta_info *link_sta =
 		sdata_dereference(sta->link[pub_link_sta->link_id], sta->sdata);
 	struct ieee80211_link_data *link =
 		sdata_dereference(sta->sdata->link[pub_link_sta->link_id],
 				  sta->sdata);
 	struct ieee80211_chanctx_conf *conf;
 	struct ieee80211_chanctx *chanctx;

 	if (WARN_ON(!link || !link_sta || link_sta->pub != pub_link_sta))
 		return;

 	conf = sdata_dereference(link->conf->chanctx_conf, sta->sdata);
 	if (WARN_ON(!conf))
 		return;

 	trace_api_finalize_rx_omi_bw(local, sta->sdata, link_sta);

 	chanctx = container_of(conf, typeof(*chanctx), conf);

 	if (link_sta->rx_omi_bw_tx != link_sta->rx_omi_bw_staging) {
 		/* rate control in finalize only when widening bandwidth */
 		WARN_ON(link_sta->rx_omi_bw_tx > link_sta->rx_omi_bw_staging);
 		link_sta->rx_omi_bw_tx = link_sta->rx_omi_bw_staging;
 		ieee80211_link_sta_rc_update_omi(link, link_sta);
 	}

 	if (link_sta->rx_omi_bw_rx != link_sta->rx_omi_bw_staging) {
 		/* channel context in finalize only when narrowing bandwidth */
 		WARN_ON(link_sta->rx_omi_bw_rx < link_sta->rx_omi_bw_staging);
 		link_sta->rx_omi_bw_rx = link_sta->rx_omi_bw_staging;
 		ieee80211_recalc_chanctx_min_def(local, chanctx, NULL, false);
 	}

 	trace_api_return_void(local);
 }
 EXPORT_SYMBOL_GPL(ieee80211_finalize_rx_omi_bw);
	// SPDX-License-Identifier: GPL-2.0-only
	/*
	* HE handling
	*
	* Copyright(c) 2017 Intel Deutschland GmbH
	* Copyright(c) 2019 - 2024 Intel Corporation
	*/

	#include "ieee80211_i.h"
	#include "rate.h"

	static void
	ieee80211_update_from_he_6ghz_capa(const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
	struct link_sta_info *link_sta)
	{
	struct sta_info *sta = link_sta->sta;
	enum ieee80211_smps_mode smps_mode;

	if (sta->sdata->vif.type == NL80211_IFTYPE_AP \|\|
	sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
	switch (le16_get_bits(he_6ghz_capa->capa,
	IEEE80211_HE_6GHZ_CAP_SM_PS)) {
	case WLAN_HT_CAP_SM_PS_INVALID:
	case WLAN_HT_CAP_SM_PS_STATIC:
	smps_mode = IEEE80211_SMPS_STATIC;
	break;
	case WLAN_HT_CAP_SM_PS_DYNAMIC:
	smps_mode = IEEE80211_SMPS_DYNAMIC;
	break;
	case WLAN_HT_CAP_SM_PS_DISABLED:
	smps_mode = IEEE80211_SMPS_OFF;
	break;
	}

	link_sta->pub->smps_mode = smps_mode;
	} else {
	link_sta->pub->smps_mode = IEEE80211_SMPS_OFF;
	}

	switch (le16_get_bits(he_6ghz_capa->capa,
	IEEE80211_HE_6GHZ_CAP_MAX_MPDU_LEN)) {
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
	link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
	break;
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
	link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
	break;
	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
	default:
	link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
	break;
	}

	ieee80211_sta_recalc_aggregates(&sta->sta);

	link_sta->pub->he_6ghz_capa = *he_6ghz_capa;
	}

	static void ieee80211_he_mcs_disable(__le16 *he_mcs)
	{
	u32 i;

	for (i = 0; i < 8; i++)
	he_mcs \|= cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i 2);
	}

	static void ieee80211_he_mcs_intersection(__le16 he_own_rx, __le16 he_peer_rx,
	__le16 he_own_tx, __le16 he_peer_tx)
	{
	u32 i;
	u16 own_rx, own_tx, peer_rx, peer_tx;

	for (i = 0; i < 8; i++) {
	own_rx = le16_to_cpu(*he_own_rx);
	own_rx = (own_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

	own_tx = le16_to_cpu(*he_own_tx);
	own_tx = (own_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

	peer_rx = le16_to_cpu(*he_peer_rx);
	peer_rx = (peer_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

	peer_tx = le16_to_cpu(*he_peer_tx);
	peer_tx = (peer_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;

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

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

	*he_peer_rx &=
	~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
	he_peer_rx \|= cpu_to_le16(peer_rx << i 2);

	*he_peer_tx &=
	~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
	he_peer_tx \|= cpu_to_le16(peer_tx << i 2);
	}
	}

	void
	ieee80211_he_cap_ie_to_sta_he_cap(struct ieee80211_sub_if_data *sdata,
	struct ieee80211_supported_band *sband,
	const u8 *he_cap_ie, u8 he_cap_len,
	const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
	struct link_sta_info *link_sta)
	{
	struct ieee80211_sta_he_cap *he_cap = &link_sta->pub->he_cap;
	const struct ieee80211_sta_he_cap *own_he_cap_ptr;
	struct ieee80211_sta_he_cap own_he_cap;
	struct ieee80211_he_cap_elem he_cap_ie_elem = (void )he_cap_ie;
	u8 he_ppe_size;
	u8 mcs_nss_size;
	u8 he_total_size;
	bool own_160, peer_160, own_80p80, peer_80p80;

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

	if (!he_cap_ie)
	return;

	own_he_cap_ptr =
	ieee80211_get_he_iftype_cap_vif(sband, &sdata->vif);
	if (!own_he_cap_ptr)
	return;

	own_he_cap = *own_he_cap_ptr;

	/* Make sure size is OK */
	mcs_nss_size = ieee80211_he_mcs_nss_size(he_cap_ie_elem);
	he_ppe_size =
	ieee80211_he_ppe_size(he_cap_ie[sizeof(he_cap->he_cap_elem) +
	mcs_nss_size],
	he_cap_ie_elem->phy_cap_info);
	he_total_size = sizeof(he_cap->he_cap_elem) + mcs_nss_size +
	he_ppe_size;
	if (he_cap_len < he_total_size)
	return;

	memcpy(&he_cap->he_cap_elem, he_cap_ie, sizeof(he_cap->he_cap_elem));

	/* HE Tx/Rx HE MCS NSS Support Field */
	memcpy(&he_cap->he_mcs_nss_supp,
	&he_cap_ie[sizeof(he_cap->he_cap_elem)], mcs_nss_size);

	/* Check if there are (optional) PPE Thresholds */
	if (he_cap->he_cap_elem.phy_cap_info[6] &
	IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT)
	memcpy(he_cap->ppe_thres,
	&he_cap_ie[sizeof(he_cap->he_cap_elem) + mcs_nss_size],
	he_ppe_size);

	he_cap->has_he = true;

	link_sta->cur_max_bandwidth = ieee80211_sta_cap_rx_bw(link_sta);
	link_sta->pub->bandwidth = ieee80211_sta_cur_vht_bw(link_sta);

	if (sband->band == NL80211_BAND_6GHZ && he_6ghz_capa)
	ieee80211_update_from_he_6ghz_capa(he_6ghz_capa, link_sta);

	ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80,
	&he_cap->he_mcs_nss_supp.rx_mcs_80,
	&own_he_cap.he_mcs_nss_supp.tx_mcs_80,
	&he_cap->he_mcs_nss_supp.tx_mcs_80);

	own_160 = own_he_cap.he_cap_elem.phy_cap_info[0] &
	IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
	peer_160 = he_cap->he_cap_elem.phy_cap_info[0] &
	IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;

	if (peer_160 && own_160) {
	ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_160,
	&he_cap->he_mcs_nss_supp.rx_mcs_160,
	&own_he_cap.he_mcs_nss_supp.tx_mcs_160,
	&he_cap->he_mcs_nss_supp.tx_mcs_160);
	} else if (peer_160 && !own_160) {
	ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_160);
	ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_160);
	he_cap->he_cap_elem.phy_cap_info[0] &=
	~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
	}

	own_80p80 = own_he_cap.he_cap_elem.phy_cap_info[0] &
	IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
	peer_80p80 = he_cap->he_cap_elem.phy_cap_info[0] &
	IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;

	if (peer_80p80 && own_80p80) {
	ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80p80,
	&he_cap->he_mcs_nss_supp.rx_mcs_80p80,
	&own_he_cap.he_mcs_nss_supp.tx_mcs_80p80,
	&he_cap->he_mcs_nss_supp.tx_mcs_80p80);
	} else if (peer_80p80 && !own_80p80) {
	ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_80p80);
	ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_80p80);
	he_cap->he_cap_elem.phy_cap_info[0] &=
	~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
	}
	}

	void
	ieee80211_he_op_ie_to_bss_conf(struct ieee80211_vif *vif,
	const struct ieee80211_he_operation *he_op_ie)
	{
	memset(&vif->bss_conf.he_oper, 0, sizeof(vif->bss_conf.he_oper));
	if (!he_op_ie)
	return;

	vif->bss_conf.he_oper.params = __le32_to_cpu(he_op_ie->he_oper_params);
	vif->bss_conf.he_oper.nss_set = __le16_to_cpu(he_op_ie->he_mcs_nss_set);
	}

	void
	ieee80211_he_spr_ie_to_bss_conf(struct ieee80211_vif *vif,
	const struct ieee80211_he_spr *he_spr_ie_elem)
	{
	struct ieee80211_he_obss_pd *he_obss_pd =
	&vif->bss_conf.he_obss_pd;
	const u8 *data;

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

	if (!he_spr_ie_elem)
	return;

	he_obss_pd->sr_ctrl = he_spr_ie_elem->he_sr_control;
	data = he_spr_ie_elem->optional;

	if (he_spr_ie_elem->he_sr_control &
	IEEE80211_HE_SPR_NON_SRG_OFFSET_PRESENT)
	he_obss_pd->non_srg_max_offset = *data++;

	if (he_spr_ie_elem->he_sr_control &
	IEEE80211_HE_SPR_SRG_INFORMATION_PRESENT) {
	he_obss_pd->min_offset = *data++;
	he_obss_pd->max_offset = *data++;
	memcpy(he_obss_pd->bss_color_bitmap, data, 8);
	data += 8;
	memcpy(he_obss_pd->partial_bssid_bitmap, data, 8);
	he_obss_pd->enable = true;
	}
	}

	static void ieee80211_link_sta_rc_update_omi(struct ieee80211_link_data *link,
	struct link_sta_info *link_sta)
	{
	struct ieee80211_sub_if_data *sdata = link->sdata;
	struct ieee80211_supported_band *sband;
	enum ieee80211_sta_rx_bandwidth new_bw;
	enum nl80211_band band;

	band = link->conf->chanreq.oper.chan->band;
	sband = sdata->local->hw.wiphy->bands[band];

	new_bw = ieee80211_sta_cur_vht_bw(link_sta);
	if (link_sta->pub->bandwidth == new_bw)
	return;

	link_sta->pub->bandwidth = new_bw;
	rate_control_rate_update(sdata->local, sband, link_sta,
	IEEE80211_RC_BW_CHANGED);
	}

	bool ieee80211_prepare_rx_omi_bw(struct ieee80211_link_sta *pub_link_sta,
	enum ieee80211_sta_rx_bandwidth bw)
	{
	struct sta_info *sta = container_of(pub_link_sta->sta,
	struct sta_info, sta);
	struct ieee80211_local *local = sta->sdata->local;
	struct link_sta_info *link_sta =
	sdata_dereference(sta->link[pub_link_sta->link_id], sta->sdata);
	struct ieee80211_link_data *link =
	sdata_dereference(sta->sdata->link[pub_link_sta->link_id],
	sta->sdata);
	struct ieee80211_chanctx_conf *conf;
	struct ieee80211_chanctx *chanctx;
	bool ret;

	if (WARN_ON(!link \|\| !link_sta \|\| link_sta->pub != pub_link_sta))
	return false;

	conf = sdata_dereference(link->conf->chanctx_conf, sta->sdata);
	if (WARN_ON(!conf))
	return false;

	trace_api_prepare_rx_omi_bw(local, sta->sdata, link_sta, bw);

	chanctx = container_of(conf, typeof(*chanctx), conf);

	if (link_sta->rx_omi_bw_staging == bw) {
	ret = false;
	goto trace;
	}

	/* must call this API in pairs */
	if (WARN_ON(link_sta->rx_omi_bw_tx != link_sta->rx_omi_bw_staging \|\|
	link_sta->rx_omi_bw_rx != link_sta->rx_omi_bw_staging)) {
	ret = false;
	goto trace;
	}

	if (bw < link_sta->rx_omi_bw_staging) {
	link_sta->rx_omi_bw_tx = bw;
	ieee80211_link_sta_rc_update_omi(link, link_sta);
	} else {
	link_sta->rx_omi_bw_rx = bw;
	ieee80211_recalc_chanctx_min_def(local, chanctx, NULL, false);
	}

	link_sta->rx_omi_bw_staging = bw;
	ret = true;
	trace:
	trace_api_return_bool(local, ret);
	return ret;
	}
	EXPORT_SYMBOL_GPL(ieee80211_prepare_rx_omi_bw);

	void ieee80211_finalize_rx_omi_bw(struct ieee80211_link_sta *pub_link_sta)
	{
	struct sta_info *sta = container_of(pub_link_sta->sta,
	struct sta_info, sta);
	struct ieee80211_local *local = sta->sdata->local;
	struct link_sta_info *link_sta =
	sdata_dereference(sta->link[pub_link_sta->link_id], sta->sdata);
	struct ieee80211_link_data *link =
	sdata_dereference(sta->sdata->link[pub_link_sta->link_id],
	sta->sdata);
	struct ieee80211_chanctx_conf *conf;
	struct ieee80211_chanctx *chanctx;

	if (WARN_ON(!link \|\| !link_sta \|\| link_sta->pub != pub_link_sta))
	return;

	conf = sdata_dereference(link->conf->chanctx_conf, sta->sdata);
	if (WARN_ON(!conf))
	return;

	trace_api_finalize_rx_omi_bw(local, sta->sdata, link_sta);

	chanctx = container_of(conf, typeof(*chanctx), conf);

	if (link_sta->rx_omi_bw_tx != link_sta->rx_omi_bw_staging) {
	/* rate control in finalize only when widening bandwidth */
	WARN_ON(link_sta->rx_omi_bw_tx > link_sta->rx_omi_bw_staging);
	link_sta->rx_omi_bw_tx = link_sta->rx_omi_bw_staging;
	ieee80211_link_sta_rc_update_omi(link, link_sta);
	}

	if (link_sta->rx_omi_bw_rx != link_sta->rx_omi_bw_staging) {
	/* channel context in finalize only when narrowing bandwidth */
	WARN_ON(link_sta->rx_omi_bw_rx < link_sta->rx_omi_bw_staging);
	link_sta->rx_omi_bw_rx = link_sta->rx_omi_bw_staging;
	ieee80211_recalc_chanctx_min_def(local, chanctx, NULL, false);
	}

	trace_api_return_void(local);
	}
	EXPORT_SYMBOL_GPL(ieee80211_finalize_rx_omi_bw);