drivers/staging/rtl8192e/rtllib_softmac.c

   1 /* IEEE 802.11 SoftMAC layer
   2  * Copyright (c) 2005 Andrea Merello <andreamrl@tiscali.it>
   3  *
   4  * Mostly extracted from the rtl8180-sa2400 driver for the
   5  * in-kernel generic ieee802.11 stack.
   6  *
   7  * Few lines might be stolen from other part of the rtllib
   8  * stack. Copyright who own it's copyright
   9  *
  10  * WPA code stolen from the ipw2200 driver.
  11  * Copyright who own it's copyright.
  12  *
  13  * released under the GPL
  14  */
  15
  16
  17 #include "rtllib.h"
  18
  19 #include <linux/random.h>
  20 #include <linux/delay.h>
  21 #include <linux/version.h>
  22 #include <linux/uaccess.h>
  23 #include "dot11d.h"
  24
  25 short rtllib_is_54g(struct rtllib_network *net)
  26 {
  27         return (net->rates_ex_len > 0) || (net->rates_len > 4);
  28 }
  29
  30 short rtllib_is_shortslot(const struct rtllib_network *net)
  31 {
  32         return net->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME;
  33 }
  34
  35 /* returns the total length needed for pleacing the RATE MFIE
  36  * tag and the EXTENDED RATE MFIE tag if needed.
  37  * It encludes two bytes per tag for the tag itself and its len
  38  */
  39 static unsigned int rtllib_MFIE_rate_len(struct rtllib_device *ieee)
  40 {
  41         unsigned int rate_len = 0;
  42
  43         if (ieee->modulation & RTLLIB_CCK_MODULATION)
  44                 rate_len = RTLLIB_CCK_RATE_LEN + 2;
  45
  46         if (ieee->modulation & RTLLIB_OFDM_MODULATION)
  47
  48                 rate_len += RTLLIB_OFDM_RATE_LEN + 2;
  49
  50         return rate_len;
  51 }
  52
  53 /* pleace the MFIE rate, tag to the memory (double) poined.
  54  * Then it updates the pointer so that
  55  * it points after the new MFIE tag added.
  56  */
  57 static void rtllib_MFIE_Brate(struct rtllib_device *ieee, u8 **tag_p)
  58 {
  59         u8 *tag = *tag_p;
  60
  61         if (ieee->modulation & RTLLIB_CCK_MODULATION) {
  62                 *tag++ = MFIE_TYPE_RATES;
  63                 *tag++ = 4;
  64                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_1MB;
  65                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_2MB;
  66                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_5MB;
  67                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_11MB;
  68         }
  69
  70         /* We may add an option for custom rates that specific HW
  71          * might support */
  72         *tag_p = tag;
  73 }
  74
  75 static void rtllib_MFIE_Grate(struct rtllib_device *ieee, u8 **tag_p)
  76 {
  77         u8 *tag = *tag_p;
  78
  79         if (ieee->modulation & RTLLIB_OFDM_MODULATION) {
  80                 *tag++ = MFIE_TYPE_RATES_EX;
  81                 *tag++ = 8;
  82                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_6MB;
  83                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_9MB;
  84                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_12MB;
  85                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_18MB;
  86                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_24MB;
  87                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_36MB;
  88                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_48MB;
  89                 *tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_54MB;
  90         }
  91         /* We may add an option for custom rates that specific HW might
  92          * support */
  93         *tag_p = tag;
  94 }
  95
  96 static void rtllib_WMM_Info(struct rtllib_device *ieee, u8 **tag_p)
  97 {
  98         u8 *tag = *tag_p;
  99
 100         *tag++ = MFIE_TYPE_GENERIC;
 101         *tag++ = 7;
 102         *tag++ = 0x00;
 103         *tag++ = 0x50;
 104         *tag++ = 0xf2;
 105         *tag++ = 0x02;
 106         *tag++ = 0x00;
 107         *tag++ = 0x01;
 108         *tag++ = MAX_SP_Len;
 109         *tag_p = tag;
 110 }
 111
 112 void rtllib_TURBO_Info(struct rtllib_device *ieee, u8 **tag_p)
 113 {
 114         u8 *tag = *tag_p;
 115
 116         *tag++ = MFIE_TYPE_GENERIC;
 117         *tag++ = 7;
 118         *tag++ = 0x00;
 119         *tag++ = 0xe0;
 120         *tag++ = 0x4c;
 121         *tag++ = 0x01;
 122         *tag++ = 0x02;
 123         *tag++ = 0x11;
 124         *tag++ = 0x00;
 125
 126         *tag_p = tag;
 127         printk(KERN_ALERT "This is enable turbo mode IE process\n");
 128 }
 129
 130 static void enqueue_mgmt(struct rtllib_device *ieee, struct sk_buff *skb)
 131 {
 132         int nh;
 133         nh = (ieee->mgmt_queue_head + 1) % MGMT_QUEUE_NUM;
 134
 135 /*
 136  * if the queue is full but we have newer frames then
 137  * just overwrites the oldest.
 138  *
 139  * if (nh == ieee->mgmt_queue_tail)
 140  *              return -1;
 141  */
 142         ieee->mgmt_queue_head = nh;
 143         ieee->mgmt_queue_ring[nh] = skb;
 144
 145 }
 146
 147 static struct sk_buff *dequeue_mgmt(struct rtllib_device *ieee)
 148 {
 149         struct sk_buff *ret;
 150
 151         if (ieee->mgmt_queue_tail == ieee->mgmt_queue_head)
 152                 return NULL;
 153
 154         ret = ieee->mgmt_queue_ring[ieee->mgmt_queue_tail];
 155
 156         ieee->mgmt_queue_tail =
 157                 (ieee->mgmt_queue_tail+1) % MGMT_QUEUE_NUM;
 158
 159         return ret;
 160 }
 161
 162 static void init_mgmt_queue(struct rtllib_device *ieee)
 163 {
 164         ieee->mgmt_queue_tail = ieee->mgmt_queue_head = 0;
 165 }
 166
 167
 168 u8
 169 MgntQuery_TxRateExcludeCCKRates(struct rtllib_device *ieee)
 170 {
 171         u16     i;
 172         u8      QueryRate = 0;
 173         u8      BasicRate;
 174
 175
 176         for (i = 0; i < ieee->current_network.rates_len; i++) {
 177                 BasicRate = ieee->current_network.rates[i]&0x7F;
 178                 if (!rtllib_is_cck_rate(BasicRate)) {
 179                         if (QueryRate == 0) {
 180                                 QueryRate = BasicRate;
 181                         } else {
 182                                 if (BasicRate < QueryRate)
 183                                         QueryRate = BasicRate;
 184                         }
 185                 }
 186         }
 187
 188         if (QueryRate == 0) {
 189                 QueryRate = 12;
 190                 printk(KERN_INFO "No BasicRate found!!\n");
 191         }
 192         return QueryRate;
 193 }
 194
 195 u8 MgntQuery_MgntFrameTxRate(struct rtllib_device *ieee)
 196 {
 197         struct rt_hi_throughput *pHTInfo = ieee->pHTInfo;
 198         u8 rate;
 199
 200         if (pHTInfo->IOTAction & HT_IOT_ACT_MGNT_USE_CCK_6M)
 201                 rate = 0x0c;
 202         else
 203                 rate = ieee->basic_rate & 0x7f;
 204
 205         if (rate == 0) {
 206                 if (ieee->mode == IEEE_A ||
 207                    ieee->mode == IEEE_N_5G ||
 208                    (ieee->mode == IEEE_N_24G && !pHTInfo->bCurSuppCCK))
 209                         rate = 0x0c;
 210                 else
 211                         rate = 0x02;
 212         }
 213
 214         return rate;
 215 }
 216
 217 inline void softmac_mgmt_xmit(struct sk_buff *skb, struct rtllib_device *ieee)
 218 {
 219         unsigned long flags;
 220         short single = ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE;
 221         struct rtllib_hdr_3addr  *header =
 222                 (struct rtllib_hdr_3addr  *) skb->data;
 223
 224         struct cb_desc *tcb_desc = (struct cb_desc *)(skb->cb + 8);
 225         spin_lock_irqsave(&ieee->lock, flags);
 226
 227         /* called with 2nd param 0, no mgmt lock required */
 228         rtllib_sta_wakeup(ieee, 0);
 229
 230         if (header->frame_ctl == RTLLIB_STYPE_BEACON)
 231                 tcb_desc->queue_index = BEACON_QUEUE;
 232         else
 233                 tcb_desc->queue_index = MGNT_QUEUE;
 234
 235         if (ieee->disable_mgnt_queue)
 236                 tcb_desc->queue_index = HIGH_QUEUE;
 237
 238         tcb_desc->data_rate = MgntQuery_MgntFrameTxRate(ieee);
 239         tcb_desc->RATRIndex = 7;
 240         tcb_desc->bTxDisableRateFallBack = 1;
 241         tcb_desc->bTxUseDriverAssingedRate = 1;
 242         if (single) {
 243                 if (ieee->queue_stop) {
 244                         enqueue_mgmt(ieee, skb);
 245                 } else {
 246                         header->seq_ctl = cpu_to_le16(ieee->seq_ctrl[0]<<4);
 247
 248                         if (ieee->seq_ctrl[0] == 0xFFF)
 249                                 ieee->seq_ctrl[0] = 0;
 250                         else
 251                                 ieee->seq_ctrl[0]++;
 252
 253                         /* avoid watchdog triggers */
 254                         ieee->softmac_data_hard_start_xmit(skb, ieee->dev,
 255                                                            ieee->basic_rate);
 256                 }
 257
 258                 spin_unlock_irqrestore(&ieee->lock, flags);
 259         } else {
 260                 spin_unlock_irqrestore(&ieee->lock, flags);
 261                 spin_lock_irqsave(&ieee->mgmt_tx_lock, flags);
 262
 263                 header->seq_ctl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
 264
 265                 if (ieee->seq_ctrl[0] == 0xFFF)
 266                         ieee->seq_ctrl[0] = 0;
 267                 else
 268                         ieee->seq_ctrl[0]++;
 269
 270                 /* check wether the managed packet queued greater than 5 */
 271                 if (!ieee->check_nic_enough_desc(ieee->dev, tcb_desc->queue_index) ||
 272                     (skb_queue_len(&ieee->skb_waitQ[tcb_desc->queue_index]) != 0) ||
 273                     (ieee->queue_stop)) {
 274                         /* insert the skb packet to the management queue */
 275                         /* as for the completion function, it does not need
 276                          * to check it any more.
 277                          * */
 278                         printk(KERN_INFO "%s():insert to waitqueue, queue_index"
 279                                ":%d!\n", __func__, tcb_desc->queue_index);
 280                         skb_queue_tail(&ieee->skb_waitQ[tcb_desc->queue_index],
 281                                        skb);
 282                 } else {
 283                         ieee->softmac_hard_start_xmit(skb, ieee->dev);
 284                 }
 285                 spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags);
 286         }
 287 }
 288
 289 inline void softmac_ps_mgmt_xmit(struct sk_buff *skb,
 290                 struct rtllib_device *ieee)
 291 {
 292         short single = ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE;
 293         struct rtllib_hdr_3addr  *header =
 294                 (struct rtllib_hdr_3addr  *) skb->data;
 295         u16 fc, type, stype;
 296         struct cb_desc *tcb_desc = (struct cb_desc *)(skb->cb + 8);
 297
 298         fc = header->frame_ctl;
 299         type = WLAN_FC_GET_TYPE(fc);
 300         stype = WLAN_FC_GET_STYPE(fc);
 301
 302
 303         if (stype != RTLLIB_STYPE_PSPOLL)
 304                 tcb_desc->queue_index = MGNT_QUEUE;
 305         else
 306                 tcb_desc->queue_index = HIGH_QUEUE;
 307
 308         if (ieee->disable_mgnt_queue)
 309                 tcb_desc->queue_index = HIGH_QUEUE;
 310
 311
 312         tcb_desc->data_rate = MgntQuery_MgntFrameTxRate(ieee);
 313         tcb_desc->RATRIndex = 7;
 314         tcb_desc->bTxDisableRateFallBack = 1;
 315         tcb_desc->bTxUseDriverAssingedRate = 1;
 316         if (single) {
 317                 if (type != RTLLIB_FTYPE_CTL) {
 318                         header->seq_ctl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
 319
 320                         if (ieee->seq_ctrl[0] == 0xFFF)
 321                                 ieee->seq_ctrl[0] = 0;
 322                         else
 323                                 ieee->seq_ctrl[0]++;
 324
 325                 }
 326                 /* avoid watchdog triggers */
 327                 ieee->softmac_data_hard_start_xmit(skb, ieee->dev,
 328                                                    ieee->basic_rate);
 329
 330         } else {
 331                 if (type != RTLLIB_FTYPE_CTL) {
 332                         header->seq_ctl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
 333
 334                         if (ieee->seq_ctrl[0] == 0xFFF)
 335                                 ieee->seq_ctrl[0] = 0;
 336                         else
 337                                 ieee->seq_ctrl[0]++;
 338                 }
 339                 ieee->softmac_hard_start_xmit(skb, ieee->dev);
 340
 341         }
 342 }
 343
 344 inline struct sk_buff *rtllib_probe_req(struct rtllib_device *ieee)
 345 {
 346         unsigned int len, rate_len;
 347         u8 *tag;
 348         struct sk_buff *skb;
 349         struct rtllib_probe_request *req;
 350
 351         len = ieee->current_network.ssid_len;
 352
 353         rate_len = rtllib_MFIE_rate_len(ieee);
 354
 355         skb = dev_alloc_skb(sizeof(struct rtllib_probe_request) +
 356                             2 + len + rate_len + ieee->tx_headroom);
 357
 358         if (!skb)
 359                 return NULL;
 360
 361         skb_reserve(skb, ieee->tx_headroom);
 362
 363         req = (struct rtllib_probe_request *) skb_put(skb,
 364               sizeof(struct rtllib_probe_request));
 365         req->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_PROBE_REQ);
 366         req->header.duration_id = 0;
 367
 368         memset(req->header.addr1, 0xff, ETH_ALEN);
 369         memcpy(req->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
 370         memset(req->header.addr3, 0xff, ETH_ALEN);
 371
 372         tag = (u8 *) skb_put(skb, len + 2 + rate_len);
 373
 374         *tag++ = MFIE_TYPE_SSID;
 375         *tag++ = len;
 376         memcpy(tag, ieee->current_network.ssid, len);
 377         tag += len;
 378
 379         rtllib_MFIE_Brate(ieee, &tag);
 380         rtllib_MFIE_Grate(ieee, &tag);
 381
 382         return skb;
 383 }
 384
 385 struct sk_buff *rtllib_get_beacon_(struct rtllib_device *ieee);
 386
 387 static void rtllib_send_beacon(struct rtllib_device *ieee)
 388 {
 389         struct sk_buff *skb;
 390         if (!ieee->ieee_up)
 391                 return;
 392         skb = rtllib_get_beacon_(ieee);
 393
 394         if (skb) {
 395                 softmac_mgmt_xmit(skb, ieee);
 396                 ieee->softmac_stats.tx_beacons++;
 397         }
 398
 399         if (ieee->beacon_txing && ieee->ieee_up)
 400                 mod_timer(&ieee->beacon_timer, jiffies +
 401                           (MSECS(ieee->current_network.beacon_interval - 5)));
 402 }
 403
 404
 405 static void rtllib_send_beacon_cb(unsigned long _ieee)
 406 {
 407         struct rtllib_device *ieee =
 408                 (struct rtllib_device *) _ieee;
 409         unsigned long flags;
 410
 411         spin_lock_irqsave(&ieee->beacon_lock, flags);
 412         rtllib_send_beacon(ieee);
 413         spin_unlock_irqrestore(&ieee->beacon_lock, flags);
 414 }
 415
 416 /*
 417  * Description:
 418  *            Enable network monitor mode, all rx packets will be received.
 419  */
 420 void rtllib_EnableNetMonitorMode(struct net_device *dev,
 421                 bool bInitState)
 422 {
 423         struct rtllib_device *ieee = netdev_priv_rsl(dev);
 424
 425         printk(KERN_INFO "========>Enter Monitor Mode\n");
 426
 427         ieee->AllowAllDestAddrHandler(dev, true, !bInitState);
 428 }
 429
 430
 431 /*
 432  *      Description:
 433  *            Disable network network monitor mode, only packets destinated to
 434  *            us will be received.
 435  */
 436 void rtllib_DisableNetMonitorMode(struct net_device *dev,
 437                 bool bInitState)
 438 {
 439         struct rtllib_device *ieee = netdev_priv_rsl(dev);
 440
 441         printk(KERN_INFO "========>Exit Monitor Mode\n");
 442
 443         ieee->AllowAllDestAddrHandler(dev, false, !bInitState);
 444 }
 445
 446
 447 /*
 448  * Description:
 449  * This enables the specialized promiscuous mode required by Intel.
 450  * In this mode, Intel intends to hear traffics from/to other STAs in the
 451  * same BSS. Therefore we don't have to disable checking BSSID and we only need
 452  * to allow all dest. BUT: if we enable checking BSSID then we can't recv
 453  * packets from other STA.
 454  */
 455 void rtllib_EnableIntelPromiscuousMode(struct net_device *dev,
 456                 bool bInitState)
 457 {
 458         bool bFilterOutNonAssociatedBSSID = false;
 459
 460         struct rtllib_device *ieee = netdev_priv_rsl(dev);
 461
 462         printk(KERN_INFO "========>Enter Intel Promiscuous Mode\n");
 463
 464         ieee->AllowAllDestAddrHandler(dev, true, !bInitState);
 465         ieee->SetHwRegHandler(dev, HW_VAR_CECHK_BSSID,
 466                              (u8 *)&bFilterOutNonAssociatedBSSID);
 467
 468         ieee->bNetPromiscuousMode = true;
 469 }
 470 EXPORT_SYMBOL(rtllib_EnableIntelPromiscuousMode);
 471
 472
 473 /*
 474  * Description:
 475  *            This disables the specialized promiscuous mode required by Intel.
 476  *            See MgntEnableIntelPromiscuousMode for detail.
 477  */
 478 void rtllib_DisableIntelPromiscuousMode(struct net_device *dev,
 479                 bool bInitState)
 480 {
 481         bool bFilterOutNonAssociatedBSSID = true;
 482
 483         struct rtllib_device *ieee = netdev_priv_rsl(dev);
 484
 485         printk(KERN_INFO "========>Exit Intel Promiscuous Mode\n");
 486
 487         ieee->AllowAllDestAddrHandler(dev, false, !bInitState);
 488         ieee->SetHwRegHandler(dev, HW_VAR_CECHK_BSSID,
 489                              (u8 *)&bFilterOutNonAssociatedBSSID);
 490
 491         ieee->bNetPromiscuousMode = false;
 492 }
 493 EXPORT_SYMBOL(rtllib_DisableIntelPromiscuousMode);
 494
 495 static void rtllib_send_probe(struct rtllib_device *ieee, u8 is_mesh)
 496 {
 497         struct sk_buff *skb;
 498         skb = rtllib_probe_req(ieee);
 499         if (skb) {
 500                 softmac_mgmt_xmit(skb, ieee);
 501                 ieee->softmac_stats.tx_probe_rq++;
 502         }
 503 }
 504
 505
 506 void rtllib_send_probe_requests(struct rtllib_device *ieee, u8 is_mesh)
 507 {
 508         if (ieee->active_scan && (ieee->softmac_features &
 509             IEEE_SOFTMAC_PROBERQ)) {
 510                 rtllib_send_probe(ieee, 0);
 511                 rtllib_send_probe(ieee, 0);
 512         }
 513 }
 514
 515 static void rtllib_softmac_hint11d_wq(void *data)
 516 {
 517 }
 518
 519 void rtllib_update_active_chan_map(struct rtllib_device *ieee)
 520 {
 521         memcpy(ieee->active_channel_map, GET_DOT11D_INFO(ieee)->channel_map,
 522                MAX_CHANNEL_NUMBER+1);
 523 }
 524
 525 /* this performs syncro scan blocking the caller until all channels
 526  * in the allowed channel map has been checked.
 527  */
 528 void rtllib_softmac_scan_syncro(struct rtllib_device *ieee, u8 is_mesh)
 529 {
 530         union iwreq_data wrqu;
 531         short ch = 0;
 532
 533         rtllib_update_active_chan_map(ieee);
 534
 535         ieee->be_scan_inprogress = true;
 536
 537         down(&ieee->scan_sem);
 538
 539         while (1) {
 540                 do {
 541                         ch++;
 542                         if (ch > MAX_CHANNEL_NUMBER)
 543                                 goto out; /* scan completed */
 544                 } while (!ieee->active_channel_map[ch]);
 545
 546                 /* this fuction can be called in two situations
 547                  * 1- We have switched to ad-hoc mode and we are
 548                  *    performing a complete syncro scan before conclude
 549                  *    there are no interesting cell and to create a
 550                  *    new one. In this case the link state is
 551                  *    RTLLIB_NOLINK until we found an interesting cell.
 552                  *    If so the ieee8021_new_net, called by the RX path
 553                  *    will set the state to RTLLIB_LINKED, so we stop
 554                  *    scanning
 555                  * 2- We are linked and the root uses run iwlist scan.
 556                  *    So we switch to RTLLIB_LINKED_SCANNING to remember
 557                  *    that we are still logically linked (not interested in
 558                  *    new network events, despite for updating the net list,
 559                  *    but we are temporarly 'unlinked' as the driver shall
 560                  *    not filter RX frames and the channel is changing.
 561                  * So the only situation in witch are interested is to check
 562                  * if the state become LINKED because of the #1 situation
 563                  */
 564
 565                 if (ieee->state == RTLLIB_LINKED)
 566                         goto out;
 567                 if (ieee->sync_scan_hurryup) {
 568                         printk(KERN_INFO "============>sync_scan_hurryup out\n");
 569                         goto out;
 570                 }
 571
 572                 ieee->set_chan(ieee->dev, ch);
 573                 if (ieee->active_channel_map[ch] == 1)
 574                         rtllib_send_probe_requests(ieee, 0);
 575
 576                 /* this prevent excessive time wait when we
 577                  * need to wait for a syncro scan to end..
 578                  */
 579                 msleep_interruptible_rsl(RTLLIB_SOFTMAC_SCAN_TIME);
 580         }
 581 out:
 582         ieee->actscanning = false;
 583         ieee->sync_scan_hurryup = 0;
 584
 585         if (ieee->state >= RTLLIB_LINKED) {
 586                 if (IS_DOT11D_ENABLE(ieee))
 587                         DOT11D_ScanComplete(ieee);
 588         }
 589         up(&ieee->scan_sem);
 590
 591         ieee->be_scan_inprogress = false;
 592
 593         memset(&wrqu, 0, sizeof(wrqu));
 594         wireless_send_event(ieee->dev, SIOCGIWSCAN, &wrqu, NULL);
 595 }
 596
 597 static void rtllib_softmac_scan_wq(void *data)
 598 {
 599         struct rtllib_device *ieee = container_of_dwork_rsl(data,
 600                                      struct rtllib_device, softmac_scan_wq);
 601         u8 last_channel = ieee->current_network.channel;
 602
 603         rtllib_update_active_chan_map(ieee);
 604
 605         if (!ieee->ieee_up)
 606                 return;
 607         if (rtllib_act_scanning(ieee, true) == true)
 608                 return;
 609
 610         down(&ieee->scan_sem);
 611
 612         if (ieee->eRFPowerState == eRfOff) {
 613                 printk(KERN_INFO "======>%s():rf state is eRfOff, return\n",
 614                        __func__);
 615                 goto out1;
 616         }
 617
 618         do {
 619                 ieee->current_network.channel =
 620                         (ieee->current_network.channel + 1) %
 621                         MAX_CHANNEL_NUMBER;
 622                 if (ieee->scan_watch_dog++ > MAX_CHANNEL_NUMBER) {
 623                         if (!ieee->active_channel_map[ieee->current_network.channel])
 624                                 ieee->current_network.channel = 6;
 625                         goto out; /* no good chans */
 626                 }
 627         } while (!ieee->active_channel_map[ieee->current_network.channel]);
 628
 629         if (ieee->scanning_continue == 0)
 630                 goto out;
 631
 632         ieee->set_chan(ieee->dev, ieee->current_network.channel);
 633
 634         if (ieee->active_channel_map[ieee->current_network.channel] == 1)
 635                 rtllib_send_probe_requests(ieee, 0);
 636
 637         queue_delayed_work_rsl(ieee->wq, &ieee->softmac_scan_wq,
 638                                MSECS(RTLLIB_SOFTMAC_SCAN_TIME));
 639
 640         up(&ieee->scan_sem);
 641         return;
 642
 643 out:
 644         if (IS_DOT11D_ENABLE(ieee))
 645                 DOT11D_ScanComplete(ieee);
 646         ieee->current_network.channel = last_channel;
 647
 648 out1:
 649         ieee->actscanning = false;
 650         ieee->scan_watch_dog = 0;
 651         ieee->scanning_continue = 0;
 652         up(&ieee->scan_sem);
 653 }
 654
 655
 656
 657 static void rtllib_beacons_start(struct rtllib_device *ieee)
 658 {
 659         unsigned long flags;
 660         spin_lock_irqsave(&ieee->beacon_lock, flags);
 661
 662         ieee->beacon_txing = 1;
 663         rtllib_send_beacon(ieee);
 664
 665         spin_unlock_irqrestore(&ieee->beacon_lock, flags);
 666 }
 667
 668 static void rtllib_beacons_stop(struct rtllib_device *ieee)
 669 {
 670         unsigned long flags;
 671
 672         spin_lock_irqsave(&ieee->beacon_lock, flags);
 673
 674         ieee->beacon_txing = 0;
 675         del_timer_sync(&ieee->beacon_timer);
 676
 677         spin_unlock_irqrestore(&ieee->beacon_lock, flags);
 678
 679 }
 680
 681
 682 void rtllib_stop_send_beacons(struct rtllib_device *ieee)
 683 {
 684         if (ieee->stop_send_beacons)
 685                 ieee->stop_send_beacons(ieee->dev);
 686         if (ieee->softmac_features & IEEE_SOFTMAC_BEACONS)
 687                 rtllib_beacons_stop(ieee);
 688 }
 689 EXPORT_SYMBOL(rtllib_stop_send_beacons);
 690
 691
 692 void rtllib_start_send_beacons(struct rtllib_device *ieee)
 693 {
 694         if (ieee->start_send_beacons)
 695                 ieee->start_send_beacons(ieee->dev);
 696         if (ieee->softmac_features & IEEE_SOFTMAC_BEACONS)
 697                 rtllib_beacons_start(ieee);
 698 }
 699 EXPORT_SYMBOL(rtllib_start_send_beacons);
 700
 701
 702 static void rtllib_softmac_stop_scan(struct rtllib_device *ieee)
 703 {
 704         down(&ieee->scan_sem);
 705         ieee->scan_watch_dog = 0;
 706         if (ieee->scanning_continue == 1) {
 707                 ieee->scanning_continue = 0;
 708                 ieee->actscanning = 0;
 709
 710                 cancel_delayed_work(&ieee->softmac_scan_wq);
 711         }
 712
 713         up(&ieee->scan_sem);
 714 }
 715
 716 void rtllib_stop_scan(struct rtllib_device *ieee)
 717 {
 718         if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
 719                 rtllib_softmac_stop_scan(ieee);
 720         } else {
 721                 if (ieee->rtllib_stop_hw_scan)
 722                         ieee->rtllib_stop_hw_scan(ieee->dev);
 723         }
 724 }
 725 EXPORT_SYMBOL(rtllib_stop_scan);
 726
 727 void rtllib_stop_scan_syncro(struct rtllib_device *ieee)
 728 {
 729         if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
 730                         ieee->sync_scan_hurryup = 1;
 731         } else {
 732                 if (ieee->rtllib_stop_hw_scan)
 733                         ieee->rtllib_stop_hw_scan(ieee->dev);
 734         }
 735 }
 736 EXPORT_SYMBOL(rtllib_stop_scan_syncro);
 737
 738 bool rtllib_act_scanning(struct rtllib_device *ieee, bool sync_scan)
 739 {
 740         if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
 741                 if (sync_scan)
 742                         return ieee->be_scan_inprogress;
 743                 else
 744                         return ieee->actscanning || ieee->be_scan_inprogress;
 745         } else {
 746                 return test_bit(STATUS_SCANNING, &ieee->status);
 747         }
 748 }
 749 EXPORT_SYMBOL(rtllib_act_scanning);
 750
 751 /* called with ieee->lock held */
 752 static void rtllib_start_scan(struct rtllib_device *ieee)
 753 {
 754         RT_TRACE(COMP_DBG, "===>%s()\n", __func__);
 755         if (ieee->rtllib_ips_leave_wq != NULL)
 756                 ieee->rtllib_ips_leave_wq(ieee->dev);
 757
 758         if (IS_DOT11D_ENABLE(ieee)) {
 759                 if (IS_COUNTRY_IE_VALID(ieee))
 760                         RESET_CIE_WATCHDOG(ieee);
 761         }
 762         if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
 763                 if (ieee->scanning_continue == 0) {
 764                         ieee->actscanning = true;
 765                         ieee->scanning_continue = 1;
 766                         queue_delayed_work_rsl(ieee->wq,
 767                                                &ieee->softmac_scan_wq, 0);
 768                 }
 769         } else {
 770                 if (ieee->rtllib_start_hw_scan)
 771                         ieee->rtllib_start_hw_scan(ieee->dev);
 772         }
 773 }
 774
 775 /* called with wx_sem held */
 776 void rtllib_start_scan_syncro(struct rtllib_device *ieee, u8 is_mesh)
 777 {
 778         if (IS_DOT11D_ENABLE(ieee)) {
 779                 if (IS_COUNTRY_IE_VALID(ieee))
 780                         RESET_CIE_WATCHDOG(ieee);
 781         }
 782         ieee->sync_scan_hurryup = 0;
 783         if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
 784                 rtllib_softmac_scan_syncro(ieee, is_mesh);
 785         } else {
 786                 if (ieee->rtllib_start_hw_scan)
 787                         ieee->rtllib_start_hw_scan(ieee->dev);
 788         }
 789 }
 790 EXPORT_SYMBOL(rtllib_start_scan_syncro);
 791
 792 inline struct sk_buff *rtllib_authentication_req(struct rtllib_network *beacon,
 793         struct rtllib_device *ieee, int challengelen, u8 *daddr)
 794 {
 795         struct sk_buff *skb;
 796         struct rtllib_authentication *auth;
 797         int  len = 0;
 798         len = sizeof(struct rtllib_authentication) + challengelen +
 799                      ieee->tx_headroom + 4;
 800         skb = dev_alloc_skb(len);
 801
 802         if (!skb)
 803                 return NULL;
 804
 805         skb_reserve(skb, ieee->tx_headroom);
 806
 807         auth = (struct rtllib_authentication *)
 808                 skb_put(skb, sizeof(struct rtllib_authentication));
 809
 810         auth->header.frame_ctl = RTLLIB_STYPE_AUTH;
 811         if (challengelen)
 812                 auth->header.frame_ctl |= RTLLIB_FCTL_WEP;
 813
 814         auth->header.duration_id = 0x013a;
 815         memcpy(auth->header.addr1, beacon->bssid, ETH_ALEN);
 816         memcpy(auth->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
 817         memcpy(auth->header.addr3, beacon->bssid, ETH_ALEN);
 818         if (ieee->auth_mode == 0)
 819                 auth->algorithm = WLAN_AUTH_OPEN;
 820         else if (ieee->auth_mode == 1)
 821                 auth->algorithm = WLAN_AUTH_SHARED_KEY;
 822         else if (ieee->auth_mode == 2)
 823                 auth->algorithm = WLAN_AUTH_OPEN;
 824         auth->transaction = cpu_to_le16(ieee->associate_seq);
 825         ieee->associate_seq++;
 826
 827         auth->status = cpu_to_le16(WLAN_STATUS_SUCCESS);
 828
 829         return skb;
 830 }
 831
 832 static struct sk_buff *rtllib_probe_resp(struct rtllib_device *ieee, u8 *dest)
 833 {
 834         u8 *tag;
 835         int beacon_size;
 836         struct rtllib_probe_response *beacon_buf;
 837         struct sk_buff *skb = NULL;
 838         int encrypt;
 839         int atim_len, erp_len;
 840         struct rtllib_crypt_data *crypt;
 841
 842         char *ssid = ieee->current_network.ssid;
 843         int ssid_len = ieee->current_network.ssid_len;
 844         int rate_len = ieee->current_network.rates_len+2;
 845         int rate_ex_len = ieee->current_network.rates_ex_len;
 846         int wpa_ie_len = ieee->wpa_ie_len;
 847         u8 erpinfo_content = 0;
 848
 849         u8 *tmp_ht_cap_buf = NULL;
 850         u8 tmp_ht_cap_len = 0;
 851         u8 *tmp_ht_info_buf = NULL;
 852         u8 tmp_ht_info_len = 0;
 853         struct rt_hi_throughput *pHTInfo = ieee->pHTInfo;
 854         u8 *tmp_generic_ie_buf = NULL;
 855         u8 tmp_generic_ie_len = 0;
 856
 857         if (rate_ex_len > 0)
 858                 rate_ex_len += 2;
 859
 860         if (ieee->current_network.capability & WLAN_CAPABILITY_IBSS)
 861                 atim_len = 4;
 862         else
 863                 atim_len = 0;
 864
 865         if ((ieee->current_network.mode == IEEE_G) ||
 866            (ieee->current_network.mode == IEEE_N_24G &&
 867            ieee->pHTInfo->bCurSuppCCK)) {
 868                 erp_len = 3;
 869                 erpinfo_content = 0;
 870                 if (ieee->current_network.buseprotection)
 871                         erpinfo_content |= ERP_UseProtection;
 872         } else
 873                 erp_len = 0;
 874
 875         crypt = ieee->crypt[ieee->tx_keyidx];
 876         encrypt = ieee->host_encrypt && crypt && crypt->ops &&
 877                 ((0 == strcmp(crypt->ops->name, "WEP") || wpa_ie_len));
 878         if (ieee->pHTInfo->bCurrentHTSupport) {
 879                 tmp_ht_cap_buf = (u8 *) &(ieee->pHTInfo->SelfHTCap);
 880                 tmp_ht_cap_len = sizeof(ieee->pHTInfo->SelfHTCap);
 881                 tmp_ht_info_buf = (u8 *) &(ieee->pHTInfo->SelfHTInfo);
 882                 tmp_ht_info_len = sizeof(ieee->pHTInfo->SelfHTInfo);
 883                 HTConstructCapabilityElement(ieee, tmp_ht_cap_buf,
 884                                              &tmp_ht_cap_len, encrypt, false);
 885                 HTConstructInfoElement(ieee, tmp_ht_info_buf, &tmp_ht_info_len,
 886                                        encrypt);
 887
 888                 if (pHTInfo->bRegRT2RTAggregation) {
 889                         tmp_generic_ie_buf = ieee->pHTInfo->szRT2RTAggBuffer;
 890                         tmp_generic_ie_len =
 891                                  sizeof(ieee->pHTInfo->szRT2RTAggBuffer);
 892                         HTConstructRT2RTAggElement(ieee, tmp_generic_ie_buf,
 893                                                    &tmp_generic_ie_len);
 894                 }
 895         }
 896
 897         beacon_size = sizeof(struct rtllib_probe_response)+2+
 898                 ssid_len + 3 + rate_len + rate_ex_len + atim_len + erp_len
 899                 + wpa_ie_len + ieee->tx_headroom;
 900         skb = dev_alloc_skb(beacon_size);
 901         if (!skb)
 902                 return NULL;
 903
 904         skb_reserve(skb, ieee->tx_headroom);
 905
 906         beacon_buf = (struct rtllib_probe_response *) skb_put(skb,
 907                      (beacon_size - ieee->tx_headroom));
 908         memcpy(beacon_buf->header.addr1, dest, ETH_ALEN);
 909         memcpy(beacon_buf->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
 910         memcpy(beacon_buf->header.addr3, ieee->current_network.bssid, ETH_ALEN);
 911
 912         beacon_buf->header.duration_id = 0;
 913         beacon_buf->beacon_interval =
 914                 cpu_to_le16(ieee->current_network.beacon_interval);
 915         beacon_buf->capability =
 916                 cpu_to_le16(ieee->current_network.capability &
 917                 WLAN_CAPABILITY_IBSS);
 918         beacon_buf->capability |=
 919                 cpu_to_le16(ieee->current_network.capability &
 920                 WLAN_CAPABILITY_SHORT_PREAMBLE);
 921
 922         if (ieee->short_slot && (ieee->current_network.capability &
 923             WLAN_CAPABILITY_SHORT_SLOT_TIME))
 924                 cpu_to_le16((beacon_buf->capability |=
 925                                  WLAN_CAPABILITY_SHORT_SLOT_TIME));
 926
 927         crypt = ieee->crypt[ieee->tx_keyidx];
 928         if (encrypt)
 929                 beacon_buf->capability |= cpu_to_le16(WLAN_CAPABILITY_PRIVACY);
 930
 931
 932         beacon_buf->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_PROBE_RESP);
 933         beacon_buf->info_element[0].id = MFIE_TYPE_SSID;
 934         beacon_buf->info_element[0].len = ssid_len;
 935
 936         tag = (u8 *) beacon_buf->info_element[0].data;
 937
 938         memcpy(tag, ssid, ssid_len);
 939
 940         tag += ssid_len;
 941
 942         *(tag++) = MFIE_TYPE_RATES;
 943         *(tag++) = rate_len-2;
 944         memcpy(tag, ieee->current_network.rates, rate_len-2);
 945         tag += rate_len-2;
 946
 947         *(tag++) = MFIE_TYPE_DS_SET;
 948         *(tag++) = 1;
 949         *(tag++) = ieee->current_network.channel;
 950
 951         if (atim_len) {
 952                 u16 val16;
 953                 *(tag++) = MFIE_TYPE_IBSS_SET;
 954                 *(tag++) = 2;
 955                  val16 = cpu_to_le16(ieee->current_network.atim_window);
 956                 memcpy((u8 *)tag, (u8 *)&val16, 2);
 957                 tag += 2;
 958         }
 959
 960         if (erp_len) {
 961                 *(tag++) = MFIE_TYPE_ERP;
 962                 *(tag++) = 1;
 963                 *(tag++) = erpinfo_content;
 964         }
 965         if (rate_ex_len) {
 966                 *(tag++) = MFIE_TYPE_RATES_EX;
 967                 *(tag++) = rate_ex_len-2;
 968                 memcpy(tag, ieee->current_network.rates_ex, rate_ex_len-2);
 969                 tag += rate_ex_len-2;
 970         }
 971
 972         if (wpa_ie_len) {
 973                 if (ieee->iw_mode == IW_MODE_ADHOC)
 974                         memcpy(&ieee->wpa_ie[14], &ieee->wpa_ie[8], 4);
 975                 memcpy(tag, ieee->wpa_ie, ieee->wpa_ie_len);
 976                 tag += ieee->wpa_ie_len;
 977         }
 978         return skb;
 979 }
 980
 981 static struct sk_buff *rtllib_assoc_resp(struct rtllib_device *ieee, u8 *dest)
 982 {
 983         struct sk_buff *skb;
 984         u8 *tag;
 985
 986         struct rtllib_crypt_data *crypt;
 987         struct rtllib_assoc_response_frame *assoc;
 988         short encrypt;
 989
 990         unsigned int rate_len = rtllib_MFIE_rate_len(ieee);
 991         int len = sizeof(struct rtllib_assoc_response_frame) + rate_len +
 992                   ieee->tx_headroom;
 993
 994         skb = dev_alloc_skb(len);
 995
 996         if (!skb)
 997                 return NULL;
 998
 999         skb_reserve(skb, ieee->tx_headroom);
1000
1001         assoc = (struct rtllib_assoc_response_frame *)
1002                 skb_put(skb, sizeof(struct rtllib_assoc_response_frame));
1003
1004         assoc->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_ASSOC_RESP);
1005         memcpy(assoc->header.addr1, dest, ETH_ALEN);
1006         memcpy(assoc->header.addr3, ieee->dev->dev_addr, ETH_ALEN);
1007         memcpy(assoc->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
1008         assoc->capability = cpu_to_le16(ieee->iw_mode == IW_MODE_MASTER ?
1009                 WLAN_CAPABILITY_ESS : WLAN_CAPABILITY_IBSS);
1010
1011
1012         if (ieee->short_slot)
1013                 assoc->capability |=
1014                                  cpu_to_le16(WLAN_CAPABILITY_SHORT_SLOT_TIME);
1015
1016         if (ieee->host_encrypt)
1017                 crypt = ieee->crypt[ieee->tx_keyidx];
1018         else
1019                 crypt = NULL;
1020
1021         encrypt = (crypt && crypt->ops);
1022
1023         if (encrypt)
1024                 assoc->capability |= cpu_to_le16(WLAN_CAPABILITY_PRIVACY);
1025
1026         assoc->status = 0;
1027         assoc->aid = cpu_to_le16(ieee->assoc_id);
1028         if (ieee->assoc_id == 0x2007)
1029                 ieee->assoc_id = 0;
1030         else
1031                 ieee->assoc_id++;
1032
1033         tag = (u8 *) skb_put(skb, rate_len);
1034         rtllib_MFIE_Brate(ieee, &tag);
1035         rtllib_MFIE_Grate(ieee, &tag);
1036
1037         return skb;
1038 }
1039
1040 static struct sk_buff *rtllib_auth_resp(struct rtllib_device *ieee, int status,
1041                                  u8 *dest)
1042 {
1043         struct sk_buff *skb = NULL;
1044         struct rtllib_authentication *auth;
1045         int len = ieee->tx_headroom + sizeof(struct rtllib_authentication) + 1;
1046         skb = dev_alloc_skb(len);
1047         if (!skb)
1048                 return NULL;
1049
1050         skb->len = sizeof(struct rtllib_authentication);
1051
1052         skb_reserve(skb, ieee->tx_headroom);
1053
1054         auth = (struct rtllib_authentication *)
1055                 skb_put(skb, sizeof(struct rtllib_authentication));
1056
1057         auth->status = cpu_to_le16(status);
1058         auth->transaction = cpu_to_le16(2);
1059         auth->algorithm = cpu_to_le16(WLAN_AUTH_OPEN);
1060
1061         memcpy(auth->header.addr3, ieee->dev->dev_addr, ETH_ALEN);
1062         memcpy(auth->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
1063         memcpy(auth->header.addr1, dest, ETH_ALEN);
1064         auth->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_AUTH);
1065         return skb;
1066
1067
1068 }
1069
1070 static struct sk_buff *rtllib_null_func(struct rtllib_device *ieee, short pwr)
1071 {
1072         struct sk_buff *skb;
1073         struct rtllib_hdr_3addr *hdr;
1074
1075         skb = dev_alloc_skb(sizeof(struct rtllib_hdr_3addr)+ieee->tx_headroom);
1076         if (!skb)
1077                 return NULL;
1078
1079         skb_reserve(skb, ieee->tx_headroom);
1080
1081         hdr = (struct rtllib_hdr_3addr *)skb_put(skb,
1082               sizeof(struct rtllib_hdr_3addr));
1083
1084         memcpy(hdr->addr1, ieee->current_network.bssid, ETH_ALEN);
1085         memcpy(hdr->addr2, ieee->dev->dev_addr, ETH_ALEN);
1086         memcpy(hdr->addr3, ieee->current_network.bssid, ETH_ALEN);
1087
1088         hdr->frame_ctl = cpu_to_le16(RTLLIB_FTYPE_DATA |
1089                 RTLLIB_STYPE_NULLFUNC | RTLLIB_FCTL_TODS |
1090                 (pwr ? RTLLIB_FCTL_PM : 0));
1091
1092         return skb;
1093
1094
1095 }
1096
1097 static struct sk_buff *rtllib_pspoll_func(struct rtllib_device *ieee)
1098 {
1099         struct sk_buff *skb;
1100         struct rtllib_pspoll_hdr *hdr;
1101
1102         skb = dev_alloc_skb(sizeof(struct rtllib_pspoll_hdr)+ieee->tx_headroom);
1103         if (!skb)
1104                 return NULL;
1105
1106         skb_reserve(skb, ieee->tx_headroom);
1107
1108         hdr = (struct rtllib_pspoll_hdr *)skb_put(skb,
1109               sizeof(struct rtllib_pspoll_hdr));
1110
1111         memcpy(hdr->bssid, ieee->current_network.bssid, ETH_ALEN);
1112         memcpy(hdr->ta, ieee->dev->dev_addr, ETH_ALEN);
1113
1114         hdr->aid = cpu_to_le16(ieee->assoc_id | 0xc000);
1115         hdr->frame_ctl = cpu_to_le16(RTLLIB_FTYPE_CTL | RTLLIB_STYPE_PSPOLL |
1116                          RTLLIB_FCTL_PM);
1117
1118         return skb;
1119
1120 }
1121
1122 static void rtllib_resp_to_assoc_rq(struct rtllib_device *ieee, u8 *dest)
1123 {
1124         struct sk_buff *buf = rtllib_assoc_resp(ieee, dest);
1125
1126         if (buf)
1127                 softmac_mgmt_xmit(buf, ieee);
1128 }
1129
1130
1131 static void rtllib_resp_to_auth(struct rtllib_device *ieee, int s, u8 *dest)
1132 {
1133         struct sk_buff *buf = rtllib_auth_resp(ieee, s, dest);
1134
1135         if (buf)
1136                 softmac_mgmt_xmit(buf, ieee);
1137 }
1138
1139
1140 static void rtllib_resp_to_probe(struct rtllib_device *ieee, u8 *dest)
1141 {
1142
1143         struct sk_buff *buf = rtllib_probe_resp(ieee, dest);
1144         if (buf)
1145                 softmac_mgmt_xmit(buf, ieee);
1146 }
1147
1148
1149 inline int SecIsInPMKIDList(struct rtllib_device *ieee, u8 *bssid)
1150 {
1151         int i = 0;
1152
1153         do {
1154                 if ((ieee->PMKIDList[i].bUsed) &&
1155                    (memcmp(ieee->PMKIDList[i].Bssid, bssid, ETH_ALEN) == 0))
1156                         break;
1157                 else
1158                         i++;
1159         } while (i < NUM_PMKID_CACHE);
1160
1161         if (i == NUM_PMKID_CACHE)
1162                 i = -1;
1163         return i;
1164 }
1165
1166 inline struct sk_buff *rtllib_association_req(struct rtllib_network *beacon,
1167                                               struct rtllib_device *ieee)
1168 {
1169         struct sk_buff *skb;
1170         struct rtllib_assoc_request_frame *hdr;
1171         u8 *tag, *ies;
1172         int i;
1173         u8 *ht_cap_buf = NULL;
1174         u8 ht_cap_len = 0;
1175         u8 *realtek_ie_buf = NULL;
1176         u8 realtek_ie_len = 0;
1177         int wpa_ie_len = ieee->wpa_ie_len;
1178         int wps_ie_len = ieee->wps_ie_len;
1179         unsigned int ckip_ie_len = 0;
1180         unsigned int ccxrm_ie_len = 0;
1181         unsigned int cxvernum_ie_len = 0;
1182         struct rtllib_crypt_data *crypt;
1183         int encrypt;
1184         int     PMKCacheIdx;
1185
1186         unsigned int rate_len = (beacon->rates_len ?
1187                                 (beacon->rates_len + 2) : 0) +
1188                                 (beacon->rates_ex_len ? (beacon->rates_ex_len) +
1189                                 2 : 0);
1190
1191         unsigned int wmm_info_len = beacon->qos_data.supported ? 9 : 0;
1192         unsigned int turbo_info_len = beacon->Turbo_Enable ? 9 : 0;
1193
1194         int len = 0;
1195         crypt = ieee->crypt[ieee->tx_keyidx];
1196         if (crypt != NULL)
1197                 encrypt = ieee->host_encrypt && crypt && crypt->ops &&
1198                           ((0 == strcmp(crypt->ops->name, "WEP") ||
1199                           wpa_ie_len));
1200         else
1201                 encrypt = 0;
1202
1203         if ((ieee->rtllib_ap_sec_type &&
1204             (ieee->rtllib_ap_sec_type(ieee) & SEC_ALG_TKIP)) ||
1205             (ieee->bForcedBgMode == true)) {
1206                 ieee->pHTInfo->bEnableHT = 0;
1207                 ieee->mode = WIRELESS_MODE_G;
1208         }
1209
1210         if (ieee->pHTInfo->bCurrentHTSupport && ieee->pHTInfo->bEnableHT) {
1211                 ht_cap_buf = (u8 *)&(ieee->pHTInfo->SelfHTCap);
1212                 ht_cap_len = sizeof(ieee->pHTInfo->SelfHTCap);
1213                 HTConstructCapabilityElement(ieee, ht_cap_buf, &ht_cap_len,
1214                                              encrypt, true);
1215                 if (ieee->pHTInfo->bCurrentRT2RTAggregation) {
1216                         realtek_ie_buf = ieee->pHTInfo->szRT2RTAggBuffer;
1217                         realtek_ie_len =
1218                                  sizeof(ieee->pHTInfo->szRT2RTAggBuffer);
1219                         HTConstructRT2RTAggElement(ieee, realtek_ie_buf,
1220                                                    &realtek_ie_len);
1221                 }
1222         }
1223
1224         if (beacon->bCkipSupported)
1225                 ckip_ie_len = 30+2;
1226         if (beacon->bCcxRmEnable)
1227                 ccxrm_ie_len = 6+2;
1228         if (beacon->BssCcxVerNumber >= 2)
1229                 cxvernum_ie_len = 5+2;
1230
1231         PMKCacheIdx = SecIsInPMKIDList(ieee, ieee->current_network.bssid);
1232         if (PMKCacheIdx >= 0) {
1233                 wpa_ie_len += 18;
1234                 printk(KERN_INFO "[PMK cache]: WPA2 IE length: %x\n",
1235                        wpa_ie_len);
1236         }
1237         len = sizeof(struct rtllib_assoc_request_frame) + 2
1238                 + beacon->ssid_len
1239                 + rate_len
1240                 + wpa_ie_len
1241                 + wps_ie_len
1242                 + wmm_info_len
1243                 + turbo_info_len
1244                 + ht_cap_len
1245                 + realtek_ie_len
1246                 + ckip_ie_len
1247                 + ccxrm_ie_len
1248                 + cxvernum_ie_len
1249                 + ieee->tx_headroom;
1250
1251         skb = dev_alloc_skb(len);
1252
1253         if (!skb)
1254                 return NULL;
1255
1256         skb_reserve(skb, ieee->tx_headroom);
1257
1258         hdr = (struct rtllib_assoc_request_frame *)
1259                 skb_put(skb, sizeof(struct rtllib_assoc_request_frame) + 2);
1260
1261
1262         hdr->header.frame_ctl = RTLLIB_STYPE_ASSOC_REQ;
1263         hdr->header.duration_id = 37;
1264         memcpy(hdr->header.addr1, beacon->bssid, ETH_ALEN);
1265         memcpy(hdr->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
1266         memcpy(hdr->header.addr3, beacon->bssid, ETH_ALEN);
1267
1268         memcpy(ieee->ap_mac_addr, beacon->bssid, ETH_ALEN);
1269
1270         hdr->capability = cpu_to_le16(WLAN_CAPABILITY_ESS);
1271         if (beacon->capability & WLAN_CAPABILITY_PRIVACY)
1272                 hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_PRIVACY);
1273
1274         if (beacon->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
1275                 hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_SHORT_PREAMBLE);
1276
1277         if (ieee->short_slot &&
1278            (beacon->capability&WLAN_CAPABILITY_SHORT_SLOT_TIME))
1279                 hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_SHORT_SLOT_TIME);
1280
1281
1282         hdr->listen_interval = beacon->listen_interval;
1283
1284         hdr->info_element[0].id = MFIE_TYPE_SSID;
1285
1286         hdr->info_element[0].len = beacon->ssid_len;
1287         tag = skb_put(skb, beacon->ssid_len);
1288         memcpy(tag, beacon->ssid, beacon->ssid_len);
1289
1290         tag = skb_put(skb, rate_len);
1291
1292         if (beacon->rates_len) {
1293                 *tag++ = MFIE_TYPE_RATES;
1294                 *tag++ = beacon->rates_len;
1295                 for (i = 0; i < beacon->rates_len; i++)
1296                         *tag++ = beacon->rates[i];
1297         }
1298
1299         if (beacon->rates_ex_len) {
1300                 *tag++ = MFIE_TYPE_RATES_EX;
1301                 *tag++ = beacon->rates_ex_len;
1302                 for (i = 0; i < beacon->rates_ex_len; i++)
1303                         *tag++ = beacon->rates_ex[i];
1304         }
1305
1306         if (beacon->bCkipSupported) {
1307                 static u8       AironetIeOui[] = {0x00, 0x01, 0x66};
1308                 u8      CcxAironetBuf[30];
1309                 struct octet_string osCcxAironetIE;
1310
1311                 memset(CcxAironetBuf, 0, 30);
1312                 osCcxAironetIE.Octet = CcxAironetBuf;
1313                 osCcxAironetIE.Length = sizeof(CcxAironetBuf);
1314                 memcpy(osCcxAironetIE.Octet, AironetIeOui,
1315                        sizeof(AironetIeOui));
1316
1317                 osCcxAironetIE.Octet[IE_CISCO_FLAG_POSITION] |=
1318                                          (SUPPORT_CKIP_PK|SUPPORT_CKIP_MIC);
1319                 tag = skb_put(skb, ckip_ie_len);
1320                 *tag++ = MFIE_TYPE_AIRONET;
1321                 *tag++ = osCcxAironetIE.Length;
1322                 memcpy(tag, osCcxAironetIE.Octet, osCcxAironetIE.Length);
1323                 tag += osCcxAironetIE.Length;
1324         }
1325
1326         if (beacon->bCcxRmEnable) {
1327                 static u8 CcxRmCapBuf[] = {0x00, 0x40, 0x96, 0x01, 0x01, 0x00};
1328                 struct octet_string osCcxRmCap;
1329
1330                 osCcxRmCap.Octet = CcxRmCapBuf;
1331                 osCcxRmCap.Length = sizeof(CcxRmCapBuf);
1332                 tag = skb_put(skb, ccxrm_ie_len);
1333                 *tag++ = MFIE_TYPE_GENERIC;
1334                 *tag++ = osCcxRmCap.Length;
1335                 memcpy(tag, osCcxRmCap.Octet, osCcxRmCap.Length);
1336                 tag += osCcxRmCap.Length;
1337         }
1338
1339         if (beacon->BssCcxVerNumber >= 2) {
1340                 u8 CcxVerNumBuf[] = {0x00, 0x40, 0x96, 0x03, 0x00};
1341                 struct octet_string osCcxVerNum;
1342                 CcxVerNumBuf[4] = beacon->BssCcxVerNumber;
1343                 osCcxVerNum.Octet = CcxVerNumBuf;
1344                 osCcxVerNum.Length = sizeof(CcxVerNumBuf);
1345                 tag = skb_put(skb, cxvernum_ie_len);
1346                 *tag++ = MFIE_TYPE_GENERIC;
1347                 *tag++ = osCcxVerNum.Length;
1348                 memcpy(tag, osCcxVerNum.Octet, osCcxVerNum.Length);
1349                 tag += osCcxVerNum.Length;
1350         }
1351         if (ieee->pHTInfo->bCurrentHTSupport && ieee->pHTInfo->bEnableHT) {
1352                 if (ieee->pHTInfo->ePeerHTSpecVer != HT_SPEC_VER_EWC) {
1353                         tag = skb_put(skb, ht_cap_len);
1354                         *tag++ = MFIE_TYPE_HT_CAP;
1355                         *tag++ = ht_cap_len - 2;
1356                         memcpy(tag, ht_cap_buf, ht_cap_len - 2);
1357                         tag += ht_cap_len - 2;
1358                 }
1359         }
1360
1361         if (wpa_ie_len) {
1362                 tag = skb_put(skb, ieee->wpa_ie_len);
1363                 memcpy(tag, ieee->wpa_ie, ieee->wpa_ie_len);
1364
1365                 if (PMKCacheIdx >= 0) {
1366                         tag = skb_put(skb, 18);
1367                         *tag = 1;
1368                         *(tag + 1) = 0;
1369                         memcpy((tag + 2), &ieee->PMKIDList[PMKCacheIdx].PMKID,
1370                                16);
1371                 }
1372         }
1373         if (wmm_info_len) {
1374                 tag = skb_put(skb, wmm_info_len);
1375                 rtllib_WMM_Info(ieee, &tag);
1376         }
1377
1378         if (wps_ie_len && ieee->wps_ie) {
1379                 tag = skb_put(skb, wps_ie_len);
1380                 memcpy(tag, ieee->wps_ie, wps_ie_len);
1381         }
1382
1383         tag = skb_put(skb, turbo_info_len);
1384         if (turbo_info_len)
1385                 rtllib_TURBO_Info(ieee, &tag);
1386
1387         if (ieee->pHTInfo->bCurrentHTSupport && ieee->pHTInfo->bEnableHT) {
1388                 if (ieee->pHTInfo->ePeerHTSpecVer == HT_SPEC_VER_EWC) {
1389                         tag = skb_put(skb, ht_cap_len);
1390                         *tag++ = MFIE_TYPE_GENERIC;
1391                         *tag++ = ht_cap_len - 2;
1392                         memcpy(tag, ht_cap_buf, ht_cap_len - 2);
1393                         tag += ht_cap_len - 2;
1394                 }
1395
1396                 if (ieee->pHTInfo->bCurrentRT2RTAggregation) {
1397                         tag = skb_put(skb, realtek_ie_len);
1398                         *tag++ = MFIE_TYPE_GENERIC;
1399                         *tag++ = realtek_ie_len - 2;
1400                         memcpy(tag, realtek_ie_buf, realtek_ie_len - 2);
1401                 }
1402         }
1403
1404         kfree(ieee->assocreq_ies);
1405         ieee->assocreq_ies = NULL;
1406         ies = &(hdr->info_element[0].id);
1407         ieee->assocreq_ies_len = (skb->data + skb->len) - ies;
1408         ieee->assocreq_ies = kmalloc(ieee->assocreq_ies_len, GFP_ATOMIC);
1409         if (ieee->assocreq_ies)
1410                 memcpy(ieee->assocreq_ies, ies, ieee->assocreq_ies_len);
1411         else {
1412                 printk(KERN_INFO "%s()Warning: can't alloc memory for assocreq"
1413                        "_ies\n", __func__);
1414                 ieee->assocreq_ies_len = 0;
1415         }
1416         return skb;
1417 }
1418
1419 void rtllib_associate_abort(struct rtllib_device *ieee)
1420 {
1421
1422         unsigned long flags;
1423         spin_lock_irqsave(&ieee->lock, flags);
1424
1425         ieee->associate_seq++;
1426
1427         /* don't scan, and avoid to have the RX path possibily
1428          * try again to associate. Even do not react to AUTH or
1429          * ASSOC response. Just wait for the retry wq to be scheduled.
1430          * Here we will check if there are good nets to associate
1431          * with, so we retry or just get back to NO_LINK and scanning
1432          */
1433         if (ieee->state == RTLLIB_ASSOCIATING_AUTHENTICATING) {
1434                 RTLLIB_DEBUG_MGMT("Authentication failed\n");
1435                 ieee->softmac_stats.no_auth_rs++;
1436         } else {
1437                 RTLLIB_DEBUG_MGMT("Association failed\n");
1438                 ieee->softmac_stats.no_ass_rs++;
1439         }
1440
1441         ieee->state = RTLLIB_ASSOCIATING_RETRY;
1442
1443         queue_delayed_work_rsl(ieee->wq, &ieee->associate_retry_wq,
1444                            RTLLIB_SOFTMAC_ASSOC_RETRY_TIME);
1445
1446         spin_unlock_irqrestore(&ieee->lock, flags);
1447 }
1448
1449 static void rtllib_associate_abort_cb(unsigned long dev)
1450 {
1451         rtllib_associate_abort((struct rtllib_device *) dev);
1452 }
1453
1454 static void rtllib_associate_step1(struct rtllib_device *ieee, u8 * daddr)
1455 {
1456         struct rtllib_network *beacon = &ieee->current_network;
1457         struct sk_buff *skb;
1458
1459         RTLLIB_DEBUG_MGMT("Stopping scan\n");
1460
1461         ieee->softmac_stats.tx_auth_rq++;
1462
1463         skb = rtllib_authentication_req(beacon, ieee, 0, daddr);
1464
1465         if (!skb)
1466                 rtllib_associate_abort(ieee);
1467         else {
1468                 ieee->state = RTLLIB_ASSOCIATING_AUTHENTICATING ;
1469                 RTLLIB_DEBUG_MGMT("Sending authentication request\n");
1470                 softmac_mgmt_xmit(skb, ieee);
1471                 if (!timer_pending(&ieee->associate_timer)) {
1472                         ieee->associate_timer.expires = jiffies + (HZ / 2);
1473                         add_timer(&ieee->associate_timer);
1474                 }
1475         }
1476 }
1477
1478 static void rtllib_auth_challenge(struct rtllib_device *ieee, u8 *challenge, int chlen)
1479 {
1480         u8 *c;
1481         struct sk_buff *skb;
1482         struct rtllib_network *beacon = &ieee->current_network;
1483
1484         ieee->associate_seq++;
1485         ieee->softmac_stats.tx_auth_rq++;
1486
1487         skb = rtllib_authentication_req(beacon, ieee, chlen + 2, beacon->bssid);
1488
1489         if (!skb)
1490                 rtllib_associate_abort(ieee);
1491         else {
1492                 c = skb_put(skb, chlen+2);
1493                 *(c++) = MFIE_TYPE_CHALLENGE;
1494                 *(c++) = chlen;
1495                 memcpy(c, challenge, chlen);
1496
1497                 RTLLIB_DEBUG_MGMT("Sending authentication challenge "
1498                                   "response\n");
1499
1500                 rtllib_encrypt_fragment(ieee, skb,
1501                                         sizeof(struct rtllib_hdr_3addr));
1502
1503                 softmac_mgmt_xmit(skb, ieee);
1504                 mod_timer(&ieee->associate_timer, jiffies + (HZ/2));
1505         }
1506         kfree(challenge);
1507 }
1508
1509 static void rtllib_associate_step2(struct rtllib_device *ieee)
1510 {
1511         struct sk_buff *skb;
1512         struct rtllib_network *beacon = &ieee->current_network;
1513
1514         del_timer_sync(&ieee->associate_timer);
1515
1516         RTLLIB_DEBUG_MGMT("Sending association request\n");
1517
1518         ieee->softmac_stats.tx_ass_rq++;
1519         skb = rtllib_association_req(beacon, ieee);
1520         if (!skb)
1521                 rtllib_associate_abort(ieee);
1522         else {
1523                 softmac_mgmt_xmit(skb, ieee);
1524                 mod_timer(&ieee->associate_timer, jiffies + (HZ/2));
1525         }
1526 }
1527
1528 #define CANCELLED  2
1529 static void rtllib_associate_complete_wq(void *data)
1530 {
1531         struct rtllib_device *ieee = (struct rtllib_device *)
1532                                      container_of_work_rsl(data,
1533                                      struct rtllib_device,
1534                                      associate_complete_wq);
1535         struct rt_pwr_save_ctrl *pPSC = (struct rt_pwr_save_ctrl *)
1536                                         (&(ieee->PowerSaveControl));
1537         printk(KERN_INFO "Associated successfully\n");
1538         if (ieee->is_silent_reset == 0) {
1539                 printk(KERN_INFO "normal associate\n");
1540                 notify_wx_assoc_event(ieee);
1541         }
1542
1543         netif_carrier_on(ieee->dev);
1544         ieee->is_roaming = false;
1545         if (rtllib_is_54g(&ieee->current_network) &&
1546            (ieee->modulation & RTLLIB_OFDM_MODULATION)) {
1547                 ieee->rate = 108;
1548                 printk(KERN_INFO"Using G rates:%d\n", ieee->rate);
1549         } else {
1550                 ieee->rate = 22;
1551                 ieee->SetWirelessMode(ieee->dev, IEEE_B);
1552                 printk(KERN_INFO"Using B rates:%d\n", ieee->rate);
1553         }
1554         if (ieee->pHTInfo->bCurrentHTSupport && ieee->pHTInfo->bEnableHT) {
1555                 printk(KERN_INFO "Successfully associated, ht enabled\n");
1556                 HTOnAssocRsp(ieee);
1557         } else {
1558                 printk(KERN_INFO "Successfully associated, ht not "
1559                        "enabled(%d, %d)\n",
1560                        ieee->pHTInfo->bCurrentHTSupport,
1561                        ieee->pHTInfo->bEnableHT);
1562                 memset(ieee->dot11HTOperationalRateSet, 0, 16);
1563         }
1564         ieee->LinkDetectInfo.SlotNum = 2 * (1 +
1565                                        ieee->current_network.beacon_interval /
1566                                        500);
1567         if (ieee->LinkDetectInfo.NumRecvBcnInPeriod == 0 ||
1568             ieee->LinkDetectInfo.NumRecvDataInPeriod == 0) {
1569                 ieee->LinkDetectInfo.NumRecvBcnInPeriod = 1;
1570                 ieee->LinkDetectInfo.NumRecvDataInPeriod = 1;
1571         }
1572         pPSC->LpsIdleCount = 0;
1573         ieee->link_change(ieee->dev);
1574
1575         if (ieee->is_silent_reset == 1) {
1576                 printk(KERN_INFO "silent reset associate\n");
1577                 ieee->is_silent_reset = 0;
1578         }
1579
1580         if (ieee->data_hard_resume)
1581                 ieee->data_hard_resume(ieee->dev);
1582
1583 }
1584
1585 static void rtllib_sta_send_associnfo(struct rtllib_device *ieee)
1586 {
1587 }
1588
1589 static void rtllib_associate_complete(struct rtllib_device *ieee)
1590 {
1591         del_timer_sync(&ieee->associate_timer);
1592
1593         ieee->state = RTLLIB_LINKED;
1594         rtllib_sta_send_associnfo(ieee);
1595
1596         queue_work_rsl(ieee->wq, &ieee->associate_complete_wq);
1597 }
1598
1599 static void rtllib_associate_procedure_wq(void *data)
1600 {
1601         struct rtllib_device *ieee = container_of_dwork_rsl(data,
1602                                      struct rtllib_device,
1603                                      associate_procedure_wq);
1604         rtllib_stop_scan_syncro(ieee);
1605         if (ieee->rtllib_ips_leave != NULL)
1606                 ieee->rtllib_ips_leave(ieee->dev);
1607         down(&ieee->wx_sem);
1608
1609         if (ieee->data_hard_stop)
1610                 ieee->data_hard_stop(ieee->dev);
1611
1612         rtllib_stop_scan(ieee);
1613         RT_TRACE(COMP_DBG, "===>%s(), chan:%d\n", __func__,
1614                  ieee->current_network.channel);
1615         HTSetConnectBwMode(ieee, HT_CHANNEL_WIDTH_20, HT_EXTCHNL_OFFSET_NO_EXT);
1616         if (ieee->eRFPowerState == eRfOff) {
1617                 RT_TRACE(COMP_DBG, "=============>%s():Rf state is eRfOff,"
1618                          " schedule ipsleave wq again,return\n", __func__);
1619                 if (ieee->rtllib_ips_leave_wq != NULL)
1620                         ieee->rtllib_ips_leave_wq(ieee->dev);
1621                 up(&ieee->wx_sem);
1622                 return;
1623         }
1624         ieee->associate_seq = 1;
1625
1626         rtllib_associate_step1(ieee, ieee->current_network.bssid);
1627
1628         up(&ieee->wx_sem);
1629 }
1630
1631 inline void rtllib_softmac_new_net(struct rtllib_device *ieee,
1632                                    struct rtllib_network *net)
1633 {
1634         u8 tmp_ssid[IW_ESSID_MAX_SIZE + 1];
1635         int tmp_ssid_len = 0;
1636
1637         short apset, ssidset, ssidbroad, apmatch, ssidmatch;
1638
1639         /* we are interested in new new only if we are not associated
1640          * and we are not associating / authenticating
1641          */
1642         if (ieee->state != RTLLIB_NOLINK)
1643                 return;
1644
1645         if ((ieee->iw_mode == IW_MODE_INFRA) && !(net->capability &
1646             WLAN_CAPABILITY_ESS))
1647                 return;
1648
1649         if ((ieee->iw_mode == IW_MODE_ADHOC) && !(net->capability &
1650              WLAN_CAPABILITY_IBSS))
1651                 return;
1652
1653         if ((ieee->iw_mode == IW_MODE_ADHOC) &&
1654             (net->channel > ieee->ibss_maxjoin_chal))
1655                 return;
1656         if (ieee->iw_mode == IW_MODE_INFRA || ieee->iw_mode == IW_MODE_ADHOC) {
1657                 /* if the user specified the AP MAC, we need also the essid
1658                  * This could be obtained by beacons or, if the network does not
1659                  * broadcast it, it can be put manually.
1660                  */
1661                 apset = ieee->wap_set;
1662                 ssidset = ieee->ssid_set;
1663                 ssidbroad =  !(net->ssid_len == 0 || net->ssid[0] == '\0');
1664                 apmatch = (memcmp(ieee->current_network.bssid, net->bssid,
1665                                   ETH_ALEN) == 0);
1666                 if (!ssidbroad) {
1667                         ssidmatch = (ieee->current_network.ssid_len ==
1668                                     net->hidden_ssid_len) &&
1669                                     (!strncmp(ieee->current_network.ssid,
1670                                     net->hidden_ssid, net->hidden_ssid_len));
1671                         if (net->hidden_ssid_len > 0) {
1672                                 strncpy(net->ssid, net->hidden_ssid,
1673                                         net->hidden_ssid_len);
1674                                 net->ssid_len = net->hidden_ssid_len;
1675                                 ssidbroad = 1;
1676                         }
1677                 } else
1678                         ssidmatch =
1679                            (ieee->current_network.ssid_len == net->ssid_len) &&
1680                            (!strncmp(ieee->current_network.ssid, net->ssid,
1681                            net->ssid_len));
1682
1683                 /* if the user set the AP check if match.
1684                  * if the network does not broadcast essid we check the
1685                  *       user supplyed ANY essid
1686                  * if the network does broadcast and the user does not set
1687                  *       essid it is OK
1688                  * if the network does broadcast and the user did set essid
1689                  * check if essid match
1690                  * if the ap is not set, check that the user set the bssid
1691                  * and the network does bradcast and that those two bssid match
1692                  */
1693                 if ((apset && apmatch &&
1694                    ((ssidset && ssidbroad && ssidmatch) ||
1695                    (ssidbroad && !ssidset) || (!ssidbroad && ssidset))) ||
1696                    (!apset && ssidset && ssidbroad && ssidmatch) ||
1697                    (ieee->is_roaming && ssidset && ssidbroad && ssidmatch)) {
1698                         /* if the essid is hidden replace it with the
1699                         * essid provided by the user.
1700                         */
1701                         if (!ssidbroad) {
1702                                 strncpy(tmp_ssid, ieee->current_network.ssid,
1703                                         IW_ESSID_MAX_SIZE);
1704                                 tmp_ssid_len = ieee->current_network.ssid_len;
1705                         }
1706                         memcpy(&ieee->current_network, net,
1707                                sizeof(struct rtllib_network));
1708                         if (!ssidbroad) {
1709                                 strncpy(ieee->current_network.ssid, tmp_ssid,
1710                                         IW_ESSID_MAX_SIZE);
1711                                 ieee->current_network.ssid_len = tmp_ssid_len;
1712                         }
1713                         printk(KERN_INFO"Linking with %s,channel:%d, qos:%d, "
1714                                "myHT:%d, networkHT:%d, mode:%x cur_net.flags"
1715                                ":0x%x\n", ieee->current_network.ssid,
1716                                ieee->current_network.channel,
1717                                ieee->current_network.qos_data.supported,
1718                                ieee->pHTInfo->bEnableHT,
1719                                ieee->current_network.bssht.bdSupportHT,
1720                                ieee->current_network.mode,
1721                                ieee->current_network.flags);
1722
1723                         if ((rtllib_act_scanning(ieee, false)) &&
1724                            !(ieee->softmac_features & IEEE_SOFTMAC_SCAN))
1725                                 rtllib_stop_scan_syncro(ieee);
1726
1727                         ieee->hwscan_ch_bk = ieee->current_network.channel;
1728                         HTResetIOTSetting(ieee->pHTInfo);
1729                         ieee->wmm_acm = 0;
1730                         if (ieee->iw_mode == IW_MODE_INFRA) {
1731                                 /* Join the network for the first time */
1732                                 ieee->AsocRetryCount = 0;
1733                                 if ((ieee->current_network.qos_data.supported == 1) &&
1734                                    ieee->current_network.bssht.bdSupportHT)
1735                                         HTResetSelfAndSavePeerSetting(ieee,
1736                                                  &(ieee->current_network));
1737                                 else
1738                                         ieee->pHTInfo->bCurrentHTSupport =
1739                                                                  false;
1740
1741                                 ieee->state = RTLLIB_ASSOCIATING;
1742                                 if (ieee->LedControlHandler != NULL)
1743                                         ieee->LedControlHandler(ieee->dev,
1744                                                          LED_CTL_START_TO_LINK);
1745                                 queue_delayed_work_rsl(ieee->wq,
1746                                            &ieee->associate_procedure_wq, 0);
1747                         } else {
1748                                 if (rtllib_is_54g(&ieee->current_network) &&
1749                                         (ieee->modulation & RTLLIB_OFDM_MODULATION)) {
1750                                         ieee->rate = 108;
1751                                         ieee->SetWirelessMode(ieee->dev, IEEE_G);
1752                                         printk(KERN_INFO"Using G rates\n");
1753                                 } else {
1754                                         ieee->rate = 22;
1755                                         ieee->SetWirelessMode(ieee->dev, IEEE_B);
1756                                         printk(KERN_INFO"Using B rates\n");
1757                                 }
1758                                 memset(ieee->dot11HTOperationalRateSet, 0, 16);
1759                                 ieee->state = RTLLIB_LINKED;
1760                         }
1761                 }
1762         }
1763 }
1764
1765 void rtllib_softmac_check_all_nets(struct rtllib_device *ieee)
1766 {
1767         unsigned long flags;
1768         struct rtllib_network *target;
1769
1770         spin_lock_irqsave(&ieee->lock, flags);
1771
1772         list_for_each_entry(target, &ieee->network_list, list) {
1773
1774                 /* if the state become different that NOLINK means
1775                  * we had found what we are searching for
1776                  */
1777
1778                 if (ieee->state != RTLLIB_NOLINK)
1779                         break;
1780
1781                 if (ieee->scan_age == 0 || time_after(target->last_scanned +
1782                     ieee->scan_age, jiffies))
1783                         rtllib_softmac_new_net(ieee, target);
1784         }
1785         spin_unlock_irqrestore(&ieee->lock, flags);
1786 }
1787
1788 static inline u16 auth_parse(struct sk_buff *skb, u8** challenge, int *chlen)
1789 {
1790         struct rtllib_authentication *a;
1791         u8 *t;
1792         if (skb->len <  (sizeof(struct rtllib_authentication) -
1793             sizeof(struct rtllib_info_element))) {
1794                 RTLLIB_DEBUG_MGMT("invalid len in auth resp: %d\n", skb->len);
1795                 return 0xcafe;
1796         }
1797         *challenge = NULL;
1798         a = (struct rtllib_authentication *) skb->data;
1799         if (skb->len > (sizeof(struct rtllib_authentication) + 3)) {
1800                 t = skb->data + sizeof(struct rtllib_authentication);
1801
1802                 if (*(t++) == MFIE_TYPE_CHALLENGE) {
1803                         *chlen = *(t++);
1804                         *challenge = kmalloc(*chlen, GFP_ATOMIC);
1805                         memcpy(*challenge, t, *chlen);  /*TODO - check here*/
1806                 }
1807         }
1808         return cpu_to_le16(a->status);
1809 }
1810
1811 static int auth_rq_parse(struct sk_buff *skb, u8 *dest)
1812 {
1813         struct rtllib_authentication *a;
1814
1815         if (skb->len <  (sizeof(struct rtllib_authentication) -
1816             sizeof(struct rtllib_info_element))) {
1817                 RTLLIB_DEBUG_MGMT("invalid len in auth request: %d\n",
1818                                   skb->len);
1819                 return -1;
1820         }
1821         a = (struct rtllib_authentication *) skb->data;
1822
1823         memcpy(dest, a->header.addr2, ETH_ALEN);
1824
1825         if (le16_to_cpu(a->algorithm) != WLAN_AUTH_OPEN)
1826                 return  WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
1827
1828         return WLAN_STATUS_SUCCESS;
1829 }
1830
1831 static short probe_rq_parse(struct rtllib_device *ieee, struct sk_buff *skb,
1832                             u8 *src)
1833 {
1834         u8 *tag;
1835         u8 *skbend;
1836         u8 *ssid = NULL;
1837         u8 ssidlen = 0;
1838         struct rtllib_hdr_3addr   *header =
1839                 (struct rtllib_hdr_3addr   *) skb->data;
1840         bool bssid_match;
1841
1842         if (skb->len < sizeof(struct rtllib_hdr_3addr))
1843                 return -1; /* corrupted */
1844
1845         bssid_match =
1846           (memcmp(header->addr3, ieee->current_network.bssid, ETH_ALEN) != 0) &&
1847           (memcmp(header->addr3, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0);
1848         if (bssid_match)
1849                 return -1;
1850
1851         memcpy(src, header->addr2, ETH_ALEN);
1852
1853         skbend = (u8 *)skb->data + skb->len;
1854
1855         tag = skb->data + sizeof(struct rtllib_hdr_3addr);
1856
1857         while (tag + 1 < skbend) {
1858                 if (*tag == 0) {
1859                         ssid = tag + 2;
1860                         ssidlen = *(tag + 1);
1861                         break;
1862                 }
1863                 tag++; /* point to the len field */
1864                 tag = tag + *(tag); /* point to the last data byte of the tag */
1865                 tag++; /* point to the next tag */
1866         }
1867
1868         if (ssidlen == 0)
1869                 return 1;
1870
1871         if (!ssid)
1872                 return 1; /* ssid not found in tagged param */
1873
1874         return !strncmp(ssid, ieee->current_network.ssid, ssidlen);
1875 }
1876
1877 static int assoc_rq_parse(struct sk_buff *skb, u8 *dest)
1878 {
1879         struct rtllib_assoc_request_frame *a;
1880
1881         if (skb->len < (sizeof(struct rtllib_assoc_request_frame) -
1882                 sizeof(struct rtllib_info_element))) {
1883
1884                 RTLLIB_DEBUG_MGMT("invalid len in auth request:%d\n", skb->len);
1885                 return -1;
1886         }
1887
1888         a = (struct rtllib_assoc_request_frame *) skb->data;
1889
1890         memcpy(dest, a->header.addr2, ETH_ALEN);
1891
1892         return 0;
1893 }
1894
1895 static inline u16 assoc_parse(struct rtllib_device *ieee, struct sk_buff *skb,
1896                               int *aid)
1897 {
1898         struct rtllib_assoc_response_frame *response_head;
1899         u16 status_code;
1900
1901         if (skb->len <  sizeof(struct rtllib_assoc_response_frame)) {
1902                 RTLLIB_DEBUG_MGMT("invalid len in auth resp: %d\n", skb->len);
1903                 return 0xcafe;
1904         }
1905
1906         response_head = (struct rtllib_assoc_response_frame *) skb->data;
1907         *aid = le16_to_cpu(response_head->aid) & 0x3fff;
1908
1909         status_code = le16_to_cpu(response_head->status);
1910         if ((status_code == WLAN_STATUS_ASSOC_DENIED_RATES ||
1911            status_code == WLAN_STATUS_CAPS_UNSUPPORTED) &&
1912            ((ieee->mode == IEEE_G) &&
1913            (ieee->current_network.mode == IEEE_N_24G) &&
1914            (ieee->AsocRetryCount++ < (RT_ASOC_RETRY_LIMIT-1)))) {
1915                 ieee->pHTInfo->IOTAction |= HT_IOT_ACT_PURE_N_MODE;
1916         } else {
1917                 ieee->AsocRetryCount = 0;
1918         }
1919
1920         return le16_to_cpu(response_head->status);
1921 }
1922
1923 void rtllib_rx_probe_rq(struct rtllib_device *ieee, struct sk_buff *skb)
1924 {
1925         u8 dest[ETH_ALEN];
1926         ieee->softmac_stats.rx_probe_rq++;
1927         if (probe_rq_parse(ieee, skb, dest) > 0) {
1928                 ieee->softmac_stats.tx_probe_rs++;
1929                 rtllib_resp_to_probe(ieee, dest);
1930         }
1931 }
1932
1933 static inline void rtllib_rx_auth_rq(struct rtllib_device *ieee,
1934                                      struct sk_buff *skb)
1935 {
1936         u8 dest[ETH_ALEN];
1937         int status;
1938         ieee->softmac_stats.rx_auth_rq++;
1939
1940         status = auth_rq_parse(skb, dest);
1941         if (status != -1)
1942                 rtllib_resp_to_auth(ieee, status, dest);
1943 }
1944
1945 static inline void rtllib_rx_assoc_rq(struct rtllib_device *ieee,
1946                                       struct sk_buff *skb)
1947 {
1948
1949         u8 dest[ETH_ALEN];
1950
1951         ieee->softmac_stats.rx_ass_rq++;
1952         if (assoc_rq_parse(skb, dest) != -1)
1953                 rtllib_resp_to_assoc_rq(ieee, dest);
1954
1955         printk(KERN_INFO"New client associated: %pM\n", dest);
1956 }
1957
1958 void rtllib_sta_ps_send_null_frame(struct rtllib_device *ieee, short pwr)
1959 {
1960
1961         struct sk_buff *buf = rtllib_null_func(ieee, pwr);
1962
1963         if (buf)
1964                 softmac_ps_mgmt_xmit(buf, ieee);
1965 }
1966 EXPORT_SYMBOL(rtllib_sta_ps_send_null_frame);
1967
1968 void rtllib_sta_ps_send_pspoll_frame(struct rtllib_device *ieee)
1969 {
1970         struct sk_buff *buf = rtllib_pspoll_func(ieee);
1971
1972         if (buf)
1973                 softmac_ps_mgmt_xmit(buf, ieee);
1974 }
1975
1976 static short rtllib_sta_ps_sleep(struct rtllib_device *ieee, u64 *time)
1977 {
1978         int timeout = ieee->ps_timeout;
1979         u8 dtim;
1980         struct rt_pwr_save_ctrl *pPSC = (struct rt_pwr_save_ctrl *)
1981                                         (&(ieee->PowerSaveControl));
1982
1983         if (ieee->LPSDelayCnt) {
1984                 ieee->LPSDelayCnt--;
1985                 return 0;
1986         }
1987
1988         dtim = ieee->current_network.dtim_data;
1989         if (!(dtim & RTLLIB_DTIM_VALID))
1990                 return 0;
1991         timeout = ieee->current_network.beacon_interval;
1992         ieee->current_network.dtim_data = RTLLIB_DTIM_INVALID;
1993         /* there's no need to nofity AP that I find you buffered
1994          * with broadcast packet */
1995         if (dtim & (RTLLIB_DTIM_UCAST & ieee->ps))
1996                 return 2;
1997
1998         if (!time_after(jiffies, ieee->dev->trans_start + MSECS(timeout)))
1999                 return 0;
2000         if (!time_after(jiffies, ieee->last_rx_ps_time + MSECS(timeout)))
2001                 return 0;
2002         if ((ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE) &&
2003             (ieee->mgmt_queue_tail != ieee->mgmt_queue_head))
2004                 return 0;
2005
2006         if (time) {
2007                 if (ieee->bAwakePktSent == true) {
2008                         pPSC->LPSAwakeIntvl = 1;
2009                 } else {
2010                         u8              MaxPeriod = 1;
2011
2012                         if (pPSC->LPSAwakeIntvl == 0)
2013                                 pPSC->LPSAwakeIntvl = 1;
2014                         if (pPSC->RegMaxLPSAwakeIntvl == 0)
2015                                 MaxPeriod = 1;
2016                         else if (pPSC->RegMaxLPSAwakeIntvl == 0xFF)
2017                                 MaxPeriod = ieee->current_network.dtim_period;
2018                         else
2019                                 MaxPeriod = pPSC->RegMaxLPSAwakeIntvl;
2020                         pPSC->LPSAwakeIntvl = (pPSC->LPSAwakeIntvl >=
2021                                                MaxPeriod) ? MaxPeriod :
2022                                                (pPSC->LPSAwakeIntvl + 1);
2023                 }
2024                 {
2025                         u8 LPSAwakeIntvl_tmp = 0;
2026                         u8 period = ieee->current_network.dtim_period;
2027                         u8 count = ieee->current_network.tim.tim_count;
2028                         if (count == 0) {
2029                                 if (pPSC->LPSAwakeIntvl > period)
2030                                         LPSAwakeIntvl_tmp = period +
2031                                                  (pPSC->LPSAwakeIntvl -
2032                                                  period) -
2033                                                  ((pPSC->LPSAwakeIntvl-period) %
2034                                                  period);
2035                                 else
2036                                         LPSAwakeIntvl_tmp = pPSC->LPSAwakeIntvl;
2037
2038                         } else {
2039                                 if (pPSC->LPSAwakeIntvl >
2040                                     ieee->current_network.tim.tim_count)
2041                                         LPSAwakeIntvl_tmp = count +
2042                                         (pPSC->LPSAwakeIntvl - count) -
2043                                         ((pPSC->LPSAwakeIntvl-count)%period);
2044                                 else
2045                                         LPSAwakeIntvl_tmp = pPSC->LPSAwakeIntvl;
2046                         }
2047
2048                 *time = ieee->current_network.last_dtim_sta_time
2049                         + MSECS(ieee->current_network.beacon_interval *
2050                         LPSAwakeIntvl_tmp);
2051         }
2052         }
2053
2054         return 1;
2055
2056
2057 }
2058
2059 static inline void rtllib_sta_ps(struct rtllib_device *ieee)
2060 {
2061         u64 time;
2062         short sleep;
2063         unsigned long flags, flags2;
2064
2065         spin_lock_irqsave(&ieee->lock, flags);
2066
2067         if ((ieee->ps == RTLLIB_PS_DISABLED ||
2068              ieee->iw_mode != IW_MODE_INFRA ||
2069              ieee->state != RTLLIB_LINKED)) {
2070                 RT_TRACE(COMP_DBG, "=====>%s(): no need to ps,wake up!! "
2071                          "ieee->ps is %d, ieee->iw_mode is %d, ieee->state"
2072                          " is %d\n", __func__, ieee->ps, ieee->iw_mode,
2073                           ieee->state);
2074                 spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
2075                 rtllib_sta_wakeup(ieee, 1);
2076
2077                 spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
2078         }
2079         sleep = rtllib_sta_ps_sleep(ieee, &time);
2080         /* 2 wake, 1 sleep, 0 do nothing */
2081         if (sleep == 0)
2082                 goto out;
2083         if (sleep == 1) {
2084                 if (ieee->sta_sleep == LPS_IS_SLEEP) {
2085                         ieee->enter_sleep_state(ieee->dev, time);
2086                 } else if (ieee->sta_sleep == LPS_IS_WAKE) {
2087                         spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
2088
2089                         if (ieee->ps_is_queue_empty(ieee->dev)) {
2090                                 ieee->sta_sleep = LPS_WAIT_NULL_DATA_SEND;
2091                                 ieee->ack_tx_to_ieee = 1;
2092                                 rtllib_sta_ps_send_null_frame(ieee, 1);
2093                                 ieee->ps_time = time;
2094                         }
2095                         spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
2096
2097                 }
2098
2099                 ieee->bAwakePktSent = false;
2100
2101         } else if (sleep == 2) {
2102                 spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
2103
2104                 rtllib_sta_wakeup(ieee, 1);
2105
2106                 spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
2107         }
2108
2109 out:
2110         spin_unlock_irqrestore(&ieee->lock, flags);
2111
2112 }
2113
2114 void rtllib_sta_wakeup(struct rtllib_device *ieee, short nl)
2115 {
2116         if (ieee->sta_sleep == LPS_IS_WAKE) {
2117                 if (nl) {
2118                         if (ieee->pHTInfo->IOTAction &
2119                             HT_IOT_ACT_NULL_DATA_POWER_SAVING) {
2120                                 ieee->ack_tx_to_ieee = 1;
2121                                 rtllib_sta_ps_send_null_frame(ieee, 0);
2122                         } else {
2123                                 ieee->ack_tx_to_ieee = 1;
2124                                 rtllib_sta_ps_send_pspoll_frame(ieee);
2125                         }
2126                 }
2127                 return;
2128
2129         }
2130
2131         if (ieee->sta_sleep == LPS_IS_SLEEP)
2132                 ieee->sta_wake_up(ieee->dev);
2133         if (nl) {
2134                 if (ieee->pHTInfo->IOTAction &
2135                     HT_IOT_ACT_NULL_DATA_POWER_SAVING) {
2136                         ieee->ack_tx_to_ieee = 1;
2137                         rtllib_sta_ps_send_null_frame(ieee, 0);
2138                 } else {
2139                         ieee->ack_tx_to_ieee = 1;
2140                         ieee->polling = true;
2141                         rtllib_sta_ps_send_pspoll_frame(ieee);
2142                 }
2143
2144         } else {
2145                 ieee->sta_sleep = LPS_IS_WAKE;
2146                 ieee->polling = false;
2147         }
2148 }
2149
2150 void rtllib_ps_tx_ack(struct rtllib_device *ieee, short success)
2151 {
2152         unsigned long flags, flags2;
2153
2154         spin_lock_irqsave(&ieee->lock, flags);
2155
2156         if (ieee->sta_sleep == LPS_WAIT_NULL_DATA_SEND) {
2157                 /* Null frame with PS bit set */
2158                 if (success) {
2159                         ieee->sta_sleep = LPS_IS_SLEEP;
2160                         ieee->enter_sleep_state(ieee->dev, ieee->ps_time);
2161                 }
2162                 /* if the card report not success we can't be sure the AP
2163                  * has not RXed so we can't assume the AP believe us awake
2164                  */
2165         } else {/* 21112005 - tx again null without PS bit if lost */
2166
2167                 if ((ieee->sta_sleep == LPS_IS_WAKE) && !success) {
2168                         spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
2169                         if (ieee->pHTInfo->IOTAction &
2170                             HT_IOT_ACT_NULL_DATA_POWER_SAVING)
2171                                 rtllib_sta_ps_send_null_frame(ieee, 0);
2172                         else
2173                                 rtllib_sta_ps_send_pspoll_frame(ieee);
2174                         spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
2175                 }
2176         }
2177         spin_unlock_irqrestore(&ieee->lock, flags);
2178 }
2179 EXPORT_SYMBOL(rtllib_ps_tx_ack);
2180
2181 static void rtllib_process_action(struct rtllib_device *ieee, struct sk_buff *skb)
2182 {
2183         struct rtllib_hdr_3addr *header = (struct rtllib_hdr_3addr *) skb->data;
2184         u8 *act = rtllib_get_payload((struct rtllib_hdr *)header);
2185         u8 category = 0;
2186
2187         if (act == NULL) {
2188                 RTLLIB_DEBUG(RTLLIB_DL_ERR, "error to get payload of "
2189                              "action frame\n");
2190                 return;
2191         }
2192
2193         category = *act;
2194         act++;
2195         switch (category) {
2196         case ACT_CAT_BA:
2197                 switch (*act) {
2198                 case ACT_ADDBAREQ:
2199                         rtllib_rx_ADDBAReq(ieee, skb);
2200                         break;
2201                 case ACT_ADDBARSP:
2202                         rtllib_rx_ADDBARsp(ieee, skb);
2203                         break;
2204                 case ACT_DELBA:
2205                         rtllib_rx_DELBA(ieee, skb);
2206                         break;
2207                 }
2208                 break;
2209         default:
2210                 break;
2211         }
2212         return;
2213 }
2214
2215 inline int rtllib_rx_assoc_resp(struct rtllib_device *ieee, struct sk_buff *skb,
2216                                 struct rtllib_rx_stats *rx_stats)
2217 {
2218         u16 errcode;
2219         int aid;
2220         u8 *ies;
2221         struct rtllib_assoc_response_frame *assoc_resp;
2222         struct rtllib_hdr_3addr *header = (struct rtllib_hdr_3addr *) skb->data;
2223
2224         RTLLIB_DEBUG_MGMT("received [RE]ASSOCIATION RESPONSE (%d)\n",
2225                           WLAN_FC_GET_STYPE(header->frame_ctl));
2226
2227         if ((ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) &&
2228              ieee->state == RTLLIB_ASSOCIATING_AUTHENTICATED &&
2229              (ieee->iw_mode == IW_MODE_INFRA)) {
2230                 errcode = assoc_parse(ieee, skb, &aid);
2231                 if (0 == errcode) {
2232                         struct rtllib_network *network =
2233                                  kzalloc(sizeof(struct rtllib_network),
2234                                  GFP_ATOMIC);
2235
2236                         if (!network)
2237                                 return 1;
2238                         memset(network, 0, sizeof(*network));
2239                         ieee->state = RTLLIB_LINKED;
2240                         ieee->assoc_id = aid;
2241                         ieee->softmac_stats.rx_ass_ok++;
2242                         /* station support qos */
2243                         /* Let the register setting default with Legacy station */
2244                         assoc_resp = (struct rtllib_assoc_response_frame *)skb->data;
2245                         if (ieee->current_network.qos_data.supported == 1) {
2246                                 if (rtllib_parse_info_param(ieee, assoc_resp->info_element,
2247                                                         rx_stats->len - sizeof(*assoc_resp),
2248                                                         network, rx_stats)) {
2249                                         kfree(network);
2250                                         return 1;
2251                                 } else {
2252                                         memcpy(ieee->pHTInfo->PeerHTCapBuf,
2253                                                network->bssht.bdHTCapBuf,
2254                                                network->bssht.bdHTCapLen);
2255                                         memcpy(ieee->pHTInfo->PeerHTInfoBuf,
2256                                                network->bssht.bdHTInfoBuf,
2257                                                network->bssht.bdHTInfoLen);
2258                                 }
2259                                 if (ieee->handle_assoc_response != NULL)
2260                                         ieee->handle_assoc_response(ieee->dev,
2261                                                  (struct rtllib_assoc_response_frame *)header,
2262                                                  network);
2263                                 kfree(network);
2264                         }
2265
2266                         kfree(ieee->assocresp_ies);
2267                         ieee->assocresp_ies = NULL;
2268                         ies = &(assoc_resp->info_element[0].id);
2269                         ieee->assocresp_ies_len = (skb->data + skb->len) - ies;
2270                         ieee->assocresp_ies = kmalloc(ieee->assocresp_ies_len,
2271                                                       GFP_ATOMIC);
2272                         if (ieee->assocresp_ies)
2273                                 memcpy(ieee->assocresp_ies, ies,
2274                                        ieee->assocresp_ies_len);
2275                         else {
2276                                 printk(KERN_INFO "%s()Warning: can't alloc "
2277                                        "memory for assocresp_ies\n", __func__);
2278                                 ieee->assocresp_ies_len = 0;
2279                         }
2280                         rtllib_associate_complete(ieee);
2281                 } else {
2282                         /* aid could not been allocated */
2283                         ieee->softmac_stats.rx_ass_err++;
2284                         printk(KERN_INFO "Association response status code 0x%x\n",
2285                                 errcode);
2286                         RTLLIB_DEBUG_MGMT(
2287                                 "Association response status code 0x%x\n",
2288                                 errcode);
2289                         if (ieee->AsocRetryCount < RT_ASOC_RETRY_LIMIT)
2290                                 queue_delayed_work_rsl(ieee->wq,
2291                                          &ieee->associate_procedure_wq, 0);
2292                         else
2293                                 rtllib_associate_abort(ieee);
2294                 }
2295         }
2296         return 0;
2297 }
2298
2299 inline int rtllib_rx_auth(struct rtllib_device *ieee, struct sk_buff *skb,
2300                           struct rtllib_rx_stats *rx_stats)
2301 {
2302         u16 errcode;
2303         u8 *challenge;
2304         int chlen = 0;
2305         bool bSupportNmode = true, bHalfSupportNmode = false;
2306
2307         if (ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) {
2308                 if (ieee->state == RTLLIB_ASSOCIATING_AUTHENTICATING &&
2309                     (ieee->iw_mode == IW_MODE_INFRA)) {
2310                         RTLLIB_DEBUG_MGMT("Received authentication response");
2311
2312                         errcode = auth_parse(skb, &challenge, &chlen);
2313                         if (0 == errcode) {
2314                                 if (ieee->open_wep || !challenge) {
2315                                         ieee->state = RTLLIB_ASSOCIATING_AUTHENTICATED;
2316                                         ieee->softmac_stats.rx_auth_rs_ok++;
2317                                         if (!(ieee->pHTInfo->IOTAction &
2318                                             HT_IOT_ACT_PURE_N_MODE)) {
2319                                                 if (!ieee->GetNmodeSupportBySecCfg(ieee->dev)) {
2320                                                         if (IsHTHalfNmodeAPs(ieee)) {
2321                                                                 bSupportNmode = true;
2322                                                                 bHalfSupportNmode = true;
2323                                                         } else {
2324                                                                 bSupportNmode = false;
2325                                                                 bHalfSupportNmode = false;
2326                                                         }
2327                                                 }
2328                                         }
2329                                         /* Dummy wirless mode setting to avoid
2330                                          * encryption issue */
2331                                         if (bSupportNmode) {
2332                                                 ieee->SetWirelessMode(ieee->dev,
2333                                                    ieee->current_network.mode);
2334                                         } else {
2335                                                 /*TODO*/
2336                                                 ieee->SetWirelessMode(ieee->dev,
2337                                                                       IEEE_G);
2338                                         }
2339
2340                                         if (ieee->current_network.mode ==
2341                                             IEEE_N_24G &&
2342                                             bHalfSupportNmode == true) {
2343                                                 printk(KERN_INFO "======>enter "
2344                                                        "half N mode\n");
2345                                                 ieee->bHalfWirelessN24GMode =
2346                                                                          true;
2347                                         } else
2348                                                 ieee->bHalfWirelessN24GMode =
2349                                                                          false;
2350
2351                                         rtllib_associate_step2(ieee);
2352                                 } else {
2353                                         rtllib_auth_challenge(ieee, challenge,
2354                                                               chlen);
2355                                 }
2356                         } else {
2357                                 ieee->softmac_stats.rx_auth_rs_err++;
2358                                 RTLLIB_DEBUG_MGMT("Authentication respose"
2359                                                   " status code 0x%x", errcode);
2360
2361                                 printk(KERN_INFO "Authentication respose "
2362                                        "status code 0x%x", errcode);
2363                                 rtllib_associate_abort(ieee);
2364                         }
2365
2366                 } else if (ieee->iw_mode == IW_MODE_MASTER) {
2367                         rtllib_rx_auth_rq(ieee, skb);
2368                 }
2369         }
2370         return 0;
2371 }
2372
2373 inline int rtllib_rx_deauth(struct rtllib_device *ieee, struct sk_buff *skb)
2374 {
2375         struct rtllib_hdr_3addr *header = (struct rtllib_hdr_3addr *) skb->data;
2376
2377         if (memcmp(header->addr3, ieee->current_network.bssid, ETH_ALEN) != 0)
2378                 return 0;
2379
2380         /* FIXME for now repeat all the association procedure
2381         * both for disassociation and deauthentication
2382         */
2383         if ((ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) &&
2384             ieee->state == RTLLIB_LINKED &&
2385             (ieee->iw_mode == IW_MODE_INFRA)) {
2386                 printk(KERN_INFO "==========>received disassoc/deauth(%x) "
2387                        "frame, reason code:%x\n",
2388                        WLAN_FC_GET_STYPE(header->frame_ctl),
2389                        ((struct rtllib_disassoc *)skb->data)->reason);
2390                 ieee->state = RTLLIB_ASSOCIATING;
2391                 ieee->softmac_stats.reassoc++;
2392                 ieee->is_roaming = true;
2393                 ieee->LinkDetectInfo.bBusyTraffic = false;
2394                 rtllib_disassociate(ieee);
2395                 RemovePeerTS(ieee, header->addr2);
2396                 if (ieee->LedControlHandler != NULL)
2397                         ieee->LedControlHandler(ieee->dev,
2398                                                 LED_CTL_START_TO_LINK);
2399
2400                 if (!(ieee->rtllib_ap_sec_type(ieee) &
2401                     (SEC_ALG_CCMP|SEC_ALG_TKIP)))
2402                         queue_delayed_work_rsl(ieee->wq,
2403                                        &ieee->associate_procedure_wq, 5);
2404         }
2405         return 0;
2406 }
2407
2408 inline int rtllib_rx_frame_softmac(struct rtllib_device *ieee,
2409                                    struct sk_buff *skb,
2410                                    struct rtllib_rx_stats *rx_stats, u16 type,
2411                                    u16 stype)
2412 {
2413         struct rtllib_hdr_3addr *header = (struct rtllib_hdr_3addr *) skb->data;
2414
2415         if (!ieee->proto_started)
2416                 return 0;
2417
2418         switch (WLAN_FC_GET_STYPE(header->frame_ctl)) {
2419         case RTLLIB_STYPE_ASSOC_RESP:
2420         case RTLLIB_STYPE_REASSOC_RESP:
2421                 if (rtllib_rx_assoc_resp(ieee, skb, rx_stats) == 1)
2422                         return 1;
2423                 break;
2424         case RTLLIB_STYPE_ASSOC_REQ:
2425         case RTLLIB_STYPE_REASSOC_REQ:
2426                 if ((ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) &&
2427                      ieee->iw_mode == IW_MODE_MASTER)
2428                         rtllib_rx_assoc_rq(ieee, skb);
2429                 break;
2430         case RTLLIB_STYPE_AUTH:
2431                 rtllib_rx_auth(ieee, skb, rx_stats);
2432                 break;
2433         case RTLLIB_STYPE_DISASSOC:
2434         case RTLLIB_STYPE_DEAUTH:
2435                 rtllib_rx_deauth(ieee, skb);
2436                 break;
2437         case RTLLIB_STYPE_MANAGE_ACT:
2438                 rtllib_process_action(ieee, skb);
2439                 break;
2440         default:
2441                 return -1;
2442                 break;
2443         }
2444         return 0;
2445 }
2446
2447 /* following are for a simplier TX queue management.
2448  * Instead of using netif_[stop/wake]_queue the driver
2449  * will uses these two function (plus a reset one), that
2450  * will internally uses the kernel netif_* and takes
2451  * care of the ieee802.11 fragmentation.
2452  * So the driver receives a fragment per time and might
2453  * call the stop function when it want without take care
2454  * to have enought room to TX an entire packet.
2455  * This might be useful if each fragment need it's own
2456  * descriptor, thus just keep a total free memory > than
2457  * the max fragmentation treshold is not enought.. If the
2458  * ieee802.11 stack passed a TXB struct then you needed
2459  * to keep N free descriptors where
2460  * N = MAX_PACKET_SIZE / MIN_FRAG_TRESHOLD
2461  * In this way you need just one and the 802.11 stack
2462  * will take care of buffering fragments and pass them to
2463  * to the driver later, when it wakes the queue.
2464  */
2465 void rtllib_softmac_xmit(struct rtllib_txb *txb, struct rtllib_device *ieee)
2466 {
2467
2468         unsigned int queue_index = txb->queue_index;
2469         unsigned long flags;
2470         int  i;
2471         struct cb_desc *tcb_desc = NULL;
2472         unsigned long queue_len = 0;
2473
2474         spin_lock_irqsave(&ieee->lock, flags);
2475
2476         /* called with 2nd parm 0, no tx mgmt lock required */
2477         rtllib_sta_wakeup(ieee, 0);
2478
2479         /* update the tx status */
2480         tcb_desc = (struct cb_desc *)(txb->fragments[0]->cb +
2481                    MAX_DEV_ADDR_SIZE);
2482         if (tcb_desc->bMulticast)
2483                 ieee->stats.multicast++;
2484
2485         /* if xmit available, just xmit it immediately, else just insert it to
2486          * the wait queue */
2487         for (i = 0; i < txb->nr_frags; i++) {
2488                 queue_len = skb_queue_len(&ieee->skb_waitQ[queue_index]);
2489                 if ((queue_len  != 0) ||\
2490                     (!ieee->check_nic_enough_desc(ieee->dev, queue_index)) ||
2491                     (ieee->queue_stop)) {
2492                         /* insert the skb packet to the wait queue */
2493                         /* as for the completion function, it does not need
2494                          * to check it any more.
2495                          * */
2496                         if (queue_len < 200)
2497                                 skb_queue_tail(&ieee->skb_waitQ[queue_index],
2498                                                txb->fragments[i]);
2499                         else
2500                                 kfree_skb(txb->fragments[i]);
2501                 } else {
2502                         ieee->softmac_data_hard_start_xmit(
2503                                         txb->fragments[i],
2504                                         ieee->dev, ieee->rate);
2505                 }
2506         }
2507
2508         rtllib_txb_free(txb);
2509
2510         spin_unlock_irqrestore(&ieee->lock, flags);
2511
2512 }
2513
2514 /* called with ieee->lock acquired */
2515 static void rtllib_resume_tx(struct rtllib_device *ieee)
2516 {
2517         int i;
2518         for (i = ieee->tx_pending.frag; i < ieee->tx_pending.txb->nr_frags;
2519              i++) {
2520
2521                 if (ieee->queue_stop) {
2522                         ieee->tx_pending.frag = i;
2523                         return;
2524                 } else {
2525
2526                         ieee->softmac_data_hard_start_xmit(
2527                                 ieee->tx_pending.txb->fragments[i],
2528                                 ieee->dev, ieee->rate);
2529                         ieee->stats.tx_packets++;
2530                 }
2531         }
2532
2533         rtllib_txb_free(ieee->tx_pending.txb);
2534         ieee->tx_pending.txb = NULL;
2535 }
2536
2537
2538 void rtllib_reset_queue(struct rtllib_device *ieee)
2539 {
2540         unsigned long flags;
2541
2542         spin_lock_irqsave(&ieee->lock, flags);
2543         init_mgmt_queue(ieee);
2544         if (ieee->tx_pending.txb) {
2545                 rtllib_txb_free(ieee->tx_pending.txb);
2546                 ieee->tx_pending.txb = NULL;
2547         }
2548         ieee->queue_stop = 0;
2549         spin_unlock_irqrestore(&ieee->lock, flags);
2550
2551 }
2552 EXPORT_SYMBOL(rtllib_reset_queue);
2553
2554 void rtllib_wake_queue(struct rtllib_device *ieee)
2555 {
2556
2557         unsigned long flags;
2558         struct sk_buff *skb;
2559         struct rtllib_hdr_3addr  *header;
2560
2561         spin_lock_irqsave(&ieee->lock, flags);
2562         if (!ieee->queue_stop)
2563                 goto exit;
2564
2565         ieee->queue_stop = 0;
2566
2567         if (ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE) {
2568                 while (!ieee->queue_stop && (skb = dequeue_mgmt(ieee))) {
2569
2570                         header = (struct rtllib_hdr_3addr  *) skb->data;
2571
2572                         header->seq_ctl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
2573
2574                         if (ieee->seq_ctrl[0] == 0xFFF)
2575                                 ieee->seq_ctrl[0] = 0;
2576                         else
2577                                 ieee->seq_ctrl[0]++;
2578
2579                         ieee->softmac_data_hard_start_xmit(skb, ieee->dev,
2580                                                            ieee->basic_rate);
2581                 }
2582         }
2583         if (!ieee->queue_stop && ieee->tx_pending.txb)
2584                 rtllib_resume_tx(ieee);
2585
2586         if (!ieee->queue_stop && netif_queue_stopped(ieee->dev)) {
2587                 ieee->softmac_stats.swtxawake++;
2588                 netif_wake_queue(ieee->dev);
2589         }
2590
2591 exit:
2592         spin_unlock_irqrestore(&ieee->lock, flags);
2593 }
2594
2595
2596 void rtllib_stop_queue(struct rtllib_device *ieee)
2597 {
2598
2599         if (!netif_queue_stopped(ieee->dev)) {
2600                 netif_stop_queue(ieee->dev);
2601                 ieee->softmac_stats.swtxstop++;
2602         }
2603         ieee->queue_stop = 1;
2604
2605 }
2606
2607 void rtllib_stop_all_queues(struct rtllib_device *ieee)
2608 {
2609         unsigned int i;
2610         for (i = 0; i < ieee->dev->num_tx_queues; i++)
2611                 netdev_get_tx_queue(ieee->dev, i)->trans_start = jiffies;
2612
2613         netif_tx_stop_all_queues(ieee->dev);
2614 }
2615
2616 void rtllib_wake_all_queues(struct rtllib_device *ieee)
2617 {
2618         netif_tx_wake_all_queues(ieee->dev);
2619 }
2620
2621 inline void rtllib_randomize_cell(struct rtllib_device *ieee)
2622 {
2623
2624         get_random_bytes(ieee->current_network.bssid, ETH_ALEN);
2625
2626         /* an IBSS cell address must have the two less significant
2627          * bits of the first byte = 2
2628          */
2629         ieee->current_network.bssid[0] &= ~0x01;
2630         ieee->current_network.bssid[0] |= 0x02;
2631 }
2632
2633 /* called in user context only */
2634 void rtllib_start_master_bss(struct rtllib_device *ieee)
2635 {
2636         ieee->assoc_id = 1;
2637
2638         if (ieee->current_network.ssid_len == 0) {
2639                 strncpy(ieee->current_network.ssid,
2640                         RTLLIB_DEFAULT_TX_ESSID,
2641                         IW_ESSID_MAX_SIZE);
2642
2643                 ieee->current_network.ssid_len =
2644                                  strlen(RTLLIB_DEFAULT_TX_ESSID);
2645                 ieee->ssid_set = 1;
2646         }
2647
2648         memcpy(ieee->current_network.bssid, ieee->dev->dev_addr, ETH_ALEN);
2649
2650         ieee->set_chan(ieee->dev, ieee->current_network.channel);
2651         ieee->state = RTLLIB_LINKED;
2652         ieee->link_change(ieee->dev);
2653         notify_wx_assoc_event(ieee);
2654
2655         if (ieee->data_hard_resume)
2656                 ieee->data_hard_resume(ieee->dev);
2657
2658         netif_carrier_on(ieee->dev);
2659 }
2660
2661 static void rtllib_start_monitor_mode(struct rtllib_device *ieee)
2662 {
2663         /* reset hardware status */
2664         if (ieee->raw_tx) {
2665                 if (ieee->data_hard_resume)
2666                         ieee->data_hard_resume(ieee->dev);
2667
2668                 netif_carrier_on(ieee->dev);
2669         }
2670 }
2671
2672 static void rtllib_start_ibss_wq(void *data)
2673 {
2674         struct rtllib_device *ieee = container_of_dwork_rsl(data,
2675                                      struct rtllib_device, start_ibss_wq);
2676         /* iwconfig mode ad-hoc will schedule this and return
2677          * on the other hand this will block further iwconfig SET
2678          * operations because of the wx_sem hold.
2679          * Anyway some most set operations set a flag to speed-up
2680          * (abort) this wq (when syncro scanning) before sleeping
2681          * on the semaphore
2682          */
2683         if (!ieee->proto_started) {
2684                 printk(KERN_INFO "==========oh driver down return\n");
2685                 return;
2686         }
2687         down(&ieee->wx_sem);
2688
2689         if (ieee->current_network.ssid_len == 0) {
2690                 strcpy(ieee->current_network.ssid, RTLLIB_DEFAULT_TX_ESSID);
2691                 ieee->current_network.ssid_len = strlen(RTLLIB_DEFAULT_TX_ESSID);
2692                 ieee->ssid_set = 1;
2693         }
2694
2695         ieee->state = RTLLIB_NOLINK;
2696         ieee->mode = IEEE_G;
2697         /* check if we have this cell in our network list */
2698         rtllib_softmac_check_all_nets(ieee);
2699
2700
2701         /* if not then the state is not linked. Maybe the user swithced to
2702          * ad-hoc mode just after being in monitor mode, or just after
2703          * being very few time in managed mode (so the card have had no
2704          * time to scan all the chans..) or we have just run up the iface
2705          * after setting ad-hoc mode. So we have to give another try..
2706          * Here, in ibss mode, should be safe to do this without extra care
2707          * (in bss mode we had to make sure no-one tryed to associate when
2708          * we had just checked the ieee->state and we was going to start the
2709          * scan) beacause in ibss mode the rtllib_new_net function, when
2710          * finds a good net, just set the ieee->state to RTLLIB_LINKED,
2711          * so, at worst, we waste a bit of time to initiate an unneeded syncro
2712          * scan, that will stop at the first round because it sees the state
2713          * associated.
2714          */
2715         if (ieee->state == RTLLIB_NOLINK)
2716                 rtllib_start_scan_syncro(ieee, 0);
2717
2718         /* the network definitively is not here.. create a new cell */
2719         if (ieee->state == RTLLIB_NOLINK) {
2720                 printk(KERN_INFO "creating new IBSS cell\n");
2721                 ieee->current_network.channel = ieee->IbssStartChnl;
2722                 if (!ieee->wap_set)
2723                         rtllib_randomize_cell(ieee);
2724
2725                 if (ieee->modulation & RTLLIB_CCK_MODULATION) {
2726
2727                         ieee->current_network.rates_len = 4;
2728
2729                         ieee->current_network.rates[0] =
2730                                  RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_1MB;
2731                         ieee->current_network.rates[1] =
2732                                  RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_2MB;
2733                         ieee->current_network.rates[2] =
2734                                  RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_5MB;
2735                         ieee->current_network.rates[3] =
2736                                  RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_11MB;
2737
2738                 } else
2739                         ieee->current_network.rates_len = 0;
2740
2741                 if (ieee->modulation & RTLLIB_OFDM_MODULATION) {
2742                         ieee->current_network.rates_ex_len = 8;
2743
2744                         ieee->current_network.rates_ex[0] =
2745                                                  RTLLIB_OFDM_RATE_6MB;
2746                         ieee->current_network.rates_ex[1] =
2747                                                  RTLLIB_OFDM_RATE_9MB;
2748                         ieee->current_network.rates_ex[2] =
2749                                                  RTLLIB_OFDM_RATE_12MB;
2750                         ieee->current_network.rates_ex[3] =
2751                                                  RTLLIB_OFDM_RATE_18MB;
2752                         ieee->current_network.rates_ex[4] =
2753                                                  RTLLIB_OFDM_RATE_24MB;
2754                         ieee->current_network.rates_ex[5] =
2755                                                  RTLLIB_OFDM_RATE_36MB;
2756                         ieee->current_network.rates_ex[6] =
2757                                                  RTLLIB_OFDM_RATE_48MB;
2758                         ieee->current_network.rates_ex[7] =
2759                                                  RTLLIB_OFDM_RATE_54MB;
2760
2761                         ieee->rate = 108;
2762                 } else {
2763                         ieee->current_network.rates_ex_len = 0;
2764                         ieee->rate = 22;
2765                 }
2766
2767                 ieee->current_network.qos_data.supported = 0;
2768                 ieee->SetWirelessMode(ieee->dev, IEEE_G);
2769                 ieee->current_network.mode = ieee->mode;
2770                 ieee->current_network.atim_window = 0;
2771                 ieee->current_network.capability = WLAN_CAPABILITY_IBSS;
2772         }
2773
2774         printk(KERN_INFO "%s(): ieee->mode = %d\n", __func__, ieee->mode);
2775         if ((ieee->mode == IEEE_N_24G) || (ieee->mode == IEEE_N_5G))
2776                 HTUseDefaultSetting(ieee);
2777         else
2778                 ieee->pHTInfo->bCurrentHTSupport = false;
2779
2780         ieee->SetHwRegHandler(ieee->dev, HW_VAR_MEDIA_STATUS,
2781                               (u8 *)(&ieee->state));
2782
2783         ieee->state = RTLLIB_LINKED;
2784         ieee->link_change(ieee->dev);
2785
2786         HTSetConnectBwMode(ieee, HT_CHANNEL_WIDTH_20, HT_EXTCHNL_OFFSET_NO_EXT);
2787         if (ieee->LedControlHandler != NULL)
2788                 ieee->LedControlHandler(ieee->dev, LED_CTL_LINK);
2789
2790         rtllib_start_send_beacons(ieee);
2791
2792         notify_wx_assoc_event(ieee);
2793
2794         if (ieee->data_hard_resume)
2795                 ieee->data_hard_resume(ieee->dev);
2796
2797         netif_carrier_on(ieee->dev);
2798
2799         up(&ieee->wx_sem);
2800 }
2801
2802 inline void rtllib_start_ibss(struct rtllib_device *ieee)
2803 {
2804         queue_delayed_work_rsl(ieee->wq, &ieee->start_ibss_wq, MSECS(150));
2805 }
2806
2807 /* this is called only in user context, with wx_sem held */
2808 void rtllib_start_bss(struct rtllib_device *ieee)
2809 {
2810         unsigned long flags;
2811         if (IS_DOT11D_ENABLE(ieee) && !IS_COUNTRY_IE_VALID(ieee)) {
2812                 if (!ieee->bGlobalDomain)
2813                         return;
2814         }
2815         /* check if we have already found the net we
2816          * are interested in (if any).
2817          * if not (we are disassociated and we are not
2818          * in associating / authenticating phase) start the background scanning.
2819          */
2820         rtllib_softmac_check_all_nets(ieee);
2821
2822         /* ensure no-one start an associating process (thus setting
2823          * the ieee->state to rtllib_ASSOCIATING) while we
2824          * have just cheked it and we are going to enable scan.
2825          * The rtllib_new_net function is always called with
2826          * lock held (from both rtllib_softmac_check_all_nets and
2827          * the rx path), so we cannot be in the middle of such function
2828          */
2829         spin_lock_irqsave(&ieee->lock, flags);
2830
2831         if (ieee->state == RTLLIB_NOLINK)
2832                 rtllib_start_scan(ieee);
2833         spin_unlock_irqrestore(&ieee->lock, flags);
2834 }
2835
2836 static void rtllib_link_change_wq(void *data)
2837 {
2838         struct rtllib_device *ieee = container_of_dwork_rsl(data,
2839                                      struct rtllib_device, link_change_wq);
2840         ieee->link_change(ieee->dev);
2841 }
2842 /* called only in userspace context */
2843 void rtllib_disassociate(struct rtllib_device *ieee)
2844 {
2845         netif_carrier_off(ieee->dev);
2846         if (ieee->softmac_features & IEEE_SOFTMAC_TX_QUEUE)
2847                         rtllib_reset_queue(ieee);
2848
2849         if (ieee->data_hard_stop)
2850                         ieee->data_hard_stop(ieee->dev);
2851         if (IS_DOT11D_ENABLE(ieee))
2852                 Dot11d_Reset(ieee);
2853         ieee->state = RTLLIB_NOLINK;
2854         ieee->is_set_key = false;
2855         ieee->wap_set = 0;
2856
2857         queue_delayed_work_rsl(ieee->wq, &ieee->link_change_wq, 0);
2858
2859         notify_wx_assoc_event(ieee);
2860 }
2861
2862 static void rtllib_associate_retry_wq(void *data)
2863 {
2864         struct rtllib_device *ieee = container_of_dwork_rsl(data,
2865                                      struct rtllib_device, associate_retry_wq);
2866         unsigned long flags;
2867
2868         down(&ieee->wx_sem);
2869         if (!ieee->proto_started)
2870                 goto exit;
2871
2872         if (ieee->state != RTLLIB_ASSOCIATING_RETRY)
2873                 goto exit;
2874
2875         /* until we do not set the state to RTLLIB_NOLINK
2876         * there are no possibility to have someone else trying
2877         * to start an association procdure (we get here with
2878         * ieee->state = RTLLIB_ASSOCIATING).
2879         * When we set the state to RTLLIB_NOLINK it is possible
2880         * that the RX path run an attempt to associate, but
2881         * both rtllib_softmac_check_all_nets and the
2882         * RX path works with ieee->lock held so there are no
2883         * problems. If we are still disassociated then start a scan.
2884         * the lock here is necessary to ensure no one try to start
2885         * an association procedure when we have just checked the
2886         * state and we are going to start the scan.
2887         */
2888         ieee->beinretry = true;
2889         ieee->state = RTLLIB_NOLINK;
2890
2891         rtllib_softmac_check_all_nets(ieee);
2892
2893         spin_lock_irqsave(&ieee->lock, flags);
2894
2895         if (ieee->state == RTLLIB_NOLINK)
2896                 rtllib_start_scan(ieee);
2897         spin_unlock_irqrestore(&ieee->lock, flags);
2898
2899         ieee->beinretry = false;
2900 exit:
2901         up(&ieee->wx_sem);
2902 }
2903
2904 struct sk_buff *rtllib_get_beacon_(struct rtllib_device *ieee)
2905 {
2906         u8 broadcast_addr[] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
2907
2908         struct sk_buff *skb;
2909         struct rtllib_probe_response *b;
2910         skb = rtllib_probe_resp(ieee, broadcast_addr);
2911
2912         if (!skb)
2913                 return NULL;
2914
2915         b = (struct rtllib_probe_response *) skb->data;
2916         b->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_BEACON);
2917
2918         return skb;
2919
2920 }
2921
2922 struct sk_buff *rtllib_get_beacon(struct rtllib_device *ieee)
2923 {
2924         struct sk_buff *skb;
2925         struct rtllib_probe_response *b;
2926
2927         skb = rtllib_get_beacon_(ieee);
2928         if (!skb)
2929                 return NULL;
2930
2931         b = (struct rtllib_probe_response *) skb->data;
2932         b->header.seq_ctl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
2933
2934         if (ieee->seq_ctrl[0] == 0xFFF)
2935                 ieee->seq_ctrl[0] = 0;
2936         else
2937                 ieee->seq_ctrl[0]++;
2938
2939         return skb;
2940 }
2941 EXPORT_SYMBOL(rtllib_get_beacon);
2942
2943 void rtllib_softmac_stop_protocol(struct rtllib_device *ieee, u8 mesh_flag,
2944                                   u8 shutdown)
2945 {
2946         rtllib_stop_scan_syncro(ieee);
2947         down(&ieee->wx_sem);
2948         rtllib_stop_protocol(ieee, shutdown);
2949         up(&ieee->wx_sem);
2950 }
2951 EXPORT_SYMBOL(rtllib_softmac_stop_protocol);
2952
2953
2954 void rtllib_stop_protocol(struct rtllib_device *ieee, u8 shutdown)
2955 {
2956         if (!ieee->proto_started)
2957                 return;
2958
2959         if (shutdown) {
2960                 ieee->proto_started = 0;
2961                 ieee->proto_stoppping = 1;
2962                 if (ieee->rtllib_ips_leave != NULL)
2963                         ieee->rtllib_ips_leave(ieee->dev);
2964         }
2965
2966         rtllib_stop_send_beacons(ieee);
2967         del_timer_sync(&ieee->associate_timer);
2968         cancel_delayed_work(&ieee->associate_retry_wq);
2969         cancel_delayed_work(&ieee->start_ibss_wq);
2970         cancel_delayed_work(&ieee->link_change_wq);
2971         rtllib_stop_scan(ieee);
2972
2973         if (ieee->state <= RTLLIB_ASSOCIATING_AUTHENTICATED)
2974                 ieee->state = RTLLIB_NOLINK;
2975
2976         if (ieee->state == RTLLIB_LINKED) {
2977                 if (ieee->iw_mode == IW_MODE_INFRA)
2978                         SendDisassociation(ieee, 1, deauth_lv_ss);
2979                 rtllib_disassociate(ieee);
2980         }
2981
2982         if (shutdown) {
2983                 RemoveAllTS(ieee);
2984                 ieee->proto_stoppping = 0;
2985         }
2986         kfree(ieee->assocreq_ies);
2987         ieee->assocreq_ies = NULL;
2988         ieee->assocreq_ies_len = 0;
2989         kfree(ieee->assocresp_ies);
2990         ieee->assocresp_ies = NULL;
2991         ieee->assocresp_ies_len = 0;
2992 }
2993
2994 void rtllib_softmac_start_protocol(struct rtllib_device *ieee, u8 mesh_flag)
2995 {
2996         down(&ieee->wx_sem);
2997         rtllib_start_protocol(ieee);
2998         up(&ieee->wx_sem);
2999 }
3000 EXPORT_SYMBOL(rtllib_softmac_start_protocol);
3001
3002 void rtllib_start_protocol(struct rtllib_device *ieee)
3003 {
3004         short ch = 0;
3005         int i = 0;
3006
3007         rtllib_update_active_chan_map(ieee);
3008
3009         if (ieee->proto_started)
3010                 return;
3011
3012         ieee->proto_started = 1;
3013
3014         if (ieee->current_network.channel == 0) {
3015                 do {
3016                         ch++;
3017                         if (ch > MAX_CHANNEL_NUMBER)
3018                                 return; /* no channel found */
3019                 } while (!ieee->active_channel_map[ch]);
3020                 ieee->current_network.channel = ch;
3021         }
3022
3023         if (ieee->current_network.beacon_interval == 0)
3024                 ieee->current_network.beacon_interval = 100;
3025
3026         for (i = 0; i < 17; i++) {
3027                 ieee->last_rxseq_num[i] = -1;
3028                 ieee->last_rxfrag_num[i] = -1;
3029                 ieee->last_packet_time[i] = 0;
3030         }
3031
3032         if (ieee->UpdateBeaconInterruptHandler)
3033                 ieee->UpdateBeaconInterruptHandler(ieee->dev, false);
3034
3035         ieee->wmm_acm = 0;
3036         /* if the user set the MAC of the ad-hoc cell and then
3037          * switch to managed mode, shall we  make sure that association
3038          * attempts does not fail just because the user provide the essid
3039          * and the nic is still checking for the AP MAC ??
3040          */
3041         if (ieee->iw_mode == IW_MODE_INFRA) {
3042                 rtllib_start_bss(ieee);
3043         } else if (ieee->iw_mode == IW_MODE_ADHOC) {
3044                 if (ieee->UpdateBeaconInterruptHandler)
3045                         ieee->UpdateBeaconInterruptHandler(ieee->dev, true);
3046
3047                 rtllib_start_ibss(ieee);
3048
3049         } else if (ieee->iw_mode == IW_MODE_MASTER) {
3050                 rtllib_start_master_bss(ieee);
3051         } else if (ieee->iw_mode == IW_MODE_MONITOR) {
3052                 rtllib_start_monitor_mode(ieee);
3053         }
3054 }
3055
3056 void rtllib_softmac_init(struct rtllib_device *ieee)
3057 {
3058         int i;
3059         memset(&ieee->current_network, 0, sizeof(struct rtllib_network));
3060
3061         ieee->state = RTLLIB_NOLINK;
3062         for (i = 0; i < 5; i++)
3063                 ieee->seq_ctrl[i] = 0;
3064         ieee->pDot11dInfo = kzalloc(sizeof(struct rt_dot11d_info), GFP_ATOMIC);
3065         if (!ieee->pDot11dInfo)
3066                 RTLLIB_DEBUG(RTLLIB_DL_ERR, "can't alloc memory for DOT11D\n");
3067         ieee->LinkDetectInfo.SlotIndex = 0;
3068         ieee->LinkDetectInfo.SlotNum = 2;
3069         ieee->LinkDetectInfo.NumRecvBcnInPeriod = 0;
3070         ieee->LinkDetectInfo.NumRecvDataInPeriod = 0;
3071         ieee->LinkDetectInfo.NumTxOkInPeriod = 0;
3072         ieee->LinkDetectInfo.NumRxOkInPeriod = 0;
3073         ieee->LinkDetectInfo.NumRxUnicastOkInPeriod = 0;
3074         ieee->bIsAggregateFrame = false;
3075         ieee->assoc_id = 0;
3076         ieee->queue_stop = 0;
3077         ieee->scanning_continue = 0;
3078         ieee->softmac_features = 0;
3079         ieee->wap_set = 0;
3080         ieee->ssid_set = 0;
3081         ieee->proto_started = 0;
3082         ieee->proto_stoppping = 0;
3083         ieee->basic_rate = RTLLIB_DEFAULT_BASIC_RATE;
3084         ieee->rate = 22;
3085         ieee->ps = RTLLIB_PS_DISABLED;
3086         ieee->sta_sleep = LPS_IS_WAKE;
3087
3088         ieee->Regdot11HTOperationalRateSet[0] = 0xff;
3089         ieee->Regdot11HTOperationalRateSet[1] = 0xff;
3090         ieee->Regdot11HTOperationalRateSet[4] = 0x01;
3091
3092         ieee->Regdot11TxHTOperationalRateSet[0] = 0xff;
3093         ieee->Regdot11TxHTOperationalRateSet[1] = 0xff;
3094         ieee->Regdot11TxHTOperationalRateSet[4] = 0x01;
3095
3096         ieee->FirstIe_InScan = false;
3097         ieee->actscanning = false;
3098         ieee->beinretry = false;
3099         ieee->is_set_key = false;
3100         init_mgmt_queue(ieee);
3101
3102         ieee->sta_edca_param[0] = 0x0000A403;
3103         ieee->sta_edca_param[1] = 0x0000A427;
3104         ieee->sta_edca_param[2] = 0x005E4342;
3105         ieee->sta_edca_param[3] = 0x002F3262;
3106         ieee->aggregation = true;
3107         ieee->enable_rx_imm_BA = 1;
3108         ieee->tx_pending.txb = NULL;
3109
3110         _setup_timer(&ieee->associate_timer,
3111                     rtllib_associate_abort_cb,
3112                     (unsigned long) ieee);
3113
3114         _setup_timer(&ieee->beacon_timer,
3115                     rtllib_send_beacon_cb,
3116                     (unsigned long) ieee);
3117
3118
3119         ieee->wq = create_workqueue(DRV_NAME);
3120
3121         INIT_DELAYED_WORK_RSL(&ieee->link_change_wq,
3122                               (void *)rtllib_link_change_wq, ieee);
3123         INIT_DELAYED_WORK_RSL(&ieee->start_ibss_wq,
3124                               (void *)rtllib_start_ibss_wq, ieee);
3125         INIT_WORK_RSL(&ieee->associate_complete_wq,
3126                       (void *)rtllib_associate_complete_wq, ieee);
3127         INIT_DELAYED_WORK_RSL(&ieee->associate_procedure_wq,
3128                               (void *)rtllib_associate_procedure_wq, ieee);
3129         INIT_DELAYED_WORK_RSL(&ieee->softmac_scan_wq,
3130                               (void *)rtllib_softmac_scan_wq, ieee);
3131         INIT_DELAYED_WORK_RSL(&ieee->softmac_hint11d_wq,
3132                               (void *)rtllib_softmac_hint11d_wq, ieee);
3133         INIT_DELAYED_WORK_RSL(&ieee->associate_retry_wq,
3134                               (void *)rtllib_associate_retry_wq, ieee);
3135         INIT_WORK_RSL(&ieee->wx_sync_scan_wq, (void *)rtllib_wx_sync_scan_wq,
3136                       ieee);
3137
3138         sema_init(&ieee->wx_sem, 1);
3139         sema_init(&ieee->scan_sem, 1);
3140         sema_init(&ieee->ips_sem, 1);
3141
3142         spin_lock_init(&ieee->mgmt_tx_lock);
3143         spin_lock_init(&ieee->beacon_lock);
3144
3145         tasklet_init(&ieee->ps_task,
3146              (void(*)(unsigned long)) rtllib_sta_ps,
3147              (unsigned long)ieee);
3148
3149 }
3150
3151 void rtllib_softmac_free(struct rtllib_device *ieee)
3152 {
3153         down(&ieee->wx_sem);
3154         kfree(ieee->pDot11dInfo);
3155         ieee->pDot11dInfo = NULL;
3156         del_timer_sync(&ieee->associate_timer);
3157
3158         cancel_delayed_work(&ieee->associate_retry_wq);
3159         destroy_workqueue(ieee->wq);
3160         up(&ieee->wx_sem);
3161 }
3162
3163 /********************************************************
3164  * Start of WPA code.                                   *
3165  * this is stolen from the ipw2200 driver               *
3166  ********************************************************/
3167
3168
3169 static int rtllib_wpa_enable(struct rtllib_device *ieee, int value)
3170 {
3171         /* This is called when wpa_supplicant loads and closes the driver
3172          * interface. */
3173         printk(KERN_INFO "%s WPA\n", value ? "enabling" : "disabling");
3174         ieee->wpa_enabled = value;
3175         memset(ieee->ap_mac_addr, 0, 6);
3176         return 0;
3177 }
3178
3179
3180 static void rtllib_wpa_assoc_frame(struct rtllib_device *ieee, char *wpa_ie,
3181                                    int wpa_ie_len)
3182 {
3183         /* make sure WPA is enabled */
3184         rtllib_wpa_enable(ieee, 1);
3185
3186         rtllib_disassociate(ieee);
3187 }
3188
3189
3190 static int rtllib_wpa_mlme(struct rtllib_device *ieee, int command, int reason)
3191 {
3192
3193         int ret = 0;
3194
3195         switch (command) {
3196         case IEEE_MLME_STA_DEAUTH:
3197                 break;
3198
3199         case IEEE_MLME_STA_DISASSOC:
3200                 rtllib_disassociate(ieee);
3201                 break;
3202
3203         default:
3204                 printk(KERN_INFO "Unknown MLME request: %d\n", command);
3205                 ret = -EOPNOTSUPP;
3206         }
3207
3208         return ret;
3209 }
3210
3211
3212 static int rtllib_wpa_set_wpa_ie(struct rtllib_device *ieee,
3213                               struct ieee_param *param, int plen)
3214 {
3215         u8 *buf;
3216
3217         if (param->u.wpa_ie.len > MAX_WPA_IE_LEN ||
3218             (param->u.wpa_ie.len && param->u.wpa_ie.data == NULL))
3219                 return -EINVAL;
3220
3221         if (param->u.wpa_ie.len) {
3222                 buf = kmemdup(param->u.wpa_ie.data, param->u.wpa_ie.len,
3223                               GFP_KERNEL);
3224                 if (buf == NULL)
3225                         return -ENOMEM;
3226
3227                 kfree(ieee->wpa_ie);
3228                 ieee->wpa_ie = buf;
3229                 ieee->wpa_ie_len = param->u.wpa_ie.len;
3230         } else {
3231                 kfree(ieee->wpa_ie);
3232                 ieee->wpa_ie = NULL;
3233                 ieee->wpa_ie_len = 0;
3234         }
3235
3236         rtllib_wpa_assoc_frame(ieee, ieee->wpa_ie, ieee->wpa_ie_len);
3237         return 0;
3238 }
3239
3240 #define AUTH_ALG_OPEN_SYSTEM                    0x1
3241 #define AUTH_ALG_SHARED_KEY                     0x2
3242 #define AUTH_ALG_LEAP                           0x4
3243 static int rtllib_wpa_set_auth_algs(struct rtllib_device *ieee, int value)
3244 {
3245
3246         struct rtllib_security sec = {
3247                 .flags = SEC_AUTH_MODE,
3248         };
3249         int ret = 0;
3250
3251         if (value & AUTH_ALG_SHARED_KEY) {
3252                 sec.auth_mode = WLAN_AUTH_SHARED_KEY;
3253                 ieee->open_wep = 0;
3254                 ieee->auth_mode = 1;
3255         } else if (value & AUTH_ALG_OPEN_SYSTEM) {
3256                 sec.auth_mode = WLAN_AUTH_OPEN;
3257                 ieee->open_wep = 1;
3258                 ieee->auth_mode = 0;
3259         } else if (value & AUTH_ALG_LEAP) {
3260                 sec.auth_mode = WLAN_AUTH_LEAP  >> 6;
3261                 ieee->open_wep = 1;
3262                 ieee->auth_mode = 2;
3263         }
3264
3265
3266         if (ieee->set_security)
3267                 ieee->set_security(ieee->dev, &sec);
3268
3269         return ret;
3270 }
3271
3272 static int rtllib_wpa_set_param(struct rtllib_device *ieee, u8 name, u32 value)
3273 {
3274         int ret = 0;
3275         unsigned long flags;
3276
3277         switch (name) {
3278         case IEEE_PARAM_WPA_ENABLED:
3279                 ret = rtllib_wpa_enable(ieee, value);
3280                 break;
3281
3282         case IEEE_PARAM_TKIP_COUNTERMEASURES:
3283                 ieee->tkip_countermeasures = value;
3284                 break;
3285
3286         case IEEE_PARAM_DROP_UNENCRYPTED:
3287         {
3288                 /* HACK:
3289                  *
3290                  * wpa_supplicant calls set_wpa_enabled when the driver
3291                  * is loaded and unloaded, regardless of if WPA is being
3292                  * used.  No other calls are made which can be used to
3293                  * determine if encryption will be used or not prior to
3294                  * association being expected.  If encryption is not being
3295                  * used, drop_unencrypted is set to false, else true -- we
3296                  * can use this to determine if the CAP_PRIVACY_ON bit should
3297                  * be set.
3298                  */
3299                 struct rtllib_security sec = {
3300                         .flags = SEC_ENABLED,
3301                         .enabled = value,
3302                 };
3303                 ieee->drop_unencrypted = value;
3304                 /* We only change SEC_LEVEL for open mode. Others
3305                  * are set by ipw_wpa_set_encryption.
3306                  */
3307                 if (!value) {
3308                         sec.flags |= SEC_LEVEL;
3309                         sec.level = SEC_LEVEL_0;
3310                 } else {
3311                         sec.flags |= SEC_LEVEL;
3312                         sec.level = SEC_LEVEL_1;
3313                 }
3314                 if (ieee->set_security)
3315                         ieee->set_security(ieee->dev, &sec);
3316                 break;
3317         }
3318
3319         case IEEE_PARAM_PRIVACY_INVOKED:
3320                 ieee->privacy_invoked = value;
3321                 break;
3322
3323         case IEEE_PARAM_AUTH_ALGS:
3324                 ret = rtllib_wpa_set_auth_algs(ieee, value);
3325                 break;
3326
3327         case IEEE_PARAM_IEEE_802_1X:
3328                 ieee->ieee802_1x = value;
3329                 break;
3330         case IEEE_PARAM_WPAX_SELECT:
3331                 spin_lock_irqsave(&ieee->wpax_suitlist_lock, flags);
3332                 spin_unlock_irqrestore(&ieee->wpax_suitlist_lock, flags);
3333                 break;
3334
3335         default:
3336                 printk(KERN_INFO "Unknown WPA param: %d\n", name);
3337                 ret = -EOPNOTSUPP;
3338         }
3339
3340         return ret;
3341 }
3342
3343 /* implementation borrowed from hostap driver */
3344 static int rtllib_wpa_set_encryption(struct rtllib_device *ieee,
3345                                   struct ieee_param *param, int param_len,
3346                                   u8 is_mesh)
3347 {
3348         int ret = 0;
3349         struct rtllib_crypto_ops *ops;
3350         struct rtllib_crypt_data **crypt;
3351
3352         struct rtllib_security sec = {
3353                 .flags = 0,
3354         };
3355
3356         param->u.crypt.err = 0;
3357         param->u.crypt.alg[IEEE_CRYPT_ALG_NAME_LEN - 1] = '\0';
3358
3359         if (param_len !=
3360             (int) ((char *) param->u.crypt.key - (char *) param) +
3361             param->u.crypt.key_len) {
3362                 printk(KERN_INFO "Len mismatch %d, %d\n", param_len,
3363                                param->u.crypt.key_len);
3364                 return -EINVAL;
3365         }
3366         if (param->sta_addr[0] == 0xff && param->sta_addr[1] == 0xff &&
3367             param->sta_addr[2] == 0xff && param->sta_addr[3] == 0xff &&
3368             param->sta_addr[4] == 0xff && param->sta_addr[5] == 0xff) {
3369                 if (param->u.crypt.idx >= WEP_KEYS)
3370                         return -EINVAL;
3371                 crypt = &ieee->crypt[param->u.crypt.idx];
3372         } else {
3373                 return -EINVAL;
3374         }
3375
3376         if (strcmp(param->u.crypt.alg, "none") == 0) {
3377                 if (crypt) {
3378                         sec.enabled = 0;
3379                         sec.level = SEC_LEVEL_0;
3380                         sec.flags |= SEC_ENABLED | SEC_LEVEL;
3381                         rtllib_crypt_delayed_deinit(ieee, crypt);
3382                 }
3383                 goto done;
3384         }
3385         sec.enabled = 1;
3386         sec.flags |= SEC_ENABLED;
3387
3388         /* IPW HW cannot build TKIP MIC, host decryption still needed. */
3389         if (!(ieee->host_encrypt || ieee->host_decrypt) &&
3390             strcmp(param->u.crypt.alg, "TKIP"))
3391                 goto skip_host_crypt;
3392
3393         ops = rtllib_get_crypto_ops(param->u.crypt.alg);
3394         if (ops == NULL && strcmp(param->u.crypt.alg, "WEP") == 0) {
3395                 request_module("rtllib_crypt_wep");
3396                 ops = rtllib_get_crypto_ops(param->u.crypt.alg);
3397         } else if (ops == NULL && strcmp(param->u.crypt.alg, "TKIP") == 0) {
3398                 request_module("rtllib_crypt_tkip");
3399                 ops = rtllib_get_crypto_ops(param->u.crypt.alg);
3400         } else if (ops == NULL && strcmp(param->u.crypt.alg, "CCMP") == 0) {
3401                 request_module("rtllib_crypt_ccmp");
3402                 ops = rtllib_get_crypto_ops(param->u.crypt.alg);
3403         }
3404         if (ops == NULL) {
3405                 printk(KERN_INFO "unknown crypto alg '%s'\n",
3406                        param->u.crypt.alg);
3407                 param->u.crypt.err = IEEE_CRYPT_ERR_UNKNOWN_ALG;
3408                 ret = -EINVAL;
3409                 goto done;
3410         }
3411         if (*crypt == NULL || (*crypt)->ops != ops) {
3412                 struct rtllib_crypt_data *new_crypt;
3413
3414                 rtllib_crypt_delayed_deinit(ieee, crypt);
3415
3416                 new_crypt = (struct rtllib_crypt_data *)
3417                         kmalloc(sizeof(*new_crypt), GFP_KERNEL);
3418                 if (new_crypt == NULL) {
3419                         ret = -ENOMEM;
3420                         goto done;
3421                 }
3422                 memset(new_crypt, 0, sizeof(struct rtllib_crypt_data));
3423                 new_crypt->ops = ops;
3424                 if (new_crypt->ops)
3425                         new_crypt->priv =
3426                                 new_crypt->ops->init(param->u.crypt.idx);
3427
3428                 if (new_crypt->priv == NULL) {
3429                         kfree(new_crypt);
3430                         param->u.crypt.err = IEEE_CRYPT_ERR_CRYPT_INIT_FAILED;
3431                         ret = -EINVAL;
3432                         goto done;
3433                 }
3434
3435                 *crypt = new_crypt;
3436         }
3437
3438         if (param->u.crypt.key_len > 0 && (*crypt)->ops->set_key &&
3439             (*crypt)->ops->set_key(param->u.crypt.key,
3440             param->u.crypt.key_len, param->u.crypt.seq,
3441             (*crypt)->priv) < 0) {
3442                 printk(KERN_INFO "key setting failed\n");
3443                 param->u.crypt.err = IEEE_CRYPT_ERR_KEY_SET_FAILED;
3444                 ret = -EINVAL;
3445                 goto done;
3446         }
3447
3448  skip_host_crypt:
3449         if (param->u.crypt.set_tx) {
3450                 ieee->tx_keyidx = param->u.crypt.idx;
3451                 sec.active_key = param->u.crypt.idx;
3452                 sec.flags |= SEC_ACTIVE_KEY;
3453         } else
3454                 sec.flags &= ~SEC_ACTIVE_KEY;
3455
3456         if (param->u.crypt.alg != NULL) {
3457                 memcpy(sec.keys[param->u.crypt.idx],
3458                        param->u.crypt.key,
3459                        param->u.crypt.key_len);
3460                 sec.key_sizes[param->u.crypt.idx] = param->u.crypt.key_len;
3461                 sec.flags |= (1 << param->u.crypt.idx);
3462
3463                 if (strcmp(param->u.crypt.alg, "WEP") == 0) {
3464                         sec.flags |= SEC_LEVEL;
3465                         sec.level = SEC_LEVEL_1;
3466                 } else if (strcmp(param->u.crypt.alg, "TKIP") == 0) {
3467                         sec.flags |= SEC_LEVEL;
3468                         sec.level = SEC_LEVEL_2;
3469                 } else if (strcmp(param->u.crypt.alg, "CCMP") == 0) {
3470                         sec.flags |= SEC_LEVEL;
3471                         sec.level = SEC_LEVEL_3;
3472                 }
3473         }
3474  done:
3475         if (ieee->set_security)
3476                 ieee->set_security(ieee->dev, &sec);
3477
3478         /* Do not reset port if card is in Managed mode since resetting will
3479          * generate new IEEE 802.11 authentication which may end up in looping
3480          * with IEEE 802.1X.  If your hardware requires a reset after WEP
3481          * configuration (for example... Prism2), implement the reset_port in
3482          * the callbacks structures used to initialize the 802.11 stack. */
3483         if (ieee->reset_on_keychange &&
3484             ieee->iw_mode != IW_MODE_INFRA &&
3485             ieee->reset_port &&
3486             ieee->reset_port(ieee->dev)) {
3487                 printk(KERN_INFO "reset_port failed\n");
3488                 param->u.crypt.err = IEEE_CRYPT_ERR_CARD_CONF_FAILED;
3489                 return -EINVAL;
3490         }
3491
3492         return ret;
3493 }
3494
3495 inline struct sk_buff *rtllib_disauth_skb(struct rtllib_network *beacon,
3496                 struct rtllib_device *ieee, u16 asRsn)
3497 {
3498         struct sk_buff *skb;
3499         struct rtllib_disauth *disauth;
3500         int len = sizeof(struct rtllib_disauth) + ieee->tx_headroom;
3501
3502         skb = dev_alloc_skb(len);
3503         if (!skb)
3504                 return NULL;
3505
3506         skb_reserve(skb, ieee->tx_headroom);
3507
3508         disauth = (struct rtllib_disauth *) skb_put(skb,
3509                   sizeof(struct rtllib_disauth));
3510         disauth->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_DEAUTH);
3511         disauth->header.duration_id = 0;
3512
3513         memcpy(disauth->header.addr1, beacon->bssid, ETH_ALEN);
3514         memcpy(disauth->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
3515         memcpy(disauth->header.addr3, beacon->bssid, ETH_ALEN);
3516
3517         disauth->reason = cpu_to_le16(asRsn);
3518         return skb;
3519 }
3520
3521 inline struct sk_buff *rtllib_disassociate_skb(struct rtllib_network *beacon,
3522                 struct rtllib_device *ieee, u16 asRsn)
3523 {
3524         struct sk_buff *skb;
3525         struct rtllib_disassoc *disass;
3526         int len = sizeof(struct rtllib_disassoc) + ieee->tx_headroom;
3527         skb = dev_alloc_skb(len);
3528
3529         if (!skb)
3530                 return NULL;
3531
3532         skb_reserve(skb, ieee->tx_headroom);
3533
3534         disass = (struct rtllib_disassoc *) skb_put(skb,
3535                                          sizeof(struct rtllib_disassoc));
3536         disass->header.frame_ctl = cpu_to_le16(RTLLIB_STYPE_DISASSOC);
3537         disass->header.duration_id = 0;
3538
3539         memcpy(disass->header.addr1, beacon->bssid, ETH_ALEN);
3540         memcpy(disass->header.addr2, ieee->dev->dev_addr, ETH_ALEN);
3541         memcpy(disass->header.addr3, beacon->bssid, ETH_ALEN);
3542
3543         disass->reason = cpu_to_le16(asRsn);
3544         return skb;
3545 }
3546
3547 void SendDisassociation(struct rtllib_device *ieee, bool deauth, u16 asRsn)
3548 {
3549         struct rtllib_network *beacon = &ieee->current_network;
3550         struct sk_buff *skb;
3551
3552         if (deauth)
3553                 skb = rtllib_disauth_skb(beacon, ieee, asRsn);
3554         else
3555                 skb = rtllib_disassociate_skb(beacon, ieee, asRsn);
3556
3557         if (skb)
3558                 softmac_mgmt_xmit(skb, ieee);
3559 }
3560
3561 u8 rtllib_ap_sec_type(struct rtllib_device *ieee)
3562 {
3563         static u8 ccmp_ie[4] = {0x00, 0x50, 0xf2, 0x04};
3564         static u8 ccmp_rsn_ie[4] = {0x00, 0x0f, 0xac, 0x04};
3565         int wpa_ie_len = ieee->wpa_ie_len;
3566         struct rtllib_crypt_data *crypt;
3567         int encrypt;
3568
3569         crypt = ieee->crypt[ieee->tx_keyidx];
3570         encrypt = (ieee->current_network.capability & WLAN_CAPABILITY_PRIVACY)
3571                   || (ieee->host_encrypt && crypt && crypt->ops &&
3572                   (0 == strcmp(crypt->ops->name, "WEP")));
3573
3574         /* simply judge  */
3575         if (encrypt && (wpa_ie_len == 0)) {
3576                 return SEC_ALG_WEP;
3577         } else if ((wpa_ie_len != 0)) {
3578                 if (((ieee->wpa_ie[0] == 0xdd) &&
3579                     (!memcmp(&(ieee->wpa_ie[14]), ccmp_ie, 4))) ||
3580                     ((ieee->wpa_ie[0] == 0x30) &&
3581                     (!memcmp(&ieee->wpa_ie[10], ccmp_rsn_ie, 4))))
3582                         return SEC_ALG_CCMP;
3583                 else
3584                         return SEC_ALG_TKIP;
3585         } else {
3586                 return SEC_ALG_NONE;
3587         }
3588 }
3589
3590 int rtllib_wpa_supplicant_ioctl(struct rtllib_device *ieee, struct iw_point *p,
3591                                 u8 is_mesh)
3592 {
3593         struct ieee_param *param;
3594         int ret = 0;
3595
3596         down(&ieee->wx_sem);
3597
3598         if (p->length < sizeof(struct ieee_param) || !p->pointer) {
3599                 ret = -EINVAL;
3600                 goto out;
3601         }
3602
3603         param = kmalloc(p->length, GFP_KERNEL);
3604         if (param == NULL) {
3605                 ret = -ENOMEM;
3606                 goto out;
3607         }
3608         if (copy_from_user(param, p->pointer, p->length)) {
3609                 kfree(param);
3610                 ret = -EFAULT;
3611                 goto out;
3612         }
3613
3614         switch (param->cmd) {
3615         case IEEE_CMD_SET_WPA_PARAM:
3616                 ret = rtllib_wpa_set_param(ieee, param->u.wpa_param.name,
3617                                         param->u.wpa_param.value);
3618                 break;
3619
3620         case IEEE_CMD_SET_WPA_IE:
3621                 ret = rtllib_wpa_set_wpa_ie(ieee, param, p->length);
3622                 break;
3623
3624         case IEEE_CMD_SET_ENCRYPTION:
3625                 ret = rtllib_wpa_set_encryption(ieee, param, p->length, 0);
3626                 break;
3627
3628         case IEEE_CMD_MLME:
3629                 ret = rtllib_wpa_mlme(ieee, param->u.mlme.command,
3630                                    param->u.mlme.reason_code);
3631                 break;
3632
3633         default:
3634                 printk(KERN_INFO "Unknown WPA supplicant request: %d\n",
3635                        param->cmd);
3636                 ret = -EOPNOTSUPP;
3637                 break;
3638         }
3639
3640         if (ret == 0 && copy_to_user(p->pointer, param, p->length))
3641                 ret = -EFAULT;
3642
3643         kfree(param);
3644 out:
3645         up(&ieee->wx_sem);
3646
3647         return ret;
3648 }
3649 EXPORT_SYMBOL(rtllib_wpa_supplicant_ioctl);
3650
3651 void rtllib_MgntDisconnectIBSS(struct rtllib_device *rtllib)
3652 {
3653         u8      OpMode;
3654         u8      i;
3655         bool    bFilterOutNonAssociatedBSSID = false;
3656
3657         rtllib->state = RTLLIB_NOLINK;
3658
3659         for (i = 0; i < 6; i++)
3660                 rtllib->current_network.bssid[i] = 0x55;
3661
3662         rtllib->OpMode = RT_OP_MODE_NO_LINK;
3663         rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_BSSID,
3664                                 rtllib->current_network.bssid);
3665         OpMode = RT_OP_MODE_NO_LINK;
3666         rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_MEDIA_STATUS, &OpMode);
3667         rtllib_stop_send_beacons(rtllib);
3668
3669         bFilterOutNonAssociatedBSSID = false;
3670         rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_CECHK_BSSID,
3671                                 (u8 *)(&bFilterOutNonAssociatedBSSID));
3672         notify_wx_assoc_event(rtllib);
3673
3674 }
3675
3676 void rtllib_MlmeDisassociateRequest(struct rtllib_device *rtllib, u8 *asSta,
3677                                     u8 asRsn)
3678 {
3679         u8 i;
3680         u8      OpMode;
3681
3682         RemovePeerTS(rtllib, asSta);
3683
3684
3685         if (memcpy(rtllib->current_network.bssid, asSta, 6) == NULL) {
3686                 rtllib->state = RTLLIB_NOLINK;
3687
3688                 for (i = 0; i < 6; i++)
3689                         rtllib->current_network.bssid[i] = 0x22;
3690                 OpMode = RT_OP_MODE_NO_LINK;
3691                 rtllib->OpMode = RT_OP_MODE_NO_LINK;
3692                 rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_MEDIA_STATUS,
3693                                         (u8 *)(&OpMode));
3694                 rtllib_disassociate(rtllib);
3695
3696                 rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_BSSID,
3697                                         rtllib->current_network.bssid);
3698
3699         }
3700
3701 }
3702
3703 void
3704 rtllib_MgntDisconnectAP(
3705         struct rtllib_device *rtllib,
3706         u8 asRsn
3707 )
3708 {
3709         bool bFilterOutNonAssociatedBSSID = false;
3710
3711         bFilterOutNonAssociatedBSSID = false;
3712         rtllib->SetHwRegHandler(rtllib->dev, HW_VAR_CECHK_BSSID,
3713                                 (u8 *)(&bFilterOutNonAssociatedBSSID));
3714         rtllib_MlmeDisassociateRequest(rtllib, rtllib->current_network.bssid,
3715                                        asRsn);
3716
3717         rtllib->state = RTLLIB_NOLINK;
3718 }
3719
3720 bool rtllib_MgntDisconnect(struct rtllib_device *rtllib, u8 asRsn)
3721 {
3722         if (rtllib->ps != RTLLIB_PS_DISABLED)
3723                 rtllib->sta_wake_up(rtllib->dev);
3724
3725         if (rtllib->state == RTLLIB_LINKED) {
3726                 if (rtllib->iw_mode == IW_MODE_ADHOC)
3727                         rtllib_MgntDisconnectIBSS(rtllib);
3728                 if (rtllib->iw_mode == IW_MODE_INFRA)
3729                         rtllib_MgntDisconnectAP(rtllib, asRsn);
3730
3731         }
3732
3733         return true;
3734 }
3735 EXPORT_SYMBOL(rtllib_MgntDisconnect);
3736
3737 void notify_wx_assoc_event(struct rtllib_device *ieee)
3738 {
3739         union iwreq_data wrqu;
3740
3741         if (ieee->cannot_notify)
3742                 return;
3743
3744         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
3745         if (ieee->state == RTLLIB_LINKED)
3746                 memcpy(wrqu.ap_addr.sa_data, ieee->current_network.bssid,
3747                        ETH_ALEN);
3748         else {
3749
3750                 printk(KERN_INFO "%s(): Tell user space disconnected\n",
3751                        __func__);
3752                 memset(wrqu.ap_addr.sa_data, 0, ETH_ALEN);
3753         }
3754         wireless_send_event(ieee->dev, SIOCGIWAP, &wrqu, NULL);
3755 }
3756 EXPORT_SYMBOL(notify_wx_assoc_event);