wps_enrollee.c revision d5e4923d04122f81300fa68fb07d64ede28fd44d
18d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt/*
28d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Wi-Fi Protected Setup - Enrollee
38d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Copyright (c) 2008, Jouni Malinen <j@w1.fi>
48d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt *
5c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * This software may be distributed under the terms of the BSD license.
6c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * See README for more details.
78d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */
88d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
98d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "includes.h"
108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "common.h"
128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "crypto/crypto.h"
138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "crypto/sha256.h"
148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "crypto/random.h"
158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "wps_i.h"
168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "wps_dev_attr.h"
178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_mac_addr(struct wps_data *wps, struct wpabuf *msg)
208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * MAC Address");
228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_MAC_ADDR);
238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ETH_ALEN);
248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->mac_addr_e, ETH_ALEN);
258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_wps_state(struct wps_data *wps, struct wpabuf *msg)
308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u8 state;
328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap)
338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		state = wps->wps->wps_state;
348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	else
358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		state = WPS_STATE_NOT_CONFIGURED;
368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * Wi-Fi Protected Setup State (%d)",
378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   state);
388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_WPS_STATE);
398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, 1);
408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_u8(msg, state);
418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_e_hash(struct wps_data *wps, struct wpabuf *msg)
468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u8 *hash;
488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	const u8 *addr[4];
498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	size_t len[4];
508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (random_get_bytes(wps->snonce, 2 * WPS_SECRET_NONCE_LEN) < 0)
528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: E-S1", wps->snonce, WPS_SECRET_NONCE_LEN);
548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: E-S2",
558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    wps->snonce + WPS_SECRET_NONCE_LEN, WPS_SECRET_NONCE_LEN);
568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->dh_pubkey_e == NULL || wps->dh_pubkey_r == NULL) {
588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: DH public keys not available for "
598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "E-Hash derivation");
608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * E-Hash1");
648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_E_HASH1);
658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, SHA256_MAC_LEN);
668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hash = wpabuf_put(msg, SHA256_MAC_LEN);
678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/* E-Hash1 = HMAC_AuthKey(E-S1 || PSK1 || PK_E || PK_R) */
688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[0] = wps->snonce;
698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[0] = WPS_SECRET_NONCE_LEN;
708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[1] = wps->psk1;
718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[1] = WPS_PSK_LEN;
728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[2] = wpabuf_head(wps->dh_pubkey_e);
738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[2] = wpabuf_len(wps->dh_pubkey_e);
748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[3] = wpabuf_head(wps->dh_pubkey_r);
758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[3] = wpabuf_len(wps->dh_pubkey_r);
768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash1", hash, SHA256_MAC_LEN);
788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * E-Hash2");
808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_E_HASH2);
818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, SHA256_MAC_LEN);
828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hash = wpabuf_put(msg, SHA256_MAC_LEN);
838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/* E-Hash2 = HMAC_AuthKey(E-S2 || PSK2 || PK_E || PK_R) */
848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[0] = wps->snonce + WPS_SECRET_NONCE_LEN;
858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[1] = wps->psk2;
868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash2", hash, SHA256_MAC_LEN);
888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_e_snonce1(struct wps_data *wps, struct wpabuf *msg)
948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * E-SNonce1");
968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_E_SNONCE1);
978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->snonce, WPS_SECRET_NONCE_LEN);
998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
1008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
1018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_e_snonce2(struct wps_data *wps, struct wpabuf *msg)
1048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
1058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * E-SNonce2");
1068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_E_SNONCE2);
1078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
1088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->snonce + WPS_SECRET_NONCE_LEN,
1098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			WPS_SECRET_NONCE_LEN);
1108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
1118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
1128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic struct wpabuf * wps_build_m1(struct wps_data *wps)
1158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
1168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg;
1178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u16 config_methods;
1188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (random_get_bytes(wps->nonce_e, WPS_NONCE_LEN) < 0)
1208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
1218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Nonce",
1228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    wps->nonce_e, WPS_NONCE_LEN);
1238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Building Message M1");
1258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	msg = wpabuf_alloc(1000);
1268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (msg == NULL)
1278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
1288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	config_methods = wps->wps->config_methods;
13087fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen	if (wps->wps->ap && !wps->pbc_in_m1 &&
13187fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen	    (wps->dev_password_len != 0 ||
13287fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen	     (config_methods & WPS_CONFIG_DISPLAY))) {
1338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/*
1348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * These are the methods that the AP supports as an Enrollee
1358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * for adding external Registrars, so remove PushButton.
13687fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen		 *
13787fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen		 * As a workaround for Windows 7 mechanism for probing WPS
13887fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen		 * capabilities from M1, leave PushButton option if no PIN
13987fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen		 * method is available or if WPS configuration enables PBC
14087fd279308af3f806848c8f2ab65ef18c6ac4c30Jouni Malinen		 * workaround.
1418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
1428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		config_methods &= ~WPS_CONFIG_PUSHBUTTON;
1438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS2
1448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		config_methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
1458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				    WPS_CONFIG_PHY_PUSHBUTTON);
1468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS2 */
1478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
1488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_build_version(msg) ||
1508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_msg_type(msg, WPS_M1) ||
1518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_uuid_e(msg, wps->uuid_e) ||
1528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_mac_addr(wps, msg) ||
1538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_enrollee_nonce(wps, msg) ||
1548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_public_key(wps, msg) ||
1558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_auth_type_flags(wps, msg) ||
1568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_encr_type_flags(wps, msg) ||
1578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_conn_type_flags(wps, msg) ||
1588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_config_methods(msg, config_methods) ||
1598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_wps_state(wps, msg) ||
1608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_device_attrs(&wps->wps->dev, msg) ||
1618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_rf_bands(&wps->wps->dev, msg) ||
1628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_assoc_state(wps, msg) ||
1638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_dev_password_id(msg, wps->dev_pw_id) ||
1648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_config_error(msg, WPS_CFG_NO_ERROR) ||
1658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_os_version(&wps->wps->dev, msg) ||
16604949598a23f501be6eec21697465fd46a28840aDmitry Shmidt	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
16704949598a23f501be6eec21697465fd46a28840aDmitry Shmidt	    wps_build_vendor_ext_m1(&wps->wps->dev, msg)) {
1688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(msg);
1698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
1708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
1718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = RECV_M2;
1738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
1748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
1758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic struct wpabuf * wps_build_m3(struct wps_data *wps)
1788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
1798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg;
1808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Building Message M3");
1828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->dev_password == NULL) {
1848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Device Password available");
1858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
1868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
1878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps_derive_psk(wps, wps->dev_password, wps->dev_password_len);
1888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	msg = wpabuf_alloc(1000);
1908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (msg == NULL)
1918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
1928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_build_version(msg) ||
1948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_msg_type(msg, WPS_M3) ||
1958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_registrar_nonce(wps, msg) ||
1968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_e_hash(wps, msg) ||
1978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
1988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_authenticator(wps, msg)) {
1998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(msg);
2008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
2018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
2028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = RECV_M4;
2048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
2058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
2068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic struct wpabuf * wps_build_m5(struct wps_data *wps)
2098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
2108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg, *plain;
2118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Building Message M5");
2138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	plain = wpabuf_alloc(200);
2158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (plain == NULL)
2168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
2178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	msg = wpabuf_alloc(1000);
2198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (msg == NULL) {
2208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(plain);
2218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
2228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
2238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_build_version(msg) ||
2258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_msg_type(msg, WPS_M5) ||
2268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_registrar_nonce(wps, msg) ||
2278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_e_snonce1(wps, plain) ||
2288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_key_wrap_auth(wps, plain) ||
2298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_encr_settings(wps, msg, plain) ||
2308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
2318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_authenticator(wps, msg)) {
2328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(plain);
2338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(msg);
2348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
2358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
2368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(plain);
2378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = RECV_M6;
2398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
2408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
2418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_cred_ssid(struct wps_data *wps, struct wpabuf *msg)
2448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
2458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * SSID");
2468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_SSID);
2478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, wps->wps->ssid_len);
2488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->wps->ssid, wps->wps->ssid_len);
2498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
2508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
2518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_cred_auth_type(struct wps_data *wps, struct wpabuf *msg)
2548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
255c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	u16 auth_type = wps->wps->auth_types;
256c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt
257c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	/* Select the best authentication type */
258c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	if (auth_type & WPS_AUTH_WPA2PSK)
259c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		auth_type = WPS_AUTH_WPA2PSK;
260c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	else if (auth_type & WPS_AUTH_WPAPSK)
261c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		auth_type = WPS_AUTH_WPAPSK;
262c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	else if (auth_type & WPS_AUTH_OPEN)
263c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		auth_type = WPS_AUTH_OPEN;
264c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	else if (auth_type & WPS_AUTH_SHARED)
265c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		auth_type = WPS_AUTH_SHARED;
266c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt
267c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * Authentication Type (0x%x)", auth_type);
2688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_AUTH_TYPE);
2698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, 2);
270c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	wpabuf_put_be16(msg, auth_type);
2718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
2728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
2738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_cred_encr_type(struct wps_data *wps, struct wpabuf *msg)
2768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
277c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	u16 encr_type = wps->wps->encr_types;
278c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt
279c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	/* Select the best encryption type */
280c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	if (wps->wps->auth_types & (WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK)) {
281c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		if (encr_type & WPS_ENCR_AES)
282c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt			encr_type = WPS_ENCR_AES;
283c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		else if (encr_type & WPS_ENCR_TKIP)
284c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt			encr_type = WPS_ENCR_TKIP;
285c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	} else {
286c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		if (encr_type & WPS_ENCR_WEP)
287c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt			encr_type = WPS_ENCR_WEP;
288c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		else if (encr_type & WPS_ENCR_NONE)
289c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt			encr_type = WPS_ENCR_NONE;
290c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	}
291c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt
292c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * Encryption Type (0x%x)", encr_type);
2938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_ENCR_TYPE);
2948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, 2);
295c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	wpabuf_put_be16(msg, encr_type);
2968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
2978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
2988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
2998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_cred_network_key(struct wps_data *wps, struct wpabuf *msg)
3018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
3028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * Network Key");
3038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_NETWORK_KEY);
3048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, wps->wps->network_key_len);
3058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->wps->network_key, wps->wps->network_key_len);
3068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
3078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
3088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_cred_mac_addr(struct wps_data *wps, struct wpabuf *msg)
3118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
3128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS:  * MAC Address (AP BSSID)");
3138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ATTR_MAC_ADDR);
3148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_be16(msg, ETH_ALEN);
3158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_put_data(msg, wps->wps->dev.mac_addr, ETH_ALEN);
3168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
3178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
3188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_build_ap_settings(struct wps_data *wps, struct wpabuf *plain)
3218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
3228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap_settings) {
3238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS:  * AP Settings (pre-configured)");
3248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_put_data(plain, wps->wps->ap_settings,
3258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				wps->wps->ap_settings_len);
3268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return 0;
3278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
3288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return wps_build_cred_ssid(wps, plain) ||
3308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_build_cred_mac_addr(wps, plain) ||
3318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_build_cred_auth_type(wps, plain) ||
3328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_build_cred_encr_type(wps, plain) ||
3338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_build_cred_network_key(wps, plain);
3348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
3358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic struct wpabuf * wps_build_m7(struct wps_data *wps)
3388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
3398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg, *plain;
3408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Building Message M7");
3428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	plain = wpabuf_alloc(500 + wps->wps->ap_settings_len);
3448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (plain == NULL)
3458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
3468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	msg = wpabuf_alloc(1000 + wps->wps->ap_settings_len);
3488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (msg == NULL) {
3498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(plain);
3508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
3518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
3528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_build_version(msg) ||
3548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_msg_type(msg, WPS_M7) ||
3558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_registrar_nonce(wps, msg) ||
3568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_e_snonce2(wps, plain) ||
3578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    (wps->wps->ap && wps_build_ap_settings(wps, plain)) ||
3588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_key_wrap_auth(wps, plain) ||
3598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_encr_settings(wps, msg, plain) ||
3608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
3618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_authenticator(wps, msg)) {
3628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(plain);
3638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(msg);
3648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
3658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
3668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(plain);
3678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap && wps->wps->registrar) {
3698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/*
3708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * If the Registrar is only learning our current configuration,
3718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * it may not continue protocol run to successful completion.
3728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * Store information here to make sure it remains available.
3738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
3748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_device_store(wps->wps->registrar, &wps->peer_dev,
3758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				 wps->uuid_r);
3768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
3778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = RECV_M8;
3798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
3808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
3818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic struct wpabuf * wps_build_wsc_done(struct wps_data *wps)
3848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
3858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg;
3868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_Done");
3888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	msg = wpabuf_alloc(1000);
3908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (msg == NULL)
3918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
3928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
3938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_build_version(msg) ||
3948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_msg_type(msg, WPS_WSC_DONE) ||
3958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_enrollee_nonce(wps, msg) ||
3968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_registrar_nonce(wps, msg) ||
3978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
3988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(msg);
3998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return NULL;
4008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap)
4038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = RECV_ACK;
4048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	else {
4058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_success_event(wps->wps);
4068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = WPS_FINISHED;
4078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
4098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
4108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstruct wpabuf * wps_enrollee_get_msg(struct wps_data *wps,
4138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				     enum wsc_op_code *op_code)
4148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
4158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *msg;
4168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	switch (wps->state) {
4188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case SEND_M1:
4198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_m1(wps);
4208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_MSG;
4218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case SEND_M3:
4238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_m3(wps);
4248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_MSG;
4258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case SEND_M5:
4278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_m5(wps);
4288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_MSG;
4298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case SEND_M7:
4318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_m7(wps);
4328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_MSG;
4338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case RECEIVED_M2D:
4358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps->wps->ap) {
4368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			msg = wps_build_wsc_nack(wps);
4378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			*op_code = WSC_NACK;
4388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			break;
4398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
4408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_wsc_ack(wps);
4418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_ACK;
4428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (msg) {
4438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			/* Another M2/M2D may be received */
4448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps->state = RECV_M2;
4458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
4468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case SEND_WSC_NACK:
4488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_wsc_nack(wps);
4498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_NACK;
4508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_MSG_DONE:
4528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = wps_build_wsc_done(wps);
4538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		*op_code = WSC_Done;
4548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	default:
4568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unsupported state %d for building "
4578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "a message", wps->state);
4588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		msg = NULL;
4598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
4608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (*op_code == WSC_MSG && msg) {
4638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/* Save a copy of the last message for Authenticator derivation
4648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
4658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(wps->last_msg);
4668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->last_msg = wpabuf_dup(msg);
4678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return msg;
4708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
4718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_registrar_nonce(struct wps_data *wps, const u8 *r_nonce)
4748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
4758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (r_nonce == NULL) {
4768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Registrar Nonce received");
4778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
4788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memcpy(wps->nonce_r, r_nonce, WPS_NONCE_LEN);
4818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: Registrar Nonce",
4828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    wps->nonce_r, WPS_NONCE_LEN);
4838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
4858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
4868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_enrollee_nonce(struct wps_data *wps, const u8 *e_nonce)
4898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
4908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (e_nonce == NULL) {
4918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Enrollee Nonce received");
4928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
4938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
4958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (os_memcmp(wps->nonce_e, e_nonce, WPS_NONCE_LEN) != 0) {
4968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Invalid Enrollee Nonce received");
4978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
4988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
4998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
5018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
5028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_uuid_r(struct wps_data *wps, const u8 *uuid_r)
5058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
5068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (uuid_r == NULL) {
5078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No UUID-R received");
5088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memcpy(wps->uuid_r, uuid_r, WPS_UUID_LEN);
5128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: UUID-R", wps->uuid_r, WPS_UUID_LEN);
5138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
5158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
5168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_pubkey(struct wps_data *wps, const u8 *pk,
5198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			      size_t pk_len)
5208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
5218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (pk == NULL || pk_len == 0) {
5228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Public Key received");
5238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(wps->dh_pubkey_r);
5278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->dh_pubkey_r = wpabuf_alloc_copy(pk, pk_len);
5288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->dh_pubkey_r == NULL)
5298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_derive_keys(wps) < 0)
5328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
5358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
5368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_r_hash1(struct wps_data *wps, const u8 *r_hash1)
5398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
5408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (r_hash1 == NULL) {
5418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No R-Hash1 received");
5428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memcpy(wps->peer_hash1, r_hash1, WPS_HASH_LEN);
5468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash1", wps->peer_hash1, WPS_HASH_LEN);
5478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
5498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
5508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_r_hash2(struct wps_data *wps, const u8 *r_hash2)
5538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
5548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (r_hash2 == NULL) {
5558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No R-Hash2 received");
5568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memcpy(wps->peer_hash2, r_hash2, WPS_HASH_LEN);
5608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash2", wps->peer_hash2, WPS_HASH_LEN);
5618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
5638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
5648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_r_snonce1(struct wps_data *wps, const u8 *r_snonce1)
5678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
5688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u8 hash[SHA256_MAC_LEN];
5698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	const u8 *addr[4];
5708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	size_t len[4];
5718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (r_snonce1 == NULL) {
5738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No R-SNonce1 received");
5748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_key(MSG_DEBUG, "WPS: R-SNonce1", r_snonce1,
5788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			WPS_SECRET_NONCE_LEN);
5798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/* R-Hash1 = HMAC_AuthKey(R-S1 || PSK1 || PK_E || PK_R) */
5818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[0] = r_snonce1;
5828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[0] = WPS_SECRET_NONCE_LEN;
5838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[1] = wps->psk1;
5848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[1] = WPS_PSK_LEN;
5858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[2] = wpabuf_head(wps->dh_pubkey_e);
5868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[2] = wpabuf_len(wps->dh_pubkey_e);
5878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[3] = wpabuf_head(wps->dh_pubkey_r);
5888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[3] = wpabuf_len(wps->dh_pubkey_r);
5898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
5908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (os_memcmp(wps->peer_hash1, hash, WPS_HASH_LEN) != 0) {
5928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: R-Hash1 derived from R-S1 does "
5938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "not match with the pre-committed value");
5948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
5958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_pwd_auth_fail_event(wps->wps, 1, 1);
5968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
5978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
5988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
5998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Registrar proved knowledge of the first "
6008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "half of the device password");
6018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
6038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
6048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_r_snonce2(struct wps_data *wps, const u8 *r_snonce2)
6078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
6088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u8 hash[SHA256_MAC_LEN];
6098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	const u8 *addr[4];
6108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	size_t len[4];
6118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (r_snonce2 == NULL) {
6138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No R-SNonce2 received");
6148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
6158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
6168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_key(MSG_DEBUG, "WPS: R-SNonce2", r_snonce2,
6188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			WPS_SECRET_NONCE_LEN);
6198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/* R-Hash2 = HMAC_AuthKey(R-S2 || PSK2 || PK_E || PK_R) */
6218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[0] = r_snonce2;
6228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[0] = WPS_SECRET_NONCE_LEN;
6238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[1] = wps->psk2;
6248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[1] = WPS_PSK_LEN;
6258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[2] = wpabuf_head(wps->dh_pubkey_e);
6268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[2] = wpabuf_len(wps->dh_pubkey_e);
6278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	addr[3] = wpabuf_head(wps->dh_pubkey_r);
6288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	len[3] = wpabuf_len(wps->dh_pubkey_r);
6298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
6308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (os_memcmp(wps->peer_hash2, hash, WPS_HASH_LEN) != 0) {
6328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: R-Hash2 derived from R-S2 does "
6338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "not match with the pre-committed value");
6348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
6358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_pwd_auth_fail_event(wps->wps, 1, 2);
6368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
6378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
6388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Registrar proved knowledge of the second "
6408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "half of the device password");
6418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
6438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
6448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_cred_e(struct wps_data *wps, const u8 *cred,
6478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			      size_t cred_len, int wps2)
6488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
6498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr attr;
6508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf msg;
651d5e4923d04122f81300fa68fb07d64ede28fd44dDmitry Shmidt	int ret = 0;
6528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received Credential");
6548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memset(&wps->cred, 0, sizeof(wps->cred));
6558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_set(&msg, cred, cred_len);
6568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(&msg, &attr) < 0 ||
6578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_cred(&attr, &wps->cred))
6588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
6598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (os_memcmp(wps->cred.mac_addr, wps->wps->dev.mac_addr, ETH_ALEN) !=
6618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    0) {
6628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: MAC Address in the Credential ("
6638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   MACSTR ") does not match with own address (" MACSTR
6648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   ")", MAC2STR(wps->cred.mac_addr),
6658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   MAC2STR(wps->wps->dev.mac_addr));
6668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/*
6678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * In theory, this could be consider fatal error, but there are
6688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * number of deployed implementations using other address here
6698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * due to unclarity in the specification. For interoperability
6708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * reasons, allow this to be processed since we do not really
6718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * use the MAC Address information for anything.
6728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
6738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS_STRICT
6748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps2) {
6758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_INFO, "WPS: Do not accept incorrect "
6768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "MAC Address in AP Settings");
6778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -1;
6788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
6798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS_STRICT */
6808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
6818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS2
6838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (!(wps->cred.encr_type &
6848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	      (WPS_ENCR_NONE | WPS_ENCR_TKIP | WPS_ENCR_AES))) {
6858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps->cred.encr_type & WPS_ENCR_WEP) {
6868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_INFO, "WPS: Reject Credential "
6878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "due to WEP configuration");
6888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps->error_indication = WPS_EI_SECURITY_WEP_PROHIBITED;
6898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -2;
6908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
6918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_INFO, "WPS: Reject Credential due to "
6938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "invalid encr_type 0x%x", wps->cred.encr_type);
6948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
6958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
6968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS2 */
6978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
6988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->cred_cb) {
6998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->cred.cred_attr = cred - 4;
7008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->cred.cred_attr_len = cred_len + 4;
701d5e4923d04122f81300fa68fb07d64ede28fd44dDmitry Shmidt		ret = wps->wps->cred_cb(wps->wps->cb_ctx, &wps->cred);
7028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->cred.cred_attr = NULL;
7038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->cred.cred_attr_len = 0;
7048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
706d5e4923d04122f81300fa68fb07d64ede28fd44dDmitry Shmidt	return ret;
7078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
7088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_creds(struct wps_data *wps, const u8 *cred[],
7118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			     size_t cred_len[], size_t num_cred, int wps2)
7128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
7138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	size_t i;
7148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	int ok = 0;
7158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap)
7178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return 0;
7188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (num_cred == 0) {
7208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Credential attributes "
7218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "received");
7228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
7238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	for (i = 0; i < num_cred; i++) {
7268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		int res;
7278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		res = wps_process_cred_e(wps, cred[i], cred_len[i], wps2);
7288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (res == 0)
7298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			ok++;
7308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		else if (res == -2)
7318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_DEBUG, "WPS: WEP credential skipped");
7328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		else
7338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -1;
7348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (ok == 0) {
7378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No valid Credential attribute "
7388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "received");
7398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
7408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
7438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
7448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic int wps_process_ap_settings_e(struct wps_data *wps,
7478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				     struct wps_parse_attr *attr,
7488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				     struct wpabuf *attrs, int wps2)
7498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
7508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_credential cred;
7518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (!wps->wps->ap)
7538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return 0;
7548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_ap_settings(attr, &cred) < 0)
7568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
7578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_INFO, "WPS: Received new AP configuration from "
7598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "Registrar");
7608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (os_memcmp(cred.mac_addr, wps->wps->dev.mac_addr, ETH_ALEN) !=
7628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    0) {
7638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: MAC Address in the AP Settings ("
7648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   MACSTR ") does not match with own address (" MACSTR
7658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   ")", MAC2STR(cred.mac_addr),
7668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   MAC2STR(wps->wps->dev.mac_addr));
7678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/*
7688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * In theory, this could be consider fatal error, but there are
7698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * number of deployed implementations using other address here
7708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * due to unclarity in the specification. For interoperability
7718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * reasons, allow this to be processed since we do not really
7728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * use the MAC Address information for anything.
7738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
7748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS_STRICT
7758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps2) {
7768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_INFO, "WPS: Do not accept incorrect "
7778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "MAC Address in AP Settings");
7788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -1;
7798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
7808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS_STRICT */
7818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS2
7848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (!(cred.encr_type & (WPS_ENCR_NONE | WPS_ENCR_TKIP | WPS_ENCR_AES)))
7858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	{
7868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (cred.encr_type & WPS_ENCR_WEP) {
7878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_INFO, "WPS: Reject new AP settings "
7888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "due to WEP configuration");
7898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps->error_indication = WPS_EI_SECURITY_WEP_PROHIBITED;
7908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -1;
7918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
7928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_INFO, "WPS: Reject new AP settings due to "
7948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "invalid encr_type 0x%x", cred.encr_type);
7958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
7968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
7978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS2 */
7988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
7998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS_STRICT
8008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps2) {
8018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if ((cred.encr_type & (WPS_ENCR_TKIP | WPS_ENCR_AES)) ==
8028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    WPS_ENCR_TKIP ||
8038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    (cred.auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) ==
8048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		    WPS_AUTH_WPAPSK) {
8058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_INFO, "WPS-STRICT: Invalid WSC 2.0 "
8068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "AP Settings: WPA-Personal/TKIP only");
8078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps->error_indication =
8088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				WPS_EI_SECURITY_TKIP_ONLY_PROHIBITED;
8098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return -1;
8108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
8118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS_STRICT */
8138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef CONFIG_WPS2
8158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if ((cred.encr_type & (WPS_ENCR_TKIP | WPS_ENCR_AES)) == WPS_ENCR_TKIP)
8168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	{
8178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Upgrade encr_type TKIP -> "
8188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "TKIP+AES");
8198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		cred.encr_type |= WPS_ENCR_AES;
8208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if ((cred.auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) ==
8238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    WPS_AUTH_WPAPSK) {
8248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Upgrade auth_type WPAPSK -> "
8258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "WPAPSK+WPA2PSK");
8268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		cred.auth_type |= WPS_AUTH_WPA2PSK;
8278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* CONFIG_WPS2 */
8298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->cred_cb) {
8318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		cred.cred_attr = wpabuf_head(attrs);
8328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		cred.cred_attr_len = wpabuf_len(attrs);
8338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->wps->cred_cb(wps->wps->cb_ctx, &cred);
8348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
8378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
8388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_m2(struct wps_data *wps,
8418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   const struct wpabuf *msg,
8428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   struct wps_parse_attr *attr)
8438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
8448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received M2");
8458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state != RECV_M2) {
8478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
8488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "receiving M2", wps->state);
8498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
8508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
8518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
8548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
8558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_uuid_r(wps, attr->uuid_r)) {
8568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
8578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
8588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/*
8618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * Stop here on an AP as an Enrollee if AP Setup is locked unless the
8628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * special locked mode is used to allow protocol run up to M7 in order
8638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * to support external Registrars that only learn the current AP
8648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * configuration without changing it.
8658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 */
8668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap &&
8678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    ((wps->wps->ap_setup_locked && wps->wps->ap_setup_locked != 2) ||
8688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	     wps->dev_password == NULL)) {
8698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: AP Setup is locked - refuse "
8708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "registration of a new Registrar");
8718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->config_error = WPS_CFG_SETUP_LOCKED;
8728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
8738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
8748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_pubkey(wps, attr->public_key, attr->public_key_len) ||
8778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_authenticator(wps, attr->authenticator, msg) ||
8788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_device_attrs(&wps->peer_dev, attr)) {
8798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
8808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
8818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = SEND_M3;
8848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_CONTINUE;
8858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
8868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_m2d(struct wps_data *wps,
8898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					    struct wps_parse_attr *attr)
8908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
8918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received M2D");
8928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
8938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state != RECV_M2) {
8948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
8958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "receiving M2D", wps->state);
8968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
8978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
8988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
8998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Manufacturer",
9018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			  attr->manufacturer, attr->manufacturer_len);
9028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Name",
9038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			  attr->model_name, attr->model_name_len);
9048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Number",
9058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			  attr->model_number, attr->model_number_len);
9068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Serial Number",
9078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			  attr->serial_number, attr->serial_number_len);
9088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Device Name",
9098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			  attr->dev_name, attr->dev_name_len);
9108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->event_cb) {
9128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		union wps_event_data data;
9138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		struct wps_event_m2d *m2d = &data.m2d;
9148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		os_memset(&data, 0, sizeof(data));
9158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (attr->config_methods)
9168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			m2d->config_methods =
9178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				WPA_GET_BE16(attr->config_methods);
9188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->manufacturer = attr->manufacturer;
9198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->manufacturer_len = attr->manufacturer_len;
9208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->model_name = attr->model_name;
9218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->model_name_len = attr->model_name_len;
9228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->model_number = attr->model_number;
9238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->model_number_len = attr->model_number_len;
9248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->serial_number = attr->serial_number;
9258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->serial_number_len = attr->serial_number_len;
9268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->dev_name = attr->dev_name;
9278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->dev_name_len = attr->dev_name_len;
9288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		m2d->primary_dev_type = attr->primary_dev_type;
9298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (attr->config_error)
9308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			m2d->config_error =
9318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				WPA_GET_BE16(attr->config_error);
9328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (attr->dev_password_id)
9338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			m2d->dev_password_id =
9348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				WPA_GET_BE16(attr->dev_password_id);
9358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->wps->event_cb(wps->wps->cb_ctx, WPS_EV_M2D, &data);
9368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = RECEIVED_M2D;
9398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_CONTINUE;
9408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
9418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_m4(struct wps_data *wps,
9448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   const struct wpabuf *msg,
9458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   struct wps_parse_attr *attr)
9468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
9478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *decrypted;
9488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr eattr;
9498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received M4");
9518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state != RECV_M4) {
9538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
9548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "receiving M4", wps->state);
9558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
9568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
9578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
9608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_authenticator(wps, attr->authenticator, msg) ||
9618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_r_hash1(wps, attr->r_hash1) ||
9628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_r_hash2(wps, attr->r_hash2)) {
9638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
9648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
9658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
9688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					      attr->encr_settings_len);
9698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (decrypted == NULL) {
9708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
9718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "Settings attribute");
9728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
9738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
9748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9761f69aa52ea2e0a73ac502565df8c666ee49cab6aDmitry Shmidt	if (wps_validate_m4_encr(decrypted, attr->version2 != NULL) < 0) {
9778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
9788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
9798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
9808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
9838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "attribute");
9848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(decrypted, &eattr) < 0 ||
9858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
9868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_r_snonce1(wps, eattr.r_snonce1)) {
9878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
9888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
9898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
9908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
9918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(decrypted);
9928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = SEND_M5;
9948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_CONTINUE;
9958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
9968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
9988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_m6(struct wps_data *wps,
9998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   const struct wpabuf *msg,
10008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   struct wps_parse_attr *attr)
10018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
10028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *decrypted;
10038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr eattr;
10048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received M6");
10068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state != RECV_M6) {
10088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
10098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "receiving M6", wps->state);
10108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
10158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_authenticator(wps, attr->authenticator, msg)) {
10168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
10218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					      attr->encr_settings_len);
10228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (decrypted == NULL) {
10238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
10248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "Settings attribute");
10258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10291f69aa52ea2e0a73ac502565df8c666ee49cab6aDmitry Shmidt	if (wps_validate_m6_encr(decrypted, attr->version2 != NULL) < 0) {
10308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
10318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
10368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "attribute");
10378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(decrypted, &eattr) < 0 ||
10388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
10398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_r_snonce2(wps, eattr.r_snonce2)) {
10408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
10418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(decrypted);
10458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
1046c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	if (wps->wps->ap)
1047c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt		wps->wps->event_cb(wps->wps->cb_ctx, WPS_EV_AP_PIN_SUCCESS,
1048c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt				   NULL);
1049c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt
10508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = SEND_M7;
10518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_CONTINUE;
10528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
10538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_m8(struct wps_data *wps,
10568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   const struct wpabuf *msg,
10578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					   struct wps_parse_attr *attr)
10588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
10598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wpabuf *decrypted;
10608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr eattr;
10618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received M8");
10638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state != RECV_M8) {
10658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
10668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "receiving M8", wps->state);
10678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
10728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_authenticator(wps, attr->authenticator, msg)) {
10738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->wps->ap && wps->wps->ap_setup_locked) {
10788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/*
10798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * Stop here if special ap_setup_locked == 2 mode allowed the
10808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * protocol to continue beyond M2. This allows ER to learn the
10818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 * current AP settings without changing them.
10828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
10838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: AP Setup is locked - refuse "
10848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "registration of a new Registrar");
10858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->config_error = WPS_CFG_SETUP_LOCKED;
10868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
10918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					      attr->encr_settings_len);
10928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (decrypted == NULL) {
10938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
10948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "Settings attribute");
10958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
10968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
10978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
10988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
10991f69aa52ea2e0a73ac502565df8c666ee49cab6aDmitry Shmidt	if (wps_validate_m8_encr(decrypted, wps->wps->ap,
11001f69aa52ea2e0a73ac502565df8c666ee49cab6aDmitry Shmidt				 attr->version2 != NULL) < 0) {
11018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
11028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
11038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
11048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
11058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
11078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "attribute");
11088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(decrypted, &eattr) < 0 ||
11098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
11108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_creds(wps, eattr.cred, eattr.cred_len,
11118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			      eattr.num_cred, attr->version2 != NULL) ||
11128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	    wps_process_ap_settings_e(wps, &eattr, decrypted,
11138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				      attr->version2 != NULL)) {
11148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(decrypted);
11158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
11168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
11178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
11188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpabuf_free(decrypted);
11198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = WPS_MSG_DONE;
11218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_CONTINUE;
11228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
11238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_wsc_msg(struct wps_data *wps,
11268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt						const struct wpabuf *msg)
11278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
11288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr attr;
11298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	enum wps_process_res ret = WPS_CONTINUE;
11308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received WSC_MSG");
11328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(msg, &attr) < 0)
11348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
11358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.enrollee_nonce == NULL ||
113761d9df3e62aaa0e87ad05452fcb95142159a17b6Dmitry Shmidt	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
11388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
11398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
11408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
11418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.msg_type == NULL) {
11438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
11448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = SEND_WSC_NACK;
11458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_CONTINUE;
11468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
11478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	switch (*attr.msg_type) {
11498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_M2:
11508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_m2(msg) < 0)
11518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
11528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		ret = wps_process_m2(wps, msg, &attr);
11538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
11548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_M2D:
11558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_m2d(msg) < 0)
11568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
11578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		ret = wps_process_m2d(wps, &attr);
11588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
11598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_M4:
11608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_m4(msg) < 0)
11618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
11628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		ret = wps_process_m4(wps, msg, &attr);
11638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
11648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps_fail_event(wps->wps, WPS_M4, wps->config_error,
11658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				       wps->error_indication);
11668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
11678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_M6:
11688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_m6(msg) < 0)
11698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
11708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		ret = wps_process_m6(wps, msg, &attr);
11718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
11728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps_fail_event(wps->wps, WPS_M6, wps->config_error,
11738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				       wps->error_indication);
11748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
11758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WPS_M8:
11768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_m8(msg) < 0)
11778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
11788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		ret = wps_process_m8(wps, msg, &attr);
11798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
11808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wps_fail_event(wps->wps, WPS_M8, wps->config_error,
11818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				       wps->error_indication);
11828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
11838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	default:
11848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unsupported Message Type %d",
11858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   *attr.msg_type);
11868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
11878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
11888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
11898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/*
11908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * Save a copy of the last message for Authenticator derivation if we
11918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * are continuing. However, skip M2D since it is not authenticated and
11928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * neither is the ACK/NACK response frame. This allows the possibly
11938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * following M2 to be processed correctly by using the previously sent
11948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * M1 in Authenticator derivation.
11958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 */
11968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (ret == WPS_CONTINUE && *attr.msg_type != WPS_M2D) {
11978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/* Save a copy of the last message for Authenticator derivation
11988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		 */
11998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpabuf_free(wps->last_msg);
12008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->last_msg = wpabuf_dup(msg);
12018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return ret;
12048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
12058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_wsc_ack(struct wps_data *wps,
12088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt						const struct wpabuf *msg)
12098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
12108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr attr;
12118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received WSC_ACK");
12138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(msg, &attr) < 0)
12158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.msg_type == NULL) {
12188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
12198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (*attr.msg_type != WPS_WSC_ACK) {
12238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
12248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   *attr.msg_type);
12258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.registrar_nonce == NULL ||
122961d9df3e62aaa0e87ad05452fcb95142159a17b6Dmitry Shmidt	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
12308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	{
12318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
12328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.enrollee_nonce == NULL ||
123661d9df3e62aaa0e87ad05452fcb95142159a17b6Dmitry Shmidt	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
12378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
12388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps->state == RECV_ACK && wps->wps->ap) {
12428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: External Registrar registration "
12438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "completed successfully");
12448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_success_event(wps->wps);
12458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps->state = WPS_FINISHED;
12468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_DONE;
12478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_FAILURE;
12508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
12518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic enum wps_process_res wps_process_wsc_nack(struct wps_data *wps,
12548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt						 const struct wpabuf *msg)
12558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
12568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr attr;
12578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	u16 config_error;
12588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Received WSC_NACK");
12608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(msg, &attr) < 0)
12628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.msg_type == NULL) {
12658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
12668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (*attr.msg_type != WPS_WSC_NACK) {
12708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
12718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   *attr.msg_type);
12728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.registrar_nonce == NULL ||
127661d9df3e62aaa0e87ad05452fcb95142159a17b6Dmitry Shmidt	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
12778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	{
12788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
12798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_hexdump(MSG_DEBUG, "WPS: Received Registrar Nonce",
12808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			    attr.registrar_nonce, WPS_NONCE_LEN);
12818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_hexdump(MSG_DEBUG, "WPS: Expected Registrar Nonce",
12828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			    wps->nonce_r, WPS_NONCE_LEN);
12838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.enrollee_nonce == NULL ||
128761d9df3e62aaa0e87ad05452fcb95142159a17b6Dmitry Shmidt	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
12888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
12898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_hexdump(MSG_DEBUG, "WPS: Received Enrollee Nonce",
12908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			    attr.enrollee_nonce, WPS_NONCE_LEN);
12918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_hexdump(MSG_DEBUG, "WPS: Expected Enrollee Nonce",
12928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			    wps->nonce_e, WPS_NONCE_LEN);
12938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
12948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
12958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
12968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.config_error == NULL) {
12978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: No Configuration Error attribute "
12988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "in WSC_NACK");
12998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
13008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
13018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	config_error = WPA_GET_BE16(attr.config_error);
13038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Registrar terminated negotiation with "
13048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "Configuration Error %d", config_error);
13058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	switch (wps->state) {
13078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case RECV_M4:
13088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_fail_event(wps->wps, WPS_M3, config_error,
13098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			       wps->error_indication);
13108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
13118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case RECV_M6:
13128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_fail_event(wps->wps, WPS_M5, config_error,
13138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			       wps->error_indication);
13148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
13158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case RECV_M8:
13168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wps_fail_event(wps->wps, WPS_M7, config_error,
13178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			       wps->error_indication);
13188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
13198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	default:
13208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		break;
13218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
13228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	/* Followed by NACK if Enrollee is Supplicant or EAP-Failure if
13248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	 * Enrollee is Authenticator */
13258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wps->state = SEND_WSC_NACK;
13268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return WPS_FAILURE;
13288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
13298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtenum wps_process_res wps_enrollee_process_msg(struct wps_data *wps,
13328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					      enum wsc_op_code op_code,
13338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt					      const struct wpabuf *msg)
13348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
13358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: Processing received message (len=%lu "
13378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   "op_code=%d)",
13388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		   (unsigned long) wpabuf_len(msg), op_code);
13398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (op_code == WSC_UPnP) {
13418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		/* Determine the OpCode based on message type attribute */
13428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		struct wps_parse_attr attr;
13438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_parse_msg(msg, &attr) == 0 && attr.msg_type) {
13448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			if (*attr.msg_type == WPS_WSC_ACK)
13458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				op_code = WSC_ACK;
13468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			else if (*attr.msg_type == WPS_WSC_NACK)
13478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				op_code = WSC_NACK;
13488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
13498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
13508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
13518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	switch (op_code) {
13528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WSC_MSG:
13538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WSC_UPnP:
13548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return wps_process_wsc_msg(wps, msg);
13558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WSC_ACK:
13568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_wsc_ack(msg) < 0)
13578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
13588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return wps_process_wsc_ack(wps, msg);
13598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	case WSC_NACK:
13608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (wps_validate_wsc_nack(msg) < 0)
13618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			return WPS_FAILURE;
13628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return wps_process_wsc_nack(wps, msg);
13638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	default:
13648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: Unsupported op_code %d", op_code);
13658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return WPS_FAILURE;
13668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
13678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
1368