1// SPDX-License-Identifier: GPL-2.0
2/* IEEE 802.11 SoftMAC layer
3 * Copyright (c) 2005 Andrea Merello <andrea.merello@gmail.com>
4 *
5 * Mostly extracted from the rtl8180-sa2400 driver for the
6 * in-kernel generic ieee802.11 stack.
7 *
8 * Few lines might be stolen from other part of the rtllib
9 * stack. Copyright who own it's copyright
10 *
11 * WPA code stolen from the ipw2200 driver.
12 * Copyright who own it's copyright.
13 */
14#include "rtllib.h"
15
16#include <linux/random.h>
17#include <linux/delay.h>
18#include <linux/uaccess.h>
19#include <linux/etherdevice.h>
20#include <linux/ieee80211.h>
21
22static void rtllib_sta_wakeup(struct rtllib_device *ieee, short nl);
23
24static short rtllib_is_54g(struct rtllib_network *net)
25{
26	return (net->rates_ex_len > 0) || (net->rates_len > 4);
27}
28
29/* returns the total length needed for placing the RATE MFIE
30 * tag and the EXTENDED RATE MFIE tag if needed.
31 * It encludes two bytes per tag for the tag itself and its len
32 */
33static unsigned int rtllib_MFIE_rate_len(struct rtllib_device *ieee)
34{
35	unsigned int rate_len = 0;
36
37	rate_len = RTLLIB_CCK_RATE_LEN + 2;
38	rate_len += RTLLIB_OFDM_RATE_LEN + 2;
39
40	return rate_len;
41}
42
43/* place the MFIE rate, tag to the memory (double) pointed.
44 * Then it updates the pointer so that
45 * it points after the new MFIE tag added.
46 */
47static void rtllib_mfie_brate(struct rtllib_device *ieee, u8 **tag_p)
48{
49	u8 *tag = *tag_p;
50
51	*tag++ = MFIE_TYPE_RATES;
52	*tag++ = 4;
53	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_1MB;
54	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_2MB;
55	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_5MB;
56	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_CCK_RATE_11MB;
57
58	/* We may add an option for custom rates that specific HW
59	 * might support
60	 */
61	*tag_p = tag;
62}
63
64static void rtllib_mfie_grate(struct rtllib_device *ieee, u8 **tag_p)
65{
66	u8 *tag = *tag_p;
67
68	*tag++ = MFIE_TYPE_RATES_EX;
69	*tag++ = 8;
70	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_6MB;
71	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_9MB;
72	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_12MB;
73	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_18MB;
74	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_24MB;
75	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_36MB;
76	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_48MB;
77	*tag++ = RTLLIB_BASIC_RATE_MASK | RTLLIB_OFDM_RATE_54MB;
78
79	/* We may add an option for custom rates that specific HW might
80	 * support
81	 */
82	*tag_p = tag;
83}
84
85static void rtllib_wmm_info(struct rtllib_device *ieee, u8 **tag_p)
86{
87	u8 *tag = *tag_p;
88
89	*tag++ = MFIE_TYPE_GENERIC;
90	*tag++ = 7;
91	*tag++ = 0x00;
92	*tag++ = 0x50;
93	*tag++ = 0xf2;
94	*tag++ = 0x02;
95	*tag++ = 0x00;
96	*tag++ = 0x01;
97	*tag++ = MAX_SP_Len;
98	*tag_p = tag;
99}
100
101static void rtllib_turbo_info(struct rtllib_device *ieee, u8 **tag_p)
102{
103	u8 *tag = *tag_p;
104
105	*tag++ = MFIE_TYPE_GENERIC;
106	*tag++ = 7;
107	*tag++ = 0x00;
108	*tag++ = 0xe0;
109	*tag++ = 0x4c;
110	*tag++ = 0x01;
111	*tag++ = 0x02;
112	*tag++ = 0x11;
113	*tag++ = 0x00;
114
115	*tag_p = tag;
116	netdev_alert(ieee->dev, "This is enable turbo mode IE process\n");
117}
118
119static void enqueue_mgmt(struct rtllib_device *ieee, struct sk_buff *skb)
120{
121	int nh;
122
123	nh = (ieee->mgmt_queue_head + 1) % MGMT_QUEUE_NUM;
124
125/* if the queue is full but we have newer frames then
126 * just overwrites the oldest.
127 *
128 * if (nh == ieee->mgmt_queue_tail)
129 *		return -1;
130 */
131	ieee->mgmt_queue_head = nh;
132	ieee->mgmt_queue_ring[nh] = skb;
133}
134
135static void init_mgmt_queue(struct rtllib_device *ieee)
136{
137	ieee->mgmt_queue_tail = 0;
138	ieee->mgmt_queue_head = 0;
139}
140
141u8 mgnt_query_tx_rate_exclude_cck_rates(struct rtllib_device *ieee)
142{
143	u16	i;
144	u8	query_rate = 0;
145	u8	basic_rate;
146
147	for (i = 0; i < ieee->current_network.rates_len; i++) {
148		basic_rate = ieee->current_network.rates[i] & 0x7F;
149		if (!rtllib_is_cck_rate(basic_rate)) {
150			if (query_rate == 0) {
151				query_rate = basic_rate;
152			} else {
153				if (basic_rate < query_rate)
154					query_rate = basic_rate;
155			}
156		}
157	}
158
159	if (query_rate == 0) {
160		query_rate = 12;
161		netdev_info(ieee->dev, "No basic_rate found!!\n");
162	}
163	return query_rate;
164}
165
166static u8 mgnt_query_mgnt_frame_tx_rate(struct rtllib_device *ieee)
167{
168	struct rt_hi_throughput *ht_info = ieee->ht_info;
169	u8 rate;
170
171	if (ht_info->iot_action & HT_IOT_ACT_MGNT_USE_CCK_6M)
172		rate = 0x0c;
173	else
174		rate = ieee->basic_rate & 0x7f;
175
176	if (rate == 0)
177		rate = 0x02;
178
179	return rate;
180}
181
182inline void softmac_mgmt_xmit(struct sk_buff *skb, struct rtllib_device *ieee)
183{
184	unsigned long flags;
185	short single = ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE;
186	struct ieee80211_hdr_3addr  *header =
187		(struct ieee80211_hdr_3addr  *)skb->data;
188
189	struct cb_desc *tcb_desc = (struct cb_desc *)(skb->cb + 8);
190
191	spin_lock_irqsave(&ieee->lock, flags);
192
193	/* called with 2nd param 0, no mgmt lock required */
194	rtllib_sta_wakeup(ieee, 0);
195
196	if (ieee80211_is_beacon(header->frame_control))
197		tcb_desc->queue_index = BEACON_QUEUE;
198	else
199		tcb_desc->queue_index = MGNT_QUEUE;
200
201	if (ieee->disable_mgnt_queue)
202		tcb_desc->queue_index = HIGH_QUEUE;
203
204	tcb_desc->data_rate = mgnt_query_mgnt_frame_tx_rate(ieee);
205	tcb_desc->ratr_index = 7;
206	tcb_desc->tx_dis_rate_fallback = 1;
207	tcb_desc->tx_use_drv_assinged_rate = 1;
208	if (single) {
209		if (ieee->queue_stop) {
210			enqueue_mgmt(ieee, skb);
211		} else {
212			header->seq_ctrl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
213
214			if (ieee->seq_ctrl[0] == 0xFFF)
215				ieee->seq_ctrl[0] = 0;
216			else
217				ieee->seq_ctrl[0]++;
218
219			/* avoid watchdog triggers */
220			ieee->softmac_data_hard_start_xmit(skb, ieee->dev,
221							   ieee->basic_rate);
222		}
223
224		spin_unlock_irqrestore(&ieee->lock, flags);
225	} else {
226		spin_unlock_irqrestore(&ieee->lock, flags);
227		spin_lock_irqsave(&ieee->mgmt_tx_lock, flags);
228
229		header->seq_ctrl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
230
231		if (ieee->seq_ctrl[0] == 0xFFF)
232			ieee->seq_ctrl[0] = 0;
233		else
234			ieee->seq_ctrl[0]++;
235
236		/* check whether the managed packet queued greater than 5 */
237		if (!ieee->check_nic_enough_desc(ieee->dev,
238						 tcb_desc->queue_index) ||
239		    skb_queue_len(&ieee->skb_waitq[tcb_desc->queue_index]) ||
240		    ieee->queue_stop) {
241			/* insert the skb packet to the management queue
242			 *
243			 * as for the completion function, it does not need
244			 * to check it any more.
245			 */
246			netdev_info(ieee->dev,
247			       "%s():insert to waitqueue, queue_index:%d!\n",
248			       __func__, tcb_desc->queue_index);
249			skb_queue_tail(&ieee->skb_waitq[tcb_desc->queue_index],
250				       skb);
251		} else {
252			ieee->softmac_hard_start_xmit(skb, ieee->dev);
253		}
254		spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags);
255	}
256}
257
258static inline void
259softmac_ps_mgmt_xmit(struct sk_buff *skb,
260		     struct rtllib_device *ieee)
261{
262	short single = ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE;
263	struct ieee80211_hdr_3addr  *header =
264		(struct ieee80211_hdr_3addr  *)skb->data;
265	u16 fc, type, stype;
266	struct cb_desc *tcb_desc = (struct cb_desc *)(skb->cb + 8);
267
268	fc = le16_to_cpu(header->frame_control);
269	type = WLAN_FC_GET_TYPE(fc);
270	stype = WLAN_FC_GET_STYPE(fc);
271
272	if (stype != IEEE80211_STYPE_PSPOLL)
273		tcb_desc->queue_index = MGNT_QUEUE;
274	else
275		tcb_desc->queue_index = HIGH_QUEUE;
276
277	if (ieee->disable_mgnt_queue)
278		tcb_desc->queue_index = HIGH_QUEUE;
279
280	tcb_desc->data_rate = mgnt_query_mgnt_frame_tx_rate(ieee);
281	tcb_desc->ratr_index = 7;
282	tcb_desc->tx_dis_rate_fallback = 1;
283	tcb_desc->tx_use_drv_assinged_rate = 1;
284	if (single) {
285		if (type != RTLLIB_FTYPE_CTL) {
286			header->seq_ctrl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
287
288			if (ieee->seq_ctrl[0] == 0xFFF)
289				ieee->seq_ctrl[0] = 0;
290			else
291				ieee->seq_ctrl[0]++;
292		}
293		/* avoid watchdog triggers */
294		ieee->softmac_data_hard_start_xmit(skb, ieee->dev,
295						   ieee->basic_rate);
296
297	} else {
298		if (type != RTLLIB_FTYPE_CTL) {
299			header->seq_ctrl = cpu_to_le16(ieee->seq_ctrl[0] << 4);
300
301			if (ieee->seq_ctrl[0] == 0xFFF)
302				ieee->seq_ctrl[0] = 0;
303			else
304				ieee->seq_ctrl[0]++;
305		}
306		ieee->softmac_hard_start_xmit(skb, ieee->dev);
307	}
308}
309
310static inline struct sk_buff *rtllib_probe_req(struct rtllib_device *ieee)
311{
312	unsigned int len, rate_len;
313	u8 *tag;
314	struct sk_buff *skb;
315	struct rtllib_probe_request *req;
316
317	len = ieee->current_network.ssid_len;
318
319	rate_len = rtllib_MFIE_rate_len(ieee);
320
321	skb = dev_alloc_skb(sizeof(struct rtllib_probe_request) +
322			    2 + len + rate_len + ieee->tx_headroom);
323
324	if (!skb)
325		return NULL;
326
327	skb_reserve(skb, ieee->tx_headroom);
328
329	req = skb_put(skb, sizeof(struct rtllib_probe_request));
330	req->header.frame_control = cpu_to_le16(IEEE80211_STYPE_PROBE_REQ);
331	req->header.duration_id = 0;
332
333	eth_broadcast_addr(req->header.addr1);
334	ether_addr_copy(req->header.addr2, ieee->dev->dev_addr);
335	eth_broadcast_addr(req->header.addr3);
336
337	tag = skb_put(skb, len + 2 + rate_len);
338
339	*tag++ = MFIE_TYPE_SSID;
340	*tag++ = len;
341	memcpy(tag, ieee->current_network.ssid, len);
342	tag += len;
343
344	rtllib_mfie_brate(ieee, &tag);
345	rtllib_mfie_grate(ieee, &tag);
346
347	return skb;
348}
349
350/* Enables network monitor mode, all rx packets will be received. */
351void rtllib_enable_net_monitor_mode(struct net_device *dev,
352		bool init_state)
353{
354	struct rtllib_device *ieee = netdev_priv_rsl(dev);
355
356	netdev_info(dev, "========>Enter Monitor Mode\n");
357
358	ieee->allow_all_dest_addr_handler(dev, true, !init_state);
359}
360
361/* Disables network monitor mode. Only packets destinated to
362 * us will be received.
363 */
364void rtllib_disable_net_monitor_mode(struct net_device *dev, bool init_state)
365{
366	struct rtllib_device *ieee = netdev_priv_rsl(dev);
367
368	netdev_info(dev, "========>Exit Monitor Mode\n");
369
370	ieee->allow_all_dest_addr_handler(dev, false, !init_state);
371}
372
373static void rtllib_send_probe(struct rtllib_device *ieee)
374{
375	struct sk_buff *skb;
376
377	skb = rtllib_probe_req(ieee);
378	if (skb) {
379		softmac_mgmt_xmit(skb, ieee);
380		ieee->softmac_stats.tx_probe_rq++;
381	}
382}
383
384static void rtllib_send_probe_requests(struct rtllib_device *ieee)
385{
386	if (ieee->softmac_features & IEEE_SOFTMAC_PROBERQ) {
387		rtllib_send_probe(ieee);
388		rtllib_send_probe(ieee);
389	}
390}
391
392/* this performs syncro scan blocking the caller until all channels
393 * in the allowed channel map has been checked.
394 */
395static void rtllib_softmac_scan_syncro(struct rtllib_device *ieee)
396{
397	union iwreq_data wrqu;
398	short ch = 0;
399
400	ieee->be_scan_inprogress = true;
401
402	mutex_lock(&ieee->scan_mutex);
403
404	while (1) {
405		do {
406			ch++;
407			if (ch > MAX_CHANNEL_NUMBER)
408				goto out; /* scan completed */
409		} while (!ieee->active_channel_map[ch]);
410
411		/* this function can be called in two situations
412		 * 1- We have switched to ad-hoc mode and we are
413		 *    performing a complete syncro scan before conclude
414		 *    there are no interesting cell and to create a
415		 *    new one. In this case the link state is
416		 *    MAC80211_NOLINK until we found an interesting cell.
417		 *    If so the ieee8021_new_net, called by the RX path
418		 *    will set the state to MAC80211_LINKED, so we stop
419		 *    scanning
420		 * 2- We are linked and the root uses run iwlist scan.
421		 *    So we switch to MAC80211_LINKED_SCANNING to remember
422		 *    that we are still logically linked (not interested in
423		 *    new network events, despite for updating the net list,
424		 *    but we are temporarly 'unlinked' as the driver shall
425		 *    not filter RX frames and the channel is changing.
426		 * So the only situation in which are interested is to check
427		 * if the state become LINKED because of the #1 situation
428		 */
429
430		if (ieee->link_state == MAC80211_LINKED)
431			goto out;
432		if (ieee->sync_scan_hurryup) {
433			netdev_info(ieee->dev,
434				    "============>sync_scan_hurryup out\n");
435			goto out;
436		}
437
438		ieee->set_chan(ieee->dev, ch);
439		if (ieee->active_channel_map[ch] == 1)
440			rtllib_send_probe_requests(ieee);
441
442		/* this prevent excessive time wait when we
443		 * need to wait for a syncro scan to end..
444		 */
445		msleep_interruptible_rsl(RTLLIB_SOFTMAC_SCAN_TIME);
446	}
447out:
448	ieee->actscanning = false;
449	ieee->sync_scan_hurryup = 0;
450
451	mutex_unlock(&ieee->scan_mutex);
452
453	ieee->be_scan_inprogress = false;
454
455	memset(&wrqu, 0, sizeof(wrqu));
456	wireless_send_event(ieee->dev, SIOCGIWSCAN, &wrqu, NULL);
457}
458
459static void rtllib_softmac_scan_wq(void *data)
460{
461	struct rtllib_device *ieee = container_of_dwork_rsl(data,
462				     struct rtllib_device, softmac_scan_wq);
463	u8 last_channel = ieee->current_network.channel;
464
465	if (!ieee->ieee_up)
466		return;
467	if (rtllib_act_scanning(ieee, true))
468		return;
469
470	mutex_lock(&ieee->scan_mutex);
471
472	if (ieee->rf_power_state == rf_off) {
473		netdev_info(ieee->dev,
474			    "======>%s():rf state is rf_off, return\n",
475			    __func__);
476		goto out1;
477	}
478
479	do {
480		ieee->current_network.channel =
481			(ieee->current_network.channel + 1) %
482			MAX_CHANNEL_NUMBER;
483		if (ieee->scan_watch_dog++ > MAX_CHANNEL_NUMBER) {
484			if (!ieee->active_channel_map[ieee->current_network.channel])
485				ieee->current_network.channel = 6;
486			goto out; /* no good chans */
487		}
488	} while (!ieee->active_channel_map[ieee->current_network.channel]);
489
490	if (ieee->scanning_continue == 0)
491		goto out;
492
493	ieee->set_chan(ieee->dev, ieee->current_network.channel);
494
495	if (ieee->active_channel_map[ieee->current_network.channel] == 1)
496		rtllib_send_probe_requests(ieee);
497
498	schedule_delayed_work(&ieee->softmac_scan_wq,
499			      msecs_to_jiffies(RTLLIB_SOFTMAC_SCAN_TIME));
500
501	mutex_unlock(&ieee->scan_mutex);
502	return;
503
504out:
505	ieee->current_network.channel = last_channel;
506
507out1:
508	ieee->actscanning = false;
509	ieee->scan_watch_dog = 0;
510	ieee->scanning_continue = 0;
511	mutex_unlock(&ieee->scan_mutex);
512}
513
514static void rtllib_softmac_stop_scan(struct rtllib_device *ieee)
515{
516	mutex_lock(&ieee->scan_mutex);
517	ieee->scan_watch_dog = 0;
518	if (ieee->scanning_continue == 1) {
519		ieee->scanning_continue = 0;
520		ieee->actscanning = false;
521		mutex_unlock(&ieee->scan_mutex);
522		cancel_delayed_work_sync(&ieee->softmac_scan_wq);
523	} else {
524		mutex_unlock(&ieee->scan_mutex);
525	}
526}
527
528void rtllib_stop_scan(struct rtllib_device *ieee)
529{
530	if (ieee->softmac_features & IEEE_SOFTMAC_SCAN)
531		rtllib_softmac_stop_scan(ieee);
532}
533EXPORT_SYMBOL(rtllib_stop_scan);
534
535void rtllib_stop_scan_syncro(struct rtllib_device *ieee)
536{
537	if (ieee->softmac_features & IEEE_SOFTMAC_SCAN)
538		ieee->sync_scan_hurryup = 1;
539}
540EXPORT_SYMBOL(rtllib_stop_scan_syncro);
541
542bool rtllib_act_scanning(struct rtllib_device *ieee, bool sync_scan)
543{
544	if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
545		if (sync_scan)
546			return ieee->be_scan_inprogress;
547		else
548			return ieee->actscanning || ieee->be_scan_inprogress;
549	} else {
550		return test_bit(STATUS_SCANNING, &ieee->status);
551	}
552}
553EXPORT_SYMBOL(rtllib_act_scanning);
554
555/* called with ieee->lock held */
556static void rtllib_start_scan(struct rtllib_device *ieee)
557{
558	ieee->rtllib_ips_leave_wq(ieee->dev);
559
560	if (ieee->softmac_features & IEEE_SOFTMAC_SCAN) {
561		if (ieee->scanning_continue == 0) {
562			ieee->actscanning = true;
563			ieee->scanning_continue = 1;
564			schedule_delayed_work(&ieee->softmac_scan_wq, 0);
565		}
566	}
567}
568
569/* called with wx_mutex held */
570void rtllib_start_scan_syncro(struct rtllib_device *ieee)
571{
572	ieee->sync_scan_hurryup = 0;
573	if (ieee->softmac_features & IEEE_SOFTMAC_SCAN)
574		rtllib_softmac_scan_syncro(ieee);
575}
576EXPORT_SYMBOL(rtllib_start_scan_syncro);
577
578static inline struct sk_buff *
579rtllib_authentication_req(struct rtllib_network *beacon,
580			  struct rtllib_device *ieee,
581			  int challengelen, u8 *daddr)
582{
583	struct sk_buff *skb;
584	struct rtllib_authentication *auth;
585	int  len;
586
587	len = sizeof(struct rtllib_authentication) + challengelen +
588		     ieee->tx_headroom + 4;
589	skb = dev_alloc_skb(len);
590
591	if (!skb)
592		return NULL;
593
594	skb_reserve(skb, ieee->tx_headroom);
595
596	auth = skb_put(skb, sizeof(struct rtllib_authentication));
597
598	auth->header.frame_control = cpu_to_le16(IEEE80211_STYPE_AUTH);
599	if (challengelen)
600		auth->header.frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
601
602	auth->header.duration_id = cpu_to_le16(0x013a);
603	ether_addr_copy(auth->header.addr1, beacon->bssid);
604	ether_addr_copy(auth->header.addr2, ieee->dev->dev_addr);
605	ether_addr_copy(auth->header.addr3, beacon->bssid);
606	if (ieee->auth_mode == 0)
607		auth->algorithm = WLAN_AUTH_OPEN;
608	else if (ieee->auth_mode == 1)
609		auth->algorithm = cpu_to_le16(WLAN_AUTH_SHARED_KEY);
610	else if (ieee->auth_mode == 2)
611		auth->algorithm = WLAN_AUTH_OPEN;
612	auth->transaction = cpu_to_le16(ieee->associate_seq);
613	ieee->associate_seq++;
614
615	auth->status = cpu_to_le16(WLAN_STATUS_SUCCESS);
616
617	return skb;
618}
619
620static struct sk_buff *rtllib_null_func(struct rtllib_device *ieee, short pwr)
621{
622	struct sk_buff *skb;
623	struct ieee80211_hdr_3addr *hdr;
624
625	skb = dev_alloc_skb(sizeof(struct ieee80211_hdr_3addr) + ieee->tx_headroom);
626	if (!skb)
627		return NULL;
628
629	skb_reserve(skb, ieee->tx_headroom);
630
631	hdr = skb_put(skb, sizeof(struct ieee80211_hdr_3addr));
632
633	ether_addr_copy(hdr->addr1, ieee->current_network.bssid);
634	ether_addr_copy(hdr->addr2, ieee->dev->dev_addr);
635	ether_addr_copy(hdr->addr3, ieee->current_network.bssid);
636
637	hdr->frame_control = cpu_to_le16(RTLLIB_FTYPE_DATA |
638		IEEE80211_STYPE_NULLFUNC | IEEE80211_FCTL_TODS |
639		(pwr ? IEEE80211_FCTL_PM : 0));
640
641	return skb;
642}
643
644static struct sk_buff *rtllib_pspoll_func(struct rtllib_device *ieee)
645{
646	struct sk_buff *skb;
647	struct ieee80211_pspoll *hdr;
648
649	skb = dev_alloc_skb(sizeof(struct ieee80211_pspoll) + ieee->tx_headroom);
650	if (!skb)
651		return NULL;
652
653	skb_reserve(skb, ieee->tx_headroom);
654
655	hdr = skb_put(skb, sizeof(struct ieee80211_pspoll));
656
657	ether_addr_copy(hdr->bssid, ieee->current_network.bssid);
658	ether_addr_copy(hdr->ta, ieee->dev->dev_addr);
659
660	hdr->aid = cpu_to_le16(ieee->assoc_id | 0xc000);
661	hdr->frame_control = cpu_to_le16(RTLLIB_FTYPE_CTL | IEEE80211_STYPE_PSPOLL |
662			 IEEE80211_FCTL_PM);
663
664	return skb;
665}
666
667static inline int sec_is_in_pmkid_list(struct rtllib_device *ieee, u8 *bssid)
668{
669	int i = 0;
670
671	do {
672		if ((ieee->pmkid_list[i].used) &&
673		    (memcmp(ieee->pmkid_list[i].bssid, bssid, ETH_ALEN) == 0))
674			break;
675		i++;
676	} while (i < NUM_PMKID_CACHE);
677
678	if (i == NUM_PMKID_CACHE)
679		i = -1;
680	return i;
681}
682
683static inline struct sk_buff *
684rtllib_association_req(struct rtllib_network *beacon,
685		       struct rtllib_device *ieee)
686{
687	struct sk_buff *skb;
688	struct rtllib_assoc_request_frame *hdr;
689	u8 *tag, *ies;
690	int i;
691	u8 *ht_cap_buf = NULL;
692	u8 ht_cap_len = 0;
693	u8 *realtek_ie_buf = NULL;
694	u8 realtek_ie_len = 0;
695	int wpa_ie_len = ieee->wpa_ie_len;
696	int wps_ie_len = ieee->wps_ie_len;
697	unsigned int ckip_ie_len = 0;
698	unsigned int ccxrm_ie_len = 0;
699	unsigned int cxvernum_ie_len = 0;
700	struct lib80211_crypt_data *crypt;
701	int encrypt;
702	int	pmk_cache_idx;
703
704	unsigned int rate_len = (beacon->rates_len ?
705				(beacon->rates_len + 2) : 0) +
706				(beacon->rates_ex_len ? (beacon->rates_ex_len) +
707				2 : 0);
708
709	unsigned int wmm_info_len = beacon->qos_data.supported ? 9 : 0;
710	unsigned int turbo_info_len = beacon->turbo_enable ? 9 : 0;
711
712	int len = 0;
713
714	crypt = ieee->crypt_info.crypt[ieee->crypt_info.tx_keyidx];
715	if (crypt)
716		encrypt = crypt && crypt->ops &&
717			  ((strcmp(crypt->ops->name, "R-WEP") == 0 ||
718			  wpa_ie_len));
719	else
720		encrypt = 0;
721
722	if ((ieee->rtllib_ap_sec_type &&
723	    (ieee->rtllib_ap_sec_type(ieee) & SEC_ALG_TKIP)) ||
724	    ieee->forced_bg_mode) {
725		ieee->ht_info->enable_ht = 0;
726		ieee->mode = WIRELESS_MODE_G;
727	}
728
729	if (ieee->ht_info->current_ht_support && ieee->ht_info->enable_ht) {
730		ht_cap_buf = (u8 *)&ieee->ht_info->self_ht_cap;
731		ht_cap_len = sizeof(ieee->ht_info->self_ht_cap);
732		ht_construct_capability_element(ieee, ht_cap_buf, &ht_cap_len,
733					     encrypt, true);
734		if (ieee->ht_info->current_rt2rt_aggregation) {
735			realtek_ie_buf = ieee->ht_info->sz_rt2rt_agg_buf;
736			realtek_ie_len =
737				 sizeof(ieee->ht_info->sz_rt2rt_agg_buf);
738			ht_construct_rt2rt_agg_element(ieee, realtek_ie_buf,
739						   &realtek_ie_len);
740		}
741	}
742
743	if (beacon->ckip_supported)
744		ckip_ie_len = 30 + 2;
745	if (beacon->ccx_rm_enable)
746		ccxrm_ie_len = 6 + 2;
747	if (beacon->bss_ccx_ver_number >= 2)
748		cxvernum_ie_len = 5 + 2;
749
750	pmk_cache_idx = sec_is_in_pmkid_list(ieee, ieee->current_network.bssid);
751	if (pmk_cache_idx >= 0) {
752		wpa_ie_len += 18;
753		netdev_info(ieee->dev, "[PMK cache]: WPA2 IE length: %x\n",
754			    wpa_ie_len);
755	}
756	len = sizeof(struct rtllib_assoc_request_frame) + 2
757		+ beacon->ssid_len
758		+ rate_len
759		+ wpa_ie_len
760		+ wps_ie_len
761		+ wmm_info_len
762		+ turbo_info_len
763		+ ht_cap_len
764		+ realtek_ie_len
765		+ ckip_ie_len
766		+ ccxrm_ie_len
767		+ cxvernum_ie_len
768		+ ieee->tx_headroom;
769
770	skb = dev_alloc_skb(len);
771
772	if (!skb)
773		return NULL;
774
775	skb_reserve(skb, ieee->tx_headroom);
776
777	hdr = skb_put(skb, sizeof(struct rtllib_assoc_request_frame) + 2);
778
779	hdr->header.frame_control = cpu_to_le16(IEEE80211_STYPE_ASSOC_REQ);
780	hdr->header.duration_id = cpu_to_le16(37);
781	ether_addr_copy(hdr->header.addr1, beacon->bssid);
782	ether_addr_copy(hdr->header.addr2, ieee->dev->dev_addr);
783	ether_addr_copy(hdr->header.addr3, beacon->bssid);
784
785	ether_addr_copy(ieee->ap_mac_addr, beacon->bssid);
786
787	hdr->capability = cpu_to_le16(WLAN_CAPABILITY_ESS);
788	if (beacon->capability & WLAN_CAPABILITY_PRIVACY)
789		hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_PRIVACY);
790
791	if (beacon->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
792		hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_SHORT_PREAMBLE);
793
794	if (beacon->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME)
795		hdr->capability |= cpu_to_le16(WLAN_CAPABILITY_SHORT_SLOT_TIME);
796
797	hdr->listen_interval = cpu_to_le16(beacon->listen_interval);
798
799	hdr->info_element[0].id = MFIE_TYPE_SSID;
800
801	hdr->info_element[0].len = beacon->ssid_len;
802	skb_put_data(skb, beacon->ssid, beacon->ssid_len);
803
804	tag = skb_put(skb, rate_len);
805
806	if (beacon->rates_len) {
807		*tag++ = MFIE_TYPE_RATES;
808		*tag++ = beacon->rates_len;
809		for (i = 0; i < beacon->rates_len; i++)
810			*tag++ = beacon->rates[i];
811	}
812
813	if (beacon->rates_ex_len) {
814		*tag++ = MFIE_TYPE_RATES_EX;
815		*tag++ = beacon->rates_ex_len;
816		for (i = 0; i < beacon->rates_ex_len; i++)
817			*tag++ = beacon->rates_ex[i];
818	}
819
820	if (beacon->ckip_supported) {
821		static const u8 aironet_ie_oui[] = {0x00, 0x01, 0x66};
822		u8	ccx_aironet_buf[30];
823		struct octet_string os_ccx_aironet_ie;
824
825		memset(ccx_aironet_buf, 0, 30);
826		os_ccx_aironet_ie.octet = ccx_aironet_buf;
827		os_ccx_aironet_ie.Length = sizeof(ccx_aironet_buf);
828		memcpy(os_ccx_aironet_ie.octet, aironet_ie_oui,
829		       sizeof(aironet_ie_oui));
830
831		os_ccx_aironet_ie.octet[IE_CISCO_FLAG_POSITION] |=
832					 (SUPPORT_CKIP_PK | SUPPORT_CKIP_MIC);
833		tag = skb_put(skb, ckip_ie_len);
834		*tag++ = MFIE_TYPE_AIRONET;
835		*tag++ = os_ccx_aironet_ie.Length;
836		memcpy(tag, os_ccx_aironet_ie.octet, os_ccx_aironet_ie.Length);
837		tag += os_ccx_aironet_ie.Length;
838	}
839
840	if (beacon->ccx_rm_enable) {
841		static const u8 ccx_rm_cap_buf[] = {0x00, 0x40, 0x96, 0x01, 0x01,
842			0x00};
843		struct octet_string os_ccx_rm_cap;
844
845		os_ccx_rm_cap.octet = (u8 *)ccx_rm_cap_buf;
846		os_ccx_rm_cap.Length = sizeof(ccx_rm_cap_buf);
847		tag = skb_put(skb, ccxrm_ie_len);
848		*tag++ = MFIE_TYPE_GENERIC;
849		*tag++ = os_ccx_rm_cap.Length;
850		memcpy(tag, os_ccx_rm_cap.octet, os_ccx_rm_cap.Length);
851		tag += os_ccx_rm_cap.Length;
852	}
853
854	if (beacon->bss_ccx_ver_number >= 2) {
855		u8 ccx_ver_num_buf[] = {0x00, 0x40, 0x96, 0x03, 0x00};
856		struct octet_string os_ccx_ver_num;
857
858		ccx_ver_num_buf[4] = beacon->bss_ccx_ver_number;
859		os_ccx_ver_num.octet = ccx_ver_num_buf;
860		os_ccx_ver_num.Length = sizeof(ccx_ver_num_buf);
861		tag = skb_put(skb, cxvernum_ie_len);
862		*tag++ = MFIE_TYPE_GENERIC;
863		*tag++ = os_ccx_ver_num.Length;
864		memcpy(tag, os_ccx_ver_num.octet, os_ccx_ver_num.Length);
865		tag += os_ccx_ver_num.Length;
866	}
867	if (ieee->ht_info->current_ht_support && ieee->ht_info->enable_ht) {
868		if (ieee->ht_info->peer_ht_spec_ver != HT_SPEC_VER_EWC) {
869			tag = skb_put(skb, ht_cap_len);
870			*tag++ = MFIE_TYPE_HT_CAP;
871			*tag++ = ht_cap_len - 2;
872			memcpy(tag, ht_cap_buf, ht_cap_len - 2);
873			tag += ht_cap_len - 2;
874		}
875	}
876
877	if (wpa_ie_len) {
878		skb_put_data(skb, ieee->wpa_ie, ieee->wpa_ie_len);
879
880		if (pmk_cache_idx >= 0) {
881			tag = skb_put(skb, 18);
882			*tag = 1;
883			*(tag + 1) = 0;
884			memcpy((tag + 2), &ieee->pmkid_list[pmk_cache_idx].PMKID,
885			       16);
886		}
887	}
888	if (wmm_info_len) {
889		tag = skb_put(skb, wmm_info_len);
890		rtllib_wmm_info(ieee, &tag);
891	}
892
893	if (wps_ie_len && ieee->wps_ie)
894		skb_put_data(skb, ieee->wps_ie, wps_ie_len);
895
896	if (turbo_info_len) {
897		tag = skb_put(skb, turbo_info_len);
898		rtllib_turbo_info(ieee, &tag);
899	}
900
901	if (ieee->ht_info->current_ht_support && ieee->ht_info->enable_ht) {
902		if (ieee->ht_info->peer_ht_spec_ver == HT_SPEC_VER_EWC) {
903			tag = skb_put(skb, ht_cap_len);
904			*tag++ = MFIE_TYPE_GENERIC;
905			*tag++ = ht_cap_len - 2;
906			memcpy(tag, ht_cap_buf, ht_cap_len - 2);
907			tag += ht_cap_len - 2;
908		}
909
910		if (ieee->ht_info->current_rt2rt_aggregation) {
911			tag = skb_put(skb, realtek_ie_len);
912			*tag++ = MFIE_TYPE_GENERIC;
913			*tag++ = realtek_ie_len - 2;
914			memcpy(tag, realtek_ie_buf, realtek_ie_len - 2);
915		}
916	}
917
918	kfree(ieee->assocreq_ies);
919	ieee->assocreq_ies = NULL;
920	ies = &hdr->info_element[0].id;
921	ieee->assocreq_ies_len = (skb->data + skb->len) - ies;
922	ieee->assocreq_ies = kmemdup(ies, ieee->assocreq_ies_len, GFP_ATOMIC);
923	if (!ieee->assocreq_ies)
924		ieee->assocreq_ies_len = 0;
925
926	return skb;
927}
928
929static void rtllib_associate_abort(struct rtllib_device *ieee)
930{
931	unsigned long flags;
932
933	spin_lock_irqsave(&ieee->lock, flags);
934
935	ieee->associate_seq++;
936
937	/* don't scan, and avoid to have the RX path possibily
938	 * try again to associate. Even do not react to AUTH or
939	 * ASSOC response. Just wait for the retry wq to be scheduled.
940	 * Here we will check if there are good nets to associate
941	 * with, so we retry or just get back to NO_LINK and scanning
942	 */
943	if (ieee->link_state == RTLLIB_ASSOCIATING_AUTHENTICATING) {
944		netdev_dbg(ieee->dev, "Authentication failed\n");
945		ieee->softmac_stats.no_auth_rs++;
946	} else {
947		netdev_dbg(ieee->dev, "Association failed\n");
948		ieee->softmac_stats.no_ass_rs++;
949	}
950
951	ieee->link_state = RTLLIB_ASSOCIATING_RETRY;
952
953	schedule_delayed_work(&ieee->associate_retry_wq,
954			      RTLLIB_SOFTMAC_ASSOC_RETRY_TIME);
955
956	spin_unlock_irqrestore(&ieee->lock, flags);
957}
958
959static void rtllib_associate_abort_cb(struct timer_list *t)
960{
961	struct rtllib_device *dev = from_timer(dev, t, associate_timer);
962
963	rtllib_associate_abort(dev);
964}
965
966static void rtllib_associate_step1(struct rtllib_device *ieee, u8 *daddr)
967{
968	struct rtllib_network *beacon = &ieee->current_network;
969	struct sk_buff *skb;
970
971	netdev_dbg(ieee->dev, "Stopping scan\n");
972
973	ieee->softmac_stats.tx_auth_rq++;
974
975	skb = rtllib_authentication_req(beacon, ieee, 0, daddr);
976
977	if (!skb) {
978		rtllib_associate_abort(ieee);
979	} else {
980		ieee->link_state = RTLLIB_ASSOCIATING_AUTHENTICATING;
981		netdev_dbg(ieee->dev, "Sending authentication request\n");
982		softmac_mgmt_xmit(skb, ieee);
983		if (!timer_pending(&ieee->associate_timer)) {
984			ieee->associate_timer.expires = jiffies + (HZ / 2);
985			add_timer(&ieee->associate_timer);
986		}
987	}
988}
989
990static void rtllib_auth_challenge(struct rtllib_device *ieee, u8 *challenge,
991				  int chlen)
992{
993	u8 *c;
994	struct sk_buff *skb;
995	struct rtllib_network *beacon = &ieee->current_network;
996
997	ieee->associate_seq++;
998	ieee->softmac_stats.tx_auth_rq++;
999
1000	skb = rtllib_authentication_req(beacon, ieee, chlen + 2, beacon->bssid);
1001
1002	if (!skb) {
1003		rtllib_associate_abort(ieee);
1004	} else {
1005		c = skb_put(skb, chlen + 2);
1006		*(c++) = MFIE_TYPE_CHALLENGE;
1007		*(c++) = chlen;
1008		memcpy(c, challenge, chlen);
1009
1010		netdev_dbg(ieee->dev,
1011			   "Sending authentication challenge response\n");
1012
1013		rtllib_encrypt_fragment(ieee, skb,
1014					sizeof(struct ieee80211_hdr_3addr));
1015
1016		softmac_mgmt_xmit(skb, ieee);
1017		mod_timer(&ieee->associate_timer, jiffies + (HZ / 2));
1018	}
1019	kfree(challenge);
1020}
1021
1022static void rtllib_associate_step2(struct rtllib_device *ieee)
1023{
1024	struct sk_buff *skb;
1025	struct rtllib_network *beacon = &ieee->current_network;
1026
1027	del_timer_sync(&ieee->associate_timer);
1028
1029	netdev_dbg(ieee->dev, "Sending association request\n");
1030
1031	ieee->softmac_stats.tx_ass_rq++;
1032	skb = rtllib_association_req(beacon, ieee);
1033	if (!skb) {
1034		rtllib_associate_abort(ieee);
1035	} else {
1036		softmac_mgmt_xmit(skb, ieee);
1037		mod_timer(&ieee->associate_timer, jiffies + (HZ / 2));
1038	}
1039}
1040
1041static void rtllib_associate_complete_wq(void *data)
1042{
1043	struct rtllib_device *ieee = (struct rtllib_device *)
1044				     container_of(data,
1045				     struct rtllib_device,
1046				     associate_complete_wq);
1047	struct rt_pwr_save_ctrl *psc = &ieee->pwr_save_ctrl;
1048
1049	netdev_info(ieee->dev, "Associated successfully with %pM\n",
1050		    ieee->current_network.bssid);
1051	netdev_info(ieee->dev, "normal associate\n");
1052	notify_wx_assoc_event(ieee);
1053
1054	netif_carrier_on(ieee->dev);
1055	ieee->is_roaming = false;
1056	if (rtllib_is_54g(&ieee->current_network)) {
1057		ieee->rate = 108;
1058		netdev_info(ieee->dev, "Using G rates:%d\n", ieee->rate);
1059	} else {
1060		ieee->rate = 22;
1061		ieee->set_wireless_mode(ieee->dev, WIRELESS_MODE_B);
1062		netdev_info(ieee->dev, "Using B rates:%d\n", ieee->rate);
1063	}
1064	if (ieee->ht_info->current_ht_support && ieee->ht_info->enable_ht) {
1065		netdev_info(ieee->dev, "Successfully associated, ht enabled\n");
1066		ht_on_assoc_rsp(ieee);
1067	} else {
1068		netdev_info(ieee->dev,
1069			    "Successfully associated, ht not enabled(%d, %d)\n",
1070			    ieee->ht_info->current_ht_support,
1071			    ieee->ht_info->enable_ht);
1072		memset(ieee->dot11ht_oper_rate_set, 0, 16);
1073	}
1074	ieee->link_detect_info.slot_num = 2 * (1 +
1075				       ieee->current_network.beacon_interval /
1076				       500);
1077	if (ieee->link_detect_info.num_recv_bcn_in_period == 0 ||
1078	    ieee->link_detect_info.num_recv_data_in_period == 0) {
1079		ieee->link_detect_info.num_recv_bcn_in_period = 1;
1080		ieee->link_detect_info.num_recv_data_in_period = 1;
1081	}
1082	psc->lps_idle_count = 0;
1083	ieee->link_change(ieee->dev);
1084}
1085
1086static void rtllib_sta_send_associnfo(struct rtllib_device *ieee)
1087{
1088}
1089
1090static void rtllib_associate_complete(struct rtllib_device *ieee)
1091{
1092	del_timer_sync(&ieee->associate_timer);
1093
1094	ieee->link_state = MAC80211_LINKED;
1095	rtllib_sta_send_associnfo(ieee);
1096
1097	schedule_work(&ieee->associate_complete_wq);
1098}
1099
1100static void rtllib_associate_procedure_wq(void *data)
1101{
1102	struct rtllib_device *ieee = container_of_dwork_rsl(data,
1103				     struct rtllib_device,
1104				     associate_procedure_wq);
1105	rtllib_stop_scan_syncro(ieee);
1106	ieee->rtllib_ips_leave(ieee->dev);
1107	mutex_lock(&ieee->wx_mutex);
1108
1109	rtllib_stop_scan(ieee);
1110	ht_set_connect_bw_mode(ieee, HT_CHANNEL_WIDTH_20, HT_EXTCHNL_OFFSET_NO_EXT);
1111	if (ieee->rf_power_state == rf_off) {
1112		ieee->rtllib_ips_leave_wq(ieee->dev);
1113		mutex_unlock(&ieee->wx_mutex);
1114		return;
1115	}
1116	ieee->associate_seq = 1;
1117
1118	rtllib_associate_step1(ieee, ieee->current_network.bssid);
1119
1120	mutex_unlock(&ieee->wx_mutex);
1121}
1122
1123inline void rtllib_softmac_new_net(struct rtllib_device *ieee,
1124				   struct rtllib_network *net)
1125{
1126	u8 tmp_ssid[IW_ESSID_MAX_SIZE + 1];
1127	int tmp_ssid_len = 0;
1128
1129	short apset, ssidset, ssidbroad, apmatch, ssidmatch;
1130
1131	/* we are interested in new only if we are not associated
1132	 * and we are not associating / authenticating
1133	 */
1134	if (ieee->link_state != MAC80211_NOLINK)
1135		return;
1136
1137	if ((ieee->iw_mode == IW_MODE_INFRA) && !(net->capability &
1138	    WLAN_CAPABILITY_ESS))
1139		return;
1140
1141	if (ieee->iw_mode == IW_MODE_INFRA) {
1142		/* if the user specified the AP MAC, we need also the essid
1143		 * This could be obtained by beacons or, if the network does not
1144		 * broadcast it, it can be put manually.
1145		 */
1146		apset = ieee->wap_set;
1147		ssidset = ieee->ssid_set;
1148		ssidbroad =  !(net->ssid_len == 0 || net->ssid[0] == '\0');
1149		apmatch = (memcmp(ieee->current_network.bssid, net->bssid,
1150				  ETH_ALEN) == 0);
1151		if (!ssidbroad) {
1152			ssidmatch = (ieee->current_network.ssid_len ==
1153				    net->hidden_ssid_len) &&
1154				    (!strncmp(ieee->current_network.ssid,
1155				    net->hidden_ssid, net->hidden_ssid_len));
1156			if (net->hidden_ssid_len > 0) {
1157				strncpy(net->ssid, net->hidden_ssid,
1158					net->hidden_ssid_len);
1159				net->ssid_len = net->hidden_ssid_len;
1160				ssidbroad = 1;
1161			}
1162		} else {
1163			ssidmatch =
1164			   (ieee->current_network.ssid_len == net->ssid_len) &&
1165			   (!strncmp(ieee->current_network.ssid, net->ssid,
1166			   net->ssid_len));
1167		}
1168
1169		/* if the user set the AP check if match.
1170		 * if the network does not broadcast essid we check the
1171		 *	 user supplied ANY essid
1172		 * if the network does broadcast and the user does not set
1173		 *	 essid it is OK
1174		 * if the network does broadcast and the user did set essid
1175		 * check if essid match
1176		 * if the ap is not set, check that the user set the bssid
1177		 * and the network does broadcast and that those two bssid match
1178		 */
1179		if ((apset && apmatch &&
1180		   ((ssidset && ssidbroad && ssidmatch) ||
1181		   (ssidbroad && !ssidset) || (!ssidbroad && ssidset))) ||
1182		   (!apset && ssidset && ssidbroad && ssidmatch) ||
1183		   (ieee->is_roaming && ssidset && ssidbroad && ssidmatch)) {
1184			/* Save the essid so that if it is hidden, it is
1185			 * replaced with the essid provided by the user.
1186			 */
1187			if (!ssidbroad) {
1188				memcpy(tmp_ssid, ieee->current_network.ssid,
1189				       ieee->current_network.ssid_len);
1190				tmp_ssid_len = ieee->current_network.ssid_len;
1191			}
1192			memcpy(&ieee->current_network, net,
1193				sizeof(ieee->current_network));
1194			if (!ssidbroad) {
1195				memcpy(ieee->current_network.ssid, tmp_ssid,
1196				       tmp_ssid_len);
1197				ieee->current_network.ssid_len = tmp_ssid_len;
1198			}
1199			netdev_info(ieee->dev,
1200				    "Linking with %s,channel:%d, qos:%d, myHT:%d, networkHT:%d, mode:%x cur_net.flags:0x%x\n",
1201				    ieee->current_network.ssid,
1202				    ieee->current_network.channel,
1203				    ieee->current_network.qos_data.supported,
1204				    ieee->ht_info->enable_ht,
1205				    ieee->current_network.bssht.bd_support_ht,
1206				    ieee->current_network.mode,
1207				    ieee->current_network.flags);
1208
1209			if ((rtllib_act_scanning(ieee, false)) &&
1210			    !(ieee->softmac_features & IEEE_SOFTMAC_SCAN))
1211				rtllib_stop_scan_syncro(ieee);
1212
1213			ht_reset_iot_setting(ieee->ht_info);
1214			ieee->wmm_acm = 0;
1215			if (ieee->iw_mode == IW_MODE_INFRA) {
1216				/* Join the network for the first time */
1217				ieee->asoc_retry_count = 0;
1218				if ((ieee->current_network.qos_data.supported == 1) &&
1219				    ieee->current_network.bssht.bd_support_ht)
1220					ht_reset_self_and_save_peer_setting(ieee,
1221						 &ieee->current_network);
1222				else
1223					ieee->ht_info->current_ht_support = false;
1224
1225				ieee->link_state = RTLLIB_ASSOCIATING;
1226				schedule_delayed_work(&ieee->associate_procedure_wq, 0);
1227			} else {
1228				if (rtllib_is_54g(&ieee->current_network)) {
1229					ieee->rate = 108;
1230					ieee->set_wireless_mode(ieee->dev, WIRELESS_MODE_G);
1231					netdev_info(ieee->dev,
1232						    "Using G rates\n");
1233				} else {
1234					ieee->rate = 22;
1235					ieee->set_wireless_mode(ieee->dev, WIRELESS_MODE_B);
1236					netdev_info(ieee->dev,
1237						    "Using B rates\n");
1238				}
1239				memset(ieee->dot11ht_oper_rate_set, 0, 16);
1240				ieee->link_state = MAC80211_LINKED;
1241			}
1242		}
1243	}
1244}
1245
1246static void rtllib_softmac_check_all_nets(struct rtllib_device *ieee)
1247{
1248	unsigned long flags;
1249	struct rtllib_network *target;
1250
1251	spin_lock_irqsave(&ieee->lock, flags);
1252
1253	list_for_each_entry(target, &ieee->network_list, list) {
1254		/* if the state become different that NOLINK means
1255		 * we had found what we are searching for
1256		 */
1257
1258		if (ieee->link_state != MAC80211_NOLINK)
1259			break;
1260
1261		if (ieee->scan_age == 0 || time_after(target->last_scanned +
1262		    ieee->scan_age, jiffies))
1263			rtllib_softmac_new_net(ieee, target);
1264	}
1265	spin_unlock_irqrestore(&ieee->lock, flags);
1266}
1267
1268static inline int auth_parse(struct net_device *dev, struct sk_buff *skb,
1269			     u8 **challenge, int *chlen)
1270{
1271	struct rtllib_authentication *a;
1272	u8 *t;
1273
1274	if (skb->len <  (sizeof(struct rtllib_authentication) -
1275	    sizeof(struct rtllib_info_element))) {
1276		netdev_dbg(dev, "invalid len in auth resp: %d\n", skb->len);
1277		return -EINVAL;
1278	}
1279	*challenge = NULL;
1280	a = (struct rtllib_authentication *)skb->data;
1281	if (skb->len > (sizeof(struct rtllib_authentication) + 3)) {
1282		t = skb->data + sizeof(struct rtllib_authentication);
1283
1284		if (*(t++) == MFIE_TYPE_CHALLENGE) {
1285			*chlen = *(t++);
1286			*challenge = kmemdup(t, *chlen, GFP_ATOMIC);
1287			if (!*challenge)
1288				return -ENOMEM;
1289		}
1290	}
1291
1292	if (a->status) {
1293		netdev_dbg(dev, "auth_parse() failed\n");
1294		return -EINVAL;
1295	}
1296
1297	return 0;
1298}
1299
1300static inline u16 assoc_parse(struct rtllib_device *ieee, struct sk_buff *skb,
1301			      int *aid)
1302{
1303	struct rtllib_assoc_response_frame *response_head;
1304	u16 status_code;
1305
1306	if (skb->len <  sizeof(struct rtllib_assoc_response_frame)) {
1307		netdev_dbg(ieee->dev, "Invalid len in auth resp: %d\n",
1308			   skb->len);
1309		return 0xcafe;
1310	}
1311
1312	response_head = (struct rtllib_assoc_response_frame *)skb->data;
1313	*aid = le16_to_cpu(response_head->aid) & 0x3fff;
1314
1315	status_code = le16_to_cpu(response_head->status);
1316	if ((status_code == WLAN_STATUS_ASSOC_DENIED_RATES ||
1317	   status_code == WLAN_STATUS_CAPS_UNSUPPORTED) &&
1318	   ((ieee->mode == WIRELESS_MODE_G) &&
1319	   (ieee->current_network.mode == WIRELESS_MODE_N_24G) &&
1320	   (ieee->asoc_retry_count++ < (RT_ASOC_RETRY_LIMIT - 1)))) {
1321		ieee->ht_info->iot_action |= HT_IOT_ACT_PURE_N_MODE;
1322	} else {
1323		ieee->asoc_retry_count = 0;
1324	}
1325
1326	return le16_to_cpu(response_head->status);
1327}
1328
1329void rtllib_sta_ps_send_null_frame(struct rtllib_device *ieee, short pwr)
1330{
1331	struct sk_buff *buf = rtllib_null_func(ieee, pwr);
1332
1333	if (buf)
1334		softmac_ps_mgmt_xmit(buf, ieee);
1335}
1336EXPORT_SYMBOL(rtllib_sta_ps_send_null_frame);
1337
1338void rtllib_sta_ps_send_pspoll_frame(struct rtllib_device *ieee)
1339{
1340	struct sk_buff *buf = rtllib_pspoll_func(ieee);
1341
1342	if (buf)
1343		softmac_ps_mgmt_xmit(buf, ieee);
1344}
1345
1346static short rtllib_sta_ps_sleep(struct rtllib_device *ieee, u64 *time)
1347{
1348	int timeout;
1349	u8 dtim;
1350	struct rt_pwr_save_ctrl *psc = &ieee->pwr_save_ctrl;
1351
1352	if (ieee->lps_delay_cnt) {
1353		ieee->lps_delay_cnt--;
1354		return 0;
1355	}
1356
1357	dtim = ieee->current_network.dtim_data;
1358	if (!(dtim & RTLLIB_DTIM_VALID))
1359		return 0;
1360	timeout = ieee->current_network.beacon_interval;
1361	ieee->current_network.dtim_data = RTLLIB_DTIM_INVALID;
1362	/* there's no need to nofity AP that I find you buffered
1363	 * with broadcast packet
1364	 */
1365	if (dtim & (RTLLIB_DTIM_UCAST & ieee->ps))
1366		return 2;
1367
1368	if (!time_after(jiffies,
1369			dev_trans_start(ieee->dev) + msecs_to_jiffies(timeout)))
1370		return 0;
1371	if (!time_after(jiffies,
1372			ieee->last_rx_ps_time + msecs_to_jiffies(timeout)))
1373		return 0;
1374	if ((ieee->softmac_features & IEEE_SOFTMAC_SINGLE_QUEUE) &&
1375	    (ieee->mgmt_queue_tail != ieee->mgmt_queue_head))
1376		return 0;
1377
1378	if (time) {
1379		if (ieee->awake_pkt_sent) {
1380			psc->lps_awake_intvl = 1;
1381		} else {
1382			u8 max_period = 5;
1383
1384			if (psc->lps_awake_intvl == 0)
1385				psc->lps_awake_intvl = 1;
1386			psc->lps_awake_intvl = (psc->lps_awake_intvl >=
1387					       max_period) ? max_period :
1388					       (psc->lps_awake_intvl + 1);
1389		}
1390		{
1391			u8 lps_awake_intvl_tmp = 0;
1392			u8 period = ieee->current_network.dtim_period;
1393			u8 count = ieee->current_network.tim.tim_count;
1394
1395			if (count == 0) {
1396				if (psc->lps_awake_intvl > period)
1397					lps_awake_intvl_tmp = period +
1398						 (psc->lps_awake_intvl -
1399						 period) -
1400						 ((psc->lps_awake_intvl - period) %
1401						 period);
1402				else
1403					lps_awake_intvl_tmp = psc->lps_awake_intvl;
1404
1405			} else {
1406				if (psc->lps_awake_intvl >
1407				    ieee->current_network.tim.tim_count)
1408					lps_awake_intvl_tmp = count +
1409					(psc->lps_awake_intvl - count) -
1410					((psc->lps_awake_intvl - count) % period);
1411				else
1412					lps_awake_intvl_tmp = psc->lps_awake_intvl;
1413			}
1414
1415		*time = ieee->current_network.last_dtim_sta_time
1416			+ msecs_to_jiffies(ieee->current_network.beacon_interval *
1417			lps_awake_intvl_tmp);
1418	}
1419	}
1420
1421	return 1;
1422}
1423
1424static inline void rtllib_sta_ps(struct work_struct *work)
1425{
1426	struct rtllib_device *ieee;
1427	u64 time;
1428	short sleep;
1429	unsigned long flags, flags2;
1430
1431	ieee = container_of(work, struct rtllib_device, ps_task);
1432
1433	spin_lock_irqsave(&ieee->lock, flags);
1434
1435	if ((ieee->ps == RTLLIB_PS_DISABLED ||
1436	     ieee->iw_mode != IW_MODE_INFRA ||
1437	     ieee->link_state != MAC80211_LINKED)) {
1438		spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
1439		rtllib_sta_wakeup(ieee, 1);
1440
1441		spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
1442	}
1443	sleep = rtllib_sta_ps_sleep(ieee, &time);
1444	/* 2 wake, 1 sleep, 0 do nothing */
1445	if (sleep == 0)
1446		goto out;
1447	if (sleep == 1) {
1448		if (ieee->sta_sleep == LPS_IS_SLEEP) {
1449			ieee->enter_sleep_state(ieee->dev, time);
1450		} else if (ieee->sta_sleep == LPS_IS_WAKE) {
1451			spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
1452
1453			if (ieee->ps_is_queue_empty(ieee->dev)) {
1454				ieee->sta_sleep = LPS_WAIT_NULL_DATA_SEND;
1455				ieee->ack_tx_to_ieee = 1;
1456				rtllib_sta_ps_send_null_frame(ieee, 1);
1457				ieee->ps_time = time;
1458			}
1459			spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
1460		}
1461
1462		ieee->awake_pkt_sent = false;
1463
1464	} else if (sleep == 2) {
1465		spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
1466
1467		rtllib_sta_wakeup(ieee, 1);
1468
1469		spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
1470	}
1471
1472out:
1473	spin_unlock_irqrestore(&ieee->lock, flags);
1474}
1475
1476static void rtllib_sta_wakeup(struct rtllib_device *ieee, short nl)
1477{
1478	if (ieee->sta_sleep == LPS_IS_WAKE) {
1479		if (nl) {
1480			if (ieee->ht_info->iot_action &
1481			    HT_IOT_ACT_NULL_DATA_POWER_SAVING) {
1482				ieee->ack_tx_to_ieee = 1;
1483				rtllib_sta_ps_send_null_frame(ieee, 0);
1484			} else {
1485				ieee->ack_tx_to_ieee = 1;
1486				rtllib_sta_ps_send_pspoll_frame(ieee);
1487			}
1488		}
1489		return;
1490	}
1491
1492	if (ieee->sta_sleep == LPS_IS_SLEEP)
1493		ieee->sta_wake_up(ieee->dev);
1494	if (nl) {
1495		if (ieee->ht_info->iot_action &
1496		    HT_IOT_ACT_NULL_DATA_POWER_SAVING) {
1497			ieee->ack_tx_to_ieee = 1;
1498			rtllib_sta_ps_send_null_frame(ieee, 0);
1499		} else {
1500			ieee->ack_tx_to_ieee = 1;
1501			ieee->polling = true;
1502			rtllib_sta_ps_send_pspoll_frame(ieee);
1503		}
1504
1505	} else {
1506		ieee->sta_sleep = LPS_IS_WAKE;
1507		ieee->polling = false;
1508	}
1509}
1510
1511void rtllib_ps_tx_ack(struct rtllib_device *ieee, short success)
1512{
1513	unsigned long flags, flags2;
1514
1515	spin_lock_irqsave(&ieee->lock, flags);
1516
1517	if (ieee->sta_sleep == LPS_WAIT_NULL_DATA_SEND) {
1518		/* Null frame with PS bit set */
1519		if (success) {
1520			ieee->sta_sleep = LPS_IS_SLEEP;
1521			ieee->enter_sleep_state(ieee->dev, ieee->ps_time);
1522		}
1523		/* if the card report not success we can't be sure the AP
1524		 * has not RXed so we can't assume the AP believe us awake
1525		 */
1526	} else {/* 21112005 - tx again null without PS bit if lost */
1527
1528		if ((ieee->sta_sleep == LPS_IS_WAKE) && !success) {
1529			spin_lock_irqsave(&ieee->mgmt_tx_lock, flags2);
1530			if (ieee->ht_info->iot_action &
1531			    HT_IOT_ACT_NULL_DATA_POWER_SAVING)
1532				rtllib_sta_ps_send_null_frame(ieee, 0);
1533			else
1534				rtllib_sta_ps_send_pspoll_frame(ieee);
1535			spin_unlock_irqrestore(&ieee->mgmt_tx_lock, flags2);
1536		}
1537	}
1538	spin_unlock_irqrestore(&ieee->lock, flags);
1539}
1540EXPORT_SYMBOL(rtllib_ps_tx_ack);
1541
1542static void rtllib_process_action(struct rtllib_device *ieee,
1543				  struct sk_buff *skb)
1544{
1545	u8 *act = skb->data + RTLLIB_3ADDR_LEN;
1546	u8 category = 0;
1547
1548	category = *act;
1549	act++;
1550	switch (category) {
1551	case ACT_CAT_BA:
1552		switch (*act) {
1553		case ACT_ADDBAREQ:
1554			rtllib_rx_add_ba_req(ieee, skb);
1555			break;
1556		case ACT_ADDBARSP:
1557			rtllib_rx_add_ba_rsp(ieee, skb);
1558			break;
1559		case ACT_DELBA:
1560			rtllib_rx_DELBA(ieee, skb);
1561			break;
1562		}
1563		break;
1564	default:
1565		break;
1566	}
1567}
1568
1569static inline int
1570rtllib_rx_assoc_resp(struct rtllib_device *ieee, struct sk_buff *skb,
1571		     struct rtllib_rx_stats *rx_stats)
1572{
1573	u16 errcode;
1574	int aid;
1575	u8 *ies;
1576	struct rtllib_assoc_response_frame *assoc_resp;
1577	struct ieee80211_hdr_3addr *header = (struct ieee80211_hdr_3addr *)skb->data;
1578	u16 frame_ctl = le16_to_cpu(header->frame_control);
1579
1580	netdev_dbg(ieee->dev, "received [RE]ASSOCIATION RESPONSE (%d)\n",
1581		   WLAN_FC_GET_STYPE(frame_ctl));
1582
1583	if ((ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) &&
1584	     ieee->link_state == RTLLIB_ASSOCIATING_AUTHENTICATED &&
1585	     (ieee->iw_mode == IW_MODE_INFRA)) {
1586		errcode = assoc_parse(ieee, skb, &aid);
1587		if (!errcode) {
1588			struct rtllib_network *network =
1589				 kzalloc(sizeof(struct rtllib_network),
1590				 GFP_ATOMIC);
1591
1592			if (!network)
1593				return 1;
1594			ieee->link_state = MAC80211_LINKED;
1595			ieee->assoc_id = aid;
1596			ieee->softmac_stats.rx_ass_ok++;
1597			/* station support qos */
1598			/* Let the register setting default with Legacy station */
1599			assoc_resp = (struct rtllib_assoc_response_frame *)skb->data;
1600			if (ieee->current_network.qos_data.supported == 1) {
1601				if (rtllib_parse_info_param(ieee, assoc_resp->info_element,
1602							rx_stats->len - sizeof(*assoc_resp),
1603							network, rx_stats)) {
1604					kfree(network);
1605					return 1;
1606				}
1607				memcpy(ieee->ht_info->peer_ht_cap_buf,
1608				       network->bssht.bd_ht_cap_buf,
1609				       network->bssht.bd_ht_cap_len);
1610				memcpy(ieee->ht_info->peer_ht_info_buf,
1611				       network->bssht.bd_ht_info_buf,
1612				       network->bssht.bd_ht_info_len);
1613				ieee->handle_assoc_response(ieee->dev,
1614					(struct rtllib_assoc_response_frame *)header, network);
1615			}
1616			kfree(network);
1617
1618			kfree(ieee->assocresp_ies);
1619			ieee->assocresp_ies = NULL;
1620			ies = &assoc_resp->info_element[0].id;
1621			ieee->assocresp_ies_len = (skb->data + skb->len) - ies;
1622			ieee->assocresp_ies = kmemdup(ies,
1623						      ieee->assocresp_ies_len,
1624						      GFP_ATOMIC);
1625			if (!ieee->assocresp_ies)
1626				ieee->assocresp_ies_len = 0;
1627
1628			rtllib_associate_complete(ieee);
1629		} else {
1630			/* aid could not been allocated */
1631			ieee->softmac_stats.rx_ass_err++;
1632			netdev_info(ieee->dev,
1633				    "Association response status code 0x%x\n",
1634				    errcode);
1635			if (ieee->asoc_retry_count < RT_ASOC_RETRY_LIMIT)
1636				schedule_delayed_work(&ieee->associate_procedure_wq, 0);
1637			else
1638				rtllib_associate_abort(ieee);
1639		}
1640	}
1641	return 0;
1642}
1643
1644static void rtllib_rx_auth_resp(struct rtllib_device *ieee, struct sk_buff *skb)
1645{
1646	int errcode;
1647	u8 *challenge;
1648	int chlen = 0;
1649	bool support_nmode = true, half_support_nmode = false;
1650
1651	errcode = auth_parse(ieee->dev, skb, &challenge, &chlen);
1652
1653	if (errcode) {
1654		ieee->softmac_stats.rx_auth_rs_err++;
1655		netdev_info(ieee->dev,
1656			    "Authentication response status code %d", errcode);
1657		rtllib_associate_abort(ieee);
1658		return;
1659	}
1660
1661	if (ieee->open_wep || !challenge) {
1662		ieee->link_state = RTLLIB_ASSOCIATING_AUTHENTICATED;
1663		ieee->softmac_stats.rx_auth_rs_ok++;
1664		if (!(ieee->ht_info->iot_action & HT_IOT_ACT_PURE_N_MODE)) {
1665			if (!ieee->get_nmode_support_by_sec_cfg(ieee->dev)) {
1666				if (is_ht_half_nmode_aps(ieee)) {
1667					support_nmode = true;
1668					half_support_nmode = true;
1669				} else {
1670					support_nmode = false;
1671					half_support_nmode = false;
1672				}
1673			}
1674		}
1675		/* Dummy wirless mode setting to avoid encryption issue */
1676		if (support_nmode) {
1677			ieee->set_wireless_mode(ieee->dev,
1678					      ieee->current_network.mode);
1679		} else {
1680			/*TODO*/
1681			ieee->set_wireless_mode(ieee->dev, WIRELESS_MODE_G);
1682		}
1683
1684		if ((ieee->current_network.mode == WIRELESS_MODE_N_24G) &&
1685		    half_support_nmode) {
1686			netdev_info(ieee->dev, "======>enter half N mode\n");
1687			ieee->half_wireless_n24g_mode = true;
1688		} else {
1689			ieee->half_wireless_n24g_mode = false;
1690		}
1691		rtllib_associate_step2(ieee);
1692	} else {
1693		rtllib_auth_challenge(ieee, challenge,  chlen);
1694	}
1695}
1696
1697static inline int
1698rtllib_rx_auth(struct rtllib_device *ieee, struct sk_buff *skb,
1699	       struct rtllib_rx_stats *rx_stats)
1700{
1701	if (ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) {
1702		if (ieee->link_state == RTLLIB_ASSOCIATING_AUTHENTICATING &&
1703		    (ieee->iw_mode == IW_MODE_INFRA)) {
1704			netdev_dbg(ieee->dev,
1705				   "Received authentication response");
1706			rtllib_rx_auth_resp(ieee, skb);
1707		}
1708	}
1709	return 0;
1710}
1711
1712static inline int
1713rtllib_rx_deauth(struct rtllib_device *ieee, struct sk_buff *skb)
1714{
1715	struct ieee80211_hdr_3addr *header = (struct ieee80211_hdr_3addr *)skb->data;
1716	u16 frame_ctl;
1717
1718	if (memcmp(header->addr3, ieee->current_network.bssid, ETH_ALEN) != 0)
1719		return 0;
1720
1721	/* FIXME for now repeat all the association procedure
1722	 * both for disassociation and deauthentication
1723	 */
1724	if ((ieee->softmac_features & IEEE_SOFTMAC_ASSOCIATE) &&
1725	    ieee->link_state == MAC80211_LINKED &&
1726	    (ieee->iw_mode == IW_MODE_INFRA)) {
1727		frame_ctl = le16_to_cpu(header->frame_control);
1728		netdev_info(ieee->dev,
1729			    "==========>received disassoc/deauth(%x) frame, reason code:%x\n",
1730			    WLAN_FC_GET_STYPE(frame_ctl),
1731			    ((struct rtllib_disassoc *)skb->data)->reason);
1732		ieee->link_state = RTLLIB_ASSOCIATING;
1733		ieee->softmac_stats.reassoc++;
1734		ieee->is_roaming = true;
1735		ieee->link_detect_info.busy_traffic = false;
1736		rtllib_disassociate(ieee);
1737		remove_peer_ts(ieee, header->addr2);
1738		if (!(ieee->rtllib_ap_sec_type(ieee) & (SEC_ALG_CCMP | SEC_ALG_TKIP)))
1739			schedule_delayed_work(&ieee->associate_procedure_wq, 5);
1740	}
1741	return 0;
1742}
1743
1744inline int rtllib_rx_frame_softmac(struct rtllib_device *ieee,
1745				   struct sk_buff *skb,
1746				   struct rtllib_rx_stats *rx_stats, u16 type,
1747				   u16 stype)
1748{
1749	struct ieee80211_hdr_3addr *header = (struct ieee80211_hdr_3addr *)skb->data;
1750	u16 frame_ctl;
1751
1752	if (!ieee->proto_started)
1753		return 0;
1754
1755	frame_ctl = le16_to_cpu(header->frame_control);
1756	switch (WLAN_FC_GET_STYPE(frame_ctl)) {
1757	case IEEE80211_STYPE_ASSOC_RESP:
1758	case IEEE80211_STYPE_REASSOC_RESP:
1759		if (rtllib_rx_assoc_resp(ieee, skb, rx_stats) == 1)
1760			return 1;
1761		break;
1762	case IEEE80211_STYPE_ASSOC_REQ:
1763	case IEEE80211_STYPE_REASSOC_REQ:
1764		break;
1765	case IEEE80211_STYPE_AUTH:
1766		rtllib_rx_auth(ieee, skb, rx_stats);
1767		break;
1768	case IEEE80211_STYPE_DISASSOC:
1769	case IEEE80211_STYPE_DEAUTH:
1770		rtllib_rx_deauth(ieee, skb);
1771		break;
1772	case IEEE80211_STYPE_ACTION:
1773		rtllib_process_action(ieee, skb);
1774		break;
1775	default:
1776		return -1;
1777	}
1778	return 0;
1779}
1780
1781/* following are for a simpler TX queue management.
1782 * Instead of using netif_[stop/wake]_queue the driver
1783 * will use these two functions (plus a reset one), that
1784 * will internally use the kernel netif_* and takes
1785 * care of the ieee802.11 fragmentation.
1786 * So the driver receives a fragment per time and might
1787 * call the stop function when it wants to not
1788 * have enough room to TX an entire packet.
1789 * This might be useful if each fragment needs it's own
1790 * descriptor, thus just keep a total free memory > than
1791 * the max fragmentation threshold is not enough.. If the
1792 * ieee802.11 stack passed a TXB struct then you need
1793 * to keep N free descriptors where
1794 * N = MAX_PACKET_SIZE / MIN_FRAG_TRESHOLD
1795 * In this way you need just one and the 802.11 stack
1796 * will take care of buffering fragments and pass them to
1797 * the driver later, when it wakes the queue.
1798 */
1799void rtllib_softmac_xmit(struct rtllib_txb *txb, struct rtllib_device *ieee)
1800{
1801	unsigned int queue_index = txb->queue_index;
1802	unsigned long flags;
1803	int  i;
1804	struct cb_desc *tcb_desc = NULL;
1805	unsigned long queue_len = 0;
1806
1807	spin_lock_irqsave(&ieee->lock, flags);
1808
1809	/* called with 2nd parm 0, no tx mgmt lock required */
1810	rtllib_sta_wakeup(ieee, 0);
1811
1812	/* update the tx status */
1813	tcb_desc = (struct cb_desc *)(txb->fragments[0]->cb +
1814		   MAX_DEV_ADDR_SIZE);
1815	if (tcb_desc->multicast)
1816		ieee->stats.multicast++;
1817
1818	/* if xmit available, just xmit it immediately, else just insert it to
1819	 * the wait queue
1820	 */
1821	for (i = 0; i < txb->nr_frags; i++) {
1822		queue_len = skb_queue_len(&ieee->skb_waitq[queue_index]);
1823		if ((queue_len  != 0) ||
1824		    (!ieee->check_nic_enough_desc(ieee->dev, queue_index)) ||
1825		    (ieee->queue_stop)) {
1826			/* insert the skb packet to the wait queue
1827			 * as for the completion function, it does not need
1828			 * to check it any more.
1829			 */
1830			if (queue_len < 200)
1831				skb_queue_tail(&ieee->skb_waitq[queue_index],
1832					       txb->fragments[i]);
1833			else
1834				kfree_skb(txb->fragments[i]);
1835		} else {
1836			ieee->softmac_data_hard_start_xmit(txb->fragments[i],
1837					ieee->dev, ieee->rate);
1838		}
1839	}
1840
1841	rtllib_txb_free(txb);
1842
1843	spin_unlock_irqrestore(&ieee->lock, flags);
1844}
1845
1846void rtllib_reset_queue(struct rtllib_device *ieee)
1847{
1848	unsigned long flags;
1849
1850	spin_lock_irqsave(&ieee->lock, flags);
1851	init_mgmt_queue(ieee);
1852	if (ieee->tx_pending.txb) {
1853		rtllib_txb_free(ieee->tx_pending.txb);
1854		ieee->tx_pending.txb = NULL;
1855	}
1856	ieee->queue_stop = 0;
1857	spin_unlock_irqrestore(&ieee->lock, flags);
1858}
1859EXPORT_SYMBOL(rtllib_reset_queue);
1860
1861void rtllib_stop_all_queues(struct rtllib_device *ieee)
1862{
1863	unsigned int i;
1864
1865	for (i = 0; i < ieee->dev->num_tx_queues; i++)
1866		txq_trans_cond_update(netdev_get_tx_queue(ieee->dev, i));
1867
1868	netif_tx_stop_all_queues(ieee->dev);
1869}
1870
1871void rtllib_wake_all_queues(struct rtllib_device *ieee)
1872{
1873	netif_tx_wake_all_queues(ieee->dev);
1874}
1875
1876/* this is called only in user context, with wx_mutex held */
1877static void rtllib_start_bss(struct rtllib_device *ieee)
1878{
1879	unsigned long flags;
1880
1881	/* check if we have already found the net we
1882	 * are interested in (if any).
1883	 * if not (we are disassociated and we are not
1884	 * in associating / authenticating phase) start the background scanning.
1885	 */
1886	rtllib_softmac_check_all_nets(ieee);
1887
1888	/* ensure no-one start an associating process (thus setting
1889	 * the ieee->link_state to rtllib_ASSOCIATING) while we
1890	 * have just checked it and we are going to enable scan.
1891	 * The rtllib_new_net function is always called with
1892	 * lock held (from both rtllib_softmac_check_all_nets and
1893	 * the rx path), so we cannot be in the middle of such function
1894	 */
1895	spin_lock_irqsave(&ieee->lock, flags);
1896
1897	if (ieee->link_state == MAC80211_NOLINK)
1898		rtllib_start_scan(ieee);
1899	spin_unlock_irqrestore(&ieee->lock, flags);
1900}
1901
1902static void rtllib_link_change_wq(void *data)
1903{
1904	struct rtllib_device *ieee = container_of_dwork_rsl(data,
1905				     struct rtllib_device, link_change_wq);
1906	ieee->link_change(ieee->dev);
1907}
1908
1909/* called only in userspace context */
1910void rtllib_disassociate(struct rtllib_device *ieee)
1911{
1912	netif_carrier_off(ieee->dev);
1913	if (ieee->softmac_features & IEEE_SOFTMAC_TX_QUEUE)
1914		rtllib_reset_queue(ieee);
1915
1916	ieee->link_state = MAC80211_NOLINK;
1917	ieee->is_set_key = false;
1918	ieee->wap_set = 0;
1919
1920	schedule_delayed_work(&ieee->link_change_wq, 0);
1921
1922	notify_wx_assoc_event(ieee);
1923}
1924
1925static void rtllib_associate_retry_wq(void *data)
1926{
1927	struct rtllib_device *ieee = container_of_dwork_rsl(data,
1928				     struct rtllib_device, associate_retry_wq);
1929	unsigned long flags;
1930
1931	mutex_lock(&ieee->wx_mutex);
1932	if (!ieee->proto_started)
1933		goto exit;
1934
1935	if (ieee->link_state != RTLLIB_ASSOCIATING_RETRY)
1936		goto exit;
1937
1938	/* until we do not set the state to MAC80211_NOLINK
1939	 * there are no possibility to have someone else trying
1940	 * to start an association procedure (we get here with
1941	 * ieee->link_state = RTLLIB_ASSOCIATING).
1942	 * When we set the state to MAC80211_NOLINK it is possible
1943	 * that the RX path run an attempt to associate, but
1944	 * both rtllib_softmac_check_all_nets and the
1945	 * RX path works with ieee->lock held so there are no
1946	 * problems. If we are still disassociated then start a scan.
1947	 * the lock here is necessary to ensure no one try to start
1948	 * an association procedure when we have just checked the
1949	 * state and we are going to start the scan.
1950	 */
1951	ieee->beinretry = true;
1952	ieee->link_state = MAC80211_NOLINK;
1953
1954	rtllib_softmac_check_all_nets(ieee);
1955
1956	spin_lock_irqsave(&ieee->lock, flags);
1957
1958	if (ieee->link_state == MAC80211_NOLINK)
1959		rtllib_start_scan(ieee);
1960	spin_unlock_irqrestore(&ieee->lock, flags);
1961
1962	ieee->beinretry = false;
1963exit:
1964	mutex_unlock(&ieee->wx_mutex);
1965}
1966
1967void rtllib_softmac_stop_protocol(struct rtllib_device *ieee)
1968{
1969	rtllib_stop_scan_syncro(ieee);
1970	mutex_lock(&ieee->wx_mutex);
1971	rtllib_stop_protocol(ieee);
1972	mutex_unlock(&ieee->wx_mutex);
1973}
1974EXPORT_SYMBOL(rtllib_softmac_stop_protocol);
1975
1976void rtllib_stop_protocol(struct rtllib_device *ieee)
1977{
1978	if (!ieee->proto_started)
1979		return;
1980
1981	ieee->proto_started = 0;
1982	ieee->proto_stoppping = 1;
1983	ieee->rtllib_ips_leave(ieee->dev);
1984
1985	del_timer_sync(&ieee->associate_timer);
1986	mutex_unlock(&ieee->wx_mutex);
1987	cancel_delayed_work_sync(&ieee->associate_retry_wq);
1988	mutex_lock(&ieee->wx_mutex);
1989	cancel_delayed_work_sync(&ieee->link_change_wq);
1990	rtllib_stop_scan(ieee);
1991
1992	if (ieee->link_state <= RTLLIB_ASSOCIATING_AUTHENTICATED)
1993		ieee->link_state = MAC80211_NOLINK;
1994
1995	if (ieee->link_state == MAC80211_LINKED) {
1996		if (ieee->iw_mode == IW_MODE_INFRA)
1997			send_disassociation(ieee, 1, WLAN_REASON_DEAUTH_LEAVING);
1998		rtllib_disassociate(ieee);
1999	}
2000
2001	remove_all_ts(ieee);
2002	ieee->proto_stoppping = 0;
2003
2004	kfree(ieee->assocreq_ies);
2005	ieee->assocreq_ies = NULL;
2006	ieee->assocreq_ies_len = 0;
2007	kfree(ieee->assocresp_ies);
2008	ieee->assocresp_ies = NULL;
2009	ieee->assocresp_ies_len = 0;
2010}
2011
2012void rtllib_softmac_start_protocol(struct rtllib_device *ieee)
2013{
2014	mutex_lock(&ieee->wx_mutex);
2015	rtllib_start_protocol(ieee);
2016	mutex_unlock(&ieee->wx_mutex);
2017}
2018EXPORT_SYMBOL(rtllib_softmac_start_protocol);
2019
2020void rtllib_start_protocol(struct rtllib_device *ieee)
2021{
2022	short ch = 0;
2023	int i = 0;
2024
2025	if (ieee->proto_started)
2026		return;
2027
2028	ieee->proto_started = 1;
2029
2030	if (ieee->current_network.channel == 0) {
2031		do {
2032			ch++;
2033			if (ch > MAX_CHANNEL_NUMBER)
2034				return; /* no channel found */
2035		} while (!ieee->active_channel_map[ch]);
2036		ieee->current_network.channel = ch;
2037	}
2038
2039	if (ieee->current_network.beacon_interval == 0)
2040		ieee->current_network.beacon_interval = 100;
2041
2042	for (i = 0; i < 17; i++) {
2043		ieee->last_rxseq_num[i] = -1;
2044		ieee->last_rxfrag_num[i] = -1;
2045		ieee->last_packet_time[i] = 0;
2046	}
2047
2048	ieee->wmm_acm = 0;
2049	/* if the user set the MAC of the ad-hoc cell and then
2050	 * switch to managed mode, shall we  make sure that association
2051	 * attempts does not fail just because the user provide the essid
2052	 * and the nic is still checking for the AP MAC ??
2053	 */
2054	switch (ieee->iw_mode) {
2055	case IW_MODE_INFRA:
2056		rtllib_start_bss(ieee);
2057		break;
2058	}
2059}
2060
2061int rtllib_softmac_init(struct rtllib_device *ieee)
2062{
2063	int i;
2064
2065	memset(&ieee->current_network, 0, sizeof(struct rtllib_network));
2066
2067	ieee->link_state = MAC80211_NOLINK;
2068	for (i = 0; i < 5; i++)
2069		ieee->seq_ctrl[i] = 0;
2070
2071	ieee->link_detect_info.slot_index = 0;
2072	ieee->link_detect_info.slot_num = 2;
2073	ieee->link_detect_info.num_recv_bcn_in_period = 0;
2074	ieee->link_detect_info.num_recv_data_in_period = 0;
2075	ieee->link_detect_info.num_tx_ok_in_period = 0;
2076	ieee->link_detect_info.num_rx_ok_in_period = 0;
2077	ieee->link_detect_info.num_rx_unicast_ok_in_period = 0;
2078	ieee->is_aggregate_frame = false;
2079	ieee->assoc_id = 0;
2080	ieee->queue_stop = 0;
2081	ieee->scanning_continue = 0;
2082	ieee->softmac_features = 0;
2083	ieee->wap_set = 0;
2084	ieee->ssid_set = 0;
2085	ieee->proto_started = 0;
2086	ieee->proto_stoppping = 0;
2087	ieee->basic_rate = RTLLIB_DEFAULT_BASIC_RATE;
2088	ieee->rate = 22;
2089	ieee->ps = RTLLIB_PS_DISABLED;
2090	ieee->sta_sleep = LPS_IS_WAKE;
2091
2092	ieee->reg_dot11ht_oper_rate_set[0] = 0xff;
2093	ieee->reg_dot11ht_oper_rate_set[1] = 0xff;
2094	ieee->reg_dot11ht_oper_rate_set[4] = 0x01;
2095
2096	ieee->reg_dot11tx_ht_oper_rate_set[0] = 0xff;
2097	ieee->reg_dot11tx_ht_oper_rate_set[1] = 0xff;
2098	ieee->reg_dot11tx_ht_oper_rate_set[4] = 0x01;
2099
2100	ieee->first_ie_in_scan = false;
2101	ieee->actscanning = false;
2102	ieee->beinretry = false;
2103	ieee->is_set_key = false;
2104	init_mgmt_queue(ieee);
2105
2106	ieee->tx_pending.txb = NULL;
2107
2108	timer_setup(&ieee->associate_timer, rtllib_associate_abort_cb, 0);
2109
2110	INIT_DELAYED_WORK(&ieee->link_change_wq, (void *)rtllib_link_change_wq);
2111	INIT_WORK(&ieee->associate_complete_wq, (void *)rtllib_associate_complete_wq);
2112	INIT_DELAYED_WORK(&ieee->associate_procedure_wq, (void *)rtllib_associate_procedure_wq);
2113	INIT_DELAYED_WORK(&ieee->softmac_scan_wq, (void *)rtllib_softmac_scan_wq);
2114	INIT_DELAYED_WORK(&ieee->associate_retry_wq, (void *)rtllib_associate_retry_wq);
2115	INIT_WORK(&ieee->wx_sync_scan_wq, (void *)rtllib_wx_sync_scan_wq);
2116
2117	mutex_init(&ieee->wx_mutex);
2118	mutex_init(&ieee->scan_mutex);
2119	mutex_init(&ieee->ips_mutex);
2120
2121	spin_lock_init(&ieee->mgmt_tx_lock);
2122	spin_lock_init(&ieee->beacon_lock);
2123
2124	INIT_WORK(&ieee->ps_task, rtllib_sta_ps);
2125
2126	return 0;
2127}
2128
2129void rtllib_softmac_free(struct rtllib_device *ieee)
2130{
2131	del_timer_sync(&ieee->associate_timer);
2132
2133	cancel_delayed_work_sync(&ieee->associate_retry_wq);
2134	cancel_delayed_work_sync(&ieee->associate_procedure_wq);
2135	cancel_delayed_work_sync(&ieee->softmac_scan_wq);
2136	cancel_delayed_work_sync(&ieee->hw_wakeup_wq);
2137	cancel_delayed_work_sync(&ieee->hw_sleep_wq);
2138	cancel_delayed_work_sync(&ieee->link_change_wq);
2139	cancel_work_sync(&ieee->associate_complete_wq);
2140	cancel_work_sync(&ieee->ips_leave_wq);
2141	cancel_work_sync(&ieee->wx_sync_scan_wq);
2142	cancel_work_sync(&ieee->ps_task);
2143}
2144
2145static inline struct sk_buff *
2146rtllib_disauth_skb(struct rtllib_network *beacon,
2147		   struct rtllib_device *ieee, u16 rsn)
2148{
2149	struct sk_buff *skb;
2150	struct rtllib_disauth *disauth;
2151	int len = sizeof(struct rtllib_disauth) + ieee->tx_headroom;
2152
2153	skb = dev_alloc_skb(len);
2154	if (!skb)
2155		return NULL;
2156
2157	skb_reserve(skb, ieee->tx_headroom);
2158
2159	disauth = skb_put(skb, sizeof(struct rtllib_disauth));
2160	disauth->header.frame_control = cpu_to_le16(IEEE80211_STYPE_DEAUTH);
2161	disauth->header.duration_id = 0;
2162
2163	ether_addr_copy(disauth->header.addr1, beacon->bssid);
2164	ether_addr_copy(disauth->header.addr2, ieee->dev->dev_addr);
2165	ether_addr_copy(disauth->header.addr3, beacon->bssid);
2166
2167	disauth->reason = cpu_to_le16(rsn);
2168	return skb;
2169}
2170
2171static inline struct sk_buff *
2172rtllib_disassociate_skb(struct rtllib_network *beacon,
2173			struct rtllib_device *ieee, u16 rsn)
2174{
2175	struct sk_buff *skb;
2176	struct rtllib_disassoc *disass;
2177	int len = sizeof(struct rtllib_disassoc) + ieee->tx_headroom;
2178
2179	skb = dev_alloc_skb(len);
2180
2181	if (!skb)
2182		return NULL;
2183
2184	skb_reserve(skb, ieee->tx_headroom);
2185
2186	disass = skb_put(skb, sizeof(struct rtllib_disassoc));
2187	disass->header.frame_control = cpu_to_le16(IEEE80211_STYPE_DISASSOC);
2188	disass->header.duration_id = 0;
2189
2190	ether_addr_copy(disass->header.addr1, beacon->bssid);
2191	ether_addr_copy(disass->header.addr2, ieee->dev->dev_addr);
2192	ether_addr_copy(disass->header.addr3, beacon->bssid);
2193
2194	disass->reason = cpu_to_le16(rsn);
2195	return skb;
2196}
2197
2198void send_disassociation(struct rtllib_device *ieee, bool deauth, u16 rsn)
2199{
2200	struct rtllib_network *beacon = &ieee->current_network;
2201	struct sk_buff *skb;
2202
2203	if (deauth)
2204		skb = rtllib_disauth_skb(beacon, ieee, rsn);
2205	else
2206		skb = rtllib_disassociate_skb(beacon, ieee, rsn);
2207
2208	if (skb)
2209		softmac_mgmt_xmit(skb, ieee);
2210}
2211
2212u8 rtllib_ap_sec_type(struct rtllib_device *ieee)
2213{
2214	static u8 ccmp_ie[4] = {0x00, 0x50, 0xf2, 0x04};
2215	static u8 ccmp_rsn_ie[4] = {0x00, 0x0f, 0xac, 0x04};
2216	int wpa_ie_len = ieee->wpa_ie_len;
2217	struct lib80211_crypt_data *crypt;
2218	int encrypt;
2219
2220	crypt = ieee->crypt_info.crypt[ieee->crypt_info.tx_keyidx];
2221	encrypt = (ieee->current_network.capability & WLAN_CAPABILITY_PRIVACY)
2222		  || (crypt && crypt->ops && (strcmp(crypt->ops->name, "R-WEP") == 0));
2223
2224	/* simply judge  */
2225	if (encrypt && (wpa_ie_len == 0)) {
2226		return SEC_ALG_WEP;
2227	} else if ((wpa_ie_len != 0)) {
2228		if (((ieee->wpa_ie[0] == 0xdd) &&
2229		    (!memcmp(&ieee->wpa_ie[14], ccmp_ie, 4))) ||
2230		    ((ieee->wpa_ie[0] == 0x30) &&
2231		    (!memcmp(&ieee->wpa_ie[10], ccmp_rsn_ie, 4))))
2232			return SEC_ALG_CCMP;
2233		else
2234			return SEC_ALG_TKIP;
2235	} else {
2236		return SEC_ALG_NONE;
2237	}
2238}
2239
2240static void rtllib_mlme_disassociate_request(struct rtllib_device *rtllib,
2241					     u8 *addr, u8 rsn)
2242{
2243	u8 i;
2244	u8	op_mode;
2245
2246	remove_peer_ts(rtllib, addr);
2247
2248	if (memcmp(rtllib->current_network.bssid, addr, 6) == 0) {
2249		rtllib->link_state = MAC80211_NOLINK;
2250
2251		for (i = 0; i < 6; i++)
2252			rtllib->current_network.bssid[i] = 0x22;
2253		op_mode = RT_OP_MODE_NO_LINK;
2254		rtllib->op_mode = RT_OP_MODE_NO_LINK;
2255		rtllib->set_hw_reg_handler(rtllib->dev, HW_VAR_MEDIA_STATUS,
2256					(u8 *)(&op_mode));
2257		rtllib_disassociate(rtllib);
2258
2259		rtllib->set_hw_reg_handler(rtllib->dev, HW_VAR_BSSID,
2260					rtllib->current_network.bssid);
2261	}
2262}
2263
2264static void rtllib_mgnt_disconnect_ap(struct rtllib_device *rtllib, u8 rsn)
2265{
2266	bool filter_out_nonassociated_bssid = false;
2267
2268	filter_out_nonassociated_bssid = false;
2269	rtllib->set_hw_reg_handler(rtllib->dev, HW_VAR_CECHK_BSSID,
2270				(u8 *)(&filter_out_nonassociated_bssid));
2271	rtllib_mlme_disassociate_request(rtllib, rtllib->current_network.bssid,
2272					 rsn);
2273
2274	rtllib->link_state = MAC80211_NOLINK;
2275}
2276
2277bool rtllib_mgnt_disconnect(struct rtllib_device *rtllib, u8 rsn)
2278{
2279	if (rtllib->ps != RTLLIB_PS_DISABLED)
2280		rtllib->sta_wake_up(rtllib->dev);
2281
2282	if (rtllib->link_state == MAC80211_LINKED) {
2283		if (rtllib->iw_mode == IW_MODE_INFRA)
2284			rtllib_mgnt_disconnect_ap(rtllib, rsn);
2285	}
2286
2287	return true;
2288}
2289EXPORT_SYMBOL(rtllib_mgnt_disconnect);
2290
2291void notify_wx_assoc_event(struct rtllib_device *ieee)
2292{
2293	union iwreq_data wrqu;
2294
2295	if (ieee->cannot_notify)
2296		return;
2297
2298	wrqu.ap_addr.sa_family = ARPHRD_ETHER;
2299	if (ieee->link_state == MAC80211_LINKED) {
2300		memcpy(wrqu.ap_addr.sa_data, ieee->current_network.bssid,
2301		       ETH_ALEN);
2302	} else {
2303		netdev_info(ieee->dev, "%s(): Tell user space disconnected\n",
2304			    __func__);
2305		eth_zero_addr(wrqu.ap_addr.sa_data);
2306	}
2307	wireless_send_event(ieee->dev, SIOCGIWAP, &wrqu, NULL);
2308}
2309EXPORT_SYMBOL(notify_wx_assoc_event);
2310