The Design and Implementation of the FreeBSD Operating System, Second Edition
Now available: The Design and Implementation of the FreeBSD Operating System (Second Edition)


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]

FreeBSD/Linux Kernel Cross Reference
sys/net80211/ieee80211_node.c

Version: -  FREEBSD  -  FREEBSD-13-STABLE  -  FREEBSD-13-0  -  FREEBSD-12-STABLE  -  FREEBSD-12-0  -  FREEBSD-11-STABLE  -  FREEBSD-11-0  -  FREEBSD-10-STABLE  -  FREEBSD-10-0  -  FREEBSD-9-STABLE  -  FREEBSD-9-0  -  FREEBSD-8-STABLE  -  FREEBSD-8-0  -  FREEBSD-7-STABLE  -  FREEBSD-7-0  -  FREEBSD-6-STABLE  -  FREEBSD-6-0  -  FREEBSD-5-STABLE  -  FREEBSD-5-0  -  FREEBSD-4-STABLE  -  FREEBSD-3-STABLE  -  FREEBSD22  -  l41  -  OPENBSD  -  linux-2.6  -  MK84  -  PLAN9  -  xnu-8792 
SearchContext: -  none  -  3  -  10 

    1 /*      $OpenBSD: ieee80211_node.c,v 1.195 2022/03/20 07:50:32 stsp Exp $       */
    2 /*      $NetBSD: ieee80211_node.c,v 1.14 2004/05/09 09:18:47 dyoung Exp $       */
    3 
    4 /*-
    5  * Copyright (c) 2001 Atsushi Onoe
    6  * Copyright (c) 2002, 2003 Sam Leffler, Errno Consulting
    7  * Copyright (c) 2008 Damien Bergamini
    8  * All rights reserved.
    9  *
   10  * Redistribution and use in source and binary forms, with or without
   11  * modification, are permitted provided that the following conditions
   12  * are met:
   13  * 1. Redistributions of source code must retain the above copyright
   14  *    notice, this list of conditions and the following disclaimer.
   15  * 2. Redistributions in binary form must reproduce the above copyright
   16  *    notice, this list of conditions and the following disclaimer in the
   17  *    documentation and/or other materials provided with the distribution.
   18  * 3. The name of the author may not be used to endorse or promote products
   19  *    derived from this software without specific prior written permission.
   20  *
   21  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
   22  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
   23  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
   24  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
   25  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
   26  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
   27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
   28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
   29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
   30  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   31  */
   32 
   33 #include "bridge.h"
   34 
   35 #include <sys/param.h>
   36 #include <sys/systm.h>
   37 #include <sys/mbuf.h>
   38 #include <sys/malloc.h>
   39 #include <sys/kernel.h>
   40 #include <sys/socket.h>
   41 #include <sys/sockio.h>
   42 #include <sys/endian.h>
   43 #include <sys/errno.h>
   44 #include <sys/sysctl.h>
   45 #include <sys/tree.h>
   46 
   47 #include <net/if.h>
   48 #include <net/if_dl.h>
   49 #include <net/if_media.h>
   50 
   51 #include <netinet/in.h>
   52 #include <netinet/if_ether.h>
   53 
   54 #if NBRIDGE > 0
   55 #include <net/if_bridge.h>
   56 #endif
   57 
   58 #include <net80211/ieee80211_var.h>
   59 #include <net80211/ieee80211_priv.h>
   60 
   61 struct ieee80211_node *ieee80211_node_alloc(struct ieee80211com *);
   62 void ieee80211_node_free(struct ieee80211com *, struct ieee80211_node *);
   63 void ieee80211_node_copy(struct ieee80211com *, struct ieee80211_node *,
   64     const struct ieee80211_node *);
   65 void ieee80211_choose_rsnparams(struct ieee80211com *);
   66 u_int8_t ieee80211_node_getrssi(struct ieee80211com *,
   67     const struct ieee80211_node *);
   68 int ieee80211_node_checkrssi(struct ieee80211com *,
   69     const struct ieee80211_node *);
   70 int ieee80211_ess_is_better(struct ieee80211com *ic, struct ieee80211_node *,
   71     struct ieee80211_node *);
   72 void ieee80211_node_set_timeouts(struct ieee80211_node *);
   73 void ieee80211_setup_node(struct ieee80211com *, struct ieee80211_node *,
   74     const u_int8_t *);
   75 struct ieee80211_node *ieee80211_alloc_node_helper(struct ieee80211com *);
   76 void ieee80211_node_free_unref_cb(struct ieee80211_node *);
   77 void ieee80211_node_tx_flushed(struct ieee80211com *, struct ieee80211_node *);
   78 void ieee80211_node_switch_bss(struct ieee80211com *, struct ieee80211_node *);
   79 void ieee80211_node_addba_request(struct ieee80211_node *, int);
   80 void ieee80211_node_addba_request_ac_be_to(void *);
   81 void ieee80211_node_addba_request_ac_bk_to(void *);
   82 void ieee80211_node_addba_request_ac_vi_to(void *);
   83 void ieee80211_node_addba_request_ac_vo_to(void *);
   84 void ieee80211_needs_auth(struct ieee80211com *, struct ieee80211_node *);
   85 #ifndef IEEE80211_STA_ONLY
   86 void ieee80211_node_join_ht(struct ieee80211com *, struct ieee80211_node *);
   87 void ieee80211_node_join_rsn(struct ieee80211com *, struct ieee80211_node *);
   88 void ieee80211_node_join_11g(struct ieee80211com *, struct ieee80211_node *);
   89 void ieee80211_node_leave_ht(struct ieee80211com *, struct ieee80211_node *);
   90 void ieee80211_node_leave_vht(struct ieee80211com *, struct ieee80211_node *);
   91 void ieee80211_node_leave_rsn(struct ieee80211com *, struct ieee80211_node *);
   92 void ieee80211_node_leave_11g(struct ieee80211com *, struct ieee80211_node *);
   93 void ieee80211_node_leave_pwrsave(struct ieee80211com *,
   94     struct ieee80211_node *);
   95 void ieee80211_inact_timeout(void *);
   96 void ieee80211_node_cache_timeout(void *);
   97 #endif
   98 void ieee80211_clean_inactive_nodes(struct ieee80211com *, int);
   99 
  100 #ifndef IEEE80211_STA_ONLY
  101 void
  102 ieee80211_inact_timeout(void *arg)
  103 {
  104         struct ieee80211com *ic = arg;
  105         struct ieee80211_node *ni, *next_ni;
  106         int s;
  107 
  108         s = splnet();
  109         for (ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
  110             ni != NULL; ni = next_ni) {
  111                 next_ni = RBT_NEXT(ieee80211_tree, ni);
  112                 if (ni->ni_refcnt > 0)
  113                         continue;
  114                 if (ni->ni_inact < IEEE80211_INACT_MAX)
  115                         ni->ni_inact++;
  116         }
  117         splx(s);
  118 
  119         timeout_add_sec(&ic->ic_inact_timeout, IEEE80211_INACT_WAIT);
  120 }
  121 
  122 void
  123 ieee80211_node_cache_timeout(void *arg)
  124 {
  125         struct ieee80211com *ic = arg;
  126 
  127         ieee80211_clean_nodes(ic, 1);
  128         timeout_add_sec(&ic->ic_node_cache_timeout, IEEE80211_CACHE_WAIT);
  129 }
  130 #endif
  131 
  132 /*
  133  * For debug purposes
  134  */
  135 void
  136 ieee80211_print_ess(struct ieee80211_ess *ess)
  137 {
  138         ieee80211_print_essid(ess->essid, ess->esslen);
  139         if (ess->flags & IEEE80211_F_RSNON) {
  140                 printf(" wpa");
  141                 if (ess->rsnprotos & IEEE80211_PROTO_RSN)
  142                         printf(",wpa2");
  143                 if (ess->rsnprotos & IEEE80211_PROTO_WPA)
  144                         printf(",wpa1");
  145 
  146                 if (ess->rsnakms & IEEE80211_AKM_8021X ||
  147                     ess->rsnakms & IEEE80211_AKM_SHA256_8021X)
  148                         printf(",802.1x");
  149                 printf(" ");
  150 
  151                 if (ess->rsnciphers & IEEE80211_CIPHER_USEGROUP)
  152                         printf(" usegroup");
  153                 if (ess->rsnciphers & IEEE80211_CIPHER_WEP40)
  154                         printf(" wep40");
  155                 if (ess->rsnciphers & IEEE80211_CIPHER_WEP104)
  156                         printf(" wep104");
  157                 if (ess->rsnciphers & IEEE80211_CIPHER_TKIP)
  158                         printf(" tkip");
  159                 if (ess->rsnciphers & IEEE80211_CIPHER_CCMP)
  160                         printf(" ccmp");
  161         }
  162         if (ess->flags & IEEE80211_F_WEPON) {
  163                 int i = ess->def_txkey;
  164 
  165                 printf(" wep,");
  166                 if (ess->nw_keys[i].k_cipher & IEEE80211_CIPHER_WEP40)
  167                         printf("wep40");
  168                 if (ess->nw_keys[i].k_cipher & IEEE80211_CIPHER_WEP104)
  169                         printf("wep104");
  170         }
  171         if (ess->flags == 0)
  172                 printf(" clear");
  173         printf("\n");
  174 }
  175 
  176 void
  177 ieee80211_print_ess_list(struct ieee80211com *ic)
  178 {
  179         struct ifnet            *ifp = &ic->ic_if;
  180         struct ieee80211_ess    *ess;
  181 
  182         printf("%s: known networks\n", ifp->if_xname);
  183         TAILQ_FOREACH(ess, &ic->ic_ess, ess_next) {
  184                 ieee80211_print_ess(ess);
  185         }
  186 }
  187 
  188 struct ieee80211_ess *
  189 ieee80211_get_ess(struct ieee80211com *ic, const char *nwid, int len)
  190 {
  191         struct ieee80211_ess    *ess;
  192 
  193         TAILQ_FOREACH(ess, &ic->ic_ess, ess_next) {
  194                 if (len == ess->esslen &&
  195                     memcmp(ess->essid, nwid, ess->esslen) == 0)
  196                         return ess;
  197         }
  198 
  199         return NULL;
  200 }
  201 
  202 void
  203 ieee80211_del_ess(struct ieee80211com *ic, char *nwid, int len, int all)
  204 {
  205         struct ieee80211_ess *ess, *next;
  206 
  207         TAILQ_FOREACH_SAFE(ess, &ic->ic_ess, ess_next, next) {
  208                 if (all == 1 || (ess->esslen == len &&
  209                     memcmp(ess->essid, nwid, len) == 0)) {
  210                         TAILQ_REMOVE(&ic->ic_ess, ess, ess_next);
  211                         explicit_bzero(ess, sizeof(*ess));
  212                         free(ess, M_DEVBUF, sizeof(*ess));
  213                         if (TAILQ_EMPTY(&ic->ic_ess))
  214                                 ic->ic_flags &= ~IEEE80211_F_AUTO_JOIN;
  215                         if (all != 1)
  216                                 return;
  217                 }
  218         }
  219 }
  220 
  221 /* Keep in sync with ieee80211_ioctl.c:ieee80211_ioctl_setnwkeys() */
  222 static int
  223 ieee80211_ess_setnwkeys(struct ieee80211_ess *ess,
  224     const struct ieee80211_nwkey *nwkey)
  225 {
  226         struct ieee80211_key *k;
  227         int error, i;
  228 
  229         if (nwkey->i_wepon == IEEE80211_NWKEY_OPEN) {
  230                 if (!(ess->flags & IEEE80211_F_WEPON))
  231                         return 0;
  232                 ess->flags &= ~IEEE80211_F_WEPON;
  233                 return ENETRESET;
  234         }
  235         if (nwkey->i_defkid < 1 || nwkey->i_defkid > IEEE80211_WEP_NKID)
  236                 return EINVAL;
  237 
  238         for (i = 0; i < IEEE80211_WEP_NKID; i++) {
  239                 if (nwkey->i_key[i].i_keylen == 0 ||
  240                     nwkey->i_key[i].i_keydat == NULL)
  241                         continue;       /* entry not set */
  242                 if (nwkey->i_key[i].i_keylen > IEEE80211_KEYBUF_SIZE)
  243                         return EINVAL;
  244 
  245                 /* map wep key to ieee80211_key */
  246                 k = &ess->nw_keys[i];
  247                 memset(k, 0, sizeof(*k));
  248                 if (nwkey->i_key[i].i_keylen <= 5)
  249                         k->k_cipher = IEEE80211_CIPHER_WEP40;
  250                 else
  251                         k->k_cipher = IEEE80211_CIPHER_WEP104;
  252                 k->k_len = ieee80211_cipher_keylen(k->k_cipher);
  253                 k->k_flags = IEEE80211_KEY_GROUP | IEEE80211_KEY_TX;
  254                 error = copyin(nwkey->i_key[i].i_keydat, k->k_key, k->k_len);
  255                 if (error != 0)
  256                         return error;
  257         }
  258         ess->def_txkey = nwkey->i_defkid - 1;
  259         ess->flags |= IEEE80211_F_WEPON;
  260 
  261         return ENETRESET;
  262 }
  263 
  264 
  265 /* Keep in sync with ieee80211_ioctl.c:ieee80211_ioctl_setwpaparms() */
  266 static int
  267 ieee80211_ess_setwpaparms(struct ieee80211_ess *ess,
  268     const struct ieee80211_wpaparams *wpa)
  269 {
  270         if (!wpa->i_enabled) {
  271                 if (!(ess->flags & IEEE80211_F_RSNON))
  272                         return 0;
  273                 ess->flags &= ~IEEE80211_F_RSNON;
  274                 ess->rsnprotos = 0;
  275                 ess->rsnakms = 0;
  276                 ess->rsngroupcipher = 0;
  277                 ess->rsnciphers = 0;
  278                 return ENETRESET;
  279         }
  280 
  281         ess->rsnprotos = 0;
  282         if (wpa->i_protos & IEEE80211_WPA_PROTO_WPA1)
  283                 ess->rsnprotos |= IEEE80211_PROTO_WPA;
  284         if (wpa->i_protos & IEEE80211_WPA_PROTO_WPA2)
  285                 ess->rsnprotos |= IEEE80211_PROTO_RSN;
  286         if (ess->rsnprotos == 0)        /* set to default (RSN) */
  287                 ess->rsnprotos = IEEE80211_PROTO_RSN;
  288 
  289         ess->rsnakms = 0;
  290         if (wpa->i_akms & IEEE80211_WPA_AKM_PSK)
  291                 ess->rsnakms |= IEEE80211_AKM_PSK;
  292         if (wpa->i_akms & IEEE80211_WPA_AKM_SHA256_PSK)
  293                 ess->rsnakms |= IEEE80211_AKM_SHA256_PSK;
  294         if (wpa->i_akms & IEEE80211_WPA_AKM_8021X)
  295                 ess->rsnakms |= IEEE80211_AKM_8021X;
  296         if (wpa->i_akms & IEEE80211_WPA_AKM_SHA256_8021X)
  297                 ess->rsnakms |= IEEE80211_AKM_SHA256_8021X;
  298         if (ess->rsnakms == 0)  /* set to default (PSK) */
  299                 ess->rsnakms = IEEE80211_AKM_PSK;
  300 
  301         if (wpa->i_groupcipher == IEEE80211_WPA_CIPHER_WEP40)
  302                 ess->rsngroupcipher = IEEE80211_CIPHER_WEP40;
  303         else if (wpa->i_groupcipher == IEEE80211_WPA_CIPHER_TKIP)
  304                 ess->rsngroupcipher = IEEE80211_CIPHER_TKIP;
  305         else if (wpa->i_groupcipher == IEEE80211_WPA_CIPHER_CCMP)
  306                 ess->rsngroupcipher = IEEE80211_CIPHER_CCMP;
  307         else if (wpa->i_groupcipher == IEEE80211_WPA_CIPHER_WEP104)
  308                 ess->rsngroupcipher = IEEE80211_CIPHER_WEP104;
  309         else  { /* set to default */
  310                 if (ess->rsnprotos & IEEE80211_PROTO_WPA)
  311                         ess->rsngroupcipher = IEEE80211_CIPHER_TKIP;
  312                 else
  313                         ess->rsngroupcipher = IEEE80211_CIPHER_CCMP;
  314         }
  315 
  316         ess->rsnciphers = 0;
  317         if (wpa->i_ciphers & IEEE80211_WPA_CIPHER_TKIP)
  318                 ess->rsnciphers |= IEEE80211_CIPHER_TKIP;
  319         if (wpa->i_ciphers & IEEE80211_WPA_CIPHER_CCMP)
  320                 ess->rsnciphers |= IEEE80211_CIPHER_CCMP;
  321         if (wpa->i_ciphers & IEEE80211_WPA_CIPHER_USEGROUP)
  322                 ess->rsnciphers = IEEE80211_CIPHER_USEGROUP;
  323         if (ess->rsnciphers == 0) { /* set to default (CCMP, TKIP if WPA1) */
  324                 ess->rsnciphers = IEEE80211_CIPHER_CCMP;
  325                 if (ess->rsnprotos & IEEE80211_PROTO_WPA)
  326                         ess->rsnciphers |= IEEE80211_CIPHER_TKIP;
  327         }
  328 
  329         ess->flags |= IEEE80211_F_RSNON;
  330 
  331         if (ess->rsnakms &
  332             (IEEE80211_AKM_8021X|IEEE80211_WPA_AKM_SHA256_8021X))
  333                 ess->flags |= IEEE80211_JOIN_8021X;
  334 
  335         return ENETRESET;
  336 }
  337 
  338 static void
  339 ieee80211_ess_clear_wep(struct ieee80211_ess *ess)
  340 {
  341         int i;
  342 
  343         /* Disable WEP */
  344         for (i = 0; i < IEEE80211_WEP_NKID; i++) {
  345                 explicit_bzero(&ess->nw_keys[i], sizeof(ess->nw_keys[0]));
  346         }
  347         ess->def_txkey = 0;
  348         ess->flags &= ~IEEE80211_F_WEPON;
  349 }
  350 
  351 static void
  352 ieee80211_ess_clear_wpa(struct ieee80211_ess *ess)
  353 {
  354         /* Disable WPA */
  355         ess->rsnprotos = ess->rsnakms = ess->rsngroupcipher =
  356             ess->rsnciphers = 0;
  357         explicit_bzero(ess->psk, sizeof(ess->psk));
  358         ess->flags &= ~(IEEE80211_F_PSK | IEEE80211_F_RSNON);
  359 }
  360 
  361 int
  362 ieee80211_add_ess(struct ieee80211com *ic, struct ieee80211_join *join)
  363 {
  364         struct ieee80211_ess *ess;
  365         int new = 0, ness = 0;
  366 
  367         /* only valid for station (aka, client) mode */
  368         if (ic->ic_opmode != IEEE80211_M_STA)
  369                 return (0);
  370 
  371         TAILQ_FOREACH(ess, &ic->ic_ess, ess_next) {
  372                 if (ess->esslen == join->i_len &&
  373                     memcmp(ess->essid, join->i_nwid, ess->esslen) == 0)
  374                         break;
  375                 ness++;
  376         }
  377 
  378         if (ess == NULL) {
  379                 /* if not found, and wpa/wep are set, then return */
  380                 if ((join->i_flags & IEEE80211_JOIN_WPA) &&
  381                     (join->i_flags & IEEE80211_JOIN_NWKEY)) {
  382                         return (EINVAL);
  383                 }
  384                 if (ness > IEEE80211_CACHE_SIZE)
  385                         return (ERANGE);
  386                 new = 1;
  387                 ess = malloc(sizeof(*ess), M_DEVBUF, M_NOWAIT|M_ZERO);
  388                 if (ess == NULL)
  389                         return (ENOMEM);
  390                 memcpy(ess->essid, join->i_nwid, join->i_len);
  391                 ess->esslen = join->i_len;
  392         }
  393 
  394         if (join->i_flags & IEEE80211_JOIN_WPA) {
  395                 if (join->i_wpaparams.i_enabled) {
  396                         if (!(ic->ic_caps & IEEE80211_C_RSN)) {
  397                                 free(ess, M_DEVBUF, sizeof(*ess));
  398                                 return ENODEV;
  399                         }
  400                         ieee80211_ess_setwpaparms(ess,
  401                             &join->i_wpaparams);
  402                         if (join->i_flags & IEEE80211_JOIN_WPAPSK) {
  403                                 ess->flags |= IEEE80211_F_PSK;
  404                                 explicit_bzero(ess->psk, sizeof(ess->psk));
  405                                 memcpy(ess->psk, &join->i_wpapsk.i_psk,
  406                                     sizeof(ess->psk));
  407                         }
  408                         ieee80211_ess_clear_wep(ess);
  409                 } else {
  410                         ieee80211_ess_clear_wpa(ess);
  411                 }
  412         } else if (join->i_flags & IEEE80211_JOIN_NWKEY) {
  413                 if (join->i_nwkey.i_wepon) {
  414                         if (!(ic->ic_caps & IEEE80211_C_WEP)) {
  415                                 free(ess, M_DEVBUF, sizeof(*ess));
  416                                 return ENODEV;
  417                         }
  418                         ieee80211_ess_setnwkeys(ess, &join->i_nwkey);
  419                         ieee80211_ess_clear_wpa(ess);
  420                 } else {
  421                         ieee80211_ess_clear_wep(ess);
  422                 }
  423         }
  424 
  425         if (new)
  426                 TAILQ_INSERT_TAIL(&ic->ic_ess, ess, ess_next);
  427 
  428         return (0);
  429 }
  430 
  431 uint8_t
  432 ieee80211_ess_adjust_rssi(struct ieee80211com *ic, struct ieee80211_node *ni)
  433 {
  434         uint8_t rssi = ni->ni_rssi;
  435 
  436         /*
  437          * Slightly punish 2 GHz RSSI values since they are usually
  438          * stronger than 5 GHz RSSI values.
  439          */
  440         if (IEEE80211_IS_CHAN_2GHZ(ni->ni_chan)) {
  441                 if (ic->ic_max_rssi) {
  442                         uint8_t p = (5 * ic->ic_max_rssi) / 100;
  443                         if (rssi >= p)
  444                                 rssi -= p; /* punish by 5% */
  445                 } else  {
  446                         if (rssi >= 8)
  447                                 rssi -= 8; /* punish by 8 dBm */
  448                 }
  449         }
  450 
  451         return rssi;
  452 }
  453 
  454 int
  455 ieee80211_ess_calculate_score(struct ieee80211com *ic,
  456     struct ieee80211_node *ni)
  457 {
  458         int score = 0;
  459         uint8_t min_5ghz_rssi;
  460 
  461         if (ic->ic_max_rssi)
  462                 min_5ghz_rssi = IEEE80211_RSSI_THRES_RATIO_5GHZ;
  463         else
  464                 min_5ghz_rssi = (uint8_t)IEEE80211_RSSI_THRES_5GHZ;
  465 
  466         /* not using join any */
  467         if (ieee80211_get_ess(ic, ni->ni_essid, ni->ni_esslen))
  468                 score += 32;
  469 
  470         /* Calculate the crypto score */
  471         if (ni->ni_rsnprotos & IEEE80211_PROTO_RSN)
  472                 score += 16;
  473         if (ni->ni_rsnprotos & IEEE80211_PROTO_WPA)
  474                 score += 8;
  475         if (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY)
  476                 score += 4;
  477 
  478         /* 5GHz with a good signal */
  479         if (IEEE80211_IS_CHAN_5GHZ(ni->ni_chan) &&
  480             ni->ni_rssi > min_5ghz_rssi)
  481                 score += 2;
  482 
  483         /* HT/VHT available */
  484         if (ieee80211_node_supports_ht(ni))
  485                 score++;
  486         if (ieee80211_node_supports_vht(ni))
  487                 score++;
  488 
  489         /* Boost this AP if it had no auth/assoc failures in the past. */
  490         if (ni->ni_fails == 0)
  491                 score += 21;
  492 
  493         return score;
  494 }
  495 
  496 /*
  497  * Given two APs, determine the "better" one of the two.
  498  * We compute a score based on the following attributes:
  499  *
  500  *  crypto: wpa2 > wpa1 > wep > open
  501  *  band: 5 GHz > 2 GHz provided 5 GHz rssi is above threshold
  502  *  supported standard revisions: 11ac > 11n > 11a/b/g
  503  *  rssi: rssi1 > rssi2 as a numeric comparison with a slight
  504  *         disadvantage for 2 GHz APs
  505  *
  506  * Crypto carries most weight, followed by band, followed by rssi.
  507  */
  508 int
  509 ieee80211_ess_is_better(struct ieee80211com *ic,
  510     struct ieee80211_node *nicur, struct ieee80211_node *nican)
  511 {
  512         struct ifnet            *ifp = &ic->ic_if;
  513         int                      score_cur = 0, score_can = 0;
  514         int                      cur_rssi, can_rssi;
  515 
  516         score_cur = ieee80211_ess_calculate_score(ic, nicur);
  517         score_can = ieee80211_ess_calculate_score(ic, nican);
  518 
  519         cur_rssi = ieee80211_ess_adjust_rssi(ic, nicur);
  520         can_rssi = ieee80211_ess_adjust_rssi(ic, nican);
  521 
  522         if (can_rssi > cur_rssi)
  523                 score_can++;
  524 
  525         if ((ifp->if_flags & IFF_DEBUG) && (score_can <= score_cur)) {
  526                 printf("%s: AP %s ", ifp->if_xname,
  527                     ether_sprintf(nican->ni_bssid));
  528                 ieee80211_print_essid(nican->ni_essid, nican->ni_esslen);
  529                 printf(" score %d\n", score_can);
  530         }
  531 
  532         return score_can > score_cur;
  533 }
  534 
  535 /* Determine whether a candidate AP belongs to a given ESS. */
  536 int
  537 ieee80211_match_ess(struct ieee80211_ess *ess, struct ieee80211_node *ni)
  538 {
  539         if (ess->esslen != 0 &&
  540             (ess->esslen != ni->ni_esslen ||
  541             memcmp(ess->essid, ni->ni_essid, ess->esslen) != 0)) {
  542                 ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_ESSID;
  543                 return 0;
  544         }
  545 
  546         if (ess->flags & (IEEE80211_F_PSK | IEEE80211_F_RSNON)) {
  547                 /* Ensure same WPA version. */
  548                 if ((ni->ni_rsnprotos & IEEE80211_PROTO_RSN) &&
  549                     (ess->rsnprotos & IEEE80211_PROTO_RSN) == 0) {
  550                         ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
  551                         return 0;
  552                 }
  553                 if ((ni->ni_rsnprotos & IEEE80211_PROTO_WPA) &&
  554                     (ess->rsnprotos & IEEE80211_PROTO_WPA) == 0) {
  555                         ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
  556                         return 0;
  557                 }
  558         } else if (ess->flags & IEEE80211_F_WEPON) {
  559                 if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0) {
  560                         ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_PRIVACY;
  561                         return 0;
  562                 }
  563         } else {
  564                 if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) != 0) {
  565                         ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_PRIVACY;
  566                         return 0;
  567                 }
  568         }
  569 
  570         if (ess->esslen == 0 &&
  571             (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) != 0) {
  572                 ni->ni_assoc_fail |= IEEE80211_NODE_ASSOCFAIL_PRIVACY;
  573                 return 0;
  574         }
  575 
  576         return 1;
  577 }
  578 
  579 void
  580 ieee80211_switch_ess(struct ieee80211com *ic)
  581 {
  582         struct ifnet            *ifp = &ic->ic_if;
  583         struct ieee80211_ess    *ess, *seless = NULL;
  584         struct ieee80211_node   *ni, *selni = NULL;
  585 
  586         if (!ISSET(ifp->if_flags, IFF_RUNNING))
  587                 return;
  588 
  589         /* Find the best AP matching an entry on our ESS join list. */
  590         RBT_FOREACH(ni, ieee80211_tree, &ic->ic_tree) {
  591                 if ((ic->ic_flags & IEEE80211_F_DESBSSID) &&
  592                     !IEEE80211_ADDR_EQ(ic->ic_des_bssid, ni->ni_bssid))
  593                         continue;
  594 
  595                 TAILQ_FOREACH(ess, &ic->ic_ess, ess_next) {
  596                         if (ieee80211_match_ess(ess, ni))
  597                                 break;
  598                 }
  599                 if (ess == NULL)
  600                         continue;
  601 
  602                 /*
  603                  * Operate only on ic_des_essid if auto-join is disabled.
  604                  * We might have a password stored for this network.
  605                  */
  606                 if (!ISSET(ic->ic_flags, IEEE80211_F_AUTO_JOIN)) {
  607                         if (ic->ic_des_esslen == ni->ni_esslen &&
  608                             memcmp(ic->ic_des_essid, ni->ni_essid,
  609                             ni->ni_esslen) == 0) {
  610                                 ieee80211_set_ess(ic, ess, ni);
  611                                 return;
  612                         }
  613                         continue;
  614                 }
  615 
  616                 if (selni == NULL) {
  617                         seless = ess;
  618                         selni = ni;
  619                         continue;
  620                 }
  621 
  622                 if (ieee80211_ess_is_better(ic, selni, ni)) {
  623                         seless = ess;
  624                         selni = ni;
  625                 }
  626         }
  627 
  628         if (selni && seless && !(selni->ni_esslen == ic->ic_des_esslen &&
  629             (memcmp(ic->ic_des_essid, selni->ni_essid,
  630              IEEE80211_NWID_LEN) == 0))) {
  631                 if (ifp->if_flags & IFF_DEBUG) {
  632                         printf("%s: best AP %s ", ifp->if_xname,
  633                             ether_sprintf(selni->ni_bssid));
  634                         ieee80211_print_essid(selni->ni_essid,
  635                             selni->ni_esslen);
  636                         printf(" score %d\n",
  637                             ieee80211_ess_calculate_score(ic, selni));
  638                         printf("%s: switching to network ", ifp->if_xname);
  639                         ieee80211_print_essid(selni->ni_essid,
  640                             selni->ni_esslen);
  641                         if (seless->esslen == 0)
  642                                 printf(" via join any");
  643                         printf("\n");
  644 
  645                 }
  646                 ieee80211_set_ess(ic, seless, selni);
  647         }
  648 }
  649 
  650 void
  651 ieee80211_set_ess(struct ieee80211com *ic, struct ieee80211_ess *ess, 
  652     struct ieee80211_node *ni)
  653 {
  654         memset(ic->ic_des_essid, 0, IEEE80211_NWID_LEN);
  655         ic->ic_des_esslen = ni->ni_esslen;
  656         memcpy(ic->ic_des_essid, ni->ni_essid, ic->ic_des_esslen);
  657 
  658         ieee80211_disable_wep(ic);
  659         ieee80211_disable_rsn(ic);
  660 
  661         if (ess->flags & IEEE80211_F_RSNON) {
  662                 explicit_bzero(ic->ic_psk, sizeof(ic->ic_psk));
  663                 memcpy(ic->ic_psk, ess->psk, sizeof(ic->ic_psk));
  664 
  665                 ic->ic_rsnprotos = ess->rsnprotos;
  666                 ic->ic_rsnakms = ess->rsnakms;
  667                 ic->ic_rsngroupcipher = ess->rsngroupcipher;
  668                 ic->ic_rsnciphers = ess->rsnciphers;
  669                 ic->ic_flags |= IEEE80211_F_RSNON;
  670                 if (ess->flags & IEEE80211_F_PSK)
  671                         ic->ic_flags |= IEEE80211_F_PSK;
  672         } else if (ess->flags & IEEE80211_F_WEPON) {
  673                 struct ieee80211_key    *k;
  674                 int                      i;
  675 
  676                 for (i = 0; i < IEEE80211_WEP_NKID; i++) {
  677                         k = &ic->ic_nw_keys[i];
  678                         if (k->k_cipher != IEEE80211_CIPHER_NONE)
  679                                 (*ic->ic_delete_key)(ic, NULL, k);
  680                         memcpy(&ic->ic_nw_keys[i], &ess->nw_keys[i],
  681                             sizeof(struct ieee80211_key));
  682                         if (k->k_cipher != IEEE80211_CIPHER_NONE)
  683                                 (*ic->ic_set_key)(ic, NULL, k);
  684                 }
  685                 ic->ic_def_txkey = ess->def_txkey;
  686                 ic->ic_flags |= IEEE80211_F_WEPON;
  687         }
  688 }
  689 
  690 void
  691 ieee80211_deselect_ess(struct ieee80211com *ic)
  692 {
  693         memset(ic->ic_des_essid, 0, IEEE80211_NWID_LEN);
  694         ic->ic_des_esslen = 0;
  695         ieee80211_disable_wep(ic);
  696         ieee80211_disable_rsn(ic);
  697 }
  698 
  699 void
  700 ieee80211_node_attach(struct ifnet *ifp)
  701 {
  702         struct ieee80211com *ic = (void *)ifp;
  703 #ifndef IEEE80211_STA_ONLY
  704         int size;
  705 #endif
  706 
  707         RBT_INIT(ieee80211_tree, &ic->ic_tree);
  708         ic->ic_node_alloc = ieee80211_node_alloc;
  709         ic->ic_node_free = ieee80211_node_free;
  710         ic->ic_node_copy = ieee80211_node_copy;
  711         ic->ic_node_getrssi = ieee80211_node_getrssi;
  712         ic->ic_node_checkrssi = ieee80211_node_checkrssi;
  713         ic->ic_scangen = 1;
  714         ic->ic_max_nnodes = ieee80211_cache_size;
  715 
  716         if (ic->ic_max_aid == 0)
  717                 ic->ic_max_aid = IEEE80211_AID_DEF;
  718         else if (ic->ic_max_aid > IEEE80211_AID_MAX)
  719                 ic->ic_max_aid = IEEE80211_AID_MAX;
  720 #ifndef IEEE80211_STA_ONLY
  721         size = howmany(ic->ic_max_aid, 32) * sizeof(u_int32_t);
  722         ic->ic_aid_bitmap = malloc(size, M_DEVBUF, M_NOWAIT | M_ZERO);
  723         if (ic->ic_aid_bitmap == NULL) {
  724                 /* XXX no way to recover */
  725                 printf("%s: no memory for AID bitmap!\n", __func__);
  726                 ic->ic_max_aid = 0;
  727         }
  728         if (ic->ic_caps & (IEEE80211_C_HOSTAP | IEEE80211_C_IBSS)) {
  729                 ic->ic_tim_len = howmany(ic->ic_max_aid, 8);
  730                 ic->ic_tim_bitmap = malloc(ic->ic_tim_len, M_DEVBUF,
  731                     M_NOWAIT | M_ZERO);
  732                 if (ic->ic_tim_bitmap == NULL) {
  733                         printf("%s: no memory for TIM bitmap!\n", __func__);
  734                         ic->ic_tim_len = 0;
  735                 } else
  736                         ic->ic_set_tim = ieee80211_set_tim;
  737                 timeout_set(&ic->ic_rsn_timeout,
  738                     ieee80211_gtk_rekey_timeout, ic);
  739                 timeout_set(&ic->ic_inact_timeout,
  740                     ieee80211_inact_timeout, ic);
  741                 timeout_set(&ic->ic_node_cache_timeout,
  742                     ieee80211_node_cache_timeout, ic);
  743         }
  744 #endif
  745         TAILQ_INIT(&ic->ic_ess);
  746 }
  747 
  748 struct ieee80211_node *
  749 ieee80211_alloc_node_helper(struct ieee80211com *ic)
  750 {
  751         struct ieee80211_node *ni;
  752         if (ic->ic_nnodes >= ic->ic_max_nnodes)
  753                 ieee80211_clean_nodes(ic, 0);
  754         if (ic->ic_nnodes >= ic->ic_max_nnodes)
  755                 return NULL;
  756         ni = (*ic->ic_node_alloc)(ic);
  757         return ni;
  758 }
  759 
  760 void
  761 ieee80211_node_lateattach(struct ifnet *ifp)
  762 {
  763         struct ieee80211com *ic = (void *)ifp;
  764         struct ieee80211_node *ni;
  765 
  766         ni = ieee80211_alloc_node_helper(ic);
  767         if (ni == NULL)
  768                 panic("unable to setup initial BSS node");
  769         ni->ni_chan = IEEE80211_CHAN_ANYC;
  770         ic->ic_bss = ieee80211_ref_node(ni);
  771         ic->ic_txpower = IEEE80211_TXPOWER_MAX;
  772 #ifndef IEEE80211_STA_ONLY
  773         mq_init(&ni->ni_savedq, IEEE80211_PS_MAX_QUEUE, IPL_NET);
  774 #endif
  775 }
  776 
  777 void
  778 ieee80211_node_detach(struct ifnet *ifp)
  779 {
  780         struct ieee80211com *ic = (void *)ifp;
  781 
  782         if (ic->ic_bss != NULL) {
  783                 (*ic->ic_node_free)(ic, ic->ic_bss);
  784                 ic->ic_bss = NULL;
  785         }
  786         ieee80211_del_ess(ic, NULL, 0, 1);
  787         ieee80211_free_allnodes(ic, 1);
  788 #ifndef IEEE80211_STA_ONLY
  789         free(ic->ic_aid_bitmap, M_DEVBUF,
  790             howmany(ic->ic_max_aid, 32) * sizeof(u_int32_t));
  791         free(ic->ic_tim_bitmap, M_DEVBUF, ic->ic_tim_len);
  792         timeout_del(&ic->ic_inact_timeout);
  793         timeout_del(&ic->ic_node_cache_timeout);
  794         timeout_del(&ic->ic_tkip_micfail_timeout);
  795 #endif
  796         timeout_del(&ic->ic_rsn_timeout);
  797 }
  798 
  799 /*
  800  * AP scanning support.
  801  */
  802 
  803 /*
  804  * Initialize the active channel set based on the set
  805  * of available channels and the current PHY mode.
  806  */
  807 void
  808 ieee80211_reset_scan(struct ifnet *ifp)
  809 {
  810         struct ieee80211com *ic = (void *)ifp;
  811 
  812         memcpy(ic->ic_chan_scan, ic->ic_chan_active,
  813                 sizeof(ic->ic_chan_active));
  814         /* NB: hack, setup so next_scan starts with the first channel */
  815         if (ic->ic_bss != NULL && ic->ic_bss->ni_chan == IEEE80211_CHAN_ANYC)
  816                 ic->ic_bss->ni_chan = &ic->ic_channels[IEEE80211_CHAN_MAX];
  817 }
  818 
  819 /*
  820  * Increase a node's inactivity counter.
  821  * This counter get reset to zero if a frame is received.
  822  * This function is intended for station mode only.
  823  * See ieee80211_node_cache_timeout() for hostap mode.
  824  */
  825 void
  826 ieee80211_node_raise_inact(void *arg, struct ieee80211_node *ni)
  827 {
  828         if (ni->ni_refcnt == 0 && ni->ni_inact < IEEE80211_INACT_SCAN)
  829                 ni->ni_inact++;
  830 }
  831 
  832 /*
  833  * Begin an active scan.
  834  */
  835 void
  836 ieee80211_begin_scan(struct ifnet *ifp)
  837 {
  838         struct ieee80211com *ic = (void *)ifp;
  839 
  840         /*
  841          * In all but hostap mode scanning starts off in
  842          * an active mode before switching to passive.
  843          */
  844 #ifndef IEEE80211_STA_ONLY
  845         if (ic->ic_opmode != IEEE80211_M_HOSTAP)
  846 #endif
  847         {
  848                 ic->ic_flags |= IEEE80211_F_ASCAN;
  849                 ic->ic_stats.is_scan_active++;
  850         }
  851 #ifndef IEEE80211_STA_ONLY
  852         else
  853                 ic->ic_stats.is_scan_passive++;
  854 #endif
  855         if (ifp->if_flags & IFF_DEBUG)
  856                 printf("%s: begin %s scan\n", ifp->if_xname,
  857                         (ic->ic_flags & IEEE80211_F_ASCAN) ?
  858                                 "active" : "passive");
  859 
  860 
  861         if (ic->ic_opmode == IEEE80211_M_STA) {
  862                 ieee80211_node_cleanup(ic, ic->ic_bss);
  863                 ieee80211_iterate_nodes(ic, ieee80211_node_raise_inact, NULL);
  864         }
  865 
  866         /*
  867          * Reset the current mode. Setting the current mode will also
  868          * reset scan state.
  869          */
  870         if (IFM_MODE(ic->ic_media.ifm_cur->ifm_media) == IFM_AUTO)
  871                 ic->ic_curmode = IEEE80211_MODE_AUTO;
  872         ieee80211_setmode(ic, ic->ic_curmode);
  873 
  874         ic->ic_scan_count = 0;
  875 
  876         /* Scan the next channel. */
  877         ieee80211_next_scan(ifp);
  878 }
  879 
  880 /*
  881  * Switch to the next channel marked for scanning.
  882  */
  883 void
  884 ieee80211_next_scan(struct ifnet *ifp)
  885 {
  886         struct ieee80211com *ic = (void *)ifp;
  887         struct ieee80211_channel *chan;
  888 
  889         chan = ic->ic_bss->ni_chan;
  890         for (;;) {
  891                 if (++chan > &ic->ic_channels[IEEE80211_CHAN_MAX])
  892                         chan = &ic->ic_channels[0];
  893                 if (isset(ic->ic_chan_scan, ieee80211_chan2ieee(ic, chan))) {
  894                         /*
  895                          * Ignore channels marked passive-only
  896                          * during an active scan.
  897                          */
  898                         if ((ic->ic_flags & IEEE80211_F_ASCAN) == 0 ||
  899                             (chan->ic_flags & IEEE80211_CHAN_PASSIVE) == 0)
  900                                 break;
  901                 }
  902                 if (chan == ic->ic_bss->ni_chan) {
  903                         ieee80211_end_scan(ifp);
  904                         return;
  905                 }
  906         }
  907         clrbit(ic->ic_chan_scan, ieee80211_chan2ieee(ic, chan));
  908         DPRINTF(("chan %d->%d\n",
  909             ieee80211_chan2ieee(ic, ic->ic_bss->ni_chan),
  910             ieee80211_chan2ieee(ic, chan)));
  911         ic->ic_bss->ni_chan = chan;
  912         ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
  913 }
  914 
  915 #ifndef IEEE80211_STA_ONLY
  916 void
  917 ieee80211_create_ibss(struct ieee80211com* ic, struct ieee80211_channel *chan)
  918 {
  919         enum ieee80211_phymode mode;
  920         struct ieee80211_node *ni;
  921         struct ifnet *ifp = &ic->ic_if;
  922 
  923         ni = ic->ic_bss;
  924         if (ifp->if_flags & IFF_DEBUG)
  925                 printf("%s: creating ibss\n", ifp->if_xname);
  926         ic->ic_flags |= IEEE80211_F_SIBSS;
  927         ni->ni_chan = chan;
  928         if ((ic->ic_flags & IEEE80211_F_VHTON) && IEEE80211_IS_CHAN_5GHZ(chan))
  929                 mode = IEEE80211_MODE_11AC;
  930         else if (ic->ic_flags & IEEE80211_F_HTON)
  931                 mode = IEEE80211_MODE_11N;
  932         else
  933                 mode = ieee80211_chan2mode(ic, ni->ni_chan);
  934         ieee80211_setmode(ic, mode);
  935         /* Pick an appropriate mode for supported legacy rates. */
  936         if (ic->ic_curmode == IEEE80211_MODE_11AC) {
  937                 mode = IEEE80211_MODE_11A;
  938         } else if (ic->ic_curmode == IEEE80211_MODE_11N) {
  939                 if (IEEE80211_IS_CHAN_5GHZ(chan))
  940                         mode = IEEE80211_MODE_11A;
  941                 else
  942                         mode = IEEE80211_MODE_11G;
  943         } else {
  944                 mode = ic->ic_curmode;
  945         }
  946         ni->ni_rates = ic->ic_sup_rates[mode];
  947         ni->ni_txrate = 0;
  948         IEEE80211_ADDR_COPY(ni->ni_macaddr, ic->ic_myaddr);
  949         IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_myaddr);
  950         if (ic->ic_opmode == IEEE80211_M_IBSS) {
  951                 if ((ic->ic_flags & IEEE80211_F_DESBSSID) != 0)
  952                         IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_des_bssid);
  953                 else
  954                         ni->ni_bssid[0] |= 0x02;        /* local bit for IBSS */
  955         }
  956         ni->ni_esslen = ic->ic_des_esslen;
  957         memcpy(ni->ni_essid, ic->ic_des_essid, ni->ni_esslen);
  958         ni->ni_rssi = 0;
  959         ni->ni_rstamp = 0;
  960         memset(ni->ni_tstamp, 0, sizeof(ni->ni_tstamp));
  961         ni->ni_intval = ic->ic_lintval;
  962         ni->ni_capinfo = IEEE80211_CAPINFO_IBSS;
  963         if (ic->ic_flags & IEEE80211_F_WEPON)
  964                 ni->ni_capinfo |= IEEE80211_CAPINFO_PRIVACY;
  965         if (ic->ic_flags & IEEE80211_F_HTON) {
  966                 const struct ieee80211_edca_ac_params *ac_qap;
  967                 struct ieee80211_edca_ac_params *ac;
  968                 int aci;
  969 
  970                 /* 
  971                  * Configure HT protection. This will be updated later
  972                  * based on the number of non-HT nodes in the node cache.
  973                  */
  974                 ic->ic_protmode = IEEE80211_PROT_NONE;
  975                 ni->ni_htop1 = IEEE80211_HTPROT_NONE;
  976                 /* Disallow Greenfield mode. None of our drivers support it. */
  977                 ni->ni_htop1 |= IEEE80211_HTOP1_NONGF_STA;
  978                 if (ic->ic_updateprot)
  979                         ic->ic_updateprot(ic);
  980 
  981                 /* Configure QoS EDCA parameters. */
  982                 for (aci = 0; aci < EDCA_NUM_AC; aci++) {
  983                         ac = &ic->ic_edca_ac[aci];
  984                         ac_qap = &ieee80211_qap_edca_table[ic->ic_curmode][aci];
  985                         ac->ac_acm       = ac_qap->ac_acm;
  986                         ac->ac_aifsn     = ac_qap->ac_aifsn;
  987                         ac->ac_ecwmin    = ac_qap->ac_ecwmin;
  988                         ac->ac_ecwmax    = ac_qap->ac_ecwmax;
  989                         ac->ac_txoplimit = ac_qap->ac_txoplimit;
  990                 }
  991                 if (ic->ic_updateedca)
  992                         (*ic->ic_updateedca)(ic);
  993         }
  994         if (ic->ic_flags & IEEE80211_F_RSNON) {
  995                 struct ieee80211_key *k;
  996 
  997                 /* initialize 256-bit global key counter to a random value */
  998                 arc4random_buf(ic->ic_globalcnt, EAPOL_KEY_NONCE_LEN);
  999 
 1000                 ni->ni_rsnprotos = ic->ic_rsnprotos;
 1001                 ni->ni_rsnakms = ic->ic_rsnakms;
 1002                 ni->ni_rsnciphers = ic->ic_rsnciphers;
 1003                 ni->ni_rsngroupcipher = ic->ic_rsngroupcipher;
 1004                 ni->ni_rsngroupmgmtcipher = ic->ic_rsngroupmgmtcipher;
 1005                 ni->ni_rsncaps = 0;
 1006                 if (ic->ic_caps & IEEE80211_C_MFP) {
 1007                         ni->ni_rsncaps |= IEEE80211_RSNCAP_MFPC;
 1008                         if (ic->ic_flags & IEEE80211_F_MFPR)
 1009                                 ni->ni_rsncaps |= IEEE80211_RSNCAP_MFPR;
 1010                 }
 1011 
 1012                 ic->ic_def_txkey = 1;
 1013                 ic->ic_flags &= ~IEEE80211_F_COUNTERM;
 1014                 k = &ic->ic_nw_keys[ic->ic_def_txkey];
 1015                 memset(k, 0, sizeof(*k));
 1016                 k->k_id = ic->ic_def_txkey;
 1017                 k->k_cipher = ni->ni_rsngroupcipher;
 1018                 k->k_flags = IEEE80211_KEY_GROUP | IEEE80211_KEY_TX;
 1019                 k->k_len = ieee80211_cipher_keylen(k->k_cipher);
 1020                 arc4random_buf(k->k_key, k->k_len);
 1021                 (*ic->ic_set_key)(ic, ni, k);   /* XXX */
 1022 
 1023                 if (ic->ic_caps & IEEE80211_C_MFP) {
 1024                         ic->ic_igtk_kid = 4;
 1025                         k = &ic->ic_nw_keys[ic->ic_igtk_kid];
 1026                         memset(k, 0, sizeof(*k));
 1027                         k->k_id = ic->ic_igtk_kid;
 1028                         k->k_cipher = ni->ni_rsngroupmgmtcipher;
 1029                         k->k_flags = IEEE80211_KEY_IGTK | IEEE80211_KEY_TX;
 1030                         k->k_len = 16;
 1031                         arc4random_buf(k->k_key, k->k_len);
 1032                         (*ic->ic_set_key)(ic, ni, k);   /* XXX */
 1033                 }
 1034                 /*
 1035                  * In HostAP mode, multicast traffic is sent using ic_bss
 1036                  * as the Tx node, so mark our node as valid so we can send
 1037                  * multicast frames using the group key we've just configured.
 1038                  */
 1039                 ni->ni_port_valid = 1;
 1040                 ni->ni_flags |= IEEE80211_NODE_TXPROT;
 1041 
 1042                 /* schedule a GTK/IGTK rekeying after 3600s */
 1043                 timeout_add_sec(&ic->ic_rsn_timeout, 3600);
 1044         }
 1045         timeout_add_sec(&ic->ic_inact_timeout, IEEE80211_INACT_WAIT);
 1046         timeout_add_sec(&ic->ic_node_cache_timeout, IEEE80211_CACHE_WAIT);
 1047         ieee80211_new_state(ic, IEEE80211_S_RUN, -1);
 1048 }
 1049 #endif  /* IEEE80211_STA_ONLY */
 1050 
 1051 int
 1052 ieee80211_match_bss(struct ieee80211com *ic, struct ieee80211_node *ni,
 1053     int bgscan)
 1054 {
 1055         u_int8_t rate;
 1056         int fail;
 1057 
 1058         fail = 0;
 1059         if ((ic->ic_flags & IEEE80211_F_BGSCAN) == 0 &&
 1060             isclr(ic->ic_chan_active, ieee80211_chan2ieee(ic, ni->ni_chan)))
 1061                 fail |= IEEE80211_NODE_ASSOCFAIL_CHAN;
 1062         if (ic->ic_des_chan != IEEE80211_CHAN_ANYC &&
 1063             ni->ni_chan != ic->ic_des_chan)
 1064                 fail |= IEEE80211_NODE_ASSOCFAIL_CHAN;
 1065 #ifndef IEEE80211_STA_ONLY
 1066         if (ic->ic_opmode == IEEE80211_M_IBSS) {
 1067                 if ((ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) == 0)
 1068                         fail |= IEEE80211_NODE_ASSOCFAIL_IBSS;
 1069         } else
 1070 #endif
 1071         {
 1072                 if ((ni->ni_capinfo & IEEE80211_CAPINFO_ESS) == 0)
 1073                         fail |= IEEE80211_NODE_ASSOCFAIL_IBSS;
 1074         }
 1075         if (ic->ic_flags & (IEEE80211_F_WEPON | IEEE80211_F_RSNON)) {
 1076                 if ((ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) == 0)
 1077                         fail |= IEEE80211_NODE_ASSOCFAIL_PRIVACY;
 1078         } else {
 1079                 if (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY)
 1080                         fail |= IEEE80211_NODE_ASSOCFAIL_PRIVACY;
 1081         }
 1082 
 1083         rate = ieee80211_fix_rate(ic, ni, IEEE80211_F_DONEGO);
 1084         if (rate & IEEE80211_RATE_BASIC)
 1085                 fail |= IEEE80211_NODE_ASSOCFAIL_BASIC_RATE;
 1086         if (ic->ic_des_esslen == 0)
 1087                 fail |= IEEE80211_NODE_ASSOCFAIL_ESSID;
 1088         if (ic->ic_des_esslen != 0 &&
 1089             (ni->ni_esslen != ic->ic_des_esslen ||
 1090              memcmp(ni->ni_essid, ic->ic_des_essid, ic->ic_des_esslen) != 0))
 1091                 fail |= IEEE80211_NODE_ASSOCFAIL_ESSID;
 1092         if ((ic->ic_flags & IEEE80211_F_DESBSSID) &&
 1093             !IEEE80211_ADDR_EQ(ic->ic_des_bssid, ni->ni_bssid))
 1094                 fail |= IEEE80211_NODE_ASSOCFAIL_BSSID;
 1095 
 1096         if (ic->ic_flags & IEEE80211_F_RSNON) {
 1097                 /*
 1098                  * If at least one RSN IE field from the AP's RSN IE fails
 1099                  * to overlap with any value the STA supports, the STA shall
 1100                  * decline to associate with that AP.
 1101                  */
 1102                 if ((ni->ni_rsnprotos & ic->ic_rsnprotos) == 0)
 1103                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1104                 if ((ni->ni_rsnakms & ic->ic_rsnakms) == 0)
 1105                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1106                 if ((ni->ni_rsnakms & ic->ic_rsnakms &
 1107                      ~(IEEE80211_AKM_PSK | IEEE80211_AKM_SHA256_PSK)) == 0) {
 1108                         /* AP only supports PSK AKMPs */
 1109                         if (!(ic->ic_flags & IEEE80211_F_PSK))
 1110                                 fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1111                 }
 1112                 if (ni->ni_rsngroupcipher != IEEE80211_CIPHER_WEP40 &&
 1113                     ni->ni_rsngroupcipher != IEEE80211_CIPHER_TKIP &&
 1114                     ni->ni_rsngroupcipher != IEEE80211_CIPHER_CCMP &&
 1115                     ni->ni_rsngroupcipher != IEEE80211_CIPHER_WEP104)
 1116                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1117                 if ((ni->ni_rsnciphers & ic->ic_rsnciphers) == 0)
 1118                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1119 
 1120                 /* we only support BIP as the IGTK cipher */
 1121                 if ((ni->ni_rsncaps & IEEE80211_RSNCAP_MFPC) &&
 1122                     ni->ni_rsngroupmgmtcipher != IEEE80211_CIPHER_BIP)
 1123                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1124 
 1125                 /* we do not support MFP but AP requires it */
 1126                 if (!(ic->ic_caps & IEEE80211_C_MFP) &&
 1127                     (ni->ni_rsncaps & IEEE80211_RSNCAP_MFPR))
 1128                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1129 
 1130                 /* we require MFP but AP does not support it */
 1131                 if ((ic->ic_caps & IEEE80211_C_MFP) &&
 1132                     (ic->ic_flags & IEEE80211_F_MFPR) &&
 1133                     !(ni->ni_rsncaps & IEEE80211_RSNCAP_MFPC))
 1134                         fail |= IEEE80211_NODE_ASSOCFAIL_WPA_PROTO;
 1135         }
 1136 
 1137         if (ic->ic_if.if_flags & IFF_DEBUG) {
 1138                 printf("%s: %c %s%c", ic->ic_if.if_xname, fail ? '-' : '+',
 1139                     ether_sprintf(ni->ni_bssid),
 1140                     fail & IEEE80211_NODE_ASSOCFAIL_BSSID ? '!' : ' ');
 1141                 printf(" %3d%c", ieee80211_chan2ieee(ic, ni->ni_chan),
 1142                         fail & IEEE80211_NODE_ASSOCFAIL_CHAN ? '!' : ' ');
 1143                 printf(" %+4d", ni->ni_rssi);
 1144                 printf(" %2dM%c", (rate & IEEE80211_RATE_VAL) / 2,
 1145                     fail & IEEE80211_NODE_ASSOCFAIL_BASIC_RATE ? '!' : ' ');
 1146                 printf(" %4s%c",
 1147                     (ni->ni_capinfo & IEEE80211_CAPINFO_ESS) ? "ess" :
 1148                     (ni->ni_capinfo & IEEE80211_CAPINFO_IBSS) ? "ibss" :
 1149                     "????",
 1150                     fail & IEEE80211_NODE_ASSOCFAIL_IBSS ? '!' : ' ');
 1151                 printf(" %7s%c ",
 1152                     (ni->ni_capinfo & IEEE80211_CAPINFO_PRIVACY) ?
 1153                     "privacy" : "no",
 1154                     fail & IEEE80211_NODE_ASSOCFAIL_PRIVACY ? '!' : ' ');
 1155                 printf(" %3s%c ",
 1156                     (ic->ic_flags & IEEE80211_F_RSNON) ?
 1157                     "rsn" : "no",
 1158                     fail & IEEE80211_NODE_ASSOCFAIL_WPA_PROTO ? '!' : ' ');
 1159                 ieee80211_print_essid(ni->ni_essid, ni->ni_esslen);
 1160                 printf("%s\n",
 1161                     fail & IEEE80211_NODE_ASSOCFAIL_ESSID ? "!" : "");
 1162         }
 1163 
 1164         /* We don't care about unrelated networks during background scans. */
 1165         if (bgscan) {
 1166                 if ((fail & IEEE80211_NODE_ASSOCFAIL_ESSID) == 0)
 1167                         ni->ni_assoc_fail = fail;
 1168         } else
 1169                 ni->ni_assoc_fail = fail;
 1170         if ((fail & IEEE80211_NODE_ASSOCFAIL_ESSID) == 0)
 1171                 ic->ic_bss->ni_assoc_fail = ni->ni_assoc_fail;
 1172 
 1173         return fail;
 1174 }
 1175 
 1176 struct ieee80211_node_switch_bss_arg {
 1177         u_int8_t cur_macaddr[IEEE80211_ADDR_LEN];
 1178         u_int8_t sel_macaddr[IEEE80211_ADDR_LEN];
 1179 };
 1180 
 1181 void
 1182 ieee80211_node_free_unref_cb(struct ieee80211_node *ni)
 1183 {
 1184         free(ni->ni_unref_arg, M_DEVBUF, ni->ni_unref_arg_size);
 1185 
 1186         /* Guard against accidental reuse. */
 1187         ni->ni_unref_cb = NULL;
 1188         ni->ni_unref_arg = NULL;
 1189         ni->ni_unref_arg_size = 0;
 1190 }
 1191 
 1192 /* Implements ni->ni_unref_cb(). */
 1193 void
 1194 ieee80211_node_tx_stopped(struct ieee80211com *ic,
 1195     struct ieee80211_node *ni)
 1196 {
 1197         splassert(IPL_NET);
 1198 
 1199         if ((ic->ic_flags & IEEE80211_F_BGSCAN) == 0)
 1200                 return;
 1201 
 1202         /* 
 1203          * Install a callback which will switch us to the new AP once
 1204          * the de-auth frame has been processed by hardware.
 1205          * Pass on the existing ni->ni_unref_arg argument.
 1206          */
 1207         ic->ic_bss->ni_unref_cb = ieee80211_node_switch_bss;
 1208 
 1209         /* 
 1210          * All data frames queued to hardware have been flushed and
 1211          * A-MPDU Tx has been stopped. We are now going to switch APs.
 1212          * Queue a de-auth frame addressed at our current AP.
 1213          */
 1214         if (IEEE80211_SEND_MGMT(ic, ic->ic_bss,
 1215             IEEE80211_FC0_SUBTYPE_DEAUTH,
 1216             IEEE80211_REASON_AUTH_LEAVE) != 0) {
 1217                 ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1218                 ieee80211_node_free_unref_cb(ni);
 1219                 ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
 1220                 return;
 1221         }
 1222 
 1223         /* F_BGSCAN flag gets cleared in ieee80211_node_join_bss(). */
 1224 }
 1225 
 1226 /* Implements ni->ni_unref_cb(). */
 1227 void
 1228 ieee80211_node_tx_flushed(struct ieee80211com *ic, struct ieee80211_node *ni)
 1229 {
 1230         splassert(IPL_NET);
 1231 
 1232         if ((ic->ic_flags & IEEE80211_F_BGSCAN) == 0)
 1233                 return;
 1234 
 1235         /* All data frames queued to hardware have been flushed. */
 1236         if (ic->ic_caps & IEEE80211_C_TX_AMPDU) {
 1237                 /* 
 1238                  * Install a callback which will switch us to the
 1239                  * new AP once Tx agg sessions have been stopped,
 1240                  * which involves sending a DELBA frame.
 1241                  * Pass on the existing ni->ni_unref_arg argument.
 1242                  */
 1243                 ic->ic_bss->ni_unref_cb = ieee80211_node_tx_stopped;
 1244                 ieee80211_stop_ampdu_tx(ic, ic->ic_bss,
 1245                     IEEE80211_FC0_SUBTYPE_DEAUTH);
 1246         } else
 1247                 ieee80211_node_tx_stopped(ic, ni);
 1248 }
 1249 
 1250 /* Implements ni->ni_unref_cb(). */
 1251 void
 1252 ieee80211_node_switch_bss(struct ieee80211com *ic, struct ieee80211_node *ni)
 1253 {
 1254         struct ifnet *ifp = &ic->ic_if;
 1255         struct ieee80211_node_switch_bss_arg *sba = ni->ni_unref_arg;
 1256         struct ieee80211_node *curbs, *selbs;
 1257 
 1258         splassert(IPL_NET);
 1259 
 1260         if ((ic->ic_flags & IEEE80211_F_BGSCAN) == 0)
 1261                 return;
 1262 
 1263         ic->ic_xflags &= ~IEEE80211_F_TX_MGMT_ONLY;
 1264 
 1265         selbs = ieee80211_find_node(ic, sba->sel_macaddr);
 1266         if (selbs == NULL) {
 1267                 ieee80211_node_free_unref_cb(ni);
 1268                 ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1269                 ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
 1270                 return;
 1271         }
 1272 
 1273         curbs = ieee80211_find_node(ic, sba->cur_macaddr);
 1274         if (curbs == NULL) {
 1275                 ieee80211_node_free_unref_cb(ni);
 1276                 ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1277                 ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
 1278                 return;
 1279         }
 1280 
 1281         if (ifp->if_flags & IFF_DEBUG) {
 1282                 printf("%s: roaming from %s chan %d ",
 1283                     ifp->if_xname, ether_sprintf(curbs->ni_macaddr),
 1284                     ieee80211_chan2ieee(ic, curbs->ni_chan));
 1285                 printf("to %s chan %d\n", ether_sprintf(selbs->ni_macaddr),
 1286                     ieee80211_chan2ieee(ic, selbs->ni_chan));
 1287         }
 1288         ieee80211_node_newstate(curbs, IEEE80211_STA_CACHE);
 1289         /*
 1290          * ieee80211_node_join_bss() frees arg and ic->ic_bss via
 1291          * ic->ic_node_copy() in ieee80211_node_cleanup().
 1292          */
 1293         ieee80211_node_join_bss(ic, selbs);
 1294 }
 1295 
 1296 void
 1297 ieee80211_node_join_bss(struct ieee80211com *ic, struct ieee80211_node *selbs)
 1298 {
 1299         enum ieee80211_phymode mode;
 1300         struct ieee80211_node *ni;
 1301         uint32_t assoc_fail = 0;
 1302 
 1303         /* Reinitialize media mode and channels if needed. */
 1304         mode = ieee80211_chan2mode(ic, selbs->ni_chan);
 1305         if (mode != ic->ic_curmode)
 1306                 ieee80211_setmode(ic, mode);
 1307 
 1308         /* Keep recorded association failures for this BSS/ESS intact. */
 1309         if (IEEE80211_ADDR_EQ(ic->ic_bss->ni_macaddr, selbs->ni_macaddr) ||
 1310             (ic->ic_des_esslen > 0 && ic->ic_des_esslen == selbs->ni_esslen &&
 1311             memcmp(ic->ic_des_essid, selbs->ni_essid, selbs->ni_esslen) == 0))
 1312                 assoc_fail = ic->ic_bss->ni_assoc_fail;
 1313 
 1314         (*ic->ic_node_copy)(ic, ic->ic_bss, selbs);
 1315         ni = ic->ic_bss;
 1316         ni->ni_assoc_fail |= assoc_fail;
 1317 
 1318         ic->ic_curmode = ieee80211_chan2mode(ic, ni->ni_chan);
 1319 
 1320         /* Make sure we send valid rates in an association request. */
 1321         if (ic->ic_opmode == IEEE80211_M_STA)
 1322                 ieee80211_fix_rate(ic, ni,
 1323                     IEEE80211_F_DOSORT | IEEE80211_F_DOFRATE |
 1324                     IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
 1325 
 1326         if (ic->ic_flags & IEEE80211_F_RSNON)
 1327                 ieee80211_choose_rsnparams(ic);
 1328         else if (ic->ic_flags & IEEE80211_F_WEPON)
 1329                 ni->ni_rsncipher = IEEE80211_CIPHER_USEGROUP;
 1330 
 1331         ieee80211_node_newstate(selbs, IEEE80211_STA_BSS);
 1332 #ifndef IEEE80211_STA_ONLY
 1333         if (ic->ic_opmode == IEEE80211_M_IBSS) {
 1334                 ieee80211_fix_rate(ic, ni, IEEE80211_F_DOFRATE |
 1335                     IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
 1336                 if (ni->ni_rates.rs_nrates == 0) {
 1337                         ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
 1338                         return;
 1339                 }
 1340                 ieee80211_new_state(ic, IEEE80211_S_RUN, -1);
 1341         } else
 1342 #endif
 1343         {
 1344                 int bgscan = ((ic->ic_flags & IEEE80211_F_BGSCAN) &&
 1345                     ic->ic_opmode == IEEE80211_M_STA &&
 1346                     ic->ic_state == IEEE80211_S_RUN);
 1347                 int auth_next = (ic->ic_opmode == IEEE80211_M_STA &&
 1348                     ic->ic_state == IEEE80211_S_AUTH);
 1349                 int mgt = -1;
 1350 
 1351                 timeout_del(&ic->ic_bgscan_timeout);
 1352                 ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1353 
 1354                 /* 
 1355                  * After a background scan, we have now switched APs.
 1356                  * Pretend we were just de-authed, which makes
 1357                  * ieee80211_new_state() try to re-auth and thus send
 1358                  * an AUTH frame to our newly selected AP.
 1359                  */
 1360                 if (bgscan)
 1361                         mgt = IEEE80211_FC0_SUBTYPE_DEAUTH;
 1362                 /*
 1363                  * If we are trying another AP after the previous one
 1364                  * failed (state transition AUTH->AUTH), ensure that
 1365                  * ieee80211_new_state() tries to send another auth frame.
 1366                  */
 1367                 else if (auth_next)
 1368                         mgt = IEEE80211_FC0_SUBTYPE_AUTH;
 1369 
 1370                 ieee80211_new_state(ic, IEEE80211_S_AUTH, mgt);
 1371         }
 1372 }
 1373 
 1374 struct ieee80211_node *
 1375 ieee80211_node_choose_bss(struct ieee80211com *ic, int bgscan,
 1376     struct ieee80211_node **curbs)
 1377 {
 1378         struct ieee80211_node *ni, *nextbs, *selbs = NULL, 
 1379             *selbs2 = NULL, *selbs5 = NULL;
 1380         uint8_t min_5ghz_rssi;
 1381 
 1382         ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
 1383 
 1384         for (; ni != NULL; ni = nextbs) {
 1385                 nextbs = RBT_NEXT(ieee80211_tree, ni);
 1386                 if (ni->ni_fails) {
 1387                         /*
 1388                          * The configuration of the access points may change
 1389                          * during my scan.  So delete the entry for the AP
 1390                          * and retry to associate if there is another beacon.
 1391                          */
 1392                         if (ni->ni_fails++ > 2)
 1393                                 ieee80211_free_node(ic, ni);
 1394                         continue;
 1395                 }
 1396 
 1397                 if (curbs && ieee80211_node_cmp(ic->ic_bss, ni) == 0)
 1398                         *curbs = ni;
 1399 
 1400                 if (ieee80211_match_bss(ic, ni, bgscan) != 0)
 1401                         continue;
 1402 
 1403                 if (ic->ic_caps & IEEE80211_C_SCANALLBAND) {
 1404                         if (IEEE80211_IS_CHAN_2GHZ(ni->ni_chan) &&
 1405                             (selbs2 == NULL || ni->ni_rssi > selbs2->ni_rssi))
 1406                                 selbs2 = ni;
 1407                         else if (IEEE80211_IS_CHAN_5GHZ(ni->ni_chan) &&
 1408                             (selbs5 == NULL || ni->ni_rssi > selbs5->ni_rssi))
 1409                                 selbs5 = ni;
 1410                 } else if (selbs == NULL || ni->ni_rssi > selbs->ni_rssi)
 1411                         selbs = ni;
 1412         }
 1413 
 1414         if (ic->ic_max_rssi)
 1415                 min_5ghz_rssi = IEEE80211_RSSI_THRES_RATIO_5GHZ;
 1416         else
 1417                 min_5ghz_rssi = (uint8_t)IEEE80211_RSSI_THRES_5GHZ;
 1418 
 1419         /*
 1420          * Prefer a 5Ghz AP even if its RSSI is weaker than the best 2Ghz AP
 1421          * (as long as it meets the minimum RSSI threshold) since the 5Ghz band
 1422          * is usually less saturated.
 1423          */
 1424         if (selbs5 && (*ic->ic_node_checkrssi)(ic, selbs5))
 1425                 selbs = selbs5;
 1426         else if (selbs5 && selbs2)
 1427                 selbs = (selbs5->ni_rssi >= selbs2->ni_rssi ? selbs5 : selbs2);
 1428         else if (selbs2)
 1429                 selbs = selbs2;
 1430         else if (selbs5)
 1431                 selbs = selbs5;
 1432 
 1433         return selbs;
 1434 }
 1435 
 1436 /*
 1437  * Complete a scan of potential channels.
 1438  */
 1439 void
 1440 ieee80211_end_scan(struct ifnet *ifp)
 1441 {
 1442         struct ieee80211com *ic = (void *)ifp;
 1443         struct ieee80211_node *ni, *selbs = NULL, *curbs = NULL;
 1444         int bgscan = ((ic->ic_flags & IEEE80211_F_BGSCAN) &&
 1445             ic->ic_opmode == IEEE80211_M_STA &&
 1446             ic->ic_state == IEEE80211_S_RUN);
 1447 
 1448         if (ifp->if_flags & IFF_DEBUG)
 1449                 printf("%s: end %s scan\n", ifp->if_xname,
 1450                     bgscan ? "background" :
 1451                     ((ic->ic_flags & IEEE80211_F_ASCAN) ?
 1452                     "active" : "passive"));
 1453 
 1454         if (ic->ic_scan_count)
 1455                 ic->ic_flags &= ~IEEE80211_F_ASCAN;
 1456 
 1457         if (ic->ic_opmode == IEEE80211_M_STA)
 1458                 ieee80211_clean_inactive_nodes(ic, IEEE80211_INACT_SCAN);
 1459 
 1460         ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
 1461 
 1462 #ifndef IEEE80211_STA_ONLY
 1463         if (ic->ic_opmode == IEEE80211_M_HOSTAP) {
 1464                 /* XXX off stack? */
 1465                 u_char occupied[howmany(IEEE80211_CHAN_MAX, NBBY)];
 1466                 int i, fail;
 1467 
 1468                 /*
 1469                  * The passive scan to look for existing AP's completed,
 1470                  * select a channel to camp on.  Identify the channels
 1471                  * that already have one or more AP's and try to locate
 1472                  * an unoccupied one.  If that fails, pick a random
 1473                  * channel from the active set.
 1474                  */
 1475                 memset(occupied, 0, sizeof(occupied));
 1476                 RBT_FOREACH(ni, ieee80211_tree, &ic->ic_tree)
 1477                         setbit(occupied, ieee80211_chan2ieee(ic, ni->ni_chan));
 1478                 for (i = 0; i < IEEE80211_CHAN_MAX; i++)
 1479                         if (isset(ic->ic_chan_active, i) && isclr(occupied, i))
 1480                                 break;
 1481                 if (i == IEEE80211_CHAN_MAX) {
 1482                         fail = arc4random() & 3;        /* random 0-3 */
 1483                         for (i = 0; i < IEEE80211_CHAN_MAX; i++)
 1484                                 if (isset(ic->ic_chan_active, i) && fail-- == 0)
 1485                                         break;
 1486                 }
 1487                 ieee80211_create_ibss(ic, &ic->ic_channels[i]);
 1488                 return;
 1489         }
 1490 #endif
 1491         if (ni == NULL) {
 1492                 DPRINTF(("no scan candidate\n"));
 1493  notfound:
 1494 
 1495 #ifndef IEEE80211_STA_ONLY
 1496                 if (ic->ic_opmode == IEEE80211_M_IBSS &&
 1497                     (ic->ic_flags & IEEE80211_F_IBSSON) &&
 1498                     ic->ic_des_esslen != 0) {
 1499                         ieee80211_create_ibss(ic, ic->ic_ibss_chan);
 1500                         return;
 1501                 }
 1502 #endif
 1503                 /*
 1504                  * Reset the list of channels to scan and scan the next mode
 1505                  * if nothing has been found.
 1506                  * If the device scans all bands in one fell swoop, return
 1507                  * current scan results to userspace regardless of mode.
 1508                  * This will loop forever until an access point is found.
 1509                  */
 1510                 ieee80211_reset_scan(ifp);
 1511                 if (ieee80211_next_mode(ifp) == IEEE80211_MODE_AUTO ||
 1512                     (ic->ic_caps & IEEE80211_C_SCANALLBAND))
 1513                         ic->ic_scan_count++;
 1514 
 1515                 ieee80211_next_scan(ifp);
 1516                 return;
 1517         }
 1518 
 1519         /* Possibly switch which ssid we are associated with */
 1520         if (!bgscan && ic->ic_opmode == IEEE80211_M_STA)
 1521                 ieee80211_switch_ess(ic);
 1522 
 1523         selbs = ieee80211_node_choose_bss(ic, bgscan, &curbs);
 1524         if (bgscan) {
 1525                 struct ieee80211_node_switch_bss_arg *arg;
 1526 
 1527                 /* AP disappeared? Should not happen. */
 1528                 if (selbs == NULL || curbs == NULL) {
 1529                         ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1530                         goto notfound;
 1531                 }
 1532 
 1533                 /* 
 1534                  * After a background scan we might end up choosing the
 1535                  * same AP again. Or the newly selected AP's RSSI level
 1536                  * might be low enough to trigger another background scan.
 1537                  * Do not change ic->ic_bss in these cases and make
 1538                  * background scans less frequent.
 1539                  */
 1540                 if (selbs == curbs || !(*ic->ic_node_checkrssi)(ic, selbs)) {
 1541                         if (ic->ic_bgscan_fail < IEEE80211_BGSCAN_FAIL_MAX) {
 1542                                 if (ic->ic_bgscan_fail <= 0)
 1543                                         ic->ic_bgscan_fail = 1;
 1544                                 else
 1545                                         ic->ic_bgscan_fail *= 2;
 1546                         }
 1547                         ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1548 
 1549                         /*
 1550                          * HT is negotiated during association so we must use
 1551                          * ic_bss to check HT. The nodes tree was re-populated
 1552                          * during background scan and therefore selbs and curbs
 1553                          * may not carry HT information.
 1554                          */
 1555                         ni = ic->ic_bss;
 1556                         if (ni->ni_flags & IEEE80211_NODE_VHT)
 1557                                 ieee80211_setmode(ic, IEEE80211_MODE_11AC);
 1558                         else if (ni->ni_flags & IEEE80211_NODE_HT)
 1559                                 ieee80211_setmode(ic, IEEE80211_MODE_11N);
 1560                         else
 1561                                 ieee80211_setmode(ic,
 1562                                     ieee80211_chan2mode(ic, ni->ni_chan));
 1563                         return;
 1564                 }
 1565         
 1566                 arg = malloc(sizeof(*arg), M_DEVBUF, M_NOWAIT | M_ZERO);
 1567                 if (arg == NULL) {
 1568                         ic->ic_flags &= ~IEEE80211_F_BGSCAN;
 1569                         return;
 1570                 }
 1571 
 1572                 ic->ic_bgscan_fail = 0;
 1573 
 1574                 /* Prevent dispatch of additional data frames to hardware. */
 1575                 ic->ic_xflags |= IEEE80211_F_TX_MGMT_ONLY;
 1576 
 1577                 IEEE80211_ADDR_COPY(arg->cur_macaddr, curbs->ni_macaddr);
 1578                 IEEE80211_ADDR_COPY(arg->sel_macaddr, selbs->ni_macaddr);
 1579 
 1580                 if (ic->ic_bgscan_done) {
 1581                         /*
 1582                          * The driver will flush its queues and allow roaming
 1583                          * to proceed once queues have been flushed.
 1584                          * On failure the driver will move back to SCAN state.
 1585                          */
 1586                         ic->ic_bgscan_done(ic, arg, sizeof(*arg));
 1587                         return;
 1588                 }
 1589 
 1590                 /* 
 1591                  * Install a callback which will switch us to the new AP once
 1592                  * all dispatched frames have been processed by hardware.
 1593                  */
 1594                 ic->ic_bss->ni_unref_arg = arg;
 1595                 ic->ic_bss->ni_unref_arg_size = sizeof(*arg);
 1596                 if (ic->ic_bss->ni_refcnt > 0)
 1597                         ic->ic_bss->ni_unref_cb = ieee80211_node_tx_flushed;
 1598                 else
 1599                         ieee80211_node_tx_flushed(ic, ni);
 1600                 /* F_BGSCAN flag gets cleared in ieee80211_node_join_bss(). */
 1601                 return;
 1602         } else if (selbs == NULL)
 1603                 goto notfound;
 1604 
 1605         ieee80211_node_join_bss(ic, selbs);
 1606 }
 1607 
 1608 /*
 1609  * Autoselect the best RSN parameters (protocol, AKMP, pairwise cipher...)
 1610  * that are supported by both peers (STA mode only).
 1611  */
 1612 void
 1613 ieee80211_choose_rsnparams(struct ieee80211com *ic)
 1614 {
 1615         struct ieee80211_node *ni = ic->ic_bss;
 1616         struct ieee80211_pmk *pmk;
 1617 
 1618         /* filter out unsupported protocol versions */
 1619         ni->ni_rsnprotos &= ic->ic_rsnprotos;
 1620         /* prefer RSN (aka WPA2) over WPA */
 1621         if (ni->ni_rsnprotos & IEEE80211_PROTO_RSN)
 1622                 ni->ni_rsnprotos = IEEE80211_PROTO_RSN;
 1623         else
 1624                 ni->ni_rsnprotos = IEEE80211_PROTO_WPA;
 1625 
 1626         /* filter out unsupported AKMPs */
 1627         ni->ni_rsnakms &= ic->ic_rsnakms;
 1628         /* prefer SHA-256 based AKMPs */
 1629         if ((ic->ic_flags & IEEE80211_F_PSK) && (ni->ni_rsnakms &
 1630             (IEEE80211_AKM_PSK | IEEE80211_AKM_SHA256_PSK))) {
 1631                 /* AP supports PSK AKMP and a PSK is configured */
 1632                 if (ni->ni_rsnakms & IEEE80211_AKM_SHA256_PSK)
 1633                         ni->ni_rsnakms = IEEE80211_AKM_SHA256_PSK;
 1634                 else
 1635                         ni->ni_rsnakms = IEEE80211_AKM_PSK;
 1636         } else {
 1637                 if (ni->ni_rsnakms & IEEE80211_AKM_SHA256_8021X)
 1638                         ni->ni_rsnakms = IEEE80211_AKM_SHA256_8021X;
 1639                 else
 1640                         ni->ni_rsnakms = IEEE80211_AKM_8021X;
 1641                 /* check if we have a cached PMK for this AP */
 1642                 if (ni->ni_rsnprotos == IEEE80211_PROTO_RSN &&
 1643                     (pmk = ieee80211_pmksa_find(ic, ni, NULL)) != NULL) {
 1644                         memcpy(ni->ni_pmkid, pmk->pmk_pmkid,
 1645                             IEEE80211_PMKID_LEN);
 1646                         ni->ni_flags |= IEEE80211_NODE_PMKID;
 1647                 }
 1648         }
 1649 
 1650         /* filter out unsupported pairwise ciphers */
 1651         ni->ni_rsnciphers &= ic->ic_rsnciphers;
 1652         /* prefer CCMP over TKIP */
 1653         if (ni->ni_rsnciphers & IEEE80211_CIPHER_CCMP)
 1654                 ni->ni_rsnciphers = IEEE80211_CIPHER_CCMP;
 1655         else
 1656                 ni->ni_rsnciphers = IEEE80211_CIPHER_TKIP;
 1657         ni->ni_rsncipher = ni->ni_rsnciphers;
 1658 
 1659         /* use MFP if we both support it */
 1660         if ((ic->ic_caps & IEEE80211_C_MFP) &&
 1661             (ni->ni_rsncaps & IEEE80211_RSNCAP_MFPC))
 1662                 ni->ni_flags |= IEEE80211_NODE_MFP;
 1663 }
 1664 
 1665 int
 1666 ieee80211_get_rate(struct ieee80211com *ic)
 1667 {
 1668         u_int8_t (*rates)[IEEE80211_RATE_MAXSIZE];
 1669         int rate;
 1670 
 1671         rates = &ic->ic_bss->ni_rates.rs_rates;
 1672 
 1673         if (ic->ic_fixed_rate != -1)
 1674                 rate = (*rates)[ic->ic_fixed_rate];
 1675         else if (ic->ic_state == IEEE80211_S_RUN)
 1676                 rate = (*rates)[ic->ic_bss->ni_txrate];
 1677         else
 1678                 rate = 0;
 1679 
 1680         return rate & IEEE80211_RATE_VAL;
 1681 }
 1682 
 1683 struct ieee80211_node *
 1684 ieee80211_node_alloc(struct ieee80211com *ic)
 1685 {
 1686         return malloc(sizeof(struct ieee80211_node), M_DEVBUF,
 1687             M_NOWAIT | M_ZERO);
 1688 }
 1689 
 1690 void
 1691 ieee80211_node_cleanup(struct ieee80211com *ic, struct ieee80211_node *ni)
 1692 {
 1693         if (ni->ni_rsnie != NULL) {
 1694                 free(ni->ni_rsnie, M_DEVBUF, 2 + ni->ni_rsnie[1]);
 1695                 ni->ni_rsnie = NULL;
 1696         }
 1697         ieee80211_ba_del(ni);
 1698 #ifndef IEEE80211_STA_ONLY
 1699         mq_purge(&ni->ni_savedq);
 1700 #endif
 1701         ieee80211_node_free_unref_cb(ni);
 1702 }
 1703 
 1704 void
 1705 ieee80211_node_free(struct ieee80211com *ic, struct ieee80211_node *ni)
 1706 {
 1707         ieee80211_node_cleanup(ic, ni);
 1708         free(ni, M_DEVBUF, 0);
 1709 }
 1710 
 1711 void
 1712 ieee80211_node_copy(struct ieee80211com *ic,
 1713         struct ieee80211_node *dst, const struct ieee80211_node *src)
 1714 {
 1715         ieee80211_node_cleanup(ic, dst);
 1716         *dst = *src;
 1717         dst->ni_rsnie = NULL;
 1718         if (src->ni_rsnie != NULL)
 1719                 ieee80211_save_ie(src->ni_rsnie, &dst->ni_rsnie);
 1720         ieee80211_node_set_timeouts(dst);
 1721 #ifndef IEEE80211_STA_ONLY
 1722         mq_init(&dst->ni_savedq, IEEE80211_PS_MAX_QUEUE, IPL_NET);
 1723 #endif
 1724 }
 1725 
 1726 u_int8_t
 1727 ieee80211_node_getrssi(struct ieee80211com *ic,
 1728     const struct ieee80211_node *ni)
 1729 {
 1730         return ni->ni_rssi;
 1731 }
 1732 
 1733 int
 1734 ieee80211_node_checkrssi(struct ieee80211com *ic,
 1735     const struct ieee80211_node *ni)
 1736 {
 1737         uint8_t thres;
 1738 
 1739         if (ni->ni_chan == IEEE80211_CHAN_ANYC)
 1740                 return 0;
 1741 
 1742         if (ic->ic_max_rssi) {
 1743                 thres = (IEEE80211_IS_CHAN_2GHZ(ni->ni_chan)) ?
 1744                     IEEE80211_RSSI_THRES_RATIO_2GHZ :
 1745                     IEEE80211_RSSI_THRES_RATIO_5GHZ;
 1746                 return ((ni->ni_rssi * 100) / ic->ic_max_rssi >= thres);
 1747         }
 1748 
 1749         thres = (IEEE80211_IS_CHAN_2GHZ(ni->ni_chan)) ?
 1750             IEEE80211_RSSI_THRES_2GHZ :
 1751             IEEE80211_RSSI_THRES_5GHZ;
 1752         return (ni->ni_rssi >= (u_int8_t)thres);
 1753 }
 1754 
 1755 void
 1756 ieee80211_node_set_timeouts(struct ieee80211_node *ni)
 1757 {
 1758         int i;
 1759 
 1760 #ifndef IEEE80211_STA_ONLY
 1761         timeout_set(&ni->ni_eapol_to, ieee80211_eapol_timeout, ni);
 1762         timeout_set(&ni->ni_sa_query_to, ieee80211_sa_query_timeout, ni);
 1763 #endif
 1764         timeout_set(&ni->ni_addba_req_to[EDCA_AC_BE],
 1765             ieee80211_node_addba_request_ac_be_to, ni);
 1766         timeout_set(&ni->ni_addba_req_to[EDCA_AC_BK],
 1767             ieee80211_node_addba_request_ac_bk_to, ni);
 1768         timeout_set(&ni->ni_addba_req_to[EDCA_AC_VI],
 1769             ieee80211_node_addba_request_ac_vi_to, ni);
 1770         timeout_set(&ni->ni_addba_req_to[EDCA_AC_VO],
 1771             ieee80211_node_addba_request_ac_vo_to, ni);
 1772         for (i = 0; i < nitems(ni->ni_addba_req_intval); i++)
 1773                 ni->ni_addba_req_intval[i] = 1;
 1774 }
 1775 
 1776 void
 1777 ieee80211_setup_node(struct ieee80211com *ic,
 1778         struct ieee80211_node *ni, const u_int8_t *macaddr)
 1779 {
 1780         int i, s;
 1781 
 1782         DPRINTF(("%s\n", ether_sprintf((u_int8_t *)macaddr)));
 1783         IEEE80211_ADDR_COPY(ni->ni_macaddr, macaddr);
 1784         ieee80211_node_newstate(ni, IEEE80211_STA_CACHE);
 1785 
 1786         ni->ni_ic = ic; /* back-pointer */
 1787         /* Initialize cached last sequence numbers with invalid values. */
 1788         ni->ni_rxseq = 0xffffU;
 1789         for (i=0; i < IEEE80211_NUM_TID; ++i)
 1790                 ni->ni_qos_rxseqs[i] = 0xffffU;
 1791 #ifndef IEEE80211_STA_ONLY
 1792         mq_init(&ni->ni_savedq, IEEE80211_PS_MAX_QUEUE, IPL_NET);
 1793 #endif
 1794         ieee80211_node_set_timeouts(ni);
 1795 
 1796         s = splnet();
 1797         RBT_INSERT(ieee80211_tree, &ic->ic_tree, ni);
 1798         ic->ic_nnodes++;
 1799         splx(s);
 1800 }
 1801 
 1802 struct ieee80211_node *
 1803 ieee80211_alloc_node(struct ieee80211com *ic, const u_int8_t *macaddr)
 1804 {
 1805         struct ieee80211_node *ni = ieee80211_alloc_node_helper(ic);
 1806         if (ni != NULL)
 1807                 ieee80211_setup_node(ic, ni, macaddr);
 1808         else
 1809                 ic->ic_stats.is_rx_nodealloc++;
 1810         return ni;
 1811 }
 1812 
 1813 struct ieee80211_node *
 1814 ieee80211_dup_bss(struct ieee80211com *ic, const u_int8_t *macaddr)
 1815 {
 1816         struct ieee80211_node *ni = ieee80211_alloc_node_helper(ic);
 1817         if (ni != NULL) {
 1818                 ieee80211_setup_node(ic, ni, macaddr);
 1819                 /*
 1820                  * Inherit from ic_bss.
 1821                  */
 1822                 IEEE80211_ADDR_COPY(ni->ni_bssid, ic->ic_bss->ni_bssid);
 1823                 ni->ni_chan = ic->ic_bss->ni_chan;
 1824         } else
 1825                 ic->ic_stats.is_rx_nodealloc++;
 1826         return ni;
 1827 }
 1828 
 1829 struct ieee80211_node *
 1830 ieee80211_find_node(struct ieee80211com *ic, const u_int8_t *macaddr)
 1831 {
 1832         struct ieee80211_node *ni;
 1833         int cmp;
 1834 
 1835         /* similar to RBT_FIND except we compare keys, not nodes */
 1836         ni = RBT_ROOT(ieee80211_tree, &ic->ic_tree);
 1837         while (ni != NULL) {
 1838                 cmp = memcmp(macaddr, ni->ni_macaddr, IEEE80211_ADDR_LEN);
 1839                 if (cmp < 0)
 1840                         ni = RBT_LEFT(ieee80211_tree, ni);
 1841                 else if (cmp > 0)
 1842                         ni = RBT_RIGHT(ieee80211_tree, ni);
 1843                 else
 1844                         break;
 1845         }
 1846         return ni;
 1847 }
 1848 
 1849 /*
 1850  * Return a reference to the appropriate node for sending
 1851  * a data frame.  This handles node discovery in adhoc networks.
 1852  *
 1853  * Drivers will call this, so increase the reference count before
 1854  * returning the node.
 1855  */
 1856 struct ieee80211_node *
 1857 ieee80211_find_txnode(struct ieee80211com *ic, const u_int8_t *macaddr)
 1858 {
 1859 #ifndef IEEE80211_STA_ONLY
 1860         struct ieee80211_node *ni;
 1861         int s;
 1862 #endif
 1863 
 1864         /*
 1865          * The destination address should be in the node table
 1866          * unless we are operating in station mode or this is a
 1867          * multicast/broadcast frame.
 1868          */
 1869         if (ic->ic_opmode == IEEE80211_M_STA || IEEE80211_IS_MULTICAST(macaddr))
 1870                 return ieee80211_ref_node(ic->ic_bss);
 1871 
 1872 #ifndef IEEE80211_STA_ONLY
 1873         s = splnet();
 1874         ni = ieee80211_find_node(ic, macaddr);
 1875         splx(s);
 1876         if (ni == NULL) {
 1877                 if (ic->ic_opmode != IEEE80211_M_IBSS &&
 1878                     ic->ic_opmode != IEEE80211_M_AHDEMO)
 1879                         return NULL;
 1880 
 1881                 /*
 1882                  * Fake up a node; this handles node discovery in
 1883                  * adhoc mode.  Note that for the driver's benefit
 1884                  * we we treat this like an association so the driver
 1885                  * has an opportunity to setup its private state.
 1886                  *
 1887                  * XXX need better way to handle this; issue probe
 1888                  *     request so we can deduce rate set, etc.
 1889                  */
 1890                 if ((ni = ieee80211_dup_bss(ic, macaddr)) == NULL)
 1891                         return NULL;
 1892                 /* XXX no rate negotiation; just dup */
 1893                 ni->ni_rates = ic->ic_bss->ni_rates;
 1894                 ni->ni_txrate = 0;
 1895                 if (ic->ic_newassoc)
 1896                         (*ic->ic_newassoc)(ic, ni, 1);
 1897         }
 1898         return ieee80211_ref_node(ni);
 1899 #else
 1900         return NULL;    /* can't get there */
 1901 #endif  /* IEEE80211_STA_ONLY */
 1902 }
 1903 
 1904 /*
 1905  * It is usually desirable to process a Rx packet using its sender's
 1906  * node-record instead of the BSS record.
 1907  *
 1908  * - AP mode: keep a node-record for every authenticated/associated
 1909  *   station *in the BSS*. For future use, we also track neighboring
 1910  *   APs, since they might belong to the same ESS.  APs in the same
 1911  *   ESS may bridge packets to each other, forming a Wireless
 1912  *   Distribution System (WDS).
 1913  *
 1914  * - IBSS mode: keep a node-record for every station *in the BSS*.
 1915  *   Also track neighboring stations by their beacons/probe responses.
 1916  *
 1917  * - monitor mode: keep a node-record for every sender, regardless
 1918  *   of BSS.
 1919  *
 1920  * - STA mode: the only available node-record is the BSS record,
 1921  *   ic->ic_bss.
 1922  *
 1923  * Of all the 802.11 Control packets, only the node-records for
 1924  * RTS packets node-record can be looked up.
 1925  *
 1926  * Return non-zero if the packet's node-record is kept, zero
 1927  * otherwise.
 1928  */
 1929 static __inline int
 1930 ieee80211_needs_rxnode(struct ieee80211com *ic,
 1931     const struct ieee80211_frame *wh, const u_int8_t **bssid)
 1932 {
 1933         int monitor, rc = 0;
 1934 
 1935         monitor = (ic->ic_opmode == IEEE80211_M_MONITOR);
 1936 
 1937         *bssid = NULL;
 1938 
 1939         switch (wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK) {
 1940         case IEEE80211_FC0_TYPE_CTL:
 1941                 if (!monitor)
 1942                         break;
 1943                 return (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) ==
 1944                     IEEE80211_FC0_SUBTYPE_RTS;
 1945         case IEEE80211_FC0_TYPE_MGT:
 1946                 *bssid = wh->i_addr3;
 1947                 switch (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) {
 1948                 case IEEE80211_FC0_SUBTYPE_BEACON:
 1949                 case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
 1950                         break;
 1951                 default:
 1952 #ifndef IEEE80211_STA_ONLY
 1953                         if (ic->ic_opmode == IEEE80211_M_STA)
 1954                                 break;
 1955                         rc = IEEE80211_ADDR_EQ(*bssid, ic->ic_bss->ni_bssid) ||
 1956                              IEEE80211_ADDR_EQ(*bssid, etherbroadcastaddr);
 1957 #endif
 1958                         break;
 1959                 }
 1960                 break;
 1961         case IEEE80211_FC0_TYPE_DATA:
 1962                 switch (wh->i_fc[1] & IEEE80211_FC1_DIR_MASK) {
 1963                 case IEEE80211_FC1_DIR_NODS:
 1964                         *bssid = wh->i_addr3;
 1965 #ifndef IEEE80211_STA_ONLY
 1966                         if (ic->ic_opmode == IEEE80211_M_IBSS ||
 1967                             ic->ic_opmode == IEEE80211_M_AHDEMO)
 1968                                 rc = IEEE80211_ADDR_EQ(*bssid,
 1969                                     ic->ic_bss->ni_bssid);
 1970 #endif
 1971                         break;
 1972                 case IEEE80211_FC1_DIR_TODS:
 1973                         *bssid = wh->i_addr1;
 1974 #ifndef IEEE80211_STA_ONLY
 1975                         if (ic->ic_opmode == IEEE80211_M_HOSTAP)
 1976                                 rc = IEEE80211_ADDR_EQ(*bssid,
 1977                                     ic->ic_bss->ni_bssid);
 1978 #endif
 1979                         break;
 1980                 case IEEE80211_FC1_DIR_FROMDS:
 1981                 case IEEE80211_FC1_DIR_DSTODS:
 1982                         *bssid = wh->i_addr2;
 1983 #ifndef IEEE80211_STA_ONLY
 1984                         rc = (ic->ic_opmode == IEEE80211_M_HOSTAP);
 1985 #endif
 1986                         break;
 1987                 }
 1988                 break;
 1989         }
 1990         return monitor || rc;
 1991 }
 1992 
 1993 /* 
 1994  * Drivers call this, so increase the reference count before returning
 1995  * the node.
 1996  */
 1997 struct ieee80211_node *
 1998 ieee80211_find_rxnode(struct ieee80211com *ic,
 1999     const struct ieee80211_frame *wh)
 2000 {
 2001         static const u_int8_t zero[] = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
 2002         struct ieee80211_node *ni;
 2003         const u_int8_t *bssid;
 2004         int s;
 2005 
 2006         if (!ieee80211_needs_rxnode(ic, wh, &bssid))
 2007                 return ieee80211_ref_node(ic->ic_bss);
 2008 
 2009         s = splnet();
 2010         ni = ieee80211_find_node(ic, wh->i_addr2);
 2011         splx(s);
 2012 
 2013         if (ni != NULL)
 2014                 return ieee80211_ref_node(ni);
 2015 #ifndef IEEE80211_STA_ONLY
 2016         if (ic->ic_opmode == IEEE80211_M_HOSTAP)
 2017                 return ieee80211_ref_node(ic->ic_bss);
 2018 #endif
 2019         /* XXX see remarks in ieee80211_find_txnode */
 2020         /* XXX no rate negotiation; just dup */
 2021         if ((ni = ieee80211_dup_bss(ic, wh->i_addr2)) == NULL)
 2022                 return ieee80211_ref_node(ic->ic_bss);
 2023 
 2024         IEEE80211_ADDR_COPY(ni->ni_bssid, (bssid != NULL) ? bssid : zero);
 2025 
 2026         ni->ni_rates = ic->ic_bss->ni_rates;
 2027         ni->ni_txrate = 0;
 2028         if (ic->ic_newassoc)
 2029                 (*ic->ic_newassoc)(ic, ni, 1);
 2030 
 2031         DPRINTF(("faked-up node %p for %s\n", ni,
 2032             ether_sprintf((u_int8_t *)wh->i_addr2)));
 2033 
 2034         return ieee80211_ref_node(ni);
 2035 }
 2036 
 2037 void
 2038 ieee80211_node_tx_ba_clear(struct ieee80211_node *ni, int tid)
 2039 {
 2040         struct ieee80211_tx_ba *ba = &ni->ni_tx_ba[tid];
 2041 
 2042         if (ba->ba_state != IEEE80211_BA_INIT) {
 2043                 if (timeout_pending(&ba->ba_to))
 2044                         timeout_del(&ba->ba_to);
 2045                 ba->ba_state = IEEE80211_BA_INIT;
 2046         }
 2047 }
 2048 
 2049 void
 2050 ieee80211_ba_del(struct ieee80211_node *ni)
 2051 {
 2052         int tid;
 2053 
 2054         for (tid = 0; tid < nitems(ni->ni_rx_ba); tid++) {
 2055                 struct ieee80211_rx_ba *ba = &ni->ni_rx_ba[tid];
 2056                 if (ba->ba_state != IEEE80211_BA_INIT) {
 2057                         if (timeout_pending(&ba->ba_to))
 2058                                 timeout_del(&ba->ba_to);
 2059                         if (timeout_pending(&ba->ba_gap_to))
 2060                                 timeout_del(&ba->ba_gap_to);
 2061                         ba->ba_state = IEEE80211_BA_INIT;
 2062                 }
 2063         }
 2064 
 2065         for (tid = 0; tid < nitems(ni->ni_tx_ba); tid++)
 2066                 ieee80211_node_tx_ba_clear(ni, tid);
 2067 
 2068         timeout_del(&ni->ni_addba_req_to[EDCA_AC_BE]);
 2069         timeout_del(&ni->ni_addba_req_to[EDCA_AC_BK]);
 2070         timeout_del(&ni->ni_addba_req_to[EDCA_AC_VI]);
 2071         timeout_del(&ni->ni_addba_req_to[EDCA_AC_VO]);
 2072 }
 2073 
 2074 void
 2075 ieee80211_free_node(struct ieee80211com *ic, struct ieee80211_node *ni)
 2076 {
 2077         if (ni == ic->ic_bss)
 2078                 panic("freeing bss node");
 2079 
 2080         splassert(IPL_NET);
 2081 
 2082         DPRINTF(("%s\n", ether_sprintf(ni->ni_macaddr)));
 2083 #ifndef IEEE80211_STA_ONLY
 2084         timeout_del(&ni->ni_eapol_to);
 2085         timeout_del(&ni->ni_sa_query_to);
 2086         IEEE80211_AID_CLR(ni->ni_associd, ic->ic_aid_bitmap);
 2087 #endif
 2088         ieee80211_ba_del(ni);
 2089         RBT_REMOVE(ieee80211_tree, &ic->ic_tree, ni);
 2090         ic->ic_nnodes--;
 2091 #ifndef IEEE80211_STA_ONLY
 2092         if (mq_purge(&ni->ni_savedq) > 0) {
 2093                 if (ic->ic_set_tim != NULL)
 2094                         (*ic->ic_set_tim)(ic, ni->ni_associd, 0);
 2095         }
 2096 #endif
 2097         (*ic->ic_node_free)(ic, ni);
 2098         /* TBD indicate to drivers that a new node can be allocated */
 2099 }
 2100 
 2101 void
 2102 ieee80211_release_node(struct ieee80211com *ic, struct ieee80211_node *ni)
 2103 {
 2104         int s;
 2105         void (*ni_unref_cb)(struct ieee80211com *, struct ieee80211_node *);
 2106 
 2107         DPRINTF(("%s refcnt %u\n", ether_sprintf(ni->ni_macaddr),
 2108             ni->ni_refcnt));
 2109         s = splnet();
 2110         if (ieee80211_node_decref(ni) == 0) {
 2111                 if (ni->ni_unref_cb) {
 2112                         /* The callback may set ni->ni_unref_cb again. */
 2113                         ni_unref_cb = ni->ni_unref_cb;
 2114                         ni->ni_unref_cb = NULL;
 2115                         /* Freed by callback if necessary: */
 2116                         (*ni_unref_cb)(ic, ni);
 2117                 }
 2118                 if (ni->ni_state == IEEE80211_STA_COLLECT)
 2119                         ieee80211_free_node(ic, ni);
 2120         }
 2121         splx(s);
 2122 }
 2123 
 2124 void
 2125 ieee80211_free_allnodes(struct ieee80211com *ic, int clear_ic_bss)
 2126 {
 2127         struct ieee80211_node *ni;
 2128         int s;
 2129 
 2130         DPRINTF(("freeing all nodes\n"));
 2131         s = splnet();
 2132         while ((ni = RBT_MIN(ieee80211_tree, &ic->ic_tree)) != NULL)
 2133                 ieee80211_free_node(ic, ni);
 2134         splx(s);
 2135 
 2136         if (clear_ic_bss && ic->ic_bss != NULL)
 2137                 ieee80211_node_cleanup(ic, ic->ic_bss);
 2138 }
 2139 
 2140 void
 2141 ieee80211_clean_cached(struct ieee80211com *ic)
 2142 {
 2143         struct ieee80211_node *ni, *next_ni;
 2144         int s;
 2145 
 2146         s = splnet();
 2147         for (ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
 2148             ni != NULL; ni = next_ni) {
 2149                 next_ni = RBT_NEXT(ieee80211_tree, ni);
 2150                 if (ni->ni_state == IEEE80211_STA_CACHE)
 2151                         ieee80211_free_node(ic, ni);
 2152         }
 2153         splx(s);
 2154 }
 2155 /*
 2156  * Timeout inactive nodes.
 2157  *
 2158  * If called because of a cache timeout, which happens only in hostap and ibss
 2159  * modes, clean all inactive cached or authenticated nodes but don't de-auth
 2160  * any associated nodes. Also update HT protection settings.
 2161  *
 2162  * Else, this function is called because a new node must be allocated but the
 2163  * node cache is full. In this case, return as soon as a free slot was made
 2164  * available. If acting as hostap, clean cached nodes regardless of their
 2165  * recent activity and also allow de-authing of authenticated nodes older
 2166  * than one cache wait interval, and de-authing of inactive associated nodes.
 2167  */
 2168 void
 2169 ieee80211_clean_nodes(struct ieee80211com *ic, int cache_timeout)
 2170 {
 2171         struct ieee80211_node *ni, *next_ni;
 2172         u_int gen = ic->ic_scangen++;           /* NB: ok 'cuz single-threaded*/
 2173         int s;
 2174 #ifndef IEEE80211_STA_ONLY
 2175         int nnodes = 0, nonht = 0, nonhtassoc = 0;
 2176         struct ifnet *ifp = &ic->ic_if;
 2177         enum ieee80211_htprot htprot = IEEE80211_HTPROT_NONE;
 2178         enum ieee80211_protmode protmode = IEEE80211_PROT_NONE;
 2179 #endif
 2180 
 2181         s = splnet();
 2182         for (ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
 2183             ni != NULL; ni = next_ni) {
 2184                 next_ni = RBT_NEXT(ieee80211_tree, ni);
 2185                 if (!cache_timeout && ic->ic_nnodes < ic->ic_max_nnodes)
 2186                         break;
 2187                 if (ni->ni_scangen == gen)      /* previously handled */
 2188                         continue;
 2189 #ifndef IEEE80211_STA_ONLY
 2190                 nnodes++;
 2191                 if ((ic->ic_flags & IEEE80211_F_HTON) && cache_timeout) {
 2192                         /*
 2193                          * Check if node supports 802.11n.
 2194                          * Only require HT capabilities IE for this check.
 2195                          * Nodes might never reveal their supported MCS to us
 2196                          * unless they go through a full association sequence.
 2197                          * ieee80211_node_supports_ht() could misclassify them.
 2198                          */
 2199                         if ((ni->ni_flags & IEEE80211_NODE_HTCAP) == 0) {
 2200                                 nonht++;
 2201                                 if (ni->ni_state == IEEE80211_STA_ASSOC)
 2202                                         nonhtassoc++;
 2203                         }
 2204                 }
 2205 #endif
 2206                 ni->ni_scangen = gen;
 2207                 if (ni->ni_refcnt > 0)
 2208                         continue;
 2209 #ifndef IEEE80211_STA_ONLY
 2210                 if ((ic->ic_opmode == IEEE80211_M_HOSTAP ||
 2211                     ic->ic_opmode == IEEE80211_M_IBSS) &&
 2212                     ic->ic_state == IEEE80211_S_RUN) {
 2213                         if (cache_timeout) {
 2214                                 if (ni->ni_state != IEEE80211_STA_COLLECT &&
 2215                                     (ni->ni_state == IEEE80211_STA_ASSOC ||
 2216                                     ni->ni_inact < IEEE80211_INACT_MAX))
 2217                                         continue;
 2218                         } else {
 2219                                 if (ic->ic_opmode == IEEE80211_M_HOSTAP &&
 2220                                     ((ni->ni_state == IEEE80211_STA_ASSOC &&
 2221                                     ni->ni_inact < IEEE80211_INACT_MAX) ||
 2222                                     (ni->ni_state == IEEE80211_STA_AUTH &&
 2223                                      ni->ni_inact == 0)))
 2224                                         continue;
 2225 
 2226                                 if (ic->ic_opmode == IEEE80211_M_IBSS &&
 2227                                     ni->ni_state != IEEE80211_STA_COLLECT &&
 2228                                     ni->ni_state != IEEE80211_STA_CACHE &&
 2229                                     ni->ni_inact < IEEE80211_INACT_MAX)
 2230                                         continue;
 2231                         }
 2232                 }
 2233                 if (ifp->if_flags & IFF_DEBUG)
 2234                         printf("%s: station %s purged from node cache\n",
 2235                             ifp->if_xname, ether_sprintf(ni->ni_macaddr));
 2236 #endif
 2237                 /*
 2238                  * If we're hostap and the node is authenticated, send
 2239                  * a deauthentication frame. The node will be freed when
 2240                  * the driver calls ieee80211_release_node().
 2241                  */
 2242 #ifndef IEEE80211_STA_ONLY
 2243                 nnodes--;
 2244                 if ((ic->ic_flags & IEEE80211_F_HTON) && cache_timeout) {
 2245                         if ((ni->ni_flags & IEEE80211_NODE_HTCAP) == 0) {
 2246                                 nonht--;
 2247                                 if (ni->ni_state == IEEE80211_STA_ASSOC)
 2248                                         nonhtassoc--;
 2249                         }
 2250                 }
 2251                 if (ic->ic_opmode == IEEE80211_M_HOSTAP &&
 2252                     ni->ni_state >= IEEE80211_STA_AUTH &&
 2253                     ni->ni_state != IEEE80211_STA_COLLECT) {
 2254                         IEEE80211_SEND_MGMT(ic, ni,
 2255                             IEEE80211_FC0_SUBTYPE_DEAUTH,
 2256                             IEEE80211_REASON_AUTH_EXPIRE);
 2257                         ieee80211_node_leave(ic, ni);
 2258                 } else
 2259 #endif
 2260                         ieee80211_free_node(ic, ni);
 2261                 ic->ic_stats.is_node_timeout++;
 2262         }
 2263 
 2264 #ifndef IEEE80211_STA_ONLY
 2265         if ((ic->ic_flags & IEEE80211_F_HTON) && cache_timeout) {
 2266                 uint16_t htop1 = ic->ic_bss->ni_htop1;
 2267 
 2268                 /* Update HT protection settings. */
 2269                 if (nonht) {
 2270                         protmode = IEEE80211_PROT_CTSONLY;
 2271                         if (nonhtassoc)
 2272                                 htprot = IEEE80211_HTPROT_NONHT_MIXED;
 2273                         else
 2274                                 htprot = IEEE80211_HTPROT_NONMEMBER;
 2275                 }
 2276                 if ((htop1 & IEEE80211_HTOP1_PROT_MASK) != htprot) {
 2277                         htop1 &= ~IEEE80211_HTOP1_PROT_MASK;
 2278                         htop1 |= htprot;
 2279                         ic->ic_bss->ni_htop1 = htop1;
 2280                         ic->ic_protmode = protmode;
 2281                         if (ic->ic_updateprot)
 2282                                 ic->ic_updateprot(ic);
 2283                 }
 2284         }
 2285 
 2286         /* 
 2287          * During a cache timeout we iterate over all nodes.
 2288          * Check for node leaks by comparing the actual number of cached
 2289          * nodes with the ic_nnodes count, which is maintained while adding
 2290          * and removing nodes from the cache.
 2291          */
 2292         if ((ifp->if_flags & IFF_DEBUG) && cache_timeout &&
 2293             nnodes != ic->ic_nnodes)
 2294                 printf("%s: number of cached nodes is %d, expected %d,"
 2295                     "possible nodes leak\n", ifp->if_xname, nnodes,
 2296                     ic->ic_nnodes);
 2297 #endif
 2298         splx(s);
 2299 }
 2300 
 2301 void
 2302 ieee80211_clean_inactive_nodes(struct ieee80211com *ic, int inact_max)
 2303 {
 2304         struct ieee80211_node *ni, *next_ni;
 2305         u_int gen = ic->ic_scangen++;   /* NB: ok 'cuz single-threaded*/
 2306         int s;
 2307 
 2308         s = splnet();
 2309         for (ni = RBT_MIN(ieee80211_tree, &ic->ic_tree);
 2310             ni != NULL; ni = next_ni) {
 2311                 next_ni = RBT_NEXT(ieee80211_tree, ni);
 2312                 if (ni->ni_scangen == gen)      /* previously handled */
 2313                         continue;
 2314                 ni->ni_scangen = gen;
 2315                 if (ni->ni_refcnt > 0 || ni->ni_inact < inact_max)
 2316                         continue;
 2317                 ieee80211_free_node(ic, ni);
 2318                 ic->ic_stats.is_node_timeout++;
 2319         }
 2320 
 2321         splx(s);
 2322 }
 2323 
 2324 void
 2325 ieee80211_iterate_nodes(struct ieee80211com *ic, ieee80211_iter_func *f,
 2326     void *arg)
 2327 {
 2328         struct ieee80211_node *ni;
 2329         int s;
 2330 
 2331         s = splnet();
 2332         RBT_FOREACH(ni, ieee80211_tree, &ic->ic_tree)
 2333                 (*f)(arg, ni);
 2334         splx(s);
 2335 }
 2336 
 2337 
 2338 /*
 2339  * Install received HT caps information in the node's state block.
 2340  */
 2341 void
 2342 ieee80211_setup_htcaps(struct ieee80211_node *ni, const uint8_t *data,
 2343     uint8_t len)
 2344 {
 2345         uint16_t rxrate;
 2346 
 2347         if (len != 26)
 2348                 return;
 2349 
 2350         ni->ni_htcaps = (data[0] | (data[1] << 8));
 2351         ni->ni_ampdu_param = data[2];
 2352 
 2353         memcpy(ni->ni_rxmcs, &data[3], sizeof(ni->ni_rxmcs));
 2354         /* clear reserved bits */
 2355         clrbit(ni->ni_rxmcs, 77);
 2356         clrbit(ni->ni_rxmcs, 78);
 2357         clrbit(ni->ni_rxmcs, 79);
 2358 
 2359         /* Max MCS Rx rate in 1Mb/s units (0 means "not specified"). */
 2360         rxrate = ((data[13] | (data[14]) << 8) & IEEE80211_MCS_RX_RATE_HIGH);
 2361         if (rxrate < 1024)
 2362                 ni->ni_max_rxrate = rxrate;
 2363 
 2364         ni->ni_tx_mcs_set = data[15];
 2365         ni->ni_htxcaps = (data[19] | (data[20] << 8));
 2366         ni->ni_txbfcaps = (data[21] | (data[22] << 8) | (data[23] << 16) |
 2367                 (data[24] << 24));
 2368         ni->ni_aselcaps = data[25];
 2369 
 2370         ni->ni_flags |= IEEE80211_NODE_HTCAP;
 2371 }
 2372 
 2373 #ifndef IEEE80211_STA_ONLY
 2374 /* 
 2375  * Handle nodes switching from 11n into legacy modes.
 2376  */
 2377 void
 2378 ieee80211_clear_htcaps(struct ieee80211_node *ni)
 2379 {
 2380         ni->ni_htcaps = 0;
 2381         ni->ni_ampdu_param = 0;
 2382         memset(ni->ni_rxmcs, 0, sizeof(ni->ni_rxmcs));
 2383         ni->ni_max_rxrate = 0;
 2384         ni->ni_tx_mcs_set = 0;
 2385         ni->ni_htxcaps = 0;
 2386         ni->ni_txbfcaps = 0;
 2387         ni->ni_aselcaps = 0;
 2388 
 2389         ni->ni_flags &= ~(IEEE80211_NODE_HT | IEEE80211_NODE_HT_SGI20 |
 2390             IEEE80211_NODE_HT_SGI40 | IEEE80211_NODE_HTCAP);
 2391 
 2392 }
 2393 #endif
 2394 
 2395 /*
 2396  * Install received HT op information in the node's state block.
 2397  */
 2398 int
 2399 ieee80211_setup_htop(struct ieee80211_node *ni, const uint8_t *data,
 2400     uint8_t len, int isprobe)
 2401 {
 2402         if (len != 22)
 2403                 return 0;
 2404 
 2405         ni->ni_primary_chan = data[0]; /* XXX corresponds to ni_chan */
 2406 
 2407         ni->ni_htop0 = data[1];
 2408         ni->ni_htop1 = (data[2] | (data[3] << 8));
 2409         ni->ni_htop2 = (data[3] | (data[4] << 8));
 2410 
 2411         /*
 2412          * According to 802.11-2012 Table 8-130 the Basic MCS set is
 2413          * only "present in Beacon, Probe Response, Mesh Peering Open
 2414          * and Mesh Peering Confirm frames. Otherwise reserved."
 2415          */
 2416         if (isprobe)
 2417                 memcpy(ni->ni_basic_mcs, &data[6], sizeof(ni->ni_basic_mcs));
 2418 
 2419         return 1;
 2420 }
 2421 
 2422 /*
 2423  * Install received VHT caps information in the node's state block.
 2424  */
 2425 void
 2426 ieee80211_setup_vhtcaps(struct ieee80211_node *ni, const uint8_t *data,
 2427     uint8_t len)
 2428 {
 2429         if (len != 12)
 2430                 return;
 2431 
 2432         ni->ni_vhtcaps = (data[0] | (data[1] << 8) | data[2] << 16 |
 2433             data[3] << 24);
 2434         ni->ni_vht_rxmcs = (data[4] | (data[5] << 8));
 2435         ni->ni_vht_rx_max_lgi_mbit_s = ((data[6] | (data[7] << 8)) &
 2436             IEEE80211_VHT_MAX_LGI_MBIT_S_MASK);
 2437         ni->ni_vht_txmcs = (data[8] | (data[9] << 8));
 2438         ni->ni_vht_tx_max_lgi_mbit_s = ((data[10] | (data[11] << 8)) &
 2439             IEEE80211_VHT_MAX_LGI_MBIT_S_MASK);
 2440 
 2441         ni->ni_flags |= IEEE80211_NODE_VHTCAP;
 2442 }
 2443 
 2444 /*
 2445  * Install received VHT op information in the node's state block.
 2446  */
 2447 int
 2448 ieee80211_setup_vhtop(struct ieee80211_node *ni, const uint8_t *data,
 2449     uint8_t len, int isprobe)
 2450 {
 2451 
 2452         if (len != 5)
 2453                 return 0;
 2454 
 2455         if (data[0] != IEEE80211_VHTOP0_CHAN_WIDTH_HT &&
 2456             data[0] != IEEE80211_VHTOP0_CHAN_WIDTH_80 &&
 2457             data[0] != IEEE80211_VHTOP0_CHAN_WIDTH_160 &&
 2458             data[0] != IEEE80211_VHTOP0_CHAN_WIDTH_8080)
 2459                 return 0;
 2460 
 2461         ni->ni_vht_chan_width = data[0];
 2462         ni->ni_vht_chan_center_freq_idx0 = data[1];
 2463         ni->ni_vht_chan_center_freq_idx1 = data[2];
 2464         ni->ni_vht_basic_mcs = (data[3] | data[4] << 8);
 2465         return 1;
 2466 }
 2467 
 2468 #ifndef IEEE80211_STA_ONLY
 2469 /* 
 2470  * Handle nodes switching from 11ac into legacy modes.
 2471  */
 2472 void
 2473 ieee80211_clear_vhtcaps(struct ieee80211_node *ni)
 2474 {
 2475         ni->ni_vhtcaps = 0;
 2476         ni->ni_vht_rxmcs = 0;
 2477         ni->ni_vht_rx_max_lgi_mbit_s = 0;
 2478         ni->ni_vht_txmcs = 0;
 2479         ni->ni_vht_tx_max_lgi_mbit_s = 0;
 2480 
 2481         ni->ni_flags &= ~(IEEE80211_NODE_VHT | IEEE80211_NODE_VHT_SGI80 |
 2482             IEEE80211_NODE_VHT_SGI160 | IEEE80211_NODE_VHTCAP);
 2483 
 2484 }
 2485 #endif
 2486 
 2487 /*
 2488  * Install received rate set information in the node's state block.
 2489  */
 2490 int
 2491 ieee80211_setup_rates(struct ieee80211com *ic, struct ieee80211_node *ni,
 2492     const u_int8_t *rates, const u_int8_t *xrates, int flags)
 2493 {
 2494         struct ieee80211_rateset *rs = &ni->ni_rates;
 2495 
 2496         memset(rs, 0, sizeof(*rs));
 2497         rs->rs_nrates = rates[1];
 2498         memcpy(rs->rs_rates, rates + 2, rs->rs_nrates);
 2499         if (xrates != NULL) {
 2500                 u_int8_t nxrates;
 2501                 /*
 2502                  * Tack on 11g extended supported rate element.
 2503                  */
 2504                 nxrates = xrates[1];
 2505                 if (rs->rs_nrates + nxrates > IEEE80211_RATE_MAXSIZE) {
 2506                         nxrates = IEEE80211_RATE_MAXSIZE - rs->rs_nrates;
 2507                         DPRINTF(("extended rate set too large; "
 2508                             "only using %u of %u rates\n",
 2509                             nxrates, xrates[1]));
 2510                         ic->ic_stats.is_rx_rstoobig++;
 2511                 }
 2512                 memcpy(rs->rs_rates + rs->rs_nrates, xrates+2, nxrates);
 2513                 rs->rs_nrates += nxrates;
 2514         }
 2515         return ieee80211_fix_rate(ic, ni, flags);
 2516 }
 2517 
 2518 void
 2519 ieee80211_node_trigger_addba_req(struct ieee80211_node *ni, int tid)
 2520 {
 2521         if (ni->ni_tx_ba[tid].ba_state == IEEE80211_BA_INIT &&
 2522             !timeout_pending(&ni->ni_addba_req_to[tid])) {
 2523                 timeout_add_sec(&ni->ni_addba_req_to[tid],
 2524                     ni->ni_addba_req_intval[tid]);
 2525         }
 2526 }
 2527 
 2528 void
 2529 ieee80211_node_addba_request(struct ieee80211_node *ni, int tid)
 2530 {
 2531         struct ieee80211com *ic = ni->ni_ic;
 2532         uint16_t ssn = ni->ni_qos_txseqs[tid];
 2533 
 2534         ieee80211_addba_request(ic, ni, ssn, tid);
 2535 }
 2536 
 2537 void
 2538 ieee80211_node_addba_request_ac_be_to(void *arg)
 2539 {
 2540         struct ieee80211_node *ni = arg;
 2541         ieee80211_node_addba_request(ni, EDCA_AC_BE);
 2542 }
 2543 
 2544 void
 2545 ieee80211_node_addba_request_ac_bk_to(void *arg)
 2546 {
 2547         struct ieee80211_node *ni = arg;
 2548         ieee80211_node_addba_request(ni, EDCA_AC_BK);
 2549 }
 2550 
 2551 void
 2552 ieee80211_node_addba_request_ac_vi_to(void *arg)
 2553 {
 2554         struct ieee80211_node *ni = arg;
 2555         ieee80211_node_addba_request(ni, EDCA_AC_VI);
 2556 }
 2557 
 2558 void
 2559 ieee80211_node_addba_request_ac_vo_to(void *arg)
 2560 {
 2561         struct ieee80211_node *ni = arg;
 2562         ieee80211_node_addba_request(ni, EDCA_AC_VO);
 2563 }
 2564 
 2565 #ifndef IEEE80211_STA_ONLY
 2566 /*
 2567  * Check if the specified node supports ERP.
 2568  */
 2569 int
 2570 ieee80211_iserp_sta(const struct ieee80211_node *ni)
 2571 {
 2572         static const u_int8_t rates[] = { 2, 4, 11, 22, 12, 24, 48 };
 2573         const struct ieee80211_rateset *rs = &ni->ni_rates;
 2574         int i, j;
 2575 
 2576         /*
 2577          * A STA supports ERP operation if it includes all the Clause 19
 2578          * mandatory rates in its supported rate set.
 2579          */
 2580         for (i = 0; i < nitems(rates); i++) {
 2581                 for (j = 0; j < rs->rs_nrates; j++) {
 2582                         if ((rs->rs_rates[j] & IEEE80211_RATE_VAL) == rates[i])
 2583                                 break;
 2584                 }
 2585                 if (j == rs->rs_nrates)
 2586                         return 0;
 2587         }
 2588         return 1;
 2589 }
 2590 
 2591 /*
 2592  * This function is called to notify the 802.1X PACP machine that a new
 2593  * 802.1X port is enabled and must be authenticated. For 802.11, a port
 2594  * becomes enabled whenever a STA successfully completes Open System
 2595  * authentication with an AP.
 2596  */
 2597 void
 2598 ieee80211_needs_auth(struct ieee80211com *ic, struct ieee80211_node *ni)
 2599 {
 2600         /*
 2601          * XXX this could be done via the route socket of via a dedicated
 2602          * EAP socket or another kernel->userland notification mechanism.
 2603          * The notification should include the MAC address (ni_macaddr).
 2604          */
 2605 }
 2606 
 2607 /*
 2608  * Handle an HT STA joining an HT network.
 2609  */
 2610 void
 2611 ieee80211_node_join_ht(struct ieee80211com *ic, struct ieee80211_node *ni)
 2612 {
 2613         enum ieee80211_htprot;
 2614 
 2615         /* Update HT protection setting. */
 2616         if ((ni->ni_flags & IEEE80211_NODE_HT) == 0) {
 2617                 uint16_t htop1 = ic->ic_bss->ni_htop1;
 2618                 htop1 &= ~IEEE80211_HTOP1_PROT_MASK;
 2619                 htop1 |= IEEE80211_HTPROT_NONHT_MIXED;
 2620                 ic->ic_bss->ni_htop1 = htop1;
 2621                 if (ic->ic_updateprot)
 2622                         ic->ic_updateprot(ic);
 2623         }
 2624 }
 2625 
 2626 /*
 2627  * Handle a station joining an RSN network.
 2628  */
 2629 void
 2630 ieee80211_node_join_rsn(struct ieee80211com *ic, struct ieee80211_node *ni)
 2631 {
 2632         DPRINTF(("station %s associated using proto %d akm 0x%x "
 2633             "cipher 0x%x groupcipher 0x%x\n", ether_sprintf(ni->ni_macaddr),
 2634             ni->ni_rsnprotos, ni->ni_rsnakms, ni->ni_rsnciphers,
 2635             ni->ni_rsngroupcipher));
 2636 
 2637         ni->ni_rsn_state = RSNA_AUTHENTICATION;
 2638 
 2639         ni->ni_key_count = 0;
 2640         ni->ni_port_valid = 0;
 2641         ni->ni_flags &= ~IEEE80211_NODE_TXRXPROT;
 2642         ni->ni_flags &= ~IEEE80211_NODE_RSN_NEW_PTK;
 2643         ni->ni_replaycnt = -1;  /* XXX */
 2644         ni->ni_rsn_retries = 0;
 2645         ni->ni_rsncipher = ni->ni_rsnciphers;
 2646 
 2647         ni->ni_rsn_state = RSNA_AUTHENTICATION_2;
 2648 
 2649         /* generate a new authenticator nonce (ANonce) */
 2650         arc4random_buf(ni->ni_nonce, EAPOL_KEY_NONCE_LEN);
 2651 
 2652         if (!ieee80211_is_8021x_akm(ni->ni_rsnakms)) {
 2653                 memcpy(ni->ni_pmk, ic->ic_psk, IEEE80211_PMK_LEN);
 2654                 ni->ni_flags |= IEEE80211_NODE_PMK;
 2655                 (void)ieee80211_send_4way_msg1(ic, ni);
 2656         } else if (ni->ni_flags & IEEE80211_NODE_PMK) {
 2657                 /* skip 802.1X auth if a cached PMK was found */
 2658                 (void)ieee80211_send_4way_msg1(ic, ni);
 2659         } else {
 2660                 /* no cached PMK found, needs full 802.1X auth */
 2661                 ieee80211_needs_auth(ic, ni);
 2662         }
 2663 }
 2664 
 2665 void
 2666 ieee80211_count_longslotsta(void *arg, struct ieee80211_node *ni)
 2667 {
 2668         int *longslotsta = arg;
 2669 
 2670         if (ni->ni_associd == 0 || ni->ni_state == IEEE80211_STA_COLLECT)
 2671                 return;
 2672 
 2673         if (!(ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME))
 2674                 (*longslotsta)++;
 2675 }
 2676 
 2677 void
 2678 ieee80211_count_nonerpsta(void *arg, struct ieee80211_node *ni)
 2679 {
 2680         int *nonerpsta = arg;
 2681 
 2682         if (ni->ni_associd == 0 || ni->ni_state == IEEE80211_STA_COLLECT)
 2683                 return;
 2684 
 2685         if (!ieee80211_iserp_sta(ni))
 2686                 (*nonerpsta)++;
 2687 }
 2688 
 2689 void
 2690 ieee80211_count_pssta(void *arg, struct ieee80211_node *ni)
 2691 {
 2692         int *pssta = arg;
 2693 
 2694         if (ni->ni_associd == 0 || ni->ni_state == IEEE80211_STA_COLLECT)
 2695                 return;
 2696 
 2697         if (ni->ni_pwrsave == IEEE80211_PS_DOZE)
 2698                 (*pssta)++;
 2699 }
 2700 
 2701 void
 2702 ieee80211_count_rekeysta(void *arg, struct ieee80211_node *ni)
 2703 {
 2704         int *rekeysta = arg;
 2705 
 2706         if (ni->ni_associd == 0 || ni->ni_state == IEEE80211_STA_COLLECT)
 2707                 return;
 2708 
 2709         if (ni->ni_flags & IEEE80211_NODE_REKEY)
 2710                 (*rekeysta)++;
 2711 }
 2712 
 2713 /*
 2714  * Handle a station joining an 11g network.
 2715  */
 2716 void
 2717 ieee80211_node_join_11g(struct ieee80211com *ic, struct ieee80211_node *ni)
 2718 {
 2719         int longslotsta = 0, nonerpsta = 0;
 2720 
 2721         if (!(ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME)) {
 2722                 /*
 2723                  * Joining STA doesn't support short slot time.  We must
 2724                  * disable the use of short slot time for all other associated
 2725                  * STAs and give the driver a chance to reconfigure the
 2726                  * hardware.
 2727                  */
 2728                 ieee80211_iterate_nodes(ic,
 2729                     ieee80211_count_longslotsta, &longslotsta);
 2730                 if (longslotsta == 1) {
 2731                         if (ic->ic_caps & IEEE80211_C_SHSLOT)
 2732                                 ieee80211_set_shortslottime(ic, 0);
 2733                 }
 2734                 DPRINTF(("[%s] station needs long slot time, count %d\n",
 2735                     ether_sprintf(ni->ni_macaddr), longslotsta));
 2736         }
 2737 
 2738         if (!ieee80211_iserp_sta(ni)) {
 2739                 /*
 2740                  * Joining STA is non-ERP.
 2741                  */
 2742                 ieee80211_iterate_nodes(ic,
 2743                     ieee80211_count_nonerpsta, &nonerpsta);
 2744                 DPRINTF(("[%s] station is non-ERP, %d non-ERP "
 2745                     "stations associated\n", ether_sprintf(ni->ni_macaddr),
 2746                     nonerpsta));
 2747                 /* must enable the use of protection */
 2748                 if (ic->ic_protmode != IEEE80211_PROT_NONE) {
 2749                         DPRINTF(("enable use of protection\n"));
 2750                         ic->ic_flags |= IEEE80211_F_USEPROT;
 2751                 }
 2752 
 2753                 if (!(ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE))
 2754                         ic->ic_flags &= ~IEEE80211_F_SHPREAMBLE;
 2755         } else
 2756                 ni->ni_flags |= IEEE80211_NODE_ERP;
 2757 }
 2758 
 2759 void
 2760 ieee80211_node_join(struct ieee80211com *ic, struct ieee80211_node *ni,
 2761     int resp)
 2762 {
 2763         int newassoc = (ni->ni_state != IEEE80211_STA_ASSOC);
 2764 
 2765         if (ni->ni_associd == 0) {
 2766                 u_int16_t aid;
 2767 
 2768                 /*
 2769                  * It would be clever to search the bitmap
 2770                  * more efficiently, but this will do for now.
 2771                  */
 2772                 for (aid = 1; aid < ic->ic_max_aid; aid++) {
 2773                         if (!IEEE80211_AID_ISSET(aid,
 2774                             ic->ic_aid_bitmap))
 2775                                 break;
 2776                 }
 2777                 if (aid >= ic->ic_max_aid) {
 2778                         IEEE80211_SEND_MGMT(ic, ni, resp,
 2779                             IEEE80211_REASON_ASSOC_TOOMANY);
 2780                         ieee80211_node_leave(ic, ni);
 2781                         return;
 2782                 }
 2783                 ni->ni_associd = aid | 0xc000;
 2784                 IEEE80211_AID_SET(ni->ni_associd, ic->ic_aid_bitmap);
 2785                 if (ic->ic_curmode == IEEE80211_MODE_11G ||
 2786                     (ic->ic_curmode == IEEE80211_MODE_11N &&
 2787                     IEEE80211_IS_CHAN_2GHZ(ic->ic_bss->ni_chan)))
 2788                         ieee80211_node_join_11g(ic, ni);
 2789         }
 2790 
 2791         DPRINTF(("station %s %s associated at aid %d\n",
 2792             ether_sprintf(ni->ni_macaddr), newassoc ? "newly" : "already",
 2793             ni->ni_associd & ~0xc000));
 2794 
 2795         ieee80211_ht_negotiate(ic, ni);
 2796         if (ic->ic_flags & IEEE80211_F_HTON)
 2797                 ieee80211_node_join_ht(ic, ni);
 2798 
 2799         /* give driver a chance to setup state like ni_txrate */
 2800         if (ic->ic_newassoc)
 2801                 (*ic->ic_newassoc)(ic, ni, newassoc);
 2802         IEEE80211_SEND_MGMT(ic, ni, resp, IEEE80211_STATUS_SUCCESS);
 2803         ieee80211_node_newstate(ni, IEEE80211_STA_ASSOC);
 2804 
 2805         if (!(ic->ic_flags & IEEE80211_F_RSNON)) {
 2806                 ni->ni_port_valid = 1;
 2807                 ni->ni_rsncipher = IEEE80211_CIPHER_USEGROUP;
 2808         } else
 2809                 ieee80211_node_join_rsn(ic, ni);
 2810 
 2811 #if NBRIDGE > 0
 2812         /*
 2813          * If the parent interface is a bridge port, learn
 2814          * the node's address dynamically on this interface.
 2815          */
 2816         if (ic->ic_if.if_bridgeidx != 0)
 2817                 bridge_update(&ic->ic_if,
 2818                     (struct ether_addr *)ni->ni_macaddr, 0);
 2819 #endif
 2820 }
 2821 
 2822 /*
 2823  * Handle an HT STA leaving an HT network.
 2824  */
 2825 void
 2826 ieee80211_node_leave_ht(struct ieee80211com *ic, struct ieee80211_node *ni)
 2827 {
 2828         struct ieee80211_rx_ba *ba;
 2829         u_int8_t tid;
 2830         int i;
 2831 
 2832         /* free all Block Ack records */
 2833         ieee80211_ba_del(ni);
 2834         for (tid = 0; tid < IEEE80211_NUM_TID; tid++) {
 2835                 ba = &ni->ni_rx_ba[tid];
 2836                 if (ba->ba_buf != NULL) {
 2837                         for (i = 0; i < IEEE80211_BA_MAX_WINSZ; i++)
 2838                                 m_freem(ba->ba_buf[i].m);
 2839                         free(ba->ba_buf, M_DEVBUF,
 2840                             IEEE80211_BA_MAX_WINSZ * sizeof(*ba->ba_buf));
 2841                         ba->ba_buf = NULL;
 2842                 }
 2843         }
 2844 
 2845         ieee80211_clear_htcaps(ni);
 2846 }
 2847 
 2848 /*
 2849  * Handle a VHT STA leaving a VHT network.
 2850  */
 2851 void
 2852 ieee80211_node_leave_vht(struct ieee80211com *ic, struct ieee80211_node *ni)
 2853 {
 2854         ieee80211_clear_vhtcaps(ni);
 2855 }
 2856 
 2857 /*
 2858  * Handle a station leaving an RSN network.
 2859  */
 2860 void
 2861 ieee80211_node_leave_rsn(struct ieee80211com *ic, struct ieee80211_node *ni)
 2862 {
 2863         int rekeysta = 0;
 2864 
 2865         ni->ni_rsn_state = RSNA_INITIALIZE;
 2866         if (ni->ni_flags & IEEE80211_NODE_REKEY) {
 2867                 ni->ni_flags &= ~IEEE80211_NODE_REKEY;
 2868                 ieee80211_iterate_nodes(ic,
 2869                     ieee80211_count_rekeysta, &rekeysta);
 2870                 if (rekeysta == 0)
 2871                         ieee80211_setkeysdone(ic);
 2872         }
 2873         ni->ni_flags &= ~IEEE80211_NODE_PMK;
 2874         ni->ni_rsn_gstate = RSNA_IDLE;
 2875 
 2876         timeout_del(&ni->ni_eapol_to);
 2877         timeout_del(&ni->ni_sa_query_to);
 2878 
 2879         ni->ni_rsn_retries = 0;
 2880         ni->ni_flags &= ~IEEE80211_NODE_TXRXPROT;
 2881         ni->ni_port_valid = 0;
 2882         (*ic->ic_delete_key)(ic, ni, &ni->ni_pairwise_key);
 2883 }
 2884 
 2885 /*
 2886  * Handle a station leaving an 11g network.
 2887  */
 2888 void
 2889 ieee80211_node_leave_11g(struct ieee80211com *ic, struct ieee80211_node *ni)
 2890 {
 2891         int longslotsta = 0, nonerpsta = 0;
 2892 
 2893         if (!(ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME)) {
 2894                 /* leaving STA did not support short slot time */
 2895                 ieee80211_iterate_nodes(ic,
 2896                     ieee80211_count_longslotsta, &longslotsta);
 2897                 if (longslotsta == 1) {
 2898                         /*
 2899                          * All associated STAs now support short slot time, so
 2900                          * enable this feature and give the driver a chance to
 2901                          * reconfigure the hardware. Notice that IBSS always
 2902                          * use a long slot time.
 2903                          */
 2904                         if ((ic->ic_caps & IEEE80211_C_SHSLOT) &&
 2905                             ic->ic_opmode != IEEE80211_M_IBSS)
 2906                                 ieee80211_set_shortslottime(ic, 1);
 2907                 }
 2908                 DPRINTF(("[%s] long slot time station leaves, count %d\n",
 2909                     ether_sprintf(ni->ni_macaddr), longslotsta));
 2910         }
 2911 
 2912         if (!(ni->ni_flags & IEEE80211_NODE_ERP)) {
 2913                 /* leaving STA was non-ERP */
 2914                 ieee80211_iterate_nodes(ic,
 2915                     ieee80211_count_nonerpsta, &nonerpsta);
 2916                 if (nonerpsta == 1) {
 2917                         /*
 2918                          * All associated STAs are now ERP capable, disable use
 2919                          * of protection and re-enable short preamble support.
 2920                          */
 2921                         ic->ic_flags &= ~IEEE80211_F_USEPROT;
 2922                         if (ic->ic_caps & IEEE80211_C_SHPREAMBLE)
 2923                                 ic->ic_flags |= IEEE80211_F_SHPREAMBLE;
 2924                 }
 2925                 DPRINTF(("[%s] non-ERP station leaves, count %d\n",
 2926                     ether_sprintf(ni->ni_macaddr), nonerpsta));
 2927         }
 2928 }
 2929 
 2930 void
 2931 ieee80211_node_leave_pwrsave(struct ieee80211com *ic,
 2932     struct ieee80211_node *ni)
 2933 {
 2934         struct mbuf_queue keep = MBUF_QUEUE_INITIALIZER(IFQ_MAXLEN, IPL_NET);
 2935         struct mbuf *m;
 2936 
 2937         if (ni->ni_pwrsave == IEEE80211_PS_DOZE)
 2938                 ni->ni_pwrsave = IEEE80211_PS_AWAKE;
 2939 
 2940         if (mq_len(&ni->ni_savedq) > 0) {
 2941                 if (ic->ic_set_tim != NULL)
 2942                         (*ic->ic_set_tim)(ic, ni->ni_associd, 0);
 2943         }
 2944         while ((m = mq_dequeue(&ni->ni_savedq)) != NULL) {
 2945                 if (ni->ni_refcnt > 0)
 2946                         ieee80211_node_decref(ni);
 2947                 m_freem(m);
 2948         }
 2949 
 2950         /* Purge frames queued for transmission during DTIM. */
 2951         while ((m = mq_dequeue(&ic->ic_pwrsaveq)) != NULL) {
 2952                 if (m->m_pkthdr.ph_cookie == ni) {
 2953                         if (ni->ni_refcnt > 0)
 2954                                 ieee80211_node_decref(ni);
 2955                         m_freem(m);
 2956                 } else
 2957                         mq_enqueue(&keep, m);
 2958         }
 2959         while ((m = mq_dequeue(&keep)) != NULL)
 2960                 mq_enqueue(&ic->ic_pwrsaveq, m);
 2961 }
 2962 
 2963 /*
 2964  * Handle bookkeeping for station deauthentication/disassociation
 2965  * when operating as an ap.
 2966  */
 2967 void
 2968 ieee80211_node_leave(struct ieee80211com *ic, struct ieee80211_node *ni)
 2969 {
 2970         if (ic->ic_opmode != IEEE80211_M_HOSTAP)
 2971                 panic("not in ap mode, mode %u", ic->ic_opmode);
 2972 
 2973         if (ni->ni_state == IEEE80211_STA_COLLECT)
 2974                 return;
 2975         /*
 2976          * If node wasn't previously associated all we need to do is
 2977          * reclaim the reference.
 2978          */
 2979         if (ni->ni_associd == 0) {
 2980                 ieee80211_node_newstate(ni, IEEE80211_STA_COLLECT);
 2981                 return;
 2982         }
 2983 
 2984         ieee80211_node_leave_pwrsave(ic, ni);
 2985 
 2986         if (ic->ic_flags & IEEE80211_F_RSNON)
 2987                 ieee80211_node_leave_rsn(ic, ni);
 2988 
 2989         if (ic->ic_curmode == IEEE80211_MODE_11G ||
 2990             (ic->ic_curmode == IEEE80211_MODE_11N &&
 2991             IEEE80211_IS_CHAN_2GHZ(ic->ic_bss->ni_chan)))
 2992                 ieee80211_node_leave_11g(ic, ni);
 2993 
 2994         if (ni->ni_flags & IEEE80211_NODE_HT)
 2995                 ieee80211_node_leave_ht(ic, ni);
 2996         if (ni->ni_flags & IEEE80211_NODE_VHT)
 2997                 ieee80211_node_leave_vht(ic, ni);
 2998 
 2999         if (ic->ic_node_leave != NULL)
 3000                 (*ic->ic_node_leave)(ic, ni);
 3001 
 3002         ieee80211_node_newstate(ni, IEEE80211_STA_COLLECT);
 3003 
 3004 #if NBRIDGE > 0
 3005         /*
 3006          * If the parent interface is a bridge port, delete
 3007          * any dynamically learned address for this node.
 3008          */
 3009         if (ic->ic_if.if_bridgeidx != 0)
 3010                 bridge_update(&ic->ic_if,
 3011                     (struct ether_addr *)ni->ni_macaddr, 1);
 3012 #endif
 3013 }
 3014 
 3015 static int
 3016 ieee80211_do_slow_print(struct ieee80211com *ic, int *did_print)
 3017 {
 3018         static const struct timeval merge_print_intvl = {
 3019                 .tv_sec = 1, .tv_usec = 0
 3020         };
 3021         if ((ic->ic_if.if_flags & IFF_LINK0) == 0)
 3022                 return 0;
 3023         if (!*did_print && (ic->ic_if.if_flags & IFF_DEBUG) == 0 &&
 3024             !ratecheck(&ic->ic_last_merge_print, &merge_print_intvl))
 3025                 return 0;
 3026 
 3027         *did_print = 1;
 3028         return 1;
 3029 }
 3030 
 3031 /* ieee80211_ibss_merge helps merge 802.11 ad hoc networks.  The
 3032  * convention, set by the Wireless Ethernet Compatibility Alliance
 3033  * (WECA), is that an 802.11 station will change its BSSID to match
 3034  * the "oldest" 802.11 ad hoc network, on the same channel, that
 3035  * has the station's desired SSID.  The "oldest" 802.11 network
 3036  * sends beacons with the greatest TSF timestamp.
 3037  *
 3038  * Return ENETRESET if the BSSID changed, 0 otherwise.
 3039  *
 3040  * XXX Perhaps we should compensate for the time that elapses
 3041  * between the MAC receiving the beacon and the host processing it
 3042  * in ieee80211_ibss_merge.
 3043  */
 3044 int
 3045 ieee80211_ibss_merge(struct ieee80211com *ic, struct ieee80211_node *ni,
 3046     u_int64_t local_tsft)
 3047 {
 3048         u_int64_t beacon_tsft;
 3049         int did_print = 0, sign;
 3050         union {
 3051                 u_int64_t       word;
 3052                 u_int8_t        tstamp[8];
 3053         } u;
 3054 
 3055         /* ensure alignment */
 3056         (void)memcpy(&u, &ni->ni_tstamp[0], sizeof(u));
 3057         beacon_tsft = letoh64(u.word);
 3058 
 3059         /* we are faster, let the other guy catch up */
 3060         if (beacon_tsft < local_tsft)
 3061                 sign = -1;
 3062         else
 3063                 sign = 1;
 3064 
 3065         if (IEEE80211_ADDR_EQ(ni->ni_bssid, ic->ic_bss->ni_bssid)) {
 3066                 if (!ieee80211_do_slow_print(ic, &did_print))
 3067                         return 0;
 3068                 printf("%s: tsft offset %s%llu\n", ic->ic_if.if_xname,
 3069                     (sign < 0) ? "-" : "",
 3070                     (sign < 0)
 3071                         ? (local_tsft - beacon_tsft)
 3072                         : (beacon_tsft - local_tsft));
 3073                 return 0;
 3074         }
 3075 
 3076         if (sign < 0)
 3077                 return 0;
 3078 
 3079         if (ieee80211_match_bss(ic, ni, 0) != 0)
 3080                 return 0;
 3081 
 3082         if (ieee80211_do_slow_print(ic, &did_print)) {
 3083                 printf("%s: ieee80211_ibss_merge: bssid mismatch %s\n",
 3084                     ic->ic_if.if_xname, ether_sprintf(ni->ni_bssid));
 3085                 printf("%s: my tsft %llu beacon tsft %llu\n",
 3086                     ic->ic_if.if_xname, local_tsft, beacon_tsft);
 3087                 printf("%s: sync TSF with %s\n",
 3088                     ic->ic_if.if_xname, ether_sprintf(ni->ni_macaddr));
 3089         }
 3090 
 3091         ic->ic_flags &= ~IEEE80211_F_SIBSS;
 3092 
 3093         /* negotiate rates with new IBSS */
 3094         ieee80211_fix_rate(ic, ni, IEEE80211_F_DOFRATE |
 3095             IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
 3096         if (ni->ni_rates.rs_nrates == 0) {
 3097                 if (ieee80211_do_slow_print(ic, &did_print)) {
 3098                         printf("%s: rates mismatch, BSSID %s\n",
 3099                             ic->ic_if.if_xname, ether_sprintf(ni->ni_bssid));
 3100                 }
 3101                 return 0;
 3102         }
 3103 
 3104         if (ieee80211_do_slow_print(ic, &did_print)) {
 3105                 printf("%s: sync BSSID %s -> ",
 3106                     ic->ic_if.if_xname, ether_sprintf(ic->ic_bss->ni_bssid));
 3107                 printf("%s ", ether_sprintf(ni->ni_bssid));
 3108                 printf("(from %s)\n", ether_sprintf(ni->ni_macaddr));
 3109         }
 3110 
 3111         ieee80211_node_newstate(ni, IEEE80211_STA_BSS);
 3112         (*ic->ic_node_copy)(ic, ic->ic_bss, ni);
 3113 
 3114         return ENETRESET;
 3115 }
 3116 
 3117 void
 3118 ieee80211_set_tim(struct ieee80211com *ic, int aid, int set)
 3119 {
 3120         if (set)
 3121                 setbit(ic->ic_tim_bitmap, aid & ~0xc000);
 3122         else
 3123                 clrbit(ic->ic_tim_bitmap, aid & ~0xc000);
 3124 }
 3125 
 3126 /*
 3127  * This function shall be called by drivers immediately after every DTIM.
 3128  * Transmit all group addressed MSDUs buffered at the AP.
 3129  */
 3130 void
 3131 ieee80211_notify_dtim(struct ieee80211com *ic)
 3132 {
 3133         /* NB: group addressed MSDUs are buffered in ic_bss */
 3134         struct ieee80211_node *ni = ic->ic_bss;
 3135         struct ifnet *ifp = &ic->ic_if;
 3136         struct ieee80211_frame *wh;
 3137         struct mbuf *m;
 3138 
 3139         KASSERT(ic->ic_opmode == IEEE80211_M_HOSTAP);
 3140 
 3141         while ((m = mq_dequeue(&ni->ni_savedq)) != NULL) {
 3142                 if (!mq_empty(&ni->ni_savedq)) {
 3143                         /* more queued frames, set the more data bit */
 3144                         wh = mtod(m, struct ieee80211_frame *);
 3145                         wh->i_fc[1] |= IEEE80211_FC1_MORE_DATA;
 3146                 }
 3147                 mq_enqueue(&ic->ic_pwrsaveq, m);
 3148                 if_start(ifp);
 3149         }
 3150         /* XXX assumes everything has been sent */
 3151         ic->ic_tim_mcast_pending = 0;
 3152 }
 3153 #endif  /* IEEE80211_STA_ONLY */
 3154 
 3155 /*
 3156  * Compare nodes in the tree by lladdr
 3157  */
 3158 int
 3159 ieee80211_node_cmp(const struct ieee80211_node *b1,
 3160     const struct ieee80211_node *b2)
 3161 {
 3162         return (memcmp(b1->ni_macaddr, b2->ni_macaddr, IEEE80211_ADDR_LEN));
 3163 }
 3164 
 3165 /*
 3166  * Compare nodes in the tree by essid
 3167  */
 3168 int
 3169 ieee80211_ess_cmp(const struct ieee80211_ess_rbt *b1,
 3170     const struct ieee80211_ess_rbt *b2)
 3171 {
 3172         return (memcmp(b1->essid, b2->essid, IEEE80211_NWID_LEN));
 3173 }
 3174 
 3175 /*
 3176  * Generate red-black tree function logic
 3177  */
 3178 RBT_GENERATE(ieee80211_tree, ieee80211_node, ni_node, ieee80211_node_cmp);
 3179 RBT_GENERATE(ieee80211_ess_tree, ieee80211_ess_rbt, ess_rbt, ieee80211_ess_cmp);

Cache object: e9e9b5e3d47540d150a0933ce60ba9ad


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]


This page is part of the FreeBSD/Linux Linux Kernel Cross-Reference, and was automatically generated using a modified version of the LXR engine.