mlme.c 120 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * BSS client mode implementation
 * Copyright 2003, Jouni Malinen <jkmaline@cc.hut.fi>
 * Copyright 2004, Instant802 Networks, Inc.
 * Copyright 2005, Devicescape Software, Inc.
 * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
 * Copyright 2007, Michael Wu <flamingice@sourmilk.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.
 */

/* TODO:
 * order BSS list by RSSI(?) ("quality of AP")
 * scan result table filtering (by capability (privacy, IBSS/BSS, WPA/RSN IE,
 *    SSID)
 */
19
#include <linux/delay.h>
20 21 22 23 24 25 26
#include <linux/if_ether.h>
#include <linux/skbuff.h>
#include <linux/netdevice.h>
#include <linux/if_arp.h>
#include <linux/wireless.h>
#include <linux/random.h>
#include <linux/etherdevice.h>
27
#include <linux/rtnetlink.h>
28 29 30 31 32
#include <net/iw_handler.h>
#include <asm/types.h>

#include <net/mac80211.h>
#include "ieee80211_i.h"
Johannes Berg's avatar
Johannes Berg committed
33 34
#include "rate.h"
#include "led.h"
35
#include "mesh.h"
36

37
#define IEEE80211_ASSOC_SCANS_MAX_TRIES 2
38 39 40 41 42
#define IEEE80211_AUTH_TIMEOUT (HZ / 5)
#define IEEE80211_AUTH_MAX_TRIES 3
#define IEEE80211_ASSOC_TIMEOUT (HZ / 5)
#define IEEE80211_ASSOC_MAX_TRIES 3
#define IEEE80211_MONITORING_INTERVAL (2 * HZ)
43
#define IEEE80211_MESH_HOUSEKEEPING_INTERVAL (60 * HZ)
44 45 46 47
#define IEEE80211_PROBE_INTERVAL (60 * HZ)
#define IEEE80211_RETRY_AUTH_INTERVAL (1 * HZ)
#define IEEE80211_SCAN_INTERVAL (2 * HZ)
#define IEEE80211_SCAN_INTERVAL_SLOW (15 * HZ)
48
#define IEEE80211_IBSS_JOIN_TIMEOUT (7 * HZ)
49 50 51 52 53 54 55

#define IEEE80211_PROBE_DELAY (HZ / 33)
#define IEEE80211_CHANNEL_TIME (HZ / 33)
#define IEEE80211_PASSIVE_CHANNEL_TIME (HZ / 5)
#define IEEE80211_SCAN_RESULT_EXPIRE (10 * HZ)
#define IEEE80211_IBSS_MERGE_INTERVAL (30 * HZ)
#define IEEE80211_IBSS_INACTIVITY_LIMIT (60 * HZ)
56
#define IEEE80211_MESH_PEER_INACTIVITY_LIMIT (1800 * HZ)
57 58 59 60 61 62

#define IEEE80211_IBSS_MAX_STA_ENTRIES 128


#define ERP_INFO_USE_PROTECTION BIT(1)

63
/* mgmt header + 1 byte category code */
64 65 66 67 68
#define IEEE80211_MIN_ACTION_SIZE (24 + 1)

#define IEEE80211_ADDBA_PARAM_POLICY_MASK 0x0002
#define IEEE80211_ADDBA_PARAM_TID_MASK 0x003C
#define IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK 0xFFA0
69 70
#define IEEE80211_DELBA_PARAM_TID_MASK 0xF000
#define IEEE80211_DELBA_PARAM_INITIATOR_MASK 0x0800
71

72 73 74 75 76
/* next values represent the buffer size for A-MPDU frame.
 * According to IEEE802.11n spec size varies from 8K to 64K (in powers of 2) */
#define IEEE80211_MIN_AMPDU_BUF 0x8
#define IEEE80211_MAX_AMPDU_BUF 0x40

77
static void ieee80211_send_probe_req(struct ieee80211_sub_if_data *sdata, u8 *dst,
78 79
				     u8 *ssid, size_t ssid_len);
static struct ieee80211_sta_bss *
80
ieee80211_rx_bss_get(struct ieee80211_local *local, u8 *bssid, int freq,
81
		     u8 *ssid, u8 ssid_len);
82
static void ieee80211_rx_bss_put(struct ieee80211_local *local,
83
				 struct ieee80211_sta_bss *bss);
84
static int ieee80211_sta_find_ibss(struct ieee80211_sub_if_data *sdata,
85
				   struct ieee80211_if_sta *ifsta);
86 87
static int ieee80211_sta_wep_configured(struct ieee80211_sub_if_data *sdata);
static int ieee80211_sta_start_scan(struct ieee80211_sub_if_data *sdata,
88
				    u8 *ssid, size_t ssid_len);
89
static int ieee80211_sta_config_auth(struct ieee80211_sub_if_data *sdata,
90
				     struct ieee80211_if_sta *ifsta);
91
static void sta_rx_agg_session_timer_expired(unsigned long data);
92 93


94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
static u8 * ieee80211_bss_get_ie(struct ieee80211_sta_bss *bss, u8 ie)
{
	u8 *end, *pos;

	pos = bss->ies;
	if (pos == NULL)
		return NULL;
	end = pos + bss->ies_len;

	while (pos + 1 < end) {
		if (pos + 2 + pos[1] > end)
			break;
		if (pos[0] == ie)
			return pos;
		pos += 2 + pos[1];
	}

	return NULL;
}


115 116
static int ecw2cw(int ecw)
{
117
	return (1 << ecw) - 1;
118 119
}

