drivers/staging/r8188eu/core/rtw_ap.c

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