1/*
2 * hostapd / WPS integration
3 * Copyright (c) 2008-2016, 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 "utils/includes.h"
10
11#include "utils/common.h"
12#include "utils/eloop.h"
13#include "utils/uuid.h"
14#include "common/wpa_ctrl.h"
15#include "common/ieee802_11_defs.h"
16#include "common/ieee802_11_common.h"
17#include "eapol_auth/eapol_auth_sm.h"
18#include "eapol_auth/eapol_auth_sm_i.h"
19#include "wps/wps.h"
20#include "wps/wps_defs.h"
21#include "wps/wps_dev_attr.h"
22#include "wps/wps_attr_parse.h"
23#include "hostapd.h"
24#include "ap_config.h"
25#include "ap_drv_ops.h"
26#include "beacon.h"
27#include "sta_info.h"
28#include "wps_hostapd.h"
29
30
31#ifdef CONFIG_WPS_UPNP
32#include "wps/wps_upnp.h"
33static int hostapd_wps_upnp_init(struct hostapd_data *hapd,
34				 struct wps_context *wps);
35static void hostapd_wps_upnp_deinit(struct hostapd_data *hapd);
36#endif /* CONFIG_WPS_UPNP */
37
38static int hostapd_wps_probe_req_rx(void *ctx, const u8 *addr, const u8 *da,
39				    const u8 *bssid,
40				    const u8 *ie, size_t ie_len,
41				    int ssi_signal);
42static void hostapd_wps_ap_pin_timeout(void *eloop_data, void *user_ctx);
43static void hostapd_wps_nfc_clear(struct wps_context *wps);
44
45
46struct wps_for_each_data {
47	int (*func)(struct hostapd_data *h, void *ctx);
48	void *ctx;
49	struct hostapd_data *calling_hapd;
50};
51
52
53static int wps_for_each(struct hostapd_iface *iface, void *ctx)
54{
55	struct wps_for_each_data *data = ctx;
56	size_t j;
57
58	if (iface == NULL)
59		return 0;
60	for (j = 0; j < iface->num_bss; j++) {
61		struct hostapd_data *hapd = iface->bss[j];
62		int ret;
63
64		if (hapd != data->calling_hapd &&
65		    (hapd->conf->wps_independent ||
66		     data->calling_hapd->conf->wps_independent))
67			continue;
68
69		ret = data->func(hapd, data->ctx);
70		if (ret)
71			return ret;
72	}
73
74	return 0;
75}
76
77
78static int hostapd_wps_for_each(struct hostapd_data *hapd,
79				int (*func)(struct hostapd_data *h, void *ctx),
80				void *ctx)
81{
82	struct hostapd_iface *iface = hapd->iface;
83	struct wps_for_each_data data;
84	data.func = func;
85	data.ctx = ctx;
86	data.calling_hapd = hapd;
87	if (iface->interfaces == NULL ||
88	    iface->interfaces->for_each_interface == NULL)
89		return wps_for_each(iface, &data);
90	return iface->interfaces->for_each_interface(iface->interfaces,
91						     wps_for_each, &data);
92}
93
94
95static int hostapd_wps_new_psk_cb(void *ctx, const u8 *mac_addr,
96				  const u8 *p2p_dev_addr, const u8 *psk,
97				  size_t psk_len)
98{
99	struct hostapd_data *hapd = ctx;
100	struct hostapd_wpa_psk *p;
101	struct hostapd_ssid *ssid = &hapd->conf->ssid;
102
103	if (is_zero_ether_addr(p2p_dev_addr)) {
104		wpa_printf(MSG_DEBUG,
105			   "Received new WPA/WPA2-PSK from WPS for STA " MACSTR,
106			   MAC2STR(mac_addr));
107	} else {
108		wpa_printf(MSG_DEBUG,
109			   "Received new WPA/WPA2-PSK from WPS for STA " MACSTR
110			   " P2P Device Addr " MACSTR,
111			   MAC2STR(mac_addr), MAC2STR(p2p_dev_addr));
112	}
113	wpa_hexdump_key(MSG_DEBUG, "Per-device PSK", psk, psk_len);
114
115	if (psk_len != PMK_LEN) {
116		wpa_printf(MSG_DEBUG, "Unexpected PSK length %lu",
117			   (unsigned long) psk_len);
118		return -1;
119	}
120
121	/* Add the new PSK to runtime PSK list */
122	p = os_zalloc(sizeof(*p));
123	if (p == NULL)
124		return -1;
125	os_memcpy(p->addr, mac_addr, ETH_ALEN);
126	os_memcpy(p->p2p_dev_addr, p2p_dev_addr, ETH_ALEN);
127	os_memcpy(p->psk, psk, PMK_LEN);
128
129	if (hapd->new_psk_cb) {
130		hapd->new_psk_cb(hapd->new_psk_cb_ctx, mac_addr, p2p_dev_addr,
131				 psk, psk_len);
132	}
133
134	p->next = ssid->wpa_psk;
135	ssid->wpa_psk = p;
136
137	if (ssid->wpa_psk_file) {
138		FILE *f;
139		char hex[PMK_LEN * 2 + 1];
140		/* Add the new PSK to PSK list file */
141		f = fopen(ssid->wpa_psk_file, "a");
142		if (f == NULL) {
143			wpa_printf(MSG_DEBUG, "Failed to add the PSK to "
144				   "'%s'", ssid->wpa_psk_file);
145			return -1;
146		}
147
148		wpa_snprintf_hex(hex, sizeof(hex), psk, psk_len);
149		fprintf(f, MACSTR " %s\n", MAC2STR(mac_addr), hex);
150		fclose(f);
151	}
152
153	return 0;
154}
155
156
157static int hostapd_wps_set_ie_cb(void *ctx, struct wpabuf *beacon_ie,
158				 struct wpabuf *probe_resp_ie)
159{
160	struct hostapd_data *hapd = ctx;
161	wpabuf_free(hapd->wps_beacon_ie);
162	hapd->wps_beacon_ie = beacon_ie;
163	wpabuf_free(hapd->wps_probe_resp_ie);
164	hapd->wps_probe_resp_ie = probe_resp_ie;
165	if (hapd->beacon_set_done)
166		ieee802_11_set_beacon(hapd);
167	return hostapd_set_ap_wps_ie(hapd);
168}
169
170
171static void hostapd_wps_pin_needed_cb(void *ctx, const u8 *uuid_e,
172				      const struct wps_device_data *dev)
173{
174	struct hostapd_data *hapd = ctx;
175	char uuid[40], txt[400];
176	int len;
177	char devtype[WPS_DEV_TYPE_BUFSIZE];
178	if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
179		return;
180	wpa_printf(MSG_DEBUG, "WPS: PIN needed for E-UUID %s", uuid);
181	len = os_snprintf(txt, sizeof(txt), WPS_EVENT_PIN_NEEDED
182			  "%s " MACSTR " [%s|%s|%s|%s|%s|%s]",
183			  uuid, MAC2STR(dev->mac_addr), dev->device_name,
184			  dev->manufacturer, dev->model_name,
185			  dev->model_number, dev->serial_number,
186			  wps_dev_type_bin2str(dev->pri_dev_type, devtype,
187					       sizeof(devtype)));
188	if (!os_snprintf_error(sizeof(txt), len))
189		wpa_msg(hapd->msg_ctx, MSG_INFO, "%s", txt);
190
191	if (hapd->conf->wps_pin_requests) {
192		FILE *f;
193		struct os_time t;
194		f = fopen(hapd->conf->wps_pin_requests, "a");
195		if (f == NULL)
196			return;
197		os_get_time(&t);
198		fprintf(f, "%ld\t%s\t" MACSTR "\t%s\t%s\t%s\t%s\t%s"
199			"\t%s\n",
200			t.sec, uuid, MAC2STR(dev->mac_addr), dev->device_name,
201			dev->manufacturer, dev->model_name, dev->model_number,
202			dev->serial_number,
203			wps_dev_type_bin2str(dev->pri_dev_type, devtype,
204					     sizeof(devtype)));
205		fclose(f);
206	}
207}
208
209
210struct wps_stop_reg_data {
211	struct hostapd_data *current_hapd;
212	const u8 *uuid_e;
213	const u8 *dev_pw;
214	size_t dev_pw_len;
215};
216
217static int wps_stop_registrar(struct hostapd_data *hapd, void *ctx)
218{
219	struct wps_stop_reg_data *data = ctx;
220	if (hapd != data->current_hapd && hapd->wps != NULL)
221		wps_registrar_complete(hapd->wps->registrar, data->uuid_e,
222				       data->dev_pw, data->dev_pw_len);
223	return 0;
224}
225
226
227static void hostapd_wps_reg_success_cb(void *ctx, const u8 *mac_addr,
228				       const u8 *uuid_e, const u8 *dev_pw,
229				       size_t dev_pw_len)
230{
231	struct hostapd_data *hapd = ctx;
232	char uuid[40];
233	struct wps_stop_reg_data data;
234	if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
235		return;
236	wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_REG_SUCCESS MACSTR " %s",
237		MAC2STR(mac_addr), uuid);
238	if (hapd->wps_reg_success_cb)
239		hapd->wps_reg_success_cb(hapd->wps_reg_success_cb_ctx,
240					 mac_addr, uuid_e);
241	data.current_hapd = hapd;
242	data.uuid_e = uuid_e;
243	data.dev_pw = dev_pw;
244	data.dev_pw_len = dev_pw_len;
245	hostapd_wps_for_each(hapd, wps_stop_registrar, &data);
246}
247
248
249static void hostapd_wps_enrollee_seen_cb(void *ctx, const u8 *addr,
250					 const u8 *uuid_e,
251					 const u8 *pri_dev_type,
252					 u16 config_methods,
253					 u16 dev_password_id, u8 request_type,
254					 const char *dev_name)
255{
256	struct hostapd_data *hapd = ctx;
257	char uuid[40];
258	char devtype[WPS_DEV_TYPE_BUFSIZE];
259	if (uuid_bin2str(uuid_e, uuid, sizeof(uuid)))
260		return;
261	if (dev_name == NULL)
262		dev_name = "";
263	wpa_msg_ctrl(hapd->msg_ctx, MSG_INFO, WPS_EVENT_ENROLLEE_SEEN MACSTR
264		     " %s %s 0x%x %u %u [%s]",
265		     MAC2STR(addr), uuid,
266		     wps_dev_type_bin2str(pri_dev_type, devtype,
267					  sizeof(devtype)),
268		     config_methods, dev_password_id, request_type, dev_name);
269}
270
271
272static int str_starts(const char *str, const char *start)
273{
274	return os_strncmp(str, start, os_strlen(start)) == 0;
275}
276
277
278static void wps_reload_config(void *eloop_data, void *user_ctx)
279{
280	struct hostapd_iface *iface = eloop_data;
281
282	wpa_printf(MSG_DEBUG, "WPS: Reload configuration data");
283	if (iface->interfaces == NULL ||
284	    iface->interfaces->reload_config(iface) < 0) {
285		wpa_printf(MSG_WARNING, "WPS: Failed to reload the updated "
286			   "configuration");
287	}
288}
289
290
291void hostapd_wps_eap_completed(struct hostapd_data *hapd)
292{
293	/*
294	 * Reduce race condition of the station trying to reconnect immediately
295	 * after AP reconfiguration through WPS by rescheduling the reload
296	 * timeout to happen after EAP completion rather than the originally
297	 * scheduled 100 ms after new configuration became known.
298	 */
299	if (eloop_deplete_timeout(0, 0, wps_reload_config, hapd->iface, NULL) ==
300	    1)
301		wpa_printf(MSG_DEBUG, "WPS: Reschedule immediate configuration reload");
302}
303
304
305static void hapd_new_ap_event(struct hostapd_data *hapd, const u8 *attr,
306			      size_t attr_len)
307{
308	size_t blen = attr_len * 2 + 1;
309	char *buf = os_malloc(blen);
310	if (buf) {
311		wpa_snprintf_hex(buf, blen, attr, attr_len);
312		wpa_msg(hapd->msg_ctx, MSG_INFO,
313			WPS_EVENT_NEW_AP_SETTINGS "%s", buf);
314		os_free(buf);
315	}
316}
317
318
319static int hapd_wps_reconfig_in_memory(struct hostapd_data *hapd,
320				       const struct wps_credential *cred)
321{
322	struct hostapd_bss_config *bss = hapd->conf;
323
324	wpa_printf(MSG_DEBUG, "WPS: Updating in-memory configuration");
325
326	bss->wps_state = 2;
327	if (cred->ssid_len <= SSID_MAX_LEN) {
328		os_memcpy(bss->ssid.ssid, cred->ssid, cred->ssid_len);
329		bss->ssid.ssid_len = cred->ssid_len;
330		bss->ssid.ssid_set = 1;
331	}
332
333	if ((cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA2PSK)) &&
334	    (cred->auth_type & (WPS_AUTH_WPA | WPS_AUTH_WPAPSK)))
335		bss->wpa = 3;
336	else if (cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA2PSK))
337		bss->wpa = 2;
338	else if (cred->auth_type & (WPS_AUTH_WPA | WPS_AUTH_WPAPSK))
339		bss->wpa = 1;
340	else
341		bss->wpa = 0;
342
343	if (bss->wpa) {
344		if (cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA))
345			bss->wpa_key_mgmt = WPA_KEY_MGMT_IEEE8021X;
346		if (cred->auth_type & (WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK))
347			bss->wpa_key_mgmt = WPA_KEY_MGMT_PSK;
348
349		bss->wpa_pairwise = 0;
350		if (cred->encr_type & WPS_ENCR_AES) {
351			if (hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211AD)
352				bss->wpa_pairwise |= WPA_CIPHER_GCMP;
353			else
354				bss->wpa_pairwise |= WPA_CIPHER_CCMP;
355		}
356		if (cred->encr_type & WPS_ENCR_TKIP)
357			bss->wpa_pairwise |= WPA_CIPHER_TKIP;
358		bss->rsn_pairwise = bss->wpa_pairwise;
359		bss->wpa_group = wpa_select_ap_group_cipher(bss->wpa,
360							    bss->wpa_pairwise,
361							    bss->rsn_pairwise);
362
363		if (cred->key_len >= 8 && cred->key_len < 64) {
364			os_free(bss->ssid.wpa_passphrase);
365			bss->ssid.wpa_passphrase = os_zalloc(cred->key_len + 1);
366			if (bss->ssid.wpa_passphrase)
367				os_memcpy(bss->ssid.wpa_passphrase, cred->key,
368					  cred->key_len);
369			hostapd_config_clear_wpa_psk(&bss->ssid.wpa_psk);
370		} else if (cred->key_len == 64) {
371			hostapd_config_clear_wpa_psk(&bss->ssid.wpa_psk);
372			bss->ssid.wpa_psk =
373				os_zalloc(sizeof(struct hostapd_wpa_psk));
374			if (bss->ssid.wpa_psk &&
375			    hexstr2bin((const char *) cred->key,
376				       bss->ssid.wpa_psk->psk, PMK_LEN) == 0) {
377				bss->ssid.wpa_psk->group = 1;
378				os_free(bss->ssid.wpa_passphrase);
379				bss->ssid.wpa_passphrase = NULL;
380			}
381		}
382		bss->auth_algs = 1;
383	} else {
384		/*
385		 * WPS 2.0 does not allow WEP to be configured, so no need to
386		 * process that option here either.
387		 */
388		bss->auth_algs = 1;
389	}
390
391	/* Schedule configuration reload after short period of time to allow
392	 * EAP-WSC to be finished.
393	 */
394	eloop_register_timeout(0, 100000, wps_reload_config, hapd->iface,
395			       NULL);
396
397	return 0;
398}
399
400
401static int hapd_wps_cred_cb(struct hostapd_data *hapd, void *ctx)
402{
403	const struct wps_credential *cred = ctx;
404	FILE *oconf, *nconf;
405	size_t len, i;
406	char *tmp_fname;
407	char buf[1024];
408	int multi_bss;
409	int wpa;
410
411	if (hapd->wps == NULL)
412		return 0;
413
414	wpa_hexdump_key(MSG_DEBUG, "WPS: Received Credential attribute",
415			cred->cred_attr, cred->cred_attr_len);
416
417	wpa_printf(MSG_DEBUG, "WPS: Received new AP Settings");
418	wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID", cred->ssid, cred->ssid_len);
419	wpa_printf(MSG_DEBUG, "WPS: Authentication Type 0x%x",
420		   cred->auth_type);
421	wpa_printf(MSG_DEBUG, "WPS: Encryption Type 0x%x", cred->encr_type);
422	wpa_printf(MSG_DEBUG, "WPS: Network Key Index %d", cred->key_idx);
423	wpa_hexdump_key(MSG_DEBUG, "WPS: Network Key",
424			cred->key, cred->key_len);
425	wpa_printf(MSG_DEBUG, "WPS: MAC Address " MACSTR,
426		   MAC2STR(cred->mac_addr));
427
428	if ((hapd->conf->wps_cred_processing == 1 ||
429	     hapd->conf->wps_cred_processing == 2) && cred->cred_attr) {
430		hapd_new_ap_event(hapd, cred->cred_attr, cred->cred_attr_len);
431	} else if (hapd->conf->wps_cred_processing == 1 ||
432		   hapd->conf->wps_cred_processing == 2) {
433		struct wpabuf *attr;
434		attr = wpabuf_alloc(200);
435		if (attr && wps_build_credential_wrap(attr, cred) == 0)
436			hapd_new_ap_event(hapd, wpabuf_head_u8(attr),
437					  wpabuf_len(attr));
438		wpabuf_free(attr);
439	} else
440		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_NEW_AP_SETTINGS);
441
442	if (hapd->conf->wps_cred_processing == 1)
443		return 0;
444
445	os_memcpy(hapd->wps->ssid, cred->ssid, cred->ssid_len);
446	hapd->wps->ssid_len = cred->ssid_len;
447	hapd->wps->encr_types = cred->encr_type;
448	hapd->wps->encr_types_rsn = cred->encr_type;
449	hapd->wps->encr_types_wpa = cred->encr_type;
450	hapd->wps->auth_types = cred->auth_type;
451	hapd->wps->ap_encr_type = cred->encr_type;
452	hapd->wps->ap_auth_type = cred->auth_type;
453	if (cred->key_len == 0) {
454		os_free(hapd->wps->network_key);
455		hapd->wps->network_key = NULL;
456		hapd->wps->network_key_len = 0;
457	} else if ((cred->auth_type & (WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK)) &&
458		   (cred->key_len < 8 || cred->key_len > 2 * PMK_LEN)) {
459		wpa_printf(MSG_INFO, "WPS: Invalid key length %lu for WPA/WPA2",
460			   (unsigned long) cred->key_len);
461		return -1;
462	} else {
463		if (hapd->wps->network_key == NULL ||
464		    hapd->wps->network_key_len < cred->key_len) {
465			hapd->wps->network_key_len = 0;
466			os_free(hapd->wps->network_key);
467			hapd->wps->network_key = os_malloc(cred->key_len);
468			if (hapd->wps->network_key == NULL)
469				return -1;
470		}
471		hapd->wps->network_key_len = cred->key_len;
472		os_memcpy(hapd->wps->network_key, cred->key, cred->key_len);
473	}
474	hapd->wps->wps_state = WPS_STATE_CONFIGURED;
475
476	if (hapd->iface->config_fname == NULL)
477		return hapd_wps_reconfig_in_memory(hapd, cred);
478	len = os_strlen(hapd->iface->config_fname) + 5;
479	tmp_fname = os_malloc(len);
480	if (tmp_fname == NULL)
481		return -1;
482	os_snprintf(tmp_fname, len, "%s-new", hapd->iface->config_fname);
483
484	oconf = fopen(hapd->iface->config_fname, "r");
485	if (oconf == NULL) {
486		wpa_printf(MSG_WARNING, "WPS: Could not open current "
487			   "configuration file");
488		os_free(tmp_fname);
489		return -1;
490	}
491
492	nconf = fopen(tmp_fname, "w");
493	if (nconf == NULL) {
494		wpa_printf(MSG_WARNING, "WPS: Could not write updated "
495			   "configuration file");
496		os_free(tmp_fname);
497		fclose(oconf);
498		return -1;
499	}
500
501	fprintf(nconf, "# WPS configuration - START\n");
502
503	fprintf(nconf, "wps_state=2\n");
504
505	if (is_hex(cred->ssid, cred->ssid_len)) {
506		fprintf(nconf, "ssid2=");
507		for (i = 0; i < cred->ssid_len; i++)
508			fprintf(nconf, "%02x", cred->ssid[i]);
509		fprintf(nconf, "\n");
510	} else {
511		fprintf(nconf, "ssid=");
512		for (i = 0; i < cred->ssid_len; i++)
513			fputc(cred->ssid[i], nconf);
514		fprintf(nconf, "\n");
515	}
516
517	if ((cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA2PSK)) &&
518	    (cred->auth_type & (WPS_AUTH_WPA | WPS_AUTH_WPAPSK)))
519		wpa = 3;
520	else if (cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA2PSK))
521		wpa = 2;
522	else if (cred->auth_type & (WPS_AUTH_WPA | WPS_AUTH_WPAPSK))
523		wpa = 1;
524	else
525		wpa = 0;
526
527	if (wpa) {
528		char *prefix;
529		fprintf(nconf, "wpa=%d\n", wpa);
530
531		fprintf(nconf, "wpa_key_mgmt=");
532		prefix = "";
533		if (cred->auth_type & (WPS_AUTH_WPA2 | WPS_AUTH_WPA)) {
534			fprintf(nconf, "WPA-EAP");
535			prefix = " ";
536		}
537		if (cred->auth_type & (WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK))
538			fprintf(nconf, "%sWPA-PSK", prefix);
539		fprintf(nconf, "\n");
540
541		fprintf(nconf, "wpa_pairwise=");
542		prefix = "";
543		if (cred->encr_type & WPS_ENCR_AES) {
544			if (hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211AD)
545				fprintf(nconf, "GCMP");
546			else
547				fprintf(nconf, "CCMP");
548
549			prefix = " ";
550		}
551		if (cred->encr_type & WPS_ENCR_TKIP) {
552			fprintf(nconf, "%sTKIP", prefix);
553		}
554		fprintf(nconf, "\n");
555
556		if (cred->key_len >= 8 && cred->key_len < 64) {
557			fprintf(nconf, "wpa_passphrase=");
558			for (i = 0; i < cred->key_len; i++)
559				fputc(cred->key[i], nconf);
560			fprintf(nconf, "\n");
561		} else if (cred->key_len == 64) {
562			fprintf(nconf, "wpa_psk=");
563			for (i = 0; i < cred->key_len; i++)
564				fputc(cred->key[i], nconf);
565			fprintf(nconf, "\n");
566		} else {
567			wpa_printf(MSG_WARNING, "WPS: Invalid key length %lu "
568				   "for WPA/WPA2",
569				   (unsigned long) cred->key_len);
570		}
571
572		fprintf(nconf, "auth_algs=1\n");
573	} else {
574		/*
575		 * WPS 2.0 does not allow WEP to be configured, so no need to
576		 * process that option here either.
577		 */
578		fprintf(nconf, "auth_algs=1\n");
579	}
580
581	fprintf(nconf, "# WPS configuration - END\n");
582
583	multi_bss = 0;
584	while (fgets(buf, sizeof(buf), oconf)) {
585		if (os_strncmp(buf, "bss=", 4) == 0)
586			multi_bss = 1;
587		if (!multi_bss &&
588		    (str_starts(buf, "ssid=") ||
589		     str_starts(buf, "ssid2=") ||
590		     str_starts(buf, "auth_algs=") ||
591		     str_starts(buf, "wep_default_key=") ||
592		     str_starts(buf, "wep_key") ||
593		     str_starts(buf, "wps_state=") ||
594		     str_starts(buf, "wpa=") ||
595		     str_starts(buf, "wpa_psk=") ||
596		     str_starts(buf, "wpa_pairwise=") ||
597		     str_starts(buf, "rsn_pairwise=") ||
598		     str_starts(buf, "wpa_key_mgmt=") ||
599		     str_starts(buf, "wpa_passphrase="))) {
600			fprintf(nconf, "#WPS# %s", buf);
601		} else
602			fprintf(nconf, "%s", buf);
603	}
604
605	fclose(nconf);
606	fclose(oconf);
607
608	if (rename(tmp_fname, hapd->iface->config_fname) < 0) {
609		wpa_printf(MSG_WARNING, "WPS: Failed to rename the updated "
610			   "configuration file: %s", strerror(errno));
611		os_free(tmp_fname);
612		return -1;
613	}
614
615	os_free(tmp_fname);
616
617	/* Schedule configuration reload after short period of time to allow
618	 * EAP-WSC to be finished.
619	 */
620	eloop_register_timeout(0, 100000, wps_reload_config, hapd->iface,
621			       NULL);
622
623	wpa_printf(MSG_DEBUG, "WPS: AP configuration updated");
624
625	return 0;
626}
627
628
629static int hostapd_wps_cred_cb(void *ctx, const struct wps_credential *cred)
630{
631	struct hostapd_data *hapd = ctx;
632	return hostapd_wps_for_each(hapd, hapd_wps_cred_cb, (void *) cred);
633}
634
635
636static void hostapd_wps_reenable_ap_pin(void *eloop_data, void *user_ctx)
637{
638	struct hostapd_data *hapd = eloop_data;
639
640	if (hapd->conf->ap_setup_locked)
641		return;
642	if (hapd->ap_pin_failures_consecutive >= 10)
643		return;
644
645	wpa_printf(MSG_DEBUG, "WPS: Re-enable AP PIN");
646	wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_AP_SETUP_UNLOCKED);
647	hapd->wps->ap_setup_locked = 0;
648	wps_registrar_update_ie(hapd->wps->registrar);
649}
650
651
652static int wps_pwd_auth_fail(struct hostapd_data *hapd, void *ctx)
653{
654	struct wps_event_pwd_auth_fail *data = ctx;
655
656	if (!data->enrollee || hapd->conf->ap_pin == NULL || hapd->wps == NULL)
657		return 0;
658
659	/*
660	 * Registrar failed to prove its knowledge of the AP PIN. Lock AP setup
661	 * for some time if this happens multiple times to slow down brute
662	 * force attacks.
663	 */
664	hapd->ap_pin_failures++;
665	hapd->ap_pin_failures_consecutive++;
666	wpa_printf(MSG_DEBUG, "WPS: AP PIN authentication failure number %u "
667		   "(%u consecutive)",
668		   hapd->ap_pin_failures, hapd->ap_pin_failures_consecutive);
669	if (hapd->ap_pin_failures < 3)
670		return 0;
671
672	wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_AP_SETUP_LOCKED);
673	hapd->wps->ap_setup_locked = 1;
674
675	wps_registrar_update_ie(hapd->wps->registrar);
676
677	if (!hapd->conf->ap_setup_locked &&
678	    hapd->ap_pin_failures_consecutive >= 10) {
679		/*
680		 * In indefinite lockdown - disable automatic AP PIN
681		 * reenablement.
682		 */
683		eloop_cancel_timeout(hostapd_wps_reenable_ap_pin, hapd, NULL);
684		wpa_printf(MSG_DEBUG, "WPS: AP PIN disabled indefinitely");
685	} else if (!hapd->conf->ap_setup_locked) {
686		if (hapd->ap_pin_lockout_time == 0)
687			hapd->ap_pin_lockout_time = 60;
688		else if (hapd->ap_pin_lockout_time < 365 * 24 * 60 * 60 &&
689			 (hapd->ap_pin_failures % 3) == 0)
690			hapd->ap_pin_lockout_time *= 2;
691
692		wpa_printf(MSG_DEBUG, "WPS: Disable AP PIN for %u seconds",
693			   hapd->ap_pin_lockout_time);
694		eloop_cancel_timeout(hostapd_wps_reenable_ap_pin, hapd, NULL);
695		eloop_register_timeout(hapd->ap_pin_lockout_time, 0,
696				       hostapd_wps_reenable_ap_pin, hapd,
697				       NULL);
698	}
699
700	return 0;
701}
702
703
704static void hostapd_pwd_auth_fail(struct hostapd_data *hapd,
705				  struct wps_event_pwd_auth_fail *data)
706{
707	/* Update WPS Status - Authentication Failure */
708	wpa_printf(MSG_DEBUG, "WPS: Authentication failure update");
709	hapd->wps_stats.status = WPS_STATUS_FAILURE;
710	hapd->wps_stats.failure_reason = WPS_EI_AUTH_FAILURE;
711	os_memcpy(hapd->wps_stats.peer_addr, data->peer_macaddr, ETH_ALEN);
712
713	hostapd_wps_for_each(hapd, wps_pwd_auth_fail, data);
714}
715
716
717static int wps_ap_pin_success(struct hostapd_data *hapd, void *ctx)
718{
719	if (hapd->conf->ap_pin == NULL || hapd->wps == NULL)
720		return 0;
721
722	if (hapd->ap_pin_failures_consecutive == 0)
723		return 0;
724
725	wpa_printf(MSG_DEBUG, "WPS: Clear consecutive AP PIN failure counter "
726		   "- total validation failures %u (%u consecutive)",
727		   hapd->ap_pin_failures, hapd->ap_pin_failures_consecutive);
728	hapd->ap_pin_failures_consecutive = 0;
729
730	return 0;
731}
732
733
734static void hostapd_wps_ap_pin_success(struct hostapd_data *hapd)
735{
736	hostapd_wps_for_each(hapd, wps_ap_pin_success, NULL);
737}
738
739
740static void hostapd_wps_event_pbc_overlap(struct hostapd_data *hapd)
741{
742	/* Update WPS Status - PBC Overlap */
743	hapd->wps_stats.pbc_status = WPS_PBC_STATUS_OVERLAP;
744}
745
746
747static void hostapd_wps_event_pbc_timeout(struct hostapd_data *hapd)
748{
749	/* Update WPS PBC Status:PBC Timeout */
750	hapd->wps_stats.pbc_status = WPS_PBC_STATUS_TIMEOUT;
751}
752
753
754static void hostapd_wps_event_pbc_active(struct hostapd_data *hapd)
755{
756	/* Update WPS PBC status - Active */
757	hapd->wps_stats.pbc_status = WPS_PBC_STATUS_ACTIVE;
758}
759
760
761static void hostapd_wps_event_pbc_disable(struct hostapd_data *hapd)
762{
763	/* Update WPS PBC status - Active */
764	hapd->wps_stats.pbc_status = WPS_PBC_STATUS_DISABLE;
765}
766
767
768static void hostapd_wps_event_success(struct hostapd_data *hapd,
769				      struct wps_event_success *success)
770{
771	/* Update WPS status - Success */
772	hapd->wps_stats.pbc_status = WPS_PBC_STATUS_DISABLE;
773	hapd->wps_stats.status = WPS_STATUS_SUCCESS;
774	os_memcpy(hapd->wps_stats.peer_addr, success->peer_macaddr, ETH_ALEN);
775}
776
777
778static void hostapd_wps_event_fail(struct hostapd_data *hapd,
779				   struct wps_event_fail *fail)
780{
781	/* Update WPS status - Failure */
782	hapd->wps_stats.status = WPS_STATUS_FAILURE;
783	os_memcpy(hapd->wps_stats.peer_addr, fail->peer_macaddr, ETH_ALEN);
784
785	hapd->wps_stats.failure_reason = fail->error_indication;
786
787	if (fail->error_indication > 0 &&
788	    fail->error_indication < NUM_WPS_EI_VALUES) {
789		wpa_msg(hapd->msg_ctx, MSG_INFO,
790			WPS_EVENT_FAIL "msg=%d config_error=%d reason=%d (%s)",
791			fail->msg, fail->config_error, fail->error_indication,
792			wps_ei_str(fail->error_indication));
793	} else {
794		wpa_msg(hapd->msg_ctx, MSG_INFO,
795			WPS_EVENT_FAIL "msg=%d config_error=%d",
796			fail->msg, fail->config_error);
797	}
798}
799
800
801static void hostapd_wps_event_cb(void *ctx, enum wps_event event,
802				 union wps_event_data *data)
803{
804	struct hostapd_data *hapd = ctx;
805
806	switch (event) {
807	case WPS_EV_M2D:
808		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_M2D);
809		break;
810	case WPS_EV_FAIL:
811		hostapd_wps_event_fail(hapd, &data->fail);
812		break;
813	case WPS_EV_SUCCESS:
814		hostapd_wps_event_success(hapd, &data->success);
815		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_SUCCESS);
816		break;
817	case WPS_EV_PWD_AUTH_FAIL:
818		hostapd_pwd_auth_fail(hapd, &data->pwd_auth_fail);
819		break;
820	case WPS_EV_PBC_OVERLAP:
821		hostapd_wps_event_pbc_overlap(hapd);
822		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_OVERLAP);
823		break;
824	case WPS_EV_PBC_TIMEOUT:
825		hostapd_wps_event_pbc_timeout(hapd);
826		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_TIMEOUT);
827		break;
828	case WPS_EV_PBC_ACTIVE:
829		hostapd_wps_event_pbc_active(hapd);
830		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_ACTIVE);
831		break;
832	case WPS_EV_PBC_DISABLE:
833		hostapd_wps_event_pbc_disable(hapd);
834		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_DISABLE);
835		break;
836	case WPS_EV_ER_AP_ADD:
837		break;
838	case WPS_EV_ER_AP_REMOVE:
839		break;
840	case WPS_EV_ER_ENROLLEE_ADD:
841		break;
842	case WPS_EV_ER_ENROLLEE_REMOVE:
843		break;
844	case WPS_EV_ER_AP_SETTINGS:
845		break;
846	case WPS_EV_ER_SET_SELECTED_REGISTRAR:
847		break;
848	case WPS_EV_AP_PIN_SUCCESS:
849		hostapd_wps_ap_pin_success(hapd);
850		break;
851	}
852	if (hapd->wps_event_cb)
853		hapd->wps_event_cb(hapd->wps_event_cb_ctx, event, data);
854}
855
856
857static int hostapd_wps_rf_band_cb(void *ctx)
858{
859	struct hostapd_data *hapd = ctx;
860
861	return hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211A ?
862		WPS_RF_50GHZ :
863		hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211AD ?
864		WPS_RF_60GHZ : WPS_RF_24GHZ; /* FIX: dualband AP */
865}
866
867
868static void hostapd_wps_clear_ies(struct hostapd_data *hapd, int deinit_only)
869{
870	wpabuf_free(hapd->wps_beacon_ie);
871	hapd->wps_beacon_ie = NULL;
872
873	wpabuf_free(hapd->wps_probe_resp_ie);
874	hapd->wps_probe_resp_ie = NULL;
875
876	if (deinit_only) {
877		if (hapd->drv_priv)
878			hostapd_reset_ap_wps_ie(hapd);
879		return;
880	}
881
882	hostapd_set_ap_wps_ie(hapd);
883}
884
885
886static int get_uuid_cb(struct hostapd_iface *iface, void *ctx)
887{
888	const u8 **uuid = ctx;
889	size_t j;
890
891	if (iface == NULL)
892		return 0;
893	for (j = 0; j < iface->num_bss; j++) {
894		struct hostapd_data *hapd = iface->bss[j];
895		if (hapd->wps && !hapd->conf->wps_independent &&
896		    !is_nil_uuid(hapd->wps->uuid)) {
897			*uuid = hapd->wps->uuid;
898			return 1;
899		}
900	}
901
902	return 0;
903}
904
905
906static const u8 * get_own_uuid(struct hostapd_iface *iface)
907{
908	const u8 *uuid;
909	if (iface->interfaces == NULL ||
910	    iface->interfaces->for_each_interface == NULL)
911		return NULL;
912	uuid = NULL;
913	iface->interfaces->for_each_interface(iface->interfaces, get_uuid_cb,
914					      &uuid);
915	return uuid;
916}
917
918
919static int count_interface_cb(struct hostapd_iface *iface, void *ctx)
920{
921	int *count= ctx;
922	(*count)++;
923	return 0;
924}
925
926
927static int interface_count(struct hostapd_iface *iface)
928{
929	int count = 0;
930	if (iface->interfaces == NULL ||
931	    iface->interfaces->for_each_interface == NULL)
932		return 0;
933	iface->interfaces->for_each_interface(iface->interfaces,
934					      count_interface_cb, &count);
935	return count;
936}
937
938
939static int hostapd_wps_set_vendor_ext(struct hostapd_data *hapd,
940				      struct wps_context *wps)
941{
942	int i;
943
944	for (i = 0; i < MAX_WPS_VENDOR_EXTENSIONS; i++) {
945		wpabuf_free(wps->dev.vendor_ext[i]);
946		wps->dev.vendor_ext[i] = NULL;
947
948		if (hapd->conf->wps_vendor_ext[i] == NULL)
949			continue;
950
951		wps->dev.vendor_ext[i] =
952			wpabuf_dup(hapd->conf->wps_vendor_ext[i]);
953		if (wps->dev.vendor_ext[i] == NULL) {
954			while (--i >= 0)
955				wpabuf_free(wps->dev.vendor_ext[i]);
956			return -1;
957		}
958	}
959
960	return 0;
961}
962
963
964static void hostapd_free_wps(struct wps_context *wps)
965{
966	int i;
967
968	for (i = 0; i < MAX_WPS_VENDOR_EXTENSIONS; i++)
969		wpabuf_free(wps->dev.vendor_ext[i]);
970	wps_device_data_free(&wps->dev);
971	os_free(wps->network_key);
972	hostapd_wps_nfc_clear(wps);
973	wpabuf_free(wps->dh_pubkey);
974	wpabuf_free(wps->dh_privkey);
975	os_free(wps);
976}
977
978
979int hostapd_init_wps(struct hostapd_data *hapd,
980		     struct hostapd_bss_config *conf)
981{
982	struct wps_context *wps;
983	struct wps_registrar_config cfg;
984
985	if (conf->wps_state == 0) {
986		hostapd_wps_clear_ies(hapd, 0);
987		return 0;
988	}
989
990	wps = os_zalloc(sizeof(*wps));
991	if (wps == NULL)
992		return -1;
993
994	wps->cred_cb = hostapd_wps_cred_cb;
995	wps->event_cb = hostapd_wps_event_cb;
996	wps->rf_band_cb = hostapd_wps_rf_band_cb;
997	wps->cb_ctx = hapd;
998
999	os_memset(&cfg, 0, sizeof(cfg));
1000	wps->wps_state = hapd->conf->wps_state;
1001	wps->ap_setup_locked = hapd->conf->ap_setup_locked;
1002	if (is_nil_uuid(hapd->conf->uuid)) {
1003		const u8 *uuid;
1004		uuid = get_own_uuid(hapd->iface);
1005		if (uuid && !conf->wps_independent) {
1006			os_memcpy(wps->uuid, uuid, UUID_LEN);
1007			wpa_hexdump(MSG_DEBUG, "WPS: Clone UUID from another "
1008				    "interface", wps->uuid, UUID_LEN);
1009		} else {
1010			uuid_gen_mac_addr(hapd->own_addr, wps->uuid);
1011			wpa_hexdump(MSG_DEBUG, "WPS: UUID based on MAC "
1012				    "address", wps->uuid, UUID_LEN);
1013		}
1014	} else {
1015		os_memcpy(wps->uuid, hapd->conf->uuid, UUID_LEN);
1016		wpa_hexdump(MSG_DEBUG, "WPS: Use configured UUID",
1017			    wps->uuid, UUID_LEN);
1018	}
1019	wps->ssid_len = hapd->conf->ssid.ssid_len;
1020	os_memcpy(wps->ssid, hapd->conf->ssid.ssid, wps->ssid_len);
1021	wps->ap = 1;
1022	os_memcpy(wps->dev.mac_addr, hapd->own_addr, ETH_ALEN);
1023	wps->dev.device_name = hapd->conf->device_name ?
1024		os_strdup(hapd->conf->device_name) : NULL;
1025	wps->dev.manufacturer = hapd->conf->manufacturer ?
1026		os_strdup(hapd->conf->manufacturer) : NULL;
1027	wps->dev.model_name = hapd->conf->model_name ?
1028		os_strdup(hapd->conf->model_name) : NULL;
1029	wps->dev.model_number = hapd->conf->model_number ?
1030		os_strdup(hapd->conf->model_number) : NULL;
1031	wps->dev.serial_number = hapd->conf->serial_number ?
1032		os_strdup(hapd->conf->serial_number) : NULL;
1033	wps->config_methods =
1034		wps_config_methods_str2bin(hapd->conf->config_methods);
1035	if ((wps->config_methods &
1036	     (WPS_CONFIG_DISPLAY | WPS_CONFIG_VIRT_DISPLAY |
1037	      WPS_CONFIG_PHY_DISPLAY)) == WPS_CONFIG_DISPLAY) {
1038		wpa_printf(MSG_INFO, "WPS: Converting display to "
1039			   "virtual_display for WPS 2.0 compliance");
1040		wps->config_methods |= WPS_CONFIG_VIRT_DISPLAY;
1041	}
1042	if ((wps->config_methods &
1043	     (WPS_CONFIG_PUSHBUTTON | WPS_CONFIG_VIRT_PUSHBUTTON |
1044	      WPS_CONFIG_PHY_PUSHBUTTON)) == WPS_CONFIG_PUSHBUTTON) {
1045		wpa_printf(MSG_INFO, "WPS: Converting push_button to "
1046			   "virtual_push_button for WPS 2.0 compliance");
1047		wps->config_methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
1048	}
1049	os_memcpy(wps->dev.pri_dev_type, hapd->conf->device_type,
1050		  WPS_DEV_TYPE_LEN);
1051
1052	if (hostapd_wps_set_vendor_ext(hapd, wps) < 0)
1053		goto fail;
1054
1055	wps->dev.os_version = WPA_GET_BE32(hapd->conf->os_version);
1056
1057	if (conf->wps_rf_bands) {
1058		wps->dev.rf_bands = conf->wps_rf_bands;
1059	} else {
1060		wps->dev.rf_bands =
1061			hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211A ?
1062			WPS_RF_50GHZ :
1063			hapd->iconf->hw_mode == HOSTAPD_MODE_IEEE80211AD ?
1064			WPS_RF_60GHZ : WPS_RF_24GHZ; /* FIX: dualband AP */
1065	}
1066
1067	if (conf->wpa & WPA_PROTO_RSN) {
1068		if (conf->wpa_key_mgmt & WPA_KEY_MGMT_PSK)
1069			wps->auth_types |= WPS_AUTH_WPA2PSK;
1070		if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X)
1071			wps->auth_types |= WPS_AUTH_WPA2;
1072
1073		if (conf->rsn_pairwise & (WPA_CIPHER_CCMP | WPA_CIPHER_GCMP)) {
1074			wps->encr_types |= WPS_ENCR_AES;
1075			wps->encr_types_rsn |= WPS_ENCR_AES;
1076		}
1077		if (conf->rsn_pairwise & WPA_CIPHER_TKIP) {
1078			wps->encr_types |= WPS_ENCR_TKIP;
1079			wps->encr_types_rsn |= WPS_ENCR_TKIP;
1080		}
1081	}
1082
1083	if (conf->wpa & WPA_PROTO_WPA) {
1084		if (conf->wpa_key_mgmt & WPA_KEY_MGMT_PSK)
1085			wps->auth_types |= WPS_AUTH_WPAPSK;
1086		if (conf->wpa_key_mgmt & WPA_KEY_MGMT_IEEE8021X)
1087			wps->auth_types |= WPS_AUTH_WPA;
1088
1089		if (conf->wpa_pairwise & WPA_CIPHER_CCMP) {
1090			wps->encr_types |= WPS_ENCR_AES;
1091			wps->encr_types_wpa |= WPS_ENCR_AES;
1092		}
1093		if (conf->wpa_pairwise & WPA_CIPHER_TKIP) {
1094			wps->encr_types |= WPS_ENCR_TKIP;
1095			wps->encr_types_wpa |= WPS_ENCR_TKIP;
1096		}
1097	}
1098
1099	if (conf->ssid.security_policy == SECURITY_PLAINTEXT) {
1100		wps->encr_types |= WPS_ENCR_NONE;
1101		wps->auth_types |= WPS_AUTH_OPEN;
1102	}
1103
1104	if (conf->ssid.wpa_psk_file) {
1105		/* Use per-device PSKs */
1106	} else if (conf->ssid.wpa_passphrase) {
1107		wps->network_key = (u8 *) os_strdup(conf->ssid.wpa_passphrase);
1108		wps->network_key_len = os_strlen(conf->ssid.wpa_passphrase);
1109	} else if (conf->ssid.wpa_psk) {
1110		wps->network_key = os_malloc(2 * PMK_LEN + 1);
1111		if (wps->network_key == NULL)
1112			goto fail;
1113		wpa_snprintf_hex((char *) wps->network_key, 2 * PMK_LEN + 1,
1114				 conf->ssid.wpa_psk->psk, PMK_LEN);
1115		wps->network_key_len = 2 * PMK_LEN;
1116	} else if (conf->ssid.wep.keys_set && conf->ssid.wep.key[0]) {
1117		wps->network_key = os_malloc(conf->ssid.wep.len[0]);
1118		if (wps->network_key == NULL)
1119			goto fail;
1120		os_memcpy(wps->network_key, conf->ssid.wep.key[0],
1121			  conf->ssid.wep.len[0]);
1122		wps->network_key_len = conf->ssid.wep.len[0];
1123	}
1124
1125	if (conf->ssid.wpa_psk) {
1126		os_memcpy(wps->psk, conf->ssid.wpa_psk->psk, PMK_LEN);
1127		wps->psk_set = 1;
1128	}
1129
1130	wps->ap_auth_type = wps->auth_types;
1131	wps->ap_encr_type = wps->encr_types;
1132	if (conf->wps_state == WPS_STATE_NOT_CONFIGURED) {
1133		/* Override parameters to enable security by default */
1134		wps->auth_types = WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK;
1135		wps->encr_types = WPS_ENCR_AES | WPS_ENCR_TKIP;
1136		wps->encr_types_rsn = WPS_ENCR_AES | WPS_ENCR_TKIP;
1137		wps->encr_types_wpa = WPS_ENCR_AES | WPS_ENCR_TKIP;
1138	}
1139
1140	wps->ap_settings = conf->ap_settings;
1141	wps->ap_settings_len = conf->ap_settings_len;
1142
1143	cfg.new_psk_cb = hostapd_wps_new_psk_cb;
1144	cfg.set_ie_cb = hostapd_wps_set_ie_cb;
1145	cfg.pin_needed_cb = hostapd_wps_pin_needed_cb;
1146	cfg.reg_success_cb = hostapd_wps_reg_success_cb;
1147	cfg.enrollee_seen_cb = hostapd_wps_enrollee_seen_cb;
1148	cfg.cb_ctx = hapd;
1149	cfg.skip_cred_build = conf->skip_cred_build;
1150	cfg.extra_cred = conf->extra_cred;
1151	cfg.extra_cred_len = conf->extra_cred_len;
1152	cfg.disable_auto_conf = (hapd->conf->wps_cred_processing == 1) &&
1153		conf->skip_cred_build;
1154	if (conf->ssid.security_policy == SECURITY_STATIC_WEP)
1155		cfg.static_wep_only = 1;
1156	cfg.dualband = interface_count(hapd->iface) > 1;
1157	if ((wps->dev.rf_bands & (WPS_RF_50GHZ | WPS_RF_24GHZ)) ==
1158	    (WPS_RF_50GHZ | WPS_RF_24GHZ))
1159		cfg.dualband = 1;
1160	if (cfg.dualband)
1161		wpa_printf(MSG_DEBUG, "WPS: Dualband AP");
1162	cfg.force_per_enrollee_psk = conf->force_per_enrollee_psk;
1163
1164	wps->registrar = wps_registrar_init(wps, &cfg);
1165	if (wps->registrar == NULL) {
1166		wpa_printf(MSG_ERROR, "Failed to initialize WPS Registrar");
1167		goto fail;
1168	}
1169
1170#ifdef CONFIG_WPS_UPNP
1171	wps->friendly_name = hapd->conf->friendly_name;
1172	wps->manufacturer_url = hapd->conf->manufacturer_url;
1173	wps->model_description = hapd->conf->model_description;
1174	wps->model_url = hapd->conf->model_url;
1175	wps->upc = hapd->conf->upc;
1176#endif /* CONFIG_WPS_UPNP */
1177
1178	hostapd_register_probereq_cb(hapd, hostapd_wps_probe_req_rx, hapd);
1179
1180	hapd->wps = wps;
1181
1182	return 0;
1183
1184fail:
1185	hostapd_free_wps(wps);
1186	return -1;
1187}
1188
1189
1190int hostapd_init_wps_complete(struct hostapd_data *hapd)
1191{
1192	struct wps_context *wps = hapd->wps;
1193
1194	if (wps == NULL)
1195		return 0;
1196
1197#ifdef CONFIG_WPS_UPNP
1198	if (hostapd_wps_upnp_init(hapd, wps) < 0) {
1199		wpa_printf(MSG_ERROR, "Failed to initialize WPS UPnP");
1200		wps_registrar_deinit(wps->registrar);
1201		hostapd_free_wps(wps);
1202		hapd->wps = NULL;
1203		return -1;
1204	}
1205#endif /* CONFIG_WPS_UPNP */
1206
1207	return 0;
1208}
1209
1210
1211static void hostapd_wps_nfc_clear(struct wps_context *wps)
1212{
1213#ifdef CONFIG_WPS_NFC
1214	wpa_printf(MSG_DEBUG, "WPS: Clear NFC Tag context %p", wps);
1215	wps->ap_nfc_dev_pw_id = 0;
1216	wpabuf_free(wps->ap_nfc_dh_pubkey);
1217	wps->ap_nfc_dh_pubkey = NULL;
1218	wpabuf_free(wps->ap_nfc_dh_privkey);
1219	wps->ap_nfc_dh_privkey = NULL;
1220	wpabuf_free(wps->ap_nfc_dev_pw);
1221	wps->ap_nfc_dev_pw = NULL;
1222#endif /* CONFIG_WPS_NFC */
1223}
1224
1225
1226void hostapd_deinit_wps(struct hostapd_data *hapd)
1227{
1228	eloop_cancel_timeout(hostapd_wps_reenable_ap_pin, hapd, NULL);
1229	eloop_cancel_timeout(hostapd_wps_ap_pin_timeout, hapd, NULL);
1230	eloop_cancel_timeout(wps_reload_config, hapd->iface, NULL);
1231	if (hapd->wps == NULL) {
1232		hostapd_wps_clear_ies(hapd, 1);
1233		return;
1234	}
1235#ifdef CONFIG_WPS_UPNP
1236	hostapd_wps_upnp_deinit(hapd);
1237#endif /* CONFIG_WPS_UPNP */
1238	wps_registrar_deinit(hapd->wps->registrar);
1239	wps_free_pending_msgs(hapd->wps->upnp_msgs);
1240	hostapd_free_wps(hapd->wps);
1241	hapd->wps = NULL;
1242	hostapd_wps_clear_ies(hapd, 1);
1243}
1244
1245
1246void hostapd_update_wps(struct hostapd_data *hapd)
1247{
1248	if (hapd->wps == NULL)
1249		return;
1250
1251#ifdef CONFIG_WPS_UPNP
1252	hapd->wps->friendly_name = hapd->conf->friendly_name;
1253	hapd->wps->manufacturer_url = hapd->conf->manufacturer_url;
1254	hapd->wps->model_description = hapd->conf->model_description;
1255	hapd->wps->model_url = hapd->conf->model_url;
1256	hapd->wps->upc = hapd->conf->upc;
1257#endif /* CONFIG_WPS_UPNP */
1258
1259	hostapd_wps_set_vendor_ext(hapd, hapd->wps);
1260
1261	if (hapd->conf->wps_state)
1262		wps_registrar_update_ie(hapd->wps->registrar);
1263	else
1264		hostapd_deinit_wps(hapd);
1265}
1266
1267
1268struct wps_add_pin_data {
1269	const u8 *addr;
1270	const u8 *uuid;
1271	const u8 *pin;
1272	size_t pin_len;
1273	int timeout;
1274	int added;
1275};
1276
1277
1278static int wps_add_pin(struct hostapd_data *hapd, void *ctx)
1279{
1280	struct wps_add_pin_data *data = ctx;
1281	int ret;
1282
1283	if (hapd->wps == NULL)
1284		return 0;
1285	ret = wps_registrar_add_pin(hapd->wps->registrar, data->addr,
1286				    data->uuid, data->pin, data->pin_len,
1287				    data->timeout);
1288	if (ret == 0)
1289		data->added++;
1290	return ret;
1291}
1292
1293
1294int hostapd_wps_add_pin(struct hostapd_data *hapd, const u8 *addr,
1295			const char *uuid, const char *pin, int timeout)
1296{
1297	u8 u[UUID_LEN];
1298	struct wps_add_pin_data data;
1299
1300	data.addr = addr;
1301	data.uuid = u;
1302	data.pin = (const u8 *) pin;
1303	data.pin_len = os_strlen(pin);
1304	data.timeout = timeout;
1305	data.added = 0;
1306
1307	if (os_strcmp(uuid, "any") == 0)
1308		data.uuid = NULL;
1309	else {
1310		if (uuid_str2bin(uuid, u))
1311			return -1;
1312		data.uuid = u;
1313	}
1314	if (hostapd_wps_for_each(hapd, wps_add_pin, &data) < 0)
1315		return -1;
1316	return data.added ? 0 : -1;
1317}
1318
1319
1320struct wps_button_pushed_ctx {
1321	const u8 *p2p_dev_addr;
1322	unsigned int count;
1323};
1324
1325static int wps_button_pushed(struct hostapd_data *hapd, void *ctx)
1326{
1327	struct wps_button_pushed_ctx *data = ctx;
1328
1329	if (hapd->wps) {
1330		data->count++;
1331		return wps_registrar_button_pushed(hapd->wps->registrar,
1332						   data->p2p_dev_addr);
1333	}
1334
1335	return 0;
1336}
1337
1338
1339int hostapd_wps_button_pushed(struct hostapd_data *hapd,
1340			      const u8 *p2p_dev_addr)
1341{
1342	struct wps_button_pushed_ctx ctx;
1343	int ret;
1344
1345	os_memset(&ctx, 0, sizeof(ctx));
1346	ctx.p2p_dev_addr = p2p_dev_addr;
1347	ret = hostapd_wps_for_each(hapd, wps_button_pushed, &ctx);
1348	if (ret == 0 && !ctx.count)
1349		ret = -1;
1350	return ret;
1351}
1352
1353
1354struct wps_cancel_ctx {
1355	unsigned int count;
1356};
1357
1358static int wps_cancel(struct hostapd_data *hapd, void *ctx)
1359{
1360	struct wps_cancel_ctx *data = ctx;
1361
1362	if (hapd->wps) {
1363		data->count++;
1364		wps_registrar_wps_cancel(hapd->wps->registrar);
1365		ap_for_each_sta(hapd, ap_sta_wps_cancel, NULL);
1366	}
1367
1368	return 0;
1369}
1370
1371
1372int hostapd_wps_cancel(struct hostapd_data *hapd)
1373{
1374	struct wps_cancel_ctx ctx;
1375	int ret;
1376
1377	os_memset(&ctx, 0, sizeof(ctx));
1378	ret = hostapd_wps_for_each(hapd, wps_cancel, &ctx);
1379	if (ret == 0 && !ctx.count)
1380		ret = -1;
1381	return ret;
1382}
1383
1384
1385static int hostapd_wps_probe_req_rx(void *ctx, const u8 *addr, const u8 *da,
1386				    const u8 *bssid,
1387				    const u8 *ie, size_t ie_len,
1388				    int ssi_signal)
1389{
1390	struct hostapd_data *hapd = ctx;
1391	struct wpabuf *wps_ie;
1392	struct ieee802_11_elems elems;
1393
1394	if (hapd->wps == NULL)
1395		return 0;
1396
1397	if (ieee802_11_parse_elems(ie, ie_len, &elems, 0) == ParseFailed) {
1398		wpa_printf(MSG_DEBUG, "WPS: Could not parse ProbeReq from "
1399			   MACSTR, MAC2STR(addr));
1400		return 0;
1401	}
1402
1403	if (elems.ssid && elems.ssid_len > 0 &&
1404	    (elems.ssid_len != hapd->conf->ssid.ssid_len ||
1405	     os_memcmp(elems.ssid, hapd->conf->ssid.ssid, elems.ssid_len) !=
1406	     0))
1407		return 0; /* Not for us */
1408
1409	wps_ie = ieee802_11_vendor_ie_concat(ie, ie_len, WPS_DEV_OUI_WFA);
1410	if (wps_ie == NULL)
1411		return 0;
1412	if (wps_validate_probe_req(wps_ie, addr) < 0) {
1413		wpabuf_free(wps_ie);
1414		return 0;
1415	}
1416
1417	if (wpabuf_len(wps_ie) > 0) {
1418		int p2p_wildcard = 0;
1419#ifdef CONFIG_P2P
1420		if (elems.ssid && elems.ssid_len == P2P_WILDCARD_SSID_LEN &&
1421		    os_memcmp(elems.ssid, P2P_WILDCARD_SSID,
1422			      P2P_WILDCARD_SSID_LEN) == 0)
1423			p2p_wildcard = 1;
1424#endif /* CONFIG_P2P */
1425		wps_registrar_probe_req_rx(hapd->wps->registrar, addr, wps_ie,
1426					   p2p_wildcard);
1427#ifdef CONFIG_WPS_UPNP
1428		/* FIX: what exactly should be included in the WLANEvent?
1429		 * WPS attributes? Full ProbeReq frame? */
1430		if (!p2p_wildcard)
1431			upnp_wps_device_send_wlan_event(
1432				hapd->wps_upnp, addr,
1433				UPNP_WPS_WLANEVENT_TYPE_PROBE, wps_ie);
1434#endif /* CONFIG_WPS_UPNP */
1435	}
1436
1437	wpabuf_free(wps_ie);
1438
1439	return 0;
1440}
1441
1442
1443#ifdef CONFIG_WPS_UPNP
1444
1445static int hostapd_rx_req_put_wlan_response(
1446	void *priv, enum upnp_wps_wlanevent_type ev_type,
1447	const u8 *mac_addr, const struct wpabuf *msg,
1448	enum wps_msg_type msg_type)
1449{
1450	struct hostapd_data *hapd = priv;
1451	struct sta_info *sta;
1452	struct upnp_pending_message *p;
1453
1454	wpa_printf(MSG_DEBUG, "WPS UPnP: PutWLANResponse ev_type=%d mac_addr="
1455		   MACSTR, ev_type, MAC2STR(mac_addr));
1456	wpa_hexdump(MSG_MSGDUMP, "WPS UPnP: PutWLANResponse NewMessage",
1457		    wpabuf_head(msg), wpabuf_len(msg));
1458	if (ev_type != UPNP_WPS_WLANEVENT_TYPE_EAP) {
1459		wpa_printf(MSG_DEBUG, "WPS UPnP: Ignored unexpected "
1460			   "PutWLANResponse WLANEventType %d", ev_type);
1461		return -1;
1462	}
1463
1464	/*
1465	 * EAP response to ongoing to WPS Registration. Send it to EAP-WSC
1466	 * server implementation for delivery to the peer.
1467	 */
1468
1469	sta = ap_get_sta(hapd, mac_addr);
1470#ifndef CONFIG_WPS_STRICT
1471	if (!sta) {
1472		/*
1473		 * Workaround - Intel wsccmd uses bogus NewWLANEventMAC:
1474		 * Pick STA that is in an ongoing WPS registration without
1475		 * checking the MAC address.
1476		 */
1477		wpa_printf(MSG_DEBUG, "WPS UPnP: No matching STA found based "
1478			   "on NewWLANEventMAC; try wildcard match");
1479		for (sta = hapd->sta_list; sta; sta = sta->next) {
1480			if (sta->eapol_sm && (sta->flags & WLAN_STA_WPS))
1481				break;
1482		}
1483	}
1484#endif /* CONFIG_WPS_STRICT */
1485
1486	if (!sta || !(sta->flags & WLAN_STA_WPS)) {
1487		wpa_printf(MSG_DEBUG, "WPS UPnP: No matching STA found");
1488		return 0;
1489	}
1490
1491	if (!sta->eapol_sm) {
1492		/*
1493		 * This can happen, e.g., if an ER sends an extra message after
1494		 * the station has disassociated (but not fully
1495		 * deauthenticated).
1496		 */
1497		wpa_printf(MSG_DEBUG, "WPS UPnP: Matching STA did not have EAPOL state machine initialized");
1498		return 0;
1499	}
1500
1501	p = os_zalloc(sizeof(*p));
1502	if (p == NULL)
1503		return -1;
1504	os_memcpy(p->addr, sta->addr, ETH_ALEN);
1505	p->msg = wpabuf_dup(msg);
1506	p->type = msg_type;
1507	p->next = hapd->wps->upnp_msgs;
1508	hapd->wps->upnp_msgs = p;
1509
1510	return eapol_auth_eap_pending_cb(sta->eapol_sm, sta->eapol_sm->eap);
1511}
1512
1513
1514static int hostapd_wps_upnp_init(struct hostapd_data *hapd,
1515				 struct wps_context *wps)
1516{
1517	struct upnp_wps_device_ctx *ctx;
1518
1519	if (!hapd->conf->upnp_iface)
1520		return 0;
1521	ctx = os_zalloc(sizeof(*ctx));
1522	if (ctx == NULL)
1523		return -1;
1524
1525	ctx->rx_req_put_wlan_response = hostapd_rx_req_put_wlan_response;
1526	if (hapd->conf->ap_pin)
1527		ctx->ap_pin = os_strdup(hapd->conf->ap_pin);
1528
1529	hapd->wps_upnp = upnp_wps_device_init(ctx, wps, hapd,
1530					      hapd->conf->upnp_iface);
1531	if (hapd->wps_upnp == NULL)
1532		return -1;
1533	wps->wps_upnp = hapd->wps_upnp;
1534
1535	return 0;
1536}
1537
1538
1539static void hostapd_wps_upnp_deinit(struct hostapd_data *hapd)
1540{
1541	upnp_wps_device_deinit(hapd->wps_upnp, hapd);
1542}
1543
1544#endif /* CONFIG_WPS_UPNP */
1545
1546
1547int hostapd_wps_get_mib_sta(struct hostapd_data *hapd, const u8 *addr,
1548			    char *buf, size_t buflen)
1549{
1550	if (hapd->wps == NULL)
1551		return 0;
1552	return wps_registrar_get_info(hapd->wps->registrar, addr, buf, buflen);
1553}
1554
1555
1556static void hostapd_wps_ap_pin_timeout(void *eloop_data, void *user_ctx)
1557{
1558	struct hostapd_data *hapd = eloop_data;
1559	wpa_printf(MSG_DEBUG, "WPS: AP PIN timed out");
1560	hostapd_wps_ap_pin_disable(hapd);
1561	wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_AP_PIN_DISABLED);
1562}
1563
1564
1565static void hostapd_wps_ap_pin_enable(struct hostapd_data *hapd, int timeout)
1566{
1567	wpa_printf(MSG_DEBUG, "WPS: Enabling AP PIN (timeout=%d)", timeout);
1568	hapd->ap_pin_failures = 0;
1569	hapd->ap_pin_failures_consecutive = 0;
1570	hapd->conf->ap_setup_locked = 0;
1571	if (hapd->wps->ap_setup_locked) {
1572		wpa_msg(hapd->msg_ctx, MSG_INFO, WPS_EVENT_AP_SETUP_UNLOCKED);
1573		hapd->wps->ap_setup_locked = 0;
1574		wps_registrar_update_ie(hapd->wps->registrar);
1575	}
1576	eloop_cancel_timeout(hostapd_wps_ap_pin_timeout, hapd, NULL);
1577	if (timeout > 0)
1578		eloop_register_timeout(timeout, 0,
1579				       hostapd_wps_ap_pin_timeout, hapd, NULL);
1580}
1581
1582
1583static int wps_ap_pin_disable(struct hostapd_data *hapd, void *ctx)
1584{
1585	os_free(hapd->conf->ap_pin);
1586	hapd->conf->ap_pin = NULL;
1587#ifdef CONFIG_WPS_UPNP
1588	upnp_wps_set_ap_pin(hapd->wps_upnp, NULL);
1589#endif /* CONFIG_WPS_UPNP */
1590	eloop_cancel_timeout(hostapd_wps_ap_pin_timeout, hapd, NULL);
1591	return 0;
1592}
1593
1594
1595void hostapd_wps_ap_pin_disable(struct hostapd_data *hapd)
1596{
1597	wpa_printf(MSG_DEBUG, "WPS: Disabling AP PIN");
1598	hostapd_wps_for_each(hapd, wps_ap_pin_disable, NULL);
1599}
1600
1601
1602struct wps_ap_pin_data {
1603	char pin_txt[9];
1604	int timeout;
1605};
1606
1607
1608static int wps_ap_pin_set(struct hostapd_data *hapd, void *ctx)
1609{
1610	struct wps_ap_pin_data *data = ctx;
1611
1612	if (!hapd->wps)
1613		return 0;
1614
1615	os_free(hapd->conf->ap_pin);
1616	hapd->conf->ap_pin = os_strdup(data->pin_txt);
1617#ifdef CONFIG_WPS_UPNP
1618	upnp_wps_set_ap_pin(hapd->wps_upnp, data->pin_txt);
1619#endif /* CONFIG_WPS_UPNP */
1620	hostapd_wps_ap_pin_enable(hapd, data->timeout);
1621	return 0;
1622}
1623
1624
1625const char * hostapd_wps_ap_pin_random(struct hostapd_data *hapd, int timeout)
1626{
1627	unsigned int pin;
1628	struct wps_ap_pin_data data;
1629
1630	if (wps_generate_pin(&pin) < 0)
1631		return NULL;
1632	os_snprintf(data.pin_txt, sizeof(data.pin_txt), "%08u", pin);
1633	data.timeout = timeout;
1634	hostapd_wps_for_each(hapd, wps_ap_pin_set, &data);
1635	return hapd->conf->ap_pin;
1636}
1637
1638
1639const char * hostapd_wps_ap_pin_get(struct hostapd_data *hapd)
1640{
1641	return hapd->conf->ap_pin;
1642}
1643
1644
1645int hostapd_wps_ap_pin_set(struct hostapd_data *hapd, const char *pin,
1646			   int timeout)
1647{
1648	struct wps_ap_pin_data data;
1649	int ret;
1650
1651	ret = os_snprintf(data.pin_txt, sizeof(data.pin_txt), "%s", pin);
1652	if (os_snprintf_error(sizeof(data.pin_txt), ret))
1653		return -1;
1654	data.timeout = timeout;
1655	return hostapd_wps_for_each(hapd, wps_ap_pin_set, &data);
1656}
1657
1658
1659static int wps_update_ie(struct hostapd_data *hapd, void *ctx)
1660{
1661	if (hapd->wps)
1662		wps_registrar_update_ie(hapd->wps->registrar);
1663	return 0;
1664}
1665
1666
1667void hostapd_wps_update_ie(struct hostapd_data *hapd)
1668{
1669	hostapd_wps_for_each(hapd, wps_update_ie, NULL);
1670}
1671
1672
1673int hostapd_wps_config_ap(struct hostapd_data *hapd, const char *ssid,
1674			  const char *auth, const char *encr, const char *key)
1675{
1676	struct wps_credential cred;
1677	size_t len;
1678
1679	os_memset(&cred, 0, sizeof(cred));
1680
1681	len = os_strlen(ssid);
1682	if ((len & 1) || len > 2 * sizeof(cred.ssid) ||
1683	    hexstr2bin(ssid, cred.ssid, len / 2))
1684		return -1;
1685	cred.ssid_len = len / 2;
1686
1687	if (os_strncmp(auth, "OPEN", 4) == 0)
1688		cred.auth_type = WPS_AUTH_OPEN;
1689	else if (os_strncmp(auth, "WPAPSK", 6) == 0)
1690		cred.auth_type = WPS_AUTH_WPAPSK;
1691	else if (os_strncmp(auth, "WPA2PSK", 7) == 0)
1692		cred.auth_type = WPS_AUTH_WPA2PSK;
1693	else
1694		return -1;
1695
1696	if (encr) {
1697		if (os_strncmp(encr, "NONE", 4) == 0)
1698			cred.encr_type = WPS_ENCR_NONE;
1699		else if (os_strncmp(encr, "TKIP", 4) == 0)
1700			cred.encr_type = WPS_ENCR_TKIP;
1701		else if (os_strncmp(encr, "CCMP", 4) == 0)
1702			cred.encr_type = WPS_ENCR_AES;
1703		else
1704			return -1;
1705	} else
1706		cred.encr_type = WPS_ENCR_NONE;
1707
1708	if (key) {
1709		len = os_strlen(key);
1710		if ((len & 1) || len > 2 * sizeof(cred.key) ||
1711		    hexstr2bin(key, cred.key, len / 2))
1712			return -1;
1713		cred.key_len = len / 2;
1714	}
1715
1716	return wps_registrar_config_ap(hapd->wps->registrar, &cred);
1717}
1718
1719
1720#ifdef CONFIG_WPS_NFC
1721
1722struct wps_nfc_password_token_data {
1723	const u8 *oob_dev_pw;
1724	size_t oob_dev_pw_len;
1725	int added;
1726};
1727
1728
1729static int wps_add_nfc_password_token(struct hostapd_data *hapd, void *ctx)
1730{
1731	struct wps_nfc_password_token_data *data = ctx;
1732	int ret;
1733
1734	if (hapd->wps == NULL)
1735		return 0;
1736	ret = wps_registrar_add_nfc_password_token(hapd->wps->registrar,
1737						   data->oob_dev_pw,
1738						   data->oob_dev_pw_len);
1739	if (ret == 0)
1740		data->added++;
1741	return ret;
1742}
1743
1744
1745static int hostapd_wps_add_nfc_password_token(struct hostapd_data *hapd,
1746					      struct wps_parse_attr *attr)
1747{
1748	struct wps_nfc_password_token_data data;
1749
1750	data.oob_dev_pw = attr->oob_dev_password;
1751	data.oob_dev_pw_len = attr->oob_dev_password_len;
1752	data.added = 0;
1753	if (hostapd_wps_for_each(hapd, wps_add_nfc_password_token, &data) < 0)
1754		return -1;
1755	return data.added ? 0 : -1;
1756}
1757
1758
1759static int hostapd_wps_nfc_tag_process(struct hostapd_data *hapd,
1760				       const struct wpabuf *wps)
1761{
1762	struct wps_parse_attr attr;
1763
1764	wpa_hexdump_buf(MSG_DEBUG, "WPS: Received NFC tag payload", wps);
1765
1766	if (wps_parse_msg(wps, &attr)) {
1767		wpa_printf(MSG_DEBUG, "WPS: Ignore invalid data from NFC tag");
1768		return -1;
1769	}
1770
1771	if (attr.oob_dev_password)
1772		return hostapd_wps_add_nfc_password_token(hapd, &attr);
1773
1774	wpa_printf(MSG_DEBUG, "WPS: Ignore unrecognized NFC tag");
1775	return -1;
1776}
1777
1778
1779int hostapd_wps_nfc_tag_read(struct hostapd_data *hapd,
1780			     const struct wpabuf *data)
1781{
1782	const struct wpabuf *wps = data;
1783	struct wpabuf *tmp = NULL;
1784	int ret;
1785
1786	if (wpabuf_len(data) < 4)
1787		return -1;
1788
1789	if (*wpabuf_head_u8(data) != 0x10) {
1790		/* Assume this contains full NDEF record */
1791		tmp = ndef_parse_wifi(data);
1792		if (tmp == NULL) {
1793			wpa_printf(MSG_DEBUG, "WPS: Could not parse NDEF");
1794			return -1;
1795		}
1796		wps = tmp;
1797	}
1798
1799	ret = hostapd_wps_nfc_tag_process(hapd, wps);
1800	wpabuf_free(tmp);
1801	return ret;
1802}
1803
1804
1805struct wpabuf * hostapd_wps_nfc_config_token(struct hostapd_data *hapd,
1806					     int ndef)
1807{
1808	struct wpabuf *ret;
1809
1810	if (hapd->wps == NULL)
1811		return NULL;
1812
1813	ret = wps_get_oob_cred(hapd->wps, hostapd_wps_rf_band_cb(hapd),
1814			       hapd->iconf->channel);
1815	if (ndef && ret) {
1816		struct wpabuf *tmp;
1817		tmp = ndef_build_wifi(ret);
1818		wpabuf_free(ret);
1819		if (tmp == NULL)
1820			return NULL;
1821		ret = tmp;
1822	}
1823
1824	return ret;
1825}
1826
1827
1828struct wpabuf * hostapd_wps_nfc_hs_cr(struct hostapd_data *hapd, int ndef)
1829{
1830	struct wpabuf *ret;
1831
1832	if (hapd->wps == NULL)
1833		return NULL;
1834
1835	if (hapd->conf->wps_nfc_dh_pubkey == NULL) {
1836		struct wps_context *wps = hapd->wps;
1837		if (wps_nfc_gen_dh(&hapd->conf->wps_nfc_dh_pubkey,
1838				   &hapd->conf->wps_nfc_dh_privkey) < 0)
1839			return NULL;
1840		hostapd_wps_nfc_clear(wps);
1841		wps->ap_nfc_dev_pw_id = DEV_PW_NFC_CONNECTION_HANDOVER;
1842		wps->ap_nfc_dh_pubkey =
1843			wpabuf_dup(hapd->conf->wps_nfc_dh_pubkey);
1844		wps->ap_nfc_dh_privkey =
1845			wpabuf_dup(hapd->conf->wps_nfc_dh_privkey);
1846		if (!wps->ap_nfc_dh_pubkey || !wps->ap_nfc_dh_privkey) {
1847			hostapd_wps_nfc_clear(wps);
1848			return NULL;
1849		}
1850	}
1851
1852	ret = wps_build_nfc_handover_sel(hapd->wps,
1853					 hapd->conf->wps_nfc_dh_pubkey,
1854					 hapd->own_addr, hapd->iface->freq);
1855
1856	if (ndef && ret) {
1857		struct wpabuf *tmp;
1858		tmp = ndef_build_wifi(ret);
1859		wpabuf_free(ret);
1860		if (tmp == NULL)
1861			return NULL;
1862		ret = tmp;
1863	}
1864
1865	return ret;
1866}
1867
1868
1869int hostapd_wps_nfc_report_handover(struct hostapd_data *hapd,
1870				    const struct wpabuf *req,
1871				    const struct wpabuf *sel)
1872{
1873	struct wpabuf *wps;
1874	int ret = -1;
1875	u16 wsc_len;
1876	const u8 *pos;
1877	struct wpabuf msg;
1878	struct wps_parse_attr attr;
1879	u16 dev_pw_id;
1880
1881	/*
1882	 * Enrollee/station is always initiator of the NFC connection handover,
1883	 * so use the request message here to find Enrollee public key hash.
1884	 */
1885	wps = ndef_parse_wifi(req);
1886	if (wps == NULL)
1887		return -1;
1888	wpa_printf(MSG_DEBUG, "WPS: Received application/vnd.wfa.wsc "
1889		   "payload from NFC connection handover");
1890	wpa_hexdump_buf(MSG_DEBUG, "WPS: NFC payload", wps);
1891	if (wpabuf_len(wps) < 2) {
1892		wpa_printf(MSG_DEBUG, "WPS: Too short Wi-Fi Handover Request "
1893			   "Message");
1894		goto out;
1895	}
1896	pos = wpabuf_head(wps);
1897	wsc_len = WPA_GET_BE16(pos);
1898	if (wsc_len > wpabuf_len(wps) - 2) {
1899		wpa_printf(MSG_DEBUG, "WPS: Invalid WSC attribute length (%u) "
1900			   "in rt Wi-Fi Handover Request Message", wsc_len);
1901		goto out;
1902	}
1903	pos += 2;
1904
1905	wpa_hexdump(MSG_DEBUG,
1906		    "WPS: WSC attributes in Wi-Fi Handover Request Message",
1907		    pos, wsc_len);
1908	if (wsc_len < wpabuf_len(wps) - 2) {
1909		wpa_hexdump(MSG_DEBUG,
1910			    "WPS: Ignore extra data after WSC attributes",
1911			    pos + wsc_len, wpabuf_len(wps) - 2 - wsc_len);
1912	}
1913
1914	wpabuf_set(&msg, pos, wsc_len);
1915	ret = wps_parse_msg(&msg, &attr);
1916	if (ret < 0) {
1917		wpa_printf(MSG_DEBUG, "WPS: Could not parse WSC attributes in "
1918			   "Wi-Fi Handover Request Message");
1919		goto out;
1920	}
1921
1922	if (attr.oob_dev_password == NULL ||
1923	    attr.oob_dev_password_len < WPS_OOB_PUBKEY_HASH_LEN + 2) {
1924		wpa_printf(MSG_DEBUG, "WPS: No Out-of-Band Device Password "
1925			   "included in Wi-Fi Handover Request Message");
1926		ret = -1;
1927		goto out;
1928	}
1929
1930	if (attr.uuid_e == NULL) {
1931		wpa_printf(MSG_DEBUG, "WPS: No UUID-E included in Wi-Fi "
1932			   "Handover Request Message");
1933		ret = -1;
1934		goto out;
1935	}
1936
1937	wpa_hexdump(MSG_DEBUG, "WPS: UUID-E", attr.uuid_e, WPS_UUID_LEN);
1938
1939	wpa_hexdump(MSG_DEBUG, "WPS: Out-of-Band Device Password",
1940		    attr.oob_dev_password, attr.oob_dev_password_len);
1941	dev_pw_id = WPA_GET_BE16(attr.oob_dev_password +
1942				 WPS_OOB_PUBKEY_HASH_LEN);
1943	if (dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER) {
1944		wpa_printf(MSG_DEBUG, "WPS: Unexpected OOB Device Password ID "
1945			   "%u in Wi-Fi Handover Request Message", dev_pw_id);
1946		ret = -1;
1947		goto out;
1948	}
1949	wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Public Key hash",
1950		    attr.oob_dev_password, WPS_OOB_PUBKEY_HASH_LEN);
1951
1952	ret = wps_registrar_add_nfc_pw_token(hapd->wps->registrar,
1953					     attr.oob_dev_password,
1954					     DEV_PW_NFC_CONNECTION_HANDOVER,
1955					     NULL, 0, 1);
1956
1957out:
1958	wpabuf_free(wps);
1959	return ret;
1960}
1961
1962
1963struct wpabuf * hostapd_wps_nfc_token_gen(struct hostapd_data *hapd, int ndef)
1964{
1965	if (hapd->conf->wps_nfc_pw_from_config) {
1966		return wps_nfc_token_build(ndef,
1967					   hapd->conf->wps_nfc_dev_pw_id,
1968					   hapd->conf->wps_nfc_dh_pubkey,
1969					   hapd->conf->wps_nfc_dev_pw);
1970	}
1971
1972	return wps_nfc_token_gen(ndef, &hapd->conf->wps_nfc_dev_pw_id,
1973				 &hapd->conf->wps_nfc_dh_pubkey,
1974				 &hapd->conf->wps_nfc_dh_privkey,
1975				 &hapd->conf->wps_nfc_dev_pw);
1976}
1977
1978
1979int hostapd_wps_nfc_token_enable(struct hostapd_data *hapd)
1980{
1981	struct wps_context *wps = hapd->wps;
1982	struct wpabuf *pw;
1983
1984	if (wps == NULL)
1985		return -1;
1986
1987	if (!hapd->conf->wps_nfc_dh_pubkey ||
1988	    !hapd->conf->wps_nfc_dh_privkey ||
1989	    !hapd->conf->wps_nfc_dev_pw ||
1990	    !hapd->conf->wps_nfc_dev_pw_id)
1991		return -1;
1992
1993	hostapd_wps_nfc_clear(wps);
1994	wpa_printf(MSG_DEBUG,
1995		   "WPS: Enable NFC Tag (Dev Pw Id %u) for AP interface %s (context %p)",
1996		   hapd->conf->wps_nfc_dev_pw_id, hapd->conf->iface, wps);
1997	wps->ap_nfc_dev_pw_id = hapd->conf->wps_nfc_dev_pw_id;
1998	wps->ap_nfc_dh_pubkey = wpabuf_dup(hapd->conf->wps_nfc_dh_pubkey);
1999	wps->ap_nfc_dh_privkey = wpabuf_dup(hapd->conf->wps_nfc_dh_privkey);
2000	pw = hapd->conf->wps_nfc_dev_pw;
2001	wps->ap_nfc_dev_pw = wpabuf_alloc(
2002		wpabuf_len(pw) * 2 + 1);
2003	if (wps->ap_nfc_dev_pw) {
2004		wpa_snprintf_hex_uppercase(
2005			(char *) wpabuf_put(wps->ap_nfc_dev_pw,
2006					    wpabuf_len(pw) * 2),
2007			wpabuf_len(pw) * 2 + 1,
2008			wpabuf_head(pw), wpabuf_len(pw));
2009	}
2010
2011	if (!wps->ap_nfc_dh_pubkey || !wps->ap_nfc_dh_privkey ||
2012	    !wps->ap_nfc_dev_pw) {
2013		hostapd_wps_nfc_clear(wps);
2014		return -1;
2015	}
2016
2017	return 0;
2018}
2019
2020
2021void hostapd_wps_nfc_token_disable(struct hostapd_data *hapd)
2022{
2023	wpa_printf(MSG_DEBUG, "WPS: Disable NFC token for AP interface %s",
2024		   hapd->conf->iface);
2025	hostapd_wps_nfc_clear(hapd->wps);
2026}
2027
2028#endif /* CONFIG_WPS_NFC */
2029