text: /*
   * hostapd / IEEE 802.11 Management
   * Copyright (c) 2002-2013, Jouni Malinen <j@w1.fi>
   *
   * This software may be distributed under the terms of the BSD license.
   * See README for more details.
   */
  
  #include "utils/includes.h"
  
  #ifndef CONFIG_NATIVE_WINDOWS
  
  #include "utils/common.h"
  #include "utils/eloop.h"
  #include "crypto/crypto.h"
  #include "crypto/sha256.h"
  #include "crypto/random.h"
  #include "drivers/driver.h"
  #include "common/ieee802_11_defs.h"
  #include "common/ieee802_11_common.h"
  #include "common/wpa_ctrl.h"
  #include "common/sae.h"
  #include "radius/radius.h"
  #include "radius/radius_client.h"
  #include "p2p/p2p.h"
  #include "wps/wps.h"
  #include "hostapd.h"
  #include "beacon.h"
  #include "ieee802_11_auth.h"
  #include "sta_info.h"
  #include "ieee802_1x.h"
  #include "wpa_auth.h"
  #include "wmm.h"
  #include "ap_list.h"
  #include "accounting.h"
  #include "ap_config.h"
  #include "ap_mlme.h"
  #include "p2p_hostapd.h"
  #include "ap_drv_ops.h"
  #include "wnm_ap.h"
  #include "ieee802_11.h"
  
  
  u8 * hostapd_eid_supp_rates(struct hostapd_data *hapd, u8 *eid)
  {
  	u8 *pos = eid;
  	int i, num, count;
  
  	if (hapd->iface->current_rates == NULL)
  		return eid;
  
  	*pos++ = WLAN_EID_SUPP_RATES;
  	num = hapd->iface->num_rates;
  	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
  		num++;
  	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht)
  		num++;
  	if (num > 8) {
  		/* rest of the rates are encoded in Extended supported
  		 * rates element */
  		num = 8;
  	}
  
  	*pos++ = num;
  	count = 0;
  	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num;
  	     i++) {
  		count++;
  		*pos = hapd->iface->current_rates[i].rate / 5;
  		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
  			*pos |= 0x80;
  		pos++;
  	}
  
  	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht && count < 8) {
  		count++;
  		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
  	}
  
  	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht && count < 8) {
  		count++;
  		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_VHT_PHY;
  	}
  
  	return pos;
  }
  
  
  u8 * hostapd_eid_ext_supp_rates(struct hostapd_data *hapd, u8 *eid)
  {
  	u8 *pos = eid;
  	int i, num, count;
  
  	if (hapd->iface->current_rates == NULL)
  		return eid;
  
  	num = hapd->iface->num_rates;
  	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
  		num++;
  	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht)
  		num++;
  	if (num <= 8)
  		return eid;
  	num -= 8;
  
  	*pos++ = WLAN_EID_EXT_SUPP_RATES;
  	*pos++ = num;
  	count = 0;
  	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num + 8;
  	     i++) {
  		count++;
  		if (count <= 8)
  			continue; /* already in SuppRates IE */
  		*pos = hapd->iface->current_rates[i].rate / 5;
  		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
  			*pos |= 0x80;
  		pos++;
  	}
  
  	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht) {
  		count++;
  		if (count > 8)
  			*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
  	}
  
  	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht) {
  		count++;
  		if (count > 8)
  			*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_VHT_PHY;
  	}
  
  	return pos;
  }
  
  
  u16 hostapd_own_capab_info(struct hostapd_data *hapd, struct sta_info *sta,
  			   int probe)
  {
  	int capab = WLAN_CAPABILITY_ESS;
  	int privacy;
  
  	if (hapd->iface->num_sta_no_short_preamble == 0 &&
  	    hapd->iconf->preamble == SHORT_PREAMBLE)
  		capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
  
  	privacy = hapd->conf->ssid.wep.keys_set;
  
  	if (hapd->conf->ieee802_1x &&
  	    (hapd->conf->default_wep_key_len ||
  	     hapd->conf->individual_wep_key_len))
  		privacy = 1;
  
  	if (hapd->conf->wpa)
  		privacy = 1;
  
  	if (sta) {
  		int policy, def_klen;
  		if (probe && sta->ssid_probe) {
  			policy = sta->ssid_probe->security_policy;
  			def_klen = sta->ssid_probe->wep.default_len;
  		} else {
  			policy = sta->ssid->security_policy;
  			def_klen = sta->ssid->wep.default_len;
  		}
  		privacy = policy != SECURITY_PLAINTEXT;
  		if (policy == SECURITY_IEEE_802_1X && def_klen == 0)
  			privacy = 0;
  	}
  
  	if (privacy)
  		capab |= WLAN_CAPABILITY_PRIVACY;
  
  	if (hapd->iface->current_mode &&
  	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G &&
  	    hapd->iface->num_sta_no_short_slot_time == 0)
  		capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
  
  	return capab;
  }
  
  
  void ieee802_11_print_ssid(char *buf, const u8 *ssid, u8 len)
  {
  	int i;
  	if (len > HOSTAPD_MAX_SSID_LEN)
  		len = HOSTAPD_MAX_SSID_LEN;
  	for (i = 0; i < len; i++) {
  		if (ssid[i] >= 32 && ssid[i] < 127)
  			buf[i] = ssid[i];
  		else
  			buf[i] = '.';
  	}
  	buf[len] = '\0';
  }
  
  
  static u16 auth_shared_key(struct hostapd_data *hapd, struct sta_info *sta,
  			   u16 auth_transaction, const u8 *challenge,
  			   int iswep)
  {
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG,
  		       "authentication (shared key, transaction %d)",
  		       auth_transaction);
  
  	if (auth_transaction == 1) {
  		if (!sta->challenge) {
  			/* Generate a pseudo-random challenge */
  			u8 key[8];
  			struct os_time now;
  			int r;
  			sta->challenge = os_zalloc(WLAN_AUTH_CHALLENGE_LEN);
  			if (sta->challenge == NULL)
  				return WLAN_STATUS_UNSPECIFIED_FAILURE;
  
  			os_get_time(&now);
  			r = os_random();
  			os_memcpy(key, &now.sec, 4);
  			os_memcpy(key + 4, &r, 4);
  			rc4_skip(key, sizeof(key), 0,
  				 sta->challenge, WLAN_AUTH_CHALLENGE_LEN);
  		}
  		return 0;
  	}
  
  	if (auth_transaction != 3)
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  
  	/* Transaction 3 */
  	if (!iswep || !sta->challenge || !challenge ||
  	    os_memcmp(sta->challenge, challenge, WLAN_AUTH_CHALLENGE_LEN)) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO,
  			       "shared key authentication - invalid "
  			       "challenge-response");
  		return WLAN_STATUS_CHALLENGE_FAIL;
  	}
  
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG,
  		       "authentication OK (shared key)");
  #ifdef IEEE80211_REQUIRE_AUTH_ACK
  	/* Station will be marked authenticated if it ACKs the
  	 * authentication reply. */
  #else
  	sta->flags |= WLAN_STA_AUTH;
  	wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
  #endif
  	os_free(sta->challenge);
  	sta->challenge = NULL;
  
  	return 0;
  }
  
  
  static void send_auth_reply(struct hostapd_data *hapd,
  			    const u8 *dst, const u8 *bssid,
  			    u16 auth_alg, u16 auth_transaction, u16 resp,
  			    const u8 *ies, size_t ies_len)
  {
  	struct ieee80211_mgmt *reply;
  	u8 *buf;
  	size_t rlen;
  
  	rlen = IEEE80211_HDRLEN + sizeof(reply->u.auth) + ies_len;
  	buf = os_zalloc(rlen);
  	if (buf == NULL)
  		return;
  
  	reply = (struct ieee80211_mgmt *) buf;
  	reply->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
  					    WLAN_FC_STYPE_AUTH);
  	os_memcpy(reply->da, dst, ETH_ALEN);
  	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
  	os_memcpy(reply->bssid, bssid, ETH_ALEN);
  
  	reply->u.auth.auth_alg = host_to_le16(auth_alg);
  	reply->u.auth.auth_transaction = host_to_le16(auth_transaction);
  	reply->u.auth.status_code = host_to_le16(resp);
  
  	if (ies && ies_len)
  		os_memcpy(reply->u.auth.variable, ies, ies_len);
  
  	wpa_printf(MSG_DEBUG, "authentication reply: STA=" MACSTR
  		   " auth_alg=%d auth_transaction=%d resp=%d (IE len=%lu)",
  		   MAC2STR(dst), auth_alg, auth_transaction,
  		   resp, (unsigned long) ies_len);
  	if (hostapd_drv_send_mlme(hapd, reply, rlen, 0) < 0)
  		perror("send_auth_reply: send");
  
  	os_free(buf);
  }
  
  
  #ifdef CONFIG_IEEE80211R
  static void handle_auth_ft_finish(void *ctx, const u8 *dst, const u8 *bssid,
  				  u16 auth_transaction, u16 status,
  				  const u8 *ies, size_t ies_len)
  {
  	struct hostapd_data *hapd = ctx;
  	struct sta_info *sta;
  
  	send_auth_reply(hapd, dst, bssid, WLAN_AUTH_FT, auth_transaction,
  			status, ies, ies_len);
  
  	if (status != WLAN_STATUS_SUCCESS)
  		return;
  
  	sta = ap_get_sta(hapd, dst);
  	if (sta == NULL)
  		return;
  
  	hostapd_logger(hapd, dst, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG, "authentication OK (FT)");
  	sta->flags |= WLAN_STA_AUTH;
  	mlme_authenticate_indication(hapd, sta);
  }
  #endif /* CONFIG_IEEE80211R */
  
  
  #ifdef CONFIG_SAE
  
  static struct wpabuf * auth_process_sae_commit(struct hostapd_data *hapd,
  					       struct sta_info *sta)
  {
  	struct wpabuf *buf;
  
  	if (hapd->conf->ssid.wpa_passphrase == NULL) {
  		wpa_printf(MSG_DEBUG, "SAE: No password available");
  		return NULL;
  	}
  
  	if (sae_prepare_commit(hapd->own_addr, sta->addr,
  			       (u8 *) hapd->conf->ssid.wpa_passphrase,
  			       os_strlen(hapd->conf->ssid.wpa_passphrase),
  			       sta->sae) < 0) {
  		wpa_printf(MSG_DEBUG, "SAE: Could not pick PWE");
  		return NULL;
  	}
  
  	if (sae_process_commit(sta->sae) < 0) {
  		wpa_printf(MSG_DEBUG, "SAE: Failed to process peer commit");
  		return NULL;
  	}
  
  	buf = wpabuf_alloc(SAE_COMMIT_MAX_LEN);
  	if (buf == NULL)
  		return NULL;
  	sae_write_commit(sta->sae, buf, NULL);
  
  	return buf;
  }
  
  
  static struct wpabuf * auth_build_sae_confirm(struct hostapd_data *hapd,
  					      struct sta_info *sta)
  {
  	struct wpabuf *buf;
  
  	buf = wpabuf_alloc(SAE_CONFIRM_MAX_LEN);
  	if (buf == NULL)
  		return NULL;
  
  	sae_write_confirm(sta->sae, buf);
  
  	return buf;
  }
  
  
  static int use_sae_anti_clogging(struct hostapd_data *hapd)
  {
  	struct sta_info *sta;
  	unsigned int open = 0;
  
  	if (hapd->conf->sae_anti_clogging_threshold == 0)
  		return 1;
  
  	for (sta = hapd->sta_list; sta; sta = sta->next) {
  		if (!sta->sae)
  			continue;
  		if (sta->sae->state != SAE_COMMITTED &&
  		    sta->sae->state != SAE_CONFIRMED)
  			continue;
  		open++;
  		if (open >= hapd->conf->sae_anti_clogging_threshold)
  			return 1;
  	}
  
  	return 0;
  }
  
  
  static int check_sae_token(struct hostapd_data *hapd, const u8 *addr,
  			   const u8 *token, size_t token_len)
  {
  	u8 mac[SHA256_MAC_LEN];
  
  	if (token_len != SHA256_MAC_LEN)
  		return -1;
  	if (hmac_sha256(hapd->sae_token_key, sizeof(hapd->sae_token_key),
  			addr, ETH_ALEN, mac) < 0 ||
  	    os_memcmp(token, mac, SHA256_MAC_LEN) != 0)
  		return -1;
  
  	return 0;
  }
  
  
  static struct wpabuf * auth_build_token_req(struct hostapd_data *hapd,
  					    const u8 *addr)
  {
  	struct wpabuf *buf;
  	u8 *token;
  	struct os_time t;
  
  	os_get_time(&t);
  	if (hapd->last_sae_token_key_update == 0 ||
  	    t.sec > hapd->last_sae_token_key_update + 60) {
  		if (random_get_bytes(hapd->sae_token_key,
  				     sizeof(hapd->sae_token_key)) < 0)
  			return NULL;
  		wpa_hexdump(MSG_DEBUG, "SAE: Updated token key",
  			    hapd->sae_token_key, sizeof(hapd->sae_token_key));
  		hapd->last_sae_token_key_update = t.sec;
  	}
  
  	buf = wpabuf_alloc(SHA256_MAC_LEN);
  	if (buf == NULL)
  		return NULL;
  
  	token = wpabuf_put(buf, SHA256_MAC_LEN);
  	hmac_sha256(hapd->sae_token_key, sizeof(hapd->sae_token_key),
  		    addr, ETH_ALEN, token);
  
  	return buf;
  }
  
  
  static void handle_auth_sae(struct hostapd_data *hapd, struct sta_info *sta,
  			    const struct ieee80211_mgmt *mgmt, size_t len,
  			    u8 auth_transaction)
  {
  	u16 resp = WLAN_STATUS_SUCCESS;
  	struct wpabuf *data = NULL;
  
  	if (!sta->sae) {
  		if (auth_transaction != 1)
  			return;
  		sta->sae = os_zalloc(sizeof(*sta->sae));
  		if (sta->sae == NULL)
  			return;
  		sta->sae->state = SAE_NOTHING;
  	}
  
  	if (auth_transaction == 1) {
  		const u8 *token = NULL;
  		size_t token_len = 0;
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "start SAE authentication (RX commit)");
  		resp = sae_parse_commit(sta->sae, mgmt->u.auth.variable,
  					((const u8 *) mgmt) + len -
  					mgmt->u.auth.variable, &token,
  					&token_len, hapd->conf->sae_groups);
  		if (token && check_sae_token(hapd, sta->addr, token, token_len)
  		    < 0) {
  			wpa_printf(MSG_DEBUG, "SAE: Drop commit message with "
  				   "incorrect token from " MACSTR,
  				   MAC2STR(sta->addr));
  			return;
  		}
  
  		if (resp == WLAN_STATUS_SUCCESS) {
  			if (!token && use_sae_anti_clogging(hapd)) {
  				wpa_printf(MSG_DEBUG, "SAE: Request anti-"
  					   "clogging token from " MACSTR,
  					   MAC2STR(sta->addr));
  				data = auth_build_token_req(hapd, sta->addr);
  				resp = WLAN_STATUS_ANTI_CLOGGING_TOKEN_REQ;
  			} else {
  				data = auth_process_sae_commit(hapd, sta);
  				if (data == NULL)
  					resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  				else
  					sta->sae->state = SAE_COMMITTED;
  			}
  		}
  	} else if (auth_transaction == 2) {
  		if (sta->sae->state != SAE_COMMITTED) {
  			hostapd_logger(hapd, sta->addr,
  				       HOSTAPD_MODULE_IEEE80211,
  				       HOSTAPD_LEVEL_DEBUG,
  				       "SAE confirm before commit");
  			resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
  		}
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "SAE authentication (RX confirm)");
  		if (sae_check_confirm(sta->sae, mgmt->u.auth.variable,
  				       ((u8 *) mgmt) + len -
  				       mgmt->u.auth.variable) < 0) {
  			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  		} else {
  			resp = WLAN_STATUS_SUCCESS;
  			sta->flags |= WLAN_STA_AUTH;
  			wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
  			sta->auth_alg = WLAN_AUTH_SAE;
  			mlme_authenticate_indication(hapd, sta);
  
  			data = auth_build_sae_confirm(hapd, sta);
  			if (data == NULL)
  				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  			else {
  				sta->sae->state = SAE_ACCEPTED;
  				sae_clear_temp_data(sta->sae);
  			}
  		}
  	} else {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "unexpected SAE authentication transaction %u",
  			       auth_transaction);
  		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
  	}
  
  	sta->auth_alg = WLAN_AUTH_SAE;
  
  	send_auth_reply(hapd, mgmt->sa, mgmt->bssid, WLAN_AUTH_SAE,
  			auth_transaction, resp,
  			data ? wpabuf_head(data) : (u8 *) "",
  			data ? wpabuf_len(data) : 0);
  	wpabuf_free(data);
  }
  #endif /* CONFIG_SAE */
  
  
  static void handle_auth(struct hostapd_data *hapd,
  			const struct ieee80211_mgmt *mgmt, size_t len)
  {
  	u16 auth_alg, auth_transaction, status_code;
  	u16 resp = WLAN_STATUS_SUCCESS;
  	struct sta_info *sta = NULL;
  	int res;
  	u16 fc;
  	const u8 *challenge = NULL;
  	u32 session_timeout, acct_interim_interval;
  	int vlan_id = 0;
  	struct hostapd_sta_wpa_psk_short *psk = NULL;
  	u8 resp_ies[2 + WLAN_AUTH_CHALLENGE_LEN];
  	size_t resp_ies_len = 0;
  	char *identity = NULL;
  	char *radius_cui = NULL;
  
  	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
  		printf("handle_auth - too short payload (len=%lu)\n",
  		       (unsigned long) len);
  		return;
  	}
  
  #ifdef CONFIG_TESTING_OPTIONS
  	if (hapd->iconf->ignore_auth_probability > 0.0d &&
  	    drand48() < hapd->iconf->ignore_auth_probability) {
  		wpa_printf(MSG_INFO,
  			   "TESTING: ignoring auth frame from " MACSTR,
  			   MAC2STR(mgmt->sa));
  		return;
  	}
  #endif /* CONFIG_TESTING_OPTIONS */
  
  	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
  	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
  	status_code = le_to_host16(mgmt->u.auth.status_code);
  	fc = le_to_host16(mgmt->frame_control);
  
  	if (len >= IEEE80211_HDRLEN + sizeof(mgmt->u.auth) +
  	    2 + WLAN_AUTH_CHALLENGE_LEN &&
  	    mgmt->u.auth.variable[0] == WLAN_EID_CHALLENGE &&
  	    mgmt->u.auth.variable[1] == WLAN_AUTH_CHALLENGE_LEN)
  		challenge = &mgmt->u.auth.variable[2];
  
  	wpa_printf(MSG_DEBUG, "authentication: STA=" MACSTR " auth_alg=%d "
  		   "auth_transaction=%d status_code=%d wep=%d%s",
  		   MAC2STR(mgmt->sa), auth_alg, auth_transaction,
  		   status_code, !!(fc & WLAN_FC_ISWEP),
  		   challenge ? " challenge" : "");
  
  	if (hapd->tkip_countermeasures) {
  		resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
  		goto fail;
  	}
  
  	if (!(((hapd->conf->auth_algs & WPA_AUTH_ALG_OPEN) &&
  	       auth_alg == WLAN_AUTH_OPEN) ||
  #ifdef CONFIG_IEEE80211R
  	      (hapd->conf->wpa && wpa_key_mgmt_ft(hapd->conf->wpa_key_mgmt) &&
  	       auth_alg == WLAN_AUTH_FT) ||
  #endif /* CONFIG_IEEE80211R */
  #ifdef CONFIG_SAE
  	      (hapd->conf->wpa && wpa_key_mgmt_sae(hapd->conf->wpa_key_mgmt) &&
  	       auth_alg == WLAN_AUTH_SAE) ||
  #endif /* CONFIG_SAE */
  	      ((hapd->conf->auth_algs & WPA_AUTH_ALG_SHARED) &&
  	       auth_alg == WLAN_AUTH_SHARED_KEY))) {
  		printf("Unsupported authentication algorithm (%d)\n",
  		       auth_alg);
  		resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
  		goto fail;
  	}
  
  	if (!(auth_transaction == 1 || auth_alg == WLAN_AUTH_SAE ||
  	      (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 3))) {
  		printf("Unknown authentication transaction number (%d)\n",
  		       auth_transaction);
  		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
  		goto fail;
  	}
  
  	if (os_memcmp(mgmt->sa, hapd->own_addr, ETH_ALEN) == 0) {
  		printf("Station " MACSTR " not allowed to authenticate.\n",
  		       MAC2STR(mgmt->sa));
  		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  		goto fail;
  	}
  
  	res = hostapd_allowed_address(hapd, mgmt->sa, (u8 *) mgmt, len,
  				      &session_timeout,
  				      &acct_interim_interval, &vlan_id,
  				      &psk, &identity, &radius_cui);
  
  	if (res == HOSTAPD_ACL_REJECT) {
  		printf("Station " MACSTR " not allowed to authenticate.\n",
  		       MAC2STR(mgmt->sa));
  		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  		goto fail;
  	}
  	if (res == HOSTAPD_ACL_PENDING) {
  		wpa_printf(MSG_DEBUG, "Authentication frame from " MACSTR
  			   " waiting for an external authentication",
  			   MAC2STR(mgmt->sa));
  		/* Authentication code will re-send the authentication frame
  		 * after it has received (and cached) information from the
  		 * external source. */
  		return;
  	}
  
  	sta = ap_sta_add(hapd, mgmt->sa);
  	if (!sta) {
  		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
  		goto fail;
  	}
  
  	if (vlan_id > 0) {
  		if (!hostapd_vlan_id_valid(hapd->conf->vlan, vlan_id)) {
  			hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
  				       HOSTAPD_LEVEL_INFO, "Invalid VLAN ID "
  				       "%d received from RADIUS server",
  				       vlan_id);
  			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  			goto fail;
  		}
  		sta->vlan_id = vlan_id;
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
  			       HOSTAPD_LEVEL_INFO, "VLAN ID %d", sta->vlan_id);
  	}
  
  	hostapd_free_psk_list(sta->psk);
  	if (hapd->conf->wpa_psk_radius != PSK_RADIUS_IGNORED) {
  		sta->psk = psk;
  		psk = NULL;
  	} else {
  		sta->psk = NULL;
  	}
  
  	sta->identity = identity;
  	identity = NULL;
  	sta->radius_cui = radius_cui;
  	radius_cui = NULL;
  
  	sta->flags &= ~WLAN_STA_PREAUTH;
  	ieee802_1x_notify_pre_auth(sta->eapol_sm, 0);
  
  	if (hapd->conf->acct_interim_interval == 0 && acct_interim_interval)
  		sta->acct_interim_interval = acct_interim_interval;
  	if (res == HOSTAPD_ACL_ACCEPT_TIMEOUT)
  		ap_sta_session_timeout(hapd, sta, session_timeout);
  	else
  		ap_sta_no_session_timeout(hapd, sta);
  
  	switch (auth_alg) {
  	case WLAN_AUTH_OPEN:
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "authentication OK (open system)");
  #ifdef IEEE80211_REQUIRE_AUTH_ACK
  		/* Station will be marked authenticated if it ACKs the
  		 * authentication reply. */
  #else
  		sta->flags |= WLAN_STA_AUTH;
  		wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
  		sta->auth_alg = WLAN_AUTH_OPEN;
  		mlme_authenticate_indication(hapd, sta);
  #endif
  		break;
  	case WLAN_AUTH_SHARED_KEY:
  		resp = auth_shared_key(hapd, sta, auth_transaction, challenge,
  				       fc & WLAN_FC_ISWEP);
  		sta->auth_alg = WLAN_AUTH_SHARED_KEY;
  		mlme_authenticate_indication(hapd, sta);
  		if (sta->challenge && auth_transaction == 1) {
  			resp_ies[0] = WLAN_EID_CHALLENGE;
  			resp_ies[1] = WLAN_AUTH_CHALLENGE_LEN;
  			os_memcpy(resp_ies + 2, sta->challenge,
  				  WLAN_AUTH_CHALLENGE_LEN);
  			resp_ies_len = 2 + WLAN_AUTH_CHALLENGE_LEN;
  		}
  		break;
  #ifdef CONFIG_IEEE80211R
  	case WLAN_AUTH_FT:
  		sta->auth_alg = WLAN_AUTH_FT;
  		if (sta->wpa_sm == NULL)
  			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
  							sta->addr, NULL);
  		if (sta->wpa_sm == NULL) {
  			wpa_printf(MSG_DEBUG, "FT: Failed to initialize WPA "
  				   "state machine");
  			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  			goto fail;
  		}
  		wpa_ft_process_auth(sta->wpa_sm, mgmt->bssid,
  				    auth_transaction, mgmt->u.auth.variable,
  				    len - IEEE80211_HDRLEN -
  				    sizeof(mgmt->u.auth),
  				    handle_auth_ft_finish, hapd);
  		/* handle_auth_ft_finish() callback will complete auth. */
  		return;
  #endif /* CONFIG_IEEE80211R */
  #ifdef CONFIG_SAE
  	case WLAN_AUTH_SAE:
  		handle_auth_sae(hapd, sta, mgmt, len, auth_transaction);
  		return;
  #endif /* CONFIG_SAE */
  	}
  
   fail:
  	os_free(identity);
  	os_free(radius_cui);
  	hostapd_free_psk_list(psk);
  
  	send_auth_reply(hapd, mgmt->sa, mgmt->bssid, auth_alg,
  			auth_transaction + 1, resp, resp_ies, resp_ies_len);
  }
  
  
  static int hostapd_get_aid(struct hostapd_data *hapd, struct sta_info *sta)
  {
  	int i, j = 32, aid;
  
  	/* get a unique AID */
  	if (sta->aid > 0) {
  		wpa_printf(MSG_DEBUG, "  old AID %d", sta->aid);
  		return 0;
  	}
  
  	for (i = 0; i < AID_WORDS; i++) {
  		if (hapd->sta_aid[i] == (u32) -1)
  			continue;
  		for (j = 0; j < 32; j++) {
  			if (!(hapd->sta_aid[i] & BIT(j)))
  				break;
  		}
  		if (j < 32)
  			break;
  	}
  	if (j == 32)
  		return -1;
  	aid = i * 32 + j + 1;
  	if (aid > 2007)
  		return -1;
  
  	sta->aid = aid;
  	hapd->sta_aid[i] |= BIT(j);
  	wpa_printf(MSG_DEBUG, "  new AID %d", sta->aid);
  	return 0;
  }
  
  
  static u16 check_ssid(struct hostapd_data *hapd, struct sta_info *sta,
  		      const u8 *ssid_ie, size_t ssid_ie_len)
  {
  	if (ssid_ie == NULL)
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  
  	if (ssid_ie_len != hapd->conf->ssid.ssid_len ||
  	    os_memcmp(ssid_ie, hapd->conf->ssid.ssid, ssid_ie_len) != 0) {
  		char ssid_txt[33];
  		ieee802_11_print_ssid(ssid_txt, ssid_ie, ssid_ie_len);
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO,
  			       "Station tried to associate with unknown SSID "
  			       "'%s'", ssid_txt);
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  	}
  
  	return WLAN_STATUS_SUCCESS;
  }
  
  
  static u16 check_wmm(struct hostapd_data *hapd, struct sta_info *sta,
  		     const u8 *wmm_ie, size_t wmm_ie_len)
  {
  	sta->flags &= ~WLAN_STA_WMM;
  	sta->qosinfo = 0;
  	if (wmm_ie && hapd->conf->wmm_enabled) {
  		struct wmm_information_element *wmm;
  
  		if (!hostapd_eid_wmm_valid(hapd, wmm_ie, wmm_ie_len)) {
  			hostapd_logger(hapd, sta->addr,
  				       HOSTAPD_MODULE_WPA,
  				       HOSTAPD_LEVEL_DEBUG,
  				       "invalid WMM element in association "
  				       "request");
  			return WLAN_STATUS_UNSPECIFIED_FAILURE;
  		}
  
  		sta->flags |= WLAN_STA_WMM;
  		wmm = (struct wmm_information_element *) wmm_ie;
  		sta->qosinfo = wmm->qos_info;
  	}
  	return WLAN_STATUS_SUCCESS;
  }
  
  
  static u16 copy_supp_rates(struct hostapd_data *hapd, struct sta_info *sta,
  			   struct ieee802_11_elems *elems)
  {
  	if (!elems->supp_rates) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "No supported rates element in AssocReq");
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  	}
  
  	if (elems->supp_rates_len + elems->ext_supp_rates_len >
  	    sizeof(sta->supported_rates)) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "Invalid supported rates element length %d+%d",
  			       elems->supp_rates_len,
  			       elems->ext_supp_rates_len);
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  	}
  
  	sta->supported_rates_len = merge_byte_arrays(
  		sta->supported_rates, sizeof(sta->supported_rates),
  		elems->supp_rates, elems->supp_rates_len,
  		elems->ext_supp_rates, elems->ext_supp_rates_len);
  
  	return WLAN_STATUS_SUCCESS;
  }
  
  
  static u16 check_assoc_ies(struct hostapd_data *hapd, struct sta_info *sta,
  			   const u8 *ies, size_t ies_len, int reassoc)
  {
  	struct ieee802_11_elems elems;
  	u16 resp;
  	const u8 *wpa_ie;
  	size_t wpa_ie_len;
  	const u8 *p2p_dev_addr = NULL;
  
  	if (ieee802_11_parse_elems(ies, ies_len, &elems, 1) == ParseFailed) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "Station sent an invalid "
  			       "association request");
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  	}
  
  	resp = check_ssid(hapd, sta, elems.ssid, elems.ssid_len);
  	if (resp != WLAN_STATUS_SUCCESS)
  		return resp;
  	resp = check_wmm(hapd, sta, elems.wmm, elems.wmm_len);
  	if (resp != WLAN_STATUS_SUCCESS)
  		return resp;
  	resp = copy_supp_rates(hapd, sta, &elems);
  	if (resp != WLAN_STATUS_SUCCESS)
  		return resp;
  #ifdef CONFIG_IEEE80211N
  	resp = copy_sta_ht_capab(hapd, sta, elems.ht_capabilities,
  				 elems.ht_capabilities_len);
  	if (resp != WLAN_STATUS_SUCCESS)
  		return resp;
  	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht &&
  	    !(sta->flags & WLAN_STA_HT)) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "Station does not support "
  			       "mandatory HT PHY - reject association");
  		return WLAN_STATUS_ASSOC_DENIED_NO_HT;
  	}
  #endif /* CONFIG_IEEE80211N */
  
  #ifdef CONFIG_IEEE80211AC
  	resp = copy_sta_vht_capab(hapd, sta, elems.vht_capabilities,
  				  elems.vht_capabilities_len);
  	if (resp != WLAN_STATUS_SUCCESS)
  		return resp;
  	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht &&
  	    !(sta->flags & WLAN_STA_VHT)) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "Station does not support "
  			       "mandatory VHT PHY - reject association");
  		return WLAN_STATUS_UNSPECIFIED_FAILURE;
  	}
  #endif /* CONFIG_IEEE80211AC */
  
  #ifdef CONFIG_P2P
  	if (elems.p2p) {
  		wpabuf_free(sta->p2p_ie);
  		sta->p2p_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
  							  P2P_IE_VENDOR_TYPE);
  		if (sta->p2p_ie)
  			p2p_dev_addr = p2p_get_go_dev_addr(sta->p2p_ie);
  	} else {
  		wpabuf_free(sta->p2p_ie);
  		sta->p2p_ie = NULL;
  	}
  #endif /* CONFIG_P2P */
  
  	if ((hapd->conf->wpa & WPA_PROTO_RSN) && elems.rsn_ie) {
  		wpa_ie = elems.rsn_ie;
  		wpa_ie_len = elems.rsn_ie_len;
  	} else if ((hapd->conf->wpa & WPA_PROTO_WPA) &&
  		   elems.wpa_ie) {
  		wpa_ie = elems.wpa_ie;
  		wpa_ie_len = elems.wpa_ie_len;
  	} else {
  		wpa_ie = NULL;
  		wpa_ie_len = 0;
  	}
  
  #ifdef CONFIG_WPS
  	sta->flags &= ~(WLAN_STA_WPS | WLAN_STA_MAYBE_WPS | WLAN_STA_WPS2);
  	if (hapd->conf->wps_state && elems.wps_ie) {
  		wpa_printf(MSG_DEBUG, "STA included WPS IE in (Re)Association "
  			   "Request - assume WPS is used");
  		sta->flags |= WLAN_STA_WPS;
  		wpabuf_free(sta->wps_ie);
  		sta->wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
  							  WPS_IE_VENDOR_TYPE);
  		if (sta->wps_ie && wps_is_20(sta->wps_ie)) {
  			wpa_printf(MSG_DEBUG, "WPS: STA supports WPS 2.0");
  			sta->flags |= WLAN_STA_WPS2;
  		}
  		wpa_ie = NULL;
  		wpa_ie_len = 0;
  		if (sta->wps_ie && wps_validate_assoc_req(sta->wps_ie) < 0) {
  			wpa_printf(MSG_DEBUG, "WPS: Invalid WPS IE in "
  				   "(Re)Association Request - reject");
  			return WLAN_STATUS_INVALID_IE;
  		}
  	} else if (hapd->conf->wps_state && wpa_ie == NULL) {
  		wpa_printf(MSG_DEBUG, "STA did not include WPA/RSN IE in "
  			   "(Re)Association Request - possible WPS use");
  		sta->flags |= WLAN_STA_MAYBE_WPS;
  	} else
  #endif /* CONFIG_WPS */
  	if (hapd->conf->wpa && wpa_ie == NULL) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO,
  			       "No WPA/RSN IE in association request");
  		return WLAN_STATUS_INVALID_IE;
  	}
  
  	if (hapd->conf->wpa && wpa_ie) {
  		int res;
  		wpa_ie -= 2;
  		wpa_ie_len += 2;
  		if (sta->wpa_sm == NULL)
  			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
  							sta->addr,
  							p2p_dev_addr);
  		if (sta->wpa_sm == NULL) {
  			wpa_printf(MSG_WARNING, "Failed to initialize WPA "
  				   "state machine");
  			return WLAN_STATUS_UNSPECIFIED_FAILURE;
  		}
  		res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
  					  wpa_ie, wpa_ie_len,
  					  elems.mdie, elems.mdie_len);
  		if (res == WPA_INVALID_GROUP)
  			resp = WLAN_STATUS_GROUP_CIPHER_NOT_VALID;
  		else if (res == WPA_INVALID_PAIRWISE)
  			resp = WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
  		else if (res == WPA_INVALID_AKMP)
  			resp = WLAN_STATUS_AKMP_NOT_VALID;
  		else if (res == WPA_ALLOC_FAIL)
  			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
  #ifdef CONFIG_IEEE80211W
  		else if (res == WPA_MGMT_FRAME_PROTECTION_VIOLATION)
  			resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
  		else if (res == WPA_INVALID_MGMT_GROUP_CIPHER)
  			resp = WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
  #endif /* CONFIG_IEEE80211W */
  		else if (res == WPA_INVALID_MDIE)
  			resp = WLAN_STATUS_INVALID_MDIE;
  		else if (res != WPA_IE_OK)
  			resp = WLAN_STATUS_INVALID_IE;
  		if (resp != WLAN_STATUS_SUCCESS)
  			return resp;
  #ifdef CONFIG_IEEE80211W
  		if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
  		    sta->sa_query_count > 0)
  			ap_check_sa_query_timeout(hapd, sta);
  		if ((sta->flags & WLAN_STA_MFP) && !sta->sa_query_timed_out &&
  		    (!reassoc || sta->auth_alg != WLAN_AUTH_FT)) {
  			/*
  			 * STA has already been associated with MFP and SA
  			 * Query timeout has not been reached. Reject the
  			 * association attempt temporarily and start SA Query,
  			 * if one is not pending.
  			 */
  
  			if (sta->sa_query_count == 0)
  				ap_sta_start_sa_query(hapd, sta);
  
  			return WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY;
  		}
  
  		if (wpa_auth_uses_mfp(sta->wpa_sm))
  			sta->flags |= WLAN_STA_MFP;
  		else
  			sta->flags &= ~WLAN_STA_MFP;
  #endif /* CONFIG_IEEE80211W */
  
  #ifdef CONFIG_IEEE80211R
  		if (sta->auth_alg == WLAN_AUTH_FT) {
  			if (!reassoc) {
  				wpa_printf(MSG_DEBUG, "FT: " MACSTR " tried "
  					   "to use association (not "
  					   "re-association) with FT auth_alg",
  					   MAC2STR(sta->addr));
  				return WLAN_STATUS_UNSPECIFIED_FAILURE;
  			}
  
  			resp = wpa_ft_validate_reassoc(sta->wpa_sm, ies,
  						       ies_len);
  			if (resp != WLAN_STATUS_SUCCESS)
  				return resp;
  		}
  #endif /* CONFIG_IEEE80211R */
  
  #ifdef CONFIG_SAE
  		if (wpa_auth_uses_sae(sta->wpa_sm) &&
  		    sta->auth_alg != WLAN_AUTH_SAE) {
  			wpa_printf(MSG_DEBUG, "SAE: " MACSTR " tried to use "
  				   "SAE AKM after non-SAE auth_alg %u",
  				   MAC2STR(sta->addr), sta->auth_alg);
  			return WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
  		}
  #endif /* CONFIG_SAE */
  
  #ifdef CONFIG_IEEE80211N
  		if ((sta->flags & (WLAN_STA_HT | WLAN_STA_VHT)) &&
  		    wpa_auth_get_pairwise(sta->wpa_sm) == WPA_CIPHER_TKIP) {
  			hostapd_logger(hapd, sta->addr,
  				       HOSTAPD_MODULE_IEEE80211,
  				       HOSTAPD_LEVEL_INFO,
  				       "Station tried to use TKIP with HT "
  				       "association");
  			return WLAN_STATUS_CIPHER_REJECTED_PER_POLICY;
  		}
  #endif /* CONFIG_IEEE80211N */
  	} else
  		wpa_auth_sta_no_wpa(sta->wpa_sm);
  
  #ifdef CONFIG_P2P
  	p2p_group_notif_assoc(hapd->p2p_group, sta->addr, ies, ies_len);
  #endif /* CONFIG_P2P */
  
  #ifdef CONFIG_HS20
  	wpabuf_free(sta->hs20_ie);
  	if (elems.hs20 && elems.hs20_len > 4) {
  		sta->hs20_ie = wpabuf_alloc_copy(elems.hs20 + 4,
  						 elems.hs20_len - 4);
  	} else
  		sta->hs20_ie = NULL;
  #endif /* CONFIG_HS20 */
  
  	return WLAN_STATUS_SUCCESS;
  }
  
  
  static void send_deauth(struct hostapd_data *hapd, const u8 *addr,
  			u16 reason_code)
  {
  	int send_len;
  	struct ieee80211_mgmt reply;
  
  	os_memset(&reply, 0, sizeof(reply));
  	reply.frame_control =
  		IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_DEAUTH);
  	os_memcpy(reply.da, addr, ETH_ALEN);
  	os_memcpy(reply.sa, hapd->own_addr, ETH_ALEN);
  	os_memcpy(reply.bssid, hapd->own_addr, ETH_ALEN);
  
  	send_len = IEEE80211_HDRLEN + sizeof(reply.u.deauth);
  	reply.u.deauth.reason_code = host_to_le16(reason_code);
  
  	if (hostapd_drv_send_mlme(hapd, &reply, send_len, 0) < 0)
  		wpa_printf(MSG_INFO, "Failed to send deauth: %s",
  			   strerror(errno));
  }
  
  
  static void send_assoc_resp(struct hostapd_data *hapd, struct sta_info *sta,
  			    u16 status_code, int reassoc, const u8 *ies,
  			    size_t ies_len)
  {
  	int send_len;
  	u8 buf[sizeof(struct ieee80211_mgmt) + 1024];
  	struct ieee80211_mgmt *reply;
  	u8 *p;
  
  	os_memset(buf, 0, sizeof(buf));
  	reply = (struct ieee80211_mgmt *) buf;
  	reply->frame_control =
  		IEEE80211_FC(WLAN_FC_TYPE_MGMT,
  			     (reassoc ? WLAN_FC_STYPE_REASSOC_RESP :
  			      WLAN_FC_STYPE_ASSOC_RESP));
  	os_memcpy(reply->da, sta->addr, ETH_ALEN);
  	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
  	os_memcpy(reply->bssid, hapd->own_addr, ETH_ALEN);
  
  	send_len = IEEE80211_HDRLEN;
  	send_len += sizeof(reply->u.assoc_resp);
  	reply->u.assoc_resp.capab_info =
  		host_to_le16(hostapd_own_capab_info(hapd, sta, 0));
  	reply->u.assoc_resp.status_code = host_to_le16(status_code);
  	reply->u.assoc_resp.aid = host_to_le16((sta ? sta->aid : 0)
  					       | BIT(14) | BIT(15));
  	/* Supported rates */
  	p = hostapd_eid_supp_rates(hapd, reply->u.assoc_resp.variable);
  	/* Extended supported rates */
  	p = hostapd_eid_ext_supp_rates(hapd, p);
  
  #ifdef CONFIG_IEEE80211R
  	if (status_code == WLAN_STATUS_SUCCESS) {
  		/* IEEE 802.11r: Mobility Domain Information, Fast BSS
  		 * Transition Information, RSN, [RIC Response] */
  		p = wpa_sm_write_assoc_resp_ies(sta->wpa_sm, p,
  						buf + sizeof(buf) - p,
  						sta->auth_alg, ies, ies_len);
  	}
  #endif /* CONFIG_IEEE80211R */
  
  #ifdef CONFIG_IEEE80211W
  	if (status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY)
  		p = hostapd_eid_assoc_comeback_time(hapd, sta, p);
  #endif /* CONFIG_IEEE80211W */
  
  #ifdef CONFIG_IEEE80211N
  	p = hostapd_eid_ht_capabilities(hapd, p);
  	p = hostapd_eid_ht_operation(hapd, p);
  #endif /* CONFIG_IEEE80211N */
  
  #ifdef CONFIG_IEEE80211AC
  	p = hostapd_eid_vht_capabilities(hapd, p);
  	p = hostapd_eid_vht_operation(hapd, p);
  #endif /* CONFIG_IEEE80211AC */
  
  	p = hostapd_eid_ext_capab(hapd, p);
  	p = hostapd_eid_bss_max_idle_period(hapd, p);
  
  	if (sta->flags & WLAN_STA_WMM)
  		p = hostapd_eid_wmm(hapd, p);
  
  #ifdef CONFIG_WPS
  	if ((sta->flags & WLAN_STA_WPS) ||
  	    ((sta->flags & WLAN_STA_MAYBE_WPS) && hapd->conf->wpa)) {
  		struct wpabuf *wps = wps_build_assoc_resp_ie();
  		if (wps) {
  			os_memcpy(p, wpabuf_head(wps), wpabuf_len(wps));
  			p += wpabuf_len(wps);
  			wpabuf_free(wps);
  		}
  	}
  #endif /* CONFIG_WPS */
  
  #ifdef CONFIG_P2P
  	if (sta->p2p_ie) {
  		struct wpabuf *p2p_resp_ie;
  		enum p2p_status_code status;
  		switch (status_code) {
  		case WLAN_STATUS_SUCCESS:
  			status = P2P_SC_SUCCESS;
  			break;
  		case WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA:
  			status = P2P_SC_FAIL_LIMIT_REACHED;
  			break;
  		default:
  			status = P2P_SC_FAIL_INVALID_PARAMS;
  			break;
  		}
  		p2p_resp_ie = p2p_group_assoc_resp_ie(hapd->p2p_group, status);
  		if (p2p_resp_ie) {
  			os_memcpy(p, wpabuf_head(p2p_resp_ie),
  				  wpabuf_len(p2p_resp_ie));
  			p += wpabuf_len(p2p_resp_ie);
  			wpabuf_free(p2p_resp_ie);
  		}
  	}
  #endif /* CONFIG_P2P */
  
  #ifdef CONFIG_P2P_MANAGER
  	if (hapd->conf->p2p & P2P_MANAGE)
  		p = hostapd_eid_p2p_manage(hapd, p);
  #endif /* CONFIG_P2P_MANAGER */
  
  	send_len += p - reply->u.assoc_resp.variable;
  
  	if (hostapd_drv_send_mlme(hapd, reply, send_len, 0) < 0)
  		wpa_printf(MSG_INFO, "Failed to send assoc resp: %s",
  			   strerror(errno));
  }
  
  
  static void handle_assoc(struct hostapd_data *hapd,
  			 const struct ieee80211_mgmt *mgmt, size_t len,
  			 int reassoc)
  {
  	u16 capab_info, listen_interval;
  	u16 resp = WLAN_STATUS_SUCCESS;
  	const u8 *pos;
  	int left, i;
  	struct sta_info *sta;
  
  	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_req) :
  				      sizeof(mgmt->u.assoc_req))) {
  		printf("handle_assoc(reassoc=%d) - too short payload (len=%lu)"
  		       "\n", reassoc, (unsigned long) len);
  		return;
  	}
  
  #ifdef CONFIG_TESTING_OPTIONS
  	if (reassoc) {
  		if (hapd->iconf->ignore_reassoc_probability > 0.0d &&
  		    drand48() < hapd->iconf->ignore_reassoc_probability) {
  			wpa_printf(MSG_INFO,
  				   "TESTING: ignoring reassoc request from "
  				   MACSTR, MAC2STR(mgmt->sa));
  			return;
  		}
  	} else {
  		if (hapd->iconf->ignore_assoc_probability > 0.0d &&
  		    drand48() < hapd->iconf->ignore_assoc_probability) {
  			wpa_printf(MSG_INFO,
  				   "TESTING: ignoring assoc request from "
  				   MACSTR, MAC2STR(mgmt->sa));
  			return;
  		}
  	}
  #endif /* CONFIG_TESTING_OPTIONS */
  
  	if (reassoc) {
  		capab_info = le_to_host16(mgmt->u.reassoc_req.capab_info);
  		listen_interval = le_to_host16(
  			mgmt->u.reassoc_req.listen_interval);
  		wpa_printf(MSG_DEBUG, "reassociation request: STA=" MACSTR
  			   " capab_info=0x%02x listen_interval=%d current_ap="
  			   MACSTR,
  			   MAC2STR(mgmt->sa), capab_info, listen_interval,
  			   MAC2STR(mgmt->u.reassoc_req.current_ap));
  		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.reassoc_req));
  		pos = mgmt->u.reassoc_req.variable;
  	} else {
  		capab_info = le_to_host16(mgmt->u.assoc_req.capab_info);
  		listen_interval = le_to_host16(
  			mgmt->u.assoc_req.listen_interval);
  		wpa_printf(MSG_DEBUG, "association request: STA=" MACSTR
  			   " capab_info=0x%02x listen_interval=%d",
  			   MAC2STR(mgmt->sa), capab_info, listen_interval);
  		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.assoc_req));
  		pos = mgmt->u.assoc_req.variable;
  	}
  
  	sta = ap_get_sta(hapd, mgmt->sa);
  #ifdef CONFIG_IEEE80211R
  	if (sta && sta->auth_alg == WLAN_AUTH_FT &&
  	    (sta->flags & WLAN_STA_AUTH) == 0) {
  		wpa_printf(MSG_DEBUG, "FT: Allow STA " MACSTR " to associate "
  			   "prior to authentication since it is using "
  			   "over-the-DS FT", MAC2STR(mgmt->sa));
  	} else
  #endif /* CONFIG_IEEE80211R */
  	if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "Station tried to "
  			       "associate before authentication "
  			       "(aid=%d flags=0x%x)",
  			       sta ? sta->aid : -1,
  			       sta ? sta->flags : 0);
  		send_deauth(hapd, mgmt->sa,
  			    WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA);
  		return;
  	}
  
  	if (hapd->tkip_countermeasures) {
  		resp = WLAN_REASON_MICHAEL_MIC_FAILURE;
  		goto fail;
  	}
  
  	if (listen_interval > hapd->conf->max_listen_interval) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "Too large Listen Interval (%d)",
  			       listen_interval);
  		resp = WLAN_STATUS_ASSOC_DENIED_LISTEN_INT_TOO_LARGE;
  		goto fail;
  	}
  
  	/* followed by SSID and Supported rates; and HT capabilities if 802.11n
  	 * is used */
  	resp = check_assoc_ies(hapd, sta, pos, left, reassoc);
  	if (resp != WLAN_STATUS_SUCCESS)
  		goto fail;
  
  	if (hostapd_get_aid(hapd, sta) < 0) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "No room for more AIDs");
  		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
  		goto fail;
  	}
  
  	sta->capability = capab_info;
  	sta->listen_interval = listen_interval;
  
  	if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G)
  		sta->flags |= WLAN_STA_NONERP;
  	for (i = 0; i < sta->supported_rates_len; i++) {
  		if ((sta->supported_rates[i] & 0x7f) > 22) {
  			sta->flags &= ~WLAN_STA_NONERP;
  			break;
  		}
  	}
  	if (sta->flags & WLAN_STA_NONERP && !sta->nonerp_set) {
  		sta->nonerp_set = 1;
  		hapd->iface->num_sta_non_erp++;
  		if (hapd->iface->num_sta_non_erp == 1)
  			ieee802_11_set_beacons(hapd->iface);
  	}
  
  	if (!(sta->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME) &&
  	    !sta->no_short_slot_time_set) {
  		sta->no_short_slot_time_set = 1;
  		hapd->iface->num_sta_no_short_slot_time++;
  		if (hapd->iface->current_mode->mode ==
  		    HOSTAPD_MODE_IEEE80211G &&
  		    hapd->iface->num_sta_no_short_slot_time == 1)
  			ieee802_11_set_beacons(hapd->iface);
  	}
  
  	if (sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
  		sta->flags |= WLAN_STA_SHORT_PREAMBLE;
  	else
  		sta->flags &= ~WLAN_STA_SHORT_PREAMBLE;
  
  	if (!(sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE) &&
  	    !sta->no_short_preamble_set) {
  		sta->no_short_preamble_set = 1;
  		hapd->iface->num_sta_no_short_preamble++;
  		if (hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
  		    && hapd->iface->num_sta_no_short_preamble == 1)
  			ieee802_11_set_beacons(hapd->iface);
  	}
  
  #ifdef CONFIG_IEEE80211N
  	update_ht_state(hapd, sta);
  #endif /* CONFIG_IEEE80211N */
  
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG,
  		       "association OK (aid %d)", sta->aid);
  	/* Station will be marked associated, after it acknowledges AssocResp
  	 */
  	sta->flags |= WLAN_STA_ASSOC_REQ_OK;
  
  #ifdef CONFIG_IEEE80211W
  	if ((sta->flags & WLAN_STA_MFP) && sta->sa_query_timed_out) {
  		wpa_printf(MSG_DEBUG, "Allowing %sassociation after timed out "
  			   "SA Query procedure", reassoc ? "re" : "");
  		/* TODO: Send a protected Disassociate frame to the STA using
  		 * the old key and Reason Code "Previous Authentication no
  		 * longer valid". Make sure this is only sent protected since
  		 * unprotected frame would be received by the STA that is now
  		 * trying to associate.
  		 */
  	}
  #endif /* CONFIG_IEEE80211W */
  
  	if (reassoc) {
  		os_memcpy(sta->previous_ap, mgmt->u.reassoc_req.current_ap,
  			  ETH_ALEN);
  	}
  
  	if (sta->last_assoc_req)
  		os_free(sta->last_assoc_req);
  	sta->last_assoc_req = os_malloc(len);
  	if (sta->last_assoc_req)
  		os_memcpy(sta->last_assoc_req, mgmt, len);
  
  	/* Make sure that the previously registered inactivity timer will not
  	 * remove the STA immediately. */
  	sta->timeout_next = STA_NULLFUNC;
  
   fail:
  	send_assoc_resp(hapd, sta, resp, reassoc, pos, left);
  }
  
  
  static void handle_disassoc(struct hostapd_data *hapd,
  			    const struct ieee80211_mgmt *mgmt, size_t len)
  {
  	struct sta_info *sta;
  
  	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.disassoc)) {
  		printf("handle_disassoc - too short payload (len=%lu)\n",
  		       (unsigned long) len);
  		return;
  	}
  
  	wpa_printf(MSG_DEBUG, "disassocation: STA=" MACSTR " reason_code=%d",
  		   MAC2STR(mgmt->sa),
  		   le_to_host16(mgmt->u.disassoc.reason_code));
  
  	sta = ap_get_sta(hapd, mgmt->sa);
  	if (sta == NULL) {
  		printf("Station " MACSTR " trying to disassociate, but it "
  		       "is not associated.\n", MAC2STR(mgmt->sa));
  		return;
  	}
  
  	ap_sta_set_authorized(hapd, sta, 0);
  	sta->flags &= ~(WLAN_STA_ASSOC | WLAN_STA_ASSOC_REQ_OK);
  	wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_INFO, "disassociated");
  	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
  	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
  	/* Stop Accounting and IEEE 802.1X sessions, but leave the STA
  	 * authenticated. */
  	accounting_sta_stop(hapd, sta);
  	ieee802_1x_free_station(sta);
  	hostapd_drv_sta_remove(hapd, sta->addr);
  
  	if (sta->timeout_next == STA_NULLFUNC ||
  	    sta->timeout_next == STA_DISASSOC) {
  		sta->timeout_next = STA_DEAUTH;
  		eloop_cancel_timeout(ap_handle_timer, hapd, sta);
  		eloop_register_timeout(AP_DEAUTH_DELAY, 0, ap_handle_timer,
  				       hapd, sta);
  	}
  
  	mlme_disassociate_indication(
  		hapd, sta, le_to_host16(mgmt->u.disassoc.reason_code));
  }
  
  
  static void handle_deauth(struct hostapd_data *hapd,
  			  const struct ieee80211_mgmt *mgmt, size_t len)
  {
  	struct sta_info *sta;
  
  	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.deauth)) {
  		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "handle_deauth - too short "
  			"payload (len=%lu)", (unsigned long) len);
  		return;
  	}
  
  	wpa_msg(hapd->msg_ctx, MSG_DEBUG, "deauthentication: STA=" MACSTR
  		" reason_code=%d",
  		MAC2STR(mgmt->sa), le_to_host16(mgmt->u.deauth.reason_code));
  
  	sta = ap_get_sta(hapd, mgmt->sa);
  	if (sta == NULL) {
  		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "Station " MACSTR " trying "
  			"to deauthenticate, but it is not authenticated",
  			MAC2STR(mgmt->sa));
  		return;
  	}
  
  	ap_sta_set_authorized(hapd, sta, 0);
  	sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC |
  			WLAN_STA_ASSOC_REQ_OK);
  	wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG, "deauthenticated");
  	mlme_deauthenticate_indication(
  		hapd, sta, le_to_host16(mgmt->u.deauth.reason_code));
  	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
  	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
  	ap_free_sta(hapd, sta);
  }
  
  
  static void handle_beacon(struct hostapd_data *hapd,
  			  const struct ieee80211_mgmt *mgmt, size_t len,
  			  struct hostapd_frame_info *fi)
  {
  	struct ieee802_11_elems elems;
  
  	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.beacon)) {
  		printf("handle_beacon - too short payload (len=%lu)\n",
  		       (unsigned long) len);
  		return;
  	}
  
  	(void) ieee802_11_parse_elems(mgmt->u.beacon.variable,
  				      len - (IEEE80211_HDRLEN +
  					     sizeof(mgmt->u.beacon)), &elems,
  				      0);
  
  	ap_list_process_beacon(hapd->iface, mgmt, &elems, fi);
  }
  
  
  #ifdef CONFIG_IEEE80211W
  
  static void hostapd_sa_query_action(struct hostapd_data *hapd,
  				    const struct ieee80211_mgmt *mgmt,
  				    size_t len)
  {
  	const u8 *end;
  
  	end = mgmt->u.action.u.sa_query_resp.trans_id +
  		WLAN_SA_QUERY_TR_ID_LEN;
  	if (((u8 *) mgmt) + len < end) {
  		wpa_printf(MSG_DEBUG, "IEEE 802.11: Too short SA Query Action "
  			   "frame (len=%lu)", (unsigned long) len);
  		return;
  	}
  
  	ieee802_11_sa_query_action(hapd, mgmt->sa,
  				   mgmt->u.action.u.sa_query_resp.action,
  				   mgmt->u.action.u.sa_query_resp.trans_id);
  }
  
  
  static int robust_action_frame(u8 category)
  {
  	return category != WLAN_ACTION_PUBLIC &&
  		category != WLAN_ACTION_HT;
  }
  #endif /* CONFIG_IEEE80211W */
  
  
  #ifdef CONFIG_WNM
  static void hostapd_wnm_action(struct hostapd_data *hapd, struct sta_info *sta,
  			       const struct ieee80211_mgmt *mgmt,
  			       size_t len)
  {
  	struct rx_action action;
  	if (len < IEEE80211_HDRLEN + 2)
  		return;
  	os_memset(&action, 0, sizeof(action));
  	action.da = mgmt->da;
  	action.sa = mgmt->sa;
  	action.bssid = mgmt->bssid;
  	action.category = mgmt->u.action.category;
  	action.data = (const u8 *) &mgmt->u.action.u.wnm_sleep_req.action;
  	action.len = len - IEEE80211_HDRLEN - 1;
  	action.freq = hapd->iface->freq;
  	ieee802_11_rx_wnm_action_ap(hapd, &action);
  }
  #endif /* CONFIG_WNM */
  
  
  static void handle_action(struct hostapd_data *hapd,
  			  const struct ieee80211_mgmt *mgmt, size_t len)
  {
  	struct sta_info *sta;
  	sta = ap_get_sta(hapd, mgmt->sa);
  
  	if (len < IEEE80211_HDRLEN + 1) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "handle_action - too short payload (len=%lu)",
  			       (unsigned long) len);
  		return;
  	}
  
  	if (mgmt->u.action.category != WLAN_ACTION_PUBLIC &&
  	    (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))) {
  		wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignored Action "
  			   "frame (category=%u) from unassociated STA " MACSTR,
  			   MAC2STR(mgmt->sa), mgmt->u.action.category);
  		return;
  	}
  
  #ifdef CONFIG_IEEE80211W
  	if (sta && (sta->flags & WLAN_STA_MFP) &&
  	    !(mgmt->frame_control & host_to_le16(WLAN_FC_ISWEP) &&
  	      robust_action_frame(mgmt->u.action.category))) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "Dropped unprotected Robust Action frame from "
  			       "an MFP STA");
  		return;
  	}
  #endif /* CONFIG_IEEE80211W */
  
  	switch (mgmt->u.action.category) {
  #ifdef CONFIG_IEEE80211R
  	case WLAN_ACTION_FT:
  		if (wpa_ft_action_rx(sta->wpa_sm, (u8 *) &mgmt->u.action,
  				     len - IEEE80211_HDRLEN))
  			break;
  		return;
  #endif /* CONFIG_IEEE80211R */
  	case WLAN_ACTION_WMM:
  		hostapd_wmm_action(hapd, mgmt, len);
  		return;
  #ifdef CONFIG_IEEE80211W
  	case WLAN_ACTION_SA_QUERY:
  		hostapd_sa_query_action(hapd, mgmt, len);
  		return;
  #endif /* CONFIG_IEEE80211W */
  #ifdef CONFIG_WNM
  	case WLAN_ACTION_WNM:
  		hostapd_wnm_action(hapd, sta, mgmt, len);
  		return;
  #endif /* CONFIG_WNM */
  	case WLAN_ACTION_PUBLIC:
  		if (hapd->public_action_cb) {
  			hapd->public_action_cb(hapd->public_action_cb_ctx,
  					       (u8 *) mgmt, len,
  					       hapd->iface->freq);
  		}
  		if (hapd->public_action_cb2) {
  			hapd->public_action_cb2(hapd->public_action_cb2_ctx,
  						(u8 *) mgmt, len,
  						hapd->iface->freq);
  		}
  		if (hapd->public_action_cb || hapd->public_action_cb2)
  			return;
  		break;
  	case WLAN_ACTION_VENDOR_SPECIFIC:
  		if (hapd->vendor_action_cb) {
  			if (hapd->vendor_action_cb(hapd->vendor_action_cb_ctx,
  						   (u8 *) mgmt, len,
  						   hapd->iface->freq) == 0)
  				return;
  		}
  		break;
  	}
  
  	hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_DEBUG,
  		       "handle_action - unknown action category %d or invalid "
  		       "frame",
  		       mgmt->u.action.category);
  	if (!(mgmt->da[0] & 0x01) && !(mgmt->u.action.category & 0x80) &&
  	    !(mgmt->sa[0] & 0x01)) {
  		struct ieee80211_mgmt *resp;
  
  		/*
  		 * IEEE 802.11-REVma/D9.0 - 7.3.1.11
  		 * Return the Action frame to the source without change
  		 * except that MSB of the Category set to 1.
  		 */
  		wpa_printf(MSG_DEBUG, "IEEE 802.11: Return unknown Action "
  			   "frame back to sender");
  		resp = os_malloc(len);
  		if (resp == NULL)
  			return;
  		os_memcpy(resp, mgmt, len);
  		os_memcpy(resp->da, resp->sa, ETH_ALEN);
  		os_memcpy(resp->sa, hapd->own_addr, ETH_ALEN);
  		os_memcpy(resp->bssid, hapd->own_addr, ETH_ALEN);
  		resp->u.action.category |= 0x80;
  
  		if (hostapd_drv_send_mlme(hapd, resp, len, 0) < 0) {
  			wpa_printf(MSG_ERROR, "IEEE 802.11: Failed to send "
  				   "Action frame");
  		}
  		os_free(resp);
  	}
  }
  
  
  /**
   * ieee802_11_mgmt - process incoming IEEE 802.11 management frames
   * @hapd: hostapd BSS data structure (the BSS to which the management frame was
   * sent to)
   * @buf: management frame data (starting from IEEE 802.11 header)
   * @len: length of frame data in octets
   * @fi: meta data about received frame (signal level, etc.)
   *
   * Process all incoming IEEE 802.11 management frames. This will be called for
   * each frame received from the kernel driver through wlan#ap interface. In
   * addition, it can be called to re-inserted pending frames (e.g., when using
   * external RADIUS server as an MAC ACL).
   */
  void ieee802_11_mgmt(struct hostapd_data *hapd, const u8 *buf, size_t len,
  		     struct hostapd_frame_info *fi)
  {
  	struct ieee80211_mgmt *mgmt;
  	int broadcast;
  	u16 fc, stype;
  
  	if (len < 24)
  		return;
  
  	mgmt = (struct ieee80211_mgmt *) buf;
  	fc = le_to_host16(mgmt->frame_control);
  	stype = WLAN_FC_GET_STYPE(fc);
  
  	if (stype == WLAN_FC_STYPE_BEACON) {
  		handle_beacon(hapd, mgmt, len, fi);
  		return;
  	}
  
  	broadcast = mgmt->bssid[0] == 0xff && mgmt->bssid[1] == 0xff &&
  		mgmt->bssid[2] == 0xff && mgmt->bssid[3] == 0xff &&
  		mgmt->bssid[4] == 0xff && mgmt->bssid[5] == 0xff;
  
  	if (!broadcast &&
  #ifdef CONFIG_P2P
  	    /* Invitation responses can be sent with the peer MAC as BSSID */
  	    !((hapd->conf->p2p & P2P_GROUP_OWNER) &&
  	      stype == WLAN_FC_STYPE_ACTION) &&
  #endif /* CONFIG_P2P */
  	    os_memcmp(mgmt->bssid, hapd->own_addr, ETH_ALEN) != 0) {
  		printf("MGMT: BSSID=" MACSTR " not our address\n",
  		       MAC2STR(mgmt->bssid));
  		return;
  	}
  
  
  	if (stype == WLAN_FC_STYPE_PROBE_REQ) {
  		handle_probe_req(hapd, mgmt, len, fi->ssi_signal);
  		return;
  	}
  
  	if (os_memcmp(mgmt->da, hapd->own_addr, ETH_ALEN) != 0) {
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "MGMT: DA=" MACSTR " not our address",
  			       MAC2STR(mgmt->da));
  		return;
  	}
  
  	switch (stype) {
  	case WLAN_FC_STYPE_AUTH:
  		wpa_printf(MSG_DEBUG, "mgmt::auth");
  		handle_auth(hapd, mgmt, len);
  		break;
  	case WLAN_FC_STYPE_ASSOC_REQ:
  		wpa_printf(MSG_DEBUG, "mgmt::assoc_req");
  		handle_assoc(hapd, mgmt, len, 0);
  		break;
  	case WLAN_FC_STYPE_REASSOC_REQ:
  		wpa_printf(MSG_DEBUG, "mgmt::reassoc_req");
  		handle_assoc(hapd, mgmt, len, 1);
  		break;
  	case WLAN_FC_STYPE_DISASSOC:
  		wpa_printf(MSG_DEBUG, "mgmt::disassoc");
  		handle_disassoc(hapd, mgmt, len);
  		break;
  	case WLAN_FC_STYPE_DEAUTH:
  		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "mgmt::deauth");
  		handle_deauth(hapd, mgmt, len);
  		break;
  	case WLAN_FC_STYPE_ACTION:
  		wpa_printf(MSG_DEBUG, "mgmt::action");
  		handle_action(hapd, mgmt, len);
  		break;
  	default:
  		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "unknown mgmt frame subtype %d", stype);
  		break;
  	}
  }
  
  
  static void handle_auth_cb(struct hostapd_data *hapd,
  			   const struct ieee80211_mgmt *mgmt,
  			   size_t len, int ok)
  {
  	u16 auth_alg, auth_transaction, status_code;
  	struct sta_info *sta;
  
  	if (!ok) {
  		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_NOTICE,
  			       "did not acknowledge authentication response");
  		return;
  	}
  
  	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
  		printf("handle_auth_cb - too short payload (len=%lu)\n",
  		       (unsigned long) len);
  		return;
  	}
  
  	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
  	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
  	status_code = le_to_host16(mgmt->u.auth.status_code);
  
  	sta = ap_get_sta(hapd, mgmt->da);
  	if (!sta) {
  		printf("handle_auth_cb: STA " MACSTR " not found\n",
  		       MAC2STR(mgmt->da));
  		return;
  	}
  
  	if (status_code == WLAN_STATUS_SUCCESS &&
  	    ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
  	     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_INFO, "authenticated");
  		sta->flags |= WLAN_STA_AUTH;
  	}
  }
  
  
  static void hostapd_set_wds_encryption(struct hostapd_data *hapd,
  				       struct sta_info *sta,
  				       char *ifname_wds)
  {
  	int i;
  	struct hostapd_ssid *ssid = sta->ssid;
  
  	if (hapd->conf->ieee802_1x || hapd->conf->wpa)
  		return;
  
  	for (i = 0; i < 4; i++) {
  		if (ssid->wep.key[i] &&
  		    hostapd_drv_set_key(ifname_wds, hapd, WPA_ALG_WEP, NULL, i,
  					i == ssid->wep.idx, NULL, 0,
  					ssid->wep.key[i], ssid->wep.len[i])) {
  			wpa_printf(MSG_WARNING,
  				   "Could not set WEP keys for WDS interface; %s",
  				   ifname_wds);
  			break;
  		}
  	}
  }
  
  
  static void handle_assoc_cb(struct hostapd_data *hapd,
  			    const struct ieee80211_mgmt *mgmt,
  			    size_t len, int reassoc, int ok)
  {
  	u16 status;
  	struct sta_info *sta;
  	int new_assoc = 1;
  	struct ieee80211_ht_capabilities ht_cap;
  	struct ieee80211_vht_capabilities vht_cap;
  
  	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_resp) :
  				      sizeof(mgmt->u.assoc_resp))) {
  		printf("handle_assoc_cb(reassoc=%d) - too short payload "
  		       "(len=%lu)\n", reassoc, (unsigned long) len);
  		return;
  	}
  
  	sta = ap_get_sta(hapd, mgmt->da);
  	if (!sta) {
  		printf("handle_assoc_cb: STA " MACSTR " not found\n",
  		       MAC2STR(mgmt->da));
  		return;
  	}
  
  	if (!ok) {
  		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_DEBUG,
  			       "did not acknowledge association response");
  		sta->flags &= ~WLAN_STA_ASSOC_REQ_OK;
  		return;
  	}
  
  	if (reassoc)
  		status = le_to_host16(mgmt->u.reassoc_resp.status_code);
  	else
  		status = le_to_host16(mgmt->u.assoc_resp.status_code);
  
  	if (status != WLAN_STATUS_SUCCESS)
  		goto fail;
  
  	/* Stop previous accounting session, if one is started, and allocate
  	 * new session id for the new session. */
  	accounting_sta_stop(hapd, sta);
  
  	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  		       HOSTAPD_LEVEL_INFO,
  		       "associated (aid %d)",
  		       sta->aid);
  
  	if (sta->flags & WLAN_STA_ASSOC)
  		new_assoc = 0;
  	sta->flags |= WLAN_STA_ASSOC;
  	if ((!hapd->conf->ieee802_1x && !hapd->conf->wpa) ||
  	    sta->auth_alg == WLAN_AUTH_FT) {
  		/*
  		 * Open, static WEP, or FT protocol; no separate authorization
  		 * step.
  		 */
  		ap_sta_set_authorized(hapd, sta, 1);
  	}
  
  	if (reassoc)
  		mlme_reassociate_indication(hapd, sta);
  	else
  		mlme_associate_indication(hapd, sta);
  
  #ifdef CONFIG_IEEE80211W
  	sta->sa_query_timed_out = 0;
  #endif /* CONFIG_IEEE80211W */
  
  	/*
  	 * Remove the STA entry in order to make sure the STA PS state gets
  	 * cleared and configuration gets updated in case of reassociation back
  	 * to the same AP.
  	 */
  	hostapd_drv_sta_remove(hapd, sta->addr);
  
  #ifdef CONFIG_IEEE80211N
  	if (sta->flags & WLAN_STA_HT)
  		hostapd_get_ht_capab(hapd, sta->ht_capabilities, &ht_cap);
  #endif /* CONFIG_IEEE80211N */
  #ifdef CONFIG_IEEE80211AC
  	if (sta->flags & WLAN_STA_VHT)
  		hostapd_get_vht_capab(hapd, sta->vht_capabilities, &vht_cap);
  #endif /* CONFIG_IEEE80211AC */
  
  	if (hostapd_sta_add(hapd, sta->addr, sta->aid, sta->capability,
  			    sta->supported_rates, sta->supported_rates_len,
  			    sta->listen_interval,
  			    sta->flags & WLAN_STA_HT ? &ht_cap : NULL,
  			    sta->flags & WLAN_STA_VHT ? &vht_cap : NULL,
  			    sta->flags, sta->qosinfo)) {
  		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
  			       HOSTAPD_LEVEL_NOTICE,
  			       "Could not add STA to kernel driver");
  
  		ap_sta_disconnect(hapd, sta, sta->addr,
  				  WLAN_REASON_DISASSOC_AP_BUSY);
  
  		goto fail;
  	}
  
  	if (sta->flags & WLAN_STA_WDS) {
  		int ret;
  		char ifname_wds[IFNAMSIZ + 1];
  
  		ret = hostapd_set_wds_sta(hapd, ifname_wds, sta->addr,
  					  sta->aid, 1);
  		if (!ret)
  			hostapd_set_wds_encryption(hapd, sta, ifname_wds);
  	}
  
  	if (sta->eapol_sm == NULL) {
  		/*
  		 * This STA does not use RADIUS server for EAP authentication,
  		 * so bind it to the selected VLAN interface now, since the
  		 * interface selection is not going to change anymore.
  		 */
  		if (ap_sta_bind_vlan(hapd, sta, 0) < 0)
  			goto fail;
  	} else if (sta->vlan_id) {
  		/* VLAN ID already set (e.g., by PMKSA caching), so bind STA */
  		if (ap_sta_bind_vlan(hapd, sta, 0) < 0)
  			goto fail;
  	}
  
  	hostapd_set_sta_flags(hapd, sta);
  
  	if (sta->auth_alg == WLAN_AUTH_FT)
  		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC_FT);
  	else
  		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
  	hapd->new_assoc_sta_cb(hapd, sta, !new_assoc);
  
  	ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
  
   fail:
  	/* Copy of the association request is not needed anymore */
  	if (sta->last_assoc_req) {
  		os_free(sta->last_assoc_req);
  		sta->last_assoc_req = NULL;
  	}
  }
  
  
  static void handle_deauth_cb(struct hostapd_data *hapd,
  			     const struct ieee80211_mgmt *mgmt,
  			     size_t len, int ok)
  {
  	struct sta_info *sta;
  	if (mgmt->da[0] & 0x01)
  		return;
  	sta = ap_get_sta(hapd, mgmt->da);
  	if (!sta) {
  		wpa_printf(MSG_DEBUG, "handle_deauth_cb: STA " MACSTR
  			   " not found", MAC2STR(mgmt->da));
  		return;
  	}
  	if (ok)
  		wpa_printf(MSG_DEBUG, "STA " MACSTR " acknowledged deauth",
  			   MAC2STR(sta->addr));
  	else
  		wpa_printf(MSG_DEBUG, "STA " MACSTR " did not acknowledge "
  			   "deauth", MAC2STR(sta->addr));
  
  	ap_sta_deauth_cb(hapd, sta);
  }
  
  
  static void handle_disassoc_cb(struct hostapd_data *hapd,
  			       const struct ieee80211_mgmt *mgmt,
  			       size_t len, int ok)
  {
  	struct sta_info *sta;
  	if (mgmt->da[0] & 0x01)
  		return;
  	sta = ap_get_sta(hapd, mgmt->da);
  	if (!sta) {
  		wpa_printf(MSG_DEBUG, "handle_disassoc_cb: STA " MACSTR
  			   " not found", MAC2STR(mgmt->da));
  		return;
  	}
  	if (ok)
  		wpa_printf(MSG_DEBUG, "STA " MACSTR " acknowledged disassoc",
  			   MAC2STR(sta->addr));
  	else
  		wpa_printf(MSG_DEBUG, "STA " MACSTR " did not acknowledge "
  			   "disassoc", MAC2STR(sta->addr));
  
  	ap_sta_disassoc_cb(hapd, sta);
  }
  
  
  /**
   * ieee802_11_mgmt_cb - Process management frame TX status callback
   * @hapd: hostapd BSS data structure (the BSS from which the management frame
   * was sent from)
   * @buf: management frame data (starting from IEEE 802.11 header)
   * @len: length of frame data in octets
   * @stype: management frame subtype from frame control field
   * @ok: Whether the frame was ACK'ed
   */
  void ieee802_11_mgmt_cb(struct hostapd_data *hapd, const u8 *buf, size_t len,
  			u16 stype, int ok)
  {
  	const struct ieee80211_mgmt *mgmt;
  	mgmt = (const struct ieee80211_mgmt *) buf;
  
  	switch (stype) {
  	case WLAN_FC_STYPE_AUTH:
  		wpa_printf(MSG_DEBUG, "mgmt::auth cb");
  		handle_auth_cb(hapd, mgmt, len, ok);
  		break;
  	case WLAN_FC_STYPE_ASSOC_RESP:
  		wpa_printf(MSG_DEBUG, "mgmt::assoc_resp cb");
  		handle_assoc_cb(hapd, mgmt, len, 0, ok);
  		break;
  	case WLAN_FC_STYPE_REASSOC_RESP:
  		wpa_printf(MSG_DEBUG, "mgmt::reassoc_resp cb");
  		handle_assoc_cb(hapd, mgmt, len, 1, ok);
  		break;
  	case WLAN_FC_STYPE_PROBE_RESP:
  		wpa_printf(MSG_EXCESSIVE, "mgmt::proberesp cb");
  		break;
  	case WLAN_FC_STYPE_DEAUTH:
  		wpa_printf(MSG_DEBUG, "mgmt::deauth cb");
  		handle_deauth_cb(hapd, mgmt, len, ok);
  		break;
  	case WLAN_FC_STYPE_DISASSOC:
  		wpa_printf(MSG_DEBUG, "mgmt::disassoc cb");
  		handle_disassoc_cb(hapd, mgmt, len, ok);
  		break;
  	case WLAN_FC_STYPE_ACTION:
  		wpa_printf(MSG_DEBUG, "mgmt::action cb");
  		break;
  	default:
  		printf("unknown mgmt cb frame subtype %d\n", stype);
  		break;
  	}
  }
  
  
  int ieee802_11_get_mib(struct hostapd_data *hapd, char *buf, size_t buflen)
  {
  	/* TODO */
  	return 0;
  }
  
  
  int ieee802_11_get_mib_sta(struct hostapd_data *hapd, struct sta_info *sta,
  			   char *buf, size_t buflen)
  {
  	/* TODO */
  	return 0;
  }
  
  
  void hostapd_tx_status(struct hostapd_data *hapd, const u8 *addr,
  		       const u8 *buf, size_t len, int ack)
  {
  	struct sta_info *sta;
  	struct hostapd_iface *iface = hapd->iface;
  
  	sta = ap_get_sta(hapd, addr);
  	if (sta == NULL && iface->num_bss > 1) {
  		size_t j;
  		for (j = 0; j < iface->num_bss; j++) {
  			hapd = iface->bss[j];
  			sta = ap_get_sta(hapd, addr);
  			if (sta)
  				break;
  		}
  	}
  	if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))
  		return;
  	if (sta->flags & WLAN_STA_PENDING_POLL) {
  		wpa_printf(MSG_DEBUG, "STA " MACSTR " %s pending "
  			   "activity poll", MAC2STR(sta->addr),
  			   ack ? "ACKed" : "did not ACK");
  		if (ack)
  			sta->flags &= ~WLAN_STA_PENDING_POLL;
  	}
  
  	ieee802_1x_tx_status(hapd, sta, buf, len, ack);
  }
  
  
  void hostapd_eapol_tx_status(struct hostapd_data *hapd, const u8 *dst,
  			     const u8 *data, size_t len, int ack)
  {
  	struct sta_info *sta;
  	struct hostapd_iface *iface = hapd->iface;
  
  	sta = ap_get_sta(hapd, dst);
  	if (sta == NULL && iface->num_bss > 1) {
  		size_t j;
  		for (j = 0; j < iface->num_bss; j++) {
  			hapd = iface->bss[j];
  			sta = ap_get_sta(hapd, dst);
  			if (sta)
  				break;
  		}
  	}
  	if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
  		wpa_printf(MSG_DEBUG, "Ignore TX status for Data frame to STA "
  			   MACSTR " that is not currently associated",
  			   MAC2STR(dst));
  		return;
  	}
  
  	ieee802_1x_eapol_tx_status(hapd, sta, data, len, ack);
  }
  
  
  void hostapd_client_poll_ok(struct hostapd_data *hapd, const u8 *addr)
  {
  	struct sta_info *sta;
  	struct hostapd_iface *iface = hapd->iface;
  
  	sta = ap_get_sta(hapd, addr);
  	if (sta == NULL && iface->num_bss > 1) {
  		size_t j;
  		for (j = 0; j < iface->num_bss; j++) {
  			hapd = iface->bss[j];
  			sta = ap_get_sta(hapd, addr);
  			if (sta)
  				break;
  		}
  	}
  	if (sta == NULL)
  		return;
  	if (!(sta->flags & WLAN_STA_PENDING_POLL))
  		return;
  
  	wpa_printf(MSG_DEBUG, "STA " MACSTR " ACKed pending "
  		   "activity poll", MAC2STR(sta->addr));
  	sta->flags &= ~WLAN_STA_PENDING_POLL;
  }
  
  
  void ieee802_11_rx_from_unknown(struct hostapd_data *hapd, const u8 *src,
  				int wds)
  {
  	struct sta_info *sta;
  
  	sta = ap_get_sta(hapd, src);
  	if (sta && (sta->flags & WLAN_STA_ASSOC)) {
  		if (!hapd->conf->wds_sta)
  			return;
  
  		if (wds && !(sta->flags & WLAN_STA_WDS)) {
  			int ret;
  			char ifname_wds[IFNAMSIZ + 1];
  
  			wpa_printf(MSG_DEBUG, "Enable 4-address WDS mode for "
  				   "STA " MACSTR " (aid %u)",
  				   MAC2STR(sta->addr), sta->aid);
  			sta->flags |= WLAN_STA_WDS;
  			ret = hostapd_set_wds_sta(hapd, ifname_wds,
  						  sta->addr, sta->aid, 1);
  			if (!ret)
  				hostapd_set_wds_encryption(hapd, sta,
  							   ifname_wds);
  		}
  		return;
  	}
  
  	wpa_printf(MSG_DEBUG, "Data/PS-poll frame from not associated STA "
  		   MACSTR, MAC2STR(src));
  	if (src[0] & 0x01) {
  		/* Broadcast bit set in SA?! Ignore the frame silently. */
  		return;
  	}
  
  	if (sta && (sta->flags & WLAN_STA_ASSOC_REQ_OK)) {
  		wpa_printf(MSG_DEBUG, "Association Response to the STA has "
  			   "already been sent, but no TX status yet known - "
  			   "ignore Class 3 frame issue with " MACSTR,
  			   MAC2STR(src));
  		return;
  	}
  
  	if (sta && (sta->flags & WLAN_STA_AUTH))
  		hostapd_drv_sta_disassoc(
  			hapd, src,
  			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
  	else
  		hostapd_drv_sta_deauth(
  			hapd, src,
  			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
  }
  
  
  #endif /* CONFIG_NATIVE_WINDOWS */
  