rx.c 70 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
/*
 * Copyright 2002-2005, Instant802 Networks, Inc.
 * Copyright 2005-2006, Devicescape Software, Inc.
 * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
 * Copyright 2007	Johannes Berg <johannes@sipsolutions.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

12
#include <linux/jiffies.h>
13 14 15 16
#include <linux/kernel.h>
#include <linux/skbuff.h>
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
17
#include <linux/rcupdate.h>
18 19 20 21
#include <net/mac80211.h>
#include <net/ieee80211_radiotap.h>

#include "ieee80211_i.h"
22
#include "driver-ops.h"
Johannes Berg's avatar
Johannes Berg committed
23
#include "led.h"
24
#include "mesh.h"
25 26 27 28 29
#include "wep.h"
#include "wpa.h"
#include "tkip.h"
#include "wme.h"

Johannes Berg's avatar
Johannes Berg committed
30 31 32 33 34
static u8 ieee80211_sta_manage_reorder_buf(struct ieee80211_hw *hw,
					   struct tid_ampdu_rx *tid_agg_rx,
					   struct sk_buff *skb,
					   u16 mpdu_seq_num,
					   int bar_req);
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
/*
 * monitor mode reception
 *
 * This function cleans up the SKB, i.e. it removes all the stuff
 * only useful for monitoring.
 */
static struct sk_buff *remove_monitor_info(struct ieee80211_local *local,
					   struct sk_buff *skb,
					   int rtap_len)
{
	skb_pull(skb, rtap_len);

	if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS) {
		if (likely(skb->len > FCS_LEN))
			skb_trim(skb, skb->len - FCS_LEN);
		else {
			/* driver bug */
			WARN_ON(1);
			dev_kfree_skb(skb);
			skb = NULL;
		}
	}

	return skb;
}

61
static inline int should_drop_frame(struct sk_buff *skb,
62 63 64
				    int present_fcs_len,
				    int radiotap_len)
{
65
	struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
66
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
67 68 69 70 71

	if (status->flag & (RX_FLAG_FAILED_FCS_CRC | RX_FLAG_FAILED_PLCP_CRC))
		return 1;
	if (unlikely(skb->len < 16 + present_fcs_len + radiotap_len))
		return 1;
72 73 74
	if (ieee80211_is_ctl(hdr->frame_control) &&
	    !ieee80211_is_pspoll(hdr->frame_control) &&
	    !ieee80211_is_back_req(hdr->frame_control))
75 76 77 78
		return 1;
	return 0;
}

79 80 81 82 83 84 85 86 87 88 89
static int
ieee80211_rx_radiotap_len(struct ieee80211_local *local,
			  struct ieee80211_rx_status *status)
{
	int len;

	/* always present fields */
	len = sizeof(struct ieee80211_radiotap_header) + 9;

	if (status->flag & RX_FLAG_TSFT)
		len += 8;
Johannes Berg's avatar
Johannes Berg committed
90
	if (local->hw.flags & IEEE80211_HW_SIGNAL_DBM)
91 92 93 94 95 96 97 98 99 100 101 102 103 104
		len += 1;
	if (local->hw.flags & IEEE80211_HW_NOISE_DBM)
		len += 1;

	if (len & 1) /* padding for RX_FLAGS if necessary */
		len++;

	/* make sure radiotap starts at a naturally aligned address */
	if (len % 8)
		len = roundup(len, 8);

	return len;
}

105
/*
106 107 108 109 110 111 112 113 114 115
 * ieee80211_add_rx_radiotap_header - add radiotap header
 *
 * add a radiotap header containing all the fields which the hardware provided.
 */
static void
ieee80211_add_rx_radiotap_header(struct ieee80211_local *local,
				 struct sk_buff *skb,
				 struct ieee80211_rate *rate,
				 int rtap_len)
{
116
	struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145
	struct ieee80211_radiotap_header *rthdr;
	unsigned char *pos;

	rthdr = (struct ieee80211_radiotap_header *)skb_push(skb, rtap_len);
	memset(rthdr, 0, rtap_len);

	/* radiotap header, set always present flags */
	rthdr->it_present =
		cpu_to_le32((1 << IEEE80211_RADIOTAP_FLAGS) |
			    (1 << IEEE80211_RADIOTAP_CHANNEL) |
			    (1 << IEEE80211_RADIOTAP_ANTENNA) |
			    (1 << IEEE80211_RADIOTAP_RX_FLAGS));
	rthdr->it_len = cpu_to_le16(rtap_len);

	pos = (unsigned char *)(rthdr+1);

	/* the order of the following fields is important */

	/* IEEE80211_RADIOTAP_TSFT */
	if (status->flag & RX_FLAG_TSFT) {
		*(__le64 *)pos = cpu_to_le64(status->mactime);
		rthdr->it_present |=
			cpu_to_le32(1 << IEEE80211_RADIOTAP_TSFT);
		pos += 8;
	}

	/* IEEE80211_RADIOTAP_FLAGS */
	if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS)
		*pos |= IEEE80211_RADIOTAP_F_FCS;
Johannes Berg's avatar
Johannes Berg committed
146 147
	if (status->flag & (RX_FLAG_FAILED_FCS_CRC | RX_FLAG_FAILED_PLCP_CRC))
		*pos |= IEEE80211_RADIOTAP_F_BADFCS;
148 149
	if (status->flag & RX_FLAG_SHORTPRE)
		*pos |= IEEE80211_RADIOTAP_F_SHORTPRE;
150 151 152
	pos++;

	/* IEEE80211_RADIOTAP_RATE */
