P2P: Allow P2P listen being offloaded to the driver/firmware
[mech_eap.git] / src / p2p / p2p.c
index 59ab71a..47abe21 100644 (file)
@@ -10,6 +10,7 @@
 
 #include "common.h"
 #include "eloop.h"
+#include "common/defs.h"
 #include "common/ieee802_11_defs.h"
 #include "common/ieee802_11_common.h"
 #include "common/wpa_ctrl.h"
@@ -48,9 +49,8 @@ static void p2p_scan_timeout(void *eloop_ctx, void *timeout_ctx);
 #define P2P_PEER_EXPIRATION_AGE 60
 #endif /* P2P_PEER_EXPIRATION_AGE */
 
-#define P2P_PEER_EXPIRATION_INTERVAL (P2P_PEER_EXPIRATION_AGE / 2)
 
-static void p2p_expire_peers(struct p2p_data *p2p)
+void p2p_expire_peers(struct p2p_data *p2p)
 {
        struct p2p_device *dev, *n;
        struct os_reltime now;
@@ -103,15 +103,6 @@ static void p2p_expire_peers(struct p2p_data *p2p)
 }
 
 
-static void p2p_expiration_timeout(void *eloop_ctx, void *timeout_ctx)
-{
-       struct p2p_data *p2p = eloop_ctx;
-       p2p_expire_peers(p2p);
-       eloop_register_timeout(P2P_PEER_EXPIRATION_INTERVAL, 0,
-                              p2p_expiration_timeout, p2p, NULL);
-}
-
-
 static const char * p2p_state_txt(int state)
 {
        switch (state) {
@@ -153,6 +144,19 @@ const char * p2p_get_state_txt(struct p2p_data *p2p)
 }
 
 
+struct p2ps_advertisement * p2p_get_p2ps_adv_list(struct p2p_data *p2p)
+{
+       return p2p ? p2p->p2ps_adv_list : NULL;
+}
+
+
+void p2p_set_intended_addr(struct p2p_data *p2p, const u8 *intended_addr)
+{
+       if (p2p && intended_addr)
+               os_memcpy(p2p->intended_addr, intended_addr, ETH_ALEN);
+}
+
+
 u16 p2p_get_provisioning_info(struct p2p_data *p2p, const u8 *addr)
 {
        struct p2p_device *dev = NULL;
@@ -284,7 +288,7 @@ static void p2p_listen_in_find(struct p2p_data *p2p, int dev_disc)
                return;
        }
 
-       ies = p2p_build_probe_resp_ies(p2p);
+       ies = p2p_build_probe_resp_ies(p2p, NULL, 0);
        if (ies == NULL)
                return;
 
@@ -333,7 +337,7 @@ int p2p_listen(struct p2p_data *p2p, unsigned int timeout)
                return 0;
        }
 
-       ies = p2p_build_probe_resp_ies(p2p);
+       ies = p2p_build_probe_resp_ies(p2p, NULL, 0);
        if (ies == NULL)
                return -1;
 
