sys/net80211/ieee80211_hostap.c

   1 /*-
   2  * Copyright (c) 2007-2008 Sam Leffler, Errno Consulting
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include <sys/cdefs.h>
  27 #ifdef __FreeBSD__
  28 __FBSDID("$FreeBSD$");
  29 #endif
  30
  31 /*
  32  * IEEE 802.11 HOSTAP mode support.
  33  */
  34 #include "opt_inet.h"
  35 #include "opt_wlan.h"
  36
  37 #include <sys/param.h>
  38 #include <sys/systm.h>
  39 #include <sys/mbuf.h>
  40 #include <sys/malloc.h>
  41 #include <sys/kernel.h>
  42
  43 #include <sys/socket.h>
  44 #include <sys/sockio.h>
  45 #include <sys/endian.h>
  46 #include <sys/errno.h>
  47 #include <sys/proc.h>
  48 #include <sys/sysctl.h>
  49
  50 #include <net/if.h>
  51 #include <net/if_media.h>
  52 #include <net/if_llc.h>
  53 #include <net/ethernet.h>
  54
  55 #include <net/bpf.h>
  56
  57 #include <net80211/ieee80211_var.h>
  58 #include <net80211/ieee80211_hostap.h>
  59 #include <net80211/ieee80211_input.h>
  60 #ifdef IEEE80211_SUPPORT_SUPERG
  61 #include <net80211/ieee80211_superg.h>
  62 #endif
  63 #include <net80211/ieee80211_wds.h>
  64
  65 #define IEEE80211_RATE2MBS(r)   (((r) & IEEE80211_RATE_VAL) / 2)
  66
  67 static  void hostap_vattach(struct ieee80211vap *);
  68 static  int hostap_newstate(struct ieee80211vap *, enum ieee80211_state, int);
  69 static  int hostap_input(struct ieee80211_node *ni, struct mbuf *m,
  70             int rssi, int nf);
  71 static void hostap_deliver_data(struct ieee80211vap *,
  72             struct ieee80211_node *, struct mbuf *);
  73 static void hostap_recv_mgmt(struct ieee80211_node *, struct mbuf *,
  74             int subtype, int rssi, int nf);
  75 static void hostap_recv_ctl(struct ieee80211_node *, struct mbuf *, int);
  76 static void hostap_recv_pspoll(struct ieee80211_node *, struct mbuf *);
  77
  78 void
  79 ieee80211_hostap_attach(struct ieee80211com *ic)
  80 {
  81         ic->ic_vattach[IEEE80211_M_HOSTAP] = hostap_vattach;
  82 }
  83
  84 void
  85 ieee80211_hostap_detach(struct ieee80211com *ic)
  86 {
  87 }
  88
  89 static void
  90 hostap_vdetach(struct ieee80211vap *vap)
  91 {
  92 }
  93
  94 static void
  95 hostap_vattach(struct ieee80211vap *vap)
  96 {
  97         vap->iv_newstate = hostap_newstate;
  98         vap->iv_input = hostap_input;
  99         vap->iv_recv_mgmt = hostap_recv_mgmt;
 100         vap->iv_recv_ctl = hostap_recv_ctl;
 101         vap->iv_opdetach = hostap_vdetach;
 102         vap->iv_deliver_data = hostap_deliver_data;
 103 }
 104
 105 static void
 106 sta_disassoc(void *arg, struct ieee80211_node *ni)
 107 {
 108         struct ieee80211vap *vap = arg;
 109
 110         if (ni->ni_vap == vap && ni->ni_associd != 0) {
 111                 IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_DISASSOC,
 112                         IEEE80211_REASON_ASSOC_LEAVE);
 113                 ieee80211_node_leave(ni);
 114         }
 115 }
 116
 117 static void
 118 sta_csa(void *arg, struct ieee80211_node *ni)
 119 {
 120         struct ieee80211vap *vap = arg;
 121
 122         if (ni->ni_vap == vap && ni->ni_associd != 0)
 123                 if (ni->ni_inact > vap->iv_inact_init) {
 124                         ni->ni_inact = vap->iv_inact_init;
 125                         IEEE80211_NOTE(vap, IEEE80211_MSG_INACT, ni,
 126                             "%s: inact %u", __func__, ni->ni_inact);
 127                 }
 128 }
 129
 130 static void
 131 sta_drop(void *arg, struct ieee80211_node *ni)
 132 {
 133         struct ieee80211vap *vap = arg;
 134
 135         if (ni->ni_vap == vap && ni->ni_associd != 0)
 136                 ieee80211_node_leave(ni);
 137 }
 138
 139 /*
 140  * Does a channel change require associated stations to re-associate
 141  * so protocol state is correct.  This is used when doing CSA across
 142  * bands or similar (e.g. HT -> legacy).
 143  */
 144 static int
 145 isbandchange(struct ieee80211com *ic)
 146 {
 147         return ((ic->ic_bsschan->ic_flags ^ ic->ic_csa_newchan->ic_flags) &
 148             (IEEE80211_CHAN_2GHZ | IEEE80211_CHAN_5GHZ | IEEE80211_CHAN_HALF |
 149              IEEE80211_CHAN_QUARTER | IEEE80211_CHAN_HT)) != 0;
 150 }
 151
 152 /*
 153  * IEEE80211_M_HOSTAP vap state machine handler.
 154  */
 155 static int
 156 hostap_newstate(struct ieee80211vap *vap, enum ieee80211_state nstate, int arg)
 157 {
 158         struct ieee80211com *ic = vap->iv_ic;
 159         enum ieee80211_state ostate;
 160
 161         IEEE80211_LOCK_ASSERT(ic);
 162
 163         ostate = vap->iv_state;
 164         IEEE80211_DPRINTF(vap, IEEE80211_MSG_STATE, "%s: %s -> %s (%d)\n",
 165             __func__, ieee80211_state_name[ostate],
 166             ieee80211_state_name[nstate], arg);
 167         vap->iv_state = nstate;                 /* state transition */
 168         if (ostate != IEEE80211_S_SCAN)
 169                 ieee80211_cancel_scan(vap);     /* background scan */
 170         switch (nstate) {
 171         case IEEE80211_S_INIT:
 172                 switch (ostate) {
 173                 case IEEE80211_S_SCAN:
 174                         ieee80211_cancel_scan(vap);
 175                         break;
 176                 case IEEE80211_S_CAC:
 177                         ieee80211_dfs_cac_stop(vap);
 178                         break;
 179                 case IEEE80211_S_RUN:
 180                         ieee80211_iterate_nodes(&ic->ic_sta, sta_disassoc, vap);
 181                         break;
 182                 default:
 183                         break;
 184                 }
 185                 if (ostate != IEEE80211_S_INIT) {
 186                         /* NB: optimize INIT -> INIT case */
 187                         ieee80211_reset_bss(vap);
 188                 }
 189                 if (vap->iv_auth->ia_detach != NULL)
 190                         vap->iv_auth->ia_detach(vap);
 191                 break;
 192         case IEEE80211_S_SCAN:
 193                 switch (ostate) {
 194                 case IEEE80211_S_CSA:
 195                 case IEEE80211_S_RUN:
 196                         ieee80211_iterate_nodes(&ic->ic_sta, sta_disassoc, vap);
 197                         /*
 198                          * Clear overlapping BSS state; the beacon frame
 199                          * will be reconstructed on transition to the RUN
 200                          * state and the timeout routines check if the flag
 201                          * is set before doing anything so this is sufficient.
 202                          */
 203                         ic->ic_flags_ext &= ~IEEE80211_FEXT_NONERP_PR;
 204                         ic->ic_flags_ht &= ~IEEE80211_FHT_NONHT_PR;
 205                         /* fall thru... */
 206                 case IEEE80211_S_CAC:
 207                         /*
 208                          * NB: We may get here because of a manual channel
 209                          *     change in which case we need to stop CAC
 210                          * XXX no need to stop if ostate RUN but it's ok
 211                          */
 212                         ieee80211_dfs_cac_stop(vap);
 213                         /* fall thru... */
 214                 case IEEE80211_S_INIT:
 215                         if (vap->iv_des_chan != IEEE80211_CHAN_ANYC &&
 216                             !IEEE80211_IS_CHAN_RADAR(vap->iv_des_chan)) {
 217                                 /*
 218                                  * Already have a channel; bypass the
 219                                  * scan and startup immediately.
 220                                  * ieee80211_create_ibss will call back to
 221                                  * move us to RUN state.
 222                                  */
 223                                 ieee80211_create_ibss(vap, vap->iv_des_chan);
 224                                 break;
 225                         }
 226                         /*
 227                          * Initiate a scan.  We can come here as a result
 228                          * of an IEEE80211_IOC_SCAN_REQ too in which case
 229                          * the vap will be marked with IEEE80211_FEXT_SCANREQ
 230                          * and the scan request parameters will be present
 231                          * in iv_scanreq.  Otherwise we do the default.
 232                          */
 233                         if (vap->iv_flags_ext & IEEE80211_FEXT_SCANREQ) {
 234                                 ieee80211_check_scan(vap,
 235                                     vap->iv_scanreq_flags,
 236                                     vap->iv_scanreq_duration,
 237                                     vap->iv_scanreq_mindwell,
 238                                     vap->iv_scanreq_maxdwell,
 239                                     vap->iv_scanreq_nssid, vap->iv_scanreq_ssid);
 240                                 vap->iv_flags_ext &= ~IEEE80211_FEXT_SCANREQ;
 241                         } else
 242                                 ieee80211_check_scan_current(vap);
 243                         break;
 244                 case IEEE80211_S_SCAN:
 245                         /*
 246                          * A state change requires a reset; scan.
 247                          */
 248                         ieee80211_check_scan_current(vap);
 249                         break;
 250                 default:
 251                         break;
 252                 }
 253                 break;
 254         case IEEE80211_S_CAC:
 255                 /*
 256                  * Start CAC on a DFS channel.  We come here when starting
 257                  * a bss on a DFS channel (see ieee80211_create_ibss).
 258                  */
 259                 ieee80211_dfs_cac_start(vap);
 260                 break;
 261         case IEEE80211_S_RUN:
 262                 if (vap->iv_flags & IEEE80211_F_WPA) {
 263                         /* XXX validate prerequisites */
 264                 }
 265                 switch (ostate) {
 266                 case IEEE80211_S_INIT:
 267                         /*
 268                          * Already have a channel; bypass the
 269                          * scan and startup immediately.
 270                          * Note that ieee80211_create_ibss will call
 271                          * back to do a RUN->RUN state change.
 272                          */
 273                         ieee80211_create_ibss(vap,
 274                             ieee80211_ht_adjust_channel(ic,
 275                                 ic->ic_curchan, vap->iv_flags_ht));
 276                         /* NB: iv_bss is changed on return */
 277                         break;
 278                 case IEEE80211_S_CAC:
 279                         /*
 280                          * NB: This is the normal state change when CAC
 281                          * expires and no radar was detected; no need to
 282                          * clear the CAC timer as it's already expired.
 283                          */
 284                         /* fall thru... */
 285                 case IEEE80211_S_CSA:
 286                         /*
 287                          * Shorten inactivity timer of associated stations
 288                          * to weed out sta's that don't follow a CSA.
 289                          */
 290                         ieee80211_iterate_nodes(&ic->ic_sta, sta_csa, vap);
 291                         /*
 292                          * Update bss node channel to reflect where
 293                          * we landed after CSA.
 294                          */
 295                         ieee80211_node_set_chan(vap->iv_bss,
 296                             ieee80211_ht_adjust_channel(ic, ic->ic_curchan,
 297                                 ieee80211_htchanflags(vap->iv_bss->ni_chan)));
 298                         /* XXX bypass debug msgs */
 299                         break;
 300                 case IEEE80211_S_SCAN:
 301                 case IEEE80211_S_RUN:
 302 #ifdef IEEE80211_DEBUG
 303                         if (ieee80211_msg_debug(vap)) {
 304                                 struct ieee80211_node *ni = vap->iv_bss;
 305                                 ieee80211_note(vap,
 306                                     "synchronized with %s ssid ",
 307                                     ether_sprintf(ni->ni_bssid));
 308                                 ieee80211_print_essid(ni->ni_essid,
 309                                     ni->ni_esslen);
 310                                 /* XXX MCS/HT */
 311                                 printf(" channel %d start %uMb\n",
 312                                     ieee80211_chan2ieee(ic, ic->ic_curchan),
 313                                     IEEE80211_RATE2MBS(ni->ni_txrate));
 314                         }
 315 #endif
 316                         break;
 317                 default:
 318                         break;
 319                 }
 320                 /*
 321                  * Start/stop the authenticator.  We delay until here
 322                  * to allow configuration to happen out of order.
 323                  */
 324                 if (vap->iv_auth->ia_attach != NULL) {
 325                         /* XXX check failure */
 326                         vap->iv_auth->ia_attach(vap);
 327                 } else if (vap->iv_auth->ia_detach != NULL) {
 328                         vap->iv_auth->ia_detach(vap);
 329                 }
 330                 ieee80211_node_authorize(vap->iv_bss);
 331                 break;
 332         case IEEE80211_S_CSA:
 333                 if (ostate == IEEE80211_S_RUN && isbandchange(ic)) {
 334                         /*
 335                          * On a ``band change'' silently drop associated
 336                          * stations as they must re-associate before they
 337                          * can pass traffic (as otherwise protocol state
 338                          * such as capabilities and the negotiated rate
 339                          * set may/will be wrong).
 340                          */
 341                         ieee80211_iterate_nodes(&ic->ic_sta, sta_drop, vap);
 342                 }
 343                 break;
 344         default:
 345                 break;
 346         }
 347         return 0;
 348 }
 349
 350 static void
 351 hostap_deliver_data(struct ieee80211vap *vap,
 352         struct ieee80211_node *ni, struct mbuf *m)
 353 {
 354         struct ether_header *eh = mtod(m, struct ether_header *);
 355         struct ifnet *ifp = vap->iv_ifp;
 356
 357         /* clear driver/net80211 flags before passing up */
 358         m->m_flags &= ~(M_80211_RX | M_MCAST | M_BCAST);
 359
 360         KASSERT(vap->iv_opmode == IEEE80211_M_HOSTAP,
 361             ("gack, opmode %d", vap->iv_opmode));
 362         /*
 363          * Do accounting.
 364          */
 365         ifp->if_ipackets++;
 366         IEEE80211_NODE_STAT(ni, rx_data);
 367         IEEE80211_NODE_STAT_ADD(ni, rx_bytes, m->m_pkthdr.len);
 368         if (ETHER_IS_MULTICAST(eh->ether_dhost)) {
 369                 m->m_flags |= M_MCAST;          /* XXX M_BCAST? */
 370                 IEEE80211_NODE_STAT(ni, rx_mcast);
 371         } else
 372                 IEEE80211_NODE_STAT(ni, rx_ucast);
 373
 374         /* perform as a bridge within the AP */
 375         if ((vap->iv_flags & IEEE80211_F_NOBRIDGE) == 0) {
 376                 struct mbuf *mcopy = NULL;
 377
 378                 if (m->m_flags & M_MCAST) {
 379                         mcopy = m_dup(m, M_DONTWAIT);
 380                         if (mcopy == NULL)
 381                                 ifp->if_oerrors++;
 382                         else
 383                                 mcopy->m_flags |= M_MCAST;
 384                 } else {
 385                         /*
 386                          * Check if the destination is associated with the
 387                          * same vap and authorized to receive traffic.
 388                          * Beware of traffic destined for the vap itself;
 389                          * sending it will not work; just let it be delivered
 390                          * normally.
 391                          */
 392                         struct ieee80211_node *sta = ieee80211_find_vap_node(
 393                              &vap->iv_ic->ic_sta, vap, eh->ether_dhost);
 394                         if (sta != NULL) {
 395                                 if (ieee80211_node_is_authorized(sta)) {
 396                                         /*
 397                                          * Beware of sending to ourself; this
 398                                          * needs to happen via the normal
 399                                          * input path.
 400                                          */
 401                                         if (sta != vap->iv_bss) {
 402                                                 mcopy = m;
 403                                                 m = NULL;
 404                                         }
 405                                 } else {
 406                                         vap->iv_stats.is_rx_unauth++;
 407                                         IEEE80211_NODE_STAT(sta, rx_unauth);
 408                                 }
 409                                 ieee80211_free_node(sta);
 410                         }
 411                 }
 412                 if (mcopy != NULL) {
 413                         int len, err;
 414                         len = mcopy->m_pkthdr.len;
 415                         err = ifp->if_transmit(ifp, mcopy);
 416                         if (err) {
 417                                 /* NB: IFQ_HANDOFF reclaims mcopy */
 418                         } else {
 419                                 ifp->if_opackets++;
 420                         }
 421                 }
 422         }
 423         if (m != NULL) {
 424                 /*
 425                  * Mark frame as coming from vap's interface.
 426                  */
 427                 m->m_pkthdr.rcvif = ifp;
 428                 if (m->m_flags & M_MCAST) {
 429                         /*
 430                          * Spam DWDS vap's w/ multicast traffic.
 431                          */
 432                         /* XXX only if dwds in use? */
 433                         ieee80211_dwds_mcast(vap, m);
 434                 }
 435                 if (ni->ni_vlan != 0) {
 436                         /* attach vlan tag */
 437                         m->m_pkthdr.ether_vtag = ni->ni_vlan;
 438                         m->m_flags |= M_VLANTAG;
 439                 }
 440                 ifp->if_input(ifp, m);
 441         }
 442 }
 443
 444 /*
 445  * Decide if a received management frame should be
 446  * printed when debugging is enabled.  This filters some
 447  * of the less interesting frames that come frequently
 448  * (e.g. beacons).
 449  */
 450 static __inline int
 451 doprint(struct ieee80211vap *vap, int subtype)
 452 {
 453         switch (subtype) {
 454         case IEEE80211_FC0_SUBTYPE_BEACON:
 455                 return (vap->iv_ic->ic_flags & IEEE80211_F_SCAN);
 456         case IEEE80211_FC0_SUBTYPE_PROBE_REQ:
 457                 return 0;
 458         }
 459         return 1;
 460 }
 461
 462 /*
 463  * Process a received frame.  The node associated with the sender
 464  * should be supplied.  If nothing was found in the node table then
 465  * the caller is assumed to supply a reference to iv_bss instead.
 466  * The RSSI and a timestamp are also supplied.  The RSSI data is used
 467  * during AP scanning to select a AP to associate with; it can have
 468  * any units so long as values have consistent units and higher values
 469  * mean ``better signal''.  The receive timestamp is currently not used
 470  * by the 802.11 layer.
 471  */
 472 static int
 473 hostap_input(struct ieee80211_node *ni, struct mbuf *m, int rssi, int nf)
 474 {
 475 #define HAS_SEQ(type)   ((type & 0x4) == 0)
 476         struct ieee80211vap *vap = ni->ni_vap;
 477         struct ieee80211com *ic = ni->ni_ic;
 478         struct ifnet *ifp = vap->iv_ifp;
 479         struct ieee80211_frame *wh;
 480         struct ieee80211_key *key;
 481         struct ether_header *eh;
 482         int hdrspace, need_tap = 1;     /* mbuf need to be tapped. */
 483         uint8_t dir, type, subtype, qos;
 484         uint8_t *bssid;
 485         uint16_t rxseq;
 486
 487         if (m->m_flags & M_AMPDU_MPDU) {
 488                 /*
 489                  * Fastpath for A-MPDU reorder q resubmission.  Frames
 490                  * w/ M_AMPDU_MPDU marked have already passed through
 491                  * here but were received out of order and been held on
 492                  * the reorder queue.  When resubmitted they are marked
 493                  * with the M_AMPDU_MPDU flag and we can bypass most of
 494                  * the normal processing.
 495                  */
 496                 wh = mtod(m, struct ieee80211_frame *);
 497                 type = IEEE80211_FC0_TYPE_DATA;
 498                 dir = wh->i_fc[1] & IEEE80211_FC1_DIR_MASK;
 499                 subtype = IEEE80211_FC0_SUBTYPE_QOS;
 500                 hdrspace = ieee80211_hdrspace(ic, wh);  /* XXX optimize? */
 501                 goto resubmit_ampdu;
 502         }
 503
 504         KASSERT(ni != NULL, ("null node"));
 505         ni->ni_inact = ni->ni_inact_reload;
 506
 507         type = -1;                      /* undefined */
 508
 509         if (m->m_pkthdr.len < sizeof(struct ieee80211_frame_min)) {
 510                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 511                     ni->ni_macaddr, NULL,
 512                     "too short (1): len %u", m->m_pkthdr.len);
 513                 vap->iv_stats.is_rx_tooshort++;
 514                 goto out;
 515         }
 516         /*
 517          * Bit of a cheat here, we use a pointer for a 3-address
 518          * frame format but don't reference fields past outside
 519          * ieee80211_frame_min w/o first validating the data is
 520          * present.
 521          */
 522         wh = mtod(m, struct ieee80211_frame *);
 523
 524         if ((wh->i_fc[0] & IEEE80211_FC0_VERSION_MASK) !=
 525             IEEE80211_FC0_VERSION_0) {
 526                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 527                     ni->ni_macaddr, NULL, "wrong version, fc %02x:%02x",
 528                     wh->i_fc[0], wh->i_fc[1]);
 529                 vap->iv_stats.is_rx_badversion++;
 530                 goto err;
 531         }
 532
 533         dir = wh->i_fc[1] & IEEE80211_FC1_DIR_MASK;
 534         type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;
 535         subtype = wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK;
 536         if ((ic->ic_flags & IEEE80211_F_SCAN) == 0) {
 537                 if (dir != IEEE80211_FC1_DIR_NODS)
 538                         bssid = wh->i_addr1;
 539                 else if (type == IEEE80211_FC0_TYPE_CTL)
 540                         bssid = wh->i_addr1;
 541                 else {
 542                         if (m->m_pkthdr.len < sizeof(struct ieee80211_frame)) {
 543                                 IEEE80211_DISCARD_MAC(vap,
 544                                     IEEE80211_MSG_ANY, ni->ni_macaddr,
 545                                     NULL, "too short (2): len %u",
 546                                     m->m_pkthdr.len);
 547                                 vap->iv_stats.is_rx_tooshort++;
 548                                 goto out;
 549                         }
 550                         bssid = wh->i_addr3;
 551                 }
 552                 /*
 553                  * Validate the bssid.
 554                  */
 555                 if (!(type == IEEE80211_FC0_TYPE_MGT &&
 556                       subtype == IEEE80211_FC0_SUBTYPE_BEACON) &&
 557                     !IEEE80211_ADDR_EQ(bssid, vap->iv_bss->ni_bssid) &&
 558                     !IEEE80211_ADDR_EQ(bssid, ifp->if_broadcastaddr)) {
 559                         /* not interested in */
 560                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 561                             bssid, NULL, "%s", "not to bss");
 562                         vap->iv_stats.is_rx_wrongbss++;
 563                         goto out;
 564                 }
 565
 566                 IEEE80211_RSSI_LPF(ni->ni_avgrssi, rssi);
 567                 ni->ni_noise = nf;
 568                 if (HAS_SEQ(type)) {
 569                         uint8_t tid = ieee80211_gettid(wh);
 570                         if (IEEE80211_QOS_HAS_SEQ(wh) &&
 571                             TID_TO_WME_AC(tid) >= WME_AC_VI)
 572                                 ic->ic_wme.wme_hipri_traffic++;
 573                         rxseq = le16toh(*(uint16_t *)wh->i_seq);
 574                         if (! ieee80211_check_rxseq(ni, wh)) {
 575                                 /* duplicate, discard */
 576                                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 577                                     bssid, "duplicate",
 578                                     "seqno <%u,%u> fragno <%u,%u> tid %u",
 579                                     rxseq >> IEEE80211_SEQ_SEQ_SHIFT,
 580                                     ni->ni_rxseqs[tid] >>
 581                                         IEEE80211_SEQ_SEQ_SHIFT,
 582                                     rxseq & IEEE80211_SEQ_FRAG_MASK,
 583                                     ni->ni_rxseqs[tid] &
 584                                         IEEE80211_SEQ_FRAG_MASK,
 585                                     tid);
 586                                 vap->iv_stats.is_rx_dup++;
 587                                 IEEE80211_NODE_STAT(ni, rx_dup);
 588                                 goto out;
 589                         }
 590                         ni->ni_rxseqs[tid] = rxseq;
 591                 }
 592         }
 593
 594         switch (type) {
 595         case IEEE80211_FC0_TYPE_DATA:
 596                 hdrspace = ieee80211_hdrspace(ic, wh);
 597                 if (m->m_len < hdrspace &&
 598                     (m = m_pullup(m, hdrspace)) == NULL) {
 599                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 600                             ni->ni_macaddr, NULL,
 601                             "data too short: expecting %u", hdrspace);
 602                         vap->iv_stats.is_rx_tooshort++;
 603                         goto out;               /* XXX */
 604                 }
 605                 if (!(dir == IEEE80211_FC1_DIR_TODS ||
 606                      (dir == IEEE80211_FC1_DIR_DSTODS &&
 607                       (vap->iv_flags & IEEE80211_F_DWDS)))) {
 608                         if (dir != IEEE80211_FC1_DIR_DSTODS) {
 609                                 IEEE80211_DISCARD(vap,
 610                                     IEEE80211_MSG_INPUT, wh, "data",
 611                                     "incorrect dir 0x%x", dir);
 612                         } else {
 613                                 IEEE80211_DISCARD(vap,
 614                                     IEEE80211_MSG_INPUT |
 615                                     IEEE80211_MSG_WDS, wh,
 616                                     "4-address data",
 617                                     "%s", "DWDS not enabled");
 618                         }
 619                         vap->iv_stats.is_rx_wrongdir++;
 620                         goto out;
 621                 }
 622                 /* check if source STA is associated */
 623                 if (ni == vap->iv_bss) {
 624                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 625                             wh, "data", "%s", "unknown src");
 626                         ieee80211_send_error(ni, wh->i_addr2,
 627                             IEEE80211_FC0_SUBTYPE_DEAUTH,
 628                             IEEE80211_REASON_NOT_AUTHED);
 629                         vap->iv_stats.is_rx_notassoc++;
 630                         goto err;
 631                 }
 632                 if (ni->ni_associd == 0) {
 633                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 634                             wh, "data", "%s", "unassoc src");
 635                         IEEE80211_SEND_MGMT(ni,
 636                             IEEE80211_FC0_SUBTYPE_DISASSOC,
 637                             IEEE80211_REASON_NOT_ASSOCED);
 638                         vap->iv_stats.is_rx_notassoc++;
 639                         goto err;
 640                 }
 641
 642                 /*
 643                  * Check for power save state change.
 644                  * XXX out-of-order A-MPDU frames?
 645                  */
 646                 if (((wh->i_fc[1] & IEEE80211_FC1_PWR_MGT) ^
 647                     (ni->ni_flags & IEEE80211_NODE_PWR_MGT)))
 648                         ieee80211_node_pwrsave(ni,
 649                                 wh->i_fc[1] & IEEE80211_FC1_PWR_MGT);
 650                 /*
 651                  * For 4-address packets handle WDS discovery
 652                  * notifications.  Once a WDS link is setup frames
 653                  * are just delivered to the WDS vap (see below).
 654                  */
 655                 if (dir == IEEE80211_FC1_DIR_DSTODS && ni->ni_wdsvap == NULL) {
 656                         if (!ieee80211_node_is_authorized(ni)) {
 657                                 IEEE80211_DISCARD(vap,
 658                                     IEEE80211_MSG_INPUT |
 659                                     IEEE80211_MSG_WDS, wh,
 660                                     "4-address data",
 661                                     "%s", "unauthorized port");
 662                                 vap->iv_stats.is_rx_unauth++;
 663                                 IEEE80211_NODE_STAT(ni, rx_unauth);
 664                                 goto err;
 665                         }
 666                         ieee80211_dwds_discover(ni, m);
 667                         return type;
 668                 }
 669
 670                 /*
 671                  * Handle A-MPDU re-ordering.  If the frame is to be
 672                  * processed directly then ieee80211_ampdu_reorder
 673                  * will return 0; otherwise it has consumed the mbuf
 674                  * and we should do nothing more with it.
 675                  */
 676                 if ((m->m_flags & M_AMPDU) &&
 677                     ieee80211_ampdu_reorder(ni, m) != 0) {
 678                         m = NULL;
 679                         goto out;
 680                 }
 681         resubmit_ampdu:
 682
 683                 /*
 684                  * Handle privacy requirements.  Note that we
 685                  * must not be preempted from here until after
 686                  * we (potentially) call ieee80211_crypto_demic;
 687                  * otherwise we may violate assumptions in the
 688                  * crypto cipher modules used to do delayed update
 689                  * of replay sequence numbers.
 690                  */
 691                 if (wh->i_fc[1] & IEEE80211_FC1_WEP) {
 692                         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
 693                                 /*
 694                                  * Discard encrypted frames when privacy is off.
 695                                  */
 696                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 697                                     wh, "WEP", "%s", "PRIVACY off");
 698                                 vap->iv_stats.is_rx_noprivacy++;
 699                                 IEEE80211_NODE_STAT(ni, rx_noprivacy);
 700                                 goto out;
 701                         }
 702                         key = ieee80211_crypto_decap(ni, m, hdrspace);
 703                         if (key == NULL) {
 704                                 /* NB: stats+msgs handled in crypto_decap */
 705                                 IEEE80211_NODE_STAT(ni, rx_wepfail);
 706                                 goto out;
 707                         }
 708                         wh = mtod(m, struct ieee80211_frame *);
 709                         wh->i_fc[1] &= ~IEEE80211_FC1_WEP;
 710                 } else {
 711                         /* XXX M_WEP and IEEE80211_F_PRIVACY */
 712                         key = NULL;
 713                 }
 714
 715                 /*
 716                  * Save QoS bits for use below--before we strip the header.
 717                  */
 718                 if (subtype == IEEE80211_FC0_SUBTYPE_QOS) {
 719                         qos = (dir == IEEE80211_FC1_DIR_DSTODS) ?
 720                             ((struct ieee80211_qosframe_addr4 *)wh)->i_qos[0] :
 721                             ((struct ieee80211_qosframe *)wh)->i_qos[0];
 722                 } else
 723                         qos = 0;
 724
 725                 /*
 726                  * Next up, any fragmentation.
 727                  */
 728                 if (!IEEE80211_IS_MULTICAST(wh->i_addr1)) {
 729                         m = ieee80211_defrag(ni, m, hdrspace);
 730                         if (m == NULL) {
 731                                 /* Fragment dropped or frame not complete yet */
 732                                 goto out;
 733                         }
 734                 }
 735                 wh = NULL;              /* no longer valid, catch any uses */
 736
 737                 /*
 738                  * Next strip any MSDU crypto bits.
 739                  */
 740                 if (key != NULL && !ieee80211_crypto_demic(vap, key, m, 0)) {
 741                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 742                             ni->ni_macaddr, "data", "%s", "demic error");
 743                         vap->iv_stats.is_rx_demicfail++;
 744                         IEEE80211_NODE_STAT(ni, rx_demicfail);
 745                         goto out;
 746                 }
 747                 /* copy to listener after decrypt */
 748                 if (ieee80211_radiotap_active_vap(vap))
 749                         ieee80211_radiotap_rx(vap, m);
 750                 need_tap = 0;
 751                 /*
 752                  * Finally, strip the 802.11 header.
 753                  */
 754                 m = ieee80211_decap(vap, m, hdrspace);
 755                 if (m == NULL) {
 756                         /* XXX mask bit to check for both */
 757                         /* don't count Null data frames as errors */
 758                         if (subtype == IEEE80211_FC0_SUBTYPE_NODATA ||
 759                             subtype == IEEE80211_FC0_SUBTYPE_QOS_NULL)
 760                                 goto out;
 761                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 762                             ni->ni_macaddr, "data", "%s", "decap error");
 763                         vap->iv_stats.is_rx_decap++;
 764                         IEEE80211_NODE_STAT(ni, rx_decap);
 765                         goto err;
 766                 }
 767                 eh = mtod(m, struct ether_header *);
 768                 if (!ieee80211_node_is_authorized(ni)) {
 769                         /*
 770                          * Deny any non-PAE frames received prior to
 771                          * authorization.  For open/shared-key
 772                          * authentication the port is mark authorized
 773                          * after authentication completes.  For 802.1x
 774                          * the port is not marked authorized by the
 775                          * authenticator until the handshake has completed.
 776                          */
 777                         if (eh->ether_type != htons(ETHERTYPE_PAE)) {
 778                                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_INPUT,
 779                                     eh->ether_shost, "data",
 780                                     "unauthorized port: ether type 0x%x len %u",
 781                                     eh->ether_type, m->m_pkthdr.len);
 782                                 vap->iv_stats.is_rx_unauth++;
 783                                 IEEE80211_NODE_STAT(ni, rx_unauth);
 784                                 goto err;
 785                         }
 786                 } else {
 787                         /*
 788                          * When denying unencrypted frames, discard
 789                          * any non-PAE frames received without encryption.
 790                          */
 791                         if ((vap->iv_flags & IEEE80211_F_DROPUNENC) &&
 792                             (key == NULL && (m->m_flags & M_WEP) == 0) &&
 793                             eh->ether_type != htons(ETHERTYPE_PAE)) {
 794                                 /*
 795                                  * Drop unencrypted frames.
 796                                  */
 797                                 vap->iv_stats.is_rx_unencrypted++;
 798                                 IEEE80211_NODE_STAT(ni, rx_unencrypted);
 799                                 goto out;
 800                         }
 801                 }
 802                 /* XXX require HT? */
 803                 if (qos & IEEE80211_QOS_AMSDU) {
 804                         m = ieee80211_decap_amsdu(ni, m);
 805                         if (m == NULL)
 806                                 return IEEE80211_FC0_TYPE_DATA;
 807                 } else {
 808 #ifdef IEEE80211_SUPPORT_SUPERG
 809                         m = ieee80211_decap_fastframe(vap, ni, m);
 810                         if (m == NULL)
 811                                 return IEEE80211_FC0_TYPE_DATA;
 812 #endif
 813                 }
 814                 if (dir == IEEE80211_FC1_DIR_DSTODS && ni->ni_wdsvap != NULL)
 815                         ieee80211_deliver_data(ni->ni_wdsvap, ni, m);
 816                 else
 817                         hostap_deliver_data(vap, ni, m);
 818                 return IEEE80211_FC0_TYPE_DATA;
 819
 820         case IEEE80211_FC0_TYPE_MGT:
 821                 vap->iv_stats.is_rx_mgmt++;
 822                 IEEE80211_NODE_STAT(ni, rx_mgmt);
 823                 if (dir != IEEE80211_FC1_DIR_NODS) {
 824                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 825                             wh, "mgt", "incorrect dir 0x%x", dir);
 826                         vap->iv_stats.is_rx_wrongdir++;
 827                         goto err;
 828                 }
 829                 if (m->m_pkthdr.len < sizeof(struct ieee80211_frame)) {
 830                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_ANY,
 831                             ni->ni_macaddr, "mgt", "too short: len %u",
 832                             m->m_pkthdr.len);
 833                         vap->iv_stats.is_rx_tooshort++;
 834                         goto out;
 835                 }
 836                 if (IEEE80211_IS_MULTICAST(wh->i_addr2)) {
 837                         /* ensure return frames are unicast */
 838                         IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
 839                             wh, NULL, "source is multicast: %s",
 840                             ether_sprintf(wh->i_addr2));
 841                         vap->iv_stats.is_rx_mgtdiscard++;       /* XXX stat */
 842                         goto out;
 843                 }
 844 #ifdef IEEE80211_DEBUG
 845                 if ((ieee80211_msg_debug(vap) && doprint(vap, subtype)) ||
 846                     ieee80211_msg_dumppkts(vap)) {
 847                         if_printf(ifp, "received %s from %s rssi %d\n",
 848                             ieee80211_mgt_subtype_name[subtype >>
 849                                 IEEE80211_FC0_SUBTYPE_SHIFT],
 850                             ether_sprintf(wh->i_addr2), rssi);
 851                 }
 852 #endif
 853                 if (wh->i_fc[1] & IEEE80211_FC1_WEP) {
 854                         if (subtype != IEEE80211_FC0_SUBTYPE_AUTH) {
 855                                 /*
 856                                  * Only shared key auth frames with a challenge
 857                                  * should be encrypted, discard all others.
 858                                  */
 859                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 860                                     wh, NULL,
 861                                     "%s", "WEP set but not permitted");
 862                                 vap->iv_stats.is_rx_mgtdiscard++; /* XXX */
 863                                 goto out;
 864                         }
 865                         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
 866                                 /*
 867                                  * Discard encrypted frames when privacy is off.
 868                                  */
 869                                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
 870                                     wh, NULL, "%s", "WEP set but PRIVACY off");
 871                                 vap->iv_stats.is_rx_noprivacy++;
 872                                 goto out;
 873                         }
 874                         hdrspace = ieee80211_hdrspace(ic, wh);
 875                         key = ieee80211_crypto_decap(ni, m, hdrspace);
 876                         if (key == NULL) {
 877                                 /* NB: stats+msgs handled in crypto_decap */
 878                                 goto out;
 879                         }
 880                         wh = mtod(m, struct ieee80211_frame *);
 881                         wh->i_fc[1] &= ~IEEE80211_FC1_WEP;
 882                 }
 883                 /*
 884                  * Pass the packet to radiotap before calling iv_recv_mgmt().
 885                  * Otherwise iv_recv_mgmt() might pass another packet to
 886                  * radiotap, resulting in out of order packet captures.
 887                  */
 888                 if (ieee80211_radiotap_active_vap(vap))
 889                         ieee80211_radiotap_rx(vap, m);
 890                 need_tap = 0;
 891                 vap->iv_recv_mgmt(ni, m, subtype, rssi, nf);
 892                 goto out;
 893
 894         case IEEE80211_FC0_TYPE_CTL:
 895                 vap->iv_stats.is_rx_ctl++;
 896                 IEEE80211_NODE_STAT(ni, rx_ctrl);
 897                 vap->iv_recv_ctl(ni, m, subtype);
 898                 goto out;
 899         default:
 900                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
 901                     wh, "bad", "frame type 0x%x", type);
 902                 /* should not come here */
 903                 break;
 904         }
 905 err:
 906         ifp->if_ierrors++;
 907 out:
 908         if (m != NULL) {
 909                 if (need_tap && ieee80211_radiotap_active_vap(vap))
 910                         ieee80211_radiotap_rx(vap, m);
 911                 m_freem(m);
 912         }
 913         return type;
 914 }
 915
 916 static void
 917 hostap_auth_open(struct ieee80211_node *ni, struct ieee80211_frame *wh,
 918     int rssi, int nf, uint16_t seq, uint16_t status)
 919 {
 920         struct ieee80211vap *vap = ni->ni_vap;
 921
 922         KASSERT(vap->iv_state == IEEE80211_S_RUN, ("state %d", vap->iv_state));
 923
 924         if (ni->ni_authmode == IEEE80211_AUTH_SHARED) {
 925                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
 926                     ni->ni_macaddr, "open auth",
 927                     "bad sta auth mode %u", ni->ni_authmode);
 928                 vap->iv_stats.is_rx_bad_auth++; /* XXX */
 929                 /*
 930                  * Clear any challenge text that may be there if
 931                  * a previous shared key auth failed and then an
 932                  * open auth is attempted.
 933                  */
 934                 if (ni->ni_challenge != NULL) {
 935                         free(ni->ni_challenge, M_80211_NODE);
 936                         ni->ni_challenge = NULL;
 937                 }
 938                 /* XXX hack to workaround calling convention */
 939                 ieee80211_send_error(ni, wh->i_addr2,
 940                     IEEE80211_FC0_SUBTYPE_AUTH,
 941                     (seq + 1) | (IEEE80211_STATUS_ALG<<16));
 942                 return;
 943         }
 944         if (seq != IEEE80211_AUTH_OPEN_REQUEST) {
 945                 vap->iv_stats.is_rx_bad_auth++;
 946                 return;
 947         }
 948         /* always accept open authentication requests */
 949         if (ni == vap->iv_bss) {
 950                 ni = ieee80211_dup_bss(vap, wh->i_addr2);
 951                 if (ni == NULL)
 952                         return;
 953         } else if ((ni->ni_flags & IEEE80211_NODE_AREF) == 0)
 954                 (void) ieee80211_ref_node(ni);
 955         /*
 956          * Mark the node as referenced to reflect that it's
 957          * reference count has been bumped to insure it remains
 958          * after the transaction completes.
 959          */
 960         ni->ni_flags |= IEEE80211_NODE_AREF;
 961         /*
 962          * Mark the node as requiring a valid association id
 963          * before outbound traffic is permitted.
 964          */
 965         ni->ni_flags |= IEEE80211_NODE_ASSOCID;
 966
 967         if (vap->iv_acl != NULL &&
 968             vap->iv_acl->iac_getpolicy(vap) == IEEE80211_MACCMD_POLICY_RADIUS) {
 969                 /*
 970                  * When the ACL policy is set to RADIUS we defer the
 971                  * authorization to a user agent.  Dispatch an event,
 972                  * a subsequent MLME call will decide the fate of the
 973                  * station.  If the user agent is not present then the
 974                  * node will be reclaimed due to inactivity.
 975                  */
 976                 IEEE80211_NOTE_MAC(vap,
 977                     IEEE80211_MSG_AUTH | IEEE80211_MSG_ACL, ni->ni_macaddr,
 978                     "%s", "station authentication defered (radius acl)");
 979                 ieee80211_notify_node_auth(ni);
 980         } else {
 981                 IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_AUTH, seq + 1);
 982                 IEEE80211_NOTE_MAC(vap,
 983                     IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH, ni->ni_macaddr,
 984                     "%s", "station authenticated (open)");
 985                 /*
 986                  * When 802.1x is not in use mark the port
 987                  * authorized at this point so traffic can flow.
 988                  */
 989                 if (ni->ni_authmode != IEEE80211_AUTH_8021X)
 990                         ieee80211_node_authorize(ni);
 991         }
 992 }
 993
 994 static void
 995 hostap_auth_shared(struct ieee80211_node *ni, struct ieee80211_frame *wh,
 996     uint8_t *frm, uint8_t *efrm, int rssi, int nf,
 997     uint16_t seq, uint16_t status)
 998 {
 999         struct ieee80211vap *vap = ni->ni_vap;
1000         uint8_t *challenge;
1001         int allocbs, estatus;
1002
1003         KASSERT(vap->iv_state == IEEE80211_S_RUN, ("state %d", vap->iv_state));
1004
1005         /*
1006          * NB: this can happen as we allow pre-shared key
1007          * authentication to be enabled w/o wep being turned
1008          * on so that configuration of these can be done
1009          * in any order.  It may be better to enforce the
1010          * ordering in which case this check would just be
1011          * for sanity/consistency.
1012          */
1013         if ((vap->iv_flags & IEEE80211_F_PRIVACY) == 0) {
1014                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1015                     ni->ni_macaddr, "shared key auth",
1016                     "%s", " PRIVACY is disabled");
1017                 estatus = IEEE80211_STATUS_ALG;
1018                 goto bad;
1019         }
1020         /*
1021          * Pre-shared key authentication is evil; accept
1022          * it only if explicitly configured (it is supported
1023          * mainly for compatibility with clients like Mac OS X).
1024          */
1025         if (ni->ni_authmode != IEEE80211_AUTH_AUTO &&
1026             ni->ni_authmode != IEEE80211_AUTH_SHARED) {
1027                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1028                     ni->ni_macaddr, "shared key auth",
1029                     "bad sta auth mode %u", ni->ni_authmode);
1030                 vap->iv_stats.is_rx_bad_auth++; /* XXX maybe a unique error? */
1031                 estatus = IEEE80211_STATUS_ALG;
1032                 goto bad;
1033         }
1034
1035         challenge = NULL;
1036         if (frm + 1 < efrm) {
1037                 if ((frm[1] + 2) > (efrm - frm)) {
1038                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1039                             ni->ni_macaddr, "shared key auth",
1040                             "ie %d/%d too long",
1041                             frm[0], (frm[1] + 2) - (efrm - frm));
1042                         vap->iv_stats.is_rx_bad_auth++;
1043                         estatus = IEEE80211_STATUS_CHALLENGE;
1044                         goto bad;
1045                 }
1046                 if (*frm == IEEE80211_ELEMID_CHALLENGE)
1047                         challenge = frm;
1048                 frm += frm[1] + 2;
1049         }
1050         switch (seq) {
1051         case IEEE80211_AUTH_SHARED_CHALLENGE:
1052         case IEEE80211_AUTH_SHARED_RESPONSE:
1053                 if (challenge == NULL) {
1054                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1055                             ni->ni_macaddr, "shared key auth",
1056                             "%s", "no challenge");
1057                         vap->iv_stats.is_rx_bad_auth++;
1058                         estatus = IEEE80211_STATUS_CHALLENGE;
1059                         goto bad;
1060                 }
1061                 if (challenge[1] != IEEE80211_CHALLENGE_LEN) {
1062                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1063                             ni->ni_macaddr, "shared key auth",
1064                             "bad challenge len %d", challenge[1]);
1065                         vap->iv_stats.is_rx_bad_auth++;
1066                         estatus = IEEE80211_STATUS_CHALLENGE;
1067                         goto bad;
1068                 }
1069         default:
1070                 break;
1071         }
1072         switch (seq) {
1073         case IEEE80211_AUTH_SHARED_REQUEST:
1074                 if (ni == vap->iv_bss) {
1075                         ni = ieee80211_dup_bss(vap, wh->i_addr2);
1076                         if (ni == NULL) {
1077                                 /* NB: no way to return an error */
1078                                 return;
1079                         }
1080                         allocbs = 1;
1081                 } else {
1082                         if ((ni->ni_flags & IEEE80211_NODE_AREF) == 0)
1083                                 (void) ieee80211_ref_node(ni);
1084                         allocbs = 0;
1085                 }
1086                 /*
1087                  * Mark the node as referenced to reflect that it's
1088                  * reference count has been bumped to insure it remains
1089                  * after the transaction completes.
1090                  */
1091                 ni->ni_flags |= IEEE80211_NODE_AREF;
1092                 /*
1093                  * Mark the node as requiring a valid associatio id
1094                  * before outbound traffic is permitted.
1095                  */
1096                 ni->ni_flags |= IEEE80211_NODE_ASSOCID;
1097                 IEEE80211_RSSI_LPF(ni->ni_avgrssi, rssi);
1098                 ni->ni_noise = nf;
1099                 if (!ieee80211_alloc_challenge(ni)) {
1100                         /* NB: don't return error so they rexmit */
1101                         return;
1102                 }
1103                 get_random_bytes(ni->ni_challenge,
1104                         IEEE80211_CHALLENGE_LEN);
1105                 IEEE80211_NOTE(vap, IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH,
1106                     ni, "shared key %sauth request", allocbs ? "" : "re");
1107                 /*
1108                  * When the ACL policy is set to RADIUS we defer the
1109                  * authorization to a user agent.  Dispatch an event,
1110                  * a subsequent MLME call will decide the fate of the
1111                  * station.  If the user agent is not present then the
1112                  * node will be reclaimed due to inactivity.
1113                  */
1114                 if (vap->iv_acl != NULL &&
1115                     vap->iv_acl->iac_getpolicy(vap) == IEEE80211_MACCMD_POLICY_RADIUS) {
1116                         IEEE80211_NOTE_MAC(vap,
1117                             IEEE80211_MSG_AUTH | IEEE80211_MSG_ACL,
1118                             ni->ni_macaddr,
1119                             "%s", "station authentication defered (radius acl)");
1120                         ieee80211_notify_node_auth(ni);
1121                         return;
1122                 }
1123                 break;
1124         case IEEE80211_AUTH_SHARED_RESPONSE:
1125                 if (ni == vap->iv_bss) {
1126                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1127                             ni->ni_macaddr, "shared key response",
1128                             "%s", "unknown station");
1129                         /* NB: don't send a response */
1130                         return;
1131                 }
1132                 if (ni->ni_challenge == NULL) {
1133                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1134                             ni->ni_macaddr, "shared key response",
1135                             "%s", "no challenge recorded");
1136                         vap->iv_stats.is_rx_bad_auth++;
1137                         estatus = IEEE80211_STATUS_CHALLENGE;
1138                         goto bad;
1139                 }
1140                 if (memcmp(ni->ni_challenge, &challenge[2],
1141                            challenge[1]) != 0) {
1142                         IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1143                             ni->ni_macaddr, "shared key response",
1144                             "%s", "challenge mismatch");
1145                         vap->iv_stats.is_rx_auth_fail++;
1146                         estatus = IEEE80211_STATUS_CHALLENGE;
1147                         goto bad;
1148                 }
1149                 IEEE80211_NOTE(vap, IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH,
1150                     ni, "%s", "station authenticated (shared key)");
1151                 ieee80211_node_authorize(ni);
1152                 break;
1153         default:
1154                 IEEE80211_DISCARD_MAC(vap, IEEE80211_MSG_AUTH,
1155                     ni->ni_macaddr, "shared key auth",
1156                     "bad seq %d", seq);
1157                 vap->iv_stats.is_rx_bad_auth++;
1158                 estatus = IEEE80211_STATUS_SEQUENCE;
1159                 goto bad;
1160         }
1161         IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_AUTH, seq + 1);
1162         return;
1163 bad:
1164         /*
1165          * Send an error response; but only when operating as an AP.
1166          */
1167         /* XXX hack to workaround calling convention */
1168         ieee80211_send_error(ni, wh->i_addr2,
1169             IEEE80211_FC0_SUBTYPE_AUTH,
1170             (seq + 1) | (estatus<<16));
1171 }
1172
1173 /*
1174  * Convert a WPA cipher selector OUI to an internal
1175  * cipher algorithm.  Where appropriate we also
1176  * record any key length.
1177  */
1178 static int
1179 wpa_cipher(const uint8_t *sel, uint8_t *keylen)
1180 {
1181 #define WPA_SEL(x)      (((x)<<24)|WPA_OUI)
1182         uint32_t w = LE_READ_4(sel);
1183
1184         switch (w) {
1185         case WPA_SEL(WPA_CSE_NULL):
1186                 return IEEE80211_CIPHER_NONE;
1187         case WPA_SEL(WPA_CSE_WEP40):
1188                 if (keylen)
1189                         *keylen = 40 / NBBY;
1190                 return IEEE80211_CIPHER_WEP;
1191         case WPA_SEL(WPA_CSE_WEP104):
1192                 if (keylen)
1193                         *keylen = 104 / NBBY;
1194                 return IEEE80211_CIPHER_WEP;
1195         case WPA_SEL(WPA_CSE_TKIP):
1196                 return IEEE80211_CIPHER_TKIP;
1197         case WPA_SEL(WPA_CSE_CCMP):
1198                 return IEEE80211_CIPHER_AES_CCM;
1199         }
1200         return 32;              /* NB: so 1<< is discarded */
1201 #undef WPA_SEL
1202 }
1203
1204 /*
1205  * Convert a WPA key management/authentication algorithm
1206  * to an internal code.
1207  */
1208 static int
1209 wpa_keymgmt(const uint8_t *sel)
1210 {
1211 #define WPA_SEL(x)      (((x)<<24)|WPA_OUI)
1212         uint32_t w = LE_READ_4(sel);
1213
1214         switch (w) {
1215         case WPA_SEL(WPA_ASE_8021X_UNSPEC):
1216                 return WPA_ASE_8021X_UNSPEC;
1217         case WPA_SEL(WPA_ASE_8021X_PSK):
1218                 return WPA_ASE_8021X_PSK;
1219         case WPA_SEL(WPA_ASE_NONE):
1220                 return WPA_ASE_NONE;
1221         }
1222         return 0;               /* NB: so is discarded */
1223 #undef WPA_SEL
1224 }
1225
1226 /*
1227  * Parse a WPA information element to collect parameters.
1228  * Note that we do not validate security parameters; that
1229  * is handled by the authenticator; the parsing done here
1230  * is just for internal use in making operational decisions.
1231  */
1232 static int
1233 ieee80211_parse_wpa(struct ieee80211vap *vap, const uint8_t *frm,
1234         struct ieee80211_rsnparms *rsn, const struct ieee80211_frame *wh)
1235 {
1236         uint8_t len = frm[1];
1237         uint32_t w;
1238         int n;
1239
1240         /*
1241          * Check the length once for fixed parts: OUI, type,
1242          * version, mcast cipher, and 2 selector counts.
1243          * Other, variable-length data, must be checked separately.
1244          */
1245         if ((vap->iv_flags & IEEE80211_F_WPA1) == 0) {
1246                 IEEE80211_DISCARD_IE(vap,
1247                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1248                     wh, "WPA", "not WPA, flags 0x%x", vap->iv_flags);
1249                 return IEEE80211_REASON_IE_INVALID;
1250         }
1251         if (len < 14) {
1252                 IEEE80211_DISCARD_IE(vap,
1253                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1254                     wh, "WPA", "too short, len %u", len);
1255                 return IEEE80211_REASON_IE_INVALID;
1256         }
1257         frm += 6, len -= 4;             /* NB: len is payload only */
1258         /* NB: iswpaoui already validated the OUI and type */
1259         w = LE_READ_2(frm);
1260         if (w != WPA_VERSION) {
1261                 IEEE80211_DISCARD_IE(vap,
1262                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1263                     wh, "WPA", "bad version %u", w);
1264                 return IEEE80211_REASON_IE_INVALID;
1265         }
1266         frm += 2, len -= 2;
1267
1268         memset(rsn, 0, sizeof(*rsn));
1269
1270         /* multicast/group cipher */
1271         rsn->rsn_mcastcipher = wpa_cipher(frm, &rsn->rsn_mcastkeylen);
1272         frm += 4, len -= 4;
1273
1274         /* unicast ciphers */
1275         n = LE_READ_2(frm);
1276         frm += 2, len -= 2;
1277         if (len < n*4+2) {
1278                 IEEE80211_DISCARD_IE(vap,
1279                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1280                     wh, "WPA", "ucast cipher data too short; len %u, n %u",
1281                     len, n);
1282                 return IEEE80211_REASON_IE_INVALID;
1283         }
1284         w = 0;
1285         for (; n > 0; n--) {
1286                 w |= 1<<wpa_cipher(frm, &rsn->rsn_ucastkeylen);
1287                 frm += 4, len -= 4;
1288         }
1289         if (w & (1<<IEEE80211_CIPHER_TKIP))
1290                 rsn->rsn_ucastcipher = IEEE80211_CIPHER_TKIP;
1291         else
1292                 rsn->rsn_ucastcipher = IEEE80211_CIPHER_AES_CCM;
1293
1294         /* key management algorithms */
1295         n = LE_READ_2(frm);
1296         frm += 2, len -= 2;
1297         if (len < n*4) {
1298                 IEEE80211_DISCARD_IE(vap,
1299                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1300                     wh, "WPA", "key mgmt alg data too short; len %u, n %u",
1301                     len, n);
1302                 return IEEE80211_REASON_IE_INVALID;
1303         }
1304         w = 0;
1305         for (; n > 0; n--) {
1306                 w |= wpa_keymgmt(frm);
1307                 frm += 4, len -= 4;
1308         }
1309         if (w & WPA_ASE_8021X_UNSPEC)
1310                 rsn->rsn_keymgmt = WPA_ASE_8021X_UNSPEC;
1311         else
1312                 rsn->rsn_keymgmt = WPA_ASE_8021X_PSK;
1313
1314         if (len > 2)            /* optional capabilities */
1315                 rsn->rsn_caps = LE_READ_2(frm);
1316
1317         return 0;
1318 }
1319
1320 /*
1321  * Convert an RSN cipher selector OUI to an internal
1322  * cipher algorithm.  Where appropriate we also
1323  * record any key length.
1324  */
1325 static int
1326 rsn_cipher(const uint8_t *sel, uint8_t *keylen)
1327 {
1328 #define RSN_SEL(x)      (((x)<<24)|RSN_OUI)
1329         uint32_t w = LE_READ_4(sel);
1330
1331         switch (w) {
1332         case RSN_SEL(RSN_CSE_NULL):
1333                 return IEEE80211_CIPHER_NONE;
1334         case RSN_SEL(RSN_CSE_WEP40):
1335                 if (keylen)
1336                         *keylen = 40 / NBBY;
1337                 return IEEE80211_CIPHER_WEP;
1338         case RSN_SEL(RSN_CSE_WEP104):
1339                 if (keylen)
1340                         *keylen = 104 / NBBY;
1341                 return IEEE80211_CIPHER_WEP;
1342         case RSN_SEL(RSN_CSE_TKIP):
1343                 return IEEE80211_CIPHER_TKIP;
1344         case RSN_SEL(RSN_CSE_CCMP):
1345                 return IEEE80211_CIPHER_AES_CCM;
1346         case RSN_SEL(RSN_CSE_WRAP):
1347                 return IEEE80211_CIPHER_AES_OCB;
1348         }
1349         return 32;              /* NB: so 1<< is discarded */
1350 #undef WPA_SEL
1351 }
1352
1353 /*
1354  * Convert an RSN key management/authentication algorithm
1355  * to an internal code.
1356  */
1357 static int
1358 rsn_keymgmt(const uint8_t *sel)
1359 {
1360 #define RSN_SEL(x)      (((x)<<24)|RSN_OUI)
1361         uint32_t w = LE_READ_4(sel);
1362
1363         switch (w) {
1364         case RSN_SEL(RSN_ASE_8021X_UNSPEC):
1365                 return RSN_ASE_8021X_UNSPEC;
1366         case RSN_SEL(RSN_ASE_8021X_PSK):
1367                 return RSN_ASE_8021X_PSK;
1368         case RSN_SEL(RSN_ASE_NONE):
1369                 return RSN_ASE_NONE;
1370         }
1371         return 0;               /* NB: so is discarded */
1372 #undef RSN_SEL
1373 }
1374
1375 /*
1376  * Parse a WPA/RSN information element to collect parameters
1377  * and validate the parameters against what has been
1378  * configured for the system.
1379  */
1380 static int
1381 ieee80211_parse_rsn(struct ieee80211vap *vap, const uint8_t *frm,
1382         struct ieee80211_rsnparms *rsn, const struct ieee80211_frame *wh)
1383 {
1384         uint8_t len = frm[1];
1385         uint32_t w;
1386         int n;
1387
1388         /*
1389          * Check the length once for fixed parts:
1390          * version, mcast cipher, and 2 selector counts.
1391          * Other, variable-length data, must be checked separately.
1392          */
1393         if ((vap->iv_flags & IEEE80211_F_WPA2) == 0) {
1394                 IEEE80211_DISCARD_IE(vap,
1395                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1396                     wh, "WPA", "not RSN, flags 0x%x", vap->iv_flags);
1397                 return IEEE80211_REASON_IE_INVALID;
1398         }
1399         if (len < 10) {
1400                 IEEE80211_DISCARD_IE(vap,
1401                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1402                     wh, "RSN", "too short, len %u", len);
1403                 return IEEE80211_REASON_IE_INVALID;
1404         }
1405         frm += 2;
1406         w = LE_READ_2(frm);
1407         if (w != RSN_VERSION) {
1408                 IEEE80211_DISCARD_IE(vap,
1409                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1410                     wh, "RSN", "bad version %u", w);
1411                 return IEEE80211_REASON_IE_INVALID;
1412         }
1413         frm += 2, len -= 2;
1414
1415         memset(rsn, 0, sizeof(*rsn));
1416
1417         /* multicast/group cipher */
1418         rsn->rsn_mcastcipher = rsn_cipher(frm, &rsn->rsn_mcastkeylen);
1419         frm += 4, len -= 4;
1420
1421         /* unicast ciphers */
1422         n = LE_READ_2(frm);
1423         frm += 2, len -= 2;
1424         if (len < n*4+2) {
1425                 IEEE80211_DISCARD_IE(vap,
1426                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1427                     wh, "RSN", "ucast cipher data too short; len %u, n %u",
1428                     len, n);
1429                 return IEEE80211_REASON_IE_INVALID;
1430         }
1431         w = 0;
1432         for (; n > 0; n--) {
1433                 w |= 1<<rsn_cipher(frm, &rsn->rsn_ucastkeylen);
1434                 frm += 4, len -= 4;
1435         }
1436         if (w & (1<<IEEE80211_CIPHER_TKIP))
1437                 rsn->rsn_ucastcipher = IEEE80211_CIPHER_TKIP;
1438         else
1439                 rsn->rsn_ucastcipher = IEEE80211_CIPHER_AES_CCM;
1440
1441         /* key management algorithms */
1442         n = LE_READ_2(frm);
1443         frm += 2, len -= 2;
1444         if (len < n*4) {
1445                 IEEE80211_DISCARD_IE(vap,
1446                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_WPA,
1447                     wh, "RSN", "key mgmt alg data too short; len %u, n %u",
1448                     len, n);
1449                 return IEEE80211_REASON_IE_INVALID;
1450         }
1451         w = 0;
1452         for (; n > 0; n--) {
1453                 w |= rsn_keymgmt(frm);
1454                 frm += 4, len -= 4;
1455         }
1456         if (w & RSN_ASE_8021X_UNSPEC)
1457                 rsn->rsn_keymgmt = RSN_ASE_8021X_UNSPEC;
1458         else
1459                 rsn->rsn_keymgmt = RSN_ASE_8021X_PSK;
1460
1461         /* optional RSN capabilities */
1462         if (len > 2)
1463                 rsn->rsn_caps = LE_READ_2(frm);
1464         /* XXXPMKID */
1465
1466         return 0;
1467 }
1468
1469 /*
1470  * WPA/802.11i assocation request processing.
1471  */
1472 static int
1473 wpa_assocreq(struct ieee80211_node *ni, struct ieee80211_rsnparms *rsnparms,
1474         const struct ieee80211_frame *wh, const uint8_t *wpa,
1475         const uint8_t *rsn, uint16_t capinfo)
1476 {
1477         struct ieee80211vap *vap = ni->ni_vap;
1478         uint8_t reason;
1479         int badwparsn;
1480
1481         ni->ni_flags &= ~(IEEE80211_NODE_WPS|IEEE80211_NODE_TSN);
1482         if (wpa == NULL && rsn == NULL) {
1483                 if (vap->iv_flags_ext & IEEE80211_FEXT_WPS) {
1484                         /*
1485                          * W-Fi Protected Setup (WPS) permits
1486                          * clients to associate and pass EAPOL frames
1487                          * to establish initial credentials.
1488                          */
1489                         ni->ni_flags |= IEEE80211_NODE_WPS;
1490                         return 1;
1491                 }
1492                 if ((vap->iv_flags_ext & IEEE80211_FEXT_TSN) &&
1493                     (capinfo & IEEE80211_CAPINFO_PRIVACY)) {
1494                         /*
1495                          * Transitional Security Network.  Permits clients
1496                          * to associate and use WEP while WPA is configured.
1497                          */
1498                         ni->ni_flags |= IEEE80211_NODE_TSN;
1499                         return 1;
1500                 }
1501                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ASSOC | IEEE80211_MSG_WPA,
1502                     wh, NULL, "%s", "no WPA/RSN IE in association request");
1503                 vap->iv_stats.is_rx_assoc_badwpaie++;
1504                 reason = IEEE80211_REASON_IE_INVALID;
1505                 goto bad;
1506         }
1507         /* assert right association security credentials */
1508         badwparsn = 0;                  /* NB: to silence compiler */
1509         switch (vap->iv_flags & IEEE80211_F_WPA) {
1510         case IEEE80211_F_WPA1:
1511                 badwparsn = (wpa == NULL);
1512                 break;
1513         case IEEE80211_F_WPA2:
1514                 badwparsn = (rsn == NULL);
1515                 break;
1516         case IEEE80211_F_WPA1|IEEE80211_F_WPA2:
1517                 badwparsn = (wpa == NULL && rsn == NULL);
1518                 break;
1519         }
1520         if (badwparsn) {
1521                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ASSOC | IEEE80211_MSG_WPA,
1522                     wh, NULL,
1523                     "%s", "missing WPA/RSN IE in association request");
1524                 vap->iv_stats.is_rx_assoc_badwpaie++;
1525                 reason = IEEE80211_REASON_IE_INVALID;
1526                 goto bad;
1527         }
1528         /*
1529          * Parse WPA/RSN information element.
1530          */
1531         if (wpa != NULL)
1532                 reason = ieee80211_parse_wpa(vap, wpa, rsnparms, wh);
1533         else
1534                 reason = ieee80211_parse_rsn(vap, rsn, rsnparms, wh);
1535         if (reason != 0) {
1536                 /* XXX distinguish WPA/RSN? */
1537                 vap->iv_stats.is_rx_assoc_badwpaie++;
1538                 goto bad;
1539         }
1540         IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC | IEEE80211_MSG_WPA, ni,
1541             "%s ie: mc %u/%u uc %u/%u key %u caps 0x%x",
1542             wpa != NULL ? "WPA" : "RSN",
1543             rsnparms->rsn_mcastcipher, rsnparms->rsn_mcastkeylen,
1544             rsnparms->rsn_ucastcipher, rsnparms->rsn_ucastkeylen,
1545             rsnparms->rsn_keymgmt, rsnparms->rsn_caps);
1546
1547         return 1;
1548 bad:
1549         ieee80211_node_deauth(ni, reason);
1550         return 0;
1551 }
1552
1553 /* XXX find a better place for definition */
1554 struct l2_update_frame {
1555         struct ether_header eh;
1556         uint8_t dsap;
1557         uint8_t ssap;
1558         uint8_t control;
1559         uint8_t xid[3];
1560 }  __packed;
1561
1562 /*
1563  * Deliver a TGf L2UF frame on behalf of a station.
1564  * This primes any bridge when the station is roaming
1565  * between ap's on the same wired network.
1566  */
1567 static void
1568 ieee80211_deliver_l2uf(struct ieee80211_node *ni)
1569 {
1570         struct ieee80211vap *vap = ni->ni_vap;
1571         struct ifnet *ifp = vap->iv_ifp;
1572         struct mbuf *m;
1573         struct l2_update_frame *l2uf;
1574         struct ether_header *eh;
1575
1576         m = m_gethdr(M_NOWAIT, MT_DATA);
1577         if (m == NULL) {
1578                 IEEE80211_NOTE(vap, IEEE80211_MSG_ASSOC, ni,
1579                     "%s", "no mbuf for l2uf frame");
1580                 vap->iv_stats.is_rx_nobuf++;    /* XXX not right */
1581                 return;
1582         }
1583         l2uf = mtod(m, struct l2_update_frame *);
1584         eh = &l2uf->eh;
1585         /* dst: Broadcast address */
1586         IEEE80211_ADDR_COPY(eh->ether_dhost, ifp->if_broadcastaddr);
1587         /* src: associated STA */
1588         IEEE80211_ADDR_COPY(eh->ether_shost, ni->ni_macaddr);
1589         eh->ether_type = htons(sizeof(*l2uf) - sizeof(*eh));
1590
1591         l2uf->dsap = 0;
1592         l2uf->ssap = 0;
1593         l2uf->control = 0xf5;
1594         l2uf->xid[0] = 0x81;
1595         l2uf->xid[1] = 0x80;
1596         l2uf->xid[2] = 0x00;
1597
1598         m->m_pkthdr.len = m->m_len = sizeof(*l2uf);
1599         hostap_deliver_data(vap, ni, m);
1600 }
1601
1602 static void
1603 ratesetmismatch(struct ieee80211_node *ni, const struct ieee80211_frame *wh,
1604         int reassoc, int resp, const char *tag, int rate)
1605 {
1606         IEEE80211_NOTE_MAC(ni->ni_vap, IEEE80211_MSG_ANY, wh->i_addr2,
1607             "deny %s request, %s rate set mismatch, rate/MCS %d",
1608             reassoc ? "reassoc" : "assoc", tag, rate & IEEE80211_RATE_VAL);
1609         IEEE80211_SEND_MGMT(ni, resp, IEEE80211_STATUS_BASIC_RATE);
1610         ieee80211_node_leave(ni);
1611 }
1612
1613 static void
1614 capinfomismatch(struct ieee80211_node *ni, const struct ieee80211_frame *wh,
1615         int reassoc, int resp, const char *tag, int capinfo)
1616 {
1617         struct ieee80211vap *vap = ni->ni_vap;
1618
1619         IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ANY, wh->i_addr2,
1620             "deny %s request, %s mismatch 0x%x",
1621             reassoc ? "reassoc" : "assoc", tag, capinfo);
1622         IEEE80211_SEND_MGMT(ni, resp, IEEE80211_STATUS_CAPINFO);
1623         ieee80211_node_leave(ni);
1624         vap->iv_stats.is_rx_assoc_capmismatch++;
1625 }
1626
1627 static void
1628 htcapmismatch(struct ieee80211_node *ni, const struct ieee80211_frame *wh,
1629         int reassoc, int resp)
1630 {
1631         IEEE80211_NOTE_MAC(ni->ni_vap, IEEE80211_MSG_ANY, wh->i_addr2,
1632             "deny %s request, %s missing HT ie", reassoc ? "reassoc" : "assoc");
1633         /* XXX no better code */
1634         IEEE80211_SEND_MGMT(ni, resp, IEEE80211_STATUS_MISSING_HT_CAPS);
1635         ieee80211_node_leave(ni);
1636 }
1637
1638 static void
1639 authalgreject(struct ieee80211_node *ni, const struct ieee80211_frame *wh,
1640         int algo, int seq, int status)
1641 {
1642         struct ieee80211vap *vap = ni->ni_vap;
1643
1644         IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
1645             wh, NULL, "unsupported alg %d", algo);
1646         vap->iv_stats.is_rx_auth_unsupported++;
1647         ieee80211_send_error(ni, wh->i_addr2, IEEE80211_FC0_SUBTYPE_AUTH,
1648             seq | (status << 16));
1649 }
1650
1651 static __inline int
1652 ishtmixed(const uint8_t *ie)
1653 {
1654         const struct ieee80211_ie_htinfo *ht =
1655             (const struct ieee80211_ie_htinfo *) ie;
1656         return (ht->hi_byte2 & IEEE80211_HTINFO_OPMODE) ==
1657             IEEE80211_HTINFO_OPMODE_MIXED;
1658 }
1659
1660 static int
1661 is11bclient(const uint8_t *rates, const uint8_t *xrates)
1662 {
1663         static const uint32_t brates = (1<<2*1)|(1<<2*2)|(1<<11)|(1<<2*11);
1664         int i;
1665
1666         /* NB: the 11b clients we care about will not have xrates */
1667         if (xrates != NULL || rates == NULL)
1668                 return 0;
1669         for (i = 0; i < rates[1]; i++) {
1670                 int r = rates[2+i] & IEEE80211_RATE_VAL;
1671                 if (r > 2*11 || ((1<<r) & brates) == 0)
1672                         return 0;
1673         }
1674         return 1;
1675 }
1676
1677 static void
1678 hostap_recv_mgmt(struct ieee80211_node *ni, struct mbuf *m0,
1679         int subtype, int rssi, int nf)
1680 {
1681         struct ieee80211vap *vap = ni->ni_vap;
1682         struct ieee80211com *ic = ni->ni_ic;
1683         struct ieee80211_frame *wh;
1684         uint8_t *frm, *efrm, *sfrm;
1685         uint8_t *ssid, *rates, *xrates, *wpa, *rsn, *wme, *ath, *htcap;
1686         int reassoc, resp;
1687         uint8_t rate;
1688
1689         wh = mtod(m0, struct ieee80211_frame *);
1690         frm = (uint8_t *)&wh[1];
1691         efrm = mtod(m0, uint8_t *) + m0->m_len;
1692         switch (subtype) {
1693         case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
1694         case IEEE80211_FC0_SUBTYPE_BEACON: {
1695                 struct ieee80211_scanparams scan;
1696                 /*
1697                  * We process beacon/probe response frames when scanning;
1698                  * otherwise we check beacon frames for overlapping non-ERP
1699                  * BSS in 11g and/or overlapping legacy BSS when in HT.
1700                  */
1701                 if ((ic->ic_flags & IEEE80211_F_SCAN) == 0 &&
1702                     subtype == IEEE80211_FC0_SUBTYPE_PROBE_RESP) {
1703                         vap->iv_stats.is_rx_mgtdiscard++;
1704                         return;
1705                 }
1706                 /* NB: accept off-channel frames */
1707                 if (ieee80211_parse_beacon(ni, m0, &scan) &~ IEEE80211_BPARSE_OFFCHAN)
1708                         return;
1709                 /*
1710                  * Count frame now that we know it's to be processed.
1711                  */
1712                 if (subtype == IEEE80211_FC0_SUBTYPE_BEACON) {
1713                         vap->iv_stats.is_rx_beacon++;           /* XXX remove */
1714                         IEEE80211_NODE_STAT(ni, rx_beacons);
1715                 } else
1716                         IEEE80211_NODE_STAT(ni, rx_proberesp);
1717                 /*
1718                  * If scanning, just pass information to the scan module.
1719                  */
1720                 if (ic->ic_flags & IEEE80211_F_SCAN) {
1721                         if (scan.status == 0 &&         /* NB: on channel */
1722                             (ic->ic_flags_ext & IEEE80211_FEXT_PROBECHAN)) {
1723                                 /*
1724                                  * Actively scanning a channel marked passive;
1725                                  * send a probe request now that we know there
1726                                  * is 802.11 traffic present.
1727                                  *
1728                                  * XXX check if the beacon we recv'd gives
1729                                  * us what we need and suppress the probe req
1730                                  */
1731                                 ieee80211_probe_curchan(vap, 1);
1732                                 ic->ic_flags_ext &= ~IEEE80211_FEXT_PROBECHAN;
1733                         }
1734                         ieee80211_add_scan(vap, &scan, wh, subtype, rssi, nf);
1735                         return;
1736                 }
1737                 /*
1738                  * Check beacon for overlapping bss w/ non ERP stations.
1739                  * If we detect one and protection is configured but not
1740                  * enabled, enable it and start a timer that'll bring us
1741                  * out if we stop seeing the bss.
1742                  */
1743                 if (IEEE80211_IS_CHAN_ANYG(ic->ic_curchan) &&
1744                     scan.status == 0 &&                 /* NB: on-channel */
1745                     ((scan.erp & 0x100) == 0 ||         /* NB: no ERP, 11b sta*/
1746                      (scan.erp & IEEE80211_ERP_NON_ERP_PRESENT))) {
1747                         ic->ic_lastnonerp = ticks;
1748                         ic->ic_flags_ext |= IEEE80211_FEXT_NONERP_PR;
1749                         if (ic->ic_protmode != IEEE80211_PROT_NONE &&
1750                             (ic->ic_flags & IEEE80211_F_USEPROT) == 0) {
1751                                 IEEE80211_NOTE_FRAME(vap,
1752                                     IEEE80211_MSG_ASSOC, wh,
1753                                     "non-ERP present on channel %d "
1754                                     "(saw erp 0x%x from channel %d), "
1755                                     "enable use of protection",
1756                                     ic->ic_curchan->ic_ieee,
1757                                     scan.erp, scan.chan);
1758                                 ic->ic_flags |= IEEE80211_F_USEPROT;
1759                                 ieee80211_notify_erp(ic);
1760                         }
1761                 }
1762                 /*
1763                  * Check beacon for non-HT station on HT channel
1764                  * and update HT BSS occupancy as appropriate.
1765                  */
1766                 if (IEEE80211_IS_CHAN_HT(ic->ic_curchan)) {
1767                         if (scan.status & IEEE80211_BPARSE_OFFCHAN) {
1768                                 /*
1769                                  * Off control channel; only check frames
1770                                  * that come in the extension channel when
1771                                  * operating w/ HT40.
1772                                  */
1773                                 if (!IEEE80211_IS_CHAN_HT40(ic->ic_curchan))
1774                                         break;
1775                                 if (scan.chan != ic->ic_curchan->ic_extieee)
1776                                         break;
1777                         }
1778                         if (scan.htinfo == NULL) {
1779                                 ieee80211_htprot_update(ic,
1780                                     IEEE80211_HTINFO_OPMODE_PROTOPT |
1781                                     IEEE80211_HTINFO_NONHT_PRESENT);
1782                         } else if (ishtmixed(scan.htinfo)) {
1783                                 /* XXX? take NONHT_PRESENT from beacon? */
1784                                 ieee80211_htprot_update(ic,
1785                                     IEEE80211_HTINFO_OPMODE_MIXED |
1786                                     IEEE80211_HTINFO_NONHT_PRESENT);
1787                         }
1788                 }
1789                 break;
1790         }
1791
1792         case IEEE80211_FC0_SUBTYPE_PROBE_REQ:
1793                 if (vap->iv_state != IEEE80211_S_RUN) {
1794                         vap->iv_stats.is_rx_mgtdiscard++;
1795                         return;
1796                 }
1797                 /*
1798                  * prreq frame format
1799                  *      [tlv] ssid
1800                  *      [tlv] supported rates
1801                  *      [tlv] extended supported rates
1802                  */
1803                 ssid = rates = xrates = NULL;
1804                 sfrm = frm;
1805                 while (efrm - frm > 1) {
1806                         IEEE80211_VERIFY_LENGTH(efrm - frm, frm[1] + 2, return);
1807                         switch (*frm) {
1808                         case IEEE80211_ELEMID_SSID:
1809                                 ssid = frm;
1810                                 break;
1811                         case IEEE80211_ELEMID_RATES:
1812                                 rates = frm;
1813                                 break;
1814                         case IEEE80211_ELEMID_XRATES:
1815                                 xrates = frm;
1816                                 break;
1817                         }
1818                         frm += frm[1] + 2;
1819                 }
1820                 IEEE80211_VERIFY_ELEMENT(rates, IEEE80211_RATE_MAXSIZE, return);
1821                 if (xrates != NULL)
1822                         IEEE80211_VERIFY_ELEMENT(xrates,
1823                                 IEEE80211_RATE_MAXSIZE - rates[1], return);
1824                 IEEE80211_VERIFY_ELEMENT(ssid, IEEE80211_NWID_LEN, return);
1825                 IEEE80211_VERIFY_SSID(vap->iv_bss, ssid, return);
1826                 if ((vap->iv_flags & IEEE80211_F_HIDESSID) && ssid[1] == 0) {
1827                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
1828                             wh, NULL,
1829                             "%s", "no ssid with ssid suppression enabled");
1830                         vap->iv_stats.is_rx_ssidmismatch++; /*XXX*/
1831                         return;
1832                 }
1833
1834                 /* XXX find a better class or define it's own */
1835                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_INPUT, wh->i_addr2,
1836                     "%s", "recv probe req");
1837                 /*
1838                  * Some legacy 11b clients cannot hack a complete
1839                  * probe response frame.  When the request includes
1840                  * only a bare-bones rate set, communicate this to
1841                  * the transmit side.
1842                  */
1843                 ieee80211_send_proberesp(vap, wh->i_addr2,
1844                     is11bclient(rates, xrates) ? IEEE80211_SEND_LEGACY_11B : 0);
1845                 break;
1846
1847         case IEEE80211_FC0_SUBTYPE_AUTH: {
1848                 uint16_t algo, seq, status;
1849
1850                 if (vap->iv_state != IEEE80211_S_RUN) {
1851                         vap->iv_stats.is_rx_mgtdiscard++;
1852                         return;
1853                 }
1854                 if (!IEEE80211_ADDR_EQ(wh->i_addr3, vap->iv_bss->ni_bssid)) {
1855                         IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
1856                             wh, NULL, "%s", "wrong bssid");
1857                         vap->iv_stats.is_rx_wrongbss++; /*XXX unique stat?*/
1858                         return;
1859                 }
1860                 /*
1861                  * auth frame format
1862                  *      [2] algorithm
1863                  *      [2] sequence
1864                  *      [2] status
1865                  *      [tlv*] challenge
1866                  */
1867                 IEEE80211_VERIFY_LENGTH(efrm - frm, 6, return);
1868                 algo   = le16toh(*(uint16_t *)frm);
1869                 seq    = le16toh(*(uint16_t *)(frm + 2));
1870                 status = le16toh(*(uint16_t *)(frm + 4));
1871                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_AUTH, wh->i_addr2,
1872                     "recv auth frame with algorithm %d seq %d", algo, seq);
1873                 /*
1874                  * Consult the ACL policy module if setup.
1875                  */
1876                 if (vap->iv_acl != NULL &&
1877                     !vap->iv_acl->iac_check(vap, wh->i_addr2)) {
1878                         IEEE80211_DISCARD(vap, IEEE80211_MSG_ACL,
1879                             wh, NULL, "%s", "disallowed by ACL");
1880                         vap->iv_stats.is_rx_acl++;
1881                         ieee80211_send_error(ni, wh->i_addr2,
1882                             IEEE80211_FC0_SUBTYPE_AUTH,
1883                             (seq+1) | (IEEE80211_STATUS_UNSPECIFIED<<16));
1884                         return;
1885                 }
1886                 if (vap->iv_flags & IEEE80211_F_COUNTERM) {
1887                         IEEE80211_DISCARD(vap,
1888                             IEEE80211_MSG_AUTH | IEEE80211_MSG_CRYPTO,
1889                             wh, NULL, "%s", "TKIP countermeasures enabled");
1890                         vap->iv_stats.is_rx_auth_countermeasures++;
1891                         ieee80211_send_error(ni, wh->i_addr2,
1892                                 IEEE80211_FC0_SUBTYPE_AUTH,
1893                                 IEEE80211_REASON_MIC_FAILURE);
1894                         return;
1895                 }
1896                 if (algo == IEEE80211_AUTH_ALG_SHARED)
1897                         hostap_auth_shared(ni, wh, frm + 6, efrm, rssi, nf,
1898                             seq, status);
1899                 else if (algo == IEEE80211_AUTH_ALG_OPEN)
1900                         hostap_auth_open(ni, wh, rssi, nf, seq, status);
1901                 else if (algo == IEEE80211_AUTH_ALG_LEAP) {
1902                         authalgreject(ni, wh, algo,
1903                             seq+1, IEEE80211_STATUS_ALG);
1904                         return;
1905                 } else {
1906                         /*
1907                          * We assume that an unknown algorithm is the result
1908                          * of a decryption failure on a shared key auth frame;
1909                          * return a status code appropriate for that instead
1910                          * of IEEE80211_STATUS_ALG.
1911                          *
1912                          * NB: a seq# of 4 is intentional; the decrypted
1913                          *     frame likely has a bogus seq value.
1914                          */
1915                         authalgreject(ni, wh, algo,
1916                             4, IEEE80211_STATUS_CHALLENGE);
1917                         return;
1918                 }
1919                 break;
1920         }
1921
1922         case IEEE80211_FC0_SUBTYPE_ASSOC_REQ:
1923         case IEEE80211_FC0_SUBTYPE_REASSOC_REQ: {
1924                 uint16_t capinfo, lintval;
1925                 struct ieee80211_rsnparms rsnparms;
1926
1927                 if (vap->iv_state != IEEE80211_S_RUN) {
1928                         vap->iv_stats.is_rx_mgtdiscard++;
1929                         return;
1930                 }
1931                 if (!IEEE80211_ADDR_EQ(wh->i_addr3, vap->iv_bss->ni_bssid)) {
1932                         IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
1933                             wh, NULL, "%s", "wrong bssid");
1934                         vap->iv_stats.is_rx_assoc_bss++;
1935                         return;
1936                 }
1937                 if (subtype == IEEE80211_FC0_SUBTYPE_REASSOC_REQ) {
1938                         reassoc = 1;
1939                         resp = IEEE80211_FC0_SUBTYPE_REASSOC_RESP;
1940                 } else {
1941                         reassoc = 0;
1942                         resp = IEEE80211_FC0_SUBTYPE_ASSOC_RESP;
1943                 }
1944                 if (ni == vap->iv_bss) {
1945                         IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_ANY, wh->i_addr2,
1946                             "deny %s request, sta not authenticated",
1947                             reassoc ? "reassoc" : "assoc");
1948                         ieee80211_send_error(ni, wh->i_addr2,
1949                             IEEE80211_FC0_SUBTYPE_DEAUTH,
1950                             IEEE80211_REASON_ASSOC_NOT_AUTHED);
1951                         vap->iv_stats.is_rx_assoc_notauth++;
1952                         return;
1953                 }
1954
1955                 /*
1956                  * asreq frame format
1957                  *      [2] capability information
1958                  *      [2] listen interval
1959                  *      [6*] current AP address (reassoc only)
1960                  *      [tlv] ssid
1961                  *      [tlv] supported rates
1962                  *      [tlv] extended supported rates
1963                  *      [tlv] WPA or RSN
1964                  *      [tlv] HT capabilities
1965                  *      [tlv] Atheros capabilities
1966                  */
1967                 IEEE80211_VERIFY_LENGTH(efrm - frm, (reassoc ? 10 : 4), return);
1968                 capinfo = le16toh(*(uint16_t *)frm);    frm += 2;
1969                 lintval = le16toh(*(uint16_t *)frm);    frm += 2;
1970                 if (reassoc)
1971                         frm += 6;       /* ignore current AP info */
1972                 ssid = rates = xrates = wpa = rsn = wme = ath = htcap = NULL;
1973                 sfrm = frm;
1974                 while (efrm - frm > 1) {
1975                         IEEE80211_VERIFY_LENGTH(efrm - frm, frm[1] + 2, return);
1976                         switch (*frm) {
1977                         case IEEE80211_ELEMID_SSID:
1978                                 ssid = frm;
1979                                 break;
1980                         case IEEE80211_ELEMID_RATES:
1981                                 rates = frm;
1982                                 break;
1983                         case IEEE80211_ELEMID_XRATES:
1984                                 xrates = frm;
1985                                 break;
1986                         case IEEE80211_ELEMID_RSN:
1987                                 rsn = frm;
1988                                 break;
1989                         case IEEE80211_ELEMID_HTCAP:
1990                                 htcap = frm;
1991                                 break;
1992                         case IEEE80211_ELEMID_VENDOR:
1993                                 if (iswpaoui(frm))
1994                                         wpa = frm;
1995                                 else if (iswmeinfo(frm))
1996                                         wme = frm;
1997 #ifdef IEEE80211_SUPPORT_SUPERG
1998                                 else if (isatherosoui(frm))
1999                                         ath = frm;
2000 #endif
2001                                 else if (vap->iv_flags_ht & IEEE80211_FHT_HTCOMPAT) {
2002                                         if (ishtcapoui(frm) && htcap == NULL)
2003                                                 htcap = frm;
2004                                 }
2005                                 break;
2006                         }
2007                         frm += frm[1] + 2;
2008                 }
2009                 IEEE80211_VERIFY_ELEMENT(rates, IEEE80211_RATE_MAXSIZE, return);
2010                 if (xrates != NULL)
2011                         IEEE80211_VERIFY_ELEMENT(xrates,
2012                                 IEEE80211_RATE_MAXSIZE - rates[1], return);
2013                 IEEE80211_VERIFY_ELEMENT(ssid, IEEE80211_NWID_LEN, return);
2014                 IEEE80211_VERIFY_SSID(vap->iv_bss, ssid, return);
2015                 if (htcap != NULL) {
2016                         IEEE80211_VERIFY_LENGTH(htcap[1],
2017                              htcap[0] == IEEE80211_ELEMID_VENDOR ?
2018                                  4 + sizeof(struct ieee80211_ie_htcap)-2 :
2019                                  sizeof(struct ieee80211_ie_htcap)-2,
2020                              return);           /* XXX just NULL out? */
2021                 }
2022
2023                 if ((vap->iv_flags & IEEE80211_F_WPA) &&
2024                     !wpa_assocreq(ni, &rsnparms, wh, wpa, rsn, capinfo))
2025                         return;
2026                 /* discard challenge after association */
2027                 if (ni->ni_challenge != NULL) {
2028                         free(ni->ni_challenge, M_80211_NODE);
2029                         ni->ni_challenge = NULL;
2030                 }
2031                 /* NB: 802.11 spec says to ignore station's privacy bit */
2032                 if ((capinfo & IEEE80211_CAPINFO_ESS) == 0) {
2033                         capinfomismatch(ni, wh, reassoc, resp,
2034                             "capability", capinfo);
2035                         return;
2036                 }
2037                 /*
2038                  * Disallow re-associate w/ invalid slot time setting.
2039                  */
2040                 if (ni->ni_associd != 0 &&
2041                     IEEE80211_IS_CHAN_ANYG(ic->ic_bsschan) &&
2042                     ((ni->ni_capinfo ^ capinfo) & IEEE80211_CAPINFO_SHORT_SLOTTIME)) {
2043                         capinfomismatch(ni, wh, reassoc, resp,
2044                             "slot time", capinfo);
2045                         return;
2046                 }
2047                 rate = ieee80211_setup_rates(ni, rates, xrates,
2048                                 IEEE80211_F_DOSORT | IEEE80211_F_DOFRATE |
2049                                 IEEE80211_F_DONEGO | IEEE80211_F_DODEL);
2050                 if (rate & IEEE80211_RATE_BASIC) {
2051                         ratesetmismatch(ni, wh, reassoc, resp, "legacy", rate);
2052                         vap->iv_stats.is_rx_assoc_norate++;
2053                         return;
2054                 }
2055                 /*
2056                  * If constrained to 11g-only stations reject an
2057                  * 11b-only station.  We cheat a bit here by looking
2058                  * at the max negotiated xmit rate and assuming anyone
2059                  * with a best rate <24Mb/s is an 11b station.
2060                  */
2061                 if ((vap->iv_flags & IEEE80211_F_PUREG) && rate < 48) {
2062                         ratesetmismatch(ni, wh, reassoc, resp, "11g", rate);
2063                         vap->iv_stats.is_rx_assoc_norate++;
2064                         return;
2065                 }
2066                 /*
2067                  * Do HT rate set handling and setup HT node state.
2068                  */
2069                 ni->ni_chan = vap->iv_bss->ni_chan;
2070                 if (IEEE80211_IS_CHAN_HT(ni->ni_chan) && htcap != NULL) {
2071                         rate = ieee80211_setup_htrates(ni, htcap,
2072                                 IEEE80211_F_DOFMCS | IEEE80211_F_DONEGO |
2073                                 IEEE80211_F_DOBRS);
2074                         if (rate & IEEE80211_RATE_BASIC) {
2075                                 ratesetmismatch(ni, wh, reassoc, resp,
2076                                     "HT", rate);
2077                                 vap->iv_stats.is_ht_assoc_norate++;
2078                                 return;
2079                         }
2080                         ieee80211_ht_node_init(ni);
2081                         ieee80211_ht_updatehtcap(ni, htcap);
2082                 } else if (ni->ni_flags & IEEE80211_NODE_HT)
2083                         ieee80211_ht_node_cleanup(ni);
2084 #ifdef IEEE80211_SUPPORT_SUPERG
2085                 else if (ni->ni_ath_flags & IEEE80211_NODE_ATH)
2086                         ieee80211_ff_node_cleanup(ni);
2087 #endif
2088                 /*
2089                  * Allow AMPDU operation only with unencrypted traffic
2090                  * or AES-CCM; the 11n spec only specifies these ciphers
2091                  * so permitting any others is undefined and can lead
2092                  * to interoperability problems.
2093                  */
2094                 if ((ni->ni_flags & IEEE80211_NODE_HT) &&
2095                     (((vap->iv_flags & IEEE80211_F_WPA) &&
2096                       rsnparms.rsn_ucastcipher != IEEE80211_CIPHER_AES_CCM) ||
2097                      (vap->iv_flags & (IEEE80211_F_WPA|IEEE80211_F_PRIVACY)) == IEEE80211_F_PRIVACY)) {
2098                         IEEE80211_NOTE(vap,
2099                             IEEE80211_MSG_ASSOC | IEEE80211_MSG_11N, ni,
2100                             "disallow HT use because WEP or TKIP requested, "
2101                             "capinfo 0x%x ucastcipher %d", capinfo,
2102                             rsnparms.rsn_ucastcipher);
2103                         ieee80211_ht_node_cleanup(ni);
2104                         vap->iv_stats.is_ht_assoc_downgrade++;
2105                 }
2106                 /*
2107                  * If constrained to 11n-only stations reject legacy stations.
2108                  */
2109                 if ((vap->iv_flags_ht & IEEE80211_FHT_PUREN) &&
2110                     (ni->ni_flags & IEEE80211_NODE_HT) == 0) {
2111                         htcapmismatch(ni, wh, reassoc, resp);
2112                         vap->iv_stats.is_ht_assoc_nohtcap++;
2113                         return;
2114                 }
2115                 IEEE80211_RSSI_LPF(ni->ni_avgrssi, rssi);
2116                 ni->ni_noise = nf;
2117                 ni->ni_intval = lintval;
2118                 ni->ni_capinfo = capinfo;
2119                 ni->ni_fhdwell = vap->iv_bss->ni_fhdwell;
2120                 ni->ni_fhindex = vap->iv_bss->ni_fhindex;
2121                 /*
2122                  * Store the IEs.
2123                  * XXX maybe better to just expand
2124                  */
2125                 if (ieee80211_ies_init(&ni->ni_ies, sfrm, efrm - sfrm)) {
2126 #define setie(_ie, _off)        ieee80211_ies_setie(ni->ni_ies, _ie, _off)
2127                         if (wpa != NULL)
2128                                 setie(wpa_ie, wpa - sfrm);
2129                         if (rsn != NULL)
2130                                 setie(rsn_ie, rsn - sfrm);
2131                         if (htcap != NULL)
2132                                 setie(htcap_ie, htcap - sfrm);
2133                         if (wme != NULL) {
2134                                 setie(wme_ie, wme - sfrm);
2135                                 /*
2136                                  * Mark node as capable of QoS.
2137                                  */
2138                                 ni->ni_flags |= IEEE80211_NODE_QOS;
2139                         } else
2140                                 ni->ni_flags &= ~IEEE80211_NODE_QOS;
2141 #ifdef IEEE80211_SUPPORT_SUPERG
2142                         if (ath != NULL) {
2143                                 setie(ath_ie, ath - sfrm);
2144                                 /*
2145                                  * Parse ATH station parameters.
2146                                  */
2147                                 ieee80211_parse_ath(ni, ni->ni_ies.ath_ie);
2148                         } else
2149 #endif
2150                                 ni->ni_ath_flags = 0;
2151 #undef setie
2152                 } else {
2153                         ni->ni_flags &= ~IEEE80211_NODE_QOS;
2154                         ni->ni_ath_flags = 0;
2155                 }
2156                 ieee80211_node_join(ni, resp);
2157                 ieee80211_deliver_l2uf(ni);
2158                 break;
2159         }
2160
2161         case IEEE80211_FC0_SUBTYPE_DEAUTH:
2162         case IEEE80211_FC0_SUBTYPE_DISASSOC: {
2163                 uint16_t reason;
2164
2165                 if (vap->iv_state != IEEE80211_S_RUN ||
2166                     /* NB: can happen when in promiscuous mode */
2167                     !IEEE80211_ADDR_EQ(wh->i_addr1, vap->iv_myaddr)) {
2168                         vap->iv_stats.is_rx_mgtdiscard++;
2169                         break;
2170                 }
2171                 /*
2172                  * deauth/disassoc frame format
2173                  *      [2] reason
2174                  */
2175                 IEEE80211_VERIFY_LENGTH(efrm - frm, 2, return);
2176                 reason = le16toh(*(uint16_t *)frm);
2177                 if (subtype == IEEE80211_FC0_SUBTYPE_DEAUTH) {
2178                         vap->iv_stats.is_rx_deauth++;
2179                         IEEE80211_NODE_STAT(ni, rx_deauth);
2180                 } else {
2181                         vap->iv_stats.is_rx_disassoc++;
2182                         IEEE80211_NODE_STAT(ni, rx_disassoc);
2183                 }
2184                 IEEE80211_NOTE(vap, IEEE80211_MSG_AUTH, ni,
2185                     "recv %s (reason %d)", ieee80211_mgt_subtype_name[subtype >>
2186                         IEEE80211_FC0_SUBTYPE_SHIFT], reason);
2187                 if (ni != vap->iv_bss)
2188                         ieee80211_node_leave(ni);
2189                 break;
2190         }
2191
2192         case IEEE80211_FC0_SUBTYPE_ACTION:
2193         case IEEE80211_FC0_SUBTYPE_ACTION_NOACK:
2194                 if (ni == vap->iv_bss) {
2195                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
2196                             wh, NULL, "%s", "unknown node");
2197                         vap->iv_stats.is_rx_mgtdiscard++;
2198                 } else if (!IEEE80211_ADDR_EQ(vap->iv_myaddr, wh->i_addr1) &&
2199                     !IEEE80211_IS_MULTICAST(wh->i_addr1)) {
2200                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
2201                             wh, NULL, "%s", "not for us");
2202                         vap->iv_stats.is_rx_mgtdiscard++;
2203                 } else if (vap->iv_state != IEEE80211_S_RUN) {
2204                         IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
2205                             wh, NULL, "wrong state %s",
2206                             ieee80211_state_name[vap->iv_state]);
2207                         vap->iv_stats.is_rx_mgtdiscard++;
2208                 } else {
2209                         if (ieee80211_parse_action(ni, m0) == 0)
2210                                 (void)ic->ic_recv_action(ni, wh, frm, efrm);
2211                 }
2212                 break;
2213
2214         case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
2215         case IEEE80211_FC0_SUBTYPE_REASSOC_RESP:
2216         case IEEE80211_FC0_SUBTYPE_ATIM:
2217                 IEEE80211_DISCARD(vap, IEEE80211_MSG_INPUT,
2218                     wh, NULL, "%s", "not handled");
2219                 vap->iv_stats.is_rx_mgtdiscard++;
2220                 break;
2221
2222         default:
2223                 IEEE80211_DISCARD(vap, IEEE80211_MSG_ANY,
2224                     wh, "mgt", "subtype 0x%x not handled", subtype);
2225                 vap->iv_stats.is_rx_badsubtype++;
2226                 break;
2227         }
2228 }
2229
2230 static void
2231 hostap_recv_ctl(struct ieee80211_node *ni, struct mbuf *m, int subtype)
2232 {
2233         switch (subtype) {
2234         case IEEE80211_FC0_SUBTYPE_PS_POLL:
2235                 hostap_recv_pspoll(ni, m);
2236                 break;
2237         case IEEE80211_FC0_SUBTYPE_BAR:
2238                 ieee80211_recv_bar(ni, m);
2239                 break;
2240         }
2241 }
2242
2243 /*
2244  * Process a received ps-poll frame.
2245  */
2246 static void
2247 hostap_recv_pspoll(struct ieee80211_node *ni, struct mbuf *m0)
2248 {
2249         struct ieee80211vap *vap = ni->ni_vap;
2250         struct ieee80211_frame_min *wh;
2251         struct ifnet *ifp;
2252         struct mbuf *m;
2253         uint16_t aid;
2254         int qlen;
2255
2256         wh = mtod(m0, struct ieee80211_frame_min *);
2257         if (ni->ni_associd == 0) {
2258                 IEEE80211_DISCARD(vap,
2259                     IEEE80211_MSG_POWER | IEEE80211_MSG_DEBUG,
2260                     (struct ieee80211_frame *) wh, NULL,
2261                     "%s", "unassociated station");
2262                 vap->iv_stats.is_ps_unassoc++;
2263                 IEEE80211_SEND_MGMT(ni, IEEE80211_FC0_SUBTYPE_DEAUTH,
2264                         IEEE80211_REASON_NOT_ASSOCED);
2265                 return;
2266         }
2267
2268         aid = le16toh(*(uint16_t *)wh->i_dur);
2269         if (aid != ni->ni_associd) {
2270                 IEEE80211_DISCARD(vap,
2271                     IEEE80211_MSG_POWER | IEEE80211_MSG_DEBUG,
2272                     (struct ieee80211_frame *) wh, NULL,
2273                     "aid mismatch: sta aid 0x%x poll aid 0x%x",
2274                     ni->ni_associd, aid);
2275                 vap->iv_stats.is_ps_badaid++;
2276                 /*
2277                  * NB: We used to deauth the station but it turns out
2278                  * the Blackberry Curve 8230 (and perhaps other devices)
2279                  * sometimes send the wrong AID when WME is negotiated.
2280                  * Being more lenient here seems ok as we already check
2281                  * the station is associated and we only return frames
2282                  * queued for the station (i.e. we don't use the AID).
2283                  */
2284                 return;
2285         }
2286
2287         /* Okay, take the first queued packet and put it out... */
2288         m = ieee80211_node_psq_dequeue(ni, &qlen);
2289         if (m == NULL) {
2290                 IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_POWER, wh->i_addr2,
2291                     "%s", "recv ps-poll, but queue empty");
2292                 ieee80211_send_nulldata(ieee80211_ref_node(ni));
2293                 vap->iv_stats.is_ps_qempty++;   /* XXX node stat */
2294                 if (vap->iv_set_tim != NULL)
2295                         vap->iv_set_tim(ni, 0); /* just in case */
2296                 return;
2297         }
2298         /*
2299          * If there are more packets, set the more packets bit
2300          * in the packet dispatched to the station; otherwise
2301          * turn off the TIM bit.
2302          */
2303         if (qlen != 0) {
2304                 IEEE80211_NOTE(vap, IEEE80211_MSG_POWER, ni,
2305                     "recv ps-poll, send packet, %u still queued", qlen);
2306                 m->m_flags |= M_MORE_DATA;
2307         } else {
2308                 IEEE80211_NOTE(vap, IEEE80211_MSG_POWER, ni,
2309                     "%s", "recv ps-poll, send packet, queue empty");
2310                 if (vap->iv_set_tim != NULL)
2311                         vap->iv_set_tim(ni, 0);
2312         }
2313         m->m_flags |= M_PWR_SAV;                /* bypass PS handling */
2314
2315         if (m->m_flags & M_ENCAP)
2316                 ifp = vap->iv_ic->ic_ifp;
2317         else
2318                 ifp = vap->iv_ifp;
2319         IF_ENQUEUE(&ifp->if_snd, m);
2320         if_start(ifp);
2321 }