153 154 155 156 157 158 159 160 161
	if (status->flag & RX_FLAG_HT) {
		/*
		 * TODO: add following information into radiotap header once
		 * suitable fields are defined for it:
		 * - MCS index (status->rate_idx)
		 * - HT40 (status->flag & RX_FLAG_40MHZ)
		 * - short-GI (status->flag & RX_FLAG_SHORT_GI)
		 */
		*pos = 0;
162
	} else {
163
		rthdr->it_present |= cpu_to_le32(1 << IEEE80211_RADIOTAP_RATE);
164
		*pos = rate->bitrate / 5;
165
	}
166 167 168 169 170 171 172 173
	pos++;

	/* IEEE80211_RADIOTAP_CHANNEL */
	*(__le16 *)pos = cpu_to_le16(status->freq);
	pos += 2;
	if (status->band == IEEE80211_BAND_5GHZ)
		*(__le16 *)pos = cpu_to_le16(IEEE80211_CHAN_OFDM |
					     IEEE80211_CHAN_5GHZ);
174 175 176
	else if (rate->flags & IEEE80211_RATE_ERP_G)
		*(__le16 *)pos = cpu_to_le16(IEEE80211_CHAN_OFDM |
					     IEEE80211_CHAN_2GHZ);
177
	else
178
		*(__le16 *)pos = cpu_to_le16(IEEE80211_CHAN_CCK |
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
					     IEEE80211_CHAN_2GHZ);
	pos += 2;

	/* IEEE80211_RADIOTAP_DBM_ANTSIGNAL */
	if (local->hw.flags & IEEE80211_HW_SIGNAL_DBM) {
		*pos = status->signal;
		rthdr->it_present |=
			cpu_to_le32(1 << IEEE80211_RADIOTAP_DBM_ANTSIGNAL);
		pos++;
	}

	/* IEEE80211_RADIOTAP_DBM_ANTNOISE */
	if (local->hw.flags & IEEE80211_HW_NOISE_DBM) {
		*pos = status->noise;
		rthdr->it_present |=
			cpu_to_le32(1 << IEEE80211_RADIOTAP_DBM_ANTNOISE);
		pos++;
	}

	/* IEEE80211_RADIOTAP_LOCK_QUALITY is missing */

	/* IEEE80211_RADIOTAP_ANTENNA */
	*pos = status->antenna;
	pos++;

	/* IEEE80211_RADIOTAP_DB_ANTNOISE is not used */

	/* IEEE80211_RADIOTAP_RX_FLAGS */
	/* ensure 2 byte alignment for the 2 byte field as required */
	if ((pos - (unsigned char *)rthdr) & 1)
		pos++;
Johannes Berg's avatar
Johannes Berg committed
210 211
	if (status->flag & RX_FLAG_FAILED_PLCP_CRC)
		*(__le16 *)pos |= cpu_to_le16(IEEE80211_RADIOTAP_F_RX_BADPLCP);
212 213 214
	pos += 2;
}

215 216 217 218 219 220 221
/*
 * This function copies a received frame to all monitor interfaces and
 * returns a cleaned-up SKB that no longer includes the FCS nor the
 * radiotap header the driver might have added.
 */
static struct sk_buff *
ieee80211_rx_monitor(struct ieee80211_local *local, struct sk_buff *origskb,
222
		     struct ieee80211_rate *rate)
223
{
224
	struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(origskb);
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
	struct ieee80211_sub_if_data *sdata;
	int needed_headroom = 0;
	struct sk_buff *skb, *skb2;
	struct net_device *prev_dev = NULL;
	int present_fcs_len = 0;
	int rtap_len = 0;

	/*
	 * First, we may need to make a copy of the skb because
	 *  (1) we need to modify it for radiotap (if not present), and
	 *  (2) the other RX handlers will modify the skb we got.
	 *
	 * We don't need to, of course, if we aren't going to return
	 * the SKB because it has a bad FCS/PLCP checksum.
	 */
	if (status->flag & RX_FLAG_RADIOTAP)
		rtap_len = ieee80211_get_radiotap_len(origskb->data);
	else
243 244
		/* room for the radiotap header based on driver features */
		needed_headroom = ieee80211_rx_radiotap_len(local, status);
245 246 247 248 249

	if (local->hw.flags & IEEE80211_HW_RX_INCLUDES_FCS)
		present_fcs_len = FCS_LEN;

	if (!local->monitors) {
250
		if (should_drop_frame(origskb, present_fcs_len, rtap_len)) {
251 252 253 254 255 256 257
			dev_kfree_skb(origskb);
			return NULL;
		}

		return remove_monitor_info(local, origskb, rtap_len);
	}

258
	if (should_drop_frame(origskb, present_fcs_len, rtap_len)) {
259 260 261 262 263 264 265 266 267 268 269 270
		/* only need to expand headroom if necessary */
		skb = origskb;
		origskb = NULL;

		/*
		 * This shouldn't trigger often because most devices have an
		 * RX header they pull before we get here, and that should
		 * be big enough for our radiotap information. We should
		 * probably export the length to drivers so that we can have
		 * them allocate enough headroom to start with.
		 */
		if (skb_headroom(skb) < needed_headroom &&
271
		    pskb_expand_head(skb, needed_headroom, 0, GFP_ATOMIC)) {
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
			dev_kfree_skb(skb);
			return NULL;
		}
	} else {
		/*
		 * Need to make a copy and possibly remove radiotap header
		 * and FCS from the original.
		 */
		skb = skb_copy_expand(origskb, needed_headroom, 0, GFP_ATOMIC);

		origskb = remove_monitor_info(local, origskb, rtap_len);

		if (!skb)
			return origskb;
	}

	/* if necessary, prepend radiotap information */
289
	if (!(status->flag & RX_FLAG_RADIOTAP))
290
		ieee80211_add_rx_radiotap_header(local, skb, rate,
291
						 needed_headroom);
292

293
	skb_reset_mac_header(skb);
294 295 296 297 298 299 300 301
	skb->ip_summed = CHECKSUM_UNNECESSARY;
	skb->pkt_type = PACKET_OTHERHOST;
	skb->protocol = htons(ETH_P_802_2);

	list_for_each_entry_rcu(sdata, &local->interfaces, list) {
		if (!netif_running(sdata->dev))
			continue;

302
		if (sdata->vif.type != NL80211_IFTYPE_MONITOR)
303 304
			continue;

305 306 307
		if (sdata->u.mntr_flags & MONITOR_FLAG_COOK_FRAMES)
			continue;

308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330
		if (prev_dev) {
			skb2 = skb_clone(skb, GFP_ATOMIC);
			if (skb2) {
				skb2->dev = prev_dev;
				netif_rx(skb2);
			}
		}

		prev_dev = sdata->dev;
		sdata->dev->stats.rx_packets++;
		sdata->dev->stats.rx_bytes += skb->len;
	}

	if (prev_dev) {
		skb->dev = prev_dev;
		netif_rx(skb);
	} else
		dev_kfree_skb(skb);

	return origskb;
}


