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