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 "common/ocv.h"
16#include "eapol_supp/eapol_supp_sm.h"
17#include "common/wpa_common.h"
18#include "common/sae.h"
19#include "common/dpp.h"
20#include "rsn_supp/wpa.h"
21#include "rsn_supp/pmksa_cache.h"
22#include "config.h"
23#include "wpa_supplicant_i.h"
24#include "driver_i.h"
25#include "wpas_glue.h"
26#include "wps_supplicant.h"
27#include "p2p_supplicant.h"
28#include "notify.h"
29#include "bss.h"
30#include "scan.h"
31#include "sme.h"
32#include "hs20_supplicant.h"
33
34#define SME_AUTH_TIMEOUT 5
35#define SME_ASSOC_TIMEOUT 5
36
37static void sme_auth_timer(void *eloop_ctx, void *timeout_ctx);
38static void sme_assoc_timer(void *eloop_ctx, void *timeout_ctx);
39static void sme_obss_scan_timeout(void *eloop_ctx, void *timeout_ctx);
40#ifdef CONFIG_IEEE80211W
41static void sme_stop_sa_query(struct wpa_supplicant *wpa_s);
42#endif /* CONFIG_IEEE80211W */
43
44
45#ifdef CONFIG_SAE
46
47static int index_within_array(const int *array, int idx)
48{
49	int i;
50	for (i = 0; i < idx; i++) {
51		if (array[i] <= 0)
52			return 0;
53	}
54	return 1;
55}
56
57
58static int sme_set_sae_group(struct wpa_supplicant *wpa_s)
59{
60	int *groups = wpa_s->conf->sae_groups;
61	int default_groups[] = { 19, 20, 21, 0 };
62
63	if (!groups || groups[0] <= 0)
64		groups = default_groups;
65
66	/* Configuration may have changed, so validate current index */
67	if (!index_within_array(groups, wpa_s->sme.sae_group_index))
68		return -1;
69
70	for (;;) {
71		int group = groups[wpa_s->sme.sae_group_index];
72		if (group <= 0)
73			break;
74		if (sae_set_group(&wpa_s->sme.sae, group) == 0) {
75			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected SAE group %d",
76				wpa_s->sme.sae.group);
77			return 0;
78		}
79		wpa_s->sme.sae_group_index++;
80	}
81
82	return -1;
83}
84
85
86static struct wpabuf * sme_auth_build_sae_commit(struct wpa_supplicant *wpa_s,
87						 struct wpa_ssid *ssid,
88						 const u8 *bssid, int external,
89						 int reuse)
90{
91	struct wpabuf *buf;
92	size_t len;
93	const char *password;
94
95#ifdef CONFIG_TESTING_OPTIONS
96	if (wpa_s->sae_commit_override) {
97		wpa_printf(MSG_DEBUG, "SAE: TESTING - commit override");
98		buf = wpabuf_alloc(4 + wpabuf_len(wpa_s->sae_commit_override));
99		if (!buf)
100			return NULL;
101		if (!external) {
102			wpabuf_put_le16(buf, 1); /* Transaction seq# */
103			wpabuf_put_le16(buf, WLAN_STATUS_SUCCESS);
104		}
105		wpabuf_put_buf(buf, wpa_s->sae_commit_override);
106		return buf;
107	}
108#endif /* CONFIG_TESTING_OPTIONS */
109
110	password = ssid->sae_password;
111	if (!password)
112		password = ssid->passphrase;
113	if (!password) {
114		wpa_printf(MSG_DEBUG, "SAE: No password available");
115		return NULL;
116	}
117
118	if (reuse && wpa_s->sme.sae.tmp &&
119	    os_memcmp(bssid, wpa_s->sme.sae.tmp->bssid, ETH_ALEN) == 0) {
120		wpa_printf(MSG_DEBUG,
121			   "SAE: Reuse previously generated PWE on a retry with the same AP");
122		goto reuse_data;
123	}
124	if (sme_set_sae_group(wpa_s) < 0) {
125		wpa_printf(MSG_DEBUG, "SAE: Failed to select group");
126		return NULL;
127	}
128
129	if (sae_prepare_commit(wpa_s->own_addr, bssid,
130			       (u8 *) password, os_strlen(password),
131			       ssid->sae_password_id,
132			       &wpa_s->sme.sae) < 0) {
133		wpa_printf(MSG_DEBUG, "SAE: Could not pick PWE");
134		return NULL;
135	}
136	if (wpa_s->sme.sae.tmp)
137		os_memcpy(wpa_s->sme.sae.tmp->bssid, bssid, ETH_ALEN);
138
139reuse_data:
140	len = wpa_s->sme.sae_token ? wpabuf_len(wpa_s->sme.sae_token) : 0;
141	if (ssid->sae_password_id)
142		len += 4 + os_strlen(ssid->sae_password_id);
143	buf = wpabuf_alloc(4 + SAE_COMMIT_MAX_LEN + len);
144	if (buf == NULL)
145		return NULL;
146	if (!external) {
147		wpabuf_put_le16(buf, 1); /* Transaction seq# */
148		wpabuf_put_le16(buf, WLAN_STATUS_SUCCESS);
149	}
150	sae_write_commit(&wpa_s->sme.sae, buf, wpa_s->sme.sae_token,
151			 ssid->sae_password_id);
152
153	return buf;
154}
155
156
157static struct wpabuf * sme_auth_build_sae_confirm(struct wpa_supplicant *wpa_s,
158						  int external)
159{
160	struct wpabuf *buf;
161
162	buf = wpabuf_alloc(4 + SAE_CONFIRM_MAX_LEN);
163	if (buf == NULL)
164		return NULL;
165
166	if (!external) {
167		wpabuf_put_le16(buf, 2); /* Transaction seq# */
168		wpabuf_put_le16(buf, WLAN_STATUS_SUCCESS);
169	}
170	sae_write_confirm(&wpa_s->sme.sae, buf);
171
172	return buf;
173}
174
175#endif /* CONFIG_SAE */
176
177
178/**
179 * sme_auth_handle_rrm - Handle RRM aspects of current authentication attempt
180 * @wpa_s: Pointer to wpa_supplicant data
181 * @bss: Pointer to the bss which is the target of authentication attempt
182 */
183static void sme_auth_handle_rrm(struct wpa_supplicant *wpa_s,
184				struct wpa_bss *bss)
185{
186	const u8 rrm_ie_len = 5;
187	u8 *pos;
188	const u8 *rrm_ie;
189
190	wpa_s->rrm.rrm_used = 0;
191
192	wpa_printf(MSG_DEBUG,
193		   "RRM: Determining whether RRM can be used - device support: 0x%x",
194		   wpa_s->drv_rrm_flags);
195
196	rrm_ie = wpa_bss_get_ie(bss, WLAN_EID_RRM_ENABLED_CAPABILITIES);
197	if (!rrm_ie || !(bss->caps & IEEE80211_CAP_RRM)) {
198		wpa_printf(MSG_DEBUG, "RRM: No RRM in network");
199		return;
200	}
201
202	if (!((wpa_s->drv_rrm_flags &
203	       WPA_DRIVER_FLAGS_DS_PARAM_SET_IE_IN_PROBES) &&
204	      (wpa_s->drv_rrm_flags & WPA_DRIVER_FLAGS_QUIET)) &&
205	    !(wpa_s->drv_rrm_flags & WPA_DRIVER_FLAGS_SUPPORT_RRM)) {
206		wpa_printf(MSG_DEBUG,
207			   "RRM: Insufficient RRM support in driver - do not use RRM");
208		return;
209	}
210
211	if (sizeof(wpa_s->sme.assoc_req_ie) <
212	    wpa_s->sme.assoc_req_ie_len + rrm_ie_len + 2) {
213		wpa_printf(MSG_INFO,
214			   "RRM: Unable to use RRM, no room for RRM IE");
215		return;
216	}
217
218	wpa_printf(MSG_DEBUG, "RRM: Adding RRM IE to Association Request");
219	pos = wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len;
220	os_memset(pos, 0, 2 + rrm_ie_len);
221	*pos++ = WLAN_EID_RRM_ENABLED_CAPABILITIES;
222	*pos++ = rrm_ie_len;
223
224	/* Set supported capabilites flags */
225	if (wpa_s->drv_rrm_flags & WPA_DRIVER_FLAGS_TX_POWER_INSERTION)
226		*pos |= WLAN_RRM_CAPS_LINK_MEASUREMENT;
227
228	*pos |= WLAN_RRM_CAPS_BEACON_REPORT_PASSIVE |
229		WLAN_RRM_CAPS_BEACON_REPORT_ACTIVE |
230		WLAN_RRM_CAPS_BEACON_REPORT_TABLE;
231
232	if (wpa_s->lci)
233		pos[1] |= WLAN_RRM_CAPS_LCI_MEASUREMENT;
234
235	wpa_s->sme.assoc_req_ie_len += rrm_ie_len + 2;
236	wpa_s->rrm.rrm_used = 1;
237}
238
239
240static void sme_send_authentication(struct wpa_supplicant *wpa_s,
241				    struct wpa_bss *bss, struct wpa_ssid *ssid,
242				    int start)
243{
244	struct wpa_driver_auth_params params;
245	struct wpa_ssid *old_ssid;
246#ifdef CONFIG_IEEE80211R
247	const u8 *ie;
248#endif /* CONFIG_IEEE80211R */
249#if defined(CONFIG_IEEE80211R) || defined(CONFIG_FILS)
250	const u8 *md = NULL;
251#endif /* CONFIG_IEEE80211R || CONFIG_FILS */
252	int i, bssid_changed;
253	struct wpabuf *resp = NULL;
254	u8 ext_capab[18];
255	int ext_capab_len;
256	int skip_auth;
257	u8 *wpa_ie;
258	size_t wpa_ie_len;
259#ifdef CONFIG_MBO
260	const u8 *mbo_ie;
261#endif /* CONFIG_MBO */
262
263	if (bss == NULL) {
264		wpa_msg(wpa_s, MSG_ERROR, "SME: No scan result available for "
265			"the network");
266		wpas_connect_work_done(wpa_s);
267		return;
268	}
269
270	skip_auth = wpa_s->conf->reassoc_same_bss_optim &&
271		wpa_s->reassoc_same_bss;
272	wpa_s->current_bss = bss;
273
274	os_memset(&params, 0, sizeof(params));
275	wpa_s->reassociate = 0;
276
277	params.freq = bss->freq;
278	params.bssid = bss->bssid;
279	params.ssid = bss->ssid;
280	params.ssid_len = bss->ssid_len;
281	params.p2p = ssid->p2p_group;
282
283	if (wpa_s->sme.ssid_len != params.ssid_len ||
284	    os_memcmp(wpa_s->sme.ssid, params.ssid, params.ssid_len) != 0)
285		wpa_s->sme.prev_bssid_set = 0;
286
287	wpa_s->sme.freq = params.freq;
288	os_memcpy(wpa_s->sme.ssid, params.ssid, params.ssid_len);
289	wpa_s->sme.ssid_len = params.ssid_len;
290
291	params.auth_alg = WPA_AUTH_ALG_OPEN;
292#ifdef IEEE8021X_EAPOL
293	if (ssid->key_mgmt & WPA_KEY_MGMT_IEEE8021X_NO_WPA) {
294		if (ssid->leap) {
295			if (ssid->non_leap == 0)
296				params.auth_alg = WPA_AUTH_ALG_LEAP;
297			else
298				params.auth_alg |= WPA_AUTH_ALG_LEAP;
299		}
300	}
301#endif /* IEEE8021X_EAPOL */
302	wpa_dbg(wpa_s, MSG_DEBUG, "Automatic auth_alg selection: 0x%x",
303		params.auth_alg);
304	if (ssid->auth_alg) {
305		params.auth_alg = ssid->auth_alg;
306		wpa_dbg(wpa_s, MSG_DEBUG, "Overriding auth_alg selection: "
307			"0x%x", params.auth_alg);
308	}
309#ifdef CONFIG_SAE
310	wpa_s->sme.sae_pmksa_caching = 0;
311	if (wpa_key_mgmt_sae(ssid->key_mgmt)) {
312		const u8 *rsn;
313		struct wpa_ie_data ied;
314
315		rsn = wpa_bss_get_ie(bss, WLAN_EID_RSN);
316		if (!rsn) {
317			wpa_dbg(wpa_s, MSG_DEBUG,
318				"SAE enabled, but target BSS does not advertise RSN");
319#ifdef CONFIG_DPP
320		} else if (wpa_parse_wpa_ie(rsn, 2 + rsn[1], &ied) == 0 &&
321			   (ssid->key_mgmt & WPA_KEY_MGMT_DPP) &&
322			   (ied.key_mgmt & WPA_KEY_MGMT_DPP)) {
323			wpa_dbg(wpa_s, MSG_DEBUG, "Prefer DPP over SAE when both are enabled");
324#endif /* CONFIG_DPP */
325		} else if (wpa_parse_wpa_ie(rsn, 2 + rsn[1], &ied) == 0 &&
326			   wpa_key_mgmt_sae(ied.key_mgmt)) {
327			wpa_dbg(wpa_s, MSG_DEBUG, "Using SAE auth_alg");
328			params.auth_alg = WPA_AUTH_ALG_SAE;
329		} else {
330			wpa_dbg(wpa_s, MSG_DEBUG,
331				"SAE enabled, but target BSS does not advertise SAE AKM for RSN");
332		}
333	}
334#endif /* CONFIG_SAE */
335
336	for (i = 0; i < NUM_WEP_KEYS; i++) {
337		if (ssid->wep_key_len[i])
338			params.wep_key[i] = ssid->wep_key[i];
339		params.wep_key_len[i] = ssid->wep_key_len[i];
340	}
341	params.wep_tx_keyidx = ssid->wep_tx_keyidx;
342
343	bssid_changed = !is_zero_ether_addr(wpa_s->bssid);
344	os_memset(wpa_s->bssid, 0, ETH_ALEN);
345	os_memcpy(wpa_s->pending_bssid, bss->bssid, ETH_ALEN);
346	if (bssid_changed)
347		wpas_notify_bssid_changed(wpa_s);
348
349	if ((wpa_bss_get_vendor_ie(bss, WPA_IE_VENDOR_TYPE) ||
350	     wpa_bss_get_ie(bss, WLAN_EID_RSN)) &&
351	    wpa_key_mgmt_wpa(ssid->key_mgmt)) {
352		int try_opportunistic;
353		const u8 *cache_id = NULL;
354
355		try_opportunistic = (ssid->proactive_key_caching < 0 ?
356				     wpa_s->conf->okc :
357				     ssid->proactive_key_caching) &&
358			(ssid->proto & WPA_PROTO_RSN);
359#ifdef CONFIG_FILS
360		if (wpa_key_mgmt_fils(ssid->key_mgmt))
361			cache_id = wpa_bss_get_fils_cache_id(bss);
362#endif /* CONFIG_FILS */
363		if (pmksa_cache_set_current(wpa_s->wpa, NULL, bss->bssid,
364					    wpa_s->current_ssid,
365					    try_opportunistic, cache_id,
366					    0) == 0)
367			eapol_sm_notify_pmkid_attempt(wpa_s->eapol);
368		wpa_s->sme.assoc_req_ie_len = sizeof(wpa_s->sme.assoc_req_ie);
369		if (wpa_supplicant_set_suites(wpa_s, bss, ssid,
370					      wpa_s->sme.assoc_req_ie,
371					      &wpa_s->sme.assoc_req_ie_len)) {
372			wpa_msg(wpa_s, MSG_WARNING, "SME: Failed to set WPA "
373				"key management and encryption suites");
374			wpas_connect_work_done(wpa_s);
375			return;
376		}
377#ifdef CONFIG_HS20
378	} else if (wpa_bss_get_vendor_ie(bss, OSEN_IE_VENDOR_TYPE) &&
379		   (ssid->key_mgmt & WPA_KEY_MGMT_OSEN)) {
380		/* No PMKSA caching, but otherwise similar to RSN/WPA */
381		wpa_s->sme.assoc_req_ie_len = sizeof(wpa_s->sme.assoc_req_ie);
382		if (wpa_supplicant_set_suites(wpa_s, bss, ssid,
383					      wpa_s->sme.assoc_req_ie,
384					      &wpa_s->sme.assoc_req_ie_len)) {
385			wpa_msg(wpa_s, MSG_WARNING, "SME: Failed to set WPA "
386				"key management and encryption suites");
387			wpas_connect_work_done(wpa_s);
388			return;
389		}
390#endif /* CONFIG_HS20 */
391	} else if ((ssid->key_mgmt & WPA_KEY_MGMT_IEEE8021X_NO_WPA) &&
392		   wpa_key_mgmt_wpa_ieee8021x(ssid->key_mgmt)) {
393		/*
394		 * Both WPA and non-WPA IEEE 802.1X enabled in configuration -
395		 * use non-WPA since the scan results did not indicate that the
396		 * AP is using WPA or WPA2.
397		 */
398		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
399		wpa_s->sme.assoc_req_ie_len = 0;
400	} else if (wpa_key_mgmt_wpa_any(ssid->key_mgmt)) {
401		wpa_s->sme.assoc_req_ie_len = sizeof(wpa_s->sme.assoc_req_ie);
402		if (wpa_supplicant_set_suites(wpa_s, NULL, ssid,
403					      wpa_s->sme.assoc_req_ie,
404					      &wpa_s->sme.assoc_req_ie_len)) {
405			wpa_msg(wpa_s, MSG_WARNING, "SME: Failed to set WPA "
406				"key management and encryption suites (no "
407				"scan results)");
408			wpas_connect_work_done(wpa_s);
409			return;
410		}
411#ifdef CONFIG_WPS
412	} else if (ssid->key_mgmt & WPA_KEY_MGMT_WPS) {
413		struct wpabuf *wps_ie;
414		wps_ie = wps_build_assoc_req_ie(wpas_wps_get_req_type(ssid));
415		if (wps_ie && wpabuf_len(wps_ie) <=
416		    sizeof(wpa_s->sme.assoc_req_ie)) {
417			wpa_s->sme.assoc_req_ie_len = wpabuf_len(wps_ie);
418			os_memcpy(wpa_s->sme.assoc_req_ie, wpabuf_head(wps_ie),
419				  wpa_s->sme.assoc_req_ie_len);
420		} else
421			wpa_s->sme.assoc_req_ie_len = 0;
422		wpabuf_free(wps_ie);
423		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
424#endif /* CONFIG_WPS */
425	} else {
426		wpa_supplicant_set_non_wpa_policy(wpa_s, ssid);
427		wpa_s->sme.assoc_req_ie_len = 0;
428	}
429
430	/* In case the WPA vendor IE is used, it should be placed after all the
431	 * non-vendor IEs, as the lower layer expects the IEs to be ordered as
432	 * defined in the standard. Store the WPA IE so it can later be
433	 * inserted at the correct location.
434	 */
435	wpa_ie = NULL;
436	wpa_ie_len = 0;
437	if (wpa_s->wpa_proto == WPA_PROTO_WPA) {
438		wpa_ie = os_memdup(wpa_s->sme.assoc_req_ie,
439				   wpa_s->sme.assoc_req_ie_len);
440		if (wpa_ie) {
441			wpa_dbg(wpa_s, MSG_DEBUG, "WPA: Storing WPA IE");
442
443			wpa_ie_len = wpa_s->sme.assoc_req_ie_len;
444			wpa_s->sme.assoc_req_ie_len = 0;
445		} else {
446			wpa_msg(wpa_s, MSG_WARNING, "WPA: Failed copy WPA IE");
447			wpas_connect_work_done(wpa_s);
448			return;
449		}
450	}
451
452#ifdef CONFIG_IEEE80211R
453	ie = wpa_bss_get_ie(bss, WLAN_EID_MOBILITY_DOMAIN);
454	if (ie && ie[1] >= MOBILITY_DOMAIN_ID_LEN)
455		md = ie + 2;
456	wpa_sm_set_ft_params(wpa_s->wpa, ie, ie ? 2 + ie[1] : 0);
457	if (md && (!wpa_key_mgmt_ft(ssid->key_mgmt) ||
458		   !wpa_key_mgmt_ft(wpa_s->key_mgmt)))
459		md = NULL;
460	if (md) {
461		/* Prepare for the next transition */
462		wpa_ft_prepare_auth_request(wpa_s->wpa, ie);
463	}
464
465	if (md) {
466		wpa_dbg(wpa_s, MSG_DEBUG, "SME: FT mobility domain %02x%02x",
467			md[0], md[1]);
468
469		if (wpa_s->sme.assoc_req_ie_len + 5 <
470		    sizeof(wpa_s->sme.assoc_req_ie)) {
471			struct rsn_mdie *mdie;
472			u8 *pos = wpa_s->sme.assoc_req_ie +
473				wpa_s->sme.assoc_req_ie_len;
474			*pos++ = WLAN_EID_MOBILITY_DOMAIN;
475			*pos++ = sizeof(*mdie);
476			mdie = (struct rsn_mdie *) pos;
477			os_memcpy(mdie->mobility_domain, md,
478				  MOBILITY_DOMAIN_ID_LEN);
479			mdie->ft_capab = md[MOBILITY_DOMAIN_ID_LEN];
480			wpa_s->sme.assoc_req_ie_len += 5;
481		}
482
483		if (wpa_s->sme.prev_bssid_set && wpa_s->sme.ft_used &&
484		    os_memcmp(md, wpa_s->sme.mobility_domain, 2) == 0 &&
485		    wpa_sm_has_ptk(wpa_s->wpa)) {
486			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying to use FT "
487				"over-the-air");
488			params.auth_alg = WPA_AUTH_ALG_FT;
489			params.ie = wpa_s->sme.ft_ies;
490			params.ie_len = wpa_s->sme.ft_ies_len;
491		}
492	}
493#endif /* CONFIG_IEEE80211R */
494
495#ifdef CONFIG_IEEE80211W
496	wpa_s->sme.mfp = wpas_get_ssid_pmf(wpa_s, ssid);
497	if (wpa_s->sme.mfp != NO_MGMT_FRAME_PROTECTION) {
498		const u8 *rsn = wpa_bss_get_ie(bss, WLAN_EID_RSN);
499		struct wpa_ie_data _ie;
500		if (rsn && wpa_parse_wpa_ie(rsn, 2 + rsn[1], &_ie) == 0 &&
501		    _ie.capabilities &
502		    (WPA_CAPABILITY_MFPC | WPA_CAPABILITY_MFPR)) {
503			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected AP supports "
504				"MFP: require MFP");
505			wpa_s->sme.mfp = MGMT_FRAME_PROTECTION_REQUIRED;
506		}
507	}
508#endif /* CONFIG_IEEE80211W */
509
510#ifdef CONFIG_P2P
511	if (wpa_s->global->p2p) {
512		u8 *pos;
513		size_t len;
514		int res;
515		pos = wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len;
516		len = sizeof(wpa_s->sme.assoc_req_ie) -
517			wpa_s->sme.assoc_req_ie_len;
518		res = wpas_p2p_assoc_req_ie(wpa_s, bss, pos, len,
519					    ssid->p2p_group);
520		if (res >= 0)
521			wpa_s->sme.assoc_req_ie_len += res;
522	}
523#endif /* CONFIG_P2P */
524
525#ifdef CONFIG_FST
526	if (wpa_s->fst_ies) {
527		int fst_ies_len = wpabuf_len(wpa_s->fst_ies);
528
529		if (wpa_s->sme.assoc_req_ie_len + fst_ies_len <=
530		    sizeof(wpa_s->sme.assoc_req_ie)) {
531			os_memcpy(wpa_s->sme.assoc_req_ie +
532				  wpa_s->sme.assoc_req_ie_len,
533				  wpabuf_head(wpa_s->fst_ies),
534				  fst_ies_len);
535			wpa_s->sme.assoc_req_ie_len += fst_ies_len;
536		}
537	}
538#endif /* CONFIG_FST */
539
540	sme_auth_handle_rrm(wpa_s, bss);
541
542	wpa_s->sme.assoc_req_ie_len += wpas_supp_op_class_ie(
543		wpa_s, ssid, bss->freq,
544		wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len,
545		sizeof(wpa_s->sme.assoc_req_ie) - wpa_s->sme.assoc_req_ie_len);
546
547	if (params.p2p)
548		wpa_drv_get_ext_capa(wpa_s, WPA_IF_P2P_CLIENT);
549	else
550		wpa_drv_get_ext_capa(wpa_s, WPA_IF_STATION);
551
552	ext_capab_len = wpas_build_ext_capab(wpa_s, ext_capab,
553					     sizeof(ext_capab));
554	if (ext_capab_len > 0) {
555		u8 *pos = wpa_s->sme.assoc_req_ie;
556		if (wpa_s->sme.assoc_req_ie_len > 0 && pos[0] == WLAN_EID_RSN)
557			pos += 2 + pos[1];
558		os_memmove(pos + ext_capab_len, pos,
559			   wpa_s->sme.assoc_req_ie_len -
560			   (pos - wpa_s->sme.assoc_req_ie));
561		wpa_s->sme.assoc_req_ie_len += ext_capab_len;
562		os_memcpy(pos, ext_capab, ext_capab_len);
563	}
564
565#ifdef CONFIG_HS20
566	if (is_hs20_network(wpa_s, ssid, bss)) {
567		struct wpabuf *hs20;
568
569		hs20 = wpabuf_alloc(20 + MAX_ROAMING_CONS_OI_LEN);
570		if (hs20) {
571			int pps_mo_id = hs20_get_pps_mo_id(wpa_s, ssid);
572			size_t len;
573
574			wpas_hs20_add_indication(hs20, pps_mo_id,
575						 get_hs20_version(bss));
576			wpas_hs20_add_roam_cons_sel(hs20, ssid);
577			len = sizeof(wpa_s->sme.assoc_req_ie) -
578				wpa_s->sme.assoc_req_ie_len;
579			if (wpabuf_len(hs20) <= len) {
580				os_memcpy(wpa_s->sme.assoc_req_ie +
581					  wpa_s->sme.assoc_req_ie_len,
582					  wpabuf_head(hs20), wpabuf_len(hs20));
583				wpa_s->sme.assoc_req_ie_len += wpabuf_len(hs20);
584			}
585			wpabuf_free(hs20);
586		}
587	}
588#endif /* CONFIG_HS20 */
589
590	if (wpa_ie) {
591		size_t len;
592
593		wpa_dbg(wpa_s, MSG_DEBUG, "WPA: Reinsert WPA IE");
594
595		len = sizeof(wpa_s->sme.assoc_req_ie) -
596			wpa_s->sme.assoc_req_ie_len;
597
598		if (len > wpa_ie_len) {
599			os_memcpy(wpa_s->sme.assoc_req_ie +
600				  wpa_s->sme.assoc_req_ie_len,
601				  wpa_ie, wpa_ie_len);
602			wpa_s->sme.assoc_req_ie_len += wpa_ie_len;
603		} else {
604			wpa_dbg(wpa_s, MSG_DEBUG, "WPA: Failed to add WPA IE");
605		}
606
607		os_free(wpa_ie);
608	}
609
610	if (wpa_s->vendor_elem[VENDOR_ELEM_ASSOC_REQ]) {
611		struct wpabuf *buf = wpa_s->vendor_elem[VENDOR_ELEM_ASSOC_REQ];
612		size_t len;
613
614		len = sizeof(wpa_s->sme.assoc_req_ie) -
615			wpa_s->sme.assoc_req_ie_len;
616		if (wpabuf_len(buf) <= len) {
617			os_memcpy(wpa_s->sme.assoc_req_ie +
618				  wpa_s->sme.assoc_req_ie_len,
619				  wpabuf_head(buf), wpabuf_len(buf));
620			wpa_s->sme.assoc_req_ie_len += wpabuf_len(buf);
621		}
622	}
623
624#ifdef CONFIG_MBO
625	mbo_ie = wpa_bss_get_vendor_ie(bss, MBO_IE_VENDOR_TYPE);
626	if (mbo_ie) {
627		int len;
628
629		len = wpas_mbo_ie(wpa_s, wpa_s->sme.assoc_req_ie +
630				  wpa_s->sme.assoc_req_ie_len,
631				  sizeof(wpa_s->sme.assoc_req_ie) -
632				  wpa_s->sme.assoc_req_ie_len,
633				  !!mbo_attr_from_mbo_ie(mbo_ie,
634							 OCE_ATTR_ID_CAPA_IND));
635		if (len >= 0)
636			wpa_s->sme.assoc_req_ie_len += len;
637	}
638#endif /* CONFIG_MBO */
639
640#ifdef CONFIG_SAE
641	if (!skip_auth && params.auth_alg == WPA_AUTH_ALG_SAE &&
642	    pmksa_cache_set_current(wpa_s->wpa, NULL, bss->bssid, ssid, 0,
643				    NULL,
644				    wpa_s->key_mgmt == WPA_KEY_MGMT_FT_SAE ?
645				    WPA_KEY_MGMT_FT_SAE :
646				    WPA_KEY_MGMT_SAE) == 0) {
647		wpa_dbg(wpa_s, MSG_DEBUG,
648			"PMKSA cache entry found - try to use PMKSA caching instead of new SAE authentication");
649		wpa_sm_set_pmk_from_pmksa(wpa_s->wpa);
650		params.auth_alg = WPA_AUTH_ALG_OPEN;
651		wpa_s->sme.sae_pmksa_caching = 1;
652	}
653
654	if (!skip_auth && params.auth_alg == WPA_AUTH_ALG_SAE) {
655		if (start)
656			resp = sme_auth_build_sae_commit(wpa_s, ssid,
657							 bss->bssid, 0,
658							 start == 2);
659		else
660			resp = sme_auth_build_sae_confirm(wpa_s, 0);
661		if (resp == NULL) {
662			wpas_connection_failed(wpa_s, bss->bssid);
663			return;
664		}
665		params.auth_data = wpabuf_head(resp);
666		params.auth_data_len = wpabuf_len(resp);
667		wpa_s->sme.sae.state = start ? SAE_COMMITTED : SAE_CONFIRMED;
668	}
669#endif /* CONFIG_SAE */
670
671	old_ssid = wpa_s->current_ssid;
672	wpa_s->current_ssid = ssid;
673	wpa_supplicant_rsn_supp_set_config(wpa_s, wpa_s->current_ssid);
674	wpa_supplicant_initiate_eapol(wpa_s);
675
676#ifdef CONFIG_FILS
677	/* TODO: FILS operations can in some cases be done between different
678	 * network_ctx (i.e., same credentials can be used with multiple
679	 * networks). */
680	if (params.auth_alg == WPA_AUTH_ALG_OPEN &&
681	    wpa_key_mgmt_fils(ssid->key_mgmt)) {
682		const u8 *indic;
683		u16 fils_info;
684		const u8 *realm, *username, *rrk;
685		size_t realm_len, username_len, rrk_len;
686		u16 next_seq_num;
687
688		/*
689		 * Check FILS Indication element (FILS Information field) bits
690		 * indicating supported authentication algorithms against local
691		 * configuration (ssid->fils_dh_group). Try to use FILS
692		 * authentication only if the AP supports the combination in the
693		 * network profile. */
694		indic = wpa_bss_get_ie(bss, WLAN_EID_FILS_INDICATION);
695		if (!indic || indic[1] < 2) {
696			wpa_printf(MSG_DEBUG, "SME: " MACSTR
697				   " does not include FILS Indication element - cannot use FILS authentication with it",
698				   MAC2STR(bss->bssid));
699			goto no_fils;
700		}
701
702		fils_info = WPA_GET_LE16(indic + 2);
703		if (ssid->fils_dh_group == 0 && !(fils_info & BIT(9))) {
704			wpa_printf(MSG_DEBUG, "SME: " MACSTR
705				   " does not support FILS SK without PFS - cannot use FILS authentication with it",
706				   MAC2STR(bss->bssid));
707			goto no_fils;
708		}
709		if (ssid->fils_dh_group != 0 && !(fils_info & BIT(10))) {
710			wpa_printf(MSG_DEBUG, "SME: " MACSTR
711				   " does not support FILS SK with PFS - cannot use FILS authentication with it",
712				   MAC2STR(bss->bssid));
713			goto no_fils;
714		}
715
716		if (wpa_s->last_con_fail_realm &&
717		    eapol_sm_get_erp_info(wpa_s->eapol, &ssid->eap,
718					  &username, &username_len,
719					  &realm, &realm_len, &next_seq_num,
720					  &rrk, &rrk_len) == 0 &&
721		    realm && realm_len == wpa_s->last_con_fail_realm_len &&
722		    os_memcmp(realm, wpa_s->last_con_fail_realm,
723			      realm_len) == 0) {
724			wpa_printf(MSG_DEBUG,
725				   "SME: FILS authentication for this realm failed last time - try to regenerate ERP key hierarchy");
726			goto no_fils;
727		}
728
729		if (pmksa_cache_set_current(wpa_s->wpa, NULL, bss->bssid,
730					    ssid, 0,
731					    wpa_bss_get_fils_cache_id(bss),
732					    0) == 0)
733			wpa_printf(MSG_DEBUG,
734				   "SME: Try to use FILS with PMKSA caching");
735		resp = fils_build_auth(wpa_s->wpa, ssid->fils_dh_group, md);
736		if (resp) {
737			int auth_alg;
738
739			if (ssid->fils_dh_group)
740				wpa_printf(MSG_DEBUG,
741					   "SME: Try to use FILS SK authentication with PFS (DH Group %u)",
742					   ssid->fils_dh_group);
743			else
744				wpa_printf(MSG_DEBUG,
745					   "SME: Try to use FILS SK authentication without PFS");
746			auth_alg = ssid->fils_dh_group ?
747				WPA_AUTH_ALG_FILS_SK_PFS : WPA_AUTH_ALG_FILS;
748			params.auth_alg = auth_alg;
749			params.auth_data = wpabuf_head(resp);
750			params.auth_data_len = wpabuf_len(resp);
751			wpa_s->sme.auth_alg = auth_alg;
752		}
753	}
754no_fils:
755#endif /* CONFIG_FILS */
756
757	wpa_supplicant_cancel_sched_scan(wpa_s);
758	wpa_supplicant_cancel_scan(wpa_s);
759
760	wpa_msg(wpa_s, MSG_INFO, "SME: Trying to authenticate with " MACSTR
761		" (SSID='%s' freq=%d MHz)", MAC2STR(params.bssid),
762		wpa_ssid_txt(params.ssid, params.ssid_len), params.freq);
763
764	eapol_sm_notify_portValid(wpa_s->eapol, FALSE);
765	wpa_clear_keys(wpa_s, bss->bssid);
766	wpa_supplicant_set_state(wpa_s, WPA_AUTHENTICATING);
767	if (old_ssid != wpa_s->current_ssid)
768		wpas_notify_network_changed(wpa_s);
769
770#ifdef CONFIG_HS20
771	hs20_configure_frame_filters(wpa_s);
772#endif /* CONFIG_HS20 */
773
774#ifdef CONFIG_P2P
775	/*
776	 * If multi-channel concurrency is not supported, check for any
777	 * frequency conflict. In case of any frequency conflict, remove the
778	 * least prioritized connection.
779	 */
780	if (wpa_s->num_multichan_concurrent < 2) {
781		int freq, num;
782		num = get_shared_radio_freqs(wpa_s, &freq, 1);
783		if (num > 0 && freq > 0 && freq != params.freq) {
784			wpa_printf(MSG_DEBUG,
785				   "Conflicting frequency found (%d != %d)",
786				   freq, params.freq);
787			if (wpas_p2p_handle_frequency_conflicts(wpa_s,
788								params.freq,
789								ssid) < 0) {
790				wpas_connection_failed(wpa_s, bss->bssid);
791				wpa_supplicant_mark_disassoc(wpa_s);
792				wpabuf_free(resp);
793				wpas_connect_work_done(wpa_s);
794				return;
795			}
796		}
797	}
798#endif /* CONFIG_P2P */
799
800	if (skip_auth) {
801		wpa_msg(wpa_s, MSG_DEBUG,
802			"SME: Skip authentication step on reassoc-to-same-BSS");
803		wpabuf_free(resp);
804		sme_associate(wpa_s, ssid->mode, bss->bssid, WLAN_AUTH_OPEN);
805		return;
806	}
807
808
809	wpa_s->sme.auth_alg = params.auth_alg;
810	if (wpa_drv_authenticate(wpa_s, &params) < 0) {
811		wpa_msg(wpa_s, MSG_INFO, "SME: Authentication request to the "
812			"driver failed");
813		wpas_connection_failed(wpa_s, bss->bssid);
814		wpa_supplicant_mark_disassoc(wpa_s);
815		wpabuf_free(resp);
816		wpas_connect_work_done(wpa_s);
817		return;
818	}
819
820	eloop_register_timeout(SME_AUTH_TIMEOUT, 0, sme_auth_timer, wpa_s,
821			       NULL);
822
823	/*
824	 * Association will be started based on the authentication event from
825	 * the driver.
826	 */
827
828	wpabuf_free(resp);
829}
830
831
832static void sme_auth_start_cb(struct wpa_radio_work *work, int deinit)
833{
834	struct wpa_connect_work *cwork = work->ctx;
835	struct wpa_supplicant *wpa_s = work->wpa_s;
836
837	if (deinit) {
838		if (work->started)
839			wpa_s->connect_work = NULL;
840
841		wpas_connect_work_free(cwork);
842		return;
843	}
844
845	wpa_s->connect_work = work;
846
847	if (cwork->bss_removed ||
848	    !wpas_valid_bss_ssid(wpa_s, cwork->bss, cwork->ssid) ||
849	    wpas_network_disabled(wpa_s, cwork->ssid)) {
850		wpa_dbg(wpa_s, MSG_DEBUG, "SME: BSS/SSID entry for authentication not valid anymore - drop connection attempt");
851		wpas_connect_work_done(wpa_s);
852		return;
853	}
854
855	/* Starting new connection, so clear the possibly used WPA IE from the
856	 * previous association. */
857	wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, NULL, 0);
858
859	sme_send_authentication(wpa_s, cwork->bss, cwork->ssid, 1);
860}
861
862
863void sme_authenticate(struct wpa_supplicant *wpa_s,
864		      struct wpa_bss *bss, struct wpa_ssid *ssid)
865{
866	struct wpa_connect_work *cwork;
867
868	if (bss == NULL || ssid == NULL)
869		return;
870	if (wpa_s->connect_work) {
871		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Reject sme_authenticate() call since connect_work exist");
872		return;
873	}
874
875	if (radio_work_pending(wpa_s, "sme-connect")) {
876		/*
877		 * The previous sme-connect work might no longer be valid due to
878		 * the fact that the BSS list was updated. In addition, it makes
879		 * sense to adhere to the 'newer' decision.
880		 */
881		wpa_dbg(wpa_s, MSG_DEBUG,
882			"SME: Remove previous pending sme-connect");
883		radio_remove_works(wpa_s, "sme-connect", 0);
884	}
885
886	wpas_abort_ongoing_scan(wpa_s);
887
888	cwork = os_zalloc(sizeof(*cwork));
889	if (cwork == NULL)
890		return;
891	cwork->bss = bss;
892	cwork->ssid = ssid;
893	cwork->sme = 1;
894
895#ifdef CONFIG_SAE
896	wpa_s->sme.sae.state = SAE_NOTHING;
897	wpa_s->sme.sae.send_confirm = 0;
898	wpa_s->sme.sae_group_index = 0;
899#endif /* CONFIG_SAE */
900
901	if (radio_add_work(wpa_s, bss->freq, "sme-connect", 1,
902			   sme_auth_start_cb, cwork) < 0)
903		wpas_connect_work_free(cwork);
904}
905
906
907#ifdef CONFIG_SAE
908
909static int sme_external_auth_build_buf(struct wpabuf *buf,
910				       struct wpabuf *params,
911				       const u8 *sa, const u8 *da,
912				       u16 auth_transaction, u16 seq_num)
913{
914	struct ieee80211_mgmt *resp;
915
916	resp = wpabuf_put(buf, offsetof(struct ieee80211_mgmt,
917					u.auth.variable));
918
919	resp->frame_control = host_to_le16((WLAN_FC_TYPE_MGMT << 2) |
920					   (WLAN_FC_STYPE_AUTH << 4));
921	os_memcpy(resp->da, da, ETH_ALEN);
922	os_memcpy(resp->sa, sa, ETH_ALEN);
923	os_memcpy(resp->bssid, da, ETH_ALEN);
924	resp->u.auth.auth_alg = host_to_le16(WLAN_AUTH_SAE);
925	resp->seq_ctrl = host_to_le16(seq_num << 4);
926	resp->u.auth.auth_transaction = host_to_le16(auth_transaction);
927	resp->u.auth.status_code = host_to_le16(WLAN_STATUS_SUCCESS);
928	if (params)
929		wpabuf_put_buf(buf, params);
930
931	return 0;
932}
933
934
935static int sme_external_auth_send_sae_commit(struct wpa_supplicant *wpa_s,
936					     const u8 *bssid,
937					     struct wpa_ssid *ssid)
938{
939	struct wpabuf *resp, *buf;
940
941	resp = sme_auth_build_sae_commit(wpa_s, ssid, bssid, 1, 0);
942	if (!resp) {
943		wpa_printf(MSG_DEBUG, "SAE: Failed to build SAE commit");
944		return -1;
945	}
946
947	wpa_s->sme.sae.state = SAE_COMMITTED;
948	buf = wpabuf_alloc(4 + SAE_COMMIT_MAX_LEN + wpabuf_len(resp));
949	if (!buf) {
950		wpabuf_free(resp);
951		return -1;
952	}
953
954	wpa_s->sme.seq_num++;
955	sme_external_auth_build_buf(buf, resp, wpa_s->own_addr,
956				    bssid, 1, wpa_s->sme.seq_num);
957	wpa_drv_send_mlme(wpa_s, wpabuf_head(buf), wpabuf_len(buf), 1, 0);
958	wpabuf_free(resp);
959	wpabuf_free(buf);
960
961	return 0;
962}
963
964
965static void sme_send_external_auth_status(struct wpa_supplicant *wpa_s,
966					  u16 status)
967{
968	struct external_auth params;
969
970	os_memset(&params, 0, sizeof(params));
971	params.status = status;
972	params.ssid = wpa_s->sme.ext_auth_ssid;
973	params.ssid_len = wpa_s->sme.ext_auth_ssid_len;
974	params.bssid = wpa_s->sme.ext_auth_bssid;
975	wpa_drv_send_external_auth_status(wpa_s, &params);
976}
977
978
979static int sme_handle_external_auth_start(struct wpa_supplicant *wpa_s,
980					  union wpa_event_data *data)
981{
982	struct wpa_ssid *ssid;
983	size_t ssid_str_len = data->external_auth.ssid_len;
984	const u8 *ssid_str = data->external_auth.ssid;
985
986	/* Get the SSID conf from the ssid string obtained */
987	for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
988		if (!wpas_network_disabled(wpa_s, ssid) &&
989		    ssid_str_len == ssid->ssid_len &&
990		    os_memcmp(ssid_str, ssid->ssid, ssid_str_len) == 0 &&
991		    (ssid->key_mgmt & (WPA_KEY_MGMT_SAE | WPA_KEY_MGMT_FT_SAE)))
992			break;
993	}
994	if (!ssid ||
995	    sme_external_auth_send_sae_commit(wpa_s, data->external_auth.bssid,
996					      ssid) < 0)
997		return -1;
998
999	return 0;
1000}
1001
1002
1003static void sme_external_auth_send_sae_confirm(struct wpa_supplicant *wpa_s,
1004					       const u8 *da)
1005{
1006	struct wpabuf *resp, *buf;
1007
1008	resp = sme_auth_build_sae_confirm(wpa_s, 1);
1009	if (!resp) {
1010		wpa_printf(MSG_DEBUG, "SAE: Confirm message buf alloc failure");
1011		return;
1012	}
1013
1014	wpa_s->sme.sae.state = SAE_CONFIRMED;
1015	buf = wpabuf_alloc(4 + SAE_CONFIRM_MAX_LEN + wpabuf_len(resp));
1016	if (!buf) {
1017		wpa_printf(MSG_DEBUG, "SAE: Auth Confirm buf alloc failure");
1018		wpabuf_free(resp);
1019		return;
1020	}
1021	wpa_s->sme.seq_num++;
1022	sme_external_auth_build_buf(buf, resp, wpa_s->own_addr,
1023				    da, 2, wpa_s->sme.seq_num);
1024	wpa_drv_send_mlme(wpa_s, wpabuf_head(buf), wpabuf_len(buf), 1, 0);
1025	wpabuf_free(resp);
1026	wpabuf_free(buf);
1027}
1028
1029
1030void sme_external_auth_trigger(struct wpa_supplicant *wpa_s,
1031			       union wpa_event_data *data)
1032{
1033	if (RSN_SELECTOR_GET(&data->external_auth.key_mgmt_suite) !=
1034	    RSN_AUTH_KEY_MGMT_SAE)
1035		return;
1036
1037	if (data->external_auth.action == EXT_AUTH_START) {
1038		if (!data->external_auth.bssid || !data->external_auth.ssid)
1039			return;
1040		os_memcpy(wpa_s->sme.ext_auth_bssid, data->external_auth.bssid,
1041			  ETH_ALEN);
1042		os_memcpy(wpa_s->sme.ext_auth_ssid, data->external_auth.ssid,
1043			  data->external_auth.ssid_len);
1044		wpa_s->sme.ext_auth_ssid_len = data->external_auth.ssid_len;
1045		wpa_s->sme.seq_num = 0;
1046		wpa_s->sme.sae.state = SAE_NOTHING;
1047		wpa_s->sme.sae.send_confirm = 0;
1048		wpa_s->sme.sae_group_index = 0;
1049		if (sme_handle_external_auth_start(wpa_s, data) < 0)
1050			sme_send_external_auth_status(wpa_s,
1051					      WLAN_STATUS_UNSPECIFIED_FAILURE);
1052	} else if (data->external_auth.action == EXT_AUTH_ABORT) {
1053		/* Report failure to driver for the wrong trigger */
1054		sme_send_external_auth_status(wpa_s,
1055					      WLAN_STATUS_UNSPECIFIED_FAILURE);
1056	}
1057}
1058
1059
1060static int sme_sae_auth(struct wpa_supplicant *wpa_s, u16 auth_transaction,
1061			u16 status_code, const u8 *data, size_t len,
1062			int external, const u8 *sa)
1063{
1064	int *groups;
1065
1066	wpa_dbg(wpa_s, MSG_DEBUG, "SME: SAE authentication transaction %u "
1067		"status code %u", auth_transaction, status_code);
1068
1069	if (auth_transaction == 1 &&
1070	    status_code == WLAN_STATUS_ANTI_CLOGGING_TOKEN_REQ &&
1071	    wpa_s->sme.sae.state == SAE_COMMITTED &&
1072	    (external || wpa_s->current_bss) && wpa_s->current_ssid) {
1073		int default_groups[] = { 19, 20, 21, 0 };
1074		u16 group;
1075
1076		groups = wpa_s->conf->sae_groups;
1077		if (!groups || groups[0] <= 0)
1078			groups = default_groups;
1079
1080		if (len < sizeof(le16)) {
1081			wpa_dbg(wpa_s, MSG_DEBUG,
1082				"SME: Too short SAE anti-clogging token request");
1083			return -1;
1084		}
1085		group = WPA_GET_LE16(data);
1086		wpa_dbg(wpa_s, MSG_DEBUG,
1087			"SME: SAE anti-clogging token requested (group %u)",
1088			group);
1089		if (sae_group_allowed(&wpa_s->sme.sae, groups, group) !=
1090		    WLAN_STATUS_SUCCESS) {
1091			wpa_dbg(wpa_s, MSG_ERROR,
1092				"SME: SAE group %u of anti-clogging request is invalid",
1093				group);
1094			return -1;
1095		}
1096		wpabuf_free(wpa_s->sme.sae_token);
1097		wpa_s->sme.sae_token = wpabuf_alloc_copy(data + sizeof(le16),
1098							 len - sizeof(le16));
1099		if (!external)
1100			sme_send_authentication(wpa_s, wpa_s->current_bss,
1101						wpa_s->current_ssid, 2);
1102		else
1103			sme_external_auth_send_sae_commit(
1104				wpa_s, wpa_s->sme.ext_auth_bssid,
1105				wpa_s->current_ssid);
1106		return 0;
1107	}
1108
1109	if (auth_transaction == 1 &&
1110	    status_code == WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED &&
1111	    wpa_s->sme.sae.state == SAE_COMMITTED &&
1112	    (external || wpa_s->current_bss) && wpa_s->current_ssid) {
1113		wpa_dbg(wpa_s, MSG_DEBUG, "SME: SAE group not supported");
1114		wpa_s->sme.sae_group_index++;
1115		if (sme_set_sae_group(wpa_s) < 0)
1116			return -1; /* no other groups enabled */
1117		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Try next enabled SAE group");
1118		if (!external)
1119			sme_send_authentication(wpa_s, wpa_s->current_bss,
1120						wpa_s->current_ssid, 1);
1121		else
1122			sme_external_auth_send_sae_commit(
1123				wpa_s, wpa_s->sme.ext_auth_bssid,
1124				wpa_s->current_ssid);
1125		return 0;
1126	}
1127
1128	if (auth_transaction == 1 &&
1129	    status_code == WLAN_STATUS_UNKNOWN_PASSWORD_IDENTIFIER) {
1130		const u8 *bssid = sa ? sa : wpa_s->pending_bssid;
1131
1132		wpa_msg(wpa_s, MSG_INFO,
1133			WPA_EVENT_SAE_UNKNOWN_PASSWORD_IDENTIFIER MACSTR,
1134			MAC2STR(bssid));
1135		return -1;
1136	}
1137
1138	if (status_code != WLAN_STATUS_SUCCESS)
1139		return -1;
1140
1141	if (auth_transaction == 1) {
1142		u16 res;
1143
1144		groups = wpa_s->conf->sae_groups;
1145
1146		wpa_dbg(wpa_s, MSG_DEBUG, "SME SAE commit");
1147		if ((!external && wpa_s->current_bss == NULL) ||
1148		    wpa_s->current_ssid == NULL)
1149			return -1;
1150		if (wpa_s->sme.sae.state != SAE_COMMITTED)
1151			return -1;
1152		if (groups && groups[0] <= 0)
1153			groups = NULL;
1154		res = sae_parse_commit(&wpa_s->sme.sae, data, len, NULL, NULL,
1155				       groups);
1156		if (res == SAE_SILENTLY_DISCARD) {
1157			wpa_printf(MSG_DEBUG,
1158				   "SAE: Drop commit message due to reflection attack");
1159			return 0;
1160		}
1161		if (res != WLAN_STATUS_SUCCESS)
1162			return -1;
1163
1164		if (sae_process_commit(&wpa_s->sme.sae) < 0) {
1165			wpa_printf(MSG_DEBUG, "SAE: Failed to process peer "
1166				   "commit");
1167			return -1;
1168		}
1169
1170		wpabuf_free(wpa_s->sme.sae_token);
1171		wpa_s->sme.sae_token = NULL;
1172		if (!external)
1173			sme_send_authentication(wpa_s, wpa_s->current_bss,
1174						wpa_s->current_ssid, 0);
1175		else
1176			sme_external_auth_send_sae_confirm(wpa_s, sa);
1177		return 0;
1178	} else if (auth_transaction == 2) {
1179		wpa_dbg(wpa_s, MSG_DEBUG, "SME SAE confirm");
1180		if (wpa_s->sme.sae.state != SAE_CONFIRMED)
1181			return -1;
1182		if (sae_check_confirm(&wpa_s->sme.sae, data, len) < 0)
1183			return -1;
1184		wpa_s->sme.sae.state = SAE_ACCEPTED;
1185		sae_clear_temp_data(&wpa_s->sme.sae);
1186
1187		if (external) {
1188			/* Report success to driver */
1189			sme_send_external_auth_status(wpa_s,
1190						      WLAN_STATUS_SUCCESS);
1191		}
1192
1193		return 1;
1194	}
1195
1196	return -1;
1197}
1198
1199
1200void sme_external_auth_mgmt_rx(struct wpa_supplicant *wpa_s,
1201			       const u8 *auth_frame, size_t len)
1202{
1203	const struct ieee80211_mgmt *header;
1204	size_t auth_length;
1205
1206	header = (const struct ieee80211_mgmt *) auth_frame;
1207	auth_length = IEEE80211_HDRLEN + sizeof(header->u.auth);
1208
1209	if (len < auth_length) {
1210		/* Notify failure to the driver */
1211		sme_send_external_auth_status(wpa_s,
1212					      WLAN_STATUS_UNSPECIFIED_FAILURE);
1213		return;
1214	}
1215
1216	if (le_to_host16(header->u.auth.auth_alg) == WLAN_AUTH_SAE) {
1217		int res;
1218
1219		res = sme_sae_auth(
1220			wpa_s, le_to_host16(header->u.auth.auth_transaction),
1221			le_to_host16(header->u.auth.status_code),
1222			header->u.auth.variable,
1223			len - auth_length, 1, header->sa);
1224		if (res < 0) {
1225			/* Notify failure to the driver */
1226			sme_send_external_auth_status(
1227				wpa_s, WLAN_STATUS_UNSPECIFIED_FAILURE);
1228			return;
1229		}
1230		if (res != 1)
1231			return;
1232
1233		wpa_printf(MSG_DEBUG,
1234			   "SME: SAE completed - setting PMK for 4-way handshake");
1235		wpa_sm_set_pmk(wpa_s->wpa, wpa_s->sme.sae.pmk, PMK_LEN,
1236			       wpa_s->sme.sae.pmkid, wpa_s->pending_bssid);
1237	}
1238}
1239
1240#endif /* CONFIG_SAE */
1241
1242
1243void sme_event_auth(struct wpa_supplicant *wpa_s, union wpa_event_data *data)
1244{
1245	struct wpa_ssid *ssid = wpa_s->current_ssid;
1246
1247	if (ssid == NULL) {
1248		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication event "
1249			"when network is not selected");
1250		return;
1251	}
1252
1253	if (wpa_s->wpa_state != WPA_AUTHENTICATING) {
1254		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication event "
1255			"when not in authenticating state");
1256		return;
1257	}
1258
1259	if (os_memcmp(wpa_s->pending_bssid, data->auth.peer, ETH_ALEN) != 0) {
1260		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Ignore authentication with "
1261			"unexpected peer " MACSTR,
1262			MAC2STR(data->auth.peer));
1263		return;
1264	}
1265
1266	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Authentication response: peer=" MACSTR
1267		" auth_type=%d auth_transaction=%d status_code=%d",
1268		MAC2STR(data->auth.peer), data->auth.auth_type,
1269		data->auth.auth_transaction, data->auth.status_code);
1270	wpa_hexdump(MSG_MSGDUMP, "SME: Authentication response IEs",
1271		    data->auth.ies, data->auth.ies_len);
1272
1273	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1274
1275#ifdef CONFIG_SAE
1276	if (data->auth.auth_type == WLAN_AUTH_SAE) {
1277		int res;
1278		res = sme_sae_auth(wpa_s, data->auth.auth_transaction,
1279				   data->auth.status_code, data->auth.ies,
1280				   data->auth.ies_len, 0, NULL);
1281		if (res < 0) {
1282			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1283			wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
1284
1285		}
1286		if (res != 1)
1287			return;
1288
1289		wpa_printf(MSG_DEBUG, "SME: SAE completed - setting PMK for "
1290			   "4-way handshake");
1291		wpa_sm_set_pmk(wpa_s->wpa, wpa_s->sme.sae.pmk, PMK_LEN,
1292			       wpa_s->sme.sae.pmkid, wpa_s->pending_bssid);
1293	}
1294#endif /* CONFIG_SAE */
1295
1296	if (data->auth.status_code != WLAN_STATUS_SUCCESS) {
1297		char *ie_txt = NULL;
1298
1299		if (data->auth.ies && data->auth.ies_len) {
1300			size_t buflen = 2 * data->auth.ies_len + 1;
1301			ie_txt = os_malloc(buflen);
1302			if (ie_txt) {
1303				wpa_snprintf_hex(ie_txt, buflen, data->auth.ies,
1304						 data->auth.ies_len);
1305			}
1306		}
1307		wpa_msg(wpa_s, MSG_INFO, WPA_EVENT_AUTH_REJECT MACSTR
1308			" auth_type=%u auth_transaction=%u status_code=%u%s%s",
1309			MAC2STR(data->auth.peer), data->auth.auth_type,
1310			data->auth.auth_transaction, data->auth.status_code,
1311			ie_txt ? " ie=" : "",
1312			ie_txt ? ie_txt : "");
1313		os_free(ie_txt);
1314
1315#ifdef CONFIG_FILS
1316		if (wpa_s->sme.auth_alg == WPA_AUTH_ALG_FILS ||
1317		    wpa_s->sme.auth_alg == WPA_AUTH_ALG_FILS_SK_PFS)
1318			fils_connection_failure(wpa_s);
1319#endif /* CONFIG_FILS */
1320
1321		if (data->auth.status_code !=
1322		    WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG ||
1323		    wpa_s->sme.auth_alg == data->auth.auth_type ||
1324		    wpa_s->current_ssid->auth_alg == WPA_AUTH_ALG_LEAP) {
1325			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1326			wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
1327			return;
1328		}
1329
1330		wpas_connect_work_done(wpa_s);
1331
1332		switch (data->auth.auth_type) {
1333		case WLAN_AUTH_OPEN:
1334			wpa_s->current_ssid->auth_alg = WPA_AUTH_ALG_SHARED;
1335
1336			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying SHARED auth");
1337			wpa_supplicant_associate(wpa_s, wpa_s->current_bss,
1338						 wpa_s->current_ssid);
1339			return;
1340
1341		case WLAN_AUTH_SHARED_KEY:
1342			wpa_s->current_ssid->auth_alg = WPA_AUTH_ALG_LEAP;
1343
1344			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Trying LEAP auth");
1345			wpa_supplicant_associate(wpa_s, wpa_s->current_bss,
1346						 wpa_s->current_ssid);
1347			return;
1348
1349		default:
1350			return;
1351		}
1352	}
1353
1354#ifdef CONFIG_IEEE80211R
1355	if (data->auth.auth_type == WLAN_AUTH_FT) {
1356		const u8 *ric_ies = NULL;
1357		size_t ric_ies_len = 0;
1358
1359		if (wpa_s->ric_ies) {
1360			ric_ies = wpabuf_head(wpa_s->ric_ies);
1361			ric_ies_len = wpabuf_len(wpa_s->ric_ies);
1362		}
1363		if (wpa_ft_process_response(wpa_s->wpa, data->auth.ies,
1364					    data->auth.ies_len, 0,
1365					    data->auth.peer,
1366					    ric_ies, ric_ies_len) < 0) {
1367			wpa_dbg(wpa_s, MSG_DEBUG,
1368				"SME: FT Authentication response processing failed");
1369			wpa_msg(wpa_s, MSG_INFO, WPA_EVENT_DISCONNECTED "bssid="
1370				MACSTR
1371				" reason=%d locally_generated=1",
1372				MAC2STR(wpa_s->pending_bssid),
1373				WLAN_REASON_DEAUTH_LEAVING);
1374			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1375			wpa_supplicant_mark_disassoc(wpa_s);
1376			return;
1377		}
1378	}
1379#endif /* CONFIG_IEEE80211R */
1380
1381#ifdef CONFIG_FILS
1382	if (data->auth.auth_type == WLAN_AUTH_FILS_SK ||
1383	    data->auth.auth_type == WLAN_AUTH_FILS_SK_PFS) {
1384		u16 expect_auth_type;
1385
1386		expect_auth_type = wpa_s->sme.auth_alg ==
1387			WPA_AUTH_ALG_FILS_SK_PFS ? WLAN_AUTH_FILS_SK_PFS :
1388			WLAN_AUTH_FILS_SK;
1389		if (data->auth.auth_type != expect_auth_type) {
1390			wpa_dbg(wpa_s, MSG_DEBUG,
1391				"SME: FILS Authentication response used different auth alg (%u; expected %u)",
1392				data->auth.auth_type, expect_auth_type);
1393			wpa_msg(wpa_s, MSG_INFO, WPA_EVENT_DISCONNECTED "bssid="
1394				MACSTR
1395				" reason=%d locally_generated=1",
1396				MAC2STR(wpa_s->pending_bssid),
1397				WLAN_REASON_DEAUTH_LEAVING);
1398			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1399			wpa_supplicant_mark_disassoc(wpa_s);
1400			return;
1401		}
1402
1403		if (fils_process_auth(wpa_s->wpa, wpa_s->pending_bssid,
1404				      data->auth.ies, data->auth.ies_len) < 0) {
1405			wpa_dbg(wpa_s, MSG_DEBUG,
1406				"SME: FILS Authentication response processing failed");
1407			wpa_msg(wpa_s, MSG_INFO, WPA_EVENT_DISCONNECTED "bssid="
1408				MACSTR
1409				" reason=%d locally_generated=1",
1410				MAC2STR(wpa_s->pending_bssid),
1411				WLAN_REASON_DEAUTH_LEAVING);
1412			wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1413			wpa_supplicant_mark_disassoc(wpa_s);
1414			return;
1415		}
1416	}
1417#endif /* CONFIG_FILS */
1418
1419	sme_associate(wpa_s, ssid->mode, data->auth.peer,
1420		      data->auth.auth_type);
1421}
1422
1423
1424#ifdef CONFIG_IEEE80211R
1425static void remove_ie(u8 *buf, size_t *len, u8 eid)
1426{
1427	u8 *pos, *next, *end;
1428
1429	pos = (u8 *) get_ie(buf, *len, eid);
1430	if (pos) {
1431		next = pos + 2 + pos[1];
1432		end = buf + *len;
1433		*len -= 2 + pos[1];
1434		os_memmove(pos, next, end - next);
1435	}
1436}
1437#endif /* CONFIG_IEEE80211R */
1438
1439
1440void sme_associate(struct wpa_supplicant *wpa_s, enum wpas_mode mode,
1441		   const u8 *bssid, u16 auth_type)
1442{
1443	struct wpa_driver_associate_params params;
1444	struct ieee802_11_elems elems;
1445#ifdef CONFIG_FILS
1446	u8 nonces[2 * FILS_NONCE_LEN];
1447#endif /* CONFIG_FILS */
1448#ifdef CONFIG_HT_OVERRIDES
1449	struct ieee80211_ht_capabilities htcaps;
1450	struct ieee80211_ht_capabilities htcaps_mask;
1451#endif /* CONFIG_HT_OVERRIDES */
1452#ifdef CONFIG_VHT_OVERRIDES
1453	struct ieee80211_vht_capabilities vhtcaps;
1454	struct ieee80211_vht_capabilities vhtcaps_mask;
1455#endif /* CONFIG_VHT_OVERRIDES */
1456
1457	os_memset(&params, 0, sizeof(params));
1458
1459#ifdef CONFIG_FILS
1460	if (auth_type == WLAN_AUTH_FILS_SK ||
1461	    auth_type == WLAN_AUTH_FILS_SK_PFS) {
1462		struct wpabuf *buf;
1463		const u8 *snonce, *anonce;
1464		const unsigned int max_hlp = 20;
1465		struct wpabuf *hlp[max_hlp];
1466		unsigned int i, num_hlp = 0;
1467		struct fils_hlp_req *req;
1468
1469		dl_list_for_each(req, &wpa_s->fils_hlp_req, struct fils_hlp_req,
1470				 list) {
1471			hlp[num_hlp] = wpabuf_alloc(2 * ETH_ALEN + 6 +
1472					      wpabuf_len(req->pkt));
1473			if (!hlp[num_hlp])
1474				break;
1475			wpabuf_put_data(hlp[num_hlp], req->dst, ETH_ALEN);
1476			wpabuf_put_data(hlp[num_hlp], wpa_s->own_addr,
1477					ETH_ALEN);
1478			wpabuf_put_data(hlp[num_hlp],
1479					"\xaa\xaa\x03\x00\x00\x00", 6);
1480			wpabuf_put_buf(hlp[num_hlp], req->pkt);
1481			num_hlp++;
1482			if (num_hlp >= max_hlp)
1483				break;
1484		}
1485
1486		buf = fils_build_assoc_req(wpa_s->wpa, &params.fils_kek,
1487					   &params.fils_kek_len, &snonce,
1488					   &anonce,
1489					   (const struct wpabuf **) hlp,
1490					   num_hlp);
1491		for (i = 0; i < num_hlp; i++)
1492			wpabuf_free(hlp[i]);
1493		if (!buf)
1494			return;
1495		wpa_hexdump(MSG_DEBUG, "FILS: assoc_req before FILS elements",
1496			    wpa_s->sme.assoc_req_ie,
1497			    wpa_s->sme.assoc_req_ie_len);
1498#ifdef CONFIG_IEEE80211R
1499		if (wpa_key_mgmt_ft(wpa_s->key_mgmt)) {
1500			/* Remove RSNE and MDE to allow them to be overridden
1501			 * with FILS+FT specific values from
1502			 * fils_build_assoc_req(). */
1503			remove_ie(wpa_s->sme.assoc_req_ie,
1504				  &wpa_s->sme.assoc_req_ie_len,
1505				  WLAN_EID_RSN);
1506			wpa_hexdump(MSG_DEBUG,
1507				    "FILS: assoc_req after RSNE removal",
1508				    wpa_s->sme.assoc_req_ie,
1509				    wpa_s->sme.assoc_req_ie_len);
1510			remove_ie(wpa_s->sme.assoc_req_ie,
1511				  &wpa_s->sme.assoc_req_ie_len,
1512				  WLAN_EID_MOBILITY_DOMAIN);
1513			wpa_hexdump(MSG_DEBUG,
1514				    "FILS: assoc_req after MDE removal",
1515				    wpa_s->sme.assoc_req_ie,
1516				    wpa_s->sme.assoc_req_ie_len);
1517		}
1518#endif /* CONFIG_IEEE80211R */
1519		/* TODO: Make wpa_s->sme.assoc_req_ie use dynamic allocation */
1520		if (wpa_s->sme.assoc_req_ie_len + wpabuf_len(buf) >
1521		    sizeof(wpa_s->sme.assoc_req_ie)) {
1522			wpa_printf(MSG_ERROR,
1523				   "FILS: Not enough buffer room for own AssocReq elements");
1524			wpabuf_free(buf);
1525			return;
1526		}
1527		os_memcpy(wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len,
1528			  wpabuf_head(buf), wpabuf_len(buf));
1529		wpa_s->sme.assoc_req_ie_len += wpabuf_len(buf);
1530		wpabuf_free(buf);
1531		wpa_hexdump(MSG_DEBUG, "FILS: assoc_req after FILS elements",
1532			    wpa_s->sme.assoc_req_ie,
1533			    wpa_s->sme.assoc_req_ie_len);
1534
1535		os_memcpy(nonces, snonce, FILS_NONCE_LEN);
1536		os_memcpy(nonces + FILS_NONCE_LEN, anonce, FILS_NONCE_LEN);
1537		params.fils_nonces = nonces;
1538		params.fils_nonces_len = sizeof(nonces);
1539	}
1540#endif /* CONFIG_FILS */
1541
1542#ifdef CONFIG_OWE
1543#ifdef CONFIG_TESTING_OPTIONS
1544	if (get_ie_ext(wpa_s->sme.assoc_req_ie, wpa_s->sme.assoc_req_ie_len,
1545		       WLAN_EID_EXT_OWE_DH_PARAM)) {
1546		wpa_printf(MSG_INFO, "TESTING: Override OWE DH element");
1547	} else
1548#endif /* CONFIG_TESTING_OPTIONS */
1549	if (auth_type == WLAN_AUTH_OPEN &&
1550	    wpa_s->key_mgmt == WPA_KEY_MGMT_OWE) {
1551		struct wpabuf *owe_ie;
1552		u16 group;
1553
1554		if (wpa_s->current_ssid && wpa_s->current_ssid->owe_group) {
1555			group = wpa_s->current_ssid->owe_group;
1556		} else if (wpa_s->assoc_status_code ==
1557			   WLAN_STATUS_FINITE_CYCLIC_GROUP_NOT_SUPPORTED) {
1558			if (wpa_s->last_owe_group == 19)
1559				group = 20;
1560			else if (wpa_s->last_owe_group == 20)
1561				group = 21;
1562			else
1563				group = OWE_DH_GROUP;
1564		} else {
1565			group = OWE_DH_GROUP;
1566		}
1567
1568		wpa_s->last_owe_group = group;
1569		wpa_printf(MSG_DEBUG, "OWE: Try to use group %u", group);
1570		owe_ie = owe_build_assoc_req(wpa_s->wpa, group);
1571		if (!owe_ie) {
1572			wpa_printf(MSG_ERROR,
1573				   "OWE: Failed to build IE for Association Request frame");
1574			return;
1575		}
1576		if (wpa_s->sme.assoc_req_ie_len + wpabuf_len(owe_ie) >
1577		    sizeof(wpa_s->sme.assoc_req_ie)) {
1578			wpa_printf(MSG_ERROR,
1579				   "OWE: Not enough buffer room for own Association Request frame elements");
1580			wpabuf_free(owe_ie);
1581			return;
1582		}
1583		os_memcpy(wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len,
1584			  wpabuf_head(owe_ie), wpabuf_len(owe_ie));
1585		wpa_s->sme.assoc_req_ie_len += wpabuf_len(owe_ie);
1586		wpabuf_free(owe_ie);
1587	}
1588#endif /* CONFIG_OWE */
1589
1590#ifdef CONFIG_DPP2
1591	if (wpa_s->key_mgmt == WPA_KEY_MGMT_DPP && wpa_s->current_ssid &&
1592	    wpa_s->current_ssid->dpp_netaccesskey) {
1593		struct wpa_ssid *ssid = wpa_s->current_ssid;
1594
1595		dpp_pfs_free(wpa_s->dpp_pfs);
1596		wpa_s->dpp_pfs = dpp_pfs_init(ssid->dpp_netaccesskey,
1597					      ssid->dpp_netaccesskey_len);
1598		if (!wpa_s->dpp_pfs) {
1599			wpa_printf(MSG_DEBUG, "DPP: Could not initialize PFS");
1600			/* Try to continue without PFS */
1601			goto pfs_fail;
1602		}
1603		if (wpa_s->sme.assoc_req_ie_len +
1604		    wpabuf_len(wpa_s->dpp_pfs->ie) >
1605		    sizeof(wpa_s->sme.assoc_req_ie)) {
1606			wpa_printf(MSG_ERROR,
1607				   "DPP: Not enough buffer room for own Association Request frame elements");
1608			dpp_pfs_free(wpa_s->dpp_pfs);
1609			wpa_s->dpp_pfs = NULL;
1610			goto pfs_fail;
1611		}
1612		os_memcpy(wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len,
1613			  wpabuf_head(wpa_s->dpp_pfs->ie),
1614			  wpabuf_len(wpa_s->dpp_pfs->ie));
1615		wpa_s->sme.assoc_req_ie_len += wpabuf_len(wpa_s->dpp_pfs->ie);
1616	}
1617pfs_fail:
1618#endif /* CONFIG_DPP2 */
1619
1620	if (wpa_s->current_ssid && wpa_s->current_ssid->multi_ap_backhaul_sta) {
1621		size_t multi_ap_ie_len;
1622
1623		multi_ap_ie_len = add_multi_ap_ie(
1624			wpa_s->sme.assoc_req_ie + wpa_s->sme.assoc_req_ie_len,
1625			sizeof(wpa_s->sme.assoc_req_ie) -
1626			wpa_s->sme.assoc_req_ie_len,
1627			MULTI_AP_BACKHAUL_STA);
1628		if (multi_ap_ie_len == 0) {
1629			wpa_printf(MSG_ERROR,
1630				   "Multi-AP: Failed to build Multi-AP IE");
1631			return;
1632		}
1633		wpa_s->sme.assoc_req_ie_len += multi_ap_ie_len;
1634	}
1635
1636	params.bssid = bssid;
1637	params.ssid = wpa_s->sme.ssid;
1638	params.ssid_len = wpa_s->sme.ssid_len;
1639	params.freq.freq = wpa_s->sme.freq;
1640	params.bg_scan_period = wpa_s->current_ssid ?
1641		wpa_s->current_ssid->bg_scan_period : -1;
1642	params.wpa_ie = wpa_s->sme.assoc_req_ie_len ?
1643		wpa_s->sme.assoc_req_ie : NULL;
1644	params.wpa_ie_len = wpa_s->sme.assoc_req_ie_len;
1645	wpa_hexdump(MSG_DEBUG, "SME: Association Request IEs",
1646		    params.wpa_ie, params.wpa_ie_len);
1647	params.pairwise_suite = wpa_s->pairwise_cipher;
1648	params.group_suite = wpa_s->group_cipher;
1649	params.mgmt_group_suite = wpa_s->mgmt_group_cipher;
1650	params.key_mgmt_suite = wpa_s->key_mgmt;
1651	params.wpa_proto = wpa_s->wpa_proto;
1652#ifdef CONFIG_HT_OVERRIDES
1653	os_memset(&htcaps, 0, sizeof(htcaps));
1654	os_memset(&htcaps_mask, 0, sizeof(htcaps_mask));
1655	params.htcaps = (u8 *) &htcaps;
1656	params.htcaps_mask = (u8 *) &htcaps_mask;
1657	wpa_supplicant_apply_ht_overrides(wpa_s, wpa_s->current_ssid, &params);
1658#endif /* CONFIG_HT_OVERRIDES */
1659#ifdef CONFIG_VHT_OVERRIDES
1660	os_memset(&vhtcaps, 0, sizeof(vhtcaps));
1661	os_memset(&vhtcaps_mask, 0, sizeof(vhtcaps_mask));
1662	params.vhtcaps = &vhtcaps;
1663	params.vhtcaps_mask = &vhtcaps_mask;
1664	wpa_supplicant_apply_vht_overrides(wpa_s, wpa_s->current_ssid, &params);
1665#endif /* CONFIG_VHT_OVERRIDES */
1666#ifdef CONFIG_IEEE80211R
1667	if (auth_type == WLAN_AUTH_FT && wpa_s->sme.ft_ies &&
1668	    get_ie(wpa_s->sme.ft_ies, wpa_s->sme.ft_ies_len,
1669		   WLAN_EID_RIC_DATA)) {
1670		/* There seems to be a pretty inconvenient bug in the Linux
1671		 * kernel IE splitting functionality when RIC is used. For now,
1672		 * skip correct behavior in IE construction here (i.e., drop the
1673		 * additional non-FT-specific IEs) to avoid kernel issues. This
1674		 * is fine since RIC is used only for testing purposes in the
1675		 * current implementation. */
1676		wpa_printf(MSG_INFO,
1677			   "SME: Linux kernel workaround - do not try to include additional IEs with RIC");
1678		params.wpa_ie = wpa_s->sme.ft_ies;
1679		params.wpa_ie_len = wpa_s->sme.ft_ies_len;
1680	} else if (auth_type == WLAN_AUTH_FT && wpa_s->sme.ft_ies) {
1681		const u8 *rm_en, *pos, *end;
1682		size_t rm_en_len = 0;
1683		u8 *rm_en_dup = NULL, *wpos;
1684
1685		/* Remove RSNE, MDE, FTE to allow them to be overridden with
1686		 * FT specific values */
1687		remove_ie(wpa_s->sme.assoc_req_ie,
1688			  &wpa_s->sme.assoc_req_ie_len,
1689			  WLAN_EID_RSN);
1690		remove_ie(wpa_s->sme.assoc_req_ie,
1691			  &wpa_s->sme.assoc_req_ie_len,
1692			  WLAN_EID_MOBILITY_DOMAIN);
1693		remove_ie(wpa_s->sme.assoc_req_ie,
1694			  &wpa_s->sme.assoc_req_ie_len,
1695			  WLAN_EID_FAST_BSS_TRANSITION);
1696		rm_en = get_ie(wpa_s->sme.assoc_req_ie,
1697			       wpa_s->sme.assoc_req_ie_len,
1698			       WLAN_EID_RRM_ENABLED_CAPABILITIES);
1699		if (rm_en) {
1700			/* Need to remove RM Enabled Capabilities element as
1701			 * well temporarily, so that it can be placed between
1702			 * RSNE and MDE. */
1703			rm_en_len = 2 + rm_en[1];
1704			rm_en_dup = os_memdup(rm_en, rm_en_len);
1705			remove_ie(wpa_s->sme.assoc_req_ie,
1706				  &wpa_s->sme.assoc_req_ie_len,
1707				  WLAN_EID_RRM_ENABLED_CAPABILITIES);
1708		}
1709		wpa_hexdump(MSG_DEBUG,
1710			    "SME: Association Request IEs after FT IE removal",
1711			    wpa_s->sme.assoc_req_ie,
1712			    wpa_s->sme.assoc_req_ie_len);
1713		if (wpa_s->sme.assoc_req_ie_len + wpa_s->sme.ft_ies_len +
1714		    rm_en_len > sizeof(wpa_s->sme.assoc_req_ie)) {
1715			wpa_printf(MSG_ERROR,
1716				   "SME: Not enough buffer room for FT IEs in Association Request frame");
1717			os_free(rm_en_dup);
1718			return;
1719		}
1720
1721		os_memmove(wpa_s->sme.assoc_req_ie + wpa_s->sme.ft_ies_len +
1722			   rm_en_len,
1723			   wpa_s->sme.assoc_req_ie,
1724			   wpa_s->sme.assoc_req_ie_len);
1725		pos = wpa_s->sme.ft_ies;
1726		end = pos + wpa_s->sme.ft_ies_len;
1727		wpos = wpa_s->sme.assoc_req_ie;
1728		if (*pos == WLAN_EID_RSN) {
1729			os_memcpy(wpos, pos, 2 + pos[1]);
1730			wpos += 2 + pos[1];
1731			pos += 2 + pos[1];
1732		}
1733		if (rm_en_dup) {
1734			os_memcpy(wpos, rm_en_dup, rm_en_len);
1735			wpos += rm_en_len;
1736			os_free(rm_en_dup);
1737		}
1738		os_memcpy(wpos, pos, end - pos);
1739		wpa_s->sme.assoc_req_ie_len += wpa_s->sme.ft_ies_len +
1740			rm_en_len;
1741		params.wpa_ie = wpa_s->sme.assoc_req_ie;
1742		params.wpa_ie_len = wpa_s->sme.assoc_req_ie_len;
1743		wpa_hexdump(MSG_DEBUG,
1744			    "SME: Association Request IEs after FT override",
1745			    params.wpa_ie, params.wpa_ie_len);
1746	}
1747#endif /* CONFIG_IEEE80211R */
1748	params.mode = mode;
1749	params.mgmt_frame_protection = wpa_s->sme.mfp;
1750	params.rrm_used = wpa_s->rrm.rrm_used;
1751	if (wpa_s->sme.prev_bssid_set)
1752		params.prev_bssid = wpa_s->sme.prev_bssid;
1753
1754	wpa_msg(wpa_s, MSG_INFO, "Trying to associate with " MACSTR
1755		" (SSID='%s' freq=%d MHz)", MAC2STR(params.bssid),
1756		params.ssid ? wpa_ssid_txt(params.ssid, params.ssid_len) : "",
1757		params.freq.freq);
1758
1759	wpa_supplicant_set_state(wpa_s, WPA_ASSOCIATING);
1760
1761	if (params.wpa_ie == NULL ||
1762	    ieee802_11_parse_elems(params.wpa_ie, params.wpa_ie_len, &elems, 0)
1763	    < 0) {
1764		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Could not parse own IEs?!");
1765		os_memset(&elems, 0, sizeof(elems));
1766	}
1767	if (elems.rsn_ie) {
1768		params.wpa_proto = WPA_PROTO_RSN;
1769		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.rsn_ie - 2,
1770					elems.rsn_ie_len + 2);
1771	} else if (elems.wpa_ie) {
1772		params.wpa_proto = WPA_PROTO_WPA;
1773		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.wpa_ie - 2,
1774					elems.wpa_ie_len + 2);
1775	} else if (elems.osen) {
1776		params.wpa_proto = WPA_PROTO_OSEN;
1777		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, elems.osen - 2,
1778					elems.osen_len + 2);
1779	} else
1780		wpa_sm_set_assoc_wpa_ie(wpa_s->wpa, NULL, 0);
1781	if (wpa_s->current_ssid && wpa_s->current_ssid->p2p_group)
1782		params.p2p = 1;
1783
1784	if (wpa_s->p2pdev->set_sta_uapsd)
1785		params.uapsd = wpa_s->p2pdev->sta_uapsd;
1786	else
1787		params.uapsd = -1;
1788
1789	if (wpa_drv_associate(wpa_s, &params) < 0) {
1790		wpa_msg(wpa_s, MSG_INFO, "SME: Association request to the "
1791			"driver failed");
1792		wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1793		wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
1794		os_memset(wpa_s->pending_bssid, 0, ETH_ALEN);
1795		return;
1796	}
1797
1798	eloop_register_timeout(SME_ASSOC_TIMEOUT, 0, sme_assoc_timer, wpa_s,
1799			       NULL);
1800
1801#ifdef CONFIG_TESTING_OPTIONS
1802	wpabuf_free(wpa_s->last_assoc_req_wpa_ie);
1803	wpa_s->last_assoc_req_wpa_ie = NULL;
1804	if (params.wpa_ie)
1805		wpa_s->last_assoc_req_wpa_ie =
1806			wpabuf_alloc_copy(params.wpa_ie, params.wpa_ie_len);
1807#endif /* CONFIG_TESTING_OPTIONS */
1808}
1809
1810
1811int sme_update_ft_ies(struct wpa_supplicant *wpa_s, const u8 *md,
1812		      const u8 *ies, size_t ies_len)
1813{
1814	if (md == NULL || ies == NULL) {
1815		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Remove mobility domain");
1816		os_free(wpa_s->sme.ft_ies);
1817		wpa_s->sme.ft_ies = NULL;
1818		wpa_s->sme.ft_ies_len = 0;
1819		wpa_s->sme.ft_used = 0;
1820		return 0;
1821	}
1822
1823	os_memcpy(wpa_s->sme.mobility_domain, md, MOBILITY_DOMAIN_ID_LEN);
1824	wpa_hexdump(MSG_DEBUG, "SME: FT IEs", ies, ies_len);
1825	os_free(wpa_s->sme.ft_ies);
1826	wpa_s->sme.ft_ies = os_memdup(ies, ies_len);
1827	if (wpa_s->sme.ft_ies == NULL)
1828		return -1;
1829	wpa_s->sme.ft_ies_len = ies_len;
1830	return 0;
1831}
1832
1833
1834static void sme_deauth(struct wpa_supplicant *wpa_s)
1835{
1836	int bssid_changed;
1837
1838	bssid_changed = !is_zero_ether_addr(wpa_s->bssid);
1839
1840	if (wpa_drv_deauthenticate(wpa_s, wpa_s->pending_bssid,
1841				   WLAN_REASON_DEAUTH_LEAVING) < 0) {
1842		wpa_msg(wpa_s, MSG_INFO, "SME: Deauth request to the driver "
1843			"failed");
1844	}
1845	wpa_s->sme.prev_bssid_set = 0;
1846
1847	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1848	wpa_supplicant_set_state(wpa_s, WPA_DISCONNECTED);
1849	os_memset(wpa_s->bssid, 0, ETH_ALEN);
1850	os_memset(wpa_s->pending_bssid, 0, ETH_ALEN);
1851	if (bssid_changed)
1852		wpas_notify_bssid_changed(wpa_s);
1853}
1854
1855
1856void sme_event_assoc_reject(struct wpa_supplicant *wpa_s,
1857			    union wpa_event_data *data)
1858{
1859	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association with " MACSTR " failed: "
1860		"status code %d", MAC2STR(wpa_s->pending_bssid),
1861		data->assoc_reject.status_code);
1862
1863	eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
1864
1865#ifdef CONFIG_SAE
1866	if (wpa_s->sme.sae_pmksa_caching && wpa_s->current_ssid &&
1867	    wpa_key_mgmt_sae(wpa_s->current_ssid->key_mgmt)) {
1868		wpa_dbg(wpa_s, MSG_DEBUG,
1869			"PMKSA caching attempt rejected - drop PMKSA cache entry and fall back to SAE authentication");
1870		wpa_sm_aborted_cached(wpa_s->wpa);
1871		wpa_sm_pmksa_cache_flush(wpa_s->wpa, wpa_s->current_ssid);
1872		if (wpa_s->current_bss) {
1873			struct wpa_bss *bss = wpa_s->current_bss;
1874			struct wpa_ssid *ssid = wpa_s->current_ssid;
1875
1876			wpa_drv_deauthenticate(wpa_s, wpa_s->pending_bssid,
1877					       WLAN_REASON_DEAUTH_LEAVING);
1878			wpas_connect_work_done(wpa_s);
1879			wpa_supplicant_mark_disassoc(wpa_s);
1880			wpa_supplicant_connect(wpa_s, bss, ssid);
1881			return;
1882		}
1883	}
1884#endif /* CONFIG_SAE */
1885
1886	/*
1887	 * For now, unconditionally terminate the previous authentication. In
1888	 * theory, this should not be needed, but mac80211 gets quite confused
1889	 * if the authentication is left pending.. Some roaming cases might
1890	 * benefit from using the previous authentication, so this could be
1891	 * optimized in the future.
1892	 */
1893	sme_deauth(wpa_s);
1894}
1895
1896
1897void sme_event_auth_timed_out(struct wpa_supplicant *wpa_s,
1898			      union wpa_event_data *data)
1899{
1900	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Authentication timed out");
1901	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1902	wpa_supplicant_mark_disassoc(wpa_s);
1903}
1904
1905
1906void sme_event_assoc_timed_out(struct wpa_supplicant *wpa_s,
1907			       union wpa_event_data *data)
1908{
1909	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association timed out");
1910	wpas_connection_failed(wpa_s, wpa_s->pending_bssid);
1911	wpa_supplicant_mark_disassoc(wpa_s);
1912}
1913
1914
1915void sme_event_disassoc(struct wpa_supplicant *wpa_s,
1916			struct disassoc_info *info)
1917{
1918	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Disassociation event received");
1919	if (wpa_s->sme.prev_bssid_set) {
1920		/*
1921		 * cfg80211/mac80211 can get into somewhat confused state if
1922		 * the AP only disassociates us and leaves us in authenticated
1923		 * state. For now, force the state to be cleared to avoid
1924		 * confusing errors if we try to associate with the AP again.
1925		 */
1926		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Deauthenticate to clear "
1927			"driver state");
1928		wpa_drv_deauthenticate(wpa_s, wpa_s->sme.prev_bssid,
1929				       WLAN_REASON_DEAUTH_LEAVING);
1930	}
1931}
1932
1933
1934static void sme_auth_timer(void *eloop_ctx, void *timeout_ctx)
1935{
1936	struct wpa_supplicant *wpa_s = eloop_ctx;
1937	if (wpa_s->wpa_state == WPA_AUTHENTICATING) {
1938		wpa_msg(wpa_s, MSG_DEBUG, "SME: Authentication timeout");
1939		sme_deauth(wpa_s);
1940	}
1941}
1942
1943
1944static void sme_assoc_timer(void *eloop_ctx, void *timeout_ctx)
1945{
1946	struct wpa_supplicant *wpa_s = eloop_ctx;
1947	if (wpa_s->wpa_state == WPA_ASSOCIATING) {
1948		wpa_msg(wpa_s, MSG_DEBUG, "SME: Association timeout");
1949		sme_deauth(wpa_s);
1950	}
1951}
1952
1953
1954void sme_state_changed(struct wpa_supplicant *wpa_s)
1955{
1956	/* Make sure timers are cleaned up appropriately. */
1957	if (wpa_s->wpa_state != WPA_ASSOCIATING)
1958		eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
1959	if (wpa_s->wpa_state != WPA_AUTHENTICATING)
1960		eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1961}
1962
1963
1964void sme_disassoc_while_authenticating(struct wpa_supplicant *wpa_s,
1965				       const u8 *prev_pending_bssid)
1966{
1967	/*
1968	 * mac80211-workaround to force deauth on failed auth cmd,
1969	 * requires us to remain in authenticating state to allow the
1970	 * second authentication attempt to be continued properly.
1971	 */
1972	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Allow pending authentication "
1973		"to proceed after disconnection event");
1974	wpa_supplicant_set_state(wpa_s, WPA_AUTHENTICATING);
1975	os_memcpy(wpa_s->pending_bssid, prev_pending_bssid, ETH_ALEN);
1976
1977	/*
1978	 * Re-arm authentication timer in case auth fails for whatever reason.
1979	 */
1980	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
1981	eloop_register_timeout(SME_AUTH_TIMEOUT, 0, sme_auth_timer, wpa_s,
1982			       NULL);
1983}
1984
1985
1986void sme_clear_on_disassoc(struct wpa_supplicant *wpa_s)
1987{
1988	wpa_s->sme.prev_bssid_set = 0;
1989#ifdef CONFIG_SAE
1990	wpabuf_free(wpa_s->sme.sae_token);
1991	wpa_s->sme.sae_token = NULL;
1992	sae_clear_data(&wpa_s->sme.sae);
1993#endif /* CONFIG_SAE */
1994#ifdef CONFIG_IEEE80211R
1995	if (wpa_s->sme.ft_ies || wpa_s->sme.ft_used)
1996		sme_update_ft_ies(wpa_s, NULL, NULL, 0);
1997#endif /* CONFIG_IEEE80211R */
1998#ifdef CONFIG_IEEE80211W
1999	sme_stop_sa_query(wpa_s);
2000#endif /* CONFIG_IEEE80211W */
2001}
2002
2003
2004void sme_deinit(struct wpa_supplicant *wpa_s)
2005{
2006	sme_clear_on_disassoc(wpa_s);
2007
2008	eloop_cancel_timeout(sme_assoc_timer, wpa_s, NULL);
2009	eloop_cancel_timeout(sme_auth_timer, wpa_s, NULL);
2010	eloop_cancel_timeout(sme_obss_scan_timeout, wpa_s, NULL);
2011}
2012
2013
2014static void sme_send_2040_bss_coex(struct wpa_supplicant *wpa_s,
2015				   const u8 *chan_list, u8 num_channels,
2016				   u8 num_intol)
2017{
2018	struct ieee80211_2040_bss_coex_ie *bc_ie;
2019	struct ieee80211_2040_intol_chan_report *ic_report;
2020	struct wpabuf *buf;
2021
2022	wpa_printf(MSG_DEBUG, "SME: Send 20/40 BSS Coexistence to " MACSTR
2023		   " (num_channels=%u num_intol=%u)",
2024		   MAC2STR(wpa_s->bssid), num_channels, num_intol);
2025	wpa_hexdump(MSG_DEBUG, "SME: 20/40 BSS Intolerant Channels",
2026		    chan_list, num_channels);
2027
2028	buf = wpabuf_alloc(2 + /* action.category + action_code */
2029			   sizeof(struct ieee80211_2040_bss_coex_ie) +
2030			   sizeof(struct ieee80211_2040_intol_chan_report) +
2031			   num_channels);
2032	if (buf == NULL)
2033		return;
2034
2035	wpabuf_put_u8(buf, WLAN_ACTION_PUBLIC);
2036	wpabuf_put_u8(buf, WLAN_PA_20_40_BSS_COEX);
2037
2038	bc_ie = wpabuf_put(buf, sizeof(*bc_ie));
2039	bc_ie->element_id = WLAN_EID_20_40_BSS_COEXISTENCE;
2040	bc_ie->length = 1;
2041	if (num_intol)
2042		bc_ie->coex_param |= WLAN_20_40_BSS_COEX_20MHZ_WIDTH_REQ;
2043
2044	if (num_channels > 0) {
2045		ic_report = wpabuf_put(buf, sizeof(*ic_report));
2046		ic_report->element_id = WLAN_EID_20_40_BSS_INTOLERANT;
2047		ic_report->length = num_channels + 1;
2048		ic_report->op_class = 0;
2049		os_memcpy(wpabuf_put(buf, num_channels), chan_list,
2050			  num_channels);
2051	}
2052
2053	if (wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
2054				wpa_s->own_addr, wpa_s->bssid,
2055				wpabuf_head(buf), wpabuf_len(buf), 0) < 0) {
2056		wpa_msg(wpa_s, MSG_INFO,
2057			"SME: Failed to send 20/40 BSS Coexistence frame");
2058	}
2059
2060	wpabuf_free(buf);
2061}
2062
2063
2064int sme_proc_obss_scan(struct wpa_supplicant *wpa_s)
2065{
2066	struct wpa_bss *bss;
2067	const u8 *ie;
2068	u16 ht_cap;
2069	u8 chan_list[P2P_MAX_CHANNELS], channel;
2070	u8 num_channels = 0, num_intol = 0, i;
2071
2072	if (!wpa_s->sme.sched_obss_scan)
2073		return 0;
2074
2075	wpa_s->sme.sched_obss_scan = 0;
2076	if (!wpa_s->current_bss || wpa_s->wpa_state != WPA_COMPLETED)
2077		return 1;
2078
2079	/*
2080	 * Check whether AP uses regulatory triplet or channel triplet in
2081	 * country info. Right now the operating class of the BSS channel
2082	 * width trigger event is "unknown" (IEEE Std 802.11-2012 10.15.12),
2083	 * based on the assumption that operating class triplet is not used in
2084	 * beacon frame. If the First Channel Number/Operating Extension
2085	 * Identifier octet has a positive integer value of 201 or greater,
2086	 * then its operating class triplet.
2087	 *
2088	 * TODO: If Supported Operating Classes element is present in beacon
2089	 * frame, have to lookup operating class in Annex E and fill them in
2090	 * 2040 coex frame.
2091	 */
2092	ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_COUNTRY);
2093	if (ie && (ie[1] >= 6) && (ie[5] >= 201))
2094		return 1;
2095
2096	os_memset(chan_list, 0, sizeof(chan_list));
2097
2098	dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
2099		/* Skip other band bss */
2100		enum hostapd_hw_mode mode;
2101		mode = ieee80211_freq_to_chan(bss->freq, &channel);
2102		if (mode != HOSTAPD_MODE_IEEE80211G &&
2103		    mode != HOSTAPD_MODE_IEEE80211B)
2104			continue;
2105
2106		ie = wpa_bss_get_ie(bss, WLAN_EID_HT_CAP);
2107		ht_cap = (ie && (ie[1] == 26)) ? WPA_GET_LE16(ie + 2) : 0;
2108		wpa_printf(MSG_DEBUG, "SME OBSS scan BSS " MACSTR
2109			   " freq=%u chan=%u ht_cap=0x%x",
2110			   MAC2STR(bss->bssid), bss->freq, channel, ht_cap);
2111
2112		if (!ht_cap || (ht_cap & HT_CAP_INFO_40MHZ_INTOLERANT)) {
2113			if (ht_cap & HT_CAP_INFO_40MHZ_INTOLERANT)
2114				num_intol++;
2115
2116			/* Check whether the channel is already considered */
2117			for (i = 0; i < num_channels; i++) {
2118				if (channel == chan_list[i])
2119					break;
2120			}
2121			if (i != num_channels)
2122				continue;
2123
2124			chan_list[num_channels++] = channel;
2125		}
2126	}
2127
2128	sme_send_2040_bss_coex(wpa_s, chan_list, num_channels, num_intol);
2129	return 1;
2130}
2131
2132
2133static void wpa_obss_scan_freqs_list(struct wpa_supplicant *wpa_s,
2134				     struct wpa_driver_scan_params *params)
2135{
2136	/* Include only affected channels */
2137	struct hostapd_hw_modes *mode;
2138	int count, i;
2139	int start, end;
2140
2141	mode = get_mode(wpa_s->hw.modes, wpa_s->hw.num_modes,
2142			HOSTAPD_MODE_IEEE80211G);
2143	if (mode == NULL) {
2144		/* No channels supported in this band - use empty list */
2145		params->freqs = os_zalloc(sizeof(int));
2146		return;
2147	}
2148
2149	if (wpa_s->sme.ht_sec_chan == HT_SEC_CHAN_UNKNOWN &&
2150	    wpa_s->current_bss) {
2151		const u8 *ie;
2152
2153		ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_HT_OPERATION);
2154		if (ie && ie[1] >= 2) {
2155			u8 o;
2156
2157			o = ie[3] & HT_INFO_HT_PARAM_SECONDARY_CHNL_OFF_MASK;
2158			if (o == HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE)
2159				wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_ABOVE;
2160			else if (o == HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW)
2161				wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_BELOW;
2162		}
2163	}
2164
2165	start = wpa_s->assoc_freq - 10;
2166	end = wpa_s->assoc_freq + 10;
2167	switch (wpa_s->sme.ht_sec_chan) {
2168	case HT_SEC_CHAN_UNKNOWN:
2169		/* HT40+ possible on channels 1..9 */
2170		if (wpa_s->assoc_freq <= 2452)
2171			start -= 20;
2172		/* HT40- possible on channels 5-13 */
2173		if (wpa_s->assoc_freq >= 2432)
2174			end += 20;
2175		break;
2176	case HT_SEC_CHAN_ABOVE:
2177		end += 20;
2178		break;
2179	case HT_SEC_CHAN_BELOW:
2180		start -= 20;
2181		break;
2182	}
2183	wpa_printf(MSG_DEBUG,
2184		   "OBSS: assoc_freq %d possible affected range %d-%d",
2185		   wpa_s->assoc_freq, start, end);
2186
2187	params->freqs = os_calloc(mode->num_channels + 1, sizeof(int));
2188	if (params->freqs == NULL)
2189		return;
2190	for (count = 0, i = 0; i < mode->num_channels; i++) {
2191		int freq;
2192
2193		if (mode->channels[i].flag & HOSTAPD_CHAN_DISABLED)
2194			continue;
2195		freq = mode->channels[i].freq;
2196		if (freq - 10 >= end || freq + 10 <= start)
2197			continue; /* not affected */
2198		params->freqs[count++] = freq;
2199	}
2200}
2201
2202
2203static void sme_obss_scan_timeout(void *eloop_ctx, void *timeout_ctx)
2204{
2205	struct wpa_supplicant *wpa_s = eloop_ctx;
2206	struct wpa_driver_scan_params params;
2207
2208	if (!wpa_s->current_bss) {
2209		wpa_printf(MSG_DEBUG, "SME OBSS: Ignore scan request");
2210		return;
2211	}
2212
2213	os_memset(&params, 0, sizeof(params));
2214	wpa_obss_scan_freqs_list(wpa_s, &params);
2215	params.low_priority = 1;
2216	wpa_printf(MSG_DEBUG, "SME OBSS: Request an OBSS scan");
2217
2218	if (wpa_supplicant_trigger_scan(wpa_s, &params))
2219		wpa_printf(MSG_DEBUG, "SME OBSS: Failed to trigger scan");
2220	else
2221		wpa_s->sme.sched_obss_scan = 1;
2222	os_free(params.freqs);
2223
2224	eloop_register_timeout(wpa_s->sme.obss_scan_int, 0,
2225			       sme_obss_scan_timeout, wpa_s, NULL);
2226}
2227
2228
2229void sme_sched_obss_scan(struct wpa_supplicant *wpa_s, int enable)
2230{
2231	const u8 *ie;
2232	struct wpa_bss *bss = wpa_s->current_bss;
2233	struct wpa_ssid *ssid = wpa_s->current_ssid;
2234	struct hostapd_hw_modes *hw_mode = NULL;
2235	int i;
2236
2237	eloop_cancel_timeout(sme_obss_scan_timeout, wpa_s, NULL);
2238	wpa_s->sme.sched_obss_scan = 0;
2239	wpa_s->sme.ht_sec_chan = HT_SEC_CHAN_UNKNOWN;
2240	if (!enable)
2241		return;
2242
2243	/*
2244	 * Schedule OBSS scan if driver is using station SME in wpa_supplicant
2245	 * or it expects OBSS scan to be performed by wpa_supplicant.
2246	 */
2247	if (!((wpa_s->drv_flags & WPA_DRIVER_FLAGS_SME) ||
2248	      (wpa_s->drv_flags & WPA_DRIVER_FLAGS_OBSS_SCAN)) ||
2249	    ssid == NULL || ssid->mode != WPAS_MODE_INFRA)
2250		return;
2251
2252	if (!wpa_s->hw.modes)
2253		return;
2254
2255	/* only HT caps in 11g mode are relevant */
2256	for (i = 0; i < wpa_s->hw.num_modes; i++) {
2257		hw_mode = &wpa_s->hw.modes[i];
2258		if (hw_mode->mode == HOSTAPD_MODE_IEEE80211G)
2259			break;
2260	}
2261
2262	/* Driver does not support HT40 for 11g or doesn't have 11g. */
2263	if (i == wpa_s->hw.num_modes || !hw_mode ||
2264	    !(hw_mode->ht_capab & HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
2265		return;
2266
2267	if (bss == NULL || bss->freq < 2400 || bss->freq > 2500)
2268		return; /* Not associated on 2.4 GHz band */
2269
2270	/* Check whether AP supports HT40 */
2271	ie = wpa_bss_get_ie(wpa_s->current_bss, WLAN_EID_HT_CAP);
2272	if (!ie || ie[1] < 2 ||
2273	    !(WPA_GET_LE16(ie + 2) & HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
2274		return; /* AP does not support HT40 */
2275
2276	ie = wpa_bss_get_ie(wpa_s->current_bss,
2277			    WLAN_EID_OVERLAPPING_BSS_SCAN_PARAMS);
2278	if (!ie || ie[1] < 14)
2279		return; /* AP does not request OBSS scans */
2280
2281	wpa_s->sme.obss_scan_int = WPA_GET_LE16(ie + 6);
2282	if (wpa_s->sme.obss_scan_int < 10) {
2283		wpa_printf(MSG_DEBUG, "SME: Invalid OBSS Scan Interval %u "
2284			   "replaced with the minimum 10 sec",
2285			   wpa_s->sme.obss_scan_int);
2286		wpa_s->sme.obss_scan_int = 10;
2287	}
2288	wpa_printf(MSG_DEBUG, "SME: OBSS Scan Interval %u sec",
2289		   wpa_s->sme.obss_scan_int);
2290	eloop_register_timeout(wpa_s->sme.obss_scan_int, 0,
2291			       sme_obss_scan_timeout, wpa_s, NULL);
2292}
2293
2294
2295#ifdef CONFIG_IEEE80211W
2296
2297static const unsigned int sa_query_max_timeout = 1000;
2298static const unsigned int sa_query_retry_timeout = 201;
2299static const unsigned int sa_query_ch_switch_max_delay = 5000; /* in usec */
2300
2301static int sme_check_sa_query_timeout(struct wpa_supplicant *wpa_s)
2302{
2303	u32 tu;
2304	struct os_reltime now, passed;
2305	os_get_reltime(&now);
2306	os_reltime_sub(&now, &wpa_s->sme.sa_query_start, &passed);
2307	tu = (passed.sec * 1000000 + passed.usec) / 1024;
2308	if (sa_query_max_timeout < tu) {
2309		wpa_dbg(wpa_s, MSG_DEBUG, "SME: SA Query timed out");
2310		sme_stop_sa_query(wpa_s);
2311		wpa_supplicant_deauthenticate(
2312			wpa_s, WLAN_REASON_PREV_AUTH_NOT_VALID);
2313		return 1;
2314	}
2315
2316	return 0;
2317}
2318
2319
2320static void sme_send_sa_query_req(struct wpa_supplicant *wpa_s,
2321				  const u8 *trans_id)
2322{
2323	u8 req[2 + WLAN_SA_QUERY_TR_ID_LEN + OCV_OCI_EXTENDED_LEN];
2324	u8 req_len = 2 + WLAN_SA_QUERY_TR_ID_LEN;
2325
2326	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Sending SA Query Request to "
2327		MACSTR, MAC2STR(wpa_s->bssid));
2328	wpa_hexdump(MSG_DEBUG, "SME: SA Query Transaction ID",
2329		    trans_id, WLAN_SA_QUERY_TR_ID_LEN);
2330	req[0] = WLAN_ACTION_SA_QUERY;
2331	req[1] = WLAN_SA_QUERY_REQUEST;
2332	os_memcpy(req + 2, trans_id, WLAN_SA_QUERY_TR_ID_LEN);
2333
2334#ifdef CONFIG_OCV
2335	if (wpa_sm_ocv_enabled(wpa_s->wpa)) {
2336		struct wpa_channel_info ci;
2337
2338		if (wpa_drv_channel_info(wpa_s, &ci) != 0) {
2339			wpa_printf(MSG_WARNING,
2340				   "Failed to get channel info for OCI element in SA Query Request frame");
2341			return;
2342		}
2343
2344		if (ocv_insert_extended_oci(&ci, req + req_len) < 0)
2345			return;
2346
2347		req_len += OCV_OCI_EXTENDED_LEN;
2348	}
2349#endif /* CONFIG_OCV */
2350
2351	if (wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
2352				wpa_s->own_addr, wpa_s->bssid,
2353				req, req_len, 0) < 0)
2354		wpa_msg(wpa_s, MSG_INFO, "SME: Failed to send SA Query "
2355			"Request");
2356}
2357
2358
2359static void sme_sa_query_timer(void *eloop_ctx, void *timeout_ctx)
2360{
2361	struct wpa_supplicant *wpa_s = eloop_ctx;
2362	unsigned int timeout, sec, usec;
2363	u8 *trans_id, *nbuf;
2364
2365	if (wpa_s->sme.sa_query_count > 0 &&
2366	    sme_check_sa_query_timeout(wpa_s))
2367		return;
2368
2369	nbuf = os_realloc_array(wpa_s->sme.sa_query_trans_id,
2370				wpa_s->sme.sa_query_count + 1,
2371				WLAN_SA_QUERY_TR_ID_LEN);
2372	if (nbuf == NULL) {
2373		sme_stop_sa_query(wpa_s);
2374		return;
2375	}
2376	if (wpa_s->sme.sa_query_count == 0) {
2377		/* Starting a new SA Query procedure */
2378		os_get_reltime(&wpa_s->sme.sa_query_start);
2379	}
2380	trans_id = nbuf + wpa_s->sme.sa_query_count * WLAN_SA_QUERY_TR_ID_LEN;
2381	wpa_s->sme.sa_query_trans_id = nbuf;
2382	wpa_s->sme.sa_query_count++;
2383
2384	if (os_get_random(trans_id, WLAN_SA_QUERY_TR_ID_LEN) < 0) {
2385		wpa_printf(MSG_DEBUG, "Could not generate SA Query ID");
2386		sme_stop_sa_query(wpa_s);
2387		return;
2388	}
2389
2390	timeout = sa_query_retry_timeout;
2391	sec = ((timeout / 1000) * 1024) / 1000;
2392	usec = (timeout % 1000) * 1024;
2393	eloop_register_timeout(sec, usec, sme_sa_query_timer, wpa_s, NULL);
2394
2395	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Association SA Query attempt %d",
2396		wpa_s->sme.sa_query_count);
2397
2398	sme_send_sa_query_req(wpa_s, trans_id);
2399}
2400
2401
2402static void sme_start_sa_query(struct wpa_supplicant *wpa_s)
2403{
2404	sme_sa_query_timer(wpa_s, NULL);
2405}
2406
2407
2408static void sme_stop_sa_query(struct wpa_supplicant *wpa_s)
2409{
2410	if (wpa_s->sme.sa_query_trans_id)
2411		wpa_dbg(wpa_s, MSG_DEBUG, "SME: Stop SA Query");
2412	eloop_cancel_timeout(sme_sa_query_timer, wpa_s, NULL);
2413	os_free(wpa_s->sme.sa_query_trans_id);
2414	wpa_s->sme.sa_query_trans_id = NULL;
2415	wpa_s->sme.sa_query_count = 0;
2416}
2417
2418
2419void sme_event_unprot_disconnect(struct wpa_supplicant *wpa_s, const u8 *sa,
2420				 const u8 *da, u16 reason_code)
2421{
2422	struct wpa_ssid *ssid;
2423	struct os_reltime now;
2424
2425	if (wpa_s->wpa_state != WPA_COMPLETED)
2426		return;
2427	ssid = wpa_s->current_ssid;
2428	if (wpas_get_ssid_pmf(wpa_s, ssid) == NO_MGMT_FRAME_PROTECTION)
2429		return;
2430	if (os_memcmp(sa, wpa_s->bssid, ETH_ALEN) != 0)
2431		return;
2432	if (reason_code != WLAN_REASON_CLASS2_FRAME_FROM_NONAUTH_STA &&
2433	    reason_code != WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA)
2434		return;
2435	if (wpa_s->sme.sa_query_count > 0)
2436		return;
2437
2438	os_get_reltime(&now);
2439	if (wpa_s->sme.last_unprot_disconnect.sec &&
2440	    !os_reltime_expired(&now, &wpa_s->sme.last_unprot_disconnect, 10))
2441		return; /* limit SA Query procedure frequency */
2442	wpa_s->sme.last_unprot_disconnect = now;
2443
2444	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Unprotected disconnect dropped - "
2445		"possible AP/STA state mismatch - trigger SA Query");
2446	sme_start_sa_query(wpa_s);
2447}
2448
2449
2450void sme_event_ch_switch(struct wpa_supplicant *wpa_s)
2451{
2452	unsigned int usec;
2453	u32 _rand;
2454
2455	if (wpa_s->wpa_state != WPA_COMPLETED ||
2456	    !wpa_sm_ocv_enabled(wpa_s->wpa))
2457		return;
2458
2459	wpa_dbg(wpa_s, MSG_DEBUG,
2460		"SME: Channel switch completed - trigger new SA Query to verify new operating channel");
2461	sme_stop_sa_query(wpa_s);
2462
2463	if (os_get_random((u8 *) &_rand, sizeof(_rand)) < 0)
2464		_rand = os_random();
2465	usec = _rand % (sa_query_ch_switch_max_delay + 1);
2466	eloop_register_timeout(0, usec, sme_sa_query_timer, wpa_s, NULL);
2467}
2468
2469
2470static void sme_process_sa_query_request(struct wpa_supplicant *wpa_s,
2471					 const u8 *sa, const u8 *data,
2472					 size_t len)
2473{
2474	u8 resp[2 + WLAN_SA_QUERY_TR_ID_LEN + OCV_OCI_EXTENDED_LEN];
2475	u8 resp_len = 2 + WLAN_SA_QUERY_TR_ID_LEN;
2476
2477	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Sending SA Query Response to "
2478		MACSTR, MAC2STR(wpa_s->bssid));
2479
2480	resp[0] = WLAN_ACTION_SA_QUERY;
2481	resp[1] = WLAN_SA_QUERY_RESPONSE;
2482	os_memcpy(resp + 2, data + 1, WLAN_SA_QUERY_TR_ID_LEN);
2483
2484#ifdef CONFIG_OCV
2485	if (wpa_sm_ocv_enabled(wpa_s->wpa)) {
2486		struct wpa_channel_info ci;
2487
2488		if (wpa_drv_channel_info(wpa_s, &ci) != 0) {
2489			wpa_printf(MSG_WARNING,
2490				   "Failed to get channel info for OCI element in SA Query Response frame");
2491			return;
2492		}
2493
2494		if (ocv_insert_extended_oci(&ci, resp + resp_len) < 0)
2495			return;
2496
2497		resp_len += OCV_OCI_EXTENDED_LEN;
2498	}
2499#endif /* CONFIG_OCV */
2500
2501	if (wpa_drv_send_action(wpa_s, wpa_s->assoc_freq, 0, wpa_s->bssid,
2502				wpa_s->own_addr, wpa_s->bssid,
2503				resp, resp_len, 0) < 0)
2504		wpa_msg(wpa_s, MSG_INFO,
2505			"SME: Failed to send SA Query Response");
2506}
2507
2508
2509static void sme_process_sa_query_response(struct wpa_supplicant *wpa_s,
2510					  const u8 *sa, const u8 *data,
2511					  size_t len)
2512{
2513	int i;
2514
2515	if (!wpa_s->sme.sa_query_trans_id)
2516		return;
2517
2518	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Received SA Query response from "
2519		MACSTR " (trans_id %02x%02x)", MAC2STR(sa), data[1], data[2]);
2520
2521	if (os_memcmp(sa, wpa_s->bssid, ETH_ALEN) != 0)
2522		return;
2523
2524	for (i = 0; i < wpa_s->sme.sa_query_count; i++) {
2525		if (os_memcmp(wpa_s->sme.sa_query_trans_id +
2526			      i * WLAN_SA_QUERY_TR_ID_LEN,
2527			      data + 1, WLAN_SA_QUERY_TR_ID_LEN) == 0)
2528			break;
2529	}
2530
2531	if (i >= wpa_s->sme.sa_query_count) {
2532		wpa_dbg(wpa_s, MSG_DEBUG, "SME: No matching SA Query "
2533			"transaction identifier found");
2534		return;
2535	}
2536
2537	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Reply to pending SA Query received "
2538		"from " MACSTR, MAC2STR(sa));
2539	sme_stop_sa_query(wpa_s);
2540}
2541
2542
2543void sme_sa_query_rx(struct wpa_supplicant *wpa_s, const u8 *sa,
2544		     const u8 *data, size_t len)
2545{
2546	if (len < 1 + WLAN_SA_QUERY_TR_ID_LEN)
2547		return;
2548
2549	wpa_dbg(wpa_s, MSG_DEBUG, "SME: Received SA Query frame from "
2550		MACSTR " (trans_id %02x%02x)", MAC2STR(sa), data[1], data[2]);
2551
2552#ifdef CONFIG_OCV
2553	if (wpa_sm_ocv_enabled(wpa_s->wpa)) {
2554		struct ieee802_11_elems elems;
2555		struct wpa_channel_info ci;
2556
2557		if (ieee802_11_parse_elems(data + 1 + WLAN_SA_QUERY_TR_ID_LEN,
2558					   len - 1 - WLAN_SA_QUERY_TR_ID_LEN,
2559					   &elems, 1) == ParseFailed) {
2560			wpa_printf(MSG_DEBUG,
2561				   "SA Query: Failed to parse elements");
2562			return;
2563		}
2564
2565		if (wpa_drv_channel_info(wpa_s, &ci) != 0) {
2566			wpa_printf(MSG_WARNING,
2567				   "Failed to get channel info to validate received OCI in SA Query Action frame");
2568			return;
2569		}
2570
2571		if (ocv_verify_tx_params(elems.oci, elems.oci_len, &ci,
2572					 channel_width_to_int(ci.chanwidth),
2573					 ci.seg1_idx) != 0) {
2574			wpa_printf(MSG_WARNING, "%s", ocv_errorstr);
2575			return;
2576		}
2577	}
2578#endif /* CONFIG_OCV */
2579
2580	if (data[0] == WLAN_SA_QUERY_REQUEST)
2581		sme_process_sa_query_request(wpa_s, sa, data, len);
2582	else if (data[0] == WLAN_SA_QUERY_RESPONSE)
2583		sme_process_sa_query_response(wpa_s, sa, data, len);
2584}
2585
2586#endif /* CONFIG_IEEE80211W */
2587