1 /*
2  * Wi-Fi Protected Setup - Enrollee
3  * Copyright (c) 2008, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "includes.h"
10 
11 #include "common.h"
12 #include "crypto/crypto.h"
13 #include "crypto/sha256.h"
14 #include "crypto/random.h"
15 #include "wps_i.h"
16 #include "wps_dev_attr.h"
17 
18 
19 static int wps_build_wps_state(struct wps_data *wps, struct wpabuf *msg)
20 {
21 	u8 state;
22 	if (wps->wps->ap)
23 		state = wps->wps->wps_state;
24 	else
25 		state = WPS_STATE_NOT_CONFIGURED;
26 	wpa_printf(MSG_DEBUG, "WPS:  * Wi-Fi Protected Setup State (%d)",
27 		   state);
28 	wpabuf_put_be16(msg, ATTR_WPS_STATE);
29 	wpabuf_put_be16(msg, 1);
30 	wpabuf_put_u8(msg, state);
31 	return 0;
32 }
33 
34 
35 static int wps_build_e_hash(struct wps_data *wps, struct wpabuf *msg)
36 {
37 	u8 *hash;
38 	const u8 *addr[4];
39 	size_t len[4];
40 
41 	if (random_get_bytes(wps->snonce, 2 * WPS_SECRET_NONCE_LEN) < 0)
42 		return -1;
43 	wpa_hexdump(MSG_DEBUG, "WPS: E-S1", wps->snonce, WPS_SECRET_NONCE_LEN);
44 	wpa_hexdump(MSG_DEBUG, "WPS: E-S2",
45 		    wps->snonce + WPS_SECRET_NONCE_LEN, WPS_SECRET_NONCE_LEN);
46 
47 	if (wps->dh_pubkey_e == NULL || wps->dh_pubkey_r == NULL) {
48 		wpa_printf(MSG_DEBUG, "WPS: DH public keys not available for "
49 			   "E-Hash derivation");
50 		return -1;
51 	}
52 
53 	wpa_printf(MSG_DEBUG, "WPS:  * E-Hash1");
54 	wpabuf_put_be16(msg, ATTR_E_HASH1);
55 	wpabuf_put_be16(msg, SHA256_MAC_LEN);
56 	hash = wpabuf_put(msg, SHA256_MAC_LEN);
57 	/* E-Hash1 = HMAC_AuthKey(E-S1 || PSK1 || PK_E || PK_R) */
58 	addr[0] = wps->snonce;
59 	len[0] = WPS_SECRET_NONCE_LEN;
60 	addr[1] = wps->psk1;
61 	len[1] = WPS_PSK_LEN;
62 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
63 	len[2] = wpabuf_len(wps->dh_pubkey_e);
64 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
65 	len[3] = wpabuf_len(wps->dh_pubkey_r);
66 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
67 	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash1", hash, SHA256_MAC_LEN);
68 
69 	wpa_printf(MSG_DEBUG, "WPS:  * E-Hash2");
70 	wpabuf_put_be16(msg, ATTR_E_HASH2);
71 	wpabuf_put_be16(msg, SHA256_MAC_LEN);
72 	hash = wpabuf_put(msg, SHA256_MAC_LEN);
73 	/* E-Hash2 = HMAC_AuthKey(E-S2 || PSK2 || PK_E || PK_R) */
74 	addr[0] = wps->snonce + WPS_SECRET_NONCE_LEN;
75 	addr[1] = wps->psk2;
76 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
77 	wpa_hexdump(MSG_DEBUG, "WPS: E-Hash2", hash, SHA256_MAC_LEN);
78 
79 	return 0;
80 }
81 
82 
83 static int wps_build_e_snonce1(struct wps_data *wps, struct wpabuf *msg)
84 {
85 	wpa_printf(MSG_DEBUG, "WPS:  * E-SNonce1");
86 	wpabuf_put_be16(msg, ATTR_E_SNONCE1);
87 	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
88 	wpabuf_put_data(msg, wps->snonce, WPS_SECRET_NONCE_LEN);
89 	return 0;
90 }
91 
92 
93 static int wps_build_e_snonce2(struct wps_data *wps, struct wpabuf *msg)
94 {
95 	wpa_printf(MSG_DEBUG, "WPS:  * E-SNonce2");
96 	wpabuf_put_be16(msg, ATTR_E_SNONCE2);
97 	wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
98 	wpabuf_put_data(msg, wps->snonce + WPS_SECRET_NONCE_LEN,
99 			WPS_SECRET_NONCE_LEN);
100 	return 0;
101 }
102 
103 
104 static struct wpabuf * wps_build_m1(struct wps_data *wps)
105 {
106 	struct wpabuf *msg;
107 	u16 config_methods;
108 
109 	if (random_get_bytes(wps->nonce_e, WPS_NONCE_LEN) < 0)
110 		return NULL;
111 	wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Nonce",
112 		    wps->nonce_e, WPS_NONCE_LEN);
113 
114 	wpa_printf(MSG_DEBUG, "WPS: Building Message M1");
115 	msg = wpabuf_alloc(1000);
116 	if (msg == NULL)
117 		return NULL;
118 
119 	config_methods = wps->wps->config_methods;
120 	if (wps->wps->ap && !wps->pbc_in_m1 &&
121 	    (wps->dev_password_len != 0 ||
122 	     (config_methods & WPS_CONFIG_DISPLAY))) {
123 		/*
124 		 * These are the methods that the AP supports as an Enrollee
125 		 * for adding external Registrars, so remove PushButton.
126 		 *
127 		 * As a workaround for Windows 7 mechanism for probing WPS
128 		 * capabilities from M1, leave PushButton option if no PIN
129 		 * method is available or if WPS configuration enables PBC
130 		 * workaround.
131 		 */
132 		config_methods &= ~WPS_CONFIG_PUSHBUTTON;
133 #ifdef CONFIG_WPS2
134 		config_methods &= ~(WPS_CONFIG_VIRT_PUSHBUTTON |
135 				    WPS_CONFIG_PHY_PUSHBUTTON);
136 #endif /* CONFIG_WPS2 */
137 	}
138 
139 	if (wps_build_version(msg) ||
140 	    wps_build_msg_type(msg, WPS_M1) ||
141 	    wps_build_uuid_e(msg, wps->uuid_e) ||
142 	    wps_build_mac_addr(msg, wps->mac_addr_e) ||
143 	    wps_build_enrollee_nonce(wps, msg) ||
144 	    wps_build_public_key(wps, msg) ||
145 	    wps_build_auth_type_flags(wps, msg) ||
146 	    wps_build_encr_type_flags(wps, msg) ||
147 	    wps_build_conn_type_flags(wps, msg) ||
148 	    wps_build_config_methods(msg, config_methods) ||
149 	    wps_build_wps_state(wps, msg) ||
150 	    wps_build_device_attrs(&wps->wps->dev, msg) ||
151 	    wps_build_rf_bands(&wps->wps->dev, msg,
152 			       wps->wps->rf_band_cb(wps->wps->cb_ctx)) ||
153 	    wps_build_assoc_state(wps, msg) ||
154 	    wps_build_dev_password_id(msg, wps->dev_pw_id) ||
155 	    wps_build_config_error(msg, WPS_CFG_NO_ERROR) ||
156 	    wps_build_os_version(&wps->wps->dev, msg) ||
157 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
158 	    wps_build_vendor_ext_m1(&wps->wps->dev, msg)) {
159 		wpabuf_free(msg);
160 		return NULL;
161 	}
162 
163 	wps->state = RECV_M2;
164 	return msg;
165 }
166 
167 
168 static struct wpabuf * wps_build_m3(struct wps_data *wps)
169 {
170 	struct wpabuf *msg;
171 
172 	wpa_printf(MSG_DEBUG, "WPS: Building Message M3");
173 
174 	if (wps->dev_password == NULL) {
175 		wpa_printf(MSG_DEBUG, "WPS: No Device Password available");
176 		return NULL;
177 	}
178 	wps_derive_psk(wps, wps->dev_password, wps->dev_password_len);
179 
180 	msg = wpabuf_alloc(1000);
181 	if (msg == NULL)
182 		return NULL;
183 
184 	if (wps_build_version(msg) ||
185 	    wps_build_msg_type(msg, WPS_M3) ||
186 	    wps_build_registrar_nonce(wps, msg) ||
187 	    wps_build_e_hash(wps, msg) ||
188 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
189 	    wps_build_authenticator(wps, msg)) {
190 		wpabuf_free(msg);
191 		return NULL;
192 	}
193 
194 	wps->state = RECV_M4;
195 	return msg;
196 }
197 
198 
199 static struct wpabuf * wps_build_m5(struct wps_data *wps)
200 {
201 	struct wpabuf *msg, *plain;
202 
203 	wpa_printf(MSG_DEBUG, "WPS: Building Message M5");
204 
205 	plain = wpabuf_alloc(200);
206 	if (plain == NULL)
207 		return NULL;
208 
209 	msg = wpabuf_alloc(1000);
210 	if (msg == NULL) {
211 		wpabuf_free(plain);
212 		return NULL;
213 	}
214 
215 	if (wps_build_version(msg) ||
216 	    wps_build_msg_type(msg, WPS_M5) ||
217 	    wps_build_registrar_nonce(wps, msg) ||
218 	    wps_build_e_snonce1(wps, plain) ||
219 	    wps_build_key_wrap_auth(wps, plain) ||
220 	    wps_build_encr_settings(wps, msg, plain) ||
221 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
222 	    wps_build_authenticator(wps, msg)) {
223 		wpabuf_free(plain);
224 		wpabuf_free(msg);
225 		return NULL;
226 	}
227 	wpabuf_free(plain);
228 
229 	wps->state = RECV_M6;
230 	return msg;
231 }
232 
233 
234 static int wps_build_cred_ssid(struct wps_data *wps, struct wpabuf *msg)
235 {
236 	wpa_printf(MSG_DEBUG, "WPS:  * SSID");
237 	wpabuf_put_be16(msg, ATTR_SSID);
238 	wpabuf_put_be16(msg, wps->wps->ssid_len);
239 	wpabuf_put_data(msg, wps->wps->ssid, wps->wps->ssid_len);
240 	return 0;
241 }
242 
243 
244 static int wps_build_cred_auth_type(struct wps_data *wps, struct wpabuf *msg)
245 {
246 	u16 auth_type = wps->wps->auth_types;
247 
248 	/* Select the best authentication type */
249 	if (auth_type & WPS_AUTH_WPA2PSK)
250 		auth_type = WPS_AUTH_WPA2PSK;
251 	else if (auth_type & WPS_AUTH_WPAPSK)
252 		auth_type = WPS_AUTH_WPAPSK;
253 	else if (auth_type & WPS_AUTH_OPEN)
254 		auth_type = WPS_AUTH_OPEN;
255 	else if (auth_type & WPS_AUTH_SHARED)
256 		auth_type = WPS_AUTH_SHARED;
257 
258 	wpa_printf(MSG_DEBUG, "WPS:  * Authentication Type (0x%x)", auth_type);
259 	wpabuf_put_be16(msg, ATTR_AUTH_TYPE);
260 	wpabuf_put_be16(msg, 2);
261 	wpabuf_put_be16(msg, auth_type);
262 	return 0;
263 }
264 
265 
266 static int wps_build_cred_encr_type(struct wps_data *wps, struct wpabuf *msg)
267 {
268 	u16 encr_type = wps->wps->encr_types;
269 
270 	/* Select the best encryption type */
271 	if (wps->wps->auth_types & (WPS_AUTH_WPA2PSK | WPS_AUTH_WPAPSK)) {
272 		if (encr_type & WPS_ENCR_AES)
273 			encr_type = WPS_ENCR_AES;
274 		else if (encr_type & WPS_ENCR_TKIP)
275 			encr_type = WPS_ENCR_TKIP;
276 	} else {
277 		if (encr_type & WPS_ENCR_WEP)
278 			encr_type = WPS_ENCR_WEP;
279 		else if (encr_type & WPS_ENCR_NONE)
280 			encr_type = WPS_ENCR_NONE;
281 	}
282 
283 	wpa_printf(MSG_DEBUG, "WPS:  * Encryption Type (0x%x)", encr_type);
284 	wpabuf_put_be16(msg, ATTR_ENCR_TYPE);
285 	wpabuf_put_be16(msg, 2);
286 	wpabuf_put_be16(msg, encr_type);
287 	return 0;
288 }
289 
290 
291 static int wps_build_cred_network_key(struct wps_data *wps, struct wpabuf *msg)
292 {
293 	wpa_printf(MSG_DEBUG, "WPS:  * Network Key");
294 	wpabuf_put_be16(msg, ATTR_NETWORK_KEY);
295 	wpabuf_put_be16(msg, wps->wps->network_key_len);
296 	wpabuf_put_data(msg, wps->wps->network_key, wps->wps->network_key_len);
297 	return 0;
298 }
299 
300 
301 static int wps_build_cred_mac_addr(struct wps_data *wps, struct wpabuf *msg)
302 {
303 	wpa_printf(MSG_DEBUG, "WPS:  * MAC Address (AP BSSID)");
304 	wpabuf_put_be16(msg, ATTR_MAC_ADDR);
305 	wpabuf_put_be16(msg, ETH_ALEN);
306 	wpabuf_put_data(msg, wps->wps->dev.mac_addr, ETH_ALEN);
307 	return 0;
308 }
309 
310 
311 static int wps_build_ap_settings(struct wps_data *wps, struct wpabuf *plain)
312 {
313 	if (wps->wps->ap_settings) {
314 		wpa_printf(MSG_DEBUG, "WPS:  * AP Settings (pre-configured)");
315 		wpabuf_put_data(plain, wps->wps->ap_settings,
316 				wps->wps->ap_settings_len);
317 		return 0;
318 	}
319 
320 	return wps_build_cred_ssid(wps, plain) ||
321 		wps_build_cred_mac_addr(wps, plain) ||
322 		wps_build_cred_auth_type(wps, plain) ||
323 		wps_build_cred_encr_type(wps, plain) ||
324 		wps_build_cred_network_key(wps, plain);
325 }
326 
327 
328 static struct wpabuf * wps_build_m7(struct wps_data *wps)
329 {
330 	struct wpabuf *msg, *plain;
331 
332 	wpa_printf(MSG_DEBUG, "WPS: Building Message M7");
333 
334 	plain = wpabuf_alloc(500 + wps->wps->ap_settings_len);
335 	if (plain == NULL)
336 		return NULL;
337 
338 	msg = wpabuf_alloc(1000 + wps->wps->ap_settings_len);
339 	if (msg == NULL) {
340 		wpabuf_free(plain);
341 		return NULL;
342 	}
343 
344 	if (wps_build_version(msg) ||
345 	    wps_build_msg_type(msg, WPS_M7) ||
346 	    wps_build_registrar_nonce(wps, msg) ||
347 	    wps_build_e_snonce2(wps, plain) ||
348 	    (wps->wps->ap && wps_build_ap_settings(wps, plain)) ||
349 	    wps_build_key_wrap_auth(wps, plain) ||
350 	    wps_build_encr_settings(wps, msg, plain) ||
351 	    wps_build_wfa_ext(msg, 0, NULL, 0) ||
352 	    wps_build_authenticator(wps, msg)) {
353 		wpabuf_free(plain);
354 		wpabuf_free(msg);
355 		return NULL;
356 	}
357 	wpabuf_free(plain);
358 
359 	if (wps->wps->ap && wps->wps->registrar) {
360 		/*
361 		 * If the Registrar is only learning our current configuration,
362 		 * it may not continue protocol run to successful completion.
363 		 * Store information here to make sure it remains available.
364 		 */
365 		wps_device_store(wps->wps->registrar, &wps->peer_dev,
366 				 wps->uuid_r);
367 	}
368 
369 	wps->state = RECV_M8;
370 	return msg;
371 }
372 
373 
374 static struct wpabuf * wps_build_wsc_done(struct wps_data *wps)
375 {
376 	struct wpabuf *msg;
377 
378 	wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_Done");
379 
380 	msg = wpabuf_alloc(1000);
381 	if (msg == NULL)
382 		return NULL;
383 
384 	if (wps_build_version(msg) ||
385 	    wps_build_msg_type(msg, WPS_WSC_DONE) ||
386 	    wps_build_enrollee_nonce(wps, msg) ||
387 	    wps_build_registrar_nonce(wps, msg) ||
388 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
389 		wpabuf_free(msg);
390 		return NULL;
391 	}
392 
393 	if (wps->wps->ap)
394 		wps->state = RECV_ACK;
395 	else {
396 		wps_success_event(wps->wps, wps->peer_dev.mac_addr);
397 		wps->state = WPS_FINISHED;
398 	}
399 	return msg;
400 }
401 
402 
403 struct wpabuf * wps_enrollee_get_msg(struct wps_data *wps,
404 				     enum wsc_op_code *op_code)
405 {
406 	struct wpabuf *msg;
407 
408 	switch (wps->state) {
409 	case SEND_M1:
410 		msg = wps_build_m1(wps);
411 		*op_code = WSC_MSG;
412 		break;
413 	case SEND_M3:
414 		msg = wps_build_m3(wps);
415 		*op_code = WSC_MSG;
416 		break;
417 	case SEND_M5:
418 		msg = wps_build_m5(wps);
419 		*op_code = WSC_MSG;
420 		break;
421 	case SEND_M7:
422 		msg = wps_build_m7(wps);
423 		*op_code = WSC_MSG;
424 		break;
425 	case RECEIVED_M2D:
426 		if (wps->wps->ap) {
427 			msg = wps_build_wsc_nack(wps);
428 			*op_code = WSC_NACK;
429 			break;
430 		}
431 		msg = wps_build_wsc_ack(wps);
432 		*op_code = WSC_ACK;
433 		if (msg) {
434 			/* Another M2/M2D may be received */
435 			wps->state = RECV_M2;
436 		}
437 		break;
438 	case SEND_WSC_NACK:
439 		msg = wps_build_wsc_nack(wps);
440 		*op_code = WSC_NACK;
441 		break;
442 	case WPS_MSG_DONE:
443 		msg = wps_build_wsc_done(wps);
444 		*op_code = WSC_Done;
445 		break;
446 	default:
447 		wpa_printf(MSG_DEBUG, "WPS: Unsupported state %d for building "
448 			   "a message", wps->state);
449 		msg = NULL;
450 		break;
451 	}
452 
453 	if (*op_code == WSC_MSG && msg) {
454 		/* Save a copy of the last message for Authenticator derivation
455 		 */
456 		wpabuf_free(wps->last_msg);
457 		wps->last_msg = wpabuf_dup(msg);
458 	}
459 
460 	return msg;
461 }
462 
463 
464 static int wps_process_registrar_nonce(struct wps_data *wps, const u8 *r_nonce)
465 {
466 	if (r_nonce == NULL) {
467 		wpa_printf(MSG_DEBUG, "WPS: No Registrar Nonce received");
468 		return -1;
469 	}
470 
471 	os_memcpy(wps->nonce_r, r_nonce, WPS_NONCE_LEN);
472 	wpa_hexdump(MSG_DEBUG, "WPS: Registrar Nonce",
473 		    wps->nonce_r, WPS_NONCE_LEN);
474 
475 	return 0;
476 }
477 
478 
479 static int wps_process_enrollee_nonce(struct wps_data *wps, const u8 *e_nonce)
480 {
481 	if (e_nonce == NULL) {
482 		wpa_printf(MSG_DEBUG, "WPS: No Enrollee Nonce received");
483 		return -1;
484 	}
485 
486 	if (os_memcmp(wps->nonce_e, e_nonce, WPS_NONCE_LEN) != 0) {
487 		wpa_printf(MSG_DEBUG, "WPS: Invalid Enrollee Nonce received");
488 		return -1;
489 	}
490 
491 	return 0;
492 }
493 
494 
495 static int wps_process_uuid_r(struct wps_data *wps, const u8 *uuid_r)
496 {
497 	if (uuid_r == NULL) {
498 		wpa_printf(MSG_DEBUG, "WPS: No UUID-R received");
499 		return -1;
500 	}
501 
502 	os_memcpy(wps->uuid_r, uuid_r, WPS_UUID_LEN);
503 	wpa_hexdump(MSG_DEBUG, "WPS: UUID-R", wps->uuid_r, WPS_UUID_LEN);
504 
505 	return 0;
506 }
507 
508 
509 static int wps_process_pubkey(struct wps_data *wps, const u8 *pk,
510 			      size_t pk_len)
511 {
512 	if (pk == NULL || pk_len == 0) {
513 		wpa_printf(MSG_DEBUG, "WPS: No Public Key received");
514 		return -1;
515 	}
516 
517 	if (wps->peer_pubkey_hash_set) {
518 		u8 hash[WPS_HASH_LEN];
519 		sha256_vector(1, &pk, &pk_len, hash);
520 		if (os_memcmp(hash, wps->peer_pubkey_hash,
521 			      WPS_OOB_PUBKEY_HASH_LEN) != 0) {
522 			wpa_printf(MSG_ERROR, "WPS: Public Key hash mismatch");
523 			wpa_hexdump(MSG_DEBUG, "WPS: Received public key",
524 				    pk, pk_len);
525 			wpa_hexdump(MSG_DEBUG, "WPS: Calculated public key "
526 				    "hash", hash, WPS_OOB_PUBKEY_HASH_LEN);
527 			wpa_hexdump(MSG_DEBUG, "WPS: Expected public key hash",
528 				    wps->peer_pubkey_hash,
529 				    WPS_OOB_PUBKEY_HASH_LEN);
530 			wps->config_error = WPS_CFG_PUBLIC_KEY_HASH_MISMATCH;
531 			return -1;
532 		}
533 	}
534 
535 	wpabuf_free(wps->dh_pubkey_r);
536 	wps->dh_pubkey_r = wpabuf_alloc_copy(pk, pk_len);
537 	if (wps->dh_pubkey_r == NULL)
538 		return -1;
539 
540 	if (wps_derive_keys(wps) < 0)
541 		return -1;
542 
543 	return 0;
544 }
545 
546 
547 static int wps_process_r_hash1(struct wps_data *wps, const u8 *r_hash1)
548 {
549 	if (r_hash1 == NULL) {
550 		wpa_printf(MSG_DEBUG, "WPS: No R-Hash1 received");
551 		return -1;
552 	}
553 
554 	os_memcpy(wps->peer_hash1, r_hash1, WPS_HASH_LEN);
555 	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash1", wps->peer_hash1, WPS_HASH_LEN);
556 
557 	return 0;
558 }
559 
560 
561 static int wps_process_r_hash2(struct wps_data *wps, const u8 *r_hash2)
562 {
563 	if (r_hash2 == NULL) {
564 		wpa_printf(MSG_DEBUG, "WPS: No R-Hash2 received");
565 		return -1;
566 	}
567 
568 	os_memcpy(wps->peer_hash2, r_hash2, WPS_HASH_LEN);
569 	wpa_hexdump(MSG_DEBUG, "WPS: R-Hash2", wps->peer_hash2, WPS_HASH_LEN);
570 
571 	return 0;
572 }
573 
574 
575 static int wps_process_r_snonce1(struct wps_data *wps, const u8 *r_snonce1)
576 {
577 	u8 hash[SHA256_MAC_LEN];
578 	const u8 *addr[4];
579 	size_t len[4];
580 
581 	if (r_snonce1 == NULL) {
582 		wpa_printf(MSG_DEBUG, "WPS: No R-SNonce1 received");
583 		return -1;
584 	}
585 
586 	wpa_hexdump_key(MSG_DEBUG, "WPS: R-SNonce1", r_snonce1,
587 			WPS_SECRET_NONCE_LEN);
588 
589 	/* R-Hash1 = HMAC_AuthKey(R-S1 || PSK1 || PK_E || PK_R) */
590 	addr[0] = r_snonce1;
591 	len[0] = WPS_SECRET_NONCE_LEN;
592 	addr[1] = wps->psk1;
593 	len[1] = WPS_PSK_LEN;
594 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
595 	len[2] = wpabuf_len(wps->dh_pubkey_e);
596 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
597 	len[3] = wpabuf_len(wps->dh_pubkey_r);
598 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
599 
600 	if (os_memcmp(wps->peer_hash1, hash, WPS_HASH_LEN) != 0) {
601 		wpa_printf(MSG_DEBUG, "WPS: R-Hash1 derived from R-S1 does "
602 			   "not match with the pre-committed value");
603 		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
604 		wps_pwd_auth_fail_event(wps->wps, 1, 1, wps->peer_dev.mac_addr);
605 		return -1;
606 	}
607 
608 	wpa_printf(MSG_DEBUG, "WPS: Registrar proved knowledge of the first "
609 		   "half of the device password");
610 
611 	return 0;
612 }
613 
614 
615 static int wps_process_r_snonce2(struct wps_data *wps, const u8 *r_snonce2)
616 {
617 	u8 hash[SHA256_MAC_LEN];
618 	const u8 *addr[4];
619 	size_t len[4];
620 
621 	if (r_snonce2 == NULL) {
622 		wpa_printf(MSG_DEBUG, "WPS: No R-SNonce2 received");
623 		return -1;
624 	}
625 
626 	wpa_hexdump_key(MSG_DEBUG, "WPS: R-SNonce2", r_snonce2,
627 			WPS_SECRET_NONCE_LEN);
628 
629 	/* R-Hash2 = HMAC_AuthKey(R-S2 || PSK2 || PK_E || PK_R) */
630 	addr[0] = r_snonce2;
631 	len[0] = WPS_SECRET_NONCE_LEN;
632 	addr[1] = wps->psk2;
633 	len[1] = WPS_PSK_LEN;
634 	addr[2] = wpabuf_head(wps->dh_pubkey_e);
635 	len[2] = wpabuf_len(wps->dh_pubkey_e);
636 	addr[3] = wpabuf_head(wps->dh_pubkey_r);
637 	len[3] = wpabuf_len(wps->dh_pubkey_r);
638 	hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
639 
640 	if (os_memcmp(wps->peer_hash2, hash, WPS_HASH_LEN) != 0) {
641 		wpa_printf(MSG_DEBUG, "WPS: R-Hash2 derived from R-S2 does "
642 			   "not match with the pre-committed value");
643 		wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
644 		wps_pwd_auth_fail_event(wps->wps, 1, 2, wps->peer_dev.mac_addr);
645 		return -1;
646 	}
647 
648 	wpa_printf(MSG_DEBUG, "WPS: Registrar proved knowledge of the second "
649 		   "half of the device password");
650 
651 	return 0;
652 }
653 
654 
655 static int wps_process_cred_e(struct wps_data *wps, const u8 *cred,
656 			      size_t cred_len, int wps2)
657 {
658 	struct wps_parse_attr attr;
659 	struct wpabuf msg;
660 	int ret = 0;
661 
662 	wpa_printf(MSG_DEBUG, "WPS: Received Credential");
663 	os_memset(&wps->cred, 0, sizeof(wps->cred));
664 	wpabuf_set(&msg, cred, cred_len);
665 	if (wps_parse_msg(&msg, &attr) < 0 ||
666 	    wps_process_cred(&attr, &wps->cred))
667 		return -1;
668 
669 	if (os_memcmp(wps->cred.mac_addr, wps->wps->dev.mac_addr, ETH_ALEN) !=
670 	    0) {
671 		wpa_printf(MSG_DEBUG, "WPS: MAC Address in the Credential ("
672 			   MACSTR ") does not match with own address (" MACSTR
673 			   ")", MAC2STR(wps->cred.mac_addr),
674 			   MAC2STR(wps->wps->dev.mac_addr));
675 		/*
676 		 * In theory, this could be consider fatal error, but there are
677 		 * number of deployed implementations using other address here
678 		 * due to unclarity in the specification. For interoperability
679 		 * reasons, allow this to be processed since we do not really
680 		 * use the MAC Address information for anything.
681 		 */
682 #ifdef CONFIG_WPS_STRICT
683 		if (wps2) {
684 			wpa_printf(MSG_INFO, "WPS: Do not accept incorrect "
685 				   "MAC Address in AP Settings");
686 			return -1;
687 		}
688 #endif /* CONFIG_WPS_STRICT */
689 	}
690 
691 #ifdef CONFIG_WPS2
692 	if (!(wps->cred.encr_type &
693 	      (WPS_ENCR_NONE | WPS_ENCR_TKIP | WPS_ENCR_AES))) {
694 		if (wps->cred.encr_type & WPS_ENCR_WEP) {
695 			wpa_printf(MSG_INFO, "WPS: Reject Credential "
696 				   "due to WEP configuration");
697 			wps->error_indication = WPS_EI_SECURITY_WEP_PROHIBITED;
698 			return -2;
699 		}
700 
701 		wpa_printf(MSG_INFO, "WPS: Reject Credential due to "
702 			   "invalid encr_type 0x%x", wps->cred.encr_type);
703 		return -1;
704 	}
705 #endif /* CONFIG_WPS2 */
706 
707 	if (wps->wps->cred_cb) {
708 		wps->cred.cred_attr = cred - 4;
709 		wps->cred.cred_attr_len = cred_len + 4;
710 		ret = wps->wps->cred_cb(wps->wps->cb_ctx, &wps->cred);
711 		wps->cred.cred_attr = NULL;
712 		wps->cred.cred_attr_len = 0;
713 	}
714 
715 	return ret;
716 }
717 
718 
719 static int wps_process_creds(struct wps_data *wps, const u8 *cred[],
720 			     size_t cred_len[], size_t num_cred, int wps2)
721 {
722 	size_t i;
723 	int ok = 0;
724 
725 	if (wps->wps->ap)
726 		return 0;
727 
728 	if (num_cred == 0) {
729 		wpa_printf(MSG_DEBUG, "WPS: No Credential attributes "
730 			   "received");
731 		return -1;
732 	}
733 
734 	for (i = 0; i < num_cred; i++) {
735 		int res;
736 		res = wps_process_cred_e(wps, cred[i], cred_len[i], wps2);
737 		if (res == 0)
738 			ok++;
739 		else if (res == -2)
740 			wpa_printf(MSG_DEBUG, "WPS: WEP credential skipped");
741 		else
742 			return -1;
743 	}
744 
745 	if (ok == 0) {
746 		wpa_printf(MSG_DEBUG, "WPS: No valid Credential attribute "
747 			   "received");
748 		return -1;
749 	}
750 
751 	return 0;
752 }
753 
754 
755 static int wps_process_ap_settings_e(struct wps_data *wps,
756 				     struct wps_parse_attr *attr,
757 				     struct wpabuf *attrs, int wps2)
758 {
759 	struct wps_credential cred;
760 
761 	if (!wps->wps->ap)
762 		return 0;
763 
764 	if (wps_process_ap_settings(attr, &cred) < 0)
765 		return -1;
766 
767 	wpa_printf(MSG_INFO, "WPS: Received new AP configuration from "
768 		   "Registrar");
769 
770 	if (os_memcmp(cred.mac_addr, wps->wps->dev.mac_addr, ETH_ALEN) !=
771 	    0) {
772 		wpa_printf(MSG_DEBUG, "WPS: MAC Address in the AP Settings ("
773 			   MACSTR ") does not match with own address (" MACSTR
774 			   ")", MAC2STR(cred.mac_addr),
775 			   MAC2STR(wps->wps->dev.mac_addr));
776 		/*
777 		 * In theory, this could be consider fatal error, but there are
778 		 * number of deployed implementations using other address here
779 		 * due to unclarity in the specification. For interoperability
780 		 * reasons, allow this to be processed since we do not really
781 		 * use the MAC Address information for anything.
782 		 */
783 #ifdef CONFIG_WPS_STRICT
784 		if (wps2) {
785 			wpa_printf(MSG_INFO, "WPS: Do not accept incorrect "
786 				   "MAC Address in AP Settings");
787 			return -1;
788 		}
789 #endif /* CONFIG_WPS_STRICT */
790 	}
791 
792 #ifdef CONFIG_WPS2
793 	if (!(cred.encr_type & (WPS_ENCR_NONE | WPS_ENCR_TKIP | WPS_ENCR_AES)))
794 	{
795 		if (cred.encr_type & WPS_ENCR_WEP) {
796 			wpa_printf(MSG_INFO, "WPS: Reject new AP settings "
797 				   "due to WEP configuration");
798 			wps->error_indication = WPS_EI_SECURITY_WEP_PROHIBITED;
799 			return -1;
800 		}
801 
802 		wpa_printf(MSG_INFO, "WPS: Reject new AP settings due to "
803 			   "invalid encr_type 0x%x", cred.encr_type);
804 		return -1;
805 	}
806 #endif /* CONFIG_WPS2 */
807 
808 #ifdef CONFIG_WPS_STRICT
809 	if (wps2) {
810 		if ((cred.encr_type & (WPS_ENCR_TKIP | WPS_ENCR_AES)) ==
811 		    WPS_ENCR_TKIP ||
812 		    (cred.auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) ==
813 		    WPS_AUTH_WPAPSK) {
814 			wpa_printf(MSG_INFO, "WPS-STRICT: Invalid WSC 2.0 "
815 				   "AP Settings: WPA-Personal/TKIP only");
816 			wps->error_indication =
817 				WPS_EI_SECURITY_TKIP_ONLY_PROHIBITED;
818 			return -1;
819 		}
820 	}
821 #endif /* CONFIG_WPS_STRICT */
822 
823 #ifdef CONFIG_WPS2
824 	if ((cred.encr_type & (WPS_ENCR_TKIP | WPS_ENCR_AES)) == WPS_ENCR_TKIP)
825 	{
826 		wpa_printf(MSG_DEBUG, "WPS: Upgrade encr_type TKIP -> "
827 			   "TKIP+AES");
828 		cred.encr_type |= WPS_ENCR_AES;
829 	}
830 
831 	if ((cred.auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) ==
832 	    WPS_AUTH_WPAPSK) {
833 		wpa_printf(MSG_DEBUG, "WPS: Upgrade auth_type WPAPSK -> "
834 			   "WPAPSK+WPA2PSK");
835 		cred.auth_type |= WPS_AUTH_WPA2PSK;
836 	}
837 #endif /* CONFIG_WPS2 */
838 
839 	if (wps->wps->cred_cb) {
840 		cred.cred_attr = wpabuf_head(attrs);
841 		cred.cred_attr_len = wpabuf_len(attrs);
842 		wps->wps->cred_cb(wps->wps->cb_ctx, &cred);
843 	}
844 
845 	return 0;
846 }
847 
848 
849 static int wps_process_dev_pw_id(struct wps_data *wps, const u8 *dev_pw_id)
850 {
851 	u16 id;
852 
853 	if (dev_pw_id == NULL) {
854 		wpa_printf(MSG_DEBUG, "WPS: Device Password ID");
855 		return -1;
856 	}
857 
858 	id = WPA_GET_BE16(dev_pw_id);
859 	if (wps->dev_pw_id == id) {
860 		wpa_printf(MSG_DEBUG, "WPS: Device Password ID %u", id);
861 		return 0;
862 	}
863 
864 #ifdef CONFIG_P2P
865 	if ((id == DEV_PW_DEFAULT &&
866 	     wps->dev_pw_id == DEV_PW_REGISTRAR_SPECIFIED) ||
867 	    (id == DEV_PW_REGISTRAR_SPECIFIED &&
868 	     wps->dev_pw_id == DEV_PW_DEFAULT)) {
869 		/*
870 		 * Common P2P use cases indicate whether the PIN is from the
871 		 * client or GO using Device Password Id in M1/M2 in a way that
872 		 * does not look fully compliant with WSC specification. Anyway,
873 		 * this is deployed and needs to be allowed, so ignore changes
874 		 * between Registrar-Specified and Default PIN.
875 		 */
876 		wpa_printf(MSG_DEBUG, "WPS: Allow PIN Device Password ID "
877 			   "change");
878 		return 0;
879 	}
880 #endif /* CONFIG_P2P */
881 
882 	wpa_printf(MSG_DEBUG, "WPS: Registrar trying to change Device Password "
883 		   "ID from %u to %u", wps->dev_pw_id, id);
884 
885 	if (wps->dev_pw_id == DEV_PW_PUSHBUTTON && id == DEV_PW_DEFAULT) {
886 		wpa_printf(MSG_DEBUG,
887 			   "WPS: Workaround - ignore PBC-to-PIN change");
888 		return 0;
889 	}
890 
891 	if (wps->alt_dev_password && wps->alt_dev_pw_id == id) {
892 		wpa_printf(MSG_DEBUG, "WPS: Found a matching Device Password");
893 		os_free(wps->dev_password);
894 		wps->dev_pw_id = wps->alt_dev_pw_id;
895 		wps->dev_password = wps->alt_dev_password;
896 		wps->dev_password_len = wps->alt_dev_password_len;
897 		wps->alt_dev_password = NULL;
898 		wps->alt_dev_password_len = 0;
899 		return 0;
900 	}
901 
902 	return -1;
903 }
904 
905 
906 static enum wps_process_res wps_process_m2(struct wps_data *wps,
907 					   const struct wpabuf *msg,
908 					   struct wps_parse_attr *attr)
909 {
910 	wpa_printf(MSG_DEBUG, "WPS: Received M2");
911 
912 	if (wps->state != RECV_M2) {
913 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
914 			   "receiving M2", wps->state);
915 		wps->state = SEND_WSC_NACK;
916 		return WPS_CONTINUE;
917 	}
918 
919 	if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
920 	    wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
921 	    wps_process_uuid_r(wps, attr->uuid_r) ||
922 	    wps_process_dev_pw_id(wps, attr->dev_password_id)) {
923 		wps->state = SEND_WSC_NACK;
924 		return WPS_CONTINUE;
925 	}
926 
927 	/*
928 	 * Stop here on an AP as an Enrollee if AP Setup is locked unless the
929 	 * special locked mode is used to allow protocol run up to M7 in order
930 	 * to support external Registrars that only learn the current AP
931 	 * configuration without changing it.
932 	 */
933 	if (wps->wps->ap &&
934 	    ((wps->wps->ap_setup_locked && wps->wps->ap_setup_locked != 2) ||
935 	     wps->dev_password == NULL)) {
936 		wpa_printf(MSG_DEBUG, "WPS: AP Setup is locked - refuse "
937 			   "registration of a new Registrar");
938 		wps->config_error = WPS_CFG_SETUP_LOCKED;
939 		wps->state = SEND_WSC_NACK;
940 		return WPS_CONTINUE;
941 	}
942 
943 	if (wps_process_pubkey(wps, attr->public_key, attr->public_key_len) ||
944 	    wps_process_authenticator(wps, attr->authenticator, msg) ||
945 	    wps_process_device_attrs(&wps->peer_dev, attr)) {
946 		wps->state = SEND_WSC_NACK;
947 		return WPS_CONTINUE;
948 	}
949 
950 #ifdef CONFIG_WPS_NFC
951 	if (wps->peer_pubkey_hash_set) {
952 		struct wpabuf *decrypted;
953 		struct wps_parse_attr eattr;
954 
955 		decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
956 						      attr->encr_settings_len);
957 		if (decrypted == NULL) {
958 			wpa_printf(MSG_DEBUG, "WPS: Failed to decrypt "
959 				   "Encrypted Settings attribute");
960 			wps->state = SEND_WSC_NACK;
961 			return WPS_CONTINUE;
962 		}
963 
964 		wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted "
965 			   "Settings attribute");
966 		if (wps_parse_msg(decrypted, &eattr) < 0 ||
967 		    wps_process_key_wrap_auth(wps, decrypted,
968 					      eattr.key_wrap_auth) ||
969 		    wps_process_creds(wps, eattr.cred, eattr.cred_len,
970 				      eattr.num_cred, attr->version2 != NULL)) {
971 			wpabuf_free(decrypted);
972 			wps->state = SEND_WSC_NACK;
973 			return WPS_CONTINUE;
974 		}
975 		wpabuf_free(decrypted);
976 
977 		wps->state = WPS_MSG_DONE;
978 		return WPS_CONTINUE;
979 	}
980 #endif /* CONFIG_WPS_NFC */
981 
982 	wps->state = SEND_M3;
983 	return WPS_CONTINUE;
984 }
985 
986 
987 static enum wps_process_res wps_process_m2d(struct wps_data *wps,
988 					    struct wps_parse_attr *attr)
989 {
990 	wpa_printf(MSG_DEBUG, "WPS: Received M2D");
991 
992 	if (wps->state != RECV_M2) {
993 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
994 			   "receiving M2D", wps->state);
995 		wps->state = SEND_WSC_NACK;
996 		return WPS_CONTINUE;
997 	}
998 
999 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Manufacturer",
1000 			  attr->manufacturer, attr->manufacturer_len);
1001 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Name",
1002 			  attr->model_name, attr->model_name_len);
1003 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Number",
1004 			  attr->model_number, attr->model_number_len);
1005 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Serial Number",
1006 			  attr->serial_number, attr->serial_number_len);
1007 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: Device Name",
1008 			  attr->dev_name, attr->dev_name_len);
1009 
1010 	if (wps->wps->event_cb) {
1011 		union wps_event_data data;
1012 		struct wps_event_m2d *m2d = &data.m2d;
1013 		os_memset(&data, 0, sizeof(data));
1014 		if (attr->config_methods)
1015 			m2d->config_methods =
1016 				WPA_GET_BE16(attr->config_methods);
1017 		m2d->manufacturer = attr->manufacturer;
1018 		m2d->manufacturer_len = attr->manufacturer_len;
1019 		m2d->model_name = attr->model_name;
1020 		m2d->model_name_len = attr->model_name_len;
1021 		m2d->model_number = attr->model_number;
1022 		m2d->model_number_len = attr->model_number_len;
1023 		m2d->serial_number = attr->serial_number;
1024 		m2d->serial_number_len = attr->serial_number_len;
1025 		m2d->dev_name = attr->dev_name;
1026 		m2d->dev_name_len = attr->dev_name_len;
1027 		m2d->primary_dev_type = attr->primary_dev_type;
1028 		if (attr->config_error)
1029 			m2d->config_error =
1030 				WPA_GET_BE16(attr->config_error);
1031 		if (attr->dev_password_id)
1032 			m2d->dev_password_id =
1033 				WPA_GET_BE16(attr->dev_password_id);
1034 		wps->wps->event_cb(wps->wps->cb_ctx, WPS_EV_M2D, &data);
1035 	}
1036 
1037 	wps->state = RECEIVED_M2D;
1038 	return WPS_CONTINUE;
1039 }
1040 
1041 
1042 static enum wps_process_res wps_process_m4(struct wps_data *wps,
1043 					   const struct wpabuf *msg,
1044 					   struct wps_parse_attr *attr)
1045 {
1046 	struct wpabuf *decrypted;
1047 	struct wps_parse_attr eattr;
1048 
1049 	wpa_printf(MSG_DEBUG, "WPS: Received M4");
1050 
1051 	if (wps->state != RECV_M4) {
1052 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
1053 			   "receiving M4", wps->state);
1054 		wps->state = SEND_WSC_NACK;
1055 		return WPS_CONTINUE;
1056 	}
1057 
1058 	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
1059 	    wps_process_authenticator(wps, attr->authenticator, msg) ||
1060 	    wps_process_r_hash1(wps, attr->r_hash1) ||
1061 	    wps_process_r_hash2(wps, attr->r_hash2)) {
1062 		wps->state = SEND_WSC_NACK;
1063 		return WPS_CONTINUE;
1064 	}
1065 
1066 	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
1067 					      attr->encr_settings_len);
1068 	if (decrypted == NULL) {
1069 		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
1070 			   "Settings attribute");
1071 		wps->state = SEND_WSC_NACK;
1072 		return WPS_CONTINUE;
1073 	}
1074 
1075 	if (wps_validate_m4_encr(decrypted, attr->version2 != NULL) < 0) {
1076 		wpabuf_free(decrypted);
1077 		wps->state = SEND_WSC_NACK;
1078 		return WPS_CONTINUE;
1079 	}
1080 
1081 	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
1082 		   "attribute");
1083 	if (wps_parse_msg(decrypted, &eattr) < 0 ||
1084 	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
1085 	    wps_process_r_snonce1(wps, eattr.r_snonce1)) {
1086 		wpabuf_free(decrypted);
1087 		wps->state = SEND_WSC_NACK;
1088 		return WPS_CONTINUE;
1089 	}
1090 	wpabuf_free(decrypted);
1091 
1092 	wps->state = SEND_M5;
1093 	return WPS_CONTINUE;
1094 }
1095 
1096 
1097 static enum wps_process_res wps_process_m6(struct wps_data *wps,
1098 					   const struct wpabuf *msg,
1099 					   struct wps_parse_attr *attr)
1100 {
1101 	struct wpabuf *decrypted;
1102 	struct wps_parse_attr eattr;
1103 
1104 	wpa_printf(MSG_DEBUG, "WPS: Received M6");
1105 
1106 	if (wps->state != RECV_M6) {
1107 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
1108 			   "receiving M6", wps->state);
1109 		wps->state = SEND_WSC_NACK;
1110 		return WPS_CONTINUE;
1111 	}
1112 
1113 	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
1114 	    wps_process_authenticator(wps, attr->authenticator, msg)) {
1115 		wps->state = SEND_WSC_NACK;
1116 		return WPS_CONTINUE;
1117 	}
1118 
1119 	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
1120 					      attr->encr_settings_len);
1121 	if (decrypted == NULL) {
1122 		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
1123 			   "Settings attribute");
1124 		wps->state = SEND_WSC_NACK;
1125 		return WPS_CONTINUE;
1126 	}
1127 
1128 	if (wps_validate_m6_encr(decrypted, attr->version2 != NULL) < 0) {
1129 		wpabuf_free(decrypted);
1130 		wps->state = SEND_WSC_NACK;
1131 		return WPS_CONTINUE;
1132 	}
1133 
1134 	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
1135 		   "attribute");
1136 	if (wps_parse_msg(decrypted, &eattr) < 0 ||
1137 	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
1138 	    wps_process_r_snonce2(wps, eattr.r_snonce2)) {
1139 		wpabuf_free(decrypted);
1140 		wps->state = SEND_WSC_NACK;
1141 		return WPS_CONTINUE;
1142 	}
1143 	wpabuf_free(decrypted);
1144 
1145 	if (wps->wps->ap)
1146 		wps->wps->event_cb(wps->wps->cb_ctx, WPS_EV_AP_PIN_SUCCESS,
1147 				   NULL);
1148 
1149 	wps->state = SEND_M7;
1150 	return WPS_CONTINUE;
1151 }
1152 
1153 
1154 static enum wps_process_res wps_process_m8(struct wps_data *wps,
1155 					   const struct wpabuf *msg,
1156 					   struct wps_parse_attr *attr)
1157 {
1158 	struct wpabuf *decrypted;
1159 	struct wps_parse_attr eattr;
1160 
1161 	wpa_printf(MSG_DEBUG, "WPS: Received M8");
1162 
1163 	if (wps->state != RECV_M8) {
1164 		wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
1165 			   "receiving M8", wps->state);
1166 		wps->state = SEND_WSC_NACK;
1167 		return WPS_CONTINUE;
1168 	}
1169 
1170 	if (wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
1171 	    wps_process_authenticator(wps, attr->authenticator, msg)) {
1172 		wps->state = SEND_WSC_NACK;
1173 		return WPS_CONTINUE;
1174 	}
1175 
1176 	if (wps->wps->ap && wps->wps->ap_setup_locked) {
1177 		/*
1178 		 * Stop here if special ap_setup_locked == 2 mode allowed the
1179 		 * protocol to continue beyond M2. This allows ER to learn the
1180 		 * current AP settings without changing them.
1181 		 */
1182 		wpa_printf(MSG_DEBUG, "WPS: AP Setup is locked - refuse "
1183 			   "registration of a new Registrar");
1184 		wps->config_error = WPS_CFG_SETUP_LOCKED;
1185 		wps->state = SEND_WSC_NACK;
1186 		return WPS_CONTINUE;
1187 	}
1188 
1189 	decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
1190 					      attr->encr_settings_len);
1191 	if (decrypted == NULL) {
1192 		wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
1193 			   "Settings attribute");
1194 		wps->state = SEND_WSC_NACK;
1195 		return WPS_CONTINUE;
1196 	}
1197 
1198 	if (wps_validate_m8_encr(decrypted, wps->wps->ap,
1199 				 attr->version2 != NULL) < 0) {
1200 		wpabuf_free(decrypted);
1201 		wps->state = SEND_WSC_NACK;
1202 		return WPS_CONTINUE;
1203 	}
1204 
1205 	wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
1206 		   "attribute");
1207 	if (wps_parse_msg(decrypted, &eattr) < 0 ||
1208 	    wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
1209 	    wps_process_creds(wps, eattr.cred, eattr.cred_len,
1210 			      eattr.num_cred, attr->version2 != NULL) ||
1211 	    wps_process_ap_settings_e(wps, &eattr, decrypted,
1212 				      attr->version2 != NULL)) {
1213 		wpabuf_free(decrypted);
1214 		wps->state = SEND_WSC_NACK;
1215 		return WPS_CONTINUE;
1216 	}
1217 	wpabuf_free(decrypted);
1218 
1219 	wps->state = WPS_MSG_DONE;
1220 	return WPS_CONTINUE;
1221 }
1222 
1223 
1224 static enum wps_process_res wps_process_wsc_msg(struct wps_data *wps,
1225 						const struct wpabuf *msg)
1226 {
1227 	struct wps_parse_attr attr;
1228 	enum wps_process_res ret = WPS_CONTINUE;
1229 
1230 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_MSG");
1231 
1232 	if (wps_parse_msg(msg, &attr) < 0)
1233 		return WPS_FAILURE;
1234 
1235 	if (attr.enrollee_nonce == NULL ||
1236 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
1237 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
1238 		return WPS_FAILURE;
1239 	}
1240 
1241 	if (attr.msg_type == NULL) {
1242 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
1243 		wps->state = SEND_WSC_NACK;
1244 		return WPS_CONTINUE;
1245 	}
1246 
1247 	switch (*attr.msg_type) {
1248 	case WPS_M2:
1249 		if (wps_validate_m2(msg) < 0)
1250 			return WPS_FAILURE;
1251 		ret = wps_process_m2(wps, msg, &attr);
1252 		break;
1253 	case WPS_M2D:
1254 		if (wps_validate_m2d(msg) < 0)
1255 			return WPS_FAILURE;
1256 		ret = wps_process_m2d(wps, &attr);
1257 		break;
1258 	case WPS_M4:
1259 		if (wps_validate_m4(msg) < 0)
1260 			return WPS_FAILURE;
1261 		ret = wps_process_m4(wps, msg, &attr);
1262 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
1263 			wps_fail_event(wps->wps, WPS_M4, wps->config_error,
1264 				       wps->error_indication,
1265 				       wps->peer_dev.mac_addr);
1266 		break;
1267 	case WPS_M6:
1268 		if (wps_validate_m6(msg) < 0)
1269 			return WPS_FAILURE;
1270 		ret = wps_process_m6(wps, msg, &attr);
1271 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
1272 			wps_fail_event(wps->wps, WPS_M6, wps->config_error,
1273 				       wps->error_indication,
1274 				       wps->peer_dev.mac_addr);
1275 		break;
1276 	case WPS_M8:
1277 		if (wps_validate_m8(msg) < 0)
1278 			return WPS_FAILURE;
1279 		ret = wps_process_m8(wps, msg, &attr);
1280 		if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
1281 			wps_fail_event(wps->wps, WPS_M8, wps->config_error,
1282 				       wps->error_indication,
1283 				       wps->peer_dev.mac_addr);
1284 		break;
1285 	default:
1286 		wpa_printf(MSG_DEBUG, "WPS: Unsupported Message Type %d",
1287 			   *attr.msg_type);
1288 		return WPS_FAILURE;
1289 	}
1290 
1291 	/*
1292 	 * Save a copy of the last message for Authenticator derivation if we
1293 	 * are continuing. However, skip M2D since it is not authenticated and
1294 	 * neither is the ACK/NACK response frame. This allows the possibly
1295 	 * following M2 to be processed correctly by using the previously sent
1296 	 * M1 in Authenticator derivation.
1297 	 */
1298 	if (ret == WPS_CONTINUE && *attr.msg_type != WPS_M2D) {
1299 		/* Save a copy of the last message for Authenticator derivation
1300 		 */
1301 		wpabuf_free(wps->last_msg);
1302 		wps->last_msg = wpabuf_dup(msg);
1303 	}
1304 
1305 	return ret;
1306 }
1307 
1308 
1309 static enum wps_process_res wps_process_wsc_ack(struct wps_data *wps,
1310 						const struct wpabuf *msg)
1311 {
1312 	struct wps_parse_attr attr;
1313 
1314 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_ACK");
1315 
1316 	if (wps_parse_msg(msg, &attr) < 0)
1317 		return WPS_FAILURE;
1318 
1319 	if (attr.msg_type == NULL) {
1320 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
1321 		return WPS_FAILURE;
1322 	}
1323 
1324 	if (*attr.msg_type != WPS_WSC_ACK) {
1325 		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
1326 			   *attr.msg_type);
1327 		return WPS_FAILURE;
1328 	}
1329 
1330 	if (attr.registrar_nonce == NULL ||
1331 	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
1332 	{
1333 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
1334 		return WPS_FAILURE;
1335 	}
1336 
1337 	if (attr.enrollee_nonce == NULL ||
1338 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
1339 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
1340 		return WPS_FAILURE;
1341 	}
1342 
1343 	if (wps->state == RECV_ACK && wps->wps->ap) {
1344 		wpa_printf(MSG_DEBUG, "WPS: External Registrar registration "
1345 			   "completed successfully");
1346 		wps_success_event(wps->wps, wps->peer_dev.mac_addr);
1347 		wps->state = WPS_FINISHED;
1348 		return WPS_DONE;
1349 	}
1350 
1351 	return WPS_FAILURE;
1352 }
1353 
1354 
1355 static enum wps_process_res wps_process_wsc_nack(struct wps_data *wps,
1356 						 const struct wpabuf *msg)
1357 {
1358 	struct wps_parse_attr attr;
1359 	u16 config_error;
1360 
1361 	wpa_printf(MSG_DEBUG, "WPS: Received WSC_NACK");
1362 
1363 	if (wps_parse_msg(msg, &attr) < 0)
1364 		return WPS_FAILURE;
1365 
1366 	if (attr.msg_type == NULL) {
1367 		wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
1368 		return WPS_FAILURE;
1369 	}
1370 
1371 	if (*attr.msg_type != WPS_WSC_NACK) {
1372 		wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
1373 			   *attr.msg_type);
1374 		return WPS_FAILURE;
1375 	}
1376 
1377 	if (attr.registrar_nonce == NULL ||
1378 	    os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN) != 0)
1379 	{
1380 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
1381 		wpa_hexdump(MSG_DEBUG, "WPS: Received Registrar Nonce",
1382 			    attr.registrar_nonce, WPS_NONCE_LEN);
1383 		wpa_hexdump(MSG_DEBUG, "WPS: Expected Registrar Nonce",
1384 			    wps->nonce_r, WPS_NONCE_LEN);
1385 		return WPS_FAILURE;
1386 	}
1387 
1388 	if (attr.enrollee_nonce == NULL ||
1389 	    os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN) != 0) {
1390 		wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
1391 		wpa_hexdump(MSG_DEBUG, "WPS: Received Enrollee Nonce",
1392 			    attr.enrollee_nonce, WPS_NONCE_LEN);
1393 		wpa_hexdump(MSG_DEBUG, "WPS: Expected Enrollee Nonce",
1394 			    wps->nonce_e, WPS_NONCE_LEN);
1395 		return WPS_FAILURE;
1396 	}
1397 
1398 	if (attr.config_error == NULL) {
1399 		wpa_printf(MSG_DEBUG, "WPS: No Configuration Error attribute "
1400 			   "in WSC_NACK");
1401 		return WPS_FAILURE;
1402 	}
1403 
1404 	config_error = WPA_GET_BE16(attr.config_error);
1405 	wpa_printf(MSG_DEBUG, "WPS: Registrar terminated negotiation with "
1406 		   "Configuration Error %d", config_error);
1407 
1408 	switch (wps->state) {
1409 	case RECV_M4:
1410 		wps_fail_event(wps->wps, WPS_M3, config_error,
1411 			       wps->error_indication, wps->peer_dev.mac_addr);
1412 		break;
1413 	case RECV_M6:
1414 		wps_fail_event(wps->wps, WPS_M5, config_error,
1415 			       wps->error_indication, wps->peer_dev.mac_addr);
1416 		break;
1417 	case RECV_M8:
1418 		wps_fail_event(wps->wps, WPS_M7, config_error,
1419 			       wps->error_indication, wps->peer_dev.mac_addr);
1420 		break;
1421 	default:
1422 		break;
1423 	}
1424 
1425 	/* Followed by NACK if Enrollee is Supplicant or EAP-Failure if
1426 	 * Enrollee is Authenticator */
1427 	wps->state = SEND_WSC_NACK;
1428 
1429 	return WPS_FAILURE;
1430 }
1431 
1432 
1433 enum wps_process_res wps_enrollee_process_msg(struct wps_data *wps,
1434 					      enum wsc_op_code op_code,
1435 					      const struct wpabuf *msg)
1436 {
1437 
1438 	wpa_printf(MSG_DEBUG, "WPS: Processing received message (len=%lu "
1439 		   "op_code=%d)",
1440 		   (unsigned long) wpabuf_len(msg), op_code);
1441 
1442 	if (op_code == WSC_UPnP) {
1443 		/* Determine the OpCode based on message type attribute */
1444 		struct wps_parse_attr attr;
1445 		if (wps_parse_msg(msg, &attr) == 0 && attr.msg_type) {
1446 			if (*attr.msg_type == WPS_WSC_ACK)
1447 				op_code = WSC_ACK;
1448 			else if (*attr.msg_type == WPS_WSC_NACK)
1449 				op_code = WSC_NACK;
1450 		}
1451 	}
1452 
1453 	switch (op_code) {
1454 	case WSC_MSG:
1455 	case WSC_UPnP:
1456 		return wps_process_wsc_msg(wps, msg);
1457 	case WSC_ACK:
1458 		if (wps_validate_wsc_ack(msg) < 0)
1459 			return WPS_FAILURE;
1460 		return wps_process_wsc_ack(wps, msg);
1461 	case WSC_NACK:
1462 		if (wps_validate_wsc_nack(msg) < 0)
1463 			return WPS_FAILURE;
1464 		return wps_process_wsc_nack(wps, msg);
1465 	default:
1466 		wpa_printf(MSG_DEBUG, "WPS: Unsupported op_code %d", op_code);
1467 		return WPS_FAILURE;
1468 	}
1469 }
1470