xref: /OK3568_Linux_fs/external/rkwifibt/drivers/rtl8852bs/core/rtw_wlan_util.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1 /******************************************************************************
2  *
3  * Copyright(c) 2007 - 2021 Realtek Corporation.
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of version 2 of the GNU General Public License as
7  * published by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
12  * more details.
13  *
14  *****************************************************************************/
15 #define _RTW_WLAN_UTIL_C_
16 
17 #include <drv_types.h>
18 
19 unsigned char ARTHEROS_OUI1[] = {0x00, 0x03, 0x7f};
20 unsigned char ARTHEROS_OUI2[] = {0x00, 0x13, 0x74};
21 
22 unsigned char BROADCOM_OUI1[] = {0x00, 0x10, 0x18};
23 unsigned char BROADCOM_OUI2[] = {0x00, 0x0a, 0xf7};
24 unsigned char BROADCOM_OUI3[] = {0x00, 0x05, 0xb5};
25 
26 
27 unsigned char CISCO_OUI[] = {0x00, 0x40, 0x96};
28 unsigned char MARVELL_OUI[] = {0x00, 0x50, 0x43};
29 unsigned char RALINK_OUI[] = {0x00, 0x0c, 0x43};
30 unsigned char REALTEK_OUI[] = {0x00, 0xe0, 0x4c};
31 unsigned char AIRGOCAP_OUI[] = {0x00, 0x0a, 0xf5};
32 
33 unsigned char REALTEK_96B_IE[] = {0x00, 0xe0, 0x4c, 0x02, 0x01, 0x20};
34 
35 extern unsigned char RTW_WPA_OUI[];
36 extern unsigned char WPA_TKIP_CIPHER[4];
37 extern unsigned char RSN_TKIP_CIPHER[4];
38 
39 #define R2T_PHY_DELAY	(0)
40 
41 /* #define WAIT_FOR_BCN_TO_MIN	(3000) */
42 #define WAIT_FOR_BCN_TO_MIN	(6000)
43 #define WAIT_FOR_BCN_TO_MAX	(20000)
44 
45 static u8 rtw_basic_rate_cck[4] = {
46 	IEEE80211_CCK_RATE_1MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_2MB | IEEE80211_BASIC_RATE_MASK,
47 	IEEE80211_CCK_RATE_5MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_11MB | IEEE80211_BASIC_RATE_MASK
48 };
49 
50 static u8 rtw_basic_rate_ofdm[3] = {
51 	IEEE80211_OFDM_RATE_6MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_OFDM_RATE_12MB | IEEE80211_BASIC_RATE_MASK,
52 	IEEE80211_OFDM_RATE_24MB | IEEE80211_BASIC_RATE_MASK
53 };
54 
55 static u8 rtw_basic_rate_mix[7] = {
56 	IEEE80211_CCK_RATE_1MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_2MB | IEEE80211_BASIC_RATE_MASK,
57 	IEEE80211_CCK_RATE_5MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_11MB | IEEE80211_BASIC_RATE_MASK,
58 	IEEE80211_OFDM_RATE_6MB | IEEE80211_BASIC_RATE_MASK, IEEE80211_OFDM_RATE_12MB | IEEE80211_BASIC_RATE_MASK,
59 	IEEE80211_OFDM_RATE_24MB | IEEE80211_BASIC_RATE_MASK
60 };
61 
62 extern u8	WIFI_CCKRATES[];
rtw_is_cck_rate(u8 rate)63 bool rtw_is_cck_rate(u8 rate)
64 {
65 	int i;
66 
67 	for (i = 0; i < 4; i++)
68 		if ((WIFI_CCKRATES[i] & 0x7F) == (rate & 0x7F))
69 			return 1;
70 	return 0;
71 }
72 
73 extern u8	WIFI_OFDMRATES[];
rtw_is_ofdm_rate(u8 rate)74 bool rtw_is_ofdm_rate(u8 rate)
75 {
76 	int i;
77 
78 	for (i = 0; i < 8; i++)
79 		if ((WIFI_OFDMRATES[i] & 0x7F) == (rate & 0x7F))
80 			return 1;
81 	return 0;
82 }
83 
84 /* test if rate is defined in rtw_basic_rate_cck */
rtw_is_basic_rate_cck(u8 rate)85 bool rtw_is_basic_rate_cck(u8 rate)
86 {
87 	int i;
88 
89 	for (i = 0; i < 4; i++)
90 		if ((rtw_basic_rate_cck[i] & 0x7F) == (rate & 0x7F))
91 			return 1;
92 	return 0;
93 }
94 
95 /* test if rate is defined in rtw_basic_rate_ofdm */
rtw_is_basic_rate_ofdm(u8 rate)96 bool rtw_is_basic_rate_ofdm(u8 rate)
97 {
98 	int i;
99 
100 	for (i = 0; i < 3; i++)
101 		if ((rtw_basic_rate_ofdm[i] & 0x7F) == (rate & 0x7F))
102 			return 1;
103 	return 0;
104 }
105 
106 /* test if rate is defined in rtw_basic_rate_mix */
rtw_is_basic_rate_mix(u8 rate)107 bool rtw_is_basic_rate_mix(u8 rate)
108 {
109 	int i;
110 
111 	for (i = 0; i < 7; i++)
112 		if ((rtw_basic_rate_mix[i] & 0x7F) == (rate & 0x7F))
113 			return 1;
114 	return 0;
115 }
cckrates_included(unsigned char * rate,int ratelen)116 int cckrates_included(unsigned char *rate, int ratelen)
117 {
118 	int	i;
119 
120 	for (i = 0; i < ratelen; i++) {
121 		if ((((rate[i]) & 0x7f) == 2)	|| (((rate[i]) & 0x7f) == 4) ||
122 		    (((rate[i]) & 0x7f) == 11)  || (((rate[i]) & 0x7f) == 22))
123 			return _TRUE;
124 	}
125 
126 	return _FALSE;
127 
128 }
129 
cckratesonly_included(unsigned char * rate,int ratelen)130 int cckratesonly_included(unsigned char *rate, int ratelen)
131 {
132 	int	i;
133 
134 	for (i = 0; i < ratelen; i++) {
135 		if ((((rate[i]) & 0x7f) != 2) && (((rate[i]) & 0x7f) != 4) &&
136 		    (((rate[i]) & 0x7f) != 11)  && (((rate[i]) & 0x7f) != 22))
137 			return _FALSE;
138 	}
139 
140 	return _TRUE;
141 }
142 
rtw_get_sta_rx_nss(_adapter * adapter,struct sta_info * psta)143 s8 rtw_get_sta_rx_nss(_adapter *adapter, struct sta_info *psta)
144 {
145 	s8 nss = 1;
146 	struct dvobj_priv *dvobj = adapter_to_dvobj(adapter);
147 
148 	if (!psta)
149 		return nss;
150 
151 	nss = get_phy_rx_nss(adapter);
152 
153 #ifdef CONFIG_80211N_HT
154 	#ifdef CONFIG_80211AC_VHT
155 	#ifdef CONFIG_80211AX_HE
156 	if (psta->hepriv.he_option)
157 		nss = psta->phl_sta->asoc_cap.nss_tx;
158 	else
159 	#endif /* CONFIG_80211AX_HE */
160 	if (psta->vhtpriv.vht_option)
161 		nss = rtw_min(nss, rtw_vht_mcsmap_to_nss(psta->vhtpriv.vht_mcs_map));
162 	else
163 	#endif /* CONFIG_80211AC_VHT */
164 	if (psta->htpriv.ht_option)
165 		nss = rtw_min(nss, rtw_ht_mcsset_to_nss(psta->htpriv.ht_cap.supp_mcs_set));
166 #endif /*CONFIG_80211N_HT*/
167 	RTW_INFO("%s: %d ss\n", __func__, nss);
168 	return nss;
169 }
170 
rtw_get_sta_tx_nss(_adapter * adapter,struct sta_info * psta)171 s8 rtw_get_sta_tx_nss(_adapter *adapter, struct sta_info *psta)
172 {
173 	s8 nss = 1;
174 
175 	if (!psta)
176 		return nss;
177 
178 	nss = get_phy_tx_nss(adapter);
179 
180 #ifdef CONFIG_80211N_HT
181 	#ifdef CONFIG_80211AC_VHT
182 	#ifdef CONFIG_80211AX_HE
183 	if (psta->hepriv.he_option)
184 		nss = psta->phl_sta->asoc_cap.nss_rx;
185 	else
186 	#endif /* CONFIG_80211AX_HE */
187 	if (psta->vhtpriv.vht_option)
188 		nss = rtw_min(nss, rtw_vht_mcsmap_to_nss(psta->vhtpriv.vht_mcs_map));
189 	else
190 	#endif /* CONFIG_80211AC_VHT */
191 	if (psta->htpriv.ht_option)
192 		nss = rtw_min(nss, rtw_ht_mcsset_to_nss(psta->htpriv.ht_cap.supp_mcs_set));
193 #endif /*CONFIG_80211N_HT*/
194 	RTW_INFO("%s: %d SS\n", __func__, nss);
195 	return nss;
196 }
197 
ratetbl_val_2wifirate(unsigned char rate)198 unsigned char ratetbl_val_2wifirate(unsigned char rate)
199 {
200 	unsigned char val = 0;
201 
202 	switch (rate & 0x7f) {
203 	case 0:
204 		val = IEEE80211_CCK_RATE_1MB;
205 		break;
206 
207 	case 1:
208 		val = IEEE80211_CCK_RATE_2MB;
209 		break;
210 
211 	case 2:
212 		val = IEEE80211_CCK_RATE_5MB;
213 		break;
214 
215 	case 3:
216 		val = IEEE80211_CCK_RATE_11MB;
217 		break;
218 
219 	case 4:
220 		val = IEEE80211_OFDM_RATE_6MB;
221 		break;
222 
223 	case 5:
224 		val = IEEE80211_OFDM_RATE_9MB;
225 		break;
226 
227 	case 6:
228 		val = IEEE80211_OFDM_RATE_12MB;
229 		break;
230 
231 	case 7:
232 		val = IEEE80211_OFDM_RATE_18MB;
233 		break;
234 
235 	case 8:
236 		val = IEEE80211_OFDM_RATE_24MB;
237 		break;
238 
239 	case 9:
240 		val = IEEE80211_OFDM_RATE_36MB;
241 		break;
242 
243 	case 10:
244 		val = IEEE80211_OFDM_RATE_48MB;
245 		break;
246 
247 	case 11:
248 		val = IEEE80211_OFDM_RATE_54MB;
249 		break;
250 
251 	}
252 
253 	return val;
254 
255 }
256 
is_basicrate(_adapter * padapter,unsigned char rate)257 int is_basicrate(_adapter *padapter, unsigned char rate)
258 {
259 	int i;
260 	unsigned char val;
261 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
262 
263 	for (i = 0; i < NumRates; i++) {
264 		val = pmlmeext->basicrate[i];
265 
266 		if ((val != 0xff) && (val != 0xfe)) {
267 			if (rate == ratetbl_val_2wifirate(val))
268 				return _TRUE;
269 		}
270 	}
271 
272 	return _FALSE;
273 }
274 
ratetbl2rateset(_adapter * padapter,unsigned char * rateset)275 unsigned int ratetbl2rateset(_adapter *padapter, unsigned char *rateset)
276 {
277 	int i;
278 	unsigned char rate;
279 	unsigned int	len = 0;
280 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
281 
282 	for (i = 0; i < NumRates; i++) {
283 		rate = pmlmeext->datarate[i];
284 
285 		if (rtw_get_oper_ch(padapter) > 14 && rate < _6M_RATE_) /*5G no support CCK rate*/
286 			continue;
287 
288 		switch (rate) {
289 		case 0xff:
290 			return len;
291 
292 		case 0xfe:
293 			continue;
294 
295 		default:
296 			rate = ratetbl_val_2wifirate(rate);
297 
298 			if (is_basicrate(padapter, rate) == _TRUE)
299 				rate |= IEEE80211_BASIC_RATE_MASK;
300 
301 			rateset[len] = rate;
302 			len++;
303 			break;
304 		}
305 	}
306 	return len;
307 }
308 
get_rate_set(_adapter * padapter,unsigned char * pbssrate,int * bssrate_len)309 void get_rate_set(_adapter *padapter, unsigned char *pbssrate, int *bssrate_len)
310 {
311 	unsigned char supportedrates[NumRates];
312 
313 	_rtw_memset(supportedrates, 0, NumRates);
314 	*bssrate_len = ratetbl2rateset(padapter, supportedrates);
315 	_rtw_memcpy(pbssrate, supportedrates, *bssrate_len);
316 }
317 
set_mcs_rate_by_mask(u8 * mcs_set,u32 mask)318 void set_mcs_rate_by_mask(u8 *mcs_set, u32 mask)
319 {
320 	u8 mcs_rate_1r = (u8)(mask & 0xff);
321 	u8 mcs_rate_2r = (u8)((mask >> 8) & 0xff);
322 	u8 mcs_rate_3r = (u8)((mask >> 16) & 0xff);
323 	u8 mcs_rate_4r = (u8)((mask >> 24) & 0xff);
324 
325 	mcs_set[0] &= mcs_rate_1r;
326 	mcs_set[1] &= mcs_rate_2r;
327 	mcs_set[2] &= mcs_rate_3r;
328 	mcs_set[3] &= mcs_rate_4r;
329 }
330 
UpdateBrateTbl(_adapter * adapter,u8 * mBratesOS)331 void UpdateBrateTbl(
332 	_adapter *adapter,
333 	u8			*mBratesOS
334 )
335 {
336 	u8	i;
337 	u8	rate;
338 
339 	/* 1M, 2M, 5.5M, 11M, 6M, 12M, 24M are mandatory. */
340 	for (i = 0; i < NDIS_802_11_LENGTH_RATES_EX; i++) {
341 		rate = mBratesOS[i] & 0x7f;
342 		switch (rate) {
343 		case IEEE80211_CCK_RATE_1MB:
344 		case IEEE80211_CCK_RATE_2MB:
345 		case IEEE80211_CCK_RATE_5MB:
346 		case IEEE80211_CCK_RATE_11MB:
347 		case IEEE80211_OFDM_RATE_6MB:
348 		case IEEE80211_OFDM_RATE_12MB:
349 		case IEEE80211_OFDM_RATE_24MB:
350 			mBratesOS[i] |= IEEE80211_BASIC_RATE_MASK;
351 			break;
352 		}
353 	}
354 
355 }
356 
UpdateBrateTblForSoftAP(u8 * bssrateset,u32 bssratelen)357 void UpdateBrateTblForSoftAP(u8 *bssrateset, u32 bssratelen)
358 {
359 	u8	i;
360 	u8	rate;
361 
362 	for (i = 0; i < bssratelen; i++) {
363 		rate = bssrateset[i] & 0x7f;
364 		switch (rate) {
365 		case IEEE80211_CCK_RATE_1MB:
366 		case IEEE80211_CCK_RATE_2MB:
367 		case IEEE80211_CCK_RATE_5MB:
368 		case IEEE80211_CCK_RATE_11MB:
369 			bssrateset[i] |= IEEE80211_BASIC_RATE_MASK;
370 			break;
371 		}
372 	}
373 
374 }
375 /*rtw_phl_mr_get_chandef(dvobj->phl, adapter->phl_role, &chandef); => mr union chan*/
376 /*rtw_phl_get_cur_hal_chdef(adapter->phl_role, &chandef) => hal chan*/
377 
rtw_get_oper_chdef(_adapter * adapter,struct rtw_chan_def * chandef)378 void rtw_get_oper_chdef(_adapter *adapter, struct rtw_chan_def *chandef)
379 {
380 	if (!adapter->phl_role)
381 		return;
382 
383 	if (rtw_phl_get_cur_hal_chdef(adapter->phl_role, chandef) != RTW_PHL_STATUS_SUCCESS)
384 		RTW_ERR("%s failed\n", __func__);
385 }
386 
rtw_get_oper_band(_adapter * adapter)387 u8 rtw_get_oper_band(_adapter *adapter)
388 {
389 	struct rtw_chan_def cur_chandef = {0};
390 
391 	rtw_get_oper_chdef(adapter, &cur_chandef);
392 	return cur_chandef.band;
393 }
394 
rtw_get_oper_ch(_adapter * adapter)395 u8 rtw_get_oper_ch(_adapter *adapter)
396 {
397 	struct rtw_chan_def cur_chandef = {0};
398 
399 	rtw_get_oper_chdef(adapter, &cur_chandef);
400 	return cur_chandef.chan;
401 }
402 
rtw_get_oper_bw(_adapter * adapter)403 u8 rtw_get_oper_bw(_adapter *adapter)
404 {
405 	struct rtw_chan_def cur_chandef = {0};
406 
407 	rtw_get_oper_chdef(adapter, &cur_chandef);
408 	return cur_chandef.bw;
409 }
410 
rtw_get_oper_choffset(_adapter * adapter)411 u8 rtw_get_oper_choffset(_adapter *adapter)
412 {
413 	struct rtw_chan_def cur_chandef = {0};
414 
415 	rtw_get_oper_chdef(adapter, &cur_chandef);
416 	return cur_chandef.offset;
417 }
418 
rtw_get_on_oper_ch_time(_adapter * adapter)419 inline systime rtw_get_on_oper_ch_time(_adapter *adapter)
420 {
421 	return adapter_to_dvobj(adapter)->on_oper_ch_time;
422 }
423 
rtw_get_on_cur_ch_time(_adapter * adapter)424 inline systime rtw_get_on_cur_ch_time(_adapter *adapter)
425 {
426 	if (adapter->mlmeextpriv.chandef.chan == rtw_get_oper_ch(adapter))
427 		return adapter_to_dvobj(adapter)->on_oper_ch_time;
428 	else
429 		return 0;
430 }
431 
set_channel_bwmode(_adapter * padapter,unsigned char channel,unsigned char channel_offset,unsigned short bwmode,u8 do_rfk)432 void set_channel_bwmode(_adapter *padapter,
433 				unsigned char channel,
434 				unsigned char channel_offset,
435 				unsigned short bwmode,
436 				u8 do_rfk)
437 {
438 	rtw_hw_set_ch_bw(padapter, channel, (enum channel_width)bwmode,
439 		channel_offset, do_rfk);
440 }
441 
get_my_bssid(WLAN_BSSID_EX * pnetwork)442 __inline u8 *get_my_bssid(WLAN_BSSID_EX *pnetwork)
443 {
444 	return pnetwork->MacAddress;
445 }
446 
get_beacon_interval(WLAN_BSSID_EX * bss)447 u16 get_beacon_interval(WLAN_BSSID_EX *bss)
448 {
449 	unsigned short val;
450 	_rtw_memcpy((unsigned char *)&val, rtw_get_beacon_interval_from_ie(bss->IEs), 2);
451 
452 	return le16_to_cpu(val);
453 
454 }
455 
is_client_associated_to_ap(_adapter * padapter)456 int is_client_associated_to_ap(_adapter *padapter)
457 {
458 	struct mlme_ext_priv	*pmlmeext;
459 	struct mlme_ext_info	*pmlmeinfo;
460 
461 	if (!padapter)
462 		return _FAIL;
463 
464 	pmlmeext = &padapter->mlmeextpriv;
465 	pmlmeinfo = &(pmlmeext->mlmext_info);
466 
467 	if ((pmlmeinfo->state & WIFI_FW_ASSOC_SUCCESS) && ((pmlmeinfo->state & 0x03) == WIFI_FW_STATION_STATE))
468 		return _TRUE;
469 	else
470 		return _FAIL;
471 }
472 
is_client_associated_to_ibss(_adapter * padapter)473 int is_client_associated_to_ibss(_adapter *padapter)
474 {
475 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
476 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
477 
478 	if ((pmlmeinfo->state & WIFI_FW_ASSOC_SUCCESS) && ((pmlmeinfo->state & 0x03) == WIFI_FW_ADHOC_STATE))
479 		return _TRUE;
480 	else
481 		return _FAIL;
482 }
483 
484 /*GEORGIA_TODO_FIXIT*/
485 #define GET_H2CCMD_MSRRPT_PARM_OPMODE(__pH2CCmd) 	LE_BITS_TO_1BYTE(((u8 *)(__pH2CCmd)), 0, 1)
486 #define GET_H2CCMD_MSRRPT_PARM_ROLE(__pH2CCmd)	LE_BITS_TO_1BYTE(((u8 *)(__pH2CCmd)), 4, 4)
487 
is_IBSS_empty(_adapter * padapter)488 int is_IBSS_empty(_adapter *padapter)
489 {
490 /* ToDo */
491 #if 0
492 	int i;
493 	struct macid_ctl_t *macid_ctl = &padapter->dvobj->macid_ctl;
494 
495 	for (i = 0; i < macid_ctl->num; i++) {
496 		if (!rtw_macid_is_used(macid_ctl, i))
497 			continue;
498 		if (!rtw_macid_is_iface_specific(macid_ctl, i, padapter))
499 			continue;
500 		if (!GET_H2CCMD_MSRRPT_PARM_OPMODE(&macid_ctl->h2c_msr[i]))
501 			continue;
502 		if (GET_H2CCMD_MSRRPT_PARM_ROLE(&macid_ctl->h2c_msr[i]) == H2C_MSR_ROLE_ADHOC)
503 			return _FAIL;
504 	}
505 #endif
506 	return _TRUE;
507 }
508 
decide_wait_for_beacon_timeout(unsigned int bcn_interval)509 unsigned int decide_wait_for_beacon_timeout(unsigned int bcn_interval)
510 {
511 	if ((bcn_interval << 2) < WAIT_FOR_BCN_TO_MIN)
512 		return WAIT_FOR_BCN_TO_MIN;
513 	else if ((bcn_interval << 2) > WAIT_FOR_BCN_TO_MAX)
514 		return WAIT_FOR_BCN_TO_MAX;
515 	else
516 		return bcn_interval << 2;
517 }
518 
519 #if defined(CONFIG_P2P) && defined(CONFIG_WFD)
rtw_process_wfd_ie(_adapter * adapter,u8 * wfd_ie,u8 wfd_ielen,const char * tag)520 void rtw_process_wfd_ie(_adapter *adapter, u8 *wfd_ie, u8 wfd_ielen, const char *tag)
521 {
522 	struct wifidirect_info *wdinfo = &adapter->wdinfo;
523 
524 	u8 *attr_content;
525 	u32 attr_contentlen = 0;
526 
527 	if (!rtw_hw_chk_wl_func(adapter_to_dvobj(adapter), WL_FUNC_MIRACAST))
528 		return;
529 
530 	RTW_INFO("[%s] Found WFD IE\n", tag);
531 	attr_content = rtw_get_wfd_attr_content(wfd_ie, wfd_ielen, WFD_ATTR_DEVICE_INFO, NULL, &attr_contentlen);
532 	if (attr_content && attr_contentlen) {
533 		wdinfo->wfd_info->peer_rtsp_ctrlport = RTW_GET_BE16(attr_content + 2);
534 		RTW_INFO("[%s] Peer PORT NUM = %d\n", tag, wdinfo->wfd_info->peer_rtsp_ctrlport);
535 	}
536 }
537 
rtw_process_wfd_ies(_adapter * adapter,u8 * ies,u8 ies_len,const char * tag)538 void rtw_process_wfd_ies(_adapter *adapter, u8 *ies, u8 ies_len, const char *tag)
539 {
540 	u8 *wfd_ie;
541 	u32	wfd_ielen;
542 
543 	if (!rtw_hw_chk_wl_func(adapter_to_dvobj(adapter), WL_FUNC_MIRACAST))
544 		return;
545 
546 	wfd_ie = rtw_get_wfd_ie(ies, ies_len, NULL, &wfd_ielen);
547 	while (wfd_ie) {
548 		rtw_process_wfd_ie(adapter, wfd_ie, wfd_ielen, tag);
549 		wfd_ie = rtw_get_wfd_ie(wfd_ie + wfd_ielen, (ies + ies_len) - (wfd_ie + wfd_ielen), NULL, &wfd_ielen);
550 	}
551 }
552 #endif /* defined(CONFIG_P2P) && defined(CONFIG_WFD) */
553 
WMM_param_handler(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)554 int WMM_param_handler(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs	pIE)
555 {
556 	/* struct registry_priv	*pregpriv = &padapter->registrypriv; */
557 	struct mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
558 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
559 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
560 
561 	if (pmlmepriv->qospriv.qos_option == 0) {
562 		pmlmeinfo->WMM_enable = 0;
563 		return _FALSE;
564 	}
565 
566 	if (_rtw_memcmp(&(pmlmeinfo->WMM_param), (pIE->data + 6), sizeof(struct WMM_para_element)))
567 		return _FALSE;
568 	else
569 		_rtw_memcpy(&(pmlmeinfo->WMM_param), (pIE->data + 6), sizeof(struct WMM_para_element));
570 	pmlmeinfo->WMM_enable = 1;
571 	return _TRUE;
572 
573 #if 0
574 	if (pregpriv->wifi_spec == 1) {
575 		if (pmlmeinfo->WMM_enable == 1) {
576 			/* todo: compare the parameter set count & decide wheher to update or not */
577 			return _FAIL;
578 		} else {
579 			pmlmeinfo->WMM_enable = 1;
580 			_rtw_rtw_memcpy(&(pmlmeinfo->WMM_param), (pIE->data + 6), sizeof(struct WMM_para_element));
581 			return _TRUE;
582 		}
583 	} else {
584 		pmlmeinfo->WMM_enable = 0;
585 		return _FAIL;
586 	}
587 #endif
588 
589 }
590 
591 #ifdef CONFIG_RTW_TOKEN_BASED_XMIT
rtw_is_tbtx_capabilty(u8 * p,u8 len)592 u8 rtw_is_tbtx_capabilty(u8 *p, u8 len){
593 	int i;
594 	u8 tbtx_cap_ie[8] = {0x00, 0xe0, 0x4c, 0x01, 0x00, 0x00, 0x00, 0x00};
595 
596 	for (i = 0; i < len; i++) {
597 		if (*(p + i) != tbtx_cap_ie[i])
598 			return _FALSE;
599 		else
600 			continue;
601 	}
602 	return _TRUE;
603 }
604 #endif
605 
WMMOnAssocRsp(_adapter * padapter)606 void WMMOnAssocRsp(_adapter *padapter)
607 {
608 	u8	ACI, ACM, AIFS, ECWMin, ECWMax, aSifsTime;
609 	u8	acm_mask;
610 	u16	TXOP;
611 	u32	acParm, i;
612 	u32	edca[4], inx[4];
613 	u8 ac_be = 0;
614 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
615 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
616 	struct xmit_priv		*pxmitpriv = &padapter->xmitpriv;
617 	struct registry_priv	*pregpriv = &padapter->registrypriv;
618 #ifdef CONFIG_WMMPS_STA
619 	struct mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
620 	struct qos_priv	*pqospriv = &pmlmepriv->qospriv;
621 #endif /* CONFIG_WMMPS_STA */
622 
623 	acm_mask = 0;
624 
625 	if (WIFI_ROLE_IS_ON_5G(padapter) ||
626 	    (pmlmeext->cur_wireless_mode & WLAN_MD_11N))
627 		aSifsTime = 16;
628 	else
629 		aSifsTime = 10;
630 
631 	if (pmlmeinfo->WMM_enable == 0) {
632 		padapter->mlmepriv.acm_mask = 0;
633 
634 		AIFS = aSifsTime + (2 * pmlmeinfo->slotTime);
635 
636 		if (pmlmeext->cur_wireless_mode & (WLAN_MD_11G | WLAN_MD_11A)) {
637 			ECWMin = 4;
638 			ECWMax = 10;
639 		} else if (pmlmeext->cur_wireless_mode & WLAN_MD_11B) {
640 			ECWMin = 5;
641 			ECWMax = 10;
642 		} else {
643 			ECWMin = 4;
644 			ECWMax = 10;
645 		}
646 
647 		TXOP = 0;
648 		acParm = AIFS | (ECWMin << 8) | (ECWMax << 12) | (TXOP << 16);
649 		rtw_hw_set_edca(padapter, 0, acParm);
650 		rtw_hw_set_edca(padapter, 1, acParm);
651 		rtw_hw_set_edca(padapter, 2, acParm);
652 
653 		ECWMin = 2;
654 		ECWMax = 3;
655 		TXOP = 0x2f;
656 		acParm = AIFS | (ECWMin << 8) | (ECWMax << 12) | (TXOP << 16);
657 		rtw_hw_set_edca(padapter, 3, acParm);
658 	} else {
659 		edca[0] = edca[1] = edca[2] = edca[3] = 0;
660 
661 		for (i = 0; i < 4; i++) {
662 			ACI = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN >> 5) & 0x03;
663 			ACM = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN >> 4) & 0x01;
664 
665 			/* AIFS = AIFSN * slot time + SIFS - r2t phy delay */
666 			AIFS = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN & 0x0f) * pmlmeinfo->slotTime + aSifsTime;
667 
668 			ECWMin = (pmlmeinfo->WMM_param.ac_param[i].CW & 0x0f);
669 			ECWMax = (pmlmeinfo->WMM_param.ac_param[i].CW & 0xf0) >> 4;
670 			TXOP = le16_to_cpu(pmlmeinfo->WMM_param.ac_param[i].TXOP_limit);
671 
672 			acParm = AIFS | (ECWMin << 8) | (ECWMax << 12) | (TXOP << 16);
673 			rtw_hw_set_edca(padapter, ACI, acParm);
674 
675 			switch (ACI) {
676 			case 0x0:
677 				acm_mask |= (ACM ? BIT(1) : 0);
678 				edca[XMIT_BE_QUEUE] = acParm;
679 				break;
680 
681 			case 0x1:
682 				/* acm_mask |= (ACM? BIT(0):0); */
683 				edca[XMIT_BK_QUEUE] = acParm;
684 				break;
685 
686 			case 0x2:
687 				acm_mask |= (ACM ? BIT(2) : 0);
688 				edca[XMIT_VI_QUEUE] = acParm;
689 				break;
690 
691 			case 0x3:
692 				acm_mask |= (ACM ? BIT(3) : 0);
693 				edca[XMIT_VO_QUEUE] = acParm;
694 				break;
695 			}
696 
697 			RTW_INFO("WMM(%x): %x, %x\n", ACI, ACM, acParm);
698 
699 			if (i == ac_be) {
700 				padapter->last_edca = acParm;
701 				acParm = rtw_get_turbo_edca(padapter, AIFS, ECWMin, ECWMax, TXOP);
702 				if (acParm) {
703 					rtw_hw_set_edca(padapter, ACI, acParm);
704 					padapter->last_edca = acParm;
705 				}
706 			}
707 
708 		}
709 
710 		if (padapter->registrypriv.acm_method == 1)
711 			rtw_hal_set_hwreg(padapter, HW_VAR_ACM_CTRL, (u8 *)(&acm_mask));
712 		else
713 			padapter->mlmepriv.acm_mask = acm_mask;
714 
715 		inx[0] = 0;
716 		inx[1] = 1;
717 		inx[2] = 2;
718 		inx[3] = 3;
719 
720 		if (pregpriv->wifi_spec == 1) {
721 			u32	j, tmp, change_inx = _FALSE;
722 
723 			/* entry indx: 0->vo, 1->vi, 2->be, 3->bk. */
724 			for (i = 0; i < 4; i++) {
725 				for (j = i + 1; j < 4; j++) {
726 					/* compare CW and AIFS */
727 					if ((edca[j] & 0xFFFF) < (edca[i] & 0xFFFF))
728 						change_inx = _TRUE;
729 					else if ((edca[j] & 0xFFFF) == (edca[i] & 0xFFFF)) {
730 						/* compare TXOP */
731 						if ((edca[j] >> 16) > (edca[i] >> 16))
732 							change_inx = _TRUE;
733 					}
734 
735 					if (change_inx) {
736 						tmp = edca[i];
737 						edca[i] = edca[j];
738 						edca[j] = tmp;
739 
740 						tmp = inx[i];
741 						inx[i] = inx[j];
742 						inx[j] = tmp;
743 
744 						change_inx = _FALSE;
745 					}
746 				}
747 			}
748 		}
749 
750 		for (i = 0; i < 4; i++) {
751 			pxmitpriv->wmm_para_seq[i] = inx[i];
752 			RTW_INFO("wmm_para_seq(%d): %d\n", i, pxmitpriv->wmm_para_seq[i]);
753 		}
754 
755 #ifdef CONFIG_WMMPS_STA
756 		/* if AP supports UAPSD function, driver must set each uapsd TID to coresponding mac register 0x693 */
757 		if (pmlmeinfo->WMM_param.QoS_info & AP_SUPPORTED_UAPSD) {
758 			pqospriv->uapsd_ap_supported = 1;
759 			rtw_hal_set_hwreg(padapter, HW_VAR_UAPSD_TID, NULL);
760 		}
761 #endif /* CONFIG_WMMPS_STA */
762 	}
763 }
764 
bwmode_update_check(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)765 static void bwmode_update_check(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
766 {
767 #ifdef CONFIG_80211N_HT
768 	unsigned char	 new_bwmode;
769 	unsigned char  new_ch_offset;
770 	struct HT_info_element	*pHT_info;
771 	struct mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
772 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
773 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
774 	struct registry_priv *pregistrypriv = &padapter->registrypriv;
775 	struct ht_priv			*phtpriv = &pmlmepriv->htpriv;
776 	u8	cbw40_enable = 0;
777 
778 	if (!pIE)
779 		return;
780 
781 	if (phtpriv->ht_option == _FALSE)
782 		return;
783 
784 	if (pmlmeext->chandef.bw >= CHANNEL_WIDTH_80)
785 		return;
786 
787 	if (pIE->Length > sizeof(struct HT_info_element))
788 		return;
789 
790 	pHT_info = (struct HT_info_element *)pIE->data;
791 
792 	if (rtw_hw_chk_bw_cap(adapter_to_dvobj(padapter), BW_CAP_40M)) {
793 		if (pmlmeext->chandef.chan > 14) {
794 			if (REGSTY_IS_BW_5G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
795 				cbw40_enable = 1;
796 		} else {
797 			if (REGSTY_IS_BW_2G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
798 				cbw40_enable = 1;
799 		}
800 	}
801 
802 	if ((pHT_info->infos[0] & BIT(2)) && cbw40_enable) {
803 		new_bwmode = CHANNEL_WIDTH_40;
804 
805 		switch (pHT_info->infos[0] & 0x3) {
806 		case 1:
807 			new_ch_offset = CHAN_OFFSET_UPPER;
808 			break;
809 
810 		case 3:
811 			new_ch_offset = CHAN_OFFSET_LOWER;
812 			break;
813 
814 		default:
815 			new_bwmode = CHANNEL_WIDTH_20;
816 			new_ch_offset = CHAN_OFFSET_NO_EXT;
817 			break;
818 		}
819 	} else {
820 		new_bwmode = CHANNEL_WIDTH_20;
821 		new_ch_offset = CHAN_OFFSET_NO_EXT;
822 	}
823 
824 
825 	if ((new_bwmode != pmlmeext->chandef.bw || new_ch_offset != pmlmeext->chandef.offset)
826 	    && new_bwmode < pmlmeext->chandef.bw
827 	   ) {
828 		pmlmeinfo->bwmode_updated = _TRUE;
829 
830 		pmlmeext->chandef.bw = new_bwmode;
831 		pmlmeext->chandef.offset = new_ch_offset;
832 
833 		/* update HT info also */
834 		HT_info_handler(padapter, pIE);
835 	} else
836 		pmlmeinfo->bwmode_updated = _FALSE;
837 
838 
839 	if (_TRUE == pmlmeinfo->bwmode_updated) {
840 		struct sta_info *psta;
841 		WLAN_BSSID_EX	*cur_network = &(pmlmeinfo->network);
842 		struct sta_priv	*pstapriv = &padapter->stapriv;
843 
844 		/* set_channel_bwmode(padapter, pmlmeext->chandef.chan, pmlmeext->chandef.offset, pmlmeext->chandef.bw); */
845 
846 
847 		/* update ap's stainfo */
848 		psta = rtw_get_stainfo(pstapriv, cur_network->MacAddress);
849 		if (psta) {
850 			struct ht_priv	*phtpriv_sta = &psta->htpriv;
851 
852 			if (phtpriv_sta->ht_option) {
853 				/* bwmode				 */
854 				psta->phl_sta->chandef.bw = pmlmeext->chandef.bw;
855 				phtpriv_sta->ch_offset = pmlmeext->chandef.offset;
856 			} else {
857 				psta->phl_sta->chandef.bw = CHANNEL_WIDTH_20;
858 				phtpriv_sta->ch_offset = CHAN_OFFSET_NO_EXT;
859 			}
860 
861 			rtw_phl_cmd_change_stainfo(adapter_to_dvobj(padapter)->phl,
862 					   psta->phl_sta,
863 					   STA_CHG_RAMASK,
864 					   NULL,
865 					   0,
866 					   PHL_CMD_NO_WAIT,
867 					   0);
868 		}
869 
870 		/* pmlmeinfo->bwmode_updated = _FALSE; */ /* bwmode_updated done, reset it! */
871 	}
872 #endif /* CONFIG_80211N_HT */
873 }
874 
875 #ifdef ROKU_PRIVATE
Supported_rate_infra_ap(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)876 void Supported_rate_infra_ap(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
877 {
878 	unsigned int	i;
879 	struct mlme_ext_priv		*pmlmeext = &padapter->mlmeextpriv;
880 	struct mlme_ext_info		*pmlmeinfo = &(pmlmeext->mlmext_info);
881 
882 	if (pIE == NULL)
883 		return;
884 
885 	for (i = 0 ; i < pIE->Length; i++)
886 		pmlmeinfo->SupportedRates_infra_ap[i] = (pIE->data[i]);
887 
888 }
889 
Extended_Supported_rate_infra_ap(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)890 void Extended_Supported_rate_infra_ap(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
891 {
892 	unsigned int i, j;
893 	struct mlme_ext_priv		*pmlmeext = &padapter->mlmeextpriv;
894 	struct mlme_ext_info		*pmlmeinfo = &(pmlmeext->mlmext_info);
895 
896 	if (pIE == NULL)
897 		return;
898 
899 	if (pIE->Length > 0) {
900 		for (i = 0; i < NDIS_802_11_LENGTH_RATES_EX; i++) {
901 			if (pmlmeinfo->SupportedRates_infra_ap[i] == 0)
902 				break;
903 		}
904 		for (j = 0; j < pIE->Length; j++)
905 			pmlmeinfo->SupportedRates_infra_ap[i+j] = (pIE->data[j]);
906 	}
907 
908 }
909 
HT_get_ss_from_mcs_set(u8 * mcs_set,u8 * Rx_ss)910 void HT_get_ss_from_mcs_set(u8 *mcs_set, u8 *Rx_ss)
911 {
912 	u8 i, j;
913 	u8 r_ss = 0, t_ss = 0;
914 
915 	for (i = 0; i < 4; i++) {
916 		if ((mcs_set[3-i] & 0xff) != 0x00) {
917 			r_ss = 4-i;
918 			break;
919 		}
920 	}
921 
922 	*Rx_ss = r_ss;
923 }
924 
HT_caps_handler_infra_ap(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)925 void HT_caps_handler_infra_ap(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
926 {
927 	unsigned int	i;
928 	u8	cur_stbc_cap_infra_ap = 0;
929 	struct mlme_priv		*pmlmepriv = &padapter->mlmepriv;
930 	struct ht_priv_infra_ap		*phtpriv = &pmlmepriv->htpriv_infra_ap;
931 
932 	struct mlme_ext_priv		*pmlmeext = &padapter->mlmeextpriv;
933 	struct mlme_ext_info		*pmlmeinfo = &(pmlmeext->mlmext_info);
934 
935 	if (pIE == NULL)
936 		return;
937 
938 	pmlmeinfo->ht_vht_received |= BIT(0);
939 
940 	/*copy MCS_SET*/
941 	for (i = 3; i < 19; i++)
942 		phtpriv->MCS_set_infra_ap[i-3] = (pIE->data[i]);
943 
944 	/*get number of stream from mcs set*/
945 	HT_get_ss_from_mcs_set(phtpriv->MCS_set_infra_ap, &phtpriv->Rx_ss_infra_ap);
946 
947 	phtpriv->rx_highest_data_rate_infra_ap = le16_to_cpu(GET_HT_CAP_ELE_RX_HIGHEST_DATA_RATE(pIE->data));
948 
949 	phtpriv->ldpc_cap_infra_ap = GET_HT_CAP_ELE_LDPC_CAP(pIE->data);
950 
951 	if (GET_HT_CAP_ELE_RX_STBC(pIE->data))
952 		SET_FLAG(cur_stbc_cap_infra_ap, STBC_HT_ENABLE_RX);
953 	if (GET_HT_CAP_ELE_TX_STBC(pIE->data))
954 		SET_FLAG(cur_stbc_cap_infra_ap, STBC_HT_ENABLE_TX);
955 	phtpriv->stbc_cap_infra_ap = cur_stbc_cap_infra_ap;
956 
957 	/*store ap info SGI 20m 40m*/
958 	phtpriv->sgi_20m_infra_ap = GET_HT_CAP_ELE_SHORT_GI20M(pIE->data);
959 	phtpriv->sgi_40m_infra_ap = GET_HT_CAP_ELE_SHORT_GI40M(pIE->data);
960 
961 	/*store ap info for supported channel bandwidth*/
962 	phtpriv->channel_width_infra_ap = GET_HT_CAP_ELE_CHL_WIDTH(pIE->data);
963 }
964 #endif /* ROKU_PRIVATE */
965 
HT_caps_handler(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)966 void HT_caps_handler(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
967 {
968 #ifdef CONFIG_80211N_HT
969 	unsigned int i;
970 	u8 max_AMPDU_len, min_MPDU_spacing;
971 	u8 cur_ldpc_cap = 0, cur_stbc_cap = 0, cur_beamform_cap = 0, rx_nss = 0;
972 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
973 	struct mlme_ext_info *pmlmeinfo = &(pmlmeext->mlmext_info);
974 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
975 	struct ht_priv *phtpriv = &pmlmepriv->htpriv;
976 	struct sta_info *sta = NULL;
977 	struct rtw_phl_stainfo_t *phl_sta = NULL;
978 	struct rtw_wifi_role_t *wrole = padapter->phl_role;
979 	struct protocol_cap_t *proto_role_cap = &(wrole->proto_role_cap);
980 #ifdef CONFIG_DISABLE_MCS13TO15
981 	struct registry_priv *pregistrypriv = &padapter->registrypriv;
982 #endif
983 
984 	if (pIE == NULL)
985 		return;
986 
987 	if (phtpriv->ht_option == _FALSE)
988 		return;
989 
990 	pmlmeinfo->HT_caps_enable = 1;
991 
992 	for (i = 0; i < (pIE->Length); i++) {
993 		if (i != 2) {
994 			/*	Commented by Albert 2010/07/12 */
995 			/*	Got the endian issue here. */
996 			pmlmeinfo->HT_caps.u.HT_cap[i] &= (pIE->data[i]);
997 		} else {
998 			/* AMPDU Parameters field */
999 
1000 			/* Get MIN of MAX AMPDU Length Exp */
1001 			if ((pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x3) > (pIE->data[i] & 0x3))
1002 				max_AMPDU_len = (pIE->data[i] & 0x3);
1003 			else
1004 				max_AMPDU_len = (pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x3);
1005 
1006 			/* Get MAX of MIN MPDU Start Spacing */
1007 			if ((pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x1c) > (pIE->data[i] & 0x1c))
1008 				min_MPDU_spacing = (pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x1c);
1009 			else
1010 				min_MPDU_spacing = (pIE->data[i] & 0x1c);
1011 
1012 			pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para = max_AMPDU_len | min_MPDU_spacing;
1013 		}
1014 	}
1015 
1016 	/*	Commented by Albert 2010/07/12 */
1017 	/*	Have to handle the endian issue after copying. */
1018 	/*	HT_ext_caps didn't be used yet.	 */
1019 	pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info = le16_to_cpu(pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info);
1020 	pmlmeinfo->HT_caps.u.HT_cap_element.HT_ext_caps = le16_to_cpu(pmlmeinfo->HT_caps.u.HT_cap_element.HT_ext_caps);
1021 
1022 	/* update the MCS set */
1023 	for (i = 0; i < 16; i++)
1024 		pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate[i] &= pmlmeext->default_supported_mcs_set[i];
1025 
1026 	rx_nss = get_phy_rx_nss(padapter);
1027 
1028 	switch (rx_nss) {
1029 	case 1:
1030 		set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_1R);
1031 		break;
1032 	case 2:
1033 		#ifdef CONFIG_DISABLE_MCS13TO15
1034 		if (pmlmeext->chandef.bw == CHANNEL_WIDTH_40 && pregistrypriv->wifi_spec != 1)
1035 			set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R_13TO15_OFF);
1036 		else
1037 		#endif
1038 			set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R);
1039 		break;
1040 	case 3:
1041 		set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_3R);
1042 		break;
1043 	case 4:
1044 		set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_4R);
1045 		break;
1046 	default:
1047 		RTW_WARN("rf_type:%d or tx_nss:%u is not expected\n", GET_HAL_RFPATH(adapter_to_dvobj(padapter)), rx_nss);
1048 	}
1049 
1050 	if (check_fwstate(pmlmepriv, WIFI_AP_STATE)) {
1051 		/* Config STBC setting */
1052 		if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX) && GET_HT_CAP_ELE_RX_STBC(pIE->data)) {
1053 			SET_FLAG(cur_stbc_cap, STBC_HT_ENABLE_TX);
1054 			RTW_INFO("Enable HT Tx STBC !\n");
1055 		}
1056 		phtpriv->stbc_cap = cur_stbc_cap;
1057 
1058 #ifdef CONFIG_BEAMFORMING
1059 		/* Config Tx beamforming setting */
1060 		if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE) &&
1061 		    GET_HT_CAP_TXBF_EXPLICIT_COMP_STEERING_CAP(pIE->data)) {
1062 			SET_FLAG(cur_beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE);
1063 			/* Shift to BEAMFORMING_HT_BEAMFORMEE_CHNL_EST_CAP*/
1064 			SET_FLAG(cur_beamform_cap, GET_HT_CAP_TXBF_CHNL_ESTIMATION_NUM_ANTENNAS(pIE->data) << 6);
1065 		}
1066 
1067 		if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE) &&
1068 		    GET_HT_CAP_TXBF_EXPLICIT_COMP_FEEDBACK_CAP(pIE->data)) {
1069 			SET_FLAG(cur_beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE);
1070 			/* Shift to BEAMFORMING_HT_BEAMFORMER_STEER_NUM*/
1071 			SET_FLAG(cur_beamform_cap, GET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(pIE->data) << 4);
1072 		}
1073 		phtpriv->beamform_cap = cur_beamform_cap;
1074 		if (cur_beamform_cap)
1075 			RTW_INFO("AP HT Beamforming Cap = 0x%02X\n", cur_beamform_cap);
1076 #endif /*CONFIG_BEAMFORMING*/
1077 	} else {
1078 		/*WIFI_STATION_STATEorI_ADHOC_STATE or WIFI_ADHOC_MASTER_STATE*/
1079 		sta = rtw_get_stainfo(&padapter->stapriv, pmlmeinfo->network.MacAddress);
1080 		if (!sta) {
1081 			RTW_ERR(FUNC_ADPT_FMT ": STA(" MAC_FMT ") not found!\n",
1082 				FUNC_ADPT_ARG(padapter), MAC_ARG(pmlmeinfo->network.MacAddress));
1083 			return;
1084 		}
1085 		if (!sta->phl_sta) {
1086 			RTW_ERR(FUNC_ADPT_FMT ": PHL STA(" MAC_FMT ") not exist!\n",
1087 				FUNC_ADPT_ARG(padapter), MAC_ARG(pmlmeinfo->network.MacAddress));
1088 			return;
1089 		}
1090 		phl_sta = sta->phl_sta;
1091 
1092 		/* Config LDPC Coding Capability */
1093 		if (TEST_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_TX) && GET_HT_CAP_ELE_LDPC_CAP(pIE->data)) {
1094 			SET_FLAG(cur_ldpc_cap, (LDPC_HT_ENABLE_TX | LDPC_HT_CAP_TX));
1095 			RTW_INFO("Enable HT Tx LDPC!\n");
1096 			phl_sta->asoc_cap.ht_ldpc = 1;
1097 		}
1098 		phtpriv->ldpc_cap = cur_ldpc_cap;
1099 
1100 		/* Config STBC setting */
1101 		if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX) && GET_HT_CAP_ELE_RX_STBC(pIE->data)) {
1102 			SET_FLAG(cur_stbc_cap, (STBC_HT_ENABLE_TX | STBC_HT_CAP_TX));
1103 			RTW_INFO("Enable HT Tx STBC!\n");
1104 			phl_sta->asoc_cap.stbc_ht_rx =
1105 				proto_role_cap->stbc_ht_tx ? GET_HT_CAP_ELE_RX_STBC(pIE->data) : 0;
1106 		}
1107 		phtpriv->stbc_cap = cur_stbc_cap;
1108 		phl_sta->asoc_cap.stbc_ht_tx = GET_HT_CAP_ELE_TX_STBC(pIE->data);
1109 
1110 #ifdef CONFIG_BEAMFORMING
1111 		/* Config beamforming setting */
1112 		if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE) &&
1113 		    GET_HT_CAP_TXBF_EXPLICIT_COMP_STEERING_CAP(pIE->data)) {
1114 			SET_FLAG(cur_beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE);
1115 			/* Shift to BEAMFORMING_HT_BEAMFORMEE_CHNL_EST_CAP*/
1116 			SET_FLAG(cur_beamform_cap, GET_HT_CAP_TXBF_CHNL_ESTIMATION_NUM_ANTENNAS(pIE->data) << 6);
1117 		}
1118 
1119 		if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE) &&
1120 		    GET_HT_CAP_TXBF_EXPLICIT_COMP_FEEDBACK_CAP(pIE->data)) {
1121 			SET_FLAG(cur_beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE);
1122 			/* Shift to BEAMFORMING_HT_BEAMFORMER_STEER_NUM*/
1123 			SET_FLAG(cur_beamform_cap, GET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(pIE->data) << 4);
1124 		}
1125 		phtpriv->beamform_cap = cur_beamform_cap;
1126 		if (cur_beamform_cap)
1127 			RTW_INFO("Client HT Beamforming Cap = 0x%02X\n", cur_beamform_cap);
1128 #endif /*CONFIG_BEAMFORMING*/
1129 	}
1130 
1131 #endif /* CONFIG_80211N_HT */
1132 }
1133 
HT_info_handler(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)1134 void HT_info_handler(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
1135 {
1136 #ifdef CONFIG_80211N_HT
1137 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
1138 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1139 	struct mlme_priv		*pmlmepriv = &padapter->mlmepriv;
1140 	struct ht_priv			*phtpriv = &pmlmepriv->htpriv;
1141 
1142 	if (pIE == NULL)
1143 		return;
1144 
1145 	if (phtpriv->ht_option == _FALSE)
1146 		return;
1147 
1148 
1149 	if (pIE->Length > sizeof(struct HT_info_element))
1150 		return;
1151 
1152 	pmlmeinfo->HT_info_enable = 1;
1153 	_rtw_memcpy(&(pmlmeinfo->HT_info), pIE->data, pIE->Length);
1154 #endif /* CONFIG_80211N_HT */
1155 	return;
1156 }
1157 
HTOnAssocRsp(_adapter * padapter)1158 void HTOnAssocRsp(_adapter *padapter)
1159 {
1160 	unsigned char		max_AMPDU_len;
1161 	unsigned char		min_MPDU_spacing;
1162 	/* struct registry_priv	 *pregpriv = &padapter->registrypriv; */
1163 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
1164 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1165 
1166 	RTW_INFO("%s\n", __FUNCTION__);
1167 
1168 	if ((pmlmeinfo->HT_info_enable) && (pmlmeinfo->HT_caps_enable))
1169 		pmlmeinfo->HT_enable = 1;
1170 	else {
1171 		pmlmeinfo->HT_enable = 0;
1172 		/* set_channel_bwmode(padapter, pmlmeext->cur_channel, pmlmeext->cur_ch_offset, pmlmeext->cur_bwmode); */
1173 		return;
1174 	}
1175 
1176 	/* handle A-MPDU parameter field */
1177 	/*
1178 		AMPDU_para [1:0]:Max AMPDU Len => 0:8k , 1:16k, 2:32k, 3:64k
1179 		AMPDU_para [4:2]:Min MPDU Start Spacing
1180 	*/
1181 	max_AMPDU_len = pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x03;
1182 
1183 	min_MPDU_spacing = (pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x1c) >> 2;
1184 
1185 	rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_MIN_SPACE, (u8 *)(&min_MPDU_spacing));
1186 #ifdef CONFIG_80211N_HT
1187 	rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_FACTOR, (u8 *)(&max_AMPDU_len));
1188 #endif /* CONFIG_80211N_HT */
1189 #if 0 /* move to rtw_update_ht_cap() */
1190 	if ((pregpriv->bw_mode > 0) &&
1191 	    (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & BIT(1)) &&
1192 	    (pmlmeinfo->HT_info.infos[0] & BIT(2))) {
1193 		/* switch to the 40M Hz mode accoring to the AP */
1194 		pmlmeext->cur_bwmode = CHANNEL_WIDTH_40;
1195 		switch ((pmlmeinfo->HT_info.infos[0] & 0x3)) {
1196 		case IEEE80211_SCA:
1197 			pmlmeext->cur_ch_offset = CHAN_OFFSET_UPPER;
1198 			break;
1199 
1200 		case IEEE80211_SCB:
1201 			pmlmeext->cur_ch_offset = CHAN_OFFSET_LOWER;
1202 			break;
1203 
1204 		default:
1205 			pmlmeext->cur_ch_offset = CHAN_OFFSET_NO_EXT;
1206 			break;
1207 		}
1208 	}
1209 #endif
1210 
1211 	/* set_channel_bwmode(padapter, pmlmeext->cur_channel, pmlmeext->cur_ch_offset, pmlmeext->cur_bwmode); */
1212 
1213 #if 0 /* move to rtw_update_ht_cap() */
1214 	/*  */
1215 	/* Config SM Power Save setting */
1216 	/*  */
1217 	pmlmeinfo->SM_PS = (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & 0x0C) >> 2;
1218 	if (pmlmeinfo->SM_PS == WLAN_HT_CAP_SM_PS_STATIC) {
1219 #if 0
1220 		u8 i;
1221 		/* update the MCS rates */
1222 		for (i = 0; i < 16; i++)
1223 			pmlmeinfo->HT_caps.HT_cap_element.MCS_rate[i] &= MCS_rate_1R[i];
1224 #endif
1225 		RTW_INFO("%s(): WLAN_HT_CAP_SM_PS_STATIC\n", __FUNCTION__);
1226 	}
1227 
1228 	/*  */
1229 	/* Config current HT Protection mode. */
1230 	/*  */
1231 	pmlmeinfo->HT_protection = pmlmeinfo->HT_info.infos[1] & 0x3;
1232 #endif
1233 
1234 }
1235 
ERP_IE_handler(_adapter * padapter,PNDIS_802_11_VARIABLE_IEs pIE)1236 void ERP_IE_handler(_adapter *padapter, PNDIS_802_11_VARIABLE_IEs pIE)
1237 {
1238 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
1239 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1240 
1241 	if (pIE->Length > 1)
1242 		return;
1243 
1244 	pmlmeinfo->ERP_enable = 1;
1245 	_rtw_memcpy(&(pmlmeinfo->ERP_IE), pIE->data, pIE->Length);
1246 }
1247 
VCS_update(_adapter * padapter,struct sta_info * psta)1248 void VCS_update(_adapter *padapter, struct sta_info *psta)
1249 {
1250 	struct registry_priv	*pregpriv = &padapter->registrypriv;
1251 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
1252 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1253 
1254 	switch (pregpriv->vrtl_carrier_sense) { /* 0:off 1:on 2:auto */
1255 	case 0: /* off */
1256 		psta->rtsen = 0;
1257 		psta->cts2self = 0;
1258 		break;
1259 
1260 	case 1: /* on */
1261 		if (pregpriv->vcs_type == 1) { /* 1:RTS/CTS 2:CTS to self */
1262 			psta->rtsen = 1;
1263 			psta->cts2self = 0;
1264 		} else {
1265 			psta->rtsen = 0;
1266 			psta->cts2self = 1;
1267 		}
1268 		break;
1269 
1270 	case 2: /* auto */
1271 	default:
1272 		if (((pmlmeinfo->ERP_enable) && (pmlmeinfo->ERP_IE & BIT(1)))
1273 			/*||(pmlmepriv->ht_op_mode & HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT)*/
1274 		) {
1275 			if (pregpriv->vcs_type == 1) {
1276 				psta->rtsen = 1;
1277 				psta->cts2self = 0;
1278 			} else {
1279 				psta->rtsen = 0;
1280 				psta->cts2self = 1;
1281 			}
1282 		} else {
1283 			psta->rtsen = 0;
1284 			psta->cts2self = 0;
1285 		}
1286 		break;
1287 	}
1288 
1289 	switch (pregpriv->hw_rts_en) { /* 0:disable 1:enable */
1290 	case 0:
1291 		psta->hw_rts_en = 0;
1292 		break;
1293 	case 1:
1294 		psta->hw_rts_en = 1;
1295 		break;
1296 	default:
1297 		RTW_WARN("%s: unexpected value(%d) for hw_rts_en.\n", __func__, pregpriv->hw_rts_en);
1298 		break;
1299 	}
1300 
1301 }
1302 
update_ldpc_stbc_cap(struct sta_info * psta)1303 void	update_ldpc_stbc_cap(struct sta_info *psta)
1304 {
1305 #ifdef CONFIG_80211N_HT
1306 
1307 #ifdef CONFIG_80211AC_VHT
1308 #ifdef CONFIG_80211AX_HE
1309 	/* CONFIG_80211AX_HE_TODO */
1310 #endif /* CONFIG_80211AX_HE */
1311 	if (psta->vhtpriv.vht_option) {
1312 		if (TEST_FLAG(psta->vhtpriv.ldpc_cap, LDPC_VHT_ENABLE_TX))
1313 			psta->phl_sta->asoc_cap.vht_ldpc = 1;
1314 		else
1315 			psta->phl_sta->asoc_cap.vht_ldpc = 0;
1316 	} else
1317 #endif /* CONFIG_80211AC_VHT */
1318 		if (psta->htpriv.ht_option) {
1319 			if (TEST_FLAG(psta->htpriv.ldpc_cap, LDPC_HT_ENABLE_TX))
1320 				psta->phl_sta->asoc_cap.ht_ldpc = 1;
1321 			else
1322 				psta->phl_sta->asoc_cap.ht_ldpc = 0;
1323 		} else {
1324 			psta->phl_sta->asoc_cap.vht_ldpc = 0;
1325 			psta->phl_sta->asoc_cap.ht_ldpc = 0;
1326 		}
1327 
1328 #endif /* CONFIG_80211N_HT */
1329 }
1330 
check_ielen(u8 * start,uint len)1331 int check_ielen(u8 *start, uint len)
1332 {
1333 	int left = len;
1334 	u8 *pos = start;
1335 	u8 id, elen;
1336 
1337 	while (left >= 2) {
1338 		id = *pos++;
1339 		elen = *pos++;
1340 		left -= 2;
1341 
1342 		if (elen > left) {
1343 			RTW_ERR("IEEE 802.11 element parse failed (id=%d elen=%d left=%lu)\n",
1344 					id, elen, (unsigned long) left);
1345 			return _FALSE;
1346 		}
1347 		if ((id == WLAN_EID_VENDOR_SPECIFIC) && (elen < 3))
1348 				return _FALSE;
1349 
1350 		left -= elen;
1351 		pos += elen;
1352 	}
1353 	if (left)
1354 		return _FALSE;
1355 
1356 	return _TRUE;
1357 }
1358 
validate_beacon_len(u8 * pframe,u32 len)1359 int validate_beacon_len(u8 *pframe, u32 len)
1360 {
1361 	u8 ie_offset = _BEACON_IE_OFFSET_ + sizeof(struct rtw_ieee80211_hdr_3addr);
1362 
1363 	if (len < ie_offset) {
1364 		RTW_ERR("%s: incorrect beacon length(%d)\n", __func__, len);
1365 		rtw_warn_on(1);
1366 		return _FALSE;
1367 	}
1368 
1369 	if (check_ielen(pframe + ie_offset, len - ie_offset) == _FALSE)
1370 		return _FALSE;
1371 
1372 	return _TRUE;
1373 }
1374 
1375 
1376 u8 support_rate_ranges[] = {
1377 	IEEE80211_CCK_RATE_1MB,
1378 	IEEE80211_CCK_RATE_2MB,
1379 	IEEE80211_CCK_RATE_5MB,
1380 	IEEE80211_CCK_RATE_11MB,
1381 	IEEE80211_OFDM_RATE_6MB,
1382 	IEEE80211_OFDM_RATE_9MB,
1383 	IEEE80211_OFDM_RATE_12MB,
1384 	IEEE80211_OFDM_RATE_18MB,
1385 	IEEE80211_PBCC_RATE_22MB,
1386 	IEEE80211_FREAK_RATE_22_5MB,
1387 	IEEE80211_OFDM_RATE_24MB,
1388 	IEEE80211_OFDM_RATE_36MB,
1389 	IEEE80211_OFDM_RATE_48MB,
1390 	IEEE80211_OFDM_RATE_54MB,
1391 };
1392 
match_ranges(u16 EID,u32 value)1393 inline bool match_ranges(u16 EID, u32 value)
1394 {
1395 	int i;
1396 	int nr_range;
1397 
1398 	switch (EID) {
1399 	case _EXT_SUPPORTEDRATES_IE_:
1400 	case _SUPPORTEDRATES_IE_:
1401 		nr_range = sizeof(support_rate_ranges)/sizeof(u8);
1402 		for (i = 0; i < nr_range; i++) {
1403 			/*	clear bit7 before searching.	*/
1404 			value &= ~BIT(7);
1405 			if (value == support_rate_ranges[i])
1406 				return _TRUE;
1407 		}
1408 		break;
1409 	default:
1410 		break;
1411 	};
1412 	return _FALSE;
1413 }
1414 
1415 /*
1416  * rtw_validate_value: validate the IE contain.
1417  *
1418  *	Input :
1419  *		EID : Element ID
1420  *		p	: IE buffer (without EID & length)
1421  *		len	: IE length
1422  *	return:
1423  * 		_TRUE	: All Values are validated.
1424  *		_FALSE	: At least one value is NOT validated.
1425  */
rtw_validate_value(u16 EID,u8 * p,u16 len)1426 bool rtw_validate_value(u16 EID, u8 *p, u16 len)
1427 {
1428 	u8 rate;
1429 	u32 i, nr_val;
1430 
1431 	switch (EID) {
1432 	case _EXT_SUPPORTEDRATES_IE_:
1433 	case _SUPPORTEDRATES_IE_:
1434 		nr_val = len;
1435 		for (i=0; i<nr_val; i++) {
1436 			rate = *(p+i);
1437 			if (match_ranges(EID, rate) == _FALSE)
1438 				return _FALSE;
1439 		}
1440 		break;
1441 	default:
1442 		break;
1443 	};
1444 	return _TRUE;
1445 }
1446 
is_hidden_ssid(char * ssid,int len)1447 bool is_hidden_ssid(char *ssid, int len)
1448 {
1449 	return len == 0 || is_all_null(ssid, len) == _TRUE;
1450 }
1451 
hidden_ssid_ap(WLAN_BSSID_EX * snetwork)1452 inline bool hidden_ssid_ap(WLAN_BSSID_EX *snetwork)
1453 {
1454 	return ((snetwork->Ssid.SsidLength == 0) ||
1455 		is_all_null(snetwork->Ssid.Ssid, snetwork->Ssid.SsidLength) == _TRUE);
1456 }
1457 
1458 /*
1459 	Get SSID if this ilegal frame(probe resp) comes from a hidden SSID AP.
1460 	Update the SSID to the corresponding pnetwork in scan queue.
1461 */
rtw_absorb_ssid_ifneed(_adapter * padapter,WLAN_BSSID_EX * bssid,u8 * pframe)1462 void rtw_absorb_ssid_ifneed(_adapter *padapter, WLAN_BSSID_EX *bssid, u8 *pframe)
1463 {
1464 	struct wlan_network *scanned = NULL;
1465 	WLAN_BSSID_EX	*snetwork;
1466 	u8 ie_offset, *p=NULL, *next_ie=NULL;
1467 	u8 *mac;
1468 	sint ssid_len_ori;
1469 	u32 remain_len = 0;
1470 	u8 backupIE[MAX_IE_SZ];
1471 	u16 subtype;
1472 
1473 	mac = get_addr2_ptr(pframe);
1474 	subtype = get_frame_sub_type(pframe);
1475 
1476 	if (subtype == WIFI_BEACON) {
1477 		bssid->Reserved[0] = BSS_TYPE_BCN;
1478 		ie_offset = _BEACON_IE_OFFSET_;
1479 	} else {
1480 		/* FIXME : more type */
1481 		if (subtype == WIFI_PROBERSP) {
1482 			ie_offset = _PROBERSP_IE_OFFSET_;
1483 			bssid->Reserved[0] = BSS_TYPE_PROB_RSP;
1484 		} else if (subtype == WIFI_PROBEREQ) {
1485 			ie_offset = _PROBEREQ_IE_OFFSET_;
1486 			bssid->Reserved[0] = BSS_TYPE_PROB_REQ;
1487 		} else {
1488 			bssid->Reserved[0] = BSS_TYPE_UNDEF;
1489 			ie_offset = _FIXED_IE_LENGTH_;
1490 		}
1491 	}
1492 
1493 	_rtw_spinlock_bh(&padapter->mlmepriv.scanned_queue.lock);
1494 	scanned = _rtw_find_network(&padapter->mlmepriv.scanned_queue, mac);
1495 	if (!scanned) {
1496 		_rtw_spinunlock_bh(&padapter->mlmepriv.scanned_queue.lock);
1497 		return;
1498 	}
1499 
1500 	snetwork = &(scanned->network);
1501 	/* scan queue records as Hidden SSID && Input frame is NOT Hidden SSID	*/
1502 	if (hidden_ssid_ap(snetwork) && !hidden_ssid_ap(bssid)) {
1503 		p = rtw_get_ie(snetwork->IEs+ie_offset, _SSID_IE_, &ssid_len_ori, snetwork->IELength-ie_offset);
1504 		if (!p) {
1505 			_rtw_spinunlock_bh(&padapter->mlmepriv.scanned_queue.lock);
1506 			return;
1507 		}
1508 		next_ie = p + 2 + ssid_len_ori;
1509 		remain_len = snetwork->IELength - (next_ie - snetwork->IEs);
1510 		scanned->network.Ssid.SsidLength = bssid->Ssid.SsidLength;
1511 		_rtw_memcpy(scanned->network.Ssid.Ssid, bssid->Ssid.Ssid, bssid->Ssid.SsidLength);
1512 
1513 		//update pnetwork->ssid, pnetwork->ssidlen
1514 		_rtw_memcpy(backupIE, next_ie, remain_len);
1515 		*(p+1) = bssid->Ssid.SsidLength;
1516 		_rtw_memcpy(p+2, bssid->Ssid.Ssid, bssid->Ssid.SsidLength);
1517 		_rtw_memcpy(p+2+bssid->Ssid.SsidLength, backupIE, remain_len);
1518 		snetwork->IELength += bssid->Ssid.SsidLength;
1519 	}
1520 	_rtw_spinunlock_bh(&padapter->mlmepriv.scanned_queue.lock);
1521 }
1522 
1523 #ifdef DBG_RX_BCN
rtw_debug_rx_bcn(_adapter * adapter,u8 * pframe,u32 packet_len)1524 void rtw_debug_rx_bcn(_adapter *adapter, u8 *pframe, u32 packet_len)
1525 {
1526 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
1527 	struct mlme_ext_info *mlmeinfo = &(pmlmeext->mlmext_info);
1528 	u16 sn = ((struct rtw_ieee80211_hdr_3addr *)pframe)->seq_ctl >> 4;
1529 	u64 tsf, tsf_offset;
1530 	u8 dtim_cnt, dtim_period, tim_bmap, tim_pvbit;
1531 
1532 	update_TSF(pmlmeext, pframe, packet_len);
1533 	tsf = pmlmeext->TSFValue;
1534 	tsf_offset = rtw_modular64(pmlmeext->TSFValue, (mlmeinfo->bcn_interval * 1024));
1535 
1536 	/*get TIM IE*/
1537 	/*DTIM Count*/
1538 	dtim_cnt = pmlmeext->tim[0];
1539 	/*DTIM Period*/
1540 	dtim_period = pmlmeext->tim[1];
1541 	/*Bitmap*/
1542 	tim_bmap = pmlmeext->tim[2];
1543 	/*Partial VBitmap AID 0 ~ 7*/
1544 	tim_pvbit = pmlmeext->tim[3];
1545 
1546 	RTW_INFO("[BCN] SN-%d, TSF-%lld(us), offset-%lld, bcn_interval-%d DTIM-%d[%d] bitmap-0x%02x-0x%02x\n",
1547 		sn, tsf, tsf_offset, mlmeinfo->bcn_interval, dtim_period, dtim_cnt, tim_bmap, tim_pvbit);
1548 }
1549 #endif
1550 
1551 /*
1552  * rtw_get_bcn_keys: get beacon keys from recv frame
1553  *
1554  * TODO:
1555  *	WLAN_EID_COUNTRY
1556  *	WLAN_EID_ERP_INFO
1557  *	WLAN_EID_CHANNEL_SWITCH
1558  *	WLAN_EID_PWR_CONSTRAINT
1559  */
rtw_get_bcn_keys(_adapter * adapter,u8 * pframe,u32 packet_len,struct beacon_keys * recv_beacon)1560 int rtw_get_bcn_keys(_adapter *adapter, u8 *pframe, u32 packet_len,
1561 		     struct beacon_keys *recv_beacon)
1562 {
1563 	int left;
1564 	u16 capability;
1565 	unsigned char *pos;
1566 	struct rtw_ieee802_11_elems elems;
1567 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
1568 
1569 	_rtw_memset(recv_beacon, 0, sizeof(*recv_beacon));
1570 
1571 	/* checking capabilities */
1572 	capability = le16_to_cpu(*(unsigned short *)(pframe + WLAN_HDR_A3_LEN + 10));
1573 
1574 	/* checking IEs */
1575 	left = packet_len - sizeof(struct rtw_ieee80211_hdr_3addr) - _BEACON_IE_OFFSET_;
1576 	pos = pframe + sizeof(struct rtw_ieee80211_hdr_3addr) + _BEACON_IE_OFFSET_;
1577 	if (rtw_ieee802_11_parse_elems(pos, left, &elems, 1) == ParseFailed)
1578 		return _FALSE;
1579 
1580 	if (elems.ht_capabilities) {
1581 		if (elems.ht_capabilities_len != 26)
1582 			return _FALSE;
1583 	}
1584 
1585 	if (elems.ht_operation) {
1586 		if (elems.ht_operation_len != 22)
1587 			return _FALSE;
1588 	}
1589 
1590 	if (elems.vht_capabilities) {
1591 		if (elems.vht_capabilities_len != 12)
1592 			return _FALSE;
1593 	}
1594 
1595 	if (elems.vht_operation) {
1596 		if (elems.vht_operation_len != 5)
1597 			return _FALSE;
1598 	}
1599 
1600 	if (rtw_ies_get_supported_rate(pos, left, recv_beacon->rate_set, &recv_beacon->rate_num) == _FAIL)
1601 		return _FALSE;
1602 
1603 	if (cckratesonly_included(recv_beacon->rate_set, recv_beacon->rate_num) == _TRUE)
1604 		recv_beacon->proto_cap |= PROTO_CAP_11B;
1605 	else if (cckrates_included(recv_beacon->rate_set, recv_beacon->rate_num) == _TRUE)
1606 		recv_beacon->proto_cap |= PROTO_CAP_11B | PROTO_CAP_11G;
1607 	else
1608 		recv_beacon->proto_cap |= PROTO_CAP_11G;
1609 
1610 	if (elems.ht_capabilities && elems.ht_operation)
1611 		recv_beacon->proto_cap |= PROTO_CAP_11N;
1612 
1613 	if (elems.vht_capabilities && elems.vht_operation)
1614 		recv_beacon->proto_cap |= PROTO_CAP_11AC;
1615 
1616 	if (elems.he_capabilities && elems.he_operation)
1617 		recv_beacon->proto_cap |= PROTO_CAP_11AX;
1618 
1619 	/* check bw and channel offset */
1620 	rtw_ies_get_chbw(pos, left, &recv_beacon->ch, &recv_beacon->bw, &recv_beacon->offset, 1, 1);
1621 	if (!recv_beacon->ch) {
1622 		/* we don't find channel IE, so don't check it */
1623 		/* RTW_INFO("Oops: %s we don't find channel IE, so don't check it\n", __func__); */
1624 		recv_beacon->ch = adapter->mlmeextpriv.chandef.chan;
1625 	}
1626 
1627 	/* checking SSID */
1628 	if (elems.ssid) {
1629 		if (elems.ssid_len > sizeof(recv_beacon->ssid))
1630 			return _FALSE;
1631 
1632 		_rtw_memcpy(recv_beacon->ssid, elems.ssid, elems.ssid_len);
1633 		recv_beacon->ssid_len = elems.ssid_len;
1634 	}
1635 
1636 	/* checking RSN first */
1637 	if (elems.rsn_ie && elems.rsn_ie_len) {
1638 		recv_beacon->encryp_protocol = ENCRYP_PROTOCOL_RSN;
1639 		rtw_parse_wpa2_ie(elems.rsn_ie - 2, elems.rsn_ie_len + 2,
1640 			&recv_beacon->group_cipher, &recv_beacon->pairwise_cipher,
1641 			NULL, &recv_beacon->akm, NULL, NULL);
1642 	}
1643 	/* checking WPA secon */
1644 	else if (elems.wpa_ie && elems.wpa_ie_len) {
1645 		recv_beacon->encryp_protocol = ENCRYP_PROTOCOL_WPA;
1646 		rtw_parse_wpa_ie(elems.wpa_ie - 2, elems.wpa_ie_len + 2,
1647 			&recv_beacon->group_cipher, &recv_beacon->pairwise_cipher,
1648 				 &recv_beacon->akm);
1649 	} else if (capability & BIT(4))
1650 		recv_beacon->encryp_protocol = ENCRYP_PROTOCOL_WEP;
1651 
1652 	if (elems.tim && elems.tim_len) {
1653 
1654 		#ifdef DBG_RX_BCN
1655 		_rtw_memcpy(pmlmeext->tim, elems.tim, 4);
1656 		#endif
1657 		pmlmeext->dtim = elems.tim[1];
1658 	}
1659 
1660 	/* checking RTW TBTX */
1661 #ifdef CONFIG_RTW_TOKEN_BASED_XMIT
1662 	if (elems.tbtx_cap && elems.tbtx_cap_len) {
1663 		struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1664 
1665 		if (rtw_is_tbtx_capabilty(elems.tbtx_cap, elems.tbtx_cap_len)){
1666 			RTW_DBG("AP support TBTX\n");
1667 		}
1668 	}
1669 #endif
1670 
1671 	return _TRUE;
1672 }
1673 
rtw_dump_bcn_keys(void * sel,struct beacon_keys * recv_beacon)1674 void rtw_dump_bcn_keys(void *sel, struct beacon_keys *recv_beacon)
1675 {
1676 	u8 ssid[IW_ESSID_MAX_SIZE + 1];
1677 
1678 	_rtw_memcpy(ssid, recv_beacon->ssid, recv_beacon->ssid_len);
1679 	ssid[recv_beacon->ssid_len] = '\0';
1680 
1681 	RTW_PRINT_SEL(sel, "ssid = %s (len = %u)\n", ssid, recv_beacon->ssid_len);
1682 	RTW_PRINT_SEL(sel, "ch = %u,%u,%u\n"
1683 		, recv_beacon->ch, recv_beacon->bw, recv_beacon->offset);
1684 	RTW_PRINT_SEL(sel, "proto_cap = 0x%02x\n", recv_beacon->proto_cap);
1685 	RTW_MAP_DUMP_SEL(sel, "rate_set = "
1686 		, recv_beacon->rate_set, recv_beacon->rate_num);
1687 	RTW_PRINT_SEL(sel, "sec = %d, group = 0x%x, pair = 0x%x, akm = 0x%08x\n"
1688 		, recv_beacon->encryp_protocol, recv_beacon->group_cipher
1689 		, recv_beacon->pairwise_cipher, recv_beacon->akm);
1690 }
1691 
rtw_check_bcn_info(_adapter * adapter,u8 * pframe,u32 packet_len)1692 int rtw_check_bcn_info(_adapter *adapter, u8 *pframe, u32 packet_len)
1693 {
1694 #define BCNKEY_VERIFY_PROTO_CAP 0
1695 #define BCNKEY_VERIFY_WHOLE_RATE_SET 0
1696 
1697 	u8 *pbssid = GetAddr3Ptr(pframe);
1698 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
1699 	struct wlan_network *cur_network = &(adapter->mlmepriv.cur_network);
1700 	struct beacon_keys *cur_beacon = &pmlmepriv->cur_beacon_keys;
1701 	struct beacon_keys recv_beacon;
1702 	int ret = 0;
1703 	u8 ifbmp_m = rtw_mi_get_ap_mesh_ifbmp(adapter);
1704 	u8 ifbmp_s = rtw_mi_get_ld_sta_ifbmp(adapter);
1705 	struct dvobj_priv *d = adapter_to_dvobj(adapter);
1706 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
1707 	struct mlme_ext_info *pmlmeinfo = &pmlmeext->mlmext_info;
1708 	u8 ssid_is_hidden = _FALSE;
1709 
1710 	if (is_client_associated_to_ap(adapter) == _FALSE)
1711 		goto exit_success;
1712 
1713 	if (rtw_get_bcn_keys(adapter, pframe, packet_len, &recv_beacon) == _FALSE)
1714 		goto exit_success; /* parsing failed => broken IE */
1715 
1716 #ifdef DBG_RX_BCN
1717 	rtw_debug_rx_bcn(adapter, pframe, packet_len);
1718 #endif
1719 
1720 	ssid_is_hidden = is_hidden_ssid(recv_beacon.ssid, recv_beacon.ssid_len);
1721 
1722 	if (recv_beacon.ssid_len != cur_beacon->ssid_len) {
1723 		pmlmeinfo->illegal_beacon_code |= SSID_LENGTH_CHANGED;
1724 		pmlmeinfo->illegal_beacon_code |= SSID_CHANGED;
1725 		if (!ssid_is_hidden) {
1726 			RTW_WARN("%s: Ignore ssid len change new %d old %d\n",
1727 				 __func__, recv_beacon.ssid_len,
1728 				 cur_beacon->ssid_len);
1729 		}
1730 	} else if ((_rtw_memcmp(recv_beacon.ssid,
1731 				cur_beacon->ssid,
1732 				cur_beacon->ssid_len) == _FALSE)) {
1733 		pmlmeinfo->illegal_beacon_code |= SSID_CHANGED;
1734 		if (!ssid_is_hidden) {
1735 			RTW_INFO_DUMP("[old ssid]: ",
1736 				      cur_beacon->ssid, cur_beacon->ssid_len);
1737 			RTW_INFO_DUMP("[new ssid]: ",
1738 				      recv_beacon.ssid, cur_beacon->ssid_len);
1739 		}
1740 	}
1741 
1742 	/* hidden ssid, replace with current beacon ssid directly */
1743 	if (ssid_is_hidden) {
1744 		_rtw_memcpy(recv_beacon.ssid, cur_beacon->ssid, cur_beacon->ssid_len);
1745 		recv_beacon.ssid_len = cur_beacon->ssid_len;
1746 	}
1747 
1748 #if CONFIG_DFS
1749 	if (check_fwstate(pmlmepriv, WIFI_CSA_UPDATE_BEACON)) {
1750 		u8 u_ch, u_offset, u_bw;
1751 		u8 bcn_ch, bcn_bw, bcn_offset;
1752 		struct sta_info *psta = NULL;
1753 		struct rtw_chan_def mr_chdef = {0};
1754 		struct rtw_chan_def new_chdef = {0};
1755 
1756 		/* get union ch/bw/offset from chan_ctx */
1757 		rtw_phl_mr_get_chandef(d->phl, adapter->phl_role, &mr_chdef);
1758 		u_ch = mr_chdef.chan;
1759 		u_offset = (u8)mr_chdef.offset;
1760 		u_bw = (u8)mr_chdef.bw;
1761 
1762 		#ifdef DBG_CSA
1763 		RTW_INFO("CSA : before update beacon, u_ch=%u, recv_beacon.ch=%u\n", u_ch, recv_beacon.ch);
1764 		#endif
1765 
1766 		if (pmlmepriv->bcn_cnts_after_csa < 5) {
1767 			if (u_ch == recv_beacon.ch)
1768 				pmlmepriv->bcn_cnts_after_csa += 1;
1769 			goto exit_success;
1770 		} else
1771 			pmlmepriv->bcn_cnts_after_csa = 0;
1772 
1773 		_rtw_memcpy(cur_beacon, &recv_beacon, sizeof(recv_beacon));
1774 		clr_fwstate(pmlmepriv, WIFI_CSA_UPDATE_BEACON);
1775 
1776 		bcn_ch = recv_beacon.ch;
1777 		bcn_bw = recv_beacon.bw;
1778 		bcn_offset = recv_beacon.offset;
1779 
1780 		#ifdef DBG_CSA
1781 		RTW_INFO("CSA : after update beacon, recv_beacon.ch=%u, recv_beacon.bw=%u, recv_beacon.offset=%u\n", \
1782 			bcn_ch, bcn_bw, bcn_offset);
1783 		rtw_dump_bcn_keys(RTW_DBGDUMP, &recv_beacon);
1784 		#endif
1785 
1786 		_cancel_timer_nowait(&pmlmeext->csa_timer);
1787 
1788 		/* beacon bw/offset is different from CSA IE */
1789 		if ((bcn_bw > u_bw) ||
1790 			(bcn_offset != u_offset &&
1791 			u_offset != CHAN_OFFSET_NO_EXT &&
1792 			bcn_offset != CHAN_OFFSET_NO_EXT)) {
1793 
1794 			pmlmeext->chandef.bw = bcn_bw;
1795 			pmlmeext->chandef.offset = bcn_offset;
1796 			/* updaet STA mode DSConfig , ap mode will update in rtw_change_bss_chbw_cmd */
1797 			pmlmepriv->cur_network.network.Configuration.DSConfig = bcn_ch;
1798 
1799 			/* update wifi role chandef */
1800 			rtw_hw_update_chan_def(adapter);
1801 
1802 			/* update chanctx */
1803 			if (rtw_phl_mr_upt_chandef(d->phl, adapter->phl_role) == RTW_PHL_STATUS_FAILURE)
1804 				RTW_ERR("CSA : update chanctx fail\n");
1805 
1806 			rtw_phl_mr_get_chandef(d->phl, adapter->phl_role, &new_chdef);
1807 			rtw_mi_update_union_chan_inf(adapter, new_chdef.chan, (u8)new_chdef.offset, (u8)new_chdef.bw);
1808 
1809 			#ifdef CONFIG_AP_MODE
1810 			if (ifbmp_m) {
1811 				rtw_change_bss_chbw_cmd(dvobj_get_primary_adapter(d), 0
1812 					, ifbmp_m, 0, new_chdef.chan, REQ_BW_ORI, REQ_OFFSET_NONE);
1813 			} else
1814 			#endif
1815 			{
1816 				#ifdef CONFIG_DFS_MASTER
1817 				rtw_dfs_rd_en_decision(adapter, MLME_OPCH_SWITCH, ifbmp_s);
1818 				#endif
1819 				rtw_set_chbw_cmd(adapter, new_chdef.chan, (u8)new_chdef.bw, (u8)new_chdef.offset, 0);
1820 			}
1821 			RTW_INFO("CSA : after update bw/offset, new_bw=%d, new_offset=%d \n", \
1822 				(u8)new_chdef.bw, (u8)new_chdef.offset);
1823 		} else {
1824 			RTW_INFO("CSA : u_ch=%d, u_bw=%d, u_offset=%d, recv_beacon.ch=%d, recv_beacon.bw=%d, recv_beacon.offset=%d\n"
1825 			, u_ch, u_bw, u_offset, bcn_ch, bcn_bw, bcn_offset);
1826 		}
1827 
1828 		/* update RA mask */
1829 		psta = rtw_get_stainfo(&adapter->stapriv, get_bssid(&adapter->mlmepriv));
1830 		if (psta) {
1831 			psta->phl_sta->chandef.bw = pmlmeext->chandef.bw;
1832 			rtw_phl_cmd_change_stainfo(GET_PHL_INFO(d),
1833 						psta->phl_sta,
1834 						STA_CHG_RAMASK,
1835 						NULL,
1836 						0,
1837 						PHL_CMD_NO_WAIT,
1838 						0);
1839 		}
1840 
1841 		RTW_INFO("CSA : update beacon done, WIFI_CSA_UPDATE_BEACON is clear\n");
1842 	}
1843 #endif /* CONFIG_DFS */
1844 
1845 	if (_rtw_memcmp(&recv_beacon, cur_beacon, sizeof(recv_beacon)) == _FALSE) {
1846 		struct beacon_keys tmp_beacon;
1847 
1848 		RTW_INFO(FUNC_ADPT_FMT" new beacon occur!!\n", FUNC_ADPT_ARG(adapter));
1849 		RTW_INFO(FUNC_ADPT_FMT" cur beacon key:\n", FUNC_ADPT_ARG(adapter));
1850 		rtw_dump_bcn_keys(RTW_DBGDUMP, cur_beacon);
1851 		RTW_INFO(FUNC_ADPT_FMT" new beacon key:\n", FUNC_ADPT_ARG(adapter));
1852 		rtw_dump_bcn_keys(RTW_DBGDUMP, &recv_beacon);
1853 
1854 		if (recv_beacon.ch != cur_beacon->ch)
1855 			pmlmeinfo->illegal_beacon_code |= BEACON_CHANNEL_CHANGED;
1856 		if (recv_beacon.encryp_protocol != cur_beacon->encryp_protocol)
1857 			pmlmeinfo->illegal_beacon_code |= ENCRYPT_PROTOCOL_CHANGED;
1858 		if (recv_beacon.pairwise_cipher != cur_beacon->pairwise_cipher)
1859 			pmlmeinfo->illegal_beacon_code |= PAIRWISE_CIPHER_CHANGED;
1860 		if (recv_beacon.group_cipher != cur_beacon->group_cipher)
1861 			pmlmeinfo->illegal_beacon_code |= GROUP_CIPHER_CHANGED;
1862 		if (recv_beacon.akm != cur_beacon->akm)
1863 			pmlmeinfo->illegal_beacon_code |= IS_8021X_CHANGED;
1864 
1865 		if (!rtw_is_chbw_grouped(cur_beacon->ch, cur_beacon->bw, cur_beacon->offset
1866 				, recv_beacon.ch, recv_beacon.bw, recv_beacon.offset))
1867 			goto exit;
1868 
1869 		_rtw_memcpy(&tmp_beacon, cur_beacon, sizeof(tmp_beacon));
1870 
1871 		/* check fields excluding below */
1872 		tmp_beacon.ch = recv_beacon.ch;
1873 		tmp_beacon.bw = recv_beacon.bw;
1874 		tmp_beacon.offset = recv_beacon.offset;
1875 		if (!BCNKEY_VERIFY_PROTO_CAP)
1876 			tmp_beacon.proto_cap = recv_beacon.proto_cap;
1877 		if (!BCNKEY_VERIFY_WHOLE_RATE_SET) {
1878 			tmp_beacon.rate_num = recv_beacon.rate_num;
1879 			_rtw_memcpy(tmp_beacon.rate_set, recv_beacon.rate_set, 12);
1880 		}
1881 		if (_rtw_memcmp(&tmp_beacon, &recv_beacon, sizeof(recv_beacon)) == _FALSE)
1882 			goto exit;
1883 
1884 		_rtw_memcpy(cur_beacon, &recv_beacon, sizeof(recv_beacon));
1885 	}
1886 
1887 exit_success:
1888 	ret = 1;
1889 
1890 exit:
1891 	return ret;
1892 }
1893 
update_beacon_info(_adapter * padapter,u8 * pframe,uint pkt_len,struct sta_info * psta)1894 void update_beacon_info(_adapter *padapter, u8 *pframe, uint pkt_len, struct sta_info *psta)
1895 {
1896 	unsigned int i;
1897 	unsigned int len;
1898 	PNDIS_802_11_VARIABLE_IEs	pIE;
1899 
1900 #ifdef CONFIG_TDLS
1901 	struct tdls_info *ptdlsinfo = &padapter->tdlsinfo;
1902 	u8 tdls_prohibited[] = { 0x00, 0x00, 0x00, 0x00, 0x10 }; /* bit(38): TDLS_prohibited */
1903 #endif /* CONFIG_TDLS */
1904 
1905 	len = pkt_len - (_BEACON_IE_OFFSET_ + WLAN_HDR_A3_LEN);
1906 
1907 	for (i = 0; i < len;) {
1908 		pIE = (PNDIS_802_11_VARIABLE_IEs)(pframe + (_BEACON_IE_OFFSET_ + WLAN_HDR_A3_LEN) + i);
1909 
1910 		switch (pIE->ElementID) {
1911 		case _VENDOR_SPECIFIC_IE_:
1912 			/* to update WMM paramter set while receiving beacon */
1913 			if (_rtw_memcmp(pIE->data, WMM_PARA_OUI, 6) && pIE->Length == WLAN_WMM_LEN)	/* WMM */
1914 				(WMM_param_handler(padapter, pIE)) ? report_wmm_edca_update(padapter) : 0;
1915 
1916 			break;
1917 
1918 		case _HT_EXTRA_INFO_IE_:	/* HT info */
1919 			/* HT_info_handler(padapter, pIE); */
1920 			bwmode_update_check(padapter, pIE);
1921 			break;
1922 #ifdef CONFIG_80211AC_VHT
1923 		case EID_OpModeNotification:
1924 			rtw_process_vht_op_mode_notify(padapter, pIE->data, psta);
1925 			break;
1926 #endif /* CONFIG_80211AC_VHT */
1927 		case _ERPINFO_IE_:
1928 			ERP_IE_handler(padapter, pIE);
1929 			VCS_update(padapter, psta);
1930 			break;
1931 
1932 #ifdef CONFIG_TDLS
1933 		case WLAN_EID_EXT_CAP:
1934 			if (check_ap_tdls_prohibited(pIE->data, pIE->Length) == _TRUE)
1935 				ptdlsinfo->ap_prohibited = _TRUE;
1936 			if (check_ap_tdls_ch_switching_prohibited(pIE->data, pIE->Length) == _TRUE)
1937 				ptdlsinfo->ch_switch_prohibited = _TRUE;
1938 			break;
1939 #endif /* CONFIG_TDLS */
1940 #ifdef CONFIG_80211AX_HE
1941 		case WLAN_EID_EXTENSION:
1942 			if (pIE->data[0] == WLAN_EID_EXTENSION_HE_OPERATION)
1943 				HE_operation_handler(padapter, pIE);
1944 			else if (pIE->data[0] == WLAN_EID_EXTENSION_HE_MU_EDCA)
1945 				HE_mu_edca_handler(padapter, pIE, _FALSE);
1946 			break;
1947 #endif
1948 		default:
1949 			break;
1950 		}
1951 
1952 		i += (pIE->Length + 2);
1953 	}
1954 }
1955 
1956 #if CONFIG_DFS
process_csa_ie(_adapter * padapter,u8 * ies,uint ies_len)1957 void process_csa_ie(_adapter *padapter, u8 *ies, uint ies_len)
1958 {
1959 #ifdef CONFIG_ECSA_PHL
1960 	struct core_ecsa_info *ecsa_info = &(padapter->ecsa_info);
1961 	struct rtw_phl_ecsa_param *ecsa_param = &(ecsa_info->phl_ecsa_param);
1962 #endif
1963 	struct rf_ctl_t *rfctl = adapter_to_rfctl(padapter);
1964 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1965 	struct mlme_ext_info *pmlmeinfo = &(pmlmeext->mlmext_info);
1966 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
1967 	unsigned int i, j, countdown;
1968 	PNDIS_802_11_VARIABLE_IEs pIE, sub_pie;
1969 	u8 ch = 0, csa_ch_offset = 0, csa_ch_width = 0, csa_ch_freq_seg0 = 0, csa_ch_freq_seg1 = 0;
1970 	u8 csa_switch_cnt = 0, csa_mode = 0;
1971 	u8 is_csa_running;
1972 
1973 #ifdef DBG_CSA
1974 	u8 *p;
1975 	u32 ie_len = 0;
1976 	p = rtw_get_ie(ies, WLAN_EID_CHANNEL_SWITCH, &ie_len, ies_len);
1977 	if (p && ie_len > 0)
1978 		RTW_INFO("CSA : %s, CH = %u, count = %u\n",__func__, *(p+2+1), *(p+2+2));
1979 #endif
1980 
1981 #ifdef CONFIG_ECSA_PHL
1982 	is_csa_running = rtw_mr_is_ecsa_running(padapter);
1983 #else
1984 	is_csa_running = (rfctl->csa_chandef.chan > 0) ? _TRUE : _FALSE;
1985 #endif
1986 
1987 	/* compare with scheduling CSA to block incoming CSA IE */
1988 	if (is_csa_running || check_fwstate(pmlmepriv, WIFI_CSA_UPDATE_BEACON))
1989 		return;
1990 
1991 	for (i = 0; i + 1 < ies_len;) {
1992 		pIE = (PNDIS_802_11_VARIABLE_IEs)(ies + i);
1993 
1994 		switch (pIE->ElementID) {
1995 		case WLAN_EID_CHANNEL_SWITCH:
1996 			csa_mode = *(pIE->data);
1997 			ch = *(pIE->data + 1);
1998 			csa_switch_cnt = *(pIE->data + 2);
1999 
2000 			RTW_INFO("CSA : mode = %u, ch = %u, switch count = %u\n",
2001 				csa_mode, ch, csa_switch_cnt);
2002 			break;
2003 		case WLAN_EID_SECONDARY_CHANNEL_OFFSET:
2004 			csa_ch_offset = *(pIE->data);
2005 			break;
2006 		case WLAN_EID_WIDE_BANDWIDTH_CHANNEL_SWITCH:
2007 			csa_ch_width = *(pIE->data);
2008 			csa_ch_freq_seg0 = *(pIE->data+1);
2009 			csa_ch_freq_seg1 = *(pIE->data+2);
2010 
2011 			RTW_INFO("CSA : bw:%02x center_freq_0:%u center_freq_1:%u, ch=%u\n"
2012 					, csa_ch_width, csa_ch_freq_seg0, csa_ch_freq_seg1, ch);
2013 			break;
2014 		case WLAN_EID_CHANNEL_SWITCH_WRAPPER:
2015 			for (j=0; j + 1 < pIE->Length;) {
2016 				sub_pie = (PNDIS_802_11_VARIABLE_IEs)(ies + i + j + 2);
2017 				if (sub_pie->ElementID == WLAN_EID_WIDE_BANDWIDTH_CHANNEL_SWITCH) {
2018 					csa_ch_width = *(sub_pie->data);
2019 					csa_ch_freq_seg0 = *(sub_pie->data+1);
2020 					csa_ch_freq_seg1 = *(sub_pie->data+2);
2021 
2022 					RTW_INFO("CSA : sub_IE:%02x IE_length:%02x bw:%02x center_freq_0:%u center_freq_1:%u, ch=%u\n"
2023 						, sub_pie->ElementID, sub_pie->Length, csa_ch_width, csa_ch_freq_seg0, csa_ch_freq_seg1, ch);
2024 				}
2025 				j += (sub_pie->Length + 2);
2026 			}
2027 			break;
2028 		default:
2029 			break;
2030 		}
2031 
2032 		i += (pIE->Length + 2);
2033 	}
2034 
2035 	/* Doesn't support switch bandwidth/offset in the same channel for now */
2036 	if (ch == rtw_mi_get_union_chan(padapter)) {
2037 		RTW_ERR("%s : receive the same channel from CSA IE, so ignore it\n", __func__);
2038 		return;
2039 	}
2040 
2041 	if (ch != 0) {
2042 		#ifdef CONFIG_ECSA_PHL
2043 		ecsa_param->ecsa_type = ECSA_TYPE_STA;
2044 		ecsa_param->mode = csa_mode;
2045 		ecsa_param->count = csa_switch_cnt;
2046 		/* ecsa_param.op_class = rfctl->op_class; */ /* TODO : ECSA */
2047 		ecsa_param->new_chan_def.band = rtw_phl_get_band_type(ch);
2048 		ecsa_param->new_chan_def.chan = ch;
2049 		/* The channel width defined in 802.11 spec */
2050 		ecsa_info->channel_width = csa_ch_width;
2051 		ecsa_param->new_chan_def.offset = csa_ch_offset;
2052 		ecsa_param->new_chan_def.center_freq1 = csa_ch_freq_seg0;
2053 		ecsa_param->new_chan_def.center_freq2 = csa_ch_freq_seg1;
2054 		ecsa_param->flag = 0;
2055 		ecsa_param->delay_start_ms = 0;
2056 
2057 		SET_ECSA_STATE(padapter, ECSA_ST_SW_START);
2058 		rtw_trigger_phl_ecsa_start(padapter);
2059 		#else
2060 		rfctl->csa_mode = csa_mode;
2061 		rfctl->csa_switch_cnt = csa_switch_cnt;
2062 		rfctl->csa_ch_width = csa_ch_width;
2063 
2064 		rfctl->csa_chandef.band = rtw_phl_get_band_type(ch);
2065 		rfctl->csa_chandef.chan = ch;
2066 		rfctl->csa_chandef.offset = csa_ch_offset;
2067 		rfctl->csa_chandef.center_freq1= csa_ch_freq_seg0;
2068 		rfctl->csa_chandef.center_freq2 = csa_ch_freq_seg1;
2069 
2070 		countdown = pmlmeinfo->network.Configuration.BeaconPeriod * (csa_switch_cnt+1); /* ms */
2071 		RTW_INFO("csa: set countdown timer to %d ms\n", countdown);
2072 		_set_timer(&pmlmeext->csa_timer, countdown);
2073 		#endif /* CONFIG_ECSA_PHL */
2074 	}
2075 }
2076 #endif /* CONFIG_DFS */
2077 
2078 #ifdef CONFIG_80211D
rtw_iface_accept_country_ie(_adapter * adapter)2079 bool rtw_iface_accept_country_ie(_adapter *adapter)
2080 {
2081 	struct rf_ctl_t *rfctl = adapter_to_rfctl(adapter);
2082 
2083 	if (!(BIT(adapter->iface_id) & rfctl->country_ie_slave_en_ifbmp))
2084 		return 0;
2085 	if (!MLME_IS_STA(adapter))
2086 		return 0;
2087 	if (!MLME_IS_GC(adapter)) {
2088 		if (!(rfctl->country_ie_slave_en_role & COUNTRY_IE_SLAVE_EN_ROLE_STA))
2089 			return 0;
2090 	} else {
2091 		if (!(rfctl->country_ie_slave_en_role & COUNTRY_IE_SLAVE_EN_ROLE_GC))
2092 			return 0;
2093 	}
2094 	return 1;
2095 }
2096 
process_country_ie(_adapter * adapter,u8 * ies,uint ies_len)2097 void process_country_ie(_adapter *adapter, u8 *ies, uint ies_len)
2098 {
2099 	struct rf_ctl_t *rfctl = adapter_to_rfctl(adapter);
2100 
2101 	if (rfctl->regd_src == REGD_SRC_RTK_PRIV
2102 		&& !rtw_rfctl_is_disable_sw_channel_plan(rfctl_to_dvobj(rfctl))
2103 		&& !rfctl->csa_chandef.chan /* don't process country ie when under CSA processing */
2104 	) {
2105 		struct mlme_priv *mlme = &adapter->mlmepriv;
2106 		const u8 *ie = NULL;
2107 		sint ie_len = 0;
2108 
2109 		if (rtw_iface_accept_country_ie(adapter)) {
2110 			ie = rtw_get_ie(ies, WLAN_EID_COUNTRY, &ie_len, ies_len);
2111 			if (ie) {
2112 				if (ie_len < 6) {
2113 					ie = NULL;
2114 					ie_len = 0;
2115 				} else
2116 					ie_len += 2;
2117 			}
2118 		}
2119 
2120 		if (!mlme->recv_country_ie && !ie)
2121 			return;
2122 		if (mlme->recv_country_ie_len == ie_len
2123 			&& _rtw_memcmp(mlme->recv_country_ie, ie, ie_len) == _TRUE)
2124 			return;
2125 
2126 		if (!ie) {
2127 			rtw_buf_free(&mlme->recv_country_ie, &mlme->recv_country_ie_len);
2128 			rtw_apply_recv_country_ie_cmd(adapter, 0, 0, 0, NULL);
2129 		} else {
2130 			char ori_alpha2[2] = {0, 0};
2131 
2132 			if (mlme->recv_country_ie)
2133 				_rtw_memcpy(ori_alpha2, mlme->recv_country_ie + 2, 2);
2134 
2135 			rtw_buf_update(&mlme->recv_country_ie, &mlme->recv_country_ie_len, ie, ie_len);
2136 			/* for now only country code is used */
2137 			if (_rtw_memcmp(ori_alpha2, mlme->recv_country_ie + 2, 2) == _TRUE)
2138 				return;
2139 			RTW_INFO(FUNC_ADPT_FMT" country_ie alpha2 changed\n", FUNC_ADPT_ARG(adapter));
2140 			rtw_apply_recv_country_ie_cmd(adapter, 0
2141 				, mlme->cur_beacon_keys.ch > 14 ? BAND_ON_5G : BAND_ON_24G
2142 				, mlme->cur_beacon_keys.ch, mlme->recv_country_ie);
2143 		}
2144 	}
2145 }
2146 #endif /* CONFIG_80211D */
2147 
parsing_eapol_packet(_adapter * padapter,u8 * key_payload,struct sta_info * psta,u8 trx_type)2148 enum eap_type parsing_eapol_packet(_adapter *padapter, u8 *key_payload, struct sta_info *psta, u8 trx_type)
2149 {
2150 	struct security_priv *psecuritypriv = &(padapter->securitypriv);
2151 	struct ieee802_1x_hdr *hdr;
2152 	struct wpa_eapol_key *key;
2153 	u16 key_info, key_data_length;
2154 	char *trx_msg = trx_type ? "send" : "recv";
2155 	enum eap_type eapol_type;
2156 
2157 	hdr = (struct ieee802_1x_hdr *) key_payload;
2158 
2159 	 /* WPS - eapol start packet */
2160 	if (hdr->type == 1 && hdr->length == 0) {
2161 		RTW_INFO("%s eapol start packet\n", trx_msg);
2162 		return EAPOL_START;
2163 	}
2164 
2165 	if (hdr->type == 0) { /* WPS - eapol packet */
2166 		RTW_INFO("%s eapol packet\n", trx_msg);
2167 		return EAPOL_PACKET;
2168 	}
2169 
2170 	key = (struct wpa_eapol_key *) (hdr + 1);
2171 	key_info = be16_to_cpu(*((u16 *)(key->key_info)));
2172 	key_data_length = be16_to_cpu(*((u16 *)(key->key_data_length)));
2173 
2174 	if (!(key_info & WPA_KEY_INFO_KEY_TYPE)) { /* WPA group key handshake */
2175 		if (key_info & WPA_KEY_INFO_ACK) {
2176 			RTW_PRINT("%s eapol packet - WPA Group Key 1/2\n", trx_msg);
2177 			eapol_type = EAPOL_WPA_GROUP_KEY_1_2;
2178 		} else {
2179 			RTW_PRINT("%s eapol packet - WPA Group Key 2/2\n", trx_msg);
2180 			eapol_type = EAPOL_WPA_GROUP_KEY_2_2;
2181 
2182 			/* WPA key-handshake has completed */
2183 			if (psecuritypriv->ndisauthtype == Ndis802_11AuthModeWPAPSK)
2184 				psta->state &= (~WIFI_UNDER_KEY_HANDSHAKE);
2185 		}
2186 	} else if (key_info & WPA_KEY_INFO_MIC) {
2187 		if (key_data_length == 0) {
2188 			RTW_PRINT("%s eapol packet 4/4\n", trx_msg);
2189 			eapol_type = EAPOL_4_4;
2190 		} else if (key_info & WPA_KEY_INFO_ACK) {
2191 			RTW_PRINT("%s eapol packet 3/4\n", trx_msg);
2192 			eapol_type = EAPOL_3_4;
2193 		} else {
2194 			RTW_PRINT("%s eapol packet 2/4\n", trx_msg);
2195 			eapol_type = EAPOL_2_4;
2196 		}
2197 	} else {
2198 		RTW_PRINT("%s eapol packet 1/4\n", trx_msg);
2199 		eapol_type = EAPOL_1_4;
2200 	}
2201 
2202 	return eapol_type;
2203 }
2204 
is_ap_in_tkip(_adapter * padapter)2205 unsigned int is_ap_in_tkip(_adapter *padapter)
2206 {
2207 	u32 i;
2208 	PNDIS_802_11_VARIABLE_IEs	pIE;
2209 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
2210 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
2211 	WLAN_BSSID_EX		*cur_network = &(pmlmeinfo->network);
2212 
2213 	if (rtw_get_capability((WLAN_BSSID_EX *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
2214 		for (i = sizeof(NDIS_802_11_FIXED_IEs); i < pmlmeinfo->network.IELength;) {
2215 			pIE = (PNDIS_802_11_VARIABLE_IEs)(pmlmeinfo->network.IEs + i);
2216 
2217 			switch (pIE->ElementID) {
2218 			case _VENDOR_SPECIFIC_IE_:
2219 				if ((_rtw_memcmp(pIE->data, RTW_WPA_OUI, 4)) && (_rtw_memcmp((pIE->data + 12), WPA_TKIP_CIPHER, 4)))
2220 					return _TRUE;
2221 				break;
2222 
2223 			case _RSN_IE_2_:
2224 				if (_rtw_memcmp((pIE->data + 8), RSN_TKIP_CIPHER, 4))
2225 					return _TRUE;
2226 
2227 			default:
2228 				break;
2229 			}
2230 
2231 			i += (pIE->Length + 2);
2232 		}
2233 
2234 		return _FALSE;
2235 	} else
2236 		return _FALSE;
2237 
2238 }
2239 
should_forbid_n_rate(_adapter * padapter)2240 unsigned int should_forbid_n_rate(_adapter *padapter)
2241 {
2242 	u32 i;
2243 	PNDIS_802_11_VARIABLE_IEs	pIE;
2244 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
2245 	WLAN_BSSID_EX  *cur_network = &pmlmepriv->cur_network.network;
2246 
2247 	if (rtw_get_capability((WLAN_BSSID_EX *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
2248 		for (i = sizeof(NDIS_802_11_FIXED_IEs); i < cur_network->IELength;) {
2249 			pIE = (PNDIS_802_11_VARIABLE_IEs)(cur_network->IEs + i);
2250 
2251 			switch (pIE->ElementID) {
2252 			case _VENDOR_SPECIFIC_IE_:
2253 				if (_rtw_memcmp(pIE->data, RTW_WPA_OUI, 4) &&
2254 				    ((_rtw_memcmp((pIE->data + 12), WPA_CIPHER_SUITE_CCMP, 4)) ||
2255 				     (_rtw_memcmp((pIE->data + 16), WPA_CIPHER_SUITE_CCMP, 4))))
2256 					return _FALSE;
2257 				break;
2258 
2259 			case _RSN_IE_2_:
2260 				if ((_rtw_memcmp((pIE->data + 8), RSN_CIPHER_SUITE_CCMP, 4))  ||
2261 				    (_rtw_memcmp((pIE->data + 12), RSN_CIPHER_SUITE_CCMP, 4)))
2262 					return _FALSE;
2263 
2264 			default:
2265 				break;
2266 			}
2267 
2268 			i += (pIE->Length + 2);
2269 		}
2270 
2271 		return _TRUE;
2272 	} else
2273 		return _FALSE;
2274 
2275 }
2276 
2277 
is_ap_in_wep(_adapter * padapter)2278 unsigned int is_ap_in_wep(_adapter *padapter)
2279 {
2280 	u32 i;
2281 	PNDIS_802_11_VARIABLE_IEs	pIE;
2282 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
2283 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
2284 	WLAN_BSSID_EX		*cur_network = &(pmlmeinfo->network);
2285 
2286 	if (rtw_get_capability((WLAN_BSSID_EX *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
2287 		for (i = sizeof(NDIS_802_11_FIXED_IEs); i < pmlmeinfo->network.IELength;) {
2288 			pIE = (PNDIS_802_11_VARIABLE_IEs)(pmlmeinfo->network.IEs + i);
2289 
2290 			switch (pIE->ElementID) {
2291 			case _VENDOR_SPECIFIC_IE_:
2292 				if (_rtw_memcmp(pIE->data, RTW_WPA_OUI, 4))
2293 					return _FALSE;
2294 				break;
2295 
2296 			case _RSN_IE_2_:
2297 				return _FALSE;
2298 
2299 			default:
2300 				break;
2301 			}
2302 
2303 			i += (pIE->Length + 2);
2304 		}
2305 
2306 		return _TRUE;
2307 	} else
2308 		return _FALSE;
2309 
2310 }
2311 
2312 int wifirate2_ratetbl_inx(unsigned char rate);
wifirate2_ratetbl_inx(unsigned char rate)2313 int wifirate2_ratetbl_inx(unsigned char rate)
2314 {
2315 	int	inx = 0;
2316 	rate = rate & 0x7f;
2317 
2318 	switch (rate) {
2319 	case 54*2:
2320 		inx = 11;
2321 		break;
2322 
2323 	case 48*2:
2324 		inx = 10;
2325 		break;
2326 
2327 	case 36*2:
2328 		inx = 9;
2329 		break;
2330 
2331 	case 24*2:
2332 		inx = 8;
2333 		break;
2334 
2335 	case 18*2:
2336 		inx = 7;
2337 		break;
2338 
2339 	case 12*2:
2340 		inx = 6;
2341 		break;
2342 
2343 	case 9*2:
2344 		inx = 5;
2345 		break;
2346 
2347 	case 6*2:
2348 		inx = 4;
2349 		break;
2350 
2351 	case 11*2:
2352 		inx = 3;
2353 		break;
2354 	case 11:
2355 		inx = 2;
2356 		break;
2357 
2358 	case 2*2:
2359 		inx = 1;
2360 		break;
2361 
2362 	case 1*2:
2363 		inx = 0;
2364 		break;
2365 
2366 	}
2367 	return inx;
2368 }
2369 
update_basic_rate(unsigned char * ptn,unsigned int ptn_sz)2370 unsigned int update_basic_rate(unsigned char *ptn, unsigned int ptn_sz)
2371 {
2372 	unsigned int i, num_of_rate;
2373 	unsigned int mask = 0;
2374 
2375 	num_of_rate = (ptn_sz > NumRates) ? NumRates : ptn_sz;
2376 
2377 	for (i = 0; i < num_of_rate; i++) {
2378 		if ((*(ptn + i)) & 0x80)
2379 			mask |= 0x1 << wifirate2_ratetbl_inx(*(ptn + i));
2380 	}
2381 	return mask;
2382 }
2383 
update_supported_rate(unsigned char * ptn,unsigned int ptn_sz)2384 unsigned int update_supported_rate(unsigned char *ptn, unsigned int ptn_sz)
2385 {
2386 	unsigned int i, num_of_rate;
2387 	unsigned int mask = 0;
2388 
2389 	num_of_rate = (ptn_sz > NumRates) ? NumRates : ptn_sz;
2390 
2391 	for (i = 0; i < num_of_rate; i++)
2392 		mask |= 0x1 << wifirate2_ratetbl_inx(*(ptn + i));
2393 
2394 	return mask;
2395 }
2396 
support_short_GI(_adapter * padapter,struct HT_caps_element * pHT_caps,u8 bwmode)2397 int support_short_GI(_adapter *padapter, struct HT_caps_element *pHT_caps, u8 bwmode)
2398 {
2399 	unsigned char					bit_offset;
2400 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
2401 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
2402 
2403 	if (!(pmlmeinfo->HT_enable))
2404 		return _FAIL;
2405 
2406 	bit_offset = (bwmode & CHANNEL_WIDTH_40) ? 6 : 5;
2407 
2408 	if (pHT_caps->u.HT_cap_element.HT_caps_info & (0x1 << bit_offset))
2409 		return _SUCCESS;
2410 	else
2411 		return _FAIL;
2412 }
2413 
get_highest_rate_idx(u64 mask)2414 unsigned char get_highest_rate_idx(u64 mask)
2415 {
2416 	int i;
2417 	unsigned char rate_idx = 0;
2418 
2419 	for (i = 63; i >= 0; i--) {
2420 		if ((mask >> i) & 0x01) {
2421 			rate_idx = i;
2422 			break;
2423 		}
2424 	}
2425 
2426 	return rate_idx;
2427 }
get_lowest_rate_idx_ex(u64 mask,int start_bit)2428 unsigned char get_lowest_rate_idx_ex(u64 mask, int start_bit)
2429 {
2430 	int i;
2431 	unsigned char rate_idx = 0;
2432 
2433 	for (i = start_bit; i < 64; i++) {
2434 		if ((mask >> i) & 0x01) {
2435 			rate_idx = i;
2436 			break;
2437 		}
2438 	}
2439 
2440 	return rate_idx;
2441 }
2442 
get_highest_bw_cap(u8 bwmode)2443 u8 get_highest_bw_cap(u8 bwmode)
2444 {
2445 	u8 hbw = CHANNEL_WIDTH_20;
2446 
2447 	if (bwmode & BW_CAP_80_80M)
2448 		hbw = CHANNEL_WIDTH_80_80;
2449 	else if (bwmode & BW_CAP_160M)
2450 		hbw = CHANNEL_WIDTH_160;
2451 	else if (bwmode & BW_CAP_80M)
2452 		hbw = CHANNEL_WIDTH_80;
2453 	else if (bwmode & BW_CAP_40M)
2454 		hbw = CHANNEL_WIDTH_40;
2455 	else if (bwmode & BW_CAP_20M)
2456 		hbw = CHANNEL_WIDTH_20;
2457 	else if (bwmode & BW_CAP_10M)
2458 		hbw = CHANNEL_WIDTH_10;
2459 	else if (bwmode & BW_CAP_5M)
2460 		hbw = CHANNEL_WIDTH_5;
2461 
2462 	return hbw;
2463 }
2464 
2465 /* Update RRSR and Rate for USERATE */
update_tx_basic_rate(_adapter * padapter,u8 wirelessmode)2466 void update_tx_basic_rate(_adapter *padapter, u8 wirelessmode)
2467 {
2468 	NDIS_802_11_RATES_EX	supported_rates;
2469 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
2470 
2471 	_rtw_memset(supported_rates, 0, NDIS_802_11_LENGTH_RATES_EX);
2472 
2473 	/* clear B mod if current channel is in 5G band, avoid tx cck rate in 5G band. */
2474 	if (pmlmeext->chandef.chan > 14)
2475 		wirelessmode &= ~(WLAN_MD_11B);
2476 
2477 	if ((wirelessmode & WLAN_MD_11B) && (wirelessmode == WLAN_MD_11B))
2478 		_rtw_memcpy(supported_rates, rtw_basic_rate_cck, 4);
2479 	else if (wirelessmode & WLAN_MD_11B)
2480 		_rtw_memcpy(supported_rates, rtw_basic_rate_mix, 7);
2481 	else
2482 		_rtw_memcpy(supported_rates, rtw_basic_rate_ofdm, 3);
2483 
2484 	if (wirelessmode & WLAN_MD_11B)
2485 		update_mgnt_tx_rate(padapter, IEEE80211_CCK_RATE_1MB);
2486 	else
2487 		update_mgnt_tx_rate(padapter, IEEE80211_OFDM_RATE_6MB);
2488 
2489 	rtw_hal_set_hwreg(padapter, HW_VAR_BASIC_RATE, supported_rates);
2490 }
2491 
check_assoc_AP(u8 * pframe,uint len)2492 unsigned char check_assoc_AP(u8 *pframe, uint len)
2493 {
2494 	unsigned int	i;
2495 	PNDIS_802_11_VARIABLE_IEs	pIE;
2496 
2497 	for (i = sizeof(NDIS_802_11_FIXED_IEs); i < len;) {
2498 		pIE = (PNDIS_802_11_VARIABLE_IEs)(pframe + i);
2499 
2500 		switch (pIE->ElementID) {
2501 		case _VENDOR_SPECIFIC_IE_:
2502 			if ((_rtw_memcmp(pIE->data, ARTHEROS_OUI1, 3)) || (_rtw_memcmp(pIE->data, ARTHEROS_OUI2, 3))) {
2503 				RTW_INFO("link to Artheros AP\n");
2504 				return HT_IOT_PEER_ATHEROS;
2505 			} else if ((_rtw_memcmp(pIE->data, BROADCOM_OUI1, 3))
2506 				   || (_rtw_memcmp(pIE->data, BROADCOM_OUI2, 3))
2507 				|| (_rtw_memcmp(pIE->data, BROADCOM_OUI3, 3))) {
2508 				RTW_INFO("link to Broadcom AP\n");
2509 				return HT_IOT_PEER_BROADCOM;
2510 			} else if (_rtw_memcmp(pIE->data, MARVELL_OUI, 3)) {
2511 				RTW_INFO("link to Marvell AP\n");
2512 				return HT_IOT_PEER_MARVELL;
2513 			} else if (_rtw_memcmp(pIE->data, RALINK_OUI, 3)) {
2514 				RTW_INFO("link to Ralink AP\n");
2515 				return HT_IOT_PEER_RALINK;
2516 			} else if (_rtw_memcmp(pIE->data, CISCO_OUI, 3)) {
2517 				RTW_INFO("link to Cisco AP\n");
2518 				return HT_IOT_PEER_CISCO;
2519 			} else if (_rtw_memcmp(pIE->data, REALTEK_OUI, 3)) {
2520 				u32	Vender = HT_IOT_PEER_REALTEK;
2521 
2522 				if (pIE->Length >= 5) {
2523 					if (pIE->data[4] == 1) {
2524 						/* if(pIE->data[5] & RT_HT_CAP_USE_LONG_PREAMBLE) */
2525 						/*	bssDesc->BssHT.RT2RT_HT_Mode |= RT_HT_CAP_USE_LONG_PREAMBLE; */
2526 
2527 						if (pIE->data[5] & RT_HT_CAP_USE_92SE) {
2528 							/* bssDesc->BssHT.RT2RT_HT_Mode |= RT_HT_CAP_USE_92SE; */
2529 							Vender = HT_IOT_PEER_REALTEK_92SE;
2530 						}
2531 					}
2532 
2533 					if (pIE->data[5] & RT_HT_CAP_USE_SOFTAP)
2534 						Vender = HT_IOT_PEER_REALTEK_SOFTAP;
2535 
2536 					if (pIE->data[4] == 2) {
2537 						if (pIE->data[6] & RT_HT_CAP_USE_JAGUAR_CBV) {
2538 							Vender = HT_IOT_PEER_REALTEK_JAGUAR_CBVAP;
2539 							RTW_INFO("link to Realtek JAGUAR_CBVAP\n");
2540 						}
2541 						if (pIE->data[6] & RT_HT_CAP_USE_JAGUAR_CCV) {
2542 							Vender = HT_IOT_PEER_REALTEK_JAGUAR_CCVAP;
2543 							RTW_INFO("link to Realtek JAGUAR_CCVAP\n");
2544 						}
2545 					}
2546 				}
2547 
2548 				RTW_INFO("link to Realtek AP\n");
2549 				return Vender;
2550 			} else if (_rtw_memcmp(pIE->data, AIRGOCAP_OUI, 3)) {
2551 				RTW_INFO("link to Airgo Cap\n");
2552 				return HT_IOT_PEER_AIRGO;
2553 			} else
2554 				break;
2555 
2556 		default:
2557 			break;
2558 		}
2559 
2560 		i += (pIE->Length + 2);
2561 	}
2562 
2563 	RTW_INFO("link to new AP\n");
2564 	return HT_IOT_PEER_UNKNOWN;
2565 }
2566 
get_assoc_AP_Vendor(char * vendor,u8 assoc_AP_vendor)2567 void get_assoc_AP_Vendor(char *vendor, u8 assoc_AP_vendor)
2568 {
2569 	switch (assoc_AP_vendor) {
2570 
2571 	case HT_IOT_PEER_UNKNOWN:
2572 	sprintf(vendor, "%s", "unknown");
2573 	break;
2574 
2575 	case HT_IOT_PEER_REALTEK:
2576 	case HT_IOT_PEER_REALTEK_92SE:
2577 	case HT_IOT_PEER_REALTEK_SOFTAP:
2578 	case HT_IOT_PEER_REALTEK_JAGUAR_CBVAP:
2579 	case HT_IOT_PEER_REALTEK_JAGUAR_CCVAP:
2580 
2581 	sprintf(vendor, "%s", "Realtek");
2582 	break;
2583 
2584 	case HT_IOT_PEER_BROADCOM:
2585 	sprintf(vendor, "%s", "Broadcom");
2586 	break;
2587 
2588 	case HT_IOT_PEER_MARVELL:
2589 	sprintf(vendor, "%s", "Marvell");
2590 	break;
2591 
2592 	case HT_IOT_PEER_RALINK:
2593 	sprintf(vendor, "%s", "Ralink");
2594 	break;
2595 
2596 	case HT_IOT_PEER_CISCO:
2597 	sprintf(vendor, "%s", "Cisco");
2598 	break;
2599 
2600 	case HT_IOT_PEER_AIRGO:
2601 	sprintf(vendor, "%s", "Airgo");
2602 	break;
2603 
2604 	case HT_IOT_PEER_ATHEROS:
2605 	sprintf(vendor, "%s", "Atheros");
2606 	break;
2607 
2608 	default:
2609 	sprintf(vendor, "%s", "unkown");
2610 	break;
2611 	}
2612 
2613 }
2614 #ifdef CONFIG_RTS_FULL_BW
rtw_parse_sta_vendor_ie_8812(_adapter * adapter,struct sta_info * sta,u8 * tlv_ies,u16 tlv_ies_len)2615 void rtw_parse_sta_vendor_ie_8812(_adapter *adapter, struct sta_info *sta, u8 *tlv_ies, u16 tlv_ies_len)
2616 {
2617 	unsigned char REALTEK_OUI[] = {0x00,0xe0, 0x4c};
2618 	u8 *p;
2619 
2620 	p = rtw_get_ie_ex(tlv_ies, tlv_ies_len, WLAN_EID_VENDOR_SPECIFIC, REALTEK_OUI, 3, NULL, NULL);
2621 	if (!p)
2622 		goto exit;
2623 	else {
2624 		if(*(p+1) > 6 ) {
2625 
2626 			if(*(p+6) != 2)
2627 				goto exit;
2628 
2629 			if(*(p+8) == RT_HT_CAP_USE_JAGUAR_CBV)
2630 				sta->vendor_8812 = TRUE;
2631 			else if (*(p+8) == RT_HT_CAP_USE_JAGUAR_CCV)
2632 				sta->vendor_8812 = TRUE;
2633 		}
2634 	}
2635 exit:
2636 	return;
2637 }
2638 #endif/*CONFIG_RTS_FULL_BW*/
2639 
2640 #ifdef CONFIG_80211AC_VHT
get_vht_bf_cap(u8 * pframe,uint len,struct vht_bf_cap * bf_cap)2641 void get_vht_bf_cap(u8 *pframe, uint len, struct vht_bf_cap *bf_cap)
2642 {
2643 	unsigned int i;
2644 	PNDIS_802_11_VARIABLE_IEs pIE;
2645 
2646 	for (i = sizeof(NDIS_802_11_FIXED_IEs); i < len;) {
2647 		pIE = (PNDIS_802_11_VARIABLE_IEs)(pframe + i);
2648 
2649 		switch (pIE->ElementID) {
2650 
2651 		case EID_VHTCapability:
2652 			bf_cap->is_mu_bfer = GET_VHT_CAPABILITY_ELE_MU_BFER(pIE->data);
2653 			bf_cap->su_sound_dim = GET_VHT_CAPABILITY_ELE_SU_BFER_SOUND_DIM_NUM(pIE->data);
2654 			break;
2655 		default:
2656 			break;
2657 		}
2658 		i += (pIE->Length + 2);
2659 	}
2660 }
2661 #endif
2662 
update_capinfo(_adapter * adapter,u16 updateCap)2663 void update_capinfo(_adapter *adapter, u16 updateCap)
2664 {
2665 	struct mlme_ext_priv	*pmlmeext = &adapter->mlmeextpriv;
2666 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
2667 	BOOLEAN		ShortPreamble;
2668 
2669 	/* Check preamble mode, 2005.01.06, by rcnjko. */
2670 	/* Mark to update preamble value forever, 2008.03.18 by lanhsin */
2671 	/* if( pMgntInfo->RegPreambleMode == PREAMBLE_AUTO ) */
2672 	{
2673 
2674 		if (updateCap & cShortPreamble) {
2675 			/* Short Preamble */
2676 			if (pmlmeinfo->preamble_mode != PREAMBLE_SHORT) { /* PREAMBLE_LONG or PREAMBLE_AUTO */
2677 				ShortPreamble = _TRUE;
2678 				pmlmeinfo->preamble_mode = PREAMBLE_SHORT;
2679 				rtw_hal_set_hwreg(adapter, HW_VAR_ACK_PREAMBLE, (u8 *)&ShortPreamble);
2680 			}
2681 		} else {
2682 			/* Long Preamble */
2683 			if (pmlmeinfo->preamble_mode != PREAMBLE_LONG) { /* PREAMBLE_SHORT or PREAMBLE_AUTO */
2684 				ShortPreamble = _FALSE;
2685 				pmlmeinfo->preamble_mode = PREAMBLE_LONG;
2686 				rtw_hal_set_hwreg(adapter, HW_VAR_ACK_PREAMBLE, (u8 *)&ShortPreamble);
2687 			}
2688 		}
2689 	}
2690 
2691 	if (updateCap & cIBSS) {
2692 		/* Filen: See 802.11-2007 p.91 */
2693 		pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
2694 	} else {
2695 		/* Filen: See 802.11-2007 p.90 */
2696 		if (pmlmeext->cur_wireless_mode & (WLAN_MD_11N | WLAN_MD_11A | WLAN_MD_11AC))
2697 			pmlmeinfo->slotTime = SHORT_SLOT_TIME;
2698 		else if (pmlmeext->cur_wireless_mode & (WLAN_MD_11G)) {
2699 			if ((updateCap & cShortSlotTime) /* && (!(pMgntInfo->pHTInfo->RT2RT_HT_Mode & RT_HT_CAP_USE_LONG_PREAMBLE)) */) {
2700 				/* Short Slot Time */
2701 				pmlmeinfo->slotTime = SHORT_SLOT_TIME;
2702 			} else {
2703 				/* Long Slot Time */
2704 				pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
2705 			}
2706 		} else {
2707 			/* B Mode */
2708 			pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
2709 		}
2710 	}
2711 
2712 	rtw_hal_set_hwreg(adapter, HW_VAR_SLOT_TIME, &pmlmeinfo->slotTime);
2713 
2714 }
2715 
2716 /*
2717 * set adapter.mlmeextpriv.mlmext_info.HT_enable
2718 * set adapter.mlmeextpriv.cur_wireless_mode
2719 * set SIFS register
2720 * set mgmt tx rate
2721 */
update_wireless_mode(_adapter * padapter)2722 void update_wireless_mode(_adapter *padapter)
2723 {
2724 	int ratelen, network_type = 0;
2725 	u32 SIFS_Timer;
2726 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
2727 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
2728 	WLAN_BSSID_EX		*cur_network = &(pmlmeinfo->network);
2729 	unsigned char			*rate = cur_network->SupportedRates;
2730 #ifdef CONFIG_P2P
2731 	struct wifidirect_info	*pwdinfo = &(padapter->wdinfo);
2732 #endif /* CONFIG_P2P */
2733 
2734 	ratelen = rtw_get_rateset_len(cur_network->SupportedRates);
2735 
2736 	if ((pmlmeinfo->HT_info_enable) && (pmlmeinfo->HT_caps_enable))
2737 		pmlmeinfo->HT_enable = 1;
2738 
2739 	if (pmlmeext->chandef.chan > 14) {
2740 		if (pmlmeinfo->HE_enable)
2741 			network_type = WLAN_MD_11AX;
2742 		else if (pmlmeinfo->VHT_enable)
2743 			network_type = WLAN_MD_11AC;
2744 		else if (pmlmeinfo->HT_enable)
2745 			network_type = WLAN_MD_11N;
2746 
2747 		network_type |= WLAN_MD_11A;
2748 	} else {
2749 		if (pmlmeinfo->HE_enable)
2750 			network_type = WLAN_MD_11AX;
2751 		else if (pmlmeinfo->VHT_enable)
2752 			network_type = WLAN_MD_11AC;
2753 		else if (pmlmeinfo->HT_enable)
2754 			network_type = WLAN_MD_11N;
2755 
2756 		if ((cckratesonly_included(rate, ratelen)) == _TRUE)
2757 			network_type |= WLAN_MD_11B;
2758 		else if ((cckrates_included(rate, ratelen)) == _TRUE)
2759 			network_type |= WLAN_MD_11BG;
2760 		else
2761 			network_type |= WLAN_MD_11G;
2762 	}
2763 
2764 	pmlmeext->cur_wireless_mode = network_type & padapter->registrypriv.wireless_mode;
2765 	/* RTW_INFO("network_type=%02x, padapter->registrypriv.wireless_mode=%02x\n", network_type, padapter->registrypriv.wireless_mode); */
2766 
2767 	if ((pmlmeext->cur_wireless_mode & WLAN_MD_11B) &&
2768 	    #ifdef CONFIG_P2P
2769 	    !rtw_p2p_chk_role(pwdinfo, P2P_ROLE_CLIENT) &&
2770 	    !rtw_p2p_chk_role(pwdinfo, P2P_ROLE_GO) &&
2771 	    #endif
2772 	    1)
2773 		update_mgnt_tx_rate(padapter, IEEE80211_CCK_RATE_1MB);
2774 	else
2775 		update_mgnt_tx_rate(padapter, IEEE80211_OFDM_RATE_6MB);
2776 }
2777 
update_sta_basic_rate(struct sta_info * psta,u8 wireless_mode)2778 void update_sta_basic_rate(struct sta_info *psta, u8 wireless_mode)
2779 {
2780 	if (is_supported_tx_cck(wireless_mode)) {
2781 		/* Only B, B/G, and B/G/N AP could use CCK rate */
2782 		_rtw_memcpy(psta->bssrateset, rtw_basic_rate_cck, 4);
2783 		psta->bssratelen = 4;
2784 	} else {
2785 		_rtw_memcpy(psta->bssrateset, rtw_basic_rate_ofdm, 3);
2786 		psta->bssratelen = 3;
2787 	}
2788 }
2789 
rtw_ies_get_supported_rate(u8 * ies,uint ies_len,u8 * rate_set,u8 * rate_num)2790 int rtw_ies_get_supported_rate(u8 *ies, uint ies_len, u8 *rate_set, u8 *rate_num)
2791 {
2792 	u8 *ie, *p;
2793 	unsigned int ie_len;
2794 	int i, j;
2795 
2796 	struct support_rate_handler support_rate_tbl[] = {
2797 		{IEEE80211_CCK_RATE_1MB, 		_FALSE,		_FALSE},
2798 		{IEEE80211_CCK_RATE_2MB, 		_FALSE,		_FALSE},
2799 		{IEEE80211_CCK_RATE_5MB, 		_FALSE,		_FALSE},
2800 		{IEEE80211_CCK_RATE_11MB,		_FALSE,		_FALSE},
2801 		{IEEE80211_OFDM_RATE_6MB,		_FALSE,		_FALSE},
2802 		{IEEE80211_OFDM_RATE_9MB,		_FALSE,		_FALSE},
2803 		{IEEE80211_OFDM_RATE_12MB,		_FALSE,		_FALSE},
2804 		{IEEE80211_OFDM_RATE_18MB,		_FALSE,		_FALSE},
2805 		{IEEE80211_OFDM_RATE_24MB,		_FALSE,		_FALSE},
2806 		{IEEE80211_OFDM_RATE_36MB,		_FALSE,		_FALSE},
2807 		{IEEE80211_OFDM_RATE_48MB,		_FALSE,		_FALSE},
2808 		{IEEE80211_OFDM_RATE_54MB,		_FALSE,		_FALSE},
2809 	};
2810 
2811 	if (!rate_set || !rate_num)
2812 		return _FALSE;
2813 
2814 	*rate_num = 0;
2815 	ie = rtw_get_ie(ies, _SUPPORTEDRATES_IE_, &ie_len, ies_len);
2816 	if (ie == NULL)
2817 		goto ext_rate;
2818 
2819 	/* get valid supported rates */
2820 	for (i = 0; i < 12; i++) {
2821 		p = ie + 2;
2822 		for (j = 0; j < ie_len; j++) {
2823 			if ((*p & ~BIT(7)) == support_rate_tbl[i].rate){
2824 				support_rate_tbl[i].existence = _TRUE;
2825 				if ((*p) & BIT(7))
2826 					support_rate_tbl[i].basic = _TRUE;
2827 			}
2828 			p++;
2829 		}
2830 	}
2831 
2832 ext_rate:
2833 	ie = rtw_get_ie(ies, _EXT_SUPPORTEDRATES_IE_, &ie_len, ies_len);
2834 	if (ie) {
2835 		/* get valid extended supported rates */
2836 		for (i = 0; i < 12; i++) {
2837 			p = ie + 2;
2838 			for (j = 0; j < ie_len; j++) {
2839 				if ((*p & ~BIT(7)) == support_rate_tbl[i].rate){
2840 					support_rate_tbl[i].existence = _TRUE;
2841 					if ((*p) & BIT(7))
2842 						support_rate_tbl[i].basic = _TRUE;
2843 				}
2844 				p++;
2845 			}
2846 		}
2847 	}
2848 
2849 	for (i = 0; i < 12; i++){
2850 		if (support_rate_tbl[i].existence){
2851 			if (support_rate_tbl[i].basic)
2852 				rate_set[*rate_num] = support_rate_tbl[i].rate | IEEE80211_BASIC_RATE_MASK;
2853 			else
2854 				rate_set[*rate_num] = support_rate_tbl[i].rate;
2855 			*rate_num += 1;
2856 		}
2857 	}
2858 
2859 	if (*rate_num == 0)
2860 		return _FAIL;
2861 
2862 	if (0) {
2863 		int i;
2864 
2865 		for (i = 0; i < *rate_num; i++)
2866 			RTW_INFO("rate:0x%02x\n", *(rate_set + i));
2867 	}
2868 
2869 	return _SUCCESS;
2870 }
2871 
process_addba_req(_adapter * padapter,u8 * paddba_req,u8 * addr)2872 void process_addba_req(_adapter *padapter, u8 *paddba_req, u8 *addr)
2873 {
2874 	struct sta_info *psta;
2875 	u16 tid, start_seq, param;
2876 	struct sta_priv *pstapriv = &padapter->stapriv;
2877 	struct ADDBA_request	*preq = (struct ADDBA_request *)paddba_req;
2878 	u8 size, accept = _FALSE;
2879 
2880 	psta = rtw_get_stainfo(pstapriv, addr);
2881 	if (!psta)
2882 		goto exit;
2883 
2884 	start_seq = le16_to_cpu(preq->BA_starting_seqctrl) >> 4;
2885 	param = le16_to_cpu(preq->BA_para_set);
2886 	tid = (param >> 2) & 0x0f;
2887 
2888 	accept = rtw_rx_ampdu_is_accept(padapter);
2889 	if (padapter->fix_rx_ampdu_size != RX_AMPDU_SIZE_INVALID)
2890 		size = padapter->fix_rx_ampdu_size;
2891 	else {
2892 		size = rtw_rx_ampdu_size(padapter);
2893 		size = rtw_min(size, rx_ampdu_size_sta_limit(padapter, psta));
2894 	}
2895 
2896 	if (accept == _TRUE)
2897 		rtw_addbarsp_cmd(padapter, addr, tid, preq, 0, size, start_seq);
2898 	else
2899 		rtw_addbarsp_cmd(padapter, addr, tid, preq, 37, size, start_seq);/* reject ADDBA Req */
2900 
2901 exit:
2902 	return;
2903 }
2904 
rtw_process_bar_frame(_adapter * padapter,union recv_frame * precv_frame)2905 void rtw_process_bar_frame(_adapter *padapter, union recv_frame *precv_frame)
2906 {
2907 	struct sta_priv *pstapriv = &padapter->stapriv;
2908 	u8 *pframe = precv_frame->u.hdr.rx_data;
2909 	struct sta_info *psta = NULL;
2910 	struct recv_reorder_ctrl *preorder_ctrl = NULL;
2911 	u8 tid = 0;
2912 	u16 start_seq=0;
2913 
2914 	psta = rtw_get_stainfo(pstapriv, get_addr2_ptr(pframe));
2915 	if (psta == NULL)
2916 		goto exit;
2917 
2918 	tid = ((cpu_to_le16((*(u16 *)(pframe + 16))) & 0xf000) >> 12);
2919 	preorder_ctrl = &psta->recvreorder_ctrl[tid];
2920 	start_seq = ((cpu_to_le16(*(u16 *)(pframe + 18))) >> 4);
2921 	preorder_ctrl->indicate_seq = start_seq;
2922 
2923 	rtw_phl_rx_bar(padapter->dvobj->phl, psta->phl_sta, tid, start_seq);
2924 	/* for Debug use */
2925 	if (0)
2926 		RTW_INFO(FUNC_ADPT_FMT" tid=%d, start_seq=%d\n", FUNC_ADPT_ARG(padapter),  tid, start_seq);
2927 
2928 exit:
2929 	return;
2930 }
2931 
update_TSF(struct mlme_ext_priv * pmlmeext,u8 * pframe,uint len)2932 void update_TSF(struct mlme_ext_priv *pmlmeext, u8 *pframe, uint len)
2933 {
2934 	u8 *pIE;
2935 	u32 *pbuf;
2936 
2937 	pIE = pframe + sizeof(struct rtw_ieee80211_hdr_3addr);
2938 	pbuf = (u32 *)pIE;
2939 
2940 	pmlmeext->TSFValue = le32_to_cpu(*(pbuf + 1));
2941 
2942 	pmlmeext->TSFValue = pmlmeext->TSFValue << 32;
2943 
2944 	pmlmeext->TSFValue |= le32_to_cpu(*pbuf);
2945 }
2946 
2947 #ifdef CONFIG_BCN_RECV_TIME
2948 /*	calculate beacon receiving time
2949 	1.RxBCNTime(CCK_1M) = [192us(preamble)] + [length of beacon(byte)*8us] + [10us]
2950 	2.RxBCNTime(OFDM_6M) = [8us(S) + 8us(L) + 4us(L-SIG)] + [(length of beacon(byte)/3 + 1] *4us] + [10us]
2951 */
_rx_bcn_time_calculate(uint bcn_len,u8 data_rate)2952 inline u16 _rx_bcn_time_calculate(uint bcn_len, u8 data_rate)
2953 {
2954 	u16 rx_bcn_time = 0;/*us*/
2955 
2956 	if (data_rate == DESC_RATE1M)
2957 		rx_bcn_time = 192 + bcn_len * 8 + 10;
2958 	else if(data_rate == DESC_RATE6M)
2959 		rx_bcn_time = 8 + 8 + 4 + (bcn_len /3 + 1) * 4 + 10;
2960 /*
2961 	else
2962 		RTW_ERR("%s invalid data rate(0x%02x)\n", __func__, data_rate);
2963 */
2964 	return rx_bcn_time;
2965 }
rtw_rx_bcn_time_update(_adapter * adapter,uint bcn_len,u8 data_rate)2966 void rtw_rx_bcn_time_update(_adapter *adapter, uint bcn_len, u8 data_rate)
2967 {
2968 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
2969 
2970 	pmlmeext->bcn_rx_time = _rx_bcn_time_calculate(bcn_len, data_rate);
2971 }
2972 #endif
2973 
beacon_timing_control(_adapter * padapter)2974 void beacon_timing_control(_adapter *padapter)
2975 {
2976 	rtw_hal_bcn_param_setting(padapter);
2977 }
2978 
rtw_collect_bcn_info(_adapter * adapter)2979 inline void rtw_collect_bcn_info(_adapter *adapter)
2980 {
2981 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
2982 
2983 	if (!is_client_associated_to_ap(adapter))
2984 		return;
2985 
2986 	pmlmeext->cur_bcn_cnt = pmlmeext->bcn_cnt - pmlmeext->last_bcn_cnt;
2987 	pmlmeext->last_bcn_cnt = pmlmeext->bcn_cnt;
2988 	/*TODO get offset of bcn's timestamp*/
2989 	/*pmlmeext->bcn_timestamp;*/
2990 }
2991 
rtw_bmp_is_set(const u8 * bmp,u8 bmp_len,u8 id)2992 inline bool rtw_bmp_is_set(const u8 *bmp, u8 bmp_len, u8 id)
2993 {
2994 	if (id / 8 >= bmp_len)
2995 		return 0;
2996 
2997 	return bmp[id / 8] & BIT(id % 8);
2998 }
2999 
rtw_bmp_set(u8 * bmp,u8 bmp_len,u8 id)3000 inline void rtw_bmp_set(u8 *bmp, u8 bmp_len, u8 id)
3001 {
3002 	if (id / 8 < bmp_len)
3003 		bmp[id / 8] |= BIT(id % 8);
3004 }
3005 
rtw_bmp_clear(u8 * bmp,u8 bmp_len,u8 id)3006 inline void rtw_bmp_clear(u8 *bmp, u8 bmp_len, u8 id)
3007 {
3008 	if (id / 8 < bmp_len)
3009 		bmp[id / 8] &= ~BIT(id % 8);
3010 }
3011 
rtw_bmp_not_empty(const u8 * bmp,u8 bmp_len)3012 inline bool rtw_bmp_not_empty(const u8 *bmp, u8 bmp_len)
3013 {
3014 	int i;
3015 
3016 	for (i = 0; i < bmp_len; i++) {
3017 		if (bmp[i])
3018 			return 1;
3019 	}
3020 
3021 	return 0;
3022 }
3023 
rtw_bmp_not_empty_exclude_bit0(const u8 * bmp,u8 bmp_len)3024 inline bool rtw_bmp_not_empty_exclude_bit0(const u8 *bmp, u8 bmp_len)
3025 {
3026 	int i;
3027 
3028 	for (i = 0; i < bmp_len; i++) {
3029 		if (i == 0) {
3030 			if (bmp[i] & 0xFE)
3031 				return 1;
3032 		} else {
3033 			if (bmp[i])
3034 				return 1;
3035 		}
3036 	}
3037 
3038 	return 0;
3039 }
3040 
3041 #ifdef CONFIG_AP_MODE
3042 /* Check the id be set or not in map , if yes , return a none zero value*/
rtw_tim_map_is_set(_adapter * padapter,const u8 * map,u8 id)3043 bool rtw_tim_map_is_set(_adapter *padapter, const u8 *map, u8 id)
3044 {
3045 	return rtw_bmp_is_set(map, padapter->stapriv.aid_bmp_len, id);
3046 }
3047 
3048 /* Set the id into map array*/
rtw_tim_map_set(_adapter * padapter,u8 * map,u8 id)3049 void rtw_tim_map_set(_adapter *padapter, u8 *map, u8 id)
3050 {
3051 	rtw_bmp_set(map, padapter->stapriv.aid_bmp_len, id);
3052 }
3053 
3054 /* Clear the id from map array*/
rtw_tim_map_clear(_adapter * padapter,u8 * map,u8 id)3055 void rtw_tim_map_clear(_adapter *padapter, u8 *map, u8 id)
3056 {
3057 	rtw_bmp_clear(map, padapter->stapriv.aid_bmp_len, id);
3058 }
3059 
3060 /* Check have anyone bit be set , if yes return true*/
rtw_tim_map_anyone_be_set(_adapter * padapter,const u8 * map)3061 bool rtw_tim_map_anyone_be_set(_adapter *padapter, const u8 *map)
3062 {
3063 	return rtw_bmp_not_empty(map, padapter->stapriv.aid_bmp_len);
3064 }
3065 
3066 /* Check have anyone bit be set exclude bit0 , if yes return true*/
rtw_tim_map_anyone_be_set_exclude_aid0(_adapter * padapter,const u8 * map)3067 bool rtw_tim_map_anyone_be_set_exclude_aid0(_adapter *padapter, const u8 *map)
3068 {
3069 	return rtw_bmp_not_empty_exclude_bit0(map, padapter->stapriv.aid_bmp_len);
3070 }
3071 #endif /* CONFIG_AP_MODE */
3072 
dvobj_get_unregisterd_adapter(struct dvobj_priv * dvobj)3073 _adapter *dvobj_get_unregisterd_adapter(struct dvobj_priv *dvobj)
3074 {
3075 	_adapter *adapter = NULL;
3076 	int i;
3077 
3078 	for (i = 0; i < dvobj->iface_nums; i++) {
3079 		if (dvobj->padapters[i]->registered == 0)
3080 			break;
3081 	}
3082 
3083 	if (i < dvobj->iface_nums)
3084 		adapter = dvobj->padapters[i];
3085 
3086 	return adapter;
3087 }
3088 
dvobj_get_adapter_by_addr(struct dvobj_priv * dvobj,u8 * addr)3089 _adapter *dvobj_get_adapter_by_addr(struct dvobj_priv *dvobj, u8 *addr)
3090 {
3091 	_adapter *adapter = NULL;
3092 	int i;
3093 
3094 	for (i = 0; i < dvobj->iface_nums; i++) {
3095 		if (_rtw_memcmp(dvobj->padapters[i]->mac_addr, addr, ETH_ALEN) == _TRUE)
3096 			break;
3097 	}
3098 
3099 	if (i < dvobj->iface_nums)
3100 		adapter = dvobj->padapters[i];
3101 
3102 	return adapter;
3103 }
3104 
3105 #ifdef CONFIG_RTW_MULTI_AP
rtw_get_ch_utilization(_adapter * adapter)3106 u8 rtw_get_ch_utilization(_adapter *adapter)
3107 {
3108 #if 0 /* FIXME */
3109 	u16 clm = rtw_phydm_clm_ratio(adapter);
3110 	u16 nhm = rtw_phydm_nhm_ratio(adapter);
3111 #else
3112 	u16 clm = 55;
3113 	u16 nhm = 55;
3114 #endif
3115 	u16 ch_util;
3116 
3117 	ch_util = clm / 3 + (2 * (nhm / 3));
3118 	/* For Multi-AP, scaling 0-100 to 0-255 */
3119 	ch_util = 255 * ch_util / 100;
3120 
3121 	return (u8)ch_util;
3122 }
3123 
rtw_ch_util_rpt(_adapter * adapter)3124 void rtw_ch_util_rpt(_adapter *adapter)
3125 {
3126 	struct dvobj_priv *dvobj = adapter_to_dvobj(adapter);
3127 	_adapter *iface;
3128 	int i, j;
3129 	u8 i_rpts = 0;
3130 	u8 *ch_util;
3131 	u8 **bssid;
3132 	u8 threshold = GET_PRIMARY_ADAPTER(adapter)->ch_util_threshold;
3133 	u8 need_rpt = 0;
3134 
3135 	if (threshold == 0)
3136 		return;
3137 
3138 	ch_util = rtw_zmalloc(sizeof(u8) * dvobj->iface_nums);
3139 	if (!ch_util)
3140 		goto err_out;
3141 
3142 	bssid = (u8 **)rtw_zmalloc(sizeof(u8 *) * dvobj->iface_nums);
3143 	if (!bssid)
3144 		goto err_out1;
3145 
3146 	for (j = 0; j < dvobj->iface_nums; j++) {
3147 		*(bssid + j) = (u8 *)rtw_zmalloc(sizeof(u8) * ETH_ALEN);
3148 		if (!(*(bssid + j)))
3149 			goto err_out2;
3150 	}
3151 
3152 	for (i = 0; i < dvobj->iface_nums; i++) {
3153 		iface = dvobj->padapters[i];
3154 		if ((iface) && MLME_IS_AP(iface)) {
3155 			*(ch_util + i_rpts) = rtw_get_ch_utilization(iface);
3156 			_rtw_memcpy(*(bssid + i_rpts), iface->mac_addr, ETH_ALEN);
3157 			if (*(ch_util + i_rpts) > threshold)
3158 				need_rpt = 1;
3159 
3160 			i_rpts++;
3161 		}
3162 	}
3163 
3164 	if (need_rpt)
3165 		rtw_nlrtw_ch_util_rpt(adapter, i_rpts, ch_util, bssid);
3166 
3167 	rtw_mfree(ch_util, sizeof(u8) * dvobj->iface_nums);
3168 	for (i = 0; i < dvobj->iface_nums; i++)
3169 		rtw_mfree(*(bssid + i), ETH_ALEN);
3170 
3171 	rtw_mfree(bssid, sizeof(u8 *) * dvobj->iface_nums);
3172 
3173 	return;
3174 
3175 err_out2:
3176 	for (i = 0; i < j; i++)
3177 		rtw_mfree(*(bssid + i), sizeof(u8) * ETH_ALEN);
3178 	rtw_mfree(bssid, sizeof(sizeof(u8 *) * dvobj->iface_nums));
3179 err_out1:
3180 	rtw_mfree(ch_util, sizeof(u8) * dvobj->iface_nums);
3181 err_out:
3182 	RTW_INFO("[%s] rtw_zmalloc fail\n", __func__);
3183 }
3184 #endif
3185