tx.c 55.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/*
 * 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.
 *
 *
 * Transmit and frame generation functions.
 */

#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/skbuff.h>
#include <linux/etherdevice.h>
#include <linux/bitmap.h>
20
#include <linux/rcupdate.h>
21
#include <net/net_namespace.h>
22 23 24 25 26 27
#include <net/ieee80211_radiotap.h>
#include <net/cfg80211.h>
#include <net/mac80211.h>
#include <asm/unaligned.h>

#include "ieee80211_i.h"
J
Johannes Berg 已提交
28
#include "led.h"
29
#include "mesh.h"
30 31 32
#include "wep.h"
#include "wpa.h"
#include "wme.h"
J
Johannes Berg 已提交
33
#include "rate.h"
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

#define IEEE80211_TX_OK		0
#define IEEE80211_TX_AGAIN	1
#define IEEE80211_TX_FRAG_AGAIN	2

/* misc utils */

static inline void ieee80211_include_sequence(struct ieee80211_sub_if_data *sdata,
					      struct ieee80211_hdr *hdr)
{
	/* Set the sequence number for this frame. */
	hdr->seq_ctrl = cpu_to_le16(sdata->sequence);

	/* Increase the sequence number. */
	sdata->sequence = (sdata->sequence + 0x10) & IEEE80211_SCTL_SEQ;
}

#ifdef CONFIG_MAC80211_LOWTX_FRAME_DUMP
static void ieee80211_dump_frame(const char *ifname, const char *title,
				 const struct sk_buff *skb)
{
	const struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
	u16 fc;
	int hdrlen;
58
	DECLARE_MAC_BUF(mac);
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73

	printk(KERN_DEBUG "%s: %s (len=%d)", ifname, title, skb->len);
	if (skb->len < 4) {
		printk("\n");
		return;
	}

	fc = le16_to_cpu(hdr->frame_control);
	hdrlen = ieee80211_get_hdrlen(fc);
	if (hdrlen > skb->len)
		hdrlen = skb->len;
	if (hdrlen >= 4)
		printk(" FC=0x%04x DUR=0x%04x",
		       fc, le16_to_cpu(hdr->duration_id));
	if (hdrlen >= 10)
74
		printk(" A1=%s", print_mac(mac, hdr->addr1));
75
	if (hdrlen >= 16)
76
		printk(" A2=%s", print_mac(mac, hdr->addr2));
77
	if (hdrlen >= 24)
78
		printk(" A3=%s", print_mac(mac, hdr->addr3));
79
	if (hdrlen >= 30)
80
		printk(" A4=%s", print_mac(mac, hdr->addr4));
81 82 83 84 85 86 87 88 89
	printk("\n");
}
#else /* CONFIG_MAC80211_LOWTX_FRAME_DUMP */
static inline void ieee80211_dump_frame(const char *ifname, const char *title,
					struct sk_buff *skb)
{
}
#endif /* CONFIG_MAC80211_LOWTX_FRAME_DUMP */

90
static u16 ieee80211_duration(struct ieee80211_tx_data *tx, int group_addr,
91 92 93
			      int next_frag_len)
{
	int rate, mrate, erp, dur, i;
94
	struct ieee80211_rate *txrate;
95
	struct ieee80211_local *local = tx->local;
96
	struct ieee80211_supported_band *sband;
97

98 99
	sband = local->hw.wiphy->bands[tx->channel->band];
	txrate = &sband->bitrates[tx->rate_idx];
100 101 102 103

	erp = 0;
	if (tx->sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
		erp = txrate->flags & IEEE80211_RATE_ERP_G;
104 105 106 107 108 109 110 111 112 113 114 115 116 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 146 147 148 149 150 151 152 153 154 155 156 157 158

	/*
	 * data and mgmt (except PS Poll):
	 * - during CFP: 32768
	 * - during contention period:
	 *   if addr1 is group address: 0
	 *   if more fragments = 0 and addr1 is individual address: time to
	 *      transmit one ACK plus SIFS
	 *   if more fragments = 1 and addr1 is individual address: time to
	 *      transmit next fragment plus 2 x ACK plus 3 x SIFS
	 *
	 * IEEE 802.11, 9.6:
	 * - control response frame (CTS or ACK) shall be transmitted using the
	 *   same rate as the immediately previous frame in the frame exchange
	 *   sequence, if this rate belongs to the PHY mandatory rates, or else
	 *   at the highest possible rate belonging to the PHY rates in the
	 *   BSSBasicRateSet
	 */

	if ((tx->fc & IEEE80211_FCTL_FTYPE) == IEEE80211_FTYPE_CTL) {
		/* TODO: These control frames are not currently sent by
		 * 80211.o, but should they be implemented, this function
		 * needs to be updated to support duration field calculation.
		 *
		 * RTS: time needed to transmit pending data/mgmt frame plus
		 *    one CTS frame plus one ACK frame plus 3 x SIFS
		 * CTS: duration of immediately previous RTS minus time
		 *    required to transmit CTS and its SIFS
		 * ACK: 0 if immediately previous directed data/mgmt had
		 *    more=0, with more=1 duration in ACK frame is duration
		 *    from previous frame minus time needed to transmit ACK
		 *    and its SIFS
		 * PS Poll: BIT(15) | BIT(14) | aid
		 */
		return 0;
	}

	/* data/mgmt */
	if (0 /* FIX: data/mgmt during CFP */)
		return 32768;

	if (group_addr) /* Group address as the destination - no ACK */
		return 0;

	/* Individual destination address:
	 * IEEE 802.11, Ch. 9.6 (after IEEE 802.11g changes)
	 * CTS and ACK frames shall be transmitted using the highest rate in
	 * basic rate set that is less than or equal to the rate of the
	 * immediately previous frame and that is using the same modulation
	 * (CCK or OFDM). If no basic rate set matches with these requirements,
	 * the highest mandatory rate of the PHY that is less than or equal to
	 * the rate of the previous frame is used.
	 * Mandatory rates for IEEE 802.11g PHY: 1, 2, 5.5, 11, 6, 12, 24 Mbps
	 */
	rate = -1;
159 160 161 162
	/* use lowest available if everything fails */
	mrate = sband->bitrates[0].bitrate;
	for (i = 0; i < sband->n_bitrates; i++) {
		struct ieee80211_rate *r = &sband->bitrates[i];
163

164 165
		if (r->bitrate > txrate->bitrate)
			break;
166

167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
		if (tx->sdata->basic_rates & BIT(i))
			rate = r->bitrate;

		switch (sband->band) {
		case IEEE80211_BAND_2GHZ: {
			u32 flag;
			if (tx->sdata->flags & IEEE80211_SDATA_OPERATING_GMODE)
				flag = IEEE80211_RATE_MANDATORY_G;
			else
				flag = IEEE80211_RATE_MANDATORY_B;
			if (r->flags & flag)
				mrate = r->bitrate;
			break;
		}
		case IEEE80211_BAND_5GHZ:
			if (r->flags & IEEE80211_RATE_MANDATORY_A)
				mrate = r->bitrate;
			break;
		case IEEE80211_NUM_BANDS:
			WARN_ON(1);
			break;
		}
189 190 191 192 193 194 195 196 197 198 199 200
	}
	if (rate == -1) {
		/* No matching basic rate found; use highest suitable mandatory
		 * PHY rate */
		rate = mrate;
	}

	/* Time needed to transmit ACK
	 * (10 bytes + 4-byte FCS = 112 bits) plus SIFS; rounded up
	 * to closest integer */

	dur = ieee80211_frame_duration(local, 10, rate, erp,
201
				tx->sdata->bss_conf.use_short_preamble);
202 203 204 205 206 207 208

	if (next_frag_len) {
		/* Frame is fragmented: duration increases with time needed to
		 * transmit next fragment plus ACK and 2 x SIFS. */
		dur *= 2; /* ACK + SIFS */
		/* next fragment */
		dur += ieee80211_frame_duration(local, next_frag_len,
209
				txrate->bitrate, erp,
210
				tx->sdata->bss_conf.use_short_preamble);
211 212 213 214 215 216 217 218 219 220 221 222 223 224
	}

	return dur;
}

static int inline is_ieee80211_device(struct net_device *dev,
				      struct net_device *master)
{
	return (wdev_priv(dev->ieee80211_ptr) ==
		wdev_priv(master->ieee80211_ptr));
}

/* tx handlers */

225
static ieee80211_tx_result
226
ieee80211_tx_h_check_assoc(struct ieee80211_tx_data *tx)
227 228
{
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
229
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;
230
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
231
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
232 233
	u32 sta_flags;

234
	if (unlikely(info->flags & IEEE80211_TX_CTL_INJECTED))
235
		return TX_CONTINUE;
236

Z
Zhu Yi 已提交
237
	if (unlikely(tx->local->sta_sw_scanning) &&
238 239
	    ((tx->fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_MGMT ||
	     (tx->fc & IEEE80211_FCTL_STYPE) != IEEE80211_STYPE_PROBE_REQ))
240
		return TX_DROP;
241

242 243 244
	if (tx->sdata->vif.type == IEEE80211_IF_TYPE_MESH_POINT)
		return TX_CONTINUE;

245
	if (tx->flags & IEEE80211_TX_PS_BUFFERED)
246
		return TX_CONTINUE;
247

248
	sta_flags = tx->sta ? get_sta_flags(tx->sta) : 0;
249

250
	if (likely(tx->flags & IEEE80211_TX_UNICAST)) {
251
		if (unlikely(!(sta_flags & WLAN_STA_ASSOC) &&
252
			     tx->sdata->vif.type != IEEE80211_IF_TYPE_IBSS &&
253 254
			     (tx->fc & IEEE80211_FCTL_FTYPE) == IEEE80211_FTYPE_DATA)) {
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
255
			DECLARE_MAC_BUF(mac);
256
			printk(KERN_DEBUG "%s: dropped data frame to not "
257 258
			       "associated station %s\n",
			       tx->dev->name, print_mac(mac, hdr->addr1));
259 260
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
			I802_DEBUG_INC(tx->local->tx_handlers_drop_not_assoc);
261
			return TX_DROP;
262 263 264 265
		}
	} else {
		if (unlikely((tx->fc & IEEE80211_FCTL_FTYPE) == IEEE80211_FTYPE_DATA &&
			     tx->local->num_sta == 0 &&
266
			     tx->sdata->vif.type != IEEE80211_IF_TYPE_IBSS)) {
267 268 269 270
			/*
			 * No associated STAs - no need to send multicast
			 * frames.
			 */
271
			return TX_DROP;
272
		}
273
		return TX_CONTINUE;
274 275
	}

276
	return TX_CONTINUE;
277 278
}

279
static ieee80211_tx_result
280
ieee80211_tx_h_sequence(struct ieee80211_tx_data *tx)
281 282 283 284 285 286
{
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)tx->skb->data;

	if (ieee80211_get_hdrlen(le16_to_cpu(hdr->frame_control)) >= 24)
		ieee80211_include_sequence(tx->sdata, hdr);

287
	return TX_CONTINUE;
288 289 290 291 292 293 294 295 296 297 298 299 300
}

/* This function is called whenever the AP is about to exceed the maximum limit
 * of buffered frames for power saving STAs. This situation should not really
 * happen often during normal operation, so dropping the oldest buffered packet
 * from each queue should be OK to make some room for new frames. */
static void purge_old_ps_buffers(struct ieee80211_local *local)
{
	int total = 0, purged = 0;
	struct sk_buff *skb;
	struct ieee80211_sub_if_data *sdata;
	struct sta_info *sta;

301 302 303 304 305 306
	/*
	 * virtual interfaces are protected by RCU
	 */
	rcu_read_lock();

	list_for_each_entry_rcu(sdata, &local->interfaces, list) {
307 308
		struct ieee80211_if_ap *ap;
		if (sdata->dev == local->mdev ||
309
		    sdata->vif.type != IEEE80211_IF_TYPE_AP)
310 311 312 313 314 315 316 317 318 319
			continue;
		ap = &sdata->u.ap;
		skb = skb_dequeue(&ap->ps_bc_buf);
		if (skb) {
			purged++;
			dev_kfree_skb(skb);
		}
		total += skb_queue_len(&ap->ps_bc_buf);
	}

320
	list_for_each_entry_rcu(sta, &local->sta_list, list) {
321 322 323 324 325 326 327
		skb = skb_dequeue(&sta->ps_tx_buf);
		if (skb) {
			purged++;
			dev_kfree_skb(skb);
		}
		total += skb_queue_len(&sta->ps_tx_buf);
	}
328 329

	rcu_read_unlock();
330 331 332

	local->total_ps_buffered = total;
	printk(KERN_DEBUG "%s: PS buffers full - purged %d frames\n",
333
	       wiphy_name(local->hw.wiphy), purged);
334 335
}

336
static ieee80211_tx_result
337
ieee80211_tx_h_multicast_ps_buf(struct ieee80211_tx_data *tx)
338
{
339 340
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);

341 342 343 344 345 346 347 348 349 350
	/*
	 * broadcast/multicast frame
	 *
	 * If any of the associated stations is in power save mode,
	 * the frame is buffered to be sent after DTIM beacon frame.
	 * This is done either by the hardware or us.
	 */

	/* not AP/IBSS or ordered frame */
	if (!tx->sdata->bss || (tx->fc & IEEE80211_FCTL_ORDER))
351
		return TX_CONTINUE;
352 353 354

	/* no stations in PS mode */
	if (!atomic_read(&tx->sdata->bss->num_sta_ps))
355
		return TX_CONTINUE;
356 357 358

	/* buffered in mac80211 */
	if (tx->local->hw.flags & IEEE80211_HW_HOST_BROADCAST_PS_BUFFERING) {
359 360 361 362 363 364 365 366 367 368 369 370 371
		if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
			purge_old_ps_buffers(tx->local);
		if (skb_queue_len(&tx->sdata->bss->ps_bc_buf) >=
		    AP_MAX_BC_BUFFER) {
			if (net_ratelimit()) {
				printk(KERN_DEBUG "%s: BC TX buffer full - "
				       "dropping the oldest frame\n",
				       tx->dev->name);
			}
			dev_kfree_skb(skb_dequeue(&tx->sdata->bss->ps_bc_buf));
		} else
			tx->local->total_ps_buffered++;
		skb_queue_tail(&tx->sdata->bss->ps_bc_buf, tx->skb);
372
		return TX_QUEUED;
373 374
	}

375
	/* buffered in hardware */
376
	info->flags |= IEEE80211_TX_CTL_SEND_AFTER_DTIM;
377

378
	return TX_CONTINUE;
379 380
}

