1 /*
2  * Wi-Fi Protected Setup - UPnP AP functionality
3  * Copyright (c) 2009, 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 "eloop.h"
13 #include "uuid.h"
14 #include "wps_i.h"
15 #include "wps_upnp.h"
16 #include "wps_upnp_i.h"
17 
18 
upnp_er_set_selected_timeout(void * eloop_ctx,void * timeout_ctx)19 static void upnp_er_set_selected_timeout(void *eloop_ctx, void *timeout_ctx)
20 {
21 	struct subscription *s = eloop_ctx;
22 	struct wps_registrar *reg = timeout_ctx;
23 	wpa_printf(MSG_DEBUG, "WPS: SetSelectedRegistrar from ER timed out");
24 	s->selected_registrar = 0;
25 	wps_registrar_selected_registrar_changed(reg, 0);
26 }
27 
28 
upnp_er_set_selected_registrar(struct wps_registrar * reg,struct subscription * s,const struct wpabuf * msg)29 int upnp_er_set_selected_registrar(struct wps_registrar *reg,
30 				   struct subscription *s,
31 				   const struct wpabuf *msg)
32 {
33 	struct wps_parse_attr attr;
34 
35 	wpa_hexdump_buf(MSG_MSGDUMP, "WPS: SetSelectedRegistrar attributes",
36 			msg);
37 	if (wps_validate_upnp_set_selected_registrar(msg) < 0)
38 		return -1;
39 
40 	if (wps_parse_msg(msg, &attr) < 0)
41 		return -1;
42 
43 	s->reg = reg;
44 	eloop_cancel_timeout(upnp_er_set_selected_timeout, s, reg);
45 
46 	os_memset(s->authorized_macs, 0, sizeof(s->authorized_macs));
47 	if (attr.selected_registrar == NULL || *attr.selected_registrar == 0) {
48 		wpa_printf(MSG_DEBUG, "WPS: SetSelectedRegistrar: Disable "
49 			   "Selected Registrar");
50 		s->selected_registrar = 0;
51 	} else {
52 		s->selected_registrar = 1;
53 		s->dev_password_id = attr.dev_password_id ?
54 			WPA_GET_BE16(attr.dev_password_id) : DEV_PW_DEFAULT;
55 		s->config_methods = attr.sel_reg_config_methods ?
56 			WPA_GET_BE16(attr.sel_reg_config_methods) : -1;
57 		if (attr.authorized_macs) {
58 			int count = attr.authorized_macs_len / ETH_ALEN;
59 			if (count > WPS_MAX_AUTHORIZED_MACS)
60 				count = WPS_MAX_AUTHORIZED_MACS;
61 			os_memcpy(s->authorized_macs, attr.authorized_macs,
62 				  count * ETH_ALEN);
63 		} else if (!attr.version2) {
64 			wpa_printf(MSG_DEBUG, "WPS: Add broadcast "
65 				   "AuthorizedMACs for WPS 1.0 ER");
66 			os_memset(s->authorized_macs, 0xff, ETH_ALEN);
67 		}
68 		eloop_register_timeout(WPS_PBC_WALK_TIME, 0,
69 				       upnp_er_set_selected_timeout, s, reg);
70 	}
71 
72 	wps_registrar_selected_registrar_changed(reg, 0);
73 
74 	return 0;
75 }
76 
77 
upnp_er_remove_notification(struct wps_registrar * reg,struct subscription * s)78 void upnp_er_remove_notification(struct wps_registrar *reg,
79 				 struct subscription *s)
80 {
81 	s->selected_registrar = 0;
82 	eloop_cancel_timeout(upnp_er_set_selected_timeout, s, reg);
83 	if (reg)
84 		wps_registrar_selected_registrar_changed(reg, 0);
85 }
86