xref: /freebsd/contrib/wpa/wpa_supplicant/sme.c (revision 7afb8adff33d47f10a11368ff54bb2eec5b30165)
1 /*
2  * wpa_supplicant - SME
3  * Copyright (c) 2009-2014, 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 "includes.h"
10 
11 #include "common.h"
12 #include "utils/eloop.h"
13 #include "common/ieee802_11_defs.h"
14 #include "common/ieee802_11_common.h"
15 #include "eapol_supp/eapol_supp_sm.h"
16 #include "common/wpa_common.h"
17 #include "common/sae.h"
18 #include "rsn_supp/wpa.h"
19 #include "rsn_supp/pmksa_cache.h"
20 #include "config.h"
21 #include "wpa_supplicant_i.h"
22 #include "driver_i.h"
23 #include "wpas_glue.h"
24 #include "wps_supplicant.h"
25 #include "p2p_supplicant.h"
26 #include "notify.h"
27 #include "bss.h"
28 #include "scan.h"
29 #include "sme.h"
30 #include "hs20_supplicant.h"
31 
32 #define SME_AUTH_TIMEOUT 5
33 #define SME_ASSOC_TIMEOUT 5
34 
35 static void sme_auth_timer(void *eloop_ctx, void *timeout_ctx);
36 static void sme_assoc_timer(void *eloop_ctx, void *timeout_ctx);
37 static void sme_obss_scan_timeout(void *eloop_ctx, void *timeout_ctx);
38 #ifdef CONFIG_IEEE80211W
39 static void sme_stop_sa_query(struct wpa_supplicant *wpa_s);
40 #endif /* CONFIG_IEEE80211W */
41 
42 
43 #ifdef CONFIG_SAE
44 
45 static int index_within_array(const int *array, int idx)
46 {
47 	int i;
48 	for (i = 0; i < idx; i++) {
49 		if (array[i] <= 0)
50 			return 0;
51 	}
52 	return 1;
53 }
54 
55 
56 static int sme_set_sae_group(struct wpa_supplicant *wpa_s)
57 {
58 	int *groups = wpa_s->conf->sae_groups;
59 	int default_groups[] = { 19, 20, 21, 25, 26, 0 };
60 
61 	if (!groups || groups[0] <= 0)
62 		groups = default_groups;
63 
64 	/* Configuration may have changed, so validate current index */
65 	if (!index_within_array(groups, wpa_s->sme.sae_group_index))
66 		return -1;
67 
68 	for (;;) {
69 		int group = groups[wpa_s->sme.sae_group_index];
70 		if (group < 0)
71 			break;
72 		if (sae_set_group(&wpa_s->sme.sae, group) == 0) {
73 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected SAE group %d",
74 				wpa_s->sme.sae.group);
75 		       return 0;
76 		}
77 		wpa_s->sme.sae_group_index++;
78 	}
79 
80 	return -1;
81 }
82 
83 
84 static struct wpabuf * sme_auth_build_sae_commit(struct wpa_supplicant *wpa_s,
85 						 struct wpa_ssid *ssid,
86 						 const u8 *bssid)
87 {
88 	struct wpabuf *buf;
89 	size_t len;
90 
91 	if (ssid->passphrase == NULL) {
92 		wpa_printf(MSG_DEBUG, "SAE: No password available");
93 		return NULL;
94 	}
95 
96 	if (sme_set_sae_group(wpa_s) < 0) {
97 		wpa_printf(MSG_DEBUG, "SAE: Failed to select group");
98 		return NULL;
99 	}
100 
101 	if (sae_prepare_commit(wpa_s->own_addr, bssid,
102 			       (u8 *) ssid->passphrase,
103 			       os_strlen(ssid->passphrase),
104 			       &wpa_s->sme.sae) < 0) {
105 		wpa_printf(MSG_DEBUG, "SAE: Could not pick PWE");
106 		return NULL;
107 	}
108 
109 	len = wpa_s->sme.sae_token ? wpabuf_len(wpa_s->sme.sae_token) : 0;
110 	buf = wpabuf_alloc(4 + SAE_COMMIT_MAX_LEN + len);
111 	if (buf == NULL)
112 		return NULL;
113 
114 	wpabuf_put_le16(buf, 1); /* Transaction seq# */
115 	wpabuf_put_le16(buf, WLAN_STATUS_SUCCESS);
116 	sae_write_commit(&wpa_s->sme.sae, buf, wpa_s->sme.sae_token);
117 
118 	return buf;
119 }
120 
121 
122 static struct wpabuf * sme_auth_build_sae_confirm(struct wpa_supplicant *wpa_s)
123 {
124 	struct wpabuf *buf;
125 
126 	buf = wpabuf_alloc(4 + SAE_CONFIRM_MAX_LEN);
127 	if (buf == NULL)
128 		return NULL;
129 
130 	wpabuf_put_le16(buf, 2); /* Transaction seq# */
131 	wpabuf_put_le16(buf, WLAN_STATUS_SUCCESS);
132 	sae_write_confirm(&wpa_s->sme.sae, buf);
133 
134 	return buf;
135 }
136 
137 #endif /* CONFIG_SAE */
138 
139 
140 /**
141  * sme_auth_handle_rrm - Handle RRM aspects of current authentication attempt
142  * @wpa_s: Pointer to wpa_supplicant data
143  * @bss: Pointer to the bss which is the target of authentication attempt
144  */
145 static void sme_auth_handle_rrm(struct wpa_supplicant *wpa_s,
146 				struct wpa_bss *bss)
147 {
148 	const u8 rrm_ie_len = 5;
149 	u8 *pos;
150 	const u8 *rrm_ie;
151 
152 	wpa_s->rrm.rrm_used = 0;
153 
154 	wpa_printf(MSG_DEBUG,
155 		   "RRM: Determining whether RRM can be used - device support: 0x%x",
156 		   wpa_s->drv_rrm_flags);
157 
158 	rrm_ie = wpa_bss_get_ie(bss, WLAN_EID_RRM_ENABLED_CAPABILITIES);
159 	if (!rrm_ie || !(bss->caps & IEEE80211_CAP_RRM)) {
160 		wpa_printf(MSG_DEBUG, "RRM: No RRM in network");
161 		return;
162 	}
163 
164 	if (!(wpa_s->drv_rrm_flags &
165 	      WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES) ||
166 	    !(wpa_s->drv_rrm_flags & WPA_DRIVER_FLAGS_QUIET)) {
167 		wpa_printf(MSG_DEBUG,
168 			   "RRM: Insufficient RRM support in driver - do not use RRM");
169 		return;
170 	}
171 
172 	if (sizeof(wpa_s->sme.assoc_req_ie) <
173 	    wpa_s->sme.assoc_req_ie_len + rrm_ie_len + 2) {
174 		wpa_printf(MSG_INFO,
175 			   "RRM: Unable to use RRM, no room for RRM IE");
176 		return;
177 	}
178 
179 	wpa_printf(MSG_DEBUG, "RRM: Adding RRM IE to Association Request");
180 	pos = wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len;
181 	os_memset(pos, 0, 2 + rrm_ie_len);
182 	*pos++ = WLAN_EID_RRM_ENABLED_CAPABILITIES;
183 	*pos++ = rrm_ie_len;
184 
185 	/* Set supported capabilites flags */
186 	if (wpa_s->drv_rrm_flags & WPA_DRIVER_FLAGS_TX_POWER_INSERTION)
187 		*pos |= WLAN_RRM_CAPS_LINK_MEASUREMENT;
188 
189 	wpa_s->sme.assoc_req_ie_len += rrm_ie_len + 2;
190 	wpa_s->rrm.rrm_used = 1;
191 }
192 
193 
194 static void sme_send_authentication(struct wpa_supplicant *wpa_s,
195 				    struct wpa_bss *bss, struct wpa_ssid *ssid,
196 				    int start)
197 {
198 	struct wpa_driver_auth_params params;
199 	struct wpa_ssid *old_ssid;
200 #ifdef CONFIG_IEEE80211R
201 	const u8 *ie;
202 #endif /* CONFIG_IEEE80211R */
203 #ifdef CONFIG_IEEE80211R
204 	const u8 *md = NULL;
205 #endif /* CONFIG_IEEE80211R */
206 	int i, bssid_changed;
207 	struct wpabuf *resp = NULL;
208 	u8 ext_capab[18];
209 	int ext_capab_len;
210 	int skip_auth;
211 
212 	if (bss == NULL) {
213 		wpa_msg(wpa_s, MSG_ERROR, "SME: No scan result available for "
214 			"the network");
215 		wpas_connect_work_done(wpa_s);
216 		return;
217 	}
218 
219 	skip_auth = wpa_s->conf->reassoc_same_bss_optim &&
220 		wpa_s->reassoc_same_bss;
221 	wpa_s->current_bss = bss;
222 
223 	os_memset(&params, 0, sizeof(params));
224 	wpa_s->reassociate = 0;
225 
226 	params.freq = bss->freq;
227 	params.bssid = bss->bssid;
228 	params.ssid = bss->ssid;
229 	params.ssid_len = bss->ssid_len;
230 	params.p2p = ssid->p2p_group;
231 
232 	if (wpa_s->sme.ssid_len != params.ssid_len ||
233 	    os_memcmp(wpa_s->sme.ssid, params.ssid, params.ssid_len) != 0)
234 		wpa_s->sme.prev_bssid_set = 0;
235 
236 	wpa_s->sme.freq = params.freq;
237 	os_memcpy(wpa_s->sme.ssid, params.ssid, params.ssid_len);
238 	wpa_s->sme.ssid_len = params.ssid_len;
239 
240 	params.auth_alg = WPA_AUTH_ALG_OPEN;
241 #ifdef IEEE8021X_EAPOL
242 	if (ssid->key_mgmt & WPA_KEY_MGMT_IEEE8021X_NO_WPA) {
243 		if (ssid->leap) {
244 			if (ssid->non_leap == 0)
245 				params.auth_alg = WPA_AUTH_ALG_LEAP;
246 			else
247 				params.auth_alg |= WPA_AUTH_ALG_LEAP;
248 		}
249 	}
250 #endif /* IEEE8021X_EAPOL */
251 	wpa_dbg(wpa_s, MSG_DEBUG, "Automatic auth_alg selection: 0x%x",
252 		params.auth_alg);
253 	if (ssid->auth_alg) {
254 		params.auth_alg = ssid->auth_alg;
255 		wpa_dbg(wpa_s, MSG_DEBUG, "Overriding auth_alg selection: "
256 			"0x%x", params.auth_alg);
257 	}
258 #ifdef CONFIG_SAE
259 	wpa_s->sme.sae_pmksa_caching = 0;
260 	if (wpa_key_mgmt_sae(ssid->key_mgmt)) {
261 		const u8 *rsn;
262 		struct wpa_ie_data ied;
263 
264 		rsn = wpa_bss_get_ie(bss, WLAN_EID_RSN);
265 		if (!rsn) {
266 			wpa_dbg(wpa_s, MSG_DEBUG,
267 				"SAE enabled, but target BSS does not advertise RSN");
268 		} else if (wpa_parse_wpa_ie(rsn, 2 + rsn[1], &ied) == 0 &&
269 			   wpa_key_mgmt_sae(ied.key_mgmt)) {
270 			wpa_dbg(wpa_s, MSG_DEBUG, "Using SAE auth_alg");
271 			params.auth_alg = WPA_AUTH_ALG_SAE;
272 		} else {
273 			wpa_dbg(wpa_s, MSG_DEBUG,
274 				"SAE enabled, but target BSS does not advertise SAE AKM for RSN");
275 		}
276 	}
277 #endif /* CONFIG_SAE */
278 
279 	for (i = 0; i < NUM_WEP_KEYS; i++) {
280 		if (ssid->wep_key_len[i])
281 			params.wep_key[i] = ssid->wep_key[i];
282 		params.wep_key_len[i] = ssid->wep_key_len[i];
283 	}
284 	params.wep_tx_keyidx = ssid->wep_tx_keyidx;
285 
286 	bssid_changed = !is_zero_ether_addr(wpa_s->bssid);
287 	os_memset(wpa_s->bssid, 0, ETH_ALEN);
288 	os_memcpy(wpa_s->pending_bssid, bss->bssid, ETH_ALEN);
289 	if (bssid_changed)
290 		wpas_notify_bssid_changed(wpa_s);
291 
292 	if ((wpa_bss_get_vendor_ie(bss, WPA_IE_VENDOR_TYPE) ||
293 	     wpa_bss_get_ie(bss, WLAN_EID_RSN)) &&
294 	    wpa_key_mgmt_wpa(ssid->key_mgmt)) {
295 		int try_opportunistic;
296 		try_opportunistic = (ssid->proactive_key_caching < 0 ?
297 				     wpa_s->conf->okc :
298 				     ssid->proactive_key_caching) &&
299 			(ssid->proto & WPA_PROTO_RSN);
300 		if (pmksa_cache_set_current(wpa_s->wpa, NULL, bss->bssid,
301 					    wpa_s->current_ssid,
302 					    try_opportunistic) == 0)
303 			eapol_sm_notify_pmkid_attempt(wpa_s->eapol);
304 		wpa_s->sme.assoc_req_ie_len = sizeof(wpa_s->sme.assoc_req_ie);
305 		if (wpa_supplicant_set_suites(wpa_s, bss, ssid,
306 					      wpa_s->sme.assoc_req_ie,
307 					      &wpa_s->sme.assoc_req_ie_len)) {
308 			wpa_msg(wpa_s, MSG_WARNING, "SME: Failed to set WPA "
309 				"key management and encryption suites");
310 			wpas_connect_work_done(wpa_s);
311 			return;
312 		}
313 	} else if ((ssid->key_mgmt & WPA_KEY_MGMT_IEEE8021X_NO_WPA) &&
314 		   wpa_key_mgmt_wpa_ieee8021x(ssid->key_mgmt)) {
315 		/*
316 		 * Both WPA and non-WPA IEEE 802.1X enabled in configuration -
317 		 * use non-WPA since the scan results did not indicate that the
318 		 * AP is using WPA or WPA2.
319 		 */
320 		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
321 		wpa_s->sme.assoc_req_ie_len = 0;
322 	} else if (wpa_key_mgmt_wpa_any(ssid->key_mgmt)) {
323 		wpa_s->sme.assoc_req_ie_len = sizeof(wpa_s->sme.assoc_req_ie);
324 		if (wpa_supplicant_set_suites(wpa_s, NULL, ssid,
325 					      wpa_s->sme.assoc_req_ie,
326 					      &wpa_s->sme.assoc_req_ie_len)) {
327 			wpa_msg(wpa_s, MSG_WARNING, "SME: Failed to set WPA "
328 				"key management and encryption suites (no "
329 				"scan results)");
330 			wpas_connect_work_done(wpa_s);
331 			return;
332 		}
333 #ifdef CONFIG_WPS
334 	} else if (ssid->key_mgmt & WPA_KEY_MGMT_WPS) {
335 		struct wpabuf *wps_ie;
336 		wps_ie = wps_build_assoc_req_ie(wpas_wps_get_req_type(ssid));
337 		if (wps_ie && wpabuf_len(wps_ie) <=
338 		    sizeof(wpa_s->sme.assoc_req_ie)) {
339 			wpa_s->sme.assoc_req_ie_len = wpabuf_len(wps_ie);
340 			os_memcpy(wpa_s->sme.assoc_req_ie, wpabuf_head(wps_ie),
341 				  wpa_s->sme.assoc_req_ie_len);
342 		} else
343 			wpa_s->sme.assoc_req_ie_len = 0;
344 		wpabuf_free(wps_ie);
345 		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
346 #endif /* CONFIG_WPS */
347 	} else {
348 		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
349 		wpa_s->sme.assoc_req_ie_len = 0;
350 	}
351 
352 #ifdef CONFIG_IEEE80211R
353 	ie = wpa_bss_get_ie(bss, WLAN_EID_MOBILITY_DOMAIN);
354 	if (ie && ie[1] >= MOBILITY_DOMAIN_ID_LEN)
355 		md = ie + 2;
356 	wpa_sm_set_ft_params(wpa_s->wpa, ie, ie ? 2 + ie[1] : 0);
357 	if (md) {
358 		/* Prepare for the next transition */
359 		wpa_ft_prepare_auth_request(wpa_s->wpa, ie);
360 	}
361 
362 	if (md && wpa_key_mgmt_ft(ssid->key_mgmt)) {
363 		if (wpa_s->sme.assoc_req_ie_len + 5 <
364 		    sizeof(wpa_s->sme.assoc_req_ie)) {
365 			struct rsn_mdie *mdie;
366 			u8 *pos = wpa_s->sme.assoc_req_ie +
367 				wpa_s->sme.assoc_req_ie_len;
368 			*pos++ = WLAN_EID_MOBILITY_DOMAIN;
369 			*pos++ = sizeof(*mdie);
370 			mdie = (struct rsn_mdie *) pos;
371 			os_memcpy(mdie->mobility_domain, md,
372 				  MOBILITY_DOMAIN_ID_LEN);
373 			mdie->ft_capab = md[MOBILITY_DOMAIN_ID_LEN];
374 			wpa_s->sme.assoc_req_ie_len += 5;
375 		}
376 
377 		if (wpa_s->sme.ft_used &&
378 		    os_memcmp(md, wpa_s->sme.mobility_domain, 2) == 0 &&
379 		    wpa_sm_has_ptk(wpa_s->wpa)) {
380 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying to use FT "
381 				"over-the-air");
382 			params.auth_alg = WPA_AUTH_ALG_FT;
383 			params.ie = wpa_s->sme.ft_ies;
384 			params.ie_len = wpa_s->sme.ft_ies_len;
385 		}
386 	}
387 #endif /* CONFIG_IEEE80211R */
388 
389 #ifdef CONFIG_IEEE80211W
390 	wpa_s->sme.mfp = wpas_get_ssid_pmf(wpa_s, ssid);
391 	if (wpa_s->sme.mfp != NO_MGMT_FRAME_PROTECTION) {
392 		const u8 *rsn = wpa_bss_get_ie(bss, WLAN_EID_RSN);
393 		struct wpa_ie_data _ie;
394 		if (rsn && wpa_parse_wpa_ie(rsn, 2 + rsn[1], &_ie) == 0 &&
395 		    _ie.capabilities &
396 		    (WPA_CAPABILITY_MFPC | WPA_CAPABILITY_MFPR)) {
397 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected AP supports "
398 				"MFP: require MFP");
399 			wpa_s->sme.mfp = MGMT_FRAME_PROTECTION_REQUIRED;
400 		}
401 	}
402 #endif /* CONFIG_IEEE80211W */
403 
404 #ifdef CONFIG_P2P
405 	if (wpa_s->global->p2p) {
406 		u8 *pos;
407 		size_t len;
408 		int res;
409 		pos = wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len;
410 		len = sizeof(wpa_s->sme.assoc_req_ie) -
411 			wpa_s->sme.assoc_req_ie_len;
412 		res = wpas_p2p_assoc_req_ie(wpa_s, bss, pos, len,
413 					    ssid->p2p_group);
414 		if (res >= 0)
415 			wpa_s->sme.assoc_req_ie_len += res;
416 	}
417 #endif /* CONFIG_P2P */
418 
419 #ifdef CONFIG_HS20
420 	if (is_hs20_network(wpa_s, ssid, bss)) {
421 		struct wpabuf *hs20;
422 		hs20 = wpabuf_alloc(20);
423 		if (hs20) {
424 			int pps_mo_id = hs20_get_pps_mo_id(wpa_s, ssid);
425 			size_t len;
426 
427 			wpas_hs20_add_indication(hs20, pps_mo_id);
428 			len = sizeof(wpa_s->sme.assoc_req_ie) -
429 				wpa_s->sme.assoc_req_ie_len;
430 			if (wpabuf_len(hs20) <= len) {
431 				os_memcpy(wpa_s->sme.assoc_req_ie +
432 					  wpa_s->sme.assoc_req_ie_len,
433 					  wpabuf_head(hs20), wpabuf_len(hs20));
434 				wpa_s->sme.assoc_req_ie_len += wpabuf_len(hs20);
435 			}
436 			wpabuf_free(hs20);
437 		}
438 	}
439 #endif /* CONFIG_HS20 */
440 
441 	ext_capab_len = wpas_build_ext_capab(wpa_s, ext_capab,
442 					     sizeof(ext_capab));
443 	if (ext_capab_len > 0) {
444 		u8 *pos = wpa_s->sme.assoc_req_ie;
445 		if (wpa_s->sme.assoc_req_ie_len > 0 && pos[0] == WLAN_EID_RSN)
446 			pos += 2 + pos[1];
447 		os_memmove(pos + ext_capab_len, pos,
448 			   wpa_s->sme.assoc_req_ie_len -
449 			   (pos - wpa_s->sme.assoc_req_ie));
450 		wpa_s->sme.assoc_req_ie_len += ext_capab_len;
451 		os_memcpy(pos, ext_capab, ext_capab_len);
452 	}
453 
454 	if (wpa_s->vendor_elem[VENDOR_ELEM_ASSOC_REQ]) {
455 		struct wpabuf *buf = wpa_s->vendor_elem[VENDOR_ELEM_ASSOC_REQ];
456 		size_t len;
457 
458 		len = sizeof(wpa_s->sme.assoc_req_ie) -
459 			wpa_s->sme.assoc_req_ie_len;
460 		if (wpabuf_len(buf) <= len) {
461 			os_memcpy(wpa_s->sme.assoc_req_ie +
462 				  wpa_s->sme.assoc_req_ie_len,
463 				  wpabuf_head(buf), wpabuf_len(buf));
464 			wpa_s->sme.assoc_req_ie_len += wpabuf_len(buf);
465 		}
466 	}
467 
468 	sme_auth_handle_rrm(wpa_s, bss);
469 
470 #ifdef CONFIG_SAE
471 	if (!skip_auth && params.auth_alg == WPA_AUTH_ALG_SAE &&
472 	    pmksa_cache_set_current(wpa_s->wpa, NULL, bss->bssid, ssid, 0) == 0)
473 	{
474 		wpa_dbg(wpa_s, MSG_DEBUG,
475 			"PMKSA cache entry found - try to use PMKSA caching instead of new SAE authentication");
476 		params.auth_alg = WPA_AUTH_ALG_OPEN;
477 		wpa_s->sme.sae_pmksa_caching = 1;
478 	}
479 
480 	if (!skip_auth && params.auth_alg == WPA_AUTH_ALG_SAE) {
481 		if (start)
482 			resp = sme_auth_build_sae_commit(wpa_s, ssid,
483 							 bss->bssid);
484 		else
485 			resp = sme_auth_build_sae_confirm(wpa_s);
486 		if (resp == NULL) {
487 			wpas_connection_failed(wpa_s, bss->bssid);
488 			return;
489 		}
490 		params.sae_data = wpabuf_head(resp);
491 		params.sae_data_len = wpabuf_len(resp);
492 		wpa_s->sme.sae.state = start ? SAE_COMMITTED : SAE_CONFIRMED;
493 	}
494 #endif /* CONFIG_SAE */
495 
496 	wpa_supplicant_cancel_sched_scan(wpa_s);
497 	wpa_supplicant_cancel_scan(wpa_s);
498 
499 	wpa_msg(wpa_s, MSG_INFO, "SME: Trying to authenticate with " MACSTR
500 		" (SSID='%s' freq=%d MHz)", MAC2STR(params.bssid),
501 		wpa_ssid_txt(params.ssid, params.ssid_len), params.freq);
502 
503 	wpa_clear_keys(wpa_s, bss->bssid);
504 	wpa_supplicant_set_state(wpa_s, WPA_AUTHENTICATING);
505 	old_ssid = wpa_s->current_ssid;
506 	wpa_s->current_ssid = ssid;
507 	wpa_supplicant_rsn_supp_set_config(wpa_s, wpa_s->current_ssid);
508 	wpa_supplicant_initiate_eapol(wpa_s);
509 	if (old_ssid != wpa_s->current_ssid)
510 		wpas_notify_network_changed(wpa_s);
511 
512 #ifdef CONFIG_P2P
513 	/*
514 	 * If multi-channel concurrency is not supported, check for any
515 	 * frequency conflict. In case of any frequency conflict, remove the
516 	 * least prioritized connection.
517 	 */
518 	if (wpa_s->num_multichan_concurrent < 2) {
519 		int freq, num;
520 		num = get_shared_radio_freqs(wpa_s, &freq, 1);
521 		if (num > 0 && freq > 0 && freq != params.freq) {
522 			wpa_printf(MSG_DEBUG,
523 				   "Conflicting frequency found (%d != %d)",
524 				   freq, params.freq);
525 			if (wpas_p2p_handle_frequency_conflicts(wpa_s,
526 								params.freq,
527 								ssid) < 0) {
528 				wpas_connection_failed(wpa_s, bss->bssid);
529 				wpa_supplicant_mark_disassoc(wpa_s);
530 				wpabuf_free(resp);
531 				wpas_connect_work_done(wpa_s);
532 				return;
533 			}
534 		}
535 	}
536 #endif /* CONFIG_P2P */
537 
538 	if (skip_auth) {
539 		wpa_msg(wpa_s, MSG_DEBUG,
540 			"SME: Skip authentication step on reassoc-to-same-BSS");
541 		wpabuf_free(resp);
542 		sme_associate(wpa_s, ssid->mode, bss->bssid, WLAN_AUTH_OPEN);
543 		return;
544 	}
545 
546 
547 	wpa_s->sme.auth_alg = params.auth_alg;
548 	if (wpa_drv_authenticate(wpa_s, &params) < 0) {
549 		wpa_msg(wpa_s, MSG_INFO, "SME: Authentication request to the "
550 			"driver failed");
551 		wpas_connection_failed(wpa_s, bss->bssid);
552 		wpa_supplicant_mark_disassoc(wpa_s);
553 		wpabuf_free(resp);
554 		wpas_connect_work_done(wpa_s);
555 		return;
556 	}
557 
558 	eloop_register_timeout(SME_AUTH_TIMEOUT, 0, sme_auth_timer, wpa_s,
559 			       NULL);
560 
561 	/*
562 	 * Association will be started based on the authentication event from
563 	 * the driver.
564 	 */
565 
566 	wpabuf_free(resp);
567 }
568 
569 
570 static void sme_auth_start_cb(struct wpa_radio_work *work, int deinit)
571 {
572 	struct wpa_connect_work *cwork = work->ctx;
573 	struct wpa_supplicant *wpa_s = work->wpa_s;
574 
575 	if (deinit) {
576 		if (work->started)
577 			wpa_s->connect_work = NULL;
578 
579 		wpas_connect_work_free(cwork);
580 		return;
581 	}
582 
583 	wpa_s->connect_work = work;
584 
585 	if (cwork->bss_removed ||
586 	    !wpas_valid_bss_ssid(wpa_s, cwork->bss, cwork->ssid)) {
587 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: BSS/SSID entry for authentication not valid anymore - drop connection attempt");
588 		wpas_connect_work_done(wpa_s);
589 		return;
590 	}
591 
592 	sme_send_authentication(wpa_s, cwork->bss, cwork->ssid, 1);
593 }
594 
595 
596 void sme_authenticate(struct wpa_supplicant *wpa_s,
597 		      struct wpa_bss *bss, struct wpa_ssid *ssid)
598 {
599 	struct wpa_connect_work *cwork;
600 
601 	if (bss == NULL || ssid == NULL)
602 		return;
603 	if (wpa_s->connect_work) {
604 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Reject sme_authenticate() call since connect_work exist");
605 		return;
606 	}
607 
608 	if (radio_work_pending(wpa_s, "sme-connect")) {
609 		/*
610 		 * The previous sme-connect work might no longer be valid due to
611 		 * the fact that the BSS list was updated. In addition, it makes
612 		 * sense to adhere to the 'newer' decision.
613 		 */
614 		wpa_dbg(wpa_s, MSG_DEBUG,
615 			"SME: Remove previous pending sme-connect");
616 		radio_remove_works(wpa_s, "sme-connect", 0);
617 	}
618 
619 	cwork = os_zalloc(sizeof(*cwork));
620 	if (cwork == NULL)
621 		return;
622 	cwork->bss = bss;
623 	cwork->ssid = ssid;
624 	cwork->sme = 1;
625 
626 #ifdef CONFIG_SAE
627 	wpa_s->sme.sae.state = SAE_NOTHING;
628 	wpa_s->sme.sae.send_confirm = 0;
629 	wpa_s->sme.sae_group_index = 0;
630 #endif /* CONFIG_SAE */
631 
632 	if (radio_add_work(wpa_s, bss->freq, "sme-connect", 1,
633 			   sme_auth_start_cb, cwork) < 0)
634 		wpas_connect_work_free(cwork);
635 }
636 
637 
638 #ifdef CONFIG_SAE
639 
640 static int sme_sae_auth(struct wpa_supplicant *wpa_s, u16 auth_transaction,
641 			u16 status_code, const u8 *data, size_t len)
642 {
643 	int *groups;
644 
645 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: SAE authentication transaction %u "
646 		"status code %u", auth_transaction, status_code);
647 
648 	if (auth_transaction == 1 &&
649 	    status_code == WLAN_STATUS_ANTI_CLOGGING_TOKEN_REQ &&
650 	    wpa_s->sme.sae.state == SAE_COMMITTED &&
651 	    wpa_s->current_bss && wpa_s->current_ssid) {
652 		int default_groups[] = { 19, 20, 21, 25, 26, 0 };
653 		u16 group;
654 
655 		groups = wpa_s->conf->sae_groups;
656 		if (!groups || groups[0] <= 0)
657 			groups = default_groups;
658 
659 		if (len < sizeof(le16)) {
660 			wpa_dbg(wpa_s, MSG_DEBUG,
661 				"SME: Too short SAE anti-clogging token request");
662 			return -1;
663 		}
664 		group = WPA_GET_LE16(data);
665 		wpa_dbg(wpa_s, MSG_DEBUG,
666 			"SME: SAE anti-clogging token requested (group %u)",
667 			group);
668 		if (sae_group_allowed(&wpa_s->sme.sae, groups, group) !=
669 		    WLAN_STATUS_SUCCESS) {
670 			wpa_dbg(wpa_s, MSG_ERROR,
671 				"SME: SAE group %u of anti-clogging request is invalid",
672 				group);
673 			return -1;
674 		}
675 		wpabuf_free(wpa_s->sme.sae_token);
676 		wpa_s->sme.sae_token = wpabuf_alloc_copy(data + sizeof(le16),
677 							 len - sizeof(le16));
678 		sme_send_authentication(wpa_s, wpa_s->current_bss,
679 					wpa_s->current_ssid, 1);
680 		return 0;
681 	}
682 
683 	if (auth_transaction == 1 &&
684 	    status_code == WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED &&
685 	    wpa_s->sme.sae.state == SAE_COMMITTED &&
686 	    wpa_s->current_bss && wpa_s->current_ssid) {
687 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: SAE group not supported");
688 		wpa_s->sme.sae_group_index++;
689 		if (sme_set_sae_group(wpa_s) < 0)
690 			return -1; /* no other groups enabled */
691 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Try next enabled SAE group");
692 		sme_send_authentication(wpa_s, wpa_s->current_bss,
693 					wpa_s->current_ssid, 1);
694 		return 0;
695 	}
696 
697 	if (status_code != WLAN_STATUS_SUCCESS)
698 		return -1;
699 
700 	if (auth_transaction == 1) {
701 		groups = wpa_s->conf->sae_groups;
702 
703 		wpa_dbg(wpa_s, MSG_DEBUG, "SME SAE commit");
704 		if (wpa_s->current_bss == NULL ||
705 		    wpa_s->current_ssid == NULL)
706 			return -1;
707 		if (wpa_s->sme.sae.state != SAE_COMMITTED)
708 			return -1;
709 		if (groups && groups[0] <= 0)
710 			groups = NULL;
711 		if (sae_parse_commit(&wpa_s->sme.sae, data, len, NULL, NULL,
712 				     groups) != WLAN_STATUS_SUCCESS)
713 			return -1;
714 
715 		if (sae_process_commit(&wpa_s->sme.sae) < 0) {
716 			wpa_printf(MSG_DEBUG, "SAE: Failed to process peer "
717 				   "commit");
718 			return -1;
719 		}
720 
721 		wpabuf_free(wpa_s->sme.sae_token);
722 		wpa_s->sme.sae_token = NULL;
723 		sme_send_authentication(wpa_s, wpa_s->current_bss,
724 					wpa_s->current_ssid, 0);
725 		return 0;
726 	} else if (auth_transaction == 2) {
727 		wpa_dbg(wpa_s, MSG_DEBUG, "SME SAE confirm");
728 		if (wpa_s->sme.sae.state != SAE_CONFIRMED)
729 			return -1;
730 		if (sae_check_confirm(&wpa_s->sme.sae, data, len) < 0)
731 			return -1;
732 		wpa_s->sme.sae.state = SAE_ACCEPTED;
733 		sae_clear_temp_data(&wpa_s->sme.sae);
734 		return 1;
735 	}
736 
737 	return -1;
738 }
739 #endif /* CONFIG_SAE */
740 
741 
742 void sme_event_auth(struct wpa_supplicant *wpa_s, union wpa_event_data *data)
743 {
744 	struct wpa_ssid *ssid = wpa_s->current_ssid;
745 
746 	if (ssid == NULL) {
747 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication event "
748 			"when network is not selected");
749 		return;
750 	}
751 
752 	if (wpa_s->wpa_state != WPA_AUTHENTICATING) {
753 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication event "
754 			"when not in authenticating state");
755 		return;
756 	}
757 
758 	if (os_memcmp(wpa_s->pending_bssid, data->auth.peer, ETH_ALEN) != 0) {
759 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication with "
760 			"unexpected peer " MACSTR,
761 			MAC2STR(data->auth.peer));
762 		return;
763 	}
764 
765 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Authentication response: peer=" MACSTR
766 		" auth_type=%d auth_transaction=%d status_code=%d",
767 		MAC2STR(data->auth.peer), data->auth.auth_type,
768 		data->auth.auth_transaction, data->auth.status_code);
769 	wpa_hexdump(MSG_MSGDUMP, "SME: Authentication response IEs",
770 		    data->auth.ies, data->auth.ies_len);
771 
772 	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
773 
774 #ifdef CONFIG_SAE
775 	if (data->auth.auth_type == WLAN_AUTH_SAE) {
776 		int res;
777 		res = sme_sae_auth(wpa_s, data->auth.auth_transaction,
778 				   data->auth.status_code, data->auth.ies,
779 				   data->auth.ies_len);
780 		if (res < 0) {
781 			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
782 			wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
783 
784 		}
785 		if (res != 1)
786 			return;
787 
788 		wpa_printf(MSG_DEBUG, "SME: SAE completed - setting PMK for "
789 			   "4-way handshake");
790 		wpa_sm_set_pmk(wpa_s->wpa, wpa_s->sme.sae.pmk, PMK_LEN,
791 			       wpa_s->pending_bssid);
792 	}
793 #endif /* CONFIG_SAE */
794 
795 	if (data->auth.status_code != WLAN_STATUS_SUCCESS) {
796 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Authentication failed (status "
797 			"code %d)", data->auth.status_code);
798 
799 		if (data->auth.status_code !=
800 		    WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG ||
801 		    wpa_s->sme.auth_alg == data->auth.auth_type ||
802 		    wpa_s->current_ssid->auth_alg == WPA_AUTH_ALG_LEAP) {
803 			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
804 			wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
805 			return;
806 		}
807 
808 		wpas_connect_work_done(wpa_s);
809 
810 		switch (data->auth.auth_type) {
811 		case WLAN_AUTH_OPEN:
812 			wpa_s->current_ssid->auth_alg = WPA_AUTH_ALG_SHARED;
813 
814 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying SHARED auth");
815 			wpa_supplicant_associate(wpa_s, wpa_s->current_bss,
816 						 wpa_s->current_ssid);
817 			return;
818 
819 		case WLAN_AUTH_SHARED_KEY:
820 			wpa_s->current_ssid->auth_alg = WPA_AUTH_ALG_LEAP;
821 
822 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying LEAP auth");
823 			wpa_supplicant_associate(wpa_s, wpa_s->current_bss,
824 						 wpa_s->current_ssid);
825 			return;
826 
827 		default:
828 			return;
829 		}
830 	}
831 
832 #ifdef CONFIG_IEEE80211R
833 	if (data->auth.auth_type == WLAN_AUTH_FT) {
834 		union wpa_event_data edata;
835 		os_memset(&edata, 0, sizeof(edata));
836 		edata.ft_ies.ies = data->auth.ies;
837 		edata.ft_ies.ies_len = data->auth.ies_len;
838 		os_memcpy(edata.ft_ies.target_ap, data->auth.peer, ETH_ALEN);
839 		wpa_supplicant_event(wpa_s, EVENT_FT_RESPONSE, &edata);
840 	}
841 #endif /* CONFIG_IEEE80211R */
842 
843 	sme_associate(wpa_s, ssid->mode, data->auth.peer,
844 		      data->auth.auth_type);
845 }
846 
847 
848 void sme_associate(struct wpa_supplicant *wpa_s, enum wpas_mode mode,
849 		   const u8 *bssid, u16 auth_type)
850 {
851 	struct wpa_driver_associate_params params;
852 	struct ieee802_11_elems elems;
853 #ifdef CONFIG_HT_OVERRIDES
854 	struct ieee80211_ht_capabilities htcaps;
855 	struct ieee80211_ht_capabilities htcaps_mask;
856 #endif /* CONFIG_HT_OVERRIDES */
857 #ifdef CONFIG_VHT_OVERRIDES
858 	struct ieee80211_vht_capabilities vhtcaps;
859 	struct ieee80211_vht_capabilities vhtcaps_mask;
860 #endif /* CONFIG_VHT_OVERRIDES */
861 
862 	os_memset(&params, 0, sizeof(params));
863 	params.bssid = bssid;
864 	params.ssid = wpa_s->sme.ssid;
865 	params.ssid_len = wpa_s->sme.ssid_len;
866 	params.freq.freq = wpa_s->sme.freq;
867 	params.bg_scan_period = wpa_s->current_ssid ?
868 		wpa_s->current_ssid->bg_scan_period : -1;
869 	params.wpa_ie = wpa_s->sme.assoc_req_ie_len ?
870 		wpa_s->sme.assoc_req_ie : NULL;
871 	params.wpa_ie_len = wpa_s->sme.assoc_req_ie_len;
872 	params.pairwise_suite = wpa_s->pairwise_cipher;
873 	params.group_suite = wpa_s->group_cipher;
874 	params.key_mgmt_suite = wpa_s->key_mgmt;
875 	params.wpa_proto = wpa_s->wpa_proto;
876 #ifdef CONFIG_HT_OVERRIDES
877 	os_memset(&htcaps, 0, sizeof(htcaps));
878 	os_memset(&htcaps_mask, 0, sizeof(htcaps_mask));
879 	params.htcaps = (u8 *) &htcaps;
880 	params.htcaps_mask = (u8 *) &htcaps_mask;
881 	wpa_supplicant_apply_ht_overrides(wpa_s, wpa_s->current_ssid, &params);
882 #endif /* CONFIG_HT_OVERRIDES */
883 #ifdef CONFIG_VHT_OVERRIDES
884 	os_memset(&vhtcaps, 0, sizeof(vhtcaps));
885 	os_memset(&vhtcaps_mask, 0, sizeof(vhtcaps_mask));
886 	params.vhtcaps = &vhtcaps;
887 	params.vhtcaps_mask = &vhtcaps_mask;
888 	wpa_supplicant_apply_vht_overrides(wpa_s, wpa_s->current_ssid, &params);
889 #endif /* CONFIG_VHT_OVERRIDES */
890 #ifdef CONFIG_IEEE80211R
891 	if (auth_type == WLAN_AUTH_FT && wpa_s->sme.ft_ies) {
892 		params.wpa_ie = wpa_s->sme.ft_ies;
893 		params.wpa_ie_len = wpa_s->sme.ft_ies_len;
894 	}
895 #endif /* CONFIG_IEEE80211R */
896 	params.mode = mode;
897 	params.mgmt_frame_protection = wpa_s->sme.mfp;
898 	params.rrm_used = wpa_s->rrm.rrm_used;
899 	if (wpa_s->sme.prev_bssid_set)
900 		params.prev_bssid = wpa_s->sme.prev_bssid;
901 
902 	wpa_msg(wpa_s, MSG_INFO, "Trying to associate with " MACSTR
903 		" (SSID='%s' freq=%d MHz)", MAC2STR(params.bssid),
904 		params.ssid ? wpa_ssid_txt(params.ssid, params.ssid_len) : "",
905 		params.freq.freq);
906 
907 	wpa_supplicant_set_state(wpa_s, WPA_ASSOCIATING);
908 
909 	if (params.wpa_ie == NULL ||
910 	    ieee802_11_parse_elems(params.wpa_ie, params.wpa_ie_len, &elems, 0)
911 	    < 0) {
912 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Could not parse own IEs?!");
913 		os_memset(&elems, 0, sizeof(elems));
914 	}
915 	if (elems.rsn_ie) {
916 		params.wpa_proto = WPA_PROTO_RSN;
917 		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.rsn_ie - 2,
918 					elems.rsn_ie_len + 2);
919 	} else if (elems.wpa_ie) {
920 		params.wpa_proto = WPA_PROTO_WPA;
921 		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.wpa_ie - 2,
922 					elems.wpa_ie_len + 2);
923 	} else if (elems.osen) {
924 		params.wpa_proto = WPA_PROTO_OSEN;
925 		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.osen - 2,
926 					elems.osen_len + 2);
927 	} else
928 		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, NULL, 0);
929 	if (wpa_s->current_ssid && wpa_s->current_ssid->p2p_group)
930 		params.p2p = 1;
931 
932 	if (wpa_s->parent->set_sta_uapsd)
933 		params.uapsd = wpa_s->parent->sta_uapsd;
934 	else
935 		params.uapsd = -1;
936 
937 	if (wpa_drv_associate(wpa_s, &params) < 0) {
938 		wpa_msg(wpa_s, MSG_INFO, "SME: Association request to the "
939 			"driver failed");
940 		wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
941 		wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
942 		os_memset(wpa_s->pending_bssid, 0, ETH_ALEN);
943 		return;
944 	}
945 
946 	eloop_register_timeout(SME_ASSOC_TIMEOUT, 0, sme_assoc_timer, wpa_s,
947 			       NULL);
948 }
949 
950 
951 int sme_update_ft_ies(struct wpa_supplicant *wpa_s, const u8 *md,
952 		      const u8 *ies, size_t ies_len)
953 {
954 	if (md == NULL || ies == NULL) {
955 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Remove mobility domain");
956 		os_free(wpa_s->sme.ft_ies);
957 		wpa_s->sme.ft_ies = NULL;
958 		wpa_s->sme.ft_ies_len = 0;
959 		wpa_s->sme.ft_used = 0;
960 		return 0;
961 	}
962 
963 	os_memcpy(wpa_s->sme.mobility_domain, md, MOBILITY_DOMAIN_ID_LEN);
964 	wpa_hexdump(MSG_DEBUG, "SME: FT IEs", ies, ies_len);
965 	os_free(wpa_s->sme.ft_ies);
966 	wpa_s->sme.ft_ies = os_malloc(ies_len);
967 	if (wpa_s->sme.ft_ies == NULL)
968 		return -1;
969 	os_memcpy(wpa_s->sme.ft_ies, ies, ies_len);
970 	wpa_s->sme.ft_ies_len = ies_len;
971 	return 0;
972 }
973 
974 
975 static void sme_deauth(struct wpa_supplicant *wpa_s)
976 {
977 	int bssid_changed;
978 
979 	bssid_changed = !is_zero_ether_addr(wpa_s->bssid);
980 
981 	if (wpa_drv_deauthenticate(wpa_s, wpa_s->pending_bssid,
982 				   WLAN_REASON_DEAUTH_LEAVING) < 0) {
983 		wpa_msg(wpa_s, MSG_INFO, "SME: Deauth request to the driver "
984 			"failed");
985 	}
986 	wpa_s->sme.prev_bssid_set = 0;
987 
988 	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
989 	wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
990 	os_memset(wpa_s->bssid, 0, ETH_ALEN);
991 	os_memset(wpa_s->pending_bssid, 0, ETH_ALEN);
992 	if (bssid_changed)
993 		wpas_notify_bssid_changed(wpa_s);
994 }
995 
996 
997 void sme_event_assoc_reject(struct wpa_supplicant *wpa_s,
998 			    union wpa_event_data *data)
999 {
1000 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association with " MACSTR " failed: "
1001 		"status code %d", MAC2STR(wpa_s->pending_bssid),
1002 		data->assoc_reject.status_code);
1003 
1004 	eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
1005 
1006 #ifdef CONFIG_SAE
1007 	if (wpa_s->sme.sae_pmksa_caching && wpa_s->current_ssid &&
1008 	    wpa_key_mgmt_sae(wpa_s->current_ssid->key_mgmt)) {
1009 		wpa_dbg(wpa_s, MSG_DEBUG,
1010 			"PMKSA caching attempt rejected - drop PMKSA cache entry and fall back to SAE authentication");
1011 		wpa_sm_aborted_cached(wpa_s->wpa);
1012 		wpa_sm_pmksa_cache_flush(wpa_s->wpa, wpa_s->current_ssid);
1013 		if (wpa_s->current_bss) {
1014 			struct wpa_bss *bss = wpa_s->current_bss;
1015 			struct wpa_ssid *ssid = wpa_s->current_ssid;
1016 
1017 			wpa_drv_deauthenticate(wpa_s, wpa_s->pending_bssid,
1018 					       WLAN_REASON_DEAUTH_LEAVING);
1019 			wpas_connect_work_done(wpa_s);
1020 			wpa_supplicant_mark_disassoc(wpa_s);
1021 			wpa_supplicant_connect(wpa_s, bss, ssid);
1022 			return;
1023 		}
1024 	}
1025 #endif /* CONFIG_SAE */
1026 
1027 	/*
1028 	 * For now, unconditionally terminate the previous authentication. In
1029 	 * theory, this should not be needed, but mac80211 gets quite confused
1030 	 * if the authentication is left pending.. Some roaming cases might
1031 	 * benefit from using the previous authentication, so this could be
1032 	 * optimized in the future.
1033 	 */
1034 	sme_deauth(wpa_s);
1035 }
1036 
1037 
1038 void sme_event_auth_timed_out(struct wpa_supplicant *wpa_s,
1039 			      union wpa_event_data *data)
1040 {
1041 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Authentication timed out");
1042 	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1043 	wpa_supplicant_mark_disassoc(wpa_s);
1044 }
1045 
1046 
1047 void sme_event_assoc_timed_out(struct wpa_supplicant *wpa_s,
1048 			       union wpa_event_data *data)
1049 {
1050 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association timed out");
1051 	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1052 	wpa_supplicant_mark_disassoc(wpa_s);
1053 }
1054 
1055 
1056 void sme_event_disassoc(struct wpa_supplicant *wpa_s,
1057 			struct disassoc_info *info)
1058 {
1059 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Disassociation event received");
1060 	if (wpa_s->sme.prev_bssid_set) {
1061 		/*
1062 		 * cfg80211/mac80211 can get into somewhat confused state if
1063 		 * the AP only disassociates us and leaves us in authenticated
1064 		 * state. For now, force the state to be cleared to avoid
1065 		 * confusing errors if we try to associate with the AP again.
1066 		 */
1067 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Deauthenticate to clear "
1068 			"driver state");
1069 		wpa_drv_deauthenticate(wpa_s, wpa_s->sme.prev_bssid,
1070 				       WLAN_REASON_DEAUTH_LEAVING);
1071 	}
1072 }
1073 
1074 
1075 static void sme_auth_timer(void *eloop_ctx, void *timeout_ctx)
1076 {
1077 	struct wpa_supplicant *wpa_s = eloop_ctx;
1078 	if (wpa_s->wpa_state == WPA_AUTHENTICATING) {
1079 		wpa_msg(wpa_s, MSG_DEBUG, "SME: Authentication timeout");
1080 		sme_deauth(wpa_s);
1081 	}
1082 }
1083 
1084 
1085 static void sme_assoc_timer(void *eloop_ctx, void *timeout_ctx)
1086 {
1087 	struct wpa_supplicant *wpa_s = eloop_ctx;
1088 	if (wpa_s->wpa_state == WPA_ASSOCIATING) {
1089 		wpa_msg(wpa_s, MSG_DEBUG, "SME: Association timeout");
1090 		sme_deauth(wpa_s);
1091 	}
1092 }
1093 
1094 
1095 void sme_state_changed(struct wpa_supplicant *wpa_s)
1096 {
1097 	/* Make sure timers are cleaned up appropriately. */
1098 	if (wpa_s->wpa_state != WPA_ASSOCIATING)
1099 		eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
1100 	if (wpa_s->wpa_state != WPA_AUTHENTICATING)
1101 		eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1102 }
1103 
1104 
1105 void sme_disassoc_while_authenticating(struct wpa_supplicant *wpa_s,
1106 				       const u8 *prev_pending_bssid)
1107 {
1108 	/*
1109 	 * mac80211-workaround to force deauth on failed auth cmd,
1110 	 * requires us to remain in authenticating state to allow the
1111 	 * second authentication attempt to be continued properly.
1112 	 */
1113 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Allow pending authentication "
1114 		"to proceed after disconnection event");
1115 	wpa_supplicant_set_state(wpa_s, WPA_AUTHENTICATING);
1116 	os_memcpy(wpa_s->pending_bssid, prev_pending_bssid, ETH_ALEN);
1117 
1118 	/*
1119 	 * Re-arm authentication timer in case auth fails for whatever reason.
1120 	 */
1121 	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1122 	eloop_register_timeout(SME_AUTH_TIMEOUT, 0, sme_auth_timer, wpa_s,
1123 			       NULL);
1124 }
1125 
1126 
1127 void sme_clear_on_disassoc(struct wpa_supplicant *wpa_s)
1128 {
1129 	wpa_s->sme.prev_bssid_set = 0;
1130 #ifdef CONFIG_SAE
1131 	wpabuf_free(wpa_s->sme.sae_token);
1132 	wpa_s->sme.sae_token = NULL;
1133 	sae_clear_data(&wpa_s->sme.sae);
1134 #endif /* CONFIG_SAE */
1135 #ifdef CONFIG_IEEE80211R
1136 	if (wpa_s->sme.ft_ies)
1137 		sme_update_ft_ies(wpa_s, NULL, NULL, 0);
1138 #endif /* CONFIG_IEEE80211R */
1139 }
1140 
1141 
1142 void sme_deinit(struct wpa_supplicant *wpa_s)
1143 {
1144 	os_free(wpa_s->sme.ft_ies);
1145 	wpa_s->sme.ft_ies = NULL;
1146 	wpa_s->sme.ft_ies_len = 0;
1147 #ifdef CONFIG_IEEE80211W
1148 	sme_stop_sa_query(wpa_s);
1149 #endif /* CONFIG_IEEE80211W */
1150 	sme_clear_on_disassoc(wpa_s);
1151 
1152 	eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
1153 	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1154 	eloop_cancel_timeout(sme_obss_scan_timeout, wpa_s, NULL);
1155 }
1156 
1157 
1158 static void sme_send_2040_bss_coex(struct wpa_supplicant *wpa_s,
1159 				   const u8 *chan_list, u8 num_channels,
1160 				   u8 num_intol)
1161 {
1162 	struct ieee80211_2040_bss_coex_ie *bc_ie;
1163 	struct ieee80211_2040_intol_chan_report *ic_report;
1164 	struct wpabuf *buf;
1165 
1166 	wpa_printf(MSG_DEBUG, "SME: Send 20/40 BSS Coexistence to " MACSTR
1167 		   " (num_channels=%u num_intol=%u)",
1168 		   MAC2STR(wpa_s->bssid), num_channels, num_intol);
1169 	wpa_hexdump(MSG_DEBUG, "SME: 20/40 BSS Intolerant Channels",
1170 		    chan_list, num_channels);
1171 
1172 	buf = wpabuf_alloc(2 + /* action.category + action_code */
1173 			   sizeof(struct ieee80211_2040_bss_coex_ie) +
1174 			   sizeof(struct ieee80211_2040_intol_chan_report) +
1175 			   num_channels);
1176 	if (buf == NULL)
1177 		return;
1178 
1179 	wpabuf_put_u8(buf, WLAN_ACTION_PUBLIC);
1180 	wpabuf_put_u8(buf, WLAN_PA_20_40_BSS_COEX);
1181 
1182 	bc_ie = wpabuf_put(buf, sizeof(*bc_ie));
1183 	bc_ie->element_id = WLAN_EID_20_40_BSS_COEXISTENCE;
1184 	bc_ie->length = 1;
1185 	if (num_intol)
1186 		bc_ie->coex_param |= WLAN_20_40_BSS_COEX_20MHZ_WIDTH_REQ;
1187 
1188 	if (num_channels > 0) {
1189 		ic_report = wpabuf_put(buf, sizeof(*ic_report));
1190 		ic_report->element_id = WLAN_EID_20_40_BSS_INTOLERANT;
1191 		ic_report->length = num_channels + 1;
1192 		ic_report->op_class = 0;
1193 		os_memcpy(wpabuf_put(buf, num_channels), chan_list,
1194 			  num_channels);
1195 	}
1196 
1197 	if (wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
1198 				wpa_s->own_addr, wpa_s->bssid,
1199 				wpabuf_head(buf), wpabuf_len(buf), 0) < 0) {
1200 		wpa_msg(wpa_s, MSG_INFO,
1201 			"SME: Failed to send 20/40 BSS Coexistence frame");
1202 	}
1203 
1204 	wpabuf_free(buf);
1205 }
1206 
1207 
1208 int sme_proc_obss_scan(struct wpa_supplicant *wpa_s)
1209 {
1210 	struct wpa_bss *bss;
1211 	const u8 *ie;
1212 	u16 ht_cap;
1213 	u8 chan_list[P2P_MAX_CHANNELS], channel;
1214 	u8 num_channels = 0, num_intol = 0, i;
1215 
1216 	if (!wpa_s->sme.sched_obss_scan)
1217 		return 0;
1218 
1219 	wpa_s->sme.sched_obss_scan = 0;
1220 	if (!wpa_s->current_bss || wpa_s->wpa_state != WPA_COMPLETED)
1221 		return 1;
1222 
1223 	/*
1224 	 * Check whether AP uses regulatory triplet or channel triplet in
1225 	 * country info. Right now the operating class of the BSS channel
1226 	 * width trigger event is "unknown" (IEEE Std 802.11-2012 10.15.12),
1227 	 * based on the assumption that operating class triplet is not used in
1228 	 * beacon frame. If the First Channel Number/Operating Extension
1229 	 * Identifier octet has a positive integer value of 201 or greater,
1230 	 * then its operating class triplet.
1231 	 *
1232 	 * TODO: If Supported Operating Classes element is present in beacon
1233 	 * frame, have to lookup operating class in Annex E and fill them in
1234 	 * 2040 coex frame.
1235 	 */
1236 	ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_COUNTRY);
1237 	if (ie && (ie[1] >= 6) && (ie[5] >= 201))
1238 		return 1;
1239 
1240 	os_memset(chan_list, 0, sizeof(chan_list));
1241 
1242 	dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1243 		/* Skip other band bss */
1244 		enum hostapd_hw_mode mode;
1245 		mode = ieee80211_freq_to_chan(bss->freq, &channel);
1246 		if (mode != HOSTAPD_MODE_IEEE80211G &&
1247 		    mode != HOSTAPD_MODE_IEEE80211B)
1248 			continue;
1249 
1250 		ie = wpa_bss_get_ie(bss, WLAN_EID_HT_CAP);
1251 		ht_cap = (ie && (ie[1] == 26)) ? WPA_GET_LE16(ie + 2) : 0;
1252 		wpa_printf(MSG_DEBUG, "SME OBSS scan BSS " MACSTR
1253 			   " freq=%u chan=%u ht_cap=0x%x",
1254 			   MAC2STR(bss->bssid), bss->freq, channel, ht_cap);
1255 
1256 		if (!ht_cap || (ht_cap & HT_CAP_INFO_40MHZ_INTOLERANT)) {
1257 			if (ht_cap & HT_CAP_INFO_40MHZ_INTOLERANT)
1258 				num_intol++;
1259 
1260 			/* Check whether the channel is already considered */
1261 			for (i = 0; i < num_channels; i++) {
1262 				if (channel == chan_list[i])
1263 					break;
1264 			}
1265 			if (i != num_channels)
1266 				continue;
1267 
1268 			chan_list[num_channels++] = channel;
1269 		}
1270 	}
1271 
1272 	sme_send_2040_bss_coex(wpa_s, chan_list, num_channels, num_intol);
1273 	return 1;
1274 }
1275 
1276 
1277 static struct hostapd_hw_modes * get_mode(struct hostapd_hw_modes *modes,
1278 					  u16 num_modes,
1279 					  enum hostapd_hw_mode mode)
1280 {
1281 	u16 i;
1282 
1283 	for (i = 0; i < num_modes; i++) {
1284 		if (modes[i].mode == mode)
1285 			return &modes[i];
1286 	}
1287 
1288 	return NULL;
1289 }
1290 
1291 
1292 static void wpa_obss_scan_freqs_list(struct wpa_supplicant *wpa_s,
1293 				     struct wpa_driver_scan_params *params)
1294 {
1295 	/* Include only affected channels */
1296 	struct hostapd_hw_modes *mode;
1297 	int count, i;
1298 	int start, end;
1299 
1300 	mode = get_mode(wpa_s->hw.modes, wpa_s->hw.num_modes,
1301 			HOSTAPD_MODE_IEEE80211G);
1302 	if (mode == NULL) {
1303 		/* No channels supported in this band - use empty list */
1304 		params->freqs = os_zalloc(sizeof(int));
1305 		return;
1306 	}
1307 
1308 	if (wpa_s->sme.ht_sec_chan == HT_SEC_CHAN_UNKNOWN &&
1309 	    wpa_s->current_bss) {
1310 		const u8 *ie;
1311 
1312 		ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_HT_OPERATION);
1313 		if (ie && ie[1] >= 2) {
1314 			u8 o;
1315 
1316 			o = ie[3] & HT_INFO_HT_PARAM_SECONDARY_CHNL_OFF_MASK;
1317 			if (o == HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE)
1318 				wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_ABOVE;
1319 			else if (o == HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW)
1320 				wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_BELOW;
1321 		}
1322 	}
1323 
1324 	start = wpa_s->assoc_freq - 10;
1325 	end = wpa_s->assoc_freq + 10;
1326 	switch (wpa_s->sme.ht_sec_chan) {
1327 	case HT_SEC_CHAN_UNKNOWN:
1328 		/* HT40+ possible on channels 1..9 */
1329 		if (wpa_s->assoc_freq <= 2452)
1330 			start -= 20;
1331 		/* HT40- possible on channels 5-13 */
1332 		if (wpa_s->assoc_freq >= 2432)
1333 			end += 20;
1334 		break;
1335 	case HT_SEC_CHAN_ABOVE:
1336 		end += 20;
1337 		break;
1338 	case HT_SEC_CHAN_BELOW:
1339 		start -= 20;
1340 		break;
1341 	}
1342 	wpa_printf(MSG_DEBUG,
1343 		   "OBSS: assoc_freq %d possible affected range %d-%d",
1344 		   wpa_s->assoc_freq, start, end);
1345 
1346 	params->freqs = os_calloc(mode->num_channels + 1, sizeof(int));
1347 	if (params->freqs == NULL)
1348 		return;
1349 	for (count = 0, i = 0; i < mode->num_channels; i++) {
1350 		int freq;
1351 
1352 		if (mode->channels[i].flag & HOSTAPD_CHAN_DISABLED)
1353 			continue;
1354 		freq = mode->channels[i].freq;
1355 		if (freq - 10 >= end || freq + 10 <= start)
1356 			continue; /* not affected */
1357 		params->freqs[count++] = freq;
1358 	}
1359 }
1360 
1361 
1362 static void sme_obss_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1363 {
1364 	struct wpa_supplicant *wpa_s = eloop_ctx;
1365 	struct wpa_driver_scan_params params;
1366 
1367 	if (!wpa_s->current_bss) {
1368 		wpa_printf(MSG_DEBUG, "SME OBSS: Ignore scan request");
1369 		return;
1370 	}
1371 
1372 	os_memset(&params, 0, sizeof(params));
1373 	wpa_obss_scan_freqs_list(wpa_s, &params);
1374 	params.low_priority = 1;
1375 	wpa_printf(MSG_DEBUG, "SME OBSS: Request an OBSS scan");
1376 
1377 	if (wpa_supplicant_trigger_scan(wpa_s, &params))
1378 		wpa_printf(MSG_DEBUG, "SME OBSS: Failed to trigger scan");
1379 	else
1380 		wpa_s->sme.sched_obss_scan = 1;
1381 	os_free(params.freqs);
1382 
1383 	eloop_register_timeout(wpa_s->sme.obss_scan_int, 0,
1384 			       sme_obss_scan_timeout, wpa_s, NULL);
1385 }
1386 
1387 
1388 void sme_sched_obss_scan(struct wpa_supplicant *wpa_s, int enable)
1389 {
1390 	const u8 *ie;
1391 	struct wpa_bss *bss = wpa_s->current_bss;
1392 	struct wpa_ssid *ssid = wpa_s->current_ssid;
1393 	struct hostapd_hw_modes *hw_mode = NULL;
1394 	int i;
1395 
1396 	eloop_cancel_timeout(sme_obss_scan_timeout, wpa_s, NULL);
1397 	wpa_s->sme.sched_obss_scan = 0;
1398 	wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_UNKNOWN;
1399 	if (!enable)
1400 		return;
1401 
1402 	/*
1403 	 * Schedule OBSS scan if driver is using station SME in wpa_supplicant
1404 	 * or it expects OBSS scan to be performed by wpa_supplicant.
1405 	 */
1406 	if (!((wpa_s->drv_flags & WPA_DRIVER_FLAGS_SME) ||
1407 	      (wpa_s->drv_flags & WPA_DRIVER_FLAGS_OBSS_SCAN)) ||
1408 	    ssid == NULL || ssid->mode != IEEE80211_MODE_INFRA)
1409 		return;
1410 
1411 	if (!wpa_s->hw.modes)
1412 		return;
1413 
1414 	/* only HT caps in 11g mode are relevant */
1415 	for (i = 0; i < wpa_s->hw.num_modes; i++) {
1416 		hw_mode = &wpa_s->hw.modes[i];
1417 		if (hw_mode->mode == HOSTAPD_MODE_IEEE80211G)
1418 			break;
1419 	}
1420 
1421 	/* Driver does not support HT40 for 11g or doesn't have 11g. */
1422 	if (i == wpa_s->hw.num_modes || !hw_mode ||
1423 	    !(hw_mode->ht_capab & HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1424 		return;
1425 
1426 	if (bss == NULL || bss->freq < 2400 || bss->freq > 2500)
1427 		return; /* Not associated on 2.4 GHz band */
1428 
1429 	/* Check whether AP supports HT40 */
1430 	ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_HT_CAP);
1431 	if (!ie || ie[1] < 2 ||
1432 	    !(WPA_GET_LE16(ie + 2) & HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
1433 		return; /* AP does not support HT40 */
1434 
1435 	ie = wpa_bss_get_ie(wpa_s->current_bss,
1436 			    WLAN_EID_OVERLAPPING_BSS_SCAN_PARAMS);
1437 	if (!ie || ie[1] < 14)
1438 		return; /* AP does not request OBSS scans */
1439 
1440 	wpa_s->sme.obss_scan_int = WPA_GET_LE16(ie + 6);
1441 	if (wpa_s->sme.obss_scan_int < 10) {
1442 		wpa_printf(MSG_DEBUG, "SME: Invalid OBSS Scan Interval %u "
1443 			   "replaced with the minimum 10 sec",
1444 			   wpa_s->sme.obss_scan_int);
1445 		wpa_s->sme.obss_scan_int = 10;
1446 	}
1447 	wpa_printf(MSG_DEBUG, "SME: OBSS Scan Interval %u sec",
1448 		   wpa_s->sme.obss_scan_int);
1449 	eloop_register_timeout(wpa_s->sme.obss_scan_int, 0,
1450 			       sme_obss_scan_timeout, wpa_s, NULL);
1451 }
1452 
1453 
1454 #ifdef CONFIG_IEEE80211W
1455 
1456 static const unsigned int sa_query_max_timeout = 1000;
1457 static const unsigned int sa_query_retry_timeout = 201;
1458 
1459 static int sme_check_sa_query_timeout(struct wpa_supplicant *wpa_s)
1460 {
1461 	u32 tu;
1462 	struct os_reltime now, passed;
1463 	os_get_reltime(&now);
1464 	os_reltime_sub(&now, &wpa_s->sme.sa_query_start, &passed);
1465 	tu = (passed.sec * 1000000 + passed.usec) / 1024;
1466 	if (sa_query_max_timeout < tu) {
1467 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: SA Query timed out");
1468 		sme_stop_sa_query(wpa_s);
1469 		wpa_supplicant_deauthenticate(
1470 			wpa_s, WLAN_REASON_PREV_AUTH_NOT_VALID);
1471 		return 1;
1472 	}
1473 
1474 	return 0;
1475 }
1476 
1477 
1478 static void sme_send_sa_query_req(struct wpa_supplicant *wpa_s,
1479 				  const u8 *trans_id)
1480 {
1481 	u8 req[2 + WLAN_SA_QUERY_TR_ID_LEN];
1482 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Sending SA Query Request to "
1483 		MACSTR, MAC2STR(wpa_s->bssid));
1484 	wpa_hexdump(MSG_DEBUG, "SME: SA Query Transaction ID",
1485 		    trans_id, WLAN_SA_QUERY_TR_ID_LEN);
1486 	req[0] = WLAN_ACTION_SA_QUERY;
1487 	req[1] = WLAN_SA_QUERY_REQUEST;
1488 	os_memcpy(req + 2, trans_id, WLAN_SA_QUERY_TR_ID_LEN);
1489 	if (wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
1490 				wpa_s->own_addr, wpa_s->bssid,
1491 				req, sizeof(req), 0) < 0)
1492 		wpa_msg(wpa_s, MSG_INFO, "SME: Failed to send SA Query "
1493 			"Request");
1494 }
1495 
1496 
1497 static void sme_sa_query_timer(void *eloop_ctx, void *timeout_ctx)
1498 {
1499 	struct wpa_supplicant *wpa_s = eloop_ctx;
1500 	unsigned int timeout, sec, usec;
1501 	u8 *trans_id, *nbuf;
1502 
1503 	if (wpa_s->sme.sa_query_count > 0 &&
1504 	    sme_check_sa_query_timeout(wpa_s))
1505 		return;
1506 
1507 	nbuf = os_realloc_array(wpa_s->sme.sa_query_trans_id,
1508 				wpa_s->sme.sa_query_count + 1,
1509 				WLAN_SA_QUERY_TR_ID_LEN);
1510 	if (nbuf == NULL)
1511 		return;
1512 	if (wpa_s->sme.sa_query_count == 0) {
1513 		/* Starting a new SA Query procedure */
1514 		os_get_reltime(&wpa_s->sme.sa_query_start);
1515 	}
1516 	trans_id = nbuf + wpa_s->sme.sa_query_count * WLAN_SA_QUERY_TR_ID_LEN;
1517 	wpa_s->sme.sa_query_trans_id = nbuf;
1518 	wpa_s->sme.sa_query_count++;
1519 
1520 	if (os_get_random(trans_id, WLAN_SA_QUERY_TR_ID_LEN) < 0) {
1521 		wpa_printf(MSG_DEBUG, "Could not generate SA Query ID");
1522 		return;
1523 	}
1524 
1525 	timeout = sa_query_retry_timeout;
1526 	sec = ((timeout / 1000) * 1024) / 1000;
1527 	usec = (timeout % 1000) * 1024;
1528 	eloop_register_timeout(sec, usec, sme_sa_query_timer, wpa_s, NULL);
1529 
1530 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association SA Query attempt %d",
1531 		wpa_s->sme.sa_query_count);
1532 
1533 	sme_send_sa_query_req(wpa_s, trans_id);
1534 }
1535 
1536 
1537 static void sme_start_sa_query(struct wpa_supplicant *wpa_s)
1538 {
1539 	sme_sa_query_timer(wpa_s, NULL);
1540 }
1541 
1542 
1543 static void sme_stop_sa_query(struct wpa_supplicant *wpa_s)
1544 {
1545 	eloop_cancel_timeout(sme_sa_query_timer, wpa_s, NULL);
1546 	os_free(wpa_s->sme.sa_query_trans_id);
1547 	wpa_s->sme.sa_query_trans_id = NULL;
1548 	wpa_s->sme.sa_query_count = 0;
1549 }
1550 
1551 
1552 void sme_event_unprot_disconnect(struct wpa_supplicant *wpa_s, const u8 *sa,
1553 				 const u8 *da, u16 reason_code)
1554 {
1555 	struct wpa_ssid *ssid;
1556 	struct os_reltime now;
1557 
1558 	if (wpa_s->wpa_state != WPA_COMPLETED)
1559 		return;
1560 	ssid = wpa_s->current_ssid;
1561 	if (wpas_get_ssid_pmf(wpa_s, ssid) == NO_MGMT_FRAME_PROTECTION)
1562 		return;
1563 	if (os_memcmp(sa, wpa_s->bssid, ETH_ALEN) != 0)
1564 		return;
1565 	if (reason_code != WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA &&
1566 	    reason_code != WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA)
1567 		return;
1568 	if (wpa_s->sme.sa_query_count > 0)
1569 		return;
1570 
1571 	os_get_reltime(&now);
1572 	if (wpa_s->sme.last_unprot_disconnect.sec &&
1573 	    !os_reltime_expired(&now, &wpa_s->sme.last_unprot_disconnect, 10))
1574 		return; /* limit SA Query procedure frequency */
1575 	wpa_s->sme.last_unprot_disconnect = now;
1576 
1577 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Unprotected disconnect dropped - "
1578 		"possible AP/STA state mismatch - trigger SA Query");
1579 	sme_start_sa_query(wpa_s);
1580 }
1581 
1582 
1583 void sme_sa_query_rx(struct wpa_supplicant *wpa_s, const u8 *sa,
1584 		     const u8 *data, size_t len)
1585 {
1586 	int i;
1587 
1588 	if (wpa_s->sme.sa_query_trans_id == NULL ||
1589 	    len < 1 + WLAN_SA_QUERY_TR_ID_LEN ||
1590 	    data[0] != WLAN_SA_QUERY_RESPONSE)
1591 		return;
1592 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Received SA Query response from "
1593 		MACSTR " (trans_id %02x%02x)", MAC2STR(sa), data[1], data[2]);
1594 
1595 	if (os_memcmp(sa, wpa_s->bssid, ETH_ALEN) != 0)
1596 		return;
1597 
1598 	for (i = 0; i < wpa_s->sme.sa_query_count; i++) {
1599 		if (os_memcmp(wpa_s->sme.sa_query_trans_id +
1600 			      i * WLAN_SA_QUERY_TR_ID_LEN,
1601 			      data + 1, WLAN_SA_QUERY_TR_ID_LEN) == 0)
1602 			break;
1603 	}
1604 
1605 	if (i >= wpa_s->sme.sa_query_count) {
1606 		wpa_dbg(wpa_s, MSG_DEBUG, "SME: No matching SA Query "
1607 			"transaction identifier found");
1608 		return;
1609 	}
1610 
1611 	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Reply to pending SA Query received "
1612 		"from " MACSTR, MAC2STR(sa));
1613 	sme_stop_sa_query(wpa_s);
1614 }
1615 
1616 #endif /* CONFIG_IEEE80211W */
1617