1 /*
2  * Wi-Fi Protected Setup - common functionality
3  * Copyright (c) 2008-2012, 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 "common/defs.h"
13 #include "common/ieee802_11_common.h"
14 #include "crypto/aes_wrap.h"
15 #include "crypto/crypto.h"
16 #include "crypto/dh_group5.h"
17 #include "crypto/sha1.h"
18 #include "crypto/sha256.h"
19 #include "crypto/random.h"
20 #include "wps_i.h"
21 #include "wps_dev_attr.h"
22 
23 
24 void wps_kdf(const u8 *key, const u8 *label_prefix, size_t label_prefix_len,
25 	     const char *label, u8 *res, size_t res_len)
26 {
27 	u8 i_buf[4], key_bits[4];
28 	const u8 *addr[4];
29 	size_t len[4];
30 	int i, iter;
31 	u8 hash[SHA256_MAC_LEN], *opos;
32 	size_t left;
33 
34 	WPA_PUT_BE32(key_bits, res_len * 8);
35 
36 	addr[0] = i_buf;
37 	len[0] = sizeof(i_buf);
38 	addr[1] = label_prefix;
39 	len[1] = label_prefix_len;
40 	addr[2] = (const u8 *) label;
41 	len[2] = os_strlen(label);
42 	addr[3] = key_bits;
43 	len[3] = sizeof(key_bits);
44 
45 	iter = (res_len + SHA256_MAC_LEN - 1) / SHA256_MAC_LEN;
46 	opos = res;
47 	left = res_len;
48 
49 	for (i = 1; i <= iter; i++) {
50 		WPA_PUT_BE32(i_buf, i);
51 		hmac_sha256_vector(key, SHA256_MAC_LEN, 4, addr, len, hash);
52 		if (i < iter) {
53 			os_memcpy(opos, hash, SHA256_MAC_LEN);
54 			opos += SHA256_MAC_LEN;
55 			left -= SHA256_MAC_LEN;
56 		} else
57 			os_memcpy(opos, hash, left);
58 	}
59 }
60 
61 
62 int wps_derive_keys(struct wps_data *wps)
63 {
64 	struct wpabuf *pubkey, *dh_shared;
65 	u8 dhkey[SHA256_MAC_LEN], kdk[SHA256_MAC_LEN];
66 	const u8 *addr[3];
67 	size_t len[3];
68 	u8 keys[WPS_AUTHKEY_LEN + WPS_KEYWRAPKEY_LEN + WPS_EMSK_LEN];
69 
70 	if (wps->dh_privkey == NULL) {
71 		wpa_printf(MSG_DEBUG, "WPS: Own DH private key not available");
72 		return -1;
73 	}
74 
75 	pubkey = wps->registrar ? wps->dh_pubkey_e : wps->dh_pubkey_r;
76 	if (pubkey == NULL) {
77 		wpa_printf(MSG_DEBUG, "WPS: Peer DH public key not available");
78 		return -1;
79 	}
80 
81 	wpa_hexdump_buf_key(MSG_DEBUG, "WPS: DH Private Key", wps->dh_privkey);
82 	wpa_hexdump_buf(MSG_DEBUG, "WPS: DH peer Public Key", pubkey);
83 	dh_shared = dh5_derive_shared(wps->dh_ctx, pubkey, wps->dh_privkey);
84 	dh5_free(wps->dh_ctx);
85 	wps->dh_ctx = NULL;
86 	dh_shared = wpabuf_zeropad(dh_shared, 192);
87 	if (dh_shared == NULL) {
88 		wpa_printf(MSG_DEBUG, "WPS: Failed to derive DH shared key");
89 		return -1;
90 	}
91 
92 	/* Own DH private key is not needed anymore */
93 	wpabuf_free(wps->dh_privkey);
94 	wps->dh_privkey = NULL;
95 
96 	wpa_hexdump_buf_key(MSG_DEBUG, "WPS: DH shared key", dh_shared);
97 
98 	/* DHKey = SHA-256(g^AB mod p) */
99 	addr[0] = wpabuf_head(dh_shared);
100 	len[0] = wpabuf_len(dh_shared);
101 	sha256_vector(1, addr, len, dhkey);
102 	wpa_hexdump_key(MSG_DEBUG, "WPS: DHKey", dhkey, sizeof(dhkey));
103 	wpabuf_free(dh_shared);
104 
105 	/* KDK = HMAC-SHA-256_DHKey(N1 || EnrolleeMAC || N2) */
106 	addr[0] = wps->nonce_e;
107 	len[0] = WPS_NONCE_LEN;
108 	addr[1] = wps->mac_addr_e;
109 	len[1] = ETH_ALEN;
110 	addr[2] = wps->nonce_r;
111 	len[2] = WPS_NONCE_LEN;
112 	hmac_sha256_vector(dhkey, sizeof(dhkey), 3, addr, len, kdk);
113 	wpa_hexdump_key(MSG_DEBUG, "WPS: KDK", kdk, sizeof(kdk));
114 
115 	wps_kdf(kdk, NULL, 0, "Wi-Fi Easy and Secure Key Derivation",
116 		keys, sizeof(keys));
117 	os_memcpy(wps->authkey, keys, WPS_AUTHKEY_LEN);
118 	os_memcpy(wps->keywrapkey, keys + WPS_AUTHKEY_LEN, WPS_KEYWRAPKEY_LEN);
119 	os_memcpy(wps->emsk, keys + WPS_AUTHKEY_LEN + WPS_KEYWRAPKEY_LEN,
120 		  WPS_EMSK_LEN);
121 
122 	wpa_hexdump_key(MSG_DEBUG, "WPS: AuthKey",
123 			wps->authkey, WPS_AUTHKEY_LEN);
124 	wpa_hexdump_key(MSG_DEBUG, "WPS: KeyWrapKey",
125 			wps->keywrapkey, WPS_KEYWRAPKEY_LEN);
126 	wpa_hexdump_key(MSG_DEBUG, "WPS: EMSK", wps->emsk, WPS_EMSK_LEN);
127 
128 	return 0;
129 }
130 
131 
132 void wps_derive_psk(struct wps_data *wps, const u8 *dev_passwd,
133 		    size_t dev_passwd_len)
134 {
135 	u8 hash[SHA256_MAC_LEN];
136 
137 	hmac_sha256(wps->authkey, WPS_AUTHKEY_LEN, dev_passwd,
138 		    (dev_passwd_len + 1) / 2, hash);
139 	os_memcpy(wps->psk1, hash, WPS_PSK_LEN);
140 	hmac_sha256(wps->authkey, WPS_AUTHKEY_LEN,
141 		    dev_passwd + (dev_passwd_len + 1) / 2,
142 		    dev_passwd_len / 2, hash);
143 	os_memcpy(wps->psk2, hash, WPS_PSK_LEN);
144 
145 	wpa_hexdump_ascii_key(MSG_DEBUG, "WPS: Device Password",
146 			      dev_passwd, dev_passwd_len);
147 	wpa_hexdump_key(MSG_DEBUG, "WPS: PSK1", wps->psk1, WPS_PSK_LEN);
148 	wpa_hexdump_key(MSG_DEBUG, "WPS: PSK2", wps->psk2, WPS_PSK_LEN);
149 }
150 
151 
152 struct wpabuf * wps_decrypt_encr_settings(struct wps_data *wps, const u8 *encr,
153 					  size_t encr_len)
154 {
155 	struct wpabuf *decrypted;
156 	const size_t block_size = 16;
157 	size_t i;
158 	u8 pad;
159 	const u8 *pos;
160 
161 	/* AES-128-CBC */
162 	if (encr == NULL || encr_len < 2 * block_size || encr_len % block_size)
163 	{
164 		wpa_printf(MSG_DEBUG, "WPS: No Encrypted Settings received");
165 		return NULL;
166 	}
167 
168 	decrypted = wpabuf_alloc(encr_len - block_size);
169 	if (decrypted == NULL)
170 		return NULL;
171 
172 	wpa_hexdump(MSG_MSGDUMP, "WPS: Encrypted Settings", encr, encr_len);
173 	wpabuf_put_data(decrypted, encr + block_size, encr_len - block_size);
174 	if (aes_128_cbc_decrypt(wps->keywrapkey, encr, wpabuf_mhead(decrypted),
175 				wpabuf_len(decrypted))) {
176 		wpabuf_free(decrypted);
177 		return NULL;
178 	}
179 
180 	wpa_hexdump_buf_key(MSG_MSGDUMP, "WPS: Decrypted Encrypted Settings",
181 			    decrypted);
182 
183 	pos = wpabuf_head_u8(decrypted) + wpabuf_len(decrypted) - 1;
184 	pad = *pos;
185 	if (pad > wpabuf_len(decrypted)) {
186 		wpa_printf(MSG_DEBUG, "WPS: Invalid PKCS#5 v2.0 pad value");
187 		wpabuf_free(decrypted);
188 		return NULL;
189 	}
190 	for (i = 0; i < pad; i++) {
191 		if (*pos-- != pad) {
192 			wpa_printf(MSG_DEBUG, "WPS: Invalid PKCS#5 v2.0 pad "
193 				   "string");
194 			wpabuf_free(decrypted);
195 			return NULL;
196 		}
197 	}
198 	decrypted->used -= pad;
199 
200 	return decrypted;
201 }
202 
203 
204 /**
205  * wps_pin_checksum - Compute PIN checksum
206  * @pin: Seven digit PIN (i.e., eight digit PIN without the checksum digit)
207  * Returns: Checksum digit
208  */
209 unsigned int wps_pin_checksum(unsigned int pin)
210 {
211 	unsigned int accum = 0;
212 	while (pin) {
213 		accum += 3 * (pin % 10);
214 		pin /= 10;
215 		accum += pin % 10;
216 		pin /= 10;
217 	}
218 
219 	return (10 - accum % 10) % 10;
220 }
221 
222 
223 /**
224  * wps_pin_valid - Check whether a PIN has a valid checksum
225  * @pin: Eight digit PIN (i.e., including the checksum digit)
226  * Returns: 1 if checksum digit is valid, or 0 if not
227  */
228 unsigned int wps_pin_valid(unsigned int pin)
229 {
230 	return wps_pin_checksum(pin / 10) == (pin % 10);
231 }
232 
233 
234 /**
235  * wps_generate_pin - Generate a random PIN
236  * Returns: Eight digit PIN (i.e., including the checksum digit)
237  */
238 unsigned int wps_generate_pin(void)
239 {
240 	unsigned int val;
241 
242 	/* Generate seven random digits for the PIN */
243 	if (random_get_bytes((unsigned char *) &val, sizeof(val)) < 0) {
244 		struct os_time now;
245 		os_get_time(&now);
246 		val = os_random() ^ now.sec ^ now.usec;
247 	}
248 	val %= 10000000;
249 
250 	/* Append checksum digit */
251 	return val * 10 + wps_pin_checksum(val);
252 }
253 
254 
255 int wps_pin_str_valid(const char *pin)
256 {
257 	const char *p;
258 	size_t len;
259 
260 	p = pin;
261 	while (*p >= '0' && *p <= '9')
262 		p++;
263 	if (*p != '\0')
264 		return 0;
265 
266 	len = p - pin;
267 	return len == 4 || len == 8;
268 }
269 
270 
271 void wps_fail_event(struct wps_context *wps, enum wps_msg_type msg,
272 		    u16 config_error, u16 error_indication, const u8 *mac_addr)
273 {
274 	union wps_event_data data;
275 
276 	if (wps->event_cb == NULL)
277 		return;
278 
279 	os_memset(&data, 0, sizeof(data));
280 	data.fail.msg = msg;
281 	data.fail.config_error = config_error;
282 	data.fail.error_indication = error_indication;
283 	os_memcpy(data.fail.peer_macaddr, mac_addr, ETH_ALEN);
284 	wps->event_cb(wps->cb_ctx, WPS_EV_FAIL, &data);
285 }
286 
287 
288 void wps_success_event(struct wps_context *wps, const u8 *mac_addr)
289 {
290 	union wps_event_data data;
291 
292 	if (wps->event_cb == NULL)
293 		return;
294 
295 	os_memset(&data, 0, sizeof(data));
296 	os_memcpy(data.success.peer_macaddr, mac_addr, ETH_ALEN);
297 	wps->event_cb(wps->cb_ctx, WPS_EV_SUCCESS, &data);
298 }
299 
300 
301 void wps_pwd_auth_fail_event(struct wps_context *wps, int enrollee, int part,
302 			     const u8 *mac_addr)
303 {
304 	union wps_event_data data;
305 
306 	if (wps->event_cb == NULL)
307 		return;
308 
309 	os_memset(&data, 0, sizeof(data));
310 	data.pwd_auth_fail.enrollee = enrollee;
311 	data.pwd_auth_fail.part = part;
312 	os_memcpy(data.pwd_auth_fail.peer_macaddr, mac_addr, ETH_ALEN);
313 	wps->event_cb(wps->cb_ctx, WPS_EV_PWD_AUTH_FAIL, &data);
314 }
315 
316 
317 void wps_pbc_overlap_event(struct wps_context *wps)
318 {
319 	if (wps->event_cb == NULL)
320 		return;
321 
322 	wps->event_cb(wps->cb_ctx, WPS_EV_PBC_OVERLAP, NULL);
323 }
324 
325 
326 void wps_pbc_timeout_event(struct wps_context *wps)
327 {
328 	if (wps->event_cb == NULL)
329 		return;
330 
331 	wps->event_cb(wps->cb_ctx, WPS_EV_PBC_TIMEOUT, NULL);
332 }
333 
334 
335 void wps_pbc_active_event(struct wps_context *wps)
336 {
337 	if (wps->event_cb == NULL)
338 		return;
339 
340 	wps->event_cb(wps->cb_ctx, WPS_EV_PBC_ACTIVE, NULL);
341 }
342 
343 
344 void wps_pbc_disable_event(struct wps_context *wps)
345 {
346 	if (wps->event_cb == NULL)
347 		return;
348 
349 	wps->event_cb(wps->cb_ctx, WPS_EV_PBC_DISABLE, NULL);
350 }
351 
352 
353 #ifdef CONFIG_WPS_OOB
354 
355 struct wpabuf * wps_get_oob_cred(struct wps_context *wps, int rf_band,
356 				 int channel)
357 {
358 	struct wps_data data;
359 	struct wpabuf *plain;
360 
361 	plain = wpabuf_alloc(500);
362 	if (plain == NULL) {
363 		wpa_printf(MSG_ERROR, "WPS: Failed to allocate memory for OOB "
364 			   "credential");
365 		return NULL;
366 	}
367 
368 	os_memset(&data, 0, sizeof(data));
369 	data.wps = wps;
370 	data.auth_type = wps->auth_types;
371 	data.encr_type = wps->encr_types;
372 	if (wps_build_cred(&data, plain) ||
373 	    (rf_band && wps_build_rf_bands_attr(plain, rf_band)) ||
374 	    (channel && wps_build_ap_channel(plain, channel)) ||
375 	    wps_build_mac_addr(plain, wps->dev.mac_addr) ||
376 	    wps_build_wfa_ext(plain, 0, NULL, 0)) {
377 		os_free(data.new_psk);
378 		wpabuf_free(plain);
379 		return NULL;
380 	}
381 
382 	if (wps->wps_state == WPS_STATE_NOT_CONFIGURED && data.new_psk &&
383 	    wps->ap) {
384 		struct wps_credential cred;
385 
386 		wpa_printf(MSG_DEBUG, "WPS: Moving to Configured state based "
387 			   "on credential token generation");
388 
389 		os_memset(&cred, 0, sizeof(cred));
390 		os_memcpy(cred.ssid, wps->ssid, wps->ssid_len);
391 		cred.ssid_len = wps->ssid_len;
392 		cred.auth_type = WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK;
393 		cred.encr_type = WPS_ENCR_TKIP | WPS_ENCR_AES;
394 		os_memcpy(cred.key, data.new_psk, data.new_psk_len);
395 		cred.key_len = data.new_psk_len;
396 
397 		wps->wps_state = WPS_STATE_CONFIGURED;
398 		wpa_hexdump_ascii_key(MSG_DEBUG,
399 				      "WPS: Generated random passphrase",
400 				      data.new_psk, data.new_psk_len);
401 		if (wps->cred_cb)
402 			wps->cred_cb(wps->cb_ctx, &cred);
403 	}
404 
405 	os_free(data.new_psk);
406 
407 	return plain;
408 }
409 
410 
411 struct wpabuf * wps_build_nfc_pw_token(u16 dev_pw_id,
412 				       const struct wpabuf *pubkey,
413 				       const struct wpabuf *dev_pw)
414 {
415 	struct wpabuf *data;
416 
417 	data = wpabuf_alloc(200);
418 	if (data == NULL)
419 		return NULL;
420 
421 	if (wps_build_oob_dev_pw(data, dev_pw_id, pubkey,
422 				 wpabuf_head(dev_pw), wpabuf_len(dev_pw)) ||
423 	    wps_build_wfa_ext(data, 0, NULL, 0)) {
424 		wpa_printf(MSG_ERROR, "WPS: Failed to build NFC password "
425 			   "token");
426 		wpabuf_free(data);
427 		return NULL;
428 	}
429 
430 	return data;
431 }
432 
433 
434 int wps_oob_use_cred(struct wps_context *wps, struct wps_parse_attr *attr)
435 {
436 	struct wpabuf msg;
437 	size_t i;
438 
439 	for (i = 0; i < attr->num_cred; i++) {
440 		struct wps_credential local_cred;
441 		struct wps_parse_attr cattr;
442 
443 		os_memset(&local_cred, 0, sizeof(local_cred));
444 		wpabuf_set(&msg, attr->cred[i], attr->cred_len[i]);
445 		if (wps_parse_msg(&msg, &cattr) < 0 ||
446 		    wps_process_cred(&cattr, &local_cred)) {
447 			wpa_printf(MSG_ERROR, "WPS: Failed to parse OOB "
448 				   "credential");
449 			return -1;
450 		}
451 		wps->cred_cb(wps->cb_ctx, &local_cred);
452 	}
453 
454 	return 0;
455 }
456 
457 
458 #endif /* CONFIG_WPS_OOB */
459 
460 
461 int wps_dev_type_str2bin(const char *str, u8 dev_type[WPS_DEV_TYPE_LEN])
462 {
463 	const char *pos;
464 
465 	/* <categ>-<OUI>-<subcateg> */
466 	WPA_PUT_BE16(dev_type, atoi(str));
467 	pos = os_strchr(str, '-');
468 	if (pos == NULL)
469 		return -1;
470 	pos++;
471 	if (hexstr2bin(pos, &dev_type[2], 4))
472 		return -1;
473 	pos = os_strchr(pos, '-');
474 	if (pos == NULL)
475 		return -1;
476 	pos++;
477 	WPA_PUT_BE16(&dev_type[6], atoi(pos));
478 
479 
480 	return 0;
481 }
482 
483 
484 char * wps_dev_type_bin2str(const u8 dev_type[WPS_DEV_TYPE_LEN], char *buf,
485 			    size_t buf_len)
486 {
487 	int ret;
488 
489 	ret = os_snprintf(buf, buf_len, "%u-%08X-%u",
490 			  WPA_GET_BE16(dev_type), WPA_GET_BE32(&dev_type[2]),
491 			  WPA_GET_BE16(&dev_type[6]));
492 	if (ret < 0 || (unsigned int) ret >= buf_len)
493 		return NULL;
494 
495 	return buf;
496 }
497 
498 
499 void uuid_gen_mac_addr(const u8 *mac_addr, u8 *uuid)
500 {
501 	const u8 *addr[2];
502 	size_t len[2];
503 	u8 hash[SHA1_MAC_LEN];
504 	u8 nsid[16] = {
505 		0x52, 0x64, 0x80, 0xf8,
506 		0xc9, 0x9b,
507 		0x4b, 0xe5,
508 		0xa6, 0x55,
509 		0x58, 0xed, 0x5f, 0x5d, 0x60, 0x84
510 	};
511 
512 	addr[0] = nsid;
513 	len[0] = sizeof(nsid);
514 	addr[1] = mac_addr;
515 	len[1] = 6;
516 	sha1_vector(2, addr, len, hash);
517 	os_memcpy(uuid, hash, 16);
518 
519 	/* Version: 5 = named-based version using SHA-1 */
520 	uuid[6] = (5 << 4) | (uuid[6] & 0x0f);
521 
522 	/* Variant specified in RFC 4122 */
523 	uuid[8] = 0x80 | (uuid[8] & 0x3f);
524 }
525 
526 
527 u16 wps_config_methods_str2bin(const char *str)
528 {
529 	u16 methods = 0;
530 
531 	if (str == NULL) {
532 		/* Default to enabling methods based on build configuration */
533 		methods |= WPS_CONFIG_DISPLAY | WPS_CONFIG_KEYPAD;
534 #ifdef CONFIG_WPS2
535 		methods |= WPS_CONFIG_VIRT_DISPLAY;
536 #endif /* CONFIG_WPS2 */
537 #ifdef CONFIG_WPS_NFC
538 		methods |= WPS_CONFIG_NFC_INTERFACE;
539 #endif /* CONFIG_WPS_NFC */
540 	} else {
541 		if (os_strstr(str, "ethernet"))
542 			methods |= WPS_CONFIG_ETHERNET;
543 		if (os_strstr(str, "label"))
544 			methods |= WPS_CONFIG_LABEL;
545 		if (os_strstr(str, "display"))
546 			methods |= WPS_CONFIG_DISPLAY;
547 		if (os_strstr(str, "ext_nfc_token"))
548 			methods |= WPS_CONFIG_EXT_NFC_TOKEN;
549 		if (os_strstr(str, "int_nfc_token"))
550 			methods |= WPS_CONFIG_INT_NFC_TOKEN;
551 		if (os_strstr(str, "nfc_interface"))
552 			methods |= WPS_CONFIG_NFC_INTERFACE;
553 		if (os_strstr(str, "push_button"))
554 			methods |= WPS_CONFIG_PUSHBUTTON;
555 		if (os_strstr(str, "keypad"))
556 			methods |= WPS_CONFIG_KEYPAD;
557 #ifdef CONFIG_WPS2
558 		if (os_strstr(str, "virtual_display"))
559 			methods |= WPS_CONFIG_VIRT_DISPLAY;
560 		if (os_strstr(str, "physical_display"))
561 			methods |= WPS_CONFIG_PHY_DISPLAY;
562 		if (os_strstr(str, "virtual_push_button"))
563 			methods |= WPS_CONFIG_VIRT_PUSHBUTTON;
564 		if (os_strstr(str, "physical_push_button"))
565 			methods |= WPS_CONFIG_PHY_PUSHBUTTON;
566 #endif /* CONFIG_WPS2 */
567 	}
568 
569 	return methods;
570 }
571 
572 
573 struct wpabuf * wps_build_wsc_ack(struct wps_data *wps)
574 {
575 	struct wpabuf *msg;
576 
577 	wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_ACK");
578 
579 	msg = wpabuf_alloc(1000);
580 	if (msg == NULL)
581 		return NULL;
582 
583 	if (wps_build_version(msg) ||
584 	    wps_build_msg_type(msg, WPS_WSC_ACK) ||
585 	    wps_build_enrollee_nonce(wps, msg) ||
586 	    wps_build_registrar_nonce(wps, msg) ||
587 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
588 		wpabuf_free(msg);
589 		return NULL;
590 	}
591 
592 	return msg;
593 }
594 
595 
596 struct wpabuf * wps_build_wsc_nack(struct wps_data *wps)
597 {
598 	struct wpabuf *msg;
599 
600 	wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_NACK");
601 
602 	msg = wpabuf_alloc(1000);
603 	if (msg == NULL)
604 		return NULL;
605 
606 	if (wps_build_version(msg) ||
607 	    wps_build_msg_type(msg, WPS_WSC_NACK) ||
608 	    wps_build_enrollee_nonce(wps, msg) ||
609 	    wps_build_registrar_nonce(wps, msg) ||
610 	    wps_build_config_error(msg, wps->config_error) ||
611 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
612 		wpabuf_free(msg);
613 		return NULL;
614 	}
615 
616 	return msg;
617 }
618 
619 
620 #ifdef CONFIG_WPS_NFC
621 
622 struct wpabuf * wps_nfc_token_build(int ndef, int id, struct wpabuf *pubkey,
623 				    struct wpabuf *dev_pw)
624 {
625 	struct wpabuf *ret;
626 
627 	if (pubkey == NULL || dev_pw == NULL)
628 		return NULL;
629 
630 	ret = wps_build_nfc_pw_token(id, pubkey, dev_pw);
631 	if (ndef && ret) {
632 		struct wpabuf *tmp;
633 		tmp = ndef_build_wifi(ret);
634 		wpabuf_free(ret);
635 		if (tmp == NULL)
636 			return NULL;
637 		ret = tmp;
638 	}
639 
640 	return ret;
641 }
642 
643 
644 int wps_nfc_gen_dh(struct wpabuf **pubkey, struct wpabuf **privkey)
645 {
646 	struct wpabuf *priv = NULL, *pub = NULL;
647 	void *dh_ctx;
648 
649 	dh_ctx = dh5_init(&priv, &pub);
650 	if (dh_ctx == NULL)
651 		return -1;
652 	pub = wpabuf_zeropad(pub, 192);
653 	if (pub == NULL) {
654 		wpabuf_free(priv);
655 		return -1;
656 	}
657 	wpa_hexdump_buf(MSG_DEBUG, "WPS: Generated new DH pubkey", pub);
658 	dh5_free(dh_ctx);
659 
660 	wpabuf_free(*pubkey);
661 	*pubkey = pub;
662 	wpabuf_free(*privkey);
663 	*privkey = priv;
664 
665 	return 0;
666 }
667 
668 
669 struct wpabuf * wps_nfc_token_gen(int ndef, int *id, struct wpabuf **pubkey,
670 				  struct wpabuf **privkey,
671 				  struct wpabuf **dev_pw)
672 {
673 	struct wpabuf *pw;
674 	u16 val;
675 
676 	pw = wpabuf_alloc(WPS_OOB_DEVICE_PASSWORD_LEN);
677 	if (pw == NULL)
678 		return NULL;
679 
680 	if (random_get_bytes(wpabuf_put(pw, WPS_OOB_DEVICE_PASSWORD_LEN),
681 			     WPS_OOB_DEVICE_PASSWORD_LEN) ||
682 	    random_get_bytes((u8 *) &val, sizeof(val))) {
683 		wpabuf_free(pw);
684 		return NULL;
685 	}
686 
687 	if (wps_nfc_gen_dh(pubkey, privkey) < 0) {
688 		wpabuf_free(pw);
689 		return NULL;
690 	}
691 
692 	*id = 0x10 + val % 0xfff0;
693 	wpabuf_free(*dev_pw);
694 	*dev_pw = pw;
695 
696 	return wps_nfc_token_build(ndef, *id, *pubkey, *dev_pw);
697 }
698 
699 
700 struct wpabuf * wps_build_nfc_handover_req(struct wps_context *ctx,
701 					   struct wpabuf *nfc_dh_pubkey)
702 {
703 	struct wpabuf *msg;
704 	void *len;
705 
706 	if (ctx == NULL)
707 		return NULL;
708 
709 	wpa_printf(MSG_DEBUG, "WPS: Building attributes for NFC connection "
710 		   "handover request");
711 
712 	if (nfc_dh_pubkey == NULL) {
713 		wpa_printf(MSG_DEBUG, "WPS: No NFC OOB Device Password "
714 			   "configured");
715 		return NULL;
716 	}
717 
718 	msg = wpabuf_alloc(1000);
719 	if (msg == NULL)
720 		return msg;
721 	len = wpabuf_put(msg, 2);
722 
723 	if (wps_build_oob_dev_pw(msg, DEV_PW_NFC_CONNECTION_HANDOVER,
724 				 nfc_dh_pubkey, NULL, 0) ||
725 	    wps_build_uuid_e(msg, ctx->uuid) ||
726 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
727 		wpabuf_free(msg);
728 		return NULL;
729 	}
730 
731 	WPA_PUT_BE16(len, wpabuf_len(msg) - 2);
732 
733 	return msg;
734 }
735 
736 
737 static int wps_build_ssid(struct wpabuf *msg, struct wps_context *wps)
738 {
739 	wpa_printf(MSG_DEBUG, "WPS:  * SSID");
740 	wpa_hexdump_ascii(MSG_DEBUG, "WPS: SSID in Connection Handover Select",
741 			  wps->ssid, wps->ssid_len);
742 	wpabuf_put_be16(msg, ATTR_SSID);
743 	wpabuf_put_be16(msg, wps->ssid_len);
744 	wpabuf_put_data(msg, wps->ssid, wps->ssid_len);
745 	return 0;
746 }
747 
748 
749 static int wps_build_ap_freq(struct wpabuf *msg, int freq)
750 {
751 	enum hostapd_hw_mode mode;
752 	u8 channel, rf_band;
753 	u16 ap_channel;
754 
755 	if (freq <= 0)
756 		return 0;
757 
758 	mode = ieee80211_freq_to_chan(freq, &channel);
759 	if (mode == NUM_HOSTAPD_MODES)
760 		return 0; /* Unknown channel */
761 
762 	if (mode == HOSTAPD_MODE_IEEE80211G || mode == HOSTAPD_MODE_IEEE80211B)
763 		rf_band = WPS_RF_24GHZ;
764 	else if (mode == HOSTAPD_MODE_IEEE80211A)
765 		rf_band = WPS_RF_50GHZ;
766 	else
767 		return 0; /* Unknown band */
768 	ap_channel = channel;
769 
770 	if (wps_build_rf_bands_attr(msg, rf_band) ||
771 	    wps_build_ap_channel(msg, ap_channel))
772 		return -1;
773 
774 	return 0;
775 }
776 
777 
778 struct wpabuf * wps_build_nfc_handover_sel(struct wps_context *ctx,
779 					   struct wpabuf *nfc_dh_pubkey,
780 					   const u8 *bssid, int freq)
781 {
782 	struct wpabuf *msg;
783 	void *len;
784 
785 	if (ctx == NULL)
786 		return NULL;
787 
788 	wpa_printf(MSG_DEBUG, "WPS: Building attributes for NFC connection "
789 		   "handover select");
790 
791 	if (nfc_dh_pubkey == NULL) {
792 		wpa_printf(MSG_DEBUG, "WPS: No NFC OOB Device Password "
793 			   "configured");
794 		return NULL;
795 	}
796 
797 	msg = wpabuf_alloc(1000);
798 	if (msg == NULL)
799 		return msg;
800 	len = wpabuf_put(msg, 2);
801 
802 	if (wps_build_oob_dev_pw(msg, DEV_PW_NFC_CONNECTION_HANDOVER,
803 				 nfc_dh_pubkey, NULL, 0) ||
804 	    wps_build_ssid(msg, ctx) ||
805 	    wps_build_ap_freq(msg, freq) ||
806 	    (bssid && wps_build_mac_addr(msg, bssid)) ||
807 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
808 		wpabuf_free(msg);
809 		return NULL;
810 	}
811 
812 	WPA_PUT_BE16(len, wpabuf_len(msg) - 2);
813 
814 	return msg;
815 }
816 
817 
818 struct wpabuf * wps_build_nfc_handover_req_p2p(struct wps_context *ctx,
819 					       struct wpabuf *nfc_dh_pubkey)
820 {
821 	struct wpabuf *msg;
822 
823 	if (ctx == NULL)
824 		return NULL;
825 
826 	wpa_printf(MSG_DEBUG, "WPS: Building attributes for NFC connection "
827 		   "handover request (P2P)");
828 
829 	if (nfc_dh_pubkey == NULL) {
830 		wpa_printf(MSG_DEBUG, "WPS: No NFC DH Public Key configured");
831 		return NULL;
832 	}
833 
834 	msg = wpabuf_alloc(1000);
835 	if (msg == NULL)
836 		return msg;
837 
838 	if (wps_build_manufacturer(&ctx->dev, msg) ||
839 	    wps_build_model_name(&ctx->dev, msg) ||
840 	    wps_build_model_number(&ctx->dev, msg) ||
841 	    wps_build_oob_dev_pw(msg, DEV_PW_NFC_CONNECTION_HANDOVER,
842 				 nfc_dh_pubkey, NULL, 0) ||
843 	    wps_build_rf_bands(&ctx->dev, msg, 0) ||
844 	    wps_build_serial_number(&ctx->dev, msg) ||
845 	    wps_build_uuid_e(msg, ctx->uuid) ||
846 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
847 		wpabuf_free(msg);
848 		return NULL;
849 	}
850 
851 	return msg;
852 }
853 
854 
855 struct wpabuf * wps_build_nfc_handover_sel_p2p(struct wps_context *ctx,
856 					       int nfc_dev_pw_id,
857 					       struct wpabuf *nfc_dh_pubkey,
858 					       struct wpabuf *nfc_dev_pw)
859 {
860 	struct wpabuf *msg;
861 	const u8 *dev_pw;
862 	size_t dev_pw_len;
863 
864 	if (ctx == NULL)
865 		return NULL;
866 
867 	wpa_printf(MSG_DEBUG, "WPS: Building attributes for NFC connection "
868 		   "handover select (P2P)");
869 
870 	if (nfc_dh_pubkey == NULL ||
871 	    (nfc_dev_pw_id != DEV_PW_NFC_CONNECTION_HANDOVER &&
872 	     nfc_dev_pw == NULL)) {
873 		wpa_printf(MSG_DEBUG, "WPS: No NFC OOB Device Password "
874 			   "configured");
875 		return NULL;
876 	}
877 
878 	msg = wpabuf_alloc(1000);
879 	if (msg == NULL)
880 		return msg;
881 
882 	if (nfc_dev_pw) {
883 		dev_pw = wpabuf_head(nfc_dev_pw);
884 		dev_pw_len = wpabuf_len(nfc_dev_pw);
885 	} else {
886 		dev_pw = NULL;
887 		dev_pw_len = 0;
888 	}
889 
890 	if (wps_build_manufacturer(&ctx->dev, msg) ||
891 	    wps_build_model_name(&ctx->dev, msg) ||
892 	    wps_build_model_number(&ctx->dev, msg) ||
893 	    wps_build_oob_dev_pw(msg, nfc_dev_pw_id, nfc_dh_pubkey,
894 				 dev_pw, dev_pw_len) ||
895 	    wps_build_rf_bands(&ctx->dev, msg, 0) ||
896 	    wps_build_serial_number(&ctx->dev, msg) ||
897 	    wps_build_uuid_e(msg, ctx->uuid) ||
898 	    wps_build_wfa_ext(msg, 0, NULL, 0)) {
899 		wpabuf_free(msg);
900 		return NULL;
901 	}
902 
903 	return msg;
904 }
905 
906 #endif /* CONFIG_WPS_NFC */
907