]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/wpa/src/wps/wps_er.c
Update hostapd/wpa_supplicant to 2.8 to fix multiple vulnerabilities.
[FreeBSD/FreeBSD.git] / contrib / wpa / src / wps / wps_er.c
1 /*
2  * Wi-Fi Protected Setup - External Registrar
3  * Copyright (c) 2009-2013, 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 "base64.h"
13 #include "uuid.h"
14 #include "eloop.h"
15 #include "httpread.h"
16 #include "http_client.h"
17 #include "http_server.h"
18 #include "upnp_xml.h"
19 #include "wps_i.h"
20 #include "wps_upnp.h"
21 #include "wps_upnp_i.h"
22 #include "wps_er.h"
23
24
25 static void wps_er_deinit_finish(void *eloop_data, void *user_ctx);
26 static void wps_er_ap_timeout(void *eloop_data, void *user_ctx);
27 static void wps_er_sta_timeout(void *eloop_data, void *user_ctx);
28 static void wps_er_ap_process(struct wps_er_ap *ap, struct wpabuf *msg);
29 static int wps_er_send_get_device_info(struct wps_er_ap *ap,
30                                        void (*m1_handler)(struct wps_er_ap *ap,
31                                                           struct wpabuf *m1));
32
33
34 static void wps_er_sta_event(struct wps_context *wps, struct wps_er_sta *sta,
35                              enum wps_event event)
36 {
37         union wps_event_data data;
38         struct wps_event_er_enrollee *ev = &data.enrollee;
39
40         if (wps->event_cb == NULL)
41                 return;
42
43         os_memset(&data, 0, sizeof(data));
44         ev->uuid = sta->uuid;
45         ev->mac_addr = sta->addr;
46         ev->m1_received = sta->m1_received;
47         ev->config_methods = sta->config_methods;
48         ev->dev_passwd_id = sta->dev_passwd_id;
49         ev->pri_dev_type = sta->pri_dev_type;
50         ev->dev_name = sta->dev_name;
51         ev->manufacturer = sta->manufacturer;
52         ev->model_name = sta->model_name;
53         ev->model_number = sta->model_number;
54         ev->serial_number = sta->serial_number;
55         wps->event_cb(wps->cb_ctx, event, &data);
56 }
57
58
59 static struct wps_er_sta * wps_er_sta_get(struct wps_er_ap *ap, const u8 *addr,
60                                           const u8 *uuid)
61 {
62         struct wps_er_sta *sta;
63         dl_list_for_each(sta, &ap->sta, struct wps_er_sta, list) {
64                 if ((addr == NULL ||
65                      os_memcmp(sta->addr, addr, ETH_ALEN) == 0) &&
66                     (uuid == NULL ||
67                      os_memcmp(uuid, sta->uuid, WPS_UUID_LEN) == 0))
68                         return sta;
69         }
70         return NULL;
71 }
72
73
74 static void wps_er_sta_free(struct wps_er_sta *sta)
75 {
76         wps_er_sta_event(sta->ap->er->wps, sta, WPS_EV_ER_ENROLLEE_REMOVE);
77         if (sta->wps)
78                 wps_deinit(sta->wps);
79         os_free(sta->manufacturer);
80         os_free(sta->model_name);
81         os_free(sta->model_number);
82         os_free(sta->serial_number);
83         os_free(sta->dev_name);
84         http_client_free(sta->http);
85         eloop_cancel_timeout(wps_er_sta_timeout, sta, NULL);
86         os_free(sta->cred);
87         os_free(sta);
88 }
89
90
91 static void wps_er_sta_remove_all(struct wps_er_ap *ap)
92 {
93         struct wps_er_sta *prev, *sta;
94         dl_list_for_each_safe(sta, prev, &ap->sta, struct wps_er_sta, list)
95                 wps_er_sta_free(sta);
96 }
97
98
99 static struct wps_er_ap * wps_er_ap_get(struct wps_er *er,
100                                         struct in_addr *addr, const u8 *uuid,
101                                         const u8 *mac_addr)
102 {
103         struct wps_er_ap *ap;
104         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
105                 if ((addr == NULL || ap->addr.s_addr == addr->s_addr) &&
106                     (uuid == NULL ||
107                      os_memcmp(uuid, ap->uuid, WPS_UUID_LEN) == 0) &&
108                     (mac_addr == NULL ||
109                      os_memcmp(mac_addr, ap->mac_addr, ETH_ALEN) == 0))
110                         return ap;
111         }
112         return NULL;
113 }
114
115
116 static struct wps_er_ap * wps_er_ap_get_id(struct wps_er *er, unsigned int id)
117 {
118         struct wps_er_ap *ap;
119         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
120                 if (ap->id == id)
121                         return ap;
122         }
123         return NULL;
124 }
125
126
127 static void wps_er_ap_event(struct wps_context *wps, struct wps_er_ap *ap,
128                             enum wps_event event)
129 {
130         union wps_event_data data;
131         struct wps_event_er_ap *evap = &data.ap;
132
133         if (wps->event_cb == NULL)
134                 return;
135
136         os_memset(&data, 0, sizeof(data));
137         evap->uuid = ap->uuid;
138         evap->friendly_name = ap->friendly_name;
139         evap->manufacturer = ap->manufacturer;
140         evap->manufacturer_url = ap->manufacturer_url;
141         evap->model_description = ap->model_description;
142         evap->model_name = ap->model_name;
143         evap->model_number = ap->model_number;
144         evap->model_url = ap->model_url;
145         evap->serial_number = ap->serial_number;
146         evap->upc = ap->upc;
147         evap->pri_dev_type = ap->pri_dev_type;
148         evap->wps_state = ap->wps_state;
149         evap->mac_addr = ap->mac_addr;
150         wps->event_cb(wps->cb_ctx, event, &data);
151 }
152
153
154 static void wps_er_ap_free(struct wps_er_ap *ap)
155 {
156         http_client_free(ap->http);
157         ap->http = NULL;
158
159         os_free(ap->location);
160         os_free(ap->friendly_name);
161         os_free(ap->manufacturer);
162         os_free(ap->manufacturer_url);
163         os_free(ap->model_description);
164         os_free(ap->model_name);
165         os_free(ap->model_number);
166         os_free(ap->model_url);
167         os_free(ap->serial_number);
168         os_free(ap->udn);
169         os_free(ap->upc);
170
171         os_free(ap->scpd_url);
172         os_free(ap->control_url);
173         os_free(ap->event_sub_url);
174
175         os_free(ap->ap_settings);
176
177         os_free(ap);
178 }
179
180
181 static void wps_er_ap_unsubscribed(struct wps_er *er, struct wps_er_ap *ap)
182 {
183         wpa_printf(MSG_DEBUG, "WPS ER: Unsubscribed from AP %s (%s)",
184                    inet_ntoa(ap->addr), ap->location);
185         dl_list_del(&ap->list);
186         wps_er_ap_free(ap);
187
188         if (er->deinitializing && dl_list_empty(&er->ap_unsubscribing))
189                 wps_er_deinit_finish(er, NULL);
190 }
191
192
193 static void wps_er_http_unsubscribe_cb(void *ctx, struct http_client *c,
194                                        enum http_client_event event)
195 {
196         struct wps_er_ap *ap = ctx;
197
198         switch (event) {
199         case HTTP_CLIENT_OK:
200                 wpa_printf(MSG_DEBUG, "WPS ER: Unsubscribed from events");
201                 ap->subscribed = 0;
202                 break;
203         case HTTP_CLIENT_FAILED:
204         case HTTP_CLIENT_INVALID_REPLY:
205         case HTTP_CLIENT_TIMEOUT:
206                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to unsubscribe from "
207                            "events");
208                 break;
209         }
210         http_client_free(ap->http);
211         ap->http = NULL;
212
213         /*
214          * Need to get rid of the AP entry regardless of whether we managed to
215          * unsubscribe cleanly or not.
216          */
217         wps_er_ap_unsubscribed(ap->er, ap);
218 }
219
220
221 static void wps_er_ap_unsubscribe(struct wps_er *er, struct wps_er_ap *ap)
222 {
223         struct wpabuf *req;
224         struct sockaddr_in dst;
225         char *url, *path;
226         char sid[100];
227
228         if (ap->event_sub_url == NULL) {
229                 wpa_printf(MSG_DEBUG, "WPS ER: No eventSubURL - cannot "
230                            "subscribe");
231                 goto fail;
232         }
233         if (ap->http) {
234                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP request - cannot "
235                            "send subscribe request");
236                 goto fail;
237         }
238
239         url = http_client_url_parse(ap->event_sub_url, &dst, &path);
240         if (url == NULL) {
241                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse eventSubURL");
242                 goto fail;
243         }
244
245         req = wpabuf_alloc(os_strlen(ap->event_sub_url) + 1000);
246         if (req == NULL) {
247                 os_free(url);
248                 goto fail;
249         }
250         uuid_bin2str(ap->sid, sid, sizeof(sid));
251         wpabuf_printf(req,
252                       "UNSUBSCRIBE %s HTTP/1.1\r\n"
253                       "HOST: %s:%d\r\n"
254                       "SID: uuid:%s\r\n"
255                       "\r\n",
256                       path, inet_ntoa(dst.sin_addr), ntohs(dst.sin_port), sid);
257         os_free(url);
258         wpa_hexdump_ascii(MSG_MSGDUMP, "WPS ER: Unsubscription request",
259                           wpabuf_head(req), wpabuf_len(req));
260
261         ap->http = http_client_addr(&dst, req, 1000,
262                                     wps_er_http_unsubscribe_cb, ap);
263         if (ap->http == NULL) {
264                 wpabuf_free(req);
265                 goto fail;
266         }
267         return;
268
269 fail:
270         /*
271          * Need to get rid of the AP entry even when we fail to unsubscribe
272          * cleanly.
273          */
274         wps_er_ap_unsubscribed(ap->er, ap);
275 }
276
277
278 static struct wps_er_ap_settings * wps_er_ap_get_settings(struct wps_er *er,
279                                                           const u8 *uuid)
280 {
281         struct wps_er_ap_settings *s;
282         dl_list_for_each(s, &er->ap_settings, struct wps_er_ap_settings, list)
283                 if (os_memcmp(uuid, s->uuid, WPS_UUID_LEN) == 0)
284                         return s;
285         return NULL;
286 }
287
288
289 int wps_er_ap_cache_settings(struct wps_er *er, struct in_addr *addr)
290 {
291         struct wps_er_ap *ap;
292         struct wps_er_ap_settings *settings;
293
294         ap = wps_er_ap_get(er, addr, NULL, NULL);
295         if (ap == NULL || ap->ap_settings == NULL)
296                 return -1;
297
298         settings = wps_er_ap_get_settings(er, ap->uuid);
299         if (!settings) {
300                 settings = os_zalloc(sizeof(*settings));
301                 if (settings == NULL)
302                         return -1;
303                 os_memcpy(settings->uuid, ap->uuid, WPS_UUID_LEN);
304                 dl_list_add(&er->ap_settings, &settings->list);
305         }
306         os_memcpy(&settings->ap_settings, ap->ap_settings,
307                   sizeof(struct wps_credential));
308
309         return 0;
310 }
311
312
313 static int wps_er_ap_use_cached_settings(struct wps_er *er,
314                                          struct wps_er_ap *ap)
315 {
316         struct wps_er_ap_settings *s;
317
318         if (ap->ap_settings)
319                 return 0;
320
321         s = wps_er_ap_get_settings(ap->er, ap->uuid);
322         if (!s)
323                 return -1;
324
325         ap->ap_settings = os_memdup(&s->ap_settings, sizeof(*ap->ap_settings));
326         if (ap->ap_settings == NULL)
327                 return -1;
328
329         wpa_printf(MSG_DEBUG, "WPS ER: Use cached AP settings");
330         return 0;
331 }
332
333
334 static void wps_er_ap_remove_entry(struct wps_er *er, struct wps_er_ap *ap)
335 {
336         wpa_printf(MSG_DEBUG, "WPS ER: Removing AP entry for %s (%s)",
337                    inet_ntoa(ap->addr), ap->location);
338         eloop_cancel_timeout(wps_er_ap_timeout, er, ap);
339         wps_er_sta_remove_all(ap);
340         wps_er_ap_event(er->wps, ap, WPS_EV_ER_AP_REMOVE);
341         http_client_free(ap->http);
342         ap->http = NULL;
343         if (ap->wps) {
344                 wps_deinit(ap->wps);
345                 ap->wps = NULL;
346         }
347
348         dl_list_del(&ap->list);
349         if (ap->subscribed) {
350                 dl_list_add(&er->ap_unsubscribing, &ap->list);
351                 wps_er_ap_unsubscribe(er, ap);
352         } else
353                 wps_er_ap_free(ap);
354 }
355
356
357 static void wps_er_ap_timeout(void *eloop_data, void *user_ctx)
358 {
359         struct wps_er *er = eloop_data;
360         struct wps_er_ap *ap = user_ctx;
361         wpa_printf(MSG_DEBUG, "WPS ER: AP advertisement timed out");
362         wps_er_ap_remove_entry(er, ap);
363 }
364
365
366 static int wps_er_get_sid(struct wps_er_ap *ap, char *sid)
367 {
368         char *pos;
369         char txt[100];
370
371         if (!sid) {
372                 wpa_printf(MSG_DEBUG, "WPS ER: No SID received from %s (%s)",
373                            inet_ntoa(ap->addr), ap->location);
374                 return -1;
375         }
376
377         pos = os_strstr(sid, "uuid:");
378         if (!pos) {
379                 wpa_printf(MSG_DEBUG, "WPS ER: Invalid SID received from "
380                            "%s (%s): '%s'", inet_ntoa(ap->addr), ap->location,
381                            sid);
382                 return -1;
383         }
384
385         pos += 5;
386         if (uuid_str2bin(pos, ap->sid) < 0) {
387                 wpa_printf(MSG_DEBUG, "WPS ER: Invalid SID received from "
388                            "%s (%s): '%s'", inet_ntoa(ap->addr), ap->location,
389                            sid);
390                 return -1;
391         }
392
393         uuid_bin2str(ap->sid, txt, sizeof(txt));
394         wpa_printf(MSG_DEBUG, "WPS ER: SID for subscription with %s (%s): %s",
395                    inet_ntoa(ap->addr), ap->location, txt);
396
397         return 0;
398 }
399
400
401 static void wps_er_http_subscribe_cb(void *ctx, struct http_client *c,
402                                      enum http_client_event event)
403 {
404         struct wps_er_ap *ap = ctx;
405
406         switch (event) {
407         case HTTP_CLIENT_OK:
408                 wpa_printf(MSG_DEBUG, "WPS ER: Subscribed to events");
409                 ap->subscribed = 1;
410                 wps_er_get_sid(ap, http_client_get_hdr_line(c, "SID"));
411                 wps_er_ap_use_cached_settings(ap->er, ap);
412                 wps_er_ap_event(ap->er->wps, ap, WPS_EV_ER_AP_ADD);
413                 break;
414         case HTTP_CLIENT_FAILED:
415         case HTTP_CLIENT_INVALID_REPLY:
416         case HTTP_CLIENT_TIMEOUT:
417                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to subscribe to events");
418                 break;
419         }
420         http_client_free(ap->http);
421         ap->http = NULL;
422 }
423
424
425 static void wps_er_subscribe(struct wps_er_ap *ap)
426 {
427         struct wpabuf *req;
428         struct sockaddr_in dst;
429         char *url, *path;
430
431         if (ap->event_sub_url == NULL) {
432                 wpa_printf(MSG_DEBUG, "WPS ER: No eventSubURL - cannot "
433                            "subscribe");
434                 return;
435         }
436         if (ap->http) {
437                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP request - cannot "
438                            "send subscribe request");
439                 return;
440         }
441
442         url = http_client_url_parse(ap->event_sub_url, &dst, &path);
443         if (url == NULL) {
444                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse eventSubURL");
445                 return;
446         }
447
448         req = wpabuf_alloc(os_strlen(ap->event_sub_url) + 1000);
449         if (req == NULL) {
450                 os_free(url);
451                 return;
452         }
453         wpabuf_printf(req,
454                       "SUBSCRIBE %s HTTP/1.1\r\n"
455                       "HOST: %s:%d\r\n"
456                       "CALLBACK: <http://%s:%d/event/%u/%u>\r\n"
457                       "NT: upnp:event\r\n"
458                       "TIMEOUT: Second-%d\r\n"
459                       "\r\n",
460                       path, inet_ntoa(dst.sin_addr), ntohs(dst.sin_port),
461                       ap->er->ip_addr_text, ap->er->http_port,
462                       ap->er->event_id, ap->id, 1800);
463         os_free(url);
464         wpa_hexdump_ascii(MSG_MSGDUMP, "WPS ER: Subscription request",
465                           wpabuf_head(req), wpabuf_len(req));
466
467         ap->http = http_client_addr(&dst, req, 1000, wps_er_http_subscribe_cb,
468                                     ap);
469         if (ap->http == NULL)
470                 wpabuf_free(req);
471 }
472
473
474 static void wps_er_ap_get_m1(struct wps_er_ap *ap, struct wpabuf *m1)
475 {
476         struct wps_parse_attr attr;
477
478         if (wps_parse_msg(m1, &attr) < 0) {
479                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse M1");
480                 return;
481         }
482         if (attr.primary_dev_type)
483                 os_memcpy(ap->pri_dev_type, attr.primary_dev_type, 8);
484         if (attr.wps_state)
485                 ap->wps_state = *attr.wps_state;
486         if (attr.mac_addr)
487                 os_memcpy(ap->mac_addr, attr.mac_addr, ETH_ALEN);
488
489         wps_er_subscribe(ap);
490 }
491
492
493 static void wps_er_get_device_info(struct wps_er_ap *ap)
494 {
495         wps_er_send_get_device_info(ap, wps_er_ap_get_m1);
496 }
497
498
499 static const char * wps_er_find_wfadevice(const char *data)
500 {
501         const char *tag, *tagname, *end;
502         char *val;
503         int found = 0;
504
505         while (!found) {
506                 /* Find next <device> */
507                 for (;;) {
508                         if (xml_next_tag(data, &tag, &tagname, &end))
509                                 return NULL;
510                         data = end;
511                         if (!os_strncasecmp(tagname, "device", 6) &&
512                             *tag != '/' &&
513                             (tagname[6] == '>' || !isgraph(tagname[6]))) {
514                                 break;
515                         }
516                 }
517
518                 /* Check whether deviceType is WFADevice */
519                 val = xml_get_first_item(data, "deviceType");
520                 if (val == NULL)
521                         return NULL;
522                 wpa_printf(MSG_DEBUG, "WPS ER: Found deviceType '%s'", val);
523                 found = os_strcasecmp(val, "urn:schemas-wifialliance-org:"
524                                       "device:WFADevice:1") == 0;
525                 os_free(val);
526         }
527
528         return data;
529 }
530
531
532 static void wps_er_parse_device_description(struct wps_er_ap *ap,
533                                             struct wpabuf *reply)
534 {
535         /* Note: reply includes null termination after the buffer data */
536         const char *tmp, *data = wpabuf_head(reply);
537         char *pos;
538
539         wpa_hexdump_ascii(MSG_MSGDUMP, "WPS ER: Device info",
540                           wpabuf_head(reply), wpabuf_len(reply));
541
542         /*
543          * The root device description may include multiple devices, so first
544          * find the beginning of the WFADevice description to allow the
545          * simplistic parser to pick the correct entries.
546          */
547         tmp = wps_er_find_wfadevice(data);
548         if (tmp == NULL) {
549                 wpa_printf(MSG_DEBUG, "WPS ER: WFADevice:1 device not found - "
550                            "trying to parse invalid data");
551         } else
552                 data = tmp;
553
554         ap->friendly_name = xml_get_first_item(data, "friendlyName");
555         wpa_printf(MSG_DEBUG, "WPS ER: friendlyName='%s'", ap->friendly_name);
556
557         ap->manufacturer = xml_get_first_item(data, "manufacturer");
558         wpa_printf(MSG_DEBUG, "WPS ER: manufacturer='%s'", ap->manufacturer);
559
560         ap->manufacturer_url = xml_get_first_item(data, "manufacturerURL");
561         wpa_printf(MSG_DEBUG, "WPS ER: manufacturerURL='%s'",
562                    ap->manufacturer_url);
563
564         ap->model_description = xml_get_first_item(data, "modelDescription");
565         wpa_printf(MSG_DEBUG, "WPS ER: modelDescription='%s'",
566                    ap->model_description);
567
568         ap->model_name = xml_get_first_item(data, "modelName");
569         wpa_printf(MSG_DEBUG, "WPS ER: modelName='%s'", ap->model_name);
570
571         ap->model_number = xml_get_first_item(data, "modelNumber");
572         wpa_printf(MSG_DEBUG, "WPS ER: modelNumber='%s'", ap->model_number);
573
574         ap->model_url = xml_get_first_item(data, "modelURL");
575         wpa_printf(MSG_DEBUG, "WPS ER: modelURL='%s'", ap->model_url);
576
577         ap->serial_number = xml_get_first_item(data, "serialNumber");
578         wpa_printf(MSG_DEBUG, "WPS ER: serialNumber='%s'", ap->serial_number);
579
580         ap->udn = xml_get_first_item(data, "UDN");
581         if (ap->udn) {
582                 wpa_printf(MSG_DEBUG, "WPS ER: UDN='%s'", ap->udn);
583                 pos = os_strstr(ap->udn, "uuid:");
584                 if (pos) {
585                         pos += 5;
586                         if (uuid_str2bin(pos, ap->uuid) < 0)
587                                 wpa_printf(MSG_DEBUG,
588                                            "WPS ER: Invalid UUID in UDN");
589                 }
590         }
591
592         ap->upc = xml_get_first_item(data, "UPC");
593         wpa_printf(MSG_DEBUG, "WPS ER: UPC='%s'", ap->upc);
594
595         ap->scpd_url = http_link_update(
596                 xml_get_first_item(data, "SCPDURL"), ap->location);
597         wpa_printf(MSG_DEBUG, "WPS ER: SCPDURL='%s'", ap->scpd_url);
598
599         ap->control_url = http_link_update(
600                 xml_get_first_item(data, "controlURL"), ap->location);
601         wpa_printf(MSG_DEBUG, "WPS ER: controlURL='%s'", ap->control_url);
602
603         ap->event_sub_url = http_link_update(
604                 xml_get_first_item(data, "eventSubURL"), ap->location);
605         wpa_printf(MSG_DEBUG, "WPS ER: eventSubURL='%s'", ap->event_sub_url);
606 }
607
608
609 static void wps_er_http_dev_desc_cb(void *ctx, struct http_client *c,
610                                     enum http_client_event event)
611 {
612         struct wps_er_ap *ap = ctx;
613         struct wpabuf *reply;
614         int ok = 0;
615
616         switch (event) {
617         case HTTP_CLIENT_OK:
618                 reply = http_client_get_body(c);
619                 if (reply == NULL)
620                         break;
621                 wps_er_parse_device_description(ap, reply);
622                 ok = 1;
623                 break;
624         case HTTP_CLIENT_FAILED:
625         case HTTP_CLIENT_INVALID_REPLY:
626         case HTTP_CLIENT_TIMEOUT:
627                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to fetch device info");
628                 break;
629         }
630         http_client_free(ap->http);
631         ap->http = NULL;
632         if (ok)
633                 wps_er_get_device_info(ap);
634 }
635
636
637 void wps_er_ap_add(struct wps_er *er, const u8 *uuid, struct in_addr *addr,
638                    const char *location, int max_age)
639 {
640         struct wps_er_ap *ap;
641
642         ap = wps_er_ap_get(er, addr, uuid, NULL);
643         if (ap) {
644                 /* Update advertisement timeout */
645                 eloop_cancel_timeout(wps_er_ap_timeout, er, ap);
646                 eloop_register_timeout(max_age, 0, wps_er_ap_timeout, er, ap);
647                 return;
648         }
649
650         ap = os_zalloc(sizeof(*ap));
651         if (ap == NULL)
652                 return;
653         dl_list_init(&ap->sta);
654         ap->er = er;
655         ap->id = ++er->next_ap_id;
656         ap->location = os_strdup(location);
657         if (ap->location == NULL) {
658                 os_free(ap);
659                 return;
660         }
661         dl_list_add(&er->ap, &ap->list);
662
663         ap->addr.s_addr = addr->s_addr;
664         os_memcpy(ap->uuid, uuid, WPS_UUID_LEN);
665         eloop_register_timeout(max_age, 0, wps_er_ap_timeout, er, ap);
666
667         wpa_printf(MSG_DEBUG, "WPS ER: Added AP entry for %s (%s)",
668                    inet_ntoa(ap->addr), ap->location);
669
670         /* Fetch device description */
671         ap->http = http_client_url(ap->location, NULL, 10000,
672                                    wps_er_http_dev_desc_cb, ap);
673 }
674
675
676 void wps_er_ap_remove(struct wps_er *er, struct in_addr *addr)
677 {
678         struct wps_er_ap *ap;
679         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
680                 if (ap->addr.s_addr == addr->s_addr) {
681                         wps_er_ap_remove_entry(er, ap);
682                         return;
683                 }
684         }
685 }
686
687
688 static void wps_er_ap_remove_all(struct wps_er *er)
689 {
690         struct wps_er_ap *prev, *ap;
691         struct wps_er_ap_settings *prev_s, *s;
692         dl_list_for_each_safe(ap, prev, &er->ap, struct wps_er_ap, list)
693                 wps_er_ap_remove_entry(er, ap);
694         dl_list_for_each_safe(s, prev_s, &er->ap_settings,
695                               struct wps_er_ap_settings, list)
696                 os_free(s);
697 }
698
699
700 static void http_put_date(struct wpabuf *buf)
701 {
702         wpabuf_put_str(buf, "Date: ");
703         format_date(buf);
704         wpabuf_put_str(buf, "\r\n");
705 }
706
707
708 static void wps_er_http_resp_not_found(struct http_request *req)
709 {
710         struct wpabuf *buf;
711         buf = wpabuf_alloc(200);
712         if (buf == NULL) {
713                 http_request_deinit(req);
714                 return;
715         }
716
717         wpabuf_put_str(buf,
718                        "HTTP/1.1 404 Not Found\r\n"
719                        "Server: unspecified, UPnP/1.0, unspecified\r\n"
720                        "Connection: close\r\n");
721         http_put_date(buf);
722         wpabuf_put_str(buf, "\r\n");
723         http_request_send_and_deinit(req, buf);
724 }
725
726
727 static void wps_er_http_resp_ok(struct http_request *req)
728 {
729         struct wpabuf *buf;
730         buf = wpabuf_alloc(200);
731         if (buf == NULL) {
732                 http_request_deinit(req);
733                 return;
734         }
735
736         wpabuf_put_str(buf,
737                        "HTTP/1.1 200 OK\r\n"
738                        "Server: unspecified, UPnP/1.0, unspecified\r\n"
739                        "Connection: close\r\n"
740                        "Content-Length: 0\r\n");
741         http_put_date(buf);
742         wpabuf_put_str(buf, "\r\n");
743         http_request_send_and_deinit(req, buf);
744 }
745
746
747 static void wps_er_sta_timeout(void *eloop_data, void *user_ctx)
748 {
749         struct wps_er_sta *sta = eloop_data;
750         wpa_printf(MSG_DEBUG, "WPS ER: STA entry timed out");
751         dl_list_del(&sta->list);
752         wps_er_sta_free(sta);
753 }
754
755
756 static struct wps_er_sta * wps_er_add_sta_data(struct wps_er_ap *ap,
757                                                const u8 *addr,
758                                                struct wps_parse_attr *attr,
759                                                int probe_req)
760 {
761         struct wps_er_sta *sta = wps_er_sta_get(ap, addr, NULL);
762         int new_sta = 0;
763         int m1;
764
765         m1 = !probe_req && attr->msg_type && *attr->msg_type == WPS_M1;
766
767         if (sta == NULL) {
768                 /*
769                  * Only allow new STA entry to be added based on Probe Request
770                  * or M1. This will filter out bogus events and anything that
771                  * may have been ongoing at the time ER subscribed for events.
772                  */
773                 if (!probe_req && !m1)
774                         return NULL;
775
776                 sta = os_zalloc(sizeof(*sta));
777                 if (sta == NULL)
778                         return NULL;
779                 os_memcpy(sta->addr, addr, ETH_ALEN);
780                 sta->ap = ap;
781                 dl_list_add(&ap->sta, &sta->list);
782                 new_sta = 1;
783         }
784
785         if (m1)
786                 sta->m1_received = 1;
787
788         if (attr->config_methods && (!probe_req || !sta->m1_received))
789                 sta->config_methods = WPA_GET_BE16(attr->config_methods);
790         if (attr->uuid_e && (!probe_req || !sta->m1_received))
791                 os_memcpy(sta->uuid, attr->uuid_e, WPS_UUID_LEN);
792         if (attr->primary_dev_type && (!probe_req || !sta->m1_received))
793                 os_memcpy(sta->pri_dev_type, attr->primary_dev_type, 8);
794         if (attr->dev_password_id && (!probe_req || !sta->m1_received))
795                 sta->dev_passwd_id = WPA_GET_BE16(attr->dev_password_id);
796
797         if (attr->manufacturer) {
798                 os_free(sta->manufacturer);
799                 sta->manufacturer = dup_binstr(attr->manufacturer,
800                                                attr->manufacturer_len);
801         }
802
803         if (attr->model_name) {
804                 os_free(sta->model_name);
805                 sta->model_name = dup_binstr(attr->model_name,
806                                              attr->model_name_len);
807         }
808
809         if (attr->model_number) {
810                 os_free(sta->model_number);
811                 sta->model_number = dup_binstr(attr->model_number,
812                                                attr->model_number_len);
813         }
814
815         if (attr->serial_number) {
816                 os_free(sta->serial_number);
817                 sta->serial_number = dup_binstr(attr->serial_number,
818                                                 attr->serial_number_len);
819         }
820
821         if (attr->dev_name) {
822                 os_free(sta->dev_name);
823                 sta->dev_name = dup_binstr(attr->dev_name, attr->dev_name_len);
824         }
825
826         eloop_cancel_timeout(wps_er_sta_timeout, sta, NULL);
827         eloop_register_timeout(300, 0, wps_er_sta_timeout, sta, NULL);
828
829         if (m1 || new_sta)
830                 wps_er_sta_event(ap->er->wps, sta, WPS_EV_ER_ENROLLEE_ADD);
831
832         return sta;
833 }
834
835
836 static void wps_er_process_wlanevent_probe_req(struct wps_er_ap *ap,
837                                                const u8 *addr,
838                                                struct wpabuf *msg)
839 {
840         struct wps_parse_attr attr;
841
842         wpa_printf(MSG_DEBUG, "WPS ER: WLANEvent - Probe Request - from "
843                    MACSTR, MAC2STR(addr));
844         wpa_hexdump_buf(MSG_MSGDUMP, "WPS ER: WLANEvent - Enrollee's message "
845                         "(TLVs from Probe Request)", msg);
846
847         if (wps_validate_probe_req(msg, addr) < 0) {
848                 wpa_printf(MSG_INFO, "WPS-STRICT: ER: Ignore invalid proxied "
849                            "Probe Request frame from " MACSTR, MAC2STR(addr));
850                 return;
851         }
852
853         if (wps_parse_msg(msg, &attr) < 0) {
854                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse TLVs in "
855                            "WLANEvent message");
856                 return;
857         }
858
859         wps_er_add_sta_data(ap, addr, &attr, 1);
860         wps_registrar_probe_req_rx(ap->er->wps->registrar, addr, msg, 0);
861 }
862
863
864 static void wps_er_http_put_wlan_response_cb(void *ctx, struct http_client *c,
865                                              enum http_client_event event)
866 {
867         struct wps_er_sta *sta = ctx;
868
869         switch (event) {
870         case HTTP_CLIENT_OK:
871                 wpa_printf(MSG_DEBUG, "WPS ER: PutWLANResponse OK");
872                 break;
873         case HTTP_CLIENT_FAILED:
874         case HTTP_CLIENT_INVALID_REPLY:
875         case HTTP_CLIENT_TIMEOUT:
876                 wpa_printf(MSG_DEBUG, "WPS ER: PutWLANResponse failed");
877                 break;
878         }
879         http_client_free(sta->http);
880         sta->http = NULL;
881 }
882
883
884 static const char *soap_prefix =
885         "<?xml version=\"1.0\"?>\n"
886         "<s:Envelope xmlns:s=\"http://schemas.xmlsoap.org/soap/envelope/\" "
887         "s:encodingStyle=\"http://schemas.xmlsoap.org/soap/encoding/\">\n"
888         "<s:Body>\n";
889 static const char *soap_postfix =
890         "</s:Body>\n</s:Envelope>\n";
891 static const char *urn_wfawlanconfig =
892         "urn:schemas-wifialliance-org:service:WFAWLANConfig:1";
893
894 static struct wpabuf * wps_er_soap_hdr(const struct wpabuf *msg,
895                                        const char *name, const char *arg_name,
896                                        const char *path,
897                                        const struct sockaddr_in *dst,
898                                        char **len_ptr, char **body_ptr)
899 {
900         unsigned char *encoded;
901         size_t encoded_len;
902         struct wpabuf *buf;
903
904         if (msg) {
905                 encoded = base64_encode(wpabuf_head(msg), wpabuf_len(msg),
906                                         &encoded_len);
907                 if (encoded == NULL)
908                         return NULL;
909         } else {
910                 encoded = NULL;
911                 encoded_len = 0;
912         }
913
914         buf = wpabuf_alloc(1000 + encoded_len);
915         if (buf == NULL) {
916                 os_free(encoded);
917                 return NULL;
918         }
919
920         wpabuf_printf(buf,
921                       "POST %s HTTP/1.1\r\n"
922                       "Host: %s:%d\r\n"
923                       "Content-Type: text/xml; charset=\"utf-8\"\r\n"
924                       "Content-Length: ",
925                       path, inet_ntoa(dst->sin_addr), ntohs(dst->sin_port));
926
927         *len_ptr = wpabuf_put(buf, 0);
928         wpabuf_printf(buf,
929                       "        \r\n"
930                       "SOAPACTION: \"%s#%s\"\r\n"
931                       "\r\n",
932                       urn_wfawlanconfig, name);
933
934         *body_ptr = wpabuf_put(buf, 0);
935
936         wpabuf_put_str(buf, soap_prefix);
937         wpabuf_printf(buf, "<u:%s xmlns:u=\"", name);
938         wpabuf_put_str(buf, urn_wfawlanconfig);
939         wpabuf_put_str(buf, "\">\n");
940         if (encoded) {
941                 wpabuf_printf(buf, "<%s>%s</%s>\n",
942                               arg_name, (char *) encoded, arg_name);
943                 os_free(encoded);
944         }
945
946         return buf;
947 }
948
949
950 static void wps_er_soap_end(struct wpabuf *buf, const char *name,
951                             char *len_ptr, char *body_ptr)
952 {
953         char len_buf[10];
954         wpabuf_printf(buf, "</u:%s>\n", name);
955         wpabuf_put_str(buf, soap_postfix);
956         os_snprintf(len_buf, sizeof(len_buf), "%d",
957                     (int) ((char *) wpabuf_put(buf, 0) - body_ptr));
958         os_memcpy(len_ptr, len_buf, os_strlen(len_buf));
959 }
960
961
962 static void wps_er_sta_send_msg(struct wps_er_sta *sta, struct wpabuf *msg)
963 {
964         struct wpabuf *buf;
965         char *len_ptr, *body_ptr;
966         struct sockaddr_in dst;
967         char *url, *path;
968
969         if (sta->http) {
970                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP request for STA - "
971                            "ignore new request");
972                 wpabuf_free(msg);
973                 return;
974         }
975
976         if (sta->ap->control_url == NULL) {
977                 wpa_printf(MSG_DEBUG, "WPS ER: No controlURL for AP");
978                 wpabuf_free(msg);
979                 return;
980         }
981
982         url = http_client_url_parse(sta->ap->control_url, &dst, &path);
983         if (url == NULL) {
984                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse controlURL");
985                 wpabuf_free(msg);
986                 return;
987         }
988
989         buf = wps_er_soap_hdr(msg, "PutWLANResponse", "NewMessage", path, &dst,
990                               &len_ptr, &body_ptr);
991         wpabuf_free(msg);
992         os_free(url);
993         if (buf == NULL)
994                 return;
995         wpabuf_printf(buf, "<NewWLANEventType>%d</NewWLANEventType>\n",
996                       UPNP_WPS_WLANEVENT_TYPE_EAP);
997         wpabuf_printf(buf, "<NewWLANEventMAC>" MACSTR "</NewWLANEventMAC>\n",
998                       MAC2STR(sta->addr));
999
1000         wps_er_soap_end(buf, "PutWLANResponse", len_ptr, body_ptr);
1001
1002         sta->http = http_client_addr(&dst, buf, 1000,
1003                                      wps_er_http_put_wlan_response_cb, sta);
1004         if (sta->http == NULL)
1005                 wpabuf_free(buf);
1006 }
1007
1008
1009 static void wps_er_sta_process(struct wps_er_sta *sta, struct wpabuf *msg,
1010                                enum wsc_op_code op_code)
1011 {
1012         enum wps_process_res res;
1013
1014         res = wps_process_msg(sta->wps, op_code, msg);
1015         if (res == WPS_CONTINUE) {
1016                 struct wpabuf *next = wps_get_msg(sta->wps, &op_code);
1017                 if (next)
1018                         wps_er_sta_send_msg(sta, next);
1019         } else {
1020                 wpa_printf(MSG_DEBUG, "WPS ER: Protocol run %s with the "
1021                            "enrollee (res=%d)",
1022                            res == WPS_DONE ? "succeeded" : "failed", res);
1023                 wps_deinit(sta->wps);
1024                 sta->wps = NULL;
1025                 if (res == WPS_DONE) {
1026                         /* Remove the STA entry after short timeout */
1027                         eloop_cancel_timeout(wps_er_sta_timeout, sta, NULL);
1028                         eloop_register_timeout(10, 0, wps_er_sta_timeout, sta,
1029                                                NULL);
1030                 }
1031         }
1032 }
1033
1034
1035 static void wps_er_sta_start(struct wps_er_sta *sta, struct wpabuf *msg)
1036 {
1037         struct wps_config cfg;
1038
1039         if (sta->wps)
1040                 wps_deinit(sta->wps);
1041
1042         os_memset(&cfg, 0, sizeof(cfg));
1043         cfg.wps = sta->ap->er->wps;
1044         cfg.registrar = 1;
1045         cfg.peer_addr = sta->addr;
1046
1047         sta->wps = wps_init(&cfg);
1048         if (sta->wps == NULL)
1049                 return;
1050         sta->wps->er = 1;
1051         sta->wps->use_cred = sta->ap->ap_settings;
1052         if (sta->ap->ap_settings) {
1053                 os_free(sta->cred);
1054                 sta->cred = os_malloc(sizeof(*sta->cred));
1055                 if (sta->cred) {
1056                         os_memcpy(sta->cred, sta->ap->ap_settings,
1057                                   sizeof(*sta->cred));
1058                         sta->cred->cred_attr = NULL;
1059                         os_memcpy(sta->cred->mac_addr, sta->addr, ETH_ALEN);
1060                         sta->wps->use_cred = sta->cred;
1061                 }
1062         }
1063
1064         wps_er_sta_process(sta, msg, WSC_MSG);
1065 }
1066
1067
1068 static void wps_er_process_wlanevent_eap(struct wps_er_ap *ap, const u8 *addr,
1069                                          struct wpabuf *msg)
1070 {
1071         struct wps_parse_attr attr;
1072         struct wps_er_sta *sta;
1073
1074         wpa_printf(MSG_DEBUG, "WPS ER: WLANEvent - EAP - from " MACSTR,
1075                    MAC2STR(addr));
1076         wpa_hexdump_buf(MSG_MSGDUMP, "WPS ER: WLANEvent - Enrollee's message "
1077                         "(TLVs from EAP-WSC)", msg);
1078
1079         if (wps_parse_msg(msg, &attr) < 0) {
1080                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse TLVs in "
1081                            "WLANEvent message");
1082                 return;
1083         }
1084
1085         sta = wps_er_add_sta_data(ap, addr, &attr, 0);
1086         if (sta == NULL)
1087                 return;
1088
1089         if (attr.msg_type && *attr.msg_type == WPS_M1)
1090                 wps_er_sta_start(sta, msg);
1091         else if (sta->wps) {
1092                 enum wsc_op_code op_code = WSC_MSG;
1093                 if (attr.msg_type) {
1094                         switch (*attr.msg_type) {
1095                         case WPS_WSC_ACK:
1096                                 op_code = WSC_ACK;
1097                                 break;
1098                         case WPS_WSC_NACK:
1099                                 op_code = WSC_NACK;
1100                                 break;
1101                         case WPS_WSC_DONE:
1102                                 op_code = WSC_Done;
1103                                 break;
1104                         }
1105                 }
1106                 wps_er_sta_process(sta, msg, op_code);
1107         }
1108 }
1109
1110
1111 static void wps_er_process_wlanevent(struct wps_er_ap *ap,
1112                                      struct wpabuf *event)
1113 {
1114         u8 *data;
1115         u8 wlan_event_type;
1116         u8 wlan_event_mac[ETH_ALEN];
1117         struct wpabuf msg;
1118
1119         wpa_hexdump(MSG_MSGDUMP, "WPS ER: Received WLANEvent",
1120                     wpabuf_head(event), wpabuf_len(event));
1121         if (wpabuf_len(event) < 1 + 17) {
1122                 wpa_printf(MSG_DEBUG, "WPS ER: Too short WLANEvent");
1123                 return;
1124         }
1125
1126         data = wpabuf_mhead(event);
1127         wlan_event_type = data[0];
1128         if (hwaddr_aton((char *) data + 1, wlan_event_mac) < 0) {
1129                 wpa_printf(MSG_DEBUG, "WPS ER: Invalid WLANEventMAC in "
1130                            "WLANEvent");
1131                 return;
1132         }
1133
1134         wpabuf_set(&msg, data + 1 + 17, wpabuf_len(event) - (1 + 17));
1135
1136         switch (wlan_event_type) {
1137         case 1:
1138                 wps_er_process_wlanevent_probe_req(ap, wlan_event_mac, &msg);
1139                 break;
1140         case 2:
1141                 wps_er_process_wlanevent_eap(ap, wlan_event_mac, &msg);
1142                 break;
1143         default:
1144                 wpa_printf(MSG_DEBUG, "WPS ER: Unknown WLANEventType %d",
1145                            wlan_event_type);
1146                 break;
1147         }
1148 }
1149
1150
1151 static void wps_er_http_event(struct wps_er *er, struct http_request *req,
1152                               unsigned int ap_id)
1153 {
1154         struct wps_er_ap *ap = wps_er_ap_get_id(er, ap_id);
1155         struct wpabuf *event;
1156         enum http_reply_code ret;
1157
1158         if (ap == NULL) {
1159                 wpa_printf(MSG_DEBUG, "WPS ER: HTTP event from unknown AP id "
1160                            "%u", ap_id);
1161                 wps_er_http_resp_not_found(req);
1162                 return;
1163         }
1164         wpa_printf(MSG_MSGDUMP, "WPS ER: HTTP event from AP id %u: %s",
1165                    ap_id, http_request_get_data(req));
1166
1167         event = xml_get_base64_item(http_request_get_data(req), "WLANEvent",
1168                                     &ret);
1169         if (event == NULL) {
1170                 wpa_printf(MSG_DEBUG, "WPS ER: Could not extract WLANEvent "
1171                            "from the event notification");
1172                 /*
1173                  * Reply with OK anyway to avoid getting unregistered from
1174                  * events.
1175                  */
1176                 wps_er_http_resp_ok(req);
1177                 return;
1178         }
1179
1180         wps_er_process_wlanevent(ap, event);
1181
1182         wpabuf_free(event);
1183         wps_er_http_resp_ok(req);
1184 }
1185
1186
1187 static void wps_er_http_notify(struct wps_er *er, struct http_request *req)
1188 {
1189         char *uri = http_request_get_uri(req);
1190
1191         if (os_strncmp(uri, "/event/", 7) == 0) {
1192                 unsigned int event_id;
1193                 char *pos;
1194                 event_id = atoi(uri + 7);
1195                 if (event_id != er->event_id) {
1196                         wpa_printf(MSG_DEBUG, "WPS ER: HTTP event for an "
1197                                    "unknown event id %u", event_id);
1198                         return;
1199                 }
1200                 pos = os_strchr(uri + 7, '/');
1201                 if (pos == NULL)
1202                         return;
1203                 pos++;
1204                 wps_er_http_event(er, req, atoi(pos));
1205         } else {
1206                 wpa_printf(MSG_DEBUG, "WPS ER: Unknown HTTP NOTIFY for '%s'",
1207                            uri);
1208                 wps_er_http_resp_not_found(req);
1209         }
1210 }
1211
1212
1213 static void wps_er_http_req(void *ctx, struct http_request *req)
1214 {
1215         struct wps_er *er = ctx;
1216         struct sockaddr_in *cli = http_request_get_cli_addr(req);
1217         enum httpread_hdr_type type = http_request_get_type(req);
1218         struct wpabuf *buf;
1219
1220         wpa_printf(MSG_DEBUG, "WPS ER: HTTP request: '%s' (type %d) from "
1221                    "%s:%d",
1222                    http_request_get_uri(req), type,
1223                    inet_ntoa(cli->sin_addr), ntohs(cli->sin_port));
1224
1225         switch (type) {
1226         case HTTPREAD_HDR_TYPE_NOTIFY:
1227                 wps_er_http_notify(er, req);
1228                 break;
1229         default:
1230                 wpa_printf(MSG_DEBUG, "WPS ER: Unsupported HTTP request type "
1231                            "%d", type);
1232                 buf = wpabuf_alloc(200);
1233                 if (buf == NULL) {
1234                         http_request_deinit(req);
1235                         return;
1236                 }
1237                 wpabuf_put_str(buf,
1238                                "HTTP/1.1 501 Unimplemented\r\n"
1239                                "Connection: close\r\n");
1240                 http_put_date(buf);
1241                 wpabuf_put_str(buf, "\r\n");
1242                 http_request_send_and_deinit(req, buf);
1243                 break;
1244         }
1245 }
1246
1247
1248 struct wps_er *
1249 wps_er_init(struct wps_context *wps, const char *ifname, const char *filter)
1250 {
1251         struct wps_er *er;
1252         struct in_addr addr;
1253
1254         er = os_zalloc(sizeof(*er));
1255         if (er == NULL)
1256                 return NULL;
1257         dl_list_init(&er->ap);
1258         dl_list_init(&er->ap_unsubscribing);
1259         dl_list_init(&er->ap_settings);
1260
1261         er->multicast_sd = -1;
1262         er->ssdp_sd = -1;
1263
1264         os_strlcpy(er->ifname, ifname, sizeof(er->ifname));
1265         er->wps = wps;
1266         if (os_get_random((unsigned char *) &er->event_id,
1267                           sizeof(er->event_id)) < 0) {
1268                 wps_er_deinit(er, NULL, NULL);
1269                 return NULL;
1270         }
1271         /* Limit event_id to < 32 bits to avoid issues with atoi() */
1272         er->event_id &= 0x0fffffff;
1273
1274         if (filter && os_strncmp(filter, "ifname=", 7) == 0) {
1275                 const char *pos, *end;
1276                 pos = filter + 7;
1277                 end = os_strchr(pos, ' ');
1278                 if (end) {
1279                         size_t len = end - pos;
1280                         os_strlcpy(er->ifname, pos, len < sizeof(er->ifname) ?
1281                                    len + 1 : sizeof(er->ifname));
1282                         filter = end + 1;
1283                 } else {
1284                         os_strlcpy(er->ifname, pos, sizeof(er->ifname));
1285                         filter = NULL;
1286                 }
1287                 er->forced_ifname = 1;
1288         }
1289
1290         if (filter) {
1291                 if (inet_aton(filter, &er->filter_addr) == 0) {
1292                         wpa_printf(MSG_INFO, "WPS UPnP: Invalid filter "
1293                                    "address %s", filter);
1294                         wps_er_deinit(er, NULL, NULL);
1295                         return NULL;
1296                 }
1297                 wpa_printf(MSG_DEBUG, "WPS UPnP: Only accepting connections "
1298                            "with %s", filter);
1299         }
1300         if (get_netif_info(er->ifname, &er->ip_addr, &er->ip_addr_text,
1301                            er->mac_addr)) {
1302                 wpa_printf(MSG_INFO, "WPS UPnP: Could not get IP/MAC address "
1303                            "for %s. Does it have IP address?", er->ifname);
1304                 wps_er_deinit(er, NULL, NULL);
1305                 return NULL;
1306         }
1307
1308         if (wps_er_ssdp_init(er) < 0) {
1309                 wpa_printf(MSG_INFO, "WPS UPnP: SSDP initialization failed");
1310                 wps_er_deinit(er, NULL, NULL);
1311                 return NULL;
1312         }
1313
1314         addr.s_addr = er->ip_addr;
1315         er->http_srv = http_server_init(&addr, -1, wps_er_http_req, er);
1316         if (er->http_srv == NULL) {
1317                 wpa_printf(MSG_INFO, "WPS UPnP: HTTP initialization failed");
1318                 wps_er_deinit(er, NULL, NULL);
1319                 return NULL;
1320         }
1321         er->http_port = http_server_get_port(er->http_srv);
1322
1323         wpa_printf(MSG_DEBUG, "WPS ER: Start (ifname=%s ip_addr=%s)",
1324                    er->ifname, er->ip_addr_text);
1325
1326         return er;
1327 }
1328
1329
1330 void wps_er_refresh(struct wps_er *er)
1331 {
1332         struct wps_er_ap *ap;
1333         struct wps_er_sta *sta;
1334
1335         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
1336                 wps_er_ap_event(er->wps, ap, WPS_EV_ER_AP_ADD);
1337                 dl_list_for_each(sta, &ap->sta, struct wps_er_sta, list)
1338                         wps_er_sta_event(er->wps, sta, WPS_EV_ER_ENROLLEE_ADD);
1339         }
1340
1341         wps_er_send_ssdp_msearch(er);
1342 }
1343
1344
1345 static void wps_er_deinit_finish(void *eloop_data, void *user_ctx)
1346 {
1347         struct wps_er *er = eloop_data;
1348         void (*deinit_done_cb)(void *ctx);
1349         void *deinit_done_ctx;
1350         struct wps_er_ap *ap, *tmp;
1351
1352         wpa_printf(MSG_DEBUG, "WPS ER: Finishing deinit");
1353
1354         dl_list_for_each_safe(ap, tmp, &er->ap_unsubscribing, struct wps_er_ap,
1355                               list) {
1356                 wpa_printf(MSG_DEBUG, "WPS ER: AP entry for %s (%s) still in ap_unsubscribing list - free it",
1357                            inet_ntoa(ap->addr), ap->location);
1358                 dl_list_del(&ap->list);
1359                 wps_er_ap_free(ap);
1360         }
1361
1362         eloop_cancel_timeout(wps_er_deinit_finish, er, NULL);
1363         deinit_done_cb = er->deinit_done_cb;
1364         deinit_done_ctx = er->deinit_done_ctx;
1365         os_free(er->ip_addr_text);
1366         os_free(er);
1367
1368         if (deinit_done_cb)
1369                 deinit_done_cb(deinit_done_ctx);
1370 }
1371
1372
1373 void wps_er_deinit(struct wps_er *er, void (*cb)(void *ctx), void *ctx)
1374 {
1375         if (er == NULL)
1376                 return;
1377         http_server_deinit(er->http_srv);
1378         wps_er_ap_remove_all(er);
1379         wps_er_ssdp_deinit(er);
1380         eloop_register_timeout(dl_list_empty(&er->ap_unsubscribing) ? 0 : 5, 0,
1381                                wps_er_deinit_finish, er, NULL);
1382         wpa_printf(MSG_DEBUG, "WPS ER: Finish deinit from timeout");
1383         er->deinitializing = 1;
1384         er->deinit_done_cb = cb;
1385         er->deinit_done_ctx = ctx;
1386 }
1387
1388
1389 static void wps_er_http_set_sel_reg_cb(void *ctx, struct http_client *c,
1390                                        enum http_client_event event)
1391 {
1392         struct wps_er_ap *ap = ctx;
1393         union wps_event_data data;
1394
1395         os_memset(&data, 0, sizeof(data));
1396
1397         switch (event) {
1398         case HTTP_CLIENT_OK:
1399                 wpa_printf(MSG_DEBUG, "WPS ER: SetSelectedRegistrar OK");
1400                 data.set_sel_reg.state = WPS_ER_SET_SEL_REG_DONE;
1401                 data.set_sel_reg.uuid = ap->uuid;
1402                 break;
1403         case HTTP_CLIENT_FAILED:
1404         case HTTP_CLIENT_INVALID_REPLY:
1405         case HTTP_CLIENT_TIMEOUT:
1406                 wpa_printf(MSG_DEBUG, "WPS ER: SetSelectedRegistrar failed");
1407                 data.set_sel_reg.state = WPS_ER_SET_SEL_REG_FAILED;
1408                 data.set_sel_reg.uuid = ap->uuid;
1409                 break;
1410         }
1411         http_client_free(ap->http);
1412         ap->http = NULL;
1413
1414         if (data.set_sel_reg.uuid)
1415                 ap->er->wps->event_cb(ap->er->wps->cb_ctx,
1416                                       WPS_EV_ER_SET_SELECTED_REGISTRAR, &data);
1417 }
1418
1419
1420 static void wps_er_send_set_sel_reg(struct wps_er_ap *ap, struct wpabuf *msg)
1421 {
1422         struct wpabuf *buf;
1423         char *len_ptr, *body_ptr;
1424         struct sockaddr_in dst;
1425         char *url, *path;
1426
1427         if (ap->control_url == NULL) {
1428                 wpa_printf(MSG_DEBUG, "WPS ER: No controlURL for AP");
1429                 return;
1430         }
1431
1432         if (ap->http) {
1433                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP request for AP - "
1434                            "ignore new request");
1435                 return;
1436         }
1437
1438         if (ap->wps) {
1439                 wpa_printf(MSG_DEBUG, "WPS ER: Pending WPS operation for AP - "
1440                            "skip SetSelectedRegistrar");
1441                 return;
1442         }
1443
1444         url = http_client_url_parse(ap->control_url, &dst, &path);
1445         if (url == NULL) {
1446                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse controlURL");
1447                 return;
1448         }
1449
1450         buf = wps_er_soap_hdr(msg, "SetSelectedRegistrar", "NewMessage", path,
1451                               &dst, &len_ptr, &body_ptr);
1452         os_free(url);
1453         if (buf == NULL)
1454                 return;
1455
1456         wps_er_soap_end(buf, "SetSelectedRegistrar", len_ptr, body_ptr);
1457
1458         ap->http = http_client_addr(&dst, buf, 1000,
1459                                     wps_er_http_set_sel_reg_cb, ap);
1460         if (ap->http == NULL)
1461                 wpabuf_free(buf);
1462 }
1463
1464
1465 static int wps_er_build_selected_registrar(struct wpabuf *msg, int sel_reg)
1466 {
1467         wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR);
1468         wpabuf_put_be16(msg, 1);
1469         wpabuf_put_u8(msg, !!sel_reg);
1470         return 0;
1471 }
1472
1473
1474 static int wps_er_build_dev_password_id(struct wpabuf *msg, u16 dev_passwd_id)
1475 {
1476         wpabuf_put_be16(msg, ATTR_DEV_PASSWORD_ID);
1477         wpabuf_put_be16(msg, 2);
1478         wpabuf_put_be16(msg, dev_passwd_id);
1479         return 0;
1480 }
1481
1482
1483 static int wps_er_build_sel_reg_config_methods(struct wpabuf *msg,
1484                                                u16 sel_reg_config_methods)
1485 {
1486         wpabuf_put_be16(msg, ATTR_SELECTED_REGISTRAR_CONFIG_METHODS);
1487         wpabuf_put_be16(msg, 2);
1488         wpabuf_put_be16(msg, sel_reg_config_methods);
1489         return 0;
1490 }
1491
1492
1493 static int wps_er_build_uuid_r(struct wpabuf *msg, const u8 *uuid_r)
1494 {
1495         wpabuf_put_be16(msg, ATTR_UUID_R);
1496         wpabuf_put_be16(msg, WPS_UUID_LEN);
1497         wpabuf_put_data(msg, uuid_r, WPS_UUID_LEN);
1498         return 0;
1499 }
1500
1501
1502 void wps_er_set_sel_reg(struct wps_er *er, int sel_reg, u16 dev_passwd_id,
1503                         u16 sel_reg_config_methods)
1504 {
1505         struct wpabuf *msg;
1506         struct wps_er_ap *ap;
1507         struct wps_registrar *reg = er->wps->registrar;
1508         const u8 *auth_macs;
1509         u8 bcast[ETH_ALEN];
1510         size_t count;
1511         union wps_event_data data;
1512
1513         if (er->skip_set_sel_reg) {
1514                 wpa_printf(MSG_DEBUG, "WPS ER: Skip SetSelectedRegistrar");
1515                 return;
1516         }
1517
1518         msg = wpabuf_alloc(500);
1519         if (msg == NULL)
1520                 return;
1521
1522         auth_macs = wps_authorized_macs(reg, &count);
1523         if (count == 0) {
1524                 os_memset(bcast, 0xff, ETH_ALEN);
1525                 auth_macs = bcast;
1526                 count = 1;
1527         }
1528
1529         if (wps_build_version(msg) ||
1530             wps_er_build_selected_registrar(msg, sel_reg) ||
1531             wps_er_build_dev_password_id(msg, dev_passwd_id) ||
1532             wps_er_build_sel_reg_config_methods(msg, sel_reg_config_methods) ||
1533             wps_build_wfa_ext(msg, 0, auth_macs, count, 0) ||
1534             wps_er_build_uuid_r(msg, er->wps->uuid)) {
1535                 wpabuf_free(msg);
1536                 return;
1537         }
1538
1539         os_memset(&data, 0, sizeof(data));
1540         data.set_sel_reg.sel_reg = sel_reg;
1541         data.set_sel_reg.dev_passwd_id = dev_passwd_id;
1542         data.set_sel_reg.sel_reg_config_methods = sel_reg_config_methods;
1543         data.set_sel_reg.state = WPS_ER_SET_SEL_REG_START;
1544
1545         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
1546                 if (er->set_sel_reg_uuid_filter &&
1547                     os_memcmp(ap->uuid, er->set_sel_reg_uuid_filter,
1548                               WPS_UUID_LEN) != 0)
1549                         continue;
1550                 data.set_sel_reg.uuid = ap->uuid;
1551                 er->wps->event_cb(er->wps->cb_ctx,
1552                                   WPS_EV_ER_SET_SELECTED_REGISTRAR, &data);
1553                 wps_er_send_set_sel_reg(ap, msg);
1554         }
1555
1556         wpabuf_free(msg);
1557 }
1558
1559
1560 int wps_er_pbc(struct wps_er *er, const u8 *uuid, const u8 *addr)
1561 {
1562         int res;
1563         struct wps_er_ap *ap;
1564
1565         if (er == NULL || er->wps == NULL)
1566                 return -1;
1567
1568         if (wps_registrar_pbc_overlap(er->wps->registrar, NULL, NULL)) {
1569                 wpa_printf(MSG_DEBUG, "WPS ER: PBC overlap - do not start PBC "
1570                            "mode");
1571                 return -2;
1572         }
1573
1574         if (uuid)
1575                 ap = wps_er_ap_get(er, NULL, uuid, NULL);
1576         else
1577                 ap = NULL;
1578         if (ap == NULL) {
1579                 struct wps_er_sta *sta = NULL;
1580                 dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
1581                         sta = wps_er_sta_get(ap, addr, uuid);
1582                         if (sta) {
1583                                 uuid = ap->uuid;
1584                                 break;
1585                         }
1586                 }
1587                 if (sta == NULL)
1588                         return -3; /* Unknown UUID */
1589         }
1590
1591         if (ap->ap_settings == NULL) {
1592                 wpa_printf(MSG_DEBUG, "WPS ER: AP settings not known");
1593                 return -4;
1594         }
1595
1596         er->set_sel_reg_uuid_filter = uuid;
1597         res = wps_registrar_button_pushed(er->wps->registrar, NULL);
1598         er->set_sel_reg_uuid_filter = NULL;
1599         if (res)
1600                 return -1;
1601
1602         return 0;
1603 }
1604
1605
1606 static void wps_er_ap_settings_cb(void *ctx, const struct wps_credential *cred)
1607 {
1608         struct wps_er_ap *ap = ctx;
1609         union wps_event_data data;
1610
1611         wpa_printf(MSG_DEBUG, "WPS ER: AP Settings received");
1612         os_free(ap->ap_settings);
1613         ap->ap_settings = os_malloc(sizeof(*cred));
1614         if (ap->ap_settings) {
1615                 os_memcpy(ap->ap_settings, cred, sizeof(*cred));
1616                 ap->ap_settings->cred_attr = NULL;
1617         }
1618
1619         os_memset(&data, 0, sizeof(data));
1620         data.ap_settings.uuid = ap->uuid;
1621         data.ap_settings.cred = cred;
1622         ap->er->wps->event_cb(ap->er->wps->cb_ctx, WPS_EV_ER_AP_SETTINGS,
1623                               &data);
1624 }
1625
1626
1627 const u8 * wps_er_get_sta_uuid(struct wps_er *er, const u8 *addr)
1628 {
1629         struct wps_er_ap *ap;
1630         dl_list_for_each(ap, &er->ap, struct wps_er_ap, list) {
1631                 struct wps_er_sta *sta;
1632                 sta = wps_er_sta_get(ap, addr, NULL);
1633                 if (sta)
1634                         return sta->uuid;
1635         }
1636         return NULL;
1637 }
1638
1639
1640 static void wps_er_http_put_message_cb(void *ctx, struct http_client *c,
1641                                        enum http_client_event event)
1642 {
1643         struct wps_er_ap *ap = ctx;
1644         struct wpabuf *reply;
1645         char *msg = NULL;
1646
1647         switch (event) {
1648         case HTTP_CLIENT_OK:
1649                 wpa_printf(MSG_DEBUG, "WPS ER: PutMessage OK");
1650                 reply = http_client_get_body(c);
1651                 if (reply)
1652                         msg = os_zalloc(wpabuf_len(reply) + 1);
1653                 if (msg == NULL) {
1654                         if (ap->wps) {
1655                                 wps_deinit(ap->wps);
1656                                 ap->wps = NULL;
1657                         }
1658                         break;
1659                 }
1660                 os_memcpy(msg, wpabuf_head(reply), wpabuf_len(reply));
1661                 break;
1662         case HTTP_CLIENT_FAILED:
1663         case HTTP_CLIENT_INVALID_REPLY:
1664         case HTTP_CLIENT_TIMEOUT:
1665                 wpa_printf(MSG_DEBUG, "WPS ER: PutMessage failed");
1666                 if (ap->wps) {
1667                         wps_deinit(ap->wps);
1668                         ap->wps = NULL;
1669                 }
1670                 break;
1671         }
1672         http_client_free(ap->http);
1673         ap->http = NULL;
1674
1675         if (msg) {
1676                 struct wpabuf *buf;
1677                 enum http_reply_code ret;
1678                 buf = xml_get_base64_item(msg, "NewOutMessage", &ret);
1679                 os_free(msg);
1680                 if (buf == NULL) {
1681                         wpa_printf(MSG_DEBUG, "WPS ER: Could not extract "
1682                                    "NewOutMessage from PutMessage response");
1683                         wps_deinit(ap->wps);
1684                         ap->wps = NULL;
1685                         return;
1686                 }
1687                 wps_er_ap_process(ap, buf);
1688                 wpabuf_free(buf);
1689         }
1690 }
1691
1692
1693 static void wps_er_ap_put_message(struct wps_er_ap *ap,
1694                                   const struct wpabuf *msg)
1695 {
1696         struct wpabuf *buf;
1697         char *len_ptr, *body_ptr;
1698         struct sockaddr_in dst;
1699         char *url, *path;
1700
1701         if (ap->http) {
1702                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP operation ongoing "
1703                            "with the AP - cannot continue learn");
1704                 return;
1705         }
1706
1707         if (ap->control_url == NULL) {
1708                 wpa_printf(MSG_DEBUG, "WPS ER: No controlURL for AP");
1709                 return;
1710         }
1711
1712         url = http_client_url_parse(ap->control_url, &dst, &path);
1713         if (url == NULL) {
1714                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse controlURL");
1715                 goto fail;
1716         }
1717
1718         buf = wps_er_soap_hdr(msg, "PutMessage", "NewInMessage", path, &dst,
1719                               &len_ptr, &body_ptr);
1720         os_free(url);
1721         if (buf == NULL)
1722                 goto fail;
1723
1724         wps_er_soap_end(buf, "PutMessage", len_ptr, body_ptr);
1725
1726         ap->http = http_client_addr(&dst, buf, 10000,
1727                                     wps_er_http_put_message_cb, ap);
1728         if (ap->http == NULL) {
1729                 wpabuf_free(buf);
1730                 goto fail;
1731         }
1732         return;
1733
1734 fail:
1735         if (ap->wps) {
1736                 wps_deinit(ap->wps);
1737                 ap->wps = NULL;
1738         }
1739 }
1740
1741
1742 static void wps_er_ap_process(struct wps_er_ap *ap, struct wpabuf *msg)
1743 {
1744         enum wps_process_res res;
1745         struct wps_parse_attr attr;
1746         enum wsc_op_code op_code;
1747
1748         op_code = WSC_MSG;
1749         if (wps_parse_msg(msg, &attr) == 0 && attr.msg_type) {
1750                 switch (*attr.msg_type) {
1751                 case WPS_WSC_ACK:
1752                         op_code = WSC_ACK;
1753                         break;
1754                 case WPS_WSC_NACK:
1755                         op_code = WSC_NACK;
1756                         break;
1757                 case WPS_WSC_DONE:
1758                         op_code = WSC_Done;
1759                         break;
1760                 }
1761         }
1762
1763         res = wps_process_msg(ap->wps, op_code, msg);
1764         if (res == WPS_CONTINUE) {
1765                 struct wpabuf *next = wps_get_msg(ap->wps, &op_code);
1766                 if (next) {
1767                         wps_er_ap_put_message(ap, next);
1768                         wpabuf_free(next);
1769                 } else {
1770                         wpa_printf(MSG_DEBUG, "WPS ER: Failed to build "
1771                                    "message");
1772                         wps_deinit(ap->wps);
1773                         ap->wps = NULL;
1774                 }
1775         } else if (res == WPS_DONE) {
1776                 wpa_printf(MSG_DEBUG, "WPS ER: Protocol run done");
1777                 wps_deinit(ap->wps);
1778                 ap->wps = NULL;
1779         } else {
1780                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to process message from "
1781                            "AP (res=%d)", res);
1782                 wps_deinit(ap->wps);
1783                 ap->wps = NULL;
1784         }
1785 }
1786
1787
1788 static void wps_er_ap_learn_m1(struct wps_er_ap *ap, struct wpabuf *m1)
1789 {
1790         struct wps_config cfg;
1791
1792         if (ap->wps) {
1793                 wpa_printf(MSG_DEBUG, "WPS ER: Protocol run already in "
1794                            "progress with this AP");
1795                 return;
1796         }
1797
1798         os_memset(&cfg, 0, sizeof(cfg));
1799         cfg.wps = ap->er->wps;
1800         cfg.registrar = 1;
1801         ap->wps = wps_init(&cfg);
1802         if (ap->wps == NULL)
1803                 return;
1804         ap->wps->ap_settings_cb = wps_er_ap_settings_cb;
1805         ap->wps->ap_settings_cb_ctx = ap;
1806
1807         wps_er_ap_process(ap, m1);
1808 }
1809
1810
1811 static void wps_er_ap_learn(struct wps_er_ap *ap, const char *dev_info)
1812 {
1813         struct wpabuf *info;
1814         enum http_reply_code ret;
1815
1816         wpa_printf(MSG_DEBUG, "WPS ER: Received GetDeviceInfo response (M1) "
1817                    "from the AP");
1818         info = xml_get_base64_item(dev_info, "NewDeviceInfo", &ret);
1819         if (info == NULL) {
1820                 wpa_printf(MSG_DEBUG, "WPS ER: Could not extract "
1821                            "NewDeviceInfo from GetDeviceInfo response");
1822                 return;
1823         }
1824
1825         ap->m1_handler(ap, info);
1826         wpabuf_free(info);
1827 }
1828
1829
1830 static void wps_er_http_get_dev_info_cb(void *ctx, struct http_client *c,
1831                                         enum http_client_event event)
1832 {
1833         struct wps_er_ap *ap = ctx;
1834         struct wpabuf *reply;
1835         char *dev_info = NULL;
1836
1837         switch (event) {
1838         case HTTP_CLIENT_OK:
1839                 wpa_printf(MSG_DEBUG, "WPS ER: GetDeviceInfo OK");
1840                 reply = http_client_get_body(c);
1841                 if (reply == NULL)
1842                         break;
1843                 dev_info = os_zalloc(wpabuf_len(reply) + 1);
1844                 if (dev_info == NULL)
1845                         break;
1846                 os_memcpy(dev_info, wpabuf_head(reply), wpabuf_len(reply));
1847                 break;
1848         case HTTP_CLIENT_FAILED:
1849         case HTTP_CLIENT_INVALID_REPLY:
1850         case HTTP_CLIENT_TIMEOUT:
1851                 wpa_printf(MSG_DEBUG, "WPS ER: GetDeviceInfo failed");
1852                 break;
1853         }
1854         http_client_free(ap->http);
1855         ap->http = NULL;
1856
1857         if (dev_info) {
1858                 wps_er_ap_learn(ap, dev_info);
1859                 os_free(dev_info);
1860         }
1861 }
1862
1863
1864 static int wps_er_send_get_device_info(struct wps_er_ap *ap,
1865                                        void (*m1_handler)(struct wps_er_ap *ap,
1866                                                           struct wpabuf *m1))
1867 {
1868         struct wpabuf *buf;
1869         char *len_ptr, *body_ptr;
1870         struct sockaddr_in dst;
1871         char *url, *path;
1872
1873         if (ap->http) {
1874                 wpa_printf(MSG_DEBUG, "WPS ER: Pending HTTP operation ongoing "
1875                            "with the AP - cannot get device info");
1876                 return -1;
1877         }
1878
1879         if (ap->control_url == NULL) {
1880                 wpa_printf(MSG_DEBUG, "WPS ER: No controlURL for AP");
1881                 return -1;
1882         }
1883
1884         url = http_client_url_parse(ap->control_url, &dst, &path);
1885         if (url == NULL) {
1886                 wpa_printf(MSG_DEBUG, "WPS ER: Failed to parse controlURL");
1887                 return -1;
1888         }
1889
1890         buf = wps_er_soap_hdr(NULL, "GetDeviceInfo", NULL, path, &dst,
1891                               &len_ptr, &body_ptr);
1892         os_free(url);
1893         if (buf == NULL)
1894                 return -1;
1895
1896         wps_er_soap_end(buf, "GetDeviceInfo", len_ptr, body_ptr);
1897
1898         ap->http = http_client_addr(&dst, buf, 10000,
1899                                     wps_er_http_get_dev_info_cb, ap);
1900         if (ap->http == NULL) {
1901                 wpabuf_free(buf);
1902                 return -1;
1903         }
1904
1905         ap->m1_handler = m1_handler;
1906
1907         return 0;
1908 }
1909
1910
1911 int wps_er_learn(struct wps_er *er, const u8 *uuid, const u8 *addr,
1912                  const u8 *pin, size_t pin_len)
1913 {
1914         struct wps_er_ap *ap;
1915
1916         if (er == NULL)
1917                 return -1;
1918
1919         ap = wps_er_ap_get(er, NULL, uuid, addr);
1920         if (ap == NULL) {
1921                 wpa_printf(MSG_DEBUG, "WPS ER: AP not found for learn "
1922                            "request");
1923                 return -1;
1924         }
1925         if (uuid == NULL)
1926                 uuid = ap->uuid;
1927         if (ap->wps) {
1928                 wpa_printf(MSG_DEBUG, "WPS ER: Pending operation ongoing "
1929                            "with the AP - cannot start learn");
1930                 return -1;
1931         }
1932
1933         if (wps_er_send_get_device_info(ap, wps_er_ap_learn_m1) < 0)
1934                 return -1;
1935
1936         er->skip_set_sel_reg = 1;
1937         wps_registrar_add_pin(er->wps->registrar, NULL, uuid, pin, pin_len, 0);
1938         er->skip_set_sel_reg = 0;
1939
1940         return 0;
1941 }
1942
1943
1944 int wps_er_set_config(struct wps_er *er, const u8 *uuid, const u8 *addr,
1945                       const struct wps_credential *cred)
1946 {
1947         struct wps_er_ap *ap;
1948
1949         if (er == NULL)
1950                 return -1;
1951
1952         ap = wps_er_ap_get(er, NULL, uuid, addr);
1953         if (ap == NULL) {
1954                 wpa_printf(MSG_DEBUG, "WPS ER: AP not found for set config "
1955                            "request");
1956                 return -1;
1957         }
1958
1959         os_free(ap->ap_settings);
1960         ap->ap_settings = os_memdup(cred, sizeof(*cred));
1961         if (ap->ap_settings == NULL)
1962                 return -1;
1963         ap->ap_settings->cred_attr = NULL;
1964         wpa_printf(MSG_DEBUG, "WPS ER: Updated local AP settings based set "
1965                    "config request");
1966
1967         return 0;
1968 }
1969
1970
1971 static void wps_er_ap_config_m1(struct wps_er_ap *ap, struct wpabuf *m1)
1972 {
1973         struct wps_config cfg;
1974
1975         if (ap->wps) {
1976                 wpa_printf(MSG_DEBUG, "WPS ER: Protocol run already in "
1977                            "progress with this AP");
1978                 return;
1979         }
1980
1981         os_memset(&cfg, 0, sizeof(cfg));
1982         cfg.wps = ap->er->wps;
1983         cfg.registrar = 1;
1984         cfg.new_ap_settings = ap->ap_settings;
1985         ap->wps = wps_init(&cfg);
1986         if (ap->wps == NULL)
1987                 return;
1988         ap->wps->ap_settings_cb = NULL;
1989         ap->wps->ap_settings_cb_ctx = NULL;
1990
1991         wps_er_ap_process(ap, m1);
1992 }
1993
1994
1995 int wps_er_config(struct wps_er *er, const u8 *uuid, const u8 *addr,
1996                   const u8 *pin, size_t pin_len,
1997                   const struct wps_credential *cred)
1998 {
1999         struct wps_er_ap *ap;
2000
2001         if (er == NULL)
2002                 return -1;
2003
2004         ap = wps_er_ap_get(er, NULL, uuid, addr);
2005         if (ap == NULL) {
2006                 wpa_printf(MSG_DEBUG, "WPS ER: AP not found for config "
2007                            "request");
2008                 return -1;
2009         }
2010         if (uuid == NULL)
2011                 uuid = ap->uuid;
2012         if (ap->wps) {
2013                 wpa_printf(MSG_DEBUG, "WPS ER: Pending operation ongoing "
2014                            "with the AP - cannot start config");
2015                 return -1;
2016         }
2017
2018         os_free(ap->ap_settings);
2019         ap->ap_settings = os_memdup(cred, sizeof(*cred));
2020         if (ap->ap_settings == NULL)
2021                 return -1;
2022         ap->ap_settings->cred_attr = NULL;
2023
2024         if (wps_er_send_get_device_info(ap, wps_er_ap_config_m1) < 0)
2025                 return -1;
2026
2027         er->skip_set_sel_reg = 1;
2028         wps_registrar_add_pin(er->wps->registrar, NULL, uuid, pin, pin_len, 0);
2029         er->skip_set_sel_reg = 0;
2030
2031         return 0;
2032 }
2033
2034
2035 #ifdef CONFIG_WPS_NFC
2036
2037 struct wpabuf * wps_er_config_token_from_cred(struct wps_context *wps,
2038                                               struct wps_credential *cred)
2039 {
2040         struct wpabuf *ret;
2041         struct wps_data data;
2042
2043         ret = wpabuf_alloc(500);
2044         if (ret == NULL)
2045                 return NULL;
2046
2047         os_memset(&data, 0, sizeof(data));
2048         data.wps = wps;
2049         data.use_cred = cred;
2050         if (wps_build_cred(&data, ret) ||
2051             wps_build_wfa_ext(ret, 0, NULL, 0, 0)) {
2052                 wpabuf_free(ret);
2053                 return NULL;
2054         }
2055
2056         return ret;
2057 }
2058
2059
2060 struct wpabuf * wps_er_nfc_config_token(struct wps_er *er, const u8 *uuid,
2061                                         const u8 *addr)
2062 {
2063         struct wps_er_ap *ap;
2064
2065         if (er == NULL)
2066                 return NULL;
2067
2068         ap = wps_er_ap_get(er, NULL, uuid, addr);
2069         if (ap == NULL)
2070                 return NULL;
2071         if (ap->ap_settings == NULL) {
2072                 wpa_printf(MSG_DEBUG, "WPS ER: No settings known for the "
2073                            "selected AP");
2074                 return NULL;
2075         }
2076
2077         return wps_er_config_token_from_cred(er->wps, ap->ap_settings);
2078 }
2079
2080
2081 struct wpabuf * wps_er_nfc_handover_sel(struct wps_er *er,
2082                                         struct wps_context *wps, const u8 *uuid,
2083                                         const u8 *addr, struct wpabuf *pubkey)
2084 {
2085         struct wps_er_ap *ap;
2086
2087         if (er == NULL)
2088                 return NULL;
2089
2090         ap = wps_er_ap_get(er, NULL, uuid, addr);
2091         if (ap == NULL)
2092                 return NULL;
2093         if (ap->ap_settings == NULL) {
2094                 wpa_printf(MSG_DEBUG, "WPS ER: No settings known for the "
2095                            "selected AP");
2096                 return NULL;
2097         }
2098
2099         os_memcpy(wps->ssid, ap->ap_settings->ssid, ap->ap_settings->ssid_len);
2100         wps->ssid_len = ap->ap_settings->ssid_len;
2101
2102         return wps_build_nfc_handover_sel(wps, pubkey, addr, 0);
2103 }
2104
2105 #endif /* CONFIG_WPS_NFC */