120

121
static void ieee80211_sta_def_wmm_params(struct ieee80211_sub_if_data *sdata,
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
					 struct ieee80211_sta_bss *bss,
					 int ibss)
{
	struct ieee80211_local *local = sdata->local;
	int i, have_higher_than_11mbit = 0;


	/* cf. IEEE 802.11 9.2.12 */
	for (i = 0; i < bss->supp_rates_len; i++)
		if ((bss->supp_rates[i] & 0x7f) * 5 > 110)
			have_higher_than_11mbit = 1;

	if (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ &&
	    have_higher_than_11mbit)
		sdata->flags |= IEEE80211_SDATA_OPERATING_GMODE;
	else
		sdata->flags &= ~IEEE80211_SDATA_OPERATING_GMODE;


	if (local->ops->conf_tx) {
		struct ieee80211_tx_queue_params qparam;

		memset(&qparam, 0, sizeof(qparam));

		qparam.aifs = 2;

		if (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ &&
		    !(sdata->flags & IEEE80211_SDATA_OPERATING_GMODE))
			qparam.cw_min = 31;
		else
			qparam.cw_min = 15;

		qparam.cw_max = 1023;
		qparam.txop = 0;

Johannes Berg's avatar
Johannes Berg committed
157 158
		for (i = 0; i < local_to_hw(local)->queues; i++)
			local->ops->conf_tx(local_to_hw(local), i, &qparam);
159 160 161
	}
}

162
static void ieee80211_sta_wmm_params(struct ieee80211_local *local,
163 164 165 166 167 168 169 170
				     struct ieee80211_if_sta *ifsta,
				     u8 *wmm_param, size_t wmm_param_len)
{
	struct ieee80211_tx_queue_params params;
	size_t left;
	int count;
	u8 *pos;

171 172 173 174 175 176
	if (!(ifsta->flags & IEEE80211_STA_WMM_ENABLED))
		return;

	if (!wmm_param)
		return;

177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
	if (wmm_param_len < 8 || wmm_param[5] /* version */ != 1)
		return;
	count = wmm_param[6] & 0x0f;
	if (count == ifsta->wmm_last_param_set)
		return;
	ifsta->wmm_last_param_set = count;

	pos = wmm_param + 8;
	left = wmm_param_len - 8;

	memset(&params, 0, sizeof(params));

	if (!local->ops->conf_tx)
		return;

	local->wmm_acm = 0;
	for (; left >= 4; left -= 4, pos += 4) {
		int aci = (pos[0] >> 5) & 0x03;
		int acm = (pos[0] >> 4) & 0x01;
		int queue;

		switch (aci) {
		case 1:
Johannes Berg's avatar
Johannes Berg committed
200
			queue = 3;
Johannes Berg's avatar
Johannes Berg committed
201
			if (acm)
202 203 204
				local->wmm_acm |= BIT(0) | BIT(3);
			break;
		case 2:
Johannes Berg's avatar
Johannes Berg committed
205
			queue = 1;
Johannes Berg's avatar
Johannes Berg committed
206
			if (acm)
207 208 209
				local->wmm_acm |= BIT(4) | BIT(5);
			break;
		case 3:
Johannes Berg's avatar
Johannes Berg committed
210
			queue = 0;
Johannes Berg's avatar
Johannes Berg committed
211
			if (acm)
212 213 214 215
				local->wmm_acm |= BIT(6) | BIT(7);
			break;
		case 0:
		default:
Johannes Berg's avatar
Johannes Berg committed
216
			queue = 2;
Johannes Berg's avatar
Johannes Berg committed
217
			if (acm)
218 219 220 221 222 223 224
				local->wmm_acm |= BIT(1) | BIT(2);
			break;
		}

		params.aifs = pos[0] & 0x0f;
		params.cw_max = ecw2cw((pos[1] & 0xf0) >> 4);
		params.cw_min = ecw2cw(pos[1] & 0x0f);
225
		params.txop = get_unaligned_le16(pos + 2);
226
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
227
		printk(KERN_DEBUG "%s: WMM queue=%d aci=%d acm=%d aifs=%d "
228
		       "cWmin=%d cWmax=%d txop=%d\n",
229
		       local->mdev->name, queue, aci, acm, params.aifs, params.cw_min,
230 231
		       params.cw_max, params.txop);
#endif
232 233 234 235
		/* TODO: handle ACM (block TX, fallback to next lowest allowed
		 * AC for now) */
		if (local->ops->conf_tx(local_to_hw(local), queue, &params)) {
			printk(KERN_DEBUG "%s: failed to set TX queue "
236
			       "parameters for queue %d\n", local->mdev->name, queue);
237 238 239 240
		}
	}
}

241 242 243
static u32 ieee80211_handle_protect_preamb(struct ieee80211_sub_if_data *sdata,
					   bool use_protection,
					   bool use_short_preamble)
