d2e473135a77de614f3eb2c4feae91af1ac8ad67
[libeap.git] / src / wps / wps_upnp.c
1 /*
2  * UPnP WPS Device
3  * Copyright (c) 2000-2003 Intel Corporation
4  * Copyright (c) 2006-2007 Sony Corporation
5  * Copyright (c) 2008-2009 Atheros Communications
6  * Copyright (c) 2009, Jouni Malinen <j@w1.fi>
7  *
8  * See below for more details on licensing and code history.
9  */
10
11 /*
12  * This has been greatly stripped down from the original file
13  * (upnp_wps_device.c) by Ted Merrill, Atheros Communications
14  * in order to eliminate use of the bulky libupnp library etc.
15  *
16  * History:
17  * upnp_wps_device.c is/was a shim layer between wps_opt_upnp.c and
18  * the libupnp library.
19  * The layering (by Sony) was well done; only a very minor modification
20  * to API of upnp_wps_device.c was required.
21  * libupnp was found to be undesirable because:
22  * -- It consumed too much code and data space
23  * -- It uses multiple threads, making debugging more difficult
24  *      and possibly reducing reliability.
25  * -- It uses static variables and only supports one instance.
26  * The shim and libupnp are here replaced by special code written
27  * specifically for the needs of hostapd.
28  * Various shortcuts can and are taken to keep the code size small.
29  * Generally, execution time is not as crucial.
30  *
31  * BUGS:
32  * -- UPnP requires that we be able to resolve domain names.
33  * While uncommon, if we have to do it then it will stall the entire
34  * hostapd program, which is bad.
35  * This is because we use the standard linux getaddrinfo() function
36  * which is syncronous.
37  * An asyncronous solution would be to use the free "ares" library.
38  * -- Does not have a robust output buffering scheme.  Uses a single
39  * fixed size output buffer per TCP/HTTP connection, with possible (although
40  * unlikely) possibility of overflow and likely excessive use of RAM.
41  * A better solution would be to write the HTTP output as a buffered stream,
42  * using chunking: (handle header specially, then) generate data with
43  * a printf-like function into a buffer, catching buffer full condition,
44  * then send it out surrounded by http chunking.
45  * -- There is some code that could be separated out into the common
46  * library to be shared with wpa_supplicant.
47  * -- Needs renaming with module prefix to avoid polluting the debugger
48  * namespace and causing possible collisions with other static fncs
49  * and structure declarations when using the debugger.
50  * -- The http error code generation is pretty bogus, hopefully noone cares.
51  *
52  * Author: Ted Merrill, Atheros Communications, based upon earlier work
53  * as explained above and below.
54  *
55  * Copyright:
56  * Copyright 2008 Atheros Communications.
57  *
58  * The original header (of upnp_wps_device.c) reads:
59  *
60  *  Copyright (c) 2006-2007 Sony Corporation. All Rights Reserved.
61  *
62  *  File Name: upnp_wps_device.c
63  *  Description: EAP-WPS UPnP device source
64  *
65  *   Redistribution and use in source and binary forms, with or without
66  *   modification, are permitted provided that the following conditions
67  *   are met:
68  *
69  *     * Redistributions of source code must retain the above copyright
70  *       notice, this list of conditions and the following disclaimer.
71  *     * Redistributions in binary form must reproduce the above copyright
72  *       notice, this list of conditions and the following disclaimer in
73  *       the documentation and/or other materials provided with the
74  *       distribution.
75  *     * Neither the name of Sony Corporation nor the names of its
76  *       contributors may be used to endorse or promote products derived
77  *       from this software without specific prior written permission.
78  *
79  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
80  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
81  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
82  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
83  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
84  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
85  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
86  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
87  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
88  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
89  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
90  *
91  * Portions from Intel libupnp files, e.g. genlib/net/http/httpreadwrite.c
92  * typical header:
93  *
94  * Copyright (c) 2000-2003 Intel Corporation
95  * All rights reserved.
96  *
97  * Redistribution and use in source and binary forms, with or without
98  * modification, are permitted provided that the following conditions are met:
99  *
100  * * Redistributions of source code must retain the above copyright notice,
101  * this list of conditions and the following disclaimer.
102  * * Redistributions in binary form must reproduce the above copyright notice,
103  * this list of conditions and the following disclaimer in the documentation
104  * and/or other materials provided with the distribution.
105  * * Neither name of Intel Corporation nor the names of its contributors
106  * may be used to endorse or promote products derived from this software
107  * without specific prior written permission.
108  *
109  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
110  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
111  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
112  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL INTEL OR
113  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
114  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
115  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
116  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
117  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
118  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
119  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
120 */
121
122 /*
123  * Overview of WPS over UPnP:
124  *
125  * UPnP is a protocol that allows devices to discover each other and control
126  * each other. In UPnP terminology, a device is either a "device" (a server
127  * that provides information about itself and allows itself to be controlled)
128  * or a "control point" (a client that controls "devices") or possibly both.
129  * This file implements a UPnP "device".
130  *
131  * For us, we use mostly basic UPnP discovery, but the control part of interest
132  * is WPS carried via UPnP messages. There is quite a bit of basic UPnP
133  * discovery to do before we can get to WPS, however.
134  *
135  * UPnP discovery begins with "devices" send out multicast UDP packets to a
136  * certain fixed multicast IP address and port, and "control points" sending
137  * out other such UDP packets.
138  *
139  * The packets sent by devices are NOTIFY packets (not to be confused with TCP
140  * NOTIFY packets that are used later) and those sent by control points are
141  * M-SEARCH packets. These packets contain a simple HTTP style header. The
142  * packets are sent redundantly to get around packet loss. Devices respond to
143  * M-SEARCH packets with HTTP-like UDP packets containing HTTP/1.1 200 OK
144  * messages, which give similar information as the UDP NOTIFY packets.
145  *
146  * The above UDP packets advertise the (arbitrary) TCP ports that the
147  * respective parties will listen to. The control point can then do a HTTP
148  * SUBSCRIBE (something like an HTTP PUT) after which the device can do a
149  * separate HTTP NOTIFY (also like an HTTP PUT) to do event messaging.
150  *
151  * The control point will also do HTTP GET of the "device file" listed in the
152  * original UDP information from the device (see UPNP_WPS_DEVICE_XML_FILE
153  * data), and based on this will do additional GETs... HTTP POSTs are done to
154  * cause an action.
155  *
156  * Beyond some basic information in HTTP headers, additional information is in
157  * the HTTP bodies, in a format set by the SOAP and XML standards, a markup
158  * language related to HTML used for web pages. This language is intended to
159  * provide the ultimate in self-documentation by providing a universal
160  * namespace based on pseudo-URLs called URIs. Note that although a URI looks
161  * like a URL (a web address), they are never accessed as such but are used
162  * only as identifiers.
163  *
164  * The POST of a GetDeviceInfo gets information similar to what might be
165  * obtained from a probe request or response on Wi-Fi. WPS messages M1-M8
166  * are passed via a POST of a PutMessage; the M1-M8 WPS messages are converted
167  * to a bin64 ascii representation for encapsulation. When proxying messages,
168  * WLANEvent and PutWLANResponse are used.
169  *
170  * This of course glosses over a lot of details.
171  */
172
173 #include "includes.h"
174
175 #include <assert.h>
176 #include <net/if.h>
177 #include <netdb.h>
178 #include <sys/ioctl.h>
179
180 #include "common.h"
181 #include "uuid.h"
182 #include "base64.h"
183 #include "wps.h"
184 #include "wps_i.h"
185 #include "wps_upnp.h"
186 #include "wps_upnp_i.h"
187
188
189 /*
190  * UPnP allows a client ("control point") to send a server like us ("device")
191  * a domain name for registration, and we are supposed to resolve it. This is
192  * bad because, using the standard Linux library, we will stall the entire
193  * hostapd waiting for resolution.
194  *
195  * The "correct" solution would be to use an event driven library for domain
196  * name resolution such as "ares". However, this would increase code size
197  * further. Since it is unlikely that we'll actually see such domain names, we
198  * can just refuse to accept them.
199  */
200 #define NO_DOMAIN_NAME_RESOLUTION 1  /* 1 to allow only dotted ip addresses */
201
202
203 /*
204  * UPnP does not scale well. If we were in a room with thousands of control
205  * points then potentially we could be expected to handle subscriptions for
206  * each of them, which would exhaust our memory. So we must set a limit. In
207  * practice we are unlikely to see more than one or two.
208  */
209 #define MAX_SUBSCRIPTIONS 4    /* how many subscribing clients we handle */
210 #define MAX_ADDR_PER_SUBSCRIPTION 8
211
212
213 /* Write the current date/time per RFC */
214 void format_date(struct wpabuf *buf)
215 {
216         const char *weekday_str = "Sun\0Mon\0Tue\0Wed\0Thu\0Fri\0Sat";
217         const char *month_str = "Jan\0Feb\0Mar\0Apr\0May\0Jun\0"
218                 "Jul\0Aug\0Sep\0Oct\0Nov\0Dec";
219         struct tm *date;
220         time_t t;
221
222         t = time(NULL);
223         date = gmtime(&t);
224         wpabuf_printf(buf, "%s, %02d %s %d %02d:%02d:%02d GMT",
225                       &weekday_str[date->tm_wday * 4], date->tm_mday,
226                       &month_str[date->tm_mon * 4], date->tm_year + 1900,
227                       date->tm_hour, date->tm_min, date->tm_sec);
228 }
229
230
231 /***************************************************************************
232  * UUIDs (unique identifiers)
233  *
234  * These are supposed to be unique in all the world.
235  * Sometimes permanent ones are used, sometimes temporary ones
236  * based on random numbers... there are different rules for valid content
237  * of different types.
238  * Each uuid is 16 bytes long.
239  **************************************************************************/
240
241 /* uuid_make -- construct a random UUID
242  * The UPnP documents don't seem to offer any guidelines as to which method to
243  * use for constructing UUIDs for subscriptions. Presumably any method from
244  * rfc4122 is good enough; I've chosen random number method.
245  */
246 static void uuid_make(u8 uuid[UUID_LEN])
247 {
248         os_get_random(uuid, UUID_LEN);
249
250         /* Replace certain bits as specified in rfc4122 or X.667 */
251         uuid[6] &= 0x0f; uuid[6] |= (4 << 4);   /* version 4 == random gen */
252         uuid[8] &= 0x3f; uuid[8] |= 0x80;
253 }
254
255
256 /*
257  * Subscriber address handling.
258  * Since a subscriber may have an arbitrary number of addresses, we have to
259  * add a bunch of code to handle them.
260  *
261  * Addresses are passed in text, and MAY be domain names instead of the (usual
262  * and expected) dotted IP addresses. Resolving domain names consumes a lot of
263  * resources. Worse, we are currently using the standard Linux getaddrinfo()
264  * which will block the entire program until complete or timeout! The proper
265  * solution would be to use the "ares" library or similar with more state
266  * machine steps etc. or just disable domain name resolution by setting
267  * NO_DOMAIN_NAME_RESOLUTION to 1 at top of this file.
268  */
269
270 /* subscr_addr_delete -- delete single unlinked subscriber address
271  * (be sure to unlink first if need be)
272  */
273 static void subscr_addr_delete(struct subscr_addr *a)
274 {
275         /*
276          * Note: do NOT free domain_and_port or path because they point to
277          * memory within the allocation of "a".
278          */
279         os_free(a);
280 }
281
282
283 /* subscr_addr_free_all -- unlink and delete list of subscriber addresses. */
284 static void subscr_addr_free_all(struct subscription *s)
285 {
286         struct subscr_addr *a, *tmp;
287         dl_list_for_each_safe(a, tmp, &s->addr_list, struct subscr_addr, list)
288         {
289                 dl_list_del(&a->list);
290                 subscr_addr_delete(a);
291         }
292 }
293
294
295 /* subscr_addr_add_url -- add address(es) for one url to subscription */
296 static void subscr_addr_add_url(struct subscription *s, const char *url)
297 {
298         int alloc_len;
299         char *scratch_mem = NULL;
300         char *mem;
301         char *domain_and_port;
302         char *delim;
303         char *path;
304         char *domain;
305         int port = 80;  /* port to send to (default is port 80) */
306         struct addrinfo hints;
307         struct addrinfo *result = NULL;
308         struct addrinfo *rp;
309         int rerr;
310         struct subscr_addr *a = NULL;
311
312         /* url MUST begin with http: */
313         if (os_strncasecmp(url, "http://", 7))
314                 goto fail;
315         url += 7;
316
317         /* allocate memory for the extra stuff we need */
318         alloc_len = (2 * (os_strlen(url) + 1));
319         scratch_mem = os_zalloc(alloc_len);
320         if (scratch_mem == NULL)
321                 goto fail;
322         mem = scratch_mem;
323         strcpy(mem, url);
324         domain_and_port = mem;
325         mem += 1 + os_strlen(mem);
326         delim = os_strchr(domain_and_port, '/');
327         if (delim) {
328                 *delim++ = 0;   /* null terminate domain and port */
329                 path = delim;
330         } else {
331                 path = domain_and_port + os_strlen(domain_and_port);
332         }
333         domain = mem;
334         strcpy(domain, domain_and_port);
335         delim = strchr(domain, ':');
336         if (delim) {
337                 *delim++ = 0;   /* null terminate domain */
338                 if (isdigit(*delim))
339                         port = atol(delim);
340         }
341
342         /*
343          * getaddrinfo does the right thing with dotted decimal notations, or
344          * will resolve domain names. Resolving domain names will unfortunately
345          * hang the entire program until it is resolved or it times out
346          * internal to getaddrinfo; fortunately we think that the use of actual
347          * domain names (vs. dotted decimal notations) should be uncommon.
348          */
349         os_memset(&hints, 0, sizeof(struct addrinfo));
350         hints.ai_family = AF_INET;      /* IPv4 */
351         hints.ai_socktype = SOCK_STREAM;
352 #if NO_DOMAIN_NAME_RESOLUTION
353         /* Suppress domain name resolutions that would halt
354          * the program for periods of time
355          */
356         hints.ai_flags = AI_NUMERICHOST;
357 #else
358         /* Allow domain name resolution. */
359         hints.ai_flags = 0;
360 #endif
361         hints.ai_protocol = 0;          /* Any protocol? */
362         rerr = getaddrinfo(domain, NULL /* fill in port ourselves */,
363                            &hints, &result);
364         if (rerr) {
365                 wpa_printf(MSG_INFO, "WPS UPnP: Resolve error %d (%s) on: %s",
366                            rerr, gai_strerror(rerr), domain);
367                 goto fail;
368         }
369         for (rp = result; rp; rp = rp->ai_next) {
370                 /* Limit no. of address to avoid denial of service attack */
371                 if (dl_list_len(&s->addr_list) >= MAX_ADDR_PER_SUBSCRIPTION) {
372                         wpa_printf(MSG_INFO, "WPS UPnP: subscr_addr_add_url: "
373                                    "Ignoring excessive addresses");
374                         break;
375                 }
376
377                 a = os_zalloc(sizeof(*a) + alloc_len);
378                 if (a == NULL)
379                         continue;
380                 mem = (void *) (a + 1);
381                 a->domain_and_port = mem;
382                 strcpy(mem, domain_and_port);
383                 mem += 1 + strlen(mem);
384                 a->path = mem;
385                 if (path[0] != '/')
386                         *mem++ = '/';
387                 strcpy(mem, path);
388                 mem += 1 + strlen(mem);
389                 os_memcpy(&a->saddr, rp->ai_addr, sizeof(a->saddr));
390                 a->saddr.sin_port = htons(port);
391
392                 dl_list_add(&s->addr_list, &a->list);
393                 a = NULL;       /* don't free it below */
394         }
395
396 fail:
397         if (result)
398                 freeaddrinfo(result);
399         os_free(scratch_mem);
400         os_free(a);
401 }
402
403
404 /* subscr_addr_list_create -- create list from urls in string.
405  *      Each url is enclosed by angle brackets.
406  */
407 static void subscr_addr_list_create(struct subscription *s,
408                                     const char *url_list)
409 {
410         char *end;
411         for (;;) {
412                 while (*url_list == ' ' || *url_list == '\t')
413                         url_list++;
414                 if (*url_list != '<')
415                         break;
416                 url_list++;
417                 end = os_strchr(url_list, '>');
418                 if (end == NULL)
419                         break;
420                 *end++ = 0;
421                 subscr_addr_add_url(s, url_list);
422                 url_list = end;
423         }
424 }
425
426
427 int send_wpabuf(int fd, struct wpabuf *buf)
428 {
429         wpa_printf(MSG_DEBUG, "WPS UPnP: Send %lu byte message",
430                    (unsigned long) wpabuf_len(buf));
431         errno = 0;
432         if (write(fd, wpabuf_head(buf), wpabuf_len(buf)) !=
433             (int) wpabuf_len(buf)) {
434                 wpa_printf(MSG_ERROR, "WPS UPnP: Failed to send buffer: "
435                            "errno=%d (%s)",
436                            errno, strerror(errno));
437                 return -1;
438         }
439
440         return 0;
441 }
442
443
444 static void wpabuf_put_property(struct wpabuf *buf, const char *name,
445                                 const char *value)
446 {
447         wpabuf_put_str(buf, "<e:property>");
448         wpabuf_printf(buf, "<%s>", name);
449         if (value)
450                 wpabuf_put_str(buf, value);
451         wpabuf_printf(buf, "</%s>", name);
452         wpabuf_put_str(buf, "</e:property>\n");
453 }
454
455
456 /**
457  * upnp_wps_device_send_event - Queue event messages for subscribers
458  * @sm: WPS UPnP state machine from upnp_wps_device_init()
459  *
460  * This function queues the last WLANEvent to be sent for all currently
461  * subscribed UPnP control points. sm->wlanevent must have been set with the
462  * encoded data before calling this function.
463  */
464 static void upnp_wps_device_send_event(struct upnp_wps_device_sm *sm)
465 {
466         /* Enqueue event message for all subscribers */
467         struct wpabuf *buf; /* holds event message */
468         int buf_size = 0;
469         struct subscription *s, *tmp;
470         /* Actually, utf-8 is the default, but it doesn't hurt to specify it */
471         const char *format_head =
472                 "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
473                 "<e:propertyset xmlns:e=\"urn:schemas-upnp-org:event-1-0\">\n";
474         const char *format_tail = "</e:propertyset>\n";
475
476         if (dl_list_empty(&sm->subscriptions)) {
477                 /* optimize */
478                 return;
479         }
480
481         /* Determine buffer size needed first */
482         buf_size += os_strlen(format_head);
483         buf_size += 50 + 2 * os_strlen("WLANEvent");
484         if (sm->wlanevent)
485                 buf_size += os_strlen(sm->wlanevent);
486         buf_size += os_strlen(format_tail);
487
488         buf = wpabuf_alloc(buf_size);
489         if (buf == NULL)
490                 return;
491         wpabuf_put_str(buf, format_head);
492         wpabuf_put_property(buf, "WLANEvent", sm->wlanevent);
493         wpabuf_put_str(buf, format_tail);
494
495         wpa_printf(MSG_MSGDUMP, "WPS UPnP: WLANEvent message:\n%s",
496                    (char *) wpabuf_head(buf));
497
498         dl_list_for_each_safe(s, tmp, &sm->subscriptions, struct subscription,
499                               list) {
500                 if (event_add(s, buf)) {
501                         wpa_printf(MSG_INFO, "WPS UPnP: Dropping "
502                                    "subscriber due to event backlog");
503                         dl_list_del(&s->list);
504                         subscription_destroy(s);
505                 }
506         }
507
508         wpabuf_free(buf);
509 }
510
511
512 /*
513  * Event subscription (subscriber machines register with us to receive event
514  * messages).
515  * This is the result of an incoming HTTP over TCP SUBSCRIBE request.
516  */
517
518 /* subscription_destroy -- destroy an unlinked subscription
519  * Be sure to unlink first if necessary.
520  */
521 void subscription_destroy(struct subscription *s)
522 {
523         wpa_printf(MSG_DEBUG, "WPS UPnP: Destroy subscription %p", s);
524         subscr_addr_free_all(s);
525         event_delete_all(s);
526         upnp_er_remove_notification(s);
527         os_free(s);
528 }
529
530
531 /* subscription_list_age -- remove expired subscriptions */
532 static void subscription_list_age(struct upnp_wps_device_sm *sm, time_t now)
533 {
534         struct subscription *s, *tmp;
535         dl_list_for_each_safe(s, tmp, &sm->subscriptions,
536                               struct subscription, list) {
537                 if (s->timeout_time > now)
538                         break;
539                 wpa_printf(MSG_DEBUG, "WPS UPnP: Removing aged subscription");
540                 dl_list_del(&s->list);
541                 subscription_destroy(s);
542         }
543 }
544
545
546 /* subscription_find -- return existing subscription matching uuid, if any
547  * returns NULL if not found
548  */
549 struct subscription * subscription_find(struct upnp_wps_device_sm *sm,
550                                         const u8 uuid[UUID_LEN])
551 {
552         struct subscription *s;
553         dl_list_for_each(s, &sm->subscriptions, struct subscription, list) {
554                 if (os_memcmp(s->uuid, uuid, UUID_LEN) == 0)
555                         return s; /* Found match */
556         }
557         return NULL;
558 }
559
560
561 static struct wpabuf * build_fake_wsc_ack(void)
562 {
563         struct wpabuf *msg = wpabuf_alloc(100);
564         if (msg == NULL)
565                 return NULL;
566         wpabuf_put_u8(msg, UPNP_WPS_WLANEVENT_TYPE_EAP);
567         wpabuf_put_str(msg, "00:00:00:00:00:00");
568         if (wps_build_version(msg) ||
569             wps_build_msg_type(msg, WPS_WSC_ACK)) {
570                 wpabuf_free(msg);
571                 return NULL;
572         }
573         /* Enrollee Nonce */
574         wpabuf_put_be16(msg, ATTR_ENROLLEE_NONCE);
575         wpabuf_put_be16(msg, WPS_NONCE_LEN);
576         wpabuf_put(msg, WPS_NONCE_LEN);
577         /* Registrar Nonce */
578         wpabuf_put_be16(msg, ATTR_REGISTRAR_NONCE);
579         wpabuf_put_be16(msg, WPS_NONCE_LEN);
580         wpabuf_put(msg, WPS_NONCE_LEN);
581         wps_build_wfa_ext(msg, 0, NULL, 0);
582         return msg;
583 }
584
585
586 /* subscription_first_event -- send format/queue event that is automatically
587  * sent on a new subscription.
588  */
589 static int subscription_first_event(struct subscription *s)
590 {
591         /*
592          * Actually, utf-8 is the default, but it doesn't hurt to specify it.
593          *
594          * APStatus is apparently a bit set,
595          * 0x1 = configuration change (but is always set?)
596          * 0x10 = ap is locked
597          *
598          * Per UPnP spec, we send out the last value of each variable, even
599          * for WLANEvent, whatever it was.
600          */
601         char *wlan_event;
602         struct wpabuf *buf;
603         int ap_status = 1;      /* TODO: add 0x10 if access point is locked */
604         const char *head =
605                 "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
606                 "<e:propertyset xmlns:e=\"urn:schemas-upnp-org:event-1-0\">\n";
607         const char *tail = "</e:propertyset>\n";
608         char txt[10];
609
610         if (s->sm->wlanevent == NULL) {
611                 /*
612                  * There has been no events before the subscription. However,
613                  * UPnP device architecture specification requires all the
614                  * evented variables to be included, so generate a dummy event
615                  * for this particular case using a WSC_ACK and all-zeros
616                  * nonces. The ER (UPnP control point) will ignore this, but at
617                  * least it will learn that WLANEvent variable will be used in
618                  * event notifications in the future.
619                  */
620                 struct wpabuf *msg;
621                 wpa_printf(MSG_DEBUG, "WPS UPnP: Use a fake WSC_ACK as the "
622                            "initial WLANEvent");
623                 msg = build_fake_wsc_ack();
624                 if (msg) {
625                         s->sm->wlanevent = (char *)
626                                 base64_encode(wpabuf_head(msg),
627                                               wpabuf_len(msg), NULL);
628                         wpabuf_free(msg);
629                 }
630         }
631
632         wlan_event = s->sm->wlanevent;
633         if (wlan_event == NULL || *wlan_event == '\0') {
634                 wpa_printf(MSG_DEBUG, "WPS UPnP: WLANEvent not known for "
635                            "initial event message");
636                 wlan_event = "";
637         }
638         buf = wpabuf_alloc(500 + os_strlen(wlan_event));
639         if (buf == NULL)
640                 return 1;
641
642         wpabuf_put_str(buf, head);
643         wpabuf_put_property(buf, "STAStatus", "1");
644         os_snprintf(txt, sizeof(txt), "%d", ap_status);
645         wpabuf_put_property(buf, "APStatus", txt);
646         if (*wlan_event)
647                 wpabuf_put_property(buf, "WLANEvent", wlan_event);
648         wpabuf_put_str(buf, tail);
649
650         if (event_add(s, buf)) {
651                 wpabuf_free(buf);
652                 return 1;
653         }
654         wpabuf_free(buf);
655
656         return 0;
657 }
658
659
660 /**
661  * subscription_start - Remember a UPnP control point to send events to.
662  * @sm: WPS UPnP state machine from upnp_wps_device_init()
663  * @callback_urls: Callback URLs
664  * Returns: %NULL on error, or pointer to new subscription structure.
665  */
666 struct subscription * subscription_start(struct upnp_wps_device_sm *sm,
667                                          const char *callback_urls)
668 {
669         struct subscription *s;
670         time_t now = time(NULL);
671         time_t expire = now + UPNP_SUBSCRIBE_SEC;
672
673         /* Get rid of expired subscriptions so we have room */
674         subscription_list_age(sm, now);
675
676         /* If too many subscriptions, remove oldest */
677         if (dl_list_len(&sm->subscriptions) >= MAX_SUBSCRIPTIONS) {
678                 s = dl_list_first(&sm->subscriptions, struct subscription,
679                                   list);
680                 wpa_printf(MSG_INFO, "WPS UPnP: Too many subscriptions, "
681                            "trashing oldest");
682                 dl_list_del(&s->list);
683                 subscription_destroy(s);
684         }
685
686         s = os_zalloc(sizeof(*s));
687         if (s == NULL)
688                 return NULL;
689         dl_list_init(&s->addr_list);
690         dl_list_init(&s->event_queue);
691
692         s->sm = sm;
693         s->timeout_time = expire;
694         uuid_make(s->uuid);
695         subscr_addr_list_create(s, callback_urls);
696         /* Add to end of list, since it has the highest expiration time */
697         dl_list_add_tail(&sm->subscriptions, &s->list);
698         /* Queue up immediate event message (our last event)
699          * as required by UPnP spec.
700          */
701         if (subscription_first_event(s)) {
702                 wpa_printf(MSG_INFO, "WPS UPnP: Dropping subscriber due to "
703                            "event backlog");
704                 dl_list_del(&s->list);
705                 subscription_destroy(s);
706                 return NULL;
707         }
708         wpa_printf(MSG_DEBUG, "WPS UPnP: Subscription %p started with %s",
709                    s, callback_urls);
710         /* Schedule sending this */
711         event_send_all_later(sm);
712         return s;
713 }
714
715
716 /* subscription_renew -- find subscription and reset timeout */
717 struct subscription * subscription_renew(struct upnp_wps_device_sm *sm,
718                                          const u8 uuid[UUID_LEN])
719 {
720         time_t now = time(NULL);
721         time_t expire = now + UPNP_SUBSCRIBE_SEC;
722         struct subscription *s = subscription_find(sm, uuid);
723         if (s == NULL)
724                 return NULL;
725         wpa_printf(MSG_DEBUG, "WPS UPnP: Subscription renewed");
726         dl_list_del(&s->list);
727         s->timeout_time = expire;
728         /* add back to end of list, since it now has highest expiry */
729         dl_list_add_tail(&sm->subscriptions, &s->list);
730         return s;
731 }
732
733
734 /**
735  * upnp_wps_device_send_wlan_event - Event notification
736  * @sm: WPS UPnP state machine from upnp_wps_device_init()
737  * @from_mac_addr: Source (Enrollee) MAC address for the event
738  * @ev_type: Event type
739  * @msg: Event data
740  * Returns: 0 on success, -1 on failure
741  *
742  * Tell external Registrars (UPnP control points) that something happened. In
743  * particular, events include WPS messages from clients that are proxied to
744  * external Registrars.
745  */
746 int upnp_wps_device_send_wlan_event(struct upnp_wps_device_sm *sm,
747                                     const u8 from_mac_addr[ETH_ALEN],
748                                     enum upnp_wps_wlanevent_type ev_type,
749                                     const struct wpabuf *msg)
750 {
751         int ret = -1;
752         char type[2];
753         const u8 *mac = from_mac_addr;
754         char mac_text[18];
755         u8 *raw = NULL;
756         size_t raw_len;
757         char *val;
758         size_t val_len;
759         int pos = 0;
760
761         if (!sm)
762                 goto fail;
763
764         os_snprintf(type, sizeof(type), "%1u", ev_type);
765
766         raw_len = 1 + 17 + (msg ? wpabuf_len(msg) : 0);
767         raw = os_zalloc(raw_len);
768         if (!raw)
769                 goto fail;
770
771         *(raw + pos) = (u8) ev_type;
772         pos += 1;
773         os_snprintf(mac_text, sizeof(mac_text), MACSTR, MAC2STR(mac));
774         wpa_printf(MSG_DEBUG, "WPS UPnP: Proxying WLANEvent from %s",
775                    mac_text);
776         os_memcpy(raw + pos, mac_text, 17);
777         pos += 17;
778         if (msg) {
779                 os_memcpy(raw + pos, wpabuf_head(msg), wpabuf_len(msg));
780                 pos += wpabuf_len(msg);
781         }
782         raw_len = pos;
783
784         val = (char *) base64_encode(raw, raw_len, &val_len);
785         if (val == NULL)
786                 goto fail;
787
788         os_free(sm->wlanevent);
789         sm->wlanevent = val;
790         upnp_wps_device_send_event(sm);
791
792         ret = 0;
793
794 fail:
795         os_free(raw);
796
797         return ret;
798 }
799
800
801 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
802 #include <sys/sysctl.h>
803 #include <net/route.h>
804 #include <net/if_dl.h>
805
806 static int eth_get(const char *device, u8 ea[ETH_ALEN])
807 {
808         struct if_msghdr *ifm;
809         struct sockaddr_dl *sdl;
810         u_char *p, *buf;
811         size_t len;
812         int mib[] = { CTL_NET, AF_ROUTE, 0, AF_LINK, NET_RT_IFLIST, 0 };
813
814         if (sysctl(mib, 6, NULL, &len, NULL, 0) < 0)
815                 return -1;
816         if ((buf = os_malloc(len)) == NULL)
817                 return -1;
818         if (sysctl(mib, 6, buf, &len, NULL, 0) < 0) {
819                 os_free(buf);
820                 return -1;
821         }
822         for (p = buf; p < buf + len; p += ifm->ifm_msglen) {
823                 ifm = (struct if_msghdr *)p;
824                 sdl = (struct sockaddr_dl *)(ifm + 1);
825                 if (ifm->ifm_type != RTM_IFINFO ||
826                     (ifm->ifm_addrs & RTA_IFP) == 0)
827                         continue;
828                 if (sdl->sdl_family != AF_LINK || sdl->sdl_nlen == 0 ||
829                     os_memcmp(sdl->sdl_data, device, sdl->sdl_nlen) != 0)
830                         continue;
831                 os_memcpy(ea, LLADDR(sdl), sdl->sdl_alen);
832                 break;
833         }
834         os_free(buf);
835
836         if (p >= buf + len) {
837                 errno = ESRCH;
838                 return -1;
839         }
840         return 0;
841 }
842 #endif /* __FreeBSD__ */
843
844
845 /**
846  * get_netif_info - Get hw and IP addresses for network device
847  * @net_if: Selected network interface name
848  * @ip_addr: Buffer for returning IP address in network byte order
849  * @ip_addr_text: Buffer for returning a pointer to allocated IP address text
850  * @mac: Buffer for returning MAC address
851  * Returns: 0 on success, -1 on failure
852  */
853 int get_netif_info(const char *net_if, unsigned *ip_addr, char **ip_addr_text,
854                    u8 mac[ETH_ALEN])
855 {
856         struct ifreq req;
857         int sock = -1;
858         struct sockaddr_in *addr;
859         struct in_addr in_addr;
860
861         *ip_addr_text = os_zalloc(16);
862         if (*ip_addr_text == NULL)
863                 goto fail;
864
865         sock = socket(AF_INET, SOCK_DGRAM, 0);
866         if (sock < 0)
867                 goto fail;
868
869         os_strlcpy(req.ifr_name, net_if, sizeof(req.ifr_name));
870         if (ioctl(sock, SIOCGIFADDR, &req) < 0) {
871                 wpa_printf(MSG_ERROR, "WPS UPnP: SIOCGIFADDR failed: %d (%s)",
872                            errno, strerror(errno));
873                 goto fail;
874         }
875         addr = (void *) &req.ifr_addr;
876         *ip_addr = addr->sin_addr.s_addr;
877         in_addr.s_addr = *ip_addr;
878         os_snprintf(*ip_addr_text, 16, "%s", inet_ntoa(in_addr));
879
880 #ifdef __linux__
881         os_strlcpy(req.ifr_name, net_if, sizeof(req.ifr_name));
882         if (ioctl(sock, SIOCGIFHWADDR, &req) < 0) {
883                 wpa_printf(MSG_ERROR, "WPS UPnP: SIOCGIFHWADDR failed: "
884                            "%d (%s)", errno, strerror(errno));
885                 goto fail;
886         }
887         os_memcpy(mac, req.ifr_addr.sa_data, 6);
888 #elif defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
889         if (eth_get(net_if, mac) < 0) {
890                 wpa_printf(MSG_ERROR, "WPS UPnP: Failed to get MAC address");
891                 goto fail;
892         }
893 #else
894 #error MAC address fetch not implemented
895 #endif
896
897         close(sock);
898         return 0;
899
900 fail:
901         if (sock >= 0)
902                 close(sock);
903         os_free(*ip_addr_text);
904         *ip_addr_text = NULL;
905         return -1;
906 }
907
908
909 static void upnp_wps_free_msearchreply(struct dl_list *head)
910 {
911         struct advertisement_state_machine *a, *tmp;
912         dl_list_for_each_safe(a, tmp, head, struct advertisement_state_machine,
913                               list)
914                 msearchreply_state_machine_stop(a);
915 }
916
917
918 static void upnp_wps_free_subscriptions(struct dl_list *head)
919 {
920         struct subscription *s, *tmp;
921         dl_list_for_each_safe(s, tmp, head, struct subscription, list) {
922                 dl_list_del(&s->list);
923                 subscription_destroy(s);
924         }
925 }
926
927
928 /**
929  * upnp_wps_device_stop - Stop WPS UPnP operations on an interface
930  * @sm: WPS UPnP state machine from upnp_wps_device_init()
931  */
932 void upnp_wps_device_stop(struct upnp_wps_device_sm *sm)
933 {
934         if (!sm || !sm->started)
935                 return;
936
937         wpa_printf(MSG_DEBUG, "WPS UPnP: Stop device");
938         web_listener_stop(sm);
939         upnp_wps_free_msearchreply(&sm->msearch_replies);
940         upnp_wps_free_subscriptions(&sm->subscriptions);
941
942         advertisement_state_machine_stop(sm, 1);
943
944         event_send_stop_all(sm);
945         os_free(sm->wlanevent);
946         sm->wlanevent = NULL;
947         os_free(sm->ip_addr_text);
948         sm->ip_addr_text = NULL;
949         if (sm->multicast_sd >= 0)
950                 close(sm->multicast_sd);
951         sm->multicast_sd = -1;
952         ssdp_listener_stop(sm);
953
954         sm->started = 0;
955 }
956
957
958 /**
959  * upnp_wps_device_start - Start WPS UPnP operations on an interface
960  * @sm: WPS UPnP state machine from upnp_wps_device_init()
961  * @net_if: Selected network interface name
962  * Returns: 0 on success, -1 on failure
963  */
964 int upnp_wps_device_start(struct upnp_wps_device_sm *sm, char *net_if)
965 {
966         if (!sm || !net_if)
967                 return -1;
968
969         if (sm->started)
970                 upnp_wps_device_stop(sm);
971
972         sm->multicast_sd = -1;
973         sm->ssdp_sd = -1;
974         sm->started = 1;
975         sm->advertise_count = 0;
976
977         /* Fix up linux multicast handling */
978         if (add_ssdp_network(net_if))
979                 goto fail;
980
981         /* Determine which IP and mac address we're using */
982         if (get_netif_info(net_if, &sm->ip_addr, &sm->ip_addr_text,
983                            sm->mac_addr)) {
984                 wpa_printf(MSG_INFO, "WPS UPnP: Could not get IP/MAC address "
985                            "for %s. Does it have IP address?", net_if);
986                 goto fail;
987         }
988
989         /* Listen for incoming TCP connections so that others
990          * can fetch our "xml files" from us.
991          */
992         if (web_listener_start(sm))
993                 goto fail;
994
995         /* Set up for receiving discovery (UDP) packets */
996         if (ssdp_listener_start(sm))
997                 goto fail;
998
999         /* Set up for sending multicast */
1000         if (ssdp_open_multicast(sm) < 0)
1001                 goto fail;
1002
1003         /*
1004          * Broadcast NOTIFY messages to let the world know we exist.
1005          * This is done via a state machine since the messages should not be
1006          * all sent out at once.
1007          */
1008         if (advertisement_state_machine_start(sm))
1009                 goto fail;
1010
1011         return 0;
1012
1013 fail:
1014         upnp_wps_device_stop(sm);
1015         return -1;
1016 }
1017
1018
1019 /**
1020  * upnp_wps_device_deinit - Deinitialize WPS UPnP
1021  * @sm: WPS UPnP state machine from upnp_wps_device_init()
1022  */
1023 void upnp_wps_device_deinit(struct upnp_wps_device_sm *sm)
1024 {
1025         if (!sm)
1026                 return;
1027
1028         upnp_wps_device_stop(sm);
1029
1030         if (sm->peer.wps)
1031                 wps_deinit(sm->peer.wps);
1032         os_free(sm->root_dir);
1033         os_free(sm->desc_url);
1034         os_free(sm->ctx->ap_pin);
1035         os_free(sm->ctx);
1036         os_free(sm);
1037 }
1038
1039
1040 /**
1041  * upnp_wps_device_init - Initialize WPS UPnP
1042  * @ctx: callback table; we must eventually free it
1043  * @wps: Pointer to longterm WPS context
1044  * @priv: External context data that will be used in callbacks
1045  * Returns: WPS UPnP state or %NULL on failure
1046  */
1047 struct upnp_wps_device_sm *
1048 upnp_wps_device_init(struct upnp_wps_device_ctx *ctx, struct wps_context *wps,
1049                      void *priv)
1050 {
1051         struct upnp_wps_device_sm *sm;
1052
1053         sm = os_zalloc(sizeof(*sm));
1054         if (!sm) {
1055                 wpa_printf(MSG_ERROR, "WPS UPnP: upnp_wps_device_init failed");
1056                 return NULL;
1057         }
1058
1059         sm->ctx = ctx;
1060         sm->wps = wps;
1061         sm->priv = priv;
1062         dl_list_init(&sm->msearch_replies);
1063         dl_list_init(&sm->subscriptions);
1064
1065         return sm;
1066 }
1067
1068
1069 /**
1070  * upnp_wps_subscribers - Check whether there are any event subscribers
1071  * @sm: WPS UPnP state machine from upnp_wps_device_init()
1072  * Returns: 0 if no subscribers, 1 if subscribers
1073  */
1074 int upnp_wps_subscribers(struct upnp_wps_device_sm *sm)
1075 {
1076         return !dl_list_empty(&sm->subscriptions);
1077 }
1078
1079
1080 int upnp_wps_set_ap_pin(struct upnp_wps_device_sm *sm, const char *ap_pin)
1081 {
1082         if (sm == NULL)
1083                 return 0;
1084
1085         os_free(sm->ctx->ap_pin);
1086         if (ap_pin) {
1087                 sm->ctx->ap_pin = os_strdup(ap_pin);
1088                 if (sm->ctx->ap_pin == NULL)
1089                         return -1;
1090         } else
1091                 sm->ctx->ap_pin = NULL;
1092
1093         return 0;
1094 }