mac80211: fix radiotap header generation
[safe/jmp/linux-2.6] / net / mac80211 / rx.c
1 /*
2  * Copyright 2002-2005, Instant802 Networks, Inc.
3  * Copyright 2005-2006, Devicescape Software, Inc.
4  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
5  * Copyright 2007       Johannes Berg <johannes@sipsolutions.net>
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  */
11
12 #include <linux/jiffies.h>
13 #include <linux/kernel.h>
14 #include <linux/skbuff.h>
15 #include <linux/netdevice.h>
16 #include <linux/etherdevice.h>
17 #include <linux/rcupdate.h>
18 #include <net/mac80211.h>
19 #include <net/ieee80211_radiotap.h>
20
21 #include "ieee80211_i.h"
22 #include "driver-ops.h"
23 #include "led.h"
24 #include "mesh.h"
25 #include "wep.h"
26 #include "wpa.h"
27 #include "tkip.h"
28 #include "wme.h"
29
30 static u8 ieee80211_sta_manage_reorder_buf(struct ieee80211_hw *hw,
31                                            struct tid_ampdu_rx *tid_agg_rx,
32                                            struct sk_buff *skb,
33                                            u16 mpdu_seq_num,
34                                            int bar_req);
35 /*
36  * monitor mode reception
37  *
38  * This function cleans up the SKB, i.e. it removes all the stuff
39  * only useful for monitoring.
40  */
41 static struct sk_buff *remove_monitor_info(struct ieee80211_local *local,
42                                            struct sk_buff *skb,
43                                            int rtap_len)
44 {
45         skb_pull(skb, rtap_len);
46
47         if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS) {
48                 if (likely(skb->len > FCS_LEN))
49                         skb_trim(skb, skb->len - FCS_LEN);
50                 else {
51                         /* driver bug */
52                         WARN_ON(1);
53                         dev_kfree_skb(skb);
54                         skb = NULL;
55                 }
56         }
57
58         return skb;
59 }
60
61 static inline int should_drop_frame(struct sk_buff *skb,
62                                     int present_fcs_len,
63                                     int radiotap_len)
64 {
65         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
66         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
67
68         if (status->flag & (RX_FLAG_FAILED_FCS_CRC | RX_FLAG_FAILED_PLCP_CRC))
69                 return 1;
70         if (unlikely(skb->len < 16 + present_fcs_len + radiotap_len))
71                 return 1;
72         if (ieee80211_is_ctl(hdr->frame_control) &&
73             !ieee80211_is_pspoll(hdr->frame_control) &&
74             !ieee80211_is_back_req(hdr->frame_control))
75                 return 1;
76         return 0;
77 }
78
79 static int
80 ieee80211_rx_radiotap_len(struct ieee80211_local *local,
81                           struct ieee80211_rx_status *status)
82 {
83         int len;
84
85         /* always present fields */
86         len = sizeof(struct ieee80211_radiotap_header) + 9;
87
88         if (status->flag & RX_FLAG_TSFT)
89                 len += 8;
90         if (local->hw.flags & IEEE80211_HW_SIGNAL_DBM)
91                 len += 1;
92         if (local->hw.flags & IEEE80211_HW_NOISE_DBM)
93                 len += 1;
94
95         if (len & 1) /* padding for RX_FLAGS if necessary */
96                 len++;
97
98         return len;
99 }
100
101 /*
102  * ieee80211_add_rx_radiotap_header - add radiotap header
103  *
104  * add a radiotap header containing all the fields which the hardware provided.
105  */
106 static void
107 ieee80211_add_rx_radiotap_header(struct ieee80211_local *local,
108                                  struct sk_buff *skb,
109                                  struct ieee80211_rate *rate,
110                                  int rtap_len)
111 {
112         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
113         struct ieee80211_radiotap_header *rthdr;
114         unsigned char *pos;
115         u16 rx_flags = 0;
116
117         rthdr = (struct ieee80211_radiotap_header *)skb_push(skb, rtap_len);
118         memset(rthdr, 0, rtap_len);
119
120         /* radiotap header, set always present flags */
121         rthdr->it_present =
122                 cpu_to_le32((1 << IEEE80211_RADIOTAP_FLAGS) |
123                             (1 << IEEE80211_RADIOTAP_CHANNEL) |
124                             (1 << IEEE80211_RADIOTAP_ANTENNA) |
125                             (1 << IEEE80211_RADIOTAP_RX_FLAGS));
126         rthdr->it_len = cpu_to_le16(rtap_len);
127
128         pos = (unsigned char *)(rthdr+1);
129
130         /* the order of the following fields is important */
131
132         /* IEEE80211_RADIOTAP_TSFT */
133         if (status->flag & RX_FLAG_TSFT) {
134                 put_unaligned_le64(status->mactime, pos);
135                 rthdr->it_present |=
136                         cpu_to_le32(1 << IEEE80211_RADIOTAP_TSFT);
137                 pos += 8;
138         }
139
140         /* IEEE80211_RADIOTAP_FLAGS */
141         if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS)
142                 *pos |= IEEE80211_RADIOTAP_F_FCS;
143         if (status->flag & (RX_FLAG_FAILED_FCS_CRC | RX_FLAG_FAILED_PLCP_CRC))
144                 *pos |= IEEE80211_RADIOTAP_F_BADFCS;
145         if (status->flag & RX_FLAG_SHORTPRE)
146                 *pos |= IEEE80211_RADIOTAP_F_SHORTPRE;
147         pos++;
148
149         /* IEEE80211_RADIOTAP_RATE */
150         if (status->flag & RX_FLAG_HT) {
151                 /*
152                  * TODO: add following information into radiotap header once
153                  * suitable fields are defined for it:
154                  * - MCS index (status->rate_idx)
155                  * - HT40 (status->flag & RX_FLAG_40MHZ)
156                  * - short-GI (status->flag & RX_FLAG_SHORT_GI)
157                  */
158                 *pos = 0;
159         } else {
160                 rthdr->it_present |= cpu_to_le32(1 << IEEE80211_RADIOTAP_RATE);
161                 *pos = rate->bitrate / 5;
162         }
163         pos++;
164
165         /* IEEE80211_RADIOTAP_CHANNEL */
166         put_unaligned_le16(status->freq, pos);
167         pos += 2;
168         if (status->band == IEEE80211_BAND_5GHZ)
169                 put_unaligned_le16(IEEE80211_CHAN_OFDM | IEEE80211_CHAN_5GHZ,
170                                    pos);
171         else if (rate->flags & IEEE80211_RATE_ERP_G)
172                 put_unaligned_le16(IEEE80211_CHAN_OFDM | IEEE80211_CHAN_2GHZ,
173                                    pos);
174         else
175                 put_unaligned_le16(IEEE80211_CHAN_CCK | IEEE80211_CHAN_2GHZ,
176                                    pos);
177         pos += 2;
178
179         /* IEEE80211_RADIOTAP_DBM_ANTSIGNAL */
180         if (local->hw.flags & IEEE80211_HW_SIGNAL_DBM) {
181                 *pos = status->signal;
182                 rthdr->it_present |=
183                         cpu_to_le32(1 << IEEE80211_RADIOTAP_DBM_ANTSIGNAL);
184                 pos++;
185         }
186
187         /* IEEE80211_RADIOTAP_DBM_ANTNOISE */
188         if (local->hw.flags & IEEE80211_HW_NOISE_DBM) {
189                 *pos = status->noise;
190                 rthdr->it_present |=
191                         cpu_to_le32(1 << IEEE80211_RADIOTAP_DBM_ANTNOISE);
192                 pos++;
193         }
194
195         /* IEEE80211_RADIOTAP_LOCK_QUALITY is missing */
196
197         /* IEEE80211_RADIOTAP_ANTENNA */
198         *pos = status->antenna;
199         pos++;
200
201         /* IEEE80211_RADIOTAP_DB_ANTNOISE is not used */
202
203         /* IEEE80211_RADIOTAP_RX_FLAGS */
204         /* ensure 2 byte alignment for the 2 byte field as required */
205         if ((pos - (u8 *)rthdr) & 1)
206                 pos++;
207         if (status->flag & RX_FLAG_FAILED_PLCP_CRC)
208                 rx_flags |= IEEE80211_RADIOTAP_F_RX_BADPLCP;
209         put_unaligned_le16(rx_flags, pos);
210         pos += 2;
211 }
212
213 /*
214  * This function copies a received frame to all monitor interfaces and
215  * returns a cleaned-up SKB that no longer includes the FCS nor the
216  * radiotap header the driver might have added.
217  */
218 static struct sk_buff *
219 ieee80211_rx_monitor(struct ieee80211_local *local, struct sk_buff *origskb,
220                      struct ieee80211_rate *rate)
221 {
222         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(origskb);
223         struct ieee80211_sub_if_data *sdata;
224         int needed_headroom = 0;
225         struct sk_buff *skb, *skb2;
226         struct net_device *prev_dev = NULL;
227         int present_fcs_len = 0;
228         int rtap_len = 0;
229
230         /*
231          * First, we may need to make a copy of the skb because
232          *  (1) we need to modify it for radiotap (if not present), and
233          *  (2) the other RX handlers will modify the skb we got.
234          *
235          * We don't need to, of course, if we aren't going to return
236          * the SKB because it has a bad FCS/PLCP checksum.
237          */
238         if (status->flag & RX_FLAG_RADIOTAP)
239                 rtap_len = ieee80211_get_radiotap_len(origskb->data);
240         else
241                 /* room for the radiotap header based on driver features */
242                 needed_headroom = ieee80211_rx_radiotap_len(local, status);
243
244         if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS)
245                 present_fcs_len = FCS_LEN;
246
247         if (!local->monitors) {
248                 if (should_drop_frame(origskb, present_fcs_len, rtap_len)) {
249                         dev_kfree_skb(origskb);
250                         return NULL;
251                 }
252
253                 return remove_monitor_info(local, origskb, rtap_len);
254         }
255
256         if (should_drop_frame(origskb, present_fcs_len, rtap_len)) {
257                 /* only need to expand headroom if necessary */
258                 skb = origskb;
259                 origskb = NULL;
260
261                 /*
262                  * This shouldn't trigger often because most devices have an
263                  * RX header they pull before we get here, and that should
264                  * be big enough for our radiotap information. We should
265                  * probably export the length to drivers so that we can have
266                  * them allocate enough headroom to start with.
267                  */
268                 if (skb_headroom(skb) < needed_headroom &&
269                     pskb_expand_head(skb, needed_headroom, 0, GFP_ATOMIC)) {
270                         dev_kfree_skb(skb);
271                         return NULL;
272                 }
273         } else {
274                 /*
275                  * Need to make a copy and possibly remove radiotap header
276                  * and FCS from the original.
277                  */
278                 skb = skb_copy_expand(origskb, needed_headroom, 0, GFP_ATOMIC);
279
280                 origskb = remove_monitor_info(local, origskb, rtap_len);
281
282                 if (!skb)
283                         return origskb;
284         }
285
286         /* if necessary, prepend radiotap information */
287         if (!(status->flag & RX_FLAG_RADIOTAP))
288                 ieee80211_add_rx_radiotap_header(local, skb, rate,
289                                                  needed_headroom);
290
291         skb_reset_mac_header(skb);
292         skb->ip_summed = CHECKSUM_UNNECESSARY;
293         skb->pkt_type = PACKET_OTHERHOST;
294         skb->protocol = htons(ETH_P_802_2);
295
296         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
297                 if (!netif_running(sdata->dev))
298                         continue;
299
300                 if (sdata->vif.type != NL80211_IFTYPE_MONITOR)
301                         continue;
302
303                 if (sdata->u.mntr_flags & MONITOR_FLAG_COOK_FRAMES)
304                         continue;
305
306                 if (prev_dev) {
307                         skb2 = skb_clone(skb, GFP_ATOMIC);
308                         if (skb2) {
309                                 skb2->dev = prev_dev;
310                                 netif_rx(skb2);
311                         }
312                 }
313
314                 prev_dev = sdata->dev;
315                 sdata->dev->stats.rx_packets++;
316                 sdata->dev->stats.rx_bytes += skb->len;
317         }
318
319         if (prev_dev) {
320                 skb->dev = prev_dev;
321                 netif_rx(skb);
322         } else
323                 dev_kfree_skb(skb);
324
325         return origskb;
326 }
327
328
329 static void ieee80211_parse_qos(struct ieee80211_rx_data *rx)
330 {
331         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
332         int tid;
333
334         /* does the frame have a qos control field? */
335         if (ieee80211_is_data_qos(hdr->frame_control)) {
336                 u8 *qc = ieee80211_get_qos_ctl(hdr);
337                 /* frame has qos control */
338                 tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
339                 if (*qc & IEEE80211_QOS_CONTROL_A_MSDU_PRESENT)
340                         rx->flags |= IEEE80211_RX_AMSDU;
341                 else
342                         rx->flags &= ~IEEE80211_RX_AMSDU;
343         } else {
344                 /*
345                  * IEEE 802.11-2007, 7.1.3.4.1 ("Sequence Number field"):
346                  *
347                  *      Sequence numbers for management frames, QoS data
348                  *      frames with a broadcast/multicast address in the
349                  *      Address 1 field, and all non-QoS data frames sent
350                  *      by QoS STAs are assigned using an additional single
351                  *      modulo-4096 counter, [...]
352                  *
353                  * We also use that counter for non-QoS STAs.
354                  */
355                 tid = NUM_RX_DATA_QUEUES - 1;
356         }
357
358         rx->queue = tid;
359         /* Set skb->priority to 1d tag if highest order bit of TID is not set.
360          * For now, set skb->priority to 0 for other cases. */
361         rx->skb->priority = (tid > 7) ? 0 : tid;
362 }
363
364 /**
365  * DOC: Packet alignment
366  *
367  * Drivers always need to pass packets that are aligned to two-byte boundaries
368  * to the stack.
369  *
370  * Additionally, should, if possible, align the payload data in a way that
371  * guarantees that the contained IP header is aligned to a four-byte
372  * boundary. In the case of regular frames, this simply means aligning the
373  * payload to a four-byte boundary (because either the IP header is directly
374  * contained, or IV/RFC1042 headers that have a length divisible by four are
375  * in front of it).
376  *
377  * With A-MSDU frames, however, the payload data address must yield two modulo
378  * four because there are 14-byte 802.3 headers within the A-MSDU frames that
379  * push the IP header further back to a multiple of four again. Thankfully, the
380  * specs were sane enough this time around to require padding each A-MSDU
381  * subframe to a length that is a multiple of four.
382  *
383  * Padding like Atheros hardware adds which is inbetween the 802.11 header and
384  * the payload is not supported, the driver is required to move the 802.11
385  * header to be directly in front of the payload in that case.
386  */
387 static void ieee80211_verify_alignment(struct ieee80211_rx_data *rx)
388 {
389         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
390         int hdrlen;
391
392 #ifndef CONFIG_MAC80211_DEBUG_PACKET_ALIGNMENT
393         return;
394 #endif
395
396         if (WARN_ONCE((unsigned long)rx->skb->data & 1,
397                       "unaligned packet at 0x%p\n", rx->skb->data))
398                 return;
399
400         if (!ieee80211_is_data_present(hdr->frame_control))
401                 return;
402
403         hdrlen = ieee80211_hdrlen(hdr->frame_control);
404         if (rx->flags & IEEE80211_RX_AMSDU)
405                 hdrlen += ETH_HLEN;
406         WARN_ONCE(((unsigned long)(rx->skb->data + hdrlen)) & 3,
407                   "unaligned IP payload at 0x%p\n", rx->skb->data + hdrlen);
408 }
409
410
411 /* rx handlers */
412
413 static ieee80211_rx_result debug_noinline
414 ieee80211_rx_h_passive_scan(struct ieee80211_rx_data *rx)
415 {
416         struct ieee80211_local *local = rx->local;
417         struct sk_buff *skb = rx->skb;
418
419         if (unlikely(test_bit(SCAN_HW_SCANNING, &local->scanning)))
420                 return ieee80211_scan_rx(rx->sdata, skb);
421
422         if (unlikely(test_bit(SCAN_SW_SCANNING, &local->scanning) &&
423                      (rx->flags & IEEE80211_RX_IN_SCAN))) {
424                 /* drop all the other packets during a software scan anyway */
425                 if (ieee80211_scan_rx(rx->sdata, skb) != RX_QUEUED)
426                         dev_kfree_skb(skb);
427                 return RX_QUEUED;
428         }
429
430         if (unlikely(rx->flags & IEEE80211_RX_IN_SCAN)) {
431                 /* scanning finished during invoking of handlers */
432                 I802_DEBUG_INC(local->rx_handlers_drop_passive_scan);
433                 return RX_DROP_UNUSABLE;
434         }
435
436         return RX_CONTINUE;
437 }
438
439
440 static int ieee80211_is_unicast_robust_mgmt_frame(struct sk_buff *skb)
441 {
442         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
443
444         if (skb->len < 24 || is_multicast_ether_addr(hdr->addr1))
445                 return 0;
446
447         return ieee80211_is_robust_mgmt_frame(hdr);
448 }
449
450
451 static int ieee80211_is_multicast_robust_mgmt_frame(struct sk_buff *skb)
452 {
453         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
454
455         if (skb->len < 24 || !is_multicast_ether_addr(hdr->addr1))
456                 return 0;
457
458         return ieee80211_is_robust_mgmt_frame(hdr);
459 }
460
461
462 /* Get the BIP key index from MMIE; return -1 if this is not a BIP frame */
463 static int ieee80211_get_mmie_keyidx(struct sk_buff *skb)
464 {
465         struct ieee80211_mgmt *hdr = (struct ieee80211_mgmt *) skb->data;
466         struct ieee80211_mmie *mmie;
467
468         if (skb->len < 24 + sizeof(*mmie) ||
469             !is_multicast_ether_addr(hdr->da))
470                 return -1;
471
472         if (!ieee80211_is_robust_mgmt_frame((struct ieee80211_hdr *) hdr))
473                 return -1; /* not a robust management frame */
474
475         mmie = (struct ieee80211_mmie *)
476                 (skb->data + skb->len - sizeof(*mmie));
477         if (mmie->element_id != WLAN_EID_MMIE ||
478             mmie->length != sizeof(*mmie) - 2)
479                 return -1;
480
481         return le16_to_cpu(mmie->key_id);
482 }
483
484
485 static ieee80211_rx_result
486 ieee80211_rx_mesh_check(struct ieee80211_rx_data *rx)
487 {
488         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
489         unsigned int hdrlen = ieee80211_hdrlen(hdr->frame_control);
490         char *dev_addr = rx->dev->dev_addr;
491
492         if (ieee80211_is_data(hdr->frame_control)) {
493                 if (is_multicast_ether_addr(hdr->addr1)) {
494                         if (ieee80211_has_tods(hdr->frame_control) ||
495                                 !ieee80211_has_fromds(hdr->frame_control))
496                                 return RX_DROP_MONITOR;
497                         if (memcmp(hdr->addr3, dev_addr, ETH_ALEN) == 0)
498                                 return RX_DROP_MONITOR;
499                 } else {
500                         if (!ieee80211_has_a4(hdr->frame_control))
501                                 return RX_DROP_MONITOR;
502                         if (memcmp(hdr->addr4, dev_addr, ETH_ALEN) == 0)
503                                 return RX_DROP_MONITOR;
504                 }
505         }
506
507         /* If there is not an established peer link and this is not a peer link
508          * establisment frame, beacon or probe, drop the frame.
509          */
510
511         if (!rx->sta || sta_plink_state(rx->sta) != PLINK_ESTAB) {
512                 struct ieee80211_mgmt *mgmt;
513
514                 if (!ieee80211_is_mgmt(hdr->frame_control))
515                         return RX_DROP_MONITOR;
516
517                 if (ieee80211_is_action(hdr->frame_control)) {
518                         mgmt = (struct ieee80211_mgmt *)hdr;
519                         if (mgmt->u.action.category != PLINK_CATEGORY)
520                                 return RX_DROP_MONITOR;
521                         return RX_CONTINUE;
522                 }
523
524                 if (ieee80211_is_probe_req(hdr->frame_control) ||
525                     ieee80211_is_probe_resp(hdr->frame_control) ||
526                     ieee80211_is_beacon(hdr->frame_control))
527                         return RX_CONTINUE;
528
529                 return RX_DROP_MONITOR;
530
531         }
532
533 #define msh_h_get(h, l) ((struct ieee80211s_hdr *) ((u8 *)h + l))
534
535         if (ieee80211_is_data(hdr->frame_control) &&
536             is_multicast_ether_addr(hdr->addr1) &&
537             mesh_rmc_check(hdr->addr3, msh_h_get(hdr, hdrlen), rx->sdata))
538                 return RX_DROP_MONITOR;
539 #undef msh_h_get
540
541         return RX_CONTINUE;
542 }
543
544
545 static ieee80211_rx_result debug_noinline
546 ieee80211_rx_h_check(struct ieee80211_rx_data *rx)
547 {
548         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
549
550         /* Drop duplicate 802.11 retransmissions (IEEE 802.11 Chap. 9.2.9) */
551         if (rx->sta && !is_multicast_ether_addr(hdr->addr1)) {
552                 if (unlikely(ieee80211_has_retry(hdr->frame_control) &&
553                              rx->sta->last_seq_ctrl[rx->queue] ==
554                              hdr->seq_ctrl)) {
555                         if (rx->flags & IEEE80211_RX_RA_MATCH) {
556                                 rx->local->dot11FrameDuplicateCount++;
557                                 rx->sta->num_duplicates++;
558                         }
559                         return RX_DROP_MONITOR;
560                 } else
561                         rx->sta->last_seq_ctrl[rx->queue] = hdr->seq_ctrl;
562         }
563
564         if (unlikely(rx->skb->len < 16)) {
565                 I802_DEBUG_INC(rx->local->rx_handlers_drop_short);
566                 return RX_DROP_MONITOR;
567         }
568
569         /* Drop disallowed frame classes based on STA auth/assoc state;
570          * IEEE 802.11, Chap 5.5.
571          *
572          * mac80211 filters only based on association state, i.e. it drops
573          * Class 3 frames from not associated stations. hostapd sends
574          * deauth/disassoc frames when needed. In addition, hostapd is
575          * responsible for filtering on both auth and assoc states.
576          */
577
578         if (ieee80211_vif_is_mesh(&rx->sdata->vif))
579                 return ieee80211_rx_mesh_check(rx);
580
581         if (unlikely((ieee80211_is_data(hdr->frame_control) ||
582                       ieee80211_is_pspoll(hdr->frame_control)) &&
583                      rx->sdata->vif.type != NL80211_IFTYPE_ADHOC &&
584                      (!rx->sta || !test_sta_flags(rx->sta, WLAN_STA_ASSOC)))) {
585                 if ((!ieee80211_has_fromds(hdr->frame_control) &&
586                      !ieee80211_has_tods(hdr->frame_control) &&
587                      ieee80211_is_data(hdr->frame_control)) ||
588                     !(rx->flags & IEEE80211_RX_RA_MATCH)) {
589                         /* Drop IBSS frames and frames for other hosts
590                          * silently. */
591                         return RX_DROP_MONITOR;
592                 }
593
594                 return RX_DROP_MONITOR;
595         }
596
597         return RX_CONTINUE;
598 }
599
600
601 static ieee80211_rx_result debug_noinline
602 ieee80211_rx_h_decrypt(struct ieee80211_rx_data *rx)
603 {
604         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
605         int keyidx;
606         int hdrlen;
607         ieee80211_rx_result result = RX_DROP_UNUSABLE;
608         struct ieee80211_key *stakey = NULL;
609         int mmie_keyidx = -1;
610
611         /*
612          * Key selection 101
613          *
614          * There are four types of keys:
615          *  - GTK (group keys)
616          *  - IGTK (group keys for management frames)
617          *  - PTK (pairwise keys)
618          *  - STK (station-to-station pairwise keys)
619          *
620          * When selecting a key, we have to distinguish between multicast
621          * (including broadcast) and unicast frames, the latter can only
622          * use PTKs and STKs while the former always use GTKs and IGTKs.
623          * Unless, of course, actual WEP keys ("pre-RSNA") are used, then
624          * unicast frames can also use key indices like GTKs. Hence, if we
625          * don't have a PTK/STK we check the key index for a WEP key.
626          *
627          * Note that in a regular BSS, multicast frames are sent by the
628          * AP only, associated stations unicast the frame to the AP first
629          * which then multicasts it on their behalf.
630          *
631          * There is also a slight problem in IBSS mode: GTKs are negotiated
632          * with each station, that is something we don't currently handle.
633          * The spec seems to expect that one negotiates the same key with
634          * every station but there's no such requirement; VLANs could be
635          * possible.
636          */
637
638         /*
639          * No point in finding a key and decrypting if the frame is neither
640          * addressed to us nor a multicast frame.
641          */
642         if (!(rx->flags & IEEE80211_RX_RA_MATCH))
643                 return RX_CONTINUE;
644
645         if (rx->sta)
646                 stakey = rcu_dereference(rx->sta->key);
647
648         if (!ieee80211_has_protected(hdr->frame_control))
649                 mmie_keyidx = ieee80211_get_mmie_keyidx(rx->skb);
650
651         if (!is_multicast_ether_addr(hdr->addr1) && stakey) {
652                 rx->key = stakey;
653                 /* Skip decryption if the frame is not protected. */
654                 if (!ieee80211_has_protected(hdr->frame_control))
655                         return RX_CONTINUE;
656         } else if (mmie_keyidx >= 0) {
657                 /* Broadcast/multicast robust management frame / BIP */
658                 if ((rx->status->flag & RX_FLAG_DECRYPTED) &&
659                     (rx->status->flag & RX_FLAG_IV_STRIPPED))
660                         return RX_CONTINUE;
661
662                 if (mmie_keyidx < NUM_DEFAULT_KEYS ||
663                     mmie_keyidx >= NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS)
664                         return RX_DROP_MONITOR; /* unexpected BIP keyidx */
665                 rx->key = rcu_dereference(rx->sdata->keys[mmie_keyidx]);
666         } else if (!ieee80211_has_protected(hdr->frame_control)) {
667                 /*
668                  * The frame was not protected, so skip decryption. However, we
669                  * need to set rx->key if there is a key that could have been
670                  * used so that the frame may be dropped if encryption would
671                  * have been expected.
672                  */
673                 struct ieee80211_key *key = NULL;
674                 if (ieee80211_is_mgmt(hdr->frame_control) &&
675                     is_multicast_ether_addr(hdr->addr1) &&
676                     (key = rcu_dereference(rx->sdata->default_mgmt_key)))
677                         rx->key = key;
678                 else if ((key = rcu_dereference(rx->sdata->default_key)))
679                         rx->key = key;
680                 return RX_CONTINUE;
681         } else {
682                 /*
683                  * The device doesn't give us the IV so we won't be
684                  * able to look up the key. That's ok though, we
685                  * don't need to decrypt the frame, we just won't
686                  * be able to keep statistics accurate.
687                  * Except for key threshold notifications, should
688                  * we somehow allow the driver to tell us which key
689                  * the hardware used if this flag is set?
690                  */
691                 if ((rx->status->flag & RX_FLAG_DECRYPTED) &&
692                     (rx->status->flag & RX_FLAG_IV_STRIPPED))
693                         return RX_CONTINUE;
694
695                 hdrlen = ieee80211_hdrlen(hdr->frame_control);
696
697                 if (rx->skb->len < 8 + hdrlen)
698                         return RX_DROP_UNUSABLE; /* TODO: count this? */
699
700                 /*
701                  * no need to call ieee80211_wep_get_keyidx,
702                  * it verifies a bunch of things we've done already
703                  */
704                 keyidx = rx->skb->data[hdrlen + 3] >> 6;
705
706                 rx->key = rcu_dereference(rx->sdata->keys[keyidx]);
707
708                 /*
709                  * RSNA-protected unicast frames should always be sent with
710                  * pairwise or station-to-station keys, but for WEP we allow
711                  * using a key index as well.
712                  */
713                 if (rx->key && rx->key->conf.alg != ALG_WEP &&
714                     !is_multicast_ether_addr(hdr->addr1))
715                         rx->key = NULL;
716         }
717
718         if (rx->key) {
719                 rx->key->tx_rx_count++;
720                 /* TODO: add threshold stuff again */
721         } else {
722                 return RX_DROP_MONITOR;
723         }
724
725         /* Check for weak IVs if possible */
726         if (rx->sta && rx->key->conf.alg == ALG_WEP &&
727             ieee80211_is_data(hdr->frame_control) &&
728             (!(rx->status->flag & RX_FLAG_IV_STRIPPED) ||
729              !(rx->status->flag & RX_FLAG_DECRYPTED)) &&
730             ieee80211_wep_is_weak_iv(rx->skb, rx->key))
731                 rx->sta->wep_weak_iv_count++;
732
733         switch (rx->key->conf.alg) {
734         case ALG_WEP:
735                 result = ieee80211_crypto_wep_decrypt(rx);
736                 break;
737         case ALG_TKIP:
738                 result = ieee80211_crypto_tkip_decrypt(rx);
739                 break;
740         case ALG_CCMP:
741                 result = ieee80211_crypto_ccmp_decrypt(rx);
742                 break;
743         case ALG_AES_CMAC:
744                 result = ieee80211_crypto_aes_cmac_decrypt(rx);
745                 break;
746         }
747
748         /* either the frame has been decrypted or will be dropped */
749         rx->status->flag |= RX_FLAG_DECRYPTED;
750
751         return result;
752 }
753
754 static ieee80211_rx_result debug_noinline
755 ieee80211_rx_h_check_more_data(struct ieee80211_rx_data *rx)
756 {
757         struct ieee80211_local *local;
758         struct ieee80211_hdr *hdr;
759         struct sk_buff *skb;
760
761         local = rx->local;
762         skb = rx->skb;
763         hdr = (struct ieee80211_hdr *) skb->data;
764
765         if (!local->pspolling)
766                 return RX_CONTINUE;
767
768         if (!ieee80211_has_fromds(hdr->frame_control))
769                 /* this is not from AP */
770                 return RX_CONTINUE;
771
772         if (!ieee80211_is_data(hdr->frame_control))
773                 return RX_CONTINUE;
774
775         if (!ieee80211_has_moredata(hdr->frame_control)) {
776                 /* AP has no more frames buffered for us */
777                 local->pspolling = false;
778                 return RX_CONTINUE;
779         }
780
781         /* more data bit is set, let's request a new frame from the AP */
782         ieee80211_send_pspoll(local, rx->sdata);
783
784         return RX_CONTINUE;
785 }
786
787 static void ap_sta_ps_start(struct sta_info *sta)
788 {
789         struct ieee80211_sub_if_data *sdata = sta->sdata;
790         struct ieee80211_local *local = sdata->local;
791
792         atomic_inc(&sdata->bss->num_sta_ps);
793         set_sta_flags(sta, WLAN_STA_PS);
794         drv_sta_notify(local, &sdata->vif, STA_NOTIFY_SLEEP, &sta->sta);
795 #ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
796         printk(KERN_DEBUG "%s: STA %pM aid %d enters power save mode\n",
797                sdata->dev->name, sta->sta.addr, sta->sta.aid);
798 #endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
799 }
800
801 static int ap_sta_ps_end(struct sta_info *sta)
802 {
803         struct ieee80211_sub_if_data *sdata = sta->sdata;
804         struct ieee80211_local *local = sdata->local;
805         int sent, buffered;
806
807         atomic_dec(&sdata->bss->num_sta_ps);
808
809         clear_sta_flags(sta, WLAN_STA_PS);
810         drv_sta_notify(local, &sdata->vif, STA_NOTIFY_AWAKE, &sta->sta);
811
812         if (!skb_queue_empty(&sta->ps_tx_buf))
813                 sta_info_clear_tim_bit(sta);
814
815 #ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
816         printk(KERN_DEBUG "%s: STA %pM aid %d exits power save mode\n",
817                sdata->dev->name, sta->sta.addr, sta->sta.aid);
818 #endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
819
820         /* Send all buffered frames to the station */
821         sent = ieee80211_add_pending_skbs(local, &sta->tx_filtered);
822         buffered = ieee80211_add_pending_skbs(local, &sta->ps_tx_buf);
823         sent += buffered;
824         local->total_ps_buffered -= buffered;
825
826 #ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
827         printk(KERN_DEBUG "%s: STA %pM aid %d sending %d filtered/%d PS frames "
828                "since STA not sleeping anymore\n", sdata->dev->name,
829                sta->sta.addr, sta->sta.aid, sent - buffered, buffered);
830 #endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
831
832         return sent;
833 }
834
835 static ieee80211_rx_result debug_noinline
836 ieee80211_rx_h_sta_process(struct ieee80211_rx_data *rx)
837 {
838         struct sta_info *sta = rx->sta;
839         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
840
841         if (!sta)
842                 return RX_CONTINUE;
843
844         /*
845          * Update last_rx only for IBSS packets which are for the current
846          * BSSID to avoid keeping the current IBSS network alive in cases
847          * where other STAs start using different BSSID.
848          */
849         if (rx->sdata->vif.type == NL80211_IFTYPE_ADHOC) {
850                 u8 *bssid = ieee80211_get_bssid(hdr, rx->skb->len,
851                                                 NL80211_IFTYPE_ADHOC);
852                 if (compare_ether_addr(bssid, rx->sdata->u.ibss.bssid) == 0)
853                         sta->last_rx = jiffies;
854         } else if (!is_multicast_ether_addr(hdr->addr1)) {
855                 /*
856                  * Mesh beacons will update last_rx when if they are found to
857                  * match the current local configuration when processed.
858                  */
859                 sta->last_rx = jiffies;
860         }
861
862         if (!(rx->flags & IEEE80211_RX_RA_MATCH))
863                 return RX_CONTINUE;
864
865         if (rx->sdata->vif.type == NL80211_IFTYPE_STATION)
866                 ieee80211_sta_rx_notify(rx->sdata, hdr);
867
868         sta->rx_fragments++;
869         sta->rx_bytes += rx->skb->len;
870         sta->last_signal = rx->status->signal;
871         sta->last_qual = rx->status->qual;
872         sta->last_noise = rx->status->noise;
873
874         /*
875          * Change STA power saving mode only at the end of a frame
876          * exchange sequence.
877          */
878         if (!ieee80211_has_morefrags(hdr->frame_control) &&
879             (rx->sdata->vif.type == NL80211_IFTYPE_AP ||
880              rx->sdata->vif.type == NL80211_IFTYPE_AP_VLAN)) {
881                 if (test_sta_flags(sta, WLAN_STA_PS)) {
882                         /*
883                          * Ignore doze->wake transitions that are
884                          * indicated by non-data frames, the standard
885                          * is unclear here, but for example going to
886                          * PS mode and then scanning would cause a
887                          * doze->wake transition for the probe request,
888                          * and that is clearly undesirable.
889                          */
890                         if (ieee80211_is_data(hdr->frame_control) &&
891                             !ieee80211_has_pm(hdr->frame_control))
892                                 rx->sent_ps_buffered += ap_sta_ps_end(sta);
893                 } else {
894                         if (ieee80211_has_pm(hdr->frame_control))
895                                 ap_sta_ps_start(sta);
896                 }
897         }
898
899         /* Drop data::nullfunc frames silently, since they are used only to
900          * control station power saving mode. */
901         if (ieee80211_is_nullfunc(hdr->frame_control)) {
902                 I802_DEBUG_INC(rx->local->rx_handlers_drop_nullfunc);
903                 /* Update counter and free packet here to avoid counting this
904                  * as a dropped packed. */
905                 sta->rx_packets++;
906                 dev_kfree_skb(rx->skb);
907                 return RX_QUEUED;
908         }
909
910         return RX_CONTINUE;
911 } /* ieee80211_rx_h_sta_process */
912
913 static inline struct ieee80211_fragment_entry *
914 ieee80211_reassemble_add(struct ieee80211_sub_if_data *sdata,
915                          unsigned int frag, unsigned int seq, int rx_queue,
916                          struct sk_buff **skb)
917 {
918         struct ieee80211_fragment_entry *entry;
919         int idx;
920
921         idx = sdata->fragment_next;
922         entry = &sdata->fragments[sdata->fragment_next++];
923         if (sdata->fragment_next >= IEEE80211_FRAGMENT_MAX)
924                 sdata->fragment_next = 0;
925
926         if (!skb_queue_empty(&entry->skb_list)) {
927 #ifdef CONFIG_MAC80211_VERBOSE_DEBUG
928                 struct ieee80211_hdr *hdr =
929                         (struct ieee80211_hdr *) entry->skb_list.next->data;
930                 printk(KERN_DEBUG "%s: RX reassembly removed oldest "
931                        "fragment entry (idx=%d age=%lu seq=%d last_frag=%d "
932                        "addr1=%pM addr2=%pM\n",
933                        sdata->dev->name, idx,
934                        jiffies - entry->first_frag_time, entry->seq,
935                        entry->last_frag, hdr->addr1, hdr->addr2);
936 #endif
937                 __skb_queue_purge(&entry->skb_list);
938         }
939
940         __skb_queue_tail(&entry->skb_list, *skb); /* no need for locking */
941         *skb = NULL;
942         entry->first_frag_time = jiffies;
943         entry->seq = seq;
944         entry->rx_queue = rx_queue;
945         entry->last_frag = frag;
946         entry->ccmp = 0;
947         entry->extra_len = 0;
948
949         return entry;
950 }
951
952 static inline struct ieee80211_fragment_entry *
953 ieee80211_reassemble_find(struct ieee80211_sub_if_data *sdata,
954                           unsigned int frag, unsigned int seq,
955                           int rx_queue, struct ieee80211_hdr *hdr)
956 {
957         struct ieee80211_fragment_entry *entry;
958         int i, idx;
959
960         idx = sdata->fragment_next;
961         for (i = 0; i < IEEE80211_FRAGMENT_MAX; i++) {
962                 struct ieee80211_hdr *f_hdr;
963
964                 idx--;
965                 if (idx < 0)
966                         idx = IEEE80211_FRAGMENT_MAX - 1;
967
968                 entry = &sdata->fragments[idx];
969                 if (skb_queue_empty(&entry->skb_list) || entry->seq != seq ||
970                     entry->rx_queue != rx_queue ||
971                     entry->last_frag + 1 != frag)
972                         continue;
973
974                 f_hdr = (struct ieee80211_hdr *)entry->skb_list.next->data;
975
976                 /*
977                  * Check ftype and addresses are equal, else check next fragment
978                  */
979                 if (((hdr->frame_control ^ f_hdr->frame_control) &
980                      cpu_to_le16(IEEE80211_FCTL_FTYPE)) ||
981                     compare_ether_addr(hdr->addr1, f_hdr->addr1) != 0 ||
982                     compare_ether_addr(hdr->addr2, f_hdr->addr2) != 0)
983                         continue;
984
985                 if (time_after(jiffies, entry->first_frag_time + 2 * HZ)) {
986                         __skb_queue_purge(&entry->skb_list);
987                         continue;
988                 }
989                 return entry;
990         }
991
992         return NULL;
993 }
994
995 static ieee80211_rx_result debug_noinline
996 ieee80211_rx_h_defragment(struct ieee80211_rx_data *rx)
997 {
998         struct ieee80211_hdr *hdr;
999         u16 sc;
1000         __le16 fc;
1001         unsigned int frag, seq;
1002         struct ieee80211_fragment_entry *entry;
1003         struct sk_buff *skb;
1004
1005         hdr = (struct ieee80211_hdr *)rx->skb->data;
1006         fc = hdr->frame_control;
1007         sc = le16_to_cpu(hdr->seq_ctrl);
1008         frag = sc & IEEE80211_SCTL_FRAG;
1009
1010         if (likely((!ieee80211_has_morefrags(fc) && frag == 0) ||
1011                    (rx->skb)->len < 24 ||
1012                    is_multicast_ether_addr(hdr->addr1))) {
1013                 /* not fragmented */
1014                 goto out;
1015         }
1016         I802_DEBUG_INC(rx->local->rx_handlers_fragments);
1017
1018         seq = (sc & IEEE80211_SCTL_SEQ) >> 4;
1019
1020         if (frag == 0) {
1021                 /* This is the first fragment of a new frame. */
1022                 entry = ieee80211_reassemble_add(rx->sdata, frag, seq,
1023                                                  rx->queue, &(rx->skb));
1024                 if (rx->key && rx->key->conf.alg == ALG_CCMP &&
1025                     ieee80211_has_protected(fc)) {
1026                         /* Store CCMP PN so that we can verify that the next
1027                          * fragment has a sequential PN value. */
1028                         entry->ccmp = 1;
1029                         memcpy(entry->last_pn,
1030                                rx->key->u.ccmp.rx_pn[rx->queue],
1031                                CCMP_PN_LEN);
1032                 }
1033                 return RX_QUEUED;
1034         }
1035
1036         /* This is a fragment for a frame that should already be pending in
1037          * fragment cache. Add this fragment to the end of the pending entry.
1038          */
1039         entry = ieee80211_reassemble_find(rx->sdata, frag, seq, rx->queue, hdr);
1040         if (!entry) {
1041                 I802_DEBUG_INC(rx->local->rx_handlers_drop_defrag);
1042                 return RX_DROP_MONITOR;
1043         }
1044
1045         /* Verify that MPDUs within one MSDU have sequential PN values.
1046          * (IEEE 802.11i, 8.3.3.4.5) */
1047         if (entry->ccmp) {
1048                 int i;
1049                 u8 pn[CCMP_PN_LEN], *rpn;
1050                 if (!rx->key || rx->key->conf.alg != ALG_CCMP)
1051                         return RX_DROP_UNUSABLE;
1052                 memcpy(pn, entry->last_pn, CCMP_PN_LEN);
1053                 for (i = CCMP_PN_LEN - 1; i >= 0; i--) {
1054                         pn[i]++;
1055                         if (pn[i])
1056                                 break;
1057                 }
1058                 rpn = rx->key->u.ccmp.rx_pn[rx->queue];
1059                 if (memcmp(pn, rpn, CCMP_PN_LEN))
1060                         return RX_DROP_UNUSABLE;
1061                 memcpy(entry->last_pn, pn, CCMP_PN_LEN);
1062         }
1063
1064         skb_pull(rx->skb, ieee80211_hdrlen(fc));
1065         __skb_queue_tail(&entry->skb_list, rx->skb);
1066         entry->last_frag = frag;
1067         entry->extra_len += rx->skb->len;
1068         if (ieee80211_has_morefrags(fc)) {
1069                 rx->skb = NULL;
1070                 return RX_QUEUED;
1071         }
1072
1073         rx->skb = __skb_dequeue(&entry->skb_list);
1074         if (skb_tailroom(rx->skb) < entry->extra_len) {
1075                 I802_DEBUG_INC(rx->local->rx_expand_skb_head2);
1076                 if (unlikely(pskb_expand_head(rx->skb, 0, entry->extra_len,
1077                                               GFP_ATOMIC))) {
1078                         I802_DEBUG_INC(rx->local->rx_handlers_drop_defrag);
1079                         __skb_queue_purge(&entry->skb_list);
1080                         return RX_DROP_UNUSABLE;
1081                 }
1082         }
1083         while ((skb = __skb_dequeue(&entry->skb_list))) {
1084                 memcpy(skb_put(rx->skb, skb->len), skb->data, skb->len);
1085                 dev_kfree_skb(skb);
1086         }
1087
1088         /* Complete frame has been reassembled - process it now */
1089         rx->flags |= IEEE80211_RX_FRAGMENTED;
1090
1091  out:
1092         if (rx->sta)
1093                 rx->sta->rx_packets++;
1094         if (is_multicast_ether_addr(hdr->addr1))
1095                 rx->local->dot11MulticastReceivedFrameCount++;
1096         else
1097                 ieee80211_led_rx(rx->local);
1098         return RX_CONTINUE;
1099 }
1100
1101 static ieee80211_rx_result debug_noinline
1102 ieee80211_rx_h_ps_poll(struct ieee80211_rx_data *rx)
1103 {
1104         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(rx->dev);
1105         struct sk_buff *skb;
1106         int no_pending_pkts;
1107         __le16 fc = ((struct ieee80211_hdr *)rx->skb->data)->frame_control;
1108
1109         if (likely(!rx->sta || !ieee80211_is_pspoll(fc) ||
1110                    !(rx->flags & IEEE80211_RX_RA_MATCH)))
1111                 return RX_CONTINUE;
1112
1113         if ((sdata->vif.type != NL80211_IFTYPE_AP) &&
1114             (sdata->vif.type != NL80211_IFTYPE_AP_VLAN))
1115                 return RX_DROP_UNUSABLE;
1116
1117         skb = skb_dequeue(&rx->sta->tx_filtered);
1118         if (!skb) {
1119                 skb = skb_dequeue(&rx->sta->ps_tx_buf);
1120                 if (skb)
1121                         rx->local->total_ps_buffered--;
1122         }
1123         no_pending_pkts = skb_queue_empty(&rx->sta->tx_filtered) &&
1124                 skb_queue_empty(&rx->sta->ps_tx_buf);
1125
1126         if (skb) {
1127                 struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1128                 struct ieee80211_hdr *hdr =
1129                         (struct ieee80211_hdr *) skb->data;
1130
1131                 /*
1132                  * Tell TX path to send this frame even though the STA may
1133                  * still remain is PS mode after this frame exchange.
1134                  */
1135                 info->flags |= IEEE80211_TX_CTL_PSPOLL_RESPONSE;
1136
1137 #ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
1138                 printk(KERN_DEBUG "STA %pM aid %d: PS Poll (entries after %d)\n",
1139                        rx->sta->sta.addr, rx->sta->sta.aid,
1140                        skb_queue_len(&rx->sta->ps_tx_buf));
1141 #endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
1142
1143                 /* Use MoreData flag to indicate whether there are more
1144                  * buffered frames for this STA */
1145                 if (no_pending_pkts)
1146                         hdr->frame_control &= cpu_to_le16(~IEEE80211_FCTL_MOREDATA);
1147                 else
1148                         hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_MOREDATA);
1149
1150                 ieee80211_add_pending_skb(rx->local, skb);
1151
1152                 if (no_pending_pkts)
1153                         sta_info_clear_tim_bit(rx->sta);
1154 #ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
1155         } else if (!rx->sent_ps_buffered) {
1156                 /*
1157                  * FIXME: This can be the result of a race condition between
1158                  *        us expiring a frame and the station polling for it.
1159                  *        Should we send it a null-func frame indicating we
1160                  *        have nothing buffered for it?
1161                  */
1162                 printk(KERN_DEBUG "%s: STA %pM sent PS Poll even "
1163                        "though there are no buffered frames for it\n",
1164                        rx->dev->name, rx->sta->sta.addr);
1165 #endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
1166         }
1167
1168         /* Free PS Poll skb here instead of returning RX_DROP that would
1169          * count as an dropped frame. */
1170         dev_kfree_skb(rx->skb);
1171
1172         return RX_QUEUED;
1173 }
1174
1175 static ieee80211_rx_result debug_noinline
1176 ieee80211_rx_h_remove_qos_control(struct ieee80211_rx_data *rx)
1177 {
1178         u8 *data = rx->skb->data;
1179         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)data;
1180
1181         if (!ieee80211_is_data_qos(hdr->frame_control))
1182                 return RX_CONTINUE;
1183
1184         /* remove the qos control field, update frame type and meta-data */
1185         memmove(data + IEEE80211_QOS_CTL_LEN, data,
1186                 ieee80211_hdrlen(hdr->frame_control) - IEEE80211_QOS_CTL_LEN);
1187         hdr = (struct ieee80211_hdr *)skb_pull(rx->skb, IEEE80211_QOS_CTL_LEN);
1188         /* change frame type to non QOS */
1189         hdr->frame_control &= ~cpu_to_le16(IEEE80211_STYPE_QOS_DATA);
1190
1191         return RX_CONTINUE;
1192 }
1193
1194 static int
1195 ieee80211_802_1x_port_control(struct ieee80211_rx_data *rx)
1196 {
1197         if (unlikely(!rx->sta ||
1198             !test_sta_flags(rx->sta, WLAN_STA_AUTHORIZED)))
1199                 return -EACCES;
1200
1201         return 0;
1202 }
1203
1204 static int
1205 ieee80211_drop_unencrypted(struct ieee80211_rx_data *rx, __le16 fc)
1206 {
1207         /*
1208          * Pass through unencrypted frames if the hardware has
1209          * decrypted them already.
1210          */
1211         if (rx->status->flag & RX_FLAG_DECRYPTED)
1212                 return 0;
1213
1214         /* Drop unencrypted frames if key is set. */
1215         if (unlikely(!ieee80211_has_protected(fc) &&
1216                      !ieee80211_is_nullfunc(fc) &&
1217                      ieee80211_is_data(fc) &&
1218                      (rx->key || rx->sdata->drop_unencrypted)))
1219                 return -EACCES;
1220         if (rx->sta && test_sta_flags(rx->sta, WLAN_STA_MFP)) {
1221                 if (unlikely(ieee80211_is_unicast_robust_mgmt_frame(rx->skb) &&
1222                              rx->key))
1223                         return -EACCES;
1224                 /* BIP does not use Protected field, so need to check MMIE */
1225                 if (unlikely(ieee80211_is_multicast_robust_mgmt_frame(rx->skb)
1226                              && ieee80211_get_mmie_keyidx(rx->skb) < 0 &&
1227                              rx->key))
1228                         return -EACCES;
1229                 /*
1230                  * When using MFP, Action frames are not allowed prior to
1231                  * having configured keys.
1232                  */
1233                 if (unlikely(ieee80211_is_action(fc) && !rx->key &&
1234                              ieee80211_is_robust_mgmt_frame(
1235                                      (struct ieee80211_hdr *) rx->skb->data)))
1236                         return -EACCES;
1237         }
1238
1239         return 0;
1240 }
1241
1242 static int
1243 __ieee80211_data_to_8023(struct ieee80211_rx_data *rx)
1244 {
1245         struct net_device *dev = rx->dev;
1246         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1247
1248         return ieee80211_data_to_8023(rx->skb, dev->dev_addr, sdata->vif.type);
1249 }
1250
1251 /*
1252  * requires that rx->skb is a frame with ethernet header
1253  */
1254 static bool ieee80211_frame_allowed(struct ieee80211_rx_data *rx, __le16 fc)
1255 {
1256         static const u8 pae_group_addr[ETH_ALEN] __aligned(2)
1257                 = { 0x01, 0x80, 0xC2, 0x00, 0x00, 0x03 };
1258         struct ethhdr *ehdr = (struct ethhdr *) rx->skb->data;
1259
1260         /*
1261          * Allow EAPOL frames to us/the PAE group address regardless
1262          * of whether the frame was encrypted or not.
1263          */
1264         if (ehdr->h_proto == htons(ETH_P_PAE) &&
1265             (compare_ether_addr(ehdr->h_dest, rx->dev->dev_addr) == 0 ||
1266              compare_ether_addr(ehdr->h_dest, pae_group_addr) == 0))
1267                 return true;
1268
1269         if (ieee80211_802_1x_port_control(rx) ||
1270             ieee80211_drop_unencrypted(rx, fc))
1271                 return false;
1272
1273         return true;
1274 }
1275
1276 /*
1277  * requires that rx->skb is a frame with ethernet header
1278  */
1279 static void
1280 ieee80211_deliver_skb(struct ieee80211_rx_data *rx)
1281 {
1282         struct net_device *dev = rx->dev;
1283         struct ieee80211_local *local = rx->local;
1284         struct sk_buff *skb, *xmit_skb;
1285         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1286         struct ethhdr *ehdr = (struct ethhdr *) rx->skb->data;
1287         struct sta_info *dsta;
1288
1289         skb = rx->skb;
1290         xmit_skb = NULL;
1291
1292         if ((sdata->vif.type == NL80211_IFTYPE_AP ||
1293              sdata->vif.type == NL80211_IFTYPE_AP_VLAN) &&
1294             !(sdata->flags & IEEE80211_SDATA_DONT_BRIDGE_PACKETS) &&
1295             (rx->flags & IEEE80211_RX_RA_MATCH)) {
1296                 if (is_multicast_ether_addr(ehdr->h_dest)) {
1297                         /*
1298                          * send multicast frames both to higher layers in
1299                          * local net stack and back to the wireless medium
1300                          */
1301                         xmit_skb = skb_copy(skb, GFP_ATOMIC);
1302                         if (!xmit_skb && net_ratelimit())
1303                                 printk(KERN_DEBUG "%s: failed to clone "
1304                                        "multicast frame\n", dev->name);
1305                 } else {
1306                         dsta = sta_info_get(local, skb->data);
1307                         if (dsta && dsta->sdata->dev == dev) {
1308                                 /*
1309                                  * The destination station is associated to
1310                                  * this AP (in this VLAN), so send the frame
1311                                  * directly to it and do not pass it to local
1312                                  * net stack.
1313                                  */
1314                                 xmit_skb = skb;
1315                                 skb = NULL;
1316                         }
1317                 }
1318         }
1319
1320         if (skb) {
1321                 int align __maybe_unused;
1322
1323 #if defined(CONFIG_MAC80211_DEBUG_PACKET_ALIGNMENT) || !defined(CONFIG_HAVE_EFFICIENT_UNALIGNED_ACCESS)
1324                 /*
1325                  * 'align' will only take the values 0 or 2 here
1326                  * since all frames are required to be aligned
1327                  * to 2-byte boundaries when being passed to
1328                  * mac80211. That also explains the __skb_push()
1329                  * below.
1330                  */
1331                 align = ((unsigned long)(skb->data + sizeof(struct ethhdr))) & 3;
1332                 if (align) {
1333                         if (WARN_ON(skb_headroom(skb) < 3)) {
1334                                 dev_kfree_skb(skb);
1335                                 skb = NULL;
1336                         } else {
1337                                 u8 *data = skb->data;
1338                                 size_t len = skb->len;
1339                                 u8 *new = __skb_push(skb, align);
1340                                 memmove(new, data, len);
1341                                 __skb_trim(skb, len);
1342                         }
1343                 }
1344 #endif
1345
1346                 if (skb) {
1347                         /* deliver to local stack */
1348                         skb->protocol = eth_type_trans(skb, dev);
1349                         memset(skb->cb, 0, sizeof(skb->cb));
1350                         netif_rx(skb);
1351                 }
1352         }
1353
1354         if (xmit_skb) {
1355                 /* send to wireless media */
1356                 xmit_skb->protocol = htons(ETH_P_802_3);
1357                 skb_reset_network_header(xmit_skb);
1358                 skb_reset_mac_header(xmit_skb);
1359                 dev_queue_xmit(xmit_skb);
1360         }
1361 }
1362
1363 static ieee80211_rx_result debug_noinline
1364 ieee80211_rx_h_amsdu(struct ieee80211_rx_data *rx)
1365 {
1366         struct net_device *dev = rx->dev;
1367         struct ieee80211_local *local = rx->local;
1368         u16 ethertype;
1369         u8 *payload;
1370         struct sk_buff *skb = rx->skb, *frame = NULL;
1371         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1372         __le16 fc = hdr->frame_control;
1373         const struct ethhdr *eth;
1374         int remaining, err;
1375         u8 dst[ETH_ALEN];
1376         u8 src[ETH_ALEN];
1377
1378         if (unlikely(!ieee80211_is_data(fc)))
1379                 return RX_CONTINUE;
1380
1381         if (unlikely(!ieee80211_is_data_present(fc)))
1382                 return RX_DROP_MONITOR;
1383
1384         if (!(rx->flags & IEEE80211_RX_AMSDU))
1385                 return RX_CONTINUE;
1386
1387         err = __ieee80211_data_to_8023(rx);
1388         if (unlikely(err))
1389                 return RX_DROP_UNUSABLE;
1390
1391         skb->dev = dev;
1392
1393         dev->stats.rx_packets++;
1394         dev->stats.rx_bytes += skb->len;
1395
1396         /* skip the wrapping header */
1397         eth = (struct ethhdr *) skb_pull(skb, sizeof(struct ethhdr));
1398         if (!eth)
1399                 return RX_DROP_UNUSABLE;
1400
1401         while (skb != frame) {
1402                 u8 padding;
1403                 __be16 len = eth->h_proto;
1404                 unsigned int subframe_len = sizeof(struct ethhdr) + ntohs(len);
1405
1406                 remaining = skb->len;
1407                 memcpy(dst, eth->h_dest, ETH_ALEN);
1408                 memcpy(src, eth->h_source, ETH_ALEN);
1409
1410                 padding = ((4 - subframe_len) & 0x3);
1411                 /* the last MSDU has no padding */
1412                 if (subframe_len > remaining)
1413                         return RX_DROP_UNUSABLE;
1414
1415                 skb_pull(skb, sizeof(struct ethhdr));
1416                 /* if last subframe reuse skb */
1417                 if (remaining <= subframe_len + padding)
1418                         frame = skb;
1419                 else {
1420                         /*
1421                          * Allocate and reserve two bytes more for payload
1422                          * alignment since sizeof(struct ethhdr) is 14.
1423                          */
1424                         frame = dev_alloc_skb(
1425                                 ALIGN(local->hw.extra_tx_headroom, 4) +
1426                                 subframe_len + 2);
1427
1428                         if (frame == NULL)
1429                                 return RX_DROP_UNUSABLE;
1430
1431                         skb_reserve(frame,
1432                                     ALIGN(local->hw.extra_tx_headroom, 4) +
1433                                     sizeof(struct ethhdr) + 2);
1434                         memcpy(skb_put(frame, ntohs(len)), skb->data,
1435                                 ntohs(len));
1436
1437                         eth = (struct ethhdr *) skb_pull(skb, ntohs(len) +
1438                                                         padding);
1439                         if (!eth) {
1440                                 dev_kfree_skb(frame);
1441                                 return RX_DROP_UNUSABLE;
1442                         }
1443                 }
1444
1445                 skb_reset_network_header(frame);
1446                 frame->dev = dev;
1447                 frame->priority = skb->priority;
1448                 rx->skb = frame;
1449
1450                 payload = frame->data;
1451                 ethertype = (payload[6] << 8) | payload[7];
1452
1453                 if (likely((compare_ether_addr(payload, rfc1042_header) == 0 &&
1454                             ethertype != ETH_P_AARP && ethertype != ETH_P_IPX) ||
1455                            compare_ether_addr(payload,
1456                                               bridge_tunnel_header) == 0)) {
1457                         /* remove RFC1042 or Bridge-Tunnel
1458                          * encapsulation and replace EtherType */
1459                         skb_pull(frame, 6);
1460                         memcpy(skb_push(frame, ETH_ALEN), src, ETH_ALEN);
1461                         memcpy(skb_push(frame, ETH_ALEN), dst, ETH_ALEN);
1462                 } else {
1463                         memcpy(skb_push(frame, sizeof(__be16)),
1464                                &len, sizeof(__be16));
1465                         memcpy(skb_push(frame, ETH_ALEN), src, ETH_ALEN);
1466                         memcpy(skb_push(frame, ETH_ALEN), dst, ETH_ALEN);
1467                 }
1468
1469                 if (!ieee80211_frame_allowed(rx, fc)) {
1470                         if (skb == frame) /* last frame */
1471                                 return RX_DROP_UNUSABLE;
1472                         dev_kfree_skb(frame);
1473                         continue;
1474                 }
1475
1476                 ieee80211_deliver_skb(rx);
1477         }
1478
1479         return RX_QUEUED;
1480 }
1481
1482 #ifdef CONFIG_MAC80211_MESH
1483 static ieee80211_rx_result
1484 ieee80211_rx_h_mesh_fwding(struct ieee80211_rx_data *rx)
1485 {
1486         struct ieee80211_hdr *hdr;
1487         struct ieee80211s_hdr *mesh_hdr;
1488         unsigned int hdrlen;
1489         struct sk_buff *skb = rx->skb, *fwd_skb;
1490         struct ieee80211_local *local = rx->local;
1491         struct ieee80211_sub_if_data *sdata;
1492
1493         hdr = (struct ieee80211_hdr *) skb->data;
1494         hdrlen = ieee80211_hdrlen(hdr->frame_control);
1495         mesh_hdr = (struct ieee80211s_hdr *) (skb->data + hdrlen);
1496         sdata = IEEE80211_DEV_TO_SUB_IF(rx->dev);
1497
1498         if (!ieee80211_is_data(hdr->frame_control))
1499                 return RX_CONTINUE;
1500
1501         if (!mesh_hdr->ttl)
1502                 /* illegal frame */
1503                 return RX_DROP_MONITOR;
1504
1505         if (mesh_hdr->flags & MESH_FLAGS_AE) {
1506                 struct mesh_path *mppath;
1507                 char *proxied_addr;
1508                 char *mpp_addr;
1509
1510                 if (is_multicast_ether_addr(hdr->addr1)) {
1511                         mpp_addr = hdr->addr3;
1512                         proxied_addr = mesh_hdr->eaddr1;
1513                 } else {
1514                         mpp_addr = hdr->addr4;
1515                         proxied_addr = mesh_hdr->eaddr2;
1516                 }
1517
1518                 rcu_read_lock();
1519                 mppath = mpp_path_lookup(proxied_addr, sdata);
1520                 if (!mppath) {
1521                         mpp_path_add(proxied_addr, mpp_addr, sdata);
1522                 } else {
1523                         spin_lock_bh(&mppath->state_lock);
1524                         mppath->exp_time = jiffies;
1525                         if (compare_ether_addr(mppath->mpp, mpp_addr) != 0)
1526                                 memcpy(mppath->mpp, mpp_addr, ETH_ALEN);
1527                         spin_unlock_bh(&mppath->state_lock);
1528                 }
1529                 rcu_read_unlock();
1530         }
1531
1532         /* Frame has reached destination.  Don't forward */
1533         if (!is_multicast_ether_addr(hdr->addr1) &&
1534                         compare_ether_addr(rx->dev->dev_addr, hdr->addr3) == 0)
1535                 return RX_CONTINUE;
1536
1537         mesh_hdr->ttl--;
1538
1539         if (rx->flags & IEEE80211_RX_RA_MATCH) {
1540                 if (!mesh_hdr->ttl)
1541                         IEEE80211_IFSTA_MESH_CTR_INC(&rx->sdata->u.mesh,
1542                                                      dropped_frames_ttl);
1543                 else {
1544                         struct ieee80211_hdr *fwd_hdr;
1545                         struct ieee80211_tx_info *info;
1546
1547                         fwd_skb = skb_copy(skb, GFP_ATOMIC);
1548
1549                         if (!fwd_skb && net_ratelimit())
1550                                 printk(KERN_DEBUG "%s: failed to clone mesh frame\n",
1551                                                    rx->dev->name);
1552
1553                         fwd_hdr =  (struct ieee80211_hdr *) fwd_skb->data;
1554                         memcpy(fwd_hdr->addr2, rx->dev->dev_addr, ETH_ALEN);
1555                         info = IEEE80211_SKB_CB(fwd_skb);
1556                         memset(info, 0, sizeof(*info));
1557                         info->flags |= IEEE80211_TX_INTFL_NEED_TXPROCESSING;
1558                         info->control.vif = &rx->sdata->vif;
1559                         ieee80211_select_queue(local, fwd_skb);
1560                         if (is_multicast_ether_addr(fwd_hdr->addr1))
1561                                 IEEE80211_IFSTA_MESH_CTR_INC(&sdata->u.mesh,
1562                                                                 fwded_mcast);
1563                         else {
1564                                 int err;
1565                                 /*
1566                                  * Save TA to addr1 to send TA a path error if a
1567                                  * suitable next hop is not found
1568                                  */
1569                                 memcpy(fwd_hdr->addr1, fwd_hdr->addr2,
1570                                                 ETH_ALEN);
1571                                 err = mesh_nexthop_lookup(fwd_skb, sdata);
1572                                 /* Failed to immediately resolve next hop:
1573                                  * fwded frame was dropped or will be added
1574                                  * later to the pending skb queue.  */
1575                                 if (err)
1576                                         return RX_DROP_MONITOR;
1577
1578                                 IEEE80211_IFSTA_MESH_CTR_INC(&sdata->u.mesh,
1579                                                                 fwded_unicast);
1580                         }
1581                         IEEE80211_IFSTA_MESH_CTR_INC(&sdata->u.mesh,
1582                                                      fwded_frames);
1583                         ieee80211_add_pending_skb(local, fwd_skb);
1584                 }
1585         }
1586
1587         if (is_multicast_ether_addr(hdr->addr1) ||
1588             rx->dev->flags & IFF_PROMISC)
1589                 return RX_CONTINUE;
1590         else
1591                 return RX_DROP_MONITOR;
1592 }
1593 #endif
1594
1595 static ieee80211_rx_result debug_noinline
1596 ieee80211_rx_h_data(struct ieee80211_rx_data *rx)
1597 {
1598         struct net_device *dev = rx->dev;
1599         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
1600         __le16 fc = hdr->frame_control;
1601         int err;
1602
1603         if (unlikely(!ieee80211_is_data(hdr->frame_control)))
1604                 return RX_CONTINUE;
1605
1606         if (unlikely(!ieee80211_is_data_present(hdr->frame_control)))
1607                 return RX_DROP_MONITOR;
1608
1609         err = __ieee80211_data_to_8023(rx);
1610         if (unlikely(err))
1611                 return RX_DROP_UNUSABLE;
1612
1613         if (!ieee80211_frame_allowed(rx, fc))
1614                 return RX_DROP_MONITOR;
1615
1616         rx->skb->dev = dev;
1617
1618         dev->stats.rx_packets++;
1619         dev->stats.rx_bytes += rx->skb->len;
1620
1621         ieee80211_deliver_skb(rx);
1622
1623         return RX_QUEUED;
1624 }
1625
1626 static ieee80211_rx_result debug_noinline
1627 ieee80211_rx_h_ctrl(struct ieee80211_rx_data *rx)
1628 {
1629         struct ieee80211_local *local = rx->local;
1630         struct ieee80211_hw *hw = &local->hw;
1631         struct sk_buff *skb = rx->skb;
1632         struct ieee80211_bar *bar = (struct ieee80211_bar *)skb->data;
1633         struct tid_ampdu_rx *tid_agg_rx;
1634         u16 start_seq_num;
1635         u16 tid;
1636
1637         if (likely(!ieee80211_is_ctl(bar->frame_control)))
1638                 return RX_CONTINUE;
1639
1640         if (ieee80211_is_back_req(bar->frame_control)) {
1641                 if (!rx->sta)
1642                         return RX_CONTINUE;
1643                 tid = le16_to_cpu(bar->control) >> 12;
1644                 if (rx->sta->ampdu_mlme.tid_state_rx[tid]
1645                                         != HT_AGG_STATE_OPERATIONAL)
1646                         return RX_CONTINUE;
1647                 tid_agg_rx = rx->sta->ampdu_mlme.tid_rx[tid];
1648
1649                 start_seq_num = le16_to_cpu(bar->start_seq_num) >> 4;
1650
1651                 /* reset session timer */
1652                 if (tid_agg_rx->timeout)
1653                         mod_timer(&tid_agg_rx->session_timer,
1654                                   TU_TO_EXP_TIME(tid_agg_rx->timeout));
1655
1656                 /* manage reordering buffer according to requested */
1657                 /* sequence number */
1658                 rcu_read_lock();
1659                 ieee80211_sta_manage_reorder_buf(hw, tid_agg_rx, NULL,
1660                                                  start_seq_num, 1);
1661                 rcu_read_unlock();
1662                 return RX_DROP_UNUSABLE;
1663         }
1664
1665         return RX_CONTINUE;
1666 }
1667
1668 static void ieee80211_process_sa_query_req(struct ieee80211_sub_if_data *sdata,
1669                                            struct ieee80211_mgmt *mgmt,
1670                                            size_t len)
1671 {
1672         struct ieee80211_local *local = sdata->local;
1673         struct sk_buff *skb;
1674         struct ieee80211_mgmt *resp;
1675
1676         if (compare_ether_addr(mgmt->da, sdata->dev->dev_addr) != 0) {
1677                 /* Not to own unicast address */
1678                 return;
1679         }
1680
1681         if (compare_ether_addr(mgmt->sa, sdata->u.mgd.bssid) != 0 ||
1682             compare_ether_addr(mgmt->bssid, sdata->u.mgd.bssid) != 0) {
1683                 /* Not from the current AP or not associated yet. */
1684                 return;
1685         }
1686
1687         if (len < 24 + 1 + sizeof(resp->u.action.u.sa_query)) {
1688                 /* Too short SA Query request frame */
1689                 return;
1690         }
1691
1692         skb = dev_alloc_skb(sizeof(*resp) + local->hw.extra_tx_headroom);
1693         if (skb == NULL)
1694                 return;
1695
1696         skb_reserve(skb, local->hw.extra_tx_headroom);
1697         resp = (struct ieee80211_mgmt *) skb_put(skb, 24);
1698         memset(resp, 0, 24);
1699         memcpy(resp->da, mgmt->sa, ETH_ALEN);
1700         memcpy(resp->sa, sdata->dev->dev_addr, ETH_ALEN);
1701         memcpy(resp->bssid, sdata->u.mgd.bssid, ETH_ALEN);
1702         resp->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
1703                                           IEEE80211_STYPE_ACTION);
1704         skb_put(skb, 1 + sizeof(resp->u.action.u.sa_query));
1705         resp->u.action.category = WLAN_CATEGORY_SA_QUERY;
1706         resp->u.action.u.sa_query.action = WLAN_ACTION_SA_QUERY_RESPONSE;
1707         memcpy(resp->u.action.u.sa_query.trans_id,
1708                mgmt->u.action.u.sa_query.trans_id,
1709                WLAN_SA_QUERY_TR_ID_LEN);
1710
1711         ieee80211_tx_skb(sdata, skb, 1);
1712 }
1713
1714 static ieee80211_rx_result debug_noinline
1715 ieee80211_rx_h_action(struct ieee80211_rx_data *rx)
1716 {
1717         struct ieee80211_local *local = rx->local;
1718         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(rx->dev);
1719         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *) rx->skb->data;
1720         int len = rx->skb->len;
1721
1722         if (!ieee80211_is_action(mgmt->frame_control))
1723                 return RX_CONTINUE;
1724
1725         if (!rx->sta)
1726                 return RX_DROP_MONITOR;
1727
1728         if (!(rx->flags & IEEE80211_RX_RA_MATCH))
1729                 return RX_DROP_MONITOR;
1730
1731         if (ieee80211_drop_unencrypted(rx, mgmt->frame_control))
1732                 return RX_DROP_MONITOR;
1733
1734         /* all categories we currently handle have action_code */
1735         if (len < IEEE80211_MIN_ACTION_SIZE + 1)
1736                 return RX_DROP_MONITOR;
1737
1738         switch (mgmt->u.action.category) {
1739         case WLAN_CATEGORY_BACK:
1740                 /*
1741                  * The aggregation code is not prepared to handle
1742                  * anything but STA/AP due to the BSSID handling;
1743                  * IBSS could work in the code but isn't supported
1744                  * by drivers or the standard.
1745                  */
1746                 if (sdata->vif.type != NL80211_IFTYPE_STATION &&
1747                     sdata->vif.type != NL80211_IFTYPE_AP_VLAN &&
1748                     sdata->vif.type != NL80211_IFTYPE_AP)
1749                         return RX_DROP_MONITOR;
1750
1751                 switch (mgmt->u.action.u.addba_req.action_code) {
1752                 case WLAN_ACTION_ADDBA_REQ:
1753                         if (len < (IEEE80211_MIN_ACTION_SIZE +
1754                                    sizeof(mgmt->u.action.u.addba_req)))
1755                                 return RX_DROP_MONITOR;
1756                         ieee80211_process_addba_request(local, rx->sta, mgmt, len);
1757                         break;
1758                 case WLAN_ACTION_ADDBA_RESP:
1759                         if (len < (IEEE80211_MIN_ACTION_SIZE +
1760                                    sizeof(mgmt->u.action.u.addba_resp)))
1761                                 return RX_DROP_MONITOR;
1762                         ieee80211_process_addba_resp(local, rx->sta, mgmt, len);
1763                         break;
1764                 case WLAN_ACTION_DELBA:
1765                         if (len < (IEEE80211_MIN_ACTION_SIZE +
1766                                    sizeof(mgmt->u.action.u.delba)))
1767                                 return RX_DROP_MONITOR;
1768                         ieee80211_process_delba(sdata, rx->sta, mgmt, len);
1769                         break;
1770                 }
1771                 break;
1772         case WLAN_CATEGORY_SPECTRUM_MGMT:
1773                 if (local->hw.conf.channel->band != IEEE80211_BAND_5GHZ)
1774                         return RX_DROP_MONITOR;
1775
1776                 if (sdata->vif.type != NL80211_IFTYPE_STATION)
1777                         return RX_DROP_MONITOR;
1778
1779                 switch (mgmt->u.action.u.measurement.action_code) {
1780                 case WLAN_ACTION_SPCT_MSR_REQ:
1781                         if (len < (IEEE80211_MIN_ACTION_SIZE +
1782                                    sizeof(mgmt->u.action.u.measurement)))
1783                                 return RX_DROP_MONITOR;
1784                         ieee80211_process_measurement_req(sdata, mgmt, len);
1785                         break;
1786                 case WLAN_ACTION_SPCT_CHL_SWITCH:
1787                         if (len < (IEEE80211_MIN_ACTION_SIZE +
1788                                    sizeof(mgmt->u.action.u.chan_switch)))
1789                                 return RX_DROP_MONITOR;
1790
1791                         if (sdata->vif.type != NL80211_IFTYPE_STATION)
1792                                 return RX_DROP_MONITOR;
1793
1794                         if (memcmp(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN))
1795                                 return RX_DROP_MONITOR;
1796
1797                         return ieee80211_sta_rx_mgmt(sdata, rx->skb);
1798                 }
1799                 break;
1800         case WLAN_CATEGORY_SA_QUERY:
1801                 if (len < (IEEE80211_MIN_ACTION_SIZE +
1802                            sizeof(mgmt->u.action.u.sa_query)))
1803                         return RX_DROP_MONITOR;
1804                 switch (mgmt->u.action.u.sa_query.action) {
1805                 case WLAN_ACTION_SA_QUERY_REQUEST:
1806                         if (sdata->vif.type != NL80211_IFTYPE_STATION)
1807                                 return RX_DROP_MONITOR;
1808                         ieee80211_process_sa_query_req(sdata, mgmt, len);
1809                         break;
1810                 case WLAN_ACTION_SA_QUERY_RESPONSE:
1811                         /*
1812                          * SA Query response is currently only used in AP mode
1813                          * and it is processed in user space.
1814                          */
1815                         return RX_CONTINUE;
1816                 }
1817                 break;
1818         default:
1819                 return RX_CONTINUE;
1820         }
1821
1822         rx->sta->rx_packets++;
1823         dev_kfree_skb(rx->skb);
1824         return RX_QUEUED;
1825 }
1826
1827 static ieee80211_rx_result debug_noinline
1828 ieee80211_rx_h_mgmt(struct ieee80211_rx_data *rx)
1829 {
1830         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(rx->dev);
1831         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *) rx->skb->data;
1832
1833         if (!(rx->flags & IEEE80211_RX_RA_MATCH))
1834                 return RX_DROP_MONITOR;
1835
1836         if (ieee80211_drop_unencrypted(rx, mgmt->frame_control))
1837                 return RX_DROP_MONITOR;
1838
1839         if (ieee80211_vif_is_mesh(&sdata->vif))
1840                 return ieee80211_mesh_rx_mgmt(sdata, rx->skb);
1841
1842         if (sdata->vif.type == NL80211_IFTYPE_ADHOC)
1843                 return ieee80211_ibss_rx_mgmt(sdata, rx->skb);
1844
1845         if (sdata->vif.type == NL80211_IFTYPE_STATION)
1846                 return ieee80211_sta_rx_mgmt(sdata, rx->skb);
1847
1848         return RX_DROP_MONITOR;
1849 }
1850
1851 static void ieee80211_rx_michael_mic_report(struct ieee80211_hdr *hdr,
1852                                             struct ieee80211_rx_data *rx)
1853 {
1854         int keyidx;
1855         unsigned int hdrlen;
1856
1857         hdrlen = ieee80211_hdrlen(hdr->frame_control);
1858         if (rx->skb->len >= hdrlen + 4)
1859                 keyidx = rx->skb->data[hdrlen + 3] >> 6;
1860         else
1861                 keyidx = -1;
1862
1863         if (!rx->sta) {
1864                 /*
1865                  * Some hardware seem to generate incorrect Michael MIC
1866                  * reports; ignore them to avoid triggering countermeasures.
1867                  */
1868                 goto ignore;
1869         }
1870
1871         if (!ieee80211_has_protected(hdr->frame_control))
1872                 goto ignore;
1873
1874         if (rx->sdata->vif.type == NL80211_IFTYPE_AP && keyidx) {
1875                 /*
1876                  * APs with pairwise keys should never receive Michael MIC
1877                  * errors for non-zero keyidx because these are reserved for
1878                  * group keys and only the AP is sending real multicast
1879                  * frames in the BSS.
1880                  */
1881                 goto ignore;
1882         }
1883
1884         if (!ieee80211_is_data(hdr->frame_control) &&
1885             !ieee80211_is_auth(hdr->frame_control))
1886                 goto ignore;
1887
1888         mac80211_ev_michael_mic_failure(rx->sdata, keyidx, hdr, NULL,
1889                                         GFP_ATOMIC);
1890  ignore:
1891         dev_kfree_skb(rx->skb);
1892         rx->skb = NULL;
1893 }
1894
1895 /* TODO: use IEEE80211_RX_FRAGMENTED */
1896 static void ieee80211_rx_cooked_monitor(struct ieee80211_rx_data *rx)
1897 {
1898         struct ieee80211_sub_if_data *sdata;
1899         struct ieee80211_local *local = rx->local;
1900         struct ieee80211_rtap_hdr {
1901                 struct ieee80211_radiotap_header hdr;
1902                 u8 flags;
1903                 u8 rate;
1904                 __le16 chan_freq;
1905                 __le16 chan_flags;
1906         } __attribute__ ((packed)) *rthdr;
1907         struct sk_buff *skb = rx->skb, *skb2;
1908         struct net_device *prev_dev = NULL;
1909         struct ieee80211_rx_status *status = rx->status;
1910
1911         if (rx->flags & IEEE80211_RX_CMNTR_REPORTED)
1912                 goto out_free_skb;
1913
1914         if (skb_headroom(skb) < sizeof(*rthdr) &&
1915             pskb_expand_head(skb, sizeof(*rthdr), 0, GFP_ATOMIC))
1916                 goto out_free_skb;
1917
1918         rthdr = (void *)skb_push(skb, sizeof(*rthdr));
1919         memset(rthdr, 0, sizeof(*rthdr));
1920         rthdr->hdr.it_len = cpu_to_le16(sizeof(*rthdr));
1921         rthdr->hdr.it_present =
1922                 cpu_to_le32((1 << IEEE80211_RADIOTAP_FLAGS) |
1923                             (1 << IEEE80211_RADIOTAP_RATE) |
1924                             (1 << IEEE80211_RADIOTAP_CHANNEL));
1925
1926         rthdr->rate = rx->rate->bitrate / 5;
1927         rthdr->chan_freq = cpu_to_le16(status->freq);
1928
1929         if (status->band == IEEE80211_BAND_5GHZ)
1930                 rthdr->chan_flags = cpu_to_le16(IEEE80211_CHAN_OFDM |
1931                                                 IEEE80211_CHAN_5GHZ);
1932         else
1933                 rthdr->chan_flags = cpu_to_le16(IEEE80211_CHAN_DYN |
1934                                                 IEEE80211_CHAN_2GHZ);
1935
1936         skb_set_mac_header(skb, 0);
1937         skb->ip_summed = CHECKSUM_UNNECESSARY;
1938         skb->pkt_type = PACKET_OTHERHOST;
1939         skb->protocol = htons(ETH_P_802_2);
1940
1941         list_for_each_entry_rcu(sdata, &local->interfaces, list) {
1942                 if (!netif_running(sdata->dev))
1943                         continue;
1944
1945                 if (sdata->vif.type != NL80211_IFTYPE_MONITOR ||
1946                     !(sdata->u.mntr_flags & MONITOR_FLAG_COOK_FRAMES))
1947                         continue;
1948
1949                 if (prev_dev) {
1950                         skb2 = skb_clone(skb, GFP_ATOMIC);
1951                         if (skb2) {
1952                                 skb2->dev = prev_dev;
1953                                 netif_rx(skb2);
1954                         }
1955                 }
1956
1957                 prev_dev = sdata->dev;
1958                 sdata->dev->stats.rx_packets++;
1959                 sdata->dev->stats.rx_bytes += skb->len;
1960         }
1961
1962         if (prev_dev) {
1963                 skb->dev = prev_dev;
1964                 netif_rx(skb);
1965                 skb = NULL;
1966         } else
1967                 goto out_free_skb;
1968
1969         rx->flags |= IEEE80211_RX_CMNTR_REPORTED;
1970         return;
1971
1972  out_free_skb:
1973         dev_kfree_skb(skb);
1974 }
1975
1976
1977 static void ieee80211_invoke_rx_handlers(struct ieee80211_sub_if_data *sdata,
1978                                          struct ieee80211_rx_data *rx,
1979                                          struct sk_buff *skb)
1980 {
1981         ieee80211_rx_result res = RX_DROP_MONITOR;
1982
1983         rx->skb = skb;
1984         rx->sdata = sdata;
1985         rx->dev = sdata->dev;
1986
1987 #define CALL_RXH(rxh)                   \
1988         do {                            \
1989                 res = rxh(rx);          \
1990                 if (res != RX_CONTINUE) \
1991                         goto rxh_done;  \
1992         } while (0);
1993
1994         CALL_RXH(ieee80211_rx_h_passive_scan)
1995         CALL_RXH(ieee80211_rx_h_check)
1996         CALL_RXH(ieee80211_rx_h_decrypt)
1997         CALL_RXH(ieee80211_rx_h_check_more_data)
1998         CALL_RXH(ieee80211_rx_h_sta_process)
1999         CALL_RXH(ieee80211_rx_h_defragment)
2000         CALL_RXH(ieee80211_rx_h_ps_poll)
2001         CALL_RXH(ieee80211_rx_h_michael_mic_verify)
2002         /* must be after MMIC verify so header is counted in MPDU mic */
2003         CALL_RXH(ieee80211_rx_h_remove_qos_control)
2004         CALL_RXH(ieee80211_rx_h_amsdu)
2005 #ifdef CONFIG_MAC80211_MESH
2006         if (ieee80211_vif_is_mesh(&sdata->vif))
2007                 CALL_RXH(ieee80211_rx_h_mesh_fwding);
2008 #endif
2009         CALL_RXH(ieee80211_rx_h_data)
2010         CALL_RXH(ieee80211_rx_h_ctrl)
2011         CALL_RXH(ieee80211_rx_h_action)
2012         CALL_RXH(ieee80211_rx_h_mgmt)
2013
2014 #undef CALL_RXH
2015
2016  rxh_done:
2017         switch (res) {
2018         case RX_DROP_MONITOR:
2019                 I802_DEBUG_INC(sdata->local->rx_handlers_drop);
2020                 if (rx->sta)
2021                         rx->sta->rx_dropped++;
2022                 /* fall through */
2023         case RX_CONTINUE:
2024                 ieee80211_rx_cooked_monitor(rx);
2025                 break;
2026         case RX_DROP_UNUSABLE:
2027                 I802_DEBUG_INC(sdata->local->rx_handlers_drop);
2028                 if (rx->sta)
2029                         rx->sta->rx_dropped++;
2030                 dev_kfree_skb(rx->skb);
2031                 break;
2032         case RX_QUEUED:
2033                 I802_DEBUG_INC(sdata->local->rx_handlers_queued);
2034                 break;
2035         }
2036 }
2037
2038 /* main receive path */
2039
2040 static int prepare_for_handlers(struct ieee80211_sub_if_data *sdata,
2041                                 struct ieee80211_rx_data *rx,
2042                                 struct ieee80211_hdr *hdr)
2043 {
2044         u8 *bssid = ieee80211_get_bssid(hdr, rx->skb->len, sdata->vif.type);
2045         int multicast = is_multicast_ether_addr(hdr->addr1);
2046
2047         switch (sdata->vif.type) {
2048         case NL80211_IFTYPE_STATION:
2049                 if (!bssid)
2050                         return 0;
2051                 if (!multicast &&
2052                     compare_ether_addr(sdata->dev->dev_addr, hdr->addr1) != 0) {
2053                         if (!(sdata->dev->flags & IFF_PROMISC))
2054                                 return 0;
2055                         rx->flags &= ~IEEE80211_RX_RA_MATCH;
2056                 }
2057                 break;
2058         case NL80211_IFTYPE_ADHOC:
2059                 if (!bssid)
2060                         return 0;
2061                 if (ieee80211_is_beacon(hdr->frame_control)) {
2062                         return 1;
2063                 }
2064                 else if (!ieee80211_bssid_match(bssid, sdata->u.ibss.bssid)) {
2065                         if (!(rx->flags & IEEE80211_RX_IN_SCAN))
2066                                 return 0;
2067                         rx->flags &= ~IEEE80211_RX_RA_MATCH;
2068                 } else if (!multicast &&
2069                            compare_ether_addr(sdata->dev->dev_addr,
2070                                               hdr->addr1) != 0) {
2071                         if (!(sdata->dev->flags & IFF_PROMISC))
2072                                 return 0;
2073                         rx->flags &= ~IEEE80211_RX_RA_MATCH;
2074                 } else if (!rx->sta) {
2075                         int rate_idx;
2076                         if (rx->status->flag & RX_FLAG_HT)
2077                                 rate_idx = 0; /* TODO: HT rates */
2078                         else
2079                                 rate_idx = rx->status->rate_idx;
2080                         rx->sta = ieee80211_ibss_add_sta(sdata, bssid, hdr->addr2,
2081                                 BIT(rate_idx));
2082                 }
2083                 break;
2084         case NL80211_IFTYPE_MESH_POINT:
2085                 if (!multicast &&
2086                     compare_ether_addr(sdata->dev->dev_addr,
2087                                        hdr->addr1) != 0) {
2088                         if (!(sdata->dev->flags & IFF_PROMISC))
2089                                 return 0;
2090
2091                         rx->flags &= ~IEEE80211_RX_RA_MATCH;
2092                 }
2093                 break;
2094         case NL80211_IFTYPE_AP_VLAN:
2095         case NL80211_IFTYPE_AP:
2096                 if (!bssid) {
2097                         if (compare_ether_addr(sdata->dev->dev_addr,
2098                                                hdr->addr1))
2099                                 return 0;
2100                 } else if (!ieee80211_bssid_match(bssid,
2101                                         sdata->dev->dev_addr)) {
2102                         if (!(rx->flags & IEEE80211_RX_IN_SCAN))
2103                                 return 0;
2104                         rx->flags &= ~IEEE80211_RX_RA_MATCH;
2105                 }
2106                 break;
2107         case NL80211_IFTYPE_WDS:
2108                 if (bssid || !ieee80211_is_data(hdr->frame_control))
2109                         return 0;
2110                 if (compare_ether_addr(sdata->u.wds.remote_addr, hdr->addr2))
2111                         return 0;
2112                 break;
2113         case NL80211_IFTYPE_MONITOR:
2114                 /* take everything */
2115                 break;
2116         case NL80211_IFTYPE_UNSPECIFIED:
2117         case __NL80211_IFTYPE_AFTER_LAST:
2118                 /* should never get here */
2119                 WARN_ON(1);
2120                 break;
2121         }
2122
2123         return 1;
2124 }
2125
2126 /*
2127  * This is the actual Rx frames handler. as it blongs to Rx path it must
2128  * be called with rcu_read_lock protection.
2129  */
2130 static void __ieee80211_rx_handle_packet(struct ieee80211_hw *hw,
2131                                          struct sk_buff *skb,
2132                                          struct ieee80211_rate *rate)
2133 {
2134         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
2135         struct ieee80211_local *local = hw_to_local(hw);
2136         struct ieee80211_sub_if_data *sdata;
2137         struct ieee80211_hdr *hdr;
2138         struct ieee80211_rx_data rx;
2139         int prepares;
2140         struct ieee80211_sub_if_data *prev = NULL;
2141         struct sk_buff *skb_new;
2142
2143         hdr = (struct ieee80211_hdr *)skb->data;
2144         memset(&rx, 0, sizeof(rx));
2145         rx.skb = skb;
2146         rx.local = local;
2147
2148         rx.status = status;
2149         rx.rate = rate;
2150
2151         if (ieee80211_is_data(hdr->frame_control) || ieee80211_is_mgmt(hdr->frame_control))
2152                 local->dot11ReceivedFragmentCount++;
2153
2154         rx.sta = sta_info_get(local, hdr->addr2);
2155         if (rx.sta) {
2156                 rx.sdata = rx.sta->sdata;
2157                 rx.dev = rx.sta->sdata->dev;
2158         }
2159
2160         if ((status->flag & RX_FLAG_MMIC_ERROR)) {
2161                 ieee80211_rx_michael_mic_report(hdr, &rx);
2162                 return;
2163         }
2164
2165         if (unlikely(test_bit(SCAN_HW_SCANNING, &local->scanning) ||
2166                      test_bit(SCAN_OFF_CHANNEL, &local->scanning)))
2167                 rx.flags |= IEEE80211_RX_IN_SCAN;
2168
2169         ieee80211_parse_qos(&rx);
2170         ieee80211_verify_alignment(&rx);
2171
2172         skb = rx.skb;
2173
2174         if (rx.sdata && ieee80211_is_data(hdr->frame_control)) {
2175                 rx.flags |= IEEE80211_RX_RA_MATCH;
2176                 prepares = prepare_for_handlers(rx.sdata, &rx, hdr);
2177                 if (prepares)
2178                         prev = rx.sdata;
2179         } else list_for_each_entry_rcu(sdata, &local->interfaces, list) {
2180                 if (!netif_running(sdata->dev))
2181                         continue;
2182
2183                 if (sdata->vif.type == NL80211_IFTYPE_MONITOR ||
2184                     sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
2185                         continue;
2186
2187                 rx.flags |= IEEE80211_RX_RA_MATCH;
2188                 prepares = prepare_for_handlers(sdata, &rx, hdr);
2189
2190                 if (!prepares)
2191                         continue;
2192
2193                 /*
2194                  * frame is destined for this interface, but if it's not
2195                  * also for the previous one we handle that after the
2196                  * loop to avoid copying the SKB once too much
2197                  */
2198
2199                 if (!prev) {
2200                         prev = sdata;
2201                         continue;
2202                 }
2203
2204                 /*
2205                  * frame was destined for the previous interface
2206                  * so invoke RX handlers for it
2207                  */
2208
2209                 skb_new = skb_copy(skb, GFP_ATOMIC);
2210                 if (!skb_new) {
2211                         if (net_ratelimit())
2212                                 printk(KERN_DEBUG "%s: failed to copy "
2213                                        "multicast frame for %s\n",
2214                                        wiphy_name(local->hw.wiphy),
2215                                        prev->dev->name);
2216                         continue;
2217                 }
2218                 ieee80211_invoke_rx_handlers(prev, &rx, skb_new);
2219                 prev = sdata;
2220         }
2221         if (prev)
2222                 ieee80211_invoke_rx_handlers(prev, &rx, skb);
2223         else
2224                 dev_kfree_skb(skb);
2225 }
2226
2227 #define SEQ_MODULO 0x1000
2228 #define SEQ_MASK   0xfff
2229
2230 static inline int seq_less(u16 sq1, u16 sq2)
2231 {
2232         return ((sq1 - sq2) & SEQ_MASK) > (SEQ_MODULO >> 1);
2233 }
2234
2235 static inline u16 seq_inc(u16 sq)
2236 {
2237         return (sq + 1) & SEQ_MASK;
2238 }
2239
2240 static inline u16 seq_sub(u16 sq1, u16 sq2)
2241 {
2242         return (sq1 - sq2) & SEQ_MASK;
2243 }
2244
2245
2246 static void ieee80211_release_reorder_frame(struct ieee80211_hw *hw,
2247                                             struct tid_ampdu_rx *tid_agg_rx,
2248                                             int index)
2249 {
2250         struct ieee80211_supported_band *sband;
2251         struct ieee80211_rate *rate;
2252         struct sk_buff *skb = tid_agg_rx->reorder_buf[index];
2253         struct ieee80211_rx_status *status;
2254
2255         if (!skb)
2256                 goto no_frame;
2257
2258         status = IEEE80211_SKB_RXCB(skb);
2259
2260         /* release the reordered frames to stack */
2261         sband = hw->wiphy->bands[status->band];
2262         if (status->flag & RX_FLAG_HT)
2263                 rate = sband->bitrates; /* TODO: HT rates */
2264         else
2265                 rate = &sband->bitrates[status->rate_idx];
2266         __ieee80211_rx_handle_packet(hw, skb, rate);
2267         tid_agg_rx->stored_mpdu_num--;
2268         tid_agg_rx->reorder_buf[index] = NULL;
2269
2270 no_frame:
2271         tid_agg_rx->head_seq_num = seq_inc(tid_agg_rx->head_seq_num);
2272 }
2273
2274
2275 /*
2276  * Timeout (in jiffies) for skb's that are waiting in the RX reorder buffer. If
2277  * the skb was added to the buffer longer than this time ago, the earlier
2278  * frames that have not yet been received are assumed to be lost and the skb
2279  * can be released for processing. This may also release other skb's from the
2280  * reorder buffer if there are no additional gaps between the frames.
2281  */
2282 #define HT_RX_REORDER_BUF_TIMEOUT (HZ / 10)
2283
2284 /*
2285  * As it function blongs to Rx path it must be called with
2286  * the proper rcu_read_lock protection for its flow.
2287  */
2288 static u8 ieee80211_sta_manage_reorder_buf(struct ieee80211_hw *hw,
2289                                            struct tid_ampdu_rx *tid_agg_rx,
2290                                            struct sk_buff *skb,
2291                                            u16 mpdu_seq_num,
2292                                            int bar_req)
2293 {
2294         u16 head_seq_num, buf_size;
2295         int index;
2296
2297         buf_size = tid_agg_rx->buf_size;
2298         head_seq_num = tid_agg_rx->head_seq_num;
2299
2300         /* frame with out of date sequence number */
2301         if (seq_less(mpdu_seq_num, head_seq_num)) {
2302                 dev_kfree_skb(skb);
2303                 return 1;
2304         }
2305
2306         /* if frame sequence number exceeds our buffering window size or
2307          * block Ack Request arrived - release stored frames */
2308         if ((!seq_less(mpdu_seq_num, head_seq_num + buf_size)) || (bar_req)) {
2309                 /* new head to the ordering buffer */
2310                 if (bar_req)
2311                         head_seq_num = mpdu_seq_num;
2312                 else
2313                         head_seq_num =
2314                                 seq_inc(seq_sub(mpdu_seq_num, buf_size));
2315                 /* release stored frames up to new head to stack */
2316                 while (seq_less(tid_agg_rx->head_seq_num, head_seq_num)) {
2317                         index = seq_sub(tid_agg_rx->head_seq_num,
2318                                 tid_agg_rx->ssn)
2319                                 % tid_agg_rx->buf_size;
2320                         ieee80211_release_reorder_frame(hw, tid_agg_rx,
2321                                                         index);
2322                 }
2323                 if (bar_req)
2324                         return 1;
2325         }
2326
2327         /* now the new frame is always in the range of the reordering */
2328         /* buffer window */
2329         index = seq_sub(mpdu_seq_num, tid_agg_rx->ssn)
2330                                 % tid_agg_rx->buf_size;
2331         /* check if we already stored this frame */
2332         if (tid_agg_rx->reorder_buf[index]) {
2333                 dev_kfree_skb(skb);
2334                 return 1;
2335         }
2336
2337         /* if arrived mpdu is in the right order and nothing else stored */
2338         /* release it immediately */
2339         if (mpdu_seq_num == tid_agg_rx->head_seq_num &&
2340                         tid_agg_rx->stored_mpdu_num == 0) {
2341                 tid_agg_rx->head_seq_num =
2342                         seq_inc(tid_agg_rx->head_seq_num);
2343                 return 0;
2344         }
2345
2346         /* put the frame in the reordering buffer */
2347         tid_agg_rx->reorder_buf[index] = skb;
2348         tid_agg_rx->reorder_time[index] = jiffies;
2349         tid_agg_rx->stored_mpdu_num++;
2350         /* release the buffer until next missing frame */
2351         index = seq_sub(tid_agg_rx->head_seq_num, tid_agg_rx->ssn)
2352                                                 % tid_agg_rx->buf_size;
2353         if (!tid_agg_rx->reorder_buf[index] &&
2354             tid_agg_rx->stored_mpdu_num > 1) {
2355                 /*
2356                  * No buffers ready to be released, but check whether any
2357                  * frames in the reorder buffer have timed out.
2358                  */
2359                 int j;
2360                 int skipped = 1;
2361                 for (j = (index + 1) % tid_agg_rx->buf_size; j != index;
2362                      j = (j + 1) % tid_agg_rx->buf_size) {
2363                         if (tid_agg_rx->reorder_buf[j] == NULL) {
2364                                 skipped++;
2365                                 continue;
2366                         }
2367                         if (!time_after(jiffies, tid_agg_rx->reorder_time[j] +
2368                                         HZ / 10))
2369                                 break;
2370
2371 #ifdef CONFIG_MAC80211_HT_DEBUG
2372                         if (net_ratelimit())
2373                                 printk(KERN_DEBUG "%s: release an RX reorder "
2374                                        "frame due to timeout on earlier "
2375                                        "frames\n",
2376                                        wiphy_name(hw->wiphy));
2377 #endif
2378                         ieee80211_release_reorder_frame(hw, tid_agg_rx, j);
2379
2380                         /*
2381                          * Increment the head seq# also for the skipped slots.
2382                          */
2383                         tid_agg_rx->head_seq_num =
2384                                 (tid_agg_rx->head_seq_num + skipped) &
2385                                 SEQ_MASK;
2386                         skipped = 0;
2387                 }
2388         } else while (tid_agg_rx->reorder_buf[index]) {
2389                 ieee80211_release_reorder_frame(hw, tid_agg_rx, index);
2390                 index = seq_sub(tid_agg_rx->head_seq_num,
2391                         tid_agg_rx->ssn) % tid_agg_rx->buf_size;
2392         }
2393         return 1;
2394 }
2395
2396 static u8 ieee80211_rx_reorder_ampdu(struct ieee80211_local *local,
2397                                      struct sk_buff *skb)
2398 {
2399         struct ieee80211_hw *hw = &local->hw;
2400         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
2401         struct sta_info *sta;
2402         struct tid_ampdu_rx *tid_agg_rx;
2403         u16 sc;
2404         u16 mpdu_seq_num;
2405         u8 ret = 0;
2406         int tid;
2407
2408         sta = sta_info_get(local, hdr->addr2);
2409         if (!sta)
2410                 return ret;
2411
2412         /* filter the QoS data rx stream according to
2413          * STA/TID and check if this STA/TID is on aggregation */
2414         if (!ieee80211_is_data_qos(hdr->frame_control))
2415                 goto end_reorder;
2416
2417         tid = *ieee80211_get_qos_ctl(hdr) & IEEE80211_QOS_CTL_TID_MASK;
2418
2419         if (sta->ampdu_mlme.tid_state_rx[tid] != HT_AGG_STATE_OPERATIONAL)
2420                 goto end_reorder;
2421
2422         tid_agg_rx = sta->ampdu_mlme.tid_rx[tid];
2423
2424         /* qos null data frames are excluded */
2425         if (unlikely(hdr->frame_control & cpu_to_le16(IEEE80211_STYPE_NULLFUNC)))
2426                 goto end_reorder;
2427
2428         /* new un-ordered ampdu frame - process it */
2429
2430         /* reset session timer */
2431         if (tid_agg_rx->timeout)
2432                 mod_timer(&tid_agg_rx->session_timer,
2433                           TU_TO_EXP_TIME(tid_agg_rx->timeout));
2434
2435         /* if this mpdu is fragmented - terminate rx aggregation session */
2436         sc = le16_to_cpu(hdr->seq_ctrl);
2437         if (sc & IEEE80211_SCTL_FRAG) {
2438                 ieee80211_sta_stop_rx_ba_session(sta->sdata, sta->sta.addr,
2439                         tid, 0, WLAN_REASON_QSTA_REQUIRE_SETUP);
2440                 ret = 1;
2441                 goto end_reorder;
2442         }
2443
2444         /* according to mpdu sequence number deal with reordering buffer */
2445         mpdu_seq_num = (sc & IEEE80211_SCTL_SEQ) >> 4;
2446         ret = ieee80211_sta_manage_reorder_buf(hw, tid_agg_rx, skb,
2447                                                 mpdu_seq_num, 0);
2448  end_reorder:
2449         return ret;
2450 }
2451
2452 /*
2453  * This is the receive path handler. It is called by a low level driver when an
2454  * 802.11 MPDU is received from the hardware.
2455  */
2456 void ieee80211_rx(struct ieee80211_hw *hw, struct sk_buff *skb)
2457 {
2458         struct ieee80211_local *local = hw_to_local(hw);
2459         struct ieee80211_rate *rate = NULL;
2460         struct ieee80211_supported_band *sband;
2461         struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
2462
2463         WARN_ON_ONCE(softirq_count() == 0);
2464
2465         if (WARN_ON(status->band < 0 ||
2466                     status->band >= IEEE80211_NUM_BANDS))
2467                 goto drop;
2468
2469         sband = local->hw.wiphy->bands[status->band];
2470         if (WARN_ON(!sband))
2471                 goto drop;
2472
2473         /*
2474          * If we're suspending, it is possible although not too likely
2475          * that we'd be receiving frames after having already partially
2476          * quiesced the stack. We can't process such frames then since
2477          * that might, for example, cause stations to be added or other
2478          * driver callbacks be invoked.
2479          */
2480         if (unlikely(local->quiescing || local->suspended))
2481                 goto drop;
2482
2483         /*
2484          * The same happens when we're not even started,
2485          * but that's worth a warning.
2486          */
2487         if (WARN_ON(!local->started))
2488                 goto drop;
2489
2490         if (status->flag & RX_FLAG_HT) {
2491                 /* rate_idx is MCS index */
2492                 if (WARN_ON(status->rate_idx < 0 ||
2493                             status->rate_idx >= 76))
2494                         goto drop;
2495                 /* HT rates are not in the table - use the highest legacy rate
2496                  * for now since other parts of mac80211 may not yet be fully
2497                  * MCS aware. */
2498                 rate = &sband->bitrates[sband->n_bitrates - 1];
2499         } else {
2500                 if (WARN_ON(status->rate_idx < 0 ||
2501                             status->rate_idx >= sband->n_bitrates))
2502                         goto drop;
2503                 rate = &sband->bitrates[status->rate_idx];
2504         }
2505
2506         /*
2507          * key references and virtual interfaces are protected using RCU
2508          * and this requires that we are in a read-side RCU section during
2509          * receive processing
2510          */
2511         rcu_read_lock();
2512
2513         /*
2514          * Frames with failed FCS/PLCP checksum are not returned,
2515          * all other frames are returned without radiotap header
2516          * if it was previously present.
2517          * Also, frames with less than 16 bytes are dropped.
2518          */
2519         skb = ieee80211_rx_monitor(local, skb, rate);
2520         if (!skb) {
2521                 rcu_read_unlock();
2522                 return;
2523         }
2524
2525         /*
2526          * In theory, the block ack reordering should happen after duplicate
2527          * removal (ieee80211_rx_h_check(), which is an RX handler). As such,
2528          * the call to ieee80211_rx_reorder_ampdu() should really be moved to
2529          * happen as a new RX handler between ieee80211_rx_h_check and
2530          * ieee80211_rx_h_decrypt. This cleanup may eventually happen, but for
2531          * the time being, the call can be here since RX reorder buf processing
2532          * will implicitly skip duplicates. We could, in theory at least,
2533          * process frames that ieee80211_rx_h_passive_scan would drop (e.g.,
2534          * frames from other than operational channel), but that should not
2535          * happen in normal networks.
2536          */
2537         if (!ieee80211_rx_reorder_ampdu(local, skb))
2538                 __ieee80211_rx_handle_packet(hw, skb, rate);
2539
2540         rcu_read_unlock();
2541
2542         return;
2543  drop:
2544         kfree_skb(skb);
2545 }
2546 EXPORT_SYMBOL(ieee80211_rx);
2547
2548 /* This is a version of the rx handler that can be called from hard irq
2549  * context. Post the skb on the queue and schedule the tasklet */
2550 void ieee80211_rx_irqsafe(struct ieee80211_hw *hw, struct sk_buff *skb)
2551 {
2552         struct ieee80211_local *local = hw_to_local(hw);
2553
2554         BUILD_BUG_ON(sizeof(struct ieee80211_rx_status) > sizeof(skb->cb));
2555
2556         skb->pkt_type = IEEE80211_RX_MSG;
2557         skb_queue_tail(&local->skb_queue, skb);
2558         tasklet_schedule(&local->tasklet);
2559 }
2560 EXPORT_SYMBOL(ieee80211_rx_irqsafe);