162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * HE handling
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright(c) 2017 Intel Deutschland GmbH
662306a36Sopenharmony_ci * Copyright(c) 2019 - 2023 Intel Corporation
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include "ieee80211_i.h"
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_cistatic void
1262306a36Sopenharmony_ciieee80211_update_from_he_6ghz_capa(const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
1362306a36Sopenharmony_ci				   struct link_sta_info *link_sta)
1462306a36Sopenharmony_ci{
1562306a36Sopenharmony_ci	struct sta_info *sta = link_sta->sta;
1662306a36Sopenharmony_ci	enum ieee80211_smps_mode smps_mode;
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_ci	if (sta->sdata->vif.type == NL80211_IFTYPE_AP ||
1962306a36Sopenharmony_ci	    sta->sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
2062306a36Sopenharmony_ci		switch (le16_get_bits(he_6ghz_capa->capa,
2162306a36Sopenharmony_ci				      IEEE80211_HE_6GHZ_CAP_SM_PS)) {
2262306a36Sopenharmony_ci		case WLAN_HT_CAP_SM_PS_INVALID:
2362306a36Sopenharmony_ci		case WLAN_HT_CAP_SM_PS_STATIC:
2462306a36Sopenharmony_ci			smps_mode = IEEE80211_SMPS_STATIC;
2562306a36Sopenharmony_ci			break;
2662306a36Sopenharmony_ci		case WLAN_HT_CAP_SM_PS_DYNAMIC:
2762306a36Sopenharmony_ci			smps_mode = IEEE80211_SMPS_DYNAMIC;
2862306a36Sopenharmony_ci			break;
2962306a36Sopenharmony_ci		case WLAN_HT_CAP_SM_PS_DISABLED:
3062306a36Sopenharmony_ci			smps_mode = IEEE80211_SMPS_OFF;
3162306a36Sopenharmony_ci			break;
3262306a36Sopenharmony_ci		}
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci		link_sta->pub->smps_mode = smps_mode;
3562306a36Sopenharmony_ci	} else {
3662306a36Sopenharmony_ci		link_sta->pub->smps_mode = IEEE80211_SMPS_OFF;
3762306a36Sopenharmony_ci	}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	switch (le16_get_bits(he_6ghz_capa->capa,
4062306a36Sopenharmony_ci			      IEEE80211_HE_6GHZ_CAP_MAX_MPDU_LEN)) {
4162306a36Sopenharmony_ci	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
4262306a36Sopenharmony_ci		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
4362306a36Sopenharmony_ci		break;
4462306a36Sopenharmony_ci	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
4562306a36Sopenharmony_ci		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
4662306a36Sopenharmony_ci		break;
4762306a36Sopenharmony_ci	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
4862306a36Sopenharmony_ci	default:
4962306a36Sopenharmony_ci		link_sta->pub->agg.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
5062306a36Sopenharmony_ci		break;
5162306a36Sopenharmony_ci	}
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_ci	ieee80211_sta_recalc_aggregates(&sta->sta);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci	link_sta->pub->he_6ghz_capa = *he_6ghz_capa;
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_cistatic void ieee80211_he_mcs_disable(__le16 *he_mcs)
5962306a36Sopenharmony_ci{
6062306a36Sopenharmony_ci	u32 i;
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci	for (i = 0; i < 8; i++)
6362306a36Sopenharmony_ci		*he_mcs |= cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
6462306a36Sopenharmony_ci}
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_cistatic void ieee80211_he_mcs_intersection(__le16 *he_own_rx, __le16 *he_peer_rx,
6762306a36Sopenharmony_ci					  __le16 *he_own_tx, __le16 *he_peer_tx)
6862306a36Sopenharmony_ci{
6962306a36Sopenharmony_ci	u32 i;
7062306a36Sopenharmony_ci	u16 own_rx, own_tx, peer_rx, peer_tx;
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci	for (i = 0; i < 8; i++) {
7362306a36Sopenharmony_ci		own_rx = le16_to_cpu(*he_own_rx);
7462306a36Sopenharmony_ci		own_rx = (own_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci		own_tx = le16_to_cpu(*he_own_tx);
7762306a36Sopenharmony_ci		own_tx = (own_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci		peer_rx = le16_to_cpu(*he_peer_rx);
8062306a36Sopenharmony_ci		peer_rx = (peer_rx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci		peer_tx = le16_to_cpu(*he_peer_tx);
8362306a36Sopenharmony_ci		peer_tx = (peer_tx >> i * 2) & IEEE80211_HE_MCS_NOT_SUPPORTED;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci		if (peer_tx != IEEE80211_HE_MCS_NOT_SUPPORTED) {
8662306a36Sopenharmony_ci			if (own_rx == IEEE80211_HE_MCS_NOT_SUPPORTED)
8762306a36Sopenharmony_ci				peer_tx = IEEE80211_HE_MCS_NOT_SUPPORTED;
8862306a36Sopenharmony_ci			else if (own_rx < peer_tx)
8962306a36Sopenharmony_ci				peer_tx = own_rx;
9062306a36Sopenharmony_ci		}
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci		if (peer_rx != IEEE80211_HE_MCS_NOT_SUPPORTED) {
9362306a36Sopenharmony_ci			if (own_tx == IEEE80211_HE_MCS_NOT_SUPPORTED)
9462306a36Sopenharmony_ci				peer_rx = IEEE80211_HE_MCS_NOT_SUPPORTED;
9562306a36Sopenharmony_ci			else if (own_tx < peer_rx)
9662306a36Sopenharmony_ci				peer_rx = own_tx;
9762306a36Sopenharmony_ci		}
9862306a36Sopenharmony_ci
9962306a36Sopenharmony_ci		*he_peer_rx &=
10062306a36Sopenharmony_ci			~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
10162306a36Sopenharmony_ci		*he_peer_rx |= cpu_to_le16(peer_rx << i * 2);
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci		*he_peer_tx &=
10462306a36Sopenharmony_ci			~cpu_to_le16(IEEE80211_HE_MCS_NOT_SUPPORTED << i * 2);
10562306a36Sopenharmony_ci		*he_peer_tx |= cpu_to_le16(peer_tx << i * 2);
10662306a36Sopenharmony_ci	}
10762306a36Sopenharmony_ci}
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_civoid
11062306a36Sopenharmony_ciieee80211_he_cap_ie_to_sta_he_cap(struct ieee80211_sub_if_data *sdata,
11162306a36Sopenharmony_ci				  struct ieee80211_supported_band *sband,
11262306a36Sopenharmony_ci				  const u8 *he_cap_ie, u8 he_cap_len,
11362306a36Sopenharmony_ci				  const struct ieee80211_he_6ghz_capa *he_6ghz_capa,
11462306a36Sopenharmony_ci				  struct link_sta_info *link_sta)
11562306a36Sopenharmony_ci{
11662306a36Sopenharmony_ci	struct ieee80211_sta_he_cap *he_cap = &link_sta->pub->he_cap;
11762306a36Sopenharmony_ci	const struct ieee80211_sta_he_cap *own_he_cap_ptr;
11862306a36Sopenharmony_ci	struct ieee80211_sta_he_cap own_he_cap;
11962306a36Sopenharmony_ci	struct ieee80211_he_cap_elem *he_cap_ie_elem = (void *)he_cap_ie;
12062306a36Sopenharmony_ci	u8 he_ppe_size;
12162306a36Sopenharmony_ci	u8 mcs_nss_size;
12262306a36Sopenharmony_ci	u8 he_total_size;
12362306a36Sopenharmony_ci	bool own_160, peer_160, own_80p80, peer_80p80;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	memset(he_cap, 0, sizeof(*he_cap));
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	if (!he_cap_ie)
12862306a36Sopenharmony_ci		return;
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	own_he_cap_ptr =
13162306a36Sopenharmony_ci		ieee80211_get_he_iftype_cap_vif(sband, &sdata->vif);
13262306a36Sopenharmony_ci	if (!own_he_cap_ptr)
13362306a36Sopenharmony_ci		return;
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_ci	own_he_cap = *own_he_cap_ptr;
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ci	/* Make sure size is OK */
13862306a36Sopenharmony_ci	mcs_nss_size = ieee80211_he_mcs_nss_size(he_cap_ie_elem);
13962306a36Sopenharmony_ci	he_ppe_size =
14062306a36Sopenharmony_ci		ieee80211_he_ppe_size(he_cap_ie[sizeof(he_cap->he_cap_elem) +
14162306a36Sopenharmony_ci						mcs_nss_size],
14262306a36Sopenharmony_ci				      he_cap_ie_elem->phy_cap_info);
14362306a36Sopenharmony_ci	he_total_size = sizeof(he_cap->he_cap_elem) + mcs_nss_size +
14462306a36Sopenharmony_ci			he_ppe_size;
14562306a36Sopenharmony_ci	if (he_cap_len < he_total_size)
14662306a36Sopenharmony_ci		return;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	memcpy(&he_cap->he_cap_elem, he_cap_ie, sizeof(he_cap->he_cap_elem));
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	/* HE Tx/Rx HE MCS NSS Support Field */
15162306a36Sopenharmony_ci	memcpy(&he_cap->he_mcs_nss_supp,
15262306a36Sopenharmony_ci	       &he_cap_ie[sizeof(he_cap->he_cap_elem)], mcs_nss_size);
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	/* Check if there are (optional) PPE Thresholds */
15562306a36Sopenharmony_ci	if (he_cap->he_cap_elem.phy_cap_info[6] &
15662306a36Sopenharmony_ci	    IEEE80211_HE_PHY_CAP6_PPE_THRESHOLD_PRESENT)
15762306a36Sopenharmony_ci		memcpy(he_cap->ppe_thres,
15862306a36Sopenharmony_ci		       &he_cap_ie[sizeof(he_cap->he_cap_elem) + mcs_nss_size],
15962306a36Sopenharmony_ci		       he_ppe_size);
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	he_cap->has_he = true;
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	link_sta->cur_max_bandwidth = ieee80211_sta_cap_rx_bw(link_sta);
16462306a36Sopenharmony_ci	link_sta->pub->bandwidth = ieee80211_sta_cur_vht_bw(link_sta);
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	if (sband->band == NL80211_BAND_6GHZ && he_6ghz_capa)
16762306a36Sopenharmony_ci		ieee80211_update_from_he_6ghz_capa(he_6ghz_capa, link_sta);
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80,
17062306a36Sopenharmony_ci				      &he_cap->he_mcs_nss_supp.rx_mcs_80,
17162306a36Sopenharmony_ci				      &own_he_cap.he_mcs_nss_supp.tx_mcs_80,
17262306a36Sopenharmony_ci				      &he_cap->he_mcs_nss_supp.tx_mcs_80);
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	own_160 = own_he_cap.he_cap_elem.phy_cap_info[0] &
17562306a36Sopenharmony_ci		  IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
17662306a36Sopenharmony_ci	peer_160 = he_cap->he_cap_elem.phy_cap_info[0] &
17762306a36Sopenharmony_ci		   IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
17862306a36Sopenharmony_ci
17962306a36Sopenharmony_ci	if (peer_160 && own_160) {
18062306a36Sopenharmony_ci		ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_160,
18162306a36Sopenharmony_ci					      &he_cap->he_mcs_nss_supp.rx_mcs_160,
18262306a36Sopenharmony_ci					      &own_he_cap.he_mcs_nss_supp.tx_mcs_160,
18362306a36Sopenharmony_ci					      &he_cap->he_mcs_nss_supp.tx_mcs_160);
18462306a36Sopenharmony_ci	} else if (peer_160 && !own_160) {
18562306a36Sopenharmony_ci		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_160);
18662306a36Sopenharmony_ci		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_160);
18762306a36Sopenharmony_ci		he_cap->he_cap_elem.phy_cap_info[0] &=
18862306a36Sopenharmony_ci			~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
18962306a36Sopenharmony_ci	}
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	own_80p80 = own_he_cap.he_cap_elem.phy_cap_info[0] &
19262306a36Sopenharmony_ci		    IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
19362306a36Sopenharmony_ci	peer_80p80 = he_cap->he_cap_elem.phy_cap_info[0] &
19462306a36Sopenharmony_ci		     IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	if (peer_80p80 && own_80p80) {
19762306a36Sopenharmony_ci		ieee80211_he_mcs_intersection(&own_he_cap.he_mcs_nss_supp.rx_mcs_80p80,
19862306a36Sopenharmony_ci					      &he_cap->he_mcs_nss_supp.rx_mcs_80p80,
19962306a36Sopenharmony_ci					      &own_he_cap.he_mcs_nss_supp.tx_mcs_80p80,
20062306a36Sopenharmony_ci					      &he_cap->he_mcs_nss_supp.tx_mcs_80p80);
20162306a36Sopenharmony_ci	} else if (peer_80p80 && !own_80p80) {
20262306a36Sopenharmony_ci		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.rx_mcs_80p80);
20362306a36Sopenharmony_ci		ieee80211_he_mcs_disable(&he_cap->he_mcs_nss_supp.tx_mcs_80p80);
20462306a36Sopenharmony_ci		he_cap->he_cap_elem.phy_cap_info[0] &=
20562306a36Sopenharmony_ci			~IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G;
20662306a36Sopenharmony_ci	}
20762306a36Sopenharmony_ci}
20862306a36Sopenharmony_ci
20962306a36Sopenharmony_civoid
21062306a36Sopenharmony_ciieee80211_he_op_ie_to_bss_conf(struct ieee80211_vif *vif,
21162306a36Sopenharmony_ci			const struct ieee80211_he_operation *he_op_ie)
21262306a36Sopenharmony_ci{
21362306a36Sopenharmony_ci	memset(&vif->bss_conf.he_oper, 0, sizeof(vif->bss_conf.he_oper));
21462306a36Sopenharmony_ci	if (!he_op_ie)
21562306a36Sopenharmony_ci		return;
21662306a36Sopenharmony_ci
21762306a36Sopenharmony_ci	vif->bss_conf.he_oper.params = __le32_to_cpu(he_op_ie->he_oper_params);
21862306a36Sopenharmony_ci	vif->bss_conf.he_oper.nss_set = __le16_to_cpu(he_op_ie->he_mcs_nss_set);
21962306a36Sopenharmony_ci}
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_civoid
22262306a36Sopenharmony_ciieee80211_he_spr_ie_to_bss_conf(struct ieee80211_vif *vif,
22362306a36Sopenharmony_ci				const struct ieee80211_he_spr *he_spr_ie_elem)
22462306a36Sopenharmony_ci{
22562306a36Sopenharmony_ci	struct ieee80211_he_obss_pd *he_obss_pd =
22662306a36Sopenharmony_ci					&vif->bss_conf.he_obss_pd;
22762306a36Sopenharmony_ci	const u8 *data;
22862306a36Sopenharmony_ci
22962306a36Sopenharmony_ci	memset(he_obss_pd, 0, sizeof(*he_obss_pd));
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	if (!he_spr_ie_elem)
23262306a36Sopenharmony_ci		return;
23362306a36Sopenharmony_ci	data = he_spr_ie_elem->optional;
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_ci	if (he_spr_ie_elem->he_sr_control &
23662306a36Sopenharmony_ci	    IEEE80211_HE_SPR_NON_SRG_OFFSET_PRESENT)
23762306a36Sopenharmony_ci		data++;
23862306a36Sopenharmony_ci	if (he_spr_ie_elem->he_sr_control &
23962306a36Sopenharmony_ci	    IEEE80211_HE_SPR_SRG_INFORMATION_PRESENT) {
24062306a36Sopenharmony_ci		he_obss_pd->max_offset = *data++;
24162306a36Sopenharmony_ci		he_obss_pd->min_offset = *data++;
24262306a36Sopenharmony_ci		he_obss_pd->enable = true;
24362306a36Sopenharmony_ci	}
24462306a36Sopenharmony_ci}
245