drivers/net/wireless/rockchip_wlan/rtl8723ds/core/rtw_mlme.c

   1 /******************************************************************************
   2  *
   3  * Copyright(c) 2007 - 2011 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_MLME_C_
  21
  22 #include <hal_data.h>
  23
  24 extern void indicate_wx_scan_complete_event(_adapter *padapter);
  25 extern u8 rtw_do_join(_adapter *padapter);
  26
  27
  28 sint    _rtw_init_mlme_priv(_adapter *padapter)
  29 {
  30         sint    i;
  31         u8      *pbuf;
  32         struct wlan_network     *pnetwork;
  33         struct mlme_priv                *pmlmepriv = &padapter->mlmepriv;
  34         sint    res = _SUCCESS;
  35
  36
  37         /* We don't need to memset padapter->XXX to zero, because adapter is allocated by rtw_zvmalloc(). */
  38         /* _rtw_memset((u8 *)pmlmepriv, 0, sizeof(struct mlme_priv)); */
  39
  40
  41         /*qos_priv*/
  42         /*pmlmepriv->qospriv.qos_option = pregistrypriv->wmm_enable;*/
  43
  44         /*ht_priv*/
  45 #ifdef CONFIG_80211N_HT
  46         pmlmepriv->htpriv.ampdu_enable = _FALSE;/*set to disabled*/
  47 #endif
  48
  49         pmlmepriv->nic_hdl = (u8 *)padapter;
  50
  51         pmlmepriv->pscanned = NULL;
  52         /*pmlmepriv->fw_state = WIFI_STATION_STATE; */ /*Must sync with rtw_wdev_alloc()*/
  53         /*init_fwstate(pmlmepriv, WIFI_STATION_STATE);*/
  54         init_fwstate(pmlmepriv, WIFI_NULL_STATE);/*assigned interface role(STA/AP) must after execute set_opmode*/
  55
  56         /* wdev->iftype = NL80211_IFTYPE_STATION*/
  57         pmlmepriv->cur_network.network.InfrastructureMode = Ndis802_11AutoUnknown;
  58         pmlmepriv->scan_mode = SCAN_ACTIVE; /* 1: active, 0: pasive. Maybe someday we should rename this varable to "active_mode" (Jeff) */
  59
  60         _rtw_spinlock_init(&(pmlmepriv->lock));
  61         _rtw_init_queue(&(pmlmepriv->free_bss_pool));
  62         _rtw_init_queue(&(pmlmepriv->scanned_queue));
  63
  64         set_scanned_network_val(pmlmepriv, 0);
  65
  66         _rtw_memset(&pmlmepriv->assoc_ssid, 0, sizeof(NDIS_802_11_SSID));
  67
  68         pbuf = rtw_zvmalloc(MAX_BSS_CNT * (sizeof(struct wlan_network)));
  69
  70         if (pbuf == NULL) {
  71                 res = _FAIL;
  72                 goto exit;
  73         }
  74         pmlmepriv->free_bss_buf = pbuf;
  75
  76         pnetwork = (struct wlan_network *)pbuf;
  77
  78         for (i = 0; i < MAX_BSS_CNT; i++) {
  79                 _rtw_init_listhead(&(pnetwork->list));
  80
  81                 rtw_list_insert_tail(&(pnetwork->list), &(pmlmepriv->free_bss_pool.queue));
  82
  83                 pnetwork++;
  84         }
  85
  86         /* allocate DMA-able/Non-Page memory for cmd_buf and rsp_buf */
  87
  88         rtw_clear_scan_deny(padapter);
  89 #ifdef CONFIG_ARP_KEEP_ALIVE
  90         pmlmepriv->bGetGateway = 0;
  91         pmlmepriv->GetGatewayTryCnt = 0;
  92 #endif
  93
  94 #ifdef CONFIG_LAYER2_ROAMING
  95 #define RTW_ROAM_SCAN_RESULT_EXP_MS (5*1000)
  96 #define RTW_ROAM_RSSI_DIFF_TH 10
  97 #define RTW_ROAM_SCAN_INTERVAL_MS (10*1000)
  98 #define RTW_ROAM_RSSI_THRESHOLD 70
  99
 100         pmlmepriv->roam_flags = 0
 101                                 | RTW_ROAM_ON_EXPIRED
 102 #ifdef CONFIG_LAYER2_ROAMING_RESUME
 103                                 | RTW_ROAM_ON_RESUME
 104 #endif
 105 #ifdef CONFIG_LAYER2_ROAMING_ACTIVE
 106                                 | RTW_ROAM_ACTIVE
 107 #endif
 108                                 ;
 109
 110         pmlmepriv->roam_scanr_exp_ms = RTW_ROAM_SCAN_RESULT_EXP_MS;
 111         pmlmepriv->roam_rssi_diff_th = RTW_ROAM_RSSI_DIFF_TH;
 112         pmlmepriv->roam_scan_int_ms = RTW_ROAM_SCAN_INTERVAL_MS;
 113         pmlmepriv->roam_rssi_threshold = RTW_ROAM_RSSI_THRESHOLD;
 114 #endif /* CONFIG_LAYER2_ROAMING */
 115
 116 #ifdef CONFIG_RTW_80211R
 117         memset(&pmlmepriv->ftpriv, 0, sizeof(ft_priv));
 118         pmlmepriv->ftpriv.ft_flags = 0
 119                 | RTW_FT_STA_SUPPORTED
 120                 | RTW_FT_STA_OVER_DS_SUPPORTED
 121                 ;
 122 #endif
 123         rtw_init_mlme_timer(padapter);
 124
 125 exit:
 126
 127
 128         return res;
 129 }
 130
 131 void rtw_mfree_mlme_priv_lock(struct mlme_priv *pmlmepriv);
 132 void rtw_mfree_mlme_priv_lock(struct mlme_priv *pmlmepriv)
 133 {
 134         _rtw_spinlock_free(&pmlmepriv->lock);
 135         _rtw_spinlock_free(&(pmlmepriv->free_bss_pool.lock));
 136         _rtw_spinlock_free(&(pmlmepriv->scanned_queue.lock));
 137 }
 138
 139 static void rtw_free_mlme_ie_data(u8 **ppie, u32 *plen)
 140 {
 141         if (*ppie) {
 142                 rtw_mfree(*ppie, *plen);
 143                 *plen = 0;
 144                 *ppie = NULL;
 145         }
 146 }
 147
 148 void rtw_free_mlme_priv_ie_data(struct mlme_priv *pmlmepriv)
 149 {
 150 #if defined(CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME)
 151         rtw_buf_free(&pmlmepriv->assoc_req, &pmlmepriv->assoc_req_len);
 152         rtw_buf_free(&pmlmepriv->assoc_rsp, &pmlmepriv->assoc_rsp_len);
 153         rtw_free_mlme_ie_data(&pmlmepriv->wps_beacon_ie, &pmlmepriv->wps_beacon_ie_len);
 154         rtw_free_mlme_ie_data(&pmlmepriv->wps_probe_req_ie, &pmlmepriv->wps_probe_req_ie_len);
 155         rtw_free_mlme_ie_data(&pmlmepriv->wps_probe_resp_ie, &pmlmepriv->wps_probe_resp_ie_len);
 156         rtw_free_mlme_ie_data(&pmlmepriv->wps_assoc_resp_ie, &pmlmepriv->wps_assoc_resp_ie_len);
 157
 158         rtw_free_mlme_ie_data(&pmlmepriv->p2p_beacon_ie, &pmlmepriv->p2p_beacon_ie_len);
 159         rtw_free_mlme_ie_data(&pmlmepriv->p2p_probe_req_ie, &pmlmepriv->p2p_probe_req_ie_len);
 160         rtw_free_mlme_ie_data(&pmlmepriv->p2p_probe_resp_ie, &pmlmepriv->p2p_probe_resp_ie_len);
 161         rtw_free_mlme_ie_data(&pmlmepriv->p2p_go_probe_resp_ie, &pmlmepriv->p2p_go_probe_resp_ie_len);
 162         rtw_free_mlme_ie_data(&pmlmepriv->p2p_assoc_req_ie, &pmlmepriv->p2p_assoc_req_ie_len);
 163         rtw_free_mlme_ie_data(&pmlmepriv->p2p_assoc_resp_ie, &pmlmepriv->p2p_assoc_resp_ie_len);
 164 #endif
 165
 166 #if defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211)
 167         rtw_free_mlme_ie_data(&pmlmepriv->wfd_beacon_ie, &pmlmepriv->wfd_beacon_ie_len);
 168         rtw_free_mlme_ie_data(&pmlmepriv->wfd_probe_req_ie, &pmlmepriv->wfd_probe_req_ie_len);
 169         rtw_free_mlme_ie_data(&pmlmepriv->wfd_probe_resp_ie, &pmlmepriv->wfd_probe_resp_ie_len);
 170         rtw_free_mlme_ie_data(&pmlmepriv->wfd_go_probe_resp_ie, &pmlmepriv->wfd_go_probe_resp_ie_len);
 171         rtw_free_mlme_ie_data(&pmlmepriv->wfd_assoc_req_ie, &pmlmepriv->wfd_assoc_req_ie_len);
 172         rtw_free_mlme_ie_data(&pmlmepriv->wfd_assoc_resp_ie, &pmlmepriv->wfd_assoc_resp_ie_len);
 173 #endif
 174
 175 #ifdef CONFIG_RTW_80211R
 176         rtw_free_mlme_ie_data(&pmlmepriv->auth_rsp, &pmlmepriv->auth_rsp_len);
 177 #endif
 178 }
 179
 180 #if defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211)
 181 int rtw_mlme_update_wfd_ie_data(struct mlme_priv *mlme, u8 type, u8 *ie, u32 ie_len)
 182 {
 183         _adapter *adapter = mlme_to_adapter(mlme);
 184         struct wifi_display_info *wfd_info = &adapter->wfd_info;
 185         u8 clear = 0;
 186         u8 **t_ie = NULL;
 187         u32 *t_ie_len = NULL;
 188         int ret = _FAIL;
 189
 190         if (!hal_chk_wl_func(adapter, WL_FUNC_MIRACAST))
 191                 goto success;
 192
 193         if (wfd_info->wfd_enable == _TRUE)
 194                 goto success; /* WFD IE is build by self */
 195
 196         if (!ie && !ie_len)
 197                 clear = 1;
 198         else if (!ie || !ie_len) {
 199                 RTW_PRINT(FUNC_ADPT_FMT" type:%u, ie:%p, ie_len:%u"
 200                           , FUNC_ADPT_ARG(adapter), type, ie, ie_len);
 201                 rtw_warn_on(1);
 202                 goto exit;
 203         }
 204
 205         switch (type) {
 206         case MLME_BEACON_IE:
 207                 t_ie = &mlme->wfd_beacon_ie;
 208                 t_ie_len = &mlme->wfd_beacon_ie_len;
 209                 break;
 210         case MLME_PROBE_REQ_IE:
 211                 t_ie = &mlme->wfd_probe_req_ie;
 212                 t_ie_len = &mlme->wfd_probe_req_ie_len;
 213                 break;
 214         case MLME_PROBE_RESP_IE:
 215                 t_ie = &mlme->wfd_probe_resp_ie;
 216                 t_ie_len = &mlme->wfd_probe_resp_ie_len;
 217                 break;
 218         case MLME_GO_PROBE_RESP_IE:
 219                 t_ie = &mlme->wfd_go_probe_resp_ie;
 220                 t_ie_len = &mlme->wfd_go_probe_resp_ie_len;
 221                 break;
 222         case MLME_ASSOC_REQ_IE:
 223                 t_ie = &mlme->wfd_assoc_req_ie;
 224                 t_ie_len = &mlme->wfd_assoc_req_ie_len;
 225                 break;
 226         case MLME_ASSOC_RESP_IE:
 227                 t_ie = &mlme->wfd_assoc_resp_ie;
 228                 t_ie_len = &mlme->wfd_assoc_resp_ie_len;
 229                 break;
 230         default:
 231                 RTW_PRINT(FUNC_ADPT_FMT" unsupported type:%u"
 232                           , FUNC_ADPT_ARG(adapter), type);
 233                 rtw_warn_on(1);
 234                 goto exit;
 235         }
 236
 237         if (*t_ie) {
 238                 u32 free_len = *t_ie_len;
 239                 *t_ie_len = 0;
 240                 rtw_mfree(*t_ie, free_len);
 241                 *t_ie = NULL;
 242         }
 243
 244         if (!clear) {
 245                 *t_ie = rtw_malloc(ie_len);
 246                 if (*t_ie == NULL) {
 247                         RTW_ERR(FUNC_ADPT_FMT" type:%u, rtw_malloc() fail\n"
 248                                 , FUNC_ADPT_ARG(adapter), type);
 249                         goto exit;
 250                 }
 251                 _rtw_memcpy(*t_ie, ie, ie_len);
 252                 *t_ie_len = ie_len;
 253         }
 254
 255         if (*t_ie && *t_ie_len) {
 256                 u8 *attr_content;
 257                 u32 attr_contentlen = 0;
 258
 259                 attr_content = rtw_get_wfd_attr_content(*t_ie, *t_ie_len, WFD_ATTR_DEVICE_INFO, NULL, &attr_contentlen);
 260                 if (attr_content && attr_contentlen) {
 261                         if (RTW_GET_BE16(attr_content + 2) != wfd_info->rtsp_ctrlport) {
 262                                 wfd_info->rtsp_ctrlport = RTW_GET_BE16(attr_content + 2);
 263                                 RTW_INFO(FUNC_ADPT_FMT" type:%u, RTSP CTRL port = %u\n"
 264                                         , FUNC_ADPT_ARG(adapter), type, wfd_info->rtsp_ctrlport);
 265                         }
 266                 }
 267         }
 268
 269 success:
 270         ret = _SUCCESS;
 271
 272 exit:
 273         return ret;
 274 }
 275 #endif /* defined(CONFIG_WFD) && defined(CONFIG_IOCTL_CFG80211) */
 276
 277 void _rtw_free_mlme_priv(struct mlme_priv *pmlmepriv)
 278 {
 279         if (NULL == pmlmepriv) {
 280                 rtw_warn_on(1);
 281                 goto exit;
 282         }
 283         rtw_free_mlme_priv_ie_data(pmlmepriv);
 284
 285         if (pmlmepriv) {
 286                 rtw_mfree_mlme_priv_lock(pmlmepriv);
 287
 288                 if (pmlmepriv->free_bss_buf)
 289                         rtw_vmfree(pmlmepriv->free_bss_buf, MAX_BSS_CNT * sizeof(struct wlan_network));
 290         }
 291 exit:
 292         return;
 293 }
 294
 295 sint    _rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork)
 296 {
 297         _irqL irqL;
 298
 299
 300         if (pnetwork == NULL)
 301                 goto exit;
 302
 303         _enter_critical_bh(&queue->lock, &irqL);
 304
 305         rtw_list_insert_tail(&pnetwork->list, &queue->queue);
 306
 307         _exit_critical_bh(&queue->lock, &irqL);
 308
 309 exit:
 310
 311
 312         return _SUCCESS;
 313 }
 314
 315 /*
 316 struct  wlan_network *_rtw_dequeue_network(_queue *queue)
 317 {
 318         _irqL irqL;
 319
 320         struct wlan_network *pnetwork;
 321
 322
 323         _enter_critical_bh(&queue->lock, &irqL);
 324
 325         if (_rtw_queue_empty(queue) == _TRUE)
 326
 327                 pnetwork = NULL;
 328
 329         else
 330         {
 331                 pnetwork = LIST_CONTAINOR(get_next(&queue->queue), struct wlan_network, list);
 332
 333                 rtw_list_delete(&(pnetwork->list));
 334         }
 335
 336         _exit_critical_bh(&queue->lock, &irqL);
 337
 338
 339         return pnetwork;
 340 }
 341 */
 342
 343 struct  wlan_network *_rtw_alloc_network(struct mlme_priv *pmlmepriv) /* (_queue *free_queue) */
 344 {
 345         _irqL   irqL;
 346         struct  wlan_network    *pnetwork;
 347         _queue *free_queue = &pmlmepriv->free_bss_pool;
 348         _list *plist = NULL;
 349
 350
 351         _enter_critical_bh(&free_queue->lock, &irqL);
 352
 353         if (_rtw_queue_empty(free_queue) == _TRUE) {
 354                 pnetwork = NULL;
 355                 goto exit;
 356         }
 357         plist = get_next(&(free_queue->queue));
 358
 359         pnetwork = LIST_CONTAINOR(plist , struct wlan_network, list);
 360
 361         rtw_list_delete(&pnetwork->list);
 362
 363         pnetwork->network_type = 0;
 364         pnetwork->fixed = _FALSE;
 365         pnetwork->last_scanned = rtw_get_current_time();
 366         pnetwork->aid = 0;
 367         pnetwork->join_res = 0;
 368
 369         pmlmepriv->num_of_scanned++;
 370
 371 exit:
 372         _exit_critical_bh(&free_queue->lock, &irqL);
 373
 374
 375         return pnetwork;
 376 }
 377
 378 void _rtw_free_network(struct   mlme_priv *pmlmepriv , struct wlan_network *pnetwork, u8 isfreeall)
 379 {
 380         u32 delta_time;
 381         u32 lifetime = SCANQUEUE_LIFETIME;
 382         _irqL irqL;
 383         _queue *free_queue = &(pmlmepriv->free_bss_pool);
 384
 385
 386         if (pnetwork == NULL)
 387                 goto exit;
 388
 389         if (pnetwork->fixed == _TRUE)
 390                 goto exit;
 391
 392         if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) == _TRUE) ||
 393             (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE))
 394                 lifetime = 1;
 395
 396         if (!isfreeall) {
 397                 delta_time = (u32) rtw_get_passing_time_ms(pnetwork->last_scanned);
 398                 if (delta_time < lifetime) /* unit:msec */
 399                         goto exit;
 400         }
 401
 402         _enter_critical_bh(&free_queue->lock, &irqL);
 403
 404         rtw_list_delete(&(pnetwork->list));
 405
 406         rtw_list_insert_tail(&(pnetwork->list), &(free_queue->queue));
 407
 408         pmlmepriv->num_of_scanned--;
 409
 410
 411         /* RTW_INFO("_rtw_free_network:SSID=%s\n", pnetwork->network.Ssid.Ssid); */
 412
 413         _exit_critical_bh(&free_queue->lock, &irqL);
 414
 415 exit:
 416         return;
 417 }
 418
 419 void _rtw_free_network_nolock(struct    mlme_priv *pmlmepriv, struct wlan_network *pnetwork)
 420 {
 421
 422         _queue *free_queue = &(pmlmepriv->free_bss_pool);
 423
 424
 425         if (pnetwork == NULL)
 426                 goto exit;
 427
 428         if (pnetwork->fixed == _TRUE)
 429                 goto exit;
 430
 431         /* _enter_critical(&free_queue->lock, &irqL); */
 432
 433         rtw_list_delete(&(pnetwork->list));
 434
 435         rtw_list_insert_tail(&(pnetwork->list), get_list_head(free_queue));
 436
 437         pmlmepriv->num_of_scanned--;
 438
 439         /* _exit_critical(&free_queue->lock, &irqL); */
 440
 441 exit:
 442         return;
 443 }
 444
 445
 446 /*
 447         return the wlan_network with the matching addr
 448
 449         Shall be calle under atomic context... to avoid possible racing condition...
 450 */
 451 struct wlan_network *_rtw_find_network(_queue *scanned_queue, u8 *addr)
 452 {
 453
 454         /* _irqL irqL; */
 455         _list   *phead, *plist;
 456         struct  wlan_network *pnetwork = NULL;
 457         u8 zero_addr[ETH_ALEN] = {0, 0, 0, 0, 0, 0};
 458
 459
 460         if (_rtw_memcmp(zero_addr, addr, ETH_ALEN)) {
 461                 pnetwork = NULL;
 462                 goto exit;
 463         }
 464
 465         /* _enter_critical_bh(&scanned_queue->lock, &irqL); */
 466
 467         phead = get_list_head(scanned_queue);
 468         plist = get_next(phead);
 469
 470         while (plist != phead) {
 471                 pnetwork = LIST_CONTAINOR(plist, struct wlan_network , list);
 472
 473                 if (_rtw_memcmp(addr, pnetwork->network.MacAddress, ETH_ALEN) == _TRUE)
 474                         break;
 475
 476                 plist = get_next(plist);
 477         }
 478
 479         if (plist == phead)
 480                 pnetwork = NULL;
 481
 482         /* _exit_critical_bh(&scanned_queue->lock, &irqL); */
 483
 484 exit:
 485
 486
 487         return pnetwork;
 488
 489 }
 490
 491
 492 void _rtw_free_network_queue(_adapter *padapter, u8 isfreeall)
 493 {
 494         _irqL irqL;
 495         _list *phead, *plist;
 496         struct wlan_network *pnetwork;
 497         struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
 498         _queue *scanned_queue = &pmlmepriv->scanned_queue;
 499
 500
 501
 502         _enter_critical_bh(&scanned_queue->lock, &irqL);
 503
 504         phead = get_list_head(scanned_queue);
 505         plist = get_next(phead);
 506
 507         while (rtw_end_of_queue_search(phead, plist) == _FALSE) {
 508
 509                 pnetwork = LIST_CONTAINOR(plist, struct wlan_network, list);
 510
 511                 plist = get_next(plist);
 512
 513                 _rtw_free_network(pmlmepriv, pnetwork, isfreeall);
 514
 515         }
 516
 517         _exit_critical_bh(&scanned_queue->lock, &irqL);
 518
 519
 520 }
 521
 522
 523
 524
 525 sint rtw_if_up(_adapter *padapter)
 526 {
 527
 528         sint res;
 529
 530         if (RTW_CANNOT_RUN(padapter) ||
 531             (check_fwstate(&padapter->mlmepriv, _FW_LINKED) == _FALSE)) {
 532                 res = _FALSE;
 533         } else
 534                 res =  _TRUE;
 535
 536         return res;
 537 }
 538
 539
 540 void rtw_generate_random_ibss(u8 *pibss)
 541 {
 542         *((u32 *)(&pibss[2])) = rtw_random32();
 543         pibss[0] = 0x02; /* in ad-hoc mode local bit must set to 1 */
 544         pibss[1] = 0x11;
 545         pibss[2] = 0x87;
 546 }
 547
 548 u8 *rtw_get_capability_from_ie(u8 *ie)
 549 {
 550         return ie + 8 + 2;
 551 }
 552
 553
 554 u16 rtw_get_capability(WLAN_BSSID_EX *bss)
 555 {
 556         u16     val;
 557
 558         _rtw_memcpy((u8 *)&val, rtw_get_capability_from_ie(bss->IEs), 2);
 559
 560         return le16_to_cpu(val);
 561 }
 562
 563 u8 *rtw_get_timestampe_from_ie(u8 *ie)
 564 {
 565         return ie + 0;
 566 }
 567
 568 u8 *rtw_get_beacon_interval_from_ie(u8 *ie)
 569 {
 570         return ie + 8;
 571 }
 572
 573
 574 int     rtw_init_mlme_priv(_adapter *padapter) /* (struct       mlme_priv *pmlmepriv) */
 575 {
 576         int     res;
 577         res = _rtw_init_mlme_priv(padapter);/* (pmlmepriv); */
 578         return res;
 579 }
 580
 581 void rtw_free_mlme_priv(struct mlme_priv *pmlmepriv)
 582 {
 583         _rtw_free_mlme_priv(pmlmepriv);
 584 }
 585
 586 int     rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork);
 587 int     rtw_enqueue_network(_queue *queue, struct wlan_network *pnetwork)
 588 {
 589         int     res;
 590         res = _rtw_enqueue_network(queue, pnetwork);
 591         return res;
 592 }
 593
 594 /*
 595 static struct   wlan_network *rtw_dequeue_network(_queue *queue)
 596 {
 597         struct wlan_network *pnetwork;
 598         pnetwork = _rtw_dequeue_network(queue);
 599         return pnetwork;
 600 }
 601 */
 602
 603 struct  wlan_network *rtw_alloc_network(struct  mlme_priv *pmlmepriv);
 604 struct  wlan_network *rtw_alloc_network(struct  mlme_priv *pmlmepriv) /* (_queue        *free_queue) */
 605 {
 606         struct  wlan_network    *pnetwork;
 607         pnetwork = _rtw_alloc_network(pmlmepriv);
 608         return pnetwork;
 609 }
 610
 611 void rtw_free_network(struct mlme_priv *pmlmepriv, struct       wlan_network *pnetwork, u8 is_freeall);
 612 void rtw_free_network(struct mlme_priv *pmlmepriv, struct       wlan_network *pnetwork, u8 is_freeall)/* (struct        wlan_network *pnetwork, _queue  *free_queue) */
 613 {
 614         _rtw_free_network(pmlmepriv, pnetwork, is_freeall);
 615 }
 616
 617 void rtw_free_network_nolock(_adapter *padapter, struct wlan_network *pnetwork);
 618 void rtw_free_network_nolock(_adapter *padapter, struct wlan_network *pnetwork)
 619 {
 620         _rtw_free_network_nolock(&(padapter->mlmepriv), pnetwork);
 621 #ifdef CONFIG_IOCTL_CFG80211
 622         rtw_cfg80211_unlink_bss(padapter, pnetwork);
 623 #endif /* CONFIG_IOCTL_CFG80211 */
 624 }
 625
 626
 627 void rtw_free_network_queue(_adapter *dev, u8 isfreeall)
 628 {
 629         _rtw_free_network_queue(dev, isfreeall);
 630 }
 631
 632 /*
 633         return the wlan_network with the matching addr
 634
 635         Shall be calle under atomic context... to avoid possible racing condition...
 636 */
 637 struct  wlan_network *rtw_find_network(_queue *scanned_queue, u8 *addr)
 638 {
 639         struct  wlan_network *pnetwork = _rtw_find_network(scanned_queue, addr);
 640
 641         return pnetwork;
 642 }
 643
 644 int rtw_is_same_ibss(_adapter *adapter, struct wlan_network *pnetwork)
 645 {
 646         int ret = _TRUE;
 647         struct security_priv *psecuritypriv = &adapter->securitypriv;
 648
 649         if ((psecuritypriv->dot11PrivacyAlgrthm != _NO_PRIVACY_) &&
 650             (pnetwork->network.Privacy == 0))
 651                 ret = _FALSE;
 652         else if ((psecuritypriv->dot11PrivacyAlgrthm == _NO_PRIVACY_) &&
 653                  (pnetwork->network.Privacy == 1))
 654                 ret = _FALSE;
 655         else
 656                 ret = _TRUE;
 657
 658         return ret;
 659
 660 }
 661
 662 inline int is_same_ess(WLAN_BSSID_EX *a, WLAN_BSSID_EX *b)
 663 {
 664         return (a->Ssid.SsidLength == b->Ssid.SsidLength)
 665                &&  _rtw_memcmp(a->Ssid.Ssid, b->Ssid.Ssid, a->Ssid.SsidLength) == _TRUE;
 666 }
 667
 668 int is_same_network(WLAN_BSSID_EX *src, WLAN_BSSID_EX *dst, u8 feature)
 669 {
 670         u16 s_cap, d_cap;
 671
 672
 673         if (rtw_bug_check(dst, src, &s_cap, &d_cap) == _FALSE)
 674                 return _FALSE;
 675
 676         _rtw_memcpy((u8 *)&s_cap, rtw_get_capability_from_ie(src->IEs), 2);
 677         _rtw_memcpy((u8 *)&d_cap, rtw_get_capability_from_ie(dst->IEs), 2);
 678
 679
 680         s_cap = le16_to_cpu(s_cap);
 681         d_cap = le16_to_cpu(d_cap);
 682
 683
 684 #ifdef CONFIG_P2P
 685         if ((feature == 1) && /* 1: P2P supported */
 686             (_rtw_memcmp(src->MacAddress, dst->MacAddress, ETH_ALEN) == _TRUE)
 687            )
 688                 return _TRUE;
 689 #endif
 690
 691         return ((src->Ssid.SsidLength == dst->Ssid.SsidLength) &&
 692                 /*      (src->Configuration.DSConfig == dst->Configuration.DSConfig) && */
 693                 ((_rtw_memcmp(src->MacAddress, dst->MacAddress, ETH_ALEN)) == _TRUE) &&
 694                 ((_rtw_memcmp(src->Ssid.Ssid, dst->Ssid.Ssid, src->Ssid.SsidLength)) == _TRUE) &&
 695                 ((s_cap & WLAN_CAPABILITY_IBSS) ==
 696                  (d_cap & WLAN_CAPABILITY_IBSS)) &&
 697                 ((s_cap & WLAN_CAPABILITY_BSS) ==
 698                  (d_cap & WLAN_CAPABILITY_BSS)));
 699
 700 }
 701
 702 struct wlan_network *_rtw_find_same_network(_queue *scanned_queue, struct wlan_network *network)
 703 {
 704         _list *phead, *plist;
 705         struct wlan_network *found = NULL;
 706
 707         phead = get_list_head(scanned_queue);
 708         plist = get_next(phead);
 709
 710         while (plist != phead) {
 711                 found = LIST_CONTAINOR(plist, struct wlan_network , list);
 712
 713                 if (is_same_network(&network->network, &found->network, 0))
 714                         break;
 715
 716                 plist = get_next(plist);
 717         }
 718
 719         if (plist == phead)
 720                 found = NULL;
 721 exit:
 722         return found;
 723 }
 724
 725 struct wlan_network *rtw_find_same_network(_queue *scanned_queue, struct wlan_network *network)
 726 {
 727         _irqL irqL;
 728         struct wlan_network *found = NULL;
 729
 730         if (scanned_queue == NULL || network == NULL)
 731                 goto exit;
 732
 733         _enter_critical_bh(&scanned_queue->lock, &irqL);
 734         found = _rtw_find_same_network(scanned_queue, network);
 735         _exit_critical_bh(&scanned_queue->lock, &irqL);
 736
 737 exit:
 738         return found;
 739 }
 740
 741 struct  wlan_network    *rtw_get_oldest_wlan_network(_queue *scanned_queue)
 742 {
 743         _list   *plist, *phead;
 744
 745
 746         struct  wlan_network    *pwlan = NULL;
 747         struct  wlan_network    *oldest = NULL;
 748         phead = get_list_head(scanned_queue);
 749
 750         plist = get_next(phead);
 751
 752         while (1) {
 753
 754                 if (rtw_end_of_queue_search(phead, plist) == _TRUE)
 755                         break;
 756
 757                 pwlan = LIST_CONTAINOR(plist, struct wlan_network, list);
 758
 759                 if (pwlan->fixed != _TRUE) {
 760                         if (oldest == NULL || time_after(oldest->last_scanned, pwlan->last_scanned))
 761                                 oldest = pwlan;
 762                 }
 763
 764                 plist = get_next(plist);
 765         }
 766         return oldest;
 767
 768 }
 769
 770 void update_network(WLAN_BSSID_EX *dst, WLAN_BSSID_EX *src,
 771                     _adapter *padapter, bool update_ie)
 772 {
 773         u8 ss_ori = dst->PhyInfo.SignalStrength;
 774         u8 sq_ori = dst->PhyInfo.SignalQuality;
 775         long rssi_ori = dst->Rssi;
 776
 777         u8 ss_smp = src->PhyInfo.SignalStrength;
 778         u8 sq_smp = src->PhyInfo.SignalQuality;
 779         long rssi_smp = src->Rssi;
 780
 781         u8 ss_final;
 782         u8 sq_final;
 783         long rssi_final;
 784
 785
 786 #ifdef CONFIG_ANTENNA_DIVERSITY
 787         rtw_hal_antdiv_rssi_compared(padapter, dst, src); /* this will update src.Rssi, need consider again */
 788 #endif
 789
 790 #if defined(DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) && 1
 791         if (strcmp(dst->Ssid.Ssid, DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) == 0) {
 792                 RTW_INFO(FUNC_ADPT_FMT" %s("MAC_FMT", ch%u) ss_ori:%3u, sq_ori:%3u, rssi_ori:%3ld, ss_smp:%3u, sq_smp:%3u, rssi_smp:%3ld\n"
 793                          , FUNC_ADPT_ARG(padapter)
 794                         , src->Ssid.Ssid, MAC_ARG(src->MacAddress), src->Configuration.DSConfig
 795                          , ss_ori, sq_ori, rssi_ori
 796                          , ss_smp, sq_smp, rssi_smp
 797                         );
 798         }
 799 #endif
 800
 801         /* The rule below is 1/5 for sample value, 4/5 for history value */
 802         if (check_fwstate(&padapter->mlmepriv, _FW_LINKED) && is_same_network(&(padapter->mlmepriv.cur_network.network), src, 0)) {
 803                 /* Take the recvpriv's value for the connected AP*/
 804                 ss_final = padapter->recvpriv.signal_strength;
 805                 sq_final = padapter->recvpriv.signal_qual;
 806                 /* the rssi value here is undecorated, and will be used for antenna diversity */
 807                 if (sq_smp != 101) /* from the right channel */
 808                         rssi_final = (src->Rssi + dst->Rssi * 4) / 5;
 809                 else
 810                         rssi_final = rssi_ori;
 811         } else {
 812                 if (sq_smp != 101) { /* from the right channel */
 813                         ss_final = ((u32)(src->PhyInfo.SignalStrength) + (u32)(dst->PhyInfo.SignalStrength) * 4) / 5;
 814                         sq_final = ((u32)(src->PhyInfo.SignalQuality) + (u32)(dst->PhyInfo.SignalQuality) * 4) / 5;
 815                         rssi_final = (src->Rssi + dst->Rssi * 4) / 5;
 816                 } else {
 817                         /* bss info not receving from the right channel, use the original RX signal infos */
 818                         ss_final = dst->PhyInfo.SignalStrength;
 819                         sq_final = dst->PhyInfo.SignalQuality;
 820                         rssi_final = dst->Rssi;
 821                 }
 822
 823         }
 824
 825         if (update_ie) {
 826                 dst->Reserved[0] = src->Reserved[0];
 827                 dst->Reserved[1] = src->Reserved[1];
 828                 _rtw_memcpy((u8 *)dst, (u8 *)src, get_WLAN_BSSID_EX_sz(src));
 829         }
 830
 831         dst->PhyInfo.SignalStrength = ss_final;
 832         dst->PhyInfo.SignalQuality = sq_final;
 833         dst->Rssi = rssi_final;
 834
 835 #if defined(DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) && 1
 836         if (strcmp(dst->Ssid.Ssid, DBG_RX_SIGNAL_DISPLAY_SSID_MONITORED) == 0) {
 837                 RTW_INFO(FUNC_ADPT_FMT" %s("MAC_FMT"), SignalStrength:%u, SignalQuality:%u, RawRSSI:%ld\n"
 838                          , FUNC_ADPT_ARG(padapter)
 839                         , dst->Ssid.Ssid, MAC_ARG(dst->MacAddress), dst->PhyInfo.SignalStrength, dst->PhyInfo.SignalQuality, dst->Rssi);
 840         }
 841 #endif
 842
 843 #if 0 /* old codes, may be useful one day...
 844  *      RTW_INFO("update_network: rssi=0x%lx dst->Rssi=%d ,dst->Rssi=0x%lx , src->Rssi=0x%lx",(dst->Rssi+src->Rssi)/2,dst->Rssi,dst->Rssi,src->Rssi); */
 845         if (check_fwstate(&padapter->mlmepriv, _FW_LINKED) && is_same_network(&(padapter->mlmepriv.cur_network.network), src)) {
 846
 847                 /* RTW_INFO("b:ssid=%s update_network: src->rssi=0x%d padapter->recvpriv.ui_rssi=%d\n",src->Ssid.Ssid,src->Rssi,padapter->recvpriv.signal); */
 848                 if (padapter->recvpriv.signal_qual_data.total_num++ >= PHY_LINKQUALITY_SLID_WIN_MAX) {
 849                         padapter->recvpriv.signal_qual_data.total_num = PHY_LINKQUALITY_SLID_WIN_MAX;
 850                         last_evm = padapter->recvpriv.signal_qual_data.elements[padapter->recvpriv.signal_qual_data.index];
 851                         padapter->recvpriv.signal_qual_data.total_val -= last_evm;
 852                 }
 853                 padapter->recvpriv.signal_qual_data.total_val += query_rx_pwr_percentage(src->Rssi);
 854
 855                 padapter->recvpriv.signal_qual_data.elements[padapter->recvpriv.signal_qual_data.index++] = query_rx_pwr_percentage(src->Rssi);
 856                 if (padapter->recvpriv.signal_qual_data.index >= PHY_LINKQUALITY_SLID_WIN_MAX)
 857                         padapter->recvpriv.signal_qual_data.index = 0;
 858
 859                 /* RTW_INFO("Total SQ=%d  pattrib->signal_qual= %d\n", padapter->recvpriv.signal_qual_data.total_val, src->Rssi); */
 860
 861                 /* <1> Showed on UI for user,in percentage. */
 862                 tmpVal = padapter->recvpriv.signal_qual_data.total_val / padapter->recvpriv.signal_qual_data.total_num;
 863                 padapter->recvpriv.signal = (u8)tmpVal; /* Link quality */
 864
 865                 src->Rssi = translate_percentage_to_dbm(padapter->recvpriv.signal) ;
 866         } else {
 867                 /*      RTW_INFO("ELSE:ssid=%s update_network: src->rssi=0x%d dst->rssi=%d\n",src->Ssid.Ssid,src->Rssi,dst->Rssi); */
 868                 src->Rssi = (src->Rssi + dst->Rssi) / 2; /* dBM */
 869         }
 870
 871         /*      RTW_INFO("a:update_network: src->rssi=0x%d padapter->recvpriv.ui_rssi=%d\n",src->Rssi,padapter->recvpriv.signal); */
 872
 873 #endif
 874
 875 }
 876
 877 static void update_current_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork)
 878 {
 879         struct  mlme_priv       *pmlmepriv = &(adapter->mlmepriv);
 880
 881
 882         rtw_bug_check(&(pmlmepriv->cur_network.network),
 883                       &(pmlmepriv->cur_network.network),
 884                       &(pmlmepriv->cur_network.network),
 885                       &(pmlmepriv->cur_network.network));
 886
 887         if ((check_fwstate(pmlmepriv, _FW_LINKED) == _TRUE) && (is_same_network(&(pmlmepriv->cur_network.network), pnetwork, 0))) {
 888
 889                 /* if(pmlmepriv->cur_network.network.IELength<= pnetwork->IELength) */
 890                 {
 891                         update_network(&(pmlmepriv->cur_network.network), pnetwork, adapter, _TRUE);
 892                         rtw_update_protection(adapter, (pmlmepriv->cur_network.network.IEs) + sizeof(NDIS_802_11_FIXED_IEs),
 893                                       pmlmepriv->cur_network.network.IELength);
 894                 }
 895         }
 896
 897
 898 }
 899
 900
 901 /*
 902
 903 Caller must hold pmlmepriv->lock first.
 904
 905
 906 */
 907 void rtw_update_scanned_network(_adapter *adapter, WLAN_BSSID_EX *target)
 908 {
 909         _irqL irqL;
 910         _list   *plist, *phead;
 911         ULONG   bssid_ex_sz;
 912         struct mlme_priv        *pmlmepriv = &(adapter->mlmepriv);
 913         struct mlme_ext_priv    *pmlmeext = &(adapter->mlmeextpriv);
 914 #ifdef CONFIG_P2P
 915         struct wifidirect_info *pwdinfo = &(adapter->wdinfo);
 916 #endif /* CONFIG_P2P */
 917         _queue  *queue  = &(pmlmepriv->scanned_queue);
 918         struct wlan_network     *pnetwork = NULL;
 919         struct wlan_network     *oldest = NULL;
 920         int target_find = 0;
 921         u8 feature = 0;
 922
 923
 924         _enter_critical_bh(&queue->lock, &irqL);
 925         phead = get_list_head(queue);
 926         plist = get_next(phead);
 927
 928 #ifdef CONFIG_P2P
 929         if (!rtw_p2p_chk_state(pwdinfo, P2P_STATE_NONE))
 930                 feature = 1; /* p2p enable */
 931 #endif
 932
 933         while (1) {
 934                 if (rtw_end_of_queue_search(phead, plist) == _TRUE)
 935                         break;
 936
 937                 pnetwork = LIST_CONTAINOR(plist, struct wlan_network, list);
 938
 939                 rtw_bug_check(pnetwork, pnetwork, pnetwork, pnetwork);
 940
 941 #ifdef CONFIG_P2P
 942                 if (!rtw_p2p_chk_state(pwdinfo, P2P_STATE_NONE) &&
 943                     (_rtw_memcmp(pnetwork->network.MacAddress, target->MacAddress, ETH_ALEN) == _TRUE)) {
 944                         target_find = 1;
 945                         break;
 946                 }
 947 #endif
 948
 949                 if (is_same_network(&(pnetwork->network), target, feature)) {
 950                         target_find = 1;
 951                         break;
 952                 }
 953
 954                 if (rtw_roam_flags(adapter)) {
 955                         /* TODO: don't  select netowrk in the same ess as oldest if it's new enough*/
 956                 }
 957
 958                 if (oldest == NULL || time_after(oldest->last_scanned, pnetwork->last_scanned))
 959                         oldest = pnetwork;
 960
 961                 plist = get_next(plist);
 962
 963         }
 964
 965
 966         /* If we didn't find a match, then get a new network slot to initialize
 967          * with this beacon's information */
 968         /* if (rtw_end_of_queue_search(phead,plist)== _TRUE) { */
 969         if (!target_find) {
 970                 if (_rtw_queue_empty(&(pmlmepriv->free_bss_pool)) == _TRUE) {
 971                         /* If there are no more slots, expire the oldest */
 972                         /* list_del_init(&oldest->list); */
 973                         pnetwork = oldest;
 974                         if (pnetwork == NULL) {
 975                                 goto exit;
 976                         }
 977 #ifdef CONFIG_ANTENNA_DIVERSITY
 978                         rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(target->PhyInfo.Optimum_antenna), NULL);
 979 #endif
 980                         _rtw_memcpy(&(pnetwork->network), target,  get_WLAN_BSSID_EX_sz(target));
 981                         /* pnetwork->last_scanned = rtw_get_current_time(); */
 982                         /* variable initialize */
 983                         pnetwork->fixed = _FALSE;
 984                         pnetwork->last_scanned = rtw_get_current_time();
 985
 986                         pnetwork->network_type = 0;
 987                         pnetwork->aid = 0;
 988                         pnetwork->join_res = 0;
 989
 990                         /* bss info not receving from the right channel */
 991                         if (pnetwork->network.PhyInfo.SignalQuality == 101)
 992                                 pnetwork->network.PhyInfo.SignalQuality = 0;
 993                 } else {
 994                         /* Otherwise just pull from the free list */
 995
 996                         pnetwork = rtw_alloc_network(pmlmepriv); /* will update scan_time */
 997
 998                         if (pnetwork == NULL) {
 999                                 goto exit;
1000                         }
1001
1002                         bssid_ex_sz = get_WLAN_BSSID_EX_sz(target);
1003                         target->Length = bssid_ex_sz;
1004 #ifdef CONFIG_ANTENNA_DIVERSITY
1005                         rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(target->PhyInfo.Optimum_antenna), NULL);
1006 #endif
1007                         _rtw_memcpy(&(pnetwork->network), target, bssid_ex_sz);
1008
1009                         pnetwork->last_scanned = rtw_get_current_time();
1010
1011                         /* bss info not receving from the right channel */
1012                         if (pnetwork->network.PhyInfo.SignalQuality == 101)
1013                                 pnetwork->network.PhyInfo.SignalQuality = 0;
1014
1015                         rtw_list_insert_tail(&(pnetwork->list), &(queue->queue));
1016
1017                 }
1018         } else {
1019                 /* we have an entry and we are going to update it. But this entry may
1020                  * be already expired. In this case we do the same as we found a new
1021                  * net and call the new_net handler
1022                  */
1023                 bool update_ie = _TRUE;
1024
1025                 pnetwork->last_scanned = rtw_get_current_time();
1026
1027                 /* target.Reserved[0]==1, means that scaned network is a bcn frame. */
1028                 if ((pnetwork->network.IELength > target->IELength) && (target->Reserved[0] == 1))
1029                         update_ie = _FALSE;
1030
1031                 /* probe resp(3) > beacon(1) > probe req(2) */
1032                 if ((target->Reserved[0] != 2) &&
1033                     (target->Reserved[0] >= pnetwork->network.Reserved[0])
1034                    )
1035                         update_ie = _TRUE;
1036                 else
1037                         update_ie = _FALSE;
1038
1039                 update_network(&(pnetwork->network), target, adapter, update_ie);
1040         }
1041
1042 exit:
1043         _exit_critical_bh(&queue->lock, &irqL);
1044
1045 }
1046
1047 void rtw_add_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork);
1048 void rtw_add_network(_adapter *adapter, WLAN_BSSID_EX *pnetwork)
1049 {
1050         _irqL irqL;
1051         struct  mlme_priv       *pmlmepriv = &(((_adapter *)adapter)->mlmepriv);
1052         /* _queue       *queue  = &(pmlmepriv->scanned_queue); */
1053
1054
1055         /* _enter_critical_bh(&queue->lock, &irqL); */
1056
1057 #if defined(CONFIG_P2P) && defined(CONFIG_P2P_REMOVE_GROUP_INFO)
1058         if (adapter->registrypriv.wifi_spec == 0)
1059                 rtw_bss_ex_del_p2p_attr(pnetwork, P2P_ATTR_GROUP_INFO);
1060 #endif
1061
1062         if (!hal_chk_wl_func(adapter, WL_FUNC_MIRACAST))
1063                 rtw_bss_ex_del_wfd_ie(pnetwork);
1064
1065         update_current_network(adapter, pnetwork);
1066
1067         rtw_update_scanned_network(adapter, pnetwork);
1068
1069         /* _exit_critical_bh(&queue->lock, &irqL); */
1070
1071 }
1072
1073 /* select the desired network based on the capability of the (i)bss.
1074  * check items: (1) security
1075  *                         (2) network_type
1076  *                         (3) WMM
1077  *                         (4) HT
1078  * (5) others */
1079 int rtw_is_desired_network(_adapter *adapter, struct wlan_network *pnetwork);
1080 int rtw_is_desired_network(_adapter *adapter, struct wlan_network *pnetwork)
1081 {
1082         struct security_priv *psecuritypriv = &adapter->securitypriv;
1083         struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
1084         u32 desired_encmode;
1085         u32 privacy;
1086
1087         /* u8 wps_ie[512]; */
1088         uint wps_ielen;
1089
1090         int bselected = _TRUE;
1091
1092         desired_encmode = psecuritypriv->ndisencryptstatus;
1093         privacy = pnetwork->network.Privacy;
1094
1095         if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
1096                 if (rtw_get_wps_ie(pnetwork->network.IEs + _FIXED_IE_LENGTH_, pnetwork->network.IELength - _FIXED_IE_LENGTH_, NULL, &wps_ielen) != NULL)
1097                         return _TRUE;
1098                 else
1099                         return _FALSE;
1100         }
1101         if (adapter->registrypriv.wifi_spec == 1) { /* for  correct flow of 8021X  to do.... */
1102                 u8 *p = NULL;
1103                 uint ie_len = 0;
1104
1105                 if ((desired_encmode == Ndis802_11EncryptionDisabled) && (privacy != 0))
1106                         bselected = _FALSE;
1107
1108                 if (psecuritypriv->ndisauthtype == Ndis802_11AuthModeWPA2PSK) {
1109                         p = rtw_get_ie(pnetwork->network.IEs + _BEACON_IE_OFFSET_, _RSN_IE_2_, &ie_len, (pnetwork->network.IELength - _BEACON_IE_OFFSET_));
1110                         if (p && ie_len > 0)
1111                                 bselected = _TRUE;
1112                         else
1113                                 bselected = _FALSE;
1114                 }
1115         }
1116
1117
1118         if ((desired_encmode != Ndis802_11EncryptionDisabled) && (privacy == 0)) {
1119                 RTW_INFO("desired_encmode: %d, privacy: %d\n", desired_encmode, privacy);
1120                 bselected = _FALSE;
1121         }
1122
1123         if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE) {
1124                 if (pnetwork->network.InfrastructureMode != pmlmepriv->cur_network.network.InfrastructureMode)
1125                         bselected = _FALSE;
1126         }
1127
1128
1129         return bselected;
1130 }
1131
1132 /* TODO: Perry : For Power Management */
1133 void rtw_atimdone_event_callback(_adapter       *adapter , u8 *pbuf)
1134 {
1135
1136         return;
1137 }
1138
1139
1140 void rtw_survey_event_callback(_adapter *adapter, u8 *pbuf)
1141 {
1142         _irqL  irqL;
1143         u32 len;
1144         WLAN_BSSID_EX *pnetwork;
1145         struct  mlme_priv       *pmlmepriv = &(adapter->mlmepriv);
1146
1147
1148         pnetwork = (WLAN_BSSID_EX *)pbuf;
1149
1150
1151 #ifdef CONFIG_RTL8712
1152         /* endian_convert */
1153         pnetwork->Length = le32_to_cpu(pnetwork->Length);
1154         pnetwork->Ssid.SsidLength = le32_to_cpu(pnetwork->Ssid.SsidLength);
1155         pnetwork->Privacy = le32_to_cpu(pnetwork->Privacy);
1156         pnetwork->Rssi = le32_to_cpu(pnetwork->Rssi);
1157         pnetwork->NetworkTypeInUse = le32_to_cpu(pnetwork->NetworkTypeInUse);
1158         pnetwork->Configuration.ATIMWindow = le32_to_cpu(pnetwork->Configuration.ATIMWindow);
1159         pnetwork->Configuration.BeaconPeriod = le32_to_cpu(pnetwork->Configuration.BeaconPeriod);
1160         pnetwork->Configuration.DSConfig = le32_to_cpu(pnetwork->Configuration.DSConfig);
1161         pnetwork->Configuration.FHConfig.DwellTime = le32_to_cpu(pnetwork->Configuration.FHConfig.DwellTime);
1162         pnetwork->Configuration.FHConfig.HopPattern = le32_to_cpu(pnetwork->Configuration.FHConfig.HopPattern);
1163         pnetwork->Configuration.FHConfig.HopSet = le32_to_cpu(pnetwork->Configuration.FHConfig.HopSet);
1164         pnetwork->Configuration.FHConfig.Length = le32_to_cpu(pnetwork->Configuration.FHConfig.Length);
1165         pnetwork->Configuration.Length = le32_to_cpu(pnetwork->Configuration.Length);
1166         pnetwork->InfrastructureMode = le32_to_cpu(pnetwork->InfrastructureMode);
1167         pnetwork->IELength = le32_to_cpu(pnetwork->IELength);
1168 #endif
1169
1170         len = get_WLAN_BSSID_EX_sz(pnetwork);
1171         if (len > (sizeof(WLAN_BSSID_EX))) {
1172                 return;
1173         }
1174
1175
1176         _enter_critical_bh(&pmlmepriv->lock, &irqL);
1177
1178         /* update IBSS_network 's timestamp */
1179         if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE)) == _TRUE) {
1180                 if (_rtw_memcmp(&(pmlmepriv->cur_network.network.MacAddress), pnetwork->MacAddress, ETH_ALEN)) {
1181                         struct wlan_network *ibss_wlan = NULL;
1182                         _irqL   irqL;
1183
1184                         _rtw_memcpy(pmlmepriv->cur_network.network.IEs, pnetwork->IEs, 8);
1185                         _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1186                         ibss_wlan = rtw_find_network(&pmlmepriv->scanned_queue,  pnetwork->MacAddress);
1187                         if (ibss_wlan) {
1188                                 _rtw_memcpy(ibss_wlan->network.IEs , pnetwork->IEs, 8);
1189                                 _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1190                                 goto exit;
1191                         }
1192                         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1193                 }
1194         }
1195
1196         /* lock pmlmepriv->lock when you accessing network_q */
1197         if ((check_fwstate(pmlmepriv, _FW_UNDER_LINKING)) == _FALSE) {
1198                 if (pnetwork->Ssid.Ssid[0] == 0)
1199                         pnetwork->Ssid.SsidLength = 0;
1200                 rtw_add_network(adapter, pnetwork);
1201         }
1202
1203 exit:
1204
1205         _exit_critical_bh(&pmlmepriv->lock, &irqL);
1206
1207
1208         return;
1209 }
1210
1211 void rtw_surveydone_event_callback(_adapter     *adapter, u8 *pbuf)
1212 {
1213         _irqL  irqL;
1214         u8 timer_cancelled;
1215         struct  mlme_priv       *pmlmepriv = &(adapter->mlmepriv);
1216 #ifdef CONFIG_RTW_80211R
1217         struct mlme_ext_priv    *pmlmeext = &adapter->mlmeextpriv;
1218 #endif
1219
1220 #ifdef CONFIG_MLME_EXT
1221         mlmeext_surveydone_event_callback(adapter);
1222 #endif
1223
1224
1225         _enter_critical_bh(&pmlmepriv->lock, &irqL);
1226         if (pmlmepriv->wps_probe_req_ie) {
1227                 u32 free_len = pmlmepriv->wps_probe_req_ie_len;
1228                 pmlmepriv->wps_probe_req_ie_len = 0;
1229                 rtw_mfree(pmlmepriv->wps_probe_req_ie, free_len);
1230                 pmlmepriv->wps_probe_req_ie = NULL;
1231         }
1232
1233
1234         if (check_fwstate(pmlmepriv, _FW_UNDER_SURVEY) == _FALSE) {
1235                 RTW_INFO(FUNC_ADPT_FMT" fw_state:0x%x\n", FUNC_ADPT_ARG(adapter), get_fwstate(pmlmepriv));
1236                 /* rtw_warn_on(1); */
1237         }
1238
1239         _clr_fwstate_(pmlmepriv, _FW_UNDER_SURVEY);
1240         _exit_critical_bh(&pmlmepriv->lock, &irqL);
1241
1242         _cancel_timer(&pmlmepriv->scan_to_timer, &timer_cancelled);
1243
1244         _enter_critical_bh(&pmlmepriv->lock, &irqL);
1245
1246 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
1247         rtw_set_signal_stat_timer(&adapter->recvpriv);
1248 #endif
1249
1250         if (pmlmepriv->to_join == _TRUE) {
1251                 if ((check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE)) {
1252                         if (check_fwstate(pmlmepriv, _FW_LINKED) == _FALSE) {
1253                                 set_fwstate(pmlmepriv, _FW_UNDER_LINKING);
1254
1255                                 if (rtw_select_and_join_from_scanned_queue(pmlmepriv) == _SUCCESS)
1256                                         _set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
1257                                 else {
1258                                         WLAN_BSSID_EX    *pdev_network = &(adapter->registrypriv.dev_network);
1259                                         u8 *pibss = adapter->registrypriv.dev_network.MacAddress;
1260
1261                                         /* pmlmepriv->fw_state ^= _FW_UNDER_SURVEY; */ /* because don't set assoc_timer */
1262                                         _clr_fwstate_(pmlmepriv, _FW_UNDER_SURVEY);
1263
1264
1265                                         _rtw_memset(&pdev_network->Ssid, 0, sizeof(NDIS_802_11_SSID));
1266                                         _rtw_memcpy(&pdev_network->Ssid, &pmlmepriv->assoc_ssid, sizeof(NDIS_802_11_SSID));
1267
1268                                         rtw_update_registrypriv_dev_network(adapter);
1269                                         rtw_generate_random_ibss(pibss);
1270
1271                                         /*pmlmepriv->fw_state = WIFI_ADHOC_MASTER_STATE;*/
1272                                         init_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE);
1273
1274                                         if (rtw_create_ibss_cmd(adapter, 0) != _SUCCESS)
1275                                                 RTW_ERR("rtw_create_ibss_cmd FAIL\n");
1276
1277                                         pmlmepriv->to_join = _FALSE;
1278                                 }
1279                         }
1280                 } else {
1281                         int s_ret;
1282                         set_fwstate(pmlmepriv, _FW_UNDER_LINKING);
1283                         pmlmepriv->to_join = _FALSE;
1284                         s_ret = rtw_select_and_join_from_scanned_queue(pmlmepriv);
1285                         if (_SUCCESS == s_ret)
1286                                 _set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
1287                         else if (s_ret == 2) { /* there is no need to wait for join */
1288                                 _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING);
1289                                 rtw_indicate_connect(adapter);
1290                         } else {
1291                                 RTW_INFO("try_to_join, but select scanning queue fail, to_roam:%d\n", rtw_to_roam(adapter));
1292
1293                                 if (rtw_to_roam(adapter) != 0) {
1294                                         if (rtw_dec_to_roam(adapter) == 0
1295                                             || _SUCCESS != rtw_sitesurvey_cmd(adapter, &pmlmepriv->assoc_ssid, 1, NULL, 0)
1296                                            ) {
1297                                                 rtw_set_to_roam(adapter, 0);
1298 #ifdef CONFIG_INTEL_WIDI
1299                                                 if (adapter->mlmepriv.widi_state == INTEL_WIDI_STATE_ROAMING) {
1300                                                         _rtw_memset(pmlmepriv->sa_ext, 0x00, L2SDTA_SERVICE_VE_LEN);
1301                                                         intel_widi_wk_cmd(adapter, INTEL_WIDI_LISTEN_WK, NULL, 0);
1302                                                         RTW_INFO("change to widi listen\n");
1303                                                 }
1304 #endif /* CONFIG_INTEL_WIDI */
1305                                                 rtw_free_assoc_resources(adapter, 1);
1306                                                 rtw_indicate_disconnect(adapter, 0, _FALSE);
1307                                         } else
1308                                                 pmlmepriv->to_join = _TRUE;
1309                                 } else
1310                                         rtw_indicate_disconnect(adapter, 0, _FALSE);
1311                                 _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING);
1312                         }
1313                 }
1314         } else {
1315                 if (rtw_chk_roam_flags(adapter, RTW_ROAM_ACTIVE)) {
1316                         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)
1317                             && check_fwstate(pmlmepriv, _FW_LINKED)) {
1318                                 if (rtw_select_roaming_candidate(pmlmepriv) == _SUCCESS) {
1319 #ifdef CONFIG_RTW_80211R
1320                                         if (rtw_chk_ft_flags(adapter, RTW_FT_OVER_DS_SUPPORTED)) {
1321                                                 start_clnt_ft_action(adapter, (u8 *)pmlmepriv->roam_network->network.MacAddress);
1322                                         } else {
1323                                                 /*wait a little time to retrieve packets buffered in the current ap while scan*/
1324                                                 _set_timer(&pmlmeext->ft_roam_timer, 30);
1325                                         }
1326 #else
1327                                         receive_disconnect(adapter, pmlmepriv->cur_network.network.MacAddress
1328                                                 , WLAN_REASON_ACTIVE_ROAM, _FALSE);
1329 #endif
1330                                 }
1331                         }
1332                 }
1333         }
1334
1335         /* RTW_INFO("scan complete in %dms\n",rtw_get_passing_time_ms(pmlmepriv->scan_start_time)); */
1336
1337         _exit_critical_bh(&pmlmepriv->lock, &irqL);
1338
1339 #ifdef CONFIG_P2P_PS
1340         if (check_fwstate(pmlmepriv, _FW_LINKED) == _TRUE)
1341                 p2p_ps_wk_cmd(adapter, P2P_PS_SCAN_DONE, 0);
1342 #endif /* CONFIG_P2P_PS */
1343
1344         rtw_mi_os_xmit_schedule(adapter);
1345
1346 #ifdef CONFIG_DRVEXT_MODULE_WSC
1347         drvext_surveydone_callback(&adapter->drvextpriv);
1348 #endif
1349
1350 #ifdef DBG_CONFIG_ERROR_DETECT
1351         {
1352                 struct mlme_ext_priv *pmlmeext = &adapter->mlmeextpriv;
1353                 if (pmlmeext->sitesurvey_res.bss_cnt == 0) {
1354                         /* rtw_hal_sreset_reset(adapter); */
1355                 }
1356         }
1357 #endif
1358
1359 #ifdef CONFIG_IOCTL_CFG80211
1360         rtw_cfg80211_surveydone_event_callback(adapter);
1361 #endif /* CONFIG_IOCTL_CFG80211 */
1362
1363         rtw_indicate_scan_done(adapter, _FALSE);
1364
1365 #if defined(CONFIG_CONCURRENT_MODE) && defined(CONFIG_IOCTL_CFG80211)
1366         rtw_cfg80211_indicate_scan_done_for_buddy(adapter, _FALSE);
1367 #endif
1368
1369 }
1370
1371 void rtw_dummy_event_callback(_adapter *adapter , u8 *pbuf)
1372 {
1373
1374 }
1375
1376 void rtw_fwdbg_event_callback(_adapter *adapter , u8 *pbuf)
1377 {
1378
1379 }
1380
1381 static void free_scanqueue(struct       mlme_priv *pmlmepriv)
1382 {
1383         _irqL irqL, irqL0;
1384         _queue *free_queue = &pmlmepriv->free_bss_pool;
1385         _queue *scan_queue = &pmlmepriv->scanned_queue;
1386         _list   *plist, *phead, *ptemp;
1387
1388
1389         _enter_critical_bh(&scan_queue->lock, &irqL0);
1390         _enter_critical_bh(&free_queue->lock, &irqL);
1391
1392         phead = get_list_head(scan_queue);
1393         plist = get_next(phead);
1394
1395         while (plist != phead) {
1396                 ptemp = get_next(plist);
1397                 rtw_list_delete(plist);
1398                 rtw_list_insert_tail(plist, &free_queue->queue);
1399                 plist = ptemp;
1400                 pmlmepriv->num_of_scanned--;
1401         }
1402
1403         _exit_critical_bh(&free_queue->lock, &irqL);
1404         _exit_critical_bh(&scan_queue->lock, &irqL0);
1405
1406 }
1407
1408 void rtw_reset_rx_info(struct debug_priv *pdbgpriv)
1409 {
1410         pdbgpriv->dbg_rx_ampdu_drop_count = 0;
1411         pdbgpriv->dbg_rx_ampdu_forced_indicate_count = 0;
1412         pdbgpriv->dbg_rx_ampdu_loss_count = 0;
1413         pdbgpriv->dbg_rx_dup_mgt_frame_drop_count = 0;
1414         pdbgpriv->dbg_rx_ampdu_window_shift_cnt = 0;
1415 }
1416
1417 /*
1418 *rtw_free_assoc_resources: the caller has to lock pmlmepriv->lock
1419 */
1420 void rtw_free_assoc_resources(_adapter *adapter, int lock_scanned_queue)
1421 {
1422         _irqL irqL;
1423         struct wlan_network *pwlan = NULL;
1424         struct  mlme_priv *pmlmepriv = &adapter->mlmepriv;
1425         struct  sta_priv *pstapriv = &adapter->stapriv;
1426         struct wlan_network *tgt_network = &pmlmepriv->cur_network;
1427         struct dvobj_priv *psdpriv = adapter->dvobj;
1428         struct debug_priv *pdbgpriv = &psdpriv->drv_dbg;
1429
1430
1431 #ifdef CONFIG_TDLS
1432         struct tdls_info *ptdlsinfo = &adapter->tdlsinfo;
1433 #endif /* CONFIG_TDLS */
1434
1435
1436         RTW_INFO("%s-"ADPT_FMT" tgt_network MacAddress=" MAC_FMT"ssid=%s\n",
1437                 __func__, ADPT_ARG(adapter), MAC_ARG(tgt_network->network.MacAddress), tgt_network->network.Ssid.Ssid);
1438
1439         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
1440                 struct sta_info *psta;
1441
1442                 psta = rtw_get_stainfo(&adapter->stapriv, tgt_network->network.MacAddress);
1443
1444 #ifdef CONFIG_TDLS
1445                 if (ptdlsinfo->link_established == _TRUE) {
1446                         rtw_tdls_cmd(adapter, NULL, TDLS_RS_RCR);
1447                         rtw_reset_tdls_info(adapter);
1448                         rtw_free_all_stainfo(adapter);
1449                         /* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
1450                 } else
1451 #endif /* CONFIG_TDLS */
1452                 {
1453                         /* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
1454                         rtw_free_stainfo(adapter,  psta);
1455                 }
1456
1457                 /* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
1458
1459         }
1460
1461         if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE | WIFI_ADHOC_MASTER_STATE)) {
1462                 struct sta_info *psta;
1463
1464                 rtw_free_all_stainfo(adapter);
1465
1466                 psta = rtw_get_bcmc_stainfo(adapter);
1467                 /* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL);                */
1468                 rtw_free_stainfo(adapter, psta);
1469                 /* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL);                 */
1470
1471                 rtw_init_bcmc_stainfo(adapter);
1472         }
1473
1474         if (lock_scanned_queue)
1475                 _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1476
1477         pwlan = _rtw_find_same_network(&pmlmepriv->scanned_queue, tgt_network);
1478         if ((pwlan)  && (!check_fwstate(pmlmepriv, WIFI_UNDER_WPS))) {
1479                 pwlan->fixed = _FALSE;
1480
1481                 RTW_INFO("free disconnecting network of scanned_queue\n");
1482                 rtw_free_network_nolock(adapter, pwlan);
1483 #ifdef CONFIG_P2P
1484                 if (!rtw_p2p_chk_state(&adapter->wdinfo, P2P_STATE_NONE)) {
1485                         rtw_mi_set_scan_deny(adapter, 2000);
1486                         /* rtw_clear_scan_deny(adapter);                         */
1487                 }
1488 #endif /* CONFIG_P2P */
1489         } else {
1490                 if (pwlan == NULL)
1491                         RTW_INFO("free disconnecting network of scanned_queue failed due to pwlan== NULL\n\n");
1492                 if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS))
1493                         RTW_INFO("donot free disconnecting network of scanned_queue when WIFI_UNDER_WPS\n\n");
1494         }
1495
1496
1497         if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) && (adapter->stapriv.asoc_sta_count == 1))
1498             /*||check_fwstate(pmlmepriv, WIFI_STATION_STATE)*/) {
1499                 if (pwlan)
1500                         rtw_free_network_nolock(adapter, pwlan);
1501         }
1502
1503         if (lock_scanned_queue)
1504                 _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1505
1506         adapter->securitypriv.key_mask = 0;
1507
1508         rtw_reset_rx_info(pdbgpriv);
1509
1510
1511 }
1512
1513 /*
1514 *rtw_indicate_connect: the caller has to lock pmlmepriv->lock
1515 */
1516 void rtw_indicate_connect(_adapter *padapter)
1517 {
1518         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
1519         struct xmit_priv        *pxmitpriv = &padapter->xmitpriv;
1520
1521
1522
1523         pmlmepriv->to_join = _FALSE;
1524
1525         if (!check_fwstate(&padapter->mlmepriv, _FW_LINKED)) {
1526
1527                 set_fwstate(pmlmepriv, _FW_LINKED);
1528
1529                 rtw_led_control(padapter, LED_CTL_LINK);
1530
1531
1532 #ifdef CONFIG_DRVEXT_MODULE
1533                 if (padapter->drvextpriv.enable_wpa)
1534                         indicate_l2_connect(padapter);
1535                 else
1536 #endif
1537                 {
1538                         rtw_os_indicate_connect(padapter);
1539                 }
1540
1541         }
1542
1543         rtw_set_to_roam(padapter, 0);
1544 #ifdef CONFIG_INTEL_WIDI
1545         if (padapter->mlmepriv.widi_state == INTEL_WIDI_STATE_ROAMING) {
1546                 _rtw_memset(pmlmepriv->sa_ext, 0x00, L2SDTA_SERVICE_VE_LEN);
1547                 intel_widi_wk_cmd(padapter, INTEL_WIDI_LISTEN_WK, NULL, 0);
1548                 RTW_INFO("change to widi listen\n");
1549         }
1550 #endif /* CONFIG_INTEL_WIDI */
1551         if (!check_fwstate(&padapter->mlmepriv, WIFI_AP_STATE))
1552                 rtw_mi_set_scan_deny(padapter, 3000);
1553
1554
1555 }
1556
1557
1558 /*
1559 *rtw_indicate_disconnect: the caller has to lock pmlmepriv->lock
1560 */
1561 void rtw_indicate_disconnect(_adapter *padapter, u16 reason, u8 locally_generated)
1562 {
1563         struct  mlme_priv *pmlmepriv = &padapter->mlmepriv;
1564         struct mlme_ext_priv *pmlmeext = &(padapter->mlmeextpriv);
1565         struct mlme_ext_info *pmlmeinfo = &(pmlmeext->mlmext_info);
1566         WLAN_BSSID_EX   *cur_network = &(pmlmeinfo->network);
1567         struct sta_info *psta;
1568         struct sta_priv *pstapriv = &padapter->stapriv;
1569         u8 *wps_ie = NULL;
1570         uint wpsie_len = 0;
1571
1572
1573
1574         _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING | WIFI_UNDER_WPS);
1575
1576         /* force to clear cur_network_scanned's SELECTED REGISTRAR */
1577         if (pmlmepriv->cur_network_scanned) {
1578                 WLAN_BSSID_EX   *current_joined_bss = &(pmlmepriv->cur_network_scanned->network);
1579                 if (current_joined_bss) {
1580                         wps_ie = rtw_get_wps_ie(current_joined_bss->IEs + _FIXED_IE_LENGTH_,
1581                                 current_joined_bss->IELength - _FIXED_IE_LENGTH_, NULL, &wpsie_len);
1582                         if (wps_ie && wpsie_len > 0) {
1583                                 u8 *attr = NULL;
1584                                 u32 attr_len;
1585                                 attr = rtw_get_wps_attr(wps_ie, wpsie_len, WPS_ATTR_SELECTED_REGISTRAR,
1586                                                         NULL, &attr_len);
1587                                 if (attr)
1588                                         *(attr + 4) = 0;
1589                         }
1590                 }
1591         }
1592         /* RTW_INFO("clear wps when %s\n", __func__); */
1593
1594         if (rtw_to_roam(padapter) > 0)
1595                 _clr_fwstate_(pmlmepriv, _FW_LINKED);
1596
1597 #ifdef CONFIG_WAPI_SUPPORT
1598         psta = rtw_get_stainfo(pstapriv, cur_network->MacAddress);
1599         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE))
1600                 rtw_wapi_return_one_sta_info(padapter, psta->hwaddr);
1601         else if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) ||
1602                  check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE))
1603                 rtw_wapi_return_all_sta_info(padapter);
1604 #endif
1605
1606         if (check_fwstate(&padapter->mlmepriv, _FW_LINKED)
1607             || (rtw_to_roam(padapter) <= 0)
1608            ) {
1609
1610                 rtw_os_indicate_disconnect(padapter, reason, locally_generated);
1611
1612                 /* set ips_deny_time to avoid enter IPS before LPS leave */
1613                 rtw_set_ips_deny(padapter, 3000);
1614
1615                 _clr_fwstate_(pmlmepriv, _FW_LINKED);
1616
1617                 rtw_led_control(padapter, LED_CTL_NO_LINK);
1618
1619                 rtw_clear_scan_deny(padapter);
1620         }
1621
1622 #ifdef CONFIG_P2P_PS
1623         p2p_ps_wk_cmd(padapter, P2P_PS_DISABLE, 1);
1624 #endif /* CONFIG_P2P_PS */
1625
1626 #ifdef CONFIG_LPS
1627         rtw_lps_ctrl_wk_cmd(padapter, LPS_CTRL_DISCONNECT, 1);
1628 #endif
1629
1630 #ifdef CONFIG_BEAMFORMING
1631         beamforming_wk_cmd(padapter, BEAMFORMING_CTRL_LEAVE, cur_network->MacAddress, ETH_ALEN, 1);
1632 #endif /*CONFIG_BEAMFORMING*/
1633
1634 }
1635
1636 inline void rtw_indicate_scan_done(_adapter *padapter, bool aborted)
1637 {
1638         RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(padapter));
1639
1640         rtw_os_indicate_scan_done(padapter, aborted);
1641
1642 #ifdef CONFIG_IPS
1643         if (is_primary_adapter(padapter)
1644             && (_FALSE == adapter_to_pwrctl(padapter)->bInSuspend)
1645             && (check_fwstate(&padapter->mlmepriv, WIFI_ASOC_STATE | WIFI_UNDER_LINKING) == _FALSE)) {
1646                 struct pwrctrl_priv *pwrpriv;
1647
1648                 pwrpriv = adapter_to_pwrctl(padapter);
1649                 rtw_set_ips_deny(padapter, 0);
1650 #ifdef CONFIG_IPS_CHECK_IN_WD
1651                 _set_timer(&adapter_to_dvobj(padapter)->dynamic_chk_timer, 1);
1652 #else /* !CONFIG_IPS_CHECK_IN_WD */
1653                 _rtw_set_pwr_state_check_timer(pwrpriv, 1);
1654 #endif /* !CONFIG_IPS_CHECK_IN_WD */
1655         }
1656 #endif /* CONFIG_IPS */
1657 }
1658
1659 static u32 _rtw_wait_scan_done(_adapter *adapter, u8 abort, u32 timeout_ms)
1660 {
1661         u32 start;
1662         u32 pass_ms;
1663         struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
1664         struct mlme_ext_priv *pmlmeext = &(adapter->mlmeextpriv);
1665
1666         start = rtw_get_current_time();
1667
1668         pmlmeext->scan_abort = abort;
1669
1670         while (check_fwstate(pmlmepriv, _FW_UNDER_SURVEY)
1671                && rtw_get_passing_time_ms(start) <= timeout_ms) {
1672
1673                 if (RTW_CANNOT_RUN(adapter))
1674                         break;
1675
1676                 RTW_INFO(FUNC_NDEV_FMT"fw_state=_FW_UNDER_SURVEY!\n", FUNC_NDEV_ARG(adapter->pnetdev));
1677                 rtw_msleep_os(20);
1678         }
1679
1680         if (_TRUE == abort) {
1681                 if (check_fwstate(pmlmepriv, _FW_UNDER_SURVEY)) {
1682                         if (!RTW_CANNOT_RUN(adapter))
1683                                 RTW_INFO(FUNC_NDEV_FMT"waiting for scan_abort time out!\n", FUNC_NDEV_ARG(adapter->pnetdev));
1684 #ifdef CONFIG_PLATFORM_MSTAR
1685                         /*_clr_fwstate_(pmlmepriv, _FW_UNDER_SURVEY);*/
1686                         set_survey_timer(pmlmeext, 0);
1687                         mlme_set_scan_to_timer(pmlmepriv, 50);
1688 #endif
1689                         rtw_indicate_scan_done(adapter, _TRUE);
1690                 }
1691         }
1692
1693         pmlmeext->scan_abort = _FALSE;
1694         pass_ms = rtw_get_passing_time_ms(start);
1695
1696         return pass_ms;
1697
1698 }
1699
1700 void rtw_scan_wait_completed(_adapter *adapter)
1701 {
1702         u32 scan_to = SCANNING_TIMEOUT;
1703
1704 #ifdef CONFIG_SCAN_BACKOP
1705         if (IsSupported5G(adapter->registrypriv.wireless_mode)
1706             && IsSupported24G(adapter->registrypriv.wireless_mode)) /*dual band*/
1707                 scan_to = CONC_SCANNING_TIMEOUT_DUAL_BAND;
1708         else /*single band*/
1709                 scan_to = CONC_SCANNING_TIMEOUT_SINGLE_BAND;
1710 #endif /* CONFIG_SCAN_BACKOP */
1711
1712         _rtw_wait_scan_done(adapter, _FALSE, scan_to);
1713 }
1714
1715 u32 rtw_scan_abort_timeout(_adapter *adapter, u32 timeout_ms)
1716 {
1717         return _rtw_wait_scan_done(adapter, _TRUE, timeout_ms);
1718 }
1719
1720 void rtw_scan_abort_no_wait(_adapter *adapter)
1721 {
1722         struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
1723         struct mlme_ext_priv *pmlmeext = &(adapter->mlmeextpriv);
1724
1725         if (check_fwstate(pmlmepriv, _FW_UNDER_SURVEY))
1726                 pmlmeext->scan_abort = _TRUE;
1727 }
1728
1729 void rtw_scan_abort(_adapter *adapter)
1730 {
1731         rtw_scan_abort_timeout(adapter, 200);
1732 }
1733
1734 static struct sta_info *rtw_joinbss_update_stainfo(_adapter *padapter, struct wlan_network *pnetwork)
1735 {
1736         int i;
1737         struct sta_info *bmc_sta, *psta = NULL;
1738         struct recv_reorder_ctrl *preorder_ctrl;
1739         struct sta_priv *pstapriv = &padapter->stapriv;
1740         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
1741
1742         psta = rtw_get_stainfo(pstapriv, pnetwork->network.MacAddress);
1743         if (psta == NULL)
1744                 psta = rtw_alloc_stainfo(pstapriv, pnetwork->network.MacAddress);
1745
1746         if (psta) { /* update ptarget_sta */
1747                 RTW_INFO("%s\n", __FUNCTION__);
1748
1749                 psta->aid  = pnetwork->join_res;
1750
1751 #if 0 /* alloc macid when call rtw_alloc_stainfo(), and release macid when call rtw_free_stainfo() */
1752 #ifdef CONFIG_CONCURRENT_MODE
1753
1754                 if (PRIMARY_ADAPTER == padapter->adapter_type)
1755                         psta->mac_id = 0;
1756                 else
1757                         psta->mac_id = 2;
1758 #else
1759                 psta->mac_id = 0;
1760 #endif
1761 #endif /* removed */
1762
1763                 update_sta_info(padapter, psta);
1764
1765                 /* update station supportRate */
1766                 psta->bssratelen = rtw_get_rateset_len(pnetwork->network.SupportedRates);
1767                 _rtw_memcpy(psta->bssrateset, pnetwork->network.SupportedRates, psta->bssratelen);
1768                 rtw_hal_update_sta_rate_mask(padapter, psta);
1769
1770                 psta->wireless_mode = pmlmeext->cur_wireless_mode;
1771                 psta->raid = rtw_hal_networktype_to_raid(padapter, psta);
1772
1773
1774                 /* sta mode */
1775                 rtw_hal_set_odm_var(padapter, HAL_ODM_STA_INFO, psta, _TRUE);
1776
1777                 /* security related */
1778 #ifdef CONFIG_RTW_80211R
1779                 if ((padapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) && (psta->ft_pairwise_key_installed == _FALSE)) {
1780 #else
1781                 if (padapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) {
1782 #endif
1783                         padapter->securitypriv.binstallGrpkey = _FALSE;
1784                         padapter->securitypriv.busetkipkey = _FALSE;
1785                         padapter->securitypriv.bgrpkey_handshake = _FALSE;
1786
1787                         psta->ieee8021x_blocked = _TRUE;
1788                         psta->dot118021XPrivacy = padapter->securitypriv.dot11PrivacyAlgrthm;
1789
1790                         _rtw_memset((u8 *)&psta->dot118021x_UncstKey, 0, sizeof(union Keytype));
1791
1792                         _rtw_memset((u8 *)&psta->dot11tkiprxmickey, 0, sizeof(union Keytype));
1793                         _rtw_memset((u8 *)&psta->dot11tkiptxmickey, 0, sizeof(union Keytype));
1794
1795                         _rtw_memset((u8 *)&psta->dot11txpn, 0, sizeof(union pn48));
1796                         psta->dot11txpn.val = psta->dot11txpn.val + 1;
1797 #ifdef CONFIG_IEEE80211W
1798                         _rtw_memset((u8 *)&psta->dot11wtxpn, 0, sizeof(union pn48));
1799 #endif /* CONFIG_IEEE80211W */
1800                         _rtw_memset((u8 *)&psta->dot11rxpn, 0, sizeof(union pn48));
1801                 }
1802
1803                 /*      Commented by Albert 2012/07/21 */
1804                 /*      When doing the WPS, the wps_ie_len won't equal to 0 */
1805                 /*      And the Wi-Fi driver shouldn't allow the data packet to be tramsmitted. */
1806                 if (padapter->securitypriv.wps_ie_len != 0) {
1807                         psta->ieee8021x_blocked = _TRUE;
1808                         padapter->securitypriv.wps_ie_len = 0;
1809                 }
1810
1811
1812                 /* for A-MPDU Rx reordering buffer control for bmc_sta & sta_info */
1813                 /* if A-MPDU Rx is enabled, reseting  rx_ordering_ctrl wstart_b(indicate_seq) to default value=0xffff */
1814                 /* todo: check if AP can send A-MPDU packets */
1815                 for (i = 0; i < 16 ; i++) {
1816                         /* preorder_ctrl = &precvpriv->recvreorder_ctrl[i]; */
1817                         preorder_ctrl = &psta->recvreorder_ctrl[i];
1818                         preorder_ctrl->enable = _FALSE;
1819                         preorder_ctrl->indicate_seq = 0xffff;
1820 #ifdef DBG_RX_SEQ
1821                         RTW_INFO("DBG_RX_SEQ %s:%d indicate_seq:%u\n", __FUNCTION__, __LINE__,
1822                                  preorder_ctrl->indicate_seq);
1823 #endif
1824                         preorder_ctrl->wend_b = 0xffff;
1825                         preorder_ctrl->wsize_b = 64;/* max_ampdu_sz; */ /* ex. 32(kbytes) -> wsize_b=32 */
1826                         preorder_ctrl->ampdu_size = RX_AMPDU_SIZE_INVALID;
1827                 }
1828
1829
1830                 bmc_sta = rtw_get_bcmc_stainfo(padapter);
1831                 if (bmc_sta) {
1832                         for (i = 0; i < 16 ; i++) {
1833                                 /* preorder_ctrl = &precvpriv->recvreorder_ctrl[i]; */
1834                                 preorder_ctrl = &bmc_sta->recvreorder_ctrl[i];
1835                                 preorder_ctrl->enable = _FALSE;
1836                                 preorder_ctrl->indicate_seq = 0xffff;
1837 #ifdef DBG_RX_SEQ
1838                                 RTW_INFO("DBG_RX_SEQ %s:%d indicate_seq:%u\n", __FUNCTION__, __LINE__,
1839                                          preorder_ctrl->indicate_seq);
1840 #endif
1841                                 preorder_ctrl->wend_b = 0xffff;
1842                                 preorder_ctrl->wsize_b = 64;/* max_ampdu_sz; */ /* ex. 32(kbytes) -> wsize_b=32 */
1843                                 preorder_ctrl->ampdu_size = RX_AMPDU_SIZE_INVALID;
1844                         }
1845                 }
1846         }
1847
1848         return psta;
1849
1850 }
1851
1852 /* pnetwork : returns from rtw_joinbss_event_callback
1853  * ptarget_wlan: found from scanned_queue */
1854 static void rtw_joinbss_update_network(_adapter *padapter, struct wlan_network *ptarget_wlan, struct wlan_network  *pnetwork)
1855 {
1856         struct mlme_priv        *pmlmepriv = &(padapter->mlmepriv);
1857         struct wlan_network  *cur_network = &(pmlmepriv->cur_network);
1858
1859         RTW_INFO("%s\n", __FUNCTION__);
1860
1861
1862
1863         /* why not use ptarget_wlan?? */
1864         _rtw_memcpy(&cur_network->network, &pnetwork->network, pnetwork->network.Length);
1865         /* some IEs in pnetwork is wrong, so we should use ptarget_wlan IEs */
1866         cur_network->network.IELength = ptarget_wlan->network.IELength;
1867         _rtw_memcpy(&cur_network->network.IEs[0], &ptarget_wlan->network.IEs[0], MAX_IE_SZ);
1868
1869         cur_network->aid = pnetwork->join_res;
1870
1871
1872 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
1873         rtw_set_signal_stat_timer(&padapter->recvpriv);
1874 #endif
1875         padapter->recvpriv.signal_strength = ptarget_wlan->network.PhyInfo.SignalStrength;
1876         padapter->recvpriv.signal_qual = ptarget_wlan->network.PhyInfo.SignalQuality;
1877         /* the ptarget_wlan->network.Rssi is raw data, we use ptarget_wlan->network.PhyInfo.SignalStrength instead (has scaled) */
1878         padapter->recvpriv.rssi = translate_percentage_to_dbm(ptarget_wlan->network.PhyInfo.SignalStrength);
1879 #if defined(DBG_RX_SIGNAL_DISPLAY_PROCESSING) && 1
1880         RTW_INFO(FUNC_ADPT_FMT" signal_strength:%3u, rssi:%3d, signal_qual:%3u"
1881                  "\n"
1882                  , FUNC_ADPT_ARG(padapter)
1883                  , padapter->recvpriv.signal_strength
1884                  , padapter->recvpriv.rssi
1885                  , padapter->recvpriv.signal_qual
1886                 );
1887 #endif
1888 #ifdef CONFIG_NEW_SIGNAL_STAT_PROCESS
1889         rtw_set_signal_stat_timer(&padapter->recvpriv);
1890 #endif
1891
1892         /* update fw_state */ /* will clr _FW_UNDER_LINKING here indirectly */
1893
1894         switch (pnetwork->network.InfrastructureMode) {
1895         case Ndis802_11Infrastructure:
1896
1897                 if (pmlmepriv->fw_state & WIFI_UNDER_WPS)
1898                         /*pmlmepriv->fw_state = WIFI_STATION_STATE|WIFI_UNDER_WPS;*/
1899                         init_fwstate(pmlmepriv, WIFI_STATION_STATE | WIFI_UNDER_WPS);
1900                 else
1901                         /*pmlmepriv->fw_state = WIFI_STATION_STATE;*/
1902                         init_fwstate(pmlmepriv, WIFI_STATION_STATE);
1903                 break;
1904         case Ndis802_11IBSS:
1905                 /*pmlmepriv->fw_state = WIFI_ADHOC_STATE;*/
1906                 init_fwstate(pmlmepriv, WIFI_ADHOC_STATE);
1907                 break;
1908         default:
1909                 /*pmlmepriv->fw_state = WIFI_NULL_STATE;*/
1910                 init_fwstate(pmlmepriv, WIFI_NULL_STATE);
1911                 break;
1912         }
1913
1914         rtw_update_protection(padapter, (cur_network->network.IEs) + sizeof(NDIS_802_11_FIXED_IEs),
1915                               (cur_network->network.IELength));
1916
1917 #ifdef CONFIG_80211N_HT
1918         rtw_update_ht_cap(padapter, cur_network->network.IEs, cur_network->network.IELength, (u8) cur_network->network.Configuration.DSConfig);
1919 #endif
1920 }
1921
1922 /* Notes: the fucntion could be > passive_level (the same context as Rx tasklet)
1923  * pnetwork : returns from rtw_joinbss_event_callback
1924  * ptarget_wlan: found from scanned_queue
1925  * if join_res > 0, for (fw_state==WIFI_STATION_STATE), we check if  "ptarget_sta" & "ptarget_wlan" exist.
1926  * if join_res > 0, for (fw_state==WIFI_ADHOC_STATE), we only check if "ptarget_wlan" exist.
1927  * if join_res > 0, update "cur_network->network" from "pnetwork->network" if (ptarget_wlan !=NULL).
1928  */
1929 /* #define REJOIN */
1930 void rtw_joinbss_event_prehandle(_adapter *adapter, u8 *pbuf)
1931 {
1932         _irqL irqL, irqL2;
1933         static u8 retry = 0;
1934         u8 timer_cancelled;
1935         struct sta_info *ptarget_sta = NULL, *pcur_sta = NULL;
1936         struct  sta_priv *pstapriv = &adapter->stapriv;
1937         struct  mlme_priv       *pmlmepriv = &(adapter->mlmepriv);
1938         struct wlan_network     *pnetwork       = (struct wlan_network *)pbuf;
1939         struct wlan_network     *cur_network = &(pmlmepriv->cur_network);
1940         struct wlan_network     *pcur_wlan = NULL, *ptarget_wlan = NULL;
1941         unsigned int            the_same_macaddr = _FALSE;
1942
1943
1944 #ifdef CONFIG_RTL8712
1945         /* endian_convert */
1946         pnetwork->join_res = le32_to_cpu(pnetwork->join_res);
1947         pnetwork->network_type = le32_to_cpu(pnetwork->network_type);
1948         pnetwork->network.Length = le32_to_cpu(pnetwork->network.Length);
1949         pnetwork->network.Ssid.SsidLength = le32_to_cpu(pnetwork->network.Ssid.SsidLength);
1950         pnetwork->network.Privacy = le32_to_cpu(pnetwork->network.Privacy);
1951         pnetwork->network.Rssi = le32_to_cpu(pnetwork->network.Rssi);
1952         pnetwork->network.NetworkTypeInUse = le32_to_cpu(pnetwork->network.NetworkTypeInUse) ;
1953         pnetwork->network.Configuration.ATIMWindow = le32_to_cpu(pnetwork->network.Configuration.ATIMWindow);
1954         pnetwork->network.Configuration.BeaconPeriod = le32_to_cpu(pnetwork->network.Configuration.BeaconPeriod);
1955         pnetwork->network.Configuration.DSConfig = le32_to_cpu(pnetwork->network.Configuration.DSConfig);
1956         pnetwork->network.Configuration.FHConfig.DwellTime = le32_to_cpu(pnetwork->network.Configuration.FHConfig.DwellTime);
1957         pnetwork->network.Configuration.FHConfig.HopPattern = le32_to_cpu(pnetwork->network.Configuration.FHConfig.HopPattern);
1958         pnetwork->network.Configuration.FHConfig.HopSet = le32_to_cpu(pnetwork->network.Configuration.FHConfig.HopSet);
1959         pnetwork->network.Configuration.FHConfig.Length = le32_to_cpu(pnetwork->network.Configuration.FHConfig.Length);
1960         pnetwork->network.Configuration.Length = le32_to_cpu(pnetwork->network.Configuration.Length);
1961         pnetwork->network.InfrastructureMode = le32_to_cpu(pnetwork->network.InfrastructureMode);
1962         pnetwork->network.IELength = le32_to_cpu(pnetwork->network.IELength);
1963 #endif
1964
1965
1966         rtw_get_encrypt_decrypt_from_registrypriv(adapter);
1967
1968
1969
1970         the_same_macaddr = _rtw_memcmp(pnetwork->network.MacAddress, cur_network->network.MacAddress, ETH_ALEN);
1971
1972         pnetwork->network.Length = get_WLAN_BSSID_EX_sz(&pnetwork->network);
1973         if (pnetwork->network.Length > sizeof(WLAN_BSSID_EX)) {
1974                 goto ignore_joinbss_callback;
1975         }
1976
1977         _enter_critical_bh(&pmlmepriv->lock, &irqL);
1978
1979         pmlmepriv->LinkDetectInfo.TrafficTransitionCount = 0;
1980         pmlmepriv->LinkDetectInfo.LowPowerTransitionCount = 0;
1981
1982
1983         if (pnetwork->join_res > 0) {
1984                 _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
1985                 retry = 0;
1986                 if (check_fwstate(pmlmepriv, _FW_UNDER_LINKING)) {
1987                         /* s1. find ptarget_wlan */
1988                         if (check_fwstate(pmlmepriv, _FW_LINKED)) {
1989                                 if (the_same_macaddr == _TRUE)
1990                                         ptarget_wlan = rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
1991                                 else {
1992                                         pcur_wlan = rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
1993                                         if (pcur_wlan)
1994                                                 pcur_wlan->fixed = _FALSE;
1995
1996                                         pcur_sta = rtw_get_stainfo(pstapriv, cur_network->network.MacAddress);
1997                                         if (pcur_sta) {
1998                                                 /* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL2); */
1999                                                 rtw_free_stainfo(adapter,  pcur_sta);
2000                                                 /* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL2); */
2001                                         }
2002
2003                                         ptarget_wlan = rtw_find_network(&pmlmepriv->scanned_queue, pnetwork->network.MacAddress);
2004                                         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2005                                                 if (ptarget_wlan)
2006                                                         ptarget_wlan->fixed = _TRUE;
2007                                         }
2008                                 }
2009
2010                         } else {
2011                                 ptarget_wlan = _rtw_find_same_network(&pmlmepriv->scanned_queue, pnetwork);
2012                                 if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2013                                         if (ptarget_wlan)
2014                                                 ptarget_wlan->fixed = _TRUE;
2015                                 }
2016                         }
2017
2018                         /* s2. update cur_network */
2019                         if (ptarget_wlan)
2020                                 rtw_joinbss_update_network(adapter, ptarget_wlan, pnetwork);
2021                         else {
2022                                 RTW_PRINT("Can't find ptarget_wlan when joinbss_event callback\n");
2023                                 _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2024                                 goto ignore_joinbss_callback;
2025                         }
2026
2027
2028                         /* s3. find ptarget_sta & update ptarget_sta after update cur_network only for station mode */
2029                         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) == _TRUE) {
2030                                 ptarget_sta = rtw_joinbss_update_stainfo(adapter, pnetwork);
2031                                 if (ptarget_sta == NULL) {
2032                                         RTW_ERR("Can't update stainfo when joinbss_event callback\n");
2033                                         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2034                                         goto ignore_joinbss_callback;
2035                                 }
2036                         }
2037
2038                         /* s4. indicate connect                  */
2039                         if (MLME_IS_STA(adapter) || MLME_IS_ADHOC(adapter)) {
2040                                 pmlmepriv->cur_network_scanned = ptarget_wlan;
2041                                 rtw_indicate_connect(adapter);
2042                         }
2043
2044                         /* s5. Cancle assoc_timer                                        */
2045                         _cancel_timer(&pmlmepriv->assoc_timer, &timer_cancelled);
2046
2047
2048                 } else {
2049                         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2050                         goto ignore_joinbss_callback;
2051                 }
2052
2053                 _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2054
2055         } else if (pnetwork->join_res == -4) {
2056                 rtw_reset_securitypriv(adapter);
2057                 _set_timer(&pmlmepriv->assoc_timer, 1);
2058
2059                 /* rtw_free_assoc_resources(adapter, 1); */
2060
2061                 if ((check_fwstate(pmlmepriv, _FW_UNDER_LINKING)) == _TRUE) {
2062                         _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING);
2063                 }
2064
2065         } else { /* if join_res < 0 (join fails), then try again */
2066
2067 #ifdef REJOIN
2068                 res = _FAIL;
2069                 if (retry < 2) {
2070                         res = rtw_select_and_join_from_scanned_queue(pmlmepriv);
2071                 }
2072
2073                 if (res == _SUCCESS) {
2074                         /* extend time of assoc_timer */
2075                         _set_timer(&pmlmepriv->assoc_timer, MAX_JOIN_TIMEOUT);
2076                         retry++;
2077                 } else if (res == 2) { /* there is no need to wait for join */
2078                         _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING);
2079                         rtw_indicate_connect(adapter);
2080                 } else {
2081 #endif
2082
2083                         _set_timer(&pmlmepriv->assoc_timer, 1);
2084                         /* rtw_free_assoc_resources(adapter, 1); */
2085                         _clr_fwstate_(pmlmepriv, _FW_UNDER_LINKING);
2086
2087 #ifdef REJOIN
2088                         retry = 0;
2089                 }
2090 #endif
2091         }
2092
2093 ignore_joinbss_callback:
2094         _exit_critical_bh(&pmlmepriv->lock, &irqL);
2095 }
2096
2097 void rtw_joinbss_event_callback(_adapter *adapter, u8 *pbuf)
2098 {
2099         struct wlan_network     *pnetwork       = (struct wlan_network *)pbuf;
2100
2101
2102         mlmeext_joinbss_event_callback(adapter, pnetwork->join_res);
2103
2104         rtw_mi_os_xmit_schedule(adapter);
2105
2106 }
2107
2108 void rtw_sta_media_status_rpt(_adapter *adapter, struct sta_info *sta, bool connected)
2109 {
2110         struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
2111         bool miracast_enabled = 0;
2112         bool miracast_sink = 0;
2113         u8 role = H2C_MSR_ROLE_RSVD;
2114
2115         if (sta == NULL) {
2116                 RTW_PRINT(FUNC_ADPT_FMT" sta is NULL\n"
2117                           , FUNC_ADPT_ARG(adapter));
2118                 rtw_warn_on(1);
2119                 return;
2120         }
2121
2122         if (sta->mac_id >= macid_ctl->num) {
2123                 RTW_PRINT(FUNC_ADPT_FMT" invalid macid:%u\n"
2124                           , FUNC_ADPT_ARG(adapter), sta->mac_id);
2125                 rtw_warn_on(1);
2126                 return;
2127         }
2128
2129         if (!rtw_macid_is_used(macid_ctl, sta->mac_id)) {
2130                 RTW_PRINT(FUNC_ADPT_FMT" macid:%u not is used, set connected to 0\n"
2131                           , FUNC_ADPT_ARG(adapter), sta->mac_id);
2132                 connected = 0;
2133                 rtw_warn_on(1);
2134         }
2135
2136         if (connected && !rtw_macid_is_bmc(macid_ctl, sta->mac_id)) {
2137                 miracast_enabled = STA_OP_WFD_MODE(sta) != 0 && is_miracast_enabled(adapter);
2138                 miracast_sink = miracast_enabled && (STA_OP_WFD_MODE(sta) & MIRACAST_SINK);
2139
2140 #ifdef CONFIG_TDLS
2141                 if (sta->tdls_sta_state & TDLS_LINKED_STATE)
2142                         role = H2C_MSR_ROLE_TDLS;
2143                 else
2144 #endif
2145                         if (MLME_IS_STA(adapter)) {
2146                                 if (MLME_IS_GC(adapter))
2147                                         role = H2C_MSR_ROLE_GO;
2148                                 else
2149                                         role = H2C_MSR_ROLE_AP;
2150                         } else if (MLME_IS_AP(adapter)) {
2151                                 if (MLME_IS_GO(adapter))
2152                                         role = H2C_MSR_ROLE_GC;
2153                                 else
2154                                         role = H2C_MSR_ROLE_STA;
2155                         } else if (MLME_IS_ADHOC(adapter) || MLME_IS_ADHOC_MASTER(adapter))
2156                                 role = H2C_MSR_ROLE_ADHOC;
2157
2158 #ifdef CONFIG_WFD
2159                 if (role == H2C_MSR_ROLE_GC
2160                     || role == H2C_MSR_ROLE_GO
2161                     || role == H2C_MSR_ROLE_TDLS
2162                    ) {
2163                         if (adapter->wfd_info.rtsp_ctrlport
2164                             || adapter->wfd_info.tdls_rtsp_ctrlport
2165                             || adapter->wfd_info.peer_rtsp_ctrlport)
2166                                 rtw_wfd_st_switch(sta, 1);
2167                 }
2168 #endif
2169         }
2170
2171         rtw_hal_set_FwMediaStatusRpt_single_cmd(adapter
2172                                                 , connected
2173                                                 , miracast_enabled
2174                                                 , miracast_sink
2175                                                 , role
2176                                                 , sta->mac_id
2177                                                );
2178 }
2179
2180 u8 rtw_sta_media_status_rpt_cmd(_adapter *adapter, struct sta_info *sta, bool connected)
2181 {
2182         struct cmd_priv *cmdpriv = &adapter->cmdpriv;
2183         struct cmd_obj *cmdobj;
2184         struct drvextra_cmd_parm *cmd_parm;
2185         struct sta_media_status_rpt_cmd_parm *rpt_parm;
2186         u8      res = _SUCCESS;
2187
2188         cmdobj = (struct cmd_obj *)rtw_zmalloc(sizeof(struct cmd_obj));
2189         if (cmdobj == NULL) {
2190                 res = _FAIL;
2191                 goto exit;
2192         }
2193
2194         cmd_parm = (struct drvextra_cmd_parm *)rtw_zmalloc(sizeof(struct drvextra_cmd_parm));
2195         if (cmd_parm == NULL) {
2196                 rtw_mfree((u8 *)cmdobj, sizeof(struct cmd_obj));
2197                 res = _FAIL;
2198                 goto exit;
2199         }
2200
2201         rpt_parm = (struct sta_media_status_rpt_cmd_parm *)rtw_zmalloc(sizeof(struct sta_media_status_rpt_cmd_parm));
2202         if (rpt_parm == NULL) {
2203                 rtw_mfree((u8 *)cmdobj, sizeof(struct cmd_obj));
2204                 rtw_mfree((u8 *)cmd_parm, sizeof(struct drvextra_cmd_parm));
2205                 res = _FAIL;
2206                 goto exit;
2207         }
2208
2209         rpt_parm->sta = sta;
2210         rpt_parm->connected = connected;
2211
2212         cmd_parm->ec_id = STA_MSTATUS_RPT_WK_CID;
2213         cmd_parm->type = 0;
2214         cmd_parm->size = sizeof(struct sta_media_status_rpt_cmd_parm);
2215         cmd_parm->pbuf = (u8 *)rpt_parm;
2216         init_h2fwcmd_w_parm_no_rsp(cmdobj, cmd_parm, GEN_CMD_CODE(_Set_Drv_Extra));
2217
2218         res = rtw_enqueue_cmd(cmdpriv, cmdobj);
2219
2220 exit:
2221         return res;
2222 }
2223
2224 inline void rtw_sta_media_status_rpt_cmd_hdl(_adapter *adapter, struct sta_media_status_rpt_cmd_parm *parm)
2225 {
2226         rtw_sta_media_status_rpt(adapter, parm->sta, parm->connected);
2227 }
2228
2229 void rtw_stassoc_event_callback(_adapter *adapter, u8 *pbuf)
2230 {
2231         _irqL irqL;
2232         struct sta_info *psta;
2233         struct mlme_priv *pmlmepriv = &(adapter->mlmepriv);
2234         struct stassoc_event    *pstassoc       = (struct stassoc_event *)pbuf;
2235         struct wlan_network     *cur_network = &(pmlmepriv->cur_network);
2236         struct wlan_network     *ptarget_wlan = NULL;
2237
2238
2239 #if CONFIG_RTW_MACADDR_ACL
2240         if (rtw_access_ctrl(adapter, pstassoc->macaddr) == _FALSE)
2241                 return;
2242 #endif
2243
2244 #if defined(CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME)
2245         if (check_fwstate(pmlmepriv, WIFI_AP_STATE)) {
2246                 psta = rtw_get_stainfo(&adapter->stapriv, pstassoc->macaddr);
2247                 if (psta) {
2248                         u8 *passoc_req = NULL;
2249                         u32 assoc_req_len = 0;
2250
2251                         rtw_sta_media_status_rpt(adapter, psta, 1);
2252
2253 #ifndef CONFIG_AUTO_AP_MODE
2254
2255                         ap_sta_info_defer_update(adapter, psta);
2256
2257                         /* report to upper layer */
2258                         RTW_INFO("indicate_sta_assoc_event to upper layer - hostapd\n");
2259 #ifdef CONFIG_IOCTL_CFG80211
2260                         _enter_critical_bh(&psta->lock, &irqL);
2261                         if (psta->passoc_req && psta->assoc_req_len > 0) {
2262                                 passoc_req = rtw_zmalloc(psta->assoc_req_len);
2263                                 if (passoc_req) {
2264                                         assoc_req_len = psta->assoc_req_len;
2265                                         _rtw_memcpy(passoc_req, psta->passoc_req, assoc_req_len);
2266
2267                                         rtw_mfree(psta->passoc_req , psta->assoc_req_len);
2268                                         psta->passoc_req = NULL;
2269                                         psta->assoc_req_len = 0;
2270                                 }
2271                         }
2272                         _exit_critical_bh(&psta->lock, &irqL);
2273
2274                         if (passoc_req && assoc_req_len > 0) {
2275                                 rtw_cfg80211_indicate_sta_assoc(adapter, passoc_req, assoc_req_len);
2276
2277                                 rtw_mfree(passoc_req, assoc_req_len);
2278                         }
2279 #else /* !CONFIG_IOCTL_CFG80211  */
2280                         rtw_indicate_sta_assoc_event(adapter, psta);
2281 #endif /* !CONFIG_IOCTL_CFG80211 */
2282 #endif /* !CONFIG_AUTO_AP_MODE */
2283
2284 #ifdef CONFIG_BEAMFORMING
2285                         beamforming_wk_cmd(adapter, BEAMFORMING_CTRL_ENTER, (u8 *)psta, sizeof(struct sta_info), 0);
2286 #endif/*CONFIG_BEAMFORMING*/
2287                         if (is_wep_enc(adapter->securitypriv.dot11PrivacyAlgrthm))
2288                                 rtw_ap_wep_pk_setting(adapter, psta);
2289                 }
2290                 goto exit;
2291         }
2292 #endif /* defined (CONFIG_AP_MODE) && defined (CONFIG_NATIVEAP_MLME) */
2293
2294         /* for AD-HOC mode */
2295         psta = rtw_get_stainfo(&adapter->stapriv, pstassoc->macaddr);
2296         if (psta == NULL) {
2297                 RTW_ERR(FUNC_ADPT_FMT" get no sta_info with "MAC_FMT"\n"
2298                         , FUNC_ADPT_ARG(adapter), MAC_ARG(pstassoc->macaddr));
2299                 rtw_warn_on(1);
2300                 goto exit;
2301         }
2302
2303         rtw_hal_set_odm_var(adapter, HAL_ODM_STA_INFO, psta, _TRUE);
2304
2305         rtw_sta_media_status_rpt(adapter, psta, 1);
2306
2307         if (adapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X)
2308                 psta->dot118021XPrivacy = adapter->securitypriv.dot11PrivacyAlgrthm;
2309
2310
2311         psta->ieee8021x_blocked = _FALSE;
2312
2313         _enter_critical_bh(&pmlmepriv->lock, &irqL);
2314
2315         if ((check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) == _TRUE) ||
2316             (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE) == _TRUE)) {
2317                 if (adapter->stapriv.asoc_sta_count == 2) {
2318                         _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2319                         ptarget_wlan = rtw_find_network(&pmlmepriv->scanned_queue, cur_network->network.MacAddress);
2320                         pmlmepriv->cur_network_scanned = ptarget_wlan;
2321                         if (ptarget_wlan)
2322                                 ptarget_wlan->fixed = _TRUE;
2323                         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2324                         /* a sta + bc/mc_stainfo (not Ibss_stainfo) */
2325                         rtw_indicate_connect(adapter);
2326                 }
2327         }
2328
2329         _exit_critical_bh(&pmlmepriv->lock, &irqL);
2330
2331
2332         mlmeext_sta_add_event_callback(adapter, psta);
2333
2334 #ifdef CONFIG_RTL8711
2335         /* submit SetStaKey_cmd to tell fw, fw will allocate an CAM entry for this sta   */
2336         rtw_setstakey_cmd(adapter, psta, GROUP_KEY, _TRUE);
2337 #endif
2338
2339 exit:
2340         return;
2341 }
2342
2343 #ifdef CONFIG_IEEE80211W
2344 void rtw_sta_timeout_event_callback(_adapter *adapter, u8 *pbuf)
2345 {
2346         _irqL irqL;
2347         struct sta_info *psta;
2348         struct stadel_event *pstadel = (struct stadel_event *)pbuf;
2349         struct sta_priv *pstapriv = &adapter->stapriv;
2350
2351
2352         psta = rtw_get_stainfo(&adapter->stapriv, pstadel->macaddr);
2353
2354         if (psta) {
2355                 u8 updated = _FALSE;
2356
2357                 _enter_critical_bh(&pstapriv->asoc_list_lock, &irqL);
2358                 if (rtw_is_list_empty(&psta->asoc_list) == _FALSE) {
2359                         rtw_list_delete(&psta->asoc_list);
2360                         pstapriv->asoc_list_cnt--;
2361                         updated = ap_free_sta(adapter, psta, _TRUE, WLAN_REASON_PREV_AUTH_NOT_VALID, _TRUE);
2362                 }
2363                 _exit_critical_bh(&pstapriv->asoc_list_lock, &irqL);
2364
2365                 associated_clients_update(adapter, updated, STA_INFO_UPDATE_ALL);
2366         }
2367
2368
2369
2370 }
2371 #endif /* CONFIG_IEEE80211W */
2372
2373 #ifdef CONFIG_RTW_80211R
2374 void rtw_update_ft_stainfo(_adapter *padapter, WLAN_BSSID_EX *pnetwork)
2375 {
2376         struct sta_priv         *pstapriv = &padapter->stapriv;
2377         struct sta_info         *psta = NULL;
2378
2379         psta = rtw_get_stainfo(pstapriv, pnetwork->MacAddress);
2380         if (psta == NULL)
2381                 psta = rtw_alloc_stainfo(pstapriv, pnetwork->MacAddress);
2382
2383         if (padapter->securitypriv.dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) {
2384
2385                 padapter->securitypriv.binstallGrpkey = _FALSE;
2386                 padapter->securitypriv.busetkipkey = _FALSE;
2387                 padapter->securitypriv.bgrpkey_handshake = _FALSE;
2388
2389                 psta->ieee8021x_blocked = _TRUE;
2390                 psta->dot118021XPrivacy = padapter->securitypriv.dot11PrivacyAlgrthm;
2391                 psta->dot11txpn.val = psta->dot11txpn.val + 1;
2392
2393                 _rtw_memset((u8 *)&psta->dot118021x_UncstKey, 0, sizeof(union Keytype));
2394                 _rtw_memset((u8 *)&psta->dot11tkiprxmickey, 0, sizeof(union Keytype));
2395                 _rtw_memset((u8 *)&psta->dot11tkiptxmickey, 0, sizeof(union Keytype));
2396                 _rtw_memset((u8 *)&psta->dot11txpn, 0, sizeof(union pn48));
2397 #ifdef CONFIG_IEEE80211W
2398                 _rtw_memset((u8 *)&psta->dot11wtxpn, 0, sizeof(union pn48));
2399 #endif
2400                 _rtw_memset((u8 *)&psta->dot11rxpn, 0, sizeof(union pn48));
2401         }
2402
2403 }
2404
2405 void rtw_ft_reassoc_event_callback(_adapter *padapter, u8 *pbuf)
2406 {
2407         struct mlme_priv                *pmlmepriv = &(padapter->mlmepriv);
2408         struct stassoc_event    *pstassoc = (struct stassoc_event *)pbuf;
2409         ft_priv *pftpriv = &pmlmepriv->ftpriv;
2410         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
2411         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
2412         WLAN_BSSID_EX           *pnetwork = (WLAN_BSSID_EX *)(&(pmlmeinfo->network));
2413         struct cfg80211_ft_event_params ft_evt_parms;
2414         _irqL irqL;
2415
2416         _rtw_memset(&ft_evt_parms, 0, sizeof(ft_evt_parms));
2417         rtw_update_ft_stainfo(padapter, pnetwork);
2418         ft_evt_parms.ies_len = pftpriv->ft_event.ies_len;
2419         ft_evt_parms.ies =  rtw_zmalloc(ft_evt_parms.ies_len);
2420         if (ft_evt_parms.ies)
2421                 _rtw_memcpy((void *)ft_evt_parms.ies, pftpriv->ft_event.ies, ft_evt_parms.ies_len);
2422          else
2423                 goto err_2;
2424
2425         ft_evt_parms.target_ap = rtw_zmalloc(ETH_ALEN);
2426         if (ft_evt_parms.target_ap)
2427                 _rtw_memcpy((void *)ft_evt_parms.target_ap, pstassoc->macaddr, ETH_ALEN);
2428         else
2429                 goto err_1;
2430
2431         ft_evt_parms.ric_ies = pftpriv->ft_event.ric_ies;
2432         ft_evt_parms.ric_ies_len = pftpriv->ft_event.ric_ies_len;
2433
2434         _enter_critical_bh(&pmlmepriv->lock, &irqL);
2435         rtw_set_ft_status(padapter, RTW_FT_AUTHENTICATED_STA);
2436         _exit_critical_bh(&pmlmepriv->lock, &irqL);
2437
2438         rtw_cfg80211_ft_event(padapter, &ft_evt_parms);
2439         RTW_INFO("%s: to "MAC_FMT"\n", __func__, MAC_ARG(ft_evt_parms.target_ap));
2440
2441         rtw_mfree((u8 *)pftpriv->ft_event.target_ap, ETH_ALEN);
2442 err_1:
2443         rtw_mfree((u8 *)ft_evt_parms.ies, ft_evt_parms.ies_len);
2444 err_2:
2445         return;
2446 }
2447 #endif
2448
2449 void rtw_sta_mstatus_disc_rpt(_adapter *adapter, u8 mac_id)
2450 {
2451         struct macid_ctl_t *macid_ctl = &adapter->dvobj->macid_ctl;
2452
2453         RTW_INFO("%s "ADPT_FMT" - mac_id=%d\n", __func__, ADPT_ARG(adapter), mac_id);
2454
2455         if (mac_id >= 0 && mac_id < macid_ctl->num) {
2456                 rtw_hal_set_FwMediaStatusRpt_single_cmd(adapter, 0, 0, 0, 0, mac_id);
2457                 /*
2458                  * For safety, prevent from keeping macid sleep.
2459                  * If we can sure all power mode enter/leave are paired,
2460                  * this check can be removed.
2461                  * Lucas@20131113
2462                  */
2463                 /* wakeup macid after disconnect. */
2464                 /*if (MLME_IS_STA(adapter))*/
2465                 rtw_hal_macid_wakeup(adapter, mac_id);
2466         } else {
2467                 RTW_PRINT(FUNC_ADPT_FMT" invalid macid:%u\n"
2468                           , FUNC_ADPT_ARG(adapter), mac_id);
2469                 rtw_warn_on(1);
2470         }
2471 }
2472 void rtw_sta_mstatus_report(_adapter *adapter)
2473 {
2474         struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
2475         struct wlan_network *tgt_network = &pmlmepriv->cur_network;
2476         struct sta_info *psta = NULL;
2477
2478         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE) &&
2479             check_fwstate(pmlmepriv, _FW_LINKED)) {
2480                 psta = rtw_get_stainfo(&adapter->stapriv, tgt_network->network.MacAddress);
2481                 if (psta)
2482                         rtw_sta_mstatus_disc_rpt(adapter, psta->mac_id);
2483                 else {
2484                         RTW_INFO("%s "ADPT_FMT" - mac_addr: "MAC_FMT" psta == NULL\n", __func__, ADPT_ARG(adapter), MAC_ARG(tgt_network->network.MacAddress));
2485                         rtw_warn_on(1);
2486                 }
2487         } else {
2488                 RTW_DBG("%s: fwstate: %08x\n", __func__, pmlmepriv->fw_state);
2489         }
2490 }
2491
2492 void rtw_stadel_event_callback(_adapter *adapter, u8 *pbuf)
2493 {
2494         _irqL irqL, irqL2;
2495
2496         struct sta_info *psta;
2497         struct wlan_network *pwlan = NULL;
2498         WLAN_BSSID_EX    *pdev_network = NULL;
2499         u8 *pibss = NULL;
2500         struct  mlme_priv       *pmlmepriv = &(adapter->mlmepriv);
2501         struct  stadel_event *pstadel   = (struct stadel_event *)pbuf;
2502         struct  sta_priv *pstapriv = &adapter->stapriv;
2503         struct wlan_network *tgt_network = &(pmlmepriv->cur_network);
2504         struct mlme_ext_priv    *pmlmeext = &adapter->mlmeextpriv;
2505         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
2506
2507
2508         RTW_INFO("%s(mac_id=%d)=" MAC_FMT "\n", __func__, pstadel->mac_id, MAC_ARG(pstadel->macaddr));
2509         rtw_sta_mstatus_disc_rpt(adapter, pstadel->mac_id);
2510
2511         psta = rtw_get_stainfo(&adapter->stapriv, pstadel->macaddr);
2512
2513         if (psta == NULL) {
2514                 RTW_INFO("%s(mac_id=%d)=" MAC_FMT " psta == NULL\n", __func__, pstadel->mac_id, MAC_ARG(pstadel->macaddr));
2515                 /*rtw_warn_on(1);*/
2516         }
2517
2518         if (psta)
2519                 rtw_wfd_st_switch(psta, 0);
2520
2521         /* if(check_fwstate(pmlmepriv, WIFI_AP_STATE)) */
2522         if ((pmlmeinfo->state & 0x03) == WIFI_FW_AP_STATE) {
2523 #ifdef CONFIG_IOCTL_CFG80211
2524 #ifdef COMPAT_KERNEL_RELEASE
2525
2526 #elif (LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 37)) || defined(CONFIG_CFG80211_FORCE_COMPATIBLE_2_6_37_UNDER)
2527                 rtw_cfg80211_indicate_sta_disassoc(adapter, pstadel->macaddr, *(u16 *)pstadel->rsvd);
2528 #endif /* (LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 37)) || defined(CONFIG_CFG80211_FORCE_COMPATIBLE_2_6_37_UNDER) */
2529 #endif /* CONFIG_IOCTL_CFG80211 */
2530
2531                 return;
2532         }
2533
2534         mlmeext_sta_del_event_callback(adapter);
2535
2536         _enter_critical_bh(&pmlmepriv->lock, &irqL2);
2537
2538         if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
2539                 u16 reason = *((unsigned short *)(pstadel->rsvd));
2540                 bool roam = _FALSE;
2541                 struct wlan_network *roam_target = NULL;
2542
2543 #ifdef CONFIG_LAYER2_ROAMING
2544 #ifdef CONFIG_RTW_80211R
2545                 if (reason == WLAN_REASON_EXPIRATION_CHK && rtw_chk_roam_flags(adapter, RTW_ROAM_ON_EXPIRED))
2546                         pmlmepriv->ftpriv.ft_roam_on_expired = _TRUE;
2547                 else
2548                         pmlmepriv->ftpriv.ft_roam_on_expired = _FALSE;
2549 #endif
2550                 if (adapter->registrypriv.wifi_spec == 1)
2551                         roam = _FALSE;
2552                 else if (reason == WLAN_REASON_EXPIRATION_CHK && rtw_chk_roam_flags(adapter, RTW_ROAM_ON_EXPIRED))
2553                         roam = _TRUE;
2554                 else if (reason == WLAN_REASON_ACTIVE_ROAM && rtw_chk_roam_flags(adapter, RTW_ROAM_ACTIVE)) {
2555                         roam = _TRUE;
2556                         roam_target = pmlmepriv->roam_network;
2557                 }
2558 #ifdef CONFIG_INTEL_WIDI
2559                 else if (adapter->mlmepriv.widi_state == INTEL_WIDI_STATE_CONNECTED)
2560                         roam = _TRUE;
2561 #endif /* CONFIG_INTEL_WIDI */
2562
2563                 if (roam == _TRUE) {
2564                         if (rtw_to_roam(adapter) > 0)
2565                                 rtw_dec_to_roam(adapter); /* this stadel_event is caused by roaming, decrease to_roam */
2566                         else if (rtw_to_roam(adapter) == 0)
2567                                 rtw_set_to_roam(adapter, adapter->registrypriv.max_roaming_times);
2568                 } else
2569                         rtw_set_to_roam(adapter, 0);
2570 #endif /* CONFIG_LAYER2_ROAMING */
2571
2572                 rtw_free_uc_swdec_pending_queue(adapter);
2573
2574                 rtw_free_assoc_resources(adapter, 1);
2575                 rtw_free_mlme_priv_ie_data(pmlmepriv);
2576
2577                 _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2578                 /* remove the network entry in scanned_queue */
2579                 pwlan = rtw_find_network(&pmlmepriv->scanned_queue, tgt_network->network.MacAddress);
2580                 if ((pwlan)  && (!check_fwstate(pmlmepriv, WIFI_UNDER_WPS))) {
2581                         pwlan->fixed = _FALSE;
2582                         rtw_free_network_nolock(adapter, pwlan);
2583                 }
2584                 _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2585
2586                 rtw_indicate_disconnect(adapter, *(u16 *)pstadel->rsvd, pstadel->locally_generated);
2587 #ifdef CONFIG_INTEL_WIDI
2588                 if (!rtw_to_roam(adapter))
2589                         process_intel_widi_disconnect(adapter, 1);
2590 #endif /* CONFIG_INTEL_WIDI */
2591
2592                 _rtw_roaming(adapter, roam_target);
2593         }
2594
2595         if (check_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE) ||
2596             check_fwstate(pmlmepriv, WIFI_ADHOC_STATE)) {
2597
2598                 /* _enter_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
2599                 rtw_free_stainfo(adapter,  psta);
2600                 /* _exit_critical_bh(&(pstapriv->sta_hash_lock), &irqL); */
2601
2602                 if (adapter->stapriv.asoc_sta_count == 1) { /* a sta + bc/mc_stainfo (not Ibss_stainfo) */
2603                         /* rtw_indicate_disconnect(adapter); */ /* removed@20091105 */
2604                         _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2605                         /* free old ibss network */
2606                         /* pwlan = rtw_find_network(&pmlmepriv->scanned_queue, pstadel->macaddr); */
2607                         pwlan = rtw_find_network(&pmlmepriv->scanned_queue, tgt_network->network.MacAddress);
2608                         if (pwlan) {
2609                                 pwlan->fixed = _FALSE;
2610                                 rtw_free_network_nolock(adapter, pwlan);
2611                         }
2612                         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
2613                         /* re-create ibss */
2614                         pdev_network = &(adapter->registrypriv.dev_network);
2615                         pibss = adapter->registrypriv.dev_network.MacAddress;
2616
2617                         _rtw_memcpy(pdev_network, &tgt_network->network, get_WLAN_BSSID_EX_sz(&tgt_network->network));
2618
2619                         _rtw_memset(&pdev_network->Ssid, 0, sizeof(NDIS_802_11_SSID));
2620                         _rtw_memcpy(&pdev_network->Ssid, &pmlmepriv->assoc_ssid, sizeof(NDIS_802_11_SSID));
2621
2622                         rtw_update_registrypriv_dev_network(adapter);
2623
2624                         rtw_generate_random_ibss(pibss);
2625
2626                         if (check_fwstate(pmlmepriv, WIFI_ADHOC_STATE)) {
2627                                 set_fwstate(pmlmepriv, WIFI_ADHOC_MASTER_STATE);
2628                                 _clr_fwstate_(pmlmepriv, WIFI_ADHOC_STATE);
2629                         }
2630
2631                         if (rtw_create_ibss_cmd(adapter, 0) != _SUCCESS)
2632                                 RTW_ERR("rtw_create_ibss_cmd FAIL\n");
2633
2634                 }
2635
2636         }
2637
2638         _exit_critical_bh(&pmlmepriv->lock, &irqL2);
2639
2640
2641 }
2642
2643
2644 void rtw_cpwm_event_callback(PADAPTER padapter, u8 *pbuf)
2645 {
2646 #ifdef CONFIG_LPS_LCLK
2647         struct reportpwrstate_parm *preportpwrstate;
2648 #endif
2649
2650
2651 #ifdef CONFIG_LPS_LCLK
2652         preportpwrstate = (struct reportpwrstate_parm *)pbuf;
2653         preportpwrstate->state |= (u8)(adapter_to_pwrctl(padapter)->cpwm_tog + 0x80);
2654         cpwm_int_hdl(padapter, preportpwrstate);
2655 #endif
2656
2657
2658 }
2659
2660
2661 void rtw_wmm_event_callback(PADAPTER padapter, u8 *pbuf)
2662 {
2663
2664         WMMOnAssocRsp(padapter);
2665
2666
2667 }
2668
2669 /*
2670 * _rtw_join_timeout_handler - Timeout/faliure handler for CMD JoinBss
2671 * @adapter: pointer to _adapter structure
2672 */
2673 void _rtw_join_timeout_handler(_adapter *adapter)
2674 {
2675         _irqL irqL;
2676         struct  mlme_priv *pmlmepriv = &adapter->mlmepriv;
2677
2678 #if 0
2679         if (rtw_is_drv_stopped(adapter)) {
2680                 _rtw_up_sema(&pmlmepriv->assoc_terminate);
2681                 return;
2682         }
2683 #endif
2684
2685
2686
2687         RTW_INFO("%s, fw_state=%x\n", __FUNCTION__, get_fwstate(pmlmepriv));
2688
2689         if (RTW_CANNOT_RUN(adapter))
2690                 return;
2691
2692
2693         _enter_critical_bh(&pmlmepriv->lock, &irqL);
2694
2695 #ifdef CONFIG_LAYER2_ROAMING
2696         if (rtw_to_roam(adapter) > 0) { /* join timeout caused by roaming */
2697                 while (1) {
2698                         rtw_dec_to_roam(adapter);
2699                         if (rtw_to_roam(adapter) != 0) { /* try another */
2700                                 int do_join_r;
2701                                 RTW_INFO("%s try another roaming\n", __FUNCTION__);
2702                                 do_join_r = rtw_do_join(adapter);
2703                                 if (_SUCCESS != do_join_r) {
2704                                         RTW_INFO("%s roaming do_join return %d\n", __FUNCTION__ , do_join_r);
2705                                         continue;
2706                                 }
2707                                 break;
2708                         } else {
2709 #ifdef CONFIG_INTEL_WIDI
2710                                 if (adapter->mlmepriv.widi_state == INTEL_WIDI_STATE_ROAMING) {
2711                                         _rtw_memset(pmlmepriv->sa_ext, 0x00, L2SDTA_SERVICE_VE_LEN);
2712                                         intel_widi_wk_cmd(adapter, INTEL_WIDI_LISTEN_WK, NULL, 0);
2713                                         RTW_INFO("change to widi listen\n");
2714                                 }
2715 #endif /* CONFIG_INTEL_WIDI */
2716                                 RTW_INFO("%s We've try roaming but fail\n", __FUNCTION__);
2717 #ifdef CONFIG_RTW_80211R
2718                                 rtw_clr_ft_flags(adapter, RTW_FT_SUPPORTED|RTW_FT_OVER_DS_SUPPORTED);
2719                                 rtw_reset_ft_status(adapter);
2720 #endif
2721                                 rtw_indicate_disconnect(adapter, 0, _FALSE);
2722                                 break;
2723                         }
2724                 }
2725
2726         } else
2727 #endif
2728         {
2729                 rtw_indicate_disconnect(adapter, 0, _FALSE);
2730                 free_scanqueue(pmlmepriv);/* ??? */
2731
2732 #ifdef CONFIG_IOCTL_CFG80211
2733                 /* indicate disconnect for the case that join_timeout and check_fwstate != FW_LINKED */
2734                 rtw_cfg80211_indicate_disconnect(adapter, 0, _FALSE);
2735 #endif /* CONFIG_IOCTL_CFG80211 */
2736
2737         }
2738
2739         _exit_critical_bh(&pmlmepriv->lock, &irqL);
2740
2741
2742 #ifdef CONFIG_DRVEXT_MODULE_WSC
2743         drvext_assoc_fail_indicate(&adapter->drvextpriv);
2744 #endif
2745
2746
2747
2748 }
2749
2750 /*
2751 * rtw_scan_timeout_handler - Timeout/Faliure handler for CMD SiteSurvey
2752 * @adapter: pointer to _adapter structure
2753 */
2754 void rtw_scan_timeout_handler(_adapter *adapter)
2755 {
2756         _irqL irqL;
2757         struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
2758         RTW_INFO(FUNC_ADPT_FMT" fw_state=%x\n", FUNC_ADPT_ARG(adapter), get_fwstate(pmlmepriv));
2759
2760         _enter_critical_bh(&pmlmepriv->lock, &irqL);
2761
2762         _clr_fwstate_(pmlmepriv, _FW_UNDER_SURVEY);
2763
2764         _exit_critical_bh(&pmlmepriv->lock, &irqL);
2765
2766 #ifdef CONFIG_IOCTL_CFG80211
2767         rtw_cfg80211_surveydone_event_callback(adapter);
2768 #endif /* CONFIG_IOCTL_CFG80211 */
2769
2770         rtw_indicate_scan_done(adapter, _TRUE);
2771
2772 #if defined(CONFIG_CONCURRENT_MODE) && defined(CONFIG_IOCTL_CFG80211)
2773         rtw_cfg80211_indicate_scan_done_for_buddy(adapter, _TRUE);
2774 #endif
2775 }
2776
2777 void rtw_mlme_reset_auto_scan_int(_adapter *adapter, u8 *reason)
2778 {
2779         struct mlme_priv *mlme = &adapter->mlmepriv;
2780         struct mlme_ext_priv    *pmlmeext = &adapter->mlmeextpriv;
2781         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
2782         u8 u_ch;
2783         u32 interval_ms = 0xffffffff; /* 0xffffffff: special value to make min() works well, also means no auto scan */
2784
2785         *reason = RTW_AUTO_SCAN_REASON_UNSPECIFIED;
2786         rtw_mi_get_ch_setting_union(adapter, &u_ch, NULL, NULL);
2787
2788         if (hal_chk_bw_cap(adapter, BW_CAP_40M)
2789             && is_client_associated_to_ap(adapter) == _TRUE
2790             && u_ch >= 1 && u_ch <= 14
2791             && adapter->registrypriv.wifi_spec
2792             /* TODO: AP Connected is 40MHz capability? */
2793            ) {
2794                 interval_ms = rtw_min(interval_ms, 60 * 1000);
2795                 *reason |= RTW_AUTO_SCAN_REASON_2040_BSS;
2796         }
2797
2798 exit:
2799         if (interval_ms == 0xffffffff)
2800                 interval_ms = 0;
2801
2802         rtw_mlme_set_auto_scan_int(adapter, interval_ms);
2803         return;
2804 }
2805
2806 void rtw_drv_scan_by_self(_adapter *padapter, u8 reason)
2807 {
2808         struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
2809         struct rtw_ieee80211_channel ch_for_2040_bss[14] = {
2810                 {1, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2811                 {2, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2812                 {3, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2813                 {4, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2814                 {5, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2815                 {6, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2816                 {7, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2817                 {8, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2818                 {9, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2819                 {10, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2820                 {11, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2821                 {12, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2822                 {13, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2823                 {14, RTW_IEEE80211_CHAN_PASSIVE_SCAN},
2824         };
2825         struct rtw_ieee80211_channel *ch_sel = NULL;
2826         int ch_num = 0;
2827
2828         if (rtw_is_scan_deny(padapter))
2829                 goto exit;
2830
2831         if (!rtw_is_adapter_up(padapter))
2832                 goto exit;
2833
2834         if (rtw_mi_busy_traffic_check(padapter, _FALSE)) {
2835 #ifdef CONFIG_LAYER2_ROAMING
2836                 if (rtw_chk_roam_flags(padapter, RTW_ROAM_ACTIVE) && pmlmepriv->need_to_roam == _TRUE) {
2837                         RTW_INFO("need to roam, don't care BusyTraffic\n");
2838                 } else
2839 #endif
2840                 {
2841                         RTW_INFO(FUNC_ADPT_FMT" exit BusyTraffic\n", FUNC_ADPT_ARG(padapter));
2842                         goto exit;
2843                 }
2844         }
2845         if (check_fwstate(pmlmepriv, WIFI_AP_STATE) && check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
2846                 RTW_INFO(FUNC_ADPT_FMT" WIFI_AP_STATE && WIFI_UNDER_WPS\n", FUNC_ADPT_ARG(padapter));
2847                 goto exit;
2848         }
2849         if (check_fwstate(pmlmepriv, (_FW_UNDER_SURVEY | _FW_UNDER_LINKING)) == _TRUE) {
2850                 RTW_INFO(FUNC_ADPT_FMT" _FW_UNDER_SURVEY|_FW_UNDER_LINKING\n", FUNC_ADPT_ARG(padapter));
2851                 goto exit;
2852         }
2853
2854 #ifdef CONFIG_CONCURRENT_MODE
2855         if (rtw_mi_buddy_check_fwstate(padapter, (_FW_UNDER_SURVEY | _FW_UNDER_LINKING | WIFI_UNDER_WPS))) {
2856                 RTW_INFO(FUNC_ADPT_FMT", but buddy_intf is under scanning or linking or wps_phase\n", FUNC_ADPT_ARG(padapter));
2857                 goto exit;
2858         }
2859 #endif
2860
2861         RTW_INFO(FUNC_ADPT_FMT" reason:0x%02x\n", FUNC_ADPT_ARG(padapter), reason);
2862
2863         /* only for 20/40 BSS */
2864         if (reason == RTW_AUTO_SCAN_REASON_2040_BSS) {
2865                 ch_sel = ch_for_2040_bss;
2866                 ch_num = 14;
2867         }
2868
2869         rtw_set_802_11_bssid_list_scan(padapter, NULL, 0, ch_sel, ch_num);
2870 exit:
2871         return;
2872 }
2873
2874 static void rtw_auto_scan_handler(_adapter *padapter)
2875 {
2876         struct mlme_priv *pmlmepriv = &padapter->mlmepriv;
2877         u8 reason = RTW_AUTO_SCAN_REASON_UNSPECIFIED;
2878
2879         rtw_mlme_reset_auto_scan_int(padapter, &reason);
2880
2881 #ifdef CONFIG_P2P
2882         if (!rtw_p2p_chk_state(&padapter->wdinfo, P2P_STATE_NONE))
2883                 goto exit;
2884 #endif
2885
2886 #ifdef CONFIG_TDLS
2887         if (padapter->tdlsinfo.link_established == _TRUE)
2888                 goto exit;
2889 #endif
2890
2891         if (pmlmepriv->auto_scan_int_ms == 0
2892             || rtw_get_passing_time_ms(pmlmepriv->scan_start_time) < pmlmepriv->auto_scan_int_ms)
2893                 goto exit;
2894
2895         rtw_drv_scan_by_self(padapter, reason);
2896
2897 exit:
2898         return;
2899 }
2900 static u8 is_drv_in_lps(_adapter *adapter)
2901 {
2902         u8 is_in_lps = _FALSE;
2903
2904         #ifdef CONFIG_LPS_LCLK_WD_TIMER /* to avoid leaving lps 32k frequently*/
2905         if ((adapter_to_pwrctl(adapter)->bFwCurrentInPSMode == _TRUE)
2906         #ifdef CONFIG_BT_COEXIST
2907                 && (rtw_btcoex_IsBtControlLps(adapter) == _FALSE)
2908         #endif
2909                 )
2910                 is_in_lps = _TRUE;
2911         #endif /* CONFIG_LPS_LCLK_WD_TIMER*/
2912         return is_in_lps;
2913 }
2914 void rtw_iface_dynamic_check_timer_handlder(_adapter *adapter)
2915 {
2916 #ifdef CONFIG_AP_MODE
2917         struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
2918 #endif /* CONFIG_AP_MODE */
2919
2920         if (adapter->net_closed == _TRUE)
2921                 return;
2922         #ifdef CONFIG_LPS_LCLK_WD_TIMER /* to avoid leaving lps 32k frequently*/
2923         if (is_drv_in_lps(adapter)) {
2924                 u8 bEnterPS;
2925
2926                 linked_status_chk(adapter, 1);
2927
2928                 bEnterPS = traffic_status_watchdog(adapter, 1);
2929                 if (bEnterPS) {
2930                         /* rtw_lps_ctrl_wk_cmd(adapter, LPS_CTRL_ENTER, 1); */
2931                         rtw_hal_dm_watchdog_in_lps(adapter);
2932                 } else {
2933                         /* call rtw_lps_ctrl_wk_cmd(padapter, LPS_CTRL_LEAVE, 1) in traffic_status_watchdog() */
2934                 }
2935         }
2936         #endif /* CONFIG_LPS_LCLK_WD_TIMER      */
2937
2938         /* auto site survey */
2939         rtw_auto_scan_handler(adapter);
2940
2941 #ifndef CONFIG_ACTIVE_KEEP_ALIVE_CHECK
2942 #ifdef CONFIG_AP_MODE
2943         if (check_fwstate(pmlmepriv, WIFI_AP_STATE) == _TRUE)
2944                 expire_timeout_chk(adapter);
2945 #endif
2946 #endif /* !CONFIG_ACTIVE_KEEP_ALIVE_CHECK */
2947
2948 #ifdef CONFIG_BR_EXT
2949
2950 #if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35))
2951         rcu_read_lock();
2952 #endif /* (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35)) */
2953
2954 #if (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35))
2955         if (adapter->pnetdev->br_port
2956 #else   /* (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35)) */
2957         if (rcu_dereference(adapter->pnetdev->rx_handler_data)
2958 #endif /* (LINUX_VERSION_CODE <= KERNEL_VERSION(2, 6, 35)) */
2959                 && (check_fwstate(pmlmepriv, WIFI_STATION_STATE | WIFI_ADHOC_STATE) == _TRUE)) {
2960                 /* expire NAT2.5 entry */
2961                 void nat25_db_expire(_adapter *priv);
2962                 nat25_db_expire(adapter);
2963
2964                 if (adapter->pppoe_connection_in_progress > 0)
2965                         adapter->pppoe_connection_in_progress--;
2966                 /* due to rtw_dynamic_check_timer_handlder() is called every 2 seconds */
2967                 if (adapter->pppoe_connection_in_progress > 0)
2968                         adapter->pppoe_connection_in_progress--;
2969         }
2970
2971 #if (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35))
2972         rcu_read_unlock();
2973 #endif /* (LINUX_VERSION_CODE > KERNEL_VERSION(2, 6, 35)) */
2974
2975 #endif /* CONFIG_BR_EXT */
2976
2977 }
2978
2979 /*#define DBG_TRAFFIC_STATISTIC*/
2980 static void collect_traffic_statistics(_adapter *padapter)
2981 {
2982         struct dvobj_priv       *pdvobjpriv = adapter_to_dvobj(padapter);
2983
2984         /*_rtw_memset(&pdvobjpriv->traffic_stat, 0, sizeof(struct rtw_traffic_statistics));*/
2985
2986         /* Tx bytes reset*/
2987         pdvobjpriv->traffic_stat.tx_bytes = 0;
2988         pdvobjpriv->traffic_stat.tx_pkts = 0;
2989         pdvobjpriv->traffic_stat.tx_drop = 0;
2990
2991         /* Rx bytes reset*/
2992         pdvobjpriv->traffic_stat.rx_bytes = 0;
2993         pdvobjpriv->traffic_stat.rx_pkts = 0;
2994         pdvobjpriv->traffic_stat.rx_drop = 0;
2995
2996         rtw_mi_traffic_statistics(padapter);
2997
2998         /* Calculate throughput in last interval */
2999         pdvobjpriv->traffic_stat.cur_tx_bytes = pdvobjpriv->traffic_stat.tx_bytes - pdvobjpriv->traffic_stat.last_tx_bytes;
3000         pdvobjpriv->traffic_stat.cur_rx_bytes = pdvobjpriv->traffic_stat.rx_bytes - pdvobjpriv->traffic_stat.last_rx_bytes;
3001         pdvobjpriv->traffic_stat.last_tx_bytes = pdvobjpriv->traffic_stat.tx_bytes;
3002         pdvobjpriv->traffic_stat.last_rx_bytes = pdvobjpriv->traffic_stat.rx_bytes;
3003
3004         pdvobjpriv->traffic_stat.cur_tx_tp = (u32)(pdvobjpriv->traffic_stat.cur_tx_bytes * 8 / 2 / 1024 / 1024);
3005         pdvobjpriv->traffic_stat.cur_rx_tp = (u32)(pdvobjpriv->traffic_stat.cur_rx_bytes * 8 / 2 / 1024 / 1024);
3006
3007         #ifdef DBG_TRAFFIC_STATISTIC
3008         RTW_INFO("\n========================\n");
3009         RTW_INFO("cur_tx_bytes:%lld\n", pdvobjpriv->traffic_stat.cur_tx_bytes);
3010         RTW_INFO("cur_rx_bytes:%lld\n", pdvobjpriv->traffic_stat.cur_rx_bytes);
3011
3012         RTW_INFO("last_tx_bytes:%lld\n", pdvobjpriv->traffic_stat.last_tx_bytes);
3013         RTW_INFO("last_rx_bytes:%lld\n", pdvobjpriv->traffic_stat.last_rx_bytes);
3014
3015         RTW_INFO("cur_tx_tp:%d\n", pdvobjpriv->traffic_stat.cur_tx_tp);
3016         RTW_INFO("cur_rx_tp:%d\n", pdvobjpriv->traffic_stat.cur_rx_tp);
3017         #endif
3018 }
3019
3020 void rtw_dynamic_check_timer_handlder(_adapter *adapter)
3021 {
3022         if (!adapter)
3023                 return;
3024
3025         if (!rtw_is_hw_init_completed(adapter))
3026                 return;
3027
3028         if (RTW_CANNOT_RUN(adapter))
3029                 return;
3030
3031         collect_traffic_statistics(adapter);
3032
3033         rtw_mi_dynamic_check_timer_handlder(adapter);
3034
3035         if (!is_drv_in_lps(adapter))
3036                 rtw_dynamic_chk_wk_cmd(adapter);
3037 }
3038
3039
3040 #ifdef CONFIG_SET_SCAN_DENY_TIMER
3041 inline bool rtw_is_scan_deny(_adapter *adapter)
3042 {
3043         struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3044         return (ATOMIC_READ(&mlmepriv->set_scan_deny) != 0) ? _TRUE : _FALSE;
3045 }
3046
3047 inline void rtw_clear_scan_deny(_adapter *adapter)
3048 {
3049         struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3050         ATOMIC_SET(&mlmepriv->set_scan_deny, 0);
3051         if (0)
3052                 RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(adapter));
3053 }
3054
3055 void rtw_set_scan_deny_timer_hdl(_adapter *adapter)
3056 {
3057         rtw_clear_scan_deny(adapter);
3058 }
3059 void rtw_set_scan_deny(_adapter *adapter, u32 ms)
3060 {
3061         struct mlme_priv *mlmepriv = &adapter->mlmepriv;
3062         if (0)
3063                 RTW_INFO(FUNC_ADPT_FMT"\n", FUNC_ADPT_ARG(adapter));
3064         ATOMIC_SET(&mlmepriv->set_scan_deny, 1);
3065         _set_timer(&mlmepriv->set_scan_deny_timer, ms);
3066 }
3067 #endif
3068
3069 #ifdef CONFIG_LAYER2_ROAMING
3070 /*
3071 * Select a new roaming candidate from the original @param candidate and @param competitor
3072 * @return _TRUE: candidate is updated
3073 * @return _FALSE: candidate is not updated
3074 */
3075 static int rtw_check_roaming_candidate(struct mlme_priv *mlme
3076         , struct wlan_network **candidate, struct wlan_network *competitor)
3077 {
3078         int updated = _FALSE;
3079         _adapter *adapter = container_of(mlme, _adapter, mlmepriv);
3080 #ifdef CONFIG_RTW_80211R
3081         ft_priv *pftpriv = &mlme->ftpriv;
3082         u32 mdie_len = 0;
3083         u8 *ptmp = NULL;
3084 #endif
3085
3086
3087         if (is_same_ess(&competitor->network, &mlme->cur_network.network) == _FALSE)
3088                 goto exit;
3089
3090         if (rtw_is_desired_network(adapter, competitor) == _FALSE)
3091                 goto exit;
3092
3093 #ifdef CONFIG_LAYER2_ROAMING
3094         if (mlme->need_to_roam == _FALSE)
3095                 goto exit;
3096 #endif
3097
3098 #ifdef CONFIG_RTW_80211R
3099         if (rtw_chk_ft_flags(adapter, RTW_FT_SUPPORTED)) {
3100                 ptmp = rtw_get_ie(&competitor->network.IEs[12], _MDIE_, &mdie_len, competitor->network.IELength-12);
3101                 if (ptmp) {
3102                         if (!_rtw_memcmp(&pftpriv->mdid, ptmp+2, 2))
3103                                 goto exit;
3104
3105                         /*The candidate don't support over-the-DS*/
3106                         if (rtw_chk_ft_flags(adapter, RTW_FT_STA_OVER_DS_SUPPORTED)) {
3107                                 if ((rtw_chk_ft_flags(adapter, RTW_FT_OVER_DS_SUPPORTED) && !(*(ptmp+4) & 0x01)) ||
3108                                         (!rtw_chk_ft_flags(adapter, RTW_FT_OVER_DS_SUPPORTED) && (*(ptmp+4) & 0x01))) {
3109                                         RTW_INFO("FT: ignore the candidate(" MAC_FMT ") for over-the-DS\n", MAC_ARG(competitor->network.MacAddress));
3110                                         rtw_clr_ft_flags(adapter, RTW_FT_OVER_DS_SUPPORTED);
3111                                         goto exit;
3112                                 }
3113                         }
3114                 } else
3115                         goto exit;
3116         }
3117 #endif
3118
3119         RTW_INFO("roam candidate:%s %s("MAC_FMT", ch%3u) rssi:%d, age:%5d\n",
3120                  (competitor == mlme->cur_network_scanned) ? "*" : " " ,
3121                  competitor->network.Ssid.Ssid,
3122                  MAC_ARG(competitor->network.MacAddress),
3123                  competitor->network.Configuration.DSConfig,
3124                  (int)competitor->network.Rssi,
3125                  rtw_get_passing_time_ms(competitor->last_scanned)
3126                 );
3127
3128         /* got specific addr to roam */
3129         if (!is_zero_mac_addr(mlme->roam_tgt_addr)) {
3130                 if (_rtw_memcmp(mlme->roam_tgt_addr, competitor->network.MacAddress, ETH_ALEN) == _TRUE)
3131                         goto update;
3132                 else
3133                         goto exit;
3134         }
3135 #if 1
3136         if (rtw_get_passing_time_ms((u32)competitor->last_scanned) >= mlme->roam_scanr_exp_ms)
3137                 goto exit;
3138
3139         if (competitor->network.Rssi - mlme->cur_network_scanned->network.Rssi < mlme->roam_rssi_diff_th)
3140                 goto exit;
3141
3142         if (*candidate != NULL && (*candidate)->network.Rssi >= competitor->network.Rssi)
3143                 goto exit;
3144 #else
3145         goto exit;
3146 #endif
3147
3148 update:
3149         *candidate = competitor;
3150         updated = _TRUE;
3151
3152 exit:
3153         return updated;
3154 }
3155
3156 int rtw_select_roaming_candidate(struct mlme_priv *mlme)
3157 {
3158         _irqL   irqL;
3159         int ret = _FAIL;
3160         _list   *phead;
3161         _adapter *adapter;
3162         _queue  *queue  = &(mlme->scanned_queue);
3163         struct  wlan_network    *pnetwork = NULL;
3164         struct  wlan_network    *candidate = NULL;
3165         u8              bSupportAntDiv = _FALSE;
3166
3167
3168         if (mlme->cur_network_scanned == NULL) {
3169                 rtw_warn_on(1);
3170                 goto exit;
3171         }
3172
3173         _enter_critical_bh(&(mlme->scanned_queue.lock), &irqL);
3174         phead = get_list_head(queue);
3175         adapter = (_adapter *)mlme->nic_hdl;
3176
3177         mlme->pscanned = get_next(phead);
3178
3179         while (!rtw_end_of_queue_search(phead, mlme->pscanned)) {
3180
3181                 pnetwork = LIST_CONTAINOR(mlme->pscanned, struct wlan_network, list);
3182                 if (pnetwork == NULL) {
3183                         ret = _FAIL;
3184                         goto exit;
3185                 }
3186
3187                 mlme->pscanned = get_next(mlme->pscanned);
3188
3189                 if (0)
3190                         RTW_INFO("%s("MAC_FMT", ch%u) rssi:%d\n"
3191                                  , pnetwork->network.Ssid.Ssid
3192                                  , MAC_ARG(pnetwork->network.MacAddress)
3193                                  , pnetwork->network.Configuration.DSConfig
3194                                  , (int)pnetwork->network.Rssi);
3195
3196                 rtw_check_roaming_candidate(mlme, &candidate, pnetwork);
3197
3198         }
3199
3200         if (candidate == NULL) {
3201                 RTW_INFO("%s: return _FAIL(candidate == NULL)\n", __FUNCTION__);
3202                 ret = _FAIL;
3203                 goto exit;
3204         } else {
3205                 RTW_INFO("%s: candidate: %s("MAC_FMT", ch:%u)\n", __FUNCTION__,
3206                         candidate->network.Ssid.Ssid, MAC_ARG(candidate->network.MacAddress),
3207                          candidate->network.Configuration.DSConfig);
3208
3209                 mlme->roam_network = candidate;
3210
3211                 if (_rtw_memcmp(candidate->network.MacAddress, mlme->roam_tgt_addr, ETH_ALEN) == _TRUE)
3212                         _rtw_memset(mlme->roam_tgt_addr, 0, ETH_ALEN);
3213         }
3214
3215         ret = _SUCCESS;
3216 exit:
3217         _exit_critical_bh(&(mlme->scanned_queue.lock), &irqL);
3218
3219         return ret;
3220 }
3221 #endif /* CONFIG_LAYER2_ROAMING */
3222
3223 /*
3224 * Select a new join candidate from the original @param candidate and @param competitor
3225 * @return _TRUE: candidate is updated
3226 * @return _FALSE: candidate is not updated
3227 */
3228 static int rtw_check_join_candidate(struct mlme_priv *mlme
3229             , struct wlan_network **candidate, struct wlan_network *competitor)
3230 {
3231         int updated = _FALSE;
3232         _adapter *adapter = container_of(mlme, _adapter, mlmepriv);
3233
3234
3235         /* check bssid, if needed */
3236         if (mlme->assoc_by_bssid == _TRUE) {
3237                 if (_rtw_memcmp(competitor->network.MacAddress, mlme->assoc_bssid, ETH_ALEN) == _FALSE)
3238                         goto exit;
3239         }
3240
3241         /* check ssid, if needed */
3242         if (mlme->assoc_ssid.Ssid[0] && mlme->assoc_ssid.SsidLength) {
3243                 if (competitor->network.Ssid.SsidLength != mlme->assoc_ssid.SsidLength
3244                     || _rtw_memcmp(competitor->network.Ssid.Ssid, mlme->assoc_ssid.Ssid, mlme->assoc_ssid.SsidLength) == _FALSE
3245                    )
3246                         goto exit;
3247         }
3248
3249         if (rtw_is_desired_network(adapter, competitor)  == _FALSE)
3250                 goto exit;
3251
3252 #ifdef CONFIG_LAYER2_ROAMING
3253         if (rtw_to_roam(adapter) > 0) {
3254                 if (rtw_get_passing_time_ms((u32)competitor->last_scanned) >= mlme->roam_scanr_exp_ms
3255                     || is_same_ess(&competitor->network, &mlme->cur_network.network) == _FALSE
3256                    )
3257                         goto exit;
3258         }
3259 #endif
3260
3261         if (*candidate == NULL || (*candidate)->network.Rssi < competitor->network.Rssi) {
3262                 *candidate = competitor;
3263                 updated = _TRUE;
3264         }
3265
3266         if (updated) {
3267                 RTW_INFO("[by_bssid:%u][assoc_ssid:%s][to_roam:%u] "
3268                          "new candidate: %s("MAC_FMT", ch%u) rssi:%d\n",
3269                          mlme->assoc_by_bssid,
3270                          mlme->assoc_ssid.Ssid,
3271                          rtw_to_roam(adapter),
3272                          (*candidate)->network.Ssid.Ssid,
3273                          MAC_ARG((*candidate)->network.MacAddress),
3274                          (*candidate)->network.Configuration.DSConfig,
3275                          (int)(*candidate)->network.Rssi
3276                         );
3277         }
3278
3279 exit:
3280         return updated;
3281 }
3282
3283 /*
3284 Calling context:
3285 The caller of the sub-routine will be in critical section...
3286
3287 The caller must hold the following spinlock
3288
3289 pmlmepriv->lock
3290
3291
3292 */
3293
3294 int rtw_select_and_join_from_scanned_queue(struct mlme_priv *pmlmepriv)
3295 {
3296         _irqL   irqL;
3297         int ret;
3298         _list   *phead;
3299         _adapter *adapter;
3300         _queue  *queue  = &(pmlmepriv->scanned_queue);
3301         struct  wlan_network    *pnetwork = NULL;
3302         struct  wlan_network    *candidate = NULL;
3303         u8              bSupportAntDiv = _FALSE;
3304
3305
3306         adapter = (_adapter *)pmlmepriv->nic_hdl;
3307
3308         _enter_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3309
3310 #ifdef CONFIG_LAYER2_ROAMING
3311         if (pmlmepriv->roam_network) {
3312                 candidate = pmlmepriv->roam_network;
3313                 pmlmepriv->roam_network = NULL;
3314                 goto candidate_exist;
3315         }
3316 #endif
3317
3318         phead = get_list_head(queue);
3319         pmlmepriv->pscanned = get_next(phead);
3320
3321         while (!rtw_end_of_queue_search(phead, pmlmepriv->pscanned)) {
3322
3323                 pnetwork = LIST_CONTAINOR(pmlmepriv->pscanned, struct wlan_network, list);
3324                 if (pnetwork == NULL) {
3325                         ret = _FAIL;
3326                         goto exit;
3327                 }
3328
3329                 pmlmepriv->pscanned = get_next(pmlmepriv->pscanned);
3330
3331                 if (0)
3332                         RTW_INFO("%s("MAC_FMT", ch%u) rssi:%d\n"
3333                                  , pnetwork->network.Ssid.Ssid
3334                                  , MAC_ARG(pnetwork->network.MacAddress)
3335                                  , pnetwork->network.Configuration.DSConfig
3336                                  , (int)pnetwork->network.Rssi);
3337
3338                 rtw_check_join_candidate(pmlmepriv, &candidate, pnetwork);
3339
3340         }
3341
3342         if (candidate == NULL) {
3343                 RTW_INFO("%s: return _FAIL(candidate == NULL)\n", __FUNCTION__);
3344 #ifdef CONFIG_WOWLAN
3345                 _clr_fwstate_(pmlmepriv, _FW_LINKED | _FW_UNDER_LINKING);
3346 #endif
3347                 ret = _FAIL;
3348                 goto exit;
3349         } else {
3350                 RTW_INFO("%s: candidate: %s("MAC_FMT", ch:%u)\n", __FUNCTION__,
3351                         candidate->network.Ssid.Ssid, MAC_ARG(candidate->network.MacAddress),
3352                          candidate->network.Configuration.DSConfig);
3353                 goto candidate_exist;
3354         }
3355
3356 candidate_exist:
3357
3358         /* check for situation of  _FW_LINKED */
3359         if (check_fwstate(pmlmepriv, _FW_LINKED) == _TRUE) {
3360                 RTW_INFO("%s: _FW_LINKED while ask_for_joinbss!!!\n", __FUNCTION__);
3361
3362 #if 0 /* for WPA/WPA2 authentication, wpa_supplicant will expect authentication from AP, it is needed to reconnect AP... */
3363                 if (is_same_network(&pmlmepriv->cur_network.network, &candidate->network)) {
3364                         RTW_INFO("%s: _FW_LINKED and is same network, it needn't join again\n", __FUNCTION__);
3365
3366                         rtw_indicate_connect(adapter);/* rtw_indicate_connect again */
3367
3368                         ret = 2;
3369                         goto exit;
3370                 } else
3371 #endif
3372                 {
3373                         rtw_disassoc_cmd(adapter, 0, _TRUE);
3374                         rtw_indicate_disconnect(adapter, 0, _FALSE);
3375                         rtw_free_assoc_resources(adapter, 0);
3376                 }
3377         }
3378
3379 #ifdef CONFIG_ANTENNA_DIVERSITY
3380         rtw_hal_get_def_var(adapter, HAL_DEF_IS_SUPPORT_ANT_DIV, &(bSupportAntDiv));
3381         if (_TRUE == bSupportAntDiv) {
3382                 u8 CurrentAntenna;
3383                 rtw_hal_get_odm_var(adapter, HAL_ODM_ANTDIV_SELECT, &(CurrentAntenna), NULL);
3384                 RTW_INFO("#### Opt_Ant_(%s) , cur_Ant(%s)\n",
3385                         (MAIN_ANT == candidate->network.PhyInfo.Optimum_antenna) ? "MAIN_ANT" : "AUX_ANT",
3386                          (MAIN_ANT == CurrentAntenna) ? "MAIN_ANT" : "AUX_ANT"
3387                         );
3388         }
3389 #endif
3390         set_fwstate(pmlmepriv, _FW_UNDER_LINKING);
3391         ret = rtw_joinbss_cmd(adapter, candidate);
3392
3393 exit:
3394         _exit_critical_bh(&(pmlmepriv->scanned_queue.lock), &irqL);
3395
3396
3397         return ret;
3398 }
3399
3400 sint rtw_set_auth(_adapter *adapter, struct security_priv *psecuritypriv)
3401 {
3402         struct  cmd_obj *pcmd;
3403         struct  setauth_parm *psetauthparm;
3404         struct  cmd_priv        *pcmdpriv = &(adapter->cmdpriv);
3405         sint            res = _SUCCESS;
3406
3407
3408         pcmd = (struct  cmd_obj *)rtw_zmalloc(sizeof(struct     cmd_obj));
3409         if (pcmd == NULL) {
3410                 res = _FAIL; /* try again */
3411                 goto exit;
3412         }
3413
3414         psetauthparm = (struct setauth_parm *)rtw_zmalloc(sizeof(struct setauth_parm));
3415         if (psetauthparm == NULL) {
3416                 rtw_mfree((unsigned char *)pcmd, sizeof(struct  cmd_obj));
3417                 res = _FAIL;
3418                 goto exit;
3419         }
3420
3421         _rtw_memset(psetauthparm, 0, sizeof(struct setauth_parm));
3422         psetauthparm->mode = (unsigned char)psecuritypriv->dot11AuthAlgrthm;
3423
3424         pcmd->cmdcode = _SetAuth_CMD_;
3425         pcmd->parmbuf = (unsigned char *)psetauthparm;
3426         pcmd->cmdsz = (sizeof(struct setauth_parm));
3427         pcmd->rsp = NULL;
3428         pcmd->rspsz = 0;
3429
3430
3431         _rtw_init_listhead(&pcmd->list);
3432
3433
3434         res = rtw_enqueue_cmd(pcmdpriv, pcmd);
3435
3436 exit:
3437
3438
3439         return res;
3440
3441 }
3442
3443
3444 sint rtw_set_key(_adapter *adapter, struct security_priv *psecuritypriv, sint keyid, u8 set_tx, bool enqueue)
3445 {
3446         u8      keylen;
3447         struct cmd_obj          *pcmd;
3448         struct setkey_parm      *psetkeyparm;
3449         struct cmd_priv         *pcmdpriv = &(adapter->cmdpriv);
3450         struct mlme_priv                *pmlmepriv = &(adapter->mlmepriv);
3451         sint    res = _SUCCESS;
3452
3453
3454         psetkeyparm = (struct setkey_parm *)rtw_zmalloc(sizeof(struct setkey_parm));
3455         if (psetkeyparm == NULL) {
3456                 res = _FAIL;
3457                 goto exit;
3458         }
3459         _rtw_memset(psetkeyparm, 0, sizeof(struct setkey_parm));
3460
3461         if (psecuritypriv->dot11AuthAlgrthm == dot11AuthAlgrthm_8021X) {
3462                 psetkeyparm->algorithm = (unsigned char)psecuritypriv->dot118021XGrpPrivacy;
3463         } else {
3464                 psetkeyparm->algorithm = (u8)psecuritypriv->dot11PrivacyAlgrthm;
3465
3466         }
3467         psetkeyparm->keyid = (u8)keyid;/* 0~3 */
3468         psetkeyparm->set_tx = set_tx;
3469         if (is_wep_enc(psetkeyparm->algorithm))
3470                 adapter->securitypriv.key_mask |= BIT(psetkeyparm->keyid);
3471
3472         RTW_INFO("==> rtw_set_key algorithm(%x),keyid(%x),key_mask(%x)\n", psetkeyparm->algorithm, psetkeyparm->keyid, adapter->securitypriv.key_mask);
3473
3474         switch (psetkeyparm->algorithm) {
3475
3476         case _WEP40_:
3477                 keylen = 5;
3478                 _rtw_memcpy(&(psetkeyparm->key[0]), &(psecuritypriv->dot11DefKey[keyid].skey[0]), keylen);
3479                 break;
3480         case _WEP104_:
3481                 keylen = 13;
3482                 _rtw_memcpy(&(psetkeyparm->key[0]), &(psecuritypriv->dot11DefKey[keyid].skey[0]), keylen);
3483                 break;
3484         case _TKIP_:
3485                 keylen = 16;
3486                 _rtw_memcpy(&psetkeyparm->key, &psecuritypriv->dot118021XGrpKey[keyid], keylen);
3487                 psetkeyparm->grpkey = 1;
3488                 break;
3489         case _AES_:
3490                 keylen = 16;
3491                 _rtw_memcpy(&psetkeyparm->key, &psecuritypriv->dot118021XGrpKey[keyid], keylen);
3492                 psetkeyparm->grpkey = 1;
3493                 break;
3494         default:
3495                 res = _FAIL;
3496                 rtw_mfree((unsigned char *)psetkeyparm, sizeof(struct setkey_parm));
3497                 goto exit;
3498         }
3499
3500
3501         if (enqueue) {
3502                 pcmd = (struct  cmd_obj *)rtw_zmalloc(sizeof(struct     cmd_obj));
3503                 if (pcmd == NULL) {
3504                         rtw_mfree((unsigned char *)psetkeyparm, sizeof(struct setkey_parm));
3505                         res = _FAIL; /* try again */
3506                         goto exit;
3507                 }
3508
3509                 pcmd->cmdcode = _SetKey_CMD_;
3510                 pcmd->parmbuf = (u8 *)psetkeyparm;
3511                 pcmd->cmdsz = (sizeof(struct setkey_parm));
3512                 pcmd->rsp = NULL;
3513                 pcmd->rspsz = 0;
3514
3515                 _rtw_init_listhead(&pcmd->list);
3516
3517                 /* _rtw_init_sema(&(pcmd->cmd_sem), 0); */
3518
3519                 res = rtw_enqueue_cmd(pcmdpriv, pcmd);
3520         } else {
3521                 setkey_hdl(adapter, (u8 *)psetkeyparm);
3522                 rtw_mfree((u8 *) psetkeyparm, sizeof(struct setkey_parm));
3523         }
3524 exit:
3525         return res;
3526
3527 }
3528
3529
3530 /* adjust IEs for rtw_joinbss_cmd in WMM */
3531 int rtw_restruct_wmm_ie(_adapter *adapter, u8 *in_ie, u8 *out_ie, uint in_len, uint initial_out_len)
3532 {
3533         unsigned        int ielength = 0;
3534         unsigned int i, j;
3535
3536         i = 12; /* after the fixed IE */
3537         while (i < in_len) {
3538                 ielength = initial_out_len;
3539
3540                 if (in_ie[i] == 0xDD && in_ie[i + 2] == 0x00 && in_ie[i + 3] == 0x50  && in_ie[i + 4] == 0xF2 && in_ie[i + 5] == 0x02 && i + 5 < in_len) { /* WMM element ID and OUI */
3541
3542                         /* Append WMM IE to the last index of out_ie */
3543 #if 0
3544                         for (j = i; j < i + (in_ie[i + 1] + 2); j++) {
3545                                 out_ie[ielength] = in_ie[j];
3546                                 ielength++;
3547                         }
3548                         out_ie[initial_out_len + 8] = 0x00; /* force the QoS Info Field to be zero */
3549 #endif
3550
3551                         for (j = i; j < i + 9; j++) {
3552                                 out_ie[ielength] = in_ie[j];
3553                                 ielength++;
3554                         }
3555                         out_ie[initial_out_len + 1] = 0x07;
3556                         out_ie[initial_out_len + 6] = 0x00;
3557                         out_ie[initial_out_len + 8] = 0x00;
3558
3559                         break;
3560                 }
3561
3562                 i += (in_ie[i + 1] + 2); /* to the next IE element */
3563         }
3564
3565         return ielength;
3566
3567 }
3568
3569
3570 /*
3571  * Ported from 8185: IsInPreAuthKeyList(). (Renamed from SecIsInPreAuthKeyList(), 2006-10-13.)
3572  * Added by Annie, 2006-05-07.
3573  *
3574  * Search by BSSID,
3575  * Return Value:
3576  *              -1              :if there is no pre-auth key in the  table
3577  *              >=0             :if there is pre-auth key, and   return the entry id
3578  *
3579  *   */
3580
3581 static int SecIsInPMKIDList(_adapter *Adapter, u8 *bssid)
3582 {
3583         struct security_priv *psecuritypriv = &Adapter->securitypriv;
3584         int i = 0;
3585
3586         do {
3587                 if ((psecuritypriv->PMKIDList[i].bUsed) &&
3588                     (_rtw_memcmp(psecuritypriv->PMKIDList[i].Bssid, bssid, ETH_ALEN) == _TRUE))
3589                         break;
3590                 else {
3591                         i++;
3592                         /* continue; */
3593                 }
3594
3595         } while (i < NUM_PMKID_CACHE);
3596
3597         if (i == NUM_PMKID_CACHE) {
3598                 i = -1;/* Could not find. */
3599         } else {
3600                 /* There is one Pre-Authentication Key for the specific BSSID. */
3601         }
3602
3603         return i;
3604
3605 }
3606
3607 /*
3608  * Check the RSN IE length
3609  * If the RSN IE length <= 20, the RSN IE didn't include the PMKID information
3610  * 0-11th element in the array are the fixed IE
3611  * 12th element in the array is the IE
3612  * 13th element in the array is the IE length
3613  *   */
3614
3615 static int rtw_append_pmkid(_adapter *adapter, int iEntry, u8 *ie, uint ie_len)
3616 {
3617         struct security_priv *sec = &adapter->securitypriv;
3618
3619         if (ie[13] > 20) {
3620                 int i;
3621                 u16 pmkid_cnt = RTW_GET_LE16(ie + 14 + 20);
3622                 if (pmkid_cnt == 1 && _rtw_memcmp(ie + 14 + 20 + 2, &sec->PMKIDList[iEntry].PMKID, 16)) {
3623                         RTW_INFO(FUNC_ADPT_FMT" has carried the same PMKID:"KEY_FMT"\n"
3624                                 , FUNC_ADPT_ARG(adapter), KEY_ARG(&sec->PMKIDList[iEntry].PMKID));
3625                         goto exit;
3626                 }
3627
3628                 RTW_INFO(FUNC_ADPT_FMT" remove original PMKID, count:%u\n"
3629                          , FUNC_ADPT_ARG(adapter), pmkid_cnt);
3630
3631                 for (i = 0; i < pmkid_cnt; i++)
3632                         RTW_INFO("    "KEY_FMT"\n", KEY_ARG(ie + 14 + 20 + 2 + i * 16));
3633
3634                 ie_len -= 2 + pmkid_cnt * 16;
3635                 ie[13] = 20;
3636         }
3637
3638         if (ie[13] <= 20) {
3639                 /* The RSN IE didn't include the PMK ID, append the PMK information */
3640
3641                 RTW_INFO(FUNC_ADPT_FMT" append PMKID:"KEY_FMT"\n"
3642                         , FUNC_ADPT_ARG(adapter), KEY_ARG(&sec->PMKIDList[iEntry].PMKID));
3643
3644                 RTW_PUT_LE16(&ie[ie_len], 1);
3645                 ie_len += 2;
3646
3647                 _rtw_memcpy(&ie[ie_len], &sec->PMKIDList[iEntry].PMKID, 16);
3648                 ie_len += 16;
3649
3650                 ie[13] += 18;/* PMKID length = 2+16 */
3651         }
3652
3653 exit:
3654         return ie_len;
3655 }
3656
3657 static int rtw_remove_pmkid(_adapter *adapter, u8 *ie, uint ie_len)
3658 {
3659         struct security_priv *sec = &adapter->securitypriv;
3660         int i;
3661         u16 pmkid_cnt = RTW_GET_LE16(ie + 14 + 20);
3662
3663         if (ie[13] <= 20)
3664                 goto exit;
3665
3666         RTW_INFO(FUNC_ADPT_FMT" remove original PMKID, count:%u\n"
3667                  , FUNC_ADPT_ARG(adapter), pmkid_cnt);
3668
3669         for (i = 0; i < pmkid_cnt; i++)
3670                 RTW_INFO("    "KEY_FMT"\n", KEY_ARG(ie + 14 + 20 + 2 + i * 16));
3671
3672         ie_len -= 2 + pmkid_cnt * 16;
3673         ie[13] = 20;
3674
3675 exit:
3676         return ie_len;
3677 }
3678
3679 sint rtw_restruct_sec_ie(_adapter *adapter, u8 *in_ie, u8 *out_ie, uint in_len)
3680 {
3681         u8 authmode = 0x0, securitytype, match;
3682         u8 sec_ie[255], uncst_oui[4], bkup_ie[255];
3683         u8 wpa_oui[4] = {0x0, 0x50, 0xf2, 0x01};
3684         uint    ielength, cnt, remove_cnt;
3685         int iEntry;
3686
3687         struct mlme_priv *pmlmepriv = &adapter->mlmepriv;
3688         struct security_priv *psecuritypriv = &adapter->securitypriv;
3689         uint    ndisauthmode = psecuritypriv->ndisauthtype;
3690         uint ndissecuritytype = psecuritypriv->ndisencryptstatus;
3691
3692
3693
3694         /* copy fixed ie only */
3695         _rtw_memcpy(out_ie, in_ie, 12);
3696         ielength = 12;
3697         if ((ndisauthmode == Ndis802_11AuthModeWPA) || (ndisauthmode == Ndis802_11AuthModeWPAPSK))
3698                 authmode = _WPA_IE_ID_;
3699         if ((ndisauthmode == Ndis802_11AuthModeWPA2) || (ndisauthmode == Ndis802_11AuthModeWPA2PSK))
3700                 authmode = _WPA2_IE_ID_;
3701
3702         if (check_fwstate(pmlmepriv, WIFI_UNDER_WPS)) {
3703                 _rtw_memcpy(out_ie + ielength, psecuritypriv->wps_ie, psecuritypriv->wps_ie_len);
3704
3705                 ielength += psecuritypriv->wps_ie_len;
3706         } else if ((authmode == _WPA_IE_ID_) || (authmode == _WPA2_IE_ID_)) {
3707                 /* copy RSN or SSN               */
3708                 _rtw_memcpy(&out_ie[ielength], &psecuritypriv->supplicant_ie[0], psecuritypriv->supplicant_ie[1] + 2);
3709                 /* debug for CONFIG_IEEE80211W
3710                 {
3711                         int jj;
3712                         printk("supplicant_ie_length=%d &&&&&&&&&&&&&&&&&&&\n", psecuritypriv->supplicant_ie[1]+2);
3713                         for(jj=0; jj < psecuritypriv->supplicant_ie[1]+2; jj++)
3714                                 printk(" %02x ", psecuritypriv->supplicant_ie[jj]);
3715                         printk("\n");
3716                 }*/
3717                 ielength += psecuritypriv->supplicant_ie[1] + 2;
3718                 rtw_report_sec_ie(adapter, authmode, psecuritypriv->supplicant_ie);
3719
3720 #ifdef CONFIG_DRVEXT_MODULE
3721                 drvext_report_sec_ie(&adapter->drvextpriv, authmode, sec_ie);
3722 #endif
3723         }
3724
3725         iEntry = SecIsInPMKIDList(adapter, pmlmepriv->assoc_bssid);
3726         if (iEntry < 0) {
3727                 if (authmode == _WPA2_IE_ID_)
3728                         ielength = rtw_remove_pmkid(adapter, out_ie, ielength);
3729         } else {
3730                 if (authmode == _WPA2_IE_ID_)
3731                         ielength = rtw_append_pmkid(adapter, iEntry, out_ie, ielength);
3732         }
3733
3734
3735         return ielength;
3736 }
3737
3738 void rtw_init_registrypriv_dev_network(_adapter *adapter)
3739 {
3740         struct registry_priv *pregistrypriv = &adapter->registrypriv;
3741         WLAN_BSSID_EX    *pdev_network = &pregistrypriv->dev_network;
3742         u8 *myhwaddr = adapter_mac_addr(adapter);
3743
3744
3745         _rtw_memcpy(pdev_network->MacAddress, myhwaddr, ETH_ALEN);
3746
3747         _rtw_memcpy(&pdev_network->Ssid, &pregistrypriv->ssid, sizeof(NDIS_802_11_SSID));
3748
3749         pdev_network->Configuration.Length = sizeof(NDIS_802_11_CONFIGURATION);
3750         pdev_network->Configuration.BeaconPeriod = 100;
3751         pdev_network->Configuration.FHConfig.Length = 0;
3752         pdev_network->Configuration.FHConfig.HopPattern = 0;
3753         pdev_network->Configuration.FHConfig.HopSet = 0;
3754         pdev_network->Configuration.FHConfig.DwellTime = 0;
3755
3756
3757
3758 }
3759
3760 void rtw_update_registrypriv_dev_network(_adapter *adapter)
3761 {
3762         int sz = 0;
3763         struct registry_priv *pregistrypriv = &adapter->registrypriv;
3764         WLAN_BSSID_EX    *pdev_network = &pregistrypriv->dev_network;
3765         struct  security_priv   *psecuritypriv = &adapter->securitypriv;
3766         struct  wlan_network    *cur_network = &adapter->mlmepriv.cur_network;
3767         /* struct       xmit_priv       *pxmitpriv = &adapter->xmitpriv; */
3768         struct mlme_ext_priv    *pmlmeext = &adapter->mlmeextpriv;
3769
3770
3771 #if 0
3772         pxmitpriv->vcs_setting = pregistrypriv->vrtl_carrier_sense;
3773         pxmitpriv->vcs = pregistrypriv->vcs_type;
3774         pxmitpriv->vcs_type = pregistrypriv->vcs_type;
3775         /* pxmitpriv->rts_thresh = pregistrypriv->rts_thresh; */
3776         pxmitpriv->frag_len = pregistrypriv->frag_thresh;
3777
3778         adapter->qospriv.qos_option = pregistrypriv->wmm_enable;
3779 #endif
3780
3781         pdev_network->Privacy = (psecuritypriv->dot11PrivacyAlgrthm > 0 ? 1 : 0) ; /* adhoc no 802.1x */
3782
3783         pdev_network->Rssi = 0;
3784
3785         switch (pregistrypriv->wireless_mode) {
3786         case WIRELESS_11B:
3787                 pdev_network->NetworkTypeInUse = (Ndis802_11DS);
3788                 break;
3789         case WIRELESS_11G:
3790         case WIRELESS_11BG:
3791         case WIRELESS_11_24N:
3792         case WIRELESS_11G_24N:
3793         case WIRELESS_11BG_24N:
3794                 pdev_network->NetworkTypeInUse = (Ndis802_11OFDM24);
3795                 break;
3796         case WIRELESS_11A:
3797         case WIRELESS_11A_5N:
3798                 pdev_network->NetworkTypeInUse = (Ndis802_11OFDM5);
3799                 break;
3800         case WIRELESS_11ABGN:
3801                 if (pregistrypriv->channel > 14)
3802                         pdev_network->NetworkTypeInUse = (Ndis802_11OFDM5);
3803                 else
3804                         pdev_network->NetworkTypeInUse = (Ndis802_11OFDM24);
3805                 break;
3806         default:
3807                 /* TODO */
3808                 break;
3809         }
3810
3811         pdev_network->Configuration.DSConfig = (pregistrypriv->channel);
3812
3813         if (cur_network->network.InfrastructureMode == Ndis802_11IBSS) {
3814                 pdev_network->Configuration.ATIMWindow = (0);
3815
3816                 if (pmlmeext->cur_channel != 0)
3817                         pdev_network->Configuration.DSConfig = pmlmeext->cur_channel;
3818                 else
3819                         pdev_network->Configuration.DSConfig = 1;
3820         }
3821
3822         pdev_network->InfrastructureMode = (cur_network->network.InfrastructureMode);
3823
3824         /* 1. Supported rates */
3825         /* 2. IE */
3826
3827         /* rtw_set_supported_rate(pdev_network->SupportedRates, pregistrypriv->wireless_mode) ; */ /* will be called in rtw_generate_ie */
3828         sz = rtw_generate_ie(pregistrypriv);
3829
3830         pdev_network->IELength = sz;
3831
3832         pdev_network->Length = get_WLAN_BSSID_EX_sz((WLAN_BSSID_EX *)pdev_network);
3833
3834         /* notes: translate IELength & Length after assign the Length to cmdsz in createbss_cmd(); */
3835         /* pdev_network->IELength = cpu_to_le32(sz); */
3836
3837
3838 }
3839
3840 void rtw_get_encrypt_decrypt_from_registrypriv(_adapter *adapter)
3841 {
3842
3843
3844
3845 }
3846
3847 /* the fucntion is at passive_level */
3848 void rtw_joinbss_reset(_adapter *padapter)
3849 {
3850         u8      threshold;
3851         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
3852         /* todo: if you want to do something io/reg/hw setting before join_bss, please add code here */
3853
3854 #ifdef CONFIG_80211N_HT
3855         struct ht_priv          *phtpriv = &pmlmepriv->htpriv;
3856
3857         pmlmepriv->num_FortyMHzIntolerant = 0;
3858
3859         pmlmepriv->num_sta_no_ht = 0;
3860
3861         phtpriv->ampdu_enable = _FALSE;/* reset to disabled */
3862
3863 #if defined(CONFIG_USB_HCI) || defined(CONFIG_SDIO_HCI)
3864         /* TH=1 => means that invalidate usb rx aggregation */
3865         /* TH=0 => means that validate usb rx aggregation, use init value. */
3866         if (phtpriv->ht_option) {
3867                 if (padapter->registrypriv.wifi_spec == 1)
3868                         threshold = 1;
3869                 else
3870                         threshold = 0;
3871                 rtw_hal_set_hwreg(padapter, HW_VAR_RXDMA_AGG_PG_TH, (u8 *)(&threshold));
3872         } else {
3873                 threshold = 1;
3874                 rtw_hal_set_hwreg(padapter, HW_VAR_RXDMA_AGG_PG_TH, (u8 *)(&threshold));
3875         }
3876 #endif/* #if defined( CONFIG_USB_HCI) || defined (CONFIG_SDIO_HCI) */
3877
3878 #endif/* #ifdef CONFIG_80211N_HT */
3879
3880 }
3881
3882
3883 #ifdef CONFIG_80211N_HT
3884 void    rtw_ht_use_default_setting(_adapter *padapter)
3885 {
3886         struct mlme_priv                *pmlmepriv = &padapter->mlmepriv;
3887         struct ht_priv          *phtpriv = &pmlmepriv->htpriv;
3888         struct registry_priv    *pregistrypriv = &padapter->registrypriv;
3889         BOOLEAN         bHwLDPCSupport = _FALSE, bHwSTBCSupport = _FALSE;
3890 #ifdef CONFIG_BEAMFORMING
3891         BOOLEAN         bHwSupportBeamformer = _FALSE, bHwSupportBeamformee = _FALSE;
3892 #endif /* CONFIG_BEAMFORMING */
3893
3894         if (pregistrypriv->wifi_spec)
3895                 phtpriv->bss_coexist = 1;
3896         else
3897                 phtpriv->bss_coexist = 0;
3898
3899         phtpriv->sgi_40m = TEST_FLAG(pregistrypriv->short_gi, BIT1) ? _TRUE : _FALSE;
3900         phtpriv->sgi_20m = TEST_FLAG(pregistrypriv->short_gi, BIT0) ? _TRUE : _FALSE;
3901
3902         /* LDPC support */
3903         rtw_hal_get_def_var(padapter, HAL_DEF_RX_LDPC, (u8 *)&bHwLDPCSupport);
3904         CLEAR_FLAGS(phtpriv->ldpc_cap);
3905         if (bHwLDPCSupport) {
3906                 if (TEST_FLAG(pregistrypriv->ldpc_cap, BIT4))
3907                         SET_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_RX);
3908         }
3909         rtw_hal_get_def_var(padapter, HAL_DEF_TX_LDPC, (u8 *)&bHwLDPCSupport);
3910         if (bHwLDPCSupport) {
3911                 if (TEST_FLAG(pregistrypriv->ldpc_cap, BIT5))
3912                         SET_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_TX);
3913         }
3914         if (phtpriv->ldpc_cap)
3915                 RTW_INFO("[HT] HAL Support LDPC = 0x%02X\n", phtpriv->ldpc_cap);
3916
3917         /* STBC */
3918         rtw_hal_get_def_var(padapter, HAL_DEF_TX_STBC, (u8 *)&bHwSTBCSupport);
3919         CLEAR_FLAGS(phtpriv->stbc_cap);
3920         if (bHwSTBCSupport) {
3921                 if (TEST_FLAG(pregistrypriv->stbc_cap, BIT5))
3922                         SET_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX);
3923         }
3924         rtw_hal_get_def_var(padapter, HAL_DEF_RX_STBC, (u8 *)&bHwSTBCSupport);
3925         if (bHwSTBCSupport) {
3926                 if (TEST_FLAG(pregistrypriv->stbc_cap, BIT4))
3927                         SET_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_RX);
3928         }
3929         if (phtpriv->stbc_cap)
3930                 RTW_INFO("[HT] HAL Support STBC = 0x%02X\n", phtpriv->stbc_cap);
3931
3932         /* Beamforming setting */
3933         CLEAR_FLAGS(phtpriv->beamform_cap);
3934 #ifdef CONFIG_BEAMFORMING
3935         rtw_hal_get_def_var(padapter, HAL_DEF_EXPLICIT_BEAMFORMER, (u8 *)&bHwSupportBeamformer);
3936         rtw_hal_get_def_var(padapter, HAL_DEF_EXPLICIT_BEAMFORMEE, (u8 *)&bHwSupportBeamformee);
3937         if (TEST_FLAG(pregistrypriv->beamform_cap, BIT4) && bHwSupportBeamformer) {
3938                 SET_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE);
3939                 RTW_INFO("[HT] HAL Support Beamformer\n");
3940         }
3941         if (TEST_FLAG(pregistrypriv->beamform_cap, BIT5) && bHwSupportBeamformee) {
3942                 SET_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE);
3943                 RTW_INFO("[HT] HAL Support Beamformee\n");
3944         }
3945 #endif /* CONFIG_BEAMFORMING */
3946 }
3947 void rtw_build_wmm_ie_ht(_adapter *padapter, u8 *out_ie, uint *pout_len)
3948 {
3949         unsigned char WMM_IE[] = {0x00, 0x50, 0xf2, 0x02, 0x00, 0x01, 0x00};
3950         int out_len;
3951         u8 *pframe;
3952
3953         if (padapter->mlmepriv.qospriv.qos_option == 0) {
3954                 out_len = *pout_len;
3955                 pframe = rtw_set_ie(out_ie + out_len, _VENDOR_SPECIFIC_IE_,
3956                                     _WMM_IE_Length_, WMM_IE, pout_len);
3957
3958                 padapter->mlmepriv.qospriv.qos_option = 1;
3959         }
3960 }
3961
3962 /* the fucntion is >= passive_level */
3963 unsigned int rtw_restructure_ht_ie(_adapter *padapter, u8 *in_ie, u8 *out_ie, uint in_len, uint *pout_len, u8 channel)
3964 {
3965         u32 ielen, out_len;
3966         u32 rx_packet_offset, max_recvbuf_sz;
3967         HT_CAP_AMPDU_FACTOR max_rx_ampdu_factor;
3968         HT_CAP_AMPDU_DENSITY best_ampdu_density;
3969         unsigned char *p, *pframe;
3970         struct rtw_ieee80211_ht_cap ht_capie;
3971         u8      cbw40_enable = 0, rf_type = 0, operation_bw = 0, rf_num = 0, rx_stbc_nss = 0;
3972         struct registry_priv *pregistrypriv = &padapter->registrypriv;
3973         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
3974         struct ht_priv          *phtpriv = &pmlmepriv->htpriv;
3975         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
3976
3977         phtpriv->ht_option = _FALSE;
3978
3979         out_len = *pout_len;
3980
3981         _rtw_memset(&ht_capie, 0, sizeof(struct rtw_ieee80211_ht_cap));
3982
3983         ht_capie.cap_info = IEEE80211_HT_CAP_DSSSCCK40;
3984
3985         if (phtpriv->sgi_20m)
3986                 ht_capie.cap_info |= IEEE80211_HT_CAP_SGI_20;
3987
3988         /* Get HT BW */
3989         if (in_ie == NULL) {
3990                 /* TDLS: TODO 20/40 issue */
3991                 if (check_fwstate(pmlmepriv, WIFI_STATION_STATE)) {
3992                         operation_bw = padapter->mlmeextpriv.cur_bwmode;
3993                         if (operation_bw > CHANNEL_WIDTH_40)
3994                                 operation_bw = CHANNEL_WIDTH_40;
3995                 } else
3996                         /* TDLS: TODO 40? */
3997                         operation_bw = CHANNEL_WIDTH_40;
3998         } else {
3999                 p = rtw_get_ie(in_ie, _HT_ADD_INFO_IE_, &ielen, in_len);
4000                 if (p && (ielen == sizeof(struct ieee80211_ht_addt_info))) {
4001                         struct HT_info_element *pht_info = (struct HT_info_element *)(p + 2);
4002                         if (pht_info->infos[0] & BIT(2)) {
4003                                 switch (pht_info->infos[0] & 0x3) {
4004                                 case 1:
4005                                 case 3:
4006                                         operation_bw = CHANNEL_WIDTH_40;
4007                                         break;
4008                                 default:
4009                                         operation_bw = CHANNEL_WIDTH_20;
4010                                         break;
4011                                 }
4012                         } else
4013                                 operation_bw = CHANNEL_WIDTH_20;
4014                 }
4015         }
4016
4017         /* to disable 40M Hz support while gd_bw_40MHz_en = 0 */
4018         if (hal_chk_bw_cap(padapter, BW_CAP_40M)) {
4019                 if (channel > 14) {
4020                         if (REGSTY_IS_BW_5G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
4021                                 cbw40_enable = 1;
4022                 } else {
4023                         if (REGSTY_IS_BW_2G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
4024                                 cbw40_enable = 1;
4025                 }
4026         }
4027
4028         if ((cbw40_enable == 1) && (operation_bw == CHANNEL_WIDTH_40)) {
4029                 ht_capie.cap_info |= IEEE80211_HT_CAP_SUP_WIDTH;
4030                 if (phtpriv->sgi_40m)
4031                         ht_capie.cap_info |= IEEE80211_HT_CAP_SGI_40;
4032         }
4033
4034         /* todo: disable SM power save mode */
4035         ht_capie.cap_info |= IEEE80211_HT_CAP_SM_PS;
4036
4037         /* RX LDPC */
4038         if (TEST_FLAG(phtpriv->ldpc_cap, LDPC_HT_ENABLE_RX)) {
4039                 ht_capie.cap_info |= IEEE80211_HT_CAP_LDPC_CODING;
4040                 RTW_INFO("[HT] Declare supporting RX LDPC\n");
4041         }
4042
4043         /* TX STBC */
4044         if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_TX)) {
4045                 ht_capie.cap_info |= IEEE80211_HT_CAP_TX_STBC;
4046                 RTW_INFO("[HT] Declare supporting TX STBC\n");
4047         }
4048
4049         /* RX STBC */
4050         if (TEST_FLAG(phtpriv->stbc_cap, STBC_HT_ENABLE_RX)) {
4051                 if ((pregistrypriv->rx_stbc == 0x3) ||                                                  /* enable for 2.4/5 GHz */
4052                     ((channel <= 14) && (pregistrypriv->rx_stbc == 0x1)) ||             /* enable for 2.4GHz */
4053                     ((channel > 14) && (pregistrypriv->rx_stbc == 0x2)) ||              /* enable for 5GHz */
4054                     (pregistrypriv->wifi_spec == 1)) {
4055                         /* HAL_DEF_RX_STBC means STBC RX spatial stream, todo: VHT 4 streams */
4056                         rtw_hal_get_def_var(padapter, HAL_DEF_RX_STBC, (u8 *)(&rx_stbc_nss));
4057                         SET_HT_CAP_ELE_RX_STBC(&ht_capie, rx_stbc_nss);
4058                         RTW_INFO("[HT] Declare supporting RX STBC = %d\n", rx_stbc_nss);
4059                 }
4060         }
4061
4062         /* fill default supported_mcs_set */
4063         _rtw_memcpy(ht_capie.supp_mcs_set, pmlmeext->default_supported_mcs_set, 16);
4064
4065         /* update default supported_mcs_set */
4066         rtw_hal_get_hwreg(padapter, HW_VAR_RF_TYPE, (u8 *)(&rf_type));
4067
4068         switch (rf_type) {
4069         case RF_1T1R:
4070                 set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_1R);
4071                 break;
4072
4073         case RF_2T2R:
4074         case RF_1T2R:
4075 #ifdef CONFIG_DISABLE_MCS13TO15
4076                 if (((cbw40_enable == 1) && (operation_bw == CHANNEL_WIDTH_40)) && (pregistrypriv->wifi_spec != 1))
4077                         set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_2R_13TO15_OFF);
4078                 else
4079                         set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_2R);
4080 #else /* CONFIG_DISABLE_MCS13TO15 */
4081                 set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_2R);
4082 #endif /* CONFIG_DISABLE_MCS13TO15 */
4083                 break;
4084         case RF_3T3R:
4085                 set_mcs_rate_by_mask(ht_capie.supp_mcs_set, MCS_RATE_3R);
4086                 break;
4087         default:
4088                 RTW_INFO("[warning] rf_type %d is not expected\n", rf_type);
4089         }
4090
4091         {
4092                 rtw_hal_get_def_var(padapter, HAL_DEF_RX_PACKET_OFFSET, &rx_packet_offset);
4093                 rtw_hal_get_def_var(padapter, HAL_DEF_MAX_RECVBUF_SZ, &max_recvbuf_sz);
4094                 if (max_recvbuf_sz - rx_packet_offset >= (8191 - 256)) {
4095                         RTW_INFO("%s IEEE80211_HT_CAP_MAX_AMSDU is set\n", __FUNCTION__);
4096                         ht_capie.cap_info = ht_capie.cap_info | IEEE80211_HT_CAP_MAX_AMSDU;
4097                 }
4098         }
4099         /*
4100         AMPDU_para [1:0]:Max AMPDU Len => 0:8k , 1:16k, 2:32k, 3:64k
4101         AMPDU_para [4:2]:Min MPDU Start Spacing
4102         */
4103
4104         /*
4105         #if defined(CONFIG_RTL8188E) && defined(CONFIG_SDIO_HCI)
4106         ht_capie.ampdu_params_info = 2;
4107         #else
4108         ht_capie.ampdu_params_info = (IEEE80211_HT_CAP_AMPDU_FACTOR&0x03);
4109         #endif
4110         */
4111
4112         if (padapter->driver_rx_ampdu_factor != 0xFF)
4113                 max_rx_ampdu_factor = (HT_CAP_AMPDU_FACTOR)padapter->driver_rx_ampdu_factor;
4114         else
4115                 rtw_hal_get_def_var(padapter, HW_VAR_MAX_RX_AMPDU_FACTOR, &max_rx_ampdu_factor);
4116
4117         /* rtw_hal_get_def_var(padapter, HW_VAR_MAX_RX_AMPDU_FACTOR, &max_rx_ampdu_factor); */
4118         ht_capie.ampdu_params_info = (max_rx_ampdu_factor & 0x03);
4119
4120         if (padapter->driver_rx_ampdu_spacing != 0xFF)
4121                 ht_capie.ampdu_params_info |= ((padapter->driver_rx_ampdu_spacing & 0x07) << 2);
4122         else {
4123                 if (padapter->securitypriv.dot11PrivacyAlgrthm == _AES_) {
4124                         /*
4125                         *       Todo : Each chip must to ask DD , this chip best ampdu_density setting
4126                         *       By yiwei.sun
4127                         */
4128                         rtw_hal_get_def_var(padapter, HW_VAR_BEST_AMPDU_DENSITY, &best_ampdu_density);
4129
4130                         ht_capie.ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY & (best_ampdu_density << 2));
4131
4132                 } else
4133                         ht_capie.ampdu_params_info |= (IEEE80211_HT_CAP_AMPDU_DENSITY & 0x00);
4134         }
4135 #ifdef CONFIG_BEAMFORMING
4136         ht_capie.tx_BF_cap_info = 0;
4137
4138         /* HT Beamformer*/
4139         if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMER_ENABLE)) {
4140                 /* Transmit NDP Capable */
4141                 SET_HT_CAP_TXBF_TRANSMIT_NDP_CAP(&ht_capie, 1);
4142                 /* Explicit Compressed Steering Capable */
4143                 SET_HT_CAP_TXBF_EXPLICIT_COMP_STEERING_CAP(&ht_capie, 1);
4144                 /* Compressed Steering Number Antennas */
4145                 SET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(&ht_capie, 1);
4146                 rtw_hal_get_def_var(padapter, HAL_DEF_BEAMFORMER_CAP, (u8 *)&rf_num);
4147                 SET_HT_CAP_TXBF_CHNL_ESTIMATION_NUM_ANTENNAS(&ht_capie, rf_num);
4148         }
4149
4150         /* HT Beamformee */
4151         if (TEST_FLAG(phtpriv->beamform_cap, BEAMFORMING_HT_BEAMFORMEE_ENABLE)) {
4152                 /* Receive NDP Capable */
4153                 SET_HT_CAP_TXBF_RECEIVE_NDP_CAP(&ht_capie, 1);
4154                 /* Explicit Compressed Beamforming Feedback Capable */
4155                 SET_HT_CAP_TXBF_EXPLICIT_COMP_FEEDBACK_CAP(&ht_capie, 2);
4156                 rtw_hal_get_def_var(padapter, HAL_DEF_BEAMFORMEE_CAP, (u8 *)&rf_num);
4157                 SET_HT_CAP_TXBF_COMP_STEERING_NUM_ANTENNAS(&ht_capie, rf_num);
4158         }
4159 #endif/*CONFIG_BEAMFORMING*/
4160
4161         pframe = rtw_set_ie(out_ie + out_len, _HT_CAPABILITY_IE_,
4162                 sizeof(struct rtw_ieee80211_ht_cap), (unsigned char *)&ht_capie, pout_len);
4163
4164         phtpriv->ht_option = _TRUE;
4165
4166         if (in_ie != NULL) {
4167                 p = rtw_get_ie(in_ie, _HT_ADD_INFO_IE_, &ielen, in_len);
4168                 if (p && (ielen == sizeof(struct ieee80211_ht_addt_info))) {
4169                         out_len = *pout_len;
4170                         pframe = rtw_set_ie(out_ie + out_len, _HT_ADD_INFO_IE_, ielen, p + 2 , pout_len);
4171                 }
4172         }
4173
4174         return phtpriv->ht_option;
4175
4176 }
4177
4178 /* the fucntion is > passive_level (in critical_section) */
4179 void rtw_update_ht_cap(_adapter *padapter, u8 *pie, uint ie_len, u8 channel)
4180 {
4181         u8 *p, max_ampdu_sz;
4182         int len;
4183         /* struct sta_info *bmc_sta, *psta; */
4184         struct rtw_ieee80211_ht_cap *pht_capie;
4185         struct ieee80211_ht_addt_info *pht_addtinfo;
4186         /* struct recv_reorder_ctrl *preorder_ctrl; */
4187         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
4188         struct ht_priv          *phtpriv = &pmlmepriv->htpriv;
4189         /* struct recv_priv *precvpriv = &padapter->recvpriv; */
4190         struct registry_priv *pregistrypriv = &padapter->registrypriv;
4191         /* struct wlan_network *pcur_network = &(pmlmepriv->cur_network);; */
4192         struct mlme_ext_priv    *pmlmeext = &padapter->mlmeextpriv;
4193         struct mlme_ext_info    *pmlmeinfo = &(pmlmeext->mlmext_info);
4194         u8 cbw40_enable = 0;
4195
4196
4197         if (!phtpriv->ht_option)
4198                 return;
4199
4200         if ((!pmlmeinfo->HT_info_enable) || (!pmlmeinfo->HT_caps_enable))
4201                 return;
4202
4203         RTW_INFO("+rtw_update_ht_cap()\n");
4204
4205         /* maybe needs check if ap supports rx ampdu. */
4206         if ((phtpriv->ampdu_enable == _FALSE) && (pregistrypriv->ampdu_enable == 1)) {
4207                 if (pregistrypriv->wifi_spec == 1) {
4208                         /* remove this part because testbed AP should disable RX AMPDU */
4209                         /* phtpriv->ampdu_enable = _FALSE; */
4210                         phtpriv->ampdu_enable = _TRUE;
4211                 } else
4212                         phtpriv->ampdu_enable = _TRUE;
4213         }
4214
4215
4216         /* check Max Rx A-MPDU Size */
4217         len = 0;
4218         p = rtw_get_ie(pie + sizeof(NDIS_802_11_FIXED_IEs), _HT_CAPABILITY_IE_, &len, ie_len - sizeof(NDIS_802_11_FIXED_IEs));
4219         if (p && len > 0) {
4220                 pht_capie = (struct rtw_ieee80211_ht_cap *)(p + 2);
4221                 max_ampdu_sz = (pht_capie->ampdu_params_info & IEEE80211_HT_CAP_AMPDU_FACTOR);
4222                 max_ampdu_sz = 1 << (max_ampdu_sz + 3); /* max_ampdu_sz (kbytes); */
4223
4224                 /* RTW_INFO("rtw_update_ht_cap(): max_ampdu_sz=%d\n", max_ampdu_sz); */
4225                 phtpriv->rx_ampdu_maxlen = max_ampdu_sz;
4226
4227         }
4228
4229
4230         len = 0;
4231         p = rtw_get_ie(pie + sizeof(NDIS_802_11_FIXED_IEs), _HT_ADD_INFO_IE_, &len, ie_len - sizeof(NDIS_802_11_FIXED_IEs));
4232         if (p && len > 0) {
4233                 pht_addtinfo = (struct ieee80211_ht_addt_info *)(p + 2);
4234                 /* todo: */
4235         }
4236
4237         if (hal_chk_bw_cap(padapter, BW_CAP_40M)) {
4238                 if (channel > 14) {
4239                         if (REGSTY_IS_BW_5G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
4240                                 cbw40_enable = 1;
4241                 } else {
4242                         if (REGSTY_IS_BW_2G_SUPPORT(pregistrypriv, CHANNEL_WIDTH_40))
4243                                 cbw40_enable = 1;
4244                 }
4245         }
4246
4247         /* update cur_bwmode & cur_ch_offset */
4248         if ((cbw40_enable) &&
4249             (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & BIT(1)) &&
4250             (pmlmeinfo->HT_info.infos[0] & BIT(2))) {
4251                 int i;
4252                 u8      rf_type = RF_1T1R;
4253
4254                 rtw_hal_get_hwreg(padapter, HW_VAR_RF_TYPE, (u8 *)(&rf_type));
4255
4256                 /* update the MCS set */
4257                 for (i = 0; i < 16; i++)
4258                         pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate[i] &= pmlmeext->default_supported_mcs_set[i];
4259
4260                 /* update the MCS rates */
4261                 switch (rf_type) {
4262                 case RF_1T1R:
4263                 case RF_1T2R:
4264                         set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_1R);
4265                         break;
4266                 case RF_2T2R:
4267 #ifdef CONFIG_DISABLE_MCS13TO15
4268                         if (pmlmeext->cur_bwmode == CHANNEL_WIDTH_40 && pregistrypriv->wifi_spec != 1)
4269                                 set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R_13TO15_OFF);
4270                         else
4271                                 set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R);
4272 #else /* CONFIG_DISABLE_MCS13TO15 */
4273                         set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_2R);
4274 #endif /* CONFIG_DISABLE_MCS13TO15 */
4275                         break;
4276                 case RF_3T3R:
4277                         set_mcs_rate_by_mask(pmlmeinfo->HT_caps.u.HT_cap_element.MCS_rate, MCS_RATE_3R);
4278                         break;
4279                 default:
4280                         RTW_INFO("[warning] rf_type %d is not expected\n", rf_type);
4281                 }
4282
4283                 /* switch to the 40M Hz mode accoring to the AP */
4284                 /* pmlmeext->cur_bwmode = CHANNEL_WIDTH_40; */
4285                 switch ((pmlmeinfo->HT_info.infos[0] & 0x3)) {
4286                 case EXTCHNL_OFFSET_UPPER:
4287                         pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_LOWER;
4288                         break;
4289
4290                 case EXTCHNL_OFFSET_LOWER:
4291                         pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_UPPER;
4292                         break;
4293
4294                 default:
4295                         pmlmeext->cur_ch_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
4296                         break;
4297                 }
4298         }
4299
4300         /*  */
4301         /* Config SM Power Save setting */
4302         /*  */
4303         pmlmeinfo->SM_PS = (pmlmeinfo->HT_caps.u.HT_cap_element.HT_caps_info & 0x0C) >> 2;
4304         if (pmlmeinfo->SM_PS == WLAN_HT_CAP_SM_PS_STATIC) {
4305 #if 0
4306                 u8 i;
4307                 /* update the MCS rates */
4308                 for (i = 0; i < 16; i++)
4309                         pmlmeinfo->HT_caps.HT_cap_element.MCS_rate[i] &= MCS_rate_1R[i];
4310 #endif
4311                 RTW_INFO("%s(): WLAN_HT_CAP_SM_PS_STATIC\n", __FUNCTION__);
4312         }
4313
4314         /*  */
4315         /* Config current HT Protection mode. */
4316         /*  */
4317         pmlmeinfo->HT_protection = pmlmeinfo->HT_info.infos[1] & 0x3;
4318
4319
4320
4321 #if 0 /* move to rtw_update_sta_info_client() */
4322         /* for A-MPDU Rx reordering buffer control for bmc_sta & sta_info */
4323         /* if A-MPDU Rx is enabled, reseting  rx_ordering_ctrl wstart_b(indicate_seq) to default value=0xffff */
4324         /* todo: check if AP can send A-MPDU packets */
4325         bmc_sta = rtw_get_bcmc_stainfo(padapter);
4326         if (bmc_sta) {
4327                 for (i = 0; i < 16 ; i++) {
4328                         /* preorder_ctrl = &precvpriv->recvreorder_ctrl[i]; */
4329                         preorder_ctrl = &bmc_sta->recvreorder_ctrl[i];
4330                         preorder_ctrl->enable = _FALSE;
4331                         preorder_ctrl->indicate_seq = 0xffff;
4332 #ifdef DBG_RX_SEQ
4333                         RTW_INFO("DBG_RX_SEQ %s:%d indicate_seq:%u\n", __FUNCTION__, __LINE__,
4334                                  preorder_ctrl->indicate_seq);
4335 #endif
4336                         preorder_ctrl->wend_b = 0xffff;
4337                         preorder_ctrl->wsize_b = 64;/* max_ampdu_sz; */ /* ex. 32(kbytes) -> wsize_b=32 */
4338                 }
4339         }
4340
4341         psta = rtw_get_stainfo(&padapter->stapriv, pcur_network->network.MacAddress);
4342         if (psta) {
4343                 for (i = 0; i < 16 ; i++) {
4344                         /* preorder_ctrl = &precvpriv->recvreorder_ctrl[i]; */
4345                         preorder_ctrl = &psta->recvreorder_ctrl[i];
4346                         preorder_ctrl->enable = _FALSE;
4347                         preorder_ctrl->indicate_seq = 0xffff;
4348 #ifdef DBG_RX_SEQ
4349                         RTW_INFO("DBG_RX_SEQ %s:%d indicate_seq:%u\n", __FUNCTION__, __LINE__,
4350                                  preorder_ctrl->indicate_seq);
4351 #endif
4352                         preorder_ctrl->wend_b = 0xffff;
4353                         preorder_ctrl->wsize_b = 64;/* max_ampdu_sz; */ /* ex. 32(kbytes) -> wsize_b=32 */
4354                 }
4355         }
4356 #endif
4357
4358 }
4359
4360 #ifdef CONFIG_TDLS
4361 void rtw_issue_addbareq_cmd_tdls(_adapter *padapter, struct xmit_frame *pxmitframe)
4362 {
4363         struct pkt_attrib *pattrib = &pxmitframe->attrib;
4364         struct sta_info *ptdls_sta = NULL;
4365         u8 issued;
4366         int priority;
4367         struct ht_priv  *phtpriv;
4368
4369         priority = pattrib->priority;
4370
4371         if (pattrib->direct_link == _TRUE) {
4372                 ptdls_sta = rtw_get_stainfo(&padapter->stapriv, pattrib->dst);
4373                 if ((ptdls_sta != NULL) && (ptdls_sta->tdls_sta_state & TDLS_LINKED_STATE)) {
4374                         phtpriv = &ptdls_sta->htpriv;
4375
4376                         if ((phtpriv->ht_option == _TRUE) && (phtpriv->ampdu_enable == _TRUE)) {
4377                                 issued = (phtpriv->agg_enable_bitmap >> priority) & 0x1;
4378                                 issued |= (phtpriv->candidate_tid_bitmap >> priority) & 0x1;
4379
4380                                 if (0 == issued) {
4381                                         RTW_INFO("[%s], p=%d\n", __FUNCTION__, priority);
4382                                         ptdls_sta->htpriv.candidate_tid_bitmap |= BIT((u8)priority);
4383                                         rtw_addbareq_cmd(padapter, (u8)priority, pattrib->dst);
4384                                 }
4385                         }
4386                 }
4387         }
4388 }
4389 #endif /* CONFIG_TDLS */
4390
4391 void rtw_issue_addbareq_cmd(_adapter *padapter, struct xmit_frame *pxmitframe)
4392 {
4393         u8 issued;
4394         int priority;
4395         struct sta_info *psta = NULL;
4396         struct ht_priv  *phtpriv;
4397         struct pkt_attrib *pattrib = &pxmitframe->attrib;
4398         s32 bmcst = IS_MCAST(pattrib->ra);
4399
4400         /* if(bmcst || (padapter->mlmepriv.LinkDetectInfo.bTxBusyTraffic == _FALSE)) */
4401         if (bmcst || (padapter->mlmepriv.LinkDetectInfo.NumTxOkInPeriod < 100))
4402                 return;
4403
4404         priority = pattrib->priority;
4405
4406 #ifdef CONFIG_TDLS
4407         rtw_issue_addbareq_cmd_tdls(padapter, pxmitframe);
4408 #endif /* CONFIG_TDLS */
4409
4410         psta = rtw_get_stainfo(&padapter->stapriv, pattrib->ra);
4411         if (pattrib->psta != psta) {
4412                 RTW_INFO("%s, pattrib->psta(%p) != psta(%p)\n", __func__, pattrib->psta, psta);
4413                 return;
4414         }
4415
4416         if (psta == NULL) {
4417                 RTW_INFO("%s, psta==NUL\n", __func__);
4418                 return;
4419         }
4420
4421         if (!(psta->state & _FW_LINKED)) {
4422                 RTW_INFO("%s, psta->state(0x%x) != _FW_LINKED\n", __func__, psta->state);
4423                 return;
4424         }
4425
4426
4427         phtpriv = &psta->htpriv;
4428
4429         if ((phtpriv->ht_option == _TRUE) && (phtpriv->ampdu_enable == _TRUE)) {
4430                 issued = (phtpriv->agg_enable_bitmap >> priority) & 0x1;
4431                 issued |= (phtpriv->candidate_tid_bitmap >> priority) & 0x1;
4432
4433                 if (0 == issued) {
4434                         RTW_INFO("rtw_issue_addbareq_cmd, p=%d\n", priority);
4435                         psta->htpriv.candidate_tid_bitmap |= BIT((u8)priority);
4436                         rtw_addbareq_cmd(padapter, (u8) priority, pattrib->ra);
4437                 }
4438         }
4439
4440 }
4441
4442 void rtw_append_exented_cap(_adapter *padapter, u8 *out_ie, uint *pout_len)
4443 {
4444         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
4445         struct ht_priv          *phtpriv = &pmlmepriv->htpriv;
4446 #ifdef CONFIG_80211AC_VHT
4447         struct vht_priv *pvhtpriv = &pmlmepriv->vhtpriv;
4448 #endif /* CONFIG_80211AC_VHT */
4449         u8      cap_content[8] = { 0 };
4450         u8      *pframe;
4451         u8   null_content[8] = {0};
4452
4453         if (phtpriv->bss_coexist)
4454                 SET_EXT_CAPABILITY_ELE_BSS_COEXIST(cap_content, 1);
4455
4456 #ifdef CONFIG_80211AC_VHT
4457         if (pvhtpriv->vht_option)
4458                 SET_EXT_CAPABILITY_ELE_OP_MODE_NOTIF(cap_content, 1);
4459 #endif /* CONFIG_80211AC_VHT */
4460         /*
4461                 From 802.11 specification,if a STA does not support any of capabilities defined
4462                 in the Extended Capabilities element, then the STA is not required to
4463                 transmit the Extended Capabilities element.
4464         */
4465         if (_FALSE == _rtw_memcmp(cap_content, null_content, 8))
4466                 pframe = rtw_set_ie(out_ie + *pout_len, EID_EXTCapability, 8, cap_content , pout_len);
4467 }
4468 #endif
4469
4470 #ifdef CONFIG_LAYER2_ROAMING
4471 inline void rtw_set_to_roam(_adapter *adapter, u8 to_roam)
4472 {
4473         if (to_roam == 0)
4474                 adapter->mlmepriv.to_join = _FALSE;
4475         adapter->mlmepriv.to_roam = to_roam;
4476 }
4477
4478 inline u8 rtw_dec_to_roam(_adapter *adapter)
4479 {
4480         adapter->mlmepriv.to_roam--;
4481         return adapter->mlmepriv.to_roam;
4482 }
4483
4484 inline u8 rtw_to_roam(_adapter *adapter)
4485 {
4486         return adapter->mlmepriv.to_roam;
4487 }
4488
4489 void rtw_roaming(_adapter *padapter, struct wlan_network *tgt_network)
4490 {
4491         _irqL irqL;
4492         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
4493
4494         _enter_critical_bh(&pmlmepriv->lock, &irqL);
4495         _rtw_roaming(padapter, tgt_network);
4496         _exit_critical_bh(&pmlmepriv->lock, &irqL);
4497 }
4498 void _rtw_roaming(_adapter *padapter, struct wlan_network *tgt_network)
4499 {
4500         struct mlme_priv        *pmlmepriv = &padapter->mlmepriv;
4501         struct wlan_network *cur_network = &pmlmepriv->cur_network;
4502         int do_join_r;
4503
4504         if (0 < rtw_to_roam(padapter)) {
4505                 RTW_INFO("roaming from %s("MAC_FMT"), length:%d\n",
4506                         cur_network->network.Ssid.Ssid, MAC_ARG(cur_network->network.MacAddress),
4507                          cur_network->network.Ssid.SsidLength);
4508                 _rtw_memcpy(&pmlmepriv->assoc_ssid, &cur_network->network.Ssid, sizeof(NDIS_802_11_SSID));
4509
4510                 pmlmepriv->assoc_by_bssid = _FALSE;
4511
4512 #ifdef CONFIG_WAPI_SUPPORT
4513                 rtw_wapi_return_all_sta_info(padapter);
4514 #endif
4515
4516                 while (1) {
4517                         do_join_r = rtw_do_join(padapter);
4518                         if (_SUCCESS == do_join_r)
4519                                 break;
4520                         else {
4521                                 RTW_INFO("roaming do_join return %d\n", do_join_r);
4522                                 rtw_dec_to_roam(padapter);
4523
4524                                 if (rtw_to_roam(padapter) > 0)
4525                                         continue;
4526                                 else {
4527                                         RTW_INFO("%s(%d) -to roaming fail, indicate_disconnect\n", __FUNCTION__, __LINE__);
4528 #ifdef CONFIG_RTW_80211R
4529                                         rtw_clr_ft_flags(padapter, RTW_FT_SUPPORTED|RTW_FT_OVER_DS_SUPPORTED);
4530                                         rtw_reset_ft_status(padapter);
4531 #endif
4532                                         rtw_indicate_disconnect(padapter, 0, _FALSE);
4533                                         break;
4534                                 }
4535                         }
4536                 }
4537         }
4538
4539 }
4540 #endif /* CONFIG_LAYER2_ROAMING */
4541
4542 #ifdef CONFIG_80211N_HT
4543 bool rtw_adjust_chbw(_adapter *adapter, u8 req_ch, u8 *req_bw, u8 *req_offset)
4544 {
4545         struct registry_priv *regsty = adapter_to_regsty(adapter);
4546         u8 allowed_bw;
4547
4548         if (req_ch <= 14)
4549                 allowed_bw = REGSTY_BW_2G(regsty);
4550         else
4551                 allowed_bw = REGSTY_BW_5G(regsty);
4552
4553         allowed_bw = hal_largest_bw(adapter, allowed_bw);
4554
4555         if (allowed_bw == CHANNEL_WIDTH_80 && *req_bw > CHANNEL_WIDTH_80)
4556                 *req_bw = CHANNEL_WIDTH_80;
4557         else if (allowed_bw == CHANNEL_WIDTH_40 && *req_bw > CHANNEL_WIDTH_40)
4558                 *req_bw = CHANNEL_WIDTH_40;
4559         else if (allowed_bw == CHANNEL_WIDTH_20 && *req_bw > CHANNEL_WIDTH_20) {
4560                 *req_bw = CHANNEL_WIDTH_20;
4561                 *req_offset = HAL_PRIME_CHNL_OFFSET_DONT_CARE;
4562         } else
4563                 return _FALSE;
4564
4565         return _TRUE;
4566 }
4567 #endif
4568
4569 sint rtw_linked_check(_adapter *padapter)
4570 {
4571         if ((check_fwstate(&padapter->mlmepriv, WIFI_AP_STATE) == _TRUE) ||
4572             (check_fwstate(&padapter->mlmepriv, WIFI_ADHOC_STATE | WIFI_ADHOC_MASTER_STATE) == _TRUE)) {
4573                 if (padapter->stapriv.asoc_sta_count > 2)
4574                         return _TRUE;
4575         } else {
4576                 /* Station mode */
4577                 if (check_fwstate(&padapter->mlmepriv, _FW_LINKED) == _TRUE)
4578                         return _TRUE;
4579         }
4580         return _FALSE;
4581 }
4582 u8 rtw_is_adapter_up(_adapter *padapter)
4583 {
4584         if (padapter == NULL)
4585                 return _FALSE;
4586
4587         if (RTW_CANNOT_RUN(padapter)) {
4588                 RTW_INFO(FUNC_ADPT_FMT "-(bSurpriseRemoved == _TRUE) || ( bDriverStopped == _TRUE)\n", FUNC_ADPT_ARG(padapter));
4589                 return _FALSE;
4590         }
4591
4592         if (!rtw_is_hw_init_completed(padapter)) {
4593                 /*RTW_INFO(FUNC_ADPT_FMT "-(hw_init_completed == _FALSE)\n", FUNC_ADPT_ARG(padapter));*/
4594                 return _FALSE;
4595         }
4596
4597         if (padapter->bup == _FALSE) {
4598                 /*RTW_INFO(FUNC_ADPT_FMT "-(bup == _FALSE)\n", FUNC_ADPT_ARG(padapter));*/
4599                 return _FALSE;
4600         }
4601
4602         return _TRUE;
4603 }
4604
4605 bool is_miracast_enabled(_adapter *adapter)
4606 {
4607         bool enabled = 0;
4608 #ifdef CONFIG_WFD
4609         struct wifi_display_info *wfdinfo = &adapter->wfd_info;
4610
4611         enabled = (wfdinfo->stack_wfd_mode & (MIRACAST_SOURCE | MIRACAST_SINK))
4612                   || (wfdinfo->op_wfd_mode & (MIRACAST_SOURCE | MIRACAST_SINK));
4613 #endif
4614
4615         return enabled;
4616 }
4617
4618 bool rtw_chk_miracast_mode(_adapter *adapter, u8 mode)
4619 {
4620         bool ret = 0;
4621 #ifdef CONFIG_WFD
4622         struct wifi_display_info *wfdinfo = &adapter->wfd_info;
4623
4624         ret = (wfdinfo->stack_wfd_mode & mode) || (wfdinfo->op_wfd_mode & mode);
4625 #endif
4626
4627         return ret;
4628 }
4629
4630 const char *get_miracast_mode_str(int mode)
4631 {
4632         if (mode == MIRACAST_SOURCE)
4633                 return "SOURCE";
4634         else if (mode == MIRACAST_SINK)
4635                 return "SINK";
4636         else if (mode == (MIRACAST_SOURCE | MIRACAST_SINK))
4637                 return "SOURCE&SINK";
4638         else if (mode == MIRACAST_DISABLED)
4639                 return "DISABLED";
4640         else
4641                 return "INVALID";
4642 }
4643
4644 #ifdef CONFIG_WFD
4645 static bool wfd_st_match_rule(_adapter *adapter, u8 *local_naddr, u8 *local_port, u8 *remote_naddr, u8 *remote_port)
4646 {
4647         struct wifi_display_info *wfdinfo = &adapter->wfd_info;
4648
4649         if (ntohs(*((u16 *)local_port)) == wfdinfo->rtsp_ctrlport
4650             || ntohs(*((u16 *)local_port)) == wfdinfo->tdls_rtsp_ctrlport
4651             || ntohs(*((u16 *)remote_port)) == wfdinfo->peer_rtsp_ctrlport)
4652                 return _TRUE;
4653         return _FALSE;
4654 }
4655
4656 static struct st_register wfd_st_reg = {
4657         .s_proto = 0x06,
4658         .rule = wfd_st_match_rule,
4659 };
4660 #endif /* CONFIG_WFD */
4661
4662 inline void rtw_wfd_st_switch(struct sta_info *sta, bool on)
4663 {
4664 #ifdef CONFIG_WFD
4665         if (on)
4666                 rtw_st_ctl_register(&sta->st_ctl, SESSION_TRACKER_REG_ID_WFD, &wfd_st_reg);
4667         else
4668                 rtw_st_ctl_unregister(&sta->st_ctl, SESSION_TRACKER_REG_ID_WFD);
4669 #endif
4670 }