381
static ieee80211_tx_result
382
ieee80211_tx_h_unicast_ps_buf(struct ieee80211_tx_data *tx)
383 384
{
	struct sta_info *sta = tx->sta;
385
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
386
	u32 staflags;
387
	DECLARE_MAC_BUF(mac);
388 389 390 391

	if (unlikely(!sta ||
		     ((tx->fc & IEEE80211_FCTL_FTYPE) == IEEE80211_FTYPE_MGMT &&
		      (tx->fc & IEEE80211_FCTL_STYPE) == IEEE80211_STYPE_PROBE_RESP)))
392
		return TX_CONTINUE;
393

394 395 396 397
	staflags = get_sta_flags(sta);

	if (unlikely((staflags & WLAN_STA_PS) &&
		     !(staflags & WLAN_STA_PSPOLL))) {
398
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
399
		printk(KERN_DEBUG "STA %s aid %d: PS buffer (entries "
400
		       "before %d)\n",
401
		       print_mac(mac, sta->addr), sta->aid,
402 403 404 405 406 407 408
		       skb_queue_len(&sta->ps_tx_buf));
#endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
		if (tx->local->total_ps_buffered >= TOTAL_MAX_TX_BUFFER)
			purge_old_ps_buffers(tx->local);
		if (skb_queue_len(&sta->ps_tx_buf) >= STA_MAX_TX_BUFFER) {
			struct sk_buff *old = skb_dequeue(&sta->ps_tx_buf);
			if (net_ratelimit()) {
409
				printk(KERN_DEBUG "%s: STA %s TX "
410
				       "buffer full - dropping oldest frame\n",
411
				       tx->dev->name, print_mac(mac, sta->addr));
412 413 414 415
			}
			dev_kfree_skb(old);
		} else
			tx->local->total_ps_buffered++;
416

417
		/* Queue frame to be sent after STA sends an PS Poll frame */
418 419 420
		if (skb_queue_empty(&sta->ps_tx_buf))
			sta_info_set_tim_bit(sta);

421
		info->control.jiffies = jiffies;
422
		skb_queue_tail(&sta->ps_tx_buf, tx->skb);
423
		return TX_QUEUED;
424 425
	}
#ifdef CONFIG_MAC80211_VERBOSE_PS_DEBUG
426
	else if (unlikely(test_sta_flags(sta, WLAN_STA_PS))) {
427
		printk(KERN_DEBUG "%s: STA %s in PS mode, but pspoll "
428
		       "set -> send frame\n", tx->dev->name,
429
		       print_mac(mac, sta->addr));
430 431
	}
#endif /* CONFIG_MAC80211_VERBOSE_PS_DEBUG */
432
	clear_sta_flags(sta, WLAN_STA_PSPOLL);
433

434
	return TX_CONTINUE;
435 436
}

437
static ieee80211_tx_result
438
ieee80211_tx_h_ps_buf(struct ieee80211_tx_data *tx)
439
{
440
	if (unlikely(tx->flags & IEEE80211_TX_PS_BUFFERED))
441
		return TX_CONTINUE;
442

443
	if (tx->flags & IEEE80211_TX_UNICAST)
444 445 446 447 448
		return ieee80211_tx_h_unicast_ps_buf(tx);
	else
		return ieee80211_tx_h_multicast_ps_buf(tx);
}