244
{
245
	struct ieee80211_bss_conf *bss_conf = &sdata->bss_conf;
246
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
247
	struct ieee80211_if_sta *ifsta = &sdata->u.sta;
248
	DECLARE_MAC_BUF(mac);
249
#endif
250
	u32 changed = 0;
251

252
	if (use_protection != bss_conf->use_cts_prot) {
253
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
254 255
		if (net_ratelimit()) {
			printk(KERN_DEBUG "%s: CTS protection %s (BSSID="
256
			       "%s)\n",
257
			       sdata->dev->name,
258
			       use_protection ? "enabled" : "disabled",
259
			       print_mac(mac, ifsta->bssid));
260
		}
261
#endif
262 263
		bss_conf->use_cts_prot = use_protection;
		changed |= BSS_CHANGED_ERP_CTS_PROT;
264
	}
265

266
	if (use_short_preamble != bss_conf->use_short_preamble) {
267
#ifdef CONFIG_MAC80211_VERBOSE_DEBUG
268 269
		if (net_ratelimit()) {
			printk(KERN_DEBUG "%s: switched to %s barker preamble"
270
			       " (BSSID=%s)\n",
271
			       sdata->dev->name,
272
			       use_short_preamble ? "short" : "long",
273
			       print_mac(mac, ifsta->bssid));
274
		}
275
#endif
276
		bss_conf->use_short_preamble = use_short_preamble;
277
		changed |= BSS_CHANGED_ERP_PREAMBLE;
278
	}
279

280
	return changed;
281 282
}

283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
static u32 ieee80211_handle_erp_ie(struct ieee80211_sub_if_data *sdata,
				   u8 erp_value)
{
	bool use_protection = (erp_value & WLAN_ERP_USE_PROTECTION) != 0;
	bool use_short_preamble = (erp_value & WLAN_ERP_BARKER_PREAMBLE) == 0;

	return ieee80211_handle_protect_preamb(sdata,
			use_protection, use_short_preamble);
}

static u32 ieee80211_handle_bss_capability(struct ieee80211_sub_if_data *sdata,
					   struct ieee80211_sta_bss *bss)
{
	u32 changed = 0;

	if (bss->has_erp_value)
		changed |= ieee80211_handle_erp_ie(sdata, bss->erp_value);
	else {
		u16 capab = bss->capability;
		changed |= ieee80211_handle_protect_preamb(sdata, false,
				(capab & WLAN_CAPABILITY_SHORT_PREAMBLE) != 0);
	}

	return changed;
}

309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
int ieee80211_ht_cap_ie_to_ht_info(struct ieee80211_ht_cap *ht_cap_ie,
				   struct ieee80211_ht_info *ht_info)
{

	if (ht_info == NULL)
		return -EINVAL;

	memset(ht_info, 0, sizeof(*ht_info));

	if (ht_cap_ie) {
		u8 ampdu_info = ht_cap_ie->ampdu_params_info;

		ht_info->ht_supported = 1;
		ht_info->cap = le16_to_cpu(ht_cap_ie->cap_info);
		ht_info->ampdu_factor =
			ampdu_info & IEEE80211_HT_CAP_AMPDU_FACTOR;
		ht_info->ampdu_density =
			(ampdu_info & IEEE80211_HT_CAP_AMPDU_DENSITY) >> 2;
		memcpy(ht_info->supp_mcs_set, ht_cap_ie->supp_mcs_set, 16);
	} else
		ht_info->ht_supported = 0;

	return 0;
}

int ieee80211_ht_addt_info_ie_to_ht_bss_info(
			struct ieee80211_ht_addt_info *ht_add_info_ie,
			struct ieee80211_ht_bss_info *bss_info)
{
	if (bss_info == NULL)
		return -EINVAL;

	memset(bss_info, 0, sizeof(*bss_info));

	if (ht_add_info_ie) {
		u16 op_mode;
		op_mode = le16_to_cpu(ht_add_info_ie->operation_mode);

		bss_info->primary_channel = ht_add_info_ie->control_chan;
		bss_info->bss_cap = ht_add_info_ie->ht_param;
		bss_info->bss_op_mode = (u8)(op_mode & 0xff);
	}

	return 0;
}
354

355 356 357 358 359 360 361 362 363 364 365
static void ieee80211_sta_send_apinfo(struct ieee80211_sub_if_data *sdata,
					struct ieee80211_if_sta *ifsta)
{
	union iwreq_data wrqu;
	memset(&wrqu, 0, sizeof(wrqu));
	if (ifsta->flags & IEEE80211_STA_ASSOCIATED)
		memcpy(wrqu.ap_addr.sa_data, sdata->u.sta.bssid, ETH_ALEN);
	wrqu.ap_addr.sa_family = ARPHRD_ETHER;
	wireless_send_event(sdata->dev, SIOCGIWAP, &wrqu, NULL);
}

366
static void ieee80211_sta_send_associnfo(struct ieee80211_sub_if_data *sdata,
367 368 369 370 371
					 struct ieee80211_if_sta *ifsta)
{
	union iwreq_data wrqu;

	if (ifsta->assocreq_ies) {
372 373
		memset(&wrqu, 0, sizeof(wrqu));
		wrqu.data.length = ifsta->assocreq_ies_len;
374
		wireless_send_event(sdata->dev, IWEVASSOCREQIE, &wrqu,
375
				    ifsta->assocreq_ies);
376
	}
377 378 379
	if (ifsta->assocresp_ies) {
		memset(&wrqu, 0, sizeof(wrqu));
		wrqu.data.length = ifsta->assocresp_ies_len;
380
		wireless_send_event(sdata->dev, IWEVASSOCRESPIE, &wrqu,
381
				    ifsta->assocresp_ies);
382 383 384 385
	}
}


386
static void ieee80211_set_associated(struct ieee80211_sub_if_data *sdata,
387
				     struct ieee80211_if_sta *ifsta)
