• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /******************************************************************************
2  *
3  * Copyright(c) 2007 - 2019 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_MLME_C_
16 
17 #include <hal_data.h>
18 #ifdef CONFIG_PLATFORM_CMAP_INTFS
19 #include "../os_dep/linux/custom_multiap_intfs/custom_multiap_intfs.h"
20 #endif
21 
22 extern void indicate_wx_scan_complete_event(_adapter *padapter);
23 extern u8 rtw_do_join(_adapter *padapter);
24 
25 
rtw_init_mlme_timer(_adapter * padapter)26 void rtw_init_mlme_timer(_adapter *padapter)
27 {
28 	struct	mlme_priv *pmlmepriv = &padapter->mlmepriv;
29 
30 	rtw_init_timer(&(pmlmepriv->assoc_timer), padapter, rtw_join_timeout_handler, padapter);
31 	rtw_init_timer(&(pmlmepriv->scan_to_timer), padapter, rtw_scan_timeout_handler, padapter);
32 
33 #ifdef CONFIG_SET_SCAN_DENY_TIMER
34 	rtw_init_timer(&(pmlmepriv->set_scan_deny_timer), padapter, rtw_set_scan_deny_timer_hdl, padapter);
35 #endif
36 
37 #ifdef RTK_DMP_PLATFORM
38 	_init_workitem(&(pmlmepriv->Linkup_workitem), Linkup_workitem_callback, padapter);
39 	_init_workitem(&(pmlmepriv->Linkdown_workitem), Linkdown_workitem_callback, padapter);
40 #endif
41 }
42 
_rtw_init_mlme_priv(_adapter * padapter)43 sint	_rtw_init_mlme_priv(_adapter *padapter)
44 {
45 	sint	i;
46 	u8	*pbuf;
47 	struct wlan_network	*pnetwork;
48 	struct mlme_priv		*pmlmepriv = &padapter->mlmepriv;
49 	struct rf_ctl_t *rfctl = adapter_to_rfctl(padapter);
50 	sint	res = _SUCCESS;
51 #ifdef CONFIG_RTW_MULTI_AP
52 	struct unassoc_sta_info *unassoc_sta;
53 #endif
54 
55 
56 	/* We don't need to memset padapter->XXX to zero, because adapter is allocated by rtw_zvmalloc(). */
57 	/* _rtw_memset((u8 *)pmlmepriv, 0, sizeof(struct mlme_priv)); */
58 
59 
60 	/*qos_priv*/
61 	/*pmlmepriv->qospriv.qos_option = pregistrypriv->wmm_enable;*/
62 
63 	/*ht_priv*/
64 #ifdef CONFIG_80211N_HT
65 	pmlmepriv->htpriv.ampdu_enable = _FALSE;/*set to disabled*/
66 #endif
67 
68 	pmlmepriv->nic_hdl = (u8 *)padapter;
69 
70 	pmlmepriv->pscanned = NULL;
71 	init_fwstate(pmlmepriv, WIFI_STATION_STATE);
72 	pmlmepriv->cur_network.network.InfrastructureMode = Ndis802_11AutoUnknown;
73 	pmlmepriv->scan_mode = SCAN_ACTIVE; /* 1: active, 0: pasive. Maybe someday we should rename this varable to "active_mode" (Jeff) */
74 
75 	_rtw_spinlock_init(&(pmlmepriv->lock));
76 	_rtw_init_queue(&(pmlmepriv->free_bss_pool));
77 	_rtw_init_queue(&(pmlmepriv->scanned_queue));
78 
79 	set_scanned_network_val(pmlmepriv, 0);
80 
81 	_rtw_memset(&pmlmepriv->assoc_ssid, 0, sizeof(NDIS_802_11_SSID));
82 
83 	if (padapter->registrypriv.max_bss_cnt != 0)
84 		pmlmepriv->max_bss_cnt = padapter->registrypriv.max_bss_cnt;
85 	else if (rfctl->max_chan_nums <= MAX_CHANNEL_NUM_2G)
86 		pmlmepriv->max_bss_cnt = MAX_BSS_CNT;
87 	else
88 		pmlmepriv->max_bss_cnt = MAX_BSS_CNT + MAX_BSS_CNT;
89 
90 
91 	pbuf = rtw_zvmalloc(pmlmepriv->max_bss_cnt * (sizeof(struct wlan_network)));
92 
93 	if (pbuf == NULL) {
94 		res = _FAIL;
95 		goto exit;
96 	}
97 	pmlmepriv->free_bss_buf = pbuf;
98 
99 	pnetwork = (struct wlan_network *)pbuf;
100 
101 	for (i = 0; i < pmlmepriv->max_bss_cnt; i++) {
102 		_rtw_init_listhead(&(pnetwork->list));
103 
104 		rtw_list_insert_tail(&(pnetwork->list), &(pmlmepriv->free_bss_pool.queue));
105 
106 		pnetwork++;
107 	}
108 
109 #ifdef CONFIG_RTW_MULTI_AP
110 	if (is_primary_adapter(padapter)) {
111 		_rtw_init_queue(&(pmlmepriv->free_unassoc_sta_queue));
112 		_rtw_init_queue(&(pmlmepriv->unassoc_sta_queue));
113 		for (i = 0; i < UNASOC_STA_SRC_NUM; i++)
114 			pmlmepriv->unassoc_sta_mode_of_stype[i] = padapter->registrypriv.unassoc_sta_mode_of_stype[i];
115 		if (padapter->registrypriv.max_unassoc_sta_cnt != 0)
116 			pmlmepriv->max_unassoc_sta_cnt = padapter->registrypriv.max_unassoc_sta_cnt;
117 		else if (rfctl->max_chan_nums <= MAX_CHANNEL_NUM_2G)
118 			pmlmepriv->max_unassoc_sta_cnt = MAX_UNASSOC_STA_CNT;
119 		else
120 			pmlmepriv->max_unassoc_sta_cnt = MAX_UNASSOC_STA_CNT * 2;
121 		pbuf = rtw_zvmalloc(pmlmepriv->max_unassoc_sta_cnt * (sizeof(struct unassoc_sta_info)));
122 		if (pbuf == NULL) {
123 			res = _FAIL;
124 			goto exit;
125 		}
126 		pmlmepriv->free_unassoc_sta_buf = pbuf;
127 		unassoc_sta = (struct unassoc_sta_info *) pbuf;
128 		for (i = 0; i < pmlmepriv->max_unassoc_sta_cnt; i++) {
129 			_rtw_init_listhead(&(unassoc_sta->list));
130 			rtw_list_insert_tail(&(unassoc_sta->list), &(pmlmepriv->free_unassoc_sta_queue.queue));
131 			unassoc_sta++;
132 		}
133 	}
134 #ifdef CONFIG_PLATFORM_CMAP_INTFS
135 	rtw_init_timer(&pmlmepriv->cmap_unassoc_sta_timer, padapter, cmap_unassoc_sta_report_info_timer, padapter);
136 #endif
137 #endif
138 	/* allocate DMA-able/Non-Page memory for cmd_buf and rsp_buf */
139 
140 	rtw_clear_scan_deny(padapter);
141 #ifdef CONFIG_ARP_KEEP_ALIVE
142 	pmlmepriv->bGetGateway = 0;
143 	pmlmepriv->GetGatewayTryCnt = 0;
144 #endif
145 
146 #ifdef CONFIG_LAYER2_ROAMING
147 #define RTW_ROAM_SCAN_RESULT_EXP_MS (5*1000)
148 #define RTW_ROAM_RSSI_DIFF_TH 10
149 #define RTW_ROAM_SCAN_INTERVAL (5)    /* 5*(2 second)*/
150 #define RTW_ROAM_RSSI_THRESHOLD 70
151 #define RTW_ROAM_DICONNECT_DELAY	20
152 	_rtw_spinlock_init(&pmlmepriv->clnt_auth_lock);
153 	pmlmepriv->roam_flags = CONFIG_ROAMING_FLAG;
154 
155 	pmlmepriv->roam_scanr_exp_ms = RTW_ROAM_SCAN_RESULT_EXP_MS;
156 	pmlmepriv->roam_rssi_diff_th = RTW_ROAM_RSSI_DIFF_TH;
157 	pmlmepriv->roam_scan_int 	 = RTW_ROAM_SCAN_INTERVAL;
158 	pmlmepriv->roam_rssi_threshold = RTW_ROAM_RSSI_THRESHOLD;
159 	pmlmepriv->need_to_roam = _FALSE;
160 	pmlmepriv->last_roaming = rtw_get_current_time();
161 #endif /* CONFIG_LAYER2_ROAMING */
162 
163 #ifdef CONFIG_RTW_80211R
164 	rtw_ft_info_init(&pmlmepriv->ft_roam);
165 #endif
166 #ifdef CONFIG_LAYER2_ROAMING
167 #if defined(CONFIG_RTW_WNM) || defined(CONFIG_RTW_80211K)
168 	rtw_roam_nb_info_init(padapter);
169 	pmlmepriv->ch_cnt = 0;
170 #endif
171 #endif
172 
173 	pmlmepriv->defs_lmt_sta = 2;
174 	pmlmepriv->defs_lmt_time = 5;
175 
176 	rtw_init_mlme_timer(padapter);
177 
178 exit:
179 
180 
181 	return res;
182 }
183 
184 void rtw_mfree_mlme_priv_lock(struct mlme_priv *pmlmepriv);
rtw_mfree_mlme_priv_lock(struct mlme_priv * pmlmepriv)185 void rtw_mfree_mlme_priv_lock(struct mlme_priv *pmlmepriv)
186 {
187 	_rtw_spinlock_free(&pmlmepriv->lock);
188 #ifdef CONFIG_LAYER2_ROAMING
189 	_rtw_spinlock_free(&pmlmepriv->clnt_auth_lock);
190 #endif
191 	_rtw_spinlock_free(&(pmlmepriv->free_bss_pool.lock));
192 	_rtw_spinlock_free(&(pmlmepriv->scanned_queue.lock));
193 #ifdef CONFIG_RTW_MULTI_AP
194 	if (is_primary_adapter(mlme_to_adapter(pmlmepriv))) {
195 		_rtw_spinlock_free(&(pmlmepriv->unassoc_sta_queue.lock));
196 		_rtw_spinlock_free(&(pmlmepriv->free_unassoc_sta_queue.lock));
197 	}
198 #endif
199 }
200 
rtw_free_mlme_priv_ie_data(struct mlme_priv * pmlmepriv)201 void rtw_free_mlme_priv_ie_data(struct mlme_priv *pmlmepriv)
202 {
203 	rtw_buf_free(&pmlmepriv->assoc_req, &pmlmepriv->assoc_req_len);
204 	rtw_buf_free(&pmlmepriv->assoc_rsp, &pmlmepriv->assoc_rsp_len);
205 #ifdef CONFIG_80211D
206 	rtw_buf_free(&pmlmepriv->recv_country_ie, &pmlmepriv->recv_country_ie_len);
207 #endif
208 #if defined(CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME)
209 	rtw_buf_free(&pmlmepriv->wps_beacon_ie, &pmlmepriv->wps_beacon_ie_len);
210 	rtw_buf_free(&pmlmepriv->wps_probe_req_ie, &pmlmepriv->wps_probe_req_ie_len);
211 	rtw_buf_free(&pmlmepriv->wps_probe_resp_ie, &pmlmepriv->wps_probe_resp_ie_len);
212 	rtw_buf_free(&pmlmepriv->wps_assoc_resp_ie, &pmlmepriv->wps_assoc_resp_ie_len);
213 
214 	rtw_buf_free(&pmlmepriv->p2p_beacon_ie, &pmlmepriv->p2p_beacon_ie_len);
215 	rtw_buf_free(&pmlmepriv->p2p_probe_req_ie, &pmlmepriv->p2p_probe_req_ie_len);
216 	rtw_buf_free(&pmlmepriv->p2p_probe_resp_ie, &pmlmepriv->p2p_probe_resp_ie_len);
217 	rtw_buf_free(&pmlmepriv->p2p_go_probe_resp_ie, &pmlmepriv->p2p_go_probe_resp_ie_len);
218 	rtw_buf_free(&pmlmepriv->p2p_assoc_req_ie, &pmlmepriv->p2p_assoc_req_ie_len);
219 	rtw_buf_free(&pmlmepriv->p2p_assoc_resp_ie, &pmlmepriv->p2p_assoc_resp_ie_len);
220 #endif
221 
222 #if defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211)
223 	rtw_buf_free(&pmlmepriv->wfd_beacon_ie, &pmlmepriv->wfd_beacon_ie_len);
224 	rtw_buf_free(&pmlmepriv->wfd_probe_req_ie, &pmlmepriv->wfd_probe_req_ie_len);
225 	rtw_buf_free(&pmlmepriv->wfd_probe_resp_ie, &pmlmepriv->wfd_probe_resp_ie_len);
226 	rtw_buf_free(&pmlmepriv->wfd_go_probe_resp_ie, &pmlmepriv->wfd_go_probe_resp_ie_len);
227 	rtw_buf_free(&pmlmepriv->wfd_assoc_req_ie, &pmlmepriv->wfd_assoc_req_ie_len);
228 	rtw_buf_free(&pmlmepriv->wfd_assoc_resp_ie, &pmlmepriv->wfd_assoc_resp_ie_len);
229 #endif
230 
231 #ifdef CONFIG_RTW_80211R
232 	rtw_buf_free(&pmlmepriv->auth_rsp, &pmlmepriv->auth_rsp_len);
233 #endif
234 #ifdef CONFIG_RTW_MBO
235 	rtw_buf_free(&pmlmepriv->pcell_data_cap_ie, &pmlmepriv->cell_data_cap_len);
236 #endif
237 }
238 
239 #if defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211)
rtw_mlme_update_wfd_ie_data(struct mlme_priv * mlme,u8 type,u8 * ie,u32 ie_len)240 int rtw_mlme_update_wfd_ie_data(struct mlme_priv *mlme, u8 type, u8 *ie, u32 ie_len)
241 {
242 	_adapter *adapter = mlme_to_adapter(mlme);
243 	struct wifi_display_info *wfd_info = &adapter->wfd_info;
244 	u8 clear = 0;
245 	u8 **t_ie = NULL;
246 	u32 *t_ie_len = NULL;
247 	int ret = _FAIL;
248 
249 	if (!hal_chk_wl_func(adapter, WL_FUNC_MIRACAST))
250 		goto success;
251 
252 	if (wfd_info->wfd_enable == _TRUE)
253 		goto success; /* WFD IE is build by self */
254 
255 	if (!ie && !ie_len)
256 		clear = 1;
257 	else if (!ie || !ie_len) {
258 		RTW_PRINT(FUNC_ADPT_FMT" type:%u, ie:%p, ie_len:%u"
259 			  , FUNC_ADPT_ARG(adapter), type, ie, ie_len);
260 		rtw_warn_on(1);
261 		goto exit;
262 	}
263 
264 	switch (type) {
265 	case MLME_BEACON_IE:
266 		t_ie = &mlme->wfd_beacon_ie;
267 		t_ie_len = &mlme->wfd_beacon_ie_len;
268 		break;
269 	case MLME_PROBE_REQ_IE:
270 		t_ie = &mlme->wfd_probe_req_ie;
271 		t_ie_len = &mlme->wfd_probe_req_ie_len;
272 		break;
273 	case MLME_PROBE_RESP_IE:
274 		t_ie = &mlme->wfd_probe_resp_ie;
275 		t_ie_len = &mlme->wfd_probe_resp_ie_len;
276 		break;
277 	case MLME_GO_PROBE_RESP_IE:
278 		t_ie = &mlme->wfd_go_probe_resp_ie;
279 		t_ie_len = &mlme->wfd_go_probe_resp_ie_len;
280 		break;
281 	case MLME_ASSOC_REQ_IE:
282 		t_ie = &mlme->wfd_assoc_req_ie;
283 		t_ie_len = &mlme->wfd_assoc_req_ie_len;
284 		break;
285 	case MLME_ASSOC_RESP_IE:
286 		t_ie = &mlme->wfd_assoc_resp_ie;
287 		t_ie_len = &mlme->wfd_assoc_resp_ie_len;
288 		break;
289 	default:
290 		RTW_PRINT(FUNC_ADPT_FMT" unsupported type:%u"
291 			  , FUNC_ADPT_ARG(adapter), type);
292 		rtw_warn_on(1);
293 		goto exit;
294 	}
295 
296 	if (*t_ie) {
297 		u32 free_len = *t_ie_len;
298 		*t_ie_len = 0;
299 		rtw_mfree(*t_ie, free_len);
300 		*t_ie = NULL;
301 	}
302 
303 	if (!clear) {
304 		*t_ie = rtw_malloc(ie_len);
305 		if (*t_ie == NULL) {
306 			RTW_ERR(FUNC_ADPT_FMT" type:%u, rtw_malloc() fail\n"
307 				, FUNC_ADPT_ARG(adapter), type);
308 			goto exit;
309 		}
310 		_rtw_memcpy(*t_ie, ie, ie_len);
311 		*t_ie_len = ie_len;
312 	}
313 
314 	if (*t_ie && *t_ie_len) {
315 		u8 *attr_content;
316 		u32 attr_contentlen = 0;
317 
318 		attr_content = rtw_get_wfd_attr_content(*t_ie, *t_ie_len, WFD_ATTR_DEVICE_INFO, NULL, &attr_contentlen);
319 		if (attr_content && attr_contentlen) {
320 			if (RTW_GET_BE16(attr_content + 2) != wfd_info->rtsp_ctrlport) {
321 				wfd_info->rtsp_ctrlport = RTW_GET_BE16(attr_content + 2);
322 				RTW_INFO(FUNC_ADPT_FMT" type:%u, RTSP CTRL port = %u\n"
323 					, FUNC_ADPT_ARG(adapter), type, wfd_info->rtsp_ctrlport);
324 			}
325 		}
326 	}
327 
328 success:
329 	ret = _SUCCESS;
330 
331 exit:
332 	return ret;
333 }
334 #endif /* defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211) */
335 
_rtw_free_mlme_priv(struct mlme_priv * pmlmepriv)336 void _rtw_free_mlme_priv(struct mlme_priv *pmlmepriv)
337 {
338 	_adapter *adapter = mlme_to_adapter(pmlmepriv);
339 	if (NULL == pmlmepriv) {
340 		rtw_warn_on(1);
341 		goto exit;
342 	}
343 	rtw_free_mlme_priv_ie_data(pmlmepriv);
344 
345 	if (pmlmepriv) {
346 		rtw_mfree_mlme_priv_lock(pmlmepriv);
347 
348 		if (pmlmepriv->free_bss_buf)
349 			rtw_vmfree(pmlmepriv->free_bss_buf, pmlmepriv->max_bss_cnt * sizeof(struct wlan_network));
350 #ifdef CONFIG_RTW_MULTI_AP
351 		if (is_primary_adapter(adapter)) {
352 			if (pmlmepriv->free_unassoc_sta_buf)
353 				rtw_vmfree(pmlmepriv->free_unassoc_sta_buf, pmlmepriv->max_unassoc_sta_cnt * sizeof(struct unassoc_sta_info));
354 		}
355 #ifdef CONFIG_PLATFORM_CMAP_INTFS
356 		_cancel_timer_ex(&pmlmepriv->cmap_unassoc_sta_timer);
357 #endif
358 #endif
359 	}
360 exit:
361 	return;
362 }
363 
_rtw_enqueue_network(_queue * queue,struct wlan_network * pnetwork)364 sint	_rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork)
365 {
366 	_irqL irqL;
367 
368 
369 	if (pnetwork == NULL)
370 		goto exit;
371 
372 	_enter_critical_bh(&queue->lock, &irqL);
373 
374 	rtw_list_insert_tail(&pnetwork->list, &queue->queue);
375 
376 	_exit_critical_bh(&queue->lock, &irqL);
377 
378 exit:
379 
380 
381 	return _SUCCESS;
382 }
383 
384 /*
385 struct	wlan_network *_rtw_dequeue_network(_queue *queue)
386 {
387 	_irqL irqL;
388 
389 	struct wlan_network *pnetwork;
390 
391 
392 	_enter_critical_bh(&queue->lock, &irqL);
393 
394 	if (_rtw_queue_empty(queue) == _TRUE)
395 
396 		pnetwork = NULL;
397 
398 	else
399 	{
400 		pnetwork = LIST_CONTAINOR(get_next(&queue->queue), struct wlan_network, list);
401 
402 		rtw_list_delete(&(pnetwork->list));
403 	}
404 
405 	_exit_critical_bh(&queue->lock, &irqL);
406 
407 
408 	return pnetwork;
409 }
410 */
411 
_rtw_alloc_network(struct mlme_priv * pmlmepriv)412 struct	wlan_network *_rtw_alloc_network(struct	mlme_priv *pmlmepriv) /* (_queue *free_queue) */
413 {
414 	_irqL	irqL;
415 	struct	wlan_network	*pnetwork;
416 	_queue *free_queue = &pmlmepriv->free_bss_pool;
417 	_list *plist = NULL;
418 
419 
420 	_enter_critical_bh(&free_queue->lock, &irqL);
421 
422 	if (_rtw_queue_empty(free_queue) == _TRUE) {
423 		pnetwork = NULL;
424 		goto exit;
425 	}
426 	plist = get_next(&(free_queue->queue));
427 
428 	pnetwork = LIST_CONTAINOR(plist , struct wlan_network, list);
429 
430 	rtw_list_delete(&pnetwork->list);
431 
432 	pnetwork->network_type = 0;
433 	pnetwork->fixed = _FALSE;
434 	pnetwork->last_scanned = rtw_get_current_time();
435 	pnetwork->last_non_hidden_ssid_ap = pnetwork->last_scanned;
436 #if defined(CONFIG_RTW_MESH) && CONFIG_RTW_MESH_ACNODE_PREVENT
437 	pnetwork->acnode_stime = 0;
438 	pnetwork->acnode_notify_etime = 0;
439 #endif
440 
441 	pnetwork->aid = 0;
442 	pnetwork->join_res = 0;
443 
444 	pmlmepriv->num_of_scanned++;
445 
446 exit:
447 	_exit_critical_bh(&free_queue->lock, &irqL);
448 
449 
450 	return pnetwork;
451 }
452 
_rtw_free_network(struct mlme_priv * pmlmepriv,struct wlan_network * pnetwork,u8 isfreeall)453 void _rtw_free_network(struct	mlme_priv *pmlmepriv , struct wlan_network *pnetwork, u8 isfreeall)
454 {
455 	u32 delta_time;
456 	u32 lifetime = SCANQUEUE_LIFETIME;
457 	_irqL irqL;
458 	_queue *free_queue = &(pmlmepriv->free_bss_pool);
459 
460 
461 	if (pnetwork == NULL)
462 		goto exit;
463 
464 	if (pnetwork->fixed == _TRUE)
465 		goto exit;
466 
467 	if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) == _TRUE) ||
468 	    (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE))
469 		lifetime = 1;
470 
471 	if (!isfreeall) {
472 		delta_time = (u32) rtw_get_passing_time_ms(pnetwork->last_scanned);
473 		if (delta_time < lifetime) /* unit:msec */
474 			goto exit;
475 	}
476 
477 	_enter_critical_bh(&free_queue->lock, &irqL);
478 
479 	rtw_list_delete(&(pnetwork->list));
480 
481 	rtw_list_insert_tail(&(pnetwork->list), &(free_queue->queue));
482 
483 	pmlmepriv->num_of_scanned--;
484 
485 
486 	/* RTW_INFO("_rtw_free_network:SSID=%s\n", pnetwork->network.Ssid.Ssid); */
487 
488 	_exit_critical_bh(&free_queue->lock, &irqL);
489 
490 exit:
491 	return;
492 }
493 
_rtw_free_network_nolock(struct mlme_priv * pmlmepriv,struct wlan_network * pnetwork)494 void _rtw_free_network_nolock(struct	mlme_priv *pmlmepriv, struct wlan_network *pnetwork)
495 {
496 
497 	_queue *free_queue = &(pmlmepriv->free_bss_pool);
498 
499 
500 	if (pnetwork == NULL)
501 		goto exit;
502 
503 	if (pnetwork->fixed == _TRUE)
504 		goto exit;
505 
506 	/* _enter_critical(&free_queue->lock, &irqL); */
507 
508 	rtw_list_delete(&(pnetwork->list));
509 
510 	rtw_list_insert_tail(&(pnetwork->list), get_list_head(free_queue));
511 
512 	pmlmepriv->num_of_scanned--;
513 
514 	/* _exit_critical(&free_queue->lock, &irqL); */
515 
516 exit:
517 	return;
518 }
519 
_rtw_free_network_queue(_adapter * padapter,u8 isfreeall)520 void _rtw_free_network_queue(_adapter *padapter, u8 isfreeall)
521 {
522 	_irqL irqL;
523 	_list *phead, *plist;
524 	struct wlan_network *pnetwork;
525 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
526 	_queue *scanned_queue = &pmlmepriv->scanned_queue;
527 
528 
529 
530 	_enter_critical_bh(&scanned_queue->lock, &irqL);
531 
532 	phead = get_list_head(scanned_queue);
533 	plist = get_next(phead);
534 
535 	while (rtw_end_of_queue_search(phead, plist) == _FALSE) {
536 
537 		pnetwork = LIST_CONTAINOR(plist, struct wlan_network, list);
538 
539 		plist = get_next(plist);
540 
541 		_rtw_free_network(pmlmepriv, pnetwork, isfreeall);
542 
543 	}
544 
545 	_exit_critical_bh(&scanned_queue->lock, &irqL);
546 
547 
548 }
549 
550 
551 
552 
rtw_if_up(_adapter * padapter)553 sint rtw_if_up(_adapter *padapter)
554 {
555 
556 	sint res;
557 
558 	if (RTW_CANNOT_RUN(padapter) ||
559 	    (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE) == _FALSE)) {
560 		res = _FALSE;
561 	} else
562 		res =  _TRUE;
563 
564 	return res;
565 }
566 
567 
rtw_generate_random_ibss(u8 * pibss)568 void rtw_generate_random_ibss(u8 *pibss)
569 {
570 	*((u32 *)(&pibss[2])) = rtw_random32();
571 	pibss[0] = 0x02; /* in ad-hoc mode local bit must set to 1 */
572 	pibss[1] = 0x11;
573 	pibss[2] = 0x87;
574 }
575 
rtw_get_capability_from_ie(u8 * ie)576 u8 *rtw_get_capability_from_ie(u8 *ie)
577 {
578 	return ie + 8 + 2;
579 }
580 
581 
rtw_get_capability(WLAN_BSSID_EX * bss)582 u16 rtw_get_capability(WLAN_BSSID_EX *bss)
583 {
584 	u16	val;
585 
586 	_rtw_memcpy((u8 *)&val, rtw_get_capability_from_ie(bss->IEs), 2);
587 
588 	return le16_to_cpu(val);
589 }
590 
rtw_get_timestampe_from_ie(u8 * ie)591 u8 *rtw_get_timestampe_from_ie(u8 *ie)
592 {
593 	return ie + 0;
594 }
595 
rtw_get_beacon_interval_from_ie(u8 * ie)596 u8 *rtw_get_beacon_interval_from_ie(u8 *ie)
597 {
598 	return ie + 8;
599 }
600 
601 
rtw_init_mlme_priv(_adapter * padapter)602 int	rtw_init_mlme_priv(_adapter *padapter) /* (struct	mlme_priv *pmlmepriv) */
603 {
604 	int	res;
605 	res = _rtw_init_mlme_priv(padapter);/* (pmlmepriv); */
606 	return res;
607 }
608 
rtw_free_mlme_priv(struct mlme_priv * pmlmepriv)609 void rtw_free_mlme_priv(struct mlme_priv *pmlmepriv)
610 {
611 	_rtw_free_mlme_priv(pmlmepriv);
612 }
613 
614 int	rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork);
rtw_enqueue_network(_queue * queue,struct wlan_network * pnetwork)615 int	rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork)
616 {
617 	int	res;
618 	res = _rtw_enqueue_network(queue, pnetwork);
619 	return res;
620 }
621 
622 /*
623 static struct	wlan_network *rtw_dequeue_network(_queue *queue)
624 {
625 	struct wlan_network *pnetwork;
626 	pnetwork = _rtw_dequeue_network(queue);
627 	return pnetwork;
628 }
629 */
630 
631 struct	wlan_network *rtw_alloc_network(struct	mlme_priv *pmlmepriv);
rtw_alloc_network(struct mlme_priv * pmlmepriv)632 struct	wlan_network *rtw_alloc_network(struct	mlme_priv *pmlmepriv) /* (_queue	*free_queue) */
633 {
634 	struct	wlan_network	*pnetwork;
635 	pnetwork = _rtw_alloc_network(pmlmepriv);
636 	return pnetwork;
637 }
638 
639 void rtw_free_network(struct mlme_priv *pmlmepriv, struct	wlan_network *pnetwork, u8 is_freeall);
rtw_free_network(struct mlme_priv * pmlmepriv,struct wlan_network * pnetwork,u8 is_freeall)640 void rtw_free_network(struct mlme_priv *pmlmepriv, struct	wlan_network *pnetwork, u8 is_freeall)/* (struct	wlan_network *pnetwork, _queue	*free_queue) */
641 {
642 	_rtw_free_network(pmlmepriv, pnetwork, is_freeall);
643 }
644 
645 void rtw_free_network_nolock(_adapter *padapter, struct wlan_network *pnetwork);
rtw_free_network_nolock(_adapter * padapter,struct wlan_network * pnetwork)646 void rtw_free_network_nolock(_adapter *padapter, struct wlan_network *pnetwork)
647 {
648 	_rtw_free_network_nolock(&(padapter->mlmepriv), pnetwork);
649 #ifdef CONFIG_IOCTL_CFG80211
650 	rtw_cfg80211_unlink_bss(padapter, pnetwork);
651 #endif /* CONFIG_IOCTL_CFG80211 */
652 }
653 
654 
rtw_free_network_queue(_adapter * dev,u8 isfreeall)655 void rtw_free_network_queue(_adapter *dev, u8 isfreeall)
656 {
657 	_rtw_free_network_queue(dev, isfreeall);
658 }
659 
_rtw_find_network(_queue * scanned_queue,const u8 * addr)660 struct wlan_network *_rtw_find_network(_queue *scanned_queue, const u8 *addr)
661 {
662 	_list	*phead, *plist;
663 	struct	wlan_network *pnetwork = NULL;
664 	u8 zero_addr[ETH_ALEN] = {0, 0, 0, 0, 0, 0};
665 
666 	if (_rtw_memcmp(zero_addr, addr, ETH_ALEN)) {
667 		pnetwork = NULL;
668 		goto exit;
669 	}
670 
671 	phead = get_list_head(scanned_queue);
672 	plist = get_next(phead);
673 
674 	while (plist != phead) {
675 		pnetwork = LIST_CONTAINOR(plist, struct wlan_network , list);
676 
677 		if (_rtw_memcmp(addr, pnetwork->network.MacAddress, ETH_ALEN) == _TRUE)
678 			break;
679 
680 		plist = get_next(plist);
681 	}
682 
683 	if (plist == phead)
684 		pnetwork = NULL;
685 
686 exit:
687 	return pnetwork;
688 }
689 
rtw_find_network(_queue * scanned_queue,const u8 * addr)690 struct wlan_network *rtw_find_network(_queue *scanned_queue, const u8 *addr)
691 {
692 	struct	wlan_network *pnetwork;
693 	_irqL irqL;
694 
695 	 _enter_critical_bh(&scanned_queue->lock, &irqL);
696 	pnetwork = _rtw_find_network(scanned_queue, addr);
697 	_exit_critical_bh(&scanned_queue->lock, &irqL);
698 
699 	return pnetwork;
700 }
701 
rtw_is_same_ibss(_adapter * adapter,struct wlan_network * pnetwork)702 int rtw_is_same_ibss(_adapter *adapter, struct wlan_network *pnetwork)
703 {
704 	int ret = _TRUE;
705 	struct security_priv *psecuritypriv = &adapter->securitypriv;
706 
707 	if ((psecuritypriv->dot11PrivacyAlgrthm != _NO_PRIVACY_) &&
708 	    (pnetwork->network.Privacy == 0))
709 		ret = _FALSE;
710 	else if ((psecuritypriv->dot11PrivacyAlgrthm == _NO_PRIVACY_) &&
711 		 (pnetwork->network.Privacy == 1))
712 		ret = _FALSE;
713 	else
714 		ret = _TRUE;
715 
716 	return ret;
717 
718 }
719 
is_same_ess(WLAN_BSSID_EX * a,WLAN_BSSID_EX * b)720 inline int is_same_ess(WLAN_BSSID_EX *a, WLAN_BSSID_EX *b)
721 {
722 	return (a->Ssid.SsidLength == b->Ssid.SsidLength)
723 	       &&  _rtw_memcmp(a->Ssid.Ssid, b->Ssid.Ssid, a->Ssid.SsidLength) == _TRUE;
724 }
725 
is_same_network(WLAN_BSSID_EX * src,WLAN_BSSID_EX * dst,u8 feature)726 int is_same_network(WLAN_BSSID_EX *src, WLAN_BSSID_EX *dst, u8 feature)
727 {
728 	u16 s_cap, d_cap;
729 
730 
731 	if (rtw_bug_check(dst, src, &s_cap, &d_cap) == _FALSE)
732 		return _FALSE;
733 
734 	_rtw_memcpy((u8 *)&s_cap, rtw_get_capability_from_ie(src->IEs), 2);
735 	_rtw_memcpy((u8 *)&d_cap, rtw_get_capability_from_ie(dst->IEs), 2);
736 
737 
738 	s_cap = le16_to_cpu(s_cap);
739 	d_cap = le16_to_cpu(d_cap);
740 
741 
742 #ifdef CONFIG_P2P
743 	if ((feature == 1) && /* 1: P2P supported */
744 	    (_rtw_memcmp(src->MacAddress, dst->MacAddress, ETH_ALEN) == _TRUE)
745 	   )
746 		return _TRUE;
747 #endif
748 
749 	/* Wi-Fi driver doesn't consider the situation of BCN and ProbRsp sent from the same hidden AP,
750 	  * it considers these two packets are sent from different AP.
751 	  * Therefore, the scan queue may store two scan results of the same hidden AP, likes below.
752 	  *
753 	  *  index            bssid              ch    RSSI   SdBm  Noise   age          flag             ssid
754 	  *    1    00:e0:4c:55:50:01    153   -73     -73        0     7044   [WPS][ESS]     RTK5G
755 	  *    3    00:e0:4c:55:50:01    153   -73     -73        0     7044   [WPS][ESS]
756 	  *
757 	  * Original rules will compare Ssid, SsidLength, MacAddress, s_cap, d_cap at the same time.
758 	  * Wi-Fi driver will assume that the BCN and ProbRsp sent from the same hidden AP are the same network
759 	  * after we add an additional rule to compare SsidLength and Ssid.
760 	  * It means the scan queue will not store two scan results of the same hidden AP, it only store ProbRsp.
761 	  * For customer request.
762 	  */
763 
764 	if (((_rtw_memcmp(src->MacAddress, dst->MacAddress, ETH_ALEN)) == _TRUE) &&
765 		((s_cap & WLAN_CAPABILITY_IBSS) == (d_cap & WLAN_CAPABILITY_IBSS)) &&
766 		((s_cap & WLAN_CAPABILITY_BSS) == (d_cap & WLAN_CAPABILITY_BSS))) {
767 		if ((src->Ssid.SsidLength == dst->Ssid.SsidLength) &&
768 			(((_rtw_memcmp(src->Ssid.Ssid, dst->Ssid.Ssid, src->Ssid.SsidLength)) == _TRUE) || //Case of normal AP
769 			(is_all_null(src->Ssid.Ssid, src->Ssid.SsidLength) == _TRUE || is_all_null(dst->Ssid.Ssid, dst->Ssid.SsidLength) == _TRUE))) //Case of hidden AP
770 			return _TRUE;
771 		else if ((src->Ssid.SsidLength == 0 || dst->Ssid.SsidLength == 0)) //Case of hidden AP
772 			return _TRUE;
773 		else
774 			return _FALSE;
775 	} else {
776 		return _FALSE;
777 	}
778 }
779 
_rtw_find_same_network(_queue * scanned_queue,struct wlan_network * network)780 struct wlan_network *_rtw_find_same_network(_queue *scanned_queue, struct wlan_network *network)
781 {
782 	_list *phead, *plist;
783 	struct wlan_network *found = NULL;
784 
785 	phead = get_list_head(scanned_queue);
786 	plist = get_next(phead);
787 
788 	while (plist != phead) {
789 		found = LIST_CONTAINOR(plist, struct wlan_network , list);
790 
791 		if (is_same_network(&network->network, &found->network, 0))
792 			break;
793 
794 		plist = get_next(plist);
795 	}
796 
797 	if (plist == phead)
798 		found = NULL;
799 
800 	return found;
801 }
802 
rtw_find_same_network(_queue * scanned_queue,struct wlan_network * network)803 struct wlan_network *rtw_find_same_network(_queue *scanned_queue, struct wlan_network *network)
804 {
805 	_irqL irqL;
806 	struct wlan_network *found = NULL;
807 
808 	if (scanned_queue == NULL || network == NULL)
809 		goto exit;
810 
811 	_enter_critical_bh(&scanned_queue->lock, &irqL);
812 	found = _rtw_find_same_network(scanned_queue, network);
813 	_exit_critical_bh(&scanned_queue->lock, &irqL);
814 
815 exit:
816 	return found;
817 }
818 
rtw_get_oldest_wlan_network(_queue * scanned_queue)819 struct	wlan_network	*rtw_get_oldest_wlan_network(_queue *scanned_queue)
820 {
821 	_list	*plist, *phead;
822 
823 
824 	struct	wlan_network	*pwlan = NULL;
825 	struct	wlan_network	*oldest = NULL;
826 	phead = get_list_head(scanned_queue);
827 
828 	plist = get_next(phead);
829 
830 	while (1) {
831 
832 		if (rtw_end_of_queue_search(phead, plist) == _TRUE)
833 			break;
834 
835 		pwlan = LIST_CONTAINOR(plist, struct wlan_network, list);
836 
837 		if (pwlan->fixed != _TRUE) {
838 			if (oldest == NULL || rtw_time_after(oldest->last_scanned, pwlan->last_scanned))
839 				oldest = pwlan;
840 		}
841 
842 		plist = get_next(plist);
843 	}
844 	return oldest;
845 
846 }
847 
update_network(WLAN_BSSID_EX * dst,WLAN_BSSID_EX * src,_adapter * padapter,bool update_ie)848 void update_network(WLAN_BSSID_EX *dst, WLAN_BSSID_EX *src,
849 		    _adapter *padapter, bool update_ie)
850 {
851 #if defined(DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) && 1
852 	u8 ss_ori = dst->PhyInfo.SignalStrength;
853 	u8 sq_ori = dst->PhyInfo.SignalQuality;
854 	u8 ss_smp = src->PhyInfo.SignalStrength;
855 	long rssi_smp = src->Rssi;
856 #endif
857 	long rssi_ori = dst->Rssi;
858 
859 	u8 sq_smp = src->PhyInfo.SignalQuality;
860 	u8 ss_final;
861 	u8 sq_final;
862 	long rssi_final;
863 
864 
865 #ifdef CONFIG_ANTENNA_DIVERSITY
866 	rtw_hal_antdiv_rssi_compared(padapter, dst, src); /* this will update src.Rssi, need consider again */
867 #endif
868 
869 #if defined(DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) && 1
870 	if (strcmp(dst->Ssid.Ssid, DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) == 0) {
871 		RTW_INFO(FUNC_ADPT_FMT" %s("MAC_FMT", ch%u) ss_ori:%3u, sq_ori:%3u, rssi_ori:%3ld, ss_smp:%3u, sq_smp:%3u, rssi_smp:%3ld\n"
872 			 , FUNC_ADPT_ARG(padapter)
873 			, src->Ssid.Ssid, MAC_ARG(src->MacAddress), src->Configuration.DSConfig
874 			 , ss_ori, sq_ori, rssi_ori
875 			 , ss_smp, sq_smp, rssi_smp
876 			);
877 	}
878 #endif
879 
880 	/* The rule below is 1/5 for sample value, 4/5 for history value */
881 	if (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE) && is_same_network(&(padapter->mlmepriv.cur_network.network), src, 0)) {
882 		/* Take the recvpriv's value for the connected AP*/
883 		ss_final = padapter->recvpriv.signal_strength;
884 		sq_final = padapter->recvpriv.signal_qual;
885 		/* the rssi value here is undecorated, and will be used for antenna diversity */
886 		if (sq_smp != 101) /* from the right channel */
887 			rssi_final = (src->Rssi + dst->Rssi * 4) / 5;
888 		else
889 			rssi_final = rssi_ori;
890 	} else {
891 		if (sq_smp != 101) { /* from the right channel */
892 			ss_final = ((u32)(src->PhyInfo.SignalStrength) + (u32)(dst->PhyInfo.SignalStrength) * 4) / 5;
893 			sq_final = ((u32)(src->PhyInfo.SignalQuality) + (u32)(dst->PhyInfo.SignalQuality) * 4) / 5;
894 			rssi_final = (src->Rssi + dst->Rssi * 4) / 5;
895 		} else {
896 			/* bss info not receving from the right channel, use the original RX signal infos */
897 			ss_final = dst->PhyInfo.SignalStrength;
898 			sq_final = dst->PhyInfo.SignalQuality;
899 			rssi_final = dst->Rssi;
900 		}
901 
902 	}
903 
904 	if (update_ie) {
905 		dst->Reserved[0] = src->Reserved[0];
906 		dst->Reserved[1] = src->Reserved[1];
907 		_rtw_memcpy((u8 *)dst, (u8 *)src, get_WLAN_BSSID_EX_sz(src));
908 	}
909 #ifdef CONFIG_LAYER2_ROAMING
910 	dst->tsf = src->tsf;
911 #endif
912 	dst->PhyInfo.SignalStrength = ss_final;
913 	dst->PhyInfo.SignalQuality = sq_final;
914 	dst->Rssi = rssi_final;
915 
916 #if defined(DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) && 1
917 	if (strcmp(dst->Ssid.Ssid, DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) == 0) {
918 		RTW_INFO(FUNC_ADPT_FMT" %s("MAC_FMT"), SignalStrength:%u, SignalQuality:%u, RawRSSI:%ld\n"
919 			 , FUNC_ADPT_ARG(padapter)
920 			, dst->Ssid.Ssid, MAC_ARG(dst->MacAddress), dst->PhyInfo.SignalStrength, dst->PhyInfo.SignalQuality, dst->Rssi);
921 	}
922 #endif
923 
924 #if 0 /* old codes, may be useful one day...
925  * 	RTW_INFO("update_network: rssi=0x%lx dst->Rssi=%d ,dst->Rssi=0x%lx , src->Rssi=0x%lx",(dst->Rssi+src->Rssi)/2,dst->Rssi,dst->Rssi,src->Rssi); */
926 	if (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE) && is_same_network(&(padapter->mlmepriv.cur_network.network), src)) {
927 
928 		/* RTW_INFO("b:ssid=%s update_network: src->rssi=0x%d padapter->recvpriv.ui_rssi=%d\n",src->Ssid.Ssid,src->Rssi,padapter->recvpriv.signal); */
929 		if (padapter->recvpriv.signal_qual_data.total_num++ >= PHY_LINKQUALITY_SLID_WIN_MAX) {
930 			padapter->recvpriv.signal_qual_data.total_num = PHY_LINKQUALITY_SLID_WIN_MAX;
931 			last_evm = padapter->recvpriv.signal_qual_data.elements[padapter->recvpriv.signal_qual_data.index];
932 			padapter->recvpriv.signal_qual_data.total_val -= last_evm;
933 		}
934 		padapter->recvpriv.signal_qual_data.total_val += query_rx_pwr_percentage(src->Rssi);
935 
936 		padapter->recvpriv.signal_qual_data.elements[padapter->recvpriv.signal_qual_data.index++] = query_rx_pwr_percentage(src->Rssi);
937 		if (padapter->recvpriv.signal_qual_data.index >= PHY_LINKQUALITY_SLID_WIN_MAX)
938 			padapter->recvpriv.signal_qual_data.index = 0;
939 
940 		/* RTW_INFO("Total SQ=%d  pattrib->signal_qual= %d\n", padapter->recvpriv.signal_qual_data.total_val, src->Rssi); */
941 
942 		/* <1> Showed on UI for user,in percentage. */
943 		tmpVal = padapter->recvpriv.signal_qual_data.total_val / padapter->recvpriv.signal_qual_data.total_num;
944 		padapter->recvpriv.signal = (u8)tmpVal; /* Link quality */
945 
946 		src->Rssi = translate_percentage_to_dbm(padapter->recvpriv.signal) ;
947 	} else {
948 		/*	RTW_INFO("ELSE:ssid=%s update_network: src->rssi=0x%d dst->rssi=%d\n",src->Ssid.Ssid,src->Rssi,dst->Rssi); */
949 		src->Rssi = (src->Rssi + dst->Rssi) / 2; /* dBM */
950 	}
951 
952 	/*	RTW_INFO("a:update_network: src->rssi=0x%d padapter->recvpriv.ui_rssi=%d\n",src->Rssi,padapter->recvpriv.signal); */
953 
954 #endif
955 
956 }
957 
update_current_network(_adapter * adapter,WLAN_BSSID_EX * pnetwork)958 static void update_current_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork)
959 {
960 	struct	mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
961 
962 
963 	rtw_bug_check(&(pmlmepriv->cur_network.network),
964 		      &(pmlmepriv->cur_network.network),
965 		      &(pmlmepriv->cur_network.network),
966 		      &(pmlmepriv->cur_network.network));
967 
968 	if ((check_fwstate(pmlmepriv, WIFI_ASOC_STATE) == _TRUE) && (is_same_network(&(pmlmepriv->cur_network.network), pnetwork, 0))) {
969 
970 		/* if(pmlmepriv->cur_network.network.IELength<= pnetwork->IELength) */
971 		{
972 			update_network(&(pmlmepriv->cur_network.network), pnetwork, adapter, _TRUE);
973 			rtw_update_protection(adapter, (pmlmepriv->cur_network.network.IEs) + sizeof(NDIS_802_11_FIXED_IEs),
974 				      pmlmepriv->cur_network.network.IELength);
975 		}
976 	}
977 
978 
979 }
980 
981 
982 /*
983 
984 Caller must hold pmlmepriv->lock first.
985 
986 
987 */
rtw_update_scanned_network(_adapter * adapter,WLAN_BSSID_EX * target)988 bool rtw_update_scanned_network(_adapter *adapter, WLAN_BSSID_EX *target)
989 {
990 	_irqL irqL;
991 	_list	*plist, *phead;
992 	u32	bssid_ex_sz;
993 	struct mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
994 #ifdef CONFIG_P2P
995 	struct wifidirect_info *pwdinfo = &(adapter->wdinfo);
996 #endif /* CONFIG_P2P */
997 	_queue	*queue	= &(pmlmepriv->scanned_queue);
998 	struct wlan_network	*pnetwork = NULL;
999 	struct wlan_network	*choice = NULL;
1000 	int target_find = 0;
1001 	u8 feature = 0;
1002 	bool update_ie = _FALSE;
1003 
1004 	_enter_critical_bh(&queue->lock, &irqL);
1005 	phead = get_list_head(queue);
1006 	plist = get_next(phead);
1007 
1008 #if 0
1009 	RTW_INFO("%s => ssid:%s , rssi:%ld , ss:%d\n",
1010 		__func__, target->Ssid.Ssid, target->Rssi, target->PhyInfo.SignalStrength);
1011 #endif
1012 
1013 #ifdef CONFIG_P2P
1014 	if (!rtw_p2p_chk_state(pwdinfo, P2P_STATE_NONE))
1015 		feature = 1; /* p2p enable */
1016 #endif
1017 
1018 	while (1) {
1019 		if (rtw_end_of_queue_search(phead, plist) == _TRUE)
1020 			break;
1021 
1022 		pnetwork = LIST_CONTAINOR(plist, struct wlan_network, list);
1023 
1024 		rtw_bug_check(pnetwork, pnetwork, pnetwork, pnetwork);
1025 
1026 #ifdef CONFIG_P2P
1027 		if (!rtw_p2p_chk_state(pwdinfo, P2P_STATE_NONE) &&
1028 		    (_rtw_memcmp(pnetwork->network.MacAddress, target->MacAddress, ETH_ALEN) == _TRUE)) {
1029 			target_find = 1;
1030 			break;
1031 		}
1032 #endif
1033 
1034 		if (is_same_network(&(pnetwork->network), target, feature)) {
1035 			target_find = 1;
1036 			break;
1037 		}
1038 
1039 		if (rtw_roam_flags(adapter)) {
1040 			/* TODO: don't  select netowrk in the same ess as choice if it's new enough*/
1041 		}
1042 		if (pnetwork->fixed) {
1043 			plist = get_next(plist);
1044 			continue;
1045 		}
1046 
1047 #ifdef CONFIG_RSSI_PRIORITY
1048 		if ((choice == NULL) || (pnetwork->network.PhyInfo.SignalStrength < choice->network.PhyInfo.SignalStrength))
1049 			#ifdef CONFIG_RTW_MESH
1050 			if (!MLME_IS_MESH(adapter) || !MLME_IS_ASOC(adapter)
1051 				|| !rtw_bss_is_same_mbss(&pmlmepriv->cur_network.network, &pnetwork->network))
1052 			#endif
1053 				choice = pnetwork;
1054 #else
1055 		if (choice == NULL || rtw_time_after(choice->last_scanned, pnetwork->last_scanned))
1056 			#ifdef CONFIG_RTW_MESH
1057 			if (!MLME_IS_MESH(adapter) || !MLME_IS_ASOC(adapter)
1058 				|| !rtw_bss_is_same_mbss(&pmlmepriv->cur_network.network, &pnetwork->network))
1059 			#endif
1060 				choice = pnetwork;
1061 #endif
1062 		plist = get_next(plist);
1063 
1064 	}
1065 
1066 
1067 	/* If we didn't find a match, then get a new network slot to initialize
1068 	 * with this beacon's information */
1069 	/* if (rtw_end_of_queue_search(phead,plist)== _TRUE) { */
1070 	if (!target_find) {
1071 		if (_rtw_queue_empty(&(pmlmepriv->free_bss_pool)) == _TRUE) {
1072 			/* If there are no more slots, expire the choice */
1073 			/* list_del_init(&choice->list); */
1074 			pnetwork = choice;
1075 			if (pnetwork == NULL)
1076 				goto unlock_scan_queue;
1077 
1078 #ifdef CONFIG_RSSI_PRIORITY
1079 		RTW_DBG("%s => ssid:%s ,bssid:"MAC_FMT"  will be deleted from scanned_queue (rssi:%ld , ss:%d)\n",
1080 			__func__, pnetwork->network.Ssid.Ssid, MAC_ARG(pnetwork->network.MacAddress), pnetwork->network.Rssi, pnetwork->network.PhyInfo.SignalStrength);
1081 #else
1082 		RTW_DBG("%s => ssid:%s ,bssid:"MAC_FMT" will be deleted from scanned_queue\n",
1083 			__func__, pnetwork->network.Ssid.Ssid, MAC_ARG(pnetwork->network.MacAddress));
1084 #endif
1085 
1086 #ifdef CONFIG_ANTENNA_DIVERSITY
1087 			rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(target->PhyInfo.Optimum_antenna), NULL);
1088 #endif
1089 			_rtw_memcpy(&(pnetwork->network), target,  get_WLAN_BSSID_EX_sz(target));
1090 			pnetwork->bcn_keys_valid = 0;
1091 			if (target->Reserved[0] == BSS_TYPE_BCN || target->Reserved[0] == BSS_TYPE_PROB_RSP)
1092 				rtw_update_bcn_keys_of_network(pnetwork);
1093 			/* variable initialize */
1094 			pnetwork->fixed = _FALSE;
1095 			pnetwork->last_scanned = rtw_get_current_time();
1096 			pnetwork->last_non_hidden_ssid_ap = pnetwork->last_scanned;
1097 			#if defined(CONFIG_RTW_MESH) && CONFIG_RTW_MESH_ACNODE_PREVENT
1098 			pnetwork->acnode_stime = 0;
1099 			pnetwork->acnode_notify_etime = 0;
1100 			#endif
1101 #ifdef	CONFIG_LAYER2_ROAMING
1102 			pnetwork->network.tsf = target->tsf;
1103 #endif
1104 			pnetwork->network_type = 0;
1105 			pnetwork->aid = 0;
1106 			pnetwork->join_res = 0;
1107 
1108 			/* bss info not receving from the right channel */
1109 			if (pnetwork->network.PhyInfo.SignalQuality == 101)
1110 				pnetwork->network.PhyInfo.SignalQuality = 0;
1111 		} else {
1112 			/* Otherwise just pull from the free list */
1113 
1114 			pnetwork = rtw_alloc_network(pmlmepriv); /* will update scan_time */
1115 			if (pnetwork == NULL)
1116 				goto unlock_scan_queue;
1117 
1118 			bssid_ex_sz = get_WLAN_BSSID_EX_sz(target);
1119 			target->Length = bssid_ex_sz;
1120 #ifdef CONFIG_ANTENNA_DIVERSITY
1121 			rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(target->PhyInfo.Optimum_antenna), NULL);
1122 #endif
1123 			_rtw_memcpy(&(pnetwork->network), target, bssid_ex_sz);
1124 			pnetwork->bcn_keys_valid = 0;
1125 			if (target->Reserved[0] == BSS_TYPE_BCN || target->Reserved[0] == BSS_TYPE_PROB_RSP)
1126 				rtw_update_bcn_keys_of_network(pnetwork);
1127 
1128 			/* bss info not receving from the right channel */
1129 			if (pnetwork->network.PhyInfo.SignalQuality == 101)
1130 				pnetwork->network.PhyInfo.SignalQuality = 0;
1131 
1132 			rtw_list_insert_tail(&(pnetwork->list), &(queue->queue));
1133 
1134 		}
1135 	} else {
1136 		/* we have an entry and we are going to update it. But this entry may
1137 		 * be already expired. In this case we do the same as we found a new
1138 		 * net and call the new_net handler
1139 		 */
1140 		#if defined(CONFIG_RTW_MESH) && CONFIG_RTW_MESH_ACNODE_PREVENT
1141 		systime last_scanned = pnetwork->last_scanned;
1142 		#endif
1143 		struct beacon_keys bcn_keys;
1144 		bool bcn_keys_valid = 0;
1145 		bool is_hidden_ssid_ap = 0;
1146 
1147 		pnetwork->last_scanned = rtw_get_current_time();
1148 
1149 		if (target->Reserved[0] == BSS_TYPE_BCN || target->Reserved[0] == BSS_TYPE_PROB_RSP) {
1150 			if (target->InfrastructureMode == Ndis802_11Infrastructure) {
1151 				is_hidden_ssid_ap = hidden_ssid_ap(target);
1152 				if (!is_hidden_ssid_ap) /* update last time it's non hidden ssid AP */
1153 					pnetwork->last_non_hidden_ssid_ap = rtw_get_current_time();
1154 			}
1155 			bcn_keys_valid = rtw_get_bcn_keys_from_bss(target, &bcn_keys);
1156 		}
1157 
1158 		if (target->InfrastructureMode == Ndis802_11_mesh
1159 			|| target->Reserved[0] >= pnetwork->network.Reserved[0])
1160 			update_ie = _TRUE;
1161 		else if (target->InfrastructureMode == Ndis802_11Infrastructure && !pnetwork->fixed
1162 			&& rtw_get_passing_time_ms(pnetwork->last_non_hidden_ssid_ap) > SCANQUEUE_LIFETIME)
1163 			update_ie = _TRUE;
1164 		else if (bcn_keys_valid) {
1165 			if (is_hidden_ssid(bcn_keys.ssid, bcn_keys.ssid_len)) {
1166 				/* hidden ssid, replace with current beacon ssid directly */
1167 				_rtw_memcpy(bcn_keys.ssid, pnetwork->bcn_keys.ssid, pnetwork->bcn_keys.ssid_len);
1168 				bcn_keys.ssid_len = pnetwork->bcn_keys.ssid_len;
1169 			}
1170 			if (rtw_bcn_key_compare(&pnetwork->bcn_keys, &bcn_keys) == _FALSE)
1171 				update_ie = _TRUE;
1172 		}
1173 
1174 		#if defined(CONFIG_RTW_MESH) && CONFIG_RTW_MESH_ACNODE_PREVENT
1175 		if (!MLME_IS_MESH(adapter) || !MLME_IS_ASOC(adapter)
1176 			|| pnetwork->network.Configuration.DSConfig != target->Configuration.DSConfig
1177 			|| rtw_get_passing_time_ms(last_scanned) > adapter->mesh_cfg.peer_sel_policy.scanr_exp_ms
1178 			|| !rtw_bss_is_same_mbss(&pnetwork->network, target)
1179 		) {
1180 			pnetwork->acnode_stime = 0;
1181 			pnetwork->acnode_notify_etime = 0;
1182 		}
1183 		#endif
1184 
1185 		if (bcn_keys_valid) {
1186 			_rtw_memcpy(&pnetwork->bcn_keys, &bcn_keys, sizeof(bcn_keys));
1187 			pnetwork->bcn_keys_valid = 1;
1188 		} else if (update_ie)
1189 			pnetwork->bcn_keys_valid = 0;
1190 
1191 		update_network(&(pnetwork->network), target, adapter, update_ie);
1192 	}
1193 
1194 	#if defined(CONFIG_RTW_MESH) && CONFIG_RTW_MESH_ACNODE_PREVENT
1195 	if (MLME_IS_MESH(adapter) && MLME_IS_ASOC(adapter))
1196 		rtw_mesh_update_scanned_acnode_status(adapter, pnetwork);
1197 	#endif
1198 
1199 unlock_scan_queue:
1200 	_exit_critical_bh(&queue->lock, &irqL);
1201 
1202 #ifdef CONFIG_RTW_MESH
1203 	if (pnetwork && MLME_IS_MESH(adapter)
1204 		&& check_fwstate(pmlmepriv, WIFI_ASOC_STATE)
1205 		&& !check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY)
1206 	)
1207 		rtw_chk_candidate_peer_notify(adapter, pnetwork);
1208 #endif
1209 
1210 	return update_ie;
1211 }
1212 
1213 void rtw_add_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork);
rtw_add_network(_adapter * adapter,WLAN_BSSID_EX * pnetwork)1214 void rtw_add_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork)
1215 {
1216 	bool update_ie;
1217 	/* _queue	*queue	= &(pmlmepriv->scanned_queue); */
1218 
1219 	/* _enter_critical_bh(&queue->lock, &irqL); */
1220 
1221 #if defined(CONFIG_P2P) && defined(CONFIG_P2P_REMOVE_GROUP_INFO)
1222 	if (adapter->registrypriv.wifi_spec == 0)
1223 		rtw_bss_ex_del_p2p_attr(pnetwork, P2P_ATTR_GROUP_INFO);
1224 #endif
1225 	if (!hal_chk_wl_func(adapter, WL_FUNC_MIRACAST))
1226 		rtw_bss_ex_del_wfd_ie(pnetwork);
1227 	/* Wi-Fi driver will update the current network if the scan result of the connected AP be updated by scan. */
1228 	update_ie = rtw_update_scanned_network(adapter, pnetwork);
1229 
1230 	if (update_ie)
1231 		update_current_network(adapter, pnetwork);
1232 
1233 	/* _exit_critical_bh(&queue->lock, &irqL); */
1234 
1235 }
1236 
1237 #ifdef CONFIG_RTW_MULTI_AP
rtw_unassoc_sta_set_mode(_adapter * adapter,u8 stype,u8 mode)1238 void rtw_unassoc_sta_set_mode(_adapter *adapter, u8 stype, u8 mode)
1239 {
1240 	if (stype >= UNASOC_STA_SRC_NUM
1241 		|| mode >= UNASOC_STA_MODE_NUM)
1242 		return;
1243 
1244 	adapter = GET_PRIMARY_ADAPTER(adapter);
1245 
1246 	if (adapter->mlmepriv.unassoc_sta_mode_of_stype[stype] == mode)
1247 		return;
1248 
1249 	adapter->mlmepriv.unassoc_sta_mode_of_stype[stype] = mode;
1250 
1251 	rtw_run_in_thread_cmd_wait(adapter, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), adapter, 2000);
1252 }
1253 
rtw_unassoc_sta_src_chk(_adapter * adapter,u8 stype)1254 bool rtw_unassoc_sta_src_chk(_adapter *adapter, u8 stype)
1255 {
1256 	if (stype >= UNASOC_STA_SRC_NUM)
1257 		return 0;
1258 
1259 	adapter = GET_PRIMARY_ADAPTER(adapter);
1260 
1261 	return adapter->mlmepriv.unassoc_sta_mode_of_stype[stype] == UNASOC_STA_MODE_ALL
1262 		|| (adapter->mlmepriv.unassoc_sta_mode_of_stype[stype] == UNASOC_STA_MODE_INTERESTED
1263 			&& adapter->mlmepriv.interested_unassoc_sta_cnt)
1264 		;
1265 }
1266 
1267 const char *unasoc_sta_src_str[] = {
1268 	"BMC",
1269 	"NMY_UC",
1270 };
1271 
1272 const char *unasoc_sta_mode_str[] = {
1273 	"DISABLED",
1274 	"INTERESTED",
1275 	"ALL",
1276 };
1277 
dump_unassoc_sta(void * sel,_adapter * adapter)1278 void dump_unassoc_sta(void *sel, _adapter *adapter)
1279 {
1280 	struct mlme_priv *mlmepriv;
1281 	_queue *queue;
1282 	_list *list, *head;
1283 	struct unassoc_sta_info **unassoc_sta_arr;
1284 	struct unassoc_sta_info *unassoc_sta;
1285 	u16 i, unassoc_sta_cnt = 0;
1286 
1287 	adapter = GET_PRIMARY_ADAPTER(adapter);
1288 	mlmepriv = &(adapter->mlmepriv);
1289 	queue = &(mlmepriv->unassoc_sta_queue);
1290 
1291 	for (i = 0; i < UNASOC_STA_SRC_NUM; i++) {
1292 		RTW_PRINT_SEL(sel, "[%u]%-6s:%u(%s)\n", i, unasoc_sta_src_str[i]
1293 			, mlmepriv->unassoc_sta_mode_of_stype[i], unasoc_sta_mode_str[mlmepriv->unassoc_sta_mode_of_stype[i]]);
1294 	}
1295 	RTW_PRINT_SEL(sel, "interested_unassoc_sta_cnt:%u\n", mlmepriv->interested_unassoc_sta_cnt);
1296 
1297 	unassoc_sta_arr = rtw_zvmalloc(mlmepriv->max_unassoc_sta_cnt * sizeof(struct unassoc_sta_info *));
1298 	if (!unassoc_sta_arr)
1299 		return;
1300 
1301 	enter_critical_bh(&queue->lock);
1302 	head = get_list_head(queue);
1303 	list = get_next(head);
1304 
1305 	while (rtw_end_of_queue_search(head, list) == _FALSE) {
1306 		unassoc_sta = LIST_CONTAINOR(list, struct unassoc_sta_info, list);
1307 		list = get_next(list);
1308 
1309 		unassoc_sta_arr[unassoc_sta_cnt++] = unassoc_sta;
1310 	}
1311 
1312 	exit_critical_bh(&queue->lock);
1313 
1314 	RTW_PRINT_SEL(sel, "  %17s %18s %6s\n", "mac_addr", "measure_delta_time", "rssi");
1315 
1316 	for (i = 0; i < unassoc_sta_cnt; i++) {
1317 		u8 rcpi;
1318 		s8 rx_power;
1319 		u32 measure_delta_time;
1320 
1321 		unassoc_sta = unassoc_sta_arr[i];
1322 
1323 		measure_delta_time = rtw_systime_to_ms(rtw_get_current_time() - unassoc_sta->time);
1324 
1325 		RTW_PRINT_SEL(sel, "%c "MAC_FMT" %18u %6d\n"
1326 			, unassoc_sta->interested ? '*' : ' '
1327 			, MAC_ARG(unassoc_sta->addr), measure_delta_time, unassoc_sta->recv_signal_power);
1328 	}
1329 
1330 	rtw_vmfree(unassoc_sta_arr, mlmepriv->max_unassoc_sta_cnt * sizeof(struct unassoc_sta_info *));
1331 }
1332 
del_unassoc_sta(struct mlme_priv * mlmepriv,struct unassoc_sta_info * unassoc_sta)1333 static void del_unassoc_sta(struct mlme_priv *mlmepriv, struct unassoc_sta_info *unassoc_sta)
1334 {
1335 	_irqL irqL;
1336 	_queue *free_queue = &(mlmepriv->free_unassoc_sta_queue);
1337 
1338 	if (unassoc_sta->interested)
1339 		mlmepriv->interested_unassoc_sta_cnt--;
1340 	if (mlmepriv->interested_unassoc_sta_cnt == 0) {
1341 		rtw_run_in_thread_cmd(mlme_to_adapter(mlmepriv)
1342 			, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), mlme_to_adapter(mlmepriv));
1343 	}
1344 
1345 	_enter_critical_bh(&free_queue->lock, &irqL);
1346 	rtw_list_delete(&(unassoc_sta->list));
1347 	rtw_list_insert_tail(&(unassoc_sta->list), &(free_queue->queue));
1348 	_exit_critical_bh(&free_queue->lock, &irqL);
1349 }
1350 
del_unassoc_sta_chk(struct mlme_priv * mlmepriv,struct unassoc_sta_info * unassoc_sta)1351 static u8 del_unassoc_sta_chk(struct mlme_priv *mlmepriv, struct unassoc_sta_info *unassoc_sta)
1352 {
1353 	systime cur, lifetime;
1354 
1355 	if (unassoc_sta == NULL)
1356 		return UNASOC_STA_DEL_CHK_SKIP;
1357 
1358 	if (unassoc_sta->interested)
1359 		return UNASOC_STA_DEL_CHK_SKIP;
1360 
1361 	cur = rtw_get_current_time();
1362 	lifetime = unassoc_sta->time + rtw_ms_to_systime(UNASSOC_STA_LIFETIME_MS);
1363 	if (rtw_time_before(cur, lifetime))
1364 		return UNASOC_STA_DEL_CHK_ALIVE;
1365 
1366 	del_unassoc_sta(mlmepriv, unassoc_sta);
1367 
1368 	return UNASOC_STA_DEL_CHK_DELETED;
1369 }
1370 
alloc_unassoc_sta(struct mlme_priv * mlmepriv)1371 static struct unassoc_sta_info *alloc_unassoc_sta(struct mlme_priv *mlmepriv)
1372 {
1373 	_irqL	irqL;
1374 	struct	unassoc_sta_info *unassoc_sta;
1375 	_queue *free_queue = &mlmepriv->free_unassoc_sta_queue;
1376 	_list *list = NULL;
1377 
1378 
1379 	_enter_critical_bh(&free_queue->lock, &irqL);
1380 
1381 	if (_rtw_queue_empty(free_queue) == _TRUE) {
1382 		unassoc_sta = NULL;
1383 		goto exit;
1384 	}
1385 	list = get_next(&(free_queue->queue));
1386 
1387 	unassoc_sta = LIST_CONTAINOR(list, struct unassoc_sta_info, list);
1388 
1389 	rtw_list_delete(&unassoc_sta->list);
1390 
1391 	_rtw_memset(unassoc_sta->addr, 0, ETH_ALEN);
1392 	unassoc_sta->recv_signal_power = 0;
1393 	unassoc_sta->time = 0;
1394 	unassoc_sta->interested = 0;
1395 exit:
1396 	_exit_critical_bh(&free_queue->lock, &irqL);
1397 
1398 	return unassoc_sta;
1399 
1400 }
1401 
rtw_del_unassoc_sta_queue(_adapter * adapter)1402 void rtw_del_unassoc_sta_queue(_adapter *adapter)
1403 {
1404 	struct unassoc_sta_info *unassoc_sta;
1405 	struct mlme_priv *mlmepriv;
1406 	_queue *queue;
1407 	_irqL irqL;
1408 	_list *head, *list;
1409 
1410 	adapter = GET_PRIMARY_ADAPTER(adapter);
1411 	mlmepriv = &(adapter->mlmepriv);
1412 	queue = &(mlmepriv->unassoc_sta_queue);
1413 
1414 	_enter_critical_bh(&queue->lock, &irqL);
1415 	head = get_list_head(queue);
1416 	list = get_next(head);
1417 
1418 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1419 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1420 		list = get_next(list);
1421 
1422 		del_unassoc_sta(mlmepriv, unassoc_sta);
1423 	}
1424 
1425 	_exit_critical_bh(&queue->lock, &irqL);
1426 
1427 }
1428 
rtw_del_unassoc_sta(_adapter * adapter,u8 * addr)1429 void rtw_del_unassoc_sta(_adapter *adapter, u8 *addr)
1430 {
1431 	struct unassoc_sta_info *unassoc_sta;
1432 	struct mlme_priv *mlmepriv;
1433 	_queue *queue;
1434 	_irqL irqL;
1435 	_list *head, *list;
1436 
1437 	adapter = GET_PRIMARY_ADAPTER(adapter);
1438 	mlmepriv = &(adapter->mlmepriv);
1439 	queue = &(mlmepriv->unassoc_sta_queue);
1440 
1441 	_enter_critical_bh(&queue->lock, &irqL);
1442 	head = get_list_head(queue);
1443 	list = get_next(head);
1444 
1445 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1446 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1447 		list = get_next(list);
1448 
1449 		if (_rtw_memcmp(addr, unassoc_sta->addr, ETH_ALEN) == _TRUE) {
1450 			del_unassoc_sta(mlmepriv, unassoc_sta);
1451 			goto unlock_unassoc_sta_queue;
1452 		}
1453 	}
1454 
1455 unlock_unassoc_sta_queue:
1456 	_exit_critical_bh(&queue->lock, &irqL);
1457 }
1458 
rtw_rx_add_unassoc_sta(_adapter * adapter,u8 stype,u8 * addr,s8 recv_signal_power)1459 void rtw_rx_add_unassoc_sta(_adapter *adapter, u8 stype, u8 *addr, s8 recv_signal_power)
1460 {
1461 	struct unassoc_sta_info *unassoc_sta;
1462 	struct unassoc_sta_info *oldest_unassoc_sta = NULL;
1463 	struct mlme_priv *mlmepriv;
1464 	_queue *queue;
1465 	_irqL irqL;
1466 	_list *head, *list;
1467 
1468 	adapter = GET_PRIMARY_ADAPTER(adapter);
1469 	mlmepriv = &(adapter->mlmepriv);
1470 	queue = &(mlmepriv->unassoc_sta_queue);
1471 
1472 	_enter_critical_bh(&queue->lock, &irqL);
1473 	head = get_list_head(queue);
1474 	list = get_next(head);
1475 
1476 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1477 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1478 		list = get_next(list);
1479 
1480 		if (_rtw_memcmp(addr, unassoc_sta->addr, ETH_ALEN) == _TRUE) {
1481 			if (unassoc_sta->interested
1482 				|| mlmepriv->unassoc_sta_mode_of_stype[stype] >= UNASOC_STA_MODE_ALL
1483 			) {
1484 				unassoc_sta->recv_signal_power = recv_signal_power;
1485 				unassoc_sta->time = rtw_get_current_time();
1486 				goto unlock_unassoc_sta_queue;
1487 			}
1488 		}
1489 
1490 		if (del_unassoc_sta_chk(mlmepriv, unassoc_sta) == UNASOC_STA_DEL_CHK_ALIVE) {
1491 			if (oldest_unassoc_sta == NULL)
1492 				oldest_unassoc_sta = unassoc_sta;
1493 			else if (rtw_time_before(unassoc_sta->time, oldest_unassoc_sta->time))
1494 				oldest_unassoc_sta = unassoc_sta;
1495 		}
1496 	}
1497 
1498 	if (mlmepriv->unassoc_sta_mode_of_stype[stype] <= UNASOC_STA_MODE_INTERESTED)
1499 		goto unlock_unassoc_sta_queue;
1500 
1501 	unassoc_sta = alloc_unassoc_sta(mlmepriv);
1502 	if (unassoc_sta == NULL) {
1503 		if (oldest_unassoc_sta) {
1504 			del_unassoc_sta(mlmepriv, oldest_unassoc_sta);
1505 			unassoc_sta = alloc_unassoc_sta(mlmepriv);
1506 		} else
1507 			goto unlock_unassoc_sta_queue;
1508 	}
1509 	_rtw_memcpy(unassoc_sta->addr, addr, ETH_ALEN);
1510 	unassoc_sta->recv_signal_power = recv_signal_power;
1511 	unassoc_sta->time = rtw_get_current_time();
1512 	rtw_list_insert_tail(&(unassoc_sta->list), &(queue->queue));
1513 
1514 unlock_unassoc_sta_queue:
1515 	_exit_critical_bh(&queue->lock, &irqL);
1516 }
1517 
rtw_add_interested_unassoc_sta(_adapter * adapter,u8 * addr)1518 void rtw_add_interested_unassoc_sta(_adapter *adapter, u8 *addr)
1519 {
1520 	struct unassoc_sta_info *unassoc_sta;
1521 	struct unassoc_sta_info *oldest_unassoc_sta = NULL;
1522 	struct mlme_priv *mlmepriv;
1523 	_queue *queue;
1524 	_irqL irqL;
1525 	_list *head, *list;
1526 
1527 	adapter = GET_PRIMARY_ADAPTER(adapter);
1528 	mlmepriv = &(adapter->mlmepriv);
1529 	queue = &(mlmepriv->unassoc_sta_queue);
1530 
1531 	_enter_critical_bh(&queue->lock, &irqL);
1532 	head = get_list_head(queue);
1533 	list = get_next(head);
1534 
1535 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1536 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1537 		list = get_next(list);
1538 
1539 		if (_rtw_memcmp(addr, unassoc_sta->addr, ETH_ALEN) == _TRUE) {
1540 			if (!unassoc_sta->interested) {
1541 				unassoc_sta->interested = 1;
1542 				mlmepriv->interested_unassoc_sta_cnt++;
1543 				if (mlmepriv->interested_unassoc_sta_cnt == 1) {
1544 					rtw_run_in_thread_cmd(mlme_to_adapter(mlmepriv)
1545 						, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), mlme_to_adapter(mlmepriv));
1546 				}
1547 			}
1548 			goto unlock_unassoc_sta_queue;
1549 		}
1550 
1551 		if (del_unassoc_sta_chk(mlmepriv, unassoc_sta) == UNASOC_STA_DEL_CHK_ALIVE) {
1552 			if (oldest_unassoc_sta == NULL)
1553 				oldest_unassoc_sta = unassoc_sta;
1554 			else if (rtw_time_after(unassoc_sta->time, oldest_unassoc_sta->time))
1555 				oldest_unassoc_sta = unassoc_sta;
1556 		}
1557 	}
1558 	unassoc_sta = alloc_unassoc_sta(mlmepriv);
1559 	if (unassoc_sta == NULL) {
1560 		RTW_INFO(FUNC_ADPT_FMT": Allocate fail\n", FUNC_ADPT_ARG(adapter));
1561 		if (oldest_unassoc_sta) {
1562 			RTW_INFO(FUNC_ADPT_FMT": Delete oldest entry and try again.\n", FUNC_ADPT_ARG(adapter));
1563 			del_unassoc_sta(mlmepriv, oldest_unassoc_sta);
1564 			unassoc_sta = alloc_unassoc_sta(mlmepriv);
1565 		} else
1566 			goto unlock_unassoc_sta_queue;
1567 	}
1568 	_rtw_memcpy(unassoc_sta->addr, addr, ETH_ALEN);
1569 	unassoc_sta->interested = 1;
1570 	unassoc_sta->recv_signal_power = 0;
1571 	unassoc_sta->time = rtw_get_current_time() - rtw_ms_to_systime(UNASSOC_STA_LIFETIME_MS);
1572 	rtw_list_insert_tail(&(unassoc_sta->list), &(queue->queue));
1573 	mlmepriv->interested_unassoc_sta_cnt++;
1574 	if (mlmepriv->interested_unassoc_sta_cnt == 1) {
1575 		rtw_run_in_thread_cmd(mlme_to_adapter(mlmepriv)
1576 			, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), mlme_to_adapter(mlmepriv));
1577 	}
1578 
1579 unlock_unassoc_sta_queue:
1580 	_exit_critical_bh(&queue->lock, &irqL);
1581 }
1582 
rtw_undo_interested_unassoc_sta(_adapter * adapter,u8 * addr)1583 void rtw_undo_interested_unassoc_sta(_adapter *adapter, u8 *addr)
1584 {
1585 	struct unassoc_sta_info *unassoc_sta;
1586 	struct mlme_priv *mlmepriv;
1587 	_queue *queue;
1588 	_irqL irqL;
1589 	_list *head, *list;
1590 
1591 	adapter = GET_PRIMARY_ADAPTER(adapter);
1592 	mlmepriv = &(adapter->mlmepriv);
1593 	queue = &(mlmepriv->unassoc_sta_queue);
1594 
1595 	_enter_critical_bh(&queue->lock, &irqL);
1596 	head = get_list_head(queue);
1597 	list = get_next(head);
1598 
1599 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1600 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1601 		list = get_next(list);
1602 
1603 		if (_rtw_memcmp(addr, unassoc_sta->addr, ETH_ALEN) == _TRUE) {
1604 			if (unassoc_sta->interested) {
1605 				unassoc_sta->interested = 0;
1606 				mlmepriv->interested_unassoc_sta_cnt--;
1607 				if (mlmepriv->interested_unassoc_sta_cnt == 0) {
1608 					rtw_run_in_thread_cmd(mlme_to_adapter(mlmepriv)
1609 						, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), mlme_to_adapter(mlmepriv));
1610 				}
1611 			}
1612 			goto unlock_unassoc_sta_queue;
1613 		}
1614 	}
1615 unlock_unassoc_sta_queue:
1616 	_exit_critical_bh(&queue->lock, &irqL);
1617 }
1618 
rtw_undo_all_interested_unassoc_sta(_adapter * adapter)1619 void rtw_undo_all_interested_unassoc_sta(_adapter *adapter)
1620 {
1621 	struct unassoc_sta_info *unassoc_sta;
1622 	struct mlme_priv *mlmepriv;
1623 	_queue *queue;
1624 	_irqL irqL;
1625 	_list *head, *list;
1626 
1627 	adapter = GET_PRIMARY_ADAPTER(adapter);
1628 	mlmepriv = &(adapter->mlmepriv);
1629 	queue = &(mlmepriv->unassoc_sta_queue);
1630 
1631 	_enter_critical_bh(&queue->lock, &irqL);
1632 	head = get_list_head(queue);
1633 	list = get_next(head);
1634 
1635 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1636 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1637 		list = get_next(list);
1638 
1639 		if (unassoc_sta->interested) {
1640 			unassoc_sta->interested = 0;
1641 			mlmepriv->interested_unassoc_sta_cnt--;
1642 			if (mlmepriv->interested_unassoc_sta_cnt == 0) {
1643 				rtw_run_in_thread_cmd(mlme_to_adapter(mlmepriv)
1644 					, ((void *)(rtw_hal_rcr_set_chk_bssid_act_non)), mlme_to_adapter(mlmepriv));
1645 				goto unlock_unassoc_sta_queue;
1646 			}
1647 		}
1648 	}
1649 unlock_unassoc_sta_queue:
1650 	_exit_critical_bh(&queue->lock, &irqL);
1651 }
1652 
rtw_search_unassoc_sta(_adapter * adapter,u8 * addr,struct unassoc_sta_info * ret_sta)1653 u8 rtw_search_unassoc_sta(_adapter *adapter, u8 *addr, struct unassoc_sta_info *ret_sta)
1654 {
1655 	struct unassoc_sta_info *unassoc_sta = NULL;
1656 	struct mlme_priv *mlmepriv;
1657 	_queue *queue;
1658 	_irqL irqL;
1659 	_list *head, *list;
1660 	u8 searched = 0;
1661 
1662 	adapter = GET_PRIMARY_ADAPTER(adapter);
1663 	mlmepriv = &(adapter->mlmepriv);
1664 	queue = &(mlmepriv->unassoc_sta_queue);
1665 
1666 	_enter_critical_bh(&queue->lock, &irqL);
1667 	head = get_list_head(queue);
1668 	list = get_next(head);
1669 
1670 	while ((rtw_end_of_queue_search(head, list)) == _FALSE) {
1671 		unassoc_sta = LIST_CONTAINOR(list , struct unassoc_sta_info, list);
1672 		list = get_next(list);
1673 
1674 		if (_rtw_memcmp(addr, unassoc_sta->addr, ETH_ALEN) == _TRUE) {
1675 			memcpy(ret_sta, unassoc_sta, sizeof(struct unassoc_sta_info));
1676 			searched = 1;
1677 			break;
1678 		}
1679 	}
1680 	_exit_critical_bh(&queue->lock, &irqL);
1681 
1682 	return searched;
1683 }
1684 #endif /* CONFIG_RTW_MULTI_AP */
1685 
1686 /* select the desired network based on the capability of the (i)bss.
1687  * check items: (1) security
1688  *			   (2) network_type
1689  *			   (3) WMM
1690  *			   (4) HT
1691  * (5) others */
1692 int rtw_is_desired_network(_adapter *adapter, struct wlan_network *pnetwork);
rtw_is_desired_network(_adapter * adapter,struct wlan_network * pnetwork)1693 int rtw_is_desired_network(_adapter *adapter, struct wlan_network *pnetwork)
1694 {
1695 	struct security_priv *psecuritypriv = &adapter->securitypriv;
1696 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
1697 	u32 desired_encmode;
1698 	u32 privacy;
1699 
1700 	/* u8 wps_ie[512]; */
1701 	uint wps_ielen;
1702 
1703 	int bselected = _TRUE;
1704 
1705 	desired_encmode = psecuritypriv->ndisencryptstatus;
1706 	privacy = pnetwork->network.Privacy;
1707 
1708 	if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
1709 		if (rtw_get_wps_ie(pnetwork->network.IEs + _FIXED_IE_LENGTH_, pnetwork->network.IELength - _FIXED_IE_LENGTH_, NULL, &wps_ielen) != NULL)
1710 			return _TRUE;
1711 		else
1712 			return _FALSE;
1713 	}
1714 	if (adapter->registrypriv.wifi_spec == 1) { /* for  correct flow of 8021X  to do.... */
1715 		u8 *p = NULL;
1716 		uint ie_len = 0;
1717 
1718 		if ((desired_encmode == Ndis802_11EncryptionDisabled) && (privacy != 0))
1719 			bselected = _FALSE;
1720 
1721 		if (psecuritypriv->ndisauthtype == Ndis802_11AuthModeWPA2PSK) {
1722 			p = rtw_get_ie(pnetwork->network.IEs + _BEACON_IE_OFFSET_, _RSN_IE_2_, &ie_len, (pnetwork->network.IELength - _BEACON_IE_OFFSET_));
1723 			if (p && ie_len > 0)
1724 				bselected = _TRUE;
1725 			else
1726 				bselected = _FALSE;
1727 		}
1728 	}
1729 
1730 
1731 	if ((desired_encmode != Ndis802_11EncryptionDisabled) && (privacy == 0)) {
1732 		RTW_INFO("desired_encmode: %d, privacy: %d\n", desired_encmode, privacy);
1733 		bselected = _FALSE;
1734 	}
1735 
1736 	if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE) {
1737 		if (pnetwork->network.InfrastructureMode != pmlmepriv->cur_network.network.InfrastructureMode)
1738 			bselected = _FALSE;
1739 	}
1740 
1741 #ifdef CONFIG_RTW_MBO
1742 	if (rtw_mbo_disallowed_network(pnetwork) == _TRUE)
1743 		bselected = _FALSE;
1744 #endif
1745 
1746 	return bselected;
1747 }
1748 
rtw_survey_event_callback(_adapter * adapter,u8 * pbuf)1749 void rtw_survey_event_callback(_adapter	*adapter, u8 *pbuf)
1750 {
1751 	_irqL  irqL;
1752 	u32 len;
1753 	u8 val8;
1754 	WLAN_BSSID_EX *pnetwork;
1755 	struct	mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
1756 
1757 	pnetwork = (WLAN_BSSID_EX *)pbuf;
1758 
1759 	len = get_WLAN_BSSID_EX_sz(pnetwork);
1760 	if (len > (sizeof(WLAN_BSSID_EX))) {
1761 		return;
1762 	}
1763 
1764 #ifdef CONFIG_RTW_80211K
1765     	val8 = 0;
1766 	rtw_hal_get_hwreg(adapter, HW_VAR_FREECNT, &val8);
1767 
1768 	/* use TSF if no free run counter */
1769 	if (val8==0)
1770 		pnetwork->PhyInfo.free_cnt = (u32)rtw_hal_get_tsftr_by_port(
1771 			adapter, rtw_hal_get_port(adapter));
1772 #endif
1773 
1774 	if (pnetwork->InfrastructureMode == Ndis802_11Infrastructure) {
1775 		if (MLME_IS_SCAN(adapter)) {
1776 			adapter->mlmeextpriv.sitesurvey_res.activate_ch_cnt
1777 				+= rtw_process_beacon_hint(adapter, pnetwork);
1778 		}
1779 	}
1780 
1781 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
1782 
1783 	/* update IBSS_network 's timestamp */
1784 	if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE)) == _TRUE) {
1785 		if (_rtw_memcmp(&(pmlmepriv->cur_network.network.MacAddress), pnetwork->MacAddress, ETH_ALEN)) {
1786 			struct wlan_network *ibss_wlan = NULL;
1787 			_irqL	irqL;
1788 
1789 			_rtw_memcpy(pmlmepriv->cur_network.network.IEs, pnetwork->IEs, 8);
1790 			_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1791 			ibss_wlan = _rtw_find_network(&pmlmepriv->scanned_queue,  pnetwork->MacAddress);
1792 			if (ibss_wlan) {
1793 				_rtw_memcpy(ibss_wlan->network.IEs , pnetwork->IEs, 8);
1794 				_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1795 				goto exit;
1796 			}
1797 			_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1798 		}
1799 	}
1800 
1801 	/* lock pmlmepriv->lock when you accessing network_q */
1802 	if ((check_fwstate(pmlmepriv, WIFI_UNDER_LINKING)) == _FALSE) {
1803 		if (pnetwork->Ssid.Ssid[0] == 0)
1804 			pnetwork->Ssid.SsidLength = 0;
1805 		rtw_add_network(adapter, pnetwork);
1806 	}
1807 
1808 exit:
1809 
1810 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
1811 
1812 
1813 	return;
1814 }
1815 
rtw_surveydone_event_callback(_adapter * adapter,u8 * pbuf)1816 void rtw_surveydone_event_callback(_adapter	*adapter, u8 *pbuf)
1817 {
1818 	_irqL  irqL;
1819 	struct surveydone_event *parm = (struct surveydone_event *)pbuf;
1820 	struct	mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
1821 	struct mlme_ext_priv	*pmlmeext = &adapter->mlmeextpriv;
1822 
1823 #ifdef CONFIG_MLME_EXT
1824 	mlmeext_surveydone_event_callback(adapter);
1825 #endif
1826 
1827 
1828 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
1829 	if (pmlmepriv->wps_probe_req_ie) {
1830 		u32 free_len = pmlmepriv->wps_probe_req_ie_len;
1831 		pmlmepriv->wps_probe_req_ie_len = 0;
1832 		rtw_mfree(pmlmepriv->wps_probe_req_ie, free_len);
1833 		pmlmepriv->wps_probe_req_ie = NULL;
1834 	}
1835 
1836 
1837 	if (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY) == _FALSE) {
1838 		RTW_INFO(FUNC_ADPT_FMT" fw_state:0x%x\n", FUNC_ADPT_ARG(adapter), get_fwstate(pmlmepriv));
1839 		/* rtw_warn_on(1); */
1840 	}
1841 
1842 	if (pmlmeext->scan_abort == _TRUE)
1843 		pmlmeext->scan_abort = _FALSE;
1844 
1845 	_clr_fwstate_(pmlmepriv, WIFI_UNDER_SURVEY);
1846 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
1847 
1848 	_cancel_timer_ex(&pmlmepriv->scan_to_timer);
1849 
1850 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
1851 
1852 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
1853 	rtw_set_signal_stat_timer(&adapter->recvpriv);
1854 #endif
1855 
1856 	if (pmlmepriv->to_join == _TRUE) {
1857 		if ((check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE)) {
1858 			if (check_fwstate(pmlmepriv, WIFI_ASOC_STATE) == _FALSE) {
1859 				set_fwstate(pmlmepriv, WIFI_UNDER_LINKING);
1860 
1861 				if (rtw_select_and_join_from_scanned_queue(pmlmepriv) == _SUCCESS)
1862 					_set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
1863 				#ifdef CONFIG_AP_MODE
1864 				else {
1865 					WLAN_BSSID_EX    *pdev_network = &(adapter->registrypriv.dev_network);
1866 					u8 *pibss = adapter->registrypriv.dev_network.MacAddress;
1867 
1868 					/* pmlmepriv->fw_state ^= WIFI_UNDER_SURVEY; */ /* because don't set assoc_timer */
1869 					_clr_fwstate_(pmlmepriv, WIFI_UNDER_SURVEY);
1870 
1871 
1872 					_rtw_memset(&pdev_network->Ssid, 0, sizeof(NDIS_802_11_SSID));
1873 					_rtw_memcpy(&pdev_network->Ssid, &pmlmepriv->assoc_ssid, sizeof(NDIS_802_11_SSID));
1874 
1875 					rtw_update_registrypriv_dev_network(adapter);
1876 					rtw_generate_random_ibss(pibss);
1877 
1878 					/*pmlmepriv->fw_state = WIFI_ADHOC_MASTER_STATE;*/
1879 					init_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE);
1880 
1881 					if (rtw_create_ibss_cmd(adapter, 0) != _SUCCESS)
1882 						RTW_ERR("rtw_create_ibss_cmd FAIL\n");
1883 
1884 					pmlmepriv->to_join = _FALSE;
1885 				}
1886 				#endif /* CONFIG_AP_MODE */
1887 			}
1888 		} else {
1889 			int s_ret;
1890 			set_fwstate(pmlmepriv, WIFI_UNDER_LINKING);
1891 			pmlmepriv->to_join = _FALSE;
1892 			s_ret = rtw_select_and_join_from_scanned_queue(pmlmepriv);
1893 			if (_SUCCESS == s_ret)
1894 				_set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
1895 			else if (s_ret == 2) { /* there is no need to wait for join */
1896 				_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING);
1897 				rtw_indicate_connect(adapter);
1898 			} else {
1899 				RTW_INFO("try_to_join, but select scanning queue fail, to_roam:%d\n", rtw_to_roam(adapter));
1900 
1901 				if (rtw_to_roam(adapter) != 0) {
1902 					struct sitesurvey_parm scan_parm;
1903 					u8 ssc_chk = rtw_sitesurvey_condition_check(adapter, _FALSE);
1904 
1905 					rtw_init_sitesurvey_parm(adapter, &scan_parm);
1906 					_rtw_memcpy(&scan_parm.ssid[0], &pmlmepriv->assoc_ssid, sizeof(NDIS_802_11_SSID));
1907 					scan_parm.ssid_num = 1;
1908 
1909 					if (rtw_dec_to_roam(adapter) == 0
1910 						|| (ssc_chk != SS_ALLOW && ssc_chk != SS_DENY_BUSY_TRAFFIC)
1911 						|| _SUCCESS != rtw_sitesurvey_cmd(adapter, &scan_parm)
1912 					   ) {
1913 						rtw_set_to_roam(adapter, 0);
1914 						rtw_free_assoc_resources(adapter, _TRUE);
1915 						rtw_indicate_disconnect(adapter, 0, _FALSE);
1916 					} else
1917 						pmlmepriv->to_join = _TRUE;
1918 				} else
1919 					rtw_indicate_disconnect(adapter, 0, _FALSE);
1920 				_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING);
1921 			}
1922 		}
1923 	} else {
1924 		if (rtw_chk_roam_flags(adapter, RTW_ROAM_ACTIVE)
1925 		#if (defined(CONFIG_RTW_WNM) && defined(CONFIG_RTW_80211R))
1926 			|| rtw_wnm_btm_roam_triggered(adapter)
1927 		#endif
1928 		) {
1929 			if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)
1930 			    && check_fwstate(pmlmepriv, WIFI_ASOC_STATE)) {
1931 				if (rtw_select_roaming_candidate(pmlmepriv) == _SUCCESS) {
1932 			#ifdef CONFIG_RTW_80211R
1933 					rtw_ft_start_roam(adapter,
1934 						(u8 *)pmlmepriv->roam_network->network.MacAddress);
1935 			#else
1936 					receive_disconnect(adapter, pmlmepriv->cur_network.network.MacAddress
1937 						, WLAN_REASON_ACTIVE_ROAM, _FALSE);
1938 			#endif
1939 				}
1940 			}
1941 		}
1942 	}
1943 
1944 	/* RTW_INFO("scan complete in %dms\n",rtw_get_passing_time_ms(pmlmepriv->scan_start_time)); */
1945 
1946 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
1947 
1948 #ifdef CONFIG_P2P_PS
1949 	if (check_fwstate(pmlmepriv, WIFI_ASOC_STATE) == _TRUE)
1950 		p2p_ps_wk_cmd(adapter, P2P_PS_SCAN_DONE, 0);
1951 #endif /* CONFIG_P2P_PS */
1952 
1953 	rtw_mi_os_xmit_schedule(adapter);
1954 
1955 #ifdef CONFIG_DRVEXT_MODULE_WSC
1956 	drvext_surveydone_callback(&adapter->drvextpriv);
1957 #endif
1958 
1959 #ifdef DBG_CONFIG_ERROR_DETECT
1960 	{
1961 		struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
1962 		if (pmlmeext->sitesurvey_res.bss_cnt == 0) {
1963 			/* rtw_hal_sreset_reset(adapter); */
1964 		}
1965 	}
1966 #endif
1967 
1968 #ifdef CONFIG_IOCTL_CFG80211
1969 	rtw_cfg80211_surveydone_event_callback(adapter);
1970 #endif /* CONFIG_IOCTL_CFG80211 */
1971 
1972 	rtw_indicate_scan_done(adapter, _FALSE);
1973 
1974 #if defined(CONFIG_CONCURRENT_MODE) && defined(CONFIG_IOCTL_CFG80211)
1975 	rtw_cfg80211_indicate_scan_done_for_buddy(adapter, _FALSE);
1976 #endif
1977 
1978 	if (parm->activate_ch_cnt) {
1979 		#ifdef CONFIG_IOCTL_CFG80211
1980 		struct get_chplan_resp *chplan;
1981 
1982 		if (rtw_get_chplan_cmd(adapter, RTW_CMDF_DIRECTLY, &chplan) != _SUCCESS
1983 			|| rtw_regd_change_complete_async(adapter_to_wiphy(adapter), chplan) != _SUCCESS)
1984 			rtw_warn_on(1);
1985 		#endif
1986 
1987 		op_class_pref_apply_regulatory(adapter, REG_BEACON_HINT);
1988 		rtw_nlrtw_reg_beacon_hint_event(adapter);
1989 	}
1990 
1991 #ifdef CONFIG_RTW_MESH
1992 	#if CONFIG_RTW_MESH_OFFCH_CAND
1993 	if (rtw_mesh_offch_candidate_accepted(adapter)) {
1994 		u8 ch;
1995 
1996 		ch = rtw_mesh_select_operating_ch(adapter);
1997 		if (ch && pmlmepriv->cur_network.network.Configuration.DSConfig != ch) {
1998 			u8 ifbmp = rtw_mi_get_ap_mesh_ifbmp(adapter);
1999 
2000 			if (ifbmp) {
2001 				/* switch to selected channel */
2002 				rtw_change_bss_chbw_cmd(adapter, RTW_CMDF_DIRECTLY, ifbmp, 0, ch, REQ_BW_ORI, REQ_OFFSET_NONE);
2003 				issue_probereq_ex(adapter, &pmlmepriv->cur_network.network.mesh_id, NULL, 0, 0, 0, 0);
2004 			} else
2005 				rtw_warn_on(1);
2006 		}
2007 	}
2008 	#endif
2009 #endif /* CONFIG_RTW_MESH */
2010 
2011 #ifdef CONFIG_RTW_ACS
2012 	if (parm->acs) {
2013 		u8 ifbmp = rtw_mi_get_ap_mesh_ifbmp(adapter);
2014 
2015 		if (ifbmp)
2016 			rtw_change_bss_chbw_cmd(adapter, RTW_CMDF_DIRECTLY, ifbmp, 0, REQ_CH_INT_INFO, REQ_BW_ORI, REQ_OFFSET_NONE);
2017 	}
2018 #endif
2019 }
2020 
_rtw_sitesurvey_condition_check(const char * caller,_adapter * adapter,bool check_sc_interval)2021 u8 _rtw_sitesurvey_condition_check(const char *caller, _adapter *adapter, bool check_sc_interval)
2022 {
2023 	u8 ss_condition = SS_ALLOW;
2024 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
2025 	struct registry_priv *registry_par = &adapter->registrypriv;
2026 
2027 
2028 #ifdef CONFIG_MP_INCLUDED
2029 	if (rtw_mp_mode_check(adapter)) {
2030 		RTW_INFO("%s ("ADPT_FMT") MP mode block Scan request\n", caller, ADPT_ARG(adapter));
2031 		ss_condition = SS_DENY_MP_MODE;
2032 		goto _exit;
2033 	}
2034 #endif
2035 
2036 #ifdef DBG_LA_MODE
2037 	if(registry_par->la_mode_en == 1 && MLME_IS_ASOC(adapter)) {
2038 		RTW_INFO("%s ("ADPT_FMT") LA debug mode block Scan request\n", caller, ADPT_ARG(adapter));
2039 		ss_condition = SS_DENY_LA_MODE;
2040 		goto _exit;
2041 	}
2042 #endif
2043 
2044 #ifdef CONFIG_RTW_REPEATER_SON
2045 	if (adapter->rtw_rson_scanstage == RSON_SCAN_PROCESS) {
2046 		RTW_INFO("%s ("ADPT_FMT") blocking scan for under rson scanning process\n", caller, ADPT_ARG(adapter));
2047 		ss_condition = SS_DENY_RSON_SCANING;
2048 		goto _exit;
2049 	}
2050 #endif
2051 #ifdef CONFIG_IOCTL_CFG80211
2052 	if (adapter_wdev_data(adapter)->block_scan == _TRUE) {
2053 		RTW_INFO("%s ("ADPT_FMT") wdev_priv.block_scan is set\n", caller, ADPT_ARG(adapter));
2054 		ss_condition = SS_DENY_BLOCK_SCAN;
2055 		goto _exit;
2056 	}
2057 #endif
2058 
2059 	if (adapter_to_dvobj(adapter)->scan_deny == _TRUE) {
2060 		RTW_INFO("%s ("ADPT_FMT") tpt mode, scan deny!\n", caller, ADPT_ARG(adapter));
2061 		ss_condition = SS_DENY_BLOCK_SCAN;
2062 		goto _exit;
2063 	}
2064 
2065 	if (rtw_is_scan_deny(adapter)) {
2066 		RTW_INFO("%s ("ADPT_FMT") : scan deny\n", caller, ADPT_ARG(adapter));
2067 		ss_condition = SS_DENY_BY_DRV;
2068 		goto _exit;
2069 	}
2070 
2071 #ifdef CONFIG_ADAPTIVITY_DENY_SCAN
2072 	if (adapter_to_rfctl(adapter)->adaptivity_en
2073 	    && rtw_phydm_get_edcca_flag(adapter)
2074 	    && rtw_is_2g_ch(GET_HAL_DATA(adapter)->current_channel)) {
2075 		RTW_WARN(FUNC_ADPT_FMT": Adaptivity block scan! (ch=%u)\n",
2076 			 FUNC_ADPT_ARG(adapter),
2077 			 GET_HAL_DATA(adapter)->current_channel);
2078 		ss_condition = SS_DENY_ADAPTIVITY;
2079 		goto _exit;
2080 	}
2081 #endif /* CONFIG_ADAPTIVITY_DENY_SCAN */
2082 
2083 	if (check_fwstate(pmlmepriv, WIFI_AP_STATE)){
2084 		if(check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
2085 			RTW_INFO("%s ("ADPT_FMT") : scan abort!! AP mode process WPS\n", caller, ADPT_ARG(adapter));
2086 			ss_condition = SS_DENY_SELF_AP_UNDER_WPS;
2087 			goto _exit;
2088 		} else if (check_fwstate(pmlmepriv, WIFI_UNDER_LINKING) == _TRUE) {
2089 			RTW_INFO("%s ("ADPT_FMT") : scan abort!!AP mode under linking (fwstate=0x%x)\n",
2090 				caller, ADPT_ARG(adapter), pmlmepriv->fw_state);
2091 			ss_condition = SS_DENY_SELF_AP_UNDER_LINKING;
2092 			goto _exit;
2093 		} else if (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY) == _TRUE) {
2094 			RTW_INFO("%s ("ADPT_FMT") : scan abort!!AP mode under survey (fwstate=0x%x)\n",
2095 				caller, ADPT_ARG(adapter), pmlmepriv->fw_state);
2096 			ss_condition = SS_DENY_SELF_AP_UNDER_SURVEY;
2097 			goto _exit;
2098 		}
2099 	} else {
2100 		if (check_fwstate(pmlmepriv, WIFI_UNDER_LINKING) == _TRUE) {
2101 			RTW_INFO("%s ("ADPT_FMT") : scan abort!!STA mode under linking (fwstate=0x%x)\n",
2102 				caller, ADPT_ARG(adapter), pmlmepriv->fw_state);
2103 			ss_condition = SS_DENY_SELF_STA_UNDER_LINKING;
2104 			goto _exit;
2105 		} else if (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY) == _TRUE) {
2106 			RTW_INFO("%s ("ADPT_FMT") : scan abort!!STA mode under survey (fwstate=0x%x)\n",
2107 				caller, ADPT_ARG(adapter), pmlmepriv->fw_state);
2108 			ss_condition = SS_DENY_SELF_STA_UNDER_SURVEY;
2109 			goto _exit;
2110 		}
2111 	}
2112 
2113 #ifdef CONFIG_CONCURRENT_MODE
2114 	if (rtw_mi_buddy_check_fwstate(adapter, WIFI_UNDER_LINKING | WIFI_UNDER_WPS)) {
2115 		RTW_INFO("%s ("ADPT_FMT") : scan abort!! buddy_intf under linking or wps\n", caller, ADPT_ARG(adapter));
2116 		ss_condition = SS_DENY_BUDDY_UNDER_LINK_WPS;
2117 		goto _exit;
2118 
2119 	} else if (rtw_mi_buddy_check_fwstate(adapter, WIFI_UNDER_SURVEY)) {
2120 		RTW_INFO("%s ("ADPT_FMT") : scan abort!! buddy_intf under survey\n", caller, ADPT_ARG(adapter));
2121 		ss_condition = SS_DENY_BUDDY_UNDER_SURVEY;
2122 		goto _exit;
2123 	}
2124 #endif /* CONFIG_CONCURRENT_MODE */
2125 
2126 #ifdef RTW_BUSY_DENY_SCAN
2127 	/*
2128 	 * busy traffic check
2129 	 * Rules:
2130 	 * 1. If (scan interval <= BUSY_TRAFFIC_SCAN_DENY_PERIOD) always allow
2131 	 *    scan, otherwise goto rule 2.
2132 	 * 2. Deny scan if any interface is busy, otherwise allow scan.
2133 	 */
2134 	if (pmlmepriv->lastscantime
2135 	    && (rtw_get_passing_time_ms(pmlmepriv->lastscantime) >
2136 		registry_par->scan_interval_thr)
2137 	    && rtw_mi_busy_traffic_check(adapter)) {
2138 		RTW_WARN("%s ("ADPT_FMT") : scan abort!! BusyTraffic\n",
2139 			 caller, ADPT_ARG(adapter));
2140 		ss_condition = SS_DENY_BUSY_TRAFFIC;
2141 		goto _exit;
2142 	}
2143 #endif /* RTW_BUSY_DENY_SCAN */
2144 
2145 _exit:
2146 	return ss_condition;
2147 }
2148 
free_scanqueue(struct mlme_priv * pmlmepriv)2149 static void free_scanqueue(struct	mlme_priv *pmlmepriv)
2150 {
2151 	_irqL irqL, irqL0;
2152 	_queue *free_queue = &pmlmepriv->free_bss_pool;
2153 	_queue *scan_queue = &pmlmepriv->scanned_queue;
2154 	_list	*plist, *phead, *ptemp;
2155 
2156 
2157 	_enter_critical_bh(&scan_queue->lock, &irqL0);
2158 	_enter_critical_bh(&free_queue->lock, &irqL);
2159 
2160 	phead = get_list_head(scan_queue);
2161 	plist = get_next(phead);
2162 
2163 	while (plist != phead) {
2164 		ptemp = get_next(plist);
2165 		rtw_list_delete(plist);
2166 		rtw_list_insert_tail(plist, &free_queue->queue);
2167 		plist = ptemp;
2168 		pmlmepriv->num_of_scanned--;
2169 	}
2170 
2171 	_exit_critical_bh(&free_queue->lock, &irqL);
2172 	_exit_critical_bh(&scan_queue->lock, &irqL0);
2173 
2174 }
2175 
rtw_reset_rx_info(_adapter * adapter)2176 void rtw_reset_rx_info(_adapter *adapter)
2177 {
2178 	struct recv_priv  *precvpriv = &adapter->recvpriv;
2179 
2180 	precvpriv->dbg_rx_ampdu_drop_count = 0;
2181 	precvpriv->dbg_rx_ampdu_forced_indicate_count = 0;
2182 	precvpriv->dbg_rx_ampdu_loss_count = 0;
2183 	precvpriv->dbg_rx_dup_mgt_frame_drop_count = 0;
2184 	precvpriv->dbg_rx_ampdu_window_shift_cnt = 0;
2185 	precvpriv->dbg_rx_drop_count = 0;
2186 	precvpriv->dbg_rx_conflic_mac_addr_cnt = 0;
2187 }
2188 
2189 /*
2190 *rtw_free_assoc_resources: the caller has to lock pmlmepriv->lock
2191 */
rtw_free_assoc_resources(_adapter * adapter,u8 lock_scanned_queue)2192 void rtw_free_assoc_resources(_adapter *adapter, u8 lock_scanned_queue)
2193 {
2194 	_irqL irqL;
2195 	struct wlan_network *pwlan = NULL;
2196 	struct	mlme_priv *pmlmepriv = &adapter->mlmepriv;
2197 	struct wlan_network *tgt_network = &pmlmepriv->cur_network;
2198 
2199 
2200 #ifdef CONFIG_TDLS
2201 	struct tdls_info *ptdlsinfo = &adapter->tdlsinfo;
2202 #endif /* CONFIG_TDLS */
2203 
2204 
2205 	RTW_INFO("%s-"ADPT_FMT" tgt_network MacAddress=" MAC_FMT" ssid=%s\n",
2206 		__func__, ADPT_ARG(adapter), MAC_ARG(tgt_network->network.MacAddress), tgt_network->network.Ssid.Ssid);
2207 
2208 	if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
2209 		struct sta_info *psta;
2210 
2211 		psta = rtw_get_stainfo(&adapter->stapriv, tgt_network->network.MacAddress);
2212 
2213 #ifdef CONFIG_TDLS
2214 		rtw_free_all_tdls_sta(adapter, _TRUE);
2215 		rtw_reset_tdls_info(adapter);
2216 
2217 		if (ptdlsinfo->link_established == _TRUE)
2218 			rtw_tdls_cmd(adapter, NULL, TDLS_RS_RCR);
2219 #endif /* CONFIG_TDLS */
2220 
2221 		/* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
2222 		rtw_free_stainfo(adapter, psta);
2223 		/* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
2224 
2225 	}
2226 
2227 	if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE | WIFI_ADHOC_MASTER_STATE)) {
2228 		struct sta_info *psta;
2229 
2230 		rtw_free_all_stainfo(adapter);
2231 
2232 		psta = rtw_get_bcmc_stainfo(adapter);
2233 		/* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL);		 */
2234 		rtw_free_stainfo(adapter, psta);
2235 		/* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL);		 */
2236 
2237 		rtw_init_bcmc_stainfo(adapter);
2238 	}
2239 
2240 	if (lock_scanned_queue)
2241 		_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2242 
2243 	if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS) || (pmlmepriv->wpa_phase == _TRUE)){
2244 		RTW_INFO("Dont free disconnecting network of scanned_queue due to uner %s %s phase\n\n",
2245 			check_fwstate(pmlmepriv, WIFI_UNDER_WPS) ? "WPS" : "",
2246 			(pmlmepriv->wpa_phase == _TRUE) ? "WPA" : "");
2247 	} else {
2248 		pwlan = _rtw_find_same_network(&pmlmepriv->scanned_queue, tgt_network);
2249 		if (pwlan) {
2250 			pwlan->fixed = _FALSE;
2251 
2252 			RTW_INFO("Free disconnecting network of scanned_queue\n");
2253 			rtw_free_network_nolock(adapter, pwlan);
2254 #ifdef CONFIG_P2P
2255 			if (!rtw_p2p_chk_state(&adapter->wdinfo, P2P_STATE_NONE)) {
2256 				rtw_set_scan_deny(adapter, 2000);
2257 				/* rtw_clear_scan_deny(adapter); */
2258 			}
2259 #endif /* CONFIG_P2P */
2260 		} else
2261 			RTW_ERR("Free disconnecting network of scanned_queue failed due to pwlan == NULL\n\n");
2262 	}
2263 
2264 	if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) && (adapter->stapriv.asoc_sta_count == 1))
2265 	    /*||check_fwstate(pmlmepriv, WIFI_STATION_STATE)*/) {
2266 		if (pwlan)
2267 			rtw_free_network_nolock(adapter, pwlan);
2268 	}
2269 
2270 	if (lock_scanned_queue)
2271 		_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2272 
2273 	adapter->securitypriv.key_mask = 0;
2274 
2275 	rtw_reset_rx_info(adapter);
2276 
2277 
2278 }
2279 
2280 /*
2281 *rtw_indicate_connect: the caller has to lock pmlmepriv->lock
2282 */
rtw_indicate_connect(_adapter * padapter)2283 void rtw_indicate_connect(_adapter *padapter)
2284 {
2285 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
2286 
2287 	pmlmepriv->to_join = _FALSE;
2288 
2289 	if (!check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE)) {
2290 
2291 		set_fwstate(pmlmepriv, WIFI_ASOC_STATE);
2292 
2293 		rtw_led_control(padapter, LED_CTL_LINK);
2294 
2295 		rtw_os_indicate_connect(padapter);
2296 
2297 		#ifdef CONFIG_RTW_WDS
2298 		if (MLME_IS_STA(padapter))
2299 			rtw_wds_gptr_tbl_init(padapter);
2300 		#endif
2301 	}
2302 
2303 	rtw_set_to_roam(padapter, 0);
2304 	if (!MLME_IS_AP(padapter) && !MLME_IS_MESH(padapter))
2305 		rtw_mi_set_scan_deny(padapter, 3000);
2306 
2307 
2308 }
2309 
2310 
2311 /*
2312 *rtw_indicate_disconnect: the caller has to lock pmlmepriv->lock
2313 */
rtw_indicate_disconnect(_adapter * padapter,u16 reason,u8 locally_generated)2314 void rtw_indicate_disconnect(_adapter *padapter, u16 reason, u8 locally_generated)
2315 {
2316 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
2317 	struct mlme_ext_priv *pmlmeext = &(padapter->mlmeextpriv);
2318 	struct mlme_ext_info *pmlmeinfo = &(pmlmeext->mlmext_info);
2319 	WLAN_BSSID_EX	*cur_network = &(pmlmeinfo->network);
2320 #ifdef CONFIG_WAPI_SUPPORT
2321 	struct sta_info *psta;
2322 	struct sta_priv *pstapriv = &padapter->stapriv;
2323 #endif
2324 	u8 *wps_ie = NULL;
2325 	uint wpsie_len = 0;
2326 
2327 	if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS))
2328 		pmlmepriv->wpa_phase = _TRUE;
2329 
2330 	_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING | WIFI_UNDER_WPS | WIFI_OP_CH_SWITCHING | WIFI_UNDER_KEY_HANDSHAKE);
2331 
2332 	/* force to clear cur_network_scanned's SELECTED REGISTRAR */
2333 	if (pmlmepriv->cur_network_scanned) {
2334 		WLAN_BSSID_EX	*current_joined_bss = &(pmlmepriv->cur_network_scanned->network);
2335 		if (current_joined_bss) {
2336 			wps_ie = rtw_get_wps_ie(current_joined_bss->IEs + _FIXED_IE_LENGTH_,
2337 				current_joined_bss->IELength - _FIXED_IE_LENGTH_, NULL, &wpsie_len);
2338 			if (wps_ie && wpsie_len > 0) {
2339 				u8 *attr = NULL;
2340 				u32 attr_len;
2341 				attr = rtw_get_wps_attr(wps_ie, wpsie_len, WPS_ATTR_SELECTED_REGISTRAR,
2342 							NULL, &attr_len);
2343 				if (attr)
2344 					*(attr + 4) = 0;
2345 			}
2346 		}
2347 	}
2348 	/* RTW_INFO("clear wps when %s\n", __func__); */
2349 
2350 	if (rtw_to_roam(padapter) > 0)
2351 		_clr_fwstate_(pmlmepriv, WIFI_ASOC_STATE);
2352 
2353 #ifdef CONFIG_WAPI_SUPPORT
2354 	psta = rtw_get_stainfo(pstapriv, cur_network->MacAddress);
2355 	if (check_fwstate(pmlmepriv, WIFI_STATION_STATE))
2356 		rtw_wapi_return_one_sta_info(padapter, psta->cmn.mac_addr);
2357 	else if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) ||
2358 		 check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE))
2359 		rtw_wapi_return_all_sta_info(padapter);
2360 #endif
2361 
2362 	if (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE)
2363 	    || (rtw_to_roam(padapter) <= 0)
2364 	   ) {
2365 		#ifdef CONFIG_RTW_WDS
2366 		adapter_set_use_wds(padapter, 0);
2367 		rtw_wds_gptr_tbl_unregister(padapter);
2368 		#endif
2369 		#ifdef CONFIG_RTW_MULTI_AP
2370 		padapter->multi_ap = 0;
2371 		#endif
2372 
2373 #ifdef CONFIG_RTW_TOKEN_BASED_XMIT
2374 		if (ATOMIC_READ(&padapter->tbtx_tx_pause) == _TRUE) {
2375 			ATOMIC_SET(&padapter->tbtx_tx_pause, _FALSE);
2376 			rtw_tx_control_cmd(padapter);
2377 		}
2378 #endif
2379 
2380 		rtw_os_indicate_disconnect(padapter, reason, locally_generated);
2381 
2382 		/* set ips_deny_time to avoid enter IPS before LPS leave */
2383 		rtw_set_ips_deny(padapter, 3000);
2384 
2385 		_clr_fwstate_(pmlmepriv, WIFI_ASOC_STATE);
2386 
2387 		rtw_led_control(padapter, LED_CTL_NO_LINK);
2388 
2389 		rtw_clear_scan_deny(padapter);
2390 	}
2391 
2392 #ifdef CONFIG_P2P_PS
2393 	p2p_ps_wk_cmd(padapter, P2P_PS_DISABLE, 1);
2394 #endif /* CONFIG_P2P_PS */
2395 
2396 #ifdef CONFIG_LPS
2397 	rtw_lps_ctrl_wk_cmd(padapter, LPS_CTRL_DISCONNECT, 0);
2398 #endif
2399 
2400 #ifdef CONFIG_BEAMFORMING
2401 	beamforming_wk_cmd(padapter, BEAMFORMING_CTRL_LEAVE, cur_network->MacAddress, ETH_ALEN, 1);
2402 #endif /*CONFIG_BEAMFORMING*/
2403 
2404 }
2405 
rtw_indicate_scan_done(_adapter * padapter,bool aborted)2406 inline void rtw_indicate_scan_done(_adapter *padapter, bool aborted)
2407 {
2408 	RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(padapter));
2409 
2410 	rtw_os_indicate_scan_done(padapter, aborted);
2411 
2412 #ifdef CONFIG_IPS
2413 	if (is_primary_adapter(padapter)
2414 	    && (_FALSE == adapter_to_pwrctl(padapter)->bInSuspend)
2415 	    && (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE | WIFI_UNDER_LINKING) == _FALSE)) {
2416 		struct pwrctrl_priv *pwrpriv;
2417 
2418 		pwrpriv = adapter_to_pwrctl(padapter);
2419 		rtw_set_ips_deny(padapter, 0);
2420 #ifdef CONFIG_IPS_CHECK_IN_WD
2421 		_set_timer(&adapter_to_dvobj(padapter)->dynamic_chk_timer, 1);
2422 #else /* !CONFIG_IPS_CHECK_IN_WD */
2423 		_rtw_set_pwr_state_check_timer(pwrpriv, 1);
2424 #endif /* !CONFIG_IPS_CHECK_IN_WD */
2425 	}
2426 #endif /* CONFIG_IPS */
2427 }
2428 
_rtw_wait_scan_done(_adapter * adapter,u8 abort,u32 timeout_ms)2429 static u32 _rtw_wait_scan_done(_adapter *adapter, u8 abort, u32 timeout_ms)
2430 {
2431 	systime start;
2432 	u32 pass_ms;
2433 	struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
2434 	struct mlme_ext_priv *pmlmeext = &(adapter->mlmeextpriv);
2435 
2436 	start = rtw_get_current_time();
2437 
2438 	pmlmeext->scan_abort = abort;
2439 
2440 	while (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY)
2441 	       && rtw_get_passing_time_ms(start) <= timeout_ms) {
2442 
2443 		if (RTW_CANNOT_RUN(adapter))
2444 			break;
2445 
2446 		RTW_INFO(FUNC_NDEV_FMT"fw_state=WIFI_UNDER_SURVEY!\n", FUNC_NDEV_ARG(adapter->pnetdev));
2447 		rtw_msleep_os(20);
2448 	}
2449 
2450 	if (_TRUE == abort) {
2451 		if (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY)) {
2452 			if (!RTW_CANNOT_RUN(adapter))
2453 				RTW_INFO(FUNC_NDEV_FMT"waiting for scan_abort time out!\n", FUNC_NDEV_ARG(adapter->pnetdev));
2454 #ifdef CONFIG_PLATFORM_MSTAR
2455 			/*_clr_fwstate_(pmlmepriv, WIFI_UNDER_SURVEY);*/
2456 			set_survey_timer(pmlmeext, 0);
2457 			mlme_set_scan_to_timer(pmlmepriv, 50);
2458 #endif
2459 			rtw_indicate_scan_done(adapter, _TRUE);
2460 		}
2461 	}
2462 
2463 	pmlmeext->scan_abort = _FALSE;
2464 	pass_ms = rtw_get_passing_time_ms(start);
2465 
2466 	return pass_ms;
2467 
2468 }
2469 
rtw_scan_wait_completed(_adapter * adapter)2470 void rtw_scan_wait_completed(_adapter *adapter)
2471 {
2472 	struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
2473 	struct ss_res *ss = &pmlmeext->sitesurvey_res;
2474 
2475 	_rtw_wait_scan_done(adapter, _FALSE, ss->scan_timeout_ms);
2476 }
2477 
rtw_scan_abort_timeout(_adapter * adapter,u32 timeout_ms)2478 u32 rtw_scan_abort_timeout(_adapter *adapter, u32 timeout_ms)
2479 {
2480 	return _rtw_wait_scan_done(adapter, _TRUE, timeout_ms);
2481 }
2482 
rtw_scan_abort_no_wait(_adapter * adapter)2483 void rtw_scan_abort_no_wait(_adapter *adapter)
2484 {
2485 	struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
2486 	struct mlme_ext_priv *pmlmeext = &(adapter->mlmeextpriv);
2487 
2488 	if (check_fwstate(pmlmepriv, WIFI_UNDER_SURVEY))
2489 		pmlmeext->scan_abort = _TRUE;
2490 }
2491 
rtw_scan_abort(_adapter * adapter)2492 void rtw_scan_abort(_adapter *adapter)
2493 {
2494 	rtw_scan_abort_timeout(adapter, 200);
2495 }
2496 
_rtw_wait_join_done(_adapter * adapter,u8 abort,u32 timeout_ms)2497 static u32 _rtw_wait_join_done(_adapter *adapter, u8 abort, u32 timeout_ms)
2498 {
2499 	systime start;
2500 	u32 pass_ms;
2501 	struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
2502 	struct mlme_ext_priv *pmlmeext = &(adapter->mlmeextpriv);
2503 
2504 	start = rtw_get_current_time();
2505 
2506 	pmlmeext->join_abort = abort;
2507 	if (abort)
2508 		set_link_timer(pmlmeext, 1);
2509 
2510 	while (rtw_get_passing_time_ms(start) <= timeout_ms
2511 		&& (check_fwstate(pmlmepriv, WIFI_UNDER_LINKING)
2512 			#ifdef CONFIG_IOCTL_CFG80211
2513 			|| rtw_cfg80211_is_connect_requested(adapter)
2514 			#endif
2515 			)
2516 	) {
2517 		if (RTW_CANNOT_RUN(adapter))
2518 			break;
2519 
2520 		RTW_INFO(FUNC_ADPT_FMT" linking...\n", FUNC_ADPT_ARG(adapter));
2521 		rtw_msleep_os(20);
2522 	}
2523 
2524 	if (abort) {
2525 		if (check_fwstate(pmlmepriv, WIFI_UNDER_LINKING)
2526 			#ifdef CONFIG_IOCTL_CFG80211
2527 			|| rtw_cfg80211_is_connect_requested(adapter)
2528 			#endif
2529 		) {
2530 			if (!RTW_CANNOT_RUN(adapter))
2531 				RTW_INFO(FUNC_ADPT_FMT" waiting for join_abort time out!\n", FUNC_ADPT_ARG(adapter));
2532 		}
2533 	}
2534 
2535 	pmlmeext->join_abort = 0;
2536 	pass_ms = rtw_get_passing_time_ms(start);
2537 
2538 	return pass_ms;
2539 }
2540 
rtw_join_abort_timeout(_adapter * adapter,u32 timeout_ms)2541 u32 rtw_join_abort_timeout(_adapter *adapter, u32 timeout_ms)
2542 {
2543 	return _rtw_wait_join_done(adapter, _TRUE, timeout_ms);
2544 }
2545 
rtw_joinbss_update_stainfo(_adapter * padapter,struct wlan_network * pnetwork)2546 static struct sta_info *rtw_joinbss_update_stainfo(_adapter *padapter, struct wlan_network *pnetwork)
2547 {
2548 	int i;
2549 	struct sta_info *psta = NULL;
2550 	struct recv_reorder_ctrl *preorder_ctrl;
2551 	struct sta_priv *pstapriv = &padapter->stapriv;
2552 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
2553 #if defined(CONFIG_RTS_FULL_BW) || defined(CONFIG_RTW_MULTI_AP)
2554 	struct	mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
2555 #endif
2556 #ifdef CONFIG_RTS_FULL_BW
2557 	struct wlan_network  *cur_network = &(pmlmepriv->cur_network);
2558 #endif
2559 
2560 	psta = rtw_get_stainfo(pstapriv, pnetwork->network.MacAddress);
2561 	if (psta == NULL)
2562 		psta = rtw_alloc_stainfo(pstapriv, pnetwork->network.MacAddress);
2563 
2564 	if (psta) { /* update ptarget_sta */
2565 		RTW_INFO("%s\n", __FUNCTION__);
2566 
2567 		psta->cmn.aid  = pnetwork->join_res;
2568 
2569 		update_sta_info(padapter, psta);
2570 
2571 		/* update station supportRate */
2572 		psta->bssratelen = rtw_get_rateset_len(pnetwork->network.SupportedRates);
2573 		_rtw_memcpy(psta->bssrateset, pnetwork->network.SupportedRates, psta->bssratelen);
2574 		rtw_hal_update_sta_ra_info(padapter, psta);
2575 
2576 		psta->wireless_mode = pmlmeext->cur_wireless_mode;
2577 		rtw_hal_update_sta_wset(padapter, psta);
2578 
2579 		/* sta mode */
2580 		rtw_hal_set_odm_var(padapter, HAL_ODM_STA_INFO, psta, _TRUE);
2581 
2582 		/* security related */
2583 #ifdef CONFIG_RTW_80211R
2584 		if ((padapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X)
2585 			&& (psta->ft_pairwise_key_installed == _FALSE)) {
2586 #else
2587 		if (padapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) {
2588 #endif
2589 			u8 *ie;
2590 			sint ie_len;
2591 			u8 mfp_opt = MFP_NO;
2592 			u8 spp_opt = 0;
2593 
2594 			padapter->securitypriv.binstallGrpkey = _FALSE;
2595 			padapter->securitypriv.busetkipkey = _FALSE;
2596 			padapter->securitypriv.bgrpkey_handshake = _FALSE;
2597 
2598 			ie = rtw_get_ie(pnetwork->network.IEs + _BEACON_IE_OFFSET_, WLAN_EID_RSN
2599 				, &ie_len, (pnetwork->network.IELength - _BEACON_IE_OFFSET_));
2600 			if (ie && ie_len > 0
2601 				&& rtw_parse_wpa2_ie(ie, ie_len + 2, NULL, NULL, NULL, NULL, &mfp_opt, &spp_opt) == _SUCCESS
2602 			) {
2603 				if (padapter->securitypriv.mfp_opt >= MFP_OPTIONAL && mfp_opt >= MFP_OPTIONAL)
2604 					psta->flags |= WLAN_STA_MFP;
2605 			}
2606 
2607 			if (padapter->securitypriv.dot11PrivacyAlgrthm != _NO_PRIVACY_ ) {
2608 				/*check if amsdu is allowed */
2609 				if (rtw_check_amsdu_disable(padapter->registrypriv.amsdu_mode, spp_opt) == _TRUE)
2610 					psta->flags |= WLAN_STA_AMSDU_DISABLE;
2611 			}
2612 			psta->ieee8021x_blocked = _TRUE;
2613 			psta->dot118021XPrivacy = padapter->securitypriv.dot11PrivacyAlgrthm;
2614 
2615 			_rtw_memset((u8 *)&psta->dot118021x_UncstKey, 0, sizeof(union Keytype));
2616 			_rtw_memset((u8 *)&psta->dot11tkiprxmickey, 0, sizeof(union Keytype));
2617 			_rtw_memset((u8 *)&psta->dot11tkiptxmickey, 0, sizeof(union Keytype));
2618 		}
2619 
2620 		/*	Commented by Albert 2012/07/21 */
2621 		/*	When doing the WPS, the wps_ie_len won't equal to 0 */
2622 		/*	And the Wi-Fi driver shouldn't allow the data packet to be tramsmitted. */
2623 		if (padapter->securitypriv.wps_ie_len != 0) {
2624 			psta->ieee8021x_blocked = _TRUE;
2625 			padapter->securitypriv.wps_ie_len = 0;
2626 		}
2627 
2628 
2629 		/* for A-MPDU Rx reordering buffer control for sta_info */
2630 		/* if A-MPDU Rx is enabled, reseting  rx_ordering_ctrl wstart_b(indicate_seq) to default value=0xffff */
2631 		/* todo: check if AP can send A-MPDU packets */
2632 		for (i = 0; i < 16 ; i++) {
2633 			/* preorder_ctrl = &precvpriv->recvreorder_ctrl[i]; */
2634 			preorder_ctrl = &psta->recvreorder_ctrl[i];
2635 			preorder_ctrl->enable = _FALSE;
2636 			preorder_ctrl->indicate_seq = 0xffff;
2637 			rtw_clear_bit(RTW_RECV_ACK_OR_TIMEOUT, &preorder_ctrl->rec_abba_rsp_ack);
2638 			#ifdef DBG_RX_SEQ
2639 			RTW_INFO("DBG_RX_SEQ "FUNC_ADPT_FMT" tid:%u SN_CLEAR indicate_seq:%u preorder_ctrl->rec_abba_rsp_ack:%lu\n"
2640 				, FUNC_ADPT_ARG(padapter)
2641 				, i
2642 				, preorder_ctrl->indicate_seq
2643 				,preorder_ctrl->rec_abba_rsp_ack
2644 				);
2645 			#endif
2646 			preorder_ctrl->wend_b = 0xffff;
2647 			preorder_ctrl->wsize_b = 64;/* max_ampdu_sz; */ /* ex. 32(kbytes) -> wsize_b=32 */
2648 			preorder_ctrl->ampdu_size = RX_AMPDU_SIZE_INVALID;
2649 		}
2650 	}
2651 
2652 #ifdef	CONFIG_RTW_80211K
2653 	_rtw_memcpy(&psta->rm_en_cap, pnetwork->network.PhyInfo.rm_en_cap, 5);
2654 #endif
2655 #ifdef CONFIG_RTW_MULTI_AP
2656 	if (padapter->multi_ap & MULTI_AP_BACKHAUL_STA) {
2657 		u8 multi_ap = rtw_get_multi_ap_ie_ext(pmlmepriv->assoc_rsp + sizeof(struct rtw_ieee80211_hdr_3addr) + 6
2658 				, pmlmepriv->assoc_rsp_len - sizeof(struct rtw_ieee80211_hdr_3addr) - 6);
2659 
2660 		if (multi_ap & MULTI_AP_BACKHAUL_BSS) /* backhaul bss, enable WDS */
2661 			psta->flags |= WLAN_STA_MULTI_AP | WLAN_STA_WDS;
2662 		else if (multi_ap & MULTI_AP_FRONTHAUL_BSS) /* fronthaul bss only */
2663 			psta->flags |= WLAN_STA_MULTI_AP;
2664 	}
2665 #endif
2666 #ifdef CONFIG_RTS_FULL_BW
2667 	rtw_parse_sta_vendor_ie_8812(padapter, psta, BSS_EX_TLV_IES(&cur_network->network), BSS_EX_TLV_IES_LEN(&cur_network->network));
2668 #endif
2669 	return psta;
2670 
2671 }
2672 
2673 /* pnetwork : returns from rtw_joinbss_event_callback
2674  * ptarget_wlan: found from scanned_queue */
2675 static void rtw_joinbss_update_network(_adapter *padapter, struct wlan_network *ptarget_wlan, struct wlan_network  *pnetwork)
2676 {
2677 	struct mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
2678 	struct security_priv *psecuritypriv = &padapter->securitypriv;
2679 	struct wlan_network  *cur_network = &(pmlmepriv->cur_network);
2680 	sint tmp_fw_state = 0x0;
2681 
2682 	RTW_INFO("%s\n", __FUNCTION__);
2683 
2684 	/* why not use ptarget_wlan?? */
2685 	_rtw_memcpy(&cur_network->network, &pnetwork->network, pnetwork->network.Length);
2686 	/* some IEs in pnetwork is wrong, so we should use ptarget_wlan IEs */
2687 	cur_network->network.IELength = ptarget_wlan->network.IELength;
2688 	_rtw_memcpy(&cur_network->network.IEs[0], &ptarget_wlan->network.IEs[0], MAX_IE_SZ);
2689 
2690 	cur_network->aid = pnetwork->join_res;
2691 
2692 
2693 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
2694 	rtw_set_signal_stat_timer(&padapter->recvpriv);
2695 #endif
2696 	padapter->recvpriv.signal_strength = ptarget_wlan->network.PhyInfo.SignalStrength;
2697 	padapter->recvpriv.signal_qual = ptarget_wlan->network.PhyInfo.SignalQuality;
2698 	/* the ptarget_wlan->network.Rssi is raw data, we use ptarget_wlan->network.PhyInfo.SignalStrength instead (has scaled) */
2699 	padapter->recvpriv.rssi = translate_percentage_to_dbm(ptarget_wlan->network.PhyInfo.SignalStrength);
2700 #if defined(DBG_RX_SIGNAL_DISPLAY_PROCESSING) && 1
2701 	RTW_INFO(FUNC_ADPT_FMT" signal_strength:%3u, rssi:%3d, signal_qual:%3u"
2702 		 "\n"
2703 		 , FUNC_ADPT_ARG(padapter)
2704 		 , padapter->recvpriv.signal_strength
2705 		 , padapter->recvpriv.rssi
2706 		 , padapter->recvpriv.signal_qual
2707 		);
2708 #endif
2709 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
2710 	rtw_set_signal_stat_timer(&padapter->recvpriv);
2711 #endif
2712 
2713 	/* update fw_state */ /* will clr WIFI_UNDER_LINKING here indirectly */
2714 
2715 	switch (pnetwork->network.InfrastructureMode) {
2716 	case Ndis802_11Infrastructure:
2717 		/* Check encryption */
2718 		if (psecuritypriv->dot11AuthAlgrthm == dot11AuthAlgrthm_8021X)
2719 			tmp_fw_state = tmp_fw_state | WIFI_UNDER_KEY_HANDSHAKE;
2720 
2721 		if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS))
2722 			tmp_fw_state = tmp_fw_state | WIFI_UNDER_WPS;
2723 
2724 		init_fwstate(pmlmepriv, WIFI_STATION_STATE | tmp_fw_state);
2725 
2726 		break;
2727 	case Ndis802_11IBSS:
2728 		/*pmlmepriv->fw_state = WIFI_ADHOC_STATE;*/
2729 		init_fwstate(pmlmepriv, WIFI_ADHOC_STATE);
2730 		break;
2731 	default:
2732 		/*pmlmepriv->fw_state = WIFI_NULL_STATE;*/
2733 		init_fwstate(pmlmepriv, WIFI_NULL_STATE);
2734 		break;
2735 	}
2736 
2737 	rtw_update_protection(padapter, (cur_network->network.IEs) + sizeof(NDIS_802_11_FIXED_IEs),
2738 			      (cur_network->network.IELength));
2739 
2740 #ifdef CONFIG_80211N_HT
2741 	rtw_update_ht_cap(padapter, cur_network->network.IEs, cur_network->network.IELength, (u8) cur_network->network.Configuration.DSConfig);
2742 #endif
2743 }
2744 
2745 /* Notes: the fucntion could be > passive_level (the same context as Rx tasklet)
2746  * pnetwork : returns from rtw_joinbss_event_callback
2747  * ptarget_wlan: found from scanned_queue
2748  * if join_res > 0, for (fw_state==WIFI_STATION_STATE), we check if  "ptarget_sta" & "ptarget_wlan" exist.
2749  * if join_res > 0, for (fw_state==WIFI_ADHOC_STATE), we only check if "ptarget_wlan" exist.
2750  * if join_res > 0, update "cur_network->network" from "pnetwork->network" if (ptarget_wlan !=NULL).
2751  */
2752 /* #define REJOIN */
2753 void rtw_joinbss_event_prehandle(_adapter *adapter, u8 *pbuf, u16 status)
2754 {
2755 	_irqL irqL;
2756 	static u8 retry = 0;
2757 	struct sta_info *ptarget_sta = NULL, *pcur_sta = NULL;
2758 	struct	sta_priv *pstapriv = &adapter->stapriv;
2759 	struct	mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
2760 	struct wlan_network	*pnetwork	= (struct wlan_network *)pbuf;
2761 	struct wlan_network	*cur_network = &(pmlmepriv->cur_network);
2762 	struct wlan_network	*pcur_wlan = NULL, *ptarget_wlan = NULL;
2763 	unsigned int		the_same_macaddr = _FALSE;
2764 
2765 	rtw_get_encrypt_decrypt_from_registrypriv(adapter);
2766 
2767 	the_same_macaddr = _rtw_memcmp(pnetwork->network.MacAddress, cur_network->network.MacAddress, ETH_ALEN);
2768 
2769 	pnetwork->network.Length = get_WLAN_BSSID_EX_sz(&pnetwork->network);
2770 	if (pnetwork->network.Length > sizeof(WLAN_BSSID_EX))
2771 		goto exit;
2772 
2773 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
2774 
2775 	pmlmepriv->LinkDetectInfo.TrafficTransitionCount = 0;
2776 	pmlmepriv->LinkDetectInfo.LowPowerTransitionCount = 0;
2777 
2778 
2779 	if (pnetwork->join_res > 0) {
2780 		_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2781 		retry = 0;
2782 		if (check_fwstate(pmlmepriv, WIFI_UNDER_LINKING)) {
2783 			/* s1. find ptarget_wlan */
2784 			if (check_fwstate(pmlmepriv, WIFI_ASOC_STATE)) {
2785 				if (the_same_macaddr == _TRUE)
2786 					ptarget_wlan = _rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
2787 				else {
2788 					pcur_wlan = _rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
2789 					if (pcur_wlan)
2790 						pcur_wlan->fixed = _FALSE;
2791 
2792 					pcur_sta = rtw_get_stainfo(pstapriv, cur_network->network.MacAddress);
2793 					if (pcur_sta) {
2794 						/* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL2); */
2795 						rtw_free_stainfo(adapter,  pcur_sta);
2796 						/* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL2); */
2797 					}
2798 
2799 					ptarget_wlan = _rtw_find_network(&pmlmepriv->scanned_queue, pnetwork->network.MacAddress);
2800 					if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2801 						if (ptarget_wlan)
2802 							ptarget_wlan->fixed = _TRUE;
2803 					}
2804 				}
2805 
2806 			} else {
2807 				ptarget_wlan = _rtw_find_same_network(&pmlmepriv->scanned_queue, pnetwork);
2808 				if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2809 					if (ptarget_wlan)
2810 						ptarget_wlan->fixed = _TRUE;
2811 				}
2812 			}
2813 
2814 			/* s2. update cur_network */
2815 			if (ptarget_wlan)
2816 				rtw_joinbss_update_network(adapter, ptarget_wlan, pnetwork);
2817 			else {
2818 				RTW_PRINT("Can't find ptarget_wlan when joinbss_event callback\n");
2819 				_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2820 				goto ignore_joinbss_callback;
2821 			}
2822 
2823 
2824 			/* s3. find ptarget_sta & update ptarget_sta after update cur_network only for station mode */
2825 			if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2826 				ptarget_sta = rtw_joinbss_update_stainfo(adapter, pnetwork);
2827 				if (ptarget_sta == NULL) {
2828 					RTW_ERR("Can't update stainfo when joinbss_event callback\n");
2829 					_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2830 					goto ignore_joinbss_callback;
2831 				}
2832 
2833 				/* Queue TX packets before FW/HW ready */
2834 				/* clear in mlmeext_joinbss_event_callback() */
2835 				rtw_xmit_queue_set(ptarget_sta);
2836 			}
2837 
2838 			/* s4. indicate connect			 */
2839 			if (MLME_IS_STA(adapter) || MLME_IS_ADHOC(adapter)) {
2840 				pmlmepriv->cur_network_scanned = ptarget_wlan;
2841 				rtw_indicate_connect(adapter);
2842 			}
2843 
2844 			/* s5. Cancle assoc_timer					 */
2845 			_cancel_timer_ex(&pmlmepriv->assoc_timer);
2846 
2847 
2848 		} else {
2849 			_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2850 			goto ignore_joinbss_callback;
2851 		}
2852 
2853 		_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2854 
2855 	} else if (pnetwork->join_res == -4) {
2856 		rtw_reset_securitypriv(adapter);
2857 		pmlmepriv->join_status = status;
2858 		_set_timer(&pmlmepriv->assoc_timer, 1);
2859 
2860 		/* rtw_free_assoc_resources(adapter, _TRUE); */
2861 
2862 		if ((check_fwstate(pmlmepriv, WIFI_UNDER_LINKING)) == _TRUE) {
2863 			_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING);
2864 		}
2865 
2866 	} else { /* if join_res < 0 (join fails), then try again */
2867 
2868 #ifdef REJOIN
2869 		res = _FAIL;
2870 		if (retry < 2) {
2871 			res = rtw_select_and_join_from_scanned_queue(pmlmepriv);
2872 		}
2873 
2874 		if (res == _SUCCESS) {
2875 			/* extend time of assoc_timer */
2876 			_set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
2877 			retry++;
2878 		} else if (res == 2) { /* there is no need to wait for join */
2879 			_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING);
2880 			rtw_indicate_connect(adapter);
2881 		} else {
2882 #endif
2883 			pmlmepriv->join_status = status;
2884 			_set_timer(&pmlmepriv->assoc_timer, 1);
2885 			/* rtw_free_assoc_resources(adapter, _TRUE); */
2886 			_clr_fwstate_(pmlmepriv, WIFI_UNDER_LINKING);
2887 
2888 #ifdef REJOIN
2889 			retry = 0;
2890 		}
2891 #endif
2892 	}
2893 
2894 ignore_joinbss_callback:
2895 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
2896 
2897 exit:
2898 	return;
2899 }
2900 
2901 void rtw_joinbss_event_callback(_adapter *adapter, u8 *pbuf)
2902 {
2903 	struct wlan_network	*pnetwork	= (struct wlan_network *)pbuf;
2904 
2905 
2906 	mlmeext_joinbss_event_callback(adapter, pnetwork->join_res);
2907 
2908 	rtw_mi_os_xmit_schedule(adapter);
2909 
2910 }
2911 
2912 void rtw_sta_media_status_rpt(_adapter *adapter, struct sta_info *sta, bool connected)
2913 {
2914 	struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
2915 	bool miracast_enabled = 0;
2916 	bool miracast_sink = 0;
2917 	u8 role = H2C_MSR_ROLE_RSVD;
2918 
2919 	if (sta == NULL) {
2920 		RTW_PRINT(FUNC_ADPT_FMT" sta is NULL\n"
2921 			  , FUNC_ADPT_ARG(adapter));
2922 		rtw_warn_on(1);
2923 		return;
2924 	}
2925 
2926 	if (sta->cmn.mac_id >= macid_ctl->num) {
2927 		RTW_PRINT(FUNC_ADPT_FMT" invalid macid:%u\n"
2928 			  , FUNC_ADPT_ARG(adapter), sta->cmn.mac_id);
2929 		rtw_warn_on(1);
2930 		return;
2931 	}
2932 
2933 	if (!rtw_macid_is_used(macid_ctl, sta->cmn.mac_id)) {
2934 		RTW_PRINT(FUNC_ADPT_FMT" macid:%u not is used, set connected to 0\n"
2935 			  , FUNC_ADPT_ARG(adapter), sta->cmn.mac_id);
2936 		connected = 0;
2937 		rtw_warn_on(1);
2938 	}
2939 
2940 	if (connected && !rtw_macid_is_bmc(macid_ctl, sta->cmn.mac_id)) {
2941 		miracast_enabled = STA_OP_WFD_MODE(sta) != 0 && is_miracast_enabled(adapter);
2942 		miracast_sink = miracast_enabled && (STA_OP_WFD_MODE(sta) & MIRACAST_SINK);
2943 
2944 #ifdef CONFIG_TDLS
2945 		if (sta->tdls_sta_state & TDLS_LINKED_STATE)
2946 			role = H2C_MSR_ROLE_TDLS;
2947 		else
2948 #endif
2949 		if (MLME_IS_STA(adapter)) {
2950 			if (MLME_IS_GC(adapter))
2951 				role = H2C_MSR_ROLE_GO;
2952 			else
2953 				role = H2C_MSR_ROLE_AP;
2954 		} else if (MLME_IS_AP(adapter)) {
2955 			if (MLME_IS_GO(adapter))
2956 				role = H2C_MSR_ROLE_GC;
2957 			else
2958 				role = H2C_MSR_ROLE_STA;
2959 		} else if (MLME_IS_ADHOC(adapter) || MLME_IS_ADHOC_MASTER(adapter))
2960 			role = H2C_MSR_ROLE_ADHOC;
2961 		else if (MLME_IS_MESH(adapter))
2962 			role = H2C_MSR_ROLE_MESH;
2963 
2964 #ifdef CONFIG_WFD
2965 		if (role == H2C_MSR_ROLE_GC
2966 			|| role == H2C_MSR_ROLE_GO
2967 			|| role == H2C_MSR_ROLE_TDLS
2968 		) {
2969 			if (adapter->wfd_info.rtsp_ctrlport
2970 				|| adapter->wfd_info.tdls_rtsp_ctrlport
2971 				|| adapter->wfd_info.peer_rtsp_ctrlport)
2972 				rtw_wfd_st_switch(sta, 1);
2973 		}
2974 #endif
2975 	}
2976 
2977 	rtw_hal_set_FwMediaStatusRpt_single_cmd(adapter
2978 		, connected
2979 		, miracast_enabled
2980 		, miracast_sink
2981 		, role
2982 		, sta->cmn.mac_id
2983 	);
2984 }
2985 
2986 u8 rtw_sta_media_status_rpt_cmd(_adapter *adapter, struct sta_info *sta, bool connected)
2987 {
2988 	struct cmd_priv	*cmdpriv = &adapter->cmdpriv;
2989 	struct cmd_obj *cmdobj;
2990 	struct drvextra_cmd_parm *cmd_parm;
2991 	struct sta_media_status_rpt_cmd_parm *rpt_parm;
2992 	u8	res = _SUCCESS;
2993 
2994 	cmdobj = (struct cmd_obj *)rtw_zmalloc(sizeof(struct cmd_obj));
2995 	if (cmdobj == NULL) {
2996 		res = _FAIL;
2997 		goto exit;
2998 	}
2999 
3000 	cmd_parm = (struct drvextra_cmd_parm *)rtw_zmalloc(sizeof(struct drvextra_cmd_parm));
3001 	if (cmd_parm == NULL) {
3002 		rtw_mfree((u8 *)cmdobj, sizeof(struct cmd_obj));
3003 		res = _FAIL;
3004 		goto exit;
3005 	}
3006 
3007 	rpt_parm = (struct sta_media_status_rpt_cmd_parm *)rtw_zmalloc(sizeof(struct sta_media_status_rpt_cmd_parm));
3008 	if (rpt_parm == NULL) {
3009 		rtw_mfree((u8 *)cmdobj, sizeof(struct cmd_obj));
3010 		rtw_mfree((u8 *)cmd_parm, sizeof(struct drvextra_cmd_parm));
3011 		res = _FAIL;
3012 		goto exit;
3013 	}
3014 
3015 	rpt_parm->sta = sta;
3016 	rpt_parm->connected = connected;
3017 
3018 	cmd_parm->ec_id = STA_MSTATUS_RPT_WK_CID;
3019 	cmd_parm->type = 0;
3020 	cmd_parm->size = sizeof(struct sta_media_status_rpt_cmd_parm);
3021 	cmd_parm->pbuf = (u8 *)rpt_parm;
3022 	init_h2fwcmd_w_parm_no_rsp(cmdobj, cmd_parm, CMD_SET_DRV_EXTRA);
3023 
3024 	res = rtw_enqueue_cmd(cmdpriv, cmdobj);
3025 
3026 exit:
3027 	return res;
3028 }
3029 
3030 inline void rtw_sta_media_status_rpt_cmd_hdl(_adapter *adapter, struct sta_media_status_rpt_cmd_parm *parm)
3031 {
3032 	rtw_sta_media_status_rpt(adapter, parm->sta, parm->connected);
3033 }
3034 
3035 void rtw_stassoc_event_callback(_adapter *adapter, u8 *pbuf)
3036 {
3037 	_irqL irqL;
3038 	struct sta_info *psta;
3039 	struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
3040 	struct stassoc_event	*pstassoc	= (struct stassoc_event *)pbuf;
3041 	struct wlan_network	*cur_network = &(pmlmepriv->cur_network);
3042 	struct wlan_network	*ptarget_wlan = NULL;
3043 
3044 
3045 #if CONFIG_RTW_MACADDR_ACL
3046 	if (rtw_access_ctrl(adapter, pstassoc->macaddr) == _FALSE)
3047 		return;
3048 #endif
3049 
3050 #if defined(CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME)
3051 	if (MLME_IS_AP(adapter) || MLME_IS_MESH(adapter)) {
3052 		psta = rtw_get_stainfo(&adapter->stapriv, pstassoc->macaddr);
3053 		if (psta) {
3054 			u8 *passoc_req = NULL;
3055 			u32 assoc_req_len = 0;
3056 
3057 			rtw_sta_media_status_rpt(adapter, psta, 1);
3058 
3059 #ifdef CONFIG_MCC_MODE
3060 			rtw_hal_mcc_update_macid_bitmap(adapter, psta->cmn.mac_id, _TRUE);
3061 #endif /* CONFIG_MCC_MODE */
3062 
3063 #ifndef CONFIG_AUTO_AP_MODE
3064 			ap_sta_info_defer_update(adapter, psta);
3065 
3066 			if (!MLME_IS_MESH(adapter)) {
3067 				/* report to upper layer */
3068 				RTW_INFO("indicate_sta_assoc_event to upper layer - hostapd\n");
3069 				#ifdef CONFIG_IOCTL_CFG80211
3070 				_enter_critical_bh(&psta->lock, &irqL);
3071 				if (psta->passoc_req && psta->assoc_req_len > 0) {
3072 					passoc_req = rtw_zmalloc(psta->assoc_req_len);
3073 					if (passoc_req) {
3074 						assoc_req_len = psta->assoc_req_len;
3075 						_rtw_memcpy(passoc_req, psta->passoc_req, assoc_req_len);
3076 					}
3077 				}
3078 				_exit_critical_bh(&psta->lock, &irqL);
3079 
3080 				if (passoc_req && assoc_req_len > 0) {
3081 					rtw_cfg80211_indicate_sta_assoc(adapter, passoc_req, assoc_req_len);
3082 					rtw_mfree(passoc_req, assoc_req_len);
3083 				}
3084 				#else /* !CONFIG_IOCTL_CFG80211	 */
3085 				rtw_indicate_sta_assoc_event(adapter, psta);
3086 				#endif /* !CONFIG_IOCTL_CFG80211 */
3087 			}
3088 #endif /* !CONFIG_AUTO_AP_MODE */
3089 
3090 #ifdef CONFIG_BEAMFORMING
3091 			beamforming_wk_cmd(adapter, BEAMFORMING_CTRL_ENTER, (u8 *)psta, sizeof(struct sta_info), 0);
3092 #endif/*CONFIG_BEAMFORMING*/
3093 			if (is_wep_enc(adapter->securitypriv.dot11PrivacyAlgrthm))
3094 				rtw_ap_wep_pk_setting(adapter, psta);
3095 
3096 			#ifdef CONFIG_PLATFORM_CMAP_INTFS
3097 			if (MLME_IS_AP(adapter)) {
3098 				cmap_intfs_nl_sta_event(psta->cmn.mac_addr, adapter_mac_addr(adapter), 1
3099 					, psta->passoc_req + IEEE80211_3ADDR_LEN, psta->assoc_req_len - IEEE80211_3ADDR_LEN);
3100 			}
3101 			#endif
3102 		}
3103 		goto exit;
3104 	}
3105 #endif /* defined (CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME) */
3106 
3107 	/* for AD-HOC mode */
3108 	psta = rtw_get_stainfo(&adapter->stapriv, pstassoc->macaddr);
3109 	if (psta == NULL) {
3110 		RTW_ERR(FUNC_ADPT_FMT" get no sta_info with "MAC_FMT"\n"
3111 			, FUNC_ADPT_ARG(adapter), MAC_ARG(pstassoc->macaddr));
3112 		rtw_warn_on(1);
3113 		goto exit;
3114 	}
3115 
3116 	rtw_sta_media_status_rpt(adapter, psta, 1);
3117 
3118 	if (adapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X)
3119 		psta->dot118021XPrivacy = adapter->securitypriv.dot11PrivacyAlgrthm;
3120 
3121 
3122 	psta->ieee8021x_blocked = _FALSE;
3123 
3124 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
3125 
3126 	if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) == _TRUE) ||
3127 	    (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE)) {
3128 		if (adapter->stapriv.asoc_sta_count == 2) {
3129 			_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3130 			ptarget_wlan = _rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
3131 			pmlmepriv->cur_network_scanned = ptarget_wlan;
3132 			if (ptarget_wlan)
3133 				ptarget_wlan->fixed = _TRUE;
3134 			_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3135 			/* a sta + bc/mc_stainfo (not Ibss_stainfo) */
3136 			rtw_indicate_connect(adapter);
3137 		}
3138 	}
3139 
3140 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
3141 
3142 
3143 	mlmeext_sta_add_event_callback(adapter, psta);
3144 
3145 #ifdef CONFIG_RTL8711
3146 	/* submit SetStaKey_cmd to tell fw, fw will allocate an CAM entry for this sta	 */
3147 	rtw_setstakey_cmd(adapter, psta, GROUP_KEY, _TRUE);
3148 #endif
3149 
3150 exit:
3151 #ifdef CONFIG_RTS_FULL_BW
3152 	rtw_set_rts_bw(adapter);
3153 #endif/*CONFIG_RTS_FULL_BW*/
3154 	return;
3155 }
3156 
3157 #ifdef CONFIG_IEEE80211W
3158 void rtw_sta_timeout_event_callback(_adapter *adapter, u8 *pbuf)
3159 {
3160 #ifdef CONFIG_AP_MODE
3161 	_irqL irqL;
3162 	struct sta_info *psta;
3163 	struct stadel_event *pstadel = (struct stadel_event *)pbuf;
3164 	struct sta_priv *pstapriv = &adapter->stapriv;
3165 
3166 	psta = rtw_get_stainfo(&adapter->stapriv, pstadel->macaddr);
3167 
3168 	if (psta) {
3169 		u8 updated = _FALSE;
3170 
3171 		_enter_critical_bh(&pstapriv->asoc_list_lock, &irqL);
3172 		if (rtw_is_list_empty(&psta->asoc_list) == _FALSE) {
3173 			rtw_list_delete(&psta->asoc_list);
3174 			pstapriv->asoc_list_cnt--;
3175 			#ifdef CONFIG_RTW_TOKEN_BASED_XMIT
3176 			if (psta->tbtx_enable)
3177 				pstapriv->tbtx_asoc_list_cnt--;
3178 			#endif
3179 			updated = ap_free_sta(adapter, psta, _TRUE, WLAN_REASON_PREV_AUTH_NOT_VALID, _TRUE);
3180 		}
3181 		_exit_critical_bh(&pstapriv->asoc_list_lock, &irqL);
3182 
3183 		associated_clients_update(adapter, updated, STA_INFO_UPDATE_ALL);
3184 	}
3185 #endif /* CONFIG_AP_MODE */
3186 }
3187 #endif /* CONFIG_IEEE80211W */
3188 
3189 void rtw_sta_mstatus_disc_rpt(_adapter *adapter, u8 mac_id)
3190 {
3191 	struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
3192 
3193 	if (mac_id < macid_ctl->num) {
3194 		u8 id_is_shared = mac_id == RTW_DEFAULT_MGMT_MACID; /* TODO: real shared macid judgment */
3195 
3196 		RTW_INFO(FUNC_ADPT_FMT" - mac_id=%d%s\n", FUNC_ADPT_ARG(adapter)
3197 			, mac_id, id_is_shared ? " shared" : "");
3198 
3199 		if (!id_is_shared) {
3200 			rtw_hal_macid_drop(adapter, mac_id);
3201 			rtw_hal_set_FwMediaStatusRpt_single_cmd(adapter, 0, 0, 0, 0, mac_id);
3202 			/*
3203 			 * For safety, prevent from keeping macid sleep.
3204 			 * If we can sure all power mode enter/leave are paired,
3205 			 * this check can be removed.
3206 			 * Lucas@20131113
3207 			 */
3208 			/* wakeup macid after disconnect. */
3209 			/*if (MLME_IS_STA(adapter))*/
3210 			rtw_hal_macid_wakeup(adapter, mac_id);
3211 		}
3212 	} else {
3213 		RTW_PRINT(FUNC_ADPT_FMT" invalid macid:%u\n"
3214 			  , FUNC_ADPT_ARG(adapter), mac_id);
3215 		rtw_warn_on(1);
3216 	}
3217 }
3218 void rtw_sta_mstatus_report(_adapter *adapter)
3219 {
3220 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
3221 	struct wlan_network *tgt_network = &pmlmepriv->cur_network;
3222 	struct sta_info *psta = NULL;
3223 
3224 	if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) && check_fwstate(pmlmepriv, WIFI_ASOC_STATE)) {
3225 		psta = rtw_get_stainfo(&adapter->stapriv, tgt_network->network.MacAddress);
3226 		if (psta)
3227 			rtw_sta_mstatus_disc_rpt(adapter, psta->cmn.mac_id);
3228 		else {
3229 			RTW_INFO("%s "ADPT_FMT" - mac_addr: "MAC_FMT" psta == NULL\n", __func__, ADPT_ARG(adapter), MAC_ARG(tgt_network->network.MacAddress));
3230 			rtw_warn_on(1);
3231 		}
3232 	}
3233 }
3234 
3235 void rtw_stadel_event_callback(_adapter *adapter, u8 *pbuf)
3236 {
3237 	_irqL irqL, irqL2;
3238 
3239 	struct sta_info *psta;
3240 	struct wlan_network *pwlan = NULL;
3241 	WLAN_BSSID_EX    *pdev_network = NULL;
3242 	u8 *pibss = NULL;
3243 	struct	mlme_priv	*pmlmepriv = &(adapter->mlmepriv);
3244 	struct	stadel_event *pstadel	= (struct stadel_event *)pbuf;
3245 	struct wlan_network *tgt_network = &(pmlmepriv->cur_network);
3246 
3247 	RTW_INFO("%s(mac_id=%d)=" MAC_FMT "\n", __func__, pstadel->mac_id, MAC_ARG(pstadel->macaddr));
3248 #ifdef CONFIG_LAYER2_ROAMING
3249 	if (pmlmepriv->roam_network)
3250 		rtw_msleep_os(RTW_ROAM_DICONNECT_DELAY);
3251 #endif
3252 	rtw_sta_mstatus_disc_rpt(adapter, pstadel->mac_id);
3253 
3254 #ifdef CONFIG_MCC_MODE
3255 	rtw_hal_mcc_update_macid_bitmap(adapter, pstadel->mac_id, _FALSE);
3256 #endif /* CONFIG_MCC_MODE */
3257 
3258 	psta = rtw_get_stainfo(&adapter->stapriv, pstadel->macaddr);
3259 
3260 	if (psta == NULL) {
3261 		RTW_INFO("%s(mac_id=%d)=" MAC_FMT " psta == NULL\n", __func__, pstadel->mac_id, MAC_ARG(pstadel->macaddr));
3262 		/*rtw_warn_on(1);*/
3263 	}
3264 
3265 	if (psta)
3266 		rtw_wfd_st_switch(psta, 0);
3267 
3268 	if (MLME_IS_MESH(adapter)) {
3269 		rtw_free_stainfo(adapter, psta);
3270 		goto exit;
3271 	}
3272 
3273 #ifdef CONFIG_AP_MODE
3274 	if (MLME_IS_AP(adapter)) {
3275 #ifdef CONFIG_IOCTL_CFG80211
3276 #ifdef COMPAT_KERNEL_RELEASE
3277 
3278 #elif (LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 37)) || defined(CONFIG_CFG80211_FORCE_COMPATIBLE_2_6_37_UNDER)
3279 		rtw_cfg80211_indicate_sta_disassoc(adapter, pstadel->macaddr, *(u16 *)pstadel->rsvd);
3280 #endif /* (LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 37)) || defined(CONFIG_CFG80211_FORCE_COMPATIBLE_2_6_37_UNDER) */
3281 #endif /* CONFIG_IOCTL_CFG80211 */
3282 
3283 		#ifdef CONFIG_PLATFORM_CMAP_INTFS
3284 		cmap_intfs_nl_sta_event(pstadel->macaddr, adapter_mac_addr(adapter), 0, NULL, 0);
3285 		#endif
3286 
3287 		rtw_free_stainfo(adapter, psta);
3288 
3289 		goto exit;
3290 	}
3291 #endif /* CONFIG_AP_MODE */
3292 
3293 	mlmeext_sta_del_event_callback(adapter);
3294 
3295 	_enter_critical_bh(&pmlmepriv->lock, &irqL2);
3296 
3297 	if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
3298 		u16 reason = *((unsigned short *)(pstadel->rsvd));
3299 		bool roam = _FALSE;
3300 		struct wlan_network *roam_target = NULL;
3301 
3302 #ifdef CONFIG_LAYER2_ROAMING
3303 #ifdef CONFIG_RTW_80211R
3304 		if (rtw_ft_roam_expired(adapter, reason))
3305 			pmlmepriv->ft_roam.ft_roam_on_expired = _TRUE;
3306 		else
3307 			pmlmepriv->ft_roam.ft_roam_on_expired = _FALSE;
3308 #endif
3309 		if (adapter->registrypriv.wifi_spec == 1)
3310 			roam = _FALSE;
3311 		else if (reason == WLAN_REASON_EXPIRATION_CHK && rtw_chk_roam_flags(adapter, RTW_ROAM_ON_EXPIRED))
3312 			roam = _TRUE;
3313 		else if (reason == WLAN_REASON_ACTIVE_ROAM && rtw_chk_roam_flags(adapter, RTW_ROAM_ACTIVE)) {
3314 			roam = _TRUE;
3315 			roam_target = pmlmepriv->roam_network;
3316 		}
3317 
3318 #ifdef CONFIG_RTW_80211R
3319 		if (reason == WLAN_REASON_ACTIVE_ROAM && rtw_ft_chk_flags(adapter, RTW_FT_BTM_ROAM)) {
3320 			roam = _TRUE;
3321 			roam_target = pmlmepriv->roam_network;
3322 		}
3323 #endif
3324 
3325 		if (roam == _TRUE) {
3326 			if (rtw_to_roam(adapter) > 0)
3327 				rtw_dec_to_roam(adapter); /* this stadel_event is caused by roaming, decrease to_roam */
3328 			else if (rtw_to_roam(adapter) == 0)
3329 				rtw_set_to_roam(adapter, adapter->registrypriv.max_roaming_times);
3330 		} else
3331 			rtw_set_to_roam(adapter, 0);
3332 #endif /* CONFIG_LAYER2_ROAMING */
3333 
3334 		rtw_free_uc_swdec_pending_queue(adapter);
3335 
3336 		rtw_free_assoc_resources(adapter, _TRUE);
3337 		rtw_free_mlme_priv_ie_data(pmlmepriv);
3338 
3339 		rtw_indicate_disconnect(adapter, *(u16 *)pstadel->rsvd, pstadel->locally_generated);
3340 
3341 #ifdef CONFIG_LAYER2_ROAMING
3342 		_rtw_roaming(adapter, roam_target);
3343 #endif
3344 	}
3345 
3346 #ifdef CONFIG_AP_MODE
3347 	if (check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) ||
3348 	    check_fwstate(pmlmepriv, WIFI_ADHOC_STATE)) {
3349 
3350 		/* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
3351 		rtw_free_stainfo(adapter,  psta);
3352 		/* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
3353 
3354 		if (adapter->stapriv.asoc_sta_count == 1) { /* a sta + bc/mc_stainfo (not Ibss_stainfo) */
3355 			/* rtw_indicate_disconnect(adapter); */ /* removed@20091105 */
3356 			_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3357 			/* free old ibss network */
3358 			/* pwlan = _rtw_find_network(&pmlmepriv->scanned_queue, pstadel->macaddr); */
3359 			pwlan = _rtw_find_network(&pmlmepriv->scanned_queue, tgt_network->network.MacAddress);
3360 			if (pwlan) {
3361 				pwlan->fixed = _FALSE;
3362 				rtw_free_network_nolock(adapter, pwlan);
3363 			}
3364 			_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3365 			/* re-create ibss */
3366 			pdev_network = &(adapter->registrypriv.dev_network);
3367 			pibss = adapter->registrypriv.dev_network.MacAddress;
3368 
3369 			_rtw_memcpy(pdev_network, &tgt_network->network, get_WLAN_BSSID_EX_sz(&tgt_network->network));
3370 
3371 			_rtw_memset(&pdev_network->Ssid, 0, sizeof(NDIS_802_11_SSID));
3372 			_rtw_memcpy(&pdev_network->Ssid, &pmlmepriv->assoc_ssid, sizeof(NDIS_802_11_SSID));
3373 
3374 			rtw_update_registrypriv_dev_network(adapter);
3375 
3376 			rtw_generate_random_ibss(pibss);
3377 
3378 			if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE)) {
3379 				set_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE);
3380 				_clr_fwstate_(pmlmepriv, WIFI_ADHOC_STATE);
3381 			}
3382 
3383 			if (rtw_create_ibss_cmd(adapter, 0) != _SUCCESS)
3384 				RTW_ERR("rtw_create_ibss_cmd FAIL\n");
3385 
3386 		}
3387 
3388 	}
3389 #endif /* CONFIG_AP_MODE */
3390 	_exit_critical_bh(&pmlmepriv->lock, &irqL2);
3391 exit:
3392 	#ifdef CONFIG_RTS_FULL_BW
3393 	rtw_set_rts_bw(adapter);
3394 	#endif/*CONFIG_RTS_FULL_BW*/
3395 	return;
3396 }
3397 
3398 void rtw_wmm_event_callback(PADAPTER padapter, u8 *pbuf)
3399 {
3400 
3401 	WMMOnAssocRsp(padapter);
3402 
3403 
3404 }
3405 
3406 /*
3407 * rtw_join_timeout_handler - Timeout/failure handler for CMD JoinBss
3408 */
3409 void rtw_join_timeout_handler(void *ctx)
3410 {
3411 	_adapter *adapter = (_adapter *)ctx;
3412 	_irqL irqL;
3413 	struct	mlme_priv *pmlmepriv = &adapter->mlmepriv;
3414 
3415 #if 0
3416 	if (rtw_is_drv_stopped(adapter)) {
3417 		_rtw_up_sema(&pmlmepriv->assoc_terminate);
3418 		return;
3419 	}
3420 #endif
3421 
3422 
3423 
3424 	RTW_INFO("%s, fw_state=%x\n", __FUNCTION__, get_fwstate(pmlmepriv));
3425 
3426 	if (RTW_CANNOT_RUN(adapter))
3427 		return;
3428 
3429 
3430 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
3431 
3432 #ifdef CONFIG_LAYER2_ROAMING
3433 	if (rtw_to_roam(adapter) > 0) { /* join timeout caused by roaming */
3434 		while (1) {
3435 			rtw_dec_to_roam(adapter);
3436 			if (rtw_to_roam(adapter) != 0) { /* try another */
3437 				int do_join_r;
3438 				RTW_INFO("%s try another roaming\n", __FUNCTION__);
3439 				do_join_r = rtw_do_join(adapter);
3440 				if (_SUCCESS != do_join_r) {
3441 					RTW_INFO("%s roaming do_join return %d\n", __FUNCTION__ , do_join_r);
3442 					continue;
3443 				}
3444 				break;
3445 			} else {
3446 				RTW_INFO("%s We've try roaming but fail\n", __FUNCTION__);
3447 #ifdef CONFIG_RTW_80211R
3448 				rtw_ft_clr_flags(adapter, RTW_FT_PEER_EN|RTW_FT_PEER_OTD_EN);
3449 				rtw_ft_reset_status(adapter);
3450 #endif
3451 				rtw_indicate_disconnect(adapter, pmlmepriv->join_status, _FALSE);
3452 				break;
3453 			}
3454 		}
3455 
3456 	} else
3457 #endif
3458 	{
3459 		rtw_indicate_disconnect(adapter, pmlmepriv->join_status, _FALSE);
3460 		free_scanqueue(pmlmepriv);/* ??? */
3461 
3462 #ifdef CONFIG_IOCTL_CFG80211
3463 		/* indicate disconnect for the case that join_timeout and check_fwstate != FW_LINKED */
3464 		rtw_cfg80211_indicate_disconnect(adapter, pmlmepriv->join_status, _FALSE);
3465 #endif /* CONFIG_IOCTL_CFG80211 */
3466 
3467 	}
3468 
3469 	pmlmepriv->join_status = 0; /* reset */
3470 
3471 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
3472 
3473 
3474 #ifdef CONFIG_DRVEXT_MODULE_WSC
3475 	drvext_assoc_fail_indicate(&adapter->drvextpriv);
3476 #endif
3477 
3478 
3479 
3480 }
3481 
3482 #ifdef CONFIG_RTL8822CS_WIFI_HDF
3483 extern void indicate_wx_scan_get_result_event(_adapter *padapter);
3484 #endif
3485 
3486 /*
3487 * rtw_scan_timeout_handler - Timeout/Faliure handler for CMD SiteSurvey
3488 * @adapter: pointer to _adapter structure
3489 */
3490 void rtw_scan_timeout_handler(void *ctx)
3491 {
3492 	_adapter *adapter = (_adapter *)ctx;
3493 	_irqL irqL;
3494 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
3495 	RTW_INFO(FUNC_ADPT_FMT" fw_state=%x\n", FUNC_ADPT_ARG(adapter), get_fwstate(pmlmepriv));
3496 
3497 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
3498 
3499 	_clr_fwstate_(pmlmepriv, WIFI_UNDER_SURVEY);
3500 
3501 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
3502 
3503 #ifdef CONFIG_RTL8822CS_WIFI_HDF
3504 	indicate_wx_scan_get_result_event(adapter);
3505 
3506 	HdfWifiEventScanDone(get_rtl_netdev(), 1);
3507 #else
3508 #ifdef CONFIG_IOCTL_CFG80211
3509 	rtw_cfg80211_surveydone_event_callback(adapter);
3510 #endif /* CONFIG_IOCTL_CFG80211 */
3511 
3512 	rtw_indicate_scan_done(adapter, _TRUE);
3513 #endif
3514 
3515 #if defined(CONFIG_CONCURRENT_MODE) && defined(CONFIG_IOCTL_CFG80211)
3516 	rtw_cfg80211_indicate_scan_done_for_buddy(adapter, _TRUE);
3517 #endif
3518 }
3519 
3520 void rtw_mlme_reset_auto_scan_int(_adapter *adapter, u8 *reason)
3521 {
3522 #if defined(CONFIG_RTW_MESH) && defined(CONFIG_DFS_MASTER)
3523 #if CONFIG_RTW_MESH_OFFCH_CAND
3524 	struct rf_ctl_t *rfctl = adapter_to_rfctl(adapter);
3525 #endif
3526 #endif
3527 	u8 u_ch;
3528 	u32 interval_ms = 0xffffffff; /* 0xffffffff: special value to make min() works well, also means no auto scan */
3529 
3530 	*reason = RTW_AUTO_SCAN_REASON_UNSPECIFIED;
3531 	rtw_mi_get_ch_setting_union(adapter, &u_ch, NULL, NULL);
3532 
3533 	if (hal_chk_bw_cap(adapter, BW_CAP_40M)
3534 		&& is_client_associated_to_ap(adapter) == _TRUE
3535 		&& u_ch >= 1 && u_ch <= 14
3536 		&& adapter->registrypriv.wifi_spec
3537 		/* TODO: AP Connected is 40MHz capability? */
3538 	) {
3539 		interval_ms = rtw_min(interval_ms, 60 * 1000);
3540 		*reason |= RTW_AUTO_SCAN_REASON_2040_BSS;
3541 	}
3542 
3543 #ifdef CONFIG_RTW_MESH
3544 	#if CONFIG_RTW_MESH_OFFCH_CAND
3545 	if (adapter->mesh_cfg.peer_sel_policy.offch_find_int_ms
3546 		&& rtw_mesh_offch_candidate_accepted(adapter)
3547 		#ifdef CONFIG_DFS_MASTER
3548 		&& (!rfctl->radar_detect_ch || (IS_CH_WAITING(rfctl) && !IS_UNDER_CAC(rfctl)))
3549 		#endif
3550 	) {
3551 		interval_ms = rtw_min(interval_ms, adapter->mesh_cfg.peer_sel_policy.offch_find_int_ms);
3552 		*reason |= RTW_AUTO_SCAN_REASON_MESH_OFFCH_CAND;
3553 	}
3554 	#endif
3555 #endif /* CONFIG_RTW_MESH */
3556 
3557 	if (interval_ms == 0xffffffff)
3558 		interval_ms = 0;
3559 
3560 	rtw_mlme_set_auto_scan_int(adapter, interval_ms);
3561 	return;
3562 }
3563 
3564 void rtw_drv_scan_by_self(_adapter *padapter, u8 reason)
3565 {
3566 	struct sitesurvey_parm parm;
3567 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
3568 	int i;
3569 #if 1
3570 	u8 ssc_chk;
3571 
3572 	ssc_chk = rtw_sitesurvey_condition_check(padapter, _FALSE);
3573 	if( ssc_chk == SS_DENY_BUSY_TRAFFIC) {
3574 		#ifdef CONFIG_LAYER2_ROAMING
3575 		if (rtw_chk_roam_flags(padapter, RTW_ROAM_ACTIVE) && pmlmepriv->need_to_roam == _TRUE)
3576 			RTW_INFO(FUNC_ADPT_FMT" need to roam, don't care BusyTraffic\n", FUNC_ADPT_ARG(padapter));
3577 		else
3578 		#endif
3579 		{
3580 			RTW_INFO(FUNC_ADPT_FMT" exit BusyTraffic\n", FUNC_ADPT_ARG(padapter));
3581 			goto exit;
3582 		}
3583 	}
3584 	else if (ssc_chk != SS_ALLOW)
3585 		goto exit;
3586 
3587 	if (!rtw_is_adapter_up(padapter))
3588 		goto exit;
3589 #else
3590 	if (rtw_is_scan_deny(padapter))
3591 		goto exit;
3592 
3593 	if (!rtw_is_adapter_up(padapter))
3594 		goto exit;
3595 
3596 	if (rtw_mi_busy_traffic_check(padapter)) {
3597 #ifdef CONFIG_LAYER2_ROAMING
3598 		if (rtw_chk_roam_flags(padapter, RTW_ROAM_ACTIVE) && pmlmepriv->need_to_roam == _TRUE) {
3599 			RTW_INFO("need to roam, don't care BusyTraffic\n");
3600 		} else
3601 #endif
3602 		{
3603 			RTW_INFO(FUNC_ADPT_FMT" exit BusyTraffic\n", FUNC_ADPT_ARG(padapter));
3604 			goto exit;
3605 		}
3606 	}
3607 	if (check_fwstate(pmlmepriv, WIFI_AP_STATE) && check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
3608 		RTW_INFO(FUNC_ADPT_FMT" WIFI_AP_STATE && WIFI_UNDER_WPS\n", FUNC_ADPT_ARG(padapter));
3609 		goto exit;
3610 	}
3611 	if (check_fwstate(pmlmepriv, (WIFI_UNDER_SURVEY | WIFI_UNDER_LINKING)) == _TRUE) {
3612 		RTW_INFO(FUNC_ADPT_FMT" WIFI_UNDER_SURVEY|WIFI_UNDER_LINKING\n", FUNC_ADPT_ARG(padapter));
3613 		goto exit;
3614 	}
3615 
3616 #ifdef CONFIG_CONCURRENT_MODE
3617 	if (rtw_mi_buddy_check_fwstate(padapter, (WIFI_UNDER_SURVEY | WIFI_UNDER_LINKING | WIFI_UNDER_WPS))) {
3618 		RTW_INFO(FUNC_ADPT_FMT", but buddy_intf is under scanning or linking or wps_phase\n", FUNC_ADPT_ARG(padapter));
3619 		goto exit;
3620 	}
3621 #endif
3622 #endif
3623 
3624 	RTW_INFO(FUNC_ADPT_FMT" reason:0x%02x\n", FUNC_ADPT_ARG(padapter), reason);
3625 
3626 	/* only for 20/40 BSS */
3627 	if (reason == RTW_AUTO_SCAN_REASON_2040_BSS) {
3628 		rtw_init_sitesurvey_parm(padapter, &parm);
3629 		for (i=0;i<14;i++) {
3630 			parm.ch[i].hw_value = i + 1;
3631 			parm.ch[i].flags = RTW_IEEE80211_CHAN_PASSIVE_SCAN;
3632 		}
3633 		parm.ch_num = 14;
3634 		rtw_set_802_11_bssid_list_scan(padapter, &parm);
3635 		goto exit;
3636 	}
3637 
3638 #if defined(CONFIG_RTW_WNM) || defined(CONFIG_RTW_80211K)
3639 	if ((reason == RTW_AUTO_SCAN_REASON_ROAM)
3640 		&& (rtw_roam_nb_scan_list_set(padapter, &parm)))
3641 		goto exit;
3642 #endif
3643 
3644 	rtw_init_sitesurvey_parm(padapter, &parm);
3645 	parm.reason = reason;
3646 	rtw_set_802_11_bssid_list_scan(padapter, &parm);
3647 exit:
3648 	return;
3649 }
3650 
3651 static void rtw_auto_scan_handler(_adapter *padapter)
3652 {
3653 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
3654 	u8 reason = RTW_AUTO_SCAN_REASON_UNSPECIFIED;
3655 
3656 	rtw_mlme_reset_auto_scan_int(padapter, &reason);
3657 
3658 #ifdef CONFIG_P2P
3659 	if (!rtw_p2p_chk_state(&padapter->wdinfo, P2P_STATE_NONE))
3660 		goto exit;
3661 #endif
3662 
3663 #ifdef CONFIG_TDLS
3664 	if (padapter->tdlsinfo.link_established == _TRUE)
3665 		goto exit;
3666 #endif
3667 
3668 	if (pmlmepriv->auto_scan_int_ms == 0
3669 	    || rtw_get_passing_time_ms(pmlmepriv->scan_start_time) < pmlmepriv->auto_scan_int_ms)
3670 		goto exit;
3671 
3672 	rtw_drv_scan_by_self(padapter, reason);
3673 
3674 exit:
3675 	return;
3676 }
3677 static u8 is_drv_in_lps(_adapter *adapter)
3678 {
3679 	u8 is_in_lps = _FALSE;
3680 
3681 	#ifdef CONFIG_LPS_LCLK_WD_TIMER /* to avoid leaving lps 32k frequently*/
3682 	if ((adapter_to_pwrctl(adapter)->bFwCurrentInPSMode == _TRUE)
3683 	#ifdef CONFIG_BT_COEXIST
3684 		&& (rtw_btcoex_IsBtControlLps(adapter) == _FALSE)
3685 	#endif
3686 		)
3687 		is_in_lps = _TRUE;
3688 	#endif /* CONFIG_LPS_LCLK_WD_TIMER*/
3689 	return is_in_lps;
3690 }
3691 void rtw_iface_dynamic_check_timer_handlder(_adapter *adapter)
3692 {
3693 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
3694 
3695 	if (adapter->net_closed == _TRUE)
3696 		return;
3697 	#ifdef CONFIG_LPS_LCLK_WD_TIMER /* to avoid leaving lps 32k frequently*/
3698 	if (is_drv_in_lps(adapter)) {
3699 		u8 bEnterPS;
3700 
3701 		linked_status_chk(adapter, 1);
3702 
3703 		bEnterPS = traffic_status_watchdog(adapter, 1);
3704 		if (bEnterPS) {
3705 			/* rtw_lps_ctrl_wk_cmd(adapter, LPS_CTRL_ENTER, 0); */
3706 			rtw_hal_dm_watchdog_in_lps(adapter);
3707 		} else {
3708 			/* call rtw_lps_ctrl_wk_cmd(padapter, LPS_CTRL_LEAVE, 0) in traffic_status_watchdog() */
3709 		}
3710 	}
3711 	#endif /* CONFIG_LPS_LCLK_WD_TIMER	*/
3712 
3713 	/* auto site survey */
3714 	rtw_auto_scan_handler(adapter);
3715 
3716 #ifdef CONFIG_AP_MODE
3717 	if (MLME_IS_AP(adapter)|| MLME_IS_MESH(adapter)) {
3718 		#ifndef CONFIG_ACTIVE_KEEP_ALIVE_CHECK
3719 		expire_timeout_chk(adapter);
3720 		#endif /* !CONFIG_ACTIVE_KEEP_ALIVE_CHECK */
3721 
3722 		#ifdef CONFIG_BMC_TX_RATE_SELECT
3723 		rtw_update_bmc_sta_tx_rate(adapter);
3724 		#endif /*CONFIG_BMC_TX_RATE_SELECT*/
3725 	}
3726 #endif /*CONFIG_AP_MODE*/
3727 
3728 
3729 #ifdef CONFIG_BR_EXT
3730 if (!adapter_use_wds(adapter)) {
3731 #if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35))
3732 	rcu_read_lock();
3733 #endif /* (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35)) */
3734 
3735 #if (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35))
3736 	if (adapter->pnetdev->br_port
3737 #else	/* (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35)) */
3738 	if (rcu_dereference(adapter->pnetdev->rx_handler_data)
3739 #endif /* (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35)) */
3740 		&& (check_fwstate(pmlmepriv, WIFI_STATION_STATE | WIFI_ADHOC_STATE) == _TRUE)) {
3741 		/* expire NAT2.5 entry */
3742 		void nat25_db_expire(_adapter *priv);
3743 		nat25_db_expire(adapter);
3744 
3745 		if (adapter->pppoe_connection_in_progress > 0)
3746 			adapter->pppoe_connection_in_progress--;
3747 		/* due to rtw_dynamic_check_timer_handlder() is called every 2 seconds */
3748 		if (adapter->pppoe_connection_in_progress > 0)
3749 			adapter->pppoe_connection_in_progress--;
3750 	}
3751 
3752 #if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35))
3753 	rcu_read_unlock();
3754 #endif /* (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35)) */
3755 }
3756 #endif /* CONFIG_BR_EXT */
3757 
3758 }
3759 
3760 /*TP_avg(t) = (1/10) * TP_avg(t-1) + (9/10) * TP(t) MBps*/
3761 static void collect_sta_traffic_statistics(_adapter *adapter)
3762 {
3763 	struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
3764 	struct sta_info *sta;
3765 	u64 curr_tx_bytes = 0, curr_rx_bytes = 0;
3766 	u32 curr_tx_mbytes = 0, curr_rx_mbytes = 0;
3767 	int i;
3768 
3769 	for (i = 0; i < MACID_NUM_SW_LIMIT; i++) {
3770 		sta = macid_ctl->sta[i];
3771 		if (sta && !is_broadcast_mac_addr(sta->cmn.mac_addr)) {
3772 			if (sta->sta_stats.last_tx_bytes > sta->sta_stats.tx_bytes)
3773 				sta->sta_stats.last_tx_bytes =  sta->sta_stats.tx_bytes;
3774 			if (sta->sta_stats.last_rx_bytes > sta->sta_stats.rx_bytes)
3775 				sta->sta_stats.last_rx_bytes = sta->sta_stats.rx_bytes;
3776 			if (sta->sta_stats.last_rx_bc_bytes > sta->sta_stats.rx_bc_bytes)
3777 				sta->sta_stats.last_rx_bc_bytes = sta->sta_stats.rx_bc_bytes;
3778 			if (sta->sta_stats.last_rx_mc_bytes > sta->sta_stats.rx_mc_bytes)
3779 				sta->sta_stats.last_rx_mc_bytes = sta->sta_stats.rx_mc_bytes;
3780 
3781 			curr_tx_bytes = sta->sta_stats.tx_bytes - sta->sta_stats.last_tx_bytes;
3782 			curr_rx_bytes = sta->sta_stats.rx_bytes - sta->sta_stats.last_rx_bytes;
3783 			sta->sta_stats.tx_tp_kbits = (curr_tx_bytes * 8 / 2) >> 10;/*Kbps*/
3784 			sta->sta_stats.rx_tp_kbits = (curr_rx_bytes * 8 / 2) >> 10;/*Kbps*/
3785 
3786 			sta->sta_stats.smooth_tx_tp_kbits = (sta->sta_stats.smooth_tx_tp_kbits * 6 / 10) + (sta->sta_stats.tx_tp_kbits * 4 / 10);/*Kbps*/
3787 			sta->sta_stats.smooth_rx_tp_kbits = (sta->sta_stats.smooth_rx_tp_kbits * 6 / 10) + (sta->sta_stats.rx_tp_kbits * 4 / 10);/*Kbps*/
3788 
3789 			curr_tx_mbytes = (curr_tx_bytes / 2) >> 20;/*MBps*/
3790 			curr_rx_mbytes = (curr_rx_bytes / 2) >> 20;/*MBps*/
3791 
3792 			sta->cmn.tx_moving_average_tp =
3793 				(sta->cmn.tx_moving_average_tp / 10) + (curr_tx_mbytes * 9 / 10); /*MBps*/
3794 
3795 			sta->cmn.rx_moving_average_tp =
3796 				(sta->cmn.rx_moving_average_tp / 10) + (curr_rx_mbytes * 9 /10); /*MBps*/
3797 
3798 			rtw_collect_bcn_info(sta->padapter);
3799 
3800 			if (adapter->bsta_tp_dump)
3801 				dump_sta_traffic(RTW_DBGDUMP, adapter, sta);
3802 
3803 			sta->sta_stats.last_tx_bytes = sta->sta_stats.tx_bytes;
3804 			sta->sta_stats.last_rx_bytes = sta->sta_stats.rx_bytes;
3805 			sta->sta_stats.last_rx_bc_bytes = sta->sta_stats.rx_bc_bytes;
3806 			sta->sta_stats.last_rx_mc_bytes = sta->sta_stats.rx_mc_bytes;
3807 		}
3808 	}
3809 }
3810 
3811 void rtw_sta_traffic_info(void *sel, _adapter *adapter)
3812 {
3813 	struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
3814 	struct sta_info *sta;
3815 	int i;
3816 
3817 	for (i = 0; i < MACID_NUM_SW_LIMIT; i++) {
3818 		sta = macid_ctl->sta[i];
3819 		if (sta && !is_broadcast_mac_addr(sta->cmn.mac_addr))
3820 			dump_sta_traffic(sel, adapter, sta);
3821 	}
3822 }
3823 
3824 /*#define DBG_TRAFFIC_STATISTIC*/
3825 static void collect_traffic_statistics(_adapter *padapter)
3826 {
3827 	struct dvobj_priv	*pdvobjpriv = adapter_to_dvobj(padapter);
3828 
3829 	/*_rtw_memset(&pdvobjpriv->traffic_stat, 0, sizeof(struct rtw_traffic_statistics));*/
3830 
3831 	/* Tx bytes reset*/
3832 	pdvobjpriv->traffic_stat.tx_bytes = 0;
3833 	pdvobjpriv->traffic_stat.tx_pkts = 0;
3834 	pdvobjpriv->traffic_stat.tx_drop = 0;
3835 
3836 	/* Rx bytes reset*/
3837 	pdvobjpriv->traffic_stat.rx_bytes = 0;
3838 	pdvobjpriv->traffic_stat.rx_pkts = 0;
3839 	pdvobjpriv->traffic_stat.rx_drop = 0;
3840 
3841 	rtw_mi_traffic_statistics(padapter);
3842 
3843 	/* Calculate throughput in last interval */
3844 	pdvobjpriv->traffic_stat.cur_tx_bytes = pdvobjpriv->traffic_stat.tx_bytes - pdvobjpriv->traffic_stat.last_tx_bytes;
3845 	pdvobjpriv->traffic_stat.cur_rx_bytes = pdvobjpriv->traffic_stat.rx_bytes - pdvobjpriv->traffic_stat.last_rx_bytes;
3846 	pdvobjpriv->traffic_stat.last_tx_bytes = pdvobjpriv->traffic_stat.tx_bytes;
3847 	pdvobjpriv->traffic_stat.last_rx_bytes = pdvobjpriv->traffic_stat.rx_bytes;
3848 
3849 	pdvobjpriv->traffic_stat.cur_tx_tp = (u32)(pdvobjpriv->traffic_stat.cur_tx_bytes * 8 / 2 / 1024 / 1024);/*Mbps*/
3850 	pdvobjpriv->traffic_stat.cur_rx_tp = (u32)(pdvobjpriv->traffic_stat.cur_rx_bytes * 8 / 2 / 1024 / 1024);/*Mbps*/
3851 
3852 	#ifdef DBG_TRAFFIC_STATISTIC
3853 	RTW_INFO("\n========================\n");
3854 	RTW_INFO("cur_tx_bytes:%lld\n", pdvobjpriv->traffic_stat.cur_tx_bytes);
3855 	RTW_INFO("cur_rx_bytes:%lld\n", pdvobjpriv->traffic_stat.cur_rx_bytes);
3856 
3857 	RTW_INFO("last_tx_bytes:%lld\n", pdvobjpriv->traffic_stat.last_tx_bytes);
3858 	RTW_INFO("last_rx_bytes:%lld\n", pdvobjpriv->traffic_stat.last_rx_bytes);
3859 
3860 	RTW_INFO("cur_tx_tp:%d (Mbps)\n", pdvobjpriv->traffic_stat.cur_tx_tp);
3861 	RTW_INFO("cur_rx_tp:%d (Mbps)\n", pdvobjpriv->traffic_stat.cur_rx_tp);
3862 	#endif
3863 
3864 #ifdef CONFIG_RTW_NAPI
3865 #ifdef CONFIG_RTW_NAPI_DYNAMIC
3866 	dynamic_napi_th_chk (padapter);
3867 #endif /* CONFIG_RTW_NAPI_DYNAMIC */
3868 #endif
3869 
3870 }
3871 
3872 void rtw_dynamic_check_timer_handlder(void *ctx)
3873 {
3874 	struct dvobj_priv *pdvobj = (struct dvobj_priv *)ctx;
3875 	_adapter *adapter = dvobj_get_primary_adapter(pdvobj);
3876 
3877 	if (!adapter)
3878 		goto exit;
3879 
3880 #if (MP_DRIVER == 1)
3881 	if (adapter->registrypriv.mp_mode == 1 && adapter->mppriv.mp_dm == 0) { /* for MP ODM dynamic Tx power tracking */
3882 		/* RTW_INFO("%s mp_dm =0 return\n", __func__); */
3883 		goto exit;
3884 	}
3885 #endif
3886 
3887 	if (!rtw_is_hw_init_completed(adapter))
3888 		goto exit;
3889 
3890 	if (RTW_CANNOT_RUN(adapter))
3891 		goto exit;
3892 
3893 	collect_traffic_statistics(adapter);
3894 	collect_sta_traffic_statistics(adapter);
3895 	rtw_mi_dynamic_check_timer_handlder(adapter);
3896 
3897 	if (!is_drv_in_lps(adapter))
3898 		rtw_dynamic_chk_wk_cmd(adapter);
3899 
3900 exit:
3901 	_set_timer(&pdvobj->dynamic_chk_timer, 2000);
3902 }
3903 
3904 
3905 #ifdef CONFIG_SET_SCAN_DENY_TIMER
3906 inline bool rtw_is_scan_deny(_adapter *adapter)
3907 {
3908 	struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3909 	return (ATOMIC_READ(&mlmepriv->set_scan_deny) != 0) ? _TRUE : _FALSE;
3910 }
3911 
3912 inline void rtw_clear_scan_deny(_adapter *adapter)
3913 {
3914 	struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3915 	ATOMIC_SET(&mlmepriv->set_scan_deny, 0);
3916 	if (0)
3917 		RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(adapter));
3918 }
3919 
3920 void rtw_set_scan_deny_timer_hdl(void *ctx)
3921 {
3922 	_adapter *adapter = (_adapter *)ctx;
3923 
3924 	rtw_clear_scan_deny(adapter);
3925 }
3926 void rtw_set_scan_deny(_adapter *adapter, u32 ms)
3927 {
3928 	struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3929 	if (0)
3930 		RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(adapter));
3931 	ATOMIC_SET(&mlmepriv->set_scan_deny, 1);
3932 	_set_timer(&mlmepriv->set_scan_deny_timer, ms);
3933 }
3934 #endif
3935 
3936 #ifdef CONFIG_LAYER2_ROAMING
3937 /*
3938 * Select a new roaming candidate from the original @param candidate and @param competitor
3939 * @return _TRUE: candidate is updated
3940 * @return _FALSE: candidate is not updated
3941 */
3942 static int rtw_check_roaming_candidate(struct mlme_priv *mlme
3943 	, struct wlan_network **candidate, struct wlan_network *competitor)
3944 {
3945 	int updated = _FALSE;
3946 	_adapter *adapter = container_of(mlme, _adapter, mlmepriv);
3947 	struct rf_ctl_t *rfctl = adapter_to_rfctl(adapter);
3948 	RT_CHANNEL_INFO *chset = rfctl->channel_set;
3949 	u8 ch = competitor->network.Configuration.DSConfig;
3950 
3951 	if (rtw_chset_search_ch(chset, ch) < 0)
3952 		goto exit;
3953 	if (IS_DFS_SLAVE_WITH_RD(rfctl)
3954 		&& !rtw_rfctl_dfs_domain_unknown(rfctl)
3955 		&& rtw_chset_is_ch_non_ocp(chset, ch))
3956 		goto exit;
3957 
3958 #if defined(CONFIG_RTW_REPEATER_SON) &&  (!defined(CONFIG_RTW_REPEATER_SON_ROOT))
3959 	if (rtw_rson_isupdate_roamcan(mlme, candidate, competitor))
3960 		goto  update;
3961 	goto exit;
3962 #endif
3963 
3964 	if (is_same_ess(&competitor->network, &mlme->cur_network.network) == _FALSE)
3965 		goto exit;
3966 
3967 	if (rtw_is_desired_network(adapter, competitor) == _FALSE)
3968 		goto exit;
3969 
3970 #ifdef CONFIG_RTW_ROAM_QUICKSCAN
3971 	if (competitor->network.PhyInfo.SignalStrength > CONFIG_RTW_ROAM_QUICKSCAN_TH)
3972 		adapter->mlmeextpriv.quickscan_next = _TRUE;
3973 #endif
3974 
3975 #ifdef CONFIG_LAYER2_ROAMING
3976 	if (mlme->need_to_roam == _FALSE)
3977 		goto exit;
3978 #endif
3979 
3980 	RTW_INFO("roam candidate:%s %s("MAC_FMT", ch%3u) rssi:%d dBm, age:%5d\n",
3981 		 (competitor == mlme->cur_network_scanned) ? "*" : " " ,
3982 		 competitor->network.Ssid.Ssid,
3983 		 MAC_ARG(competitor->network.MacAddress),
3984 		 competitor->network.Configuration.DSConfig,
3985 		 (int)competitor->network.Rssi,
3986 		 rtw_get_passing_time_ms(competitor->last_scanned)
3987 		);
3988 
3989 	/* got specific addr to roam */
3990 	if (!is_zero_mac_addr(mlme->roam_tgt_addr)) {
3991 		if (_rtw_memcmp(mlme->roam_tgt_addr, competitor->network.MacAddress, ETH_ALEN) == _TRUE)
3992 			goto update;
3993 		else
3994 			goto exit;
3995 	}
3996 
3997 #ifdef CONFIG_RTW_80211R
3998 	if (rtw_ft_chk_flags(adapter, RTW_FT_PEER_EN)) {
3999 		if (rtw_ft_chk_roaming_candidate(adapter, competitor) == _FALSE)
4000 		goto exit;
4001 	}
4002 
4003 #ifdef CONFIG_RTW_WNM
4004 	if (rtw_wnm_btm_diff_bss(adapter) &&
4005 		rtw_wnm_btm_roam_candidate(adapter, competitor)) {
4006 		goto update;
4007 	}
4008 #endif
4009 #endif
4010 
4011 #if 1
4012 	if (rtw_get_passing_time_ms(competitor->last_scanned) >= mlme->roam_scanr_exp_ms)
4013 		goto exit;
4014 
4015 	if (competitor->network.Rssi - mlme->cur_network_scanned->network.Rssi < mlme->roam_rssi_diff_th)
4016 		goto exit;
4017 
4018 	if (*candidate != NULL && (*candidate)->network.Rssi >= competitor->network.Rssi)
4019 		goto exit;
4020 #else
4021 	goto exit;
4022 #endif
4023 
4024 update:
4025 	*candidate = competitor;
4026 	updated = _TRUE;
4027 
4028 exit:
4029 	return updated;
4030 }
4031 
4032 int rtw_select_roaming_candidate(struct mlme_priv *mlme)
4033 {
4034 	_irqL	irqL;
4035 	int ret = _FAIL;
4036 	_list	*phead;
4037 	_adapter *adapter;
4038 	_queue	*queue	= &(mlme->scanned_queue);
4039 	struct	wlan_network	*pnetwork = NULL;
4040 	struct	wlan_network	*candidate = NULL;
4041 
4042 	if (mlme->cur_network_scanned == NULL) {
4043 		rtw_warn_on(1);
4044 		goto exit;
4045 	}
4046 
4047 	_enter_critical_bh(&(mlme->scanned_queue.lock), &irqL);
4048 	phead = get_list_head(queue);
4049 	adapter = (_adapter *)mlme->nic_hdl;
4050 
4051 	mlme->pscanned = get_next(phead);
4052 
4053 	while (!rtw_end_of_queue_search(phead, mlme->pscanned)) {
4054 
4055 		pnetwork = LIST_CONTAINOR(mlme->pscanned, struct wlan_network, list);
4056 		if (pnetwork == NULL) {
4057 			ret = _FAIL;
4058 			goto exit;
4059 		}
4060 
4061 		mlme->pscanned = get_next(mlme->pscanned);
4062 
4063 		if (0)
4064 			RTW_INFO("%s("MAC_FMT", ch%u) rssi:%d\n"
4065 				 , pnetwork->network.Ssid.Ssid
4066 				 , MAC_ARG(pnetwork->network.MacAddress)
4067 				 , pnetwork->network.Configuration.DSConfig
4068 				 , (int)pnetwork->network.Rssi);
4069 
4070 		rtw_check_roaming_candidate(mlme, &candidate, pnetwork);
4071 
4072 	}
4073 
4074 	if (candidate == NULL) {
4075 	/*	if parent note lost the path to root and there is no other cadidate, report disconnection	*/
4076 #if defined(CONFIG_RTW_REPEATER_SON) &&  (!defined(CONFIG_RTW_REPEATER_SON_ROOT))
4077 		struct rtw_rson_struct  rson_curr;
4078 		u8 rson_score;
4079 
4080 		rtw_get_rson_struct(&(mlme->cur_network_scanned->network), &rson_curr);
4081 		rson_score = rtw_cal_rson_score(&rson_curr, mlme->cur_network_scanned->network.Rssi);
4082 		if (check_fwstate(mlme, WIFI_ASOC_STATE)
4083 			&& ((rson_score == RTW_RSON_SCORE_NOTCNNT)
4084 			|| (rson_score == RTW_RSON_SCORE_NOTSUP)))
4085 			receive_disconnect(adapter, mlme->cur_network_scanned->network.MacAddress
4086 								, WLAN_REASON_EXPIRATION_CHK, _FALSE);
4087 #endif
4088 		RTW_INFO("%s: return _FAIL(candidate == NULL)\n", __FUNCTION__);
4089 		ret = _FAIL;
4090 		goto exit;
4091 	} else {
4092 #if defined(CONFIG_RTW_REPEATER_SON) &&  (!defined(CONFIG_RTW_REPEATER_SON_ROOT))
4093 		struct rtw_rson_struct  rson_curr;
4094 		u8 rson_score;
4095 
4096 		rtw_get_rson_struct(&(candidate->network), &rson_curr);
4097 		rson_score = rtw_cal_rson_score(&rson_curr, candidate->network.Rssi);
4098 		RTW_INFO("%s: candidate: %s("MAC_FMT", ch:%u) rson_score:%d\n", __FUNCTION__,
4099 			candidate->network.Ssid.Ssid, MAC_ARG(candidate->network.MacAddress),
4100 			 candidate->network.Configuration.DSConfig, rson_score);
4101 #else
4102 		RTW_INFO("%s: candidate: %s("MAC_FMT", ch:%u)\n", __FUNCTION__,
4103 			candidate->network.Ssid.Ssid, MAC_ARG(candidate->network.MacAddress),
4104 			 candidate->network.Configuration.DSConfig);
4105 #endif
4106 		mlme->roam_network = candidate;
4107 
4108 		if (_rtw_memcmp(candidate->network.MacAddress, mlme->roam_tgt_addr, ETH_ALEN) == _TRUE)
4109 			_rtw_memset(mlme->roam_tgt_addr, 0, ETH_ALEN);
4110 	}
4111 
4112 	ret = _SUCCESS;
4113 exit:
4114 	_exit_critical_bh(&(mlme->scanned_queue.lock), &irqL);
4115 
4116 	return ret;
4117 }
4118 #endif /* CONFIG_LAYER2_ROAMING */
4119 
4120 /*
4121 * Select a new join candidate from the original @param candidate and @param competitor
4122 * @return _TRUE: candidate is updated
4123 * @return _FALSE: candidate is not updated
4124 */
4125 static int rtw_check_join_candidate(struct mlme_priv *mlme
4126 	    , struct wlan_network **candidate, struct wlan_network *competitor)
4127 {
4128 	int updated = _FALSE;
4129 	_adapter *adapter = container_of(mlme, _adapter, mlmepriv);
4130 	struct rf_ctl_t *rfctl = adapter_to_rfctl(adapter);
4131 	RT_CHANNEL_INFO *chset = rfctl->channel_set;
4132 	u8 ch = competitor->network.Configuration.DSConfig;
4133 
4134 	if (rtw_chset_search_ch(chset, ch) < 0)
4135 		goto exit;
4136 	if (IS_DFS_SLAVE_WITH_RD(rfctl)
4137 		&& !rtw_rfctl_dfs_domain_unknown(rfctl)
4138 		&& rtw_chset_is_ch_non_ocp(chset, ch))
4139 		goto exit;
4140 
4141 #if defined(CONFIG_RTW_REPEATER_SON) &&  (!defined(CONFIG_RTW_REPEATER_SON_ROOT))
4142 	s16 rson_score;
4143 	struct rtw_rson_struct  rson_data;
4144 
4145 	if (rtw_rson_choose(candidate, competitor)) {
4146 		*candidate = competitor;
4147 		rtw_get_rson_struct(&((*candidate)->network), &rson_data);
4148 		rson_score = rtw_cal_rson_score(&rson_data, (*candidate)->network.Rssi);
4149 		RTW_INFO("[assoc_ssid:%s] new candidate: %s("MAC_FMT", ch%u) rson_score:%d\n",
4150 			 mlme->assoc_ssid.Ssid,
4151 			 (*candidate)->network.Ssid.Ssid,
4152 			 MAC_ARG((*candidate)->network.MacAddress),
4153 			 (*candidate)->network.Configuration.DSConfig,
4154 			 rson_score);
4155 		return _TRUE;
4156 	}
4157 	return _FALSE;
4158 #endif
4159 
4160 	/* check bssid, if needed */
4161 	if (mlme->assoc_by_bssid == _TRUE) {
4162 		if (_rtw_memcmp(competitor->network.MacAddress, mlme->assoc_bssid, ETH_ALEN) == _FALSE)
4163 			goto exit;
4164 	}
4165 
4166 	/* check ssid, if needed */
4167 	if (mlme->assoc_ssid.Ssid[0] && mlme->assoc_ssid.SsidLength) {
4168 		if (competitor->network.Ssid.SsidLength != mlme->assoc_ssid.SsidLength
4169 		    || _rtw_memcmp(competitor->network.Ssid.Ssid, mlme->assoc_ssid.Ssid, mlme->assoc_ssid.SsidLength) == _FALSE
4170 		   )
4171 			goto exit;
4172 	}
4173 
4174 	if (rtw_is_desired_network(adapter, competitor)  == _FALSE)
4175 		goto exit;
4176 
4177 #ifdef CONFIG_LAYER2_ROAMING
4178 	if (rtw_to_roam(adapter) > 0) {
4179 		if (rtw_get_passing_time_ms(competitor->last_scanned) >= mlme->roam_scanr_exp_ms
4180 		    || is_same_ess(&competitor->network, &mlme->cur_network.network) == _FALSE
4181 		   )
4182 			goto exit;
4183 	}
4184 #endif
4185 
4186 	if (*candidate == NULL || (*candidate)->network.Rssi < competitor->network.Rssi) {
4187 		*candidate = competitor;
4188 		updated = _TRUE;
4189 	}
4190 
4191 	if (updated) {
4192 		RTW_INFO("[by_bssid:%u][assoc_ssid:%s][to_roam:%u] "
4193 			 "new candidate: %s("MAC_FMT", ch%u) rssi:%d dBm\n",
4194 			 mlme->assoc_by_bssid,
4195 			 mlme->assoc_ssid.Ssid,
4196 			 rtw_to_roam(adapter),
4197 			 (*candidate)->network.Ssid.Ssid,
4198 			 MAC_ARG((*candidate)->network.MacAddress),
4199 			 (*candidate)->network.Configuration.DSConfig,
4200 			 (int)(*candidate)->network.Rssi
4201 			);
4202 	}
4203 
4204 exit:
4205 	return updated;
4206 }
4207 
4208 /*
4209 Calling context:
4210 The caller of the sub-routine will be in critical section...
4211 
4212 The caller must hold the following spinlock
4213 
4214 pmlmepriv->lock
4215 
4216 
4217 */
4218 
4219 int rtw_select_and_join_from_scanned_queue(struct mlme_priv *pmlmepriv)
4220 {
4221 	_irqL	irqL;
4222 	int ret;
4223 	_list	*phead;
4224 	_adapter *adapter;
4225 	_queue	*queue	= &(pmlmepriv->scanned_queue);
4226 	struct	wlan_network	*pnetwork = NULL;
4227 	struct	wlan_network	*candidate = NULL;
4228 #ifdef CONFIG_ANTENNA_DIVERSITY
4229 	u8		bSupportAntDiv = _FALSE;
4230 #endif
4231 
4232 	adapter = (_adapter *)pmlmepriv->nic_hdl;
4233 
4234 	_enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
4235 
4236 #ifdef CONFIG_LAYER2_ROAMING
4237 	if (pmlmepriv->roam_network) {
4238 		candidate = pmlmepriv->roam_network;
4239 		goto candidate_exist;
4240 	}
4241 #endif
4242 
4243 	phead = get_list_head(queue);
4244 	pmlmepriv->pscanned = get_next(phead);
4245 
4246 	while (!rtw_end_of_queue_search(phead, pmlmepriv->pscanned)) {
4247 
4248 		pnetwork = LIST_CONTAINOR(pmlmepriv->pscanned, struct wlan_network, list);
4249 		if (pnetwork == NULL) {
4250 			ret = _FAIL;
4251 			goto exit;
4252 		}
4253 
4254 		pmlmepriv->pscanned = get_next(pmlmepriv->pscanned);
4255 
4256 		if (0)
4257 			RTW_INFO("%s("MAC_FMT", ch%u) rssi:%d\n"
4258 				 , pnetwork->network.Ssid.Ssid
4259 				 , MAC_ARG(pnetwork->network.MacAddress)
4260 				 , pnetwork->network.Configuration.DSConfig
4261 				 , (int)pnetwork->network.Rssi);
4262 
4263 		rtw_check_join_candidate(pmlmepriv, &candidate, pnetwork);
4264 
4265 	}
4266 
4267 	if (candidate == NULL) {
4268 		RTW_INFO("%s: return _FAIL(candidate == NULL)\n", __FUNCTION__);
4269 #ifdef CONFIG_WOWLAN
4270 		_clr_fwstate_(pmlmepriv, WIFI_ASOC_STATE | WIFI_UNDER_LINKING);
4271 #endif
4272 		ret = _FAIL;
4273 		goto exit;
4274 	} else {
4275 		RTW_INFO("%s: candidate: %s("MAC_FMT", ch:%u)\n", __FUNCTION__,
4276 			candidate->network.Ssid.Ssid, MAC_ARG(candidate->network.MacAddress),
4277 			 candidate->network.Configuration.DSConfig);
4278 		goto candidate_exist;
4279 	}
4280 
4281 candidate_exist:
4282 
4283 	/* check for situation of  WIFI_ASOC_STATE */
4284 	if (check_fwstate(pmlmepriv, WIFI_ASOC_STATE) == _TRUE) {
4285 		RTW_INFO("%s: WIFI_ASOC_STATE while ask_for_joinbss!!!\n", __FUNCTION__);
4286 
4287 #if 0 /* for WPA/WPA2 authentication, wpa_supplicant will expect authentication from AP, it is needed to reconnect AP... */
4288 		if (is_same_network(&pmlmepriv->cur_network.network, &candidate->network)) {
4289 			RTW_INFO("%s: WIFI_ASOC_STATE and is same network, it needn't join again\n", __FUNCTION__);
4290 
4291 			rtw_indicate_connect(adapter);/* rtw_indicate_connect again */
4292 
4293 			ret = 2;
4294 			goto exit;
4295 		} else
4296 #endif
4297 		{
4298 			rtw_disassoc_cmd(adapter, 0, 0);
4299 			rtw_indicate_disconnect(adapter, 0, _FALSE);
4300 			rtw_free_assoc_resources_cmd(adapter, _TRUE, 0);
4301 		}
4302 	}
4303 
4304 #ifdef CONFIG_ANTENNA_DIVERSITY
4305 	rtw_hal_get_def_var(adapter, HAL_DEF_IS_SUPPORT_ANT_DIV, &(bSupportAntDiv));
4306 	if (_TRUE == bSupportAntDiv) {
4307 		u8 CurrentAntenna;
4308 		rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(CurrentAntenna), NULL);
4309 		RTW_INFO("#### Opt_Ant_(%s) , cur_Ant(%s)\n",
4310 			(MAIN_ANT == candidate->network.PhyInfo.Optimum_antenna) ? "MAIN_ANT" : "AUX_ANT",
4311 			 (MAIN_ANT == CurrentAntenna) ? "MAIN_ANT" : "AUX_ANT"
4312 			);
4313 	}
4314 #endif
4315 	set_fwstate(pmlmepriv, WIFI_UNDER_LINKING);
4316 
4317 #if defined(CONFIG_CONCURRENT_MODE) && defined(CONFIG_AP_MODE)
4318 {
4319 	u8 ifbmp = 0;
4320 	u8 csa_cnt;
4321 	struct mi_state mstate;
4322 	struct rf_ctl_t *rfctl;
4323 
4324 	rfctl = adapter_to_rfctl(adapter);
4325 	csa_cnt = rfctl->ap_csa_cnt_input;
4326 	ifbmp = rtw_mi_get_ap_mesh_ifbmp(adapter);
4327 	rtw_mi_status_no_self(adapter, &mstate);
4328 
4329 	if (csa_cnt > 0 && ifbmp && MSTATE_AP_LD_NUM(&mstate) &&
4330 		rtw_mi_get_union_chan(adapter) != candidate->network.Configuration.DSConfig) {
4331 		rfctl->ap_csa_en = CSA_STA_JOINBSS;
4332 		rfctl->ap_csa_switch_cnt = csa_cnt;
4333 		rtw_bss_get_chbw(&(candidate->network), &rfctl->ap_csa_ch
4334 			, &rfctl->ap_csa_ch_width, &rfctl->ap_csa_ch_offset, 1, 1);
4335 		rtw_set_ap_csa_cmd(adapter);
4336 
4337 		/* Store candidata network until softap switch channel done */
4338 		_rtw_memcpy(&(pmlmepriv->candidate_network), candidate, sizeof(struct wlan_network));
4339 		ret = _SUCCESS;
4340 		goto exit;
4341 	}
4342 }
4343 #endif
4344 
4345 	ret = rtw_joinbss_cmd(adapter, candidate);
4346 
4347 exit:
4348 	_exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
4349 
4350 
4351 	return ret;
4352 }
4353 
4354 sint rtw_set_auth(_adapter *adapter, struct security_priv *psecuritypriv)
4355 {
4356 	struct	cmd_obj *pcmd;
4357 	struct	setauth_parm *psetauthparm;
4358 	struct	cmd_priv	*pcmdpriv = &(adapter->cmdpriv);
4359 	sint		res = _SUCCESS;
4360 
4361 
4362 	pcmd = (struct	cmd_obj *)rtw_zmalloc(sizeof(struct	cmd_obj));
4363 	if (pcmd == NULL) {
4364 		res = _FAIL; /* try again */
4365 		goto exit;
4366 	}
4367 
4368 	psetauthparm = (struct setauth_parm *)rtw_zmalloc(sizeof(struct setauth_parm));
4369 	if (psetauthparm == NULL) {
4370 		rtw_mfree((unsigned char *)pcmd, sizeof(struct	cmd_obj));
4371 		res = _FAIL;
4372 		goto exit;
4373 	}
4374 
4375 	_rtw_memset(psetauthparm, 0, sizeof(struct setauth_parm));
4376 	psetauthparm->mode = (unsigned char)psecuritypriv->dot11AuthAlgrthm;
4377 
4378 	pcmd->cmdcode = CMD_SET_AUTH;
4379 	pcmd->parmbuf = (unsigned char *)psetauthparm;
4380 	pcmd->cmdsz = (sizeof(struct setauth_parm));
4381 	pcmd->rsp = NULL;
4382 	pcmd->rspsz = 0;
4383 
4384 
4385 	_rtw_init_listhead(&pcmd->list);
4386 
4387 
4388 	res = rtw_enqueue_cmd(pcmdpriv, pcmd);
4389 
4390 exit:
4391 
4392 
4393 	return res;
4394 
4395 }
4396 
4397 
4398 sint rtw_set_key(_adapter *adapter, struct security_priv *psecuritypriv, sint keyid, u8 set_tx, bool enqueue)
4399 {
4400 	u8	keylen;
4401 	struct cmd_obj		*pcmd;
4402 	struct setkey_parm	*psetkeyparm;
4403 	struct cmd_priv		*pcmdpriv = &(adapter->cmdpriv);
4404 	sint	res = _SUCCESS;
4405 
4406 
4407 	psetkeyparm = (struct setkey_parm *)rtw_zmalloc(sizeof(struct setkey_parm));
4408 	if (psetkeyparm == NULL) {
4409 		res = _FAIL;
4410 		goto exit;
4411 	}
4412 	_rtw_memset(psetkeyparm, 0, sizeof(struct setkey_parm));
4413 
4414 	if (psecuritypriv->dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) {
4415 		psetkeyparm->algorithm = (unsigned char)psecuritypriv->dot118021XGrpPrivacy;
4416 	} else {
4417 		psetkeyparm->algorithm = (u8)psecuritypriv->dot11PrivacyAlgrthm;
4418 
4419 	}
4420 	psetkeyparm->keyid = (u8)keyid;/* 0~3 */
4421 	psetkeyparm->set_tx = set_tx;
4422 	if (is_wep_enc(psetkeyparm->algorithm))
4423 		adapter->securitypriv.key_mask |= BIT(psetkeyparm->keyid);
4424 
4425 	RTW_INFO("==> rtw_set_key algorithm(%x),keyid(%x),key_mask(%x)\n", psetkeyparm->algorithm, psetkeyparm->keyid, adapter->securitypriv.key_mask);
4426 
4427 	switch (psetkeyparm->algorithm) {
4428 
4429 	case _WEP40_:
4430 		keylen = 5;
4431 		_rtw_memcpy(&(psetkeyparm->key[0]), &(psecuritypriv->dot11DefKey[keyid].skey[0]), keylen);
4432 		break;
4433 	case _WEP104_:
4434 		keylen = 13;
4435 		_rtw_memcpy(&(psetkeyparm->key[0]), &(psecuritypriv->dot11DefKey[keyid].skey[0]), keylen);
4436 		break;
4437 	case _TKIP_:
4438 		keylen = 16;
4439 		_rtw_memcpy(&psetkeyparm->key, &psecuritypriv->dot118021XGrpKey[keyid], keylen);
4440 		break;
4441 	case _AES_:
4442 	case _GCMP_:
4443 		keylen = 16;
4444 		_rtw_memcpy(&psetkeyparm->key, &psecuritypriv->dot118021XGrpKey[keyid], keylen);
4445 		break;
4446 	case _GCMP_256_:
4447 	case _CCMP_256_:
4448 		keylen = 32;
4449 		_rtw_memcpy(&psetkeyparm->key, &psecuritypriv->dot118021XGrpKey[keyid], keylen);
4450 		break;
4451 	default:
4452 		res = _FAIL;
4453 		rtw_mfree((unsigned char *)psetkeyparm, sizeof(struct setkey_parm));
4454 		goto exit;
4455 	}
4456 
4457 
4458 	if (enqueue) {
4459 		pcmd = (struct	cmd_obj *)rtw_zmalloc(sizeof(struct	cmd_obj));
4460 		if (pcmd == NULL) {
4461 			rtw_mfree((unsigned char *)psetkeyparm, sizeof(struct setkey_parm));
4462 			res = _FAIL; /* try again */
4463 			goto exit;
4464 		}
4465 
4466 		pcmd->cmdcode =CMD_SET_KEY;
4467 		pcmd->parmbuf = (u8 *)psetkeyparm;
4468 		pcmd->cmdsz = (sizeof(struct setkey_parm));
4469 		pcmd->rsp = NULL;
4470 		pcmd->rspsz = 0;
4471 
4472 		_rtw_init_listhead(&pcmd->list);
4473 
4474 		/* _rtw_init_sema(&(pcmd->cmd_sem), 0); */
4475 
4476 		res = rtw_enqueue_cmd(pcmdpriv, pcmd);
4477 	} else {
4478 		setkey_hdl(adapter, (u8 *)psetkeyparm);
4479 		rtw_mfree((u8 *) psetkeyparm, sizeof(struct setkey_parm));
4480 	}
4481 exit:
4482 	return res;
4483 
4484 }
4485 
4486 #ifdef CONFIG_WMMPS_STA
4487 /*
4488  * rtw_uapsd_use_default_setting
4489  * This function is used for setting default uapsd max sp length to uapsd_max_sp_len
4490  * in qos_priv data structure from registry. In additional, it will also map default uapsd
4491  * ac to each uapsd TID, delivery-enabled and trigger-enabled of corresponding TID.
4492  *
4493  * Arguments:
4494  * @padapter: _adapter pointer.
4495  *
4496  * Auther: Arvin Liu
4497  * Date: 2017/05/03
4498  */
4499 void	rtw_uapsd_use_default_setting(_adapter *padapter)
4500 {
4501 	struct mlme_priv		*pmlmepriv = &padapter->mlmepriv;
4502 	struct qos_priv		*pqospriv = &pmlmepriv->qospriv;
4503 	struct registry_priv		*pregistrypriv = &padapter->registrypriv;
4504 
4505 	if (pregistrypriv->uapsd_ac_enable != 0) {
4506 		pqospriv->uapsd_max_sp_len = pregistrypriv->uapsd_max_sp_len;
4507 
4508 		CLEAR_FLAGS(pqospriv->uapsd_tid);
4509 		CLEAR_FLAGS(pqospriv->uapsd_tid_delivery_enabled);
4510 		CLEAR_FLAGS(pqospriv->uapsd_tid_trigger_enabled);
4511 
4512 		/* check the uapsd setting of AC_VO from registry then map these setting to each TID if necessary  */
4513 		if(TEST_FLAG(pregistrypriv->uapsd_ac_enable, DRV_CFG_UAPSD_VO)) {
4514 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID7);
4515 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID7);
4516 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID7);
4517 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID6);
4518 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID6);
4519 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID6);
4520 		}
4521 
4522 		/* check the uapsd setting of AC_VI from registry then map these setting to each TID if necessary  */
4523 		if(TEST_FLAG(pregistrypriv->uapsd_ac_enable, DRV_CFG_UAPSD_VI)) {
4524 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID5);
4525 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID5);
4526 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID5);
4527 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID4);
4528 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID4);
4529 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID4);
4530 		}
4531 
4532 		/* check the uapsd setting of AC_BK from registry then map these setting to each TID if necessary  */
4533 		if(TEST_FLAG(pregistrypriv->uapsd_ac_enable, DRV_CFG_UAPSD_BK)) {
4534 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID2);
4535 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID2);
4536 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID2);
4537 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID1);
4538 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID1);
4539 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID1);
4540 		}
4541 
4542 		/* check the uapsd setting of AC_BE from registry then map these setting to each TID if necessary  */
4543 		if(TEST_FLAG(pregistrypriv->uapsd_ac_enable, DRV_CFG_UAPSD_BE)) {
4544 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID3);
4545 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID3);
4546 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID3);
4547 			SET_FLAG(pqospriv->uapsd_tid, WMM_TID0);
4548 			SET_FLAG(pqospriv->uapsd_tid_delivery_enabled, WMM_TID0);
4549 			SET_FLAG(pqospriv->uapsd_tid_trigger_enabled, WMM_TID0);
4550 		}
4551 
4552 		RTW_INFO("[WMMPS] UAPSD MAX SP Len = 0x%02x, UAPSD TID enabled = 0x%02x\n",
4553 			pqospriv->uapsd_max_sp_len, (u8)pqospriv->uapsd_tid);
4554 	}
4555 
4556 }
4557 
4558 /*
4559  * rtw_is_wmmps_mode
4560  * This function is used for checking whether Driver and an AP support uapsd function or not.
4561  * If both of them support uapsd function, it will return true. Otherwise returns false.
4562  *
4563  * Arguments:
4564  * @padapter: _adapter pointer.
4565  *
4566  * Auther: Arvin Liu
4567  * Date: 2017/06/12
4568  */
4569 bool rtw_is_wmmps_mode(_adapter *padapter)
4570 {
4571 	struct mlme_priv	*pmlmepriv = &(padapter->mlmepriv);
4572 	struct qos_priv	*pqospriv = &pmlmepriv->qospriv;
4573 
4574 	if ((pqospriv->uapsd_ap_supported) && ((pqospriv->uapsd_tid & BIT_MASK_TID_TC)  != 0))
4575 		return _TRUE;
4576 
4577 	return _FALSE;
4578 }
4579 #endif /* CONFIG_WMMPS_STA */
4580 
4581 /* adjust IEs for rtw_joinbss_cmd in WMM */
4582 int rtw_restruct_wmm_ie(_adapter *adapter, u8 *in_ie, u8 *out_ie, uint in_len, uint initial_out_len)
4583 {
4584 #ifdef CONFIG_WMMPS_STA
4585 	struct mlme_priv		*pmlmepriv = &adapter->mlmepriv;
4586 	struct qos_priv		*pqospriv = &pmlmepriv->qospriv;
4587 #endif /* CONFIG_WMMPS_STA */
4588 	unsigned	int ielength = 0;
4589 	unsigned int i, j;
4590 	u8 qos_info = 0;
4591 
4592 	i = 12; /* after the fixed IE */
4593 	while (i < in_len) {
4594 		ielength = initial_out_len;
4595 
4596 		if (in_ie[i] == 0xDD && in_ie[i + 2] == 0x00 && in_ie[i + 3] == 0x50  && in_ie[i + 4] == 0xF2 && in_ie[i + 5] == 0x02 && i + 5 < in_len) { /* WMM element ID and OUI */
4597 
4598 			/* Append WMM IE to the last index of out_ie */
4599 #if 0
4600 			for (j = i; j < i + (in_ie[i + 1] + 2); j++) {
4601 				out_ie[ielength] = in_ie[j];
4602 				ielength++;
4603 			}
4604 			out_ie[initial_out_len + 8] = 0x00; /* force the QoS Info Field to be zero */
4605 #endif
4606 
4607 			for (j = i; j < i + 9; j++) {
4608 				out_ie[ielength] = in_ie[j];
4609 				ielength++;
4610 			}
4611 			out_ie[initial_out_len + 1] = 0x07;
4612 			out_ie[initial_out_len + 6] = 0x00;
4613 
4614 #ifdef CONFIG_WMMPS_STA
4615 			switch(pqospriv->uapsd_max_sp_len) {
4616 				case NO_LIMIT:
4617 					/* do nothing */
4618 					break;
4619 				case TWO_MSDU:
4620 					SET_FLAG(qos_info, BIT5);
4621 					break;
4622 				case FOUR_MSDU:
4623 					SET_FLAG(qos_info, BIT6);
4624 					break;
4625 				case SIX_MSDU:
4626 					SET_FLAG(qos_info, BIT5);
4627 					SET_FLAG(qos_info, BIT6);
4628 					break;
4629 				default:
4630 					/* do nothing */
4631 					break;
4632 			};
4633 
4634 			/* check TID7 and TID6 for AC_VO to set corresponding Qos_info bit in WMM IE  */
4635 			if((TEST_FLAG(pqospriv->uapsd_tid, WMM_TID7)) && (TEST_FLAG(pqospriv->uapsd_tid, WMM_TID6)))
4636 				SET_FLAG(qos_info, WMM_IE_UAPSD_VO);
4637 			/* check TID5 and TID4 for AC_VI to set corresponding Qos_info bit in WMM IE  */
4638 			if((TEST_FLAG(pqospriv->uapsd_tid, WMM_TID5)) && (TEST_FLAG(pqospriv->uapsd_tid, WMM_TID4)))
4639 				SET_FLAG(qos_info, WMM_IE_UAPSD_VI);
4640 			/* check TID2 and TID1 for AC_BK to set corresponding Qos_info bit in WMM IE  */
4641 			if((TEST_FLAG(pqospriv->uapsd_tid, WMM_TID2)) && (TEST_FLAG(pqospriv->uapsd_tid, WMM_TID1)))
4642 				SET_FLAG(qos_info, WMM_IE_UAPSD_BK);
4643 			/* check TID3 and TID0 for AC_BE to set corresponding Qos_info bit in WMM IE  */
4644 			if((TEST_FLAG(pqospriv->uapsd_tid, WMM_TID3)) && (TEST_FLAG(pqospriv->uapsd_tid, WMM_TID0)))
4645 				SET_FLAG(qos_info, WMM_IE_UAPSD_BE);
4646 #endif /* CONFIG_WMMPS_STA */
4647 
4648 			out_ie[initial_out_len + 8] = qos_info;
4649 
4650 			break;
4651 		}
4652 
4653 		i += (in_ie[i + 1] + 2); /* to the next IE element */
4654 	}
4655 
4656 	return ielength;
4657 
4658 }
4659 
4660 
4661 /*
4662  * Ported from 8185: IsInPreAuthKeyList(). (Renamed from SecIsInPreAuthKeyList(), 2006-10-13.)
4663  * Added by Annie, 2006-05-07.
4664  *
4665  * Search by BSSID,
4666  * Return Value:
4667  *		-1		:if there is no pre-auth key in the  table
4668  *		>=0		:if there is pre-auth key, and   return the entry id
4669  *
4670  *   */
4671 
4672 static int SecIsInPMKIDList(_adapter *Adapter, u8 *bssid)
4673 {
4674 	struct security_priv *psecuritypriv = &Adapter->securitypriv;
4675 	int i = 0;
4676 
4677 	do {
4678 		if ((psecuritypriv->PMKIDList[i].bUsed) &&
4679 		    (_rtw_memcmp(psecuritypriv->PMKIDList[i].Bssid, bssid, ETH_ALEN) == _TRUE))
4680 			break;
4681 		else {
4682 			i++;
4683 			/* continue; */
4684 		}
4685 
4686 	} while (i < NUM_PMKID_CACHE);
4687 
4688 	if (i == NUM_PMKID_CACHE) {
4689 		i = -1;/* Could not find. */
4690 	} else {
4691 		/* There is one Pre-Authentication Key for the specific BSSID. */
4692 	}
4693 
4694 	return i;
4695 
4696 }
4697 
4698 int rtw_cached_pmkid(_adapter *Adapter, u8 *bssid)
4699 {
4700 	return SecIsInPMKIDList(Adapter, bssid);
4701 }
4702 
4703 int rtw_rsn_sync_pmkid(_adapter *adapter, u8 *ie, uint ie_len, int i_ent)
4704 {
4705 	struct security_priv *sec = &adapter->securitypriv;
4706 	struct rsne_info info;
4707 	u8 gm_cs[4];
4708 	int i;
4709 
4710 	rtw_rsne_info_parse(ie, ie_len, &info);
4711 
4712 	if (info.err) {
4713 		RTW_WARN(FUNC_ADPT_FMT" rtw_rsne_info_parse error\n"
4714 			, FUNC_ADPT_ARG(adapter));
4715 		return 0;
4716 	}
4717 
4718 	if (i_ent < 0 && info.pmkid_cnt == 0)
4719 		goto exit;
4720 
4721 	if (info.pmkid_list == NULL)
4722 		goto exit;
4723 
4724 	if (i_ent >= 0 && info.pmkid_cnt == 1 && _rtw_memcmp(info.pmkid_list, sec->PMKIDList[i_ent].PMKID, 16)) {
4725 		RTW_INFO(FUNC_ADPT_FMT" has carried the same PMKID:"KEY_FMT"\n"
4726 			, FUNC_ADPT_ARG(adapter), KEY_ARG(&sec->PMKIDList[i_ent].PMKID));
4727 		goto exit;
4728 	}
4729 
4730 	/* bakcup group mgmt cs */
4731 	if (info.gmcs)
4732 		_rtw_memcpy(gm_cs, info.gmcs, 4);
4733 
4734 	if (info.pmkid_cnt) {
4735 		RTW_INFO(FUNC_ADPT_FMT" remove original PMKID, count:%u\n"
4736 			 , FUNC_ADPT_ARG(adapter), info.pmkid_cnt);
4737 		for (i = 0; i < info.pmkid_cnt; i++)
4738 			RTW_INFO("    "KEY_FMT"\n", KEY_ARG(info.pmkid_list + i * 16));
4739 	}
4740 
4741 	if (i_ent >= 0) {
4742 		RTW_INFO(FUNC_ADPT_FMT" append PMKID:"KEY_FMT"\n"
4743 			, FUNC_ADPT_ARG(adapter), KEY_ARG(sec->PMKIDList[i_ent].PMKID));
4744 
4745 		info.pmkid_cnt = 1; /* update new pmkid_cnt */
4746 		_rtw_memcpy(info.pmkid_list, sec->PMKIDList[i_ent].PMKID, 16);
4747 	} else
4748 		info.pmkid_cnt = 0; /* update new pmkid_cnt */
4749 
4750 	RTW_PUT_LE16(info.pmkid_list - 2, info.pmkid_cnt);
4751 	if (info.gmcs)
4752 		_rtw_memcpy(info.pmkid_list + 16 * info.pmkid_cnt, gm_cs, 4);
4753 
4754 	ie_len = 1 + 1 + 2 + 4
4755 		+ 2 + 4 * info.pcs_cnt
4756 		+ 2 + 4 * info.akm_cnt
4757 		+ 2
4758 		+ 2 + 16 * info.pmkid_cnt
4759 		+ (info.gmcs ? 4 : 0)
4760 		;
4761 
4762 	ie[1] = (u8)(ie_len - 2);
4763 
4764 exit:
4765 	return ie_len;
4766 }
4767 
4768 sint rtw_restruct_sec_ie(_adapter *adapter, u8 *out_ie)
4769 {
4770 	u8 authmode = 0x0;
4771 	uint	ielength = 0;
4772 	int iEntry;
4773 
4774 	struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
4775 	struct security_priv *psecuritypriv = &adapter->securitypriv;
4776 	uint	ndisauthmode = psecuritypriv->ndisauthtype;
4777 
4778 	if ((ndisauthmode == Ndis802_11AuthModeWPA) || (ndisauthmode == Ndis802_11AuthModeWPAPSK))
4779 		authmode = _WPA_IE_ID_;
4780 	if ((ndisauthmode == Ndis802_11AuthModeWPA2) || (ndisauthmode == Ndis802_11AuthModeWPA2PSK))
4781 		authmode = _WPA2_IE_ID_;
4782 
4783 	if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
4784 		_rtw_memcpy(out_ie, psecuritypriv->wps_ie, psecuritypriv->wps_ie_len);
4785 		ielength = psecuritypriv->wps_ie_len;
4786 
4787 	} else if ((authmode == _WPA_IE_ID_) || (authmode == _WPA2_IE_ID_)) {
4788 		/* copy RSN or SSN		 */
4789 		_rtw_memcpy(out_ie, psecuritypriv->supplicant_ie, psecuritypriv->supplicant_ie[1] + 2);
4790 		/* debug for CONFIG_IEEE80211W
4791 		{
4792 			int jj;
4793 			printk("supplicant_ie_length=%d &&&&&&&&&&&&&&&&&&&\n", psecuritypriv->supplicant_ie[1]+2);
4794 			for(jj=0; jj < psecuritypriv->supplicant_ie[1]+2; jj++)
4795 				printk(" %02x ", psecuritypriv->supplicant_ie[jj]);
4796 			printk("\n");
4797 		}*/
4798 		ielength = psecuritypriv->supplicant_ie[1] + 2;
4799 		rtw_report_sec_ie(adapter, authmode, psecuritypriv->supplicant_ie);
4800 	}
4801 
4802 	if (authmode == WLAN_EID_RSN) {
4803 		iEntry = SecIsInPMKIDList(adapter, pmlmepriv->assoc_bssid);
4804 		ielength = rtw_rsn_sync_pmkid(adapter, out_ie, ielength, iEntry);
4805 	}
4806 
4807 	if ((psecuritypriv->auth_type == MLME_AUTHTYPE_SAE) &&
4808 		(psecuritypriv->rsnx_ie_len >= 3)) {
4809 		u8 *_pos = out_ie + (psecuritypriv->supplicant_ie[1] + 2);
4810 		_rtw_memcpy(_pos, psecuritypriv->rsnx_ie,
4811 			psecuritypriv->rsnx_ie_len);
4812 		ielength += psecuritypriv->rsnx_ie_len;
4813 		RTW_INFO_DUMP("update IE for RSNX :", out_ie, ielength);
4814 	}
4815 
4816 	return ielength;
4817 }
4818 
4819 void rtw_init_registrypriv_dev_network(_adapter *adapter)
4820 {
4821 	struct registry_priv *pregistrypriv = &adapter->registrypriv;
4822 	WLAN_BSSID_EX    *pdev_network = &pregistrypriv->dev_network;
4823 	u8 *myhwaddr = adapter_mac_addr(adapter);
4824 
4825 
4826 	_rtw_memcpy(pdev_network->MacAddress, myhwaddr, ETH_ALEN);
4827 
4828 	_rtw_memcpy(&pdev_network->Ssid, &pregistrypriv->ssid, sizeof(NDIS_802_11_SSID));
4829 
4830 	pdev_network->Configuration.Length = sizeof(NDIS_802_11_CONFIGURATION);
4831 	pdev_network->Configuration.BeaconPeriod = 100;
4832 }
4833 
4834 void rtw_update_registrypriv_dev_network(_adapter *adapter)
4835 {
4836 	int sz = 0;
4837 	struct registry_priv *pregistrypriv = &adapter->registrypriv;
4838 	WLAN_BSSID_EX    *pdev_network = &pregistrypriv->dev_network;
4839 	struct	security_priv	*psecuritypriv = &adapter->securitypriv;
4840 	struct	wlan_network	*cur_network = &adapter->mlmepriv.cur_network;
4841 	/* struct	xmit_priv	*pxmitpriv = &adapter->xmitpriv; */
4842 	struct mlme_ext_priv	*pmlmeext = &adapter->mlmeextpriv;
4843 
4844 
4845 #if 0
4846 	pxmitpriv->vcs_setting = pregistrypriv->vrtl_carrier_sense;
4847 	pxmitpriv->vcs = pregistrypriv->vcs_type;
4848 	pxmitpriv->vcs_type = pregistrypriv->vcs_type;
4849 	/* pxmitpriv->rts_thresh = pregistrypriv->rts_thresh; */
4850 	pxmitpriv->frag_len = pregistrypriv->frag_thresh;
4851 
4852 	adapter->qospriv.qos_option = pregistrypriv->wmm_enable;
4853 #endif
4854 
4855 	pdev_network->Privacy = (psecuritypriv->dot11PrivacyAlgrthm > 0 ? 1 : 0) ; /* adhoc no 802.1x */
4856 
4857 	pdev_network->Rssi = 0;
4858 
4859 	pdev_network->Configuration.DSConfig = (pregistrypriv->channel);
4860 
4861 	if (cur_network->network.InfrastructureMode == Ndis802_11IBSS) {
4862 		pdev_network->Configuration.ATIMWindow = (0);
4863 
4864 		if (pmlmeext->cur_channel != 0)
4865 			pdev_network->Configuration.DSConfig = pmlmeext->cur_channel;
4866 		else
4867 			pdev_network->Configuration.DSConfig = 1;
4868 	}
4869 
4870 	pdev_network->InfrastructureMode = (cur_network->network.InfrastructureMode);
4871 
4872 	/* 1. Supported rates */
4873 	/* 2. IE */
4874 
4875 	/* rtw_set_supported_rate(pdev_network->SupportedRates, pregistrypriv->wireless_mode) ; */ /* will be called in rtw_generate_ie */
4876 	sz = rtw_generate_ie(pregistrypriv);
4877 
4878 	pdev_network->IELength = sz;
4879 
4880 	pdev_network->Length = get_WLAN_BSSID_EX_sz((WLAN_BSSID_EX *)pdev_network);
4881 
4882 	/* notes: translate IELength & Length after assign the Length to cmdsz in createbss_cmd(); */
4883 	/* pdev_network->IELength = cpu_to_le32(sz); */
4884 
4885 
4886 }
4887 
4888 void rtw_get_encrypt_decrypt_from_registrypriv(_adapter *adapter)
4889 {
4890 
4891 
4892 
4893 }
4894 
4895 /* the fucntion is at passive_level */
4896 void rtw_joinbss_reset(_adapter *padapter)
4897 {
4898 	u8	threshold;
4899 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
4900 	/* todo: if you want to do something io/reg/hw setting before join_bss, please add code here */
4901 
4902 #ifdef CONFIG_80211N_HT
4903 	struct ht_priv		*phtpriv = &pmlmepriv->htpriv;
4904 
4905 	pmlmepriv->num_FortyMHzIntolerant = 0;
4906 
4907 	pmlmepriv->num_sta_no_ht = 0;
4908 
4909 	phtpriv->ampdu_enable = _FALSE;/* reset to disabled */
4910 
4911 #if defined(CONFIG_USB_HCI) || defined(CONFIG_SDIO_HCI)
4912 	/* TH=1 => means that invalidate usb rx aggregation */
4913 	/* TH=0 => means that validate usb rx aggregation, use init value. */
4914 	if (phtpriv->ht_option) {
4915 		if (padapter->registrypriv.wifi_spec == 1)
4916 			threshold = 1;
4917 		else
4918 			threshold = 0;
4919 		rtw_hal_set_hwreg(padapter, HW_VAR_RXDMA_AGG_PG_TH, (u8 *)(&threshold));
4920 	} else {
4921 		threshold = 1;
4922 		rtw_hal_set_hwreg(padapter, HW_VAR_RXDMA_AGG_PG_TH, (u8 *)(&threshold));
4923 	}
4924 #endif/* #if defined( CONFIG_USB_HCI) || defined (CONFIG_SDIO_HCI) */
4925 
4926 #endif/* #ifdef CONFIG_80211N_HT */
4927 
4928 }
4929 
4930 
4931 #ifdef CONFIG_80211N_HT
4932 void	rtw_ht_use_default_setting(_adapter *padapter)
4933 {
4934 	struct mlme_priv		*pmlmepriv = &padapter->mlmepriv;
4935 	struct ht_priv		*phtpriv = &pmlmepriv->htpriv;
4936 	struct registry_priv	*pregistrypriv = &padapter->registrypriv;
4937 	BOOLEAN		bHwLDPCSupport = _FALSE, bHwSTBCSupport = _FALSE;
4938 #ifdef CONFIG_BEAMFORMING
4939 	BOOLEAN		bHwSupportBeamformer = _FALSE, bHwSupportBeamformee = _FALSE;
4940 #endif /* CONFIG_BEAMFORMING */
4941 
4942 	if (pregistrypriv->wifi_spec)
4943 		phtpriv->bss_coexist = 1;
4944 	else
4945 		phtpriv->bss_coexist = 0;
4946 
4947 	phtpriv->sgi_40m = TEST_FLAG(pregistrypriv->short_gi, BIT1) ? _TRUE : _FALSE;
4948 	phtpriv->sgi_20m = TEST_FLAG(pregistrypriv->short_gi, BIT0) ? _TRUE : _FALSE;
4949 
4950 	/* LDPC support */
4951 	rtw_hal_get_def_var(padapter, HAL_DEF_RX_LDPC, (u8 *)&bHwLDPCSupport);
4952 	CLEAR_FLAGS(phtpriv->ldpc_cap);
4953 	if (bHwLDPCSupport) {
4954 		if (TEST_FLAG(pregistrypriv->ldpc_cap, BIT4))
4955 			SET_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_RX);
4956 	}
4957 	rtw_hal_get_def_var(padapter, HAL_DEF_TX_LDPC, (u8 *)&bHwLDPCSupport);
4958 	if (bHwLDPCSupport) {
4959 		if (TEST_FLAG(pregistrypriv->ldpc_cap, BIT5))
4960 			SET_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_TX);
4961 	}
4962 	if (phtpriv->ldpc_cap)
4963 		RTW_INFO("[HT] HAL Support LDPC = 0x%02X\n", phtpriv->ldpc_cap);
4964 
4965 	/* STBC */
4966 	rtw_hal_get_def_var(padapter, HAL_DEF_TX_STBC, (u8 *)&bHwSTBCSupport);
4967 	CLEAR_FLAGS(phtpriv->stbc_cap);
4968 	if (bHwSTBCSupport) {
4969 		if (TEST_FLAG(pregistrypriv->stbc_cap, BIT5))
4970 			SET_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX);
4971 	}
4972 	rtw_hal_get_def_var(padapter, HAL_DEF_RX_STBC, (u8 *)&bHwSTBCSupport);
4973 	if (bHwSTBCSupport) {
4974 		if (TEST_FLAG(pregistrypriv->stbc_cap, BIT4))
4975 			SET_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_RX);
4976 	}
4977 	if (phtpriv->stbc_cap)
4978 		RTW_INFO("[HT] HAL Support STBC = 0x%02X\n", phtpriv->stbc_cap);
4979 
4980 	/* Beamforming setting */
4981 	CLEAR_FLAGS(phtpriv->beamform_cap);
4982 #ifdef CONFIG_BEAMFORMING
4983 #ifdef RTW_BEAMFORMING_VERSION_2
4984 #ifdef CONFIG_CONCURRENT_MODE
4985 	/* only enable beamforming in STA client mode */
4986 	if (MLME_IS_STA(padapter) && !MLME_IS_GC(padapter))
4987 #else
4988 	if ((MLME_IS_AP(padapter) && !MLME_IS_GO(padapter)) ||
4989 	    (MLME_IS_STA(padapter) && !MLME_IS_GC(padapter)))
4990 #endif
4991 #endif
4992 	{
4993 		rtw_hal_get_def_var(padapter, HAL_DEF_EXPLICIT_BEAMFORMER, (u8 *)&bHwSupportBeamformer);
4994 		rtw_hal_get_def_var(padapter, HAL_DEF_EXPLICIT_BEAMFORMEE, (u8 *)&bHwSupportBeamformee);
4995 		if (TEST_FLAG(pregistrypriv->beamform_cap, BIT4) && bHwSupportBeamformer) {
4996 			SET_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE);
4997 			RTW_INFO("[HT] HAL Support Beamformer\n");
4998 		}
4999 		if (TEST_FLAG(pregistrypriv->beamform_cap, BIT5) && bHwSupportBeamformee) {
5000 			SET_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE);
5001 			RTW_INFO("[HT] HAL Support Beamformee\n");
5002 		}
5003 	}
5004 #endif /* CONFIG_BEAMFORMING */
5005 }
5006 void rtw_build_wmm_ie_ht(_adapter *padapter, u8 *out_ie, uint *pout_len)
5007 {
5008 	unsigned char WMM_IE[] = {0x00, 0x50, 0xf2, 0x02, 0x00, 0x01, 0x00};
5009 	int out_len;
5010 	u8 *pframe;
5011 
5012 	if (padapter->mlmepriv.qospriv.qos_option == 0) {
5013 		out_len = *pout_len;
5014 		pframe = rtw_set_ie(out_ie + out_len, _VENDOR_SPECIFIC_IE_,
5015 				    _WMM_IE_Length_, WMM_IE, pout_len);
5016 
5017 		padapter->mlmepriv.qospriv.qos_option = 1;
5018 	}
5019 }
5020 #if defined(CONFIG_80211N_HT)
5021 /* the fucntion is >= passive_level */
5022 unsigned int rtw_restructure_ht_ie(_adapter *padapter, u8 *in_ie, u8 *out_ie, uint in_len, uint *pout_len, u8 channel, struct country_chplan *req_chplan)
5023 {
5024 	u32 ielen, out_len;
5025 	u32 rx_packet_offset, max_recvbuf_sz;
5026 	HT_CAP_AMPDU_FACTOR max_rx_ampdu_factor;
5027 	HT_CAP_AMPDU_DENSITY best_ampdu_density;
5028 	unsigned char *p, *pframe;
5029 	BAND_TYPE band = channel > 14 ? BAND_ON_5G : BAND_ON_2_4G;
5030 	struct rtw_ieee80211_ht_cap ht_capie;
5031 	u8	cbw40_enable = 0, rf_num = 0, rx_stbc_nss = 0, rx_nss = 0;
5032 	struct registry_priv *pregistrypriv = &padapter->registrypriv;
5033 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
5034 	struct ht_priv		*phtpriv = &pmlmepriv->htpriv;
5035 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
5036 #ifdef CONFIG_80211AC_VHT
5037 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
5038 	struct vht_priv	*pvhtpriv = &pmlmepriv->vhtpriv;
5039 #endif /* CONFIG_80211AC_VHT */
5040 
5041 	phtpriv->ht_option = _FALSE;
5042 
5043 	out_len = *pout_len;
5044 
5045 	_rtw_memset(&ht_capie, 0, sizeof(struct rtw_ieee80211_ht_cap));
5046 
5047 	ht_capie.cap_info = IEEE80211_HT_CAP_DSSSCCK40;
5048 
5049 	if (phtpriv->sgi_20m)
5050 		ht_capie.cap_info |= IEEE80211_HT_CAP_SGI_20;
5051 
5052 	/* check if 40MHz is allowed according to hal cap and registry */
5053 	if (hal_chk_bw_cap(padapter, BW_CAP_40M)) {
5054 		if (band == BAND_ON_5G) {
5055 			if (REGSTY_IS_BW_5G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
5056 				cbw40_enable = 1;
5057 		} else if (band == BAND_ON_2_4G) {
5058 			if (REGSTY_IS_BW_2G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
5059 				cbw40_enable = 1;
5060 		}
5061 	}
5062 
5063 	if (cbw40_enable) {
5064 		struct rf_ctl_t *rfctl = adapter_to_rfctl(padapter);
5065 		RT_CHANNEL_INFO *chset = rfctl->channel_set;
5066 		u8 oper_bw = CHANNEL_WIDTH_20, oper_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
5067 
5068 		if (in_ie == NULL) {
5069 			/* TDLS: TODO 20/40 issue */
5070 			if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
5071 				oper_bw = padapter->mlmeextpriv.cur_bwmode;
5072 				if (oper_bw > CHANNEL_WIDTH_40)
5073 					oper_bw = CHANNEL_WIDTH_40;
5074 			} else
5075 				/* TDLS: TODO 40? */
5076 				oper_bw = CHANNEL_WIDTH_40;
5077 		} else {
5078 			p = rtw_get_ie(in_ie, WLAN_EID_HT_OPERATION, &ielen, in_len);
5079 			if (p && ielen == HT_OP_IE_LEN) {
5080 				if (GET_HT_OP_ELE_STA_CHL_WIDTH(p + 2)) {
5081 					switch (GET_HT_OP_ELE_2ND_CHL_OFFSET(p + 2)) {
5082 					case SCA:
5083 						oper_bw = CHANNEL_WIDTH_40;
5084 						oper_offset = HAL_PRIME_CHNL_OFFSET_LOWER;
5085 						break;
5086 					case SCB:
5087 						oper_bw = CHANNEL_WIDTH_40;
5088 						oper_offset = HAL_PRIME_CHNL_OFFSET_UPPER;
5089 						break;
5090 					}
5091 				}
5092 			}
5093 
5094 			/* IOT issue : AP TP-link Archer C7, correct the secondary channel offset of 5G channel */
5095 			if (oper_bw == CHANNEL_WIDTH_40 && p && ielen == HT_OP_IE_LEN) {
5096 				u8 primary_ch = GET_HT_OP_ELE_PRI_CHL(p + 2);
5097 				u8 offset_bak = oper_offset;
5098 
5099 				if (primary_ch > 14) {
5100 					rtw_get_offset_by_chbw(primary_ch, oper_bw, &oper_offset);
5101 
5102 					/* secondary channel offset is wrong */
5103 					if (offset_bak != oper_offset) {
5104 						RTW_INFO(FUNC_ADPT_FMT": secondary channel offset is wrong, so workaround\n", FUNC_ADPT_ARG(padapter));
5105 						switch (oper_offset) {
5106 						case HAL_PRIME_CHNL_OFFSET_LOWER:
5107 							SET_HT_OP_ELE_2ND_CHL_OFFSET(p + 2, SCA);
5108 							break;
5109 						case HAL_PRIME_CHNL_OFFSET_UPPER:
5110 							SET_HT_OP_ELE_2ND_CHL_OFFSET(p + 2, SCB);
5111 							break;
5112 						}
5113 					}
5114 				}
5115 			}
5116 
5117 			// IOT issue : AP TP-Link WDR6500
5118 			if(oper_bw == CHANNEL_WIDTH_40){
5119 				p = rtw_get_ie(in_ie, WLAN_EID_HT_CAP, &ielen, in_len);
5120 				if (p && ielen == HT_CAP_IE_LEN) {
5121 					oper_bw = GET_HT_CAP_ELE_CHL_WIDTH(p + 2)  ? CHANNEL_WIDTH_40 : CHANNEL_WIDTH_20;
5122 					if(oper_bw == CHANNEL_WIDTH_20)
5123 						oper_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
5124 				}
5125 			}
5126 		}
5127 
5128 		/* adjust bw to fit in channel plan setting */
5129 		if (oper_bw == CHANNEL_WIDTH_40
5130 			&& oper_offset != HAL_PRIME_CHNL_OFFSET_DONT_CARE /* check this because TDLS has no info to set offset */
5131 		) {
5132 			if ((req_chplan && !rtw_country_chplan_is_chbw_valid(req_chplan, band, channel, oper_bw, oper_offset, 1, 1, pregistrypriv))
5133 				|| (!req_chplan && !rtw_chset_is_chbw_valid(chset, channel, oper_bw, oper_offset, 1, 1))
5134 				|| (IS_DFS_SLAVE_WITH_RD(rfctl)
5135 					&& !rtw_rfctl_dfs_domain_unknown(rfctl)
5136 					&& rtw_chset_is_chbw_non_ocp(chset, channel, oper_bw, oper_offset))
5137 			) {
5138 				oper_bw = CHANNEL_WIDTH_20;
5139 				oper_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
5140 				rtw_warn_on(req_chplan && !rtw_country_chplan_is_chbw_valid(req_chplan, band, channel, oper_bw, oper_offset, 1, 1, pregistrypriv));
5141 				rtw_warn_on(!req_chplan && !rtw_chset_is_chbw_valid(chset, channel, oper_bw, oper_offset, 1, 1));
5142 				if (IS_DFS_SLAVE_WITH_RD(rfctl) && !rtw_rfctl_dfs_domain_unknown(rfctl))
5143 					rtw_warn_on(rtw_chset_is_chbw_non_ocp(chset, channel, oper_bw, oper_offset));
5144 			}
5145 		}
5146 
5147 		if (oper_bw == CHANNEL_WIDTH_40) {
5148 			ht_capie.cap_info |= IEEE80211_HT_CAP_SUP_WIDTH;
5149 			if (phtpriv->sgi_40m)
5150 				ht_capie.cap_info |= IEEE80211_HT_CAP_SGI_40;
5151 		}
5152 
5153 		cbw40_enable = oper_bw == CHANNEL_WIDTH_40 ? 1 : 0;
5154 	}
5155 
5156 	/* todo: disable SM power save mode */
5157 	ht_capie.cap_info |= IEEE80211_HT_CAP_SM_PS;
5158 
5159 	/* RX LDPC */
5160 	if (TEST_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_RX)) {
5161 		ht_capie.cap_info |= IEEE80211_HT_CAP_LDPC_CODING;
5162 		RTW_INFO("[HT] Declare supporting RX LDPC\n");
5163 	}
5164 
5165 	/* TX STBC */
5166 	if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX)) {
5167 		ht_capie.cap_info |= IEEE80211_HT_CAP_TX_STBC;
5168 		RTW_INFO("[HT] Declare supporting TX STBC\n");
5169 	}
5170 
5171 	/* RX STBC */
5172 	if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_RX)) {
5173 		if ((pregistrypriv->rx_stbc == 0x3) ||							/* enable for 2.4/5 GHz */
5174 		    ((channel <= 14) && (pregistrypriv->rx_stbc == 0x1)) ||		/* enable for 2.4GHz */
5175 		    ((channel > 14) && (pregistrypriv->rx_stbc == 0x2)) ||		/* enable for 5GHz */
5176 		    (pregistrypriv->wifi_spec == 1)) {
5177 			/* HAL_DEF_RX_STBC means STBC RX spatial stream, todo: VHT 4 streams */
5178 			rtw_hal_get_def_var(padapter, HAL_DEF_RX_STBC, (u8 *)(&rx_stbc_nss));
5179 			SET_HT_CAP_ELE_RX_STBC(&ht_capie, rx_stbc_nss);
5180 			RTW_INFO("[HT] Declare supporting RX STBC = %d\n", rx_stbc_nss);
5181 		}
5182 	}
5183 
5184 	/* fill default supported_mcs_set */
5185 	_rtw_memcpy(ht_capie.supp_mcs_set, pmlmeext->default_supported_mcs_set, 16);
5186 
5187 	/* update default supported_mcs_set */
5188 	rx_nss = GET_HAL_RX_NSS(padapter);
5189 
5190 	switch (rx_nss) {
5191 	case 1:
5192 		set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_1R);
5193 		break;
5194 	case 2:
5195 		#ifdef CONFIG_DISABLE_MCS13TO15
5196 		if (cbw40_enable && pregistrypriv->wifi_spec != 1)
5197 			set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_2R_13TO15_OFF);
5198 		else
5199 		#endif
5200 			set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_2R);
5201 		break;
5202 	case 3:
5203 		set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_3R);
5204 		break;
5205 	case 4:
5206 		set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_4R);
5207 		break;
5208 	default:
5209 		RTW_WARN("rf_type:%d or rx_nss:%u is not expected\n", GET_HAL_RFPATH(padapter), rx_nss);
5210 	}
5211 
5212 	{
5213 		rtw_hal_get_def_var(padapter, HAL_DEF_RX_PACKET_OFFSET, &rx_packet_offset);
5214 		rtw_hal_get_def_var(padapter, HAL_DEF_MAX_RECVBUF_SZ, &max_recvbuf_sz);
5215 		if (max_recvbuf_sz - rx_packet_offset >= (8191 - 256)) {
5216 			RTW_INFO("%s IEEE80211_HT_CAP_MAX_AMSDU is set\n", __FUNCTION__);
5217 			ht_capie.cap_info = ht_capie.cap_info | IEEE80211_HT_CAP_MAX_AMSDU;
5218 		}
5219 	}
5220 	/*
5221 	AMPDU_para [1:0]:Max AMPDU Len => 0:8k , 1:16k, 2:32k, 3:64k
5222 	AMPDU_para [4:2]:Min MPDU Start Spacing
5223 	*/
5224 
5225 	/*
5226 	#if defined(CONFIG_RTL8188E) && defined(CONFIG_SDIO_HCI)
5227 	ht_capie.ampdu_params_info = 2;
5228 	#else
5229 	ht_capie.ampdu_params_info = (IEEE80211_HT_CAP_AMPDU_FACTOR&0x03);
5230 	#endif
5231 	*/
5232 
5233 	if (padapter->driver_rx_ampdu_factor != 0xFF)
5234 		max_rx_ampdu_factor = (HT_CAP_AMPDU_FACTOR)padapter->driver_rx_ampdu_factor;
5235 	else
5236 		rtw_hal_get_def_var(padapter, HW_VAR_MAX_RX_AMPDU_FACTOR, &max_rx_ampdu_factor);
5237 
5238 	/* rtw_hal_get_def_var(padapter, HW_VAR_MAX_RX_AMPDU_FACTOR, &max_rx_ampdu_factor); */
5239 	ht_capie.ampdu_params_info = (max_rx_ampdu_factor & 0x03);
5240 
5241 	if (padapter->driver_rx_ampdu_spacing != 0xFF)
5242 		ht_capie.ampdu_params_info |= ((padapter->driver_rx_ampdu_spacing & 0x07) << 2);
5243 	else {
5244 		if (padapter->securitypriv.dot11PrivacyAlgrthm == _AES_) {
5245 			/*
5246 			*	Todo : Each chip must to ask DD , this chip best ampdu_density setting
5247 			*	By yiwei.sun
5248 			*/
5249 			rtw_hal_get_def_var(padapter, HW_VAR_BEST_AMPDU_DENSITY, &best_ampdu_density);
5250 
5251 			ht_capie.ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY & (best_ampdu_density << 2));
5252 
5253 		} else
5254 			ht_capie.ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY & 0x00);
5255 	}
5256 #ifdef CONFIG_BEAMFORMING
5257 	ht_capie.tx_BF_cap_info = 0;
5258 
5259 	/* HT Beamformer*/
5260 	if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE)) {
5261 		/* Transmit NDP Capable */
5262 		SET_HT_CAP_TXBF_TRANSMIT_NDP_CAP(&ht_capie, 1);
5263 		/* Explicit Compressed Steering Capable */
5264 		SET_HT_CAP_TXBF_EXPLICIT_COMP_STEERING_CAP(&ht_capie, 1);
5265 		/* Compressed Steering Number Antennas */
5266 		SET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(&ht_capie, 1);
5267 		rtw_hal_get_def_var(padapter, HAL_DEF_BEAMFORMER_CAP, (u8 *)&rf_num);
5268 		SET_HT_CAP_TXBF_CHNL_ESTIMATION_NUM_ANTENNAS(&ht_capie, rf_num);
5269 	}
5270 
5271 	/* HT Beamformee */
5272 	if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE)) {
5273 		/* Receive NDP Capable */
5274 		SET_HT_CAP_TXBF_RECEIVE_NDP_CAP(&ht_capie, 1);
5275 		/* Explicit Compressed Beamforming Feedback Capable */
5276 		SET_HT_CAP_TXBF_EXPLICIT_COMP_FEEDBACK_CAP(&ht_capie, 2);
5277 
5278 		rtw_hal_get_def_var(padapter, HAL_DEF_BEAMFORMEE_CAP, (u8 *)&rf_num);
5279 #ifdef CONFIG_80211AC_VHT
5280 		/* IOT action suggested by Yu Chen 2017/3/3 */
5281 		if ((pmlmeinfo->assoc_AP_vendor == HT_IOT_PEER_BROADCOM) &&
5282 			!pvhtpriv->ap_bf_cap.is_mu_bfer &&
5283 			pvhtpriv->ap_bf_cap.su_sound_dim == 2)
5284 			rf_num = (rf_num >= 2 ? 2 : rf_num);
5285 #endif
5286 		SET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(&ht_capie, rf_num);
5287 	}
5288 #endif/*CONFIG_BEAMFORMING*/
5289 
5290 	pframe = rtw_set_ie(out_ie + out_len, _HT_CAPABILITY_IE_,
5291 		sizeof(struct rtw_ieee80211_ht_cap), (unsigned char *)&ht_capie, pout_len);
5292 
5293 	phtpriv->ht_option = _TRUE;
5294 
5295 	if (in_ie != NULL) {
5296 		p = rtw_get_ie(in_ie, _HT_ADD_INFO_IE_, &ielen, in_len);
5297 		if (p && (ielen == sizeof(struct ieee80211_ht_addt_info))) {
5298 			out_len = *pout_len;
5299 			pframe = rtw_set_ie(out_ie + out_len, _HT_ADD_INFO_IE_, ielen, p + 2 , pout_len);
5300 		}
5301 	}
5302 
5303 	return phtpriv->ht_option;
5304 
5305 }
5306 
5307 /* the fucntion is > passive_level (in critical_section) */
5308 void rtw_update_ht_cap(_adapter *padapter, u8 *pie, uint ie_len, u8 channel)
5309 {
5310 	u8 *p, max_ampdu_sz;
5311 	int len;
5312 	/* struct sta_info *bmc_sta, *psta; */
5313 	struct rtw_ieee80211_ht_cap *pht_capie;
5314 	struct ieee80211_ht_addt_info *pht_addtinfo;
5315 	/* struct recv_reorder_ctrl *preorder_ctrl; */
5316 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
5317 	struct ht_priv		*phtpriv = &pmlmepriv->htpriv;
5318 	/* struct recv_priv *precvpriv = &padapter->recvpriv; */
5319 	struct registry_priv *pregistrypriv = &padapter->registrypriv;
5320 	/* struct wlan_network *pcur_network = &(pmlmepriv->cur_network);; */
5321 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
5322 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
5323 	u8 cbw40_enable = 0;
5324 
5325 
5326 	if (!phtpriv->ht_option)
5327 		return;
5328 
5329 	if ((!pmlmeinfo->HT_info_enable) || (!pmlmeinfo->HT_caps_enable))
5330 		return;
5331 
5332 	RTW_INFO("+rtw_update_ht_cap()\n");
5333 
5334 	/* maybe needs check if ap supports rx ampdu. */
5335 	if ((phtpriv->ampdu_enable == _FALSE) && (pregistrypriv->ampdu_enable == 1)) {
5336 		if (pregistrypriv->wifi_spec == 1) {
5337 			/* remove this part because testbed AP should disable RX AMPDU */
5338 			/* phtpriv->ampdu_enable = _FALSE; */
5339 			phtpriv->ampdu_enable = _TRUE;
5340 		} else
5341 			phtpriv->ampdu_enable = _TRUE;
5342 	}
5343 
5344 
5345 	/* check Max Rx A-MPDU Size */
5346 	len = 0;
5347 	p = rtw_get_ie(pie + sizeof(NDIS_802_11_FIXED_IEs), _HT_CAPABILITY_IE_, &len, ie_len - sizeof(NDIS_802_11_FIXED_IEs));
5348 	if (p && len > 0) {
5349 		pht_capie = (struct rtw_ieee80211_ht_cap *)(p + 2);
5350 		max_ampdu_sz = (pht_capie->ampdu_params_info & IEEE80211_HT_CAP_AMPDU_FACTOR);
5351 		max_ampdu_sz = 1 << (max_ampdu_sz + 3); /* max_ampdu_sz (kbytes); */
5352 
5353 		/* RTW_INFO("rtw_update_ht_cap(): max_ampdu_sz=%d\n", max_ampdu_sz); */
5354 		phtpriv->rx_ampdu_maxlen = max_ampdu_sz;
5355 
5356 	}
5357 
5358 
5359 	len = 0;
5360 	p = rtw_get_ie(pie + sizeof(NDIS_802_11_FIXED_IEs), _HT_ADD_INFO_IE_, &len, ie_len - sizeof(NDIS_802_11_FIXED_IEs));
5361 	if (p && len > 0) {
5362 		pht_addtinfo = (struct ieee80211_ht_addt_info *)(p + 2);
5363 		/* todo: */
5364 	}
5365 
5366 	if (hal_chk_bw_cap(padapter, BW_CAP_40M)) {
5367 		if (channel > 14) {
5368 			if (REGSTY_IS_BW_5G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
5369 				cbw40_enable = 1;
5370 		} else {
5371 			if (REGSTY_IS_BW_2G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
5372 				cbw40_enable = 1;
5373 		}
5374 	}
5375 
5376 	/* update cur_bwmode & cur_ch_offset */
5377 	if ((cbw40_enable) &&
5378 	    (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & BIT(1)) &&
5379 	    (pmlmeinfo->HT_info.infos[0] & BIT(2))) {
5380 		int i;
5381 		u8 rx_nss = 0;
5382 
5383 		rx_nss = GET_HAL_RX_NSS(padapter);
5384 
5385 		/* update the MCS set */
5386 		for (i = 0; i < 16; i++)
5387 			pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate[i] &= pmlmeext->default_supported_mcs_set[i];
5388 
5389 		/* update the MCS rates */
5390 		switch (rx_nss) {
5391 		case 1:
5392 			set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_1R);
5393 			break;
5394 		case 2:
5395 			#ifdef CONFIG_DISABLE_MCS13TO15
5396 			if (pmlmeext->cur_bwmode == CHANNEL_WIDTH_40 && pregistrypriv->wifi_spec != 1)
5397 				set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R_13TO15_OFF);
5398 			else
5399 			#endif
5400 				set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R);
5401 			break;
5402 		case 3:
5403 			set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_3R);
5404 			break;
5405 		case 4:
5406 			set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_4R);
5407 			break;
5408 		default:
5409 			RTW_WARN("rx_nss:%u is not expected\n", rx_nss);
5410 		}
5411 
5412 		/* switch to the 40M Hz mode accoring to the AP */
5413 		/* pmlmeext->cur_bwmode = CHANNEL_WIDTH_40; */
5414 		switch ((pmlmeinfo->HT_info.infos[0] & 0x3)) {
5415 		case EXTCHNL_OFFSET_UPPER:
5416 			pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_LOWER;
5417 			break;
5418 
5419 		case EXTCHNL_OFFSET_LOWER:
5420 			pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_UPPER;
5421 			break;
5422 
5423 		default:
5424 			pmlmeext->cur_bwmode = CHANNEL_WIDTH_20;
5425 			pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
5426 			RTW_INFO("%s : ch offset is not assigned for HT40 mod , update cur_bwmode=%u, cur_ch_offset=%u\n",
5427 					__func__, pmlmeext->cur_bwmode, pmlmeext->cur_ch_offset);
5428 			break;
5429 		}
5430 	}
5431 
5432 	/*  */
5433 	/* Config SM Power Save setting */
5434 	/*  */
5435 	pmlmeinfo->SM_PS = (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & 0x0C) >> 2;
5436 	if (pmlmeinfo->SM_PS == WLAN_HT_CAP_SM_PS_STATIC) {
5437 #if 0
5438 		u8 i;
5439 		/* update the MCS rates */
5440 		for (i = 0; i < 16; i++)
5441 			pmlmeinfo->HT_caps.HT_cap_element.MCS_rate[i] &= MCS_rate_1R[i];
5442 #endif
5443 		RTW_INFO("%s(): WLAN_HT_CAP_SM_PS_STATIC\n", __FUNCTION__);
5444 	}
5445 
5446 	/*  */
5447 	/* Config current HT Protection mode. */
5448 	/*  */
5449 	pmlmeinfo->HT_protection = pmlmeinfo->HT_info.infos[1] & 0x3;
5450 }
5451 #endif
5452 
5453 #ifdef CONFIG_TDLS
5454 void rtw_issue_addbareq_cmd_tdls(_adapter *padapter, struct xmit_frame *pxmitframe)
5455 {
5456 	struct pkt_attrib *pattrib = &pxmitframe->attrib;
5457 	struct sta_info *ptdls_sta = NULL;
5458 	u8 issued;
5459 	int priority;
5460 	struct ht_priv	*phtpriv;
5461 
5462 	priority = pattrib->priority;
5463 
5464 	if (pattrib->direct_link == _TRUE) {
5465 		ptdls_sta = rtw_get_stainfo(&padapter->stapriv, pattrib->dst);
5466 		if ((ptdls_sta != NULL) && (ptdls_sta->tdls_sta_state & TDLS_LINKED_STATE)) {
5467 			phtpriv = &ptdls_sta->htpriv;
5468 
5469 			if ((phtpriv->ht_option == _TRUE) && (phtpriv->ampdu_enable == _TRUE)) {
5470 				issued = (phtpriv->agg_enable_bitmap >> priority) & 0x1;
5471 				issued |= (phtpriv->candidate_tid_bitmap >> priority) & 0x1;
5472 
5473 				if (0 == issued) {
5474 					RTW_INFO("[%s], p=%d\n", __FUNCTION__, priority);
5475 					ptdls_sta->htpriv.candidate_tid_bitmap |= BIT((u8)priority);
5476 					rtw_addbareq_cmd(padapter, (u8)priority, pattrib->dst);
5477 				}
5478 			}
5479 		}
5480 	}
5481 }
5482 #endif /* CONFIG_TDLS */
5483 
5484 #ifdef CONFIG_80211N_HT
5485 static u8 rtw_issue_addbareq_check(_adapter *padapter, struct xmit_frame *pxmitframe, u8 issue_when_busy)
5486 {
5487 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
5488 	struct registry_priv *pregistry = &padapter->registrypriv;
5489 	struct pkt_attrib *pattrib = &pxmitframe->attrib;
5490 	s32 bmcst = IS_MCAST(pattrib->ra);
5491 
5492 	if (bmcst)
5493 		return _FALSE;
5494 
5495 	if (pregistry->tx_quick_addba_req == 0) {
5496 		if ((issue_when_busy == _TRUE) && (pmlmepriv->LinkDetectInfo.bBusyTraffic == _FALSE))
5497 			return _FALSE;
5498 
5499 		if (pmlmepriv->LinkDetectInfo.NumTxOkInPeriod < 100)
5500 			return _FALSE;
5501 	}
5502 
5503 	return _TRUE;
5504 }
5505 
5506 void rtw_issue_addbareq_cmd(_adapter *padapter, struct xmit_frame *pxmitframe, u8 issue_when_busy)
5507 {
5508 	u8 issued;
5509 	int priority;
5510 	struct sta_info *psta = NULL;
5511 	struct ht_priv	*phtpriv;
5512 	struct pkt_attrib *pattrib = &pxmitframe->attrib;
5513 
5514 	if (rtw_issue_addbareq_check(padapter,pxmitframe, issue_when_busy) == _FALSE)
5515 		return;
5516 
5517 	priority = pattrib->priority;
5518 
5519 #ifdef CONFIG_TDLS
5520 	rtw_issue_addbareq_cmd_tdls(padapter, pxmitframe);
5521 #endif /* CONFIG_TDLS */
5522 
5523 	psta = rtw_get_stainfo(&padapter->stapriv, pattrib->ra);
5524 	if (pattrib->psta != psta) {
5525 		RTW_INFO("%s, pattrib->psta(%p) != psta(%p)\n", __func__, pattrib->psta, psta);
5526 		return;
5527 	}
5528 
5529 	if (psta == NULL) {
5530 		RTW_INFO("%s, psta==NUL\n", __func__);
5531 		return;
5532 	}
5533 
5534 	if (!(psta->state & WIFI_ASOC_STATE)) {
5535 		RTW_INFO("%s, psta->state(0x%x) != WIFI_ASOC_STATE\n", __func__, psta->state);
5536 		return;
5537 	}
5538 
5539 
5540 	phtpriv = &psta->htpriv;
5541 
5542 	if ((phtpriv->ht_option == _TRUE) && (phtpriv->ampdu_enable == _TRUE)) {
5543 		issued = (phtpriv->agg_enable_bitmap >> priority) & 0x1;
5544 		issued |= (phtpriv->candidate_tid_bitmap >> priority) & 0x1;
5545 
5546 		if (0 == issued) {
5547 			RTW_INFO("rtw_issue_addbareq_cmd, p=%d\n", priority);
5548 			psta->htpriv.candidate_tid_bitmap |= BIT((u8)priority);
5549 			rtw_addbareq_cmd(padapter, (u8) priority, pattrib->ra);
5550 		}
5551 	}
5552 
5553 }
5554 #endif /* CONFIG_80211N_HT */
5555 void rtw_append_extended_cap(_adapter *padapter, u8 *out_ie, uint *pout_len)
5556 {
5557 	struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
5558 	struct ht_priv *phtpriv = &pmlmepriv->htpriv;
5559 #ifdef CONFIG_80211AC_VHT
5560 	struct vht_priv *pvhtpriv = &pmlmepriv->vhtpriv;
5561 #endif /* CONFIG_80211AC_VHT */
5562 	u8 *ext_cap_data = pmlmepriv->ext_capab_ie_data;
5563 	u8 *ext_cap_data_len = &(pmlmepriv->ext_capab_ie_len);
5564 
5565 	if (phtpriv->bss_coexist)
5566 		rtw_add_ext_cap_info(ext_cap_data, ext_cap_data_len, BSS_COEXT);
5567 #ifdef CONFIG_ECSA
5568 	rtw_add_ext_cap_info(ext_cap_data, ext_cap_data_len, EXT_CH_SWITCH);
5569 #endif
5570 #ifdef CONFIG_80211AC_VHT
5571 	if (pvhtpriv->vht_option)
5572 		rtw_add_ext_cap_info(ext_cap_data, ext_cap_data_len, OP_MODE_NOTIFICATION);
5573 #endif /* CONFIG_80211AC_VHT */
5574 #ifdef CONFIG_RTW_WNM
5575 	rtw_add_ext_cap_info(ext_cap_data, ext_cap_data_len, BSS_TRANSITION);
5576 #endif
5577 
5578 #ifdef CONFIG_RTW_MBO
5579 	rtw_add_ext_cap_info(ext_cap_data, ext_cap_data_len, INTERWORKING);
5580 #endif
5581 
5582 	/*
5583 		From 802.11 specification,if a STA does not support any of capabilities defined
5584 		in the Extended Capabilities element, then the STA is not required to
5585 		transmit the Extended Capabilities element.
5586 	*/
5587 	rtw_update_ext_cap_ie(ext_cap_data, *ext_cap_data_len, out_ie, pout_len, _BEACON_IE_OFFSET_);
5588 }
5589 #endif
5590 
5591 #ifdef CONFIG_LAYER2_ROAMING
5592 inline void rtw_set_to_roam(_adapter *adapter, u8 to_roam)
5593 {
5594 	if (to_roam == 0)
5595 		adapter->mlmepriv.to_join = _FALSE;
5596 	adapter->mlmepriv.to_roam = to_roam;
5597 }
5598 
5599 inline u8 rtw_dec_to_roam(_adapter *adapter)
5600 {
5601 	adapter->mlmepriv.to_roam--;
5602 	return adapter->mlmepriv.to_roam;
5603 }
5604 
5605 inline u8 rtw_to_roam(_adapter *adapter)
5606 {
5607 	return adapter->mlmepriv.to_roam;
5608 }
5609 
5610 void rtw_roaming(_adapter *padapter, struct wlan_network *tgt_network)
5611 {
5612 	_irqL irqL;
5613 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
5614 
5615 	_enter_critical_bh(&pmlmepriv->lock, &irqL);
5616 	_rtw_roaming(padapter, tgt_network);
5617 	_exit_critical_bh(&pmlmepriv->lock, &irqL);
5618 }
5619 void _rtw_roaming(_adapter *padapter, struct wlan_network *tgt_network)
5620 {
5621 	struct mlme_priv	*pmlmepriv = &padapter->mlmepriv;
5622 	struct wlan_network *cur_network = &pmlmepriv->cur_network;
5623 	int do_join_r;
5624 
5625 	if (0 < rtw_to_roam(padapter)) {
5626 		RTW_INFO("roaming from %s("MAC_FMT"), length:%d\n",
5627 			cur_network->network.Ssid.Ssid, MAC_ARG(cur_network->network.MacAddress),
5628 			 cur_network->network.Ssid.SsidLength);
5629 		_rtw_memcpy(&pmlmepriv->assoc_ssid, &cur_network->network.Ssid, sizeof(NDIS_802_11_SSID));
5630 		pmlmepriv->assoc_ch = 0;
5631 		pmlmepriv->assoc_by_bssid = _FALSE;
5632 
5633 #ifdef CONFIG_WAPI_SUPPORT
5634 		rtw_wapi_return_all_sta_info(padapter);
5635 #endif
5636 
5637 		while (1) {
5638 			do_join_r = rtw_do_join(padapter);
5639 			if (_SUCCESS == do_join_r)
5640 				break;
5641 			else {
5642 				RTW_INFO("roaming do_join return %d\n", do_join_r);
5643 				rtw_dec_to_roam(padapter);
5644 
5645 				if (rtw_to_roam(padapter) > 0)
5646 					continue;
5647 				else {
5648 					RTW_INFO("%s(%d) -to roaming fail, indicate_disconnect\n", __FUNCTION__, __LINE__);
5649 #ifdef CONFIG_RTW_80211R
5650 					rtw_ft_clr_flags(padapter, RTW_FT_PEER_EN|RTW_FT_PEER_OTD_EN);
5651 					rtw_ft_reset_status(padapter);
5652 #endif
5653 					rtw_indicate_disconnect(padapter, 0, _FALSE);
5654 					break;
5655 				}
5656 			}
5657 		}
5658 	}
5659 
5660 }
5661 #endif /* CONFIG_LAYER2_ROAMING */
5662 
5663 bool rtw_adjust_chbw(_adapter *adapter, u8 req_ch, u8 *req_bw, u8 *req_offset)
5664 {
5665 	struct registry_priv *regsty = adapter_to_regsty(adapter);
5666 	u8 allowed_bw;
5667 
5668 	if (req_ch < 14)
5669 		allowed_bw = REGSTY_BW_2G(regsty);
5670 	else if (req_ch == 14)
5671 		allowed_bw = CHANNEL_WIDTH_20;
5672 	else
5673 		allowed_bw = REGSTY_BW_5G(regsty);
5674 
5675 	allowed_bw = hal_largest_bw(adapter, allowed_bw);
5676 
5677 	if (allowed_bw == CHANNEL_WIDTH_80 && *req_bw > CHANNEL_WIDTH_80)
5678 		*req_bw = CHANNEL_WIDTH_80;
5679 	else if (allowed_bw == CHANNEL_WIDTH_40 && *req_bw > CHANNEL_WIDTH_40)
5680 		*req_bw = CHANNEL_WIDTH_40;
5681 	else if (allowed_bw == CHANNEL_WIDTH_20 && *req_bw > CHANNEL_WIDTH_20) {
5682 		*req_bw = CHANNEL_WIDTH_20;
5683 		*req_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
5684 	} else
5685 		return _FALSE;
5686 
5687 	return _TRUE;
5688 }
5689 
5690 sint rtw_linked_check(_adapter *padapter)
5691 {
5692 	if (MLME_IS_AP(padapter) || MLME_IS_MESH(padapter)
5693 		|| MLME_IS_ADHOC(padapter) || MLME_IS_ADHOC_MASTER(padapter)
5694 	) {
5695 		if (padapter->stapriv.asoc_sta_count > 2)
5696 			return _TRUE;
5697 	} else {
5698 		/* Station mode */
5699 		if (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE) == _TRUE)
5700 			return _TRUE;
5701 	}
5702 	return _FALSE;
5703 }
5704 /*#define DBG_ADAPTER_STATE_CHK*/
5705 u8 rtw_is_adapter_up(_adapter *padapter)
5706 {
5707 	if (padapter == NULL)
5708 		return _FALSE;
5709 
5710 	if (RTW_CANNOT_RUN(padapter)) {
5711 		#ifdef DBG_ADAPTER_STATE_CHK
5712 		RTW_INFO(FUNC_ADPT_FMT " FALSE -bDriverStopped(%s) bSurpriseRemoved(%s)\n"
5713 			, FUNC_ADPT_ARG(padapter)
5714 			, rtw_is_drv_stopped(padapter) ? "True" : "False"
5715 			, rtw_is_surprise_removed(padapter) ? "True" : "False");
5716 		#endif
5717 		return _FALSE;
5718 	}
5719 
5720 	if (!rtw_is_hw_init_completed(padapter)) {
5721 		#ifdef DBG_ADAPTER_STATE_CHK
5722 		RTW_INFO(FUNC_ADPT_FMT " FALSE -(hw_init_completed == _FALSE)\n", FUNC_ADPT_ARG(padapter));
5723 		#endif
5724 		return _FALSE;
5725 	}
5726 
5727 	if (padapter->bup == _FALSE) {
5728 		#ifdef DBG_ADAPTER_STATE_CHK
5729 		RTW_INFO(FUNC_ADPT_FMT " FALSE -(bup == _FALSE)\n", FUNC_ADPT_ARG(padapter));
5730 		#endif
5731 		return _FALSE;
5732 	}
5733 
5734 	return _TRUE;
5735 }
5736 
5737 bool is_miracast_enabled(_adapter *adapter)
5738 {
5739 	bool enabled = 0;
5740 #ifdef CONFIG_WFD
5741 	struct wifi_display_info *wfdinfo = &adapter->wfd_info;
5742 
5743 	enabled = (wfdinfo->stack_wfd_mode & (MIRACAST_SOURCE | MIRACAST_SINK))
5744 		  || (wfdinfo->op_wfd_mode & (MIRACAST_SOURCE | MIRACAST_SINK));
5745 #endif
5746 
5747 	return enabled;
5748 }
5749 
5750 bool rtw_chk_miracast_mode(_adapter *adapter, u8 mode)
5751 {
5752 	bool ret = 0;
5753 #ifdef CONFIG_WFD
5754 	struct wifi_display_info *wfdinfo = &adapter->wfd_info;
5755 
5756 	ret = (wfdinfo->stack_wfd_mode & mode) || (wfdinfo->op_wfd_mode & mode);
5757 #endif
5758 
5759 	return ret;
5760 }
5761 
5762 const char *get_miracast_mode_str(int mode)
5763 {
5764 	if (mode == MIRACAST_SOURCE)
5765 		return "SOURCE";
5766 	else if (mode == MIRACAST_SINK)
5767 		return "SINK";
5768 	else if (mode == (MIRACAST_SOURCE | MIRACAST_SINK))
5769 		return "SOURCE&SINK";
5770 	else if (mode == MIRACAST_DISABLED)
5771 		return "DISABLED";
5772 	else
5773 		return "INVALID";
5774 }
5775 
5776 #ifdef CONFIG_WFD
5777 static bool wfd_st_match_rule(_adapter *adapter, u8 *local_naddr, u8 *local_port, u8 *remote_naddr, u8 *remote_port)
5778 {
5779 	struct wifi_display_info *wfdinfo = &adapter->wfd_info;
5780 
5781 	if (ntohs(*((u16 *)local_port)) == wfdinfo->rtsp_ctrlport
5782 	    || ntohs(*((u16 *)local_port)) == wfdinfo->tdls_rtsp_ctrlport
5783 	    || ntohs(*((u16 *)remote_port)) == wfdinfo->peer_rtsp_ctrlport)
5784 		return _TRUE;
5785 	return _FALSE;
5786 }
5787 
5788 static struct st_register wfd_st_reg = {
5789 	.s_proto = 0x06,
5790 	.rule = wfd_st_match_rule,
5791 };
5792 #endif /* CONFIG_WFD */
5793 
5794 inline void rtw_wfd_st_switch(struct sta_info *sta, bool on)
5795 {
5796 #ifdef CONFIG_WFD
5797 	if (on)
5798 		rtw_st_ctl_register(&sta->st_ctl, SESSION_TRACKER_REG_ID_WFD, &wfd_st_reg);
5799 	else
5800 		rtw_st_ctl_unregister(&sta->st_ctl, SESSION_TRACKER_REG_ID_WFD);
5801 #endif
5802 }
5803 
5804 void dump_arp_pkt(void *sel, u8 *da, u8 *sa, u8 *arp, bool tx)
5805 {
5806 	RTW_PRINT_SEL(sel, "%s ARP da="MAC_FMT", sa="MAC_FMT"\n"
5807 		, tx ? "send" : "recv", MAC_ARG(da), MAC_ARG(sa));
5808 	RTW_PRINT_SEL(sel, "htype=%u, ptype=0x%04x, hlen=%u, plen=%u, oper=%u\n"
5809 		, GET_ARP_HTYPE(arp), GET_ARP_PTYPE(arp), GET_ARP_HLEN(arp)
5810 		, GET_ARP_PLEN(arp), GET_ARP_OPER(arp));
5811 	RTW_PRINT_SEL(sel, "sha="MAC_FMT", spa="IP_FMT"\n"
5812 		, MAC_ARG(ARP_SENDER_MAC_ADDR(arp)), IP_ARG(ARP_SENDER_IP_ADDR(arp)));
5813 	RTW_PRINT_SEL(sel, "tha="MAC_FMT", tpa="IP_FMT"\n"
5814 		, MAC_ARG(ARP_TARGET_MAC_ADDR(arp)), IP_ARG(ARP_TARGET_IP_ADDR(arp)));
5815 }
5816 
5817