449
static ieee80211_tx_result
450
ieee80211_tx_h_select_key(struct ieee80211_tx_data *tx)
451
{
452
	struct ieee80211_key *key;
453
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
454
	u16 fc = tx->fc;
455

456
	if (unlikely(info->flags & IEEE80211_TX_CTL_DO_NOT_ENCRYPT))
457
		tx->key = NULL;
458 459 460 461
	else if (tx->sta && (key = rcu_dereference(tx->sta->key)))
		tx->key = key;
	else if ((key = rcu_dereference(tx->sdata->default_key)))
		tx->key = key;
462
	else if (tx->sdata->drop_unencrypted &&
463 464
		 !(info->flags & IEEE80211_TX_CTL_EAPOL_FRAME) &&
		 !(info->flags & IEEE80211_TX_CTL_INJECTED)) {
465
		I802_DEBUG_INC(tx->local->tx_handlers_drop_unencrypted);
466
		return TX_DROP;
467
	} else
468 469 470
		tx->key = NULL;

	if (tx->key) {
471 472
		u16 ftype, stype;

473
		tx->key->tx_rx_count++;
474
		/* TODO: add threshold stuff again */
475 476 477 478 479 480 481 482 483 484 485 486 487 488 489

		switch (tx->key->conf.alg) {
		case ALG_WEP:
			ftype = fc & IEEE80211_FCTL_FTYPE;
			stype = fc & IEEE80211_FCTL_STYPE;

			if (ftype == IEEE80211_FTYPE_MGMT &&
			    stype == IEEE80211_STYPE_AUTH)
				break;
		case ALG_TKIP:
		case ALG_CCMP:
			if (!WLAN_FC_DATA_PRESENT(fc))
				tx->key = NULL;
			break;
		}
490 491
	}

492
	if (!tx->key || !(tx->key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
493
		info->flags |= IEEE80211_TX_CTL_DO_NOT_ENCRYPT;
494

495
	return TX_CONTINUE;
496 497
}

498
static ieee80211_tx_result
499
ieee80211_tx_h_rate_ctrl(struct ieee80211_tx_data *tx)
500
{
501
	struct rate_selection rsel;
502
	struct ieee80211_supported_band *sband;
503
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
504

505
	sband = tx->local->hw.wiphy->bands[tx->channel->band];
506

507
	if (likely(tx->rate_idx < 0)) {
508
		rate_control_get_rate(tx->dev, sband, tx->skb, &rsel);
509 510
		tx->rate_idx = rsel.rate_idx;
		if (unlikely(rsel.probe_idx >= 0)) {
511
			info->flags |= IEEE80211_TX_CTL_RATE_CTRL_PROBE;
512
			tx->flags |= IEEE80211_TX_PROBE_LAST_FRAG;
513
			info->control.alt_retry_rate_idx = tx->rate_idx;
514
			tx->rate_idx = rsel.probe_idx;
515
		} else
516
			info->control.alt_retry_rate_idx = -1;
517

518
		if (unlikely(tx->rate_idx < 0))
519
			return TX_DROP;
520
	} else
521
		info->control.alt_retry_rate_idx = -1;
522

523
	if (tx->sdata->bss_conf.use_cts_prot &&
524 525 526
	    (tx->flags & IEEE80211_TX_FRAGMENTED) && (rsel.nonerp_idx >= 0)) {
		tx->last_frag_rate_idx = tx->rate_idx;
		if (rsel.probe_idx >= 0)
527
			tx->flags &= ~IEEE80211_TX_PROBE_LAST_FRAG;
528
		else
529
			tx->flags |= IEEE80211_TX_PROBE_LAST_FRAG;
530
		tx->rate_idx = rsel.nonerp_idx;
531 532
		info->tx_rate_idx = rsel.nonerp_idx;
		info->flags &= ~IEEE80211_TX_CTL_RATE_CTRL_PROBE;
533
	} else {
534
		tx->last_frag_rate_idx = tx->rate_idx;
535
		info->tx_rate_idx = tx->rate_idx;
536
	}
537
	info->tx_rate_idx = tx->rate_idx;
538

539
	return TX_CONTINUE;
540 541
}

542
static ieee80211_tx_result
543
ieee80211_tx_h_misc(struct ieee80211_tx_data *tx)
544 545
{
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) tx->skb->data;
546
	u16 fc = le16_to_cpu(hdr->frame_control);
547
	u16 dur;
548
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(tx->skb);
549 550 551
	struct ieee80211_supported_band *sband;

	sband = tx->local->hw.wiphy->bands[tx->channel->band];
552

553 554 555 556
	if (tx->sta)
		info->control.aid = tx->sta->aid;

	if (!info->control.retry_limit) {
557
		if (!is_multicast_ether_addr(hdr->addr1)) {
558 559 560
			int len = min_t(int, tx->skb->len + FCS_LEN,
					tx->local->fragmentation_threshold);
			if (len > tx->local->rts_threshold
561
			    && tx->local->rts_threshold <
562 563 564 565 566
						IEEE80211_MAX_RTS_THRESHOLD) {
				info->flags |= IEEE80211_TX_CTL_USE_RTS_CTS;
				info->flags |=
					IEEE80211_TX_CTL_LONG_RETRY_LIMIT;
				info->control.retry_limit =
567 568
					tx->local->long_retry_limit;
			} else {
569
				info->control.retry_limit =
570 571
					tx->local->short_retry_limit;
			}
572
		} else {
573
			info->control.retry_limit = 1;
574 575 576
		}
	}

577
	if (tx->flags & IEEE80211_TX_FRAGMENTED) {
578 579 580 581
		/* Do not use multiple retry rates when sending fragmented
		 * frames.
		 * TODO: The last fragment could still use multiple retry
		 * rates. */
582
		info->control.alt_retry_rate_idx = -1;
583 584 585 586 587
	}

	/* Use CTS protection for unicast frames sent using extended rates if
	 * there are associated non-ERP stations and RTS/CTS is not configured
	 * for the frame. */
588
	if ((tx->sdata->flags & IEEE80211_SDATA_OPERATING_GMODE) &&
589
	    (sband->bitrates[tx->rate_idx].flags & IEEE80211_RATE_ERP_G) &&
590
	    (tx->flags & IEEE80211_TX_UNICAST) &&
591
	    tx->sdata->bss_conf.use_cts_prot &&
592 593
	    !(info->flags & IEEE80211_TX_CTL_USE_RTS_CTS))
		info->flags |= IEEE80211_TX_CTL_USE_CTS_PROTECT;
594

595 596 597 598
	/* Transmit data frames using short preambles if the driver supports
	 * short preambles at the selected rate and short preambles are
	 * available on the network at the current point in time. */
	if (((fc & IEEE80211_FCTL_FTYPE) == IEEE80211_FTYPE_DATA) &&
599
	    (sband->bitrates[tx->rate_idx].flags & IEEE80211_RATE_SHORT_PREAMBLE) &&
600
	    tx->sdata->bss_conf.use_short_preamble &&
601
	    (!tx->sta || test_sta_flags(tx->sta, WLAN_STA_SHORT_PREAMBLE))) {
602
		info->flags |= IEEE80211_TX_CTL_SHORT_PREAMBLE;
603 604
	}

605 606 607 608
	/* Setup duration field for the first fragment of the frame. Duration
	 * for remaining fragments will be updated when they are being sent
	 * to low-level driver in ieee80211_tx(). */
	dur = ieee80211_duration(tx, is_multicast_ether_addr(hdr->addr1),
609 610
				 (tx->flags & IEEE80211_TX_FRAGMENTED) ?
				 tx->extra_frag[0]->len : 0);
611 612
	hdr->duration_id = cpu_to_le16(dur);

613 614
	if ((info->flags & IEEE80211_TX_CTL_USE_RTS_CTS) ||
	    (info->flags & IEEE80211_TX_CTL_USE_CTS_PROTECT)) {
615 616
		struct ieee80211_rate *rate;
		s8 baserate = -1;
617 618
		int idx;

619
		/* Do not use multiple retry rates when using RTS/CTS */
620
		info->control.alt_retry_rate_idx = -1;
621 622

		/* Use min(data rate, max base rate) as CTS/RTS rate */
623
		rate = &sband->bitrates[tx->rate_idx];
624 625 626 627 628

		for (idx = 0; idx < sband->n_bitrates; idx++) {
			if (sband->bitrates[idx].bitrate > rate->bitrate)
				continue;
			if (tx->sdata->basic_rates & BIT(idx) &&
629 630 631 632
			    (baserate < 0 ||
			     (sband->bitrates[baserate].bitrate
			      < sband->bitrates[idx].bitrate)))
				baserate = idx;
633
		}
634

635
		if (baserate >= 0)
636
			info->control.rts_cts_rate_idx = baserate;
637
		else
638
			info->control.rts_cts_rate_idx = 0;
639 640
	}

641
	if (tx->sta)
642
		info->control.aid = tx->sta->aid;
643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660

	return TX_CONTINUE;
}

static ieee80211_tx_result
ieee80211_tx_h_fragment(struct ieee80211_tx_data *tx)
{
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) tx->skb->data;
	size_t hdrlen, per_fragm, num_fragm, payload_len, left;
	struct sk_buff **frags, *first, *frag;
	int i;
	u16 seq;
	u8 *pos;
	int frag_threshold = tx->local->fragmentation_threshold;

	if (!(tx->flags & IEEE80211_TX_FRAGMENTED))
		return TX_CONTINUE;

661 662
	/*
	 * Warn when submitting a fragmented A-MPDU frame and drop it.
663 664
	 * This scenario is handled in __ieee80211_tx_prepare but extra
	 * caution taken here as fragmented ampdu may cause Tx stop.
665 666
	 */
	if (WARN_ON(tx->flags & IEEE80211_TX_CTL_AMPDU ||
667 668
		    skb_get_queue_mapping(tx->skb) >=
			ieee80211_num_regular_queues(&tx->local->hw)))
669 670
		return TX_DROP;

671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716
	first = tx->skb;

	hdrlen = ieee80211_get_hdrlen(tx->fc);
	payload_len = first->len - hdrlen;
	per_fragm = frag_threshold - hdrlen - FCS_LEN;
	num_fragm = DIV_ROUND_UP(payload_len, per_fragm);

	frags = kzalloc(num_fragm * sizeof(struct sk_buff *), GFP_ATOMIC);
	if (!frags)
		goto fail;

	hdr->frame_control |= cpu_to_le16(IEEE80211_FCTL_MOREFRAGS);
	seq = le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_SEQ;
	pos = first->data + hdrlen + per_fragm;
	left = payload_len - per_fragm;
	for (i = 0; i < num_fragm - 1; i++) {
		struct ieee80211_hdr *fhdr;
		size_t copylen;

		if (left <= 0)
			goto fail;

		/* reserve enough extra head and tail room for possible
		 * encryption */
		frag = frags[i] =
			dev_alloc_skb(tx->local->tx_headroom +
				      frag_threshold +
				      IEEE80211_ENCRYPT_HEADROOM +
				      IEEE80211_ENCRYPT_TAILROOM);
		if (!frag)
			goto fail;
		/* Make sure that all fragments use the same priority so
		 * that they end up using the same TX queue */
		frag->priority = first->priority;
		skb_reserve(frag, tx->local->tx_headroom +
				  IEEE80211_ENCRYPT_HEADROOM);
		fhdr = (struct ieee80211_hdr *) skb_put(frag, hdrlen);
		memcpy(fhdr, first->data, hdrlen);
		if (i == num_fragm - 2)
			fhdr->frame_control &= cpu_to_le16(~IEEE80211_FCTL_MOREFRAGS);
		fhdr->seq_ctrl = cpu_to_le16(seq | ((i + 1) & IEEE80211_SCTL_FRAG));
		copylen = left > per_fragm ? per_fragm : left;
		memcpy(skb_put(frag, copylen), pos, copylen);

		pos += copylen;
		left -= copylen;
717
	}
718 719 720 721
	skb_trim(first, hdrlen + per_fragm);

	tx->num_extra_frag = num_fragm - 1;
	tx->extra_frag = frags;
722

723
	return TX_CONTINUE;
724 725 726 727 728 729 730 731 732 733 734

 fail:
	printk(KERN_DEBUG "%s: failed to fragment frame\n", tx->dev->name);
	if (frags) {
		for (i = 0; i < num_fragm - 1; i++)
			if (frags[i])
				dev_kfree_skb(frags[i]);
		kfree(frags);
	}
	I802_DEBUG_INC(tx->local->tx_handlers_drop_fragment);
	return TX_DROP;
735 736
}

