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