388
{
389
	struct ieee80211_local *local = sdata->local;
Tomas Winkler's avatar
Tomas Winkler committed
390
	struct ieee80211_conf *conf = &local_to_hw(local)->conf;
391
	u32 changed = BSS_CHANGED_ASSOC;
392

393
	struct ieee80211_sta_bss *bss;
394

395
	ifsta->flags |= IEEE80211_STA_ASSOCIATED;
396

397 398
	if (sdata->vif.type != IEEE80211_IF_TYPE_STA)
		return;
399

400 401 402 403 404 405 406 407
	bss = ieee80211_rx_bss_get(local, ifsta->bssid,
				   conf->channel->center_freq,
				   ifsta->ssid, ifsta->ssid_len);
	if (bss) {
		/* set timing information */
		sdata->bss_conf.beacon_int = bss->beacon_int;
		sdata->bss_conf.timestamp = bss->timestamp;
		sdata->bss_conf.dtim_period = bss->dtim_period;
408

409
		changed |= ieee80211_handle_bss_capability(sdata, bss);
410

411 412
		ieee80211_rx_bss_put(local, bss);
	}
Tomas Winkler's avatar
Tomas Winkler committed
413

414 415 416 417 418 419
	if (conf->flags & IEEE80211_CONF_SUPPORT_HT_MODE) {
		changed |= BSS_CHANGED_HT;
		sdata->bss_conf.assoc_ht = 1;
		sdata->bss_conf.ht_conf = &conf->ht_conf;
		sdata->bss_conf.ht_bss_conf = &conf->ht_bss_conf;
	}
Tomas Winkler's avatar
Tomas Winkler committed
420

421 422 423
	ifsta->flags |= IEEE80211_STA_PREV_BSSID_SET;
	memcpy(ifsta->prev_bssid, sdata->u.sta.bssid, ETH_ALEN);
	ieee80211_sta_send_associnfo(sdata, ifsta);
Tomas Winkler's avatar
Tomas Winkler committed
424

425
	ifsta->last_probe = jiffies;
426
	ieee80211_led_assoc(local, 1);
427

428
	sdata->bss_conf.assoc = 1;
429
	ieee80211_bss_info_change_notify(sdata, changed);
430

431
	netif_tx_start_all_queues(sdata->dev);
432
	netif_carrier_on(sdata->dev);
433

434
	ieee80211_sta_send_apinfo(sdata, ifsta);
435 436
}

437
void ieee80211_sta_tx(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb,
438
		      int encrypt)
439 440 441 442 443 444
{
	skb->dev = sdata->local->mdev;
	skb_set_mac_header(skb, 0);
	skb_set_network_header(skb, 0);
	skb_set_transport_header(skb, 0);

445 446
	skb->iif = sdata->dev->ifindex;
	skb->do_not_encrypt = !encrypt;
447 448 449 450 451

	dev_queue_xmit(skb);
}


452
static void ieee80211_send_auth(struct ieee80211_sub_if_data *sdata,
453 454 455 456
				struct ieee80211_if_sta *ifsta,
				int transaction, u8 *extra, size_t extra_len,
				int encrypt)
{
457
	struct ieee80211_local *local = sdata->local;
458 459 460 461 462 463 464
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;

	skb = dev_alloc_skb(local->hw.extra_tx_headroom +
			    sizeof(*mgmt) + 6 + extra_len);
	if (!skb) {
		printk(KERN_DEBUG "%s: failed to allocate buffer for auth "
465
		       "frame\n", sdata->dev->name);
466 467 468 469 470 471
		return;
	}
	skb_reserve(skb, local->hw.extra_tx_headroom);

	mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24 + 6);
	memset(mgmt, 0, 24 + 6);
472 473
	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
					  IEEE80211_STYPE_AUTH);
474 475 476
	if (encrypt)
		mgmt->frame_control |= cpu_to_le16(IEEE80211_FCTL_PROTECTED);
	memcpy(mgmt->da, ifsta->bssid, ETH_ALEN);
477
	memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
478 479 480 481 482 483 484 485
	memcpy(mgmt->bssid, ifsta->bssid, ETH_ALEN);
	mgmt->u.auth.auth_alg = cpu_to_le16(ifsta->auth_alg);
	mgmt->u.auth.auth_transaction = cpu_to_le16(transaction);
	ifsta->auth_transaction = transaction + 1;
	mgmt->u.auth.status_code = cpu_to_le16(0);
	if (extra)
		memcpy(skb_put(skb, extra_len), extra, extra_len);

486
	ieee80211_sta_tx(sdata, skb, encrypt);
487 488
}

489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518
static void ieee80211_direct_probe(struct ieee80211_sub_if_data *sdata,
				   struct ieee80211_if_sta *ifsta)
{
	DECLARE_MAC_BUF(mac);

	ifsta->direct_probe_tries++;
	if (ifsta->direct_probe_tries > IEEE80211_AUTH_MAX_TRIES) {
		printk(KERN_DEBUG "%s: direct probe to AP %s timed out\n",
		       sdata->dev->name, print_mac(mac, ifsta->bssid));
		ifsta->state = IEEE80211_STA_MLME_DISABLED;
		return;
	}

	printk(KERN_DEBUG "%s: direct probe to AP %s try %d\n",
			sdata->dev->name, print_mac(mac, ifsta->bssid),
			ifsta->direct_probe_tries);

	ifsta->state = IEEE80211_STA_MLME_DIRECT_PROBE;

	set_bit(IEEE80211_STA_REQ_DIRECT_PROBE, &ifsta->request);

	/* Direct probe is sent to broadcast address as some APs
	 * will not answer to direct packet in unassociated state.
	 */
	ieee80211_send_probe_req(sdata, NULL,
				 ifsta->ssid, ifsta->ssid_len);

	mod_timer(&ifsta->timer, jiffies + IEEE80211_AUTH_TIMEOUT);
}