737
static ieee80211_tx_result
738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
ieee80211_tx_h_encrypt(struct ieee80211_tx_data *tx)
{
	if (!tx->key)
		return TX_CONTINUE;

	switch (tx->key->conf.alg) {
	case ALG_WEP:
		return ieee80211_crypto_wep_encrypt(tx);
	case ALG_TKIP:
		return ieee80211_crypto_tkip_encrypt(tx);
	case ALG_CCMP:
		return ieee80211_crypto_ccmp_encrypt(tx);
	}

	/* not reached */
	WARN_ON(1);
	return TX_DROP;
}

static ieee80211_tx_result
ieee80211_tx_h_stats(struct ieee80211_tx_data *tx)
759
{
760
	int i;
761

762 763
	if (!tx->sta)
		return TX_CONTINUE;
764

765 766 767
	tx->sta->tx_packets++;
	tx->sta->tx_fragments++;
	tx->sta->tx_bytes += tx->skb->len;
768
	if (tx->extra_frag) {
769 770 771
		tx->sta->tx_fragments += tx->num_extra_frag;
		for (i = 0; i < tx->num_extra_frag; i++)
			tx->sta->tx_bytes += tx->extra_frag[i]->len;
772 773
	}

774
	return TX_CONTINUE;
775 776 777
}


778
typedef ieee80211_tx_result (*ieee80211_tx_handler)(struct ieee80211_tx_data *);
779
static ieee80211_tx_handler ieee80211_tx_handlers[] =
780 781 782 783 784 785 786 787
{
	ieee80211_tx_h_check_assoc,
	ieee80211_tx_h_sequence,
	ieee80211_tx_h_ps_buf,
	ieee80211_tx_h_select_key,
	ieee80211_tx_h_michael_mic_add,
	ieee80211_tx_h_rate_ctrl,
	ieee80211_tx_h_misc,
788
	ieee80211_tx_h_fragment,
789
	/* handlers after fragment must be aware of tx info fragmentation! */
790 791
	ieee80211_tx_h_encrypt,
	ieee80211_tx_h_stats,
792 793 794 795 796 797 798 799 800
	NULL
};

/* actual transmit path */

/*
 * deal with packet injection down monitor interface
 * with Radiotap Header -- only called for monitor mode interface
 */
801
static ieee80211_tx_result
802
__ieee80211_parse_tx_radiotap(struct ieee80211_tx_data *tx,
803
			      struct sk_buff *skb)
804 805 806 807 808 809 810 811 812 813 814 815
{
	/*
	 * this is the moment to interpret and discard the radiotap header that
	 * must be at the start of the packet injected in Monitor mode
	 *
	 * Need to take some care with endian-ness since radiotap
	 * args are little-endian
	 */

	struct ieee80211_radiotap_iterator iterator;
	struct ieee80211_radiotap_header *rthdr =
		(struct ieee80211_radiotap_header *) skb->data;
816
	struct ieee80211_supported_band *sband;
817
	int ret = ieee80211_radiotap_iterator_init(&iterator, rthdr, skb->len);
818
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
819

820
	sband = tx->local->hw.wiphy->bands[tx->channel->band];
821

822 823
	info->flags |= IEEE80211_TX_CTL_DO_NOT_ENCRYPT;
	info->flags |= IEEE80211_TX_CTL_INJECTED;
824
	tx->flags &= ~IEEE80211_TX_FRAGMENTED;
825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853

	/*
	 * for every radiotap entry that is present
	 * (ieee80211_radiotap_iterator_next returns -ENOENT when no more
	 * entries present, or -EINVAL on error)
	 */

	while (!ret) {
		int i, target_rate;

		ret = ieee80211_radiotap_iterator_next(&iterator);

		if (ret)
			continue;

		/* see if this argument is something we can use */
		switch (iterator.this_arg_index) {
		/*
		 * You must take care when dereferencing iterator.this_arg
		 * for multibyte types... the pointer is not aligned.  Use
		 * get_unaligned((type *)iterator.this_arg) to dereference
		 * iterator.this_arg for type "type" safely on all arches.
		*/
		case IEEE80211_RADIOTAP_RATE:
			/*
			 * radiotap rate u8 is in 500kbps units eg, 0x02=1Mbps
			 * ieee80211 rate int is in 100kbps units eg, 0x0a=1Mbps
			 */
			target_rate = (*iterator.this_arg) * 5;
854 855
			for (i = 0; i < sband->n_bitrates; i++) {
				struct ieee80211_rate *r;
856

857 858 859
				r = &sband->bitrates[i];

				if (r->bitrate == target_rate) {
860
					tx->rate_idx = i;
861 862
					break;
				}
863 864 865 866 867 868 869 870
			}
			break;

		case IEEE80211_RADIOTAP_ANTENNA:
			/*
			 * radiotap uses 0 for 1st ant, mac80211 is 1 for
			 * 1st ant
			 */
871
			info->antenna_sel_tx = (*iterator.this_arg) + 1;
872 873
			break;

874
#if 0
875 876 877
		case IEEE80211_RADIOTAP_DBM_TX_POWER:
			control->power_level = *iterator.this_arg;
			break;
878
#endif
879 880 881 882 883 884 885 886 887 888 889

		case IEEE80211_RADIOTAP_FLAGS:
			if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FCS) {
				/*
				 * this indicates that the skb we have been
				 * handed has the 32-bit FCS CRC at the end...
				 * we should react to that by snipping it off
				 * because it will be recomputed and added
				 * on transmission
				 */
				if (skb->len < (iterator.max_length + FCS_LEN))
890
					return TX_DROP;
891 892 893

				skb_trim(skb, skb->len - FCS_LEN);
			}
894
			if (*iterator.this_arg & IEEE80211_RADIOTAP_F_WEP)
895 896
				info->flags &=
					~IEEE80211_TX_CTL_DO_NOT_ENCRYPT;
897
			if (*iterator.this_arg & IEEE80211_RADIOTAP_F_FRAG)
898
				tx->flags |= IEEE80211_TX_FRAGMENTED;
899 900
			break;

901 902 903 904 905 906
		/*
		 * Please update the file
		 * Documentation/networking/mac80211-injection.txt
		 * when parsing new fields here.
		 */

907 908 909 910 911 912
		default:
			break;
		}
	}

	if (ret != -ENOENT) /* ie, if we didn't simply run out of fields */
913
		return TX_DROP;
914 915 916 917 918 919 920 921

	/*
	 * remove the radiotap header
	 * iterator->max_length was sanity-checked against
	 * skb->len by iterator init
	 */
	skb_pull(skb, iterator.max_length);

922
	return TX_CONTINUE;
923 924
}

925 926 927
/*
 * initialises @tx
 */
928
static ieee80211_tx_result
929
__ieee80211_tx_prepare(struct ieee80211_tx_data *tx,
930
		       struct sk_buff *skb,
931
		       struct net_device *dev)
932 933
{
	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
934
	struct ieee80211_hdr *hdr;
935
	struct ieee80211_sub_if_data *sdata;
936
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
937 938 939 940 941 942 943 944

	int hdrlen;

	memset(tx, 0, sizeof(*tx));
	tx->skb = skb;
	tx->dev = dev; /* use original interface */
	tx->local = local;
	tx->sdata = IEEE80211_DEV_TO_SUB_IF(dev);
945
	tx->channel = local->hw.conf.channel;
946 947
	tx->rate_idx = -1;
	tx->last_frag_rate_idx = -1;
948
	/*
949 950
	 * Set this flag (used below to indicate "automatic fragmentation"),
	 * it will be cleared/left by radiotap as desired.
951
	 */
952
	tx->flags |= IEEE80211_TX_FRAGMENTED;
953 954 955

	/* process and remove the injection radiotap header */
	sdata = IEEE80211_DEV_TO_SUB_IF(dev);
956
	if (unlikely(sdata->vif.type == IEEE80211_IF_TYPE_MNTR)) {
957 958
		if (__ieee80211_parse_tx_radiotap(tx, skb) == TX_DROP)
			return TX_DROP;
959

960
		/*
961 962 963
		 * __ieee80211_parse_tx_radiotap has now removed
		 * the radiotap header that was present and pre-filled
		 * 'tx' with tx control information.
964 965 966
		 */
	}

967 968
	hdr = (struct ieee80211_hdr *) skb->data;

969 970
	tx->sta = sta_info_get(local, hdr->addr1);
	tx->fc = le16_to_cpu(hdr->frame_control);
971

972
	if (is_multicast_ether_addr(hdr->addr1)) {
973
		tx->flags &= ~IEEE80211_TX_UNICAST;
974
		info->flags |= IEEE80211_TX_CTL_NO_ACK;
975
	} else {
976
		tx->flags |= IEEE80211_TX_UNICAST;
977
		info->flags &= ~IEEE80211_TX_CTL_NO_ACK;
978
	}
979

980 981
	if (tx->flags & IEEE80211_TX_FRAGMENTED) {
		if ((tx->flags & IEEE80211_TX_UNICAST) &&
982
		    skb->len + FCS_LEN > local->fragmentation_threshold &&
983 984
		    !local->ops->set_frag_threshold &&
		    !(info->flags & IEEE80211_TX_CTL_AMPDU))
985
			tx->flags |= IEEE80211_TX_FRAGMENTED;
986
		else
987
			tx->flags &= ~IEEE80211_TX_FRAGMENTED;
988 989
	}

990
	if (!tx->sta)
991
		info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
992
	else if (test_and_clear_sta_flags(tx->sta, WLAN_STA_CLEAR_PS_FILT))
993
		info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
994

995 996 997 998 999
	hdrlen = ieee80211_get_hdrlen(tx->fc);
	if (skb->len > hdrlen + sizeof(rfc1042_header) + 2) {
		u8 *pos = &skb->data[hdrlen + sizeof(rfc1042_header)];
		tx->ethertype = (pos[0] << 8) | pos[1];
	}
1000
	info->flags |= IEEE80211_TX_CTL_FIRST_FRAGMENT;
1001

1002
	return TX_CONTINUE;
1003 1004
}

1005
/*
1006 1007
 * NB: @tx is uninitialised when passed in here
 */
1008
static int ieee80211_tx_prepare(struct ieee80211_tx_data *tx,
1009
				struct sk_buff *skb,
1010
				struct net_device *mdev)
