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