331
static void ieee80211_parse_qos(struct ieee80211_rx_data *rx)
332
{
333
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
334 335 336
	int tid;

	/* does the frame have a qos control field? */
337 338
	if (ieee80211_is_data_qos(hdr->frame_control)) {
		u8 *qc = ieee80211_get_qos_ctl(hdr);
339
		/* frame has qos control */
340 341
		tid = *qc & IEEE80211_QOS_CTL_TID_MASK;
		if (*qc & IEEE80211_QOS_CONTROL_A_MSDU_PRESENT)
342
			rx->flags |= IEEE80211_RX_AMSDU;
343
		else
344
			rx->flags &= ~IEEE80211_RX_AMSDU;
345
	} else {
346 347 348 349 350 351 352 353 354 355 356 357
		/*
		 * IEEE 802.11-2007, 7.1.3.4.1 ("Sequence Number field"):
		 *
		 *	Sequence numbers for management frames, QoS data
		 *	frames with a broadcast/multicast address in the
		 *	Address 1 field, and all non-QoS data frames sent
		 *	by QoS STAs are assigned using an additional single
		 *	modulo-4096 counter, [...]
		 *
		 * We also use that counter for non-QoS STAs.
		 */
		tid = NUM_RX_DATA_QUEUES - 1;
358
	}
359

360
	rx->queue = tid;
361 362 363
	/* Set skb->priority to 1d tag if highest order bit of TID is not set.
	 * For now, set skb->priority to 0 for other cases. */
	rx->skb->priority = (tid > 7) ? 0 : tid;
364
}
365

366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389
/**
 * DOC: Packet alignment
 *
 * Drivers always need to pass packets that are aligned to two-byte boundaries
 * to the stack.
 *
 * Additionally, should, if possible, align the payload data in a way that
 * guarantees that the contained IP header is aligned to a four-byte
 * boundary. In the case of regular frames, this simply means aligning the
 * payload to a four-byte boundary (because either the IP header is directly
 * contained, or IV/RFC1042 headers that have a length divisible by four are
 * in front of it).
 *
 * With A-MSDU frames, however, the payload data address must yield two modulo
 * four because there are 14-byte 802.3 headers within the A-MSDU frames that
 * push the IP header further back to a multiple of four again. Thankfully, the
 * specs were sane enough this time around to require padding each A-MSDU
 * subframe to a length that is a multiple of four.
 *
 * Padding like Atheros hardware adds which is inbetween the 802.11 header and
 * the payload is not supported, the driver is required to move the 802.11
 * header to be directly in front of the payload in that case.
 */
static void ieee80211_verify_alignment(struct ieee80211_rx_data *rx)
390
{
391
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
392 393
	int hdrlen;

394 395 396 397 398 399 400 401
#ifndef CONFIG_MAC80211_DEBUG_PACKET_ALIGNMENT
	return;
#endif

	if (WARN_ONCE((unsigned long)rx->skb->data & 1,
		      "unaligned packet at 0x%p\n", rx->skb->data))
		return;

402
	if (!ieee80211_is_data_present(hdr->frame_control))
403 404
		return;

405
	hdrlen = ieee80211_hdrlen(hdr->frame_control);
406
	if (rx->flags & IEEE80211_RX_AMSDU)
407
		hdrlen += ETH_HLEN;
408 409
	WARN_ONCE(((unsigned long)(rx->skb->data + hdrlen)) & 3,
		  "unaligned IP payload at 0x%p\n", rx->skb->data + hdrlen);
410 411
}

412

413 414
/* rx handlers */

415
static ieee80211_rx_result debug_noinline
416
ieee80211_rx_h_passive_scan(struct ieee80211_rx_data *rx)
417 418 419 420
{
	struct ieee80211_local *local = rx->local;
	struct sk_buff *skb = rx->skb;

421
	if (unlikely(test_bit(SCAN_HW_SCANNING, &local->scanning)))
422
		return ieee80211_scan_rx(rx->sdata, skb);
Zhu Yi's avatar
Zhu Yi committed
423

424 425
	if (unlikely(test_bit(SCAN_SW_SCANNING, &local->scanning) &&
		     (rx->flags & IEEE80211_RX_IN_SCAN))) {
Zhu Yi's avatar
Zhu Yi committed
426
		/* drop all the other packets during a software scan anyway */
427
		if (ieee80211_scan_rx(rx->sdata, skb) != RX_QUEUED)
Zhu Yi's avatar
Zhu Yi committed
428
			dev_kfree_skb(skb);
429
		return RX_QUEUED;
430 431
	}

432
	if (unlikely(rx->flags & IEEE80211_RX_IN_SCAN)) {
433 434
		/* scanning finished during invoking of handlers */
		I802_DEBUG_INC(local->rx_handlers_drop_passive_scan);
Johannes Berg's avatar
Johannes Berg committed
435
		return RX_DROP_UNUSABLE;
436 437
	}

438
	return RX_CONTINUE;
439 440
}