1011
{
1012
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1013 1014
	struct net_device *dev;

1015
	dev = dev_get_by_index(&init_net, info->control.ifindex);
1016 1017 1018 1019 1020 1021
	if (unlikely(dev && !is_ieee80211_device(dev, mdev))) {
		dev_put(dev);
		dev = NULL;
	}
	if (unlikely(!dev))
		return -ENODEV;
1022
	/* initialises tx with control */
1023
	__ieee80211_tx_prepare(tx, skb, dev);
1024
	dev_put(dev);
1025 1026 1027 1028
	return 0;
}

static int __ieee80211_tx(struct ieee80211_local *local, struct sk_buff *skb,
1029
			  struct ieee80211_tx_data *tx)
1030
{
1031
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1032 1033
	int ret, i;

1034
	if (netif_subqueue_stopped(local->mdev, skb))
1035
		return IEEE80211_TX_AGAIN;
1036

1037
	if (skb) {
1038 1039
		ieee80211_dump_frame(wiphy_name(local->hw.wiphy),
				     "TX to low-level driver", skb);
1040
		ret = local->ops->tx(local_to_hw(local), skb);
1041 1042 1043 1044 1045
		if (ret)
			return IEEE80211_TX_AGAIN;
		local->mdev->trans_start = jiffies;
		ieee80211_led_tx(local, 1);
	}
1046 1047 1048
	if (tx->extra_frag) {
		for (i = 0; i < tx->num_extra_frag; i++) {
			if (!tx->extra_frag[i])
1049
				continue;
1050 1051 1052 1053 1054
			info = IEEE80211_SKB_CB(tx->extra_frag[i]);
			info->flags &= ~(IEEE80211_TX_CTL_USE_RTS_CTS |
					 IEEE80211_TX_CTL_USE_CTS_PROTECT |
					 IEEE80211_TX_CTL_CLEAR_PS_FILT |
					 IEEE80211_TX_CTL_FIRST_FRAGMENT);
1055 1056
			if (netif_subqueue_stopped(local->mdev,
						   tx->extra_frag[i]))
1057
				return IEEE80211_TX_FRAG_AGAIN;
1058
			if (i == tx->num_extra_frag) {
1059
				info->tx_rate_idx = tx->last_frag_rate_idx;
1060

1061
				if (tx->flags & IEEE80211_TX_PROBE_LAST_FRAG)
1062 1063
					info->flags |=
						IEEE80211_TX_CTL_RATE_CTRL_PROBE;
1064
				else
1065 1066
					info->flags &=
						~IEEE80211_TX_CTL_RATE_CTRL_PROBE;
1067 1068
			}

1069
			ieee80211_dump_frame(wiphy_name(local->hw.wiphy),
1070
					     "TX to low-level driver",
1071
					     tx->extra_frag[i]);
1072
			ret = local->ops->tx(local_to_hw(local),
1073
					    tx->extra_frag[i]);
1074 1075 1076 1077
			if (ret)
				return IEEE80211_TX_FRAG_AGAIN;
			local->mdev->trans_start = jiffies;
			ieee80211_led_tx(local, 1);
1078
			tx->extra_frag[i] = NULL;
1079
		}
1080 1081
		kfree(tx->extra_frag);
		tx->extra_frag = NULL;
1082 1083 1084 1085
	}
	return IEEE80211_TX_OK;
}

1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
/*
 * Invoke TX handlers, return 0 on success and non-zero if the
 * frame was dropped or queued.
 */
static int invoke_tx_handlers(struct ieee80211_tx_data *tx)
{
	struct ieee80211_local *local = tx->local;
	struct sk_buff *skb = tx->skb;
	ieee80211_tx_handler *handler;
	ieee80211_tx_result res = TX_DROP;
	int i;

	for (handler = ieee80211_tx_handlers; *handler != NULL; handler++) {
		res = (*handler)(tx);
		if (res != TX_CONTINUE)
			break;
	}

	if (unlikely(res == TX_DROP)) {
		I802_DEBUG_INC(local->tx_handlers_drop);
		dev_kfree_skb(skb);
		for (i = 0; i < tx->num_extra_frag; i++)
			if (tx->extra_frag[i])
				dev_kfree_skb(tx->extra_frag[i]);
		kfree(tx->extra_frag);
		return -1;
	} else if (unlikely(res == TX_QUEUED)) {
		I802_DEBUG_INC(local->tx_handlers_queued);
		return -1;
	}

	return 0;
}

1120
static int ieee80211_tx(struct net_device *dev, struct sk_buff *skb)
1121 1122 1123
{
	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
	struct sta_info *sta;
1124
	struct ieee80211_tx_data tx;
1125
	ieee80211_tx_result res_prepare;
1126
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1127
	int ret, i;
1128
	u16 queue;
1129

1130 1131 1132
	queue = skb_get_queue_mapping(skb);

	WARN_ON(test_bit(queue, local->queues_pending));
1133 1134 1135 1136 1137 1138

	if (unlikely(skb->len < 10)) {
		dev_kfree_skb(skb);
		return 0;
	}

1139 1140
	rcu_read_lock();

1141
	/* initialises tx */
1142
	res_prepare = __ieee80211_tx_prepare(&tx, skb, dev);
1143

1144
	if (res_prepare == TX_DROP) {
1145
		dev_kfree_skb(skb);
1146
		rcu_read_unlock();
1147 1148 1149 1150
		return 0;
	}

	sta = tx.sta;
1151
	tx.channel = local->hw.conf.channel;
1152
	info->band = tx.channel->band;
1153

1154 1155
	if (invoke_tx_handlers(&tx))
		goto out;
1156

1157 1158
	if (tx.extra_frag) {
		for (i = 0; i < tx.num_extra_frag; i++) {
1159 1160 1161
			int next_len, dur;
			struct ieee80211_hdr *hdr =
				(struct ieee80211_hdr *)
1162
				tx.extra_frag[i]->data;
1163

1164 1165
			if (i + 1 < tx.num_extra_frag) {
				next_len = tx.extra_frag[i + 1]->len;
1166 1167
			} else {
				next_len = 0;
1168
				tx.rate_idx = tx.last_frag_rate_idx;
1169 1170 1171 1172 1173 1174 1175 1176 1177
			}
			dur = ieee80211_duration(&tx, 0, next_len);
			hdr->duration_id = cpu_to_le16(dur);
		}
	}

retry:
	ret = __ieee80211_tx(local, skb, &tx);
	if (ret) {
1178 1179 1180 1181 1182 1183 1184
		struct ieee80211_tx_stored_packet *store;

		/*
		 * Since there are no fragmented frames on A-MPDU
		 * queues, there's no reason for a driver to reject
		 * a frame there, warn and drop it.
		 */
1185
		if (WARN_ON(queue >= ieee80211_num_regular_queues(&local->hw)))
1186 1187 1188
			goto drop;

		store = &local->pending_packet[queue];
1189 1190 1191

		if (ret == IEEE80211_TX_FRAG_AGAIN)
			skb = NULL;
1192
		set_bit(queue, local->queues_pending);
1193
		smp_mb();
1194 1195 1196 1197 1198
		/*
		 * When the driver gets out of buffers during sending of
		 * fragments and calls ieee80211_stop_queue, the netif
		 * subqueue is stopped. There is, however, a small window
		 * in which the PENDING bit is not yet set. If a buffer
1199 1200
		 * gets available in that window (i.e. driver calls
		 * ieee80211_wake_queue), we would end up with ieee80211_tx
1201
		 * called with the PENDING bit still set. Prevent this by
1202
		 * continuing transmitting here when that situation is
1203 1204 1205 1206
		 * possible to have happened.
		 */
		if (!__netif_subqueue_stopped(local->mdev, queue)) {
			clear_bit(queue, local->queues_pending);
1207 1208 1209
			goto retry;
		}
		store->skb = skb;
1210 1211
		store->extra_frag = tx.extra_frag;
		store->num_extra_frag = tx.num_extra_frag;
1212
		store->last_frag_rate_idx = tx.last_frag_rate_idx;
1213
		store->last_frag_rate_ctrl_probe =
1214
			!!(tx.flags & IEEE80211_TX_PROBE_LAST_FRAG);
1215
	}
1216
 out:
1217
	rcu_read_unlock();
1218 1219 1220 1221 1222
	return 0;

 drop:
	if (skb)
		dev_kfree_skb(skb);
1223 1224 1225 1226
	for (i = 0; i < tx.num_extra_frag; i++)
		if (tx.extra_frag[i])
			dev_kfree_skb(tx.extra_frag[i]);
	kfree(tx.extra_frag);
1227
	rcu_read_unlock();
1228 1229 1230 1231 1232
	return 0;
}

/* device xmit handlers */

1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271
static int ieee80211_skb_resize(struct ieee80211_local *local,
				struct sk_buff *skb,
				int head_need, bool may_encrypt)
{
	int tail_need = 0;

	/*
	 * This could be optimised, devices that do full hardware
	 * crypto (including TKIP MMIC) need no tailroom... But we
	 * have no drivers for such devices currently.
	 */
	if (may_encrypt) {
		tail_need = IEEE80211_ENCRYPT_TAILROOM;
		tail_need -= skb_tailroom(skb);
		tail_need = max_t(int, tail_need, 0);
	}

	if (head_need || tail_need) {
		/* Sorry. Can't account for this any more */
		skb_orphan(skb);
	}

	if (skb_header_cloned(skb))
		I802_DEBUG_INC(local->tx_expand_skb_head_cloned);
	else
		I802_DEBUG_INC(local->tx_expand_skb_head);

	if (pskb_expand_head(skb, head_need, tail_need, GFP_ATOMIC)) {
		printk(KERN_DEBUG "%s: failed to reallocate TX buffer\n",
		       wiphy_name(local->hw.wiphy));
		return -ENOMEM;
	}

	/* update truesize too */
	skb->truesize += head_need + tail_need;

	return 0;
}

