1 /******************************************************************************
2  *
3  * Copyright(c) 2007 - 2012 Realtek Corporation. All rights reserved.
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  * You should have received a copy of the GNU General Public License along with
15  * this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA
17  *
18  *
19  ******************************************************************************/
20 #define _RTW_AP_C_
21 
22 #include <osdep_service.h>
23 #include <drv_types.h>
24 #include <wifi.h>
25 #include <ieee80211.h>
26 #include <asm/unaligned.h>
27 
28 #ifdef CONFIG_88EU_AP_MODE
29 
init_mlme_ap_info(struct adapter * padapter)30 void init_mlme_ap_info(struct adapter *padapter)
31 {
32 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
33 	struct sta_priv *pstapriv = &padapter->stapriv;
34 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
35 
36 
37 	spin_lock_init(&pmlmepriv->bcn_update_lock);
38 
39 	/* for ACL */
40 	_rtw_init_queue(&pacl_list->acl_node_q);
41 
42 	start_ap_mode(padapter);
43 }
44 
free_mlme_ap_info(struct adapter * padapter)45 void free_mlme_ap_info(struct adapter *padapter)
46 {
47 	struct sta_info *psta = NULL;
48 	struct sta_priv *pstapriv = &padapter->stapriv;
49 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
50 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
51 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
52 
53 	pmlmepriv->update_bcn = false;
54 	pmlmeext->bstart_bss = false;
55 
56 	rtw_sta_flush(padapter);
57 
58 	pmlmeinfo->state = _HW_STATE_NOLINK_;
59 
60 	/* free_assoc_sta_resources */
61 	rtw_free_all_stainfo(padapter);
62 
63 	/* free bc/mc sta_info */
64 	psta = rtw_get_bcmc_stainfo(padapter);
65 	spin_lock_bh(&(pstapriv->sta_hash_lock));
66 	rtw_free_stainfo(padapter, psta);
67 	spin_unlock_bh(&(pstapriv->sta_hash_lock));
68 }
69 
update_BCNTIM(struct adapter * padapter)70 static void update_BCNTIM(struct adapter *padapter)
71 {
72 	struct sta_priv *pstapriv = &padapter->stapriv;
73 	struct mlme_ext_priv *pmlmeext = &(padapter->mlmeextpriv);
74 	struct mlme_ext_info *pmlmeinfo = &(pmlmeext->mlmext_info);
75 	struct wlan_bssid_ex *pnetwork_mlmeext = &(pmlmeinfo->network);
76 	unsigned char *pie = pnetwork_mlmeext->IEs;
77 
78 	/* update TIM IE */
79 	if (true) {
80 		u8 *p, *dst_ie, *premainder_ie = NULL;
81 		u8 *pbackup_remainder_ie = NULL;
82 		uint offset, tmp_len, tim_ielen, tim_ie_offset, remainder_ielen;
83 
84 		p = rtw_get_ie(pie + _FIXED_IE_LENGTH_, _TIM_IE_, &tim_ielen, pnetwork_mlmeext->IELength - _FIXED_IE_LENGTH_);
85 		if (p != NULL && tim_ielen > 0) {
86 			tim_ielen += 2;
87 			premainder_ie = p+tim_ielen;
88 			tim_ie_offset = (int)(p - pie);
89 			remainder_ielen = pnetwork_mlmeext->IELength - tim_ie_offset - tim_ielen;
90 			/* append TIM IE from dst_ie offset */
91 			dst_ie = p;
92 		} else {
93 			tim_ielen = 0;
94 
95 			/* calculate head_len */
96 			offset = _FIXED_IE_LENGTH_;
97 			offset += pnetwork_mlmeext->Ssid.SsidLength + 2;
98 
99 			/*  get supported rates len */
100 			p = rtw_get_ie(pie + _BEACON_IE_OFFSET_, _SUPPORTEDRATES_IE_, &tmp_len, (pnetwork_mlmeext->IELength - _BEACON_IE_OFFSET_));
101 			if (p !=  NULL)
102 				offset += tmp_len+2;
103 
104 			/* DS Parameter Set IE, len = 3 */
105 			offset += 3;
106 
107 			premainder_ie = pie + offset;
108 
109 			remainder_ielen = pnetwork_mlmeext->IELength - offset - tim_ielen;
110 
111 			/* append TIM IE from offset */
112 			dst_ie = pie + offset;
113 		}
114 
115 		if (remainder_ielen > 0) {
116 			pbackup_remainder_ie = rtw_malloc(remainder_ielen);
117 			if (pbackup_remainder_ie && premainder_ie)
118 				memcpy(pbackup_remainder_ie, premainder_ie, remainder_ielen);
119 		}
120 		*dst_ie++ = _TIM_IE_;
121 
122 		if ((pstapriv->tim_bitmap&0xff00) && (pstapriv->tim_bitmap&0x00fc))
123 			tim_ielen = 5;
124 		else
125 			tim_ielen = 4;
126 
127 		*dst_ie++ = tim_ielen;
128 
129 		*dst_ie++ = 0;/* DTIM count */
130 		*dst_ie++ = 1;/* DTIM period */
131 
132 		if (pstapriv->tim_bitmap&BIT(0))/* for bc/mc frames */
133 			*dst_ie++ = BIT(0);/* bitmap ctrl */
134 		else
135 			*dst_ie++ = 0;
136 
137 		if (tim_ielen == 4) {
138 			*dst_ie++ = pstapriv->tim_bitmap & 0xff;
139 		} else if (tim_ielen == 5) {
140 			put_unaligned_le16(pstapriv->tim_bitmap, dst_ie);
141 			dst_ie += 2;
142 		}
143 
144 		/* copy remainder IE */
145 		if (pbackup_remainder_ie) {
146 			memcpy(dst_ie, pbackup_remainder_ie, remainder_ielen);
147 
148 			kfree(pbackup_remainder_ie);
149 		}
150 		offset =  (uint)(dst_ie - pie);
151 		pnetwork_mlmeext->IELength = offset + remainder_ielen;
152 	}
153 
154 	set_tx_beacon_cmd(padapter);
155 }
156 
rtw_add_bcn_ie(struct adapter * padapter,struct wlan_bssid_ex * pnetwork,u8 index,u8 * data,u8 len)157 void rtw_add_bcn_ie(struct adapter *padapter, struct wlan_bssid_ex *pnetwork, u8 index, u8 *data, u8 len)
158 {
159 	struct ndis_802_11_var_ie *pIE;
160 	u8 bmatch = false;
161 	u8 *pie = pnetwork->IEs;
162 	u8 *p = NULL, *dst_ie = NULL, *premainder_ie = NULL;
163 	u8 *pbackup_remainder_ie = NULL;
164 	u32 i, offset, ielen = 0, ie_offset, remainder_ielen = 0;
165 
166 	for (i = sizeof(struct ndis_802_11_fixed_ie); i < pnetwork->IELength;) {
167 		pIE = (struct ndis_802_11_var_ie *)(pnetwork->IEs + i);
168 
169 		if (pIE->ElementID > index) {
170 			break;
171 		} else if (pIE->ElementID == index) { /*  already exist the same IE */
172 			p = (u8 *)pIE;
173 			ielen = pIE->Length;
174 			bmatch = true;
175 			break;
176 		}
177 		p = (u8 *)pIE;
178 		ielen = pIE->Length;
179 		i += (pIE->Length + 2);
180 	}
181 
182 	if (p != NULL && ielen > 0) {
183 		ielen += 2;
184 
185 		premainder_ie = p+ielen;
186 
187 		ie_offset = (int)(p - pie);
188 
189 		remainder_ielen = pnetwork->IELength - ie_offset - ielen;
190 
191 		if (bmatch)
192 			dst_ie = p;
193 		else
194 			dst_ie = (p+ielen);
195 	}
196 
197 	if (remainder_ielen > 0) {
198 		pbackup_remainder_ie = rtw_malloc(remainder_ielen);
199 		if (pbackup_remainder_ie && premainder_ie)
200 			memcpy(pbackup_remainder_ie, premainder_ie, remainder_ielen);
201 	}
202 
203 	*dst_ie++ = index;
204 	*dst_ie++ = len;
205 
206 	memcpy(dst_ie, data, len);
207 	dst_ie += len;
208 
209 	/* copy remainder IE */
210 	if (pbackup_remainder_ie) {
211 		memcpy(dst_ie, pbackup_remainder_ie, remainder_ielen);
212 
213 		kfree(pbackup_remainder_ie);
214 	}
215 
216 	offset =  (uint)(dst_ie - pie);
217 	pnetwork->IELength = offset + remainder_ielen;
218 }
219 
rtw_remove_bcn_ie(struct adapter * padapter,struct wlan_bssid_ex * pnetwork,u8 index)220 void rtw_remove_bcn_ie(struct adapter *padapter, struct wlan_bssid_ex *pnetwork, u8 index)
221 {
222 	u8 *p, *dst_ie = NULL, *premainder_ie = NULL;
223 	u8 *pbackup_remainder_ie = NULL;
224 	uint offset, ielen, ie_offset, remainder_ielen = 0;
225 	u8	*pie = pnetwork->IEs;
226 
227 	p = rtw_get_ie(pie + _FIXED_IE_LENGTH_, index, &ielen,
228 		       pnetwork->IELength - _FIXED_IE_LENGTH_);
229 	if (p != NULL && ielen > 0) {
230 		ielen += 2;
231 
232 		premainder_ie = p+ielen;
233 
234 		ie_offset = (int)(p - pie);
235 
236 		remainder_ielen = pnetwork->IELength - ie_offset - ielen;
237 
238 		dst_ie = p;
239 	}
240 
241 	if (remainder_ielen > 0) {
242 		pbackup_remainder_ie = rtw_malloc(remainder_ielen);
243 		if (pbackup_remainder_ie && premainder_ie)
244 			memcpy(pbackup_remainder_ie, premainder_ie, remainder_ielen);
245 	}
246 
247 	/* copy remainder IE */
248 	if (pbackup_remainder_ie) {
249 		memcpy(dst_ie, pbackup_remainder_ie, remainder_ielen);
250 
251 		kfree(pbackup_remainder_ie);
252 	}
253 
254 	offset =  (uint)(dst_ie - pie);
255 	pnetwork->IELength = offset + remainder_ielen;
256 }
257 
chk_sta_is_alive(struct sta_info * psta)258 static u8 chk_sta_is_alive(struct sta_info *psta)
259 {
260 	u8 ret = false;
261 
262 	if ((psta->sta_stats.last_rx_data_pkts + psta->sta_stats.last_rx_ctrl_pkts) ==
263 	    (psta->sta_stats.rx_data_pkts + psta->sta_stats.rx_ctrl_pkts))
264 		;
265 	else
266 		ret = true;
267 
268 	sta_update_last_rx_pkts(psta);
269 
270 	return ret;
271 }
272 
expire_timeout_chk(struct adapter * padapter)273 void	expire_timeout_chk(struct adapter *padapter)
274 {
275 	struct list_head *phead, *plist;
276 	u8 updated = 0;
277 	struct sta_info *psta = NULL;
278 	struct sta_priv *pstapriv = &padapter->stapriv;
279 	u8 chk_alive_num = 0;
280 	char chk_alive_list[NUM_STA];
281 	int i;
282 
283 	spin_lock_bh(&pstapriv->auth_list_lock);
284 
285 	phead = &pstapriv->auth_list;
286 	plist = phead->next;
287 
288 	/* check auth_queue */
289 	while (phead != plist) {
290 		psta = container_of(plist, struct sta_info, auth_list);
291 		plist = plist->next;
292 
293 		if (psta->expire_to > 0) {
294 			psta->expire_to--;
295 			if (psta->expire_to == 0) {
296 				list_del_init(&psta->auth_list);
297 				pstapriv->auth_list_cnt--;
298 
299 				DBG_88E("auth expire %6ph\n",
300 					psta->hwaddr);
301 
302 				spin_unlock_bh(&pstapriv->auth_list_lock);
303 
304 				spin_lock_bh(&(pstapriv->sta_hash_lock));
305 				rtw_free_stainfo(padapter, psta);
306 				spin_unlock_bh(&(pstapriv->sta_hash_lock));
307 
308 				spin_lock_bh(&pstapriv->auth_list_lock);
309 			}
310 		}
311 
312 	}
313 	spin_unlock_bh(&pstapriv->auth_list_lock);
314 
315 	psta = NULL;
316 
317 	spin_lock_bh(&pstapriv->asoc_list_lock);
318 
319 	phead = &pstapriv->asoc_list;
320 	plist = phead->next;
321 
322 	/* check asoc_queue */
323 	while (phead != plist) {
324 		psta = container_of(plist, struct sta_info, asoc_list);
325 		plist = plist->next;
326 
327 		if (chk_sta_is_alive(psta) || !psta->expire_to) {
328 			psta->expire_to = pstapriv->expire_to;
329 			psta->keep_alive_trycnt = 0;
330 			psta->under_exist_checking = 0;
331 		} else {
332 			psta->expire_to--;
333 		}
334 
335 		if (psta->expire_to <= 0) {
336 			struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
337 
338 			if (padapter->registrypriv.wifi_spec == 1) {
339 				psta->expire_to = pstapriv->expire_to;
340 				continue;
341 			}
342 
343 			if (psta->state & WIFI_SLEEP_STATE) {
344 				if (!(psta->state & WIFI_STA_ALIVE_CHK_STATE)) {
345 					/* to check if alive by another methods if station is at ps mode. */
346 					psta->expire_to = pstapriv->expire_to;
347 					psta->state |= WIFI_STA_ALIVE_CHK_STATE;
348 
349 					/* to update bcn with tim_bitmap for this station */
350 					pstapriv->tim_bitmap |= BIT(psta->aid);
351 					update_beacon(padapter, _TIM_IE_, NULL, false);
352 
353 					if (!pmlmeext->active_keep_alive_check)
354 						continue;
355 				}
356 			}
357 			if (pmlmeext->active_keep_alive_check) {
358 				int stainfo_offset;
359 
360 				stainfo_offset = rtw_stainfo_offset(pstapriv, psta);
361 				if (stainfo_offset_valid(stainfo_offset))
362 					chk_alive_list[chk_alive_num++] = stainfo_offset;
363 				continue;
364 			}
365 
366 			list_del_init(&psta->asoc_list);
367 			pstapriv->asoc_list_cnt--;
368 
369 			DBG_88E("asoc expire %pM, state = 0x%x\n", (psta->hwaddr), psta->state);
370 			updated = ap_free_sta(padapter, psta, true, WLAN_REASON_DEAUTH_LEAVING);
371 		} else {
372 			/* TODO: Aging mechanism to digest frames in sleep_q to avoid running out of xmitframe */
373 			if (psta->sleepq_len > (NR_XMITFRAME/pstapriv->asoc_list_cnt) &&
374 			    padapter->xmitpriv.free_xmitframe_cnt < (NR_XMITFRAME/pstapriv->asoc_list_cnt/2)) {
375 				DBG_88E("%s sta:%pM, sleepq_len:%u, free_xmitframe_cnt:%u, asoc_list_cnt:%u, clear sleep_q\n", __func__,
376 					(psta->hwaddr), psta->sleepq_len,
377 					padapter->xmitpriv.free_xmitframe_cnt,
378 					pstapriv->asoc_list_cnt);
379 				wakeup_sta_to_xmit(padapter, psta);
380 			}
381 		}
382 	}
383 
384 	spin_unlock_bh(&pstapriv->asoc_list_lock);
385 
386 	if (chk_alive_num) {
387 		u8 backup_oper_channel = 0;
388 		struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
389 		/* switch to correct channel of current network  before issue keep-alive frames */
390 		if (rtw_get_oper_ch(padapter) != pmlmeext->cur_channel) {
391 			backup_oper_channel = rtw_get_oper_ch(padapter);
392 			SelectChannel(padapter, pmlmeext->cur_channel);
393 		}
394 
395 		/* issue null data to check sta alive*/
396 		for (i = 0; i < chk_alive_num; i++) {
397 			int ret = _FAIL;
398 
399 			psta = rtw_get_stainfo_by_offset(pstapriv, chk_alive_list[i]);
400 
401 			if (psta->state & WIFI_SLEEP_STATE)
402 				ret = issue_nulldata(padapter, psta->hwaddr, 0, 1, 50);
403 			else
404 				ret = issue_nulldata(padapter, psta->hwaddr, 0, 3, 50);
405 
406 			psta->keep_alive_trycnt++;
407 			if (ret == _SUCCESS) {
408 				DBG_88E("asoc check, sta(%pM) is alive\n", (psta->hwaddr));
409 				psta->expire_to = pstapriv->expire_to;
410 				psta->keep_alive_trycnt = 0;
411 				continue;
412 			} else if (psta->keep_alive_trycnt <= 3) {
413 				DBG_88E("ack check for asoc expire, keep_alive_trycnt =%d\n", psta->keep_alive_trycnt);
414 				psta->expire_to = 1;
415 				continue;
416 			}
417 
418 			psta->keep_alive_trycnt = 0;
419 
420 			DBG_88E("asoc expire %pM, state = 0x%x\n", (psta->hwaddr), psta->state);
421 			spin_lock_bh(&pstapriv->asoc_list_lock);
422 			list_del_init(&psta->asoc_list);
423 			pstapriv->asoc_list_cnt--;
424 			updated = ap_free_sta(padapter, psta, true, WLAN_REASON_DEAUTH_LEAVING);
425 			spin_unlock_bh(&pstapriv->asoc_list_lock);
426 		}
427 
428 		if (backup_oper_channel > 0) /* back to the original operation channel */
429 			SelectChannel(padapter, backup_oper_channel);
430 	}
431 
432 	associated_clients_update(padapter, updated);
433 }
434 
add_RATid(struct adapter * padapter,struct sta_info * psta,u8 rssi_level)435 void add_RATid(struct adapter *padapter, struct sta_info *psta, u8 rssi_level)
436 {
437 	int i;
438 	u8 rf_type;
439 	u32 init_rate = 0;
440 	unsigned char sta_band = 0, raid, shortGIrate = false;
441 	unsigned char limit;
442 	unsigned int tx_ra_bitmap = 0;
443 	struct ht_priv	*psta_ht = NULL;
444 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
445 	struct wlan_bssid_ex *pcur_network = (struct wlan_bssid_ex *)&pmlmepriv->cur_network.network;
446 
447 	if (psta)
448 		psta_ht = &psta->htpriv;
449 	else
450 		return;
451 
452 	if (!(psta->state & _FW_LINKED))
453 		return;
454 
455 	/* b/g mode ra_bitmap */
456 	for (i = 0; i < sizeof(psta->bssrateset); i++) {
457 		if (psta->bssrateset[i])
458 			tx_ra_bitmap |= rtw_get_bit_value_from_ieee_value(psta->bssrateset[i]&0x7f);
459 	}
460 	/* n mode ra_bitmap */
461 	if (psta_ht->ht_option) {
462 		rtw_hal_get_hwreg(padapter, HW_VAR_RF_TYPE, (u8 *)(&rf_type));
463 		if (rf_type == RF_2T2R)
464 			limit = 16;/*  2R */
465 		else
466 			limit = 8;/*   1R */
467 
468 		for (i = 0; i < limit; i++) {
469 			if (psta_ht->ht_cap.supp_mcs_set[i/8] & BIT(i%8))
470 				tx_ra_bitmap |= BIT(i+12);
471 		}
472 
473 		/* max short GI rate */
474 		shortGIrate = psta_ht->sgi;
475 	}
476 
477 	if (pcur_network->Configuration.DSConfig > 14) {
478 		/*  5G band */
479 		if (tx_ra_bitmap & 0xffff000)
480 			sta_band |= WIRELESS_11_5N | WIRELESS_11A;
481 		else
482 			sta_band |= WIRELESS_11A;
483 	} else {
484 		if (tx_ra_bitmap & 0xffff000)
485 			sta_band |= WIRELESS_11_24N | WIRELESS_11G | WIRELESS_11B;
486 		else if (tx_ra_bitmap & 0xff0)
487 			sta_band |= WIRELESS_11G | WIRELESS_11B;
488 		else
489 			sta_band |= WIRELESS_11B;
490 	}
491 
492 	psta->wireless_mode = sta_band;
493 
494 	raid = networktype_to_raid(sta_band);
495 	init_rate = get_highest_rate_idx(tx_ra_bitmap&0x0fffffff)&0x3f;
496 
497 	if (psta->aid < NUM_STA) {
498 		u8 arg = 0;
499 
500 		arg = psta->mac_id&0x1f;
501 
502 		arg |= BIT(7);/* support entry 2~31 */
503 
504 		if (shortGIrate)
505 			arg |= BIT(5);
506 
507 		tx_ra_bitmap |= ((raid<<28)&0xf0000000);
508 
509 		DBG_88E("%s => mac_id:%d , raid:%d , bitmap = 0x%x, arg = 0x%x\n",
510 			__func__, psta->mac_id, raid, tx_ra_bitmap, arg);
511 
512 		/* bitmap[0:27] = tx_rate_bitmap */
513 		/* bitmap[28:31]= Rate Adaptive id */
514 		/* arg[0:4] = macid */
515 		/* arg[5] = Short GI */
516 		rtw_hal_add_ra_tid(padapter, tx_ra_bitmap, arg, rssi_level);
517 
518 		if (shortGIrate)
519 			init_rate |= BIT(6);
520 
521 		/* set ra_id, init_rate */
522 		psta->raid = raid;
523 		psta->init_rate = init_rate;
524 
525 	} else {
526 		DBG_88E("station aid %d exceed the max number\n", psta->aid);
527 	}
528 }
529 
update_bmc_sta(struct adapter * padapter)530 static void update_bmc_sta(struct adapter *padapter)
531 {
532 	u32 init_rate = 0;
533 	unsigned char	network_type, raid;
534 	int i, supportRateNum = 0;
535 	unsigned int tx_ra_bitmap = 0;
536 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
537 	struct wlan_bssid_ex *pcur_network = (struct wlan_bssid_ex *)&pmlmepriv->cur_network.network;
538 	struct sta_info *psta = rtw_get_bcmc_stainfo(padapter);
539 
540 	if (psta) {
541 		psta->aid = 0;/* default set to 0 */
542 		psta->mac_id = psta->aid + 1;
543 
544 		psta->qos_option = 0;
545 		psta->htpriv.ht_option = false;
546 
547 		psta->ieee8021x_blocked = 0;
548 
549 		memset((void *)&psta->sta_stats, 0, sizeof(struct stainfo_stats));
550 
551 		/* prepare for add_RATid */
552 		supportRateNum = rtw_get_rateset_len((u8 *)&pcur_network->SupportedRates);
553 		network_type = rtw_check_network_type((u8 *)&pcur_network->SupportedRates, supportRateNum, 1);
554 
555 		memcpy(psta->bssrateset, &pcur_network->SupportedRates, supportRateNum);
556 		psta->bssratelen = supportRateNum;
557 
558 		/* b/g mode ra_bitmap */
559 		for (i = 0; i < supportRateNum; i++) {
560 			if (psta->bssrateset[i])
561 				tx_ra_bitmap |= rtw_get_bit_value_from_ieee_value(psta->bssrateset[i]&0x7f);
562 		}
563 
564 		if (pcur_network->Configuration.DSConfig > 14) {
565 			/* force to A mode. 5G doesn't support CCK rates */
566 			network_type = WIRELESS_11A;
567 			tx_ra_bitmap = 0x150; /*  6, 12, 24 Mbps */
568 		} else {
569 			/* force to b mode */
570 			network_type = WIRELESS_11B;
571 			tx_ra_bitmap = 0xf;
572 		}
573 
574 		raid = networktype_to_raid(network_type);
575 		init_rate = get_highest_rate_idx(tx_ra_bitmap&0x0fffffff)&0x3f;
576 
577 		/* ap mode */
578 		rtw_hal_set_odm_var(padapter, HAL_ODM_STA_INFO, psta, true);
579 
580 		{
581 			u8 arg = 0;
582 
583 			arg = psta->mac_id&0x1f;
584 			arg |= BIT(7);
585 			tx_ra_bitmap |= ((raid<<28)&0xf0000000);
586 			DBG_88E("update_bmc_sta, mask = 0x%x, arg = 0x%x\n", tx_ra_bitmap, arg);
587 
588 			/* bitmap[0:27] = tx_rate_bitmap */
589 			/* bitmap[28:31]= Rate Adaptive id */
590 			/* arg[0:4] = macid */
591 			/* arg[5] = Short GI */
592 			rtw_hal_add_ra_tid(padapter, tx_ra_bitmap, arg, 0);
593 		}
594 		/* set ra_id, init_rate */
595 		psta->raid = raid;
596 		psta->init_rate = init_rate;
597 
598 		rtw_stassoc_hw_rpt(padapter, psta);
599 
600 		spin_lock_bh(&psta->lock);
601 		psta->state = _FW_LINKED;
602 		spin_unlock_bh(&psta->lock);
603 
604 	} else {
605 		DBG_88E("add_RATid_bmc_sta error!\n");
606 	}
607 }
608 
609 /* notes: */
610 /* AID: 1~MAX for sta and 0 for bc/mc in ap/adhoc mode */
611 /* MAC_ID = AID+1 for sta in ap/adhoc mode */
612 /* MAC_ID = 1 for bc/mc for sta/ap/adhoc */
613 /* MAC_ID = 0 for bssid for sta/ap/adhoc */
614 /* CAM_ID = 0~3 for default key, cmd_id = macid + 3, macid = aid+1; */
615 
update_sta_info_apmode(struct adapter * padapter,struct sta_info * psta)616 void update_sta_info_apmode(struct adapter *padapter, struct sta_info *psta)
617 {
618 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
619 	struct security_priv *psecuritypriv = &padapter->securitypriv;
620 	struct mlme_ext_priv	*pmlmeext = &(padapter->mlmeextpriv);
621 	struct ht_priv	*phtpriv_ap = &pmlmepriv->htpriv;
622 	struct ht_priv	*phtpriv_sta = &psta->htpriv;
623 
624 	psta->mac_id = psta->aid+1;
625 	DBG_88E("%s\n", __func__);
626 
627 	/* ap mode */
628 	rtw_hal_set_odm_var(padapter, HAL_ODM_STA_INFO, psta, true);
629 
630 	if (psecuritypriv->dot11AuthAlgrthm == dot11AuthAlgrthm_8021X)
631 		psta->ieee8021x_blocked = true;
632 	else
633 		psta->ieee8021x_blocked = false;
634 
635 
636 	/* update sta's cap */
637 
638 	/* ERP */
639 	VCS_update(padapter, psta);
640 	/* HT related cap */
641 	if (phtpriv_sta->ht_option) {
642 		/* check if sta supports rx ampdu */
643 		phtpriv_sta->ampdu_enable = phtpriv_ap->ampdu_enable;
644 
645 		/* check if sta support s Short GI */
646 		if ((phtpriv_sta->ht_cap.cap_info & phtpriv_ap->ht_cap.cap_info) & (IEEE80211_HT_CAP_SGI_20 | IEEE80211_HT_CAP_SGI_40))
647 			phtpriv_sta->sgi = true;
648 
649 		/*  bwmode */
650 		if ((phtpriv_sta->ht_cap.cap_info & phtpriv_ap->ht_cap.cap_info) & IEEE80211_HT_CAP_SUP_WIDTH) {
651 			phtpriv_sta->bwmode = pmlmeext->cur_bwmode;
652 			phtpriv_sta->ch_offset = pmlmeext->cur_ch_offset;
653 		}
654 		psta->qos_option = true;
655 	} else {
656 		phtpriv_sta->ampdu_enable = false;
657 		phtpriv_sta->sgi = false;
658 		phtpriv_sta->bwmode = HT_CHANNEL_WIDTH_20;
659 		phtpriv_sta->ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
660 	}
661 
662 	/* Rx AMPDU */
663 	send_delba(padapter, 0, psta->hwaddr);/*  recipient */
664 
665 	/* TX AMPDU */
666 	send_delba(padapter, 1, psta->hwaddr);/* originator */
667 	phtpriv_sta->agg_enable_bitmap = 0x0;/* reset */
668 	phtpriv_sta->candidate_tid_bitmap = 0x0;/* reset */
669 
670 	/* todo: init other variables */
671 
672 	memset((void *)&psta->sta_stats, 0, sizeof(struct stainfo_stats));
673 
674 	spin_lock_bh(&psta->lock);
675 	psta->state |= _FW_LINKED;
676 	spin_unlock_bh(&psta->lock);
677 }
678 
update_hw_ht_param(struct adapter * padapter)679 static void update_hw_ht_param(struct adapter *padapter)
680 {
681 	unsigned char		max_AMPDU_len;
682 	unsigned char		min_MPDU_spacing;
683 	struct mlme_ext_priv	*pmlmeext = &padapter->mlmeextpriv;
684 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
685 
686 	DBG_88E("%s\n", __func__);
687 
688 	/* handle A-MPDU parameter field */
689 	/*
690 		AMPDU_para [1:0]:Max AMPDU Len => 0:8k , 1:16k, 2:32k, 3:64k
691 		AMPDU_para [4:2]:Min MPDU Start Spacing
692 	*/
693 	max_AMPDU_len = pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x03;
694 
695 	min_MPDU_spacing = (pmlmeinfo->HT_caps.u.HT_cap_element.AMPDU_para & 0x1c) >> 2;
696 
697 	rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_MIN_SPACE, (u8 *)(&min_MPDU_spacing));
698 
699 	rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_FACTOR, (u8 *)(&max_AMPDU_len));
700 
701 	/*  */
702 	/*  Config SM Power Save setting */
703 	/*  */
704 	pmlmeinfo->SM_PS = (le16_to_cpu(pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info) & 0x0C) >> 2;
705 	if (pmlmeinfo->SM_PS == WLAN_HT_CAP_SM_PS_STATIC)
706 		DBG_88E("%s(): WLAN_HT_CAP_SM_PS_STATIC\n", __func__);
707 }
708 
start_bss_network(struct adapter * padapter,u8 * pbuf)709 static void start_bss_network(struct adapter *padapter, u8 *pbuf)
710 {
711 	u8 *p;
712 	u8 val8, cur_channel, cur_bwmode, cur_ch_offset;
713 	u16 bcn_interval;
714 	u32	acparm;
715 	int	ie_len;
716 	struct registry_priv	 *pregpriv = &padapter->registrypriv;
717 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
718 	struct security_priv *psecuritypriv = &(padapter->securitypriv);
719 	struct wlan_bssid_ex *pnetwork = (struct wlan_bssid_ex *)&pmlmepriv->cur_network.network;
720 	struct mlme_ext_priv	*pmlmeext = &(padapter->mlmeextpriv);
721 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
722 	struct wlan_bssid_ex *pnetwork_mlmeext = &(pmlmeinfo->network);
723 	struct HT_info_element *pht_info = NULL;
724 
725 	bcn_interval = (u16)pnetwork->Configuration.BeaconPeriod;
726 	cur_channel = pnetwork->Configuration.DSConfig;
727 	cur_bwmode = HT_CHANNEL_WIDTH_20;
728 	cur_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
729 
730 
731 	/* check if there is wps ie, */
732 	/* if there is wpsie in beacon, the hostapd will update beacon twice when stating hostapd, */
733 	/* and at first time the security ie (RSN/WPA IE) will not include in beacon. */
734 	if (!rtw_get_wps_ie(pnetwork->IEs+_FIXED_IE_LENGTH_, pnetwork->IELength-_FIXED_IE_LENGTH_, NULL, NULL))
735 		pmlmeext->bstart_bss = true;
736 
737 	/* todo: update wmm, ht cap */
738 	if (pmlmepriv->qospriv.qos_option)
739 		pmlmeinfo->WMM_enable = true;
740 	if (pmlmepriv->htpriv.ht_option) {
741 		pmlmeinfo->WMM_enable = true;
742 		pmlmeinfo->HT_enable = true;
743 
744 		update_hw_ht_param(padapter);
745 	}
746 
747 	if (pmlmepriv->cur_network.join_res != true) { /* setting only at  first time */
748 		/* WEP Key will be set before this function, do not clear CAM. */
749 		if ((psecuritypriv->dot11PrivacyAlgrthm != _WEP40_) &&
750 		    (psecuritypriv->dot11PrivacyAlgrthm != _WEP104_))
751 			flush_all_cam_entry(padapter);	/* clear CAM */
752 	}
753 
754 	/* set MSR to AP_Mode */
755 	Set_MSR(padapter, _HW_STATE_AP_);
756 
757 	/* Set BSSID REG */
758 	rtw_hal_set_hwreg(padapter, HW_VAR_BSSID, pnetwork->MacAddress);
759 
760 	/* Set EDCA param reg */
761 	acparm = 0x002F3217; /*  VO */
762 	rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_VO, (u8 *)(&acparm));
763 	acparm = 0x005E4317; /*  VI */
764 	rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_VI, (u8 *)(&acparm));
765 	acparm = 0x005ea42b;
766 	rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_BE, (u8 *)(&acparm));
767 	acparm = 0x0000A444; /*  BK */
768 	rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_BK, (u8 *)(&acparm));
769 
770 	/* Set Security */
771 	val8 = (psecuritypriv->dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) ? 0xcc : 0xcf;
772 	rtw_hal_set_hwreg(padapter, HW_VAR_SEC_CFG, (u8 *)(&val8));
773 
774 	/* Beacon Control related register */
775 	rtw_hal_set_hwreg(padapter, HW_VAR_BEACON_INTERVAL, (u8 *)(&bcn_interval));
776 
777 	UpdateBrateTbl(padapter, pnetwork->SupportedRates);
778 	rtw_hal_set_hwreg(padapter, HW_VAR_BASIC_RATE, pnetwork->SupportedRates);
779 
780 	if (!pmlmepriv->cur_network.join_res) { /* setting only at  first time */
781 		/* turn on all dynamic functions */
782 		Switch_DM_Func(padapter, DYNAMIC_ALL_FUNC_ENABLE, true);
783 	}
784 	/* set channel, bwmode */
785 	p = rtw_get_ie((pnetwork->IEs + sizeof(struct ndis_802_11_fixed_ie)), _HT_ADD_INFO_IE_, &ie_len, (pnetwork->IELength - sizeof(struct ndis_802_11_fixed_ie)));
786 	if (p && ie_len) {
787 		pht_info = (struct HT_info_element *)(p+2);
788 
789 		if ((pregpriv->cbw40_enable) &&	 (pht_info->infos[0] & BIT(2))) {
790 			/* switch to the 40M Hz mode */
791 			cur_bwmode = HT_CHANNEL_WIDTH_40;
792 			switch (pht_info->infos[0] & 0x3) {
793 			case 1:
794 				cur_ch_offset = HAL_PRIME_CHNL_OFFSET_LOWER;
795 				break;
796 			case 3:
797 				cur_ch_offset = HAL_PRIME_CHNL_OFFSET_UPPER;
798 				break;
799 			default:
800 				cur_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
801 				break;
802 			}
803 		}
804 	}
805 	/* TODO: need to judge the phy parameters on concurrent mode for single phy */
806 	set_channel_bwmode(padapter, cur_channel, cur_ch_offset, cur_bwmode);
807 
808 	DBG_88E("CH =%d, BW =%d, offset =%d\n", cur_channel, cur_bwmode, cur_ch_offset);
809 
810 	/*  */
811 	pmlmeext->cur_channel = cur_channel;
812 	pmlmeext->cur_bwmode = cur_bwmode;
813 	pmlmeext->cur_ch_offset = cur_ch_offset;
814 	pmlmeext->cur_wireless_mode = pmlmepriv->cur_network.network_type;
815 
816 	/* update cur_wireless_mode */
817 	update_wireless_mode(padapter);
818 
819 	/* update capability after cur_wireless_mode updated */
820 	update_capinfo(padapter, rtw_get_capability((struct wlan_bssid_ex *)pnetwork));
821 
822 	/* let pnetwork_mlmeext == pnetwork_mlme. */
823 	memcpy(pnetwork_mlmeext, pnetwork, pnetwork->Length);
824 
825 	if (pmlmeext->bstart_bss) {
826 		update_beacon(padapter, _TIM_IE_, NULL, false);
827 
828 		/* issue beacon frame */
829 		if (send_beacon(padapter) == _FAIL)
830 			DBG_88E("issue_beacon, fail!\n");
831 	}
832 
833 	/* update bc/mc sta_info */
834 	update_bmc_sta(padapter);
835 }
836 
rtw_check_beacon_data(struct adapter * padapter,u8 * pbuf,int len)837 int rtw_check_beacon_data(struct adapter *padapter, u8 *pbuf,  int len)
838 {
839 	int ret = _SUCCESS;
840 	u8 *p;
841 	u8 *pHT_caps_ie = NULL;
842 	u8 *pHT_info_ie = NULL;
843 	struct sta_info *psta = NULL;
844 	u16 cap, ht_cap = false;
845 	uint ie_len = 0;
846 	int group_cipher, pairwise_cipher;
847 	u8	channel, network_type, supportRate[NDIS_802_11_LENGTH_RATES_EX];
848 	int supportRateNum = 0;
849 	u8 OUI1[] = {0x00, 0x50, 0xf2, 0x01};
850 	u8 WMM_PARA_IE[] = {0x00, 0x50, 0xf2, 0x02, 0x01, 0x01};
851 	struct registry_priv *pregistrypriv = &padapter->registrypriv;
852 	struct security_priv *psecuritypriv = &padapter->securitypriv;
853 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
854 	struct wlan_bssid_ex *pbss_network = (struct wlan_bssid_ex *)&pmlmepriv->cur_network.network;
855 	u8 *ie = pbss_network->IEs;
856 
857 	/* SSID */
858 	/* Supported rates */
859 	/* DS Params */
860 	/* WLAN_EID_COUNTRY */
861 	/* ERP Information element */
862 	/* Extended supported rates */
863 	/* WPA/WPA2 */
864 	/* Wi-Fi Wireless Multimedia Extensions */
865 	/* ht_capab, ht_oper */
866 	/* WPS IE */
867 
868 	DBG_88E("%s, len =%d\n", __func__, len);
869 
870 	if (check_fwstate(pmlmepriv, WIFI_AP_STATE) != true)
871 		return _FAIL;
872 
873 
874 	if (len > MAX_IE_SZ)
875 		return _FAIL;
876 
877 	pbss_network->IELength = len;
878 
879 	memset(ie, 0, MAX_IE_SZ);
880 
881 	memcpy(ie, pbuf, pbss_network->IELength);
882 
883 
884 	if (pbss_network->InfrastructureMode != Ndis802_11APMode)
885 		return _FAIL;
886 
887 	pbss_network->Rssi = 0;
888 
889 	ether_addr_copy(pbss_network->MacAddress, myid(&(padapter->eeprompriv)));
890 
891 	/* beacon interval */
892 	p = rtw_get_beacon_interval_from_ie(ie);/* 8: TimeStamp, 2: Beacon Interval 2:Capability */
893 	pbss_network->Configuration.BeaconPeriod = get_unaligned_le16(p);
894 
895 	/* capability */
896 	cap = get_unaligned_le16(ie);
897 
898 	/* SSID */
899 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _SSID_IE_, &ie_len, (pbss_network->IELength - _BEACON_IE_OFFSET_));
900 	if (p && ie_len > 0) {
901 		memset(&pbss_network->Ssid, 0, sizeof(struct ndis_802_11_ssid));
902 		memcpy(pbss_network->Ssid.Ssid, (p + 2), ie_len);
903 		pbss_network->Ssid.SsidLength = ie_len;
904 	}
905 
906 	/* channel */
907 	channel = 0;
908 	pbss_network->Configuration.Length = 0;
909 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _DSSET_IE_, &ie_len, (pbss_network->IELength - _BEACON_IE_OFFSET_));
910 	if (p && ie_len > 0)
911 		channel = *(p + 2);
912 
913 	pbss_network->Configuration.DSConfig = channel;
914 
915 	memset(supportRate, 0, NDIS_802_11_LENGTH_RATES_EX);
916 	/*  get supported rates */
917 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _SUPPORTEDRATES_IE_, &ie_len, (pbss_network->IELength - _BEACON_IE_OFFSET_));
918 	if (p !=  NULL) {
919 		memcpy(supportRate, p+2, ie_len);
920 		supportRateNum = ie_len;
921 	}
922 
923 	/* get ext_supported rates */
924 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _EXT_SUPPORTEDRATES_IE_, &ie_len, pbss_network->IELength - _BEACON_IE_OFFSET_);
925 	if (p !=  NULL) {
926 		memcpy(supportRate+supportRateNum, p+2, ie_len);
927 		supportRateNum += ie_len;
928 	}
929 
930 	network_type = rtw_check_network_type(supportRate, supportRateNum, channel);
931 
932 	rtw_set_supported_rate(pbss_network->SupportedRates, network_type);
933 
934 	/* parsing ERP_IE */
935 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _ERPINFO_IE_, &ie_len, (pbss_network->IELength - _BEACON_IE_OFFSET_));
936 	if (p && ie_len > 0)
937 		ERP_IE_handler(padapter, (struct ndis_802_11_var_ie *)p);
938 
939 	/* update privacy/security */
940 	if (cap & BIT(4))
941 		pbss_network->Privacy = 1;
942 	else
943 		pbss_network->Privacy = 0;
944 
945 	psecuritypriv->wpa_psk = 0;
946 
947 	/* wpa2 */
948 	group_cipher = 0;
949 	pairwise_cipher = 0;
950 	psecuritypriv->wpa2_group_cipher = _NO_PRIVACY_;
951 	psecuritypriv->wpa2_pairwise_cipher = _NO_PRIVACY_;
952 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _RSN_IE_2_, &ie_len, (pbss_network->IELength - _BEACON_IE_OFFSET_));
953 	if (p && ie_len > 0) {
954 		if (rtw_parse_wpa2_ie(p, ie_len+2, &group_cipher, &pairwise_cipher, NULL) == _SUCCESS) {
955 			psecuritypriv->dot11AuthAlgrthm = dot11AuthAlgrthm_8021X;
956 
957 			psecuritypriv->dot8021xalg = 1;/* psk,  todo:802.1x */
958 			psecuritypriv->wpa_psk |= BIT(1);
959 
960 			psecuritypriv->wpa2_group_cipher = group_cipher;
961 			psecuritypriv->wpa2_pairwise_cipher = pairwise_cipher;
962 		}
963 	}
964 	/* wpa */
965 	ie_len = 0;
966 	group_cipher = 0;
967 	pairwise_cipher = 0;
968 	psecuritypriv->wpa_group_cipher = _NO_PRIVACY_;
969 	psecuritypriv->wpa_pairwise_cipher = _NO_PRIVACY_;
970 	for (p = ie + _BEACON_IE_OFFSET_;; p += (ie_len + 2)) {
971 		p = rtw_get_ie(p, _SSN_IE_1_, &ie_len,
972 			       (pbss_network->IELength - _BEACON_IE_OFFSET_ - (ie_len + 2)));
973 		if ((p) && (!memcmp(p+2, OUI1, 4))) {
974 			if (rtw_parse_wpa_ie(p, ie_len+2, &group_cipher,
975 					     &pairwise_cipher, NULL) == _SUCCESS) {
976 				psecuritypriv->dot11AuthAlgrthm = dot11AuthAlgrthm_8021X;
977 
978 				psecuritypriv->dot8021xalg = 1;/* psk,  todo:802.1x */
979 
980 				psecuritypriv->wpa_psk |= BIT(0);
981 
982 				psecuritypriv->wpa_group_cipher = group_cipher;
983 				psecuritypriv->wpa_pairwise_cipher = pairwise_cipher;
984 			}
985 			break;
986 		}
987 		if ((p == NULL) || (ie_len == 0))
988 			break;
989 	}
990 
991 	/* wmm */
992 	ie_len = 0;
993 	pmlmepriv->qospriv.qos_option = 0;
994 	if (pregistrypriv->wmm_enable) {
995 		for (p = ie + _BEACON_IE_OFFSET_;; p += (ie_len + 2)) {
996 			p = rtw_get_ie(p, _VENDOR_SPECIFIC_IE_, &ie_len,
997 				       (pbss_network->IELength - _BEACON_IE_OFFSET_ - (ie_len + 2)));
998 			if ((p) && !memcmp(p+2, WMM_PARA_IE, 6)) {
999 				pmlmepriv->qospriv.qos_option = 1;
1000 
1001 				*(p+8) |= BIT(7);/* QoS Info, support U-APSD */
1002 
1003 				/* disable all ACM bits since the WMM admission control is not supported */
1004 				*(p + 10) &= ~BIT(4); /* BE */
1005 				*(p + 14) &= ~BIT(4); /* BK */
1006 				*(p + 18) &= ~BIT(4); /* VI */
1007 				*(p + 22) &= ~BIT(4); /* VO */
1008 				break;
1009 			}
1010 
1011 			if ((p == NULL) || (ie_len == 0))
1012 				break;
1013 		}
1014 	}
1015 	/* parsing HT_CAP_IE */
1016 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _HT_CAPABILITY_IE_, &ie_len,
1017 		       (pbss_network->IELength - _BEACON_IE_OFFSET_));
1018 	if (p && ie_len > 0) {
1019 		u8 rf_type;
1020 		struct rtw_ieee80211_ht_cap *pht_cap = (struct rtw_ieee80211_ht_cap *)(p+2);
1021 
1022 		pHT_caps_ie = p;
1023 		ht_cap = true;
1024 		network_type |= WIRELESS_11_24N;
1025 
1026 		rtw_hal_get_hwreg(padapter, HW_VAR_RF_TYPE, (u8 *)(&rf_type));
1027 
1028 		if ((psecuritypriv->wpa_pairwise_cipher & WPA_CIPHER_CCMP) ||
1029 		    (psecuritypriv->wpa2_pairwise_cipher & WPA_CIPHER_CCMP))
1030 			pht_cap->ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY&(0x07<<2));
1031 		else
1032 			pht_cap->ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY&0x00);
1033 
1034 		/* set  Max Rx AMPDU size  to 64K */
1035 		pht_cap->ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_FACTOR & 0x03);
1036 
1037 		if (rf_type == RF_1T1R) {
1038 			pht_cap->supp_mcs_set[0] = 0xff;
1039 			pht_cap->supp_mcs_set[1] = 0x0;
1040 		}
1041 		memcpy(&pmlmepriv->htpriv.ht_cap, p+2, ie_len);
1042 	}
1043 
1044 	/* parsing HT_INFO_IE */
1045 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _HT_ADD_INFO_IE_, &ie_len,
1046 		       (pbss_network->IELength - _BEACON_IE_OFFSET_));
1047 	if (p && ie_len > 0)
1048 		pHT_info_ie = p;
1049 	switch (network_type) {
1050 	case WIRELESS_11B:
1051 		pbss_network->NetworkTypeInUse = Ndis802_11DS;
1052 		break;
1053 	case WIRELESS_11G:
1054 	case WIRELESS_11BG:
1055 	case WIRELESS_11G_24N:
1056 	case WIRELESS_11BG_24N:
1057 		pbss_network->NetworkTypeInUse = Ndis802_11OFDM24;
1058 		break;
1059 	case WIRELESS_11A:
1060 		pbss_network->NetworkTypeInUse = Ndis802_11OFDM5;
1061 		break;
1062 	default:
1063 		pbss_network->NetworkTypeInUse = Ndis802_11OFDM24;
1064 		break;
1065 	}
1066 
1067 	pmlmepriv->cur_network.network_type = network_type;
1068 
1069 	pmlmepriv->htpriv.ht_option = false;
1070 
1071 	if ((psecuritypriv->wpa2_pairwise_cipher & WPA_CIPHER_TKIP) ||
1072 	    (psecuritypriv->wpa_pairwise_cipher & WPA_CIPHER_TKIP)) {
1073 		/* todo: */
1074 		/* ht_cap = false; */
1075 	}
1076 
1077 	/* ht_cap */
1078 	if (pregistrypriv->ht_enable && ht_cap) {
1079 		pmlmepriv->htpriv.ht_option = true;
1080 		pmlmepriv->qospriv.qos_option = 1;
1081 
1082 		if (pregistrypriv->ampdu_enable == 1)
1083 			pmlmepriv->htpriv.ampdu_enable = true;
1084 		HT_caps_handler(padapter, (struct ndis_802_11_var_ie *)pHT_caps_ie);
1085 
1086 		HT_info_handler(padapter, (struct ndis_802_11_var_ie *)pHT_info_ie);
1087 	}
1088 
1089 	pbss_network->Length = get_wlan_bssid_ex_sz((struct wlan_bssid_ex  *)pbss_network);
1090 
1091 	/* issue beacon to start bss network */
1092 	start_bss_network(padapter, (u8 *)pbss_network);
1093 
1094 	/* alloc sta_info for ap itself */
1095 	psta = rtw_get_stainfo(&padapter->stapriv, pbss_network->MacAddress);
1096 	if (!psta) {
1097 		psta = rtw_alloc_stainfo(&padapter->stapriv, pbss_network->MacAddress);
1098 		if (psta == NULL)
1099 			return _FAIL;
1100 	}
1101 
1102 	/* fix bug of flush_cam_entry at STOP AP mode */
1103 	psta->state |= WIFI_AP_STATE;
1104 	rtw_indicate_connect(padapter);
1105 	pmlmepriv->cur_network.join_res = true;/* for check if already set beacon */
1106 	return ret;
1107 }
1108 
rtw_set_macaddr_acl(struct adapter * padapter,int mode)1109 void rtw_set_macaddr_acl(struct adapter *padapter, int mode)
1110 {
1111 	struct sta_priv *pstapriv = &padapter->stapriv;
1112 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
1113 
1114 	DBG_88E("%s, mode =%d\n", __func__, mode);
1115 
1116 	pacl_list->mode = mode;
1117 }
1118 
rtw_acl_add_sta(struct adapter * padapter,u8 * addr)1119 int rtw_acl_add_sta(struct adapter *padapter, u8 *addr)
1120 {
1121 	struct list_head *plist, *phead;
1122 	u8 added = false;
1123 	int i, ret = 0;
1124 	struct rtw_wlan_acl_node *paclnode;
1125 	struct sta_priv *pstapriv = &padapter->stapriv;
1126 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
1127 	struct __queue *pacl_node_q = &pacl_list->acl_node_q;
1128 
1129 	DBG_88E("%s(acl_num =%d) =%pM\n", __func__, pacl_list->num, (addr));
1130 
1131 	if ((NUM_ACL-1) < pacl_list->num)
1132 		return -1;
1133 
1134 	spin_lock_bh(&(pacl_node_q->lock));
1135 
1136 	phead = get_list_head(pacl_node_q);
1137 	plist = phead->next;
1138 
1139 	while (phead != plist) {
1140 		paclnode = container_of(plist, struct rtw_wlan_acl_node, list);
1141 		plist = plist->next;
1142 
1143 		if (!memcmp(paclnode->addr, addr, ETH_ALEN)) {
1144 			if (paclnode->valid) {
1145 				added = true;
1146 				DBG_88E("%s, sta has been added\n", __func__);
1147 				break;
1148 			}
1149 		}
1150 	}
1151 
1152 	spin_unlock_bh(&(pacl_node_q->lock));
1153 
1154 	if (added)
1155 		return ret;
1156 
1157 	spin_lock_bh(&(pacl_node_q->lock));
1158 
1159 	for (i = 0; i < NUM_ACL; i++) {
1160 		paclnode = &pacl_list->aclnode[i];
1161 
1162 		if (!paclnode->valid) {
1163 			INIT_LIST_HEAD(&paclnode->list);
1164 
1165 			ether_addr_copy(paclnode->addr, addr);
1166 
1167 			paclnode->valid = true;
1168 
1169 			list_add_tail(&paclnode->list, get_list_head(pacl_node_q));
1170 
1171 			pacl_list->num++;
1172 
1173 			break;
1174 		}
1175 	}
1176 
1177 	DBG_88E("%s, acl_num =%d\n", __func__, pacl_list->num);
1178 
1179 	spin_unlock_bh(&(pacl_node_q->lock));
1180 
1181 	return ret;
1182 }
1183 
rtw_acl_remove_sta(struct adapter * padapter,u8 * addr)1184 int rtw_acl_remove_sta(struct adapter *padapter, u8 *addr)
1185 {
1186 	struct list_head *plist, *phead;
1187 	struct rtw_wlan_acl_node *paclnode;
1188 	struct sta_priv *pstapriv = &padapter->stapriv;
1189 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
1190 	struct __queue *pacl_node_q = &pacl_list->acl_node_q;
1191 
1192 	DBG_88E("%s(acl_num =%d) =%pM\n", __func__, pacl_list->num, (addr));
1193 
1194 	spin_lock_bh(&(pacl_node_q->lock));
1195 
1196 	phead = get_list_head(pacl_node_q);
1197 	plist = phead->next;
1198 
1199 	while (phead != plist) {
1200 		paclnode = container_of(plist, struct rtw_wlan_acl_node, list);
1201 		plist = plist->next;
1202 
1203 		if (!memcmp(paclnode->addr, addr, ETH_ALEN)) {
1204 			if (paclnode->valid) {
1205 				paclnode->valid = false;
1206 
1207 				list_del_init(&paclnode->list);
1208 
1209 				pacl_list->num--;
1210 			}
1211 		}
1212 	}
1213 
1214 	spin_unlock_bh(&(pacl_node_q->lock));
1215 
1216 	DBG_88E("%s, acl_num =%d\n", __func__, pacl_list->num);
1217 	return 0;
1218 }
1219 
update_bcn_fixed_ie(struct adapter * padapter)1220 static void update_bcn_fixed_ie(struct adapter *padapter)
1221 {
1222 	DBG_88E("%s\n", __func__);
1223 }
1224 
update_bcn_erpinfo_ie(struct adapter * padapter)1225 static void update_bcn_erpinfo_ie(struct adapter *padapter)
1226 {
1227 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1228 	struct mlme_ext_priv	*pmlmeext = &(padapter->mlmeextpriv);
1229 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1230 	struct wlan_bssid_ex *pnetwork = &(pmlmeinfo->network);
1231 	unsigned char *p, *ie = pnetwork->IEs;
1232 	u32 len = 0;
1233 
1234 	DBG_88E("%s, ERP_enable =%d\n", __func__, pmlmeinfo->ERP_enable);
1235 
1236 	if (!pmlmeinfo->ERP_enable)
1237 		return;
1238 
1239 	/* parsing ERP_IE */
1240 	p = rtw_get_ie(ie + _BEACON_IE_OFFSET_, _ERPINFO_IE_, &len,
1241 		       (pnetwork->IELength - _BEACON_IE_OFFSET_));
1242 	if (p && len > 0) {
1243 		struct ndis_802_11_var_ie *pIE = (struct ndis_802_11_var_ie *)p;
1244 
1245 		if (pmlmepriv->num_sta_non_erp == 1)
1246 			pIE->data[0] |= RTW_ERP_INFO_NON_ERP_PRESENT|RTW_ERP_INFO_USE_PROTECTION;
1247 		else
1248 			pIE->data[0] &= ~(RTW_ERP_INFO_NON_ERP_PRESENT|RTW_ERP_INFO_USE_PROTECTION);
1249 
1250 		if (pmlmepriv->num_sta_no_short_preamble > 0)
1251 			pIE->data[0] |= RTW_ERP_INFO_BARKER_PREAMBLE_MODE;
1252 		else
1253 			pIE->data[0] &= ~(RTW_ERP_INFO_BARKER_PREAMBLE_MODE);
1254 
1255 		ERP_IE_handler(padapter, pIE);
1256 	}
1257 }
1258 
update_bcn_htcap_ie(struct adapter * padapter)1259 static void update_bcn_htcap_ie(struct adapter *padapter)
1260 {
1261 	DBG_88E("%s\n", __func__);
1262 }
1263 
update_bcn_htinfo_ie(struct adapter * padapter)1264 static void update_bcn_htinfo_ie(struct adapter *padapter)
1265 {
1266 	DBG_88E("%s\n", __func__);
1267 }
1268 
update_bcn_rsn_ie(struct adapter * padapter)1269 static void update_bcn_rsn_ie(struct adapter *padapter)
1270 {
1271 	DBG_88E("%s\n", __func__);
1272 }
1273 
update_bcn_wpa_ie(struct adapter * padapter)1274 static void update_bcn_wpa_ie(struct adapter *padapter)
1275 {
1276 	DBG_88E("%s\n", __func__);
1277 }
1278 
update_bcn_wmm_ie(struct adapter * padapter)1279 static void update_bcn_wmm_ie(struct adapter *padapter)
1280 {
1281 	DBG_88E("%s\n", __func__);
1282 }
1283 
update_bcn_wps_ie(struct adapter * padapter)1284 static void update_bcn_wps_ie(struct adapter *padapter)
1285 {
1286 	u8 *pwps_ie = NULL, *pwps_ie_src;
1287 	u8 *premainder_ie, *pbackup_remainder_ie = NULL;
1288 	uint wps_ielen = 0, wps_offset, remainder_ielen;
1289 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1290 	struct mlme_ext_priv	*pmlmeext = &(padapter->mlmeextpriv);
1291 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1292 	struct wlan_bssid_ex *pnetwork = &(pmlmeinfo->network);
1293 	unsigned char *ie = pnetwork->IEs;
1294 	u32 ielen = pnetwork->IELength;
1295 
1296 	DBG_88E("%s\n", __func__);
1297 
1298 	pwps_ie_src = pmlmepriv->wps_beacon_ie;
1299 	if (pwps_ie_src == NULL)
1300 		return;
1301 
1302 	pwps_ie = rtw_get_wps_ie(ie+_FIXED_IE_LENGTH_, ielen-_FIXED_IE_LENGTH_, NULL, &wps_ielen);
1303 
1304 	if (pwps_ie == NULL || wps_ielen == 0)
1305 		return;
1306 
1307 	wps_offset = (uint)(pwps_ie-ie);
1308 
1309 	premainder_ie = pwps_ie + wps_ielen;
1310 
1311 	remainder_ielen = ielen - wps_offset - wps_ielen;
1312 
1313 	if (remainder_ielen > 0) {
1314 		pbackup_remainder_ie = rtw_malloc(remainder_ielen);
1315 		if (pbackup_remainder_ie)
1316 			memcpy(pbackup_remainder_ie, premainder_ie, remainder_ielen);
1317 	}
1318 
1319 	wps_ielen = (uint)pwps_ie_src[1];/* to get ie data len */
1320 	if ((wps_offset+wps_ielen+2+remainder_ielen) <= MAX_IE_SZ) {
1321 		memcpy(pwps_ie, pwps_ie_src, wps_ielen+2);
1322 		pwps_ie += (wps_ielen+2);
1323 
1324 		if (pbackup_remainder_ie)
1325 			memcpy(pwps_ie, pbackup_remainder_ie, remainder_ielen);
1326 
1327 		/* update IELength */
1328 		pnetwork->IELength = wps_offset + (wps_ielen+2) + remainder_ielen;
1329 	}
1330 
1331 	kfree(pbackup_remainder_ie);
1332 }
1333 
update_bcn_p2p_ie(struct adapter * padapter)1334 static void update_bcn_p2p_ie(struct adapter *padapter)
1335 {
1336 }
1337 
update_bcn_vendor_spec_ie(struct adapter * padapter,u8 * oui)1338 static void update_bcn_vendor_spec_ie(struct adapter *padapter, u8 *oui)
1339 {
1340 	DBG_88E("%s\n", __func__);
1341 
1342 	if (!memcmp(RTW_WPA_OUI, oui, 4))
1343 		update_bcn_wpa_ie(padapter);
1344 	else if (!memcmp(WMM_OUI, oui, 4))
1345 		update_bcn_wmm_ie(padapter);
1346 	else if (!memcmp(WPS_OUI, oui, 4))
1347 		update_bcn_wps_ie(padapter);
1348 	else if (!memcmp(P2P_OUI, oui, 4))
1349 		update_bcn_p2p_ie(padapter);
1350 	else
1351 		DBG_88E("unknown OUI type!\n");
1352 }
1353 
update_beacon(struct adapter * padapter,u8 ie_id,u8 * oui,u8 tx)1354 void update_beacon(struct adapter *padapter, u8 ie_id, u8 *oui, u8 tx)
1355 {
1356 	struct mlme_priv *pmlmepriv;
1357 	struct mlme_ext_priv	*pmlmeext;
1358 
1359 	if (!padapter)
1360 		return;
1361 
1362 	pmlmepriv = &(padapter->mlmepriv);
1363 	pmlmeext = &(padapter->mlmeextpriv);
1364 
1365 	if (!pmlmeext->bstart_bss)
1366 		return;
1367 
1368 	spin_lock_bh(&pmlmepriv->bcn_update_lock);
1369 
1370 	switch (ie_id) {
1371 	case 0xFF:
1372 		update_bcn_fixed_ie(padapter);/* 8: TimeStamp, 2: Beacon Interval 2:Capability */
1373 		break;
1374 	case _TIM_IE_:
1375 		update_BCNTIM(padapter);
1376 		break;
1377 	case _ERPINFO_IE_:
1378 		update_bcn_erpinfo_ie(padapter);
1379 		break;
1380 	case _HT_CAPABILITY_IE_:
1381 		update_bcn_htcap_ie(padapter);
1382 		break;
1383 	case _RSN_IE_2_:
1384 		update_bcn_rsn_ie(padapter);
1385 		break;
1386 	case _HT_ADD_INFO_IE_:
1387 		update_bcn_htinfo_ie(padapter);
1388 		break;
1389 	case _VENDOR_SPECIFIC_IE_:
1390 		update_bcn_vendor_spec_ie(padapter, oui);
1391 		break;
1392 	default:
1393 		break;
1394 	}
1395 
1396 	pmlmepriv->update_bcn = true;
1397 
1398 	spin_unlock_bh(&pmlmepriv->bcn_update_lock);
1399 
1400 	if (tx)
1401 		set_tx_beacon_cmd(padapter);
1402 }
1403 
1404 /*
1405 op_mode
1406 Set to 0 (HT pure) under the following conditions
1407 	- all STAs in the BSS are 20/40 MHz HT in 20/40 MHz BSS or
1408 	- all STAs in the BSS are 20 MHz HT in 20 MHz BSS
1409 Set to 1 (HT non-member protection) if there may be non-HT STAs
1410 	in both the primary and the secondary channel
1411 Set to 2 if only HT STAs are associated in BSS,
1412 	however and at least one 20 MHz HT STA is associated
1413 Set to 3 (HT mixed mode) when one or more non-HT STAs are associated
1414 	(currently non-GF HT station is considered as non-HT STA also)
1415 */
rtw_ht_operation_update(struct adapter * padapter)1416 static int rtw_ht_operation_update(struct adapter *padapter)
1417 {
1418 	u16 cur_op_mode, new_op_mode;
1419 	int op_mode_changes = 0;
1420 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1421 	struct ht_priv	*phtpriv_ap = &pmlmepriv->htpriv;
1422 
1423 	if (pmlmepriv->htpriv.ht_option)
1424 		return 0;
1425 
1426 	DBG_88E("%s current operation mode = 0x%X\n",
1427 		__func__, pmlmepriv->ht_op_mode);
1428 
1429 	if (!(pmlmepriv->ht_op_mode & HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT) &&
1430 	    pmlmepriv->num_sta_ht_no_gf) {
1431 		pmlmepriv->ht_op_mode |=
1432 			HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT;
1433 		op_mode_changes++;
1434 	} else if ((pmlmepriv->ht_op_mode &
1435 		   HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT) &&
1436 		   pmlmepriv->num_sta_ht_no_gf == 0) {
1437 		pmlmepriv->ht_op_mode &=
1438 			~HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT;
1439 		op_mode_changes++;
1440 	}
1441 
1442 	if (!(pmlmepriv->ht_op_mode & HT_INFO_OPERATION_MODE_NON_HT_STA_PRESENT) &&
1443 	    (pmlmepriv->num_sta_no_ht || pmlmepriv->olbc_ht)) {
1444 		pmlmepriv->ht_op_mode |= HT_INFO_OPERATION_MODE_NON_HT_STA_PRESENT;
1445 		op_mode_changes++;
1446 	} else if ((pmlmepriv->ht_op_mode &
1447 		    HT_INFO_OPERATION_MODE_NON_HT_STA_PRESENT) &&
1448 		   (pmlmepriv->num_sta_no_ht == 0 && !pmlmepriv->olbc_ht)) {
1449 		pmlmepriv->ht_op_mode &=
1450 			~HT_INFO_OPERATION_MODE_NON_HT_STA_PRESENT;
1451 		op_mode_changes++;
1452 	}
1453 
1454 	/* Note: currently we switch to the MIXED op mode if HT non-greenfield
1455 	 * station is associated. Probably it's a theoretical case, since
1456 	 * it looks like all known HT STAs support greenfield.
1457 	 */
1458 	new_op_mode = 0;
1459 	if (pmlmepriv->num_sta_no_ht ||
1460 	    (pmlmepriv->ht_op_mode & HT_INFO_OPERATION_MODE_NON_GF_DEVS_PRESENT))
1461 		new_op_mode = OP_MODE_MIXED;
1462 	else if ((phtpriv_ap->ht_cap.cap_info & IEEE80211_HT_CAP_SUP_WIDTH) &&
1463 		 pmlmepriv->num_sta_ht_20mhz)
1464 		new_op_mode = OP_MODE_20MHZ_HT_STA_ASSOCED;
1465 	else if (pmlmepriv->olbc_ht)
1466 		new_op_mode = OP_MODE_MAY_BE_LEGACY_STAS;
1467 	else
1468 		new_op_mode = OP_MODE_PURE;
1469 
1470 	cur_op_mode = pmlmepriv->ht_op_mode & HT_INFO_OPERATION_MODE_OP_MODE_MASK;
1471 	if (cur_op_mode != new_op_mode) {
1472 		pmlmepriv->ht_op_mode &= ~HT_INFO_OPERATION_MODE_OP_MODE_MASK;
1473 		pmlmepriv->ht_op_mode |= new_op_mode;
1474 		op_mode_changes++;
1475 	}
1476 
1477 	DBG_88E("%s new operation mode = 0x%X changes =%d\n",
1478 		__func__, pmlmepriv->ht_op_mode, op_mode_changes);
1479 
1480 	return op_mode_changes;
1481 }
1482 
associated_clients_update(struct adapter * padapter,u8 updated)1483 void associated_clients_update(struct adapter *padapter, u8 updated)
1484 {
1485 	/* update associated stations cap. */
1486 	if (updated) {
1487 		struct list_head *phead, *plist;
1488 		struct sta_info *psta = NULL;
1489 		struct sta_priv *pstapriv = &padapter->stapriv;
1490 
1491 		spin_lock_bh(&pstapriv->asoc_list_lock);
1492 
1493 		phead = &pstapriv->asoc_list;
1494 		plist = phead->next;
1495 
1496 		/* check asoc_queue */
1497 		while (phead != plist) {
1498 			psta = container_of(plist, struct sta_info, asoc_list);
1499 
1500 			plist = plist->next;
1501 
1502 			VCS_update(padapter, psta);
1503 		}
1504 		spin_unlock_bh(&pstapriv->asoc_list_lock);
1505 	}
1506 }
1507 
1508 /* called > TSR LEVEL for USB or SDIO Interface*/
bss_cap_update_on_sta_join(struct adapter * padapter,struct sta_info * psta)1509 void bss_cap_update_on_sta_join(struct adapter *padapter, struct sta_info *psta)
1510 {
1511 	u8 beacon_updated = false;
1512 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1513 	struct mlme_ext_priv *pmlmeext = &(padapter->mlmeextpriv);
1514 
1515 	if (!(psta->flags & WLAN_STA_SHORT_PREAMBLE)) {
1516 		if (!psta->no_short_preamble_set) {
1517 			psta->no_short_preamble_set = 1;
1518 
1519 			pmlmepriv->num_sta_no_short_preamble++;
1520 
1521 			if ((pmlmeext->cur_wireless_mode > WIRELESS_11B) &&
1522 			    (pmlmepriv->num_sta_no_short_preamble == 1)) {
1523 				beacon_updated = true;
1524 				update_beacon(padapter, 0xFF, NULL, true);
1525 			}
1526 		}
1527 	} else {
1528 		if (psta->no_short_preamble_set) {
1529 			psta->no_short_preamble_set = 0;
1530 
1531 			pmlmepriv->num_sta_no_short_preamble--;
1532 
1533 			if ((pmlmeext->cur_wireless_mode > WIRELESS_11B) &&
1534 			    (pmlmepriv->num_sta_no_short_preamble == 0)) {
1535 				beacon_updated = true;
1536 				update_beacon(padapter, 0xFF, NULL, true);
1537 			}
1538 		}
1539 	}
1540 
1541 	if (psta->flags & WLAN_STA_NONERP) {
1542 		if (!psta->nonerp_set) {
1543 			psta->nonerp_set = 1;
1544 
1545 			pmlmepriv->num_sta_non_erp++;
1546 
1547 			if (pmlmepriv->num_sta_non_erp == 1) {
1548 				beacon_updated = true;
1549 				update_beacon(padapter, _ERPINFO_IE_, NULL, true);
1550 			}
1551 		}
1552 	} else {
1553 		if (psta->nonerp_set) {
1554 			psta->nonerp_set = 0;
1555 
1556 			pmlmepriv->num_sta_non_erp--;
1557 
1558 			if (pmlmepriv->num_sta_non_erp == 0) {
1559 				beacon_updated = true;
1560 				update_beacon(padapter, _ERPINFO_IE_, NULL, true);
1561 			}
1562 		}
1563 	}
1564 
1565 	if (!(psta->capability & WLAN_CAPABILITY_SHORT_SLOT)) {
1566 		if (!psta->no_short_slot_time_set) {
1567 			psta->no_short_slot_time_set = 1;
1568 
1569 			pmlmepriv->num_sta_no_short_slot_time++;
1570 
1571 			if ((pmlmeext->cur_wireless_mode > WIRELESS_11B) &&
1572 			    (pmlmepriv->num_sta_no_short_slot_time == 1)) {
1573 				beacon_updated = true;
1574 				update_beacon(padapter, 0xFF, NULL, true);
1575 			}
1576 		}
1577 	} else {
1578 		if (psta->no_short_slot_time_set) {
1579 			psta->no_short_slot_time_set = 0;
1580 
1581 			pmlmepriv->num_sta_no_short_slot_time--;
1582 
1583 			if ((pmlmeext->cur_wireless_mode > WIRELESS_11B) &&
1584 			    (pmlmepriv->num_sta_no_short_slot_time == 0)) {
1585 				beacon_updated = true;
1586 				update_beacon(padapter, 0xFF, NULL, true);
1587 			}
1588 		}
1589 	}
1590 
1591 	if (psta->flags & WLAN_STA_HT) {
1592 		u16 ht_capab = psta->htpriv.ht_cap.cap_info;
1593 
1594 		DBG_88E("HT: STA %pM HT Capabilities Info: 0x%04x\n",
1595 			(psta->hwaddr), ht_capab);
1596 
1597 		if (psta->no_ht_set) {
1598 			psta->no_ht_set = 0;
1599 			pmlmepriv->num_sta_no_ht--;
1600 		}
1601 
1602 		if ((ht_capab & IEEE80211_HT_CAP_GRN_FLD) == 0) {
1603 			if (!psta->no_ht_gf_set) {
1604 				psta->no_ht_gf_set = 1;
1605 				pmlmepriv->num_sta_ht_no_gf++;
1606 			}
1607 			DBG_88E("%s STA %pM - no greenfield, num of non-gf stations %d\n",
1608 				   __func__, (psta->hwaddr),
1609 				   pmlmepriv->num_sta_ht_no_gf);
1610 		}
1611 
1612 		if ((ht_capab & IEEE80211_HT_CAP_SUP_WIDTH) == 0) {
1613 			if (!psta->ht_20mhz_set) {
1614 				psta->ht_20mhz_set = 1;
1615 				pmlmepriv->num_sta_ht_20mhz++;
1616 			}
1617 			DBG_88E("%s STA %pM - 20 MHz HT, num of 20MHz HT STAs %d\n",
1618 				   __func__, (psta->hwaddr),
1619 				   pmlmepriv->num_sta_ht_20mhz);
1620 		}
1621 	} else {
1622 		if (!psta->no_ht_set) {
1623 			psta->no_ht_set = 1;
1624 			pmlmepriv->num_sta_no_ht++;
1625 		}
1626 		if (pmlmepriv->htpriv.ht_option) {
1627 			DBG_88E("%s STA %pM - no HT, num of non-HT stations %d\n",
1628 				__func__, (psta->hwaddr),
1629 				pmlmepriv->num_sta_no_ht);
1630 		}
1631 	}
1632 
1633 	if (rtw_ht_operation_update(padapter) > 0) {
1634 		update_beacon(padapter, _HT_CAPABILITY_IE_, NULL, false);
1635 		update_beacon(padapter, _HT_ADD_INFO_IE_, NULL, true);
1636 	}
1637 
1638 	/* update associated stations cap. */
1639 	associated_clients_update(padapter,  beacon_updated);
1640 
1641 	DBG_88E("%s, updated =%d\n", __func__, beacon_updated);
1642 }
1643 
bss_cap_update_on_sta_leave(struct adapter * padapter,struct sta_info * psta)1644 u8 bss_cap_update_on_sta_leave(struct adapter *padapter, struct sta_info *psta)
1645 {
1646 	u8 beacon_updated = false;
1647 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1648 	struct mlme_ext_priv *pmlmeext = &(padapter->mlmeextpriv);
1649 
1650 	if (!psta)
1651 		return beacon_updated;
1652 
1653 	if (psta->no_short_preamble_set) {
1654 		psta->no_short_preamble_set = 0;
1655 		pmlmepriv->num_sta_no_short_preamble--;
1656 		if (pmlmeext->cur_wireless_mode > WIRELESS_11B &&
1657 		    pmlmepriv->num_sta_no_short_preamble == 0) {
1658 			beacon_updated = true;
1659 			update_beacon(padapter, 0xFF, NULL, true);
1660 		}
1661 	}
1662 
1663 	if (psta->nonerp_set) {
1664 		psta->nonerp_set = 0;
1665 		pmlmepriv->num_sta_non_erp--;
1666 		if (pmlmepriv->num_sta_non_erp == 0) {
1667 			beacon_updated = true;
1668 			update_beacon(padapter, _ERPINFO_IE_, NULL, true);
1669 		}
1670 	}
1671 
1672 	if (psta->no_short_slot_time_set) {
1673 		psta->no_short_slot_time_set = 0;
1674 		pmlmepriv->num_sta_no_short_slot_time--;
1675 		if (pmlmeext->cur_wireless_mode > WIRELESS_11B &&
1676 		    pmlmepriv->num_sta_no_short_slot_time == 0) {
1677 			beacon_updated = true;
1678 			update_beacon(padapter, 0xFF, NULL, true);
1679 		}
1680 	}
1681 
1682 	if (psta->no_ht_gf_set) {
1683 		psta->no_ht_gf_set = 0;
1684 		pmlmepriv->num_sta_ht_no_gf--;
1685 	}
1686 
1687 	if (psta->no_ht_set) {
1688 		psta->no_ht_set = 0;
1689 		pmlmepriv->num_sta_no_ht--;
1690 	}
1691 
1692 	if (psta->ht_20mhz_set) {
1693 		psta->ht_20mhz_set = 0;
1694 		pmlmepriv->num_sta_ht_20mhz--;
1695 	}
1696 
1697 	if (rtw_ht_operation_update(padapter) > 0) {
1698 		update_beacon(padapter, _HT_CAPABILITY_IE_, NULL, false);
1699 		update_beacon(padapter, _HT_ADD_INFO_IE_, NULL, true);
1700 	}
1701 
1702 	/* update associated stations cap. */
1703 
1704 	DBG_88E("%s, updated =%d\n", __func__, beacon_updated);
1705 
1706 	return beacon_updated;
1707 }
1708 
ap_free_sta(struct adapter * padapter,struct sta_info * psta,bool active,u16 reason)1709 u8 ap_free_sta(struct adapter *padapter, struct sta_info *psta,
1710 	       bool active, u16 reason)
1711 {
1712 	u8 beacon_updated = false;
1713 	struct sta_priv *pstapriv = &padapter->stapriv;
1714 
1715 	if (!psta)
1716 		return beacon_updated;
1717 
1718 	/* tear down Rx AMPDU */
1719 	send_delba(padapter, 0, psta->hwaddr);/*  recipient */
1720 
1721 	/* tear down TX AMPDU */
1722 	send_delba(padapter, 1, psta->hwaddr);/*  originator */
1723 	psta->htpriv.agg_enable_bitmap = 0x0;/* reset */
1724 	psta->htpriv.candidate_tid_bitmap = 0x0;/* reset */
1725 
1726 	if (active)
1727 		issue_deauth(padapter, psta->hwaddr, reason);
1728 
1729 	/* clear cam entry / key */
1730 	rtw_clearstakey_cmd(padapter, (u8 *)psta, (u8)(psta->mac_id + 3), true);
1731 
1732 
1733 	spin_lock_bh(&psta->lock);
1734 	psta->state &= ~_FW_LINKED;
1735 	spin_unlock_bh(&psta->lock);
1736 
1737 	rtw_indicate_sta_disassoc_event(padapter, psta);
1738 
1739 	report_del_sta_event(padapter, psta->hwaddr, reason);
1740 
1741 	beacon_updated = bss_cap_update_on_sta_leave(padapter, psta);
1742 
1743 	spin_lock_bh(&(pstapriv->sta_hash_lock));
1744 	rtw_free_stainfo(padapter, psta);
1745 	spin_unlock_bh(&(pstapriv->sta_hash_lock));
1746 
1747 	return beacon_updated;
1748 }
1749 
rtw_ap_inform_ch_switch(struct adapter * padapter,u8 new_ch,u8 ch_offset)1750 int rtw_ap_inform_ch_switch(struct adapter *padapter, u8 new_ch, u8 ch_offset)
1751 {
1752 	struct list_head *phead, *plist;
1753 	struct sta_info *psta = NULL;
1754 	struct sta_priv *pstapriv = &padapter->stapriv;
1755 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1756 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1757 	u8 bc_addr[ETH_ALEN] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
1758 
1759 	if ((pmlmeinfo->state&0x03) != WIFI_FW_AP_STATE)
1760 		return 0;
1761 
1762 	DBG_88E(FUNC_NDEV_FMT" with ch:%u, offset:%u\n",
1763 		FUNC_NDEV_ARG(padapter->pnetdev), new_ch, ch_offset);
1764 
1765 	spin_lock_bh(&pstapriv->asoc_list_lock);
1766 	phead = &pstapriv->asoc_list;
1767 	plist = phead->next;
1768 
1769 	/* for each sta in asoc_queue */
1770 	while (phead != plist) {
1771 		psta = container_of(plist, struct sta_info, asoc_list);
1772 		plist = plist->next;
1773 
1774 		issue_action_spct_ch_switch(padapter, psta->hwaddr, new_ch, ch_offset);
1775 		psta->expire_to = ((pstapriv->expire_to * 2) > 5) ? 5 : (pstapriv->expire_to * 2);
1776 	}
1777 	spin_unlock_bh(&pstapriv->asoc_list_lock);
1778 
1779 	issue_action_spct_ch_switch(padapter, bc_addr, new_ch, ch_offset);
1780 
1781 	return 0;
1782 }
1783 
rtw_sta_flush(struct adapter * padapter)1784 int rtw_sta_flush(struct adapter *padapter)
1785 {
1786 	struct list_head *phead, *plist;
1787 	struct sta_info *psta = NULL;
1788 	struct sta_priv *pstapriv = &padapter->stapriv;
1789 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1790 	struct mlme_ext_info	*pmlmeinfo = &(pmlmeext->mlmext_info);
1791 	u8 bc_addr[ETH_ALEN] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
1792 
1793 	DBG_88E(FUNC_NDEV_FMT"\n", FUNC_NDEV_ARG(padapter->pnetdev));
1794 
1795 	if ((pmlmeinfo->state&0x03) != WIFI_FW_AP_STATE)
1796 		return 0;
1797 
1798 	spin_lock_bh(&pstapriv->asoc_list_lock);
1799 	phead = &pstapriv->asoc_list;
1800 	plist = phead->next;
1801 
1802 	/* free sta asoc_queue */
1803 	while (phead != plist) {
1804 		psta = container_of(plist, struct sta_info, asoc_list);
1805 
1806 		plist = plist->next;
1807 
1808 		list_del_init(&psta->asoc_list);
1809 		pstapriv->asoc_list_cnt--;
1810 
1811 		ap_free_sta(padapter, psta, true, WLAN_REASON_DEAUTH_LEAVING);
1812 	}
1813 	spin_unlock_bh(&pstapriv->asoc_list_lock);
1814 
1815 
1816 	issue_deauth(padapter, bc_addr, WLAN_REASON_DEAUTH_LEAVING);
1817 
1818 	associated_clients_update(padapter, true);
1819 
1820 	return 0;
1821 }
1822 
1823 /* called > TSR LEVEL for USB or SDIO Interface*/
sta_info_update(struct adapter * padapter,struct sta_info * psta)1824 void sta_info_update(struct adapter *padapter, struct sta_info *psta)
1825 {
1826 	int flags = psta->flags;
1827 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1828 
1829 	/* update wmm cap. */
1830 	if (WLAN_STA_WME&flags)
1831 		psta->qos_option = 1;
1832 	else
1833 		psta->qos_option = 0;
1834 
1835 	if (pmlmepriv->qospriv.qos_option == 0)
1836 		psta->qos_option = 0;
1837 
1838 	/* update 802.11n ht cap. */
1839 	if (WLAN_STA_HT&flags) {
1840 		psta->htpriv.ht_option = true;
1841 		psta->qos_option = 1;
1842 	} else {
1843 		psta->htpriv.ht_option = false;
1844 	}
1845 
1846 	if (!pmlmepriv->htpriv.ht_option)
1847 		psta->htpriv.ht_option = false;
1848 
1849 	update_sta_info_apmode(padapter, psta);
1850 }
1851 
1852 /* called >= TSR LEVEL for USB or SDIO Interface*/
ap_sta_info_defer_update(struct adapter * padapter,struct sta_info * psta)1853 void ap_sta_info_defer_update(struct adapter *padapter, struct sta_info *psta)
1854 {
1855 	if (psta->state & _FW_LINKED) {
1856 		/* add ratid */
1857 		add_RATid(padapter, psta, 0);/* DM_RATR_STA_INIT */
1858 	}
1859 }
1860 
start_ap_mode(struct adapter * padapter)1861 void start_ap_mode(struct adapter *padapter)
1862 {
1863 	int i;
1864 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1865 	struct sta_priv *pstapriv = &padapter->stapriv;
1866 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1867 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
1868 
1869 	pmlmepriv->update_bcn = false;
1870 
1871 	pmlmeext->bstart_bss = false;
1872 
1873 	pmlmepriv->num_sta_non_erp = 0;
1874 
1875 	pmlmepriv->num_sta_no_short_slot_time = 0;
1876 
1877 	pmlmepriv->num_sta_no_short_preamble = 0;
1878 
1879 	pmlmepriv->num_sta_ht_no_gf = 0;
1880 	pmlmepriv->num_sta_no_ht = 0;
1881 	pmlmepriv->num_sta_ht_20mhz = 0;
1882 
1883 	pmlmepriv->olbc = false;
1884 
1885 	pmlmepriv->olbc_ht = false;
1886 
1887 	pmlmepriv->ht_op_mode = 0;
1888 
1889 	for (i = 0; i < NUM_STA; i++)
1890 		pstapriv->sta_aid[i] = NULL;
1891 
1892 	pmlmepriv->wps_beacon_ie = NULL;
1893 	pmlmepriv->wps_probe_resp_ie = NULL;
1894 	pmlmepriv->wps_assoc_resp_ie = NULL;
1895 
1896 	pmlmepriv->p2p_beacon_ie = NULL;
1897 	pmlmepriv->p2p_probe_resp_ie = NULL;
1898 
1899 	/* for ACL */
1900 	INIT_LIST_HEAD(&(pacl_list->acl_node_q.queue));
1901 	pacl_list->num = 0;
1902 	pacl_list->mode = 0;
1903 	for (i = 0; i < NUM_ACL; i++) {
1904 		INIT_LIST_HEAD(&pacl_list->aclnode[i].list);
1905 		pacl_list->aclnode[i].valid = false;
1906 	}
1907 }
1908 
stop_ap_mode(struct adapter * padapter)1909 void stop_ap_mode(struct adapter *padapter)
1910 {
1911 	struct list_head *phead, *plist;
1912 	struct rtw_wlan_acl_node *paclnode;
1913 	struct sta_info *psta = NULL;
1914 	struct sta_priv *pstapriv = &padapter->stapriv;
1915 	struct mlme_priv *pmlmepriv = &(padapter->mlmepriv);
1916 	struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1917 	struct wlan_acl_pool *pacl_list = &pstapriv->acl_list;
1918 	struct __queue *pacl_node_q = &pacl_list->acl_node_q;
1919 
1920 	pmlmepriv->update_bcn = false;
1921 	pmlmeext->bstart_bss = false;
1922 
1923 	/* reset and init security priv , this can refine with rtw_reset_securitypriv */
1924 	memset((unsigned char *)&padapter->securitypriv, 0, sizeof(struct security_priv));
1925 	padapter->securitypriv.ndisauthtype = Ndis802_11AuthModeOpen;
1926 	padapter->securitypriv.ndisencryptstatus = Ndis802_11WEPDisabled;
1927 
1928 	/* for ACL */
1929 	spin_lock_bh(&(pacl_node_q->lock));
1930 	phead = get_list_head(pacl_node_q);
1931 	plist = phead->next;
1932 	while (phead != plist) {
1933 		paclnode = container_of(plist, struct rtw_wlan_acl_node, list);
1934 		plist = plist->next;
1935 
1936 		if (paclnode->valid) {
1937 			paclnode->valid = false;
1938 
1939 			list_del_init(&paclnode->list);
1940 
1941 			pacl_list->num--;
1942 		}
1943 	}
1944 	spin_unlock_bh(&(pacl_node_q->lock));
1945 
1946 	DBG_88E("%s, free acl_node_queue, num =%d\n", __func__, pacl_list->num);
1947 
1948 	rtw_sta_flush(padapter);
1949 
1950 	/* free_assoc_sta_resources */
1951 	rtw_free_all_stainfo(padapter);
1952 
1953 	psta = rtw_get_bcmc_stainfo(padapter);
1954 	spin_lock_bh(&(pstapriv->sta_hash_lock));
1955 	rtw_free_stainfo(padapter, psta);
1956 	spin_unlock_bh(&(pstapriv->sta_hash_lock));
1957 
1958 	rtw_init_bcmc_stainfo(padapter);
1959 
1960 	rtw_free_mlme_priv_ie_data(pmlmepriv);
1961 }
1962 
1963 #endif /* CONFIG_88EU_AP_MODE */
1964