1 /*
2  * Intersil Prism2 driver with Host AP (software access point) support
3  * Copyright (c) 2001-2002, SSH Communications Security Corp and Jouni Malinen
4  * <j@w1.fi>
5  * Copyright (c) 2002-2005, Jouni Malinen <j@w1.fi>
6  *
7  * This file is to be included into hostap.c when S/W AP functionality is
8  * compiled.
9  *
10  * AP:  FIX:
11  * - if unicast Class 2 (assoc,reassoc,disassoc) frame received from
12  *   unauthenticated STA, send deauth. frame (8802.11: 5.5)
13  * - if unicast Class 3 (data with to/from DS,deauth,pspoll) frame received
14  *   from authenticated, but unassoc STA, send disassoc frame (8802.11: 5.5)
15  * - if unicast Class 3 received from unauthenticated STA, send deauth. frame
16  *   (8802.11: 5.5)
17  */
18 
19 #include <linux/proc_fs.h>
20 #include <linux/delay.h>
21 #include <linux/random.h>
22 #include <linux/if_arp.h>
23 #include <linux/slab.h>
24 
25 #include "hostap_wlan.h"
26 #include "hostap.h"
27 #include "hostap_ap.h"
28 
29 static int other_ap_policy[MAX_PARM_DEVICES] = { AP_OTHER_AP_SKIP_ALL,
30 						 DEF_INTS };
31 module_param_array(other_ap_policy, int, NULL, 0444);
32 MODULE_PARM_DESC(other_ap_policy, "Other AP beacon monitoring policy (0-3)");
33 
34 static int ap_max_inactivity[MAX_PARM_DEVICES] = { AP_MAX_INACTIVITY_SEC,
35 						   DEF_INTS };
36 module_param_array(ap_max_inactivity, int, NULL, 0444);
37 MODULE_PARM_DESC(ap_max_inactivity, "AP timeout (in seconds) for station "
38 		 "inactivity");
39 
40 static int ap_bridge_packets[MAX_PARM_DEVICES] = { 1, DEF_INTS };
41 module_param_array(ap_bridge_packets, int, NULL, 0444);
42 MODULE_PARM_DESC(ap_bridge_packets, "Bridge packets directly between "
43 		 "stations");
44 
45 static int autom_ap_wds[MAX_PARM_DEVICES] = { 0, DEF_INTS };
46 module_param_array(autom_ap_wds, int, NULL, 0444);
47 MODULE_PARM_DESC(autom_ap_wds, "Add WDS connections to other APs "
48 		 "automatically");
49 
50 
51 static struct sta_info* ap_get_sta(struct ap_data *ap, u8 *sta);
52 static void hostap_event_expired_sta(struct net_device *dev,
53 				     struct sta_info *sta);
54 static void handle_add_proc_queue(struct work_struct *work);
55 
56 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
57 static void handle_wds_oper_queue(struct work_struct *work);
58 static void prism2_send_mgmt(struct net_device *dev,
59 			     u16 type_subtype, char *body,
60 			     int body_len, u8 *addr, u16 tx_cb_idx);
61 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
62 
63 
64 #ifndef PRISM2_NO_PROCFS_DEBUG
ap_debug_proc_read(char * page,char ** start,off_t off,int count,int * eof,void * data)65 static int ap_debug_proc_read(char *page, char **start, off_t off,
66 			      int count, int *eof, void *data)
67 {
68 	char *p = page;
69 	struct ap_data *ap = (struct ap_data *) data;
70 
71 	if (off != 0) {
72 		*eof = 1;
73 		return 0;
74 	}
75 
76 	p += sprintf(p, "BridgedUnicastFrames=%u\n", ap->bridged_unicast);
77 	p += sprintf(p, "BridgedMulticastFrames=%u\n", ap->bridged_multicast);
78 	p += sprintf(p, "max_inactivity=%u\n", ap->max_inactivity / HZ);
79 	p += sprintf(p, "bridge_packets=%u\n", ap->bridge_packets);
80 	p += sprintf(p, "nullfunc_ack=%u\n", ap->nullfunc_ack);
81 	p += sprintf(p, "autom_ap_wds=%u\n", ap->autom_ap_wds);
82 	p += sprintf(p, "auth_algs=%u\n", ap->local->auth_algs);
83 	p += sprintf(p, "tx_drop_nonassoc=%u\n", ap->tx_drop_nonassoc);
84 
85 	return (p - page);
86 }
87 #endif /* PRISM2_NO_PROCFS_DEBUG */
88 
89 
ap_sta_hash_add(struct ap_data * ap,struct sta_info * sta)90 static void ap_sta_hash_add(struct ap_data *ap, struct sta_info *sta)
91 {
92 	sta->hnext = ap->sta_hash[STA_HASH(sta->addr)];
93 	ap->sta_hash[STA_HASH(sta->addr)] = sta;
94 }
95 
ap_sta_hash_del(struct ap_data * ap,struct sta_info * sta)96 static void ap_sta_hash_del(struct ap_data *ap, struct sta_info *sta)
97 {
98 	struct sta_info *s;
99 
100 	s = ap->sta_hash[STA_HASH(sta->addr)];
101 	if (s == NULL) return;
102 	if (memcmp(s->addr, sta->addr, ETH_ALEN) == 0) {
103 		ap->sta_hash[STA_HASH(sta->addr)] = s->hnext;
104 		return;
105 	}
106 
107 	while (s->hnext != NULL && memcmp(s->hnext->addr, sta->addr, ETH_ALEN)
108 	       != 0)
109 		s = s->hnext;
110 	if (s->hnext != NULL)
111 		s->hnext = s->hnext->hnext;
112 	else
113 		printk("AP: could not remove STA %pM from hash table\n",
114 		       sta->addr);
115 }
116 
ap_free_sta(struct ap_data * ap,struct sta_info * sta)117 static void ap_free_sta(struct ap_data *ap, struct sta_info *sta)
118 {
119 	if (sta->ap && sta->local)
120 		hostap_event_expired_sta(sta->local->dev, sta);
121 
122 	if (ap->proc != NULL) {
123 		char name[20];
124 		sprintf(name, "%pM", sta->addr);
125 		remove_proc_entry(name, ap->proc);
126 	}
127 
128 	if (sta->crypt) {
129 		sta->crypt->ops->deinit(sta->crypt->priv);
130 		kfree(sta->crypt);
131 		sta->crypt = NULL;
132 	}
133 
134 	skb_queue_purge(&sta->tx_buf);
135 
136 	ap->num_sta--;
137 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
138 	if (sta->aid > 0)
139 		ap->sta_aid[sta->aid - 1] = NULL;
140 
141 	if (!sta->ap && sta->u.sta.challenge)
142 		kfree(sta->u.sta.challenge);
143 	del_timer(&sta->timer);
144 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
145 
146 	kfree(sta);
147 }
148 
149 
hostap_set_tim(local_info_t * local,int aid,int set)150 static void hostap_set_tim(local_info_t *local, int aid, int set)
151 {
152 	if (local->func->set_tim)
153 		local->func->set_tim(local->dev, aid, set);
154 }
155 
156 
hostap_event_new_sta(struct net_device * dev,struct sta_info * sta)157 static void hostap_event_new_sta(struct net_device *dev, struct sta_info *sta)
158 {
159 	union iwreq_data wrqu;
160 	memset(&wrqu, 0, sizeof(wrqu));
161 	memcpy(wrqu.addr.sa_data, sta->addr, ETH_ALEN);
162 	wrqu.addr.sa_family = ARPHRD_ETHER;
163 	wireless_send_event(dev, IWEVREGISTERED, &wrqu, NULL);
164 }
165 
166 
hostap_event_expired_sta(struct net_device * dev,struct sta_info * sta)167 static void hostap_event_expired_sta(struct net_device *dev,
168 				     struct sta_info *sta)
169 {
170 	union iwreq_data wrqu;
171 	memset(&wrqu, 0, sizeof(wrqu));
172 	memcpy(wrqu.addr.sa_data, sta->addr, ETH_ALEN);
173 	wrqu.addr.sa_family = ARPHRD_ETHER;
174 	wireless_send_event(dev, IWEVEXPIRED, &wrqu, NULL);
175 }
176 
177 
178 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
179 
ap_handle_timer(unsigned long data)180 static void ap_handle_timer(unsigned long data)
181 {
182 	struct sta_info *sta = (struct sta_info *) data;
183 	local_info_t *local;
184 	struct ap_data *ap;
185 	unsigned long next_time = 0;
186 	int was_assoc;
187 
188 	if (sta == NULL || sta->local == NULL || sta->local->ap == NULL) {
189 		PDEBUG(DEBUG_AP, "ap_handle_timer() called with NULL data\n");
190 		return;
191 	}
192 
193 	local = sta->local;
194 	ap = local->ap;
195 	was_assoc = sta->flags & WLAN_STA_ASSOC;
196 
197 	if (atomic_read(&sta->users) != 0)
198 		next_time = jiffies + HZ;
199 	else if ((sta->flags & WLAN_STA_PERM) && !(sta->flags & WLAN_STA_AUTH))
200 		next_time = jiffies + ap->max_inactivity;
201 
202 	if (time_before(jiffies, sta->last_rx + ap->max_inactivity)) {
203 		/* station activity detected; reset timeout state */
204 		sta->timeout_next = STA_NULLFUNC;
205 		next_time = sta->last_rx + ap->max_inactivity;
206 	} else if (sta->timeout_next == STA_DISASSOC &&
207 		   !(sta->flags & WLAN_STA_PENDING_POLL)) {
208 		/* STA ACKed data nullfunc frame poll */
209 		sta->timeout_next = STA_NULLFUNC;
210 		next_time = jiffies + ap->max_inactivity;
211 	}
212 
213 	if (next_time) {
214 		sta->timer.expires = next_time;
215 		add_timer(&sta->timer);
216 		return;
217 	}
218 
219 	if (sta->ap)
220 		sta->timeout_next = STA_DEAUTH;
221 
222 	if (sta->timeout_next == STA_DEAUTH && !(sta->flags & WLAN_STA_PERM)) {
223 		spin_lock(&ap->sta_table_lock);
224 		ap_sta_hash_del(ap, sta);
225 		list_del(&sta->list);
226 		spin_unlock(&ap->sta_table_lock);
227 		sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
228 	} else if (sta->timeout_next == STA_DISASSOC)
229 		sta->flags &= ~WLAN_STA_ASSOC;
230 
231 	if (was_assoc && !(sta->flags & WLAN_STA_ASSOC) && !sta->ap)
232 		hostap_event_expired_sta(local->dev, sta);
233 
234 	if (sta->timeout_next == STA_DEAUTH && sta->aid > 0 &&
235 	    !skb_queue_empty(&sta->tx_buf)) {
236 		hostap_set_tim(local, sta->aid, 0);
237 		sta->flags &= ~WLAN_STA_TIM;
238 	}
239 
240 	if (sta->ap) {
241 		if (ap->autom_ap_wds) {
242 			PDEBUG(DEBUG_AP, "%s: removing automatic WDS "
243 			       "connection to AP %pM\n",
244 			       local->dev->name, sta->addr);
245 			hostap_wds_link_oper(local, sta->addr, WDS_DEL);
246 		}
247 	} else if (sta->timeout_next == STA_NULLFUNC) {
248 		/* send data frame to poll STA and check whether this frame
249 		 * is ACKed */
250 		/* FIX: IEEE80211_STYPE_NULLFUNC would be more appropriate, but
251 		 * it is apparently not retried so TX Exc events are not
252 		 * received for it */
253 		sta->flags |= WLAN_STA_PENDING_POLL;
254 		prism2_send_mgmt(local->dev, IEEE80211_FTYPE_DATA |
255 				 IEEE80211_STYPE_DATA, NULL, 0,
256 				 sta->addr, ap->tx_callback_poll);
257 	} else {
258 		int deauth = sta->timeout_next == STA_DEAUTH;
259 		__le16 resp;
260 		PDEBUG(DEBUG_AP, "%s: sending %s info to STA %pM"
261 		       "(last=%lu, jiffies=%lu)\n",
262 		       local->dev->name,
263 		       deauth ? "deauthentication" : "disassociation",
264 		       sta->addr, sta->last_rx, jiffies);
265 
266 		resp = cpu_to_le16(deauth ? WLAN_REASON_PREV_AUTH_NOT_VALID :
267 				   WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY);
268 		prism2_send_mgmt(local->dev, IEEE80211_FTYPE_MGMT |
269 				 (deauth ? IEEE80211_STYPE_DEAUTH :
270 				  IEEE80211_STYPE_DISASSOC),
271 				 (char *) &resp, 2, sta->addr, 0);
272 	}
273 
274 	if (sta->timeout_next == STA_DEAUTH) {
275 		if (sta->flags & WLAN_STA_PERM) {
276 			PDEBUG(DEBUG_AP, "%s: STA %pM"
277 			       " would have been removed, "
278 			       "but it has 'perm' flag\n",
279 			       local->dev->name, sta->addr);
280 		} else
281 			ap_free_sta(ap, sta);
282 		return;
283 	}
284 
285 	if (sta->timeout_next == STA_NULLFUNC) {
286 		sta->timeout_next = STA_DISASSOC;
287 		sta->timer.expires = jiffies + AP_DISASSOC_DELAY;
288 	} else {
289 		sta->timeout_next = STA_DEAUTH;
290 		sta->timer.expires = jiffies + AP_DEAUTH_DELAY;
291 	}
292 
293 	add_timer(&sta->timer);
294 }
295 
296 
hostap_deauth_all_stas(struct net_device * dev,struct ap_data * ap,int resend)297 void hostap_deauth_all_stas(struct net_device *dev, struct ap_data *ap,
298 			    int resend)
299 {
300 	u8 addr[ETH_ALEN];
301 	__le16 resp;
302 	int i;
303 
304 	PDEBUG(DEBUG_AP, "%s: Deauthenticate all stations\n", dev->name);
305 	memset(addr, 0xff, ETH_ALEN);
306 
307 	resp = cpu_to_le16(WLAN_REASON_PREV_AUTH_NOT_VALID);
308 
309 	/* deauth message sent; try to resend it few times; the message is
310 	 * broadcast, so it may be delayed until next DTIM; there is not much
311 	 * else we can do at this point since the driver is going to be shut
312 	 * down */
313 	for (i = 0; i < 5; i++) {
314 		prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
315 				 IEEE80211_STYPE_DEAUTH,
316 				 (char *) &resp, 2, addr, 0);
317 
318 		if (!resend || ap->num_sta <= 0)
319 			return;
320 
321 		mdelay(50);
322 	}
323 }
324 
325 
ap_control_proc_read(char * page,char ** start,off_t off,int count,int * eof,void * data)326 static int ap_control_proc_read(char *page, char **start, off_t off,
327 				int count, int *eof, void *data)
328 {
329 	char *p = page;
330 	struct ap_data *ap = (struct ap_data *) data;
331 	char *policy_txt;
332 	struct mac_entry *entry;
333 
334 	if (off != 0) {
335 		*eof = 1;
336 		return 0;
337 	}
338 
339 	switch (ap->mac_restrictions.policy) {
340 	case MAC_POLICY_OPEN:
341 		policy_txt = "open";
342 		break;
343 	case MAC_POLICY_ALLOW:
344 		policy_txt = "allow";
345 		break;
346 	case MAC_POLICY_DENY:
347 		policy_txt = "deny";
348 		break;
349 	default:
350 		policy_txt = "unknown";
351 		break;
352 	}
353 	p += sprintf(p, "MAC policy: %s\n", policy_txt);
354 	p += sprintf(p, "MAC entries: %u\n", ap->mac_restrictions.entries);
355 	p += sprintf(p, "MAC list:\n");
356 	spin_lock_bh(&ap->mac_restrictions.lock);
357 	list_for_each_entry(entry, &ap->mac_restrictions.mac_list, list) {
358 		if (p - page > PAGE_SIZE - 80) {
359 			p += sprintf(p, "All entries did not fit one page.\n");
360 			break;
361 		}
362 
363 		p += sprintf(p, "%pM\n", entry->addr);
364 	}
365 	spin_unlock_bh(&ap->mac_restrictions.lock);
366 
367 	return (p - page);
368 }
369 
370 
ap_control_add_mac(struct mac_restrictions * mac_restrictions,u8 * mac)371 int ap_control_add_mac(struct mac_restrictions *mac_restrictions, u8 *mac)
372 {
373 	struct mac_entry *entry;
374 
375 	entry = kmalloc(sizeof(struct mac_entry), GFP_KERNEL);
376 	if (entry == NULL)
377 		return -1;
378 
379 	memcpy(entry->addr, mac, ETH_ALEN);
380 
381 	spin_lock_bh(&mac_restrictions->lock);
382 	list_add_tail(&entry->list, &mac_restrictions->mac_list);
383 	mac_restrictions->entries++;
384 	spin_unlock_bh(&mac_restrictions->lock);
385 
386 	return 0;
387 }
388 
389 
ap_control_del_mac(struct mac_restrictions * mac_restrictions,u8 * mac)390 int ap_control_del_mac(struct mac_restrictions *mac_restrictions, u8 *mac)
391 {
392 	struct list_head *ptr;
393 	struct mac_entry *entry;
394 
395 	spin_lock_bh(&mac_restrictions->lock);
396 	for (ptr = mac_restrictions->mac_list.next;
397 	     ptr != &mac_restrictions->mac_list; ptr = ptr->next) {
398 		entry = list_entry(ptr, struct mac_entry, list);
399 
400 		if (memcmp(entry->addr, mac, ETH_ALEN) == 0) {
401 			list_del(ptr);
402 			kfree(entry);
403 			mac_restrictions->entries--;
404 			spin_unlock_bh(&mac_restrictions->lock);
405 			return 0;
406 		}
407 	}
408 	spin_unlock_bh(&mac_restrictions->lock);
409 	return -1;
410 }
411 
412 
ap_control_mac_deny(struct mac_restrictions * mac_restrictions,u8 * mac)413 static int ap_control_mac_deny(struct mac_restrictions *mac_restrictions,
414 			       u8 *mac)
415 {
416 	struct mac_entry *entry;
417 	int found = 0;
418 
419 	if (mac_restrictions->policy == MAC_POLICY_OPEN)
420 		return 0;
421 
422 	spin_lock_bh(&mac_restrictions->lock);
423 	list_for_each_entry(entry, &mac_restrictions->mac_list, list) {
424 		if (memcmp(entry->addr, mac, ETH_ALEN) == 0) {
425 			found = 1;
426 			break;
427 		}
428 	}
429 	spin_unlock_bh(&mac_restrictions->lock);
430 
431 	if (mac_restrictions->policy == MAC_POLICY_ALLOW)
432 		return !found;
433 	else
434 		return found;
435 }
436 
437 
ap_control_flush_macs(struct mac_restrictions * mac_restrictions)438 void ap_control_flush_macs(struct mac_restrictions *mac_restrictions)
439 {
440 	struct list_head *ptr, *n;
441 	struct mac_entry *entry;
442 
443 	if (mac_restrictions->entries == 0)
444 		return;
445 
446 	spin_lock_bh(&mac_restrictions->lock);
447 	for (ptr = mac_restrictions->mac_list.next, n = ptr->next;
448 	     ptr != &mac_restrictions->mac_list;
449 	     ptr = n, n = ptr->next) {
450 		entry = list_entry(ptr, struct mac_entry, list);
451 		list_del(ptr);
452 		kfree(entry);
453 	}
454 	mac_restrictions->entries = 0;
455 	spin_unlock_bh(&mac_restrictions->lock);
456 }
457 
458 
ap_control_kick_mac(struct ap_data * ap,struct net_device * dev,u8 * mac)459 int ap_control_kick_mac(struct ap_data *ap, struct net_device *dev, u8 *mac)
460 {
461 	struct sta_info *sta;
462 	__le16 resp;
463 
464 	spin_lock_bh(&ap->sta_table_lock);
465 	sta = ap_get_sta(ap, mac);
466 	if (sta) {
467 		ap_sta_hash_del(ap, sta);
468 		list_del(&sta->list);
469 	}
470 	spin_unlock_bh(&ap->sta_table_lock);
471 
472 	if (!sta)
473 		return -EINVAL;
474 
475 	resp = cpu_to_le16(WLAN_REASON_PREV_AUTH_NOT_VALID);
476 	prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_DEAUTH,
477 			 (char *) &resp, 2, sta->addr, 0);
478 
479 	if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
480 		hostap_event_expired_sta(dev, sta);
481 
482 	ap_free_sta(ap, sta);
483 
484 	return 0;
485 }
486 
487 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
488 
489 
ap_control_kickall(struct ap_data * ap)490 void ap_control_kickall(struct ap_data *ap)
491 {
492 	struct list_head *ptr, *n;
493 	struct sta_info *sta;
494 
495 	spin_lock_bh(&ap->sta_table_lock);
496 	for (ptr = ap->sta_list.next, n = ptr->next; ptr != &ap->sta_list;
497 	     ptr = n, n = ptr->next) {
498 		sta = list_entry(ptr, struct sta_info, list);
499 		ap_sta_hash_del(ap, sta);
500 		list_del(&sta->list);
501 		if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
502 			hostap_event_expired_sta(sta->local->dev, sta);
503 		ap_free_sta(ap, sta);
504 	}
505 	spin_unlock_bh(&ap->sta_table_lock);
506 }
507 
508 
509 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
510 
511 #define PROC_LIMIT (PAGE_SIZE - 80)
512 
prism2_ap_proc_read(char * page,char ** start,off_t off,int count,int * eof,void * data)513 static int prism2_ap_proc_read(char *page, char **start, off_t off,
514 			       int count, int *eof, void *data)
515 {
516 	char *p = page;
517 	struct ap_data *ap = (struct ap_data *) data;
518 	struct sta_info *sta;
519 	int i;
520 
521 	if (off > PROC_LIMIT) {
522 		*eof = 1;
523 		return 0;
524 	}
525 
526 	p += sprintf(p, "# BSSID CHAN SIGNAL NOISE RATE SSID FLAGS\n");
527 	spin_lock_bh(&ap->sta_table_lock);
528 	list_for_each_entry(sta, &ap->sta_list, list) {
529 		if (!sta->ap)
530 			continue;
531 
532 		p += sprintf(p, "%pM %d %d %d %d '",
533 			     sta->addr,
534 			     sta->u.ap.channel, sta->last_rx_signal,
535 			     sta->last_rx_silence, sta->last_rx_rate);
536 		for (i = 0; i < sta->u.ap.ssid_len; i++)
537 			p += sprintf(p, ((sta->u.ap.ssid[i] >= 32 &&
538 					  sta->u.ap.ssid[i] < 127) ?
539 					 "%c" : "<%02x>"),
540 				     sta->u.ap.ssid[i]);
541 		p += sprintf(p, "'");
542 		if (sta->capability & WLAN_CAPABILITY_ESS)
543 			p += sprintf(p, " [ESS]");
544 		if (sta->capability & WLAN_CAPABILITY_IBSS)
545 			p += sprintf(p, " [IBSS]");
546 		if (sta->capability & WLAN_CAPABILITY_PRIVACY)
547 			p += sprintf(p, " [WEP]");
548 		p += sprintf(p, "\n");
549 
550 		if ((p - page) > PROC_LIMIT) {
551 			printk(KERN_DEBUG "hostap: ap proc did not fit\n");
552 			break;
553 		}
554 	}
555 	spin_unlock_bh(&ap->sta_table_lock);
556 
557 	if ((p - page) <= off) {
558 		*eof = 1;
559 		return 0;
560 	}
561 
562 	*start = page + off;
563 
564 	return (p - page - off);
565 }
566 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
567 
568 
hostap_check_sta_fw_version(struct ap_data * ap,int sta_fw_ver)569 void hostap_check_sta_fw_version(struct ap_data *ap, int sta_fw_ver)
570 {
571 	if (!ap)
572 		return;
573 
574 	if (sta_fw_ver == PRISM2_FW_VER(0,8,0)) {
575 		PDEBUG(DEBUG_AP, "Using data::nullfunc ACK workaround - "
576 		       "firmware upgrade recommended\n");
577 		ap->nullfunc_ack = 1;
578 	} else
579 		ap->nullfunc_ack = 0;
580 
581 	if (sta_fw_ver == PRISM2_FW_VER(1,4,2)) {
582 		printk(KERN_WARNING "%s: Warning: secondary station firmware "
583 		       "version 1.4.2 does not seem to work in Host AP mode\n",
584 		       ap->local->dev->name);
585 	}
586 }
587 
588 
589 /* Called only as a tasklet (software IRQ) */
hostap_ap_tx_cb(struct sk_buff * skb,int ok,void * data)590 static void hostap_ap_tx_cb(struct sk_buff *skb, int ok, void *data)
591 {
592 	struct ap_data *ap = data;
593 	struct ieee80211_hdr *hdr;
594 
595 	if (!ap->local->hostapd || !ap->local->apdev) {
596 		dev_kfree_skb(skb);
597 		return;
598 	}
599 
600 	/* Pass the TX callback frame to the hostapd; use 802.11 header version
601 	 * 1 to indicate failure (no ACK) and 2 success (frame ACKed) */
602 
603 	hdr = (struct ieee80211_hdr *) skb->data;
604 	hdr->frame_control &= cpu_to_le16(~IEEE80211_FCTL_VERS);
605 	hdr->frame_control |= cpu_to_le16(ok ? BIT(1) : BIT(0));
606 
607 	skb->dev = ap->local->apdev;
608 	skb_pull(skb, hostap_80211_get_hdrlen(hdr->frame_control));
609 	skb->pkt_type = PACKET_OTHERHOST;
610 	skb->protocol = cpu_to_be16(ETH_P_802_2);
611 	memset(skb->cb, 0, sizeof(skb->cb));
612 	netif_rx(skb);
613 }
614 
615 
616 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
617 /* Called only as a tasklet (software IRQ) */
hostap_ap_tx_cb_auth(struct sk_buff * skb,int ok,void * data)618 static void hostap_ap_tx_cb_auth(struct sk_buff *skb, int ok, void *data)
619 {
620 	struct ap_data *ap = data;
621 	struct net_device *dev = ap->local->dev;
622 	struct ieee80211_hdr *hdr;
623 	u16 auth_alg, auth_transaction, status;
624 	__le16 *pos;
625 	struct sta_info *sta = NULL;
626 	char *txt = NULL;
627 
628 	if (ap->local->hostapd) {
629 		dev_kfree_skb(skb);
630 		return;
631 	}
632 
633 	hdr = (struct ieee80211_hdr *) skb->data;
634 	if (!ieee80211_is_auth(hdr->frame_control) ||
635 	    skb->len < IEEE80211_MGMT_HDR_LEN + 6) {
636 		printk(KERN_DEBUG "%s: hostap_ap_tx_cb_auth received invalid "
637 		       "frame\n", dev->name);
638 		dev_kfree_skb(skb);
639 		return;
640 	}
641 
642 	pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
643 	auth_alg = le16_to_cpu(*pos++);
644 	auth_transaction = le16_to_cpu(*pos++);
645 	status = le16_to_cpu(*pos++);
646 
647 	if (!ok) {
648 		txt = "frame was not ACKed";
649 		goto done;
650 	}
651 
652 	spin_lock(&ap->sta_table_lock);
653 	sta = ap_get_sta(ap, hdr->addr1);
654 	if (sta)
655 		atomic_inc(&sta->users);
656 	spin_unlock(&ap->sta_table_lock);
657 
658 	if (!sta) {
659 		txt = "STA not found";
660 		goto done;
661 	}
662 
663 	if (status == WLAN_STATUS_SUCCESS &&
664 	    ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
665 	     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
666 		txt = "STA authenticated";
667 		sta->flags |= WLAN_STA_AUTH;
668 		sta->last_auth = jiffies;
669 	} else if (status != WLAN_STATUS_SUCCESS)
670 		txt = "authentication failed";
671 
672  done:
673 	if (sta)
674 		atomic_dec(&sta->users);
675 	if (txt) {
676 		PDEBUG(DEBUG_AP, "%s: %pM auth_cb - alg=%d "
677 		       "trans#=%d status=%d - %s\n",
678 		       dev->name, hdr->addr1,
679 		       auth_alg, auth_transaction, status, txt);
680 	}
681 	dev_kfree_skb(skb);
682 }
683 
684 
685 /* Called only as a tasklet (software IRQ) */
hostap_ap_tx_cb_assoc(struct sk_buff * skb,int ok,void * data)686 static void hostap_ap_tx_cb_assoc(struct sk_buff *skb, int ok, void *data)
687 {
688 	struct ap_data *ap = data;
689 	struct net_device *dev = ap->local->dev;
690 	struct ieee80211_hdr *hdr;
691 	u16 status;
692 	__le16 *pos;
693 	struct sta_info *sta = NULL;
694 	char *txt = NULL;
695 
696 	if (ap->local->hostapd) {
697 		dev_kfree_skb(skb);
698 		return;
699 	}
700 
701 	hdr = (struct ieee80211_hdr *) skb->data;
702 	if ((!ieee80211_is_assoc_resp(hdr->frame_control) &&
703 	     !ieee80211_is_reassoc_resp(hdr->frame_control)) ||
704 	    skb->len < IEEE80211_MGMT_HDR_LEN + 4) {
705 		printk(KERN_DEBUG "%s: hostap_ap_tx_cb_assoc received invalid "
706 		       "frame\n", dev->name);
707 		dev_kfree_skb(skb);
708 		return;
709 	}
710 
711 	if (!ok) {
712 		txt = "frame was not ACKed";
713 		goto done;
714 	}
715 
716 	spin_lock(&ap->sta_table_lock);
717 	sta = ap_get_sta(ap, hdr->addr1);
718 	if (sta)
719 		atomic_inc(&sta->users);
720 	spin_unlock(&ap->sta_table_lock);
721 
722 	if (!sta) {
723 		txt = "STA not found";
724 		goto done;
725 	}
726 
727 	pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
728 	pos++;
729 	status = le16_to_cpu(*pos++);
730 	if (status == WLAN_STATUS_SUCCESS) {
731 		if (!(sta->flags & WLAN_STA_ASSOC))
732 			hostap_event_new_sta(dev, sta);
733 		txt = "STA associated";
734 		sta->flags |= WLAN_STA_ASSOC;
735 		sta->last_assoc = jiffies;
736 	} else
737 		txt = "association failed";
738 
739  done:
740 	if (sta)
741 		atomic_dec(&sta->users);
742 	if (txt) {
743 		PDEBUG(DEBUG_AP, "%s: %pM assoc_cb - %s\n",
744 		       dev->name, hdr->addr1, txt);
745 	}
746 	dev_kfree_skb(skb);
747 }
748 
749 /* Called only as a tasklet (software IRQ); TX callback for poll frames used
750  * in verifying whether the STA is still present. */
hostap_ap_tx_cb_poll(struct sk_buff * skb,int ok,void * data)751 static void hostap_ap_tx_cb_poll(struct sk_buff *skb, int ok, void *data)
752 {
753 	struct ap_data *ap = data;
754 	struct ieee80211_hdr *hdr;
755 	struct sta_info *sta;
756 
757 	if (skb->len < 24)
758 		goto fail;
759 	hdr = (struct ieee80211_hdr *) skb->data;
760 	if (ok) {
761 		spin_lock(&ap->sta_table_lock);
762 		sta = ap_get_sta(ap, hdr->addr1);
763 		if (sta)
764 			sta->flags &= ~WLAN_STA_PENDING_POLL;
765 		spin_unlock(&ap->sta_table_lock);
766 	} else {
767 		PDEBUG(DEBUG_AP,
768 		       "%s: STA %pM did not ACK activity poll frame\n",
769 		       ap->local->dev->name, hdr->addr1);
770 	}
771 
772  fail:
773 	dev_kfree_skb(skb);
774 }
775 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
776 
777 
hostap_init_data(local_info_t * local)778 void hostap_init_data(local_info_t *local)
779 {
780 	struct ap_data *ap = local->ap;
781 
782 	if (ap == NULL) {
783 		printk(KERN_WARNING "hostap_init_data: ap == NULL\n");
784 		return;
785 	}
786 	memset(ap, 0, sizeof(struct ap_data));
787 	ap->local = local;
788 
789 	ap->ap_policy = GET_INT_PARM(other_ap_policy, local->card_idx);
790 	ap->bridge_packets = GET_INT_PARM(ap_bridge_packets, local->card_idx);
791 	ap->max_inactivity =
792 		GET_INT_PARM(ap_max_inactivity, local->card_idx) * HZ;
793 	ap->autom_ap_wds = GET_INT_PARM(autom_ap_wds, local->card_idx);
794 
795 	spin_lock_init(&ap->sta_table_lock);
796 	INIT_LIST_HEAD(&ap->sta_list);
797 
798 	/* Initialize task queue structure for AP management */
799 	INIT_WORK(&local->ap->add_sta_proc_queue, handle_add_proc_queue);
800 
801 	ap->tx_callback_idx =
802 		hostap_tx_callback_register(local, hostap_ap_tx_cb, ap);
803 	if (ap->tx_callback_idx == 0)
804 		printk(KERN_WARNING "%s: failed to register TX callback for "
805 		       "AP\n", local->dev->name);
806 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
807 	INIT_WORK(&local->ap->wds_oper_queue, handle_wds_oper_queue);
808 
809 	ap->tx_callback_auth =
810 		hostap_tx_callback_register(local, hostap_ap_tx_cb_auth, ap);
811 	ap->tx_callback_assoc =
812 		hostap_tx_callback_register(local, hostap_ap_tx_cb_assoc, ap);
813 	ap->tx_callback_poll =
814 		hostap_tx_callback_register(local, hostap_ap_tx_cb_poll, ap);
815 	if (ap->tx_callback_auth == 0 || ap->tx_callback_assoc == 0 ||
816 		ap->tx_callback_poll == 0)
817 		printk(KERN_WARNING "%s: failed to register TX callback for "
818 		       "AP\n", local->dev->name);
819 
820 	spin_lock_init(&ap->mac_restrictions.lock);
821 	INIT_LIST_HEAD(&ap->mac_restrictions.mac_list);
822 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
823 
824 	ap->initialized = 1;
825 }
826 
827 
hostap_init_ap_proc(local_info_t * local)828 void hostap_init_ap_proc(local_info_t *local)
829 {
830 	struct ap_data *ap = local->ap;
831 
832 	ap->proc = local->proc;
833 	if (ap->proc == NULL)
834 		return;
835 
836 #ifndef PRISM2_NO_PROCFS_DEBUG
837 	create_proc_read_entry("ap_debug", 0, ap->proc,
838 			       ap_debug_proc_read, ap);
839 #endif /* PRISM2_NO_PROCFS_DEBUG */
840 
841 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
842 	create_proc_read_entry("ap_control", 0, ap->proc,
843 			       ap_control_proc_read, ap);
844 	create_proc_read_entry("ap", 0, ap->proc,
845 			       prism2_ap_proc_read, ap);
846 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
847 
848 }
849 
850 
hostap_free_data(struct ap_data * ap)851 void hostap_free_data(struct ap_data *ap)
852 {
853 	struct sta_info *n, *sta;
854 
855 	if (ap == NULL || !ap->initialized) {
856 		printk(KERN_DEBUG "hostap_free_data: ap has not yet been "
857 		       "initialized - skip resource freeing\n");
858 		return;
859 	}
860 
861 	flush_work_sync(&ap->add_sta_proc_queue);
862 
863 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
864 	flush_work_sync(&ap->wds_oper_queue);
865 	if (ap->crypt)
866 		ap->crypt->deinit(ap->crypt_priv);
867 	ap->crypt = ap->crypt_priv = NULL;
868 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
869 
870 	list_for_each_entry_safe(sta, n, &ap->sta_list, list) {
871 		ap_sta_hash_del(ap, sta);
872 		list_del(&sta->list);
873 		if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
874 			hostap_event_expired_sta(sta->local->dev, sta);
875 		ap_free_sta(ap, sta);
876 	}
877 
878 #ifndef PRISM2_NO_PROCFS_DEBUG
879 	if (ap->proc != NULL) {
880 		remove_proc_entry("ap_debug", ap->proc);
881 	}
882 #endif /* PRISM2_NO_PROCFS_DEBUG */
883 
884 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
885 	if (ap->proc != NULL) {
886 	  remove_proc_entry("ap", ap->proc);
887 		remove_proc_entry("ap_control", ap->proc);
888 	}
889 	ap_control_flush_macs(&ap->mac_restrictions);
890 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
891 
892 	ap->initialized = 0;
893 }
894 
895 
896 /* caller should have mutex for AP STA list handling */
ap_get_sta(struct ap_data * ap,u8 * sta)897 static struct sta_info* ap_get_sta(struct ap_data *ap, u8 *sta)
898 {
899 	struct sta_info *s;
900 
901 	s = ap->sta_hash[STA_HASH(sta)];
902 	while (s != NULL && memcmp(s->addr, sta, ETH_ALEN) != 0)
903 		s = s->hnext;
904 	return s;
905 }
906 
907 
908 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
909 
910 /* Called from timer handler and from scheduled AP queue handlers */
prism2_send_mgmt(struct net_device * dev,u16 type_subtype,char * body,int body_len,u8 * addr,u16 tx_cb_idx)911 static void prism2_send_mgmt(struct net_device *dev,
912 			     u16 type_subtype, char *body,
913 			     int body_len, u8 *addr, u16 tx_cb_idx)
914 {
915 	struct hostap_interface *iface;
916 	local_info_t *local;
917 	struct ieee80211_hdr *hdr;
918 	u16 fc;
919 	struct sk_buff *skb;
920 	struct hostap_skb_tx_data *meta;
921 	int hdrlen;
922 
923 	iface = netdev_priv(dev);
924 	local = iface->local;
925 	dev = local->dev; /* always use master radio device */
926 	iface = netdev_priv(dev);
927 
928 	if (!(dev->flags & IFF_UP)) {
929 		PDEBUG(DEBUG_AP, "%s: prism2_send_mgmt - device is not UP - "
930 		       "cannot send frame\n", dev->name);
931 		return;
932 	}
933 
934 	skb = dev_alloc_skb(sizeof(*hdr) + body_len);
935 	if (skb == NULL) {
936 		PDEBUG(DEBUG_AP, "%s: prism2_send_mgmt failed to allocate "
937 		       "skb\n", dev->name);
938 		return;
939 	}
940 
941 	fc = type_subtype;
942 	hdrlen = hostap_80211_get_hdrlen(cpu_to_le16(type_subtype));
943 	hdr = (struct ieee80211_hdr *) skb_put(skb, hdrlen);
944 	if (body)
945 		memcpy(skb_put(skb, body_len), body, body_len);
946 
947 	memset(hdr, 0, hdrlen);
948 
949 	/* FIX: ctrl::ack sending used special HFA384X_TX_CTRL_802_11
950 	 * tx_control instead of using local->tx_control */
951 
952 
953 	memcpy(hdr->addr1, addr, ETH_ALEN); /* DA / RA */
954 	if (ieee80211_is_data(hdr->frame_control)) {
955 		fc |= IEEE80211_FCTL_FROMDS;
956 		memcpy(hdr->addr2, dev->dev_addr, ETH_ALEN); /* BSSID */
957 		memcpy(hdr->addr3, dev->dev_addr, ETH_ALEN); /* SA */
958 	} else if (ieee80211_is_ctl(hdr->frame_control)) {
959 		/* control:ACK does not have addr2 or addr3 */
960 		memset(hdr->addr2, 0, ETH_ALEN);
961 		memset(hdr->addr3, 0, ETH_ALEN);
962 	} else {
963 		memcpy(hdr->addr2, dev->dev_addr, ETH_ALEN); /* SA */
964 		memcpy(hdr->addr3, dev->dev_addr, ETH_ALEN); /* BSSID */
965 	}
966 
967 	hdr->frame_control = cpu_to_le16(fc);
968 
969 	meta = (struct hostap_skb_tx_data *) skb->cb;
970 	memset(meta, 0, sizeof(*meta));
971 	meta->magic = HOSTAP_SKB_TX_DATA_MAGIC;
972 	meta->iface = iface;
973 	meta->tx_cb_idx = tx_cb_idx;
974 
975 	skb->dev = dev;
976 	skb_reset_mac_header(skb);
977 	skb_reset_network_header(skb);
978 	dev_queue_xmit(skb);
979 }
980 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
981 
982 
prism2_sta_proc_read(char * page,char ** start,off_t off,int count,int * eof,void * data)983 static int prism2_sta_proc_read(char *page, char **start, off_t off,
984 				int count, int *eof, void *data)
985 {
986 	char *p = page;
987 	struct sta_info *sta = (struct sta_info *) data;
988 	int i;
989 
990 	/* FIX: possible race condition.. the STA data could have just expired,
991 	 * but proc entry was still here so that the read could have started;
992 	 * some locking should be done here.. */
993 
994 	if (off != 0) {
995 		*eof = 1;
996 		return 0;
997 	}
998 
999 	p += sprintf(p, "%s=%pM\nusers=%d\naid=%d\n"
1000 		     "flags=0x%04x%s%s%s%s%s%s%s\n"
1001 		     "capability=0x%02x\nlisten_interval=%d\nsupported_rates=",
1002 		     sta->ap ? "AP" : "STA",
1003 		     sta->addr, atomic_read(&sta->users), sta->aid,
1004 		     sta->flags,
1005 		     sta->flags & WLAN_STA_AUTH ? " AUTH" : "",
1006 		     sta->flags & WLAN_STA_ASSOC ? " ASSOC" : "",
1007 		     sta->flags & WLAN_STA_PS ? " PS" : "",
1008 		     sta->flags & WLAN_STA_TIM ? " TIM" : "",
1009 		     sta->flags & WLAN_STA_PERM ? " PERM" : "",
1010 		     sta->flags & WLAN_STA_AUTHORIZED ? " AUTHORIZED" : "",
1011 		     sta->flags & WLAN_STA_PENDING_POLL ? " POLL" : "",
1012 		     sta->capability, sta->listen_interval);
1013 	/* supported_rates: 500 kbit/s units with msb ignored */
1014 	for (i = 0; i < sizeof(sta->supported_rates); i++)
1015 		if (sta->supported_rates[i] != 0)
1016 			p += sprintf(p, "%d%sMbps ",
1017 				     (sta->supported_rates[i] & 0x7f) / 2,
1018 				     sta->supported_rates[i] & 1 ? ".5" : "");
1019 	p += sprintf(p, "\njiffies=%lu\nlast_auth=%lu\nlast_assoc=%lu\n"
1020 		     "last_rx=%lu\nlast_tx=%lu\nrx_packets=%lu\n"
1021 		     "tx_packets=%lu\n"
1022 		     "rx_bytes=%lu\ntx_bytes=%lu\nbuffer_count=%d\n"
1023 		     "last_rx: silence=%d dBm signal=%d dBm rate=%d%s Mbps\n"
1024 		     "tx_rate=%d\ntx[1M]=%d\ntx[2M]=%d\ntx[5.5M]=%d\n"
1025 		     "tx[11M]=%d\n"
1026 		     "rx[1M]=%d\nrx[2M]=%d\nrx[5.5M]=%d\nrx[11M]=%d\n",
1027 		     jiffies, sta->last_auth, sta->last_assoc, sta->last_rx,
1028 		     sta->last_tx,
1029 		     sta->rx_packets, sta->tx_packets, sta->rx_bytes,
1030 		     sta->tx_bytes, skb_queue_len(&sta->tx_buf),
1031 		     sta->last_rx_silence,
1032 		     sta->last_rx_signal, sta->last_rx_rate / 10,
1033 		     sta->last_rx_rate % 10 ? ".5" : "",
1034 		     sta->tx_rate, sta->tx_count[0], sta->tx_count[1],
1035 		     sta->tx_count[2], sta->tx_count[3],  sta->rx_count[0],
1036 		     sta->rx_count[1], sta->rx_count[2], sta->rx_count[3]);
1037 	if (sta->crypt && sta->crypt->ops && sta->crypt->ops->print_stats)
1038 		p = sta->crypt->ops->print_stats(p, sta->crypt->priv);
1039 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1040 	if (sta->ap) {
1041 		if (sta->u.ap.channel >= 0)
1042 			p += sprintf(p, "channel=%d\n", sta->u.ap.channel);
1043 		p += sprintf(p, "ssid=");
1044 		for (i = 0; i < sta->u.ap.ssid_len; i++)
1045 			p += sprintf(p, ((sta->u.ap.ssid[i] >= 32 &&
1046 					  sta->u.ap.ssid[i] < 127) ?
1047 					 "%c" : "<%02x>"),
1048 				     sta->u.ap.ssid[i]);
1049 		p += sprintf(p, "\n");
1050 	}
1051 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1052 
1053 	return (p - page);
1054 }
1055 
1056 
handle_add_proc_queue(struct work_struct * work)1057 static void handle_add_proc_queue(struct work_struct *work)
1058 {
1059 	struct ap_data *ap = container_of(work, struct ap_data,
1060 					  add_sta_proc_queue);
1061 	struct sta_info *sta;
1062 	char name[20];
1063 	struct add_sta_proc_data *entry, *prev;
1064 
1065 	entry = ap->add_sta_proc_entries;
1066 	ap->add_sta_proc_entries = NULL;
1067 
1068 	while (entry) {
1069 		spin_lock_bh(&ap->sta_table_lock);
1070 		sta = ap_get_sta(ap, entry->addr);
1071 		if (sta)
1072 			atomic_inc(&sta->users);
1073 		spin_unlock_bh(&ap->sta_table_lock);
1074 
1075 		if (sta) {
1076 			sprintf(name, "%pM", sta->addr);
1077 			sta->proc = create_proc_read_entry(
1078 				name, 0, ap->proc,
1079 				prism2_sta_proc_read, sta);
1080 
1081 			atomic_dec(&sta->users);
1082 		}
1083 
1084 		prev = entry;
1085 		entry = entry->next;
1086 		kfree(prev);
1087 	}
1088 }
1089 
1090 
ap_add_sta(struct ap_data * ap,u8 * addr)1091 static struct sta_info * ap_add_sta(struct ap_data *ap, u8 *addr)
1092 {
1093 	struct sta_info *sta;
1094 
1095 	sta = kzalloc(sizeof(struct sta_info), GFP_ATOMIC);
1096 	if (sta == NULL) {
1097 		PDEBUG(DEBUG_AP, "AP: kmalloc failed\n");
1098 		return NULL;
1099 	}
1100 
1101 	/* initialize STA info data */
1102 	sta->local = ap->local;
1103 	skb_queue_head_init(&sta->tx_buf);
1104 	memcpy(sta->addr, addr, ETH_ALEN);
1105 
1106 	atomic_inc(&sta->users);
1107 	spin_lock_bh(&ap->sta_table_lock);
1108 	list_add(&sta->list, &ap->sta_list);
1109 	ap->num_sta++;
1110 	ap_sta_hash_add(ap, sta);
1111 	spin_unlock_bh(&ap->sta_table_lock);
1112 
1113 	if (ap->proc) {
1114 		struct add_sta_proc_data *entry;
1115 		/* schedule a non-interrupt context process to add a procfs
1116 		 * entry for the STA since procfs code use GFP_KERNEL */
1117 		entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
1118 		if (entry) {
1119 			memcpy(entry->addr, sta->addr, ETH_ALEN);
1120 			entry->next = ap->add_sta_proc_entries;
1121 			ap->add_sta_proc_entries = entry;
1122 			schedule_work(&ap->add_sta_proc_queue);
1123 		} else
1124 			printk(KERN_DEBUG "Failed to add STA proc data\n");
1125 	}
1126 
1127 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1128 	init_timer(&sta->timer);
1129 	sta->timer.expires = jiffies + ap->max_inactivity;
1130 	sta->timer.data = (unsigned long) sta;
1131 	sta->timer.function = ap_handle_timer;
1132 	if (!ap->local->hostapd)
1133 		add_timer(&sta->timer);
1134 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1135 
1136 	return sta;
1137 }
1138 
1139 
ap_tx_rate_ok(int rateidx,struct sta_info * sta,local_info_t * local)1140 static int ap_tx_rate_ok(int rateidx, struct sta_info *sta,
1141 			 local_info_t *local)
1142 {
1143 	if (rateidx > sta->tx_max_rate ||
1144 	    !(sta->tx_supp_rates & (1 << rateidx)))
1145 		return 0;
1146 
1147 	if (local->tx_rate_control != 0 &&
1148 	    !(local->tx_rate_control & (1 << rateidx)))
1149 		return 0;
1150 
1151 	return 1;
1152 }
1153 
1154 
prism2_check_tx_rates(struct sta_info * sta)1155 static void prism2_check_tx_rates(struct sta_info *sta)
1156 {
1157 	int i;
1158 
1159 	sta->tx_supp_rates = 0;
1160 	for (i = 0; i < sizeof(sta->supported_rates); i++) {
1161 		if ((sta->supported_rates[i] & 0x7f) == 2)
1162 			sta->tx_supp_rates |= WLAN_RATE_1M;
1163 		if ((sta->supported_rates[i] & 0x7f) == 4)
1164 			sta->tx_supp_rates |= WLAN_RATE_2M;
1165 		if ((sta->supported_rates[i] & 0x7f) == 11)
1166 			sta->tx_supp_rates |= WLAN_RATE_5M5;
1167 		if ((sta->supported_rates[i] & 0x7f) == 22)
1168 			sta->tx_supp_rates |= WLAN_RATE_11M;
1169 	}
1170 	sta->tx_max_rate = sta->tx_rate = sta->tx_rate_idx = 0;
1171 	if (sta->tx_supp_rates & WLAN_RATE_1M) {
1172 		sta->tx_max_rate = 0;
1173 		if (ap_tx_rate_ok(0, sta, sta->local)) {
1174 			sta->tx_rate = 10;
1175 			sta->tx_rate_idx = 0;
1176 		}
1177 	}
1178 	if (sta->tx_supp_rates & WLAN_RATE_2M) {
1179 		sta->tx_max_rate = 1;
1180 		if (ap_tx_rate_ok(1, sta, sta->local)) {
1181 			sta->tx_rate = 20;
1182 			sta->tx_rate_idx = 1;
1183 		}
1184 	}
1185 	if (sta->tx_supp_rates & WLAN_RATE_5M5) {
1186 		sta->tx_max_rate = 2;
1187 		if (ap_tx_rate_ok(2, sta, sta->local)) {
1188 			sta->tx_rate = 55;
1189 			sta->tx_rate_idx = 2;
1190 		}
1191 	}
1192 	if (sta->tx_supp_rates & WLAN_RATE_11M) {
1193 		sta->tx_max_rate = 3;
1194 		if (ap_tx_rate_ok(3, sta, sta->local)) {
1195 			sta->tx_rate = 110;
1196 			sta->tx_rate_idx = 3;
1197 		}
1198 	}
1199 }
1200 
1201 
1202 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1203 
ap_crypt_init(struct ap_data * ap)1204 static void ap_crypt_init(struct ap_data *ap)
1205 {
1206 	ap->crypt = lib80211_get_crypto_ops("WEP");
1207 
1208 	if (ap->crypt) {
1209 		if (ap->crypt->init) {
1210 			ap->crypt_priv = ap->crypt->init(0);
1211 			if (ap->crypt_priv == NULL)
1212 				ap->crypt = NULL;
1213 			else {
1214 				u8 key[WEP_KEY_LEN];
1215 				get_random_bytes(key, WEP_KEY_LEN);
1216 				ap->crypt->set_key(key, WEP_KEY_LEN, NULL,
1217 						   ap->crypt_priv);
1218 			}
1219 		}
1220 	}
1221 
1222 	if (ap->crypt == NULL) {
1223 		printk(KERN_WARNING "AP could not initialize WEP: load module "
1224 		       "lib80211_crypt_wep.ko\n");
1225 	}
1226 }
1227 
1228 
1229 /* Generate challenge data for shared key authentication. IEEE 802.11 specifies
1230  * that WEP algorithm is used for generating challenge. This should be unique,
1231  * but otherwise there is not really need for randomness etc. Initialize WEP
1232  * with pseudo random key and then use increasing IV to get unique challenge
1233  * streams.
1234  *
1235  * Called only as a scheduled task for pending AP frames.
1236  */
ap_auth_make_challenge(struct ap_data * ap)1237 static char * ap_auth_make_challenge(struct ap_data *ap)
1238 {
1239 	char *tmpbuf;
1240 	struct sk_buff *skb;
1241 
1242 	if (ap->crypt == NULL) {
1243 		ap_crypt_init(ap);
1244 		if (ap->crypt == NULL)
1245 			return NULL;
1246 	}
1247 
1248 	tmpbuf = kmalloc(WLAN_AUTH_CHALLENGE_LEN, GFP_ATOMIC);
1249 	if (tmpbuf == NULL) {
1250 		PDEBUG(DEBUG_AP, "AP: kmalloc failed for challenge\n");
1251 		return NULL;
1252 	}
1253 
1254 	skb = dev_alloc_skb(WLAN_AUTH_CHALLENGE_LEN +
1255 			    ap->crypt->extra_mpdu_prefix_len +
1256 			    ap->crypt->extra_mpdu_postfix_len);
1257 	if (skb == NULL) {
1258 		kfree(tmpbuf);
1259 		return NULL;
1260 	}
1261 
1262 	skb_reserve(skb, ap->crypt->extra_mpdu_prefix_len);
1263 	memset(skb_put(skb, WLAN_AUTH_CHALLENGE_LEN), 0,
1264 	       WLAN_AUTH_CHALLENGE_LEN);
1265 	if (ap->crypt->encrypt_mpdu(skb, 0, ap->crypt_priv)) {
1266 		dev_kfree_skb(skb);
1267 		kfree(tmpbuf);
1268 		return NULL;
1269 	}
1270 
1271 	skb_copy_from_linear_data_offset(skb, ap->crypt->extra_mpdu_prefix_len,
1272 					 tmpbuf, WLAN_AUTH_CHALLENGE_LEN);
1273 	dev_kfree_skb(skb);
1274 
1275 	return tmpbuf;
1276 }
1277 
1278 
1279 /* Called only as a scheduled task for pending AP frames. */
handle_authen(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)1280 static void handle_authen(local_info_t *local, struct sk_buff *skb,
1281 			  struct hostap_80211_rx_status *rx_stats)
1282 {
1283 	struct net_device *dev = local->dev;
1284 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1285 	size_t hdrlen;
1286 	struct ap_data *ap = local->ap;
1287 	char body[8 + WLAN_AUTH_CHALLENGE_LEN], *challenge = NULL;
1288 	int len, olen;
1289 	u16 auth_alg, auth_transaction, status_code;
1290 	__le16 *pos;
1291 	u16 resp = WLAN_STATUS_SUCCESS;
1292 	struct sta_info *sta = NULL;
1293 	struct lib80211_crypt_data *crypt;
1294 	char *txt = "";
1295 
1296 	len = skb->len - IEEE80211_MGMT_HDR_LEN;
1297 
1298 	hdrlen = hostap_80211_get_hdrlen(hdr->frame_control);
1299 
1300 	if (len < 6) {
1301 		PDEBUG(DEBUG_AP, "%s: handle_authen - too short payload "
1302 		       "(len=%d) from %pM\n", dev->name, len, hdr->addr2);
1303 		return;
1304 	}
1305 
1306 	spin_lock_bh(&local->ap->sta_table_lock);
1307 	sta = ap_get_sta(local->ap, hdr->addr2);
1308 	if (sta)
1309 		atomic_inc(&sta->users);
1310 	spin_unlock_bh(&local->ap->sta_table_lock);
1311 
1312 	if (sta && sta->crypt)
1313 		crypt = sta->crypt;
1314 	else {
1315 		int idx = 0;
1316 		if (skb->len >= hdrlen + 3)
1317 			idx = skb->data[hdrlen + 3] >> 6;
1318 		crypt = local->crypt_info.crypt[idx];
1319 	}
1320 
1321 	pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1322 	auth_alg = __le16_to_cpu(*pos);
1323 	pos++;
1324 	auth_transaction = __le16_to_cpu(*pos);
1325 	pos++;
1326 	status_code = __le16_to_cpu(*pos);
1327 	pos++;
1328 
1329 	if (memcmp(dev->dev_addr, hdr->addr2, ETH_ALEN) == 0 ||
1330 	    ap_control_mac_deny(&ap->mac_restrictions, hdr->addr2)) {
1331 		txt = "authentication denied";
1332 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1333 		goto fail;
1334 	}
1335 
1336 	if (((local->auth_algs & PRISM2_AUTH_OPEN) &&
1337 	     auth_alg == WLAN_AUTH_OPEN) ||
1338 	    ((local->auth_algs & PRISM2_AUTH_SHARED_KEY) &&
1339 	     crypt && auth_alg == WLAN_AUTH_SHARED_KEY)) {
1340 	} else {
1341 		txt = "unsupported algorithm";
1342 		resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
1343 		goto fail;
1344 	}
1345 
1346 	if (len >= 8) {
1347 		u8 *u = (u8 *) pos;
1348 		if (*u == WLAN_EID_CHALLENGE) {
1349 			if (*(u + 1) != WLAN_AUTH_CHALLENGE_LEN) {
1350 				txt = "invalid challenge len";
1351 				resp = WLAN_STATUS_CHALLENGE_FAIL;
1352 				goto fail;
1353 			}
1354 			if (len - 8 < WLAN_AUTH_CHALLENGE_LEN) {
1355 				txt = "challenge underflow";
1356 				resp = WLAN_STATUS_CHALLENGE_FAIL;
1357 				goto fail;
1358 			}
1359 			challenge = (char *) (u + 2);
1360 		}
1361 	}
1362 
1363 	if (sta && sta->ap) {
1364 		if (time_after(jiffies, sta->u.ap.last_beacon +
1365 			       (10 * sta->listen_interval * HZ) / 1024)) {
1366 			PDEBUG(DEBUG_AP, "%s: no beacons received for a while,"
1367 			       " assuming AP %pM is now STA\n",
1368 			       dev->name, sta->addr);
1369 			sta->ap = 0;
1370 			sta->flags = 0;
1371 			sta->u.sta.challenge = NULL;
1372 		} else {
1373 			txt = "AP trying to authenticate?";
1374 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1375 			goto fail;
1376 		}
1377 	}
1378 
1379 	if ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 1) ||
1380 	    (auth_alg == WLAN_AUTH_SHARED_KEY &&
1381 	     (auth_transaction == 1 ||
1382 	      (auth_transaction == 3 && sta != NULL &&
1383 	       sta->u.sta.challenge != NULL)))) {
1384 	} else {
1385 		txt = "unknown authentication transaction number";
1386 		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
1387 		goto fail;
1388 	}
1389 
1390 	if (sta == NULL) {
1391 		txt = "new STA";
1392 
1393 		if (local->ap->num_sta >= MAX_STA_COUNT) {
1394 			/* FIX: might try to remove some old STAs first? */
1395 			txt = "no more room for new STAs";
1396 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1397 			goto fail;
1398 		}
1399 
1400 		sta = ap_add_sta(local->ap, hdr->addr2);
1401 		if (sta == NULL) {
1402 			txt = "ap_add_sta failed";
1403 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1404 			goto fail;
1405 		}
1406 	}
1407 
1408 	switch (auth_alg) {
1409 	case WLAN_AUTH_OPEN:
1410 		txt = "authOK";
1411 		/* IEEE 802.11 standard is not completely clear about
1412 		 * whether STA is considered authenticated after
1413 		 * authentication OK frame has been send or after it
1414 		 * has been ACKed. In order to reduce interoperability
1415 		 * issues, mark the STA authenticated before ACK. */
1416 		sta->flags |= WLAN_STA_AUTH;
1417 		break;
1418 
1419 	case WLAN_AUTH_SHARED_KEY:
1420 		if (auth_transaction == 1) {
1421 			if (sta->u.sta.challenge == NULL) {
1422 				sta->u.sta.challenge =
1423 					ap_auth_make_challenge(local->ap);
1424 				if (sta->u.sta.challenge == NULL) {
1425 					resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1426 					goto fail;
1427 				}
1428 			}
1429 		} else {
1430 			if (sta->u.sta.challenge == NULL ||
1431 			    challenge == NULL ||
1432 			    memcmp(sta->u.sta.challenge, challenge,
1433 				   WLAN_AUTH_CHALLENGE_LEN) != 0 ||
1434 			    !ieee80211_has_protected(hdr->frame_control)) {
1435 				txt = "challenge response incorrect";
1436 				resp = WLAN_STATUS_CHALLENGE_FAIL;
1437 				goto fail;
1438 			}
1439 
1440 			txt = "challenge OK - authOK";
1441 			/* IEEE 802.11 standard is not completely clear about
1442 			 * whether STA is considered authenticated after
1443 			 * authentication OK frame has been send or after it
1444 			 * has been ACKed. In order to reduce interoperability
1445 			 * issues, mark the STA authenticated before ACK. */
1446 			sta->flags |= WLAN_STA_AUTH;
1447 			kfree(sta->u.sta.challenge);
1448 			sta->u.sta.challenge = NULL;
1449 		}
1450 		break;
1451 	}
1452 
1453  fail:
1454 	pos = (__le16 *) body;
1455 	*pos = cpu_to_le16(auth_alg);
1456 	pos++;
1457 	*pos = cpu_to_le16(auth_transaction + 1);
1458 	pos++;
1459 	*pos = cpu_to_le16(resp); /* status_code */
1460 	pos++;
1461 	olen = 6;
1462 
1463 	if (resp == WLAN_STATUS_SUCCESS && sta != NULL &&
1464 	    sta->u.sta.challenge != NULL &&
1465 	    auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 1) {
1466 		u8 *tmp = (u8 *) pos;
1467 		*tmp++ = WLAN_EID_CHALLENGE;
1468 		*tmp++ = WLAN_AUTH_CHALLENGE_LEN;
1469 		pos++;
1470 		memcpy(pos, sta->u.sta.challenge, WLAN_AUTH_CHALLENGE_LEN);
1471 		olen += 2 + WLAN_AUTH_CHALLENGE_LEN;
1472 	}
1473 
1474 	prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_AUTH,
1475 			 body, olen, hdr->addr2, ap->tx_callback_auth);
1476 
1477 	if (sta) {
1478 		sta->last_rx = jiffies;
1479 		atomic_dec(&sta->users);
1480 	}
1481 
1482 	if (resp) {
1483 		PDEBUG(DEBUG_AP, "%s: %pM auth (alg=%d "
1484 		       "trans#=%d stat=%d len=%d fc=%04x) ==> %d (%s)\n",
1485 		       dev->name, hdr->addr2,
1486 		       auth_alg, auth_transaction, status_code, len,
1487 		       le16_to_cpu(hdr->frame_control), resp, txt);
1488 	}
1489 }
1490 
1491 
1492 /* Called only as a scheduled task for pending AP frames. */
handle_assoc(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats,int reassoc)1493 static void handle_assoc(local_info_t *local, struct sk_buff *skb,
1494 			 struct hostap_80211_rx_status *rx_stats, int reassoc)
1495 {
1496 	struct net_device *dev = local->dev;
1497 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1498 	char body[12], *p, *lpos;
1499 	int len, left;
1500 	__le16 *pos;
1501 	u16 resp = WLAN_STATUS_SUCCESS;
1502 	struct sta_info *sta = NULL;
1503 	int send_deauth = 0;
1504 	char *txt = "";
1505 	u8 prev_ap[ETH_ALEN];
1506 
1507 	left = len = skb->len - IEEE80211_MGMT_HDR_LEN;
1508 
1509 	if (len < (reassoc ? 10 : 4)) {
1510 		PDEBUG(DEBUG_AP, "%s: handle_assoc - too short payload "
1511 		       "(len=%d, reassoc=%d) from %pM\n",
1512 		       dev->name, len, reassoc, hdr->addr2);
1513 		return;
1514 	}
1515 
1516 	spin_lock_bh(&local->ap->sta_table_lock);
1517 	sta = ap_get_sta(local->ap, hdr->addr2);
1518 	if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
1519 		spin_unlock_bh(&local->ap->sta_table_lock);
1520 		txt = "trying to associate before authentication";
1521 		send_deauth = 1;
1522 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1523 		sta = NULL; /* do not decrement sta->users */
1524 		goto fail;
1525 	}
1526 	atomic_inc(&sta->users);
1527 	spin_unlock_bh(&local->ap->sta_table_lock);
1528 
1529 	pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1530 	sta->capability = __le16_to_cpu(*pos);
1531 	pos++; left -= 2;
1532 	sta->listen_interval = __le16_to_cpu(*pos);
1533 	pos++; left -= 2;
1534 
1535 	if (reassoc) {
1536 		memcpy(prev_ap, pos, ETH_ALEN);
1537 		pos++; pos++; pos++; left -= 6;
1538 	} else
1539 		memset(prev_ap, 0, ETH_ALEN);
1540 
1541 	if (left >= 2) {
1542 		unsigned int ileft;
1543 		unsigned char *u = (unsigned char *) pos;
1544 
1545 		if (*u == WLAN_EID_SSID) {
1546 			u++; left--;
1547 			ileft = *u;
1548 			u++; left--;
1549 
1550 			if (ileft > left || ileft > MAX_SSID_LEN) {
1551 				txt = "SSID overflow";
1552 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1553 				goto fail;
1554 			}
1555 
1556 			if (ileft != strlen(local->essid) ||
1557 			    memcmp(local->essid, u, ileft) != 0) {
1558 				txt = "not our SSID";
1559 				resp = WLAN_STATUS_ASSOC_DENIED_UNSPEC;
1560 				goto fail;
1561 			}
1562 
1563 			u += ileft;
1564 			left -= ileft;
1565 		}
1566 
1567 		if (left >= 2 && *u == WLAN_EID_SUPP_RATES) {
1568 			u++; left--;
1569 			ileft = *u;
1570 			u++; left--;
1571 
1572 			if (ileft > left || ileft == 0 ||
1573 			    ileft > WLAN_SUPP_RATES_MAX) {
1574 				txt = "SUPP_RATES len error";
1575 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1576 				goto fail;
1577 			}
1578 
1579 			memset(sta->supported_rates, 0,
1580 			       sizeof(sta->supported_rates));
1581 			memcpy(sta->supported_rates, u, ileft);
1582 			prism2_check_tx_rates(sta);
1583 
1584 			u += ileft;
1585 			left -= ileft;
1586 		}
1587 
1588 		if (left > 0) {
1589 			PDEBUG(DEBUG_AP, "%s: assoc from %pM"
1590 			       " with extra data (%d bytes) [",
1591 			       dev->name, hdr->addr2, left);
1592 			while (left > 0) {
1593 				PDEBUG2(DEBUG_AP, "<%02x>", *u);
1594 				u++; left--;
1595 			}
1596 			PDEBUG2(DEBUG_AP, "]\n");
1597 		}
1598 	} else {
1599 		txt = "frame underflow";
1600 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1601 		goto fail;
1602 	}
1603 
1604 	/* get a unique AID */
1605 	if (sta->aid > 0)
1606 		txt = "OK, old AID";
1607 	else {
1608 		spin_lock_bh(&local->ap->sta_table_lock);
1609 		for (sta->aid = 1; sta->aid <= MAX_AID_TABLE_SIZE; sta->aid++)
1610 			if (local->ap->sta_aid[sta->aid - 1] == NULL)
1611 				break;
1612 		if (sta->aid > MAX_AID_TABLE_SIZE) {
1613 			sta->aid = 0;
1614 			spin_unlock_bh(&local->ap->sta_table_lock);
1615 			resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
1616 			txt = "no room for more AIDs";
1617 		} else {
1618 			local->ap->sta_aid[sta->aid - 1] = sta;
1619 			spin_unlock_bh(&local->ap->sta_table_lock);
1620 			txt = "OK, new AID";
1621 		}
1622 	}
1623 
1624  fail:
1625 	pos = (__le16 *) body;
1626 
1627 	if (send_deauth) {
1628 		*pos = cpu_to_le16(WLAN_REASON_STA_REQ_ASSOC_WITHOUT_AUTH);
1629 		pos++;
1630 	} else {
1631 		/* FIX: CF-Pollable and CF-PollReq should be set to match the
1632 		 * values in beacons/probe responses */
1633 		/* FIX: how about privacy and WEP? */
1634 		/* capability */
1635 		*pos = cpu_to_le16(WLAN_CAPABILITY_ESS);
1636 		pos++;
1637 
1638 		/* status_code */
1639 		*pos = cpu_to_le16(resp);
1640 		pos++;
1641 
1642 		*pos = cpu_to_le16((sta && sta->aid > 0 ? sta->aid : 0) |
1643 				     BIT(14) | BIT(15)); /* AID */
1644 		pos++;
1645 
1646 		/* Supported rates (Information element) */
1647 		p = (char *) pos;
1648 		*p++ = WLAN_EID_SUPP_RATES;
1649 		lpos = p;
1650 		*p++ = 0; /* len */
1651 		if (local->tx_rate_control & WLAN_RATE_1M) {
1652 			*p++ = local->basic_rates & WLAN_RATE_1M ? 0x82 : 0x02;
1653 			(*lpos)++;
1654 		}
1655 		if (local->tx_rate_control & WLAN_RATE_2M) {
1656 			*p++ = local->basic_rates & WLAN_RATE_2M ? 0x84 : 0x04;
1657 			(*lpos)++;
1658 		}
1659 		if (local->tx_rate_control & WLAN_RATE_5M5) {
1660 			*p++ = local->basic_rates & WLAN_RATE_5M5 ?
1661 				0x8b : 0x0b;
1662 			(*lpos)++;
1663 		}
1664 		if (local->tx_rate_control & WLAN_RATE_11M) {
1665 			*p++ = local->basic_rates & WLAN_RATE_11M ?
1666 				0x96 : 0x16;
1667 			(*lpos)++;
1668 		}
1669 		pos = (__le16 *) p;
1670 	}
1671 
1672 	prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
1673 			 (send_deauth ? IEEE80211_STYPE_DEAUTH :
1674 			  (reassoc ? IEEE80211_STYPE_REASSOC_RESP :
1675 			   IEEE80211_STYPE_ASSOC_RESP)),
1676 			 body, (u8 *) pos - (u8 *) body,
1677 			 hdr->addr2,
1678 			 send_deauth ? 0 : local->ap->tx_callback_assoc);
1679 
1680 	if (sta) {
1681 		if (resp == WLAN_STATUS_SUCCESS) {
1682 			sta->last_rx = jiffies;
1683 			/* STA will be marked associated from TX callback, if
1684 			 * AssocResp is ACKed */
1685 		}
1686 		atomic_dec(&sta->users);
1687 	}
1688 
1689 #if 0
1690 	PDEBUG(DEBUG_AP, "%s: %pM %sassoc (len=%d "
1691 	       "prev_ap=%pM) => %d(%d) (%s)\n",
1692 	       dev->name,
1693 	       hdr->addr2,
1694 	       reassoc ? "re" : "", len,
1695 	       prev_ap,
1696 	       resp, send_deauth, txt);
1697 #endif
1698 }
1699 
1700 
1701 /* Called only as a scheduled task for pending AP frames. */
handle_deauth(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)1702 static void handle_deauth(local_info_t *local, struct sk_buff *skb,
1703 			  struct hostap_80211_rx_status *rx_stats)
1704 {
1705 	struct net_device *dev = local->dev;
1706 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1707 	char *body = (char *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1708 	int len;
1709 	u16 reason_code;
1710 	__le16 *pos;
1711 	struct sta_info *sta = NULL;
1712 
1713 	len = skb->len - IEEE80211_MGMT_HDR_LEN;
1714 
1715 	if (len < 2) {
1716 		printk("handle_deauth - too short payload (len=%d)\n", len);
1717 		return;
1718 	}
1719 
1720 	pos = (__le16 *) body;
1721 	reason_code = le16_to_cpu(*pos);
1722 
1723 	PDEBUG(DEBUG_AP, "%s: deauthentication: %pM len=%d, "
1724 	       "reason_code=%d\n", dev->name, hdr->addr2,
1725 	       len, reason_code);
1726 
1727 	spin_lock_bh(&local->ap->sta_table_lock);
1728 	sta = ap_get_sta(local->ap, hdr->addr2);
1729 	if (sta != NULL) {
1730 		if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
1731 			hostap_event_expired_sta(local->dev, sta);
1732 		sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
1733 	}
1734 	spin_unlock_bh(&local->ap->sta_table_lock);
1735 	if (sta == NULL) {
1736 		printk("%s: deauthentication from %pM, "
1737 	       "reason_code=%d, but STA not authenticated\n", dev->name,
1738 		       hdr->addr2, reason_code);
1739 	}
1740 }
1741 
1742 
1743 /* Called only as a scheduled task for pending AP frames. */
handle_disassoc(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)1744 static void handle_disassoc(local_info_t *local, struct sk_buff *skb,
1745 			    struct hostap_80211_rx_status *rx_stats)
1746 {
1747 	struct net_device *dev = local->dev;
1748 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1749 	char *body = skb->data + IEEE80211_MGMT_HDR_LEN;
1750 	int len;
1751 	u16 reason_code;
1752 	__le16 *pos;
1753 	struct sta_info *sta = NULL;
1754 
1755 	len = skb->len - IEEE80211_MGMT_HDR_LEN;
1756 
1757 	if (len < 2) {
1758 		printk("handle_disassoc - too short payload (len=%d)\n", len);
1759 		return;
1760 	}
1761 
1762 	pos = (__le16 *) body;
1763 	reason_code = le16_to_cpu(*pos);
1764 
1765 	PDEBUG(DEBUG_AP, "%s: disassociation: %pM len=%d, "
1766 	       "reason_code=%d\n", dev->name, hdr->addr2,
1767 	       len, reason_code);
1768 
1769 	spin_lock_bh(&local->ap->sta_table_lock);
1770 	sta = ap_get_sta(local->ap, hdr->addr2);
1771 	if (sta != NULL) {
1772 		if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
1773 			hostap_event_expired_sta(local->dev, sta);
1774 		sta->flags &= ~WLAN_STA_ASSOC;
1775 	}
1776 	spin_unlock_bh(&local->ap->sta_table_lock);
1777 	if (sta == NULL) {
1778 		printk("%s: disassociation from %pM, "
1779 		       "reason_code=%d, but STA not authenticated\n",
1780 		       dev->name, hdr->addr2, reason_code);
1781 	}
1782 }
1783 
1784 
1785 /* Called only as a scheduled task for pending AP frames. */
ap_handle_data_nullfunc(local_info_t * local,struct ieee80211_hdr * hdr)1786 static void ap_handle_data_nullfunc(local_info_t *local,
1787 				    struct ieee80211_hdr *hdr)
1788 {
1789 	struct net_device *dev = local->dev;
1790 
1791 	/* some STA f/w's seem to require control::ACK frame for
1792 	 * data::nullfunc, but at least Prism2 station f/w version 0.8.0 does
1793 	 * not send this..
1794 	 * send control::ACK for the data::nullfunc */
1795 
1796 	printk(KERN_DEBUG "Sending control::ACK for data::nullfunc\n");
1797 	prism2_send_mgmt(dev, IEEE80211_FTYPE_CTL | IEEE80211_STYPE_ACK,
1798 			 NULL, 0, hdr->addr2, 0);
1799 }
1800 
1801 
1802 /* Called only as a scheduled task for pending AP frames. */
ap_handle_dropped_data(local_info_t * local,struct ieee80211_hdr * hdr)1803 static void ap_handle_dropped_data(local_info_t *local,
1804 				   struct ieee80211_hdr *hdr)
1805 {
1806 	struct net_device *dev = local->dev;
1807 	struct sta_info *sta;
1808 	__le16 reason;
1809 
1810 	spin_lock_bh(&local->ap->sta_table_lock);
1811 	sta = ap_get_sta(local->ap, hdr->addr2);
1812 	if (sta)
1813 		atomic_inc(&sta->users);
1814 	spin_unlock_bh(&local->ap->sta_table_lock);
1815 
1816 	if (sta != NULL && (sta->flags & WLAN_STA_ASSOC)) {
1817 		PDEBUG(DEBUG_AP, "ap_handle_dropped_data: STA is now okay?\n");
1818 		atomic_dec(&sta->users);
1819 		return;
1820 	}
1821 
1822 	reason = cpu_to_le16(WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1823 	prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
1824 			 ((sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) ?
1825 			  IEEE80211_STYPE_DEAUTH : IEEE80211_STYPE_DISASSOC),
1826 			 (char *) &reason, sizeof(reason), hdr->addr2, 0);
1827 
1828 	if (sta)
1829 		atomic_dec(&sta->users);
1830 }
1831 
1832 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1833 
1834 
1835 /* Called only as a scheduled task for pending AP frames. */
pspoll_send_buffered(local_info_t * local,struct sta_info * sta,struct sk_buff * skb)1836 static void pspoll_send_buffered(local_info_t *local, struct sta_info *sta,
1837 				 struct sk_buff *skb)
1838 {
1839 	struct hostap_skb_tx_data *meta;
1840 
1841 	if (!(sta->flags & WLAN_STA_PS)) {
1842 		/* Station has moved to non-PS mode, so send all buffered
1843 		 * frames using normal device queue. */
1844 		dev_queue_xmit(skb);
1845 		return;
1846 	}
1847 
1848 	/* add a flag for hostap_handle_sta_tx() to know that this skb should
1849 	 * be passed through even though STA is using PS */
1850 	meta = (struct hostap_skb_tx_data *) skb->cb;
1851 	meta->flags |= HOSTAP_TX_FLAGS_BUFFERED_FRAME;
1852 	if (!skb_queue_empty(&sta->tx_buf)) {
1853 		/* indicate to STA that more frames follow */
1854 		meta->flags |= HOSTAP_TX_FLAGS_ADD_MOREDATA;
1855 	}
1856 	dev_queue_xmit(skb);
1857 }
1858 
1859 
1860 /* Called only as a scheduled task for pending AP frames. */
handle_pspoll(local_info_t * local,struct ieee80211_hdr * hdr,struct hostap_80211_rx_status * rx_stats)1861 static void handle_pspoll(local_info_t *local,
1862 			  struct ieee80211_hdr *hdr,
1863 			  struct hostap_80211_rx_status *rx_stats)
1864 {
1865 	struct net_device *dev = local->dev;
1866 	struct sta_info *sta;
1867 	u16 aid;
1868 	struct sk_buff *skb;
1869 
1870 	PDEBUG(DEBUG_PS2, "handle_pspoll: BSSID=%pM, TA=%pM PWRMGT=%d\n",
1871 	       hdr->addr1, hdr->addr2, !!ieee80211_has_pm(hdr->frame_control));
1872 
1873 	if (memcmp(hdr->addr1, dev->dev_addr, ETH_ALEN)) {
1874 		PDEBUG(DEBUG_AP,
1875 		       "handle_pspoll - addr1(BSSID)=%pM not own MAC\n",
1876 		       hdr->addr1);
1877 		return;
1878 	}
1879 
1880 	aid = le16_to_cpu(hdr->duration_id);
1881 	if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14))) {
1882 		PDEBUG(DEBUG_PS, "   PSPOLL and AID[15:14] not set\n");
1883 		return;
1884 	}
1885 	aid &= ~(BIT(15) | BIT(14));
1886 	if (aid == 0 || aid > MAX_AID_TABLE_SIZE) {
1887 		PDEBUG(DEBUG_PS, "   invalid aid=%d\n", aid);
1888 		return;
1889 	}
1890 	PDEBUG(DEBUG_PS2, "   aid=%d\n", aid);
1891 
1892 	spin_lock_bh(&local->ap->sta_table_lock);
1893 	sta = ap_get_sta(local->ap, hdr->addr2);
1894 	if (sta)
1895 		atomic_inc(&sta->users);
1896 	spin_unlock_bh(&local->ap->sta_table_lock);
1897 
1898 	if (sta == NULL) {
1899 		PDEBUG(DEBUG_PS, "   STA not found\n");
1900 		return;
1901 	}
1902 	if (sta->aid != aid) {
1903 		PDEBUG(DEBUG_PS, "   received aid=%i does not match with "
1904 		       "assoc.aid=%d\n", aid, sta->aid);
1905 		return;
1906 	}
1907 
1908 	/* FIX: todo:
1909 	 * - add timeout for buffering (clear aid in TIM vector if buffer timed
1910 	 *   out (expiry time must be longer than ListenInterval for
1911 	 *   the corresponding STA; "8802-11: 11.2.1.9 AP aging function"
1912 	 * - what to do, if buffered, pspolled, and sent frame is not ACKed by
1913 	 *   sta; store buffer for later use and leave TIM aid bit set? use
1914 	 *   TX event to check whether frame was ACKed?
1915 	 */
1916 
1917 	while ((skb = skb_dequeue(&sta->tx_buf)) != NULL) {
1918 		/* send buffered frame .. */
1919 		PDEBUG(DEBUG_PS2, "Sending buffered frame to STA after PS POLL"
1920 		       " (buffer_count=%d)\n", skb_queue_len(&sta->tx_buf));
1921 
1922 		pspoll_send_buffered(local, sta, skb);
1923 
1924 		if (sta->flags & WLAN_STA_PS) {
1925 			/* send only one buffered packet per PS Poll */
1926 			/* FIX: should ignore further PS Polls until the
1927 			 * buffered packet that was just sent is acknowledged
1928 			 * (Tx or TxExc event) */
1929 			break;
1930 		}
1931 	}
1932 
1933 	if (skb_queue_empty(&sta->tx_buf)) {
1934 		/* try to clear aid from TIM */
1935 		if (!(sta->flags & WLAN_STA_TIM))
1936 			PDEBUG(DEBUG_PS2,  "Re-unsetting TIM for aid %d\n",
1937 			       aid);
1938 		hostap_set_tim(local, aid, 0);
1939 		sta->flags &= ~WLAN_STA_TIM;
1940 	}
1941 
1942 	atomic_dec(&sta->users);
1943 }
1944 
1945 
1946 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1947 
handle_wds_oper_queue(struct work_struct * work)1948 static void handle_wds_oper_queue(struct work_struct *work)
1949 {
1950 	struct ap_data *ap = container_of(work, struct ap_data,
1951 					  wds_oper_queue);
1952 	local_info_t *local = ap->local;
1953 	struct wds_oper_data *entry, *prev;
1954 
1955 	spin_lock_bh(&local->lock);
1956 	entry = local->ap->wds_oper_entries;
1957 	local->ap->wds_oper_entries = NULL;
1958 	spin_unlock_bh(&local->lock);
1959 
1960 	while (entry) {
1961 		PDEBUG(DEBUG_AP, "%s: %s automatic WDS connection "
1962 		       "to AP %pM\n",
1963 		       local->dev->name,
1964 		       entry->type == WDS_ADD ? "adding" : "removing",
1965 		       entry->addr);
1966 		if (entry->type == WDS_ADD)
1967 			prism2_wds_add(local, entry->addr, 0);
1968 		else if (entry->type == WDS_DEL)
1969 			prism2_wds_del(local, entry->addr, 0, 1);
1970 
1971 		prev = entry;
1972 		entry = entry->next;
1973 		kfree(prev);
1974 	}
1975 }
1976 
1977 
1978 /* Called only as a scheduled task for pending AP frames. */
handle_beacon(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)1979 static void handle_beacon(local_info_t *local, struct sk_buff *skb,
1980 			  struct hostap_80211_rx_status *rx_stats)
1981 {
1982 	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1983 	char *body = skb->data + IEEE80211_MGMT_HDR_LEN;
1984 	int len, left;
1985 	u16 beacon_int, capability;
1986 	__le16 *pos;
1987 	char *ssid = NULL;
1988 	unsigned char *supp_rates = NULL;
1989 	int ssid_len = 0, supp_rates_len = 0;
1990 	struct sta_info *sta = NULL;
1991 	int new_sta = 0, channel = -1;
1992 
1993 	len = skb->len - IEEE80211_MGMT_HDR_LEN;
1994 
1995 	if (len < 8 + 2 + 2) {
1996 		printk(KERN_DEBUG "handle_beacon - too short payload "
1997 		       "(len=%d)\n", len);
1998 		return;
1999 	}
2000 
2001 	pos = (__le16 *) body;
2002 	left = len;
2003 
2004 	/* Timestamp (8 octets) */
2005 	pos += 4; left -= 8;
2006 	/* Beacon interval (2 octets) */
2007 	beacon_int = le16_to_cpu(*pos);
2008 	pos++; left -= 2;
2009 	/* Capability information (2 octets) */
2010 	capability = le16_to_cpu(*pos);
2011 	pos++; left -= 2;
2012 
2013 	if (local->ap->ap_policy != AP_OTHER_AP_EVEN_IBSS &&
2014 	    capability & WLAN_CAPABILITY_IBSS)
2015 		return;
2016 
2017 	if (left >= 2) {
2018 		unsigned int ileft;
2019 		unsigned char *u = (unsigned char *) pos;
2020 
2021 		if (*u == WLAN_EID_SSID) {
2022 			u++; left--;
2023 			ileft = *u;
2024 			u++; left--;
2025 
2026 			if (ileft > left || ileft > MAX_SSID_LEN) {
2027 				PDEBUG(DEBUG_AP, "SSID: overflow\n");
2028 				return;
2029 			}
2030 
2031 			if (local->ap->ap_policy == AP_OTHER_AP_SAME_SSID &&
2032 			    (ileft != strlen(local->essid) ||
2033 			     memcmp(local->essid, u, ileft) != 0)) {
2034 				/* not our SSID */
2035 				return;
2036 			}
2037 
2038 			ssid = u;
2039 			ssid_len = ileft;
2040 
2041 			u += ileft;
2042 			left -= ileft;
2043 		}
2044 
2045 		if (*u == WLAN_EID_SUPP_RATES) {
2046 			u++; left--;
2047 			ileft = *u;
2048 			u++; left--;
2049 
2050 			if (ileft > left || ileft == 0 || ileft > 8) {
2051 				PDEBUG(DEBUG_AP, " - SUPP_RATES len error\n");
2052 				return;
2053 			}
2054 
2055 			supp_rates = u;
2056 			supp_rates_len = ileft;
2057 
2058 			u += ileft;
2059 			left -= ileft;
2060 		}
2061 
2062 		if (*u == WLAN_EID_DS_PARAMS) {
2063 			u++; left--;
2064 			ileft = *u;
2065 			u++; left--;
2066 
2067 			if (ileft > left || ileft != 1) {
2068 				PDEBUG(DEBUG_AP, " - DS_PARAMS len error\n");
2069 				return;
2070 			}
2071 
2072 			channel = *u;
2073 
2074 			u += ileft;
2075 			left -= ileft;
2076 		}
2077 	}
2078 
2079 	spin_lock_bh(&local->ap->sta_table_lock);
2080 	sta = ap_get_sta(local->ap, hdr->addr2);
2081 	if (sta != NULL)
2082 		atomic_inc(&sta->users);
2083 	spin_unlock_bh(&local->ap->sta_table_lock);
2084 
2085 	if (sta == NULL) {
2086 		/* add new AP */
2087 		new_sta = 1;
2088 		sta = ap_add_sta(local->ap, hdr->addr2);
2089 		if (sta == NULL) {
2090 			printk(KERN_INFO "prism2: kmalloc failed for AP "
2091 			       "data structure\n");
2092 			return;
2093 		}
2094 		hostap_event_new_sta(local->dev, sta);
2095 
2096 		/* mark APs authentication and associated for pseudo ad-hoc
2097 		 * style communication */
2098 		sta->flags = WLAN_STA_AUTH | WLAN_STA_ASSOC;
2099 
2100 		if (local->ap->autom_ap_wds) {
2101 			hostap_wds_link_oper(local, sta->addr, WDS_ADD);
2102 		}
2103 	}
2104 
2105 	sta->ap = 1;
2106 	if (ssid) {
2107 		sta->u.ap.ssid_len = ssid_len;
2108 		memcpy(sta->u.ap.ssid, ssid, ssid_len);
2109 		sta->u.ap.ssid[ssid_len] = '\0';
2110 	} else {
2111 		sta->u.ap.ssid_len = 0;
2112 		sta->u.ap.ssid[0] = '\0';
2113 	}
2114 	sta->u.ap.channel = channel;
2115 	sta->rx_packets++;
2116 	sta->rx_bytes += len;
2117 	sta->u.ap.last_beacon = sta->last_rx = jiffies;
2118 	sta->capability = capability;
2119 	sta->listen_interval = beacon_int;
2120 
2121 	atomic_dec(&sta->users);
2122 
2123 	if (new_sta) {
2124 		memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
2125 		memcpy(sta->supported_rates, supp_rates, supp_rates_len);
2126 		prism2_check_tx_rates(sta);
2127 	}
2128 }
2129 
2130 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2131 
2132 
2133 /* Called only as a tasklet. */
handle_ap_item(local_info_t * local,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)2134 static void handle_ap_item(local_info_t *local, struct sk_buff *skb,
2135 			   struct hostap_80211_rx_status *rx_stats)
2136 {
2137 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2138 	struct net_device *dev = local->dev;
2139 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2140 	u16 fc, type, stype;
2141 	struct ieee80211_hdr *hdr;
2142 
2143 	/* FIX: should give skb->len to handler functions and check that the
2144 	 * buffer is long enough */
2145 	hdr = (struct ieee80211_hdr *) skb->data;
2146 	fc = le16_to_cpu(hdr->frame_control);
2147 	type = fc & IEEE80211_FCTL_FTYPE;
2148 	stype = fc & IEEE80211_FCTL_STYPE;
2149 
2150 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2151 	if (!local->hostapd && type == IEEE80211_FTYPE_DATA) {
2152 		PDEBUG(DEBUG_AP, "handle_ap_item - data frame\n");
2153 
2154 		if (!(fc & IEEE80211_FCTL_TODS) ||
2155 		    (fc & IEEE80211_FCTL_FROMDS)) {
2156 			if (stype == IEEE80211_STYPE_NULLFUNC) {
2157 				/* no ToDS nullfunc seems to be used to check
2158 				 * AP association; so send reject message to
2159 				 * speed up re-association */
2160 				ap_handle_dropped_data(local, hdr);
2161 				goto done;
2162 			}
2163 			PDEBUG(DEBUG_AP, "   not ToDS frame (fc=0x%04x)\n",
2164 			       fc);
2165 			goto done;
2166 		}
2167 
2168 		if (memcmp(hdr->addr1, dev->dev_addr, ETH_ALEN)) {
2169 			PDEBUG(DEBUG_AP, "handle_ap_item - addr1(BSSID)=%pM"
2170 			       " not own MAC\n", hdr->addr1);
2171 			goto done;
2172 		}
2173 
2174 		if (local->ap->nullfunc_ack &&
2175 		    stype == IEEE80211_STYPE_NULLFUNC)
2176 			ap_handle_data_nullfunc(local, hdr);
2177 		else
2178 			ap_handle_dropped_data(local, hdr);
2179 		goto done;
2180 	}
2181 
2182 	if (type == IEEE80211_FTYPE_MGMT && stype == IEEE80211_STYPE_BEACON) {
2183 		handle_beacon(local, skb, rx_stats);
2184 		goto done;
2185 	}
2186 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2187 
2188 	if (type == IEEE80211_FTYPE_CTL && stype == IEEE80211_STYPE_PSPOLL) {
2189 		handle_pspoll(local, hdr, rx_stats);
2190 		goto done;
2191 	}
2192 
2193 	if (local->hostapd) {
2194 		PDEBUG(DEBUG_AP, "Unknown frame in AP queue: type=0x%02x "
2195 		       "subtype=0x%02x\n", type, stype);
2196 		goto done;
2197 	}
2198 
2199 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2200 	if (type != IEEE80211_FTYPE_MGMT) {
2201 		PDEBUG(DEBUG_AP, "handle_ap_item - not a management frame?\n");
2202 		goto done;
2203 	}
2204 
2205 	if (memcmp(hdr->addr1, dev->dev_addr, ETH_ALEN)) {
2206 		PDEBUG(DEBUG_AP, "handle_ap_item - addr1(DA)=%pM"
2207 		       " not own MAC\n", hdr->addr1);
2208 		goto done;
2209 	}
2210 
2211 	if (memcmp(hdr->addr3, dev->dev_addr, ETH_ALEN)) {
2212 		PDEBUG(DEBUG_AP, "handle_ap_item - addr3(BSSID)=%pM"
2213 		       " not own MAC\n", hdr->addr3);
2214 		goto done;
2215 	}
2216 
2217 	switch (stype) {
2218 	case IEEE80211_STYPE_ASSOC_REQ:
2219 		handle_assoc(local, skb, rx_stats, 0);
2220 		break;
2221 	case IEEE80211_STYPE_ASSOC_RESP:
2222 		PDEBUG(DEBUG_AP, "==> ASSOC RESP (ignored)\n");
2223 		break;
2224 	case IEEE80211_STYPE_REASSOC_REQ:
2225 		handle_assoc(local, skb, rx_stats, 1);
2226 		break;
2227 	case IEEE80211_STYPE_REASSOC_RESP:
2228 		PDEBUG(DEBUG_AP, "==> REASSOC RESP (ignored)\n");
2229 		break;
2230 	case IEEE80211_STYPE_ATIM:
2231 		PDEBUG(DEBUG_AP, "==> ATIM (ignored)\n");
2232 		break;
2233 	case IEEE80211_STYPE_DISASSOC:
2234 		handle_disassoc(local, skb, rx_stats);
2235 		break;
2236 	case IEEE80211_STYPE_AUTH:
2237 		handle_authen(local, skb, rx_stats);
2238 		break;
2239 	case IEEE80211_STYPE_DEAUTH:
2240 		handle_deauth(local, skb, rx_stats);
2241 		break;
2242 	default:
2243 		PDEBUG(DEBUG_AP, "Unknown mgmt frame subtype 0x%02x\n",
2244 		       stype >> 4);
2245 		break;
2246 	}
2247 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2248 
2249  done:
2250 	dev_kfree_skb(skb);
2251 }
2252 
2253 
2254 /* Called only as a tasklet (software IRQ) */
hostap_rx(struct net_device * dev,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats)2255 void hostap_rx(struct net_device *dev, struct sk_buff *skb,
2256 	       struct hostap_80211_rx_status *rx_stats)
2257 {
2258 	struct hostap_interface *iface;
2259 	local_info_t *local;
2260 	struct ieee80211_hdr *hdr;
2261 
2262 	iface = netdev_priv(dev);
2263 	local = iface->local;
2264 
2265 	if (skb->len < 16)
2266 		goto drop;
2267 
2268 	dev->stats.rx_packets++;
2269 
2270 	hdr = (struct ieee80211_hdr *) skb->data;
2271 
2272 	if (local->ap->ap_policy == AP_OTHER_AP_SKIP_ALL &&
2273 	    ieee80211_is_beacon(hdr->frame_control))
2274 		goto drop;
2275 
2276 	skb->protocol = cpu_to_be16(ETH_P_HOSTAP);
2277 	handle_ap_item(local, skb, rx_stats);
2278 	return;
2279 
2280  drop:
2281 	dev_kfree_skb(skb);
2282 }
2283 
2284 
2285 /* Called only as a tasklet (software IRQ) */
schedule_packet_send(local_info_t * local,struct sta_info * sta)2286 static void schedule_packet_send(local_info_t *local, struct sta_info *sta)
2287 {
2288 	struct sk_buff *skb;
2289 	struct ieee80211_hdr *hdr;
2290 	struct hostap_80211_rx_status rx_stats;
2291 
2292 	if (skb_queue_empty(&sta->tx_buf))
2293 		return;
2294 
2295 	skb = dev_alloc_skb(16);
2296 	if (skb == NULL) {
2297 		printk(KERN_DEBUG "%s: schedule_packet_send: skb alloc "
2298 		       "failed\n", local->dev->name);
2299 		return;
2300 	}
2301 
2302 	hdr = (struct ieee80211_hdr *) skb_put(skb, 16);
2303 
2304 	/* Generate a fake pspoll frame to start packet delivery */
2305 	hdr->frame_control = cpu_to_le16(
2306 		IEEE80211_FTYPE_CTL | IEEE80211_STYPE_PSPOLL);
2307 	memcpy(hdr->addr1, local->dev->dev_addr, ETH_ALEN);
2308 	memcpy(hdr->addr2, sta->addr, ETH_ALEN);
2309 	hdr->duration_id = cpu_to_le16(sta->aid | BIT(15) | BIT(14));
2310 
2311 	PDEBUG(DEBUG_PS2,
2312 	       "%s: Scheduling buffered packet delivery for STA %pM\n",
2313 	       local->dev->name, sta->addr);
2314 
2315 	skb->dev = local->dev;
2316 
2317 	memset(&rx_stats, 0, sizeof(rx_stats));
2318 	hostap_rx(local->dev, skb, &rx_stats);
2319 }
2320 
2321 
prism2_ap_get_sta_qual(local_info_t * local,struct sockaddr addr[],struct iw_quality qual[],int buf_size,int aplist)2322 int prism2_ap_get_sta_qual(local_info_t *local, struct sockaddr addr[],
2323 			   struct iw_quality qual[], int buf_size,
2324 			   int aplist)
2325 {
2326 	struct ap_data *ap = local->ap;
2327 	struct list_head *ptr;
2328 	int count = 0;
2329 
2330 	spin_lock_bh(&ap->sta_table_lock);
2331 
2332 	for (ptr = ap->sta_list.next; ptr != NULL && ptr != &ap->sta_list;
2333 	     ptr = ptr->next) {
2334 		struct sta_info *sta = (struct sta_info *) ptr;
2335 
2336 		if (aplist && !sta->ap)
2337 			continue;
2338 		addr[count].sa_family = ARPHRD_ETHER;
2339 		memcpy(addr[count].sa_data, sta->addr, ETH_ALEN);
2340 		if (sta->last_rx_silence == 0)
2341 			qual[count].qual = sta->last_rx_signal < 27 ?
2342 				0 : (sta->last_rx_signal - 27) * 92 / 127;
2343 		else
2344 			qual[count].qual = sta->last_rx_signal -
2345 				sta->last_rx_silence - 35;
2346 		qual[count].level = HFA384X_LEVEL_TO_dBm(sta->last_rx_signal);
2347 		qual[count].noise = HFA384X_LEVEL_TO_dBm(sta->last_rx_silence);
2348 		qual[count].updated = sta->last_rx_updated;
2349 
2350 		sta->last_rx_updated = IW_QUAL_DBM;
2351 
2352 		count++;
2353 		if (count >= buf_size)
2354 			break;
2355 	}
2356 	spin_unlock_bh(&ap->sta_table_lock);
2357 
2358 	return count;
2359 }
2360 
2361 
2362 /* Translate our list of Access Points & Stations to a card independent
2363  * format that the Wireless Tools will understand - Jean II */
prism2_ap_translate_scan(struct net_device * dev,struct iw_request_info * info,char * buffer)2364 int prism2_ap_translate_scan(struct net_device *dev,
2365 			     struct iw_request_info *info, char *buffer)
2366 {
2367 	struct hostap_interface *iface;
2368 	local_info_t *local;
2369 	struct ap_data *ap;
2370 	struct list_head *ptr;
2371 	struct iw_event iwe;
2372 	char *current_ev = buffer;
2373 	char *end_buf = buffer + IW_SCAN_MAX_DATA;
2374 #if !defined(PRISM2_NO_KERNEL_IEEE80211_MGMT)
2375 	char buf[64];
2376 #endif
2377 
2378 	iface = netdev_priv(dev);
2379 	local = iface->local;
2380 	ap = local->ap;
2381 
2382 	spin_lock_bh(&ap->sta_table_lock);
2383 
2384 	for (ptr = ap->sta_list.next; ptr != NULL && ptr != &ap->sta_list;
2385 	     ptr = ptr->next) {
2386 		struct sta_info *sta = (struct sta_info *) ptr;
2387 
2388 		/* First entry *MUST* be the AP MAC address */
2389 		memset(&iwe, 0, sizeof(iwe));
2390 		iwe.cmd = SIOCGIWAP;
2391 		iwe.u.ap_addr.sa_family = ARPHRD_ETHER;
2392 		memcpy(iwe.u.ap_addr.sa_data, sta->addr, ETH_ALEN);
2393 		iwe.len = IW_EV_ADDR_LEN;
2394 		current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2395 						  &iwe, IW_EV_ADDR_LEN);
2396 
2397 		/* Use the mode to indicate if it's a station or
2398 		 * an Access Point */
2399 		memset(&iwe, 0, sizeof(iwe));
2400 		iwe.cmd = SIOCGIWMODE;
2401 		if (sta->ap)
2402 			iwe.u.mode = IW_MODE_MASTER;
2403 		else
2404 			iwe.u.mode = IW_MODE_INFRA;
2405 		iwe.len = IW_EV_UINT_LEN;
2406 		current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2407 						  &iwe, IW_EV_UINT_LEN);
2408 
2409 		/* Some quality */
2410 		memset(&iwe, 0, sizeof(iwe));
2411 		iwe.cmd = IWEVQUAL;
2412 		if (sta->last_rx_silence == 0)
2413 			iwe.u.qual.qual = sta->last_rx_signal < 27 ?
2414 				0 : (sta->last_rx_signal - 27) * 92 / 127;
2415 		else
2416 			iwe.u.qual.qual = sta->last_rx_signal -
2417 				sta->last_rx_silence - 35;
2418 		iwe.u.qual.level = HFA384X_LEVEL_TO_dBm(sta->last_rx_signal);
2419 		iwe.u.qual.noise = HFA384X_LEVEL_TO_dBm(sta->last_rx_silence);
2420 		iwe.u.qual.updated = sta->last_rx_updated;
2421 		iwe.len = IW_EV_QUAL_LEN;
2422 		current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2423 						  &iwe, IW_EV_QUAL_LEN);
2424 
2425 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2426 		if (sta->ap) {
2427 			memset(&iwe, 0, sizeof(iwe));
2428 			iwe.cmd = SIOCGIWESSID;
2429 			iwe.u.data.length = sta->u.ap.ssid_len;
2430 			iwe.u.data.flags = 1;
2431 			current_ev = iwe_stream_add_point(info, current_ev,
2432 							  end_buf, &iwe,
2433 							  sta->u.ap.ssid);
2434 
2435 			memset(&iwe, 0, sizeof(iwe));
2436 			iwe.cmd = SIOCGIWENCODE;
2437 			if (sta->capability & WLAN_CAPABILITY_PRIVACY)
2438 				iwe.u.data.flags =
2439 					IW_ENCODE_ENABLED | IW_ENCODE_NOKEY;
2440 			else
2441 				iwe.u.data.flags = IW_ENCODE_DISABLED;
2442 			current_ev = iwe_stream_add_point(info, current_ev,
2443 							  end_buf, &iwe,
2444 							  sta->u.ap.ssid);
2445 
2446 			if (sta->u.ap.channel > 0 &&
2447 			    sta->u.ap.channel <= FREQ_COUNT) {
2448 				memset(&iwe, 0, sizeof(iwe));
2449 				iwe.cmd = SIOCGIWFREQ;
2450 				iwe.u.freq.m = freq_list[sta->u.ap.channel - 1]
2451 					* 100000;
2452 				iwe.u.freq.e = 1;
2453 				current_ev = iwe_stream_add_event(
2454 					info, current_ev, end_buf, &iwe,
2455 					IW_EV_FREQ_LEN);
2456 			}
2457 
2458 			memset(&iwe, 0, sizeof(iwe));
2459 			iwe.cmd = IWEVCUSTOM;
2460 			sprintf(buf, "beacon_interval=%d",
2461 				sta->listen_interval);
2462 			iwe.u.data.length = strlen(buf);
2463 			current_ev = iwe_stream_add_point(info, current_ev,
2464 							  end_buf, &iwe, buf);
2465 		}
2466 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2467 
2468 		sta->last_rx_updated = IW_QUAL_DBM;
2469 
2470 		/* To be continued, we should make good use of IWEVCUSTOM */
2471 	}
2472 
2473 	spin_unlock_bh(&ap->sta_table_lock);
2474 
2475 	return current_ev - buffer;
2476 }
2477 
2478 
prism2_hostapd_add_sta(struct ap_data * ap,struct prism2_hostapd_param * param)2479 static int prism2_hostapd_add_sta(struct ap_data *ap,
2480 				  struct prism2_hostapd_param *param)
2481 {
2482 	struct sta_info *sta;
2483 
2484 	spin_lock_bh(&ap->sta_table_lock);
2485 	sta = ap_get_sta(ap, param->sta_addr);
2486 	if (sta)
2487 		atomic_inc(&sta->users);
2488 	spin_unlock_bh(&ap->sta_table_lock);
2489 
2490 	if (sta == NULL) {
2491 		sta = ap_add_sta(ap, param->sta_addr);
2492 		if (sta == NULL)
2493 			return -1;
2494 	}
2495 
2496 	if (!(sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
2497 		hostap_event_new_sta(sta->local->dev, sta);
2498 
2499 	sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC;
2500 	sta->last_rx = jiffies;
2501 	sta->aid = param->u.add_sta.aid;
2502 	sta->capability = param->u.add_sta.capability;
2503 	sta->tx_supp_rates = param->u.add_sta.tx_supp_rates;
2504 	if (sta->tx_supp_rates & WLAN_RATE_1M)
2505 		sta->supported_rates[0] = 2;
2506 	if (sta->tx_supp_rates & WLAN_RATE_2M)
2507 		sta->supported_rates[1] = 4;
2508  	if (sta->tx_supp_rates & WLAN_RATE_5M5)
2509 		sta->supported_rates[2] = 11;
2510 	if (sta->tx_supp_rates & WLAN_RATE_11M)
2511 		sta->supported_rates[3] = 22;
2512 	prism2_check_tx_rates(sta);
2513 	atomic_dec(&sta->users);
2514 	return 0;
2515 }
2516 
2517 
prism2_hostapd_remove_sta(struct ap_data * ap,struct prism2_hostapd_param * param)2518 static int prism2_hostapd_remove_sta(struct ap_data *ap,
2519 				     struct prism2_hostapd_param *param)
2520 {
2521 	struct sta_info *sta;
2522 
2523 	spin_lock_bh(&ap->sta_table_lock);
2524 	sta = ap_get_sta(ap, param->sta_addr);
2525 	if (sta) {
2526 		ap_sta_hash_del(ap, sta);
2527 		list_del(&sta->list);
2528 	}
2529 	spin_unlock_bh(&ap->sta_table_lock);
2530 
2531 	if (!sta)
2532 		return -ENOENT;
2533 
2534 	if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
2535 		hostap_event_expired_sta(sta->local->dev, sta);
2536 	ap_free_sta(ap, sta);
2537 
2538 	return 0;
2539 }
2540 
2541 
prism2_hostapd_get_info_sta(struct ap_data * ap,struct prism2_hostapd_param * param)2542 static int prism2_hostapd_get_info_sta(struct ap_data *ap,
2543 				       struct prism2_hostapd_param *param)
2544 {
2545 	struct sta_info *sta;
2546 
2547 	spin_lock_bh(&ap->sta_table_lock);
2548 	sta = ap_get_sta(ap, param->sta_addr);
2549 	if (sta)
2550 		atomic_inc(&sta->users);
2551 	spin_unlock_bh(&ap->sta_table_lock);
2552 
2553 	if (!sta)
2554 		return -ENOENT;
2555 
2556 	param->u.get_info_sta.inactive_sec = (jiffies - sta->last_rx) / HZ;
2557 
2558 	atomic_dec(&sta->users);
2559 
2560 	return 1;
2561 }
2562 
2563 
prism2_hostapd_set_flags_sta(struct ap_data * ap,struct prism2_hostapd_param * param)2564 static int prism2_hostapd_set_flags_sta(struct ap_data *ap,
2565 					struct prism2_hostapd_param *param)
2566 {
2567 	struct sta_info *sta;
2568 
2569 	spin_lock_bh(&ap->sta_table_lock);
2570 	sta = ap_get_sta(ap, param->sta_addr);
2571 	if (sta) {
2572 		sta->flags |= param->u.set_flags_sta.flags_or;
2573 		sta->flags &= param->u.set_flags_sta.flags_and;
2574 	}
2575 	spin_unlock_bh(&ap->sta_table_lock);
2576 
2577 	if (!sta)
2578 		return -ENOENT;
2579 
2580 	return 0;
2581 }
2582 
2583 
prism2_hostapd_sta_clear_stats(struct ap_data * ap,struct prism2_hostapd_param * param)2584 static int prism2_hostapd_sta_clear_stats(struct ap_data *ap,
2585 					  struct prism2_hostapd_param *param)
2586 {
2587 	struct sta_info *sta;
2588 	int rate;
2589 
2590 	spin_lock_bh(&ap->sta_table_lock);
2591 	sta = ap_get_sta(ap, param->sta_addr);
2592 	if (sta) {
2593 		sta->rx_packets = sta->tx_packets = 0;
2594 		sta->rx_bytes = sta->tx_bytes = 0;
2595 		for (rate = 0; rate < WLAN_RATE_COUNT; rate++) {
2596 			sta->tx_count[rate] = 0;
2597 			sta->rx_count[rate] = 0;
2598 		}
2599 	}
2600 	spin_unlock_bh(&ap->sta_table_lock);
2601 
2602 	if (!sta)
2603 		return -ENOENT;
2604 
2605 	return 0;
2606 }
2607 
2608 
prism2_hostapd(struct ap_data * ap,struct prism2_hostapd_param * param)2609 int prism2_hostapd(struct ap_data *ap, struct prism2_hostapd_param *param)
2610 {
2611 	switch (param->cmd) {
2612 	case PRISM2_HOSTAPD_FLUSH:
2613 		ap_control_kickall(ap);
2614 		return 0;
2615 	case PRISM2_HOSTAPD_ADD_STA:
2616 		return prism2_hostapd_add_sta(ap, param);
2617 	case PRISM2_HOSTAPD_REMOVE_STA:
2618 		return prism2_hostapd_remove_sta(ap, param);
2619 	case PRISM2_HOSTAPD_GET_INFO_STA:
2620 		return prism2_hostapd_get_info_sta(ap, param);
2621 	case PRISM2_HOSTAPD_SET_FLAGS_STA:
2622 		return prism2_hostapd_set_flags_sta(ap, param);
2623 	case PRISM2_HOSTAPD_STA_CLEAR_STATS:
2624 		return prism2_hostapd_sta_clear_stats(ap, param);
2625 	default:
2626 		printk(KERN_WARNING "prism2_hostapd: unknown cmd=%d\n",
2627 		       param->cmd);
2628 		return -EOPNOTSUPP;
2629 	}
2630 }
2631 
2632 
2633 /* Update station info for host-based TX rate control and return current
2634  * TX rate */
ap_update_sta_tx_rate(struct sta_info * sta,struct net_device * dev)2635 static int ap_update_sta_tx_rate(struct sta_info *sta, struct net_device *dev)
2636 {
2637 	int ret = sta->tx_rate;
2638 	struct hostap_interface *iface;
2639 	local_info_t *local;
2640 
2641 	iface = netdev_priv(dev);
2642 	local = iface->local;
2643 
2644 	sta->tx_count[sta->tx_rate_idx]++;
2645 	sta->tx_since_last_failure++;
2646 	sta->tx_consecutive_exc = 0;
2647 	if (sta->tx_since_last_failure >= WLAN_RATE_UPDATE_COUNT &&
2648 	    sta->tx_rate_idx < sta->tx_max_rate) {
2649 		/* use next higher rate */
2650 		int old_rate, new_rate;
2651 		old_rate = new_rate = sta->tx_rate_idx;
2652 		while (new_rate < sta->tx_max_rate) {
2653 			new_rate++;
2654 			if (ap_tx_rate_ok(new_rate, sta, local)) {
2655 				sta->tx_rate_idx = new_rate;
2656 				break;
2657 			}
2658 		}
2659 		if (old_rate != sta->tx_rate_idx) {
2660 			switch (sta->tx_rate_idx) {
2661 			case 0: sta->tx_rate = 10; break;
2662 			case 1: sta->tx_rate = 20; break;
2663 			case 2: sta->tx_rate = 55; break;
2664 			case 3: sta->tx_rate = 110; break;
2665 			default: sta->tx_rate = 0; break;
2666 			}
2667 			PDEBUG(DEBUG_AP, "%s: STA %pM TX rate raised to %d\n",
2668 			       dev->name, sta->addr, sta->tx_rate);
2669 		}
2670 		sta->tx_since_last_failure = 0;
2671 	}
2672 
2673 	return ret;
2674 }
2675 
2676 
2677 /* Called only from software IRQ. Called for each TX frame prior possible
2678  * encryption and transmit. */
hostap_handle_sta_tx(local_info_t * local,struct hostap_tx_data * tx)2679 ap_tx_ret hostap_handle_sta_tx(local_info_t *local, struct hostap_tx_data *tx)
2680 {
2681 	struct sta_info *sta = NULL;
2682 	struct sk_buff *skb = tx->skb;
2683 	int set_tim, ret;
2684 	struct ieee80211_hdr *hdr;
2685 	struct hostap_skb_tx_data *meta;
2686 
2687 	meta = (struct hostap_skb_tx_data *) skb->cb;
2688 	ret = AP_TX_CONTINUE;
2689 	if (local->ap == NULL || skb->len < 10 ||
2690 	    meta->iface->type == HOSTAP_INTERFACE_STA)
2691 		goto out;
2692 
2693 	hdr = (struct ieee80211_hdr *) skb->data;
2694 
2695 	if (hdr->addr1[0] & 0x01) {
2696 		/* broadcast/multicast frame - no AP related processing */
2697 		if (local->ap->num_sta <= 0)
2698 			ret = AP_TX_DROP;
2699 		goto out;
2700 	}
2701 
2702 	/* unicast packet - check whether destination STA is associated */
2703 	spin_lock(&local->ap->sta_table_lock);
2704 	sta = ap_get_sta(local->ap, hdr->addr1);
2705 	if (sta)
2706 		atomic_inc(&sta->users);
2707 	spin_unlock(&local->ap->sta_table_lock);
2708 
2709 	if (local->iw_mode == IW_MODE_MASTER && sta == NULL &&
2710 	    !(meta->flags & HOSTAP_TX_FLAGS_WDS) &&
2711 	    meta->iface->type != HOSTAP_INTERFACE_MASTER &&
2712 	    meta->iface->type != HOSTAP_INTERFACE_AP) {
2713 #if 0
2714 		/* This can happen, e.g., when wlan0 is added to a bridge and
2715 		 * bridging code does not know which port is the correct target
2716 		 * for a unicast frame. In this case, the packet is send to all
2717 		 * ports of the bridge. Since this is a valid scenario, do not
2718 		 * print out any errors here. */
2719 		if (net_ratelimit()) {
2720 			printk(KERN_DEBUG "AP: drop packet to non-associated "
2721 			       "STA %pM\n", hdr->addr1);
2722 		}
2723 #endif
2724 		local->ap->tx_drop_nonassoc++;
2725 		ret = AP_TX_DROP;
2726 		goto out;
2727 	}
2728 
2729 	if (sta == NULL)
2730 		goto out;
2731 
2732 	if (!(sta->flags & WLAN_STA_AUTHORIZED))
2733 		ret = AP_TX_CONTINUE_NOT_AUTHORIZED;
2734 
2735 	/* Set tx_rate if using host-based TX rate control */
2736 	if (!local->fw_tx_rate_control)
2737 		local->ap->last_tx_rate = meta->rate =
2738 			ap_update_sta_tx_rate(sta, local->dev);
2739 
2740 	if (local->iw_mode != IW_MODE_MASTER)
2741 		goto out;
2742 
2743 	if (!(sta->flags & WLAN_STA_PS))
2744 		goto out;
2745 
2746 	if (meta->flags & HOSTAP_TX_FLAGS_ADD_MOREDATA) {
2747 		/* indicate to STA that more frames follow */
2748 		hdr->frame_control |=
2749 			cpu_to_le16(IEEE80211_FCTL_MOREDATA);
2750 	}
2751 
2752 	if (meta->flags & HOSTAP_TX_FLAGS_BUFFERED_FRAME) {
2753 		/* packet was already buffered and now send due to
2754 		 * PS poll, so do not rebuffer it */
2755 		goto out;
2756 	}
2757 
2758 	if (skb_queue_len(&sta->tx_buf) >= STA_MAX_TX_BUFFER) {
2759 		PDEBUG(DEBUG_PS, "%s: No more space in STA (%pM)'s"
2760 		       "PS mode buffer\n",
2761 		       local->dev->name, sta->addr);
2762 		/* Make sure that TIM is set for the station (it might not be
2763 		 * after AP wlan hw reset). */
2764 		/* FIX: should fix hw reset to restore bits based on STA
2765 		 * buffer state.. */
2766 		hostap_set_tim(local, sta->aid, 1);
2767 		sta->flags |= WLAN_STA_TIM;
2768 		ret = AP_TX_DROP;
2769 		goto out;
2770 	}
2771 
2772 	/* STA in PS mode, buffer frame for later delivery */
2773 	set_tim = skb_queue_empty(&sta->tx_buf);
2774 	skb_queue_tail(&sta->tx_buf, skb);
2775 	/* FIX: could save RX time to skb and expire buffered frames after
2776 	 * some time if STA does not poll for them */
2777 
2778 	if (set_tim) {
2779 		if (sta->flags & WLAN_STA_TIM)
2780 			PDEBUG(DEBUG_PS2, "Re-setting TIM for aid %d\n",
2781 			       sta->aid);
2782 		hostap_set_tim(local, sta->aid, 1);
2783 		sta->flags |= WLAN_STA_TIM;
2784 	}
2785 
2786 	ret = AP_TX_BUFFERED;
2787 
2788  out:
2789 	if (sta != NULL) {
2790 		if (ret == AP_TX_CONTINUE ||
2791 		    ret == AP_TX_CONTINUE_NOT_AUTHORIZED) {
2792 			sta->tx_packets++;
2793 			sta->tx_bytes += skb->len;
2794 			sta->last_tx = jiffies;
2795 		}
2796 
2797 		if ((ret == AP_TX_CONTINUE ||
2798 		     ret == AP_TX_CONTINUE_NOT_AUTHORIZED) &&
2799 		    sta->crypt && tx->host_encrypt) {
2800 			tx->crypt = sta->crypt;
2801 			tx->sta_ptr = sta; /* hostap_handle_sta_release() will
2802 					    * be called to release sta info
2803 					    * later */
2804 		} else
2805 			atomic_dec(&sta->users);
2806 	}
2807 
2808 	return ret;
2809 }
2810 
2811 
hostap_handle_sta_release(void * ptr)2812 void hostap_handle_sta_release(void *ptr)
2813 {
2814 	struct sta_info *sta = ptr;
2815 	atomic_dec(&sta->users);
2816 }
2817 
2818 
2819 /* Called only as a tasklet (software IRQ) */
hostap_handle_sta_tx_exc(local_info_t * local,struct sk_buff * skb)2820 void hostap_handle_sta_tx_exc(local_info_t *local, struct sk_buff *skb)
2821 {
2822 	struct sta_info *sta;
2823 	struct ieee80211_hdr *hdr;
2824 	struct hostap_skb_tx_data *meta;
2825 
2826 	hdr = (struct ieee80211_hdr *) skb->data;
2827 	meta = (struct hostap_skb_tx_data *) skb->cb;
2828 
2829 	spin_lock(&local->ap->sta_table_lock);
2830 	sta = ap_get_sta(local->ap, hdr->addr1);
2831 	if (!sta) {
2832 		spin_unlock(&local->ap->sta_table_lock);
2833 		PDEBUG(DEBUG_AP, "%s: Could not find STA %pM"
2834 		       " for this TX error (@%lu)\n",
2835 		       local->dev->name, hdr->addr1, jiffies);
2836 		return;
2837 	}
2838 
2839 	sta->tx_since_last_failure = 0;
2840 	sta->tx_consecutive_exc++;
2841 
2842 	if (sta->tx_consecutive_exc >= WLAN_RATE_DECREASE_THRESHOLD &&
2843 	    sta->tx_rate_idx > 0 && meta->rate <= sta->tx_rate) {
2844 		/* use next lower rate */
2845 		int old, rate;
2846 		old = rate = sta->tx_rate_idx;
2847 		while (rate > 0) {
2848 			rate--;
2849 			if (ap_tx_rate_ok(rate, sta, local)) {
2850 				sta->tx_rate_idx = rate;
2851 				break;
2852 			}
2853 		}
2854 		if (old != sta->tx_rate_idx) {
2855 			switch (sta->tx_rate_idx) {
2856 			case 0: sta->tx_rate = 10; break;
2857 			case 1: sta->tx_rate = 20; break;
2858 			case 2: sta->tx_rate = 55; break;
2859 			case 3: sta->tx_rate = 110; break;
2860 			default: sta->tx_rate = 0; break;
2861 			}
2862 			PDEBUG(DEBUG_AP,
2863 			       "%s: STA %pM TX rate lowered to %d\n",
2864 			       local->dev->name, sta->addr, sta->tx_rate);
2865 		}
2866 		sta->tx_consecutive_exc = 0;
2867 	}
2868 	spin_unlock(&local->ap->sta_table_lock);
2869 }
2870 
2871 
hostap_update_sta_ps2(local_info_t * local,struct sta_info * sta,int pwrmgt,int type,int stype)2872 static void hostap_update_sta_ps2(local_info_t *local, struct sta_info *sta,
2873 				  int pwrmgt, int type, int stype)
2874 {
2875 	if (pwrmgt && !(sta->flags & WLAN_STA_PS)) {
2876 		sta->flags |= WLAN_STA_PS;
2877 		PDEBUG(DEBUG_PS2, "STA %pM changed to use PS "
2878 		       "mode (type=0x%02X, stype=0x%02X)\n",
2879 		       sta->addr, type >> 2, stype >> 4);
2880 	} else if (!pwrmgt && (sta->flags & WLAN_STA_PS)) {
2881 		sta->flags &= ~WLAN_STA_PS;
2882 		PDEBUG(DEBUG_PS2, "STA %pM changed to not use "
2883 		       "PS mode (type=0x%02X, stype=0x%02X)\n",
2884 		       sta->addr, type >> 2, stype >> 4);
2885 		if (type != IEEE80211_FTYPE_CTL ||
2886 		    stype != IEEE80211_STYPE_PSPOLL)
2887 			schedule_packet_send(local, sta);
2888 	}
2889 }
2890 
2891 
2892 /* Called only as a tasklet (software IRQ). Called for each RX frame to update
2893  * STA power saving state. pwrmgt is a flag from 802.11 frame_control field. */
hostap_update_sta_ps(local_info_t * local,struct ieee80211_hdr * hdr)2894 int hostap_update_sta_ps(local_info_t *local, struct ieee80211_hdr *hdr)
2895 {
2896 	struct sta_info *sta;
2897 	u16 fc;
2898 
2899 	spin_lock(&local->ap->sta_table_lock);
2900 	sta = ap_get_sta(local->ap, hdr->addr2);
2901 	if (sta)
2902 		atomic_inc(&sta->users);
2903 	spin_unlock(&local->ap->sta_table_lock);
2904 
2905 	if (!sta)
2906 		return -1;
2907 
2908 	fc = le16_to_cpu(hdr->frame_control);
2909 	hostap_update_sta_ps2(local, sta, fc & IEEE80211_FCTL_PM,
2910 			      fc & IEEE80211_FCTL_FTYPE,
2911 			      fc & IEEE80211_FCTL_STYPE);
2912 
2913 	atomic_dec(&sta->users);
2914 	return 0;
2915 }
2916 
2917 
2918 /* Called only as a tasklet (software IRQ). Called for each RX frame after
2919  * getting RX header and payload from hardware. */
hostap_handle_sta_rx(local_info_t * local,struct net_device * dev,struct sk_buff * skb,struct hostap_80211_rx_status * rx_stats,int wds)2920 ap_rx_ret hostap_handle_sta_rx(local_info_t *local, struct net_device *dev,
2921 			       struct sk_buff *skb,
2922 			       struct hostap_80211_rx_status *rx_stats,
2923 			       int wds)
2924 {
2925 	int ret;
2926 	struct sta_info *sta;
2927 	u16 fc, type, stype;
2928 	struct ieee80211_hdr *hdr;
2929 
2930 	if (local->ap == NULL)
2931 		return AP_RX_CONTINUE;
2932 
2933 	hdr = (struct ieee80211_hdr *) skb->data;
2934 
2935 	fc = le16_to_cpu(hdr->frame_control);
2936 	type = fc & IEEE80211_FCTL_FTYPE;
2937 	stype = fc & IEEE80211_FCTL_STYPE;
2938 
2939 	spin_lock(&local->ap->sta_table_lock);
2940 	sta = ap_get_sta(local->ap, hdr->addr2);
2941 	if (sta)
2942 		atomic_inc(&sta->users);
2943 	spin_unlock(&local->ap->sta_table_lock);
2944 
2945 	if (sta && !(sta->flags & WLAN_STA_AUTHORIZED))
2946 		ret = AP_RX_CONTINUE_NOT_AUTHORIZED;
2947 	else
2948 		ret = AP_RX_CONTINUE;
2949 
2950 
2951 	if (fc & IEEE80211_FCTL_TODS) {
2952 		if (!wds && (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))) {
2953 			if (local->hostapd) {
2954 				prism2_rx_80211(local->apdev, skb, rx_stats,
2955 						PRISM2_RX_NON_ASSOC);
2956 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2957 			} else {
2958 				printk(KERN_DEBUG "%s: dropped received packet"
2959 				       " from non-associated STA %pM"
2960 				       " (type=0x%02x, subtype=0x%02x)\n",
2961 				       dev->name, hdr->addr2,
2962 				       type >> 2, stype >> 4);
2963 				hostap_rx(dev, skb, rx_stats);
2964 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2965 			}
2966 			ret = AP_RX_EXIT;
2967 			goto out;
2968 		}
2969 	} else if (fc & IEEE80211_FCTL_FROMDS) {
2970 		if (!wds) {
2971 			/* FromDS frame - not for us; probably
2972 			 * broadcast/multicast in another BSS - drop */
2973 			if (memcmp(hdr->addr1, dev->dev_addr, ETH_ALEN) == 0) {
2974 				printk(KERN_DEBUG "Odd.. FromDS packet "
2975 				       "received with own BSSID\n");
2976 				hostap_dump_rx_80211(dev->name, skb, rx_stats);
2977 			}
2978 			ret = AP_RX_DROP;
2979 			goto out;
2980 		}
2981 	} else if (stype == IEEE80211_STYPE_NULLFUNC && sta == NULL &&
2982 		   memcmp(hdr->addr1, dev->dev_addr, ETH_ALEN) == 0) {
2983 
2984 		if (local->hostapd) {
2985 			prism2_rx_80211(local->apdev, skb, rx_stats,
2986 					PRISM2_RX_NON_ASSOC);
2987 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2988 		} else {
2989 			/* At least Lucent f/w seems to send data::nullfunc
2990 			 * frames with no ToDS flag when the current AP returns
2991 			 * after being unavailable for some time. Speed up
2992 			 * re-association by informing the station about it not
2993 			 * being associated. */
2994 			printk(KERN_DEBUG "%s: rejected received nullfunc frame"
2995 			       " without ToDS from not associated STA %pM\n",
2996 			       dev->name, hdr->addr2);
2997 			hostap_rx(dev, skb, rx_stats);
2998 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2999 		}
3000 		ret = AP_RX_EXIT;
3001 		goto out;
3002 	} else if (stype == IEEE80211_STYPE_NULLFUNC) {
3003 		/* At least Lucent cards seem to send periodic nullfunc
3004 		 * frames with ToDS. Let these through to update SQ
3005 		 * stats and PS state. Nullfunc frames do not contain
3006 		 * any data and they will be dropped below. */
3007 	} else {
3008 		/* If BSSID (Addr3) is foreign, this frame is a normal
3009 		 * broadcast frame from an IBSS network. Drop it silently.
3010 		 * If BSSID is own, report the dropping of this frame. */
3011 		if (memcmp(hdr->addr3, dev->dev_addr, ETH_ALEN) == 0) {
3012 			printk(KERN_DEBUG "%s: dropped received packet from %pM"
3013 			       " with no ToDS flag "
3014 			       "(type=0x%02x, subtype=0x%02x)\n", dev->name,
3015 			       hdr->addr2, type >> 2, stype >> 4);
3016 			hostap_dump_rx_80211(dev->name, skb, rx_stats);
3017 		}
3018 		ret = AP_RX_DROP;
3019 		goto out;
3020 	}
3021 
3022 	if (sta) {
3023 		hostap_update_sta_ps2(local, sta, fc & IEEE80211_FCTL_PM,
3024 				      type, stype);
3025 
3026 		sta->rx_packets++;
3027 		sta->rx_bytes += skb->len;
3028 		sta->last_rx = jiffies;
3029 	}
3030 
3031 	if (local->ap->nullfunc_ack && stype == IEEE80211_STYPE_NULLFUNC &&
3032 	    fc & IEEE80211_FCTL_TODS) {
3033 		if (local->hostapd) {
3034 			prism2_rx_80211(local->apdev, skb, rx_stats,
3035 					PRISM2_RX_NULLFUNC_ACK);
3036 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
3037 		} else {
3038 			/* some STA f/w's seem to require control::ACK frame
3039 			 * for data::nullfunc, but Prism2 f/w 0.8.0 (at least
3040 			 * from Compaq) does not send this.. Try to generate
3041 			 * ACK for these frames from the host driver to make
3042 			 * power saving work with, e.g., Lucent WaveLAN f/w */
3043 			hostap_rx(dev, skb, rx_stats);
3044 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
3045 		}
3046 		ret = AP_RX_EXIT;
3047 		goto out;
3048 	}
3049 
3050  out:
3051 	if (sta)
3052 		atomic_dec(&sta->users);
3053 
3054 	return ret;
3055 }
3056 
3057 
3058 /* Called only as a tasklet (software IRQ) */
hostap_handle_sta_crypto(local_info_t * local,struct ieee80211_hdr * hdr,struct lib80211_crypt_data ** crypt,void ** sta_ptr)3059 int hostap_handle_sta_crypto(local_info_t *local,
3060 			     struct ieee80211_hdr *hdr,
3061 			     struct lib80211_crypt_data **crypt,
3062 			     void **sta_ptr)
3063 {
3064 	struct sta_info *sta;
3065 
3066 	spin_lock(&local->ap->sta_table_lock);
3067 	sta = ap_get_sta(local->ap, hdr->addr2);
3068 	if (sta)
3069 		atomic_inc(&sta->users);
3070 	spin_unlock(&local->ap->sta_table_lock);
3071 
3072 	if (!sta)
3073 		return -1;
3074 
3075 	if (sta->crypt) {
3076 		*crypt = sta->crypt;
3077 		*sta_ptr = sta;
3078 		/* hostap_handle_sta_release() will be called to release STA
3079 		 * info */
3080 	} else
3081 		atomic_dec(&sta->users);
3082 
3083 	return 0;
3084 }
3085 
3086 
3087 /* Called only as a tasklet (software IRQ) */
hostap_is_sta_assoc(struct ap_data * ap,u8 * sta_addr)3088 int hostap_is_sta_assoc(struct ap_data *ap, u8 *sta_addr)
3089 {
3090 	struct sta_info *sta;
3091 	int ret = 0;
3092 
3093 	spin_lock(&ap->sta_table_lock);
3094 	sta = ap_get_sta(ap, sta_addr);
3095 	if (sta != NULL && (sta->flags & WLAN_STA_ASSOC) && !sta->ap)
3096 		ret = 1;
3097 	spin_unlock(&ap->sta_table_lock);
3098 
3099 	return ret;
3100 }
3101 
3102 
3103 /* Called only as a tasklet (software IRQ) */
hostap_is_sta_authorized(struct ap_data * ap,u8 * sta_addr)3104 int hostap_is_sta_authorized(struct ap_data *ap, u8 *sta_addr)
3105 {
3106 	struct sta_info *sta;
3107 	int ret = 0;
3108 
3109 	spin_lock(&ap->sta_table_lock);
3110 	sta = ap_get_sta(ap, sta_addr);
3111 	if (sta != NULL && (sta->flags & WLAN_STA_ASSOC) && !sta->ap &&
3112 	    ((sta->flags & WLAN_STA_AUTHORIZED) ||
3113 	     ap->local->ieee_802_1x == 0))
3114 		ret = 1;
3115 	spin_unlock(&ap->sta_table_lock);
3116 
3117 	return ret;
3118 }
3119 
3120 
3121 /* Called only as a tasklet (software IRQ) */
hostap_add_sta(struct ap_data * ap,u8 * sta_addr)3122 int hostap_add_sta(struct ap_data *ap, u8 *sta_addr)
3123 {
3124 	struct sta_info *sta;
3125 	int ret = 1;
3126 
3127 	if (!ap)
3128 		return -1;
3129 
3130 	spin_lock(&ap->sta_table_lock);
3131 	sta = ap_get_sta(ap, sta_addr);
3132 	if (sta)
3133 		ret = 0;
3134 	spin_unlock(&ap->sta_table_lock);
3135 
3136 	if (ret == 1) {
3137 		sta = ap_add_sta(ap, sta_addr);
3138 		if (!sta)
3139 			return -1;
3140 		sta->flags = WLAN_STA_AUTH | WLAN_STA_ASSOC;
3141 		sta->ap = 1;
3142 		memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
3143 		/* No way of knowing which rates are supported since we did not
3144 		 * get supported rates element from beacon/assoc req. Assume
3145 		 * that remote end supports all 802.11b rates. */
3146 		sta->supported_rates[0] = 0x82;
3147 		sta->supported_rates[1] = 0x84;
3148 		sta->supported_rates[2] = 0x0b;
3149 		sta->supported_rates[3] = 0x16;
3150 		sta->tx_supp_rates = WLAN_RATE_1M | WLAN_RATE_2M |
3151 			WLAN_RATE_5M5 | WLAN_RATE_11M;
3152 		sta->tx_rate = 110;
3153 		sta->tx_max_rate = sta->tx_rate_idx = 3;
3154 	}
3155 
3156 	return ret;
3157 }
3158 
3159 
3160 /* Called only as a tasklet (software IRQ) */
hostap_update_rx_stats(struct ap_data * ap,struct ieee80211_hdr * hdr,struct hostap_80211_rx_status * rx_stats)3161 int hostap_update_rx_stats(struct ap_data *ap,
3162 			   struct ieee80211_hdr *hdr,
3163 			   struct hostap_80211_rx_status *rx_stats)
3164 {
3165 	struct sta_info *sta;
3166 
3167 	if (!ap)
3168 		return -1;
3169 
3170 	spin_lock(&ap->sta_table_lock);
3171 	sta = ap_get_sta(ap, hdr->addr2);
3172 	if (sta) {
3173 		sta->last_rx_silence = rx_stats->noise;
3174 		sta->last_rx_signal = rx_stats->signal;
3175 		sta->last_rx_rate = rx_stats->rate;
3176 		sta->last_rx_updated = IW_QUAL_ALL_UPDATED | IW_QUAL_DBM;
3177 		if (rx_stats->rate == 10)
3178 			sta->rx_count[0]++;
3179 		else if (rx_stats->rate == 20)
3180 			sta->rx_count[1]++;
3181 		else if (rx_stats->rate == 55)
3182 			sta->rx_count[2]++;
3183 		else if (rx_stats->rate == 110)
3184 			sta->rx_count[3]++;
3185 	}
3186 	spin_unlock(&ap->sta_table_lock);
3187 
3188 	return sta ? 0 : -1;
3189 }
3190 
3191 
hostap_update_rates(local_info_t * local)3192 void hostap_update_rates(local_info_t *local)
3193 {
3194 	struct sta_info *sta;
3195 	struct ap_data *ap = local->ap;
3196 
3197 	if (!ap)
3198 		return;
3199 
3200 	spin_lock_bh(&ap->sta_table_lock);
3201 	list_for_each_entry(sta, &ap->sta_list, list) {
3202 		prism2_check_tx_rates(sta);
3203 	}
3204 	spin_unlock_bh(&ap->sta_table_lock);
3205 }
3206 
3207 
ap_crypt_get_ptrs(struct ap_data * ap,u8 * addr,int permanent,struct lib80211_crypt_data *** crypt)3208 void * ap_crypt_get_ptrs(struct ap_data *ap, u8 *addr, int permanent,
3209 			 struct lib80211_crypt_data ***crypt)
3210 {
3211 	struct sta_info *sta;
3212 
3213 	spin_lock_bh(&ap->sta_table_lock);
3214 	sta = ap_get_sta(ap, addr);
3215 	if (sta)
3216 		atomic_inc(&sta->users);
3217 	spin_unlock_bh(&ap->sta_table_lock);
3218 
3219 	if (!sta && permanent)
3220 		sta = ap_add_sta(ap, addr);
3221 
3222 	if (!sta)
3223 		return NULL;
3224 
3225 	if (permanent)
3226 		sta->flags |= WLAN_STA_PERM;
3227 
3228 	*crypt = &sta->crypt;
3229 
3230 	return sta;
3231 }
3232 
3233 
hostap_add_wds_links(local_info_t * local)3234 void hostap_add_wds_links(local_info_t *local)
3235 {
3236 	struct ap_data *ap = local->ap;
3237 	struct sta_info *sta;
3238 
3239 	spin_lock_bh(&ap->sta_table_lock);
3240 	list_for_each_entry(sta, &ap->sta_list, list) {
3241 		if (sta->ap)
3242 			hostap_wds_link_oper(local, sta->addr, WDS_ADD);
3243 	}
3244 	spin_unlock_bh(&ap->sta_table_lock);
3245 
3246 	schedule_work(&local->ap->wds_oper_queue);
3247 }
3248 
3249 
hostap_wds_link_oper(local_info_t * local,u8 * addr,wds_oper_type type)3250 void hostap_wds_link_oper(local_info_t *local, u8 *addr, wds_oper_type type)
3251 {
3252 	struct wds_oper_data *entry;
3253 
3254 	entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
3255 	if (!entry)
3256 		return;
3257 	memcpy(entry->addr, addr, ETH_ALEN);
3258 	entry->type = type;
3259 	spin_lock_bh(&local->lock);
3260 	entry->next = local->ap->wds_oper_entries;
3261 	local->ap->wds_oper_entries = entry;
3262 	spin_unlock_bh(&local->lock);
3263 
3264 	schedule_work(&local->ap->wds_oper_queue);
3265 }
3266 
3267 
3268 EXPORT_SYMBOL(hostap_init_data);
3269 EXPORT_SYMBOL(hostap_init_ap_proc);
3270 EXPORT_SYMBOL(hostap_free_data);
3271 EXPORT_SYMBOL(hostap_check_sta_fw_version);
3272 EXPORT_SYMBOL(hostap_handle_sta_tx_exc);
3273 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
3274 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
3275