1/* 2 * Wi-Fi Protected Setup - attribute building 3 * Copyright (c) 2008-2016, Jouni Malinen <j@w1.fi> 4 * 5 * This software may be distributed under the terms of the BSD license. 6 * See README for more details. 7 */ 8 9#include "includes.h" 10 11#include "common.h" 12#include "crypto/aes_wrap.h" 13#include "crypto/crypto.h" 14#include "crypto/dh_group5.h" 15#include "crypto/sha256.h" 16#include "crypto/random.h" 17#include "common/ieee802_11_defs.h" 18#include "wps_i.h" 19 20 21int wps_build_public_key(struct wps_data *wps, struct wpabuf *msg) 22{ 23 struct wpabuf *pubkey = NULL; 24 25 wpa_printf(MSG_DEBUG, "WPS: * Public Key"); 26 wpabuf_clear_free(wps->dh_privkey); 27 wps->dh_privkey = NULL; 28 if (wps->dev_pw_id != DEV_PW_DEFAULT && wps->wps->dh_privkey && 29 wps->wps->dh_ctx) { 30 wpa_printf(MSG_DEBUG, "WPS: Using pre-configured DH keys"); 31 if (wps->wps->dh_pubkey == NULL) { 32 wpa_printf(MSG_DEBUG, 33 "WPS: wps->wps->dh_pubkey == NULL"); 34 return -1; 35 } 36 wps->dh_privkey = wpabuf_dup(wps->wps->dh_privkey); 37 wps->dh_ctx = wps->wps->dh_ctx; 38 wps->wps->dh_ctx = NULL; 39 pubkey = wpabuf_dup(wps->wps->dh_pubkey); 40#ifdef CONFIG_WPS_NFC 41 } else if ((wps->dev_pw_id >= 0x10 || 42 wps->dev_pw_id == DEV_PW_NFC_CONNECTION_HANDOVER) && 43 (wps->wps->ap || 44 (wps->wps->ap_nfc_dh_pubkey && 45 wps->wps->ap_nfc_dev_pw_id == 46 DEV_PW_NFC_CONNECTION_HANDOVER && 47 wps->dev_pw_id == DEV_PW_NFC_CONNECTION_HANDOVER)) && 48 (wps->dev_pw_id == wps->wps->ap_nfc_dev_pw_id || 49 wps->wps->ap_nfc_dh_pubkey)) { 50 wpa_printf(MSG_DEBUG, "WPS: Using NFC password token DH keys"); 51 if (wps->wps->ap_nfc_dh_privkey == NULL) { 52 wpa_printf(MSG_DEBUG, 53 "WPS: wps->wps->ap_nfc_dh_privkey == NULL"); 54 return -1; 55 } 56 if (wps->wps->ap_nfc_dh_pubkey == NULL) { 57 wpa_printf(MSG_DEBUG, 58 "WPS: wps->wps->ap_nfc_dh_pubkey == NULL"); 59 return -1; 60 } 61 wps->dh_privkey = wpabuf_dup(wps->wps->ap_nfc_dh_privkey); 62 pubkey = wpabuf_dup(wps->wps->ap_nfc_dh_pubkey); 63 wps->dh_ctx = dh5_init_fixed(wps->dh_privkey, pubkey); 64#endif /* CONFIG_WPS_NFC */ 65 } else { 66 wpa_printf(MSG_DEBUG, "WPS: Generate new DH keys"); 67 dh5_free(wps->dh_ctx); 68 wps->dh_ctx = dh5_init(&wps->dh_privkey, &pubkey); 69 pubkey = wpabuf_zeropad(pubkey, 192); 70 } 71 if (wps->dh_ctx == NULL || wps->dh_privkey == NULL || pubkey == NULL) { 72 wpa_printf(MSG_DEBUG, "WPS: Failed to initialize " 73 "Diffie-Hellman handshake"); 74 wpabuf_free(pubkey); 75 return -1; 76 } 77 wpa_hexdump_buf_key(MSG_DEBUG, "WPS: DH Private Key", wps->dh_privkey); 78 wpa_hexdump_buf(MSG_DEBUG, "WPS: DH own Public Key", pubkey); 79 80 wpabuf_put_be16(msg, ATTR_PUBLIC_KEY); 81 wpabuf_put_be16(msg, wpabuf_len(pubkey)); 82 wpabuf_put_buf(msg, pubkey); 83 84 if (wps->registrar) { 85 wpabuf_free(wps->dh_pubkey_r); 86 wps->dh_pubkey_r = pubkey; 87 } else { 88 wpabuf_free(wps->dh_pubkey_e); 89 wps->dh_pubkey_e = pubkey; 90 } 91 92 return 0; 93} 94 95 96int wps_build_req_type(struct wpabuf *msg, enum wps_request_type type) 97{ 98 wpa_printf(MSG_DEBUG, "WPS: * Request Type"); 99 wpabuf_put_be16(msg, ATTR_REQUEST_TYPE); 100 wpabuf_put_be16(msg, 1); 101 wpabuf_put_u8(msg, type); 102 return 0; 103} 104 105 106int wps_build_resp_type(struct wpabuf *msg, enum wps_response_type type) 107{ 108 wpa_printf(MSG_DEBUG, "WPS: * Response Type (%d)", type); 109 wpabuf_put_be16(msg, ATTR_RESPONSE_TYPE); 110 wpabuf_put_be16(msg, 1); 111 wpabuf_put_u8(msg, type); 112 return 0; 113} 114 115 116int wps_build_config_methods(struct wpabuf *msg, u16 methods) 117{ 118 wpa_printf(MSG_DEBUG, "WPS: * Config Methods (%x)", methods); 119 wpabuf_put_be16(msg, ATTR_CONFIG_METHODS); 120 wpabuf_put_be16(msg, 2); 121 wpabuf_put_be16(msg, methods); 122 return 0; 123} 124 125 126int wps_build_uuid_e(struct wpabuf *msg, const u8 *uuid) 127{ 128 if (wpabuf_tailroom(msg) < 4 + WPS_UUID_LEN) 129 return -1; 130 wpa_printf(MSG_DEBUG, "WPS: * UUID-E"); 131 wpabuf_put_be16(msg, ATTR_UUID_E); 132 wpabuf_put_be16(msg, WPS_UUID_LEN); 133 wpabuf_put_data(msg, uuid, WPS_UUID_LEN); 134 return 0; 135} 136 137 138int wps_build_dev_password_id(struct wpabuf *msg, u16 id) 139{ 140 wpa_printf(MSG_DEBUG, "WPS: * Device Password ID (%d)", id); 141 wpabuf_put_be16(msg, ATTR_DEV_PASSWORD_ID); 142 wpabuf_put_be16(msg, 2); 143 wpabuf_put_be16(msg, id); 144 return 0; 145} 146 147 148int wps_build_config_error(struct wpabuf *msg, u16 err) 149{ 150 wpa_printf(MSG_DEBUG, "WPS: * Configuration Error (%d)", err); 151 wpabuf_put_be16(msg, ATTR_CONFIG_ERROR); 152 wpabuf_put_be16(msg, 2); 153 wpabuf_put_be16(msg, err); 154 return 0; 155} 156 157 158int wps_build_authenticator(struct wps_data *wps, struct wpabuf *msg) 159{ 160 u8 hash[SHA256_MAC_LEN]; 161 const u8 *addr[2]; 162 size_t len[2]; 163 164 if (wps->last_msg == NULL) { 165 wpa_printf(MSG_DEBUG, "WPS: Last message not available for " 166 "building authenticator"); 167 return -1; 168 } 169 170 /* Authenticator = HMAC-SHA256_AuthKey(M_prev || M_curr*) 171 * (M_curr* is M_curr without the Authenticator attribute) 172 */ 173 addr[0] = wpabuf_head(wps->last_msg); 174 len[0] = wpabuf_len(wps->last_msg); 175 addr[1] = wpabuf_head(msg); 176 len[1] = wpabuf_len(msg); 177 hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 2, addr, len, hash); 178 179 wpa_printf(MSG_DEBUG, "WPS: * Authenticator"); 180 wpabuf_put_be16(msg, ATTR_AUTHENTICATOR); 181 wpabuf_put_be16(msg, WPS_AUTHENTICATOR_LEN); 182 wpabuf_put_data(msg, hash, WPS_AUTHENTICATOR_LEN); 183 184 return 0; 185} 186 187 188int wps_build_version(struct wpabuf *msg) 189{ 190 /* 191 * Note: This attribute is deprecated and set to hardcoded 0x10 for 192 * backwards compatibility reasons. The real version negotiation is 193 * done with Version2. 194 */ 195 if (wpabuf_tailroom(msg) < 5) 196 return -1; 197 wpa_printf(MSG_DEBUG, "WPS: * Version (hardcoded 0x10)"); 198 wpabuf_put_be16(msg, ATTR_VERSION); 199 wpabuf_put_be16(msg, 1); 200 wpabuf_put_u8(msg, 0x10); 201 return 0; 202} 203 204 205int wps_build_wfa_ext(struct wpabuf *msg, int req_to_enroll, 206 const u8 *auth_macs, size_t auth_macs_count) 207{ 208 u8 *len; 209 210#ifdef CONFIG_WPS_TESTING 211 if (WPS_VERSION == 0x10) 212 return 0; 213#endif /* CONFIG_WPS_TESTING */ 214 215 if (wpabuf_tailroom(msg) < 216 7 + 3 + (req_to_enroll ? 3 : 0) + 217 (auth_macs ? 2 + auth_macs_count * ETH_ALEN : 0)) 218 return -1; 219 wpabuf_put_be16(msg, ATTR_VENDOR_EXT); 220 len = wpabuf_put(msg, 2); /* to be filled */ 221 wpabuf_put_be24(msg, WPS_VENDOR_ID_WFA); 222 223 wpa_printf(MSG_DEBUG, "WPS: * Version2 (0x%x)", WPS_VERSION); 224 wpabuf_put_u8(msg, WFA_ELEM_VERSION2); 225 wpabuf_put_u8(msg, 1); 226 wpabuf_put_u8(msg, WPS_VERSION); 227 228 if (req_to_enroll) { 229 wpa_printf(MSG_DEBUG, "WPS: * Request to Enroll (1)"); 230 wpabuf_put_u8(msg, WFA_ELEM_REQUEST_TO_ENROLL); 231 wpabuf_put_u8(msg, 1); 232 wpabuf_put_u8(msg, 1); 233 } 234 235 if (auth_macs && auth_macs_count) { 236 size_t i; 237 wpa_printf(MSG_DEBUG, "WPS: * AuthorizedMACs (count=%d)", 238 (int) auth_macs_count); 239 wpabuf_put_u8(msg, WFA_ELEM_AUTHORIZEDMACS); 240 wpabuf_put_u8(msg, auth_macs_count * ETH_ALEN); 241 wpabuf_put_data(msg, auth_macs, auth_macs_count * ETH_ALEN); 242 for (i = 0; i < auth_macs_count; i++) 243 wpa_printf(MSG_DEBUG, "WPS: AuthorizedMAC: " MACSTR, 244 MAC2STR(&auth_macs[i * ETH_ALEN])); 245 } 246 247 WPA_PUT_BE16(len, (u8 *) wpabuf_put(msg, 0) - len - 2); 248 249#ifdef CONFIG_WPS_TESTING 250 if (WPS_VERSION > 0x20) { 251 if (wpabuf_tailroom(msg) < 5) 252 return -1; 253 wpa_printf(MSG_DEBUG, "WPS: * Extensibility Testing - extra " 254 "attribute"); 255 wpabuf_put_be16(msg, ATTR_EXTENSIBILITY_TEST); 256 wpabuf_put_be16(msg, 1); 257 wpabuf_put_u8(msg, 42); 258 } 259#endif /* CONFIG_WPS_TESTING */ 260 return 0; 261} 262 263 264int wps_build_msg_type(struct wpabuf *msg, enum wps_msg_type msg_type) 265{ 266 wpa_printf(MSG_DEBUG, "WPS: * Message Type (%d)", msg_type); 267 wpabuf_put_be16(msg, ATTR_MSG_TYPE); 268 wpabuf_put_be16(msg, 1); 269 wpabuf_put_u8(msg, msg_type); 270 return 0; 271} 272 273 274int wps_build_enrollee_nonce(struct wps_data *wps, struct wpabuf *msg) 275{ 276 wpa_printf(MSG_DEBUG, "WPS: * Enrollee Nonce"); 277 wpabuf_put_be16(msg, ATTR_ENROLLEE_NONCE); 278 wpabuf_put_be16(msg, WPS_NONCE_LEN); 279 wpabuf_put_data(msg, wps->nonce_e, WPS_NONCE_LEN); 280 return 0; 281} 282 283 284int wps_build_registrar_nonce(struct wps_data *wps, struct wpabuf *msg) 285{ 286 wpa_printf(MSG_DEBUG, "WPS: * Registrar Nonce"); 287 wpabuf_put_be16(msg, ATTR_REGISTRAR_NONCE); 288 wpabuf_put_be16(msg, WPS_NONCE_LEN); 289 wpabuf_put_data(msg, wps->nonce_r, WPS_NONCE_LEN); 290 return 0; 291} 292 293 294int wps_build_auth_type_flags(struct wps_data *wps, struct wpabuf *msg) 295{ 296 u16 auth_types = WPS_AUTH_TYPES; 297 /* WPA/WPA2-Enterprise enrollment not supported through WPS */ 298 auth_types &= ~WPS_AUTH_WPA; 299 auth_types &= ~WPS_AUTH_WPA2; 300 auth_types &= ~WPS_AUTH_SHARED; 301#ifdef CONFIG_WPS_TESTING 302 if (wps_force_auth_types_in_use) { 303 wpa_printf(MSG_DEBUG, 304 "WPS: Testing - replace auth type 0x%x with 0x%x", 305 auth_types, wps_force_auth_types); 306 auth_types = wps_force_auth_types; 307 } 308#endif /* CONFIG_WPS_TESTING */ 309 wpa_printf(MSG_DEBUG, "WPS: * Authentication Type Flags (0x%x)", 310 auth_types); 311 wpabuf_put_be16(msg, ATTR_AUTH_TYPE_FLAGS); 312 wpabuf_put_be16(msg, 2); 313 wpabuf_put_be16(msg, auth_types); 314 return 0; 315} 316 317 318int wps_build_encr_type_flags(struct wps_data *wps, struct wpabuf *msg) 319{ 320 u16 encr_types = WPS_ENCR_TYPES; 321 encr_types &= ~WPS_ENCR_WEP; 322#ifdef CONFIG_WPS_TESTING 323 if (wps_force_encr_types_in_use) { 324 wpa_printf(MSG_DEBUG, 325 "WPS: Testing - replace encr type 0x%x with 0x%x", 326 encr_types, wps_force_encr_types); 327 encr_types = wps_force_encr_types; 328 } 329#endif /* CONFIG_WPS_TESTING */ 330 wpa_printf(MSG_DEBUG, "WPS: * Encryption Type Flags (0x%x)", 331 encr_types); 332 wpabuf_put_be16(msg, ATTR_ENCR_TYPE_FLAGS); 333 wpabuf_put_be16(msg, 2); 334 wpabuf_put_be16(msg, encr_types); 335 return 0; 336} 337 338 339int wps_build_conn_type_flags(struct wps_data *wps, struct wpabuf *msg) 340{ 341 wpa_printf(MSG_DEBUG, "WPS: * Connection Type Flags"); 342 wpabuf_put_be16(msg, ATTR_CONN_TYPE_FLAGS); 343 wpabuf_put_be16(msg, 1); 344 wpabuf_put_u8(msg, WPS_CONN_ESS); 345 return 0; 346} 347 348 349int wps_build_assoc_state(struct wps_data *wps, struct wpabuf *msg) 350{ 351 wpa_printf(MSG_DEBUG, "WPS: * Association State"); 352 wpabuf_put_be16(msg, ATTR_ASSOC_STATE); 353 wpabuf_put_be16(msg, 2); 354 wpabuf_put_be16(msg, WPS_ASSOC_NOT_ASSOC); 355 return 0; 356} 357 358 359int wps_build_key_wrap_auth(struct wps_data *wps, struct wpabuf *msg) 360{ 361 u8 hash[SHA256_MAC_LEN]; 362 363 wpa_printf(MSG_DEBUG, "WPS: * Key Wrap Authenticator"); 364 hmac_sha256(wps->authkey, WPS_AUTHKEY_LEN, wpabuf_head(msg), 365 wpabuf_len(msg), hash); 366 367 wpabuf_put_be16(msg, ATTR_KEY_WRAP_AUTH); 368 wpabuf_put_be16(msg, WPS_KWA_LEN); 369 wpabuf_put_data(msg, hash, WPS_KWA_LEN); 370 return 0; 371} 372 373 374int wps_build_encr_settings(struct wps_data *wps, struct wpabuf *msg, 375 struct wpabuf *plain) 376{ 377 size_t pad_len; 378 const size_t block_size = 16; 379 u8 *iv, *data; 380 381 wpa_printf(MSG_DEBUG, "WPS: * Encrypted Settings"); 382 383 /* PKCS#5 v2.0 pad */ 384 pad_len = block_size - wpabuf_len(plain) % block_size; 385 os_memset(wpabuf_put(plain, pad_len), pad_len, pad_len); 386 387 wpabuf_put_be16(msg, ATTR_ENCR_SETTINGS); 388 wpabuf_put_be16(msg, block_size + wpabuf_len(plain)); 389 390 iv = wpabuf_put(msg, block_size); 391 if (random_get_bytes(iv, block_size) < 0) 392 return -1; 393 394 data = wpabuf_put(msg, 0); 395 wpabuf_put_buf(msg, plain); 396 if (aes_128_cbc_encrypt(wps->keywrapkey, iv, data, wpabuf_len(plain))) 397 return -1; 398 399 return 0; 400} 401 402 403#ifdef CONFIG_WPS_OOB 404int wps_build_oob_dev_pw(struct wpabuf *msg, u16 dev_pw_id, 405 const struct wpabuf *pubkey, const u8 *dev_pw, 406 size_t dev_pw_len) 407{ 408 size_t hash_len; 409 const u8 *addr[1]; 410 u8 pubkey_hash[WPS_HASH_LEN]; 411 412 wpa_printf(MSG_DEBUG, "WPS: * OOB Device Password (dev_pw_id=%u)", 413 dev_pw_id); 414 addr[0] = wpabuf_head(pubkey); 415 hash_len = wpabuf_len(pubkey); 416 if (sha256_vector(1, addr, &hash_len, pubkey_hash) < 0) 417 return -1; 418#ifdef CONFIG_WPS_TESTING 419 if (wps_corrupt_pkhash) { 420 wpa_hexdump(MSG_DEBUG, "WPS: Real Public Key Hash", 421 pubkey_hash, WPS_OOB_PUBKEY_HASH_LEN); 422 wpa_printf(MSG_INFO, "WPS: Testing - corrupt public key hash"); 423 pubkey_hash[WPS_OOB_PUBKEY_HASH_LEN - 2]++; 424 } 425#endif /* CONFIG_WPS_TESTING */ 426 427 wpabuf_put_be16(msg, ATTR_OOB_DEVICE_PASSWORD); 428 wpabuf_put_be16(msg, WPS_OOB_PUBKEY_HASH_LEN + 2 + dev_pw_len); 429 wpa_hexdump(MSG_DEBUG, "WPS: Public Key Hash", 430 pubkey_hash, WPS_OOB_PUBKEY_HASH_LEN); 431 wpabuf_put_data(msg, pubkey_hash, WPS_OOB_PUBKEY_HASH_LEN); 432 wpabuf_put_be16(msg, dev_pw_id); 433 if (dev_pw) { 434 wpa_hexdump_key(MSG_DEBUG, "WPS: OOB Device Password", 435 dev_pw, dev_pw_len); 436 wpabuf_put_data(msg, dev_pw, dev_pw_len); 437 } 438 439 return 0; 440} 441#endif /* CONFIG_WPS_OOB */ 442 443 444/* Encapsulate WPS IE data with one (or more, if needed) IE headers */ 445struct wpabuf * wps_ie_encapsulate(struct wpabuf *data) 446{ 447 struct wpabuf *ie; 448 const u8 *pos, *end; 449 450 ie = wpabuf_alloc(wpabuf_len(data) + 100); 451 if (ie == NULL) { 452 wpabuf_free(data); 453 return NULL; 454 } 455 456 pos = wpabuf_head(data); 457 end = pos + wpabuf_len(data); 458 459 while (end > pos) { 460 size_t frag_len = end - pos; 461 if (frag_len > 251) 462 frag_len = 251; 463 wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC); 464 wpabuf_put_u8(ie, 4 + frag_len); 465 wpabuf_put_be32(ie, WPS_DEV_OUI_WFA); 466 wpabuf_put_data(ie, pos, frag_len); 467 pos += frag_len; 468 } 469 470 wpabuf_free(data); 471 472 return ie; 473} 474 475 476int wps_build_mac_addr(struct wpabuf *msg, const u8 *addr) 477{ 478 wpa_printf(MSG_DEBUG, "WPS: * MAC Address (" MACSTR ")", 479 MAC2STR(addr)); 480 wpabuf_put_be16(msg, ATTR_MAC_ADDR); 481 wpabuf_put_be16(msg, ETH_ALEN); 482 wpabuf_put_data(msg, addr, ETH_ALEN); 483 return 0; 484} 485 486 487int wps_build_rf_bands_attr(struct wpabuf *msg, u8 rf_bands) 488{ 489 wpa_printf(MSG_DEBUG, "WPS: * RF Bands (%x)", rf_bands); 490 wpabuf_put_be16(msg, ATTR_RF_BANDS); 491 wpabuf_put_be16(msg, 1); 492 wpabuf_put_u8(msg, rf_bands); 493 return 0; 494} 495 496 497int wps_build_ap_channel(struct wpabuf *msg, u16 ap_channel) 498{ 499 wpa_printf(MSG_DEBUG, "WPS: * AP Channel (%u)", ap_channel); 500 wpabuf_put_be16(msg, ATTR_AP_CHANNEL); 501 wpabuf_put_be16(msg, 2); 502 wpabuf_put_be16(msg, ap_channel); 503 return 0; 504} 505