1272 1273 1274
int ieee80211_master_start_xmit(struct sk_buff *skb,
				struct net_device *dev)
{
1275
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1276 1277 1278
	struct net_device *odev = NULL;
	struct ieee80211_sub_if_data *osdata;
	int headroom;
1279
	bool may_encrypt;
1280 1281
	int ret;

1282 1283
	if (info->control.ifindex)
		odev = dev_get_by_index(&init_net, info->control.ifindex);
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295
	if (unlikely(odev && !is_ieee80211_device(odev, dev))) {
		dev_put(odev);
		odev = NULL;
	}
	if (unlikely(!odev)) {
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
		printk(KERN_DEBUG "%s: Discarded packet with nonexistent "
		       "originating device\n", dev->name);
#endif
		dev_kfree_skb(skb);
		return 0;
	}
1296

1297 1298
	osdata = IEEE80211_DEV_TO_SUB_IF(odev);

1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310
	may_encrypt = !(info->flags & IEEE80211_TX_CTL_DO_NOT_ENCRYPT);

	headroom = osdata->local->tx_headroom;
	if (may_encrypt)
		headroom += IEEE80211_ENCRYPT_HEADROOM;
	headroom -= skb_headroom(skb);
	headroom = max_t(int, 0, headroom);

	if (ieee80211_skb_resize(osdata->local, skb, headroom, may_encrypt)) {
		dev_kfree_skb(skb);
		dev_put(odev);
		return 0;
1311 1312
	}

1313 1314
	info->control.vif = &osdata->vif;
	ret = ieee80211_tx(odev, skb);
1315 1316 1317 1318 1319 1320 1321 1322 1323
	dev_put(odev);

	return ret;
}

int ieee80211_monitor_start_xmit(struct sk_buff *skb,
				 struct net_device *dev)
{
	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
1324
	struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1325 1326
	struct ieee80211_radiotap_header *prthdr =
		(struct ieee80211_radiotap_header *)skb->data;
1327
	u16 len_rthdr;
1328

1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342
	/* check for not even having the fixed radiotap header part */
	if (unlikely(skb->len < sizeof(struct ieee80211_radiotap_header)))
		goto fail; /* too short to be possibly valid */

	/* is it a header version we can trust to find length from? */
	if (unlikely(prthdr->it_version))
		goto fail; /* only version 0 is supported */

	/* then there must be a radiotap header with a length we can use */
	len_rthdr = ieee80211_get_radiotap_len(skb->data);

	/* does the skb contain enough to deliver on the alleged length? */
	if (unlikely(skb->len < len_rthdr))
		goto fail; /* skb too short for claimed rt header extent */
1343 1344 1345

	skb->dev = local->mdev;

1346
	/* needed because we set skb device to master */
1347
	info->control.ifindex = dev->ifindex;
1348

1349
	info->flags |= IEEE80211_TX_CTL_DO_NOT_ENCRYPT;
1350
	/* Interfaces should always request a status report */
1351
	info->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
1352 1353 1354 1355 1356 1357 1358

	/*
	 * fix up the pointers accounting for the radiotap
	 * header still being in there.  We are being given
	 * a precooked IEEE80211 header so no need for
	 * normal processing
	 */
1359
	skb_set_mac_header(skb, len_rthdr);
1360
	/*
1361 1362
	 * these are just fixed to the end of the rt area since we
	 * don't have any better information and at this point, nobody cares
1363
	 */
1364 1365
	skb_set_network_header(skb, len_rthdr);
	skb_set_transport_header(skb, len_rthdr);
1366

1367 1368
	/* pass the radiotap header up to the next stage intact */
	dev_queue_xmit(skb);
1369
	return NETDEV_TX_OK;
1370 1371 1372 1373

fail:
	dev_kfree_skb(skb);
	return NETDEV_TX_OK; /* meaning, we dealt with the skb */
1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
}

/**
 * ieee80211_subif_start_xmit - netif start_xmit function for Ethernet-type
 * subinterfaces (wlan#, WDS, and VLAN interfaces)
 * @skb: packet to be sent
 * @dev: incoming interface
 *
 * Returns: 0 on success (and frees skb in this case) or 1 on failure (skb will
 * not be freed, and caller is responsible for either retrying later or freeing
 * skb).
 *
 * This function takes in an Ethernet header and encapsulates it with suitable
 * IEEE 802.11 header based on which interface the packet is coming in. The
 * encapsulated packet will then be passed to master interface, wlan#.11, for
 * transmission (through low-level driver).
 */
int ieee80211_subif_start_xmit(struct sk_buff *skb,
			       struct net_device *dev)
{
	struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
1395
	struct ieee80211_tx_info *info;
1396 1397
	struct ieee80211_sub_if_data *sdata;
	int ret = 1, head_need;
1398
	u16 ethertype, hdrlen,  meshhdrlen = 0, fc;
1399
	struct ieee80211_hdr hdr;
1400
	struct ieee80211s_hdr mesh_hdr;
1401 1402
	const u8 *encaps_data;
	int encaps_len, skip_header_bytes;
1403
	int nh_pos, h_pos;
1404
	struct sta_info *sta;
1405
	u32 sta_flags = 0;
1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422

	sdata = IEEE80211_DEV_TO_SUB_IF(dev);
	if (unlikely(skb->len < ETH_HLEN)) {
		printk(KERN_DEBUG "%s: short skb (len=%d)\n",
		       dev->name, skb->len);
		ret = 0;
		goto fail;
	}

	nh_pos = skb_network_header(skb) - skb->data;
	h_pos = skb_transport_header(skb) - skb->data;

	/* convert Ethernet header to proper 802.11 header (based on
	 * operation mode) */
	ethertype = (skb->data[12] << 8) | skb->data[13];
	fc = IEEE80211_FTYPE_DATA | IEEE80211_STYPE_DATA;

1423
	switch (sdata->vif.type) {
1424 1425
	case IEEE80211_IF_TYPE_AP:
	case IEEE80211_IF_TYPE_VLAN:
1426 1427 1428 1429 1430 1431
		fc |= IEEE80211_FCTL_FROMDS;
		/* DA BSSID SA */
		memcpy(hdr.addr1, skb->data, ETH_ALEN);
		memcpy(hdr.addr2, dev->dev_addr, ETH_ALEN);
		memcpy(hdr.addr3, skb->data + ETH_ALEN, ETH_ALEN);
		hdrlen = 24;
1432 1433
		break;
	case IEEE80211_IF_TYPE_WDS:
1434 1435 1436 1437 1438 1439 1440
		fc |= IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS;
		/* RA TA DA SA */
		memcpy(hdr.addr1, sdata->u.wds.remote_addr, ETH_ALEN);
		memcpy(hdr.addr2, dev->dev_addr, ETH_ALEN);
		memcpy(hdr.addr3, skb->data, ETH_ALEN);
		memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
		hdrlen = 30;
1441
		break;
1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
#ifdef CONFIG_MAC80211_MESH
	case IEEE80211_IF_TYPE_MESH_POINT:
		fc |= IEEE80211_FCTL_FROMDS | IEEE80211_FCTL_TODS;
		/* RA TA DA SA */
		if (is_multicast_ether_addr(skb->data))
			memcpy(hdr.addr1, skb->data, ETH_ALEN);
		else if (mesh_nexthop_lookup(hdr.addr1, skb, dev))
				return 0;
		memcpy(hdr.addr2, dev->dev_addr, ETH_ALEN);
		memcpy(hdr.addr3, skb->data, ETH_ALEN);
		memcpy(hdr.addr4, skb->data + ETH_ALEN, ETH_ALEN);
		if (skb->pkt_type == PACKET_OTHERHOST) {
			/* Forwarded frame, keep mesh ttl and seqnum */
			struct ieee80211s_hdr *prev_meshhdr;
			prev_meshhdr = ((struct ieee80211s_hdr *)skb->cb);
			meshhdrlen = ieee80211_get_mesh_hdrlen(prev_meshhdr);
			memcpy(&mesh_hdr, prev_meshhdr, meshhdrlen);
			sdata->u.sta.mshstats.fwded_frames++;
		} else {
			if (!sdata->u.sta.mshcfg.dot11MeshTTL) {
				/* Do not send frames with mesh_ttl == 0 */
				sdata->u.sta.mshstats.dropped_frames_ttl++;
				ret = 0;
				goto fail;
			}
			meshhdrlen = ieee80211_new_mesh_header(&mesh_hdr,
J
Johannes Berg 已提交
1468
							       sdata);
1469 1470 1471 1472
		}
		hdrlen = 30;
		break;
#endif
1473
	case IEEE80211_IF_TYPE_STA:
1474 1475 1476 1477 1478 1479
		fc |= IEEE80211_FCTL_TODS;
		/* BSSID SA DA */
		memcpy(hdr.addr1, sdata->u.sta.bssid, ETH_ALEN);
		memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
		memcpy(hdr.addr3, skb->data, ETH_ALEN);
		hdrlen = 24;
1480 1481
		break;
	case IEEE80211_IF_TYPE_IBSS:
1482 1483 1484 1485 1486
		/* DA SA BSSID */
		memcpy(hdr.addr1, skb->data, ETH_ALEN);
		memcpy(hdr.addr2, skb->data + ETH_ALEN, ETH_ALEN);
		memcpy(hdr.addr3, sdata->u.sta.bssid, ETH_ALEN);
		hdrlen = 24;
1487 1488
		break;
	default:
1489 1490 1491 1492
		ret = 0;
		goto fail;
	}

1493 1494 1495 1496 1497 1498
	/*
	 * There's no need to try to look up the destination
	 * if it is a multicast address (which can only happen
	 * in AP mode)
	 */
	if (!is_multicast_ether_addr(hdr.addr1)) {
1499
		rcu_read_lock();
1500
		sta = sta_info_get(local, hdr.addr1);
1501
		if (sta)
1502
			sta_flags = get_sta_flags(sta);
1503
		rcu_read_unlock();
1504 1505
	}

1506
	/* receiver and we are QoS enabled, use a QoS type frame */
1507 1508
	if (sta_flags & WLAN_STA_WME &&
	    ieee80211_num_regular_queues(&local->hw) >= 4) {
1509 1510 1511 1512 1513
		fc |= IEEE80211_STYPE_QOS_DATA;
		hdrlen += 2;
	}