441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486

static int ieee80211_is_unicast_robust_mgmt_frame(struct sk_buff *skb)
{
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;

	if (skb->len < 24 || is_multicast_ether_addr(hdr->addr1))
		return 0;

	return ieee80211_is_robust_mgmt_frame(hdr);
}


static int ieee80211_is_multicast_robust_mgmt_frame(struct sk_buff *skb)
{
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;

	if (skb->len < 24 || !is_multicast_ether_addr(hdr->addr1))
		return 0;

	return ieee80211_is_robust_mgmt_frame(hdr);
}


/* Get the BIP key index from MMIE; return -1 if this is not a BIP frame */
static int ieee80211_get_mmie_keyidx(struct sk_buff *skb)
{
	struct ieee80211_mgmt *hdr = (struct ieee80211_mgmt *) skb->data;
	struct ieee80211_mmie *mmie;

	if (skb->len < 24 + sizeof(*mmie) ||
	    !is_multicast_ether_addr(hdr->da))
		return -1;

	if (!ieee80211_is_robust_mgmt_frame((struct ieee80211_hdr *) hdr))
		return -1; /* not a robust management frame */

	mmie = (struct ieee80211_mmie *)
		(skb->data + skb->len - sizeof(*mmie));
	if (mmie->element_id != WLAN_EID_MMIE ||
	    mmie->length != sizeof(*mmie) - 2)
		return -1;

	return le16_to_cpu(mmie->key_id);
}


487
static ieee80211_rx_result
488
ieee80211_rx_mesh_check(struct ieee80211_rx_data *rx)
489
{
490 491
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
	unsigned int hdrlen = ieee80211_hdrlen(hdr->frame_control);
492

493 494
	if (ieee80211_is_data(hdr->frame_control)) {
		if (!ieee80211_has_a4(hdr->frame_control))
495 496 497 498 499 500 501 502 503
			return RX_DROP_MONITOR;
		if (memcmp(hdr->addr4, rx->dev->dev_addr, ETH_ALEN) == 0)
			return RX_DROP_MONITOR;
	}

	/* If there is not an established peer link and this is not a peer link
	 * establisment frame, beacon or probe, drop the frame.
	 */

504
	if (!rx->sta || sta_plink_state(rx->sta) != PLINK_ESTAB) {
505
		struct ieee80211_mgmt *mgmt;
506

507
		if (!ieee80211_is_mgmt(hdr->frame_control))
508 509
			return RX_DROP_MONITOR;

510
		if (ieee80211_is_action(hdr->frame_control)) {
511 512 513 514 515 516
			mgmt = (struct ieee80211_mgmt *)hdr;
			if (mgmt->u.action.category != PLINK_CATEGORY)
				return RX_DROP_MONITOR;
			return RX_CONTINUE;
		}

517 518 519 520 521 522 523 524 525 526 527 528 529
		if (ieee80211_is_probe_req(hdr->frame_control) ||
		    ieee80211_is_probe_resp(hdr->frame_control) ||
		    ieee80211_is_beacon(hdr->frame_control))
			return RX_CONTINUE;

		return RX_DROP_MONITOR;

	}

#define msh_h_get(h, l) ((struct ieee80211s_hdr *) ((u8 *)h + l))

	if (ieee80211_is_data(hdr->frame_control) &&
	    is_multicast_ether_addr(hdr->addr1) &&
530
	    mesh_rmc_check(hdr->addr4, msh_h_get(hdr, hdrlen), rx->sdata))
531
		return RX_DROP_MONITOR;
Johannes Berg's avatar
Johannes Berg committed
532
#undef msh_h_get
533

Johannes Berg's avatar
Johannes Berg committed
534 535
	return RX_CONTINUE;
}
536 537


538
static ieee80211_rx_result debug_noinline
539
ieee80211_rx_h_check(struct ieee80211_rx_data *rx)
540
{
541
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
542 543 544

	/* Drop duplicate 802.11 retransmissions (IEEE 802.11 Chap. 9.2.9) */
	if (rx->sta && !is_multicast_ether_addr(hdr->addr1)) {
545
		if (unlikely(ieee80211_has_retry(hdr->frame_control) &&
546
			     rx->sta->last_seq_ctrl[rx->queue] ==
547
			     hdr->seq_ctrl)) {
548
			if (rx->flags & IEEE80211_RX_RA_MATCH) {
549 550 551
				rx->local->dot11FrameDuplicateCount++;
				rx->sta->num_duplicates++;
			}
Johannes Berg's avatar
Johannes Berg committed
552
			return RX_DROP_MONITOR;
553
		} else
554
			rx->sta->last_seq_ctrl[rx->queue] = hdr->seq_ctrl;
555 556 557 558
	}

	if (unlikely(rx->skb->len < 16)) {
		I802_DEBUG_INC(rx->local->rx_handlers_drop_short);
Johannes Berg's avatar
Johannes Berg committed
559
		return RX_DROP_MONITOR;
560 561 562 563 564
	}

	/* Drop disallowed frame classes based on STA auth/assoc state;
	 * IEEE 802.11, Chap 5.5.
	 *
565 566
	 * mac80211 filters only based on association state, i.e. it drops
	 * Class 3 frames from not associated stations. hostapd sends
567 568 569
	 * deauth/disassoc frames when needed. In addition, hostapd is
	 * responsible for filtering on both auth and assoc states.
	 */
570

Johannes Berg's avatar
Johannes Berg committed
571
	if (ieee80211_vif_is_mesh(&rx->sdata->vif))
572 573
		return ieee80211_rx_mesh_check(rx);

574 575
	if (unlikely((ieee80211_is_data(hdr->frame_control) ||
		      ieee80211_is_pspoll(hdr->frame_control)) &&
576
		     rx->sdata->vif.type != NL80211_IFTYPE_ADHOC &&
577
		     (!rx->sta || !test_sta_flags(rx->sta, WLAN_STA_ASSOC)))) {
578 579 580 581
		if ((!ieee80211_has_fromds(hdr->frame_control) &&
		     !ieee80211_has_tods(hdr->frame_control) &&
		     ieee80211_is_data(hdr->frame_control)) ||
		    !(rx->flags & IEEE80211_RX_RA_MATCH)) {
582 583
			/* Drop IBSS frames and frames for other hosts
			 * silently. */
Johannes Berg's avatar
Johannes Berg committed
584
			return RX_DROP_MONITOR;
585 586
		}

Johannes Berg's avatar
Johannes Berg committed
587
		return RX_DROP_MONITOR;
588 589
	}

590
	return RX_CONTINUE;
591 592 593
}


