sys/net80211/ieee80211_sta.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 2007-2008 Sam Leffler, Errno Consulting
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <sys/cdefs.h>
  29 #ifdef __FreeBSD__
  30 __FBSDID("$FreeBSD$");
  31 #endif
  32
  33 /*
  34  * IEEE 802.11 Station mode support.
  35  */
  36 #include "opt_inet.h"
  37 #include "opt_wlan.h"
  38
  39 #include <sys/param.h>
  40 #include <sys/systm.h>
  41 #include <sys/mbuf.h>
  42 #include <sys/malloc.h>
  43 #include <sys/kernel.h>
  44
  45 #include <sys/socket.h>
  46 #include <sys/sockio.h>
  47 #include <sys/endian.h>
  48 #include <sys/errno.h>
  49 #include <sys/proc.h>
  50 #include <sys/sysctl.h>
  51
  52 #include <net/if.h>
  53 #include <net/if_media.h>
  54 #include <net/if_llc.h>
  55 #include <net/if_dl.h>
  56 #include <net/if_var.h>
  57 #include <net/ethernet.h>
  58
  59 #include <net/bpf.h>
  60
  61 #include <net80211/ieee80211_var.h>
  62 #include <net80211/ieee80211_sta.h>
  63 #include <net80211/ieee80211_input.h>
  64 #ifdef IEEE80211_SUPPORT_SUPERG
  65 #include <net80211/ieee80211_superg.h>
  66 #endif
  67 #include <net80211/ieee80211_ratectl.h>
  68 #include <net80211/ieee80211_sta.h>
  69 #include <net80211/ieee80211_vht.h>
  70
  71 #define IEEE80211_RATE2MBS(r)   (((r) & IEEE80211_RATE_VAL) / 2)
  72
  73 static  void sta_vattach(struct ieee80211vap *);
  74 static  void sta_beacon_miss(struct ieee80211vap *);
  75 static  int sta_newstate(struct ieee80211vap *, enum ieee80211_state, int);
  76 static  int sta_input(struct ieee80211_node *, struct mbuf *,
  77             const struct ieee80211_rx_stats *, int, int);
  78 static void sta_recv_mgmt(struct ieee80211_node *, struct mbuf *,
  79             int subtype, const struct ieee80211_rx_stats *, int rssi, int nf);
  80 static void sta_recv_ctl(struct ieee80211_node *, struct mbuf *, int subtype);
  81
  82 void
  83 ieee80211_sta_attach(struct ieee80211com *ic)
  84 {
  85         ic->ic_vattach[IEEE80211_M_STA] = sta_vattach;
  86 }
  87
  88 void
  89 ieee80211_sta_detach(struct ieee80211com *ic)
  90 {
  91 }
  92
  93 static void
  94 sta_vdetach(struct ieee80211vap *vap)
  95 {
  96 }
  97
  98 static void
  99 sta_vattach(struct ieee80211vap *vap)
 100 {
 101         vap->iv_newstate = sta_newstate;
 102         vap->iv_input = sta_input;
 103         vap->iv_recv_mgmt = sta_recv_mgmt;
 104         vap->iv_recv_ctl = sta_recv_ctl;
 105         vap->iv_opdetach = sta_vdetach;
 106         vap->iv_bmiss = sta_beacon_miss;
 107 }
 108
 109 /*
 110  * Handle a beacon miss event.  The common code filters out
 111  * spurious events that can happen when scanning and/or before
 112  * reaching RUN state.
 113  */
 114 static void
 115 sta_beacon_miss(struct ieee80211vap *vap)
 116 {
 117         struct ieee80211com *ic = vap->iv_ic;
 118
 119         IEEE80211_LOCK_ASSERT(ic);
 120
 121         KASSERT((ic->ic_flags & IEEE80211_F_SCAN) == 0, ("scanning"));
 122         KASSERT(vap->iv_state >= IEEE80211_S_RUN,
 123             ("wrong state %s", ieee80211_state_name[vap->iv_state]));
 124
 125         IEEE80211_DPRINTF(vap, IEEE80211_MSG_STATE | IEEE80211_MSG_DEBUG,
 126             "beacon miss, mode %s state %s\n",
 127             ieee80211_opmode_name[vap->iv_opmode],
 128             ieee80211_state_name[vap->iv_state]);
 129
 130         if (vap->iv_state == IEEE80211_S_CSA) {
 131                 /*
 132                  * A Channel Switch is pending; assume we missed the
 133                  * beacon that would've completed the process and just
 134                  * force the switch.  If we made a mistake we'll not
 135                  * find the AP on the new channel and fall back to a
 136                  * normal scan.
 137                  */
 138                 ieee80211_csa_completeswitch(ic);
 139                 return;
 140         }
 141         if (++vap->iv_bmiss_count < vap->iv_bmiss_max) {
 142                 /*
 143                  * Send a directed probe req before falling back to a
 144                  * scan; if we receive a response ic_bmiss_count will
 145                  * be reset.  Some cards mistakenly report beacon miss
 146                  * so this avoids the expensive scan if the ap is
 147                  * still there.
 148                  */
 149                 ieee80211_send_probereq(vap->iv_bss, vap->iv_myaddr,
 150                         vap->iv_bss->ni_bssid, vap->iv_bss->ni_bssid,
 151                         vap->iv_bss->ni_essid, vap->iv_bss->ni_esslen);
 152                 return;
 153         }
 154
 155         callout_stop(&vap->iv_swbmiss);
 156         vap->iv_bmiss_count = 0;
 157         vap->iv_stats.is_beacon_miss++;
 158         if (vap->iv_roaming == IEEE80211_ROAMING_AUTO) {
 159 #ifdef IEEE80211_SUPPORT_SUPERG
 160
 161                 /*
 162                  * If we receive a beacon miss interrupt when using
 163                  * dynamic turbo, attempt to switch modes before
 164                  * reassociating.
 165                  */
 166                 if (IEEE80211_ATH_CAP(vap, vap->iv_bss, IEEE80211_NODE_TURBOP))
 167                         ieee80211_dturbo_switch(vap,
 168                             ic->ic_bsschan->ic_flags ^ IEEE80211_CHAN_TURBO);
 169 #endif
 170                 /*
 171                  * Try to reassociate before scanning for a new ap.
 172                  */
 173                 ieee80211_new_state(vap, IEEE80211_S_ASSOC, 1);
 174         } else {
 175                 /*
 176                  * Somebody else is controlling state changes (e.g.
 177                  * a user-mode app) don't do anything that would
 178                  * confuse them; just drop into scan mode so they'll
 179                  * notified of the state change and given control.
 180                  */
 181                 ieee80211_new_state(vap, IEEE80211_S_SCAN, 0);
 182         }
 183 }
 184
 185 /*
 186  * Handle deauth with reason.  We retry only for
 187  * the cases where we might succeed.  Otherwise
 188  * we downgrade the ap and scan.
 189  */
 190 static void
 191 sta_authretry(struct ieee80211vap *vap, struct ieee80211_node *ni, int reason)
 192 {
 193         switch (reason) {
 194         case IEEE80211_STATUS_SUCCESS:          /* NB: MLME assoc */
 195         case IEEE80211_STATUS_TIMEOUT:
 196         case IEEE80211_REASON_ASSOC_EXPIRE:
 197         case IEEE80211_REASON_NOT_AUTHED:
 198         case IEEE80211_REASON_NOT_ASSOCED:
 199         case IEEE80211_REASON_ASSOC_LEAVE:
 200         case IEEE80211_REASON_ASSOC_NOT_AUTHED:
 201                 IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_AUTH, 1);
 202                 break;
 203         default:
 204                 ieee80211_scan_assoc_fail(vap, vap->iv_bss->ni_macaddr, reason);
 205                 if (vap->iv_roaming == IEEE80211_ROAMING_AUTO)
 206                         ieee80211_check_scan_current(vap);
 207                 break;
 208         }
 209 }
 210
 211 static void
 212 sta_swbmiss_start(struct ieee80211vap *vap)
 213 {
 214
 215         if (vap->iv_flags_ext & IEEE80211_FEXT_SWBMISS) {
 216                 /*
 217                  * Start s/w beacon miss timer for devices w/o
 218                  * hardware support.  We fudge a bit here since
 219                  * we're doing this in software.
 220                  */
 221                 vap->iv_swbmiss_period = IEEE80211_TU_TO_TICKS(
 222                     2 * vap->iv_bmissthreshold * vap->iv_bss->ni_intval);
 223                 vap->iv_swbmiss_count = 0;
 224                 callout_reset(&vap->iv_swbmiss, vap->iv_swbmiss_period,
 225                     ieee80211_swbmiss, vap);
 226         }
 227 }
 228
 229 /*
 230  * IEEE80211_M_STA vap state machine handler.
 231  * This routine handles the main states in the 802.11 protocol.
 232  */
 233 static int
 234 sta_newstate(struct ieee80211vap *vap, enum ieee80211_state nstate, int arg)
 235 {
 236         struct ieee80211com *ic = vap->iv_ic;
 237         struct ieee80211_node *ni;
 238         enum ieee80211_state ostate;
 239
 240         IEEE80211_LOCK_ASSERT(ic);
 241
 242         ostate = vap->iv_state;
 243         IEEE80211_DPRINTF(vap, IEEE80211_MSG_STATE, "%s: %s -> %s (%d)\n",
 244             __func__, ieee80211_state_name[ostate],
 245             ieee80211_state_name[nstate], arg);
 246         vap->iv_state = nstate;                 /* state transition */
 247         callout_stop(&vap->iv_mgtsend);         /* XXX callout_drain */
 248         if (ostate != IEEE80211_S_SCAN)
 249                 ieee80211_cancel_scan(vap);     /* background scan */
 250         ni = vap->iv_bss;                       /* NB: no reference held */
 251         if (vap->iv_flags_ext & IEEE80211_FEXT_SWBMISS)
 252                 callout_stop(&vap->iv_swbmiss);
 253         switch (nstate) {
 254         case IEEE80211_S_INIT:
 255                 switch (ostate) {
 256                 case IEEE80211_S_SLEEP:
 257                         /* XXX wakeup */
 258                         /* XXX driver hook to wakeup the hardware? */
 259                 case IEEE80211_S_RUN:
 260                         IEEE80211_SEND_MGMT(ni,
 261                             IEEE80211_FC0_SUBTYPE_DISASSOC,
 262                             IEEE80211_REASON_ASSOC_LEAVE);
 263                         ieee80211_sta_leave(ni);
 264                         break;
 265                 case IEEE80211_S_ASSOC:
 266                         IEEE80211_SEND_MGMT(ni,
 267                             IEEE80211_FC0_SUBTYPE_DEAUTH,
 268                             IEEE80211_REASON_AUTH_LEAVE);
 269                         break;
 270                 case IEEE80211_S_SCAN:
 271                         ieee80211_cancel_scan(vap);
 272                         break;
 273                 default:
 274                         break;
 275                 }
 276                 if (ostate != IEEE80211_S_INIT) {
 277                         /* NB: optimize INIT -> INIT case */
 278                         ieee80211_reset_bss(vap);
 279                 }
 280                 if (vap->iv_auth->ia_detach != NULL)
 281                         vap->iv_auth->ia_detach(vap);
 282                 break;
 283         case IEEE80211_S_SCAN:
 284                 switch (ostate) {
 285                 case IEEE80211_S_INIT:
 286                         /*
 287                          * Initiate a scan.  We can come here as a result
 288                          * of an IEEE80211_IOC_SCAN_REQ too in which case
 289                          * the vap will be marked with IEEE80211_FEXT_SCANREQ
 290                          * and the scan request parameters will be present
 291                          * in iv_scanreq.  Otherwise we do the default.
 292                          */
 293                         if (vap->iv_flags_ext & IEEE80211_FEXT_SCANREQ) {
 294                                 ieee80211_check_scan(vap,
 295                                     vap->iv_scanreq_flags,
 296                                     vap->iv_scanreq_duration,
 297                                     vap->iv_scanreq_mindwell,
 298                                     vap->iv_scanreq_maxdwell,
 299                                     vap->iv_scanreq_nssid, vap->iv_scanreq_ssid);
 300                                 vap->iv_flags_ext &= ~IEEE80211_FEXT_SCANREQ;
 301                         } else
 302                                 ieee80211_check_scan_current(vap);
 303                         break;
 304                 case IEEE80211_S_SCAN:
 305                 case IEEE80211_S_AUTH:
 306                 case IEEE80211_S_ASSOC:
 307                         /*
 308                          * These can happen either because of a timeout
 309                          * on an assoc/auth response or because of a
 310                          * change in state that requires a reset.  For
 311                          * the former we're called with a non-zero arg
 312                          * that is the cause for the failure; pass this
 313                          * to the scan code so it can update state.
 314                          * Otherwise trigger a new scan unless we're in
 315                          * manual roaming mode in which case an application
 316                          * must issue an explicit scan request.
 317                          */
 318                         if (arg != 0)
 319                                 ieee80211_scan_assoc_fail(vap,
 320                                         vap->iv_bss->ni_macaddr, arg);
 321                         if (vap->iv_roaming == IEEE80211_ROAMING_AUTO)
 322                                 ieee80211_check_scan_current(vap);
 323                         break;
 324                 case IEEE80211_S_SLEEP:         /* beacon miss */
 325                         /*
 326                          * XXX if in sleep we need to wakeup the hardware.
 327                          */
 328                         /* FALLTHROUGH */
 329                 case IEEE80211_S_RUN:           /* beacon miss */
 330                         /*
 331                          * Beacon miss.  Notify user space and if not
 332                          * under control of a user application (roaming
 333                          * manual) kick off a scan to re-connect.
 334                          */
 335
 336                         ieee80211_sta_leave(ni);
 337                         if (vap->iv_roaming == IEEE80211_ROAMING_AUTO)
 338                                 ieee80211_check_scan_current(vap);
 339                         break;
 340                 default:
 341                         goto invalid;
 342                 }
 343                 break;
 344         case IEEE80211_S_AUTH:
 345                 switch (ostate) {
 346                 case IEEE80211_S_INIT:
 347                 case IEEE80211_S_SCAN:
 348                         IEEE80211_SEND_MGMT(ni,
 349                             IEEE80211_FC0_SUBTYPE_AUTH, 1);
 350                         break;
 351                 case IEEE80211_S_AUTH:
 352                 case IEEE80211_S_ASSOC:
 353                         switch (arg & 0xff) {
 354                         case IEEE80211_FC0_SUBTYPE_AUTH:
 355                                 /* ??? */
 356                                 IEEE80211_SEND_MGMT(ni,
 357                                     IEEE80211_FC0_SUBTYPE_AUTH, 2);
 358                                 break;
 359                         case IEEE80211_FC0_SUBTYPE_DEAUTH:
 360                                 sta_authretry(vap, ni, arg>>8);
 361                                 break;
 362                         }
 363                         break;
 364                 case IEEE80211_S_SLEEP:
 365                 case IEEE80211_S_RUN:
 366                         switch (arg & 0xff) {
 367                         case IEEE80211_FC0_SUBTYPE_AUTH:
 368                                 IEEE80211_SEND_MGMT(ni,
 369                                     IEEE80211_FC0_SUBTYPE_AUTH, 2);
 370                                 vap->iv_state = IEEE80211_S_RUN; /* stay RUN */
 371                                 break;
 372                         case IEEE80211_FC0_SUBTYPE_DEAUTH:
 373                                 ieee80211_sta_leave(ni);
 374                                 if (vap->iv_roaming == IEEE80211_ROAMING_AUTO) {
 375                                         /* try to reauth */
 376                                         IEEE80211_SEND_MGMT(ni,
 377                                             IEEE80211_FC0_SUBTYPE_AUTH, 1);
 378                                 }
 379                                 break;
 380                         }
 381                         break;
 382                 default:
 383                         goto invalid;
 384                 }
 385                 break;
 386         case IEEE80211_S_ASSOC:
 387                 switch (ostate) {
 388                 case IEEE80211_S_AUTH:
 389                 case IEEE80211_S_ASSOC:
 390                         IEEE80211_SEND_MGMT(ni,
 391                             IEEE80211_FC0_SUBTYPE_ASSOC_REQ, 0);
 392                         break;
 393                 case IEEE80211_S_SLEEP:         /* cannot happen */
 394                 case IEEE80211_S_RUN:
 395                         ieee80211_sta_leave(ni);
 396                         if (vap->iv_roaming == IEEE80211_ROAMING_AUTO) {
 397                                 IEEE80211_SEND_MGMT(ni, arg ?
 398                                     IEEE80211_FC0_SUBTYPE_REASSOC_REQ :
 399                                     IEEE80211_FC0_SUBTYPE_ASSOC_REQ, 0);
 400                         }
 401                         break;
 402                 default:
 403                         goto invalid;
 404                 }
 405                 break;
 406         case IEEE80211_S_RUN:
 407                 if (vap->iv_flags & IEEE80211_F_WPA) {
 408                         /* XXX validate prerequisites */
 409                 }
 410                 switch (ostate) {
 411                 case IEEE80211_S_RUN:
 412                 case IEEE80211_S_CSA:
 413                         break;
 414                 case IEEE80211_S_AUTH:          /* when join is done in fw */
 415                 case IEEE80211_S_ASSOC:
 416 #ifdef IEEE80211_DEBUG
 417                         if (ieee80211_msg_debug(vap)) {
 418                                 ieee80211_note(vap, "%s with %s ssid ",
 419                                     (vap->iv_opmode == IEEE80211_M_STA ?
 420                                     "associated" : "synchronized"),
 421                                     ether_sprintf(ni->ni_bssid));
 422                                 ieee80211_print_essid(vap->iv_bss->ni_essid,
 423                                     ni->ni_esslen);
 424                                 /* XXX MCS/HT */
 425                                 printf(" channel %d start %uMb\n",
 426                                     ieee80211_chan2ieee(ic, ic->ic_curchan),
 427                                     IEEE80211_RATE2MBS(ni->ni_txrate));
 428                         }
 429 #endif
 430                         ieee80211_scan_assoc_success(vap, ni->ni_macaddr);
 431                         ieee80211_notify_node_join(ni,
 432                             arg == IEEE80211_FC0_SUBTYPE_ASSOC_RESP);
 433                         break;
 434                 case IEEE80211_S_SLEEP:
 435                         /* Wake up from sleep */
 436                         vap->iv_sta_ps(vap, 0);
 437                         break;
 438                 default:
 439                         goto invalid;
 440                 }
 441                 ieee80211_sync_curchan(ic);
 442                 if (ostate != IEEE80211_S_RUN)
 443                         sta_swbmiss_start(vap);
 444                 /*
 445                  * When 802.1x is not in use mark the port authorized
 446                  * at this point so traffic can flow.
 447                  */
 448                 if (ni->ni_authmode != IEEE80211_AUTH_8021X)
 449                         ieee80211_node_authorize(ni);
 450                 /*
 451                  * Fake association when joining an existing bss.
 452                  *
 453                  * Don't do this if we're doing SLEEP->RUN.
 454                  */
 455                 if (ic->ic_newassoc != NULL && ostate != IEEE80211_S_SLEEP)
 456                         ic->ic_newassoc(vap->iv_bss, (ostate != IEEE80211_S_RUN));
 457                 break;
 458         case IEEE80211_S_CSA:
 459                 if (ostate != IEEE80211_S_RUN)
 460                         goto invalid;
 461                 break;
 462         case IEEE80211_S_SLEEP:
 463                 sta_swbmiss_start(vap);
 464                 vap->iv_sta_ps(vap, 1);
 465                 break;
 466         default:
 467         invalid:
 468                 IEEE80211_DPRINTF(vap, IEEE80211_MSG_STATE,
 469                     "%s: unexpected state transition %s -> %s\n", __func__,
 470                     ieee80211_state_name[ostate], ieee80211_state_name[nstate]);
 471                 break;
 472         }
 473         return 0;
 474 }
 475
 476 /*
 477  * Return non-zero if the frame is an echo of a multicast
 478  * frame sent by ourself.  The dir is known to be DSTODS.
 479  */
 480 static __inline int
 481 isdstods_mcastecho(struct ieee80211vap *vap, const struct ieee80211_frame *wh)
 482 {
 483 #define QWH4(wh)        ((const struct ieee80211_qosframe_addr4 *)wh)
 484 #define WH4(wh)         ((const struct ieee80211_frame_addr4 *)wh)
 485         const uint8_t *sa;
 486
 487         KASSERT(vap->iv_opmode == IEEE80211_M_STA, ("wrong mode"));
 488
 489         if (!IEEE80211_IS_MULTICAST(wh->i_addr3))
 490                 return 0;
 491         sa = IEEE80211_QOS_HAS_SEQ(wh) ? QWH4(wh)->i_addr4 : WH4(wh)->i_addr4;
 492         return IEEE80211_ADDR_EQ(sa, vap->iv_myaddr);
 493 #undef WH4
 494 #undef QWH4
 495 }
 496
 497 /*
 498  * Return non-zero if the frame is an echo of a multicast
 499  * frame sent by ourself.  The dir is known to be FROMDS.
 500  */
 501 static __inline int
 502 isfromds_mcastecho(struct ieee80211vap *vap, const struct ieee80211_frame *wh)
 503 {
 504         KASSERT(vap->iv_opmode == IEEE80211_M_STA, ("wrong mode"));
 505
 506         if (!IEEE80211_IS_MULTICAST(wh->i_addr1))
 507                 return 0;
 508         return IEEE80211_ADDR_EQ(wh->i_addr3, vap->iv_myaddr);
 509 }
 510
 511 /*
 512  * Decide if a received management frame should be
 513  * printed when debugging is enabled.  This filters some
 514  * of the less interesting frames that come frequently
 515  * (e.g. beacons).
 516  */
 517 static __inline int
 518 doprint(struct ieee80211vap *vap, int subtype)
 519 {
 520         switch (subtype) {
 521         case IEEE80211_FC0_SUBTYPE_BEACON:
 522                 return (vap->iv_ic->ic_flags & IEEE80211_F_SCAN);
 523         case IEEE80211_FC0_SUBTYPE_PROBE_REQ:
 524                 return 0;
 525         }
 526         return 1;
 527 }
 528
 529 /*
 530  * Process a received frame.  The node associated with the sender
 531  * should be supplied.  If nothing was found in the node table then
 532  * the caller is assumed to supply a reference to iv_bss instead.
 533  * The RSSI and a timestamp are also supplied.  The RSSI data is used
 534  * during AP scanning to select a AP to associate with; it can have
 535  * any units so long as values have consistent units and higher values
 536  * mean ``better signal''.  The receive timestamp is currently not used
 537  * by the 802.11 layer.
 538  */
 539 static int
 540 sta_input(struct ieee80211_node *ni, struct mbuf *m,
 541     const struct ieee80211_rx_stats *rxs, int rssi, int nf)
 542 {
 543         struct ieee80211vap *vap = ni->ni_vap;
 544         struct ieee80211com *ic = ni->ni_ic;
 545         struct ifnet *ifp = vap->iv_ifp;
 546         struct ieee80211_frame *wh;
 547         struct ieee80211_key *key;
 548         struct ether_header *eh;
 549         int hdrspace, need_tap = 1;     /* mbuf need to be tapped. */
 550         uint8_t dir, type, subtype, qos;
 551         uint8_t *bssid;
 552         int is_hw_decrypted = 0;
 553         int has_decrypted = 0;
 554
 555         /*
 556          * Some devices do hardware decryption all the way through
 557          * to pretending the frame wasn't encrypted in the first place.
 558          * So, tag it appropriately so it isn't discarded inappropriately.
 559          */
 560         if ((rxs != NULL) && (rxs->c_pktflags & IEEE80211_RX_F_DECRYPTED))
 561                 is_hw_decrypted = 1;
 562
 563         if (m->m_flags & M_AMPDU_MPDU) {
 564                 /*
 565                  * Fastpath for A-MPDU reorder q resubmission.  Frames
 566                  * w/ M_AMPDU_MPDU marked have already passed through
 567                  * here but were received out of order and been held on
 568                  * the reorder queue.  When resubmitted they are marked
 569                  * with the M_AMPDU_MPDU flag and we can bypass most of
 570                  * the normal processing.
 571                  */
 572                 wh = mtod(m, struct ieee80211_frame *);
 573                 type = IEEE80211_FC0_TYPE_DATA;
 574                 dir = wh->i_fc[1] & IEEE80211_FC1_DIR_MASK;
 575                 subtype = IEEE80211_FC0_SUBTYPE_QOS;
 576                 hdrspace = ieee80211_hdrspace(ic, wh);  /* XXX optimize? */
 577                 goto resubmit_ampdu;
 578         }
 579
 580         KASSERT(ni != NULL, ("null node"));
 581         ni->ni_inact = ni->ni_inact_reload;
 582
 583         type = -1;                      /* undefined */
 584
 585         if (m->m_pkthdr.len < sizeof(struct ieee80211_frame_min)) {
 586                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 587                     ni->ni_macaddr, NULL,
 588                     "too short (1): len %u", m->m_pkthdr.len);
 589                 vap->iv_stats.is_rx_tooshort++;
 590                 goto out;
 591         }
 592         /*
 593          * Bit of a cheat here, we use a pointer for a 3-address
 594          * frame format but don't reference fields past outside
 595          * ieee80211_frame_min w/o first validating the data is
 596          * present.
 597          */
 598         wh = mtod(m, struct ieee80211_frame *);
 599
 600         if ((wh->i_fc[0] & IEEE80211_FC0_VERSION_MASK) !=
 601             IEEE80211_FC0_VERSION_0) {
 602                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 603                     ni->ni_macaddr, NULL, "wrong version, fc %02x:%02x",
 604                     wh->i_fc[0], wh->i_fc[1]);
 605                 vap->iv_stats.is_rx_badversion++;
 606                 goto err;
 607         }
 608
 609         dir = wh->i_fc[1] & IEEE80211_FC1_DIR_MASK;
 610         type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;
 611         subtype = wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK;
 612         if ((ic->ic_flags & IEEE80211_F_SCAN) == 0) {
 613                 bssid = wh->i_addr2;
 614                 if (!IEEE80211_ADDR_EQ(bssid, ni->ni_bssid)) {
 615                         /* not interested in */
 616                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 617                             bssid, NULL, "%s", "not to bss");
 618                         vap->iv_stats.is_rx_wrongbss++;
 619                         goto out;
 620                 }
 621
 622                 /*
 623                  * Some devices may be in a promiscuous mode
 624                  * where they receive frames for multiple station
 625                  * addresses.
 626                  *
 627                  * If we receive a data frame that isn't
 628                  * destined to our VAP MAC, drop it.
 629                  *
 630                  * XXX TODO: This is only enforced when not scanning;
 631                  * XXX it assumes a software-driven scan will put the NIC
 632                  * XXX into a "no data frames" mode before setting this
 633                  * XXX flag. Otherwise it may be possible that we'll still
 634                  * XXX process data frames whilst scanning.
 635                  */
 636                 if ((! IEEE80211_IS_MULTICAST(wh->i_addr1))
 637                     && (! IEEE80211_ADDR_EQ(wh->i_addr1, IF_LLADDR(ifp)))) {
 638                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 639                             bssid, NULL, "not to cur sta: lladdr=%6D, addr1=%6D",
 640                             IF_LLADDR(ifp), ":", wh->i_addr1, ":");
 641                         vap->iv_stats.is_rx_wrongbss++;
 642                         goto out;
 643                 }
 644
 645                 IEEE80211_RSSI_LPF(ni->ni_avgrssi, rssi);
 646                 ni->ni_noise = nf;
 647                 if ( IEEE80211_HAS_SEQ(type, subtype) &&
 648                     !IEEE80211_IS_MULTICAST(wh->i_addr1)) {
 649                         uint8_t tid = ieee80211_gettid(wh);
 650                         if (IEEE80211_QOS_HAS_SEQ(wh) &&
 651                             TID_TO_WME_AC(tid) >= WME_AC_VI)
 652                                 ic->ic_wme.wme_hipri_traffic++;
 653                         if (! ieee80211_check_rxseq(ni, wh, bssid, rxs))
 654                                 goto out;
 655                 }
 656         }
 657
 658         switch (type) {
 659         case IEEE80211_FC0_TYPE_DATA:
 660                 hdrspace = ieee80211_hdrspace(ic, wh);
 661                 if (m->m_len < hdrspace &&
 662                     (m = m_pullup(m, hdrspace)) == NULL) {
 663                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 664                             ni->ni_macaddr, NULL,
 665                             "data too short: expecting %u", hdrspace);
 666                         vap->iv_stats.is_rx_tooshort++;
 667                         goto out;               /* XXX */
 668                 }
 669                 /*
 670                  * Handle A-MPDU re-ordering.  If the frame is to be
 671                  * processed directly then ieee80211_ampdu_reorder
 672                  * will return 0; otherwise it has consumed the mbuf
 673                  * and we should do nothing more with it.
 674                  */
 675                 if ((m->m_flags & M_AMPDU) &&
 676                     (dir == IEEE80211_FC1_DIR_FROMDS ||
 677                      dir == IEEE80211_FC1_DIR_DSTODS) &&
 678                     ieee80211_ampdu_reorder(ni, m, rxs) != 0) {
 679                         m = NULL;
 680                         goto out;
 681                 }
 682         resubmit_ampdu:
 683                 if (dir == IEEE80211_FC1_DIR_FROMDS) {
 684                         if ((ifp->if_flags & IFF_SIMPLEX) &&
 685                             isfromds_mcastecho(vap, wh)) {
 686                                 /*
 687                                  * In IEEE802.11 network, multicast
 688                                  * packets sent from "me" are broadcast
 689                                  * from the AP; silently discard for
 690                                  * SIMPLEX interface.
 691                                  */
 692                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 693                                     wh, "data", "%s", "multicast echo");
 694                                 vap->iv_stats.is_rx_mcastecho++;
 695                                 goto out;
 696                         }
 697                         if ((vap->iv_flags & IEEE80211_F_DWDS) &&
 698                             IEEE80211_IS_MULTICAST(wh->i_addr1)) {
 699                                 /*
 700                                  * DWDS sta's must drop 3-address mcast frames
 701                                  * as they will be sent separately as a 4-addr
 702                                  * frame.  Accepting the 3-addr frame will
 703                                  * confuse the bridge into thinking the sending
 704                                  * sta is located at the end of WDS link.
 705                                  */
 706                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT, wh,
 707                                     "3-address data", "%s", "DWDS enabled");
 708                                 vap->iv_stats.is_rx_mcastecho++;
 709                                 goto out;
 710                         }
 711                 } else if (dir == IEEE80211_FC1_DIR_DSTODS) {
 712                         if ((vap->iv_flags & IEEE80211_F_DWDS) == 0) {
 713                                 IEEE80211_DISCARD(vap,
 714                                     IEEE80211_MSG_INPUT, wh, "4-address data",
 715                                     "%s", "DWDS not enabled");
 716                                 vap->iv_stats.is_rx_wrongdir++;
 717                                 goto out;
 718                         }
 719                         if ((ifp->if_flags & IFF_SIMPLEX) &&
 720                             isdstods_mcastecho(vap, wh)) {
 721                                 /*
 722                                  * In IEEE802.11 network, multicast
 723                                  * packets sent from "me" are broadcast
 724                                  * from the AP; silently discard for
 725                                  * SIMPLEX interface.
 726                                  */
 727                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT, wh,
 728                                     "4-address data", "%s", "multicast echo");
 729                                 vap->iv_stats.is_rx_mcastecho++;
 730                                 goto out;
 731                         }
 732                 } else {
 733                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT, wh,
 734                             "data", "incorrect dir 0x%x", dir);
 735                         vap->iv_stats.is_rx_wrongdir++;
 736                         goto out;
 737                 }
 738
 739                 /*
 740                  * Handle privacy requirements for hardware decryption
 741                  * devices.
 742                  *
 743                  * For those devices, a handful of things happen.
 744                  *
 745                  * + If IV has been stripped, then we can't run
 746                  *   ieee80211_crypto_decap() - none of the key
 747                  * + If MIC has been stripped, we can't validate
 748                  *   MIC here.
 749                  * + If MIC fails, then we need to communicate a
 750                  *   MIC failure up to the stack - but we don't know
 751                  *   which key was used.
 752                  */
 753
 754                 /*
 755                  * Handle privacy requirements.  Note that we
 756                  * must not be preempted from here until after
 757                  * we (potentially) call ieee80211_crypto_demic;
 758                  * otherwise we may violate assumptions in the
 759                  * crypto cipher modules used to do delayed update
 760                  * of replay sequence numbers.
 761                  */
 762                 if (is_hw_decrypted || wh->i_fc[1] & IEEE80211_FC1_PROTECTED) {
 763                         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
 764                                 /*
 765                                  * Discard encrypted frames when privacy is off.
 766                                  */
 767                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 768                                     wh, "WEP", "%s", "PRIVACY off");
 769                                 vap->iv_stats.is_rx_noprivacy++;
 770                                 IEEE80211_NODE_STAT(ni, rx_noprivacy);
 771                                 goto out;
 772                         }
 773                         if (ieee80211_crypto_decap(ni, m, hdrspace, &key) == 0) {
 774                                 /* NB: stats+msgs handled in crypto_decap */
 775                                 IEEE80211_NODE_STAT(ni, rx_wepfail);
 776                                 goto out;
 777                         }
 778                         wh = mtod(m, struct ieee80211_frame *);
 779                         wh->i_fc[1] &= ~IEEE80211_FC1_PROTECTED;
 780                         has_decrypted = 1;
 781                 } else {
 782                         /* XXX M_WEP and IEEE80211_F_PRIVACY */
 783                         key = NULL;
 784                 }
 785
 786                 /*
 787                  * Save QoS bits for use below--before we strip the header.
 788                  */
 789                 if (subtype == IEEE80211_FC0_SUBTYPE_QOS)
 790                         qos = ieee80211_getqos(wh)[0];
 791                 else
 792                         qos = 0;
 793
 794                 /*
 795                  * Next up, any fragmentation.
 796                  */
 797                 if (!IEEE80211_IS_MULTICAST(wh->i_addr1)) {
 798                         m = ieee80211_defrag(ni, m, hdrspace);
 799                         if (m == NULL) {
 800                                 /* Fragment dropped or frame not complete yet */
 801                                 goto out;
 802                         }
 803                 }
 804                 wh = NULL;              /* no longer valid, catch any uses */
 805
 806                 /*
 807                  * Next strip any MSDU crypto bits.
 808                  *
 809                  * Note: we can't do MIC stripping/verification if the
 810                  * upper layer has stripped it.  We have to check MIC
 811                  * ourselves.  So, key may be NULL, but we have to check
 812                  * the RX status.
 813                  */
 814                 if (!ieee80211_crypto_demic(vap, key, m, 0)) {
 815                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 816                             ni->ni_macaddr, "data", "%s", "demic error");
 817                         vap->iv_stats.is_rx_demicfail++;
 818                         IEEE80211_NODE_STAT(ni, rx_demicfail);
 819                         goto out;
 820                 }
 821
 822                 /* copy to listener after decrypt */
 823                 if (ieee80211_radiotap_active_vap(vap))
 824                         ieee80211_radiotap_rx(vap, m);
 825                 need_tap = 0;
 826
 827                 /*
 828                  * Finally, strip the 802.11 header.
 829                  */
 830                 m = ieee80211_decap(vap, m, hdrspace);
 831                 if (m == NULL) {
 832                         /* XXX mask bit to check for both */
 833                         /* don't count Null data frames as errors */
 834                         if (subtype == IEEE80211_FC0_SUBTYPE_NODATA ||
 835                             subtype == IEEE80211_FC0_SUBTYPE_QOS_NULL)
 836                                 goto out;
 837                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 838                             ni->ni_macaddr, "data", "%s", "decap error");
 839                         vap->iv_stats.is_rx_decap++;
 840                         IEEE80211_NODE_STAT(ni, rx_decap);
 841                         goto err;
 842                 }
 843                 eh = mtod(m, struct ether_header *);
 844                 if (!ieee80211_node_is_authorized(ni)) {
 845                         /*
 846                          * Deny any non-PAE frames received prior to
 847                          * authorization.  For open/shared-key
 848                          * authentication the port is mark authorized
 849                          * after authentication completes.  For 802.1x
 850                          * the port is not marked authorized by the
 851                          * authenticator until the handshake has completed.
 852                          */
 853                         if (eh->ether_type != htons(ETHERTYPE_PAE)) {
 854                                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 855                                     eh->ether_shost, "data",
 856                                     "unauthorized port: ether type 0x%x len %u",
 857                                     eh->ether_type, m->m_pkthdr.len);
 858                                 vap->iv_stats.is_rx_unauth++;
 859                                 IEEE80211_NODE_STAT(ni, rx_unauth);
 860                                 goto err;
 861                         }
 862                 } else {
 863                         /*
 864                          * When denying unencrypted frames, discard
 865                          * any non-PAE frames received without encryption.
 866                          */
 867                         if ((vap->iv_flags & IEEE80211_F_DROPUNENC) &&
 868                             ((has_decrypted == 0) && (m->m_flags & M_WEP) == 0) &&
 869                             (is_hw_decrypted == 0) &&
 870                             eh->ether_type != htons(ETHERTYPE_PAE)) {
 871                                 /*
 872                                  * Drop unencrypted frames.
 873                                  */
 874                                 vap->iv_stats.is_rx_unencrypted++;
 875                                 IEEE80211_NODE_STAT(ni, rx_unencrypted);
 876                                 goto out;
 877                         }
 878                 }
 879                 /* XXX require HT? */
 880                 if (qos & IEEE80211_QOS_AMSDU) {
 881                         m = ieee80211_decap_amsdu(ni, m);
 882                         if (m == NULL)
 883                                 return IEEE80211_FC0_TYPE_DATA;
 884                 } else {
 885 #ifdef IEEE80211_SUPPORT_SUPERG
 886                         m = ieee80211_decap_fastframe(vap, ni, m);
 887                         if (m == NULL)
 888                                 return IEEE80211_FC0_TYPE_DATA;
 889 #endif
 890                 }
 891                 ieee80211_deliver_data(vap, ni, m);
 892                 return IEEE80211_FC0_TYPE_DATA;
 893
 894         case IEEE80211_FC0_TYPE_MGT:
 895                 vap->iv_stats.is_rx_mgmt++;
 896                 IEEE80211_NODE_STAT(ni, rx_mgmt);
 897                 if (dir != IEEE80211_FC1_DIR_NODS) {
 898                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 899                             wh, "data", "incorrect dir 0x%x", dir);
 900                         vap->iv_stats.is_rx_wrongdir++;
 901                         goto err;
 902                 }
 903                 if (m->m_pkthdr.len < sizeof(struct ieee80211_frame)) {
 904                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 905                             ni->ni_macaddr, "mgt", "too short: len %u",
 906                             m->m_pkthdr.len);
 907                         vap->iv_stats.is_rx_tooshort++;
 908                         goto out;
 909                 }
 910 #ifdef IEEE80211_DEBUG
 911                 if ((ieee80211_msg_debug(vap) && doprint(vap, subtype)) ||
 912                     ieee80211_msg_dumppkts(vap)) {
 913                         if_printf(ifp, "received %s from %s rssi %d\n",
 914                             ieee80211_mgt_subtype_name(subtype),
 915                             ether_sprintf(wh->i_addr2), rssi);
 916                 }
 917 #endif
 918
 919                 /*
 920                  * Note: See above for hardware offload privacy requirements.
 921                  *       It also applies here.
 922                  */
 923
 924                 /*
 925                  * Again, having encrypted flag set check would be good, but
 926                  * then we have to also handle crypto_decap() like above.
 927                  */
 928                 if (wh->i_fc[1] & IEEE80211_FC1_PROTECTED) {
 929                         if (subtype != IEEE80211_FC0_SUBTYPE_AUTH) {
 930                                 /*
 931                                  * Only shared key auth frames with a challenge
 932                                  * should be encrypted, discard all others.
 933                                  */
 934                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 935                                     wh, ieee80211_mgt_subtype_name(subtype),
 936                                     "%s", "WEP set but not permitted");
 937                                 vap->iv_stats.is_rx_mgtdiscard++; /* XXX */
 938                                 goto out;
 939                         }
 940                         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
 941                                 /*
 942                                  * Discard encrypted frames when privacy is off.
 943                                  */
 944                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 945                                     wh, "mgt", "%s", "WEP set but PRIVACY off");
 946                                 vap->iv_stats.is_rx_noprivacy++;
 947                                 goto out;
 948                         }
 949                         hdrspace = ieee80211_hdrspace(ic, wh);
 950
 951                         /*
 952                          * Again, if IV/MIC was stripped, then this whole
 953                          * setup will fail.  That's going to need some poking.
 954                          */
 955                         if (ieee80211_crypto_decap(ni, m, hdrspace, &key) == 0) {
 956                                 /* NB: stats+msgs handled in crypto_decap */
 957                                 goto out;
 958                         }
 959                         has_decrypted = 1;
 960                         wh = mtod(m, struct ieee80211_frame *);
 961                         wh->i_fc[1] &= ~IEEE80211_FC1_PROTECTED;
 962                 }
 963                 vap->iv_recv_mgmt(ni, m, subtype, rxs, rssi, nf);
 964                 goto out;
 965
 966         case IEEE80211_FC0_TYPE_CTL:
 967                 vap->iv_stats.is_rx_ctl++;
 968                 IEEE80211_NODE_STAT(ni, rx_ctrl);
 969                 vap->iv_recv_ctl(ni, m, subtype);
 970                 goto out;
 971
 972         default:
 973                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
 974                     wh, NULL, "bad frame type 0x%x", type);
 975                 /* should not come here */
 976                 break;
 977         }
 978 err:
 979         if_inc_counter(ifp, IFCOUNTER_IERRORS, 1);
 980 out:
 981         if (m != NULL) {
 982                 if (need_tap && ieee80211_radiotap_active_vap(vap))
 983                         ieee80211_radiotap_rx(vap, m);
 984                 m_freem(m);
 985         }
 986         return type;
 987 }
 988
 989 static void
 990 sta_auth_open(struct ieee80211_node *ni, struct ieee80211_frame *wh,
 991     int rssi, int nf, uint16_t seq, uint16_t status)
 992 {
 993         struct ieee80211vap *vap = ni->ni_vap;
 994
 995         if (ni->ni_authmode == IEEE80211_AUTH_SHARED) {
 996                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
 997                     ni->ni_macaddr, "open auth",
 998                     "bad sta auth mode %u", ni->ni_authmode);
 999                 vap->iv_stats.is_rx_bad_auth++; /* XXX */
1000                 return;
1001         }
1002         if (vap->iv_state != IEEE80211_S_AUTH ||
1003             seq != IEEE80211_AUTH_OPEN_RESPONSE) {
1004                 vap->iv_stats.is_rx_bad_auth++;
1005                 return;
1006         }
1007         if (status != 0) {
1008                 IEEE80211_NOTE(vap, IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH,
1009                     ni, "open auth failed (reason %d)", status);
1010                 vap->iv_stats.is_rx_auth_fail++;
1011                 vap->iv_stats.is_rx_authfail_code = status;
1012                 ieee80211_new_state(vap, IEEE80211_S_SCAN,
1013                     IEEE80211_SCAN_FAIL_STATUS);
1014         } else
1015                 ieee80211_new_state(vap, IEEE80211_S_ASSOC, 0);
1016 }
1017
1018 static void
1019 sta_auth_shared(struct ieee80211_node *ni, struct ieee80211_frame *wh,
1020     uint8_t *frm, uint8_t *efrm, int rssi, int nf,
1021     uint16_t seq, uint16_t status)
1022 {
1023         struct ieee80211vap *vap = ni->ni_vap;
1024         uint8_t *challenge;
1025
1026         /*
1027          * NB: this can happen as we allow pre-shared key
1028          * authentication to be enabled w/o wep being turned
1029          * on so that configuration of these can be done
1030          * in any order.  It may be better to enforce the
1031          * ordering in which case this check would just be
1032          * for sanity/consistency.
1033          */
1034         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
1035                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1036                     ni->ni_macaddr, "shared key auth",
1037                     "%s", " PRIVACY is disabled");
1038                 goto bad;
1039         }
1040         /*
1041          * Pre-shared key authentication is evil; accept
1042          * it only if explicitly configured (it is supported
1043          * mainly for compatibility with clients like OS X).
1044          */
1045         if (ni->ni_authmode != IEEE80211_AUTH_AUTO &&
1046             ni->ni_authmode != IEEE80211_AUTH_SHARED) {
1047                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1048                     ni->ni_macaddr, "shared key auth",
1049                     "bad sta auth mode %u", ni->ni_authmode);
1050                 vap->iv_stats.is_rx_bad_auth++; /* XXX maybe a unique error? */
1051                 goto bad;
1052         }
1053
1054         challenge = NULL;
1055         if (frm + 1 < efrm) {
1056                 if ((frm[1] + 2) > (efrm - frm)) {
1057                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1058                             ni->ni_macaddr, "shared key auth",
1059                             "ie %d/%d too long",
1060                             frm[0], (frm[1] + 2) - (efrm - frm));
1061                         vap->iv_stats.is_rx_bad_auth++;
1062                         goto bad;
1063                 }
1064                 if (*frm == IEEE80211_ELEMID_CHALLENGE)
1065                         challenge = frm;
1066                 frm += frm[1] + 2;
1067         }
1068         switch (seq) {
1069         case IEEE80211_AUTH_SHARED_CHALLENGE:
1070         case IEEE80211_AUTH_SHARED_RESPONSE:
1071                 if (challenge == NULL) {
1072                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1073                             ni->ni_macaddr, "shared key auth",
1074                             "%s", "no challenge");
1075                         vap->iv_stats.is_rx_bad_auth++;
1076                         goto bad;
1077                 }
1078                 if (challenge[1] != IEEE80211_CHALLENGE_LEN) {
1079                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1080                             ni->ni_macaddr, "shared key auth",
1081                             "bad challenge len %d", challenge[1]);
1082                         vap->iv_stats.is_rx_bad_auth++;
1083                         goto bad;
1084                 }
1085         default:
1086                 break;
1087         }
1088         if (vap->iv_state != IEEE80211_S_AUTH)
1089                 return;
1090         switch (seq) {
1091         case IEEE80211_AUTH_SHARED_PASS:
1092                 if (ni->ni_challenge != NULL) {
1093                         IEEE80211_FREE(ni->ni_challenge, M_80211_NODE);
1094                         ni->ni_challenge = NULL;
1095                 }
1096                 if (status != 0) {
1097                         IEEE80211_NOTE_FRAME(vap,
1098                             IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH, wh,
1099                             "shared key auth failed (reason %d)", status);
1100                         vap->iv_stats.is_rx_auth_fail++;
1101                         vap->iv_stats.is_rx_authfail_code = status;
1102                         return;
1103                 }
1104                 ieee80211_new_state(vap, IEEE80211_S_ASSOC, 0);
1105                 break;
1106         case IEEE80211_AUTH_SHARED_CHALLENGE:
1107                 if (!ieee80211_alloc_challenge(ni))
1108                         return;
1109                 /* XXX could optimize by passing recvd challenge */
1110                 memcpy(ni->ni_challenge, &challenge[2], challenge[1]);
1111                 IEEE80211_SEND_MGMT(ni,
1112                         IEEE80211_FC0_SUBTYPE_AUTH, seq + 1);
1113                 break;
1114         default:
1115                 IEEE80211_DISCARD(vap, IEEE80211_MSG_AUTH,
1116                     wh, "shared key auth", "bad seq %d", seq);
1117                 vap->iv_stats.is_rx_bad_auth++;
1118                 return;
1119         }
1120         return;
1121 bad:
1122         /*
1123          * Kick the state machine.  This short-circuits
1124          * using the mgt frame timeout to trigger the
1125          * state transition.
1126          */
1127         if (vap->iv_state == IEEE80211_S_AUTH)
1128                 ieee80211_new_state(vap, IEEE80211_S_SCAN,
1129                     IEEE80211_SCAN_FAIL_STATUS);
1130 }
1131
1132 int
1133 ieee80211_parse_wmeparams(struct ieee80211vap *vap, uint8_t *frm,
1134         const struct ieee80211_frame *wh)
1135 {
1136 #define MS(_v, _f)      (((_v) & _f) >> _f##_S)
1137         struct ieee80211_wme_state *wme = &vap->iv_ic->ic_wme;
1138         u_int len = frm[1], qosinfo;
1139         int i;
1140
1141         if (len < sizeof(struct ieee80211_wme_param)-2) {
1142                 IEEE80211_DISCARD_IE(vap,
1143                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WME,
1144                     wh, "WME", "too short, len %u", len);
1145                 return -1;
1146         }
1147         qosinfo = frm[__offsetof(struct ieee80211_wme_param, param_qosInfo)];
1148         qosinfo &= WME_QOSINFO_COUNT;
1149         /* XXX do proper check for wraparound */
1150         if (qosinfo == wme->wme_wmeChanParams.cap_info)
1151                 return 0;
1152         frm += __offsetof(struct ieee80211_wme_param, params_acParams);
1153         for (i = 0; i < WME_NUM_AC; i++) {
1154                 struct wmeParams *wmep =
1155                         &wme->wme_wmeChanParams.cap_wmeParams[i];
1156                 /* NB: ACI not used */
1157                 wmep->wmep_acm = MS(frm[0], WME_PARAM_ACM);
1158                 wmep->wmep_aifsn = MS(frm[0], WME_PARAM_AIFSN);
1159                 wmep->wmep_logcwmin = MS(frm[1], WME_PARAM_LOGCWMIN);
1160                 wmep->wmep_logcwmax = MS(frm[1], WME_PARAM_LOGCWMAX);
1161                 wmep->wmep_txopLimit = le16dec(frm+2);
1162                 frm += 4;
1163         }
1164         wme->wme_wmeChanParams.cap_info = qosinfo;
1165         return 1;
1166 #undef MS
1167 }
1168
1169 /*
1170  * Process 11h Channel Switch Announcement (CSA) ie.  If this
1171  * is the first CSA then initiate the switch.  Otherwise we
1172  * track state and trigger completion and/or cancel of the switch.
1173  * XXX should be public for IBSS use
1174  */
1175 static void
1176 ieee80211_parse_csaparams(struct ieee80211vap *vap, uint8_t *frm,
1177         const struct ieee80211_frame *wh)
1178 {
1179         struct ieee80211com *ic = vap->iv_ic;
1180         const struct ieee80211_csa_ie *csa =
1181             (const struct ieee80211_csa_ie *) frm;
1182
1183         KASSERT(vap->iv_state >= IEEE80211_S_RUN,
1184             ("state %s", ieee80211_state_name[vap->iv_state]));
1185
1186         if (csa->csa_mode > 1) {
1187                 IEEE80211_DISCARD_IE(vap,
1188                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_DOTH,
1189                     wh, "CSA", "invalid mode %u", csa->csa_mode);
1190                 return;
1191         }
1192         IEEE80211_LOCK(ic);
1193         if ((ic->ic_flags & IEEE80211_F_CSAPENDING) == 0) {
1194                 /*
1195                  * Convert the channel number to a channel reference.  We
1196                  * try first to preserve turbo attribute of the current
1197                  * channel then fallback.  Note this will not work if the
1198                  * CSA specifies a channel that requires a band switch (e.g.
1199                  * 11a => 11g).  This is intentional as 11h is defined only
1200                  * for 5GHz/11a and because the switch does not involve a
1201                  * reassociation, protocol state (capabilities, negotated
1202                  * rates, etc) may/will be wrong.
1203                  */
1204                 struct ieee80211_channel *c =
1205                     ieee80211_find_channel_byieee(ic, csa->csa_newchan,
1206                         (ic->ic_bsschan->ic_flags & IEEE80211_CHAN_ALLTURBO));
1207                 if (c == NULL) {
1208                         c = ieee80211_find_channel_byieee(ic,
1209                             csa->csa_newchan,
1210                             (ic->ic_bsschan->ic_flags & IEEE80211_CHAN_ALL));
1211                         if (c == NULL) {
1212                                 IEEE80211_DISCARD_IE(vap,
1213                                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_DOTH,
1214                                     wh, "CSA", "invalid channel %u",
1215                                     csa->csa_newchan);
1216                                 goto done;
1217                         }
1218                 }
1219 #if IEEE80211_CSA_COUNT_MIN > 0
1220                 if (csa->csa_count < IEEE80211_CSA_COUNT_MIN) {
1221                         /*
1222                          * Require at least IEEE80211_CSA_COUNT_MIN count to
1223                          * reduce the risk of being redirected by a fabricated
1224                          * CSA.  If a valid CSA is dropped we'll still get a
1225                          * beacon miss when the AP leaves the channel so we'll
1226                          * eventually follow to the new channel.
1227                          *
1228                          * NOTE: this violates the 11h spec that states that
1229                          * count may be any value and if 0 then a switch
1230                          * should happen asap.
1231                          */
1232                         IEEE80211_DISCARD_IE(vap,
1233                             IEEE80211_MSG_ELEMID | IEEE80211_MSG_DOTH,
1234                             wh, "CSA", "count %u too small, must be >= %u",
1235                             csa->csa_count, IEEE80211_CSA_COUNT_MIN);
1236                         goto done;
1237                 }
1238 #endif
1239                 ieee80211_csa_startswitch(ic, c, csa->csa_mode, csa->csa_count);
1240         } else {
1241                 /*
1242                  * Validate this ie against the initial CSA.  We require
1243                  * mode and channel not change and the count must be
1244                  * monotonically decreasing.  This may be pointless and
1245                  * canceling the switch as a result may be too paranoid but
1246                  * in the worst case if we drop out of CSA because of this
1247                  * and the AP does move then we'll just end up taking a
1248                  * beacon miss and scan to find the AP.
1249                  *
1250                  * XXX may want <= on count as we also process ProbeResp
1251                  * frames and those may come in w/ the same count as the
1252                  * previous beacon; but doing so leaves us open to a stuck
1253                  * count until we add a dead-man timer
1254                  */
1255                 if (!(csa->csa_count < ic->ic_csa_count &&
1256                       csa->csa_mode == ic->ic_csa_mode &&
1257                       csa->csa_newchan == ieee80211_chan2ieee(ic, ic->ic_csa_newchan))) {
1258                         IEEE80211_NOTE_FRAME(vap, IEEE80211_MSG_DOTH, wh,
1259                             "CSA ie mismatch, initial ie <%d,%d,%d>, "
1260                             "this ie <%d,%d,%d>", ic->ic_csa_mode,
1261                             ic->ic_csa_newchan, ic->ic_csa_count,
1262                             csa->csa_mode, csa->csa_newchan, csa->csa_count);
1263                         ieee80211_csa_cancelswitch(ic);
1264                 } else {
1265                         if (csa->csa_count <= 1)
1266                                 ieee80211_csa_completeswitch(ic);
1267                         else
1268                                 ic->ic_csa_count = csa->csa_count;
1269                 }
1270         }
1271 done:
1272         IEEE80211_UNLOCK(ic);
1273 }
1274
1275 /*
1276  * Return non-zero if a background scan may be continued:
1277  * o bg scan is active
1278  * o no channel switch is pending
1279  * o there has not been any traffic recently
1280  * o no full-offload scan support (no need for explicitly continuing scan then)
1281  *
1282  * Note we do not check if there is an administrative enable;
1283  * this is only done to start the scan.  We assume that any
1284  * change in state will be accompanied by a request to cancel
1285  * active scans which will otherwise cause this test to fail.
1286  */
1287 static __inline int
1288 contbgscan(struct ieee80211vap *vap)
1289 {
1290         struct ieee80211com *ic = vap->iv_ic;
1291
1292         return ((ic->ic_flags_ext & IEEE80211_FEXT_BGSCAN) &&
1293             (ic->ic_flags & IEEE80211_F_CSAPENDING) == 0 &&
1294             !(vap->iv_flags_ext & IEEE80211_FEXT_SCAN_OFFLOAD) &&
1295             vap->iv_state == IEEE80211_S_RUN &&         /* XXX? */
1296             ieee80211_time_after(ticks, ic->ic_lastdata + vap->iv_bgscanidle));
1297 }
1298
1299 /*
1300  * Return non-zero if a backgrond scan may be started:
1301  * o bg scanning is administratively enabled
1302  * o no channel switch is pending
1303  * o we are not boosted on a dynamic turbo channel
1304  * o there has not been a scan recently
1305  * o there has not been any traffic recently (don't check if full-offload scan)
1306  */
1307 static __inline int
1308 startbgscan(struct ieee80211vap *vap)
1309 {
1310         struct ieee80211com *ic = vap->iv_ic;
1311
1312         return ((vap->iv_flags & IEEE80211_F_BGSCAN) &&
1313             (ic->ic_flags & IEEE80211_F_CSAPENDING) == 0 &&
1314 #ifdef IEEE80211_SUPPORT_SUPERG
1315             !IEEE80211_IS_CHAN_DTURBO(ic->ic_curchan) &&
1316 #endif
1317             ieee80211_time_after(ticks, ic->ic_lastscan + vap->iv_bgscanintvl) &&
1318             ((vap->iv_flags_ext & IEEE80211_FEXT_SCAN_OFFLOAD) ||
1319              ieee80211_time_after(ticks, ic->ic_lastdata + vap->iv_bgscanidle)));
1320 }
1321
1322 #ifdef  notyet
1323 /*
1324  * Compare two quiet IEs and return if they are equivalent.
1325  *
1326  * The tbttcount isnt checked - that's not part of the configuration.
1327  */
1328 static int
1329 compare_quiet_ie(const struct ieee80211_quiet_ie *q1,
1330     const struct ieee80211_quiet_ie *q2)
1331 {
1332
1333         if (q1->period != q2->period)
1334                 return (0);
1335         if (le16dec(&q1->duration) != le16dec(&q2->duration))
1336                 return (0);
1337         if (le16dec(&q1->offset) != le16dec(&q2->offset))
1338                 return (0);
1339         return (1);
1340 }
1341 #endif
1342
1343 static void
1344 sta_recv_mgmt(struct ieee80211_node *ni, struct mbuf *m0, int subtype,
1345     const struct ieee80211_rx_stats *rxs,
1346     int rssi, int nf)
1347 {
1348 #define ISREASSOC(_st)  ((_st) == IEEE80211_FC0_SUBTYPE_REASSOC_RESP)
1349         struct ieee80211vap *vap = ni->ni_vap;
1350         struct ieee80211com *ic = ni->ni_ic;
1351         struct ieee80211_channel *rxchan = ic->ic_curchan;
1352         struct ieee80211_frame *wh;
1353         uint8_t *frm, *efrm;
1354         uint8_t *rates, *xrates, *wme, *htcap, *htinfo;
1355         uint8_t *vhtcap, *vhtopmode;
1356         uint8_t rate;
1357         int ht_state_change = 0, do_ht = 0;
1358
1359         wh = mtod(m0, struct ieee80211_frame *);
1360         frm = (uint8_t *)&wh[1];
1361         efrm = mtod(m0, uint8_t *) + m0->m_len;
1362         switch (subtype) {
1363         case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
1364         case IEEE80211_FC0_SUBTYPE_BEACON: {
1365                 struct ieee80211_scanparams scan;
1366                 struct ieee80211_channel *c;
1367                 /*
1368                  * We process beacon/probe response frames:
1369                  *    o when scanning, or
1370                  *    o station mode when associated (to collect state
1371                  *      updates such as 802.11g slot time)
1372                  * Frames otherwise received are discarded.
1373                  */
1374                 if (!((ic->ic_flags & IEEE80211_F_SCAN) || ni->ni_associd)) {
1375                         vap->iv_stats.is_rx_mgtdiscard++;
1376                         return;
1377                 }
1378
1379                 /* Override RX channel as appropriate */
1380                 if (rxs != NULL) {
1381                         c = ieee80211_lookup_channel_rxstatus(vap, rxs);
1382                         if (c != NULL)
1383                                 rxchan = c;
1384                 }
1385
1386                 /* XXX probe response in sta mode when !scanning? */
1387                 if (ieee80211_parse_beacon(ni, m0, rxchan, &scan) != 0) {
1388                         if (! (ic->ic_flags & IEEE80211_F_SCAN))
1389                                 vap->iv_stats.is_beacon_bad++;
1390                         return;
1391                 }
1392
1393                 /*
1394                  * Count frame now that we know it's to be processed.
1395                  */
1396                 if (subtype == IEEE80211_FC0_SUBTYPE_BEACON) {
1397                         vap->iv_stats.is_rx_beacon++;           /* XXX remove */
1398                         IEEE80211_NODE_STAT(ni, rx_beacons);
1399                 } else
1400                         IEEE80211_NODE_STAT(ni, rx_proberesp);
1401                 /*
1402                  * When operating in station mode, check for state updates.
1403                  * Be careful to ignore beacons received while doing a
1404                  * background scan.  We consider only 11g/WMM stuff right now.
1405                  */
1406                 if (ni->ni_associd != 0 &&
1407                     ((ic->ic_flags & IEEE80211_F_SCAN) == 0 ||
1408                      IEEE80211_ADDR_EQ(wh->i_addr2, ni->ni_bssid))) {
1409                         /* record tsf of last beacon */
1410                         memcpy(ni->ni_tstamp.data, scan.tstamp,
1411                                 sizeof(ni->ni_tstamp));
1412                         /* count beacon frame for s/w bmiss handling */
1413                         vap->iv_swbmiss_count++;
1414                         vap->iv_bmiss_count = 0;
1415                         if (ni->ni_erp != scan.erp) {
1416                                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ASSOC,
1417                                     wh->i_addr2,
1418                                     "erp change: was 0x%x, now 0x%x",
1419                                     ni->ni_erp, scan.erp);
1420                                 if (IEEE80211_IS_CHAN_ANYG(ic->ic_curchan) &&
1421                                     (ni->ni_erp & IEEE80211_ERP_USE_PROTECTION))
1422                                         ic->ic_flags |= IEEE80211_F_USEPROT;
1423                                 else
1424                                         ic->ic_flags &= ~IEEE80211_F_USEPROT;
1425                                 ni->ni_erp = scan.erp;
1426                                 /* XXX statistic */
1427                                 /* XXX driver notification */
1428                         }
1429                         if ((ni->ni_capinfo ^ scan.capinfo) & IEEE80211_CAPINFO_SHORT_SLOTTIME) {
1430                                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ASSOC,
1431                                     wh->i_addr2,
1432                                     "capabilities change: was 0x%x, now 0x%x",
1433                                     ni->ni_capinfo, scan.capinfo);
1434                                 /*
1435                                  * NB: we assume short preamble doesn't
1436                                  *     change dynamically
1437                                  */
1438                                 ieee80211_set_shortslottime(ic,
1439                                         IEEE80211_IS_CHAN_A(ic->ic_bsschan) ||
1440                                         (scan.capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME));
1441                                 ni->ni_capinfo = (ni->ni_capinfo &~ IEEE80211_CAPINFO_SHORT_SLOTTIME)
1442                                                | (scan.capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME);
1443                                 /* XXX statistic */
1444                         }
1445                         if (scan.wme != NULL &&
1446                             (ni->ni_flags & IEEE80211_NODE_QOS) &&
1447                             ieee80211_parse_wmeparams(vap, scan.wme, wh) > 0)
1448                                 ieee80211_wme_updateparams(vap);
1449 #ifdef IEEE80211_SUPPORT_SUPERG
1450                         if (scan.ath != NULL)
1451                                 ieee80211_parse_athparams(ni, scan.ath, wh);
1452 #endif
1453                         if (scan.htcap != NULL && scan.htinfo != NULL &&
1454                             (vap->iv_flags_ht & IEEE80211_FHT_HT)) {
1455                                 /* XXX state changes? */
1456                                 ieee80211_ht_updateparams(ni,
1457                                     scan.htcap, scan.htinfo);
1458                                 do_ht = 1;
1459                         }
1460                         if (scan.vhtcap != NULL && scan.vhtopmode != NULL &&
1461                             (vap->iv_flags_vht & IEEE80211_FVHT_VHT)) {
1462                                 /* XXX state changes? */
1463                                 ieee80211_vht_updateparams(ni,
1464                                     scan.vhtcap, scan.vhtopmode);
1465                                 do_ht = 1;
1466                         }
1467                         if (do_ht) {
1468                                 if (ieee80211_ht_updateparams_final(ni,
1469                                     scan.htcap, scan.htinfo))
1470                                         ht_state_change = 1;
1471                         }
1472
1473                         /*
1474                          * If we have a quiet time IE then report it up to
1475                          * the driver.
1476                          *
1477                          * Otherwise, inform the driver that the quiet time
1478                          * IE has disappeared - only do that once rather than
1479                          * spamming it each time.
1480                          */
1481                         if (scan.quiet) {
1482                                 ic->ic_set_quiet(ni, scan.quiet);
1483                                 ni->ni_quiet_ie_set = 1;
1484                                 memcpy(&ni->ni_quiet_ie, scan.quiet,
1485                                     sizeof(struct ieee80211_quiet_ie));
1486                         } else {
1487                                 if (ni->ni_quiet_ie_set == 1)
1488                                         ic->ic_set_quiet(ni, NULL);
1489                                 ni->ni_quiet_ie_set = 0;
1490                                 bzero(&ni->ni_quiet_ie,
1491                                     sizeof(struct ieee80211_quiet_ie));
1492                         }
1493
1494                         if (scan.tim != NULL) {
1495                                 struct ieee80211_tim_ie *tim =
1496                                     (struct ieee80211_tim_ie *) scan.tim;
1497                                 /*
1498                                  * XXX Check/debug this code; see if it's about
1499                                  * the right time to force the VAP awake if we
1500                                  * receive a frame destined for us?
1501                                  */
1502                                 int aid = IEEE80211_AID(ni->ni_associd);
1503                                 int ix = aid / NBBY;
1504                                 int min = tim->tim_bitctl &~ 1;
1505                                 int max = tim->tim_len + min - 4;
1506                                 int tim_ucast = 0, tim_mcast = 0;
1507
1508                                 /*
1509                                  * Only do this for unicast traffic in the TIM
1510                                  * The multicast traffic notification for
1511                                  * the scan notification stuff should occur
1512                                  * differently.
1513                                  */
1514                                 if (min <= ix && ix <= max &&
1515                                      isset(tim->tim_bitmap - min, aid)) {
1516                                         tim_ucast = 1;
1517                                 }
1518
1519                                 /*
1520                                  * Do a separate notification
1521                                  * for the multicast bit being set.
1522                                  */
1523                                 if (tim->tim_bitctl & 1) {
1524                                         tim_mcast = 1;
1525                                 }
1526
1527                                 /*
1528                                  * If the TIM indicates there's traffic for
1529                                  * us then get us out of STA mode powersave.
1530                                  */
1531                                 if (tim_ucast == 1) {
1532
1533                                         /*
1534                                          * Wake us out of SLEEP state if we're
1535                                          * in it; and if we're doing bgscan
1536                                          * then wake us out of STA powersave.
1537                                          */
1538                                         ieee80211_sta_tim_notify(vap, 1);
1539
1540                                         /*
1541                                          * This is preventing us from
1542                                          * continuing a bgscan; because it
1543                                          * tricks the contbgscan()
1544                                          * routine to think there's always
1545                                          * traffic for us.
1546                                          *
1547                                          * I think we need both an RX and
1548                                          * TX ic_lastdata field.
1549                                          */
1550                                         ic->ic_lastdata = ticks;
1551                                 }
1552
1553                                 ni->ni_dtim_count = tim->tim_count;
1554                                 ni->ni_dtim_period = tim->tim_period;
1555                         }
1556                         if (scan.csa != NULL &&
1557                             (vap->iv_flags & IEEE80211_F_DOTH))
1558                                 ieee80211_parse_csaparams(vap, scan.csa, wh);
1559                         else if (ic->ic_flags & IEEE80211_F_CSAPENDING) {
1560                                 /*
1561                                  * No CSA ie or 11h disabled, but a channel
1562                                  * switch is pending; drop out so we aren't
1563                                  * stuck in CSA state.  If the AP really is
1564                                  * moving we'll get a beacon miss and scan.
1565                                  */
1566                                 IEEE80211_LOCK(ic);
1567                                 ieee80211_csa_cancelswitch(ic);
1568                                 IEEE80211_UNLOCK(ic);
1569                         }
1570                         /*
1571                          * If scanning, pass the info to the scan module.
1572                          * Otherwise, check if it's the right time to do
1573                          * a background scan.  Background scanning must
1574                          * be enabled and we must not be operating in the
1575                          * turbo phase of dynamic turbo mode.  Then,
1576                          * it's been a while since the last background
1577                          * scan and if no data frames have come through
1578                          * recently, kick off a scan.  Note that this
1579                          * is the mechanism by which a background scan
1580                          * is started _and_ continued each time we
1581                          * return on-channel to receive a beacon from
1582                          * our ap.
1583                          */
1584                         if (ic->ic_flags & IEEE80211_F_SCAN) {
1585                                 ieee80211_add_scan(vap, rxchan,
1586                                     &scan, wh, subtype, rssi, nf);
1587                         } else if (contbgscan(vap)) {
1588                                 ieee80211_bg_scan(vap, 0);
1589                         } else if (startbgscan(vap)) {
1590                                 vap->iv_stats.is_scan_bg++;
1591 #if 0
1592                                 /* wakeup if we are sleeing */
1593                                 ieee80211_set_pwrsave(vap, 0);
1594 #endif
1595                                 ieee80211_bg_scan(vap, 0);
1596                         }
1597
1598                         /*
1599                          * Put the station to sleep if we haven't seen
1600                          * traffic in a while.
1601                          */
1602                         IEEE80211_LOCK(ic);
1603                         ieee80211_sta_ps_timer_check(vap);
1604                         IEEE80211_UNLOCK(ic);
1605
1606                         /*
1607                          * If we've had a channel width change (eg HT20<->HT40)
1608                          * then schedule a delayed driver notification.
1609                          */
1610                         if (ht_state_change)
1611                                 ieee80211_update_chw(ic);
1612                         return;
1613                 }
1614                 /*
1615                  * If scanning, just pass information to the scan module.
1616                  */
1617                 if (ic->ic_flags & IEEE80211_F_SCAN) {
1618                         if (ic->ic_flags_ext & IEEE80211_FEXT_PROBECHAN) {
1619                                 /*
1620                                  * Actively scanning a channel marked passive;
1621                                  * send a probe request now that we know there
1622                                  * is 802.11 traffic present.
1623                                  *
1624                                  * XXX check if the beacon we recv'd gives
1625                                  * us what we need and suppress the probe req
1626                                  */
1627                                 ieee80211_probe_curchan(vap, 1);
1628                                 ic->ic_flags_ext &= ~IEEE80211_FEXT_PROBECHAN;
1629                         }
1630                         ieee80211_add_scan(vap, rxchan, &scan, wh,
1631                             subtype, rssi, nf);
1632                         return;
1633                 }
1634                 break;
1635         }
1636
1637         case IEEE80211_FC0_SUBTYPE_AUTH: {
1638                 uint16_t algo, seq, status;
1639                 /*
1640                  * auth frame format
1641                  *      [2] algorithm
1642                  *      [2] sequence
1643                  *      [2] status
1644                  *      [tlv*] challenge
1645                  */
1646                 IEEE80211_VERIFY_LENGTH(efrm - frm, 6, return);
1647                 algo   = le16toh(*(uint16_t *)frm);
1648                 seq    = le16toh(*(uint16_t *)(frm + 2));
1649                 status = le16toh(*(uint16_t *)(frm + 4));
1650                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_AUTH, wh->i_addr2,
1651                     "recv auth frame with algorithm %d seq %d", algo, seq);
1652
1653                 if (vap->iv_flags & IEEE80211_F_COUNTERM) {
1654                         IEEE80211_DISCARD(vap,
1655                             IEEE80211_MSG_AUTH | IEEE80211_MSG_CRYPTO,
1656                             wh, "auth", "%s", "TKIP countermeasures enabled");
1657                         vap->iv_stats.is_rx_auth_countermeasures++;
1658                         if (vap->iv_opmode == IEEE80211_M_HOSTAP) {
1659                                 ieee80211_send_error(ni, wh->i_addr2,
1660                                         IEEE80211_FC0_SUBTYPE_AUTH,
1661                                         IEEE80211_REASON_MIC_FAILURE);
1662                         }
1663                         return;
1664                 }
1665                 if (algo == IEEE80211_AUTH_ALG_SHARED)
1666                         sta_auth_shared(ni, wh, frm + 6, efrm, rssi, nf,
1667                             seq, status);
1668                 else if (algo == IEEE80211_AUTH_ALG_OPEN)
1669                         sta_auth_open(ni, wh, rssi, nf, seq, status);
1670                 else {
1671                         IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
1672                             wh, "auth", "unsupported alg %d", algo);
1673                         vap->iv_stats.is_rx_auth_unsupported++;
1674                         return;
1675                 }
1676                 break;
1677         }
1678
1679         case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
1680         case IEEE80211_FC0_SUBTYPE_REASSOC_RESP: {
1681                 uint16_t capinfo, associd;
1682                 uint16_t status;
1683
1684                 if (vap->iv_state != IEEE80211_S_ASSOC) {
1685                         vap->iv_stats.is_rx_mgtdiscard++;
1686                         return;
1687                 }
1688
1689                 /*
1690                  * asresp frame format
1691                  *      [2] capability information
1692                  *      [2] status
1693                  *      [2] association ID
1694                  *      [tlv] supported rates
1695                  *      [tlv] extended supported rates
1696                  *      [tlv] WME
1697                  *      [tlv] HT capabilities
1698                  *      [tlv] HT info
1699                  */
1700                 IEEE80211_VERIFY_LENGTH(efrm - frm, 6, return);
1701                 ni = vap->iv_bss;
1702                 capinfo = le16toh(*(uint16_t *)frm);
1703                 frm += 2;
1704                 status = le16toh(*(uint16_t *)frm);
1705                 frm += 2;
1706                 if (status != 0) {
1707                         IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ASSOC,
1708                             wh->i_addr2, "%sassoc failed (reason %d)",
1709                             ISREASSOC(subtype) ?  "re" : "", status);
1710                         vap->iv_stats.is_rx_auth_fail++;        /* XXX */
1711                         return;
1712                 }
1713                 associd = le16toh(*(uint16_t *)frm);
1714                 frm += 2;
1715
1716                 rates = xrates = wme = htcap = htinfo = NULL;
1717                 vhtcap = vhtopmode = NULL;
1718                 while (efrm - frm > 1) {
1719                         IEEE80211_VERIFY_LENGTH(efrm - frm, frm[1] + 2, return);
1720                         switch (*frm) {
1721                         case IEEE80211_ELEMID_RATES:
1722                                 rates = frm;
1723                                 break;
1724                         case IEEE80211_ELEMID_XRATES:
1725                                 xrates = frm;
1726                                 break;
1727                         case IEEE80211_ELEMID_HTCAP:
1728                                 htcap = frm;
1729                                 break;
1730                         case IEEE80211_ELEMID_HTINFO:
1731                                 htinfo = frm;
1732                                 break;
1733                         case IEEE80211_ELEMID_VENDOR:
1734                                 if (iswmeoui(frm))
1735                                         wme = frm;
1736                                 else if (vap->iv_flags_ht & IEEE80211_FHT_HTCOMPAT) {
1737                                         /*
1738                                          * Accept pre-draft HT ie's if the
1739                                          * standard ones have not been seen.
1740                                          */
1741                                         if (ishtcapoui(frm)) {
1742                                                 if (htcap == NULL)
1743                                                         htcap = frm;
1744                                         } else if (ishtinfooui(frm)) {
1745                                                 if (htinfo == NULL)
1746                                                         htinfo = frm;
1747                                         }
1748                                 }
1749                                 /* XXX Atheros OUI support */
1750                                 break;
1751                         case IEEE80211_ELEMID_VHT_CAP:
1752                                 vhtcap = frm;
1753                                 break;
1754                         case IEEE80211_ELEMID_VHT_OPMODE:
1755                                 vhtopmode = frm;
1756                                 break;
1757                         }
1758                         frm += frm[1] + 2;
1759                 }
1760
1761                 IEEE80211_VERIFY_ELEMENT(rates, IEEE80211_RATE_MAXSIZE, return);
1762                 if (xrates != NULL)
1763                         IEEE80211_VERIFY_ELEMENT(xrates,
1764                                 IEEE80211_RATE_MAXSIZE - rates[1], return);
1765                 rate = ieee80211_setup_rates(ni, rates, xrates,
1766                                 IEEE80211_F_JOIN |
1767                                 IEEE80211_F_DOSORT | IEEE80211_F_DOFRATE |
1768                                 IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
1769                 if (rate & IEEE80211_RATE_BASIC) {
1770                         IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ASSOC,
1771                             wh->i_addr2,
1772                             "%sassoc failed (rate set mismatch)",
1773                             ISREASSOC(subtype) ?  "re" : "");
1774                         vap->iv_stats.is_rx_assoc_norate++;
1775                         ieee80211_new_state(vap, IEEE80211_S_SCAN,
1776                             IEEE80211_SCAN_FAIL_STATUS);
1777                         return;
1778                 }
1779
1780                 ni->ni_capinfo = capinfo;
1781                 ni->ni_associd = associd;
1782                 if (ni->ni_jointime == 0)
1783                         ni->ni_jointime = time_uptime;
1784                 if (wme != NULL &&
1785                     ieee80211_parse_wmeparams(vap, wme, wh) >= 0) {
1786                         ni->ni_flags |= IEEE80211_NODE_QOS;
1787                         ieee80211_wme_updateparams(vap);
1788                 } else
1789                         ni->ni_flags &= ~IEEE80211_NODE_QOS;
1790                 /*
1791                  * Setup HT state according to the negotiation.
1792                  *
1793                  * NB: shouldn't need to check if HT use is enabled but some
1794                  *     ap's send back HT ie's even when we don't indicate we
1795                  *     are HT capable in our AssocReq.
1796                  */
1797                 if (htcap != NULL && htinfo != NULL &&
1798                     (vap->iv_flags_ht & IEEE80211_FHT_HT)) {
1799                         ieee80211_ht_node_init(ni);
1800                         ieee80211_ht_updateparams(ni, htcap, htinfo);
1801
1802                         if ((vhtcap != NULL) && (vhtopmode != NULL) &
1803                             (vap->iv_flags_vht & IEEE80211_FVHT_VHT)) {
1804                                 /*
1805                                  * Log if we get a VHT assoc/reassoc response.
1806                                  * We aren't ready for 2GHz VHT support.
1807                                  */
1808                                 if (IEEE80211_IS_CHAN_2GHZ(ni->ni_chan)) {
1809                                         printf("%s: peer %6D: VHT on 2GHz, ignoring\n",
1810                                             __func__,
1811                                             ni->ni_macaddr,
1812                                             ":");
1813                                 } else {
1814                                         ieee80211_vht_node_init(ni);
1815                                         ieee80211_vht_updateparams(ni, vhtcap, vhtopmode);
1816                                         ieee80211_setup_vht_rates(ni, vhtcap, vhtopmode);
1817                                 }
1818                         }
1819
1820                         ieee80211_ht_updateparams_final(ni, htcap, htinfo);
1821                         ieee80211_setup_htrates(ni, htcap,
1822                              IEEE80211_F_JOIN | IEEE80211_F_DOBRS);
1823                         ieee80211_setup_basic_htrates(ni, htinfo);
1824
1825                         ieee80211_node_setuptxparms(ni);
1826                         ieee80211_ratectl_node_init(ni);
1827                 }
1828
1829                 /*
1830                  * Always initialise FF/superg state; we can use this
1831                  * for doing A-MSDU encapsulation as well.
1832                  */
1833 #ifdef  IEEE80211_SUPPORT_SUPERG
1834                 ieee80211_ff_node_init(ni);
1835 #endif
1836
1837                 /*
1838                  * Configure state now that we are associated.
1839                  *
1840                  * XXX may need different/additional driver callbacks?
1841                  */
1842                 if (IEEE80211_IS_CHAN_A(ic->ic_curchan) ||
1843                     (ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE)) {
1844                         ic->ic_flags |= IEEE80211_F_SHPREAMBLE;
1845                         ic->ic_flags &= ~IEEE80211_F_USEBARKER;
1846                 } else {
1847                         ic->ic_flags &= ~IEEE80211_F_SHPREAMBLE;
1848                         ic->ic_flags |= IEEE80211_F_USEBARKER;
1849                 }
1850                 ieee80211_set_shortslottime(ic,
1851                         IEEE80211_IS_CHAN_A(ic->ic_curchan) ||
1852                         (ni->ni_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME));
1853                 /*
1854                  * Honor ERP protection.
1855                  *
1856                  * NB: ni_erp should zero for non-11g operation.
1857                  */
1858                 if (IEEE80211_IS_CHAN_ANYG(ic->ic_curchan) &&
1859                     (ni->ni_erp & IEEE80211_ERP_USE_PROTECTION))
1860                         ic->ic_flags |= IEEE80211_F_USEPROT;
1861                 else
1862                         ic->ic_flags &= ~IEEE80211_F_USEPROT;
1863                 IEEE80211_NOTE_MAC(vap,
1864                     IEEE80211_MSG_ASSOC | IEEE80211_MSG_DEBUG, wh->i_addr2,
1865                     "%sassoc success at aid %d: %s preamble, %s slot time%s%s%s%s%s%s%s%s",
1866                     ISREASSOC(subtype) ? "re" : "",
1867                     IEEE80211_NODE_AID(ni),
1868                     ic->ic_flags&IEEE80211_F_SHPREAMBLE ? "short" : "long",
1869                     ic->ic_flags&IEEE80211_F_SHSLOT ? "short" : "long",
1870                     ic->ic_flags&IEEE80211_F_USEPROT ? ", protection" : "",
1871                     ni->ni_flags & IEEE80211_NODE_QOS ? ", QoS" : "",
1872                     ni->ni_flags & IEEE80211_NODE_HT ?
1873                         (ni->ni_chw == 40 ? ", HT40" : ", HT20") : "",
1874                     ni->ni_flags & IEEE80211_NODE_AMPDU ? " (+AMPDU)" : "",
1875                     ni->ni_flags & IEEE80211_NODE_MIMO_RTS ? " (+SMPS-DYN)" :
1876                         ni->ni_flags & IEEE80211_NODE_MIMO_PS ? " (+SMPS)" : "",
1877                     ni->ni_flags & IEEE80211_NODE_RIFS ? " (+RIFS)" : "",
1878                     IEEE80211_ATH_CAP(vap, ni, IEEE80211_NODE_FF) ?
1879                         ", fast-frames" : "",
1880                     IEEE80211_ATH_CAP(vap, ni, IEEE80211_NODE_TURBOP) ?
1881                         ", turbo" : ""
1882                 );
1883                 ieee80211_new_state(vap, IEEE80211_S_RUN, subtype);
1884                 break;
1885         }
1886
1887         case IEEE80211_FC0_SUBTYPE_DEAUTH: {
1888                 uint16_t reason;
1889
1890                 if (vap->iv_state == IEEE80211_S_SCAN) {
1891                         vap->iv_stats.is_rx_mgtdiscard++;
1892                         return;
1893                 }
1894                 if (!IEEE80211_ADDR_EQ(wh->i_addr1, vap->iv_myaddr)) {
1895                         /* NB: can happen when in promiscuous mode */
1896                         vap->iv_stats.is_rx_mgtdiscard++;
1897                         break;
1898                 }
1899
1900                 /*
1901                  * deauth frame format
1902                  *      [2] reason
1903                  */
1904                 IEEE80211_VERIFY_LENGTH(efrm - frm, 2, return);
1905                 reason = le16toh(*(uint16_t *)frm);
1906
1907                 vap->iv_stats.is_rx_deauth++;
1908                 vap->iv_stats.is_rx_deauth_code = reason;
1909                 IEEE80211_NODE_STAT(ni, rx_deauth);
1910
1911                 IEEE80211_NOTE(vap, IEEE80211_MSG_AUTH, ni,
1912                     "recv deauthenticate (reason: %d (%s))", reason,
1913                     ieee80211_reason_to_string(reason));
1914                 ieee80211_new_state(vap, IEEE80211_S_AUTH,
1915                     (reason << 8) | IEEE80211_FC0_SUBTYPE_DEAUTH);
1916                 break;
1917         }
1918
1919         case IEEE80211_FC0_SUBTYPE_DISASSOC: {
1920                 uint16_t reason;
1921
1922                 if (vap->iv_state != IEEE80211_S_RUN &&
1923                     vap->iv_state != IEEE80211_S_ASSOC &&
1924                     vap->iv_state != IEEE80211_S_AUTH) {
1925                         vap->iv_stats.is_rx_mgtdiscard++;
1926                         return;
1927                 }
1928                 if (!IEEE80211_ADDR_EQ(wh->i_addr1, vap->iv_myaddr)) {
1929                         /* NB: can happen when in promiscuous mode */
1930                         vap->iv_stats.is_rx_mgtdiscard++;
1931                         break;
1932                 }
1933
1934                 /*
1935                  * disassoc frame format
1936                  *      [2] reason
1937                  */
1938                 IEEE80211_VERIFY_LENGTH(efrm - frm, 2, return);
1939                 reason = le16toh(*(uint16_t *)frm);
1940
1941                 vap->iv_stats.is_rx_disassoc++;
1942                 vap->iv_stats.is_rx_disassoc_code = reason;
1943                 IEEE80211_NODE_STAT(ni, rx_disassoc);
1944
1945                 IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC, ni,
1946                     "recv disassociate (reason: %d (%s))", reason,
1947                     ieee80211_reason_to_string(reason));
1948                 ieee80211_new_state(vap, IEEE80211_S_ASSOC, 0);
1949                 break;
1950         }
1951
1952         case IEEE80211_FC0_SUBTYPE_ACTION:
1953         case IEEE80211_FC0_SUBTYPE_ACTION_NOACK:
1954                 if (!IEEE80211_ADDR_EQ(vap->iv_myaddr, wh->i_addr1) &&
1955                     !IEEE80211_IS_MULTICAST(wh->i_addr1)) {
1956                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
1957                             wh, NULL, "%s", "not for us");
1958                         vap->iv_stats.is_rx_mgtdiscard++;
1959                 } else if (vap->iv_state != IEEE80211_S_RUN) {
1960                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
1961                             wh, NULL, "wrong state %s",
1962                             ieee80211_state_name[vap->iv_state]);
1963                         vap->iv_stats.is_rx_mgtdiscard++;
1964                 } else {
1965                         if (ieee80211_parse_action(ni, m0) == 0)
1966                                 (void)ic->ic_recv_action(ni, wh, frm, efrm);
1967                 }
1968                 break;
1969
1970         case IEEE80211_FC0_SUBTYPE_ASSOC_REQ:
1971         case IEEE80211_FC0_SUBTYPE_REASSOC_REQ:
1972         case IEEE80211_FC0_SUBTYPE_PROBE_REQ:
1973         case IEEE80211_FC0_SUBTYPE_TIMING_ADV:
1974         case IEEE80211_FC0_SUBTYPE_ATIM:
1975                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
1976                     wh, NULL, "%s", "not handled");
1977                 vap->iv_stats.is_rx_mgtdiscard++;
1978                 break;
1979
1980         default:
1981                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
1982                     wh, "mgt", "subtype 0x%x not handled", subtype);
1983                 vap->iv_stats.is_rx_badsubtype++;
1984                 break;
1985         }
1986 #undef ISREASSOC
1987 }
1988
1989 static void
1990 sta_recv_ctl(struct ieee80211_node *ni, struct mbuf *m, int subtype)
1991 {
1992         switch (subtype) {
1993         case IEEE80211_FC0_SUBTYPE_BAR:
1994                 ieee80211_recv_bar(ni, m);
1995                 break;
1996         }
1997 }