1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * VHT handling
4  *
5  * Portions of this file
6  * Copyright(c) 2015 - 2016 Intel Deutschland GmbH
7  * Copyright (C) 2018 - 2022 Intel Corporation
8  */
9 
10 #include <linux/ieee80211.h>
11 #include <linux/export.h>
12 #include <net/mac80211.h>
13 #include "ieee80211_i.h"
14 #include "rate.h"
15 
16 
__check_vhtcap_disable(struct ieee80211_sub_if_data * sdata,struct ieee80211_sta_vht_cap * vht_cap,u32 flag)17 static void __check_vhtcap_disable(struct ieee80211_sub_if_data *sdata,
18 				   struct ieee80211_sta_vht_cap *vht_cap,
19 				   u32 flag)
20 {
21 	__le32 le_flag = cpu_to_le32(flag);
22 
23 	if (sdata->u.mgd.vht_capa_mask.vht_cap_info & le_flag &&
24 	    !(sdata->u.mgd.vht_capa.vht_cap_info & le_flag))
25 		vht_cap->cap &= ~flag;
26 }
27 
ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data * sdata,struct ieee80211_sta_vht_cap * vht_cap)28 void ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data *sdata,
29 				      struct ieee80211_sta_vht_cap *vht_cap)
30 {
31 	int i;
32 	u16 rxmcs_mask, rxmcs_cap, rxmcs_n, txmcs_mask, txmcs_cap, txmcs_n;
33 
34 	if (!vht_cap->vht_supported)
35 		return;
36 
37 	if (sdata->vif.type != NL80211_IFTYPE_STATION)
38 		return;
39 
40 	__check_vhtcap_disable(sdata, vht_cap,
41 			       IEEE80211_VHT_CAP_RXLDPC);
42 	__check_vhtcap_disable(sdata, vht_cap,
43 			       IEEE80211_VHT_CAP_SHORT_GI_80);
44 	__check_vhtcap_disable(sdata, vht_cap,
45 			       IEEE80211_VHT_CAP_SHORT_GI_160);
46 	__check_vhtcap_disable(sdata, vht_cap,
47 			       IEEE80211_VHT_CAP_TXSTBC);
48 	__check_vhtcap_disable(sdata, vht_cap,
49 			       IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE);
50 	__check_vhtcap_disable(sdata, vht_cap,
51 			       IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE);
52 	__check_vhtcap_disable(sdata, vht_cap,
53 			       IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN);
54 	__check_vhtcap_disable(sdata, vht_cap,
55 			       IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN);
56 
57 	/* Allow user to decrease AMPDU length exponent */
58 	if (sdata->u.mgd.vht_capa_mask.vht_cap_info &
59 	    cpu_to_le32(IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK)) {
60 		u32 cap, n;
61 
62 		n = le32_to_cpu(sdata->u.mgd.vht_capa.vht_cap_info) &
63 			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
64 		n >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
65 		cap = vht_cap->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
66 		cap >>= IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
67 
68 		if (n < cap) {
69 			vht_cap->cap &=
70 				~IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
71 			vht_cap->cap |=
72 				n << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
73 		}
74 	}
75 
76 	/* Allow the user to decrease MCSes */
77 	rxmcs_mask =
78 		le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.rx_mcs_map);
79 	rxmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.rx_mcs_map);
80 	rxmcs_n &= rxmcs_mask;
81 	rxmcs_cap = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
82 
83 	txmcs_mask =
84 		le16_to_cpu(sdata->u.mgd.vht_capa_mask.supp_mcs.tx_mcs_map);
85 	txmcs_n = le16_to_cpu(sdata->u.mgd.vht_capa.supp_mcs.tx_mcs_map);
86 	txmcs_n &= txmcs_mask;
87 	txmcs_cap = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
88 	for (i = 0; i < 8; i++) {
89 		u8 m, n, c;
90 
91 		m = (rxmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
92 		n = (rxmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
93 		c = (rxmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
94 
95 		if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
96 			  n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
97 			rxmcs_cap &= ~(3 << 2*i);
98 			rxmcs_cap |= (rxmcs_n & (3 << 2*i));
99 		}
100 
101 		m = (txmcs_mask >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
102 		n = (txmcs_n >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
103 		c = (txmcs_cap >> 2*i) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
104 
105 		if (m && ((c != IEEE80211_VHT_MCS_NOT_SUPPORTED && n < c) ||
106 			  n == IEEE80211_VHT_MCS_NOT_SUPPORTED)) {
107 			txmcs_cap &= ~(3 << 2*i);
108 			txmcs_cap |= (txmcs_n & (3 << 2*i));
109 		}
110 	}
111 	vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(rxmcs_cap);
112 	vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(txmcs_cap);
113 }
114 
115 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)116 ieee80211_vht_cap_ie_to_sta_vht_cap(struct ieee80211_sub_if_data *sdata,
117 				    struct ieee80211_supported_band *sband,
118 				    const struct ieee80211_vht_cap *vht_cap_ie,
119 				    struct sta_info *sta)
120 {
121 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
122 	struct ieee80211_sta_vht_cap own_cap;
123 	u32 cap_info, i;
124 	bool have_80mhz;
125 
126 	memset(vht_cap, 0, sizeof(*vht_cap));
127 
128 	if (!sta->sta.deflink.ht_cap.ht_supported)
129 		return;
130 
131 	if (!vht_cap_ie || !sband->vht_cap.vht_supported)
132 		return;
133 
134 	/* Allow VHT if at least one channel on the sband supports 80 MHz */
135 	have_80mhz = false;
136 	for (i = 0; i < sband->n_channels; i++) {
137 		if (sband->channels[i].flags & (IEEE80211_CHAN_DISABLED |
138 						IEEE80211_CHAN_NO_80MHZ))
139 			continue;
140 
141 		have_80mhz = true;
142 		break;
143 	}
144 
145 	if (!have_80mhz)
146 		return;
147 
148 	/*
149 	 * A VHT STA must support 40 MHz, but if we verify that here
150 	 * then we break a few things - some APs (e.g. Netgear R6300v2
151 	 * and others based on the BCM4360 chipset) will unset this
152 	 * capability bit when operating in 20 MHz.
153 	 */
154 
155 	vht_cap->vht_supported = true;
156 
157 	own_cap = sband->vht_cap;
158 	/*
159 	 * If user has specified capability overrides, take care
160 	 * of that if the station we're setting up is the AP that
161 	 * we advertised a restricted capability set to. Override
162 	 * our own capabilities and then use those below.
163 	 */
164 	if (sdata->vif.type == NL80211_IFTYPE_STATION &&
165 	    !test_sta_flag(sta, WLAN_STA_TDLS_PEER))
166 		ieee80211_apply_vhtcap_overrides(sdata, &own_cap);
167 
168 	/* take some capabilities as-is */
169 	cap_info = le32_to_cpu(vht_cap_ie->vht_cap_info);
170 	vht_cap->cap = cap_info;
171 	vht_cap->cap &= IEEE80211_VHT_CAP_RXLDPC |
172 			IEEE80211_VHT_CAP_VHT_TXOP_PS |
173 			IEEE80211_VHT_CAP_HTC_VHT |
174 			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK |
175 			IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_UNSOL_MFB |
176 			IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB |
177 			IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
178 			IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN;
179 
180 	vht_cap->cap |= min_t(u32, cap_info & IEEE80211_VHT_CAP_MAX_MPDU_MASK,
181 			      own_cap.cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK);
182 
183 	/* and some based on our own capabilities */
184 	switch (own_cap.cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
185 	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
186 		vht_cap->cap |= cap_info &
187 				IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ;
188 		break;
189 	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
190 		vht_cap->cap |= cap_info &
191 				IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
192 		break;
193 	default:
194 		/* nothing */
195 		break;
196 	}
197 
198 	/* symmetric capabilities */
199 	vht_cap->cap |= cap_info & own_cap.cap &
200 			(IEEE80211_VHT_CAP_SHORT_GI_80 |
201 			 IEEE80211_VHT_CAP_SHORT_GI_160);
202 
203 	/* remaining ones */
204 	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
205 		vht_cap->cap |= cap_info &
206 				(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
207 				 IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
208 
209 	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
210 		vht_cap->cap |= cap_info &
211 				(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
212 				 IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
213 
214 	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE)
215 		vht_cap->cap |= cap_info &
216 				IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE;
217 
218 	if (own_cap.cap & IEEE80211_VHT_CAP_MU_BEAMFORMEE_CAPABLE)
219 		vht_cap->cap |= cap_info &
220 				IEEE80211_VHT_CAP_MU_BEAMFORMER_CAPABLE;
221 
222 	if (own_cap.cap & IEEE80211_VHT_CAP_TXSTBC)
223 		vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_RXSTBC_MASK;
224 
225 	if (own_cap.cap & IEEE80211_VHT_CAP_RXSTBC_MASK)
226 		vht_cap->cap |= cap_info & IEEE80211_VHT_CAP_TXSTBC;
227 
228 	/* Copy peer MCS info, the driver might need them. */
229 	memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
230 	       sizeof(struct ieee80211_vht_mcs_info));
231 
232 	/* copy EXT_NSS_BW Support value or remove the capability */
233 	if (ieee80211_hw_check(&sdata->local->hw, SUPPORTS_VHT_EXT_NSS_BW))
234 		vht_cap->cap |= (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK);
235 	else
236 		vht_cap->vht_mcs.tx_highest &=
237 			~cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE);
238 
239 	/* but also restrict MCSes */
240 	for (i = 0; i < 8; i++) {
241 		u16 own_rx, own_tx, peer_rx, peer_tx;
242 
243 		own_rx = le16_to_cpu(own_cap.vht_mcs.rx_mcs_map);
244 		own_rx = (own_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
245 
246 		own_tx = le16_to_cpu(own_cap.vht_mcs.tx_mcs_map);
247 		own_tx = (own_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
248 
249 		peer_rx = le16_to_cpu(vht_cap->vht_mcs.rx_mcs_map);
250 		peer_rx = (peer_rx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
251 
252 		peer_tx = le16_to_cpu(vht_cap->vht_mcs.tx_mcs_map);
253 		peer_tx = (peer_tx >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
254 
255 		if (peer_tx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
256 			if (own_rx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
257 				peer_tx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
258 			else if (own_rx < peer_tx)
259 				peer_tx = own_rx;
260 		}
261 
262 		if (peer_rx != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
263 			if (own_tx == IEEE80211_VHT_MCS_NOT_SUPPORTED)
264 				peer_rx = IEEE80211_VHT_MCS_NOT_SUPPORTED;
265 			else if (own_tx < peer_rx)
266 				peer_rx = own_tx;
267 		}
268 
269 		vht_cap->vht_mcs.rx_mcs_map &=
270 			~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
271 		vht_cap->vht_mcs.rx_mcs_map |= cpu_to_le16(peer_rx << i * 2);
272 
273 		vht_cap->vht_mcs.tx_mcs_map &=
274 			~cpu_to_le16(IEEE80211_VHT_MCS_NOT_SUPPORTED << i * 2);
275 		vht_cap->vht_mcs.tx_mcs_map |= cpu_to_le16(peer_tx << i * 2);
276 	}
277 
278 	/*
279 	 * This is a workaround for VHT-enabled STAs which break the spec
280 	 * and have the VHT-MCS Rx map filled in with value 3 for all eight
281 	 * spacial streams, an example is AR9462.
282 	 *
283 	 * As per spec, in section 22.1.1 Introduction to the VHT PHY
284 	 * A VHT STA shall support at least single spactial stream VHT-MCSs
285 	 * 0 to 7 (transmit and receive) in all supported channel widths.
286 	 */
287 	if (vht_cap->vht_mcs.rx_mcs_map == cpu_to_le16(0xFFFF)) {
288 		vht_cap->vht_supported = false;
289 		sdata_info(sdata, "Ignoring VHT IE from %pM due to invalid rx_mcs_map\n",
290 			   sta->addr);
291 		return;
292 	}
293 
294 	/* finally set up the bandwidth */
295 	switch (vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
296 	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
297 	case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
298 		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
299 		break;
300 	default:
301 		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
302 
303 		if (!(vht_cap->vht_mcs.tx_highest &
304 				cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE)))
305 			break;
306 
307 		/*
308 		 * If this is non-zero, then it does support 160 MHz after all,
309 		 * in one form or the other. We don't distinguish here (or even
310 		 * above) between 160 and 80+80 yet.
311 		 */
312 		if (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
313 			sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
314 	}
315 
316 	sta->sta.deflink.bandwidth = ieee80211_sta_cur_vht_bw(sta);
317 
318 	switch (vht_cap->cap & IEEE80211_VHT_CAP_MAX_MPDU_MASK) {
319 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
320 		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_11454;
321 		break;
322 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
323 		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_7991;
324 		break;
325 	case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
326 	default:
327 		sta->sta.max_amsdu_len = IEEE80211_MAX_MPDU_LEN_VHT_3895;
328 		break;
329 	}
330 }
331 
332 /* FIXME: move this to some better location - parses HE/EHT now */
ieee80211_sta_cap_rx_bw(struct sta_info * sta)333 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
334 {
335 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
336 	struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
337 	struct ieee80211_sta_eht_cap *eht_cap = &sta->sta.deflink.eht_cap;
338 	u32 cap_width;
339 
340 	if (he_cap->has_he) {
341 		u8 info;
342 
343 		if (eht_cap->has_eht &&
344 		    sta->sdata->vif.bss_conf.chandef.chan->band ==
345 		    NL80211_BAND_6GHZ) {
346 			info = eht_cap->eht_cap_elem.phy_cap_info[0];
347 
348 			if (info & IEEE80211_EHT_PHY_CAP0_320MHZ_IN_6GHZ)
349 				return IEEE80211_STA_RX_BW_320;
350 		}
351 
352 		info = he_cap->he_cap_elem.phy_cap_info[0];
353 
354 		if (sta->sdata->vif.bss_conf.chandef.chan->band ==
355 				NL80211_BAND_2GHZ) {
356 			if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_IN_2G)
357 				return IEEE80211_STA_RX_BW_40;
358 			else
359 				return IEEE80211_STA_RX_BW_20;
360 		}
361 
362 		if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G ||
363 		    info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_80PLUS80_MHZ_IN_5G)
364 			return IEEE80211_STA_RX_BW_160;
365 		else if (info & IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_40MHZ_80MHZ_IN_5G)
366 			return IEEE80211_STA_RX_BW_80;
367 
368 		return IEEE80211_STA_RX_BW_20;
369 	}
370 
371 	if (!vht_cap->vht_supported)
372 		return sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
373 				IEEE80211_STA_RX_BW_40 :
374 				IEEE80211_STA_RX_BW_20;
375 
376 	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
377 
378 	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ ||
379 	    cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
380 		return IEEE80211_STA_RX_BW_160;
381 
382 	/*
383 	 * If this is non-zero, then it does support 160 MHz after all,
384 	 * in one form or the other. We don't distinguish here (or even
385 	 * above) between 160 and 80+80 yet.
386 	 */
387 	if (vht_cap->cap & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
388 		return IEEE80211_STA_RX_BW_160;
389 
390 	return IEEE80211_STA_RX_BW_80;
391 }
392 
ieee80211_sta_cap_chan_bw(struct sta_info * sta)393 enum nl80211_chan_width ieee80211_sta_cap_chan_bw(struct sta_info *sta)
394 {
395 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
396 	u32 cap_width;
397 
398 	if (!vht_cap->vht_supported) {
399 		if (!sta->sta.deflink.ht_cap.ht_supported)
400 			return NL80211_CHAN_WIDTH_20_NOHT;
401 
402 		return sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40 ?
403 				NL80211_CHAN_WIDTH_40 : NL80211_CHAN_WIDTH_20;
404 	}
405 
406 	cap_width = vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
407 
408 	if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
409 		return NL80211_CHAN_WIDTH_160;
410 	else if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ)
411 		return NL80211_CHAN_WIDTH_80P80;
412 
413 	return NL80211_CHAN_WIDTH_80;
414 }
415 
416 enum nl80211_chan_width
ieee80211_sta_rx_bw_to_chan_width(struct sta_info * sta)417 ieee80211_sta_rx_bw_to_chan_width(struct sta_info *sta)
418 {
419 	enum ieee80211_sta_rx_bandwidth cur_bw = sta->sta.deflink.bandwidth;
420 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.deflink.vht_cap;
421 	u32 cap_width;
422 
423 	switch (cur_bw) {
424 	case IEEE80211_STA_RX_BW_20:
425 		if (!sta->sta.deflink.ht_cap.ht_supported)
426 			return NL80211_CHAN_WIDTH_20_NOHT;
427 		else
428 			return NL80211_CHAN_WIDTH_20;
429 	case IEEE80211_STA_RX_BW_40:
430 		return NL80211_CHAN_WIDTH_40;
431 	case IEEE80211_STA_RX_BW_80:
432 		return NL80211_CHAN_WIDTH_80;
433 	case IEEE80211_STA_RX_BW_160:
434 		cap_width =
435 			vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
436 
437 		if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
438 			return NL80211_CHAN_WIDTH_160;
439 
440 		return NL80211_CHAN_WIDTH_80P80;
441 	default:
442 		return NL80211_CHAN_WIDTH_20;
443 	}
444 }
445 
446 enum ieee80211_sta_rx_bandwidth
ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)447 ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
448 {
449 	switch (width) {
450 	case NL80211_CHAN_WIDTH_20_NOHT:
451 	case NL80211_CHAN_WIDTH_20:
452 		return IEEE80211_STA_RX_BW_20;
453 	case NL80211_CHAN_WIDTH_40:
454 		return IEEE80211_STA_RX_BW_40;
455 	case NL80211_CHAN_WIDTH_80:
456 		return IEEE80211_STA_RX_BW_80;
457 	case NL80211_CHAN_WIDTH_160:
458 	case NL80211_CHAN_WIDTH_80P80:
459 		return IEEE80211_STA_RX_BW_160;
460 	case NL80211_CHAN_WIDTH_320:
461 		return IEEE80211_STA_RX_BW_320;
462 	default:
463 		WARN_ON_ONCE(1);
464 		return IEEE80211_STA_RX_BW_20;
465 	}
466 }
467 
468 /* FIXME: rename/move - this deals with everything not just VHT */
ieee80211_sta_cur_vht_bw(struct sta_info * sta)469 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
470 {
471 	struct ieee80211_sub_if_data *sdata = sta->sdata;
472 	enum ieee80211_sta_rx_bandwidth bw;
473 	enum nl80211_chan_width bss_width = sdata->vif.bss_conf.chandef.width;
474 
475 	bw = ieee80211_sta_cap_rx_bw(sta);
476 	bw = min(bw, sta->deflink.cur_max_bandwidth);
477 
478 	/* Don't consider AP's bandwidth for TDLS peers, section 11.23.1 of
479 	 * IEEE80211-2016 specification makes higher bandwidth operation
480 	 * possible on the TDLS link if the peers have wider bandwidth
481 	 * capability.
482 	 *
483 	 * However, in this case, and only if the TDLS peer is authorized,
484 	 * limit to the tdls_chandef so that the configuration here isn't
485 	 * wider than what's actually requested on the channel context.
486 	 */
487 	if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
488 	    test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW) &&
489 	    test_sta_flag(sta, WLAN_STA_AUTHORIZED) &&
490 	    sta->tdls_chandef.chan)
491 		bw = min(bw, ieee80211_chan_width_to_rx_bw(sta->tdls_chandef.width));
492 	else
493 		bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
494 
495 	return bw;
496 }
497 
ieee80211_sta_set_rx_nss(struct sta_info * sta)498 void ieee80211_sta_set_rx_nss(struct sta_info *sta)
499 {
500 	u8 ht_rx_nss = 0, vht_rx_nss = 0, he_rx_nss = 0, eht_rx_nss = 0, rx_nss;
501 	bool support_160;
502 
503 	/* if we received a notification already don't overwrite it */
504 	if (sta->sta.deflink.rx_nss)
505 		return;
506 
507 	if (sta->sta.deflink.eht_cap.has_eht) {
508 		int i;
509 		const u8 *rx_nss_mcs = (void *)&sta->sta.deflink.eht_cap.eht_mcs_nss_supp;
510 
511 		/* get the max nss for EHT over all possible bandwidths and mcs */
512 		for (i = 0; i < sizeof(struct ieee80211_eht_mcs_nss_supp); i++)
513 			eht_rx_nss = max_t(u8, eht_rx_nss,
514 					   u8_get_bits(rx_nss_mcs[i],
515 						       IEEE80211_EHT_MCS_NSS_RX));
516 	}
517 
518 	if (sta->sta.deflink.he_cap.has_he) {
519 		int i;
520 		u8 rx_mcs_80 = 0, rx_mcs_160 = 0;
521 		const struct ieee80211_sta_he_cap *he_cap = &sta->sta.deflink.he_cap;
522 		u16 mcs_160_map =
523 			le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_160);
524 		u16 mcs_80_map = le16_to_cpu(he_cap->he_mcs_nss_supp.rx_mcs_80);
525 
526 		for (i = 7; i >= 0; i--) {
527 			u8 mcs_160 = (mcs_160_map >> (2 * i)) & 3;
528 
529 			if (mcs_160 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
530 				rx_mcs_160 = i + 1;
531 				break;
532 			}
533 		}
534 		for (i = 7; i >= 0; i--) {
535 			u8 mcs_80 = (mcs_80_map >> (2 * i)) & 3;
536 
537 			if (mcs_80 != IEEE80211_HE_MCS_NOT_SUPPORTED) {
538 				rx_mcs_80 = i + 1;
539 				break;
540 			}
541 		}
542 
543 		support_160 = he_cap->he_cap_elem.phy_cap_info[0] &
544 			      IEEE80211_HE_PHY_CAP0_CHANNEL_WIDTH_SET_160MHZ_IN_5G;
545 
546 		if (support_160)
547 			he_rx_nss = min(rx_mcs_80, rx_mcs_160);
548 		else
549 			he_rx_nss = rx_mcs_80;
550 	}
551 
552 	if (sta->sta.deflink.ht_cap.ht_supported) {
553 		if (sta->sta.deflink.ht_cap.mcs.rx_mask[0])
554 			ht_rx_nss++;
555 		if (sta->sta.deflink.ht_cap.mcs.rx_mask[1])
556 			ht_rx_nss++;
557 		if (sta->sta.deflink.ht_cap.mcs.rx_mask[2])
558 			ht_rx_nss++;
559 		if (sta->sta.deflink.ht_cap.mcs.rx_mask[3])
560 			ht_rx_nss++;
561 		/* FIXME: consider rx_highest? */
562 	}
563 
564 	if (sta->sta.deflink.vht_cap.vht_supported) {
565 		int i;
566 		u16 rx_mcs_map;
567 
568 		rx_mcs_map = le16_to_cpu(sta->sta.deflink.vht_cap.vht_mcs.rx_mcs_map);
569 
570 		for (i = 7; i >= 0; i--) {
571 			u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
572 
573 			if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
574 				vht_rx_nss = i + 1;
575 				break;
576 			}
577 		}
578 		/* FIXME: consider rx_highest? */
579 	}
580 
581 	rx_nss = max(vht_rx_nss, ht_rx_nss);
582 	rx_nss = max(he_rx_nss, rx_nss);
583 	rx_nss = max(eht_rx_nss, rx_nss);
584 	sta->sta.deflink.rx_nss = max_t(u8, 1, rx_nss);
585 }
586 
__ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,u8 opmode,enum nl80211_band band)587 u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
588 				  struct sta_info *sta, u8 opmode,
589 				  enum nl80211_band band)
590 {
591 	enum ieee80211_sta_rx_bandwidth new_bw;
592 	struct sta_opmode_info sta_opmode = {};
593 	u32 changed = 0;
594 	u8 nss;
595 
596 	/* ignore - no support for BF yet */
597 	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
598 		return 0;
599 
600 	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
601 	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
602 	nss += 1;
603 
604 	if (sta->sta.deflink.rx_nss != nss) {
605 		sta->sta.deflink.rx_nss = nss;
606 		sta_opmode.rx_nss = nss;
607 		changed |= IEEE80211_RC_NSS_CHANGED;
608 		sta_opmode.changed |= STA_OPMODE_N_SS_CHANGED;
609 	}
610 
611 	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
612 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
613 		/* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
614 		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
615 		break;
616 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
617 		/* ignore IEEE80211_OPMODE_NOTIF_BW_160_80P80 must not be set */
618 		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
619 		break;
620 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
621 		if (opmode & IEEE80211_OPMODE_NOTIF_BW_160_80P80)
622 			sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
623 		else
624 			sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
625 		break;
626 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
627 		/* legacy only, no longer used by newer spec */
628 		sta->deflink.cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
629 		break;
630 	}
631 
632 	new_bw = ieee80211_sta_cur_vht_bw(sta);
633 	if (new_bw != sta->sta.deflink.bandwidth) {
634 		sta->sta.deflink.bandwidth = new_bw;
635 		sta_opmode.bw = ieee80211_sta_rx_bw_to_chan_width(sta);
636 		changed |= IEEE80211_RC_BW_CHANGED;
637 		sta_opmode.changed |= STA_OPMODE_MAX_BW_CHANGED;
638 	}
639 
640 	if (sta_opmode.changed)
641 		cfg80211_sta_opmode_change_notify(sdata->dev, sta->addr,
642 						  &sta_opmode, GFP_KERNEL);
643 
644 	return changed;
645 }
646 
ieee80211_process_mu_groups(struct ieee80211_sub_if_data * sdata,struct ieee80211_mgmt * mgmt)647 void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
648 				 struct ieee80211_mgmt *mgmt)
649 {
650 	struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
651 
652 	if (!sdata->vif.bss_conf.mu_mimo_owner)
653 		return;
654 
655 	if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
656 		    bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
657 	    !memcmp(mgmt->u.action.u.vht_group_notif.membership,
658 		    bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
659 		return;
660 
661 	memcpy(bss_conf->mu_group.membership,
662 	       mgmt->u.action.u.vht_group_notif.membership,
663 	       WLAN_MEMBERSHIP_LEN);
664 	memcpy(bss_conf->mu_group.position,
665 	       mgmt->u.action.u.vht_group_notif.position,
666 	       WLAN_USER_POSITION_LEN);
667 
668 	ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_MU_GROUPS);
669 }
670 
ieee80211_update_mu_groups(struct ieee80211_vif * vif,const u8 * membership,const u8 * position)671 void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
672 				const u8 *membership, const u8 *position)
673 {
674 	struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
675 
676 	if (WARN_ON_ONCE(!vif->bss_conf.mu_mimo_owner))
677 		return;
678 
679 	memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
680 	memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
681 }
682 EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
683 
ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,u8 opmode,enum nl80211_band band)684 void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
685 				 struct sta_info *sta, u8 opmode,
686 				 enum nl80211_band band)
687 {
688 	struct ieee80211_local *local = sdata->local;
689 	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
690 
691 	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);
692 
693 	if (changed > 0) {
694 		ieee80211_recalc_min_chandef(sdata);
695 		rate_control_rate_update(local, sband, sta, changed);
696 	}
697 }
698 
ieee80211_get_vht_mask_from_cap(__le16 vht_cap,u16 vht_mask[NL80211_VHT_NSS_MAX])699 void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
700 				     u16 vht_mask[NL80211_VHT_NSS_MAX])
701 {
702 	int i;
703 	u16 mask, cap = le16_to_cpu(vht_cap);
704 
705 	for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
706 		mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
707 		switch (mask) {
708 		case IEEE80211_VHT_MCS_SUPPORT_0_7:
709 			vht_mask[i] = 0x00FF;
710 			break;
711 		case IEEE80211_VHT_MCS_SUPPORT_0_8:
712 			vht_mask[i] = 0x01FF;
713 			break;
714 		case IEEE80211_VHT_MCS_SUPPORT_0_9:
715 			vht_mask[i] = 0x03FF;
716 			break;
717 		case IEEE80211_VHT_MCS_NOT_SUPPORTED:
718 		default:
719 			vht_mask[i] = 0;
720 			break;
721 		}
722 	}
723 }
724