594
static ieee80211_rx_result debug_noinline
595
ieee80211_rx_h_decrypt(struct ieee80211_rx_data *rx)
596
{
597
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
598 599
	int keyidx;
	int hdrlen;
Johannes Berg's avatar
Johannes Berg committed
600
	ieee80211_rx_result result = RX_DROP_UNUSABLE;
601
	struct ieee80211_key *stakey = NULL;
602
	int mmie_keyidx = -1;
603

604 605 606
	/*
	 * Key selection 101
	 *
607
	 * There are four types of keys:
608
	 *  - GTK (group keys)
609
	 *  - IGTK (group keys for management frames)
610 611 612 613 614
	 *  - PTK (pairwise keys)
	 *  - STK (station-to-station pairwise keys)
	 *
	 * When selecting a key, we have to distinguish between multicast
	 * (including broadcast) and unicast frames, the latter can only
615 616 617 618
	 * use PTKs and STKs while the former always use GTKs and IGTKs.
	 * Unless, of course, actual WEP keys ("pre-RSNA") are used, then
	 * unicast frames can also use key indices like GTKs. Hence, if we
	 * don't have a PTK/STK we check the key index for a WEP key.
619
	 *
620 621 622 623
	 * Note that in a regular BSS, multicast frames are sent by the
	 * AP only, associated stations unicast the frame to the AP first
	 * which then multicasts it on their behalf.
	 *
624 625
	 * There is also a slight problem in IBSS mode: GTKs are negotiated
	 * with each station, that is something we don't currently handle.
626 627 628
	 * The spec seems to expect that one negotiates the same key with
	 * every station but there's no such requirement; VLANs could be
	 * possible.
629 630 631
	 */

	/*
632
	 * No point in finding a key and decrypting if the frame is neither
633 634
	 * addressed to us nor a multicast frame.
	 */
635
	if (!(rx->flags & IEEE80211_RX_RA_MATCH))
636
		return RX_CONTINUE;
637

638 639 640
	if (rx->sta)
		stakey = rcu_dereference(rx->sta->key);

641 642 643
	if (!ieee80211_has_protected(hdr->frame_control))
		mmie_keyidx = ieee80211_get_mmie_keyidx(rx->skb);

644 645
	if (!is_multicast_ether_addr(hdr->addr1) && stakey) {
		rx->key = stakey;
646 647 648
		/* Skip decryption if the frame is not protected. */
		if (!ieee80211_has_protected(hdr->frame_control))
			return RX_CONTINUE;
649 650 651 652 653 654 655 656 657 658
	} else if (mmie_keyidx >= 0) {
		/* Broadcast/multicast robust management frame / BIP */
		if ((rx->status->flag & RX_FLAG_DECRYPTED) &&
		    (rx->status->flag & RX_FLAG_IV_STRIPPED))
			return RX_CONTINUE;

		if (mmie_keyidx < NUM_DEFAULT_KEYS ||
		    mmie_keyidx >= NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS)
			return RX_DROP_MONITOR; /* unexpected BIP keyidx */
		rx->key = rcu_dereference(rx->sdata->keys[mmie_keyidx]);
659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
	} else if (!ieee80211_has_protected(hdr->frame_control)) {
		/*
		 * The frame was not protected, so skip decryption. However, we
		 * need to set rx->key if there is a key that could have been
		 * used so that the frame may be dropped if encryption would
		 * have been expected.
		 */
		struct ieee80211_key *key = NULL;
		if (ieee80211_is_mgmt(hdr->frame_control) &&
		    is_multicast_ether_addr(hdr->addr1) &&
		    (key = rcu_dereference(rx->sdata->default_mgmt_key)))
			rx->key = key;
		else if ((key = rcu_dereference(rx->sdata->default_key)))
			rx->key = key;
		return RX_CONTINUE;
674
	} else {
675 676 677 678 679 680 681 682 683
		/*
		 * The device doesn't give us the IV so we won't be
		 * able to look up the key. That's ok though, we
		 * don't need to decrypt the frame, we just won't
		 * be able to keep statistics accurate.
		 * Except for key threshold notifications, should
		 * we somehow allow the driver to tell us which key
		 * the hardware used if this flag is set?
		 */
684 685
		if ((rx->status->flag & RX_FLAG_DECRYPTED) &&
		    (rx->status->flag & RX_FLAG_IV_STRIPPED))
686
			return RX_CONTINUE;
687

688
		hdrlen = ieee80211_hdrlen(hdr->frame_control);
689 690

		if (rx->skb->len < 8 + hdrlen)
Johannes Berg's avatar
Johannes Berg committed
691
			return RX_DROP_UNUSABLE; /* TODO: count this? */
692 693 694 695 696 697 698

		/*
		 * no need to call ieee80211_wep_get_keyidx,
		 * it verifies a bunch of things we've done already
		 */
		keyidx = rx->skb->data[hdrlen + 3] >> 6;

699
		rx->key = rcu_dereference(rx->sdata->keys[keyidx]);
700 701 702 703 704 705

		/*
		 * RSNA-protected unicast frames should always be sent with
		 * pairwise or station-to-station keys, but for WEP we allow
		 * using a key index as well.
		 */
706
		if (rx->key && rx->key->conf.alg != ALG_WEP &&
707 708
		    !is_multicast_ether_addr(hdr->addr1))
			rx->key = NULL;
709 710
	}

711
	if (rx->key) {
712
		rx->key->tx_rx_count++;
713
		/* TODO: add threshold stuff again */
714
	} else {
Johannes Berg's avatar
Johannes Berg committed
715
		return RX_DROP_MONITOR;
716 717
	}

718 719
	/* Check for weak IVs if possible */
	if (rx->sta && rx->key->conf.alg == ALG_WEP &&
720
	    ieee80211_is_data(hdr->frame_control) &&
721 722
	    (!(rx->status->flag & RX_FLAG_IV_STRIPPED) ||
	     !(rx->status->flag & RX_FLAG_DECRYPTED)) &&
723 724 725
	    ieee80211_wep_is_weak_iv(rx->skb, rx->key))
		rx->sta->wep_weak_iv_count++;

726 727
	switch (rx->key->conf.alg) {
	case ALG_WEP:
728 729
		result = ieee80211_crypto_wep_decrypt(rx);
		break;
730
	case ALG_TKIP:
731 732
		result = ieee80211_crypto_tkip_decrypt(rx);
		break;
733
	case ALG_CCMP:
734 735
		result = ieee80211_crypto_ccmp_decrypt(rx);
		break;
736 737 738
	case ALG_AES_CMAC:
		result = ieee80211_crypto_aes_cmac_decrypt(rx);
		break;
739 740
	}

741
	/* either the frame has been decrypted or will be dropped */
742
	rx->status->flag |= RX_FLAG_DECRYPTED;
743 744

	return result;
745 746
}

