List.mui: Update entries count prior to range change
[AROS.git] / workbench / network / WirelessManager / src / wps / wps_registrar.c
blob81ddf3a2b57b18e6b25bee6aa4ac17f962c6f805
1 /*
2 * Wi-Fi Protected Setup - Registrar
3 * Copyright (c) 2008-2009, Jouni Malinen <j@w1.fi>
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.
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
12 * See README and COPYING for more details.
15 #include "utils/includes.h"
17 #include "utils/common.h"
18 #include "utils/base64.h"
19 #include "utils/eloop.h"
20 #include "utils/uuid.h"
21 #include "utils/list.h"
22 #include "crypto/crypto.h"
23 #include "crypto/sha256.h"
24 #include "common/ieee802_11_defs.h"
25 #include "wps_i.h"
26 #include "wps_dev_attr.h"
27 #include "wps_upnp.h"
28 #include "wps_upnp_i.h"
30 #define WPS_WORKAROUNDS
32 struct wps_uuid_pin {
33 struct dl_list list;
34 u8 uuid[WPS_UUID_LEN];
35 int wildcard_uuid;
36 u8 *pin;
37 size_t pin_len;
38 #define PIN_LOCKED BIT(0)
39 #define PIN_EXPIRES BIT(1)
40 int flags;
41 struct os_time expiration;
45 static void wps_free_pin(struct wps_uuid_pin *pin)
47 os_free(pin->pin);
48 os_free(pin);
52 static void wps_remove_pin(struct wps_uuid_pin *pin)
54 dl_list_del(&pin->list);
55 wps_free_pin(pin);
59 static void wps_free_pins(struct dl_list *pins)
61 struct wps_uuid_pin *pin, *prev;
62 dl_list_for_each_safe(pin, prev, pins, struct wps_uuid_pin, list)
63 wps_remove_pin(pin);
67 struct wps_pbc_session {
68 struct wps_pbc_session *next;
69 u8 addr[ETH_ALEN];
70 u8 uuid_e[WPS_UUID_LEN];
71 struct os_time timestamp;
75 static void wps_free_pbc_sessions(struct wps_pbc_session *pbc)
77 struct wps_pbc_session *prev;
79 while (pbc) {
80 prev = pbc;
81 pbc = pbc->next;
82 os_free(prev);
87 struct wps_registrar_device {
88 struct wps_registrar_device *next;
89 struct wps_device_data dev;
90 u8 uuid[WPS_UUID_LEN];
94 struct wps_registrar {
95 struct wps_context *wps;
97 int pbc;
98 int selected_registrar;
100 int (*new_psk_cb)(void *ctx, const u8 *mac_addr, const u8 *psk,
101 size_t psk_len);
102 int (*set_ie_cb)(void *ctx, struct wpabuf *beacon_ie,
103 struct wpabuf *probe_resp_ie);
104 void (*pin_needed_cb)(void *ctx, const u8 *uuid_e,
105 const struct wps_device_data *dev);
106 void (*reg_success_cb)(void *ctx, const u8 *mac_addr,
107 const u8 *uuid_e);
108 void (*set_sel_reg_cb)(void *ctx, int sel_reg, u16 dev_passwd_id,
109 u16 sel_reg_config_methods);
110 void (*enrollee_seen_cb)(void *ctx, const u8 *addr, const u8 *uuid_e,
111 const u8 *pri_dev_type, u16 config_methods,
112 u16 dev_password_id, u8 request_type,
113 const char *dev_name);
114 void *cb_ctx;
116 struct dl_list pins;
117 struct wps_pbc_session *pbc_sessions;
119 int skip_cred_build;
120 struct wpabuf *extra_cred;
121 int disable_auto_conf;
122 int sel_reg_union;
123 int sel_reg_dev_password_id_override;
124 int sel_reg_config_methods_override;
125 int static_wep_only;
127 struct wps_registrar_device *devices;
129 int force_pbc_overlap;
133 static int wps_set_ie(struct wps_registrar *reg);
134 static void wps_registrar_pbc_timeout(void *eloop_ctx, void *timeout_ctx);
135 static void wps_registrar_set_selected_timeout(void *eloop_ctx,
136 void *timeout_ctx);
139 static void wps_free_devices(struct wps_registrar_device *dev)
141 struct wps_registrar_device *prev;
143 while (dev) {
144 prev = dev;
145 dev = dev->next;
146 wps_device_data_free(&prev->dev);
147 os_free(prev);
152 static struct wps_registrar_device * wps_device_get(struct wps_registrar *reg,
153 const u8 *addr)
155 struct wps_registrar_device *dev;
157 for (dev = reg->devices; dev; dev = dev->next) {
158 if (os_memcmp(dev->dev.mac_addr, addr, ETH_ALEN) == 0)
159 return dev;
161 return NULL;
165 static void wps_device_clone_data(struct wps_device_data *dst,
166 struct wps_device_data *src)
168 os_memcpy(dst->mac_addr, src->mac_addr, ETH_ALEN);
169 os_memcpy(dst->pri_dev_type, src->pri_dev_type, WPS_DEV_TYPE_LEN);
171 #define WPS_STRDUP(n) \
172 os_free(dst->n); \
173 dst->n = src->n ? os_strdup(src->n) : NULL
175 WPS_STRDUP(device_name);
176 WPS_STRDUP(manufacturer);
177 WPS_STRDUP(model_name);
178 WPS_STRDUP(model_number);
179 WPS_STRDUP(serial_number);
180 #undef WPS_STRDUP
184 int wps_device_store(struct wps_registrar *reg,
185 struct wps_device_data *dev, const u8 *uuid)
187 struct wps_registrar_device *d;
189 d = wps_device_get(reg, dev->mac_addr);
190 if (d == NULL) {
191 d = os_zalloc(sizeof(*d));
192 if (d == NULL)
193 return -1;
194 d->next = reg->devices;
195 reg->devices = d;
198 wps_device_clone_data(&d->dev, dev);
199 os_memcpy(d->uuid, uuid, WPS_UUID_LEN);
201 return 0;
205 static void wps_registrar_add_pbc_session(struct wps_registrar *reg,
206 const u8 *addr, const u8 *uuid_e)
208 struct wps_pbc_session *pbc, *prev = NULL;
209 struct os_time now;
211 os_get_time(&now);
213 pbc = reg->pbc_sessions;
214 while (pbc) {
215 if (os_memcmp(pbc->addr, addr, ETH_ALEN) == 0 &&
216 os_memcmp(pbc->uuid_e, uuid_e, WPS_UUID_LEN) == 0) {
217 if (prev)
218 prev->next = pbc->next;
219 else
220 reg->pbc_sessions = pbc->next;
221 break;
223 prev = pbc;
224 pbc = pbc->next;
227 if (!pbc) {
228 pbc = os_zalloc(sizeof(*pbc));
229 if (pbc == NULL)
230 return;
231 os_memcpy(pbc->addr, addr, ETH_ALEN);
232 if (uuid_e)
233 os_memcpy(pbc->uuid_e, uuid_e, WPS_UUID_LEN);
236 pbc->next = reg->pbc_sessions;
237 reg->pbc_sessions = pbc;
238 pbc->timestamp = now;
240 /* remove entries that have timed out */
241 prev = pbc;
242 pbc = pbc->next;
244 while (pbc) {
245 if (now.sec > pbc->timestamp.sec + WPS_PBC_WALK_TIME) {
246 prev->next = NULL;
247 wps_free_pbc_sessions(pbc);
248 break;
250 prev = pbc;
251 pbc = pbc->next;
256 static void wps_registrar_remove_pbc_session(struct wps_registrar *reg,
257 const u8 *addr, const u8 *uuid_e)
259 struct wps_pbc_session *pbc, *prev = NULL;
261 pbc = reg->pbc_sessions;
262 while (pbc) {
263 if (os_memcmp(pbc->addr, addr, ETH_ALEN) == 0 &&
264 os_memcmp(pbc->uuid_e, uuid_e, WPS_UUID_LEN) == 0) {
265 if (prev)
266 prev->next = pbc->next;
267 else
268 reg->pbc_sessions = pbc->next;
269 os_free(pbc);
270 break;
272 prev = pbc;
273 pbc = pbc->next;
278 static int wps_registrar_pbc_overlap(struct wps_registrar *reg,
279 const u8 *addr, const u8 *uuid_e)
281 int count = 0;
282 struct wps_pbc_session *pbc;
283 struct os_time now;
285 os_get_time(&now);
287 for (pbc = reg->pbc_sessions; pbc; pbc = pbc->next) {
288 if (now.sec > pbc->timestamp.sec + WPS_PBC_WALK_TIME)
289 break;
290 if (addr == NULL || os_memcmp(addr, pbc->addr, ETH_ALEN) ||
291 uuid_e == NULL ||
292 os_memcmp(uuid_e, pbc->uuid_e, WPS_UUID_LEN))
293 count++;
296 if (addr || uuid_e)
297 count++;
299 return count > 1 ? 1 : 0;
303 static int wps_build_wps_state(struct wps_context *wps, struct wpabuf *msg)
305 wpa_printf(MSG_DEBUG, "WPS: * Wi-Fi Protected Setup State (%d)",
306 wps->wps_state);
307 wpabuf_put_be16(msg, ATTR_WPS_STATE);
308 wpabuf_put_be16(msg, 1);
309 wpabuf_put_u8(msg, wps->wps_state);
310 return 0;
314 #ifdef CONFIG_WPS_UPNP
315 static void wps_registrar_free_pending_m2(struct wps_context *wps)
317 struct upnp_pending_message *p, *p2, *prev = NULL;
318 p = wps->upnp_msgs;
319 while (p) {
320 if (p->type == WPS_M2 || p->type == WPS_M2D) {
321 if (prev == NULL)
322 wps->upnp_msgs = p->next;
323 else
324 prev->next = p->next;
325 wpa_printf(MSG_DEBUG, "WPS UPnP: Drop pending M2/M2D");
326 p2 = p;
327 p = p->next;
328 wpabuf_free(p2->msg);
329 os_free(p2);
330 continue;
332 prev = p;
333 p = p->next;
336 #endif /* CONFIG_WPS_UPNP */
339 static int wps_build_ap_setup_locked(struct wps_context *wps,
340 struct wpabuf *msg)
342 if (wps->ap_setup_locked) {
343 wpa_printf(MSG_DEBUG, "WPS: * AP Setup Locked");
344 wpabuf_put_be16(msg, ATTR_AP_SETUP_LOCKED);
345 wpabuf_put_be16(msg, 1);
346 wpabuf_put_u8(msg, 1);
348 return 0;
352 static int wps_build_selected_registrar(struct wps_registrar *reg,
353 struct wpabuf *msg)
355 if (!reg->sel_reg_union)
356 return 0;
357 wpa_printf(MSG_DEBUG, "WPS: * Selected Registrar");
358 wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR);
359 wpabuf_put_be16(msg, 1);
360 wpabuf_put_u8(msg, 1);
361 return 0;
365 static int wps_build_sel_reg_dev_password_id(struct wps_registrar *reg,
366 struct wpabuf *msg)
368 u16 id = reg->pbc ? DEV_PW_PUSHBUTTON : DEV_PW_DEFAULT;
369 if (!reg->sel_reg_union)
370 return 0;
371 if (reg->sel_reg_dev_password_id_override >= 0)
372 id = reg->sel_reg_dev_password_id_override;
373 wpa_printf(MSG_DEBUG, "WPS: * Device Password ID (%d)", id);
374 wpabuf_put_be16(msg, ATTR_DEV_PASSWORD_ID);
375 wpabuf_put_be16(msg, 2);
376 wpabuf_put_be16(msg, id);
377 return 0;
381 static int wps_build_sel_reg_config_methods(struct wps_registrar *reg,
382 struct wpabuf *msg)
384 u16 methods;
385 if (!reg->sel_reg_union)
386 return 0;
387 methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
388 if (reg->pbc)
389 methods |= WPS_CONFIG_PUSHBUTTON;
390 if (reg->sel_reg_config_methods_override >= 0)
391 methods = reg->sel_reg_config_methods_override;
392 wpa_printf(MSG_DEBUG, "WPS: * Selected Registrar Config Methods (%x)",
393 methods);
394 wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR_CONFIG_METHODS);
395 wpabuf_put_be16(msg, 2);
396 wpabuf_put_be16(msg, methods);
397 return 0;
401 static int wps_build_probe_config_methods(struct wps_registrar *reg,
402 struct wpabuf *msg)
404 u16 methods;
406 * These are the methods that the AP supports as an Enrollee for adding
407 * external Registrars.
409 methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
410 wpa_printf(MSG_DEBUG, "WPS: * Config Methods (%x)", methods);
411 wpabuf_put_be16(msg, ATTR_CONFIG_METHODS);
412 wpabuf_put_be16(msg, 2);
413 wpabuf_put_be16(msg, methods);
414 return 0;
418 static int wps_build_config_methods_r(struct wps_registrar *reg,
419 struct wpabuf *msg)
421 u16 methods;
422 methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
423 if (reg->pbc)
424 methods |= WPS_CONFIG_PUSHBUTTON;
425 return wps_build_config_methods(msg, methods);
430 * wps_registrar_init - Initialize WPS Registrar data
431 * @wps: Pointer to longterm WPS context
432 * @cfg: Registrar configuration
433 * Returns: Pointer to allocated Registrar data or %NULL on failure
435 * This function is used to initialize WPS Registrar functionality. It can be
436 * used for a single Registrar run (e.g., when run in a supplicant) or multiple
437 * runs (e.g., when run as an internal Registrar in an AP). Caller is
438 * responsible for freeing the returned data with wps_registrar_deinit() when
439 * Registrar functionality is not needed anymore.
441 struct wps_registrar *
442 wps_registrar_init(struct wps_context *wps,
443 const struct wps_registrar_config *cfg)
445 struct wps_registrar *reg = os_zalloc(sizeof(*reg));
446 if (reg == NULL)
447 return NULL;
449 dl_list_init(&reg->pins);
450 reg->wps = wps;
451 reg->new_psk_cb = cfg->new_psk_cb;
452 reg->set_ie_cb = cfg->set_ie_cb;
453 reg->pin_needed_cb = cfg->pin_needed_cb;
454 reg->reg_success_cb = cfg->reg_success_cb;
455 reg->set_sel_reg_cb = cfg->set_sel_reg_cb;
456 reg->enrollee_seen_cb = cfg->enrollee_seen_cb;
457 reg->cb_ctx = cfg->cb_ctx;
458 reg->skip_cred_build = cfg->skip_cred_build;
459 if (cfg->extra_cred) {
460 reg->extra_cred = wpabuf_alloc_copy(cfg->extra_cred,
461 cfg->extra_cred_len);
462 if (reg->extra_cred == NULL) {
463 os_free(reg);
464 return NULL;
467 reg->disable_auto_conf = cfg->disable_auto_conf;
468 reg->sel_reg_dev_password_id_override = -1;
469 reg->sel_reg_config_methods_override = -1;
470 reg->static_wep_only = cfg->static_wep_only;
472 if (wps_set_ie(reg)) {
473 wps_registrar_deinit(reg);
474 return NULL;
477 return reg;
482 * wps_registrar_deinit - Deinitialize WPS Registrar data
483 * @reg: Registrar data from wps_registrar_init()
485 void wps_registrar_deinit(struct wps_registrar *reg)
487 if (reg == NULL)
488 return;
489 eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
490 eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
491 wps_free_pins(&reg->pins);
492 wps_free_pbc_sessions(reg->pbc_sessions);
493 wpabuf_free(reg->extra_cred);
494 wps_free_devices(reg->devices);
495 os_free(reg);
500 * wps_registrar_add_pin - Configure a new PIN for Registrar
501 * @reg: Registrar data from wps_registrar_init()
502 * @uuid: UUID-E or %NULL for wildcard (any UUID)
503 * @pin: PIN (Device Password)
504 * @pin_len: Length of pin in octets
505 * @timeout: Time (in seconds) when the PIN will be invalidated; 0 = no timeout
506 * Returns: 0 on success, -1 on failure
508 int wps_registrar_add_pin(struct wps_registrar *reg, const u8 *uuid,
509 const u8 *pin, size_t pin_len, int timeout)
511 struct wps_uuid_pin *p;
513 p = os_zalloc(sizeof(*p));
514 if (p == NULL)
515 return -1;
516 if (uuid == NULL)
517 p->wildcard_uuid = 1;
518 else
519 os_memcpy(p->uuid, uuid, WPS_UUID_LEN);
520 p->pin = os_malloc(pin_len);
521 if (p->pin == NULL) {
522 os_free(p);
523 return -1;
525 os_memcpy(p->pin, pin, pin_len);
526 p->pin_len = pin_len;
528 if (timeout) {
529 p->flags |= PIN_EXPIRES;
530 os_get_time(&p->expiration);
531 p->expiration.sec += timeout;
534 dl_list_add(&reg->pins, &p->list);
536 wpa_printf(MSG_DEBUG, "WPS: A new PIN configured (timeout=%d)",
537 timeout);
538 wpa_hexdump(MSG_DEBUG, "WPS: UUID", uuid, WPS_UUID_LEN);
539 wpa_hexdump_ascii_key(MSG_DEBUG, "WPS: PIN", pin, pin_len);
540 reg->selected_registrar = 1;
541 reg->pbc = 0;
542 wps_registrar_selected_registrar_changed(reg);
543 eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
544 eloop_register_timeout(WPS_PBC_WALK_TIME, 0,
545 wps_registrar_set_selected_timeout,
546 reg, NULL);
548 return 0;
552 static void wps_registrar_expire_pins(struct wps_registrar *reg)
554 struct wps_uuid_pin *pin, *prev;
555 struct os_time now;
557 os_get_time(&now);
558 dl_list_for_each_safe(pin, prev, &reg->pins, struct wps_uuid_pin, list)
560 if ((pin->flags & PIN_EXPIRES) &&
561 os_time_before(&pin->expiration, &now)) {
562 wpa_hexdump(MSG_DEBUG, "WPS: Expired PIN for UUID",
563 pin->uuid, WPS_UUID_LEN);
564 wps_remove_pin(pin);
571 * wps_registrar_invalidate_pin - Invalidate a PIN for a specific UUID-E
572 * @reg: Registrar data from wps_registrar_init()
573 * @uuid: UUID-E
574 * Returns: 0 on success, -1 on failure (e.g., PIN not found)
576 int wps_registrar_invalidate_pin(struct wps_registrar *reg, const u8 *uuid)
578 struct wps_uuid_pin *pin, *prev;
580 dl_list_for_each_safe(pin, prev, &reg->pins, struct wps_uuid_pin, list)
582 if (os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
583 wpa_hexdump(MSG_DEBUG, "WPS: Invalidated PIN for UUID",
584 pin->uuid, WPS_UUID_LEN);
585 wps_remove_pin(pin);
586 return 0;
590 return -1;
594 static const u8 * wps_registrar_get_pin(struct wps_registrar *reg,
595 const u8 *uuid, size_t *pin_len)
597 struct wps_uuid_pin *pin, *found = NULL;
599 wps_registrar_expire_pins(reg);
601 dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
602 if (!pin->wildcard_uuid &&
603 os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
604 found = pin;
605 break;
609 if (!found) {
610 /* Check for wildcard UUIDs since none of the UUID-specific
611 * PINs matched */
612 dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
613 if (pin->wildcard_uuid == 1) {
614 wpa_printf(MSG_DEBUG, "WPS: Found a wildcard "
615 "PIN. Assigned it for this UUID-E");
616 pin->wildcard_uuid = 2;
617 os_memcpy(pin->uuid, uuid, WPS_UUID_LEN);
618 found = pin;
619 break;
624 if (!found)
625 return NULL;
628 * Lock the PIN to avoid attacks based on concurrent re-use of the PIN
629 * that could otherwise avoid PIN invalidations.
631 if (found->flags & PIN_LOCKED) {
632 wpa_printf(MSG_DEBUG, "WPS: Selected PIN locked - do not "
633 "allow concurrent re-use");
634 return NULL;
636 *pin_len = found->pin_len;
637 found->flags |= PIN_LOCKED;
638 return found->pin;
643 * wps_registrar_unlock_pin - Unlock a PIN for a specific UUID-E
644 * @reg: Registrar data from wps_registrar_init()
645 * @uuid: UUID-E
646 * Returns: 0 on success, -1 on failure
648 * PINs are locked to enforce only one concurrent use. This function unlocks a
649 * PIN to allow it to be used again. If the specified PIN was configured using
650 * a wildcard UUID, it will be removed instead of allowing multiple uses.
652 int wps_registrar_unlock_pin(struct wps_registrar *reg, const u8 *uuid)
654 struct wps_uuid_pin *pin;
656 dl_list_for_each(pin, &reg->pins, struct wps_uuid_pin, list) {
657 if (os_memcmp(pin->uuid, uuid, WPS_UUID_LEN) == 0) {
658 if (pin->wildcard_uuid == 2) {
659 wpa_printf(MSG_DEBUG, "WPS: Invalidating used "
660 "wildcard PIN");
661 return wps_registrar_invalidate_pin(reg, uuid);
663 pin->flags &= ~PIN_LOCKED;
664 return 0;
668 return -1;
672 static void wps_registrar_stop_pbc(struct wps_registrar *reg)
674 reg->selected_registrar = 0;
675 reg->pbc = 0;
676 wps_registrar_selected_registrar_changed(reg);
680 static void wps_registrar_pbc_timeout(void *eloop_ctx, void *timeout_ctx)
682 struct wps_registrar *reg = eloop_ctx;
684 wpa_printf(MSG_DEBUG, "WPS: PBC timed out - disable PBC mode");
685 wps_pbc_timeout_event(reg->wps);
686 wps_registrar_stop_pbc(reg);
691 * wps_registrar_button_pushed - Notify Registrar that AP button was pushed
692 * @reg: Registrar data from wps_registrar_init()
693 * Returns: 0 on success, -1 on failure
695 * This function is called on an AP when a push button is pushed to activate
696 * PBC mode. The PBC mode will be stopped after walk time (2 minutes) timeout
697 * or when a PBC registration is completed.
699 int wps_registrar_button_pushed(struct wps_registrar *reg)
701 if (wps_registrar_pbc_overlap(reg, NULL, NULL)) {
702 wpa_printf(MSG_DEBUG, "WPS: PBC overlap - do not start PBC "
703 "mode");
704 wps_pbc_overlap_event(reg->wps);
705 return -1;
707 wpa_printf(MSG_DEBUG, "WPS: Button pushed - PBC mode started");
708 reg->force_pbc_overlap = 0;
709 reg->selected_registrar = 1;
710 reg->pbc = 1;
711 wps_registrar_selected_registrar_changed(reg);
713 eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
714 eloop_register_timeout(WPS_PBC_WALK_TIME, 0, wps_registrar_pbc_timeout,
715 reg, NULL);
716 return 0;
720 static void wps_registrar_pbc_completed(struct wps_registrar *reg)
722 wpa_printf(MSG_DEBUG, "WPS: PBC completed - stopping PBC mode");
723 eloop_cancel_timeout(wps_registrar_pbc_timeout, reg, NULL);
724 wps_registrar_stop_pbc(reg);
728 static void wps_registrar_pin_completed(struct wps_registrar *reg)
730 wpa_printf(MSG_DEBUG, "WPS: PIN completed using internal Registrar");
731 eloop_cancel_timeout(wps_registrar_set_selected_timeout, reg, NULL);
732 reg->selected_registrar = 0;
733 wps_registrar_selected_registrar_changed(reg);
738 * wps_registrar_probe_req_rx - Notify Registrar of Probe Request
739 * @reg: Registrar data from wps_registrar_init()
740 * @addr: MAC address of the Probe Request sender
741 * @wps_data: WPS IE contents
743 * This function is called on an AP when a Probe Request with WPS IE is
744 * received. This is used to track PBC mode use and to detect possible overlap
745 * situation with other WPS APs.
747 void wps_registrar_probe_req_rx(struct wps_registrar *reg, const u8 *addr,
748 const struct wpabuf *wps_data)
750 struct wps_parse_attr attr;
752 wpa_hexdump_buf(MSG_MSGDUMP,
753 "WPS: Probe Request with WPS data received",
754 wps_data);
756 if (wps_parse_msg(wps_data, &attr) < 0)
757 return;
758 if (!wps_version_supported(attr.version)) {
759 wpa_printf(MSG_DEBUG, "WPS: Unsupported ProbeReq WPS IE "
760 "version 0x%x", attr.version ? *attr.version : 0);
761 return;
764 if (attr.config_methods == NULL) {
765 wpa_printf(MSG_DEBUG, "WPS: No Config Methods attribute in "
766 "Probe Request");
767 return;
770 if (attr.dev_password_id == NULL) {
771 wpa_printf(MSG_DEBUG, "WPS: No Device Password Id attribute "
772 "in Probe Request");
773 return;
776 if (reg->enrollee_seen_cb && attr.uuid_e &&
777 attr.primary_dev_type && attr.request_type) {
778 char *dev_name = NULL;
779 if (attr.dev_name) {
780 dev_name = os_zalloc(attr.dev_name_len + 1);
781 if (dev_name) {
782 os_memcpy(dev_name, attr.dev_name,
783 attr.dev_name_len);
786 reg->enrollee_seen_cb(reg->cb_ctx, addr, attr.uuid_e,
787 attr.primary_dev_type,
788 WPA_GET_BE16(attr.config_methods),
789 WPA_GET_BE16(attr.dev_password_id),
790 *attr.request_type, dev_name);
791 os_free(dev_name);
794 if (WPA_GET_BE16(attr.dev_password_id) != DEV_PW_PUSHBUTTON)
795 return; /* Not PBC */
797 wpa_printf(MSG_DEBUG, "WPS: Probe Request for PBC received from "
798 MACSTR, MAC2STR(addr));
799 if (attr.uuid_e == NULL) {
800 wpa_printf(MSG_DEBUG, "WPS: Invalid Probe Request WPS IE: No "
801 "UUID-E included");
802 return;
805 wps_registrar_add_pbc_session(reg, addr, attr.uuid_e);
806 if (wps_registrar_pbc_overlap(reg, addr, attr.uuid_e)) {
807 wpa_printf(MSG_DEBUG, "WPS: PBC session overlap detected");
808 reg->force_pbc_overlap = 1;
809 wps_pbc_overlap_event(reg->wps);
814 static int wps_cb_new_psk(struct wps_registrar *reg, const u8 *mac_addr,
815 const u8 *psk, size_t psk_len)
817 if (reg->new_psk_cb == NULL)
818 return 0;
820 return reg->new_psk_cb(reg->cb_ctx, mac_addr, psk, psk_len);
824 static void wps_cb_pin_needed(struct wps_registrar *reg, const u8 *uuid_e,
825 const struct wps_device_data *dev)
827 if (reg->pin_needed_cb == NULL)
828 return;
830 reg->pin_needed_cb(reg->cb_ctx, uuid_e, dev);
834 static void wps_cb_reg_success(struct wps_registrar *reg, const u8 *mac_addr,
835 const u8 *uuid_e)
837 if (reg->reg_success_cb == NULL)
838 return;
840 reg->reg_success_cb(reg->cb_ctx, mac_addr, uuid_e);
844 static int wps_cb_set_ie(struct wps_registrar *reg, struct wpabuf *beacon_ie,
845 struct wpabuf *probe_resp_ie)
847 return reg->set_ie_cb(reg->cb_ctx, beacon_ie, probe_resp_ie);
851 static void wps_cb_set_sel_reg(struct wps_registrar *reg)
853 u16 methods = 0;
854 if (reg->set_sel_reg_cb == NULL)
855 return;
857 if (reg->selected_registrar) {
858 methods = reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
859 if (reg->pbc)
860 methods |= WPS_CONFIG_PUSHBUTTON;
863 reg->set_sel_reg_cb(reg->cb_ctx, reg->selected_registrar,
864 reg->pbc ? DEV_PW_PUSHBUTTON : DEV_PW_DEFAULT,
865 methods);
869 /* Encapsulate WPS IE data with one (or more, if needed) IE headers */
870 static struct wpabuf * wps_ie_encapsulate(struct wpabuf *data)
872 struct wpabuf *ie;
873 const u8 *pos, *end;
875 ie = wpabuf_alloc(wpabuf_len(data) + 100);
876 if (ie == NULL) {
877 wpabuf_free(data);
878 return NULL;
881 pos = wpabuf_head(data);
882 end = pos + wpabuf_len(data);
884 while (end > pos) {
885 size_t frag_len = end - pos;
886 if (frag_len > 251)
887 frag_len = 251;
888 wpabuf_put_u8(ie, WLAN_EID_VENDOR_SPECIFIC);
889 wpabuf_put_u8(ie, 4 + frag_len);
890 wpabuf_put_be32(ie, WPS_DEV_OUI_WFA);
891 wpabuf_put_data(ie, pos, frag_len);
892 pos += frag_len;
895 wpabuf_free(data);
897 return ie;
901 static int wps_set_ie(struct wps_registrar *reg)
903 struct wpabuf *beacon;
904 struct wpabuf *probe;
906 if (reg->set_ie_cb == NULL)
907 return 0;
909 wpa_printf(MSG_DEBUG, "WPS: Build Beacon and Probe Response IEs");
911 beacon = wpabuf_alloc(300);
912 if (beacon == NULL)
913 return -1;
914 probe = wpabuf_alloc(400);
915 if (probe == NULL) {
916 wpabuf_free(beacon);
917 return -1;
920 if (wps_build_version(beacon) ||
921 wps_build_wps_state(reg->wps, beacon) ||
922 wps_build_ap_setup_locked(reg->wps, beacon) ||
923 wps_build_selected_registrar(reg, beacon) ||
924 wps_build_sel_reg_dev_password_id(reg, beacon) ||
925 wps_build_sel_reg_config_methods(reg, beacon) ||
926 wps_build_version(probe) ||
927 wps_build_wps_state(reg->wps, probe) ||
928 wps_build_ap_setup_locked(reg->wps, probe) ||
929 wps_build_selected_registrar(reg, probe) ||
930 wps_build_sel_reg_dev_password_id(reg, probe) ||
931 wps_build_sel_reg_config_methods(reg, probe) ||
932 wps_build_resp_type(probe, reg->wps->ap ? WPS_RESP_AP :
933 WPS_RESP_REGISTRAR) ||
934 wps_build_uuid_e(probe, reg->wps->uuid) ||
935 wps_build_device_attrs(&reg->wps->dev, probe) ||
936 wps_build_probe_config_methods(reg, probe) ||
937 wps_build_rf_bands(&reg->wps->dev, probe)) {
938 wpabuf_free(beacon);
939 wpabuf_free(probe);
940 return -1;
943 beacon = wps_ie_encapsulate(beacon);
944 probe = wps_ie_encapsulate(probe);
946 if (!beacon || !probe) {
947 wpabuf_free(beacon);
948 wpabuf_free(probe);
949 return -1;
952 if (reg->static_wep_only) {
954 * Windows XP and Vista clients can get confused about
955 * EAP-Identity/Request when they probe the network with
956 * EAPOL-Start. In such a case, they may assume the network is
957 * using IEEE 802.1X and prompt user for a certificate while
958 * the correct (non-WPS) behavior would be to ask for the
959 * static WEP key. As a workaround, use Microsoft Provisioning
960 * IE to advertise that legacy 802.1X is not supported.
962 const u8 ms_wps[7] = {
963 WLAN_EID_VENDOR_SPECIFIC, 5,
964 /* Microsoft Provisioning IE (00:50:f2:5) */
965 0x00, 0x50, 0xf2, 5,
966 0x00 /* no legacy 802.1X or MS WPS */
968 wpa_printf(MSG_DEBUG, "WPS: Add Microsoft Provisioning IE "
969 "into Beacon/Probe Response frames");
970 wpabuf_put_data(beacon, ms_wps, sizeof(ms_wps));
971 wpabuf_put_data(probe, ms_wps, sizeof(ms_wps));
974 return wps_cb_set_ie(reg, beacon, probe);
978 static int wps_get_dev_password(struct wps_data *wps)
980 const u8 *pin;
981 size_t pin_len = 0;
983 os_free(wps->dev_password);
984 wps->dev_password = NULL;
986 if (wps->pbc) {
987 wpa_printf(MSG_DEBUG, "WPS: Use default PIN for PBC");
988 pin = (const u8 *) "00000000";
989 pin_len = 8;
990 } else {
991 pin = wps_registrar_get_pin(wps->wps->registrar, wps->uuid_e,
992 &pin_len);
994 if (pin == NULL) {
995 wpa_printf(MSG_DEBUG, "WPS: No Device Password available for "
996 "the Enrollee");
997 wps_cb_pin_needed(wps->wps->registrar, wps->uuid_e,
998 &wps->peer_dev);
999 return -1;
1002 wps->dev_password = os_malloc(pin_len);
1003 if (wps->dev_password == NULL)
1004 return -1;
1005 os_memcpy(wps->dev_password, pin, pin_len);
1006 wps->dev_password_len = pin_len;
1008 return 0;
1012 static int wps_build_uuid_r(struct wps_data *wps, struct wpabuf *msg)
1014 wpa_printf(MSG_DEBUG, "WPS: * UUID-R");
1015 wpabuf_put_be16(msg, ATTR_UUID_R);
1016 wpabuf_put_be16(msg, WPS_UUID_LEN);
1017 wpabuf_put_data(msg, wps->uuid_r, WPS_UUID_LEN);
1018 return 0;
1022 static int wps_build_r_hash(struct wps_data *wps, struct wpabuf *msg)
1024 u8 *hash;
1025 const u8 *addr[4];
1026 size_t len[4];
1028 if (os_get_random(wps->snonce, 2 * WPS_SECRET_NONCE_LEN) < 0)
1029 return -1;
1030 wpa_hexdump(MSG_DEBUG, "WPS: R-S1", wps->snonce, WPS_SECRET_NONCE_LEN);
1031 wpa_hexdump(MSG_DEBUG, "WPS: R-S2",
1032 wps->snonce + WPS_SECRET_NONCE_LEN, WPS_SECRET_NONCE_LEN);
1034 if (wps->dh_pubkey_e == NULL || wps->dh_pubkey_r == NULL) {
1035 wpa_printf(MSG_DEBUG, "WPS: DH public keys not available for "
1036 "R-Hash derivation");
1037 return -1;
1040 wpa_printf(MSG_DEBUG, "WPS: * R-Hash1");
1041 wpabuf_put_be16(msg, ATTR_R_HASH1);
1042 wpabuf_put_be16(msg, SHA256_MAC_LEN);
1043 hash = wpabuf_put(msg, SHA256_MAC_LEN);
1044 /* R-Hash1 = HMAC_AuthKey(R-S1 || PSK1 || PK_E || PK_R) */
1045 addr[0] = wps->snonce;
1046 len[0] = WPS_SECRET_NONCE_LEN;
1047 addr[1] = wps->psk1;
1048 len[1] = WPS_PSK_LEN;
1049 addr[2] = wpabuf_head(wps->dh_pubkey_e);
1050 len[2] = wpabuf_len(wps->dh_pubkey_e);
1051 addr[3] = wpabuf_head(wps->dh_pubkey_r);
1052 len[3] = wpabuf_len(wps->dh_pubkey_r);
1053 hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
1054 wpa_hexdump(MSG_DEBUG, "WPS: R-Hash1", hash, SHA256_MAC_LEN);
1056 wpa_printf(MSG_DEBUG, "WPS: * R-Hash2");
1057 wpabuf_put_be16(msg, ATTR_R_HASH2);
1058 wpabuf_put_be16(msg, SHA256_MAC_LEN);
1059 hash = wpabuf_put(msg, SHA256_MAC_LEN);
1060 /* R-Hash2 = HMAC_AuthKey(R-S2 || PSK2 || PK_E || PK_R) */
1061 addr[0] = wps->snonce + WPS_SECRET_NONCE_LEN;
1062 addr[1] = wps->psk2;
1063 hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
1064 wpa_hexdump(MSG_DEBUG, "WPS: R-Hash2", hash, SHA256_MAC_LEN);
1066 return 0;
1070 static int wps_build_r_snonce1(struct wps_data *wps, struct wpabuf *msg)
1072 wpa_printf(MSG_DEBUG, "WPS: * R-SNonce1");
1073 wpabuf_put_be16(msg, ATTR_R_SNONCE1);
1074 wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
1075 wpabuf_put_data(msg, wps->snonce, WPS_SECRET_NONCE_LEN);
1076 return 0;
1080 static int wps_build_r_snonce2(struct wps_data *wps, struct wpabuf *msg)
1082 wpa_printf(MSG_DEBUG, "WPS: * R-SNonce2");
1083 wpabuf_put_be16(msg, ATTR_R_SNONCE2);
1084 wpabuf_put_be16(msg, WPS_SECRET_NONCE_LEN);
1085 wpabuf_put_data(msg, wps->snonce + WPS_SECRET_NONCE_LEN,
1086 WPS_SECRET_NONCE_LEN);
1087 return 0;
1091 static int wps_build_cred_network_idx(struct wpabuf *msg,
1092 const struct wps_credential *cred)
1094 wpa_printf(MSG_DEBUG, "WPS: * Network Index");
1095 wpabuf_put_be16(msg, ATTR_NETWORK_INDEX);
1096 wpabuf_put_be16(msg, 1);
1097 wpabuf_put_u8(msg, 1);
1098 return 0;
1102 static int wps_build_cred_ssid(struct wpabuf *msg,
1103 const struct wps_credential *cred)
1105 wpa_printf(MSG_DEBUG, "WPS: * SSID");
1106 wpabuf_put_be16(msg, ATTR_SSID);
1107 wpabuf_put_be16(msg, cred->ssid_len);
1108 wpabuf_put_data(msg, cred->ssid, cred->ssid_len);
1109 return 0;
1113 static int wps_build_cred_auth_type(struct wpabuf *msg,
1114 const struct wps_credential *cred)
1116 wpa_printf(MSG_DEBUG, "WPS: * Authentication Type (0x%x)",
1117 cred->auth_type);
1118 wpabuf_put_be16(msg, ATTR_AUTH_TYPE);
1119 wpabuf_put_be16(msg, 2);
1120 wpabuf_put_be16(msg, cred->auth_type);
1121 return 0;
1125 static int wps_build_cred_encr_type(struct wpabuf *msg,
1126 const struct wps_credential *cred)
1128 wpa_printf(MSG_DEBUG, "WPS: * Encryption Type (0x%x)",
1129 cred->encr_type);
1130 wpabuf_put_be16(msg, ATTR_ENCR_TYPE);
1131 wpabuf_put_be16(msg, 2);
1132 wpabuf_put_be16(msg, cred->encr_type);
1133 return 0;
1137 static int wps_build_cred_network_key(struct wpabuf *msg,
1138 const struct wps_credential *cred)
1140 wpa_printf(MSG_DEBUG, "WPS: * Network Key (len=%d)",
1141 (int) cred->key_len);
1142 wpabuf_put_be16(msg, ATTR_NETWORK_KEY);
1143 wpabuf_put_be16(msg, cred->key_len);
1144 wpabuf_put_data(msg, cred->key, cred->key_len);
1145 return 0;
1149 static int wps_build_cred_mac_addr(struct wpabuf *msg,
1150 const struct wps_credential *cred)
1152 wpa_printf(MSG_DEBUG, "WPS: * MAC Address (" MACSTR ")",
1153 MAC2STR(cred->mac_addr));
1154 wpabuf_put_be16(msg, ATTR_MAC_ADDR);
1155 wpabuf_put_be16(msg, ETH_ALEN);
1156 wpabuf_put_data(msg, cred->mac_addr, ETH_ALEN);
1157 return 0;
1161 static int wps_build_credential(struct wpabuf *msg,
1162 const struct wps_credential *cred)
1164 if (wps_build_cred_network_idx(msg, cred) ||
1165 wps_build_cred_ssid(msg, cred) ||
1166 wps_build_cred_auth_type(msg, cred) ||
1167 wps_build_cred_encr_type(msg, cred) ||
1168 wps_build_cred_network_key(msg, cred) ||
1169 wps_build_cred_mac_addr(msg, cred))
1170 return -1;
1171 return 0;
1175 int wps_build_cred(struct wps_data *wps, struct wpabuf *msg)
1177 struct wpabuf *cred;
1179 if (wps->wps->registrar->skip_cred_build)
1180 goto skip_cred_build;
1182 wpa_printf(MSG_DEBUG, "WPS: * Credential");
1183 if (wps->use_cred) {
1184 os_memcpy(&wps->cred, wps->use_cred, sizeof(wps->cred));
1185 goto use_provided;
1187 os_memset(&wps->cred, 0, sizeof(wps->cred));
1189 os_memcpy(wps->cred.ssid, wps->wps->ssid, wps->wps->ssid_len);
1190 wps->cred.ssid_len = wps->wps->ssid_len;
1192 /* Select the best authentication and encryption type */
1193 if (wps->auth_type & WPS_AUTH_WPA2PSK)
1194 wps->auth_type = WPS_AUTH_WPA2PSK;
1195 else if (wps->auth_type & WPS_AUTH_WPAPSK)
1196 wps->auth_type = WPS_AUTH_WPAPSK;
1197 else if (wps->auth_type & WPS_AUTH_OPEN)
1198 wps->auth_type = WPS_AUTH_OPEN;
1199 else if (wps->auth_type & WPS_AUTH_SHARED)
1200 wps->auth_type = WPS_AUTH_SHARED;
1201 else {
1202 wpa_printf(MSG_DEBUG, "WPS: Unsupported auth_type 0x%x",
1203 wps->auth_type);
1204 return -1;
1206 wps->cred.auth_type = wps->auth_type;
1208 if (wps->auth_type == WPS_AUTH_WPA2PSK ||
1209 wps->auth_type == WPS_AUTH_WPAPSK) {
1210 if (wps->encr_type & WPS_ENCR_AES)
1211 wps->encr_type = WPS_ENCR_AES;
1212 else if (wps->encr_type & WPS_ENCR_TKIP)
1213 wps->encr_type = WPS_ENCR_TKIP;
1214 else {
1215 wpa_printf(MSG_DEBUG, "WPS: No suitable encryption "
1216 "type for WPA/WPA2");
1217 return -1;
1219 } else {
1220 if (wps->encr_type & WPS_ENCR_WEP)
1221 wps->encr_type = WPS_ENCR_WEP;
1222 else if (wps->encr_type & WPS_ENCR_NONE)
1223 wps->encr_type = WPS_ENCR_NONE;
1224 else {
1225 wpa_printf(MSG_DEBUG, "WPS: No suitable encryption "
1226 "type for non-WPA/WPA2 mode");
1227 return -1;
1230 wps->cred.encr_type = wps->encr_type;
1232 * Set MAC address in the Credential to be the Enrollee's MAC address
1234 os_memcpy(wps->cred.mac_addr, wps->mac_addr_e, ETH_ALEN);
1236 if (wps->wps->wps_state == WPS_STATE_NOT_CONFIGURED && wps->wps->ap &&
1237 !wps->wps->registrar->disable_auto_conf) {
1238 u8 r[16];
1239 /* Generate a random passphrase */
1240 if (os_get_random(r, sizeof(r)) < 0)
1241 return -1;
1242 os_free(wps->new_psk);
1243 wps->new_psk = base64_encode(r, sizeof(r), &wps->new_psk_len);
1244 if (wps->new_psk == NULL)
1245 return -1;
1246 wps->new_psk_len--; /* remove newline */
1247 while (wps->new_psk_len &&
1248 wps->new_psk[wps->new_psk_len - 1] == '=')
1249 wps->new_psk_len--;
1250 wpa_hexdump_ascii_key(MSG_DEBUG, "WPS: Generated passphrase",
1251 wps->new_psk, wps->new_psk_len);
1252 os_memcpy(wps->cred.key, wps->new_psk, wps->new_psk_len);
1253 wps->cred.key_len = wps->new_psk_len;
1254 } else if (wps->use_psk_key && wps->wps->psk_set) {
1255 char hex[65];
1256 wpa_printf(MSG_DEBUG, "WPS: Use PSK format for Network Key");
1257 wpa_snprintf_hex(hex, sizeof(hex), wps->wps->psk, 32);
1258 os_memcpy(wps->cred.key, hex, 32 * 2);
1259 wps->cred.key_len = 32 * 2;
1260 } else if (wps->wps->network_key) {
1261 os_memcpy(wps->cred.key, wps->wps->network_key,
1262 wps->wps->network_key_len);
1263 wps->cred.key_len = wps->wps->network_key_len;
1264 } else if (wps->auth_type & (WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK)) {
1265 char hex[65];
1266 /* Generate a random per-device PSK */
1267 os_free(wps->new_psk);
1268 wps->new_psk_len = 32;
1269 wps->new_psk = os_malloc(wps->new_psk_len);
1270 if (wps->new_psk == NULL)
1271 return -1;
1272 if (os_get_random(wps->new_psk, wps->new_psk_len) < 0) {
1273 os_free(wps->new_psk);
1274 wps->new_psk = NULL;
1275 return -1;
1277 wpa_hexdump_key(MSG_DEBUG, "WPS: Generated per-device PSK",
1278 wps->new_psk, wps->new_psk_len);
1279 wpa_snprintf_hex(hex, sizeof(hex), wps->new_psk,
1280 wps->new_psk_len);
1281 os_memcpy(wps->cred.key, hex, wps->new_psk_len * 2);
1282 wps->cred.key_len = wps->new_psk_len * 2;
1285 use_provided:
1286 cred = wpabuf_alloc(200);
1287 if (cred == NULL)
1288 return -1;
1290 if (wps_build_credential(cred, &wps->cred)) {
1291 wpabuf_free(cred);
1292 return -1;
1295 wpabuf_put_be16(msg, ATTR_CRED);
1296 wpabuf_put_be16(msg, wpabuf_len(cred));
1297 wpabuf_put_buf(msg, cred);
1298 wpabuf_free(cred);
1300 skip_cred_build:
1301 if (wps->wps->registrar->extra_cred) {
1302 wpa_printf(MSG_DEBUG, "WPS: * Credential (pre-configured)");
1303 wpabuf_put_buf(msg, wps->wps->registrar->extra_cred);
1306 return 0;
1310 static int wps_build_ap_settings(struct wps_data *wps, struct wpabuf *msg)
1312 wpa_printf(MSG_DEBUG, "WPS: * AP Settings");
1314 if (wps_build_credential(msg, &wps->cred))
1315 return -1;
1317 return 0;
1321 static struct wpabuf * wps_build_m2(struct wps_data *wps)
1323 struct wpabuf *msg;
1325 if (os_get_random(wps->nonce_r, WPS_NONCE_LEN) < 0)
1326 return NULL;
1327 wpa_hexdump(MSG_DEBUG, "WPS: Registrar Nonce",
1328 wps->nonce_r, WPS_NONCE_LEN);
1329 wpa_hexdump(MSG_DEBUG, "WPS: UUID-R", wps->uuid_r, WPS_UUID_LEN);
1331 wpa_printf(MSG_DEBUG, "WPS: Building Message M2");
1332 msg = wpabuf_alloc(1000);
1333 if (msg == NULL)
1334 return NULL;
1336 if (wps_build_version(msg) ||
1337 wps_build_msg_type(msg, WPS_M2) ||
1338 wps_build_enrollee_nonce(wps, msg) ||
1339 wps_build_registrar_nonce(wps, msg) ||
1340 wps_build_uuid_r(wps, msg) ||
1341 wps_build_public_key(wps, msg) ||
1342 wps_derive_keys(wps) ||
1343 wps_build_auth_type_flags(wps, msg) ||
1344 wps_build_encr_type_flags(wps, msg) ||
1345 wps_build_conn_type_flags(wps, msg) ||
1346 wps_build_config_methods_r(wps->wps->registrar, msg) ||
1347 wps_build_device_attrs(&wps->wps->dev, msg) ||
1348 wps_build_rf_bands(&wps->wps->dev, msg) ||
1349 wps_build_assoc_state(wps, msg) ||
1350 wps_build_config_error(msg, WPS_CFG_NO_ERROR) ||
1351 wps_build_dev_password_id(msg, wps->dev_pw_id) ||
1352 wps_build_os_version(&wps->wps->dev, msg) ||
1353 wps_build_authenticator(wps, msg)) {
1354 wpabuf_free(msg);
1355 return NULL;
1358 wps->int_reg = 1;
1359 wps->state = RECV_M3;
1360 return msg;
1364 static struct wpabuf * wps_build_m2d(struct wps_data *wps)
1366 struct wpabuf *msg;
1367 u16 err = wps->config_error;
1369 wpa_printf(MSG_DEBUG, "WPS: Building Message M2D");
1370 msg = wpabuf_alloc(1000);
1371 if (msg == NULL)
1372 return NULL;
1374 if (wps->wps->ap && wps->wps->ap_setup_locked &&
1375 err == WPS_CFG_NO_ERROR)
1376 err = WPS_CFG_SETUP_LOCKED;
1378 if (wps_build_version(msg) ||
1379 wps_build_msg_type(msg, WPS_M2D) ||
1380 wps_build_enrollee_nonce(wps, msg) ||
1381 wps_build_registrar_nonce(wps, msg) ||
1382 wps_build_uuid_r(wps, msg) ||
1383 wps_build_auth_type_flags(wps, msg) ||
1384 wps_build_encr_type_flags(wps, msg) ||
1385 wps_build_conn_type_flags(wps, msg) ||
1386 wps_build_config_methods_r(wps->wps->registrar, msg) ||
1387 wps_build_device_attrs(&wps->wps->dev, msg) ||
1388 wps_build_rf_bands(&wps->wps->dev, msg) ||
1389 wps_build_assoc_state(wps, msg) ||
1390 wps_build_config_error(msg, err) ||
1391 wps_build_os_version(&wps->wps->dev, msg)) {
1392 wpabuf_free(msg);
1393 return NULL;
1396 wps->state = RECV_M2D_ACK;
1397 return msg;
1401 static struct wpabuf * wps_build_m4(struct wps_data *wps)
1403 struct wpabuf *msg, *plain;
1405 wpa_printf(MSG_DEBUG, "WPS: Building Message M4");
1407 wps_derive_psk(wps, wps->dev_password, wps->dev_password_len);
1409 plain = wpabuf_alloc(200);
1410 if (plain == NULL)
1411 return NULL;
1413 msg = wpabuf_alloc(1000);
1414 if (msg == NULL) {
1415 wpabuf_free(plain);
1416 return NULL;
1419 if (wps_build_version(msg) ||
1420 wps_build_msg_type(msg, WPS_M4) ||
1421 wps_build_enrollee_nonce(wps, msg) ||
1422 wps_build_r_hash(wps, msg) ||
1423 wps_build_r_snonce1(wps, plain) ||
1424 wps_build_key_wrap_auth(wps, plain) ||
1425 wps_build_encr_settings(wps, msg, plain) ||
1426 wps_build_authenticator(wps, msg)) {
1427 wpabuf_free(plain);
1428 wpabuf_free(msg);
1429 return NULL;
1431 wpabuf_free(plain);
1433 wps->state = RECV_M5;
1434 return msg;
1438 static struct wpabuf * wps_build_m6(struct wps_data *wps)
1440 struct wpabuf *msg, *plain;
1442 wpa_printf(MSG_DEBUG, "WPS: Building Message M6");
1444 plain = wpabuf_alloc(200);
1445 if (plain == NULL)
1446 return NULL;
1448 msg = wpabuf_alloc(1000);
1449 if (msg == NULL) {
1450 wpabuf_free(plain);
1451 return NULL;
1454 if (wps_build_version(msg) ||
1455 wps_build_msg_type(msg, WPS_M6) ||
1456 wps_build_enrollee_nonce(wps, msg) ||
1457 wps_build_r_snonce2(wps, plain) ||
1458 wps_build_key_wrap_auth(wps, plain) ||
1459 wps_build_encr_settings(wps, msg, plain) ||
1460 wps_build_authenticator(wps, msg)) {
1461 wpabuf_free(plain);
1462 wpabuf_free(msg);
1463 return NULL;
1465 wpabuf_free(plain);
1467 wps->wps_pin_revealed = 1;
1468 wps->state = RECV_M7;
1469 return msg;
1473 static struct wpabuf * wps_build_m8(struct wps_data *wps)
1475 struct wpabuf *msg, *plain;
1477 wpa_printf(MSG_DEBUG, "WPS: Building Message M8");
1479 plain = wpabuf_alloc(500);
1480 if (plain == NULL)
1481 return NULL;
1483 msg = wpabuf_alloc(1000);
1484 if (msg == NULL) {
1485 wpabuf_free(plain);
1486 return NULL;
1489 if (wps_build_version(msg) ||
1490 wps_build_msg_type(msg, WPS_M8) ||
1491 wps_build_enrollee_nonce(wps, msg) ||
1492 ((wps->wps->ap || wps->er) && wps_build_cred(wps, plain)) ||
1493 (!wps->wps->ap && !wps->er && wps_build_ap_settings(wps, plain)) ||
1494 wps_build_key_wrap_auth(wps, plain) ||
1495 wps_build_encr_settings(wps, msg, plain) ||
1496 wps_build_authenticator(wps, msg)) {
1497 wpabuf_free(plain);
1498 wpabuf_free(msg);
1499 return NULL;
1501 wpabuf_free(plain);
1503 wps->state = RECV_DONE;
1504 return msg;
1508 static struct wpabuf * wps_build_wsc_ack(struct wps_data *wps)
1510 struct wpabuf *msg;
1512 wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_ACK");
1514 msg = wpabuf_alloc(1000);
1515 if (msg == NULL)
1516 return NULL;
1518 if (wps_build_version(msg) ||
1519 wps_build_msg_type(msg, WPS_WSC_ACK) ||
1520 wps_build_enrollee_nonce(wps, msg) ||
1521 wps_build_registrar_nonce(wps, msg)) {
1522 wpabuf_free(msg);
1523 return NULL;
1526 return msg;
1530 static struct wpabuf * wps_build_wsc_nack(struct wps_data *wps)
1532 struct wpabuf *msg;
1534 wpa_printf(MSG_DEBUG, "WPS: Building Message WSC_NACK");
1536 msg = wpabuf_alloc(1000);
1537 if (msg == NULL)
1538 return NULL;
1540 if (wps_build_version(msg) ||
1541 wps_build_msg_type(msg, WPS_WSC_NACK) ||
1542 wps_build_enrollee_nonce(wps, msg) ||
1543 wps_build_registrar_nonce(wps, msg) ||
1544 wps_build_config_error(msg, wps->config_error)) {
1545 wpabuf_free(msg);
1546 return NULL;
1549 return msg;
1553 struct wpabuf * wps_registrar_get_msg(struct wps_data *wps,
1554 enum wsc_op_code *op_code)
1556 struct wpabuf *msg;
1558 #ifdef CONFIG_WPS_UPNP
1559 if (!wps->int_reg && wps->wps->wps_upnp) {
1560 struct upnp_pending_message *p, *prev = NULL;
1561 if (wps->ext_reg > 1)
1562 wps_registrar_free_pending_m2(wps->wps);
1563 p = wps->wps->upnp_msgs;
1564 /* TODO: check pending message MAC address */
1565 while (p && p->next) {
1566 prev = p;
1567 p = p->next;
1569 if (p) {
1570 wpa_printf(MSG_DEBUG, "WPS: Use pending message from "
1571 "UPnP");
1572 if (prev)
1573 prev->next = NULL;
1574 else
1575 wps->wps->upnp_msgs = NULL;
1576 msg = p->msg;
1577 switch (p->type) {
1578 case WPS_WSC_ACK:
1579 *op_code = WSC_ACK;
1580 break;
1581 case WPS_WSC_NACK:
1582 *op_code = WSC_NACK;
1583 break;
1584 default:
1585 *op_code = WSC_MSG;
1586 break;
1588 os_free(p);
1589 if (wps->ext_reg == 0)
1590 wps->ext_reg = 1;
1591 return msg;
1594 if (wps->ext_reg) {
1595 wpa_printf(MSG_DEBUG, "WPS: Using external Registrar, but no "
1596 "pending message available");
1597 return NULL;
1599 #endif /* CONFIG_WPS_UPNP */
1601 switch (wps->state) {
1602 case SEND_M2:
1603 if (wps_get_dev_password(wps) < 0)
1604 msg = wps_build_m2d(wps);
1605 else
1606 msg = wps_build_m2(wps);
1607 *op_code = WSC_MSG;
1608 break;
1609 case SEND_M2D:
1610 msg = wps_build_m2d(wps);
1611 *op_code = WSC_MSG;
1612 break;
1613 case SEND_M4:
1614 msg = wps_build_m4(wps);
1615 *op_code = WSC_MSG;
1616 break;
1617 case SEND_M6:
1618 msg = wps_build_m6(wps);
1619 *op_code = WSC_MSG;
1620 break;
1621 case SEND_M8:
1622 msg = wps_build_m8(wps);
1623 *op_code = WSC_MSG;
1624 break;
1625 case RECV_DONE:
1626 msg = wps_build_wsc_ack(wps);
1627 *op_code = WSC_ACK;
1628 break;
1629 case SEND_WSC_NACK:
1630 msg = wps_build_wsc_nack(wps);
1631 *op_code = WSC_NACK;
1632 break;
1633 default:
1634 wpa_printf(MSG_DEBUG, "WPS: Unsupported state %d for building "
1635 "a message", wps->state);
1636 msg = NULL;
1637 break;
1640 if (*op_code == WSC_MSG && msg) {
1641 /* Save a copy of the last message for Authenticator derivation
1643 wpabuf_free(wps->last_msg);
1644 wps->last_msg = wpabuf_dup(msg);
1647 return msg;
1651 static int wps_process_enrollee_nonce(struct wps_data *wps, const u8 *e_nonce)
1653 if (e_nonce == NULL) {
1654 wpa_printf(MSG_DEBUG, "WPS: No Enrollee Nonce received");
1655 return -1;
1658 os_memcpy(wps->nonce_e, e_nonce, WPS_NONCE_LEN);
1659 wpa_hexdump(MSG_DEBUG, "WPS: Enrollee Nonce",
1660 wps->nonce_e, WPS_NONCE_LEN);
1662 return 0;
1666 static int wps_process_registrar_nonce(struct wps_data *wps, const u8 *r_nonce)
1668 if (r_nonce == NULL) {
1669 wpa_printf(MSG_DEBUG, "WPS: No Registrar Nonce received");
1670 return -1;
1673 if (os_memcmp(wps->nonce_r, r_nonce, WPS_NONCE_LEN) != 0) {
1674 wpa_printf(MSG_DEBUG, "WPS: Invalid Registrar Nonce received");
1675 return -1;
1678 return 0;
1682 static int wps_process_uuid_e(struct wps_data *wps, const u8 *uuid_e)
1684 if (uuid_e == NULL) {
1685 wpa_printf(MSG_DEBUG, "WPS: No UUID-E received");
1686 return -1;
1689 os_memcpy(wps->uuid_e, uuid_e, WPS_UUID_LEN);
1690 wpa_hexdump(MSG_DEBUG, "WPS: UUID-E", wps->uuid_e, WPS_UUID_LEN);
1692 return 0;
1696 static int wps_process_dev_password_id(struct wps_data *wps, const u8 *pw_id)
1698 if (pw_id == NULL) {
1699 wpa_printf(MSG_DEBUG, "WPS: No Device Password ID received");
1700 return -1;
1703 wps->dev_pw_id = WPA_GET_BE16(pw_id);
1704 wpa_printf(MSG_DEBUG, "WPS: Device Password ID %d", wps->dev_pw_id);
1706 return 0;
1710 static int wps_process_e_hash1(struct wps_data *wps, const u8 *e_hash1)
1712 if (e_hash1 == NULL) {
1713 wpa_printf(MSG_DEBUG, "WPS: No E-Hash1 received");
1714 return -1;
1717 os_memcpy(wps->peer_hash1, e_hash1, WPS_HASH_LEN);
1718 wpa_hexdump(MSG_DEBUG, "WPS: E-Hash1", wps->peer_hash1, WPS_HASH_LEN);
1720 return 0;
1724 static int wps_process_e_hash2(struct wps_data *wps, const u8 *e_hash2)
1726 if (e_hash2 == NULL) {
1727 wpa_printf(MSG_DEBUG, "WPS: No E-Hash2 received");
1728 return -1;
1731 os_memcpy(wps->peer_hash2, e_hash2, WPS_HASH_LEN);
1732 wpa_hexdump(MSG_DEBUG, "WPS: E-Hash2", wps->peer_hash2, WPS_HASH_LEN);
1734 return 0;
1738 static int wps_process_e_snonce1(struct wps_data *wps, const u8 *e_snonce1)
1740 u8 hash[SHA256_MAC_LEN];
1741 const u8 *addr[4];
1742 size_t len[4];
1744 if (e_snonce1 == NULL) {
1745 wpa_printf(MSG_DEBUG, "WPS: No E-SNonce1 received");
1746 return -1;
1749 wpa_hexdump_key(MSG_DEBUG, "WPS: E-SNonce1", e_snonce1,
1750 WPS_SECRET_NONCE_LEN);
1752 /* E-Hash1 = HMAC_AuthKey(E-S1 || PSK1 || PK_E || PK_R) */
1753 addr[0] = e_snonce1;
1754 len[0] = WPS_SECRET_NONCE_LEN;
1755 addr[1] = wps->psk1;
1756 len[1] = WPS_PSK_LEN;
1757 addr[2] = wpabuf_head(wps->dh_pubkey_e);
1758 len[2] = wpabuf_len(wps->dh_pubkey_e);
1759 addr[3] = wpabuf_head(wps->dh_pubkey_r);
1760 len[3] = wpabuf_len(wps->dh_pubkey_r);
1761 hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
1763 if (os_memcmp(wps->peer_hash1, hash, WPS_HASH_LEN) != 0) {
1764 wpa_printf(MSG_DEBUG, "WPS: E-Hash1 derived from E-S1 does "
1765 "not match with the pre-committed value");
1766 wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
1767 wps_pwd_auth_fail_event(wps->wps, 0, 1);
1768 return -1;
1771 wpa_printf(MSG_DEBUG, "WPS: Enrollee proved knowledge of the first "
1772 "half of the device password");
1774 return 0;
1778 static int wps_process_e_snonce2(struct wps_data *wps, const u8 *e_snonce2)
1780 u8 hash[SHA256_MAC_LEN];
1781 const u8 *addr[4];
1782 size_t len[4];
1784 if (e_snonce2 == NULL) {
1785 wpa_printf(MSG_DEBUG, "WPS: No E-SNonce2 received");
1786 return -1;
1789 wpa_hexdump_key(MSG_DEBUG, "WPS: E-SNonce2", e_snonce2,
1790 WPS_SECRET_NONCE_LEN);
1792 /* E-Hash2 = HMAC_AuthKey(E-S2 || PSK2 || PK_E || PK_R) */
1793 addr[0] = e_snonce2;
1794 len[0] = WPS_SECRET_NONCE_LEN;
1795 addr[1] = wps->psk2;
1796 len[1] = WPS_PSK_LEN;
1797 addr[2] = wpabuf_head(wps->dh_pubkey_e);
1798 len[2] = wpabuf_len(wps->dh_pubkey_e);
1799 addr[3] = wpabuf_head(wps->dh_pubkey_r);
1800 len[3] = wpabuf_len(wps->dh_pubkey_r);
1801 hmac_sha256_vector(wps->authkey, WPS_AUTHKEY_LEN, 4, addr, len, hash);
1803 if (os_memcmp(wps->peer_hash2, hash, WPS_HASH_LEN) != 0) {
1804 wpa_printf(MSG_DEBUG, "WPS: E-Hash2 derived from E-S2 does "
1805 "not match with the pre-committed value");
1806 wps_registrar_invalidate_pin(wps->wps->registrar, wps->uuid_e);
1807 wps->config_error = WPS_CFG_DEV_PASSWORD_AUTH_FAILURE;
1808 wps_pwd_auth_fail_event(wps->wps, 0, 2);
1809 return -1;
1812 wpa_printf(MSG_DEBUG, "WPS: Enrollee proved knowledge of the second "
1813 "half of the device password");
1814 wps->wps_pin_revealed = 0;
1815 wps_registrar_unlock_pin(wps->wps->registrar, wps->uuid_e);
1817 return 0;
1821 static int wps_process_mac_addr(struct wps_data *wps, const u8 *mac_addr)
1823 if (mac_addr == NULL) {
1824 wpa_printf(MSG_DEBUG, "WPS: No MAC Address received");
1825 return -1;
1828 wpa_printf(MSG_DEBUG, "WPS: Enrollee MAC Address " MACSTR,
1829 MAC2STR(mac_addr));
1830 os_memcpy(wps->mac_addr_e, mac_addr, ETH_ALEN);
1831 os_memcpy(wps->peer_dev.mac_addr, mac_addr, ETH_ALEN);
1833 return 0;
1837 static int wps_process_pubkey(struct wps_data *wps, const u8 *pk,
1838 size_t pk_len)
1840 if (pk == NULL || pk_len == 0) {
1841 wpa_printf(MSG_DEBUG, "WPS: No Public Key received");
1842 return -1;
1845 #ifdef CONFIG_WPS_OOB
1846 if (wps->wps->oob_conf.pubkey_hash != NULL) {
1847 const u8 *addr[1];
1848 u8 hash[WPS_HASH_LEN];
1850 addr[0] = pk;
1851 sha256_vector(1, addr, &pk_len, hash);
1852 if (os_memcmp(hash,
1853 wpabuf_head(wps->wps->oob_conf.pubkey_hash),
1854 WPS_OOB_PUBKEY_HASH_LEN) != 0) {
1855 wpa_printf(MSG_ERROR, "WPS: Public Key hash error");
1856 return -1;
1859 #endif /* CONFIG_WPS_OOB */
1861 wpabuf_free(wps->dh_pubkey_e);
1862 wps->dh_pubkey_e = wpabuf_alloc_copy(pk, pk_len);
1863 if (wps->dh_pubkey_e == NULL)
1864 return -1;
1866 return 0;
1870 static int wps_process_auth_type_flags(struct wps_data *wps, const u8 *auth)
1872 u16 auth_types;
1874 if (auth == NULL) {
1875 wpa_printf(MSG_DEBUG, "WPS: No Authentication Type flags "
1876 "received");
1877 return -1;
1880 auth_types = WPA_GET_BE16(auth);
1882 wpa_printf(MSG_DEBUG, "WPS: Enrollee Authentication Type flags 0x%x",
1883 auth_types);
1884 wps->auth_type = wps->wps->auth_types & auth_types;
1885 if (wps->auth_type == 0) {
1886 wpa_printf(MSG_DEBUG, "WPS: No match in supported "
1887 "authentication types (own 0x%x Enrollee 0x%x)",
1888 wps->wps->auth_types, auth_types);
1889 #ifdef WPS_WORKAROUNDS
1891 * Some deployed implementations seem to advertise incorrect
1892 * information in this attribute. For example, Linksys WRT350N
1893 * seems to have a byteorder bug that breaks this negotiation.
1894 * In order to interoperate with existing implementations,
1895 * assume that the Enrollee supports everything we do.
1897 wpa_printf(MSG_DEBUG, "WPS: Workaround - assume Enrollee "
1898 "does not advertise supported authentication types "
1899 "correctly");
1900 wps->auth_type = wps->wps->auth_types;
1901 #else /* WPS_WORKAROUNDS */
1902 return -1;
1903 #endif /* WPS_WORKAROUNDS */
1906 return 0;
1910 static int wps_process_encr_type_flags(struct wps_data *wps, const u8 *encr)
1912 u16 encr_types;
1914 if (encr == NULL) {
1915 wpa_printf(MSG_DEBUG, "WPS: No Encryption Type flags "
1916 "received");
1917 return -1;
1920 encr_types = WPA_GET_BE16(encr);
1922 wpa_printf(MSG_DEBUG, "WPS: Enrollee Encryption Type flags 0x%x",
1923 encr_types);
1924 wps->encr_type = wps->wps->encr_types & encr_types;
1925 if (wps->encr_type == 0) {
1926 wpa_printf(MSG_DEBUG, "WPS: No match in supported "
1927 "encryption types (own 0x%x Enrollee 0x%x)",
1928 wps->wps->encr_types, encr_types);
1929 #ifdef WPS_WORKAROUNDS
1931 * Some deployed implementations seem to advertise incorrect
1932 * information in this attribute. For example, Linksys WRT350N
1933 * seems to have a byteorder bug that breaks this negotiation.
1934 * In order to interoperate with existing implementations,
1935 * assume that the Enrollee supports everything we do.
1937 wpa_printf(MSG_DEBUG, "WPS: Workaround - assume Enrollee "
1938 "does not advertise supported encryption types "
1939 "correctly");
1940 wps->encr_type = wps->wps->encr_types;
1941 #else /* WPS_WORKAROUNDS */
1942 return -1;
1943 #endif /* WPS_WORKAROUNDS */
1946 return 0;
1950 static int wps_process_conn_type_flags(struct wps_data *wps, const u8 *conn)
1952 if (conn == NULL) {
1953 wpa_printf(MSG_DEBUG, "WPS: No Connection Type flags "
1954 "received");
1955 return -1;
1958 wpa_printf(MSG_DEBUG, "WPS: Enrollee Connection Type flags 0x%x",
1959 *conn);
1961 return 0;
1965 static int wps_process_config_methods(struct wps_data *wps, const u8 *methods)
1967 u16 m;
1969 if (methods == NULL) {
1970 wpa_printf(MSG_DEBUG, "WPS: No Config Methods received");
1971 return -1;
1974 m = WPA_GET_BE16(methods);
1976 wpa_printf(MSG_DEBUG, "WPS: Enrollee Config Methods 0x%x"
1977 "%s%s%s%s%s%s%s%s%s", m,
1978 m & WPS_CONFIG_USBA ? " [USBA]" : "",
1979 m & WPS_CONFIG_ETHERNET ? " [Ethernet]" : "",
1980 m & WPS_CONFIG_LABEL ? " [Label]" : "",
1981 m & WPS_CONFIG_DISPLAY ? " [Display]" : "",
1982 m & WPS_CONFIG_EXT_NFC_TOKEN ? " [Ext NFC Token]" : "",
1983 m & WPS_CONFIG_INT_NFC_TOKEN ? " [Int NFC Token]" : "",
1984 m & WPS_CONFIG_NFC_INTERFACE ? " [NFC]" : "",
1985 m & WPS_CONFIG_PUSHBUTTON ? " [PBC]" : "",
1986 m & WPS_CONFIG_KEYPAD ? " [Keypad]" : "");
1988 if (!(m & WPS_CONFIG_DISPLAY) && !wps->use_psk_key) {
1990 * The Enrollee does not have a display so it is unlikely to be
1991 * able to show the passphrase to a user and as such, could
1992 * benefit from receiving PSK to reduce key derivation time.
1994 wpa_printf(MSG_DEBUG, "WPS: Prefer PSK format key due to "
1995 "Enrollee not supporting display");
1996 wps->use_psk_key = 1;
1999 return 0;
2003 static int wps_process_wps_state(struct wps_data *wps, const u8 *state)
2005 if (state == NULL) {
2006 wpa_printf(MSG_DEBUG, "WPS: No Wi-Fi Protected Setup State "
2007 "received");
2008 return -1;
2011 wpa_printf(MSG_DEBUG, "WPS: Enrollee Wi-Fi Protected Setup State %d",
2012 *state);
2014 return 0;
2018 static int wps_process_assoc_state(struct wps_data *wps, const u8 *assoc)
2020 u16 a;
2022 if (assoc == NULL) {
2023 wpa_printf(MSG_DEBUG, "WPS: No Association State received");
2024 return -1;
2027 a = WPA_GET_BE16(assoc);
2028 wpa_printf(MSG_DEBUG, "WPS: Enrollee Association State %d", a);
2030 return 0;
2034 static int wps_process_config_error(struct wps_data *wps, const u8 *err)
2036 u16 e;
2038 if (err == NULL) {
2039 wpa_printf(MSG_DEBUG, "WPS: No Configuration Error received");
2040 return -1;
2043 e = WPA_GET_BE16(err);
2044 wpa_printf(MSG_DEBUG, "WPS: Enrollee Configuration Error %d", e);
2046 return 0;
2050 static enum wps_process_res wps_process_m1(struct wps_data *wps,
2051 struct wps_parse_attr *attr)
2053 wpa_printf(MSG_DEBUG, "WPS: Received M1");
2055 if (wps->state != RECV_M1) {
2056 wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
2057 "receiving M1", wps->state);
2058 return WPS_FAILURE;
2061 if (wps_process_uuid_e(wps, attr->uuid_e) ||
2062 wps_process_mac_addr(wps, attr->mac_addr) ||
2063 wps_process_enrollee_nonce(wps, attr->enrollee_nonce) ||
2064 wps_process_pubkey(wps, attr->public_key, attr->public_key_len) ||
2065 wps_process_auth_type_flags(wps, attr->auth_type_flags) ||
2066 wps_process_encr_type_flags(wps, attr->encr_type_flags) ||
2067 wps_process_conn_type_flags(wps, attr->conn_type_flags) ||
2068 wps_process_config_methods(wps, attr->config_methods) ||
2069 wps_process_wps_state(wps, attr->wps_state) ||
2070 wps_process_device_attrs(&wps->peer_dev, attr) ||
2071 wps_process_rf_bands(&wps->peer_dev, attr->rf_bands) ||
2072 wps_process_assoc_state(wps, attr->assoc_state) ||
2073 wps_process_dev_password_id(wps, attr->dev_password_id) ||
2074 wps_process_config_error(wps, attr->config_error) ||
2075 wps_process_os_version(&wps->peer_dev, attr->os_version))
2076 return WPS_FAILURE;
2078 if (wps->dev_pw_id < 0x10 &&
2079 wps->dev_pw_id != DEV_PW_DEFAULT &&
2080 wps->dev_pw_id != DEV_PW_USER_SPECIFIED &&
2081 wps->dev_pw_id != DEV_PW_MACHINE_SPECIFIED &&
2082 wps->dev_pw_id != DEV_PW_REGISTRAR_SPECIFIED &&
2083 (wps->dev_pw_id != DEV_PW_PUSHBUTTON ||
2084 !wps->wps->registrar->pbc)) {
2085 wpa_printf(MSG_DEBUG, "WPS: Unsupported Device Password ID %d",
2086 wps->dev_pw_id);
2087 wps->state = SEND_M2D;
2088 return WPS_CONTINUE;
2091 #ifdef CONFIG_WPS_OOB
2092 if (wps->dev_pw_id >= 0x10 &&
2093 wps->dev_pw_id != wps->wps->oob_dev_pw_id) {
2094 wpa_printf(MSG_DEBUG, "WPS: OOB Device Password ID "
2095 "%d mismatch", wps->dev_pw_id);
2096 wps->state = SEND_M2D;
2097 return WPS_CONTINUE;
2099 #endif /* CONFIG_WPS_OOB */
2101 if (wps->dev_pw_id == DEV_PW_PUSHBUTTON) {
2102 if (wps->wps->registrar->force_pbc_overlap ||
2103 wps_registrar_pbc_overlap(wps->wps->registrar,
2104 wps->mac_addr_e, wps->uuid_e)) {
2105 wpa_printf(MSG_DEBUG, "WPS: PBC overlap - deny PBC "
2106 "negotiation");
2107 wps->state = SEND_M2D;
2108 wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
2109 wps_pbc_overlap_event(wps->wps);
2110 wps->wps->registrar->force_pbc_overlap = 1;
2111 return WPS_CONTINUE;
2113 wps_registrar_add_pbc_session(wps->wps->registrar,
2114 wps->mac_addr_e, wps->uuid_e);
2115 wps->pbc = 1;
2118 #ifdef WPS_WORKAROUNDS
2120 * It looks like Mac OS X 10.6.3 and 10.6.4 do not like Network Key in
2121 * passphrase format. To avoid interop issues, force PSK format to be
2122 * used.
2124 if (!wps->use_psk_key &&
2125 wps->peer_dev.manufacturer &&
2126 os_strncmp(wps->peer_dev.manufacturer, "Apple ", 6) == 0 &&
2127 wps->peer_dev.model_name &&
2128 os_strcmp(wps->peer_dev.model_name, "AirPort") == 0) {
2129 wpa_printf(MSG_DEBUG, "WPS: Workaround - Force Network Key in "
2130 "PSK format");
2131 wps->use_psk_key = 1;
2133 #endif /* WPS_WORKAROUNDS */
2135 wps->state = SEND_M2;
2136 return WPS_CONTINUE;
2140 static enum wps_process_res wps_process_m3(struct wps_data *wps,
2141 const struct wpabuf *msg,
2142 struct wps_parse_attr *attr)
2144 wpa_printf(MSG_DEBUG, "WPS: Received M3");
2146 if (wps->state != RECV_M3) {
2147 wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
2148 "receiving M3", wps->state);
2149 wps->state = SEND_WSC_NACK;
2150 return WPS_CONTINUE;
2153 if (wps->pbc && wps->wps->registrar->force_pbc_overlap) {
2154 wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
2155 "session overlap");
2156 wps->state = SEND_WSC_NACK;
2157 wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
2158 return WPS_CONTINUE;
2161 if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
2162 wps_process_authenticator(wps, attr->authenticator, msg) ||
2163 wps_process_e_hash1(wps, attr->e_hash1) ||
2164 wps_process_e_hash2(wps, attr->e_hash2)) {
2165 wps->state = SEND_WSC_NACK;
2166 return WPS_CONTINUE;
2169 wps->state = SEND_M4;
2170 return WPS_CONTINUE;
2174 static enum wps_process_res wps_process_m5(struct wps_data *wps,
2175 const struct wpabuf *msg,
2176 struct wps_parse_attr *attr)
2178 struct wpabuf *decrypted;
2179 struct wps_parse_attr eattr;
2181 wpa_printf(MSG_DEBUG, "WPS: Received M5");
2183 if (wps->state != RECV_M5) {
2184 wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
2185 "receiving M5", wps->state);
2186 wps->state = SEND_WSC_NACK;
2187 return WPS_CONTINUE;
2190 if (wps->pbc && wps->wps->registrar->force_pbc_overlap) {
2191 wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
2192 "session overlap");
2193 wps->state = SEND_WSC_NACK;
2194 wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
2195 return WPS_CONTINUE;
2198 if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
2199 wps_process_authenticator(wps, attr->authenticator, msg)) {
2200 wps->state = SEND_WSC_NACK;
2201 return WPS_CONTINUE;
2204 decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
2205 attr->encr_settings_len);
2206 if (decrypted == NULL) {
2207 wpa_printf(MSG_DEBUG, "WPS: Failed to decrypted Encrypted "
2208 "Settings attribute");
2209 wps->state = SEND_WSC_NACK;
2210 return WPS_CONTINUE;
2213 wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
2214 "attribute");
2215 if (wps_parse_msg(decrypted, &eattr) < 0 ||
2216 wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
2217 wps_process_e_snonce1(wps, eattr.e_snonce1)) {
2218 wpabuf_free(decrypted);
2219 wps->state = SEND_WSC_NACK;
2220 return WPS_CONTINUE;
2222 wpabuf_free(decrypted);
2224 wps->state = SEND_M6;
2225 return WPS_CONTINUE;
2229 static void wps_sta_cred_cb(struct wps_data *wps)
2232 * Update credential to only include a single authentication and
2233 * encryption type in case the AP configuration includes more than one
2234 * option.
2236 if (wps->cred.auth_type & WPS_AUTH_WPA2PSK)
2237 wps->cred.auth_type = WPS_AUTH_WPA2PSK;
2238 else if (wps->cred.auth_type & WPS_AUTH_WPAPSK)
2239 wps->cred.auth_type = WPS_AUTH_WPAPSK;
2240 if (wps->cred.encr_type & WPS_ENCR_AES)
2241 wps->cred.encr_type = WPS_ENCR_AES;
2242 else if (wps->cred.encr_type & WPS_ENCR_TKIP)
2243 wps->cred.encr_type = WPS_ENCR_TKIP;
2244 wpa_printf(MSG_DEBUG, "WPS: Update local configuration based on the "
2245 "AP configuration");
2246 if (wps->wps->cred_cb)
2247 wps->wps->cred_cb(wps->wps->cb_ctx, &wps->cred);
2251 static void wps_cred_update(struct wps_credential *dst,
2252 struct wps_credential *src)
2254 os_memcpy(dst->ssid, src->ssid, sizeof(dst->ssid));
2255 dst->ssid_len = src->ssid_len;
2256 dst->auth_type = src->auth_type;
2257 dst->encr_type = src->encr_type;
2258 dst->key_idx = src->key_idx;
2259 os_memcpy(dst->key, src->key, sizeof(dst->key));
2260 dst->key_len = src->key_len;
2264 static int wps_process_ap_settings_r(struct wps_data *wps,
2265 struct wps_parse_attr *attr)
2267 if (wps->wps->ap || wps->er)
2268 return 0;
2270 /* AP Settings Attributes in M7 when Enrollee is an AP */
2271 if (wps_process_ap_settings(attr, &wps->cred) < 0)
2272 return -1;
2274 wpa_printf(MSG_INFO, "WPS: Received old AP configuration from AP");
2276 if (wps->new_ap_settings) {
2277 wpa_printf(MSG_INFO, "WPS: Update AP configuration based on "
2278 "new settings");
2279 wps_cred_update(&wps->cred, wps->new_ap_settings);
2280 return 0;
2281 } else {
2283 * Use the AP PIN only to receive the current AP settings, not
2284 * to reconfigure the AP.
2286 if (wps->ap_settings_cb) {
2287 wps->ap_settings_cb(wps->ap_settings_cb_ctx,
2288 &wps->cred);
2289 return 1;
2291 wps_sta_cred_cb(wps);
2292 return 1;
2297 static enum wps_process_res wps_process_m7(struct wps_data *wps,
2298 const struct wpabuf *msg,
2299 struct wps_parse_attr *attr)
2301 struct wpabuf *decrypted;
2302 struct wps_parse_attr eattr;
2304 wpa_printf(MSG_DEBUG, "WPS: Received M7");
2306 if (wps->state != RECV_M7) {
2307 wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
2308 "receiving M7", wps->state);
2309 wps->state = SEND_WSC_NACK;
2310 return WPS_CONTINUE;
2313 if (wps->pbc && wps->wps->registrar->force_pbc_overlap) {
2314 wpa_printf(MSG_DEBUG, "WPS: Reject negotiation due to PBC "
2315 "session overlap");
2316 wps->state = SEND_WSC_NACK;
2317 wps->config_error = WPS_CFG_MULTIPLE_PBC_DETECTED;
2318 return WPS_CONTINUE;
2321 if (wps_process_registrar_nonce(wps, attr->registrar_nonce) ||
2322 wps_process_authenticator(wps, attr->authenticator, msg)) {
2323 wps->state = SEND_WSC_NACK;
2324 return WPS_CONTINUE;
2327 decrypted = wps_decrypt_encr_settings(wps, attr->encr_settings,
2328 attr->encr_settings_len);
2329 if (decrypted == NULL) {
2330 wpa_printf(MSG_DEBUG, "WPS: Failed to decrypt Encrypted "
2331 "Settings attribute");
2332 wps->state = SEND_WSC_NACK;
2333 return WPS_CONTINUE;
2336 wpa_printf(MSG_DEBUG, "WPS: Processing decrypted Encrypted Settings "
2337 "attribute");
2338 if (wps_parse_msg(decrypted, &eattr) < 0 ||
2339 wps_process_key_wrap_auth(wps, decrypted, eattr.key_wrap_auth) ||
2340 wps_process_e_snonce2(wps, eattr.e_snonce2) ||
2341 wps_process_ap_settings_r(wps, &eattr)) {
2342 wpabuf_free(decrypted);
2343 wps->state = SEND_WSC_NACK;
2344 return WPS_CONTINUE;
2347 wpabuf_free(decrypted);
2349 wps->state = SEND_M8;
2350 return WPS_CONTINUE;
2354 static enum wps_process_res wps_process_wsc_msg(struct wps_data *wps,
2355 const struct wpabuf *msg)
2357 struct wps_parse_attr attr;
2358 enum wps_process_res ret = WPS_CONTINUE;
2360 wpa_printf(MSG_DEBUG, "WPS: Received WSC_MSG");
2362 if (wps_parse_msg(msg, &attr) < 0)
2363 return WPS_FAILURE;
2365 if (!wps_version_supported(attr.version)) {
2366 wpa_printf(MSG_DEBUG, "WPS: Unsupported message version 0x%x",
2367 attr.version ? *attr.version : 0);
2368 return WPS_FAILURE;
2371 if (attr.msg_type == NULL) {
2372 wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
2373 return WPS_FAILURE;
2376 if (*attr.msg_type != WPS_M1 &&
2377 (attr.registrar_nonce == NULL ||
2378 os_memcmp(wps->nonce_r, attr.registrar_nonce,
2379 WPS_NONCE_LEN != 0))) {
2380 wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
2381 return WPS_FAILURE;
2384 switch (*attr.msg_type) {
2385 case WPS_M1:
2386 #ifdef CONFIG_WPS_UPNP
2387 if (wps->wps->wps_upnp && attr.mac_addr) {
2388 /* Remove old pending messages when starting new run */
2389 wps_free_pending_msgs(wps->wps->upnp_msgs);
2390 wps->wps->upnp_msgs = NULL;
2392 upnp_wps_device_send_wlan_event(
2393 wps->wps->wps_upnp, attr.mac_addr,
2394 UPNP_WPS_WLANEVENT_TYPE_EAP, msg);
2396 #endif /* CONFIG_WPS_UPNP */
2397 ret = wps_process_m1(wps, &attr);
2398 break;
2399 case WPS_M3:
2400 ret = wps_process_m3(wps, msg, &attr);
2401 if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
2402 wps_fail_event(wps->wps, WPS_M3);
2403 break;
2404 case WPS_M5:
2405 ret = wps_process_m5(wps, msg, &attr);
2406 if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
2407 wps_fail_event(wps->wps, WPS_M5);
2408 break;
2409 case WPS_M7:
2410 ret = wps_process_m7(wps, msg, &attr);
2411 if (ret == WPS_FAILURE || wps->state == SEND_WSC_NACK)
2412 wps_fail_event(wps->wps, WPS_M7);
2413 break;
2414 default:
2415 wpa_printf(MSG_DEBUG, "WPS: Unsupported Message Type %d",
2416 *attr.msg_type);
2417 return WPS_FAILURE;
2420 if (ret == WPS_CONTINUE) {
2421 /* Save a copy of the last message for Authenticator derivation
2423 wpabuf_free(wps->last_msg);
2424 wps->last_msg = wpabuf_dup(msg);
2427 return ret;
2431 static enum wps_process_res wps_process_wsc_ack(struct wps_data *wps,
2432 const struct wpabuf *msg)
2434 struct wps_parse_attr attr;
2436 wpa_printf(MSG_DEBUG, "WPS: Received WSC_ACK");
2438 if (wps_parse_msg(msg, &attr) < 0)
2439 return WPS_FAILURE;
2441 if (!wps_version_supported(attr.version)) {
2442 wpa_printf(MSG_DEBUG, "WPS: Unsupported message version 0x%x",
2443 attr.version ? *attr.version : 0);
2444 return WPS_FAILURE;
2447 if (attr.msg_type == NULL) {
2448 wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
2449 return WPS_FAILURE;
2452 if (*attr.msg_type != WPS_WSC_ACK) {
2453 wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
2454 *attr.msg_type);
2455 return WPS_FAILURE;
2458 #ifdef CONFIG_WPS_UPNP
2459 if (wps->wps->wps_upnp && wps->ext_reg && wps->state == RECV_M2D_ACK &&
2460 upnp_wps_subscribers(wps->wps->wps_upnp)) {
2461 if (wps->wps->upnp_msgs)
2462 return WPS_CONTINUE;
2463 wpa_printf(MSG_DEBUG, "WPS: Wait for response from an "
2464 "external Registrar");
2465 return WPS_PENDING;
2467 #endif /* CONFIG_WPS_UPNP */
2469 if (attr.registrar_nonce == NULL ||
2470 os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN != 0))
2472 wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
2473 return WPS_FAILURE;
2476 if (attr.enrollee_nonce == NULL ||
2477 os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN != 0)) {
2478 wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
2479 return WPS_FAILURE;
2482 if (wps->state == RECV_M2D_ACK) {
2483 #ifdef CONFIG_WPS_UPNP
2484 if (wps->wps->wps_upnp &&
2485 upnp_wps_subscribers(wps->wps->wps_upnp)) {
2486 if (wps->wps->upnp_msgs)
2487 return WPS_CONTINUE;
2488 if (wps->ext_reg == 0)
2489 wps->ext_reg = 1;
2490 wpa_printf(MSG_DEBUG, "WPS: Wait for response from an "
2491 "external Registrar");
2492 return WPS_PENDING;
2494 #endif /* CONFIG_WPS_UPNP */
2496 wpa_printf(MSG_DEBUG, "WPS: No more registrars available - "
2497 "terminate negotiation");
2500 return WPS_FAILURE;
2504 static enum wps_process_res wps_process_wsc_nack(struct wps_data *wps,
2505 const struct wpabuf *msg)
2507 struct wps_parse_attr attr;
2508 int old_state;
2510 wpa_printf(MSG_DEBUG, "WPS: Received WSC_NACK");
2512 old_state = wps->state;
2513 wps->state = SEND_WSC_NACK;
2515 if (wps_parse_msg(msg, &attr) < 0)
2516 return WPS_FAILURE;
2518 if (!wps_version_supported(attr.version)) {
2519 wpa_printf(MSG_DEBUG, "WPS: Unsupported message version 0x%x",
2520 attr.version ? *attr.version : 0);
2521 return WPS_FAILURE;
2524 if (attr.msg_type == NULL) {
2525 wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
2526 return WPS_FAILURE;
2529 if (*attr.msg_type != WPS_WSC_NACK) {
2530 wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
2531 *attr.msg_type);
2532 return WPS_FAILURE;
2535 #ifdef CONFIG_WPS_UPNP
2536 if (wps->wps->wps_upnp && wps->ext_reg) {
2537 wpa_printf(MSG_DEBUG, "WPS: Negotiation using external "
2538 "Registrar terminated by the Enrollee");
2539 return WPS_FAILURE;
2541 #endif /* CONFIG_WPS_UPNP */
2543 if (attr.registrar_nonce == NULL ||
2544 os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN != 0))
2546 wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
2547 return WPS_FAILURE;
2550 if (attr.enrollee_nonce == NULL ||
2551 os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN != 0)) {
2552 wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
2553 return WPS_FAILURE;
2556 if (attr.config_error == NULL) {
2557 wpa_printf(MSG_DEBUG, "WPS: No Configuration Error attribute "
2558 "in WSC_NACK");
2559 return WPS_FAILURE;
2562 wpa_printf(MSG_DEBUG, "WPS: Enrollee terminated negotiation with "
2563 "Configuration Error %d", WPA_GET_BE16(attr.config_error));
2565 switch (old_state) {
2566 case RECV_M3:
2567 wps_fail_event(wps->wps, WPS_M2);
2568 break;
2569 case RECV_M5:
2570 wps_fail_event(wps->wps, WPS_M4);
2571 break;
2572 case RECV_M7:
2573 wps_fail_event(wps->wps, WPS_M6);
2574 break;
2575 case RECV_DONE:
2576 wps_fail_event(wps->wps, WPS_M8);
2577 break;
2578 default:
2579 break;
2582 return WPS_FAILURE;
2586 static enum wps_process_res wps_process_wsc_done(struct wps_data *wps,
2587 const struct wpabuf *msg)
2589 struct wps_parse_attr attr;
2591 wpa_printf(MSG_DEBUG, "WPS: Received WSC_Done");
2593 if (wps->state != RECV_DONE &&
2594 (!wps->wps->wps_upnp || !wps->ext_reg)) {
2595 wpa_printf(MSG_DEBUG, "WPS: Unexpected state (%d) for "
2596 "receiving WSC_Done", wps->state);
2597 return WPS_FAILURE;
2600 if (wps_parse_msg(msg, &attr) < 0)
2601 return WPS_FAILURE;
2603 if (!wps_version_supported(attr.version)) {
2604 wpa_printf(MSG_DEBUG, "WPS: Unsupported message version 0x%x",
2605 attr.version ? *attr.version : 0);
2606 return WPS_FAILURE;
2609 if (attr.msg_type == NULL) {
2610 wpa_printf(MSG_DEBUG, "WPS: No Message Type attribute");
2611 return WPS_FAILURE;
2614 if (*attr.msg_type != WPS_WSC_DONE) {
2615 wpa_printf(MSG_DEBUG, "WPS: Invalid Message Type %d",
2616 *attr.msg_type);
2617 return WPS_FAILURE;
2620 #ifdef CONFIG_WPS_UPNP
2621 if (wps->wps->wps_upnp && wps->ext_reg) {
2622 wpa_printf(MSG_DEBUG, "WPS: Negotiation using external "
2623 "Registrar completed successfully");
2624 wps_device_store(wps->wps->registrar, &wps->peer_dev,
2625 wps->uuid_e);
2626 return WPS_DONE;
2628 #endif /* CONFIG_WPS_UPNP */
2630 if (attr.registrar_nonce == NULL ||
2631 os_memcmp(wps->nonce_r, attr.registrar_nonce, WPS_NONCE_LEN != 0))
2633 wpa_printf(MSG_DEBUG, "WPS: Mismatch in registrar nonce");
2634 return WPS_FAILURE;
2637 if (attr.enrollee_nonce == NULL ||
2638 os_memcmp(wps->nonce_e, attr.enrollee_nonce, WPS_NONCE_LEN != 0)) {
2639 wpa_printf(MSG_DEBUG, "WPS: Mismatch in enrollee nonce");
2640 return WPS_FAILURE;
2643 wpa_printf(MSG_DEBUG, "WPS: Negotiation completed successfully");
2644 wps_device_store(wps->wps->registrar, &wps->peer_dev,
2645 wps->uuid_e);
2647 if (wps->wps->wps_state == WPS_STATE_NOT_CONFIGURED && wps->new_psk &&
2648 wps->wps->ap && !wps->wps->registrar->disable_auto_conf) {
2649 struct wps_credential cred;
2651 wpa_printf(MSG_DEBUG, "WPS: Moving to Configured state based "
2652 "on first Enrollee connection");
2654 os_memset(&cred, 0, sizeof(cred));
2655 os_memcpy(cred.ssid, wps->wps->ssid, wps->wps->ssid_len);
2656 cred.ssid_len = wps->wps->ssid_len;
2657 cred.auth_type = WPS_AUTH_WPAPSK | WPS_AUTH_WPA2PSK;
2658 cred.encr_type = WPS_ENCR_TKIP | WPS_ENCR_AES;
2659 os_memcpy(cred.key, wps->new_psk, wps->new_psk_len);
2660 cred.key_len = wps->new_psk_len;
2662 wps->wps->wps_state = WPS_STATE_CONFIGURED;
2663 wpa_hexdump_ascii_key(MSG_DEBUG,
2664 "WPS: Generated random passphrase",
2665 wps->new_psk, wps->new_psk_len);
2666 if (wps->wps->cred_cb)
2667 wps->wps->cred_cb(wps->wps->cb_ctx, &cred);
2669 os_free(wps->new_psk);
2670 wps->new_psk = NULL;
2673 if (!wps->wps->ap && !wps->er)
2674 wps_sta_cred_cb(wps);
2676 if (wps->new_psk) {
2677 if (wps_cb_new_psk(wps->wps->registrar, wps->mac_addr_e,
2678 wps->new_psk, wps->new_psk_len)) {
2679 wpa_printf(MSG_DEBUG, "WPS: Failed to configure the "
2680 "new PSK");
2682 os_free(wps->new_psk);
2683 wps->new_psk = NULL;
2686 wps_cb_reg_success(wps->wps->registrar, wps->mac_addr_e, wps->uuid_e);
2688 if (wps->pbc) {
2689 wps_registrar_remove_pbc_session(wps->wps->registrar,
2690 wps->mac_addr_e, wps->uuid_e);
2691 wps_registrar_pbc_completed(wps->wps->registrar);
2692 } else {
2693 wps_registrar_pin_completed(wps->wps->registrar);
2696 wps_success_event(wps->wps);
2698 return WPS_DONE;
2702 enum wps_process_res wps_registrar_process_msg(struct wps_data *wps,
2703 enum wsc_op_code op_code,
2704 const struct wpabuf *msg)
2706 enum wps_process_res ret;
2708 wpa_printf(MSG_DEBUG, "WPS: Processing received message (len=%lu "
2709 "op_code=%d)",
2710 (unsigned long) wpabuf_len(msg), op_code);
2712 #ifdef CONFIG_WPS_UPNP
2713 if (wps->wps->wps_upnp && op_code == WSC_MSG && wps->ext_reg == 1) {
2714 struct wps_parse_attr attr;
2715 if (wps_parse_msg(msg, &attr) == 0 && attr.msg_type &&
2716 *attr.msg_type == WPS_M3)
2717 wps->ext_reg = 2; /* past M2/M2D phase */
2719 if (wps->ext_reg > 1)
2720 wps_registrar_free_pending_m2(wps->wps);
2721 if (wps->wps->wps_upnp && wps->ext_reg &&
2722 wps->wps->upnp_msgs == NULL &&
2723 (op_code == WSC_MSG || op_code == WSC_Done || op_code == WSC_NACK))
2725 struct wps_parse_attr attr;
2726 int type;
2727 if (wps_parse_msg(msg, &attr) < 0 || attr.msg_type == NULL)
2728 type = -1;
2729 else
2730 type = *attr.msg_type;
2731 wpa_printf(MSG_DEBUG, "WPS: Sending received message (type %d)"
2732 " to external Registrar for processing", type);
2733 upnp_wps_device_send_wlan_event(wps->wps->wps_upnp,
2734 wps->mac_addr_e,
2735 UPNP_WPS_WLANEVENT_TYPE_EAP,
2736 msg);
2737 if (op_code == WSC_MSG)
2738 return WPS_PENDING;
2739 } else if (wps->wps->wps_upnp && wps->ext_reg && op_code == WSC_MSG) {
2740 wpa_printf(MSG_DEBUG, "WPS: Skip internal processing - using "
2741 "external Registrar");
2742 return WPS_CONTINUE;
2744 #endif /* CONFIG_WPS_UPNP */
2746 switch (op_code) {
2747 case WSC_MSG:
2748 return wps_process_wsc_msg(wps, msg);
2749 case WSC_ACK:
2750 return wps_process_wsc_ack(wps, msg);
2751 case WSC_NACK:
2752 return wps_process_wsc_nack(wps, msg);
2753 case WSC_Done:
2754 ret = wps_process_wsc_done(wps, msg);
2755 if (ret == WPS_FAILURE) {
2756 wps->state = SEND_WSC_NACK;
2757 wps_fail_event(wps->wps, WPS_WSC_DONE);
2759 return ret;
2760 default:
2761 wpa_printf(MSG_DEBUG, "WPS: Unsupported op_code %d", op_code);
2762 return WPS_FAILURE;
2767 int wps_registrar_update_ie(struct wps_registrar *reg)
2769 return wps_set_ie(reg);
2773 static void wps_registrar_set_selected_timeout(void *eloop_ctx,
2774 void *timeout_ctx)
2776 struct wps_registrar *reg = eloop_ctx;
2778 wpa_printf(MSG_DEBUG, "WPS: Selected Registrar timeout - "
2779 "unselect internal Registrar");
2780 reg->selected_registrar = 0;
2781 reg->pbc = 0;
2782 wps_registrar_selected_registrar_changed(reg);
2786 #ifdef CONFIG_WPS_UPNP
2787 static void wps_registrar_sel_reg_add(struct wps_registrar *reg,
2788 struct subscription *s)
2790 wpa_printf(MSG_DEBUG, "WPS: External Registrar selected (dev_pw_id=%d "
2791 "config_methods=0x%x)",
2792 s->dev_password_id, s->config_methods);
2793 reg->sel_reg_union = 1;
2794 if (reg->sel_reg_dev_password_id_override != DEV_PW_PUSHBUTTON)
2795 reg->sel_reg_dev_password_id_override = s->dev_password_id;
2796 if (reg->sel_reg_config_methods_override == -1)
2797 reg->sel_reg_config_methods_override = 0;
2798 reg->sel_reg_config_methods_override |= s->config_methods;
2800 #endif /* CONFIG_WPS_UPNP */
2803 static void wps_registrar_sel_reg_union(struct wps_registrar *reg)
2805 #ifdef CONFIG_WPS_UPNP
2806 struct subscription *s;
2808 if (reg->wps->wps_upnp == NULL)
2809 return;
2811 dl_list_for_each(s, &reg->wps->wps_upnp->subscriptions,
2812 struct subscription, list) {
2813 struct subscr_addr *sa;
2814 sa = dl_list_first(&s->addr_list, struct subscr_addr, list);
2815 if (sa) {
2816 wpa_printf(MSG_DEBUG, "WPS: External Registrar %s:%d",
2817 inet_ntoa(sa->saddr.sin_addr),
2818 ntohs(sa->saddr.sin_port));
2820 if (s->selected_registrar)
2821 wps_registrar_sel_reg_add(reg, s);
2822 else
2823 wpa_printf(MSG_DEBUG, "WPS: External Registrar not "
2824 "selected");
2826 #endif /* CONFIG_WPS_UPNP */
2831 * wps_registrar_selected_registrar_changed - SetSelectedRegistrar change
2832 * @reg: Registrar data from wps_registrar_init()
2834 * This function is called when selected registrar state changes, e.g., when an
2835 * AP receives a SetSelectedRegistrar UPnP message.
2837 void wps_registrar_selected_registrar_changed(struct wps_registrar *reg)
2839 wpa_printf(MSG_DEBUG, "WPS: Selected registrar information changed");
2841 reg->sel_reg_union = reg->selected_registrar;
2842 reg->sel_reg_dev_password_id_override = -1;
2843 reg->sel_reg_config_methods_override = -1;
2844 if (reg->selected_registrar) {
2845 reg->sel_reg_config_methods_override =
2846 reg->wps->config_methods & ~WPS_CONFIG_PUSHBUTTON;
2847 if (reg->pbc) {
2848 reg->sel_reg_dev_password_id_override =
2849 DEV_PW_PUSHBUTTON;
2850 reg->sel_reg_config_methods_override |=
2851 WPS_CONFIG_PUSHBUTTON;
2853 wpa_printf(MSG_DEBUG, "WPS: Internal Registrar selected "
2854 "(pbc=%d)", reg->pbc);
2855 } else
2856 wpa_printf(MSG_DEBUG, "WPS: Internal Registrar not selected");
2858 wps_registrar_sel_reg_union(reg);
2860 wps_set_ie(reg);
2861 wps_cb_set_sel_reg(reg);
2865 int wps_registrar_get_info(struct wps_registrar *reg, const u8 *addr,
2866 char *buf, size_t buflen)
2868 struct wps_registrar_device *d;
2869 int len = 0, ret;
2870 char uuid[40];
2871 char devtype[WPS_DEV_TYPE_BUFSIZE];
2873 d = wps_device_get(reg, addr);
2874 if (d == NULL)
2875 return 0;
2876 if (uuid_bin2str(d->uuid, uuid, sizeof(uuid)))
2877 return 0;
2879 ret = os_snprintf(buf + len, buflen - len,
2880 "wpsUuid=%s\n"
2881 "wpsPrimaryDeviceType=%s\n"
2882 "wpsDeviceName=%s\n"
2883 "wpsManufacturer=%s\n"
2884 "wpsModelName=%s\n"
2885 "wpsModelNumber=%s\n"
2886 "wpsSerialNumber=%s\n",
2887 uuid,
2888 wps_dev_type_bin2str(d->dev.pri_dev_type, devtype,
2889 sizeof(devtype)),
2890 d->dev.device_name ? d->dev.device_name : "",
2891 d->dev.manufacturer ? d->dev.manufacturer : "",
2892 d->dev.model_name ? d->dev.model_name : "",
2893 d->dev.model_number ? d->dev.model_number : "",
2894 d->dev.serial_number ? d->dev.serial_number : "");
2895 if (ret < 0 || (size_t) ret >= buflen - len)
2896 return len;
2897 len += ret;
2899 return len;