xref: /freebsd/contrib/wpa/src/ap/ieee802_11.c (revision e8d8bef961a50d4dc22501cde4fb9fb0be1b2532)
1 /*
2  * hostapd / IEEE 802.11 Management
3  * Copyright (c) 2002-2017, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "utils/includes.h"
10 
11 #ifndef CONFIG_NATIVE_WINDOWS
12 
13 #include "utils/common.h"
14 #include "utils/eloop.h"
15 #include "crypto/crypto.h"
16 #include "crypto/sha256.h"
17 #include "crypto/sha384.h"
18 #include "crypto/sha512.h"
19 #include "crypto/random.h"
20 #include "common/ieee802_11_defs.h"
21 #include "common/ieee802_11_common.h"
22 #include "common/wpa_ctrl.h"
23 #include "common/sae.h"
24 #include "common/dpp.h"
25 #include "common/ocv.h"
26 #include "common/wpa_common.h"
27 #include "radius/radius.h"
28 #include "radius/radius_client.h"
29 #include "p2p/p2p.h"
30 #include "wps/wps.h"
31 #include "fst/fst.h"
32 #include "hostapd.h"
33 #include "beacon.h"
34 #include "ieee802_11_auth.h"
35 #include "sta_info.h"
36 #include "ieee802_1x.h"
37 #include "wpa_auth.h"
38 #include "pmksa_cache_auth.h"
39 #include "wmm.h"
40 #include "ap_list.h"
41 #include "accounting.h"
42 #include "ap_config.h"
43 #include "ap_mlme.h"
44 #include "p2p_hostapd.h"
45 #include "ap_drv_ops.h"
46 #include "wnm_ap.h"
47 #include "hw_features.h"
48 #include "ieee802_11.h"
49 #include "dfs.h"
50 #include "mbo_ap.h"
51 #include "rrm.h"
52 #include "taxonomy.h"
53 #include "fils_hlp.h"
54 #include "dpp_hostapd.h"
55 #include "gas_query_ap.h"
56 
57 
58 #ifdef CONFIG_FILS
59 static struct wpabuf *
60 prepare_auth_resp_fils(struct hostapd_data *hapd,
61 		       struct sta_info *sta, u16 *resp,
62 		       struct rsn_pmksa_cache_entry *pmksa,
63 		       struct wpabuf *erp_resp,
64 		       const u8 *msk, size_t msk_len,
65 		       int *is_pub);
66 #endif /* CONFIG_FILS */
67 static void handle_auth(struct hostapd_data *hapd,
68 			const struct ieee80211_mgmt *mgmt, size_t len,
69 			int rssi, int from_queue);
70 
71 
72 u8 * hostapd_eid_multi_ap(struct hostapd_data *hapd, u8 *eid)
73 {
74 	u8 multi_ap_val = 0;
75 
76 	if (!hapd->conf->multi_ap)
77 		return eid;
78 	if (hapd->conf->multi_ap & BACKHAUL_BSS)
79 		multi_ap_val |= MULTI_AP_BACKHAUL_BSS;
80 	if (hapd->conf->multi_ap & FRONTHAUL_BSS)
81 		multi_ap_val |= MULTI_AP_FRONTHAUL_BSS;
82 
83 	return eid + add_multi_ap_ie(eid, 9, multi_ap_val);
84 }
85 
86 
87 u8 * hostapd_eid_supp_rates(struct hostapd_data *hapd, u8 *eid)
88 {
89 	u8 *pos = eid;
90 	int i, num, count;
91 
92 	if (hapd->iface->current_rates == NULL)
93 		return eid;
94 
95 	*pos++ = WLAN_EID_SUPP_RATES;
96 	num = hapd->iface->num_rates;
97 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
98 		num++;
99 	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht)
100 		num++;
101 	if (num > 8) {
102 		/* rest of the rates are encoded in Extended supported
103 		 * rates element */
104 		num = 8;
105 	}
106 
107 	*pos++ = num;
108 	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num;
109 	     i++) {
110 		count++;
111 		*pos = hapd->iface->current_rates[i].rate / 5;
112 		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
113 			*pos |= 0x80;
114 		pos++;
115 	}
116 
117 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht && count < 8) {
118 		count++;
119 		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
120 	}
121 
122 	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht && count < 8) {
123 		count++;
124 		*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_VHT_PHY;
125 	}
126 
127 	return pos;
128 }
129 
130 
131 u8 * hostapd_eid_ext_supp_rates(struct hostapd_data *hapd, u8 *eid)
132 {
133 	u8 *pos = eid;
134 	int i, num, count;
135 
136 	if (hapd->iface->current_rates == NULL)
137 		return eid;
138 
139 	num = hapd->iface->num_rates;
140 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht)
141 		num++;
142 	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht)
143 		num++;
144 	if (num <= 8)
145 		return eid;
146 	num -= 8;
147 
148 	*pos++ = WLAN_EID_EXT_SUPP_RATES;
149 	*pos++ = num;
150 	for (i = 0, count = 0; i < hapd->iface->num_rates && count < num + 8;
151 	     i++) {
152 		count++;
153 		if (count <= 8)
154 			continue; /* already in SuppRates IE */
155 		*pos = hapd->iface->current_rates[i].rate / 5;
156 		if (hapd->iface->current_rates[i].flags & HOSTAPD_RATE_BASIC)
157 			*pos |= 0x80;
158 		pos++;
159 	}
160 
161 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht) {
162 		count++;
163 		if (count > 8)
164 			*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_HT_PHY;
165 	}
166 
167 	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht) {
168 		count++;
169 		if (count > 8)
170 			*pos++ = 0x80 | BSS_MEMBERSHIP_SELECTOR_VHT_PHY;
171 	}
172 
173 	return pos;
174 }
175 
176 
177 u16 hostapd_own_capab_info(struct hostapd_data *hapd)
178 {
179 	int capab = WLAN_CAPABILITY_ESS;
180 	int privacy;
181 	int dfs;
182 	int i;
183 
184 	/* Check if any of configured channels require DFS */
185 	dfs = hostapd_is_dfs_required(hapd->iface);
186 	if (dfs < 0) {
187 		wpa_printf(MSG_WARNING, "Failed to check if DFS is required; ret=%d",
188 			   dfs);
189 		dfs = 0;
190 	}
191 
192 	if (hapd->iface->num_sta_no_short_preamble == 0 &&
193 	    hapd->iconf->preamble == SHORT_PREAMBLE)
194 		capab |= WLAN_CAPABILITY_SHORT_PREAMBLE;
195 
196 	privacy = hapd->conf->ssid.wep.keys_set;
197 
198 	if (hapd->conf->ieee802_1x &&
199 	    (hapd->conf->default_wep_key_len ||
200 	     hapd->conf->individual_wep_key_len))
201 		privacy = 1;
202 
203 	if (hapd->conf->wpa)
204 		privacy = 1;
205 
206 #ifdef CONFIG_HS20
207 	if (hapd->conf->osen)
208 		privacy = 1;
209 #endif /* CONFIG_HS20 */
210 
211 	if (privacy)
212 		capab |= WLAN_CAPABILITY_PRIVACY;
213 
214 	if (hapd->iface->current_mode &&
215 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G &&
216 	    hapd->iface->num_sta_no_short_slot_time == 0)
217 		capab |= WLAN_CAPABILITY_SHORT_SLOT_TIME;
218 
219 	/*
220 	 * Currently, Spectrum Management capability bit is set when directly
221 	 * requested in configuration by spectrum_mgmt_required or when AP is
222 	 * running on DFS channel.
223 	 * TODO: Also consider driver support for TPC to set Spectrum Mgmt bit
224 	 */
225 	if (hapd->iface->current_mode &&
226 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211A &&
227 	    (hapd->iconf->spectrum_mgmt_required || dfs))
228 		capab |= WLAN_CAPABILITY_SPECTRUM_MGMT;
229 
230 	for (i = 0; i < RRM_CAPABILITIES_IE_LEN; i++) {
231 		if (hapd->conf->radio_measurements[i]) {
232 			capab |= IEEE80211_CAP_RRM;
233 			break;
234 		}
235 	}
236 
237 	return capab;
238 }
239 
240 
241 #ifndef CONFIG_NO_RC4
242 static u16 auth_shared_key(struct hostapd_data *hapd, struct sta_info *sta,
243 			   u16 auth_transaction, const u8 *challenge,
244 			   int iswep)
245 {
246 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
247 		       HOSTAPD_LEVEL_DEBUG,
248 		       "authentication (shared key, transaction %d)",
249 		       auth_transaction);
250 
251 	if (auth_transaction == 1) {
252 		if (!sta->challenge) {
253 			/* Generate a pseudo-random challenge */
254 			u8 key[8];
255 
256 			sta->challenge = os_zalloc(WLAN_AUTH_CHALLENGE_LEN);
257 			if (sta->challenge == NULL)
258 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
259 
260 			if (os_get_random(key, sizeof(key)) < 0) {
261 				os_free(sta->challenge);
262 				sta->challenge = NULL;
263 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
264 			}
265 
266 			rc4_skip(key, sizeof(key), 0,
267 				 sta->challenge, WLAN_AUTH_CHALLENGE_LEN);
268 		}
269 		return 0;
270 	}
271 
272 	if (auth_transaction != 3)
273 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
274 
275 	/* Transaction 3 */
276 	if (!iswep || !sta->challenge || !challenge ||
277 	    os_memcmp_const(sta->challenge, challenge,
278 			    WLAN_AUTH_CHALLENGE_LEN)) {
279 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
280 			       HOSTAPD_LEVEL_INFO,
281 			       "shared key authentication - invalid "
282 			       "challenge-response");
283 		return WLAN_STATUS_CHALLENGE_FAIL;
284 	}
285 
286 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
287 		       HOSTAPD_LEVEL_DEBUG,
288 		       "authentication OK (shared key)");
289 	sta->flags |= WLAN_STA_AUTH;
290 	wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
291 	os_free(sta->challenge);
292 	sta->challenge = NULL;
293 
294 	return 0;
295 }
296 #endif /* CONFIG_NO_RC4 */
297 
298 
299 static int send_auth_reply(struct hostapd_data *hapd,
300 			   const u8 *dst, const u8 *bssid,
301 			   u16 auth_alg, u16 auth_transaction, u16 resp,
302 			   const u8 *ies, size_t ies_len, const char *dbg)
303 {
304 	struct ieee80211_mgmt *reply;
305 	u8 *buf;
306 	size_t rlen;
307 	int reply_res = WLAN_STATUS_UNSPECIFIED_FAILURE;
308 
309 	rlen = IEEE80211_HDRLEN + sizeof(reply->u.auth) + ies_len;
310 	buf = os_zalloc(rlen);
311 	if (buf == NULL)
312 		return -1;
313 
314 	reply = (struct ieee80211_mgmt *) buf;
315 	reply->frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
316 					    WLAN_FC_STYPE_AUTH);
317 	os_memcpy(reply->da, dst, ETH_ALEN);
318 	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
319 	os_memcpy(reply->bssid, bssid, ETH_ALEN);
320 
321 	reply->u.auth.auth_alg = host_to_le16(auth_alg);
322 	reply->u.auth.auth_transaction = host_to_le16(auth_transaction);
323 	reply->u.auth.status_code = host_to_le16(resp);
324 
325 	if (ies && ies_len)
326 		os_memcpy(reply->u.auth.variable, ies, ies_len);
327 
328 	wpa_printf(MSG_DEBUG, "authentication reply: STA=" MACSTR
329 		   " auth_alg=%d auth_transaction=%d resp=%d (IE len=%lu) (dbg=%s)",
330 		   MAC2STR(dst), auth_alg, auth_transaction,
331 		   resp, (unsigned long) ies_len, dbg);
332 	if (hostapd_drv_send_mlme(hapd, reply, rlen, 0) < 0)
333 		wpa_printf(MSG_INFO, "send_auth_reply: send failed");
334 	else
335 		reply_res = WLAN_STATUS_SUCCESS;
336 
337 	os_free(buf);
338 
339 	return reply_res;
340 }
341 
342 
343 #ifdef CONFIG_IEEE80211R_AP
344 static void handle_auth_ft_finish(void *ctx, const u8 *dst, const u8 *bssid,
345 				  u16 auth_transaction, u16 status,
346 				  const u8 *ies, size_t ies_len)
347 {
348 	struct hostapd_data *hapd = ctx;
349 	struct sta_info *sta;
350 	int reply_res;
351 
352 	reply_res = send_auth_reply(hapd, dst, bssid, WLAN_AUTH_FT,
353 				    auth_transaction, status, ies, ies_len,
354 				    "auth-ft-finish");
355 
356 	sta = ap_get_sta(hapd, dst);
357 	if (sta == NULL)
358 		return;
359 
360 	if (sta->added_unassoc && (reply_res != WLAN_STATUS_SUCCESS ||
361 				   status != WLAN_STATUS_SUCCESS)) {
362 		hostapd_drv_sta_remove(hapd, sta->addr);
363 		sta->added_unassoc = 0;
364 		return;
365 	}
366 
367 	if (status != WLAN_STATUS_SUCCESS)
368 		return;
369 
370 	hostapd_logger(hapd, dst, HOSTAPD_MODULE_IEEE80211,
371 		       HOSTAPD_LEVEL_DEBUG, "authentication OK (FT)");
372 	sta->flags |= WLAN_STA_AUTH;
373 	mlme_authenticate_indication(hapd, sta);
374 }
375 #endif /* CONFIG_IEEE80211R_AP */
376 
377 
378 #ifdef CONFIG_SAE
379 
380 static void sae_set_state(struct sta_info *sta, enum sae_state state,
381 			  const char *reason)
382 {
383 	wpa_printf(MSG_DEBUG, "SAE: State %s -> %s for peer " MACSTR " (%s)",
384 		   sae_state_txt(sta->sae->state), sae_state_txt(state),
385 		   MAC2STR(sta->addr), reason);
386 	sta->sae->state = state;
387 }
388 
389 
390 static struct wpabuf * auth_build_sae_commit(struct hostapd_data *hapd,
391 					     struct sta_info *sta, int update)
392 {
393 	struct wpabuf *buf;
394 	const char *password = NULL;
395 	struct sae_password_entry *pw;
396 	const char *rx_id = NULL;
397 
398 	if (sta->sae->tmp)
399 		rx_id = sta->sae->tmp->pw_id;
400 
401 	for (pw = hapd->conf->sae_passwords; pw; pw = pw->next) {
402 		if (!is_broadcast_ether_addr(pw->peer_addr) &&
403 		    os_memcmp(pw->peer_addr, sta->addr, ETH_ALEN) != 0)
404 			continue;
405 		if ((rx_id && !pw->identifier) || (!rx_id && pw->identifier))
406 			continue;
407 		if (rx_id && pw->identifier &&
408 		    os_strcmp(rx_id, pw->identifier) != 0)
409 			continue;
410 		password = pw->password;
411 		break;
412 	}
413 	if (!password)
414 		password = hapd->conf->ssid.wpa_passphrase;
415 	if (!password) {
416 		wpa_printf(MSG_DEBUG, "SAE: No password available");
417 		return NULL;
418 	}
419 
420 	if (update &&
421 	    sae_prepare_commit(hapd->own_addr, sta->addr,
422 			       (u8 *) password, os_strlen(password), rx_id,
423 			       sta->sae) < 0) {
424 		wpa_printf(MSG_DEBUG, "SAE: Could not pick PWE");
425 		return NULL;
426 	}
427 
428 	if (pw && pw->vlan_id) {
429 		if (!sta->sae->tmp) {
430 			wpa_printf(MSG_INFO,
431 				   "SAE: No temporary data allocated - cannot store VLAN ID");
432 			return NULL;
433 		}
434 		sta->sae->tmp->vlan_id = pw->vlan_id;
435 	}
436 
437 	buf = wpabuf_alloc(SAE_COMMIT_MAX_LEN +
438 			   (rx_id ? 3 + os_strlen(rx_id) : 0));
439 	if (buf == NULL)
440 		return NULL;
441 	sae_write_commit(sta->sae, buf, sta->sae->tmp ?
442 			 sta->sae->tmp->anti_clogging_token : NULL, rx_id);
443 
444 	return buf;
445 }
446 
447 
448 static struct wpabuf * auth_build_sae_confirm(struct hostapd_data *hapd,
449 					      struct sta_info *sta)
450 {
451 	struct wpabuf *buf;
452 
453 	buf = wpabuf_alloc(SAE_CONFIRM_MAX_LEN);
454 	if (buf == NULL)
455 		return NULL;
456 
457 	sae_write_confirm(sta->sae, buf);
458 
459 	return buf;
460 }
461 
462 
463 static int auth_sae_send_commit(struct hostapd_data *hapd,
464 				struct sta_info *sta,
465 				const u8 *bssid, int update)
466 {
467 	struct wpabuf *data;
468 	int reply_res;
469 
470 	data = auth_build_sae_commit(hapd, sta, update);
471 	if (!data && sta->sae->tmp && sta->sae->tmp->pw_id)
472 		return WLAN_STATUS_UNKNOWN_PASSWORD_IDENTIFIER;
473 	if (data == NULL)
474 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
475 
476 	reply_res = send_auth_reply(hapd, sta->addr, bssid, WLAN_AUTH_SAE, 1,
477 				    WLAN_STATUS_SUCCESS, wpabuf_head(data),
478 				    wpabuf_len(data), "sae-send-commit");
479 
480 	wpabuf_free(data);
481 
482 	return reply_res;
483 }
484 
485 
486 static int auth_sae_send_confirm(struct hostapd_data *hapd,
487 				 struct sta_info *sta,
488 				 const u8 *bssid)
489 {
490 	struct wpabuf *data;
491 	int reply_res;
492 
493 	data = auth_build_sae_confirm(hapd, sta);
494 	if (data == NULL)
495 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
496 
497 	reply_res = send_auth_reply(hapd, sta->addr, bssid, WLAN_AUTH_SAE, 2,
498 				    WLAN_STATUS_SUCCESS, wpabuf_head(data),
499 				    wpabuf_len(data), "sae-send-confirm");
500 
501 	wpabuf_free(data);
502 
503 	return reply_res;
504 }
505 
506 
507 static int use_sae_anti_clogging(struct hostapd_data *hapd)
508 {
509 	struct sta_info *sta;
510 	unsigned int open = 0;
511 
512 	if (hapd->conf->sae_anti_clogging_threshold == 0)
513 		return 1;
514 
515 	for (sta = hapd->sta_list; sta; sta = sta->next) {
516 		if (!sta->sae)
517 			continue;
518 		if (sta->sae->state != SAE_COMMITTED &&
519 		    sta->sae->state != SAE_CONFIRMED)
520 			continue;
521 		open++;
522 		if (open >= hapd->conf->sae_anti_clogging_threshold)
523 			return 1;
524 	}
525 
526 	/* In addition to already existing open SAE sessions, check whether
527 	 * there are enough pending commit messages in the processing queue to
528 	 * potentially result in too many open sessions. */
529 	if (open + dl_list_len(&hapd->sae_commit_queue) >=
530 	    hapd->conf->sae_anti_clogging_threshold)
531 		return 1;
532 
533 	return 0;
534 }
535 
536 
537 static u8 sae_token_hash(struct hostapd_data *hapd, const u8 *addr)
538 {
539 	u8 hash[SHA256_MAC_LEN];
540 
541 	hmac_sha256(hapd->sae_token_key, sizeof(hapd->sae_token_key),
542 		    addr, ETH_ALEN, hash);
543 	return hash[0];
544 }
545 
546 
547 static int check_sae_token(struct hostapd_data *hapd, const u8 *addr,
548 			   const u8 *token, size_t token_len)
549 {
550 	u8 mac[SHA256_MAC_LEN];
551 	const u8 *addrs[2];
552 	size_t len[2];
553 	u16 token_idx;
554 	u8 idx;
555 
556 	if (token_len != SHA256_MAC_LEN)
557 		return -1;
558 	idx = sae_token_hash(hapd, addr);
559 	token_idx = hapd->sae_pending_token_idx[idx];
560 	if (token_idx == 0 || token_idx != WPA_GET_BE16(token)) {
561 		wpa_printf(MSG_DEBUG, "SAE: Invalid anti-clogging token from "
562 			   MACSTR " - token_idx 0x%04x, expected 0x%04x",
563 			   MAC2STR(addr), WPA_GET_BE16(token), token_idx);
564 		return -1;
565 	}
566 
567 	addrs[0] = addr;
568 	len[0] = ETH_ALEN;
569 	addrs[1] = token;
570 	len[1] = 2;
571 	if (hmac_sha256_vector(hapd->sae_token_key, sizeof(hapd->sae_token_key),
572 			       2, addrs, len, mac) < 0 ||
573 	    os_memcmp_const(token + 2, &mac[2], SHA256_MAC_LEN - 2) != 0)
574 		return -1;
575 
576 	hapd->sae_pending_token_idx[idx] = 0; /* invalidate used token */
577 
578 	return 0;
579 }
580 
581 
582 static struct wpabuf * auth_build_token_req(struct hostapd_data *hapd,
583 					    int group, const u8 *addr)
584 {
585 	struct wpabuf *buf;
586 	u8 *token;
587 	struct os_reltime now;
588 	u8 idx[2];
589 	const u8 *addrs[2];
590 	size_t len[2];
591 	u8 p_idx;
592 	u16 token_idx;
593 
594 	os_get_reltime(&now);
595 	if (!os_reltime_initialized(&hapd->last_sae_token_key_update) ||
596 	    os_reltime_expired(&now, &hapd->last_sae_token_key_update, 60) ||
597 	    hapd->sae_token_idx == 0xffff) {
598 		if (random_get_bytes(hapd->sae_token_key,
599 				     sizeof(hapd->sae_token_key)) < 0)
600 			return NULL;
601 		wpa_hexdump(MSG_DEBUG, "SAE: Updated token key",
602 			    hapd->sae_token_key, sizeof(hapd->sae_token_key));
603 		hapd->last_sae_token_key_update = now;
604 		hapd->sae_token_idx = 0;
605 		os_memset(hapd->sae_pending_token_idx, 0,
606 			  sizeof(hapd->sae_pending_token_idx));
607 	}
608 
609 	buf = wpabuf_alloc(sizeof(le16) + SHA256_MAC_LEN);
610 	if (buf == NULL)
611 		return NULL;
612 
613 	wpabuf_put_le16(buf, group); /* Finite Cyclic Group */
614 
615 	p_idx = sae_token_hash(hapd, addr);
616 	token_idx = hapd->sae_pending_token_idx[p_idx];
617 	if (!token_idx) {
618 		hapd->sae_token_idx++;
619 		token_idx = hapd->sae_token_idx;
620 		hapd->sae_pending_token_idx[p_idx] = token_idx;
621 	}
622 	WPA_PUT_BE16(idx, token_idx);
623 	token = wpabuf_put(buf, SHA256_MAC_LEN);
624 	addrs[0] = addr;
625 	len[0] = ETH_ALEN;
626 	addrs[1] = idx;
627 	len[1] = sizeof(idx);
628 	if (hmac_sha256_vector(hapd->sae_token_key, sizeof(hapd->sae_token_key),
629 			       2, addrs, len, token) < 0) {
630 		wpabuf_free(buf);
631 		return NULL;
632 	}
633 	WPA_PUT_BE16(token, token_idx);
634 
635 	return buf;
636 }
637 
638 
639 static int sae_check_big_sync(struct hostapd_data *hapd, struct sta_info *sta)
640 {
641 	if (sta->sae->sync > hapd->conf->sae_sync) {
642 		sae_set_state(sta, SAE_NOTHING, "Sync > dot11RSNASAESync");
643 		sta->sae->sync = 0;
644 		return -1;
645 	}
646 	return 0;
647 }
648 
649 
650 static void auth_sae_retransmit_timer(void *eloop_ctx, void *eloop_data)
651 {
652 	struct hostapd_data *hapd = eloop_ctx;
653 	struct sta_info *sta = eloop_data;
654 	int ret;
655 
656 	if (sae_check_big_sync(hapd, sta))
657 		return;
658 	sta->sae->sync++;
659 	wpa_printf(MSG_DEBUG, "SAE: Auth SAE retransmit timer for " MACSTR
660 		   " (sync=%d state=%s)",
661 		   MAC2STR(sta->addr), sta->sae->sync,
662 		   sae_state_txt(sta->sae->state));
663 
664 	switch (sta->sae->state) {
665 	case SAE_COMMITTED:
666 		ret = auth_sae_send_commit(hapd, sta, hapd->own_addr, 0);
667 		eloop_register_timeout(0,
668 				       hapd->dot11RSNASAERetransPeriod * 1000,
669 				       auth_sae_retransmit_timer, hapd, sta);
670 		break;
671 	case SAE_CONFIRMED:
672 		ret = auth_sae_send_confirm(hapd, sta, hapd->own_addr);
673 		eloop_register_timeout(0,
674 				       hapd->dot11RSNASAERetransPeriod * 1000,
675 				       auth_sae_retransmit_timer, hapd, sta);
676 		break;
677 	default:
678 		ret = -1;
679 		break;
680 	}
681 
682 	if (ret != WLAN_STATUS_SUCCESS)
683 		wpa_printf(MSG_INFO, "SAE: Failed to retransmit: ret=%d", ret);
684 }
685 
686 
687 void sae_clear_retransmit_timer(struct hostapd_data *hapd, struct sta_info *sta)
688 {
689 	eloop_cancel_timeout(auth_sae_retransmit_timer, hapd, sta);
690 }
691 
692 
693 static void sae_set_retransmit_timer(struct hostapd_data *hapd,
694 				     struct sta_info *sta)
695 {
696 	if (!(hapd->conf->mesh & MESH_ENABLED))
697 		return;
698 
699 	eloop_cancel_timeout(auth_sae_retransmit_timer, hapd, sta);
700 	eloop_register_timeout(0, hapd->dot11RSNASAERetransPeriod * 1000,
701 			       auth_sae_retransmit_timer, hapd, sta);
702 }
703 
704 
705 static void sae_sme_send_external_auth_status(struct hostapd_data *hapd,
706 					      struct sta_info *sta, u16 status)
707 {
708 	struct external_auth params;
709 
710 	os_memset(&params, 0, sizeof(params));
711 	params.status = status;
712 	params.bssid = sta->addr;
713 	if (status == WLAN_STATUS_SUCCESS && sta->sae &&
714 	    !hapd->conf->disable_pmksa_caching)
715 		params.pmkid = sta->sae->pmkid;
716 
717 	hostapd_drv_send_external_auth_status(hapd, &params);
718 }
719 
720 
721 void sae_accept_sta(struct hostapd_data *hapd, struct sta_info *sta)
722 {
723 #ifndef CONFIG_NO_VLAN
724 	struct vlan_description vlan_desc;
725 
726 	if (sta->sae->tmp && sta->sae->tmp->vlan_id > 0) {
727 		wpa_printf(MSG_DEBUG, "SAE: Assign STA " MACSTR
728 			   " to VLAN ID %d",
729 			   MAC2STR(sta->addr), sta->sae->tmp->vlan_id);
730 
731 		os_memset(&vlan_desc, 0, sizeof(vlan_desc));
732 		vlan_desc.notempty = 1;
733 		vlan_desc.untagged = sta->sae->tmp->vlan_id;
734 		if (!hostapd_vlan_valid(hapd->conf->vlan, &vlan_desc)) {
735 			wpa_printf(MSG_INFO,
736 				   "Invalid VLAN ID %d in sae_password",
737 				   sta->sae->tmp->vlan_id);
738 			return;
739 		}
740 
741 		if (ap_sta_set_vlan(hapd, sta, &vlan_desc) < 0 ||
742 		    ap_sta_bind_vlan(hapd, sta) < 0) {
743 			wpa_printf(MSG_INFO,
744 				   "Failed to assign VLAN ID %d from sae_password to "
745 				   MACSTR, sta->sae->tmp->vlan_id,
746 				   MAC2STR(sta->addr));
747 			return;
748 		}
749 	}
750 #endif /* CONFIG_NO_VLAN */
751 
752 	sta->flags |= WLAN_STA_AUTH;
753 	sta->auth_alg = WLAN_AUTH_SAE;
754 	mlme_authenticate_indication(hapd, sta);
755 	wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
756 	sae_set_state(sta, SAE_ACCEPTED, "Accept Confirm");
757 	wpa_auth_pmksa_add_sae(hapd->wpa_auth, sta->addr,
758 			       sta->sae->pmk, sta->sae->pmkid);
759 	sae_sme_send_external_auth_status(hapd, sta, WLAN_STATUS_SUCCESS);
760 }
761 
762 
763 static int sae_sm_step(struct hostapd_data *hapd, struct sta_info *sta,
764 		       const u8 *bssid, u8 auth_transaction, int allow_reuse,
765 		       int *sta_removed)
766 {
767 	int ret;
768 
769 	*sta_removed = 0;
770 
771 	if (auth_transaction != 1 && auth_transaction != 2)
772 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
773 
774 	wpa_printf(MSG_DEBUG, "SAE: Peer " MACSTR " state=%s auth_trans=%u",
775 		   MAC2STR(sta->addr), sae_state_txt(sta->sae->state),
776 		   auth_transaction);
777 	switch (sta->sae->state) {
778 	case SAE_NOTHING:
779 		if (auth_transaction == 1) {
780 			ret = auth_sae_send_commit(hapd, sta, bssid,
781 						   !allow_reuse);
782 			if (ret)
783 				return ret;
784 			sae_set_state(sta, SAE_COMMITTED, "Sent Commit");
785 
786 			if (sae_process_commit(sta->sae) < 0)
787 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
788 
789 			/*
790 			 * In mesh case, both Commit and Confirm can be sent
791 			 * immediately. In infrastructure BSS, only a single
792 			 * Authentication frame (Commit) is expected from the AP
793 			 * here and the second one (Confirm) will be sent once
794 			 * the STA has sent its second Authentication frame
795 			 * (Confirm).
796 			 */
797 			if (hapd->conf->mesh & MESH_ENABLED) {
798 				/*
799 				 * Send both Commit and Confirm immediately
800 				 * based on SAE finite state machine
801 				 * Nothing -> Confirm transition.
802 				 */
803 				ret = auth_sae_send_confirm(hapd, sta, bssid);
804 				if (ret)
805 					return ret;
806 				sae_set_state(sta, SAE_CONFIRMED,
807 					      "Sent Confirm (mesh)");
808 			} else {
809 				/*
810 				 * For infrastructure BSS, send only the Commit
811 				 * message now to get alternating sequence of
812 				 * Authentication frames between the AP and STA.
813 				 * Confirm will be sent in
814 				 * Committed -> Confirmed/Accepted transition
815 				 * when receiving Confirm from STA.
816 				 */
817 			}
818 			sta->sae->sync = 0;
819 			sae_set_retransmit_timer(hapd, sta);
820 		} else {
821 			hostapd_logger(hapd, sta->addr,
822 				       HOSTAPD_MODULE_IEEE80211,
823 				       HOSTAPD_LEVEL_DEBUG,
824 				       "SAE confirm before commit");
825 		}
826 		break;
827 	case SAE_COMMITTED:
828 		sae_clear_retransmit_timer(hapd, sta);
829 		if (auth_transaction == 1) {
830 			if (sae_process_commit(sta->sae) < 0)
831 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
832 
833 			ret = auth_sae_send_confirm(hapd, sta, bssid);
834 			if (ret)
835 				return ret;
836 			sae_set_state(sta, SAE_CONFIRMED, "Sent Confirm");
837 			sta->sae->sync = 0;
838 			sae_set_retransmit_timer(hapd, sta);
839 		} else if (hapd->conf->mesh & MESH_ENABLED) {
840 			/*
841 			 * In mesh case, follow SAE finite state machine and
842 			 * send Commit now, if sync count allows.
843 			 */
844 			if (sae_check_big_sync(hapd, sta))
845 				return WLAN_STATUS_SUCCESS;
846 			sta->sae->sync++;
847 
848 			ret = auth_sae_send_commit(hapd, sta, bssid, 0);
849 			if (ret)
850 				return ret;
851 
852 			sae_set_retransmit_timer(hapd, sta);
853 		} else {
854 			/*
855 			 * For instructure BSS, send the postponed Confirm from
856 			 * Nothing -> Confirmed transition that was reduced to
857 			 * Nothing -> Committed above.
858 			 */
859 			ret = auth_sae_send_confirm(hapd, sta, bssid);
860 			if (ret)
861 				return ret;
862 
863 			sae_set_state(sta, SAE_CONFIRMED, "Sent Confirm");
864 
865 			/*
866 			 * Since this was triggered on Confirm RX, run another
867 			 * step to get to Accepted without waiting for
868 			 * additional events.
869 			 */
870 			return sae_sm_step(hapd, sta, bssid, auth_transaction,
871 					   0, sta_removed);
872 		}
873 		break;
874 	case SAE_CONFIRMED:
875 		sae_clear_retransmit_timer(hapd, sta);
876 		if (auth_transaction == 1) {
877 			if (sae_check_big_sync(hapd, sta))
878 				return WLAN_STATUS_SUCCESS;
879 			sta->sae->sync++;
880 
881 			ret = auth_sae_send_commit(hapd, sta, bssid, 1);
882 			if (ret)
883 				return ret;
884 
885 			if (sae_process_commit(sta->sae) < 0)
886 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
887 
888 			ret = auth_sae_send_confirm(hapd, sta, bssid);
889 			if (ret)
890 				return ret;
891 
892 			sae_set_retransmit_timer(hapd, sta);
893 		} else {
894 			sta->sae->send_confirm = 0xffff;
895 			sae_accept_sta(hapd, sta);
896 		}
897 		break;
898 	case SAE_ACCEPTED:
899 		if (auth_transaction == 1 &&
900 		    (hapd->conf->mesh & MESH_ENABLED)) {
901 			wpa_printf(MSG_DEBUG, "SAE: remove the STA (" MACSTR
902 				   ") doing reauthentication",
903 				   MAC2STR(sta->addr));
904 			wpa_auth_pmksa_remove(hapd->wpa_auth, sta->addr);
905 			ap_free_sta(hapd, sta);
906 			*sta_removed = 1;
907 		} else if (auth_transaction == 1) {
908 			wpa_printf(MSG_DEBUG, "SAE: Start reauthentication");
909 			ret = auth_sae_send_commit(hapd, sta, bssid, 1);
910 			if (ret)
911 				return ret;
912 			sae_set_state(sta, SAE_COMMITTED, "Sent Commit");
913 
914 			if (sae_process_commit(sta->sae) < 0)
915 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
916 			sta->sae->sync = 0;
917 			sae_set_retransmit_timer(hapd, sta);
918 		} else {
919 			if (sae_check_big_sync(hapd, sta))
920 				return WLAN_STATUS_SUCCESS;
921 			sta->sae->sync++;
922 
923 			ret = auth_sae_send_confirm(hapd, sta, bssid);
924 			sae_clear_temp_data(sta->sae);
925 			if (ret)
926 				return ret;
927 		}
928 		break;
929 	default:
930 		wpa_printf(MSG_ERROR, "SAE: invalid state %d",
931 			   sta->sae->state);
932 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
933 	}
934 	return WLAN_STATUS_SUCCESS;
935 }
936 
937 
938 static void sae_pick_next_group(struct hostapd_data *hapd, struct sta_info *sta)
939 {
940 	struct sae_data *sae = sta->sae;
941 	int i, *groups = hapd->conf->sae_groups;
942 	int default_groups[] = { 19, 0 };
943 
944 	if (sae->state != SAE_COMMITTED)
945 		return;
946 
947 	wpa_printf(MSG_DEBUG, "SAE: Previously selected group: %d", sae->group);
948 
949 	if (!groups)
950 		groups = default_groups;
951 	for (i = 0; groups[i] > 0; i++) {
952 		if (sae->group == groups[i])
953 			break;
954 	}
955 
956 	if (groups[i] <= 0) {
957 		wpa_printf(MSG_DEBUG,
958 			   "SAE: Previously selected group not found from the current configuration");
959 		return;
960 	}
961 
962 	for (;;) {
963 		i++;
964 		if (groups[i] <= 0) {
965 			wpa_printf(MSG_DEBUG,
966 				   "SAE: No alternative group enabled");
967 			return;
968 		}
969 
970 		if (sae_set_group(sae, groups[i]) < 0)
971 			continue;
972 
973 		break;
974 	}
975 	wpa_printf(MSG_DEBUG, "SAE: Selected new group: %d", groups[i]);
976 }
977 
978 
979 static void handle_auth_sae(struct hostapd_data *hapd, struct sta_info *sta,
980 			    const struct ieee80211_mgmt *mgmt, size_t len,
981 			    u16 auth_transaction, u16 status_code)
982 {
983 	int resp = WLAN_STATUS_SUCCESS;
984 	struct wpabuf *data = NULL;
985 	int *groups = hapd->conf->sae_groups;
986 	int default_groups[] = { 19, 0 };
987 	const u8 *pos, *end;
988 	int sta_removed = 0;
989 
990 	if (!groups)
991 		groups = default_groups;
992 
993 #ifdef CONFIG_TESTING_OPTIONS
994 	if (hapd->conf->sae_reflection_attack && auth_transaction == 1) {
995 		wpa_printf(MSG_DEBUG, "SAE: TESTING - reflection attack");
996 		pos = mgmt->u.auth.variable;
997 		end = ((const u8 *) mgmt) + len;
998 		send_auth_reply(hapd, mgmt->sa, mgmt->bssid, WLAN_AUTH_SAE,
999 				auth_transaction, resp, pos, end - pos,
1000 				"auth-sae-reflection-attack");
1001 		goto remove_sta;
1002 	}
1003 
1004 	if (hapd->conf->sae_commit_override && auth_transaction == 1) {
1005 		wpa_printf(MSG_DEBUG, "SAE: TESTING - commit override");
1006 		send_auth_reply(hapd, mgmt->sa, mgmt->bssid, WLAN_AUTH_SAE,
1007 				auth_transaction, resp,
1008 				wpabuf_head(hapd->conf->sae_commit_override),
1009 				wpabuf_len(hapd->conf->sae_commit_override),
1010 				"sae-commit-override");
1011 		goto remove_sta;
1012 	}
1013 #endif /* CONFIG_TESTING_OPTIONS */
1014 	if (!sta->sae) {
1015 		if (auth_transaction != 1 ||
1016 		    status_code != WLAN_STATUS_SUCCESS) {
1017 			resp = -1;
1018 			goto remove_sta;
1019 		}
1020 		sta->sae = os_zalloc(sizeof(*sta->sae));
1021 		if (!sta->sae) {
1022 			resp = -1;
1023 			goto remove_sta;
1024 		}
1025 		sae_set_state(sta, SAE_NOTHING, "Init");
1026 		sta->sae->sync = 0;
1027 	}
1028 
1029 	if (sta->mesh_sae_pmksa_caching) {
1030 		wpa_printf(MSG_DEBUG,
1031 			   "SAE: Cancel use of mesh PMKSA caching because peer starts SAE authentication");
1032 		wpa_auth_pmksa_remove(hapd->wpa_auth, sta->addr);
1033 		sta->mesh_sae_pmksa_caching = 0;
1034 	}
1035 
1036 	if (auth_transaction == 1) {
1037 		const u8 *token = NULL;
1038 		size_t token_len = 0;
1039 		int allow_reuse = 0;
1040 
1041 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1042 			       HOSTAPD_LEVEL_DEBUG,
1043 			       "start SAE authentication (RX commit, status=%u (%s))",
1044 			       status_code, status2str(status_code));
1045 
1046 		if ((hapd->conf->mesh & MESH_ENABLED) &&
1047 		    status_code == WLAN_STATUS_ANTI_CLOGGING_TOKEN_REQ &&
1048 		    sta->sae->tmp) {
1049 			pos = mgmt->u.auth.variable;
1050 			end = ((const u8 *) mgmt) + len;
1051 			if (pos + sizeof(le16) > end) {
1052 				wpa_printf(MSG_ERROR,
1053 					   "SAE: Too short anti-clogging token request");
1054 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1055 				goto reply;
1056 			}
1057 			resp = sae_group_allowed(sta->sae, groups,
1058 						 WPA_GET_LE16(pos));
1059 			if (resp != WLAN_STATUS_SUCCESS) {
1060 				wpa_printf(MSG_ERROR,
1061 					   "SAE: Invalid group in anti-clogging token request");
1062 				goto reply;
1063 			}
1064 			pos += sizeof(le16);
1065 
1066 			wpabuf_free(sta->sae->tmp->anti_clogging_token);
1067 			sta->sae->tmp->anti_clogging_token =
1068 				wpabuf_alloc_copy(pos, end - pos);
1069 			if (sta->sae->tmp->anti_clogging_token == NULL) {
1070 				wpa_printf(MSG_ERROR,
1071 					   "SAE: Failed to alloc for anti-clogging token");
1072 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1073 				goto remove_sta;
1074 			}
1075 
1076 			/*
1077 			 * IEEE Std 802.11-2012, 11.3.8.6.4: If the Status code
1078 			 * is 76, a new Commit Message shall be constructed
1079 			 * with the Anti-Clogging Token from the received
1080 			 * Authentication frame, and the commit-scalar and
1081 			 * COMMIT-ELEMENT previously sent.
1082 			 */
1083 			resp = auth_sae_send_commit(hapd, sta, mgmt->bssid, 0);
1084 			if (resp != WLAN_STATUS_SUCCESS) {
1085 				wpa_printf(MSG_ERROR,
1086 					   "SAE: Failed to send commit message");
1087 				goto remove_sta;
1088 			}
1089 			sae_set_state(sta, SAE_COMMITTED,
1090 				      "Sent Commit (anti-clogging token case in mesh)");
1091 			sta->sae->sync = 0;
1092 			sae_set_retransmit_timer(hapd, sta);
1093 			return;
1094 		}
1095 
1096 		if ((hapd->conf->mesh & MESH_ENABLED) &&
1097 		    status_code ==
1098 		    WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED &&
1099 		    sta->sae->tmp) {
1100 			wpa_printf(MSG_DEBUG,
1101 				   "SAE: Peer did not accept our SAE group");
1102 			sae_pick_next_group(hapd, sta);
1103 			goto remove_sta;
1104 		}
1105 
1106 		if (status_code != WLAN_STATUS_SUCCESS)
1107 			goto remove_sta;
1108 
1109 		if (!(hapd->conf->mesh & MESH_ENABLED) &&
1110 		    sta->sae->state == SAE_COMMITTED) {
1111 			/* This is needed in the infrastructure BSS case to
1112 			 * address a sequence where a STA entry may remain in
1113 			 * hostapd across two attempts to do SAE authentication
1114 			 * by the same STA. The second attempt may end up trying
1115 			 * to use a different group and that would not be
1116 			 * allowed if we remain in Committed state with the
1117 			 * previously set parameters. */
1118 			pos = mgmt->u.auth.variable;
1119 			end = ((const u8 *) mgmt) + len;
1120 			if (end - pos >= (int) sizeof(le16) &&
1121 			    sae_group_allowed(sta->sae, groups,
1122 					      WPA_GET_LE16(pos)) ==
1123 			    WLAN_STATUS_SUCCESS) {
1124 				/* Do not waste resources deriving the same PWE
1125 				 * again since the same group is reused. */
1126 				sae_set_state(sta, SAE_NOTHING,
1127 					      "Allow previous PWE to be reused");
1128 				allow_reuse = 1;
1129 			} else {
1130 				sae_set_state(sta, SAE_NOTHING,
1131 					      "Clear existing state to allow restart");
1132 				sae_clear_data(sta->sae);
1133 			}
1134 		}
1135 
1136 		resp = sae_parse_commit(sta->sae, mgmt->u.auth.variable,
1137 					((const u8 *) mgmt) + len -
1138 					mgmt->u.auth.variable, &token,
1139 					&token_len, groups);
1140 		if (resp == SAE_SILENTLY_DISCARD) {
1141 			wpa_printf(MSG_DEBUG,
1142 				   "SAE: Drop commit message from " MACSTR " due to reflection attack",
1143 				   MAC2STR(sta->addr));
1144 			goto remove_sta;
1145 		}
1146 
1147 		if (resp == WLAN_STATUS_UNKNOWN_PASSWORD_IDENTIFIER) {
1148 			wpa_msg(hapd->msg_ctx, MSG_INFO,
1149 				WPA_EVENT_SAE_UNKNOWN_PASSWORD_IDENTIFIER
1150 				MACSTR, MAC2STR(sta->addr));
1151 			sae_clear_retransmit_timer(hapd, sta);
1152 			sae_set_state(sta, SAE_NOTHING,
1153 				      "Unknown Password Identifier");
1154 			goto remove_sta;
1155 		}
1156 
1157 		if (token && check_sae_token(hapd, sta->addr, token, token_len)
1158 		    < 0) {
1159 			wpa_printf(MSG_DEBUG, "SAE: Drop commit message with "
1160 				   "incorrect token from " MACSTR,
1161 				   MAC2STR(sta->addr));
1162 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1163 			goto remove_sta;
1164 		}
1165 
1166 		if (resp != WLAN_STATUS_SUCCESS)
1167 			goto reply;
1168 
1169 		if (!token && use_sae_anti_clogging(hapd) && !allow_reuse) {
1170 			wpa_printf(MSG_DEBUG,
1171 				   "SAE: Request anti-clogging token from "
1172 				   MACSTR, MAC2STR(sta->addr));
1173 			data = auth_build_token_req(hapd, sta->sae->group,
1174 						    sta->addr);
1175 			resp = WLAN_STATUS_ANTI_CLOGGING_TOKEN_REQ;
1176 			if (hapd->conf->mesh & MESH_ENABLED)
1177 				sae_set_state(sta, SAE_NOTHING,
1178 					      "Request anti-clogging token case in mesh");
1179 			goto reply;
1180 		}
1181 
1182 		resp = sae_sm_step(hapd, sta, mgmt->bssid, auth_transaction,
1183 				   allow_reuse, &sta_removed);
1184 	} else if (auth_transaction == 2) {
1185 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1186 			       HOSTAPD_LEVEL_DEBUG,
1187 			       "SAE authentication (RX confirm, status=%u (%s))",
1188 			       status_code, status2str(status_code));
1189 		if (status_code != WLAN_STATUS_SUCCESS)
1190 			goto remove_sta;
1191 		if (sta->sae->state >= SAE_CONFIRMED ||
1192 		    !(hapd->conf->mesh & MESH_ENABLED)) {
1193 			const u8 *var;
1194 			size_t var_len;
1195 			u16 peer_send_confirm;
1196 
1197 			var = mgmt->u.auth.variable;
1198 			var_len = ((u8 *) mgmt) + len - mgmt->u.auth.variable;
1199 			if (var_len < 2) {
1200 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1201 				goto reply;
1202 			}
1203 
1204 			peer_send_confirm = WPA_GET_LE16(var);
1205 
1206 			if (sta->sae->state == SAE_ACCEPTED &&
1207 			    (peer_send_confirm <= sta->sae->rc ||
1208 			     peer_send_confirm == 0xffff)) {
1209 				wpa_printf(MSG_DEBUG,
1210 					   "SAE: Silently ignore unexpected Confirm from peer "
1211 					   MACSTR
1212 					   " (peer-send-confirm=%u Rc=%u)",
1213 					   MAC2STR(sta->addr),
1214 					   peer_send_confirm, sta->sae->rc);
1215 				return;
1216 			}
1217 
1218 			if (sae_check_confirm(sta->sae, var, var_len) < 0) {
1219 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1220 				goto reply;
1221 			}
1222 			sta->sae->rc = peer_send_confirm;
1223 		}
1224 		resp = sae_sm_step(hapd, sta, mgmt->bssid, auth_transaction, 0,
1225 			&sta_removed);
1226 	} else {
1227 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1228 			       HOSTAPD_LEVEL_DEBUG,
1229 			       "unexpected SAE authentication transaction %u (status=%u (%s))",
1230 			       auth_transaction, status_code,
1231 			       status2str(status_code));
1232 		if (status_code != WLAN_STATUS_SUCCESS)
1233 			goto remove_sta;
1234 		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
1235 	}
1236 
1237 reply:
1238 	if (!sta_removed && resp != WLAN_STATUS_SUCCESS) {
1239 		pos = mgmt->u.auth.variable;
1240 		end = ((const u8 *) mgmt) + len;
1241 
1242 		/* Copy the Finite Cyclic Group field from the request if we
1243 		 * rejected it as unsupported group. */
1244 		if (resp == WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED &&
1245 		    !data && end - pos >= 2)
1246 			data = wpabuf_alloc_copy(pos, 2);
1247 
1248 		sae_sme_send_external_auth_status(hapd, sta, resp);
1249 		send_auth_reply(hapd, mgmt->sa, mgmt->bssid, WLAN_AUTH_SAE,
1250 				auth_transaction, resp,
1251 				data ? wpabuf_head(data) : (u8 *) "",
1252 				data ? wpabuf_len(data) : 0, "auth-sae");
1253 	}
1254 
1255 remove_sta:
1256 	if (!sta_removed && sta->added_unassoc &&
1257 	    (resp != WLAN_STATUS_SUCCESS ||
1258 	     status_code != WLAN_STATUS_SUCCESS)) {
1259 		hostapd_drv_sta_remove(hapd, sta->addr);
1260 		sta->added_unassoc = 0;
1261 	}
1262 	wpabuf_free(data);
1263 }
1264 
1265 
1266 /**
1267  * auth_sae_init_committed - Send COMMIT and start SAE in committed state
1268  * @hapd: BSS data for the device initiating the authentication
1269  * @sta: the peer to which commit authentication frame is sent
1270  *
1271  * This function implements Init event handling (IEEE Std 802.11-2012,
1272  * 11.3.8.6.3) in which initial COMMIT message is sent. Prior to calling, the
1273  * sta->sae structure should be initialized appropriately via a call to
1274  * sae_prepare_commit().
1275  */
1276 int auth_sae_init_committed(struct hostapd_data *hapd, struct sta_info *sta)
1277 {
1278 	int ret;
1279 
1280 	if (!sta->sae || !sta->sae->tmp)
1281 		return -1;
1282 
1283 	if (sta->sae->state != SAE_NOTHING)
1284 		return -1;
1285 
1286 	ret = auth_sae_send_commit(hapd, sta, hapd->own_addr, 0);
1287 	if (ret)
1288 		return -1;
1289 
1290 	sae_set_state(sta, SAE_COMMITTED, "Init and sent commit");
1291 	sta->sae->sync = 0;
1292 	sae_set_retransmit_timer(hapd, sta);
1293 
1294 	return 0;
1295 }
1296 
1297 
1298 void auth_sae_process_commit(void *eloop_ctx, void *user_ctx)
1299 {
1300 	struct hostapd_data *hapd = eloop_ctx;
1301 	struct hostapd_sae_commit_queue *q;
1302 	unsigned int queue_len;
1303 
1304 	q = dl_list_first(&hapd->sae_commit_queue,
1305 			  struct hostapd_sae_commit_queue, list);
1306 	if (!q)
1307 		return;
1308 	wpa_printf(MSG_DEBUG,
1309 		   "SAE: Process next available message from queue");
1310 	dl_list_del(&q->list);
1311 	handle_auth(hapd, (const struct ieee80211_mgmt *) q->msg, q->len,
1312 		    q->rssi, 1);
1313 	os_free(q);
1314 
1315 	if (eloop_is_timeout_registered(auth_sae_process_commit, hapd, NULL))
1316 		return;
1317 	queue_len = dl_list_len(&hapd->sae_commit_queue);
1318 	eloop_register_timeout(0, queue_len * 10000, auth_sae_process_commit,
1319 			       hapd, NULL);
1320 }
1321 
1322 
1323 static void auth_sae_queue(struct hostapd_data *hapd,
1324 			   const struct ieee80211_mgmt *mgmt, size_t len,
1325 			   int rssi)
1326 {
1327 	struct hostapd_sae_commit_queue *q, *q2;
1328 	unsigned int queue_len;
1329 	const struct ieee80211_mgmt *mgmt2;
1330 
1331 	queue_len = dl_list_len(&hapd->sae_commit_queue);
1332 	if (queue_len >= 15) {
1333 		wpa_printf(MSG_DEBUG,
1334 			   "SAE: No more room in message queue - drop the new frame from "
1335 			   MACSTR, MAC2STR(mgmt->sa));
1336 		return;
1337 	}
1338 
1339 	wpa_printf(MSG_DEBUG, "SAE: Queue Authentication message from "
1340 		   MACSTR " for processing (queue_len %u)", MAC2STR(mgmt->sa),
1341 		   queue_len);
1342 	q = os_zalloc(sizeof(*q) + len);
1343 	if (!q)
1344 		return;
1345 	q->rssi = rssi;
1346 	q->len = len;
1347 	os_memcpy(q->msg, mgmt, len);
1348 
1349 	/* Check whether there is already a queued Authentication frame from the
1350 	 * same station with the same transaction number and if so, replace that
1351 	 * queue entry with the new one. This avoids issues with a peer that
1352 	 * sends multiple times (e.g., due to frequent SAE retries). There is no
1353 	 * point in us trying to process the old attempts after a new one has
1354 	 * obsoleted them. */
1355 	dl_list_for_each(q2, &hapd->sae_commit_queue,
1356 			 struct hostapd_sae_commit_queue, list) {
1357 		mgmt2 = (const struct ieee80211_mgmt *) q2->msg;
1358 		if (os_memcmp(mgmt->sa, mgmt2->sa, ETH_ALEN) == 0 &&
1359 		    mgmt->u.auth.auth_transaction ==
1360 		    mgmt2->u.auth.auth_transaction) {
1361 			wpa_printf(MSG_DEBUG,
1362 				   "SAE: Replace queued message from same STA with same transaction number");
1363 			dl_list_add(&q2->list, &q->list);
1364 			dl_list_del(&q2->list);
1365 			os_free(q2);
1366 			goto queued;
1367 		}
1368 	}
1369 
1370 	/* No pending identical entry, so add to the end of the queue */
1371 	dl_list_add_tail(&hapd->sae_commit_queue, &q->list);
1372 
1373 queued:
1374 	if (eloop_is_timeout_registered(auth_sae_process_commit, hapd, NULL))
1375 		return;
1376 	eloop_register_timeout(0, queue_len * 10000, auth_sae_process_commit,
1377 			       hapd, NULL);
1378 }
1379 
1380 
1381 static int auth_sae_queued_addr(struct hostapd_data *hapd, const u8 *addr)
1382 {
1383 	struct hostapd_sae_commit_queue *q;
1384 	const struct ieee80211_mgmt *mgmt;
1385 
1386 	dl_list_for_each(q, &hapd->sae_commit_queue,
1387 			 struct hostapd_sae_commit_queue, list) {
1388 		mgmt = (const struct ieee80211_mgmt *) q->msg;
1389 		if (os_memcmp(addr, mgmt->sa, ETH_ALEN) == 0)
1390 			return 1;
1391 	}
1392 
1393 	return 0;
1394 }
1395 
1396 #endif /* CONFIG_SAE */
1397 
1398 
1399 static u16 wpa_res_to_status_code(int res)
1400 {
1401 	if (res == WPA_INVALID_GROUP)
1402 		return WLAN_STATUS_GROUP_CIPHER_NOT_VALID;
1403 	if (res == WPA_INVALID_PAIRWISE)
1404 		return WLAN_STATUS_PAIRWISE_CIPHER_NOT_VALID;
1405 	if (res == WPA_INVALID_AKMP)
1406 		return WLAN_STATUS_AKMP_NOT_VALID;
1407 	if (res == WPA_ALLOC_FAIL)
1408 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1409 #ifdef CONFIG_IEEE80211W
1410 	if (res == WPA_MGMT_FRAME_PROTECTION_VIOLATION)
1411 		return WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
1412 	if (res == WPA_INVALID_MGMT_GROUP_CIPHER)
1413 		return WLAN_STATUS_CIPHER_REJECTED_PER_POLICY;
1414 #endif /* CONFIG_IEEE80211W */
1415 	if (res == WPA_INVALID_MDIE)
1416 		return WLAN_STATUS_INVALID_MDIE;
1417 	if (res == WPA_INVALID_PMKID)
1418 		return WLAN_STATUS_INVALID_PMKID;
1419 	if (res != WPA_IE_OK)
1420 		return WLAN_STATUS_INVALID_IE;
1421 	return WLAN_STATUS_SUCCESS;
1422 }
1423 
1424 
1425 #ifdef CONFIG_FILS
1426 
1427 static void handle_auth_fils_finish(struct hostapd_data *hapd,
1428 				    struct sta_info *sta, u16 resp,
1429 				    struct wpabuf *data, int pub);
1430 
1431 void handle_auth_fils(struct hostapd_data *hapd, struct sta_info *sta,
1432 		      const u8 *pos, size_t len, u16 auth_alg,
1433 		      u16 auth_transaction, u16 status_code,
1434 		      void (*cb)(struct hostapd_data *hapd,
1435 				 struct sta_info *sta, u16 resp,
1436 				 struct wpabuf *data, int pub))
1437 {
1438 	u16 resp = WLAN_STATUS_SUCCESS;
1439 	const u8 *end;
1440 	struct ieee802_11_elems elems;
1441 	int res;
1442 	struct wpa_ie_data rsn;
1443 	struct rsn_pmksa_cache_entry *pmksa = NULL;
1444 
1445 	if (auth_transaction != 1 || status_code != WLAN_STATUS_SUCCESS)
1446 		return;
1447 
1448 	end = pos + len;
1449 
1450 	wpa_hexdump(MSG_DEBUG, "FILS: Authentication frame fields",
1451 		    pos, end - pos);
1452 
1453 	/* TODO: FILS PK */
1454 #ifdef CONFIG_FILS_SK_PFS
1455 	if (auth_alg == WLAN_AUTH_FILS_SK_PFS) {
1456 		u16 group;
1457 		struct wpabuf *pub;
1458 		size_t elem_len;
1459 
1460 		/* Using FILS PFS */
1461 
1462 		/* Finite Cyclic Group */
1463 		if (end - pos < 2) {
1464 			wpa_printf(MSG_DEBUG,
1465 				   "FILS: No room for Finite Cyclic Group");
1466 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1467 			goto fail;
1468 		}
1469 		group = WPA_GET_LE16(pos);
1470 		pos += 2;
1471 		if (group != hapd->conf->fils_dh_group) {
1472 			wpa_printf(MSG_DEBUG,
1473 				   "FILS: Unsupported Finite Cyclic Group: %u (expected %u)",
1474 				   group, hapd->conf->fils_dh_group);
1475 			resp = WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED;
1476 			goto fail;
1477 		}
1478 
1479 		crypto_ecdh_deinit(sta->fils_ecdh);
1480 		sta->fils_ecdh = crypto_ecdh_init(group);
1481 		if (!sta->fils_ecdh) {
1482 			wpa_printf(MSG_INFO,
1483 				   "FILS: Could not initialize ECDH with group %d",
1484 				   group);
1485 			resp = WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED;
1486 			goto fail;
1487 		}
1488 
1489 		pub = crypto_ecdh_get_pubkey(sta->fils_ecdh, 1);
1490 		if (!pub) {
1491 			wpa_printf(MSG_DEBUG,
1492 				   "FILS: Failed to derive ECDH public key");
1493 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1494 			goto fail;
1495 		}
1496 		elem_len = wpabuf_len(pub);
1497 		wpabuf_free(pub);
1498 
1499 		/* Element */
1500 		if ((size_t) (end - pos) < elem_len) {
1501 			wpa_printf(MSG_DEBUG, "FILS: No room for Element");
1502 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1503 			goto fail;
1504 		}
1505 
1506 		wpabuf_free(sta->fils_g_sta);
1507 		sta->fils_g_sta = wpabuf_alloc_copy(pos, elem_len);
1508 		wpabuf_clear_free(sta->fils_dh_ss);
1509 		sta->fils_dh_ss = crypto_ecdh_set_peerkey(sta->fils_ecdh, 1,
1510 							  pos, elem_len);
1511 		if (!sta->fils_dh_ss) {
1512 			wpa_printf(MSG_DEBUG, "FILS: ECDH operation failed");
1513 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1514 			goto fail;
1515 		}
1516 		wpa_hexdump_buf_key(MSG_DEBUG, "FILS: DH_SS", sta->fils_dh_ss);
1517 		pos += elem_len;
1518 	} else {
1519 		crypto_ecdh_deinit(sta->fils_ecdh);
1520 		sta->fils_ecdh = NULL;
1521 		wpabuf_clear_free(sta->fils_dh_ss);
1522 		sta->fils_dh_ss = NULL;
1523 	}
1524 #endif /* CONFIG_FILS_SK_PFS */
1525 
1526 	wpa_hexdump(MSG_DEBUG, "FILS: Remaining IEs", pos, end - pos);
1527 	if (ieee802_11_parse_elems(pos, end - pos, &elems, 1) == ParseFailed) {
1528 		wpa_printf(MSG_DEBUG, "FILS: Could not parse elements");
1529 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1530 		goto fail;
1531 	}
1532 
1533 	/* RSNE */
1534 	wpa_hexdump(MSG_DEBUG, "FILS: RSN element",
1535 		    elems.rsn_ie, elems.rsn_ie_len);
1536 	if (!elems.rsn_ie ||
1537 	    wpa_parse_wpa_ie_rsn(elems.rsn_ie - 2, elems.rsn_ie_len + 2,
1538 				 &rsn) < 0) {
1539 		wpa_printf(MSG_DEBUG, "FILS: No valid RSN element");
1540 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1541 		goto fail;
1542 	}
1543 
1544 	if (!sta->wpa_sm)
1545 		sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth, sta->addr,
1546 						NULL);
1547 	if (!sta->wpa_sm) {
1548 		wpa_printf(MSG_DEBUG,
1549 			   "FILS: Failed to initialize RSN state machine");
1550 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1551 		goto fail;
1552 	}
1553 
1554 	res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
1555 				  hapd->iface->freq,
1556 				  elems.rsn_ie - 2, elems.rsn_ie_len + 2,
1557 				  elems.mdie, elems.mdie_len, NULL, 0);
1558 	resp = wpa_res_to_status_code(res);
1559 	if (resp != WLAN_STATUS_SUCCESS)
1560 		goto fail;
1561 
1562 	if (!elems.fils_nonce) {
1563 		wpa_printf(MSG_DEBUG, "FILS: No FILS Nonce field");
1564 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1565 		goto fail;
1566 	}
1567 	wpa_hexdump(MSG_DEBUG, "FILS: SNonce", elems.fils_nonce,
1568 		    FILS_NONCE_LEN);
1569 	os_memcpy(sta->fils_snonce, elems.fils_nonce, FILS_NONCE_LEN);
1570 
1571 	/* PMKID List */
1572 	if (rsn.pmkid && rsn.num_pmkid > 0) {
1573 		u8 num;
1574 		const u8 *pmkid;
1575 
1576 		wpa_hexdump(MSG_DEBUG, "FILS: PMKID List",
1577 			    rsn.pmkid, rsn.num_pmkid * PMKID_LEN);
1578 
1579 		pmkid = rsn.pmkid;
1580 		num = rsn.num_pmkid;
1581 		while (num) {
1582 			wpa_hexdump(MSG_DEBUG, "FILS: PMKID", pmkid, PMKID_LEN);
1583 			pmksa = wpa_auth_pmksa_get(hapd->wpa_auth, sta->addr,
1584 						   pmkid);
1585 			if (pmksa)
1586 				break;
1587 			pmksa = wpa_auth_pmksa_get_fils_cache_id(hapd->wpa_auth,
1588 								 sta->addr,
1589 								 pmkid);
1590 			if (pmksa)
1591 				break;
1592 			pmkid += PMKID_LEN;
1593 			num--;
1594 		}
1595 	}
1596 	if (pmksa && wpa_auth_sta_key_mgmt(sta->wpa_sm) != pmksa->akmp) {
1597 		wpa_printf(MSG_DEBUG,
1598 			   "FILS: Matching PMKSA cache entry has different AKMP (0x%x != 0x%x) - ignore",
1599 			   wpa_auth_sta_key_mgmt(sta->wpa_sm), pmksa->akmp);
1600 		pmksa = NULL;
1601 	}
1602 	if (pmksa)
1603 		wpa_printf(MSG_DEBUG, "FILS: Found matching PMKSA cache entry");
1604 
1605 	/* FILS Session */
1606 	if (!elems.fils_session) {
1607 		wpa_printf(MSG_DEBUG, "FILS: No FILS Session element");
1608 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1609 		goto fail;
1610 	}
1611 	wpa_hexdump(MSG_DEBUG, "FILS: FILS Session", elems.fils_session,
1612 		    FILS_SESSION_LEN);
1613 	os_memcpy(sta->fils_session, elems.fils_session, FILS_SESSION_LEN);
1614 
1615 	/* FILS Wrapped Data */
1616 	if (elems.fils_wrapped_data) {
1617 		wpa_hexdump(MSG_DEBUG, "FILS: Wrapped Data",
1618 			    elems.fils_wrapped_data,
1619 			    elems.fils_wrapped_data_len);
1620 		if (!pmksa) {
1621 #ifndef CONFIG_NO_RADIUS
1622 			if (!sta->eapol_sm) {
1623 				sta->eapol_sm =
1624 					ieee802_1x_alloc_eapol_sm(hapd, sta);
1625 			}
1626 			wpa_printf(MSG_DEBUG,
1627 				   "FILS: Forward EAP-Initiate/Re-auth to authentication server");
1628 			ieee802_1x_encapsulate_radius(
1629 				hapd, sta, elems.fils_wrapped_data,
1630 				elems.fils_wrapped_data_len);
1631 			sta->fils_pending_cb = cb;
1632 			wpa_printf(MSG_DEBUG,
1633 				   "FILS: Will send Authentication frame once the response from authentication server is available");
1634 			sta->flags |= WLAN_STA_PENDING_FILS_ERP;
1635 			/* Calculate pending PMKID here so that we do not need
1636 			 * to maintain a copy of the EAP-Initiate/Reauth
1637 			 * message. */
1638 			if (fils_pmkid_erp(wpa_auth_sta_key_mgmt(sta->wpa_sm),
1639 					   elems.fils_wrapped_data,
1640 					   elems.fils_wrapped_data_len,
1641 					   sta->fils_erp_pmkid) == 0)
1642 				sta->fils_erp_pmkid_set = 1;
1643 			return;
1644 #else /* CONFIG_NO_RADIUS */
1645 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1646 			goto fail;
1647 #endif /* CONFIG_NO_RADIUS */
1648 		}
1649 	}
1650 
1651 fail:
1652 	if (cb) {
1653 		struct wpabuf *data;
1654 		int pub = 0;
1655 
1656 		data = prepare_auth_resp_fils(hapd, sta, &resp, pmksa, NULL,
1657 					      NULL, 0, &pub);
1658 		if (!data) {
1659 			wpa_printf(MSG_DEBUG,
1660 				   "%s: prepare_auth_resp_fils() returned failure",
1661 				   __func__);
1662 		}
1663 
1664 		cb(hapd, sta, resp, data, pub);
1665 	}
1666 }
1667 
1668 
1669 static struct wpabuf *
1670 prepare_auth_resp_fils(struct hostapd_data *hapd,
1671 		       struct sta_info *sta, u16 *resp,
1672 		       struct rsn_pmksa_cache_entry *pmksa,
1673 		       struct wpabuf *erp_resp,
1674 		       const u8 *msk, size_t msk_len,
1675 		       int *is_pub)
1676 {
1677 	u8 fils_nonce[FILS_NONCE_LEN];
1678 	size_t ielen;
1679 	struct wpabuf *data = NULL;
1680 	const u8 *ie;
1681 	u8 *ie_buf = NULL;
1682 	const u8 *pmk = NULL;
1683 	size_t pmk_len = 0;
1684 	u8 pmk_buf[PMK_LEN_MAX];
1685 	struct wpabuf *pub = NULL;
1686 
1687 	if (*resp != WLAN_STATUS_SUCCESS)
1688 		goto fail;
1689 
1690 	ie = wpa_auth_get_wpa_ie(hapd->wpa_auth, &ielen);
1691 	if (!ie) {
1692 		*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1693 		goto fail;
1694 	}
1695 
1696 	if (pmksa) {
1697 		/* Add PMKID of the selected PMKSA into RSNE */
1698 		ie_buf = os_malloc(ielen + 2 + 2 + PMKID_LEN);
1699 		if (!ie_buf) {
1700 			*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1701 			goto fail;
1702 		}
1703 
1704 		os_memcpy(ie_buf, ie, ielen);
1705 		if (wpa_insert_pmkid(ie_buf, &ielen, pmksa->pmkid) < 0) {
1706 			*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1707 			goto fail;
1708 		}
1709 		ie = ie_buf;
1710 	}
1711 
1712 	if (random_get_bytes(fils_nonce, FILS_NONCE_LEN) < 0) {
1713 		*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1714 		goto fail;
1715 	}
1716 	wpa_hexdump(MSG_DEBUG, "RSN: Generated FILS Nonce",
1717 		    fils_nonce, FILS_NONCE_LEN);
1718 
1719 #ifdef CONFIG_FILS_SK_PFS
1720 	if (sta->fils_dh_ss && sta->fils_ecdh) {
1721 		pub = crypto_ecdh_get_pubkey(sta->fils_ecdh, 1);
1722 		if (!pub) {
1723 			*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1724 			goto fail;
1725 		}
1726 	}
1727 #endif /* CONFIG_FILS_SK_PFS */
1728 
1729 	data = wpabuf_alloc(1000 + ielen + (pub ? wpabuf_len(pub) : 0));
1730 	if (!data) {
1731 		*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1732 		goto fail;
1733 	}
1734 
1735 	/* TODO: FILS PK */
1736 #ifdef CONFIG_FILS_SK_PFS
1737 	if (pub) {
1738 		/* Finite Cyclic Group */
1739 		wpabuf_put_le16(data, hapd->conf->fils_dh_group);
1740 
1741 		/* Element */
1742 		wpabuf_put_buf(data, pub);
1743 	}
1744 #endif /* CONFIG_FILS_SK_PFS */
1745 
1746 	/* RSNE */
1747 	wpabuf_put_data(data, ie, ielen);
1748 
1749 	/* MDE when using FILS+FT (already included in ie,ielen with RSNE) */
1750 
1751 #ifdef CONFIG_IEEE80211R_AP
1752 	if (wpa_key_mgmt_ft(wpa_auth_sta_key_mgmt(sta->wpa_sm))) {
1753 		/* FTE[R1KH-ID,R0KH-ID] when using FILS+FT */
1754 		int res;
1755 		int use_sha384 = wpa_key_mgmt_sha384(
1756 			wpa_auth_sta_key_mgmt(sta->wpa_sm));
1757 
1758 		res = wpa_auth_write_fte(hapd->wpa_auth, use_sha384,
1759 					 wpabuf_put(data, 0),
1760 					 wpabuf_tailroom(data));
1761 		if (res < 0) {
1762 			*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1763 			goto fail;
1764 		}
1765 		wpabuf_put(data, res);
1766 	}
1767 #endif /* CONFIG_IEEE80211R_AP */
1768 
1769 	/* FILS Nonce */
1770 	wpabuf_put_u8(data, WLAN_EID_EXTENSION); /* Element ID */
1771 	wpabuf_put_u8(data, 1 + FILS_NONCE_LEN); /* Length */
1772 	/* Element ID Extension */
1773 	wpabuf_put_u8(data, WLAN_EID_EXT_FILS_NONCE);
1774 	wpabuf_put_data(data, fils_nonce, FILS_NONCE_LEN);
1775 
1776 	/* FILS Session */
1777 	wpabuf_put_u8(data, WLAN_EID_EXTENSION); /* Element ID */
1778 	wpabuf_put_u8(data, 1 + FILS_SESSION_LEN); /* Length */
1779 	/* Element ID Extension */
1780 	wpabuf_put_u8(data, WLAN_EID_EXT_FILS_SESSION);
1781 	wpabuf_put_data(data, sta->fils_session, FILS_SESSION_LEN);
1782 
1783 	/* FILS Wrapped Data */
1784 	if (!pmksa && erp_resp) {
1785 		wpabuf_put_u8(data, WLAN_EID_EXTENSION); /* Element ID */
1786 		wpabuf_put_u8(data, 1 + wpabuf_len(erp_resp)); /* Length */
1787 		/* Element ID Extension */
1788 		wpabuf_put_u8(data, WLAN_EID_EXT_FILS_WRAPPED_DATA);
1789 		wpabuf_put_buf(data, erp_resp);
1790 
1791 		if (fils_rmsk_to_pmk(wpa_auth_sta_key_mgmt(sta->wpa_sm),
1792 				     msk, msk_len, sta->fils_snonce, fils_nonce,
1793 				     sta->fils_dh_ss ?
1794 				     wpabuf_head(sta->fils_dh_ss) : NULL,
1795 				     sta->fils_dh_ss ?
1796 				     wpabuf_len(sta->fils_dh_ss) : 0,
1797 				     pmk_buf, &pmk_len)) {
1798 			wpa_printf(MSG_DEBUG, "FILS: Failed to derive PMK");
1799 			*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1800 			wpabuf_free(data);
1801 			data = NULL;
1802 			goto fail;
1803 		}
1804 		pmk = pmk_buf;
1805 
1806 		/* Don't use DHss in PTK derivation if PMKSA caching is not
1807 		 * used. */
1808 		wpabuf_clear_free(sta->fils_dh_ss);
1809 		sta->fils_dh_ss = NULL;
1810 
1811 		if (sta->fils_erp_pmkid_set) {
1812 			/* TODO: get PMKLifetime from WPA parameters */
1813 			unsigned int dot11RSNAConfigPMKLifetime = 43200;
1814 			int session_timeout;
1815 
1816 			session_timeout = dot11RSNAConfigPMKLifetime;
1817 			if (sta->session_timeout_set) {
1818 				struct os_reltime now, diff;
1819 
1820 				os_get_reltime(&now);
1821 				os_reltime_sub(&sta->session_timeout, &now,
1822 					       &diff);
1823 				session_timeout = diff.sec;
1824 			}
1825 
1826 			sta->fils_erp_pmkid_set = 0;
1827 			wpa_auth_add_fils_pmk_pmkid(sta->wpa_sm, pmk, pmk_len,
1828 						    sta->fils_erp_pmkid);
1829 			if (!hapd->conf->disable_pmksa_caching &&
1830 			    wpa_auth_pmksa_add2(
1831 				    hapd->wpa_auth, sta->addr,
1832 				    pmk, pmk_len,
1833 				    sta->fils_erp_pmkid,
1834 				    session_timeout,
1835 				    wpa_auth_sta_key_mgmt(sta->wpa_sm)) < 0) {
1836 				wpa_printf(MSG_ERROR,
1837 					   "FILS: Failed to add PMKSA cache entry based on ERP");
1838 			}
1839 		}
1840 	} else if (pmksa) {
1841 		pmk = pmksa->pmk;
1842 		pmk_len = pmksa->pmk_len;
1843 	}
1844 
1845 	if (!pmk) {
1846 		wpa_printf(MSG_DEBUG, "FILS: No PMK available");
1847 		*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1848 		wpabuf_free(data);
1849 		data = NULL;
1850 		goto fail;
1851 	}
1852 
1853 	if (fils_auth_pmk_to_ptk(sta->wpa_sm, pmk, pmk_len,
1854 				 sta->fils_snonce, fils_nonce,
1855 				 sta->fils_dh_ss ?
1856 				 wpabuf_head(sta->fils_dh_ss) : NULL,
1857 				 sta->fils_dh_ss ?
1858 				 wpabuf_len(sta->fils_dh_ss) : 0,
1859 				 sta->fils_g_sta, pub) < 0) {
1860 		*resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1861 		wpabuf_free(data);
1862 		data = NULL;
1863 		goto fail;
1864 	}
1865 
1866 fail:
1867 	if (is_pub)
1868 		*is_pub = pub != NULL;
1869 	os_free(ie_buf);
1870 	wpabuf_free(pub);
1871 	wpabuf_clear_free(sta->fils_dh_ss);
1872 	sta->fils_dh_ss = NULL;
1873 #ifdef CONFIG_FILS_SK_PFS
1874 	crypto_ecdh_deinit(sta->fils_ecdh);
1875 	sta->fils_ecdh = NULL;
1876 #endif /* CONFIG_FILS_SK_PFS */
1877 	return data;
1878 }
1879 
1880 
1881 static void handle_auth_fils_finish(struct hostapd_data *hapd,
1882 				    struct sta_info *sta, u16 resp,
1883 				    struct wpabuf *data, int pub)
1884 {
1885 	u16 auth_alg;
1886 
1887 	auth_alg = (pub ||
1888 		    resp == WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED) ?
1889 		WLAN_AUTH_FILS_SK_PFS : WLAN_AUTH_FILS_SK;
1890 	send_auth_reply(hapd, sta->addr, hapd->own_addr, auth_alg, 2, resp,
1891 			data ? wpabuf_head(data) : (u8 *) "",
1892 			data ? wpabuf_len(data) : 0, "auth-fils-finish");
1893 	wpabuf_free(data);
1894 
1895 	if (resp == WLAN_STATUS_SUCCESS) {
1896 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
1897 			       HOSTAPD_LEVEL_DEBUG,
1898 			       "authentication OK (FILS)");
1899 		sta->flags |= WLAN_STA_AUTH;
1900 		wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
1901 		sta->auth_alg = pub ? WLAN_AUTH_FILS_SK_PFS : WLAN_AUTH_FILS_SK;
1902 		mlme_authenticate_indication(hapd, sta);
1903 	}
1904 }
1905 
1906 
1907 void ieee802_11_finish_fils_auth(struct hostapd_data *hapd,
1908 				 struct sta_info *sta, int success,
1909 				 struct wpabuf *erp_resp,
1910 				 const u8 *msk, size_t msk_len)
1911 {
1912 	struct wpabuf *data;
1913 	int pub = 0;
1914 	u16 resp;
1915 
1916 	sta->flags &= ~WLAN_STA_PENDING_FILS_ERP;
1917 
1918 	if (!sta->fils_pending_cb)
1919 		return;
1920 	resp = success ? WLAN_STATUS_SUCCESS : WLAN_STATUS_UNSPECIFIED_FAILURE;
1921 	data = prepare_auth_resp_fils(hapd, sta, &resp, NULL, erp_resp,
1922 				      msk, msk_len, &pub);
1923 	if (!data) {
1924 		wpa_printf(MSG_DEBUG,
1925 			   "%s: prepare_auth_resp_fils() returned failure",
1926 			   __func__);
1927 	}
1928 	sta->fils_pending_cb(hapd, sta, resp, data, pub);
1929 }
1930 
1931 #endif /* CONFIG_FILS */
1932 
1933 
1934 int
1935 ieee802_11_allowed_address(struct hostapd_data *hapd, const u8 *addr,
1936 			   const u8 *msg, size_t len, u32 *session_timeout,
1937 			   u32 *acct_interim_interval,
1938 			   struct vlan_description *vlan_id,
1939 			   struct hostapd_sta_wpa_psk_short **psk,
1940 			   char **identity, char **radius_cui, int is_probe_req)
1941 {
1942 	int res;
1943 
1944 	os_memset(vlan_id, 0, sizeof(*vlan_id));
1945 	res = hostapd_allowed_address(hapd, addr, msg, len,
1946 				      session_timeout, acct_interim_interval,
1947 				      vlan_id, psk, identity, radius_cui,
1948 				      is_probe_req);
1949 
1950 	if (res == HOSTAPD_ACL_REJECT) {
1951 		if (!is_probe_req)
1952 			wpa_printf(MSG_DEBUG,
1953 				   "Station " MACSTR
1954 				   " not allowed to authenticate",
1955 				   MAC2STR(addr));
1956 		return HOSTAPD_ACL_REJECT;
1957 	}
1958 
1959 	if (res == HOSTAPD_ACL_PENDING) {
1960 		wpa_printf(MSG_DEBUG, "Authentication frame from " MACSTR
1961 			   " waiting for an external authentication",
1962 			   MAC2STR(addr));
1963 		/* Authentication code will re-send the authentication frame
1964 		 * after it has received (and cached) information from the
1965 		 * external source. */
1966 		return HOSTAPD_ACL_PENDING;
1967 	}
1968 
1969 	return res;
1970 }
1971 
1972 
1973 static int
1974 ieee802_11_set_radius_info(struct hostapd_data *hapd, struct sta_info *sta,
1975 			   int res, u32 session_timeout,
1976 			   u32 acct_interim_interval,
1977 			   struct vlan_description *vlan_id,
1978 			   struct hostapd_sta_wpa_psk_short **psk,
1979 			   char **identity, char **radius_cui)
1980 {
1981 	if (vlan_id->notempty &&
1982 	    !hostapd_vlan_valid(hapd->conf->vlan, vlan_id)) {
1983 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
1984 			       HOSTAPD_LEVEL_INFO,
1985 			       "Invalid VLAN %d%s received from RADIUS server",
1986 			       vlan_id->untagged,
1987 			       vlan_id->tagged[0] ? "+" : "");
1988 		return -1;
1989 	}
1990 	if (ap_sta_set_vlan(hapd, sta, vlan_id) < 0)
1991 		return -1;
1992 	if (sta->vlan_id)
1993 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_RADIUS,
1994 			       HOSTAPD_LEVEL_INFO, "VLAN ID %d", sta->vlan_id);
1995 
1996 	hostapd_free_psk_list(sta->psk);
1997 	if (hapd->conf->wpa_psk_radius != PSK_RADIUS_IGNORED) {
1998 		sta->psk = *psk;
1999 		*psk = NULL;
2000 	} else {
2001 		sta->psk = NULL;
2002 	}
2003 
2004 	os_free(sta->identity);
2005 	sta->identity = *identity;
2006 	*identity = NULL;
2007 
2008 	os_free(sta->radius_cui);
2009 	sta->radius_cui = *radius_cui;
2010 	*radius_cui = NULL;
2011 
2012 	if (hapd->conf->acct_interim_interval == 0 && acct_interim_interval)
2013 		sta->acct_interim_interval = acct_interim_interval;
2014 	if (res == HOSTAPD_ACL_ACCEPT_TIMEOUT) {
2015 		sta->session_timeout_set = 1;
2016 		os_get_reltime(&sta->session_timeout);
2017 		sta->session_timeout.sec += session_timeout;
2018 		ap_sta_session_timeout(hapd, sta, session_timeout);
2019 	} else {
2020 		sta->session_timeout_set = 0;
2021 		ap_sta_no_session_timeout(hapd, sta);
2022 	}
2023 
2024 	return 0;
2025 }
2026 
2027 
2028 static void handle_auth(struct hostapd_data *hapd,
2029 			const struct ieee80211_mgmt *mgmt, size_t len,
2030 			int rssi, int from_queue)
2031 {
2032 	u16 auth_alg, auth_transaction, status_code;
2033 	u16 resp = WLAN_STATUS_SUCCESS;
2034 	struct sta_info *sta = NULL;
2035 	int res, reply_res;
2036 	u16 fc;
2037 	const u8 *challenge = NULL;
2038 	u32 session_timeout, acct_interim_interval;
2039 	struct vlan_description vlan_id;
2040 	struct hostapd_sta_wpa_psk_short *psk = NULL;
2041 	u8 resp_ies[2 + WLAN_AUTH_CHALLENGE_LEN];
2042 	size_t resp_ies_len = 0;
2043 	char *identity = NULL;
2044 	char *radius_cui = NULL;
2045 	u16 seq_ctrl;
2046 
2047 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
2048 		wpa_printf(MSG_INFO, "handle_auth - too short payload (len=%lu)",
2049 			   (unsigned long) len);
2050 		return;
2051 	}
2052 
2053 #ifdef CONFIG_TESTING_OPTIONS
2054 	if (hapd->iconf->ignore_auth_probability > 0.0 &&
2055 	    drand48() < hapd->iconf->ignore_auth_probability) {
2056 		wpa_printf(MSG_INFO,
2057 			   "TESTING: ignoring auth frame from " MACSTR,
2058 			   MAC2STR(mgmt->sa));
2059 		return;
2060 	}
2061 #endif /* CONFIG_TESTING_OPTIONS */
2062 
2063 	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
2064 	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
2065 	status_code = le_to_host16(mgmt->u.auth.status_code);
2066 	fc = le_to_host16(mgmt->frame_control);
2067 	seq_ctrl = le_to_host16(mgmt->seq_ctrl);
2068 
2069 	if (len >= IEEE80211_HDRLEN + sizeof(mgmt->u.auth) +
2070 	    2 + WLAN_AUTH_CHALLENGE_LEN &&
2071 	    mgmt->u.auth.variable[0] == WLAN_EID_CHALLENGE &&
2072 	    mgmt->u.auth.variable[1] == WLAN_AUTH_CHALLENGE_LEN)
2073 		challenge = &mgmt->u.auth.variable[2];
2074 
2075 	wpa_printf(MSG_DEBUG, "authentication: STA=" MACSTR " auth_alg=%d "
2076 		   "auth_transaction=%d status_code=%d wep=%d%s "
2077 		   "seq_ctrl=0x%x%s%s",
2078 		   MAC2STR(mgmt->sa), auth_alg, auth_transaction,
2079 		   status_code, !!(fc & WLAN_FC_ISWEP),
2080 		   challenge ? " challenge" : "",
2081 		   seq_ctrl, (fc & WLAN_FC_RETRY) ? " retry" : "",
2082 		   from_queue ? " (from queue)" : "");
2083 
2084 #ifdef CONFIG_NO_RC4
2085 	if (auth_alg == WLAN_AUTH_SHARED_KEY) {
2086 		wpa_printf(MSG_INFO,
2087 			   "Unsupported authentication algorithm (%d)",
2088 			   auth_alg);
2089 		resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
2090 		goto fail;
2091 	}
2092 #endif /* CONFIG_NO_RC4 */
2093 
2094 	if (hapd->tkip_countermeasures) {
2095 		wpa_printf(MSG_DEBUG,
2096 			   "Ongoing TKIP countermeasures (Michael MIC failure) - reject authentication");
2097 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2098 		goto fail;
2099 	}
2100 
2101 	if (!(((hapd->conf->auth_algs & WPA_AUTH_ALG_OPEN) &&
2102 	       auth_alg == WLAN_AUTH_OPEN) ||
2103 #ifdef CONFIG_IEEE80211R_AP
2104 	      (hapd->conf->wpa && wpa_key_mgmt_ft(hapd->conf->wpa_key_mgmt) &&
2105 	       auth_alg == WLAN_AUTH_FT) ||
2106 #endif /* CONFIG_IEEE80211R_AP */
2107 #ifdef CONFIG_SAE
2108 	      (hapd->conf->wpa && wpa_key_mgmt_sae(hapd->conf->wpa_key_mgmt) &&
2109 	       auth_alg == WLAN_AUTH_SAE) ||
2110 #endif /* CONFIG_SAE */
2111 #ifdef CONFIG_FILS
2112 	      (hapd->conf->wpa && wpa_key_mgmt_fils(hapd->conf->wpa_key_mgmt) &&
2113 	       auth_alg == WLAN_AUTH_FILS_SK) ||
2114 	      (hapd->conf->wpa && wpa_key_mgmt_fils(hapd->conf->wpa_key_mgmt) &&
2115 	       hapd->conf->fils_dh_group &&
2116 	       auth_alg == WLAN_AUTH_FILS_SK_PFS) ||
2117 #endif /* CONFIG_FILS */
2118 	      ((hapd->conf->auth_algs & WPA_AUTH_ALG_SHARED) &&
2119 	       auth_alg == WLAN_AUTH_SHARED_KEY))) {
2120 		wpa_printf(MSG_INFO, "Unsupported authentication algorithm (%d)",
2121 			   auth_alg);
2122 		resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
2123 		goto fail;
2124 	}
2125 
2126 	if (!(auth_transaction == 1 || auth_alg == WLAN_AUTH_SAE ||
2127 	      (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 3))) {
2128 		wpa_printf(MSG_INFO, "Unknown authentication transaction number (%d)",
2129 			   auth_transaction);
2130 		resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
2131 		goto fail;
2132 	}
2133 
2134 	if (os_memcmp(mgmt->sa, hapd->own_addr, ETH_ALEN) == 0) {
2135 		wpa_printf(MSG_INFO, "Station " MACSTR " not allowed to authenticate",
2136 			   MAC2STR(mgmt->sa));
2137 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2138 		goto fail;
2139 	}
2140 
2141 	if (hapd->conf->no_auth_if_seen_on) {
2142 		struct hostapd_data *other;
2143 
2144 		other = sta_track_seen_on(hapd->iface, mgmt->sa,
2145 					  hapd->conf->no_auth_if_seen_on);
2146 		if (other) {
2147 			u8 *pos;
2148 			u32 info;
2149 			u8 op_class, channel, phytype;
2150 
2151 			wpa_printf(MSG_DEBUG, "%s: Reject authentication from "
2152 				   MACSTR " since STA has been seen on %s",
2153 				   hapd->conf->iface, MAC2STR(mgmt->sa),
2154 				   hapd->conf->no_auth_if_seen_on);
2155 
2156 			resp = WLAN_STATUS_REJECTED_WITH_SUGGESTED_BSS_TRANSITION;
2157 			pos = &resp_ies[0];
2158 			*pos++ = WLAN_EID_NEIGHBOR_REPORT;
2159 			*pos++ = 13;
2160 			os_memcpy(pos, other->own_addr, ETH_ALEN);
2161 			pos += ETH_ALEN;
2162 			info = 0; /* TODO: BSSID Information */
2163 			WPA_PUT_LE32(pos, info);
2164 			pos += 4;
2165 			if (other->iconf->hw_mode == HOSTAPD_MODE_IEEE80211AD)
2166 				phytype = 8; /* dmg */
2167 			else if (other->iconf->ieee80211ac)
2168 				phytype = 9; /* vht */
2169 			else if (other->iconf->ieee80211n)
2170 				phytype = 7; /* ht */
2171 			else if (other->iconf->hw_mode ==
2172 				 HOSTAPD_MODE_IEEE80211A)
2173 				phytype = 4; /* ofdm */
2174 			else if (other->iconf->hw_mode ==
2175 				 HOSTAPD_MODE_IEEE80211G)
2176 				phytype = 6; /* erp */
2177 			else
2178 				phytype = 5; /* hrdsss */
2179 			if (ieee80211_freq_to_channel_ext(
2180 				    hostapd_hw_get_freq(other,
2181 							other->iconf->channel),
2182 				    other->iconf->secondary_channel,
2183 				    other->iconf->ieee80211ac,
2184 				    &op_class, &channel) == NUM_HOSTAPD_MODES) {
2185 				op_class = 0;
2186 				channel = other->iconf->channel;
2187 			}
2188 			*pos++ = op_class;
2189 			*pos++ = channel;
2190 			*pos++ = phytype;
2191 			resp_ies_len = pos - &resp_ies[0];
2192 			goto fail;
2193 		}
2194 	}
2195 
2196 	res = ieee802_11_allowed_address(
2197 		hapd, mgmt->sa, (const u8 *) mgmt, len, &session_timeout,
2198 		&acct_interim_interval, &vlan_id, &psk, &identity, &radius_cui,
2199 		0);
2200 	if (res == HOSTAPD_ACL_REJECT) {
2201 		wpa_msg(hapd->msg_ctx, MSG_DEBUG,
2202 			"Ignore Authentication frame from " MACSTR
2203 			" due to ACL reject", MAC2STR(mgmt->sa));
2204 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2205 		goto fail;
2206 	}
2207 	if (res == HOSTAPD_ACL_PENDING)
2208 		return;
2209 
2210 #ifdef CONFIG_SAE
2211 	if (auth_alg == WLAN_AUTH_SAE && !from_queue &&
2212 	    (auth_transaction == 1 ||
2213 	     (auth_transaction == 2 && auth_sae_queued_addr(hapd, mgmt->sa)))) {
2214 		/* Handle SAE Authentication commit message through a queue to
2215 		 * provide more control for postponing the needed heavy
2216 		 * processing under a possible DoS attack scenario. In addition,
2217 		 * queue SAE Authentication confirm message if there happens to
2218 		 * be a queued commit message from the same peer. This is needed
2219 		 * to avoid reordering Authentication frames within the same
2220 		 * SAE exchange. */
2221 		auth_sae_queue(hapd, mgmt, len, rssi);
2222 		return;
2223 	}
2224 #endif /* CONFIG_SAE */
2225 
2226 	sta = ap_get_sta(hapd, mgmt->sa);
2227 	if (sta) {
2228 		sta->flags &= ~WLAN_STA_PENDING_FILS_ERP;
2229 		sta->ft_over_ds = 0;
2230 		if ((fc & WLAN_FC_RETRY) &&
2231 		    sta->last_seq_ctrl != WLAN_INVALID_MGMT_SEQ &&
2232 		    sta->last_seq_ctrl == seq_ctrl &&
2233 		    sta->last_subtype == WLAN_FC_STYPE_AUTH) {
2234 			hostapd_logger(hapd, sta->addr,
2235 				       HOSTAPD_MODULE_IEEE80211,
2236 				       HOSTAPD_LEVEL_DEBUG,
2237 				       "Drop repeated authentication frame seq_ctrl=0x%x",
2238 				       seq_ctrl);
2239 			return;
2240 		}
2241 #ifdef CONFIG_MESH
2242 		if ((hapd->conf->mesh & MESH_ENABLED) &&
2243 		    sta->plink_state == PLINK_BLOCKED) {
2244 			wpa_printf(MSG_DEBUG, "Mesh peer " MACSTR
2245 				   " is blocked - drop Authentication frame",
2246 				   MAC2STR(mgmt->sa));
2247 			return;
2248 		}
2249 #endif /* CONFIG_MESH */
2250 	} else {
2251 #ifdef CONFIG_MESH
2252 		if (hapd->conf->mesh & MESH_ENABLED) {
2253 			/* if the mesh peer is not available, we don't do auth.
2254 			 */
2255 			wpa_printf(MSG_DEBUG, "Mesh peer " MACSTR
2256 				   " not yet known - drop Authentication frame",
2257 				   MAC2STR(mgmt->sa));
2258 			/*
2259 			 * Save a copy of the frame so that it can be processed
2260 			 * if a new peer entry is added shortly after this.
2261 			 */
2262 			wpabuf_free(hapd->mesh_pending_auth);
2263 			hapd->mesh_pending_auth = wpabuf_alloc_copy(mgmt, len);
2264 			os_get_reltime(&hapd->mesh_pending_auth_time);
2265 			return;
2266 		}
2267 #endif /* CONFIG_MESH */
2268 
2269 		sta = ap_sta_add(hapd, mgmt->sa);
2270 		if (!sta) {
2271 			wpa_printf(MSG_DEBUG, "ap_sta_add() failed");
2272 			resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
2273 			goto fail;
2274 		}
2275 	}
2276 	sta->last_seq_ctrl = seq_ctrl;
2277 	sta->last_subtype = WLAN_FC_STYPE_AUTH;
2278 #ifdef CONFIG_MBO
2279 	sta->auth_rssi = rssi;
2280 #endif /* CONFIG_MBO */
2281 
2282 	res = ieee802_11_set_radius_info(
2283 		hapd, sta, res, session_timeout, acct_interim_interval,
2284 		&vlan_id, &psk, &identity, &radius_cui);
2285 	if (res) {
2286 		wpa_printf(MSG_DEBUG, "ieee802_11_set_radius_info() failed");
2287 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2288 		goto fail;
2289 	}
2290 
2291 	sta->flags &= ~WLAN_STA_PREAUTH;
2292 	ieee802_1x_notify_pre_auth(sta->eapol_sm, 0);
2293 
2294 	/*
2295 	 * If the driver supports full AP client state, add a station to the
2296 	 * driver before sending authentication reply to make sure the driver
2297 	 * has resources, and not to go through the entire authentication and
2298 	 * association handshake, and fail it at the end.
2299 	 *
2300 	 * If this is not the first transaction, in a multi-step authentication
2301 	 * algorithm, the station already exists in the driver
2302 	 * (sta->added_unassoc = 1) so skip it.
2303 	 *
2304 	 * In mesh mode, the station was already added to the driver when the
2305 	 * NEW_PEER_CANDIDATE event is received.
2306 	 *
2307 	 * If PMF was negotiated for the existing association, skip this to
2308 	 * avoid dropping the STA entry and the associated keys. This is needed
2309 	 * to allow the original connection work until the attempt can complete
2310 	 * (re)association, so that unprotected Authentication frame cannot be
2311 	 * used to bypass PMF protection.
2312 	 */
2313 	if (FULL_AP_CLIENT_STATE_SUPP(hapd->iface->drv_flags) &&
2314 	    (!(sta->flags & WLAN_STA_MFP) || !ap_sta_is_authorized(sta)) &&
2315 	    !(hapd->conf->mesh & MESH_ENABLED) &&
2316 	    !(sta->added_unassoc)) {
2317 		/*
2318 		 * If a station that is already associated to the AP, is trying
2319 		 * to authenticate again, remove the STA entry, in order to make
2320 		 * sure the STA PS state gets cleared and configuration gets
2321 		 * updated. To handle this, station's added_unassoc flag is
2322 		 * cleared once the station has completed association.
2323 		 */
2324 		ap_sta_set_authorized(hapd, sta, 0);
2325 		hostapd_drv_sta_remove(hapd, sta->addr);
2326 		sta->flags &= ~(WLAN_STA_ASSOC | WLAN_STA_AUTH |
2327 				WLAN_STA_AUTHORIZED);
2328 
2329 		if (hostapd_sta_add(hapd, sta->addr, 0, 0,
2330 				    sta->supported_rates,
2331 				    sta->supported_rates_len,
2332 				    0, NULL, NULL, NULL, 0,
2333 				    sta->flags, 0, 0, 0, 0)) {
2334 			hostapd_logger(hapd, sta->addr,
2335 				       HOSTAPD_MODULE_IEEE80211,
2336 				       HOSTAPD_LEVEL_NOTICE,
2337 				       "Could not add STA to kernel driver");
2338 			resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
2339 			goto fail;
2340 		}
2341 
2342 		sta->added_unassoc = 1;
2343 	}
2344 
2345 	switch (auth_alg) {
2346 	case WLAN_AUTH_OPEN:
2347 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2348 			       HOSTAPD_LEVEL_DEBUG,
2349 			       "authentication OK (open system)");
2350 		sta->flags |= WLAN_STA_AUTH;
2351 		wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
2352 		sta->auth_alg = WLAN_AUTH_OPEN;
2353 		mlme_authenticate_indication(hapd, sta);
2354 		break;
2355 #ifndef CONFIG_NO_RC4
2356 	case WLAN_AUTH_SHARED_KEY:
2357 		resp = auth_shared_key(hapd, sta, auth_transaction, challenge,
2358 				       fc & WLAN_FC_ISWEP);
2359 		if (resp != 0)
2360 			wpa_printf(MSG_DEBUG,
2361 				   "auth_shared_key() failed: status=%d", resp);
2362 		sta->auth_alg = WLAN_AUTH_SHARED_KEY;
2363 		mlme_authenticate_indication(hapd, sta);
2364 		if (sta->challenge && auth_transaction == 1) {
2365 			resp_ies[0] = WLAN_EID_CHALLENGE;
2366 			resp_ies[1] = WLAN_AUTH_CHALLENGE_LEN;
2367 			os_memcpy(resp_ies + 2, sta->challenge,
2368 				  WLAN_AUTH_CHALLENGE_LEN);
2369 			resp_ies_len = 2 + WLAN_AUTH_CHALLENGE_LEN;
2370 		}
2371 		break;
2372 #endif /* CONFIG_NO_RC4 */
2373 #ifdef CONFIG_IEEE80211R_AP
2374 	case WLAN_AUTH_FT:
2375 		sta->auth_alg = WLAN_AUTH_FT;
2376 		if (sta->wpa_sm == NULL)
2377 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
2378 							sta->addr, NULL);
2379 		if (sta->wpa_sm == NULL) {
2380 			wpa_printf(MSG_DEBUG, "FT: Failed to initialize WPA "
2381 				   "state machine");
2382 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2383 			goto fail;
2384 		}
2385 		wpa_ft_process_auth(sta->wpa_sm, mgmt->bssid,
2386 				    auth_transaction, mgmt->u.auth.variable,
2387 				    len - IEEE80211_HDRLEN -
2388 				    sizeof(mgmt->u.auth),
2389 				    handle_auth_ft_finish, hapd);
2390 		/* handle_auth_ft_finish() callback will complete auth. */
2391 		return;
2392 #endif /* CONFIG_IEEE80211R_AP */
2393 #ifdef CONFIG_SAE
2394 	case WLAN_AUTH_SAE:
2395 #ifdef CONFIG_MESH
2396 		if (status_code == WLAN_STATUS_SUCCESS &&
2397 		    hapd->conf->mesh & MESH_ENABLED) {
2398 			if (sta->wpa_sm == NULL)
2399 				sta->wpa_sm =
2400 					wpa_auth_sta_init(hapd->wpa_auth,
2401 							  sta->addr, NULL);
2402 			if (sta->wpa_sm == NULL) {
2403 				wpa_printf(MSG_DEBUG,
2404 					   "SAE: Failed to initialize WPA state machine");
2405 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
2406 				goto fail;
2407 			}
2408 		}
2409 #endif /* CONFIG_MESH */
2410 		handle_auth_sae(hapd, sta, mgmt, len, auth_transaction,
2411 				status_code);
2412 		return;
2413 #endif /* CONFIG_SAE */
2414 #ifdef CONFIG_FILS
2415 	case WLAN_AUTH_FILS_SK:
2416 	case WLAN_AUTH_FILS_SK_PFS:
2417 		handle_auth_fils(hapd, sta, mgmt->u.auth.variable,
2418 				 len - IEEE80211_HDRLEN - sizeof(mgmt->u.auth),
2419 				 auth_alg, auth_transaction, status_code,
2420 				 handle_auth_fils_finish);
2421 		return;
2422 #endif /* CONFIG_FILS */
2423 	}
2424 
2425  fail:
2426 	os_free(identity);
2427 	os_free(radius_cui);
2428 	hostapd_free_psk_list(psk);
2429 
2430 	reply_res = send_auth_reply(hapd, mgmt->sa, mgmt->bssid, auth_alg,
2431 				    auth_transaction + 1, resp, resp_ies,
2432 				    resp_ies_len, "handle-auth");
2433 
2434 	if (sta && sta->added_unassoc && (resp != WLAN_STATUS_SUCCESS ||
2435 					  reply_res != WLAN_STATUS_SUCCESS)) {
2436 		hostapd_drv_sta_remove(hapd, sta->addr);
2437 		sta->added_unassoc = 0;
2438 	}
2439 }
2440 
2441 
2442 int hostapd_get_aid(struct hostapd_data *hapd, struct sta_info *sta)
2443 {
2444 	int i, j = 32, aid;
2445 
2446 	/* get a unique AID */
2447 	if (sta->aid > 0) {
2448 		wpa_printf(MSG_DEBUG, "  old AID %d", sta->aid);
2449 		return 0;
2450 	}
2451 
2452 	if (TEST_FAIL())
2453 		return -1;
2454 
2455 	for (i = 0; i < AID_WORDS; i++) {
2456 		if (hapd->sta_aid[i] == (u32) -1)
2457 			continue;
2458 		for (j = 0; j < 32; j++) {
2459 			if (!(hapd->sta_aid[i] & BIT(j)))
2460 				break;
2461 		}
2462 		if (j < 32)
2463 			break;
2464 	}
2465 	if (j == 32)
2466 		return -1;
2467 	aid = i * 32 + j + 1;
2468 	if (aid > 2007)
2469 		return -1;
2470 
2471 	sta->aid = aid;
2472 	hapd->sta_aid[i] |= BIT(j);
2473 	wpa_printf(MSG_DEBUG, "  new AID %d", sta->aid);
2474 	return 0;
2475 }
2476 
2477 
2478 static u16 check_ssid(struct hostapd_data *hapd, struct sta_info *sta,
2479 		      const u8 *ssid_ie, size_t ssid_ie_len)
2480 {
2481 	if (ssid_ie == NULL)
2482 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2483 
2484 	if (ssid_ie_len != hapd->conf->ssid.ssid_len ||
2485 	    os_memcmp(ssid_ie, hapd->conf->ssid.ssid, ssid_ie_len) != 0) {
2486 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2487 			       HOSTAPD_LEVEL_INFO,
2488 			       "Station tried to associate with unknown SSID "
2489 			       "'%s'", wpa_ssid_txt(ssid_ie, ssid_ie_len));
2490 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2491 	}
2492 
2493 	return WLAN_STATUS_SUCCESS;
2494 }
2495 
2496 
2497 static u16 check_wmm(struct hostapd_data *hapd, struct sta_info *sta,
2498 		     const u8 *wmm_ie, size_t wmm_ie_len)
2499 {
2500 	sta->flags &= ~WLAN_STA_WMM;
2501 	sta->qosinfo = 0;
2502 	if (wmm_ie && hapd->conf->wmm_enabled) {
2503 		struct wmm_information_element *wmm;
2504 
2505 		if (!hostapd_eid_wmm_valid(hapd, wmm_ie, wmm_ie_len)) {
2506 			hostapd_logger(hapd, sta->addr,
2507 				       HOSTAPD_MODULE_WPA,
2508 				       HOSTAPD_LEVEL_DEBUG,
2509 				       "invalid WMM element in association "
2510 				       "request");
2511 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
2512 		}
2513 
2514 		sta->flags |= WLAN_STA_WMM;
2515 		wmm = (struct wmm_information_element *) wmm_ie;
2516 		sta->qosinfo = wmm->qos_info;
2517 	}
2518 	return WLAN_STATUS_SUCCESS;
2519 }
2520 
2521 static u16 check_multi_ap(struct hostapd_data *hapd, struct sta_info *sta,
2522 			  const u8 *multi_ap_ie, size_t multi_ap_len)
2523 {
2524 	u8 multi_ap_value = 0;
2525 
2526 	sta->flags &= ~WLAN_STA_MULTI_AP;
2527 
2528 	if (!hapd->conf->multi_ap)
2529 		return WLAN_STATUS_SUCCESS;
2530 
2531 	if (multi_ap_ie) {
2532 		const u8 *multi_ap_subelem;
2533 
2534 		multi_ap_subelem = get_ie(multi_ap_ie + 4,
2535 					  multi_ap_len - 4,
2536 					  MULTI_AP_SUB_ELEM_TYPE);
2537 		if (multi_ap_subelem && multi_ap_subelem[1] == 1) {
2538 			multi_ap_value = multi_ap_subelem[2];
2539 		} else {
2540 			hostapd_logger(hapd, sta->addr,
2541 				       HOSTAPD_MODULE_IEEE80211,
2542 				       HOSTAPD_LEVEL_INFO,
2543 				       "Multi-AP IE has missing or invalid Multi-AP subelement");
2544 			return WLAN_STATUS_INVALID_IE;
2545 		}
2546 	}
2547 
2548 	if (multi_ap_value && multi_ap_value != MULTI_AP_BACKHAUL_STA)
2549 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2550 			       HOSTAPD_LEVEL_INFO,
2551 			       "Multi-AP IE with unexpected value 0x%02x",
2552 			       multi_ap_value);
2553 
2554 	if (!(multi_ap_value & MULTI_AP_BACKHAUL_STA)) {
2555 		if (hapd->conf->multi_ap & FRONTHAUL_BSS)
2556 			return WLAN_STATUS_SUCCESS;
2557 
2558 		hostapd_logger(hapd, sta->addr,
2559 			       HOSTAPD_MODULE_IEEE80211,
2560 			       HOSTAPD_LEVEL_INFO,
2561 			       "Non-Multi-AP STA tries to associate with backhaul-only BSS");
2562 		return WLAN_STATUS_ASSOC_DENIED_UNSPEC;
2563 	}
2564 
2565 	if (!(hapd->conf->multi_ap & BACKHAUL_BSS))
2566 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2567 			       HOSTAPD_LEVEL_DEBUG,
2568 			       "Backhaul STA tries to associate with fronthaul-only BSS");
2569 
2570 	sta->flags |= WLAN_STA_MULTI_AP;
2571 	return WLAN_STATUS_SUCCESS;
2572 }
2573 
2574 
2575 static u16 copy_supp_rates(struct hostapd_data *hapd, struct sta_info *sta,
2576 			   struct ieee802_11_elems *elems)
2577 {
2578 	/* Supported rates not used in IEEE 802.11ad/DMG */
2579 	if (hapd->iface->current_mode &&
2580 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211AD)
2581 		return WLAN_STATUS_SUCCESS;
2582 
2583 	if (!elems->supp_rates) {
2584 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2585 			       HOSTAPD_LEVEL_DEBUG,
2586 			       "No supported rates element in AssocReq");
2587 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2588 	}
2589 
2590 	if (elems->supp_rates_len + elems->ext_supp_rates_len >
2591 	    sizeof(sta->supported_rates)) {
2592 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2593 			       HOSTAPD_LEVEL_DEBUG,
2594 			       "Invalid supported rates element length %d+%d",
2595 			       elems->supp_rates_len,
2596 			       elems->ext_supp_rates_len);
2597 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2598 	}
2599 
2600 	sta->supported_rates_len = merge_byte_arrays(
2601 		sta->supported_rates, sizeof(sta->supported_rates),
2602 		elems->supp_rates, elems->supp_rates_len,
2603 		elems->ext_supp_rates, elems->ext_supp_rates_len);
2604 
2605 	return WLAN_STATUS_SUCCESS;
2606 }
2607 
2608 
2609 static u16 check_ext_capab(struct hostapd_data *hapd, struct sta_info *sta,
2610 			   const u8 *ext_capab_ie, size_t ext_capab_ie_len)
2611 {
2612 #ifdef CONFIG_INTERWORKING
2613 	/* check for QoS Map support */
2614 	if (ext_capab_ie_len >= 5) {
2615 		if (ext_capab_ie[4] & 0x01)
2616 			sta->qos_map_enabled = 1;
2617 	}
2618 #endif /* CONFIG_INTERWORKING */
2619 
2620 	if (ext_capab_ie_len > 0) {
2621 		sta->ecsa_supported = !!(ext_capab_ie[0] & BIT(2));
2622 		os_free(sta->ext_capability);
2623 		sta->ext_capability = os_malloc(1 + ext_capab_ie_len);
2624 		if (sta->ext_capability) {
2625 			sta->ext_capability[0] = ext_capab_ie_len;
2626 			os_memcpy(sta->ext_capability + 1, ext_capab_ie,
2627 				  ext_capab_ie_len);
2628 		}
2629 	}
2630 
2631 	return WLAN_STATUS_SUCCESS;
2632 }
2633 
2634 
2635 #ifdef CONFIG_OWE
2636 
2637 static int owe_group_supported(struct hostapd_data *hapd, u16 group)
2638 {
2639 	int i;
2640 	int *groups = hapd->conf->owe_groups;
2641 
2642 	if (group != 19 && group != 20 && group != 21)
2643 		return 0;
2644 
2645 	if (!groups)
2646 		return 1;
2647 
2648 	for (i = 0; groups[i] > 0; i++) {
2649 		if (groups[i] == group)
2650 			return 1;
2651 	}
2652 
2653 	return 0;
2654 }
2655 
2656 
2657 static u16 owe_process_assoc_req(struct hostapd_data *hapd,
2658 				 struct sta_info *sta, const u8 *owe_dh,
2659 				 u8 owe_dh_len)
2660 {
2661 	struct wpabuf *secret, *pub, *hkey;
2662 	int res;
2663 	u8 prk[SHA512_MAC_LEN], pmkid[SHA512_MAC_LEN];
2664 	const char *info = "OWE Key Generation";
2665 	const u8 *addr[2];
2666 	size_t len[2];
2667 	u16 group;
2668 	size_t hash_len, prime_len;
2669 
2670 	if (wpa_auth_sta_get_pmksa(sta->wpa_sm)) {
2671 		wpa_printf(MSG_DEBUG, "OWE: Using PMKSA caching");
2672 		return WLAN_STATUS_SUCCESS;
2673 	}
2674 
2675 	group = WPA_GET_LE16(owe_dh);
2676 	if (!owe_group_supported(hapd, group)) {
2677 		wpa_printf(MSG_DEBUG, "OWE: Unsupported DH group %u", group);
2678 		return WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED;
2679 	}
2680 	if (group == 19)
2681 		prime_len = 32;
2682 	else if (group == 20)
2683 		prime_len = 48;
2684 	else if (group == 21)
2685 		prime_len = 66;
2686 	else
2687 		return WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED;
2688 
2689 	crypto_ecdh_deinit(sta->owe_ecdh);
2690 	sta->owe_ecdh = crypto_ecdh_init(group);
2691 	if (!sta->owe_ecdh)
2692 		return WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED;
2693 	sta->owe_group = group;
2694 
2695 	secret = crypto_ecdh_set_peerkey(sta->owe_ecdh, 0, owe_dh + 2,
2696 					 owe_dh_len - 2);
2697 	secret = wpabuf_zeropad(secret, prime_len);
2698 	if (!secret) {
2699 		wpa_printf(MSG_DEBUG, "OWE: Invalid peer DH public key");
2700 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2701 	}
2702 	wpa_hexdump_buf_key(MSG_DEBUG, "OWE: DH shared secret", secret);
2703 
2704 	/* prk = HKDF-extract(C | A | group, z) */
2705 
2706 	pub = crypto_ecdh_get_pubkey(sta->owe_ecdh, 0);
2707 	if (!pub) {
2708 		wpabuf_clear_free(secret);
2709 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2710 	}
2711 
2712 	/* PMKID = Truncate-128(Hash(C | A)) */
2713 	addr[0] = owe_dh + 2;
2714 	len[0] = owe_dh_len - 2;
2715 	addr[1] = wpabuf_head(pub);
2716 	len[1] = wpabuf_len(pub);
2717 	if (group == 19) {
2718 		res = sha256_vector(2, addr, len, pmkid);
2719 		hash_len = SHA256_MAC_LEN;
2720 	} else if (group == 20) {
2721 		res = sha384_vector(2, addr, len, pmkid);
2722 		hash_len = SHA384_MAC_LEN;
2723 	} else if (group == 21) {
2724 		res = sha512_vector(2, addr, len, pmkid);
2725 		hash_len = SHA512_MAC_LEN;
2726 	} else {
2727 		wpabuf_free(pub);
2728 		wpabuf_clear_free(secret);
2729 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2730 	}
2731 	pub = wpabuf_zeropad(pub, prime_len);
2732 	if (res < 0 || !pub) {
2733 		wpabuf_free(pub);
2734 		wpabuf_clear_free(secret);
2735 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2736 	}
2737 
2738 	hkey = wpabuf_alloc(owe_dh_len - 2 + wpabuf_len(pub) + 2);
2739 	if (!hkey) {
2740 		wpabuf_free(pub);
2741 		wpabuf_clear_free(secret);
2742 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2743 	}
2744 
2745 	wpabuf_put_data(hkey, owe_dh + 2, owe_dh_len - 2); /* C */
2746 	wpabuf_put_buf(hkey, pub); /* A */
2747 	wpabuf_free(pub);
2748 	wpabuf_put_le16(hkey, group); /* group */
2749 	if (group == 19)
2750 		res = hmac_sha256(wpabuf_head(hkey), wpabuf_len(hkey),
2751 				  wpabuf_head(secret), wpabuf_len(secret), prk);
2752 	else if (group == 20)
2753 		res = hmac_sha384(wpabuf_head(hkey), wpabuf_len(hkey),
2754 				  wpabuf_head(secret), wpabuf_len(secret), prk);
2755 	else if (group == 21)
2756 		res = hmac_sha512(wpabuf_head(hkey), wpabuf_len(hkey),
2757 				  wpabuf_head(secret), wpabuf_len(secret), prk);
2758 	wpabuf_clear_free(hkey);
2759 	wpabuf_clear_free(secret);
2760 	if (res < 0)
2761 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2762 
2763 	wpa_hexdump_key(MSG_DEBUG, "OWE: prk", prk, hash_len);
2764 
2765 	/* PMK = HKDF-expand(prk, "OWE Key Generation", n) */
2766 
2767 	os_free(sta->owe_pmk);
2768 	sta->owe_pmk = os_malloc(hash_len);
2769 	if (!sta->owe_pmk) {
2770 		os_memset(prk, 0, SHA512_MAC_LEN);
2771 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2772 	}
2773 
2774 	if (group == 19)
2775 		res = hmac_sha256_kdf(prk, hash_len, NULL, (const u8 *) info,
2776 				      os_strlen(info), sta->owe_pmk, hash_len);
2777 	else if (group == 20)
2778 		res = hmac_sha384_kdf(prk, hash_len, NULL, (const u8 *) info,
2779 				      os_strlen(info), sta->owe_pmk, hash_len);
2780 	else if (group == 21)
2781 		res = hmac_sha512_kdf(prk, hash_len, NULL, (const u8 *) info,
2782 				      os_strlen(info), sta->owe_pmk, hash_len);
2783 	os_memset(prk, 0, SHA512_MAC_LEN);
2784 	if (res < 0) {
2785 		os_free(sta->owe_pmk);
2786 		sta->owe_pmk = NULL;
2787 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2788 	}
2789 	sta->owe_pmk_len = hash_len;
2790 
2791 	wpa_hexdump_key(MSG_DEBUG, "OWE: PMK", sta->owe_pmk, sta->owe_pmk_len);
2792 	wpa_hexdump(MSG_DEBUG, "OWE: PMKID", pmkid, PMKID_LEN);
2793 	wpa_auth_pmksa_add2(hapd->wpa_auth, sta->addr, sta->owe_pmk,
2794 			    sta->owe_pmk_len, pmkid, 0, WPA_KEY_MGMT_OWE);
2795 
2796 	return WLAN_STATUS_SUCCESS;
2797 }
2798 
2799 
2800 u16 owe_validate_request(struct hostapd_data *hapd, const u8 *peer,
2801 			 const u8 *rsn_ie, size_t rsn_ie_len,
2802 			 const u8 *owe_dh, size_t owe_dh_len)
2803 {
2804 	struct wpa_ie_data data;
2805 	int res;
2806 
2807 	if (!rsn_ie || rsn_ie_len < 2) {
2808 		wpa_printf(MSG_DEBUG, "OWE: Invalid RSNE from " MACSTR,
2809 			   MAC2STR(peer));
2810 		return WLAN_STATUS_INVALID_IE;
2811 	}
2812 	rsn_ie -= 2;
2813 	rsn_ie_len += 2;
2814 
2815 	res = wpa_parse_wpa_ie_rsn(rsn_ie, rsn_ie_len, &data);
2816 	if (res) {
2817 		wpa_printf(MSG_DEBUG, "Failed to parse RSNE from " MACSTR
2818 			   " (res=%d)", MAC2STR(peer), res);
2819 		wpa_hexdump(MSG_DEBUG, "RSNE", rsn_ie, rsn_ie_len);
2820 		return wpa_res_to_status_code(res);
2821 	}
2822 	if (!(data.key_mgmt & WPA_KEY_MGMT_OWE)) {
2823 		wpa_printf(MSG_DEBUG,
2824 			   "OWE: Unexpected key mgmt 0x%x from " MACSTR,
2825 			   (unsigned int) data.key_mgmt, MAC2STR(peer));
2826 		return WLAN_STATUS_AKMP_NOT_VALID;
2827 	}
2828 	if (!owe_dh) {
2829 		wpa_printf(MSG_DEBUG,
2830 			   "OWE: No Diffie-Hellman Parameter element from "
2831 			   MACSTR, MAC2STR(peer));
2832 		return WLAN_STATUS_AKMP_NOT_VALID;
2833 	}
2834 
2835 	return WLAN_STATUS_SUCCESS;
2836 }
2837 
2838 
2839 u16 owe_process_rsn_ie(struct hostapd_data *hapd,
2840 		       struct sta_info *sta,
2841 		       const u8 *rsn_ie, size_t rsn_ie_len,
2842 		       const u8 *owe_dh, size_t owe_dh_len)
2843 {
2844 	u16 status;
2845 	u8 *owe_buf, ie[256 * 2];
2846 	size_t ie_len = 0;
2847 	int res;
2848 
2849 	if (!rsn_ie || rsn_ie_len < 2) {
2850 		wpa_printf(MSG_DEBUG, "OWE: No RSNE in (Re)AssocReq");
2851 		status = WLAN_STATUS_INVALID_IE;
2852 		goto end;
2853 	}
2854 
2855 	if (!sta->wpa_sm)
2856 		sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,	sta->addr,
2857 						NULL);
2858 	if (!sta->wpa_sm) {
2859 		wpa_printf(MSG_WARNING,
2860 			   "OWE: Failed to initialize WPA state machine");
2861 		status = WLAN_STATUS_UNSPECIFIED_FAILURE;
2862 		goto end;
2863 	}
2864 	rsn_ie -= 2;
2865 	rsn_ie_len += 2;
2866 	res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
2867 				  hapd->iface->freq, rsn_ie, rsn_ie_len,
2868 				  NULL, 0, owe_dh, owe_dh_len);
2869 	status = wpa_res_to_status_code(res);
2870 	if (status != WLAN_STATUS_SUCCESS)
2871 		goto end;
2872 	status = owe_process_assoc_req(hapd, sta, owe_dh, owe_dh_len);
2873 	if (status != WLAN_STATUS_SUCCESS)
2874 		goto end;
2875 	owe_buf = wpa_auth_write_assoc_resp_owe(sta->wpa_sm, ie, sizeof(ie),
2876 						NULL, 0);
2877 	if (!owe_buf) {
2878 		status = WLAN_STATUS_UNSPECIFIED_FAILURE;
2879 		goto end;
2880 	}
2881 
2882 	if (sta->owe_ecdh) {
2883 		struct wpabuf *pub;
2884 
2885 		pub = crypto_ecdh_get_pubkey(sta->owe_ecdh, 0);
2886 		if (!pub) {
2887 			status = WLAN_STATUS_UNSPECIFIED_FAILURE;
2888 			goto end;
2889 		}
2890 
2891 		/* OWE Diffie-Hellman Parameter element */
2892 		*owe_buf++ = WLAN_EID_EXTENSION; /* Element ID */
2893 		*owe_buf++ = 1 + 2 + wpabuf_len(pub); /* Length */
2894 		*owe_buf++ = WLAN_EID_EXT_OWE_DH_PARAM; /* Element ID Extension
2895 							 */
2896 		WPA_PUT_LE16(owe_buf, sta->owe_group);
2897 		owe_buf += 2;
2898 		os_memcpy(owe_buf, wpabuf_head(pub), wpabuf_len(pub));
2899 		owe_buf += wpabuf_len(pub);
2900 		wpabuf_free(pub);
2901 		sta->external_dh_updated = 1;
2902 	}
2903 	ie_len = owe_buf - ie;
2904 
2905 end:
2906 	wpa_printf(MSG_DEBUG, "OWE: Update status %d, ie len %d for peer "
2907 			      MACSTR, status, (unsigned int) ie_len,
2908 			      MAC2STR(sta->addr));
2909 	hostapd_drv_update_dh_ie(hapd, sta->addr, status,
2910 				 status == WLAN_STATUS_SUCCESS ? ie : NULL,
2911 				 ie_len);
2912 
2913 	return status;
2914 }
2915 
2916 #endif /* CONFIG_OWE */
2917 
2918 
2919 static u16 check_assoc_ies(struct hostapd_data *hapd, struct sta_info *sta,
2920 			   const u8 *ies, size_t ies_len, int reassoc)
2921 {
2922 	struct ieee802_11_elems elems;
2923 	u16 resp;
2924 	const u8 *wpa_ie;
2925 	size_t wpa_ie_len;
2926 	const u8 *p2p_dev_addr = NULL;
2927 
2928 	if (ieee802_11_parse_elems(ies, ies_len, &elems, 1) == ParseFailed) {
2929 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2930 			       HOSTAPD_LEVEL_INFO, "Station sent an invalid "
2931 			       "association request");
2932 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
2933 	}
2934 
2935 	resp = check_ssid(hapd, sta, elems.ssid, elems.ssid_len);
2936 	if (resp != WLAN_STATUS_SUCCESS)
2937 		return resp;
2938 	resp = check_wmm(hapd, sta, elems.wmm, elems.wmm_len);
2939 	if (resp != WLAN_STATUS_SUCCESS)
2940 		return resp;
2941 	resp = check_ext_capab(hapd, sta, elems.ext_capab, elems.ext_capab_len);
2942 	if (resp != WLAN_STATUS_SUCCESS)
2943 		return resp;
2944 	resp = copy_supp_rates(hapd, sta, &elems);
2945 	if (resp != WLAN_STATUS_SUCCESS)
2946 		return resp;
2947 
2948 	resp = check_multi_ap(hapd, sta, elems.multi_ap, elems.multi_ap_len);
2949 	if (resp != WLAN_STATUS_SUCCESS)
2950 		return resp;
2951 
2952 #ifdef CONFIG_IEEE80211N
2953 	resp = copy_sta_ht_capab(hapd, sta, elems.ht_capabilities);
2954 	if (resp != WLAN_STATUS_SUCCESS)
2955 		return resp;
2956 	if (hapd->iconf->ieee80211n && hapd->iconf->require_ht &&
2957 	    !(sta->flags & WLAN_STA_HT)) {
2958 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2959 			       HOSTAPD_LEVEL_INFO, "Station does not support "
2960 			       "mandatory HT PHY - reject association");
2961 		return WLAN_STATUS_ASSOC_DENIED_NO_HT;
2962 	}
2963 #endif /* CONFIG_IEEE80211N */
2964 
2965 #ifdef CONFIG_IEEE80211AC
2966 	if (hapd->iconf->ieee80211ac) {
2967 		resp = copy_sta_vht_capab(hapd, sta, elems.vht_capabilities);
2968 		if (resp != WLAN_STATUS_SUCCESS)
2969 			return resp;
2970 
2971 		resp = set_sta_vht_opmode(hapd, sta, elems.vht_opmode_notif);
2972 		if (resp != WLAN_STATUS_SUCCESS)
2973 			return resp;
2974 	}
2975 
2976 	if (hapd->iconf->ieee80211ac && hapd->iconf->require_vht &&
2977 	    !(sta->flags & WLAN_STA_VHT)) {
2978 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
2979 			       HOSTAPD_LEVEL_INFO, "Station does not support "
2980 			       "mandatory VHT PHY - reject association");
2981 		return WLAN_STATUS_ASSOC_DENIED_NO_VHT;
2982 	}
2983 
2984 	if (hapd->conf->vendor_vht && !elems.vht_capabilities) {
2985 		resp = copy_sta_vendor_vht(hapd, sta, elems.vendor_vht,
2986 					   elems.vendor_vht_len);
2987 		if (resp != WLAN_STATUS_SUCCESS)
2988 			return resp;
2989 	}
2990 #endif /* CONFIG_IEEE80211AC */
2991 #ifdef CONFIG_IEEE80211AX
2992 	if (hapd->iconf->ieee80211ax) {
2993 		resp = copy_sta_he_capab(hapd, sta, IEEE80211_MODE_AP,
2994 					 elems.he_capabilities,
2995 					 elems.he_capabilities_len);
2996 		if (resp != WLAN_STATUS_SUCCESS)
2997 			return resp;
2998 	}
2999 #endif /* CONFIG_IEEE80211AX */
3000 
3001 #ifdef CONFIG_P2P
3002 	if (elems.p2p) {
3003 		wpabuf_free(sta->p2p_ie);
3004 		sta->p2p_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
3005 							  P2P_IE_VENDOR_TYPE);
3006 		if (sta->p2p_ie)
3007 			p2p_dev_addr = p2p_get_go_dev_addr(sta->p2p_ie);
3008 	} else {
3009 		wpabuf_free(sta->p2p_ie);
3010 		sta->p2p_ie = NULL;
3011 	}
3012 #endif /* CONFIG_P2P */
3013 
3014 	if ((hapd->conf->wpa & WPA_PROTO_RSN) && elems.rsn_ie) {
3015 		wpa_ie = elems.rsn_ie;
3016 		wpa_ie_len = elems.rsn_ie_len;
3017 	} else if ((hapd->conf->wpa & WPA_PROTO_WPA) &&
3018 		   elems.wpa_ie) {
3019 		wpa_ie = elems.wpa_ie;
3020 		wpa_ie_len = elems.wpa_ie_len;
3021 	} else {
3022 		wpa_ie = NULL;
3023 		wpa_ie_len = 0;
3024 	}
3025 
3026 #ifdef CONFIG_WPS
3027 	sta->flags &= ~(WLAN_STA_WPS | WLAN_STA_MAYBE_WPS | WLAN_STA_WPS2);
3028 	if (hapd->conf->wps_state && elems.wps_ie) {
3029 		wpa_printf(MSG_DEBUG, "STA included WPS IE in (Re)Association "
3030 			   "Request - assume WPS is used");
3031 		sta->flags |= WLAN_STA_WPS;
3032 		wpabuf_free(sta->wps_ie);
3033 		sta->wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len,
3034 							  WPS_IE_VENDOR_TYPE);
3035 		if (sta->wps_ie && wps_is_20(sta->wps_ie)) {
3036 			wpa_printf(MSG_DEBUG, "WPS: STA supports WPS 2.0");
3037 			sta->flags |= WLAN_STA_WPS2;
3038 		}
3039 		wpa_ie = NULL;
3040 		wpa_ie_len = 0;
3041 		if (sta->wps_ie && wps_validate_assoc_req(sta->wps_ie) < 0) {
3042 			wpa_printf(MSG_DEBUG, "WPS: Invalid WPS IE in "
3043 				   "(Re)Association Request - reject");
3044 			return WLAN_STATUS_INVALID_IE;
3045 		}
3046 	} else if (hapd->conf->wps_state && wpa_ie == NULL) {
3047 		wpa_printf(MSG_DEBUG, "STA did not include WPA/RSN IE in "
3048 			   "(Re)Association Request - possible WPS use");
3049 		sta->flags |= WLAN_STA_MAYBE_WPS;
3050 	} else
3051 #endif /* CONFIG_WPS */
3052 	if (hapd->conf->wpa && wpa_ie == NULL) {
3053 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
3054 			       HOSTAPD_LEVEL_INFO,
3055 			       "No WPA/RSN IE in association request");
3056 		return WLAN_STATUS_INVALID_IE;
3057 	}
3058 
3059 	if (hapd->conf->wpa && wpa_ie) {
3060 		int res;
3061 		wpa_ie -= 2;
3062 		wpa_ie_len += 2;
3063 		if (sta->wpa_sm == NULL)
3064 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
3065 							sta->addr,
3066 							p2p_dev_addr);
3067 		if (sta->wpa_sm == NULL) {
3068 			wpa_printf(MSG_WARNING, "Failed to initialize WPA "
3069 				   "state machine");
3070 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3071 		}
3072 		wpa_auth_set_auth_alg(sta->wpa_sm, sta->auth_alg);
3073 		res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
3074 					  hapd->iface->freq,
3075 					  wpa_ie, wpa_ie_len,
3076 					  elems.mdie, elems.mdie_len,
3077 					  elems.owe_dh, elems.owe_dh_len);
3078 		resp = wpa_res_to_status_code(res);
3079 		if (resp != WLAN_STATUS_SUCCESS)
3080 			return resp;
3081 #ifdef CONFIG_IEEE80211W
3082 		if ((sta->flags & (WLAN_STA_ASSOC | WLAN_STA_MFP)) ==
3083 		    (WLAN_STA_ASSOC | WLAN_STA_MFP) &&
3084 		    !sta->sa_query_timed_out &&
3085 		    sta->sa_query_count > 0)
3086 			ap_check_sa_query_timeout(hapd, sta);
3087 		if ((sta->flags & (WLAN_STA_ASSOC | WLAN_STA_MFP)) ==
3088 		    (WLAN_STA_ASSOC | WLAN_STA_MFP) &&
3089 		    !sta->sa_query_timed_out &&
3090 		    (!reassoc || sta->auth_alg != WLAN_AUTH_FT)) {
3091 			/*
3092 			 * STA has already been associated with MFP and SA
3093 			 * Query timeout has not been reached. Reject the
3094 			 * association attempt temporarily and start SA Query,
3095 			 * if one is not pending.
3096 			 */
3097 
3098 			if (sta->sa_query_count == 0)
3099 				ap_sta_start_sa_query(hapd, sta);
3100 
3101 			return WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY;
3102 		}
3103 
3104 		if (wpa_auth_uses_mfp(sta->wpa_sm))
3105 			sta->flags |= WLAN_STA_MFP;
3106 		else
3107 			sta->flags &= ~WLAN_STA_MFP;
3108 #endif /* CONFIG_IEEE80211W */
3109 
3110 #ifdef CONFIG_IEEE80211R_AP
3111 		if (sta->auth_alg == WLAN_AUTH_FT) {
3112 			if (!reassoc) {
3113 				wpa_printf(MSG_DEBUG, "FT: " MACSTR " tried "
3114 					   "to use association (not "
3115 					   "re-association) with FT auth_alg",
3116 					   MAC2STR(sta->addr));
3117 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
3118 			}
3119 
3120 			resp = wpa_ft_validate_reassoc(sta->wpa_sm, ies,
3121 						       ies_len);
3122 			if (resp != WLAN_STATUS_SUCCESS)
3123 				return resp;
3124 		}
3125 #endif /* CONFIG_IEEE80211R_AP */
3126 
3127 #ifdef CONFIG_SAE
3128 		if (wpa_auth_uses_sae(sta->wpa_sm) && sta->sae &&
3129 		    sta->sae->state == SAE_ACCEPTED)
3130 			wpa_auth_add_sae_pmkid(sta->wpa_sm, sta->sae->pmkid);
3131 
3132 		if (wpa_auth_uses_sae(sta->wpa_sm) &&
3133 		    sta->auth_alg == WLAN_AUTH_OPEN) {
3134 			struct rsn_pmksa_cache_entry *sa;
3135 			sa = wpa_auth_sta_get_pmksa(sta->wpa_sm);
3136 			if (!sa || sa->akmp != WPA_KEY_MGMT_SAE) {
3137 				wpa_printf(MSG_DEBUG,
3138 					   "SAE: No PMKSA cache entry found for "
3139 					   MACSTR, MAC2STR(sta->addr));
3140 				return WLAN_STATUS_INVALID_PMKID;
3141 			}
3142 			wpa_printf(MSG_DEBUG, "SAE: " MACSTR
3143 				   " using PMKSA caching", MAC2STR(sta->addr));
3144 		} else if (wpa_auth_uses_sae(sta->wpa_sm) &&
3145 			   sta->auth_alg != WLAN_AUTH_SAE &&
3146 			   !(sta->auth_alg == WLAN_AUTH_FT &&
3147 			     wpa_auth_uses_ft_sae(sta->wpa_sm))) {
3148 			wpa_printf(MSG_DEBUG, "SAE: " MACSTR " tried to use "
3149 				   "SAE AKM after non-SAE auth_alg %u",
3150 				   MAC2STR(sta->addr), sta->auth_alg);
3151 			return WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
3152 		}
3153 #endif /* CONFIG_SAE */
3154 
3155 #ifdef CONFIG_OWE
3156 		if ((hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_OWE) &&
3157 		    wpa_auth_sta_key_mgmt(sta->wpa_sm) == WPA_KEY_MGMT_OWE &&
3158 		    elems.owe_dh) {
3159 			resp = owe_process_assoc_req(hapd, sta, elems.owe_dh,
3160 						     elems.owe_dh_len);
3161 			if (resp != WLAN_STATUS_SUCCESS)
3162 				return resp;
3163 		}
3164 #endif /* CONFIG_OWE */
3165 
3166 #ifdef CONFIG_DPP2
3167 		dpp_pfs_free(sta->dpp_pfs);
3168 		sta->dpp_pfs = NULL;
3169 
3170 		if ((hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_DPP) &&
3171 		    hapd->conf->dpp_netaccesskey && sta->wpa_sm &&
3172 		    wpa_auth_sta_key_mgmt(sta->wpa_sm) == WPA_KEY_MGMT_DPP &&
3173 		    elems.owe_dh) {
3174 			sta->dpp_pfs = dpp_pfs_init(
3175 				wpabuf_head(hapd->conf->dpp_netaccesskey),
3176 				wpabuf_len(hapd->conf->dpp_netaccesskey));
3177 			if (!sta->dpp_pfs) {
3178 				wpa_printf(MSG_DEBUG,
3179 					   "DPP: Could not initialize PFS");
3180 				/* Try to continue without PFS */
3181 				goto pfs_fail;
3182 			}
3183 
3184 			if (dpp_pfs_process(sta->dpp_pfs, elems.owe_dh,
3185 					    elems.owe_dh_len) < 0) {
3186 				dpp_pfs_free(sta->dpp_pfs);
3187 				sta->dpp_pfs = NULL;
3188 				return WLAN_STATUS_UNSPECIFIED_FAILURE;
3189 			}
3190 		}
3191 
3192 		wpa_auth_set_dpp_z(sta->wpa_sm, sta->dpp_pfs ?
3193 				   sta->dpp_pfs->secret : NULL);
3194 	pfs_fail:
3195 #endif /* CONFIG_DPP2 */
3196 
3197 #ifdef CONFIG_IEEE80211N
3198 		if ((sta->flags & (WLAN_STA_HT | WLAN_STA_VHT)) &&
3199 		    wpa_auth_get_pairwise(sta->wpa_sm) == WPA_CIPHER_TKIP) {
3200 			hostapd_logger(hapd, sta->addr,
3201 				       HOSTAPD_MODULE_IEEE80211,
3202 				       HOSTAPD_LEVEL_INFO,
3203 				       "Station tried to use TKIP with HT "
3204 				       "association");
3205 			return WLAN_STATUS_CIPHER_REJECTED_PER_POLICY;
3206 		}
3207 #endif /* CONFIG_IEEE80211N */
3208 #ifdef CONFIG_HS20
3209 	} else if (hapd->conf->osen) {
3210 		if (elems.osen == NULL) {
3211 			hostapd_logger(
3212 				hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
3213 				HOSTAPD_LEVEL_INFO,
3214 				"No HS 2.0 OSEN element in association request");
3215 			return WLAN_STATUS_INVALID_IE;
3216 		}
3217 
3218 		wpa_printf(MSG_DEBUG, "HS 2.0: OSEN association");
3219 		if (sta->wpa_sm == NULL)
3220 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
3221 							sta->addr, NULL);
3222 		if (sta->wpa_sm == NULL) {
3223 			wpa_printf(MSG_WARNING, "Failed to initialize WPA "
3224 				   "state machine");
3225 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3226 		}
3227 		if (wpa_validate_osen(hapd->wpa_auth, sta->wpa_sm,
3228 				      elems.osen - 2, elems.osen_len + 2) < 0)
3229 			return WLAN_STATUS_INVALID_IE;
3230 #endif /* CONFIG_HS20 */
3231 	} else
3232 		wpa_auth_sta_no_wpa(sta->wpa_sm);
3233 
3234 #ifdef CONFIG_P2P
3235 	p2p_group_notif_assoc(hapd->p2p_group, sta->addr, ies, ies_len);
3236 #endif /* CONFIG_P2P */
3237 
3238 #ifdef CONFIG_HS20
3239 	wpabuf_free(sta->hs20_ie);
3240 	if (elems.hs20 && elems.hs20_len > 4) {
3241 		int release;
3242 
3243 		sta->hs20_ie = wpabuf_alloc_copy(elems.hs20 + 4,
3244 						 elems.hs20_len - 4);
3245 		release = ((elems.hs20[4] >> 4) & 0x0f) + 1;
3246 		if (release >= 2 && !wpa_auth_uses_mfp(sta->wpa_sm)) {
3247 			wpa_printf(MSG_DEBUG,
3248 				   "HS 2.0: PMF not negotiated by release %d station "
3249 				   MACSTR, release, MAC2STR(sta->addr));
3250 			return WLAN_STATUS_ROBUST_MGMT_FRAME_POLICY_VIOLATION;
3251 		}
3252 	} else {
3253 		sta->hs20_ie = NULL;
3254 	}
3255 
3256 	wpabuf_free(sta->roaming_consortium);
3257 	if (elems.roaming_cons_sel)
3258 		sta->roaming_consortium = wpabuf_alloc_copy(
3259 			elems.roaming_cons_sel + 4,
3260 			elems.roaming_cons_sel_len - 4);
3261 	else
3262 		sta->roaming_consortium = NULL;
3263 #endif /* CONFIG_HS20 */
3264 
3265 #ifdef CONFIG_FST
3266 	wpabuf_free(sta->mb_ies);
3267 	if (hapd->iface->fst)
3268 		sta->mb_ies = mb_ies_by_info(&elems.mb_ies);
3269 	else
3270 		sta->mb_ies = NULL;
3271 #endif /* CONFIG_FST */
3272 
3273 #ifdef CONFIG_MBO
3274 	mbo_ap_check_sta_assoc(hapd, sta, &elems);
3275 
3276 	if (hapd->conf->mbo_enabled && (hapd->conf->wpa & 2) &&
3277 	    elems.mbo && sta->cell_capa && !(sta->flags & WLAN_STA_MFP) &&
3278 	    hapd->conf->ieee80211w != NO_MGMT_FRAME_PROTECTION) {
3279 		wpa_printf(MSG_INFO,
3280 			   "MBO: Reject WPA2 association without PMF");
3281 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3282 	}
3283 #endif /* CONFIG_MBO */
3284 
3285 #if defined(CONFIG_FILS) && defined(CONFIG_OCV)
3286 	if (wpa_auth_uses_ocv(sta->wpa_sm) &&
3287 	    (sta->auth_alg == WLAN_AUTH_FILS_SK ||
3288 	     sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
3289 	     sta->auth_alg == WLAN_AUTH_FILS_PK)) {
3290 		struct wpa_channel_info ci;
3291 		int tx_chanwidth;
3292 		int tx_seg1_idx;
3293 
3294 		if (hostapd_drv_channel_info(hapd, &ci) != 0) {
3295 			wpa_printf(MSG_WARNING,
3296 				   "Failed to get channel info to validate received OCI in FILS (Re)Association Request frame");
3297 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3298 		}
3299 
3300 		if (get_sta_tx_parameters(sta->wpa_sm,
3301 					  channel_width_to_int(ci.chanwidth),
3302 					  ci.seg1_idx, &tx_chanwidth,
3303 					  &tx_seg1_idx) < 0)
3304 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3305 
3306 		if (ocv_verify_tx_params(elems.oci, elems.oci_len, &ci,
3307 					 tx_chanwidth, tx_seg1_idx) != 0) {
3308 			wpa_printf(MSG_WARNING, "FILS: %s", ocv_errorstr);
3309 			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3310 		}
3311 	}
3312 #endif /* CONFIG_FILS && CONFIG_OCV */
3313 
3314 	ap_copy_sta_supp_op_classes(sta, elems.supp_op_classes,
3315 				    elems.supp_op_classes_len);
3316 
3317 	if ((sta->capability & WLAN_CAPABILITY_RADIO_MEASUREMENT) &&
3318 	    elems.rrm_enabled &&
3319 	    elems.rrm_enabled_len >= sizeof(sta->rrm_enabled_capa))
3320 		os_memcpy(sta->rrm_enabled_capa, elems.rrm_enabled,
3321 			  sizeof(sta->rrm_enabled_capa));
3322 
3323 	if (elems.power_capab) {
3324 		sta->min_tx_power = elems.power_capab[0];
3325 		sta->max_tx_power = elems.power_capab[1];
3326 		sta->power_capab = 1;
3327 	} else {
3328 		sta->power_capab = 0;
3329 	}
3330 
3331 	return WLAN_STATUS_SUCCESS;
3332 }
3333 
3334 
3335 static void send_deauth(struct hostapd_data *hapd, const u8 *addr,
3336 			u16 reason_code)
3337 {
3338 	int send_len;
3339 	struct ieee80211_mgmt reply;
3340 
3341 	os_memset(&reply, 0, sizeof(reply));
3342 	reply.frame_control =
3343 		IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_DEAUTH);
3344 	os_memcpy(reply.da, addr, ETH_ALEN);
3345 	os_memcpy(reply.sa, hapd->own_addr, ETH_ALEN);
3346 	os_memcpy(reply.bssid, hapd->own_addr, ETH_ALEN);
3347 
3348 	send_len = IEEE80211_HDRLEN + sizeof(reply.u.deauth);
3349 	reply.u.deauth.reason_code = host_to_le16(reason_code);
3350 
3351 	if (hostapd_drv_send_mlme(hapd, &reply, send_len, 0) < 0)
3352 		wpa_printf(MSG_INFO, "Failed to send deauth: %s",
3353 			   strerror(errno));
3354 }
3355 
3356 
3357 static int add_associated_sta(struct hostapd_data *hapd,
3358 			      struct sta_info *sta, int reassoc)
3359 {
3360 	struct ieee80211_ht_capabilities ht_cap;
3361 	struct ieee80211_vht_capabilities vht_cap;
3362 	struct ieee80211_he_capabilities he_cap;
3363 	int set = 1;
3364 
3365 	/*
3366 	 * Remove the STA entry to ensure the STA PS state gets cleared and
3367 	 * configuration gets updated. This is relevant for cases, such as
3368 	 * FT-over-the-DS, where a station re-associates back to the same AP but
3369 	 * skips the authentication flow, or if working with a driver that
3370 	 * does not support full AP client state.
3371 	 *
3372 	 * Skip this if the STA has already completed FT reassociation and the
3373 	 * TK has been configured since the TX/RX PN must not be reset to 0 for
3374 	 * the same key.
3375 	 *
3376 	 * FT-over-the-DS has a special case where the STA entry (and as such,
3377 	 * the TK) has not yet been configured to the driver depending on which
3378 	 * driver interface is used. For that case, allow add-STA operation to
3379 	 * be used (instead of set-STA). This is needed to allow mac80211-based
3380 	 * drivers to accept the STA parameter configuration. Since this is
3381 	 * after a new FT-over-DS exchange, a new TK has been derived, so key
3382 	 * reinstallation is not a concern for this case.
3383 	 */
3384 	wpa_printf(MSG_DEBUG, "Add associated STA " MACSTR
3385 		   " (added_unassoc=%d auth_alg=%u ft_over_ds=%u reassoc=%d authorized=%d ft_tk=%d fils_tk=%d)",
3386 		   MAC2STR(sta->addr), sta->added_unassoc, sta->auth_alg,
3387 		   sta->ft_over_ds, reassoc,
3388 		   !!(sta->flags & WLAN_STA_AUTHORIZED),
3389 		   wpa_auth_sta_ft_tk_already_set(sta->wpa_sm),
3390 		   wpa_auth_sta_fils_tk_already_set(sta->wpa_sm));
3391 
3392 	if (!sta->added_unassoc &&
3393 	    (!(sta->flags & WLAN_STA_AUTHORIZED) ||
3394 	     (reassoc && sta->ft_over_ds && sta->auth_alg == WLAN_AUTH_FT) ||
3395 	     (!wpa_auth_sta_ft_tk_already_set(sta->wpa_sm) &&
3396 	      !wpa_auth_sta_fils_tk_already_set(sta->wpa_sm)))) {
3397 		hostapd_drv_sta_remove(hapd, sta->addr);
3398 		wpa_auth_sm_event(sta->wpa_sm, WPA_DRV_STA_REMOVED);
3399 		set = 0;
3400 
3401 		 /* Do not allow the FT-over-DS exception to be used more than
3402 		  * once per authentication exchange to guarantee a new TK is
3403 		  * used here */
3404 		sta->ft_over_ds = 0;
3405 	}
3406 
3407 #ifdef CONFIG_IEEE80211N
3408 	if (sta->flags & WLAN_STA_HT)
3409 		hostapd_get_ht_capab(hapd, sta->ht_capabilities, &ht_cap);
3410 #endif /* CONFIG_IEEE80211N */
3411 #ifdef CONFIG_IEEE80211AC
3412 	if (sta->flags & WLAN_STA_VHT)
3413 		hostapd_get_vht_capab(hapd, sta->vht_capabilities, &vht_cap);
3414 #endif /* CONFIG_IEEE80211AC */
3415 #ifdef CONFIG_IEEE80211AX
3416 	if (sta->flags & WLAN_STA_HE) {
3417 		hostapd_get_he_capab(hapd, sta->he_capab, &he_cap,
3418 				     sta->he_capab_len);
3419 	}
3420 #endif /* CONFIG_IEEE80211AX */
3421 
3422 	/*
3423 	 * Add the station with forced WLAN_STA_ASSOC flag. The sta->flags
3424 	 * will be set when the ACK frame for the (Re)Association Response frame
3425 	 * is processed (TX status driver event).
3426 	 */
3427 	if (hostapd_sta_add(hapd, sta->addr, sta->aid, sta->capability,
3428 			    sta->supported_rates, sta->supported_rates_len,
3429 			    sta->listen_interval,
3430 			    sta->flags & WLAN_STA_HT ? &ht_cap : NULL,
3431 			    sta->flags & WLAN_STA_VHT ? &vht_cap : NULL,
3432 			    sta->flags & WLAN_STA_HE ? &he_cap : NULL,
3433 			    sta->flags & WLAN_STA_HE ? sta->he_capab_len : 0,
3434 			    sta->flags | WLAN_STA_ASSOC, sta->qosinfo,
3435 			    sta->vht_opmode, sta->p2p_ie ? 1 : 0,
3436 			    set)) {
3437 		hostapd_logger(hapd, sta->addr,
3438 			       HOSTAPD_MODULE_IEEE80211, HOSTAPD_LEVEL_NOTICE,
3439 			       "Could not %s STA to kernel driver",
3440 			       set ? "set" : "add");
3441 
3442 		if (sta->added_unassoc) {
3443 			hostapd_drv_sta_remove(hapd, sta->addr);
3444 			sta->added_unassoc = 0;
3445 		}
3446 
3447 		return -1;
3448 	}
3449 
3450 	sta->added_unassoc = 0;
3451 
3452 	return 0;
3453 }
3454 
3455 
3456 static u16 send_assoc_resp(struct hostapd_data *hapd, struct sta_info *sta,
3457 			   const u8 *addr, u16 status_code, int reassoc,
3458 			   const u8 *ies, size_t ies_len, int rssi)
3459 {
3460 	int send_len;
3461 	u8 *buf;
3462 	size_t buflen;
3463 	struct ieee80211_mgmt *reply;
3464 	u8 *p;
3465 	u16 res = WLAN_STATUS_SUCCESS;
3466 
3467 	buflen = sizeof(struct ieee80211_mgmt) + 1024;
3468 #ifdef CONFIG_FILS
3469 	if (sta && sta->fils_hlp_resp)
3470 		buflen += wpabuf_len(sta->fils_hlp_resp);
3471 	if (sta)
3472 		buflen += 150;
3473 #endif /* CONFIG_FILS */
3474 #ifdef CONFIG_OWE
3475 	if (sta && (hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_OWE))
3476 		buflen += 150;
3477 #endif /* CONFIG_OWE */
3478 #ifdef CONFIG_DPP2
3479 	if (sta && sta->dpp_pfs)
3480 		buflen += 5 + sta->dpp_pfs->curve->prime_len;
3481 #endif /* CONFIG_DPP2 */
3482 	buf = os_zalloc(buflen);
3483 	if (!buf) {
3484 		res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3485 		goto done;
3486 	}
3487 	reply = (struct ieee80211_mgmt *) buf;
3488 	reply->frame_control =
3489 		IEEE80211_FC(WLAN_FC_TYPE_MGMT,
3490 			     (reassoc ? WLAN_FC_STYPE_REASSOC_RESP :
3491 			      WLAN_FC_STYPE_ASSOC_RESP));
3492 	os_memcpy(reply->da, addr, ETH_ALEN);
3493 	os_memcpy(reply->sa, hapd->own_addr, ETH_ALEN);
3494 	os_memcpy(reply->bssid, hapd->own_addr, ETH_ALEN);
3495 
3496 	send_len = IEEE80211_HDRLEN;
3497 	send_len += sizeof(reply->u.assoc_resp);
3498 	reply->u.assoc_resp.capab_info =
3499 		host_to_le16(hostapd_own_capab_info(hapd));
3500 	reply->u.assoc_resp.status_code = host_to_le16(status_code);
3501 
3502 	reply->u.assoc_resp.aid = host_to_le16((sta ? sta->aid : 0) |
3503 					       BIT(14) | BIT(15));
3504 	/* Supported rates */
3505 	p = hostapd_eid_supp_rates(hapd, reply->u.assoc_resp.variable);
3506 	/* Extended supported rates */
3507 	p = hostapd_eid_ext_supp_rates(hapd, p);
3508 
3509 #ifdef CONFIG_MBO
3510 	if (status_code == WLAN_STATUS_DENIED_POOR_CHANNEL_CONDITIONS &&
3511 	    rssi != 0) {
3512 		int delta = hapd->iconf->rssi_reject_assoc_rssi - rssi;
3513 
3514 		p = hostapd_eid_mbo_rssi_assoc_rej(hapd, p, buf + buflen - p,
3515 						   delta);
3516 	}
3517 #endif /* CONFIG_MBO */
3518 
3519 #ifdef CONFIG_IEEE80211R_AP
3520 	if (sta && status_code == WLAN_STATUS_SUCCESS) {
3521 		/* IEEE 802.11r: Mobility Domain Information, Fast BSS
3522 		 * Transition Information, RSN, [RIC Response] */
3523 		p = wpa_sm_write_assoc_resp_ies(sta->wpa_sm, p,
3524 						buf + buflen - p,
3525 						sta->auth_alg, ies, ies_len);
3526 		if (!p) {
3527 			wpa_printf(MSG_DEBUG,
3528 				   "FT: Failed to write AssocResp IEs");
3529 			res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3530 			goto done;
3531 		}
3532 	}
3533 #endif /* CONFIG_IEEE80211R_AP */
3534 #ifdef CONFIG_FILS
3535 	if (sta && status_code == WLAN_STATUS_SUCCESS &&
3536 	    (sta->auth_alg == WLAN_AUTH_FILS_SK ||
3537 	     sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
3538 	     sta->auth_alg == WLAN_AUTH_FILS_PK))
3539 		p = wpa_auth_write_assoc_resp_fils(sta->wpa_sm, p,
3540 						   buf + buflen - p,
3541 						   ies, ies_len);
3542 #endif /* CONFIG_FILS */
3543 
3544 #ifdef CONFIG_OWE
3545 	if (sta && status_code == WLAN_STATUS_SUCCESS &&
3546 	    (hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_OWE))
3547 		p = wpa_auth_write_assoc_resp_owe(sta->wpa_sm, p,
3548 						  buf + buflen - p,
3549 						  ies, ies_len);
3550 #endif /* CONFIG_OWE */
3551 
3552 #ifdef CONFIG_IEEE80211W
3553 	if (sta && status_code == WLAN_STATUS_ASSOC_REJECTED_TEMPORARILY)
3554 		p = hostapd_eid_assoc_comeback_time(hapd, sta, p);
3555 #endif /* CONFIG_IEEE80211W */
3556 
3557 #ifdef CONFIG_IEEE80211N
3558 	p = hostapd_eid_ht_capabilities(hapd, p);
3559 	p = hostapd_eid_ht_operation(hapd, p);
3560 #endif /* CONFIG_IEEE80211N */
3561 
3562 #ifdef CONFIG_IEEE80211AC
3563 	if (hapd->iconf->ieee80211ac && !hapd->conf->disable_11ac) {
3564 		u32 nsts = 0, sta_nsts;
3565 
3566 		if (sta && hapd->conf->use_sta_nsts && sta->vht_capabilities) {
3567 			struct ieee80211_vht_capabilities *capa;
3568 
3569 			nsts = (hapd->iface->conf->vht_capab >>
3570 				VHT_CAP_BEAMFORMEE_STS_OFFSET) & 7;
3571 			capa = sta->vht_capabilities;
3572 			sta_nsts = (le_to_host32(capa->vht_capabilities_info) >>
3573 				    VHT_CAP_BEAMFORMEE_STS_OFFSET) & 7;
3574 
3575 			if (nsts < sta_nsts)
3576 				nsts = 0;
3577 			else
3578 				nsts = sta_nsts;
3579 		}
3580 		p = hostapd_eid_vht_capabilities(hapd, p, nsts);
3581 		p = hostapd_eid_vht_operation(hapd, p);
3582 	}
3583 #endif /* CONFIG_IEEE80211AC */
3584 
3585 #ifdef CONFIG_IEEE80211AX
3586 	if (hapd->iconf->ieee80211ax) {
3587 		p = hostapd_eid_he_capab(hapd, p, IEEE80211_MODE_AP);
3588 		p = hostapd_eid_he_operation(hapd, p);
3589 		p = hostapd_eid_spatial_reuse(hapd, p);
3590 		p = hostapd_eid_he_mu_edca_parameter_set(hapd, p);
3591 	}
3592 #endif /* CONFIG_IEEE80211AX */
3593 
3594 	p = hostapd_eid_ext_capab(hapd, p);
3595 	p = hostapd_eid_bss_max_idle_period(hapd, p);
3596 	if (sta && sta->qos_map_enabled)
3597 		p = hostapd_eid_qos_map_set(hapd, p);
3598 
3599 #ifdef CONFIG_FST
3600 	if (hapd->iface->fst_ies) {
3601 		os_memcpy(p, wpabuf_head(hapd->iface->fst_ies),
3602 			  wpabuf_len(hapd->iface->fst_ies));
3603 		p += wpabuf_len(hapd->iface->fst_ies);
3604 	}
3605 #endif /* CONFIG_FST */
3606 
3607 #ifdef CONFIG_OWE
3608 	if ((hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_OWE) &&
3609 	    sta && sta->owe_ecdh && status_code == WLAN_STATUS_SUCCESS &&
3610 	    wpa_auth_sta_key_mgmt(sta->wpa_sm) == WPA_KEY_MGMT_OWE) {
3611 		struct wpabuf *pub;
3612 
3613 		pub = crypto_ecdh_get_pubkey(sta->owe_ecdh, 0);
3614 		if (!pub) {
3615 			res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3616 			goto done;
3617 		}
3618 		/* OWE Diffie-Hellman Parameter element */
3619 		*p++ = WLAN_EID_EXTENSION; /* Element ID */
3620 		*p++ = 1 + 2 + wpabuf_len(pub); /* Length */
3621 		*p++ = WLAN_EID_EXT_OWE_DH_PARAM; /* Element ID Extension */
3622 		WPA_PUT_LE16(p, sta->owe_group);
3623 		p += 2;
3624 		os_memcpy(p, wpabuf_head(pub), wpabuf_len(pub));
3625 		p += wpabuf_len(pub);
3626 		wpabuf_free(pub);
3627 	}
3628 #endif /* CONFIG_OWE */
3629 
3630 #ifdef CONFIG_DPP2
3631 	if ((hapd->conf->wpa_key_mgmt & WPA_KEY_MGMT_DPP) &&
3632 	    sta && sta->dpp_pfs && status_code == WLAN_STATUS_SUCCESS &&
3633 	    wpa_auth_sta_key_mgmt(sta->wpa_sm) == WPA_KEY_MGMT_DPP) {
3634 		os_memcpy(p, wpabuf_head(sta->dpp_pfs->ie),
3635 			  wpabuf_len(sta->dpp_pfs->ie));
3636 		p += wpabuf_len(sta->dpp_pfs->ie);
3637 	}
3638 #endif /* CONFIG_DPP2 */
3639 
3640 #ifdef CONFIG_IEEE80211AC
3641 	if (sta && hapd->conf->vendor_vht && (sta->flags & WLAN_STA_VENDOR_VHT))
3642 		p = hostapd_eid_vendor_vht(hapd, p);
3643 #endif /* CONFIG_IEEE80211AC */
3644 
3645 	if (sta && (sta->flags & WLAN_STA_WMM))
3646 		p = hostapd_eid_wmm(hapd, p);
3647 
3648 #ifdef CONFIG_WPS
3649 	if (sta &&
3650 	    ((sta->flags & WLAN_STA_WPS) ||
3651 	     ((sta->flags & WLAN_STA_MAYBE_WPS) && hapd->conf->wpa))) {
3652 		struct wpabuf *wps = wps_build_assoc_resp_ie();
3653 		if (wps) {
3654 			os_memcpy(p, wpabuf_head(wps), wpabuf_len(wps));
3655 			p += wpabuf_len(wps);
3656 			wpabuf_free(wps);
3657 		}
3658 	}
3659 #endif /* CONFIG_WPS */
3660 
3661 	if (sta && (sta->flags & WLAN_STA_MULTI_AP))
3662 		p = hostapd_eid_multi_ap(hapd, p);
3663 
3664 #ifdef CONFIG_P2P
3665 	if (sta && sta->p2p_ie && hapd->p2p_group) {
3666 		struct wpabuf *p2p_resp_ie;
3667 		enum p2p_status_code status;
3668 		switch (status_code) {
3669 		case WLAN_STATUS_SUCCESS:
3670 			status = P2P_SC_SUCCESS;
3671 			break;
3672 		case WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA:
3673 			status = P2P_SC_FAIL_LIMIT_REACHED;
3674 			break;
3675 		default:
3676 			status = P2P_SC_FAIL_INVALID_PARAMS;
3677 			break;
3678 		}
3679 		p2p_resp_ie = p2p_group_assoc_resp_ie(hapd->p2p_group, status);
3680 		if (p2p_resp_ie) {
3681 			os_memcpy(p, wpabuf_head(p2p_resp_ie),
3682 				  wpabuf_len(p2p_resp_ie));
3683 			p += wpabuf_len(p2p_resp_ie);
3684 			wpabuf_free(p2p_resp_ie);
3685 		}
3686 	}
3687 #endif /* CONFIG_P2P */
3688 
3689 #ifdef CONFIG_P2P_MANAGER
3690 	if (hapd->conf->p2p & P2P_MANAGE)
3691 		p = hostapd_eid_p2p_manage(hapd, p);
3692 #endif /* CONFIG_P2P_MANAGER */
3693 
3694 	p = hostapd_eid_mbo(hapd, p, buf + buflen - p);
3695 
3696 	if (hapd->conf->assocresp_elements &&
3697 	    (size_t) (buf + buflen - p) >=
3698 	    wpabuf_len(hapd->conf->assocresp_elements)) {
3699 		os_memcpy(p, wpabuf_head(hapd->conf->assocresp_elements),
3700 			  wpabuf_len(hapd->conf->assocresp_elements));
3701 		p += wpabuf_len(hapd->conf->assocresp_elements);
3702 	}
3703 
3704 	send_len += p - reply->u.assoc_resp.variable;
3705 
3706 #ifdef CONFIG_FILS
3707 	if (sta &&
3708 	    (sta->auth_alg == WLAN_AUTH_FILS_SK ||
3709 	     sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
3710 	     sta->auth_alg == WLAN_AUTH_FILS_PK) &&
3711 	    status_code == WLAN_STATUS_SUCCESS) {
3712 		struct ieee802_11_elems elems;
3713 
3714 		if (ieee802_11_parse_elems(ies, ies_len, &elems, 0) ==
3715 		    ParseFailed || !elems.fils_session) {
3716 			res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3717 			goto done;
3718 		}
3719 
3720 		/* FILS Session */
3721 		*p++ = WLAN_EID_EXTENSION; /* Element ID */
3722 		*p++ = 1 + FILS_SESSION_LEN; /* Length */
3723 		*p++ = WLAN_EID_EXT_FILS_SESSION; /* Element ID Extension */
3724 		os_memcpy(p, elems.fils_session, FILS_SESSION_LEN);
3725 		send_len += 2 + 1 + FILS_SESSION_LEN;
3726 
3727 		send_len = fils_encrypt_assoc(sta->wpa_sm, buf, send_len,
3728 					      buflen, sta->fils_hlp_resp);
3729 		if (send_len < 0) {
3730 			res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3731 			goto done;
3732 		}
3733 	}
3734 #endif /* CONFIG_FILS */
3735 
3736 	if (hostapd_drv_send_mlme(hapd, reply, send_len, 0) < 0) {
3737 		wpa_printf(MSG_INFO, "Failed to send assoc resp: %s",
3738 			   strerror(errno));
3739 		res = WLAN_STATUS_UNSPECIFIED_FAILURE;
3740 	}
3741 
3742 done:
3743 	os_free(buf);
3744 	return res;
3745 }
3746 
3747 
3748 #ifdef CONFIG_OWE
3749 u8 * owe_assoc_req_process(struct hostapd_data *hapd, struct sta_info *sta,
3750 			   const u8 *owe_dh, u8 owe_dh_len,
3751 			   u8 *owe_buf, size_t owe_buf_len, u16 *reason)
3752 {
3753 #ifdef CONFIG_TESTING_OPTIONS
3754 	if (hapd->conf->own_ie_override) {
3755 		wpa_printf(MSG_DEBUG, "OWE: Using IE override");
3756 		*reason = WLAN_STATUS_SUCCESS;
3757 		return wpa_auth_write_assoc_resp_owe(sta->wpa_sm, owe_buf,
3758 						     owe_buf_len, NULL, 0);
3759 	}
3760 #endif /* CONFIG_TESTING_OPTIONS */
3761 
3762 	if (wpa_auth_sta_get_pmksa(sta->wpa_sm)) {
3763 		wpa_printf(MSG_DEBUG, "OWE: Using PMKSA caching");
3764 		owe_buf = wpa_auth_write_assoc_resp_owe(sta->wpa_sm, owe_buf,
3765 							owe_buf_len, NULL, 0);
3766 		*reason = WLAN_STATUS_SUCCESS;
3767 		return owe_buf;
3768 	}
3769 
3770 	if (sta->owe_pmk && sta->external_dh_updated) {
3771 		wpa_printf(MSG_DEBUG, "OWE: Using previously derived PMK");
3772 		*reason = WLAN_STATUS_SUCCESS;
3773 		return owe_buf;
3774 	}
3775 
3776 	*reason = owe_process_assoc_req(hapd, sta, owe_dh, owe_dh_len);
3777 	if (*reason != WLAN_STATUS_SUCCESS)
3778 		return NULL;
3779 
3780 	owe_buf = wpa_auth_write_assoc_resp_owe(sta->wpa_sm, owe_buf,
3781 						owe_buf_len, NULL, 0);
3782 
3783 	if (sta->owe_ecdh && owe_buf) {
3784 		struct wpabuf *pub;
3785 
3786 		pub = crypto_ecdh_get_pubkey(sta->owe_ecdh, 0);
3787 		if (!pub) {
3788 			*reason = WLAN_STATUS_UNSPECIFIED_FAILURE;
3789 			return owe_buf;
3790 		}
3791 
3792 		/* OWE Diffie-Hellman Parameter element */
3793 		*owe_buf++ = WLAN_EID_EXTENSION; /* Element ID */
3794 		*owe_buf++ = 1 + 2 + wpabuf_len(pub); /* Length */
3795 		*owe_buf++ = WLAN_EID_EXT_OWE_DH_PARAM; /* Element ID Extension
3796 							 */
3797 		WPA_PUT_LE16(owe_buf, sta->owe_group);
3798 		owe_buf += 2;
3799 		os_memcpy(owe_buf, wpabuf_head(pub), wpabuf_len(pub));
3800 		owe_buf += wpabuf_len(pub);
3801 		wpabuf_free(pub);
3802 	}
3803 
3804 	return owe_buf;
3805 }
3806 #endif /* CONFIG_OWE */
3807 
3808 
3809 #ifdef CONFIG_FILS
3810 
3811 void fils_hlp_finish_assoc(struct hostapd_data *hapd, struct sta_info *sta)
3812 {
3813 	u16 reply_res;
3814 
3815 	wpa_printf(MSG_DEBUG, "FILS: Finish association with " MACSTR,
3816 		   MAC2STR(sta->addr));
3817 	eloop_cancel_timeout(fils_hlp_timeout, hapd, sta);
3818 	if (!sta->fils_pending_assoc_req)
3819 		return;
3820 	reply_res = send_assoc_resp(hapd, sta, sta->addr, WLAN_STATUS_SUCCESS,
3821 				    sta->fils_pending_assoc_is_reassoc,
3822 				    sta->fils_pending_assoc_req,
3823 				    sta->fils_pending_assoc_req_len, 0);
3824 	os_free(sta->fils_pending_assoc_req);
3825 	sta->fils_pending_assoc_req = NULL;
3826 	sta->fils_pending_assoc_req_len = 0;
3827 	wpabuf_free(sta->fils_hlp_resp);
3828 	sta->fils_hlp_resp = NULL;
3829 	wpabuf_free(sta->hlp_dhcp_discover);
3830 	sta->hlp_dhcp_discover = NULL;
3831 
3832 	/*
3833 	 * Remove the station in case transmission of a success response fails.
3834 	 * At this point the station was already added associated to the driver.
3835 	 */
3836 	if (reply_res != WLAN_STATUS_SUCCESS)
3837 		hostapd_drv_sta_remove(hapd, sta->addr);
3838 }
3839 
3840 
3841 void fils_hlp_timeout(void *eloop_ctx, void *eloop_data)
3842 {
3843 	struct hostapd_data *hapd = eloop_ctx;
3844 	struct sta_info *sta = eloop_data;
3845 
3846 	wpa_printf(MSG_DEBUG,
3847 		   "FILS: HLP response timeout - continue with association response for "
3848 		   MACSTR, MAC2STR(sta->addr));
3849 	if (sta->fils_drv_assoc_finish)
3850 		hostapd_notify_assoc_fils_finish(hapd, sta);
3851 	else
3852 		fils_hlp_finish_assoc(hapd, sta);
3853 }
3854 
3855 #endif /* CONFIG_FILS */
3856 
3857 
3858 static void handle_assoc(struct hostapd_data *hapd,
3859 			 const struct ieee80211_mgmt *mgmt, size_t len,
3860 			 int reassoc, int rssi)
3861 {
3862 	u16 capab_info, listen_interval, seq_ctrl, fc;
3863 	u16 resp = WLAN_STATUS_SUCCESS, reply_res;
3864 	const u8 *pos;
3865 	int left, i;
3866 	struct sta_info *sta;
3867 	u8 *tmp = NULL;
3868 	struct hostapd_sta_wpa_psk_short *psk = NULL;
3869 	char *identity = NULL;
3870 	char *radius_cui = NULL;
3871 #ifdef CONFIG_FILS
3872 	int delay_assoc = 0;
3873 #endif /* CONFIG_FILS */
3874 
3875 	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_req) :
3876 				      sizeof(mgmt->u.assoc_req))) {
3877 		wpa_printf(MSG_INFO, "handle_assoc(reassoc=%d) - too short payload (len=%lu)",
3878 			   reassoc, (unsigned long) len);
3879 		return;
3880 	}
3881 
3882 #ifdef CONFIG_TESTING_OPTIONS
3883 	if (reassoc) {
3884 		if (hapd->iconf->ignore_reassoc_probability > 0.0 &&
3885 		    drand48() < hapd->iconf->ignore_reassoc_probability) {
3886 			wpa_printf(MSG_INFO,
3887 				   "TESTING: ignoring reassoc request from "
3888 				   MACSTR, MAC2STR(mgmt->sa));
3889 			return;
3890 		}
3891 	} else {
3892 		if (hapd->iconf->ignore_assoc_probability > 0.0 &&
3893 		    drand48() < hapd->iconf->ignore_assoc_probability) {
3894 			wpa_printf(MSG_INFO,
3895 				   "TESTING: ignoring assoc request from "
3896 				   MACSTR, MAC2STR(mgmt->sa));
3897 			return;
3898 		}
3899 	}
3900 #endif /* CONFIG_TESTING_OPTIONS */
3901 
3902 	fc = le_to_host16(mgmt->frame_control);
3903 	seq_ctrl = le_to_host16(mgmt->seq_ctrl);
3904 
3905 	if (reassoc) {
3906 		capab_info = le_to_host16(mgmt->u.reassoc_req.capab_info);
3907 		listen_interval = le_to_host16(
3908 			mgmt->u.reassoc_req.listen_interval);
3909 		wpa_printf(MSG_DEBUG, "reassociation request: STA=" MACSTR
3910 			   " capab_info=0x%02x listen_interval=%d current_ap="
3911 			   MACSTR " seq_ctrl=0x%x%s",
3912 			   MAC2STR(mgmt->sa), capab_info, listen_interval,
3913 			   MAC2STR(mgmt->u.reassoc_req.current_ap),
3914 			   seq_ctrl, (fc & WLAN_FC_RETRY) ? " retry" : "");
3915 		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.reassoc_req));
3916 		pos = mgmt->u.reassoc_req.variable;
3917 	} else {
3918 		capab_info = le_to_host16(mgmt->u.assoc_req.capab_info);
3919 		listen_interval = le_to_host16(
3920 			mgmt->u.assoc_req.listen_interval);
3921 		wpa_printf(MSG_DEBUG, "association request: STA=" MACSTR
3922 			   " capab_info=0x%02x listen_interval=%d "
3923 			   "seq_ctrl=0x%x%s",
3924 			   MAC2STR(mgmt->sa), capab_info, listen_interval,
3925 			   seq_ctrl, (fc & WLAN_FC_RETRY) ? " retry" : "");
3926 		left = len - (IEEE80211_HDRLEN + sizeof(mgmt->u.assoc_req));
3927 		pos = mgmt->u.assoc_req.variable;
3928 	}
3929 
3930 	sta = ap_get_sta(hapd, mgmt->sa);
3931 #ifdef CONFIG_IEEE80211R_AP
3932 	if (sta && sta->auth_alg == WLAN_AUTH_FT &&
3933 	    (sta->flags & WLAN_STA_AUTH) == 0) {
3934 		wpa_printf(MSG_DEBUG, "FT: Allow STA " MACSTR " to associate "
3935 			   "prior to authentication since it is using "
3936 			   "over-the-DS FT", MAC2STR(mgmt->sa));
3937 
3938 		/*
3939 		 * Mark station as authenticated, to avoid adding station
3940 		 * entry in the driver as associated and not authenticated
3941 		 */
3942 		sta->flags |= WLAN_STA_AUTH;
3943 	} else
3944 #endif /* CONFIG_IEEE80211R_AP */
3945 	if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
3946 		if (hapd->iface->current_mode &&
3947 		    hapd->iface->current_mode->mode ==
3948 			HOSTAPD_MODE_IEEE80211AD) {
3949 			int acl_res;
3950 			u32 session_timeout, acct_interim_interval;
3951 			struct vlan_description vlan_id;
3952 
3953 			acl_res = ieee802_11_allowed_address(
3954 				hapd, mgmt->sa, (const u8 *) mgmt, len,
3955 				&session_timeout, &acct_interim_interval,
3956 				&vlan_id, &psk, &identity, &radius_cui, 0);
3957 			if (acl_res == HOSTAPD_ACL_REJECT) {
3958 				wpa_msg(hapd->msg_ctx, MSG_DEBUG,
3959 					"Ignore Association Request frame from "
3960 					MACSTR " due to ACL reject",
3961 					MAC2STR(mgmt->sa));
3962 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
3963 				goto fail;
3964 			}
3965 			if (acl_res == HOSTAPD_ACL_PENDING)
3966 				return;
3967 
3968 			/* DMG/IEEE 802.11ad does not use authentication.
3969 			 * Allocate sta entry upon association. */
3970 			sta = ap_sta_add(hapd, mgmt->sa);
3971 			if (!sta) {
3972 				hostapd_logger(hapd, mgmt->sa,
3973 					       HOSTAPD_MODULE_IEEE80211,
3974 					       HOSTAPD_LEVEL_INFO,
3975 					       "Failed to add STA");
3976 				resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
3977 				goto fail;
3978 			}
3979 
3980 			acl_res = ieee802_11_set_radius_info(
3981 				hapd, sta, acl_res, session_timeout,
3982 				acct_interim_interval, &vlan_id, &psk,
3983 				&identity, &radius_cui);
3984 			if (acl_res) {
3985 				resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
3986 				goto fail;
3987 			}
3988 
3989 			hostapd_logger(hapd, sta->addr,
3990 				       HOSTAPD_MODULE_IEEE80211,
3991 				       HOSTAPD_LEVEL_DEBUG,
3992 				       "Skip authentication for DMG/IEEE 802.11ad");
3993 			sta->flags |= WLAN_STA_AUTH;
3994 			wpa_auth_sm_event(sta->wpa_sm, WPA_AUTH);
3995 			sta->auth_alg = WLAN_AUTH_OPEN;
3996 		} else {
3997 			hostapd_logger(hapd, mgmt->sa,
3998 				       HOSTAPD_MODULE_IEEE80211,
3999 				       HOSTAPD_LEVEL_INFO,
4000 				       "Station tried to associate before authentication (aid=%d flags=0x%x)",
4001 				       sta ? sta->aid : -1,
4002 				       sta ? sta->flags : 0);
4003 			send_deauth(hapd, mgmt->sa,
4004 				    WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA);
4005 			return;
4006 		}
4007 	}
4008 
4009 	if ((fc & WLAN_FC_RETRY) &&
4010 	    sta->last_seq_ctrl != WLAN_INVALID_MGMT_SEQ &&
4011 	    sta->last_seq_ctrl == seq_ctrl &&
4012 	    sta->last_subtype == (reassoc ? WLAN_FC_STYPE_REASSOC_REQ :
4013 				  WLAN_FC_STYPE_ASSOC_REQ)) {
4014 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4015 			       HOSTAPD_LEVEL_DEBUG,
4016 			       "Drop repeated association frame seq_ctrl=0x%x",
4017 			       seq_ctrl);
4018 		return;
4019 	}
4020 	sta->last_seq_ctrl = seq_ctrl;
4021 	sta->last_subtype = reassoc ? WLAN_FC_STYPE_REASSOC_REQ :
4022 		WLAN_FC_STYPE_ASSOC_REQ;
4023 
4024 	if (hapd->tkip_countermeasures) {
4025 		resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
4026 		goto fail;
4027 	}
4028 
4029 	if (listen_interval > hapd->conf->max_listen_interval) {
4030 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4031 			       HOSTAPD_LEVEL_DEBUG,
4032 			       "Too large Listen Interval (%d)",
4033 			       listen_interval);
4034 		resp = WLAN_STATUS_ASSOC_DENIED_LISTEN_INT_TOO_LARGE;
4035 		goto fail;
4036 	}
4037 
4038 #ifdef CONFIG_MBO
4039 	if (hapd->conf->mbo_enabled && hapd->mbo_assoc_disallow) {
4040 		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
4041 		goto fail;
4042 	}
4043 
4044 	if (hapd->iconf->rssi_reject_assoc_rssi && rssi &&
4045 	    rssi < hapd->iconf->rssi_reject_assoc_rssi &&
4046 	    (sta->auth_rssi == 0 ||
4047 	     sta->auth_rssi < hapd->iconf->rssi_reject_assoc_rssi)) {
4048 		resp = WLAN_STATUS_DENIED_POOR_CHANNEL_CONDITIONS;
4049 		goto fail;
4050 	}
4051 #endif /* CONFIG_MBO */
4052 
4053 	/*
4054 	 * sta->capability is used in check_assoc_ies() for RRM enabled
4055 	 * capability element.
4056 	 */
4057 	sta->capability = capab_info;
4058 
4059 #ifdef CONFIG_FILS
4060 	if (sta->auth_alg == WLAN_AUTH_FILS_SK ||
4061 	    sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
4062 	    sta->auth_alg == WLAN_AUTH_FILS_PK) {
4063 		int res;
4064 
4065 		/* The end of the payload is encrypted. Need to decrypt it
4066 		 * before parsing. */
4067 
4068 		tmp = os_memdup(pos, left);
4069 		if (!tmp) {
4070 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
4071 			goto fail;
4072 		}
4073 
4074 		res = fils_decrypt_assoc(sta->wpa_sm, sta->fils_session, mgmt,
4075 					 len, tmp, left);
4076 		if (res < 0) {
4077 			resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
4078 			goto fail;
4079 		}
4080 		pos = tmp;
4081 		left = res;
4082 	}
4083 #endif /* CONFIG_FILS */
4084 
4085 	/* followed by SSID and Supported rates; and HT capabilities if 802.11n
4086 	 * is used */
4087 	resp = check_assoc_ies(hapd, sta, pos, left, reassoc);
4088 	if (resp != WLAN_STATUS_SUCCESS)
4089 		goto fail;
4090 
4091 	if (hostapd_get_aid(hapd, sta) < 0) {
4092 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4093 			       HOSTAPD_LEVEL_INFO, "No room for more AIDs");
4094 		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
4095 		goto fail;
4096 	}
4097 
4098 	sta->listen_interval = listen_interval;
4099 
4100 	if (hapd->iface->current_mode &&
4101 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G)
4102 		sta->flags |= WLAN_STA_NONERP;
4103 	for (i = 0; i < sta->supported_rates_len; i++) {
4104 		if ((sta->supported_rates[i] & 0x7f) > 22) {
4105 			sta->flags &= ~WLAN_STA_NONERP;
4106 			break;
4107 		}
4108 	}
4109 	if (sta->flags & WLAN_STA_NONERP && !sta->nonerp_set) {
4110 		sta->nonerp_set = 1;
4111 		hapd->iface->num_sta_non_erp++;
4112 		if (hapd->iface->num_sta_non_erp == 1)
4113 			ieee802_11_set_beacons(hapd->iface);
4114 	}
4115 
4116 	if (!(sta->capability & WLAN_CAPABILITY_SHORT_SLOT_TIME) &&
4117 	    !sta->no_short_slot_time_set) {
4118 		sta->no_short_slot_time_set = 1;
4119 		hapd->iface->num_sta_no_short_slot_time++;
4120 		if (hapd->iface->current_mode &&
4121 		    hapd->iface->current_mode->mode ==
4122 		    HOSTAPD_MODE_IEEE80211G &&
4123 		    hapd->iface->num_sta_no_short_slot_time == 1)
4124 			ieee802_11_set_beacons(hapd->iface);
4125 	}
4126 
4127 	if (sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE)
4128 		sta->flags |= WLAN_STA_SHORT_PREAMBLE;
4129 	else
4130 		sta->flags &= ~WLAN_STA_SHORT_PREAMBLE;
4131 
4132 	if (!(sta->capability & WLAN_CAPABILITY_SHORT_PREAMBLE) &&
4133 	    !sta->no_short_preamble_set) {
4134 		sta->no_short_preamble_set = 1;
4135 		hapd->iface->num_sta_no_short_preamble++;
4136 		if (hapd->iface->current_mode &&
4137 		    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211G
4138 		    && hapd->iface->num_sta_no_short_preamble == 1)
4139 			ieee802_11_set_beacons(hapd->iface);
4140 	}
4141 
4142 #ifdef CONFIG_IEEE80211N
4143 	update_ht_state(hapd, sta);
4144 #endif /* CONFIG_IEEE80211N */
4145 
4146 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4147 		       HOSTAPD_LEVEL_DEBUG,
4148 		       "association OK (aid %d)", sta->aid);
4149 	/* Station will be marked associated, after it acknowledges AssocResp
4150 	 */
4151 	sta->flags |= WLAN_STA_ASSOC_REQ_OK;
4152 
4153 #ifdef CONFIG_IEEE80211W
4154 	if ((sta->flags & WLAN_STA_MFP) && sta->sa_query_timed_out) {
4155 		wpa_printf(MSG_DEBUG, "Allowing %sassociation after timed out "
4156 			   "SA Query procedure", reassoc ? "re" : "");
4157 		/* TODO: Send a protected Disassociate frame to the STA using
4158 		 * the old key and Reason Code "Previous Authentication no
4159 		 * longer valid". Make sure this is only sent protected since
4160 		 * unprotected frame would be received by the STA that is now
4161 		 * trying to associate.
4162 		 */
4163 	}
4164 #endif /* CONFIG_IEEE80211W */
4165 
4166 	/* Make sure that the previously registered inactivity timer will not
4167 	 * remove the STA immediately. */
4168 	sta->timeout_next = STA_NULLFUNC;
4169 
4170 #ifdef CONFIG_TAXONOMY
4171 	taxonomy_sta_info_assoc_req(hapd, sta, pos, left);
4172 #endif /* CONFIG_TAXONOMY */
4173 
4174 	sta->pending_wds_enable = 0;
4175 
4176 #ifdef CONFIG_FILS
4177 	if (sta->auth_alg == WLAN_AUTH_FILS_SK ||
4178 	    sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
4179 	    sta->auth_alg == WLAN_AUTH_FILS_PK) {
4180 		if (fils_process_hlp(hapd, sta, pos, left) > 0)
4181 			delay_assoc = 1;
4182 	}
4183 #endif /* CONFIG_FILS */
4184 
4185  fail:
4186 	os_free(identity);
4187 	os_free(radius_cui);
4188 	hostapd_free_psk_list(psk);
4189 
4190 	/*
4191 	 * In case of a successful response, add the station to the driver.
4192 	 * Otherwise, the kernel may ignore Data frames before we process the
4193 	 * ACK frame (TX status). In case of a failure, this station will be
4194 	 * removed.
4195 	 *
4196 	 * Note that this is not compliant with the IEEE 802.11 standard that
4197 	 * states that a non-AP station should transition into the
4198 	 * authenticated/associated state only after the station acknowledges
4199 	 * the (Re)Association Response frame. However, still do this as:
4200 	 *
4201 	 * 1. In case the station does not acknowledge the (Re)Association
4202 	 *    Response frame, it will be removed.
4203 	 * 2. Data frames will be dropped in the kernel until the station is
4204 	 *    set into authorized state, and there are no significant known
4205 	 *    issues with processing other non-Data Class 3 frames during this
4206 	 *    window.
4207 	 */
4208 	if (resp == WLAN_STATUS_SUCCESS && sta &&
4209 	    add_associated_sta(hapd, sta, reassoc))
4210 		resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
4211 
4212 #ifdef CONFIG_FILS
4213 	if (sta && delay_assoc && resp == WLAN_STATUS_SUCCESS &&
4214 	    eloop_is_timeout_registered(fils_hlp_timeout, hapd, sta) &&
4215 	    sta->fils_pending_assoc_req) {
4216 		/* Do not reschedule fils_hlp_timeout in case the station
4217 		 * retransmits (Re)Association Request frame while waiting for
4218 		 * the previously started FILS HLP wait, so that the timeout can
4219 		 * be determined from the first pending attempt. */
4220 		wpa_printf(MSG_DEBUG,
4221 			   "FILS: Continue waiting for HLP processing before sending (Re)Association Response frame to "
4222 			   MACSTR, MAC2STR(sta->addr));
4223 		os_free(tmp);
4224 		return;
4225 	}
4226 	if (sta) {
4227 		eloop_cancel_timeout(fils_hlp_timeout, hapd, sta);
4228 		os_free(sta->fils_pending_assoc_req);
4229 		sta->fils_pending_assoc_req = NULL;
4230 		sta->fils_pending_assoc_req_len = 0;
4231 		wpabuf_free(sta->fils_hlp_resp);
4232 		sta->fils_hlp_resp = NULL;
4233 	}
4234 	if (sta && delay_assoc && resp == WLAN_STATUS_SUCCESS) {
4235 		sta->fils_pending_assoc_req = tmp;
4236 		sta->fils_pending_assoc_req_len = left;
4237 		sta->fils_pending_assoc_is_reassoc = reassoc;
4238 		sta->fils_drv_assoc_finish = 0;
4239 		wpa_printf(MSG_DEBUG,
4240 			   "FILS: Waiting for HLP processing before sending (Re)Association Response frame to "
4241 			   MACSTR, MAC2STR(sta->addr));
4242 		eloop_cancel_timeout(fils_hlp_timeout, hapd, sta);
4243 		eloop_register_timeout(0, hapd->conf->fils_hlp_wait_time * 1024,
4244 				       fils_hlp_timeout, hapd, sta);
4245 		return;
4246 	}
4247 #endif /* CONFIG_FILS */
4248 
4249 	reply_res = send_assoc_resp(hapd, sta, mgmt->sa, resp, reassoc, pos,
4250 				    left, rssi);
4251 	os_free(tmp);
4252 
4253 	/*
4254 	 * Remove the station in case tranmission of a success response fails
4255 	 * (the STA was added associated to the driver) or if the station was
4256 	 * previously added unassociated.
4257 	 */
4258 	if (sta && ((reply_res != WLAN_STATUS_SUCCESS &&
4259 		     resp == WLAN_STATUS_SUCCESS) || sta->added_unassoc)) {
4260 		hostapd_drv_sta_remove(hapd, sta->addr);
4261 		sta->added_unassoc = 0;
4262 	}
4263 }
4264 
4265 
4266 static void handle_disassoc(struct hostapd_data *hapd,
4267 			    const struct ieee80211_mgmt *mgmt, size_t len)
4268 {
4269 	struct sta_info *sta;
4270 
4271 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.disassoc)) {
4272 		wpa_printf(MSG_INFO, "handle_disassoc - too short payload (len=%lu)",
4273 			   (unsigned long) len);
4274 		return;
4275 	}
4276 
4277 	wpa_printf(MSG_DEBUG, "disassocation: STA=" MACSTR " reason_code=%d",
4278 		   MAC2STR(mgmt->sa),
4279 		   le_to_host16(mgmt->u.disassoc.reason_code));
4280 
4281 	sta = ap_get_sta(hapd, mgmt->sa);
4282 	if (sta == NULL) {
4283 		wpa_printf(MSG_INFO, "Station " MACSTR " trying to disassociate, but it is not associated",
4284 			   MAC2STR(mgmt->sa));
4285 		return;
4286 	}
4287 
4288 	ap_sta_set_authorized(hapd, sta, 0);
4289 	sta->last_seq_ctrl = WLAN_INVALID_MGMT_SEQ;
4290 	sta->flags &= ~(WLAN_STA_ASSOC | WLAN_STA_ASSOC_REQ_OK);
4291 	wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
4292 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4293 		       HOSTAPD_LEVEL_INFO, "disassociated");
4294 	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
4295 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
4296 	/* Stop Accounting and IEEE 802.1X sessions, but leave the STA
4297 	 * authenticated. */
4298 	accounting_sta_stop(hapd, sta);
4299 	ieee802_1x_free_station(hapd, sta);
4300 	if (sta->ipaddr)
4301 		hostapd_drv_br_delete_ip_neigh(hapd, 4, (u8 *) &sta->ipaddr);
4302 	ap_sta_ip6addr_del(hapd, sta);
4303 	hostapd_drv_sta_remove(hapd, sta->addr);
4304 	sta->added_unassoc = 0;
4305 
4306 	if (sta->timeout_next == STA_NULLFUNC ||
4307 	    sta->timeout_next == STA_DISASSOC) {
4308 		sta->timeout_next = STA_DEAUTH;
4309 		eloop_cancel_timeout(ap_handle_timer, hapd, sta);
4310 		eloop_register_timeout(AP_DEAUTH_DELAY, 0, ap_handle_timer,
4311 				       hapd, sta);
4312 	}
4313 
4314 	mlme_disassociate_indication(
4315 		hapd, sta, le_to_host16(mgmt->u.disassoc.reason_code));
4316 
4317 	/* DMG/IEEE 802.11ad does not use deauthication. Deallocate sta upon
4318 	 * disassociation. */
4319 	if (hapd->iface->current_mode &&
4320 	    hapd->iface->current_mode->mode == HOSTAPD_MODE_IEEE80211AD) {
4321 		sta->flags &= ~WLAN_STA_AUTH;
4322 		wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
4323 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4324 			       HOSTAPD_LEVEL_DEBUG, "deauthenticated");
4325 		ap_free_sta(hapd, sta);
4326 	}
4327 }
4328 
4329 
4330 static void handle_deauth(struct hostapd_data *hapd,
4331 			  const struct ieee80211_mgmt *mgmt, size_t len)
4332 {
4333 	struct sta_info *sta;
4334 
4335 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.deauth)) {
4336 		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "handle_deauth - too short "
4337 			"payload (len=%lu)", (unsigned long) len);
4338 		return;
4339 	}
4340 
4341 	wpa_msg(hapd->msg_ctx, MSG_DEBUG, "deauthentication: STA=" MACSTR
4342 		" reason_code=%d",
4343 		MAC2STR(mgmt->sa), le_to_host16(mgmt->u.deauth.reason_code));
4344 
4345 	sta = ap_get_sta(hapd, mgmt->sa);
4346 	if (sta == NULL) {
4347 		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "Station " MACSTR " trying "
4348 			"to deauthenticate, but it is not authenticated",
4349 			MAC2STR(mgmt->sa));
4350 		return;
4351 	}
4352 
4353 	ap_sta_set_authorized(hapd, sta, 0);
4354 	sta->last_seq_ctrl = WLAN_INVALID_MGMT_SEQ;
4355 	sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC |
4356 			WLAN_STA_ASSOC_REQ_OK);
4357 	wpa_auth_sm_event(sta->wpa_sm, WPA_DEAUTH);
4358 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4359 		       HOSTAPD_LEVEL_DEBUG, "deauthenticated");
4360 	mlme_deauthenticate_indication(
4361 		hapd, sta, le_to_host16(mgmt->u.deauth.reason_code));
4362 	sta->acct_terminate_cause = RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
4363 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
4364 	ap_free_sta(hapd, sta);
4365 }
4366 
4367 
4368 static void handle_beacon(struct hostapd_data *hapd,
4369 			  const struct ieee80211_mgmt *mgmt, size_t len,
4370 			  struct hostapd_frame_info *fi)
4371 {
4372 	struct ieee802_11_elems elems;
4373 
4374 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.beacon)) {
4375 		wpa_printf(MSG_INFO, "handle_beacon - too short payload (len=%lu)",
4376 			   (unsigned long) len);
4377 		return;
4378 	}
4379 
4380 	(void) ieee802_11_parse_elems(mgmt->u.beacon.variable,
4381 				      len - (IEEE80211_HDRLEN +
4382 					     sizeof(mgmt->u.beacon)), &elems,
4383 				      0);
4384 
4385 	ap_list_process_beacon(hapd->iface, mgmt, &elems, fi);
4386 }
4387 
4388 
4389 #ifdef CONFIG_IEEE80211W
4390 static int robust_action_frame(u8 category)
4391 {
4392 	return category != WLAN_ACTION_PUBLIC &&
4393 		category != WLAN_ACTION_HT;
4394 }
4395 #endif /* CONFIG_IEEE80211W */
4396 
4397 
4398 static int handle_action(struct hostapd_data *hapd,
4399 			 const struct ieee80211_mgmt *mgmt, size_t len,
4400 			 unsigned int freq)
4401 {
4402 	struct sta_info *sta;
4403 	u8 *action __maybe_unused;
4404 
4405 	if (len < IEEE80211_HDRLEN + 2 + 1) {
4406 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4407 			       HOSTAPD_LEVEL_DEBUG,
4408 			       "handle_action - too short payload (len=%lu)",
4409 			       (unsigned long) len);
4410 		return 0;
4411 	}
4412 
4413 	action = (u8 *) &mgmt->u.action.u;
4414 	wpa_printf(MSG_DEBUG, "RX_ACTION category %u action %u sa " MACSTR
4415 		   " da " MACSTR " len %d freq %u",
4416 		   mgmt->u.action.category, *action,
4417 		   MAC2STR(mgmt->sa), MAC2STR(mgmt->da), (int) len, freq);
4418 
4419 	sta = ap_get_sta(hapd, mgmt->sa);
4420 
4421 	if (mgmt->u.action.category != WLAN_ACTION_PUBLIC &&
4422 	    (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))) {
4423 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Ignored Action "
4424 			   "frame (category=%u) from unassociated STA " MACSTR,
4425 			   mgmt->u.action.category, MAC2STR(mgmt->sa));
4426 		return 0;
4427 	}
4428 
4429 #ifdef CONFIG_IEEE80211W
4430 	if (sta && (sta->flags & WLAN_STA_MFP) &&
4431 	    !(mgmt->frame_control & host_to_le16(WLAN_FC_ISWEP)) &&
4432 	    robust_action_frame(mgmt->u.action.category)) {
4433 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4434 			       HOSTAPD_LEVEL_DEBUG,
4435 			       "Dropped unprotected Robust Action frame from "
4436 			       "an MFP STA");
4437 		return 0;
4438 	}
4439 #endif /* CONFIG_IEEE80211W */
4440 
4441 	if (sta) {
4442 		u16 fc = le_to_host16(mgmt->frame_control);
4443 		u16 seq_ctrl = le_to_host16(mgmt->seq_ctrl);
4444 
4445 		if ((fc & WLAN_FC_RETRY) &&
4446 		    sta->last_seq_ctrl != WLAN_INVALID_MGMT_SEQ &&
4447 		    sta->last_seq_ctrl == seq_ctrl &&
4448 		    sta->last_subtype == WLAN_FC_STYPE_ACTION) {
4449 			hostapd_logger(hapd, sta->addr,
4450 				       HOSTAPD_MODULE_IEEE80211,
4451 				       HOSTAPD_LEVEL_DEBUG,
4452 				       "Drop repeated action frame seq_ctrl=0x%x",
4453 				       seq_ctrl);
4454 			return 1;
4455 		}
4456 
4457 		sta->last_seq_ctrl = seq_ctrl;
4458 		sta->last_subtype = WLAN_FC_STYPE_ACTION;
4459 	}
4460 
4461 	switch (mgmt->u.action.category) {
4462 #ifdef CONFIG_IEEE80211R_AP
4463 	case WLAN_ACTION_FT:
4464 		if (!sta ||
4465 		    wpa_ft_action_rx(sta->wpa_sm, (u8 *) &mgmt->u.action,
4466 				     len - IEEE80211_HDRLEN))
4467 			break;
4468 		return 1;
4469 #endif /* CONFIG_IEEE80211R_AP */
4470 	case WLAN_ACTION_WMM:
4471 		hostapd_wmm_action(hapd, mgmt, len);
4472 		return 1;
4473 #ifdef CONFIG_IEEE80211W
4474 	case WLAN_ACTION_SA_QUERY:
4475 		ieee802_11_sa_query_action(hapd, mgmt, len);
4476 		return 1;
4477 #endif /* CONFIG_IEEE80211W */
4478 #ifdef CONFIG_WNM_AP
4479 	case WLAN_ACTION_WNM:
4480 		ieee802_11_rx_wnm_action_ap(hapd, mgmt, len);
4481 		return 1;
4482 #endif /* CONFIG_WNM_AP */
4483 #ifdef CONFIG_FST
4484 	case WLAN_ACTION_FST:
4485 		if (hapd->iface->fst)
4486 			fst_rx_action(hapd->iface->fst, mgmt, len);
4487 		else
4488 			wpa_printf(MSG_DEBUG,
4489 				   "FST: Ignore FST Action frame - no FST attached");
4490 		return 1;
4491 #endif /* CONFIG_FST */
4492 	case WLAN_ACTION_PUBLIC:
4493 	case WLAN_ACTION_PROTECTED_DUAL:
4494 #ifdef CONFIG_IEEE80211N
4495 		if (len >= IEEE80211_HDRLEN + 2 &&
4496 		    mgmt->u.action.u.public_action.action ==
4497 		    WLAN_PA_20_40_BSS_COEX) {
4498 			hostapd_2040_coex_action(hapd, mgmt, len);
4499 			return 1;
4500 		}
4501 #endif /* CONFIG_IEEE80211N */
4502 #ifdef CONFIG_DPP
4503 		if (len >= IEEE80211_HDRLEN + 6 &&
4504 		    mgmt->u.action.u.vs_public_action.action ==
4505 		    WLAN_PA_VENDOR_SPECIFIC &&
4506 		    WPA_GET_BE24(mgmt->u.action.u.vs_public_action.oui) ==
4507 		    OUI_WFA &&
4508 		    mgmt->u.action.u.vs_public_action.variable[0] ==
4509 		    DPP_OUI_TYPE) {
4510 			const u8 *pos, *end;
4511 
4512 			pos = mgmt->u.action.u.vs_public_action.oui;
4513 			end = ((const u8 *) mgmt) + len;
4514 			hostapd_dpp_rx_action(hapd, mgmt->sa, pos, end - pos,
4515 					      freq);
4516 			return 1;
4517 		}
4518 		if (len >= IEEE80211_HDRLEN + 2 &&
4519 		    (mgmt->u.action.u.public_action.action ==
4520 		     WLAN_PA_GAS_INITIAL_RESP ||
4521 		     mgmt->u.action.u.public_action.action ==
4522 		     WLAN_PA_GAS_COMEBACK_RESP)) {
4523 			const u8 *pos, *end;
4524 
4525 			pos = &mgmt->u.action.u.public_action.action;
4526 			end = ((const u8 *) mgmt) + len;
4527 			gas_query_ap_rx(hapd->gas, mgmt->sa,
4528 					mgmt->u.action.category,
4529 					pos, end - pos, hapd->iface->freq);
4530 			return 1;
4531 		}
4532 #endif /* CONFIG_DPP */
4533 		if (hapd->public_action_cb) {
4534 			hapd->public_action_cb(hapd->public_action_cb_ctx,
4535 					       (u8 *) mgmt, len,
4536 					       hapd->iface->freq);
4537 		}
4538 		if (hapd->public_action_cb2) {
4539 			hapd->public_action_cb2(hapd->public_action_cb2_ctx,
4540 						(u8 *) mgmt, len,
4541 						hapd->iface->freq);
4542 		}
4543 		if (hapd->public_action_cb || hapd->public_action_cb2)
4544 			return 1;
4545 		break;
4546 	case WLAN_ACTION_VENDOR_SPECIFIC:
4547 		if (hapd->vendor_action_cb) {
4548 			if (hapd->vendor_action_cb(hapd->vendor_action_cb_ctx,
4549 						   (u8 *) mgmt, len,
4550 						   hapd->iface->freq) == 0)
4551 				return 1;
4552 		}
4553 		break;
4554 	case WLAN_ACTION_RADIO_MEASUREMENT:
4555 		hostapd_handle_radio_measurement(hapd, (const u8 *) mgmt, len);
4556 		return 1;
4557 	}
4558 
4559 	hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4560 		       HOSTAPD_LEVEL_DEBUG,
4561 		       "handle_action - unknown action category %d or invalid "
4562 		       "frame",
4563 		       mgmt->u.action.category);
4564 	if (!is_multicast_ether_addr(mgmt->da) &&
4565 	    !(mgmt->u.action.category & 0x80) &&
4566 	    !is_multicast_ether_addr(mgmt->sa)) {
4567 		struct ieee80211_mgmt *resp;
4568 
4569 		/*
4570 		 * IEEE 802.11-REVma/D9.0 - 7.3.1.11
4571 		 * Return the Action frame to the source without change
4572 		 * except that MSB of the Category set to 1.
4573 		 */
4574 		wpa_printf(MSG_DEBUG, "IEEE 802.11: Return unknown Action "
4575 			   "frame back to sender");
4576 		resp = os_memdup(mgmt, len);
4577 		if (resp == NULL)
4578 			return 0;
4579 		os_memcpy(resp->da, resp->sa, ETH_ALEN);
4580 		os_memcpy(resp->sa, hapd->own_addr, ETH_ALEN);
4581 		os_memcpy(resp->bssid, hapd->own_addr, ETH_ALEN);
4582 		resp->u.action.category |= 0x80;
4583 
4584 		if (hostapd_drv_send_mlme(hapd, resp, len, 0) < 0) {
4585 			wpa_printf(MSG_ERROR, "IEEE 802.11: Failed to send "
4586 				   "Action frame");
4587 		}
4588 		os_free(resp);
4589 	}
4590 
4591 	return 1;
4592 }
4593 
4594 
4595 /**
4596  * ieee802_11_mgmt - process incoming IEEE 802.11 management frames
4597  * @hapd: hostapd BSS data structure (the BSS to which the management frame was
4598  * sent to)
4599  * @buf: management frame data (starting from IEEE 802.11 header)
4600  * @len: length of frame data in octets
4601  * @fi: meta data about received frame (signal level, etc.)
4602  *
4603  * Process all incoming IEEE 802.11 management frames. This will be called for
4604  * each frame received from the kernel driver through wlan#ap interface. In
4605  * addition, it can be called to re-inserted pending frames (e.g., when using
4606  * external RADIUS server as an MAC ACL).
4607  */
4608 int ieee802_11_mgmt(struct hostapd_data *hapd, const u8 *buf, size_t len,
4609 		    struct hostapd_frame_info *fi)
4610 {
4611 	struct ieee80211_mgmt *mgmt;
4612 	u16 fc, stype;
4613 	int ret = 0;
4614 	unsigned int freq;
4615 	int ssi_signal = fi ? fi->ssi_signal : 0;
4616 
4617 	if (len < 24)
4618 		return 0;
4619 
4620 	if (fi && fi->freq)
4621 		freq = fi->freq;
4622 	else
4623 		freq = hapd->iface->freq;
4624 
4625 	mgmt = (struct ieee80211_mgmt *) buf;
4626 	fc = le_to_host16(mgmt->frame_control);
4627 	stype = WLAN_FC_GET_STYPE(fc);
4628 
4629 	if (stype == WLAN_FC_STYPE_BEACON) {
4630 		handle_beacon(hapd, mgmt, len, fi);
4631 		return 1;
4632 	}
4633 
4634 	if (!is_broadcast_ether_addr(mgmt->bssid) &&
4635 #ifdef CONFIG_P2P
4636 	    /* Invitation responses can be sent with the peer MAC as BSSID */
4637 	    !((hapd->conf->p2p & P2P_GROUP_OWNER) &&
4638 	      stype == WLAN_FC_STYPE_ACTION) &&
4639 #endif /* CONFIG_P2P */
4640 #ifdef CONFIG_MESH
4641 	    !(hapd->conf->mesh & MESH_ENABLED) &&
4642 #endif /* CONFIG_MESH */
4643 	    os_memcmp(mgmt->bssid, hapd->own_addr, ETH_ALEN) != 0) {
4644 		wpa_printf(MSG_INFO, "MGMT: BSSID=" MACSTR " not our address",
4645 			   MAC2STR(mgmt->bssid));
4646 		return 0;
4647 	}
4648 
4649 
4650 	if (stype == WLAN_FC_STYPE_PROBE_REQ) {
4651 		handle_probe_req(hapd, mgmt, len, ssi_signal);
4652 		return 1;
4653 	}
4654 
4655 	if ((!is_broadcast_ether_addr(mgmt->da) ||
4656 	     stype != WLAN_FC_STYPE_ACTION) &&
4657 	    os_memcmp(mgmt->da, hapd->own_addr, ETH_ALEN) != 0) {
4658 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4659 			       HOSTAPD_LEVEL_DEBUG,
4660 			       "MGMT: DA=" MACSTR " not our address",
4661 			       MAC2STR(mgmt->da));
4662 		return 0;
4663 	}
4664 
4665 	if (hapd->iconf->track_sta_max_num)
4666 		sta_track_add(hapd->iface, mgmt->sa, ssi_signal);
4667 
4668 	switch (stype) {
4669 	case WLAN_FC_STYPE_AUTH:
4670 		wpa_printf(MSG_DEBUG, "mgmt::auth");
4671 		handle_auth(hapd, mgmt, len, ssi_signal, 0);
4672 		ret = 1;
4673 		break;
4674 	case WLAN_FC_STYPE_ASSOC_REQ:
4675 		wpa_printf(MSG_DEBUG, "mgmt::assoc_req");
4676 		handle_assoc(hapd, mgmt, len, 0, ssi_signal);
4677 		ret = 1;
4678 		break;
4679 	case WLAN_FC_STYPE_REASSOC_REQ:
4680 		wpa_printf(MSG_DEBUG, "mgmt::reassoc_req");
4681 		handle_assoc(hapd, mgmt, len, 1, ssi_signal);
4682 		ret = 1;
4683 		break;
4684 	case WLAN_FC_STYPE_DISASSOC:
4685 		wpa_printf(MSG_DEBUG, "mgmt::disassoc");
4686 		handle_disassoc(hapd, mgmt, len);
4687 		ret = 1;
4688 		break;
4689 	case WLAN_FC_STYPE_DEAUTH:
4690 		wpa_msg(hapd->msg_ctx, MSG_DEBUG, "mgmt::deauth");
4691 		handle_deauth(hapd, mgmt, len);
4692 		ret = 1;
4693 		break;
4694 	case WLAN_FC_STYPE_ACTION:
4695 		wpa_printf(MSG_DEBUG, "mgmt::action");
4696 		ret = handle_action(hapd, mgmt, len, freq);
4697 		break;
4698 	default:
4699 		hostapd_logger(hapd, mgmt->sa, HOSTAPD_MODULE_IEEE80211,
4700 			       HOSTAPD_LEVEL_DEBUG,
4701 			       "unknown mgmt frame subtype %d", stype);
4702 		break;
4703 	}
4704 
4705 	return ret;
4706 }
4707 
4708 
4709 static void handle_auth_cb(struct hostapd_data *hapd,
4710 			   const struct ieee80211_mgmt *mgmt,
4711 			   size_t len, int ok)
4712 {
4713 	u16 auth_alg, auth_transaction, status_code;
4714 	struct sta_info *sta;
4715 
4716 	sta = ap_get_sta(hapd, mgmt->da);
4717 	if (!sta) {
4718 		wpa_printf(MSG_DEBUG, "handle_auth_cb: STA " MACSTR
4719 			   " not found",
4720 			   MAC2STR(mgmt->da));
4721 		return;
4722 	}
4723 
4724 	auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
4725 	auth_transaction = le_to_host16(mgmt->u.auth.auth_transaction);
4726 	status_code = le_to_host16(mgmt->u.auth.status_code);
4727 
4728 	if (!ok) {
4729 		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
4730 			       HOSTAPD_LEVEL_NOTICE,
4731 			       "did not acknowledge authentication response");
4732 		goto fail;
4733 	}
4734 
4735 	if (len < IEEE80211_HDRLEN + sizeof(mgmt->u.auth)) {
4736 		wpa_printf(MSG_INFO, "handle_auth_cb - too short payload (len=%lu)",
4737 			   (unsigned long) len);
4738 		goto fail;
4739 	}
4740 
4741 	if (status_code == WLAN_STATUS_SUCCESS &&
4742 	    ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
4743 	     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
4744 		hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4745 			       HOSTAPD_LEVEL_INFO, "authenticated");
4746 		sta->flags |= WLAN_STA_AUTH;
4747 		if (sta->added_unassoc)
4748 			hostapd_set_sta_flags(hapd, sta);
4749 		return;
4750 	}
4751 
4752 fail:
4753 	if (status_code != WLAN_STATUS_SUCCESS && sta->added_unassoc) {
4754 		hostapd_drv_sta_remove(hapd, sta->addr);
4755 		sta->added_unassoc = 0;
4756 	}
4757 }
4758 
4759 
4760 static void hostapd_set_wds_encryption(struct hostapd_data *hapd,
4761 				       struct sta_info *sta,
4762 				       char *ifname_wds)
4763 {
4764 	int i;
4765 	struct hostapd_ssid *ssid = &hapd->conf->ssid;
4766 
4767 	if (hapd->conf->ieee802_1x || hapd->conf->wpa)
4768 		return;
4769 
4770 	for (i = 0; i < 4; i++) {
4771 		if (ssid->wep.key[i] &&
4772 		    hostapd_drv_set_key(ifname_wds, hapd, WPA_ALG_WEP, NULL, i,
4773 					i == ssid->wep.idx, NULL, 0,
4774 					ssid->wep.key[i], ssid->wep.len[i])) {
4775 			wpa_printf(MSG_WARNING,
4776 				   "Could not set WEP keys for WDS interface; %s",
4777 				   ifname_wds);
4778 			break;
4779 		}
4780 	}
4781 }
4782 
4783 
4784 static void handle_assoc_cb(struct hostapd_data *hapd,
4785 			    const struct ieee80211_mgmt *mgmt,
4786 			    size_t len, int reassoc, int ok)
4787 {
4788 	u16 status;
4789 	struct sta_info *sta;
4790 	int new_assoc = 1;
4791 
4792 	sta = ap_get_sta(hapd, mgmt->da);
4793 	if (!sta) {
4794 		wpa_printf(MSG_INFO, "handle_assoc_cb: STA " MACSTR " not found",
4795 			   MAC2STR(mgmt->da));
4796 		return;
4797 	}
4798 
4799 	if (len < IEEE80211_HDRLEN + (reassoc ? sizeof(mgmt->u.reassoc_resp) :
4800 				      sizeof(mgmt->u.assoc_resp))) {
4801 		wpa_printf(MSG_INFO,
4802 			   "handle_assoc_cb(reassoc=%d) - too short payload (len=%lu)",
4803 			   reassoc, (unsigned long) len);
4804 		hostapd_drv_sta_remove(hapd, sta->addr);
4805 		return;
4806 	}
4807 
4808 	if (reassoc)
4809 		status = le_to_host16(mgmt->u.reassoc_resp.status_code);
4810 	else
4811 		status = le_to_host16(mgmt->u.assoc_resp.status_code);
4812 
4813 	if (!ok) {
4814 		hostapd_logger(hapd, mgmt->da, HOSTAPD_MODULE_IEEE80211,
4815 			       HOSTAPD_LEVEL_DEBUG,
4816 			       "did not acknowledge association response");
4817 		sta->flags &= ~WLAN_STA_ASSOC_REQ_OK;
4818 		/* The STA is added only in case of SUCCESS */
4819 		if (status == WLAN_STATUS_SUCCESS)
4820 			hostapd_drv_sta_remove(hapd, sta->addr);
4821 
4822 		return;
4823 	}
4824 
4825 	if (status != WLAN_STATUS_SUCCESS)
4826 		return;
4827 
4828 	/* Stop previous accounting session, if one is started, and allocate
4829 	 * new session id for the new session. */
4830 	accounting_sta_stop(hapd, sta);
4831 
4832 	hostapd_logger(hapd, sta->addr, HOSTAPD_MODULE_IEEE80211,
4833 		       HOSTAPD_LEVEL_INFO,
4834 		       "associated (aid %d)",
4835 		       sta->aid);
4836 
4837 	if (sta->flags & WLAN_STA_ASSOC)
4838 		new_assoc = 0;
4839 	sta->flags |= WLAN_STA_ASSOC;
4840 	sta->flags &= ~WLAN_STA_WNM_SLEEP_MODE;
4841 	if ((!hapd->conf->ieee802_1x && !hapd->conf->wpa &&
4842 	     !hapd->conf->osen) ||
4843 	    sta->auth_alg == WLAN_AUTH_FILS_SK ||
4844 	    sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
4845 	    sta->auth_alg == WLAN_AUTH_FILS_PK ||
4846 	    sta->auth_alg == WLAN_AUTH_FT) {
4847 		/*
4848 		 * Open, static WEP, FT protocol, or FILS; no separate
4849 		 * authorization step.
4850 		 */
4851 		ap_sta_set_authorized(hapd, sta, 1);
4852 	}
4853 
4854 	if (reassoc)
4855 		mlme_reassociate_indication(hapd, sta);
4856 	else
4857 		mlme_associate_indication(hapd, sta);
4858 
4859 #ifdef CONFIG_IEEE80211W
4860 	sta->sa_query_timed_out = 0;
4861 #endif /* CONFIG_IEEE80211W */
4862 
4863 	if (sta->eapol_sm == NULL) {
4864 		/*
4865 		 * This STA does not use RADIUS server for EAP authentication,
4866 		 * so bind it to the selected VLAN interface now, since the
4867 		 * interface selection is not going to change anymore.
4868 		 */
4869 		if (ap_sta_bind_vlan(hapd, sta) < 0)
4870 			return;
4871 	} else if (sta->vlan_id) {
4872 		/* VLAN ID already set (e.g., by PMKSA caching), so bind STA */
4873 		if (ap_sta_bind_vlan(hapd, sta) < 0)
4874 			return;
4875 	}
4876 
4877 	hostapd_set_sta_flags(hapd, sta);
4878 
4879 	if (!(sta->flags & WLAN_STA_WDS) && sta->pending_wds_enable) {
4880 		wpa_printf(MSG_DEBUG, "Enable 4-address WDS mode for STA "
4881 			   MACSTR " based on pending request",
4882 			   MAC2STR(sta->addr));
4883 		sta->pending_wds_enable = 0;
4884 		sta->flags |= WLAN_STA_WDS;
4885 	}
4886 
4887 	if (sta->flags & (WLAN_STA_WDS | WLAN_STA_MULTI_AP)) {
4888 		int ret;
4889 		char ifname_wds[IFNAMSIZ + 1];
4890 
4891 		wpa_printf(MSG_DEBUG, "Reenable 4-address WDS mode for STA "
4892 			   MACSTR " (aid %u)",
4893 			   MAC2STR(sta->addr), sta->aid);
4894 		ret = hostapd_set_wds_sta(hapd, ifname_wds, sta->addr,
4895 					  sta->aid, 1);
4896 		if (!ret)
4897 			hostapd_set_wds_encryption(hapd, sta, ifname_wds);
4898 	}
4899 
4900 	if (sta->auth_alg == WLAN_AUTH_FT)
4901 		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC_FT);
4902 	else
4903 		wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
4904 	hapd->new_assoc_sta_cb(hapd, sta, !new_assoc);
4905 	ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
4906 
4907 #ifdef CONFIG_FILS
4908 	if ((sta->auth_alg == WLAN_AUTH_FILS_SK ||
4909 	     sta->auth_alg == WLAN_AUTH_FILS_SK_PFS ||
4910 	     sta->auth_alg == WLAN_AUTH_FILS_PK) &&
4911 	    fils_set_tk(sta->wpa_sm) < 0) {
4912 		wpa_printf(MSG_DEBUG, "FILS: TK configuration failed");
4913 		ap_sta_disconnect(hapd, sta, sta->addr,
4914 				  WLAN_REASON_UNSPECIFIED);
4915 		return;
4916 	}
4917 #endif /* CONFIG_FILS */
4918 
4919 	if (sta->pending_eapol_rx) {
4920 		struct os_reltime now, age;
4921 
4922 		os_get_reltime(&now);
4923 		os_reltime_sub(&now, &sta->pending_eapol_rx->rx_time, &age);
4924 		if (age.sec == 0 && age.usec < 200000) {
4925 			wpa_printf(MSG_DEBUG,
4926 				   "Process pending EAPOL frame that was received from " MACSTR " just before association notification",
4927 				   MAC2STR(sta->addr));
4928 			ieee802_1x_receive(
4929 				hapd, mgmt->da,
4930 				wpabuf_head(sta->pending_eapol_rx->buf),
4931 				wpabuf_len(sta->pending_eapol_rx->buf));
4932 		}
4933 		wpabuf_free(sta->pending_eapol_rx->buf);
4934 		os_free(sta->pending_eapol_rx);
4935 		sta->pending_eapol_rx = NULL;
4936 	}
4937 }
4938 
4939 
4940 static void handle_deauth_cb(struct hostapd_data *hapd,
4941 			     const struct ieee80211_mgmt *mgmt,
4942 			     size_t len, int ok)
4943 {
4944 	struct sta_info *sta;
4945 	if (is_multicast_ether_addr(mgmt->da))
4946 		return;
4947 	sta = ap_get_sta(hapd, mgmt->da);
4948 	if (!sta) {
4949 		wpa_printf(MSG_DEBUG, "handle_deauth_cb: STA " MACSTR
4950 			   " not found", MAC2STR(mgmt->da));
4951 		return;
4952 	}
4953 	if (ok)
4954 		wpa_printf(MSG_DEBUG, "STA " MACSTR " acknowledged deauth",
4955 			   MAC2STR(sta->addr));
4956 	else
4957 		wpa_printf(MSG_DEBUG, "STA " MACSTR " did not acknowledge "
4958 			   "deauth", MAC2STR(sta->addr));
4959 
4960 	ap_sta_deauth_cb(hapd, sta);
4961 }
4962 
4963 
4964 static void handle_disassoc_cb(struct hostapd_data *hapd,
4965 			       const struct ieee80211_mgmt *mgmt,
4966 			       size_t len, int ok)
4967 {
4968 	struct sta_info *sta;
4969 	if (is_multicast_ether_addr(mgmt->da))
4970 		return;
4971 	sta = ap_get_sta(hapd, mgmt->da);
4972 	if (!sta) {
4973 		wpa_printf(MSG_DEBUG, "handle_disassoc_cb: STA " MACSTR
4974 			   " not found", MAC2STR(mgmt->da));
4975 		return;
4976 	}
4977 	if (ok)
4978 		wpa_printf(MSG_DEBUG, "STA " MACSTR " acknowledged disassoc",
4979 			   MAC2STR(sta->addr));
4980 	else
4981 		wpa_printf(MSG_DEBUG, "STA " MACSTR " did not acknowledge "
4982 			   "disassoc", MAC2STR(sta->addr));
4983 
4984 	ap_sta_disassoc_cb(hapd, sta);
4985 }
4986 
4987 
4988 static void handle_action_cb(struct hostapd_data *hapd,
4989 			     const struct ieee80211_mgmt *mgmt,
4990 			     size_t len, int ok)
4991 {
4992 	struct sta_info *sta;
4993 	const struct rrm_measurement_report_element *report;
4994 
4995 	if (is_multicast_ether_addr(mgmt->da))
4996 		return;
4997 #ifdef CONFIG_DPP
4998 	if (len >= IEEE80211_HDRLEN + 6 &&
4999 	    mgmt->u.action.category == WLAN_ACTION_PUBLIC &&
5000 	    mgmt->u.action.u.vs_public_action.action ==
5001 	    WLAN_PA_VENDOR_SPECIFIC &&
5002 	    WPA_GET_BE24(mgmt->u.action.u.vs_public_action.oui) ==
5003 	    OUI_WFA &&
5004 	    mgmt->u.action.u.vs_public_action.variable[0] ==
5005 	    DPP_OUI_TYPE) {
5006 		const u8 *pos, *end;
5007 
5008 		pos = &mgmt->u.action.u.vs_public_action.variable[1];
5009 		end = ((const u8 *) mgmt) + len;
5010 		hostapd_dpp_tx_status(hapd, mgmt->da, pos, end - pos, ok);
5011 		return;
5012 	}
5013 	if (len >= IEEE80211_HDRLEN + 2 &&
5014 	    mgmt->u.action.category == WLAN_ACTION_PUBLIC &&
5015 	    (mgmt->u.action.u.public_action.action ==
5016 	     WLAN_PA_GAS_INITIAL_REQ ||
5017 	     mgmt->u.action.u.public_action.action ==
5018 	     WLAN_PA_GAS_COMEBACK_REQ)) {
5019 		const u8 *pos, *end;
5020 
5021 		pos = mgmt->u.action.u.public_action.variable;
5022 		end = ((const u8 *) mgmt) + len;
5023 		gas_query_ap_tx_status(hapd->gas, mgmt->da, pos, end - pos, ok);
5024 		return;
5025 	}
5026 #endif /* CONFIG_DPP */
5027 	sta = ap_get_sta(hapd, mgmt->da);
5028 	if (!sta) {
5029 		wpa_printf(MSG_DEBUG, "handle_action_cb: STA " MACSTR
5030 			   " not found", MAC2STR(mgmt->da));
5031 		return;
5032 	}
5033 
5034 	if (len < 24 + 5 + sizeof(*report))
5035 		return;
5036 	report = (const struct rrm_measurement_report_element *)
5037 		&mgmt->u.action.u.rrm.variable[2];
5038 	if (mgmt->u.action.category == WLAN_ACTION_RADIO_MEASUREMENT &&
5039 	    mgmt->u.action.u.rrm.action == WLAN_RRM_RADIO_MEASUREMENT_REQUEST &&
5040 	    report->eid == WLAN_EID_MEASURE_REQUEST &&
5041 	    report->len >= 3 &&
5042 	    report->type == MEASURE_TYPE_BEACON)
5043 		hostapd_rrm_beacon_req_tx_status(hapd, mgmt, len, ok);
5044 }
5045 
5046 
5047 /**
5048  * ieee802_11_mgmt_cb - Process management frame TX status callback
5049  * @hapd: hostapd BSS data structure (the BSS from which the management frame
5050  * was sent from)
5051  * @buf: management frame data (starting from IEEE 802.11 header)
5052  * @len: length of frame data in octets
5053  * @stype: management frame subtype from frame control field
5054  * @ok: Whether the frame was ACK'ed
5055  */
5056 void ieee802_11_mgmt_cb(struct hostapd_data *hapd, const u8 *buf, size_t len,
5057 			u16 stype, int ok)
5058 {
5059 	const struct ieee80211_mgmt *mgmt;
5060 	mgmt = (const struct ieee80211_mgmt *) buf;
5061 
5062 #ifdef CONFIG_TESTING_OPTIONS
5063 	if (hapd->ext_mgmt_frame_handling) {
5064 		size_t hex_len = 2 * len + 1;
5065 		char *hex = os_malloc(hex_len);
5066 
5067 		if (hex) {
5068 			wpa_snprintf_hex(hex, hex_len, buf, len);
5069 			wpa_msg(hapd->msg_ctx, MSG_INFO,
5070 				"MGMT-TX-STATUS stype=%u ok=%d buf=%s",
5071 				stype, ok, hex);
5072 			os_free(hex);
5073 		}
5074 		return;
5075 	}
5076 #endif /* CONFIG_TESTING_OPTIONS */
5077 
5078 	switch (stype) {
5079 	case WLAN_FC_STYPE_AUTH:
5080 		wpa_printf(MSG_DEBUG, "mgmt::auth cb");
5081 		handle_auth_cb(hapd, mgmt, len, ok);
5082 		break;
5083 	case WLAN_FC_STYPE_ASSOC_RESP:
5084 		wpa_printf(MSG_DEBUG, "mgmt::assoc_resp cb");
5085 		handle_assoc_cb(hapd, mgmt, len, 0, ok);
5086 		break;
5087 	case WLAN_FC_STYPE_REASSOC_RESP:
5088 		wpa_printf(MSG_DEBUG, "mgmt::reassoc_resp cb");
5089 		handle_assoc_cb(hapd, mgmt, len, 1, ok);
5090 		break;
5091 	case WLAN_FC_STYPE_PROBE_RESP:
5092 		wpa_printf(MSG_EXCESSIVE, "mgmt::proberesp cb ok=%d", ok);
5093 		break;
5094 	case WLAN_FC_STYPE_DEAUTH:
5095 		wpa_printf(MSG_DEBUG, "mgmt::deauth cb");
5096 		handle_deauth_cb(hapd, mgmt, len, ok);
5097 		break;
5098 	case WLAN_FC_STYPE_DISASSOC:
5099 		wpa_printf(MSG_DEBUG, "mgmt::disassoc cb");
5100 		handle_disassoc_cb(hapd, mgmt, len, ok);
5101 		break;
5102 	case WLAN_FC_STYPE_ACTION:
5103 		wpa_printf(MSG_DEBUG, "mgmt::action cb ok=%d", ok);
5104 		handle_action_cb(hapd, mgmt, len, ok);
5105 		break;
5106 	default:
5107 		wpa_printf(MSG_INFO, "unknown mgmt cb frame subtype %d", stype);
5108 		break;
5109 	}
5110 }
5111 
5112 
5113 int ieee802_11_get_mib(struct hostapd_data *hapd, char *buf, size_t buflen)
5114 {
5115 	/* TODO */
5116 	return 0;
5117 }
5118 
5119 
5120 int ieee802_11_get_mib_sta(struct hostapd_data *hapd, struct sta_info *sta,
5121 			   char *buf, size_t buflen)
5122 {
5123 	/* TODO */
5124 	return 0;
5125 }
5126 
5127 
5128 void hostapd_tx_status(struct hostapd_data *hapd, const u8 *addr,
5129 		       const u8 *buf, size_t len, int ack)
5130 {
5131 	struct sta_info *sta;
5132 	struct hostapd_iface *iface = hapd->iface;
5133 
5134 	sta = ap_get_sta(hapd, addr);
5135 	if (sta == NULL && iface->num_bss > 1) {
5136 		size_t j;
5137 		for (j = 0; j < iface->num_bss; j++) {
5138 			hapd = iface->bss[j];
5139 			sta = ap_get_sta(hapd, addr);
5140 			if (sta)
5141 				break;
5142 		}
5143 	}
5144 	if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))
5145 		return;
5146 	if (sta->flags & WLAN_STA_PENDING_POLL) {
5147 		wpa_printf(MSG_DEBUG, "STA " MACSTR " %s pending "
5148 			   "activity poll", MAC2STR(sta->addr),
5149 			   ack ? "ACKed" : "did not ACK");
5150 		if (ack)
5151 			sta->flags &= ~WLAN_STA_PENDING_POLL;
5152 	}
5153 
5154 	ieee802_1x_tx_status(hapd, sta, buf, len, ack);
5155 }
5156 
5157 
5158 void hostapd_eapol_tx_status(struct hostapd_data *hapd, const u8 *dst,
5159 			     const u8 *data, size_t len, int ack)
5160 {
5161 	struct sta_info *sta;
5162 	struct hostapd_iface *iface = hapd->iface;
5163 
5164 	sta = ap_get_sta(hapd, dst);
5165 	if (sta == NULL && iface->num_bss > 1) {
5166 		size_t j;
5167 		for (j = 0; j < iface->num_bss; j++) {
5168 			hapd = iface->bss[j];
5169 			sta = ap_get_sta(hapd, dst);
5170 			if (sta)
5171 				break;
5172 		}
5173 	}
5174 	if (sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) {
5175 		wpa_printf(MSG_DEBUG, "Ignore TX status for Data frame to STA "
5176 			   MACSTR " that is not currently associated",
5177 			   MAC2STR(dst));
5178 		return;
5179 	}
5180 
5181 	ieee802_1x_eapol_tx_status(hapd, sta, data, len, ack);
5182 }
5183 
5184 
5185 void hostapd_client_poll_ok(struct hostapd_data *hapd, const u8 *addr)
5186 {
5187 	struct sta_info *sta;
5188 	struct hostapd_iface *iface = hapd->iface;
5189 
5190 	sta = ap_get_sta(hapd, addr);
5191 	if (sta == NULL && iface->num_bss > 1) {
5192 		size_t j;
5193 		for (j = 0; j < iface->num_bss; j++) {
5194 			hapd = iface->bss[j];
5195 			sta = ap_get_sta(hapd, addr);
5196 			if (sta)
5197 				break;
5198 		}
5199 	}
5200 	if (sta == NULL)
5201 		return;
5202 	wpa_msg(hapd->msg_ctx, MSG_INFO, AP_STA_POLL_OK MACSTR,
5203 		MAC2STR(sta->addr));
5204 	if (!(sta->flags & WLAN_STA_PENDING_POLL))
5205 		return;
5206 
5207 	wpa_printf(MSG_DEBUG, "STA " MACSTR " ACKed pending "
5208 		   "activity poll", MAC2STR(sta->addr));
5209 	sta->flags &= ~WLAN_STA_PENDING_POLL;
5210 }
5211 
5212 
5213 void ieee802_11_rx_from_unknown(struct hostapd_data *hapd, const u8 *src,
5214 				int wds)
5215 {
5216 	struct sta_info *sta;
5217 
5218 	sta = ap_get_sta(hapd, src);
5219 	if (sta &&
5220 	    ((sta->flags & WLAN_STA_ASSOC) ||
5221 	     ((sta->flags & WLAN_STA_ASSOC_REQ_OK) && wds))) {
5222 		if (!hapd->conf->wds_sta)
5223 			return;
5224 
5225 		if ((sta->flags & (WLAN_STA_ASSOC | WLAN_STA_ASSOC_REQ_OK)) ==
5226 		    WLAN_STA_ASSOC_REQ_OK) {
5227 			wpa_printf(MSG_DEBUG,
5228 				   "Postpone 4-address WDS mode enabling for STA "
5229 				   MACSTR " since TX status for AssocResp is not yet known",
5230 				   MAC2STR(sta->addr));
5231 			sta->pending_wds_enable = 1;
5232 			return;
5233 		}
5234 
5235 		if (wds && !(sta->flags & WLAN_STA_WDS)) {
5236 			int ret;
5237 			char ifname_wds[IFNAMSIZ + 1];
5238 
5239 			wpa_printf(MSG_DEBUG, "Enable 4-address WDS mode for "
5240 				   "STA " MACSTR " (aid %u)",
5241 				   MAC2STR(sta->addr), sta->aid);
5242 			sta->flags |= WLAN_STA_WDS;
5243 			ret = hostapd_set_wds_sta(hapd, ifname_wds,
5244 						  sta->addr, sta->aid, 1);
5245 			if (!ret)
5246 				hostapd_set_wds_encryption(hapd, sta,
5247 							   ifname_wds);
5248 		}
5249 		return;
5250 	}
5251 
5252 	wpa_printf(MSG_DEBUG, "Data/PS-poll frame from not associated STA "
5253 		   MACSTR, MAC2STR(src));
5254 	if (is_multicast_ether_addr(src)) {
5255 		/* Broadcast bit set in SA?! Ignore the frame silently. */
5256 		return;
5257 	}
5258 
5259 	if (sta && (sta->flags & WLAN_STA_ASSOC_REQ_OK)) {
5260 		wpa_printf(MSG_DEBUG, "Association Response to the STA has "
5261 			   "already been sent, but no TX status yet known - "
5262 			   "ignore Class 3 frame issue with " MACSTR,
5263 			   MAC2STR(src));
5264 		return;
5265 	}
5266 
5267 	if (sta && (sta->flags & WLAN_STA_AUTH))
5268 		hostapd_drv_sta_disassoc(
5269 			hapd, src,
5270 			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
5271 	else
5272 		hostapd_drv_sta_deauth(
5273 			hapd, src,
5274 			WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
5275 }
5276 
5277 
5278 #endif /* CONFIG_NATIVE_WINDOWS */
5279