747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779
static ieee80211_rx_result debug_noinline
ieee80211_rx_h_check_more_data(struct ieee80211_rx_data *rx)
{
	struct ieee80211_local *local;
	struct ieee80211_hdr *hdr;
	struct sk_buff *skb;

	local = rx->local;
	skb = rx->skb;
	hdr = (struct ieee80211_hdr *) skb->data;

	if (!local->pspolling)
		return RX_CONTINUE;

	if (!ieee80211_has_fromds(hdr->frame_control))
		/* this is not from AP */
		return RX_CONTINUE;

	if (!ieee80211_is_data(hdr->frame_control))
		return RX_CONTINUE;

	if (!ieee80211_has_moredata(hdr->frame_control)) {
		/* AP has no more frames buffered for us */
		local->pspolling = false;
		return RX_CONTINUE;
	}

	/* more data bit is set, let's request a new frame from the AP */
	ieee80211_send_pspoll(local, rx->sdata);

	return RX_CONTINUE;
}

780
static void ap_sta_ps_start(struct sta_info *sta)
781
{
782
	struct ieee80211_sub_if_data *sdata = sta->sdata;
783
	struct ieee80211_local *local = sdata->local;
784

785
	atomic_inc(&sdata->bss->num_sta_ps);
786
	set_sta_flags(sta, WLAN_STA_PS);
787
	drv_sta_notify(local, &sdata->vif, STA_NOTIFY_SLEEP, &sta->sta);
788
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
789 790
	printk(KERN_DEBUG "%s: STA %pM aid %d enters power save mode\n",
	       sdata->dev->name, sta->sta.addr, sta->sta.aid);
791 792 793
#endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
}

794
static int ap_sta_ps_end(struct sta_info *sta)
795
{
796 797
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	struct ieee80211_local *local = sdata->local;
798
	int sent, buffered;
799

800
	atomic_dec(&sdata->bss->num_sta_ps);
801

802
	clear_sta_flags(sta, WLAN_STA_PS);
803
	drv_sta_notify(local, &sdata->vif, STA_NOTIFY_AWAKE, &sta->sta);
804 805 806 807

	if (!skb_queue_empty(&sta->ps_tx_buf))
		sta_info_clear_tim_bit(sta);

808
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
809 810
	printk(KERN_DEBUG "%s: STA %pM aid %d exits power save mode\n",
	       sdata->dev->name, sta->sta.addr, sta->sta.aid);
811
#endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
812

813
	/* Send all buffered frames to the station */
814 815 816 817 818
	sent = ieee80211_add_pending_skbs(local, &sta->tx_filtered);
	buffered = ieee80211_add_pending_skbs(local, &sta->ps_tx_buf);
	sent += buffered;
	local->total_ps_buffered -= buffered;

819
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
820 821 822
	printk(KERN_DEBUG "%s: STA %pM aid %d sending %d filtered/%d PS frames "
	       "since STA not sleeping anymore\n", sdata->dev->name,
	       sta->sta.addr, sta->sta.aid, sent - buffered, buffered);
823 824 825 826 827
#endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */

	return sent;
}

828
static ieee80211_rx_result debug_noinline
829
ieee80211_rx_h_sta_process(struct ieee80211_rx_data *rx)
830 831
{
	struct sta_info *sta = rx->sta;
832
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)rx->skb->data;
833 834

	if (!sta)
835
		return RX_CONTINUE;
836

837 838 839 840 841
	/*
	 * Update last_rx only for IBSS packets which are for the current
	 * BSSID to avoid keeping the current IBSS network alive in cases
	 * where other STAs start using different BSSID.
	 */
842
	if (rx->sdata->vif.type == NL80211_IFTYPE_ADHOC) {
843
		u8 *bssid = ieee80211_get_bssid(hdr, rx->skb->len,
844
						NL80211_IFTYPE_ADHOC);
845
		if (compare_ether_addr(bssid, rx->sdata->u.ibss.bssid) == 0)
846
			sta->last_rx = jiffies;
847 848
	} else if (!is_multicast_ether_addr(hdr->addr1)) {
		/*
849 850
		 * Mesh beacons will update last_rx when if they are found to
		 * match the current local configuration when processed.
851
		 */
852
		sta->last_rx = jiffies;
853 854
	}