	/*
1514 1515
	 * Drop unicast frames to unauthorised stations unless they are
	 * EAPOL frames from the local station.
1516
	 */
1517
	if (unlikely(!is_multicast_ether_addr(hdr.addr1) &&
1518 1519
		      !(sta_flags & WLAN_STA_AUTHORIZED) &&
		      !(ethertype == ETH_P_PAE &&
1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536
		       compare_ether_addr(dev->dev_addr,
					  skb->data + ETH_ALEN) == 0))) {
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
		DECLARE_MAC_BUF(mac);

		if (net_ratelimit())
			printk(KERN_DEBUG "%s: dropped frame to %s"
			       " (unauthorized port)\n", dev->name,
			       print_mac(mac, hdr.addr1));
#endif

		I802_DEBUG_INC(local->tx_handlers_drop_unauth_port);

		ret = 0;
		goto fail;
	}

1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571
	hdr.frame_control = cpu_to_le16(fc);
	hdr.duration_id = 0;
	hdr.seq_ctrl = 0;

	skip_header_bytes = ETH_HLEN;
	if (ethertype == ETH_P_AARP || ethertype == ETH_P_IPX) {
		encaps_data = bridge_tunnel_header;
		encaps_len = sizeof(bridge_tunnel_header);
		skip_header_bytes -= 2;
	} else if (ethertype >= 0x600) {
		encaps_data = rfc1042_header;
		encaps_len = sizeof(rfc1042_header);
		skip_header_bytes -= 2;
	} else {
		encaps_data = NULL;
		encaps_len = 0;
	}

	skb_pull(skb, skip_header_bytes);
	nh_pos -= skip_header_bytes;
	h_pos -= skip_header_bytes;

	/* TODO: implement support for fragments so that there is no need to
	 * reallocate and copy payload; it might be enough to support one
	 * extra fragment that would be copied in the beginning of the frame
	 * data.. anyway, it would be nice to include this into skb structure
	 * somehow
	 *
	 * There are few options for this:
	 * use skb->cb as an extra space for 802.11 header
	 * allocate new buffer if not enough headroom
	 * make sure that there is enough headroom in every skb by increasing
	 * build in headroom in __dev_alloc_skb() (linux/skbuff.h) and
	 * alloc_skb() (net/core/skbuff.c)
	 */
1572
	head_need = hdrlen + encaps_len + meshhdrlen - skb_headroom(skb);
1573

1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
	/*
	 * So we need to modify the skb header and hence need a copy of
	 * that. The head_need variable above doesn't, so far, include
	 * the needed header space that we don't need right away. If we
	 * can, then we don't reallocate right now but only after the
	 * frame arrives at the master device (if it does...)
	 *
	 * If we cannot, however, then we will reallocate to include all
	 * the ever needed space. Also, if we need to reallocate it anyway,
	 * make it big enough for everything we may ever need.
	 */
1585

1586
	if (head_need > 0 || skb_cloned(skb)) {
1587 1588 1589 1590
		head_need += IEEE80211_ENCRYPT_HEADROOM;
		head_need += local->tx_headroom;
		head_need = max_t(int, 0, head_need);
		if (ieee80211_skb_resize(local, skb, head_need, true))
1591 1592 1593 1594 1595 1596 1597 1598
			goto fail;
	}

	if (encaps_data) {
		memcpy(skb_push(skb, encaps_len), encaps_data, encaps_len);
		nh_pos += encaps_len;
		h_pos += encaps_len;
	}
1599

1600 1601 1602 1603 1604 1605
	if (meshhdrlen > 0) {
		memcpy(skb_push(skb, meshhdrlen), &mesh_hdr, meshhdrlen);
		nh_pos += meshhdrlen;
		h_pos += meshhdrlen;
	}

1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
	if (fc & IEEE80211_STYPE_QOS_DATA) {
		__le16 *qos_control;

		qos_control = (__le16*) skb_push(skb, 2);
		memcpy(skb_push(skb, hdrlen - 2), &hdr, hdrlen - 2);
		/*
		 * Maybe we could actually set some fields here, for now just
		 * initialise to zero to indicate no special operation.
		 */
		*qos_control = 0;
	} else
		memcpy(skb_push(skb, hdrlen), &hdr, hdrlen);

1619 1620 1621
	nh_pos += hdrlen;
	h_pos += hdrlen;

1622 1623 1624
	info = IEEE80211_SKB_CB(skb);
	memset(info, 0, sizeof(*info));
	info->control.ifindex = dev->ifindex;
1625
	if (ethertype == ETH_P_PAE)
1626
		info->flags |= IEEE80211_TX_CTL_EAPOL_FRAME;
1627

1628
	/* Interfaces should always request a status report */
1629
	info->flags |= IEEE80211_TX_CTL_REQ_TX_STATUS;
1630

1631
	skb->dev = local->mdev;
1632 1633
	dev->stats.tx_packets++;
	dev->stats.tx_bytes += skb->len;
1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654

	/* Update skb pointers to various headers since this modified frame
	 * is going to go through Linux networking code that may potentially
	 * need things like pointer to IP header. */
	skb_set_mac_header(skb, 0);
	skb_set_network_header(skb, nh_pos);
	skb_set_transport_header(skb, h_pos);

	dev->trans_start = jiffies;
	dev_queue_xmit(skb);

	return 0;

 fail:
	if (!ret)
		dev_kfree_skb(skb);

	return ret;
}


1655 1656 1657 1658
/*
 * ieee80211_clear_tx_pending may not be called in a context where
 * it is possible that it packets could come in again.
 */
1659 1660 1661 1662 1663
void ieee80211_clear_tx_pending(struct ieee80211_local *local)
{
	int i, j;
	struct ieee80211_tx_stored_packet *store;

1664 1665
	for (i = 0; i < ieee80211_num_regular_queues(&local->hw); i++) {
		if (!test_bit(i, local->queues_pending))
1666 1667 1668 1669 1670 1671
			continue;
		store = &local->pending_packet[i];
		kfree_skb(store->skb);
		for (j = 0; j < store->num_extra_frag; j++)
			kfree_skb(store->extra_frag[j]);
		kfree(store->extra_frag);
1672
		clear_bit(i, local->queues_pending);
1673 1674 1675
	}
}

1676 1677 1678 1679
/*
 * Transmit all pending packets. Called from tasklet, locks master device
 * TX lock so that no new packets can come in.
 */
1680 1681 1682 1683 1684
void ieee80211_tx_pending(unsigned long data)
{
	struct ieee80211_local *local = (struct ieee80211_local *)data;
	struct net_device *dev = local->mdev;
	struct ieee80211_tx_stored_packet *store;
1685
	struct ieee80211_tx_data tx;
1686
	int i, ret;
1687 1688

	netif_tx_lock_bh(dev);
1689 1690 1691
	for (i = 0; i < ieee80211_num_regular_queues(&local->hw); i++) {
		/* Check that this queue is ok */
		if (__netif_subqueue_stopped(local->mdev, i))
1692
			continue;
1693 1694 1695

		if (!test_bit(i, local->queues_pending)) {
			ieee80211_wake_queue(&local->hw, i);
1696 1697
			continue;
		}
1698

1699
		store = &local->pending_packet[i];
1700 1701
		tx.extra_frag = store->extra_frag;
		tx.num_extra_frag = store->num_extra_frag;
1702
		tx.last_frag_rate_idx = store->last_frag_rate_idx;
1703 1704
		tx.flags = 0;
		if (store->last_frag_rate_ctrl_probe)
1705
			tx.flags |= IEEE80211_TX_PROBE_LAST_FRAG;
1706 1707 1708 1709 1710
		ret = __ieee80211_tx(local, store->skb, &tx);
		if (ret) {
			if (ret == IEEE80211_TX_FRAG_AGAIN)
				store->skb = NULL;
		} else {
1711 1712
			clear_bit(i, local->queues_pending);
			ieee80211_wake_queue(&local->hw, i);
1713 1714 1715 1716 1717 1718 1719 1720 1721
		}
	}
	netif_tx_unlock_bh(dev);
}

/* functions for drivers to get certain frames */

static void ieee80211_beacon_add_tim(struct ieee80211_local *local,
				     struct ieee80211_if_ap *bss,
1722 1723
				     struct sk_buff *skb,
				     struct beacon_data *beacon)
1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737
{
	u8 *pos, *tim;
	int aid0 = 0;
	int i, have_bits = 0, n1, n2;

	/* Generate bitmap for TIM only if there are any STAs in power save
	 * mode. */
	if (atomic_read(&bss->num_sta_ps) > 0)
		/* in the hope that this is faster than
		 * checking byte-for-byte */
		have_bits = !bitmap_empty((unsigned long*)bss->tim,
					  IEEE80211_MAX_AID+1);

	if (bss->dtim_count == 0)
1738
		bss->dtim_count = beacon->dtim_period - 1;
1739 1740 1741 1742 1743 1744 1745
	else
		bss->dtim_count--;

	tim = pos = (u8 *) skb_put(skb, 6);
	*pos++ = WLAN_EID_TIM;
	*pos++ = 4;
	*pos++ = bss->dtim_count;
1746
	*pos++ = beacon->dtim_period;
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782

	if (bss->dtim_count == 0 && !skb_queue_empty(&bss->ps_bc_buf))
		aid0 = 1;

	if (have_bits) {
		/* Find largest even number N1 so that bits numbered 1 through
		 * (N1 x 8) - 1 in the bitmap are 0 and number N2 so that bits
		 * (N2 + 1) x 8 through 2007 are 0. */
		n1 = 0;
		for (i = 0; i < IEEE80211_MAX_TIM_LEN; i++) {
			if (bss->tim[i]) {
				n1 = i & 0xfe;
				break;
			}
		}
		n2 = n1;
		for (i = IEEE80211_MAX_TIM_LEN - 1; i >= n1; i--) {
			if (bss->tim[i]) {
				n2 = i;
				break;
			}
		}

		/* Bitmap control */
		*pos++ = n1 | aid0;
		/* Part Virt Bitmap */
		memcpy(pos, bss->tim + n1, n2 - n1 + 1);

		tim[1] = n2 - n1 + 4;
		skb_put(skb, n2 - n1);
	} else {
		*pos++ = aid0; /* Bitmap control */
		*pos++ = 0; /* Part Virt Bitmap */
	}
}

1783
struct sk_buff *ieee80211_beacon_get(struct ieee80211_hw *hw,
1784
				     struct ieee80211_vif *vif)
