1/*
2 * wpa_supplicant / WPS integration
3 * Copyright (c) 2008-2012, 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 "eloop.h"
13#include "uuid.h"
14#include "crypto/random.h"
15#include "crypto/dh_group5.h"
16#include "common/ieee802_11_defs.h"
17#include "common/ieee802_11_common.h"
18#include "common/wpa_common.h"
19#include "common/wpa_ctrl.h"
20#include "eap_common/eap_wsc_common.h"
21#include "eap_peer/eap.h"
22#include "eapol_supp/eapol_supp_sm.h"
23#include "rsn_supp/wpa.h"
24#include "wps/wps_attr_parse.h"
25#include "config.h"
26#include "wpa_supplicant_i.h"
27#include "driver_i.h"
28#include "notify.h"
29#include "blacklist.h"
30#include "bss.h"
31#include "scan.h"
32#include "ap.h"
33#include "p2p/p2p.h"
34#include "p2p_supplicant.h"
35#include "wps_supplicant.h"
36
37
38#ifndef WPS_PIN_SCAN_IGNORE_SEL_REG
39#define WPS_PIN_SCAN_IGNORE_SEL_REG 3
40#endif /* WPS_PIN_SCAN_IGNORE_SEL_REG */
41
42static void wpas_wps_timeout(void *eloop_ctx, void *timeout_ctx);
43static void wpas_clear_wps(struct wpa_supplicant *wpa_s);
44
45
46static void wpas_wps_clear_ap_info(struct wpa_supplicant *wpa_s)
47{
48	os_free(wpa_s->wps_ap);
49	wpa_s->wps_ap = NULL;
50	wpa_s->num_wps_ap = 0;
51	wpa_s->wps_ap_iter = 0;
52}
53
54
55int wpas_wps_eapol_cb(struct wpa_supplicant *wpa_s)
56{
57	if (!wpa_s->wps_success &&
58	    wpa_s->current_ssid &&
59	    eap_is_wps_pin_enrollee(&wpa_s->current_ssid->eap)) {
60		const u8 *bssid = wpa_s->bssid;
61		if (is_zero_ether_addr(bssid))
62			bssid = wpa_s->pending_bssid;
63
64		wpa_printf(MSG_DEBUG, "WPS: PIN registration with " MACSTR
65			   " did not succeed - continue trying to find "
66			   "suitable AP", MAC2STR(bssid));
67		wpa_blacklist_add(wpa_s, bssid);
68
69		wpa_supplicant_deauthenticate(wpa_s,
70					      WLAN_REASON_DEAUTH_LEAVING);
71		wpa_s->reassociate = 1;
72		wpa_supplicant_req_scan(wpa_s,
73					wpa_s->blacklist_cleared ? 5 : 0, 0);
74		wpa_s->blacklist_cleared = 0;
75		return 1;
76	}
77
78	wpas_wps_clear_ap_info(wpa_s);
79	eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
80	if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && !wpa_s->wps_success)
81		wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_FAIL);
82
83	if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid &&
84	    !(wpa_s->current_ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
85		int disabled = wpa_s->current_ssid->disabled;
86		unsigned int freq = wpa_s->assoc_freq;
87		wpa_printf(MSG_DEBUG, "WPS: Network configuration replaced - "
88			   "try to associate with the received credential "
89			   "(freq=%u)", freq);
90		wpa_supplicant_deauthenticate(wpa_s,
91					      WLAN_REASON_DEAUTH_LEAVING);
92		if (disabled) {
93			wpa_printf(MSG_DEBUG, "WPS: Current network is "
94				   "disabled - wait for user to enable");
95			return 1;
96		}
97		wpa_s->after_wps = 5;
98		wpa_s->wps_freq = freq;
99		wpa_s->normal_scans = 0;
100		wpa_s->reassociate = 1;
101		wpa_supplicant_req_scan(wpa_s, 0, 0);
102		return 1;
103	}
104
105	if (wpa_s->key_mgmt == WPA_KEY_MGMT_WPS && wpa_s->current_ssid) {
106		wpa_printf(MSG_DEBUG, "WPS: Registration completed - waiting "
107			   "for external credential processing");
108		wpas_clear_wps(wpa_s);
109		wpa_supplicant_deauthenticate(wpa_s,
110					      WLAN_REASON_DEAUTH_LEAVING);
111		return 1;
112	}
113
114	return 0;
115}
116
117
118static void wpas_wps_security_workaround(struct wpa_supplicant *wpa_s,
119					 struct wpa_ssid *ssid,
120					 const struct wps_credential *cred)
121{
122	struct wpa_driver_capa capa;
123	struct wpa_bss *bss;
124	const u8 *ie;
125	struct wpa_ie_data adv;
126	int wpa2 = 0, ccmp = 0;
127
128	/*
129	 * Many existing WPS APs do not know how to negotiate WPA2 or CCMP in
130	 * case they are configured for mixed mode operation (WPA+WPA2 and
131	 * TKIP+CCMP). Try to use scan results to figure out whether the AP
132	 * actually supports stronger security and select that if the client
133	 * has support for it, too.
134	 */
135
136	if (wpa_drv_get_capa(wpa_s, &capa))
137		return; /* Unknown what driver supports */
138
139	if (ssid->ssid == NULL)
140		return;
141	bss = wpa_bss_get(wpa_s, cred->mac_addr, ssid->ssid, ssid->ssid_len);
142	if (bss == NULL) {
143		wpa_printf(MSG_DEBUG, "WPS: The AP was not found from BSS "
144			   "table - use credential as-is");
145		return;
146	}
147
148	wpa_printf(MSG_DEBUG, "WPS: AP found from BSS table");
149
150	ie = wpa_bss_get_ie(bss, WLAN_EID_RSN);
151	if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0) {
152		wpa2 = 1;
153		if (adv.pairwise_cipher & WPA_CIPHER_CCMP)
154			ccmp = 1;
155	} else {
156		ie = wpa_bss_get_vendor_ie(bss, WPA_IE_VENDOR_TYPE);
157		if (ie && wpa_parse_wpa_ie(ie, 2 + ie[1], &adv) == 0 &&
158		    adv.pairwise_cipher & WPA_CIPHER_CCMP)
159			ccmp = 1;
160	}
161
162	if (ie == NULL && (ssid->proto & WPA_PROTO_WPA) &&
163	    (ssid->pairwise_cipher & WPA_CIPHER_TKIP)) {
164		/*
165		 * TODO: This could be the initial AP configuration and the
166		 * Beacon contents could change shortly. Should request a new
167		 * scan and delay addition of the network until the updated
168		 * scan results are available.
169		 */
170		wpa_printf(MSG_DEBUG, "WPS: The AP did not yet advertise WPA "
171			   "support - use credential as-is");
172		return;
173	}
174
175	if (ccmp && !(ssid->pairwise_cipher & WPA_CIPHER_CCMP) &&
176	    (ssid->pairwise_cipher & WPA_CIPHER_TKIP) &&
177	    (capa.key_mgmt & WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK)) {
178		wpa_printf(MSG_DEBUG, "WPS: Add CCMP into the credential "
179			   "based on scan results");
180		if (wpa_s->conf->ap_scan == 1)
181			ssid->pairwise_cipher |= WPA_CIPHER_CCMP;
182		else
183			ssid->pairwise_cipher = WPA_CIPHER_CCMP;
184	}
185
186	if (wpa2 && !(ssid->proto & WPA_PROTO_RSN) &&
187	    (ssid->proto & WPA_PROTO_WPA) &&
188	    (capa.enc & WPA_DRIVER_CAPA_ENC_CCMP)) {
189		wpa_printf(MSG_DEBUG, "WPS: Add WPA2 into the credential "
190			   "based on scan results");
191		if (wpa_s->conf->ap_scan == 1)
192			ssid->proto |= WPA_PROTO_RSN;
193		else
194			ssid->proto = WPA_PROTO_RSN;
195	}
196}
197
198
199static int wpa_supplicant_wps_cred(void *ctx,
200				   const struct wps_credential *cred)
201{
202	struct wpa_supplicant *wpa_s = ctx;
203	struct wpa_ssid *ssid = wpa_s->current_ssid;
204	u8 key_idx = 0;
205	u16 auth_type;
206#ifdef CONFIG_WPS_REG_DISABLE_OPEN
207	int registrar = 0;
208#endif /* CONFIG_WPS_REG_DISABLE_OPEN */
209
210	if ((wpa_s->conf->wps_cred_processing == 1 ||
211	     wpa_s->conf->wps_cred_processing == 2) && cred->cred_attr) {
212		size_t blen = cred->cred_attr_len * 2 + 1;
213		char *buf = os_malloc(blen);
214		if (buf) {
215			wpa_snprintf_hex(buf, blen,
216					 cred->cred_attr, cred->cred_attr_len);
217			wpa_msg(wpa_s, MSG_INFO, "%s%s",
218				WPS_EVENT_CRED_RECEIVED, buf);
219			os_free(buf);
220		}
221
222		wpas_notify_wps_credential(wpa_s, cred);
223	} else
224		wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_CRED_RECEIVED);
225
226	wpa_hexdump_key(MSG_DEBUG, "WPS: Received Credential attribute",
227			cred->cred_attr, cred->cred_attr_len);
228
229	if (wpa_s->conf->wps_cred_processing == 1)
230		return 0;
231
232	wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID", cred->ssid, cred->ssid_len);
233	wpa_printf(MSG_DEBUG, "WPS: Authentication Type 0x%x",
234		   cred->auth_type);
235	wpa_printf(MSG_DEBUG, "WPS: Encryption Type 0x%x", cred->encr_type);
236	wpa_printf(MSG_DEBUG, "WPS: Network Key Index %d", cred->key_idx);
237	wpa_hexdump_key(MSG_DEBUG, "WPS: Network Key",
238			cred->key, cred->key_len);
239	wpa_printf(MSG_DEBUG, "WPS: MAC Address " MACSTR,
240		   MAC2STR(cred->mac_addr));
241
242	auth_type = cred->auth_type;
243	if (auth_type == (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
244		wpa_printf(MSG_DEBUG, "WPS: Workaround - convert mixed-mode "
245			   "auth_type into WPA2PSK");
246		auth_type = WPS_AUTH_WPA2PSK;
247	}
248
249	if (auth_type != WPS_AUTH_OPEN &&
250	    auth_type != WPS_AUTH_SHARED &&
251	    auth_type != WPS_AUTH_WPAPSK &&
252	    auth_type != WPS_AUTH_WPA2PSK) {
253		wpa_printf(MSG_DEBUG, "WPS: Ignored credentials for "
254			   "unsupported authentication type 0x%x",
255			   auth_type);
256		return 0;
257	}
258
259	if (ssid && (ssid->key_mgmt & WPA_KEY_MGMT_WPS)) {
260		wpa_printf(MSG_DEBUG, "WPS: Replace WPS network block based "
261			   "on the received credential");
262#ifdef CONFIG_WPS_REG_DISABLE_OPEN
263		if (ssid->eap.identity &&
264		    ssid->eap.identity_len == WSC_ID_REGISTRAR_LEN &&
265		    os_memcmp(ssid->eap.identity, WSC_ID_REGISTRAR,
266			      WSC_ID_REGISTRAR_LEN) == 0)
267			registrar = 1;
268#endif /* CONFIG_WPS_REG_DISABLE_OPEN */
269		os_free(ssid->eap.identity);
270		ssid->eap.identity = NULL;
271		ssid->eap.identity_len = 0;
272		os_free(ssid->eap.phase1);
273		ssid->eap.phase1 = NULL;
274		os_free(ssid->eap.eap_methods);
275		ssid->eap.eap_methods = NULL;
276		if (!ssid->p2p_group) {
277			ssid->temporary = 0;
278			ssid->bssid_set = 0;
279		}
280		ssid->disabled_until.sec = 0;
281		ssid->disabled_until.usec = 0;
282		ssid->auth_failures = 0;
283	} else {
284		wpa_printf(MSG_DEBUG, "WPS: Create a new network based on the "
285			   "received credential");
286		ssid = wpa_config_add_network(wpa_s->conf);
287		if (ssid == NULL)
288			return -1;
289		wpas_notify_network_added(wpa_s, ssid);
290	}
291
292	wpa_config_set_network_defaults(ssid);
293
294	os_free(ssid->ssid);
295	ssid->ssid = os_malloc(cred->ssid_len);
296	if (ssid->ssid) {
297		os_memcpy(ssid->ssid, cred->ssid, cred->ssid_len);
298		ssid->ssid_len = cred->ssid_len;
299	}
300
301	switch (cred->encr_type) {
302	case WPS_ENCR_NONE:
303		break;
304	case WPS_ENCR_WEP:
305		if (cred->key_len <= 0)
306			break;
307		if (cred->key_len != 5 && cred->key_len != 13 &&
308		    cred->key_len != 10 && cred->key_len != 26) {
309			wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key length "
310				   "%lu", (unsigned long) cred->key_len);
311			return -1;
312		}
313		if (cred->key_idx > NUM_WEP_KEYS) {
314			wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key index %d",
315				   cred->key_idx);
316			return -1;
317		}
318		if (cred->key_idx)
319			key_idx = cred->key_idx - 1;
320		if (cred->key_len == 10 || cred->key_len == 26) {
321			if (hexstr2bin((char *) cred->key,
322				       ssid->wep_key[key_idx],
323				       cred->key_len / 2) < 0) {
324				wpa_printf(MSG_ERROR, "WPS: Invalid WEP Key "
325					   "%d", key_idx);
326				return -1;
327			}
328			ssid->wep_key_len[key_idx] = cred->key_len / 2;
329		} else {
330			os_memcpy(ssid->wep_key[key_idx], cred->key,
331				  cred->key_len);
332			ssid->wep_key_len[key_idx] = cred->key_len;
333		}
334		ssid->wep_tx_keyidx = key_idx;
335		break;
336	case WPS_ENCR_TKIP:
337		ssid->pairwise_cipher = WPA_CIPHER_TKIP;
338		break;
339	case WPS_ENCR_AES:
340		ssid->pairwise_cipher = WPA_CIPHER_CCMP;
341		break;
342	}
343
344	switch (auth_type) {
345	case WPS_AUTH_OPEN:
346		ssid->auth_alg = WPA_AUTH_ALG_OPEN;
347		ssid->key_mgmt = WPA_KEY_MGMT_NONE;
348		ssid->proto = 0;
349#ifdef CONFIG_WPS_REG_DISABLE_OPEN
350		if (registrar) {
351			wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_OPEN_NETWORK
352				"id=%d - Credentials for an open "
353				"network disabled by default - use "
354				"'select_network %d' to enable",
355				ssid->id, ssid->id);
356			ssid->disabled = 1;
357		}
358#endif /* CONFIG_WPS_REG_DISABLE_OPEN */
359		break;
360	case WPS_AUTH_SHARED:
361		ssid->auth_alg = WPA_AUTH_ALG_SHARED;
362		ssid->key_mgmt = WPA_KEY_MGMT_NONE;
363		ssid->proto = 0;
364		break;
365	case WPS_AUTH_WPAPSK:
366		ssid->auth_alg = WPA_AUTH_ALG_OPEN;
367		ssid->key_mgmt = WPA_KEY_MGMT_PSK;
368		ssid->proto = WPA_PROTO_WPA;
369		break;
370	case WPS_AUTH_WPA2PSK:
371		ssid->auth_alg = WPA_AUTH_ALG_OPEN;
372		ssid->key_mgmt = WPA_KEY_MGMT_PSK;
373		ssid->proto = WPA_PROTO_RSN;
374		break;
375	}
376
377	if (ssid->key_mgmt == WPA_KEY_MGMT_PSK) {
378		if (cred->key_len == 2 * PMK_LEN) {
379			if (hexstr2bin((const char *) cred->key, ssid->psk,
380				       PMK_LEN)) {
381				wpa_printf(MSG_ERROR, "WPS: Invalid Network "
382					   "Key");
383				return -1;
384			}
385			ssid->psk_set = 1;
386			ssid->export_keys = 1;
387		} else if (cred->key_len >= 8 && cred->key_len < 2 * PMK_LEN) {
388			os_free(ssid->passphrase);
389			ssid->passphrase = os_malloc(cred->key_len + 1);
390			if (ssid->passphrase == NULL)
391				return -1;
392			os_memcpy(ssid->passphrase, cred->key, cred->key_len);
393			ssid->passphrase[cred->key_len] = '\0';
394			wpa_config_update_psk(ssid);
395			ssid->export_keys = 1;
396		} else {
397			wpa_printf(MSG_ERROR, "WPS: Invalid Network Key "
398				   "length %lu",
399				   (unsigned long) cred->key_len);
400			return -1;
401		}
402	}
403
404	wpas_wps_security_workaround(wpa_s, ssid, cred);
405
406#ifndef CONFIG_NO_CONFIG_WRITE
407	if (wpa_s->conf->update_config &&
408	    wpa_config_write(wpa_s->confname, wpa_s->conf)) {
409		wpa_printf(MSG_DEBUG, "WPS: Failed to update configuration");
410		return -1;
411	}
412#endif /* CONFIG_NO_CONFIG_WRITE */
413
414	return 0;
415}
416
417
418#ifdef CONFIG_P2P
419static void wpas_wps_pbc_overlap_cb(void *eloop_ctx, void *timeout_ctx)
420{
421	struct wpa_supplicant *wpa_s = eloop_ctx;
422	wpas_p2p_notif_pbc_overlap(wpa_s);
423}
424#endif /* CONFIG_P2P */
425
426
427static void wpa_supplicant_wps_event_m2d(struct wpa_supplicant *wpa_s,
428					 struct wps_event_m2d *m2d)
429{
430	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_M2D
431		"dev_password_id=%d config_error=%d",
432		m2d->dev_password_id, m2d->config_error);
433	wpas_notify_wps_event_m2d(wpa_s, m2d);
434#ifdef CONFIG_P2P
435	if (wpa_s->parent && wpa_s->parent != wpa_s) {
436		wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_M2D
437			"dev_password_id=%d config_error=%d",
438			m2d->dev_password_id, m2d->config_error);
439	}
440	if (m2d->config_error == WPS_CFG_MULTIPLE_PBC_DETECTED) {
441		/*
442		 * Notify P2P from eloop timeout to avoid issues with the
443		 * interface getting removed while processing a message.
444		 */
445		eloop_register_timeout(0, 0, wpas_wps_pbc_overlap_cb, wpa_s,
446				       NULL);
447	}
448#endif /* CONFIG_P2P */
449}
450
451
452static const char * wps_event_fail_reason[NUM_WPS_EI_VALUES] = {
453	"No Error", /* WPS_EI_NO_ERROR */
454	"TKIP Only Prohibited", /* WPS_EI_SECURITY_TKIP_ONLY_PROHIBITED */
455	"WEP Prohibited" /* WPS_EI_SECURITY_WEP_PROHIBITED */
456};
457
458static void wpa_supplicant_wps_event_fail(struct wpa_supplicant *wpa_s,
459					  struct wps_event_fail *fail)
460{
461	if (fail->error_indication > 0 &&
462	    fail->error_indication < NUM_WPS_EI_VALUES) {
463		wpa_msg(wpa_s, MSG_INFO,
464			WPS_EVENT_FAIL "msg=%d config_error=%d reason=%d (%s)",
465			fail->msg, fail->config_error, fail->error_indication,
466			wps_event_fail_reason[fail->error_indication]);
467		if (wpa_s->parent && wpa_s->parent != wpa_s)
468			wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
469				"msg=%d config_error=%d reason=%d (%s)",
470				fail->msg, fail->config_error,
471				fail->error_indication,
472				wps_event_fail_reason[fail->error_indication]);
473	} else {
474		wpa_msg(wpa_s, MSG_INFO,
475			WPS_EVENT_FAIL "msg=%d config_error=%d",
476			fail->msg, fail->config_error);
477		if (wpa_s->parent && wpa_s->parent != wpa_s)
478			wpa_msg(wpa_s->parent, MSG_INFO, WPS_EVENT_FAIL
479				"msg=%d config_error=%d",
480				fail->msg, fail->config_error);
481	}
482	wpas_clear_wps(wpa_s);
483	wpas_notify_wps_event_fail(wpa_s, fail);
484#ifdef CONFIG_P2P
485	wpas_p2p_wps_failed(wpa_s, fail);
486#endif /* CONFIG_P2P */
487}
488
489
490static void wpa_supplicant_wps_event_success(struct wpa_supplicant *wpa_s)
491{
492	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_SUCCESS);
493	wpa_s->wps_success = 1;
494	wpas_notify_wps_event_success(wpa_s);
495#ifdef CONFIG_P2P
496	wpas_p2p_wps_success(wpa_s, wpa_s->bssid, 0);
497#endif /* CONFIG_P2P */
498}
499
500
501static void wpa_supplicant_wps_event_er_ap_add(struct wpa_supplicant *wpa_s,
502					       struct wps_event_er_ap *ap)
503{
504	char uuid_str[100];
505	char dev_type[WPS_DEV_TYPE_BUFSIZE];
506
507	uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
508	if (ap->pri_dev_type)
509		wps_dev_type_bin2str(ap->pri_dev_type, dev_type,
510				     sizeof(dev_type));
511	else
512		dev_type[0] = '\0';
513
514	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_ADD "%s " MACSTR
515		" pri_dev_type=%s wps_state=%d |%s|%s|%s|%s|%s|%s|",
516		uuid_str, MAC2STR(ap->mac_addr), dev_type, ap->wps_state,
517		ap->friendly_name ? ap->friendly_name : "",
518		ap->manufacturer ? ap->manufacturer : "",
519		ap->model_description ? ap->model_description : "",
520		ap->model_name ? ap->model_name : "",
521		ap->manufacturer_url ? ap->manufacturer_url : "",
522		ap->model_url ? ap->model_url : "");
523}
524
525
526static void wpa_supplicant_wps_event_er_ap_remove(struct wpa_supplicant *wpa_s,
527						  struct wps_event_er_ap *ap)
528{
529	char uuid_str[100];
530	uuid_bin2str(ap->uuid, uuid_str, sizeof(uuid_str));
531	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_REMOVE "%s", uuid_str);
532}
533
534
535static void wpa_supplicant_wps_event_er_enrollee_add(
536	struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
537{
538	char uuid_str[100];
539	char dev_type[WPS_DEV_TYPE_BUFSIZE];
540
541	uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
542	if (enrollee->pri_dev_type)
543		wps_dev_type_bin2str(enrollee->pri_dev_type, dev_type,
544				     sizeof(dev_type));
545	else
546		dev_type[0] = '\0';
547
548	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_ADD "%s " MACSTR
549		" M1=%d config_methods=0x%x dev_passwd_id=%d pri_dev_type=%s "
550		"|%s|%s|%s|%s|%s|",
551		uuid_str, MAC2STR(enrollee->mac_addr), enrollee->m1_received,
552		enrollee->config_methods, enrollee->dev_passwd_id, dev_type,
553		enrollee->dev_name ? enrollee->dev_name : "",
554		enrollee->manufacturer ? enrollee->manufacturer : "",
555		enrollee->model_name ? enrollee->model_name : "",
556		enrollee->model_number ? enrollee->model_number : "",
557		enrollee->serial_number ? enrollee->serial_number : "");
558}
559
560
561static void wpa_supplicant_wps_event_er_enrollee_remove(
562	struct wpa_supplicant *wpa_s, struct wps_event_er_enrollee *enrollee)
563{
564	char uuid_str[100];
565	uuid_bin2str(enrollee->uuid, uuid_str, sizeof(uuid_str));
566	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_ENROLLEE_REMOVE "%s " MACSTR,
567		uuid_str, MAC2STR(enrollee->mac_addr));
568}
569
570
571static void wpa_supplicant_wps_event_er_ap_settings(
572	struct wpa_supplicant *wpa_s,
573	struct wps_event_er_ap_settings *ap_settings)
574{
575	char uuid_str[100];
576	char key_str[65];
577	const struct wps_credential *cred = ap_settings->cred;
578
579	key_str[0] = '\0';
580	if (cred->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
581		if (cred->key_len >= 8 && cred->key_len <= 64) {
582			os_memcpy(key_str, cred->key, cred->key_len);
583			key_str[cred->key_len] = '\0';
584		}
585	}
586
587	uuid_bin2str(ap_settings->uuid, uuid_str, sizeof(uuid_str));
588	/* Use wpa_msg_ctrl to avoid showing the key in debug log */
589	wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_ER_AP_SETTINGS
590		     "uuid=%s ssid=%s auth_type=0x%04x encr_type=0x%04x "
591		     "key=%s",
592		     uuid_str, wpa_ssid_txt(cred->ssid, cred->ssid_len),
593		     cred->auth_type, cred->encr_type, key_str);
594}
595
596
597static void wpa_supplicant_wps_event_er_set_sel_reg(
598	struct wpa_supplicant *wpa_s,
599	struct wps_event_er_set_selected_registrar *ev)
600{
601	char uuid_str[100];
602
603	uuid_bin2str(ev->uuid, uuid_str, sizeof(uuid_str));
604	switch (ev->state) {
605	case WPS_ER_SET_SEL_REG_START:
606		wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
607			"uuid=%s state=START sel_reg=%d dev_passwd_id=%u "
608			"sel_reg_config_methods=0x%x",
609			uuid_str, ev->sel_reg, ev->dev_passwd_id,
610			ev->sel_reg_config_methods);
611		break;
612	case WPS_ER_SET_SEL_REG_DONE:
613		wpa_msg(wpa_s, MSG_DEBUG, WPS_EVENT_ER_SET_SEL_REG
614			"uuid=%s state=DONE", uuid_str);
615		break;
616	case WPS_ER_SET_SEL_REG_FAILED:
617		wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_ER_SET_SEL_REG
618			"uuid=%s state=FAILED", uuid_str);
619		break;
620	}
621}
622
623
624static void wpa_supplicant_wps_event(void *ctx, enum wps_event event,
625				     union wps_event_data *data)
626{
627	struct wpa_supplicant *wpa_s = ctx;
628	switch (event) {
629	case WPS_EV_M2D:
630		wpa_supplicant_wps_event_m2d(wpa_s, &data->m2d);
631		break;
632	case WPS_EV_FAIL:
633		wpa_supplicant_wps_event_fail(wpa_s, &data->fail);
634		break;
635	case WPS_EV_SUCCESS:
636		wpa_supplicant_wps_event_success(wpa_s);
637		break;
638	case WPS_EV_PWD_AUTH_FAIL:
639#ifdef CONFIG_AP
640		if (wpa_s->ap_iface && data->pwd_auth_fail.enrollee)
641			wpa_supplicant_ap_pwd_auth_fail(wpa_s);
642#endif /* CONFIG_AP */
643		break;
644	case WPS_EV_PBC_OVERLAP:
645		break;
646	case WPS_EV_PBC_TIMEOUT:
647		break;
648	case WPS_EV_ER_AP_ADD:
649		wpa_supplicant_wps_event_er_ap_add(wpa_s, &data->ap);
650		break;
651	case WPS_EV_ER_AP_REMOVE:
652		wpa_supplicant_wps_event_er_ap_remove(wpa_s, &data->ap);
653		break;
654	case WPS_EV_ER_ENROLLEE_ADD:
655		wpa_supplicant_wps_event_er_enrollee_add(wpa_s,
656							 &data->enrollee);
657		break;
658	case WPS_EV_ER_ENROLLEE_REMOVE:
659		wpa_supplicant_wps_event_er_enrollee_remove(wpa_s,
660							    &data->enrollee);
661		break;
662	case WPS_EV_ER_AP_SETTINGS:
663		wpa_supplicant_wps_event_er_ap_settings(wpa_s,
664							&data->ap_settings);
665		break;
666	case WPS_EV_ER_SET_SELECTED_REGISTRAR:
667		wpa_supplicant_wps_event_er_set_sel_reg(wpa_s,
668							&data->set_sel_reg);
669		break;
670	case WPS_EV_AP_PIN_SUCCESS:
671		break;
672	}
673}
674
675
676enum wps_request_type wpas_wps_get_req_type(struct wpa_ssid *ssid)
677{
678	if (eap_is_wps_pbc_enrollee(&ssid->eap) ||
679	    eap_is_wps_pin_enrollee(&ssid->eap))
680		return WPS_REQ_ENROLLEE;
681	else
682		return WPS_REQ_REGISTRAR;
683}
684
685
686static void wpas_clear_wps(struct wpa_supplicant *wpa_s)
687{
688	int id;
689	struct wpa_ssid *ssid, *remove_ssid = NULL, *prev_current;
690
691	prev_current = wpa_s->current_ssid;
692
693	eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
694
695	/* Remove any existing WPS network from configuration */
696	ssid = wpa_s->conf->ssid;
697	while (ssid) {
698		if (ssid->key_mgmt & WPA_KEY_MGMT_WPS) {
699			if (ssid == wpa_s->current_ssid) {
700				wpa_s->current_ssid = NULL;
701				if (ssid != NULL)
702					wpas_notify_network_changed(wpa_s);
703			}
704			id = ssid->id;
705			remove_ssid = ssid;
706		} else
707			id = -1;
708		ssid = ssid->next;
709		if (id >= 0) {
710			if (prev_current == remove_ssid) {
711				wpa_sm_set_config(wpa_s->wpa, NULL);
712				eapol_sm_notify_config(wpa_s->eapol, NULL,
713						       NULL);
714			}
715			wpas_notify_network_removed(wpa_s, remove_ssid);
716			wpa_config_remove_network(wpa_s->conf, id);
717		}
718	}
719
720	wpas_wps_clear_ap_info(wpa_s);
721}
722
723
724static void wpas_wps_timeout(void *eloop_ctx, void *timeout_ctx)
725{
726	struct wpa_supplicant *wpa_s = eloop_ctx;
727	wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_TIMEOUT "Requested operation timed "
728		"out");
729	wpas_clear_wps(wpa_s);
730}
731
732
733static struct wpa_ssid * wpas_wps_add_network(struct wpa_supplicant *wpa_s,
734					      int registrar, const u8 *bssid)
735{
736	struct wpa_ssid *ssid;
737
738	ssid = wpa_config_add_network(wpa_s->conf);
739	if (ssid == NULL)
740		return NULL;
741	wpas_notify_network_added(wpa_s, ssid);
742	wpa_config_set_network_defaults(ssid);
743	ssid->temporary = 1;
744	if (wpa_config_set(ssid, "key_mgmt", "WPS", 0) < 0 ||
745	    wpa_config_set(ssid, "eap", "WSC", 0) < 0 ||
746	    wpa_config_set(ssid, "identity", registrar ?
747			   "\"" WSC_ID_REGISTRAR "\"" :
748			   "\"" WSC_ID_ENROLLEE "\"", 0) < 0) {
749		wpas_notify_network_removed(wpa_s, ssid);
750		wpa_config_remove_network(wpa_s->conf, ssid->id);
751		return NULL;
752	}
753
754	if (bssid) {
755#ifndef CONFIG_P2P
756		struct wpa_bss *bss;
757		int count = 0;
758#endif /* CONFIG_P2P */
759
760		os_memcpy(ssid->bssid, bssid, ETH_ALEN);
761		ssid->bssid_set = 1;
762
763		/*
764		 * Note: With P2P, the SSID may change at the time the WPS
765		 * provisioning is started, so better not filter the AP based
766		 * on the current SSID in the scan results.
767		 */
768#ifndef CONFIG_P2P
769		dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
770			if (os_memcmp(bssid, bss->bssid, ETH_ALEN) != 0)
771				continue;
772
773			os_free(ssid->ssid);
774			ssid->ssid = os_malloc(bss->ssid_len);
775			if (ssid->ssid == NULL)
776				break;
777			os_memcpy(ssid->ssid, bss->ssid, bss->ssid_len);
778			ssid->ssid_len = bss->ssid_len;
779			wpa_hexdump_ascii(MSG_DEBUG, "WPS: Picked SSID from "
780					  "scan results",
781					  ssid->ssid, ssid->ssid_len);
782			count++;
783		}
784
785		if (count > 1) {
786			wpa_printf(MSG_DEBUG, "WPS: More than one SSID found "
787				   "for the AP; use wildcard");
788			os_free(ssid->ssid);
789			ssid->ssid = NULL;
790			ssid->ssid_len = 0;
791		}
792#endif /* CONFIG_P2P */
793	}
794
795	return ssid;
796}
797
798
799static void wpas_wps_reassoc(struct wpa_supplicant *wpa_s,
800			     struct wpa_ssid *selected, const u8 *bssid)
801{
802	struct wpa_ssid *ssid;
803	struct wpa_bss *bss;
804
805	wpa_s->known_wps_freq = 0;
806	if (bssid) {
807		bss = wpa_bss_get_bssid(wpa_s, bssid);
808		if (bss && bss->freq > 0) {
809			wpa_s->known_wps_freq = 1;
810			wpa_s->wps_freq = bss->freq;
811		}
812	}
813
814	if (wpa_s->current_ssid)
815		wpa_supplicant_deauthenticate(
816			wpa_s, WLAN_REASON_DEAUTH_LEAVING);
817
818	/* Mark all other networks disabled and trigger reassociation */
819	ssid = wpa_s->conf->ssid;
820	while (ssid) {
821		int was_disabled = ssid->disabled;
822		/*
823		 * In case the network object corresponds to a persistent group
824		 * then do not send out network disabled signal. In addition,
825		 * do not change disabled status of persistent network objects
826		 * from 2 to 1 should we connect to another network.
827		 */
828		if (was_disabled != 2) {
829			ssid->disabled = ssid != selected;
830			if (was_disabled != ssid->disabled)
831				wpas_notify_network_enabled_changed(wpa_s,
832								    ssid);
833		}
834		ssid = ssid->next;
835	}
836	wpa_s->disconnected = 0;
837	wpa_s->reassociate = 1;
838	wpa_s->scan_runs = 0;
839	wpa_s->normal_scans = 0;
840	wpa_s->wps_success = 0;
841	wpa_s->blacklist_cleared = 0;
842	wpa_supplicant_req_scan(wpa_s, 0, 0);
843}
844
845
846int wpas_wps_start_pbc(struct wpa_supplicant *wpa_s, const u8 *bssid,
847		       int p2p_group)
848{
849	struct wpa_ssid *ssid;
850	wpas_clear_wps(wpa_s);
851	ssid = wpas_wps_add_network(wpa_s, 0, bssid);
852	if (ssid == NULL)
853		return -1;
854	ssid->temporary = 1;
855	ssid->p2p_group = p2p_group;
856#ifdef CONFIG_P2P
857	if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
858		ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
859		if (ssid->ssid) {
860			ssid->ssid_len = wpa_s->go_params->ssid_len;
861			os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
862				  ssid->ssid_len);
863			wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
864					  "SSID", ssid->ssid, ssid->ssid_len);
865		}
866	}
867#endif /* CONFIG_P2P */
868	wpa_config_set(ssid, "phase1", "\"pbc=1\"", 0);
869	if (wpa_s->wps_fragment_size)
870		ssid->eap.fragment_size = wpa_s->wps_fragment_size;
871	eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
872			       wpa_s, NULL);
873	wpas_wps_reassoc(wpa_s, ssid, bssid);
874	return 0;
875}
876
877
878int wpas_wps_start_pin(struct wpa_supplicant *wpa_s, const u8 *bssid,
879		       const char *pin, int p2p_group, u16 dev_pw_id)
880{
881	struct wpa_ssid *ssid;
882	char val[128];
883	unsigned int rpin = 0;
884
885	wpas_clear_wps(wpa_s);
886	ssid = wpas_wps_add_network(wpa_s, 0, bssid);
887	if (ssid == NULL)
888		return -1;
889	ssid->temporary = 1;
890	ssid->p2p_group = p2p_group;
891#ifdef CONFIG_P2P
892	if (p2p_group && wpa_s->go_params && wpa_s->go_params->ssid_len) {
893		ssid->ssid = os_zalloc(wpa_s->go_params->ssid_len + 1);
894		if (ssid->ssid) {
895			ssid->ssid_len = wpa_s->go_params->ssid_len;
896			os_memcpy(ssid->ssid, wpa_s->go_params->ssid,
897				  ssid->ssid_len);
898			wpa_hexdump_ascii(MSG_DEBUG, "WPS: Use specific AP "
899					  "SSID", ssid->ssid, ssid->ssid_len);
900		}
901	}
902#endif /* CONFIG_P2P */
903	if (pin)
904		os_snprintf(val, sizeof(val), "\"pin=%s dev_pw_id=%u\"",
905			    pin, dev_pw_id);
906	else {
907		rpin = wps_generate_pin();
908		os_snprintf(val, sizeof(val), "\"pin=%08d dev_pw_id=%u\"",
909			    rpin, dev_pw_id);
910	}
911	wpa_config_set(ssid, "phase1", val, 0);
912	if (wpa_s->wps_fragment_size)
913		ssid->eap.fragment_size = wpa_s->wps_fragment_size;
914	eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
915			       wpa_s, NULL);
916	wpa_s->wps_ap_iter = 1;
917	wpas_wps_reassoc(wpa_s, ssid, bssid);
918	return rpin;
919}
920
921
922/* Cancel the wps pbc/pin requests */
923int wpas_wps_cancel(struct wpa_supplicant *wpa_s)
924{
925#ifdef CONFIG_AP
926	if (wpa_s->ap_iface) {
927		wpa_printf(MSG_DEBUG, "WPS: Cancelling in AP mode");
928		return wpa_supplicant_ap_wps_cancel(wpa_s);
929	}
930#endif /* CONFIG_AP */
931
932	if (wpa_s->wpa_state == WPA_SCANNING ||
933	    wpa_s->wpa_state == WPA_DISCONNECTED) {
934		wpa_printf(MSG_DEBUG, "WPS: Cancel operation - cancel scan");
935		wpa_supplicant_cancel_scan(wpa_s);
936		wpas_clear_wps(wpa_s);
937	} else if (wpa_s->wpa_state >= WPA_ASSOCIATED) {
938		wpa_printf(MSG_DEBUG, "WPS: Cancel operation - "
939			   "deauthenticate");
940		wpa_supplicant_deauthenticate(wpa_s,
941					      WLAN_REASON_DEAUTH_LEAVING);
942		wpas_clear_wps(wpa_s);
943	} else
944		wpas_wps_clear_ap_info(wpa_s);
945
946	return 0;
947}
948
949
950#ifdef CONFIG_WPS_OOB
951int wpas_wps_start_oob(struct wpa_supplicant *wpa_s, char *device_type,
952		       char *path, char *method, char *name)
953{
954	struct wps_context *wps = wpa_s->wps;
955	struct oob_device_data *oob_dev;
956
957	oob_dev = wps_get_oob_device(device_type);
958	if (oob_dev == NULL)
959		return -1;
960	oob_dev->device_path = path;
961	oob_dev->device_name = name;
962	wps->oob_conf.oob_method = wps_get_oob_method(method);
963
964	if (wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_E) {
965		/*
966		 * Use pre-configured DH keys in order to be able to write the
967		 * key hash into the OOB file.
968		 */
969		wpabuf_free(wps->dh_pubkey);
970		wpabuf_free(wps->dh_privkey);
971		wps->dh_privkey = NULL;
972		wps->dh_pubkey = NULL;
973		dh5_free(wps->dh_ctx);
974		wps->dh_ctx = dh5_init(&wps->dh_privkey, &wps->dh_pubkey);
975		wps->dh_pubkey = wpabuf_zeropad(wps->dh_pubkey, 192);
976		if (wps->dh_ctx == NULL || wps->dh_pubkey == NULL) {
977			wpa_printf(MSG_ERROR, "WPS: Failed to initialize "
978				   "Diffie-Hellman handshake");
979			return -1;
980		}
981	}
982
983	if (wps->oob_conf.oob_method == OOB_METHOD_CRED)
984		wpas_clear_wps(wpa_s);
985
986	if (wps_process_oob(wps, oob_dev, 0) < 0)
987		return -1;
988
989	if ((wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_E ||
990	     wps->oob_conf.oob_method == OOB_METHOD_DEV_PWD_R) &&
991	    wpas_wps_start_pin(wpa_s, NULL,
992			       wpabuf_head(wps->oob_conf.dev_password), 0,
993			       DEV_PW_DEFAULT) < 0)
994			return -1;
995
996	return 0;
997}
998#endif /* CONFIG_WPS_OOB */
999
1000
1001int wpas_wps_start_reg(struct wpa_supplicant *wpa_s, const u8 *bssid,
1002		       const char *pin, struct wps_new_ap_settings *settings)
1003{
1004	struct wpa_ssid *ssid;
1005	char val[200];
1006	char *pos, *end;
1007	int res;
1008
1009	if (!pin)
1010		return -1;
1011	wpas_clear_wps(wpa_s);
1012	ssid = wpas_wps_add_network(wpa_s, 1, bssid);
1013	if (ssid == NULL)
1014		return -1;
1015	ssid->temporary = 1;
1016	pos = val;
1017	end = pos + sizeof(val);
1018	res = os_snprintf(pos, end - pos, "\"pin=%s", pin);
1019	if (res < 0 || res >= end - pos)
1020		return -1;
1021	pos += res;
1022	if (settings) {
1023		res = os_snprintf(pos, end - pos, " new_ssid=%s new_auth=%s "
1024				  "new_encr=%s new_key=%s",
1025				  settings->ssid_hex, settings->auth,
1026				  settings->encr, settings->key_hex);
1027		if (res < 0 || res >= end - pos)
1028			return -1;
1029		pos += res;
1030	}
1031	res = os_snprintf(pos, end - pos, "\"");
1032	if (res < 0 || res >= end - pos)
1033		return -1;
1034	wpa_config_set(ssid, "phase1", val, 0);
1035	if (wpa_s->wps_fragment_size)
1036		ssid->eap.fragment_size = wpa_s->wps_fragment_size;
1037	eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wpas_wps_timeout,
1038			       wpa_s, NULL);
1039	wpas_wps_reassoc(wpa_s, ssid, bssid);
1040	return 0;
1041}
1042
1043
1044static int wpas_wps_new_psk_cb(void *ctx, const u8 *mac_addr, const u8 *psk,
1045			       size_t psk_len)
1046{
1047	wpa_printf(MSG_DEBUG, "WPS: Received new WPA/WPA2-PSK from WPS for "
1048		   "STA " MACSTR, MAC2STR(mac_addr));
1049	wpa_hexdump_key(MSG_DEBUG, "Per-device PSK", psk, psk_len);
1050
1051	/* TODO */
1052
1053	return 0;
1054}
1055
1056
1057static void wpas_wps_pin_needed_cb(void *ctx, const u8 *uuid_e,
1058				   const struct wps_device_data *dev)
1059{
1060	char uuid[40], txt[400];
1061	int len;
1062	char devtype[WPS_DEV_TYPE_BUFSIZE];
1063	if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
1064		return;
1065	wpa_printf(MSG_DEBUG, "WPS: PIN needed for UUID-E %s", uuid);
1066	len = os_snprintf(txt, sizeof(txt), "WPS-EVENT-PIN-NEEDED %s " MACSTR
1067			  " [%s|%s|%s|%s|%s|%s]",
1068			  uuid, MAC2STR(dev->mac_addr), dev->device_name,
1069			  dev->manufacturer, dev->model_name,
1070			  dev->model_number, dev->serial_number,
1071			  wps_dev_type_bin2str(dev->pri_dev_type, devtype,
1072					       sizeof(devtype)));
1073	if (len > 0 && len < (int) sizeof(txt))
1074		wpa_printf(MSG_INFO, "%s", txt);
1075}
1076
1077
1078static void wpas_wps_set_sel_reg_cb(void *ctx, int sel_reg, u16 dev_passwd_id,
1079				    u16 sel_reg_config_methods)
1080{
1081#ifdef CONFIG_WPS_ER
1082	struct wpa_supplicant *wpa_s = ctx;
1083
1084	if (wpa_s->wps_er == NULL)
1085		return;
1086	wpa_printf(MSG_DEBUG, "WPS ER: SetSelectedRegistrar - sel_reg=%d "
1087		   "dev_password_id=%u sel_reg_config_methods=0x%x",
1088		   sel_reg, dev_passwd_id, sel_reg_config_methods);
1089	wps_er_set_sel_reg(wpa_s->wps_er, sel_reg, dev_passwd_id,
1090			   sel_reg_config_methods);
1091#endif /* CONFIG_WPS_ER */
1092}
1093
1094
1095static u16 wps_fix_config_methods(u16 config_methods)
1096{
1097#ifdef CONFIG_WPS2
1098	if ((config_methods &
1099	     (WPS_CONFIG_DISPLAY | WPS_CONFIG_VIRT_DISPLAY |
1100	      WPS_CONFIG_PHY_DISPLAY)) == WPS_CONFIG_DISPLAY) {
1101		wpa_printf(MSG_INFO, "WPS: Converting display to "
1102			   "virtual_display for WPS 2.0 compliance");
1103		config_methods |= WPS_CONFIG_VIRT_DISPLAY;
1104	}
1105	if ((config_methods &
1106	     (WPS_CONFIG_PUSHBUTTON | WPS_CONFIG_VIRT_PUSHBUTTON |
1107	      WPS_CONFIG_PHY_PUSHBUTTON)) == WPS_CONFIG_PUSHBUTTON) {
1108		wpa_printf(MSG_INFO, "WPS: Converting push_button to "
1109			   "virtual_push_button for WPS 2.0 compliance");
1110		config_methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
1111	}
1112#endif /* CONFIG_WPS2 */
1113
1114	return config_methods;
1115}
1116
1117
1118static void wpas_wps_set_uuid(struct wpa_supplicant *wpa_s,
1119			      struct wps_context *wps)
1120{
1121	wpa_printf(MSG_DEBUG, "WPS: Set UUID for interface %s", wpa_s->ifname);
1122	if (is_nil_uuid(wpa_s->conf->uuid)) {
1123		struct wpa_supplicant *first;
1124		first = wpa_s->global->ifaces;
1125		while (first && first->next)
1126			first = first->next;
1127		if (first && first != wpa_s) {
1128			if (wps != wpa_s->global->ifaces->wps)
1129				os_memcpy(wps->uuid,
1130					  wpa_s->global->ifaces->wps->uuid,
1131					  WPS_UUID_LEN);
1132			wpa_hexdump(MSG_DEBUG, "WPS: UUID from the first "
1133				    "interface", wps->uuid, WPS_UUID_LEN);
1134		} else {
1135			uuid_gen_mac_addr(wpa_s->own_addr, wps->uuid);
1136			wpa_hexdump(MSG_DEBUG, "WPS: UUID based on MAC "
1137				    "address", wps->uuid, WPS_UUID_LEN);
1138		}
1139	} else {
1140		os_memcpy(wps->uuid, wpa_s->conf->uuid, WPS_UUID_LEN);
1141		wpa_hexdump(MSG_DEBUG, "WPS: UUID based on configuration",
1142			    wps->uuid, WPS_UUID_LEN);
1143	}
1144}
1145
1146
1147static void wpas_wps_set_vendor_ext_m1(struct wpa_supplicant *wpa_s,
1148				       struct wps_context *wps)
1149{
1150	wpabuf_free(wps->dev.vendor_ext_m1);
1151	wps->dev.vendor_ext_m1 = NULL;
1152
1153	if (wpa_s->conf->wps_vendor_ext_m1) {
1154		wps->dev.vendor_ext_m1 =
1155			wpabuf_dup(wpa_s->conf->wps_vendor_ext_m1);
1156		if (!wps->dev.vendor_ext_m1) {
1157			wpa_printf(MSG_ERROR, "WPS: Cannot "
1158				   "allocate memory for vendor_ext_m1");
1159		}
1160	}
1161}
1162
1163
1164int wpas_wps_init(struct wpa_supplicant *wpa_s)
1165{
1166	struct wps_context *wps;
1167	struct wps_registrar_config rcfg;
1168	struct hostapd_hw_modes *modes;
1169	u16 m;
1170
1171	wps = os_zalloc(sizeof(*wps));
1172	if (wps == NULL)
1173		return -1;
1174
1175	wps->cred_cb = wpa_supplicant_wps_cred;
1176	wps->event_cb = wpa_supplicant_wps_event;
1177	wps->cb_ctx = wpa_s;
1178
1179	wps->dev.device_name = wpa_s->conf->device_name;
1180	wps->dev.manufacturer = wpa_s->conf->manufacturer;
1181	wps->dev.model_name = wpa_s->conf->model_name;
1182	wps->dev.model_number = wpa_s->conf->model_number;
1183	wps->dev.serial_number = wpa_s->conf->serial_number;
1184	wps->config_methods =
1185		wps_config_methods_str2bin(wpa_s->conf->config_methods);
1186	if ((wps->config_methods & (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
1187	    (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
1188		wpa_printf(MSG_ERROR, "WPS: Both Label and Display config "
1189			   "methods are not allowed at the same time");
1190		os_free(wps);
1191		return -1;
1192	}
1193	wps->config_methods = wps_fix_config_methods(wps->config_methods);
1194	wps->dev.config_methods = wps->config_methods;
1195	os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
1196		  WPS_DEV_TYPE_LEN);
1197
1198	wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
1199	os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
1200		  WPS_DEV_TYPE_LEN * wps->dev.num_sec_dev_types);
1201
1202	wpas_wps_set_vendor_ext_m1(wpa_s, wps);
1203
1204	wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
1205	modes = wpa_s->hw.modes;
1206	if (modes) {
1207		for (m = 0; m < wpa_s->hw.num_modes; m++) {
1208			if (modes[m].mode == HOSTAPD_MODE_IEEE80211B ||
1209			    modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1210				wps->dev.rf_bands |= WPS_RF_24GHZ;
1211			else if (modes[m].mode == HOSTAPD_MODE_IEEE80211A)
1212				wps->dev.rf_bands |= WPS_RF_50GHZ;
1213		}
1214	}
1215	if (wps->dev.rf_bands == 0) {
1216		/*
1217		 * Default to claiming support for both bands if the driver
1218		 * does not provide support for fetching supported bands.
1219		 */
1220		wps->dev.rf_bands = WPS_RF_24GHZ | WPS_RF_50GHZ;
1221	}
1222	os_memcpy(wps->dev.mac_addr, wpa_s->own_addr, ETH_ALEN);
1223	wpas_wps_set_uuid(wpa_s, wps);
1224
1225	wps->auth_types = WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK;
1226	wps->encr_types = WPS_ENCR_AES | WPS_ENCR_TKIP;
1227
1228	os_memset(&rcfg, 0, sizeof(rcfg));
1229	rcfg.new_psk_cb = wpas_wps_new_psk_cb;
1230	rcfg.pin_needed_cb = wpas_wps_pin_needed_cb;
1231	rcfg.set_sel_reg_cb = wpas_wps_set_sel_reg_cb;
1232	rcfg.cb_ctx = wpa_s;
1233
1234	wps->registrar = wps_registrar_init(wps, &rcfg);
1235	if (wps->registrar == NULL) {
1236		wpa_printf(MSG_DEBUG, "Failed to initialize WPS Registrar");
1237		os_free(wps);
1238		return -1;
1239	}
1240
1241	wpa_s->wps = wps;
1242
1243	return 0;
1244}
1245
1246
1247void wpas_wps_deinit(struct wpa_supplicant *wpa_s)
1248{
1249	eloop_cancel_timeout(wpas_wps_timeout, wpa_s, NULL);
1250	wpas_wps_clear_ap_info(wpa_s);
1251
1252	if (wpa_s->wps == NULL)
1253		return;
1254
1255#ifdef CONFIG_WPS_ER
1256	wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1257	wpa_s->wps_er = NULL;
1258#endif /* CONFIG_WPS_ER */
1259
1260	wps_registrar_deinit(wpa_s->wps->registrar);
1261	wpabuf_free(wpa_s->wps->dh_pubkey);
1262	wpabuf_free(wpa_s->wps->dh_privkey);
1263	wpabuf_free(wpa_s->wps->oob_conf.pubkey_hash);
1264	wpabuf_free(wpa_s->wps->oob_conf.dev_password);
1265	wpabuf_free(wpa_s->wps->dev.vendor_ext_m1);
1266	os_free(wpa_s->wps->network_key);
1267	os_free(wpa_s->wps);
1268	wpa_s->wps = NULL;
1269}
1270
1271
1272int wpas_wps_ssid_bss_match(struct wpa_supplicant *wpa_s,
1273			    struct wpa_ssid *ssid, struct wpa_bss *bss)
1274{
1275	struct wpabuf *wps_ie;
1276
1277	if (!(ssid->key_mgmt & WPA_KEY_MGMT_WPS))
1278		return -1;
1279
1280	wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1281	if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1282		if (!wps_ie) {
1283			wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1284			return 0;
1285		}
1286
1287		if (!wps_is_selected_pbc_registrar(wps_ie)) {
1288			wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1289				   "without active PBC Registrar");
1290			wpabuf_free(wps_ie);
1291			return 0;
1292		}
1293
1294		/* TODO: overlap detection */
1295		wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1296			   "(Active PBC)");
1297		wpabuf_free(wps_ie);
1298		return 1;
1299	}
1300
1301	if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1302		if (!wps_ie) {
1303			wpa_printf(MSG_DEBUG, "   skip - non-WPS AP");
1304			return 0;
1305		}
1306
1307		/*
1308		 * Start with WPS APs that advertise our address as an
1309		 * authorized MAC (v2.0) or active PIN Registrar (v1.0) and
1310		 * allow any WPS AP after couple of scans since some APs do not
1311		 * set Selected Registrar attribute properly when using
1312		 * external Registrar.
1313		 */
1314		if (!wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1)) {
1315			if (wpa_s->scan_runs < WPS_PIN_SCAN_IGNORE_SEL_REG) {
1316				wpa_printf(MSG_DEBUG, "   skip - WPS AP "
1317					   "without active PIN Registrar");
1318				wpabuf_free(wps_ie);
1319				return 0;
1320			}
1321			wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1322		} else {
1323			wpa_printf(MSG_DEBUG, "   selected based on WPS IE "
1324				   "(Authorized MAC or Active PIN)");
1325		}
1326		wpabuf_free(wps_ie);
1327		return 1;
1328	}
1329
1330	if (wps_ie) {
1331		wpa_printf(MSG_DEBUG, "   selected based on WPS IE");
1332		wpabuf_free(wps_ie);
1333		return 1;
1334	}
1335
1336	return -1;
1337}
1338
1339
1340int wpas_wps_ssid_wildcard_ok(struct wpa_supplicant *wpa_s,
1341			      struct wpa_ssid *ssid,
1342			      struct wpa_bss *bss)
1343{
1344	struct wpabuf *wps_ie = NULL;
1345	int ret = 0;
1346
1347	if (eap_is_wps_pbc_enrollee(&ssid->eap)) {
1348		wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1349		if (wps_ie && wps_is_selected_pbc_registrar(wps_ie)) {
1350			/* allow wildcard SSID for WPS PBC */
1351			ret = 1;
1352		}
1353	} else if (eap_is_wps_pin_enrollee(&ssid->eap)) {
1354		wps_ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1355		if (wps_ie &&
1356		    (wps_is_addr_authorized(wps_ie, wpa_s->own_addr, 1) ||
1357		     wpa_s->scan_runs >= WPS_PIN_SCAN_IGNORE_SEL_REG)) {
1358			/* allow wildcard SSID for WPS PIN */
1359			ret = 1;
1360		}
1361	}
1362
1363	if (!ret && ssid->bssid_set &&
1364	    os_memcmp(ssid->bssid, bss->bssid, ETH_ALEN) == 0) {
1365		/* allow wildcard SSID due to hardcoded BSSID match */
1366		ret = 1;
1367	}
1368
1369#ifdef CONFIG_WPS_STRICT
1370	if (wps_ie) {
1371		if (wps_validate_beacon_probe_resp(wps_ie, bss->beacon_ie_len >
1372						   0, bss->bssid) < 0)
1373			ret = 0;
1374		if (bss->beacon_ie_len) {
1375			struct wpabuf *bcn_wps;
1376			bcn_wps = wpa_bss_get_vendor_ie_multi_beacon(
1377				bss, WPS_IE_VENDOR_TYPE);
1378			if (bcn_wps == NULL) {
1379				wpa_printf(MSG_DEBUG, "WPS: Mandatory WPS IE "
1380					   "missing from AP Beacon");
1381				ret = 0;
1382			} else {
1383				if (wps_validate_beacon(wps_ie) < 0)
1384					ret = 0;
1385				wpabuf_free(bcn_wps);
1386			}
1387		}
1388	}
1389#endif /* CONFIG_WPS_STRICT */
1390
1391	wpabuf_free(wps_ie);
1392
1393	return ret;
1394}
1395
1396
1397int wpas_wps_scan_pbc_overlap(struct wpa_supplicant *wpa_s,
1398			      struct wpa_bss *selected, struct wpa_ssid *ssid)
1399{
1400	const u8 *sel_uuid, *uuid;
1401	struct wpabuf *wps_ie;
1402	int ret = 0;
1403	struct wpa_bss *bss;
1404
1405	if (!eap_is_wps_pbc_enrollee(&ssid->eap))
1406		return 0;
1407
1408	wpa_printf(MSG_DEBUG, "WPS: Check whether PBC session overlap is "
1409		   "present in scan results; selected BSSID " MACSTR,
1410		   MAC2STR(selected->bssid));
1411
1412	/* Make sure that only one AP is in active PBC mode */
1413	wps_ie = wpa_bss_get_vendor_ie_multi(selected, WPS_IE_VENDOR_TYPE);
1414	if (wps_ie) {
1415		sel_uuid = wps_get_uuid_e(wps_ie);
1416		wpa_hexdump(MSG_DEBUG, "WPS: UUID of the selected BSS",
1417			    sel_uuid, UUID_LEN);
1418	} else {
1419		wpa_printf(MSG_DEBUG, "WPS: Selected BSS does not include "
1420			   "WPS IE?!");
1421		sel_uuid = NULL;
1422	}
1423
1424	dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1425		struct wpabuf *ie;
1426		if (bss == selected)
1427			continue;
1428		ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1429		if (!ie)
1430			continue;
1431		if (!wps_is_selected_pbc_registrar(ie)) {
1432			wpabuf_free(ie);
1433			continue;
1434		}
1435		wpa_printf(MSG_DEBUG, "WPS: Another BSS in active PBC mode: "
1436			   MACSTR, MAC2STR(bss->bssid));
1437		uuid = wps_get_uuid_e(ie);
1438		wpa_hexdump(MSG_DEBUG, "WPS: UUID of the other BSS",
1439			    uuid, UUID_LEN);
1440		if (sel_uuid == NULL || uuid == NULL ||
1441		    os_memcmp(sel_uuid, uuid, UUID_LEN) != 0) {
1442			ret = 1; /* PBC overlap */
1443			wpa_msg(wpa_s, MSG_INFO, "WPS: PBC overlap detected: "
1444				MACSTR " and " MACSTR,
1445				MAC2STR(selected->bssid),
1446				MAC2STR(bss->bssid));
1447			wpabuf_free(ie);
1448			break;
1449		}
1450
1451		/* TODO: verify that this is reasonable dual-band situation */
1452
1453		wpabuf_free(ie);
1454	}
1455
1456	wpabuf_free(wps_ie);
1457
1458	return ret;
1459}
1460
1461
1462void wpas_wps_notify_scan_results(struct wpa_supplicant *wpa_s)
1463{
1464	struct wpa_bss *bss;
1465	unsigned int pbc = 0, auth = 0, pin = 0, wps = 0;
1466
1467	if (wpa_s->disconnected || wpa_s->wpa_state >= WPA_ASSOCIATED)
1468		return;
1469
1470	dl_list_for_each(bss, &wpa_s->bss, struct wpa_bss, list) {
1471		struct wpabuf *ie;
1472		ie = wpa_bss_get_vendor_ie_multi(bss, WPS_IE_VENDOR_TYPE);
1473		if (!ie)
1474			continue;
1475		if (wps_is_selected_pbc_registrar(ie))
1476			pbc++;
1477		else if (wps_is_addr_authorized(ie, wpa_s->own_addr, 0))
1478			auth++;
1479		else if (wps_is_selected_pin_registrar(ie))
1480			pin++;
1481		else
1482			wps++;
1483		wpabuf_free(ie);
1484	}
1485
1486	if (pbc)
1487		wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PBC);
1488	else if (auth)
1489		wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_AUTH);
1490	else if (pin)
1491		wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE_PIN);
1492	else if (wps)
1493		wpa_msg_ctrl(wpa_s, MSG_INFO, WPS_EVENT_AP_AVAILABLE);
1494}
1495
1496
1497int wpas_wps_searching(struct wpa_supplicant *wpa_s)
1498{
1499	struct wpa_ssid *ssid;
1500
1501	for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1502		if ((ssid->key_mgmt & WPA_KEY_MGMT_WPS) && !ssid->disabled)
1503			return 1;
1504	}
1505
1506	return 0;
1507}
1508
1509
1510int wpas_wps_scan_result_text(const u8 *ies, size_t ies_len, char *buf,
1511			      char *end)
1512{
1513	struct wpabuf *wps_ie;
1514	int ret;
1515
1516	wps_ie = ieee802_11_vendor_ie_concat(ies, ies_len, WPS_DEV_OUI_WFA);
1517	if (wps_ie == NULL)
1518		return 0;
1519
1520	ret = wps_attr_text(wps_ie, buf, end);
1521	wpabuf_free(wps_ie);
1522	return ret;
1523}
1524
1525
1526int wpas_wps_er_start(struct wpa_supplicant *wpa_s, const char *filter)
1527{
1528#ifdef CONFIG_WPS_ER
1529	if (wpa_s->wps_er) {
1530		wps_er_refresh(wpa_s->wps_er);
1531		return 0;
1532	}
1533	wpa_s->wps_er = wps_er_init(wpa_s->wps, wpa_s->ifname, filter);
1534	if (wpa_s->wps_er == NULL)
1535		return -1;
1536	return 0;
1537#else /* CONFIG_WPS_ER */
1538	return 0;
1539#endif /* CONFIG_WPS_ER */
1540}
1541
1542
1543int wpas_wps_er_stop(struct wpa_supplicant *wpa_s)
1544{
1545#ifdef CONFIG_WPS_ER
1546	wps_er_deinit(wpa_s->wps_er, NULL, NULL);
1547	wpa_s->wps_er = NULL;
1548#endif /* CONFIG_WPS_ER */
1549	return 0;
1550}
1551
1552
1553#ifdef CONFIG_WPS_ER
1554int wpas_wps_er_add_pin(struct wpa_supplicant *wpa_s, const u8 *addr,
1555			const char *uuid, const char *pin)
1556{
1557	u8 u[UUID_LEN];
1558	int any = 0;
1559
1560	if (os_strcmp(uuid, "any") == 0)
1561		any = 1;
1562	else if (uuid_str2bin(uuid, u))
1563		return -1;
1564	return wps_registrar_add_pin(wpa_s->wps->registrar, addr,
1565				     any ? NULL : u,
1566				     (const u8 *) pin, os_strlen(pin), 300);
1567}
1568
1569
1570int wpas_wps_er_pbc(struct wpa_supplicant *wpa_s, const char *uuid)
1571{
1572	u8 u[UUID_LEN];
1573
1574	if (uuid_str2bin(uuid, u))
1575		return -1;
1576	return wps_er_pbc(wpa_s->wps_er, u);
1577}
1578
1579
1580int wpas_wps_er_learn(struct wpa_supplicant *wpa_s, const char *uuid,
1581		      const char *pin)
1582{
1583	u8 u[UUID_LEN];
1584
1585	if (uuid_str2bin(uuid, u))
1586		return -1;
1587	return wps_er_learn(wpa_s->wps_er, u, (const u8 *) pin,
1588			    os_strlen(pin));
1589}
1590
1591
1592int wpas_wps_er_set_config(struct wpa_supplicant *wpa_s, const char *uuid,
1593			   int id)
1594{
1595	u8 u[UUID_LEN];
1596	struct wpa_ssid *ssid;
1597	struct wps_credential cred;
1598
1599	if (uuid_str2bin(uuid, u))
1600		return -1;
1601	ssid = wpa_config_get_network(wpa_s->conf, id);
1602	if (ssid == NULL || ssid->ssid == NULL)
1603		return -1;
1604
1605	os_memset(&cred, 0, sizeof(cred));
1606	if (ssid->ssid_len > 32)
1607		return -1;
1608	os_memcpy(cred.ssid, ssid->ssid, ssid->ssid_len);
1609	cred.ssid_len = ssid->ssid_len;
1610	if (ssid->key_mgmt & WPA_KEY_MGMT_PSK) {
1611		cred.auth_type = (ssid->proto & WPA_PROTO_RSN) ?
1612			WPS_AUTH_WPA2PSK : WPS_AUTH_WPAPSK;
1613		if (ssid->pairwise_cipher & WPA_CIPHER_CCMP)
1614			cred.encr_type = WPS_ENCR_AES;
1615		else
1616			cred.encr_type = WPS_ENCR_TKIP;
1617		if (ssid->passphrase) {
1618			cred.key_len = os_strlen(ssid->passphrase);
1619			if (cred.key_len >= 64)
1620				return -1;
1621			os_memcpy(cred.key, ssid->passphrase, cred.key_len);
1622		} else if (ssid->psk_set) {
1623			cred.key_len = 32;
1624			os_memcpy(cred.key, ssid->psk, 32);
1625		} else
1626			return -1;
1627	} else {
1628		cred.auth_type = WPS_AUTH_OPEN;
1629		cred.encr_type = WPS_ENCR_NONE;
1630	}
1631	return wps_er_set_config(wpa_s->wps_er, u, &cred);
1632}
1633
1634
1635int wpas_wps_er_config(struct wpa_supplicant *wpa_s, const char *uuid,
1636		       const char *pin, struct wps_new_ap_settings *settings)
1637{
1638	u8 u[UUID_LEN];
1639	struct wps_credential cred;
1640	size_t len;
1641
1642	if (uuid_str2bin(uuid, u))
1643		return -1;
1644	if (settings->ssid_hex == NULL || settings->auth == NULL ||
1645	    settings->encr == NULL || settings->key_hex == NULL)
1646		return -1;
1647
1648	os_memset(&cred, 0, sizeof(cred));
1649	len = os_strlen(settings->ssid_hex);
1650	if ((len & 1) || len > 2 * sizeof(cred.ssid) ||
1651	    hexstr2bin(settings->ssid_hex, cred.ssid, len / 2))
1652		return -1;
1653	cred.ssid_len = len / 2;
1654
1655	len = os_strlen(settings->key_hex);
1656	if ((len & 1) || len > 2 * sizeof(cred.key) ||
1657	    hexstr2bin(settings->key_hex, cred.key, len / 2))
1658		return -1;
1659	cred.key_len = len / 2;
1660
1661	if (os_strcmp(settings->auth, "OPEN") == 0)
1662		cred.auth_type = WPS_AUTH_OPEN;
1663	else if (os_strcmp(settings->auth, "WPAPSK") == 0)
1664		cred.auth_type = WPS_AUTH_WPAPSK;
1665	else if (os_strcmp(settings->auth, "WPA2PSK") == 0)
1666		cred.auth_type = WPS_AUTH_WPA2PSK;
1667	else
1668		return -1;
1669
1670	if (os_strcmp(settings->encr, "NONE") == 0)
1671		cred.encr_type = WPS_ENCR_NONE;
1672	else if (os_strcmp(settings->encr, "WEP") == 0)
1673		cred.encr_type = WPS_ENCR_WEP;
1674	else if (os_strcmp(settings->encr, "TKIP") == 0)
1675		cred.encr_type = WPS_ENCR_TKIP;
1676	else if (os_strcmp(settings->encr, "CCMP") == 0)
1677		cred.encr_type = WPS_ENCR_AES;
1678	else
1679		return -1;
1680
1681	return wps_er_config(wpa_s->wps_er, u, (const u8 *) pin,
1682			     os_strlen(pin), &cred);
1683}
1684
1685
1686#ifdef CONFIG_WPS_NFC
1687struct wpabuf * wpas_wps_er_nfc_config_token(struct wpa_supplicant *wpa_s,
1688					     int ndef, const char *uuid)
1689{
1690	struct wpabuf *ret;
1691	u8 u[UUID_LEN];
1692
1693	if (!wpa_s->wps_er)
1694		return NULL;
1695
1696	if (uuid_str2bin(uuid, u))
1697		return NULL;
1698
1699	ret = wps_er_nfc_config_token(wpa_s->wps_er, u);
1700	if (ndef && ret) {
1701		struct wpabuf *tmp;
1702		tmp = ndef_build_wifi(ret);
1703		wpabuf_free(ret);
1704		if (tmp == NULL)
1705			return NULL;
1706		ret = tmp;
1707	}
1708
1709	return ret;
1710}
1711#endif /* CONFIG_WPS_NFC */
1712
1713
1714static int callbacks_pending = 0;
1715
1716static void wpas_wps_terminate_cb(void *ctx)
1717{
1718	wpa_printf(MSG_DEBUG, "WPS ER: Terminated");
1719	if (--callbacks_pending <= 0)
1720		eloop_terminate();
1721}
1722#endif /* CONFIG_WPS_ER */
1723
1724
1725int wpas_wps_terminate_pending(struct wpa_supplicant *wpa_s)
1726{
1727#ifdef CONFIG_WPS_ER
1728	if (wpa_s->wps_er) {
1729		callbacks_pending++;
1730		wps_er_deinit(wpa_s->wps_er, wpas_wps_terminate_cb, wpa_s);
1731		wpa_s->wps_er = NULL;
1732		return 1;
1733	}
1734#endif /* CONFIG_WPS_ER */
1735	return 0;
1736}
1737
1738
1739int wpas_wps_in_progress(struct wpa_supplicant *wpa_s)
1740{
1741	struct wpa_ssid *ssid;
1742
1743	for (ssid = wpa_s->conf->ssid; ssid; ssid = ssid->next) {
1744		if (!ssid->disabled && ssid->key_mgmt == WPA_KEY_MGMT_WPS)
1745			return 1;
1746	}
1747
1748	return 0;
1749}
1750
1751
1752void wpas_wps_update_config(struct wpa_supplicant *wpa_s)
1753{
1754	struct wps_context *wps = wpa_s->wps;
1755
1756	if (wps == NULL)
1757		return;
1758
1759	if (wpa_s->conf->changed_parameters & CFG_CHANGED_CONFIG_METHODS) {
1760		wps->config_methods = wps_config_methods_str2bin(
1761			wpa_s->conf->config_methods);
1762		if ((wps->config_methods &
1763		     (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) ==
1764		    (WPS_CONFIG_DISPLAY | WPS_CONFIG_LABEL)) {
1765			wpa_printf(MSG_ERROR, "WPS: Both Label and Display "
1766				   "config methods are not allowed at the "
1767				   "same time");
1768			wps->config_methods &= ~WPS_CONFIG_LABEL;
1769		}
1770	}
1771	wps->config_methods = wps_fix_config_methods(wps->config_methods);
1772	wps->dev.config_methods = wps->config_methods;
1773
1774	if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_TYPE)
1775		os_memcpy(wps->dev.pri_dev_type, wpa_s->conf->device_type,
1776			  WPS_DEV_TYPE_LEN);
1777
1778	if (wpa_s->conf->changed_parameters & CFG_CHANGED_SEC_DEVICE_TYPE) {
1779		wps->dev.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
1780		os_memcpy(wps->dev.sec_dev_type, wpa_s->conf->sec_device_type,
1781			  wps->dev.num_sec_dev_types * WPS_DEV_TYPE_LEN);
1782	}
1783
1784	if (wpa_s->conf->changed_parameters & CFG_CHANGED_VENDOR_EXTENSION)
1785		wpas_wps_set_vendor_ext_m1(wpa_s, wps);
1786
1787	if (wpa_s->conf->changed_parameters & CFG_CHANGED_OS_VERSION)
1788		wps->dev.os_version = WPA_GET_BE32(wpa_s->conf->os_version);
1789
1790	if (wpa_s->conf->changed_parameters & CFG_CHANGED_UUID)
1791		wpas_wps_set_uuid(wpa_s, wps);
1792
1793	if (wpa_s->conf->changed_parameters &
1794	    (CFG_CHANGED_DEVICE_NAME | CFG_CHANGED_WPS_STRING)) {
1795		/* Update pointers to make sure they refer current values */
1796		wps->dev.device_name = wpa_s->conf->device_name;
1797		wps->dev.manufacturer = wpa_s->conf->manufacturer;
1798		wps->dev.model_name = wpa_s->conf->model_name;
1799		wps->dev.model_number = wpa_s->conf->model_number;
1800		wps->dev.serial_number = wpa_s->conf->serial_number;
1801	}
1802}
1803
1804
1805#ifdef CONFIG_WPS_NFC
1806
1807struct wpabuf * wpas_wps_nfc_token(struct wpa_supplicant *wpa_s, int ndef)
1808{
1809	return wps_nfc_token_gen(ndef, &wpa_s->conf->wps_nfc_dev_pw_id,
1810				 &wpa_s->conf->wps_nfc_dh_pubkey,
1811				 &wpa_s->conf->wps_nfc_dh_privkey,
1812				 &wpa_s->conf->wps_nfc_dev_pw);
1813}
1814
1815
1816int wpas_wps_start_nfc(struct wpa_supplicant *wpa_s, const u8 *bssid)
1817{
1818	struct wps_context *wps = wpa_s->wps;
1819	char pw[32 * 2 + 1];
1820
1821	if (wpa_s->conf->wps_nfc_dh_pubkey == NULL ||
1822	    wpa_s->conf->wps_nfc_dh_privkey == NULL ||
1823	    wpa_s->conf->wps_nfc_dev_pw == NULL)
1824		return -1;
1825
1826	dh5_free(wps->dh_ctx);
1827	wpabuf_free(wps->dh_pubkey);
1828	wpabuf_free(wps->dh_privkey);
1829	wps->dh_privkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_privkey);
1830	wps->dh_pubkey = wpabuf_dup(wpa_s->conf->wps_nfc_dh_pubkey);
1831	if (wps->dh_privkey == NULL || wps->dh_pubkey == NULL) {
1832		wps->dh_ctx = NULL;
1833		wpabuf_free(wps->dh_pubkey);
1834		wps->dh_pubkey = NULL;
1835		wpabuf_free(wps->dh_privkey);
1836		wps->dh_privkey = NULL;
1837		return -1;
1838	}
1839	wps->dh_ctx = dh5_init_fixed(wps->dh_privkey, wps->dh_pubkey);
1840	if (wps->dh_ctx == NULL)
1841		return -1;
1842
1843	wpa_snprintf_hex_uppercase(pw, sizeof(pw),
1844				   wpabuf_head(wpa_s->conf->wps_nfc_dev_pw),
1845				   wpabuf_len(wpa_s->conf->wps_nfc_dev_pw));
1846	return wpas_wps_start_pin(wpa_s, bssid, pw, 0,
1847				  wpa_s->conf->wps_nfc_dev_pw_id);
1848}
1849
1850
1851static int wpas_wps_use_cred(struct wpa_supplicant *wpa_s,
1852			     struct wps_parse_attr *attr)
1853{
1854	if (wps_oob_use_cred(wpa_s->wps, attr) < 0)
1855		return -1;
1856
1857	if (wpa_s->wpa_state == WPA_INTERFACE_DISABLED)
1858		return 0;
1859
1860	wpa_printf(MSG_DEBUG, "WPS: Request reconnection with new network "
1861		   "based on the received credential added");
1862	wpa_s->normal_scans = 0;
1863	wpa_supplicant_reinit_autoscan(wpa_s);
1864	wpa_s->disconnected = 0;
1865	wpa_s->reassociate = 1;
1866	wpa_supplicant_req_scan(wpa_s, 0, 0);
1867
1868	return 0;
1869}
1870
1871
1872#ifdef CONFIG_WPS_ER
1873static int wpas_wps_add_nfc_password_token(struct wpa_supplicant *wpa_s,
1874					   struct wps_parse_attr *attr)
1875{
1876	return wps_registrar_add_nfc_password_token(
1877		wpa_s->wps->registrar, attr->oob_dev_password,
1878		attr->oob_dev_password_len);
1879}
1880#endif /* CONFIG_WPS_ER */
1881
1882
1883static int wpas_wps_nfc_tag_process(struct wpa_supplicant *wpa_s,
1884				    const struct wpabuf *wps)
1885{
1886	struct wps_parse_attr attr;
1887
1888	wpa_hexdump_buf(MSG_DEBUG, "WPS: Received NFC tag payload", wps);
1889
1890	if (wps_parse_msg(wps, &attr)) {
1891		wpa_printf(MSG_DEBUG, "WPS: Ignore invalid data from NFC tag");
1892		return -1;
1893	}
1894
1895	if (attr.num_cred)
1896		return wpas_wps_use_cred(wpa_s, &attr);
1897
1898#ifdef CONFIG_WPS_ER
1899	if (attr.oob_dev_password)
1900		return wpas_wps_add_nfc_password_token(wpa_s, &attr);
1901#endif /* CONFIG_WPS_ER */
1902
1903	wpa_printf(MSG_DEBUG, "WPS: Ignore unrecognized NFC tag");
1904	return -1;
1905}
1906
1907
1908int wpas_wps_nfc_tag_read(struct wpa_supplicant *wpa_s,
1909			  const struct wpabuf *data)
1910{
1911	const struct wpabuf *wps = data;
1912	struct wpabuf *tmp = NULL;
1913	int ret;
1914
1915	if (wpabuf_len(data) < 4)
1916		return -1;
1917
1918	if (*wpabuf_head_u8(data) != 0x10) {
1919		/* Assume this contains full NDEF record */
1920		tmp = ndef_parse_wifi(data);
1921		if (tmp == NULL) {
1922			wpa_printf(MSG_DEBUG, "WPS: Could not parse NDEF");
1923			return -1;
1924		}
1925		wps = tmp;
1926	}
1927
1928	ret = wpas_wps_nfc_tag_process(wpa_s, wps);
1929	wpabuf_free(tmp);
1930	return ret;
1931}
1932
1933#endif /* CONFIG_WPS_NFC */
1934
1935
1936extern int wpa_debug_level;
1937
1938static void wpas_wps_dump_ap_info(struct wpa_supplicant *wpa_s)
1939{
1940	size_t i;
1941	struct os_time now;
1942
1943	if (wpa_debug_level > MSG_DEBUG)
1944		return;
1945
1946	if (wpa_s->wps_ap == NULL)
1947		return;
1948
1949	os_get_time(&now);
1950
1951	for (i = 0; i < wpa_s->num_wps_ap; i++) {
1952		struct wps_ap_info *ap = &wpa_s->wps_ap[i];
1953		struct wpa_blacklist *e = wpa_blacklist_get(wpa_s, ap->bssid);
1954
1955		wpa_printf(MSG_DEBUG, "WPS: AP[%d] " MACSTR " type=%d "
1956			   "tries=%d last_attempt=%d sec ago blacklist=%d",
1957			   (int) i, MAC2STR(ap->bssid), ap->type, ap->tries,
1958			   ap->last_attempt.sec > 0 ?
1959			   (int) now.sec - (int) ap->last_attempt.sec : -1,
1960			   e ? e->count : 0);
1961	}
1962}
1963
1964
1965static struct wps_ap_info * wpas_wps_get_ap_info(struct wpa_supplicant *wpa_s,
1966						 const u8 *bssid)
1967{
1968	size_t i;
1969
1970	if (wpa_s->wps_ap == NULL)
1971		return NULL;
1972
1973	for (i = 0; i < wpa_s->num_wps_ap; i++) {
1974		struct wps_ap_info *ap = &wpa_s->wps_ap[i];
1975		if (os_memcmp(ap->bssid, bssid, ETH_ALEN) == 0)
1976			return ap;
1977	}
1978
1979	return NULL;
1980}
1981
1982
1983static void wpas_wps_update_ap_info_bss(struct wpa_supplicant *wpa_s,
1984					struct wpa_scan_res *res)
1985{
1986	struct wpabuf *wps;
1987	enum wps_ap_info_type type;
1988	struct wps_ap_info *ap;
1989	int r;
1990
1991	if (wpa_scan_get_vendor_ie(res, WPS_IE_VENDOR_TYPE) == NULL)
1992		return;
1993
1994	wps = wpa_scan_get_vendor_ie_multi(res, WPS_IE_VENDOR_TYPE);
1995	if (wps == NULL)
1996		return;
1997
1998	r = wps_is_addr_authorized(wps, wpa_s->own_addr, 1);
1999	if (r == 2)
2000		type = WPS_AP_SEL_REG_OUR;
2001	else if (r == 1)
2002		type = WPS_AP_SEL_REG;
2003	else
2004		type = WPS_AP_NOT_SEL_REG;
2005
2006	wpabuf_free(wps);
2007
2008	ap = wpas_wps_get_ap_info(wpa_s, res->bssid);
2009	if (ap) {
2010		if (ap->type != type) {
2011			wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR
2012				   " changed type %d -> %d",
2013				   MAC2STR(res->bssid), ap->type, type);
2014			ap->type = type;
2015			if (type != WPS_AP_NOT_SEL_REG)
2016				wpa_blacklist_del(wpa_s, ap->bssid);
2017		}
2018		return;
2019	}
2020
2021	ap = os_realloc_array(wpa_s->wps_ap, wpa_s->num_wps_ap + 1,
2022			      sizeof(struct wps_ap_info));
2023	if (ap == NULL)
2024		return;
2025
2026	wpa_s->wps_ap = ap;
2027	ap = &wpa_s->wps_ap[wpa_s->num_wps_ap];
2028	wpa_s->num_wps_ap++;
2029
2030	os_memset(ap, 0, sizeof(*ap));
2031	os_memcpy(ap->bssid, res->bssid, ETH_ALEN);
2032	ap->type = type;
2033	wpa_printf(MSG_DEBUG, "WPS: AP " MACSTR " type %d added",
2034		   MAC2STR(ap->bssid), ap->type);
2035}
2036
2037
2038void wpas_wps_update_ap_info(struct wpa_supplicant *wpa_s,
2039			     struct wpa_scan_results *scan_res)
2040{
2041	size_t i;
2042
2043	for (i = 0; i < scan_res->num; i++)
2044		wpas_wps_update_ap_info_bss(wpa_s, scan_res->res[i]);
2045
2046	wpas_wps_dump_ap_info(wpa_s);
2047}
2048
2049
2050void wpas_wps_notify_assoc(struct wpa_supplicant *wpa_s, const u8 *bssid)
2051{
2052	struct wps_ap_info *ap;
2053	if (!wpa_s->wps_ap_iter)
2054		return;
2055	ap = wpas_wps_get_ap_info(wpa_s, bssid);
2056	if (ap == NULL)
2057		return;
2058	ap->tries++;
2059	os_get_time(&ap->last_attempt);
2060}
2061