519

520
static void ieee80211_authenticate(struct ieee80211_sub_if_data *sdata,
521 522
				   struct ieee80211_if_sta *ifsta)
{
523 524
	DECLARE_MAC_BUF(mac);

525 526
	ifsta->auth_tries++;
	if (ifsta->auth_tries > IEEE80211_AUTH_MAX_TRIES) {
527
		printk(KERN_DEBUG "%s: authentication with AP %s"
528
		       " timed out\n",
529
		       sdata->dev->name, print_mac(mac, ifsta->bssid));
530
		ifsta->state = IEEE80211_STA_MLME_DISABLED;
531 532 533
		return;
	}

534
	ifsta->state = IEEE80211_STA_MLME_AUTHENTICATE;
535
	printk(KERN_DEBUG "%s: authenticate with AP %s\n",
536
	       sdata->dev->name, print_mac(mac, ifsta->bssid));
537

538
	ieee80211_send_auth(sdata, ifsta, 1, NULL, 0, 0);
539 540 541 542

	mod_timer(&ifsta->timer, jiffies + IEEE80211_AUTH_TIMEOUT);
}

543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562
static int ieee80211_compatible_rates(struct ieee80211_sta_bss *bss,
				      struct ieee80211_supported_band *sband,
				      u64 *rates)
{
	int i, j, count;
	*rates = 0;
	count = 0;
	for (i = 0; i < bss->supp_rates_len; i++) {
		int rate = (bss->supp_rates[i] & 0x7F) * 5;

		for (j = 0; j < sband->n_bitrates; j++)
			if (sband->bitrates[j].bitrate == rate) {
				*rates |= BIT(j);
				count++;
				break;
			}
	}

	return count;
}
563

564
static void ieee80211_send_assoc(struct ieee80211_sub_if_data *sdata,
565 566
				 struct ieee80211_if_sta *ifsta)
{
567
	struct ieee80211_local *local = sdata->local;
568 569
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;
570
	u8 *pos, *ies, *ht_add_ie;
571
	int i, len, count, rates_len, supp_rates_len;
572 573 574
	u16 capab;
	struct ieee80211_sta_bss *bss;
	int wmm = 0;
575
	struct ieee80211_supported_band *sband;
576
	u64 rates = 0;
577 578 579 580 581 582

	skb = dev_alloc_skb(local->hw.extra_tx_headroom +
			    sizeof(*mgmt) + 200 + ifsta->extra_ie_len +
			    ifsta->ssid_len);
	if (!skb) {
		printk(KERN_DEBUG "%s: failed to allocate buffer for assoc "
583
		       "frame\n", sdata->dev->name);
584 585 586 587
		return;
	}
	skb_reserve(skb, local->hw.extra_tx_headroom);

588 589
	sband = local->hw.wiphy->bands[local->hw.conf.channel->band];

590
	capab = ifsta->capab;
591 592 593 594 595 596

	if (local->hw.conf.channel->band == IEEE80211_BAND_2GHZ) {
		if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_SLOT_INCAPABLE))
			capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
		if (!(local->hw.flags & IEEE80211_HW_2GHZ_SHORT_PREAMBLE_INCAPABLE))
			capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
597
	}
598

599
	bss = ieee80211_rx_bss_get(local, ifsta->bssid,
600
				   local->hw.conf.channel->center_freq,
601
				   ifsta->ssid, ifsta->ssid_len);
602 603 604
	if (bss) {
		if (bss->capability & WLAN_CAPABILITY_PRIVACY)
			capab |= WLAN_CAPABILITY_PRIVACY;
605
		if (bss->wmm_used)
606
			wmm = 1;
607 608 609 610 611 612 613

		/* get all rates supported by the device and the AP as
		 * some APs don't like getting a superset of their rates
		 * in the association request (e.g. D-Link DAP 1353 in
		 * b-only mode) */
		rates_len = ieee80211_compatible_rates(bss, sband, &rates);

614 615 616 617
		if ((bss->capability & WLAN_CAPABILITY_SPECTRUM_MGMT) &&
		    (local->hw.flags & IEEE80211_HW_SPECTRUM_MGMT))
			capab |= WLAN_CAPABILITY_SPECTRUM_MGMT;

618
		ieee80211_rx_bss_put(local, bss);
619 620 621
	} else {
		rates = ~0;
		rates_len = sband->n_bitrates;
622 623 624 625 626
	}

	mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
	memset(mgmt, 0, 24);
	memcpy(mgmt->da, ifsta->bssid, ETH_ALEN);
627
	memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
628 629
	memcpy(mgmt->bssid, ifsta->bssid, ETH_ALEN);

630
	if (ifsta->flags & IEEE80211_STA_PREV_BSSID_SET) {
631
		skb_put(skb, 10);
632 633
		mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
						  IEEE80211_STYPE_REASSOC_REQ);
634
		mgmt->u.reassoc_req.capab_info = cpu_to_le16(capab);
635 636
		mgmt->u.reassoc_req.listen_interval =
				cpu_to_le16(local->hw.conf.listen_interval);
637 638 639 640
		memcpy(mgmt->u.reassoc_req.current_ap, ifsta->prev_bssid,
		       ETH_ALEN);
	} else {
		skb_put(skb, 4);
641 642
		mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
						  IEEE80211_STYPE_ASSOC_REQ);
643
		mgmt->u.assoc_req.capab_info = cpu_to_le16(capab);
644 645
		mgmt->u.reassoc_req.listen_interval =
				cpu_to_le16(local->hw.conf.listen_interval);