855
	if (!(rx->flags & IEEE80211_RX_RA_MATCH))
856
		return RX_CONTINUE;
857

858 859 860
	if (rx->sdata->vif.type == NL80211_IFTYPE_STATION)
		ieee80211_sta_rx_notify(rx->sdata, hdr);

861 862
	sta->rx_fragments++;
	sta->rx_bytes += rx->skb->len;
863
	sta->last_signal = rx->status->signal;
864
	sta->last_qual = rx->status->qual;
865
	sta->last_noise = rx->status->noise;
866

867 868 869 870
	/*
	 * Change STA power saving mode only at the end of a frame
	 * exchange sequence.
	 */
871
	if (!ieee80211_has_morefrags(hdr->frame_control) &&
872 873
	    (rx->sdata->vif.type == NL80211_IFTYPE_AP ||
	     rx->sdata->vif.type == NL80211_IFTYPE_AP_VLAN)) {
874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889
		if (test_sta_flags(sta, WLAN_STA_PS)) {
			/*
			 * Ignore doze->wake transitions that are
			 * indicated by non-data frames, the standard
			 * is unclear here, but for example going to
			 * PS mode and then scanning would cause a
			 * doze->wake transition for the probe request,
			 * and that is clearly undesirable.
			 */
			if (ieee80211_is_data(hdr->frame_control) &&
			    !ieee80211_has_pm(hdr->frame_control))
				rx->sent_ps_buffered += ap_sta_ps_end(sta);
		} else {
			if (ieee80211_has_pm(hdr->frame_control))
				ap_sta_ps_start(sta);
		}
890 891 892 893
	}

	/* Drop data::nullfunc frames silently, since they are used only to
	 * control station power saving mode. */
894
	if (ieee80211_is_nullfunc(hdr->frame_control)) {
895 896 897 898 899
		I802_DEBUG_INC(rx->local->rx_handlers_drop_nullfunc);
		/* Update counter and free packet here to avoid counting this
		 * as a dropped packed. */
		sta->rx_packets++;
		dev_kfree_skb(rx->skb);
900
		return RX_QUEUED;
901 902
	}

903
	return RX_CONTINUE;
904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919
} /* ieee80211_rx_h_sta_process */

static inline struct ieee80211_fragment_entry *
ieee80211_reassemble_add(struct ieee80211_sub_if_data *sdata,
			 unsigned int frag, unsigned int seq, int rx_queue,
			 struct sk_buff **skb)
{
	struct ieee80211_fragment_entry *entry;
	int idx;

	idx = sdata->fragment_next;
	entry = &sdata->fragments[sdata->fragment_next++];
	if (sdata->fragment_next >= IEEE80211_FRAGMENT_MAX)
		sdata->fragment_next = 0;

	if (!skb_queue_empty(&entry->skb_list)) {
920
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
921 922 923 924
		struct ieee80211_hdr *hdr =
			(struct ieee80211_hdr *) entry->skb_list.next->data;
		printk(KERN_DEBUG "%s: RX reassembly removed oldest "
		       "fragment entry (idx=%d age=%lu seq=%d last_frag=%d "
925
		       "addr1=%pM addr2=%pM\n",
926 927
		       sdata->dev->name, idx,
		       jiffies - entry->first_frag_time, entry->seq,
928
		       entry->last_frag, hdr->addr1, hdr->addr2);
929
#endif
930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946
		__skb_queue_purge(&entry->skb_list);
	}

	__skb_queue_tail(&entry->skb_list, *skb); /* no need for locking */
	*skb = NULL;
	entry->first_frag_time = jiffies;
	entry->seq = seq;
	entry->rx_queue = rx_queue;
	entry->last_frag = frag;
	entry->ccmp = 0;
	entry->extra_len = 0;

	return entry;
}

static inline struct ieee80211_fragment_entry *
ieee80211_reassemble_find(struct ieee80211_sub_if_data *sdata,
947
			  unsigned int frag, unsigned int seq,
948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966
			  int rx_queue, struct ieee80211_hdr *hdr)
{
	struct ieee80211_fragment_entry *entry;
	int i, idx;

	idx = sdata->fragment_next;
	for (i = 0; i < IEEE80211_FRAGMENT_MAX; i++) {
		struct ieee80211_hdr *f_hdr;

		idx--;
		if (idx < 0)
			idx = IEEE80211_FRAGMENT_MAX - 1;

		entry = &sdata->fragments[idx];
		if (skb_queue_empty(&entry->skb_list) || entry->seq != seq ||
		    entry->rx_queue != rx_queue ||
		    entry->last_frag + 1 != frag)
			continue;

967
		f_hdr = (struct ieee80211_hdr *)entry->skb_list.next->data;
968

969 970 971 972 973
		/*
		 * Check ftype and addresses are equal, else check next fragment
		 */
		if (((hdr->frame_control ^ f_hdr->frame_control) &
		     cpu_to_le16(IEEE80211_FCTL_FTYPE)) ||
974 975 976 977
		    compare_ether_addr(hdr->addr1, f_hdr->addr1) != 0 ||
		    compare_ether_addr(hdr->addr2, f_hdr->addr2) != 0)
			continue;

978
		if (time_after(jiffies, entry->first_frag_time + 2 * HZ)) {
979 980 981 982 983 984 985 986 987
			__skb_queue_purge(&entry->skb_list);
			continue;
		}
		return entry;
	}

	return NULL;
}

