18d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt/*
28d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Wi-Fi Protected Setup - UPnP AP functionality
38d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
48d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt *
5c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * This software may be distributed under the terms of the BSD license.
6c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * See README for more details.
78d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */
88d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
98d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "includes.h"
108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "common.h"
128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "eloop.h"
138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "uuid.h"
148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "wps_i.h"
158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "wps_upnp.h"
168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "wps_upnp_i.h"
178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic void upnp_er_set_selected_timeout(void *eloop_ctx, void *timeout_ctx)
208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct subscription *s = eloop_ctx;
22c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	struct wps_registrar *reg = timeout_ctx;
238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_printf(MSG_DEBUG, "WPS: SetSelectedRegistrar from ER timed out");
248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	s->selected_registrar = 0;
254b06059785b935dd1f4f09314e4e12c417d2c6a4Dmitry Shmidt	wps_registrar_selected_registrar_changed(reg, 0);
268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtint upnp_er_set_selected_registrar(struct wps_registrar *reg,
308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   struct subscription *s,
318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   const struct wpabuf *msg)
328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	struct wps_parse_attr attr;
348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	wpa_hexdump_buf(MSG_MSGDUMP, "WPS: SetSelectedRegistrar attributes",
368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			msg);
378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_validate_upnp_set_selected_registrar(msg) < 0)
388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (wps_parse_msg(msg, &attr) < 0)
418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		return -1;
428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	s->reg = reg;
44c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	eloop_cancel_timeout(upnp_er_set_selected_timeout, s, reg);
458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	os_memset(s->authorized_macs, 0, sizeof(s->authorized_macs));
478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	if (attr.selected_registrar == NULL || *attr.selected_registrar == 0) {
488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		wpa_printf(MSG_DEBUG, "WPS: SetSelectedRegistrar: Disable "
498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			   "Selected Registrar");
508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		s->selected_registrar = 0;
518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	} else {
528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		s->selected_registrar = 1;
538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		s->dev_password_id = attr.dev_password_id ?
548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			WPA_GET_BE16(attr.dev_password_id) : DEV_PW_DEFAULT;
558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		s->config_methods = attr.sel_reg_config_methods ?
568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			WPA_GET_BE16(attr.sel_reg_config_methods) : -1;
578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		if (attr.authorized_macs) {
588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			int count = attr.authorized_macs_len / ETH_ALEN;
598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			if (count > WPS_MAX_AUTHORIZED_MACS)
608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				count = WPS_MAX_AUTHORIZED_MACS;
618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			os_memcpy(s->authorized_macs, attr.authorized_macs,
628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				  count * ETH_ALEN);
638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		} else if (!attr.version2) {
648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			wpa_printf(MSG_DEBUG, "WPS: Add broadcast "
658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt				   "AuthorizedMACs for WPS 1.0 ER");
668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt			os_memset(s->authorized_macs, 0xff, ETH_ALEN);
678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		}
688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt		eloop_register_timeout(WPS_PBC_WALK_TIME, 0,
69c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt				       upnp_er_set_selected_timeout, s, reg);
708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	}
718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
724b06059785b935dd1f4f09314e4e12c417d2c6a4Dmitry Shmidt	wps_registrar_selected_registrar_changed(reg, 0);
738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	return 0;
758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt
78c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidtvoid upnp_er_remove_notification(struct wps_registrar *reg,
79c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt				 struct subscription *s)
808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{
818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt	s->selected_registrar = 0;
82c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	eloop_cancel_timeout(upnp_er_set_selected_timeout, s, reg);
83c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt	if (reg)
844b06059785b935dd1f4f09314e4e12c417d2c6a4Dmitry Shmidt		wps_registrar_selected_registrar_changed(reg, 0);
858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt}
86