646 647 648 649 650 651 652 653
	}

	/* SSID */
	ies = pos = skb_put(skb, 2 + ifsta->ssid_len);
	*pos++ = WLAN_EID_SSID;
	*pos++ = ifsta->ssid_len;
	memcpy(pos, ifsta->ssid, ifsta->ssid_len);

654
	/* add all rates which were marked to be used above */
655 656 657 658
	supp_rates_len = rates_len;
	if (supp_rates_len > 8)
		supp_rates_len = 8;

659
	len = sband->n_bitrates;
660
	pos = skb_put(skb, supp_rates_len + 2);
661
	*pos++ = WLAN_EID_SUPP_RATES;
662
	*pos++ = supp_rates_len;
663

664 665 666
	count = 0;
	for (i = 0; i < sband->n_bitrates; i++) {
		if (BIT(i) & rates) {
667
			int rate = sband->bitrates[i].bitrate;
668
			*pos++ = (u8) (rate / 5);
669 670 671 672 673
			if (++count == 8)
				break;
		}
	}

674
	if (rates_len > count) {
675 676 677 678 679 680 681 682 683
		pos = skb_put(skb, rates_len - count + 2);
		*pos++ = WLAN_EID_EXT_SUPP_RATES;
		*pos++ = rates_len - count;

		for (i++; i < sband->n_bitrates; i++) {
			if (BIT(i) & rates) {
				int rate = sband->bitrates[i].bitrate;
				*pos++ = (u8) (rate / 5);
			}
684 685 686
		}
	}

687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706
	if (capab & WLAN_CAPABILITY_SPECTRUM_MGMT) {
		/* 1. power capabilities */
		pos = skb_put(skb, 4);
		*pos++ = WLAN_EID_PWR_CAPABILITY;
		*pos++ = 2;
		*pos++ = 0; /* min tx power */
		*pos++ = local->hw.conf.channel->max_power; /* max tx power */

		/* 2. supported channels */
		/* TODO: get this in reg domain format */
		pos = skb_put(skb, 2 * sband->n_channels + 2);
		*pos++ = WLAN_EID_SUPPORTED_CHANNELS;
		*pos++ = 2 * sband->n_channels;
		for (i = 0; i < sband->n_channels; i++) {
			*pos++ = ieee80211_frequency_to_channel(
					sband->channels[i].center_freq);
			*pos++ = 1; /* one channel in the subband*/
		}
	}

707 708 709 710 711
	if (ifsta->extra_ie) {
		pos = skb_put(skb, ifsta->extra_ie_len);
		memcpy(pos, ifsta->extra_ie, ifsta->extra_ie_len);
	}

712
	if (wmm && (ifsta->flags & IEEE80211_STA_WMM_ENABLED)) {
713 714 715 716 717 718 719 720 721 722 723
		pos = skb_put(skb, 9);
		*pos++ = WLAN_EID_VENDOR_SPECIFIC;
		*pos++ = 7; /* len */
		*pos++ = 0x00; /* Microsoft OUI 00:50:F2 */
		*pos++ = 0x50;
		*pos++ = 0xf2;
		*pos++ = 2; /* WME */
		*pos++ = 0; /* WME info */
		*pos++ = 1; /* WME ver */
		*pos++ = 0;
	}
724

725
	/* wmm support is a must to HT */
726
	if (wmm && (ifsta->flags & IEEE80211_STA_WMM_ENABLED) &&
727 728
	    sband->ht_info.ht_supported &&
	    (ht_add_ie = ieee80211_bss_get_ie(bss, WLAN_EID_HT_EXTRA_INFO))) {
729
		struct ieee80211_ht_addt_info *ht_add_info =
730
			(struct ieee80211_ht_addt_info *)ht_add_ie;
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750
		u16 cap = sband->ht_info.cap;
		__le16 tmp;
		u32 flags = local->hw.conf.channel->flags;

		switch (ht_add_info->ht_param & IEEE80211_HT_IE_CHA_SEC_OFFSET) {
		case IEEE80211_HT_IE_CHA_SEC_ABOVE:
			if (flags & IEEE80211_CHAN_NO_FAT_ABOVE) {
				cap &= ~IEEE80211_HT_CAP_SUP_WIDTH;
				cap &= ~IEEE80211_HT_CAP_SGI_40;
			}
			break;
		case IEEE80211_HT_IE_CHA_SEC_BELOW:
			if (flags & IEEE80211_CHAN_NO_FAT_BELOW) {
				cap &= ~IEEE80211_HT_CAP_SUP_WIDTH;
				cap &= ~IEEE80211_HT_CAP_SGI_40;
			}
			break;
		}

		tmp = cpu_to_le16(cap);
751 752 753 754 755 756
		pos = skb_put(skb, sizeof(struct ieee80211_ht_cap)+2);
		*pos++ = WLAN_EID_HT_CAPABILITY;
		*pos++ = sizeof(struct ieee80211_ht_cap);
		memset(pos, 0, sizeof(struct ieee80211_ht_cap));
		memcpy(pos, &tmp, sizeof(u16));
		pos += sizeof(u16);
757 758 759 760
		/* TODO: needs a define here for << 2 */
		*pos++ = sband->ht_info.ampdu_factor |
			 (sband->ht_info.ampdu_density << 2);
		memcpy(pos, sband->ht_info.supp_mcs_set, 16);
761
	}
762 763 764

	kfree(ifsta->assocreq_ies);
	ifsta->assocreq_ies_len = (skb->data + skb->len) - ies;
