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