8b1d5c4bde7c7d7389e24196042199300c2e3998
[libeap.git] / wpa_supplicant / p2p_supplicant.c
1 /*
2  * wpa_supplicant - P2P
3  * Copyright (c) 2009-2010, Atheros Communications
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "includes.h"
16
17 #include "common.h"
18 #include "eloop.h"
19 #include "common/ieee802_11_common.h"
20 #include "common/ieee802_11_defs.h"
21 #include "common/wpa_ctrl.h"
22 #include "wps/wps_i.h"
23 #include "p2p/p2p.h"
24 #include "ap/hostapd.h"
25 #include "ap/p2p_hostapd.h"
26 #include "wpa_supplicant_i.h"
27 #include "driver_i.h"
28 #include "ap.h"
29 #include "config_ssid.h"
30 #include "config.h"
31 #include "mlme.h"
32 #include "notify.h"
33 #include "scan.h"
34 #include "bss.h"
35 #include "wps_supplicant.h"
36 #include "p2p_supplicant.h"
37
38
39 static void wpas_p2p_long_listen_timeout(void *eloop_ctx, void *timeout_ctx);
40 static struct wpa_supplicant *
41 wpas_p2p_get_group_iface(struct wpa_supplicant *wpa_s, int addr_allocated,
42                          int go);
43 static int wpas_p2p_join_start(struct wpa_supplicant *wpa_s);
44 static void wpas_p2p_join_scan(void *eloop_ctx, void *timeout_ctx);
45 static int wpas_p2p_join(struct wpa_supplicant *wpa_s, const u8 *iface_addr,
46                          const u8 *dev_addr, enum p2p_wps_method wps_method);
47 static int wpas_p2p_create_iface(struct wpa_supplicant *wpa_s);
48 static void wpas_p2p_cross_connect_setup(struct wpa_supplicant *wpa_s);
49
50
51 static void wpas_p2p_scan_res_handler(struct wpa_supplicant *wpa_s,
52                                       struct wpa_scan_results *scan_res)
53 {
54         size_t i;
55
56         if (wpa_s->global->p2p_disabled)
57                 return;
58
59         wpa_printf(MSG_DEBUG, "P2P: Scan results received (%d BSS)",
60                    (int) scan_res->num);
61
62         for (i = 0; i < scan_res->num; i++) {
63                 struct wpa_scan_res *bss = scan_res->res[i];
64                 if (p2p_scan_res_handler(wpa_s->global->p2p, bss->bssid,
65                                          bss->freq, bss->level,
66                                          (const u8 *) (bss + 1),
67                                          bss->ie_len) > 0)
68                         return;
69         }
70
71         p2p_scan_res_handled(wpa_s->global->p2p);
72 }
73
74
75 static int wpas_p2p_scan(void *ctx, enum p2p_scan_type type, int freq)
76 {
77         struct wpa_supplicant *wpa_s = ctx;
78         struct wpa_driver_scan_params params;
79         int ret;
80         struct wpabuf *wps_ie, *ies;
81         int social_channels[] = { 2412, 2437, 2462, 0, 0 };
82
83         os_memset(&params, 0, sizeof(params));
84
85         /* P2P Wildcard SSID */
86         params.num_ssids = 1;
87         params.ssids[0].ssid = (u8 *) P2P_WILDCARD_SSID;
88         params.ssids[0].ssid_len = P2P_WILDCARD_SSID_LEN;
89
90         wpa_s->wps->dev.p2p = 1;
91         wps_ie = wps_build_probe_req_ie(0, &wpa_s->wps->dev, wpa_s->wps->uuid,
92                                         WPS_REQ_ENROLLEE);
93         if (wps_ie == NULL)
94                 return -1;
95
96         ies = wpabuf_alloc(wpabuf_len(wps_ie) + 100);
97         if (ies == NULL) {
98                 wpabuf_free(wps_ie);
99                 return -1;
100         }
101         wpabuf_put_buf(ies, wps_ie);
102         wpabuf_free(wps_ie);
103
104         p2p_scan_ie(wpa_s->global->p2p, ies);
105
106         params.extra_ies = wpabuf_head(ies);
107         params.extra_ies_len = wpabuf_len(ies);
108
109         switch (type) {
110         case P2P_SCAN_SOCIAL:
111                 params.freqs = social_channels;
112                 break;
113         case P2P_SCAN_FULL:
114                 break;
115         case P2P_SCAN_SPECIFIC:
116                 social_channels[0] = freq;
117                 social_channels[1] = 0;
118                 params.freqs = social_channels;
119                 break;
120         case P2P_SCAN_SOCIAL_PLUS_ONE:
121                 social_channels[3] = freq;
122                 params.freqs = social_channels;
123                 break;
124         }
125
126         wpa_s->scan_res_handler = wpas_p2p_scan_res_handler;
127         if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
128                 ret = ieee80211_sta_req_scan(wpa_s, &params);
129         else
130                 ret = wpa_drv_scan(wpa_s, &params);
131
132         wpabuf_free(ies);
133
134         return ret;
135 }
136
137
138 #ifdef CONFIG_CLIENT_MLME
139 static void p2p_rx_action_mlme(void *ctx, const u8 *buf, size_t len, int freq)
140 {
141         struct wpa_supplicant *wpa_s = ctx;
142         const struct ieee80211_mgmt *mgmt;
143         size_t hdr_len;
144
145         if (wpa_s->global->p2p_disabled)
146                 return;
147         mgmt = (const struct ieee80211_mgmt *) buf;
148         hdr_len = (const u8 *) &mgmt->u.action.u.vs_public_action.action - buf;
149         if (hdr_len > len)
150                 return;
151         p2p_rx_action(wpa_s->global->p2p, mgmt->da, mgmt->sa, mgmt->bssid,
152                       mgmt->u.action.category,
153                       &mgmt->u.action.u.vs_public_action.action,
154                       len - hdr_len, freq);
155 }
156 #endif /* CONFIG_CLIENT_MLME */
157
158
159 static enum wpa_driver_if_type wpas_p2p_if_type(int p2p_group_interface)
160 {
161         switch (p2p_group_interface) {
162         case P2P_GROUP_INTERFACE_PENDING:
163                 return WPA_IF_P2P_GROUP;
164         case P2P_GROUP_INTERFACE_GO:
165                 return WPA_IF_P2P_GO;
166         case P2P_GROUP_INTERFACE_CLIENT:
167                 return WPA_IF_P2P_CLIENT;
168         }
169
170         return WPA_IF_P2P_GROUP;
171 }
172
173
174 static void wpas_p2p_group_delete(struct wpa_supplicant *wpa_s)
175 {
176         struct wpa_ssid *ssid;
177         char *gtype;
178
179         ssid = wpa_s->current_ssid;
180         if (ssid == NULL) {
181                 /*
182                  * The current SSID was not known, but there may still be a
183                  * pending P2P group interface waiting for provisioning.
184                  */
185                 ssid = wpa_s->conf->ssid;
186                 while (ssid) {
187                         if (ssid->p2p_group &&
188                             (ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION ||
189                              (ssid->key_mgmt & WPA_KEY_MGMT_WPS)))
190                                 break;
191                         ssid = ssid->next;
192                 }
193         }
194         if (wpa_s->p2p_group_interface == P2P_GROUP_INTERFACE_GO)
195                 gtype = "GO";
196         else if (wpa_s->p2p_group_interface == P2P_GROUP_INTERFACE_CLIENT ||
197                  (ssid && ssid->mode == WPAS_MODE_INFRA)) {
198                 wpa_s->reassociate = 0;
199                 wpa_s->disconnected = 1;
200                 wpa_supplicant_deauthenticate(wpa_s,
201                                               WLAN_REASON_DEAUTH_LEAVING);
202                 gtype = "client";
203         } else
204                 gtype = "GO";
205         if (wpa_s->cross_connect_in_use) {
206                 wpa_s->cross_connect_in_use = 0;
207                 wpa_msg(wpa_s->parent, MSG_INFO,
208                         P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
209                         wpa_s->ifname, wpa_s->cross_connect_uplink);
210         }
211         wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_REMOVED "%s %s",
212                 wpa_s->ifname, gtype);
213         if (wpa_s->p2p_group_interface != NOT_P2P_GROUP_INTERFACE) {
214                 struct wpa_global *global;
215                 char *ifname;
216                 enum wpa_driver_if_type type;
217                 wpa_printf(MSG_DEBUG, "P2P: Remove group interface %s",
218                         wpa_s->ifname);
219                 global = wpa_s->global;
220                 ifname = os_strdup(wpa_s->ifname);
221                 type = wpas_p2p_if_type(wpa_s->p2p_group_interface);
222                 wpa_supplicant_remove_iface(wpa_s->global, wpa_s);
223                 wpa_s = global->ifaces;
224                 if (wpa_s && ifname)
225                         wpa_drv_if_remove(wpa_s, type, ifname);
226                 os_free(ifname);
227                 return;
228         }
229
230         wpa_printf(MSG_DEBUG, "P2P: Remove temporary group network");
231         if (ssid && (ssid->p2p_group ||
232                      ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION ||
233                      (ssid->key_mgmt & WPA_KEY_MGMT_WPS))) {
234                 int id = ssid->id;
235                 if (ssid == wpa_s->current_ssid)
236                         wpa_s->current_ssid = NULL;
237                 wpas_notify_network_removed(wpa_s, ssid);
238                 wpa_config_remove_network(wpa_s->conf, id);
239                 wpa_supplicant_clear_status(wpa_s);
240         } else {
241                 wpa_printf(MSG_DEBUG, "P2P: Temporary group network not "
242                            "found");
243         }
244         wpa_supplicant_ap_deinit(wpa_s);
245 }
246
247
248 static int wpas_p2p_persistent_group(struct wpa_supplicant *wpa_s,
249                                      u8 *go_dev_addr,
250                                      const u8 *ssid, size_t ssid_len)
251 {
252         struct wpa_bss *bss;
253         const u8 *bssid;
254         struct wpabuf *p2p;
255         u8 group_capab;
256         const u8 *addr;
257
258         if (wpa_s->go_params)
259                 bssid = wpa_s->go_params->peer_interface_addr;
260         else
261                 bssid = wpa_s->bssid;
262
263         bss = wpa_bss_get(wpa_s, bssid, ssid, ssid_len);
264         if (bss == NULL) {
265                 u8 iface_addr[ETH_ALEN];
266                 if (p2p_get_interface_addr(wpa_s->global->p2p, bssid,
267                                            iface_addr) == 0)
268                         bss = wpa_bss_get(wpa_s, iface_addr, ssid, ssid_len);
269         }
270         if (bss == NULL) {
271                 wpa_printf(MSG_DEBUG, "P2P: Could not figure out whether "
272                            "group is persistent - BSS " MACSTR " not found",
273                            MAC2STR(bssid));
274                 return 0;
275         }
276
277         p2p = wpa_bss_get_vendor_ie_multi(bss, P2P_IE_VENDOR_TYPE);
278         if (p2p == NULL) {
279                 wpa_printf(MSG_DEBUG, "P2P: Could not figure out whether "
280                            "group is persistent - BSS " MACSTR
281                            " did not include P2P IE", MAC2STR(bssid));
282                 wpa_hexdump(MSG_DEBUG, "P2P: Probe Response IEs",
283                             (u8 *) (bss + 1), bss->ie_len);
284                 wpa_hexdump(MSG_DEBUG, "P2P: Beacon IEs",
285                             ((u8 *) bss + 1) + bss->ie_len,
286                             bss->beacon_ie_len);
287                 return 0;
288         }
289
290         group_capab = p2p_get_group_capab(p2p);
291         addr = p2p_get_go_dev_addr(p2p);
292         wpa_printf(MSG_DEBUG, "P2P: Checking whether group is persistent: "
293                    "group_capab=0x%x", group_capab);
294         if (addr) {
295                 os_memcpy(go_dev_addr, addr, ETH_ALEN);
296                 wpa_printf(MSG_DEBUG, "P2P: GO Device Address " MACSTR,
297                            MAC2STR(addr));
298         } else
299                 os_memset(go_dev_addr, 0, ETH_ALEN);
300         wpabuf_free(p2p);
301
302         wpa_printf(MSG_DEBUG, "P2P: BSS " MACSTR " group_capab=0x%x "
303                    "go_dev_addr=" MACSTR,
304                    MAC2STR(bssid), group_capab, MAC2STR(go_dev_addr));
305
306         return group_capab & P2P_GROUP_CAPAB_PERSISTENT_GROUP;
307 }
308
309
310 static void wpas_p2p_store_persistent_group(struct wpa_supplicant *wpa_s,
311                                             struct wpa_ssid *ssid,
312                                             const u8 *go_dev_addr)
313 {
314         struct wpa_ssid *s;
315         int changed = 0;
316
317         wpa_printf(MSG_DEBUG, "P2P: Storing credentials for a persistent "
318                    "group (GO Dev Addr " MACSTR ")", MAC2STR(go_dev_addr));
319         for (s = wpa_s->conf->ssid; s; s = s->next) {
320                 if (s->disabled == 2 &&
321                     os_memcmp(go_dev_addr, s->bssid, ETH_ALEN) == 0 &&
322                     s->ssid_len == ssid->ssid_len &&
323                     os_memcmp(ssid->ssid, s->ssid, ssid->ssid_len) == 0)
324                         break;
325         }
326
327         if (s) {
328                 wpa_printf(MSG_DEBUG, "P2P: Update existing persistent group "
329                            "entry");
330                 if (ssid->passphrase && !s->passphrase)
331                         changed = 1;
332                 else if (ssid->passphrase && s->passphrase &&
333                          os_strcmp(ssid->passphrase, s->passphrase) != 0)
334                         changed = 1;
335         } else {
336                 wpa_printf(MSG_DEBUG, "P2P: Create a new persistent group "
337                            "entry");
338                 changed = 1;
339                 s = wpa_config_add_network(wpa_s->conf);
340                 if (s == NULL)
341                         return;
342                 wpa_config_set_network_defaults(s);
343         }
344
345         s->p2p_group = 1;
346         s->p2p_persistent_group = 1;
347         s->disabled = 2;
348         s->bssid_set = 1;
349         os_memcpy(s->bssid, go_dev_addr, ETH_ALEN);
350         s->mode = ssid->mode;
351         s->auth_alg = WPA_AUTH_ALG_OPEN;
352         s->key_mgmt = WPA_KEY_MGMT_PSK;
353         s->proto = WPA_PROTO_RSN;
354         s->pairwise_cipher = WPA_CIPHER_CCMP;
355         if (ssid->passphrase) {
356                 os_free(s->passphrase);
357                 s->passphrase = os_strdup(ssid->passphrase);
358         }
359         if (ssid->psk_set) {
360                 s->psk_set = 1;
361                 os_memcpy(s->psk, ssid->psk, 32);
362         }
363         if (s->passphrase && !s->psk_set)
364                 wpa_config_update_psk(s);
365         if (s->ssid == NULL || s->ssid_len < ssid->ssid_len) {
366                 os_free(s->ssid);
367                 s->ssid = os_malloc(ssid->ssid_len);
368         }
369         if (s->ssid) {
370                 s->ssid_len = ssid->ssid_len;
371                 os_memcpy(s->ssid, ssid->ssid, s->ssid_len);
372         }
373
374 #ifndef CONFIG_NO_CONFIG_WRITE
375         if (changed && wpa_s->conf->update_config &&
376             wpa_config_write(wpa_s->confname, wpa_s->conf)) {
377                 wpa_printf(MSG_DEBUG, "P2P: Failed to update configuration");
378         }
379 #endif /* CONFIG_NO_CONFIG_WRITE */
380 }
381
382
383 static void wpas_group_formation_completed(struct wpa_supplicant *wpa_s,
384                                            int success)
385 {
386         struct wpa_ssid *ssid;
387         const char *ssid_txt;
388         int client;
389         int persistent;
390         u8 go_dev_addr[ETH_ALEN];
391
392         /*
393          * This callback is likely called for the main interface. Update wpa_s
394          * to use the group interface if a new interface was created for the
395          * group.
396          */
397         if (wpa_s->global->p2p_group_formation)
398                 wpa_s = wpa_s->global->p2p_group_formation;
399         wpa_s->p2p_in_provisioning = 0;
400
401         if (!success) {
402                 wpa_msg(wpa_s->parent, MSG_INFO,
403                         P2P_EVENT_GROUP_FORMATION_FAILURE);
404                 wpas_p2p_group_delete(wpa_s);
405                 return;
406         }
407
408         wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_FORMATION_SUCCESS);
409
410         ssid = wpa_s->current_ssid;
411         if (ssid && ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION) {
412                 ssid->mode = WPAS_MODE_P2P_GO;
413                 p2p_group_notif_formation_done(wpa_s->p2p_group);
414                 wpa_supplicant_ap_mac_addr_filter(wpa_s, NULL);
415         }
416
417         persistent = 0;
418         if (ssid) {
419                 ssid_txt = wpa_ssid_txt(ssid->ssid, ssid->ssid_len);
420                 client = ssid->mode == WPAS_MODE_INFRA;
421                 if (ssid->mode == WPAS_MODE_P2P_GO) {
422                         persistent = ssid->p2p_persistent_group;
423                         os_memcpy(go_dev_addr, wpa_s->parent->own_addr,
424                                   ETH_ALEN);
425                 } else
426                         persistent = wpas_p2p_persistent_group(wpa_s,
427                                                                go_dev_addr,
428                                                                ssid->ssid,
429                                                                ssid->ssid_len);
430         } else {
431                 ssid_txt = "";
432                 client = wpa_s->p2p_group_interface ==
433                         P2P_GROUP_INTERFACE_CLIENT;
434         }
435
436         wpa_s->show_group_started = 0;
437         if (client) {
438                 /*
439                  * Indicate event only after successfully completed 4-way
440                  * handshake, i.e., when the interface is ready for data
441                  * packets.
442                  */
443                 wpa_s->show_group_started = 1;
444         } else if (ssid && ssid->passphrase == NULL && ssid->psk_set) {
445                 char psk[65];
446                 wpa_snprintf_hex(psk, sizeof(psk), ssid->psk, 32);
447                 wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
448                         "%s GO ssid=\"%s\" psk=%s go_dev_addr=" MACSTR "%s",
449                         wpa_s->ifname, ssid_txt, psk, MAC2STR(go_dev_addr),
450                         persistent ? " [PERSISTENT]" : "");
451                 wpas_p2p_cross_connect_setup(wpa_s);
452         } else {
453                 wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
454                         "%s GO ssid=\"%s\" passphrase=\"%s\" go_dev_addr="
455                         MACSTR "%s",
456                         wpa_s->ifname, ssid_txt,
457                         ssid && ssid->passphrase ? ssid->passphrase : "",
458                         MAC2STR(go_dev_addr),
459                         persistent ? " [PERSISTENT]" : "");
460                 wpas_p2p_cross_connect_setup(wpa_s);
461         }
462
463         if (persistent)
464                 wpas_p2p_store_persistent_group(wpa_s->parent, ssid,
465                                                 go_dev_addr);
466 }
467
468
469 static void wpas_send_action_cb(void *eloop_ctx, void *timeout_ctx)
470 {
471         struct wpa_supplicant *wpa_s = eloop_ctx;
472         struct wpa_supplicant *iface;
473         int res;
474         int without_roc;
475
476         without_roc = wpa_s->pending_action_without_roc;
477         wpa_s->pending_action_without_roc = 0;
478
479         if (wpa_s->pending_action_tx == NULL)
480                 return;
481
482         if (wpa_s->off_channel_freq != wpa_s->pending_action_freq &&
483             wpa_s->pending_action_freq != 0) {
484                 wpa_printf(MSG_DEBUG, "P2P: Pending Action frame TX "
485                            "waiting for another freq=%u (off_channel_freq=%u)",
486                            wpa_s->pending_action_freq,
487                            wpa_s->off_channel_freq);
488                 if (without_roc && wpa_s->off_channel_freq == 0) {
489                         /*
490                          * We may get here if wpas_send_action() found us to be
491                          * on the correct channel, but remain-on-channel cancel
492                          * event was received before getting here.
493                          */
494                         wpa_printf(MSG_DEBUG, "P2P: Schedule "
495                                    "remain-on-channel to send Action frame");
496                         if (wpa_drv_remain_on_channel(
497                                     wpa_s, wpa_s->pending_action_freq, 200) <
498                             0) {
499                                 wpa_printf(MSG_DEBUG, "P2P: Failed to request "
500                                            "driver to remain on channel (%u "
501                                            "MHz) for Action Frame TX",
502                                            wpa_s->pending_action_freq);
503                         } else
504                                 wpa_s->roc_waiting_drv_freq =
505                                         wpa_s->pending_action_freq;
506                 }
507                 return;
508         }
509
510         /*
511          * This call is likely going to be on the P2P device instance if the
512          * driver uses a separate interface for that purpose. However, some
513          * Action frames are actually sent within a P2P Group and when that is
514          * the case, we need to follow power saving (e.g., GO buffering the
515          * frame for a client in PS mode or a client following the advertised
516          * NoA from its GO). To make that easier for the driver, select the
517          * correct group interface here.
518          */
519         if (os_memcmp(wpa_s->pending_action_src, wpa_s->own_addr, ETH_ALEN) !=
520             0) {
521                 /*
522                  * Try to find a group interface that matches with the source
523                  * address.
524                  */
525                 iface = wpa_s->global->ifaces;
526                 while (iface) {
527                         if (os_memcmp(wpa_s->pending_action_src,
528                                       iface->own_addr, ETH_ALEN) == 0)
529                                 break;
530                         iface = iface->next;
531                 }
532                 if (iface) {
533                         wpa_printf(MSG_DEBUG, "P2P: Use group interface %s "
534                                    "instead of interface %s for Action TX",
535                                    iface->ifname, wpa_s->ifname);
536                 } else
537                         iface = wpa_s;
538         } else
539                 iface = wpa_s;
540
541         wpa_printf(MSG_DEBUG, "P2P: Sending pending Action frame to "
542                    MACSTR " using interface %s",
543                    MAC2STR(wpa_s->pending_action_dst), iface->ifname);
544         res = wpa_drv_send_action(iface, wpa_s->pending_action_freq,
545                                   wpa_s->pending_action_dst,
546                                   wpa_s->pending_action_src,
547                                   wpa_s->pending_action_bssid,
548                                   wpabuf_head(wpa_s->pending_action_tx),
549                                   wpabuf_len(wpa_s->pending_action_tx));
550         if (res) {
551                 wpa_printf(MSG_DEBUG, "P2P: Failed to send the pending "
552                            "Action frame");
553                 /*
554                  * Use fake TX status event to allow P2P state machine to
555                  * continue.
556                  */
557                 wpas_send_action_tx_status(
558                         wpa_s, wpa_s->pending_action_dst,
559                         wpabuf_head(wpa_s->pending_action_tx),
560                         wpabuf_len(wpa_s->pending_action_tx), 0);
561         }
562 }
563
564
565 void wpas_send_action_tx_status(struct wpa_supplicant *wpa_s, const u8 *dst,
566                                 const u8 *data, size_t data_len, int ack)
567 {
568         if (wpa_s->global->p2p_disabled)
569                 return;
570
571         if (wpa_s->pending_action_tx == NULL) {
572                 wpa_printf(MSG_DEBUG, "P2P: Ignore Action TX status - no "
573                            "pending operation");
574                 return;
575         }
576
577         if (os_memcmp(dst, wpa_s->pending_action_dst, ETH_ALEN) != 0) {
578                 wpa_printf(MSG_DEBUG, "P2P: Ignore Action TX status - unknown "
579                            "destination address");
580                 return;
581         }
582
583         wpabuf_free(wpa_s->pending_action_tx);
584         wpa_s->pending_action_tx = NULL;
585
586         p2p_send_action_cb(wpa_s->global->p2p, wpa_s->pending_action_freq,
587                            wpa_s->pending_action_dst,
588                            wpa_s->pending_action_src,
589                            wpa_s->pending_action_bssid,
590                            ack);
591
592         if (wpa_s->pending_pd_before_join &&
593             (os_memcmp(wpa_s->pending_action_dst, wpa_s->pending_join_dev_addr,
594                        ETH_ALEN) == 0 ||
595              os_memcmp(wpa_s->pending_action_dst,
596                        wpa_s->pending_join_iface_addr, ETH_ALEN) == 0)) {
597                 wpa_s->pending_pd_before_join = 0;
598                 wpa_printf(MSG_DEBUG, "P2P: Starting pending "
599                            "join-existing-group operation");
600                 wpas_p2p_join_start(wpa_s);
601         }
602 }
603
604
605 static int wpas_send_action(void *ctx, unsigned int freq, const u8 *dst,
606                             const u8 *src, const u8 *bssid, const u8 *buf,
607                             size_t len, unsigned int wait_time)
608 {
609         struct wpa_supplicant *wpa_s = ctx;
610
611         wpa_printf(MSG_DEBUG, "P2P: Send action frame: freq=%d dst=" MACSTR
612                    " src=" MACSTR " bssid=" MACSTR,
613                    freq, MAC2STR(dst), MAC2STR(src), MAC2STR(bssid));
614
615         if (wpa_s->pending_action_tx) {
616                 wpa_printf(MSG_DEBUG, "P2P: Dropped pending Action frame TX "
617                            "to " MACSTR, MAC2STR(wpa_s->pending_action_dst));
618                 wpabuf_free(wpa_s->pending_action_tx);
619         }
620         wpa_s->pending_action_tx = wpabuf_alloc(len);
621         if (wpa_s->pending_action_tx == NULL)
622                 return -1;
623         wpabuf_put_data(wpa_s->pending_action_tx, buf, len);
624         os_memcpy(wpa_s->pending_action_src, src, ETH_ALEN);
625         os_memcpy(wpa_s->pending_action_dst, dst, ETH_ALEN);
626         os_memcpy(wpa_s->pending_action_bssid, bssid, ETH_ALEN);
627         wpa_s->pending_action_freq = freq;
628
629         if (wpa_s->off_channel_freq == freq || freq == 0) {
630                 /* Already on requested channel; send immediately */
631                 /* TODO: Would there ever be need to extend the current
632                  * duration on the channel? */
633                 wpa_s->pending_action_without_roc = 1;
634                 eloop_cancel_timeout(wpas_send_action_cb, wpa_s, NULL);
635                 eloop_register_timeout(0, 0, wpas_send_action_cb, wpa_s, NULL);
636                 return 0;
637         }
638         wpa_s->pending_action_without_roc = 0;
639
640         if (wpa_s->roc_waiting_drv_freq == freq) {
641                 wpa_printf(MSG_DEBUG, "P2P: Already waiting for driver to get "
642                            "to frequency %u MHz; continue waiting to send the "
643                            "Action frame", freq);
644                 return 0;
645         }
646
647         wpa_printf(MSG_DEBUG, "P2P: Schedule Action frame to be transmitted "
648                    "once the driver gets to the requested channel");
649         if (wait_time > wpa_s->max_remain_on_chan)
650                 wait_time = wpa_s->max_remain_on_chan;
651         if (wpa_drv_remain_on_channel(wpa_s, freq, wait_time) < 0) {
652                 wpa_printf(MSG_DEBUG, "P2P: Failed to request driver "
653                            "to remain on channel (%u MHz) for Action "
654                            "Frame TX", freq);
655                 return -1;
656         }
657         wpa_s->roc_waiting_drv_freq = freq;
658
659         return 0;
660 }
661
662
663 static void wpas_send_action_done(void *ctx)
664 {
665         struct wpa_supplicant *wpa_s = ctx;
666         wpa_printf(MSG_DEBUG, "P2P: Action frame sequence done notification");
667         wpabuf_free(wpa_s->pending_action_tx);
668         wpa_s->pending_action_tx = NULL;
669         if (wpa_s->off_channel_freq) {
670                 wpa_drv_cancel_remain_on_channel(wpa_s);
671                 wpa_s->off_channel_freq = 0;
672                 wpa_s->roc_waiting_drv_freq = 0;
673         }
674 }
675
676
677 static int wpas_copy_go_neg_results(struct wpa_supplicant *wpa_s,
678                                     struct p2p_go_neg_results *params)
679 {
680         if (wpa_s->go_params == NULL) {
681                 wpa_s->go_params = os_malloc(sizeof(*params));
682                 if (wpa_s->go_params == NULL)
683                         return -1;
684         }
685         os_memcpy(wpa_s->go_params, params, sizeof(*params));
686         return 0;
687 }
688
689
690 static void wpas_start_wps_enrollee(struct wpa_supplicant *wpa_s,
691                                     struct p2p_go_neg_results *res)
692 {
693         wpa_supplicant_ap_deinit(wpa_s);
694         wpas_copy_go_neg_results(wpa_s, res);
695         if (res->wps_method == WPS_PBC)
696                 wpas_wps_start_pbc(wpa_s, NULL /* res->peer_interface_addr */,
697                                    1);
698         else {
699                 u16 dev_pw_id = DEV_PW_DEFAULT;
700                 if (wpa_s->p2p_wps_method == WPS_PIN_KEYPAD)
701                         dev_pw_id = DEV_PW_REGISTRAR_SPECIFIED;
702                 wpas_wps_start_pin(wpa_s, res->peer_interface_addr,
703                                    wpa_s->p2p_pin, 1, dev_pw_id);
704         }
705 }
706
707
708 static void p2p_go_configured(void *ctx, void *data)
709 {
710         struct wpa_supplicant *wpa_s = ctx;
711         struct p2p_go_neg_results *params = data;
712         struct wpa_ssid *ssid;
713
714         ssid = wpa_s->current_ssid;
715         if (ssid && ssid->mode == WPAS_MODE_P2P_GO) {
716                 wpa_printf(MSG_DEBUG, "P2P: Group setup without provisioning");
717                 wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
718                         "%s GO ssid=\"%s\" passphrase=\"%s\" go_dev_addr="
719                         MACSTR "%s",
720                         wpa_s->ifname,
721                         wpa_ssid_txt(ssid->ssid, ssid->ssid_len),
722                         params->passphrase ? params->passphrase : "",
723                         MAC2STR(wpa_s->parent->own_addr),
724                         params->persistent_group ? " [PERSISTENT]" : "");
725                 if (params->persistent_group)
726                         wpas_p2p_store_persistent_group(
727                                 wpa_s->parent, ssid,
728                                 wpa_s->parent->own_addr);
729                 wpas_p2p_cross_connect_setup(wpa_s);
730                 return;
731         }
732
733         wpa_printf(MSG_DEBUG, "P2P: Setting up WPS for GO provisioning");
734         if (wpa_supplicant_ap_mac_addr_filter(wpa_s,
735                                               params->peer_interface_addr)) {
736                 wpa_printf(MSG_DEBUG, "P2P: Failed to setup MAC address "
737                            "filtering");
738                 return;
739         }
740         if (params->wps_method == WPS_PBC)
741                 wpa_supplicant_ap_wps_pbc(wpa_s, params->peer_interface_addr);
742         else if (wpa_s->p2p_pin[0])
743                 wpa_supplicant_ap_wps_pin(wpa_s, params->peer_interface_addr,
744                                           wpa_s->p2p_pin, NULL, 0);
745         os_free(wpa_s->go_params);
746         wpa_s->go_params = NULL;
747 }
748
749
750 static void wpas_start_wps_go(struct wpa_supplicant *wpa_s,
751                               struct p2p_go_neg_results *params,
752                               int group_formation)
753 {
754         struct wpa_ssid *ssid;
755
756         if (wpas_copy_go_neg_results(wpa_s, params) < 0)
757                 return;
758
759         ssid = wpa_config_add_network(wpa_s->conf);
760         if (ssid == NULL)
761                 return;
762
763         wpa_config_set_network_defaults(ssid);
764         ssid->temporary = 1;
765         ssid->p2p_group = 1;
766         ssid->p2p_persistent_group = params->persistent_group;
767         ssid->mode = group_formation ? WPAS_MODE_P2P_GROUP_FORMATION :
768                 WPAS_MODE_P2P_GO;
769         ssid->frequency = params->freq;
770         ssid->ssid = os_zalloc(params->ssid_len + 1);
771         if (ssid->ssid) {
772                 os_memcpy(ssid->ssid, params->ssid, params->ssid_len);
773                 ssid->ssid_len = params->ssid_len;
774         }
775         ssid->auth_alg = WPA_AUTH_ALG_OPEN;
776         ssid->key_mgmt = WPA_KEY_MGMT_PSK;
777         ssid->proto = WPA_PROTO_RSN;
778         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
779         ssid->passphrase = os_strdup(params->passphrase);
780
781         wpa_s->ap_configured_cb = p2p_go_configured;
782         wpa_s->ap_configured_cb_ctx = wpa_s;
783         wpa_s->ap_configured_cb_data = wpa_s->go_params;
784         wpa_s->connect_without_scan = 1;
785         wpa_s->reassociate = 1;
786         wpa_s->disconnected = 0;
787         wpa_supplicant_req_scan(wpa_s, 0, 0);
788 }
789
790
791 static void wpas_p2p_clone_config(struct wpa_supplicant *dst,
792                                   const struct wpa_supplicant *src)
793 {
794         struct wpa_config *d;
795         const struct wpa_config *s;
796
797         d = dst->conf;
798         s = src->conf;
799
800 #define C(n) if (s->n) d->n = os_strdup(s->n)
801         C(device_name);
802         C(manufacturer);
803         C(model_name);
804         C(model_number);
805         C(serial_number);
806         C(device_type);
807         C(config_methods);
808 #undef C
809 }
810
811
812 static int wpas_p2p_add_group_interface(struct wpa_supplicant *wpa_s,
813                                         enum wpa_driver_if_type type)
814 {
815         char ifname[120], force_ifname[120];
816
817         if (wpa_s->pending_interface_name[0]) {
818                 wpa_printf(MSG_DEBUG, "P2P: Pending virtual interface exists "
819                            "- skip creation of a new one");
820                 if (is_zero_ether_addr(wpa_s->pending_interface_addr)) {
821                         wpa_printf(MSG_DEBUG, "P2P: Pending virtual address "
822                                    "unknown?! ifname='%s'",
823                                    wpa_s->pending_interface_name);
824                         return -1;
825                 }
826                 return 0;
827         }
828
829         os_snprintf(ifname, sizeof(ifname), "%s-p2p-%d", wpa_s->ifname,
830                     wpa_s->p2p_group_idx);
831         force_ifname[0] = '\0';
832
833         wpa_printf(MSG_DEBUG, "P2P: Create a new interface %s for the group",
834                    ifname);
835         wpa_s->p2p_group_idx++;
836
837         wpa_s->pending_interface_type = type;
838         if (wpa_drv_if_add(wpa_s, type, ifname, NULL, NULL, force_ifname,
839                            wpa_s->pending_interface_addr) < 0) {
840                 wpa_printf(MSG_ERROR, "P2P: Failed to create new group "
841                            "interface");
842                 return -1;
843         }
844
845         if (force_ifname[0]) {
846                 wpa_printf(MSG_DEBUG, "P2P: Driver forced interface name %s",
847                            force_ifname);
848                 os_strlcpy(wpa_s->pending_interface_name, force_ifname,
849                            sizeof(wpa_s->pending_interface_name));
850         } else
851                 os_strlcpy(wpa_s->pending_interface_name, ifname,
852                            sizeof(wpa_s->pending_interface_name));
853         wpa_printf(MSG_DEBUG, "P2P: Created pending virtual interface %s addr "
854                    MACSTR, wpa_s->pending_interface_name,
855                    MAC2STR(wpa_s->pending_interface_addr));
856
857         return 0;
858 }
859
860
861 static void wpas_p2p_remove_pending_group_interface(
862         struct wpa_supplicant *wpa_s)
863 {
864         if (!wpa_s->pending_interface_name[0] ||
865             is_zero_ether_addr(wpa_s->pending_interface_addr))
866                 return; /* No pending virtual interface */
867
868         wpa_printf(MSG_DEBUG, "P2P: Removing pending group interface %s",
869                    wpa_s->pending_interface_name);
870         wpa_drv_if_remove(wpa_s, wpa_s->pending_interface_type,
871                           wpa_s->pending_interface_name);
872         os_memset(wpa_s->pending_interface_addr, 0, ETH_ALEN);
873         wpa_s->pending_interface_name[0] = '\0';
874 }
875
876
877 static struct wpa_supplicant *
878 wpas_p2p_init_group_interface(struct wpa_supplicant *wpa_s, int go)
879 {
880         struct wpa_interface iface;
881         struct wpa_supplicant *group_wpa_s;
882
883         if (!wpa_s->pending_interface_name[0]) {
884                 wpa_printf(MSG_ERROR, "P2P: No pending group interface");
885                 return NULL;
886         }
887
888         os_memset(&iface, 0, sizeof(iface));
889         iface.ifname = wpa_s->pending_interface_name;
890         iface.driver = wpa_s->driver->name;
891         iface.ctrl_interface = wpa_s->conf->ctrl_interface;
892         iface.driver_param = wpa_s->conf->driver_param;
893         group_wpa_s = wpa_supplicant_add_iface(wpa_s->global, &iface);
894         if (group_wpa_s == NULL) {
895                 wpa_printf(MSG_ERROR, "P2P: Failed to create new "
896                            "wpa_supplicant interface");
897                 return NULL;
898         }
899         wpa_s->pending_interface_name[0] = '\0';
900         group_wpa_s->parent = wpa_s;
901         group_wpa_s->p2p_group_interface = go ? P2P_GROUP_INTERFACE_GO :
902                 P2P_GROUP_INTERFACE_CLIENT;
903         wpa_s->global->p2p_group_formation = group_wpa_s;
904
905         wpas_p2p_clone_config(group_wpa_s, wpa_s);
906
907         return group_wpa_s;
908 }
909
910
911 static void wpas_p2p_group_formation_timeout(void *eloop_ctx,
912                                              void *timeout_ctx)
913 {
914         struct wpa_supplicant *wpa_s = eloop_ctx;
915         wpa_printf(MSG_DEBUG, "P2P: Group Formation timed out");
916         if (wpa_s->global->p2p)
917                 p2p_group_formation_failed(wpa_s->global->p2p);
918         wpas_group_formation_completed(wpa_s, 0);
919 }
920
921
922 void wpas_go_neg_completed(void *ctx, struct p2p_go_neg_results *res)
923 {
924         struct wpa_supplicant *wpa_s = ctx;
925
926         if (wpa_s->off_channel_freq) {
927                 wpa_drv_cancel_remain_on_channel(wpa_s);
928                 wpa_s->off_channel_freq = 0;
929                 wpa_s->roc_waiting_drv_freq = 0;
930         }
931
932         if (res->status) {
933                 wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_GO_NEG_FAILURE "status=%d",
934                         res->status);
935                 wpas_p2p_remove_pending_group_interface(wpa_s);
936                 return;
937         }
938
939         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_GO_NEG_SUCCESS);
940
941         if (wpa_s->create_p2p_iface) {
942                 struct wpa_supplicant *group_wpa_s =
943                         wpas_p2p_init_group_interface(wpa_s, res->role_go);
944                 if (group_wpa_s == NULL) {
945                         wpas_p2p_remove_pending_group_interface(wpa_s);
946                         return;
947                 }
948                 if (group_wpa_s != wpa_s) {
949                         os_memcpy(group_wpa_s->p2p_pin, wpa_s->p2p_pin,
950                                   sizeof(group_wpa_s->p2p_pin));
951                         group_wpa_s->p2p_wps_method = wpa_s->p2p_wps_method;
952                 }
953                 os_memset(wpa_s->pending_interface_addr, 0, ETH_ALEN);
954                 wpa_s->pending_interface_name[0] = '\0';
955                 group_wpa_s->p2p_in_provisioning = 1;
956
957                 if (res->role_go)
958                         wpas_start_wps_go(group_wpa_s, res, 1);
959                 else
960                         wpas_start_wps_enrollee(group_wpa_s, res);
961         } else {
962                 wpa_s->p2p_in_provisioning = 1;
963                 wpa_s->global->p2p_group_formation = wpa_s;
964
965                 if (res->role_go)
966                         wpas_start_wps_go(wpa_s, res, 1);
967                 else
968                         wpas_start_wps_enrollee(ctx, res);
969         }
970
971         wpa_s->p2p_long_listen = 0;
972         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
973
974         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s, NULL);
975         /* TODO: add peer Config Timeout */
976         eloop_register_timeout(15, 0, wpas_p2p_group_formation_timeout, wpa_s,
977                                NULL);
978 }
979
980
981 void wpas_go_neg_req_rx(void *ctx, const u8 *src, u16 dev_passwd_id)
982 {
983         struct wpa_supplicant *wpa_s = ctx;
984         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_GO_NEG_REQUEST MACSTR
985                 " dev_passwd_id=%u", MAC2STR(src), dev_passwd_id);
986 }
987
988
989 void wpas_dev_found(void *ctx, const u8 *addr, const u8 *dev_addr,
990                     const u8 *pri_dev_type, const char *dev_name,
991                     u16 config_methods, u8 dev_capab, u8 group_capab)
992 {
993         struct wpa_supplicant *wpa_s = ctx;
994         char devtype[WPS_DEV_TYPE_BUFSIZE];
995         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_DEVICE_FOUND MACSTR
996                 " p2p_dev_addr=" MACSTR
997                 " pri_dev_type=%s name='%s' config_methods=0x%x "
998                 "dev_capab=0x%x group_capab=0x%x",
999                 MAC2STR(addr), MAC2STR(dev_addr),
1000                 wps_dev_type_bin2str(pri_dev_type, devtype, sizeof(devtype)),
1001                 dev_name, config_methods, dev_capab, group_capab);
1002 }
1003
1004
1005 static int wpas_start_listen(void *ctx, unsigned int freq,
1006                              unsigned int duration,
1007                              const struct wpabuf *probe_resp_ie)
1008 {
1009         struct wpa_supplicant *wpa_s = ctx;
1010
1011         wpa_drv_set_ap_wps_ie(wpa_s, NULL, probe_resp_ie);
1012
1013         if (wpa_drv_probe_req_report(wpa_s, 1) < 0) {
1014                 wpa_printf(MSG_DEBUG, "P2P: Failed to request the driver to "
1015                            "report received Probe Request frames");
1016                 return -1;
1017         }
1018
1019         wpa_s->pending_listen_freq = freq;
1020         wpa_s->pending_listen_duration = duration;
1021
1022         if (wpa_drv_remain_on_channel(wpa_s, freq, duration) < 0) {
1023                 wpa_printf(MSG_DEBUG, "P2P: Failed to request the driver "
1024                            "to remain on channel (%u MHz) for Listen "
1025                            "state", freq);
1026                 wpa_s->pending_listen_freq = 0;
1027                 return -1;
1028         }
1029         wpa_s->roc_waiting_drv_freq = freq;
1030
1031         return 0;
1032 }
1033
1034
1035 static void wpas_stop_listen(void *ctx)
1036 {
1037         struct wpa_supplicant *wpa_s = ctx;
1038         if (wpa_s->off_channel_freq) {
1039                 wpa_drv_cancel_remain_on_channel(wpa_s);
1040                 wpa_s->off_channel_freq = 0;
1041                 wpa_s->roc_waiting_drv_freq = 0;
1042         }
1043         wpa_drv_probe_req_report(wpa_s, 0);
1044 }
1045
1046
1047 static int wpas_send_probe_resp(void *ctx, const struct wpabuf *buf)
1048 {
1049         struct wpa_supplicant *wpa_s = ctx;
1050         return wpa_drv_send_mlme(wpa_s, wpabuf_head(buf), wpabuf_len(buf));
1051 }
1052
1053
1054 static struct p2p_srv_bonjour *
1055 wpas_p2p_service_get_bonjour(struct wpa_supplicant *wpa_s,
1056                              const struct wpabuf *query)
1057 {
1058         struct p2p_srv_bonjour *bsrv;
1059         size_t len;
1060
1061         len = wpabuf_len(query);
1062         dl_list_for_each(bsrv, &wpa_s->global->p2p_srv_bonjour,
1063                          struct p2p_srv_bonjour, list) {
1064                 if (len == wpabuf_len(bsrv->query) &&
1065                     os_memcmp(wpabuf_head(query), wpabuf_head(bsrv->query),
1066                               len) == 0)
1067                         return bsrv;
1068         }
1069         return NULL;
1070 }
1071
1072
1073 static struct p2p_srv_upnp *
1074 wpas_p2p_service_get_upnp(struct wpa_supplicant *wpa_s, u8 version,
1075                           const char *service)
1076 {
1077         struct p2p_srv_upnp *usrv;
1078
1079         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
1080                          struct p2p_srv_upnp, list) {
1081                 if (version == usrv->version &&
1082                     os_strcmp(service, usrv->service) == 0)
1083                         return usrv;
1084         }
1085         return NULL;
1086 }
1087
1088
1089 static void wpas_sd_add_proto_not_avail(struct wpabuf *resp, u8 srv_proto,
1090                                         u8 srv_trans_id)
1091 {
1092         u8 *len_pos;
1093
1094         if (wpabuf_tailroom(resp) < 5)
1095                 return;
1096
1097         /* Length (to be filled) */
1098         len_pos = wpabuf_put(resp, 2);
1099         wpabuf_put_u8(resp, srv_proto);
1100         wpabuf_put_u8(resp, srv_trans_id);
1101         /* Status Code */
1102         wpabuf_put_u8(resp, P2P_SD_PROTO_NOT_AVAILABLE);
1103         /* Response Data: empty */
1104         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
1105 }
1106
1107
1108 static void wpas_sd_all_bonjour(struct wpa_supplicant *wpa_s,
1109                                 struct wpabuf *resp, u8 srv_trans_id)
1110 {
1111         struct p2p_srv_bonjour *bsrv;
1112         u8 *len_pos;
1113
1114         wpa_printf(MSG_DEBUG, "P2P: SD Request for all Bonjour services");
1115
1116         if (dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
1117                 wpa_printf(MSG_DEBUG, "P2P: Bonjour protocol not available");
1118                 return;
1119         }
1120
1121         dl_list_for_each(bsrv, &wpa_s->global->p2p_srv_bonjour,
1122                          struct p2p_srv_bonjour, list) {
1123                 if (wpabuf_tailroom(resp) <
1124                     5 + wpabuf_len(bsrv->query) + wpabuf_len(bsrv->resp))
1125                         return;
1126                 /* Length (to be filled) */
1127                 len_pos = wpabuf_put(resp, 2);
1128                 wpabuf_put_u8(resp, P2P_SERV_BONJOUR);
1129                 wpabuf_put_u8(resp, srv_trans_id);
1130                 /* Status Code */
1131                 wpabuf_put_u8(resp, P2P_SD_SUCCESS);
1132                 wpa_hexdump_ascii(MSG_DEBUG, "P2P: Matching Bonjour service",
1133                                   wpabuf_head(bsrv->resp),
1134                                   wpabuf_len(bsrv->resp));
1135                 /* Response Data */
1136                 wpabuf_put_buf(resp, bsrv->query); /* Key */
1137                 wpabuf_put_buf(resp, bsrv->resp); /* Value */
1138                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
1139                              2);
1140         }
1141 }
1142
1143
1144 static void wpas_sd_req_bonjour(struct wpa_supplicant *wpa_s,
1145                                 struct wpabuf *resp, u8 srv_trans_id,
1146                                 const u8 *query, size_t query_len)
1147 {
1148         struct p2p_srv_bonjour *bsrv;
1149         struct wpabuf buf;
1150         u8 *len_pos;
1151
1152         wpa_hexdump_ascii(MSG_DEBUG, "P2P: SD Request for Bonjour",
1153                           query, query_len);
1154         if (dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
1155                 wpa_printf(MSG_DEBUG, "P2P: Bonjour protocol not available");
1156                 wpas_sd_add_proto_not_avail(resp, P2P_SERV_BONJOUR,
1157                                             srv_trans_id);
1158                 return;
1159         }
1160
1161         if (query_len == 0) {
1162                 wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
1163                 return;
1164         }
1165
1166         if (wpabuf_tailroom(resp) < 5)
1167                 return;
1168         /* Length (to be filled) */
1169         len_pos = wpabuf_put(resp, 2);
1170         wpabuf_put_u8(resp, P2P_SERV_BONJOUR);
1171         wpabuf_put_u8(resp, srv_trans_id);
1172
1173         wpabuf_set(&buf, query, query_len);
1174         bsrv = wpas_p2p_service_get_bonjour(wpa_s, &buf);
1175         if (bsrv == NULL) {
1176                 wpa_printf(MSG_DEBUG, "P2P: Requested Bonjour service not "
1177                            "available");
1178
1179                 /* Status Code */
1180                 wpabuf_put_u8(resp, P2P_SD_REQUESTED_INFO_NOT_AVAILABLE);
1181                 /* Response Data: empty */
1182                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
1183                              2);
1184                 return;
1185         }
1186
1187         /* Status Code */
1188         wpabuf_put_u8(resp, P2P_SD_SUCCESS);
1189         wpa_hexdump_ascii(MSG_DEBUG, "P2P: Matching Bonjour service",
1190                           wpabuf_head(bsrv->resp), wpabuf_len(bsrv->resp));
1191
1192         if (wpabuf_tailroom(resp) >=
1193             wpabuf_len(bsrv->query) + wpabuf_len(bsrv->resp)) {
1194                 /* Response Data */
1195                 wpabuf_put_buf(resp, bsrv->query); /* Key */
1196                 wpabuf_put_buf(resp, bsrv->resp); /* Value */
1197         }
1198         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
1199 }
1200
1201
1202 static void wpas_sd_all_upnp(struct wpa_supplicant *wpa_s,
1203                              struct wpabuf *resp, u8 srv_trans_id)
1204 {
1205         struct p2p_srv_upnp *usrv;
1206         u8 *len_pos;
1207
1208         wpa_printf(MSG_DEBUG, "P2P: SD Request for all UPnP services");
1209
1210         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp)) {
1211                 wpa_printf(MSG_DEBUG, "P2P: UPnP protocol not available");
1212                 return;
1213         }
1214
1215         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
1216                          struct p2p_srv_upnp, list) {
1217                 if (wpabuf_tailroom(resp) < 5 + 1 + os_strlen(usrv->service))
1218                         return;
1219
1220                 /* Length (to be filled) */
1221                 len_pos = wpabuf_put(resp, 2);
1222                 wpabuf_put_u8(resp, P2P_SERV_UPNP);
1223                 wpabuf_put_u8(resp, srv_trans_id);
1224
1225                 /* Status Code */
1226                 wpabuf_put_u8(resp, P2P_SD_SUCCESS);
1227                 /* Response Data */
1228                 wpabuf_put_u8(resp, usrv->version);
1229                 wpa_printf(MSG_DEBUG, "P2P: Matching UPnP Service: %s",
1230                            usrv->service);
1231                 wpabuf_put_str(resp, usrv->service);
1232                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
1233                              2);
1234         }
1235 }
1236
1237
1238 static void wpas_sd_req_upnp(struct wpa_supplicant *wpa_s,
1239                              struct wpabuf *resp, u8 srv_trans_id,
1240                              const u8 *query, size_t query_len)
1241 {
1242         struct p2p_srv_upnp *usrv;
1243         u8 *len_pos;
1244         u8 version;
1245         char *str;
1246         int count = 0;
1247
1248         wpa_hexdump_ascii(MSG_DEBUG, "P2P: SD Request for UPnP",
1249                           query, query_len);
1250
1251         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp)) {
1252                 wpa_printf(MSG_DEBUG, "P2P: UPnP protocol not available");
1253                 wpas_sd_add_proto_not_avail(resp, P2P_SERV_UPNP,
1254                                             srv_trans_id);
1255                 return;
1256         }
1257
1258         if (query_len == 0) {
1259                 wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
1260                 return;
1261         }
1262
1263         version = query[0];
1264         str = os_malloc(query_len);
1265         if (str == NULL)
1266                 return;
1267         os_memcpy(str, query + 1, query_len - 1);
1268         str[query_len - 1] = '\0';
1269
1270         if (wpabuf_tailroom(resp) < 5)
1271                 return;
1272
1273         /* Length (to be filled) */
1274         len_pos = wpabuf_put(resp, 2);
1275         wpabuf_put_u8(resp, P2P_SERV_UPNP);
1276         wpabuf_put_u8(resp, srv_trans_id);
1277
1278         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
1279                          struct p2p_srv_upnp, list) {
1280                 if (version != usrv->version)
1281                         continue;
1282
1283                 if (os_strcmp(str, "ssdp:all") != 0 &&
1284                     os_strstr(usrv->service, str) == NULL)
1285                         continue;
1286
1287                 if (wpabuf_tailroom(resp) < 2)
1288                         break;
1289                 if (count == 0) {
1290                         /* Status Code */
1291                         wpabuf_put_u8(resp, P2P_SD_SUCCESS);
1292                         /* Response Data */
1293                         wpabuf_put_u8(resp, version);
1294                 } else
1295                         wpabuf_put_u8(resp, ',');
1296
1297                 count++;
1298
1299                 wpa_printf(MSG_DEBUG, "P2P: Matching UPnP Service: %s",
1300                            usrv->service);
1301                 if (wpabuf_tailroom(resp) < os_strlen(usrv->service))
1302                         break;
1303                 wpabuf_put_str(resp, usrv->service);
1304         }
1305
1306         if (count == 0) {
1307                 wpa_printf(MSG_DEBUG, "P2P: Requested UPnP service not "
1308                            "available");
1309                 /* Status Code */
1310                 wpabuf_put_u8(resp, P2P_SD_REQUESTED_INFO_NOT_AVAILABLE);
1311                 /* Response Data: empty */
1312         }
1313
1314         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
1315 }
1316
1317
1318 void wpas_sd_request(void *ctx, int freq, const u8 *sa, u8 dialog_token,
1319                      u16 update_indic, const u8 *tlvs, size_t tlvs_len)
1320 {
1321         struct wpa_supplicant *wpa_s = ctx;
1322         const u8 *pos = tlvs;
1323         const u8 *end = tlvs + tlvs_len;
1324         const u8 *tlv_end;
1325         u16 slen;
1326         struct wpabuf *resp;
1327         u8 srv_proto, srv_trans_id;
1328         size_t buf_len;
1329         char *buf;
1330
1331         wpa_hexdump(MSG_MSGDUMP, "P2P: Service Discovery Request TLVs",
1332                     tlvs, tlvs_len);
1333         buf_len = 2 * tlvs_len + 1;
1334         buf = os_malloc(buf_len);
1335         if (buf) {
1336                 wpa_snprintf_hex(buf, buf_len, tlvs, tlvs_len);
1337                 wpa_msg_ctrl(wpa_s, MSG_INFO, P2P_EVENT_SERV_DISC_REQ "%d "
1338                              MACSTR " %u %u %s",
1339                              freq, MAC2STR(sa), dialog_token, update_indic,
1340                              buf);
1341                 os_free(buf);
1342         }
1343
1344         if (wpa_s->p2p_sd_over_ctrl_iface)
1345                 return; /* to be processed by an external program */
1346
1347         resp = wpabuf_alloc(10000);
1348         if (resp == NULL)
1349                 return;
1350
1351         while (pos + 1 < end) {
1352                 wpa_printf(MSG_DEBUG, "P2P: Service Request TLV");
1353                 slen = WPA_GET_LE16(pos);
1354                 pos += 2;
1355                 if (pos + slen > end || slen < 2) {
1356                         wpa_printf(MSG_DEBUG, "P2P: Unexpected Query Data "
1357                                    "length");
1358                         wpabuf_free(resp);
1359                         return;
1360                 }
1361                 tlv_end = pos + slen;
1362
1363                 srv_proto = *pos++;
1364                 wpa_printf(MSG_DEBUG, "P2P: Service Protocol Type %u",
1365                            srv_proto);
1366                 srv_trans_id = *pos++;
1367                 wpa_printf(MSG_DEBUG, "P2P: Service Transaction ID %u",
1368                            srv_trans_id);
1369
1370                 wpa_hexdump(MSG_MSGDUMP, "P2P: Query Data",
1371                             pos, tlv_end - pos);
1372
1373
1374                 if (wpa_s->force_long_sd) {
1375                         wpa_printf(MSG_DEBUG, "P2P: SD test - force long "
1376                                    "response");
1377                         wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
1378                         wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
1379                         goto done;
1380                 }
1381
1382                 switch (srv_proto) {
1383                 case P2P_SERV_ALL_SERVICES:
1384                         wpa_printf(MSG_DEBUG, "P2P: Service Discovery Request "
1385                                    "for all services");
1386                         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp) &&
1387                             dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
1388                                 wpa_printf(MSG_DEBUG, "P2P: No service "
1389                                            "discovery protocols available");
1390                                 wpas_sd_add_proto_not_avail(
1391                                         resp, P2P_SERV_ALL_SERVICES,
1392                                         srv_trans_id);
1393                                 break;
1394                         }
1395                         wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
1396                         wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
1397                         break;
1398                 case P2P_SERV_BONJOUR:
1399                         wpas_sd_req_bonjour(wpa_s, resp, srv_trans_id,
1400                                             pos, tlv_end - pos);
1401                         break;
1402                 case P2P_SERV_UPNP:
1403                         wpas_sd_req_upnp(wpa_s, resp, srv_trans_id,
1404                                          pos, tlv_end - pos);
1405                         break;
1406                 default:
1407                         wpa_printf(MSG_DEBUG, "P2P: Unavailable service "
1408                                    "protocol %u", srv_proto);
1409                         wpas_sd_add_proto_not_avail(resp, srv_proto,
1410                                                     srv_trans_id);
1411                         break;
1412                 }
1413
1414                 pos = tlv_end;
1415         }
1416
1417 done:
1418         wpas_p2p_sd_response(wpa_s, freq, sa, dialog_token, resp);
1419
1420         wpabuf_free(resp);
1421 }
1422
1423
1424 void wpas_sd_response(void *ctx, const u8 *sa, u16 update_indic,
1425                       const u8 *tlvs, size_t tlvs_len)
1426 {
1427         struct wpa_supplicant *wpa_s = ctx;
1428         const u8 *pos = tlvs;
1429         const u8 *end = tlvs + tlvs_len;
1430         const u8 *tlv_end;
1431         u16 slen;
1432         size_t buf_len;
1433         char *buf;
1434
1435         wpa_hexdump(MSG_MSGDUMP, "P2P: Service Discovery Response TLVs",
1436                     tlvs, tlvs_len);
1437         if (tlvs_len > 1500) {
1438                 /* TODO: better way for handling this */
1439                 wpa_msg_ctrl(wpa_s, MSG_INFO,
1440                              P2P_EVENT_SERV_DISC_RESP MACSTR
1441                              " %u <long response: %u bytes>",
1442                              MAC2STR(sa), update_indic,
1443                              (unsigned int) tlvs_len);
1444         } else {
1445                 buf_len = 2 * tlvs_len + 1;
1446                 buf = os_malloc(buf_len);
1447                 if (buf) {
1448                         wpa_snprintf_hex(buf, buf_len, tlvs, tlvs_len);
1449                         wpa_msg_ctrl(wpa_s, MSG_INFO,
1450                                      P2P_EVENT_SERV_DISC_RESP MACSTR " %u %s",
1451                                      MAC2STR(sa), update_indic, buf);
1452                         os_free(buf);
1453                 }
1454         }
1455
1456         while (pos < end) {
1457                 u8 srv_proto, srv_trans_id, status;
1458
1459                 wpa_printf(MSG_DEBUG, "P2P: Service Response TLV");
1460                 slen = WPA_GET_LE16(pos);
1461                 pos += 2;
1462                 if (pos + slen > end || slen < 3) {
1463                         wpa_printf(MSG_DEBUG, "P2P: Unexpected Response Data "
1464                                    "length");
1465                         return;
1466                 }
1467                 tlv_end = pos + slen;
1468
1469                 srv_proto = *pos++;
1470                 wpa_printf(MSG_DEBUG, "P2P: Service Protocol Type %u",
1471                            srv_proto);
1472                 srv_trans_id = *pos++;
1473                 wpa_printf(MSG_DEBUG, "P2P: Service Transaction ID %u",
1474                            srv_trans_id);
1475                 status = *pos++;
1476                 wpa_printf(MSG_DEBUG, "P2P: Status Code ID %u",
1477                            status);
1478
1479                 wpa_hexdump(MSG_MSGDUMP, "P2P: Response Data",
1480                             pos, tlv_end - pos);
1481
1482                 pos = tlv_end;
1483         }
1484 }
1485
1486
1487 void * wpas_p2p_sd_request(struct wpa_supplicant *wpa_s, const u8 *dst,
1488                            const struct wpabuf *tlvs)
1489 {
1490         return p2p_sd_request(wpa_s->global->p2p, dst, tlvs);
1491 }
1492
1493
1494 void * wpas_p2p_sd_request_upnp(struct wpa_supplicant *wpa_s, const u8 *dst,
1495                                 u8 version, const char *query)
1496 {
1497         struct wpabuf *tlvs;
1498         void *ret;
1499
1500         tlvs = wpabuf_alloc(2 + 1 + 1 + 1 + os_strlen(query));
1501         if (tlvs == NULL)
1502                 return NULL;
1503         wpabuf_put_le16(tlvs, 1 + 1 + 1 + os_strlen(query));
1504         wpabuf_put_u8(tlvs, P2P_SERV_UPNP); /* Service Protocol Type */
1505         wpabuf_put_u8(tlvs, 1); /* Service Transaction ID */
1506         wpabuf_put_u8(tlvs, version);
1507         wpabuf_put_str(tlvs, query);
1508         ret = wpas_p2p_sd_request(wpa_s, dst, tlvs);
1509         wpabuf_free(tlvs);
1510         return ret;
1511 }
1512
1513
1514 int wpas_p2p_sd_cancel_request(struct wpa_supplicant *wpa_s, void *req)
1515 {
1516         return p2p_sd_cancel_request(wpa_s->global->p2p, req);
1517 }
1518
1519
1520 void wpas_p2p_sd_response(struct wpa_supplicant *wpa_s, int freq,
1521                           const u8 *dst, u8 dialog_token,
1522                           const struct wpabuf *resp_tlvs)
1523 {
1524         p2p_sd_response(wpa_s->global->p2p, freq, dst, dialog_token,
1525                         resp_tlvs);
1526 }
1527
1528
1529 void wpas_p2p_sd_service_update(struct wpa_supplicant *wpa_s)
1530 {
1531         p2p_sd_service_update(wpa_s->global->p2p);
1532 }
1533
1534
1535 static void wpas_p2p_srv_bonjour_free(struct p2p_srv_bonjour *bsrv)
1536 {
1537         dl_list_del(&bsrv->list);
1538         wpabuf_free(bsrv->query);
1539         wpabuf_free(bsrv->resp);
1540         os_free(bsrv);
1541 }
1542
1543
1544 static void wpas_p2p_srv_upnp_free(struct p2p_srv_upnp *usrv)
1545 {
1546         dl_list_del(&usrv->list);
1547         os_free(usrv->service);
1548         os_free(usrv);
1549 }
1550
1551
1552 void wpas_p2p_service_flush(struct wpa_supplicant *wpa_s)
1553 {
1554         struct p2p_srv_bonjour *bsrv, *bn;
1555         struct p2p_srv_upnp *usrv, *un;
1556
1557         dl_list_for_each_safe(bsrv, bn, &wpa_s->global->p2p_srv_bonjour,
1558                               struct p2p_srv_bonjour, list)
1559                 wpas_p2p_srv_bonjour_free(bsrv);
1560
1561         dl_list_for_each_safe(usrv, un, &wpa_s->global->p2p_srv_upnp,
1562                               struct p2p_srv_upnp, list)
1563                 wpas_p2p_srv_upnp_free(usrv);
1564
1565         wpas_p2p_sd_service_update(wpa_s);
1566 }
1567
1568
1569 int wpas_p2p_service_add_bonjour(struct wpa_supplicant *wpa_s,
1570                                  struct wpabuf *query, struct wpabuf *resp)
1571 {
1572         struct p2p_srv_bonjour *bsrv;
1573
1574         bsrv = wpas_p2p_service_get_bonjour(wpa_s, query);
1575         if (bsrv) {
1576                 wpabuf_free(query);
1577                 wpabuf_free(bsrv->resp);
1578                 bsrv->resp = resp;
1579                 return 0;
1580         }
1581
1582         bsrv = os_zalloc(sizeof(*bsrv));
1583         if (bsrv == NULL)
1584                 return -1;
1585         bsrv->query = query;
1586         bsrv->resp = resp;
1587         dl_list_add(&wpa_s->global->p2p_srv_bonjour, &bsrv->list);
1588
1589         wpas_p2p_sd_service_update(wpa_s);
1590         return 0;
1591 }
1592
1593
1594 int wpas_p2p_service_del_bonjour(struct wpa_supplicant *wpa_s,
1595                                  const struct wpabuf *query)
1596 {
1597         struct p2p_srv_bonjour *bsrv;
1598
1599         bsrv = wpas_p2p_service_get_bonjour(wpa_s, query);
1600         if (bsrv == NULL)
1601                 return -1;
1602         wpas_p2p_srv_bonjour_free(bsrv);
1603         wpas_p2p_sd_service_update(wpa_s);
1604         return 0;
1605 }
1606
1607
1608 int wpas_p2p_service_add_upnp(struct wpa_supplicant *wpa_s, u8 version,
1609                               const char *service)
1610 {
1611         struct p2p_srv_upnp *usrv;
1612
1613         if (wpas_p2p_service_get_upnp(wpa_s, version, service))
1614                 return 0; /* Already listed */
1615         usrv = os_zalloc(sizeof(*usrv));
1616         if (usrv == NULL)
1617                 return -1;
1618         usrv->version = version;
1619         usrv->service = os_strdup(service);
1620         if (usrv->service == NULL) {
1621                 os_free(usrv);
1622                 return -1;
1623         }
1624         dl_list_add(&wpa_s->global->p2p_srv_upnp, &usrv->list);
1625
1626         wpas_p2p_sd_service_update(wpa_s);
1627         return 0;
1628 }
1629
1630
1631 int wpas_p2p_service_del_upnp(struct wpa_supplicant *wpa_s, u8 version,
1632                               const char *service)
1633 {
1634         struct p2p_srv_upnp *usrv;
1635
1636         usrv = wpas_p2p_service_get_upnp(wpa_s, version, service);
1637         if (usrv == NULL)
1638                 return -1;
1639         wpas_p2p_srv_upnp_free(usrv);
1640         wpas_p2p_sd_service_update(wpa_s);
1641         return 0;
1642 }
1643
1644
1645 static void wpas_prov_disc_local_display(struct wpa_supplicant *wpa_s,
1646                                          const u8 *peer, const char *params)
1647 {
1648         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_SHOW_PIN MACSTR " %08d%s",
1649                 MAC2STR(peer), wps_generate_pin(), params);
1650 }
1651
1652
1653 static void wpas_prov_disc_local_keypad(struct wpa_supplicant *wpa_s,
1654                                         const u8 *peer, const char *params)
1655 {
1656         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_ENTER_PIN MACSTR "%s",
1657                 MAC2STR(peer), params);
1658 }
1659
1660
1661 void wpas_prov_disc_req(void *ctx, const u8 *peer, u16 config_methods,
1662                         const u8 *dev_addr, const u8 *pri_dev_type,
1663                         const char *dev_name, u16 supp_config_methods,
1664                         u8 dev_capab, u8 group_capab)
1665 {
1666         struct wpa_supplicant *wpa_s = ctx;
1667         char devtype[WPS_DEV_TYPE_BUFSIZE];
1668         char params[200];
1669         u8 empty_dev_type[8];
1670
1671         if (pri_dev_type == NULL) {
1672                 os_memset(empty_dev_type, 0, sizeof(empty_dev_type));
1673                 pri_dev_type = empty_dev_type;
1674         }
1675         os_snprintf(params, sizeof(params), " p2p_dev_addr=" MACSTR
1676                     " pri_dev_type=%s name='%s' config_methods=0x%x "
1677                     "dev_capab=0x%x group_capab=0x%x",
1678                     MAC2STR(dev_addr),
1679                     wps_dev_type_bin2str(pri_dev_type, devtype,
1680                                          sizeof(devtype)),
1681                     dev_name, supp_config_methods, dev_capab, group_capab);
1682         params[sizeof(params) - 1] = '\0';
1683
1684         if (config_methods & WPS_CONFIG_DISPLAY)
1685                 wpas_prov_disc_local_display(wpa_s, peer, params);
1686         else if (config_methods & WPS_CONFIG_KEYPAD)
1687                 wpas_prov_disc_local_keypad(wpa_s, peer, params);
1688         else if (config_methods & WPS_CONFIG_PUSHBUTTON)
1689                 wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_PBC_REQ MACSTR
1690                         "%s", MAC2STR(peer), params);
1691 }
1692
1693
1694 void wpas_prov_disc_resp(void *ctx, const u8 *peer, u16 config_methods)
1695 {
1696         struct wpa_supplicant *wpa_s = ctx;
1697         if (config_methods & WPS_CONFIG_DISPLAY)
1698                 wpas_prov_disc_local_keypad(wpa_s, peer, "");
1699         else if (config_methods & WPS_CONFIG_KEYPAD)
1700                 wpas_prov_disc_local_display(wpa_s, peer, "");
1701         else if (config_methods & WPS_CONFIG_PUSHBUTTON)
1702                 wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_PBC_RESP MACSTR,
1703                         MAC2STR(peer));
1704 }
1705
1706
1707 static u8 wpas_invitation_process(void *ctx, const u8 *sa, const u8 *bssid,
1708                                   const u8 *go_dev_addr, const u8 *ssid,
1709                                   size_t ssid_len, int *go, u8 *group_bssid,
1710                                   int *force_freq, int persistent_group)
1711 {
1712         struct wpa_supplicant *wpa_s = ctx;
1713         struct wpa_ssid *s;
1714         u8 cur_bssid[ETH_ALEN];
1715         int res;
1716
1717         if (!persistent_group) {
1718                 wpa_printf(MSG_DEBUG, "P2P: Invitation from " MACSTR
1719                            " to join an active group", MAC2STR(sa));
1720                 if (!is_zero_ether_addr(wpa_s->p2p_auth_invite) &&
1721                     (os_memcmp(go_dev_addr, wpa_s->p2p_auth_invite, ETH_ALEN)
1722                      == 0 ||
1723                      os_memcmp(sa, wpa_s->p2p_auth_invite, ETH_ALEN) == 0)) {
1724                         wpa_printf(MSG_DEBUG, "P2P: Accept previously "
1725                                    "authorized invitation");
1726                         goto accept_inv;
1727                 }
1728                 /*
1729                  * Do not accept the invitation automatically; notify user and
1730                  * request approval.
1731                  */
1732                 return P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
1733         }
1734
1735         if (!wpa_s->conf->persistent_reconnect)
1736                 return P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
1737
1738         for (s = wpa_s->conf->ssid; s; s = s->next) {
1739                 if (s->disabled == 2 &&
1740                     os_memcmp(s->bssid, go_dev_addr, ETH_ALEN) == 0 &&
1741                     s->ssid_len == ssid_len &&
1742                     os_memcmp(ssid, s->ssid, ssid_len) == 0)
1743                         break;
1744         }
1745
1746         if (!s) {
1747                 wpa_printf(MSG_DEBUG, "P2P: Invitation from " MACSTR
1748                            " requested reinvocation of an unknown group",
1749                            MAC2STR(sa));
1750                 return P2P_SC_FAIL_UNKNOWN_GROUP;
1751         }
1752
1753         if (s->mode == WPAS_MODE_P2P_GO && !wpas_p2p_create_iface(wpa_s)) {
1754                 *go = 1;
1755                 if (wpa_s->wpa_state >= WPA_AUTHENTICATING) {
1756                         wpa_printf(MSG_DEBUG, "P2P: The only available "
1757                                    "interface is already in use - reject "
1758                                    "invitation");
1759                         return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
1760                 }
1761                 os_memcpy(group_bssid, wpa_s->own_addr, ETH_ALEN);
1762         } else if (s->mode == WPAS_MODE_P2P_GO) {
1763                 *go = 1;
1764                 if (wpas_p2p_add_group_interface(wpa_s, WPA_IF_P2P_GO) < 0)
1765                 {
1766                         wpa_printf(MSG_ERROR, "P2P: Failed to allocate a new "
1767                                    "interface address for the group");
1768                         return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
1769                 }
1770                 os_memcpy(group_bssid, wpa_s->pending_interface_addr,
1771                           ETH_ALEN);
1772         }
1773
1774 accept_inv:
1775         if (wpa_s->current_ssid && wpa_drv_get_bssid(wpa_s, cur_bssid) == 0 &&
1776             wpa_s->assoc_freq) {
1777                 wpa_printf(MSG_DEBUG, "P2P: Trying to force channel to match "
1778                            "the channel we are already using");
1779                 *force_freq = wpa_s->assoc_freq;
1780         }
1781
1782         res = wpa_drv_shared_freq(wpa_s);
1783         if (res > 0) {
1784                 wpa_printf(MSG_DEBUG, "P2P: Trying to force channel to match "
1785                            "with the channel we are already using on a "
1786                            "shared interface");
1787                 *force_freq = res;
1788         }
1789
1790         return P2P_SC_SUCCESS;
1791 }
1792
1793
1794 static void wpas_invitation_received(void *ctx, const u8 *sa, const u8 *bssid,
1795                                      const u8 *ssid, size_t ssid_len,
1796                                      const u8 *go_dev_addr, u8 status,
1797                                      int op_freq)
1798 {
1799         struct wpa_supplicant *wpa_s = ctx;
1800         struct wpa_ssid *s;
1801
1802         for (s = wpa_s->conf->ssid; s; s = s->next) {
1803                 if (s->disabled == 2 &&
1804                     s->ssid_len == ssid_len &&
1805                     os_memcmp(ssid, s->ssid, ssid_len) == 0)
1806                         break;
1807         }
1808
1809         if (status == P2P_SC_SUCCESS) {
1810                 wpa_printf(MSG_DEBUG, "P2P: Invitation from peer " MACSTR
1811                            " was accepted; op_freq=%d MHz",
1812                            MAC2STR(sa), op_freq);
1813                 if (s) {
1814                         wpas_p2p_group_add_persistent(
1815                                 wpa_s, s, s->mode == WPAS_MODE_P2P_GO, 0);
1816                 } else {
1817                         wpas_p2p_join(wpa_s, bssid, go_dev_addr,
1818                                       wpa_s->p2p_wps_method);
1819                 }
1820                 return;
1821         }
1822
1823         if (status != P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE) {
1824                 wpa_printf(MSG_DEBUG, "P2P: Invitation from peer " MACSTR
1825                            " was rejected (status %u)", MAC2STR(sa), status);
1826                 return;
1827         }
1828
1829         if (!s) {
1830                 if (bssid) {
1831                         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RECEIVED
1832                                 "sa=" MACSTR " go_dev_addr=" MACSTR
1833                                 " bssid=" MACSTR " unknown-network",
1834                                 MAC2STR(sa), MAC2STR(go_dev_addr),
1835                                 MAC2STR(bssid));
1836                 } else {
1837                         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RECEIVED
1838                                 "sa=" MACSTR " go_dev_addr=" MACSTR
1839                                 " unknown-network",
1840                                 MAC2STR(sa), MAC2STR(go_dev_addr));
1841                 }
1842                 return;
1843         }
1844
1845         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RECEIVED "sa=" MACSTR
1846                 " persistent=%d", MAC2STR(sa), s->id);
1847 }
1848
1849
1850 static void wpas_invitation_result(void *ctx, int status, const u8 *bssid)
1851 {
1852         struct wpa_supplicant *wpa_s = ctx;
1853         struct wpa_ssid *ssid;
1854
1855         if (bssid) {
1856                 wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RESULT
1857                         "status=%d " MACSTR,
1858                         status, MAC2STR(bssid));
1859         } else {
1860                 wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RESULT
1861                         "status=%d ", status);
1862         }
1863
1864         if (wpa_s->pending_invite_ssid_id == -1)
1865                 return; /* Invitation to active group */
1866
1867         if (status != P2P_SC_SUCCESS) {
1868                 wpas_p2p_remove_pending_group_interface(wpa_s);
1869                 return;
1870         }
1871
1872         ssid = wpa_config_get_network(wpa_s->conf,
1873                                       wpa_s->pending_invite_ssid_id);
1874         if (ssid == NULL) {
1875                 wpa_printf(MSG_ERROR, "P2P: Could not find persistent group "
1876                            "data matching with invitation");
1877                 return;
1878         }
1879
1880         wpas_p2p_group_add_persistent(wpa_s, ssid,
1881                                       ssid->mode == WPAS_MODE_P2P_GO, 0);
1882 }
1883
1884
1885 static int wpas_p2p_setup_channels(struct wpa_supplicant *wpa_s,
1886                                    struct p2p_config *p2p)
1887 {
1888         struct hostapd_hw_modes *modes;
1889         u16 num_modes, flags;
1890         int i, cla;
1891         int band24 = 0, band5_low = 0, band5_high = 0;
1892
1893         /* TODO: more detailed selection of channels within reg_class based on
1894          * driver capabilities */
1895
1896         modes = wpa_drv_get_hw_feature_data(wpa_s, &num_modes, &flags);
1897         if (modes == NULL) {
1898                 wpa_printf(MSG_DEBUG, "P2P: Driver did not support fetching "
1899                            "of all supported channels; assume dualband "
1900                            "support");
1901                 band24 = band5_low = band5_high = 1;
1902         } else {
1903                 for (i = 0; i < num_modes; i++) {
1904                         struct hostapd_hw_modes *mode;
1905                         mode = &modes[i];
1906                         if (mode->mode == HOSTAPD_MODE_IEEE80211G) {
1907                                 band24 = 1;
1908                         } else if (mode->mode == HOSTAPD_MODE_IEEE80211A) {
1909                                 int j;
1910                                 for (j = 0; j < mode->num_channels; j++) {
1911                                         struct hostapd_channel_data *ch;
1912                                         ch = &mode->channels[j];
1913                                         if (ch->chan == 36)
1914                                                 band5_low = 1;
1915                                         else if (ch->chan == 157)
1916                                                 band5_high = 1;
1917                                 }
1918                         }
1919                 }
1920         }
1921
1922         cla = 0;
1923
1924         if (band24) {
1925                 wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for "
1926                            "2.4 GHz band");
1927
1928                 /* Operating class 81 - 2.4 GHz band channels 1..13 */
1929                 p2p->channels.reg_class[cla].reg_class = 81;
1930                 p2p->channels.reg_class[cla].channels = 13;
1931                 for (i = 0; i < 13; i++)
1932                         p2p->channels.reg_class[cla].channel[i] = i + 1;
1933                 cla++;
1934
1935                 /* Operating class 82 - 2.4 GHz band channel 14 */
1936                 p2p->channels.reg_class[cla].reg_class = 82;
1937                 p2p->channels.reg_class[cla].channels = 1;
1938                 p2p->channels.reg_class[cla].channel[0] = 14;
1939                 cla++;
1940
1941 #if 0
1942                 /* Operating class 83 - 2.4 GHz band channels 1..9; 40 MHz */
1943                 p2p->channels.reg_class[cla].reg_class = 83;
1944                 p2p->channels.reg_class[cla].channels = 9;
1945                 for (i = 0; i < 9; i++)
1946                         p2p->channels.reg_class[cla].channel[i] = i + 1;
1947                 cla++;
1948
1949                 /* Operating class 84 - 2.4 GHz band channels 5..13; 40 MHz */
1950                 p2p->channels.reg_class[cla].reg_class = 84;
1951                 p2p->channels.reg_class[cla].channels = 9;
1952                 for (i = 0; i < 9; i++)
1953                         p2p->channels.reg_class[cla].channel[i] = i + 5;
1954                 cla++;
1955 #endif
1956         }
1957
1958         if (band5_low) {
1959                 wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for "
1960                            "lower 5 GHz band");
1961
1962                 /* Operating class 115 - 5 GHz, channels 36-48 */
1963                 p2p->channels.reg_class[cla].reg_class = 115;
1964                 p2p->channels.reg_class[cla].channels = 4;
1965                 p2p->channels.reg_class[cla].channel[0] = 36;
1966                 p2p->channels.reg_class[cla].channel[1] = 40;
1967                 p2p->channels.reg_class[cla].channel[2] = 44;
1968                 p2p->channels.reg_class[cla].channel[3] = 48;
1969                 cla++;
1970
1971 #if 0
1972                 /* Operating class 116 - 5 GHz, channels 36,44; 40 MHz */
1973                 p2p->channels.reg_class[cla].reg_class = 116;
1974                 p2p->channels.reg_class[cla].channels = 2;
1975                 p2p->channels.reg_class[cla].channel[0] = 36;
1976                 p2p->channels.reg_class[cla].channel[1] = 44;
1977                 cla++;
1978
1979                 /* Operating class 117 - 5 GHz, channels 40,48; 40 MHz */
1980                 p2p->channels.reg_class[cla].reg_class = 117;
1981                 p2p->channels.reg_class[cla].channels = 2;
1982                 p2p->channels.reg_class[cla].channel[0] = 40;
1983                 p2p->channels.reg_class[cla].channel[1] = 48;
1984                 cla++;
1985 #endif
1986         }
1987
1988         if (band5_high) {
1989                 wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for "
1990                            "higher 5 GHz band");
1991
1992                 /* Operating class 124 - 5 GHz, channels 149,153,157,161 */
1993                 p2p->channels.reg_class[cla].reg_class = 124;
1994                 p2p->channels.reg_class[cla].channels = 4;
1995                 p2p->channels.reg_class[cla].channel[0] = 149;
1996                 p2p->channels.reg_class[cla].channel[1] = 153;
1997                 p2p->channels.reg_class[cla].channel[2] = 157;
1998                 p2p->channels.reg_class[cla].channel[3] = 161;
1999                 cla++;
2000
2001 #if 0
2002                 /* Operating class 126 - 5 GHz, channels 149,157; 40 MHz */
2003                 p2p->channels.reg_class[cla].reg_class = 126;
2004                 p2p->channels.reg_class[cla].channels = 2;
2005                 p2p->channels.reg_class[cla].channel[0] = 149;
2006                 p2p->channels.reg_class[cla].channel[1] = 157;
2007                 cla++;
2008
2009                 /* Operating class 127 - 5 GHz, channels 153,161; 40 MHz */
2010                 p2p->channels.reg_class[cla].reg_class = 127;
2011                 p2p->channels.reg_class[cla].channels = 2;
2012                 p2p->channels.reg_class[cla].channel[0] = 153;
2013                 p2p->channels.reg_class[cla].channel[1] = 161;
2014                 cla++;
2015 #endif
2016         }
2017
2018         p2p->channels.reg_classes = cla;
2019
2020         if (modes)
2021                 ieee80211_sta_free_hw_features(modes, num_modes);
2022
2023         return 0;
2024 }
2025
2026
2027 static int wpas_get_noa(void *ctx, const u8 *interface_addr, u8 *buf,
2028                         size_t buf_len)
2029 {
2030         struct wpa_supplicant *wpa_s = ctx;
2031
2032         for (wpa_s = wpa_s->global->ifaces; wpa_s; wpa_s = wpa_s->next) {
2033                 if (os_memcmp(wpa_s->own_addr, interface_addr, ETH_ALEN) == 0)
2034                         break;
2035         }
2036         if (wpa_s == NULL)
2037                 return -1;
2038
2039         return wpa_drv_get_noa(wpa_s, buf, buf_len);
2040 }
2041
2042
2043 /**
2044  * wpas_p2p_init - Initialize P2P module for %wpa_supplicant
2045  * @global: Pointer to global data from wpa_supplicant_init()
2046  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2047  * Returns: 0 on success, -1 on failure
2048  */
2049 int wpas_p2p_init(struct wpa_global *global, struct wpa_supplicant *wpa_s)
2050 {
2051         struct p2p_config p2p;
2052         unsigned int r;
2053         int i;
2054
2055         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_CAPABLE))
2056                 return 0;
2057
2058 #ifdef CONFIG_CLIENT_MLME
2059         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_MGMT)) {
2060                 wpa_s->mlme.public_action_cb = p2p_rx_action_mlme;
2061                 wpa_s->mlme.public_action_cb_ctx = wpa_s;
2062         }
2063 #endif /* CONFIG_CLIENT_MLME */
2064
2065         if (wpa_drv_disable_11b_rates(wpa_s, 1) < 0) {
2066                 wpa_printf(MSG_DEBUG, "P2P: Failed to disable 11b rates");
2067                 /* Continue anyway; this is not really a fatal error */
2068         }
2069
2070         if (global->p2p)
2071                 return 0;
2072
2073         os_memset(&p2p, 0, sizeof(p2p));
2074         p2p.msg_ctx = wpa_s;
2075         p2p.cb_ctx = wpa_s;
2076         p2p.p2p_scan = wpas_p2p_scan;
2077         p2p.send_action = wpas_send_action;
2078         p2p.send_action_done = wpas_send_action_done;
2079         p2p.go_neg_completed = wpas_go_neg_completed;
2080         p2p.go_neg_req_rx = wpas_go_neg_req_rx;
2081         p2p.dev_found = wpas_dev_found;
2082         p2p.start_listen = wpas_start_listen;
2083         p2p.stop_listen = wpas_stop_listen;
2084         p2p.send_probe_resp = wpas_send_probe_resp;
2085         p2p.sd_request = wpas_sd_request;
2086         p2p.sd_response = wpas_sd_response;
2087         p2p.prov_disc_req = wpas_prov_disc_req;
2088         p2p.prov_disc_resp = wpas_prov_disc_resp;
2089         p2p.invitation_process = wpas_invitation_process;
2090         p2p.invitation_received = wpas_invitation_received;
2091         p2p.invitation_result = wpas_invitation_result;
2092         p2p.get_noa = wpas_get_noa;
2093
2094         os_memcpy(wpa_s->global->p2p_dev_addr, wpa_s->own_addr, ETH_ALEN);
2095         os_memcpy(p2p.dev_addr, wpa_s->own_addr, ETH_ALEN);
2096         p2p.dev_name = wpa_s->conf->device_name;
2097
2098         if (wpa_s->conf->p2p_listen_reg_class &&
2099             wpa_s->conf->p2p_listen_channel) {
2100                 p2p.reg_class = wpa_s->conf->p2p_listen_reg_class;
2101                 p2p.channel = wpa_s->conf->p2p_listen_channel;
2102         } else {
2103                 p2p.reg_class = 81;
2104                 /*
2105                  * Pick one of the social channels randomly as the listen
2106                  * channel.
2107                  */
2108                 os_get_random((u8 *) &r, sizeof(r));
2109                 p2p.channel = 1 + (r % 3) * 5;
2110         }
2111
2112         if (wpa_s->conf->p2p_oper_reg_class &&
2113             wpa_s->conf->p2p_oper_channel) {
2114                 p2p.op_reg_class = wpa_s->conf->p2p_oper_reg_class;
2115                 p2p.op_channel = wpa_s->conf->p2p_oper_channel;
2116         } else {
2117                 p2p.op_reg_class = 81;
2118                 /*
2119                  * For initial tests, pick the operation channel randomly.
2120                  * TODO: Use scan results (etc.) to select the best channel.
2121                  */
2122                 p2p.op_channel = 1 + r % 11;
2123         }
2124         wpa_printf(MSG_DEBUG, "P2P: Own listen channel: %d  "
2125                    "Own preferred operation channel: %d",
2126                    p2p.channel, p2p.op_channel);
2127         if (wpa_s->conf->country[0] && wpa_s->conf->country[1]) {
2128                 os_memcpy(p2p.country, wpa_s->conf->country, 2);
2129                 p2p.country[2] = 0x04;
2130         } else
2131                 os_memcpy(p2p.country, "US\x04", 3);
2132
2133         if (wpas_p2p_setup_channels(wpa_s, &p2p)) {
2134                 wpa_printf(MSG_ERROR, "P2P: Failed to configure supported "
2135                            "channel list");
2136                 return -1;
2137         }
2138
2139         if (wpa_s->conf->device_type &&
2140             wps_dev_type_str2bin(wpa_s->conf->device_type, p2p.pri_dev_type) <
2141             0) {
2142                 wpa_printf(MSG_ERROR, "P2P: Invalid device_type");
2143                 return -1;
2144         }
2145
2146         for (i = 0; i < MAX_SEC_DEVICE_TYPES; i++) {
2147                 if (wpa_s->conf->sec_device_type[i] == NULL)
2148                         continue;
2149                 if (wps_dev_type_str2bin(
2150                             wpa_s->conf->sec_device_type[i],
2151                             p2p.sec_dev_type[p2p.num_sec_dev_types]) < 0) {
2152                         wpa_printf(MSG_ERROR, "P2P: Invalid sec_device_type");
2153                         return -1;
2154                 }
2155                 p2p.num_sec_dev_types++;
2156                 if (p2p.num_sec_dev_types == P2P_SEC_DEVICE_TYPES)
2157                         break;
2158         }
2159
2160         p2p.concurrent_operations = !!(wpa_s->drv_flags &
2161                                        WPA_DRIVER_FLAGS_P2P_CONCURRENT);
2162
2163         p2p.max_peers = wpa_s->max_stations ? wpa_s->max_stations : 100;
2164
2165         if (wpa_s->conf->p2p_ssid_postfix) {
2166                 p2p.ssid_postfix_len =
2167                         os_strlen(wpa_s->conf->p2p_ssid_postfix);
2168                 if (p2p.ssid_postfix_len > sizeof(p2p.ssid_postfix))
2169                         p2p.ssid_postfix_len = sizeof(p2p.ssid_postfix);
2170                 os_memcpy(p2p.ssid_postfix, wpa_s->conf->p2p_ssid_postfix,
2171                           p2p.ssid_postfix_len);
2172         }
2173
2174         global->p2p = p2p_init(&p2p);
2175         if (global->p2p == NULL)
2176                 return -1;
2177
2178         return 0;
2179 }
2180
2181
2182 /**
2183  * wpas_p2p_deinit - Deinitialize per-interface P2P data
2184  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2185  *
2186  * This function deinitialize per-interface P2P data.
2187  */
2188 void wpas_p2p_deinit(struct wpa_supplicant *wpa_s)
2189 {
2190         if (wpa_s->driver && wpa_s->drv_priv)
2191                 wpa_drv_probe_req_report(wpa_s, 0);
2192         os_free(wpa_s->go_params);
2193         wpa_s->go_params = NULL;
2194         wpabuf_free(wpa_s->pending_action_tx);
2195         wpa_s->pending_action_tx = NULL;
2196         eloop_cancel_timeout(wpas_send_action_cb, wpa_s, NULL);
2197         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s, NULL);
2198         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
2199         wpa_s->p2p_long_listen = 0;
2200         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
2201         wpas_p2p_remove_pending_group_interface(wpa_s);
2202
2203         /* TODO: remove group interface from the driver if this wpa_s instance
2204          * is on top of a P2P group interface */
2205 }
2206
2207
2208 /**
2209  * wpas_p2p_deinit_global - Deinitialize global P2P module
2210  * @global: Pointer to global data from wpa_supplicant_init()
2211  *
2212  * This function deinitializes the global (per device) P2P module.
2213  */
2214 void wpas_p2p_deinit_global(struct wpa_global *global)
2215 {
2216         struct wpa_supplicant *wpa_s, *tmp;
2217         char *ifname;
2218
2219         if (global->p2p == NULL)
2220                 return;
2221
2222         /* Remove remaining P2P group interfaces */
2223         wpa_s = global->ifaces;
2224         if (wpa_s)
2225                 wpas_p2p_service_flush(wpa_s);
2226         while (wpa_s && wpa_s->p2p_group_interface != NOT_P2P_GROUP_INTERFACE)
2227                 wpa_s = wpa_s->next;
2228         while (wpa_s) {
2229                 enum wpa_driver_if_type type;
2230                 tmp = global->ifaces;
2231                 while (tmp &&
2232                        (tmp == wpa_s ||
2233                         tmp->p2p_group_interface == NOT_P2P_GROUP_INTERFACE)) {
2234                         tmp = tmp->next;
2235                 }
2236                 if (tmp == NULL)
2237                         break;
2238                 ifname = os_strdup(tmp->ifname);
2239                 type = wpas_p2p_if_type(tmp->p2p_group_interface);
2240                 wpa_supplicant_remove_iface(global, tmp);
2241                 if (ifname)
2242                         wpa_drv_if_remove(wpa_s, type, ifname);
2243                 os_free(ifname);
2244         }
2245
2246         p2p_deinit(global->p2p);
2247         global->p2p = NULL;
2248 }
2249
2250
2251 static int wpas_p2p_create_iface(struct wpa_supplicant *wpa_s)
2252 {
2253         if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_DEDICATED_INTERFACE)
2254                 return 1; /* P2P group requires a new interface in every case
2255                            */
2256         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_CONCURRENT))
2257                 return 0; /* driver does not support concurrent operations */
2258         if (wpa_s->global->ifaces->next)
2259                 return 1; /* more that one interface already in use */
2260         if (wpa_s->wpa_state >= WPA_AUTHENTICATING)
2261                 return 1; /* this interface is already in use */
2262         return 0;
2263 }
2264
2265
2266 static int wpas_p2p_start_go_neg(struct wpa_supplicant *wpa_s,
2267                                  const u8 *peer_addr,
2268                                  enum p2p_wps_method wps_method,
2269                                  int go_intent, const u8 *own_interface_addr,
2270                                  unsigned int force_freq, int persistent_group)
2271 {
2272         return p2p_connect(wpa_s->global->p2p, peer_addr, wps_method,
2273                            go_intent, own_interface_addr, force_freq,
2274                            persistent_group);
2275 }
2276
2277
2278 static int wpas_p2p_auth_go_neg(struct wpa_supplicant *wpa_s,
2279                                 const u8 *peer_addr,
2280                                 enum p2p_wps_method wps_method,
2281                                 int go_intent, const u8 *own_interface_addr,
2282                                 unsigned int force_freq, int persistent_group)
2283 {
2284         return p2p_authorize(wpa_s->global->p2p, peer_addr, wps_method,
2285                              go_intent, own_interface_addr, force_freq,
2286                              persistent_group);
2287 }
2288
2289
2290 static void wpas_p2p_scan_res_join(struct wpa_supplicant *wpa_s,
2291                                    struct wpa_scan_results *scan_res)
2292 {
2293         struct wpa_bss *bss;
2294
2295         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
2296
2297         if (wpa_s->global->p2p_disabled)
2298                 return;
2299
2300         wpa_printf(MSG_DEBUG, "P2P: Scan results received (%d BSS) for join",
2301                    scan_res ? (int) scan_res->num : -1);
2302
2303         if (scan_res)
2304                 wpas_p2p_scan_res_handler(wpa_s, scan_res);
2305
2306         bss = wpa_bss_get_bssid(wpa_s, wpa_s->pending_join_iface_addr);
2307         if (bss) {
2308                 u16 method;
2309
2310                 wpa_printf(MSG_DEBUG, "P2P: Send Provision Discovery Request "
2311                            "prior to joining an existing group (GO " MACSTR
2312                            " freq=%u MHz)",
2313                            MAC2STR(wpa_s->pending_join_dev_addr), bss->freq);
2314                 wpa_s->pending_pd_before_join = 1;
2315
2316                 switch (wpa_s->pending_join_wps_method) {
2317                 case WPS_PIN_LABEL:
2318                 case WPS_PIN_DISPLAY:
2319                         method = WPS_CONFIG_KEYPAD;
2320                         break;
2321                 case WPS_PIN_KEYPAD:
2322                         method = WPS_CONFIG_DISPLAY;
2323                         break;
2324                 case WPS_PBC:
2325                         method = WPS_CONFIG_PUSHBUTTON;
2326                         break;
2327                 default:
2328                         method = 0;
2329                         break;
2330                 }
2331
2332                 if (p2p_prov_disc_req(wpa_s->global->p2p,
2333                                       wpa_s->pending_join_dev_addr, method, 1)
2334                     < 0) {
2335                         wpa_printf(MSG_DEBUG, "P2P: Failed to send Provision "
2336                                    "Discovery Request before joining an "
2337                                    "existing group");
2338                         wpa_s->pending_pd_before_join = 0;
2339                         goto start;
2340                 }
2341
2342                 /*
2343                  * Actual join operation will be started from the Action frame
2344                  * TX status callback.
2345                  */
2346                 return;
2347         }
2348
2349         wpa_printf(MSG_DEBUG, "P2P: Target BSS/GO not yet in BSS table - "
2350                    "cannot send Provision Discovery Request");
2351
2352 start:
2353         /* Start join operation immediately */
2354         wpas_p2p_join_start(wpa_s);
2355 }
2356
2357
2358 static void wpas_p2p_join_scan(void *eloop_ctx, void *timeout_ctx)
2359 {
2360         struct wpa_supplicant *wpa_s = eloop_ctx;
2361         int ret;
2362         struct wpa_driver_scan_params params;
2363         struct wpabuf *wps_ie, *ies;
2364
2365         os_memset(&params, 0, sizeof(params));
2366
2367         /* P2P Wildcard SSID */
2368         params.num_ssids = 1;
2369         params.ssids[0].ssid = (u8 *) P2P_WILDCARD_SSID;
2370         params.ssids[0].ssid_len = P2P_WILDCARD_SSID_LEN;
2371
2372         wpa_s->wps->dev.p2p = 1;
2373         wps_ie = wps_build_probe_req_ie(0, &wpa_s->wps->dev, wpa_s->wps->uuid,
2374                                         WPS_REQ_ENROLLEE);
2375         if (wps_ie == NULL) {
2376                 wpas_p2p_scan_res_join(wpa_s, NULL);
2377                 return;
2378         }
2379
2380         ies = wpabuf_alloc(wpabuf_len(wps_ie) + 100);
2381         if (ies == NULL) {
2382                 wpabuf_free(wps_ie);
2383                 wpas_p2p_scan_res_join(wpa_s, NULL);
2384                 return;
2385         }
2386         wpabuf_put_buf(ies, wps_ie);
2387         wpabuf_free(wps_ie);
2388
2389         p2p_scan_ie(wpa_s->global->p2p, ies);
2390
2391         params.extra_ies = wpabuf_head(ies);
2392         params.extra_ies_len = wpabuf_len(ies);
2393
2394         /*
2395          * Run a scan to update BSS table and start Provision Discovery once
2396          * the new scan results become available.
2397          */
2398         wpa_s->scan_res_handler = wpas_p2p_scan_res_join;
2399         if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
2400                 ret = ieee80211_sta_req_scan(wpa_s, &params);
2401         else
2402                 ret = wpa_drv_scan(wpa_s, &params);
2403
2404         wpabuf_free(ies);
2405
2406         if (ret) {
2407                 wpa_printf(MSG_DEBUG, "P2P: Failed to start scan for join - "
2408                            "try again later");
2409                 eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
2410                 eloop_register_timeout(1, 0, wpas_p2p_join_scan, wpa_s, NULL);
2411         }
2412 }
2413
2414
2415 static int wpas_p2p_join(struct wpa_supplicant *wpa_s, const u8 *iface_addr,
2416                          const u8 *dev_addr, enum p2p_wps_method wps_method)
2417 {
2418         wpa_printf(MSG_DEBUG, "P2P: Request to join existing group (iface "
2419                    MACSTR " dev " MACSTR ")",
2420                    MAC2STR(iface_addr), MAC2STR(dev_addr));
2421
2422         os_memcpy(wpa_s->pending_join_iface_addr, iface_addr, ETH_ALEN);
2423         os_memcpy(wpa_s->pending_join_dev_addr, dev_addr, ETH_ALEN);
2424         wpa_s->pending_join_wps_method = wps_method;
2425
2426         /* Make sure we are not running find during connection establishment */
2427         wpas_p2p_stop_find(wpa_s);
2428
2429         wpas_p2p_join_scan(wpa_s, NULL);
2430         return 0;
2431 }
2432
2433
2434 static int wpas_p2p_join_start(struct wpa_supplicant *wpa_s)
2435 {
2436         struct wpa_supplicant *group;
2437         struct p2p_go_neg_results res;
2438
2439         group = wpas_p2p_get_group_iface(wpa_s, 0, 0);
2440         if (group == NULL)
2441                 return -1;
2442         if (group != wpa_s) {
2443                 os_memcpy(group->p2p_pin, wpa_s->p2p_pin,
2444                           sizeof(group->p2p_pin));
2445                 group->p2p_wps_method = wpa_s->p2p_wps_method;
2446         }
2447
2448         group->p2p_in_provisioning = 1;
2449
2450         os_memset(&res, 0, sizeof(res));
2451         os_memcpy(res.peer_interface_addr, wpa_s->pending_join_iface_addr,
2452                   ETH_ALEN);
2453         res.wps_method = wpa_s->pending_join_wps_method;
2454         wpas_start_wps_enrollee(group, &res);
2455
2456         return 0;
2457 }
2458
2459
2460 /**
2461  * wpas_p2p_connect - Request P2P Group Formation to be started
2462  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2463  * @peer_addr: Address of the peer P2P Device
2464  * @pin: PIN to use during provisioning or %NULL to indicate PBC mode
2465  * @persistent_group: Whether to create a persistent group
2466  * @join: Whether to join an existing group (as a client) instead of starting
2467  *      Group Owner negotiation; @peer_addr is BSSID in that case
2468  * @auth: Whether to only authorize the connection instead of doing that and
2469  *      initiating Group Owner negotiation
2470  * @go_intent: GO Intent or -1 to use default
2471  * @freq: Frequency for the group or 0 for auto-selection
2472  * Returns: 0 or new PIN (if pin was %NULL) on success, -1 on failure
2473  */
2474 int wpas_p2p_connect(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
2475                      const char *pin, enum p2p_wps_method wps_method,
2476                      int persistent_group, int join, int auth, int go_intent,
2477                      int freq)
2478 {
2479         int force_freq = 0;
2480         u8 bssid[ETH_ALEN];
2481         int ret = 0;
2482         enum wpa_driver_if_type iftype;
2483
2484         if (go_intent < 0)
2485                 go_intent = wpa_s->conf->p2p_go_intent;
2486
2487         if (!auth)
2488                 wpa_s->p2p_long_listen = 0;
2489
2490         wpa_s->p2p_wps_method = wps_method;
2491
2492         if (pin)
2493                 os_strlcpy(wpa_s->p2p_pin, pin, sizeof(wpa_s->p2p_pin));
2494         else if (wps_method == WPS_PIN_DISPLAY) {
2495                 ret = wps_generate_pin();
2496                 os_snprintf(wpa_s->p2p_pin, sizeof(wpa_s->p2p_pin), "%08d",
2497                             ret);
2498                 wpa_printf(MSG_DEBUG, "P2P: Randomly generated PIN: %s",
2499                            wpa_s->p2p_pin);
2500         } else
2501                 wpa_s->p2p_pin[0] = '\0';
2502
2503         if (join) {
2504                 u8 iface_addr[ETH_ALEN], dev_addr[ETH_ALEN];
2505                 if (auth) {
2506                         wpa_printf(MSG_DEBUG, "P2P: Authorize invitation to "
2507                                    "connect a running group from " MACSTR,
2508                                    MAC2STR(peer_addr));
2509                         os_memcpy(wpa_s->p2p_auth_invite, peer_addr, ETH_ALEN);
2510                         return ret;
2511                 }
2512                 os_memcpy(dev_addr, peer_addr, ETH_ALEN);
2513                 if (p2p_get_interface_addr(wpa_s->global->p2p, peer_addr,
2514                                            iface_addr) < 0) {
2515                         os_memcpy(iface_addr, peer_addr, ETH_ALEN);
2516                         p2p_get_dev_addr(wpa_s->global->p2p, peer_addr,
2517                                          dev_addr);
2518                 }
2519                 if (wpas_p2p_join(wpa_s, iface_addr, dev_addr, wps_method) <
2520                     0)
2521                         return -1;
2522                 return ret;
2523         }
2524
2525         if (freq > 0)
2526                 force_freq = freq;
2527         else if (wpa_s->current_ssid && wpa_drv_get_bssid(wpa_s, bssid) == 0 &&
2528                  wpa_s->assoc_freq)
2529                 force_freq = wpa_s->assoc_freq;
2530         else {
2531                 force_freq = wpa_drv_shared_freq(wpa_s);
2532                 if (force_freq < 0)
2533                         force_freq = 0;
2534         }
2535
2536         if (force_freq > 0) {
2537                 wpa_printf(MSG_DEBUG, "P2P: Trying to force us to use the "
2538                            "channel we are already using (%u MHz) on another "
2539                            "interface", force_freq);
2540         }
2541
2542         wpa_s->create_p2p_iface = wpas_p2p_create_iface(wpa_s);
2543
2544         if (!wpa_s->create_p2p_iface) {
2545                 if (auth) {
2546                         if (wpas_p2p_auth_go_neg(wpa_s, peer_addr, wps_method,
2547                                                  go_intent, wpa_s->own_addr,
2548                                                  force_freq, persistent_group)
2549                             < 0)
2550                                 return -1;
2551                         return ret;
2552                 }
2553                 if (wpas_p2p_start_go_neg(wpa_s, peer_addr, wps_method,
2554                                           go_intent, wpa_s->own_addr,
2555                                           force_freq, persistent_group) < 0)
2556                         return -1;
2557                 return ret;
2558         }
2559
2560         /* Prepare to add a new interface for the group */
2561         iftype = WPA_IF_P2P_GROUP;
2562         if (join)
2563                 iftype = WPA_IF_P2P_CLIENT;
2564         else if (go_intent == 15)
2565                 iftype = WPA_IF_P2P_GO;
2566         if (wpas_p2p_add_group_interface(wpa_s, iftype) < 0) {
2567                 wpa_printf(MSG_ERROR, "P2P: Failed to allocate a new "
2568                            "interface for the group");
2569                 return -1;
2570         }
2571
2572         if (auth) {
2573                 if (wpas_p2p_auth_go_neg(wpa_s, peer_addr, wps_method,
2574                                          go_intent,
2575                                          wpa_s->pending_interface_addr,
2576                                          force_freq, persistent_group) < 0)
2577                         return -1;
2578                 return ret;
2579         }
2580         if (wpas_p2p_start_go_neg(wpa_s, peer_addr, wps_method, go_intent,
2581                                   wpa_s->pending_interface_addr,
2582                                   force_freq, persistent_group) < 0) {
2583                 wpas_p2p_remove_pending_group_interface(wpa_s);
2584                 return -1;
2585         }
2586         return ret;
2587 }
2588
2589
2590 /**
2591  * wpas_p2p_remain_on_channel_cb - Indication of remain-on-channel start
2592  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2593  * @freq: Frequency of the channel in MHz
2594  * @duration: Duration of the stay on the channel in milliseconds
2595  *
2596  * This callback is called when the driver indicates that it has started the
2597  * requested remain-on-channel duration.
2598  */
2599 void wpas_p2p_remain_on_channel_cb(struct wpa_supplicant *wpa_s,
2600                                    unsigned int freq, unsigned int duration)
2601 {
2602         wpa_s->roc_waiting_drv_freq = 0;
2603         wpa_s->off_channel_freq = freq;
2604         wpas_send_action_cb(wpa_s, NULL);
2605         if (wpa_s->off_channel_freq == wpa_s->pending_listen_freq) {
2606                 p2p_listen_cb(wpa_s->global->p2p, wpa_s->pending_listen_freq,
2607                               wpa_s->pending_listen_duration);
2608                 wpa_s->pending_listen_freq = 0;
2609         }
2610 }
2611
2612
2613 static int wpas_p2p_listen_start(struct wpa_supplicant *wpa_s,
2614                                  unsigned int timeout)
2615 {
2616         /* Limit maximum Listen state time based on driver limitation. */
2617         if (timeout > wpa_s->max_remain_on_chan)
2618                 timeout = wpa_s->max_remain_on_chan;
2619
2620         return p2p_listen(wpa_s->global->p2p, timeout);
2621 }
2622
2623
2624 /**
2625  * wpas_p2p_cancel_remain_on_channel_cb - Remain-on-channel timeout
2626  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2627  * @freq: Frequency of the channel in MHz
2628  *
2629  * This callback is called when the driver indicates that a remain-on-channel
2630  * operation has been completed, i.e., the duration on the requested channel
2631  * has timed out.
2632  */
2633 void wpas_p2p_cancel_remain_on_channel_cb(struct wpa_supplicant *wpa_s,
2634                                           unsigned int freq)
2635 {
2636         wpa_s->off_channel_freq = 0;
2637         if (p2p_listen_end(wpa_s->global->p2p, freq) > 0)
2638                 return; /* P2P module started a new operation */
2639         if (wpa_s->pending_action_tx)
2640                 return;
2641         if (wpa_s->p2p_long_listen > 0)
2642                 wpa_s->p2p_long_listen -= 5;
2643         if (wpa_s->p2p_long_listen > 0) {
2644                 wpa_printf(MSG_DEBUG, "P2P: Continuing long Listen state");
2645                 wpas_p2p_listen_start(wpa_s, wpa_s->p2p_long_listen * 1000);
2646         }
2647 }
2648
2649
2650 /**
2651  * wpas_p2p_group_remove - Remove a P2P group
2652  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2653  * @ifname: Network interface name of the group interface or "*" to remove all
2654  *      groups
2655  * Returns: 0 on success, -1 on failure
2656  *
2657  * This function is used to remove a P2P group. This can be used to disconnect
2658  * from a group in which the local end is a P2P Client or to end a P2P Group in
2659  * case the local end is the Group Owner. If a virtual network interface was
2660  * created for this group, that interface will be removed. Otherwise, only the
2661  * configured P2P group network will be removed from the interface.
2662  */
2663 int wpas_p2p_group_remove(struct wpa_supplicant *wpa_s, const char *ifname)
2664 {
2665         struct wpa_global *global = wpa_s->global;
2666
2667         if (os_strcmp(ifname, "*") == 0) {
2668                 struct wpa_supplicant *prev;
2669                 wpa_s = global->ifaces;
2670                 while (wpa_s) {
2671                         prev = wpa_s;
2672                         wpa_s = wpa_s->next;
2673                         wpas_p2p_group_delete(prev);
2674                 }
2675                 return 0;
2676         }
2677
2678         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
2679                 if (os_strcmp(wpa_s->ifname, ifname) == 0)
2680                         break;
2681         }
2682
2683         if (wpa_s == NULL)
2684                 return -1;
2685
2686         wpas_p2p_group_delete(wpa_s);
2687
2688         return 0;
2689 }
2690
2691
2692 static void wpas_p2p_init_go_params(struct wpa_supplicant *wpa_s,
2693                                     struct p2p_go_neg_results *params,
2694                                     int freq)
2695 {
2696         u8 bssid[ETH_ALEN];
2697         int res;
2698
2699         os_memset(params, 0, sizeof(*params));
2700         params->role_go = 1;
2701         params->freq = 2412;
2702         if (freq)
2703                 params->freq = freq;
2704         else if (wpa_s->conf->p2p_oper_reg_class == 81 &&
2705                  wpa_s->conf->p2p_oper_channel >= 1 &&
2706                  wpa_s->conf->p2p_oper_channel <= 11)
2707                 params->freq = 2407 + 5 * wpa_s->conf->p2p_oper_channel;
2708         else if (wpa_s->conf->p2p_oper_reg_class == 115 ||
2709                  wpa_s->conf->p2p_oper_reg_class == 118)
2710                 params->freq = 5000 + 5 * wpa_s->conf->p2p_oper_channel;
2711         if (wpa_s->current_ssid && wpa_drv_get_bssid(wpa_s, bssid) == 0 &&
2712             wpa_s->assoc_freq && !freq) {
2713                 wpa_printf(MSG_DEBUG, "P2P: Force GO on the channel we are "
2714                            "already using");
2715                 params->freq = wpa_s->assoc_freq;
2716         }
2717
2718         res = wpa_drv_shared_freq(wpa_s);
2719         if (res > 0 && !freq) {
2720                 wpa_printf(MSG_DEBUG, "P2P: Force GO on the channel we are "
2721                            "already using on a shared interface");
2722                 params->freq = res;
2723         }
2724 }
2725
2726
2727 static struct wpa_supplicant *
2728 wpas_p2p_get_group_iface(struct wpa_supplicant *wpa_s, int addr_allocated,
2729                          int go)
2730 {
2731         struct wpa_supplicant *group_wpa_s;
2732
2733         if (!wpas_p2p_create_iface(wpa_s))
2734                 return wpa_s;
2735
2736         if (wpas_p2p_add_group_interface(wpa_s, go ? WPA_IF_P2P_GO :
2737                                          WPA_IF_P2P_CLIENT) < 0)
2738                 return NULL;
2739         group_wpa_s = wpas_p2p_init_group_interface(wpa_s, go);
2740         if (group_wpa_s == NULL) {
2741                 wpas_p2p_remove_pending_group_interface(wpa_s);
2742                 return NULL;
2743         }
2744
2745         return group_wpa_s;
2746 }
2747
2748
2749 /**
2750  * wpas_p2p_group_add - Add a new P2P group with local end as Group Owner
2751  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
2752  * @persistent_group: Whether to create a persistent group
2753  * @freq: Frequency for the group or 0 to indicate no hardcoding
2754  * Returns: 0 on success, -1 on failure
2755  *
2756  * This function creates a new P2P group with the local end as the Group Owner,
2757  * i.e., without using Group Owner Negotiation.
2758  */
2759 int wpas_p2p_group_add(struct wpa_supplicant *wpa_s, int persistent_group,
2760                        int freq)
2761 {
2762         struct p2p_go_neg_results params;
2763
2764         wpas_p2p_init_go_params(wpa_s, &params, freq);
2765         p2p_go_params(wpa_s->global->p2p, &params);
2766         params.persistent_group = persistent_group;
2767
2768         wpa_s = wpas_p2p_get_group_iface(wpa_s, 0, 1);
2769         if (wpa_s == NULL)
2770                 return -1;
2771         wpas_start_wps_go(wpa_s, &params, 0);
2772
2773         return 0;
2774 }
2775
2776
2777 static int wpas_start_p2p_client(struct wpa_supplicant *wpa_s,
2778                                  struct wpa_ssid *params, int addr_allocated)
2779 {
2780         struct wpa_ssid *ssid;
2781
2782         wpa_s = wpas_p2p_get_group_iface(wpa_s, addr_allocated, 0);
2783         if (wpa_s == NULL)
2784                 return -1;
2785
2786         wpa_supplicant_ap_deinit(wpa_s);
2787
2788         ssid = wpa_config_add_network(wpa_s->conf);
2789         if (ssid == NULL)
2790                 return -1;
2791         wpas_notify_network_added(wpa_s, ssid);
2792         wpa_config_set_network_defaults(ssid);
2793         ssid->temporary = 1;
2794         ssid->proto = WPA_PROTO_RSN;
2795         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
2796         ssid->group_cipher = WPA_CIPHER_CCMP;
2797         ssid->key_mgmt = WPA_KEY_MGMT_PSK;
2798         ssid->ssid = os_malloc(params->ssid_len);
2799         if (ssid->ssid == NULL) {
2800                 wpas_notify_network_removed(wpa_s, ssid);
2801                 wpa_config_remove_network(wpa_s->conf, ssid->id);
2802                 return -1;
2803         }
2804         os_memcpy(ssid->ssid, params->ssid, params->ssid_len);
2805         ssid->ssid_len = params->ssid_len;
2806         ssid->p2p_group = 1;
2807         if (params->psk_set) {
2808                 os_memcpy(ssid->psk, params->psk, 32);
2809                 ssid->psk_set = 1;
2810         }
2811         if (params->passphrase)
2812                 ssid->passphrase = os_strdup(params->passphrase);
2813
2814         wpa_supplicant_select_network(wpa_s, ssid);
2815
2816         wpa_s->show_group_started = 1;
2817
2818         return 0;
2819 }
2820
2821
2822 int wpas_p2p_group_add_persistent(struct wpa_supplicant *wpa_s,
2823                                   struct wpa_ssid *ssid, int addr_allocated,
2824                                   int freq)
2825 {
2826         struct p2p_go_neg_results params;
2827
2828         if (ssid->disabled != 2 || ssid->ssid == NULL)
2829                 return -1;
2830
2831         wpa_s->p2p_long_listen = 0;
2832         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
2833
2834         if (ssid->mode == WPAS_MODE_INFRA)
2835                 return wpas_start_p2p_client(wpa_s, ssid, addr_allocated);
2836
2837         if (ssid->mode != WPAS_MODE_P2P_GO)
2838                 return -1;
2839
2840         wpas_p2p_init_go_params(wpa_s, &params, freq);
2841
2842         params.role_go = 1;
2843         if (ssid->passphrase == NULL ||
2844             os_strlen(ssid->passphrase) >= sizeof(params.passphrase)) {
2845                 wpa_printf(MSG_DEBUG, "P2P: Invalid passphrase in persistent "
2846                            "group");
2847                 return -1;
2848         }
2849         os_strlcpy(params.passphrase, ssid->passphrase,
2850                    sizeof(params.passphrase));
2851         os_memcpy(params.ssid, ssid->ssid, ssid->ssid_len);
2852         params.ssid_len = ssid->ssid_len;
2853         params.persistent_group = 1;
2854
2855         wpa_s = wpas_p2p_get_group_iface(wpa_s, addr_allocated, 1);
2856         if (wpa_s == NULL)
2857                 return -1;
2858
2859         wpas_start_wps_go(wpa_s, &params, 0);
2860
2861         return 0;
2862 }
2863
2864
2865 static void wpas_p2p_ie_update(void *ctx, struct wpabuf *beacon_ies,
2866                                struct wpabuf *proberesp_ies)
2867 {
2868         struct wpa_supplicant *wpa_s = ctx;
2869         if (wpa_s->ap_iface) {
2870                 struct hostapd_data *hapd = wpa_s->ap_iface->bss[0];
2871                 if (beacon_ies) {
2872                         wpabuf_free(hapd->p2p_beacon_ie);
2873                         hapd->p2p_beacon_ie = beacon_ies;
2874                 }
2875                 wpabuf_free(hapd->p2p_probe_resp_ie);
2876                 hapd->p2p_probe_resp_ie = proberesp_ies;
2877         } else {
2878                 wpabuf_free(beacon_ies);
2879                 wpabuf_free(proberesp_ies);
2880         }
2881         wpa_supplicant_ap_update_beacon(wpa_s);
2882 }
2883
2884
2885 struct p2p_group * wpas_p2p_group_init(struct wpa_supplicant *wpa_s,
2886                                        int persistent_group,
2887                                        int group_formation)
2888 {
2889         struct p2p_group *group;
2890         struct p2p_group_config *cfg;
2891
2892         cfg = os_zalloc(sizeof(*cfg));
2893         if (cfg == NULL)
2894                 return NULL;
2895
2896         cfg->persistent_group = persistent_group;
2897         os_memcpy(cfg->interface_addr, wpa_s->own_addr, ETH_ALEN);
2898         cfg->cb_ctx = wpa_s;
2899         cfg->ie_update = wpas_p2p_ie_update;
2900
2901         group = p2p_group_init(wpa_s->global->p2p, cfg);
2902         if (group == NULL)
2903                 os_free(cfg);
2904         if (!group_formation)
2905                 p2p_group_notif_formation_done(group);
2906         wpa_s->p2p_group = group;
2907         return group;
2908 }
2909
2910
2911 void wpas_p2p_wps_success(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
2912                           int registrar)
2913 {
2914         if (!wpa_s->p2p_in_provisioning) {
2915                 wpa_printf(MSG_DEBUG, "P2P: Ignore WPS success event - P2P "
2916                            "provisioning not in progress");
2917                 return;
2918         }
2919
2920         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s->parent,
2921                              NULL);
2922         if (wpa_s->global->p2p)
2923                 p2p_wps_success_cb(wpa_s->global->p2p, peer_addr);
2924         wpas_group_formation_completed(wpa_s, 1);
2925 }
2926
2927
2928 int wpas_p2p_prov_disc(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
2929                        const char *config_method)
2930 {
2931         u16 config_methods;
2932
2933         if (os_strcmp(config_method, "display") == 0)
2934                 config_methods = WPS_CONFIG_DISPLAY;
2935         else if (os_strcmp(config_method, "keypad") == 0)
2936                 config_methods = WPS_CONFIG_KEYPAD;
2937         else if (os_strcmp(config_method, "pbc") == 0 ||
2938                  os_strcmp(config_method, "pushbutton") == 0)
2939                 config_methods = WPS_CONFIG_PUSHBUTTON;
2940         else
2941                 return -1;
2942
2943         if (wpa_s->global->p2p == NULL)
2944                 return -1;
2945
2946         return p2p_prov_disc_req(wpa_s->global->p2p, peer_addr,
2947                                  config_methods, 0);
2948 }
2949
2950
2951 int wpas_p2p_scan_result_text(const u8 *ies, size_t ies_len, char *buf,
2952                               char *end)
2953 {
2954         return p2p_scan_result_text(ies, ies_len, buf, end);
2955 }
2956
2957
2958 int wpas_p2p_find(struct wpa_supplicant *wpa_s, unsigned int timeout,
2959                   enum p2p_discovery_type type)
2960 {
2961         wpa_s->p2p_long_listen = 0;
2962
2963         return p2p_find(wpa_s->global->p2p, timeout, type);
2964 }
2965
2966
2967 void wpas_p2p_stop_find(struct wpa_supplicant *wpa_s)
2968 {
2969         wpa_s->p2p_long_listen = 0;
2970         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
2971
2972         p2p_stop_find(wpa_s->global->p2p);
2973
2974         wpas_p2p_remove_pending_group_interface(wpa_s);
2975 }
2976
2977
2978 static void wpas_p2p_long_listen_timeout(void *eloop_ctx, void *timeout_ctx)
2979 {
2980         struct wpa_supplicant *wpa_s = eloop_ctx;
2981         wpa_s->p2p_long_listen = 0;
2982 }
2983
2984
2985 int wpas_p2p_listen(struct wpa_supplicant *wpa_s, unsigned int timeout)
2986 {
2987         int res;
2988
2989         if (timeout == 0) {
2990                 /*
2991                  * This is a request for unlimited Listen state. However, at
2992                  * least for now, this is mapped to a Listen state for one
2993                  * hour.
2994                  */
2995                 timeout = 3600;
2996         }
2997         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
2998         wpa_s->p2p_long_listen = 0;
2999
3000         res = wpas_p2p_listen_start(wpa_s, timeout * 1000);
3001         if (res == 0 && timeout * 1000 > wpa_s->max_remain_on_chan) {
3002                 wpa_s->p2p_long_listen = timeout;
3003                 eloop_register_timeout(timeout, 0,
3004                                        wpas_p2p_long_listen_timeout,
3005                                        wpa_s, NULL);
3006         }
3007
3008         return res;
3009 }
3010
3011
3012 int wpas_p2p_assoc_req_ie(struct wpa_supplicant *wpa_s, struct wpa_bss *bss,
3013                           u8 *buf, size_t len, int p2p_group)
3014 {
3015         struct wpabuf *p2p_ie;
3016         int ret;
3017
3018         if (wpa_s->global->p2p_disabled)
3019                 return -1;
3020         if (wpa_s->global->p2p == NULL)
3021                 return -1;
3022
3023         p2p_ie = wpa_bss_get_vendor_ie_multi(bss, P2P_IE_VENDOR_TYPE);
3024         ret = p2p_assoc_req_ie(wpa_s->global->p2p, bss->bssid, buf, len,
3025                                p2p_group, p2p_ie);
3026         wpabuf_free(p2p_ie);
3027
3028         return ret;
3029 }
3030
3031
3032 int wpas_p2p_probe_req_rx(struct wpa_supplicant *wpa_s, const u8 *addr,
3033                           const u8 *ie, size_t ie_len)
3034 {
3035         if (wpa_s->global->p2p_disabled)
3036                 return 0;
3037         if (wpa_s->global->p2p == NULL)
3038                 return 0;
3039
3040         return p2p_probe_req_rx(wpa_s->global->p2p, addr, ie, ie_len);
3041 }
3042
3043
3044 void wpas_p2p_rx_action(struct wpa_supplicant *wpa_s, const u8 *da,
3045                         const u8 *sa, const u8 *bssid,
3046                         u8 category, const u8 *data, size_t len, int freq)
3047 {
3048         if (wpa_s->global->p2p_disabled)
3049                 return;
3050         if (wpa_s->global->p2p == NULL)
3051                 return;
3052
3053         p2p_rx_action(wpa_s->global->p2p, da, sa, bssid, category, data, len,
3054                       freq);
3055 }
3056
3057
3058 void wpas_p2p_scan_ie(struct wpa_supplicant *wpa_s, struct wpabuf *ies)
3059 {
3060         if (wpa_s->global->p2p_disabled)
3061                 return;
3062         if (wpa_s->global->p2p == NULL)
3063                 return;
3064
3065         p2p_scan_ie(wpa_s->global->p2p, ies);
3066 }
3067
3068
3069 void wpas_p2p_group_deinit(struct wpa_supplicant *wpa_s)
3070 {
3071         p2p_group_deinit(wpa_s->p2p_group);
3072         wpa_s->p2p_group = NULL;
3073 }
3074
3075
3076 int wpas_p2p_reject(struct wpa_supplicant *wpa_s, const u8 *addr)
3077 {
3078         wpa_s->p2p_long_listen = 0;
3079
3080         return p2p_reject(wpa_s->global->p2p, addr);
3081 }
3082
3083
3084 /* Invite to reinvoke a persistent group */
3085 int wpas_p2p_invite(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
3086                     struct wpa_ssid *ssid, const u8 *go_dev_addr)
3087 {
3088         enum p2p_invite_role role;
3089         u8 *bssid = NULL;
3090
3091         if (ssid->mode == WPAS_MODE_P2P_GO) {
3092                 role = P2P_INVITE_ROLE_GO;
3093                 if (peer_addr == NULL) {
3094                         wpa_printf(MSG_DEBUG, "P2P: Missing peer "
3095                                    "address in invitation command");
3096                         return -1;
3097                 }
3098                 if (wpas_p2p_create_iface(wpa_s)) {
3099                         if (wpas_p2p_add_group_interface(wpa_s,
3100                                                          WPA_IF_P2P_GO) < 0) {
3101                                 wpa_printf(MSG_ERROR, "P2P: Failed to "
3102                                            "allocate a new interface for the "
3103                                            "group");
3104                                 return -1;
3105                         }
3106                         bssid = wpa_s->pending_interface_addr;
3107                 } else
3108                         bssid = wpa_s->own_addr;
3109         } else {
3110                 role = P2P_INVITE_ROLE_CLIENT;
3111                 peer_addr = ssid->bssid;
3112         }
3113         wpa_s->pending_invite_ssid_id = ssid->id;
3114
3115         return p2p_invite(wpa_s->global->p2p, peer_addr, role, bssid,
3116                           ssid->ssid, ssid->ssid_len, 0, go_dev_addr, 1);
3117 }
3118
3119
3120 /* Invite to join an active group */
3121 int wpas_p2p_invite_group(struct wpa_supplicant *wpa_s, const char *ifname,
3122                           const u8 *peer_addr, const u8 *go_dev_addr)
3123 {
3124         struct wpa_global *global = wpa_s->global;
3125         enum p2p_invite_role role;
3126         u8 *bssid = NULL;
3127         struct wpa_ssid *ssid;
3128
3129         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
3130                 if (os_strcmp(wpa_s->ifname, ifname) == 0)
3131                         break;
3132         }
3133         if (wpa_s == NULL) {
3134                 wpa_printf(MSG_DEBUG, "P2P: Interface '%s' not found", ifname);
3135                 return -1;
3136         }
3137
3138         ssid = wpa_s->current_ssid;
3139         if (ssid == NULL) {
3140                 wpa_printf(MSG_DEBUG, "P2P: No current SSID to use for "
3141                            "invitation");
3142                 return -1;
3143         }
3144
3145         if (ssid->mode == WPAS_MODE_P2P_GO) {
3146                 role = P2P_INVITE_ROLE_ACTIVE_GO;
3147                 bssid = wpa_s->own_addr;
3148                 if (go_dev_addr == NULL)
3149                         go_dev_addr = wpa_s->parent->own_addr;
3150         } else {
3151                 role = P2P_INVITE_ROLE_CLIENT;
3152                 if (wpa_s->wpa_state < WPA_ASSOCIATED) {
3153                         wpa_printf(MSG_DEBUG, "P2P: Not associated - cannot "
3154                                    "invite to current group");
3155                         return -1;
3156                 }
3157                 bssid = wpa_s->bssid;
3158                 if (go_dev_addr == NULL &&
3159                     !is_zero_ether_addr(wpa_s->go_dev_addr))
3160                         go_dev_addr = wpa_s->go_dev_addr;
3161         }
3162         wpa_s->pending_invite_ssid_id = -1;
3163
3164         return p2p_invite(wpa_s->global->p2p, peer_addr, role, bssid,
3165                           ssid->ssid, ssid->ssid_len, wpa_s->assoc_freq,
3166                           go_dev_addr, 0);
3167 }
3168
3169
3170 void wpas_p2p_completed(struct wpa_supplicant *wpa_s)
3171 {
3172         struct wpa_ssid *ssid = wpa_s->current_ssid;
3173         const char *ssid_txt;
3174         u8 go_dev_addr[ETH_ALEN];
3175         int persistent;
3176
3177         if (!wpa_s->show_group_started || !ssid)
3178                 return;
3179
3180         wpa_s->show_group_started = 0;
3181
3182         ssid_txt = wpa_ssid_txt(ssid->ssid, ssid->ssid_len);
3183         os_memset(go_dev_addr, 0, ETH_ALEN);
3184         if (ssid->bssid_set)
3185                 os_memcpy(go_dev_addr, ssid->bssid, ETH_ALEN);
3186         persistent = wpas_p2p_persistent_group(wpa_s, go_dev_addr, ssid->ssid,
3187                                                ssid->ssid_len);
3188         os_memcpy(wpa_s->go_dev_addr, go_dev_addr, ETH_ALEN);
3189
3190         if (ssid->passphrase == NULL && ssid->psk_set) {
3191                 char psk[65];
3192                 wpa_snprintf_hex(psk, sizeof(psk), ssid->psk, 32);
3193                 wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
3194                         "%s client ssid=\"%s\" psk=%s go_dev_addr=" MACSTR
3195                         "%s",
3196                         wpa_s->ifname, ssid_txt, psk, MAC2STR(go_dev_addr),
3197                         persistent ? " [PERSISTENT]" : "");
3198         } else {
3199                 wpa_msg(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
3200                         "%s client ssid=\"%s\" passphrase=\"%s\" go_dev_addr="
3201                         MACSTR "%s",
3202                         wpa_s->ifname, ssid_txt,
3203                         ssid->passphrase ? ssid->passphrase : "",
3204                         MAC2STR(go_dev_addr),
3205                         persistent ? " [PERSISTENT]" : "");
3206         }
3207
3208         if (persistent)
3209                 wpas_p2p_store_persistent_group(wpa_s->parent, ssid,
3210                                                 go_dev_addr);
3211 }
3212
3213
3214 int wpas_p2p_presence_req(struct wpa_supplicant *wpa_s, u32 duration1,
3215                           u32 interval1, u32 duration2, u32 interval2)
3216 {
3217         if (wpa_s->wpa_state < WPA_ASSOCIATED ||
3218             wpa_s->current_ssid == NULL ||
3219             wpa_s->current_ssid->mode != WPAS_MODE_INFRA)
3220                 return -1;
3221
3222         return p2p_presence_req(wpa_s->global->p2p, wpa_s->bssid,
3223                                 wpa_s->own_addr, wpa_s->assoc_freq,
3224                                 duration1, interval1, duration2, interval2);
3225 }
3226
3227
3228 int wpas_p2p_ext_listen(struct wpa_supplicant *wpa_s, unsigned int period,
3229                         unsigned int interval)
3230 {
3231         return p2p_ext_listen(wpa_s->global->p2p, period, interval);
3232 }
3233
3234
3235 void wpas_p2p_deauth_notif(struct wpa_supplicant *wpa_s, const u8 *bssid,
3236                            u16 reason_code, const u8 *ie, size_t ie_len)
3237 {
3238         if (wpa_s->global->p2p_disabled)
3239                 return;
3240
3241         p2p_deauth_notif(wpa_s->global->p2p, bssid, reason_code, ie, ie_len);
3242 }
3243
3244
3245 void wpas_p2p_disassoc_notif(struct wpa_supplicant *wpa_s, const u8 *bssid,
3246                              u16 reason_code, const u8 *ie, size_t ie_len)
3247 {
3248         if (wpa_s->global->p2p_disabled)
3249                 return;
3250
3251         p2p_disassoc_notif(wpa_s->global->p2p, bssid, reason_code, ie, ie_len);
3252 }
3253
3254
3255 void wpas_p2p_update_config(struct wpa_supplicant *wpa_s)
3256 {
3257         struct p2p_data *p2p = wpa_s->global->p2p;
3258
3259         if (p2p == NULL)
3260                 return;
3261
3262         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_NAME)
3263                 p2p_set_dev_name(p2p, wpa_s->conf->device_name);
3264
3265         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_TYPE) {
3266                 u8 pri_dev_type[8];
3267                 if (wpa_s->conf->device_type) {
3268                         if (wps_dev_type_str2bin(wpa_s->conf->device_type,
3269                                                  pri_dev_type) < 0) {
3270                                 wpa_printf(MSG_ERROR, "P2P: Invalid "
3271                                            "device_type");
3272                         } else
3273                                 p2p_set_pri_dev_type(p2p, pri_dev_type);
3274                 }
3275         }
3276
3277         if (wpa_s->conf->changed_parameters & CFG_CHANGED_SEC_DEVICE_TYPE) {
3278                 u8 sec_dev_type[P2P_SEC_DEVICE_TYPES][8];
3279                 size_t num = 0;
3280                 int i;
3281                 for (i = 0; i < MAX_SEC_DEVICE_TYPES; i++) {
3282                         if (wpa_s->conf->sec_device_type[i] == NULL)
3283                                 continue;
3284                         if (wps_dev_type_str2bin(
3285                                     wpa_s->conf->sec_device_type[i],
3286                                     sec_dev_type[num]) < 0) {
3287                                 wpa_printf(MSG_ERROR, "P2P: Invalid "
3288                                            "sec_device_type");
3289                                 continue;
3290                         }
3291                         num++;
3292                         if (num == P2P_SEC_DEVICE_TYPES)
3293                                 break;
3294                 }
3295                 p2p_set_sec_dev_types(p2p, (void *) sec_dev_type, num);
3296         }
3297
3298         if ((wpa_s->conf->changed_parameters & CFG_CHANGED_COUNTRY) &&
3299             wpa_s->conf->country[0] && wpa_s->conf->country[1]) {
3300                 char country[3];
3301                 country[0] = wpa_s->conf->country[0];
3302                 country[1] = wpa_s->conf->country[1];
3303                 country[2] = 0x04;
3304                 p2p_set_country(p2p, country);
3305         }
3306
3307         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_SSID_POSTFIX) {
3308                 p2p_set_ssid_postfix(p2p, (u8 *) wpa_s->conf->p2p_ssid_postfix,
3309                                      wpa_s->conf->p2p_ssid_postfix ?
3310                                      os_strlen(wpa_s->conf->p2p_ssid_postfix) :
3311                                      0);
3312         }
3313 }
3314
3315
3316 int wpas_p2p_set_noa(struct wpa_supplicant *wpa_s, u8 count, int start,
3317                      int duration)
3318 {
3319         if (!wpa_s->ap_iface)
3320                 return -1;
3321         return hostapd_p2p_set_noa(wpa_s->ap_iface->bss[0], count, start,
3322                                    duration);
3323 }
3324
3325
3326 int wpas_p2p_set_cross_connect(struct wpa_supplicant *wpa_s, int enabled)
3327 {
3328         if (wpa_s->global->p2p_disabled)
3329                 return -1;
3330         if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_MGMT)
3331                 return -1;
3332
3333         wpa_s->global->cross_connection = enabled;
3334         p2p_set_cross_connect(wpa_s->global->p2p, enabled);
3335
3336         if (!enabled) {
3337                 struct wpa_supplicant *iface;
3338
3339                 for (iface = wpa_s->global->ifaces; iface; iface = iface->next)
3340                 {
3341                         if (iface->cross_connect_enabled == 0)
3342                                 continue;
3343
3344                         iface->cross_connect_enabled = 0;
3345                         iface->cross_connect_in_use = 0;
3346                         wpa_msg(iface->parent, MSG_INFO,
3347                                 P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
3348                                 iface->ifname, iface->cross_connect_uplink);
3349                 }
3350         }
3351
3352         return 0;
3353 }
3354
3355
3356 static void wpas_p2p_enable_cross_connect(struct wpa_supplicant *uplink)
3357 {
3358         struct wpa_supplicant *iface;
3359
3360         if (!uplink->global->cross_connection)
3361                 return;
3362
3363         for (iface = uplink->global->ifaces; iface; iface = iface->next) {
3364                 if (!iface->cross_connect_enabled)
3365                         continue;
3366                 if (os_strcmp(uplink->ifname, iface->cross_connect_uplink) !=
3367                     0)
3368                         continue;
3369                 if (iface->ap_iface == NULL)
3370                         continue;
3371                 if (iface->cross_connect_in_use)
3372                         continue;
3373
3374                 iface->cross_connect_in_use = 1;
3375                 wpa_msg(iface->parent, MSG_INFO,
3376                         P2P_EVENT_CROSS_CONNECT_ENABLE "%s %s",
3377                         iface->ifname, iface->cross_connect_uplink);
3378         }
3379 }
3380
3381
3382 static void wpas_p2p_disable_cross_connect(struct wpa_supplicant *uplink)
3383 {
3384         struct wpa_supplicant *iface;
3385
3386         for (iface = uplink->global->ifaces; iface; iface = iface->next) {
3387                 if (!iface->cross_connect_enabled)
3388                         continue;
3389                 if (os_strcmp(uplink->ifname, iface->cross_connect_uplink) !=
3390                     0)
3391                         continue;
3392                 if (!iface->cross_connect_in_use)
3393                         continue;
3394
3395                 wpa_msg(iface->parent, MSG_INFO,
3396                         P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
3397                         iface->ifname, iface->cross_connect_uplink);
3398                 iface->cross_connect_in_use = 0;
3399         }
3400 }
3401
3402
3403 void wpas_p2p_notif_connected(struct wpa_supplicant *wpa_s)
3404 {
3405         if (wpa_s->ap_iface || wpa_s->current_ssid == NULL ||
3406             wpa_s->current_ssid->mode != WPAS_MODE_INFRA ||
3407             wpa_s->cross_connect_disallowed)
3408                 wpas_p2p_disable_cross_connect(wpa_s);
3409         else
3410                 wpas_p2p_enable_cross_connect(wpa_s);
3411 }
3412
3413
3414 void wpas_p2p_notif_disconnected(struct wpa_supplicant *wpa_s)
3415 {
3416         wpas_p2p_disable_cross_connect(wpa_s);
3417 }
3418
3419
3420 static void wpas_p2p_cross_connect_setup(struct wpa_supplicant *wpa_s)
3421 {
3422         struct wpa_supplicant *iface;
3423
3424         if (!wpa_s->global->cross_connection)
3425                 return;
3426
3427         for (iface = wpa_s->global->ifaces; iface; iface = iface->next) {
3428                 if (iface == wpa_s)
3429                         continue;
3430                 if (iface->drv_flags &
3431                     WPA_DRIVER_FLAGS_P2P_DEDICATED_INTERFACE)
3432                         continue;
3433                 if (iface->drv_flags & WPA_DRIVER_FLAGS_P2P_CAPABLE)
3434                         continue;
3435
3436                 wpa_s->cross_connect_enabled = 1;
3437                 os_strlcpy(wpa_s->cross_connect_uplink, iface->ifname,
3438                            sizeof(wpa_s->cross_connect_uplink));
3439                 wpa_printf(MSG_DEBUG, "P2P: Enable cross connection from "
3440                            "%s to %s whenever uplink is available",
3441                            wpa_s->ifname, wpa_s->cross_connect_uplink);
3442
3443                 if (iface->ap_iface || iface->current_ssid == NULL ||
3444                     iface->current_ssid->mode != WPAS_MODE_INFRA ||
3445                     iface->cross_connect_disallowed ||
3446                     iface->wpa_state != WPA_COMPLETED)
3447                         break;
3448
3449                 wpa_s->cross_connect_in_use = 1;
3450                 wpa_msg(wpa_s->parent, MSG_INFO,
3451                         P2P_EVENT_CROSS_CONNECT_ENABLE "%s %s",
3452                         wpa_s->ifname, wpa_s->cross_connect_uplink);
3453                 break;
3454         }
3455 }