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