765
	ifsta->assocreq_ies = kmalloc(ifsta->assocreq_ies_len, GFP_KERNEL);
766 767 768
	if (ifsta->assocreq_ies)
		memcpy(ifsta->assocreq_ies, ies, ifsta->assocreq_ies_len);

769
	ieee80211_sta_tx(sdata, skb, 0);
770 771 772
}


773
static void ieee80211_send_deauth(struct ieee80211_sub_if_data *sdata,
774 775
				  struct ieee80211_if_sta *ifsta, u16 reason)
{
776
	struct ieee80211_local *local = sdata->local;
777 778 779 780 781 782
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;

	skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*mgmt));
	if (!skb) {
		printk(KERN_DEBUG "%s: failed to allocate buffer for deauth "
783
		       "frame\n", sdata->dev->name);
784 785 786 787 788 789 790
		return;
	}
	skb_reserve(skb, local->hw.extra_tx_headroom);

	mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
	memset(mgmt, 0, 24);
	memcpy(mgmt->da, ifsta->bssid, ETH_ALEN);
791
	memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
792
	memcpy(mgmt->bssid, ifsta->bssid, ETH_ALEN);
793 794
	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
					  IEEE80211_STYPE_DEAUTH);
795 796 797
	skb_put(skb, 2);
	mgmt->u.deauth.reason_code = cpu_to_le16(reason);

798
	ieee80211_sta_tx(sdata, skb, 0);
799 800 801
}


802
static void ieee80211_send_disassoc(struct ieee80211_sub_if_data *sdata,
803 804
				    struct ieee80211_if_sta *ifsta, u16 reason)
{
805
	struct ieee80211_local *local = sdata->local;
806 807 808 809 810 811
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;

	skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*mgmt));
	if (!skb) {
		printk(KERN_DEBUG "%s: failed to allocate buffer for disassoc "
812
		       "frame\n", sdata->dev->name);
813 814 815 816 817 818 819
		return;
	}
	skb_reserve(skb, local->hw.extra_tx_headroom);

	mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
	memset(mgmt, 0, 24);
	memcpy(mgmt->da, ifsta->bssid, ETH_ALEN);
820
	memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
821
	memcpy(mgmt->bssid, ifsta->bssid, ETH_ALEN);
822 823
	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
					  IEEE80211_STYPE_DISASSOC);
824 825 826
	skb_put(skb, 2);
	mgmt->u.disassoc.reason_code = cpu_to_le16(reason);

827
	ieee80211_sta_tx(sdata, skb, 0);
828 829
}

830 831 832 833 834 835
static void ieee80211_set_disassoc(struct ieee80211_sub_if_data *sdata,
				   struct ieee80211_if_sta *ifsta, bool deauth,
				   bool self_disconnected, u16 reason)
{
	struct ieee80211_local *local = sdata->local;
	struct sta_info *sta;
836
	u32 changed = BSS_CHANGED_ASSOC;
837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852

	rcu_read_lock();

	sta = sta_info_get(local, ifsta->bssid);
	if (!sta) {
		rcu_read_unlock();
		return;
	}

	if (deauth) {
		ifsta->direct_probe_tries = 0;
		ifsta->auth_tries = 0;
	}
	ifsta->assoc_scan_tries = 0;
	ifsta->assoc_tries = 0;

853
	netif_tx_stop_all_queues(sdata->dev);
854 855 856 857 858 859 860 861 862 863 864
	netif_carrier_off(sdata->dev);

	ieee80211_sta_tear_down_BA_sessions(sdata, sta->addr);

	if (self_disconnected) {
		if (deauth)
			ieee80211_send_deauth(sdata, ifsta, reason);
		else
			ieee80211_send_disassoc(sdata, ifsta, reason);
	}

865 866 867 868 869 870 871 872 873 874 875 876 877 878
	ifsta->flags &= ~IEEE80211_STA_ASSOCIATED;
	changed |= ieee80211_reset_erp_info(sdata);

	if (sdata->bss_conf.assoc_ht)
		changed |= BSS_CHANGED_HT;

	sdata->bss_conf.assoc_ht = 0;
	sdata->bss_conf.ht_conf = NULL;
	sdata->bss_conf.ht_bss_conf = NULL;

	ieee80211_led_assoc(local, 0);
	sdata->bss_conf.assoc = 0;

	ieee80211_sta_send_apinfo(sdata, ifsta);
879 880 881 882 883 884 885 886 887 888

	if (self_disconnected)
		ifsta->state = IEEE80211_STA_MLME_DISABLED;

	sta_info_unlink(&sta);

	rcu_read_unlock();

	sta_info_destroy(sta);
}
889

890
static int ieee80211_privacy_mismatch(struct ieee80211_sub_if_data *sdata,
891 892
				      struct ieee80211_if_sta *ifsta)
{
893
	struct ieee80211_local *local = sdata->local;
894
	struct ieee80211_sta_bss *bss;
895 896 897
	int bss_privacy;
	int wep_privacy;
	int privacy_invoked;
898

899
	if (!ifsta || (ifsta->flags & IEEE80211_STA_MIXED_CELL))
900 901
		return 0;

902
	bss = ieee80211_rx_bss_get(local, ifsta->bssid,
903
				   local->hw.conf.channel->center_freq,
904
				   ifsta->ssid, ifsta->ssid_len);
905 906 907
	if (!bss)
		return 0;

908
	bss_privacy = !!(bss->capability & WLAN_CAPABILITY_PRIVACY);
909
	wep_privacy = !!ieee80211_sta_wep_configured(sdata);
910
	privacy_invoked = !!(ifsta->flags & IEEE80211_STA_PRIVACY_INVOKED);
911

912
	ieee80211_rx_bss_put(local, bss);
913

914 915 916 917
	if ((bss_privacy == wep_privacy) || (bss_privacy == privacy_invoked))
		return 0;