988
static ieee80211_rx_result debug_noinline
989
ieee80211_rx_h_defragment(struct ieee80211_rx_data *rx)
990 991 992
{
	struct ieee80211_hdr *hdr;
	u16 sc;
993
	__le16 fc;
994 995 996 997
	unsigned int frag, seq;
	struct ieee80211_fragment_entry *entry;
	struct sk_buff *skb;

998
	hdr = (struct ieee80211_hdr *)rx->skb->data;
999
	fc = hdr->frame_control;
1000 1001 1002
	sc = le16_to_cpu(hdr->seq_ctrl);
	frag = sc & IEEE80211_SCTL_FRAG;

1003
	if (likely((!ieee80211_has_morefrags(fc) && frag == 0) ||
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015
		   (rx->skb)->len < 24 ||
		   is_multicast_ether_addr(hdr->addr1))) {
		/* not fragmented */
		goto out;
	}
	I802_DEBUG_INC(rx->local->rx_handlers_fragments);

	seq = (sc & IEEE80211_SCTL_SEQ) >> 4;

	if (frag == 0) {
		/* This is the first fragment of a new frame. */
		entry = ieee80211_reassemble_add(rx->sdata, frag, seq,
1016
						 rx->queue, &(rx->skb));
1017
		if (rx->key && rx->key->conf.alg == ALG_CCMP &&
1018
		    ieee80211_has_protected(fc)) {
1019 1020 1021 1022
			/* Store CCMP PN so that we can verify that the next
			 * fragment has a sequential PN value. */
			entry->ccmp = 1;
			memcpy(entry->last_pn,
1023
			       rx->key->u.ccmp.rx_pn[rx->queue],
1024 1025
			       CCMP_PN_LEN);
		}
1026
		return RX_QUEUED;
1027 1028 1029 1030 1031
	}

	/* This is a fragment for a frame that should already be pending in
	 * fragment cache. Add this fragment to the end of the pending entry.
	 */
1032
	entry = ieee80211_reassemble_find(rx->sdata, frag, seq, rx->queue, hdr);
1033 1034
	if (!entry) {
		I802_DEBUG_INC(rx->local->rx_handlers_drop_defrag);
Johannes Berg's avatar
Johannes Berg committed
1035
		return RX_DROP_MONITOR;
1036 1037 1038 1039 1040 1041 1042
	}

	/* Verify that MPDUs within one MSDU have sequential PN values.
	 * (IEEE 802.11i, 8.3.3.4.5) */
	if (entry->ccmp) {
		int i;
		u8 pn[CCMP_PN_LEN], *rpn;
1043
		if (!rx->key || rx->key->conf.alg != ALG_CCMP)
Johannes Berg's avatar
Johannes Berg committed
1044
			return RX_DROP_UNUSABLE;
1045 1046 1047 1048 1049 1050
		memcpy(pn, entry->last_pn, CCMP_PN_LEN);
		for (i = CCMP_PN_LEN - 1; i >= 0; i--) {
			pn[i]++;
			if (pn[i])
				break;
		}
1051
		rpn = rx->key->u.ccmp.rx_pn[rx->queue];
1052
		if (memcmp(pn, rpn, CCMP_PN_LEN))
Johannes Berg's avatar
Johannes Berg committed
1053
			return RX_DROP_UNUSABLE;
1054 1055 1056
		memcpy(entry->last_pn, pn, CCMP_PN_LEN);
	}

1057
	skb_pull(rx->skb, ieee80211_hdrlen(fc));
1058 1059 1060
	__skb_queue_tail(&entry->skb_list, rx->skb);
	entry->last_frag = frag;
	entry->extra_len += rx->skb->len;
1061
	if (ieee80211_has_morefrags(fc)) {
1062
		rx->skb = NULL;
1063
		return RX_QUEUED;
1064 1065 1066 1067 1068 1069 1070 1071 1072
	}

	rx->skb = __skb_dequeue(&entry->skb_list);
	if (skb_tailroom(rx->skb) < entry->extra_len) {
		I802_DEBUG_INC(rx->local->rx_expand_skb_head2);
		if (unlikely(pskb_expand_head(rx->skb, 0, entry->extra_len,
					      GFP_ATOMIC))) {
			I802_DEBUG_INC(rx->local->rx_handlers_drop_defrag);
			__skb_queue_purge(&entry->skb_list);
Johannes Berg's avatar
Johannes Berg committed
1073
			return RX_DROP_UNUSABLE;
1074 1075 1076 1077 1078 1079 1080 1081
		}
	}
	while ((skb = __skb_dequeue(&entry->skb_list))) {
		memcpy(skb_put(rx->skb, skb->len), skb->data, skb->len);
		dev_kfree_skb(skb);
	}

	/* Complete frame has been reassembled - process it now */
1082
	rx->flags |= IEEE80211_RX_FRAGMENTED;
1083 1084 1085 1086 1087 1088 1089 1090

 out:
	if (rx->sta)
		rx->sta->rx_packets++;
	if (is_multicast_ether_addr(hdr->addr1))
		rx->local->dot11MulticastReceivedFrameCount++;
	else
		ieee80211_led_rx(rx->local);
1091
	return RX_CONTINUE;
1092 1093
}

1094
static ieee80211_rx_result debug_noinline
1095
ieee80211_rx_h_ps_poll(struct ieee80211_rx_data *rx)
1096
{
1097
	struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(rx->dev);
1098 1099
	struct sk_buff *skb;
	int no_pending_pkts;
1100
	__le16 fc = ((struct ieee80211_hdr *)rx->skb->data)->frame_control;
1101

1102
	if (likely(!rx->sta || !ieee80211_is_pspoll(fc) ||
1103
		   !(rx->flags & IEEE80211_RX_RA_MATCH)))
1104
		return RX_CONTINUE;
1105

1106 1107
	if ((sdata->vif.type != NL80211_IFTYPE_AP) &&
	    (sdata->vif.type != NL80211_IFTYPE_AP_VLAN))
Johannes Berg's avatar
Johannes Berg committed
1108
		return RX_DROP_UNUSABLE;