Home | History | Annotate | Download | only in wps
      1 /*
      2  * Wi-Fi Protected Setup - Registrar
      3  * Copyright (c) 2008-2013, 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/base64.h"
     13 #include "utils/eloop.h"
     14 #include "utils/uuid.h"
     15 #include "utils/list.h"
     16 #include "crypto/crypto.h"
     17 #include "crypto/sha256.h"
     18 #include "crypto/random.h"
     19 #include "common/ieee802_11_defs.h"
     20 #include "wps_i.h"
     21 #include "wps_dev_attr.h"
     22 #include "wps_upnp.h"
     23 #include "wps_upnp_i.h"
     24 
     25 #ifndef CONFIG_WPS_STRICT
     26 #define WPS_WORKAROUNDS
     27 #endif /* CONFIG_WPS_STRICT */
     28 
     29 #ifdef CONFIG_WPS_NFC
     30 
     31 struct wps_nfc_pw_token {
     32 	struct dl_list list;
     33 	u8 pubkey_hash[WPS_OOB_PUBKEY_HASH_LEN];
     34 	unsigned int peer_pk_hash_known:1;
     35 	u16 pw_id;
     36 	u8 dev_pw[WPS_OOB_DEVICE_PASSWORD_LEN * 2 + 1];
     37 	size_t dev_pw_len;
     38 	int pk_hash_provided_oob; /* whether own PK hash was provided OOB */
     39 };
     40 
     41 
     42 static void wps_remove_nfc_pw_token(struct wps_nfc_pw_token *token)
     43 {
     44 	dl_list_del(&token->list);
     45 	bin_clear_free(token, sizeof(*token));
     46 }
     47 
     48 
     49 static void wps_free_nfc_pw_tokens(struct dl_list *tokens, u16 pw_id)
     50 {
     51 	struct wps_nfc_pw_token *token, *prev;
     52 	dl_list_for_each_safe(token, prev, tokens, struct wps_nfc_pw_token,
     53 			      list) {
     54 		if (pw_id == 0 || pw_id == token->pw_id)
     55 			wps_remove_nfc_pw_token(token);
     56 	}
     57 }
     58 
     59 
     60 static struct wps_nfc_pw_token * wps_get_nfc_pw_token(struct dl_list *tokens,
     61 						      u16 pw_id)
     62 {
     63 	struct wps_nfc_pw_token *token;
     64 	dl_list_for_each(token, tokens, struct wps_nfc_pw_token, list) {
     65 		if (pw_id == token->pw_id)
     66 			return token;
     67 	}
     68 	return NULL;
     69 }
     70 
     71 #else /* CONFIG_WPS_NFC */
     72 
     73 #define wps_free_nfc_pw_tokens(t, p) do { } while (0)
     74 
     75 #endif /* CONFIG_WPS_NFC */
     76 
     77 
     78 struct wps_uuid_pin {
     79 	struct dl_list list;
     80 	u8 uuid[WPS_UUID_LEN];
     81 	int wildcard_uuid;
     82 	u8 *pin;
     83 	size_t pin_len;
     84 #define PIN_LOCKED BIT(0)
     85 #define PIN_EXPIRES BIT(1)
     86 	int flags;
     87 	struct os_reltime expiration;
     88 	u8 enrollee_addr[ETH_ALEN];
     89 };
     90 
     91 
     92 static void wps_free_pin(struct wps_uuid_pin *pin)
     93 {
     94 	bin_clear_free(pin->pin, pin->pin_len);
     95 	os_free(pin);
     96 }
     97 
     98 
     99 static void wps_remove_pin(struct wps_uuid_pin *pin)
    100 {
    101 	dl_list_del(&pin->list);
    102 	wps_free_pin(pin);
    103 }
    104 
    105 
    106 static void wps_free_pins(struct dl_list *pins)
    107 {
    108 	struct wps_uuid_pin *pin, *prev;
    109 	dl_list_for_each_safe(pin, prev, pins, struct wps_uuid_pin, list)
    110 		wps_remove_pin(pin);
    111 }
    112 
    113 
    114 struct wps_pbc_session {
    115 	struct wps_pbc_session *next;
    116 	u8 addr[ETH_ALEN];
    117 	u8 uuid_e[WPS_UUID_LEN];
    118 	struct os_reltime timestamp;
    119 };
    120 
    121 
    122 static void wps_free_pbc_sessions(struct wps_pbc_session *pbc)
    123 {
    124 	struct wps_pbc_session *prev;
    125 
    126 	while (pbc) {
    127 		prev = pbc;
    128 		pbc = pbc->next;
    129 		os_free(prev);
    130 	}
    131 }
    132 
    133 
    134 struct wps_registrar_device {
    135 	struct wps_registrar_device *next;
    136 	struct wps_device_data dev;
    137 	u8 uuid[WPS_UUID_LEN];
    138 };
    139 
    140 
    141 struct wps_registrar {
    142 	struct wps_context *wps;
    143 
    144 	int pbc;
    145 	int selected_registrar;
    146 
    147 	int (*new_psk_cb)(void *ctx, const u8 *mac_addr, const u8 *p2p_dev_addr,
    148 			  const u8 *psk, size_t psk_len);
    149 	int (*set_ie_cb)(void *ctx, struct wpabuf *beacon_ie,
    150 			 struct wpabuf *probe_resp_ie);
    151 	void (*pin_needed_cb)(void *ctx, const u8 *uuid_e,
    152 			      const struct wps_device_data *dev);
    153 	void (*reg_success_cb)(void *ctx, const u8 *mac_addr,
    154 			       const u8 *uuid_e, const u8 *dev_pw,
    155 			       size_t dev_pw_len);
    156 	void (*set_sel_reg_cb)(void *ctx, int sel_reg, u16 dev_passwd_id,
    157 			       u16 sel_reg_config_methods);
    158 	void (*enrollee_seen_cb)(void *ctx, const u8 *addr, const u8 *uuid_e,
    159 				 const u8 *pri_dev_type, u16 config_methods,
    160 				 u16 dev_password_id, u8 request_type,
    161 				 const char *dev_name);
    162 	void *cb_ctx;
    163 
    164 	struct dl_list pins;
    165 	struct dl_list nfc_pw_tokens;
    166 	struct wps_pbc_session *pbc_sessions;
    167 
    168 	int skip_cred_build;
    169 	struct wpabuf *extra_cred;
    170 	int disable_auto_conf;
    171 	int sel_reg_union;
    172 	int sel_reg_dev_password_id_override;
    173 	int sel_reg_config_methods_override;
    174 	int static_wep_only;
    175 	int dualband;
    176 	int force_per_enrollee_psk;
    177 
    178 	struct wps_registrar_device *devices;
    179 
    180 	int force_pbc_overlap;
    181 
    182 	u8 authorized_macs[WPS_MAX_AUTHORIZED_MACS][ETH_ALEN];
    183 	u8 authorized_macs_union[WPS_MAX_AUTHORIZED_MACS][ETH_ALEN];
    184 
    185 	u8 p2p_dev_addr[ETH_ALEN];
    186 
    187 	u8 pbc_ignore_uuid[WPS_UUID_LEN];
    188 #ifdef WPS_WORKAROUNDS
    189 	struct os_reltime pbc_ignore_start;
    190 #endif /* WPS_WORKAROUNDS */
    191 };
    192 
    193 
    194 static int wps_set_ie(struct wps_registrar *reg);
    195 static void wps_registrar_pbc_timeout(void *eloop_ctx, void *timeout_ctx);
    196 static void wps_registrar_set_selected_timeout(void *eloop_ctx,
    197 					       void *timeout_ctx);
    198 static void wps_registrar_remove_pin(struct wps_registrar *reg,
    199 				     struct wps_uuid_pin *pin);
    200 
    201 
    202 static void wps_registrar_add_authorized_mac(struct wps_registrar *reg,
    203 					     const u8 *addr)
    204 {
    205 	int i;
    206 	wpa_printf(MSG_DEBUG, "WPS: Add authorized MAC " MACSTR,
    207 		   MAC2STR(addr));
    208 	for (i = 0; i < WPS_MAX_AUTHORIZED_MACS; i++)
    209 		if (os_memcmp(reg->authorized_macs[i], addr, ETH_ALEN) == 0) {
    210 			wpa_printf(MSG_DEBUG, "WPS: Authorized MAC was "
    211 				   "already in the list");
    212 			return; /* already in list */
    213 		}
    214 	for (i = WPS_MAX_AUTHORIZED_MACS - 1; i > 0; i--)
    215 		os_memcpy(reg->authorized_macs[i], reg->authorized_macs[i - 1],
    216 			  ETH_ALEN);
    217 	os_memcpy(reg->authorized_macs[0], addr, ETH_ALEN);
    218 	wpa_hexdump(MSG_DEBUG, "WPS: Authorized MACs",
    219 		    (u8 *) reg->authorized_macs, sizeof(reg->authorized_macs));
    220 }
    221 
    222 
    223 static void wps_registrar_remove_authorized_mac(struct wps_registrar *reg,
    224 						const u8 *addr)
    225 {
    226 	int i;
    227 	wpa_printf(MSG_DEBUG, "WPS: Remove authorized MAC " MACSTR,
    228 		   MAC2STR(addr));
    229 	for (i = 0; i < WPS_MAX_AUTHORIZED_MACS; i++) {
    230 		if (os_memcmp(reg->authorized_macs, addr, ETH_ALEN) == 0)
    231 			break;
    232 	}
    233 	if (i == WPS_MAX_AUTHORIZED_MACS) {
    234 		wpa_printf(MSG_DEBUG, "WPS: Authorized MAC was not in the "
    235 			   "list");
    236 		return; /* not in the list */
    237 	}
    238 	for (; i + 1 < WPS_MAX_AUTHORIZED_MACS; i++)
    239 		os_memcpy(reg->authorized_macs[i], reg->authorized_macs[i + 1],
    240 			  ETH_ALEN);
    241 	os_memset(reg->authorized_macs[WPS_MAX_AUTHORIZED_MACS - 1], 0,
    242 		  ETH_ALEN);
    243 	wpa_hexdump(MSG_DEBUG, "WPS: Authorized MACs",
    244 		    (u8 *) reg->authorized_macs, sizeof(reg->authorized_macs));
    245 }
    246 
    247 
    248 static void wps_free_devices(struct wps_registrar_device *dev)
    249 {
    250 	struct wps_registrar_device *prev;
    251 
    252 	while (dev) {
    253 		prev = dev;
    254 		dev = dev->next;
    255 		wps_device_data_free(&prev->dev);
    256 		os_free(prev);
    257 	}
    258 }
    259 
    260 
    261 static struct wps_registrar_device * wps_device_get(struct wps_registrar *reg,
    262 						    const u8 *addr)
    263 {
    264 	struct wps_registrar_device *dev;
    265 
    266 	for (dev = reg->devices; dev; dev = dev->next) {
    267 		if (os_memcmp(dev->dev.mac_addr, addr, ETH_ALEN) == 0)
    268 			return dev;
    269 	}
    270 	return NULL;
    271 }
    272 
    273 
    274 static void wps_device_clone_data(struct wps_device_data *dst,
    275 				  struct wps_device_data *src)
    276 {
    277 	os_memcpy(dst->mac_addr, src->mac_addr, ETH_ALEN);
    278 	os_memcpy(dst->pri_dev_type, src->pri_dev_type, WPS_DEV_TYPE_LEN);
    279 
    280 #define WPS_STRDUP(n) \
    281 	os_free(dst->n); \
    282 	dst->n = src->n ? os_strdup(src->n) : NULL
    283 
    284 	WPS_STRDUP(device_name);
    285 	WPS_STRDUP(manufacturer);
    286 	WPS_STRDUP(model_name);
    287 	WPS_STRDUP(model_number);
    288 	WPS_STRDUP(serial_number);
    289 #undef WPS_STRDUP
    290 }
    291 
    292 
    293 int wps_device_store(struct wps_registrar *reg,
    294 		     struct wps_device_data *dev, const u8 *uuid)
    295 {
    296 	struct wps_registrar_device *d;
    297 
    298 	d = wps_device_get(reg, dev->mac_addr);
    299 	if (d == NULL) {
    300 		d = os_zalloc(sizeof(*d));
    301 		if (d == NULL)
    302 			return -1;
    303 		d->next = reg->devices;
    304 		reg->devices = d;
    305 	}
    306 
    307 	wps_device_clone_data(&d->dev, dev);
    308 	os_memcpy(d->uuid, uuid, WPS_UUID_LEN);
    309 
    310 	return 0;
    311 }
    312 
    313 
    314 static void wps_registrar_add_pbc_session(struct wps_registrar *reg,
    315 					  const u8 *addr, const u8 *uuid_e)
    316 {
    317 	struct wps_pbc_session *pbc, *prev = NULL;
    318 	struct os_reltime now;
    319 
    320 	os_get_reltime(&now);
    321 
    322 	pbc = reg->pbc_sessions;
    323 	while (pbc) {
    324 		if (os_memcmp(pbc->addr, addr, ETH_ALEN) == 0 &&
    325 		    os_memcmp(pbc->uuid_e, uuid_e, WPS_UUID_LEN) == 0) {
    326 			if (prev)
    327 				prev->next = pbc->next;
    328 			else
    329 				reg->pbc_sessions = pbc->next;
    330 			break;
    331 		}
    332 		prev = pbc;
    333 		pbc = pbc->next;
    334 	}
    335 
    336 	if (!pbc) {
    337 		pbc = os_zalloc(sizeof(*pbc));
    338 		if (pbc == NULL)
    339 			return;
    340 		os_memcpy(pbc->addr, addr, ETH_ALEN);
    341 		if (uuid_e)
    342 			os_memcpy(pbc->uuid_e, uuid_e, WPS_UUID_LEN);
    343 	}
    344 
    345 	pbc->next = reg->pbc_sessions;
    346 	reg->pbc_sessions = pbc;
    347 	pbc->timestamp = now;
    348 
    349 	/* remove entries that have timed out */
    350 	prev = pbc;
    351 	pbc = pbc->next;
    352 
    353 	while (pbc) {
    354 		if (os_reltime_expired(&now, &pbc->timestamp,
    355 				       WPS_PBC_WALK_TIME)) {
    356 			prev->next = NULL;
    357 			wps_free_pbc_sessions(pbc);
    358 			break;
    359 		}
    360 		prev = pbc;
    361 		pbc = pbc->next;
    362 	}
    363 }
    364 
    365 
    366 static void wps_registrar_remove_pbc_session(struct wps_registrar *reg,
    367 					     const u8 *uuid_e,
    368 					     const u8 *p2p_dev_addr)
    369 {
    370 	struct wps_pbc_session *pbc, *prev = NULL, *tmp;
    371 
    372 	pbc = reg->pbc_sessions;
    373 	while (pbc) {
    374 		if (os_memcmp(pbc->uuid_e, uuid_e, WPS_UUID_LEN) == 0 ||
    375 		    (p2p_dev_addr && !is_zero_ether_addr(reg->p2p_dev_addr) &&
    376 		     os_memcmp(reg->p2p_dev_addr, p2p_dev_addr, ETH_ALEN) ==
    377 		     0)) {
    378 			if (prev)
    379 				prev->next = pbc->next;
    380 			else
    381 				reg->pbc_sessions = pbc->next;
    382 			tmp = pbc;
    383 			pbc = pbc->next;
    384 			wpa_printf(MSG_DEBUG, "WPS: Removing PBC session for "
    385 				   "addr=" MACSTR, MAC2STR(tmp->addr));
    386 			wpa_hexdump(MSG_DEBUG, "WPS: Removed UUID-E",
    387 				    tmp->uuid_e, WPS_UUID_LEN);
    388 			os_free(tmp);
    389 			continue;
    390 		}
    391 		prev = pbc;
    392 		pbc = pbc->next;
    393 	}
    394 }
    395 
    396 
    397 int wps_registrar_pbc_overlap(struct wps_registrar *reg,
    398 			      const u8 *addr, const u8 *uuid_e)
    399 {
    400 	int count = 0;
    401 	struct wps_pbc_session *pbc;
    402 	struct wps_pbc_session *first = NULL;
    403 	struct os_reltime now;
    404 
    405 	os_get_reltime(&now);
    406 
    407 	wpa_printf(MSG_DEBUG, "WPS: Checking active PBC sessions for overlap");
    408 
    409 	if (uuid_e) {
    410 		wpa_printf(MSG_DEBUG, "WPS: Add one for the requested UUID");
    411 		wpa_hexdump(MSG_DEBUG, "WPS: Requested UUID",
    412 			    uuid_e, WPS_UUID_LEN);
    413 		count++;
    414 	}
    415 
    416 	for (pbc = reg->pbc_sessions; pbc; pbc = pbc->next) {
    417 		wpa_printf(MSG_DEBUG, "WPS: Consider PBC session with " MACSTR,
    418 			   MAC2STR(pbc->addr));
    419 		wpa_hexdump(MSG_DEBUG, "WPS: UUID-E",
    420 			    pbc->uuid_e, WPS_UUID_LEN);
    421 		if (os_reltime_expired(&now, &pbc->timestamp,
    422 				       WPS_PBC_WALK_TIME)) {
    423 			wpa_printf(MSG_DEBUG, "WPS: PBC walk time has expired");
    424 			break;
    425 		}
    426 		if (first &&
    427 		    os_memcmp(pbc->uuid_e, first->uuid_e, WPS_UUID_LEN) == 0) {
    428 			wpa_printf(MSG_DEBUG, "WPS: Same Enrollee");
    429 			continue; /* same Enrollee */
    430 		}
    431 		if (uuid_e == NULL ||
    432 		    os_memcmp(uuid_e, pbc->uuid_e, WPS_UUID_LEN)) {
    433 			wpa_printf(MSG_DEBUG, "WPS: New Enrollee");
    434 			count++;
    435 		}
    436 		if (first == NULL)
    437 			first = pbc;
    438 	}
    439 
    440 	wpa_printf(MSG_DEBUG, "WPS: %u active PBC session(s) found", count);
    441 
    442 	return count > 1 ? 1 : 0;
    443 }
    444 
    445 
    446 static int wps_build_wps_state(struct wps_context *wps, struct wpabuf *msg)
    447 {
    448 	wpa_printf(MSG_DEBUG, "WPS:  * Wi-Fi Protected Setup State (%d)",
    449 		   wps->wps_state);
    450 	wpabuf_put_be16(msg, ATTR_WPS_STATE);
    451 	wpabuf_put_be16(msg, 1);
    452 	wpabuf_put_u8(msg, wps->wps_state);
    453 	return 0;
    454 }
    455 
    456 
    457 #ifdef CONFIG_WPS_UPNP
    458 static void wps_registrar_free_pending_m2(struct wps_context *wps)
    459 {
    460 	struct upnp_pending_message *p, *p2, *prev = NULL;
    461 	p = wps->upnp_msgs;
    462 	while (p) {
    463 		if (p->type == WPS_M2 || p->type == WPS_M2D) {
    464 			if (prev == NULL)
    465 				wps->upnp_msgs = p->next;
    466 			else
    467 				prev->next = p->next;
    468 			wpa_printf(MSG_DEBUG, "WPS UPnP: Drop pending M2/M2D");
    469 			p2 = p;
    470 			p = p->next;
    471 			wpabuf_free(p2->msg);
    472 			os_free(p2);
    473 			continue;
    474 		}
    475 		prev = p;
    476 		p = p->next;
    477 	}
    478 }
    479 #endif /* CONFIG_WPS_UPNP */
    480 
    481 
    482 static int wps_build_ap_setup_locked(struct wps_context *wps,
    483 				     struct wpabuf *msg)
    484 {
    485 	if (wps->ap_setup_locked && wps->ap_setup_locked != 2) {
    486 		wpa_printf(MSG_DEBUG, "WPS:  * AP Setup Locked");
    487 		wpabuf_put_be16(msg, ATTR_AP_SETUP_LOCKED);
    488 		wpabuf_put_be16(msg, 1);
    489 		wpabuf_put_u8(msg, 1);
    490 	}
    491 	return 0;
    492 }
    493 
    494 
    495 static int wps_build_selected_registrar(struct wps_registrar *reg,
    496 					struct wpabuf *msg)
    497 {
    498 	if (!reg->sel_reg_union)
    499 		return 0;
    500 	wpa_printf(MSG_DEBUG, "WPS:  * Selected Registrar");
    501 	wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR);
    502 	wpabuf_put_be16(msg, 1);
    503 	wpabuf_put_u8(msg, 1);
    504 	return 0;
    505 }
    506 
    507 
    508 static int wps_build_sel_reg_dev_password_id(struct wps_registrar *reg,
    509 					     struct wpabuf *msg)
    510 {
    511 	u16 id = reg->pbc ? DEV_PW_PUSHBUTTON : DEV_PW_DEFAULT;
    512 	if (!reg->sel_reg_union)
    513 		return 0;
    514 	if (reg->sel_reg_dev_password_id_override >= 0)
    515 		id = reg->sel_reg_dev_password_id_override;
    516 	wpa_printf(MSG_DEBUG, "WPS:  * Device Password ID (%d)", id);
    517 	wpabuf_put_be16(msg, ATTR_DEV_PASSWORD_ID);
    518 	wpabuf_put_be16(msg, 2);
    519 	wpabuf_put_be16(msg, id);
    520 	return 0;
    521 }
    522 
    523 
    524 static int wps_build_sel_pbc_reg_uuid_e(struct wps_registrar *reg,
    525 					struct wpabuf *msg)
    526 {
    527 	u16 id = reg->pbc ? DEV_PW_PUSHBUTTON : DEV_PW_DEFAULT;
    528 	if (!reg->sel_reg_union)
    529 		return 0;
    530 	if (reg->sel_reg_dev_password_id_override >= 0)
    531 		id = reg->sel_reg_dev_password_id_override;
    532 	if (id != DEV_PW_PUSHBUTTON || !reg->dualband)
    533 		return 0;
    534 	return wps_build_uuid_e(msg, reg->wps->uuid);
    535 }
    536 
    537 
    538 static void wps_set_pushbutton(u16 *methods, u16 conf_methods)
    539 {
    540 	*methods |= WPS_CONFIG_PUSHBUTTON;
    541 	if ((conf_methods & WPS_CONFIG_VIRT_PUSHBUTTON) ==
    542 	    WPS_CONFIG_VIRT_PUSHBUTTON)
    543 		*methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
    544 	if ((conf_methods & WPS_CONFIG_PHY_PUSHBUTTON) ==
    545 	    WPS_CONFIG_PHY_PUSHBUTTON)
    546 		*methods |= WPS_CONFIG_PHY_PUSHBUTTON;
    547 	if ((*methods & WPS_CONFIG_VIRT_PUSHBUTTON) !=
    548 	    WPS_CONFIG_VIRT_PUSHBUTTON &&
    549 	    (*methods & WPS_CONFIG_PHY_PUSHBUTTON) !=
    550 	    WPS_CONFIG_PHY_PUSHBUTTON) {
    551 		/*
    552 		 * Required to include virtual/physical flag, but we were not
    553 		 * configured with push button type, so have to default to one
    554 		 * of them.
    555 		 */
    556 		*methods |= WPS_CONFIG_PHY_PUSHBUTTON;
    557 	}
    558 }
    559 
    560 
    561 static int wps_build_sel_reg_config_methods(struct wps_registrar *reg,
    562 					    struct wpabuf *msg)
    563 {
    564 	u16 methods;
    565 	if (!reg->sel_reg_union)
    566 		return 0;
    567 	methods = reg->wps->config_methods;
    568 	methods &= ~WPS_CONFIG_PUSHBUTTON;
    569 	methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
    570 		     WPS_CONFIG_PHY_PUSHBUTTON);
    571 	if (reg->pbc)
    572 		wps_set_pushbutton(&methods, reg->wps->config_methods);
    573 	if (reg->sel_reg_config_methods_override >= 0)
    574 		methods = reg->sel_reg_config_methods_override;
    575 	wpa_printf(MSG_DEBUG, "WPS:  * Selected Registrar Config Methods (%x)",
    576 		   methods);
    577 	wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR_CONFIG_METHODS);
    578 	wpabuf_put_be16(msg, 2);
    579 	wpabuf_put_be16(msg, methods);
    580 	return 0;
    581 }
    582 
    583 
    584 static int wps_build_probe_config_methods(struct wps_registrar *reg,
    585 					  struct wpabuf *msg)
    586 {
    587 	u16 methods;
    588 	/*
    589 	 * These are the methods that the AP supports as an Enrollee for adding
    590 	 * external Registrars.
    591 	 */
    592 	methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
    593 	methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
    594 		     WPS_CONFIG_PHY_PUSHBUTTON);
    595 	wpa_printf(MSG_DEBUG, "WPS:  * Config Methods (%x)", methods);
    596 	wpabuf_put_be16(msg, ATTR_CONFIG_METHODS);
    597 	wpabuf_put_be16(msg, 2);
    598 	wpabuf_put_be16(msg, methods);
    599 	return 0;
    600 }
    601 
    602 
    603 static int wps_build_config_methods_r(struct wps_registrar *reg,
    604 				      struct wpabuf *msg)
    605 {
    606 	return wps_build_config_methods(msg, reg->wps->config_methods);
    607 }
    608 
    609 
    610 const u8 * wps_authorized_macs(struct wps_registrar *reg, size_t *count)
    611 {
    612 	*count = 0;
    613 
    614 	while (*count < WPS_MAX_AUTHORIZED_MACS) {
    615 		if (is_zero_ether_addr(reg->authorized_macs_union[*count]))
    616 			break;
    617 		(*count)++;
    618 	}
    619 
    620 	return (const u8 *) reg->authorized_macs_union;
    621 }
    622 
    623 
    624 /**
    625  * wps_registrar_init - Initialize WPS Registrar data
    626  * @wps: Pointer to longterm WPS context
    627  * @cfg: Registrar configuration
    628  * Returns: Pointer to allocated Registrar data or %NULL on failure
    629  *
    630  * This function is used to initialize WPS Registrar functionality. It can be
    631  * used for a single Registrar run (e.g., when run in a supplicant) or multiple
    632  * runs (e.g., when run as an internal Registrar in an AP). Caller is
    633  * responsible for freeing the returned data with wps_registrar_deinit() when
    634  * Registrar functionality is not needed anymore.
    635  */
    636 struct wps_registrar *
    637 wps_registrar_init(struct wps_context *wps,
    638 		   const struct wps_registrar_config *cfg)
    639 {
    640 	struct wps_registrar *reg = os_zalloc(sizeof(*reg));
    641 	if (reg == NULL)
    642 		return NULL;
    643 
    644 	dl_list_init(&reg->pins);
    645 	dl_list_init(&reg->nfc_pw_tokens);
    646 	reg->wps = wps;
    647 	reg->new_psk_cb = cfg->new_psk_cb;
    648 	reg->set_ie_cb = cfg->set_ie_cb;
    649 	reg->pin_needed_cb = cfg->pin_needed_cb;
    650 	reg->reg_success_cb = cfg->reg_success_cb;
    651 	reg->set_sel_reg_cb = cfg->set_sel_reg_cb;
    652 	reg->enrollee_seen_cb = cfg->enrollee_seen_cb;
    653 	reg->cb_ctx = cfg->cb_ctx;
    654 	reg->skip_cred_build = cfg->skip_cred_build;
    655 	if (cfg->extra_cred) {
    656 		reg->extra_cred = wpabuf_alloc_copy(cfg->extra_cred,
    657 						    cfg->extra_cred_len);
    658 		if (reg->extra_cred == NULL) {
    659 			os_free(reg);
    660 			return NULL;
    661 		}
    662 	}
    663 	reg->disable_auto_conf = cfg->disable_auto_conf;
    664 	reg->sel_reg_dev_password_id_override = -1;
    665 	reg->sel_reg_config_methods_override = -1;
    666 	reg->static_wep_only = cfg->static_wep_only;
    667 	reg->dualband = cfg->dualband;
    668 	reg->force_per_enrollee_psk = cfg->force_per_enrollee_psk;
    669 
    670 	if (wps_set_ie(reg)) {
    671 		wps_registrar_deinit(reg);
    672 		return NULL;
    673 	}
    674 
    675 	return reg;
    676 }
    677 
    678 
    679 /**
    680  * wps_registrar_deinit - Deinitialize WPS Registrar data
    681  * @reg: Registrar data from wps_registrar_init()
    682  */
    683 void wps_registrar_deinit(struct wps_registrar *reg)
    684 {
    685 	if (reg == NULL)
    686 		return;
    687 	eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
    688 	eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
    689 	wps_free_pins(&reg->pins);
    690 	wps_free_nfc_pw_tokens(&reg->nfc_pw_tokens, 0);
    691 	wps_free_pbc_sessions(reg->pbc_sessions);
    692 	wpabuf_free(reg->extra_cred);
    693 	wps_free_devices(reg->devices);
    694 	os_free(reg);
    695 }
    696 
    697 
    698 static void wps_registrar_invalidate_unused(struct wps_registrar *reg)
    699 {
    700 	struct wps_uuid_pin *pin;
    701 
    702 	dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
    703 		if (pin->wildcard_uuid == 1 && !(pin->flags & PIN_LOCKED)) {
    704 			wpa_printf(MSG_DEBUG, "WPS: Invalidate previously "
    705 				   "configured wildcard PIN");
    706 			wps_registrar_remove_pin(reg, pin);
    707 			break;
    708 		}
    709 	}
    710 }
    711 
    712 
    713 /**
    714  * wps_registrar_add_pin - Configure a new PIN for Registrar
    715  * @reg: Registrar data from wps_registrar_init()
    716  * @addr: Enrollee MAC address or %NULL if not known
    717  * @uuid: UUID-E or %NULL for wildcard (any UUID)
    718  * @pin: PIN (Device Password)
    719  * @pin_len: Length of pin in octets
    720  * @timeout: Time (in seconds) when the PIN will be invalidated; 0 = no timeout
    721  * Returns: 0 on success, -1 on failure
    722  */
    723 int wps_registrar_add_pin(struct wps_registrar *reg, const u8 *addr,
    724 			  const u8 *uuid, const u8 *pin, size_t pin_len,
    725 			  int timeout)
    726 {
    727 	struct wps_uuid_pin *p;
    728 
    729 	p = os_zalloc(sizeof(*p));
    730 	if (p == NULL)
    731 		return -1;
    732 	if (addr)
    733 		os_memcpy(p->enrollee_addr, addr, ETH_ALEN);
    734 	if (uuid == NULL)
    735 		p->wildcard_uuid = 1;
    736 	else
    737 		os_memcpy(p->uuid, uuid, WPS_UUID_LEN);
    738 	p->pin = os_malloc(pin_len);
    739 	if (p->pin == NULL) {
    740 		os_free(p);
    741 		return -1;
    742 	}
    743 	os_memcpy(p->pin, pin, pin_len);
    744 	p->pin_len = pin_len;
    745 
    746 	if (timeout) {
    747 		p->flags |= PIN_EXPIRES;
    748 		os_get_reltime(&p->expiration);
    749 		p->expiration.sec += timeout;
    750 	}
    751 
    752 	if (p->wildcard_uuid)
    753 		wps_registrar_invalidate_unused(reg);
    754 
    755 	dl_list_add(&reg->pins, &p->list);
    756 
    757 	wpa_printf(MSG_DEBUG, "WPS: A new PIN configured (timeout=%d)",
    758 		   timeout);
    759 	wpa_hexdump(MSG_DEBUG, "WPS: UUID", uuid, WPS_UUID_LEN);
    760 	wpa_hexdump_ascii_key(MSG_DEBUG, "WPS: PIN", pin, pin_len);
    761 	reg->selected_registrar = 1;
    762 	reg->pbc = 0;
    763 	if (addr)
    764 		wps_registrar_add_authorized_mac(reg, addr);
    765 	else
    766 		wps_registrar_add_authorized_mac(
    767 			reg, (u8 *) "\xff\xff\xff\xff\xff\xff");
    768 	wps_registrar_selected_registrar_changed(reg, 0);
    769 	eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
    770 	eloop_register_timeout(WPS_PBC_WALK_TIME, 0,
    771 			       wps_registrar_set_selected_timeout,
    772 			       reg, NULL);
    773 
    774 	return 0;
    775 }
    776 
    777 
    778 static void wps_registrar_remove_pin(struct wps_registrar *reg,
    779 				     struct wps_uuid_pin *pin)
    780 {
    781 	u8 *addr;
    782 	u8 bcast[ETH_ALEN] = { 0xff, 0xff, 0xff, 0xff, 0xff, 0xff };
    783 
    784 	if (is_zero_ether_addr(pin->enrollee_addr))
    785 		addr = bcast;
    786 	else
    787 		addr = pin->enrollee_addr;
    788 	wps_registrar_remove_authorized_mac(reg, addr);
    789 	wps_remove_pin(pin);
    790 	wps_registrar_selected_registrar_changed(reg, 0);
    791 }
    792 
    793 
    794 static void wps_registrar_expire_pins(struct wps_registrar *reg)
    795 {
    796 	struct wps_uuid_pin *pin, *prev;
    797 	struct os_reltime now;
    798 
    799 	os_get_reltime(&now);
    800 	dl_list_for_each_safe(pin, prev, &reg->pins, struct wps_uuid_pin, list)
    801 	{
    802 		if ((pin->flags & PIN_EXPIRES) &&
    803 		    os_reltime_before(&pin->expiration, &now)) {
    804 			wpa_hexdump(MSG_DEBUG, "WPS: Expired PIN for UUID",
    805 				    pin->uuid, WPS_UUID_LEN);
    806 			wps_registrar_remove_pin(reg, pin);
    807 		}
    808 	}
    809 }
    810 
    811 
    812 /**
    813  * wps_registrar_invalidate_wildcard_pin - Invalidate a wildcard PIN
    814  * @reg: Registrar data from wps_registrar_init()
    815  * @dev_pw: PIN to search for or %NULL to match any
    816  * @dev_pw_len: Length of dev_pw in octets
    817  * Returns: 0 on success, -1 if not wildcard PIN is enabled
    818  */
    819 static int wps_registrar_invalidate_wildcard_pin(struct wps_registrar *reg,
    820 						 const u8 *dev_pw,
    821 						 size_t dev_pw_len)
    822 {
    823 	struct wps_uuid_pin *pin, *prev;
    824 
    825 	dl_list_for_each_safe(pin, prev, &reg->pins, struct wps_uuid_pin, list)
    826 	{
    827 		if (dev_pw && pin->pin &&
    828 		    (dev_pw_len != pin->pin_len ||
    829 		     os_memcmp_const(dev_pw, pin->pin, dev_pw_len) != 0))
    830 			continue; /* different PIN */
    831 		if (pin->wildcard_uuid) {
    832 			wpa_hexdump(MSG_DEBUG, "WPS: Invalidated PIN for UUID",
    833 				    pin->uuid, WPS_UUID_LEN);
    834 			wps_registrar_remove_pin(reg, pin);
    835 			return 0;
    836 		}
    837 	}
    838 
    839 	return -1;
    840 }
    841 
    842 
    843 /**
    844  * wps_registrar_invalidate_pin - Invalidate a PIN for a specific UUID-E
    845  * @reg: Registrar data from wps_registrar_init()
    846  * @uuid: UUID-E
    847  * Returns: 0 on success, -1 on failure (e.g., PIN not found)
    848  */
    849 int wps_registrar_invalidate_pin(struct wps_registrar *reg, const u8 *uuid)
    850 {
    851 	struct wps_uuid_pin *pin, *prev;
    852 
    853 	dl_list_for_each_safe(pin, prev, &reg->pins, struct wps_uuid_pin, list)
    854 	{
    855 		if (os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
    856 			wpa_hexdump(MSG_DEBUG, "WPS: Invalidated PIN for UUID",
    857 				    pin->uuid, WPS_UUID_LEN);
    858 			wps_registrar_remove_pin(reg, pin);
    859 			return 0;
    860 		}
    861 	}
    862 
    863 	return -1;
    864 }
    865 
    866 
    867 static const u8 * wps_registrar_get_pin(struct wps_registrar *reg,
    868 					const u8 *uuid, size_t *pin_len)
    869 {
    870 	struct wps_uuid_pin *pin, *found = NULL;
    871 
    872 	wps_registrar_expire_pins(reg);
    873 
    874 	dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
    875 		if (!pin->wildcard_uuid &&
    876 		    os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
    877 			found = pin;
    878 			break;
    879 		}
    880 	}
    881 
    882 	if (!found) {
    883 		/* Check for wildcard UUIDs since none of the UUID-specific
    884 		 * PINs matched */
    885 		dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
    886 			if (pin->wildcard_uuid == 1 ||
    887 			    pin->wildcard_uuid == 2) {
    888 				wpa_printf(MSG_DEBUG, "WPS: Found a wildcard "
    889 					   "PIN. Assigned it for this UUID-E");
    890 				pin->wildcard_uuid++;
    891 				os_memcpy(pin->uuid, uuid, WPS_UUID_LEN);
    892 				found = pin;
    893 				break;
    894 			}
    895 		}
    896 	}
    897 
    898 	if (!found)
    899 		return NULL;
    900 
    901 	/*
    902 	 * Lock the PIN to avoid attacks based on concurrent re-use of the PIN
    903 	 * that could otherwise avoid PIN invalidations.
    904 	 */
    905 	if (found->flags & PIN_LOCKED) {
    906 		wpa_printf(MSG_DEBUG, "WPS: Selected PIN locked - do not "
    907 			   "allow concurrent re-use");
    908 		return NULL;
    909 	}
    910 	*pin_len = found->pin_len;
    911 	found->flags |= PIN_LOCKED;
    912 	return found->pin;
    913 }
    914 
    915 
    916 /**
    917  * wps_registrar_unlock_pin - Unlock a PIN for a specific UUID-E
    918  * @reg: Registrar data from wps_registrar_init()
    919  * @uuid: UUID-E
    920  * Returns: 0 on success, -1 on failure
    921  *
    922  * PINs are locked to enforce only one concurrent use. This function unlocks a
    923  * PIN to allow it to be used again. If the specified PIN was configured using
    924  * a wildcard UUID, it will be removed instead of allowing multiple uses.
    925  */
    926 int wps_registrar_unlock_pin(struct wps_registrar *reg, const u8 *uuid)
    927 {
    928 	struct wps_uuid_pin *pin;
    929 
    930 	dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
    931 		if (os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
    932 			if (pin->wildcard_uuid == 3) {
    933 				wpa_printf(MSG_DEBUG, "WPS: Invalidating used "
    934 					   "wildcard PIN");
    935 				return wps_registrar_invalidate_pin(reg, uuid);
    936 			}
    937 			pin->flags &= ~PIN_LOCKED;
    938 			return 0;
    939 		}
    940 	}
    941 
    942 	return -1;
    943 }
    944 
    945 
    946 static void wps_registrar_stop_pbc(struct wps_registrar *reg)
    947 {
    948 	reg->selected_registrar = 0;
    949 	reg->pbc = 0;
    950 	os_memset(reg->p2p_dev_addr, 0, ETH_ALEN);
    951 	wps_registrar_remove_authorized_mac(reg,
    952 					    (u8 *) "\xff\xff\xff\xff\xff\xff");
    953 	wps_registrar_selected_registrar_changed(reg, 0);
    954 }
    955 
    956 
    957 static void wps_registrar_pbc_timeout(void *eloop_ctx, void *timeout_ctx)
    958 {
    959 	struct wps_registrar *reg = eloop_ctx;
    960 
    961 	wpa_printf(MSG_DEBUG, "WPS: PBC timed out - disable PBC mode");
    962 	wps_pbc_timeout_event(reg->wps);
    963 	wps_registrar_stop_pbc(reg);
    964 }
    965 
    966 
    967 /**
    968  * wps_registrar_button_pushed - Notify Registrar that AP button was pushed
    969  * @reg: Registrar data from wps_registrar_init()
    970  * @p2p_dev_addr: Limit allowed PBC devices to the specified P2P device, %NULL
    971  *	indicates no such filtering
    972  * Returns: 0 on success, -1 on failure, -2 on session overlap
    973  *
    974  * This function is called on an AP when a push button is pushed to activate
    975  * PBC mode. The PBC mode will be stopped after walk time (2 minutes) timeout
    976  * or when a PBC registration is completed. If more than one Enrollee in active
    977  * PBC mode has been detected during the monitor time (previous 2 minutes), the
    978  * PBC mode is not activated and -2 is returned to indicate session overlap.
    979  * This is skipped if a specific Enrollee is selected.
    980  */
    981 int wps_registrar_button_pushed(struct wps_registrar *reg,
    982 				const u8 *p2p_dev_addr)
    983 {
    984 	if (p2p_dev_addr == NULL &&
    985 	    wps_registrar_pbc_overlap(reg, NULL, NULL)) {
    986 		wpa_printf(MSG_DEBUG, "WPS: PBC overlap - do not start PBC "
    987 			   "mode");
    988 		wps_pbc_overlap_event(reg->wps);
    989 		return -2;
    990 	}
    991 	wpa_printf(MSG_DEBUG, "WPS: Button pushed - PBC mode started");
    992 	reg->force_pbc_overlap = 0;
    993 	reg->selected_registrar = 1;
    994 	reg->pbc = 1;
    995 	if (p2p_dev_addr)
    996 		os_memcpy(reg->p2p_dev_addr, p2p_dev_addr, ETH_ALEN);
    997 	else
    998 		os_memset(reg->p2p_dev_addr, 0, ETH_ALEN);
    999 	wps_registrar_add_authorized_mac(reg,
   1000 					 (u8 *) "\xff\xff\xff\xff\xff\xff");
   1001 	wps_registrar_selected_registrar_changed(reg, 0);
   1002 
   1003 	wps_pbc_active_event(reg->wps);
   1004 	eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
   1005 	eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
   1006 	eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wps_registrar_pbc_timeout,
   1007 			       reg, NULL);
   1008 	return 0;
   1009 }
   1010 
   1011 
   1012 static void wps_registrar_pbc_completed(struct wps_registrar *reg)
   1013 {
   1014 	wpa_printf(MSG_DEBUG, "WPS: PBC completed - stopping PBC mode");
   1015 	eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
   1016 	wps_registrar_stop_pbc(reg);
   1017 	wps_pbc_disable_event(reg->wps);
   1018 }
   1019 
   1020 
   1021 static void wps_registrar_pin_completed(struct wps_registrar *reg)
   1022 {
   1023 	wpa_printf(MSG_DEBUG, "WPS: PIN completed using internal Registrar");
   1024 	eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
   1025 	reg->selected_registrar = 0;
   1026 	wps_registrar_selected_registrar_changed(reg, 0);
   1027 }
   1028 
   1029 
   1030 void wps_registrar_complete(struct wps_registrar *registrar, const u8 *uuid_e,
   1031 			    const u8 *dev_pw, size_t dev_pw_len)
   1032 {
   1033 	if (registrar->pbc) {
   1034 		wps_registrar_remove_pbc_session(registrar,
   1035 						 uuid_e, NULL);
   1036 		wps_registrar_pbc_completed(registrar);
   1037 #ifdef WPS_WORKAROUNDS
   1038 		os_get_reltime(&registrar->pbc_ignore_start);
   1039 #endif /* WPS_WORKAROUNDS */
   1040 		os_memcpy(registrar->pbc_ignore_uuid, uuid_e, WPS_UUID_LEN);
   1041 	} else {
   1042 		wps_registrar_pin_completed(registrar);
   1043 	}
   1044 
   1045 	if (dev_pw &&
   1046 	    wps_registrar_invalidate_wildcard_pin(registrar, dev_pw,
   1047 						  dev_pw_len) == 0) {
   1048 		wpa_hexdump_key(MSG_DEBUG, "WPS: Invalidated wildcard PIN",
   1049 				dev_pw, dev_pw_len);
   1050 	}
   1051 }
   1052 
   1053 
   1054 int wps_registrar_wps_cancel(struct wps_registrar *reg)
   1055 {
   1056 	if (reg->pbc) {
   1057 		wpa_printf(MSG_DEBUG, "WPS: PBC is set - cancelling it");
   1058 		wps_registrar_pbc_timeout(reg, NULL);
   1059 		eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
   1060 		return 1;
   1061 	} else if (reg->selected_registrar) {
   1062 		/* PIN Method */
   1063 		wpa_printf(MSG_DEBUG, "WPS: PIN is set - cancelling it");
   1064 		wps_registrar_pin_completed(reg);
   1065 		wps_registrar_invalidate_wildcard_pin(reg, NULL, 0);
   1066 		return 1;
   1067 	}
   1068 	return 0;
   1069 }
   1070 
   1071 
   1072 /**
   1073  * wps_registrar_probe_req_rx - Notify Registrar of Probe Request
   1074  * @reg: Registrar data from wps_registrar_init()
   1075  * @addr: MAC address of the Probe Request sender
   1076  * @wps_data: WPS IE contents
   1077  *
   1078  * This function is called on an AP when a Probe Request with WPS IE is
   1079  * received. This is used to track PBC mode use and to detect possible overlap
   1080  * situation with other WPS APs.
   1081  */
   1082 void wps_registrar_probe_req_rx(struct wps_registrar *reg, const u8 *addr,
   1083 				const struct wpabuf *wps_data,
   1084 				int p2p_wildcard)
   1085 {
   1086 	struct wps_parse_attr attr;
   1087 	int skip_add = 0;
   1088 
   1089 	wpa_hexdump_buf(MSG_MSGDUMP,
   1090 			"WPS: Probe Request with WPS data received",
   1091 			wps_data);
   1092 
   1093 	if (wps_parse_msg(wps_data, &attr) < 0)
   1094 		return;
   1095 
   1096 	if (attr.config_methods == NULL) {
   1097 		wpa_printf(MSG_DEBUG, "WPS: No Config Methods attribute in "
   1098 			   "Probe Request");
   1099 		return;
   1100 	}
   1101 
   1102 	if (attr.dev_password_id == NULL) {
   1103 		wpa_printf(MSG_DEBUG, "WPS: No Device Password Id attribute "
   1104 			   "in Probe Request");
   1105 		return;
   1106 	}
   1107 
   1108 	if (reg->enrollee_seen_cb && attr.uuid_e &&
   1109 	    attr.primary_dev_type && attr.request_type && !p2p_wildcard) {
   1110 		char *dev_name = NULL;
   1111 		if (attr.dev_name) {
   1112 			dev_name = os_zalloc(attr.dev_name_len + 1);
   1113 			if (dev_name) {
   1114 				os_memcpy(dev_name, attr.dev_name,
   1115 					  attr.dev_name_len);
   1116 			}
   1117 		}
   1118 		reg->enrollee_seen_cb(reg->cb_ctx, addr, attr.uuid_e,
   1119 				      attr.primary_dev_type,
   1120 				      WPA_GET_BE16(attr.config_methods),
   1121 				      WPA_GET_BE16(attr.dev_password_id),
   1122 				      *attr.request_type, dev_name);
   1123 		os_free(dev_name);
   1124 	}
   1125 
   1126 	if (WPA_GET_BE16(attr.dev_password_id) != DEV_PW_PUSHBUTTON)
   1127 		return; /* Not PBC */
   1128 
   1129 	wpa_printf(MSG_DEBUG, "WPS: Probe Request for PBC received from "
   1130 		   MACSTR, MAC2STR(addr));
   1131 	if (attr.uuid_e == NULL) {
   1132 		wpa_printf(MSG_DEBUG, "WPS: Invalid Probe Request WPS IE: No "
   1133 			   "UUID-E included");
   1134 		return;
   1135 	}
   1136 	wpa_hexdump(MSG_DEBUG, "WPS: UUID-E from Probe Request", attr.uuid_e,
   1137 		    WPS_UUID_LEN);
   1138 
   1139 #ifdef WPS_WORKAROUNDS
   1140 	if (reg->pbc_ignore_start.sec &&
   1141 	    os_memcmp(attr.uuid_e, reg->pbc_ignore_uuid, WPS_UUID_LEN) == 0) {
   1142 		struct os_reltime now, dur;
   1143 		os_get_reltime(&now);
   1144 		os_reltime_sub(&now, &reg->pbc_ignore_start, &dur);
   1145 		if (dur.sec >= 0 && dur.sec < 5) {
   1146 			wpa_printf(MSG_DEBUG, "WPS: Ignore PBC activation "
   1147 				   "based on Probe Request from the Enrollee "
   1148 				   "that just completed PBC provisioning");
   1149 			skip_add = 1;
   1150 		} else
   1151 			reg->pbc_ignore_start.sec = 0;
   1152 	}
   1153 #endif /* WPS_WORKAROUNDS */
   1154 
   1155 	if (!skip_add)
   1156 		wps_registrar_add_pbc_session(reg, addr, attr.uuid_e);
   1157 	if (wps_registrar_pbc_overlap(reg, addr, attr.uuid_e)) {
   1158 		wpa_printf(MSG_DEBUG, "WPS: PBC session overlap detected");
   1159 		reg->force_pbc_overlap = 1;
   1160 		wps_pbc_overlap_event(reg->wps);
   1161 	}
   1162 }
   1163 
   1164 
   1165 int wps_cb_new_psk(struct wps_registrar *reg, const u8 *mac_addr,
   1166 		   const u8 *p2p_dev_addr, const u8 *psk, size_t psk_len)
   1167 {
   1168 	if (reg->new_psk_cb == NULL)
   1169 		return 0;
   1170 
   1171 	return reg->new_psk_cb(reg->cb_ctx, mac_addr, p2p_dev_addr, psk,
   1172 			       psk_len);
   1173 }
   1174 
   1175 
   1176 static void wps_cb_pin_needed(struct wps_registrar *reg, const u8 *uuid_e,
   1177 			      const struct wps_device_data *dev)
   1178 {
   1179 	if (reg->pin_needed_cb == NULL)
   1180 		return;
   1181 
   1182 	reg->pin_needed_cb(reg->cb_ctx, uuid_e, dev);
   1183 }
   1184 
   1185 
   1186 static void wps_cb_reg_success(struct wps_registrar *reg, const u8 *mac_addr,
   1187 			       const u8 *uuid_e, const u8 *dev_pw,
   1188 			       size_t dev_pw_len)
   1189 {
   1190 	if (reg->reg_success_cb == NULL)
   1191 		return;
   1192 
   1193 	reg->reg_success_cb(reg->cb_ctx, mac_addr, uuid_e, dev_pw, dev_pw_len);
   1194 }
   1195 
   1196 
   1197 static int wps_cb_set_ie(struct wps_registrar *reg, struct wpabuf *beacon_ie,
   1198 			 struct wpabuf *probe_resp_ie)
   1199 {
   1200 	return reg->set_ie_cb(reg->cb_ctx, beacon_ie, probe_resp_ie);
   1201 }
   1202 
   1203 
   1204 static void wps_cb_set_sel_reg(struct wps_registrar *reg)
   1205 {
   1206 	u16 methods = 0;
   1207 	if (reg->set_sel_reg_cb == NULL)
   1208 		return;
   1209 
   1210 	if (reg->selected_registrar) {
   1211 		methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
   1212 		methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
   1213 			     WPS_CONFIG_PHY_PUSHBUTTON);
   1214 		if (reg->pbc)
   1215 			wps_set_pushbutton(&methods, reg->wps->config_methods);
   1216 	}
   1217 
   1218 	wpa_printf(MSG_DEBUG, "WPS: wps_cb_set_sel_reg: sel_reg=%d "
   1219 		   "config_methods=0x%x pbc=%d methods=0x%x",
   1220 		   reg->selected_registrar, reg->wps->config_methods,
   1221 		   reg->pbc, methods);
   1222 
   1223 	reg->set_sel_reg_cb(reg->cb_ctx, reg->selected_registrar,
   1224 			    reg->pbc ? DEV_PW_PUSHBUTTON : DEV_PW_DEFAULT,
   1225 			    methods);
   1226 }
   1227 
   1228 
   1229 static int wps_set_ie(struct wps_registrar *reg)
   1230 {
   1231 	struct wpabuf *beacon;
   1232 	struct wpabuf *probe;
   1233 	const u8 *auth_macs;
   1234 	size_t count;
   1235 	size_t vendor_len = 0;
   1236 	int i;
   1237 
   1238 	if (reg->set_ie_cb == NULL)
   1239 		return 0;
   1240 
   1241 	for (i = 0; i < MAX_WPS_VENDOR_EXTENSIONS; i++) {
   1242 		if (reg->wps->dev.vendor_ext[i]) {
   1243 			vendor_len += 2 + 2;
   1244 			vendor_len += wpabuf_len(reg->wps->dev.vendor_ext[i]);
   1245 		}
   1246 	}
   1247 
   1248 	beacon = wpabuf_alloc(400 + vendor_len);
   1249 	if (beacon == NULL)
   1250 		return -1;
   1251 	probe = wpabuf_alloc(500 + vendor_len);
   1252 	if (probe == NULL) {
   1253 		wpabuf_free(beacon);
   1254 		return -1;
   1255 	}
   1256 
   1257 	auth_macs = wps_authorized_macs(reg, &count);
   1258 
   1259 	wpa_printf(MSG_DEBUG, "WPS: Build Beacon IEs");
   1260 
   1261 	if (wps_build_version(beacon) ||
   1262 	    wps_build_wps_state(reg->wps, beacon) ||
   1263 	    wps_build_ap_setup_locked(reg->wps, beacon) ||
   1264 	    wps_build_selected_registrar(reg, beacon) ||
   1265 	    wps_build_sel_reg_dev_password_id(reg, beacon) ||
   1266 	    wps_build_sel_reg_config_methods(reg, beacon) ||
   1267 	    wps_build_sel_pbc_reg_uuid_e(reg, beacon) ||
   1268 	    (reg->dualband && wps_build_rf_bands(&reg->wps->dev, beacon, 0)) ||
   1269 	    wps_build_wfa_ext(beacon, 0, auth_macs, count) ||
   1270 	    wps_build_vendor_ext(&reg->wps->dev, beacon)) {
   1271 		wpabuf_free(beacon);
   1272 		wpabuf_free(probe);
   1273 		return -1;
   1274 	}
   1275 
   1276 #ifdef CONFIG_P2P
   1277 	if (wps_build_dev_name(&reg->wps->dev, beacon) ||
   1278 	    wps_build_primary_dev_type(&reg->wps->dev, beacon)) {
   1279 		wpabuf_free(beacon);
   1280 		wpabuf_free(probe);
   1281 		return -1;
   1282 	}
   1283 #endif /* CONFIG_P2P */
   1284 
   1285 	wpa_printf(MSG_DEBUG, "WPS: Build Probe Response IEs");
   1286 
   1287 	if (wps_build_version(probe) ||
   1288 	    wps_build_wps_state(reg->wps, probe) ||
   1289 	    wps_build_ap_setup_locked(reg->wps, probe) ||
   1290 	    wps_build_selected_registrar(reg, probe) ||
   1291 	    wps_build_sel_reg_dev_password_id(reg, probe) ||
   1292 	    wps_build_sel_reg_config_methods(reg, probe) ||
   1293 	    wps_build_resp_type(probe, reg->wps->ap ? WPS_RESP_AP :
   1294 				WPS_RESP_REGISTRAR) ||
   1295 	    wps_build_uuid_e(probe, reg->wps->uuid) ||
   1296 	    wps_build_device_attrs(&reg->wps->dev, probe) ||
   1297 	    wps_build_probe_config_methods(reg, probe) ||
   1298 	    (reg->dualband && wps_build_rf_bands(&reg->wps->dev, probe, 0)) ||
   1299 	    wps_build_wfa_ext(probe, 0, auth_macs, count) ||
   1300 	    wps_build_vendor_ext(&reg->wps->dev, probe)) {
   1301 		wpabuf_free(beacon);
   1302 		wpabuf_free(probe);
   1303 		return -1;
   1304 	}
   1305 
   1306 	beacon = wps_ie_encapsulate(beacon);
   1307 	probe = wps_ie_encapsulate(probe);
   1308 
   1309 	if (!beacon || !probe) {
   1310 		wpabuf_free(beacon);
   1311 		wpabuf_free(probe);
   1312 		return -1;
   1313 	}
   1314 
   1315 	if (reg->static_wep_only) {
   1316 		/*
   1317 		 * Windows XP and Vista clients can get confused about
   1318 		 * EAP-Identity/Request when they probe the network with
   1319 		 * EAPOL-Start. In such a case, they may assume the network is
   1320 		 * using IEEE 802.1X and prompt user for a certificate while
   1321 		 * the correct (non-WPS) behavior would be to ask for the
   1322 		 * static WEP key. As a workaround, use Microsoft Provisioning
   1323 		 * IE to advertise that legacy 802.1X is not supported.
   1324 		 */
   1325 		const u8 ms_wps[7] = {
   1326 			WLAN_EID_VENDOR_SPECIFIC, 5,
   1327 			/* Microsoft Provisioning IE (00:50:f2:5) */
   1328 			0x00, 0x50, 0xf2, 5,
   1329 			0x00 /* no legacy 802.1X or MS WPS */
   1330 		};
   1331 		wpa_printf(MSG_DEBUG, "WPS: Add Microsoft Provisioning IE "
   1332 			   "into Beacon/Probe Response frames");
   1333 		wpabuf_put_data(beacon, ms_wps, sizeof(ms_wps));
   1334 		wpabuf_put_data(probe, ms_wps, sizeof(ms_wps));
   1335 	}
   1336 
   1337 	return wps_cb_set_ie(reg, beacon, probe);
   1338 }
   1339 
   1340 
   1341 static int wps_get_dev_password(struct wps_data *wps)
   1342 {
   1343 	const u8 *pin;
   1344 	size_t pin_len = 0;
   1345 
   1346 	bin_clear_free(wps->dev_password, wps->dev_password_len);
   1347 	wps->dev_password = NULL;
   1348 
   1349 	if (wps->pbc) {
   1350 		wpa_printf(MSG_DEBUG, "WPS: Use default PIN for PBC");
   1351 		pin = (const u8 *) "00000000";
   1352 		pin_len = 8;
   1353 #ifdef CONFIG_WPS_NFC
   1354 	} else if (wps->nfc_pw_token) {
   1355 		if (wps->nfc_pw_token->pw_id == DEV_PW_NFC_CONNECTION_HANDOVER)
   1356 		{
   1357 			wpa_printf(MSG_DEBUG, "WPS: Using NFC connection "
   1358 				   "handover and abbreviated WPS handshake "
   1359 				   "without Device Password");
   1360 			return 0;
   1361 		}
   1362 		wpa_printf(MSG_DEBUG, "WPS: Use OOB Device Password from NFC "
   1363 			   "Password Token");
   1364 		pin = wps->nfc_pw_token->dev_pw;
   1365 		pin_len = wps->nfc_pw_token->dev_pw_len;
   1366 	} else if (wps->dev_pw_id >= 0x10 &&
   1367 		   wps->wps->ap_nfc_dev_pw_id == wps->dev_pw_id &&
   1368 		   wps->wps->ap_nfc_dev_pw) {
   1369 		wpa_printf(MSG_DEBUG, "WPS: Use OOB Device Password from own NFC Password Token");
   1370 		pin = wpabuf_head(wps->wps->ap_nfc_dev_pw);
   1371 		pin_len = wpabuf_len(wps->wps->ap_nfc_dev_pw);
   1372 #endif /* CONFIG_WPS_NFC */
   1373 	} else {
   1374 		pin = wps_registrar_get_pin(wps->wps->registrar, wps->uuid_e,
   1375 					    &pin_len);
   1376 		if (pin && wps->dev_pw_id >= 0x10) {
   1377 			wpa_printf(MSG_DEBUG, "WPS: No match for OOB Device "
   1378 				   "Password ID, but PIN found");
   1379 			/*
   1380 			 * See whether Enrollee is willing to use PIN instead.
   1381 			 */
   1382 			wps->dev_pw_id = DEV_PW_DEFAULT;
   1383 		}
   1384 	}
   1385 	if (pin == NULL) {
   1386 		wpa_printf(MSG_DEBUG, "WPS: No Device Password available for "
   1387 			   "the Enrollee (context %p registrar %p)",
   1388 			   wps->wps, wps->wps->registrar);
   1389 		wps_cb_pin_needed(wps->wps->registrar, wps->uuid_e,
   1390 				  &wps->peer_dev);
   1391 		return -1;
   1392 	}
   1393 
   1394 	wps->dev_password = os_malloc(pin_len);
   1395 	if (wps->dev_password == NULL)
   1396 		return -1;
   1397 	os_memcpy(wps->dev_password, pin, pin_len);
   1398 	wps->dev_password_len = pin_len;
   1399 
   1400 	return 0;
   1401 }
   1402 
   1403 
   1404 static int wps_build_uuid_r(struct wps_data *wps, struct wpabuf *msg)
   1405 {
   1406 	wpa_printf(MSG_DEBUG, "WPS:  * UUID-R");
   1407 	wpabuf_put_be16(msg, ATTR_UUID_R);
   1408 	wpabuf_put_be16(msg, WPS_UUID_LEN);
   1409 	wpabuf_put_data(msg, wps->uuid_r, WPS_UUID_LEN);
   1410 	return 0;
   1411 }
   1412 
   1413 
   1414 static int wps_build_r_hash(struct wps_data *wps, struct wpabuf *msg)
   1415 {
   1416 	u8 *hash;
   1417 	const u8 *addr[4];
   1418 	size_t len[4];
   1419 
   1420 	if (random_get_bytes(wps->snonce, 2 * WPS_SECRET_NONCE_LEN) < 0)
   1421 		return -1;
   1422 	wpa_hexdump(MSG_DEBUG, "WPS: R-S1", wps->snonce, WPS_SECRET_NONCE_LEN);
   1423 	wpa_hexdump(MSG_DEBUG, "WPS: R-S2",
   1424 		    wps->snonce + WPS_SECRET_NONCE_LEN, WPS_SECRET_NONCE_LEN);
   1425 
   1426 	if (wps->dh_pubkey_e == NULL || wps->dh_pubkey_r == NULL) {
   1427 		wpa_printf(MSG_DEBUG, "WPS: DH public keys not available for "
   1428 			   "R-Hash derivation");
   1429 		return -1;
   1430 	}
   1431 
   1432 	wpa_printf(MSG_DEBUG, "WPS:  * R-Hash1");
   1433 	wpabuf_put_be16(msg, ATTR_R_HASH1);
   1434 	wpabuf_put_be16(msg, SHA256_MAC_LEN);
   1435 	hash = wpabuf_put(msg, SHA256_MAC_LEN);
   1436 	/* R-Hash1 = HMAC_AuthKey(R-S1 || PSK1 || PK_E || PK_R) */
   1437 	addr[0] = wps->snonce;
   1438 	len[0] = WPS_SECRET_NONCE_LEN;
   1439 	addr[1] = wps->psk1;
   1440 	len[1] = WPS_PSK_LEN;
   1441 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
   1442 	len[2] = wpabuf_len(wps->dh_pubkey_e);
   1443 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
   1444 	len[3] = wpabuf_len(wps->dh_pubkey_r);
   1445 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
   1446 	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash1", hash, SHA256_MAC_LEN);
   1447 
   1448 	wpa_printf(MSG_DEBUG, "WPS:  * R-Hash2");
   1449 	wpabuf_put_be16(msg, ATTR_R_HASH2);
   1450 	wpabuf_put_be16(msg, SHA256_MAC_LEN);
   1451 	hash = wpabuf_put(msg, SHA256_MAC_LEN);
   1452 	/* R-Hash2 = HMAC_AuthKey(R-S2 || PSK2 || PK_E || PK_R) */
   1453 	addr[0] = wps->snonce + WPS_SECRET_NONCE_LEN;
   1454 	addr[1] = wps->psk2;
   1455 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
   1456 	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash2", hash, SHA256_MAC_LEN);
   1457 
   1458 	return 0;
   1459 }
   1460 
   1461 
   1462 static int wps_build_r_snonce1(struct wps_data *wps, struct wpabuf *msg)
   1463 {
   1464 	wpa_printf(MSG_DEBUG, "WPS:  * R-SNonce1");
   1465 	wpabuf_put_be16(msg, ATTR_R_SNONCE1);
   1466 	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
   1467 	wpabuf_put_data(msg, wps->snonce, WPS_SECRET_NONCE_LEN);
   1468 	return 0;
   1469 }
   1470 
   1471 
   1472 static int wps_build_r_snonce2(struct wps_data *wps, struct wpabuf *msg)
   1473 {
   1474 	wpa_printf(MSG_DEBUG, "WPS:  * R-SNonce2");
   1475 	wpabuf_put_be16(msg, ATTR_R_SNONCE2);
   1476 	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
   1477 	wpabuf_put_data(msg, wps->snonce + WPS_SECRET_NONCE_LEN,
   1478 			WPS_SECRET_NONCE_LEN);
   1479 	return 0;
   1480 }
   1481 
   1482 
   1483 static int wps_build_cred_network_idx(struct wpabuf *msg,
   1484 				      const struct wps_credential *cred)
   1485 {
   1486 	wpa_printf(MSG_DEBUG, "WPS:  * Network Index (1)");
   1487 	wpabuf_put_be16(msg, ATTR_NETWORK_INDEX);
   1488 	wpabuf_put_be16(msg, 1);
   1489 	wpabuf_put_u8(msg, 1);
   1490 	return 0;
   1491 }
   1492 
   1493 
   1494 static int wps_build_cred_ssid(struct wpabuf *msg,
   1495 			       const struct wps_credential *cred)
   1496 {
   1497 	wpa_printf(MSG_DEBUG, "WPS:  * SSID");
   1498 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID for Credential",
   1499 			  cred->ssid, cred->ssid_len);
   1500 	wpabuf_put_be16(msg, ATTR_SSID);
   1501 	wpabuf_put_be16(msg, cred->ssid_len);
   1502 	wpabuf_put_data(msg, cred->ssid, cred->ssid_len);
   1503 	return 0;
   1504 }
   1505 
   1506 
   1507 static int wps_build_cred_auth_type(struct wpabuf *msg,
   1508 				    const struct wps_credential *cred)
   1509 {
   1510 	wpa_printf(MSG_DEBUG, "WPS:  * Authentication Type (0x%x)",
   1511 		   cred->auth_type);
   1512 	wpabuf_put_be16(msg, ATTR_AUTH_TYPE);
   1513 	wpabuf_put_be16(msg, 2);
   1514 	wpabuf_put_be16(msg, cred->auth_type);
   1515 	return 0;
   1516 }
   1517 
   1518 
   1519 static int wps_build_cred_encr_type(struct wpabuf *msg,
   1520 				    const struct wps_credential *cred)
   1521 {
   1522 	wpa_printf(MSG_DEBUG, "WPS:  * Encryption Type (0x%x)",
   1523 		   cred->encr_type);
   1524 	wpabuf_put_be16(msg, ATTR_ENCR_TYPE);
   1525 	wpabuf_put_be16(msg, 2);
   1526 	wpabuf_put_be16(msg, cred->encr_type);
   1527 	return 0;
   1528 }
   1529 
   1530 
   1531 static int wps_build_cred_network_key(struct wpabuf *msg,
   1532 				      const struct wps_credential *cred)
   1533 {
   1534 	wpa_printf(MSG_DEBUG, "WPS:  * Network Key (len=%d)",
   1535 		   (int) cred->key_len);
   1536 	wpa_hexdump_key(MSG_DEBUG, "WPS: Network Key",
   1537 			cred->key, cred->key_len);
   1538 	wpabuf_put_be16(msg, ATTR_NETWORK_KEY);
   1539 	wpabuf_put_be16(msg, cred->key_len);
   1540 	wpabuf_put_data(msg, cred->key, cred->key_len);
   1541 	return 0;
   1542 }
   1543 
   1544 
   1545 static int wps_build_credential(struct wpabuf *msg,
   1546 				const struct wps_credential *cred)
   1547 {
   1548 	if (wps_build_cred_network_idx(msg, cred) ||
   1549 	    wps_build_cred_ssid(msg, cred) ||
   1550 	    wps_build_cred_auth_type(msg, cred) ||
   1551 	    wps_build_cred_encr_type(msg, cred) ||
   1552 	    wps_build_cred_network_key(msg, cred) ||
   1553 	    wps_build_mac_addr(msg, cred->mac_addr))
   1554 		return -1;
   1555 	return 0;
   1556 }
   1557 
   1558 
   1559 int wps_build_credential_wrap(struct wpabuf *msg,
   1560 			      const struct wps_credential *cred)
   1561 {
   1562 	struct wpabuf *wbuf;
   1563 	wbuf = wpabuf_alloc(200);
   1564 	if (wbuf == NULL)
   1565 		return -1;
   1566 	if (wps_build_credential(wbuf, cred)) {
   1567 		wpabuf_free(wbuf);
   1568 		return -1;
   1569 	}
   1570 	wpabuf_put_be16(msg, ATTR_CRED);
   1571 	wpabuf_put_be16(msg, wpabuf_len(wbuf));
   1572 	wpabuf_put_buf(msg, wbuf);
   1573 	wpabuf_free(wbuf);
   1574 	return 0;
   1575 }
   1576 
   1577 
   1578 int wps_build_cred(struct wps_data *wps, struct wpabuf *msg)
   1579 {
   1580 	struct wpabuf *cred;
   1581 
   1582 	if (wps->wps->registrar->skip_cred_build)
   1583 		goto skip_cred_build;
   1584 
   1585 	wpa_printf(MSG_DEBUG, "WPS:  * Credential");
   1586 	if (wps->use_cred) {
   1587 		os_memcpy(&wps->cred, wps->use_cred, sizeof(wps->cred));
   1588 		goto use_provided;
   1589 	}
   1590 	os_memset(&wps->cred, 0, sizeof(wps->cred));
   1591 
   1592 	os_memcpy(wps->cred.ssid, wps->wps->ssid, wps->wps->ssid_len);
   1593 	wps->cred.ssid_len = wps->wps->ssid_len;
   1594 
   1595 	/* Select the best authentication and encryption type */
   1596 	if (wps->auth_type & WPS_AUTH_WPA2PSK)
   1597 		wps->auth_type = WPS_AUTH_WPA2PSK;
   1598 	else if (wps->auth_type & WPS_AUTH_WPAPSK)
   1599 		wps->auth_type = WPS_AUTH_WPAPSK;
   1600 	else if (wps->auth_type & WPS_AUTH_OPEN)
   1601 		wps->auth_type = WPS_AUTH_OPEN;
   1602 	else {
   1603 		wpa_printf(MSG_DEBUG, "WPS: Unsupported auth_type 0x%x",
   1604 			   wps->auth_type);
   1605 		return -1;
   1606 	}
   1607 	wps->cred.auth_type = wps->auth_type;
   1608 
   1609 	if (wps->auth_type == WPS_AUTH_WPA2PSK ||
   1610 	    wps->auth_type == WPS_AUTH_WPAPSK) {
   1611 		if (wps->encr_type & WPS_ENCR_AES)
   1612 			wps->encr_type = WPS_ENCR_AES;
   1613 		else if (wps->encr_type & WPS_ENCR_TKIP)
   1614 			wps->encr_type = WPS_ENCR_TKIP;
   1615 		else {
   1616 			wpa_printf(MSG_DEBUG, "WPS: No suitable encryption "
   1617 				   "type for WPA/WPA2");
   1618 			return -1;
   1619 		}
   1620 	} else {
   1621 		if (wps->encr_type & WPS_ENCR_NONE)
   1622 			wps->encr_type = WPS_ENCR_NONE;
   1623 #ifdef CONFIG_TESTING_OPTIONS
   1624 		else if (wps->encr_type & WPS_ENCR_WEP)
   1625 			wps->encr_type = WPS_ENCR_WEP;
   1626 #endif /* CONFIG_TESTING_OPTIONS */
   1627 		else {
   1628 			wpa_printf(MSG_DEBUG, "WPS: No suitable encryption "
   1629 				   "type for non-WPA/WPA2 mode");
   1630 			return -1;
   1631 		}
   1632 	}
   1633 	wps->cred.encr_type = wps->encr_type;
   1634 	/*
   1635 	 * Set MAC address in the Credential to be the Enrollee's MAC address
   1636 	 */
   1637 	os_memcpy(wps->cred.mac_addr, wps->mac_addr_e, ETH_ALEN);
   1638 
   1639 	if (wps->wps->wps_state == WPS_STATE_NOT_CONFIGURED && wps->wps->ap &&
   1640 	    !wps->wps->registrar->disable_auto_conf) {
   1641 		u8 r[16];
   1642 		/* Generate a random passphrase */
   1643 		if (random_pool_ready() != 1 ||
   1644 		    random_get_bytes(r, sizeof(r)) < 0) {
   1645 			wpa_printf(MSG_INFO,
   1646 				   "WPS: Could not generate random PSK");
   1647 			return -1;
   1648 		}
   1649 		os_free(wps->new_psk);
   1650 		wps->new_psk = base64_encode(r, sizeof(r), &wps->new_psk_len);
   1651 		if (wps->new_psk == NULL)
   1652 			return -1;
   1653 		wps->new_psk_len--; /* remove newline */
   1654 		while (wps->new_psk_len &&
   1655 		       wps->new_psk[wps->new_psk_len - 1] == '=')
   1656 			wps->new_psk_len--;
   1657 		wpa_hexdump_ascii_key(MSG_DEBUG, "WPS: Generated passphrase",
   1658 				      wps->new_psk, wps->new_psk_len);
   1659 		os_memcpy(wps->cred.key, wps->new_psk, wps->new_psk_len);
   1660 		wps->cred.key_len = wps->new_psk_len;
   1661 	} else if (!wps->wps->registrar->force_per_enrollee_psk &&
   1662 		   wps->use_psk_key && wps->wps->psk_set) {
   1663 		char hex[65];
   1664 		wpa_printf(MSG_DEBUG, "WPS: Use PSK format for Network Key");
   1665 		wpa_snprintf_hex(hex, sizeof(hex), wps->wps->psk, 32);
   1666 		os_memcpy(wps->cred.key, hex, 32 * 2);
   1667 		wps->cred.key_len = 32 * 2;
   1668 	} else if (!wps->wps->registrar->force_per_enrollee_psk &&
   1669 		   wps->wps->network_key) {
   1670 		os_memcpy(wps->cred.key, wps->wps->network_key,
   1671 			  wps->wps->network_key_len);
   1672 		wps->cred.key_len = wps->wps->network_key_len;
   1673 	} else if (wps->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
   1674 		char hex[65];
   1675 		/* Generate a random per-device PSK */
   1676 		os_free(wps->new_psk);
   1677 		wps->new_psk_len = 32;
   1678 		wps->new_psk = os_malloc(wps->new_psk_len);
   1679 		if (wps->new_psk == NULL)
   1680 			return -1;
   1681 		if (random_pool_ready() != 1 ||
   1682 		    random_get_bytes(wps->new_psk, wps->new_psk_len) < 0) {
   1683 			wpa_printf(MSG_INFO,
   1684 				   "WPS: Could not generate random PSK");
   1685 			os_free(wps->new_psk);
   1686 			wps->new_psk = NULL;
   1687 			return -1;
   1688 		}
   1689 		wpa_hexdump_key(MSG_DEBUG, "WPS: Generated per-device PSK",
   1690 				wps->new_psk, wps->new_psk_len);
   1691 		wpa_snprintf_hex(hex, sizeof(hex), wps->new_psk,
   1692 				 wps->new_psk_len);
   1693 		os_memcpy(wps->cred.key, hex, wps->new_psk_len * 2);
   1694 		wps->cred.key_len = wps->new_psk_len * 2;
   1695 	}
   1696 
   1697 use_provided:
   1698 #ifdef CONFIG_WPS_TESTING
   1699 	if (wps_testing_dummy_cred)
   1700 		cred = wpabuf_alloc(200);
   1701 	else
   1702 		cred = NULL;
   1703 	if (cred) {
   1704 		struct wps_credential dummy;
   1705 		wpa_printf(MSG_DEBUG, "WPS: Add dummy credential");
   1706 		os_memset(&dummy, 0, sizeof(dummy));
   1707 		os_memcpy(dummy.ssid, "dummy", 5);
   1708 		dummy.ssid_len = 5;
   1709 		dummy.auth_type = WPS_AUTH_WPA2PSK;
   1710 		dummy.encr_type = WPS_ENCR_AES;
   1711 		os_memcpy(dummy.key, "dummy psk", 9);
   1712 		dummy.key_len = 9;
   1713 		os_memcpy(dummy.mac_addr, wps->mac_addr_e, ETH_ALEN);
   1714 		wps_build_credential(cred, &dummy);
   1715 		wpa_hexdump_buf(MSG_DEBUG, "WPS: Dummy Credential", cred);
   1716 
   1717 		wpabuf_put_be16(msg, ATTR_CRED);
   1718 		wpabuf_put_be16(msg, wpabuf_len(cred));
   1719 		wpabuf_put_buf(msg, cred);
   1720 
   1721 		wpabuf_free(cred);
   1722 	}
   1723 #endif /* CONFIG_WPS_TESTING */
   1724 
   1725 	cred = wpabuf_alloc(200);
   1726 	if (cred == NULL)
   1727 		return -1;
   1728 
   1729 	if (wps_build_credential(cred, &wps->cred)) {
   1730 		wpabuf_free(cred);
   1731 		return -1;
   1732 	}
   1733 
   1734 	wpabuf_put_be16(msg, ATTR_CRED);
   1735 	wpabuf_put_be16(msg, wpabuf_len(cred));
   1736 	wpabuf_put_buf(msg, cred);
   1737 	wpabuf_free(cred);
   1738 
   1739 skip_cred_build:
   1740 	if (wps->wps->registrar->extra_cred) {
   1741 		wpa_printf(MSG_DEBUG, "WPS:  * Credential (pre-configured)");
   1742 		wpabuf_put_buf(msg, wps->wps->registrar->extra_cred);
   1743 	}
   1744 
   1745 	return 0;
   1746 }
   1747 
   1748 
   1749 static int wps_build_ap_settings(struct wps_data *wps, struct wpabuf *msg)
   1750 {
   1751 	wpa_printf(MSG_DEBUG, "WPS:  * AP Settings");
   1752 
   1753 	if (wps_build_credential(msg, &wps->cred))
   1754 		return -1;
   1755 
   1756 	return 0;
   1757 }
   1758 
   1759 
   1760 static struct wpabuf * wps_build_ap_cred(struct wps_data *wps)
   1761 {
   1762 	struct wpabuf *msg, *plain;
   1763 
   1764 	msg = wpabuf_alloc(1000);
   1765 	if (msg == NULL)
   1766 		return NULL;
   1767 
   1768 	plain = wpabuf_alloc(200);
   1769 	if (plain == NULL) {
   1770 		wpabuf_free(msg);
   1771 		return NULL;
   1772 	}
   1773 
   1774 	if (wps_build_ap_settings(wps, plain)) {
   1775 		wpabuf_free(plain);
   1776 		wpabuf_free(msg);
   1777 		return NULL;
   1778 	}
   1779 
   1780 	wpabuf_put_be16(msg, ATTR_CRED);
   1781 	wpabuf_put_be16(msg, wpabuf_len(plain));
   1782 	wpabuf_put_buf(msg, plain);
   1783 	wpabuf_free(plain);
   1784 
   1785 	return msg;
   1786 }
   1787 
   1788 
   1789 static struct wpabuf * wps_build_m2(struct wps_data *wps)
   1790 {
   1791 	struct wpabuf *msg;
   1792 	int config_in_m2 = 0;
   1793 
   1794 	if (random_get_bytes(wps->nonce_r, WPS_NONCE_LEN) < 0)
   1795 		return NULL;
   1796 	wpa_hexdump(MSG_DEBUG, "WPS: Registrar Nonce",
   1797 		    wps->nonce_r, WPS_NONCE_LEN);
   1798 	wpa_hexdump(MSG_DEBUG, "WPS: UUID-R", wps->uuid_r, WPS_UUID_LEN);
   1799 
   1800 	wpa_printf(MSG_DEBUG, "WPS: Building Message M2");
   1801 	msg = wpabuf_alloc(1000);
   1802 	if (msg == NULL)
   1803 		return NULL;
   1804 
   1805 	if (wps_build_version(msg) ||
   1806 	    wps_build_msg_type(msg, WPS_M2) ||
   1807 	    wps_build_enrollee_nonce(wps, msg) ||
   1808 	    wps_build_registrar_nonce(wps, msg) ||
   1809 	    wps_build_uuid_r(wps, msg) ||
   1810 	    wps_build_public_key(wps, msg) ||
   1811 	    wps_derive_keys(wps) ||
   1812 	    wps_build_auth_type_flags(wps, msg) ||
   1813 	    wps_build_encr_type_flags(wps, msg) ||
   1814 	    wps_build_conn_type_flags(wps, msg) ||
   1815 	    wps_build_config_methods_r(wps->wps->registrar, msg) ||
   1816 	    wps_build_device_attrs(&wps->wps->dev, msg) ||
   1817 	    wps_build_rf_bands(&wps->wps->dev, msg,
   1818 			       wps->wps->rf_band_cb(wps->wps->cb_ctx)) ||
   1819 	    wps_build_assoc_state(wps, msg) ||
   1820 	    wps_build_config_error(msg, WPS_CFG_NO_ERROR) ||
   1821 	    wps_build_dev_password_id(msg, wps->dev_pw_id) ||
   1822 	    wps_build_os_version(&wps->wps->dev, msg) ||
   1823 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
   1824 		wpabuf_free(msg);
   1825 		return NULL;
   1826 	}
   1827 
   1828 #ifdef CONFIG_WPS_NFC
   1829 	if (wps->nfc_pw_token && wps->nfc_pw_token->pk_hash_provided_oob &&
   1830 	    wps->nfc_pw_token->pw_id == DEV_PW_NFC_CONNECTION_HANDOVER) {
   1831 		/*
   1832 		 * Use abbreviated handshake since public key hash allowed
   1833 		 * Enrollee to validate our public key similarly to how Enrollee
   1834 		 * public key was validated. There is no need to validate Device
   1835 		 * Password in this case.
   1836 		 */
   1837 		struct wpabuf *plain = wpabuf_alloc(500);
   1838 		if (plain == NULL ||
   1839 		    wps_build_cred(wps, plain) ||
   1840 		    wps_build_key_wrap_auth(wps, plain) ||
   1841 		    wps_build_encr_settings(wps, msg, plain)) {
   1842 			wpabuf_free(msg);
   1843 			wpabuf_free(plain);
   1844 			return NULL;
   1845 		}
   1846 		wpabuf_free(plain);
   1847 		config_in_m2 = 1;
   1848 	}
   1849 #endif /* CONFIG_WPS_NFC */
   1850 
   1851 	if (wps_build_authenticator(wps, msg)) {
   1852 		wpabuf_free(msg);
   1853 		return NULL;
   1854 	}
   1855 
   1856 	wps->int_reg = 1;
   1857 	wps->state = config_in_m2 ? RECV_DONE : RECV_M3;
   1858 	return msg;
   1859 }
   1860 
   1861 
   1862 static struct wpabuf * wps_build_m2d(struct wps_data *wps)
   1863 {
   1864 	struct wpabuf *msg;
   1865 	u16 err = wps->config_error;
   1866 
   1867 	wpa_printf(MSG_DEBUG, "WPS: Building Message M2D");
   1868 	msg = wpabuf_alloc(1000);
   1869 	if (msg == NULL)
   1870 		return NULL;
   1871 
   1872 	if (wps->wps->ap && wps->wps->ap_setup_locked &&
   1873 	    err == WPS_CFG_NO_ERROR)
   1874 		err = WPS_CFG_SETUP_LOCKED;
   1875 
   1876 	if (wps_build_version(msg) ||
   1877 	    wps_build_msg_type(msg, WPS_M2D) ||
   1878 	    wps_build_enrollee_nonce(wps, msg) ||
   1879 	    wps_build_registrar_nonce(wps, msg) ||
   1880 	    wps_build_uuid_r(wps, msg) ||
   1881 	    wps_build_auth_type_flags(wps, msg) ||
   1882 	    wps_build_encr_type_flags(wps, msg) ||
   1883 	    wps_build_conn_type_flags(wps, msg) ||
   1884 	    wps_build_config_methods_r(wps->wps->registrar, msg) ||
   1885 	    wps_build_device_attrs(&wps->wps->dev, msg) ||
   1886 	    wps_build_rf_bands(&wps->wps->dev, msg,
   1887 			       wps->wps->rf_band_cb(wps->wps->cb_ctx)) ||
   1888 	    wps_build_assoc_state(wps, msg) ||
   1889 	    wps_build_config_error(msg, err) ||
   1890 	    wps_build_os_version(&wps->wps->dev, msg) ||
   1891 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
   1892 		wpabuf_free(msg);
   1893 		return NULL;
   1894 	}
   1895 
   1896 	wps->state = RECV_M2D_ACK;
   1897 	return msg;
   1898 }
   1899 
   1900 
   1901 static struct wpabuf * wps_build_m4(struct wps_data *wps)
   1902 {
   1903 	struct wpabuf *msg, *plain;
   1904 
   1905 	wpa_printf(MSG_DEBUG, "WPS: Building Message M4");
   1906 
   1907 	wps_derive_psk(wps, wps->dev_password, wps->dev_password_len);
   1908 
   1909 	plain = wpabuf_alloc(200);
   1910 	if (plain == NULL)
   1911 		return NULL;
   1912 
   1913 	msg = wpabuf_alloc(1000);
   1914 	if (msg == NULL) {
   1915 		wpabuf_free(plain);
   1916 		return NULL;
   1917 	}
   1918 
   1919 	if (wps_build_version(msg) ||
   1920 	    wps_build_msg_type(msg, WPS_M4) ||
   1921 	    wps_build_enrollee_nonce(wps, msg) ||
   1922 	    wps_build_r_hash(wps, msg) ||
   1923 	    wps_build_r_snonce1(wps, plain) ||
   1924 	    wps_build_key_wrap_auth(wps, plain) ||
   1925 	    wps_build_encr_settings(wps, msg, plain) ||
   1926 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
   1927 	    wps_build_authenticator(wps, msg)) {
   1928 		wpabuf_free(plain);
   1929 		wpabuf_free(msg);
   1930 		return NULL;
   1931 	}
   1932 	wpabuf_free(plain);
   1933 
   1934 	wps->state = RECV_M5;
   1935 	return msg;
   1936 }
   1937 
   1938 
   1939 static struct wpabuf * wps_build_m6(struct wps_data *wps)
   1940 {
   1941 	struct wpabuf *msg, *plain;
   1942 
   1943 	wpa_printf(MSG_DEBUG, "WPS: Building Message M6");
   1944 
   1945 	plain = wpabuf_alloc(200);
   1946 	if (plain == NULL)
   1947 		return NULL;
   1948 
   1949 	msg = wpabuf_alloc(1000);
   1950 	if (msg == NULL) {
   1951 		wpabuf_free(plain);
   1952 		return NULL;
   1953 	}
   1954 
   1955 	if (wps_build_version(msg) ||
   1956 	    wps_build_msg_type(msg, WPS_M6) ||
   1957 	    wps_build_enrollee_nonce(wps, msg) ||
   1958 	    wps_build_r_snonce2(wps, plain) ||
   1959 	    wps_build_key_wrap_auth(wps, plain) ||
   1960 	    wps_build_encr_settings(wps, msg, plain) ||
   1961 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
   1962 	    wps_build_authenticator(wps, msg)) {
   1963 		wpabuf_free(plain);
   1964 		wpabuf_free(msg);
   1965 		return NULL;
   1966 	}
   1967 	wpabuf_free(plain);
   1968 
   1969 	wps->wps_pin_revealed = 1;
   1970 	wps->state = RECV_M7;
   1971 	return msg;
   1972 }
   1973 
   1974 
   1975 static struct wpabuf * wps_build_m8(struct wps_data *wps)
   1976 {
   1977 	struct wpabuf *msg, *plain;
   1978 
   1979 	wpa_printf(MSG_DEBUG, "WPS: Building Message M8");
   1980 
   1981 	plain = wpabuf_alloc(500);
   1982 	if (plain == NULL)
   1983 		return NULL;
   1984 
   1985 	msg = wpabuf_alloc(1000);
   1986 	if (msg == NULL) {
   1987 		wpabuf_free(plain);
   1988 		return NULL;
   1989 	}
   1990 
   1991 	if (wps_build_version(msg) ||
   1992 	    wps_build_msg_type(msg, WPS_M8) ||
   1993 	    wps_build_enrollee_nonce(wps, msg) ||
   1994 	    ((wps->wps->ap || wps->er) && wps_build_cred(wps, plain)) ||
   1995 	    (!wps->wps->ap && !wps->er && wps_build_ap_settings(wps, plain)) ||
   1996 	    wps_build_key_wrap_auth(wps, plain) ||
   1997 	    wps_build_encr_settings(wps, msg, plain) ||
   1998 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
   1999 	    wps_build_authenticator(wps, msg)) {
   2000 		wpabuf_free(plain);
   2001 		wpabuf_free(msg);
   2002 		return NULL;
   2003 	}
   2004 	wpabuf_free(plain);
   2005 
   2006 	wps->state = RECV_DONE;
   2007 	return msg;
   2008 }
   2009 
   2010 
   2011 struct wpabuf * wps_registrar_get_msg(struct wps_data *wps,
   2012 				      enum wsc_op_code *op_code)
   2013 {
   2014 	struct wpabuf *msg;
   2015 
   2016 #ifdef CONFIG_WPS_UPNP
   2017 	if (!wps->int_reg && wps->wps->wps_upnp) {
   2018 		struct upnp_pending_message *p, *prev = NULL;
   2019 		if (wps->ext_reg > 1)
   2020 			wps_registrar_free_pending_m2(wps->wps);
   2021 		p = wps->wps->upnp_msgs;
   2022 		/* TODO: check pending message MAC address */
   2023 		while (p && p->next) {
   2024 			prev = p;
   2025 			p = p->next;
   2026 		}
   2027 		if (p) {
   2028 			wpa_printf(MSG_DEBUG, "WPS: Use pending message from "
   2029 				   "UPnP");
   2030 			if (prev)
   2031 				prev->next = NULL;
   2032 			else
   2033 				wps->wps->upnp_msgs = NULL;
   2034 			msg = p->msg;
   2035 			switch (p->type) {
   2036 			case WPS_WSC_ACK:
   2037 				*op_code = WSC_ACK;
   2038 				break;
   2039 			case WPS_WSC_NACK:
   2040 				*op_code = WSC_NACK;
   2041 				break;
   2042 			default:
   2043 				*op_code = WSC_MSG;
   2044 				break;
   2045 			}
   2046 			os_free(p);
   2047 			if (wps->ext_reg == 0)
   2048 				wps->ext_reg = 1;
   2049 			return msg;
   2050 		}
   2051 	}
   2052 	if (wps->ext_reg) {
   2053 		wpa_printf(MSG_DEBUG, "WPS: Using external Registrar, but no "
   2054 			   "pending message available");
   2055 		return NULL;
   2056 	}
   2057 #endif /* CONFIG_WPS_UPNP */
   2058 
   2059 	switch (wps->state) {
   2060 	case SEND_M2:
   2061 		if (wps_get_dev_password(wps) < 0)
   2062 			msg = wps_build_m2d(wps);
   2063 		else
   2064 			msg = wps_build_m2(wps);
   2065 		*op_code = WSC_MSG;
   2066 		break;
   2067 	case SEND_M2D:
   2068 		msg = wps_build_m2d(wps);
   2069 		*op_code = WSC_MSG;
   2070 		break;
   2071 	case SEND_M4:
   2072 		msg = wps_build_m4(wps);
   2073 		*op_code = WSC_MSG;
   2074 		break;
   2075 	case SEND_M6:
   2076 		msg = wps_build_m6(wps);
   2077 		*op_code = WSC_MSG;
   2078 		break;
   2079 	case SEND_M8:
   2080 		msg = wps_build_m8(wps);
   2081 		*op_code = WSC_MSG;
   2082 		break;
   2083 	case RECV_DONE:
   2084 		msg = wps_build_wsc_ack(wps);
   2085 		*op_code = WSC_ACK;
   2086 		break;
   2087 	case SEND_WSC_NACK:
   2088 		msg = wps_build_wsc_nack(wps);
   2089 		*op_code = WSC_NACK;
   2090 		break;
   2091 	default:
   2092 		wpa_printf(MSG_DEBUG, "WPS: Unsupported state %d for building "
   2093 			   "a message", wps->state);
   2094 		msg = NULL;
   2095 		break;
   2096 	}
   2097 
   2098 	if (*op_code == WSC_MSG && msg) {
   2099 		/* Save a copy of the last message for Authenticator derivation
   2100 		 */
   2101 		wpabuf_free(wps->last_msg);
   2102 		wps->last_msg = wpabuf_dup(msg);
   2103 	}
   2104 
   2105 	return msg;
   2106 }
   2107 
   2108 
   2109 static int wps_process_enrollee_nonce(struct wps_data *wps, const u8 *e_nonce)
   2110 {
   2111 	if (e_nonce == NULL) {
   2112 		wpa_printf(MSG_DEBUG, "WPS: No Enrollee Nonce received");
   2113 		return -1;
   2114 	}
   2115 
   2116 	os_memcpy(wps->nonce_e, e_nonce, WPS_NONCE_LEN);
   2117 	wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Nonce",
   2118 		    wps->nonce_e, WPS_NONCE_LEN);
   2119 
   2120 	return 0;
   2121 }
   2122 
   2123 
   2124 static int wps_process_registrar_nonce(struct wps_data *wps, const u8 *r_nonce)
   2125 {
   2126 	if (r_nonce == NULL) {
   2127 		wpa_printf(MSG_DEBUG, "WPS: No Registrar Nonce received");
   2128 		return -1;
   2129 	}
   2130 
   2131 	if (os_memcmp(wps->nonce_r, r_nonce, WPS_NONCE_LEN) != 0) {
   2132 		wpa_printf(MSG_DEBUG, "WPS: Invalid Registrar Nonce received");
   2133 		return -1;
   2134 	}
   2135 
   2136 	return 0;
   2137 }
   2138 
   2139 
   2140 static int wps_process_uuid_e(struct wps_data *wps, const u8 *uuid_e)
   2141 {
   2142 	if (uuid_e == NULL) {
   2143 		wpa_printf(MSG_DEBUG, "WPS: No UUID-E received");
   2144 		return -1;
   2145 	}
   2146 
   2147 	os_memcpy(wps->uuid_e, uuid_e, WPS_UUID_LEN);
   2148 	wpa_hexdump(MSG_DEBUG, "WPS: UUID-E", wps->uuid_e, WPS_UUID_LEN);
   2149 
   2150 	return 0;
   2151 }
   2152 
   2153 
   2154 static int wps_process_dev_password_id(struct wps_data *wps, const u8 *pw_id)
   2155 {
   2156 	if (pw_id == NULL) {
   2157 		wpa_printf(MSG_DEBUG, "WPS: No Device Password ID received");
   2158 		return -1;
   2159 	}
   2160 
   2161 	wps->dev_pw_id = WPA_GET_BE16(pw_id);
   2162 	wpa_printf(MSG_DEBUG, "WPS: Device Password ID %d", wps->dev_pw_id);
   2163 
   2164 	return 0;
   2165 }
   2166 
   2167 
   2168 static int wps_process_e_hash1(struct wps_data *wps, const u8 *e_hash1)
   2169 {
   2170 	if (e_hash1 == NULL) {
   2171 		wpa_printf(MSG_DEBUG, "WPS: No E-Hash1 received");
   2172 		return -1;
   2173 	}
   2174 
   2175 	os_memcpy(wps->peer_hash1, e_hash1, WPS_HASH_LEN);
   2176 	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash1", wps->peer_hash1, WPS_HASH_LEN);
   2177 
   2178 	return 0;
   2179 }
   2180 
   2181 
   2182 static int wps_process_e_hash2(struct wps_data *wps, const u8 *e_hash2)
   2183 {
   2184 	if (e_hash2 == NULL) {
   2185 		wpa_printf(MSG_DEBUG, "WPS: No E-Hash2 received");
   2186 		return -1;
   2187 	}
   2188 
   2189 	os_memcpy(wps->peer_hash2, e_hash2, WPS_HASH_LEN);
   2190 	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash2", wps->peer_hash2, WPS_HASH_LEN);
   2191 
   2192 	return 0;
   2193 }
   2194 
   2195 
   2196 static int wps_process_e_snonce1(struct wps_data *wps, const u8 *e_snonce1)
   2197 {
   2198 	u8 hash[SHA256_MAC_LEN];
   2199 	const u8 *addr[4];
   2200 	size_t len[4];
   2201 
   2202 	if (e_snonce1 == NULL) {
   2203 		wpa_printf(MSG_DEBUG, "WPS: No E-SNonce1 received");
   2204 		return -1;
   2205 	}
   2206 
   2207 	wpa_hexdump_key(MSG_DEBUG, "WPS: E-SNonce1", e_snonce1,
   2208 			WPS_SECRET_NONCE_LEN);
   2209 
   2210 	/* E-Hash1 = HMAC_AuthKey(E-S1 || PSK1 || PK_E || PK_R) */
   2211 	addr[0] = e_snonce1;
   2212 	len[0] = WPS_SECRET_NONCE_LEN;
   2213 	addr[1] = wps->psk1;
   2214 	len[1] = WPS_PSK_LEN;
   2215 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
   2216 	len[2] = wpabuf_len(wps->dh_pubkey_e);
   2217 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
   2218 	len[3] = wpabuf_len(wps->dh_pubkey_r);
   2219 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
   2220 
   2221 	if (os_memcmp_const(wps->peer_hash1, hash, WPS_HASH_LEN) != 0) {
   2222 		wpa_printf(MSG_DEBUG, "WPS: E-Hash1 derived from E-S1 does "
   2223 			   "not match with the pre-committed value");
   2224 		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
   2225 		wps_pwd_auth_fail_event(wps->wps, 0, 1, wps->mac_addr_e);
   2226 		return -1;
   2227 	}
   2228 
   2229 	wpa_printf(MSG_DEBUG, "WPS: Enrollee proved knowledge of the first "
   2230 		   "half of the device password");
   2231 
   2232 	return 0;
   2233 }
   2234 
   2235 
   2236 static int wps_process_e_snonce2(struct wps_data *wps, const u8 *e_snonce2)
   2237 {
   2238 	u8 hash[SHA256_MAC_LEN];
   2239 	const u8 *addr[4];
   2240 	size_t len[4];
   2241 
   2242 	if (e_snonce2 == NULL) {
   2243 		wpa_printf(MSG_DEBUG, "WPS: No E-SNonce2 received");
   2244 		return -1;
   2245 	}
   2246 
   2247 	wpa_hexdump_key(MSG_DEBUG, "WPS: E-SNonce2", e_snonce2,
   2248 			WPS_SECRET_NONCE_LEN);
   2249 
   2250 	/* E-Hash2 = HMAC_AuthKey(E-S2 || PSK2 || PK_E || PK_R) */
   2251 	addr[0] = e_snonce2;
   2252 	len[0] = WPS_SECRET_NONCE_LEN;
   2253 	addr[1] = wps->psk2;
   2254 	len[1] = WPS_PSK_LEN;
   2255 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
   2256 	len[2] = wpabuf_len(wps->dh_pubkey_e);
   2257 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
   2258 	len[3] = wpabuf_len(wps->dh_pubkey_r);
   2259 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
   2260 
   2261 	if (os_memcmp_const(wps->peer_hash2, hash, WPS_HASH_LEN) != 0) {
   2262 		wpa_printf(MSG_DEBUG, "WPS: E-Hash2 derived from E-S2 does "
   2263 			   "not match with the pre-committed value");
   2264 		wps_registrar_invalidate_pin(wps->wps->registrar, wps->uuid_e);
   2265 		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
   2266 		wps_pwd_auth_fail_event(wps->wps, 0, 2, wps->mac_addr_e);
   2267 		return -1;
   2268 	}
   2269 
   2270 	wpa_printf(MSG_DEBUG, "WPS: Enrollee proved knowledge of the second "
   2271 		   "half of the device password");
   2272 	wps->wps_pin_revealed = 0;
   2273 	wps_registrar_unlock_pin(wps->wps->registrar, wps->uuid_e);
   2274 
   2275 	/*
   2276 	 * In case wildcard PIN is used and WPS handshake succeeds in the first
   2277 	 * attempt, wps_registrar_unlock_pin() would not free the PIN, so make
   2278 	 * sure the PIN gets invalidated here.
   2279 	 */
   2280 	wps_registrar_invalidate_pin(wps->wps->registrar, wps->uuid_e);
   2281 
   2282 	return 0;
   2283 }
   2284 
   2285 
   2286 static int wps_process_mac_addr(struct wps_data *wps, const u8 *mac_addr)
   2287 {
   2288 	if (mac_addr == NULL) {
   2289 		wpa_printf(MSG_DEBUG, "WPS: No MAC Address received");
   2290 		return -1;
   2291 	}
   2292 
   2293 	wpa_printf(MSG_DEBUG, "WPS: Enrollee MAC Address " MACSTR,
   2294 		   MAC2STR(mac_addr));
   2295 	os_memcpy(wps->mac_addr_e, mac_addr, ETH_ALEN);
   2296 	os_memcpy(wps->peer_dev.mac_addr, mac_addr, ETH_ALEN);
   2297 
   2298 	return 0;
   2299 }
   2300 
   2301 
   2302 static int wps_process_pubkey(struct wps_data *wps, const u8 *pk,
   2303 			      size_t pk_len)
   2304 {
   2305 	if (pk == NULL || pk_len == 0) {
   2306 		wpa_printf(MSG_DEBUG, "WPS: No Public Key received");
   2307 		return -1;
   2308 	}
   2309 
   2310 	wpabuf_free(wps->dh_pubkey_e);
   2311 	wps->dh_pubkey_e = wpabuf_alloc_copy(pk, pk_len);
   2312 	if (wps->dh_pubkey_e == NULL)
   2313 		return -1;
   2314 
   2315 	return 0;
   2316 }
   2317 
   2318 
   2319 static int wps_process_auth_type_flags(struct wps_data *wps, const u8 *auth)
   2320 {
   2321 	u16 auth_types;
   2322 
   2323 	if (auth == NULL) {
   2324 		wpa_printf(MSG_DEBUG, "WPS: No Authentication Type flags "
   2325 			   "received");
   2326 		return -1;
   2327 	}
   2328 
   2329 	auth_types = WPA_GET_BE16(auth);
   2330 
   2331 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Authentication Type flags 0x%x",
   2332 		   auth_types);
   2333 	wps->auth_type = wps->wps->auth_types & auth_types;
   2334 	if (wps->auth_type == 0) {
   2335 		wpa_printf(MSG_DEBUG, "WPS: No match in supported "
   2336 			   "authentication types (own 0x%x Enrollee 0x%x)",
   2337 			   wps->wps->auth_types, auth_types);
   2338 #ifdef WPS_WORKAROUNDS
   2339 		/*
   2340 		 * Some deployed implementations seem to advertise incorrect
   2341 		 * information in this attribute. For example, Linksys WRT350N
   2342 		 * seems to have a byteorder bug that breaks this negotiation.
   2343 		 * In order to interoperate with existing implementations,
   2344 		 * assume that the Enrollee supports everything we do.
   2345 		 */
   2346 		wpa_printf(MSG_DEBUG, "WPS: Workaround - assume Enrollee "
   2347 			   "does not advertise supported authentication types "
   2348 			   "correctly");
   2349 		wps->auth_type = wps->wps->auth_types;
   2350 #else /* WPS_WORKAROUNDS */
   2351 		return -1;
   2352 #endif /* WPS_WORKAROUNDS */
   2353 	}
   2354 
   2355 	return 0;
   2356 }
   2357 
   2358 
   2359 static int wps_process_encr_type_flags(struct wps_data *wps, const u8 *encr)
   2360 {
   2361 	u16 encr_types;
   2362 
   2363 	if (encr == NULL) {
   2364 		wpa_printf(MSG_DEBUG, "WPS: No Encryption Type flags "
   2365 			   "received");
   2366 		return -1;
   2367 	}
   2368 
   2369 	encr_types = WPA_GET_BE16(encr);
   2370 
   2371 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Encryption Type flags 0x%x",
   2372 		   encr_types);
   2373 	wps->encr_type = wps->wps->encr_types & encr_types;
   2374 	if (wps->encr_type == 0) {
   2375 		wpa_printf(MSG_DEBUG, "WPS: No match in supported "
   2376 			   "encryption types (own 0x%x Enrollee 0x%x)",
   2377 			   wps->wps->encr_types, encr_types);
   2378 #ifdef WPS_WORKAROUNDS
   2379 		/*
   2380 		 * Some deployed implementations seem to advertise incorrect
   2381 		 * information in this attribute. For example, Linksys WRT350N
   2382 		 * seems to have a byteorder bug that breaks this negotiation.
   2383 		 * In order to interoperate with existing implementations,
   2384 		 * assume that the Enrollee supports everything we do.
   2385 		 */
   2386 		wpa_printf(MSG_DEBUG, "WPS: Workaround - assume Enrollee "
   2387 			   "does not advertise supported encryption types "
   2388 			   "correctly");
   2389 		wps->encr_type = wps->wps->encr_types;
   2390 #else /* WPS_WORKAROUNDS */
   2391 		return -1;
   2392 #endif /* WPS_WORKAROUNDS */
   2393 	}
   2394 
   2395 	return 0;
   2396 }
   2397 
   2398 
   2399 static int wps_process_conn_type_flags(struct wps_data *wps, const u8 *conn)
   2400 {
   2401 	if (conn == NULL) {
   2402 		wpa_printf(MSG_DEBUG, "WPS: No Connection Type flags "
   2403 			   "received");
   2404 		return -1;
   2405 	}
   2406 
   2407 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Connection Type flags 0x%x",
   2408 		   *conn);
   2409 
   2410 	return 0;
   2411 }
   2412 
   2413 
   2414 static int wps_process_config_methods(struct wps_data *wps, const u8 *methods)
   2415 {
   2416 	u16 m;
   2417 
   2418 	if (methods == NULL) {
   2419 		wpa_printf(MSG_DEBUG, "WPS: No Config Methods received");
   2420 		return -1;
   2421 	}
   2422 
   2423 	m = WPA_GET_BE16(methods);
   2424 
   2425 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Config Methods 0x%x"
   2426 		   "%s%s%s%s%s%s%s%s%s", m,
   2427 		   m & WPS_CONFIG_USBA ? " [USBA]" : "",
   2428 		   m & WPS_CONFIG_ETHERNET ? " [Ethernet]" : "",
   2429 		   m & WPS_CONFIG_LABEL ? " [Label]" : "",
   2430 		   m & WPS_CONFIG_DISPLAY ? " [Display]" : "",
   2431 		   m & WPS_CONFIG_EXT_NFC_TOKEN ? " [Ext NFC Token]" : "",
   2432 		   m & WPS_CONFIG_INT_NFC_TOKEN ? " [Int NFC Token]" : "",
   2433 		   m & WPS_CONFIG_NFC_INTERFACE ? " [NFC]" : "",
   2434 		   m & WPS_CONFIG_PUSHBUTTON ? " [PBC]" : "",
   2435 		   m & WPS_CONFIG_KEYPAD ? " [Keypad]" : "");
   2436 
   2437 	if (!(m & WPS_CONFIG_DISPLAY) && !wps->use_psk_key) {
   2438 		/*
   2439 		 * The Enrollee does not have a display so it is unlikely to be
   2440 		 * able to show the passphrase to a user and as such, could
   2441 		 * benefit from receiving PSK to reduce key derivation time.
   2442 		 */
   2443 		wpa_printf(MSG_DEBUG, "WPS: Prefer PSK format key due to "
   2444 			   "Enrollee not supporting display");
   2445 		wps->use_psk_key = 1;
   2446 	}
   2447 
   2448 	return 0;
   2449 }
   2450 
   2451 
   2452 static int wps_process_wps_state(struct wps_data *wps, const u8 *state)
   2453 {
   2454 	if (state == NULL) {
   2455 		wpa_printf(MSG_DEBUG, "WPS: No Wi-Fi Protected Setup State "
   2456 			   "received");
   2457 		return -1;
   2458 	}
   2459 
   2460 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Wi-Fi Protected Setup State %d",
   2461 		   *state);
   2462 
   2463 	return 0;
   2464 }
   2465 
   2466 
   2467 static int wps_process_assoc_state(struct wps_data *wps, const u8 *assoc)
   2468 {
   2469 	u16 a;
   2470 
   2471 	if (assoc == NULL) {
   2472 		wpa_printf(MSG_DEBUG, "WPS: No Association State received");
   2473 		return -1;
   2474 	}
   2475 
   2476 	a = WPA_GET_BE16(assoc);
   2477 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Association State %d", a);
   2478 
   2479 	return 0;
   2480 }
   2481 
   2482 
   2483 static int wps_process_config_error(struct wps_data *wps, const u8 *err)
   2484 {
   2485 	u16 e;
   2486 
   2487 	if (err == NULL) {
   2488 		wpa_printf(MSG_DEBUG, "WPS: No Configuration Error received");
   2489 		return -1;
   2490 	}
   2491 
   2492 	e = WPA_GET_BE16(err);
   2493 	wpa_printf(MSG_DEBUG, "WPS: Enrollee Configuration Error %d", e);
   2494 
   2495 	return 0;
   2496 }
   2497 
   2498 
   2499 static int wps_registrar_p2p_dev_addr_match(struct wps_data *wps)
   2500 {
   2501 #ifdef CONFIG_P2P
   2502 	struct wps_registrar *reg = wps->wps->registrar;
   2503 
   2504 	if (is_zero_ether_addr(reg->p2p_dev_addr))
   2505 		return 1; /* no filtering in use */
   2506 
   2507 	if (os_memcmp(reg->p2p_dev_addr, wps->p2p_dev_addr, ETH_ALEN) != 0) {
   2508 		wpa_printf(MSG_DEBUG, "WPS: No match on P2P Device Address "
   2509 			   "filtering for PBC: expected " MACSTR " was "
   2510 			   MACSTR " - indicate PBC session overlap",
   2511 			   MAC2STR(reg->p2p_dev_addr),
   2512 			   MAC2STR(wps->p2p_dev_addr));
   2513 		return 0;
   2514 	}
   2515 #endif /* CONFIG_P2P */
   2516 	return 1;
   2517 }
   2518 
   2519 
   2520 static int wps_registrar_skip_overlap(struct wps_data *wps)
   2521 {
   2522 #ifdef CONFIG_P2P
   2523 	struct wps_registrar *reg = wps->wps->registrar;
   2524 
   2525 	if (is_zero_ether_addr(reg->p2p_dev_addr))
   2526 		return 0; /* no specific Enrollee selected */
   2527 
   2528 	if (os_memcmp(reg->p2p_dev_addr, wps->p2p_dev_addr, ETH_ALEN) == 0) {
   2529 		wpa_printf(MSG_DEBUG, "WPS: Skip PBC overlap due to selected "
   2530 			   "Enrollee match");
   2531 		return 1;
   2532 	}
   2533 #endif /* CONFIG_P2P */
   2534 	return 0;
   2535 }
   2536 
   2537 
   2538 static enum wps_process_res wps_process_m1(struct wps_data *wps,
   2539 					   struct wps_parse_attr *attr)
   2540 {
   2541 	wpa_printf(MSG_DEBUG, "WPS: Received M1");
   2542 
   2543 	if (wps->state != RECV_M1) {
   2544 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
   2545 			   "receiving M1", wps->state);
   2546 		return WPS_FAILURE;
   2547 	}
   2548 
   2549 	if (wps_process_uuid_e(wps, attr->uuid_e) ||
   2550 	    wps_process_mac_addr(wps, attr->mac_addr) ||
   2551 	    wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
   2552 	    wps_process_pubkey(wps, attr->public_key, attr->public_key_len) ||
   2553 	    wps_process_auth_type_flags(wps, attr->auth_type_flags) ||
   2554 	    wps_process_encr_type_flags(wps, attr->encr_type_flags) ||
   2555 	    wps_process_conn_type_flags(wps, attr->conn_type_flags) ||
   2556 	    wps_process_config_methods(wps, attr->config_methods) ||
   2557 	    wps_process_wps_state(wps, attr->wps_state) ||
   2558 	    wps_process_device_attrs(&wps->peer_dev, attr) ||
   2559 	    wps_process_rf_bands(&wps->peer_dev, attr->rf_bands) ||
   2560 	    wps_process_assoc_state(wps, attr->assoc_state) ||
   2561 	    wps_process_dev_password_id(wps, attr->dev_password_id) ||
   2562 	    wps_process_config_error(wps, attr->config_error) ||
   2563 	    wps_process_os_version(&wps->peer_dev, attr->os_version))
   2564 		return WPS_FAILURE;
   2565 
   2566 	if (wps->dev_pw_id < 0x10 &&
   2567 	    wps->dev_pw_id != DEV_PW_DEFAULT &&
   2568 	    wps->dev_pw_id != DEV_PW_USER_SPECIFIED &&
   2569 	    wps->dev_pw_id != DEV_PW_MACHINE_SPECIFIED &&
   2570 	    wps->dev_pw_id != DEV_PW_REGISTRAR_SPECIFIED &&
   2571 #ifdef CONFIG_WPS_NFC
   2572 	    wps->dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER &&
   2573 #endif /* CONFIG_WPS_NFC */
   2574 	    (wps->dev_pw_id != DEV_PW_PUSHBUTTON ||
   2575 	     !wps->wps->registrar->pbc)) {
   2576 		wpa_printf(MSG_DEBUG, "WPS: Unsupported Device Password ID %d",
   2577 			   wps->dev_pw_id);
   2578 		wps->state = SEND_M2D;
   2579 		return WPS_CONTINUE;
   2580 	}
   2581 
   2582 #ifdef CONFIG_WPS_NFC
   2583 	if (wps->dev_pw_id >= 0x10 ||
   2584 	    wps->dev_pw_id == DEV_PW_NFC_CONNECTION_HANDOVER) {
   2585 		struct wps_nfc_pw_token *token;
   2586 		const u8 *addr[1];
   2587 		u8 hash[WPS_HASH_LEN];
   2588 
   2589 		wpa_printf(MSG_DEBUG, "WPS: Searching for NFC token match for id=%d (ctx %p registrar %p)",
   2590 			   wps->dev_pw_id, wps->wps, wps->wps->registrar);
   2591 		token = wps_get_nfc_pw_token(
   2592 			&wps->wps->registrar->nfc_pw_tokens, wps->dev_pw_id);
   2593 		if (token && token->peer_pk_hash_known) {
   2594 			wpa_printf(MSG_DEBUG, "WPS: Found matching NFC "
   2595 				   "Password Token");
   2596 			dl_list_del(&token->list);
   2597 			wps->nfc_pw_token = token;
   2598 
   2599 			addr[0] = attr->public_key;
   2600 			sha256_vector(1, addr, &attr->public_key_len, hash);
   2601 			if (os_memcmp_const(hash,
   2602 					    wps->nfc_pw_token->pubkey_hash,
   2603 					    WPS_OOB_PUBKEY_HASH_LEN) != 0) {
   2604 				wpa_printf(MSG_ERROR, "WPS: Public Key hash "
   2605 					   "mismatch");
   2606 				wps->state = SEND_M2D;
   2607 				wps->config_error =
   2608 					WPS_CFG_PUBLIC_KEY_HASH_MISMATCH;
   2609 				return WPS_CONTINUE;
   2610 			}
   2611 		} else if (token) {
   2612 			wpa_printf(MSG_DEBUG, "WPS: Found matching NFC "
   2613 				   "Password Token (no peer PK hash)");
   2614 			wps->nfc_pw_token = token;
   2615 		} else if (wps->dev_pw_id >= 0x10 &&
   2616 			   wps->wps->ap_nfc_dev_pw_id == wps->dev_pw_id &&
   2617 			   wps->wps->ap_nfc_dev_pw) {
   2618 			wpa_printf(MSG_DEBUG, "WPS: Found match with own NFC Password Token");
   2619 		}
   2620 	}
   2621 #endif /* CONFIG_WPS_NFC */
   2622 
   2623 	if (wps->dev_pw_id == DEV_PW_PUSHBUTTON) {
   2624 		if ((wps->wps->registrar->force_pbc_overlap ||
   2625 		     wps_registrar_pbc_overlap(wps->wps->registrar,
   2626 					       wps->mac_addr_e, wps->uuid_e) ||
   2627 		     !wps_registrar_p2p_dev_addr_match(wps)) &&
   2628 		    !wps_registrar_skip_overlap(wps)) {
   2629 			wpa_printf(MSG_DEBUG, "WPS: PBC overlap - deny PBC "
   2630 				   "negotiation");
   2631 			wps->state = SEND_M2D;
   2632 			wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
   2633 			wps_pbc_overlap_event(wps->wps);
   2634 			wps_fail_event(wps->wps, WPS_M1,
   2635 				       WPS_CFG_MULTIPLE_PBC_DETECTED,
   2636 				       WPS_EI_NO_ERROR, wps->mac_addr_e);
   2637 			wps->wps->registrar->force_pbc_overlap = 1;
   2638 			return WPS_CONTINUE;
   2639 		}
   2640 		wps_registrar_add_pbc_session(wps->wps->registrar,
   2641 					      wps->mac_addr_e, wps->uuid_e);
   2642 		wps->pbc = 1;
   2643 	}
   2644 
   2645 #ifdef WPS_WORKAROUNDS
   2646 	/*
   2647 	 * It looks like Mac OS X 10.6.3 and 10.6.4 do not like Network Key in
   2648 	 * passphrase format. To avoid interop issues, force PSK format to be
   2649 	 * used.
   2650 	 */
   2651 	if (!wps->use_psk_key &&
   2652 	    wps->peer_dev.manufacturer &&
   2653 	    os_strncmp(wps->peer_dev.manufacturer, "Apple ", 6) == 0 &&
   2654 	    wps->peer_dev.model_name &&
   2655 	    os_strcmp(wps->peer_dev.model_name, "AirPort") == 0) {
   2656 		wpa_printf(MSG_DEBUG, "WPS: Workaround - Force Network Key in "
   2657 			   "PSK format");
   2658 		wps->use_psk_key = 1;
   2659 	}
   2660 #endif /* WPS_WORKAROUNDS */
   2661 
   2662 	wps->state = SEND_M2;
   2663 	return WPS_CONTINUE;
   2664 }
   2665 
   2666 
   2667 static enum wps_process_res wps_process_m3(struct wps_data *wps,
   2668 					   const struct wpabuf *msg,
   2669 					   struct wps_parse_attr *attr)
   2670 {
   2671 	wpa_printf(MSG_DEBUG, "WPS: Received M3");
   2672 
   2673 	if (wps->state != RECV_M3) {
   2674 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
   2675 			   "receiving M3", wps->state);
   2676 		wps->state = SEND_WSC_NACK;
   2677 		return WPS_CONTINUE;
   2678 	}
   2679 
   2680 	if (wps->pbc && wps->wps->registrar->force_pbc_overlap &&
   2681 	    !wps_registrar_skip_overlap(wps)) {
   2682 		wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
   2683 			   "session overlap");
   2684 		wps->state = SEND_WSC_NACK;
   2685 		wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
   2686 		return WPS_CONTINUE;
   2687 	}
   2688 
   2689 	if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
   2690 	    wps_process_authenticator(wps, attr->authenticator, msg) ||
   2691 	    wps_process_e_hash1(wps, attr->e_hash1) ||
   2692 	    wps_process_e_hash2(wps, attr->e_hash2)) {
   2693 		wps->state = SEND_WSC_NACK;
   2694 		return WPS_CONTINUE;
   2695 	}
   2696 
   2697 	wps->state = SEND_M4;
   2698 	return WPS_CONTINUE;
   2699 }
   2700 
   2701 
   2702 static enum wps_process_res wps_process_m5(struct wps_data *wps,
   2703 					   const struct wpabuf *msg,
   2704 					   struct wps_parse_attr *attr)
   2705 {
   2706 	struct wpabuf *decrypted;
   2707 	struct wps_parse_attr eattr;
   2708 
   2709 	wpa_printf(MSG_DEBUG, "WPS: Received M5");
   2710 
   2711 	if (wps->state != RECV_M5) {
   2712 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
   2713 			   "receiving M5", wps->state);
   2714 		wps->state = SEND_WSC_NACK;
   2715 		return WPS_CONTINUE;
   2716 	}
   2717 
   2718 	if (wps->pbc && wps->wps->registrar->force_pbc_overlap &&
   2719 	    !wps_registrar_skip_overlap(wps)) {
   2720 		wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
   2721 			   "session overlap");
   2722 		wps->state = SEND_WSC_NACK;
   2723 		wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
   2724 		return WPS_CONTINUE;
   2725 	}
   2726 
   2727 	if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
   2728 	    wps_process_authenticator(wps, attr->authenticator, msg)) {
   2729 		wps->state = SEND_WSC_NACK;
   2730 		return WPS_CONTINUE;
   2731 	}
   2732 
   2733 	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
   2734 					      attr->encr_settings_len);
   2735 	if (decrypted == NULL) {
   2736 		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
   2737 			   "Settings attribute");
   2738 		wps->state = SEND_WSC_NACK;
   2739 		return WPS_CONTINUE;
   2740 	}
   2741 
   2742 	if (wps_validate_m5_encr(decrypted, attr->version2 != NULL) < 0) {
   2743 		wpabuf_free(decrypted);
   2744 		wps->state = SEND_WSC_NACK;
   2745 		return WPS_CONTINUE;
   2746 	}
   2747 
   2748 	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
   2749 		   "attribute");
   2750 	if (wps_parse_msg(decrypted, &eattr) < 0 ||
   2751 	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
   2752 	    wps_process_e_snonce1(wps, eattr.e_snonce1)) {
   2753 		wpabuf_free(decrypted);
   2754 		wps->state = SEND_WSC_NACK;
   2755 		return WPS_CONTINUE;
   2756 	}
   2757 	wpabuf_free(decrypted);
   2758 
   2759 	wps->state = SEND_M6;
   2760 	return WPS_CONTINUE;
   2761 }
   2762 
   2763 
   2764 static void wps_sta_cred_cb(struct wps_data *wps)
   2765 {
   2766 	/*
   2767 	 * Update credential to only include a single authentication and
   2768 	 * encryption type in case the AP configuration includes more than one
   2769 	 * option.
   2770 	 */
   2771 	if (wps->cred.auth_type & WPS_AUTH_WPA2PSK)
   2772 		wps->cred.auth_type = WPS_AUTH_WPA2PSK;
   2773 	else if (wps->cred.auth_type & WPS_AUTH_WPAPSK)
   2774 		wps->cred.auth_type = WPS_AUTH_WPAPSK;
   2775 	if (wps->cred.encr_type & WPS_ENCR_AES)
   2776 		wps->cred.encr_type = WPS_ENCR_AES;
   2777 	else if (wps->cred.encr_type & WPS_ENCR_TKIP)
   2778 		wps->cred.encr_type = WPS_ENCR_TKIP;
   2779 	wpa_printf(MSG_DEBUG, "WPS: Update local configuration based on the "
   2780 		   "AP configuration");
   2781 	if (wps->wps->cred_cb)
   2782 		wps->wps->cred_cb(wps->wps->cb_ctx, &wps->cred);
   2783 }
   2784 
   2785 
   2786 static void wps_cred_update(struct wps_credential *dst,
   2787 			    struct wps_credential *src)
   2788 {
   2789 	os_memcpy(dst->ssid, src->ssid, sizeof(dst->ssid));
   2790 	dst->ssid_len = src->ssid_len;
   2791 	dst->auth_type = src->auth_type;
   2792 	dst->encr_type = src->encr_type;
   2793 	dst->key_idx = src->key_idx;
   2794 	os_memcpy(dst->key, src->key, sizeof(dst->key));
   2795 	dst->key_len = src->key_len;
   2796 }
   2797 
   2798 
   2799 static int wps_process_ap_settings_r(struct wps_data *wps,
   2800 				     struct wps_parse_attr *attr)
   2801 {
   2802 	struct wpabuf *msg;
   2803 
   2804 	if (wps->wps->ap || wps->er)
   2805 		return 0;
   2806 
   2807 	/* AP Settings Attributes in M7 when Enrollee is an AP */
   2808 	if (wps_process_ap_settings(attr, &wps->cred) < 0)
   2809 		return -1;
   2810 
   2811 	wpa_printf(MSG_INFO, "WPS: Received old AP configuration from AP");
   2812 
   2813 	if (wps->new_ap_settings) {
   2814 		wpa_printf(MSG_INFO, "WPS: Update AP configuration based on "
   2815 			   "new settings");
   2816 		wps_cred_update(&wps->cred, wps->new_ap_settings);
   2817 		return 0;
   2818 	} else {
   2819 		/*
   2820 		 * Use the AP PIN only to receive the current AP settings, not
   2821 		 * to reconfigure the AP.
   2822 		 */
   2823 
   2824 		/*
   2825 		 * Clear selected registrar here since we do not get to
   2826 		 * WSC_Done in this protocol run.
   2827 		 */
   2828 		wps_registrar_pin_completed(wps->wps->registrar);
   2829 
   2830 		msg = wps_build_ap_cred(wps);
   2831 		if (msg == NULL)
   2832 			return -1;
   2833 		wps->cred.cred_attr = wpabuf_head(msg);
   2834 		wps->cred.cred_attr_len = wpabuf_len(msg);
   2835 
   2836 		if (wps->ap_settings_cb) {
   2837 			wps->ap_settings_cb(wps->ap_settings_cb_ctx,
   2838 					    &wps->cred);
   2839 			wpabuf_free(msg);
   2840 			return 1;
   2841 		}
   2842 		wps_sta_cred_cb(wps);
   2843 
   2844 		wps->cred.cred_attr = NULL;
   2845 		wps->cred.cred_attr_len = 0;
   2846 		wpabuf_free(msg);
   2847 
   2848 		return 1;
   2849 	}
   2850 }
   2851 
   2852 
   2853 static enum wps_process_res wps_process_m7(struct wps_data *wps,
   2854 					   const struct wpabuf *msg,
   2855 					   struct wps_parse_attr *attr)
   2856 {
   2857 	struct wpabuf *decrypted;
   2858 	struct wps_parse_attr eattr;
   2859 
   2860 	wpa_printf(MSG_DEBUG, "WPS: Received M7");
   2861 
   2862 	if (wps->state != RECV_M7) {
   2863 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
   2864 			   "receiving M7", wps->state);
   2865 		wps->state = SEND_WSC_NACK;
   2866 		return WPS_CONTINUE;
   2867 	}
   2868 
   2869 	if (wps->pbc && wps->wps->registrar->force_pbc_overlap &&
   2870 	    !wps_registrar_skip_overlap(wps)) {
   2871 		wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
   2872 			   "session overlap");
   2873 		wps->state = SEND_WSC_NACK;
   2874 		wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
   2875 		return WPS_CONTINUE;
   2876 	}
   2877 
   2878 	if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
   2879 	    wps_process_authenticator(wps, attr->authenticator, msg)) {
   2880 		wps->state = SEND_WSC_NACK;
   2881 		return WPS_CONTINUE;
   2882 	}
   2883 
   2884 	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
   2885 					      attr->encr_settings_len);
   2886 	if (decrypted == NULL) {
   2887 		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypt Encrypted "
   2888 			   "Settings attribute");
   2889 		wps->state = SEND_WSC_NACK;
   2890 		return WPS_CONTINUE;
   2891 	}
   2892 
   2893 	if (wps_validate_m7_encr(decrypted, wps->wps->ap || wps->er,
   2894 				 attr->version2 != NULL) < 0) {
   2895 		wpabuf_free(decrypted);
   2896 		wps->state = SEND_WSC_NACK;
   2897 		return WPS_CONTINUE;
   2898 	}
   2899 
   2900 	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
   2901 		   "attribute");
   2902 	if (wps_parse_msg(decrypted, &eattr) < 0 ||
   2903 	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
   2904 	    wps_process_e_snonce2(wps, eattr.e_snonce2) ||
   2905 	    wps_process_ap_settings_r(wps, &eattr)) {
   2906 		wpabuf_free(decrypted);
   2907 		wps->state = SEND_WSC_NACK;
   2908 		return WPS_CONTINUE;
   2909 	}
   2910 
   2911 	wpabuf_free(decrypted);
   2912 
   2913 	wps->state = SEND_M8;
   2914 	return WPS_CONTINUE;
   2915 }
   2916 
   2917 
   2918 static enum wps_process_res wps_process_wsc_msg(struct wps_data *wps,
   2919 						const struct wpabuf *msg)
   2920 {
   2921 	struct wps_parse_attr attr;
   2922 	enum wps_process_res ret = WPS_CONTINUE;
   2923 
   2924 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_MSG");
   2925 
   2926 	if (wps_parse_msg(msg, &attr) < 0)
   2927 		return WPS_FAILURE;
   2928 
   2929 	if (attr.msg_type == NULL) {
   2930 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
   2931 		wps->state = SEND_WSC_NACK;
   2932 		return WPS_CONTINUE;
   2933 	}
   2934 
   2935 	if (*attr.msg_type != WPS_M1 &&
   2936 	    (attr.registrar_nonce == NULL ||
   2937 	     os_memcmp(wps->nonce_r, attr.registrar_nonce,
   2938 		       WPS_NONCE_LEN) != 0)) {
   2939 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
   2940 		return WPS_FAILURE;
   2941 	}
   2942 
   2943 	switch (*attr.msg_type) {
   2944 	case WPS_M1:
   2945 		if (wps_validate_m1(msg) < 0)
   2946 			return WPS_FAILURE;
   2947 #ifdef CONFIG_WPS_UPNP
   2948 		if (wps->wps->wps_upnp && attr.mac_addr) {
   2949 			/* Remove old pending messages when starting new run */
   2950 			wps_free_pending_msgs(wps->wps->upnp_msgs);
   2951 			wps->wps->upnp_msgs = NULL;
   2952 
   2953 			upnp_wps_device_send_wlan_event(
   2954 				wps->wps->wps_upnp, attr.mac_addr,
   2955 				UPNP_WPS_WLANEVENT_TYPE_EAP, msg);
   2956 		}
   2957 #endif /* CONFIG_WPS_UPNP */
   2958 		ret = wps_process_m1(wps, &attr);
   2959 		break;
   2960 	case WPS_M3:
   2961 		if (wps_validate_m3(msg) < 0)
   2962 			return WPS_FAILURE;
   2963 		ret = wps_process_m3(wps, msg, &attr);
   2964 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
   2965 			wps_fail_event(wps->wps, WPS_M3, wps->config_error,
   2966 				       wps->error_indication, wps->mac_addr_e);
   2967 		break;
   2968 	case WPS_M5:
   2969 		if (wps_validate_m5(msg) < 0)
   2970 			return WPS_FAILURE;
   2971 		ret = wps_process_m5(wps, msg, &attr);
   2972 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
   2973 			wps_fail_event(wps->wps, WPS_M5, wps->config_error,
   2974 				       wps->error_indication, wps->mac_addr_e);
   2975 		break;
   2976 	case WPS_M7:
   2977 		if (wps_validate_m7(msg) < 0)
   2978 			return WPS_FAILURE;
   2979 		ret = wps_process_m7(wps, msg, &attr);
   2980 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
   2981 			wps_fail_event(wps->wps, WPS_M7, wps->config_error,
   2982 				       wps->error_indication, wps->mac_addr_e);
   2983 		break;
   2984 	default:
   2985 		wpa_printf(MSG_DEBUG, "WPS: Unsupported Message Type %d",
   2986 			   *attr.msg_type);
   2987 		return WPS_FAILURE;
   2988 	}
   2989 
   2990 	if (ret == WPS_CONTINUE) {
   2991 		/* Save a copy of the last message for Authenticator derivation
   2992 		 */
   2993 		wpabuf_free(wps->last_msg);
   2994 		wps->last_msg = wpabuf_dup(msg);
   2995 	}
   2996 
   2997 	return ret;
   2998 }
   2999 
   3000 
   3001 static enum wps_process_res wps_process_wsc_ack(struct wps_data *wps,
   3002 						const struct wpabuf *msg)
   3003 {
   3004 	struct wps_parse_attr attr;
   3005 
   3006 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_ACK");
   3007 
   3008 	if (wps_parse_msg(msg, &attr) < 0)
   3009 		return WPS_FAILURE;
   3010 
   3011 	if (attr.msg_type == NULL) {
   3012 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
   3013 		return WPS_FAILURE;
   3014 	}
   3015 
   3016 	if (*attr.msg_type != WPS_WSC_ACK) {
   3017 		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
   3018 			   *attr.msg_type);
   3019 		return WPS_FAILURE;
   3020 	}
   3021 
   3022 #ifdef CONFIG_WPS_UPNP
   3023 	if (wps->wps->wps_upnp && wps->ext_reg && wps->state == RECV_M2D_ACK &&
   3024 	    upnp_wps_subscribers(wps->wps->wps_upnp)) {
   3025 		if (wps->wps->upnp_msgs)
   3026 			return WPS_CONTINUE;
   3027 		wpa_printf(MSG_DEBUG, "WPS: Wait for response from an "
   3028 			   "external Registrar");
   3029 		return WPS_PENDING;
   3030 	}
   3031 #endif /* CONFIG_WPS_UPNP */
   3032 
   3033 	if (attr.registrar_nonce == NULL ||
   3034 	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
   3035 	{
   3036 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
   3037 		return WPS_FAILURE;
   3038 	}
   3039 
   3040 	if (attr.enrollee_nonce == NULL ||
   3041 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
   3042 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
   3043 		return WPS_FAILURE;
   3044 	}
   3045 
   3046 	if (wps->state == RECV_M2D_ACK) {
   3047 #ifdef CONFIG_WPS_UPNP
   3048 		if (wps->wps->wps_upnp &&
   3049 		    upnp_wps_subscribers(wps->wps->wps_upnp)) {
   3050 			if (wps->wps->upnp_msgs)
   3051 				return WPS_CONTINUE;
   3052 			if (wps->ext_reg == 0)
   3053 				wps->ext_reg = 1;
   3054 			wpa_printf(MSG_DEBUG, "WPS: Wait for response from an "
   3055 				   "external Registrar");
   3056 			return WPS_PENDING;
   3057 		}
   3058 #endif /* CONFIG_WPS_UPNP */
   3059 
   3060 		wpa_printf(MSG_DEBUG, "WPS: No more registrars available - "
   3061 			   "terminate negotiation");
   3062 	}
   3063 
   3064 	return WPS_FAILURE;
   3065 }
   3066 
   3067 
   3068 static enum wps_process_res wps_process_wsc_nack(struct wps_data *wps,
   3069 						 const struct wpabuf *msg)
   3070 {
   3071 	struct wps_parse_attr attr;
   3072 	int old_state;
   3073 	u16 config_error;
   3074 
   3075 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_NACK");
   3076 
   3077 	old_state = wps->state;
   3078 	wps->state = SEND_WSC_NACK;
   3079 
   3080 	if (wps_parse_msg(msg, &attr) < 0)
   3081 		return WPS_FAILURE;
   3082 
   3083 	if (attr.msg_type == NULL) {
   3084 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
   3085 		return WPS_FAILURE;
   3086 	}
   3087 
   3088 	if (*attr.msg_type != WPS_WSC_NACK) {
   3089 		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
   3090 			   *attr.msg_type);
   3091 		return WPS_FAILURE;
   3092 	}
   3093 
   3094 #ifdef CONFIG_WPS_UPNP
   3095 	if (wps->wps->wps_upnp && wps->ext_reg) {
   3096 		wpa_printf(MSG_DEBUG, "WPS: Negotiation using external "
   3097 			   "Registrar terminated by the Enrollee");
   3098 		return WPS_FAILURE;
   3099 	}
   3100 #endif /* CONFIG_WPS_UPNP */
   3101 
   3102 	if (attr.registrar_nonce == NULL ||
   3103 	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
   3104 	{
   3105 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
   3106 		return WPS_FAILURE;
   3107 	}
   3108 
   3109 	if (attr.enrollee_nonce == NULL ||
   3110 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
   3111 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
   3112 		return WPS_FAILURE;
   3113 	}
   3114 
   3115 	if (attr.config_error == NULL) {
   3116 		wpa_printf(MSG_DEBUG, "WPS: No Configuration Error attribute "
   3117 			   "in WSC_NACK");
   3118 		return WPS_FAILURE;
   3119 	}
   3120 
   3121 	config_error = WPA_GET_BE16(attr.config_error);
   3122 	wpa_printf(MSG_DEBUG, "WPS: Enrollee terminated negotiation with "
   3123 		   "Configuration Error %d", config_error);
   3124 
   3125 	switch (old_state) {
   3126 	case RECV_M3:
   3127 		wps_fail_event(wps->wps, WPS_M2, config_error,
   3128 			       wps->error_indication, wps->mac_addr_e);
   3129 		break;
   3130 	case RECV_M5:
   3131 		wps_fail_event(wps->wps, WPS_M4, config_error,
   3132 			       wps->error_indication, wps->mac_addr_e);
   3133 		break;
   3134 	case RECV_M7:
   3135 		wps_fail_event(wps->wps, WPS_M6, config_error,
   3136 			       wps->error_indication, wps->mac_addr_e);
   3137 		break;
   3138 	case RECV_DONE:
   3139 		wps_fail_event(wps->wps, WPS_M8, config_error,
   3140 			       wps->error_indication, wps->mac_addr_e);
   3141 		break;
   3142 	default:
   3143 		break;
   3144 	}
   3145 
   3146 	return WPS_FAILURE;
   3147 }
   3148 
   3149 
   3150 static enum wps_process_res wps_process_wsc_done(struct wps_data *wps,
   3151 						 const struct wpabuf *msg)
   3152 {
   3153 	struct wps_parse_attr attr;
   3154 
   3155 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_Done");
   3156 
   3157 	if (wps->state != RECV_DONE &&
   3158 	    (!wps->wps->wps_upnp || !wps->ext_reg)) {
   3159 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
   3160 			   "receiving WSC_Done", wps->state);
   3161 		return WPS_FAILURE;
   3162 	}
   3163 
   3164 	if (wps_parse_msg(msg, &attr) < 0)
   3165 		return WPS_FAILURE;
   3166 
   3167 	if (attr.msg_type == NULL) {
   3168 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
   3169 		return WPS_FAILURE;
   3170 	}
   3171 
   3172 	if (*attr.msg_type != WPS_WSC_DONE) {
   3173 		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
   3174 			   *attr.msg_type);
   3175 		return WPS_FAILURE;
   3176 	}
   3177 
   3178 #ifdef CONFIG_WPS_UPNP
   3179 	if (wps->wps->wps_upnp && wps->ext_reg) {
   3180 		wpa_printf(MSG_DEBUG, "WPS: Negotiation using external "
   3181 			   "Registrar completed successfully");
   3182 		wps_device_store(wps->wps->registrar, &wps->peer_dev,
   3183 				 wps->uuid_e);
   3184 		return WPS_DONE;
   3185 	}
   3186 #endif /* CONFIG_WPS_UPNP */
   3187 
   3188 	if (attr.registrar_nonce == NULL ||
   3189 	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
   3190 	{
   3191 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
   3192 		return WPS_FAILURE;
   3193 	}
   3194 
   3195 	if (attr.enrollee_nonce == NULL ||
   3196 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
   3197 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
   3198 		return WPS_FAILURE;
   3199 	}
   3200 
   3201 	wpa_printf(MSG_DEBUG, "WPS: Negotiation completed successfully");
   3202 	wps_device_store(wps->wps->registrar, &wps->peer_dev,
   3203 			 wps->uuid_e);
   3204 
   3205 	if (wps->wps->wps_state == WPS_STATE_NOT_CONFIGURED && wps->new_psk &&
   3206 	    wps->wps->ap && !wps->wps->registrar->disable_auto_conf) {
   3207 		struct wps_credential cred;
   3208 
   3209 		wpa_printf(MSG_DEBUG, "WPS: Moving to Configured state based "
   3210 			   "on first Enrollee connection");
   3211 
   3212 		os_memset(&cred, 0, sizeof(cred));
   3213 		os_memcpy(cred.ssid, wps->wps->ssid, wps->wps->ssid_len);
   3214 		cred.ssid_len = wps->wps->ssid_len;
   3215 		cred.auth_type = WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK;
   3216 		cred.encr_type = WPS_ENCR_TKIP | WPS_ENCR_AES;
   3217 		os_memcpy(cred.key, wps->new_psk, wps->new_psk_len);
   3218 		cred.key_len = wps->new_psk_len;
   3219 
   3220 		wps->wps->wps_state = WPS_STATE_CONFIGURED;
   3221 		wpa_hexdump_ascii_key(MSG_DEBUG,
   3222 				      "WPS: Generated random passphrase",
   3223 				      wps->new_psk, wps->new_psk_len);
   3224 		if (wps->wps->cred_cb)
   3225 			wps->wps->cred_cb(wps->wps->cb_ctx, &cred);
   3226 
   3227 		os_free(wps->new_psk);
   3228 		wps->new_psk = NULL;
   3229 	}
   3230 
   3231 	if (!wps->wps->ap && !wps->er)
   3232 		wps_sta_cred_cb(wps);
   3233 
   3234 	if (wps->new_psk) {
   3235 		if (wps_cb_new_psk(wps->wps->registrar, wps->mac_addr_e,
   3236 				   wps->p2p_dev_addr, wps->new_psk,
   3237 				   wps->new_psk_len)) {
   3238 			wpa_printf(MSG_DEBUG, "WPS: Failed to configure the "
   3239 				   "new PSK");
   3240 		}
   3241 		os_free(wps->new_psk);
   3242 		wps->new_psk = NULL;
   3243 	}
   3244 
   3245 	wps_cb_reg_success(wps->wps->registrar, wps->mac_addr_e, wps->uuid_e,
   3246 			   wps->dev_password, wps->dev_password_len);
   3247 
   3248 	if (wps->pbc) {
   3249 		wps_registrar_remove_pbc_session(wps->wps->registrar,
   3250 						 wps->uuid_e,
   3251 						 wps->p2p_dev_addr);
   3252 		wps_registrar_pbc_completed(wps->wps->registrar);
   3253 #ifdef WPS_WORKAROUNDS
   3254 		os_get_reltime(&wps->wps->registrar->pbc_ignore_start);
   3255 #endif /* WPS_WORKAROUNDS */
   3256 		os_memcpy(wps->wps->registrar->pbc_ignore_uuid, wps->uuid_e,
   3257 			  WPS_UUID_LEN);
   3258 	} else {
   3259 		wps_registrar_pin_completed(wps->wps->registrar);
   3260 	}
   3261 	/* TODO: maintain AuthorizedMACs somewhere separately for each ER and
   3262 	 * merge them into APs own list.. */
   3263 
   3264 	wps_success_event(wps->wps, wps->mac_addr_e);
   3265 
   3266 	return WPS_DONE;
   3267 }
   3268 
   3269 
   3270 enum wps_process_res wps_registrar_process_msg(struct wps_data *wps,
   3271 					       enum wsc_op_code op_code,
   3272 					       const struct wpabuf *msg)
   3273 {
   3274 	enum wps_process_res ret;
   3275 
   3276 	wpa_printf(MSG_DEBUG, "WPS: Processing received message (len=%lu "
   3277 		   "op_code=%d)",
   3278 		   (unsigned long) wpabuf_len(msg), op_code);
   3279 
   3280 #ifdef CONFIG_WPS_UPNP
   3281 	if (wps->wps->wps_upnp && op_code == WSC_MSG && wps->ext_reg == 1) {
   3282 		struct wps_parse_attr attr;
   3283 		if (wps_parse_msg(msg, &attr) == 0 && attr.msg_type &&
   3284 		    *attr.msg_type == WPS_M3)
   3285 			wps->ext_reg = 2; /* past M2/M2D phase */
   3286 	}
   3287 	if (wps->ext_reg > 1)
   3288 		wps_registrar_free_pending_m2(wps->wps);
   3289 	if (wps->wps->wps_upnp && wps->ext_reg &&
   3290 	    wps->wps->upnp_msgs == NULL &&
   3291 	    (op_code == WSC_MSG || op_code == WSC_Done || op_code == WSC_NACK))
   3292 	{
   3293 		struct wps_parse_attr attr;
   3294 		int type;
   3295 		if (wps_parse_msg(msg, &attr) < 0 || attr.msg_type == NULL)
   3296 			type = -1;
   3297 		else
   3298 			type = *attr.msg_type;
   3299 		wpa_printf(MSG_DEBUG, "WPS: Sending received message (type %d)"
   3300 			   " to external Registrar for processing", type);
   3301 		upnp_wps_device_send_wlan_event(wps->wps->wps_upnp,
   3302 						wps->mac_addr_e,
   3303 						UPNP_WPS_WLANEVENT_TYPE_EAP,
   3304 						msg);
   3305 		if (op_code == WSC_MSG)
   3306 			return WPS_PENDING;
   3307 	} else if (wps->wps->wps_upnp && wps->ext_reg && op_code == WSC_MSG) {
   3308 		wpa_printf(MSG_DEBUG, "WPS: Skip internal processing - using "
   3309 			   "external Registrar");
   3310 		return WPS_CONTINUE;
   3311 	}
   3312 #endif /* CONFIG_WPS_UPNP */
   3313 
   3314 	switch (op_code) {
   3315 	case WSC_MSG:
   3316 		return wps_process_wsc_msg(wps, msg);
   3317 	case WSC_ACK:
   3318 		if (wps_validate_wsc_ack(msg) < 0)
   3319 			return WPS_FAILURE;
   3320 		return wps_process_wsc_ack(wps, msg);
   3321 	case WSC_NACK:
   3322 		if (wps_validate_wsc_nack(msg) < 0)
   3323 			return WPS_FAILURE;
   3324 		return wps_process_wsc_nack(wps, msg);
   3325 	case WSC_Done:
   3326 		if (wps_validate_wsc_done(msg) < 0)
   3327 			return WPS_FAILURE;
   3328 		ret = wps_process_wsc_done(wps, msg);
   3329 		if (ret == WPS_FAILURE) {
   3330 			wps->state = SEND_WSC_NACK;
   3331 			wps_fail_event(wps->wps, WPS_WSC_DONE,
   3332 				       wps->config_error,
   3333 				       wps->error_indication, wps->mac_addr_e);
   3334 		}
   3335 		return ret;
   3336 	default:
   3337 		wpa_printf(MSG_DEBUG, "WPS: Unsupported op_code %d", op_code);
   3338 		return WPS_FAILURE;
   3339 	}
   3340 }
   3341 
   3342 
   3343 int wps_registrar_update_ie(struct wps_registrar *reg)
   3344 {
   3345 	return wps_set_ie(reg);
   3346 }
   3347 
   3348 
   3349 static void wps_registrar_set_selected_timeout(void *eloop_ctx,
   3350 					       void *timeout_ctx)
   3351 {
   3352 	struct wps_registrar *reg = eloop_ctx;
   3353 
   3354 	wpa_printf(MSG_DEBUG, "WPS: Selected Registrar timeout - "
   3355 		   "unselect internal Registrar");
   3356 	reg->selected_registrar = 0;
   3357 	reg->pbc = 0;
   3358 	wps_registrar_selected_registrar_changed(reg, 0);
   3359 }
   3360 
   3361 
   3362 #ifdef CONFIG_WPS_UPNP
   3363 static void wps_registrar_sel_reg_add(struct wps_registrar *reg,
   3364 				      struct subscription *s)
   3365 {
   3366 	int i, j;
   3367 	wpa_printf(MSG_DEBUG, "WPS: External Registrar selected (dev_pw_id=%d "
   3368 		   "config_methods=0x%x)",
   3369 		   s->dev_password_id, s->config_methods);
   3370 	reg->sel_reg_union = 1;
   3371 	if (reg->sel_reg_dev_password_id_override != DEV_PW_PUSHBUTTON)
   3372 		reg->sel_reg_dev_password_id_override = s->dev_password_id;
   3373 	if (reg->sel_reg_config_methods_override == -1)
   3374 		reg->sel_reg_config_methods_override = 0;
   3375 	reg->sel_reg_config_methods_override |= s->config_methods;
   3376 	for (i = 0; i < WPS_MAX_AUTHORIZED_MACS; i++)
   3377 		if (is_zero_ether_addr(reg->authorized_macs_union[i]))
   3378 			break;
   3379 	for (j = 0; i < WPS_MAX_AUTHORIZED_MACS && j < WPS_MAX_AUTHORIZED_MACS;
   3380 	     j++) {
   3381 		if (is_zero_ether_addr(s->authorized_macs[j]))
   3382 			break;
   3383 		wpa_printf(MSG_DEBUG, "WPS: Add authorized MAC into union: "
   3384 			   MACSTR, MAC2STR(s->authorized_macs[j]));
   3385 		os_memcpy(reg->authorized_macs_union[i],
   3386 			  s->authorized_macs[j], ETH_ALEN);
   3387 		i++;
   3388 	}
   3389 	wpa_hexdump(MSG_DEBUG, "WPS: Authorized MACs union",
   3390 		    (u8 *) reg->authorized_macs_union,
   3391 		    sizeof(reg->authorized_macs_union));
   3392 }
   3393 #endif /* CONFIG_WPS_UPNP */
   3394 
   3395 
   3396 static void wps_registrar_sel_reg_union(struct wps_registrar *reg)
   3397 {
   3398 #ifdef CONFIG_WPS_UPNP
   3399 	struct subscription *s;
   3400 
   3401 	if (reg->wps->wps_upnp == NULL)
   3402 		return;
   3403 
   3404 	dl_list_for_each(s, &reg->wps->wps_upnp->subscriptions,
   3405 			 struct subscription, list) {
   3406 		struct subscr_addr *sa;
   3407 		sa = dl_list_first(&s->addr_list, struct subscr_addr, list);
   3408 		if (sa) {
   3409 			wpa_printf(MSG_DEBUG, "WPS: External Registrar %s:%d",
   3410 				   inet_ntoa(sa->saddr.sin_addr),
   3411 				   ntohs(sa->saddr.sin_port));
   3412 		}
   3413 		if (s->selected_registrar)
   3414 			wps_registrar_sel_reg_add(reg, s);
   3415 		else
   3416 			wpa_printf(MSG_DEBUG, "WPS: External Registrar not "
   3417 				   "selected");
   3418 	}
   3419 #endif /* CONFIG_WPS_UPNP */
   3420 }
   3421 
   3422 
   3423 /**
   3424  * wps_registrar_selected_registrar_changed - SetSelectedRegistrar change
   3425  * @reg: Registrar data from wps_registrar_init()
   3426  *
   3427  * This function is called when selected registrar state changes, e.g., when an
   3428  * AP receives a SetSelectedRegistrar UPnP message.
   3429  */
   3430 void wps_registrar_selected_registrar_changed(struct wps_registrar *reg,
   3431 					      u16 dev_pw_id)
   3432 {
   3433 	wpa_printf(MSG_DEBUG, "WPS: Selected registrar information changed");
   3434 
   3435 	reg->sel_reg_union = reg->selected_registrar;
   3436 	reg->sel_reg_dev_password_id_override = -1;
   3437 	reg->sel_reg_config_methods_override = -1;
   3438 	os_memcpy(reg->authorized_macs_union, reg->authorized_macs,
   3439 		  WPS_MAX_AUTHORIZED_MACS * ETH_ALEN);
   3440 	wpa_hexdump(MSG_DEBUG, "WPS: Authorized MACs union (start with own)",
   3441 		    (u8 *) reg->authorized_macs_union,
   3442 		    sizeof(reg->authorized_macs_union));
   3443 	if (reg->selected_registrar) {
   3444 		u16 methods;
   3445 
   3446 		methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
   3447 		methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
   3448 			     WPS_CONFIG_PHY_PUSHBUTTON);
   3449 		if (reg->pbc) {
   3450 			reg->sel_reg_dev_password_id_override =
   3451 				DEV_PW_PUSHBUTTON;
   3452 			wps_set_pushbutton(&methods, reg->wps->config_methods);
   3453 		} else if (dev_pw_id)
   3454 			reg->sel_reg_dev_password_id_override = dev_pw_id;
   3455 		wpa_printf(MSG_DEBUG, "WPS: Internal Registrar selected "
   3456 			   "(pbc=%d)", reg->pbc);
   3457 		reg->sel_reg_config_methods_override = methods;
   3458 	} else
   3459 		wpa_printf(MSG_DEBUG, "WPS: Internal Registrar not selected");
   3460 
   3461 	wps_registrar_sel_reg_union(reg);
   3462 
   3463 	wps_set_ie(reg);
   3464 	wps_cb_set_sel_reg(reg);
   3465 }
   3466 
   3467 
   3468 int wps_registrar_get_info(struct wps_registrar *reg, const u8 *addr,
   3469 			   char *buf, size_t buflen)
   3470 {
   3471 	struct wps_registrar_device *d;
   3472 	int len = 0, ret;
   3473 	char uuid[40];
   3474 	char devtype[WPS_DEV_TYPE_BUFSIZE];
   3475 
   3476 	d = wps_device_get(reg, addr);
   3477 	if (d == NULL)
   3478 		return 0;
   3479 	if (uuid_bin2str(d->uuid, uuid, sizeof(uuid)))
   3480 		return 0;
   3481 
   3482 	ret = os_snprintf(buf + len, buflen - len,
   3483 			  "wpsUuid=%s\n"
   3484 			  "wpsPrimaryDeviceType=%s\n"
   3485 			  "wpsDeviceName=%s\n"
   3486 			  "wpsManufacturer=%s\n"
   3487 			  "wpsModelName=%s\n"
   3488 			  "wpsModelNumber=%s\n"
   3489 			  "wpsSerialNumber=%s\n",
   3490 			  uuid,
   3491 			  wps_dev_type_bin2str(d->dev.pri_dev_type, devtype,
   3492 					       sizeof(devtype)),
   3493 			  d->dev.device_name ? d->dev.device_name : "",
   3494 			  d->dev.manufacturer ? d->dev.manufacturer : "",
   3495 			  d->dev.model_name ? d->dev.model_name : "",
   3496 			  d->dev.model_number ? d->dev.model_number : "",
   3497 			  d->dev.serial_number ? d->dev.serial_number : "");
   3498 	if (ret < 0 || (size_t) ret >= buflen - len)
   3499 		return len;
   3500 	len += ret;
   3501 
   3502 	return len;
   3503 }
   3504 
   3505 
   3506 int wps_registrar_config_ap(struct wps_registrar *reg,
   3507 			    struct wps_credential *cred)
   3508 {
   3509 	wpa_printf(MSG_DEBUG, "WPS: encr_type=0x%x", cred->encr_type);
   3510 	if (!(cred->encr_type & (WPS_ENCR_NONE | WPS_ENCR_TKIP |
   3511 				 WPS_ENCR_AES))) {
   3512 		if (cred->encr_type & WPS_ENCR_WEP) {
   3513 			wpa_printf(MSG_INFO, "WPS: Reject new AP settings "
   3514 				   "due to WEP configuration");
   3515 			return -1;
   3516 		}
   3517 
   3518 		wpa_printf(MSG_INFO, "WPS: Reject new AP settings due to "
   3519 			   "invalid encr_type 0x%x", cred->encr_type);
   3520 		return -1;
   3521 	}
   3522 
   3523 	if ((cred->encr_type & (WPS_ENCR_TKIP | WPS_ENCR_AES)) ==
   3524 	    WPS_ENCR_TKIP) {
   3525 		wpa_printf(MSG_DEBUG, "WPS: Upgrade encr_type TKIP -> "
   3526 			   "TKIP+AES");
   3527 		cred->encr_type |= WPS_ENCR_AES;
   3528 	}
   3529 
   3530 	if ((cred->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) ==
   3531 	    WPS_AUTH_WPAPSK) {
   3532 		wpa_printf(MSG_DEBUG, "WPS: Upgrade auth_type WPAPSK -> "
   3533 			   "WPAPSK+WPA2PSK");
   3534 		cred->auth_type |= WPS_AUTH_WPA2PSK;
   3535 	}
   3536 
   3537 	if (reg->wps->cred_cb)
   3538 		return reg->wps->cred_cb(reg->wps->cb_ctx, cred);
   3539 
   3540 	return -1;
   3541 }
   3542 
   3543 
   3544 #ifdef CONFIG_WPS_NFC
   3545 
   3546 int wps_registrar_add_nfc_pw_token(struct wps_registrar *reg,
   3547 				   const u8 *pubkey_hash, u16 pw_id,
   3548 				   const u8 *dev_pw, size_t dev_pw_len,
   3549 				   int pk_hash_provided_oob)
   3550 {
   3551 	struct wps_nfc_pw_token *token;
   3552 
   3553 	if (dev_pw_len > WPS_OOB_DEVICE_PASSWORD_LEN)
   3554 		return -1;
   3555 
   3556 	if (pw_id == DEV_PW_NFC_CONNECTION_HANDOVER &&
   3557 	    (pubkey_hash == NULL || !pk_hash_provided_oob)) {
   3558 		wpa_printf(MSG_DEBUG, "WPS: Unexpected NFC Password Token "
   3559 			   "addition - missing public key hash");
   3560 		return -1;
   3561 	}
   3562 
   3563 	wps_free_nfc_pw_tokens(&reg->nfc_pw_tokens, pw_id);
   3564 
   3565 	token = os_zalloc(sizeof(*token));
   3566 	if (token == NULL)
   3567 		return -1;
   3568 
   3569 	token->peer_pk_hash_known = pubkey_hash != NULL;
   3570 	if (pubkey_hash)
   3571 		os_memcpy(token->pubkey_hash, pubkey_hash,
   3572 			  WPS_OOB_PUBKEY_HASH_LEN);
   3573 	token->pw_id = pw_id;
   3574 	token->pk_hash_provided_oob = pk_hash_provided_oob;
   3575 	if (dev_pw) {
   3576 		wpa_snprintf_hex_uppercase((char *) token->dev_pw,
   3577 					   sizeof(token->dev_pw),
   3578 					   dev_pw, dev_pw_len);
   3579 		token->dev_pw_len = dev_pw_len * 2;
   3580 	}
   3581 
   3582 	dl_list_add(&reg->nfc_pw_tokens, &token->list);
   3583 
   3584 	reg->selected_registrar = 1;
   3585 	reg->pbc = 0;
   3586 	wps_registrar_add_authorized_mac(reg,
   3587 					 (u8 *) "\xff\xff\xff\xff\xff\xff");
   3588 	wps_registrar_selected_registrar_changed(reg, pw_id);
   3589 	eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
   3590 	eloop_register_timeout(WPS_PBC_WALK_TIME, 0,
   3591 			       wps_registrar_set_selected_timeout,
   3592 			       reg, NULL);
   3593 
   3594 	wpa_printf(MSG_DEBUG, "WPS: Added NFC Device Password %u to Registrar",
   3595 		   pw_id);
   3596 
   3597 	return 0;
   3598 }
   3599 
   3600 
   3601 int wps_registrar_add_nfc_password_token(struct wps_registrar *reg,
   3602 					 const u8 *oob_dev_pw,
   3603 					 size_t oob_dev_pw_len)
   3604 {
   3605 	const u8 *pos, *hash, *dev_pw;
   3606 	u16 id;
   3607 	size_t dev_pw_len;
   3608 
   3609 	if (oob_dev_pw_len < WPS_OOB_PUBKEY_HASH_LEN + 2 ||
   3610 	    oob_dev_pw_len > WPS_OOB_PUBKEY_HASH_LEN + 2 +
   3611 	    WPS_OOB_DEVICE_PASSWORD_LEN)
   3612 		return -1;
   3613 
   3614 	hash = oob_dev_pw;
   3615 	pos = oob_dev_pw + WPS_OOB_PUBKEY_HASH_LEN;
   3616 	id = WPA_GET_BE16(pos);
   3617 	dev_pw = pos + 2;
   3618 	dev_pw_len = oob_dev_pw + oob_dev_pw_len - dev_pw;
   3619 
   3620 	wpa_printf(MSG_DEBUG, "WPS: Add NFC Password Token for Password ID %u",
   3621 		   id);
   3622 
   3623 	wpa_hexdump(MSG_DEBUG, "WPS: Public Key Hash",
   3624 		    hash, WPS_OOB_PUBKEY_HASH_LEN);
   3625 	wpa_hexdump_key(MSG_DEBUG, "WPS: Device Password", dev_pw, dev_pw_len);
   3626 
   3627 	return wps_registrar_add_nfc_pw_token(reg, hash, id, dev_pw,
   3628 					      dev_pw_len, 0);
   3629 }
   3630 
   3631 
   3632 void wps_registrar_remove_nfc_pw_token(struct wps_registrar *reg,
   3633 				       struct wps_nfc_pw_token *token)
   3634 {
   3635 	wps_registrar_remove_authorized_mac(reg,
   3636 					    (u8 *) "\xff\xff\xff\xff\xff\xff");
   3637 	wps_registrar_selected_registrar_changed(reg, 0);
   3638 
   3639 	/*
   3640 	 * Free the NFC password token if it was used only for a single protocol
   3641 	 * run. The static handover case uses the same password token multiple
   3642 	 * times, so do not free that case here.
   3643 	 */
   3644 	if (token->peer_pk_hash_known)
   3645 		os_free(token);
   3646 }
   3647 
   3648 #endif /* CONFIG_WPS_NFC */
   3649