ieee80211_node.c revision 183255
1/*-
2 * Copyright (c) 2001 Atsushi Onoe
3 * Copyright (c) 2002-2008 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 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: head/sys/net80211/ieee80211_node.c 183255 2008-09-21 23:59:14Z sam $");
29
30#include "opt_wlan.h"
31
32#include <sys/param.h>
33#include <sys/systm.h>
34#include <sys/mbuf.h>
35#include <sys/malloc.h>
36#include <sys/kernel.h>
37
38#include <sys/socket.h>
39
40#include <net/if.h>
41#include <net/if_media.h>
42#include <net/ethernet.h>
43
44#include <net80211/ieee80211_var.h>
45#include <net80211/ieee80211_input.h>
46#include <net80211/ieee80211_wds.h>
47
48#include <net/bpf.h>
49
50/*
51 * Association id's are managed with a bit vector.
52 */
53#define	IEEE80211_AID_SET(_vap, b) \
54	((_vap)->iv_aid_bitmap[IEEE80211_AID(b) / 32] |= \
55		(1 << (IEEE80211_AID(b) % 32)))
56#define	IEEE80211_AID_CLR(_vap, b) \
57	((_vap)->iv_aid_bitmap[IEEE80211_AID(b) / 32] &= \
58		~(1 << (IEEE80211_AID(b) % 32)))
59#define	IEEE80211_AID_ISSET(_vap, b) \
60	((_vap)->iv_aid_bitmap[IEEE80211_AID(b) / 32] & (1 << (IEEE80211_AID(b) % 32)))
61
62#ifdef IEEE80211_DEBUG_REFCNT
63#define REFCNT_LOC "%s (%s:%u) %p<%s> refcnt %d\n", __func__, func, line
64#else
65#define REFCNT_LOC "%s %p<%s> refcnt %d\n", __func__
66#endif
67
68static int ieee80211_sta_join1(struct ieee80211_node *);
69
70static struct ieee80211_node *node_alloc(struct ieee80211vap *,
71	const uint8_t [IEEE80211_ADDR_LEN]);
72static void node_cleanup(struct ieee80211_node *);
73static void node_free(struct ieee80211_node *);
74static void node_age(struct ieee80211_node *);
75static int8_t node_getrssi(const struct ieee80211_node *);
76static void node_getsignal(const struct ieee80211_node *, int8_t *, int8_t *);
77static void node_getmimoinfo(const struct ieee80211_node *,
78	struct ieee80211_mimo_info *);
79
80static void _ieee80211_free_node(struct ieee80211_node *);
81
82static void ieee80211_node_table_init(struct ieee80211com *ic,
83	struct ieee80211_node_table *nt, const char *name,
84	int inact, int keymaxix);
85static void ieee80211_node_table_reset(struct ieee80211_node_table *,
86	struct ieee80211vap *);
87static void ieee80211_node_reclaim(struct ieee80211_node *);
88static void ieee80211_node_table_cleanup(struct ieee80211_node_table *nt);
89static void ieee80211_erp_timeout(struct ieee80211com *);
90
91MALLOC_DEFINE(M_80211_NODE, "80211node", "802.11 node state");
92MALLOC_DEFINE(M_80211_NODE_IE, "80211nodeie", "802.11 node ie");
93
94void
95ieee80211_node_attach(struct ieee80211com *ic)
96{
97	ieee80211_node_table_init(ic, &ic->ic_sta, "station",
98		IEEE80211_INACT_INIT, ic->ic_max_keyix);
99	callout_init(&ic->ic_inact, CALLOUT_MPSAFE);
100	callout_reset(&ic->ic_inact, IEEE80211_INACT_WAIT*hz,
101		ieee80211_node_timeout, ic);
102
103	ic->ic_node_alloc = node_alloc;
104	ic->ic_node_free = node_free;
105	ic->ic_node_cleanup = node_cleanup;
106	ic->ic_node_age = node_age;
107	ic->ic_node_drain = node_age;		/* NB: same as age */
108	ic->ic_node_getrssi = node_getrssi;
109	ic->ic_node_getsignal = node_getsignal;
110	ic->ic_node_getmimoinfo = node_getmimoinfo;
111
112	/*
113	 * Set flags to be propagated to all vap's;
114	 * these define default behaviour/configuration.
115	 */
116	ic->ic_flags_ext |= IEEE80211_FEXT_INACT; /* inactivity processing */
117}
118
119void
120ieee80211_node_detach(struct ieee80211com *ic)
121{
122
123	callout_drain(&ic->ic_inact);
124	ieee80211_node_table_cleanup(&ic->ic_sta);
125}
126
127void
128ieee80211_node_vattach(struct ieee80211vap *vap)
129{
130	/* NB: driver can override */
131	vap->iv_max_aid = IEEE80211_AID_DEF;
132
133	/* default station inactivity timer setings */
134	vap->iv_inact_init = IEEE80211_INACT_INIT;
135	vap->iv_inact_auth = IEEE80211_INACT_AUTH;
136	vap->iv_inact_run = IEEE80211_INACT_RUN;
137	vap->iv_inact_probe = IEEE80211_INACT_PROBE;
138}
139
140void
141ieee80211_node_latevattach(struct ieee80211vap *vap)
142{
143	if (vap->iv_opmode == IEEE80211_M_HOSTAP) {
144		/* XXX should we allow max aid to be zero? */
145		if (vap->iv_max_aid < IEEE80211_AID_MIN) {
146			vap->iv_max_aid = IEEE80211_AID_MIN;
147			if_printf(vap->iv_ifp,
148			    "WARNING: max aid too small, changed to %d\n",
149			    vap->iv_max_aid);
150		}
151		MALLOC(vap->iv_aid_bitmap, uint32_t *,
152			howmany(vap->iv_max_aid, 32) * sizeof(uint32_t),
153			M_80211_NODE, M_NOWAIT | M_ZERO);
154		if (vap->iv_aid_bitmap == NULL) {
155			/* XXX no way to recover */
156			printf("%s: no memory for AID bitmap, max aid %d!\n",
157			    __func__, vap->iv_max_aid);
158			vap->iv_max_aid = 0;
159		}
160	}
161
162	ieee80211_reset_bss(vap);
163
164	vap->iv_auth = ieee80211_authenticator_get(vap->iv_bss->ni_authmode);
165}
166
167void
168ieee80211_node_vdetach(struct ieee80211vap *vap)
169{
170	struct ieee80211com *ic = vap->iv_ic;
171
172	ieee80211_node_table_reset(&ic->ic_sta, vap);
173	if (vap->iv_bss != NULL) {
174		ieee80211_free_node(vap->iv_bss);
175		vap->iv_bss = NULL;
176	}
177	if (vap->iv_aid_bitmap != NULL) {
178		FREE(vap->iv_aid_bitmap, M_80211_NODE);
179		vap->iv_aid_bitmap = NULL;
180	}
181}
182
183/*
184 * Port authorize/unauthorize interfaces for use by an authenticator.
185 */
186
187void
188ieee80211_node_authorize(struct ieee80211_node *ni)
189{
190	ni->ni_flags |= IEEE80211_NODE_AUTH;
191	ni->ni_inact_reload = ni->ni_vap->iv_inact_run;
192	ni->ni_inact = ni->ni_inact_reload;
193}
194
195void
196ieee80211_node_unauthorize(struct ieee80211_node *ni)
197{
198	ni->ni_flags &= ~IEEE80211_NODE_AUTH;
199	ni->ni_inact_reload = ni->ni_vap->iv_inact_auth;
200	if (ni->ni_inact > ni->ni_inact_reload)
201		ni->ni_inact = ni->ni_inact_reload;
202}
203
204/*
205 * Fix tx parameters for a node according to ``association state''.
206 */
207static void
208node_setuptxparms(struct ieee80211_node *ni)
209{
210	struct ieee80211vap *vap = ni->ni_vap;
211
212	if (ni->ni_flags & IEEE80211_NODE_HT) {
213		if (IEEE80211_IS_CHAN_5GHZ(ni->ni_chan))
214			ni->ni_txparms = &vap->iv_txparms[IEEE80211_MODE_11NA];
215		else
216			ni->ni_txparms = &vap->iv_txparms[IEEE80211_MODE_11NG];
217	} else {				/* legacy rate handling */
218		if (IEEE80211_IS_CHAN_A(ni->ni_chan))
219			ni->ni_txparms = &vap->iv_txparms[IEEE80211_MODE_11A];
220		else if (ni->ni_flags & IEEE80211_NODE_ERP)
221			ni->ni_txparms = &vap->iv_txparms[IEEE80211_MODE_11G];
222		else
223			ni->ni_txparms = &vap->iv_txparms[IEEE80211_MODE_11B];
224	}
225}
226
227/*
228 * Set/change the channel.  The rate set is also updated as
229 * to insure a consistent view by drivers.
230 * XXX should be private but hostap needs it to deal with CSA
231 */
232void
233ieee80211_node_set_chan(struct ieee80211_node *ni,
234	struct ieee80211_channel *chan)
235{
236	struct ieee80211com *ic = ni->ni_ic;
237	struct ieee80211vap *vap = ni->ni_vap;
238	enum ieee80211_phymode mode;
239
240	KASSERT(chan != IEEE80211_CHAN_ANYC, ("no channel"));
241
242	ni->ni_chan = chan;
243	mode = ieee80211_chan2mode(chan);
244	if (IEEE80211_IS_CHAN_HT(chan)) {
245		/*
246		 * XXX Gotta be careful here; the rate set returned by
247		 * ieee80211_get_suprates is actually any HT rate
248		 * set so blindly copying it will be bad.  We must
249		 * install the legacy rate est in ni_rates and the
250		 * HT rate set in ni_htrates.
251		 */
252		ni->ni_htrates = *ieee80211_get_suphtrates(ic, chan);
253		/*
254		 * Setup bss tx parameters based on operating mode.  We
255		 * use legacy rates when operating in a mixed HT+non-HT bss
256		 * and non-ERP rates in 11g for mixed ERP+non-ERP bss.
257		 */
258		if (mode == IEEE80211_MODE_11NA &&
259		    (vap->iv_flags_ext & IEEE80211_FEXT_PUREN) == 0)
260			mode = IEEE80211_MODE_11A;
261		else if (mode == IEEE80211_MODE_11NG &&
262		    (vap->iv_flags_ext & IEEE80211_FEXT_PUREN) == 0)
263			mode = IEEE80211_MODE_11G;
264		if (mode == IEEE80211_MODE_11G &&
265		    (vap->iv_flags & IEEE80211_F_PUREG) == 0)
266			mode = IEEE80211_MODE_11B;
267	}
268	ni->ni_txparms = &vap->iv_txparms[mode];
269	ni->ni_rates = *ieee80211_get_suprates(ic, chan);
270}
271
272static __inline void
273copy_bss(struct ieee80211_node *nbss, const struct ieee80211_node *obss)
274{
275	/* propagate useful state */
276	nbss->ni_authmode = obss->ni_authmode;
277	nbss->ni_txpower = obss->ni_txpower;
278	nbss->ni_vlan = obss->ni_vlan;
279	/* XXX statistics? */
280	/* XXX legacy WDS bssid? */
281}
282
283void
284ieee80211_create_ibss(struct ieee80211vap* vap, struct ieee80211_channel *chan)
285{
286	struct ieee80211com *ic = vap->iv_ic;
287	struct ieee80211_node *ni;
288
289	IEEE80211_DPRINTF(vap, IEEE80211_MSG_SCAN,
290		"%s: creating ibss on channel %u\n", __func__,
291		ieee80211_chan2ieee(ic, chan));
292
293	ni = ieee80211_alloc_node(&ic->ic_sta, vap, vap->iv_myaddr);
294	if (ni == NULL) {
295		/* XXX recovery? */
296		return;
297	}
298	IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_myaddr);
299	ni->ni_esslen = vap->iv_des_ssid[0].len;
300	memcpy(ni->ni_essid, vap->iv_des_ssid[0].ssid, ni->ni_esslen);
301	if (vap->iv_bss != NULL)
302		copy_bss(ni, vap->iv_bss);
303	ni->ni_intval = ic->ic_bintval;
304	if (vap->iv_flags & IEEE80211_F_PRIVACY)
305		ni->ni_capinfo |= IEEE80211_CAPINFO_PRIVACY;
306	if (ic->ic_phytype == IEEE80211_T_FH) {
307		ni->ni_fhdwell = 200;	/* XXX */
308		ni->ni_fhindex = 1;
309	}
310	if (vap->iv_opmode == IEEE80211_M_IBSS) {
311		vap->iv_flags |= IEEE80211_F_SIBSS;
312		ni->ni_capinfo |= IEEE80211_CAPINFO_IBSS;	/* XXX */
313		if (vap->iv_flags & IEEE80211_F_DESBSSID)
314			IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_des_bssid);
315		else {
316			get_random_bytes(ni->ni_bssid, IEEE80211_ADDR_LEN);
317			/* clear group bit, add local bit */
318			ni->ni_bssid[0] = (ni->ni_bssid[0] &~ 0x01) | 0x02;
319		}
320	} else if (vap->iv_opmode == IEEE80211_M_AHDEMO) {
321		if (vap->iv_flags & IEEE80211_F_DESBSSID)
322			IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_des_bssid);
323		else
324			memset(ni->ni_bssid, 0, IEEE80211_ADDR_LEN);
325	}
326	/*
327	 * Fix the channel and related attributes.
328	 */
329	/* clear DFS CAC state on previous channel */
330	if (ic->ic_bsschan != IEEE80211_CHAN_ANYC &&
331	    ic->ic_bsschan->ic_freq != chan->ic_freq &&
332	    IEEE80211_IS_CHAN_CACDONE(ic->ic_bsschan))
333		ieee80211_dfs_cac_clear(ic, ic->ic_bsschan);
334	ic->ic_bsschan = chan;
335	ieee80211_node_set_chan(ni, chan);
336	ic->ic_curmode = ieee80211_chan2mode(chan);
337	/*
338	 * Do mode-specific setup.
339	 */
340	if (IEEE80211_IS_CHAN_FULL(chan)) {
341		if (IEEE80211_IS_CHAN_ANYG(chan)) {
342			/*
343			 * Use a mixed 11b/11g basic rate set.
344			 */
345			ieee80211_setbasicrates(&ni->ni_rates,
346			    IEEE80211_MODE_11G);
347			if (vap->iv_flags & IEEE80211_F_PUREG) {
348				/*
349				 * Also mark OFDM rates basic so 11b
350				 * stations do not join (WiFi compliance).
351				 */
352				ieee80211_addbasicrates(&ni->ni_rates,
353				    IEEE80211_MODE_11A);
354			}
355		} else if (IEEE80211_IS_CHAN_B(chan)) {
356			/*
357			 * Force pure 11b rate set.
358			 */
359			ieee80211_setbasicrates(&ni->ni_rates,
360				IEEE80211_MODE_11B);
361		}
362	}
363
364	(void) ieee80211_sta_join1(ieee80211_ref_node(ni));
365}
366
367/*
368 * Reset bss state on transition to the INIT state.
369 * Clear any stations from the table (they have been
370 * deauth'd) and reset the bss node (clears key, rate
371 * etc. state).
372 */
373void
374ieee80211_reset_bss(struct ieee80211vap *vap)
375{
376	struct ieee80211com *ic = vap->iv_ic;
377	struct ieee80211_node *ni, *obss;
378
379	ieee80211_node_table_reset(&ic->ic_sta, vap);
380	/* XXX multi-bss: wrong */
381	ieee80211_reset_erp(ic);
382
383	ni = ieee80211_alloc_node(&ic->ic_sta, vap, vap->iv_myaddr);
384	KASSERT(ni != NULL, ("unable to setup inital BSS node"));
385	obss = vap->iv_bss;
386	vap->iv_bss = ieee80211_ref_node(ni);
387	if (obss != NULL) {
388		copy_bss(ni, obss);
389		ni->ni_intval = ic->ic_bintval;
390		ieee80211_free_node(obss);
391	} else
392		IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_myaddr);
393}
394
395static int
396match_ssid(const struct ieee80211_node *ni,
397	int nssid, const struct ieee80211_scan_ssid ssids[])
398{
399	int i;
400
401	for (i = 0; i < nssid; i++) {
402		if (ni->ni_esslen == ssids[i].len &&
403		     memcmp(ni->ni_essid, ssids[i].ssid, ni->ni_esslen) == 0)
404			return 1;
405	}
406	return 0;
407}
408
409/*
410 * Test a node for suitability/compatibility.
411 */
412static int
413check_bss(struct ieee80211vap *vap, struct ieee80211_node *ni)
414{
415	struct ieee80211com *ic = ni->ni_ic;
416        uint8_t rate;
417
418	if (isclr(ic->ic_chan_active, ieee80211_chan2ieee(ic, ni->ni_chan)))
419		return 0;
420	if (vap->iv_opmode == IEEE80211_M_IBSS) {
421		if ((ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) == 0)
422			return 0;
423	} else {
424		if ((ni->ni_capinfo & IEEE80211_CAPINFO_ESS) == 0)
425			return 0;
426	}
427	if (vap->iv_flags & IEEE80211_F_PRIVACY) {
428		if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0)
429			return 0;
430	} else {
431		/* XXX does this mean privacy is supported or required? */
432		if (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY)
433			return 0;
434	}
435	rate = ieee80211_fix_rate(ni, &ni->ni_rates,
436	    IEEE80211_F_JOIN | IEEE80211_F_DONEGO | IEEE80211_F_DOFRATE);
437	if (rate & IEEE80211_RATE_BASIC)
438		return 0;
439	if (vap->iv_des_nssid != 0 &&
440	    !match_ssid(ni, vap->iv_des_nssid, vap->iv_des_ssid))
441		return 0;
442	if ((vap->iv_flags & IEEE80211_F_DESBSSID) &&
443	    !IEEE80211_ADDR_EQ(vap->iv_des_bssid, ni->ni_bssid))
444		return 0;
445	return 1;
446}
447
448#ifdef IEEE80211_DEBUG
449/*
450 * Display node suitability/compatibility.
451 */
452static void
453check_bss_debug(struct ieee80211vap *vap, struct ieee80211_node *ni)
454{
455	struct ieee80211com *ic = ni->ni_ic;
456        uint8_t rate;
457        int fail;
458
459	fail = 0;
460	if (isclr(ic->ic_chan_active, ieee80211_chan2ieee(ic, ni->ni_chan)))
461		fail |= 0x01;
462	if (vap->iv_opmode == IEEE80211_M_IBSS) {
463		if ((ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) == 0)
464			fail |= 0x02;
465	} else {
466		if ((ni->ni_capinfo & IEEE80211_CAPINFO_ESS) == 0)
467			fail |= 0x02;
468	}
469	if (vap->iv_flags & IEEE80211_F_PRIVACY) {
470		if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0)
471			fail |= 0x04;
472	} else {
473		/* XXX does this mean privacy is supported or required? */
474		if (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY)
475			fail |= 0x04;
476	}
477	rate = ieee80211_fix_rate(ni, &ni->ni_rates,
478	     IEEE80211_F_JOIN | IEEE80211_F_DONEGO | IEEE80211_F_DOFRATE);
479	if (rate & IEEE80211_RATE_BASIC)
480		fail |= 0x08;
481	if (vap->iv_des_nssid != 0 &&
482	    !match_ssid(ni, vap->iv_des_nssid, vap->iv_des_ssid))
483		fail |= 0x10;
484	if ((vap->iv_flags & IEEE80211_F_DESBSSID) &&
485	    !IEEE80211_ADDR_EQ(vap->iv_des_bssid, ni->ni_bssid))
486		fail |= 0x20;
487
488	printf(" %c %s", fail ? '-' : '+', ether_sprintf(ni->ni_macaddr));
489	printf(" %s%c", ether_sprintf(ni->ni_bssid), fail & 0x20 ? '!' : ' ');
490	printf(" %3d%c",
491	    ieee80211_chan2ieee(ic, ni->ni_chan), fail & 0x01 ? '!' : ' ');
492	printf(" %2dM%c", (rate & IEEE80211_RATE_VAL) / 2,
493	    fail & 0x08 ? '!' : ' ');
494	printf(" %4s%c",
495	    (ni->ni_capinfo & IEEE80211_CAPINFO_ESS) ? "ess" :
496	    (ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) ? "ibss" :
497	    "????",
498	    fail & 0x02 ? '!' : ' ');
499	printf(" %3s%c ",
500	    (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) ?  "wep" : "no",
501	    fail & 0x04 ? '!' : ' ');
502	ieee80211_print_essid(ni->ni_essid, ni->ni_esslen);
503	printf("%s\n", fail & 0x10 ? "!" : "");
504}
505#endif /* IEEE80211_DEBUG */
506
507/*
508 * Handle 802.11 ad hoc network merge.  The
509 * convention, set by the Wireless Ethernet Compatibility Alliance
510 * (WECA), is that an 802.11 station will change its BSSID to match
511 * the "oldest" 802.11 ad hoc network, on the same channel, that
512 * has the station's desired SSID.  The "oldest" 802.11 network
513 * sends beacons with the greatest TSF timestamp.
514 *
515 * The caller is assumed to validate TSF's before attempting a merge.
516 *
517 * Return !0 if the BSSID changed, 0 otherwise.
518 */
519int
520ieee80211_ibss_merge(struct ieee80211_node *ni)
521{
522	struct ieee80211vap *vap = ni->ni_vap;
523#ifdef IEEE80211_DEBUG
524	struct ieee80211com *ic = ni->ni_ic;
525#endif
526
527	if (ni == vap->iv_bss ||
528	    IEEE80211_ADDR_EQ(ni->ni_bssid, vap->iv_bss->ni_bssid)) {
529		/* unchanged, nothing to do */
530		return 0;
531	}
532	if (!check_bss(vap, ni)) {
533		/* capabilities mismatch */
534		IEEE80211_DPRINTF(vap, IEEE80211_MSG_ASSOC,
535		    "%s: merge failed, capabilities mismatch\n", __func__);
536#ifdef IEEE80211_DEBUG
537		if (ieee80211_msg_assoc(vap))
538			check_bss_debug(vap, ni);
539#endif
540		vap->iv_stats.is_ibss_capmismatch++;
541		return 0;
542	}
543	IEEE80211_DPRINTF(vap, IEEE80211_MSG_ASSOC,
544		"%s: new bssid %s: %s preamble, %s slot time%s\n", __func__,
545		ether_sprintf(ni->ni_bssid),
546		ic->ic_flags&IEEE80211_F_SHPREAMBLE ? "short" : "long",
547		ic->ic_flags&IEEE80211_F_SHSLOT ? "short" : "long",
548		ic->ic_flags&IEEE80211_F_USEPROT ? ", protection" : ""
549	);
550	return ieee80211_sta_join1(ieee80211_ref_node(ni));
551}
552
553/*
554 * Calculate HT channel promotion flags for all vaps.
555 * This assumes ni_chan have been setup for each vap.
556 */
557static int
558gethtadjustflags(struct ieee80211com *ic)
559{
560	struct ieee80211vap *vap;
561	int flags;
562
563	flags = 0;
564	/* XXX locking */
565	TAILQ_FOREACH(vap, &ic->ic_vaps, iv_next) {
566		if (vap->iv_state < IEEE80211_S_RUN)
567			continue;
568		switch (vap->iv_opmode) {
569		case IEEE80211_M_WDS:
570		case IEEE80211_M_STA:
571		case IEEE80211_M_AHDEMO:
572		case IEEE80211_M_HOSTAP:
573		case IEEE80211_M_IBSS:
574			flags |= ieee80211_htchanflags(vap->iv_bss->ni_chan);
575			break;
576		default:
577			break;
578		}
579	}
580	return flags;
581}
582
583/*
584 * Check if the current channel needs to change based on whether
585 * any vap's are using HT20/HT40.  This is used sync the state of
586 * ic_curchan after a channel width change on a running vap.
587 */
588void
589ieee80211_sync_curchan(struct ieee80211com *ic)
590{
591	struct ieee80211_channel *c;
592
593	c = ieee80211_ht_adjust_channel(ic, ic->ic_curchan, gethtadjustflags(ic));
594	if (c != ic->ic_curchan) {
595		ic->ic_curchan = c;
596		ic->ic_curmode = ieee80211_chan2mode(ic->ic_curchan);
597		ic->ic_set_channel(ic);
598	}
599}
600
601/*
602 * Change the current channel.  The request channel may be
603 * promoted if other vap's are operating with HT20/HT40.
604 */
605void
606ieee80211_setcurchan(struct ieee80211com *ic, struct ieee80211_channel *c)
607{
608	if (ic->ic_htcaps & IEEE80211_HTC_HT) {
609		int flags = gethtadjustflags(ic);
610		/*
611		 * Check for channel promotion required to support the
612		 * set of running vap's.  This assumes we are called
613		 * after ni_chan is setup for each vap.
614		 */
615		/* NB: this assumes IEEE80211_FEXT_USEHT40 > IEEE80211_FEXT_HT */
616		if (flags > ieee80211_htchanflags(c))
617			c = ieee80211_ht_adjust_channel(ic, c, flags);
618	}
619	ic->ic_bsschan = ic->ic_curchan = c;
620	ic->ic_curmode = ieee80211_chan2mode(ic->ic_curchan);
621	ic->ic_set_channel(ic);
622}
623
624/*
625 * Join the specified IBSS/BSS network.  The node is assumed to
626 * be passed in with a held reference.
627 */
628static int
629ieee80211_sta_join1(struct ieee80211_node *selbs)
630{
631	struct ieee80211vap *vap = selbs->ni_vap;
632	struct ieee80211com *ic = selbs->ni_ic;
633	struct ieee80211_node *obss;
634	int canreassoc;
635
636	/*
637	 * Committed to selbs, setup state.
638	 */
639	obss = vap->iv_bss;
640	/*
641	 * Check if old+new node have the same address in which
642	 * case we can reassociate when operating in sta mode.
643	 */
644	canreassoc = (obss != NULL &&
645		vap->iv_state == IEEE80211_S_RUN &&
646		IEEE80211_ADDR_EQ(obss->ni_macaddr, selbs->ni_macaddr));
647	vap->iv_bss = selbs;		/* NB: caller assumed to bump refcnt */
648	if (obss != NULL) {
649		copy_bss(selbs, obss);
650		ieee80211_node_reclaim(obss);
651		obss = NULL;		/* NB: guard against later use */
652	}
653
654	/*
655	 * Delete unusable rates; we've already checked
656	 * that the negotiated rate set is acceptable.
657	 */
658	ieee80211_fix_rate(vap->iv_bss, &vap->iv_bss->ni_rates,
659		IEEE80211_F_DODEL | IEEE80211_F_JOIN);
660
661	ieee80211_setcurchan(ic, selbs->ni_chan);
662	/*
663	 * Set the erp state (mostly the slot time) to deal with
664	 * the auto-select case; this should be redundant if the
665	 * mode is locked.
666	 */
667	ieee80211_reset_erp(ic);
668	ieee80211_wme_initparams(vap);
669
670	if (vap->iv_opmode == IEEE80211_M_STA) {
671		if (canreassoc) {
672			/* Reassociate */
673			ieee80211_new_state(vap, IEEE80211_S_ASSOC, 1);
674		} else {
675			/*
676			 * Act as if we received a DEAUTH frame in case we
677			 * are invoked from the RUN state.  This will cause
678			 * us to try to re-authenticate if we are operating
679			 * as a station.
680			 */
681			ieee80211_new_state(vap, IEEE80211_S_AUTH,
682				IEEE80211_FC0_SUBTYPE_DEAUTH);
683		}
684	} else
685		ieee80211_new_state(vap, IEEE80211_S_RUN, -1);
686	return 1;
687}
688
689int
690ieee80211_sta_join(struct ieee80211vap *vap,
691	const struct ieee80211_scan_entry *se)
692{
693	struct ieee80211com *ic = vap->iv_ic;
694	struct ieee80211_node *ni;
695
696	ni = ieee80211_alloc_node(&ic->ic_sta, vap, se->se_macaddr);
697	if (ni == NULL) {
698		/* XXX msg */
699		return 0;
700	}
701	/*
702	 * Expand scan state into node's format.
703	 * XXX may not need all this stuff
704	 */
705	IEEE80211_ADDR_COPY(ni->ni_bssid, se->se_bssid);
706	ni->ni_esslen = se->se_ssid[1];
707	memcpy(ni->ni_essid, se->se_ssid+2, ni->ni_esslen);
708	ni->ni_rstamp = se->se_rstamp;
709	ni->ni_tstamp.tsf = se->se_tstamp.tsf;
710	ni->ni_intval = se->se_intval;
711	ni->ni_capinfo = se->se_capinfo;
712	ni->ni_chan = se->se_chan;
713	ni->ni_timoff = se->se_timoff;
714	ni->ni_fhdwell = se->se_fhdwell;
715	ni->ni_fhindex = se->se_fhindex;
716	ni->ni_erp = se->se_erp;
717	IEEE80211_RSSI_LPF(ni->ni_avgrssi, se->se_rssi);
718	ni->ni_noise = se->se_noise;
719
720	if (ieee80211_ies_init(&ni->ni_ies, se->se_ies.data, se->se_ies.len)) {
721		ieee80211_ies_expand(&ni->ni_ies);
722		if (ni->ni_ies.ath_ie != NULL)
723			ieee80211_parse_ath(ni, ni->ni_ies.ath_ie);
724		if (ni->ni_ies.htcap_ie != NULL)
725			ieee80211_parse_htcap(ni, ni->ni_ies.htcap_ie);
726		if (ni->ni_ies.htinfo_ie != NULL)
727			ieee80211_parse_htinfo(ni, ni->ni_ies.htinfo_ie);
728	}
729
730	vap->iv_dtim_period = se->se_dtimperiod;
731	vap->iv_dtim_count = 0;
732
733	/* NB: must be after ni_chan is setup */
734	ieee80211_setup_rates(ni, se->se_rates, se->se_xrates,
735		IEEE80211_F_DOSORT);
736
737	return ieee80211_sta_join1(ieee80211_ref_node(ni));
738}
739
740/*
741 * Leave the specified IBSS/BSS network.  The node is assumed to
742 * be passed in with a held reference.
743 */
744void
745ieee80211_sta_leave(struct ieee80211_node *ni)
746{
747	struct ieee80211com *ic = ni->ni_ic;
748
749	ic->ic_node_cleanup(ni);
750	ieee80211_notify_node_leave(ni);
751}
752
753/*
754 * Send a deauthenticate frame and drop the station.
755 */
756void
757ieee80211_node_deauth(struct ieee80211_node *ni, int reason)
758{
759	/* NB: bump the refcnt to be sure temporay nodes are not reclaimed */
760	ieee80211_ref_node(ni);
761	if (ni->ni_associd != 0)
762		IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_DEAUTH, reason);
763	ieee80211_node_leave(ni);
764	ieee80211_free_node(ni);
765}
766
767static struct ieee80211_node *
768node_alloc(struct ieee80211vap *vap, const uint8_t macaddr[IEEE80211_ADDR_LEN])
769{
770	struct ieee80211_node *ni;
771
772	MALLOC(ni, struct ieee80211_node *, sizeof(struct ieee80211_node),
773		M_80211_NODE, M_NOWAIT | M_ZERO);
774	return ni;
775}
776
777/*
778 * Initialize an ie blob with the specified data.  If previous
779 * data exists re-use the data block.  As a side effect we clear
780 * all references to specific ie's; the caller is required to
781 * recalculate them.
782 */
783int
784ieee80211_ies_init(struct ieee80211_ies *ies, const uint8_t *data, int len)
785{
786	/* NB: assumes data+len are the last fields */
787	memset(ies, 0, offsetof(struct ieee80211_ies, data));
788	if (ies->data != NULL && ies->len != len) {
789		/* data size changed */
790		FREE(ies->data, M_80211_NODE_IE);
791		ies->data = NULL;
792	}
793	if (ies->data == NULL) {
794		MALLOC(ies->data, uint8_t *, len, M_80211_NODE_IE, M_NOWAIT);
795		if (ies->data == NULL) {
796			ies->len = 0;
797			/* NB: pointers have already been zero'd above */
798			return 0;
799		}
800	}
801	memcpy(ies->data, data, len);
802	ies->len = len;
803	return 1;
804}
805
806/*
807 * Reclaim storage for an ie blob.
808 */
809void
810ieee80211_ies_cleanup(struct ieee80211_ies *ies)
811{
812	if (ies->data != NULL)
813		FREE(ies->data, M_80211_NODE_IE);
814}
815
816/*
817 * Expand an ie blob data contents and to fillin individual
818 * ie pointers.  The data blob is assumed to be well-formed;
819 * we don't do any validity checking of ie lengths.
820 */
821void
822ieee80211_ies_expand(struct ieee80211_ies *ies)
823{
824	uint8_t *ie;
825	int ielen;
826
827	ie = ies->data;
828	ielen = ies->len;
829	while (ielen > 0) {
830		switch (ie[0]) {
831		case IEEE80211_ELEMID_VENDOR:
832			if (iswpaoui(ie))
833				ies->wpa_ie = ie;
834			else if (iswmeoui(ie))
835				ies->wme_ie = ie;
836			else if (isatherosoui(ie))
837				ies->ath_ie = ie;
838			break;
839		case IEEE80211_ELEMID_RSN:
840			ies->rsn_ie = ie;
841			break;
842		case IEEE80211_ELEMID_HTCAP:
843			ies->htcap_ie = ie;
844			break;
845		}
846		ielen -= 2 + ie[1];
847		ie += 2 + ie[1];
848	}
849}
850
851/*
852 * Reclaim any resources in a node and reset any critical
853 * state.  Typically nodes are free'd immediately after,
854 * but in some cases the storage may be reused so we need
855 * to insure consistent state (should probably fix that).
856 */
857static void
858node_cleanup(struct ieee80211_node *ni)
859{
860#define	N(a)	(sizeof(a)/sizeof(a[0]))
861	struct ieee80211vap *vap = ni->ni_vap;
862	int i;
863
864	/* NB: preserve ni_table */
865	if (ni->ni_flags & IEEE80211_NODE_PWR_MGT) {
866		if (vap->iv_opmode != IEEE80211_M_STA)
867			vap->iv_ps_sta--;
868		ni->ni_flags &= ~IEEE80211_NODE_PWR_MGT;
869		IEEE80211_NOTE(vap, IEEE80211_MSG_POWER, ni,
870		    "power save mode off, %u sta's in ps mode", vap->iv_ps_sta);
871	}
872	/*
873	 * Cleanup any HT-related state.
874	 */
875	if (ni->ni_flags & IEEE80211_NODE_HT)
876		ieee80211_ht_node_cleanup(ni);
877	/*
878	 * Clear AREF flag that marks the authorization refcnt bump
879	 * has happened.  This is probably not needed as the node
880	 * should always be removed from the table so not found but
881	 * do it just in case.
882	 */
883	ni->ni_flags &= ~IEEE80211_NODE_AREF;
884
885	/*
886	 * Drain power save queue and, if needed, clear TIM.
887	 */
888	if (ieee80211_node_saveq_drain(ni) != 0 && vap->iv_set_tim != NULL)
889		vap->iv_set_tim(ni, 0);
890
891	ni->ni_associd = 0;
892	if (ni->ni_challenge != NULL) {
893		FREE(ni->ni_challenge, M_80211_NODE);
894		ni->ni_challenge = NULL;
895	}
896	/*
897	 * Preserve SSID, WPA, and WME ie's so the bss node is
898	 * reusable during a re-auth/re-assoc state transition.
899	 * If we remove these data they will not be recreated
900	 * because they come from a probe-response or beacon frame
901	 * which cannot be expected prior to the association-response.
902	 * This should not be an issue when operating in other modes
903	 * as stations leaving always go through a full state transition
904	 * which will rebuild this state.
905	 *
906	 * XXX does this leave us open to inheriting old state?
907	 */
908	for (i = 0; i < N(ni->ni_rxfrag); i++)
909		if (ni->ni_rxfrag[i] != NULL) {
910			m_freem(ni->ni_rxfrag[i]);
911			ni->ni_rxfrag[i] = NULL;
912		}
913	/*
914	 * Must be careful here to remove any key map entry w/o a LOR.
915	 */
916	ieee80211_node_delucastkey(ni);
917#undef N
918}
919
920static void
921node_free(struct ieee80211_node *ni)
922{
923	struct ieee80211com *ic = ni->ni_ic;
924
925	ic->ic_node_cleanup(ni);
926	ieee80211_ies_cleanup(&ni->ni_ies);
927	IEEE80211_NODE_SAVEQ_DESTROY(ni);
928	IEEE80211_NODE_WDSQ_DESTROY(ni);
929	FREE(ni, M_80211_NODE);
930}
931
932static void
933node_age(struct ieee80211_node *ni)
934{
935	struct ieee80211vap *vap = ni->ni_vap;
936#if 0
937	IEEE80211_NODE_LOCK_ASSERT(&ic->ic_sta);
938#endif
939	/*
940	 * Age frames on the power save queue.
941	 */
942	if (ieee80211_node_saveq_age(ni) != 0 &&
943	    IEEE80211_NODE_SAVEQ_QLEN(ni) == 0 &&
944	    vap->iv_set_tim != NULL)
945		vap->iv_set_tim(ni, 0);
946	/*
947	 * Age frames on the wds pending queue.
948	 */
949	if (IEEE80211_NODE_WDSQ_QLEN(ni) != 0)
950		ieee80211_node_wdsq_age(ni);
951	/*
952	 * Age out HT resources (e.g. frames on the
953	 * A-MPDU reorder queues).
954	 */
955	if (ni->ni_associd != 0 && (ni->ni_flags & IEEE80211_NODE_HT))
956		ieee80211_ht_node_age(ni);
957}
958
959static int8_t
960node_getrssi(const struct ieee80211_node *ni)
961{
962	uint32_t avgrssi = ni->ni_avgrssi;
963	int32_t rssi;
964
965	if (avgrssi == IEEE80211_RSSI_DUMMY_MARKER)
966		return 0;
967	rssi = IEEE80211_RSSI_GET(avgrssi);
968	return rssi < 0 ? 0 : rssi > 127 ? 127 : rssi;
969}
970
971static void
972node_getsignal(const struct ieee80211_node *ni, int8_t *rssi, int8_t *noise)
973{
974	*rssi = node_getrssi(ni);
975	*noise = ni->ni_noise;
976}
977
978static void
979node_getmimoinfo(const struct ieee80211_node *ni,
980	struct ieee80211_mimo_info *info)
981{
982	/* XXX zero data? */
983}
984
985struct ieee80211_node *
986ieee80211_alloc_node(struct ieee80211_node_table *nt,
987	struct ieee80211vap *vap, const uint8_t macaddr[IEEE80211_ADDR_LEN])
988{
989	struct ieee80211com *ic = nt->nt_ic;
990	struct ieee80211_node *ni;
991	int hash;
992
993	ni = ic->ic_node_alloc(vap, macaddr);
994	if (ni == NULL) {
995		vap->iv_stats.is_rx_nodealloc++;
996		return NULL;
997	}
998
999	IEEE80211_DPRINTF(vap, IEEE80211_MSG_NODE,
1000		"%s %p<%s> in %s table\n", __func__, ni,
1001		ether_sprintf(macaddr), nt->nt_name);
1002
1003	IEEE80211_ADDR_COPY(ni->ni_macaddr, macaddr);
1004	hash = IEEE80211_NODE_HASH(macaddr);
1005	ieee80211_node_initref(ni);		/* mark referenced */
1006	ni->ni_chan = IEEE80211_CHAN_ANYC;
1007	ni->ni_authmode = IEEE80211_AUTH_OPEN;
1008	ni->ni_txpower = ic->ic_txpowlimit;	/* max power */
1009	ni->ni_txparms = &vap->iv_txparms[ieee80211_chan2mode(ic->ic_curchan)];
1010	ieee80211_crypto_resetkey(vap, &ni->ni_ucastkey, IEEE80211_KEYIX_NONE);
1011	ni->ni_avgrssi = IEEE80211_RSSI_DUMMY_MARKER;
1012	ni->ni_inact_reload = nt->nt_inact_init;
1013	ni->ni_inact = ni->ni_inact_reload;
1014	ni->ni_ath_defkeyix = 0x7fff;
1015	IEEE80211_NODE_SAVEQ_INIT(ni, "unknown");
1016	IEEE80211_NODE_WDSQ_INIT(ni, "unknown");
1017
1018	IEEE80211_NODE_LOCK(nt);
1019	TAILQ_INSERT_TAIL(&nt->nt_node, ni, ni_list);
1020	LIST_INSERT_HEAD(&nt->nt_hash[hash], ni, ni_hash);
1021	ni->ni_table = nt;
1022	ni->ni_vap = vap;
1023	ni->ni_ic = ic;
1024	IEEE80211_NODE_UNLOCK(nt);
1025
1026	return ni;
1027}
1028
1029/*
1030 * Craft a temporary node suitable for sending a management frame
1031 * to the specified station.  We craft only as much state as we
1032 * need to do the work since the node will be immediately reclaimed
1033 * once the send completes.
1034 */
1035struct ieee80211_node *
1036ieee80211_tmp_node(struct ieee80211vap *vap,
1037	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1038{
1039	struct ieee80211com *ic = vap->iv_ic;
1040	struct ieee80211_node *ni;
1041
1042	ni = ic->ic_node_alloc(vap, macaddr);
1043	if (ni != NULL) {
1044		IEEE80211_DPRINTF(vap, IEEE80211_MSG_NODE,
1045			"%s %p<%s>\n", __func__, ni, ether_sprintf(macaddr));
1046
1047		ni->ni_table = NULL;		/* NB: pedantic */
1048		ni->ni_ic = ic;			/* NB: needed to set channel */
1049		ni->ni_vap = vap;
1050
1051		IEEE80211_ADDR_COPY(ni->ni_macaddr, macaddr);
1052		IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_bss->ni_bssid);
1053		ieee80211_node_initref(ni);		/* mark referenced */
1054		/* NB: required by ieee80211_fix_rate */
1055		ieee80211_node_set_chan(ni, vap->iv_bss->ni_chan);
1056		ieee80211_crypto_resetkey(vap, &ni->ni_ucastkey,
1057			IEEE80211_KEYIX_NONE);
1058		/* XXX optimize away */
1059		IEEE80211_NODE_SAVEQ_INIT(ni, "unknown");
1060		IEEE80211_NODE_WDSQ_INIT(ni, "unknown");
1061	} else {
1062		/* XXX msg */
1063		vap->iv_stats.is_rx_nodealloc++;
1064	}
1065	return ni;
1066}
1067
1068struct ieee80211_node *
1069ieee80211_dup_bss(struct ieee80211vap *vap,
1070	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1071{
1072	struct ieee80211com *ic = vap->iv_ic;
1073	struct ieee80211_node *ni;
1074
1075	ni = ieee80211_alloc_node(&ic->ic_sta, vap, macaddr);
1076	if (ni != NULL) {
1077		/*
1078		 * Inherit from iv_bss.
1079		 */
1080		ni->ni_authmode = vap->iv_bss->ni_authmode;
1081		ni->ni_vlan = vap->iv_bss->ni_vlan;	/* XXX?? */
1082		IEEE80211_ADDR_COPY(ni->ni_bssid, vap->iv_bss->ni_bssid);
1083		ieee80211_node_set_chan(ni, vap->iv_bss->ni_chan);
1084	}
1085	return ni;
1086}
1087
1088/*
1089 * Create a bss node for a legacy WDS vap.  The far end does
1090 * not associate so we just create create a new node and
1091 * simulate an association.  The caller is responsible for
1092 * installing the node as the bss node and handling any further
1093 * setup work like authorizing the port.
1094 */
1095struct ieee80211_node *
1096ieee80211_node_create_wds(struct ieee80211vap *vap,
1097	const uint8_t bssid[IEEE80211_ADDR_LEN], struct ieee80211_channel *chan)
1098{
1099	struct ieee80211com *ic = vap->iv_ic;
1100	struct ieee80211_node *ni;
1101
1102	/* XXX check if node already in sta table? */
1103	ni = ieee80211_alloc_node(&ic->ic_sta, vap, bssid);
1104	if (ni != NULL) {
1105		ni->ni_wdsvap = vap;
1106		IEEE80211_ADDR_COPY(ni->ni_bssid, bssid);
1107		/*
1108		 * Inherit any manually configured settings.
1109		 */
1110		ni->ni_authmode = vap->iv_bss->ni_authmode;
1111		ni->ni_vlan = vap->iv_bss->ni_vlan;
1112		ieee80211_node_set_chan(ni, chan);
1113		/* NB: propagate ssid so available to WPA supplicant */
1114		ni->ni_esslen = vap->iv_des_ssid[0].len;
1115		memcpy(ni->ni_essid, vap->iv_des_ssid[0].ssid, ni->ni_esslen);
1116		/* NB: no associd for peer */
1117		/*
1118		 * There are no management frames to use to
1119		 * discover neighbor capabilities, so blindly
1120		 * propagate the local configuration.
1121		 */
1122		if (vap->iv_flags & IEEE80211_F_WME)
1123			ni->ni_flags |= IEEE80211_NODE_QOS;
1124		if (vap->iv_flags & IEEE80211_F_FF)
1125			ni->ni_flags |= IEEE80211_NODE_FF;
1126		if ((ic->ic_htcaps & IEEE80211_HTC_HT) &&
1127		    (vap->iv_flags_ext & IEEE80211_FEXT_HT)) {
1128			/*
1129			 * Device is HT-capable and HT is enabled for
1130			 * the vap; setup HT operation.  On return
1131			 * ni_chan will be adjusted to an HT channel.
1132			 */
1133			ieee80211_ht_wds_init(ni);
1134		} else {
1135			struct ieee80211_channel *c = ni->ni_chan;
1136			/*
1137			 * Force a legacy channel to be used.
1138			 */
1139			c = ieee80211_find_channel(ic,
1140			    c->ic_freq, c->ic_flags &~ IEEE80211_CHAN_HT);
1141			KASSERT(c != NULL, ("no legacy channel, %u/%x",
1142			    ni->ni_chan->ic_freq, ni->ni_chan->ic_flags));
1143			ni->ni_chan = c;
1144		}
1145	}
1146	return ni;
1147}
1148
1149struct ieee80211_node *
1150#ifdef IEEE80211_DEBUG_REFCNT
1151ieee80211_find_node_locked_debug(struct ieee80211_node_table *nt,
1152	const uint8_t macaddr[IEEE80211_ADDR_LEN], const char *func, int line)
1153#else
1154ieee80211_find_node_locked(struct ieee80211_node_table *nt,
1155	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1156#endif
1157{
1158	struct ieee80211_node *ni;
1159	int hash;
1160
1161	IEEE80211_NODE_LOCK_ASSERT(nt);
1162
1163	hash = IEEE80211_NODE_HASH(macaddr);
1164	LIST_FOREACH(ni, &nt->nt_hash[hash], ni_hash) {
1165		if (IEEE80211_ADDR_EQ(ni->ni_macaddr, macaddr)) {
1166			ieee80211_ref_node(ni);	/* mark referenced */
1167#ifdef IEEE80211_DEBUG_REFCNT
1168			IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1169			    "%s (%s:%u) %p<%s> refcnt %d\n", __func__,
1170			    func, line,
1171			    ni, ether_sprintf(ni->ni_macaddr),
1172			    ieee80211_node_refcnt(ni));
1173#endif
1174			return ni;
1175		}
1176	}
1177	return NULL;
1178}
1179
1180struct ieee80211_node *
1181#ifdef IEEE80211_DEBUG_REFCNT
1182ieee80211_find_node_debug(struct ieee80211_node_table *nt,
1183	const uint8_t macaddr[IEEE80211_ADDR_LEN], const char *func, int line)
1184#else
1185ieee80211_find_node(struct ieee80211_node_table *nt,
1186	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1187#endif
1188{
1189	struct ieee80211_node *ni;
1190
1191	IEEE80211_NODE_LOCK(nt);
1192	ni = ieee80211_find_node_locked(nt, macaddr);
1193	IEEE80211_NODE_UNLOCK(nt);
1194	return ni;
1195}
1196
1197struct ieee80211_node *
1198#ifdef IEEE80211_DEBUG_REFCNT
1199ieee80211_find_vap_node_locked_debug(struct ieee80211_node_table *nt,
1200	const struct ieee80211vap *vap,
1201	const uint8_t macaddr[IEEE80211_ADDR_LEN], const char *func, int line)
1202#else
1203ieee80211_find_vap_node_locked(struct ieee80211_node_table *nt,
1204	const struct ieee80211vap *vap,
1205	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1206#endif
1207{
1208	struct ieee80211_node *ni;
1209	int hash;
1210
1211	IEEE80211_NODE_LOCK_ASSERT(nt);
1212
1213	hash = IEEE80211_NODE_HASH(macaddr);
1214	LIST_FOREACH(ni, &nt->nt_hash[hash], ni_hash) {
1215		if (ni->ni_vap == vap &&
1216		    IEEE80211_ADDR_EQ(ni->ni_macaddr, macaddr)) {
1217			ieee80211_ref_node(ni);	/* mark referenced */
1218#ifdef IEEE80211_DEBUG_REFCNT
1219			IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1220			    "%s (%s:%u) %p<%s> refcnt %d\n", __func__,
1221			    func, line,
1222			    ni, ether_sprintf(ni->ni_macaddr),
1223			    ieee80211_node_refcnt(ni));
1224#endif
1225			return ni;
1226		}
1227	}
1228	return NULL;
1229}
1230
1231struct ieee80211_node *
1232#ifdef IEEE80211_DEBUG_REFCNT
1233ieee80211_find_vap_node_debug(struct ieee80211_node_table *nt,
1234	const struct ieee80211vap *vap,
1235	const uint8_t macaddr[IEEE80211_ADDR_LEN], const char *func, int line)
1236#else
1237ieee80211_find_vap_node(struct ieee80211_node_table *nt,
1238	const struct ieee80211vap *vap,
1239	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1240#endif
1241{
1242	struct ieee80211_node *ni;
1243
1244	IEEE80211_NODE_LOCK(nt);
1245	ni = ieee80211_find_vap_node_locked(nt, vap, macaddr);
1246	IEEE80211_NODE_UNLOCK(nt);
1247	return ni;
1248}
1249
1250/*
1251 * Fake up a node; this handles node discovery in adhoc mode.
1252 * Note that for the driver's benefit we we treat this like
1253 * an association so the driver has an opportunity to setup
1254 * it's private state.
1255 */
1256struct ieee80211_node *
1257ieee80211_fakeup_adhoc_node(struct ieee80211vap *vap,
1258	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1259{
1260	struct ieee80211_node *ni;
1261
1262	IEEE80211_DPRINTF(vap, IEEE80211_MSG_NODE,
1263	    "%s: mac<%s>\n", __func__, ether_sprintf(macaddr));
1264	ni = ieee80211_dup_bss(vap, macaddr);
1265	if (ni != NULL) {
1266		struct ieee80211com *ic = vap->iv_ic;
1267
1268		/* XXX no rate negotiation; just dup */
1269		ni->ni_rates = vap->iv_bss->ni_rates;
1270		if (vap->iv_opmode == IEEE80211_M_AHDEMO) {
1271			/*
1272			 * In adhoc demo mode there are no management
1273			 * frames to use to discover neighbor capabilities,
1274			 * so blindly propagate the local configuration
1275			 * so we can do interesting things (e.g. use
1276			 * WME to disable ACK's).
1277			 */
1278			if (vap->iv_flags & IEEE80211_F_WME)
1279				ni->ni_flags |= IEEE80211_NODE_QOS;
1280			if (vap->iv_flags & IEEE80211_F_FF)
1281				ni->ni_flags |= IEEE80211_NODE_FF;
1282		}
1283		node_setuptxparms(ni);
1284		if (ic->ic_newassoc != NULL)
1285			ic->ic_newassoc(ni, 1);
1286		/* XXX not right for 802.1x/WPA */
1287		ieee80211_node_authorize(ni);
1288	}
1289	return ni;
1290}
1291
1292void
1293ieee80211_init_neighbor(struct ieee80211_node *ni,
1294	const struct ieee80211_frame *wh,
1295	const struct ieee80211_scanparams *sp)
1296{
1297	ni->ni_esslen = sp->ssid[1];
1298	memcpy(ni->ni_essid, sp->ssid + 2, sp->ssid[1]);
1299	IEEE80211_ADDR_COPY(ni->ni_bssid, wh->i_addr3);
1300	memcpy(ni->ni_tstamp.data, sp->tstamp, sizeof(ni->ni_tstamp));
1301	ni->ni_intval = sp->bintval;
1302	ni->ni_capinfo = sp->capinfo;
1303	ni->ni_chan = ni->ni_ic->ic_curchan;
1304	ni->ni_fhdwell = sp->fhdwell;
1305	ni->ni_fhindex = sp->fhindex;
1306	ni->ni_erp = sp->erp;
1307	ni->ni_timoff = sp->timoff;
1308
1309	if (ieee80211_ies_init(&ni->ni_ies, sp->ies, sp->ies_len)) {
1310		ieee80211_ies_expand(&ni->ni_ies);
1311		if (ni->ni_ies.ath_ie != NULL)
1312			ieee80211_parse_ath(ni, ni->ni_ies.ath_ie);
1313	}
1314
1315	/* NB: must be after ni_chan is setup */
1316	ieee80211_setup_rates(ni, sp->rates, sp->xrates,
1317		IEEE80211_F_DOSORT | IEEE80211_F_DOFRATE |
1318		IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
1319}
1320
1321/*
1322 * Do node discovery in adhoc mode on receipt of a beacon
1323 * or probe response frame.  Note that for the driver's
1324 * benefit we we treat this like an association so the
1325 * driver has an opportunity to setup it's private state.
1326 */
1327struct ieee80211_node *
1328ieee80211_add_neighbor(struct ieee80211vap *vap,
1329	const struct ieee80211_frame *wh,
1330	const struct ieee80211_scanparams *sp)
1331{
1332	struct ieee80211_node *ni;
1333
1334	IEEE80211_DPRINTF(vap, IEEE80211_MSG_NODE,
1335	    "%s: mac<%s>\n", __func__, ether_sprintf(wh->i_addr2));
1336	ni = ieee80211_dup_bss(vap, wh->i_addr2);/* XXX alloc_node? */
1337	if (ni != NULL) {
1338		struct ieee80211com *ic = vap->iv_ic;
1339
1340		ieee80211_init_neighbor(ni, wh, sp);
1341		node_setuptxparms(ni);
1342		if (ic->ic_newassoc != NULL)
1343			ic->ic_newassoc(ni, 1);
1344		/* XXX not right for 802.1x/WPA */
1345		ieee80211_node_authorize(ni);
1346	}
1347	return ni;
1348}
1349
1350#define	IS_CTL(wh) \
1351	((wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK) == IEEE80211_FC0_TYPE_CTL)
1352#define	IS_PSPOLL(wh) \
1353	((wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) == IEEE80211_FC0_SUBTYPE_PS_POLL)
1354#define	IS_BAR(wh) \
1355	((wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) == IEEE80211_FC0_SUBTYPE_BAR)
1356#define	IS_PROBEREQ(wh) \
1357	((wh->i_fc[0] & (IEEE80211_FC0_TYPE_MASK|IEEE80211_FC0_SUBTYPE_MASK)) \
1358	    == (IEEE80211_FC0_TYPE_MGT | IEEE80211_FC0_SUBTYPE_PROBE_REQ))
1359#define	IS_BCAST_PROBEREQ(wh) \
1360	(IS_PROBEREQ(wh) && IEEE80211_IS_MULTICAST( \
1361	    ((const struct ieee80211_frame *)(wh))->i_addr3))
1362
1363static __inline struct ieee80211_node *
1364_find_rxnode(struct ieee80211_node_table *nt,
1365    const struct ieee80211_frame_min *wh)
1366{
1367	/* XXX 4-address frames? */
1368	if (IS_CTL(wh) && !IS_PSPOLL(wh) && !IS_BAR(wh) /*&& !IS_RTS(ah)*/)
1369		return ieee80211_find_node_locked(nt, wh->i_addr1);
1370	if (IS_BCAST_PROBEREQ(wh))
1371		return NULL;		/* spam bcast probe req to all vap's */
1372	return ieee80211_find_node_locked(nt, wh->i_addr2);
1373}
1374
1375/*
1376 * Locate the node for sender, track state, and then pass the
1377 * (referenced) node up to the 802.11 layer for its use.  Note
1378 * we can return NULL if the sender is not in the table.
1379 */
1380struct ieee80211_node *
1381#ifdef IEEE80211_DEBUG_REFCNT
1382ieee80211_find_rxnode_debug(struct ieee80211com *ic,
1383	const struct ieee80211_frame_min *wh, const char *func, int line)
1384#else
1385ieee80211_find_rxnode(struct ieee80211com *ic,
1386	const struct ieee80211_frame_min *wh)
1387#endif
1388{
1389	struct ieee80211_node_table *nt;
1390	struct ieee80211_node *ni;
1391
1392	nt = &ic->ic_sta;
1393	IEEE80211_NODE_LOCK(nt);
1394	ni = _find_rxnode(nt, wh);
1395	IEEE80211_NODE_UNLOCK(nt);
1396
1397	return ni;
1398}
1399
1400/*
1401 * Like ieee80211_find_rxnode but use the supplied h/w
1402 * key index as a hint to locate the node in the key
1403 * mapping table.  If an entry is present at the key
1404 * index we return it; otherwise do a normal lookup and
1405 * update the mapping table if the station has a unicast
1406 * key assigned to it.
1407 */
1408struct ieee80211_node *
1409#ifdef IEEE80211_DEBUG_REFCNT
1410ieee80211_find_rxnode_withkey_debug(struct ieee80211com *ic,
1411	const struct ieee80211_frame_min *wh, ieee80211_keyix keyix,
1412	const char *func, int line)
1413#else
1414ieee80211_find_rxnode_withkey(struct ieee80211com *ic,
1415	const struct ieee80211_frame_min *wh, ieee80211_keyix keyix)
1416#endif
1417{
1418	struct ieee80211_node_table *nt;
1419	struct ieee80211_node *ni;
1420
1421	nt = &ic->ic_sta;
1422	IEEE80211_NODE_LOCK(nt);
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		ni = _find_rxnode(nt, wh);
1429		if (ni != NULL && nt->nt_keyixmap != NULL) {
1430			/*
1431			 * If the station has a unicast key cache slot
1432			 * assigned update the key->node mapping table.
1433			 */
1434			keyix = ni->ni_ucastkey.wk_rxkeyix;
1435			/* XXX can keyixmap[keyix] != NULL? */
1436			if (keyix < nt->nt_keyixmax &&
1437			    nt->nt_keyixmap[keyix] == NULL) {
1438				IEEE80211_DPRINTF(ni->ni_vap,
1439				    IEEE80211_MSG_NODE,
1440				    "%s: add key map entry %p<%s> refcnt %d\n",
1441				    __func__, ni, ether_sprintf(ni->ni_macaddr),
1442				    ieee80211_node_refcnt(ni)+1);
1443				nt->nt_keyixmap[keyix] = ieee80211_ref_node(ni);
1444			}
1445		}
1446	} else {
1447		if (IS_BCAST_PROBEREQ(wh))
1448			ni = NULL;	/* spam bcast probe req to all vap's */
1449		else
1450			ieee80211_ref_node(ni);
1451	}
1452	IEEE80211_NODE_UNLOCK(nt);
1453
1454	return ni;
1455}
1456#undef IS_BCAST_PROBEREQ
1457#undef IS_PROBEREQ
1458#undef IS_BAR
1459#undef IS_PSPOLL
1460#undef IS_CTL
1461
1462/*
1463 * Return a reference to the appropriate node for sending
1464 * a data frame.  This handles node discovery in adhoc networks.
1465 */
1466struct ieee80211_node *
1467#ifdef IEEE80211_DEBUG_REFCNT
1468ieee80211_find_txnode_debug(struct ieee80211vap *vap,
1469	const uint8_t macaddr[IEEE80211_ADDR_LEN],
1470	const char *func, int line)
1471#else
1472ieee80211_find_txnode(struct ieee80211vap *vap,
1473	const uint8_t macaddr[IEEE80211_ADDR_LEN])
1474#endif
1475{
1476	struct ieee80211_node_table *nt = &vap->iv_ic->ic_sta;
1477	struct ieee80211_node *ni;
1478
1479	/*
1480	 * The destination address should be in the node table
1481	 * unless this is a multicast/broadcast frame.  We can
1482	 * also optimize station mode operation, all frames go
1483	 * to the bss node.
1484	 */
1485	/* XXX can't hold lock across dup_bss 'cuz of recursive locking */
1486	IEEE80211_NODE_LOCK(nt);
1487	if (vap->iv_opmode == IEEE80211_M_STA ||
1488	    vap->iv_opmode == IEEE80211_M_WDS ||
1489	    IEEE80211_IS_MULTICAST(macaddr))
1490		ni = ieee80211_ref_node(vap->iv_bss);
1491	else {
1492		ni = ieee80211_find_node_locked(nt, macaddr);
1493		if (vap->iv_opmode == IEEE80211_M_HOSTAP &&
1494		    (ni != NULL && ni->ni_associd == 0)) {
1495			/*
1496			 * Station is not associated; don't permit the
1497			 * data frame to be sent by returning NULL.  This
1498			 * is kinda a kludge but the least intrusive way
1499			 * to add this check into all drivers.
1500			 */
1501			ieee80211_unref_node(&ni);	/* NB: null's ni */
1502		}
1503	}
1504	IEEE80211_NODE_UNLOCK(nt);
1505
1506	if (ni == NULL) {
1507		if (vap->iv_opmode == IEEE80211_M_IBSS ||
1508		    vap->iv_opmode == IEEE80211_M_AHDEMO) {
1509			/*
1510			 * In adhoc mode cons up a node for the destination.
1511			 * Note that we need an additional reference for the
1512			 * caller to be consistent with
1513			 * ieee80211_find_node_locked.
1514			 */
1515			ni = ieee80211_fakeup_adhoc_node(vap, macaddr);
1516			if (ni != NULL)
1517				(void) ieee80211_ref_node(ni);
1518		} else {
1519			IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_OUTPUT, macaddr,
1520			    "no node, discard frame (%s)", __func__);
1521			vap->iv_stats.is_tx_nonode++;
1522		}
1523	}
1524	return ni;
1525}
1526
1527static void
1528_ieee80211_free_node(struct ieee80211_node *ni)
1529{
1530	struct ieee80211_node_table *nt = ni->ni_table;
1531
1532	/*
1533	 * NB: careful about referencing the vap as it may be
1534	 * gone if the last reference was held by a driver.
1535	 * We know the com will always be present so it's safe
1536	 * to use ni_ic below to reclaim resources.
1537	 */
1538#if 0
1539	IEEE80211_DPRINTF(vap, IEEE80211_MSG_NODE,
1540		"%s %p<%s> in %s table\n", __func__, ni,
1541		ether_sprintf(ni->ni_macaddr),
1542		nt != NULL ? nt->nt_name : "<gone>");
1543#endif
1544	if (ni->ni_associd != 0) {
1545		struct ieee80211vap *vap = ni->ni_vap;
1546		if (vap->iv_aid_bitmap != NULL)
1547			IEEE80211_AID_CLR(vap, ni->ni_associd);
1548	}
1549	if (nt != NULL) {
1550		TAILQ_REMOVE(&nt->nt_node, ni, ni_list);
1551		LIST_REMOVE(ni, ni_hash);
1552	}
1553	ni->ni_ic->ic_node_free(ni);
1554}
1555
1556void
1557#ifdef IEEE80211_DEBUG_REFCNT
1558ieee80211_free_node_debug(struct ieee80211_node *ni, const char *func, int line)
1559#else
1560ieee80211_free_node(struct ieee80211_node *ni)
1561#endif
1562{
1563	struct ieee80211_node_table *nt = ni->ni_table;
1564
1565#ifdef IEEE80211_DEBUG_REFCNT
1566	IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1567		"%s (%s:%u) %p<%s> refcnt %d\n", __func__, func, line, ni,
1568		 ether_sprintf(ni->ni_macaddr), ieee80211_node_refcnt(ni)-1);
1569#endif
1570	if (nt != NULL) {
1571		IEEE80211_NODE_LOCK(nt);
1572		if (ieee80211_node_dectestref(ni)) {
1573			/*
1574			 * Last reference, reclaim state.
1575			 */
1576			_ieee80211_free_node(ni);
1577		} else if (ieee80211_node_refcnt(ni) == 1 &&
1578		    nt->nt_keyixmap != NULL) {
1579			ieee80211_keyix keyix;
1580			/*
1581			 * Check for a last reference in the key mapping table.
1582			 */
1583			keyix = ni->ni_ucastkey.wk_rxkeyix;
1584			if (keyix < nt->nt_keyixmax &&
1585			    nt->nt_keyixmap[keyix] == ni) {
1586				IEEE80211_DPRINTF(ni->ni_vap,
1587				    IEEE80211_MSG_NODE,
1588				    "%s: %p<%s> clear key map entry", __func__,
1589				    ni, ether_sprintf(ni->ni_macaddr));
1590				nt->nt_keyixmap[keyix] = NULL;
1591				ieee80211_node_decref(ni); /* XXX needed? */
1592				_ieee80211_free_node(ni);
1593			}
1594		}
1595		IEEE80211_NODE_UNLOCK(nt);
1596	} else {
1597		if (ieee80211_node_dectestref(ni))
1598			_ieee80211_free_node(ni);
1599	}
1600}
1601
1602/*
1603 * Reclaim a unicast key and clear any key cache state.
1604 */
1605int
1606ieee80211_node_delucastkey(struct ieee80211_node *ni)
1607{
1608	struct ieee80211com *ic = ni->ni_ic;
1609	struct ieee80211_node_table *nt = &ic->ic_sta;
1610	struct ieee80211_node *nikey;
1611	ieee80211_keyix keyix;
1612	int isowned, status;
1613
1614	/*
1615	 * NB: We must beware of LOR here; deleting the key
1616	 * can cause the crypto layer to block traffic updates
1617	 * which can generate a LOR against the node table lock;
1618	 * grab it here and stash the key index for our use below.
1619	 *
1620	 * Must also beware of recursion on the node table lock.
1621	 * When called from node_cleanup we may already have
1622	 * the node table lock held.  Unfortunately there's no
1623	 * way to separate out this path so we must do this
1624	 * conditionally.
1625	 */
1626	isowned = IEEE80211_NODE_IS_LOCKED(nt);
1627	if (!isowned)
1628		IEEE80211_NODE_LOCK(nt);
1629	nikey = NULL;
1630	status = 1;		/* NB: success */
1631	if (!IEEE80211_KEY_UNDEFINED(&ni->ni_ucastkey)) {
1632		keyix = ni->ni_ucastkey.wk_rxkeyix;
1633		status = ieee80211_crypto_delkey(ni->ni_vap, &ni->ni_ucastkey);
1634		if (nt->nt_keyixmap != NULL && keyix < nt->nt_keyixmax) {
1635			nikey = nt->nt_keyixmap[keyix];
1636			nt->nt_keyixmap[keyix] = NULL;;
1637		}
1638	}
1639	if (!isowned)
1640		IEEE80211_NODE_UNLOCK(nt);
1641
1642	if (nikey != NULL) {
1643		KASSERT(nikey == ni,
1644			("key map out of sync, ni %p nikey %p", ni, nikey));
1645		IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1646			"%s: delete key map entry %p<%s> refcnt %d\n",
1647			__func__, ni, ether_sprintf(ni->ni_macaddr),
1648			ieee80211_node_refcnt(ni)-1);
1649		ieee80211_free_node(ni);
1650	}
1651	return status;
1652}
1653
1654/*
1655 * Reclaim a node.  If this is the last reference count then
1656 * do the normal free work.  Otherwise remove it from the node
1657 * table and mark it gone by clearing the back-reference.
1658 */
1659static void
1660node_reclaim(struct ieee80211_node_table *nt, struct ieee80211_node *ni)
1661{
1662	ieee80211_keyix keyix;
1663
1664	IEEE80211_NODE_LOCK_ASSERT(nt);
1665
1666	IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1667		"%s: remove %p<%s> from %s table, refcnt %d\n",
1668		__func__, ni, ether_sprintf(ni->ni_macaddr),
1669		nt->nt_name, ieee80211_node_refcnt(ni)-1);
1670	/*
1671	 * Clear any entry in the unicast key mapping table.
1672	 * We need to do it here so rx lookups don't find it
1673	 * in the mapping table even if it's not in the hash
1674	 * table.  We cannot depend on the mapping table entry
1675	 * being cleared because the node may not be free'd.
1676	 */
1677	keyix = ni->ni_ucastkey.wk_rxkeyix;
1678	if (nt->nt_keyixmap != NULL && keyix < nt->nt_keyixmax &&
1679	    nt->nt_keyixmap[keyix] == ni) {
1680		IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1681			"%s: %p<%s> clear key map entry\n",
1682			__func__, ni, ether_sprintf(ni->ni_macaddr));
1683		nt->nt_keyixmap[keyix] = NULL;
1684		ieee80211_node_decref(ni);	/* NB: don't need free */
1685	}
1686	if (!ieee80211_node_dectestref(ni)) {
1687		/*
1688		 * Other references are present, just remove the
1689		 * node from the table so it cannot be found.  When
1690		 * the references are dropped storage will be
1691		 * reclaimed.
1692		 */
1693		TAILQ_REMOVE(&nt->nt_node, ni, ni_list);
1694		LIST_REMOVE(ni, ni_hash);
1695		ni->ni_table = NULL;		/* clear reference */
1696	} else
1697		_ieee80211_free_node(ni);
1698}
1699
1700/*
1701 * Reclaim a (bss) node.  Decrement the refcnt and reclaim
1702 * the node if the only other reference to it is in the sta
1703 * table.  This is effectively ieee80211_free_node followed
1704 * by node_reclaim when the refcnt is 1 (after the free).
1705 */
1706static void
1707ieee80211_node_reclaim(struct ieee80211_node *ni)
1708{
1709	struct ieee80211_node_table *nt = ni->ni_table;
1710
1711	KASSERT(nt != NULL, ("reclaim node not in table"));
1712
1713#ifdef IEEE80211_DEBUG_REFCNT
1714	IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_NODE,
1715		"%s %p<%s> refcnt %d\n", __func__, ni,
1716		 ether_sprintf(ni->ni_macaddr), ieee80211_node_refcnt(ni)-1);
1717#endif
1718	IEEE80211_NODE_LOCK(nt);
1719	if (ieee80211_node_dectestref(ni)) {
1720		/*
1721		 * Last reference, reclaim state.
1722		 */
1723		_ieee80211_free_node(ni);
1724		nt = NULL;
1725	} else if (ieee80211_node_refcnt(ni) == 1 &&
1726	    nt->nt_keyixmap != NULL) {
1727		ieee80211_keyix keyix;
1728		/*
1729		 * Check for a last reference in the key mapping table.
1730		 */
1731		keyix = ni->ni_ucastkey.wk_rxkeyix;
1732		if (keyix < nt->nt_keyixmax &&
1733		    nt->nt_keyixmap[keyix] == ni) {
1734			IEEE80211_DPRINTF(ni->ni_vap,
1735			    IEEE80211_MSG_NODE,
1736			    "%s: %p<%s> clear key map entry", __func__,
1737			    ni, ether_sprintf(ni->ni_macaddr));
1738			nt->nt_keyixmap[keyix] = NULL;
1739			ieee80211_node_decref(ni); /* XXX needed? */
1740			_ieee80211_free_node(ni);
1741			nt = NULL;
1742		}
1743	}
1744	if (nt != NULL && ieee80211_node_refcnt(ni) == 1) {
1745		/*
1746		 * Last reference is in the sta table; complete
1747		 * the reclaim.  This handles bss nodes being
1748		 * recycled: the node has two references, one for
1749		 * iv_bss and one for the table.  After dropping
1750		 * the iv_bss ref above we need to reclaim the sta
1751		 * table reference.
1752		 */
1753		ieee80211_node_decref(ni);	/* NB: be pendantic */
1754		_ieee80211_free_node(ni);
1755	}
1756	IEEE80211_NODE_UNLOCK(nt);
1757}
1758
1759/*
1760 * Node table support.
1761 */
1762
1763static void
1764ieee80211_node_table_init(struct ieee80211com *ic,
1765	struct ieee80211_node_table *nt,
1766	const char *name, int inact, int keyixmax)
1767{
1768	struct ifnet *ifp = ic->ic_ifp;
1769
1770	nt->nt_ic = ic;
1771	IEEE80211_NODE_LOCK_INIT(nt, ifp->if_xname);
1772	IEEE80211_NODE_ITERATE_LOCK_INIT(nt, ifp->if_xname);
1773	TAILQ_INIT(&nt->nt_node);
1774	nt->nt_name = name;
1775	nt->nt_scangen = 1;
1776	nt->nt_inact_init = inact;
1777	nt->nt_keyixmax = keyixmax;
1778	if (nt->nt_keyixmax > 0) {
1779		MALLOC(nt->nt_keyixmap, struct ieee80211_node **,
1780			keyixmax * sizeof(struct ieee80211_node *),
1781			M_80211_NODE, M_NOWAIT | M_ZERO);
1782		if (nt->nt_keyixmap == NULL)
1783			if_printf(ic->ic_ifp,
1784			    "Cannot allocate key index map with %u entries\n",
1785			    keyixmax);
1786	} else
1787		nt->nt_keyixmap = NULL;
1788}
1789
1790static void
1791ieee80211_node_table_reset(struct ieee80211_node_table *nt,
1792	struct ieee80211vap *match)
1793{
1794	struct ieee80211_node *ni, *next;
1795
1796	IEEE80211_NODE_LOCK(nt);
1797	TAILQ_FOREACH_SAFE(ni, &nt->nt_node, ni_list, next) {
1798		if (match != NULL && ni->ni_vap != match)
1799			continue;
1800		/* XXX can this happen?  if so need's work */
1801		if (ni->ni_associd != 0) {
1802			struct ieee80211vap *vap = ni->ni_vap;
1803
1804			if (vap->iv_auth->ia_node_leave != NULL)
1805				vap->iv_auth->ia_node_leave(ni);
1806			if (vap->iv_aid_bitmap != NULL)
1807				IEEE80211_AID_CLR(vap, ni->ni_associd);
1808		}
1809		ni->ni_wdsvap = NULL;		/* clear reference */
1810		node_reclaim(nt, ni);
1811	}
1812	if (match != NULL && match->iv_opmode == IEEE80211_M_WDS) {
1813		/*
1814		 * Make a separate pass to clear references to this vap
1815		 * held by DWDS entries.  They will not be matched above
1816		 * because ni_vap will point to the ap vap but we still
1817		 * need to clear ni_wdsvap when the WDS vap is destroyed
1818		 * and/or reset.
1819		 */
1820		TAILQ_FOREACH_SAFE(ni, &nt->nt_node, ni_list, next)
1821			if (ni->ni_wdsvap == match)
1822				ni->ni_wdsvap = NULL;
1823	}
1824	IEEE80211_NODE_UNLOCK(nt);
1825}
1826
1827static void
1828ieee80211_node_table_cleanup(struct ieee80211_node_table *nt)
1829{
1830	ieee80211_node_table_reset(nt, NULL);
1831	if (nt->nt_keyixmap != NULL) {
1832#ifdef DIAGNOSTIC
1833		/* XXX verify all entries are NULL */
1834		int i;
1835		for (i = 0; i < nt->nt_keyixmax; i++)
1836			if (nt->nt_keyixmap[i] != NULL)
1837				printf("%s: %s[%u] still active\n", __func__,
1838					nt->nt_name, i);
1839#endif
1840		FREE(nt->nt_keyixmap, M_80211_NODE);
1841		nt->nt_keyixmap = NULL;
1842	}
1843	IEEE80211_NODE_ITERATE_LOCK_DESTROY(nt);
1844	IEEE80211_NODE_LOCK_DESTROY(nt);
1845}
1846
1847/*
1848 * Timeout inactive stations and do related housekeeping.
1849 * Note that we cannot hold the node lock while sending a
1850 * frame as this would lead to a LOR.  Instead we use a
1851 * generation number to mark nodes that we've scanned and
1852 * drop the lock and restart a scan if we have to time out
1853 * a node.  Since we are single-threaded by virtue of
1854 * controlling the inactivity timer we can be sure this will
1855 * process each node only once.
1856 */
1857static void
1858ieee80211_timeout_stations(struct ieee80211com *ic)
1859{
1860	struct ieee80211_node_table *nt = &ic->ic_sta;
1861	struct ieee80211vap *vap;
1862	struct ieee80211_node *ni;
1863	int gen = 0;
1864
1865	IEEE80211_NODE_ITERATE_LOCK(nt);
1866	gen = ++nt->nt_scangen;
1867restart:
1868	IEEE80211_NODE_LOCK(nt);
1869	TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
1870		if (ni->ni_scangen == gen)	/* previously handled */
1871			continue;
1872		ni->ni_scangen = gen;
1873		/*
1874		 * Ignore entries for which have yet to receive an
1875		 * authentication frame.  These are transient and
1876		 * will be reclaimed when the last reference to them
1877		 * goes away (when frame xmits complete).
1878		 */
1879		vap = ni->ni_vap;
1880		/*
1881		 * Only process stations when in RUN state.  This
1882		 * insures, for example, that we don't timeout an
1883		 * inactive station during CAC.  Note that CSA state
1884		 * is actually handled in ieee80211_node_timeout as
1885		 * it applies to more than timeout processing.
1886		 */
1887		if (vap->iv_state != IEEE80211_S_RUN)
1888			continue;
1889		/* XXX can vap be NULL? */
1890		if ((vap->iv_opmode == IEEE80211_M_HOSTAP ||
1891		     vap->iv_opmode == IEEE80211_M_STA) &&
1892		    (ni->ni_flags & IEEE80211_NODE_AREF) == 0)
1893			continue;
1894		/*
1895		 * Free fragment if not needed anymore
1896		 * (last fragment older than 1s).
1897		 * XXX doesn't belong here, move to node_age
1898		 */
1899		if (ni->ni_rxfrag[0] != NULL &&
1900		    ticks > ni->ni_rxfragstamp + hz) {
1901			m_freem(ni->ni_rxfrag[0]);
1902			ni->ni_rxfrag[0] = NULL;
1903		}
1904		if (ni->ni_inact > 0)
1905			ni->ni_inact--;
1906		/*
1907		 * Special case ourself; we may be idle for extended periods
1908		 * of time and regardless reclaiming our state is wrong.
1909		 * XXX run ic_node_age
1910		 */
1911		if (ni == vap->iv_bss)
1912			continue;
1913		if (ni->ni_associd != 0 ||
1914		    (vap->iv_opmode == IEEE80211_M_IBSS ||
1915		     vap->iv_opmode == IEEE80211_M_AHDEMO)) {
1916			/*
1917			 * Age/drain resources held by the station.
1918			 */
1919			ic->ic_node_age(ni);
1920			/*
1921			 * Probe the station before time it out.  We
1922			 * send a null data frame which may not be
1923			 * universally supported by drivers (need it
1924			 * for ps-poll support so it should be...).
1925			 *
1926			 * XXX don't probe the station unless we've
1927			 *     received a frame from them (and have
1928			 *     some idea of the rates they are capable
1929			 *     of); this will get fixed more properly
1930			 *     soon with better handling of the rate set.
1931			 */
1932			if ((vap->iv_flags_ext & IEEE80211_FEXT_INACT) &&
1933			    (0 < ni->ni_inact &&
1934			     ni->ni_inact <= vap->iv_inact_probe) &&
1935			    ni->ni_rates.rs_nrates != 0) {
1936				IEEE80211_NOTE(vap,
1937				    IEEE80211_MSG_INACT | IEEE80211_MSG_NODE,
1938				    ni, "%s",
1939				    "probe station due to inactivity");
1940				/*
1941				 * Grab a reference before unlocking the table
1942				 * so the node cannot be reclaimed before we
1943				 * send the frame. ieee80211_send_nulldata
1944				 * understands we've done this and reclaims the
1945				 * ref for us as needed.
1946				 */
1947				ieee80211_ref_node(ni);
1948				IEEE80211_NODE_UNLOCK(nt);
1949				ieee80211_send_nulldata(ni);
1950				/* XXX stat? */
1951				goto restart;
1952			}
1953		}
1954		if ((vap->iv_flags_ext & IEEE80211_FEXT_INACT) &&
1955		    ni->ni_inact <= 0) {
1956			IEEE80211_NOTE(vap,
1957			    IEEE80211_MSG_INACT | IEEE80211_MSG_NODE, ni,
1958			    "station timed out due to inactivity "
1959			    "(refcnt %u)", ieee80211_node_refcnt(ni));
1960			/*
1961			 * Send a deauthenticate frame and drop the station.
1962			 * This is somewhat complicated due to reference counts
1963			 * and locking.  At this point a station will typically
1964			 * have a reference count of 1.  ieee80211_node_leave
1965			 * will do a "free" of the node which will drop the
1966			 * reference count.  But in the meantime a reference
1967			 * wil be held by the deauth frame.  The actual reclaim
1968			 * of the node will happen either after the tx is
1969			 * completed or by ieee80211_node_leave.
1970			 *
1971			 * Separately we must drop the node lock before sending
1972			 * in case the driver takes a lock, as this can result
1973			 * in a LOR between the node lock and the driver lock.
1974			 */
1975			ieee80211_ref_node(ni);
1976			IEEE80211_NODE_UNLOCK(nt);
1977			if (ni->ni_associd != 0) {
1978				IEEE80211_SEND_MGMT(ni,
1979				    IEEE80211_FC0_SUBTYPE_DEAUTH,
1980				    IEEE80211_REASON_AUTH_EXPIRE);
1981			}
1982			ieee80211_node_leave(ni);
1983			ieee80211_free_node(ni);
1984			vap->iv_stats.is_node_timeout++;
1985			goto restart;
1986		}
1987	}
1988	IEEE80211_NODE_UNLOCK(nt);
1989
1990	IEEE80211_NODE_ITERATE_UNLOCK(nt);
1991}
1992
1993/*
1994 * Aggressively reclaim resources.  This should be used
1995 * only in a critical situation to reclaim mbuf resources.
1996 */
1997void
1998ieee80211_drain(struct ieee80211com *ic)
1999{
2000	struct ieee80211_node_table *nt = &ic->ic_sta;
2001	struct ieee80211vap *vap;
2002	struct ieee80211_node *ni;
2003
2004	IEEE80211_NODE_LOCK(nt);
2005	TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
2006		/*
2007		 * Ignore entries for which have yet to receive an
2008		 * authentication frame.  These are transient and
2009		 * will be reclaimed when the last reference to them
2010		 * goes away (when frame xmits complete).
2011		 */
2012		vap = ni->ni_vap;
2013		/*
2014		 * Only process stations when in RUN state.  This
2015		 * insures, for example, that we don't timeout an
2016		 * inactive station during CAC.  Note that CSA state
2017		 * is actually handled in ieee80211_node_timeout as
2018		 * it applies to more than timeout processing.
2019		 */
2020		if (vap->iv_state != IEEE80211_S_RUN)
2021			continue;
2022		/* XXX can vap be NULL? */
2023		if ((vap->iv_opmode == IEEE80211_M_HOSTAP ||
2024		     vap->iv_opmode == IEEE80211_M_STA) &&
2025		    (ni->ni_flags & IEEE80211_NODE_AREF) == 0)
2026			continue;
2027		/*
2028		 * Free fragments.
2029		 * XXX doesn't belong here, move to node_drain
2030		 */
2031		if (ni->ni_rxfrag[0] != NULL) {
2032			m_freem(ni->ni_rxfrag[0]);
2033			ni->ni_rxfrag[0] = NULL;
2034		}
2035		/*
2036		 * Drain resources held by the station.
2037		 */
2038		ic->ic_node_drain(ni);
2039	}
2040	IEEE80211_NODE_UNLOCK(nt);
2041}
2042
2043/*
2044 * Per-ieee80211com inactivity timer callback.
2045 */
2046void
2047ieee80211_node_timeout(void *arg)
2048{
2049	struct ieee80211com *ic = arg;
2050
2051	/*
2052	 * Defer timeout processing if a channel switch is pending.
2053	 * We typically need to be mute so not doing things that
2054	 * might generate frames is good to handle in one place.
2055	 * Supressing the station timeout processing may extend the
2056	 * lifetime of inactive stations (by not decrementing their
2057	 * idle counters) but this should be ok unless the CSA is
2058	 * active for an unusually long time.
2059	 */
2060	if ((ic->ic_flags & IEEE80211_F_CSAPENDING) == 0) {
2061		ieee80211_scan_timeout(ic);
2062		ieee80211_timeout_stations(ic);
2063
2064		IEEE80211_LOCK(ic);
2065		ieee80211_erp_timeout(ic);
2066		ieee80211_ht_timeout(ic);
2067		IEEE80211_UNLOCK(ic);
2068	}
2069	callout_reset(&ic->ic_inact, IEEE80211_INACT_WAIT*hz,
2070		ieee80211_node_timeout, ic);
2071}
2072
2073void
2074ieee80211_iterate_nodes(struct ieee80211_node_table *nt,
2075	ieee80211_iter_func *f, void *arg)
2076{
2077	struct ieee80211_node *ni;
2078	u_int gen;
2079
2080	IEEE80211_NODE_ITERATE_LOCK(nt);
2081	gen = ++nt->nt_scangen;
2082restart:
2083	IEEE80211_NODE_LOCK(nt);
2084	TAILQ_FOREACH(ni, &nt->nt_node, ni_list) {
2085		if (ni->ni_scangen != gen) {
2086			ni->ni_scangen = gen;
2087			(void) ieee80211_ref_node(ni);
2088			IEEE80211_NODE_UNLOCK(nt);
2089			(*f)(arg, ni);
2090			ieee80211_free_node(ni);
2091			goto restart;
2092		}
2093	}
2094	IEEE80211_NODE_UNLOCK(nt);
2095
2096	IEEE80211_NODE_ITERATE_UNLOCK(nt);
2097}
2098
2099void
2100ieee80211_dump_node(struct ieee80211_node_table *nt, struct ieee80211_node *ni)
2101{
2102	printf("0x%p: mac %s refcnt %d\n", ni,
2103		ether_sprintf(ni->ni_macaddr), ieee80211_node_refcnt(ni));
2104	printf("\tscangen %u authmode %u flags 0x%x\n",
2105		ni->ni_scangen, ni->ni_authmode, ni->ni_flags);
2106	printf("\tassocid 0x%x txpower %u vlan %u\n",
2107		ni->ni_associd, ni->ni_txpower, ni->ni_vlan);
2108	printf("\ttxseq %u rxseq %u fragno %u rxfragstamp %u\n",
2109		ni->ni_txseqs[IEEE80211_NONQOS_TID],
2110		ni->ni_rxseqs[IEEE80211_NONQOS_TID] >> IEEE80211_SEQ_SEQ_SHIFT,
2111		ni->ni_rxseqs[IEEE80211_NONQOS_TID] & IEEE80211_SEQ_FRAG_MASK,
2112		ni->ni_rxfragstamp);
2113	printf("\trstamp %u rssi %d noise %d intval %u capinfo 0x%x\n",
2114		ni->ni_rstamp, node_getrssi(ni), ni->ni_noise,
2115		ni->ni_intval, ni->ni_capinfo);
2116	printf("\tbssid %s essid \"%.*s\" channel %u:0x%x\n",
2117		ether_sprintf(ni->ni_bssid),
2118		ni->ni_esslen, ni->ni_essid,
2119		ni->ni_chan->ic_freq, ni->ni_chan->ic_flags);
2120	printf("\tinact %u txrate %u\n",
2121		ni->ni_inact, ni->ni_txrate);
2122	printf("\thtcap %x htparam %x htctlchan %u ht2ndchan %u\n",
2123		ni->ni_htcap, ni->ni_htparam,
2124		ni->ni_htctlchan, ni->ni_ht2ndchan);
2125	printf("\thtopmode %x htstbc %x chw %u\n",
2126		ni->ni_htopmode, ni->ni_htstbc, ni->ni_chw);
2127}
2128
2129void
2130ieee80211_dump_nodes(struct ieee80211_node_table *nt)
2131{
2132	ieee80211_iterate_nodes(nt,
2133		(ieee80211_iter_func *) ieee80211_dump_node, nt);
2134}
2135
2136static void
2137ieee80211_notify_erp_locked(struct ieee80211com *ic)
2138{
2139	struct ieee80211vap *vap;
2140
2141	IEEE80211_LOCK_ASSERT(ic);
2142
2143	TAILQ_FOREACH(vap, &ic->ic_vaps, iv_next)
2144		if (vap->iv_opmode == IEEE80211_M_HOSTAP)
2145			ieee80211_beacon_notify(vap, IEEE80211_BEACON_ERP);
2146}
2147
2148void
2149ieee80211_notify_erp(struct ieee80211com *ic)
2150{
2151	IEEE80211_LOCK(ic);
2152	ieee80211_notify_erp_locked(ic);
2153	IEEE80211_UNLOCK(ic);
2154}
2155
2156/*
2157 * Handle a station joining an 11g network.
2158 */
2159static void
2160ieee80211_node_join_11g(struct ieee80211_node *ni)
2161{
2162	struct ieee80211com *ic = ni->ni_ic;
2163
2164	IEEE80211_LOCK_ASSERT(ic);
2165
2166	/*
2167	 * Station isn't capable of short slot time.  Bump
2168	 * the count of long slot time stations and disable
2169	 * use of short slot time.  Note that the actual switch
2170	 * over to long slot time use may not occur until the
2171	 * next beacon transmission (per sec. 7.3.1.4 of 11g).
2172	 */
2173	if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME) == 0) {
2174		ic->ic_longslotsta++;
2175		IEEE80211_NOTE(ni->ni_vap, IEEE80211_MSG_ASSOC, ni,
2176		    "station needs long slot time, count %d",
2177		    ic->ic_longslotsta);
2178		/* XXX vap's w/ conflicting needs won't work */
2179		if (!IEEE80211_IS_CHAN_108G(ic->ic_bsschan)) {
2180			/*
2181			 * Don't force slot time when switched to turbo
2182			 * mode as non-ERP stations won't be present; this
2183			 * need only be done when on the normal G channel.
2184			 */
2185			ieee80211_set_shortslottime(ic, 0);
2186		}
2187	}
2188	/*
2189	 * If the new station is not an ERP station
2190	 * then bump the counter and enable protection
2191	 * if configured.
2192	 */
2193	if (!ieee80211_iserp_rateset(&ni->ni_rates)) {
2194		ic->ic_nonerpsta++;
2195		IEEE80211_NOTE(ni->ni_vap, IEEE80211_MSG_ASSOC, ni,
2196		    "station is !ERP, %d non-ERP stations associated",
2197		    ic->ic_nonerpsta);
2198		/*
2199		 * If station does not support short preamble
2200		 * then we must enable use of Barker preamble.
2201		 */
2202		if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE) == 0) {
2203			IEEE80211_NOTE(ni->ni_vap, IEEE80211_MSG_ASSOC, ni,
2204			    "%s", "station needs long preamble");
2205			ic->ic_flags |= IEEE80211_F_USEBARKER;
2206			ic->ic_flags &= ~IEEE80211_F_SHPREAMBLE;
2207		}
2208		/*
2209		 * If protection is configured and this is the first
2210		 * indication we should use protection, enable it.
2211		 */
2212		if (ic->ic_protmode != IEEE80211_PROT_NONE &&
2213		    ic->ic_nonerpsta == 1 &&
2214		    (ic->ic_flags_ext & IEEE80211_FEXT_NONERP_PR) == 0) {
2215			IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_ASSOC,
2216			    "%s: enable use of protection\n", __func__);
2217			ic->ic_flags |= IEEE80211_F_USEPROT;
2218			ieee80211_notify_erp_locked(ic);
2219		}
2220	} else
2221		ni->ni_flags |= IEEE80211_NODE_ERP;
2222}
2223
2224void
2225ieee80211_node_join(struct ieee80211_node *ni, int resp)
2226{
2227	struct ieee80211com *ic = ni->ni_ic;
2228	struct ieee80211vap *vap = ni->ni_vap;
2229	int newassoc;
2230
2231	if (ni->ni_associd == 0) {
2232		uint16_t aid;
2233
2234		KASSERT(vap->iv_aid_bitmap != NULL, ("no aid bitmap"));
2235		/*
2236		 * It would be good to search the bitmap
2237		 * more efficiently, but this will do for now.
2238		 */
2239		for (aid = 1; aid < vap->iv_max_aid; aid++) {
2240			if (!IEEE80211_AID_ISSET(vap, aid))
2241				break;
2242		}
2243		if (aid >= vap->iv_max_aid) {
2244			IEEE80211_SEND_MGMT(ni, resp, IEEE80211_STATUS_TOOMANY);
2245			ieee80211_node_leave(ni);
2246			return;
2247		}
2248		ni->ni_associd = aid | 0xc000;
2249		ni->ni_jointime = time_uptime;
2250		IEEE80211_LOCK(ic);
2251		IEEE80211_AID_SET(vap, ni->ni_associd);
2252		vap->iv_sta_assoc++;
2253		ic->ic_sta_assoc++;
2254
2255		if (IEEE80211_IS_CHAN_HT(ic->ic_bsschan))
2256			ieee80211_ht_node_join(ni);
2257		if (IEEE80211_IS_CHAN_ANYG(ic->ic_bsschan) &&
2258		    IEEE80211_IS_CHAN_FULL(ic->ic_bsschan))
2259			ieee80211_node_join_11g(ni);
2260		IEEE80211_UNLOCK(ic);
2261
2262		newassoc = 1;
2263	} else
2264		newassoc = 0;
2265
2266	IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG, ni,
2267	    "station associated at aid %d: %s preamble, %s slot time%s%s%s%s%s%s%s",
2268	    IEEE80211_NODE_AID(ni),
2269	    ic->ic_flags & IEEE80211_F_SHPREAMBLE ? "short" : "long",
2270	    ic->ic_flags & IEEE80211_F_SHSLOT ? "short" : "long",
2271	    ic->ic_flags & IEEE80211_F_USEPROT ? ", protection" : "",
2272	    ni->ni_flags & IEEE80211_NODE_QOS ? ", QoS" : "",
2273	    ni->ni_flags & IEEE80211_NODE_HT ?
2274		(ni->ni_chw == 40 ? ", HT40" : ", HT20") : "",
2275	    ni->ni_flags & IEEE80211_NODE_AMPDU ? " (+AMPDU)" : "",
2276	    ni->ni_flags & IEEE80211_NODE_MIMO_RTS ? " (+SMPS-DYN)" :
2277	        ni->ni_flags & IEEE80211_NODE_MIMO_PS ? " (+SMPS)" : "",
2278	    IEEE80211_ATH_CAP(vap, ni, IEEE80211_NODE_FF) ?
2279		", fast-frames" : "",
2280	    IEEE80211_ATH_CAP(vap, ni, IEEE80211_NODE_TURBOP) ?
2281		", turbo" : ""
2282	);
2283
2284	node_setuptxparms(ni);
2285	/* give driver a chance to setup state like ni_txrate */
2286	if (ic->ic_newassoc != NULL)
2287		ic->ic_newassoc(ni, newassoc);
2288	IEEE80211_SEND_MGMT(ni, resp, IEEE80211_STATUS_SUCCESS);
2289	/* tell the authenticator about new station */
2290	if (vap->iv_auth->ia_node_join != NULL)
2291		vap->iv_auth->ia_node_join(ni);
2292	ieee80211_notify_node_join(ni,
2293	    resp == IEEE80211_FC0_SUBTYPE_ASSOC_RESP);
2294}
2295
2296static void
2297disable_protection(struct ieee80211com *ic)
2298{
2299	KASSERT(ic->ic_nonerpsta == 0 &&
2300	    (ic->ic_flags_ext & IEEE80211_FEXT_NONERP_PR) == 0,
2301	   ("%d non ERP stations, flags 0x%x", ic->ic_nonerpsta,
2302	   ic->ic_flags_ext));
2303
2304	ic->ic_flags &= ~IEEE80211_F_USEPROT;
2305	/* XXX verify mode? */
2306	if (ic->ic_caps & IEEE80211_C_SHPREAMBLE) {
2307		ic->ic_flags |= IEEE80211_F_SHPREAMBLE;
2308		ic->ic_flags &= ~IEEE80211_F_USEBARKER;
2309	}
2310	ieee80211_notify_erp_locked(ic);
2311}
2312
2313/*
2314 * Handle a station leaving an 11g network.
2315 */
2316static void
2317ieee80211_node_leave_11g(struct ieee80211_node *ni)
2318{
2319	struct ieee80211com *ic = ni->ni_ic;
2320
2321	IEEE80211_LOCK_ASSERT(ic);
2322
2323	KASSERT(IEEE80211_IS_CHAN_ANYG(ic->ic_bsschan),
2324	     ("not in 11g, bss %u:0x%x", ic->ic_bsschan->ic_freq,
2325	      ic->ic_bsschan->ic_flags));
2326
2327	/*
2328	 * If a long slot station do the slot time bookkeeping.
2329	 */
2330	if ((ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME) == 0) {
2331		KASSERT(ic->ic_longslotsta > 0,
2332		    ("bogus long slot station count %d", ic->ic_longslotsta));
2333		ic->ic_longslotsta--;
2334		IEEE80211_NOTE(ni->ni_vap, IEEE80211_MSG_ASSOC, ni,
2335		    "long slot time station leaves, count now %d",
2336		    ic->ic_longslotsta);
2337		if (ic->ic_longslotsta == 0) {
2338			/*
2339			 * Re-enable use of short slot time if supported
2340			 * and not operating in IBSS mode (per spec).
2341			 */
2342			if ((ic->ic_caps & IEEE80211_C_SHSLOT) &&
2343			    ic->ic_opmode != IEEE80211_M_IBSS) {
2344				IEEE80211_DPRINTF(ni->ni_vap,
2345				    IEEE80211_MSG_ASSOC,
2346				    "%s: re-enable use of short slot time\n",
2347				    __func__);
2348				ieee80211_set_shortslottime(ic, 1);
2349			}
2350		}
2351	}
2352	/*
2353	 * If a non-ERP station do the protection-related bookkeeping.
2354	 */
2355	if ((ni->ni_flags & IEEE80211_NODE_ERP) == 0) {
2356		KASSERT(ic->ic_nonerpsta > 0,
2357		    ("bogus non-ERP station count %d", ic->ic_nonerpsta));
2358		ic->ic_nonerpsta--;
2359		IEEE80211_NOTE(ni->ni_vap, IEEE80211_MSG_ASSOC, ni,
2360		    "non-ERP station leaves, count now %d%s", ic->ic_nonerpsta,
2361		    (ic->ic_flags_ext & IEEE80211_FEXT_NONERP_PR) ?
2362			" (non-ERP sta present)" : "");
2363		if (ic->ic_nonerpsta == 0 &&
2364		    (ic->ic_flags_ext & IEEE80211_FEXT_NONERP_PR) == 0) {
2365			IEEE80211_DPRINTF(ni->ni_vap, IEEE80211_MSG_ASSOC,
2366				"%s: disable use of protection\n", __func__);
2367			disable_protection(ic);
2368		}
2369	}
2370}
2371
2372/*
2373 * Time out presence of an overlapping bss with non-ERP
2374 * stations.  When operating in hostap mode we listen for
2375 * beacons from other stations and if we identify a non-ERP
2376 * station is present we enable protection.  To identify
2377 * when all non-ERP stations are gone we time out this
2378 * condition.
2379 */
2380static void
2381ieee80211_erp_timeout(struct ieee80211com *ic)
2382{
2383
2384	IEEE80211_LOCK_ASSERT(ic);
2385
2386	if ((ic->ic_flags_ext & IEEE80211_FEXT_NONERP_PR) &&
2387	    time_after(ticks, ic->ic_lastnonerp + IEEE80211_NONERP_PRESENT_AGE)) {
2388#if 0
2389		IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC, ni,
2390		    "%s", "age out non-ERP sta present on channel");
2391#endif
2392		ic->ic_flags_ext &= ~IEEE80211_FEXT_NONERP_PR;
2393		if (ic->ic_nonerpsta == 0)
2394			disable_protection(ic);
2395	}
2396}
2397
2398/*
2399 * Handle bookkeeping for station deauthentication/disassociation
2400 * when operating as an ap.
2401 */
2402void
2403ieee80211_node_leave(struct ieee80211_node *ni)
2404{
2405	struct ieee80211com *ic = ni->ni_ic;
2406	struct ieee80211vap *vap = ni->ni_vap;
2407	struct ieee80211_node_table *nt = ni->ni_table;
2408
2409	IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG, ni,
2410	    "station with aid %d leaves", IEEE80211_NODE_AID(ni));
2411
2412	KASSERT(vap->iv_opmode != IEEE80211_M_STA,
2413		("unexpected operating mode %u", vap->iv_opmode));
2414	/*
2415	 * If node wasn't previously associated all
2416	 * we need to do is reclaim the reference.
2417	 */
2418	/* XXX ibss mode bypasses 11g and notification */
2419	if (ni->ni_associd == 0)
2420		goto done;
2421	/*
2422	 * Tell the authenticator the station is leaving.
2423	 * Note that we must do this before yanking the
2424	 * association id as the authenticator uses the
2425	 * associd to locate it's state block.
2426	 */
2427	if (vap->iv_auth->ia_node_leave != NULL)
2428		vap->iv_auth->ia_node_leave(ni);
2429
2430	IEEE80211_LOCK(ic);
2431	IEEE80211_AID_CLR(vap, ni->ni_associd);
2432	ni->ni_associd = 0;
2433	vap->iv_sta_assoc--;
2434	ic->ic_sta_assoc--;
2435
2436	if (IEEE80211_IS_CHAN_HT(ic->ic_bsschan))
2437		ieee80211_ht_node_leave(ni);
2438	if (IEEE80211_IS_CHAN_ANYG(ic->ic_bsschan) &&
2439	    IEEE80211_IS_CHAN_FULL(ic->ic_bsschan))
2440		ieee80211_node_leave_11g(ni);
2441	IEEE80211_UNLOCK(ic);
2442	/*
2443	 * Cleanup station state.  In particular clear various
2444	 * state that might otherwise be reused if the node
2445	 * is reused before the reference count goes to zero
2446	 * (and memory is reclaimed).
2447	 */
2448	ieee80211_sta_leave(ni);
2449done:
2450	/*
2451	 * Remove the node from any table it's recorded in and
2452	 * drop the caller's reference.  Removal from the table
2453	 * is important to insure the node is not reprocessed
2454	 * for inactivity.
2455	 */
2456	if (nt != NULL) {
2457		IEEE80211_NODE_LOCK(nt);
2458		node_reclaim(nt, ni);
2459		IEEE80211_NODE_UNLOCK(nt);
2460	} else
2461		ieee80211_free_node(ni);
2462}
2463
2464struct rssiinfo {
2465	struct ieee80211vap *vap;
2466	int	rssi_samples;
2467	uint32_t rssi_total;
2468};
2469
2470static void
2471get_hostap_rssi(void *arg, struct ieee80211_node *ni)
2472{
2473	struct rssiinfo *info = arg;
2474	struct ieee80211vap *vap = ni->ni_vap;
2475	int8_t rssi;
2476
2477	if (info->vap != vap)
2478		return;
2479	/* only associated stations */
2480	if (ni->ni_associd == 0)
2481		return;
2482	rssi = vap->iv_ic->ic_node_getrssi(ni);
2483	if (rssi != 0) {
2484		info->rssi_samples++;
2485		info->rssi_total += rssi;
2486	}
2487}
2488
2489static void
2490get_adhoc_rssi(void *arg, struct ieee80211_node *ni)
2491{
2492	struct rssiinfo *info = arg;
2493	struct ieee80211vap *vap = ni->ni_vap;
2494	int8_t rssi;
2495
2496	if (info->vap != vap)
2497		return;
2498	/* only neighbors */
2499	/* XXX check bssid */
2500	if ((ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) == 0)
2501		return;
2502	rssi = vap->iv_ic->ic_node_getrssi(ni);
2503	if (rssi != 0) {
2504		info->rssi_samples++;
2505		info->rssi_total += rssi;
2506	}
2507}
2508
2509int8_t
2510ieee80211_getrssi(struct ieee80211vap *vap)
2511{
2512#define	NZ(x)	((x) == 0 ? 1 : (x))
2513	struct ieee80211com *ic = vap->iv_ic;
2514	struct rssiinfo info;
2515
2516	info.rssi_total = 0;
2517	info.rssi_samples = 0;
2518	info.vap = vap;
2519	switch (vap->iv_opmode) {
2520	case IEEE80211_M_IBSS:		/* average of all ibss neighbors */
2521	case IEEE80211_M_AHDEMO:	/* average of all neighbors */
2522		ieee80211_iterate_nodes(&ic->ic_sta, get_adhoc_rssi, &info);
2523		break;
2524	case IEEE80211_M_HOSTAP:	/* average of all associated stations */
2525		ieee80211_iterate_nodes(&ic->ic_sta, get_hostap_rssi, &info);
2526		break;
2527	case IEEE80211_M_MONITOR:	/* XXX */
2528	case IEEE80211_M_STA:		/* use stats from associated ap */
2529	default:
2530		if (vap->iv_bss != NULL)
2531			info.rssi_total = ic->ic_node_getrssi(vap->iv_bss);
2532		info.rssi_samples = 1;
2533		break;
2534	}
2535	return info.rssi_total / NZ(info.rssi_samples);
2536#undef NZ
2537}
2538
2539void
2540ieee80211_getsignal(struct ieee80211vap *vap, int8_t *rssi, int8_t *noise)
2541{
2542
2543	if (vap->iv_bss == NULL)		/* NB: shouldn't happen */
2544		return;
2545	vap->iv_ic->ic_node_getsignal(vap->iv_bss, rssi, noise);
2546	/* for non-station mode return avg'd rssi accounting */
2547	if (vap->iv_opmode != IEEE80211_M_STA)
2548		*rssi = ieee80211_getrssi(vap);
2549}
2550