@@ -442,8 +446,9 @@ static struct p2p_device * p2p_create_device(struct p2p_data *p2p,
 static void p2p_copy_client_info(struct p2p_device *dev,
                                 struct p2p_client_info *cli)
 {
-       os_memcpy(dev->info.device_name, cli->dev_name, cli->dev_name_len);
-       dev->info.device_name[cli->dev_name_len] = '\0';
+       p2p_copy_filter_devname(dev->info.device_name,
+                               sizeof(dev->info.device_name),
+                               cli->dev_name, cli->dev_name_len);
        dev->info.dev_capab = cli->dev_capab;
        dev->info.config_methods = cli->config_methods;
        os_memcpy(dev->info.pri_dev_type, cli->pri_dev_type, 8);
@@ -455,7 +460,8 @@ static void p2p_copy_client_info(struct p2p_device *dev,
 
 static int p2p_add_group_clients(struct p2p_data *p2p, const u8 *go_dev_addr,
                                 const u8 *go_interface_addr, int freq,
-                                const u8 *gi, size_t gi_len)
+                                const u8 *gi, size_t gi_len,
+                                struct os_reltime *rx_time)
 {
        struct p2p_group_info info;
        size_t c;
@@ -523,10 +529,11 @@ static int p2p_add_group_clients(struct p2p_data *p2p, const u8 *go_dev_addr,
 
                os_memcpy(dev->interface_addr, cli->p2p_interface_addr,
                          ETH_ALEN);
-               os_get_reltime(&dev->last_seen);
+               os_memcpy(&dev->last_seen, rx_time, sizeof(struct os_reltime));
                os_memcpy(dev->member_in_go_dev, go_dev_addr, ETH_ALEN);
                os_memcpy(dev->member_in_go_iface, go_interface_addr,
                          ETH_ALEN);
+               dev->flags |= P2P_DEV_LAST_SEEN_AS_GROUP_CLIENT;
        }
 
        return 0;
@@ -631,11 +638,11 @@ static void p2p_update_peer_vendor_elems(struct p2p_device *dev, const u8 *ies,
 
        end = ies + ies_len;
 
-       for (pos = ies; pos + 1 < end; pos += len) {
+       for (pos = ies; end - pos > 1; pos += len) {
                id = *pos++;
                len = *pos++;
 
-               if (pos + len > end)
+               if (len > end - pos)
                        break;
 
                if (id != WLAN_EID_VENDOR_SPECIFIC || len < 3)
@@ -745,26 +752,35 @@ int p2p_add_device(struct p2p_data *p2p, const u8 *addr, int freq,
 
        /*
         * Update the device entry only if the new peer
-        * entry is newer than the one previously stored.
+        * entry is newer than the one previously stored, or if
+        * the device was previously seen as a P2P Client in a group
+        * and the new entry isn't older than a threshold.
         */
        if (dev->last_seen.sec > 0 &&
-           os_reltime_before(rx_time, &dev->last_seen)) {
-               p2p_dbg(p2p, "Do not update peer entry based on old frame (rx_time=%u.%06u last_seen=%u.%06u)",
+           os_reltime_before(rx_time, &dev->last_seen) &&
+           (!(dev->flags & P2P_DEV_LAST_SEEN_AS_GROUP_CLIENT) ||
+            os_reltime_expired(&dev->last_seen, rx_time,
+                               P2P_DEV_GROUP_CLIENT_RESP_THRESHOLD))) {
+               p2p_dbg(p2p,
+                       "Do not update peer entry based on old frame (rx_time=%u.%06u last_seen=%u.%06u flags=0x%x)",
                        (unsigned int) rx_time->sec,
                        (unsigned int) rx_time->usec,
                        (unsigned int) dev->last_seen.sec,
-                       (unsigned int) dev->last_seen.usec);
+                       (unsigned int) dev->last_seen.usec,
+                       dev->flags);
                p2p_parse_free(&msg);
                return -1;
        }
 
        os_memcpy(&dev->last_seen, rx_time, sizeof(struct os_reltime));
 
-       dev->flags &= ~(P2P_DEV_PROBE_REQ_ONLY | P2P_DEV_GROUP_CLIENT_ONLY);
+       dev->flags &= ~(P2P_DEV_PROBE_REQ_ONLY | P2P_DEV_GROUP_CLIENT_ONLY |
+                       P2P_DEV_LAST_SEEN_AS_GROUP_CLIENT);
 
        if (os_memcmp(addr, p2p_dev_addr, ETH_ALEN) != 0)
                os_memcpy(dev->interface_addr, addr, ETH_ALEN);
        if (msg.ssid &&
+           msg.ssid[1] <= sizeof(dev->oper_ssid) &&
            (msg.ssid[1] != P2P_WILDCARD_SSID_LEN ||
             os_memcmp(msg.ssid + 2, P2P_WILDCARD_SSID, P2P_WILDCARD_SSID_LEN)
             != 0)) {
@@ -772,6 +788,12 @@ int p2p_add_device(struct p2p_data *p2p, const u8 *addr, int freq,
                dev->oper_ssid_len = msg.ssid[1];
        }
 
+       if (msg.adv_service_instance && msg.adv_service_instance_len) {
+               wpabuf_free(dev->info.p2ps_instance);
+               dev->info.p2ps_instance = wpabuf_alloc_copy(
+                       msg.adv_service_instance, msg.adv_service_instance_len);
+       }
+
        if (freq >= 2412 && freq <= 2484 && msg.ds_params &&
            *msg.ds_params >= 1 && *msg.ds_params <= 14) {
                int ds_freq;
@@ -824,14 +846,17 @@ int p2p_add_device(struct p2p_data *p2p, const u8 *addr, int freq,
 
        if (scan_res) {
                p2p_add_group_clients(p2p, p2p_dev_addr, addr, freq,
-                                     msg.group_info, msg.group_info_len);
+                                     msg.group_info, msg.group_info_len,
+                                     rx_time);
        }
 
        p2p_parse_free(&msg);
 
        p2p_update_peer_vendor_elems(dev, ies, ies_len);
 
-       if (dev->flags & P2P_DEV_REPORTED && !wfd_changed)
+       if (dev->flags & P2P_DEV_REPORTED && !wfd_changed &&
+           (!msg.adv_service_instance ||
+            (dev->flags & P2P_DEV_P2PS_REPORTED)))
                return 0;
 
        p2p_dbg(p2p, "Peer found with Listen frequency %d MHz (rx_time=%u.%06u)",
@@ -867,6 +892,9 @@ int p2p_add_device(struct p2p_data *p2p, const u8 *addr, int freq,
                            !(dev->flags & P2P_DEV_REPORTED_ONCE));
        dev->flags |= P2P_DEV_REPORTED | P2P_DEV_REPORTED_ONCE;
 
+       if (msg.adv_service_instance)
+               dev->flags |= P2P_DEV_P2PS_REPORTED;
+
        return 0;
 }
 
@@ -901,6 +929,7 @@ static void p2p_device_free(struct p2p_data *p2p, struct p2p_device *dev)
        wpabuf_free(dev->info.wfd_subelems);
        wpabuf_free(dev->info.vendor_elems);
        wpabuf_free(dev->go_neg_conf);
+       wpabuf_free(dev->info.p2ps_instance);
 
        os_free(dev);
 }
@@ -1102,8 +1131,10 @@ static int p2ps_gen_hash(struct p2p_data *p2p, const char *str, u8 *hash)
 
        adv_array = (u8 *) str_buf;
        adv_len = os_strlen(str);
+       if (adv_len >= sizeof(str_buf))
+               return 0;
 
-       for (i = 0; str[i] && i < adv_len; i++) {
+       for (i = 0; i < adv_len; i++) {
                if (str[i] >= 'A' && str[i] <= 'Z')
                        str_buf[i] = str[i] - 'A' + 'a';
                else
@@ -1122,7 +1153,7 @@ int p2p_find(struct p2p_data *p2p, unsigned int timeout,
             enum p2p_discovery_type type,
             unsigned int num_req_dev_types, const u8 *req_dev_types,
             const u8 *dev_id, unsigned int search_delay,
-            u8 seek_count, const char **seek)
+            u8 seek_count, const char **seek, int freq)
 {
        int res;
 
@@ -1157,27 +1188,25 @@ int p2p_find(struct p2p_data *p2p, unsigned int timeout,
                 * An empty seek string means no hash values, but still an ASP
                 * search.
                 */
+               p2p_dbg(p2p, "ASP search");
                p2p->p2ps_seek_count = 0;
                p2p->p2ps_seek = 1;
        } else if (seek && seek_count <= P2P_MAX_QUERY_HASH) {
                u8 buf[P2PS_HASH_LEN];
-               int i;
+               int i, count = 0;
 
-               p2p->p2ps_seek_count = seek_count;
                for (i = 0; i < seek_count; i++) {
                        if (!p2ps_gen_hash(p2p, seek[i], buf))
                                continue;
 
-                       /* If asking for wildcard, don't do others */
-                       if (os_memcmp(buf, p2p->wild_card_hash,
-                                     P2PS_HASH_LEN) == 0) {
-                               p2p->p2ps_seek_count = 0;
-                               break;
-                       }
-
-                       os_memcpy(&p2p->query_hash[i * P2PS_HASH_LEN], buf,
-                                 P2PS_HASH_LEN);
+                       p2p_dbg(p2p, "Seek service %s hash " MACSTR,
+                               seek[i], MAC2STR(buf));
+                       os_memcpy(&p2p->p2ps_seek_hash[count * P2PS_HASH_LEN],
+                                 buf, P2PS_HASH_LEN);
+                       count++;
                }
+
+               p2p->p2ps_seek_count = count;
                p2p->p2ps_seek = 1;
        } else {
                p2p->p2ps_seek_count = 0;
@@ -1187,14 +1216,20 @@ int p2p_find(struct p2p_data *p2p, unsigned int timeout,
        /* Special case to perform wildcard search */
        if (p2p->p2ps_seek_count == 0 && p2p->p2ps_seek) {
                p2p->p2ps_seek_count = 1;
-               os_memcpy(&p2p->query_hash, p2p->wild_card_hash, P2PS_HASH_LEN);
+               os_memcpy(&p2p->p2ps_seek_hash, p2p->wild_card_hash,
+                         P2PS_HASH_LEN);
        }
 
        p2p->start_after_scan = P2P_AFTER_SCAN_NOTHING;
        p2p_clear_timeout(p2p);
+       if (p2p->pending_listen_freq) {
+               p2p_dbg(p2p, "Clear pending_listen_freq for p2p_find");
+               p2p->pending_listen_freq = 0;
+       }
        p2p->cfg->stop_listen(p2p->cfg->cb_ctx);
        p2p->find_type = type;
        p2p_device_clear_reported(p2p);
+       os_memset(p2p->sd_query_no_ack, 0, ETH_ALEN);
        p2p_set_state(p2p, P2P_SEARCH);
        p2p->search_delay = search_delay;
        p2p->in_search_delay = 0;
@@ -1205,6 +1240,19 @@ int p2p_find(struct p2p_data *p2p, unsigned int timeout,
                                       p2p, NULL);
        switch (type) {
        case P2P_FIND_START_WITH_FULL:
+               if (freq > 0) {
+                       /*
+                        * Start with the specified channel and then move to
+                        * social channels only scans.
+                        */
+                       res = p2p->cfg->p2p_scan(p2p->cfg->cb_ctx,
+                                                P2P_SCAN_SPECIFIC, freq,
+                                                p2p->num_req_dev_types,
+                                                p2p->req_dev_types, dev_id,
+                                                DEV_PW_DEFAULT);
+                       break;
+               }
+               /* fall through */
        case P2P_FIND_PROGRESSIVE:
                res = p2p->cfg->p2p_scan(p2p->cfg->cb_ctx, P2P_SCAN_FULL, 0,
                                         p2p->num_req_dev_types,
@@ -1240,7 +1288,7 @@ void p2p_stop_find_for_freq(struct p2p_data *p2p, int freq)
        p2p_dbg(p2p, "Stopping find");
        eloop_cancel_timeout(p2p_find_timeout, p2p, NULL);
        p2p_clear_timeout(p2p);
-       if (p2p->state == P2P_SEARCH)
+       if (p2p->state == P2P_SEARCH || p2p->state == P2P_SD_DURING_FIND)
                p2p->cfg->find_stopped(p2p->cfg->cb_ctx);
 
        p2p->p2ps_seek_count = 0;
@@ -1254,6 +1302,7 @@ void p2p_stop_find_for_freq(struct p2p_data *p2p, int freq)
        p2p->sd_peer = NULL;
        p2p->invite_peer = NULL;
        p2p_stop_listen_for_freq(p2p, freq);
+       p2p->send_action_in_progress = 0;
 }
 
 
@@ -1341,7 +1390,7 @@ static int p2p_prepare_channel_pref(struct p2p_data *p2p,
 static void p2p_prepare_channel_best(struct p2p_data *p2p)
 {
        u8 op_class, op_channel;
-       const int op_classes_5ghz[] = { 124, 115, 0 };
+       const int op_classes_5ghz[] = { 124, 125, 115, 0 };
        const int op_classes_ht40[] = { 126, 127, 116, 117, 0 };
        const int op_classes_vht[] = { 128, 0 };
 
@@ -1417,7 +1466,7 @@ static void p2p_prepare_channel_best(struct p2p_data *p2p)
 
 
 /**
- * p2p_prepare_channel - Select operating channel for GO Negotiation
+ * p2p_prepare_channel - Select operating channel for GO Negotiation or P2PS PD
  * @p2p: P2P module context from p2p_init()
  * @dev: Selected peer device
  * @force_freq: Forced frequency in MHz or 0 if not forced
@@ -1426,9 +1475,9 @@ static void p2p_prepare_channel_best(struct p2p_data *p2p)
  * Returns: 0 on success, -1 on failure (channel not supported for P2P)
  *
  * This function is used to do initial operating channel selection for GO
- * Negotiation prior to having received peer information. The selected channel
- * may be further optimized in p2p_reselect_channel() once the peer information
- * is available.
+ * Negotiation prior to having received peer information or for P2PS PD
+ * signalling. The selected channel may be further optimized in
+ * p2p_reselect_channel() once the peer information is available.
  */
 int p2p_prepare_channel(struct p2p_data *p2p, struct p2p_device *dev,
                        unsigned int force_freq, unsigned int pref_freq, int go)
@@ -1717,7 +1766,14 @@ void p2p_build_ssid(struct p2p_data *p2p, u8 *ssid, size_t *ssid_len)
 
 int p2p_go_params(struct p2p_data *p2p, struct p2p_go_neg_results *params)
 {
-       p2p_build_ssid(p2p, params->ssid, &params->ssid_len);
+       if (p2p->ssid_set) {
+               os_memcpy(params->ssid, p2p->ssid, p2p->ssid_len);
+               params->ssid_len = p2p->ssid_len;
+       } else {
+               p2p_build_ssid(p2p, params->ssid, &params->ssid_len);
+       }
+       p2p->ssid_set = 0;
+
        p2p_random(params->passphrase, p2p->cfg->passphrase_len);
        return 0;
 }
@@ -1979,8 +2035,23 @@ static void p2p_add_dev_from_probe_req(struct p2p_data *p2p, const u8 *addr,
 
        dev = p2p_get_device(p2p, addr);
        if (dev) {
-               if (dev->country[0] == 0 && msg.listen_channel)
-                       os_memcpy(dev->country, msg.listen_channel, 3);
+               if (msg.listen_channel) {
+                       int freq;
+
+                       if (dev->country[0] == 0)
+                               os_memcpy(dev->country, msg.listen_channel, 3);
+
+                       freq = p2p_channel_to_freq(msg.listen_channel[3],
+                                                  msg.listen_channel[4]);
+
+                       if (freq > 0 && dev->listen_freq != freq) {
+                               p2p_dbg(p2p,
+                                       "Updated peer " MACSTR " Listen channel (Probe Request): %d -> %d MHz",
+                                       MAC2STR(addr), dev->listen_freq, freq);
+                               dev->listen_freq = freq;
+                       }
+               }
+
                os_get_reltime(&dev->last_seen);
                p2p_parse_free(&msg);
                return; /* already known */
@@ -2101,7 +2172,9 @@ int p2p_match_dev_type(struct p2p_data *p2p, struct wpabuf *wps)
 }
 
 
-struct wpabuf * p2p_build_probe_resp_ies(struct p2p_data *p2p)
+struct wpabuf * p2p_build_probe_resp_ies(struct p2p_data *p2p,
+                                        const u8 *query_hash,
+                                        u8 query_count)
 {
        struct wpabuf *buf;
        u8 *len;
@@ -2116,6 +2189,9 @@ struct wpabuf * p2p_build_probe_resp_ies(struct p2p_data *p2p)
        if (p2p->vendor_elem && p2p->vendor_elem[VENDOR_ELEM_PROBE_RESP_P2P])
                extra += wpabuf_len(p2p->vendor_elem[VENDOR_ELEM_PROBE_RESP_P2P]);
 
+       if (query_count)
+               extra += MAX_SVC_ADV_IE_LEN;
+
        buf = wpabuf_alloc(1000 + extra);
        if (buf == NULL)
                return NULL;
@@ -2150,27 +2226,102 @@ struct wpabuf * p2p_build_probe_resp_ies(struct p2p_data *p2p)
        p2p_buf_add_device_info(buf, p2p, NULL);
        p2p_buf_update_ie_hdr(buf, len);
 
+       if (query_count) {
+               p2p_buf_add_service_instance(buf, p2p, query_count, query_hash,
+                                            p2p->p2ps_adv_list);
+       }
+
        return buf;
 }
 
+static int p2p_build_probe_resp_buf(struct p2p_data *p2p, struct wpabuf *buf,
+                                   struct wpabuf *ies,
+                                   const u8 *addr, int rx_freq)
+{
+       struct ieee80211_mgmt *resp;
+       u8 channel, op_class;
+
+       resp = wpabuf_put(buf, offsetof(struct ieee80211_mgmt,
+                                       u.probe_resp.variable));
+
+       resp->frame_control = host_to_le16((WLAN_FC_TYPE_MGMT << 2) |
+                                          (WLAN_FC_STYPE_PROBE_RESP << 4));
+       os_memcpy(resp->da, addr, ETH_ALEN);
+       os_memcpy(resp->sa, p2p->cfg->dev_addr, ETH_ALEN);
+       os_memcpy(resp->bssid, p2p->cfg->dev_addr, ETH_ALEN);
+       resp->u.probe_resp.beacon_int = host_to_le16(100);
+       /* hardware or low-level driver will setup seq_ctrl and timestamp */
+       resp->u.probe_resp.capab_info =
+           host_to_le16(WLAN_CAPABILITY_SHORT_PREAMBLE |
+                    WLAN_CAPABILITY_PRIVACY |
+                    WLAN_CAPABILITY_SHORT_SLOT_TIME);
+
+       wpabuf_put_u8(buf, WLAN_EID_SSID);
+       wpabuf_put_u8(buf, P2P_WILDCARD_SSID_LEN);
+       wpabuf_put_data(buf, P2P_WILDCARD_SSID, P2P_WILDCARD_SSID_LEN);
+
+       wpabuf_put_u8(buf, WLAN_EID_SUPP_RATES);
+       wpabuf_put_u8(buf, 8);
+       wpabuf_put_u8(buf, (60 / 5) | 0x80);
+       wpabuf_put_u8(buf, 90 / 5);
+       wpabuf_put_u8(buf, (120 / 5) | 0x80);
+       wpabuf_put_u8(buf, 180 / 5);
+       wpabuf_put_u8(buf, (240 / 5) | 0x80);
+       wpabuf_put_u8(buf, 360 / 5);
+       wpabuf_put_u8(buf, 480 / 5);
+       wpabuf_put_u8(buf, 540 / 5);
+
+       if (!rx_freq) {
+               channel = p2p->cfg->channel;
+       } else if (p2p_freq_to_channel(rx_freq, &op_class, &channel)) {
+               p2p_err(p2p, "Failed to convert freq to channel");
+               return -1;
+       }
+
+       wpabuf_put_u8(buf, WLAN_EID_DS_PARAMS);
+       wpabuf_put_u8(buf, 1);
+       wpabuf_put_u8(buf, channel);
+
+       wpabuf_put_buf(buf, ies);
+
+       return 0;
+}
+
+static int p2p_service_find_asp(struct p2p_data *p2p, const u8 *hash)
+{
+       struct p2ps_advertisement *adv_data;
+       int any_wfa;
+
+       p2p_dbg(p2p, "ASP find - ASP list: %p", p2p->p2ps_adv_list);
+
+       /* Wildcard org.wi-fi.wfds matches any WFA spec defined service */
+       any_wfa = os_memcmp(hash, p2p->wild_card_hash, P2PS_HASH_LEN) == 0;
+
+       adv_data = p2p->p2ps_adv_list;
+       while (adv_data) {
+               if (os_memcmp(hash, adv_data->hash, P2PS_HASH_LEN) == 0)
+                       return 1; /* exact hash match */
+               if (any_wfa &&
+                   os_strncmp(adv_data->svc_name, P2PS_WILD_HASH_STR,
+                              os_strlen(P2PS_WILD_HASH_STR)) == 0)
+                       return 1; /* WFA service match */
+               adv_data = adv_data->next;
+       }
+
+       return 0;
+}
+
 
 static enum p2p_probe_req_status
 p2p_reply_probe(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
-               const u8 *bssid, const u8 *ie, size_t ie_len)
+               const u8 *bssid, const u8 *ie, size_t ie_len,
+               unsigned int rx_freq)
 {
        struct ieee802_11_elems elems;
        struct wpabuf *buf;
-       struct ieee80211_mgmt *resp;
        struct p2p_message msg;
        struct wpabuf *ies;
 
-       if (!p2p->in_listen || !p2p->drv_in_listen) {
-               /* not in Listen state - ignore Probe Request */
-               p2p_dbg(p2p, "Not in Listen state (in_listen=%d drv_in_listen=%d) - ignore Probe Request",
-                       p2p->in_listen, p2p->drv_in_listen);
-               return P2P_PREQ_NOT_LISTEN;
-       }
-
        if (ieee802_11_parse_elems((u8 *) ie, ie_len, &elems, 0) ==
            ParseFailed) {
                /* Ignore invalid Probe Request frames */
@@ -2221,6 +2372,53 @@ p2p_reply_probe(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
                return P2P_PREQ_NOT_P2P;
        }
 
+       if (msg.service_hash && msg.service_hash_count) {
+               const u8 *hash = msg.service_hash;
+               u8 i;
+               int p2ps_svc_found = 0;
+
+               p2p_dbg(p2p, "in_listen=%d drv_in_listen=%d when received P2PS Probe Request at %u MHz; own Listen channel %u, pending listen freq %u MHz",
+                       p2p->in_listen, p2p->drv_in_listen, rx_freq,
+                       p2p->cfg->channel, p2p->pending_listen_freq);
+
+               if (!p2p->in_listen && !p2p->drv_in_listen &&
+                   p2p->pending_listen_freq && rx_freq &&
+                   rx_freq != p2p->pending_listen_freq) {
+                       p2p_dbg(p2p, "Do not reply to Probe Request frame that was received on %u MHz while waiting to start Listen state on %u MHz",
+                               rx_freq, p2p->pending_listen_freq);
+                       p2p_parse_free(&msg);
+                       return P2P_PREQ_NOT_LISTEN;
+               }
+
+               for (i = 0; i < msg.service_hash_count; i++) {
+                       if (p2p_service_find_asp(p2p, hash)) {
+                               p2p_dbg(p2p, "Service Hash match found: "
+                                       MACSTR, MAC2STR(hash));
+                               p2ps_svc_found = 1;
+                               break;
+                       }
+                       hash += P2PS_HASH_LEN;
+               }
+
+               /* Probed hash unknown */
+               if (!p2ps_svc_found) {
+                       p2p_dbg(p2p, "No Service Hash match found");
+                       p2p_parse_free(&msg);
+                       return P2P_PREQ_NOT_PROCESSED;
+               }
+       } else {
+               /* This is not a P2PS Probe Request */
+               p2p_dbg(p2p, "No P2PS Hash in Probe Request");
+
+               if (!p2p->in_listen || !p2p->drv_in_listen) {
+                       /* not in Listen state - ignore Probe Request */
+                       p2p_dbg(p2p, "Not in Listen state (in_listen=%d drv_in_listen=%d) - ignore Probe Request",
+                               p2p->in_listen, p2p->drv_in_listen);
+                       p2p_parse_free(&msg);
+                       return P2P_PREQ_NOT_LISTEN;
+               }
+       }
+
        if (msg.device_id &&
            os_memcmp(msg.device_id, p2p->cfg->dev_addr, ETH_ALEN) != 0) {
                /* Device ID did not match */
@@ -2238,11 +2436,11 @@ p2p_reply_probe(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
                p2p_parse_free(&msg);
                return P2P_PREQ_NOT_PROCESSED;
        }
-       p2p_parse_free(&msg);
 
        if (!p2p->cfg->send_probe_resp) {
                /* Response generated elsewhere */
                p2p_dbg(p2p, "Probe Resp generated elsewhere - do not generate additional response");
+               p2p_parse_free(&msg);
                return P2P_PREQ_NOT_PROCESSED;
        }
 
@@ -2254,7 +2452,9 @@ p2p_reply_probe(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
         * really only used for discovery purposes, not to learn exact BSS
         * parameters.
         */
-       ies = p2p_build_probe_resp_ies(p2p);
+       ies = p2p_build_probe_resp_ies(p2p, msg.service_hash,
+                                      msg.service_hash_count);
+       p2p_parse_free(&msg);
        if (ies == NULL)
                return P2P_PREQ_NOT_PROCESSED;
 
@@ -2264,61 +2464,49 @@ p2p_reply_probe(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
                return P2P_PREQ_NOT_PROCESSED;
        }
 
-       resp = NULL;
-       resp = wpabuf_put(buf, resp->u.probe_resp.variable - (u8 *) resp);
-
-       resp->frame_control = host_to_le16((WLAN_FC_TYPE_MGMT << 2) |
-                                          (WLAN_FC_STYPE_PROBE_RESP << 4));
-       os_memcpy(resp->da, addr, ETH_ALEN);
-       os_memcpy(resp->sa, p2p->cfg->dev_addr, ETH_ALEN);
-       os_memcpy(resp->bssid, p2p->cfg->dev_addr, ETH_ALEN);
-       resp->u.probe_resp.beacon_int = host_to_le16(100);
-       /* hardware or low-level driver will setup seq_ctrl and timestamp */
-       resp->u.probe_resp.capab_info =
-               host_to_le16(WLAN_CAPABILITY_SHORT_PREAMBLE |
-                            WLAN_CAPABILITY_PRIVACY |
-                            WLAN_CAPABILITY_SHORT_SLOT_TIME);
-
-       wpabuf_put_u8(buf, WLAN_EID_SSID);
-       wpabuf_put_u8(buf, P2P_WILDCARD_SSID_LEN);
-       wpabuf_put_data(buf, P2P_WILDCARD_SSID, P2P_WILDCARD_SSID_LEN);
-
-       wpabuf_put_u8(buf, WLAN_EID_SUPP_RATES);
-       wpabuf_put_u8(buf, 8);
-       wpabuf_put_u8(buf, (60 / 5) | 0x80);
-       wpabuf_put_u8(buf, 90 / 5);
-       wpabuf_put_u8(buf, (120 / 5) | 0x80);
-       wpabuf_put_u8(buf, 180 / 5);
-       wpabuf_put_u8(buf, (240 / 5) | 0x80);
-       wpabuf_put_u8(buf, 360 / 5);
-       wpabuf_put_u8(buf, 480 / 5);
-       wpabuf_put_u8(buf, 540 / 5);
-
-       wpabuf_put_u8(buf, WLAN_EID_DS_PARAMS);
-       wpabuf_put_u8(buf, 1);
-       wpabuf_put_u8(buf, p2p->cfg->channel);
+       if (p2p_build_probe_resp_buf(p2p, buf, ies, addr, rx_freq)) {
+               wpabuf_free(ies);
+               wpabuf_free(buf);
+               return P2P_PREQ_NOT_PROCESSED;
+       }
 
-       wpabuf_put_buf(buf, ies);
        wpabuf_free(ies);
 
-       p2p->cfg->send_probe_resp(p2p->cfg->cb_ctx, buf);
+       p2p->cfg->send_probe_resp(p2p->cfg->cb_ctx, buf, rx_freq);
 
        wpabuf_free(buf);
 
-       return P2P_PREQ_NOT_PROCESSED;
+       return P2P_PREQ_PROCESSED;
 }
 
 
 enum p2p_probe_req_status
 p2p_probe_req_rx(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
-                const u8 *bssid, const u8 *ie, size_t ie_len)
+                const u8 *bssid, const u8 *ie, size_t ie_len,
+                unsigned int rx_freq, int p2p_lo_started)
 {
        enum p2p_probe_req_status res;
 
        p2p_add_dev_from_probe_req(p2p, addr, ie, ie_len);
 
-       res = p2p_reply_probe(p2p, addr, dst, bssid, ie, ie_len);
+       if (p2p_lo_started) {
+               p2p_dbg(p2p,
+                       "Probe Response is offloaded, do not reply Probe Request");
+               return P2P_PREQ_PROCESSED;
+       }
+
+       res = p2p_reply_probe(p2p, addr, dst, bssid, ie, ie_len, rx_freq);
+       if (res != P2P_PREQ_PROCESSED && res != P2P_PREQ_NOT_PROCESSED)
+               return res;
 
+       /*
+        * Activate a pending GO Negotiation/Invite flow if a received Probe
+        * Request frame is from an expected peer. Some devices may share the
+        * same address for P2P and non-P2P STA running simultaneously. The
+        * P2P_PREQ_PROCESSED and P2P_PREQ_NOT_PROCESSED p2p_reply_probe()
+        * return values verified above ensure we are handling a Probe Request
+        * frame from a P2P peer.
+        */
        if ((p2p->state == P2P_CONNECT || p2p->state == P2P_CONNECT_LISTEN) &&
            p2p->go_neg_peer &&
            os_memcmp(addr, p2p->go_neg_peer->info.p2p_device_addr, ETH_ALEN)
@@ -2328,7 +2516,7 @@ p2p_probe_req_rx(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
                p2p_dbg(p2p, "Found GO Negotiation peer - try to start GO negotiation from timeout");
                eloop_cancel_timeout(p2p_go_neg_start, p2p, NULL);
                eloop_register_timeout(0, 0, p2p_go_neg_start, p2p, NULL);
-               return P2P_PREQ_PROCESSED;
+               return res;
        }
 
        if ((p2p->state == P2P_INVITE || p2p->state == P2P_INVITE_LISTEN) &&
@@ -2340,7 +2528,7 @@ p2p_probe_req_rx(struct p2p_data *p2p, const u8 *addr, const u8 *dst,
                p2p_dbg(p2p, "Found Invite peer - try to start Invite from timeout");
                eloop_cancel_timeout(p2p_invite_start, p2p, NULL);
                eloop_register_timeout(0, 0, p2p_invite_start, p2p, NULL);
-               return P2P_PREQ_PROCESSED;
+               return res;
        }
 
        return res;
@@ -2355,10 +2543,21 @@ static int p2p_assoc_req_ie_wlan_ap(struct p2p_data *p2p, const u8 *bssid,
        size_t tmplen;
        int res;
        u8 group_capab;
+       struct p2p_message msg;
 
        if (p2p_ie == NULL)
                return 0; /* WLAN AP is not a P2P manager */
 
+       os_memset(&msg, 0, sizeof(msg));
+       if (p2p_parse_p2p_ie(p2p_ie, &msg) < 0)
+               return 0;
+
+       p2p_dbg(p2p, "BSS P2P manageability %s",
+               msg.manageability ? "enabled" : "disabled");
+
+       if (!msg.manageability)
+               return 0;
+
        /*
         * (Re)Association Request - P2P IE
         * P2P Capability attribute (shall be present)
@@ -2521,13 +2720,14 @@ int p2p_service_del_asp(struct p2p_data *p2p, u32 adv_id)
 
 int p2p_service_add_asp(struct p2p_data *p2p, int auto_accept, u32 adv_id,
                        const char *adv_str, u8 svc_state, u16 config_methods,
-                       const char *svc_info)
+                       const char *svc_info, const u8 *cpt_priority)
 {
        struct p2ps_advertisement *adv_data, *tmp, **prev;
        u8 buf[P2PS_HASH_LEN];
        size_t adv_data_len, adv_len, info_len = 0;
+       int i;
 
-       if (!p2p || !adv_str || !adv_str[0])
+       if (!p2p || !adv_str || !adv_str[0] || !cpt_priority)
                return -1;
 
        if (!(config_methods & p2p->cfg->config_methods)) {
@@ -2556,6 +2756,11 @@ int p2p_service_add_asp(struct p2p_data *p2p, int auto_accept, u32 adv_id,
        adv_data->auto_accept = (u8) auto_accept;
        os_memcpy(adv_data->svc_name, adv_str, adv_len);
 
+       for (i = 0; cpt_priority[i] && i < P2PS_FEATURE_CAPAB_CPT_MAX; i++) {
+               adv_data->cpt_priority[i] = cpt_priority[i];
+               adv_data->cpt_mask |= cpt_priority[i];
+       }
+
        if (svc_info && info_len) {
                adv_data->svc_info = &adv_data->svc_name[adv_len + 1];
                os_memcpy(adv_data->svc_info, svc_info, info_len);
@@ -2594,13 +2799,33 @@ int p2p_service_add_asp(struct p2p_data *p2p, int auto_accept, u32 adv_id,
 
 inserted:
        p2p_dbg(p2p,
-               "Added ASP advertisement adv_id=0x%x config_methods=0x%x svc_state=0x%x adv_str='%s'",
-               adv_id, adv_data->config_methods, svc_state, adv_str);
+               "Added ASP advertisement adv_id=0x%x config_methods=0x%x svc_state=0x%x adv_str='%s' cpt_mask=0x%x",
+               adv_id, adv_data->config_methods, svc_state, adv_str,
+               adv_data->cpt_mask);
 
        return 0;
 }
 
 
+void p2p_service_flush_asp(struct p2p_data *p2p)
+{
+       struct p2ps_advertisement *adv, *prev;
+
+       if (!p2p)
+               return;
+
+       adv = p2p->p2ps_adv_list;
+       while (adv) {
+               prev = adv;
+               adv = adv->next;
+               os_free(prev);
+       }
+
+       p2p->p2ps_adv_list = NULL;
+       p2p_dbg(p2p, "All ASP advertisements flushed");
+}
+
+
 int p2p_parse_dev_addr_in_p2p_ie(struct wpabuf *p2p_ie, u8 *dev_addr)
 {
        struct p2p_message msg;
@@ -2732,9 +2957,6 @@ struct p2p_data * p2p_init(const struct p2p_config *cfg)
 
        dl_list_init(&p2p->devices);
 
-       eloop_register_timeout(P2P_PEER_EXPIRATION_INTERVAL, 0,
-                              p2p_expiration_timeout, p2p, NULL);
-
        p2p->go_timeout = 100;
        p2p->client_timeout = 20;
        p2p->num_p2p_sd_queries = 0;
@@ -2749,8 +2971,6 @@ struct p2p_data * p2p_init(const struct p2p_config *cfg)
 
 void p2p_deinit(struct p2p_data *p2p)
 {
-       struct p2ps_advertisement *adv, *prev;
-
 #ifdef CONFIG_WIFI_DISPLAY
        wpabuf_free(p2p->wfd_ie_beacon);
        wpabuf_free(p2p->wfd_ie_probe_req);
@@ -2765,8 +2985,6 @@ void p2p_deinit(struct p2p_data *p2p)
        wpabuf_free(p2p->wfd_coupled_sink_info);
 #endif /* CONFIG_WIFI_DISPLAY */
 
-       eloop_cancel_timeout(p2p_expiration_timeout, p2p, NULL);
-       eloop_cancel_timeout(p2p_ext_listen_timeout, p2p, NULL);
        eloop_cancel_timeout(p2p_scan_timeout, p2p, NULL);
        eloop_cancel_timeout(p2p_go_neg_start, p2p, NULL);
        eloop_cancel_timeout(p2p_go_neg_wait_timeout, p2p, NULL);
@@ -2779,17 +2997,12 @@ void p2p_deinit(struct p2p_data *p2p)
        os_free(p2p->cfg->serial_number);
        os_free(p2p->cfg->pref_chan);
        os_free(p2p->groups);
+       p2ps_prov_free(p2p);
        wpabuf_free(p2p->sd_resp);
        os_free(p2p->after_scan_tx);
        p2p_remove_wps_vendor_extensions(p2p);
        os_free(p2p->no_go_freq.range);
-
-       adv = p2p->p2ps_adv_list;
-       while (adv) {
-               prev = adv;
-               adv = adv->next;
-               os_free(prev);
-       }
+       p2p_service_flush_asp(p2p);
 
        os_free(p2p);
 }
@@ -2798,6 +3011,8 @@ void p2p_deinit(struct p2p_data *p2p)
 void p2p_flush(struct p2p_data *p2p)
 {
        struct p2p_device *dev, *prev;
+
+       p2p_ext_listen(p2p, 0, 0);
        p2p_stop_find(p2p);
        dl_list_for_each_safe(dev, prev, &p2p->devices, struct p2p_device,
                              list) {
@@ -2807,6 +3022,7 @@ void p2p_flush(struct p2p_data *p2p)
        p2p_free_sd_queries(p2p);
        os_free(p2p->after_scan_tx);
        p2p->after_scan_tx = NULL;
+       p2p->ssid_set = 0;
 }
 
 
@@ -2983,13 +3199,18 @@ int p2p_set_country(struct p2p_data *p2p, const char *country)
 
 static int p2p_pre_find_operation(struct p2p_data *p2p, struct p2p_device *dev)
 {
+       int res;
+
        if (dev->sd_pending_bcast_queries == 0) {
                /* Initialize with total number of registered broadcast
                 * SD queries. */
                dev->sd_pending_bcast_queries = p2p->num_p2p_sd_queries;
        }
 
-       if (p2p_start_sd(p2p, dev) == 0)
+       res = p2p_start_sd(p2p, dev);
+       if (res == -2)
+               return -2;
+       if (res == 0)
                return 1;
 
        if (dev->req_config_methods &&
@@ -3009,7 +3230,7 @@ static int p2p_pre_find_operation(struct p2p_data *p2p, struct p2p_device *dev)
 void p2p_continue_find(struct p2p_data *p2p)
 {
        struct p2p_device *dev;
-       int found;
+       int found, res;
 
        p2p_set_state(p2p, P2P_SEARCH);
 
@@ -3022,10 +3243,13 @@ void p2p_continue_find(struct p2p_data *p2p)
                }
                if (!found)
                        continue;
-               if (p2p_pre_find_operation(p2p, dev) > 0) {
+               res = p2p_pre_find_operation(p2p, dev);
+               if (res > 0) {
                        p2p->last_p2p_find_oper = dev;
                        return;
                }
+               if (res == -2)
+                       goto skip_sd;
        }
 
        /*
@@ -3033,14 +3257,19 @@ void p2p_continue_find(struct p2p_data *p2p)
         * iteration device.
         */
        dl_list_for_each(dev, &p2p->devices, struct p2p_device, list) {
-               if (p2p_pre_find_operation(p2p, dev) > 0) {
+               res = p2p_pre_find_operation(p2p, dev);
+               if (res > 0) {
                        p2p->last_p2p_find_oper = dev;
                        return;
                }
+               if (res == -2)
+                       goto skip_sd;
                if (dev == p2p->last_p2p_find_oper)
                        break;
        }
 
+skip_sd:
+       os_memset(p2p->sd_query_no_ack, 0, ETH_ALEN);
        p2p_listen_in_find(p2p, 1);
 }
 
@@ -3052,16 +3281,27 @@ static void p2p_sd_cb(struct p2p_data *p2p, int success)
        p2p->pending_action_state = P2P_NO_PENDING_ACTION;
 
        if (!success) {
-               if (p2p->sd_peer)
+               if (p2p->sd_peer) {
+                       if (is_zero_ether_addr(p2p->sd_query_no_ack)) {
+                               os_memcpy(p2p->sd_query_no_ack,
+                                         p2p->sd_peer->info.p2p_device_addr,
+                                         ETH_ALEN);
+                               p2p_dbg(p2p,
+                                       "First SD Query no-ACK in this search iteration: "
+                                       MACSTR, MAC2STR(p2p->sd_query_no_ack));
+                       }
                        p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
+               }
                p2p->sd_peer = NULL;
-               p2p_continue_find(p2p);
+               if (p2p->state != P2P_IDLE)
+                       p2p_continue_find(p2p);
                return;
        }
 
        if (p2p->sd_peer == NULL) {
                p2p_dbg(p2p, "No SD peer entry known");
-               p2p_continue_find(p2p);
+               if (p2p->state != P2P_IDLE)
+                       p2p_continue_find(p2p);
                return;
        }
 
@@ -3092,9 +3332,6 @@ static void p2p_retry_pd(struct p2p_data *p2p)
 {
        struct p2p_device *dev;
 
-       if (p2p->state != P2P_IDLE)
-               return;
-
        /*
         * Retry the prov disc req attempt only for the peer that the user had
         * requested.
@@ -3153,6 +3390,43 @@ static void p2p_prov_disc_cb(struct p2p_data *p2p, int success)
        }
 
        /*
+        * If after PD Request the peer doesn't expect to receive PD Response
+        * the PD Request ACK indicates a completion of the current PD. This
+        * happens only on the advertiser side sending the follow-on PD Request
+        * with the status different than 12 (Success: accepted by user).
+        */
+       if (p2p->p2ps_prov && !p2p->p2ps_prov->pd_seeker &&
+           p2p->p2ps_prov->status != P2P_SC_SUCCESS_DEFERRED) {
+               p2p_dbg(p2p, "P2PS PD completion on Follow-on PD Request ACK");
+
+               if (p2p->send_action_in_progress) {
+                       p2p->send_action_in_progress = 0;
+                       p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
+               }
+
+               p2p->pending_action_state = P2P_NO_PENDING_ACTION;
+
+               if (p2p->cfg->p2ps_prov_complete) {
+                       p2p->cfg->p2ps_prov_complete(
+                               p2p->cfg->cb_ctx,
+                               p2p->p2ps_prov->status,
+                               p2p->p2ps_prov->adv_mac,
+                               p2p->p2ps_prov->adv_mac,
+                               p2p->p2ps_prov->session_mac,
+                               NULL, p2p->p2ps_prov->adv_id,
+                               p2p->p2ps_prov->session_id,
+                               0, 0, NULL, 0, 0, 0,
+                               NULL, NULL, 0, 0, NULL, 0);
+               }
+
+               if (p2p->user_initiated_pd)
+                       p2p_reset_pending_pd(p2p);
+
+               p2ps_prov_free(p2p);
+               return;
+       }
+
+       /*
         * This postponing, of resetting pending_action_state, needs to be
         * done only for user initiated PD requests and not internal ones.
         */
@@ -3168,6 +3442,51 @@ static void p2p_prov_disc_cb(struct p2p_data *p2p, int success)
 }
 
 
+static int p2p_check_after_scan_tx_continuation(struct p2p_data *p2p)
+{
+       if (p2p->after_scan_tx_in_progress) {
+               p2p->after_scan_tx_in_progress = 0;
+               if (p2p->start_after_scan != P2P_AFTER_SCAN_NOTHING &&
+                   p2p_run_after_scan(p2p))
+                       return 1;
+               if (p2p->state == P2P_SEARCH) {
+                       p2p_dbg(p2p, "Continue find after after_scan_tx completion");
+                       p2p_continue_find(p2p);
+               }
+       }
+
+       return 0;
+}
+
+
+static void p2p_prov_disc_resp_cb(struct p2p_data *p2p, int success)
+{
+       p2p_dbg(p2p, "Provision Discovery Response TX callback: success=%d",
+               success);
+
+       if (p2p->send_action_in_progress) {
+               p2p->send_action_in_progress = 0;
+               p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
+       }
+
+       p2p->pending_action_state = P2P_NO_PENDING_ACTION;
+
+       if (!success)
+               goto continue_search;
+
+       if (!p2p->cfg->prov_disc_resp_cb ||
+           p2p->cfg->prov_disc_resp_cb(p2p->cfg->cb_ctx) < 1)
+               goto continue_search;
+
+       p2p_dbg(p2p,
+               "Post-Provision Discovery operations started - do not try to continue other P2P operations");
+       return;
+
+continue_search:
+       p2p_check_after_scan_tx_continuation(p2p);
+}
+
+
 int p2p_scan_res_handler(struct p2p_data *p2p, const u8 *bssid, int freq,
                         struct os_reltime *rx_time, int level, const u8 *ies,
                         size_t ies_len)
@@ -3181,9 +3500,11 @@ int p2p_scan_res_handler(struct p2p_data *p2p, const u8 *bssid, int freq,
                 * operation was started.
                 */
                p2p_dbg(p2p, "Ignore old scan result for " MACSTR
-                       " (rx_time=%u.%06u)",
+                       " (rx_time=%u.%06u find_start=%u.%06u)",
                        MAC2STR(bssid), (unsigned int) rx_time->sec,
-                       (unsigned int) rx_time->usec);
+                       (unsigned int) rx_time->usec,
+                       (unsigned int) p2p->find_start.sec,
+                       (unsigned int) p2p->find_start.usec);
                return 0;
        }
 
@@ -3208,7 +3529,8 @@ void p2p_scan_res_handled(struct p2p_data *p2p)
 }
 
 
-void p2p_scan_ie(struct p2p_data *p2p, struct wpabuf *ies, const u8 *dev_id)
+void p2p_scan_ie(struct p2p_data *p2p, struct wpabuf *ies, const u8 *dev_id,
+                unsigned int bands)
 {
        u8 dev_capab;
        u8 *len;
@@ -3242,6 +3564,9 @@ void p2p_scan_ie(struct p2p_data *p2p, struct wpabuf *ies, const u8 *dev_id)
                p2p_buf_add_ext_listen_timing(ies, p2p->ext_listen_period,
                                              p2p->ext_listen_interval);
 
+       if (bands & BAND_60_GHZ)
+               p2p_buf_add_device_info(ies, p2p, NULL);
+
        if (p2p->p2ps_seek && p2p->p2ps_seek_count)
                p2p_buf_add_service_hash(ies, p2p);
 
@@ -3347,13 +3672,19 @@ static void p2p_go_neg_resp_failure_cb(struct p2p_data *p2p, int success,
        p2p_dbg(p2p, "GO Negotiation Response (failure) TX callback: success=%d", success);
        if (p2p->go_neg_peer && p2p->go_neg_peer->status != P2P_SC_SUCCESS) {
                p2p_go_neg_failed(p2p, p2p->go_neg_peer->status);
-       } else if (success) {
+               return;
+       }
+
+       if (success) {
                struct p2p_device *dev;
                dev = p2p_get_device(p2p, addr);
                if (dev &&
                    dev->status == P2P_SC_FAIL_INFO_CURRENTLY_UNAVAILABLE)
                        dev->flags |= P2P_DEV_PEER_WAITING_RESPONSE;
        }
+
+       if (p2p->state == P2P_SEARCH || p2p->state == P2P_SD_DURING_FIND)
+               p2p_continue_find(p2p);
 }
 
 
@@ -3442,16 +3773,7 @@ void p2p_send_action_cb(struct p2p_data *p2p, unsigned int freq, const u8 *dst,
                        p2p->send_action_in_progress = 0;
                        p2p->cfg->send_action_done(p2p->cfg->cb_ctx);
                }
-               if (p2p->after_scan_tx_in_progress) {
-                       p2p->after_scan_tx_in_progress = 0;
-                       if (p2p->start_after_scan != P2P_AFTER_SCAN_NOTHING &&
-                           p2p_run_after_scan(p2p))
-                               break;
-                       if (p2p->state == P2P_SEARCH) {
-                               p2p_dbg(p2p, "Continue find after after_scan_tx completion");
-                               p2p_continue_find(p2p);
-                       }
-               }
+               p2p_check_after_scan_tx_continuation(p2p);
                break;
        case P2P_PENDING_GO_NEG_REQUEST:
                p2p_go_neg_req_cb(p2p, success);
@@ -3471,11 +3793,16 @@ void p2p_send_action_cb(struct p2p_data *p2p, unsigned int freq, const u8 *dst,
        case P2P_PENDING_PD:
                p2p_prov_disc_cb(p2p, success);
                break;
+       case P2P_PENDING_PD_RESPONSE:
+               p2p_prov_disc_resp_cb(p2p, success);
+               break;
        case P2P_PENDING_INVITATION_REQUEST:
                p2p_invitation_req_cb(p2p, success);
                break;
        case P2P_PENDING_INVITATION_RESPONSE:
                p2p_invitation_resp_cb(p2p, success);
+               if (p2p->inv_status != P2P_SC_SUCCESS)
+                       p2p_check_after_scan_tx_continuation(p2p);
                break;
        case P2P_PENDING_DEV_DISC_REQUEST:
                p2p_dev_disc_req_cb(p2p, success);
@@ -3677,6 +4004,9 @@ static void p2p_timeout_prov_disc_during_find(struct p2p_data *p2p)
 
 static void p2p_timeout_prov_disc_req(struct p2p_data *p2p)
 {
+       u32 adv_id = 0;
+       u8 *adv_mac = NULL;
+
        p2p->pending_action_state = P2P_NO_PENDING_ACTION;
 
        /*
@@ -3705,12 +4035,18 @@ static void p2p_timeout_prov_disc_req(struct p2p_data *p2p)
                                for_join = 1;
                }
 
+               if (p2p->p2ps_prov) {
+                       adv_id = p2p->p2ps_prov->adv_id;
+                       adv_mac = p2p->p2ps_prov->adv_mac;
+               }
+
                if (p2p->cfg->prov_disc_fail)
                        p2p->cfg->prov_disc_fail(p2p->cfg->cb_ctx,
                                                 p2p->pending_pd_devaddr,
                                                 for_join ?
                                                 P2P_PROV_DISC_TIMEOUT_JOIN :
-                                                P2P_PROV_DISC_TIMEOUT);
+                                                P2P_PROV_DISC_TIMEOUT,
+                                                adv_id, adv_mac, NULL);
                p2p_reset_pending_pd(p2p);
        }
 }
@@ -3937,7 +4273,7 @@ int p2p_get_peer_info_txt(const struct p2p_peer_info *info,
                          "country=%c%c\n"
                          "oper_freq=%d\n"
                          "req_config_methods=0x%x\n"
-                         "flags=%s%s%s%s%s%s%s%s%s%s%s%s%s\n"
+                         "flags=%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s\n"
                          "status=%d\n"
                          "invitation_reqs=%u\n",
                          (int) (now.sec - dev->last_seen.sec),
@@ -3963,6 +4299,8 @@ int p2p_get_peer_info_txt(const struct p2p_peer_info *info,
                          "[PD_PEER_DISPLAY]" : "",
                          dev->flags & P2P_DEV_PD_PEER_KEYPAD ?
                          "[PD_PEER_KEYPAD]" : "",
+                         dev->flags & P2P_DEV_PD_PEER_P2PS ?
+                         "[PD_PEER_P2PS]" : "",
                          dev->flags & P2P_DEV_USER_REJECTED ?
                          "[USER_REJECTED]" : "",
                          dev->flags & P2P_DEV_PEER_WAITING_RESPONSE ?
@@ -3979,6 +4317,8 @@ int p2p_get_peer_info_txt(const struct p2p_peer_info *info,
                          "[FORCE_FREQ]" : "",
                          dev->flags & P2P_DEV_PD_FOR_JOIN ?
                          "[PD_FOR_JOIN]" : "",
+                         dev->flags & P2P_DEV_LAST_SEEN_AS_GROUP_CLIENT ?
+                         "[LAST_SEEN_AS_GROUP_CLIENT]" : "",
                          dev->status,
                          dev->invitation_reqs);
        if (os_snprintf_error(end - pos, res))
@@ -4394,15 +4734,18 @@ int p2p_set_listen_channel(struct p2p_data *p2p, u8 reg_class, u8 channel,
        if (p2p_channel_to_freq(reg_class, channel) < 0)
                return -1;
 
-       p2p_dbg(p2p, "Set Listen channel: reg_class %u channel %u",
-               reg_class, channel);
-
        /*
         * Listen channel was set in configuration or set by control interface;
         * cannot override it.
         */
-       if (p2p->cfg->channel_forced && forced == 0)
+       if (p2p->cfg->channel_forced && forced == 0) {
+               p2p_dbg(p2p,
+                       "Listen channel was previously configured - do not override based on optimization");
                return -1;
+       }
+
+       p2p_dbg(p2p, "Set Listen channel: reg_class %u channel %u",
+               reg_class, channel);
 
        if (p2p->state == P2P_IDLE) {
                p2p->cfg->reg_class = reg_class;
@@ -5027,6 +5370,7 @@ int p2p_process_nfc_connection_handover(struct p2p_data *p2p,
 
        if (!msg.oob_go_neg_channel) {
                p2p_dbg(p2p, "OOB GO Negotiation Channel attribute not included");
+               p2p_parse_free(&msg);
                return -1;
        }
 
@@ -5038,6 +5382,7 @@ int p2p_process_nfc_connection_handover(struct p2p_data *p2p,
                                           msg.oob_go_neg_channel[4]);
        if (freq < 0) {
                p2p_dbg(p2p, "Unknown peer OOB GO Neg channel");
+               p2p_parse_free(&msg);
                return -1;
        }
        role = msg.oob_go_neg_channel[5];
@@ -5058,6 +5403,7 @@ int p2p_process_nfc_connection_handover(struct p2p_data *p2p,
                                           p2p->cfg->channel);
                if (freq < 0) {
                        p2p_dbg(p2p, "Own listen channel not known");
+                       p2p_parse_free(&msg);
                        return -1;
                }
                p2p_dbg(p2p, "Use own Listen channel as OOB GO Neg channel: %u MHz", freq);
@@ -5146,3 +5492,51 @@ void p2p_go_neg_wait_timeout(void *eloop_ctx, void *timeout_ctx)
                "Timeout on waiting peer to become ready for GO Negotiation");
        p2p_go_neg_failed(p2p, -1);
 }
+
+
+void p2p_set_own_pref_freq_list(struct p2p_data *p2p,
+                               const unsigned int *pref_freq_list,
+                               unsigned int size)
+{
+       unsigned int i;
+
+       if (size > P2P_MAX_PREF_CHANNELS)
+               size = P2P_MAX_PREF_CHANNELS;
+       p2p->num_pref_freq = size;
+       for (i = 0; i < size; i++) {
+               p2p->pref_freq_list[i] = pref_freq_list[i];
+               p2p_dbg(p2p, "Own preferred frequency list[%u]=%u MHz",
+                       i, p2p->pref_freq_list[i]);
+       }
+}
+
+
+struct wpabuf * p2p_build_probe_resp_template(struct p2p_data *p2p,
+                                             unsigned int freq)
+{
+       struct wpabuf *ies, *buf;
+       u8 addr[] = { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF };
+       int ret;
+
+       ies = p2p_build_probe_resp_ies(p2p, NULL, 0);
+       if (!ies) {
+               wpa_printf(MSG_ERROR,
+                          "CTRL: Failed to build Probe Response IEs");
+               return NULL;
+       }
+
+       buf = wpabuf_alloc(200 + wpabuf_len(ies));
+       if (!buf) {
+               wpabuf_free(ies);
+               return NULL;
+       }
+
+       ret = p2p_build_probe_resp_buf(p2p, buf, ies, addr, freq);
+       wpabuf_free(ies);
+       if (ret) {
+               wpabuf_free(buf);
+               return NULL;
+       }
+
+       return buf;
+}