wpa_supplicant/p2p_supplicant.c

   1 /*
   2  * wpa_supplicant - P2P
   3  * Copyright (c) 2009-2010, Atheros Communications
   4  * Copyright (c) 2010-2014, Jouni Malinen <j@w1.fi>
   5  *
   6  * This software may be distributed under the terms of the BSD license.
   7  * See README for more details.
   8  */
   9
  10 #include "includes.h"
  11
  12 #include "common.h"
  13 #include "eloop.h"
  14 #include "common/ieee802_11_common.h"
  15 #include "common/ieee802_11_defs.h"
  16 #include "common/wpa_ctrl.h"
  17 #include "wps/wps_i.h"
  18 #include "p2p/p2p.h"
  19 #include "ap/hostapd.h"
  20 #include "ap/ap_config.h"
  21 #include "ap/sta_info.h"
  22 #include "ap/ap_drv_ops.h"
  23 #include "ap/p2p_hostapd.h"
  24 #include "eapol_supp/eapol_supp_sm.h"
  25 #include "rsn_supp/wpa.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 "notify.h"
  32 #include "scan.h"
  33 #include "bss.h"
  34 #include "offchannel.h"
  35 #include "wps_supplicant.h"
  36 #include "p2p_supplicant.h"
  37 #include "wifi_display.h"
  38
  39
  40 /*
  41  * How many times to try to scan to find the GO before giving up on join
  42  * request.
  43  */
  44 #define P2P_MAX_JOIN_SCAN_ATTEMPTS 10
  45
  46 #define P2P_AUTO_PD_SCAN_ATTEMPTS 5
  47
  48 #ifndef P2P_MAX_CLIENT_IDLE
  49 /*
  50  * How many seconds to try to reconnect to the GO when connection in P2P client
  51  * role has been lost.
  52  */
  53 #define P2P_MAX_CLIENT_IDLE 10
  54 #endif /* P2P_MAX_CLIENT_IDLE */
  55
  56 #ifndef P2P_MAX_INITIAL_CONN_WAIT
  57 /*
  58  * How many seconds to wait for initial 4-way handshake to get completed after
  59  * WPS provisioning step.
  60  */
  61 #define P2P_MAX_INITIAL_CONN_WAIT 10
  62 #endif /* P2P_MAX_INITIAL_CONN_WAIT */
  63
  64 #ifndef P2P_MAX_INITIAL_CONN_WAIT_GO
  65 /*
  66  * How many seconds to wait for initial 4-way handshake to get completed after
  67  * WPS provisioning step on the GO. This controls the extra time the P2P
  68  * operation is considered to be in progress (e.g., to delay other scans) after
  69  * WPS provisioning has been completed on the GO during group formation.
  70  */
  71 #define P2P_MAX_INITIAL_CONN_WAIT_GO 10
  72 #endif /* P2P_MAX_INITIAL_CONN_WAIT_GO */
  73
  74 #ifndef P2P_MAX_INITIAL_CONN_WAIT_GO_REINVOKE
  75 /*
  76  * How many seconds to wait for initial 4-way handshake to get completed after
  77  * re-invocation of a persistent group on the GO when the client is expected
  78  * to connect automatically (no user interaction).
  79  */
  80 #define P2P_MAX_INITIAL_CONN_WAIT_GO_REINVOKE 15
  81 #endif /* P2P_MAX_INITIAL_CONN_WAIT_GO_REINVOKE */
  82
  83 #ifndef P2P_CONCURRENT_SEARCH_DELAY
  84 #define P2P_CONCURRENT_SEARCH_DELAY 500
  85 #endif /* P2P_CONCURRENT_SEARCH_DELAY */
  86
  87 #define P2P_MGMT_DEVICE_PREFIX          "p2p-dev-"
  88
  89 enum p2p_group_removal_reason {
  90         P2P_GROUP_REMOVAL_UNKNOWN,
  91         P2P_GROUP_REMOVAL_SILENT,
  92         P2P_GROUP_REMOVAL_FORMATION_FAILED,
  93         P2P_GROUP_REMOVAL_REQUESTED,
  94         P2P_GROUP_REMOVAL_IDLE_TIMEOUT,
  95         P2P_GROUP_REMOVAL_UNAVAILABLE,
  96         P2P_GROUP_REMOVAL_GO_ENDING_SESSION,
  97         P2P_GROUP_REMOVAL_PSK_FAILURE,
  98         P2P_GROUP_REMOVAL_FREQ_CONFLICT
  99 };
 100
 101
 102 static void wpas_p2p_long_listen_timeout(void *eloop_ctx, void *timeout_ctx);
 103 static struct wpa_supplicant *
 104 wpas_p2p_get_group_iface(struct wpa_supplicant *wpa_s, int addr_allocated,
 105                          int go);
 106 static int wpas_p2p_join_start(struct wpa_supplicant *wpa_s);
 107 static void wpas_p2p_join_scan_req(struct wpa_supplicant *wpa_s, int freq,
 108                                    const u8 *ssid, size_t ssid_len);
 109 static void wpas_p2p_join_scan(void *eloop_ctx, void *timeout_ctx);
 110 static int wpas_p2p_join(struct wpa_supplicant *wpa_s, const u8 *iface_addr,
 111                          const u8 *dev_addr, enum p2p_wps_method wps_method,
 112                          int auto_join, const u8 *ssid, size_t ssid_len);
 113 static int wpas_p2p_create_iface(struct wpa_supplicant *wpa_s);
 114 static void wpas_p2p_cross_connect_setup(struct wpa_supplicant *wpa_s);
 115 static void wpas_p2p_group_idle_timeout(void *eloop_ctx, void *timeout_ctx);
 116 static void wpas_p2p_set_group_idle_timeout(struct wpa_supplicant *wpa_s);
 117 static void wpas_p2p_group_formation_timeout(void *eloop_ctx,
 118                                              void *timeout_ctx);
 119 static void wpas_p2p_group_freq_conflict(void *eloop_ctx, void *timeout_ctx);
 120 static void wpas_p2p_fallback_to_go_neg(struct wpa_supplicant *wpa_s,
 121                                         int group_added);
 122 static int wpas_p2p_stop_find_oper(struct wpa_supplicant *wpa_s);
 123
 124
 125 /*
 126  * Get the number of concurrent channels that the HW can operate, but that are
 127  * currently not in use by any of the wpa_supplicant interfaces.
 128  */
 129 static int wpas_p2p_num_unused_channels(struct wpa_supplicant *wpa_s)
 130 {
 131         int *freqs;
 132         int num, unused;
 133
 134         freqs = os_calloc(wpa_s->num_multichan_concurrent, sizeof(int));
 135         if (!freqs)
 136                 return -1;
 137
 138         num = get_shared_radio_freqs(wpa_s, freqs,
 139                                      wpa_s->num_multichan_concurrent);
 140         os_free(freqs);
 141
 142         unused = wpa_s->num_multichan_concurrent - num;
 143         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: num_unused_channels: %d", unused);
 144         return unused;
 145 }
 146
 147
 148 /*
 149  * Get the frequencies that are currently in use by one or more of the virtual
 150  * interfaces, and that are also valid for P2P operation.
 151  */
 152 static int wpas_p2p_valid_oper_freqs(struct wpa_supplicant *wpa_s,
 153                                      int *p2p_freqs, unsigned int len)
 154 {
 155         int *freqs;
 156         unsigned int num, i, j;
 157
 158         freqs = os_calloc(wpa_s->num_multichan_concurrent, sizeof(int));
 159         if (!freqs)
 160                 return -1;
 161
 162         num = get_shared_radio_freqs(wpa_s, freqs,
 163                                      wpa_s->num_multichan_concurrent);
 164
 165         os_memset(p2p_freqs, 0, sizeof(int) * len);
 166
 167         for (i = 0, j = 0; i < num && j < len; i++) {
 168                 if (p2p_supported_freq(wpa_s->global->p2p, freqs[i]))
 169                         p2p_freqs[j++] = freqs[i];
 170         }
 171
 172         os_free(freqs);
 173
 174         dump_freq_array(wpa_s, "valid for P2P", p2p_freqs, j);
 175
 176         return j;
 177 }
 178
 179
 180 static void wpas_p2p_set_own_freq_preference(struct wpa_supplicant *wpa_s,
 181                                              int freq)
 182 {
 183         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
 184                 return;
 185         if (wpa_s->parent->conf->p2p_ignore_shared_freq &&
 186             freq > 0 && wpa_s->num_multichan_concurrent > 1 &&
 187             wpas_p2p_num_unused_channels(wpa_s) > 0) {
 188                 wpa_printf(MSG_DEBUG, "P2P: Ignore own channel preference %d MHz due to p2p_ignore_shared_freq=1 configuration",
 189                            freq);
 190                 freq = 0;
 191         }
 192         p2p_set_own_freq_preference(wpa_s->global->p2p, freq);
 193 }
 194
 195
 196 static void wpas_p2p_scan_res_handler(struct wpa_supplicant *wpa_s,
 197                                       struct wpa_scan_results *scan_res)
 198 {
 199         size_t i;
 200
 201         if (wpa_s->p2p_scan_work) {
 202                 struct wpa_radio_work *work = wpa_s->p2p_scan_work;
 203                 wpa_s->p2p_scan_work = NULL;
 204                 radio_work_done(work);
 205         }
 206
 207         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
 208                 return;
 209
 210         wpa_printf(MSG_DEBUG, "P2P: Scan results received (%d BSS)",
 211                    (int) scan_res->num);
 212
 213         for (i = 0; i < scan_res->num; i++) {
 214                 struct wpa_scan_res *bss = scan_res->res[i];
 215                 struct os_reltime time_tmp_age, entry_ts;
 216                 const u8 *ies;
 217                 size_t ies_len;
 218
 219                 time_tmp_age.sec = bss->age / 1000;
 220                 time_tmp_age.usec = (bss->age % 1000) * 1000;
 221                 os_reltime_sub(&scan_res->fetch_time, &time_tmp_age, &entry_ts);
 222
 223                 ies = (const u8 *) (bss + 1);
 224                 ies_len = bss->ie_len;
 225                 if (bss->beacon_ie_len > 0 &&
 226                     !wpa_scan_get_vendor_ie(bss, P2P_IE_VENDOR_TYPE) &&
 227                     wpa_scan_get_vendor_ie_beacon(bss, P2P_IE_VENDOR_TYPE)) {
 228                         wpa_printf(MSG_DEBUG, "P2P: Use P2P IE(s) from Beacon frame since no P2P IE(s) in Probe Response frames received for "
 229                                    MACSTR, MAC2STR(bss->bssid));
 230                         ies = ies + ies_len;
 231                         ies_len = bss->beacon_ie_len;
 232                 }
 233
 234
 235                 if (p2p_scan_res_handler(wpa_s->global->p2p, bss->bssid,
 236                                          bss->freq, &entry_ts, bss->level,
 237                                          ies, ies_len) > 0)
 238                         break;
 239         }
 240
 241         p2p_scan_res_handled(wpa_s->global->p2p);
 242 }
 243
 244
 245 static void wpas_p2p_trigger_scan_cb(struct wpa_radio_work *work, int deinit)
 246 {
 247         struct wpa_supplicant *wpa_s = work->wpa_s;
 248         struct wpa_driver_scan_params *params = work->ctx;
 249         int ret;
 250
 251         if (deinit) {
 252                 wpa_scan_free_params(params);
 253                 return;
 254         }
 255
 256         ret = wpa_drv_scan(wpa_s, params);
 257         wpa_scan_free_params(params);
 258         work->ctx = NULL;
 259         if (ret) {
 260                 radio_work_done(work);
 261                 return;
 262         }
 263
 264         os_get_reltime(&wpa_s->scan_trigger_time);
 265         wpa_s->scan_res_handler = wpas_p2p_scan_res_handler;
 266         wpa_s->own_scan_requested = 1;
 267         wpa_s->p2p_scan_work = work;
 268 }
 269
 270
 271 static int wpas_p2p_scan(void *ctx, enum p2p_scan_type type, int freq,
 272                          unsigned int num_req_dev_types,
 273                          const u8 *req_dev_types, const u8 *dev_id, u16 pw_id)
 274 {
 275         struct wpa_supplicant *wpa_s = ctx;
 276         struct wpa_driver_scan_params *params = NULL;
 277         struct wpabuf *wps_ie, *ies;
 278         size_t ielen;
 279         u8 *n;
 280
 281         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
 282                 return -1;
 283
 284         if (wpa_s->p2p_scan_work) {
 285                 wpa_dbg(wpa_s, MSG_INFO, "P2P: Reject scan trigger since one is already pending");
 286                 return -1;
 287         }
 288
 289         params = os_zalloc(sizeof(*params));
 290         if (params == NULL)
 291                 return -1;
 292
 293         /* P2P Wildcard SSID */
 294         params->num_ssids = 1;
 295         n = os_malloc(P2P_WILDCARD_SSID_LEN);
 296         if (n == NULL)
 297                 goto fail;
 298         os_memcpy(n, P2P_WILDCARD_SSID, P2P_WILDCARD_SSID_LEN);
 299         params->ssids[0].ssid = n;
 300         params->ssids[0].ssid_len = P2P_WILDCARD_SSID_LEN;
 301
 302         wpa_s->wps->dev.p2p = 1;
 303         wps_ie = wps_build_probe_req_ie(pw_id, &wpa_s->wps->dev,
 304                                         wpa_s->wps->uuid, WPS_REQ_ENROLLEE,
 305                                         num_req_dev_types, req_dev_types);
 306         if (wps_ie == NULL)
 307                 goto fail;
 308
 309         ielen = p2p_scan_ie_buf_len(wpa_s->global->p2p);
 310         ies = wpabuf_alloc(wpabuf_len(wps_ie) + ielen);
 311         if (ies == NULL) {
 312                 wpabuf_free(wps_ie);
 313                 goto fail;
 314         }
 315         wpabuf_put_buf(ies, wps_ie);
 316         wpabuf_free(wps_ie);
 317
 318         p2p_scan_ie(wpa_s->global->p2p, ies, dev_id);
 319
 320         params->p2p_probe = 1;
 321         n = os_malloc(wpabuf_len(ies));
 322         if (n == NULL) {
 323                 wpabuf_free(ies);
 324                 goto fail;
 325         }
 326         os_memcpy(n, wpabuf_head(ies), wpabuf_len(ies));
 327         params->extra_ies = n;
 328         params->extra_ies_len = wpabuf_len(ies);
 329         wpabuf_free(ies);
 330
 331         switch (type) {
 332         case P2P_SCAN_SOCIAL:
 333                 params->freqs = os_malloc(4 * sizeof(int));
 334                 if (params->freqs == NULL)
 335                         goto fail;
 336                 params->freqs[0] = 2412;
 337                 params->freqs[1] = 2437;
 338                 params->freqs[2] = 2462;
 339                 params->freqs[3] = 0;
 340                 break;
 341         case P2P_SCAN_FULL:
 342                 break;
 343         case P2P_SCAN_SOCIAL_PLUS_ONE:
 344                 params->freqs = os_malloc(5 * sizeof(int));
 345                 if (params->freqs == NULL)
 346                         goto fail;
 347                 params->freqs[0] = 2412;
 348                 params->freqs[1] = 2437;
 349                 params->freqs[2] = 2462;
 350                 params->freqs[3] = freq;
 351                 params->freqs[4] = 0;
 352                 break;
 353         }
 354
 355         radio_remove_unstarted_work(wpa_s, "p2p-scan");
 356         if (radio_add_work(wpa_s, 0, "p2p-scan", 0, wpas_p2p_trigger_scan_cb,
 357                            params) < 0)
 358                 goto fail;
 359         return 0;
 360
 361 fail:
 362         wpa_scan_free_params(params);
 363         return -1;
 364 }
 365
 366
 367 static enum wpa_driver_if_type wpas_p2p_if_type(int p2p_group_interface)
 368 {
 369         switch (p2p_group_interface) {
 370         case P2P_GROUP_INTERFACE_PENDING:
 371                 return WPA_IF_P2P_GROUP;
 372         case P2P_GROUP_INTERFACE_GO:
 373                 return WPA_IF_P2P_GO;
 374         case P2P_GROUP_INTERFACE_CLIENT:
 375                 return WPA_IF_P2P_CLIENT;
 376         }
 377
 378         return WPA_IF_P2P_GROUP;
 379 }
 380
 381
 382 static struct wpa_supplicant * wpas_get_p2p_group(struct wpa_supplicant *wpa_s,
 383                                                   const u8 *ssid,
 384                                                   size_t ssid_len, int *go)
 385 {
 386         struct wpa_ssid *s;
 387
 388         for (wpa_s = wpa_s->global->ifaces; wpa_s; wpa_s = wpa_s->next) {
 389                 for (s = wpa_s->conf->ssid; s; s = s->next) {
 390                         if (s->disabled != 0 || !s->p2p_group ||
 391                             s->ssid_len != ssid_len ||
 392                             os_memcmp(ssid, s->ssid, ssid_len) != 0)
 393                                 continue;
 394                         if (s->mode == WPAS_MODE_P2P_GO &&
 395                             s != wpa_s->current_ssid)
 396                                 continue;
 397                         if (go)
 398                                 *go = s->mode == WPAS_MODE_P2P_GO;
 399                         return wpa_s;
 400                 }
 401         }
 402
 403         return NULL;
 404 }
 405
 406
 407 static int wpas_p2p_group_delete(struct wpa_supplicant *wpa_s,
 408                                  enum p2p_group_removal_reason removal_reason)
 409 {
 410         struct wpa_ssid *ssid;
 411         char *gtype;
 412         const char *reason;
 413
 414         ssid = wpa_s->current_ssid;
 415         if (ssid == NULL) {
 416                 /*
 417                  * The current SSID was not known, but there may still be a
 418                  * pending P2P group interface waiting for provisioning or a
 419                  * P2P group that is trying to reconnect.
 420                  */
 421                 ssid = wpa_s->conf->ssid;
 422                 while (ssid) {
 423                         if (ssid->p2p_group && ssid->disabled != 2)
 424                                 break;
 425                         ssid = ssid->next;
 426                 }
 427                 if (ssid == NULL &&
 428                         wpa_s->p2p_group_interface == NOT_P2P_GROUP_INTERFACE)
 429                 {
 430                         wpa_printf(MSG_ERROR, "P2P: P2P group interface "
 431                                    "not found");
 432                         return -1;
 433                 }
 434         }
 435         if (wpa_s->p2p_group_interface == P2P_GROUP_INTERFACE_GO)
 436                 gtype = "GO";
 437         else if (wpa_s->p2p_group_interface == P2P_GROUP_INTERFACE_CLIENT ||
 438                  (ssid && ssid->mode == WPAS_MODE_INFRA)) {
 439                 wpa_s->reassociate = 0;
 440                 wpa_s->disconnected = 1;
 441                 wpa_supplicant_deauthenticate(wpa_s,
 442                                               WLAN_REASON_DEAUTH_LEAVING);
 443                 gtype = "client";
 444         } else
 445                 gtype = "GO";
 446         if (wpa_s->cross_connect_in_use) {
 447                 wpa_s->cross_connect_in_use = 0;
 448                 wpa_msg_global(wpa_s->parent, MSG_INFO,
 449                                P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
 450                                wpa_s->ifname, wpa_s->cross_connect_uplink);
 451         }
 452         switch (removal_reason) {
 453         case P2P_GROUP_REMOVAL_REQUESTED:
 454                 reason = " reason=REQUESTED";
 455                 break;
 456         case P2P_GROUP_REMOVAL_FORMATION_FAILED:
 457                 reason = " reason=FORMATION_FAILED";
 458                 break;
 459         case P2P_GROUP_REMOVAL_IDLE_TIMEOUT:
 460                 reason = " reason=IDLE";
 461                 break;
 462         case P2P_GROUP_REMOVAL_UNAVAILABLE:
 463                 reason = " reason=UNAVAILABLE";
 464                 break;
 465         case P2P_GROUP_REMOVAL_GO_ENDING_SESSION:
 466                 reason = " reason=GO_ENDING_SESSION";
 467                 break;
 468         case P2P_GROUP_REMOVAL_PSK_FAILURE:
 469                 reason = " reason=PSK_FAILURE";
 470                 break;
 471         case P2P_GROUP_REMOVAL_FREQ_CONFLICT:
 472                 reason = " reason=FREQ_CONFLICT";
 473                 break;
 474         default:
 475                 reason = "";
 476                 break;
 477         }
 478         if (removal_reason != P2P_GROUP_REMOVAL_SILENT) {
 479                 wpa_msg_global(wpa_s->parent, MSG_INFO,
 480                                P2P_EVENT_GROUP_REMOVED "%s %s%s",
 481                                wpa_s->ifname, gtype, reason);
 482         }
 483
 484         if (eloop_cancel_timeout(wpas_p2p_group_freq_conflict, wpa_s, NULL) > 0)
 485                 wpa_printf(MSG_DEBUG, "P2P: Cancelled P2P group freq_conflict timeout");
 486         if (eloop_cancel_timeout(wpas_p2p_group_idle_timeout, wpa_s, NULL) > 0)
 487                 wpa_printf(MSG_DEBUG, "P2P: Cancelled P2P group idle timeout");
 488         if (eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
 489                                  wpa_s->parent, NULL) > 0) {
 490                 wpa_printf(MSG_DEBUG, "P2P: Cancelled P2P group formation "
 491                            "timeout");
 492                 wpa_s->p2p_in_provisioning = 0;
 493         }
 494
 495         /*
 496          * Make sure wait for the first client does not remain active after the
 497          * group has been removed.
 498          */
 499         wpa_s->global->p2p_go_wait_client.sec = 0;
 500
 501         if (removal_reason != P2P_GROUP_REMOVAL_SILENT && ssid)
 502                 wpas_notify_p2p_group_removed(wpa_s, ssid, gtype);
 503
 504         if (wpa_s->p2p_group_interface != NOT_P2P_GROUP_INTERFACE) {
 505                 struct wpa_global *global;
 506                 char *ifname;
 507                 enum wpa_driver_if_type type;
 508                 wpa_printf(MSG_DEBUG, "P2P: Remove group interface %s",
 509                         wpa_s->ifname);
 510                 global = wpa_s->global;
 511                 ifname = os_strdup(wpa_s->ifname);
 512                 type = wpas_p2p_if_type(wpa_s->p2p_group_interface);
 513                 wpa_supplicant_remove_iface(wpa_s->global, wpa_s, 0);
 514                 wpa_s = global->ifaces;
 515                 if (wpa_s && ifname)
 516                         wpa_drv_if_remove(wpa_s, type, ifname);
 517                 os_free(ifname);
 518                 return 1;
 519         }
 520
 521         if (!wpa_s->p2p_go_group_formation_completed) {
 522                 wpa_s->global->p2p_group_formation = NULL;
 523                 wpa_s->p2p_in_provisioning = 0;
 524         }
 525
 526         wpa_s->show_group_started = 0;
 527         os_free(wpa_s->go_params);
 528         wpa_s->go_params = NULL;
 529
 530         wpa_s->waiting_presence_resp = 0;
 531
 532         wpa_printf(MSG_DEBUG, "P2P: Remove temporary group network");
 533         if (ssid && (ssid->p2p_group ||
 534                      ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION ||
 535                      (ssid->key_mgmt & WPA_KEY_MGMT_WPS))) {
 536                 int id = ssid->id;
 537                 if (ssid == wpa_s->current_ssid) {
 538                         wpa_sm_set_config(wpa_s->wpa, NULL);
 539                         eapol_sm_notify_config(wpa_s->eapol, NULL, NULL);
 540                         wpa_s->current_ssid = NULL;
 541                 }
 542                 /*
 543                  * Networks objects created during any P2P activities are not
 544                  * exposed out as they might/will confuse certain non-P2P aware
 545                  * applications since these network objects won't behave like
 546                  * regular ones.
 547                  *
 548                  * Likewise, we don't send out network removed signals for such
 549                  * network objects.
 550                  */
 551                 wpa_config_remove_network(wpa_s->conf, id);
 552                 wpa_supplicant_clear_status(wpa_s);
 553                 wpa_supplicant_cancel_sched_scan(wpa_s);
 554         } else {
 555                 wpa_printf(MSG_DEBUG, "P2P: Temporary group network not "
 556                            "found");
 557         }
 558         if (wpa_s->ap_iface)
 559                 wpa_supplicant_ap_deinit(wpa_s);
 560         else
 561                 wpa_drv_deinit_p2p_cli(wpa_s);
 562
 563         return 0;
 564 }
 565
 566
 567 static int wpas_p2p_persistent_group(struct wpa_supplicant *wpa_s,
 568                                      u8 *go_dev_addr,
 569                                      const u8 *ssid, size_t ssid_len)
 570 {
 571         struct wpa_bss *bss;
 572         const u8 *bssid;
 573         struct wpabuf *p2p;
 574         u8 group_capab;
 575         const u8 *addr;
 576
 577         if (wpa_s->go_params)
 578                 bssid = wpa_s->go_params->peer_interface_addr;
 579         else
 580                 bssid = wpa_s->bssid;
 581
 582         bss = wpa_bss_get(wpa_s, bssid, ssid, ssid_len);
 583         if (bss == NULL) {
 584                 u8 iface_addr[ETH_ALEN];
 585                 if (p2p_get_interface_addr(wpa_s->global->p2p, bssid,
 586                                            iface_addr) == 0)
 587                         bss = wpa_bss_get(wpa_s, iface_addr, ssid, ssid_len);
 588         }
 589         if (bss == NULL) {
 590                 wpa_printf(MSG_DEBUG, "P2P: Could not figure out whether "
 591                            "group is persistent - BSS " MACSTR " not found",
 592                            MAC2STR(bssid));
 593                 return 0;
 594         }
 595
 596         p2p = wpa_bss_get_vendor_ie_multi(bss, P2P_IE_VENDOR_TYPE);
 597         if (p2p == NULL)
 598                 p2p = wpa_bss_get_vendor_ie_multi_beacon(bss,
 599                                                          P2P_IE_VENDOR_TYPE);
 600         if (p2p == NULL) {
 601                 wpa_printf(MSG_DEBUG, "P2P: Could not figure out whether "
 602                            "group is persistent - BSS " MACSTR
 603                            " did not include P2P IE", MAC2STR(bssid));
 604                 wpa_hexdump(MSG_DEBUG, "P2P: Probe Response IEs",
 605                             (u8 *) (bss + 1), bss->ie_len);
 606                 wpa_hexdump(MSG_DEBUG, "P2P: Beacon IEs",
 607                             ((u8 *) bss + 1) + bss->ie_len,
 608                             bss->beacon_ie_len);
 609                 return 0;
 610         }
 611
 612         group_capab = p2p_get_group_capab(p2p);
 613         addr = p2p_get_go_dev_addr(p2p);
 614         wpa_printf(MSG_DEBUG, "P2P: Checking whether group is persistent: "
 615                    "group_capab=0x%x", group_capab);
 616         if (addr) {
 617                 os_memcpy(go_dev_addr, addr, ETH_ALEN);
 618                 wpa_printf(MSG_DEBUG, "P2P: GO Device Address " MACSTR,
 619                            MAC2STR(addr));
 620         } else
 621                 os_memset(go_dev_addr, 0, ETH_ALEN);
 622         wpabuf_free(p2p);
 623
 624         wpa_printf(MSG_DEBUG, "P2P: BSS " MACSTR " group_capab=0x%x "
 625                    "go_dev_addr=" MACSTR,
 626                    MAC2STR(bssid), group_capab, MAC2STR(go_dev_addr));
 627
 628         return group_capab & P2P_GROUP_CAPAB_PERSISTENT_GROUP;
 629 }
 630
 631
 632 static int wpas_p2p_store_persistent_group(struct wpa_supplicant *wpa_s,
 633                                            struct wpa_ssid *ssid,
 634                                            const u8 *go_dev_addr)
 635 {
 636         struct wpa_ssid *s;
 637         int changed = 0;
 638
 639         wpa_printf(MSG_DEBUG, "P2P: Storing credentials for a persistent "
 640                    "group (GO Dev Addr " MACSTR ")", MAC2STR(go_dev_addr));
 641         for (s = wpa_s->conf->ssid; s; s = s->next) {
 642                 if (s->disabled == 2 &&
 643                     os_memcmp(go_dev_addr, s->bssid, ETH_ALEN) == 0 &&
 644                     s->ssid_len == ssid->ssid_len &&
 645                     os_memcmp(ssid->ssid, s->ssid, ssid->ssid_len) == 0)
 646                         break;
 647         }
 648
 649         if (s) {
 650                 wpa_printf(MSG_DEBUG, "P2P: Update existing persistent group "
 651                            "entry");
 652                 if (ssid->passphrase && !s->passphrase)
 653                         changed = 1;
 654                 else if (ssid->passphrase && s->passphrase &&
 655                          os_strcmp(ssid->passphrase, s->passphrase) != 0)
 656                         changed = 1;
 657         } else {
 658                 wpa_printf(MSG_DEBUG, "P2P: Create a new persistent group "
 659                            "entry");
 660                 changed = 1;
 661                 s = wpa_config_add_network(wpa_s->conf);
 662                 if (s == NULL)
 663                         return -1;
 664
 665                 /*
 666                  * Instead of network_added we emit persistent_group_added
 667                  * notification. Also to keep the defense checks in
 668                  * persistent_group obj registration method, we set the
 669                  * relevant flags in s to designate it as a persistent group.
 670                  */
 671                 s->p2p_group = 1;
 672                 s->p2p_persistent_group = 1;
 673                 wpas_notify_persistent_group_added(wpa_s, s);
 674                 wpa_config_set_network_defaults(s);
 675         }
 676
 677         s->p2p_group = 1;
 678         s->p2p_persistent_group = 1;
 679         s->disabled = 2;
 680         s->bssid_set = 1;
 681         os_memcpy(s->bssid, go_dev_addr, ETH_ALEN);
 682         s->mode = ssid->mode;
 683         s->auth_alg = WPA_AUTH_ALG_OPEN;
 684         s->key_mgmt = WPA_KEY_MGMT_PSK;
 685         s->proto = WPA_PROTO_RSN;
 686         s->pairwise_cipher = WPA_CIPHER_CCMP;
 687         s->export_keys = 1;
 688         if (ssid->passphrase) {
 689                 os_free(s->passphrase);
 690                 s->passphrase = os_strdup(ssid->passphrase);
 691         }
 692         if (ssid->psk_set) {
 693                 s->psk_set = 1;
 694                 os_memcpy(s->psk, ssid->psk, 32);
 695         }
 696         if (s->passphrase && !s->psk_set)
 697                 wpa_config_update_psk(s);
 698         if (s->ssid == NULL || s->ssid_len < ssid->ssid_len) {
 699                 os_free(s->ssid);
 700                 s->ssid = os_malloc(ssid->ssid_len);
 701         }
 702         if (s->ssid) {
 703                 s->ssid_len = ssid->ssid_len;
 704                 os_memcpy(s->ssid, ssid->ssid, s->ssid_len);
 705         }
 706         if (ssid->mode == WPAS_MODE_P2P_GO && wpa_s->global->add_psk) {
 707                 dl_list_add(&s->psk_list, &wpa_s->global->add_psk->list);
 708                 wpa_s->global->add_psk = NULL;
 709                 changed = 1;
 710         }
 711
 712 #ifndef CONFIG_NO_CONFIG_WRITE
 713         if (changed && wpa_s->conf->update_config &&
 714             wpa_config_write(wpa_s->confname, wpa_s->conf)) {
 715                 wpa_printf(MSG_DEBUG, "P2P: Failed to update configuration");
 716         }
 717 #endif /* CONFIG_NO_CONFIG_WRITE */
 718
 719         return s->id;
 720 }
 721
 722
 723 static void wpas_p2p_add_persistent_group_client(struct wpa_supplicant *wpa_s,
 724                                                  const u8 *addr)
 725 {
 726         struct wpa_ssid *ssid, *s;
 727         u8 *n;
 728         size_t i;
 729         int found = 0;
 730
 731         ssid = wpa_s->current_ssid;
 732         if (ssid == NULL || ssid->mode != WPAS_MODE_P2P_GO ||
 733             !ssid->p2p_persistent_group)
 734                 return;
 735
 736         for (s = wpa_s->parent->conf->ssid; s; s = s->next) {
 737                 if (s->disabled != 2 || s->mode != WPAS_MODE_P2P_GO)
 738                         continue;
 739
 740                 if (s->ssid_len == ssid->ssid_len &&
 741                     os_memcmp(s->ssid, ssid->ssid, s->ssid_len) == 0)
 742                         break;
 743         }
 744
 745         if (s == NULL)
 746                 return;
 747
 748         for (i = 0; s->p2p_client_list && i < s->num_p2p_clients; i++) {
 749                 if (os_memcmp(s->p2p_client_list + i * ETH_ALEN, addr,
 750                               ETH_ALEN) != 0)
 751                         continue;
 752
 753                 if (i == s->num_p2p_clients - 1)
 754                         return; /* already the most recent entry */
 755
 756                 /* move the entry to mark it most recent */
 757                 os_memmove(s->p2p_client_list + i * ETH_ALEN,
 758                            s->p2p_client_list + (i + 1) * ETH_ALEN,
 759                            (s->num_p2p_clients - i - 1) * ETH_ALEN);
 760                 os_memcpy(s->p2p_client_list +
 761                           (s->num_p2p_clients - 1) * ETH_ALEN, addr, ETH_ALEN);
 762                 found = 1;
 763                 break;
 764         }
 765
 766         if (!found && s->num_p2p_clients < P2P_MAX_STORED_CLIENTS) {
 767                 n = os_realloc_array(s->p2p_client_list,
 768                                      s->num_p2p_clients + 1, ETH_ALEN);
 769                 if (n == NULL)
 770                         return;
 771                 os_memcpy(n + s->num_p2p_clients * ETH_ALEN, addr, ETH_ALEN);
 772                 s->p2p_client_list = n;
 773                 s->num_p2p_clients++;
 774         } else if (!found) {
 775                 /* Not enough room for an additional entry - drop the oldest
 776                  * entry */
 777                 os_memmove(s->p2p_client_list,
 778                            s->p2p_client_list + ETH_ALEN,
 779                            (s->num_p2p_clients - 1) * ETH_ALEN);
 780                 os_memcpy(s->p2p_client_list +
 781                           (s->num_p2p_clients - 1) * ETH_ALEN,
 782                           addr, ETH_ALEN);
 783         }
 784
 785 #ifndef CONFIG_NO_CONFIG_WRITE
 786         if (wpa_s->parent->conf->update_config &&
 787             wpa_config_write(wpa_s->parent->confname, wpa_s->parent->conf))
 788                 wpa_printf(MSG_DEBUG, "P2P: Failed to update configuration");
 789 #endif /* CONFIG_NO_CONFIG_WRITE */
 790 }
 791
 792
 793 static void wpas_group_formation_completed(struct wpa_supplicant *wpa_s,
 794                                            int success)
 795 {
 796         struct wpa_ssid *ssid;
 797         const char *ssid_txt;
 798         int client;
 799         int persistent;
 800         u8 go_dev_addr[ETH_ALEN];
 801         int network_id = -1;
 802
 803         /*
 804          * This callback is likely called for the main interface. Update wpa_s
 805          * to use the group interface if a new interface was created for the
 806          * group.
 807          */
 808         if (wpa_s->global->p2p_group_formation)
 809                 wpa_s = wpa_s->global->p2p_group_formation;
 810         if (wpa_s->p2p_go_group_formation_completed) {
 811                 wpa_s->global->p2p_group_formation = NULL;
 812                 wpa_s->p2p_in_provisioning = 0;
 813         }
 814
 815         if (!success) {
 816                 wpa_msg_global(wpa_s->parent, MSG_INFO,
 817                                P2P_EVENT_GROUP_FORMATION_FAILURE);
 818                 wpas_p2p_group_delete(wpa_s,
 819                                       P2P_GROUP_REMOVAL_FORMATION_FAILED);
 820                 return;
 821         }
 822
 823         wpa_msg_global(wpa_s->parent, MSG_INFO,
 824                        P2P_EVENT_GROUP_FORMATION_SUCCESS);
 825
 826         ssid = wpa_s->current_ssid;
 827         if (ssid && ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION) {
 828                 ssid->mode = WPAS_MODE_P2P_GO;
 829                 p2p_group_notif_formation_done(wpa_s->p2p_group);
 830                 wpa_supplicant_ap_mac_addr_filter(wpa_s, NULL);
 831         }
 832
 833         persistent = 0;
 834         if (ssid) {
 835                 ssid_txt = wpa_ssid_txt(ssid->ssid, ssid->ssid_len);
 836                 client = ssid->mode == WPAS_MODE_INFRA;
 837                 if (ssid->mode == WPAS_MODE_P2P_GO) {
 838                         persistent = ssid->p2p_persistent_group;
 839                         os_memcpy(go_dev_addr, wpa_s->global->p2p_dev_addr,
 840                                   ETH_ALEN);
 841                 } else
 842                         persistent = wpas_p2p_persistent_group(wpa_s,
 843                                                                go_dev_addr,
 844                                                                ssid->ssid,
 845                                                                ssid->ssid_len);
 846         } else {
 847                 ssid_txt = "";
 848                 client = wpa_s->p2p_group_interface ==
 849                         P2P_GROUP_INTERFACE_CLIENT;
 850                 os_memset(go_dev_addr, 0, ETH_ALEN);
 851         }
 852
 853         wpa_s->show_group_started = 0;
 854         if (client) {
 855                 /*
 856                  * Indicate event only after successfully completed 4-way
 857                  * handshake, i.e., when the interface is ready for data
 858                  * packets.
 859                  */
 860                 wpa_s->show_group_started = 1;
 861         } else if (ssid && ssid->passphrase == NULL && ssid->psk_set) {
 862                 char psk[65];
 863                 wpa_snprintf_hex(psk, sizeof(psk), ssid->psk, 32);
 864                 wpa_msg_global(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
 865                                "%s GO ssid=\"%s\" freq=%d psk=%s go_dev_addr="
 866                                MACSTR "%s",
 867                                wpa_s->ifname, ssid_txt, ssid->frequency, psk,
 868                                MAC2STR(go_dev_addr),
 869                                persistent ? " [PERSISTENT]" : "");
 870                 wpas_p2p_cross_connect_setup(wpa_s);
 871                 wpas_p2p_set_group_idle_timeout(wpa_s);
 872         } else {
 873                 wpa_msg_global(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
 874                                "%s GO ssid=\"%s\" freq=%d passphrase=\"%s\" "
 875                                "go_dev_addr=" MACSTR "%s",
 876                                wpa_s->ifname, ssid_txt,
 877                                ssid ? ssid->frequency : 0,
 878                                ssid && ssid->passphrase ? ssid->passphrase : "",
 879                                MAC2STR(go_dev_addr),
 880                                persistent ? " [PERSISTENT]" : "");
 881                 wpas_p2p_cross_connect_setup(wpa_s);
 882                 wpas_p2p_set_group_idle_timeout(wpa_s);
 883         }
 884
 885         if (persistent)
 886                 network_id = wpas_p2p_store_persistent_group(wpa_s->parent,
 887                                                              ssid, go_dev_addr);
 888         else {
 889                 os_free(wpa_s->global->add_psk);
 890                 wpa_s->global->add_psk = NULL;
 891         }
 892         if (network_id < 0 && ssid)
 893                 network_id = ssid->id;
 894         if (!client) {
 895                 wpas_notify_p2p_group_started(wpa_s, ssid, network_id, 0);
 896                 os_get_reltime(&wpa_s->global->p2p_go_wait_client);
 897         }
 898 }
 899
 900
 901 struct send_action_work {
 902         unsigned int freq;
 903         u8 dst[ETH_ALEN];
 904         u8 src[ETH_ALEN];
 905         u8 bssid[ETH_ALEN];
 906         size_t len;
 907         unsigned int wait_time;
 908         u8 buf[0];
 909 };
 910
 911
 912 static void wpas_p2p_send_action_work_timeout(void *eloop_ctx,
 913                                               void *timeout_ctx)
 914 {
 915         struct wpa_supplicant *wpa_s = eloop_ctx;
 916
 917         if (!wpa_s->p2p_send_action_work)
 918                 return;
 919
 920         wpa_printf(MSG_DEBUG, "P2P: Send Action frame radio work timed out");
 921         os_free(wpa_s->p2p_send_action_work->ctx);
 922         radio_work_done(wpa_s->p2p_send_action_work);
 923         wpa_s->p2p_send_action_work = NULL;
 924 }
 925
 926
 927 static void wpas_p2p_send_action_tx_status(struct wpa_supplicant *wpa_s,
 928                                            unsigned int freq,
 929                                            const u8 *dst, const u8 *src,
 930                                            const u8 *bssid,
 931                                            const u8 *data, size_t data_len,
 932                                            enum offchannel_send_action_result
 933                                            result)
 934 {
 935         enum p2p_send_action_result res = P2P_SEND_ACTION_SUCCESS;
 936
 937         if (wpa_s->p2p_send_action_work) {
 938                 struct send_action_work *awork;
 939                 awork = wpa_s->p2p_send_action_work->ctx;
 940                 if (awork->wait_time == 0) {
 941                         os_free(awork);
 942                         radio_work_done(wpa_s->p2p_send_action_work);
 943                         wpa_s->p2p_send_action_work = NULL;
 944                 } else {
 945                         /*
 946                          * In theory, this should not be needed, but number of
 947                          * places in the P2P code is still using non-zero wait
 948                          * time for the last Action frame in the sequence and
 949                          * some of these do not call send_action_done().
 950                          */
 951                         eloop_cancel_timeout(wpas_p2p_send_action_work_timeout,
 952                                              wpa_s, NULL);
 953                         eloop_register_timeout(
 954                                 0, awork->wait_time * 1000,
 955                                 wpas_p2p_send_action_work_timeout,
 956                                 wpa_s, NULL);
 957                 }
 958         }
 959
 960         if (wpa_s->global->p2p == NULL || wpa_s->global->p2p_disabled)
 961                 return;
 962
 963         switch (result) {
 964         case OFFCHANNEL_SEND_ACTION_SUCCESS:
 965                 res = P2P_SEND_ACTION_SUCCESS;
 966                 break;
 967         case OFFCHANNEL_SEND_ACTION_NO_ACK:
 968                 res = P2P_SEND_ACTION_NO_ACK;
 969                 break;
 970         case OFFCHANNEL_SEND_ACTION_FAILED:
 971                 res = P2P_SEND_ACTION_FAILED;
 972                 break;
 973         }
 974
 975         p2p_send_action_cb(wpa_s->global->p2p, freq, dst, src, bssid, res);
 976
 977         if (result != OFFCHANNEL_SEND_ACTION_SUCCESS &&
 978             wpa_s->pending_pd_before_join &&
 979             (os_memcmp(dst, wpa_s->pending_join_dev_addr, ETH_ALEN) == 0 ||
 980              os_memcmp(dst, wpa_s->pending_join_iface_addr, ETH_ALEN) == 0) &&
 981             wpa_s->p2p_fallback_to_go_neg) {
 982                 wpa_s->pending_pd_before_join = 0;
 983                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: No ACK for PD Req "
 984                         "during p2p_connect-auto");
 985                 wpas_p2p_fallback_to_go_neg(wpa_s, 0);
 986                 return;
 987         }
 988 }
 989
 990
 991 static void wpas_send_action_cb(struct wpa_radio_work *work, int deinit)
 992 {
 993         struct wpa_supplicant *wpa_s = work->wpa_s;
 994         struct send_action_work *awork = work->ctx;
 995
 996         if (deinit) {
 997                 os_free(awork);
 998                 return;
 999         }
1000
1001         if (offchannel_send_action(wpa_s, awork->freq, awork->dst, awork->src,
1002                                    awork->bssid, awork->buf, awork->len,
1003                                    awork->wait_time,
1004                                    wpas_p2p_send_action_tx_status, 1) < 0) {
1005                 os_free(awork);
1006                 radio_work_done(work);
1007                 return;
1008         }
1009         wpa_s->p2p_send_action_work = work;
1010 }
1011
1012
1013 static int wpas_send_action_work(struct wpa_supplicant *wpa_s,
1014                                  unsigned int freq, const u8 *dst,
1015                                  const u8 *src, const u8 *bssid, const u8 *buf,
1016                                  size_t len, unsigned int wait_time)
1017 {
1018         struct send_action_work *awork;
1019
1020         if (wpa_s->p2p_send_action_work) {
1021                 wpa_printf(MSG_DEBUG, "P2P: Cannot schedule new p2p-send-action work since one is already pending");
1022                 return -1;
1023         }
1024
1025         awork = os_zalloc(sizeof(*awork) + len);
1026         if (awork == NULL)
1027                 return -1;
1028
1029         awork->freq = freq;
1030         os_memcpy(awork->dst, dst, ETH_ALEN);
1031         os_memcpy(awork->src, src, ETH_ALEN);
1032         os_memcpy(awork->bssid, bssid, ETH_ALEN);
1033         awork->len = len;
1034         awork->wait_time = wait_time;
1035         os_memcpy(awork->buf, buf, len);
1036
1037         if (radio_add_work(wpa_s, freq, "p2p-send-action", 0,
1038                            wpas_send_action_cb, awork) < 0) {
1039                 os_free(awork);
1040                 return -1;
1041         }
1042
1043         return 0;
1044 }
1045
1046
1047 static int wpas_send_action(void *ctx, unsigned int freq, const u8 *dst,
1048                             const u8 *src, const u8 *bssid, const u8 *buf,
1049                             size_t len, unsigned int wait_time)
1050 {
1051         struct wpa_supplicant *wpa_s = ctx;
1052         int listen_freq = -1, send_freq = -1;
1053
1054         if (wpa_s->p2p_listen_work)
1055                 listen_freq = wpa_s->p2p_listen_work->freq;
1056         if (wpa_s->p2p_send_action_work)
1057                 send_freq = wpa_s->p2p_send_action_work->freq;
1058         if (listen_freq != (int) freq && send_freq != (int) freq) {
1059                 wpa_printf(MSG_DEBUG, "P2P: Schedule new radio work for Action frame TX (listen_freq=%d send_freq=%d)",
1060                            listen_freq, send_freq);
1061                 return wpas_send_action_work(wpa_s, freq, dst, src, bssid, buf,
1062                                              len, wait_time);
1063         }
1064
1065         wpa_printf(MSG_DEBUG, "P2P: Use ongoing radio work for Action frame TX");
1066         return offchannel_send_action(wpa_s, freq, dst, src, bssid, buf, len,
1067                                       wait_time,
1068                                       wpas_p2p_send_action_tx_status, 1);
1069 }
1070
1071
1072 static void wpas_send_action_done(void *ctx)
1073 {
1074         struct wpa_supplicant *wpa_s = ctx;
1075
1076         if (wpa_s->p2p_send_action_work) {
1077                 eloop_cancel_timeout(wpas_p2p_send_action_work_timeout,
1078                                      wpa_s, NULL);
1079                 os_free(wpa_s->p2p_send_action_work->ctx);
1080                 radio_work_done(wpa_s->p2p_send_action_work);
1081                 wpa_s->p2p_send_action_work = NULL;
1082         }
1083
1084         offchannel_send_action_done(wpa_s);
1085 }
1086
1087
1088 static int wpas_copy_go_neg_results(struct wpa_supplicant *wpa_s,
1089                                     struct p2p_go_neg_results *params)
1090 {
1091         if (wpa_s->go_params == NULL) {
1092                 wpa_s->go_params = os_malloc(sizeof(*params));
1093                 if (wpa_s->go_params == NULL)
1094                         return -1;
1095         }
1096         os_memcpy(wpa_s->go_params, params, sizeof(*params));
1097         return 0;
1098 }
1099
1100
1101 static void wpas_start_wps_enrollee(struct wpa_supplicant *wpa_s,
1102                                     struct p2p_go_neg_results *res)
1103 {
1104         wpa_printf(MSG_DEBUG, "P2P: Start WPS Enrollee for peer " MACSTR,
1105                    MAC2STR(res->peer_interface_addr));
1106         wpa_hexdump_ascii(MSG_DEBUG, "P2P: Start WPS Enrollee for SSID",
1107                           res->ssid, res->ssid_len);
1108         wpa_supplicant_ap_deinit(wpa_s);
1109         wpas_copy_go_neg_results(wpa_s, res);
1110         if (res->wps_method == WPS_PBC)
1111                 wpas_wps_start_pbc(wpa_s, res->peer_interface_addr, 1);
1112         else {
1113                 u16 dev_pw_id = DEV_PW_DEFAULT;
1114                 if (wpa_s->p2p_wps_method == WPS_PIN_KEYPAD)
1115                         dev_pw_id = DEV_PW_REGISTRAR_SPECIFIED;
1116                 wpas_wps_start_pin(wpa_s, res->peer_interface_addr,
1117                                    wpa_s->p2p_pin, 1, dev_pw_id);
1118         }
1119 }
1120
1121
1122 static void wpas_p2p_add_psk_list(struct wpa_supplicant *wpa_s,
1123                                   struct wpa_ssid *ssid)
1124 {
1125         struct wpa_ssid *persistent;
1126         struct psk_list_entry *psk;
1127         struct hostapd_data *hapd;
1128
1129         if (!wpa_s->ap_iface)
1130                 return;
1131
1132         persistent = wpas_p2p_get_persistent(wpa_s->parent, NULL, ssid->ssid,
1133                                              ssid->ssid_len);
1134         if (persistent == NULL)
1135                 return;
1136
1137         hapd = wpa_s->ap_iface->bss[0];
1138
1139         dl_list_for_each(psk, &persistent->psk_list, struct psk_list_entry,
1140                          list) {
1141                 struct hostapd_wpa_psk *hpsk;
1142
1143                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Add persistent group PSK entry for "
1144                         MACSTR " psk=%d",
1145                         MAC2STR(psk->addr), psk->p2p);
1146                 hpsk = os_zalloc(sizeof(*hpsk));
1147                 if (hpsk == NULL)
1148                         break;
1149                 os_memcpy(hpsk->psk, psk->psk, PMK_LEN);
1150                 if (psk->p2p)
1151                         os_memcpy(hpsk->p2p_dev_addr, psk->addr, ETH_ALEN);
1152                 else
1153                         os_memcpy(hpsk->addr, psk->addr, ETH_ALEN);
1154                 hpsk->next = hapd->conf->ssid.wpa_psk;
1155                 hapd->conf->ssid.wpa_psk = hpsk;
1156         }
1157 }
1158
1159
1160 static void p2p_go_configured(void *ctx, void *data)
1161 {
1162         struct wpa_supplicant *wpa_s = ctx;
1163         struct p2p_go_neg_results *params = data;
1164         struct wpa_ssid *ssid;
1165         int network_id = -1;
1166
1167         ssid = wpa_s->current_ssid;
1168         if (ssid && ssid->mode == WPAS_MODE_P2P_GO) {
1169                 wpa_printf(MSG_DEBUG, "P2P: Group setup without provisioning");
1170                 if (wpa_s->global->p2p_group_formation == wpa_s)
1171                         wpa_s->global->p2p_group_formation = NULL;
1172                 if (os_strlen(params->passphrase) > 0) {
1173                         wpa_msg_global(wpa_s->parent, MSG_INFO,
1174                                        P2P_EVENT_GROUP_STARTED
1175                                        "%s GO ssid=\"%s\" freq=%d "
1176                                        "passphrase=\"%s\" go_dev_addr=" MACSTR
1177                                        "%s", wpa_s->ifname,
1178                                        wpa_ssid_txt(ssid->ssid, ssid->ssid_len),
1179                                        ssid->frequency, params->passphrase,
1180                                        MAC2STR(wpa_s->global->p2p_dev_addr),
1181                                        params->persistent_group ?
1182                                        " [PERSISTENT]" : "");
1183                 } else {
1184                         char psk[65];
1185                         wpa_snprintf_hex(psk, sizeof(psk), params->psk,
1186                                          sizeof(params->psk));
1187                         wpa_msg_global(wpa_s->parent, MSG_INFO,
1188                                        P2P_EVENT_GROUP_STARTED
1189                                        "%s GO ssid=\"%s\" freq=%d psk=%s "
1190                                        "go_dev_addr=" MACSTR "%s",
1191                                        wpa_s->ifname,
1192                                        wpa_ssid_txt(ssid->ssid, ssid->ssid_len),
1193                                        ssid->frequency, psk,
1194                                        MAC2STR(wpa_s->global->p2p_dev_addr),
1195                                        params->persistent_group ?
1196                                        " [PERSISTENT]" : "");
1197                 }
1198
1199                 os_get_reltime(&wpa_s->global->p2p_go_wait_client);
1200                 if (params->persistent_group) {
1201                         network_id = wpas_p2p_store_persistent_group(
1202                                 wpa_s->parent, ssid,
1203                                 wpa_s->global->p2p_dev_addr);
1204                         wpas_p2p_add_psk_list(wpa_s, ssid);
1205                 }
1206                 if (network_id < 0)
1207                         network_id = ssid->id;
1208                 wpas_notify_p2p_group_started(wpa_s, ssid, network_id, 0);
1209                 wpas_p2p_cross_connect_setup(wpa_s);
1210                 wpas_p2p_set_group_idle_timeout(wpa_s);
1211
1212                 if (wpa_s->p2p_first_connection_timeout) {
1213                         wpa_dbg(wpa_s, MSG_DEBUG,
1214                                 "P2P: Start group formation timeout of %d seconds until first data connection on GO",
1215                                 wpa_s->p2p_first_connection_timeout);
1216                         wpa_s->p2p_go_group_formation_completed = 0;
1217                         wpa_s->global->p2p_group_formation = wpa_s;
1218                         eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
1219                                              wpa_s->parent, NULL);
1220                         eloop_register_timeout(
1221                                 wpa_s->p2p_first_connection_timeout, 0,
1222                                 wpas_p2p_group_formation_timeout,
1223                                 wpa_s->parent, NULL);
1224                 }
1225
1226                 return;
1227         }
1228
1229         wpa_printf(MSG_DEBUG, "P2P: Setting up WPS for GO provisioning");
1230         if (wpa_supplicant_ap_mac_addr_filter(wpa_s,
1231                                               params->peer_interface_addr)) {
1232                 wpa_printf(MSG_DEBUG, "P2P: Failed to setup MAC address "
1233                            "filtering");
1234                 return;
1235         }
1236         if (params->wps_method == WPS_PBC)
1237                 wpa_supplicant_ap_wps_pbc(wpa_s, params->peer_interface_addr,
1238                                           params->peer_device_addr);
1239         else if (wpa_s->p2p_pin[0])
1240                 wpa_supplicant_ap_wps_pin(wpa_s, params->peer_interface_addr,
1241                                           wpa_s->p2p_pin, NULL, 0, 0);
1242         os_free(wpa_s->go_params);
1243         wpa_s->go_params = NULL;
1244 }
1245
1246
1247 static void wpas_start_wps_go(struct wpa_supplicant *wpa_s,
1248                               struct p2p_go_neg_results *params,
1249                               int group_formation)
1250 {
1251         struct wpa_ssid *ssid;
1252
1253         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Starting GO");
1254         if (wpas_copy_go_neg_results(wpa_s, params) < 0) {
1255                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Could not copy GO Negotiation "
1256                         "results");
1257                 return;
1258         }
1259
1260         ssid = wpa_config_add_network(wpa_s->conf);
1261         if (ssid == NULL) {
1262                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Could not add network for GO");
1263                 return;
1264         }
1265
1266         wpa_s->show_group_started = 0;
1267
1268         wpa_config_set_network_defaults(ssid);
1269         ssid->temporary = 1;
1270         ssid->p2p_group = 1;
1271         ssid->p2p_persistent_group = params->persistent_group;
1272         ssid->mode = group_formation ? WPAS_MODE_P2P_GROUP_FORMATION :
1273                 WPAS_MODE_P2P_GO;
1274         ssid->frequency = params->freq;
1275         ssid->ht40 = params->ht40;
1276         ssid->vht = params->vht;
1277         ssid->ssid = os_zalloc(params->ssid_len + 1);
1278         if (ssid->ssid) {
1279                 os_memcpy(ssid->ssid, params->ssid, params->ssid_len);
1280                 ssid->ssid_len = params->ssid_len;
1281         }
1282         ssid->auth_alg = WPA_AUTH_ALG_OPEN;
1283         ssid->key_mgmt = WPA_KEY_MGMT_PSK;
1284         ssid->proto = WPA_PROTO_RSN;
1285         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
1286         if (os_strlen(params->passphrase) > 0) {
1287                 ssid->passphrase = os_strdup(params->passphrase);
1288                 if (ssid->passphrase == NULL) {
1289                         wpa_msg_global(wpa_s, MSG_ERROR,
1290                                        "P2P: Failed to copy passphrase for GO");
1291                         wpa_config_remove_network(wpa_s->conf, ssid->id);
1292                         return;
1293                 }
1294         } else
1295                 ssid->passphrase = NULL;
1296         ssid->psk_set = params->psk_set;
1297         if (ssid->psk_set)
1298                 os_memcpy(ssid->psk, params->psk, sizeof(ssid->psk));
1299         else if (ssid->passphrase)
1300                 wpa_config_update_psk(ssid);
1301         ssid->ap_max_inactivity = wpa_s->parent->conf->p2p_go_max_inactivity;
1302
1303         wpa_s->ap_configured_cb = p2p_go_configured;
1304         wpa_s->ap_configured_cb_ctx = wpa_s;
1305         wpa_s->ap_configured_cb_data = wpa_s->go_params;
1306         wpa_s->connect_without_scan = ssid;
1307         wpa_s->reassociate = 1;
1308         wpa_s->disconnected = 0;
1309         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Request scan (that will be skipped) to "
1310                 "start GO)");
1311         wpa_supplicant_req_scan(wpa_s, 0, 0);
1312 }
1313
1314
1315 static void wpas_p2p_clone_config(struct wpa_supplicant *dst,
1316                                   const struct wpa_supplicant *src)
1317 {
1318         struct wpa_config *d;
1319         const struct wpa_config *s;
1320
1321         d = dst->conf;
1322         s = src->conf;
1323
1324 #define C(n) if (s->n) d->n = os_strdup(s->n)
1325         C(device_name);
1326         C(manufacturer);
1327         C(model_name);
1328         C(model_number);
1329         C(serial_number);
1330         C(config_methods);
1331 #undef C
1332
1333         os_memcpy(d->device_type, s->device_type, WPS_DEV_TYPE_LEN);
1334         os_memcpy(d->sec_device_type, s->sec_device_type,
1335                   sizeof(d->sec_device_type));
1336         d->num_sec_device_types = s->num_sec_device_types;
1337
1338         d->p2p_group_idle = s->p2p_group_idle;
1339         d->p2p_intra_bss = s->p2p_intra_bss;
1340         d->persistent_reconnect = s->persistent_reconnect;
1341         d->max_num_sta = s->max_num_sta;
1342         d->pbc_in_m1 = s->pbc_in_m1;
1343         d->ignore_old_scan_res = s->ignore_old_scan_res;
1344         d->beacon_int = s->beacon_int;
1345         d->dtim_period = s->dtim_period;
1346         d->disassoc_low_ack = s->disassoc_low_ack;
1347         d->disable_scan_offload = s->disable_scan_offload;
1348 }
1349
1350
1351 static void wpas_p2p_get_group_ifname(struct wpa_supplicant *wpa_s,
1352                                       char *ifname, size_t len)
1353 {
1354         char *ifname_ptr = wpa_s->ifname;
1355
1356         if (os_strncmp(wpa_s->ifname, P2P_MGMT_DEVICE_PREFIX,
1357                        os_strlen(P2P_MGMT_DEVICE_PREFIX)) == 0) {
1358                 ifname_ptr = os_strrchr(wpa_s->ifname, '-') + 1;
1359         }
1360
1361         os_snprintf(ifname, len, "p2p-%s-%d", ifname_ptr, wpa_s->p2p_group_idx);
1362         if (os_strlen(ifname) >= IFNAMSIZ &&
1363             os_strlen(wpa_s->ifname) < IFNAMSIZ) {
1364                 /* Try to avoid going over the IFNAMSIZ length limit */
1365                 os_snprintf(ifname, len, "p2p-%d", wpa_s->p2p_group_idx);
1366         }
1367 }
1368
1369
1370 static int wpas_p2p_add_group_interface(struct wpa_supplicant *wpa_s,
1371                                         enum wpa_driver_if_type type)
1372 {
1373         char ifname[120], force_ifname[120];
1374
1375         if (wpa_s->pending_interface_name[0]) {
1376                 wpa_printf(MSG_DEBUG, "P2P: Pending virtual interface exists "
1377                            "- skip creation of a new one");
1378                 if (is_zero_ether_addr(wpa_s->pending_interface_addr)) {
1379                         wpa_printf(MSG_DEBUG, "P2P: Pending virtual address "
1380                                    "unknown?! ifname='%s'",
1381                                    wpa_s->pending_interface_name);
1382                         return -1;
1383                 }
1384                 return 0;
1385         }
1386
1387         wpas_p2p_get_group_ifname(wpa_s, ifname, sizeof(ifname));
1388         force_ifname[0] = '\0';
1389
1390         wpa_printf(MSG_DEBUG, "P2P: Create a new interface %s for the group",
1391                    ifname);
1392         wpa_s->p2p_group_idx++;
1393
1394         wpa_s->pending_interface_type = type;
1395         if (wpa_drv_if_add(wpa_s, type, ifname, NULL, NULL, force_ifname,
1396                            wpa_s->pending_interface_addr, NULL) < 0) {
1397                 wpa_printf(MSG_ERROR, "P2P: Failed to create new group "
1398                            "interface");
1399                 return -1;
1400         }
1401
1402         if (force_ifname[0]) {
1403                 wpa_printf(MSG_DEBUG, "P2P: Driver forced interface name %s",
1404                            force_ifname);
1405                 os_strlcpy(wpa_s->pending_interface_name, force_ifname,
1406                            sizeof(wpa_s->pending_interface_name));
1407         } else
1408                 os_strlcpy(wpa_s->pending_interface_name, ifname,
1409                            sizeof(wpa_s->pending_interface_name));
1410         wpa_printf(MSG_DEBUG, "P2P: Created pending virtual interface %s addr "
1411                    MACSTR, wpa_s->pending_interface_name,
1412                    MAC2STR(wpa_s->pending_interface_addr));
1413
1414         return 0;
1415 }
1416
1417
1418 static void wpas_p2p_remove_pending_group_interface(
1419         struct wpa_supplicant *wpa_s)
1420 {
1421         if (!wpa_s->pending_interface_name[0] ||
1422             is_zero_ether_addr(wpa_s->pending_interface_addr))
1423                 return; /* No pending virtual interface */
1424
1425         wpa_printf(MSG_DEBUG, "P2P: Removing pending group interface %s",
1426                    wpa_s->pending_interface_name);
1427         wpa_drv_if_remove(wpa_s, wpa_s->pending_interface_type,
1428                           wpa_s->pending_interface_name);
1429         os_memset(wpa_s->pending_interface_addr, 0, ETH_ALEN);
1430         wpa_s->pending_interface_name[0] = '\0';
1431 }
1432
1433
1434 static struct wpa_supplicant *
1435 wpas_p2p_init_group_interface(struct wpa_supplicant *wpa_s, int go)
1436 {
1437         struct wpa_interface iface;
1438         struct wpa_supplicant *group_wpa_s;
1439
1440         if (!wpa_s->pending_interface_name[0]) {
1441                 wpa_printf(MSG_ERROR, "P2P: No pending group interface");
1442                 if (!wpas_p2p_create_iface(wpa_s))
1443                         return NULL;
1444                 /*
1445                  * Something has forced us to remove the pending interface; try
1446                  * to create a new one and hope for the best that we will get
1447                  * the same local address.
1448                  */
1449                 if (wpas_p2p_add_group_interface(wpa_s, go ? WPA_IF_P2P_GO :
1450                                                  WPA_IF_P2P_CLIENT) < 0)
1451                         return NULL;
1452         }
1453
1454         os_memset(&iface, 0, sizeof(iface));
1455         iface.ifname = wpa_s->pending_interface_name;
1456         iface.driver = wpa_s->driver->name;
1457         if (wpa_s->conf->ctrl_interface == NULL &&
1458             wpa_s->parent != wpa_s &&
1459             wpa_s->p2p_mgmt &&
1460             (wpa_s->drv_flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE))
1461                 iface.ctrl_interface = wpa_s->parent->conf->ctrl_interface;
1462         else
1463                 iface.ctrl_interface = wpa_s->conf->ctrl_interface;
1464         iface.driver_param = wpa_s->conf->driver_param;
1465         group_wpa_s = wpa_supplicant_add_iface(wpa_s->global, &iface);
1466         if (group_wpa_s == NULL) {
1467                 wpa_printf(MSG_ERROR, "P2P: Failed to create new "
1468                            "wpa_supplicant interface");
1469                 return NULL;
1470         }
1471         wpa_s->pending_interface_name[0] = '\0';
1472         group_wpa_s->parent = wpa_s;
1473         group_wpa_s->p2p_group_interface = go ? P2P_GROUP_INTERFACE_GO :
1474                 P2P_GROUP_INTERFACE_CLIENT;
1475         wpa_s->global->p2p_group_formation = group_wpa_s;
1476
1477         wpas_p2p_clone_config(group_wpa_s, wpa_s);
1478
1479         return group_wpa_s;
1480 }
1481
1482
1483 static void wpas_p2p_group_formation_timeout(void *eloop_ctx,
1484                                              void *timeout_ctx)
1485 {
1486         struct wpa_supplicant *wpa_s = eloop_ctx;
1487         wpa_printf(MSG_DEBUG, "P2P: Group Formation timed out");
1488         wpas_p2p_group_formation_failed(wpa_s);
1489 }
1490
1491
1492 void wpas_p2p_group_formation_failed(struct wpa_supplicant *wpa_s)
1493 {
1494         eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
1495                              wpa_s->parent, NULL);
1496         if (wpa_s->global->p2p)
1497                 p2p_group_formation_failed(wpa_s->global->p2p);
1498         wpas_group_formation_completed(wpa_s, 0);
1499 }
1500
1501
1502 static void wpas_p2p_grpform_fail_after_wps(struct wpa_supplicant *wpa_s)
1503 {
1504         wpa_printf(MSG_DEBUG, "P2P: Reject group formation due to WPS provisioning failure");
1505         eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
1506                              wpa_s->parent, NULL);
1507         eloop_register_timeout(0, 0, wpas_p2p_group_formation_timeout,
1508                                wpa_s->parent, NULL);
1509         wpa_s->global->p2p_fail_on_wps_complete = 0;
1510 }
1511
1512
1513 void wpas_p2p_ap_setup_failed(struct wpa_supplicant *wpa_s)
1514 {
1515         if (wpa_s->global->p2p_group_formation != wpa_s)
1516                 return;
1517         /* Speed up group formation timeout since this cannot succeed */
1518         eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
1519                              wpa_s->parent, NULL);
1520         eloop_register_timeout(0, 0, wpas_p2p_group_formation_timeout,
1521                                wpa_s->parent, NULL);
1522 }
1523
1524
1525 static void wpas_go_neg_completed(void *ctx, struct p2p_go_neg_results *res)
1526 {
1527         struct wpa_supplicant *wpa_s = ctx;
1528
1529         if (wpa_s->off_channel_freq || wpa_s->roc_waiting_drv_freq) {
1530                 wpa_drv_cancel_remain_on_channel(wpa_s);
1531                 wpa_s->off_channel_freq = 0;
1532                 wpa_s->roc_waiting_drv_freq = 0;
1533         }
1534
1535         if (res->status) {
1536                 wpa_msg_global(wpa_s, MSG_INFO,
1537                                P2P_EVENT_GO_NEG_FAILURE "status=%d",
1538                                res->status);
1539                 wpas_notify_p2p_go_neg_completed(wpa_s, res);
1540                 wpas_p2p_remove_pending_group_interface(wpa_s);
1541                 return;
1542         }
1543
1544         if (wpa_s->p2p_go_ht40)
1545                 res->ht40 = 1;
1546         if (wpa_s->p2p_go_vht)
1547                 res->vht = 1;
1548
1549         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_GO_NEG_SUCCESS "role=%s "
1550                        "freq=%d ht40=%d peer_dev=" MACSTR " peer_iface=" MACSTR
1551                        " wps_method=%s",
1552                        res->role_go ? "GO" : "client", res->freq, res->ht40,
1553                        MAC2STR(res->peer_device_addr),
1554                        MAC2STR(res->peer_interface_addr),
1555                        p2p_wps_method_text(res->wps_method));
1556         wpas_notify_p2p_go_neg_completed(wpa_s, res);
1557
1558         if (res->role_go && wpa_s->p2p_persistent_id >= 0) {
1559                 struct wpa_ssid *ssid;
1560                 ssid = wpa_config_get_network(wpa_s->conf,
1561                                               wpa_s->p2p_persistent_id);
1562                 if (ssid && ssid->disabled == 2 &&
1563                     ssid->mode == WPAS_MODE_P2P_GO && ssid->passphrase) {
1564                         size_t len = os_strlen(ssid->passphrase);
1565                         wpa_printf(MSG_DEBUG, "P2P: Override passphrase based "
1566                                    "on requested persistent group");
1567                         os_memcpy(res->passphrase, ssid->passphrase, len);
1568                         res->passphrase[len] = '\0';
1569                 }
1570         }
1571
1572         if (wpa_s->create_p2p_iface) {
1573                 struct wpa_supplicant *group_wpa_s =
1574                         wpas_p2p_init_group_interface(wpa_s, res->role_go);
1575                 if (group_wpa_s == NULL) {
1576                         wpas_p2p_remove_pending_group_interface(wpa_s);
1577                         return;
1578                 }
1579                 if (group_wpa_s != wpa_s) {
1580                         os_memcpy(group_wpa_s->p2p_pin, wpa_s->p2p_pin,
1581                                   sizeof(group_wpa_s->p2p_pin));
1582                         group_wpa_s->p2p_wps_method = wpa_s->p2p_wps_method;
1583                 }
1584                 os_memset(wpa_s->pending_interface_addr, 0, ETH_ALEN);
1585                 wpa_s->pending_interface_name[0] = '\0';
1586                 group_wpa_s->p2p_in_provisioning = 1;
1587
1588                 if (res->role_go)
1589                         wpas_start_wps_go(group_wpa_s, res, 1);
1590                 else
1591                         wpas_start_wps_enrollee(group_wpa_s, res);
1592         } else {
1593                 wpa_s->p2p_in_provisioning = 1;
1594                 wpa_s->global->p2p_group_formation = wpa_s;
1595
1596                 if (res->role_go)
1597                         wpas_start_wps_go(wpa_s, res, 1);
1598                 else
1599                         wpas_start_wps_enrollee(ctx, res);
1600         }
1601
1602         wpa_s->p2p_long_listen = 0;
1603         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
1604
1605         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s, NULL);
1606         eloop_register_timeout(15 + res->peer_config_timeout / 100,
1607                                (res->peer_config_timeout % 100) * 10000,
1608                                wpas_p2p_group_formation_timeout, wpa_s, NULL);
1609 }
1610
1611
1612 static void wpas_go_neg_req_rx(void *ctx, const u8 *src, u16 dev_passwd_id)
1613 {
1614         struct wpa_supplicant *wpa_s = ctx;
1615         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_GO_NEG_REQUEST MACSTR
1616                        " dev_passwd_id=%u", MAC2STR(src), dev_passwd_id);
1617
1618         wpas_notify_p2p_go_neg_req(wpa_s, src, dev_passwd_id);
1619 }
1620
1621
1622 static void wpas_dev_found(void *ctx, const u8 *addr,
1623                            const struct p2p_peer_info *info,
1624                            int new_device)
1625 {
1626 #ifndef CONFIG_NO_STDOUT_DEBUG
1627         struct wpa_supplicant *wpa_s = ctx;
1628         char devtype[WPS_DEV_TYPE_BUFSIZE];
1629         char *wfd_dev_info_hex = NULL;
1630
1631 #ifdef CONFIG_WIFI_DISPLAY
1632         wfd_dev_info_hex = wifi_display_subelem_hex(info->wfd_subelems,
1633                                                     WFD_SUBELEM_DEVICE_INFO);
1634 #endif /* CONFIG_WIFI_DISPLAY */
1635
1636         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_DEVICE_FOUND MACSTR
1637                        " p2p_dev_addr=" MACSTR
1638                        " pri_dev_type=%s name='%s' config_methods=0x%x "
1639                        "dev_capab=0x%x group_capab=0x%x%s%s",
1640                        MAC2STR(addr), MAC2STR(info->p2p_device_addr),
1641                        wps_dev_type_bin2str(info->pri_dev_type, devtype,
1642                                             sizeof(devtype)),
1643                        info->device_name, info->config_methods,
1644                        info->dev_capab, info->group_capab,
1645                        wfd_dev_info_hex ? " wfd_dev_info=0x" : "",
1646                        wfd_dev_info_hex ? wfd_dev_info_hex : "");
1647 #endif /* CONFIG_NO_STDOUT_DEBUG */
1648
1649         os_free(wfd_dev_info_hex);
1650
1651         wpas_notify_p2p_device_found(ctx, info->p2p_device_addr, new_device);
1652 }
1653
1654
1655 static void wpas_dev_lost(void *ctx, const u8 *dev_addr)
1656 {
1657         struct wpa_supplicant *wpa_s = ctx;
1658
1659         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_DEVICE_LOST
1660                        "p2p_dev_addr=" MACSTR, MAC2STR(dev_addr));
1661
1662         wpas_notify_p2p_device_lost(wpa_s, dev_addr);
1663 }
1664
1665
1666 static void wpas_find_stopped(void *ctx)
1667 {
1668         struct wpa_supplicant *wpa_s = ctx;
1669         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_FIND_STOPPED);
1670 }
1671
1672
1673 struct wpas_p2p_listen_work {
1674         unsigned int freq;
1675         unsigned int duration;
1676         struct wpabuf *probe_resp_ie;
1677 };
1678
1679
1680 static void wpas_p2p_listen_work_free(struct wpas_p2p_listen_work *lwork)
1681 {
1682         if (lwork == NULL)
1683                 return;
1684         wpabuf_free(lwork->probe_resp_ie);
1685         os_free(lwork);
1686 }
1687
1688
1689 static void wpas_p2p_listen_work_done(struct wpa_supplicant *wpa_s)
1690 {
1691         struct wpas_p2p_listen_work *lwork;
1692
1693         if (!wpa_s->p2p_listen_work)
1694                 return;
1695
1696         lwork = wpa_s->p2p_listen_work->ctx;
1697         wpas_p2p_listen_work_free(lwork);
1698         radio_work_done(wpa_s->p2p_listen_work);
1699         wpa_s->p2p_listen_work = NULL;
1700 }
1701
1702
1703 static void wpas_start_listen_cb(struct wpa_radio_work *work, int deinit)
1704 {
1705         struct wpa_supplicant *wpa_s = work->wpa_s;
1706         struct wpas_p2p_listen_work *lwork = work->ctx;
1707
1708         if (deinit) {
1709                 wpas_p2p_listen_work_free(lwork);
1710                 return;
1711         }
1712
1713         wpa_s->p2p_listen_work = work;
1714
1715         wpa_drv_set_ap_wps_ie(wpa_s, NULL, lwork->probe_resp_ie, NULL);
1716
1717         if (wpa_drv_probe_req_report(wpa_s, 1) < 0) {
1718                 wpa_printf(MSG_DEBUG, "P2P: Failed to request the driver to "
1719                            "report received Probe Request frames");
1720                 wpas_p2p_listen_work_done(wpa_s);
1721                 return;
1722         }
1723
1724         wpa_s->pending_listen_freq = lwork->freq;
1725         wpa_s->pending_listen_duration = lwork->duration;
1726
1727         if (wpa_drv_remain_on_channel(wpa_s, lwork->freq, lwork->duration) < 0)
1728         {
1729                 wpa_printf(MSG_DEBUG, "P2P: Failed to request the driver "
1730                            "to remain on channel (%u MHz) for Listen "
1731                            "state", lwork->freq);
1732                 wpas_p2p_listen_work_done(wpa_s);
1733                 wpa_s->pending_listen_freq = 0;
1734                 return;
1735         }
1736         wpa_s->off_channel_freq = 0;
1737         wpa_s->roc_waiting_drv_freq = lwork->freq;
1738 }
1739
1740
1741 static int wpas_start_listen(void *ctx, unsigned int freq,
1742                              unsigned int duration,
1743                              const struct wpabuf *probe_resp_ie)
1744 {
1745         struct wpa_supplicant *wpa_s = ctx;
1746         struct wpas_p2p_listen_work *lwork;
1747
1748         if (wpa_s->p2p_listen_work) {
1749                 wpa_printf(MSG_DEBUG, "P2P: Reject start_listen since p2p_listen_work already exists");
1750                 return -1;
1751         }
1752
1753         lwork = os_zalloc(sizeof(*lwork));
1754         if (lwork == NULL)
1755                 return -1;
1756         lwork->freq = freq;
1757         lwork->duration = duration;
1758         if (probe_resp_ie) {
1759                 lwork->probe_resp_ie = wpabuf_dup(probe_resp_ie);
1760                 if (lwork->probe_resp_ie == NULL) {
1761                         wpas_p2p_listen_work_free(lwork);
1762                         return -1;
1763                 }
1764         }
1765
1766         if (radio_add_work(wpa_s, freq, "p2p-listen", 0, wpas_start_listen_cb,
1767                            lwork) < 0) {
1768                 wpas_p2p_listen_work_free(lwork);
1769                 return -1;
1770         }
1771
1772         return 0;
1773 }
1774
1775
1776 static void wpas_stop_listen(void *ctx)
1777 {
1778         struct wpa_supplicant *wpa_s = ctx;
1779         if (wpa_s->off_channel_freq || wpa_s->roc_waiting_drv_freq) {
1780                 wpa_drv_cancel_remain_on_channel(wpa_s);
1781                 wpa_s->off_channel_freq = 0;
1782                 wpa_s->roc_waiting_drv_freq = 0;
1783         }
1784         wpa_drv_set_ap_wps_ie(wpa_s, NULL, NULL, NULL);
1785         wpa_drv_probe_req_report(wpa_s, 0);
1786         wpas_p2p_listen_work_done(wpa_s);
1787 }
1788
1789
1790 static int wpas_send_probe_resp(void *ctx, const struct wpabuf *buf)
1791 {
1792         struct wpa_supplicant *wpa_s = ctx;
1793         return wpa_drv_send_mlme(wpa_s, wpabuf_head(buf), wpabuf_len(buf), 1);
1794 }
1795
1796
1797 /*
1798  * DNS Header section is used only to calculate compression pointers, so the
1799  * contents of this data does not matter, but the length needs to be reserved
1800  * in the virtual packet.
1801  */
1802 #define DNS_HEADER_LEN 12
1803
1804 /*
1805  * 27-octet in-memory packet from P2P specification containing two implied
1806  * queries for _tcp.lcoal. PTR IN and _udp.local. PTR IN
1807  */
1808 #define P2P_SD_IN_MEMORY_LEN 27
1809
1810 static int p2p_sd_dns_uncompress_label(char **upos, char *uend, u8 *start,
1811                                        u8 **spos, const u8 *end)
1812 {
1813         while (*spos < end) {
1814                 u8 val = ((*spos)[0] & 0xc0) >> 6;
1815                 int len;
1816
1817                 if (val == 1 || val == 2) {
1818                         /* These are reserved values in RFC 1035 */
1819                         wpa_printf(MSG_DEBUG, "P2P: Invalid domain name "
1820                                    "sequence starting with 0x%x", val);
1821                         return -1;
1822                 }
1823
1824                 if (val == 3) {
1825                         u16 offset;
1826                         u8 *spos_tmp;
1827
1828                         /* Offset */
1829                         if (*spos + 2 > end) {
1830                                 wpa_printf(MSG_DEBUG, "P2P: No room for full "
1831                                            "DNS offset field");
1832                                 return -1;
1833                         }
1834
1835                         offset = (((*spos)[0] & 0x3f) << 8) | (*spos)[1];
1836                         if (offset >= *spos - start) {
1837                                 wpa_printf(MSG_DEBUG, "P2P: Invalid DNS "
1838                                            "pointer offset %u", offset);
1839                                 return -1;
1840                         }
1841
1842                         (*spos) += 2;
1843                         spos_tmp = start + offset;
1844                         return p2p_sd_dns_uncompress_label(upos, uend, start,
1845                                                            &spos_tmp,
1846                                                            *spos - 2);
1847                 }
1848
1849                 /* Label */
1850                 len = (*spos)[0] & 0x3f;
1851                 if (len == 0)
1852                         return 0;
1853
1854                 (*spos)++;
1855                 if (*spos + len > end) {
1856                         wpa_printf(MSG_DEBUG, "P2P: Invalid domain name "
1857                                    "sequence - no room for label with length "
1858                                    "%u", len);
1859                         return -1;
1860                 }
1861
1862                 if (*upos + len + 2 > uend)
1863                         return -2;
1864
1865                 os_memcpy(*upos, *spos, len);
1866                 *spos += len;
1867                 *upos += len;
1868                 (*upos)[0] = '.';
1869                 (*upos)++;
1870                 (*upos)[0] = '\0';
1871         }
1872
1873         return 0;
1874 }
1875
1876
1877 /* Uncompress domain names per RFC 1035 using the P2P SD in-memory packet.
1878  * Returns -1 on parsing error (invalid input sequence), -2 if output buffer is
1879  * not large enough */
1880 static int p2p_sd_dns_uncompress(char *buf, size_t buf_len, const u8 *msg,
1881                                  size_t msg_len, size_t offset)
1882 {
1883         /* 27-octet in-memory packet from P2P specification */
1884         const char *prefix = "\x04_tcp\x05local\x00\x00\x0C\x00\x01"
1885                 "\x04_udp\xC0\x11\x00\x0C\x00\x01";
1886         u8 *tmp, *end, *spos;
1887         char *upos, *uend;
1888         int ret = 0;
1889
1890         if (buf_len < 2)
1891                 return -1;
1892         if (offset > msg_len)
1893                 return -1;
1894
1895         tmp = os_malloc(DNS_HEADER_LEN + P2P_SD_IN_MEMORY_LEN + msg_len);
1896         if (tmp == NULL)
1897                 return -1;
1898         spos = tmp + DNS_HEADER_LEN + P2P_SD_IN_MEMORY_LEN;
1899         end = spos + msg_len;
1900         spos += offset;
1901
1902         os_memset(tmp, 0, DNS_HEADER_LEN);
1903         os_memcpy(tmp + DNS_HEADER_LEN, prefix, P2P_SD_IN_MEMORY_LEN);
1904         os_memcpy(tmp + DNS_HEADER_LEN + P2P_SD_IN_MEMORY_LEN, msg, msg_len);
1905
1906         upos = buf;
1907         uend = buf + buf_len;
1908
1909         ret = p2p_sd_dns_uncompress_label(&upos, uend, tmp, &spos, end);
1910         if (ret) {
1911                 os_free(tmp);
1912                 return ret;
1913         }
1914
1915         if (upos == buf) {
1916                 upos[0] = '.';
1917                 upos[1] = '\0';
1918         } else if (upos[-1] == '.')
1919                 upos[-1] = '\0';
1920
1921         os_free(tmp);
1922         return 0;
1923 }
1924
1925
1926 static struct p2p_srv_bonjour *
1927 wpas_p2p_service_get_bonjour(struct wpa_supplicant *wpa_s,
1928                              const struct wpabuf *query)
1929 {
1930         struct p2p_srv_bonjour *bsrv;
1931         size_t len;
1932
1933         len = wpabuf_len(query);
1934         dl_list_for_each(bsrv, &wpa_s->global->p2p_srv_bonjour,
1935                          struct p2p_srv_bonjour, list) {
1936                 if (len == wpabuf_len(bsrv->query) &&
1937                     os_memcmp(wpabuf_head(query), wpabuf_head(bsrv->query),
1938                               len) == 0)
1939                         return bsrv;
1940         }
1941         return NULL;
1942 }
1943
1944
1945 static struct p2p_srv_upnp *
1946 wpas_p2p_service_get_upnp(struct wpa_supplicant *wpa_s, u8 version,
1947                           const char *service)
1948 {
1949         struct p2p_srv_upnp *usrv;
1950
1951         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
1952                          struct p2p_srv_upnp, list) {
1953                 if (version == usrv->version &&
1954                     os_strcmp(service, usrv->service) == 0)
1955                         return usrv;
1956         }
1957         return NULL;
1958 }
1959
1960
1961 static void wpas_sd_add_proto_not_avail(struct wpabuf *resp, u8 srv_proto,
1962                                         u8 srv_trans_id)
1963 {
1964         u8 *len_pos;
1965
1966         if (wpabuf_tailroom(resp) < 5)
1967                 return;
1968
1969         /* Length (to be filled) */
1970         len_pos = wpabuf_put(resp, 2);
1971         wpabuf_put_u8(resp, srv_proto);
1972         wpabuf_put_u8(resp, srv_trans_id);
1973         /* Status Code */
1974         wpabuf_put_u8(resp, P2P_SD_PROTO_NOT_AVAILABLE);
1975         /* Response Data: empty */
1976         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
1977 }
1978
1979
1980 static void wpas_sd_all_bonjour(struct wpa_supplicant *wpa_s,
1981                                 struct wpabuf *resp, u8 srv_trans_id)
1982 {
1983         struct p2p_srv_bonjour *bsrv;
1984         u8 *len_pos;
1985
1986         wpa_printf(MSG_DEBUG, "P2P: SD Request for all Bonjour services");
1987
1988         if (dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
1989                 wpa_printf(MSG_DEBUG, "P2P: Bonjour protocol not available");
1990                 return;
1991         }
1992
1993         dl_list_for_each(bsrv, &wpa_s->global->p2p_srv_bonjour,
1994                          struct p2p_srv_bonjour, list) {
1995                 if (wpabuf_tailroom(resp) <
1996                     5 + wpabuf_len(bsrv->query) + wpabuf_len(bsrv->resp))
1997                         return;
1998                 /* Length (to be filled) */
1999                 len_pos = wpabuf_put(resp, 2);
2000                 wpabuf_put_u8(resp, P2P_SERV_BONJOUR);
2001                 wpabuf_put_u8(resp, srv_trans_id);
2002                 /* Status Code */
2003                 wpabuf_put_u8(resp, P2P_SD_SUCCESS);
2004                 wpa_hexdump_ascii(MSG_DEBUG, "P2P: Matching Bonjour service",
2005                                   wpabuf_head(bsrv->resp),
2006                                   wpabuf_len(bsrv->resp));
2007                 /* Response Data */
2008                 wpabuf_put_buf(resp, bsrv->query); /* Key */
2009                 wpabuf_put_buf(resp, bsrv->resp); /* Value */
2010                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
2011                              2);
2012         }
2013 }
2014
2015
2016 static int match_bonjour_query(struct p2p_srv_bonjour *bsrv, const u8 *query,
2017                                size_t query_len)
2018 {
2019         char str_rx[256], str_srv[256];
2020
2021         if (query_len < 3 || wpabuf_len(bsrv->query) < 3)
2022                 return 0; /* Too short to include DNS Type and Version */
2023         if (os_memcmp(query + query_len - 3,
2024                       wpabuf_head_u8(bsrv->query) + wpabuf_len(bsrv->query) - 3,
2025                       3) != 0)
2026                 return 0; /* Mismatch in DNS Type or Version */
2027         if (query_len == wpabuf_len(bsrv->query) &&
2028             os_memcmp(query, wpabuf_head(bsrv->query), query_len - 3) == 0)
2029                 return 1; /* Binary match */
2030
2031         if (p2p_sd_dns_uncompress(str_rx, sizeof(str_rx), query, query_len - 3,
2032                                   0))
2033                 return 0; /* Failed to uncompress query */
2034         if (p2p_sd_dns_uncompress(str_srv, sizeof(str_srv),
2035                                   wpabuf_head(bsrv->query),
2036                                   wpabuf_len(bsrv->query) - 3, 0))
2037                 return 0; /* Failed to uncompress service */
2038
2039         return os_strcmp(str_rx, str_srv) == 0;
2040 }
2041
2042
2043 static void wpas_sd_req_bonjour(struct wpa_supplicant *wpa_s,
2044                                 struct wpabuf *resp, u8 srv_trans_id,
2045                                 const u8 *query, size_t query_len)
2046 {
2047         struct p2p_srv_bonjour *bsrv;
2048         u8 *len_pos;
2049         int matches = 0;
2050
2051         wpa_hexdump_ascii(MSG_DEBUG, "P2P: SD Request for Bonjour",
2052                           query, query_len);
2053         if (dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
2054                 wpa_printf(MSG_DEBUG, "P2P: Bonjour protocol not available");
2055                 wpas_sd_add_proto_not_avail(resp, P2P_SERV_BONJOUR,
2056                                             srv_trans_id);
2057                 return;
2058         }
2059
2060         if (query_len == 0) {
2061                 wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
2062                 return;
2063         }
2064
2065         dl_list_for_each(bsrv, &wpa_s->global->p2p_srv_bonjour,
2066                          struct p2p_srv_bonjour, list) {
2067                 if (!match_bonjour_query(bsrv, query, query_len))
2068                         continue;
2069
2070                 if (wpabuf_tailroom(resp) <
2071                     5 + query_len + wpabuf_len(bsrv->resp))
2072                         return;
2073
2074                 matches++;
2075
2076                 /* Length (to be filled) */
2077                 len_pos = wpabuf_put(resp, 2);
2078                 wpabuf_put_u8(resp, P2P_SERV_BONJOUR);
2079                 wpabuf_put_u8(resp, srv_trans_id);
2080
2081                 /* Status Code */
2082                 wpabuf_put_u8(resp, P2P_SD_SUCCESS);
2083                 wpa_hexdump_ascii(MSG_DEBUG, "P2P: Matching Bonjour service",
2084                                   wpabuf_head(bsrv->resp),
2085                                   wpabuf_len(bsrv->resp));
2086
2087                 /* Response Data */
2088                 wpabuf_put_data(resp, query, query_len); /* Key */
2089                 wpabuf_put_buf(resp, bsrv->resp); /* Value */
2090
2091                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
2092         }
2093
2094         if (matches == 0) {
2095                 wpa_printf(MSG_DEBUG, "P2P: Requested Bonjour service not "
2096                            "available");
2097                 if (wpabuf_tailroom(resp) < 5)
2098                         return;
2099
2100                 /* Length (to be filled) */
2101                 len_pos = wpabuf_put(resp, 2);
2102                 wpabuf_put_u8(resp, P2P_SERV_BONJOUR);
2103                 wpabuf_put_u8(resp, srv_trans_id);
2104
2105                 /* Status Code */
2106                 wpabuf_put_u8(resp, P2P_SD_REQUESTED_INFO_NOT_AVAILABLE);
2107                 /* Response Data: empty */
2108                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
2109                              2);
2110         }
2111 }
2112
2113
2114 static void wpas_sd_all_upnp(struct wpa_supplicant *wpa_s,
2115                              struct wpabuf *resp, u8 srv_trans_id)
2116 {
2117         struct p2p_srv_upnp *usrv;
2118         u8 *len_pos;
2119
2120         wpa_printf(MSG_DEBUG, "P2P: SD Request for all UPnP services");
2121
2122         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp)) {
2123                 wpa_printf(MSG_DEBUG, "P2P: UPnP protocol not available");
2124                 return;
2125         }
2126
2127         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
2128                          struct p2p_srv_upnp, list) {
2129                 if (wpabuf_tailroom(resp) < 5 + 1 + os_strlen(usrv->service))
2130                         return;
2131
2132                 /* Length (to be filled) */
2133                 len_pos = wpabuf_put(resp, 2);
2134                 wpabuf_put_u8(resp, P2P_SERV_UPNP);
2135                 wpabuf_put_u8(resp, srv_trans_id);
2136
2137                 /* Status Code */
2138                 wpabuf_put_u8(resp, P2P_SD_SUCCESS);
2139                 /* Response Data */
2140                 wpabuf_put_u8(resp, usrv->version);
2141                 wpa_printf(MSG_DEBUG, "P2P: Matching UPnP Service: %s",
2142                            usrv->service);
2143                 wpabuf_put_str(resp, usrv->service);
2144                 WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos -
2145                              2);
2146         }
2147 }
2148
2149
2150 static void wpas_sd_req_upnp(struct wpa_supplicant *wpa_s,
2151                              struct wpabuf *resp, u8 srv_trans_id,
2152                              const u8 *query, size_t query_len)
2153 {
2154         struct p2p_srv_upnp *usrv;
2155         u8 *len_pos;
2156         u8 version;
2157         char *str;
2158         int count = 0;
2159
2160         wpa_hexdump_ascii(MSG_DEBUG, "P2P: SD Request for UPnP",
2161                           query, query_len);
2162
2163         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp)) {
2164                 wpa_printf(MSG_DEBUG, "P2P: UPnP protocol not available");
2165                 wpas_sd_add_proto_not_avail(resp, P2P_SERV_UPNP,
2166                                             srv_trans_id);
2167                 return;
2168         }
2169
2170         if (query_len == 0) {
2171                 wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
2172                 return;
2173         }
2174
2175         if (wpabuf_tailroom(resp) < 5)
2176                 return;
2177
2178         /* Length (to be filled) */
2179         len_pos = wpabuf_put(resp, 2);
2180         wpabuf_put_u8(resp, P2P_SERV_UPNP);
2181         wpabuf_put_u8(resp, srv_trans_id);
2182
2183         version = query[0];
2184         str = os_malloc(query_len);
2185         if (str == NULL)
2186                 return;
2187         os_memcpy(str, query + 1, query_len - 1);
2188         str[query_len - 1] = '\0';
2189
2190         dl_list_for_each(usrv, &wpa_s->global->p2p_srv_upnp,
2191                          struct p2p_srv_upnp, list) {
2192                 if (version != usrv->version)
2193                         continue;
2194
2195                 if (os_strcmp(str, "ssdp:all") != 0 &&
2196                     os_strstr(usrv->service, str) == NULL)
2197                         continue;
2198
2199                 if (wpabuf_tailroom(resp) < 2)
2200                         break;
2201                 if (count == 0) {
2202                         /* Status Code */
2203                         wpabuf_put_u8(resp, P2P_SD_SUCCESS);
2204                         /* Response Data */
2205                         wpabuf_put_u8(resp, version);
2206                 } else
2207                         wpabuf_put_u8(resp, ',');
2208
2209                 count++;
2210
2211                 wpa_printf(MSG_DEBUG, "P2P: Matching UPnP Service: %s",
2212                            usrv->service);
2213                 if (wpabuf_tailroom(resp) < os_strlen(usrv->service))
2214                         break;
2215                 wpabuf_put_str(resp, usrv->service);
2216         }
2217         os_free(str);
2218
2219         if (count == 0) {
2220                 wpa_printf(MSG_DEBUG, "P2P: Requested UPnP service not "
2221                            "available");
2222                 /* Status Code */
2223                 wpabuf_put_u8(resp, P2P_SD_REQUESTED_INFO_NOT_AVAILABLE);
2224                 /* Response Data: empty */
2225         }
2226
2227         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
2228 }
2229
2230
2231 #ifdef CONFIG_WIFI_DISPLAY
2232 static void wpas_sd_req_wfd(struct wpa_supplicant *wpa_s,
2233                             struct wpabuf *resp, u8 srv_trans_id,
2234                             const u8 *query, size_t query_len)
2235 {
2236         const u8 *pos;
2237         u8 role;
2238         u8 *len_pos;
2239
2240         wpa_hexdump(MSG_DEBUG, "P2P: SD Request for WFD", query, query_len);
2241
2242         if (!wpa_s->global->wifi_display) {
2243                 wpa_printf(MSG_DEBUG, "P2P: WFD protocol not available");
2244                 wpas_sd_add_proto_not_avail(resp, P2P_SERV_WIFI_DISPLAY,
2245                                             srv_trans_id);
2246                 return;
2247         }
2248
2249         if (query_len < 1) {
2250                 wpa_printf(MSG_DEBUG, "P2P: Missing WFD Requested Device "
2251                            "Role");
2252                 return;
2253         }
2254
2255         if (wpabuf_tailroom(resp) < 5)
2256                 return;
2257
2258         pos = query;
2259         role = *pos++;
2260         wpa_printf(MSG_DEBUG, "P2P: WSD for device role 0x%x", role);
2261
2262         /* TODO: role specific handling */
2263
2264         /* Length (to be filled) */
2265         len_pos = wpabuf_put(resp, 2);
2266         wpabuf_put_u8(resp, P2P_SERV_WIFI_DISPLAY);
2267         wpabuf_put_u8(resp, srv_trans_id);
2268         wpabuf_put_u8(resp, P2P_SD_SUCCESS); /* Status Code */
2269
2270         while (pos < query + query_len) {
2271                 if (*pos < MAX_WFD_SUBELEMS &&
2272                     wpa_s->global->wfd_subelem[*pos] &&
2273                     wpabuf_tailroom(resp) >=
2274                     wpabuf_len(wpa_s->global->wfd_subelem[*pos])) {
2275                         wpa_printf(MSG_DEBUG, "P2P: Add WSD response "
2276                                    "subelement %u", *pos);
2277                         wpabuf_put_buf(resp, wpa_s->global->wfd_subelem[*pos]);
2278                 }
2279                 pos++;
2280         }
2281
2282         WPA_PUT_LE16(len_pos, (u8 *) wpabuf_put(resp, 0) - len_pos - 2);
2283 }
2284 #endif /* CONFIG_WIFI_DISPLAY */
2285
2286
2287 static void wpas_sd_request(void *ctx, int freq, const u8 *sa, u8 dialog_token,
2288                             u16 update_indic, const u8 *tlvs, size_t tlvs_len)
2289 {
2290         struct wpa_supplicant *wpa_s = ctx;
2291         const u8 *pos = tlvs;
2292         const u8 *end = tlvs + tlvs_len;
2293         const u8 *tlv_end;
2294         u16 slen;
2295         struct wpabuf *resp;
2296         u8 srv_proto, srv_trans_id;
2297         size_t buf_len;
2298         char *buf;
2299
2300         wpa_hexdump(MSG_MSGDUMP, "P2P: Service Discovery Request TLVs",
2301                     tlvs, tlvs_len);
2302         buf_len = 2 * tlvs_len + 1;
2303         buf = os_malloc(buf_len);
2304         if (buf) {
2305                 wpa_snprintf_hex(buf, buf_len, tlvs, tlvs_len);
2306                 wpa_msg_ctrl(wpa_s, MSG_INFO, P2P_EVENT_SERV_DISC_REQ "%d "
2307                              MACSTR " %u %u %s",
2308                              freq, MAC2STR(sa), dialog_token, update_indic,
2309                              buf);
2310                 os_free(buf);
2311         }
2312
2313         if (wpa_s->p2p_sd_over_ctrl_iface) {
2314                 wpas_notify_p2p_sd_request(wpa_s, freq, sa, dialog_token,
2315                                            update_indic, tlvs, tlvs_len);
2316                 return; /* to be processed by an external program */
2317         }
2318
2319         resp = wpabuf_alloc(10000);
2320         if (resp == NULL)
2321                 return;
2322
2323         while (pos + 1 < end) {
2324                 wpa_printf(MSG_DEBUG, "P2P: Service Request TLV");
2325                 slen = WPA_GET_LE16(pos);
2326                 pos += 2;
2327                 if (pos + slen > end || slen < 2) {
2328                         wpa_printf(MSG_DEBUG, "P2P: Unexpected Query Data "
2329                                    "length");
2330                         wpabuf_free(resp);
2331                         return;
2332                 }
2333                 tlv_end = pos + slen;
2334
2335                 srv_proto = *pos++;
2336                 wpa_printf(MSG_DEBUG, "P2P: Service Protocol Type %u",
2337                            srv_proto);
2338                 srv_trans_id = *pos++;
2339                 wpa_printf(MSG_DEBUG, "P2P: Service Transaction ID %u",
2340                            srv_trans_id);
2341
2342                 wpa_hexdump(MSG_MSGDUMP, "P2P: Query Data",
2343                             pos, tlv_end - pos);
2344
2345
2346                 if (wpa_s->force_long_sd) {
2347                         wpa_printf(MSG_DEBUG, "P2P: SD test - force long "
2348                                    "response");
2349                         wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
2350                         wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
2351                         goto done;
2352                 }
2353
2354                 switch (srv_proto) {
2355                 case P2P_SERV_ALL_SERVICES:
2356                         wpa_printf(MSG_DEBUG, "P2P: Service Discovery Request "
2357                                    "for all services");
2358                         if (dl_list_empty(&wpa_s->global->p2p_srv_upnp) &&
2359                             dl_list_empty(&wpa_s->global->p2p_srv_bonjour)) {
2360                                 wpa_printf(MSG_DEBUG, "P2P: No service "
2361                                            "discovery protocols available");
2362                                 wpas_sd_add_proto_not_avail(
2363                                         resp, P2P_SERV_ALL_SERVICES,
2364                                         srv_trans_id);
2365                                 break;
2366                         }
2367                         wpas_sd_all_bonjour(wpa_s, resp, srv_trans_id);
2368                         wpas_sd_all_upnp(wpa_s, resp, srv_trans_id);
2369                         break;
2370                 case P2P_SERV_BONJOUR:
2371                         wpas_sd_req_bonjour(wpa_s, resp, srv_trans_id,
2372                                             pos, tlv_end - pos);
2373                         break;
2374                 case P2P_SERV_UPNP:
2375                         wpas_sd_req_upnp(wpa_s, resp, srv_trans_id,
2376                                          pos, tlv_end - pos);
2377                         break;
2378 #ifdef CONFIG_WIFI_DISPLAY
2379                 case P2P_SERV_WIFI_DISPLAY:
2380                         wpas_sd_req_wfd(wpa_s, resp, srv_trans_id,
2381                                         pos, tlv_end - pos);
2382                         break;
2383 #endif /* CONFIG_WIFI_DISPLAY */
2384                 default:
2385                         wpa_printf(MSG_DEBUG, "P2P: Unavailable service "
2386                                    "protocol %u", srv_proto);
2387                         wpas_sd_add_proto_not_avail(resp, srv_proto,
2388                                                     srv_trans_id);
2389                         break;
2390                 }
2391
2392                 pos = tlv_end;
2393         }
2394
2395 done:
2396         wpas_notify_p2p_sd_request(wpa_s, freq, sa, dialog_token,
2397                                    update_indic, tlvs, tlvs_len);
2398
2399         wpas_p2p_sd_response(wpa_s, freq, sa, dialog_token, resp);
2400
2401         wpabuf_free(resp);
2402 }
2403
2404
2405 static void wpas_sd_response(void *ctx, const u8 *sa, u16 update_indic,
2406                              const u8 *tlvs, size_t tlvs_len)
2407 {
2408         struct wpa_supplicant *wpa_s = ctx;
2409         const u8 *pos = tlvs;
2410         const u8 *end = tlvs + tlvs_len;
2411         const u8 *tlv_end;
2412         u16 slen;
2413         size_t buf_len;
2414         char *buf;
2415
2416         wpa_hexdump(MSG_MSGDUMP, "P2P: Service Discovery Response TLVs",
2417                     tlvs, tlvs_len);
2418         if (tlvs_len > 1500) {
2419                 /* TODO: better way for handling this */
2420                 wpa_msg_ctrl(wpa_s, MSG_INFO,
2421                              P2P_EVENT_SERV_DISC_RESP MACSTR
2422                              " %u <long response: %u bytes>",
2423                              MAC2STR(sa), update_indic,
2424                              (unsigned int) tlvs_len);
2425         } else {
2426                 buf_len = 2 * tlvs_len + 1;
2427                 buf = os_malloc(buf_len);
2428                 if (buf) {
2429                         wpa_snprintf_hex(buf, buf_len, tlvs, tlvs_len);
2430                         wpa_msg_ctrl(wpa_s, MSG_INFO,
2431                                      P2P_EVENT_SERV_DISC_RESP MACSTR " %u %s",
2432                                      MAC2STR(sa), update_indic, buf);
2433                         os_free(buf);
2434                 }
2435         }
2436
2437         while (pos < end) {
2438                 u8 srv_proto, srv_trans_id, status;
2439
2440                 wpa_printf(MSG_DEBUG, "P2P: Service Response TLV");
2441                 slen = WPA_GET_LE16(pos);
2442                 pos += 2;
2443                 if (pos + slen > end || slen < 3) {
2444                         wpa_printf(MSG_DEBUG, "P2P: Unexpected Response Data "
2445                                    "length");
2446                         return;
2447                 }
2448                 tlv_end = pos + slen;
2449
2450                 srv_proto = *pos++;
2451                 wpa_printf(MSG_DEBUG, "P2P: Service Protocol Type %u",
2452                            srv_proto);
2453                 srv_trans_id = *pos++;
2454                 wpa_printf(MSG_DEBUG, "P2P: Service Transaction ID %u",
2455                            srv_trans_id);
2456                 status = *pos++;
2457                 wpa_printf(MSG_DEBUG, "P2P: Status Code ID %u",
2458                            status);
2459
2460                 wpa_hexdump(MSG_MSGDUMP, "P2P: Response Data",
2461                             pos, tlv_end - pos);
2462
2463                 pos = tlv_end;
2464         }
2465
2466         wpas_notify_p2p_sd_response(wpa_s, sa, update_indic, tlvs, tlvs_len);
2467 }
2468
2469
2470 u64 wpas_p2p_sd_request(struct wpa_supplicant *wpa_s, const u8 *dst,
2471                         const struct wpabuf *tlvs)
2472 {
2473         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
2474                 return 0;
2475         return (uintptr_t) p2p_sd_request(wpa_s->global->p2p, dst, tlvs);
2476 }
2477
2478
2479 u64 wpas_p2p_sd_request_upnp(struct wpa_supplicant *wpa_s, const u8 *dst,
2480                              u8 version, const char *query)
2481 {
2482         struct wpabuf *tlvs;
2483         u64 ret;
2484
2485         tlvs = wpabuf_alloc(2 + 1 + 1 + 1 + os_strlen(query));
2486         if (tlvs == NULL)
2487                 return 0;
2488         wpabuf_put_le16(tlvs, 1 + 1 + 1 + os_strlen(query));
2489         wpabuf_put_u8(tlvs, P2P_SERV_UPNP); /* Service Protocol Type */
2490         wpabuf_put_u8(tlvs, 1); /* Service Transaction ID */
2491         wpabuf_put_u8(tlvs, version);
2492         wpabuf_put_str(tlvs, query);
2493         ret = wpas_p2p_sd_request(wpa_s, dst, tlvs);
2494         wpabuf_free(tlvs);
2495         return ret;
2496 }
2497
2498
2499 #ifdef CONFIG_WIFI_DISPLAY
2500
2501 static u64 wpas_p2p_sd_request_wfd(struct wpa_supplicant *wpa_s, const u8 *dst,
2502                                    const struct wpabuf *tlvs)
2503 {
2504         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
2505                 return 0;
2506         return (uintptr_t) p2p_sd_request_wfd(wpa_s->global->p2p, dst, tlvs);
2507 }
2508
2509
2510 #define MAX_WFD_SD_SUBELEMS 20
2511
2512 static void wfd_add_sd_req_role(struct wpabuf *tlvs, u8 id, u8 role,
2513                                 const char *subelems)
2514 {
2515         u8 *len;
2516         const char *pos;
2517         int val;
2518         int count = 0;
2519
2520         len = wpabuf_put(tlvs, 2);
2521         wpabuf_put_u8(tlvs, P2P_SERV_WIFI_DISPLAY); /* Service Protocol Type */
2522         wpabuf_put_u8(tlvs, id); /* Service Transaction ID */
2523
2524         wpabuf_put_u8(tlvs, role);
2525
2526         pos = subelems;
2527         while (*pos) {
2528                 val = atoi(pos);
2529                 if (val >= 0 && val < 256) {
2530                         wpabuf_put_u8(tlvs, val);
2531                         count++;
2532                         if (count == MAX_WFD_SD_SUBELEMS)
2533                                 break;
2534                 }
2535                 pos = os_strchr(pos + 1, ',');
2536                 if (pos == NULL)
2537                         break;
2538                 pos++;
2539         }
2540
2541         WPA_PUT_LE16(len, (u8 *) wpabuf_put(tlvs, 0) - len - 2);
2542 }
2543
2544
2545 u64 wpas_p2p_sd_request_wifi_display(struct wpa_supplicant *wpa_s,
2546                                      const u8 *dst, const char *role)
2547 {
2548         struct wpabuf *tlvs;
2549         u64 ret;
2550         const char *subelems;
2551         u8 id = 1;
2552
2553         subelems = os_strchr(role, ' ');
2554         if (subelems == NULL)
2555                 return 0;
2556         subelems++;
2557
2558         tlvs = wpabuf_alloc(4 * (2 + 1 + 1 + 1 + MAX_WFD_SD_SUBELEMS));
2559         if (tlvs == NULL)
2560                 return 0;
2561
2562         if (os_strstr(role, "[source]"))
2563                 wfd_add_sd_req_role(tlvs, id++, 0x00, subelems);
2564         if (os_strstr(role, "[pri-sink]"))
2565                 wfd_add_sd_req_role(tlvs, id++, 0x01, subelems);
2566         if (os_strstr(role, "[sec-sink]"))
2567                 wfd_add_sd_req_role(tlvs, id++, 0x02, subelems);
2568         if (os_strstr(role, "[source+sink]"))
2569                 wfd_add_sd_req_role(tlvs, id++, 0x03, subelems);
2570
2571         ret = wpas_p2p_sd_request_wfd(wpa_s, dst, tlvs);
2572         wpabuf_free(tlvs);
2573         return ret;
2574 }
2575
2576 #endif /* CONFIG_WIFI_DISPLAY */
2577
2578
2579 int wpas_p2p_sd_cancel_request(struct wpa_supplicant *wpa_s, u64 req)
2580 {
2581         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
2582                 return -1;
2583         return p2p_sd_cancel_request(wpa_s->global->p2p,
2584                                      (void *) (uintptr_t) req);
2585 }
2586
2587
2588 void wpas_p2p_sd_response(struct wpa_supplicant *wpa_s, int freq,
2589                           const u8 *dst, u8 dialog_token,
2590                           const struct wpabuf *resp_tlvs)
2591 {
2592         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
2593                 return;
2594         p2p_sd_response(wpa_s->global->p2p, freq, dst, dialog_token,
2595                         resp_tlvs);
2596 }
2597
2598
2599 void wpas_p2p_sd_service_update(struct wpa_supplicant *wpa_s)
2600 {
2601         if (wpa_s->global->p2p)
2602                 p2p_sd_service_update(wpa_s->global->p2p);
2603 }
2604
2605
2606 static void wpas_p2p_srv_bonjour_free(struct p2p_srv_bonjour *bsrv)
2607 {
2608         dl_list_del(&bsrv->list);
2609         wpabuf_free(bsrv->query);
2610         wpabuf_free(bsrv->resp);
2611         os_free(bsrv);
2612 }
2613
2614
2615 static void wpas_p2p_srv_upnp_free(struct p2p_srv_upnp *usrv)
2616 {
2617         dl_list_del(&usrv->list);
2618         os_free(usrv->service);
2619         os_free(usrv);
2620 }
2621
2622
2623 void wpas_p2p_service_flush(struct wpa_supplicant *wpa_s)
2624 {
2625         struct p2p_srv_bonjour *bsrv, *bn;
2626         struct p2p_srv_upnp *usrv, *un;
2627
2628         dl_list_for_each_safe(bsrv, bn, &wpa_s->global->p2p_srv_bonjour,
2629                               struct p2p_srv_bonjour, list)
2630                 wpas_p2p_srv_bonjour_free(bsrv);
2631
2632         dl_list_for_each_safe(usrv, un, &wpa_s->global->p2p_srv_upnp,
2633                               struct p2p_srv_upnp, list)
2634                 wpas_p2p_srv_upnp_free(usrv);
2635
2636         wpas_p2p_sd_service_update(wpa_s);
2637 }
2638
2639
2640 int wpas_p2p_service_add_bonjour(struct wpa_supplicant *wpa_s,
2641                                  struct wpabuf *query, struct wpabuf *resp)
2642 {
2643         struct p2p_srv_bonjour *bsrv;
2644
2645         bsrv = os_zalloc(sizeof(*bsrv));
2646         if (bsrv == NULL)
2647                 return -1;
2648         bsrv->query = query;
2649         bsrv->resp = resp;
2650         dl_list_add(&wpa_s->global->p2p_srv_bonjour, &bsrv->list);
2651
2652         wpas_p2p_sd_service_update(wpa_s);
2653         return 0;
2654 }
2655
2656
2657 int wpas_p2p_service_del_bonjour(struct wpa_supplicant *wpa_s,
2658                                  const struct wpabuf *query)
2659 {
2660         struct p2p_srv_bonjour *bsrv;
2661
2662         bsrv = wpas_p2p_service_get_bonjour(wpa_s, query);
2663         if (bsrv == NULL)
2664                 return -1;
2665         wpas_p2p_srv_bonjour_free(bsrv);
2666         wpas_p2p_sd_service_update(wpa_s);
2667         return 0;
2668 }
2669
2670
2671 int wpas_p2p_service_add_upnp(struct wpa_supplicant *wpa_s, u8 version,
2672                               const char *service)
2673 {
2674         struct p2p_srv_upnp *usrv;
2675
2676         if (wpas_p2p_service_get_upnp(wpa_s, version, service))
2677                 return 0; /* Already listed */
2678         usrv = os_zalloc(sizeof(*usrv));
2679         if (usrv == NULL)
2680                 return -1;
2681         usrv->version = version;
2682         usrv->service = os_strdup(service);
2683         if (usrv->service == NULL) {
2684                 os_free(usrv);
2685                 return -1;
2686         }
2687         dl_list_add(&wpa_s->global->p2p_srv_upnp, &usrv->list);
2688
2689         wpas_p2p_sd_service_update(wpa_s);
2690         return 0;
2691 }
2692
2693
2694 int wpas_p2p_service_del_upnp(struct wpa_supplicant *wpa_s, u8 version,
2695                               const char *service)
2696 {
2697         struct p2p_srv_upnp *usrv;
2698
2699         usrv = wpas_p2p_service_get_upnp(wpa_s, version, service);
2700         if (usrv == NULL)
2701                 return -1;
2702         wpas_p2p_srv_upnp_free(usrv);
2703         wpas_p2p_sd_service_update(wpa_s);
2704         return 0;
2705 }
2706
2707
2708 static void wpas_prov_disc_local_display(struct wpa_supplicant *wpa_s,
2709                                          const u8 *peer, const char *params,
2710                                          unsigned int generated_pin)
2711 {
2712         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_SHOW_PIN MACSTR
2713                        " %08d%s", MAC2STR(peer), generated_pin, params);
2714 }
2715
2716
2717 static void wpas_prov_disc_local_keypad(struct wpa_supplicant *wpa_s,
2718                                         const u8 *peer, const char *params)
2719 {
2720         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_ENTER_PIN MACSTR
2721                        "%s", MAC2STR(peer), params);
2722 }
2723
2724
2725 static void wpas_prov_disc_req(void *ctx, const u8 *peer, u16 config_methods,
2726                                const u8 *dev_addr, const u8 *pri_dev_type,
2727                                const char *dev_name, u16 supp_config_methods,
2728                                u8 dev_capab, u8 group_capab, const u8 *group_id,
2729                                size_t group_id_len)
2730 {
2731         struct wpa_supplicant *wpa_s = ctx;
2732         char devtype[WPS_DEV_TYPE_BUFSIZE];
2733         char params[300];
2734         u8 empty_dev_type[8];
2735         unsigned int generated_pin = 0;
2736         struct wpa_supplicant *group = NULL;
2737
2738         if (group_id) {
2739                 for (group = wpa_s->global->ifaces; group; group = group->next)
2740                 {
2741                         struct wpa_ssid *s = group->current_ssid;
2742                         if (s != NULL &&
2743                             s->mode == WPAS_MODE_P2P_GO &&
2744                             group_id_len - ETH_ALEN == s->ssid_len &&
2745                             os_memcmp(group_id + ETH_ALEN, s->ssid,
2746                                       s->ssid_len) == 0)
2747                                 break;
2748                 }
2749         }
2750
2751         if (pri_dev_type == NULL) {
2752                 os_memset(empty_dev_type, 0, sizeof(empty_dev_type));
2753                 pri_dev_type = empty_dev_type;
2754         }
2755         os_snprintf(params, sizeof(params), " p2p_dev_addr=" MACSTR
2756                     " pri_dev_type=%s name='%s' config_methods=0x%x "
2757                     "dev_capab=0x%x group_capab=0x%x%s%s",
2758                     MAC2STR(dev_addr),
2759                     wps_dev_type_bin2str(pri_dev_type, devtype,
2760                                          sizeof(devtype)),
2761                     dev_name, supp_config_methods, dev_capab, group_capab,
2762                     group ? " group=" : "",
2763                     group ? group->ifname : "");
2764         params[sizeof(params) - 1] = '\0';
2765
2766         if (config_methods & WPS_CONFIG_DISPLAY) {
2767                 generated_pin = wps_generate_pin();
2768                 wpas_prov_disc_local_display(wpa_s, peer, params,
2769                                              generated_pin);
2770         } else if (config_methods & WPS_CONFIG_KEYPAD)
2771                 wpas_prov_disc_local_keypad(wpa_s, peer, params);
2772         else if (config_methods & WPS_CONFIG_PUSHBUTTON)
2773                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_PBC_REQ
2774                                MACSTR "%s", MAC2STR(peer), params);
2775
2776         wpas_notify_p2p_provision_discovery(wpa_s, peer, 1 /* request */,
2777                                             P2P_PROV_DISC_SUCCESS,
2778                                             config_methods, generated_pin);
2779 }
2780
2781
2782 static void wpas_prov_disc_resp(void *ctx, const u8 *peer, u16 config_methods)
2783 {
2784         struct wpa_supplicant *wpa_s = ctx;
2785         unsigned int generated_pin = 0;
2786         char params[20];
2787
2788         if (wpa_s->pending_pd_before_join &&
2789             (os_memcmp(peer, wpa_s->pending_join_dev_addr, ETH_ALEN) == 0 ||
2790              os_memcmp(peer, wpa_s->pending_join_iface_addr, ETH_ALEN) == 0)) {
2791                 wpa_s->pending_pd_before_join = 0;
2792                 wpa_printf(MSG_DEBUG, "P2P: Starting pending "
2793                            "join-existing-group operation");
2794                 wpas_p2p_join_start(wpa_s);
2795                 return;
2796         }
2797
2798         if (wpa_s->pending_pd_use == AUTO_PD_JOIN ||
2799             wpa_s->pending_pd_use == AUTO_PD_GO_NEG)
2800                 os_snprintf(params, sizeof(params), " peer_go=%d",
2801                             wpa_s->pending_pd_use == AUTO_PD_JOIN);
2802         else
2803                 params[0] = '\0';
2804
2805         if (config_methods & WPS_CONFIG_DISPLAY)
2806                 wpas_prov_disc_local_keypad(wpa_s, peer, params);
2807         else if (config_methods & WPS_CONFIG_KEYPAD) {
2808                 generated_pin = wps_generate_pin();
2809                 wpas_prov_disc_local_display(wpa_s, peer, params,
2810                                              generated_pin);
2811         } else if (config_methods & WPS_CONFIG_PUSHBUTTON)
2812                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_PBC_RESP
2813                                MACSTR "%s", MAC2STR(peer), params);
2814
2815         wpas_notify_p2p_provision_discovery(wpa_s, peer, 0 /* response */,
2816                                             P2P_PROV_DISC_SUCCESS,
2817                                             config_methods, generated_pin);
2818 }
2819
2820
2821 static void wpas_prov_disc_fail(void *ctx, const u8 *peer,
2822                                 enum p2p_prov_disc_status status)
2823 {
2824         struct wpa_supplicant *wpa_s = ctx;
2825
2826         if (wpa_s->p2p_fallback_to_go_neg) {
2827                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: PD for p2p_connect-auto "
2828                         "failed - fall back to GO Negotiation");
2829                 wpas_p2p_fallback_to_go_neg(wpa_s, 0);
2830                 return;
2831         }
2832
2833         if (status == P2P_PROV_DISC_TIMEOUT_JOIN) {
2834                 wpa_s->pending_pd_before_join = 0;
2835                 wpa_printf(MSG_DEBUG, "P2P: Starting pending "
2836                            "join-existing-group operation (no ACK for PD "
2837                            "Req attempts)");
2838                 wpas_p2p_join_start(wpa_s);
2839                 return;
2840         }
2841
2842         wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_PROV_DISC_FAILURE
2843                        " p2p_dev_addr=" MACSTR " status=%d",
2844                        MAC2STR(peer), status);
2845
2846         wpas_notify_p2p_provision_discovery(wpa_s, peer, 0 /* response */,
2847                                             status, 0, 0);
2848 }
2849
2850
2851 static int freq_included(const struct p2p_channels *channels, unsigned int freq)
2852 {
2853         if (channels == NULL)
2854                 return 1; /* Assume no restrictions */
2855         return p2p_channels_includes_freq(channels, freq);
2856
2857 }
2858
2859
2860 static u8 wpas_invitation_process(void *ctx, const u8 *sa, const u8 *bssid,
2861                                   const u8 *go_dev_addr, const u8 *ssid,
2862                                   size_t ssid_len, int *go, u8 *group_bssid,
2863                                   int *force_freq, int persistent_group,
2864                                   const struct p2p_channels *channels)
2865 {
2866         struct wpa_supplicant *wpa_s = ctx;
2867         struct wpa_ssid *s;
2868         int res;
2869         struct wpa_supplicant *grp;
2870
2871         if (!persistent_group) {
2872                 wpa_printf(MSG_DEBUG, "P2P: Invitation from " MACSTR
2873                            " to join an active group (SSID: %s)",
2874                            MAC2STR(sa), wpa_ssid_txt(ssid, ssid_len));
2875                 if (!is_zero_ether_addr(wpa_s->p2p_auth_invite) &&
2876                     (os_memcmp(go_dev_addr, wpa_s->p2p_auth_invite, ETH_ALEN)
2877                      == 0 ||
2878                      os_memcmp(sa, wpa_s->p2p_auth_invite, ETH_ALEN) == 0)) {
2879                         wpa_printf(MSG_DEBUG, "P2P: Accept previously "
2880                                    "authorized invitation");
2881                         goto accept_inv;
2882                 }
2883                 /*
2884                  * Do not accept the invitation automatically; notify user and
2885                  * request approval.
2886                  */
2887                 return P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
2888         }
2889
2890         grp = wpas_get_p2p_group(wpa_s, ssid, ssid_len, go);
2891         if (grp) {
2892                 wpa_printf(MSG_DEBUG, "P2P: Accept invitation to already "
2893                            "running persistent group");
2894                 if (*go)
2895                         os_memcpy(group_bssid, grp->own_addr, ETH_ALEN);
2896                 goto accept_inv;
2897         }
2898
2899         if (!is_zero_ether_addr(wpa_s->p2p_auth_invite) &&
2900             os_memcmp(sa, wpa_s->p2p_auth_invite, ETH_ALEN) == 0) {
2901                 wpa_printf(MSG_DEBUG, "P2P: Accept previously initiated "
2902                            "invitation to re-invoke a persistent group");
2903         } else if (!wpa_s->conf->persistent_reconnect)
2904                 return P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE;
2905
2906         for (s = wpa_s->conf->ssid; s; s = s->next) {
2907                 if (s->disabled == 2 &&
2908                     os_memcmp(s->bssid, go_dev_addr, ETH_ALEN) == 0 &&
2909                     s->ssid_len == ssid_len &&
2910                     os_memcmp(ssid, s->ssid, ssid_len) == 0)
2911                         break;
2912         }
2913
2914         if (!s) {
2915                 wpa_printf(MSG_DEBUG, "P2P: Invitation from " MACSTR
2916                            " requested reinvocation of an unknown group",
2917                            MAC2STR(sa));
2918                 return P2P_SC_FAIL_UNKNOWN_GROUP;
2919         }
2920
2921         if (s->mode == WPAS_MODE_P2P_GO && !wpas_p2p_create_iface(wpa_s)) {
2922                 *go = 1;
2923                 if (wpa_s->wpa_state >= WPA_AUTHENTICATING) {
2924                         wpa_printf(MSG_DEBUG, "P2P: The only available "
2925                                    "interface is already in use - reject "
2926                                    "invitation");
2927                         return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
2928                 }
2929                 os_memcpy(group_bssid, wpa_s->own_addr, ETH_ALEN);
2930         } else if (s->mode == WPAS_MODE_P2P_GO) {
2931                 *go = 1;
2932                 if (wpas_p2p_add_group_interface(wpa_s, WPA_IF_P2P_GO) < 0)
2933                 {
2934                         wpa_printf(MSG_ERROR, "P2P: Failed to allocate a new "
2935                                    "interface address for the group");
2936                         return P2P_SC_FAIL_UNABLE_TO_ACCOMMODATE;
2937                 }
2938                 os_memcpy(group_bssid, wpa_s->pending_interface_addr,
2939                           ETH_ALEN);
2940         }
2941
2942 accept_inv:
2943         wpas_p2p_set_own_freq_preference(wpa_s, 0);
2944
2945         /* Get one of the frequencies currently in use */
2946         if (wpas_p2p_valid_oper_freqs(wpa_s, &res, 1) > 0) {
2947                 wpa_printf(MSG_DEBUG, "P2P: Trying to prefer a channel already used by one of the interfaces");
2948                 wpas_p2p_set_own_freq_preference(wpa_s, res);
2949
2950                 if (wpa_s->num_multichan_concurrent < 2 ||
2951                     wpas_p2p_num_unused_channels(wpa_s) < 1) {
2952                         wpa_printf(MSG_DEBUG, "P2P: No extra channels available - trying to force channel to match a channel already used by one of the interfaces");
2953                         *force_freq = res;
2954                 }
2955         }
2956
2957         if (*force_freq > 0 && wpa_s->num_multichan_concurrent > 1 &&
2958             wpas_p2p_num_unused_channels(wpa_s) > 0) {
2959                 if (*go == 0) {
2960                         /* We are the client */
2961                         wpa_printf(MSG_DEBUG, "P2P: Peer was found to be "
2962                                    "running a GO but we are capable of MCC, "
2963                                    "figure out the best channel to use");
2964                         *force_freq = 0;
2965                 } else if (!freq_included(channels, *force_freq)) {
2966                         /* We are the GO, and *force_freq is not in the
2967                          * intersection */
2968                         wpa_printf(MSG_DEBUG, "P2P: Forced GO freq %d MHz not "
2969                                    "in intersection but we are capable of MCC, "
2970                                    "figure out the best channel to use",
2971                                    *force_freq);
2972                         *force_freq = 0;
2973                 }
2974         }
2975
2976         return P2P_SC_SUCCESS;
2977 }
2978
2979
2980 static void wpas_invitation_received(void *ctx, const u8 *sa, const u8 *bssid,
2981                                      const u8 *ssid, size_t ssid_len,
2982                                      const u8 *go_dev_addr, u8 status,
2983                                      int op_freq)
2984 {
2985         struct wpa_supplicant *wpa_s = ctx;
2986         struct wpa_ssid *s;
2987
2988         for (s = wpa_s->conf->ssid; s; s = s->next) {
2989                 if (s->disabled == 2 &&
2990                     s->ssid_len == ssid_len &&
2991                     os_memcmp(ssid, s->ssid, ssid_len) == 0)
2992                         break;
2993         }
2994
2995         if (status == P2P_SC_SUCCESS) {
2996                 wpa_printf(MSG_DEBUG, "P2P: Invitation from peer " MACSTR
2997                            " was accepted; op_freq=%d MHz, SSID=%s",
2998                            MAC2STR(sa), op_freq, wpa_ssid_txt(ssid, ssid_len));
2999                 if (s) {
3000                         int go = s->mode == WPAS_MODE_P2P_GO;
3001                         wpas_p2p_group_add_persistent(
3002                                 wpa_s, s, go, go ? op_freq : 0, 0, 0, NULL,
3003                                 go ? P2P_MAX_INITIAL_CONN_WAIT_GO_REINVOKE : 0);
3004                 } else if (bssid) {
3005                         wpa_s->user_initiated_pd = 0;
3006                         wpas_p2p_join(wpa_s, bssid, go_dev_addr,
3007                                       wpa_s->p2p_wps_method, 0,
3008                                       ssid, ssid_len);
3009                 }
3010                 return;
3011         }
3012
3013         if (status != P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE) {
3014                 wpa_printf(MSG_DEBUG, "P2P: Invitation from peer " MACSTR
3015                            " was rejected (status %u)", MAC2STR(sa), status);
3016                 return;
3017         }
3018
3019         if (!s) {
3020                 if (bssid) {
3021                         wpa_msg_global(wpa_s, MSG_INFO,
3022                                        P2P_EVENT_INVITATION_RECEIVED
3023                                        "sa=" MACSTR " go_dev_addr=" MACSTR
3024                                        " bssid=" MACSTR " unknown-network",
3025                                        MAC2STR(sa), MAC2STR(go_dev_addr),
3026                                        MAC2STR(bssid));
3027                 } else {
3028                         wpa_msg_global(wpa_s, MSG_INFO,
3029                                        P2P_EVENT_INVITATION_RECEIVED
3030                                        "sa=" MACSTR " go_dev_addr=" MACSTR
3031                                        " unknown-network",
3032                                        MAC2STR(sa), MAC2STR(go_dev_addr));
3033                 }
3034                 return;
3035         }
3036
3037         if (s->mode == WPAS_MODE_P2P_GO && op_freq) {
3038                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RECEIVED
3039                                "sa=" MACSTR " persistent=%d freq=%d",
3040                                MAC2STR(sa), s->id, op_freq);
3041         } else {
3042                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RECEIVED
3043                                "sa=" MACSTR " persistent=%d",
3044                                MAC2STR(sa), s->id);
3045         }
3046 }
3047
3048
3049 static void wpas_remove_persistent_peer(struct wpa_supplicant *wpa_s,
3050                                         struct wpa_ssid *ssid,
3051                                         const u8 *peer, int inv)
3052 {
3053         size_t i;
3054
3055         if (ssid == NULL)
3056                 return;
3057
3058         for (i = 0; ssid->p2p_client_list && i < ssid->num_p2p_clients; i++) {
3059                 if (os_memcmp(ssid->p2p_client_list + i * ETH_ALEN, peer,
3060                               ETH_ALEN) == 0)
3061                         break;
3062         }
3063         if (i >= ssid->num_p2p_clients) {
3064                 if (ssid->mode != WPAS_MODE_P2P_GO &&
3065                     os_memcmp(ssid->bssid, peer, ETH_ALEN) == 0) {
3066                         wpa_printf(MSG_DEBUG, "P2P: Remove persistent group %d "
3067                                    "due to invitation result", ssid->id);
3068                         wpas_notify_network_removed(wpa_s, ssid);
3069                         wpa_config_remove_network(wpa_s->conf, ssid->id);
3070                         return;
3071                 }
3072                 return; /* Peer not found in client list */
3073         }
3074
3075         wpa_printf(MSG_DEBUG, "P2P: Remove peer " MACSTR " from persistent "
3076                    "group %d client list%s",
3077                    MAC2STR(peer), ssid->id,
3078                    inv ? " due to invitation result" : "");
3079         os_memmove(ssid->p2p_client_list + i * ETH_ALEN,
3080                    ssid->p2p_client_list + (i + 1) * ETH_ALEN,
3081                    (ssid->num_p2p_clients - i - 1) * ETH_ALEN);
3082         ssid->num_p2p_clients--;
3083 #ifndef CONFIG_NO_CONFIG_WRITE
3084         if (wpa_s->parent->conf->update_config &&
3085             wpa_config_write(wpa_s->parent->confname, wpa_s->parent->conf))
3086                 wpa_printf(MSG_DEBUG, "P2P: Failed to update configuration");
3087 #endif /* CONFIG_NO_CONFIG_WRITE */
3088 }
3089
3090
3091 static void wpas_remove_persistent_client(struct wpa_supplicant *wpa_s,
3092                                           const u8 *peer)
3093 {
3094         struct wpa_ssid *ssid;
3095
3096         wpa_s = wpa_s->global->p2p_invite_group;
3097         if (wpa_s == NULL)
3098                 return; /* No known invitation group */
3099         ssid = wpa_s->current_ssid;
3100         if (ssid == NULL || ssid->mode != WPAS_MODE_P2P_GO ||
3101             !ssid->p2p_persistent_group)
3102                 return; /* Not operating as a GO in persistent group */
3103         ssid = wpas_p2p_get_persistent(wpa_s->parent, peer,
3104                                        ssid->ssid, ssid->ssid_len);
3105         wpas_remove_persistent_peer(wpa_s, ssid, peer, 1);
3106 }
3107
3108
3109 static void wpas_invitation_result(void *ctx, int status, const u8 *bssid,
3110                                    const struct p2p_channels *channels,
3111                                    const u8 *peer, int neg_freq)
3112 {
3113         struct wpa_supplicant *wpa_s = ctx;
3114         struct wpa_ssid *ssid;
3115         int freq;
3116
3117         if (bssid) {
3118                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RESULT
3119                                "status=%d " MACSTR,
3120                                status, MAC2STR(bssid));
3121         } else {
3122                 wpa_msg_global(wpa_s, MSG_INFO, P2P_EVENT_INVITATION_RESULT
3123                                "status=%d ", status);
3124         }
3125         wpas_notify_p2p_invitation_result(wpa_s, status, bssid);
3126
3127         wpa_printf(MSG_DEBUG, "P2P: Invitation result - status=%d peer=" MACSTR,
3128                    status, MAC2STR(peer));
3129         if (wpa_s->pending_invite_ssid_id == -1) {
3130                 if (status == P2P_SC_FAIL_UNKNOWN_GROUP)
3131                         wpas_remove_persistent_client(wpa_s, peer);
3132                 return; /* Invitation to active group */
3133         }
3134
3135         if (status == P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE) {
3136                 wpa_printf(MSG_DEBUG, "P2P: Waiting for peer to start another "
3137                            "invitation exchange to indicate readiness for "
3138                            "re-invocation");
3139         }
3140
3141         if (status != P2P_SC_SUCCESS) {
3142                 if (status == P2P_SC_FAIL_UNKNOWN_GROUP) {
3143                         ssid = wpa_config_get_network(
3144                                 wpa_s->conf, wpa_s->pending_invite_ssid_id);
3145                         wpas_remove_persistent_peer(wpa_s, ssid, peer, 1);
3146                 }
3147                 wpas_p2p_remove_pending_group_interface(wpa_s);
3148                 return;
3149         }
3150
3151         ssid = wpa_config_get_network(wpa_s->conf,
3152                                       wpa_s->pending_invite_ssid_id);
3153         if (ssid == NULL) {
3154                 wpa_printf(MSG_ERROR, "P2P: Could not find persistent group "
3155                            "data matching with invitation");
3156                 return;
3157         }
3158
3159         /*
3160          * The peer could have missed our ctrl::ack frame for Invitation
3161          * Response and continue retransmitting the frame. To reduce the
3162          * likelihood of the peer not getting successful TX status for the
3163          * Invitation Response frame, wait a short time here before starting
3164          * the persistent group so that we will remain on the current channel to
3165          * acknowledge any possible retransmission from the peer.
3166          */
3167         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: 50 ms wait on current channel before "
3168                 "starting persistent group");
3169         os_sleep(0, 50000);
3170
3171         freq = wpa_s->p2p_persistent_go_freq;
3172         if (neg_freq > 0 && ssid->mode == WPAS_MODE_P2P_GO &&
3173             freq_included(channels, neg_freq)) {
3174                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Use frequence %d MHz from invitation for GO mode",
3175                         neg_freq);
3176                 freq = neg_freq;
3177         }
3178
3179         wpas_p2p_group_add_persistent(wpa_s, ssid,
3180                                       ssid->mode == WPAS_MODE_P2P_GO,
3181                                       freq,
3182                                       wpa_s->p2p_go_ht40, wpa_s->p2p_go_vht,
3183                                       channels,
3184                                       ssid->mode == WPAS_MODE_P2P_GO ?
3185                                       P2P_MAX_INITIAL_CONN_WAIT_GO_REINVOKE :
3186                                       0);
3187 }
3188
3189
3190 static int wpas_p2p_disallowed_freq(struct wpa_global *global,
3191                                     unsigned int freq)
3192 {
3193         if (freq_range_list_includes(&global->p2p_go_avoid_freq, freq))
3194                 return 1;
3195         return freq_range_list_includes(&global->p2p_disallow_freq, freq);
3196 }
3197
3198
3199 static void wpas_p2p_add_chan(struct p2p_reg_class *reg, u8 chan)
3200 {
3201         reg->channel[reg->channels] = chan;
3202         reg->channels++;
3203 }
3204
3205
3206 static int wpas_p2p_default_channels(struct wpa_supplicant *wpa_s,
3207                                      struct p2p_channels *chan,
3208                                      struct p2p_channels *cli_chan)
3209 {
3210         int i, cla = 0;
3211
3212         os_memset(cli_chan, 0, sizeof(*cli_chan));
3213
3214         wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for 2.4 GHz "
3215                    "band");
3216
3217         /* Operating class 81 - 2.4 GHz band channels 1..13 */
3218         chan->reg_class[cla].reg_class = 81;
3219         chan->reg_class[cla].channels = 0;
3220         for (i = 0; i < 11; i++) {
3221                 if (!wpas_p2p_disallowed_freq(wpa_s->global, 2412 + i * 5))
3222                         wpas_p2p_add_chan(&chan->reg_class[cla], i + 1);
3223         }
3224         if (chan->reg_class[cla].channels)
3225                 cla++;
3226
3227         wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for lower 5 GHz "
3228                    "band");
3229
3230         /* Operating class 115 - 5 GHz, channels 36-48 */
3231         chan->reg_class[cla].reg_class = 115;
3232         chan->reg_class[cla].channels = 0;
3233         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 36 * 5))
3234                 wpas_p2p_add_chan(&chan->reg_class[cla], 36);
3235         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 40 * 5))
3236                 wpas_p2p_add_chan(&chan->reg_class[cla], 40);
3237         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 44 * 5))
3238                 wpas_p2p_add_chan(&chan->reg_class[cla], 44);
3239         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 48 * 5))
3240                 wpas_p2p_add_chan(&chan->reg_class[cla], 48);
3241         if (chan->reg_class[cla].channels)
3242                 cla++;
3243
3244         wpa_printf(MSG_DEBUG, "P2P: Enable operating classes for higher 5 GHz "
3245                    "band");
3246
3247         /* Operating class 124 - 5 GHz, channels 149,153,157,161 */
3248         chan->reg_class[cla].reg_class = 124;
3249         chan->reg_class[cla].channels = 0;
3250         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 149 * 5))
3251                 wpas_p2p_add_chan(&chan->reg_class[cla], 149);
3252         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 153 * 5))
3253                 wpas_p2p_add_chan(&chan->reg_class[cla], 153);
3254         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 156 * 5))
3255                 wpas_p2p_add_chan(&chan->reg_class[cla], 157);
3256         if (!wpas_p2p_disallowed_freq(wpa_s->global, 5000 + 161 * 5))
3257                 wpas_p2p_add_chan(&chan->reg_class[cla], 161);
3258         if (chan->reg_class[cla].channels)
3259                 cla++;
3260
3261         chan->reg_classes = cla;
3262         return 0;
3263 }
3264
3265
3266 static struct hostapd_hw_modes * get_mode(struct hostapd_hw_modes *modes,
3267                                           u16 num_modes,
3268                                           enum hostapd_hw_mode mode)
3269 {
3270         u16 i;
3271
3272         for (i = 0; i < num_modes; i++) {
3273                 if (modes[i].mode == mode)
3274                         return &modes[i];
3275         }
3276
3277         return NULL;
3278 }
3279
3280
3281 enum chan_allowed {
3282         NOT_ALLOWED, PASSIVE_ONLY, ALLOWED
3283 };
3284
3285 static int has_channel(struct wpa_global *global,
3286                        struct hostapd_hw_modes *mode, u8 chan, int *flags)
3287 {
3288         int i;
3289         unsigned int freq;
3290
3291         freq = (mode->mode == HOSTAPD_MODE_IEEE80211A ? 5000 : 2407) +
3292                 chan * 5;
3293         if (wpas_p2p_disallowed_freq(global, freq))
3294                 return NOT_ALLOWED;
3295
3296         for (i = 0; i < mode->num_channels; i++) {
3297                 if (mode->channels[i].chan == chan) {
3298                         if (flags)
3299                                 *flags = mode->channels[i].flag;
3300                         if (mode->channels[i].flag &
3301                             (HOSTAPD_CHAN_DISABLED |
3302                              HOSTAPD_CHAN_RADAR))
3303                                 return NOT_ALLOWED;
3304                         if (mode->channels[i].flag &
3305                             (HOSTAPD_CHAN_PASSIVE_SCAN |
3306                              HOSTAPD_CHAN_NO_IBSS))
3307                                 return PASSIVE_ONLY;
3308                         return ALLOWED;
3309                 }
3310         }
3311
3312         return NOT_ALLOWED;
3313 }
3314
3315
3316 struct p2p_oper_class_map {
3317         enum hostapd_hw_mode mode;
3318         u8 op_class;
3319         u8 min_chan;
3320         u8 max_chan;
3321         u8 inc;
3322         enum { BW20, BW40PLUS, BW40MINUS, BW80 } bw;
3323 };
3324
3325 static struct p2p_oper_class_map op_class[] = {
3326         { HOSTAPD_MODE_IEEE80211G, 81, 1, 13, 1, BW20 },
3327 #if 0 /* Do not enable HT40 on 2 GHz for now */
3328         { HOSTAPD_MODE_IEEE80211G, 83, 1, 9, 1, BW40PLUS },
3329         { HOSTAPD_MODE_IEEE80211G, 84, 5, 13, 1, BW40MINUS },
3330 #endif
3331         { HOSTAPD_MODE_IEEE80211A, 115, 36, 48, 4, BW20 },
3332         { HOSTAPD_MODE_IEEE80211A, 124, 149, 161, 4, BW20 },
3333         { HOSTAPD_MODE_IEEE80211A, 116, 36, 44, 8, BW40PLUS },
3334         { HOSTAPD_MODE_IEEE80211A, 117, 40, 48, 8, BW40MINUS },
3335         { HOSTAPD_MODE_IEEE80211A, 126, 149, 157, 8, BW40PLUS },
3336         { HOSTAPD_MODE_IEEE80211A, 127, 153, 161, 8, BW40MINUS },
3337
3338         /*
3339          * IEEE P802.11ac/D7.0 Table E-4 actually talks about channel center
3340          * frequency index 42, 58, 106, 122, 138, 155 with channel spacing of
3341          * 80 MHz, but currently use the following definition for simplicity
3342          * (these center frequencies are not actual channels, which makes
3343          * has_channel() fail). wpas_p2p_verify_80mhz() should take care of
3344          * removing invalid channels.
3345          */
3346         { HOSTAPD_MODE_IEEE80211A, 128, 36, 161, 4, BW80 },
3347         { -1, 0, 0, 0, 0, BW20 }
3348 };
3349
3350
3351 static int wpas_p2p_get_center_80mhz(struct wpa_supplicant *wpa_s,
3352                                      struct hostapd_hw_modes *mode,
3353                                      u8 channel)
3354 {
3355         u8 center_channels[] = { 42, 58, 106, 122, 138, 155 };
3356         unsigned int i;
3357
3358         if (mode->mode != HOSTAPD_MODE_IEEE80211A)
3359                 return 0;
3360
3361         for (i = 0; i < ARRAY_SIZE(center_channels); i++)
3362                 /*
3363                  * In 80 MHz, the bandwidth "spans" 12 channels (e.g., 36-48),
3364                  * so the center channel is 6 channels away from the start/end.
3365                  */
3366                 if (channel >= center_channels[i] - 6 &&
3367                     channel <= center_channels[i] + 6)
3368                         return center_channels[i];
3369
3370         return 0;
3371 }
3372
3373
3374 static enum chan_allowed wpas_p2p_verify_80mhz(struct wpa_supplicant *wpa_s,
3375                                                struct hostapd_hw_modes *mode,
3376                                                u8 channel, u8 bw)
3377 {
3378         u8 center_chan;
3379         int i, flags;
3380         enum chan_allowed res, ret = ALLOWED;
3381
3382         center_chan = wpas_p2p_get_center_80mhz(wpa_s, mode, channel);
3383         if (!center_chan)
3384                 return NOT_ALLOWED;
3385         if (center_chan >= 58 && center_chan <= 138)
3386                 return NOT_ALLOWED; /* Do not allow DFS channels for P2P */
3387
3388         /* check all the channels are available */
3389         for (i = 0; i < 4; i++) {
3390                 int adj_chan = center_chan - 6 + i * 4;
3391
3392                 res = has_channel(wpa_s->global, mode, adj_chan, &flags);
3393                 if (res == NOT_ALLOWED)
3394                         return NOT_ALLOWED;
3395                 if (res == PASSIVE_ONLY)
3396                         ret = PASSIVE_ONLY;
3397
3398                 if (i == 0 && !(flags & HOSTAPD_CHAN_VHT_10_70))
3399                         return NOT_ALLOWED;
3400                 if (i == 1 && !(flags & HOSTAPD_CHAN_VHT_30_50))
3401                         return NOT_ALLOWED;
3402                 if (i == 2 && !(flags & HOSTAPD_CHAN_VHT_50_30))
3403                         return NOT_ALLOWED;
3404                 if (i == 3 && !(flags & HOSTAPD_CHAN_VHT_70_10))
3405                         return NOT_ALLOWED;
3406         }
3407
3408         return ret;
3409 }
3410
3411
3412 static enum chan_allowed wpas_p2p_verify_channel(struct wpa_supplicant *wpa_s,
3413                                                  struct hostapd_hw_modes *mode,
3414                                                  u8 channel, u8 bw)
3415 {
3416         int flag;
3417         enum chan_allowed res, res2;
3418
3419         res2 = res = has_channel(wpa_s->global, mode, channel, &flag);
3420         if (bw == BW40MINUS) {
3421                 if (!(flag & HOSTAPD_CHAN_HT40MINUS))
3422                         return NOT_ALLOWED;
3423                 res2 = has_channel(wpa_s->global, mode, channel - 4, NULL);
3424         } else if (bw == BW40PLUS) {
3425                 if (!(flag & HOSTAPD_CHAN_HT40PLUS))
3426                         return NOT_ALLOWED;
3427                 res2 = has_channel(wpa_s->global, mode, channel + 4, NULL);
3428         } else if (bw == BW80) {
3429                 res2 = wpas_p2p_verify_80mhz(wpa_s, mode, channel, bw);
3430         }
3431
3432         if (res == NOT_ALLOWED || res2 == NOT_ALLOWED)
3433                 return NOT_ALLOWED;
3434         if (res == PASSIVE_ONLY || res2 == PASSIVE_ONLY)
3435                 return PASSIVE_ONLY;
3436         return res;
3437 }
3438
3439
3440 static int wpas_p2p_setup_channels(struct wpa_supplicant *wpa_s,
3441                                    struct p2p_channels *chan,
3442                                    struct p2p_channels *cli_chan)
3443 {
3444         struct hostapd_hw_modes *mode;
3445         int cla, op, cli_cla;
3446
3447         if (wpa_s->hw.modes == NULL) {
3448                 wpa_printf(MSG_DEBUG, "P2P: Driver did not support fetching "
3449                            "of all supported channels; assume dualband "
3450                            "support");
3451                 return wpas_p2p_default_channels(wpa_s, chan, cli_chan);
3452         }
3453
3454         cla = cli_cla = 0;
3455
3456         for (op = 0; op_class[op].op_class; op++) {
3457                 struct p2p_oper_class_map *o = &op_class[op];
3458                 u8 ch;
3459                 struct p2p_reg_class *reg = NULL, *cli_reg = NULL;
3460
3461                 mode = get_mode(wpa_s->hw.modes, wpa_s->hw.num_modes, o->mode);
3462                 if (mode == NULL)
3463                         continue;
3464                 for (ch = o->min_chan; ch <= o->max_chan; ch += o->inc) {
3465                         enum chan_allowed res;
3466                         res = wpas_p2p_verify_channel(wpa_s, mode, ch, o->bw);
3467                         if (res == ALLOWED) {
3468                                 if (reg == NULL) {
3469                                         wpa_printf(MSG_DEBUG, "P2P: Add operating class %u",
3470                                                    o->op_class);
3471                                         reg = &chan->reg_class[cla];
3472                                         cla++;
3473                                         reg->reg_class = o->op_class;
3474                                 }
3475                                 reg->channel[reg->channels] = ch;
3476                                 reg->channels++;
3477                         } else if (res == PASSIVE_ONLY &&
3478                                    wpa_s->conf->p2p_add_cli_chan) {
3479                                 if (cli_reg == NULL) {
3480                                         wpa_printf(MSG_DEBUG, "P2P: Add operating class %u (client only)",
3481                                                    o->op_class);
3482                                         cli_reg = &cli_chan->reg_class[cli_cla];
3483                                         cli_cla++;
3484                                         cli_reg->reg_class = o->op_class;
3485                                 }
3486                                 cli_reg->channel[cli_reg->channels] = ch;
3487                                 cli_reg->channels++;
3488                         }
3489                 }
3490                 if (reg) {
3491                         wpa_hexdump(MSG_DEBUG, "P2P: Channels",
3492                                     reg->channel, reg->channels);
3493                 }
3494                 if (cli_reg) {
3495                         wpa_hexdump(MSG_DEBUG, "P2P: Channels (client only)",
3496                                     cli_reg->channel, cli_reg->channels);
3497                 }
3498         }
3499
3500         chan->reg_classes = cla;
3501         cli_chan->reg_classes = cli_cla;
3502
3503         return 0;
3504 }
3505
3506
3507 int wpas_p2p_get_ht40_mode(struct wpa_supplicant *wpa_s,
3508                            struct hostapd_hw_modes *mode, u8 channel)
3509 {
3510         int op;
3511         enum chan_allowed ret;
3512
3513         for (op = 0; op_class[op].op_class; op++) {
3514                 struct p2p_oper_class_map *o = &op_class[op];
3515                 u8 ch;
3516
3517                 for (ch = o->min_chan; ch <= o->max_chan; ch += o->inc) {
3518                         if (o->mode != HOSTAPD_MODE_IEEE80211A ||
3519                             o->bw == BW20 || ch != channel)
3520                                 continue;
3521                         ret = wpas_p2p_verify_channel(wpa_s, mode, ch, o->bw);
3522                         if (ret == ALLOWED)
3523                                 return (o->bw == BW40MINUS) ? -1 : 1;
3524                 }
3525         }
3526         return 0;
3527 }
3528
3529
3530 int wpas_p2p_get_vht80_center(struct wpa_supplicant *wpa_s,
3531                               struct hostapd_hw_modes *mode, u8 channel)
3532 {
3533         if (!wpas_p2p_verify_channel(wpa_s, mode, channel, BW80))
3534                 return 0;
3535
3536         return wpas_p2p_get_center_80mhz(wpa_s, mode, channel);
3537 }
3538
3539
3540 static int wpas_get_noa(void *ctx, const u8 *interface_addr, u8 *buf,
3541                         size_t buf_len)
3542 {
3543         struct wpa_supplicant *wpa_s = ctx;
3544
3545         for (wpa_s = wpa_s->global->ifaces; wpa_s; wpa_s = wpa_s->next) {
3546                 if (os_memcmp(wpa_s->own_addr, interface_addr, ETH_ALEN) == 0)
3547                         break;
3548         }
3549         if (wpa_s == NULL)
3550                 return -1;
3551
3552         return wpa_drv_get_noa(wpa_s, buf, buf_len);
3553 }
3554
3555
3556 static int wpas_go_connected(void *ctx, const u8 *dev_addr)
3557 {
3558         struct wpa_supplicant *wpa_s = ctx;
3559
3560         for (wpa_s = wpa_s->global->ifaces; wpa_s; wpa_s = wpa_s->next) {
3561                 struct wpa_ssid *ssid = wpa_s->current_ssid;
3562                 if (ssid == NULL)
3563                         continue;
3564                 if (ssid->mode != WPAS_MODE_INFRA)
3565                         continue;
3566                 if (wpa_s->wpa_state != WPA_COMPLETED &&
3567                     wpa_s->wpa_state != WPA_GROUP_HANDSHAKE)
3568                         continue;
3569                 if (os_memcmp(wpa_s->go_dev_addr, dev_addr, ETH_ALEN) == 0)
3570                         return 1;
3571         }
3572
3573         return 0;
3574 }
3575
3576
3577 static int wpas_is_concurrent_session_active(void *ctx)
3578 {
3579         struct wpa_supplicant *wpa_s = ctx;
3580         struct wpa_supplicant *ifs;
3581
3582         for (ifs = wpa_s->global->ifaces; ifs; ifs = ifs->next) {
3583                 if (ifs == wpa_s)
3584                         continue;
3585                 if (ifs->wpa_state > WPA_ASSOCIATED)
3586                         return 1;
3587         }
3588         return 0;
3589 }
3590
3591
3592 static void wpas_p2p_debug_print(void *ctx, int level, const char *msg)
3593 {
3594         struct wpa_supplicant *wpa_s = ctx;
3595         wpa_msg_global(wpa_s, level, "P2P: %s", msg);
3596 }
3597
3598
3599 int wpas_p2p_add_p2pdev_interface(struct wpa_supplicant *wpa_s)
3600 {
3601         struct wpa_interface iface;
3602         struct wpa_supplicant *p2pdev_wpa_s;
3603         char ifname[100];
3604         char force_name[100];
3605         int ret;
3606
3607         os_snprintf(ifname, sizeof(ifname), P2P_MGMT_DEVICE_PREFIX "%s",
3608                     wpa_s->ifname);
3609         force_name[0] = '\0';
3610         wpa_s->pending_interface_type = WPA_IF_P2P_DEVICE;
3611         ret = wpa_drv_if_add(wpa_s, WPA_IF_P2P_DEVICE, ifname, NULL, NULL,
3612                              force_name, wpa_s->pending_interface_addr, NULL);
3613         if (ret < 0) {
3614                 wpa_printf(MSG_DEBUG, "P2P: Failed to create P2P Device interface");
3615                 return ret;
3616         }
3617         os_strlcpy(wpa_s->pending_interface_name, ifname,
3618                    sizeof(wpa_s->pending_interface_name));
3619
3620         os_memset(&iface, 0, sizeof(iface));
3621         iface.p2p_mgmt = 1;
3622         iface.ifname = wpa_s->pending_interface_name;
3623         iface.driver = wpa_s->driver->name;
3624         iface.driver_param = wpa_s->conf->driver_param;
3625         iface.confname = wpa_s->confname;
3626         p2pdev_wpa_s = wpa_supplicant_add_iface(wpa_s->global, &iface);
3627         if (!p2pdev_wpa_s) {
3628                 wpa_printf(MSG_DEBUG, "P2P: Failed to add P2P Device interface");
3629                 return -1;
3630         }
3631         p2pdev_wpa_s->parent = wpa_s;
3632
3633         wpa_s->pending_interface_name[0] = '\0';
3634         return 0;
3635 }
3636
3637
3638 static void wpas_presence_resp(void *ctx, const u8 *src, u8 status,
3639                                const u8 *noa, size_t noa_len)
3640 {
3641         struct wpa_supplicant *wpa_s, *intf = ctx;
3642         char hex[100];
3643
3644         for (wpa_s = intf->global->ifaces; wpa_s; wpa_s = wpa_s->next) {
3645                 if (wpa_s->waiting_presence_resp)
3646                         break;
3647         }
3648         if (!wpa_s) {
3649                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: No group interface was waiting for presence response");
3650                 return;
3651         }
3652         wpa_s->waiting_presence_resp = 0;
3653
3654         wpa_snprintf_hex(hex, sizeof(hex), noa, noa_len);
3655         wpa_msg(wpa_s, MSG_INFO, P2P_EVENT_PRESENCE_RESPONSE "src=" MACSTR
3656                 " status=%u noa=%s", MAC2STR(src), status, hex);
3657 }
3658
3659
3660 /**
3661  * wpas_p2p_init - Initialize P2P module for %wpa_supplicant
3662  * @global: Pointer to global data from wpa_supplicant_init()
3663  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
3664  * Returns: 0 on success, -1 on failure
3665  */
3666 int wpas_p2p_init(struct wpa_global *global, struct wpa_supplicant *wpa_s)
3667 {
3668         struct p2p_config p2p;
3669         unsigned int r;
3670         int i;
3671
3672         if (wpa_s->conf->p2p_disabled)
3673                 return 0;
3674
3675         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_CAPABLE))
3676                 return 0;
3677
3678         if (global->p2p)
3679                 return 0;
3680
3681         os_memset(&p2p, 0, sizeof(p2p));
3682         p2p.cb_ctx = wpa_s;
3683         p2p.debug_print = wpas_p2p_debug_print;
3684         p2p.p2p_scan = wpas_p2p_scan;
3685         p2p.send_action = wpas_send_action;
3686         p2p.send_action_done = wpas_send_action_done;
3687         p2p.go_neg_completed = wpas_go_neg_completed;
3688         p2p.go_neg_req_rx = wpas_go_neg_req_rx;
3689         p2p.dev_found = wpas_dev_found;
3690         p2p.dev_lost = wpas_dev_lost;
3691         p2p.find_stopped = wpas_find_stopped;
3692         p2p.start_listen = wpas_start_listen;
3693         p2p.stop_listen = wpas_stop_listen;
3694         p2p.send_probe_resp = wpas_send_probe_resp;
3695         p2p.sd_request = wpas_sd_request;
3696         p2p.sd_response = wpas_sd_response;
3697         p2p.prov_disc_req = wpas_prov_disc_req;
3698         p2p.prov_disc_resp = wpas_prov_disc_resp;
3699         p2p.prov_disc_fail = wpas_prov_disc_fail;
3700         p2p.invitation_process = wpas_invitation_process;
3701         p2p.invitation_received = wpas_invitation_received;
3702         p2p.invitation_result = wpas_invitation_result;
3703         p2p.get_noa = wpas_get_noa;
3704         p2p.go_connected = wpas_go_connected;
3705         p2p.presence_resp = wpas_presence_resp;
3706         p2p.is_concurrent_session_active = wpas_is_concurrent_session_active;
3707
3708         os_memcpy(wpa_s->global->p2p_dev_addr, wpa_s->own_addr, ETH_ALEN);
3709         os_memcpy(p2p.dev_addr, wpa_s->global->p2p_dev_addr, ETH_ALEN);
3710         p2p.dev_name = wpa_s->conf->device_name;
3711         p2p.manufacturer = wpa_s->conf->manufacturer;
3712         p2p.model_name = wpa_s->conf->model_name;
3713         p2p.model_number = wpa_s->conf->model_number;
3714         p2p.serial_number = wpa_s->conf->serial_number;
3715         if (wpa_s->wps) {
3716                 os_memcpy(p2p.uuid, wpa_s->wps->uuid, 16);
3717                 p2p.config_methods = wpa_s->wps->config_methods;
3718         }
3719
3720         if (wpa_s->conf->p2p_listen_reg_class &&
3721             wpa_s->conf->p2p_listen_channel) {
3722                 p2p.reg_class = wpa_s->conf->p2p_listen_reg_class;
3723                 p2p.channel = wpa_s->conf->p2p_listen_channel;
3724         } else {
3725                 p2p.reg_class = 81;
3726                 /*
3727                  * Pick one of the social channels randomly as the listen
3728                  * channel.
3729                  */
3730                 os_get_random((u8 *) &r, sizeof(r));
3731                 p2p.channel = 1 + (r % 3) * 5;
3732         }
3733         wpa_printf(MSG_DEBUG, "P2P: Own listen channel: %d", p2p.channel);
3734
3735         if (wpa_s->conf->p2p_oper_reg_class &&
3736             wpa_s->conf->p2p_oper_channel) {
3737                 p2p.op_reg_class = wpa_s->conf->p2p_oper_reg_class;
3738                 p2p.op_channel = wpa_s->conf->p2p_oper_channel;
3739                 p2p.cfg_op_channel = 1;
3740                 wpa_printf(MSG_DEBUG, "P2P: Configured operating channel: "
3741                            "%d:%d", p2p.op_reg_class, p2p.op_channel);
3742
3743         } else {
3744                 p2p.op_reg_class = 81;
3745                 /*
3746                  * Use random operation channel from (1, 6, 11) if no other
3747                  * preference is indicated.
3748                  */
3749                 os_get_random((u8 *) &r, sizeof(r));
3750                 p2p.op_channel = 1 + (r % 3) * 5;
3751                 p2p.cfg_op_channel = 0;
3752                 wpa_printf(MSG_DEBUG, "P2P: Random operating channel: "
3753                            "%d:%d", p2p.op_reg_class, p2p.op_channel);
3754         }
3755
3756         if (wpa_s->conf->p2p_pref_chan && wpa_s->conf->num_p2p_pref_chan) {
3757                 p2p.pref_chan = wpa_s->conf->p2p_pref_chan;
3758                 p2p.num_pref_chan = wpa_s->conf->num_p2p_pref_chan;
3759         }
3760
3761         if (wpa_s->conf->country[0] && wpa_s->conf->country[1]) {
3762                 os_memcpy(p2p.country, wpa_s->conf->country, 2);
3763                 p2p.country[2] = 0x04;
3764         } else
3765                 os_memcpy(p2p.country, "XX\x04", 3);
3766
3767         if (wpas_p2p_setup_channels(wpa_s, &p2p.channels, &p2p.cli_channels)) {
3768                 wpa_printf(MSG_ERROR, "P2P: Failed to configure supported "
3769                            "channel list");
3770                 return -1;
3771         }
3772
3773         os_memcpy(p2p.pri_dev_type, wpa_s->conf->device_type,
3774                   WPS_DEV_TYPE_LEN);
3775
3776         p2p.num_sec_dev_types = wpa_s->conf->num_sec_device_types;
3777         os_memcpy(p2p.sec_dev_type, wpa_s->conf->sec_device_type,
3778                   p2p.num_sec_dev_types * WPS_DEV_TYPE_LEN);
3779
3780         p2p.concurrent_operations = !!(wpa_s->drv_flags &
3781                                        WPA_DRIVER_FLAGS_P2P_CONCURRENT);
3782
3783         p2p.max_peers = 100;
3784
3785         if (wpa_s->conf->p2p_ssid_postfix) {
3786                 p2p.ssid_postfix_len =
3787                         os_strlen(wpa_s->conf->p2p_ssid_postfix);
3788                 if (p2p.ssid_postfix_len > sizeof(p2p.ssid_postfix))
3789                         p2p.ssid_postfix_len = sizeof(p2p.ssid_postfix);
3790                 os_memcpy(p2p.ssid_postfix, wpa_s->conf->p2p_ssid_postfix,
3791                           p2p.ssid_postfix_len);
3792         }
3793
3794         p2p.p2p_intra_bss = wpa_s->conf->p2p_intra_bss;
3795
3796         p2p.max_listen = wpa_s->max_remain_on_chan;
3797
3798         global->p2p = p2p_init(&p2p);
3799         if (global->p2p == NULL)
3800                 return -1;
3801         global->p2p_init_wpa_s = wpa_s;
3802
3803         for (i = 0; i < MAX_WPS_VENDOR_EXT; i++) {
3804                 if (wpa_s->conf->wps_vendor_ext[i] == NULL)
3805                         continue;
3806                 p2p_add_wps_vendor_extension(
3807                         global->p2p, wpa_s->conf->wps_vendor_ext[i]);
3808         }
3809
3810         p2p_set_no_go_freq(global->p2p, &wpa_s->conf->p2p_no_go_freq);
3811
3812         return 0;
3813 }
3814
3815
3816 /**
3817  * wpas_p2p_deinit - Deinitialize per-interface P2P data
3818  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
3819  *
3820  * This function deinitialize per-interface P2P data.
3821  */
3822 void wpas_p2p_deinit(struct wpa_supplicant *wpa_s)
3823 {
3824         if (wpa_s->driver && wpa_s->drv_priv)
3825                 wpa_drv_probe_req_report(wpa_s, 0);
3826
3827         if (wpa_s->go_params) {
3828                 /* Clear any stored provisioning info */
3829                 p2p_clear_provisioning_info(
3830                         wpa_s->global->p2p,
3831                         wpa_s->go_params->peer_device_addr);
3832         }
3833
3834         os_free(wpa_s->go_params);
3835         wpa_s->go_params = NULL;
3836         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s, NULL);
3837         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
3838         wpa_s->p2p_long_listen = 0;
3839         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
3840         eloop_cancel_timeout(wpas_p2p_group_idle_timeout, wpa_s, NULL);
3841         wpas_p2p_remove_pending_group_interface(wpa_s);
3842         eloop_cancel_timeout(wpas_p2p_group_freq_conflict, wpa_s, NULL);
3843         wpas_p2p_listen_work_done(wpa_s);
3844         if (wpa_s->p2p_send_action_work) {
3845                 os_free(wpa_s->p2p_send_action_work->ctx);
3846                 radio_work_done(wpa_s->p2p_send_action_work);
3847                 wpa_s->p2p_send_action_work = NULL;
3848         }
3849         eloop_cancel_timeout(wpas_p2p_send_action_work_timeout, wpa_s, NULL);
3850
3851         /* TODO: remove group interface from the driver if this wpa_s instance
3852          * is on top of a P2P group interface */
3853 }
3854
3855
3856 /**
3857  * wpas_p2p_deinit_global - Deinitialize global P2P module
3858  * @global: Pointer to global data from wpa_supplicant_init()
3859  *
3860  * This function deinitializes the global (per device) P2P module.
3861  */
3862 void wpas_p2p_deinit_global(struct wpa_global *global)
3863 {
3864         struct wpa_supplicant *wpa_s, *tmp;
3865
3866         wpa_s = global->ifaces;
3867         if (wpa_s)
3868                 wpas_p2p_service_flush(wpa_s);
3869
3870         if (global->p2p == NULL)
3871                 return;
3872
3873         /* Remove remaining P2P group interfaces */
3874         while (wpa_s && wpa_s->p2p_group_interface != NOT_P2P_GROUP_INTERFACE)
3875                 wpa_s = wpa_s->next;
3876         while (wpa_s) {
3877                 tmp = global->ifaces;
3878                 while (tmp &&
3879                        (tmp == wpa_s ||
3880                         tmp->p2p_group_interface == NOT_P2P_GROUP_INTERFACE)) {
3881                         tmp = tmp->next;
3882                 }
3883                 if (tmp == NULL)
3884                         break;
3885                 /* Disconnect from the P2P group and deinit the interface */
3886                 wpas_p2p_disconnect(tmp);
3887         }
3888
3889         /*
3890          * Deinit GO data on any possibly remaining interface (if main
3891          * interface is used as GO).
3892          */
3893         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
3894                 if (wpa_s->ap_iface)
3895                         wpas_p2p_group_deinit(wpa_s);
3896         }
3897
3898         p2p_deinit(global->p2p);
3899         global->p2p = NULL;
3900         global->p2p_init_wpa_s = NULL;
3901 }
3902
3903
3904 static int wpas_p2p_create_iface(struct wpa_supplicant *wpa_s)
3905 {
3906         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_DEDICATED_P2P_DEVICE) &&
3907             wpa_s->conf->p2p_no_group_iface)
3908                 return 0; /* separate interface disabled per configuration */
3909         if (wpa_s->drv_flags &
3910             (WPA_DRIVER_FLAGS_P2P_DEDICATED_INTERFACE |
3911              WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P))
3912                 return 1; /* P2P group requires a new interface in every case
3913                            */
3914         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_CONCURRENT))
3915                 return 0; /* driver does not support concurrent operations */
3916         if (wpa_s->global->ifaces->next)
3917                 return 1; /* more that one interface already in use */
3918         if (wpa_s->wpa_state >= WPA_AUTHENTICATING)
3919                 return 1; /* this interface is already in use */
3920         return 0;
3921 }
3922
3923
3924 static int wpas_p2p_start_go_neg(struct wpa_supplicant *wpa_s,
3925                                  const u8 *peer_addr,
3926                                  enum p2p_wps_method wps_method,
3927                                  int go_intent, const u8 *own_interface_addr,
3928                                  unsigned int force_freq, int persistent_group,
3929                                  struct wpa_ssid *ssid, unsigned int pref_freq)
3930 {
3931         if (persistent_group && wpa_s->conf->persistent_reconnect)
3932                 persistent_group = 2;
3933
3934         /*
3935          * Increase GO config timeout if HT40 is used since it takes some time
3936          * to scan channels for coex purposes before the BSS can be started.
3937          */
3938         p2p_set_config_timeout(wpa_s->global->p2p,
3939                                wpa_s->p2p_go_ht40 ? 255 : 100, 20);
3940
3941         return p2p_connect(wpa_s->global->p2p, peer_addr, wps_method,
3942                            go_intent, own_interface_addr, force_freq,
3943                            persistent_group, ssid ? ssid->ssid : NULL,
3944                            ssid ? ssid->ssid_len : 0,
3945                            wpa_s->p2p_pd_before_go_neg, pref_freq, 0);
3946 }
3947
3948
3949 static int wpas_p2p_auth_go_neg(struct wpa_supplicant *wpa_s,
3950                                 const u8 *peer_addr,
3951                                 enum p2p_wps_method wps_method,
3952                                 int go_intent, const u8 *own_interface_addr,
3953                                 unsigned int force_freq, int persistent_group,
3954                                 struct wpa_ssid *ssid, unsigned int pref_freq)
3955 {
3956         if (persistent_group && wpa_s->conf->persistent_reconnect)
3957                 persistent_group = 2;
3958
3959         return p2p_authorize(wpa_s->global->p2p, peer_addr, wps_method,
3960                              go_intent, own_interface_addr, force_freq,
3961                              persistent_group, ssid ? ssid->ssid : NULL,
3962                              ssid ? ssid->ssid_len : 0, pref_freq, 0);
3963 }
3964
3965
3966 static void wpas_p2p_check_join_scan_limit(struct wpa_supplicant *wpa_s)
3967 {
3968         wpa_s->p2p_join_scan_count++;
3969         wpa_printf(MSG_DEBUG, "P2P: Join scan attempt %d",
3970                    wpa_s->p2p_join_scan_count);
3971         if (wpa_s->p2p_join_scan_count > P2P_MAX_JOIN_SCAN_ATTEMPTS) {
3972                 wpa_printf(MSG_DEBUG, "P2P: Failed to find GO " MACSTR
3973                            " for join operationg - stop join attempt",
3974                            MAC2STR(wpa_s->pending_join_iface_addr));
3975                 eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
3976                 if (wpa_s->p2p_auto_pd) {
3977                         wpa_s->p2p_auto_pd = 0;
3978                         wpa_msg_global(wpa_s, MSG_INFO,
3979                                        P2P_EVENT_PROV_DISC_FAILURE
3980                                        " p2p_dev_addr=" MACSTR " status=N/A",
3981                                        MAC2STR(wpa_s->pending_join_dev_addr));
3982                         return;
3983                 }
3984                 wpa_msg_global(wpa_s->parent, MSG_INFO,
3985                                P2P_EVENT_GROUP_FORMATION_FAILURE);
3986         }
3987 }
3988
3989
3990 static int wpas_check_freq_conflict(struct wpa_supplicant *wpa_s, int freq)
3991 {
3992         int *freqs, res, num, i;
3993
3994         if (wpas_p2p_num_unused_channels(wpa_s) > 0) {
3995                 /* Multiple channels are supported and not all are in use */
3996                 return 0;
3997         }
3998
3999         freqs = os_calloc(wpa_s->num_multichan_concurrent, sizeof(int));
4000         if (!freqs)
4001                 return 1;
4002
4003         num = wpas_p2p_valid_oper_freqs(wpa_s, freqs,
4004                                         wpa_s->num_multichan_concurrent);
4005         if (num < 0) {
4006                 res = 1;
4007                 goto exit_free;
4008         }
4009
4010         for (i = 0; i < num; i++) {
4011                 if (freqs[i] == freq) {
4012                         wpa_printf(MSG_DEBUG, "P2P: Frequency %d MHz in use by another virtual interface and can be used",
4013                                    freq);
4014                         res = 0;
4015                         goto exit_free;
4016                 }
4017         }
4018
4019         res = 1;
4020
4021 exit_free:
4022         os_free(freqs);
4023         return res;
4024 }
4025
4026
4027 static int wpas_p2p_peer_go(struct wpa_supplicant *wpa_s,
4028                             const u8 *peer_dev_addr)
4029 {
4030         struct wpa_bss *bss;
4031         int updated;
4032
4033         bss = wpa_bss_get_p2p_dev_addr(wpa_s, peer_dev_addr);
4034         if (bss == NULL)
4035                 return -1;
4036         if (bss->last_update_idx < wpa_s->bss_update_idx) {
4037                 wpa_printf(MSG_DEBUG, "P2P: Peer BSS entry not updated in the "
4038                            "last scan");
4039                 return 0;
4040         }
4041
4042         updated = os_reltime_before(&wpa_s->p2p_auto_started,
4043                                     &bss->last_update);
4044         wpa_printf(MSG_DEBUG, "P2P: Current BSS entry for peer updated at "
4045                    "%ld.%06ld (%supdated in last scan)",
4046                    bss->last_update.sec, bss->last_update.usec,
4047                    updated ? "": "not ");
4048
4049         return updated;
4050 }
4051
4052
4053 static void wpas_p2p_scan_res_join(struct wpa_supplicant *wpa_s,
4054                                    struct wpa_scan_results *scan_res)
4055 {
4056         struct wpa_bss *bss = NULL;
4057         int freq;
4058         u8 iface_addr[ETH_ALEN];
4059
4060         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
4061
4062         if (wpa_s->global->p2p_disabled)
4063                 return;
4064
4065         wpa_printf(MSG_DEBUG, "P2P: Scan results received (%d BSS) for %sjoin",
4066                    scan_res ? (int) scan_res->num : -1,
4067                    wpa_s->p2p_auto_join ? "auto_" : "");
4068
4069         if (scan_res)
4070                 wpas_p2p_scan_res_handler(wpa_s, scan_res);
4071
4072         if (wpa_s->p2p_auto_pd) {
4073                 int join = wpas_p2p_peer_go(wpa_s,
4074                                             wpa_s->pending_join_dev_addr);
4075                 if (join == 0 &&
4076                     wpa_s->auto_pd_scan_retry < P2P_AUTO_PD_SCAN_ATTEMPTS) {
4077                         wpa_s->auto_pd_scan_retry++;
4078                         bss = wpa_bss_get_bssid_latest(
4079                                 wpa_s, wpa_s->pending_join_dev_addr);
4080                         if (bss) {
4081                                 freq = bss->freq;
4082                                 wpa_printf(MSG_DEBUG, "P2P: Scan retry %d for "
4083                                            "the peer " MACSTR " at %d MHz",
4084                                            wpa_s->auto_pd_scan_retry,
4085                                            MAC2STR(wpa_s->
4086                                                    pending_join_dev_addr),
4087                                            freq);
4088                                 wpas_p2p_join_scan_req(wpa_s, freq, NULL, 0);
4089                                 return;
4090                         }
4091                 }
4092
4093                 if (join < 0)
4094                         join = 0;
4095
4096                 wpa_s->p2p_auto_pd = 0;
4097                 wpa_s->pending_pd_use = join ? AUTO_PD_JOIN : AUTO_PD_GO_NEG;
4098                 wpa_printf(MSG_DEBUG, "P2P: Auto PD with " MACSTR " join=%d",
4099                            MAC2STR(wpa_s->pending_join_dev_addr), join);
4100                 if (p2p_prov_disc_req(wpa_s->global->p2p,
4101                                       wpa_s->pending_join_dev_addr,
4102                                       wpa_s->pending_pd_config_methods, join,
4103                                       0, wpa_s->user_initiated_pd) < 0) {
4104                         wpa_s->p2p_auto_pd = 0;
4105                         wpa_msg_global(wpa_s, MSG_INFO,
4106                                        P2P_EVENT_PROV_DISC_FAILURE
4107                                        " p2p_dev_addr=" MACSTR " status=N/A",
4108                                        MAC2STR(wpa_s->pending_join_dev_addr));
4109                 }
4110                 return;
4111         }
4112
4113         if (wpa_s->p2p_auto_join) {
4114                 int join = wpas_p2p_peer_go(wpa_s,
4115                                             wpa_s->pending_join_dev_addr);
4116                 if (join < 0) {
4117                         wpa_printf(MSG_DEBUG, "P2P: Peer was not found to be "
4118                                    "running a GO -> use GO Negotiation");
4119                         wpas_p2p_connect(wpa_s, wpa_s->pending_join_dev_addr,
4120                                          wpa_s->p2p_pin, wpa_s->p2p_wps_method,
4121                                          wpa_s->p2p_persistent_group, 0, 0, 0,
4122                                          wpa_s->p2p_go_intent,
4123                                          wpa_s->p2p_connect_freq,
4124                                          wpa_s->p2p_persistent_id,
4125                                          wpa_s->p2p_pd_before_go_neg,
4126                                          wpa_s->p2p_go_ht40,
4127                                          wpa_s->p2p_go_vht);
4128                         return;
4129                 }
4130
4131                 wpa_printf(MSG_DEBUG, "P2P: Peer was found running GO%s -> "
4132                            "try to join the group", join ? "" :
4133                            " in older scan");
4134                 if (!join)
4135                         wpa_s->p2p_fallback_to_go_neg = 1;
4136         }
4137
4138         freq = p2p_get_oper_freq(wpa_s->global->p2p,
4139                                  wpa_s->pending_join_iface_addr);
4140         if (freq < 0 &&
4141             p2p_get_interface_addr(wpa_s->global->p2p,
4142                                    wpa_s->pending_join_dev_addr,
4143                                    iface_addr) == 0 &&
4144             os_memcmp(iface_addr, wpa_s->pending_join_dev_addr, ETH_ALEN) != 0)
4145         {
4146                 wpa_printf(MSG_DEBUG, "P2P: Overwrite pending interface "
4147                            "address for join from " MACSTR " to " MACSTR
4148                            " based on newly discovered P2P peer entry",
4149                            MAC2STR(wpa_s->pending_join_iface_addr),
4150                            MAC2STR(iface_addr));
4151                 os_memcpy(wpa_s->pending_join_iface_addr, iface_addr,
4152                           ETH_ALEN);
4153
4154                 freq = p2p_get_oper_freq(wpa_s->global->p2p,
4155                                          wpa_s->pending_join_iface_addr);
4156         }
4157         if (freq >= 0) {
4158                 wpa_printf(MSG_DEBUG, "P2P: Target GO operating frequency "
4159                            "from P2P peer table: %d MHz", freq);
4160         }
4161         if (wpa_s->p2p_join_ssid_len) {
4162                 wpa_printf(MSG_DEBUG, "P2P: Trying to find target GO BSS entry based on BSSID "
4163                            MACSTR " and SSID %s",
4164                            MAC2STR(wpa_s->pending_join_iface_addr),
4165                            wpa_ssid_txt(wpa_s->p2p_join_ssid,
4166                                         wpa_s->p2p_join_ssid_len));
4167                 bss = wpa_bss_get(wpa_s, wpa_s->pending_join_iface_addr,
4168                                   wpa_s->p2p_join_ssid,
4169                                   wpa_s->p2p_join_ssid_len);
4170         }
4171         if (!bss) {
4172                 wpa_printf(MSG_DEBUG, "P2P: Trying to find target GO BSS entry based on BSSID "
4173                            MACSTR, MAC2STR(wpa_s->pending_join_iface_addr));
4174                 bss = wpa_bss_get_bssid_latest(wpa_s,
4175                                                wpa_s->pending_join_iface_addr);
4176         }
4177         if (bss) {
4178                 freq = bss->freq;
4179                 wpa_printf(MSG_DEBUG, "P2P: Target GO operating frequency "
4180                            "from BSS table: %d MHz (SSID %s)", freq,
4181                            wpa_ssid_txt(bss->ssid, bss->ssid_len));
4182         }
4183         if (freq > 0) {
4184                 u16 method;
4185
4186                 if (wpas_check_freq_conflict(wpa_s, freq) > 0) {
4187                         wpa_msg_global(wpa_s->parent, MSG_INFO,
4188                                        P2P_EVENT_GROUP_FORMATION_FAILURE
4189                                        "reason=FREQ_CONFLICT");
4190                         return;
4191                 }
4192
4193                 wpa_printf(MSG_DEBUG, "P2P: Send Provision Discovery Request "
4194                            "prior to joining an existing group (GO " MACSTR
4195                            " freq=%u MHz)",
4196                            MAC2STR(wpa_s->pending_join_dev_addr), freq);
4197                 wpa_s->pending_pd_before_join = 1;
4198
4199                 switch (wpa_s->pending_join_wps_method) {
4200                 case WPS_PIN_DISPLAY:
4201                         method = WPS_CONFIG_KEYPAD;
4202                         break;
4203                 case WPS_PIN_KEYPAD:
4204                         method = WPS_CONFIG_DISPLAY;
4205                         break;
4206                 case WPS_PBC:
4207                         method = WPS_CONFIG_PUSHBUTTON;
4208                         break;
4209                 default:
4210                         method = 0;
4211                         break;
4212                 }
4213
4214                 if ((p2p_get_provisioning_info(wpa_s->global->p2p,
4215                                                wpa_s->pending_join_dev_addr) ==
4216                      method)) {
4217                         /*
4218                          * We have already performed provision discovery for
4219                          * joining the group. Proceed directly to join
4220                          * operation without duplicated provision discovery. */
4221                         wpa_printf(MSG_DEBUG, "P2P: Provision discovery "
4222                                    "with " MACSTR " already done - proceed to "
4223                                    "join",
4224                                    MAC2STR(wpa_s->pending_join_dev_addr));
4225                         wpa_s->pending_pd_before_join = 0;
4226                         goto start;
4227                 }
4228
4229                 if (p2p_prov_disc_req(wpa_s->global->p2p,
4230                                       wpa_s->pending_join_dev_addr, method, 1,
4231                                       freq, wpa_s->user_initiated_pd) < 0) {
4232                         wpa_printf(MSG_DEBUG, "P2P: Failed to send Provision "
4233                                    "Discovery Request before joining an "
4234                                    "existing group");
4235                         wpa_s->pending_pd_before_join = 0;
4236                         goto start;
4237                 }
4238                 return;
4239         }
4240
4241         wpa_printf(MSG_DEBUG, "P2P: Failed to find BSS/GO - try again later");
4242         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
4243         eloop_register_timeout(1, 0, wpas_p2p_join_scan, wpa_s, NULL);
4244         wpas_p2p_check_join_scan_limit(wpa_s);
4245         return;
4246
4247 start:
4248         /* Start join operation immediately */
4249         wpas_p2p_join_start(wpa_s);
4250 }
4251
4252
4253 static void wpas_p2p_join_scan_req(struct wpa_supplicant *wpa_s, int freq,
4254                                    const u8 *ssid, size_t ssid_len)
4255 {
4256         int ret;
4257         struct wpa_driver_scan_params params;
4258         struct wpabuf *wps_ie, *ies;
4259         size_t ielen;
4260         int freqs[2] = { 0, 0 };
4261
4262         os_memset(&params, 0, sizeof(params));
4263
4264         /* P2P Wildcard SSID */
4265         params.num_ssids = 1;
4266         if (ssid && ssid_len) {
4267                 params.ssids[0].ssid = ssid;
4268                 params.ssids[0].ssid_len = ssid_len;
4269                 os_memcpy(wpa_s->p2p_join_ssid, ssid, ssid_len);
4270                 wpa_s->p2p_join_ssid_len = ssid_len;
4271         } else {
4272                 params.ssids[0].ssid = (u8 *) P2P_WILDCARD_SSID;
4273                 params.ssids[0].ssid_len = P2P_WILDCARD_SSID_LEN;
4274                 wpa_s->p2p_join_ssid_len = 0;
4275         }
4276
4277         wpa_s->wps->dev.p2p = 1;
4278         wps_ie = wps_build_probe_req_ie(DEV_PW_DEFAULT, &wpa_s->wps->dev,
4279                                         wpa_s->wps->uuid, WPS_REQ_ENROLLEE, 0,
4280                                         NULL);
4281         if (wps_ie == NULL) {
4282                 wpas_p2p_scan_res_join(wpa_s, NULL);
4283                 return;
4284         }
4285
4286         ielen = p2p_scan_ie_buf_len(wpa_s->global->p2p);
4287         ies = wpabuf_alloc(wpabuf_len(wps_ie) + ielen);
4288         if (ies == NULL) {
4289                 wpabuf_free(wps_ie);
4290                 wpas_p2p_scan_res_join(wpa_s, NULL);
4291                 return;
4292         }
4293         wpabuf_put_buf(ies, wps_ie);
4294         wpabuf_free(wps_ie);
4295
4296         p2p_scan_ie(wpa_s->global->p2p, ies, NULL);
4297
4298         params.p2p_probe = 1;
4299         params.extra_ies = wpabuf_head(ies);
4300         params.extra_ies_len = wpabuf_len(ies);
4301
4302         if (!freq) {
4303                 int oper_freq;
4304                 /*
4305                  * If freq is not provided, check the operating freq of the GO
4306                  * and use a single channel scan on if possible.
4307                  */
4308                 oper_freq = p2p_get_oper_freq(wpa_s->global->p2p,
4309                                               wpa_s->pending_join_iface_addr);
4310                 if (oper_freq > 0)
4311                         freq = oper_freq;
4312         }
4313         if (freq > 0) {
4314                 freqs[0] = freq;
4315                 params.freqs = freqs;
4316         }
4317
4318         /*
4319          * Run a scan to update BSS table and start Provision Discovery once
4320          * the new scan results become available.
4321          */
4322         ret = wpa_drv_scan(wpa_s, &params);
4323         if (!ret) {
4324                 os_get_reltime(&wpa_s->scan_trigger_time);
4325                 wpa_s->scan_res_handler = wpas_p2p_scan_res_join;
4326                 wpa_s->own_scan_requested = 1;
4327         }
4328
4329         wpabuf_free(ies);
4330
4331         if (ret) {
4332                 wpa_printf(MSG_DEBUG, "P2P: Failed to start scan for join - "
4333                            "try again later");
4334                 eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
4335                 eloop_register_timeout(1, 0, wpas_p2p_join_scan, wpa_s, NULL);
4336                 wpas_p2p_check_join_scan_limit(wpa_s);
4337         }
4338 }
4339
4340
4341 static void wpas_p2p_join_scan(void *eloop_ctx, void *timeout_ctx)
4342 {
4343         struct wpa_supplicant *wpa_s = eloop_ctx;
4344         wpas_p2p_join_scan_req(wpa_s, 0, NULL, 0);
4345 }
4346
4347
4348 static int wpas_p2p_join(struct wpa_supplicant *wpa_s, const u8 *iface_addr,
4349                          const u8 *dev_addr, enum p2p_wps_method wps_method,
4350                          int auto_join, const u8 *ssid, size_t ssid_len)
4351 {
4352         wpa_printf(MSG_DEBUG, "P2P: Request to join existing group (iface "
4353                    MACSTR " dev " MACSTR ")%s",
4354                    MAC2STR(iface_addr), MAC2STR(dev_addr),
4355                    auto_join ? " (auto_join)" : "");
4356         if (ssid && ssid_len) {
4357                 wpa_printf(MSG_DEBUG, "P2P: Group SSID specified: %s",
4358                            wpa_ssid_txt(ssid, ssid_len));
4359         }
4360
4361         wpa_s->p2p_auto_pd = 0;
4362         wpa_s->p2p_auto_join = !!auto_join;
4363         os_memcpy(wpa_s->pending_join_iface_addr, iface_addr, ETH_ALEN);
4364         os_memcpy(wpa_s->pending_join_dev_addr, dev_addr, ETH_ALEN);
4365         wpa_s->pending_join_wps_method = wps_method;
4366
4367         /* Make sure we are not running find during connection establishment */
4368         wpas_p2p_stop_find(wpa_s);
4369
4370         wpa_s->p2p_join_scan_count = 0;
4371         wpas_p2p_join_scan_req(wpa_s, 0, ssid, ssid_len);
4372         return 0;
4373 }
4374
4375
4376 static int wpas_p2p_join_start(struct wpa_supplicant *wpa_s)
4377 {
4378         struct wpa_supplicant *group;
4379         struct p2p_go_neg_results res;
4380         struct wpa_bss *bss;
4381
4382         group = wpas_p2p_get_group_iface(wpa_s, 0, 0);
4383         if (group == NULL)
4384                 return -1;
4385         if (group != wpa_s) {
4386                 os_memcpy(group->p2p_pin, wpa_s->p2p_pin,
4387                           sizeof(group->p2p_pin));
4388                 group->p2p_wps_method = wpa_s->p2p_wps_method;
4389         } else {
4390                 /*
4391                  * Need to mark the current interface for p2p_group_formation
4392                  * when a separate group interface is not used. This is needed
4393                  * to allow p2p_cancel stop a pending p2p_connect-join.
4394                  * wpas_p2p_init_group_interface() addresses this for the case
4395                  * where a separate group interface is used.
4396                  */
4397                 wpa_s->global->p2p_group_formation = wpa_s;
4398         }
4399
4400         group->p2p_in_provisioning = 1;
4401         group->p2p_fallback_to_go_neg = wpa_s->p2p_fallback_to_go_neg;
4402
4403         os_memset(&res, 0, sizeof(res));
4404         os_memcpy(res.peer_interface_addr, wpa_s->pending_join_iface_addr,
4405                   ETH_ALEN);
4406         res.wps_method = wpa_s->pending_join_wps_method;
4407         bss = wpa_bss_get_bssid_latest(wpa_s, wpa_s->pending_join_iface_addr);
4408         if (bss) {
4409                 res.freq = bss->freq;
4410                 res.ssid_len = bss->ssid_len;
4411                 os_memcpy(res.ssid, bss->ssid, bss->ssid_len);
4412                 wpa_printf(MSG_DEBUG, "P2P: Join target GO operating frequency "
4413                            "from BSS table: %d MHz (SSID %s)", bss->freq,
4414                            wpa_ssid_txt(bss->ssid, bss->ssid_len));
4415         }
4416
4417         if (wpa_s->off_channel_freq || wpa_s->roc_waiting_drv_freq) {
4418                 wpa_printf(MSG_DEBUG, "P2P: Cancel remain-on-channel prior to "
4419                            "starting client");
4420                 wpa_drv_cancel_remain_on_channel(wpa_s);
4421                 wpa_s->off_channel_freq = 0;
4422                 wpa_s->roc_waiting_drv_freq = 0;
4423         }
4424         wpas_start_wps_enrollee(group, &res);
4425
4426         /*
4427          * Allow a longer timeout for join-a-running-group than normal 15
4428          * second group formation timeout since the GO may not have authorized
4429          * our connection yet.
4430          */
4431         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s, NULL);
4432         eloop_register_timeout(60, 0, wpas_p2p_group_formation_timeout,
4433                                wpa_s, NULL);
4434
4435         return 0;
4436 }
4437
4438
4439 static int wpas_p2p_setup_freqs(struct wpa_supplicant *wpa_s, int freq,
4440                                 int *force_freq, int *pref_freq, int go)
4441 {
4442         int *freqs, res;
4443         unsigned int freq_in_use = 0, num, i;
4444
4445         freqs = os_calloc(wpa_s->num_multichan_concurrent, sizeof(int));
4446         if (!freqs)
4447                 return -1;
4448
4449         num = get_shared_radio_freqs(wpa_s, freqs,
4450                                      wpa_s->num_multichan_concurrent);
4451         wpa_printf(MSG_DEBUG,
4452                    "P2P: Setup freqs: freq=%d num_MCC=%d shared_freqs=%u",
4453                    freq, wpa_s->num_multichan_concurrent, num);
4454
4455         if (freq > 0) {
4456                 int ret;
4457                 if (go)
4458                         ret = p2p_supported_freq(wpa_s->global->p2p, freq);
4459                 else
4460                         ret = p2p_supported_freq_cli(wpa_s->global->p2p, freq);
4461                 if (!ret) {
4462                         wpa_printf(MSG_DEBUG, "P2P: The forced channel "
4463                                    "(%u MHz) is not supported for P2P uses",
4464                                    freq);
4465                         res = -3;
4466                         goto exit_free;
4467                 }
4468
4469                 for (i = 0; i < num; i++) {
4470                         if (freqs[i] == freq)
4471                                 freq_in_use = 1;
4472                 }
4473
4474                 if (num == wpa_s->num_multichan_concurrent && !freq_in_use) {
4475                         wpa_printf(MSG_DEBUG, "P2P: Cannot start P2P group on %u MHz as there are no available channels",
4476                                    freq);
4477                         res = -2;
4478                         goto exit_free;
4479                 }
4480                 wpa_printf(MSG_DEBUG, "P2P: Trying to force us to use the "
4481                            "requested channel (%u MHz)", freq);
4482                 *force_freq = freq;
4483                 goto exit_ok;
4484         }
4485
4486         for (i = 0; i < num; i++) {
4487                 if (!p2p_supported_freq(wpa_s->global->p2p, freqs[i]))
4488                         continue;
4489
4490                 if (*pref_freq == 0 && num < wpa_s->num_multichan_concurrent) {
4491                         wpa_printf(MSG_DEBUG, "P2P: Try to prefer a frequency (%u MHz) we are already using",
4492                                    freqs[i]);
4493                         *pref_freq = freqs[i];
4494                 } else {
4495                         wpa_printf(MSG_DEBUG, "P2P: Try to force us to use frequency (%u MHz) which is already in use",
4496                                    freqs[i]);
4497                         *force_freq = freqs[i];
4498                 }
4499                 break;
4500         }
4501
4502         if (i == num) {
4503                 if (num < wpa_s->num_multichan_concurrent && num > 0) {
4504                         wpa_printf(MSG_DEBUG, "P2P: Current operating channels are not available for P2P. Try to use another channel");
4505                         *force_freq = 0;
4506                 } else if (num < wpa_s->num_multichan_concurrent) {
4507                         wpa_printf(MSG_DEBUG, "P2P: No current operating channels - try to use a new channel");
4508                         *force_freq = 0;
4509                 } else {
4510                         wpa_printf(MSG_DEBUG, "P2P: All channels are in use and none of them are P2P enabled. Cannot start P2P group");
4511                         res = -2;
4512                         goto exit_free;
4513                 }
4514         }
4515
4516 exit_ok:
4517         res = 0;
4518 exit_free:
4519         os_free(freqs);
4520         return res;
4521 }
4522
4523
4524 /**
4525  * wpas_p2p_connect - Request P2P Group Formation to be started
4526  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
4527  * @peer_addr: Address of the peer P2P Device
4528  * @pin: PIN to use during provisioning or %NULL to indicate PBC mode
4529  * @persistent_group: Whether to create a persistent group
4530  * @auto_join: Whether to select join vs. GO Negotiation automatically
4531  * @join: Whether to join an existing group (as a client) instead of starting
4532  *      Group Owner negotiation; @peer_addr is BSSID in that case
4533  * @auth: Whether to only authorize the connection instead of doing that and
4534  *      initiating Group Owner negotiation
4535  * @go_intent: GO Intent or -1 to use default
4536  * @freq: Frequency for the group or 0 for auto-selection
4537  * @persistent_id: Persistent group credentials to use for forcing GO
4538  *      parameters or -1 to generate new values (SSID/passphrase)
4539  * @pd: Whether to send Provision Discovery prior to GO Negotiation as an
4540  *      interoperability workaround when initiating group formation
4541  * @ht40: Start GO with 40 MHz channel width
4542  * @vht:  Start GO with VHT support
4543  * Returns: 0 or new PIN (if pin was %NULL) on success, -1 on unspecified
4544  *      failure, -2 on failure due to channel not currently available,
4545  *      -3 if forced channel is not supported
4546  */
4547 int wpas_p2p_connect(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
4548                      const char *pin, enum p2p_wps_method wps_method,
4549                      int persistent_group, int auto_join, int join, int auth,
4550                      int go_intent, int freq, int persistent_id, int pd,
4551                      int ht40, int vht)
4552 {
4553         int force_freq = 0, pref_freq = 0;
4554         int ret = 0, res;
4555         enum wpa_driver_if_type iftype;
4556         const u8 *if_addr;
4557         struct wpa_ssid *ssid = NULL;
4558
4559         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
4560                 return -1;
4561
4562         if (persistent_id >= 0) {
4563                 ssid = wpa_config_get_network(wpa_s->conf, persistent_id);
4564                 if (ssid == NULL || ssid->disabled != 2 ||
4565                     ssid->mode != WPAS_MODE_P2P_GO)
4566                         return -1;
4567         }
4568
4569         os_free(wpa_s->global->add_psk);
4570         wpa_s->global->add_psk = NULL;
4571
4572         wpa_s->global->p2p_fail_on_wps_complete = 0;
4573
4574         if (go_intent < 0)
4575                 go_intent = wpa_s->conf->p2p_go_intent;
4576
4577         if (!auth)
4578                 wpa_s->p2p_long_listen = 0;
4579
4580         wpa_s->p2p_wps_method = wps_method;
4581         wpa_s->p2p_persistent_group = !!persistent_group;
4582         wpa_s->p2p_persistent_id = persistent_id;
4583         wpa_s->p2p_go_intent = go_intent;
4584         wpa_s->p2p_connect_freq = freq;
4585         wpa_s->p2p_fallback_to_go_neg = 0;
4586         wpa_s->p2p_pd_before_go_neg = !!pd;
4587         wpa_s->p2p_go_ht40 = !!ht40;
4588         wpa_s->p2p_go_vht = !!vht;
4589
4590         if (pin)
4591                 os_strlcpy(wpa_s->p2p_pin, pin, sizeof(wpa_s->p2p_pin));
4592         else if (wps_method == WPS_PIN_DISPLAY) {
4593                 ret = wps_generate_pin();
4594                 os_snprintf(wpa_s->p2p_pin, sizeof(wpa_s->p2p_pin), "%08d",
4595                             ret);
4596                 wpa_printf(MSG_DEBUG, "P2P: Randomly generated PIN: %s",
4597                            wpa_s->p2p_pin);
4598         } else
4599                 wpa_s->p2p_pin[0] = '\0';
4600
4601         if (join || auto_join) {
4602                 u8 iface_addr[ETH_ALEN], dev_addr[ETH_ALEN];
4603                 if (auth) {
4604                         wpa_printf(MSG_DEBUG, "P2P: Authorize invitation to "
4605                                    "connect a running group from " MACSTR,
4606                                    MAC2STR(peer_addr));
4607                         os_memcpy(wpa_s->p2p_auth_invite, peer_addr, ETH_ALEN);
4608                         return ret;
4609                 }
4610                 os_memcpy(dev_addr, peer_addr, ETH_ALEN);
4611                 if (p2p_get_interface_addr(wpa_s->global->p2p, peer_addr,
4612                                            iface_addr) < 0) {
4613                         os_memcpy(iface_addr, peer_addr, ETH_ALEN);
4614                         p2p_get_dev_addr(wpa_s->global->p2p, peer_addr,
4615                                          dev_addr);
4616                 }
4617                 if (auto_join) {
4618                         os_get_reltime(&wpa_s->p2p_auto_started);
4619                         wpa_printf(MSG_DEBUG, "P2P: Auto join started at "
4620                                    "%ld.%06ld",
4621                                    wpa_s->p2p_auto_started.sec,
4622                                    wpa_s->p2p_auto_started.usec);
4623                 }
4624                 wpa_s->user_initiated_pd = 1;
4625                 if (wpas_p2p_join(wpa_s, iface_addr, dev_addr, wps_method,
4626                                   auto_join, NULL, 0) < 0)
4627                         return -1;
4628                 return ret;
4629         }
4630
4631         res = wpas_p2p_setup_freqs(wpa_s, freq, &force_freq, &pref_freq,
4632                                    go_intent == 15);
4633         if (res)
4634                 return res;
4635         wpas_p2p_set_own_freq_preference(wpa_s,
4636                                          force_freq ? force_freq : pref_freq);
4637
4638         wpa_s->create_p2p_iface = wpas_p2p_create_iface(wpa_s);
4639
4640         if (wpa_s->create_p2p_iface) {
4641                 /* Prepare to add a new interface for the group */
4642                 iftype = WPA_IF_P2P_GROUP;
4643                 if (go_intent == 15)
4644                         iftype = WPA_IF_P2P_GO;
4645                 if (wpas_p2p_add_group_interface(wpa_s, iftype) < 0) {
4646                         wpa_printf(MSG_ERROR, "P2P: Failed to allocate a new "
4647                                    "interface for the group");
4648                         return -1;
4649                 }
4650
4651                 if_addr = wpa_s->pending_interface_addr;
4652         } else
4653                 if_addr = wpa_s->own_addr;
4654
4655         if (auth) {
4656                 if (wpas_p2p_auth_go_neg(wpa_s, peer_addr, wps_method,
4657                                          go_intent, if_addr,
4658                                          force_freq, persistent_group, ssid,
4659                                          pref_freq) < 0)
4660                         return -1;
4661                 return ret;
4662         }
4663
4664         if (wpas_p2p_start_go_neg(wpa_s, peer_addr, wps_method,
4665                                   go_intent, if_addr, force_freq,
4666                                   persistent_group, ssid, pref_freq) < 0) {
4667                 if (wpa_s->create_p2p_iface)
4668                         wpas_p2p_remove_pending_group_interface(wpa_s);
4669                 return -1;
4670         }
4671         return ret;
4672 }
4673
4674
4675 /**
4676  * wpas_p2p_remain_on_channel_cb - Indication of remain-on-channel start
4677  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
4678  * @freq: Frequency of the channel in MHz
4679  * @duration: Duration of the stay on the channel in milliseconds
4680  *
4681  * This callback is called when the driver indicates that it has started the
4682  * requested remain-on-channel duration.
4683  */
4684 void wpas_p2p_remain_on_channel_cb(struct wpa_supplicant *wpa_s,
4685                                    unsigned int freq, unsigned int duration)
4686 {
4687         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
4688                 return;
4689         if (wpa_s->off_channel_freq == wpa_s->pending_listen_freq) {
4690                 p2p_listen_cb(wpa_s->global->p2p, wpa_s->pending_listen_freq,
4691                               wpa_s->pending_listen_duration);
4692                 wpa_s->pending_listen_freq = 0;
4693         }
4694 }
4695
4696
4697 static int wpas_p2p_listen_start(struct wpa_supplicant *wpa_s,
4698                                  unsigned int timeout)
4699 {
4700         /* Limit maximum Listen state time based on driver limitation. */
4701         if (timeout > wpa_s->max_remain_on_chan)
4702                 timeout = wpa_s->max_remain_on_chan;
4703
4704         return p2p_listen(wpa_s->global->p2p, timeout);
4705 }
4706
4707
4708 /**
4709  * wpas_p2p_cancel_remain_on_channel_cb - Remain-on-channel timeout
4710  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
4711  * @freq: Frequency of the channel in MHz
4712  *
4713  * This callback is called when the driver indicates that a remain-on-channel
4714  * operation has been completed, i.e., the duration on the requested channel
4715  * has timed out.
4716  */
4717 void wpas_p2p_cancel_remain_on_channel_cb(struct wpa_supplicant *wpa_s,
4718                                           unsigned int freq)
4719 {
4720         wpa_printf(MSG_DEBUG, "P2P: Cancel remain-on-channel callback "
4721                    "(p2p_long_listen=%d ms pending_action_tx=%p)",
4722                    wpa_s->p2p_long_listen, offchannel_pending_action_tx(wpa_s));
4723         wpas_p2p_listen_work_done(wpa_s);
4724         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
4725                 return;
4726         if (p2p_listen_end(wpa_s->global->p2p, freq) > 0)
4727                 return; /* P2P module started a new operation */
4728         if (offchannel_pending_action_tx(wpa_s))
4729                 return;
4730         if (wpa_s->p2p_long_listen > 0)
4731                 wpa_s->p2p_long_listen -= wpa_s->max_remain_on_chan;
4732         if (wpa_s->p2p_long_listen > 0) {
4733                 wpa_printf(MSG_DEBUG, "P2P: Continuing long Listen state");
4734                 wpas_p2p_listen_start(wpa_s, wpa_s->p2p_long_listen);
4735         } else {
4736                 /*
4737                  * When listen duration is over, stop listen & update p2p_state
4738                  * to IDLE.
4739                  */
4740                 p2p_stop_listen(wpa_s->global->p2p);
4741         }
4742 }
4743
4744
4745 /**
4746  * wpas_p2p_group_remove - Remove a P2P group
4747  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
4748  * @ifname: Network interface name of the group interface or "*" to remove all
4749  *      groups
4750  * Returns: 0 on success, -1 on failure
4751  *
4752  * This function is used to remove a P2P group. This can be used to disconnect
4753  * from a group in which the local end is a P2P Client or to end a P2P Group in
4754  * case the local end is the Group Owner. If a virtual network interface was
4755  * created for this group, that interface will be removed. Otherwise, only the
4756  * configured P2P group network will be removed from the interface.
4757  */
4758 int wpas_p2p_group_remove(struct wpa_supplicant *wpa_s, const char *ifname)
4759 {
4760         struct wpa_global *global = wpa_s->global;
4761
4762         if (os_strcmp(ifname, "*") == 0) {
4763                 struct wpa_supplicant *prev;
4764                 wpa_s = global->ifaces;
4765                 while (wpa_s) {
4766                         prev = wpa_s;
4767                         wpa_s = wpa_s->next;
4768                         if (prev->p2p_group_interface !=
4769                             NOT_P2P_GROUP_INTERFACE ||
4770                             (prev->current_ssid &&
4771                              prev->current_ssid->p2p_group))
4772                                 wpas_p2p_disconnect(prev);
4773                 }
4774                 return 0;
4775         }
4776
4777         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
4778                 if (os_strcmp(wpa_s->ifname, ifname) == 0)
4779                         break;
4780         }
4781
4782         return wpas_p2p_disconnect(wpa_s);
4783 }
4784
4785
4786 static int wpas_p2p_select_go_freq(struct wpa_supplicant *wpa_s, int freq)
4787 {
4788         unsigned int r;
4789
4790         if (freq == 2) {
4791                 wpa_printf(MSG_DEBUG, "P2P: Request to start GO on 2.4 GHz "
4792                            "band");
4793                 if (wpa_s->best_24_freq > 0 &&
4794                     p2p_supported_freq_go(wpa_s->global->p2p,
4795                                           wpa_s->best_24_freq)) {
4796                         freq = wpa_s->best_24_freq;
4797                         wpa_printf(MSG_DEBUG, "P2P: Use best 2.4 GHz band "
4798                                    "channel: %d MHz", freq);
4799                 } else {
4800                         os_get_random((u8 *) &r, sizeof(r));
4801                         freq = 2412 + (r % 3) * 25;
4802                         wpa_printf(MSG_DEBUG, "P2P: Use random 2.4 GHz band "
4803                                    "channel: %d MHz", freq);
4804                 }
4805         }
4806
4807         if (freq == 5) {
4808                 wpa_printf(MSG_DEBUG, "P2P: Request to start GO on 5 GHz "
4809                            "band");
4810                 if (wpa_s->best_5_freq > 0 &&
4811                     p2p_supported_freq_go(wpa_s->global->p2p,
4812                                        wpa_s->best_5_freq)) {
4813                         freq = wpa_s->best_5_freq;
4814                         wpa_printf(MSG_DEBUG, "P2P: Use best 5 GHz band "
4815                                    "channel: %d MHz", freq);
4816                 } else {
4817                         os_get_random((u8 *) &r, sizeof(r));
4818                         freq = 5180 + (r % 4) * 20;
4819                         if (!p2p_supported_freq_go(wpa_s->global->p2p, freq)) {
4820                                 wpa_printf(MSG_DEBUG, "P2P: Could not select "
4821                                            "5 GHz channel for P2P group");
4822                                 return -1;
4823                         }
4824                         wpa_printf(MSG_DEBUG, "P2P: Use random 5 GHz band "
4825                                    "channel: %d MHz", freq);
4826                 }
4827         }
4828
4829         if (freq > 0 && !p2p_supported_freq_go(wpa_s->global->p2p, freq)) {
4830                 wpa_printf(MSG_DEBUG, "P2P: The forced channel for GO "
4831                            "(%u MHz) is not supported for P2P uses",
4832                            freq);
4833                 return -1;
4834         }
4835
4836         return freq;
4837 }
4838
4839
4840 static int wpas_p2p_init_go_params(struct wpa_supplicant *wpa_s,
4841                                    struct p2p_go_neg_results *params,
4842                                    int freq, int ht40, int vht,
4843                                    const struct p2p_channels *channels)
4844 {
4845         int res, *freqs;
4846         unsigned int pref_freq;
4847         unsigned int num, i;
4848
4849         os_memset(params, 0, sizeof(*params));
4850         params->role_go = 1;
4851         params->ht40 = ht40;
4852         params->vht = vht;
4853         if (freq) {
4854                 if (!freq_included(channels, freq)) {
4855                         wpa_printf(MSG_DEBUG, "P2P: Forced GO freq %d MHz not "
4856                                    "accepted", freq);
4857                         return -1;
4858                 }
4859                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on forced "
4860                            "frequency %d MHz", freq);
4861                 params->freq = freq;
4862         } else if (wpa_s->conf->p2p_oper_reg_class == 81 &&
4863                    wpa_s->conf->p2p_oper_channel >= 1 &&
4864                    wpa_s->conf->p2p_oper_channel <= 11 &&
4865                    freq_included(channels,
4866                                  2407 + 5 * wpa_s->conf->p2p_oper_channel)) {
4867                 params->freq = 2407 + 5 * wpa_s->conf->p2p_oper_channel;
4868                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on configured "
4869                            "frequency %d MHz", params->freq);
4870         } else if ((wpa_s->conf->p2p_oper_reg_class == 115 ||
4871                     wpa_s->conf->p2p_oper_reg_class == 116 ||
4872                     wpa_s->conf->p2p_oper_reg_class == 117 ||
4873                     wpa_s->conf->p2p_oper_reg_class == 124 ||
4874                     wpa_s->conf->p2p_oper_reg_class == 126 ||
4875                     wpa_s->conf->p2p_oper_reg_class == 127) &&
4876                    freq_included(channels,
4877                                  5000 + 5 * wpa_s->conf->p2p_oper_channel)) {
4878                 params->freq = 5000 + 5 * wpa_s->conf->p2p_oper_channel;
4879                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on configured "
4880                            "frequency %d MHz", params->freq);
4881         } else if (wpa_s->conf->p2p_oper_channel == 0 &&
4882                    wpa_s->best_overall_freq > 0 &&
4883                    p2p_supported_freq_go(wpa_s->global->p2p,
4884                                          wpa_s->best_overall_freq) &&
4885                    freq_included(channels, wpa_s->best_overall_freq)) {
4886                 params->freq = wpa_s->best_overall_freq;
4887                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on best overall "
4888                            "channel %d MHz", params->freq);
4889         } else if (wpa_s->conf->p2p_oper_channel == 0 &&
4890                    wpa_s->best_24_freq > 0 &&
4891                    p2p_supported_freq_go(wpa_s->global->p2p,
4892                                          wpa_s->best_24_freq) &&
4893                    freq_included(channels, wpa_s->best_24_freq)) {
4894                 params->freq = wpa_s->best_24_freq;
4895                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on best 2.4 GHz "
4896                            "channel %d MHz", params->freq);
4897         } else if (wpa_s->conf->p2p_oper_channel == 0 &&
4898                    wpa_s->best_5_freq > 0 &&
4899                    p2p_supported_freq_go(wpa_s->global->p2p,
4900                                          wpa_s->best_5_freq) &&
4901                    freq_included(channels, wpa_s->best_5_freq)) {
4902                 params->freq = wpa_s->best_5_freq;
4903                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq based on best 5 GHz "
4904                            "channel %d MHz", params->freq);
4905         } else if ((pref_freq = p2p_get_pref_freq(wpa_s->global->p2p,
4906                                                   channels))) {
4907                 params->freq = pref_freq;
4908                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq %d MHz from preferred "
4909                            "channels", params->freq);
4910         } else {
4911                 int chan;
4912                 for (chan = 0; chan < 11; chan++) {
4913                         params->freq = 2412 + chan * 5;
4914                         if (!wpas_p2p_disallowed_freq(wpa_s->global,
4915                                                       params->freq) &&
4916                             freq_included(channels, params->freq))
4917                                 break;
4918                 }
4919                 if (chan == 11) {
4920                         wpa_printf(MSG_DEBUG, "P2P: No 2.4 GHz channel "
4921                                    "allowed");
4922                         return -1;
4923                 }
4924                 wpa_printf(MSG_DEBUG, "P2P: Set GO freq %d MHz (no preference "
4925                            "known)", params->freq);
4926         }
4927
4928         freqs = os_calloc(wpa_s->num_multichan_concurrent, sizeof(int));
4929         if (!freqs)
4930                 return -1;
4931
4932         res = wpas_p2p_valid_oper_freqs(wpa_s, freqs,
4933                                         wpa_s->num_multichan_concurrent);
4934         if (res < 0) {
4935                 os_free(freqs);
4936                 return -1;
4937         }
4938         num = res;
4939
4940         for (i = 0; i < num; i++) {
4941                 if (freq && freqs[i] == freq)
4942                         break;
4943                 if (!freq && freq_included(channels, freqs[i])) {
4944                         wpa_printf(MSG_DEBUG, "P2P: Force GO on a channel we are already using (%u MHz)",
4945                                    freqs[i]);
4946                         params->freq = freqs[i];
4947                         break;
4948                 }
4949         }
4950
4951         if (i == num) {
4952                 if (wpas_p2p_num_unused_channels(wpa_s) <= 0) {
4953                         if (freq)
4954                                 wpa_printf(MSG_DEBUG, "P2P: Cannot force GO on freq (%u MHz) as all the channels are in use", freq);
4955                         else
4956                                 wpa_printf(MSG_DEBUG, "P2P: Cannot force GO on any of the channels we are already using");
4957                         os_free(freqs);
4958                         return -1;
4959                 } else if (num == 0) {
4960                         wpa_printf(MSG_DEBUG, "P2P: Use one of the free channels");
4961                 } else {
4962                         wpa_printf(MSG_DEBUG, "P2P: Cannot force GO on any of the channels we are already using. Use one of the free channels");
4963                 }
4964         }
4965
4966         os_free(freqs);
4967         return 0;
4968 }
4969
4970
4971 static struct wpa_supplicant *
4972 wpas_p2p_get_group_iface(struct wpa_supplicant *wpa_s, int addr_allocated,
4973                          int go)
4974 {
4975         struct wpa_supplicant *group_wpa_s;
4976
4977         if (!wpas_p2p_create_iface(wpa_s)) {
4978                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Use same interface for group "
4979                         "operations");
4980                 wpa_s->p2p_first_connection_timeout = 0;
4981                 return wpa_s;
4982         }
4983
4984         if (wpas_p2p_add_group_interface(wpa_s, go ? WPA_IF_P2P_GO :
4985                                          WPA_IF_P2P_CLIENT) < 0) {
4986                 wpa_msg_global(wpa_s, MSG_ERROR,
4987                                "P2P: Failed to add group interface");
4988                 return NULL;
4989         }
4990         group_wpa_s = wpas_p2p_init_group_interface(wpa_s, go);
4991         if (group_wpa_s == NULL) {
4992                 wpa_msg_global(wpa_s, MSG_ERROR,
4993                                "P2P: Failed to initialize group interface");
4994                 wpas_p2p_remove_pending_group_interface(wpa_s);
4995                 return NULL;
4996         }
4997
4998         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Use separate group interface %s",
4999                 group_wpa_s->ifname);
5000         group_wpa_s->p2p_first_connection_timeout = 0;
5001         return group_wpa_s;
5002 }
5003
5004
5005 /**
5006  * wpas_p2p_group_add - Add a new P2P group with local end as Group Owner
5007  * @wpa_s: Pointer to wpa_supplicant data from wpa_supplicant_add_iface()
5008  * @persistent_group: Whether to create a persistent group
5009  * @freq: Frequency for the group or 0 to indicate no hardcoding
5010  * @ht40: Start GO with 40 MHz channel width
5011  * @vht:  Start GO with VHT support
5012  * Returns: 0 on success, -1 on failure
5013  *
5014  * This function creates a new P2P group with the local end as the Group Owner,
5015  * i.e., without using Group Owner Negotiation.
5016  */
5017 int wpas_p2p_group_add(struct wpa_supplicant *wpa_s, int persistent_group,
5018                        int freq, int ht40, int vht)
5019 {
5020         struct p2p_go_neg_results params;
5021
5022         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5023                 return -1;
5024
5025         os_free(wpa_s->global->add_psk);
5026         wpa_s->global->add_psk = NULL;
5027
5028         /* Make sure we are not running find during connection establishment */
5029         wpa_printf(MSG_DEBUG, "P2P: Stop any on-going P2P FIND");
5030         wpas_p2p_stop_find_oper(wpa_s);
5031
5032         freq = wpas_p2p_select_go_freq(wpa_s, freq);
5033         if (freq < 0)
5034                 return -1;
5035
5036         if (wpas_p2p_init_go_params(wpa_s, &params, freq, ht40, vht, NULL))
5037                 return -1;
5038         if (params.freq &&
5039             !p2p_supported_freq_go(wpa_s->global->p2p, params.freq)) {
5040                 wpa_printf(MSG_DEBUG, "P2P: The selected channel for GO "
5041                            "(%u MHz) is not supported for P2P uses",
5042                            params.freq);
5043                 return -1;
5044         }
5045         p2p_go_params(wpa_s->global->p2p, &params);
5046         params.persistent_group = persistent_group;
5047
5048         wpa_s = wpas_p2p_get_group_iface(wpa_s, 0, 1);
5049         if (wpa_s == NULL)
5050                 return -1;
5051         wpas_start_wps_go(wpa_s, &params, 0);
5052
5053         return 0;
5054 }
5055
5056
5057 static int wpas_start_p2p_client(struct wpa_supplicant *wpa_s,
5058                                  struct wpa_ssid *params, int addr_allocated)
5059 {
5060         struct wpa_ssid *ssid;
5061
5062         wpa_s = wpas_p2p_get_group_iface(wpa_s, addr_allocated, 0);
5063         if (wpa_s == NULL)
5064                 return -1;
5065         wpa_s->p2p_last_4way_hs_fail = NULL;
5066
5067         wpa_supplicant_ap_deinit(wpa_s);
5068
5069         ssid = wpa_config_add_network(wpa_s->conf);
5070         if (ssid == NULL)
5071                 return -1;
5072         wpa_config_set_network_defaults(ssid);
5073         ssid->temporary = 1;
5074         ssid->proto = WPA_PROTO_RSN;
5075         ssid->pairwise_cipher = WPA_CIPHER_CCMP;
5076         ssid->group_cipher = WPA_CIPHER_CCMP;
5077         ssid->key_mgmt = WPA_KEY_MGMT_PSK;
5078         ssid->ssid = os_malloc(params->ssid_len);
5079         if (ssid->ssid == NULL) {
5080                 wpa_config_remove_network(wpa_s->conf, ssid->id);
5081                 return -1;
5082         }
5083         os_memcpy(ssid->ssid, params->ssid, params->ssid_len);
5084         ssid->ssid_len = params->ssid_len;
5085         ssid->p2p_group = 1;
5086         ssid->export_keys = 1;
5087         if (params->psk_set) {
5088                 os_memcpy(ssid->psk, params->psk, 32);
5089                 ssid->psk_set = 1;
5090         }
5091         if (params->passphrase)
5092                 ssid->passphrase = os_strdup(params->passphrase);
5093
5094         wpa_s->show_group_started = 1;
5095
5096         wpa_supplicant_select_network(wpa_s, ssid);
5097
5098         return 0;
5099 }
5100
5101
5102 int wpas_p2p_group_add_persistent(struct wpa_supplicant *wpa_s,
5103                                   struct wpa_ssid *ssid, int addr_allocated,
5104                                   int freq, int ht40, int vht,
5105                                   const struct p2p_channels *channels,
5106                                   int connection_timeout)
5107 {
5108         struct p2p_go_neg_results params;
5109         int go = 0;
5110
5111         if (ssid->disabled != 2 || ssid->ssid == NULL)
5112                 return -1;
5113
5114         if (wpas_get_p2p_group(wpa_s, ssid->ssid, ssid->ssid_len, &go) &&
5115             go == (ssid->mode == WPAS_MODE_P2P_GO)) {
5116                 wpa_printf(MSG_DEBUG, "P2P: Requested persistent group is "
5117                            "already running");
5118                 return 0;
5119         }
5120
5121         os_free(wpa_s->global->add_psk);
5122         wpa_s->global->add_psk = NULL;
5123
5124         /* Make sure we are not running find during connection establishment */
5125         wpas_p2p_stop_find_oper(wpa_s);
5126
5127         wpa_s->p2p_fallback_to_go_neg = 0;
5128
5129         if (ssid->mode == WPAS_MODE_INFRA)
5130                 return wpas_start_p2p_client(wpa_s, ssid, addr_allocated);
5131
5132         if (ssid->mode != WPAS_MODE_P2P_GO)
5133                 return -1;
5134
5135         freq = wpas_p2p_select_go_freq(wpa_s, freq);
5136         if (freq < 0)
5137                 return -1;
5138
5139         if (wpas_p2p_init_go_params(wpa_s, &params, freq, ht40, vht, channels))
5140                 return -1;
5141
5142         params.role_go = 1;
5143         params.psk_set = ssid->psk_set;
5144         if (params.psk_set)
5145                 os_memcpy(params.psk, ssid->psk, sizeof(params.psk));
5146         if (ssid->passphrase) {
5147                 if (os_strlen(ssid->passphrase) >= sizeof(params.passphrase)) {
5148                         wpa_printf(MSG_ERROR, "P2P: Invalid passphrase in "
5149                                    "persistent group");
5150                         return -1;
5151                 }
5152                 os_strlcpy(params.passphrase, ssid->passphrase,
5153                            sizeof(params.passphrase));
5154         }
5155         os_memcpy(params.ssid, ssid->ssid, ssid->ssid_len);
5156         params.ssid_len = ssid->ssid_len;
5157         params.persistent_group = 1;
5158
5159         wpa_s = wpas_p2p_get_group_iface(wpa_s, addr_allocated, 1);
5160         if (wpa_s == NULL)
5161                 return -1;
5162
5163         wpa_s->p2p_first_connection_timeout = connection_timeout;
5164         wpas_start_wps_go(wpa_s, &params, 0);
5165
5166         return 0;
5167 }
5168
5169
5170 static void wpas_p2p_ie_update(void *ctx, struct wpabuf *beacon_ies,
5171                                struct wpabuf *proberesp_ies)
5172 {
5173         struct wpa_supplicant *wpa_s = ctx;
5174         if (wpa_s->ap_iface) {
5175                 struct hostapd_data *hapd = wpa_s->ap_iface->bss[0];
5176                 if (!(hapd->conf->p2p & P2P_GROUP_OWNER)) {
5177                         wpabuf_free(beacon_ies);
5178                         wpabuf_free(proberesp_ies);
5179                         return;
5180                 }
5181                 if (beacon_ies) {
5182                         wpabuf_free(hapd->p2p_beacon_ie);
5183                         hapd->p2p_beacon_ie = beacon_ies;
5184                 }
5185                 wpabuf_free(hapd->p2p_probe_resp_ie);
5186                 hapd->p2p_probe_resp_ie = proberesp_ies;
5187         } else {
5188                 wpabuf_free(beacon_ies);
5189                 wpabuf_free(proberesp_ies);
5190         }
5191         wpa_supplicant_ap_update_beacon(wpa_s);
5192 }
5193
5194
5195 static void wpas_p2p_idle_update(void *ctx, int idle)
5196 {
5197         struct wpa_supplicant *wpa_s = ctx;
5198         if (!wpa_s->ap_iface)
5199                 return;
5200         wpa_printf(MSG_DEBUG, "P2P: GO - group %sidle", idle ? "" : "not ");
5201         if (idle) {
5202                 if (wpa_s->global->p2p_fail_on_wps_complete &&
5203                     wpa_s->p2p_in_provisioning) {
5204                         wpas_p2p_grpform_fail_after_wps(wpa_s);
5205                         return;
5206                 }
5207                 wpas_p2p_set_group_idle_timeout(wpa_s);
5208         } else
5209                 eloop_cancel_timeout(wpas_p2p_group_idle_timeout, wpa_s, NULL);
5210 }
5211
5212
5213 struct p2p_group * wpas_p2p_group_init(struct wpa_supplicant *wpa_s,
5214                                        struct wpa_ssid *ssid)
5215 {
5216         struct p2p_group *group;
5217         struct p2p_group_config *cfg;
5218
5219         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5220                 return NULL;
5221
5222         cfg = os_zalloc(sizeof(*cfg));
5223         if (cfg == NULL)
5224                 return NULL;
5225
5226         if (ssid->p2p_persistent_group && wpa_s->conf->persistent_reconnect)
5227                 cfg->persistent_group = 2;
5228         else if (ssid->p2p_persistent_group)
5229                 cfg->persistent_group = 1;
5230         os_memcpy(cfg->interface_addr, wpa_s->own_addr, ETH_ALEN);
5231         if (wpa_s->max_stations &&
5232             wpa_s->max_stations < wpa_s->conf->max_num_sta)
5233                 cfg->max_clients = wpa_s->max_stations;
5234         else
5235                 cfg->max_clients = wpa_s->conf->max_num_sta;
5236         os_memcpy(cfg->ssid, ssid->ssid, ssid->ssid_len);
5237         cfg->ssid_len = ssid->ssid_len;
5238         cfg->freq = ssid->frequency;
5239         cfg->cb_ctx = wpa_s;
5240         cfg->ie_update = wpas_p2p_ie_update;
5241         cfg->idle_update = wpas_p2p_idle_update;
5242
5243         group = p2p_group_init(wpa_s->global->p2p, cfg);
5244         if (group == NULL)
5245                 os_free(cfg);
5246         if (ssid->mode != WPAS_MODE_P2P_GROUP_FORMATION)
5247                 p2p_group_notif_formation_done(group);
5248         wpa_s->p2p_group = group;
5249         return group;
5250 }
5251
5252
5253 void wpas_p2p_wps_success(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
5254                           int registrar)
5255 {
5256         struct wpa_ssid *ssid = wpa_s->current_ssid;
5257
5258         if (!wpa_s->p2p_in_provisioning) {
5259                 wpa_printf(MSG_DEBUG, "P2P: Ignore WPS success event - P2P "
5260                            "provisioning not in progress");
5261                 return;
5262         }
5263
5264         if (ssid && ssid->mode == WPAS_MODE_INFRA) {
5265                 u8 go_dev_addr[ETH_ALEN];
5266                 os_memcpy(go_dev_addr, wpa_s->bssid, ETH_ALEN);
5267                 wpas_p2p_persistent_group(wpa_s, go_dev_addr, ssid->ssid,
5268                                           ssid->ssid_len);
5269                 /* Clear any stored provisioning info */
5270                 p2p_clear_provisioning_info(wpa_s->global->p2p, go_dev_addr);
5271         }
5272
5273         eloop_cancel_timeout(wpas_p2p_group_formation_timeout, wpa_s->parent,
5274                              NULL);
5275         wpa_s->p2p_go_group_formation_completed = 1;
5276         if (ssid && ssid->mode == WPAS_MODE_INFRA) {
5277                 /*
5278                  * Use a separate timeout for initial data connection to
5279                  * complete to allow the group to be removed automatically if
5280                  * something goes wrong in this step before the P2P group idle
5281                  * timeout mechanism is taken into use.
5282                  */
5283                 wpa_dbg(wpa_s, MSG_DEBUG,
5284                         "P2P: Re-start group formation timeout (%d seconds) as client for initial connection",
5285                         P2P_MAX_INITIAL_CONN_WAIT);
5286                 eloop_register_timeout(P2P_MAX_INITIAL_CONN_WAIT, 0,
5287                                        wpas_p2p_group_formation_timeout,
5288                                        wpa_s->parent, NULL);
5289         } else if (ssid) {
5290                 /*
5291                  * Use a separate timeout for initial data connection to
5292                  * complete to allow the group to be removed automatically if
5293                  * the client does not complete data connection successfully.
5294                  */
5295                 wpa_dbg(wpa_s, MSG_DEBUG,
5296                         "P2P: Re-start group formation timeout (%d seconds) as GO for initial connection",
5297                         P2P_MAX_INITIAL_CONN_WAIT_GO);
5298                 eloop_register_timeout(P2P_MAX_INITIAL_CONN_WAIT_GO, 0,
5299                                        wpas_p2p_group_formation_timeout,
5300                                        wpa_s->parent, NULL);
5301                 /*
5302                  * Complete group formation on first successful data connection
5303                  */
5304                 wpa_s->p2p_go_group_formation_completed = 0;
5305         }
5306         if (wpa_s->global->p2p)
5307                 p2p_wps_success_cb(wpa_s->global->p2p, peer_addr);
5308         wpas_group_formation_completed(wpa_s, 1);
5309 }
5310
5311
5312 void wpas_p2p_wps_failed(struct wpa_supplicant *wpa_s,
5313                          struct wps_event_fail *fail)
5314 {
5315         if (!wpa_s->p2p_in_provisioning) {
5316                 wpa_printf(MSG_DEBUG, "P2P: Ignore WPS fail event - P2P "
5317                            "provisioning not in progress");
5318                 return;
5319         }
5320
5321         if (wpa_s->go_params) {
5322                 p2p_clear_provisioning_info(
5323                         wpa_s->global->p2p,
5324                         wpa_s->go_params->peer_device_addr);
5325         }
5326
5327         wpas_notify_p2p_wps_failed(wpa_s, fail);
5328
5329         if (wpa_s == wpa_s->global->p2p_group_formation) {
5330                 /*
5331                  * Allow some time for the failed WPS negotiation exchange to
5332                  * complete, but remove the group since group formation cannot
5333                  * succeed after provisioning failure.
5334                  */
5335                 wpa_printf(MSG_DEBUG, "P2P: WPS step failed during group formation - reject connection from timeout");
5336                 wpa_s->global->p2p_fail_on_wps_complete = 1;
5337                 eloop_deplete_timeout(0, 50000,
5338                                       wpas_p2p_group_formation_timeout,
5339                                       wpa_s->parent, NULL);
5340         }
5341 }
5342
5343
5344 int wpas_p2p_wps_eapol_cb(struct wpa_supplicant *wpa_s)
5345 {
5346         if (!wpa_s->global->p2p_fail_on_wps_complete ||
5347             !wpa_s->p2p_in_provisioning)
5348                 return 0;
5349
5350         wpas_p2p_grpform_fail_after_wps(wpa_s);
5351
5352         return 1;
5353 }
5354
5355
5356 int wpas_p2p_prov_disc(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
5357                        const char *config_method,
5358                        enum wpas_p2p_prov_disc_use use)
5359 {
5360         u16 config_methods;
5361
5362         wpa_s->p2p_fallback_to_go_neg = 0;
5363         wpa_s->pending_pd_use = NORMAL_PD;
5364         if (os_strncmp(config_method, "display", 7) == 0)
5365                 config_methods = WPS_CONFIG_DISPLAY;
5366         else if (os_strncmp(config_method, "keypad", 6) == 0)
5367                 config_methods = WPS_CONFIG_KEYPAD;
5368         else if (os_strncmp(config_method, "pbc", 3) == 0 ||
5369                  os_strncmp(config_method, "pushbutton", 10) == 0)
5370                 config_methods = WPS_CONFIG_PUSHBUTTON;
5371         else {
5372                 wpa_printf(MSG_DEBUG, "P2P: Unknown config method");
5373                 return -1;
5374         }
5375
5376         if (use == WPAS_P2P_PD_AUTO) {
5377                 os_memcpy(wpa_s->pending_join_dev_addr, peer_addr, ETH_ALEN);
5378                 wpa_s->pending_pd_config_methods = config_methods;
5379                 wpa_s->p2p_auto_pd = 1;
5380                 wpa_s->p2p_auto_join = 0;
5381                 wpa_s->pending_pd_before_join = 0;
5382                 wpa_s->auto_pd_scan_retry = 0;
5383                 wpas_p2p_stop_find(wpa_s);
5384                 wpa_s->p2p_join_scan_count = 0;
5385                 os_get_reltime(&wpa_s->p2p_auto_started);
5386                 wpa_printf(MSG_DEBUG, "P2P: Auto PD started at %ld.%06ld",
5387                            wpa_s->p2p_auto_started.sec,
5388                            wpa_s->p2p_auto_started.usec);
5389                 wpas_p2p_join_scan(wpa_s, NULL);
5390                 return 0;
5391         }
5392
5393         if (wpa_s->global->p2p == NULL || wpa_s->global->p2p_disabled)
5394                 return -1;
5395
5396         return p2p_prov_disc_req(wpa_s->global->p2p, peer_addr,
5397                                  config_methods, use == WPAS_P2P_PD_FOR_JOIN,
5398                                  0, 1);
5399 }
5400
5401
5402 int wpas_p2p_scan_result_text(const u8 *ies, size_t ies_len, char *buf,
5403                               char *end)
5404 {
5405         return p2p_scan_result_text(ies, ies_len, buf, end);
5406 }
5407
5408
5409 static void wpas_p2p_clear_pending_action_tx(struct wpa_supplicant *wpa_s)
5410 {
5411         if (!offchannel_pending_action_tx(wpa_s))
5412                 return;
5413
5414         wpa_printf(MSG_DEBUG, "P2P: Drop pending Action TX due to new "
5415                    "operation request");
5416         offchannel_clear_pending_action_tx(wpa_s);
5417 }
5418
5419
5420 int wpas_p2p_find(struct wpa_supplicant *wpa_s, unsigned int timeout,
5421                   enum p2p_discovery_type type,
5422                   unsigned int num_req_dev_types, const u8 *req_dev_types,
5423                   const u8 *dev_id, unsigned int search_delay)
5424 {
5425         wpas_p2p_clear_pending_action_tx(wpa_s);
5426         wpa_s->p2p_long_listen = 0;
5427
5428         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL ||
5429             wpa_s->p2p_in_provisioning)
5430                 return -1;
5431
5432         wpa_supplicant_cancel_sched_scan(wpa_s);
5433
5434         return p2p_find(wpa_s->global->p2p, timeout, type,
5435                         num_req_dev_types, req_dev_types, dev_id,
5436                         search_delay);
5437 }
5438
5439
5440 static int wpas_p2p_stop_find_oper(struct wpa_supplicant *wpa_s)
5441 {
5442         wpas_p2p_clear_pending_action_tx(wpa_s);
5443         wpa_s->p2p_long_listen = 0;
5444         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
5445         eloop_cancel_timeout(wpas_p2p_join_scan, wpa_s, NULL);
5446
5447         if (wpa_s->global->p2p)
5448                 p2p_stop_find(wpa_s->global->p2p);
5449
5450         return 0;
5451 }
5452
5453
5454 void wpas_p2p_stop_find(struct wpa_supplicant *wpa_s)
5455 {
5456         if (wpas_p2p_stop_find_oper(wpa_s) > 0)
5457                 return;
5458         wpas_p2p_remove_pending_group_interface(wpa_s);
5459 }
5460
5461
5462 static void wpas_p2p_long_listen_timeout(void *eloop_ctx, void *timeout_ctx)
5463 {
5464         struct wpa_supplicant *wpa_s = eloop_ctx;
5465         wpa_s->p2p_long_listen = 0;
5466 }
5467
5468
5469 int wpas_p2p_listen(struct wpa_supplicant *wpa_s, unsigned int timeout)
5470 {
5471         int res;
5472
5473         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5474                 return -1;
5475
5476         wpa_supplicant_cancel_sched_scan(wpa_s);
5477         wpas_p2p_clear_pending_action_tx(wpa_s);
5478
5479         if (timeout == 0) {
5480                 /*
5481                  * This is a request for unlimited Listen state. However, at
5482                  * least for now, this is mapped to a Listen state for one
5483                  * hour.
5484                  */
5485                 timeout = 3600;
5486         }
5487         eloop_cancel_timeout(wpas_p2p_long_listen_timeout, wpa_s, NULL);
5488         wpa_s->p2p_long_listen = 0;
5489
5490         /*
5491          * Stop previous find/listen operation to avoid trying to request a new
5492          * remain-on-channel operation while the driver is still running the
5493          * previous one.
5494          */
5495         if (wpa_s->global->p2p)
5496                 p2p_stop_find(wpa_s->global->p2p);
5497
5498         res = wpas_p2p_listen_start(wpa_s, timeout * 1000);
5499         if (res == 0 && timeout * 1000 > wpa_s->max_remain_on_chan) {
5500                 wpa_s->p2p_long_listen = timeout * 1000;
5501                 eloop_register_timeout(timeout, 0,
5502                                        wpas_p2p_long_listen_timeout,
5503                                        wpa_s, NULL);
5504         }
5505
5506         return res;
5507 }
5508
5509
5510 int wpas_p2p_assoc_req_ie(struct wpa_supplicant *wpa_s, struct wpa_bss *bss,
5511                           u8 *buf, size_t len, int p2p_group)
5512 {
5513         struct wpabuf *p2p_ie;
5514         int ret;
5515
5516         if (wpa_s->global->p2p_disabled)
5517                 return -1;
5518         if (wpa_s->global->p2p == NULL)
5519                 return -1;
5520         if (bss == NULL)
5521                 return -1;
5522
5523         p2p_ie = wpa_bss_get_vendor_ie_multi(bss, P2P_IE_VENDOR_TYPE);
5524         ret = p2p_assoc_req_ie(wpa_s->global->p2p, bss->bssid, buf, len,
5525                                p2p_group, p2p_ie);
5526         wpabuf_free(p2p_ie);
5527
5528         return ret;
5529 }
5530
5531
5532 int wpas_p2p_probe_req_rx(struct wpa_supplicant *wpa_s, const u8 *addr,
5533                           const u8 *dst, const u8 *bssid,
5534                           const u8 *ie, size_t ie_len, int ssi_signal)
5535 {
5536         if (wpa_s->global->p2p_disabled)
5537                 return 0;
5538         if (wpa_s->global->p2p == NULL)
5539                 return 0;
5540
5541         switch (p2p_probe_req_rx(wpa_s->global->p2p, addr, dst, bssid,
5542                                  ie, ie_len)) {
5543         case P2P_PREQ_NOT_P2P:
5544                 wpas_notify_preq(wpa_s, addr, dst, bssid, ie, ie_len,
5545                                  ssi_signal);
5546                 /* fall through */
5547         case P2P_PREQ_MALFORMED:
5548         case P2P_PREQ_NOT_LISTEN:
5549         case P2P_PREQ_NOT_PROCESSED:
5550         default: /* make gcc happy */
5551                 return 0;
5552         case P2P_PREQ_PROCESSED:
5553                 return 1;
5554         }
5555 }
5556
5557
5558 void wpas_p2p_rx_action(struct wpa_supplicant *wpa_s, const u8 *da,
5559                         const u8 *sa, const u8 *bssid,
5560                         u8 category, const u8 *data, size_t len, int freq)
5561 {
5562         if (wpa_s->global->p2p_disabled)
5563                 return;
5564         if (wpa_s->global->p2p == NULL)
5565                 return;
5566
5567         p2p_rx_action(wpa_s->global->p2p, da, sa, bssid, category, data, len,
5568                       freq);
5569 }
5570
5571
5572 void wpas_p2p_scan_ie(struct wpa_supplicant *wpa_s, struct wpabuf *ies)
5573 {
5574         if (wpa_s->global->p2p_disabled)
5575                 return;
5576         if (wpa_s->global->p2p == NULL)
5577                 return;
5578
5579         p2p_scan_ie(wpa_s->global->p2p, ies, NULL);
5580 }
5581
5582
5583 void wpas_p2p_group_deinit(struct wpa_supplicant *wpa_s)
5584 {
5585         p2p_group_deinit(wpa_s->p2p_group);
5586         wpa_s->p2p_group = NULL;
5587
5588         wpa_s->ap_configured_cb = NULL;
5589         wpa_s->ap_configured_cb_ctx = NULL;
5590         wpa_s->ap_configured_cb_data = NULL;
5591         wpa_s->connect_without_scan = NULL;
5592 }
5593
5594
5595 int wpas_p2p_reject(struct wpa_supplicant *wpa_s, const u8 *addr)
5596 {
5597         wpa_s->p2p_long_listen = 0;
5598
5599         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5600                 return -1;
5601
5602         return p2p_reject(wpa_s->global->p2p, addr);
5603 }
5604
5605
5606 /* Invite to reinvoke a persistent group */
5607 int wpas_p2p_invite(struct wpa_supplicant *wpa_s, const u8 *peer_addr,
5608                     struct wpa_ssid *ssid, const u8 *go_dev_addr, int freq,
5609                     int ht40, int vht, int pref_freq)
5610 {
5611         enum p2p_invite_role role;
5612         u8 *bssid = NULL;
5613         int force_freq = 0;
5614         int res;
5615         int no_pref_freq_given = pref_freq == 0;
5616
5617         wpa_s->global->p2p_invite_group = NULL;
5618         if (peer_addr)
5619                 os_memcpy(wpa_s->p2p_auth_invite, peer_addr, ETH_ALEN);
5620         else
5621                 os_memset(wpa_s->p2p_auth_invite, 0, ETH_ALEN);
5622
5623         wpa_s->p2p_persistent_go_freq = freq;
5624         wpa_s->p2p_go_ht40 = !!ht40;
5625         if (ssid->mode == WPAS_MODE_P2P_GO) {
5626                 role = P2P_INVITE_ROLE_GO;
5627                 if (peer_addr == NULL) {
5628                         wpa_printf(MSG_DEBUG, "P2P: Missing peer "
5629                                    "address in invitation command");
5630                         return -1;
5631                 }
5632                 if (wpas_p2p_create_iface(wpa_s)) {
5633                         if (wpas_p2p_add_group_interface(wpa_s,
5634                                                          WPA_IF_P2P_GO) < 0) {
5635                                 wpa_printf(MSG_ERROR, "P2P: Failed to "
5636                                            "allocate a new interface for the "
5637                                            "group");
5638                                 return -1;
5639                         }
5640                         bssid = wpa_s->pending_interface_addr;
5641                 } else
5642                         bssid = wpa_s->own_addr;
5643         } else {
5644                 role = P2P_INVITE_ROLE_CLIENT;
5645                 peer_addr = ssid->bssid;
5646         }
5647         wpa_s->pending_invite_ssid_id = ssid->id;
5648
5649         res = wpas_p2p_setup_freqs(wpa_s, freq, &force_freq, &pref_freq,
5650                                    role == P2P_INVITE_ROLE_GO);
5651         if (res)
5652                 return res;
5653
5654         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5655                 return -1;
5656
5657         if (wpa_s->parent->conf->p2p_ignore_shared_freq &&
5658             no_pref_freq_given && pref_freq > 0 &&
5659             wpa_s->num_multichan_concurrent > 1 &&
5660             wpas_p2p_num_unused_channels(wpa_s) > 0) {
5661                 wpa_printf(MSG_DEBUG, "P2P: Ignore own channel preference %d MHz for invitation due to p2p_ignore_shared_freq=1 configuration",
5662                            pref_freq);
5663                 pref_freq = 0;
5664         }
5665
5666         return p2p_invite(wpa_s->global->p2p, peer_addr, role, bssid,
5667                           ssid->ssid, ssid->ssid_len, force_freq, go_dev_addr,
5668                           1, pref_freq);
5669 }
5670
5671
5672 /* Invite to join an active group */
5673 int wpas_p2p_invite_group(struct wpa_supplicant *wpa_s, const char *ifname,
5674                           const u8 *peer_addr, const u8 *go_dev_addr)
5675 {
5676         struct wpa_global *global = wpa_s->global;
5677         enum p2p_invite_role role;
5678         u8 *bssid = NULL;
5679         struct wpa_ssid *ssid;
5680         int persistent;
5681         int freq = 0, force_freq = 0, pref_freq = 0;
5682         int res;
5683
5684         wpa_s->p2p_persistent_go_freq = 0;
5685         wpa_s->p2p_go_ht40 = 0;
5686         wpa_s->p2p_go_vht = 0;
5687
5688         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
5689                 if (os_strcmp(wpa_s->ifname, ifname) == 0)
5690                         break;
5691         }
5692         if (wpa_s == NULL) {
5693                 wpa_printf(MSG_DEBUG, "P2P: Interface '%s' not found", ifname);
5694                 return -1;
5695         }
5696
5697         ssid = wpa_s->current_ssid;
5698         if (ssid == NULL) {
5699                 wpa_printf(MSG_DEBUG, "P2P: No current SSID to use for "
5700                            "invitation");
5701                 return -1;
5702         }
5703
5704         wpa_s->global->p2p_invite_group = wpa_s;
5705         persistent = ssid->p2p_persistent_group &&
5706                 wpas_p2p_get_persistent(wpa_s->parent, peer_addr,
5707                                         ssid->ssid, ssid->ssid_len);
5708
5709         if (ssid->mode == WPAS_MODE_P2P_GO) {
5710                 role = P2P_INVITE_ROLE_ACTIVE_GO;
5711                 bssid = wpa_s->own_addr;
5712                 if (go_dev_addr == NULL)
5713                         go_dev_addr = wpa_s->global->p2p_dev_addr;
5714                 freq = ssid->frequency;
5715         } else {
5716                 role = P2P_INVITE_ROLE_CLIENT;
5717                 if (wpa_s->wpa_state < WPA_ASSOCIATED) {
5718                         wpa_printf(MSG_DEBUG, "P2P: Not associated - cannot "
5719                                    "invite to current group");
5720                         return -1;
5721                 }
5722                 bssid = wpa_s->bssid;
5723                 if (go_dev_addr == NULL &&
5724                     !is_zero_ether_addr(wpa_s->go_dev_addr))
5725                         go_dev_addr = wpa_s->go_dev_addr;
5726                 freq = wpa_s->current_bss ? wpa_s->current_bss->freq :
5727                         (int) wpa_s->assoc_freq;
5728         }
5729         wpa_s->parent->pending_invite_ssid_id = -1;
5730
5731         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5732                 return -1;
5733
5734         res = wpas_p2p_setup_freqs(wpa_s, freq, &force_freq, &pref_freq,
5735                                    role == P2P_INVITE_ROLE_ACTIVE_GO);
5736         if (res)
5737                 return res;
5738         wpas_p2p_set_own_freq_preference(wpa_s, force_freq);
5739
5740         return p2p_invite(wpa_s->global->p2p, peer_addr, role, bssid,
5741                           ssid->ssid, ssid->ssid_len, force_freq,
5742                           go_dev_addr, persistent, pref_freq);
5743 }
5744
5745
5746 void wpas_p2p_completed(struct wpa_supplicant *wpa_s)
5747 {
5748         struct wpa_ssid *ssid = wpa_s->current_ssid;
5749         const char *ssid_txt;
5750         u8 go_dev_addr[ETH_ALEN];
5751         int network_id = -1;
5752         int persistent;
5753         int freq;
5754
5755         if (ssid == NULL || ssid->mode != WPAS_MODE_P2P_GROUP_FORMATION) {
5756                 eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
5757                                      wpa_s->parent, NULL);
5758         }
5759
5760         if (!wpa_s->show_group_started || !ssid)
5761                 return;
5762
5763         wpa_s->show_group_started = 0;
5764
5765         ssid_txt = wpa_ssid_txt(ssid->ssid, ssid->ssid_len);
5766         os_memset(go_dev_addr, 0, ETH_ALEN);
5767         if (ssid->bssid_set)
5768                 os_memcpy(go_dev_addr, ssid->bssid, ETH_ALEN);
5769         persistent = wpas_p2p_persistent_group(wpa_s, go_dev_addr, ssid->ssid,
5770                                                ssid->ssid_len);
5771         os_memcpy(wpa_s->go_dev_addr, go_dev_addr, ETH_ALEN);
5772
5773         if (wpa_s->global->p2p_group_formation == wpa_s)
5774                 wpa_s->global->p2p_group_formation = NULL;
5775
5776         freq = wpa_s->current_bss ? wpa_s->current_bss->freq :
5777                 (int) wpa_s->assoc_freq;
5778         if (ssid->passphrase == NULL && ssid->psk_set) {
5779                 char psk[65];
5780                 wpa_snprintf_hex(psk, sizeof(psk), ssid->psk, 32);
5781                 wpa_msg_global(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
5782                                "%s client ssid=\"%s\" freq=%d psk=%s "
5783                                "go_dev_addr=" MACSTR "%s",
5784                                wpa_s->ifname, ssid_txt, freq, psk,
5785                                MAC2STR(go_dev_addr),
5786                                persistent ? " [PERSISTENT]" : "");
5787         } else {
5788                 wpa_msg_global(wpa_s->parent, MSG_INFO, P2P_EVENT_GROUP_STARTED
5789                                "%s client ssid=\"%s\" freq=%d "
5790                                "passphrase=\"%s\" go_dev_addr=" MACSTR "%s",
5791                                wpa_s->ifname, ssid_txt, freq,
5792                                ssid->passphrase ? ssid->passphrase : "",
5793                                MAC2STR(go_dev_addr),
5794                                persistent ? " [PERSISTENT]" : "");
5795         }
5796
5797         if (persistent)
5798                 network_id = wpas_p2p_store_persistent_group(wpa_s->parent,
5799                                                              ssid, go_dev_addr);
5800         if (network_id < 0)
5801                 network_id = ssid->id;
5802         wpas_notify_p2p_group_started(wpa_s, ssid, network_id, 1);
5803 }
5804
5805
5806 int wpas_p2p_presence_req(struct wpa_supplicant *wpa_s, u32 duration1,
5807                           u32 interval1, u32 duration2, u32 interval2)
5808 {
5809         int ret;
5810
5811         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5812                 return -1;
5813
5814         if (wpa_s->wpa_state < WPA_ASSOCIATED ||
5815             wpa_s->current_ssid == NULL ||
5816             wpa_s->current_ssid->mode != WPAS_MODE_INFRA)
5817                 return -1;
5818
5819         ret = p2p_presence_req(wpa_s->global->p2p, wpa_s->bssid,
5820                                wpa_s->own_addr, wpa_s->assoc_freq,
5821                                duration1, interval1, duration2, interval2);
5822         if (ret == 0)
5823                 wpa_s->waiting_presence_resp = 1;
5824
5825         return ret;
5826 }
5827
5828
5829 int wpas_p2p_ext_listen(struct wpa_supplicant *wpa_s, unsigned int period,
5830                         unsigned int interval)
5831 {
5832         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5833                 return -1;
5834
5835         return p2p_ext_listen(wpa_s->global->p2p, period, interval);
5836 }
5837
5838
5839 static int wpas_p2p_is_client(struct wpa_supplicant *wpa_s)
5840 {
5841         if (wpa_s->current_ssid == NULL) {
5842                 /*
5843                  * current_ssid can be cleared when P2P client interface gets
5844                  * disconnected, so assume this interface was used as P2P
5845                  * client.
5846                  */
5847                 return 1;
5848         }
5849         return wpa_s->current_ssid->p2p_group &&
5850                 wpa_s->current_ssid->mode == WPAS_MODE_INFRA;
5851 }
5852
5853
5854 static void wpas_p2p_group_idle_timeout(void *eloop_ctx, void *timeout_ctx)
5855 {
5856         struct wpa_supplicant *wpa_s = eloop_ctx;
5857
5858         if (wpa_s->conf->p2p_group_idle == 0 && !wpas_p2p_is_client(wpa_s)) {
5859                 wpa_printf(MSG_DEBUG, "P2P: Ignore group idle timeout - "
5860                            "disabled");
5861                 return;
5862         }
5863
5864         wpa_printf(MSG_DEBUG, "P2P: Group idle timeout reached - terminate "
5865                    "group");
5866         wpas_p2p_group_delete(wpa_s, P2P_GROUP_REMOVAL_IDLE_TIMEOUT);
5867 }
5868
5869
5870 static void wpas_p2p_set_group_idle_timeout(struct wpa_supplicant *wpa_s)
5871 {
5872         int timeout;
5873
5874         if (eloop_cancel_timeout(wpas_p2p_group_idle_timeout, wpa_s, NULL) > 0)
5875                 wpa_printf(MSG_DEBUG, "P2P: Cancelled P2P group idle timeout");
5876
5877         if (wpa_s->current_ssid == NULL || !wpa_s->current_ssid->p2p_group)
5878                 return;
5879
5880         timeout = wpa_s->conf->p2p_group_idle;
5881         if (wpa_s->current_ssid->mode == WPAS_MODE_INFRA &&
5882             (timeout == 0 || timeout > P2P_MAX_CLIENT_IDLE))
5883             timeout = P2P_MAX_CLIENT_IDLE;
5884
5885         if (timeout == 0)
5886                 return;
5887
5888         if (timeout < 0) {
5889                 if (wpa_s->current_ssid->mode == WPAS_MODE_INFRA)
5890                         timeout = 0; /* special client mode no-timeout */
5891                 else
5892                         return;
5893         }
5894
5895         if (wpa_s->p2p_in_provisioning) {
5896                 /*
5897                  * Use the normal group formation timeout during the
5898                  * provisioning phase to avoid terminating this process too
5899                  * early due to group idle timeout.
5900                  */
5901                 wpa_printf(MSG_DEBUG, "P2P: Do not use P2P group idle timeout "
5902                            "during provisioning");
5903                 return;
5904         }
5905
5906         if (wpa_s->show_group_started) {
5907                 /*
5908                  * Use the normal group formation timeout between the end of
5909                  * the provisioning phase and completion of 4-way handshake to
5910                  * avoid terminating this process too early due to group idle
5911                  * timeout.
5912                  */
5913                 wpa_printf(MSG_DEBUG, "P2P: Do not use P2P group idle timeout "
5914                            "while waiting for initial 4-way handshake to "
5915                            "complete");
5916                 return;
5917         }
5918
5919         wpa_printf(MSG_DEBUG, "P2P: Set P2P group idle timeout to %u seconds",
5920                    timeout);
5921         eloop_register_timeout(timeout, 0, wpas_p2p_group_idle_timeout,
5922                                wpa_s, NULL);
5923 }
5924
5925
5926 /* Returns 1 if the interface was removed */
5927 int wpas_p2p_deauth_notif(struct wpa_supplicant *wpa_s, const u8 *bssid,
5928                           u16 reason_code, const u8 *ie, size_t ie_len,
5929                           int locally_generated)
5930 {
5931         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5932                 return 0;
5933
5934         if (!locally_generated)
5935                 p2p_deauth_notif(wpa_s->global->p2p, bssid, reason_code, ie,
5936                                  ie_len);
5937
5938         if (reason_code == WLAN_REASON_DEAUTH_LEAVING && !locally_generated &&
5939             wpa_s->current_ssid &&
5940             wpa_s->current_ssid->p2p_group &&
5941             wpa_s->current_ssid->mode == WPAS_MODE_INFRA) {
5942                 wpa_printf(MSG_DEBUG, "P2P: GO indicated that the P2P Group "
5943                            "session is ending");
5944                 if (wpas_p2p_group_delete(wpa_s,
5945                                           P2P_GROUP_REMOVAL_GO_ENDING_SESSION)
5946                     > 0)
5947                         return 1;
5948         }
5949
5950         return 0;
5951 }
5952
5953
5954 void wpas_p2p_disassoc_notif(struct wpa_supplicant *wpa_s, const u8 *bssid,
5955                              u16 reason_code, const u8 *ie, size_t ie_len,
5956                              int locally_generated)
5957 {
5958         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
5959                 return;
5960
5961         if (!locally_generated)
5962                 p2p_disassoc_notif(wpa_s->global->p2p, bssid, reason_code, ie,
5963                                    ie_len);
5964 }
5965
5966
5967 void wpas_p2p_update_config(struct wpa_supplicant *wpa_s)
5968 {
5969         struct p2p_data *p2p = wpa_s->global->p2p;
5970
5971         if (p2p == NULL)
5972                 return;
5973
5974         if (!(wpa_s->drv_flags & WPA_DRIVER_FLAGS_P2P_CAPABLE))
5975                 return;
5976
5977         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_NAME)
5978                 p2p_set_dev_name(p2p, wpa_s->conf->device_name);
5979
5980         if (wpa_s->conf->changed_parameters & CFG_CHANGED_DEVICE_TYPE)
5981                 p2p_set_pri_dev_type(p2p, wpa_s->conf->device_type);
5982
5983         if (wpa_s->wps &&
5984             (wpa_s->conf->changed_parameters & CFG_CHANGED_CONFIG_METHODS))
5985                 p2p_set_config_methods(p2p, wpa_s->wps->config_methods);
5986
5987         if (wpa_s->wps && (wpa_s->conf->changed_parameters & CFG_CHANGED_UUID))
5988                 p2p_set_uuid(p2p, wpa_s->wps->uuid);
5989
5990         if (wpa_s->conf->changed_parameters & CFG_CHANGED_WPS_STRING) {
5991                 p2p_set_manufacturer(p2p, wpa_s->conf->manufacturer);
5992                 p2p_set_model_name(p2p, wpa_s->conf->model_name);
5993                 p2p_set_model_number(p2p, wpa_s->conf->model_number);
5994                 p2p_set_serial_number(p2p, wpa_s->conf->serial_number);
5995         }
5996
5997         if (wpa_s->conf->changed_parameters & CFG_CHANGED_SEC_DEVICE_TYPE)
5998                 p2p_set_sec_dev_types(p2p,
5999                                       (void *) wpa_s->conf->sec_device_type,
6000                                       wpa_s->conf->num_sec_device_types);
6001
6002         if (wpa_s->conf->changed_parameters & CFG_CHANGED_VENDOR_EXTENSION) {
6003                 int i;
6004                 p2p_remove_wps_vendor_extensions(p2p);
6005                 for (i = 0; i < MAX_WPS_VENDOR_EXT; i++) {
6006                         if (wpa_s->conf->wps_vendor_ext[i] == NULL)
6007                                 continue;
6008                         p2p_add_wps_vendor_extension(
6009                                 p2p, wpa_s->conf->wps_vendor_ext[i]);
6010                 }
6011         }
6012
6013         if ((wpa_s->conf->changed_parameters & CFG_CHANGED_COUNTRY) &&
6014             wpa_s->conf->country[0] && wpa_s->conf->country[1]) {
6015                 char country[3];
6016                 country[0] = wpa_s->conf->country[0];
6017                 country[1] = wpa_s->conf->country[1];
6018                 country[2] = 0x04;
6019                 p2p_set_country(p2p, country);
6020         }
6021
6022         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_SSID_POSTFIX) {
6023                 p2p_set_ssid_postfix(p2p, (u8 *) wpa_s->conf->p2p_ssid_postfix,
6024                                      wpa_s->conf->p2p_ssid_postfix ?
6025                                      os_strlen(wpa_s->conf->p2p_ssid_postfix) :
6026                                      0);
6027         }
6028
6029         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_INTRA_BSS)
6030                 p2p_set_intra_bss_dist(p2p, wpa_s->conf->p2p_intra_bss);
6031
6032         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_LISTEN_CHANNEL) {
6033                 u8 reg_class, channel;
6034                 int ret;
6035                 unsigned int r;
6036                 if (wpa_s->conf->p2p_listen_reg_class &&
6037                     wpa_s->conf->p2p_listen_channel) {
6038                         reg_class = wpa_s->conf->p2p_listen_reg_class;
6039                         channel = wpa_s->conf->p2p_listen_channel;
6040                 } else {
6041                         reg_class = 81;
6042                         /*
6043                          * Pick one of the social channels randomly as the
6044                          * listen channel.
6045                          */
6046                         os_get_random((u8 *) &r, sizeof(r));
6047                         channel = 1 + (r % 3) * 5;
6048                 }
6049                 ret = p2p_set_listen_channel(p2p, reg_class, channel);
6050                 if (ret)
6051                         wpa_printf(MSG_ERROR, "P2P: Own listen channel update "
6052                                    "failed: %d", ret);
6053         }
6054         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_OPER_CHANNEL) {
6055                 u8 op_reg_class, op_channel, cfg_op_channel;
6056                 int ret = 0;
6057                 unsigned int r;
6058                 if (wpa_s->conf->p2p_oper_reg_class &&
6059                     wpa_s->conf->p2p_oper_channel) {
6060                         op_reg_class = wpa_s->conf->p2p_oper_reg_class;
6061                         op_channel = wpa_s->conf->p2p_oper_channel;
6062                         cfg_op_channel = 1;
6063                 } else {
6064                         op_reg_class = 81;
6065                         /*
6066                          * Use random operation channel from (1, 6, 11)
6067                          *if no other preference is indicated.
6068                          */
6069                         os_get_random((u8 *) &r, sizeof(r));
6070                         op_channel = 1 + (r % 3) * 5;
6071                         cfg_op_channel = 0;
6072                 }
6073                 ret = p2p_set_oper_channel(p2p, op_reg_class, op_channel,
6074                                            cfg_op_channel);
6075                 if (ret)
6076                         wpa_printf(MSG_ERROR, "P2P: Own oper channel update "
6077                                    "failed: %d", ret);
6078         }
6079
6080         if (wpa_s->conf->changed_parameters & CFG_CHANGED_P2P_PREF_CHAN) {
6081                 if (p2p_set_pref_chan(p2p, wpa_s->conf->num_p2p_pref_chan,
6082                                       wpa_s->conf->p2p_pref_chan) < 0) {
6083                         wpa_printf(MSG_ERROR, "P2P: Preferred channel list "
6084                                    "update failed");
6085                 }
6086
6087                 if (p2p_set_no_go_freq(p2p, &wpa_s->conf->p2p_no_go_freq) < 0) {
6088                         wpa_printf(MSG_ERROR, "P2P: No GO channel list "
6089                                    "update failed");
6090                 }
6091         }
6092 }
6093
6094
6095 int wpas_p2p_set_noa(struct wpa_supplicant *wpa_s, u8 count, int start,
6096                      int duration)
6097 {
6098         if (!wpa_s->ap_iface)
6099                 return -1;
6100         return hostapd_p2p_set_noa(wpa_s->ap_iface->bss[0], count, start,
6101                                    duration);
6102 }
6103
6104
6105 int wpas_p2p_set_cross_connect(struct wpa_supplicant *wpa_s, int enabled)
6106 {
6107         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
6108                 return -1;
6109
6110         wpa_s->global->cross_connection = enabled;
6111         p2p_set_cross_connect(wpa_s->global->p2p, enabled);
6112
6113         if (!enabled) {
6114                 struct wpa_supplicant *iface;
6115
6116                 for (iface = wpa_s->global->ifaces; iface; iface = iface->next)
6117                 {
6118                         if (iface->cross_connect_enabled == 0)
6119                                 continue;
6120
6121                         iface->cross_connect_enabled = 0;
6122                         iface->cross_connect_in_use = 0;
6123                         wpa_msg_global(iface->parent, MSG_INFO,
6124                                        P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
6125                                        iface->ifname,
6126                                        iface->cross_connect_uplink);
6127                 }
6128         }
6129
6130         return 0;
6131 }
6132
6133
6134 static void wpas_p2p_enable_cross_connect(struct wpa_supplicant *uplink)
6135 {
6136         struct wpa_supplicant *iface;
6137
6138         if (!uplink->global->cross_connection)
6139                 return;
6140
6141         for (iface = uplink->global->ifaces; iface; iface = iface->next) {
6142                 if (!iface->cross_connect_enabled)
6143                         continue;
6144                 if (os_strcmp(uplink->ifname, iface->cross_connect_uplink) !=
6145                     0)
6146                         continue;
6147                 if (iface->ap_iface == NULL)
6148                         continue;
6149                 if (iface->cross_connect_in_use)
6150                         continue;
6151
6152                 iface->cross_connect_in_use = 1;
6153                 wpa_msg_global(iface->parent, MSG_INFO,
6154                                P2P_EVENT_CROSS_CONNECT_ENABLE "%s %s",
6155                                iface->ifname, iface->cross_connect_uplink);
6156         }
6157 }
6158
6159
6160 static void wpas_p2p_disable_cross_connect(struct wpa_supplicant *uplink)
6161 {
6162         struct wpa_supplicant *iface;
6163
6164         for (iface = uplink->global->ifaces; iface; iface = iface->next) {
6165                 if (!iface->cross_connect_enabled)
6166                         continue;
6167                 if (os_strcmp(uplink->ifname, iface->cross_connect_uplink) !=
6168                     0)
6169                         continue;
6170                 if (!iface->cross_connect_in_use)
6171                         continue;
6172
6173                 wpa_msg_global(iface->parent, MSG_INFO,
6174                                P2P_EVENT_CROSS_CONNECT_DISABLE "%s %s",
6175                                iface->ifname, iface->cross_connect_uplink);
6176                 iface->cross_connect_in_use = 0;
6177         }
6178 }
6179
6180
6181 void wpas_p2p_notif_connected(struct wpa_supplicant *wpa_s)
6182 {
6183         if (wpa_s->ap_iface || wpa_s->current_ssid == NULL ||
6184             wpa_s->current_ssid->mode != WPAS_MODE_INFRA ||
6185             wpa_s->cross_connect_disallowed)
6186                 wpas_p2p_disable_cross_connect(wpa_s);
6187         else
6188                 wpas_p2p_enable_cross_connect(wpa_s);
6189         if (!wpa_s->ap_iface &&
6190             eloop_cancel_timeout(wpas_p2p_group_idle_timeout, wpa_s, NULL) > 0)
6191                 wpa_printf(MSG_DEBUG, "P2P: Cancelled P2P group idle timeout");
6192 }
6193
6194
6195 void wpas_p2p_notif_disconnected(struct wpa_supplicant *wpa_s)
6196 {
6197         wpas_p2p_disable_cross_connect(wpa_s);
6198         if (!wpa_s->ap_iface &&
6199             !eloop_is_timeout_registered(wpas_p2p_group_idle_timeout,
6200                                          wpa_s, NULL))
6201                 wpas_p2p_set_group_idle_timeout(wpa_s);
6202 }
6203
6204
6205 static void wpas_p2p_cross_connect_setup(struct wpa_supplicant *wpa_s)
6206 {
6207         struct wpa_supplicant *iface;
6208
6209         if (!wpa_s->global->cross_connection)
6210                 return;
6211
6212         for (iface = wpa_s->global->ifaces; iface; iface = iface->next) {
6213                 if (iface == wpa_s)
6214                         continue;
6215                 if (iface->drv_flags &
6216                     WPA_DRIVER_FLAGS_P2P_DEDICATED_INTERFACE)
6217                         continue;
6218                 if (iface->drv_flags & WPA_DRIVER_FLAGS_P2P_CAPABLE)
6219                         continue;
6220
6221                 wpa_s->cross_connect_enabled = 1;
6222                 os_strlcpy(wpa_s->cross_connect_uplink, iface->ifname,
6223                            sizeof(wpa_s->cross_connect_uplink));
6224                 wpa_printf(MSG_DEBUG, "P2P: Enable cross connection from "
6225                            "%s to %s whenever uplink is available",
6226                            wpa_s->ifname, wpa_s->cross_connect_uplink);
6227
6228                 if (iface->ap_iface || iface->current_ssid == NULL ||
6229                     iface->current_ssid->mode != WPAS_MODE_INFRA ||
6230                     iface->cross_connect_disallowed ||
6231                     iface->wpa_state != WPA_COMPLETED)
6232                         break;
6233
6234                 wpa_s->cross_connect_in_use = 1;
6235                 wpa_msg_global(wpa_s->parent, MSG_INFO,
6236                                P2P_EVENT_CROSS_CONNECT_ENABLE "%s %s",
6237                                wpa_s->ifname, wpa_s->cross_connect_uplink);
6238                 break;
6239         }
6240 }
6241
6242
6243 int wpas_p2p_notif_pbc_overlap(struct wpa_supplicant *wpa_s)
6244 {
6245         if (wpa_s->p2p_group_interface != P2P_GROUP_INTERFACE_CLIENT &&
6246             !wpa_s->p2p_in_provisioning)
6247                 return 0; /* not P2P client operation */
6248
6249         wpa_printf(MSG_DEBUG, "P2P: Terminate connection due to WPS PBC "
6250                    "session overlap");
6251         if (wpa_s != wpa_s->parent)
6252                 wpa_msg_ctrl(wpa_s->parent, MSG_INFO, WPS_EVENT_OVERLAP);
6253         wpas_p2p_group_formation_failed(wpa_s);
6254         return 1;
6255 }
6256
6257
6258 void wpas_p2p_update_channel_list(struct wpa_supplicant *wpa_s)
6259 {
6260         struct p2p_channels chan, cli_chan;
6261
6262         if (wpa_s->global == NULL || wpa_s->global->p2p == NULL)
6263                 return;
6264
6265         os_memset(&chan, 0, sizeof(chan));
6266         os_memset(&cli_chan, 0, sizeof(cli_chan));
6267         if (wpas_p2p_setup_channels(wpa_s, &chan, &cli_chan)) {
6268                 wpa_printf(MSG_ERROR, "P2P: Failed to update supported "
6269                            "channel list");
6270                 return;
6271         }
6272
6273         p2p_update_channel_list(wpa_s->global->p2p, &chan, &cli_chan);
6274 }
6275
6276
6277 static void wpas_p2p_scan_res_ignore(struct wpa_supplicant *wpa_s,
6278                                      struct wpa_scan_results *scan_res)
6279 {
6280         wpa_printf(MSG_DEBUG, "P2P: Ignore scan results");
6281 }
6282
6283
6284 int wpas_p2p_cancel(struct wpa_supplicant *wpa_s)
6285 {
6286         struct wpa_global *global = wpa_s->global;
6287         int found = 0;
6288         const u8 *peer;
6289
6290         if (global->p2p == NULL)
6291                 return -1;
6292
6293         wpa_printf(MSG_DEBUG, "P2P: Request to cancel group formation");
6294
6295         if (wpa_s->pending_interface_name[0] &&
6296             !is_zero_ether_addr(wpa_s->pending_interface_addr))
6297                 found = 1;
6298
6299         peer = p2p_get_go_neg_peer(global->p2p);
6300         if (peer) {
6301                 wpa_printf(MSG_DEBUG, "P2P: Unauthorize pending GO Neg peer "
6302                            MACSTR, MAC2STR(peer));
6303                 p2p_unauthorize(global->p2p, peer);
6304                 found = 1;
6305         }
6306
6307         if (wpa_s->scan_res_handler == wpas_p2p_scan_res_join) {
6308                 wpa_printf(MSG_DEBUG, "P2P: Stop pending scan for join");
6309                 wpa_s->scan_res_handler = wpas_p2p_scan_res_ignore;
6310                 found = 1;
6311         }
6312
6313         if (wpa_s->pending_pd_before_join) {
6314                 wpa_printf(MSG_DEBUG, "P2P: Stop pending PD before join");
6315                 wpa_s->pending_pd_before_join = 0;
6316                 found = 1;
6317         }
6318
6319         wpas_p2p_stop_find(wpa_s);
6320
6321         for (wpa_s = global->ifaces; wpa_s; wpa_s = wpa_s->next) {
6322                 if (wpa_s == global->p2p_group_formation &&
6323                     (wpa_s->p2p_in_provisioning ||
6324                      wpa_s->parent->pending_interface_type ==
6325                      WPA_IF_P2P_CLIENT)) {
6326                         wpa_printf(MSG_DEBUG, "P2P: Interface %s in group "
6327                                    "formation found - cancelling",
6328                                    wpa_s->ifname);
6329                         found = 1;
6330                         eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
6331                                              wpa_s->parent, NULL);
6332                         if (wpa_s->p2p_in_provisioning) {
6333                                 wpas_group_formation_completed(wpa_s, 0);
6334                                 break;
6335                         }
6336                         wpas_p2p_group_delete(wpa_s,
6337                                               P2P_GROUP_REMOVAL_REQUESTED);
6338                         break;
6339                 }
6340         }
6341
6342         if (!found) {
6343                 wpa_printf(MSG_DEBUG, "P2P: No ongoing group formation found");
6344                 return -1;
6345         }
6346
6347         return 0;
6348 }
6349
6350
6351 void wpas_p2p_interface_unavailable(struct wpa_supplicant *wpa_s)
6352 {
6353         if (wpa_s->current_ssid == NULL || !wpa_s->current_ssid->p2p_group)
6354                 return;
6355
6356         wpa_printf(MSG_DEBUG, "P2P: Remove group due to driver resource not "
6357                    "being available anymore");
6358         wpas_p2p_group_delete(wpa_s, P2P_GROUP_REMOVAL_UNAVAILABLE);
6359 }
6360
6361
6362 void wpas_p2p_update_best_channels(struct wpa_supplicant *wpa_s,
6363                                    int freq_24, int freq_5, int freq_overall)
6364 {
6365         struct p2p_data *p2p = wpa_s->global->p2p;
6366         if (p2p == NULL)
6367                 return;
6368         p2p_set_best_channels(p2p, freq_24, freq_5, freq_overall);
6369 }
6370
6371
6372 int wpas_p2p_unauthorize(struct wpa_supplicant *wpa_s, const char *addr)
6373 {
6374         u8 peer[ETH_ALEN];
6375         struct p2p_data *p2p = wpa_s->global->p2p;
6376
6377         if (p2p == NULL)
6378                 return -1;
6379
6380         if (hwaddr_aton(addr, peer))
6381                 return -1;
6382
6383         return p2p_unauthorize(p2p, peer);
6384 }
6385
6386
6387 /**
6388  * wpas_p2p_disconnect - Disconnect from a P2P Group
6389  * @wpa_s: Pointer to wpa_supplicant data
6390  * Returns: 0 on success, -1 on failure
6391  *
6392  * This can be used to disconnect from a group in which the local end is a P2P
6393  * Client or to end a P2P Group in case the local end is the Group Owner. If a
6394  * virtual network interface was created for this group, that interface will be
6395  * removed. Otherwise, only the configured P2P group network will be removed
6396  * from the interface.
6397  */
6398 int wpas_p2p_disconnect(struct wpa_supplicant *wpa_s)
6399 {
6400
6401         if (wpa_s == NULL)
6402                 return -1;
6403
6404         return wpas_p2p_group_delete(wpa_s, P2P_GROUP_REMOVAL_REQUESTED) < 0 ?
6405                 -1 : 0;
6406 }
6407
6408
6409 int wpas_p2p_in_progress(struct wpa_supplicant *wpa_s)
6410 {
6411         int ret;
6412
6413         if (wpa_s->global->p2p_disabled || wpa_s->global->p2p == NULL)
6414                 return 0;
6415
6416         ret = p2p_in_progress(wpa_s->global->p2p);
6417         if (ret == 0) {
6418                 /*
6419                  * Check whether there is an ongoing WPS provisioning step (or
6420                  * other parts of group formation) on another interface since
6421                  * p2p_in_progress() does not report this to avoid issues for
6422                  * scans during such provisioning step.
6423                  */
6424                 if (wpa_s->global->p2p_group_formation &&
6425                     wpa_s->global->p2p_group_formation != wpa_s) {
6426                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Another interface (%s) "
6427                                 "in group formation",
6428                                 wpa_s->global->p2p_group_formation->ifname);
6429                         ret = 1;
6430                 }
6431         }
6432
6433         if (!ret && wpa_s->global->p2p_go_wait_client.sec) {
6434                 struct os_reltime now;
6435                 os_get_reltime(&now);
6436                 if (os_reltime_expired(&now, &wpa_s->global->p2p_go_wait_client,
6437                                        P2P_MAX_INITIAL_CONN_WAIT_GO)) {
6438                         /* Wait for the first client has expired */
6439                         wpa_s->global->p2p_go_wait_client.sec = 0;
6440                 } else {
6441                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Waiting for initial client connection during group formation");
6442                         ret = 1;
6443                 }
6444         }
6445
6446         return ret;
6447 }
6448
6449
6450 void wpas_p2p_network_removed(struct wpa_supplicant *wpa_s,
6451                               struct wpa_ssid *ssid)
6452 {
6453         if (wpa_s->p2p_in_provisioning && ssid->p2p_group &&
6454             eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
6455                                  wpa_s->parent, NULL) > 0) {
6456                 /**
6457                  * Remove the network by scheduling the group formation
6458                  * timeout to happen immediately. The teardown code
6459                  * needs to be scheduled to run asynch later so that we
6460                  * don't delete data from under ourselves unexpectedly.
6461                  * Calling wpas_p2p_group_formation_timeout directly
6462                  * causes a series of crashes in WPS failure scenarios.
6463                  */
6464                 wpa_printf(MSG_DEBUG, "P2P: Canceled group formation due to "
6465                            "P2P group network getting removed");
6466                 eloop_register_timeout(0, 0, wpas_p2p_group_formation_timeout,
6467                                        wpa_s->parent, NULL);
6468         }
6469 }
6470
6471
6472 struct wpa_ssid * wpas_p2p_get_persistent(struct wpa_supplicant *wpa_s,
6473                                           const u8 *addr, const u8 *ssid,
6474                                           size_t ssid_len)
6475 {
6476         struct wpa_ssid *s;
6477         size_t i;
6478
6479         for (s = wpa_s->conf->ssid; s; s = s->next) {
6480                 if (s->disabled != 2)
6481                         continue;
6482                 if (ssid &&
6483                     (ssid_len != s->ssid_len ||
6484                      os_memcmp(ssid, s->ssid, ssid_len) != 0))
6485                         continue;
6486                 if (addr == NULL) {
6487                         if (s->mode == WPAS_MODE_P2P_GO)
6488                                 return s;
6489                         continue;
6490                 }
6491                 if (os_memcmp(s->bssid, addr, ETH_ALEN) == 0)
6492                         return s; /* peer is GO in the persistent group */
6493                 if (s->mode != WPAS_MODE_P2P_GO || s->p2p_client_list == NULL)
6494                         continue;
6495                 for (i = 0; i < s->num_p2p_clients; i++) {
6496                         if (os_memcmp(s->p2p_client_list + i * ETH_ALEN,
6497                                       addr, ETH_ALEN) == 0)
6498                                 return s; /* peer is P2P client in persistent
6499                                            * group */
6500                 }
6501         }
6502
6503         return NULL;
6504 }
6505
6506
6507 void wpas_p2p_notify_ap_sta_authorized(struct wpa_supplicant *wpa_s,
6508                                        const u8 *addr)
6509 {
6510         if (eloop_cancel_timeout(wpas_p2p_group_formation_timeout,
6511                                  wpa_s->parent, NULL) > 0) {
6512                 /*
6513                  * This can happen if WPS provisioning step is not terminated
6514                  * cleanly (e.g., P2P Client does not send WSC_Done). Since the
6515                  * peer was able to connect, there is no need to time out group
6516                  * formation after this, though. In addition, this is used with
6517                  * the initial connection wait on the GO as a separate formation
6518                  * timeout and as such, expected to be hit after the initial WPS
6519                  * provisioning step.
6520                  */
6521                 wpa_printf(MSG_DEBUG, "P2P: Canceled P2P group formation timeout on data connection");
6522         }
6523         if (!wpa_s->p2p_go_group_formation_completed) {
6524                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Marking group formation completed on GO on first data connection");
6525                 wpa_s->p2p_go_group_formation_completed = 1;
6526                 wpa_s->global->p2p_group_formation = NULL;
6527                 wpa_s->p2p_in_provisioning = 0;
6528         }
6529         wpa_s->global->p2p_go_wait_client.sec = 0;
6530         if (addr == NULL)
6531                 return;
6532         wpas_p2p_add_persistent_group_client(wpa_s, addr);
6533 }
6534
6535
6536 static void wpas_p2p_fallback_to_go_neg(struct wpa_supplicant *wpa_s,
6537                                         int group_added)
6538 {
6539         struct wpa_supplicant *group = wpa_s;
6540         if (wpa_s->global->p2p_group_formation)
6541                 group = wpa_s->global->p2p_group_formation;
6542         wpa_s = wpa_s->parent;
6543         offchannel_send_action_done(wpa_s);
6544         if (group_added)
6545                 wpas_p2p_group_delete(group, P2P_GROUP_REMOVAL_SILENT);
6546         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Fall back to GO Negotiation");
6547         wpas_p2p_connect(wpa_s, wpa_s->pending_join_dev_addr, wpa_s->p2p_pin,
6548                          wpa_s->p2p_wps_method, wpa_s->p2p_persistent_group, 0,
6549                          0, 0, wpa_s->p2p_go_intent, wpa_s->p2p_connect_freq,
6550                          wpa_s->p2p_persistent_id,
6551                          wpa_s->p2p_pd_before_go_neg,
6552                          wpa_s->p2p_go_ht40,
6553                          wpa_s->p2p_go_vht);
6554 }
6555
6556
6557 int wpas_p2p_scan_no_go_seen(struct wpa_supplicant *wpa_s)
6558 {
6559         if (!wpa_s->p2p_fallback_to_go_neg ||
6560             wpa_s->p2p_in_provisioning <= 5)
6561                 return 0;
6562
6563         if (wpas_p2p_peer_go(wpa_s, wpa_s->pending_join_dev_addr) > 0)
6564                 return 0; /* peer operating as a GO */
6565
6566         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: GO not found for p2p_connect-auto - "
6567                 "fallback to GO Negotiation");
6568         wpas_p2p_fallback_to_go_neg(wpa_s, 1);
6569
6570         return 1;
6571 }
6572
6573
6574 unsigned int wpas_p2p_search_delay(struct wpa_supplicant *wpa_s)
6575 {
6576         struct wpa_supplicant *ifs;
6577
6578         if (wpa_s->wpa_state > WPA_SCANNING) {
6579                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Use %u ms search delay due to "
6580                         "concurrent operation",
6581                         P2P_CONCURRENT_SEARCH_DELAY);
6582                 return P2P_CONCURRENT_SEARCH_DELAY;
6583         }
6584
6585         dl_list_for_each(ifs, &wpa_s->radio->ifaces, struct wpa_supplicant,
6586                          radio_list) {
6587                 if (ifs != wpa_s && ifs->wpa_state > WPA_SCANNING) {
6588                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Use %u ms search "
6589                                 "delay due to concurrent operation on "
6590                                 "interface %s",
6591                                 P2P_CONCURRENT_SEARCH_DELAY, ifs->ifname);
6592                         return P2P_CONCURRENT_SEARCH_DELAY;
6593                 }
6594         }
6595
6596         return 0;
6597 }
6598
6599
6600 static int wpas_p2p_remove_psk_entry(struct wpa_supplicant *wpa_s,
6601                                      struct wpa_ssid *s, const u8 *addr,
6602                                      int iface_addr)
6603 {
6604         struct psk_list_entry *psk, *tmp;
6605         int changed = 0;
6606
6607         dl_list_for_each_safe(psk, tmp, &s->psk_list, struct psk_list_entry,
6608                               list) {
6609                 if ((iface_addr && !psk->p2p &&
6610                      os_memcmp(addr, psk->addr, ETH_ALEN) == 0) ||
6611                     (!iface_addr && psk->p2p &&
6612                      os_memcmp(addr, psk->addr, ETH_ALEN) == 0)) {
6613                         wpa_dbg(wpa_s, MSG_DEBUG,
6614                                 "P2P: Remove persistent group PSK list entry for "
6615                                 MACSTR " p2p=%u",
6616                                 MAC2STR(psk->addr), psk->p2p);
6617                         dl_list_del(&psk->list);
6618                         os_free(psk);
6619                         changed++;
6620                 }
6621         }
6622
6623         return changed;
6624 }
6625
6626
6627 void wpas_p2p_new_psk_cb(struct wpa_supplicant *wpa_s, const u8 *mac_addr,
6628                          const u8 *p2p_dev_addr,
6629                          const u8 *psk, size_t psk_len)
6630 {
6631         struct wpa_ssid *ssid = wpa_s->current_ssid;
6632         struct wpa_ssid *persistent;
6633         struct psk_list_entry *p;
6634
6635         if (psk_len != sizeof(p->psk))
6636                 return;
6637
6638         if (p2p_dev_addr) {
6639                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: New PSK for addr=" MACSTR
6640                         " p2p_dev_addr=" MACSTR,
6641                         MAC2STR(mac_addr), MAC2STR(p2p_dev_addr));
6642                 if (is_zero_ether_addr(p2p_dev_addr))
6643                         p2p_dev_addr = NULL;
6644         } else {
6645                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: New PSK for addr=" MACSTR,
6646                         MAC2STR(mac_addr));
6647         }
6648
6649         if (ssid->mode == WPAS_MODE_P2P_GROUP_FORMATION) {
6650                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: new_psk_cb during group formation");
6651                 /* To be added to persistent group once created */
6652                 if (wpa_s->global->add_psk == NULL) {
6653                         wpa_s->global->add_psk = os_zalloc(sizeof(*p));
6654                         if (wpa_s->global->add_psk == NULL)
6655                                 return;
6656                 }
6657                 p = wpa_s->global->add_psk;
6658                 if (p2p_dev_addr) {
6659                         p->p2p = 1;
6660                         os_memcpy(p->addr, p2p_dev_addr, ETH_ALEN);
6661                 } else {
6662                         p->p2p = 0;
6663                         os_memcpy(p->addr, mac_addr, ETH_ALEN);
6664                 }
6665                 os_memcpy(p->psk, psk, psk_len);
6666                 return;
6667         }
6668
6669         if (ssid->mode != WPAS_MODE_P2P_GO || !ssid->p2p_persistent_group) {
6670                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Ignore new_psk_cb on not-persistent GO");
6671                 return;
6672         }
6673
6674         persistent = wpas_p2p_get_persistent(wpa_s->parent, NULL, ssid->ssid,
6675                                              ssid->ssid_len);
6676         if (!persistent) {
6677                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Could not find persistent group information to store the new PSK");
6678                 return;
6679         }
6680
6681         p = os_zalloc(sizeof(*p));
6682         if (p == NULL)
6683                 return;
6684         if (p2p_dev_addr) {
6685                 p->p2p = 1;
6686                 os_memcpy(p->addr, p2p_dev_addr, ETH_ALEN);
6687         } else {
6688                 p->p2p = 0;
6689                 os_memcpy(p->addr, mac_addr, ETH_ALEN);
6690         }
6691         os_memcpy(p->psk, psk, psk_len);
6692
6693         if (dl_list_len(&persistent->psk_list) > P2P_MAX_STORED_CLIENTS) {
6694                 struct psk_list_entry *last;
6695                 last = dl_list_last(&persistent->psk_list,
6696                                     struct psk_list_entry, list);
6697                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Remove oldest PSK entry for "
6698                         MACSTR " (p2p=%u) to make room for a new one",
6699                         MAC2STR(last->addr), last->p2p);
6700                 dl_list_del(&last->list);
6701                 os_free(last);
6702         }
6703
6704         wpas_p2p_remove_psk_entry(wpa_s->parent, persistent,
6705                                   p2p_dev_addr ? p2p_dev_addr : mac_addr,
6706                                   p2p_dev_addr == NULL);
6707         if (p2p_dev_addr) {
6708                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Add new PSK for p2p_dev_addr="
6709                         MACSTR, MAC2STR(p2p_dev_addr));
6710         } else {
6711                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Add new PSK for addr=" MACSTR,
6712                         MAC2STR(mac_addr));
6713         }
6714         dl_list_add(&persistent->psk_list, &p->list);
6715
6716 #ifndef CONFIG_NO_CONFIG_WRITE
6717         if (wpa_s->parent->conf->update_config &&
6718             wpa_config_write(wpa_s->parent->confname, wpa_s->parent->conf))
6719                 wpa_printf(MSG_DEBUG, "P2P: Failed to update configuration");
6720 #endif /* CONFIG_NO_CONFIG_WRITE */
6721 }
6722
6723
6724 static void wpas_p2p_remove_psk(struct wpa_supplicant *wpa_s,
6725                                 struct wpa_ssid *s, const u8 *addr,
6726                                 int iface_addr)
6727 {
6728         int res;
6729
6730         res = wpas_p2p_remove_psk_entry(wpa_s, s, addr, iface_addr);
6731         if (res > 0) {
6732 #ifndef CONFIG_NO_CONFIG_WRITE
6733                 if (wpa_s->conf->update_config &&
6734                     wpa_config_write(wpa_s->confname, wpa_s->conf))
6735                         wpa_dbg(wpa_s, MSG_DEBUG,
6736                                 "P2P: Failed to update configuration");
6737 #endif /* CONFIG_NO_CONFIG_WRITE */
6738         }
6739 }
6740
6741
6742 static void wpas_p2p_remove_client_go(struct wpa_supplicant *wpa_s,
6743                                       const u8 *peer, int iface_addr)
6744 {
6745         struct hostapd_data *hapd;
6746         struct hostapd_wpa_psk *psk, *prev, *rem;
6747         struct sta_info *sta;
6748
6749         if (wpa_s->ap_iface == NULL || wpa_s->current_ssid == NULL ||
6750             wpa_s->current_ssid->mode != WPAS_MODE_P2P_GO)
6751                 return;
6752
6753         /* Remove per-station PSK entry */
6754         hapd = wpa_s->ap_iface->bss[0];
6755         prev = NULL;
6756         psk = hapd->conf->ssid.wpa_psk;
6757         while (psk) {
6758                 if ((iface_addr && os_memcmp(peer, psk->addr, ETH_ALEN) == 0) ||
6759                     (!iface_addr &&
6760                      os_memcmp(peer, psk->p2p_dev_addr, ETH_ALEN) == 0)) {
6761                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Remove operating group PSK entry for "
6762                                 MACSTR " iface_addr=%d",
6763                                 MAC2STR(peer), iface_addr);
6764                         if (prev)
6765                                 prev->next = psk->next;
6766                         else
6767                                 hapd->conf->ssid.wpa_psk = psk->next;
6768                         rem = psk;
6769                         psk = psk->next;
6770                         os_free(rem);
6771                 } else {
6772                         prev = psk;
6773                         psk = psk->next;
6774                 }
6775         }
6776
6777         /* Disconnect from group */
6778         if (iface_addr)
6779                 sta = ap_get_sta(hapd, peer);
6780         else
6781                 sta = ap_get_sta_p2p(hapd, peer);
6782         if (sta) {
6783                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Disconnect peer " MACSTR
6784                         " (iface_addr=%d) from group",
6785                         MAC2STR(peer), iface_addr);
6786                 hostapd_drv_sta_deauth(hapd, sta->addr,
6787                                        WLAN_REASON_DEAUTH_LEAVING);
6788                 ap_sta_deauthenticate(hapd, sta, WLAN_REASON_DEAUTH_LEAVING);
6789         }
6790 }
6791
6792
6793 void wpas_p2p_remove_client(struct wpa_supplicant *wpa_s, const u8 *peer,
6794                             int iface_addr)
6795 {
6796         struct wpa_ssid *s;
6797         struct wpa_supplicant *w;
6798
6799         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Remove client " MACSTR, MAC2STR(peer));
6800
6801         /* Remove from any persistent group */
6802         for (s = wpa_s->parent->conf->ssid; s; s = s->next) {
6803                 if (s->disabled != 2 || s->mode != WPAS_MODE_P2P_GO)
6804                         continue;
6805                 if (!iface_addr)
6806                         wpas_remove_persistent_peer(wpa_s, s, peer, 0);
6807                 wpas_p2p_remove_psk(wpa_s->parent, s, peer, iface_addr);
6808         }
6809
6810         /* Remove from any operating group */
6811         for (w = wpa_s->global->ifaces; w; w = w->next)
6812                 wpas_p2p_remove_client_go(w, peer, iface_addr);
6813 }
6814
6815
6816 static void wpas_p2p_psk_failure_removal(void *eloop_ctx, void *timeout_ctx)
6817 {
6818         struct wpa_supplicant *wpa_s = eloop_ctx;
6819         wpas_p2p_group_delete(wpa_s, P2P_GROUP_REMOVAL_PSK_FAILURE);
6820 }
6821
6822
6823 static void wpas_p2p_group_freq_conflict(void *eloop_ctx, void *timeout_ctx)
6824 {
6825         struct wpa_supplicant *wpa_s = eloop_ctx;
6826
6827         wpa_printf(MSG_DEBUG, "P2P: Frequency conflict - terminate group");
6828         wpas_p2p_group_delete(wpa_s, P2P_GROUP_REMOVAL_FREQ_CONFLICT);
6829 }
6830
6831
6832 int wpas_p2p_handle_frequency_conflicts(struct wpa_supplicant *wpa_s, int freq,
6833                                         struct wpa_ssid *ssid)
6834 {
6835         struct wpa_supplicant *iface;
6836
6837         for (iface = wpa_s->global->ifaces; iface; iface = iface->next) {
6838                 if (!iface->current_ssid ||
6839                     iface->current_ssid->frequency == freq ||
6840                     (iface->p2p_group_interface == NOT_P2P_GROUP_INTERFACE &&
6841                      !iface->current_ssid->p2p_group))
6842                         continue;
6843
6844                 /* Remove the connection with least priority */
6845                 if (!wpas_is_p2p_prioritized(iface)) {
6846                         /* STA connection has priority over existing
6847                          * P2P connection, so remove the interface. */
6848                         wpa_printf(MSG_DEBUG, "P2P: Removing P2P connection due to single channel concurrent mode frequency conflict");
6849                         eloop_register_timeout(0, 0,
6850                                                wpas_p2p_group_freq_conflict,
6851                                                iface, NULL);
6852                         /* If connection in progress is P2P connection, do not
6853                          * proceed for the connection. */
6854                         if (wpa_s == iface)
6855                                 return -1;
6856                         else
6857                                 return 0;
6858                 } else {
6859                         /* P2P connection has priority, disable the STA network
6860                          */
6861                         wpa_supplicant_disable_network(wpa_s->global->ifaces,
6862                                                        ssid);
6863                         wpa_msg(wpa_s->global->ifaces, MSG_INFO,
6864                                 WPA_EVENT_FREQ_CONFLICT " id=%d", ssid->id);
6865                         os_memset(wpa_s->global->ifaces->pending_bssid, 0,
6866                                   ETH_ALEN);
6867                         /* If P2P connection is in progress, continue
6868                          * connecting...*/
6869                         if (wpa_s == iface)
6870                                 return 0;
6871                         else
6872                                 return -1;
6873                 }
6874         }
6875
6876         return 0;
6877 }
6878
6879
6880 int wpas_p2p_4way_hs_failed(struct wpa_supplicant *wpa_s)
6881 {
6882         struct wpa_ssid *ssid = wpa_s->current_ssid;
6883
6884         if (ssid == NULL || !ssid->p2p_group)
6885                 return 0;
6886
6887         if (wpa_s->p2p_last_4way_hs_fail &&
6888             wpa_s->p2p_last_4way_hs_fail == ssid) {
6889                 u8 go_dev_addr[ETH_ALEN];
6890                 struct wpa_ssid *persistent;
6891
6892                 if (wpas_p2p_persistent_group(wpa_s, go_dev_addr,
6893                                               ssid->ssid,
6894                                               ssid->ssid_len) <= 0) {
6895                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Could not determine whether 4-way handshake failures were for a persistent group");
6896                         goto disconnect;
6897                 }
6898
6899                 wpa_dbg(wpa_s, MSG_DEBUG, "P2P: Two 4-way handshake failures for a P2P group - go_dev_addr="
6900                         MACSTR, MAC2STR(go_dev_addr));
6901                 persistent = wpas_p2p_get_persistent(wpa_s->parent, go_dev_addr,
6902                                                      ssid->ssid,
6903                                                      ssid->ssid_len);
6904                 if (persistent == NULL || persistent->mode != WPAS_MODE_INFRA) {
6905                         wpa_dbg(wpa_s, MSG_DEBUG, "P2P: No matching persistent group stored");
6906                         goto disconnect;
6907                 }
6908                 wpa_msg_global(wpa_s->parent, MSG_INFO,
6909                                P2P_EVENT_PERSISTENT_PSK_FAIL "%d",
6910                                persistent->id);
6911         disconnect:
6912                 wpa_s->p2p_last_4way_hs_fail = NULL;
6913                 /*
6914                  * Remove the group from a timeout to avoid issues with caller
6915                  * continuing to use the interface if this is on a P2P group
6916                  * interface.
6917                  */
6918                 eloop_register_timeout(0, 0, wpas_p2p_psk_failure_removal,
6919                                        wpa_s, NULL);
6920                 return 1;
6921         }
6922
6923         wpa_s->p2p_last_4way_hs_fail = ssid;
6924         return 0;
6925 }