1 /*
2  * VHT handling
3  *
4  * Portions of this file
5  * Copyright(c) 2015 - 2016 Intel Deutschland GmbH
6  *
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"
16 #include "rate.h"
17 
18 
__check_vhtcap_disable(struct ieee80211_sub_if_data * sdata,struct ieee80211_sta_vht_cap * vht_cap,u32 flag)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 
ieee80211_apply_vhtcap_overrides(struct ieee80211_sub_if_data * sdata,struct ieee80211_sta_vht_cap * vht_cap)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 
117 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)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)
122 {
123 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
124 	struct ieee80211_sta_vht_cap own_cap;
125 	u32 cap_info, i;
126 	bool have_80mhz;
127 
128 	memset(vht_cap, 0, sizeof(*vht_cap));
129 
130 	if (!sta->sta.ht_cap.ht_supported)
131 		return;
132 
133 	if (!vht_cap_ie || !sband->vht_cap.vht_supported)
134 		return;
135 
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 
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 	 */
156 
157 	vht_cap->vht_supported = true;
158 
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 */
206 	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE)
207 		vht_cap->cap |= cap_info &
208 				(IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE |
209 				 IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK);
210 
211 	if (own_cap.cap & IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE)
212 		vht_cap->cap |= cap_info &
213 				(IEEE80211_VHT_CAP_SU_BEAMFORMEE_CAPABLE |
214 				 IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK);
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;
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));
233 
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 
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 
289 	/* finally set up the bandwidth */
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 
299 	sta->sta.bandwidth = ieee80211_sta_cur_vht_bw(sta);
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 	}
317 }
318 
ieee80211_sta_cap_rx_bw(struct sta_info * sta)319 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cap_rx_bw(struct sta_info *sta)
320 {
321 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
322 	u32 cap_width;
323 
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;
328 
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 
ieee80211_sta_cap_chan_bw(struct sta_info * sta)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 nl80211_chan_width
ieee80211_sta_rx_bw_to_chan_width(struct sta_info * sta)362 ieee80211_sta_rx_bw_to_chan_width(struct sta_info *sta)
363 {
364 	enum ieee80211_sta_rx_bandwidth cur_bw = sta->sta.bandwidth;
365 	struct ieee80211_sta_vht_cap *vht_cap = &sta->sta.vht_cap;
366 	u32 cap_width;
367 
368 	switch (cur_bw) {
369 	case IEEE80211_STA_RX_BW_20:
370 		if (!sta->sta.ht_cap.ht_supported)
371 			return NL80211_CHAN_WIDTH_20_NOHT;
372 		else
373 			return NL80211_CHAN_WIDTH_20;
374 	case IEEE80211_STA_RX_BW_40:
375 		return NL80211_CHAN_WIDTH_40;
376 	case IEEE80211_STA_RX_BW_80:
377 		return NL80211_CHAN_WIDTH_80;
378 	case IEEE80211_STA_RX_BW_160:
379 		cap_width =
380 			vht_cap->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK;
381 
382 		if (cap_width == IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ)
383 			return NL80211_CHAN_WIDTH_160;
384 
385 		return NL80211_CHAN_WIDTH_80P80;
386 	default:
387 		return NL80211_CHAN_WIDTH_20;
388 	}
389 }
390 
391 enum ieee80211_sta_rx_bandwidth
ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)392 ieee80211_chan_width_to_rx_bw(enum nl80211_chan_width width)
393 {
394 	switch (width) {
395 	case NL80211_CHAN_WIDTH_20_NOHT:
396 	case NL80211_CHAN_WIDTH_20:
397 		return IEEE80211_STA_RX_BW_20;
398 	case NL80211_CHAN_WIDTH_40:
399 		return IEEE80211_STA_RX_BW_40;
400 	case NL80211_CHAN_WIDTH_80:
401 		return IEEE80211_STA_RX_BW_80;
402 	case NL80211_CHAN_WIDTH_160:
403 	case NL80211_CHAN_WIDTH_80P80:
404 		return IEEE80211_STA_RX_BW_160;
405 	default:
406 		WARN_ON_ONCE(1);
407 		return IEEE80211_STA_RX_BW_20;
408 	}
409 }
410 
ieee80211_sta_cur_vht_bw(struct sta_info * sta)411 enum ieee80211_sta_rx_bandwidth ieee80211_sta_cur_vht_bw(struct sta_info *sta)
412 {
413 	struct ieee80211_sub_if_data *sdata = sta->sdata;
414 	enum ieee80211_sta_rx_bandwidth bw;
415 	enum nl80211_chan_width bss_width = sdata->vif.bss_conf.chandef.width;
416 
417 	bw = ieee80211_sta_cap_rx_bw(sta);
418 	bw = min(bw, sta->cur_max_bandwidth);
419 
420 	/* Don't consider AP's bandwidth for TDLS peers, section 11.23.1 of
421 	 * IEEE80211-2016 specification makes higher bandwidth operation
422 	 * possible on the TDLS link if the peers have wider bandwidth
423 	 * capability.
424 	 */
425 	if (test_sta_flag(sta, WLAN_STA_TDLS_PEER) &&
426 	    test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW))
427 		return bw;
428 
429 	bw = min(bw, ieee80211_chan_width_to_rx_bw(bss_width));
430 
431 	return bw;
432 }
433 
ieee80211_sta_set_rx_nss(struct sta_info * sta)434 void ieee80211_sta_set_rx_nss(struct sta_info *sta)
435 {
436 	u8 ht_rx_nss = 0, vht_rx_nss = 0;
437 
438 	/* if we received a notification already don't overwrite it */
439 	if (sta->sta.rx_nss)
440 		return;
441 
442 	if (sta->sta.ht_cap.ht_supported) {
443 		if (sta->sta.ht_cap.mcs.rx_mask[0])
444 			ht_rx_nss++;
445 		if (sta->sta.ht_cap.mcs.rx_mask[1])
446 			ht_rx_nss++;
447 		if (sta->sta.ht_cap.mcs.rx_mask[2])
448 			ht_rx_nss++;
449 		if (sta->sta.ht_cap.mcs.rx_mask[3])
450 			ht_rx_nss++;
451 		/* FIXME: consider rx_highest? */
452 	}
453 
454 	if (sta->sta.vht_cap.vht_supported) {
455 		int i;
456 		u16 rx_mcs_map;
457 
458 		rx_mcs_map = le16_to_cpu(sta->sta.vht_cap.vht_mcs.rx_mcs_map);
459 
460 		for (i = 7; i >= 0; i--) {
461 			u8 mcs = (rx_mcs_map >> (2 * i)) & 3;
462 
463 			if (mcs != IEEE80211_VHT_MCS_NOT_SUPPORTED) {
464 				vht_rx_nss = i + 1;
465 				break;
466 			}
467 		}
468 		/* FIXME: consider rx_highest? */
469 	}
470 
471 	ht_rx_nss = max(ht_rx_nss, vht_rx_nss);
472 	sta->sta.rx_nss = max_t(u8, 1, ht_rx_nss);
473 }
474 
__ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,u8 opmode,enum nl80211_band band)475 u32 __ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
476 				  struct sta_info *sta, u8 opmode,
477 				  enum nl80211_band band)
478 {
479 	enum ieee80211_sta_rx_bandwidth new_bw;
480 	struct sta_opmode_info sta_opmode = {};
481 	u32 changed = 0;
482 	u8 nss;
483 
484 	/* ignore - no support for BF yet */
485 	if (opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_TYPE_BF)
486 		return 0;
487 
488 	nss = opmode & IEEE80211_OPMODE_NOTIF_RX_NSS_MASK;
489 	nss >>= IEEE80211_OPMODE_NOTIF_RX_NSS_SHIFT;
490 	nss += 1;
491 
492 	if (sta->sta.rx_nss != nss) {
493 		sta->sta.rx_nss = nss;
494 		sta_opmode.rx_nss = nss;
495 		changed |= IEEE80211_RC_NSS_CHANGED;
496 		sta_opmode.changed |= STA_OPMODE_N_SS_CHANGED;
497 	}
498 
499 	switch (opmode & IEEE80211_OPMODE_NOTIF_CHANWIDTH_MASK) {
500 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_20MHZ:
501 		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_20;
502 		break;
503 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_40MHZ:
504 		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_40;
505 		break;
506 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_80MHZ:
507 		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
508 		break;
509 	case IEEE80211_OPMODE_NOTIF_CHANWIDTH_160MHZ:
510 		sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
511 		break;
512 	}
513 
514 	new_bw = ieee80211_sta_cur_vht_bw(sta);
515 	if (new_bw != sta->sta.bandwidth) {
516 		sta->sta.bandwidth = new_bw;
517 		sta_opmode.bw = ieee80211_sta_rx_bw_to_chan_width(sta);
518 		changed |= IEEE80211_RC_BW_CHANGED;
519 		sta_opmode.changed |= STA_OPMODE_MAX_BW_CHANGED;
520 	}
521 
522 	if (sta_opmode.changed)
523 		cfg80211_sta_opmode_change_notify(sdata->dev, sta->addr,
524 						  &sta_opmode, GFP_KERNEL);
525 
526 	return changed;
527 }
528 
ieee80211_process_mu_groups(struct ieee80211_sub_if_data * sdata,struct ieee80211_mgmt * mgmt)529 void ieee80211_process_mu_groups(struct ieee80211_sub_if_data *sdata,
530 				 struct ieee80211_mgmt *mgmt)
531 {
532 	struct ieee80211_bss_conf *bss_conf = &sdata->vif.bss_conf;
533 
534 	if (!sdata->vif.mu_mimo_owner)
535 		return;
536 
537 	if (!memcmp(mgmt->u.action.u.vht_group_notif.position,
538 		    bss_conf->mu_group.position, WLAN_USER_POSITION_LEN) &&
539 	    !memcmp(mgmt->u.action.u.vht_group_notif.membership,
540 		    bss_conf->mu_group.membership, WLAN_MEMBERSHIP_LEN))
541 		return;
542 
543 	memcpy(bss_conf->mu_group.membership,
544 	       mgmt->u.action.u.vht_group_notif.membership,
545 	       WLAN_MEMBERSHIP_LEN);
546 	memcpy(bss_conf->mu_group.position,
547 	       mgmt->u.action.u.vht_group_notif.position,
548 	       WLAN_USER_POSITION_LEN);
549 
550 	ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_MU_GROUPS);
551 }
552 
ieee80211_update_mu_groups(struct ieee80211_vif * vif,const u8 * membership,const u8 * position)553 void ieee80211_update_mu_groups(struct ieee80211_vif *vif,
554 				const u8 *membership, const u8 *position)
555 {
556 	struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
557 
558 	if (WARN_ON_ONCE(!vif->mu_mimo_owner))
559 		return;
560 
561 	memcpy(bss_conf->mu_group.membership, membership, WLAN_MEMBERSHIP_LEN);
562 	memcpy(bss_conf->mu_group.position, position, WLAN_USER_POSITION_LEN);
563 }
564 EXPORT_SYMBOL_GPL(ieee80211_update_mu_groups);
565 
ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data * sdata,struct sta_info * sta,u8 opmode,enum nl80211_band band)566 void ieee80211_vht_handle_opmode(struct ieee80211_sub_if_data *sdata,
567 				 struct sta_info *sta, u8 opmode,
568 				 enum nl80211_band band)
569 {
570 	struct ieee80211_local *local = sdata->local;
571 	struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
572 
573 	u32 changed = __ieee80211_vht_handle_opmode(sdata, sta, opmode, band);
574 
575 	if (changed > 0) {
576 		ieee80211_recalc_min_chandef(sdata);
577 		rate_control_rate_update(local, sband, sta, changed);
578 	}
579 }
580 
ieee80211_get_vht_mask_from_cap(__le16 vht_cap,u16 vht_mask[NL80211_VHT_NSS_MAX])581 void ieee80211_get_vht_mask_from_cap(__le16 vht_cap,
582 				     u16 vht_mask[NL80211_VHT_NSS_MAX])
583 {
584 	int i;
585 	u16 mask, cap = le16_to_cpu(vht_cap);
586 
587 	for (i = 0; i < NL80211_VHT_NSS_MAX; i++) {
588 		mask = (cap >> i * 2) & IEEE80211_VHT_MCS_NOT_SUPPORTED;
589 		switch (mask) {
590 		case IEEE80211_VHT_MCS_SUPPORT_0_7:
591 			vht_mask[i] = 0x00FF;
592 			break;
593 		case IEEE80211_VHT_MCS_SUPPORT_0_8:
594 			vht_mask[i] = 0x01FF;
595 			break;
596 		case IEEE80211_VHT_MCS_SUPPORT_0_9:
597 			vht_mask[i] = 0x03FF;
598 			break;
599 		case IEEE80211_VHT_MCS_NOT_SUPPORTED:
600 		default:
601 			vht_mask[i] = 0;
602 			break;
603 		}
604 	}
605 }
606