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