sys/net80211/ieee80211_node.c

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