Home | History | Annotate | Download | only in ap
      1 /*
      2  * hostapd / IEEE 802.11 Management
      3  * Copyright (c) 2002-2010, Jouni Malinen <j (at) w1.fi>
      4  *
      5  * This program is free software; you can redistribute it and/or modify
      6  * it under the terms of the GNU General Public License version 2 as
      7  * published by the Free Software Foundation.
      8  *
      9  * Alternatively, this software may be distributed under the terms of BSD
     10  * license.
     11  *
     12  * See README and COPYING for more details.
     13  */
     14 
     15 #include "utils/includes.h"
     16 
     17 #ifndef CONFIG_NATIVE_WINDOWS
     18 
     19 #include "utils/common.h"
     20 #include "utils/eloop.h"
     21 #include "crypto/crypto.h"
     22 #include "drivers/driver.h"
     23 #include "common/ieee802_11_defs.h"
     24 #include "common/ieee802_11_common.h"
     25 #include "common/wpa_ctrl.h"
     26 #include "radius/radius.h"
     27 #include "radius/radius_client.h"
     28 #include "p2p/p2p.h"
     29 #include "wps/wps.h"
     30 #include "hostapd.h"
     31 #include "beacon.h"
     32 #include "ieee802_11_auth.h"
     33 #include "sta_info.h"
     34 #include "ieee802_1x.h"
     35 #include "wpa_auth.h"
     36 #include "wmm.h"
     37 #include "ap_list.h"
     38 #include "accounting.h"
     39 #include "ap_config.h"
     40 #include "ap_mlme.h"
     41 #include "p2p_hostapd.h"
     42 #include "ap_drv_ops.h"
     43 #include "ieee802_11.h"
     44 
     45 
     46 u8 * hostapd_eid_supp_rates(struct hostapd_data *hapd, u8 *eid)
     47 {
     48 	u8 *pos = eid;
     49 	int i, num, count;
     50 
     51 	if (hapd->iface->current_rates == NULL)
     52 		return eid;
     53 
     54 	*pos++ = WLAN_EID_SUPP_RATES;
     55 	num = hapd->iface->num_rates;
     56 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
     57 		num++;
     58 	if (num > 8) {
     59 		/* rest of the rates are encoded in Extended supported
     60 		 * rates element */
     61 		num = 8;
     62 	}
     63 
     64 	*pos++ = num;
     65 	count = 0;
     66 	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num;
     67 	     i++) {
     68 		count++;
     69 		*pos = hapd->iface->current_rates[i].rate / 5;
     70 		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
     71 			*pos |= 0x80;
     72 		pos++;
     73 	}
     74 
     75 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht &&
     76 	    hapd->iface->num_rates < 8)
     77 		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
     78 
     79 	return pos;
     80 }
     81 
     82 
     83 u8 * hostapd_eid_ext_supp_rates(struct hostapd_data *hapd, u8 *eid)
     84 {
     85 	u8 *pos = eid;
     86 	int i, num, count;
     87 
     88 	if (hapd->iface->current_rates == NULL)
     89 		return eid;
     90 
     91 	num = hapd->iface->num_rates;
     92 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
     93 		num++;
     94 	if (num <= 8)
     95 		return eid;
     96 	num -= 8;
     97 
     98 	*pos++ = WLAN_EID_EXT_SUPP_RATES;
     99 	*pos++ = num;
    100 	count = 0;
    101 	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num + 8;
    102 	     i++) {
    103 		count++;
    104 		if (count <= 8)
    105 			continue; /* already in SuppRates IE */
    106 		*pos = hapd->iface->current_rates[i].rate / 5;
    107 		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
    108 			*pos |= 0x80;
    109 		pos++;
    110 	}
    111 
    112 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht &&
    113 	    hapd->iface->num_rates >= 8)
    114 		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
    115 
    116 	return pos;
    117 }
    118 
    119 
    120 u16 hostapd_own_capab_info(struct hostapd_data *hapd, struct sta_info *sta,
    121 			   int probe)
    122 {
    123 	int capab = WLAN_CAPABILITY_ESS;
    124 	int privacy;
    125 
    126 	if (hapd->iface->num_sta_no_short_preamble == 0 &&
    127 	    hapd->iconf->preamble == SHORT_PREAMBLE)
    128 		capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
    129 
    130 	privacy = hapd->conf->ssid.wep.keys_set;
    131 
    132 	if (hapd->conf->ieee802_1x &&
    133 	    (hapd->conf->default_wep_key_len ||
    134 	     hapd->conf->individual_wep_key_len))
    135 		privacy = 1;
    136 
    137 	if (hapd->conf->wpa)
    138 		privacy = 1;
    139 
    140 	if (sta) {
    141 		int policy, def_klen;
    142 		if (probe && sta->ssid_probe) {
    143 			policy = sta->ssid_probe->security_policy;
    144 			def_klen = sta->ssid_probe->wep.default_len;
    145 		} else {
    146 			policy = sta->ssid->security_policy;
    147 			def_klen = sta->ssid->wep.default_len;
    148 		}
    149 		privacy = policy != SECURITY_PLAINTEXT;
    150 		if (policy == SECURITY_IEEE_802_1X && def_klen == 0)
    151 			privacy = 0;
    152 	}
    153 
    154 	if (privacy)
    155 		capab |= WLAN_CAPABILITY_PRIVACY;
    156 
    157 	if (hapd->iface->current_mode &&
    158 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G &&
    159 	    hapd->iface->num_sta_no_short_slot_time == 0)
    160 		capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
    161 
    162 	return capab;
    163 }
    164 
    165 
    166 u8 * hostapd_eid_ext_capab(struct hostapd_data *hapd, u8 *eid)
    167 {
    168 	u8 *pos = eid;
    169 
    170 	if ((hapd->conf->tdls & (TDLS_PROHIBIT | TDLS_PROHIBIT_CHAN_SWITCH)) ==
    171 	    0)
    172 		return eid;
    173 
    174 	*pos++ = WLAN_EID_EXT_CAPAB;
    175 	*pos++ = 5;
    176 	*pos++ = 0x00;
    177 	*pos++ = 0x00;
    178 	*pos++ = 0x00;
    179 	*pos++ = 0x00;
    180 	*pos = 0x00;
    181 	if (hapd->conf->tdls & TDLS_PROHIBIT)
    182 		*pos |= 0x40; /* Bit 38 - TDLS Prohibited */
    183 	if (hapd->conf->tdls & TDLS_PROHIBIT_CHAN_SWITCH)
    184 		*pos |= 0x80; /* Bit 39 - TDLS Channel Switching Prohibited */
    185 	pos++;
    186 
    187 	return pos;
    188 }
    189 
    190 
    191 #ifdef CONFIG_IEEE80211W
    192 static u8 * hostapd_eid_assoc_comeback_time(struct hostapd_data *hapd,
    193 					    struct sta_info *sta, u8 *eid)
    194 {
    195 	u8 *pos = eid;
    196 	u32 timeout, tu;
    197 	struct os_time now, passed;
    198 
    199 	*pos++ = WLAN_EID_TIMEOUT_INTERVAL;
    200 	*pos++ = 5;
    201 	*pos++ = WLAN_TIMEOUT_ASSOC_COMEBACK;
    202 	os_get_time(&now);
    203 	os_time_sub(&now, &sta->sa_query_start, &passed);
    204 	tu = (passed.sec * 1000000 + passed.usec) / 1024;
    205 	if (hapd->conf->assoc_sa_query_max_timeout > tu)
    206 		timeout = hapd->conf->assoc_sa_query_max_timeout - tu;
    207 	else
    208 		timeout = 0;
    209 	if (timeout < hapd->conf->assoc_sa_query_max_timeout)
    210 		timeout++; /* add some extra time for local timers */
    211 	WPA_PUT_LE32(pos, timeout);
    212 	pos += 4;
    213 
    214 	return pos;
    215 }
    216 #endif /* CONFIG_IEEE80211W */
    217 
    218 
    219 void ieee802_11_print_ssid(char *buf, const u8 *ssid, u8 len)
    220 {
    221 	int i;
    222 	if (len > HOSTAPD_MAX_SSID_LEN)
    223 		len = HOSTAPD_MAX_SSID_LEN;
    224 	for (i = 0; i < len; i++) {
    225 		if (ssid[i] >= 32 && ssid[i] < 127)
    226 			buf[i] = ssid[i];
    227 		else
    228 			buf[i] = '.';
    229 	}
    230 	buf[len] = '\0';
    231 }
    232 
    233 
    234 static u16 auth_shared_key(struct hostapd_data *hapd, struct sta_info *sta,
    235 			   u16 auth_transaction, const u8 *challenge,
    236 			   int iswep)
    237 {
    238 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    239 		       HOSTAPD_LEVEL_DEBUG,
    240 		       "authentication (shared key, transaction %d)",
    241 		       auth_transaction);
    242 
    243 	if (auth_transaction == 1) {
    244 		if (!sta->challenge) {
    245 			/* Generate a pseudo-random challenge */
    246 			u8 key[8];
    247 			struct os_time now;
    248 			int r;
    249 			sta->challenge = os_zalloc(WLAN_AUTH_CHALLENGE_LEN);
    250 			if (sta->challenge == NULL)
    251 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
    252 
    253 			os_get_time(&now);
    254 			r = os_random();
    255 			os_memcpy(key, &now.sec, 4);
    256 			os_memcpy(key + 4, &r, 4);
    257 			rc4_skip(key, sizeof(key), 0,
    258 				 sta->challenge, WLAN_AUTH_CHALLENGE_LEN);
    259 		}
    260 		return 0;
    261 	}
    262 
    263 	if (auth_transaction != 3)
    264 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    265 
    266 	/* Transaction 3 */
    267 	if (!iswep || !sta->challenge || !challenge ||
    268 	    os_memcmp(sta->challenge, challenge, WLAN_AUTH_CHALLENGE_LEN)) {
    269 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    270 			       HOSTAPD_LEVEL_INFO,
    271 			       "shared key authentication - invalid "
    272 			       "challenge-response");
    273 		return WLAN_STATUS_CHALLENGE_FAIL;
    274 	}
    275 
    276 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    277 		       HOSTAPD_LEVEL_DEBUG,
    278 		       "authentication OK (shared key)");
    279 #ifdef IEEE80211_REQUIRE_AUTH_ACK
    280 	/* Station will be marked authenticated if it ACKs the
    281 	 * authentication reply. */
    282 #else
    283 	sta->flags |= WLAN_STA_AUTH;
    284 	wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
    285 #endif
    286 	os_free(sta->challenge);
    287 	sta->challenge = NULL;
    288 
    289 	return 0;
    290 }
    291 
    292 
    293 static void send_auth_reply(struct hostapd_data *hapd,
    294 			    const u8 *dst, const u8 *bssid,
    295 			    u16 auth_alg, u16 auth_transaction, u16 resp,
    296 			    const u8 *ies, size_t ies_len)
    297 {
    298 	struct ieee80211_mgmt *reply;
    299 	u8 *buf;
    300 	size_t rlen;
    301 
    302 	rlen = IEEE80211_HDRLEN + sizeof(reply->u.auth) + ies_len;
    303 	buf = os_zalloc(rlen);
    304 	if (buf == NULL)
    305 		return;
    306 
    307 	reply = (struct ieee80211_mgmt *) buf;
    308 	reply->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
    309 					    WLAN_FC_STYPE_AUTH);
    310 	os_memcpy(reply->da, dst, ETH_ALEN);
    311 	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
    312 	os_memcpy(reply->bssid, bssid, ETH_ALEN);
    313 
    314 	reply->u.auth.auth_alg = host_to_le16(auth_alg);
    315 	reply->u.auth.auth_transaction = host_to_le16(auth_transaction);
    316 	reply->u.auth.status_code = host_to_le16(resp);
    317 
    318 	if (ies && ies_len)
    319 		os_memcpy(reply->u.auth.variable, ies, ies_len);
    320 
    321 	wpa_printf(MSG_DEBUG, "authentication reply: STA=" MACSTR
    322 		   " auth_alg=%d auth_transaction=%d resp=%d (IE len=%lu)",
    323 		   MAC2STR(dst), auth_alg, auth_transaction,
    324 		   resp, (unsigned long) ies_len);
    325 	if (hostapd_drv_send_mlme(hapd, reply, rlen) < 0)
    326 		perror("send_auth_reply: send");
    327 
    328 	os_free(buf);
    329 }
    330 
    331 
    332 #ifdef CONFIG_IEEE80211R
    333 static void handle_auth_ft_finish(void *ctx, const u8 *dst, const u8 *bssid,
    334 				  u16 auth_transaction, u16 status,
    335 				  const u8 *ies, size_t ies_len)
    336 {
    337 	struct hostapd_data *hapd = ctx;
    338 	struct sta_info *sta;
    339 
    340 	send_auth_reply(hapd, dst, bssid, WLAN_AUTH_FT, auth_transaction,
    341 			status, ies, ies_len);
    342 
    343 	if (status != WLAN_STATUS_SUCCESS)
    344 		return;
    345 
    346 	sta = ap_get_sta(hapd, dst);
    347 	if (sta == NULL)
    348 		return;
    349 
    350 	hostapd_logger(hapd, dst, HOSTAPD_MODULE_IEEE80211,
    351 		       HOSTAPD_LEVEL_DEBUG, "authentication OK (FT)");
    352 	sta->flags |= WLAN_STA_AUTH;
    353 	mlme_authenticate_indication(hapd, sta);
    354 }
    355 #endif /* CONFIG_IEEE80211R */
    356 
    357 
    358 static void handle_auth(struct hostapd_data *hapd,
    359 			const struct ieee80211_mgmt *mgmt, size_t len)
    360 {
    361 	u16 auth_alg, auth_transaction, status_code;
    362 	u16 resp = WLAN_STATUS_SUCCESS;
    363 	struct sta_info *sta = NULL;
    364 	int res;
    365 	u16 fc;
    366 	const u8 *challenge = NULL;
    367 	u32 session_timeout, acct_interim_interval;
    368 	int vlan_id = 0;
    369 	u8 resp_ies[2 + WLAN_AUTH_CHALLENGE_LEN];
    370 	size_t resp_ies_len = 0;
    371 
    372 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
    373 		printf("handle_auth - too short payload (len=%lu)\n",
    374 		       (unsigned long) len);
    375 		return;
    376 	}
    377 
    378 	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
    379 	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
    380 	status_code = le_to_host16(mgmt->u.auth.status_code);
    381 	fc = le_to_host16(mgmt->frame_control);
    382 
    383 	if (len >= IEEE80211_HDRLEN + sizeof(mgmt->u.auth) +
    384 	    2 + WLAN_AUTH_CHALLENGE_LEN &&
    385 	    mgmt->u.auth.variable[0] == WLAN_EID_CHALLENGE &&
    386 	    mgmt->u.auth.variable[1] == WLAN_AUTH_CHALLENGE_LEN)
    387 		challenge = &mgmt->u.auth.variable[2];
    388 
    389 	wpa_printf(MSG_DEBUG, "authentication: STA=" MACSTR " auth_alg=%d "
    390 		   "auth_transaction=%d status_code=%d wep=%d%s",
    391 		   MAC2STR(mgmt->sa), auth_alg, auth_transaction,
    392 		   status_code, !!(fc & WLAN_FC_ISWEP),
    393 		   challenge ? " challenge" : "");
    394 
    395 	if (hapd->tkip_countermeasures) {
    396 		resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
    397 		goto fail;
    398 	}
    399 
    400 	if (!(((hapd->conf->auth_algs & WPA_AUTH_ALG_OPEN) &&
    401 	       auth_alg == WLAN_AUTH_OPEN) ||
    402 #ifdef CONFIG_IEEE80211R
    403 	      (hapd->conf->wpa &&
    404 	       (hapd->conf->wpa_key_mgmt &
    405 		(WPA_KEY_MGMT_FT_IEEE8021X | WPA_KEY_MGMT_FT_PSK)) &&
    406 	       auth_alg == WLAN_AUTH_FT) ||
    407 #endif /* CONFIG_IEEE80211R */
    408 	      ((hapd->conf->auth_algs & WPA_AUTH_ALG_SHARED) &&
    409 	       auth_alg == WLAN_AUTH_SHARED_KEY))) {
    410 		printf("Unsupported authentication algorithm (%d)\n",
    411 		       auth_alg);
    412 		resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
    413 		goto fail;
    414 	}
    415 
    416 	if (!(auth_transaction == 1 ||
    417 	      (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 3))) {
    418 		printf("Unknown authentication transaction number (%d)\n",
    419 		       auth_transaction);
    420 		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
    421 		goto fail;
    422 	}
    423 
    424 	if (os_memcmp(mgmt->sa, hapd->own_addr, ETH_ALEN) == 0) {
    425 		printf("Station " MACSTR " not allowed to authenticate.\n",
    426 		       MAC2STR(mgmt->sa));
    427 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    428 		goto fail;
    429 	}
    430 
    431 	res = hostapd_allowed_address(hapd, mgmt->sa, (u8 *) mgmt, len,
    432 				      &session_timeout,
    433 				      &acct_interim_interval, &vlan_id);
    434 	if (res == HOSTAPD_ACL_REJECT) {
    435 		printf("Station " MACSTR " not allowed to authenticate.\n",
    436 		       MAC2STR(mgmt->sa));
    437 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    438 		goto fail;
    439 	}
    440 	if (res == HOSTAPD_ACL_PENDING) {
    441 		wpa_printf(MSG_DEBUG, "Authentication frame from " MACSTR
    442 			   " waiting for an external authentication",
    443 			   MAC2STR(mgmt->sa));
    444 		/* Authentication code will re-send the authentication frame
    445 		 * after it has received (and cached) information from the
    446 		 * external source. */
    447 		return;
    448 	}
    449 
    450 	sta = ap_sta_add(hapd, mgmt->sa);
    451 	if (!sta) {
    452 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    453 		goto fail;
    454 	}
    455 
    456 	if (vlan_id > 0) {
    457 		if (hostapd_get_vlan_id_ifname(hapd->conf->vlan,
    458 					       vlan_id) == NULL) {
    459 			hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
    460 				       HOSTAPD_LEVEL_INFO, "Invalid VLAN ID "
    461 				       "%d received from RADIUS server",
    462 				       vlan_id);
    463 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    464 			goto fail;
    465 		}
    466 		sta->vlan_id = vlan_id;
    467 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
    468 			       HOSTAPD_LEVEL_INFO, "VLAN ID %d", sta->vlan_id);
    469 	}
    470 
    471 	sta->flags &= ~WLAN_STA_PREAUTH;
    472 	ieee802_1x_notify_pre_auth(sta->eapol_sm, 0);
    473 
    474 	if (hapd->conf->acct_interim_interval == 0 && acct_interim_interval)
    475 		sta->acct_interim_interval = acct_interim_interval;
    476 	if (res == HOSTAPD_ACL_ACCEPT_TIMEOUT)
    477 		ap_sta_session_timeout(hapd, sta, session_timeout);
    478 	else
    479 		ap_sta_no_session_timeout(hapd, sta);
    480 
    481 	switch (auth_alg) {
    482 	case WLAN_AUTH_OPEN:
    483 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    484 			       HOSTAPD_LEVEL_DEBUG,
    485 			       "authentication OK (open system)");
    486 #ifdef IEEE80211_REQUIRE_AUTH_ACK
    487 		/* Station will be marked authenticated if it ACKs the
    488 		 * authentication reply. */
    489 #else
    490 		sta->flags |= WLAN_STA_AUTH;
    491 		wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
    492 		sta->auth_alg = WLAN_AUTH_OPEN;
    493 		mlme_authenticate_indication(hapd, sta);
    494 #endif
    495 		break;
    496 	case WLAN_AUTH_SHARED_KEY:
    497 		resp = auth_shared_key(hapd, sta, auth_transaction, challenge,
    498 				       fc & WLAN_FC_ISWEP);
    499 		sta->auth_alg = WLAN_AUTH_SHARED_KEY;
    500 		mlme_authenticate_indication(hapd, sta);
    501 		if (sta->challenge && auth_transaction == 1) {
    502 			resp_ies[0] = WLAN_EID_CHALLENGE;
    503 			resp_ies[1] = WLAN_AUTH_CHALLENGE_LEN;
    504 			os_memcpy(resp_ies + 2, sta->challenge,
    505 				  WLAN_AUTH_CHALLENGE_LEN);
    506 			resp_ies_len = 2 + WLAN_AUTH_CHALLENGE_LEN;
    507 		}
    508 		break;
    509 #ifdef CONFIG_IEEE80211R
    510 	case WLAN_AUTH_FT:
    511 		sta->auth_alg = WLAN_AUTH_FT;
    512 		if (sta->wpa_sm == NULL)
    513 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
    514 							sta->addr);
    515 		if (sta->wpa_sm == NULL) {
    516 			wpa_printf(MSG_DEBUG, "FT: Failed to initialize WPA "
    517 				   "state machine");
    518 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    519 			goto fail;
    520 		}
    521 		wpa_ft_process_auth(sta->wpa_sm, mgmt->bssid,
    522 				    auth_transaction, mgmt->u.auth.variable,
    523 				    len - IEEE80211_HDRLEN -
    524 				    sizeof(mgmt->u.auth),
    525 				    handle_auth_ft_finish, hapd);
    526 		/* handle_auth_ft_finish() callback will complete auth. */
    527 		return;
    528 #endif /* CONFIG_IEEE80211R */
    529 	}
    530 
    531  fail:
    532 	send_auth_reply(hapd, mgmt->sa, mgmt->bssid, auth_alg,
    533 			auth_transaction + 1, resp, resp_ies, resp_ies_len);
    534 }
    535 
    536 
    537 static int hostapd_get_aid(struct hostapd_data *hapd, struct sta_info *sta)
    538 {
    539 	int i, j = 32, aid;
    540 
    541 	/* get a unique AID */
    542 	if (sta->aid > 0) {
    543 		wpa_printf(MSG_DEBUG, "  old AID %d", sta->aid);
    544 		return 0;
    545 	}
    546 
    547 	for (i = 0; i < AID_WORDS; i++) {
    548 		if (hapd->sta_aid[i] == (u32) -1)
    549 			continue;
    550 		for (j = 0; j < 32; j++) {
    551 			if (!(hapd->sta_aid[i] & BIT(j)))
    552 				break;
    553 		}
    554 		if (j < 32)
    555 			break;
    556 	}
    557 	if (j == 32)
    558 		return -1;
    559 	aid = i * 32 + j + 1;
    560 	if (aid > 2007)
    561 		return -1;
    562 
    563 	sta->aid = aid;
    564 	hapd->sta_aid[i] |= BIT(j);
    565 	wpa_printf(MSG_DEBUG, "  new AID %d", sta->aid);
    566 	return 0;
    567 }
    568 
    569 
    570 static u16 check_ssid(struct hostapd_data *hapd, struct sta_info *sta,
    571 		      const u8 *ssid_ie, size_t ssid_ie_len)
    572 {
    573 	if (ssid_ie == NULL)
    574 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    575 
    576 	if (ssid_ie_len != hapd->conf->ssid.ssid_len ||
    577 	    os_memcmp(ssid_ie, hapd->conf->ssid.ssid, ssid_ie_len) != 0) {
    578 		char ssid_txt[33];
    579 		ieee802_11_print_ssid(ssid_txt, ssid_ie, ssid_ie_len);
    580 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    581 			       HOSTAPD_LEVEL_INFO,
    582 			       "Station tried to associate with unknown SSID "
    583 			       "'%s'", ssid_txt);
    584 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    585 	}
    586 
    587 	return WLAN_STATUS_SUCCESS;
    588 }
    589 
    590 
    591 static u16 check_wmm(struct hostapd_data *hapd, struct sta_info *sta,
    592 		     const u8 *wmm_ie, size_t wmm_ie_len)
    593 {
    594 	sta->flags &= ~WLAN_STA_WMM;
    595 	if (wmm_ie && hapd->conf->wmm_enabled) {
    596 		if (hostapd_eid_wmm_valid(hapd, wmm_ie, wmm_ie_len))
    597 			hostapd_logger(hapd, sta->addr,
    598 				       HOSTAPD_MODULE_WPA,
    599 				       HOSTAPD_LEVEL_DEBUG,
    600 				       "invalid WMM element in association "
    601 				       "request");
    602 		else
    603 			sta->flags |= WLAN_STA_WMM;
    604 	}
    605 	return WLAN_STATUS_SUCCESS;
    606 }
    607 
    608 
    609 static u16 copy_supp_rates(struct hostapd_data *hapd, struct sta_info *sta,
    610 			   struct ieee802_11_elems *elems)
    611 {
    612 	if (!elems->supp_rates) {
    613 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    614 			       HOSTAPD_LEVEL_DEBUG,
    615 			       "No supported rates element in AssocReq");
    616 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    617 	}
    618 
    619 	if (elems->supp_rates_len > sizeof(sta->supported_rates)) {
    620 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    621 			       HOSTAPD_LEVEL_DEBUG,
    622 			       "Invalid supported rates element length %d",
    623 			       elems->supp_rates_len);
    624 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    625 	}
    626 
    627 	os_memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
    628 	os_memcpy(sta->supported_rates, elems->supp_rates,
    629 		  elems->supp_rates_len);
    630 	sta->supported_rates_len = elems->supp_rates_len;
    631 
    632 	if (elems->ext_supp_rates) {
    633 		if (elems->supp_rates_len + elems->ext_supp_rates_len >
    634 		    sizeof(sta->supported_rates)) {
    635 			hostapd_logger(hapd, sta->addr,
    636 				       HOSTAPD_MODULE_IEEE80211,
    637 				       HOSTAPD_LEVEL_DEBUG,
    638 				       "Invalid supported rates element length"
    639 				       " %d+%d", elems->supp_rates_len,
    640 				       elems->ext_supp_rates_len);
    641 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
    642 		}
    643 
    644 		os_memcpy(sta->supported_rates + elems->supp_rates_len,
    645 			  elems->ext_supp_rates, elems->ext_supp_rates_len);
    646 		sta->supported_rates_len += elems->ext_supp_rates_len;
    647 	}
    648 
    649 	return WLAN_STATUS_SUCCESS;
    650 }
    651 
    652 
    653 static u16 check_assoc_ies(struct hostapd_data *hapd, struct sta_info *sta,
    654 			   const u8 *ies, size_t ies_len, int reassoc)
    655 {
    656 	struct ieee802_11_elems elems;
    657 	u16 resp;
    658 	const u8 *wpa_ie;
    659 	size_t wpa_ie_len;
    660 
    661 	if (ieee802_11_parse_elems(ies, ies_len, &elems, 1) == ParseFailed) {
    662 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    663 			       HOSTAPD_LEVEL_INFO, "Station sent an invalid "
    664 			       "association request");
    665 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
    666 	}
    667 
    668 	resp = check_ssid(hapd, sta, elems.ssid, elems.ssid_len);
    669 	if (resp != WLAN_STATUS_SUCCESS)
    670 		return resp;
    671 	resp = check_wmm(hapd, sta, elems.wmm, elems.wmm_len);
    672 	if (resp != WLAN_STATUS_SUCCESS)
    673 		return resp;
    674 	resp = copy_supp_rates(hapd, sta, &elems);
    675 	if (resp != WLAN_STATUS_SUCCESS)
    676 		return resp;
    677 #ifdef CONFIG_IEEE80211N
    678 	resp = copy_sta_ht_capab(hapd, sta, elems.ht_capabilities,
    679 				 elems.ht_capabilities_len);
    680 	if (resp != WLAN_STATUS_SUCCESS)
    681 		return resp;
    682 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht &&
    683 	    !(sta->flags & WLAN_STA_HT)) {
    684 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    685 			       HOSTAPD_LEVEL_INFO, "Station does not support "
    686 			       "mandatory HT PHY - reject association");
    687 		return WLAN_STATUS_ASSOC_DENIED_NO_HT;
    688 	}
    689 #endif /* CONFIG_IEEE80211N */
    690 
    691 	if ((hapd->conf->wpa & WPA_PROTO_RSN) && elems.rsn_ie) {
    692 		wpa_ie = elems.rsn_ie;
    693 		wpa_ie_len = elems.rsn_ie_len;
    694 	} else if ((hapd->conf->wpa & WPA_PROTO_WPA) &&
    695 		   elems.wpa_ie) {
    696 		wpa_ie = elems.wpa_ie;
    697 		wpa_ie_len = elems.wpa_ie_len;
    698 	} else {
    699 		wpa_ie = NULL;
    700 		wpa_ie_len = 0;
    701 	}
    702 
    703 #ifdef CONFIG_WPS
    704 	sta->flags &= ~(WLAN_STA_WPS | WLAN_STA_MAYBE_WPS);
    705 	if (hapd->conf->wps_state && elems.wps_ie) {
    706 		wpa_printf(MSG_DEBUG, "STA included WPS IE in (Re)Association "
    707 			   "Request - assume WPS is used");
    708 		sta->flags |= WLAN_STA_WPS;
    709 		wpabuf_free(sta->wps_ie);
    710 		sta->wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
    711 							  WPS_IE_VENDOR_TYPE);
    712 		wpa_ie = NULL;
    713 		wpa_ie_len = 0;
    714 		if (sta->wps_ie && wps_validate_assoc_req(sta->wps_ie) < 0) {
    715 			wpa_printf(MSG_DEBUG, "WPS: Invalid WPS IE in "
    716 				   "(Re)Association Request - reject");
    717 			return WLAN_STATUS_INVALID_IE;
    718 		}
    719 	} else if (hapd->conf->wps_state && wpa_ie == NULL) {
    720 		wpa_printf(MSG_DEBUG, "STA did not include WPA/RSN IE in "
    721 			   "(Re)Association Request - possible WPS use");
    722 		sta->flags |= WLAN_STA_MAYBE_WPS;
    723 	} else
    724 #endif /* CONFIG_WPS */
    725 	if (hapd->conf->wpa && wpa_ie == NULL) {
    726 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
    727 			       HOSTAPD_LEVEL_INFO,
    728 			       "No WPA/RSN IE in association request");
    729 		return WLAN_STATUS_INVALID_IE;
    730 	}
    731 
    732 	if (hapd->conf->wpa && wpa_ie) {
    733 		int res;
    734 		wpa_ie -= 2;
    735 		wpa_ie_len += 2;
    736 		if (sta->wpa_sm == NULL)
    737 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
    738 							sta->addr);
    739 		if (sta->wpa_sm == NULL) {
    740 			wpa_printf(MSG_WARNING, "Failed to initialize WPA "
    741 				   "state machine");
    742 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
    743 		}
    744 		res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
    745 					  wpa_ie, wpa_ie_len,
    746 					  elems.mdie, elems.mdie_len);
    747 		if (res == WPA_INVALID_GROUP)
    748 			resp = WLAN_STATUS_GROUP_CIPHER_NOT_VALID;
    749 		else if (res == WPA_INVALID_PAIRWISE)
    750 			resp = WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
    751 		else if (res == WPA_INVALID_AKMP)
    752 			resp = WLAN_STATUS_AKMP_NOT_VALID;
    753 		else if (res == WPA_ALLOC_FAIL)
    754 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
    755 #ifdef CONFIG_IEEE80211W
    756 		else if (res == WPA_MGMT_FRAME_PROTECTION_VIOLATION)
    757 			resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
    758 		else if (res == WPA_INVALID_MGMT_GROUP_CIPHER)
    759 			resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
    760 #endif /* CONFIG_IEEE80211W */
    761 		else if (res == WPA_INVALID_MDIE)
    762 			resp = WLAN_STATUS_INVALID_MDIE;
    763 		else if (res != WPA_IE_OK)
    764 			resp = WLAN_STATUS_INVALID_IE;
    765 		if (resp != WLAN_STATUS_SUCCESS)
    766 			return resp;
    767 #ifdef CONFIG_IEEE80211W
    768 		if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
    769 		    sta->sa_query_count > 0)
    770 			ap_check_sa_query_timeout(hapd, sta);
    771 		if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
    772 		    (!reassoc || sta->auth_alg != WLAN_AUTH_FT)) {
    773 			/*
    774 			 * STA has already been associated with MFP and SA
    775 			 * Query timeout has not been reached. Reject the
    776 			 * association attempt temporarily and start SA Query,
    777 			 * if one is not pending.
    778 			 */
    779 
    780 			if (sta->sa_query_count == 0)
    781 				ap_sta_start_sa_query(hapd, sta);
    782 
    783 			return WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY;
    784 		}
    785 
    786 		if (wpa_auth_uses_mfp(sta->wpa_sm))
    787 			sta->flags |= WLAN_STA_MFP;
    788 		else
    789 			sta->flags &= ~WLAN_STA_MFP;
    790 #endif /* CONFIG_IEEE80211W */
    791 
    792 #ifdef CONFIG_IEEE80211R
    793 		if (sta->auth_alg == WLAN_AUTH_FT) {
    794 			if (!reassoc) {
    795 				wpa_printf(MSG_DEBUG, "FT: " MACSTR " tried "
    796 					   "to use association (not "
    797 					   "re-association) with FT auth_alg",
    798 					   MAC2STR(sta->addr));
    799 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
    800 			}
    801 
    802 			resp = wpa_ft_validate_reassoc(sta->wpa_sm, ies,
    803 						       ies_len);
    804 			if (resp != WLAN_STATUS_SUCCESS)
    805 				return resp;
    806 		}
    807 #endif /* CONFIG_IEEE80211R */
    808 
    809 #ifdef CONFIG_IEEE80211N
    810 		if ((sta->flags & WLAN_STA_HT) &&
    811 		    wpa_auth_get_pairwise(sta->wpa_sm) == WPA_CIPHER_TKIP) {
    812 			hostapd_logger(hapd, sta->addr,
    813 				       HOSTAPD_MODULE_IEEE80211,
    814 				       HOSTAPD_LEVEL_INFO,
    815 				       "Station tried to use TKIP with HT "
    816 				       "association");
    817 			return WLAN_STATUS_CIPHER_REJECTED_PER_POLICY;
    818 		}
    819 #endif /* CONFIG_IEEE80211N */
    820 	} else
    821 		wpa_auth_sta_no_wpa(sta->wpa_sm);
    822 
    823 #ifdef CONFIG_P2P
    824 	if (elems.p2p) {
    825 		wpabuf_free(sta->p2p_ie);
    826 		sta->p2p_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
    827 							  P2P_IE_VENDOR_TYPE);
    828 
    829 	} else {
    830 		wpabuf_free(sta->p2p_ie);
    831 		sta->p2p_ie = NULL;
    832 	}
    833 
    834 	p2p_group_notif_assoc(hapd->p2p_group, sta->addr, ies, ies_len);
    835 #endif /* CONFIG_P2P */
    836 
    837 	return WLAN_STATUS_SUCCESS;
    838 }
    839 
    840 
    841 static void send_deauth(struct hostapd_data *hapd, const u8 *addr,
    842 			u16 reason_code)
    843 {
    844 	int send_len;
    845 	struct ieee80211_mgmt reply;
    846 
    847 	os_memset(&reply, 0, sizeof(reply));
    848 	reply.frame_control =
    849 		IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_DEAUTH);
    850 	os_memcpy(reply.da, addr, ETH_ALEN);
    851 	os_memcpy(reply.sa, hapd->own_addr, ETH_ALEN);
    852 	os_memcpy(reply.bssid, hapd->own_addr, ETH_ALEN);
    853 
    854 	send_len = IEEE80211_HDRLEN + sizeof(reply.u.deauth);
    855 	reply.u.deauth.reason_code = host_to_le16(reason_code);
    856 
    857 	if (hostapd_drv_send_mlme(hapd, &reply, send_len) < 0)
    858 		wpa_printf(MSG_INFO, "Failed to send deauth: %s",
    859 			   strerror(errno));
    860 }
    861 
    862 
    863 static void send_assoc_resp(struct hostapd_data *hapd, struct sta_info *sta,
    864 			    u16 status_code, int reassoc, const u8 *ies,
    865 			    size_t ies_len)
    866 {
    867 	int send_len;
    868 	u8 buf[sizeof(struct ieee80211_mgmt) + 1024];
    869 	struct ieee80211_mgmt *reply;
    870 	u8 *p;
    871 
    872 	os_memset(buf, 0, sizeof(buf));
    873 	reply = (struct ieee80211_mgmt *) buf;
    874 	reply->frame_control =
    875 		IEEE80211_FC(WLAN_FC_TYPE_MGMT,
    876 			     (reassoc ? WLAN_FC_STYPE_REASSOC_RESP :
    877 			      WLAN_FC_STYPE_ASSOC_RESP));
    878 	os_memcpy(reply->da, sta->addr, ETH_ALEN);
    879 	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
    880 	os_memcpy(reply->bssid, hapd->own_addr, ETH_ALEN);
    881 
    882 	send_len = IEEE80211_HDRLEN;
    883 	send_len += sizeof(reply->u.assoc_resp);
    884 	reply->u.assoc_resp.capab_info =
    885 		host_to_le16(hostapd_own_capab_info(hapd, sta, 0));
    886 	reply->u.assoc_resp.status_code = host_to_le16(status_code);
    887 	reply->u.assoc_resp.aid = host_to_le16((sta ? sta->aid : 0)
    888 					       | BIT(14) | BIT(15));
    889 	/* Supported rates */
    890 	p = hostapd_eid_supp_rates(hapd, reply->u.assoc_resp.variable);
    891 	/* Extended supported rates */
    892 	p = hostapd_eid_ext_supp_rates(hapd, p);
    893 
    894 #ifdef CONFIG_IEEE80211R
    895 	if (status_code == WLAN_STATUS_SUCCESS) {
    896 		/* IEEE 802.11r: Mobility Domain Information, Fast BSS
    897 		 * Transition Information, RSN, [RIC Response] */
    898 		p = wpa_sm_write_assoc_resp_ies(sta->wpa_sm, p,
    899 						buf + sizeof(buf) - p,
    900 						sta->auth_alg, ies, ies_len);
    901 	}
    902 #endif /* CONFIG_IEEE80211R */
    903 
    904 #ifdef CONFIG_IEEE80211W
    905 	if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY)
    906 		p = hostapd_eid_assoc_comeback_time(hapd, sta, p);
    907 #endif /* CONFIG_IEEE80211W */
    908 
    909 #ifdef CONFIG_IEEE80211N
    910 	p = hostapd_eid_ht_capabilities(hapd, p);
    911 	p = hostapd_eid_ht_operation(hapd, p);
    912 #endif /* CONFIG_IEEE80211N */
    913 
    914 	p = hostapd_eid_ext_capab(hapd, p);
    915 
    916 	if (sta->flags & WLAN_STA_WMM)
    917 		p = hostapd_eid_wmm(hapd, p);
    918 
    919 #ifdef CONFIG_WPS
    920 	if (sta->flags & WLAN_STA_WPS) {
    921 		struct wpabuf *wps = wps_build_assoc_resp_ie();
    922 		if (wps) {
    923 			os_memcpy(p, wpabuf_head(wps), wpabuf_len(wps));
    924 			p += wpabuf_len(wps);
    925 			wpabuf_free(wps);
    926 		}
    927 	}
    928 #endif /* CONFIG_WPS */
    929 
    930 #ifdef CONFIG_P2P
    931 	if (sta->p2p_ie) {
    932 		struct wpabuf *p2p_resp_ie;
    933 		enum p2p_status_code status;
    934 		switch (status_code) {
    935 		case WLAN_STATUS_SUCCESS:
    936 			status = P2P_SC_SUCCESS;
    937 			break;
    938 		case WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA:
    939 			status = P2P_SC_FAIL_LIMIT_REACHED;
    940 			break;
    941 		default:
    942 			status = P2P_SC_FAIL_INVALID_PARAMS;
    943 			break;
    944 		}
    945 		p2p_resp_ie = p2p_group_assoc_resp_ie(hapd->p2p_group, status);
    946 		if (p2p_resp_ie) {
    947 			os_memcpy(p, wpabuf_head(p2p_resp_ie),
    948 				  wpabuf_len(p2p_resp_ie));
    949 			p += wpabuf_len(p2p_resp_ie);
    950 			wpabuf_free(p2p_resp_ie);
    951 		}
    952 	}
    953 #endif /* CONFIG_P2P */
    954 
    955 #ifdef CONFIG_P2P_MANAGER
    956 	if (hapd->conf->p2p & P2P_MANAGE)
    957 		p = hostapd_eid_p2p_manage(hapd, p);
    958 #endif /* CONFIG_P2P_MANAGER */
    959 
    960 	send_len += p - reply->u.assoc_resp.variable;
    961 
    962 	if (hostapd_drv_send_mlme(hapd, reply, send_len) < 0)
    963 		wpa_printf(MSG_INFO, "Failed to send assoc resp: %s",
    964 			   strerror(errno));
    965 }
    966 
    967 
    968 static void handle_assoc(struct hostapd_data *hapd,
    969 			 const struct ieee80211_mgmt *mgmt, size_t len,
    970 			 int reassoc)
    971 {
    972 	u16 capab_info, listen_interval;
    973 	u16 resp = WLAN_STATUS_SUCCESS;
    974 	const u8 *pos;
    975 	int left, i;
    976 	struct sta_info *sta;
    977 
    978 	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_req) :
    979 				      sizeof(mgmt->u.assoc_req))) {
    980 		printf("handle_assoc(reassoc=%d) - too short payload (len=%lu)"
    981 		       "\n", reassoc, (unsigned long) len);
    982 		return;
    983 	}
    984 
    985 	if (reassoc) {
    986 		capab_info = le_to_host16(mgmt->u.reassoc_req.capab_info);
    987 		listen_interval = le_to_host16(
    988 			mgmt->u.reassoc_req.listen_interval);
    989 		wpa_printf(MSG_DEBUG, "reassociation request: STA=" MACSTR
    990 			   " capab_info=0x%02x listen_interval=%d current_ap="
    991 			   MACSTR,
    992 			   MAC2STR(mgmt->sa), capab_info, listen_interval,
    993 			   MAC2STR(mgmt->u.reassoc_req.current_ap));
    994 		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.reassoc_req));
    995 		pos = mgmt->u.reassoc_req.variable;
    996 	} else {
    997 		capab_info = le_to_host16(mgmt->u.assoc_req.capab_info);
    998 		listen_interval = le_to_host16(
    999 			mgmt->u.assoc_req.listen_interval);
   1000 		wpa_printf(MSG_DEBUG, "association request: STA=" MACSTR
   1001 			   " capab_info=0x%02x listen_interval=%d",
   1002 			   MAC2STR(mgmt->sa), capab_info, listen_interval);
   1003 		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.assoc_req));
   1004 		pos = mgmt->u.assoc_req.variable;
   1005 	}
   1006 
   1007 	sta = ap_get_sta(hapd, mgmt->sa);
   1008 #ifdef CONFIG_IEEE80211R
   1009 	if (sta && sta->auth_alg == WLAN_AUTH_FT &&
   1010 	    (sta->flags & WLAN_STA_AUTH) == 0) {
   1011 		wpa_printf(MSG_DEBUG, "FT: Allow STA " MACSTR " to associate "
   1012 			   "prior to authentication since it is using "
   1013 			   "over-the-DS FT", MAC2STR(mgmt->sa));
   1014 	} else
   1015 #endif /* CONFIG_IEEE80211R */
   1016 	if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
   1017 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1018 			       HOSTAPD_LEVEL_INFO, "Station tried to "
   1019 			       "associate before authentication "
   1020 			       "(aid=%d flags=0x%x)",
   1021 			       sta ? sta->aid : -1,
   1022 			       sta ? sta->flags : 0);
   1023 		send_deauth(hapd, mgmt->sa,
   1024 			    WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA);
   1025 		return;
   1026 	}
   1027 
   1028 	if (hapd->tkip_countermeasures) {
   1029 		resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
   1030 		goto fail;
   1031 	}
   1032 
   1033 	if (listen_interval > hapd->conf->max_listen_interval) {
   1034 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1035 			       HOSTAPD_LEVEL_DEBUG,
   1036 			       "Too large Listen Interval (%d)",
   1037 			       listen_interval);
   1038 		resp = WLAN_STATUS_ASSOC_DENIED_LISTEN_INT_TOO_LARGE;
   1039 		goto fail;
   1040 	}
   1041 
   1042 	/* followed by SSID and Supported rates; and HT capabilities if 802.11n
   1043 	 * is used */
   1044 	resp = check_assoc_ies(hapd, sta, pos, left, reassoc);
   1045 	if (resp != WLAN_STATUS_SUCCESS)
   1046 		goto fail;
   1047 
   1048 	if (hostapd_get_aid(hapd, sta) < 0) {
   1049 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1050 			       HOSTAPD_LEVEL_INFO, "No room for more AIDs");
   1051 		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
   1052 		goto fail;
   1053 	}
   1054 
   1055 	sta->capability = capab_info;
   1056 	sta->listen_interval = listen_interval;
   1057 
   1058 	if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G)
   1059 		sta->flags |= WLAN_STA_NONERP;
   1060 	for (i = 0; i < sta->supported_rates_len; i++) {
   1061 		if ((sta->supported_rates[i] & 0x7f) > 22) {
   1062 			sta->flags &= ~WLAN_STA_NONERP;
   1063 			break;
   1064 		}
   1065 	}
   1066 	if (sta->flags & WLAN_STA_NONERP && !sta->nonerp_set) {
   1067 		sta->nonerp_set = 1;
   1068 		hapd->iface->num_sta_non_erp++;
   1069 		if (hapd->iface->num_sta_non_erp == 1)
   1070 			ieee802_11_set_beacons(hapd->iface);
   1071 	}
   1072 
   1073 	if (!(sta->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME) &&
   1074 	    !sta->no_short_slot_time_set) {
   1075 		sta->no_short_slot_time_set = 1;
   1076 		hapd->iface->num_sta_no_short_slot_time++;
   1077 		if (hapd->iface->current_mode->mode ==
   1078 		    HOSTAPD_MODE_IEEE80211G &&
   1079 		    hapd->iface->num_sta_no_short_slot_time == 1)
   1080 			ieee802_11_set_beacons(hapd->iface);
   1081 	}
   1082 
   1083 	if (sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
   1084 		sta->flags |= WLAN_STA_SHORT_PREAMBLE;
   1085 	else
   1086 		sta->flags &= ~WLAN_STA_SHORT_PREAMBLE;
   1087 
   1088 	if (!(sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE) &&
   1089 	    !sta->no_short_preamble_set) {
   1090 		sta->no_short_preamble_set = 1;
   1091 		hapd->iface->num_sta_no_short_preamble++;
   1092 		if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
   1093 		    && hapd->iface->num_sta_no_short_preamble == 1)
   1094 			ieee802_11_set_beacons(hapd->iface);
   1095 	}
   1096 
   1097 #ifdef CONFIG_IEEE80211N
   1098 	update_ht_state(hapd, sta);
   1099 #endif /* CONFIG_IEEE80211N */
   1100 
   1101 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1102 		       HOSTAPD_LEVEL_DEBUG,
   1103 		       "association OK (aid %d)", sta->aid);
   1104 	/* Station will be marked associated, after it acknowledges AssocResp
   1105 	 */
   1106 	sta->flags |= WLAN_STA_ASSOC_REQ_OK;
   1107 
   1108 #ifdef CONFIG_IEEE80211W
   1109 	if ((sta->flags & WLAN_STA_MFP) && sta->sa_query_timed_out) {
   1110 		wpa_printf(MSG_DEBUG, "Allowing %sassociation after timed out "
   1111 			   "SA Query procedure", reassoc ? "re" : "");
   1112 		/* TODO: Send a protected Disassociate frame to the STA using
   1113 		 * the old key and Reason Code "Previous Authentication no
   1114 		 * longer valid". Make sure this is only sent protected since
   1115 		 * unprotected frame would be received by the STA that is now
   1116 		 * trying to associate.
   1117 		 */
   1118 	}
   1119 #endif /* CONFIG_IEEE80211W */
   1120 
   1121 	if (reassoc) {
   1122 		os_memcpy(sta->previous_ap, mgmt->u.reassoc_req.current_ap,
   1123 			  ETH_ALEN);
   1124 	}
   1125 
   1126 	if (sta->last_assoc_req)
   1127 		os_free(sta->last_assoc_req);
   1128 	sta->last_assoc_req = os_malloc(len);
   1129 	if (sta->last_assoc_req)
   1130 		os_memcpy(sta->last_assoc_req, mgmt, len);
   1131 
   1132 	/* Make sure that the previously registered inactivity timer will not
   1133 	 * remove the STA immediately. */
   1134 	sta->timeout_next = STA_NULLFUNC;
   1135 
   1136  fail:
   1137 	send_assoc_resp(hapd, sta, resp, reassoc, pos, left);
   1138 }
   1139 
   1140 
   1141 static void handle_disassoc(struct hostapd_data *hapd,
   1142 			    const struct ieee80211_mgmt *mgmt, size_t len)
   1143 {
   1144 	struct sta_info *sta;
   1145 
   1146 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.disassoc)) {
   1147 		printf("handle_disassoc - too short payload (len=%lu)\n",
   1148 		       (unsigned long) len);
   1149 		return;
   1150 	}
   1151 
   1152 	wpa_printf(MSG_DEBUG, "disassocation: STA=" MACSTR " reason_code=%d",
   1153 		   MAC2STR(mgmt->sa),
   1154 		   le_to_host16(mgmt->u.disassoc.reason_code));
   1155 
   1156 	sta = ap_get_sta(hapd, mgmt->sa);
   1157 	if (sta == NULL) {
   1158 		printf("Station " MACSTR " trying to disassociate, but it "
   1159 		       "is not associated.\n", MAC2STR(mgmt->sa));
   1160 		return;
   1161 	}
   1162 
   1163 	sta->flags &= ~(WLAN_STA_ASSOC | WLAN_STA_ASSOC_REQ_OK);
   1164 	wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
   1165 		MAC2STR(sta->addr));
   1166 #ifdef ANDROID_BRCM_P2P_PATCH
   1167 	if(hapd->msg_ctx_parent)
   1168 		wpa_msg(hapd->msg_ctx_parent, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
   1169 			MAC2STR(sta->addr));
   1170 #endif /* ANDROID_BRCM_P2P_PATCH */
   1171 
   1172 	wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
   1173 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1174 		       HOSTAPD_LEVEL_INFO, "disassociated");
   1175 	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
   1176 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
   1177 	/* Stop Accounting and IEEE 802.1X sessions, but leave the STA
   1178 	 * authenticated. */
   1179 	accounting_sta_stop(hapd, sta);
   1180 	ieee802_1x_free_station(sta);
   1181 	hostapd_drv_sta_remove(hapd, sta->addr);
   1182 
   1183 	if (sta->timeout_next == STA_NULLFUNC ||
   1184 	    sta->timeout_next == STA_DISASSOC) {
   1185 		sta->timeout_next = STA_DEAUTH;
   1186 		eloop_cancel_timeout(ap_handle_timer, hapd, sta);
   1187 		eloop_register_timeout(AP_DEAUTH_DELAY, 0, ap_handle_timer,
   1188 				       hapd, sta);
   1189 	}
   1190 
   1191 	mlme_disassociate_indication(
   1192 		hapd, sta, le_to_host16(mgmt->u.disassoc.reason_code));
   1193 }
   1194 
   1195 
   1196 static void handle_deauth(struct hostapd_data *hapd,
   1197 			  const struct ieee80211_mgmt *mgmt, size_t len)
   1198 {
   1199 	struct sta_info *sta;
   1200 
   1201 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.deauth)) {
   1202 		wpa_msg(hapd, MSG_DEBUG, "handle_deauth - too short payload "
   1203 			"(len=%lu)", (unsigned long) len);
   1204 		return;
   1205 	}
   1206 
   1207 	wpa_msg(hapd, MSG_DEBUG, "deauthentication: STA=" MACSTR
   1208 		" reason_code=%d",
   1209 		MAC2STR(mgmt->sa), le_to_host16(mgmt->u.deauth.reason_code));
   1210 
   1211 	sta = ap_get_sta(hapd, mgmt->sa);
   1212 	if (sta == NULL) {
   1213 		wpa_msg(hapd, MSG_DEBUG, "Station " MACSTR " trying to "
   1214 			"deauthenticate, but it is not authenticated",
   1215 			MAC2STR(mgmt->sa));
   1216 		return;
   1217 	}
   1218 
   1219 	sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC |
   1220 			WLAN_STA_ASSOC_REQ_OK);
   1221 	wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
   1222 		MAC2STR(sta->addr));
   1223 #ifdef ANDROID_BRCM_P2P_PATCH
   1224 	if(hapd->msg_ctx_parent)
   1225 		wpa_msg(hapd->msg_ctx_parent, MSG_INFO, AP_STA_DISCONNECTED MACSTR,
   1226 			MAC2STR(sta->addr));
   1227 #endif /* ANDROID_BRCM_P2P_PATCH */
   1228 	wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
   1229 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1230 		       HOSTAPD_LEVEL_DEBUG, "deauthenticated");
   1231 	mlme_deauthenticate_indication(
   1232 		hapd, sta, le_to_host16(mgmt->u.deauth.reason_code));
   1233 	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
   1234 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
   1235 	ap_free_sta(hapd, sta);
   1236 }
   1237 
   1238 
   1239 static void handle_beacon(struct hostapd_data *hapd,
   1240 			  const struct ieee80211_mgmt *mgmt, size_t len,
   1241 			  struct hostapd_frame_info *fi)
   1242 {
   1243 	struct ieee802_11_elems elems;
   1244 
   1245 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.beacon)) {
   1246 		printf("handle_beacon - too short payload (len=%lu)\n",
   1247 		       (unsigned long) len);
   1248 		return;
   1249 	}
   1250 
   1251 	(void) ieee802_11_parse_elems(mgmt->u.beacon.variable,
   1252 				      len - (IEEE80211_HDRLEN +
   1253 					     sizeof(mgmt->u.beacon)), &elems,
   1254 				      0);
   1255 
   1256 	ap_list_process_beacon(hapd->iface, mgmt, &elems, fi);
   1257 }
   1258 
   1259 
   1260 #ifdef CONFIG_IEEE80211W
   1261 
   1262 /* MLME-SAQuery.request */
   1263 void ieee802_11_send_sa_query_req(struct hostapd_data *hapd,
   1264 				  const u8 *addr, const u8 *trans_id)
   1265 {
   1266 	struct ieee80211_mgmt mgmt;
   1267 	u8 *end;
   1268 
   1269 	wpa_printf(MSG_DEBUG, "IEEE 802.11: Sending SA Query Request to "
   1270 		   MACSTR, MAC2STR(addr));
   1271 	wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
   1272 		    trans_id, WLAN_SA_QUERY_TR_ID_LEN);
   1273 
   1274 	os_memset(&mgmt, 0, sizeof(mgmt));
   1275 	mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
   1276 					  WLAN_FC_STYPE_ACTION);
   1277 	os_memcpy(mgmt.da, addr, ETH_ALEN);
   1278 	os_memcpy(mgmt.sa, hapd->own_addr, ETH_ALEN);
   1279 	os_memcpy(mgmt.bssid, hapd->own_addr, ETH_ALEN);
   1280 	mgmt.u.action.category = WLAN_ACTION_SA_QUERY;
   1281 	mgmt.u.action.u.sa_query_req.action = WLAN_SA_QUERY_REQUEST;
   1282 	os_memcpy(mgmt.u.action.u.sa_query_req.trans_id, trans_id,
   1283 		  WLAN_SA_QUERY_TR_ID_LEN);
   1284 	end = mgmt.u.action.u.sa_query_req.trans_id + WLAN_SA_QUERY_TR_ID_LEN;
   1285 	if (hostapd_drv_send_mlme(hapd, &mgmt, end - (u8 *) &mgmt) < 0)
   1286 		perror("ieee802_11_send_sa_query_req: send");
   1287 }
   1288 
   1289 
   1290 static void hostapd_sa_query_request(struct hostapd_data *hapd,
   1291 				     const struct ieee80211_mgmt *mgmt)
   1292 {
   1293 	struct sta_info *sta;
   1294 	struct ieee80211_mgmt resp;
   1295 	u8 *end;
   1296 
   1297 	wpa_printf(MSG_DEBUG, "IEEE 802.11: Received SA Query Request from "
   1298 		   MACSTR, MAC2STR(mgmt->sa));
   1299 	wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
   1300 		    mgmt->u.action.u.sa_query_resp.trans_id,
   1301 		    WLAN_SA_QUERY_TR_ID_LEN);
   1302 
   1303 	sta = ap_get_sta(hapd, mgmt->sa);
   1304 	if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
   1305 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignore SA Query Request "
   1306 			   "from unassociated STA " MACSTR, MAC2STR(mgmt->sa));
   1307 		return;
   1308 	}
   1309 
   1310 	wpa_printf(MSG_DEBUG, "IEEE 802.11: Sending SA Query Response to "
   1311 		   MACSTR, MAC2STR(mgmt->sa));
   1312 
   1313 	os_memset(&resp, 0, sizeof(resp));
   1314 	resp.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
   1315 					  WLAN_FC_STYPE_ACTION);
   1316 	os_memcpy(resp.da, mgmt->sa, ETH_ALEN);
   1317 	os_memcpy(resp.sa, hapd->own_addr, ETH_ALEN);
   1318 	os_memcpy(resp.bssid, hapd->own_addr, ETH_ALEN);
   1319 	resp.u.action.category = WLAN_ACTION_SA_QUERY;
   1320 	resp.u.action.u.sa_query_req.action = WLAN_SA_QUERY_RESPONSE;
   1321 	os_memcpy(resp.u.action.u.sa_query_req.trans_id,
   1322 		  mgmt->u.action.u.sa_query_req.trans_id,
   1323 		  WLAN_SA_QUERY_TR_ID_LEN);
   1324 	end = resp.u.action.u.sa_query_req.trans_id + WLAN_SA_QUERY_TR_ID_LEN;
   1325 	if (hostapd_drv_send_mlme(hapd, &resp, end - (u8 *) &resp) < 0)
   1326 		perror("hostapd_sa_query_request: send");
   1327 }
   1328 
   1329 
   1330 static void hostapd_sa_query_action(struct hostapd_data *hapd,
   1331 				    const struct ieee80211_mgmt *mgmt,
   1332 				    size_t len)
   1333 {
   1334 	struct sta_info *sta;
   1335 	const u8 *end;
   1336 	int i;
   1337 
   1338 	end = mgmt->u.action.u.sa_query_resp.trans_id +
   1339 		WLAN_SA_QUERY_TR_ID_LEN;
   1340 	if (((u8 *) mgmt) + len < end) {
   1341 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Too short SA Query Action "
   1342 			   "frame (len=%lu)", (unsigned long) len);
   1343 		return;
   1344 	}
   1345 
   1346 	if (mgmt->u.action.u.sa_query_resp.action == WLAN_SA_QUERY_REQUEST) {
   1347 		hostapd_sa_query_request(hapd, mgmt);
   1348 		return;
   1349 	}
   1350 
   1351 	if (mgmt->u.action.u.sa_query_resp.action != WLAN_SA_QUERY_RESPONSE) {
   1352 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Unexpected SA Query "
   1353 			   "Action %d", mgmt->u.action.u.sa_query_resp.action);
   1354 		return;
   1355 	}
   1356 
   1357 	wpa_printf(MSG_DEBUG, "IEEE 802.11: Received SA Query Response from "
   1358 		   MACSTR, MAC2STR(mgmt->sa));
   1359 	wpa_hexdump(MSG_DEBUG, "IEEE 802.11: SA Query Transaction ID",
   1360 		    mgmt->u.action.u.sa_query_resp.trans_id,
   1361 		    WLAN_SA_QUERY_TR_ID_LEN);
   1362 
   1363 	/* MLME-SAQuery.confirm */
   1364 
   1365 	sta = ap_get_sta(hapd, mgmt->sa);
   1366 	if (sta == NULL || sta->sa_query_trans_id == NULL) {
   1367 		wpa_printf(MSG_DEBUG, "IEEE 802.11: No matching STA with "
   1368 			   "pending SA Query request found");
   1369 		return;
   1370 	}
   1371 
   1372 	for (i = 0; i < sta->sa_query_count; i++) {
   1373 		if (os_memcmp(sta->sa_query_trans_id +
   1374 			      i * WLAN_SA_QUERY_TR_ID_LEN,
   1375 			      mgmt->u.action.u.sa_query_resp.trans_id,
   1376 			      WLAN_SA_QUERY_TR_ID_LEN) == 0)
   1377 			break;
   1378 	}
   1379 
   1380 	if (i >= sta->sa_query_count) {
   1381 		wpa_printf(MSG_DEBUG, "IEEE 802.11: No matching SA Query "
   1382 			   "transaction identifier found");
   1383 		return;
   1384 	}
   1385 
   1386 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1387 		       HOSTAPD_LEVEL_DEBUG,
   1388 		       "Reply to pending SA Query received");
   1389 	ap_sta_stop_sa_query(hapd, sta);
   1390 }
   1391 
   1392 
   1393 static int robust_action_frame(u8 category)
   1394 {
   1395 	return category != WLAN_ACTION_PUBLIC &&
   1396 		category != WLAN_ACTION_HT;
   1397 }
   1398 #endif /* CONFIG_IEEE80211W */
   1399 
   1400 
   1401 static void handle_action(struct hostapd_data *hapd,
   1402 			  const struct ieee80211_mgmt *mgmt, size_t len)
   1403 {
   1404 	struct sta_info *sta;
   1405 
   1406 	if (len < IEEE80211_HDRLEN + 1) {
   1407 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1408 			       HOSTAPD_LEVEL_DEBUG,
   1409 			       "handle_action - too short payload (len=%lu)",
   1410 			       (unsigned long) len);
   1411 		return;
   1412 	}
   1413 
   1414 	sta = ap_get_sta(hapd, mgmt->sa);
   1415 #ifdef CONFIG_IEEE80211W
   1416 	if (sta && (sta->flags & WLAN_STA_MFP) &&
   1417 	    !(mgmt->frame_control & host_to_le16(WLAN_FC_ISWEP) &&
   1418 	      robust_action_frame(mgmt->u.action.category))) {
   1419 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1420 			       HOSTAPD_LEVEL_DEBUG,
   1421 			       "Dropped unprotected Robust Action frame from "
   1422 			       "an MFP STA");
   1423 		return;
   1424 	}
   1425 #endif /* CONFIG_IEEE80211W */
   1426 
   1427 	switch (mgmt->u.action.category) {
   1428 #ifdef CONFIG_IEEE80211R
   1429 	case WLAN_ACTION_FT:
   1430 	{
   1431 		if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
   1432 			wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignored FT Action "
   1433 				   "frame from unassociated STA " MACSTR,
   1434 				   MAC2STR(mgmt->sa));
   1435 			return;
   1436 		}
   1437 
   1438 		if (wpa_ft_action_rx(sta->wpa_sm, (u8 *) &mgmt->u.action,
   1439 				     len - IEEE80211_HDRLEN))
   1440 			break;
   1441 
   1442 		return;
   1443 	}
   1444 #endif /* CONFIG_IEEE80211R */
   1445 	case WLAN_ACTION_WMM:
   1446 		hostapd_wmm_action(hapd, mgmt, len);
   1447 		return;
   1448 #ifdef CONFIG_IEEE80211W
   1449 	case WLAN_ACTION_SA_QUERY:
   1450 		hostapd_sa_query_action(hapd, mgmt, len);
   1451 		return;
   1452 #endif /* CONFIG_IEEE80211W */
   1453 	case WLAN_ACTION_PUBLIC:
   1454 		if (hapd->public_action_cb) {
   1455 			hapd->public_action_cb(hapd->public_action_cb_ctx,
   1456 					       (u8 *) mgmt, len,
   1457 					       hapd->iface->freq);
   1458 			return;
   1459 		}
   1460 		break;
   1461 	case WLAN_ACTION_VENDOR_SPECIFIC:
   1462 		if (hapd->vendor_action_cb) {
   1463 			if (hapd->vendor_action_cb(hapd->vendor_action_cb_ctx,
   1464 						   (u8 *) mgmt, len,
   1465 						   hapd->iface->freq) == 0)
   1466 				return;
   1467 		}
   1468 		break;
   1469 	}
   1470 
   1471 	hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1472 		       HOSTAPD_LEVEL_DEBUG,
   1473 		       "handle_action - unknown action category %d or invalid "
   1474 		       "frame",
   1475 		       mgmt->u.action.category);
   1476 	if (!(mgmt->da[0] & 0x01) && !(mgmt->u.action.category & 0x80) &&
   1477 	    !(mgmt->sa[0] & 0x01)) {
   1478 		struct ieee80211_mgmt *resp;
   1479 
   1480 		/*
   1481 		 * IEEE 802.11-REVma/D9.0 - 7.3.1.11
   1482 		 * Return the Action frame to the source without change
   1483 		 * except that MSB of the Category set to 1.
   1484 		 */
   1485 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Return unknown Action "
   1486 			   "frame back to sender");
   1487 		resp = os_malloc(len);
   1488 		if (resp == NULL)
   1489 			return;
   1490 		os_memcpy(resp, mgmt, len);
   1491 		os_memcpy(resp->da, resp->sa, ETH_ALEN);
   1492 		os_memcpy(resp->sa, hapd->own_addr, ETH_ALEN);
   1493 		os_memcpy(resp->bssid, hapd->own_addr, ETH_ALEN);
   1494 		resp->u.action.category |= 0x80;
   1495 
   1496 		hostapd_drv_send_mlme(hapd, resp, len);
   1497 		os_free(resp);
   1498 	}
   1499 }
   1500 
   1501 
   1502 /**
   1503  * ieee802_11_mgmt - process incoming IEEE 802.11 management frames
   1504  * @hapd: hostapd BSS data structure (the BSS to which the management frame was
   1505  * sent to)
   1506  * @buf: management frame data (starting from IEEE 802.11 header)
   1507  * @len: length of frame data in octets
   1508  * @fi: meta data about received frame (signal level, etc.)
   1509  *
   1510  * Process all incoming IEEE 802.11 management frames. This will be called for
   1511  * each frame received from the kernel driver through wlan#ap interface. In
   1512  * addition, it can be called to re-inserted pending frames (e.g., when using
   1513  * external RADIUS server as an MAC ACL).
   1514  */
   1515 void ieee802_11_mgmt(struct hostapd_data *hapd, const u8 *buf, size_t len,
   1516 		     struct hostapd_frame_info *fi)
   1517 {
   1518 	struct ieee80211_mgmt *mgmt;
   1519 	int broadcast;
   1520 	u16 fc, stype;
   1521 
   1522 	if (len < 24)
   1523 		return;
   1524 
   1525 	mgmt = (struct ieee80211_mgmt *) buf;
   1526 	fc = le_to_host16(mgmt->frame_control);
   1527 	stype = WLAN_FC_GET_STYPE(fc);
   1528 
   1529 	if (stype == WLAN_FC_STYPE_BEACON) {
   1530 		handle_beacon(hapd, mgmt, len, fi);
   1531 		return;
   1532 	}
   1533 
   1534 	broadcast = mgmt->bssid[0] == 0xff && mgmt->bssid[1] == 0xff &&
   1535 		mgmt->bssid[2] == 0xff && mgmt->bssid[3] == 0xff &&
   1536 		mgmt->bssid[4] == 0xff && mgmt->bssid[5] == 0xff;
   1537 
   1538 	if (!broadcast &&
   1539 	    os_memcmp(mgmt->bssid, hapd->own_addr, ETH_ALEN) != 0) {
   1540 		printf("MGMT: BSSID=" MACSTR " not our address\n",
   1541 		       MAC2STR(mgmt->bssid));
   1542 		return;
   1543 	}
   1544 
   1545 
   1546 	if (stype == WLAN_FC_STYPE_PROBE_REQ) {
   1547 		handle_probe_req(hapd, mgmt, len);
   1548 		return;
   1549 	}
   1550 
   1551 	if (os_memcmp(mgmt->da, hapd->own_addr, ETH_ALEN) != 0) {
   1552 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1553 			       HOSTAPD_LEVEL_DEBUG,
   1554 			       "MGMT: DA=" MACSTR " not our address",
   1555 			       MAC2STR(mgmt->da));
   1556 		return;
   1557 	}
   1558 
   1559 	switch (stype) {
   1560 	case WLAN_FC_STYPE_AUTH:
   1561 		wpa_printf(MSG_DEBUG, "mgmt::auth");
   1562 		handle_auth(hapd, mgmt, len);
   1563 		break;
   1564 	case WLAN_FC_STYPE_ASSOC_REQ:
   1565 		wpa_printf(MSG_DEBUG, "mgmt::assoc_req");
   1566 		handle_assoc(hapd, mgmt, len, 0);
   1567 		break;
   1568 	case WLAN_FC_STYPE_REASSOC_REQ:
   1569 		wpa_printf(MSG_DEBUG, "mgmt::reassoc_req");
   1570 		handle_assoc(hapd, mgmt, len, 1);
   1571 		break;
   1572 	case WLAN_FC_STYPE_DISASSOC:
   1573 		wpa_printf(MSG_DEBUG, "mgmt::disassoc");
   1574 		handle_disassoc(hapd, mgmt, len);
   1575 		break;
   1576 	case WLAN_FC_STYPE_DEAUTH:
   1577 		wpa_msg(hapd, MSG_DEBUG, "mgmt::deauth");
   1578 		handle_deauth(hapd, mgmt, len);
   1579 		break;
   1580 	case WLAN_FC_STYPE_ACTION:
   1581 		wpa_printf(MSG_DEBUG, "mgmt::action");
   1582 		handle_action(hapd, mgmt, len);
   1583 		break;
   1584 	default:
   1585 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
   1586 			       HOSTAPD_LEVEL_DEBUG,
   1587 			       "unknown mgmt frame subtype %d", stype);
   1588 		break;
   1589 	}
   1590 }
   1591 
   1592 
   1593 static void handle_auth_cb(struct hostapd_data *hapd,
   1594 			   const struct ieee80211_mgmt *mgmt,
   1595 			   size_t len, int ok)
   1596 {
   1597 	u16 auth_alg, auth_transaction, status_code;
   1598 	struct sta_info *sta;
   1599 
   1600 	if (!ok) {
   1601 		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
   1602 			       HOSTAPD_LEVEL_NOTICE,
   1603 			       "did not acknowledge authentication response");
   1604 		return;
   1605 	}
   1606 
   1607 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
   1608 		printf("handle_auth_cb - too short payload (len=%lu)\n",
   1609 		       (unsigned long) len);
   1610 		return;
   1611 	}
   1612 
   1613 	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
   1614 	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
   1615 	status_code = le_to_host16(mgmt->u.auth.status_code);
   1616 
   1617 	sta = ap_get_sta(hapd, mgmt->da);
   1618 	if (!sta) {
   1619 		printf("handle_auth_cb: STA " MACSTR " not found\n",
   1620 		       MAC2STR(mgmt->da));
   1621 		return;
   1622 	}
   1623 
   1624 	if (status_code == WLAN_STATUS_SUCCESS &&
   1625 	    ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
   1626 	     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
   1627 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1628 			       HOSTAPD_LEVEL_INFO, "authenticated");
   1629 		sta->flags |= WLAN_STA_AUTH;
   1630 	}
   1631 }
   1632 
   1633 
   1634 static void handle_assoc_cb(struct hostapd_data *hapd,
   1635 			    const struct ieee80211_mgmt *mgmt,
   1636 			    size_t len, int reassoc, int ok)
   1637 {
   1638 	u16 status;
   1639 	struct sta_info *sta;
   1640 	int new_assoc = 1;
   1641 	struct ieee80211_ht_capabilities ht_cap;
   1642 
   1643 	if (!ok) {
   1644 		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
   1645 			       HOSTAPD_LEVEL_DEBUG,
   1646 			       "did not acknowledge association response");
   1647 		return;
   1648 	}
   1649 
   1650 	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_resp) :
   1651 				      sizeof(mgmt->u.assoc_resp))) {
   1652 		printf("handle_assoc_cb(reassoc=%d) - too short payload "
   1653 		       "(len=%lu)\n", reassoc, (unsigned long) len);
   1654 		return;
   1655 	}
   1656 
   1657 	if (reassoc)
   1658 		status = le_to_host16(mgmt->u.reassoc_resp.status_code);
   1659 	else
   1660 		status = le_to_host16(mgmt->u.assoc_resp.status_code);
   1661 
   1662 	sta = ap_get_sta(hapd, mgmt->da);
   1663 	if (!sta) {
   1664 		printf("handle_assoc_cb: STA " MACSTR " not found\n",
   1665 		       MAC2STR(mgmt->da));
   1666 		return;
   1667 	}
   1668 
   1669 	if (status != WLAN_STATUS_SUCCESS)
   1670 		goto fail;
   1671 
   1672 	/* Stop previous accounting session, if one is started, and allocate
   1673 	 * new session id for the new session. */
   1674 	accounting_sta_stop(hapd, sta);
   1675 
   1676 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1677 		       HOSTAPD_LEVEL_INFO,
   1678 		       "associated (aid %d)",
   1679 		       sta->aid);
   1680 
   1681 	if (sta->flags & WLAN_STA_ASSOC)
   1682 		new_assoc = 0;
   1683 	sta->flags |= WLAN_STA_ASSOC;
   1684 	if ((!hapd->conf->ieee802_1x && !hapd->conf->wpa) ||
   1685 	    sta->auth_alg == WLAN_AUTH_FT) {
   1686 		/*
   1687 		 * Open, static WEP, or FT protocol; no separate authorization
   1688 		 * step.
   1689 		 */
   1690 		ap_sta_set_authorized(hapd, sta, 1);
   1691 		wpa_msg(hapd->msg_ctx, MSG_INFO,
   1692 			AP_STA_CONNECTED MACSTR, MAC2STR(sta->addr));
   1693 #ifdef ANDROID_BRCM_P2P_PATCH
   1694 		/* Sending the event to parent is required as SSL listens on parent ctrl iface */
   1695 		if(hapd->msg_ctx_parent)
   1696 			wpa_msg(hapd->msg_ctx_parent, MSG_INFO,
   1697 				AP_STA_CONNECTED MACSTR, MAC2STR(sta->addr));
   1698 #endif /* ANDROID_BRCM_P2P_PATCH */
   1699 	}
   1700 
   1701 	if (reassoc)
   1702 		mlme_reassociate_indication(hapd, sta);
   1703 	else
   1704 		mlme_associate_indication(hapd, sta);
   1705 
   1706 #ifdef CONFIG_IEEE80211W
   1707 	sta->sa_query_timed_out = 0;
   1708 #endif /* CONFIG_IEEE80211W */
   1709 
   1710 	/*
   1711 	 * Remove the STA entry in order to make sure the STA PS state gets
   1712 	 * cleared and configuration gets updated in case of reassociation back
   1713 	 * to the same AP.
   1714 	 */
   1715 	hostapd_drv_sta_remove(hapd, sta->addr);
   1716 
   1717 #ifdef CONFIG_IEEE80211N
   1718 	if (sta->flags & WLAN_STA_HT)
   1719 		hostapd_get_ht_capab(hapd, sta->ht_capabilities, &ht_cap);
   1720 #endif /* CONFIG_IEEE80211N */
   1721 
   1722 	if (hostapd_sta_add(hapd, sta->addr, sta->aid, sta->capability,
   1723 			    sta->supported_rates, sta->supported_rates_len,
   1724 			    sta->listen_interval,
   1725 			    sta->flags & WLAN_STA_HT ? &ht_cap : NULL)) {
   1726 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
   1727 			       HOSTAPD_LEVEL_NOTICE,
   1728 			       "Could not add STA to kernel driver");
   1729 	}
   1730 
   1731 	if (sta->flags & WLAN_STA_WDS)
   1732 		hostapd_set_wds_sta(hapd, sta->addr, sta->aid, 1);
   1733 
   1734 	if (sta->eapol_sm == NULL) {
   1735 		/*
   1736 		 * This STA does not use RADIUS server for EAP authentication,
   1737 		 * so bind it to the selected VLAN interface now, since the
   1738 		 * interface selection is not going to change anymore.
   1739 		 */
   1740 		if (ap_sta_bind_vlan(hapd, sta, 0) < 0)
   1741 			goto fail;
   1742 	} else if (sta->vlan_id) {
   1743 		/* VLAN ID already set (e.g., by PMKSA caching), so bind STA */
   1744 		if (ap_sta_bind_vlan(hapd, sta, 0) < 0)
   1745 			goto fail;
   1746 	}
   1747 
   1748 	hostapd_set_sta_flags(hapd, sta);
   1749 
   1750 	if (sta->auth_alg == WLAN_AUTH_FT)
   1751 		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC_FT);
   1752 	else
   1753 		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
   1754 	hapd->new_assoc_sta_cb(hapd, sta, !new_assoc);
   1755 
   1756 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
   1757 
   1758  fail:
   1759 	/* Copy of the association request is not needed anymore */
   1760 	if (sta->last_assoc_req) {
   1761 		os_free(sta->last_assoc_req);
   1762 		sta->last_assoc_req = NULL;
   1763 	}
   1764 }
   1765 
   1766 
   1767 /**
   1768  * ieee802_11_mgmt_cb - Process management frame TX status callback
   1769  * @hapd: hostapd BSS data structure (the BSS from which the management frame
   1770  * was sent from)
   1771  * @buf: management frame data (starting from IEEE 802.11 header)
   1772  * @len: length of frame data in octets
   1773  * @stype: management frame subtype from frame control field
   1774  * @ok: Whether the frame was ACK'ed
   1775  */
   1776 void ieee802_11_mgmt_cb(struct hostapd_data *hapd, const u8 *buf, size_t len,
   1777 			u16 stype, int ok)
   1778 {
   1779 	const struct ieee80211_mgmt *mgmt;
   1780 	mgmt = (const struct ieee80211_mgmt *) buf;
   1781 
   1782 	switch (stype) {
   1783 	case WLAN_FC_STYPE_AUTH:
   1784 		wpa_printf(MSG_DEBUG, "mgmt::auth cb");
   1785 		handle_auth_cb(hapd, mgmt, len, ok);
   1786 		break;
   1787 	case WLAN_FC_STYPE_ASSOC_RESP:
   1788 		wpa_printf(MSG_DEBUG, "mgmt::assoc_resp cb");
   1789 		handle_assoc_cb(hapd, mgmt, len, 0, ok);
   1790 		break;
   1791 	case WLAN_FC_STYPE_REASSOC_RESP:
   1792 		wpa_printf(MSG_DEBUG, "mgmt::reassoc_resp cb");
   1793 		handle_assoc_cb(hapd, mgmt, len, 1, ok);
   1794 		break;
   1795 	case WLAN_FC_STYPE_PROBE_RESP:
   1796 		wpa_printf(MSG_EXCESSIVE, "mgmt::proberesp cb");
   1797 		break;
   1798 	case WLAN_FC_STYPE_DEAUTH:
   1799 		/* ignore */
   1800 		break;
   1801 	case WLAN_FC_STYPE_ACTION:
   1802 		wpa_printf(MSG_DEBUG, "mgmt::action cb");
   1803 		break;
   1804 	default:
   1805 		printf("unknown mgmt cb frame subtype %d\n", stype);
   1806 		break;
   1807 	}
   1808 }
   1809 
   1810 
   1811 int ieee802_11_get_mib(struct hostapd_data *hapd, char *buf, size_t buflen)
   1812 {
   1813 	/* TODO */
   1814 	return 0;
   1815 }
   1816 
   1817 
   1818 int ieee802_11_get_mib_sta(struct hostapd_data *hapd, struct sta_info *sta,
   1819 			   char *buf, size_t buflen)
   1820 {
   1821 	/* TODO */
   1822 	return 0;
   1823 }
   1824 
   1825 
   1826 void hostapd_tx_status(struct hostapd_data *hapd, const u8 *addr,
   1827 		       const u8 *buf, size_t len, int ack)
   1828 {
   1829 	struct sta_info *sta;
   1830 	struct hostapd_iface *iface = hapd->iface;
   1831 
   1832 	sta = ap_get_sta(hapd, addr);
   1833 	if (sta == NULL && iface->num_bss > 1) {
   1834 		size_t j;
   1835 		for (j = 0; j < iface->num_bss; j++) {
   1836 			hapd = iface->bss[j];
   1837 			sta = ap_get_sta(hapd, addr);
   1838 			if (sta)
   1839 				break;
   1840 		}
   1841 	}
   1842 	if (sta == NULL)
   1843 		return;
   1844 	if (sta->flags & WLAN_STA_PENDING_POLL) {
   1845 		wpa_printf(MSG_DEBUG, "STA " MACSTR " %s pending "
   1846 			   "activity poll", MAC2STR(sta->addr),
   1847 			   ack ? "ACKed" : "did not ACK");
   1848 		if (ack)
   1849 			sta->flags &= ~WLAN_STA_PENDING_POLL;
   1850 	}
   1851 
   1852 	ieee802_1x_tx_status(hapd, sta, buf, len, ack);
   1853 }
   1854 
   1855 
   1856 void ieee802_11_rx_from_unknown(struct hostapd_data *hapd, const u8 *src,
   1857 				int wds)
   1858 {
   1859 	struct sta_info *sta;
   1860 
   1861 	sta = ap_get_sta(hapd, src);
   1862 	if (sta && (sta->flags & WLAN_STA_ASSOC)) {
   1863 		if (wds && !(sta->flags & WLAN_STA_WDS)) {
   1864 			wpa_printf(MSG_DEBUG, "Enable 4-address WDS mode for "
   1865 				   "STA " MACSTR " (aid %u)",
   1866 				   MAC2STR(sta->addr), sta->aid);
   1867 			sta->flags |= WLAN_STA_WDS;
   1868 			hostapd_set_wds_sta(hapd, sta->addr, sta->aid, 1);
   1869 		}
   1870 		return;
   1871 	}
   1872 
   1873 	wpa_printf(MSG_DEBUG, "Data/PS-poll frame from not associated STA "
   1874 		   MACSTR, MAC2STR(src));
   1875 	if (src[0] & 0x01) {
   1876 		/* Broadcast bit set in SA?! Ignore the frame silently. */
   1877 		return;
   1878 	}
   1879 
   1880 	if (sta && (sta->flags & WLAN_STA_ASSOC_REQ_OK)) {
   1881 		wpa_printf(MSG_DEBUG, "Association Response to the STA has "
   1882 			   "already been sent, but no TX status yet known - "
   1883 			   "ignore Class 3 frame issue with " MACSTR,
   1884 			   MAC2STR(src));
   1885 		return;
   1886 	}
   1887 
   1888 	if (sta && (sta->flags & WLAN_STA_AUTH))
   1889 		hostapd_drv_sta_disassoc(
   1890 			hapd, src,
   1891 			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
   1892 	else
   1893 		hostapd_drv_sta_deauth(
   1894 			hapd, src,
   1895 			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
   1896 }
   1897 
   1898 
   1899 #endif /* CONFIG_NATIVE_WINDOWS */
   1900