drivers/staging/rtl8192e/rtllib_softmac.c

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