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