drivers/staging/rtl8188eu/core/rtw_ap.c

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