1785 1786 1787
{
	struct ieee80211_local *local = hw_to_local(hw);
	struct sk_buff *skb;
1788
	struct ieee80211_tx_info *info;
1789 1790 1791
	struct net_device *bdev;
	struct ieee80211_sub_if_data *sdata = NULL;
	struct ieee80211_if_ap *ap = NULL;
1792
	struct rate_selection rsel;
1793
	struct beacon_data *beacon;
1794
	struct ieee80211_supported_band *sband;
J
Johannes Berg 已提交
1795
	struct ieee80211_mgmt *mgmt;
1796
	int *num_beacons;
J
Johannes Berg 已提交
1797
	bool err = true;
1798
	enum ieee80211_band band = local->hw.conf.channel->band;
J
Johannes Berg 已提交
1799
	u8 *pos;
1800

1801
	sband = local->hw.wiphy->bands[band];
1802 1803

	rcu_read_lock();
1804

1805 1806
	sdata = vif_to_sdata(vif);
	bdev = sdata->dev;
1807

J
Johannes Berg 已提交
1808
	if (sdata->vif.type == IEEE80211_IF_TYPE_AP) {
1809 1810
		ap = &sdata->u.ap;
		beacon = rcu_dereference(ap->beacon);
J
Johannes Berg 已提交
1811 1812 1813 1814 1815 1816 1817 1818 1819 1820
		if (ap && beacon) {
			/*
			 * headroom, head length,
			 * tail length and maximum TIM length
			 */
			skb = dev_alloc_skb(local->tx_headroom +
					    beacon->head_len +
					    beacon->tail_len + 256);
			if (!skb)
				goto out;
1821

J
Johannes Berg 已提交
1822 1823 1824
			skb_reserve(skb, local->tx_headroom);
			memcpy(skb_put(skb, beacon->head_len), beacon->head,
			       beacon->head_len);
1825

J
Johannes Berg 已提交
1826 1827
			ieee80211_include_sequence(sdata,
					(struct ieee80211_hdr *)skb->data);
1828

1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844
			/*
			 * Not very nice, but we want to allow the driver to call
			 * ieee80211_beacon_get() as a response to the set_tim()
			 * callback. That, however, is already invoked under the
			 * sta_lock to guarantee consistent and race-free update
			 * of the tim bitmap in mac80211 and the driver.
			 */
			if (local->tim_in_locked_section) {
				ieee80211_beacon_add_tim(local, ap, skb, beacon);
			} else {
				unsigned long flags;

				spin_lock_irqsave(&local->sta_lock, flags);
				ieee80211_beacon_add_tim(local, ap, skb, beacon);
				spin_unlock_irqrestore(&local->sta_lock, flags);
			}
1845

J
Johannes Berg 已提交
1846 1847 1848
			if (beacon->tail)
				memcpy(skb_put(skb, beacon->tail_len),
				       beacon->tail, beacon->tail_len);
1849

J
Johannes Berg 已提交
1850
			num_beacons = &ap->num_beacons;
1851

J
Johannes Berg 已提交
1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
			err = false;
		}
	} else if (ieee80211_vif_is_mesh(&sdata->vif)) {
		/* headroom, head length, tail length and maximum TIM length */
		skb = dev_alloc_skb(local->tx_headroom + 400);
		if (!skb)
			goto out;

		skb_reserve(skb, local->hw.extra_tx_headroom);
		mgmt = (struct ieee80211_mgmt *)
			skb_put(skb, 24 + sizeof(mgmt->u.beacon));
		memset(mgmt, 0, 24 + sizeof(mgmt->u.beacon));
		mgmt->frame_control = IEEE80211_FC(IEEE80211_FTYPE_MGMT,
						   IEEE80211_STYPE_BEACON);
		memset(mgmt->da, 0xff, ETH_ALEN);
		memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
		/* BSSID is left zeroed, wildcard value */
		mgmt->u.beacon.beacon_int =
			cpu_to_le16(local->hw.conf.beacon_int);
		mgmt->u.beacon.capab_info = 0x0; /* 0x0 for MPs */

		pos = skb_put(skb, 2);
		*pos++ = WLAN_EID_SSID;
		*pos++ = 0x0;

		mesh_mgmt_ies_add(skb, sdata->dev);
1878 1879 1880

		num_beacons = &sdata->u.sta.num_beacons;

J
Johannes Berg 已提交
1881
		err = false;
1882
	}
1883

1884
	if (err) {
1885 1886
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
		if (net_ratelimit())
1887 1888
			printk(KERN_DEBUG "no beacon data avail for %s\n",
			       bdev->name);
1889
#endif /* CONFIG_MAC80211_VERBOSE_DEBUG */
1890 1891
		skb = NULL;
		goto out;
1892 1893
	}

1894 1895 1896 1897
	info = IEEE80211_SKB_CB(skb);

	info->band = band;
	rate_control_get_rate(local->mdev, sband, skb, &rsel);
1898

1899 1900 1901 1902 1903 1904 1905 1906 1907
	if (unlikely(rsel.rate_idx < 0)) {
		if (net_ratelimit()) {
			printk(KERN_DEBUG "%s: ieee80211_beacon_get: "
			       "no rate found\n",
			       wiphy_name(local->hw.wiphy));
		}
		dev_kfree_skb(skb);
		skb = NULL;
		goto out;
1908
	}
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919

	info->control.vif = vif;
	info->tx_rate_idx = rsel.rate_idx;
	if (sdata->bss_conf.use_short_preamble &&
	    sband->bitrates[rsel.rate_idx].flags & IEEE80211_RATE_SHORT_PREAMBLE)
		info->flags |= IEEE80211_TX_CTL_SHORT_PREAMBLE;
	info->antenna_sel_tx = local->hw.conf.antenna_sel_tx;
	info->flags |= IEEE80211_TX_CTL_NO_ACK;
	info->flags |= IEEE80211_TX_CTL_DO_NOT_ENCRYPT;
	info->control.retry_limit = 1;
	info->flags |= IEEE80211_TX_CTL_CLEAR_PS_FILT;
1920 1921
	(*num_beacons)++;
out:
1922
	rcu_read_unlock();
1923 1924 1925 1926
	return skb;
}
EXPORT_SYMBOL(ieee80211_beacon_get);

1927
void ieee80211_rts_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1928
		       const void *frame, size_t frame_len,
1929
		       const struct ieee80211_tx_info *frame_txctl,
1930 1931 1932 1933 1934 1935 1936
		       struct ieee80211_rts *rts)
{
	const struct ieee80211_hdr *hdr = frame;
	u16 fctl;

	fctl = IEEE80211_FTYPE_CTL | IEEE80211_STYPE_RTS;
	rts->frame_control = cpu_to_le16(fctl);
1937 1938
	rts->duration = ieee80211_rts_duration(hw, vif, frame_len,
					       frame_txctl);
1939 1940 1941 1942 1943
	memcpy(rts->ra, hdr->addr1, sizeof(rts->ra));
	memcpy(rts->ta, hdr->addr2, sizeof(rts->ta));
}
EXPORT_SYMBOL(ieee80211_rts_get);

1944
void ieee80211_ctstoself_get(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
1945
			     const void *frame, size_t frame_len,
1946
			     const struct ieee80211_tx_info *frame_txctl,
1947 1948 1949 1950 1951 1952 1953
			     struct ieee80211_cts *cts)
{
	const struct ieee80211_hdr *hdr = frame;
	u16 fctl;

	fctl = IEEE80211_FTYPE_CTL | IEEE80211_STYPE_CTS;
	cts->frame_control = cpu_to_le16(fctl);
1954 1955
	cts->duration = ieee80211_ctstoself_duration(hw, vif,
						     frame_len, frame_txctl);
1956 1957 1958 1959 1960
	memcpy(cts->ra, hdr->addr1, sizeof(cts->ra));
}
EXPORT_SYMBOL(ieee80211_ctstoself_get);

struct sk_buff *
1961
ieee80211_get_buffered_bc(struct ieee80211_hw *hw,
1962
			  struct ieee80211_vif *vif)
1963 1964
{
	struct ieee80211_local *local = hw_to_local(hw);
1965
	struct sk_buff *skb = NULL;
1966
	struct sta_info *sta;
1967
	struct ieee80211_tx_data tx;
1968 1969 1970
	struct net_device *bdev;
	struct ieee80211_sub_if_data *sdata;
	struct ieee80211_if_ap *bss = NULL;
1971
	struct beacon_data *beacon;
1972
	struct ieee80211_tx_info *info;
1973

1974 1975
	sdata = vif_to_sdata(vif);
	bdev = sdata->dev;
1976
	bss = &sdata->u.ap;
1977 1978

	if (!bss)
1979 1980
		return NULL;

1981 1982 1983
	rcu_read_lock();
	beacon = rcu_dereference(bss->beacon);

1984 1985
	if (sdata->vif.type != IEEE80211_IF_TYPE_AP || !beacon || !beacon->head)
		goto out;
1986

1987
	if (bss->dtim_count != 0)
1988
		goto out; /* send buffered bc/mc only after DTIM beacon */
1989

1990 1991 1992
	while (1) {
		skb = skb_dequeue(&bss->ps_bc_buf);
		if (!skb)
1993
			goto out;
1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005
		local->total_ps_buffered--;

		if (!skb_queue_empty(&bss->ps_bc_buf) && skb->len >= 2) {
			struct ieee80211_hdr *hdr =
				(struct ieee80211_hdr *) skb->data;
			/* more buffered multicast/broadcast frames ==> set
			 * MoreData flag in IEEE 802.11 header to inform PS
			 * STAs */
			hdr->frame_control |=
				cpu_to_le16(IEEE80211_FCTL_MOREDATA);
		}

2006
		if (!ieee80211_tx_prepare(&tx, skb, local->mdev))
2007 2008 2009
			break;
		dev_kfree_skb_any(skb);
	}
2010 2011 2012

	info = IEEE80211_SKB_CB(skb);

2013
	sta = tx.sta;
2014 2015
	tx.flags |= IEEE80211_TX_PS_BUFFERED;
	tx.channel = local->hw.conf.channel;
2016
	info->band = tx.channel->band;
2017

2018
	if (invoke_tx_handlers(&tx))
2019
		skb = NULL;
2020
 out:
2021
	rcu_read_unlock();
2022 2023 2024 2025

	return skb;
}
EXPORT_SYMBOL(ieee80211_get_buffered_bc);