	return 1;
918 919 920
}


921
static void ieee80211_associate(struct ieee80211_sub_if_data *sdata,
922 923
				struct ieee80211_if_sta *ifsta)
{
924 925
	DECLARE_MAC_BUF(mac);

926 927
	ifsta->assoc_tries++;
	if (ifsta->assoc_tries > IEEE80211_ASSOC_MAX_TRIES) {
928
		printk(KERN_DEBUG "%s: association with AP %s"
929
		       " timed out\n",
930
		       sdata->dev->name, print_mac(mac, ifsta->bssid));
931
		ifsta->state = IEEE80211_STA_MLME_DISABLED;
932 933 934
		return;
	}

935
	ifsta->state = IEEE80211_STA_MLME_ASSOCIATE;
936
	printk(KERN_DEBUG "%s: associate with AP %s\n",
937 938
	       sdata->dev->name, print_mac(mac, ifsta->bssid));
	if (ieee80211_privacy_mismatch(sdata, ifsta)) {
939
		printk(KERN_DEBUG "%s: mismatch in privacy configuration and "
940
		       "mixed-cell disabled - abort association\n", sdata->dev->name);
941
		ifsta->state = IEEE80211_STA_MLME_DISABLED;
942 943 944
		return;
	}

945
	ieee80211_send_assoc(sdata, ifsta);
946 947 948 949 950

	mod_timer(&ifsta->timer, jiffies + IEEE80211_ASSOC_TIMEOUT);
}


951
static void ieee80211_associated(struct ieee80211_sub_if_data *sdata,
952 953
				 struct ieee80211_if_sta *ifsta)
{
954
	struct ieee80211_local *local = sdata->local;
955 956
	struct sta_info *sta;
	int disassoc;
957
	DECLARE_MAC_BUF(mac);
958 959 960 961 962 963

	/* TODO: start monitoring current AP signal quality and number of
	 * missed beacons. Scan other channels every now and then and search
	 * for better APs. */
	/* TODO: remove expired BSSes */

964
	ifsta->state = IEEE80211_STA_MLME_ASSOCIATED;
965

966 967
	rcu_read_lock();

968 969
	sta = sta_info_get(local, ifsta->bssid);
	if (!sta) {
970
		printk(KERN_DEBUG "%s: No STA entry for own AP %s\n",
971
		       sdata->dev->name, print_mac(mac, ifsta->bssid));
972 973 974 975 976
		disassoc = 1;
	} else {
		disassoc = 0;
		if (time_after(jiffies,
			       sta->last_rx + IEEE80211_MONITORING_INTERVAL)) {
977
			if (ifsta->flags & IEEE80211_STA_PROBEREQ_POLL) {
978
				printk(KERN_DEBUG "%s: No ProbeResp from "
979
				       "current AP %s - assume out of "
980
				       "range\n",
981
				       sdata->dev->name, print_mac(mac, ifsta->bssid));
982
				disassoc = 1;
983
			} else
984
				ieee80211_send_probe_req(sdata, ifsta->bssid,
985 986
							 local->scan_ssid,
							 local->scan_ssid_len);
987
			ifsta->flags ^= IEEE80211_STA_PROBEREQ_POLL;
988
		} else {
989
			ifsta->flags &= ~IEEE80211_STA_PROBEREQ_POLL;
990 991 992
			if (time_after(jiffies, ifsta->last_probe +
				       IEEE80211_PROBE_INTERVAL)) {
				ifsta->last_probe = jiffies;
993
				ieee80211_send_probe_req(sdata, ifsta->bssid,
994 995 996 997 998
							 ifsta->ssid,
							 ifsta->ssid_len);
			}
		}
	}
999 1000 1001

	rcu_read_unlock();

1002 1003 1004 1005
	if (disassoc)
		ieee80211_set_disassoc(sdata, ifsta, true, true,
					WLAN_REASON_PREV_AUTH_NOT_VALID);
	else
1006 1007 1008 1009 1010
		mod_timer(&ifsta->timer, jiffies +
				      IEEE80211_MONITORING_INTERVAL);
}


1011
static void ieee80211_send_probe_req(struct ieee80211_sub_if_data *sdata, u8 *dst,
1012 1013
				     u8 *ssid, size_t ssid_len)
{
1014
	struct ieee80211_local *local = sdata->local;
1015
	struct ieee80211_supported_band *sband;
1016 1017 1018 1019 1020 1021 1022 1023
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;
	u8 *pos, *supp_rates, *esupp_rates = NULL;
	int i;

	skb = dev_alloc_skb(local->hw.extra_tx_headroom + sizeof(*mgmt) + 200);
	if (!skb) {
		printk(KERN_DEBUG "%s: failed to allocate buffer for probe "
1024
		       "request\n", sdata->dev->name);
1025 1026 1027 1028 1029 1030
		return;
	}
	skb_reserve(skb, local->hw.extra_tx_headroom);

	mgmt = (struct ieee80211_mgmt *) skb_put(skb, 24);
	memset(mgmt, 0, 24);
1031 1032
	mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
					  IEEE80211_STYPE_PROBE_REQ);
1033
	memcpy(mgmt->sa, sdata->dev->dev_addr, ETH_ALEN);
1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045