GNU Linux-libre 4.14.290-gnu1
[releases.git] / drivers / staging / rtl8188eu / core / rtw_wlan_util.c
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  ******************************************************************************/
15 #define _RTW_WLAN_UTIL_C_
16
17 #include <linux/ieee80211.h>
18
19 #include <osdep_service.h>
20 #include <drv_types.h>
21 #include <wifi.h>
22
23 static unsigned char ARTHEROS_OUI1[] = {0x00, 0x03, 0x7f};
24 static unsigned char ARTHEROS_OUI2[] = {0x00, 0x13, 0x74};
25
26 static unsigned char BROADCOM_OUI1[] = {0x00, 0x10, 0x18};
27 static unsigned char BROADCOM_OUI2[] = {0x00, 0x0a, 0xf7};
28
29 static unsigned char CISCO_OUI[] = {0x00, 0x40, 0x96};
30 static unsigned char MARVELL_OUI[] = {0x00, 0x50, 0x43};
31 static unsigned char RALINK_OUI[] = {0x00, 0x0c, 0x43};
32 static unsigned char REALTEK_OUI[] = {0x00, 0xe0, 0x4c};
33 static unsigned char AIRGOCAP_OUI[] = {0x00, 0x0a, 0xf5};
34 static unsigned char EPIGRAM_OUI[] = {0x00, 0x90, 0x4c};
35
36 unsigned char REALTEK_96B_IE[] = {0x00, 0xe0, 0x4c, 0x02, 0x01, 0x20};
37
38 #define R2T_PHY_DELAY   (0)
39
40 /* define WAIT_FOR_BCN_TO_M     (3000) */
41 #define WAIT_FOR_BCN_TO_MIN     (6000)
42 #define WAIT_FOR_BCN_TO_MAX     (20000)
43
44 static u8 rtw_basic_rate_cck[4] = {
45         IEEE80211_CCK_RATE_1MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_2MB|IEEE80211_BASIC_RATE_MASK,
46         IEEE80211_CCK_RATE_5MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_11MB|IEEE80211_BASIC_RATE_MASK
47 };
48
49 static u8 rtw_basic_rate_ofdm[3] = {
50         IEEE80211_OFDM_RATE_6MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_OFDM_RATE_12MB|IEEE80211_BASIC_RATE_MASK,
51         IEEE80211_OFDM_RATE_24MB|IEEE80211_BASIC_RATE_MASK
52 };
53
54 static u8 rtw_basic_rate_mix[7] = {
55         IEEE80211_CCK_RATE_1MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_2MB|IEEE80211_BASIC_RATE_MASK,
56         IEEE80211_CCK_RATE_5MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_CCK_RATE_11MB|IEEE80211_BASIC_RATE_MASK,
57         IEEE80211_OFDM_RATE_6MB|IEEE80211_BASIC_RATE_MASK, IEEE80211_OFDM_RATE_12MB|IEEE80211_BASIC_RATE_MASK,
58         IEEE80211_OFDM_RATE_24MB|IEEE80211_BASIC_RATE_MASK
59 };
60
61 int cckrates_included(unsigned char *rate, int ratelen)
62 {
63         int     i;
64
65         for (i = 0; i < ratelen; i++) {
66                 if  ((((rate[i]) & 0x7f) == 2)  || (((rate[i]) & 0x7f) == 4) ||
67                      (((rate[i]) & 0x7f) == 11)  || (((rate[i]) & 0x7f) == 22))
68                         return true;
69         }
70         return false;
71 }
72
73 int cckratesonly_included(unsigned char *rate, int ratelen)
74 {
75         int     i;
76
77         for (i = 0; i < ratelen; i++) {
78                 if  ((((rate[i]) & 0x7f) != 2) && (((rate[i]) & 0x7f) != 4) &&
79                            (((rate[i]) & 0x7f) != 11)  && (((rate[i]) & 0x7f) != 22))
80                         return false;
81         }
82
83         return true;
84 }
85
86 unsigned char networktype_to_raid(unsigned char network_type)
87 {
88         switch (network_type) {
89         case WIRELESS_11B:
90                 return RATR_INX_WIRELESS_B;
91         case WIRELESS_11A:
92         case WIRELESS_11G:
93                 return RATR_INX_WIRELESS_G;
94         case WIRELESS_11BG:
95                 return RATR_INX_WIRELESS_GB;
96         case WIRELESS_11_24N:
97         case WIRELESS_11_5N:
98                 return RATR_INX_WIRELESS_N;
99         case WIRELESS_11A_5N:
100         case WIRELESS_11G_24N:
101                 return  RATR_INX_WIRELESS_NG;
102         case WIRELESS_11BG_24N:
103                 return RATR_INX_WIRELESS_NGB;
104         default:
105                 return RATR_INX_WIRELESS_GB;
106         }
107 }
108
109 u8 judge_network_type(struct adapter *padapter, unsigned char *rate, int ratelen)
110 {
111         u8 network_type = 0;
112         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
113         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
114
115         if (pmlmeext->cur_channel > 14) {
116                 if (pmlmeinfo->HT_enable)
117                         network_type = WIRELESS_11_5N;
118
119                 network_type |= WIRELESS_11A;
120         } else {
121                 if (pmlmeinfo->HT_enable)
122                         network_type = WIRELESS_11_24N;
123
124                 if ((cckratesonly_included(rate, ratelen)) == true)
125                         network_type |= WIRELESS_11B;
126                 else if ((cckrates_included(rate, ratelen)) == true)
127                         network_type |= WIRELESS_11BG;
128                 else
129                         network_type |= WIRELESS_11G;
130         }
131         return  network_type;
132 }
133
134 static unsigned char ratetbl_val_2wifirate(unsigned char rate)
135 {
136         switch (rate & 0x7f) {
137         case 0:
138                 return IEEE80211_CCK_RATE_1MB;
139         case 1:
140                 return IEEE80211_CCK_RATE_2MB;
141         case 2:
142                 return IEEE80211_CCK_RATE_5MB;
143         case 3:
144                 return IEEE80211_CCK_RATE_11MB;
145         case 4:
146                 return IEEE80211_OFDM_RATE_6MB;
147         case 5:
148                 return IEEE80211_OFDM_RATE_9MB;
149         case 6:
150                 return IEEE80211_OFDM_RATE_12MB;
151         case 7:
152                 return IEEE80211_OFDM_RATE_18MB;
153         case 8:
154                 return IEEE80211_OFDM_RATE_24MB;
155         case 9:
156                 return IEEE80211_OFDM_RATE_36MB;
157         case 10:
158                 return IEEE80211_OFDM_RATE_48MB;
159         case 11:
160                 return IEEE80211_OFDM_RATE_54MB;
161         default:
162                 return 0;
163         }
164 }
165
166 static int is_basicrate(struct adapter *padapter, unsigned char rate)
167 {
168         int i;
169         unsigned char val;
170         struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
171
172         for (i = 0; i < NumRates; i++) {
173                 val = pmlmeext->basicrate[i];
174
175                 if ((val != 0xff) && (val != 0xfe)) {
176                         if (rate == ratetbl_val_2wifirate(val))
177                                 return true;
178                 }
179         }
180         return false;
181 }
182
183 static unsigned int ratetbl2rateset(struct adapter *padapter, unsigned char *rateset)
184 {
185         int i;
186         unsigned char rate;
187         unsigned int    len = 0;
188         struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
189
190         for (i = 0; i < NumRates; i++) {
191                 rate = pmlmeext->datarate[i];
192
193                 switch (rate) {
194                 case 0xff:
195                         return len;
196                 case 0xfe:
197                         continue;
198                 default:
199                         rate = ratetbl_val_2wifirate(rate);
200
201                         if (is_basicrate(padapter, rate) == true)
202                                 rate |= IEEE80211_BASIC_RATE_MASK;
203
204                         rateset[len] = rate;
205                         len++;
206                         break;
207                 }
208         }
209         return len;
210 }
211
212 void get_rate_set(struct adapter *padapter, unsigned char *pbssrate, int *bssrate_len)
213 {
214         unsigned char supportedrates[NumRates];
215
216         memset(supportedrates, 0, NumRates);
217         *bssrate_len = ratetbl2rateset(padapter, supportedrates);
218         memcpy(pbssrate, supportedrates, *bssrate_len);
219 }
220
221 void UpdateBrateTbl(struct adapter *Adapter, u8 *mbrate)
222 {
223         u8      i;
224         u8      rate;
225
226         /*  1M, 2M, 5.5M, 11M, 6M, 12M, 24M are mandatory. */
227         for (i = 0; i < NDIS_802_11_LENGTH_RATES_EX; i++) {
228                 rate = mbrate[i] & 0x7f;
229                 switch (rate) {
230                 case IEEE80211_CCK_RATE_1MB:
231                 case IEEE80211_CCK_RATE_2MB:
232                 case IEEE80211_CCK_RATE_5MB:
233                 case IEEE80211_CCK_RATE_11MB:
234                 case IEEE80211_OFDM_RATE_6MB:
235                 case IEEE80211_OFDM_RATE_12MB:
236                 case IEEE80211_OFDM_RATE_24MB:
237                         mbrate[i] |= IEEE80211_BASIC_RATE_MASK;
238                         break;
239                 }
240         }
241 }
242
243 void UpdateBrateTblForSoftAP(u8 *bssrateset, u32 bssratelen)
244 {
245         u8      i;
246         u8      rate;
247
248         for (i = 0; i < bssratelen; i++) {
249                 rate = bssrateset[i] & 0x7f;
250                 switch (rate) {
251                 case IEEE80211_CCK_RATE_1MB:
252                 case IEEE80211_CCK_RATE_2MB:
253                 case IEEE80211_CCK_RATE_5MB:
254                 case IEEE80211_CCK_RATE_11MB:
255                         bssrateset[i] |= IEEE80211_BASIC_RATE_MASK;
256                         break;
257                 }
258         }
259 }
260
261 void Save_DM_Func_Flag(struct adapter *padapter)
262 {
263         u8      saveflag = true;
264
265         rtw_hal_set_hwreg(padapter, HW_VAR_DM_FUNC_OP, (u8 *)(&saveflag));
266 }
267
268 void Restore_DM_Func_Flag(struct adapter *padapter)
269 {
270         u8      saveflag = false;
271
272         rtw_hal_set_hwreg(padapter, HW_VAR_DM_FUNC_OP, (u8 *)(&saveflag));
273 }
274
275 void Switch_DM_Func(struct adapter *padapter, u32 mode, u8 enable)
276 {
277         if (enable)
278                 rtw_hal_set_hwreg(padapter, HW_VAR_DM_FUNC_SET, (u8 *)(&mode));
279         else
280                 rtw_hal_set_hwreg(padapter, HW_VAR_DM_FUNC_CLR, (u8 *)(&mode));
281 }
282
283 static void Set_NETYPE0_MSR(struct adapter *padapter, u8 type)
284 {
285         rtw_hal_set_hwreg(padapter, HW_VAR_MEDIA_STATUS, (u8 *)(&type));
286 }
287
288 void Set_MSR(struct adapter *padapter, u8 type)
289 {
290         Set_NETYPE0_MSR(padapter, type);
291 }
292
293 inline u8 rtw_get_oper_ch(struct adapter *adapter)
294 {
295         return adapter->mlmeextpriv.oper_channel;
296 }
297
298 inline void rtw_set_oper_ch(struct adapter *adapter, u8 ch)
299 {
300         adapter->mlmeextpriv.oper_channel = ch;
301 }
302
303 inline u8 rtw_get_oper_bw(struct adapter *adapter)
304 {
305         return adapter->mlmeextpriv.oper_bwmode;
306 }
307
308 inline void rtw_set_oper_bw(struct adapter *adapter, u8 bw)
309 {
310         adapter->mlmeextpriv.oper_bwmode = bw;
311 }
312
313 inline u8 rtw_get_oper_choffset(struct adapter *adapter)
314 {
315         return adapter->mlmeextpriv.oper_ch_offset;
316 }
317
318 inline void rtw_set_oper_choffset(struct adapter *adapter, u8 offset)
319 {
320         adapter->mlmeextpriv.oper_ch_offset = offset;
321 }
322
323 void SelectChannel(struct adapter *padapter, unsigned char channel)
324 {
325         /* saved channel info */
326         rtw_set_oper_ch(padapter, channel);
327         rtw_hal_set_chan(padapter, channel);
328 }
329
330 void SetBWMode(struct adapter *padapter, unsigned short bwmode,
331                unsigned char channel_offset)
332 {
333         /* saved bw info */
334         rtw_set_oper_bw(padapter, bwmode);
335         rtw_set_oper_choffset(padapter, channel_offset);
336
337         rtw_hal_set_bwmode(padapter, (enum ht_channel_width)bwmode, channel_offset);
338 }
339
340 void set_channel_bwmode(struct adapter *padapter, unsigned char channel, unsigned char channel_offset, unsigned short bwmode)
341 {
342         u8 center_ch;
343
344         if ((bwmode == HT_CHANNEL_WIDTH_20) ||
345             (channel_offset == HAL_PRIME_CHNL_OFFSET_DONT_CARE)) {
346                 /* SelectChannel(padapter, channel); */
347                 center_ch = channel;
348         } else {
349                 /* switch to the proper channel */
350                 if (channel_offset == HAL_PRIME_CHNL_OFFSET_LOWER) {
351                         /* SelectChannel(padapter, channel + 2); */
352                         center_ch = channel + 2;
353                 } else {
354                         /* SelectChannel(padapter, channel - 2); */
355                         center_ch = channel - 2;
356                 }
357         }
358
359         /* set Channel */
360         /* saved channel/bw info */
361         rtw_set_oper_ch(padapter, channel);
362         rtw_set_oper_bw(padapter, bwmode);
363         rtw_set_oper_choffset(padapter, channel_offset);
364
365         rtw_hal_set_chan(padapter, center_ch); /*  set center channel */
366         SetBWMode(padapter, bwmode, channel_offset);
367 }
368
369 int get_bsstype(unsigned short capability)
370 {
371         if (capability & BIT(0))
372                 return WIFI_FW_AP_STATE;
373         else if (capability & BIT(1))
374                 return WIFI_FW_ADHOC_STATE;
375         else
376                 return 0;
377 }
378
379 u16 get_beacon_interval(struct wlan_bssid_ex *bss)
380 {
381         __le16 val;
382
383         memcpy((unsigned char *)&val, rtw_get_beacon_interval_from_ie(bss->IEs), 2);
384
385         return le16_to_cpu(val);
386 }
387
388 int is_client_associated_to_ap(struct adapter *padapter)
389 {
390         struct mlme_ext_priv    *pmlmeext;
391         struct mlme_ext_info    *pmlmeinfo;
392
393         if (!padapter)
394                 return _FAIL;
395
396         pmlmeext = &padapter->mlmeextpriv;
397         pmlmeinfo = &(pmlmeext->mlmext_info);
398
399         if ((pmlmeinfo->state & WIFI_FW_ASSOC_SUCCESS) && ((pmlmeinfo->state&0x03) == WIFI_FW_STATION_STATE))
400                 return true;
401         else
402                 return _FAIL;
403 }
404
405 int is_client_associated_to_ibss(struct adapter *padapter)
406 {
407         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
408         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
409
410         if ((pmlmeinfo->state & WIFI_FW_ASSOC_SUCCESS) && ((pmlmeinfo->state&0x03) == WIFI_FW_ADHOC_STATE))
411                 return true;
412         else
413                 return _FAIL;
414 }
415
416 int is_IBSS_empty(struct adapter *padapter)
417 {
418         unsigned int i;
419         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
420         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
421
422         for (i = IBSS_START_MAC_ID; i < NUM_STA; i++) {
423                 if (pmlmeinfo->FW_sta_info[i].status == 1)
424                         return _FAIL;
425         }
426         return true;
427 }
428
429 unsigned int decide_wait_for_beacon_timeout(unsigned int bcn_interval)
430 {
431         if ((bcn_interval << 2) < WAIT_FOR_BCN_TO_MIN)
432                 return WAIT_FOR_BCN_TO_MIN;
433         else if ((bcn_interval << 2) > WAIT_FOR_BCN_TO_MAX)
434                 return WAIT_FOR_BCN_TO_MAX;
435         else
436                 return bcn_interval << 2;
437 }
438
439 void CAM_empty_entry(struct adapter *Adapter, u8 ucIndex)
440 {
441         rtw_hal_set_hwreg(Adapter, HW_VAR_CAM_EMPTY_ENTRY, (u8 *)(&ucIndex));
442 }
443
444 void invalidate_cam_all(struct adapter *padapter)
445 {
446         rtw_hal_set_hwreg(padapter, HW_VAR_CAM_INVALID_ALL, NULL);
447 }
448
449 void write_cam(struct adapter *padapter, u8 entry, u16 ctrl, u8 *mac, u8 *key)
450 {
451         unsigned int    i, val, addr;
452         int j;
453         u32     cam_val[2];
454
455         addr = entry << 3;
456
457         for (j = 5; j >= 0; j--) {
458                 switch (j) {
459                 case 0:
460                         val = ctrl | (mac[0] << 16) | (mac[1] << 24);
461                         break;
462                 case 1:
463                         val = mac[2] | (mac[3] << 8) | (mac[4] << 16) | (mac[5] << 24);
464                         break;
465                 default:
466                         i = (j - 2) << 2;
467                         val = key[i] | (key[i+1] << 8) | (key[i+2] << 16) | (key[i+3] << 24);
468                         break;
469                 }
470
471                 cam_val[0] = val;
472                 cam_val[1] = addr + (unsigned int)j;
473
474                 rtw_hal_set_hwreg(padapter, HW_VAR_CAM_WRITE, (u8 *)cam_val);
475         }
476 }
477
478 void clear_cam_entry(struct adapter *padapter, u8 entry)
479 {
480         unsigned char null_sta[] = {0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
481         unsigned char null_key[] = {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
482                                     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
483
484         write_cam(padapter, entry, 0, null_sta, null_key);
485 }
486
487 int allocate_fw_sta_entry(struct adapter *padapter)
488 {
489         unsigned int mac_id;
490         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
491         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
492
493         for (mac_id = IBSS_START_MAC_ID; mac_id < NUM_STA; mac_id++) {
494                 if (pmlmeinfo->FW_sta_info[mac_id].status == 0) {
495                         pmlmeinfo->FW_sta_info[mac_id].status = 1;
496                         pmlmeinfo->FW_sta_info[mac_id].retry = 0;
497                         break;
498                 }
499         }
500
501         return mac_id;
502 }
503
504 void flush_all_cam_entry(struct adapter *padapter)
505 {
506         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
507         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
508
509         rtw_hal_set_hwreg(padapter, HW_VAR_CAM_INVALID_ALL, NULL);
510
511         memset((u8 *)(pmlmeinfo->FW_sta_info), 0, sizeof(pmlmeinfo->FW_sta_info));
512 }
513
514 int WMM_param_handler(struct adapter *padapter, struct ndis_802_11_var_ie *pIE)
515 {
516         /* struct registry_priv *pregpriv = &padapter->registrypriv; */
517         struct mlme_priv        *pmlmepriv = &(padapter->mlmepriv);
518         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
519         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
520
521         if (pmlmepriv->qospriv.qos_option == 0) {
522                 pmlmeinfo->WMM_enable = 0;
523                 return _FAIL;
524         }
525
526         pmlmeinfo->WMM_enable = 1;
527         memcpy(&(pmlmeinfo->WMM_param), (pIE->data + 6), sizeof(struct WMM_para_element));
528         return true;
529 }
530
531 void WMMOnAssocRsp(struct adapter *padapter)
532 {
533         u8      ACI, ACM, AIFS, ECWMin, ECWMax, aSifsTime;
534         u8      acm_mask;
535         u16     TXOP;
536         u32     acParm, i;
537         u32     edca[4], inx[4];
538         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
539         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
540         struct xmit_priv                *pxmitpriv = &padapter->xmitpriv;
541         struct registry_priv    *pregpriv = &padapter->registrypriv;
542
543         if (pmlmeinfo->WMM_enable == 0) {
544                 padapter->mlmepriv.acm_mask = 0;
545                 return;
546         }
547
548         acm_mask = 0;
549
550         if (pmlmeext->cur_wireless_mode == WIRELESS_11B)
551                 aSifsTime = 10;
552         else
553                 aSifsTime = 16;
554
555         for (i = 0; i < 4; i++) {
556                 ACI = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN >> 5) & 0x03;
557                 ACM = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN >> 4) & 0x01;
558
559                 /* AIFS = AIFSN * slot time + SIFS - r2t phy delay */
560                 AIFS = (pmlmeinfo->WMM_param.ac_param[i].ACI_AIFSN & 0x0f) * pmlmeinfo->slotTime + aSifsTime;
561
562                 ECWMin = pmlmeinfo->WMM_param.ac_param[i].CW & 0x0f;
563                 ECWMax = (pmlmeinfo->WMM_param.ac_param[i].CW & 0xf0) >> 4;
564                 TXOP = le16_to_cpu(pmlmeinfo->WMM_param.ac_param[i].TXOP_limit);
565
566                 acParm = AIFS | (ECWMin << 8) | (ECWMax << 12) | (TXOP << 16);
567
568                 switch (ACI) {
569                 case 0x0:
570                         rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_BE, (u8 *)(&acParm));
571                         acm_mask |= (ACM ? BIT(1) : 0);
572                         edca[XMIT_BE_QUEUE] = acParm;
573                         break;
574                 case 0x1:
575                         rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_BK, (u8 *)(&acParm));
576                         edca[XMIT_BK_QUEUE] = acParm;
577                         break;
578                 case 0x2:
579                         rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_VI, (u8 *)(&acParm));
580                         acm_mask |= (ACM ? BIT(2) : 0);
581                         edca[XMIT_VI_QUEUE] = acParm;
582                         break;
583                 case 0x3:
584                         rtw_hal_set_hwreg(padapter, HW_VAR_AC_PARAM_VO, (u8 *)(&acParm));
585                         acm_mask |= (ACM ? BIT(3) : 0);
586                         edca[XMIT_VO_QUEUE] = acParm;
587                         break;
588                 }
589
590                 DBG_88E("WMM(%x): %x, %x\n", ACI, ACM, acParm);
591         }
592
593         if (padapter->registrypriv.acm_method == 1)
594                 rtw_hal_set_hwreg(padapter, HW_VAR_ACM_CTRL, (u8 *)(&acm_mask));
595         else
596                 padapter->mlmepriv.acm_mask = acm_mask;
597
598         inx[0] = 0; inx[1] = 1; inx[2] = 2; inx[3] = 3;
599
600         if (pregpriv->wifi_spec == 1) {
601                 u32     j, change_inx = false;
602
603                 /* entry indx: 0->vo, 1->vi, 2->be, 3->bk. */
604                 for (i = 0; i < 4; i++) {
605                         for (j = i+1; j < 4; j++) {
606                                 /* compare CW and AIFS */
607                                 if ((edca[j] & 0xFFFF) < (edca[i] & 0xFFFF)) {
608                                         change_inx = true;
609                                 } else if ((edca[j] & 0xFFFF) == (edca[i] & 0xFFFF)) {
610                                         /* compare TXOP */
611                                         if ((edca[j] >> 16) > (edca[i] >> 16))
612                                                 change_inx = true;
613                                 }
614
615                                 if (change_inx) {
616                                         swap(edca[i], edca[j]);
617                                         swap(inx[i], inx[j]);
618                                         change_inx = false;
619                                 }
620                         }
621                 }
622         }
623
624         for (i = 0; i < 4; i++) {
625                 pxmitpriv->wmm_para_seq[i] = inx[i];
626                 DBG_88E("wmm_para_seq(%d): %d\n", i, pxmitpriv->wmm_para_seq[i]);
627         }
628 }
629
630 static void bwmode_update_check(struct adapter *padapter, struct ndis_802_11_var_ie *pIE)
631 {
632         unsigned char    new_bwmode;
633         unsigned char  new_ch_offset;
634         struct HT_info_element   *pHT_info;
635         struct mlme_priv        *pmlmepriv = &(padapter->mlmepriv);
636         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
637         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
638         struct registry_priv *pregistrypriv = &padapter->registrypriv;
639         struct ht_priv                  *phtpriv = &pmlmepriv->htpriv;
640
641         if (!pIE)
642                 return;
643
644         if (!phtpriv)
645                 return;
646
647         if (pIE->Length > sizeof(struct HT_info_element))
648                 return;
649
650         pHT_info = (struct HT_info_element *)pIE->data;
651
652         if ((pHT_info->infos[0] & BIT(2)) && pregistrypriv->cbw40_enable) {
653                 new_bwmode = HT_CHANNEL_WIDTH_40;
654
655                 switch (pHT_info->infos[0] & 0x3) {
656                 case 1:
657                         new_ch_offset = HAL_PRIME_CHNL_OFFSET_LOWER;
658                         break;
659                 case 3:
660                         new_ch_offset = HAL_PRIME_CHNL_OFFSET_UPPER;
661                         break;
662                 default:
663                         new_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
664                         break;
665                 }
666         } else {
667                 new_bwmode = HT_CHANNEL_WIDTH_20;
668                 new_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
669         }
670
671         if ((new_bwmode != pmlmeext->cur_bwmode) ||
672             (new_ch_offset != pmlmeext->cur_ch_offset)) {
673                 pmlmeinfo->bwmode_updated = true;
674
675                 pmlmeext->cur_bwmode = new_bwmode;
676                 pmlmeext->cur_ch_offset = new_ch_offset;
677
678                 /* update HT info also */
679                 HT_info_handler(padapter, pIE);
680         } else {
681                 pmlmeinfo->bwmode_updated = false;
682         }
683
684         if (pmlmeinfo->bwmode_updated) {
685                 struct sta_info *psta;
686                 struct wlan_bssid_ex    *cur_network = &(pmlmeinfo->network);
687                 struct sta_priv *pstapriv = &padapter->stapriv;
688
689                 /* set_channel_bwmode(padapter, pmlmeext->cur_channel, pmlmeext->cur_ch_offset, pmlmeext->cur_bwmode); */
690
691                 /* update ap's stainfo */
692                 psta = rtw_get_stainfo(pstapriv, cur_network->MacAddress);
693                 if (psta) {
694                         struct ht_priv  *phtpriv_sta = &psta->htpriv;
695
696                         if (phtpriv_sta->ht_option) {
697                                 /*  bwmode */
698                                 phtpriv_sta->bwmode = pmlmeext->cur_bwmode;
699                                 phtpriv_sta->ch_offset = pmlmeext->cur_ch_offset;
700                         } else {
701                                 phtpriv_sta->bwmode = HT_CHANNEL_WIDTH_20;
702                                 phtpriv_sta->ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
703                         }
704                 }
705         }
706 }
707
708 void HT_caps_handler(struct adapter *padapter, struct ndis_802_11_var_ie *pIE)
709 {
710         unsigned int    i;
711         u8      max_AMPDU_len, min_MPDU_spacing;
712         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
713         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
714         struct mlme_priv                *pmlmepriv = &padapter->mlmepriv;
715         struct ht_priv                  *phtpriv = &pmlmepriv->htpriv;
716         u8 *HT_cap = (u8 *)(&pmlmeinfo->HT_caps);
717
718         if (pIE == NULL)
719                 return;
720
721         if (!phtpriv->ht_option)
722                 return;
723
724         pmlmeinfo->HT_caps_enable = 1;
725
726         for (i = 0; i < (pIE->Length); i++) {
727                 if (i != 2) {
728                         /*      Got the endian issue here. */
729                         HT_cap[i] &= (pIE->data[i]);
730                 } else {
731                         /* modify from  fw by Thomas 2010/11/17 */
732                         if ((pmlmeinfo->HT_caps.ampdu_params_info & 0x3) > (pIE->data[i] & 0x3))
733                                 max_AMPDU_len = pIE->data[i] & 0x3;
734                         else
735                                 max_AMPDU_len = pmlmeinfo->HT_caps.ampdu_params_info & 0x3;
736
737                         if ((pmlmeinfo->HT_caps.ampdu_params_info & 0x1c) > (pIE->data[i] & 0x1c))
738                                 min_MPDU_spacing = pmlmeinfo->HT_caps.ampdu_params_info & 0x1c;
739                         else
740                                 min_MPDU_spacing = pIE->data[i] & 0x1c;
741
742                         pmlmeinfo->HT_caps.ampdu_params_info = max_AMPDU_len | min_MPDU_spacing;
743                 }
744         }
745
746         /* update the MCS rates */
747         for (i = 0; i < 16; i++)
748                 ((u8 *)&pmlmeinfo->HT_caps.mcs)[i] &= MCS_rate_1R[i];
749 }
750
751 void HT_info_handler(struct adapter *padapter, struct ndis_802_11_var_ie *pIE)
752 {
753         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
754         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
755         struct mlme_priv                *pmlmepriv = &padapter->mlmepriv;
756         struct ht_priv                  *phtpriv = &pmlmepriv->htpriv;
757
758         if (pIE == NULL)
759                 return;
760
761         if (!phtpriv->ht_option)
762                 return;
763
764         if (pIE->Length > sizeof(struct HT_info_element))
765                 return;
766
767         pmlmeinfo->HT_info_enable = 1;
768         memcpy(&(pmlmeinfo->HT_info), pIE->data, pIE->Length);
769 }
770
771 void HTOnAssocRsp(struct adapter *padapter)
772 {
773         unsigned char           max_AMPDU_len;
774         unsigned char           min_MPDU_spacing;
775         /* struct registry_priv  *pregpriv = &padapter->registrypriv; */
776         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
777         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
778
779         DBG_88E("%s\n", __func__);
780
781         if ((pmlmeinfo->HT_info_enable) && (pmlmeinfo->HT_caps_enable)) {
782                 pmlmeinfo->HT_enable = 1;
783         } else {
784                 pmlmeinfo->HT_enable = 0;
785                 return;
786         }
787
788         /* handle A-MPDU parameter field */
789         /*
790                 AMPDU_para [1:0]:Max AMPDU Len => 0:8k , 1:16k, 2:32k, 3:64k
791                 AMPDU_para [4:2]:Min MPDU Start Spacing
792         */
793         max_AMPDU_len = pmlmeinfo->HT_caps.ampdu_params_info & 0x03;
794
795         min_MPDU_spacing = (pmlmeinfo->HT_caps.ampdu_params_info & 0x1c) >> 2;
796
797         rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_MIN_SPACE, (u8 *)(&min_MPDU_spacing));
798
799         rtw_hal_set_hwreg(padapter, HW_VAR_AMPDU_FACTOR, (u8 *)(&max_AMPDU_len));
800 }
801
802 void ERP_IE_handler(struct adapter *padapter, struct ndis_802_11_var_ie *pIE)
803 {
804         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
805         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
806
807         if (pIE->Length > 1)
808                 return;
809
810         pmlmeinfo->ERP_enable = 1;
811         memcpy(&(pmlmeinfo->ERP_IE), pIE->data, pIE->Length);
812 }
813
814 void VCS_update(struct adapter *padapter, struct sta_info *psta)
815 {
816         struct registry_priv     *pregpriv = &padapter->registrypriv;
817         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
818         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
819
820         switch (pregpriv->vrtl_carrier_sense) { /* 0:off 1:on 2:auto */
821         case 0: /* off */
822                 psta->rtsen = 0;
823                 psta->cts2self = 0;
824                 break;
825         case 1: /* on */
826                 if (pregpriv->vcs_type == 1) { /* 1:RTS/CTS 2:CTS to self */
827                         psta->rtsen = 1;
828                         psta->cts2self = 0;
829                 } else {
830                         psta->rtsen = 0;
831                         psta->cts2self = 1;
832                 }
833                 break;
834         case 2: /* auto */
835         default:
836                 if ((pmlmeinfo->ERP_enable) && (pmlmeinfo->ERP_IE & BIT(1))) {
837                         if (pregpriv->vcs_type == 1) {
838                                 psta->rtsen = 1;
839                                 psta->cts2self = 0;
840                         } else {
841                                 psta->rtsen = 0;
842                                 psta->cts2self = 1;
843                         }
844                 } else {
845                         psta->rtsen = 0;
846                         psta->cts2self = 0;
847                 }
848                 break;
849         }
850 }
851
852 int rtw_check_bcn_info(struct adapter  *Adapter, u8 *pframe, u32 packet_len)
853 {
854         unsigned int            len;
855         unsigned char           *p;
856         unsigned short  val16, subtype;
857         struct wlan_network *cur_network = &(Adapter->mlmepriv.cur_network);
858         /* u8 wpa_ie[255], rsn_ie[255]; */
859         u16 wpa_len = 0, rsn_len = 0;
860         u8 encryp_protocol = 0;
861         struct wlan_bssid_ex *bssid;
862         int group_cipher = 0, pairwise_cipher = 0, is_8021x = 0;
863         unsigned char *pbuf;
864         u32 wpa_ielen = 0;
865         u8 *pbssid = GetAddr3Ptr(pframe);
866         struct HT_info_element *pht_info = NULL;
867         u32 bcn_channel;
868         unsigned short  ht_cap_info;
869         unsigned char   ht_info_infos_0;
870         int ssid_len;
871
872         if (is_client_associated_to_ap(Adapter) == false)
873                 return true;
874
875         len = packet_len - sizeof(struct ieee80211_hdr_3addr);
876
877         if (len > MAX_IE_SZ) {
878                 DBG_88E("%s IE too long for survey event\n", __func__);
879                 return _FAIL;
880         }
881
882         if (!memcmp(cur_network->network.MacAddress, pbssid, 6) == false) {
883                 DBG_88E("Oops: rtw_check_network_encrypt linked but recv other bssid bcn\n%pM %pM\n",
884                         (pbssid), (cur_network->network.MacAddress));
885                 return true;
886         }
887
888         bssid = kzalloc(sizeof(struct wlan_bssid_ex), GFP_ATOMIC);
889         if (!bssid)
890                 return _FAIL;
891
892         subtype = GetFrameSubType(pframe) >> 4;
893
894         if (subtype == WIFI_BEACON)
895                 bssid->Reserved[0] = 1;
896
897         bssid->Length = sizeof(struct wlan_bssid_ex) - MAX_IE_SZ + len;
898
899         /* below is to copy the information element */
900         bssid->IELength = len;
901         memcpy(bssid->IEs, (pframe + sizeof(struct ieee80211_hdr_3addr)), bssid->IELength);
902
903         /* check bw and channel offset */
904         /* parsing HT_CAP_IE */
905         p = rtw_get_ie(bssid->IEs + _FIXED_IE_LENGTH_, _HT_CAPABILITY_IE_, &len, bssid->IELength - _FIXED_IE_LENGTH_);
906         if (p && len > 0) {
907                 struct ieee80211_ht_cap *ht_cap =
908                         (struct ieee80211_ht_cap *)(p + 2);
909
910                 ht_cap_info = le16_to_cpu(ht_cap->cap_info);
911         } else {
912                 ht_cap_info = 0;
913         }
914         /* parsing HT_INFO_IE */
915         p = rtw_get_ie(bssid->IEs + _FIXED_IE_LENGTH_, _HT_ADD_INFO_IE_, &len, bssid->IELength - _FIXED_IE_LENGTH_);
916         if (p && len > 0) {
917                         pht_info = (struct HT_info_element *)(p + 2);
918                         ht_info_infos_0 = pht_info->infos[0];
919         } else {
920                         ht_info_infos_0 = 0;
921         }
922         if (ht_cap_info != cur_network->BcnInfo.ht_cap_info ||
923             ((ht_info_infos_0&0x03) != (cur_network->BcnInfo.ht_info_infos_0&0x03))) {
924                         DBG_88E("%s bcn now: ht_cap_info:%x ht_info_infos_0:%x\n", __func__,
925                                 ht_cap_info, ht_info_infos_0);
926                         DBG_88E("%s bcn link: ht_cap_info:%x ht_info_infos_0:%x\n", __func__,
927                                 cur_network->BcnInfo.ht_cap_info, cur_network->BcnInfo.ht_info_infos_0);
928                         DBG_88E("%s bw mode change, disconnect\n", __func__);
929                         /* bcn_info_update */
930                         cur_network->BcnInfo.ht_cap_info = ht_cap_info;
931                         cur_network->BcnInfo.ht_info_infos_0 = ht_info_infos_0;
932                         /* to do : need to check that whether modify related register of BB or not */
933                         /* goto _mismatch; */
934         }
935
936         /* Checking for channel */
937         p = rtw_get_ie(bssid->IEs + _FIXED_IE_LENGTH_, _DSSET_IE_, &len, bssid->IELength - _FIXED_IE_LENGTH_);
938         if (p) {
939                         bcn_channel = *(p + 2);
940         } else {/* In 5G, some ap do not have DSSET IE checking HT info for channel */
941                         p = rtw_get_ie(bssid->IEs + _FIXED_IE_LENGTH_, _HT_ADD_INFO_IE_, &len, bssid->IELength - _FIXED_IE_LENGTH_);
942                         if (pht_info) {
943                                         bcn_channel = pht_info->primary_channel;
944                         } else { /* we don't find channel IE, so don't check it */
945                                         DBG_88E("Oops: %s we don't find channel IE, so don't check it\n", __func__);
946                                         bcn_channel = Adapter->mlmeextpriv.cur_channel;
947                         }
948         }
949         if (bcn_channel != Adapter->mlmeextpriv.cur_channel) {
950                         DBG_88E("%s beacon channel:%d cur channel:%d disconnect\n", __func__,
951                                 bcn_channel, Adapter->mlmeextpriv.cur_channel);
952                         goto _mismatch;
953         }
954
955         /* checking SSID */
956         ssid_len = 0;
957         p = rtw_get_ie(bssid->IEs + _FIXED_IE_LENGTH_, _SSID_IE_, &len, bssid->IELength - _FIXED_IE_LENGTH_);
958         if (p) {
959                 ssid_len = *(p + 1);
960                 if (ssid_len > NDIS_802_11_LENGTH_SSID)
961                         ssid_len = 0;
962         }
963         memcpy(bssid->Ssid.Ssid, (p + 2), ssid_len);
964         bssid->Ssid.SsidLength = ssid_len;
965
966         RT_TRACE(_module_rtl871x_mlme_c_, _drv_info_, ("%s bssid.Ssid.Ssid:%s bssid.Ssid.SsidLength:%d "
967                                 "cur_network->network.Ssid.Ssid:%s len:%d\n", __func__, bssid->Ssid.Ssid,
968                                 bssid->Ssid.SsidLength, cur_network->network.Ssid.Ssid,
969                                 cur_network->network.Ssid.SsidLength));
970
971         if (memcmp(bssid->Ssid.Ssid, cur_network->network.Ssid.Ssid, 32) ||
972             bssid->Ssid.SsidLength != cur_network->network.Ssid.SsidLength) {
973                 if (bssid->Ssid.Ssid[0] != '\0' && bssid->Ssid.SsidLength != 0) { /* not hidden ssid */
974                         DBG_88E("%s(), SSID is not match return FAIL\n", __func__);
975                         goto _mismatch;
976                 }
977         }
978
979         /* check encryption info */
980         val16 = rtw_get_capability((struct wlan_bssid_ex *)bssid);
981
982         if (val16 & BIT(4))
983                 bssid->Privacy = 1;
984         else
985                 bssid->Privacy = 0;
986
987         RT_TRACE(_module_rtl871x_mlme_c_, _drv_info_,
988                  ("%s(): cur_network->network.Privacy is %d, bssid.Privacy is %d\n",
989                  __func__, cur_network->network.Privacy, bssid->Privacy));
990         if (cur_network->network.Privacy != bssid->Privacy) {
991                 DBG_88E("%s(), privacy is not match return FAIL\n", __func__);
992                 goto _mismatch;
993         }
994
995         rtw_get_sec_ie(bssid->IEs, bssid->IELength, NULL, &rsn_len, NULL, &wpa_len);
996
997         if (rsn_len > 0) {
998                 encryp_protocol = ENCRYP_PROTOCOL_WPA2;
999         } else if (wpa_len > 0) {
1000                 encryp_protocol = ENCRYP_PROTOCOL_WPA;
1001         } else {
1002                 if (bssid->Privacy)
1003                         encryp_protocol = ENCRYP_PROTOCOL_WEP;
1004         }
1005
1006         if (cur_network->BcnInfo.encryp_protocol != encryp_protocol) {
1007                 DBG_88E("%s(): encryption protocol is not match , return FAIL\n", __func__);
1008                 goto _mismatch;
1009         }
1010
1011         if (encryp_protocol == ENCRYP_PROTOCOL_WPA || encryp_protocol == ENCRYP_PROTOCOL_WPA2) {
1012                 pbuf = rtw_get_wpa_ie(&bssid->IEs[12], &wpa_ielen, bssid->IELength-12);
1013                 if (pbuf && (wpa_ielen > 0)) {
1014                         if (_SUCCESS == rtw_parse_wpa_ie(pbuf, wpa_ielen+2, &group_cipher, &pairwise_cipher, &is_8021x)) {
1015                                 RT_TRACE(_module_rtl871x_mlme_c_, _drv_info_,
1016                                          ("%s pnetwork->pairwise_cipher: %d, group_cipher is %d, is_8021x is %d\n", __func__,
1017                                          pairwise_cipher, group_cipher, is_8021x));
1018                         }
1019                 } else {
1020                         pbuf = rtw_get_wpa2_ie(&bssid->IEs[12], &wpa_ielen, bssid->IELength-12);
1021
1022                         if (pbuf && (wpa_ielen > 0)) {
1023                                 if (_SUCCESS == rtw_parse_wpa2_ie(pbuf, wpa_ielen+2, &group_cipher, &pairwise_cipher, &is_8021x)) {
1024                                         RT_TRACE(_module_rtl871x_mlme_c_, _drv_info_,
1025                                                  ("%s pnetwork->pairwise_cipher: %d, pnetwork->group_cipher is %d, is_802x is %d\n",
1026                                                   __func__, pairwise_cipher, group_cipher, is_8021x));
1027                                 }
1028                         }
1029                 }
1030
1031                 RT_TRACE(_module_rtl871x_mlme_c_, _drv_err_,
1032                          ("%s cur_network->group_cipher is %d: %d\n", __func__, cur_network->BcnInfo.group_cipher, group_cipher));
1033                 if (pairwise_cipher != cur_network->BcnInfo.pairwise_cipher || group_cipher != cur_network->BcnInfo.group_cipher) {
1034                         DBG_88E("%s pairwise_cipher(%x:%x) or group_cipher(%x:%x) is not match , return FAIL\n", __func__,
1035                                 pairwise_cipher, cur_network->BcnInfo.pairwise_cipher,
1036                                 group_cipher, cur_network->BcnInfo.group_cipher);
1037                         goto _mismatch;
1038                 }
1039
1040                 if (is_8021x != cur_network->BcnInfo.is_8021x) {
1041                         DBG_88E("%s authentication is not match , return FAIL\n", __func__);
1042                         goto _mismatch;
1043                 }
1044         }
1045
1046         kfree(bssid);
1047         return _SUCCESS;
1048
1049 _mismatch:
1050         kfree(bssid);
1051         return _FAIL;
1052 }
1053
1054 void update_beacon_info(struct adapter *padapter, u8 *pframe, uint pkt_len, struct sta_info *psta)
1055 {
1056         unsigned int i;
1057         unsigned int len;
1058         struct ndis_802_11_var_ie *pIE;
1059
1060         len = pkt_len - (_BEACON_IE_OFFSET_ + WLAN_HDR_A3_LEN);
1061
1062         for (i = 0; i < len;) {
1063                 pIE = (struct ndis_802_11_var_ie *)(pframe + (_BEACON_IE_OFFSET_ + WLAN_HDR_A3_LEN) + i);
1064
1065                 switch (pIE->ElementID) {
1066                 case _HT_EXTRA_INFO_IE_:        /* HT info */
1067                         /* HT_info_handler(padapter, pIE); */
1068                         bwmode_update_check(padapter, pIE);
1069                         break;
1070                 case _ERPINFO_IE_:
1071                         ERP_IE_handler(padapter, pIE);
1072                         VCS_update(padapter, psta);
1073                         break;
1074                 default:
1075                         break;
1076                 }
1077
1078                 i += (pIE->Length + 2);
1079         }
1080 }
1081
1082 unsigned int is_ap_in_tkip(struct adapter *padapter)
1083 {
1084         u32 i;
1085         struct ndis_802_11_var_ie *pIE;
1086         struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1087         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1088         struct wlan_bssid_ex            *cur_network = &(pmlmeinfo->network);
1089
1090         if (rtw_get_capability((struct wlan_bssid_ex *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
1091                 for (i = sizeof(struct ndis_802_11_fixed_ie); i < pmlmeinfo->network.IELength;) {
1092                         pIE = (struct ndis_802_11_var_ie *)(pmlmeinfo->network.IEs + i);
1093
1094                         switch (pIE->ElementID) {
1095                         case _VENDOR_SPECIFIC_IE_:
1096                                 if ((!memcmp(pIE->data, RTW_WPA_OUI, 4)) && (!memcmp((pIE->data + 12), WPA_TKIP_CIPHER, 4)))
1097                                         return true;
1098                                 break;
1099                         case _RSN_IE_2_:
1100                                 if (!memcmp((pIE->data + 8), RSN_TKIP_CIPHER, 4))
1101                                         return true;
1102                         default:
1103                                 break;
1104                         }
1105
1106                         i += (pIE->Length + 2);
1107                 }
1108                 return false;
1109         } else {
1110                 return false;
1111         }
1112 }
1113
1114 unsigned int should_forbid_n_rate(struct adapter *padapter)
1115 {
1116         u32 i;
1117         struct ndis_802_11_var_ie *pIE;
1118         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
1119         struct wlan_bssid_ex  *cur_network = &pmlmepriv->cur_network.network;
1120
1121         if (rtw_get_capability((struct wlan_bssid_ex *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
1122                 for (i = sizeof(struct ndis_802_11_fixed_ie); i < cur_network->IELength;) {
1123                         pIE = (struct ndis_802_11_var_ie *)(cur_network->IEs + i);
1124
1125                         switch (pIE->ElementID) {
1126                         case _VENDOR_SPECIFIC_IE_:
1127                                 if (!memcmp(pIE->data, RTW_WPA_OUI, 4) &&
1128                                     ((!memcmp((pIE->data + 12), WPA_CIPHER_SUITE_CCMP, 4)) ||
1129                                     (!memcmp((pIE->data + 16), WPA_CIPHER_SUITE_CCMP, 4))))
1130                                         return false;
1131                                 break;
1132                         case _RSN_IE_2_:
1133                                 if  ((!memcmp((pIE->data + 8), RSN_CIPHER_SUITE_CCMP, 4))  ||
1134                                        (!memcmp((pIE->data + 12), RSN_CIPHER_SUITE_CCMP, 4)))
1135                                         return false;
1136                         default:
1137                                 break;
1138                         }
1139
1140                         i += (pIE->Length + 2);
1141                 }
1142
1143                 return true;
1144         } else {
1145                 return false;
1146         }
1147 }
1148
1149 unsigned int is_ap_in_wep(struct adapter *padapter)
1150 {
1151         u32 i;
1152         struct ndis_802_11_var_ie *pIE;
1153         struct mlme_ext_priv *pmlmeext = &padapter->mlmeextpriv;
1154         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1155         struct wlan_bssid_ex            *cur_network = &(pmlmeinfo->network);
1156
1157         if (rtw_get_capability((struct wlan_bssid_ex *)cur_network) & WLAN_CAPABILITY_PRIVACY) {
1158                 for (i = sizeof(struct ndis_802_11_fixed_ie); i < pmlmeinfo->network.IELength;) {
1159                         pIE = (struct ndis_802_11_var_ie *)(pmlmeinfo->network.IEs + i);
1160
1161                         switch (pIE->ElementID) {
1162                         case _VENDOR_SPECIFIC_IE_:
1163                                 if (!memcmp(pIE->data, RTW_WPA_OUI, 4))
1164                                         return false;
1165                                 break;
1166                         case _RSN_IE_2_:
1167                                 return false;
1168                         default:
1169                                 break;
1170                         }
1171                         i += (pIE->Length + 2);
1172                 }
1173                 return true;
1174         } else {
1175                 return false;
1176         }
1177 }
1178
1179 static int wifirate2_ratetbl_inx(unsigned char rate)
1180 {
1181         rate = rate & 0x7f;
1182
1183         switch (rate) {
1184         case 54*2:
1185                 return 11;
1186         case 48*2:
1187                 return 10;
1188         case 36*2:
1189                 return 9;
1190         case 24*2:
1191                 return 8;
1192         case 18*2:
1193                 return 7;
1194         case 12*2:
1195                 return 6;
1196         case 9*2:
1197                 return 5;
1198         case 6*2:
1199                 return 4;
1200         case 11*2:
1201                 return 3;
1202         case 11:
1203                 return 2;
1204         case 2*2:
1205                 return 1;
1206         case 1*2:
1207                 return 0;
1208         default:
1209                 return 0;
1210         }
1211 }
1212
1213 unsigned int update_basic_rate(unsigned char *ptn, unsigned int ptn_sz)
1214 {
1215         unsigned int i, num_of_rate;
1216         unsigned int mask = 0;
1217
1218         num_of_rate = min_t(unsigned int, ptn_sz, NumRates);
1219
1220         for (i = 0; i < num_of_rate; i++) {
1221                 if ((*(ptn + i)) & 0x80)
1222                         mask |= 0x1 << wifirate2_ratetbl_inx(*(ptn + i));
1223         }
1224         return mask;
1225 }
1226
1227 unsigned int update_supported_rate(unsigned char *ptn, unsigned int ptn_sz)
1228 {
1229         unsigned int i, num_of_rate;
1230         unsigned int mask = 0;
1231
1232         num_of_rate = min_t(unsigned int, ptn_sz, NumRates);
1233
1234         for (i = 0; i < num_of_rate; i++)
1235                 mask |= 0x1 << wifirate2_ratetbl_inx(*(ptn + i));
1236         return mask;
1237 }
1238
1239 unsigned int update_MSC_rate(struct ieee80211_ht_cap *pHT_caps)
1240 {
1241         unsigned int mask = 0;
1242
1243         mask = (pHT_caps->mcs.rx_mask[0] << 12) |
1244                (pHT_caps->mcs.rx_mask[1] << 20);
1245
1246         return mask;
1247 }
1248
1249 int support_short_GI(struct adapter *padapter, struct ieee80211_ht_cap *pHT_caps)
1250 {
1251         unsigned char                                   bit_offset;
1252         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
1253         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1254
1255         if (!(pmlmeinfo->HT_enable))
1256                 return _FAIL;
1257
1258         if (pmlmeinfo->assoc_AP_vendor == HT_IOT_PEER_RALINK)
1259                 return _FAIL;
1260
1261         bit_offset = (pmlmeext->cur_bwmode & HT_CHANNEL_WIDTH_40) ? 6 : 5;
1262
1263         if (__le16_to_cpu(pHT_caps->cap_info) & (0x1 << bit_offset))
1264                 return _SUCCESS;
1265         else
1266                 return _FAIL;
1267 }
1268
1269 unsigned char get_highest_rate_idx(u32 mask)
1270 {
1271         int i;
1272         unsigned char rate_idx = 0;
1273
1274         for (i = 27; i >= 0; i--) {
1275                 if (mask & BIT(i)) {
1276                         rate_idx = i;
1277                         break;
1278                 }
1279         }
1280         return rate_idx;
1281 }
1282
1283 void Update_RA_Entry(struct adapter *padapter, u32 mac_id)
1284 {
1285         rtw_hal_update_ra_mask(padapter, mac_id, 0);
1286 }
1287
1288 static void enable_rate_adaptive(struct adapter *padapter, u32 mac_id)
1289 {
1290         Update_RA_Entry(padapter, mac_id);
1291 }
1292
1293 void set_sta_rate(struct adapter *padapter, struct sta_info *psta)
1294 {
1295         /* rate adaptive */
1296         enable_rate_adaptive(padapter, psta->mac_id);
1297 }
1298
1299 /*  Update RRSR and Rate for USERATE */
1300 void update_tx_basic_rate(struct adapter *padapter, u8 wirelessmode)
1301 {
1302         unsigned char supported_rates[NDIS_802_11_LENGTH_RATES_EX];
1303
1304         memset(supported_rates, 0, NDIS_802_11_LENGTH_RATES_EX);
1305
1306         if ((wirelessmode & WIRELESS_11B) && (wirelessmode == WIRELESS_11B))
1307                 memcpy(supported_rates, rtw_basic_rate_cck, 4);
1308         else if (wirelessmode & WIRELESS_11B)
1309                 memcpy(supported_rates, rtw_basic_rate_mix, 7);
1310         else
1311                 memcpy(supported_rates, rtw_basic_rate_ofdm, 3);
1312
1313
1314         if (wirelessmode & WIRELESS_11B)
1315                 update_mgnt_tx_rate(padapter, IEEE80211_CCK_RATE_1MB);
1316         else
1317                 update_mgnt_tx_rate(padapter, IEEE80211_OFDM_RATE_6MB);
1318
1319         rtw_hal_set_hwreg(padapter, HW_VAR_BASIC_RATE, supported_rates);
1320 }
1321
1322 unsigned char check_assoc_AP(u8 *pframe, uint len)
1323 {
1324         unsigned int i;
1325         struct ndis_802_11_var_ie *pIE;
1326         u8      epigram_vendor_flag;
1327         u8      ralink_vendor_flag;
1328
1329         epigram_vendor_flag = 0;
1330         ralink_vendor_flag = 0;
1331
1332         for (i = sizeof(struct ndis_802_11_fixed_ie); i < len;) {
1333                 pIE = (struct ndis_802_11_var_ie *)(pframe + i);
1334
1335                 switch (pIE->ElementID) {
1336                 case _VENDOR_SPECIFIC_IE_:
1337                         if ((!memcmp(pIE->data, ARTHEROS_OUI1, 3)) ||
1338                             (!memcmp(pIE->data, ARTHEROS_OUI2, 3))) {
1339                                 DBG_88E("link to Artheros AP\n");
1340                                 return HT_IOT_PEER_ATHEROS;
1341                         } else if ((!memcmp(pIE->data, BROADCOM_OUI1, 3)) ||
1342                                    (!memcmp(pIE->data, BROADCOM_OUI2, 3))) {
1343                                 DBG_88E("link to Broadcom AP\n");
1344                                 return HT_IOT_PEER_BROADCOM;
1345                         } else if (!memcmp(pIE->data, MARVELL_OUI, 3)) {
1346                                 DBG_88E("link to Marvell AP\n");
1347                                 return HT_IOT_PEER_MARVELL;
1348                         } else if (!memcmp(pIE->data, RALINK_OUI, 3)) {
1349                                 if (!ralink_vendor_flag) {
1350                                         ralink_vendor_flag = 1;
1351                                 } else {
1352                                         DBG_88E("link to Ralink AP\n");
1353                                         return HT_IOT_PEER_RALINK;
1354                                 }
1355                         } else if (!memcmp(pIE->data, CISCO_OUI, 3)) {
1356                                 DBG_88E("link to Cisco AP\n");
1357                                 return HT_IOT_PEER_CISCO;
1358                         } else if (!memcmp(pIE->data, REALTEK_OUI, 3)) {
1359                                 DBG_88E("link to Realtek 96B\n");
1360                                 return HT_IOT_PEER_REALTEK;
1361                         } else if (!memcmp(pIE->data, AIRGOCAP_OUI, 3)) {
1362                                 DBG_88E("link to Airgo Cap\n");
1363                                 return HT_IOT_PEER_AIRGO;
1364                         } else if (!memcmp(pIE->data, EPIGRAM_OUI, 3)) {
1365                                 epigram_vendor_flag = 1;
1366                                 if (ralink_vendor_flag) {
1367                                         DBG_88E("link to Tenda W311R AP\n");
1368                                         return HT_IOT_PEER_TENDA;
1369                                 } else {
1370                                         DBG_88E("Capture EPIGRAM_OUI\n");
1371                                 }
1372                         } else {
1373                                 break;
1374                         }
1375
1376                 default:
1377                         break;
1378                 }
1379                 i += (pIE->Length + 2);
1380         }
1381
1382         if (ralink_vendor_flag && !epigram_vendor_flag) {
1383                 DBG_88E("link to Ralink AP\n");
1384                 return HT_IOT_PEER_RALINK;
1385         } else if (ralink_vendor_flag && epigram_vendor_flag) {
1386                 DBG_88E("link to Tenda W311R AP\n");
1387                 return HT_IOT_PEER_TENDA;
1388         } else {
1389                 DBG_88E("link to new AP\n");
1390                 return HT_IOT_PEER_UNKNOWN;
1391         }
1392 }
1393
1394 void update_IOT_info(struct adapter *padapter)
1395 {
1396         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
1397         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1398
1399         switch (pmlmeinfo->assoc_AP_vendor) {
1400         case HT_IOT_PEER_MARVELL:
1401                 pmlmeinfo->turboMode_cts2self = 1;
1402                 pmlmeinfo->turboMode_rtsen = 0;
1403                 break;
1404         case HT_IOT_PEER_RALINK:
1405                 pmlmeinfo->turboMode_cts2self = 0;
1406                 pmlmeinfo->turboMode_rtsen = 1;
1407                 /* disable high power */
1408                 Switch_DM_Func(padapter, (u32)(~DYNAMIC_BB_DYNAMIC_TXPWR),
1409                                false);
1410                 break;
1411         case HT_IOT_PEER_REALTEK:
1412                 /* rtw_write16(padapter, 0x4cc, 0xffff); */
1413                 /* rtw_write16(padapter, 0x546, 0x01c0); */
1414                 /* disable high power */
1415                 Switch_DM_Func(padapter, (u32)(~DYNAMIC_BB_DYNAMIC_TXPWR),
1416                                false);
1417                 break;
1418         default:
1419                 pmlmeinfo->turboMode_cts2self = 0;
1420                 pmlmeinfo->turboMode_rtsen = 1;
1421                 break;
1422         }
1423 }
1424
1425 void update_capinfo(struct adapter *Adapter, u16 updateCap)
1426 {
1427         struct mlme_ext_priv    *pmlmeext = &Adapter->mlmeextpriv;
1428         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1429         bool            ShortPreamble;
1430
1431         /*  Check preamble mode, 2005.01.06, by rcnjko. */
1432         /*  Mark to update preamble value forever, 2008.03.18 by lanhsin */
1433
1434         if (updateCap & cShortPreamble) { /*  Short Preamble */
1435                 if (pmlmeinfo->preamble_mode != PREAMBLE_SHORT) { /*  PREAMBLE_LONG or PREAMBLE_AUTO */
1436                         ShortPreamble = true;
1437                         pmlmeinfo->preamble_mode = PREAMBLE_SHORT;
1438                         rtw_hal_set_hwreg(Adapter, HW_VAR_ACK_PREAMBLE, (u8 *)&ShortPreamble);
1439                 }
1440         } else { /*  Long Preamble */
1441                 if (pmlmeinfo->preamble_mode != PREAMBLE_LONG) {  /*  PREAMBLE_SHORT or PREAMBLE_AUTO */
1442                         ShortPreamble = false;
1443                         pmlmeinfo->preamble_mode = PREAMBLE_LONG;
1444                         rtw_hal_set_hwreg(Adapter, HW_VAR_ACK_PREAMBLE, (u8 *)&ShortPreamble);
1445                 }
1446         }
1447
1448         if (updateCap & cIBSS) {
1449                 /* Filen: See 802.11-2007 p.91 */
1450                 pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
1451         } else { /* Filen: See 802.11-2007 p.90 */
1452                 if (pmlmeext->cur_wireless_mode & (WIRELESS_11G | WIRELESS_11_24N)) {
1453                         if (updateCap & cShortSlotTime) { /*  Short Slot Time */
1454                                 if (pmlmeinfo->slotTime != SHORT_SLOT_TIME)
1455                                         pmlmeinfo->slotTime = SHORT_SLOT_TIME;
1456                         } else { /*  Long Slot Time */
1457                                 if (pmlmeinfo->slotTime != NON_SHORT_SLOT_TIME)
1458                                         pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
1459                         }
1460                 } else if (pmlmeext->cur_wireless_mode & (WIRELESS_11A | WIRELESS_11_5N)) {
1461                         pmlmeinfo->slotTime = SHORT_SLOT_TIME;
1462                 } else {
1463                         /* B Mode */
1464                         pmlmeinfo->slotTime = NON_SHORT_SLOT_TIME;
1465                 }
1466         }
1467
1468         rtw_hal_set_hwreg(Adapter, HW_VAR_SLOT_TIME, &pmlmeinfo->slotTime);
1469 }
1470
1471 void update_wireless_mode(struct adapter *padapter)
1472 {
1473         int ratelen, network_type = 0;
1474         u32 SIFS_Timer;
1475         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
1476         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1477         struct wlan_bssid_ex    *cur_network = &(pmlmeinfo->network);
1478         unsigned char           *rate = cur_network->SupportedRates;
1479
1480         ratelen = rtw_get_rateset_len(cur_network->SupportedRates);
1481
1482         if ((pmlmeinfo->HT_info_enable) && (pmlmeinfo->HT_caps_enable))
1483                 pmlmeinfo->HT_enable = 1;
1484
1485         if (pmlmeext->cur_channel > 14) {
1486                 if (pmlmeinfo->HT_enable)
1487                         network_type = WIRELESS_11_5N;
1488
1489                 network_type |= WIRELESS_11A;
1490         } else {
1491                 if (pmlmeinfo->HT_enable)
1492                         network_type = WIRELESS_11_24N;
1493
1494                 if ((cckratesonly_included(rate, ratelen)) == true)
1495                         network_type |= WIRELESS_11B;
1496                 else if ((cckrates_included(rate, ratelen)) == true)
1497                         network_type |= WIRELESS_11BG;
1498                 else
1499                         network_type |= WIRELESS_11G;
1500         }
1501
1502         pmlmeext->cur_wireless_mode = network_type & padapter->registrypriv.wireless_mode;
1503
1504         SIFS_Timer = 0x0a0a0808;/* 0x0808 -> for CCK, 0x0a0a -> for OFDM */
1505                                 /* change this value if having IOT issues. */
1506
1507         rtw_hal_set_hwreg(padapter, HW_VAR_RESP_SIFS,  (u8 *)&SIFS_Timer);
1508
1509         if (pmlmeext->cur_wireless_mode & WIRELESS_11B)
1510                 update_mgnt_tx_rate(padapter, IEEE80211_CCK_RATE_1MB);
1511          else
1512                 update_mgnt_tx_rate(padapter, IEEE80211_OFDM_RATE_6MB);
1513 }
1514
1515 void update_bmc_sta_support_rate(struct adapter *padapter, u32 mac_id)
1516 {
1517         struct mlme_ext_priv    *pmlmeext = &(padapter->mlmeextpriv);
1518         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1519
1520         if (pmlmeext->cur_wireless_mode & WIRELESS_11B) {
1521                 /*  Only B, B/G, and B/G/N AP could use CCK rate */
1522                 memcpy((pmlmeinfo->FW_sta_info[mac_id].SupportedRates), rtw_basic_rate_cck, 4);
1523         } else {
1524                 memcpy((pmlmeinfo->FW_sta_info[mac_id].SupportedRates), rtw_basic_rate_ofdm, 3);
1525         }
1526 }
1527
1528 int update_sta_support_rate(struct adapter *padapter, u8 *pvar_ie, uint var_ie_len, int cam_idx)
1529 {
1530         unsigned int    ie_len;
1531         struct ndis_802_11_var_ie *pIE;
1532         int     supportRateNum = 0;
1533         struct mlme_ext_priv    *pmlmeext = &(padapter->mlmeextpriv);
1534         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1535
1536         pIE = (struct ndis_802_11_var_ie *)rtw_get_ie(pvar_ie, _SUPPORTEDRATES_IE_, &ie_len, var_ie_len);
1537         if (pIE == NULL)
1538                 return _FAIL;
1539         if (ie_len > NDIS_802_11_LENGTH_RATES_EX)
1540                 return _FAIL;
1541
1542         memcpy(pmlmeinfo->FW_sta_info[cam_idx].SupportedRates, pIE->data, ie_len);
1543         supportRateNum = ie_len;
1544
1545         pIE = (struct ndis_802_11_var_ie *)rtw_get_ie(pvar_ie, _EXT_SUPPORTEDRATES_IE_, &ie_len, var_ie_len);
1546         if (pIE) {
1547                 if (supportRateNum + ie_len > NDIS_802_11_LENGTH_RATES_EX)
1548                         return _FAIL;
1549                 memcpy((pmlmeinfo->FW_sta_info[cam_idx].SupportedRates + supportRateNum), pIE->data, ie_len);
1550         }
1551
1552         return _SUCCESS;
1553 }
1554
1555 void process_addba_req(struct adapter *padapter, u8 *paddba_req, u8 *addr)
1556 {
1557         struct sta_info *psta;
1558         u16 tid;
1559         u16 param;
1560         struct recv_reorder_ctrl *preorder_ctrl;
1561         struct sta_priv *pstapriv = &padapter->stapriv;
1562         struct ADDBA_request    *preq = (struct ADDBA_request *)paddba_req;
1563         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
1564         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
1565
1566         psta = rtw_get_stainfo(pstapriv, addr);
1567
1568         if (psta) {
1569                 param = le16_to_cpu(preq->BA_para_set);
1570                 tid = (param>>2)&0x0f;
1571                 preorder_ctrl = &psta->recvreorder_ctrl[tid];
1572                 preorder_ctrl->indicate_seq = 0xffff;
1573                 preorder_ctrl->enable = (pmlmeinfo->accept_addba_req) ? true
1574                                                                       : false;
1575         }
1576 }
1577
1578 void update_TSF(struct mlme_ext_priv *pmlmeext, u8 *pframe, uint len)
1579 {
1580         u8 *pIE;
1581         __le32 *pbuf;
1582
1583         pIE = pframe + sizeof(struct ieee80211_hdr_3addr);
1584         pbuf = (__le32 *)pIE;
1585
1586         pmlmeext->TSFValue = le32_to_cpu(*(pbuf+1));
1587
1588         pmlmeext->TSFValue = pmlmeext->TSFValue << 32;
1589
1590         pmlmeext->TSFValue |= le32_to_cpu(*pbuf);
1591 }
1592
1593 void correct_TSF(struct adapter *padapter, struct mlme_ext_priv *pmlmeext)
1594 {
1595         rtw_hal_set_hwreg(padapter, HW_VAR_CORRECT_TSF, NULL);
1596 }
1597
1598 void beacon_timing_control(struct adapter *padapter)
1599 {
1600         rtw_hal_bcn_related_reg_setting(padapter);
1601 }