1 /*
2  * Copyright (c) 2001 Atsushi Onoe
3  * Copyright (c) 2002-2005 Sam Leffler, Errno Consulting
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. The name of the author may not be used to endorse or promote products
15  *    derived from this software without specific prior written permission.
16  *
17  * Alternatively, this software may be distributed under the terms of the
18  * GNU General Public License ("GPL") version 2 as published by the Free
19  * Software Foundation.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
22  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
23  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
24  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
25  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
26  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
30  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  *
32  * $FreeBSD: src/sys/net80211/ieee80211_node.c,v 1.48.2.12 2006/07/10 00:46:27 sam Exp $
33  * $DragonFly: src/sys/netproto/802_11/wlan/ieee80211_node.c,v 1.16 2007/03/06 14:51:22 sephe Exp $
34  */
35 
36 #include <sys/param.h>
37 #include <sys/systm.h>
38 #include <sys/mbuf.h>
39 #include <sys/malloc.h>
40 #include <sys/kernel.h>
41 
42 #include <sys/socket.h>
43 
44 #include <net/if.h>
45 #include <net/if_arp.h>
46 #include <net/if_media.h>
47 #include <net/ethernet.h>
48 
49 #include <netproto/802_11/ieee80211_var.h>
50 
51 #include <net/bpf.h>
52 
53 /*
54  * Association id's are managed with a bit vector.
55  */
56 #define	IEEE80211_AID_SET(b, w) \
57 	((w)[IEEE80211_AID(b) / 32] |= (1 << (IEEE80211_AID(b) % 32)))
58 #define	IEEE80211_AID_CLR(b, w) \
59 	((w)[IEEE80211_AID(b) / 32] &= ~(1 << (IEEE80211_AID(b) % 32)))
60 #define	IEEE80211_AID_ISSET(b, w) \
61 	((w)[IEEE80211_AID(b) / 32] & (1 << (IEEE80211_AID(b) % 32)))
62 
63 #ifdef IEEE80211_DEBUG_REFCNT
64 #define REFCNT_LOC "%s (%s:%u) %p<%s> refcnt %d\n", __func__, func, line
65 #else
66 #define REFCNT_LOC "%s %p<%s> refcnt %d\n", __func__
67 #endif
68 
69 static struct ieee80211_node *node_alloc(struct ieee80211_node_table *);
70 static void node_cleanup(struct ieee80211_node *);
71 static void node_free(struct ieee80211_node *);
72 static uint8_t node_getrssi(const struct ieee80211_node *);
73 
74 static void ieee80211_setup_node(struct ieee80211_node_table *,
75 		struct ieee80211_node *, const uint8_t *);
76 static void _ieee80211_free_node(struct ieee80211_node *);
77 static void ieee80211_free_allnodes(struct ieee80211_node_table *);
78 
79 static void ieee80211_timeout_scan_candidates(struct ieee80211_node_table *);
80 static void ieee80211_timeout_stations(struct ieee80211_node_table *);
81 
82 static void ieee80211_set_tim(struct ieee80211_node *, int set);
83 
84 static void ieee80211_node_table_init(struct ieee80211com *ic,
85 	struct ieee80211_node_table *nt, const char *name,
86 	int inact, int keyixmax,
87 	void (*timeout)(struct ieee80211_node_table *));
88 static void ieee80211_node_table_cleanup(struct ieee80211_node_table *nt);
89 
90 MALLOC_DEFINE(M_80211_NODE, "80211node", "802.11 node state");
91 
92 void
93 ieee80211_node_attach(struct ieee80211com *ic)
94 {
95 	ic->ic_node_alloc = node_alloc;
96 	ic->ic_node_free = node_free;
97 	ic->ic_node_cleanup = node_cleanup;
98 	ic->ic_node_getrssi = node_getrssi;
99 
100 	/* default station inactivity timer setings */
101 	ic->ic_inact_init = IEEE80211_INACT_INIT;
102 	ic->ic_inact_auth = IEEE80211_INACT_AUTH;
103 	ic->ic_inact_run = IEEE80211_INACT_RUN;
104 	ic->ic_inact_probe = IEEE80211_INACT_PROBE;
105 
106 	/* NB: driver should override */
107 	ic->ic_max_aid = IEEE80211_AID_DEF;
108 	ic->ic_set_tim = ieee80211_set_tim;
109 }
110 
111 void
112 ieee80211_node_lateattach(struct ieee80211com *ic)
113 {
114 	struct ieee80211_rsnparms *rsn;
115 
116 	if (ic->ic_max_aid > IEEE80211_AID_MAX)
117 		ic->ic_max_aid = IEEE80211_AID_MAX;
118 
119 	ic->ic_aid_bitmap =
120 		kmalloc(howmany(ic->ic_max_aid, 32) * sizeof(uint32_t),
121 		       M_DEVBUF, M_WAITOK | M_ZERO);
122 
123 	/* XXX defer until using hostap/ibss mode */
124 	ic->ic_tim_len = howmany(ic->ic_max_aid, 8) * sizeof(uint8_t);
125 	ic->ic_tim_bitmap = kmalloc(ic->ic_tim_len, M_DEVBUF,
126 				   M_WAITOK | M_ZERO);
127 
128 	ieee80211_node_table_init(ic, &ic->ic_sta, "station",
129 		IEEE80211_INACT_INIT, ic->ic_crypto.cs_max_keyix,
130 		ieee80211_timeout_stations);
131 	ieee80211_node_table_init(ic, &ic->ic_scan, "scan",
132 		IEEE80211_INACT_SCAN, 0,
133 		ieee80211_timeout_scan_candidates);
134 
135 	ieee80211_reset_bss(ic);
136 	/*
137 	 * Setup "global settings" in the bss node so that
138 	 * each new station automatically inherits them.
139 	 */
140 	rsn = &ic->ic_bss->ni_rsn;
141 	/* WEP, TKIP, and AES-CCM are always supported */
142 	rsn->rsn_ucastcipherset |= 1<<IEEE80211_CIPHER_WEP;
143 	rsn->rsn_ucastcipherset |= 1<<IEEE80211_CIPHER_TKIP;
144 	rsn->rsn_ucastcipherset |= 1<<IEEE80211_CIPHER_AES_CCM;
145 	if (ic->ic_caps & IEEE80211_C_AES)
146 		rsn->rsn_ucastcipherset |= 1<<IEEE80211_CIPHER_AES_OCB;
147 	if (ic->ic_caps & IEEE80211_C_CKIP)
148 		rsn->rsn_ucastcipherset |= 1<<IEEE80211_CIPHER_CKIP;
149 	/*
150 	 * Default unicast cipher to WEP for 802.1x use.  If
151 	 * WPA is enabled the management code will set these
152 	 * values to reflect.
153 	 */
154 	rsn->rsn_ucastcipher = IEEE80211_CIPHER_WEP;
155 	rsn->rsn_ucastkeylen = 104 / NBBY;
156 	/*
157 	 * WPA says the multicast cipher is the lowest unicast
158 	 * cipher supported.  But we skip WEP which would
159 	 * otherwise be used based on this criteria.
160 	 */
161 	rsn->rsn_mcastcipher = IEEE80211_CIPHER_TKIP;
162 	rsn->rsn_mcastkeylen = 128 / NBBY;
163 
164 	/*
165 	 * We support both WPA-PSK and 802.1x; the one used
166 	 * is determined by the authentication mode and the
167 	 * setting of the PSK state.
168 	 */
169 	rsn->rsn_keymgmtset = WPA_ASE_8021X_UNSPEC | WPA_ASE_8021X_PSK;
170 	rsn->rsn_keymgmt = WPA_ASE_8021X_PSK;
171 
172 	ic->ic_auth = ieee80211_authenticator_get(ic->ic_bss->ni_authmode);
173 }
174 
175 void
176 ieee80211_node_detach(struct ieee80211com *ic)
177 {
178 	if (ic->ic_bss != NULL) {
179 		ieee80211_free_node(ic->ic_bss);
180 		ic->ic_bss = NULL;
181 	}
182 	ieee80211_node_table_cleanup(&ic->ic_scan);
183 	ieee80211_node_table_cleanup(&ic->ic_sta);
184 	if (ic->ic_aid_bitmap != NULL) {
185 		kfree(ic->ic_aid_bitmap, M_DEVBUF);
186 		ic->ic_aid_bitmap = NULL;
187 	}
188 	if (ic->ic_tim_bitmap != NULL) {
189 		kfree(ic->ic_tim_bitmap, M_DEVBUF);
190 		ic->ic_tim_bitmap = NULL;
191 	}
192 }
193 
194 /*
195  * Port authorize/unauthorize interfaces for use by an authenticator.
196  */
197 
198 void
199 ieee80211_node_authorize(struct ieee80211_node *ni)
200 {
201 	struct ieee80211com *ic = ni->ni_ic;
202 
203 	ni->ni_flags |= IEEE80211_NODE_AUTH;
204 	ni->ni_inact_reload = ic->ic_inact_run;
205 }
206 
207 void
208 ieee80211_node_unauthorize(struct ieee80211_node *ni)
209 {
210 	ni->ni_flags &= ~IEEE80211_NODE_AUTH;
211 }
212 
213 /*
214  * Set/change the channel.  The rate set is also updated as
215  * to insure a consistent view by drivers.
216  */
217 static void
218 ieee80211_set_chan(struct ieee80211com *ic,
219 	struct ieee80211_node *ni, struct ieee80211_channel *chan)
220 {
221 	if (chan == IEEE80211_CHAN_ANYC)	/* XXX while scanning */
222 		chan = ic->ic_curchan;
223 	ni->ni_chan = chan;
224 	ni->ni_rates = ic->ic_sup_rates[ieee80211_chan2mode(ic, chan)];
225 }
226 
227 /*
228  * AP scanning support.
229  */
230 
231 #ifdef IEEE80211_DEBUG
232 static void
233 dump_chanlist(const u_char chans[])
234 {
235 	const char *sep;
236 	int i;
237 
238 	sep = " ";
239 	for (i = 0; i < IEEE80211_CHAN_MAX; i++)
240 		if (isset(chans, i)) {
241 			kprintf("%s%u", sep, i);
242 			sep = ", ";
243 		}
244 }
245 #endif /* IEEE80211_DEBUG */
246 
247 /*
248  * Initialize the channel set to scan based on the
249  * of available channels and the current PHY mode.
250  */
251 static void
252 ieee80211_reset_scan(struct ieee80211com *ic)
253 {
254 
255 	/* XXX ic_des_chan should be handled with ic_chan_active */
256 	if (ic->ic_des_chan != IEEE80211_CHAN_ANYC) {
257 		memset(ic->ic_chan_scan, 0, sizeof(ic->ic_chan_scan));
258 		setbit(ic->ic_chan_scan,
259 			ieee80211_chan2ieee(ic, ic->ic_des_chan));
260 	} else
261 		memcpy(ic->ic_chan_scan, ic->ic_chan_active,
262 			sizeof(ic->ic_chan_active));
263 #ifdef IEEE80211_DEBUG
264 	if (ieee80211_msg_scan(ic)) {
265 		kprintf("%s: scan set:", __func__);
266 		dump_chanlist(ic->ic_chan_scan);
267 		kprintf(" start chan %u\n",
268 			ieee80211_chan2ieee(ic, ic->ic_curchan));
269 	}
270 #endif /* IEEE80211_DEBUG */
271 }
272 
273 /*
274  * Begin an active scan.
275  */
276 void
277 ieee80211_begin_scan(struct ieee80211com *ic, int reset)
278 {
279 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
280 
281 	/*
282 	 * In all but hostap mode scanning starts off in
283 	 * an active mode before switching to passive.
284 	 */
285 	if (ic->ic_opmode != IEEE80211_M_HOSTAP) {
286 		ic->ic_flags |= IEEE80211_F_ASCAN;
287 		ic->ic_stats.is_scan_active++;
288 	} else
289 		ic->ic_stats.is_scan_passive++;
290 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN,
291 		"begin %s scan in %s mode\n",
292 		(ic->ic_flags & IEEE80211_F_ASCAN) ?  "active" : "passive",
293 		ieee80211_phymode_name[ic->ic_curmode]);
294 	/*
295 	 * Clear scan state and flush any previously seen AP's.
296 	 */
297 	ieee80211_reset_scan(ic);
298 	if (reset)
299 		ieee80211_free_allnodes(&ic->ic_scan);
300 
301 	ic->ic_flags |= IEEE80211_F_SCAN;
302 
303 	/* Scan the next channel. */
304 	ieee80211_next_scan(ic);
305 }
306 
307 /*
308  * Switch to the next channel marked for scanning.
309  */
310 int
311 ieee80211_next_scan(struct ieee80211com *ic)
312 {
313 	struct ieee80211_channel *chan;
314 
315 	/*
316 	 * Insure any previous mgt frame timeouts don't fire.
317 	 * This assumes the driver does the right thing in
318 	 * flushing anything queued in the driver and below.
319 	 */
320 	ic->ic_mgt_timer = 0;
321 	ic->ic_flags_ext &= ~IEEE80211_FEXT_PROBECHAN;
322 
323 	chan = ic->ic_curchan;
324 	do {
325 		if (++chan > &ic->ic_channels[IEEE80211_CHAN_MAX])
326 			chan = &ic->ic_channels[0];
327 		if (isset(ic->ic_chan_scan, ieee80211_chan2ieee(ic, chan))) {
328 			clrbit(ic->ic_chan_scan, ieee80211_chan2ieee(ic, chan));
329 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN,
330 			    "%s: chan %d->%d\n", __func__,
331 			    ieee80211_chan2ieee(ic, ic->ic_curchan),
332 			    ieee80211_chan2ieee(ic, chan));
333 			ic->ic_curchan = chan;
334 			/*
335 			 * XXX drivers should do this as needed,
336 			 * XXX for now maintain compatibility
337 			 */
338 			ic->ic_bss->ni_rates =
339 				ic->ic_sup_rates[ieee80211_chan2mode(ic, chan)];
340 			ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
341 			return 1;
342 		}
343 	} while (chan != ic->ic_curchan);
344 	ieee80211_end_scan(ic);
345 	return 0;
346 }
347 
348 /*
349  * Probe the curent channel, if allowed, while scanning.
350  * If the channel is not marked passive-only then send
351  * a probe request immediately.  Otherwise mark state and
352  * listen for beacons on the channel; if we receive something
353  * then we'll transmit a probe request.
354  */
355 void
356 ieee80211_probe_curchan(struct ieee80211com *ic, int force)
357 {
358 	struct ifnet *ifp = ic->ic_ifp;
359 
360 	if ((ic->ic_curchan->ic_flags & IEEE80211_CHAN_PASSIVE) == 0 || force) {
361 		/*
362 		 * XXX send both broadcast+directed probe request
363 		 */
364 		ieee80211_send_probereq(ic->ic_bss,
365 			ic->ic_myaddr, ifp->if_broadcastaddr,
366 			ifp->if_broadcastaddr,
367 			ic->ic_des_essid, ic->ic_des_esslen,
368 			ic->ic_opt_ie, ic->ic_opt_ie_len);
369 	} else
370 		ic->ic_flags_ext |= IEEE80211_FEXT_PROBECHAN;
371 }
372 
373 static __inline void
374 copy_bss(struct ieee80211_node *nbss, const struct ieee80211_node *obss)
375 {
376 	/* propagate useful state */
377 	nbss->ni_authmode = obss->ni_authmode;
378 	nbss->ni_txpower = obss->ni_txpower;
379 	nbss->ni_vlan = obss->ni_vlan;
380 	nbss->ni_rsn = obss->ni_rsn;
381 	ieee80211_ratectl_data_dup(obss, nbss);
382 	/* XXX statistics? */
383 }
384 
385 void
386 ieee80211_create_ibss(struct ieee80211com* ic, struct ieee80211_channel *chan)
387 {
388 	struct ieee80211_node_table *nt;
389 	struct ieee80211_node *ni;
390 
391 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
392 
393 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN,
394 		"%s: creating ibss\n", __func__);
395 
396 	/*
397 	 * Create the station/neighbor table.  Note that for adhoc
398 	 * mode we make the initial inactivity timer longer since
399 	 * we create nodes only through discovery and they typically
400 	 * are long-lived associations.
401 	 */
402 	nt = &ic->ic_sta;
403 	if (ic->ic_opmode == IEEE80211_M_HOSTAP) {
404 		nt->nt_name = "station";
405 		nt->nt_inact_init = ic->ic_inact_init;
406 	} else {
407 		nt->nt_name = "neighbor";
408 		nt->nt_inact_init = ic->ic_inact_run;
409 	}
410 
411 	ni = ieee80211_alloc_node(&ic->ic_sta, ic->ic_myaddr);
412 	if (ni == NULL) {
413 		/* XXX recovery? */
414 		return;
415 	}
416 	IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_myaddr);
417 	ni->ni_esslen = ic->ic_des_esslen;
418 	memcpy(ni->ni_essid, ic->ic_des_essid, ni->ni_esslen);
419 	copy_bss(ni, ic->ic_bss);
420 	ni->ni_intval = ic->ic_bintval;
421 	if (ic->ic_flags & IEEE80211_F_PRIVACY)
422 		ni->ni_capinfo |= IEEE80211_CAPINFO_PRIVACY;
423 	if (ic->ic_phytype == IEEE80211_T_FH) {
424 		ni->ni_fhdwell = 200;	/* XXX */
425 		ni->ni_fhindex = 1;
426 	}
427 	if (ic->ic_opmode == IEEE80211_M_IBSS) {
428 		ic->ic_flags |= IEEE80211_F_SIBSS;
429 		ni->ni_capinfo |= IEEE80211_CAPINFO_IBSS;	/* XXX */
430 		if (ic->ic_flags & IEEE80211_F_DESBSSID) {
431 			IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_des_bssid);
432 		} else {
433 			get_random_bytes(ni->ni_bssid, IEEE80211_ADDR_LEN);
434 			/* Clear group bit, add local bit */
435 			ni->ni_bssid[0] = (ni->ni_bssid[0] &~ 0x01) | 0x02;
436 		}
437 	} else if (ic->ic_opmode == IEEE80211_M_AHDEMO) {
438 		if (ic->ic_flags & IEEE80211_F_DESBSSID)
439 			IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_des_bssid);
440 		else
441 			memset(ni->ni_bssid, 0, IEEE80211_ADDR_LEN);
442 	}
443 	/*
444 	 * Fix the channel and related attributes.
445 	 */
446 	ieee80211_set_chan(ic, ni, chan);
447 	ic->ic_curchan = chan;
448 	ic->ic_curmode = ieee80211_chan2mode(ic, chan);
449 	/*
450 	 * Do mode-specific rate setup.
451 	 */
452 	ieee80211_set_basicrates(&ni->ni_rates, ic->ic_curmode,
453 				 ic->ic_flags & IEEE80211_F_PUREG);
454 
455 	ieee80211_sta_join(ic, ieee80211_ref_node(ni));
456 }
457 
458 void
459 ieee80211_reset_bss(struct ieee80211com *ic)
460 {
461 	struct ieee80211_node *ni, *obss;
462 
463 	ieee80211_node_table_reset(&ic->ic_scan);
464 	ieee80211_node_table_reset(&ic->ic_sta);
465 
466 	ni = ieee80211_alloc_node(&ic->ic_scan, ic->ic_myaddr);
467 	KASSERT(ni != NULL, ("unable to setup inital BSS node"));
468 	obss = ic->ic_bss;
469 	ic->ic_bss = ieee80211_ref_node(ni);
470 	if (obss != NULL) {
471 		copy_bss(ni, obss);
472 		ni->ni_intval = ic->ic_bintval;
473 		ieee80211_free_node(obss);
474 	}
475 	ic->ic_nbasicrates = 0;
476 }
477 
478 /* XXX tunable */
479 #define	STA_FAILS_MAX	2		/* assoc failures before ignored */
480 
481 static int
482 ieee80211_match_bss(struct ieee80211com *ic, struct ieee80211_node *ni)
483 {
484         uint8_t rate;
485         int fail;
486 
487 	IEEE80211_PRINT_NODERATES(ic, ni, 0);
488 
489 	fail = 0;
490 	if (isclr(ic->ic_chan_active, ieee80211_chan2ieee(ic, ni->ni_chan)))
491 		fail |= 0x01;
492 	if (ic->ic_des_chan != IEEE80211_CHAN_ANYC &&
493 	    ni->ni_chan != ic->ic_des_chan)
494 		fail |= 0x01;
495 	if (ic->ic_opmode == IEEE80211_M_IBSS) {
496 		if ((ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) == 0)
497 			fail |= 0x02;
498 	} else {
499 		if ((ni->ni_capinfo & IEEE80211_CAPINFO_ESS) == 0)
500 			fail |= 0x02;
501 	}
502 	if (ic->ic_flags & IEEE80211_F_PRIVACY) {
503 		if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0)
504 			fail |= 0x04;
505 	} else {
506 		/* XXX does this mean privacy is supported or required? */
507 		if (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY)
508 			fail |= 0x04;
509 	}
510 	rate = ieee80211_fix_rate(ni,
511 		IEEE80211_F_DONEGO | IEEE80211_F_DOFRATE, 1);
512 	if (rate & IEEE80211_RATE_BASIC)
513 		fail |= 0x08;
514 	if (ic->ic_des_esslen != 0 &&
515 	    (ni->ni_esslen != ic->ic_des_esslen ||
516 	     memcmp(ni->ni_essid, ic->ic_des_essid, ic->ic_des_esslen) != 0))
517 		fail |= 0x10;
518 	if ((ic->ic_flags & IEEE80211_F_DESBSSID) &&
519 	    !IEEE80211_ADDR_EQ(ic->ic_des_bssid, ni->ni_bssid))
520 		fail |= 0x20;
521 	if (ni->ni_fails >= STA_FAILS_MAX)
522 		fail |= 0x40;
523 #ifdef IEEE80211_DEBUG
524 	if (ieee80211_msg_scan(ic)) {
525 		kprintf(" %c %6D",
526 		    fail & 0x40 ? '=' : fail & 0x80 ? '^' : fail ? '-' : '+',
527 		    ni->ni_macaddr, ":");
528 		kprintf(" %6D%c", ni->ni_bssid, ":",
529 		    fail & 0x20 ? '!' : ' ');
530 		kprintf(" %3d%c", ieee80211_chan2ieee(ic, ni->ni_chan),
531 			fail & 0x01 ? '!' : ' ');
532 		kprintf(" %+4d", ni->ni_rssi);
533 		kprintf(" %2dM%c", (rate & IEEE80211_RATE_VAL) / 2,
534 		    fail & 0x08 ? '!' : ' ');
535 		kprintf(" %4s%c",
536 		    (ni->ni_capinfo & IEEE80211_CAPINFO_ESS) ? "ess" :
537 		    (ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) ? "ibss" :
538 		    "????",
539 		    fail & 0x02 ? '!' : ' ');
540 		kprintf(" %3s%c ",
541 		    (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) ?
542 		    "wep" : "no",
543 		    fail & 0x04 ? '!' : ' ');
544 		ieee80211_print_essid(ni->ni_essid, ni->ni_esslen);
545 		kprintf("%s\n", fail & 0x10 ? "!" : "");
546 	}
547 #endif
548 	return fail;
549 }
550 
551 static __inline uint8_t
552 maxrate(const struct ieee80211_node *ni)
553 {
554 	const struct ieee80211_rateset *rs = &ni->ni_rates;
555 	/* NB: assumes rate set is sorted (happens on frame receive) */
556 	return rs->rs_rates[rs->rs_nrates-1] & IEEE80211_RATE_VAL;
557 }
558 
559 /*
560  * Compare the capabilities of two nodes and decide which is
561  * more desirable (return >0 if a is considered better).  Note
562  * that we assume compatibility/usability has already been checked
563  * so we don't need to (e.g. validate whether privacy is supported).
564  * Used to select the best scan candidate for association in a BSS.
565  */
566 static int
567 ieee80211_node_compare(struct ieee80211com *ic,
568 		       const struct ieee80211_node *a,
569 		       const struct ieee80211_node *b)
570 {
571 #define ABS(a)	((a) < 0 ? -(a) : (a))
572 	uint8_t maxa, maxb;
573 	uint8_t rssia, rssib;
574 	int weight;
575 
576 	/* privacy support preferred */
577 	if ((a->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) &&
578 	    (b->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0)
579 		return 1;
580 	if ((a->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0 &&
581 	    (b->ni_capinfo & IEEE80211_CAPINFO_PRIVACY))
582 		return -1;
583 
584 	/* compare count of previous failures */
585 	weight = b->ni_fails - a->ni_fails;
586 	if (ABS(weight) > 1)
587 		return weight;
588 
589 	rssia = ic->ic_node_getrssi(a);
590 	rssib = ic->ic_node_getrssi(b);
591 	if (ABS(rssib - rssia) < 5) {
592 		/* best/max rate preferred if signal level close enough XXX */
593 		maxa = maxrate(a);
594 		maxb = maxrate(b);
595 		if (maxa != maxb)
596 			return maxa - maxb;
597 		/* XXX use freq for channel preference */
598 		/* for now just prefer 5Ghz band to all other bands */
599 		if (IEEE80211_IS_CHAN_5GHZ(a->ni_chan) &&
600 		   !IEEE80211_IS_CHAN_5GHZ(b->ni_chan))
601 			return 1;
602 		if (!IEEE80211_IS_CHAN_5GHZ(a->ni_chan) &&
603 		     IEEE80211_IS_CHAN_5GHZ(b->ni_chan))
604 			return -1;
605 	}
606 	/* all things being equal, use signal level */
607 	return rssia - rssib;
608 #undef ABS
609 }
610 
611 /*
612  * Mark an ongoing scan stopped.
613  */
614 void
615 ieee80211_cancel_scan(struct ieee80211com *ic)
616 {
617 
618 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN, "%s: end %s scan\n",
619 		__func__,
620 		(ic->ic_flags & IEEE80211_F_ASCAN) ?  "active" : "passive");
621 
622 	ic->ic_flags &= ~(IEEE80211_F_SCAN | IEEE80211_F_ASCAN);
623 	ic->ic_flags_ext &= ~IEEE80211_FEXT_PROBECHAN;
624 }
625 
626 /*
627  * Complete a scan of potential channels.
628  */
629 void
630 ieee80211_end_scan(struct ieee80211com *ic)
631 {
632 	struct ieee80211_node_table *nt = &ic->ic_scan;
633 	struct ieee80211_node *ni, *selbs;
634 
635 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
636 
637 	ieee80211_cancel_scan(ic);
638 	ieee80211_notify_scan_done(ic);
639 
640 	if (ic->ic_opmode == IEEE80211_M_HOSTAP) {
641 		uint8_t maxrssi[IEEE80211_CHAN_MAX];	/* XXX off stack? */
642 		int i, bestchan;
643 		uint8_t rssi;
644 
645 		/*
646 		 * The passive scan to look for existing AP's completed,
647 		 * select a channel to camp on.  Identify the channels
648 		 * that already have one or more AP's and try to locate
649 		 * an unoccupied one.  If that fails, pick a channel that
650 		 * looks to be quietest.
651 		 */
652 		memset(maxrssi, 0, sizeof(maxrssi));
653 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
654 			rssi = ic->ic_node_getrssi(ni);
655 			i = ieee80211_chan2ieee(ic, ni->ni_chan);
656 			if (rssi > maxrssi[i])
657 				maxrssi[i] = rssi;
658 		}
659 		/* XXX select channel more intelligently */
660 		bestchan = -1;
661 		for (i = 0; i < IEEE80211_CHAN_MAX; i++)
662 			if (isset(ic->ic_chan_active, i)) {
663 				/*
664 				 * If the channel is unoccupied the max rssi
665 				 * should be zero; just take it.  Otherwise
666 				 * track the channel with the lowest rssi and
667 				 * use that when all channels appear occupied.
668 				 */
669 				if (maxrssi[i] == 0) {
670 					bestchan = i;
671 					break;
672 				}
673 				if (bestchan == -1 ||
674 				    maxrssi[i] < maxrssi[bestchan])
675 					bestchan = i;
676 			}
677 		if (bestchan != -1) {
678 			ieee80211_create_ibss(ic, &ic->ic_channels[bestchan]);
679 			return;
680 		}
681 		/* no suitable channel, should not happen */
682 	}
683 
684 	/*
685 	 * When manually sequencing the state machine; scan just once
686 	 * regardless of whether we have a candidate or not.  The
687 	 * controlling application is expected to setup state and
688 	 * initiate an association.
689 	 */
690 	if (ic->ic_roaming == IEEE80211_ROAMING_MANUAL)
691 		return;
692 	/*
693 	 * Automatic sequencing; look for a candidate and
694 	 * if found join the network.
695 	 */
696 	/* NB: unlocked read should be ok */
697 	if (TAILQ_FIRST(&nt->nt_node) == NULL) {
698 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN,
699 			"%s: no scan candidate\n", __func__);
700   notfound:
701 		if (ic->ic_opmode == IEEE80211_M_IBSS &&
702 		    (ic->ic_flags & IEEE80211_F_IBSSON) &&
703 		    ic->ic_des_esslen != 0) {
704 			ieee80211_create_ibss(ic, ic->ic_ibss_chan);
705 			return;
706 		}
707 		/*
708 		 * Decrement the failure counts so entries will be
709 		 * reconsidered the next time around.  We really want
710 		 * to do this only for sta's where we've previously
711 		 * had some success.
712 		 */
713 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list)
714 			if (ni->ni_fails)
715 				ni->ni_fails--;
716 		/*
717 		 * Reset the list of channels to scan and start again.
718 		 */
719 		ieee80211_reset_scan(ic);
720 		ic->ic_flags |= IEEE80211_F_SCAN;
721 		ieee80211_next_scan(ic);
722 		return;
723 	}
724 	selbs = NULL;
725 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_SCAN, "\t%s\n",
726 	    "macaddr          bssid         chan  rssi rate flag  wep  essid");
727 	TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
728 		if (ieee80211_match_bss(ic, ni) == 0) {
729 			if (selbs == NULL)
730 				selbs = ni;
731 			else if (ieee80211_node_compare(ic, ni, selbs) > 0)
732 				selbs = ni;
733 		}
734 	}
735 	if (selbs != NULL)		/* NB: grab ref while dropping lock */
736 		ieee80211_ref_node(selbs);
737 	if (selbs == NULL)
738 		goto notfound;
739 	if (!ieee80211_sta_join(ic, selbs)) {
740 		ieee80211_free_node(selbs);
741 		goto notfound;
742 	}
743 }
744 
745 /*
746  * Handle 802.11 ad hoc network merge.  The
747  * convention, set by the Wireless Ethernet Compatibility Alliance
748  * (WECA), is that an 802.11 station will change its BSSID to match
749  * the "oldest" 802.11 ad hoc network, on the same channel, that
750  * has the station's desired SSID.  The "oldest" 802.11 network
751  * sends beacons with the greatest TSF timestamp.
752  *
753  * The caller is assumed to validate TSF's before attempting a merge.
754  *
755  * Return !0 if the BSSID changed, 0 otherwise.
756  */
757 int
758 ieee80211_ibss_merge(struct ieee80211_node *ni)
759 {
760 	struct ieee80211com *ic = ni->ni_ic;
761 
762 	if (ni == ic->ic_bss ||
763 	    IEEE80211_ADDR_EQ(ni->ni_bssid, ic->ic_bss->ni_bssid)) {
764 		/* unchanged, nothing to do */
765 		return 0;
766 	}
767 	if (ieee80211_match_bss(ic, ni) != 0) {	/* capabilities mismatch */
768 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
769 		    "%s: merge failed, capabilities mismatch\n", __func__);
770 		ic->ic_stats.is_ibss_capmismatch++;
771 		return 0;
772 	}
773 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
774 		"%6D: new bssid %s: %s preamble, %s slot time%s\n", __func__,
775 		ni->ni_bssid, ":",
776 		ic->ic_flags&IEEE80211_F_SHPREAMBLE ? "short" : "long",
777 		ic->ic_flags&IEEE80211_F_SHSLOT ? "short" : "long",
778 		ic->ic_flags&IEEE80211_F_USEPROT ? ", protection" : ""
779 	);
780 	return ieee80211_sta_join(ic, ieee80211_ref_node(ni));
781 }
782 
783 /*
784  * Join the specified IBSS/BSS network.  The node is assumed to
785  * be passed in with a held reference.
786  */
787 int
788 ieee80211_sta_join(struct ieee80211com *ic, struct ieee80211_node *selbs)
789 {
790 	struct ieee80211_node *obss;
791 
792 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
793 
794 	if (ic->ic_opmode == IEEE80211_M_IBSS ||
795 	    ic->ic_opmode == IEEE80211_M_STA) {
796 		/*
797 		 * Delete unusable rates; we've already checked
798 		 * that the negotiated rate set is acceptable.
799 		 */
800 		ieee80211_fix_rate(selbs, IEEE80211_F_DODEL, 1);
801 		IEEE80211_PRINT_NODERATES(ic, selbs, 0);
802 
803 		if (ic->ic_opmode == IEEE80211_M_IBSS) {
804 			struct ieee80211_node_table *nt;
805 
806 			/*
807 			 * Fillin the neighbor table; it will already
808 			 * exist if we are simply switching mastership.
809 			 * XXX ic_sta always setup so this is unnecessary?
810 			 */
811 			nt = &ic->ic_sta;
812 			nt->nt_name = "neighbor";
813 			nt->nt_inact_init = ic->ic_inact_run;
814 		}
815 	}
816 
817 	/*
818 	 * Committed to selbs, setup state.
819 	 */
820 	obss = ic->ic_bss;
821 	ic->ic_bss = selbs;		/* NB: caller assumed to bump refcnt */
822 	if (obss != NULL) {
823 		copy_bss(selbs, obss);
824 		ieee80211_free_node(obss);
825 	}
826 	/*
827 	 * Set the erp state (mostly the slot time) to deal with
828 	 * the auto-select case; this should be redundant if the
829 	 * mode is locked.
830 	 */
831 	ic->ic_curmode = ieee80211_chan2mode(ic, selbs->ni_chan);
832 	ic->ic_curchan = selbs->ni_chan;
833 	ieee80211_reset_erp(ic);
834 	ieee80211_wme_initparams(ic);
835 
836 	/*
837 	 * Copy BSS basic rate set.
838 	 */
839 	ic->ic_nbasicrates =
840 		ieee80211_copy_basicrates(&ic->ic_sup_rates[ic->ic_curmode],
841 					  &selbs->ni_rates);
842 #ifdef IEEE80211_DEBUG
843 	if (ieee80211_msg(ic, IEEE80211_MSG_XRATE)) {
844 		ieee80211_note(ic, "number basic rates %d, "
845 			       "supported rates (mode %d): ",
846 			       ic->ic_nbasicrates, ic->ic_curmode);
847 		ieee80211_print_rateset(&ic->ic_sup_rates[ic->ic_curmode]);
848 		kprintf("\n");
849 	}
850 #endif
851 
852 	if (ic->ic_opmode == IEEE80211_M_STA)
853 		ieee80211_new_state(ic, IEEE80211_S_AUTH, -1);
854 	else
855 		ieee80211_new_state(ic, IEEE80211_S_RUN, -1);
856 	return 1;
857 }
858 
859 /*
860  * Leave the specified IBSS/BSS network.  The node is assumed to
861  * be passed in with a held reference.
862  */
863 void
864 ieee80211_sta_leave(struct ieee80211com *ic, struct ieee80211_node *ni)
865 {
866 	ic->ic_node_cleanup(ni);
867 	ieee80211_notify_node_leave(ic, ni);
868 }
869 
870 static struct ieee80211_node *
871 node_alloc(struct ieee80211_node_table *nt)
872 {
873 	struct ieee80211_node *ni;
874 
875 	ni = kmalloc(sizeof(struct ieee80211_node), M_80211_NODE,
876 		    M_NOWAIT | M_ZERO);
877 	return ni;
878 }
879 
880 /*
881  * Reclaim any resources in a node and reset any critical
882  * state.  Typically nodes are free'd immediately after,
883  * but in some cases the storage may be reused so we need
884  * to insure consistent state (should probably fix that).
885  */
886 static void
887 node_cleanup(struct ieee80211_node *ni)
888 {
889 #define	N(a)	(sizeof(a)/sizeof(a[0]))
890 	struct ieee80211com *ic = ni->ni_ic;
891 	int i, qlen;
892 
893 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
894 
895 	/* NB: preserve ni_table */
896 	if (ni->ni_flags & IEEE80211_NODE_PWR_MGT) {
897 		ic->ic_ps_sta--;
898 		ni->ni_flags &= ~IEEE80211_NODE_PWR_MGT;
899 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_POWER,
900 		    "[%6D] power save mode off, %u sta's in ps mode\n",
901 		    ni->ni_macaddr, ":", ic->ic_ps_sta);
902 	}
903 	/*
904 	 * Clear AREF flag that marks the authorization refcnt bump
905 	 * has happened.  This is probably not needed as the node
906 	 * should always be removed from the table so not found but
907 	 * do it just in case.
908 	 */
909 	ni->ni_flags &= ~IEEE80211_NODE_AREF;
910 
911 	/*
912 	 * Drain power save queue and, if needed, clear TIM.
913 	 */
914 	IEEE80211_NODE_SAVEQ_DRAIN(ni, qlen);
915 	if (qlen != 0 && ic->ic_set_tim != NULL)
916 		ic->ic_set_tim(ni, 0);
917 
918 	ni->ni_associd = 0;
919 	if (ni->ni_challenge != NULL) {
920 		kfree(ni->ni_challenge, M_DEVBUF);
921 		ni->ni_challenge = NULL;
922 	}
923 	/*
924 	 * Preserve SSID, WPA, and WME ie's so the bss node is
925 	 * reusable during a re-auth/re-assoc state transition.
926 	 * If we remove these data they will not be recreated
927 	 * because they come from a probe-response or beacon frame
928 	 * which cannot be expected prior to the association-response.
929 	 * This should not be an issue when operating in other modes
930 	 * as stations leaving always go through a full state transition
931 	 * which will rebuild this state.
932 	 *
933 	 * XXX does this leave us open to inheriting old state?
934 	 */
935 	for (i = 0; i < N(ni->ni_rxfrag); i++)
936 		if (ni->ni_rxfrag[i] != NULL) {
937 			m_freem(ni->ni_rxfrag[i]);
938 			ni->ni_rxfrag[i] = NULL;
939 		}
940 	/*
941 	 * Must be careful here to remove any key map entry w/o a LOR.
942 	 */
943 	ieee80211_node_delucastkey(ni);
944 #undef N
945 }
946 
947 static void
948 node_free(struct ieee80211_node *ni)
949 {
950 	struct ieee80211com *ic = ni->ni_ic;
951 
952 	ic->ic_node_cleanup(ni);
953 	if (ni->ni_wpa_ie != NULL)
954 		kfree(ni->ni_wpa_ie, M_DEVBUF);
955 	if (ni->ni_wme_ie != NULL)
956 		kfree(ni->ni_wme_ie, M_DEVBUF);
957 	IEEE80211_NODE_SAVEQ_DESTROY(ni);
958 	kfree(ni, M_80211_NODE);
959 }
960 
961 static uint8_t
962 node_getrssi(const struct ieee80211_node *ni)
963 {
964 	return ni->ni_rssi;
965 }
966 
967 static void
968 ieee80211_setup_node(struct ieee80211_node_table *nt,
969 	struct ieee80211_node *ni, const uint8_t *macaddr)
970 {
971 	struct ieee80211com *ic = nt->nt_ic;
972 	int hash;
973 
974 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
975 
976 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
977 		"%s %p<%6D> in %s table\n", __func__, ni,
978 		macaddr, ":", nt->nt_name);
979 
980 	IEEE80211_ADDR_COPY(ni->ni_macaddr, macaddr);
981 	hash = IEEE80211_NODE_HASH(macaddr);
982 	ieee80211_node_initref(ni);		/* mark referenced */
983 	ni->ni_chan = IEEE80211_CHAN_ANYC;
984 	ni->ni_authmode = IEEE80211_AUTH_OPEN;
985 	ni->ni_txpower = ic->ic_txpowlimit;	/* max power */
986 	ieee80211_crypto_resetkey(ic, &ni->ni_ucastkey, IEEE80211_KEYIX_NONE);
987 	ni->ni_inact_reload = nt->nt_inact_init;
988 	ni->ni_inact = ni->ni_inact_reload;
989 	IEEE80211_NODE_SAVEQ_INIT(ni, "unknown");
990 
991 	TAILQ_INSERT_TAIL(&nt->nt_node, ni, ni_list);
992 	LIST_INSERT_HEAD(&nt->nt_hash[hash], ni, ni_hash);
993 	ni->ni_table = nt;
994 	ni->ni_ic = ic;
995 
996 	ieee80211_ratectl_data_alloc(ni);
997 }
998 
999 struct ieee80211_node *
1000 ieee80211_alloc_node(struct ieee80211_node_table *nt, const uint8_t *macaddr)
1001 {
1002 	struct ieee80211com *ic = nt->nt_ic;
1003 	struct ieee80211_node *ni;
1004 
1005 	ni = ic->ic_node_alloc(nt);
1006 	if (ni != NULL)
1007 		ieee80211_setup_node(nt, ni, macaddr);
1008 	else
1009 		ic->ic_stats.is_rx_nodealloc++;
1010 	return ni;
1011 }
1012 
1013 /*
1014  * Craft a temporary node suitable for sending a management frame
1015  * to the specified station.  We craft only as much state as we
1016  * need to do the work since the node will be immediately reclaimed
1017  * once the send completes.
1018  */
1019 struct ieee80211_node *
1020 ieee80211_tmp_node(struct ieee80211com *ic, const uint8_t *macaddr)
1021 {
1022 	struct ieee80211_node *ni;
1023 
1024 	ni = ic->ic_node_alloc(&ic->ic_sta);
1025 	if (ni != NULL) {
1026 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1027 			"%s %p<%6D>\n", __func__, ni, macaddr, ":");
1028 
1029 		IEEE80211_ADDR_COPY(ni->ni_macaddr, macaddr);
1030 		IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_bss->ni_bssid);
1031 		ieee80211_node_initref(ni);		/* mark referenced */
1032 		ni->ni_txpower = ic->ic_bss->ni_txpower;
1033 		/* NB: required by ieee80211_fix_rate */
1034 		ieee80211_set_chan(ic, ni, ic->ic_bss->ni_chan);
1035 		ieee80211_crypto_resetkey(ic, &ni->ni_ucastkey,
1036 			IEEE80211_KEYIX_NONE);
1037 		/* XXX optimize away */
1038 		IEEE80211_NODE_SAVEQ_INIT(ni, "unknown");
1039 
1040 		ni->ni_table = NULL;		/* NB: pedantic */
1041 		ni->ni_ic = ic;
1042 	} else {
1043 		/* XXX msg */
1044 		ic->ic_stats.is_rx_nodealloc++;
1045 	}
1046 	return ni;
1047 }
1048 
1049 struct ieee80211_node *
1050 ieee80211_dup_bss(struct ieee80211_node_table *nt, const uint8_t *macaddr)
1051 {
1052 	struct ieee80211com *ic = nt->nt_ic;
1053 	struct ieee80211_node *ni;
1054 
1055 	ni = ic->ic_node_alloc(nt);
1056 	if (ni != NULL) {
1057 		ieee80211_setup_node(nt, ni, macaddr);
1058 		/*
1059 		 * Inherit from ic_bss.
1060 		 */
1061 		ni->ni_authmode = ic->ic_bss->ni_authmode;
1062 		ni->ni_txpower = ic->ic_bss->ni_txpower;
1063 		ni->ni_vlan = ic->ic_bss->ni_vlan;	/* XXX?? */
1064 		IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_bss->ni_bssid);
1065 		ieee80211_set_chan(ic, ni, ic->ic_bss->ni_chan);
1066 		ni->ni_rsn = ic->ic_bss->ni_rsn;
1067 	} else
1068 		ic->ic_stats.is_rx_nodealloc++;
1069 	return ni;
1070 }
1071 
1072 static struct ieee80211_node *
1073 #ifdef IEEE80211_DEBUG_REFCNT
1074 _ieee80211_find_node_debug(struct ieee80211_node_table *nt,
1075 	const uint8_t *macaddr, const char *func, int line)
1076 #else
1077 _ieee80211_find_node(struct ieee80211_node_table *nt,
1078 	const uint8_t *macaddr)
1079 #endif
1080 {
1081 	struct ieee80211_node *ni;
1082 	int hash;
1083 
1084 	hash = IEEE80211_NODE_HASH(macaddr);
1085 	LIST_FOREACH(ni, &nt->nt_hash[hash], ni_hash) {
1086 		if (IEEE80211_ADDR_EQ(ni->ni_macaddr, macaddr)) {
1087 			ieee80211_ref_node(ni);	/* mark referenced */
1088 #ifdef IEEE80211_DEBUG_REFCNT
1089 			IEEE80211_DPRINTF(nt->nt_ic, IEEE80211_MSG_NODE,
1090 			    "%s (%s:%u) %p<%6D> refcnt %d\n", __func__,
1091 			    func, line,
1092 			    ni, ni->ni_macaddr, ":",
1093 			    ieee80211_node_refcnt(ni));
1094 #endif
1095 			return ni;
1096 		}
1097 	}
1098 	return NULL;
1099 }
1100 #ifdef IEEE80211_DEBUG_REFCNT
1101 #define	_ieee80211_find_node(nt, mac) \
1102 	_ieee80211_find_node_debug(nt, mac, func, line)
1103 #endif
1104 
1105 struct ieee80211_node *
1106 #ifdef IEEE80211_DEBUG_REFCNT
1107 ieee80211_find_node_debug(struct ieee80211_node_table *nt,
1108 	const uint8_t *macaddr, const char *func, int line)
1109 #else
1110 ieee80211_find_node(struct ieee80211_node_table *nt, const uint8_t *macaddr)
1111 #endif
1112 {
1113 	struct ieee80211_node *ni;
1114 
1115 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
1116 
1117 	ni = _ieee80211_find_node(nt, macaddr);
1118 	return ni;
1119 }
1120 
1121 /*
1122  * Fake up a node; this handles node discovery in adhoc mode.
1123  * Note that for the driver's benefit we we treat this like
1124  * an association so the driver has an opportunity to setup
1125  * it's private state.
1126  */
1127 struct ieee80211_node *
1128 ieee80211_fakeup_adhoc_node(struct ieee80211_node_table *nt,
1129 	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1130 {
1131 	struct ieee80211com *ic = nt->nt_ic;
1132 	struct ieee80211_node *ni;
1133 
1134 	IEEE80211_DPRINTF(nt->nt_ic, IEEE80211_MSG_NODE,
1135 	    "%s: mac<%6D>\n", __func__, macaddr, ":");
1136 	ni = ieee80211_dup_bss(nt, macaddr);
1137 	if (ni != NULL) {
1138 		/* XXX no rate negotiation; just dup */
1139 		ni->ni_rates = ic->ic_bss->ni_rates;
1140 
1141 		ieee80211_ratectl_newassoc(ni, 1);
1142 
1143 		if (ic->ic_newassoc != NULL)
1144 			ic->ic_newassoc(ni, 1);
1145 
1146 		/* XXX not right for 802.1x/WPA */
1147 		ieee80211_node_authorize(ni);
1148 		if (ic->ic_opmode == IEEE80211_M_AHDEMO) {
1149 			/*
1150 			 * Blindly propagate capabilities based on the
1151 			 * local configuration.  In particular this permits
1152 			 * us to use QoS to disable ACK's.
1153 			 */
1154 			if (ic->ic_flags & IEEE80211_F_WME)
1155 				ni->ni_flags |= IEEE80211_NODE_QOS;
1156 		}
1157 	}
1158 	return ni;
1159 }
1160 
1161 #ifdef IEEE80211_DEBUG
1162 static void
1163 dump_probe_beacon(uint8_t subtype, int isnew,
1164 	const uint8_t mac[IEEE80211_ADDR_LEN],
1165 	const struct ieee80211_scanparams *sp)
1166 {
1167 
1168 	kprintf("[%6D] %s%s on chan %u (bss chan %u) ",
1169 	    mac, ":", isnew ? "new " : "",
1170 	    ieee80211_mgt_subtype_name[subtype >> IEEE80211_FC0_SUBTYPE_SHIFT],
1171 	    sp->chan, sp->bchan);
1172 	ieee80211_print_essid(sp->ssid + 2, sp->ssid[1]);
1173 	kprintf("\n");
1174 
1175 	if (isnew) {
1176 		kprintf("[%6D] caps 0x%x bintval %u erp 0x%x",
1177 			mac, ":", sp->capinfo, sp->bintval, sp->erp);
1178 		if (sp->country != NULL) {
1179 #if defined(__FreeBSD__) || defined(__DragonFly__)
1180 			kprintf(" country info %*D",
1181 				sp->country[1], sp->country+2, " ");
1182 #else
1183 			int i;
1184 			kprintf(" country info");
1185 			for (i = 0; i < sp->country[1]; i++)
1186 				kprintf(" %02x", sp->country[i+2]);
1187 #endif
1188 		}
1189 		kprintf("\n");
1190 	}
1191 }
1192 #endif /* IEEE80211_DEBUG */
1193 
1194 static void
1195 saveie(uint8_t **iep, const uint8_t *ie)
1196 {
1197 
1198 	if (ie == NULL)
1199 		*iep = NULL;
1200 	else
1201 		ieee80211_saveie(iep, ie);
1202 }
1203 
1204 /*
1205  * Process a beacon or probe response frame.
1206  */
1207 void
1208 ieee80211_add_scan(struct ieee80211com *ic,
1209 	const struct ieee80211_scanparams *sp,
1210 	const struct ieee80211_frame *wh,
1211 	int subtype, int rssi, int rstamp)
1212 {
1213 #define	ISPROBE(_st)	((_st) == IEEE80211_FC0_SUBTYPE_PROBE_RESP)
1214 	struct ieee80211_node_table *nt = &ic->ic_scan;
1215 	struct ieee80211_node *ni;
1216 	int newnode = 0;
1217 
1218 	ni = ieee80211_find_node(nt, wh->i_addr2);
1219 	if (ni == NULL) {
1220 		/*
1221 		 * Create a new entry.
1222 		 */
1223 		ni = ic->ic_node_alloc(nt);
1224 		if (ni == NULL) {
1225 			ic->ic_stats.is_rx_nodealloc++;
1226 			return;
1227 		}
1228 		ieee80211_setup_node(nt, ni, wh->i_addr2);
1229 		/*
1230 		 * XXX inherit from ic_bss.
1231 		 */
1232 		ni->ni_authmode = ic->ic_bss->ni_authmode;
1233 		ni->ni_txpower = ic->ic_bss->ni_txpower;
1234 		ni->ni_vlan = ic->ic_bss->ni_vlan;	/* XXX?? */
1235 		ieee80211_set_chan(ic, ni, ic->ic_curchan);
1236 		ni->ni_rsn = ic->ic_bss->ni_rsn;
1237 		newnode = 1;
1238 	}
1239 #ifdef IEEE80211_DEBUG
1240 	if (ieee80211_msg_scan(ic) && (ic->ic_flags & IEEE80211_F_SCAN))
1241 		dump_probe_beacon(subtype, newnode, wh->i_addr2, sp);
1242 #endif
1243 	/* XXX ap beaconing multiple ssid w/ same bssid */
1244 	if (sp->ssid[1] != 0 &&
1245 	    (ISPROBE(subtype) || ni->ni_esslen == 0)) {
1246 		ni->ni_esslen = sp->ssid[1];
1247 		memset(ni->ni_essid, 0, sizeof(ni->ni_essid));
1248 		memcpy(ni->ni_essid, sp->ssid + 2, sp->ssid[1]);
1249 	}
1250 	IEEE80211_ADDR_COPY(ni->ni_bssid, wh->i_addr3);
1251 	ni->ni_rssi = rssi;
1252 	ni->ni_rstamp = rstamp;
1253 	memcpy(ni->ni_tstamp.data, sp->tstamp, sizeof(ni->ni_tstamp));
1254 	ni->ni_intval = sp->bintval;
1255 	ni->ni_capinfo = sp->capinfo;
1256 	ni->ni_chan = &ic->ic_channels[sp->chan];
1257 	ni->ni_fhdwell = sp->fhdwell;
1258 	ni->ni_fhindex = sp->fhindex;
1259 	ni->ni_erp = sp->erp;
1260 	if (sp->tim != NULL) {
1261 		struct ieee80211_tim_ie *ie =
1262 		    (struct ieee80211_tim_ie *) sp->tim;
1263 
1264 		ni->ni_dtim_count = ie->tim_count;
1265 		ni->ni_dtim_period = ie->tim_period;
1266 	}
1267 	/*
1268 	 * Record the byte offset from the mac header to
1269 	 * the start of the TIM information element for
1270 	 * use by hardware and/or to speedup software
1271 	 * processing of beacon frames.
1272 	 */
1273 	ni->ni_timoff = sp->timoff;
1274 	/*
1275 	 * Record optional information elements that might be
1276 	 * used by applications or drivers.
1277 	 */
1278 	saveie(&ni->ni_wme_ie, sp->wme);
1279 	saveie(&ni->ni_wpa_ie, sp->wpa);
1280 
1281 	/* NB: must be after ni_chan is setup */
1282 	ieee80211_setup_rates(ni, sp->rates, sp->xrates, IEEE80211_F_DOSORT, 0);
1283 
1284 	if (!newnode)
1285 		ieee80211_free_node(ni);
1286 #undef ISPROBE
1287 }
1288 
1289 void
1290 ieee80211_init_neighbor(struct ieee80211_node *ni,
1291 	const struct ieee80211_frame *wh,
1292 	const struct ieee80211_scanparams *sp)
1293 {
1294 	IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1295 	    "%s: %p<%6D>\n", __func__, ni, ni->ni_macaddr, ":");
1296 	ni->ni_esslen = sp->ssid[1];
1297 	memcpy(ni->ni_essid, sp->ssid + 2, sp->ssid[1]);
1298 	IEEE80211_ADDR_COPY(ni->ni_bssid, wh->i_addr3);
1299 	memcpy(ni->ni_tstamp.data, sp->tstamp, sizeof(ni->ni_tstamp));
1300 	ni->ni_intval = sp->bintval;
1301 	ni->ni_capinfo = sp->capinfo;
1302 	ni->ni_chan = ni->ni_ic->ic_curchan;
1303 	ni->ni_fhdwell = sp->fhdwell;
1304 	ni->ni_fhindex = sp->fhindex;
1305 	ni->ni_erp = sp->erp;
1306 	ni->ni_timoff = sp->timoff;
1307 	if (sp->wme != NULL)
1308 		ieee80211_saveie(&ni->ni_wme_ie, sp->wme);
1309 	if (sp->wpa != NULL)
1310 		ieee80211_saveie(&ni->ni_wpa_ie, sp->wpa);
1311 
1312 	/* NB: must be after ni_chan is setup */
1313 	ieee80211_setup_rates(ni, sp->rates, sp->xrates,
1314 		IEEE80211_F_DOSORT | IEEE80211_F_DOFRATE |
1315 		IEEE80211_F_DONEGO | IEEE80211_F_DODEL, 0);
1316 	IEEE80211_PRINT_NODERATES(ni->ni_ic, ni, IEEE80211_MSG_NODE);
1317 }
1318 
1319 /*
1320  * Do node discovery in adhoc mode on receipt of a beacon
1321  * or probe response frame.  Note that for the driver's
1322  * benefit we we treat this like an association so the
1323  * driver has an opportunity to setup it's private state.
1324  */
1325 struct ieee80211_node *
1326 ieee80211_add_neighbor(struct ieee80211com *ic,
1327 	const struct ieee80211_frame *wh,
1328 	const struct ieee80211_scanparams *sp)
1329 {
1330 	struct ieee80211_node *ni;
1331 
1332 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1333 	    "%s: mac<%s>\n", __func__, wh->i_addr2, ":");
1334 	ni = ieee80211_dup_bss(&ic->ic_sta, wh->i_addr2);/* XXX alloc_node? */
1335 	if (ni != NULL) {
1336 		ieee80211_init_neighbor(ni, wh, sp);
1337 
1338 		ieee80211_ratectl_newassoc(ni, 1);
1339 
1340 		if (ic->ic_newassoc != NULL)
1341 			ic->ic_newassoc(ni, 1);
1342 
1343 		/* XXX not right for 802.1x/WPA */
1344 		ieee80211_node_authorize(ni);
1345 	}
1346 	return ni;
1347 }
1348 
1349 #define	IS_CTL(wh) \
1350 	((wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK) == IEEE80211_FC0_TYPE_CTL)
1351 #define	IS_PSPOLL(wh) \
1352 	((wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) == IEEE80211_FC0_SUBTYPE_PS_POLL)
1353 /*
1354  * Locate the node for sender, track state, and then pass the
1355  * (referenced) node up to the 802.11 layer for its use.  We
1356  * are required to pass some node so we fall back to ic_bss
1357  * when this frame is from an unknown sender.  The 802.11 layer
1358  * knows this means the sender wasn't in the node table and
1359  * acts accordingly.
1360  */
1361 struct ieee80211_node *
1362 #ifdef IEEE80211_DEBUG_REFCNT
1363 ieee80211_find_rxnode_debug(struct ieee80211com *ic,
1364 	const struct ieee80211_frame_min *wh, const char *func, int line)
1365 #else
1366 ieee80211_find_rxnode(struct ieee80211com *ic,
1367 	const struct ieee80211_frame_min *wh)
1368 #endif
1369 {
1370 	struct ieee80211_node_table *nt;
1371 	struct ieee80211_node *ni;
1372 
1373 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1374 
1375 	/* XXX may want scanned nodes in the neighbor table for adhoc */
1376 	if (ic->ic_opmode == IEEE80211_M_STA ||
1377 	    ic->ic_opmode == IEEE80211_M_MONITOR ||
1378 	    (ic->ic_flags & IEEE80211_F_SCAN))
1379 		nt = &ic->ic_scan;
1380 	else
1381 		nt = &ic->ic_sta;
1382 	/* XXX check ic_bss first in station mode */
1383 	/* XXX 4-address frames? */
1384 	if (IS_CTL(wh) && !IS_PSPOLL(wh) /*&& !IS_RTS(ah)*/)
1385 		ni = _ieee80211_find_node(nt, wh->i_addr1);
1386 	else
1387 		ni = _ieee80211_find_node(nt, wh->i_addr2);
1388 	if (ni == NULL)
1389 		ni = ieee80211_ref_node(ic->ic_bss);
1390 
1391 	return ni;
1392 }
1393 
1394 /*
1395  * Like ieee80211_find_rxnode but use the supplied h/w
1396  * key index as a hint to locate the node in the key
1397  * mapping table.  If an entry is present at the key
1398  * index we return it; otherwise do a normal lookup and
1399  * update the mapping table if the station has a unicast
1400  * key assigned to it.
1401  */
1402 struct ieee80211_node *
1403 #ifdef IEEE80211_DEBUG_REFCNT
1404 ieee80211_find_rxnode_withkey_debug(struct ieee80211com *ic,
1405 	const struct ieee80211_frame_min *wh, ieee80211_keyix keyix,
1406 	const char *func, int line)
1407 #else
1408 ieee80211_find_rxnode_withkey(struct ieee80211com *ic,
1409 	const struct ieee80211_frame_min *wh, ieee80211_keyix keyix)
1410 #endif
1411 {
1412 	struct ieee80211_node_table *nt;
1413 	struct ieee80211_node *ni;
1414 
1415 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1416 
1417 	if (ic->ic_opmode == IEEE80211_M_STA ||
1418 	    ic->ic_opmode == IEEE80211_M_MONITOR ||
1419 	    (ic->ic_flags & IEEE80211_F_SCAN))
1420 		nt = &ic->ic_scan;
1421 	else
1422 		nt = &ic->ic_sta;
1423 	if (nt->nt_keyixmap != NULL && keyix < nt->nt_keyixmax)
1424 		ni = nt->nt_keyixmap[keyix];
1425 	else
1426 		ni = NULL;
1427 	if (ni == NULL) {
1428 		if (IS_CTL(wh) && !IS_PSPOLL(wh) /*&& !IS_RTS(ah)*/)
1429 			ni = _ieee80211_find_node(nt, wh->i_addr1);
1430 		else
1431 			ni = _ieee80211_find_node(nt, wh->i_addr2);
1432 		if (ni == NULL)
1433 			ni = ieee80211_ref_node(ic->ic_bss);
1434 		if (nt->nt_keyixmap != NULL) {
1435 			/*
1436 			 * If the station has a unicast key cache slot
1437 			 * assigned update the key->node mapping table.
1438 			 */
1439 			keyix = ni->ni_ucastkey.wk_rxkeyix;
1440 			/* XXX can keyixmap[keyix] != NULL? */
1441 			if (keyix < nt->nt_keyixmax &&
1442 			    nt->nt_keyixmap[keyix] == NULL) {
1443 				IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1444 				    "%s: add key map entry %p<%6D> refcnt %d\n",
1445 				    __func__, ni, ni->ni_macaddr, ":",
1446 				    ieee80211_node_refcnt(ni)+1);
1447 				nt->nt_keyixmap[keyix] = ieee80211_ref_node(ni);
1448 			}
1449 		}
1450 	} else {
1451 		ieee80211_ref_node(ni);
1452 	}
1453 
1454 	return ni;
1455 }
1456 #undef IS_PSPOLL
1457 #undef IS_CTL
1458 
1459 /*
1460  * Return a reference to the appropriate node for sending
1461  * a data frame.  This handles node discovery in adhoc networks.
1462  */
1463 struct ieee80211_node *
1464 #ifdef IEEE80211_DEBUG_REFCNT
1465 ieee80211_find_txnode_debug(struct ieee80211com *ic, const uint8_t *macaddr,
1466 	const char *func, int line)
1467 #else
1468 ieee80211_find_txnode(struct ieee80211com *ic, const uint8_t *macaddr)
1469 #endif
1470 {
1471 	struct ieee80211_node_table *nt = &ic->ic_sta;
1472 	struct ieee80211_node *ni;
1473 
1474 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1475 
1476 	/*
1477 	 * The destination address should be in the node table
1478 	 * unless this is a multicast/broadcast frame.  We can
1479 	 * also optimize station mode operation, all frames go
1480 	 * to the bss node.
1481 	 */
1482 	if (ic->ic_opmode == IEEE80211_M_STA ||
1483 	    IEEE80211_IS_MULTICAST(macaddr)) {
1484 		ni = ieee80211_ref_node(ic->ic_bss);
1485 	} else {
1486 		ni = _ieee80211_find_node(nt, macaddr);
1487 		if (ic->ic_opmode == IEEE80211_M_HOSTAP &&
1488 		    (ni != NULL && ni->ni_associd == 0)) {
1489 			/*
1490 			 * Station is not associated; don't permit the
1491 			 * data frame to be sent by returning NULL.  This
1492 			 * is kinda a kludge but the least intrusive way
1493 			 * to add this check into all drivers.
1494 			 */
1495 			ieee80211_unref_node(&ni);      /* NB: null's ni */
1496 		}
1497 	}
1498 
1499 	if (ni == NULL) {
1500 		if (ic->ic_opmode == IEEE80211_M_IBSS ||
1501 		    ic->ic_opmode == IEEE80211_M_AHDEMO) {
1502 			/*
1503 			 * In adhoc mode cons up a node for the destination.
1504 			 * Note that we need an additional reference for the
1505 			 * caller to be consistent with _ieee80211_find_node.
1506 			 */
1507 			ni = ieee80211_fakeup_adhoc_node(nt, macaddr);
1508 			if (ni != NULL)
1509 				ieee80211_ref_node(ni);
1510 		} else {
1511 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_OUTPUT,
1512 				"[%6D] no node, discard frame (%s)\n",
1513 				macaddr, ":", __func__);
1514 			ic->ic_stats.is_tx_nonode++;
1515 		}
1516 	}
1517 	return ni;
1518 }
1519 
1520 /*
1521  * Like find but search based on the channel too.
1522  */
1523 struct ieee80211_node *
1524 #ifdef IEEE80211_DEBUG_REFCNT
1525 ieee80211_find_node_with_channel_debug(struct ieee80211_node_table *nt,
1526 	const uint8_t *macaddr, struct ieee80211_channel *chan,
1527 	const char *func, int line)
1528 #else
1529 ieee80211_find_node_with_channel(struct ieee80211_node_table *nt,
1530 	const uint8_t *macaddr, struct ieee80211_channel *chan)
1531 #endif
1532 {
1533 	struct ieee80211_node *ni;
1534 	int hash;
1535 
1536 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
1537 
1538 	hash = IEEE80211_NODE_HASH(macaddr);
1539 	LIST_FOREACH(ni, &nt->nt_hash[hash], ni_hash) {
1540 		if (IEEE80211_ADDR_EQ(ni->ni_macaddr, macaddr) &&
1541 		    ni->ni_chan == chan) {
1542 			ieee80211_ref_node(ni);		/* mark referenced */
1543 			IEEE80211_DPRINTF(nt->nt_ic, IEEE80211_MSG_NODE,
1544 			    REFCNT_LOC, ni, ni->ni_macaddr, ":",
1545 			    ieee80211_node_refcnt(ni));
1546 			break;
1547 		}
1548 	}
1549 	return ni;
1550 }
1551 
1552 /*
1553  * Like find but search based on the ssid too.
1554  */
1555 struct ieee80211_node *
1556 #ifdef IEEE80211_DEBUG_REFCNT
1557 ieee80211_find_node_with_ssid_debug(struct ieee80211_node_table *nt,
1558 	const uint8_t *macaddr, u_int ssidlen, const uint8_t *ssid,
1559 	const char *func, int line)
1560 #else
1561 ieee80211_find_node_with_ssid(struct ieee80211_node_table *nt,
1562 	const uint8_t *macaddr, u_int ssidlen, const uint8_t *ssid)
1563 #endif
1564 {
1565 #define	MATCH_SSID(ni, ssid, ssidlen) \
1566 	(ni->ni_esslen == ssidlen && memcmp(ni->ni_essid, ssid, ssidlen) == 0)
1567 	static const uint8_t zeromac[IEEE80211_ADDR_LEN];
1568 	struct ieee80211com *ic = nt->nt_ic;
1569 	struct ieee80211_node *ni;
1570 	int hash;
1571 
1572 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1573 
1574 	/*
1575 	 * A mac address that is all zero means match only the ssid;
1576 	 * otherwise we must match both.
1577 	 */
1578 	if (IEEE80211_ADDR_EQ(macaddr, zeromac)) {
1579 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
1580 			if (MATCH_SSID(ni, ssid, ssidlen))
1581 				break;
1582 		}
1583 	} else {
1584 		hash = IEEE80211_NODE_HASH(macaddr);
1585 		LIST_FOREACH(ni, &nt->nt_hash[hash], ni_hash) {
1586 			if (IEEE80211_ADDR_EQ(ni->ni_macaddr, macaddr) &&
1587 			    MATCH_SSID(ni, ssid, ssidlen))
1588 				break;
1589 		}
1590 	}
1591 	if (ni != NULL) {
1592 		ieee80211_ref_node(ni);	/* mark referenced */
1593 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1594 		     REFCNT_LOC, ni, ni->ni_macaddr, ":",
1595 		     ieee80211_node_refcnt(ni));
1596 	}
1597 	return ni;
1598 #undef MATCH_SSID
1599 }
1600 
1601 static void
1602 _ieee80211_free_node(struct ieee80211_node *ni)
1603 {
1604 	struct ieee80211com *ic = ni->ni_ic;
1605 	struct ieee80211_node_table *nt = ni->ni_table;
1606 
1607 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1608 		"%s %p<%6D> in %s table\n", __func__, ni,
1609 		ni->ni_macaddr, ":",
1610 		nt != NULL ? nt->nt_name : "<gone>");
1611 
1612 	ieee80211_ratectl_data_free(ni);
1613 
1614 	IEEE80211_AID_CLR(ni->ni_associd, ic->ic_aid_bitmap);
1615 	if (nt != NULL) {
1616 		TAILQ_REMOVE(&nt->nt_node, ni, ni_list);
1617 		LIST_REMOVE(ni, ni_hash);
1618 	}
1619 	ic->ic_node_free(ni);
1620 }
1621 
1622 void
1623 #ifdef IEEE80211_DEBUG_REFCNT
1624 ieee80211_free_node_debug(struct ieee80211_node *ni, const char *func, int line)
1625 #else
1626 ieee80211_free_node(struct ieee80211_node *ni)
1627 #endif
1628 {
1629 	struct ieee80211_node_table *nt = ni->ni_table;
1630 
1631 	ASSERT_SERIALIZED(ni->ni_ic->ic_ifp->if_serializer);
1632 
1633 #ifdef IEEE80211_DEBUG_REFCNT
1634 	IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1635 		"%s (%s:%u) %p<%6D> refcnt %d\n", __func__, func, line, ni,
1636 		 ni->ni_macaddr, ":", ieee80211_node_refcnt(ni) - 1);
1637 #endif
1638 	if (nt != NULL) {
1639 		if (ieee80211_node_dectestref(ni)) {
1640 			/*
1641 			 * Last reference, reclaim state.
1642 			 */
1643 			_ieee80211_free_node(ni);
1644 		} else if (ieee80211_node_refcnt(ni) == 1 &&
1645 		    nt->nt_keyixmap != NULL) {
1646 			ieee80211_keyix keyix;
1647 			/*
1648 			 * Check for a last reference in the key mapping table.
1649 			 */
1650 			keyix = ni->ni_ucastkey.wk_rxkeyix;
1651 			if (keyix < nt->nt_keyixmax &&
1652 			    nt->nt_keyixmap[keyix] == ni) {
1653 				IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1654 				    "%s: %p<%6D> clear key map entry", __func__,
1655 				    ni, ni->ni_macaddr, ":");
1656 				nt->nt_keyixmap[keyix] = NULL;
1657 				ieee80211_node_decref(ni); /* XXX needed? */
1658 				_ieee80211_free_node(ni);
1659 			}
1660 		}
1661 	} else {
1662 		if (ieee80211_node_dectestref(ni))
1663 			_ieee80211_free_node(ni);
1664 	}
1665 }
1666 
1667 /*
1668  * Reclaim a unicast key and clear any key cache state.
1669  */
1670 int
1671 ieee80211_node_delucastkey(struct ieee80211_node *ni)
1672 {
1673 	struct ieee80211com *ic = ni->ni_ic;
1674 	struct ieee80211_node_table *nt = &ic->ic_sta;
1675 	struct ieee80211_node *nikey;
1676 	ieee80211_keyix keyix;
1677 	int status;
1678 
1679 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1680 
1681 	keyix = ni->ni_ucastkey.wk_rxkeyix;
1682 	status = ieee80211_crypto_delkey(ic, &ni->ni_ucastkey);
1683 	if (nt->nt_keyixmap != NULL && keyix < nt->nt_keyixmax) {
1684 		nikey = nt->nt_keyixmap[keyix];
1685 		nt->nt_keyixmap[keyix] = NULL;
1686 	} else
1687 		nikey = NULL;
1688 
1689 	if (nikey != NULL) {
1690 		KASSERT(nikey == ni,
1691 			("key map out of sync, ni %p nikey %p", ni, nikey));
1692 		IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1693 			"%s: delete key map entry %p<%6D> refcnt %d\n",
1694 			__func__, ni, ni->ni_macaddr, ":",
1695 			ieee80211_node_refcnt(ni)-1);
1696 		ieee80211_free_node(ni);
1697 	}
1698 	return status;
1699 }
1700 
1701 /*
1702  * Reclaim a node.  If this is the last reference count then
1703  * do the normal free work.  Otherwise remove it from the node
1704  * table and mark it gone by clearing the back-reference.
1705  */
1706 static void
1707 node_reclaim(struct ieee80211_node_table *nt, struct ieee80211_node *ni)
1708 {
1709 	ieee80211_keyix keyix;
1710 
1711 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
1712 
1713 	IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1714 		"%s: remove %p<%6D> from %s table, refcnt %d\n",
1715 		__func__, ni, ni->ni_macaddr, ":",
1716 		nt->nt_name, ieee80211_node_refcnt(ni)-1);
1717 
1718 	/*
1719 	 * Clear any entry in the unicast key mapping table.
1720 	 * We need to do it here so rx lookups don't find it
1721 	 * in the mapping table even if it's not in the hash
1722 	 * table.  We cannot depend on the mapping table entry
1723 	 * being cleared because the node may not be free'd.
1724 	 */
1725 	keyix = ni->ni_ucastkey.wk_rxkeyix;
1726 	if (nt->nt_keyixmap != NULL && keyix < nt->nt_keyixmax &&
1727 	    nt->nt_keyixmap[keyix] == ni) {
1728 		IEEE80211_DPRINTF(ni->ni_ic, IEEE80211_MSG_NODE,
1729 			"%s: %p<%6D> clear key map entry\n",
1730 			__func__, ni, ni->ni_macaddr, ":");
1731 		nt->nt_keyixmap[keyix] = NULL;
1732 		ieee80211_node_decref(ni);	/* NB: don't need free */
1733 	}
1734 	if (!ieee80211_node_dectestref(ni)) {
1735 		/*
1736 		 * Other references are present, just remove the
1737 		 * node from the table so it cannot be found.  When
1738 		 * the references are dropped storage will be
1739 		 * reclaimed.
1740 		 */
1741 		TAILQ_REMOVE(&nt->nt_node, ni, ni_list);
1742 		LIST_REMOVE(ni, ni_hash);
1743 		ni->ni_table = NULL;		/* clear reference */
1744 
1745 		/*
1746 		 * XXX
1747 		 * We may want to put reclaimed node on <gone> table
1748 		 * so that ratectl modules can find them and free
1749 		 * the resources in their detach routines instead of
1750 		 * freeing the resources here.
1751 		 */
1752 		ieee80211_ratectl_data_free(ni);
1753 	} else
1754 		_ieee80211_free_node(ni);
1755 }
1756 
1757 static void
1758 ieee80211_free_allnodes(struct ieee80211_node_table *nt)
1759 {
1760 	struct ieee80211com *ic = nt->nt_ic;
1761 	struct ieee80211_node *ni;
1762 
1763 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1764 		"%s: free all nodes in %s table\n", __func__, nt->nt_name);
1765 
1766 	while ((ni = TAILQ_FIRST(&nt->nt_node)) != NULL) {
1767 		if (ni->ni_associd != 0) {
1768 			if (ic->ic_auth->ia_node_leave != NULL)
1769 				ic->ic_auth->ia_node_leave(ic, ni);
1770 			IEEE80211_AID_CLR(ni->ni_associd, ic->ic_aid_bitmap);
1771 		}
1772 		node_reclaim(nt, ni);
1773 	}
1774 	ieee80211_reset_erp(ic);
1775 }
1776 
1777 /*
1778  * Timeout entries in the scan cache.
1779  */
1780 static void
1781 ieee80211_timeout_scan_candidates(struct ieee80211_node_table *nt)
1782 {
1783 	struct ieee80211com *ic = nt->nt_ic;
1784 	struct ieee80211_node *ni, *tni;
1785 
1786 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1787 
1788 	ni = ic->ic_bss;
1789 	/* XXX belongs elsewhere */
1790 	if (ni->ni_rxfrag[0] != NULL && ticks > ni->ni_rxfragstamp + hz) {
1791 		m_freem(ni->ni_rxfrag[0]);
1792 		ni->ni_rxfrag[0] = NULL;
1793 	}
1794 	TAILQ_FOREACH_MUTABLE(ni, &nt->nt_node, ni_list, tni) {
1795 		if (ni->ni_inact && --ni->ni_inact == 0) {
1796 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
1797 			    "[%6D] scan candidate purged from cache "
1798 			    "(refcnt %u)\n", ni->ni_macaddr, ":",
1799 			    ieee80211_node_refcnt(ni));
1800 			node_reclaim(nt, ni);
1801 		}
1802 	}
1803 
1804 	nt->nt_inact_timer = IEEE80211_INACT_WAIT;
1805 }
1806 
1807 /*
1808  * Timeout inactive stations and do related housekeeping.
1809  * Note that we cannot hold the node lock while sending a
1810  * frame as this would lead to a LOR.  Instead we use a
1811  * generation number to mark nodes that we've scanned and
1812  * drop the lock and restart a scan if we have to time out
1813  * a node.  Since we are single-threaded by virtue of
1814  * controlling the inactivity timer we can be sure this will
1815  * process each node only once.
1816  */
1817 static void
1818 ieee80211_timeout_stations(struct ieee80211_node_table *nt)
1819 {
1820 	struct ieee80211com *ic = nt->nt_ic;
1821 	struct ieee80211_node *ni, *next;
1822 	int isadhoc;
1823 
1824 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
1825 
1826 	isadhoc = (ic->ic_opmode == IEEE80211_M_IBSS ||
1827 		   ic->ic_opmode == IEEE80211_M_AHDEMO);
1828 
1829 	TAILQ_FOREACH_MUTABLE(ni, &nt->nt_node, ni_list, next) {
1830 		/*
1831 		 * Ignore entries for which have yet to receive an
1832 		 * authentication frame.  These are transient and
1833 		 * will be reclaimed when the last reference to them
1834 		 * goes away (when frame xmits complete).
1835 		 */
1836 		if (ic->ic_opmode == IEEE80211_M_HOSTAP &&
1837 		    (ni->ni_flags & IEEE80211_NODE_AREF) == 0)
1838 			continue;
1839 		/*
1840 		 * Free fragment if not needed anymore
1841 		 * (last fragment older than 1s).
1842 		 * XXX doesn't belong here
1843 		 */
1844 		if (ni->ni_rxfrag[0] != NULL &&
1845 		    ticks > ni->ni_rxfragstamp + hz) {
1846 			m_freem(ni->ni_rxfrag[0]);
1847 			ni->ni_rxfrag[0] = NULL;
1848 		}
1849 		/*
1850 		 * Special case ourself; we may be idle for extended periods
1851 		 * of time and regardless reclaiming our state is wrong.
1852 		 */
1853 		if (ni == ic->ic_bss)
1854 			continue;
1855 		ni->ni_inact--;
1856 		if (ni->ni_associd != 0 || isadhoc) {
1857 			/*
1858 			 * Age frames on the power save queue. The
1859 			 * aging interval is 4 times the listen
1860 			 * interval specified by the station.  This
1861 			 * number is factored into the age calculations
1862 			 * when the frame is placed on the queue.  We
1863 			 * store ages as time differences we can check
1864 			 * and/or adjust only the head of the list.
1865 			 */
1866 			if (IEEE80211_NODE_SAVEQ_QLEN(ni) != 0) {
1867 				struct mbuf *m;
1868 				int discard = 0;
1869 
1870 				while (IF_POLL(&ni->ni_savedq, m) != NULL &&
1871 				     M_AGE_GET(m) < IEEE80211_INACT_WAIT) {
1872 					IEEE80211_DPRINTF(ic,
1873 					    IEEE80211_MSG_POWER,
1874 					    "[%6D] discard frame, age %u\n",
1875 					    ni->ni_macaddr, ":",
1876 					    M_AGE_GET(m));/*XXX*/
1877 					_IEEE80211_NODE_SAVEQ_DEQUEUE_HEAD(ni, m);
1878 					m_freem(m);
1879 					discard++;
1880 				}
1881 				if (m != NULL)
1882 					M_AGE_SUB(m, IEEE80211_INACT_WAIT);
1883 
1884 				if (discard != 0) {
1885 					IEEE80211_DPRINTF(ic,
1886 					    IEEE80211_MSG_POWER,
1887 					    "[%6D] discard %u frames for age\n",
1888 					    ni->ni_macaddr, ":",
1889 					    discard);
1890 					IEEE80211_NODE_STAT_ADD(ni,
1891 						ps_discard, discard);
1892 					if (IEEE80211_NODE_SAVEQ_QLEN(ni) == 0)
1893 						ic->ic_set_tim(ni, 0);
1894 				}
1895 			}
1896 			/*
1897 			 * Probe the station before time it out.  We
1898 			 * send a null data frame which may not be
1899 			 * universally supported by drivers (need it
1900 			 * for ps-poll support so it should be...).
1901 			 */
1902 			if (0 < ni->ni_inact &&
1903 			    ni->ni_inact <= ic->ic_inact_probe) {
1904 				IEEE80211_NOTE(ic,
1905 				    IEEE80211_MSG_INACT | IEEE80211_MSG_NODE,
1906 				    ni, "%s",
1907 				    "probe station due to inactivity");
1908 				/*
1909 				 * Grab a reference before unlocking the table
1910 				 * so the node cannot be reclaimed before we
1911 				 * send the frame. ieee80211_send_nulldata
1912 				 * understands we've done this and reclaims the
1913 				 * ref for us as needed.
1914 				 */
1915 				ieee80211_ref_node(ni);
1916 				ieee80211_send_nulldata(ni);
1917 				/* XXX stat? */
1918 				continue;
1919 			}
1920 		}
1921 		if (ni->ni_inact <= 0) {
1922 			IEEE80211_NOTE(ic,
1923 			    IEEE80211_MSG_INACT | IEEE80211_MSG_NODE, ni,
1924 			    "station timed out due to inactivity "
1925 			    "(refcnt %u)", ieee80211_node_refcnt(ni));
1926 			/*
1927 			 * Send a deauthenticate frame and drop the station.
1928 			 * This is somewhat complicated due to reference counts
1929 			 * and locking.  At this point a station will typically
1930 			 * have a reference count of 1.  ieee80211_node_leave
1931 			 * will do a "free" of the node which will drop the
1932 			 * reference count.  But in the meantime a reference
1933 			 * wil be held by the deauth frame.  The actual reclaim
1934 			 * of the node will happen either after the tx is
1935 			 * completed or by ieee80211_node_leave.
1936 			 */
1937 			if (ni->ni_associd != 0) {
1938 				IEEE80211_SEND_MGMT(ic, ni,
1939 				    IEEE80211_FC0_SUBTYPE_DEAUTH,
1940 				    IEEE80211_REASON_AUTH_EXPIRE);
1941 			}
1942 			ieee80211_node_leave(ic, ni);
1943 			ic->ic_stats.is_node_timeout++;
1944 			continue;
1945 		}
1946 	}
1947 
1948 	nt->nt_inact_timer = IEEE80211_INACT_WAIT;
1949 }
1950 
1951 void
1952 ieee80211_iterate_nodes(struct ieee80211_node_table *nt, ieee80211_iter_func *f, void *arg)
1953 {
1954 	struct ieee80211_node *ni, *next;
1955 
1956 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
1957 
1958 	TAILQ_FOREACH_MUTABLE(ni, &nt->nt_node, ni_list, next)
1959 		f(arg, ni);
1960 }
1961 
1962 void
1963 ieee80211_dump_node(struct ieee80211_node_table *nt, struct ieee80211_node *ni)
1964 {
1965 	kprintf("0x%p: mac %6D refcnt %d\n", ni,
1966 		ni->ni_macaddr, ":", ieee80211_node_refcnt(ni));
1967 	kprintf("\tauthmode %u flags 0x%x\n",
1968 		ni->ni_authmode, ni->ni_flags);
1969 	kprintf("\tassocid 0x%x txpower %u vlan %u\n",
1970 		ni->ni_associd, ni->ni_txpower, ni->ni_vlan);
1971 	kprintf("\ttxseq %u rxseq %u fragno %u rxfragstamp %u\n",
1972 		ni->ni_txseqs[0],
1973 		ni->ni_rxseqs[0] >> IEEE80211_SEQ_SEQ_SHIFT,
1974 		ni->ni_rxseqs[0] & IEEE80211_SEQ_FRAG_MASK,
1975 		ni->ni_rxfragstamp);
1976 	kprintf("\trstamp %u rssi %u intval %u capinfo 0x%x\n",
1977 		ni->ni_rstamp, ni->ni_rssi, ni->ni_intval, ni->ni_capinfo);
1978 	kprintf("\tbssid %6D essid \"%.*s\" channel %u:0x%x\n",
1979 		ni->ni_bssid, ":",
1980 		ni->ni_esslen, ni->ni_essid,
1981 		ni->ni_chan->ic_freq, ni->ni_chan->ic_flags);
1982 	kprintf("\tfails %u inact %u txrate %u\n",
1983 		ni->ni_fails, ni->ni_inact, ni->ni_txrate);
1984 }
1985 
1986 void
1987 ieee80211_dump_nodes(struct ieee80211_node_table *nt)
1988 {
1989 	ieee80211_iterate_nodes(nt,
1990 		(ieee80211_iter_func *) ieee80211_dump_node, nt);
1991 }
1992 
1993 /*
1994  * Handle a station joining an 11g network.
1995  */
1996 static void
1997 ieee80211_node_join_11g(struct ieee80211com *ic, struct ieee80211_node *ni)
1998 {
1999 
2000 	/*
2001 	 * Station isn't capable of short slot time.  Bump
2002 	 * the count of long slot time stations and disable
2003 	 * use of short slot time.  Note that the actual switch
2004 	 * over to long slot time use may not occur until the
2005 	 * next beacon transmission (per sec. 7.3.1.4 of 11g).
2006 	 */
2007 	if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME) == 0) {
2008 		ic->ic_longslotsta++;
2009 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2010 		    "[%6D] station needs long slot time, count %d\n",
2011 		    ni->ni_macaddr, ":", ic->ic_longslotsta);
2012 		/* XXX vap's w/ conflicting needs won't work */
2013 		ieee80211_set_shortslottime(ic, 0);
2014 	}
2015 	/*
2016 	 * If the new station is not an ERP station
2017 	 * then bump the counter and enable protection
2018 	 * if configured.
2019 	 */
2020 	if (!ieee80211_iserp_rateset(ic, &ni->ni_rates)) {
2021 		ic->ic_nonerpsta++;
2022 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2023 		    "[%6D] station is !ERP, %d non-ERP stations associated\n",
2024 		    ni->ni_macaddr, ":", ic->ic_nonerpsta);
2025 		/*
2026 		 * If protection is configured, enable it.
2027 		 */
2028 		if (ic->ic_protmode != IEEE80211_PROT_NONE) {
2029 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2030 			    "%s: enable use of protection\n", __func__);
2031 			ic->ic_flags |= IEEE80211_F_USEPROT;
2032 		}
2033 		/*
2034 		 * If station does not support short preamble
2035 		 * then we must enable use of Barker preamble.
2036 		 */
2037 		if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE) == 0) {
2038 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2039 			    "[%6D] station needs long preamble\n",
2040 			    ni->ni_macaddr, ":");
2041 			ieee80211_set_shortpreamble(ic, 0);
2042 		}
2043 		if (ic->ic_nonerpsta == 1)
2044 			ic->ic_flags_ext |= IEEE80211_FEXT_ERPUPDATE;
2045 	} else
2046 		ni->ni_flags |= IEEE80211_NODE_ERP;
2047 }
2048 
2049 void
2050 ieee80211_node_join(struct ieee80211com *ic, struct ieee80211_node *ni, int resp)
2051 {
2052 	int newassoc;
2053 
2054 	if (ni->ni_associd == 0) {
2055 		uint16_t aid;
2056 
2057 		/*
2058 		 * It would be good to search the bitmap
2059 		 * more efficiently, but this will do for now.
2060 		 */
2061 		for (aid = 1; aid < ic->ic_max_aid; aid++) {
2062 			if (!IEEE80211_AID_ISSET(aid,
2063 			    ic->ic_aid_bitmap))
2064 				break;
2065 		}
2066 		if (aid >= ic->ic_max_aid) {
2067 			IEEE80211_SEND_MGMT(ic, ni, resp,
2068 			    IEEE80211_REASON_ASSOC_TOOMANY);
2069 			ieee80211_node_leave(ic, ni);
2070 			return;
2071 		}
2072 		ni->ni_associd = aid | 0xc000;
2073 		IEEE80211_AID_SET(ni->ni_associd, ic->ic_aid_bitmap);
2074 		ic->ic_sta_assoc++;
2075 		newassoc = 1;
2076 		if (ic->ic_curmode == IEEE80211_MODE_11G)
2077 			ieee80211_node_join_11g(ic, ni);
2078 	} else
2079 		newassoc = 0;
2080 
2081 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG,
2082 	    "[%6D] station %sassociated at aid %d: %s preamble, %s slot time%s%s\n",
2083 	    ni->ni_macaddr, ":", newassoc ? "" : "re",
2084 	    IEEE80211_NODE_AID(ni),
2085 	    ic->ic_flags & IEEE80211_F_SHPREAMBLE ? "short" : "long",
2086 	    ic->ic_flags & IEEE80211_F_SHSLOT ? "short" : "long",
2087 	    ic->ic_flags & IEEE80211_F_USEPROT ? ", protection" : "",
2088 	    ni->ni_flags & IEEE80211_NODE_QOS ? ", QoS" : ""
2089 	);
2090 
2091 	IEEE80211_PRINT_NODERATES(ic, ni,
2092 		IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG);
2093 
2094 	ieee80211_ratectl_newassoc(ni, newassoc);
2095 
2096 	/* give driver a chance to setup state like ni_txrate */
2097 	if (ic->ic_newassoc != NULL)
2098 		ic->ic_newassoc(ni, newassoc);
2099 
2100 	ni->ni_inact_reload = ic->ic_inact_auth;
2101 	ni->ni_inact = ni->ni_inact_reload;
2102 	IEEE80211_SEND_MGMT(ic, ni, resp, IEEE80211_STATUS_SUCCESS);
2103 	/* tell the authenticator about new station */
2104 	if (ic->ic_auth->ia_node_join != NULL)
2105 		ic->ic_auth->ia_node_join(ic, ni);
2106 	ieee80211_notify_node_join(ic, ni, newassoc);
2107 }
2108 
2109 /*
2110  * Handle a station leaving an 11g network.
2111  */
2112 static void
2113 ieee80211_node_leave_11g(struct ieee80211com *ic, struct ieee80211_node *ni)
2114 {
2115 
2116 	KASSERT(ic->ic_curmode == IEEE80211_MODE_11G,
2117 	     ("not in 11g, bss %u:0x%x, curmode %u", ni->ni_chan->ic_freq,
2118 	      ni->ni_chan->ic_flags, ic->ic_curmode));
2119 
2120 	/*
2121 	 * If a long slot station do the slot time bookkeeping.
2122 	 */
2123 	if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME) == 0) {
2124 		KASSERT(ic->ic_longslotsta > 0,
2125 		    ("bogus long slot station count %d", ic->ic_longslotsta));
2126 		ic->ic_longslotsta--;
2127 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2128 		    "[%6D] long slot time station leaves, count now %d\n",
2129 		    ni->ni_macaddr, ":", ic->ic_longslotsta);
2130 		if (ic->ic_longslotsta == 0) {
2131 			/*
2132 			 * Re-enable use of short slot time if supported
2133 			 * and not operating in IBSS mode (per spec).
2134 			 */
2135 			if ((ic->ic_caps & IEEE80211_C_SHSLOT) &&
2136 			    ic->ic_opmode != IEEE80211_M_IBSS) {
2137 				IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2138 				    "%s: re-enable use of short slot time\n",
2139 				    __func__);
2140 				ieee80211_set_shortslottime(ic, 1);
2141 			}
2142 		}
2143 	}
2144 	/*
2145 	 * If a non-ERP station do the protection-related bookkeeping.
2146 	 */
2147 	if ((ni->ni_flags & IEEE80211_NODE_ERP) == 0) {
2148 		KASSERT(ic->ic_nonerpsta > 0,
2149 		    ("bogus non-ERP station count %d", ic->ic_nonerpsta));
2150 		ic->ic_nonerpsta--;
2151 		IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2152 		    "[%6D] non-ERP station leaves, count now %d\n",
2153 		    ni->ni_macaddr, ":", ic->ic_nonerpsta);
2154 		if (ic->ic_nonerpsta == 0) {
2155 			IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2156 				"%s: disable use of protection\n", __func__);
2157 			ic->ic_flags &= ~IEEE80211_F_USEPROT;
2158 			/* XXX verify mode? */
2159 			if (ic->ic_caps & IEEE80211_C_SHPREAMBLE) {
2160 				IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC,
2161 				    "%s: re-enable use of short preamble\n",
2162 				    __func__);
2163 				ieee80211_set_shortpreamble(ic, 1);
2164 			}
2165 			ic->ic_flags_ext |= IEEE80211_FEXT_ERPUPDATE;
2166 		}
2167 	}
2168 }
2169 
2170 /*
2171  * Handle bookkeeping for station deauthentication/disassociation
2172  * when operating as an ap.
2173  */
2174 void
2175 ieee80211_node_leave(struct ieee80211com *ic, struct ieee80211_node *ni)
2176 {
2177 	struct ieee80211_node_table *nt = ni->ni_table;
2178 
2179 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
2180 
2181 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG,
2182 	    "[%6D] station with aid %d leaves\n",
2183 	    ni->ni_macaddr, ":", IEEE80211_NODE_AID(ni));
2184 
2185 	KASSERT(ic->ic_opmode == IEEE80211_M_HOSTAP ||
2186 		ic->ic_opmode == IEEE80211_M_IBSS ||
2187 		ic->ic_opmode == IEEE80211_M_AHDEMO,
2188 		("unexpected operating mode %u", ic->ic_opmode));
2189 	/*
2190 	 * If node wasn't previously associated all
2191 	 * we need to do is reclaim the reference.
2192 	 */
2193 	/* XXX ibss mode bypasses 11g and notification */
2194 	if (ni->ni_associd == 0)
2195 		goto done;
2196 	/*
2197 	 * Tell the authenticator the station is leaving.
2198 	 * Note that we must do this before yanking the
2199 	 * association id as the authenticator uses the
2200 	 * associd to locate it's state block.
2201 	 */
2202 	if (ic->ic_auth->ia_node_leave != NULL)
2203 		ic->ic_auth->ia_node_leave(ic, ni);
2204 	IEEE80211_AID_CLR(ni->ni_associd, ic->ic_aid_bitmap);
2205 	ni->ni_associd = 0;
2206 	ic->ic_sta_assoc--;
2207 
2208 	if (ic->ic_curmode == IEEE80211_MODE_11G)
2209 		ieee80211_node_leave_11g(ic, ni);
2210 	/*
2211 	 * Cleanup station state.  In particular clear various
2212 	 * state that might otherwise be reused if the node
2213 	 * is reused before the reference count goes to zero
2214 	 * (and memory is reclaimed).
2215 	 */
2216 	ieee80211_sta_leave(ic, ni);
2217 done:
2218 	/*
2219 	 * Remove the node from any table it's recorded in and
2220 	 * drop the caller's reference.  Removal from the table
2221 	 * is important to insure the node is not reprocessed
2222 	 * for inactivity.
2223 	 */
2224 	if (nt != NULL)
2225 		node_reclaim(nt, ni);
2226 	else
2227 		ieee80211_free_node(ni);
2228 }
2229 
2230 uint8_t
2231 ieee80211_getrssi(struct ieee80211com *ic)
2232 {
2233 #define	NZ(x)	((x) == 0 ? 1 : (x))
2234 	struct ieee80211_node_table *nt = &ic->ic_sta;
2235 	uint32_t rssi_samples, rssi_total;
2236 	struct ieee80211_node *ni;
2237 
2238 	rssi_total = 0;
2239 	rssi_samples = 0;
2240 	switch (ic->ic_opmode) {
2241 	case IEEE80211_M_IBSS:		/* average of all ibss neighbors */
2242 		/* XXX locking */
2243 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list)
2244 			if (ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) {
2245 				rssi_samples++;
2246 				rssi_total += ic->ic_node_getrssi(ni);
2247 			}
2248 		break;
2249 	case IEEE80211_M_AHDEMO:	/* average of all neighbors */
2250 		/* XXX locking */
2251 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
2252 			rssi_samples++;
2253 			rssi_total += ic->ic_node_getrssi(ni);
2254 		}
2255 		break;
2256 	case IEEE80211_M_HOSTAP:	/* average of all associated stations */
2257 		/* XXX locking */
2258 		TAILQ_FOREACH(ni, &nt->nt_node, ni_list)
2259 			if (IEEE80211_AID(ni->ni_associd) != 0) {
2260 				rssi_samples++;
2261 				rssi_total += ic->ic_node_getrssi(ni);
2262 			}
2263 		break;
2264 	case IEEE80211_M_MONITOR:	/* XXX */
2265 	case IEEE80211_M_STA:		/* use stats from associated ap */
2266 	default:
2267 		if (ic->ic_bss != NULL)
2268 			rssi_total = ic->ic_node_getrssi(ic->ic_bss);
2269 		rssi_samples = 1;
2270 		break;
2271 	}
2272 	return rssi_total / NZ(rssi_samples);
2273 #undef NZ
2274 }
2275 
2276 /*
2277  * Indicate whether there are frames queued for a station in power-save mode.
2278  */
2279 static void
2280 ieee80211_set_tim(struct ieee80211_node *ni, int set)
2281 {
2282 	struct ieee80211com *ic = ni->ni_ic;
2283 	uint16_t aid;
2284 
2285 	ASSERT_SERIALIZED(ic->ic_ifp->if_serializer);
2286 
2287 	KASSERT(ic->ic_opmode == IEEE80211_M_HOSTAP ||
2288 		ic->ic_opmode == IEEE80211_M_IBSS,
2289 		("operating mode %u", ic->ic_opmode));
2290 
2291 	aid = IEEE80211_AID(ni->ni_associd);
2292 	KASSERT(aid < ic->ic_max_aid,
2293 		("bogus aid %u, max %u", aid, ic->ic_max_aid));
2294 
2295 	if (set != (isset(ic->ic_tim_bitmap, aid) != 0)) {
2296 		if (set) {
2297 			setbit(ic->ic_tim_bitmap, aid);
2298 			ic->ic_ps_pending++;
2299 		} else {
2300 			clrbit(ic->ic_tim_bitmap, aid);
2301 			ic->ic_ps_pending--;
2302 		}
2303 		ic->ic_flags |= IEEE80211_F_TIMUPDATE;
2304 	}
2305 }
2306 
2307 /*
2308  * Node table support.
2309  */
2310 
2311 static void
2312 ieee80211_node_table_init(struct ieee80211com *ic,
2313 	struct ieee80211_node_table *nt,
2314 	const char *name, int inact, int keyixmax,
2315 	void (*timeout)(struct ieee80211_node_table *))
2316 {
2317 	IEEE80211_DPRINTF(ic, IEEE80211_MSG_NODE,
2318 		"%s %s table, inact %u\n", __func__, name, inact);
2319 
2320 	nt->nt_ic = ic;
2321 	TAILQ_INIT(&nt->nt_node);
2322 	nt->nt_name = name;
2323 	nt->nt_inact_init = inact;
2324 	nt->nt_timeout = timeout;
2325 	nt->nt_keyixmax = keyixmax;
2326 	if (nt->nt_keyixmax > 0) {
2327 		nt->nt_keyixmap =
2328 			kmalloc(keyixmax * sizeof(struct ieee80211_node *),
2329 			       M_80211_NODE, M_WAITOK | M_ZERO);
2330 	} else {
2331 		nt->nt_keyixmap = NULL;
2332 	}
2333 }
2334 
2335 void
2336 ieee80211_node_table_reset(struct ieee80211_node_table *nt)
2337 {
2338 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
2339 
2340 	IEEE80211_DPRINTF(nt->nt_ic, IEEE80211_MSG_NODE,
2341 		"%s %s table\n", __func__, nt->nt_name);
2342 
2343 	nt->nt_inact_timer = 0;
2344 	ieee80211_free_allnodes(nt);
2345 }
2346 
2347 static void
2348 ieee80211_node_table_cleanup(struct ieee80211_node_table *nt)
2349 {
2350 	ASSERT_SERIALIZED(nt->nt_ic->ic_ifp->if_serializer);
2351 
2352 	IEEE80211_DPRINTF(nt->nt_ic, IEEE80211_MSG_NODE,
2353 		"%s %s table\n", __func__, nt->nt_name);
2354 
2355 	ieee80211_free_allnodes(nt);
2356 	if (nt->nt_keyixmap != NULL) {
2357 		/* XXX verify all entries are NULL */
2358 		int i;
2359 		for (i = 0; i < nt->nt_keyixmax; i++)
2360 			if (nt->nt_keyixmap[i] != NULL) {
2361 				kprintf("%s: %s[%u] still active\n", __func__,
2362 				       nt->nt_name, i);
2363 			}
2364 		kfree(nt->nt_keyixmap, M_80211_NODE);
2365 		nt->nt_keyixmap = NULL;
2366 	}
2367 }
2368 
2369 /*
2370  * Update short preamble state
2371  */
2372 void
2373 ieee80211_update_shpreamble(struct ieee80211com *ic,
2374 			    const struct ieee80211_node *ni)
2375 {
2376 	int shpreamble = 0;
2377 
2378 	switch (ic->ic_curmode) {
2379 	case IEEE80211_MODE_11A:
2380 		shpreamble = 1;
2381 		break;
2382 	case IEEE80211_MODE_11G:
2383 		if (ni->ni_erp & IEEE80211_ERP_LONG_PREAMBLE) {
2384 			/*
2385 			 * According to IEEE Std 802.11g-2003 subclause
2386 			 * 7.3.2.13, page 10:
2387 			 * Short preamble should not be used, if barker
2388 			 * preamble mode bit is 1 in ERP informarion,
2389 			 * _regardless_ of the short preamble bit in
2390 			 * capability information.
2391 			 */
2392 			break;
2393 		}
2394 		/* FALL THROUGH */
2395 	default:
2396 		if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE) &&
2397 		    (ic->ic_caps & IEEE80211_C_SHPREAMBLE))
2398 			shpreamble = 1;
2399 		break;
2400 	}
2401 	ieee80211_set_shortpreamble(ic, shpreamble);
2402 }
2403