kernel: Move us to using M_NOWAIT and M_WAITOK for mbuf functions.
[dragonfly.git] / sys / netproto / 802_11 / wlan / ieee80211_input.c
1 /*-
2  * Copyright (c) 2001 Atsushi Onoe
3  * Copyright (c) 2002-2009 Sam Leffler, Errno Consulting
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29
30 #include "opt_wlan.h"
31
32 #include <sys/param.h>
33 #include <sys/systm.h>
34 #include <sys/mbuf.h>   
35 #include <sys/malloc.h>
36 #include <sys/endian.h>
37 #include <sys/kernel.h>
38  
39 #include <sys/socket.h>
40  
41 #include <net/ethernet.h>
42 #include <net/if.h>
43 #include <net/if_var.h>
44 #include <net/if_llc.h>
45 #include <net/if_media.h>
46 #include <net/vlan/if_vlan_var.h>
47
48 #include <netproto/802_11/ieee80211_var.h>
49 #include <netproto/802_11/ieee80211_input.h>
50 #ifdef IEEE80211_SUPPORT_MESH
51 #include <netproto/802_11/ieee80211_mesh.h>
52 #endif
53
54 #include <net/bpf.h>
55
56 #ifdef INET
57 #include <netinet/in.h>
58 #include <net/ethernet.h>
59 #endif
60
61 static void
62 ieee80211_process_mimo(struct ieee80211_node *ni, struct ieee80211_rx_stats *rx)
63 {
64         int i;
65
66         /* Verify the required MIMO bits are set */
67         if ((rx->r_flags & (IEEE80211_R_C_CHAIN | IEEE80211_R_C_NF | IEEE80211_R_C_RSSI)) !=
68             (IEEE80211_R_C_CHAIN | IEEE80211_R_C_NF | IEEE80211_R_C_RSSI))
69                 return;
70
71         /* XXX This assumes the MIMO radios have both ctl and ext chains */
72         for (i = 0; i < MIN(rx->c_chain, IEEE80211_MAX_CHAINS); i++) {
73                 IEEE80211_RSSI_LPF(ni->ni_mimo_rssi_ctl[i], rx->c_rssi_ctl[i]);
74                 IEEE80211_RSSI_LPF(ni->ni_mimo_rssi_ext[i], rx->c_rssi_ext[i]);
75         }
76
77         /* XXX This also assumes the MIMO radios have both ctl and ext chains */
78         for(i = 0; i < MIN(rx->c_chain, IEEE80211_MAX_CHAINS); i++) {
79                 ni->ni_mimo_noise_ctl[i] = rx->c_nf_ctl[i];
80                 ni->ni_mimo_noise_ext[i] = rx->c_nf_ext[i];
81         }
82         ni->ni_mimo_chains = rx->c_chain;
83 }
84
85 int
86 ieee80211_input_mimo(struct ieee80211_node *ni, struct mbuf *m,
87     struct ieee80211_rx_stats *rx)
88 {
89         /* XXX should assert IEEE80211_R_NF and IEEE80211_R_RSSI are set */
90         ieee80211_process_mimo(ni, rx);
91         return ieee80211_input(ni, m, rx->rssi, rx->nf);
92 }
93
94 int
95 ieee80211_input_all(struct ieee80211com *ic, struct mbuf *m, int rssi, int nf)
96 {
97         struct ieee80211_rx_stats rx;
98
99         rx.r_flags = IEEE80211_R_NF | IEEE80211_R_RSSI;
100         rx.nf = nf;
101         rx.rssi = rssi;
102         return ieee80211_input_mimo_all(ic, m, &rx);
103 }
104
105 int
106 ieee80211_input_mimo_all(struct ieee80211com *ic, struct mbuf *m,
107     struct ieee80211_rx_stats *rx)
108 {
109         struct ieee80211vap *vap;
110         int type = -1;
111
112         m->m_flags |= M_BCAST;          /* NB: mark for bpf tap'ing */
113
114         /* XXX locking */
115         TAILQ_FOREACH(vap, &ic->ic_vaps, iv_next) {
116                 struct ieee80211_node *ni;
117                 struct mbuf *mcopy;
118
119                 /* NB: could check for IFF_UP but this is cheaper */
120                 if (vap->iv_state == IEEE80211_S_INIT)
121                         continue;
122                 /*
123                  * WDS vap's only receive directed traffic from the
124                  * station at the ``far end''.  That traffic should
125                  * be passed through the AP vap the station is associated
126                  * to--so don't spam them with mcast frames.
127                  */
128                 if (vap->iv_opmode == IEEE80211_M_WDS)
129                         continue;
130                 if (TAILQ_NEXT(vap, iv_next) != NULL) {
131                         /*
132                          * Packet contents are changed by ieee80211_decap
133                          * so do a deep copy of the packet.
134                          */
135                         mcopy = m_dup(m, M_NOWAIT);
136                         if (mcopy == NULL) {
137                                 /* XXX stat+msg */
138                                 continue;
139                         }
140                 } else {
141                         mcopy = m;
142                         m = NULL;
143                 }
144                 ni = ieee80211_ref_node(vap->iv_bss);
145                 type = ieee80211_input_mimo(ni, mcopy, rx);
146                 ieee80211_free_node(ni);
147         }
148         if (m != NULL)                  /* no vaps, reclaim mbuf */
149                 m_freem(m);
150         return type;
151 }
152
153 /*
154  * This function reassembles fragments.
155  *
156  * XXX should handle 3 concurrent reassemblies per-spec.
157  */
158 struct mbuf *
159 ieee80211_defrag(struct ieee80211_node *ni, struct mbuf *m, int hdrspace)
160 {
161         struct ieee80211vap *vap = ni->ni_vap;
162         struct ieee80211_frame *wh = mtod(m, struct ieee80211_frame *);
163         struct ieee80211_frame *lwh;
164         uint16_t rxseq;
165         uint8_t fragno;
166         uint8_t more_frag = wh->i_fc[1] & IEEE80211_FC1_MORE_FRAG;
167         struct mbuf *mfrag;
168
169         KASSERT(!IEEE80211_IS_MULTICAST(wh->i_addr1), ("multicast fragm?"));
170
171         rxseq = le16toh(*(uint16_t *)wh->i_seq);
172         fragno = rxseq & IEEE80211_SEQ_FRAG_MASK;
173
174         /* Quick way out, if there's nothing to defragment */
175         if (!more_frag && fragno == 0 && ni->ni_rxfrag[0] == NULL)
176                 return m;
177
178         /*
179          * Remove frag to insure it doesn't get reaped by timer.
180          */
181         if (ni->ni_table == NULL) {
182                 /*
183                  * Should never happen.  If the node is orphaned (not in
184                  * the table) then input packets should not reach here.
185                  * Otherwise, a concurrent request that yanks the table
186                  * should be blocked by other interlocking and/or by first
187                  * shutting the driver down.  Regardless, be defensive
188                  * here and just bail
189                  */
190                 /* XXX need msg+stat */
191                 m_freem(m);
192                 return NULL;
193         }
194         IEEE80211_NODE_LOCK(ni->ni_table);
195         mfrag = ni->ni_rxfrag[0];
196         ni->ni_rxfrag[0] = NULL;
197         IEEE80211_NODE_UNLOCK(ni->ni_table);
198
199         /*
200          * Validate new fragment is in order and
201          * related to the previous ones.
202          */
203         if (mfrag != NULL) {
204                 uint16_t last_rxseq;
205
206                 lwh = mtod(mfrag, struct ieee80211_frame *);
207                 last_rxseq = le16toh(*(uint16_t *)lwh->i_seq);
208                 /* NB: check seq # and frag together */
209                 if (rxseq != last_rxseq+1 ||
210                     !IEEE80211_ADDR_EQ(wh->i_addr1, lwh->i_addr1) ||
211                     !IEEE80211_ADDR_EQ(wh->i_addr2, lwh->i_addr2)) {
212                         /*
213                          * Unrelated fragment or no space for it,
214                          * clear current fragments.
215                          */
216                         m_freem(mfrag);
217                         mfrag = NULL;
218                 }
219         }
220
221         if (mfrag == NULL) {
222                 if (fragno != 0) {              /* !first fragment, discard */
223                         vap->iv_stats.is_rx_defrag++;
224                         IEEE80211_NODE_STAT(ni, rx_defrag);
225                         m_freem(m);
226                         return NULL;
227                 }
228                 mfrag = m;
229         } else {                                /* concatenate */
230                 m_adj(m, hdrspace);             /* strip header */
231                 m_cat(mfrag, m);
232                 /* NB: m_cat doesn't update the packet header */
233                 mfrag->m_pkthdr.len += m->m_pkthdr.len;
234                 /* track last seqnum and fragno */
235                 lwh = mtod(mfrag, struct ieee80211_frame *);
236                 *(uint16_t *) lwh->i_seq = *(uint16_t *) wh->i_seq;
237         }
238         if (more_frag) {                        /* more to come, save */
239                 ni->ni_rxfragstamp = ticks;
240                 ni->ni_rxfrag[0] = mfrag;
241                 mfrag = NULL;
242         }
243         return mfrag;
244 }
245
246 void
247 ieee80211_deliver_data(struct ieee80211vap *vap,
248         struct ieee80211_node *ni, struct mbuf *m)
249 {
250         struct ether_header *eh = mtod(m, struct ether_header *);
251         struct ifnet *ifp = vap->iv_ifp;
252
253         /* clear driver/net80211 flags before passing up */
254         m->m_flags &= ~(M_MCAST | M_BCAST);
255 #if __FreeBSD_version >= 1000046
256         m_clrprotoflags(m);
257 #endif
258
259         /* NB: see hostap_deliver_data, this path doesn't handle hostap */
260         KASSERT(vap->iv_opmode != IEEE80211_M_HOSTAP, ("gack, hostap"));
261         /*
262          * Do accounting.
263          */
264         IFNET_STAT_INC(ifp, ipackets, 1);
265         IEEE80211_NODE_STAT(ni, rx_data);
266         IEEE80211_NODE_STAT_ADD(ni, rx_bytes, m->m_pkthdr.len);
267         if (ETHER_IS_MULTICAST(eh->ether_dhost)) {
268                 m->m_flags |= M_MCAST;          /* XXX M_BCAST? */
269                 IEEE80211_NODE_STAT(ni, rx_mcast);
270         } else
271                 IEEE80211_NODE_STAT(ni, rx_ucast);
272         m->m_pkthdr.rcvif = ifp;
273
274         if (ni->ni_vlan != 0) {
275                 /* attach vlan tag */
276 #if defined(__DragonFly__)
277                 m->m_pkthdr.ether_vlantag = ni->ni_vlan;
278 #else
279                 m->m_pkthdr.ether_vtag = ni->ni_vlan;
280 #endif
281                 m->m_flags |= M_VLANTAG;
282         }
283         ifp->if_input(ifp, m, NULL, -1);
284 }
285
286 struct mbuf *
287 ieee80211_decap(struct ieee80211vap *vap, struct mbuf *m, int hdrlen)
288 {
289         struct ieee80211_qosframe_addr4 wh;
290         struct ether_header *eh;
291         struct llc *llc;
292
293         KASSERT(hdrlen <= sizeof(wh),
294             ("hdrlen %d > max %zd", hdrlen, sizeof(wh)));
295
296         if (m->m_len < hdrlen + sizeof(*llc) &&
297             (m = m_pullup(m, hdrlen + sizeof(*llc))) == NULL) {
298                 vap->iv_stats.is_rx_tooshort++;
299                 /* XXX msg */
300                 return NULL;
301         }
302         memcpy(&wh, mtod(m, caddr_t), hdrlen);
303         llc = (struct llc *)(mtod(m, caddr_t) + hdrlen);
304         if (llc->llc_dsap == LLC_SNAP_LSAP && llc->llc_ssap == LLC_SNAP_LSAP &&
305             llc->llc_control == LLC_UI && llc->llc_snap.org_code[0] == 0 &&
306             llc->llc_snap.org_code[1] == 0 && llc->llc_snap.org_code[2] == 0 &&
307             /* NB: preserve AppleTalk frames that have a native SNAP hdr */
308             !(llc->llc_snap.ether_type == htons(ETHERTYPE_AARP) ||
309               llc->llc_snap.ether_type == htons(ETHERTYPE_IPX))) {
310                 m_adj(m, hdrlen + sizeof(struct llc) - sizeof(*eh));
311                 llc = NULL;
312         } else {
313                 m_adj(m, hdrlen - sizeof(*eh));
314         }
315         eh = mtod(m, struct ether_header *);
316         switch (wh.i_fc[1] & IEEE80211_FC1_DIR_MASK) {
317         case IEEE80211_FC1_DIR_NODS:
318                 IEEE80211_ADDR_COPY(eh->ether_dhost, wh.i_addr1);
319                 IEEE80211_ADDR_COPY(eh->ether_shost, wh.i_addr2);
320                 break;
321         case IEEE80211_FC1_DIR_TODS:
322                 IEEE80211_ADDR_COPY(eh->ether_dhost, wh.i_addr3);
323                 IEEE80211_ADDR_COPY(eh->ether_shost, wh.i_addr2);
324                 break;
325         case IEEE80211_FC1_DIR_FROMDS:
326                 IEEE80211_ADDR_COPY(eh->ether_dhost, wh.i_addr1);
327                 IEEE80211_ADDR_COPY(eh->ether_shost, wh.i_addr3);
328                 break;
329         case IEEE80211_FC1_DIR_DSTODS:
330                 IEEE80211_ADDR_COPY(eh->ether_dhost, wh.i_addr3);
331                 IEEE80211_ADDR_COPY(eh->ether_shost, wh.i_addr4);
332                 break;
333         }
334 #ifndef __NO_STRICT_ALIGNMENT
335         if (!ALIGNED_POINTER(mtod(m, caddr_t) + sizeof(*eh), uint32_t)) {
336                 m = ieee80211_realign(vap, m, sizeof(*eh));
337                 if (m == NULL)
338                         return NULL;
339         }
340 #endif /* !__NO_STRICT_ALIGNMENT */
341         if (llc != NULL) {
342                 eh = mtod(m, struct ether_header *);
343                 eh->ether_type = htons(m->m_pkthdr.len - sizeof(*eh));
344         }
345         return m;
346 }
347
348 /*
349  * Decap a frame encapsulated in a fast-frame/A-MSDU.
350  */
351 struct mbuf *
352 ieee80211_decap1(struct mbuf *m, int *framelen)
353 {
354 #define FF_LLC_SIZE     (sizeof(struct ether_header) + sizeof(struct llc))
355         struct ether_header *eh;
356         struct llc *llc;
357
358         /*
359          * The frame has an 802.3 header followed by an 802.2
360          * LLC header.  The encapsulated frame length is in the
361          * first header type field; save that and overwrite it 
362          * with the true type field found in the second.  Then
363          * copy the 802.3 header up to where it belongs and
364          * adjust the mbuf contents to remove the void.
365          */
366         if (m->m_len < FF_LLC_SIZE && (m = m_pullup(m, FF_LLC_SIZE)) == NULL)
367                 return NULL;
368         eh = mtod(m, struct ether_header *);    /* 802.3 header is first */
369         llc = (struct llc *)&eh[1];             /* 802.2 header follows */
370         *framelen = ntohs(eh->ether_type)       /* encap'd frame size */
371                   + sizeof(struct ether_header) - sizeof(struct llc);
372         eh->ether_type = llc->llc_un.type_snap.ether_type;
373         ovbcopy(eh, mtod(m, uint8_t *) + sizeof(struct llc),
374                 sizeof(struct ether_header));
375         m_adj(m, sizeof(struct llc));
376         return m;
377 #undef FF_LLC_SIZE
378 }
379
380 /*
381  * Install received rate set information in the node's state block.
382  */
383 int
384 ieee80211_setup_rates(struct ieee80211_node *ni,
385         const uint8_t *rates, const uint8_t *xrates, int flags)
386 {
387         struct ieee80211vap *vap = ni->ni_vap;
388         struct ieee80211_rateset *rs = &ni->ni_rates;
389
390         memset(rs, 0, sizeof(*rs));
391         rs->rs_nrates = rates[1];
392         memcpy(rs->rs_rates, rates + 2, rs->rs_nrates);
393         if (xrates != NULL) {
394                 uint8_t nxrates;
395                 /*
396                  * Tack on 11g extended supported rate element.
397                  */
398                 nxrates = xrates[1];
399                 if (rs->rs_nrates + nxrates > IEEE80211_RATE_MAXSIZE) {
400                         nxrates = IEEE80211_RATE_MAXSIZE - rs->rs_nrates;
401                         IEEE80211_NOTE(vap, IEEE80211_MSG_XRATE, ni,
402                             "extended rate set too large; only using "
403                             "%u of %u rates", nxrates, xrates[1]);
404                         vap->iv_stats.is_rx_rstoobig++;
405                 }
406                 memcpy(rs->rs_rates + rs->rs_nrates, xrates+2, nxrates);
407                 rs->rs_nrates += nxrates;
408         }
409         return ieee80211_fix_rate(ni, rs, flags);
410 }
411
412 /*
413  * Send a management frame error response to the specified
414  * station.  If ni is associated with the station then use
415  * it; otherwise allocate a temporary node suitable for
416  * transmitting the frame and then free the reference so
417  * it will go away as soon as the frame has been transmitted.
418  */
419 void
420 ieee80211_send_error(struct ieee80211_node *ni,
421         const uint8_t mac[IEEE80211_ADDR_LEN], int subtype, int arg)
422 {
423         struct ieee80211vap *vap = ni->ni_vap;
424         int istmp;
425
426         if (ni == vap->iv_bss) {
427                 if (vap->iv_state != IEEE80211_S_RUN) {
428                         /*
429                          * XXX hack until we get rid of this routine.
430                          * We can be called prior to the vap reaching
431                          * run state under certain conditions in which
432                          * case iv_bss->ni_chan will not be setup.
433                          * Check for this explicitly and and just ignore
434                          * the request.
435                          */
436                         return;
437                 }
438                 ni = ieee80211_tmp_node(vap, mac);
439                 if (ni == NULL) {
440                         /* XXX msg */
441                         return;
442                 }
443                 istmp = 1;
444         } else
445                 istmp = 0;
446         IEEE80211_SEND_MGMT(ni, subtype, arg);
447         if (istmp)
448                 ieee80211_free_node(ni);
449 }
450
451 int
452 ieee80211_alloc_challenge(struct ieee80211_node *ni)
453 {
454         if (ni->ni_challenge == NULL)
455                 ni->ni_challenge = (uint32_t *) kmalloc(IEEE80211_CHALLENGE_LEN,
456                     M_80211_NODE, M_INTWAIT);
457         if (ni->ni_challenge == NULL) {
458                 IEEE80211_NOTE(ni->ni_vap,
459                     IEEE80211_MSG_DEBUG | IEEE80211_MSG_AUTH, ni,
460                     "%s", "shared key challenge alloc failed");
461                 /* XXX statistic */
462         }
463         return (ni->ni_challenge != NULL);
464 }
465
466 /*
467  * Parse a Beacon or ProbeResponse frame and return the
468  * useful information in an ieee80211_scanparams structure.
469  * Status is set to 0 if no problems were found; otherwise
470  * a bitmask of IEEE80211_BPARSE_* items is returned that
471  * describes the problems detected.
472  */
473 int
474 ieee80211_parse_beacon(struct ieee80211_node *ni, struct mbuf *m,
475         struct ieee80211_scanparams *scan)
476 {
477         struct ieee80211vap *vap = ni->ni_vap;
478         struct ieee80211com *ic = ni->ni_ic;
479         struct ieee80211_frame *wh;
480         uint8_t *frm, *efrm;
481
482         wh = mtod(m, struct ieee80211_frame *);
483         frm = (uint8_t *)&wh[1];
484         efrm = mtod(m, uint8_t *) + m->m_len;
485         scan->status = 0;
486         /*
487          * beacon/probe response frame format
488          *      [8] time stamp
489          *      [2] beacon interval
490          *      [2] capability information
491          *      [tlv] ssid
492          *      [tlv] supported rates
493          *      [tlv] country information
494          *      [tlv] channel switch announcement (CSA)
495          *      [tlv] parameter set (FH/DS)
496          *      [tlv] erp information
497          *      [tlv] extended supported rates
498          *      [tlv] WME
499          *      [tlv] WPA or RSN
500          *      [tlv] HT capabilities
501          *      [tlv] HT information
502          *      [tlv] Atheros capabilities
503          *      [tlv] Mesh ID
504          *      [tlv] Mesh Configuration
505          */
506         IEEE80211_VERIFY_LENGTH(efrm - frm, 12,
507             return (scan->status = IEEE80211_BPARSE_BADIELEN));
508         memset(scan, 0, sizeof(*scan));
509         scan->tstamp  = frm;                            frm += 8;
510         scan->bintval = le16toh(*(uint16_t *)frm);      frm += 2;
511         scan->capinfo = le16toh(*(uint16_t *)frm);      frm += 2;
512         scan->bchan = ieee80211_chan2ieee(ic, ic->ic_curchan);
513         scan->chan = scan->bchan;
514         scan->ies = frm;
515         scan->ies_len = efrm - frm;
516
517         while (efrm - frm > 1) {
518                 IEEE80211_VERIFY_LENGTH(efrm - frm, frm[1] + 2,
519                     return (scan->status = IEEE80211_BPARSE_BADIELEN));
520                 switch (*frm) {
521                 case IEEE80211_ELEMID_SSID:
522                         scan->ssid = frm;
523                         break;
524                 case IEEE80211_ELEMID_RATES:
525                         scan->rates = frm;
526                         break;
527                 case IEEE80211_ELEMID_COUNTRY:
528                         scan->country = frm;
529                         break;
530                 case IEEE80211_ELEMID_CSA:
531                         scan->csa = frm;
532                         break;
533                 case IEEE80211_ELEMID_QUIET:
534                         scan->quiet = frm;
535                         break;
536                 case IEEE80211_ELEMID_FHPARMS:
537                         if (ic->ic_phytype == IEEE80211_T_FH) {
538                                 scan->fhdwell = LE_READ_2(&frm[2]);
539                                 scan->chan = IEEE80211_FH_CHAN(frm[4], frm[5]);
540                                 scan->fhindex = frm[6];
541                         }
542                         break;
543                 case IEEE80211_ELEMID_DSPARMS:
544                         /*
545                          * XXX hack this since depending on phytype
546                          * is problematic for multi-mode devices.
547                          */
548                         if (ic->ic_phytype != IEEE80211_T_FH)
549                                 scan->chan = frm[2];
550                         break;
551                 case IEEE80211_ELEMID_TIM:
552                         /* XXX ATIM? */
553                         scan->tim = frm;
554                         scan->timoff = frm - mtod(m, uint8_t *);
555                         break;
556                 case IEEE80211_ELEMID_IBSSPARMS:
557                 case IEEE80211_ELEMID_CFPARMS:
558                 case IEEE80211_ELEMID_PWRCNSTR:
559                         /* NB: avoid debugging complaints */
560                         break;
561                 case IEEE80211_ELEMID_XRATES:
562                         scan->xrates = frm;
563                         break;
564                 case IEEE80211_ELEMID_ERP:
565                         if (frm[1] != 1) {
566                                 IEEE80211_DISCARD_IE(vap,
567                                     IEEE80211_MSG_ELEMID, wh, "ERP",
568                                     "bad len %u", frm[1]);
569                                 vap->iv_stats.is_rx_elem_toobig++;
570                                 break;
571                         }
572                         scan->erp = frm[2] | 0x100;
573                         break;
574                 case IEEE80211_ELEMID_HTCAP:
575                         scan->htcap = frm;
576                         break;
577                 case IEEE80211_ELEMID_RSN:
578                         scan->rsn = frm;
579                         break;
580                 case IEEE80211_ELEMID_HTINFO:
581                         scan->htinfo = frm;
582                         break;
583 #ifdef IEEE80211_SUPPORT_MESH
584                 case IEEE80211_ELEMID_MESHID:
585                         scan->meshid = frm;
586                         break;
587                 case IEEE80211_ELEMID_MESHCONF:
588                         scan->meshconf = frm;
589                         break;
590 #endif
591                 case IEEE80211_ELEMID_VENDOR:
592                         if (iswpaoui(frm))
593                                 scan->wpa = frm;
594                         else if (iswmeparam(frm) || iswmeinfo(frm))
595                                 scan->wme = frm;
596 #ifdef IEEE80211_SUPPORT_SUPERG
597                         else if (isatherosoui(frm))
598                                 scan->ath = frm;
599 #endif
600 #ifdef IEEE80211_SUPPORT_TDMA
601                         else if (istdmaoui(frm))
602                                 scan->tdma = frm;
603 #endif
604                         else if (vap->iv_flags_ht & IEEE80211_FHT_HTCOMPAT) {
605                                 /*
606                                  * Accept pre-draft HT ie's if the
607                                  * standard ones have not been seen.
608                                  */
609                                 if (ishtcapoui(frm)) {
610                                         if (scan->htcap == NULL)
611                                                 scan->htcap = frm;
612                                 } else if (ishtinfooui(frm)) {
613                                         if (scan->htinfo == NULL)
614                                                 scan->htcap = frm;
615                                 }
616                         }
617                         break;
618                 default:
619                         IEEE80211_DISCARD_IE(vap, IEEE80211_MSG_ELEMID,
620                             wh, "unhandled",
621                             "id %u, len %u", *frm, frm[1]);
622                         vap->iv_stats.is_rx_elem_unknown++;
623                         break;
624                 }
625                 frm += frm[1] + 2;
626         }
627         IEEE80211_VERIFY_ELEMENT(scan->rates, IEEE80211_RATE_MAXSIZE,
628             scan->status |= IEEE80211_BPARSE_RATES_INVALID);
629         if (scan->rates != NULL && scan->xrates != NULL) {
630                 /*
631                  * NB: don't process XRATES if RATES is missing.  This
632                  * avoids a potential null ptr deref and should be ok
633                  * as the return code will already note RATES is missing
634                  * (so callers shouldn't otherwise process the frame).
635                  */
636                 IEEE80211_VERIFY_ELEMENT(scan->xrates,
637                     IEEE80211_RATE_MAXSIZE - scan->rates[1],
638                     scan->status |= IEEE80211_BPARSE_XRATES_INVALID);
639         }
640         IEEE80211_VERIFY_ELEMENT(scan->ssid, IEEE80211_NWID_LEN,
641             scan->status |= IEEE80211_BPARSE_SSID_INVALID);
642         if (scan->chan != scan->bchan && ic->ic_phytype != IEEE80211_T_FH) {
643                 /*
644                  * Frame was received on a channel different from the
645                  * one indicated in the DS params element id;
646                  * silently discard it.
647                  *
648                  * NB: this can happen due to signal leakage.
649                  *     But we should take it for FH phy because
650                  *     the rssi value should be correct even for
651                  *     different hop pattern in FH.
652                  */
653                 IEEE80211_DISCARD(vap,
654                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_INPUT,
655                     wh, NULL, "for off-channel %u", scan->chan);
656                 vap->iv_stats.is_rx_chanmismatch++;
657                 scan->status |= IEEE80211_BPARSE_OFFCHAN;
658         }
659         if (!(IEEE80211_BINTVAL_MIN <= scan->bintval &&
660               scan->bintval <= IEEE80211_BINTVAL_MAX)) {
661                 IEEE80211_DISCARD(vap,
662                     IEEE80211_MSG_ELEMID | IEEE80211_MSG_INPUT,
663                     wh, NULL, "bogus beacon interval (%d TU)",
664                     (int) scan->bintval);
665                 vap->iv_stats.is_rx_badbintval++;
666                 scan->status |= IEEE80211_BPARSE_BINTVAL_INVALID;
667         }
668         if (scan->country != NULL) {
669                 /*
670                  * Validate we have at least enough data to extract
671                  * the country code.  Not sure if we should return an
672                  * error instead of discarding the IE; consider this
673                  * being lenient as we don't depend on the data for
674                  * correct operation.
675                  */
676                 IEEE80211_VERIFY_LENGTH(scan->country[1], 3 * sizeof(uint8_t),
677                     scan->country = NULL);
678         }
679         if (scan->csa != NULL) {
680                 /*
681                  * Validate Channel Switch Announcement; this must
682                  * be the correct length or we toss the frame.
683                  */
684                 IEEE80211_VERIFY_LENGTH(scan->csa[1], 3 * sizeof(uint8_t),
685                     scan->status |= IEEE80211_BPARSE_CSA_INVALID);
686         }
687         /*
688          * Process HT ie's.  This is complicated by our
689          * accepting both the standard ie's and the pre-draft
690          * vendor OUI ie's that some vendors still use/require.
691          */
692         if (scan->htcap != NULL) {
693                 IEEE80211_VERIFY_LENGTH(scan->htcap[1],
694                      scan->htcap[0] == IEEE80211_ELEMID_VENDOR ?
695                          4 + sizeof(struct ieee80211_ie_htcap)-2 :
696                          sizeof(struct ieee80211_ie_htcap)-2,
697                      scan->htcap = NULL);
698         }
699         if (scan->htinfo != NULL) {
700                 IEEE80211_VERIFY_LENGTH(scan->htinfo[1],
701                      scan->htinfo[0] == IEEE80211_ELEMID_VENDOR ?
702                          4 + sizeof(struct ieee80211_ie_htinfo)-2 :
703                          sizeof(struct ieee80211_ie_htinfo)-2,
704                      scan->htinfo = NULL);
705         }
706         return scan->status;
707 }
708
709 /*
710  * Parse an Action frame.  Return 0 on success, non-zero on failure.
711  */
712 int
713 ieee80211_parse_action(struct ieee80211_node *ni, struct mbuf *m)
714 {
715         struct ieee80211vap *vap = ni->ni_vap;
716         const struct ieee80211_action *ia;
717         struct ieee80211_frame *wh;
718         uint8_t *frm, *efrm;
719
720         /*
721          * action frame format:
722          *      [1] category
723          *      [1] action
724          *      [tlv] parameters
725          */
726         wh = mtod(m, struct ieee80211_frame *);
727         frm = (u_int8_t *)&wh[1];
728         efrm = mtod(m, u_int8_t *) + m->m_len;
729         IEEE80211_VERIFY_LENGTH(efrm - frm,
730                 sizeof(struct ieee80211_action), return EINVAL);
731         ia = (const struct ieee80211_action *) frm;
732
733         vap->iv_stats.is_rx_action++;
734         IEEE80211_NODE_STAT(ni, rx_action);
735
736         /* verify frame payloads but defer processing */
737         switch (ia->ia_category) {
738         case IEEE80211_ACTION_CAT_BA:
739                 switch (ia->ia_action) {
740                 case IEEE80211_ACTION_BA_ADDBA_REQUEST:
741                         IEEE80211_VERIFY_LENGTH(efrm - frm,
742                             sizeof(struct ieee80211_action_ba_addbarequest),
743                             return EINVAL);
744                         break;
745                 case IEEE80211_ACTION_BA_ADDBA_RESPONSE:
746                         IEEE80211_VERIFY_LENGTH(efrm - frm,
747                             sizeof(struct ieee80211_action_ba_addbaresponse),
748                             return EINVAL);
749                         break;
750                 case IEEE80211_ACTION_BA_DELBA:
751                         IEEE80211_VERIFY_LENGTH(efrm - frm,
752                             sizeof(struct ieee80211_action_ba_delba),
753                             return EINVAL);
754                         break;
755                 }
756                 break;
757         case IEEE80211_ACTION_CAT_HT:
758                 switch (ia->ia_action) {
759                 case IEEE80211_ACTION_HT_TXCHWIDTH:
760                         IEEE80211_VERIFY_LENGTH(efrm - frm,
761                             sizeof(struct ieee80211_action_ht_txchwidth),
762                             return EINVAL);
763                         break;
764                 case IEEE80211_ACTION_HT_MIMOPWRSAVE:
765                         IEEE80211_VERIFY_LENGTH(efrm - frm,
766                             sizeof(struct ieee80211_action_ht_mimopowersave),
767                             return EINVAL);
768                         break;
769                 }
770                 break;
771 #ifdef IEEE80211_SUPPORT_MESH
772         case IEEE80211_ACTION_CAT_MESH:
773                 switch (ia->ia_action) {
774                 case IEEE80211_ACTION_MESH_LMETRIC:
775                         /*
776                          * XXX: verification is true only if we are using
777                          * Airtime link metric (default)
778                          */
779                         IEEE80211_VERIFY_LENGTH(efrm - frm,
780                             sizeof(struct ieee80211_meshlmetric_ie),
781                             return EINVAL);
782                         break;
783                 case IEEE80211_ACTION_MESH_HWMP:
784                         /* verify something */
785                         break;
786                 case IEEE80211_ACTION_MESH_GANN:
787                         IEEE80211_VERIFY_LENGTH(efrm - frm,
788                             sizeof(struct ieee80211_meshgann_ie),
789                             return EINVAL);
790                         break;
791                 case IEEE80211_ACTION_MESH_CC:
792                 case IEEE80211_ACTION_MESH_MCCA_SREQ:
793                 case IEEE80211_ACTION_MESH_MCCA_SREP:
794                 case IEEE80211_ACTION_MESH_MCCA_AREQ:
795                 case IEEE80211_ACTION_MESH_MCCA_ADVER:
796                 case IEEE80211_ACTION_MESH_MCCA_TRDOWN:
797                 case IEEE80211_ACTION_MESH_TBTT_REQ:
798                 case IEEE80211_ACTION_MESH_TBTT_RES:
799                         /* reject these early on, not implemented */
800                         IEEE80211_DISCARD(vap,
801                             IEEE80211_MSG_ELEMID | IEEE80211_MSG_INPUT,
802                             wh, NULL, "not implemented yet, act=0x%02X",
803                             ia->ia_action);
804                         return EINVAL;
805                 }
806                 break;
807         case IEEE80211_ACTION_CAT_SELF_PROT:
808                 /* If TA or RA group address discard silently */
809                 if (IEEE80211_IS_MULTICAST(wh->i_addr1) ||
810                         IEEE80211_IS_MULTICAST(wh->i_addr2))
811                         return EINVAL;
812                 /*
813                  * XXX: Should we verify complete length now or it is
814                  * to varying in sizes?
815                  */
816                 switch (ia->ia_action) {
817                 case IEEE80211_ACTION_MESHPEERING_CONFIRM:
818                 case IEEE80211_ACTION_MESHPEERING_CLOSE:
819                         /* is not a peering candidate (yet) */
820                         if (ni == vap->iv_bss)
821                                 return EINVAL;
822                         break;
823                 }
824                 break;
825 #endif
826         }
827         return 0;
828 }
829
830 #ifdef IEEE80211_DEBUG
831 /*
832  * Debugging support.
833  */
834 void
835 ieee80211_ssid_mismatch(struct ieee80211vap *vap, const char *tag,
836         uint8_t mac[IEEE80211_ADDR_LEN], uint8_t *ssid)
837 {
838         kprintf("[%s] discard %s frame, ssid mismatch: ",
839                 ether_sprintf(mac), tag);
840         ieee80211_print_essid(ssid + 2, ssid[1]);
841         kprintf("\n");
842 }
843
844 /*
845  * Return the bssid of a frame.
846  */
847 static const uint8_t *
848 ieee80211_getbssid(const struct ieee80211vap *vap,
849         const struct ieee80211_frame *wh)
850 {
851         if (vap->iv_opmode == IEEE80211_M_STA)
852                 return wh->i_addr2;
853         if ((wh->i_fc[1] & IEEE80211_FC1_DIR_MASK) != IEEE80211_FC1_DIR_NODS)
854                 return wh->i_addr1;
855         if ((wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) == IEEE80211_FC0_SUBTYPE_PS_POLL)
856                 return wh->i_addr1;
857         return wh->i_addr3;
858 }
859
860 #include <machine/stdarg.h>
861
862 void
863 ieee80211_note(const struct ieee80211vap *vap, const char *fmt, ...)
864 {
865         char buf[128];          /* XXX */
866         osdep_va_list ap;
867
868         osdep_va_start(ap, fmt);
869         kvsnprintf(buf, sizeof(buf), fmt, ap);
870         osdep_va_end(ap);
871
872         if_printf(vap->iv_ifp, "%s", buf);      /* NB: no \n */
873 }
874
875 void
876 ieee80211_note_frame(const struct ieee80211vap *vap,
877         const struct ieee80211_frame *wh,
878         const char *fmt, ...)
879 {
880         char buf[128];          /* XXX */
881         osdep_va_list ap;
882
883         osdep_va_start(ap, fmt);
884         kvsnprintf(buf, sizeof(buf), fmt, ap);
885         osdep_va_end(ap);
886         if_printf(vap->iv_ifp, "[%s] %s\n",
887                 ether_sprintf(ieee80211_getbssid(vap, wh)), buf);
888 }
889
890 void
891 ieee80211_note_mac(const struct ieee80211vap *vap,
892         const uint8_t mac[IEEE80211_ADDR_LEN],
893         const char *fmt, ...)
894 {
895         char buf[128];          /* XXX */
896         osdep_va_list ap;
897
898         osdep_va_start(ap, fmt);
899         kvsnprintf(buf, sizeof(buf), fmt, ap);
900         osdep_va_end(ap);
901         if_printf(vap->iv_ifp, "[%s] %s\n", ether_sprintf(mac), buf);
902 }
903
904 void
905 ieee80211_discard_frame(const struct ieee80211vap *vap,
906         const struct ieee80211_frame *wh,
907         const char *type, const char *fmt, ...)
908 {
909         osdep_va_list ap;
910
911         if_printf(vap->iv_ifp, "[%s] discard ",
912                 ether_sprintf(ieee80211_getbssid(vap, wh)));
913         if (type == NULL) {
914                 kprintf("%s frame, ", ieee80211_mgt_subtype_name[
915                         (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK) >>
916                         IEEE80211_FC0_SUBTYPE_SHIFT]);
917         } else
918                 kprintf("%s frame, ", type);
919         osdep_va_start(ap, fmt);
920         kvprintf(fmt, ap);
921         osdep_va_end(ap);
922         kprintf("\n");
923 }
924
925 void
926 ieee80211_discard_ie(const struct ieee80211vap *vap,
927         const struct ieee80211_frame *wh,
928         const char *type, const char *fmt, ...)
929 {
930         osdep_va_list ap;
931
932         if_printf(vap->iv_ifp, "[%s] discard ",
933                 ether_sprintf(ieee80211_getbssid(vap, wh)));
934         if (type != NULL)
935                 kprintf("%s information element, ", type);
936         else
937                 kprintf("information element, ");
938         osdep_va_start(ap, fmt);
939         kvprintf(fmt, ap);
940         osdep_va_end(ap);
941         kprintf("\n");
942 }
943
944 void
945 ieee80211_discard_mac(const struct ieee80211vap *vap,
946         const uint8_t mac[IEEE80211_ADDR_LEN],
947         const char *type, const char *fmt, ...)
948 {
949         osdep_va_list ap;
950
951         if_printf(vap->iv_ifp, "[%s] discard ", ether_sprintf(mac));
952         if (type != NULL)
953                 kprintf("%s frame, ", type);
954         else
955                 kprintf("frame, ");
956         osdep_va_start(ap, fmt);
957         kvprintf(fmt, ap);
958         osdep_va_end(ap);
959         kprintf("\n");
960 }
961 #endif /* IEEE80211_DEBUG */