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