WPS: Use blacklist more aggressively during WPS provisioning
[libeap.git] / wpa_supplicant / events.c
index 167e633..dfded7d 100644 (file)
@@ -36,6 +36,7 @@
 #include "wps_supplicant.h"
 #include "ibss_rsn.h"
 #include "sme.h"
+#include "p2p_supplicant.h"
 #include "bgscan.h"
 #include "ap.h"
 #include "bss.h"
@@ -462,6 +463,11 @@ static struct wpa_ssid * wpa_scan_res_match(struct wpa_supplicant *wpa_s,
                }
 
 #ifdef CONFIG_WPS
+               if ((ssid->key_mgmt & WPA_KEY_MGMT_WPS) && e && e->count > 0) {
+                       wpa_printf(MSG_DEBUG, "   skip - blacklisted (WPS)");
+                       continue;
+               }
+
                if (wpa && ssid->ssid_len == 0 &&
                    wpas_wps_ssid_wildcard_ok(wpa_s, ssid, bss))
                        check_ssid = 0;
@@ -622,7 +628,11 @@ void wpa_supplicant_connect(struct wpa_supplicant *wpa_s,
        if (wpas_wps_scan_pbc_overlap(wpa_s, selected, ssid)) {
                wpa_msg(wpa_s, MSG_INFO, WPS_EVENT_OVERLAP
                        "PBC session overlap");
-               wpa_supplicant_req_new_scan(wpa_s, 10, 0);
+#ifdef CONFIG_P2P
+               if (wpas_p2p_notif_pbc_overlap(wpa_s) == 1)
+                       return;
+#endif /* CONFIG_P2P */
+               wpas_wps_cancel(wpa_s);
                return;
        }
 
@@ -1220,8 +1230,19 @@ static void wpa_supplicant_event_disassoc(struct wpa_supplicant *wpa_s,
                wpa_msg(wpa_s, MSG_INFO, "WPA: 4-Way Handshake failed - "
                        "pre-shared key may be incorrect");
        }
-       if (wpa_s->wpa_state >= WPA_ASSOCIATED)
-               wpa_supplicant_req_scan(wpa_s, 0, 100000);
+       if (!wpa_s->auto_reconnect_disabled ||
+           wpa_s->key_mgmt == WPA_KEY_MGMT_WPS) {
+               wpa_printf(MSG_DEBUG, "WPA: Auto connect enabled: try to "
+                          "reconnect (wps=%d)",
+                          wpa_s->key_mgmt == WPA_KEY_MGMT_WPS);
+               if (wpa_s->wpa_state >= WPA_ASSOCIATING)
+                       wpa_supplicant_req_scan(wpa_s, 0, 100000);
+       } else {
+               wpa_printf(MSG_DEBUG, "WPA: Auto connect disabled: do not try "
+                          "to re-connect");
+               wpa_s->reassociate = 0;
+               wpa_s->disconnected = 1;
+       }
        bssid = wpa_s->bssid;
        if (is_zero_ether_addr(bssid))
                bssid = wpa_s->pending_bssid;
@@ -1546,8 +1567,18 @@ void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
                        break;
                }
 #endif /* CONFIG_AP */
-               if (data)
-                       reason_code = data->deauth_info.reason_code;
+               if (data) {
+                       reason_code = data->disassoc_info.reason_code;
+                       wpa_hexdump(MSG_DEBUG, "Disassociation frame IE(s)",
+                                   data->disassoc_info.ie,
+                                   data->disassoc_info.ie_len);
+#ifdef CONFIG_P2P
+                       wpas_p2p_disassoc_notif(
+                               wpa_s, data->disassoc_info.addr, reason_code,
+                               data->disassoc_info.ie,
+                               data->disassoc_info.ie_len);
+#endif /* CONFIG_P2P */
+               }
                if (wpa_s->drv_flags & WPA_DRIVER_FLAGS_SME)
                        sme_event_disassoc(wpa_s, data);
                /* fall through */
@@ -1564,6 +1595,17 @@ void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
                                                   MAC2STR(data->deauth_info.
                                                           addr));
                                }
+                               wpa_hexdump(MSG_DEBUG,
+                                           "Deauthentication frame IE(s)",
+                                           data->deauth_info.ie,
+                                           data->deauth_info.ie_len);
+#ifdef CONFIG_P2P
+                               wpas_p2p_deauth_notif(
+                                       wpa_s, data->deauth_info.addr,
+                                       reason_code,
+                                       data->deauth_info.ie,
+                                       data->deauth_info.ie_len);
+#endif /* CONFIG_P2P */
                        }
                }
 #ifdef CONFIG_AP
@@ -1618,8 +1660,40 @@ void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
                break;
 #ifdef CONFIG_AP
        case EVENT_TX_STATUS:
-               if (wpa_s->ap_iface == NULL)
+               wpa_printf(MSG_DEBUG, "EVENT_TX_STATUS on %s dst=" MACSTR
+                          " type=%d stype=%d pending_dst=" MACSTR,
+                          wpa_s->ifname, MAC2STR(data->tx_status.dst),
+                          data->tx_status.type, data->tx_status.stype,
+                          MAC2STR(wpa_s->parent->pending_action_dst));
+               if (wpa_s->ap_iface == NULL) {
+#ifdef CONFIG_P2P
+                       if (data->tx_status.type == WLAN_FC_TYPE_MGMT &&
+                           data->tx_status.stype == WLAN_FC_STYPE_ACTION)
+                               wpas_send_action_tx_status(
+                                       wpa_s, data->tx_status.dst,
+                                       data->tx_status.data,
+                                       data->tx_status.data_len,
+                                       data->tx_status.ack);
+#endif /* CONFIG_P2P */
                        break;
+               }
+#ifdef CONFIG_P2P
+               /*
+                * Catch TX status events for Action frames we sent via group
+                * interface in GO mode.
+                */
+               if (data->tx_status.type == WLAN_FC_TYPE_MGMT &&
+                   data->tx_status.stype == WLAN_FC_STYPE_ACTION &&
+                   os_memcmp(wpa_s->parent->pending_action_dst,
+                             data->tx_status.dst, ETH_ALEN) == 0) {
+                       wpas_send_action_tx_status(
+                               wpa_s->parent, data->tx_status.dst,
+                               data->tx_status.data,
+                               data->tx_status.data_len,
+                               data->tx_status.ack);
+                       break;
+               }
+#endif /* CONFIG_P2P */
                switch (data->tx_status.type) {
                case WLAN_FC_TYPE_MGMT:
                        ap_mgmt_tx_cb(wpa_s, data->tx_status.data,
@@ -1642,8 +1716,29 @@ void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
                                       data->rx_from_unknown.len);
                break;
        case EVENT_RX_MGMT:
-               if (wpa_s->ap_iface == NULL)
+               if (wpa_s->ap_iface == NULL) {
+#ifdef CONFIG_P2P
+                       u16 fc, stype;
+                       const struct ieee80211_mgmt *mgmt;
+                       mgmt = (const struct ieee80211_mgmt *)
+                               data->rx_mgmt.frame;
+                       fc = le_to_host16(mgmt->frame_control);
+                       stype = WLAN_FC_GET_STYPE(fc);
+                       if (stype == WLAN_FC_STYPE_PROBE_REQ &&
+                           data->rx_mgmt.frame_len > 24) {
+                               const u8 *src = mgmt->sa;
+                               const u8 *ie = mgmt->u.probe_req.variable;
+                               size_t ie_len = data->rx_mgmt.frame_len -
+                                       (mgmt->u.probe_req.variable -
+                                        data->rx_mgmt.frame);
+                               wpas_p2p_probe_req_rx(wpa_s, src, ie, ie_len);
+                               break;
+                       }
+#endif /* CONFIG_P2P */
+                       wpa_printf(MSG_DEBUG, "AP: ignore received management "
+                                  "frame in non-AP mode");
                        break;
+               }
                ap_mgmt_rx(wpa_s, &data->rx_mgmt);
                break;
 #endif /* CONFIG_AP */
@@ -1660,7 +1755,31 @@ void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
                        break;
                }
 #endif /* CONFIG_IEEE80211R */
+#ifdef CONFIG_P2P
+               wpas_p2p_rx_action(wpa_s, data->rx_action.da,
+                                  data->rx_action.sa,
+                                  data->rx_action.bssid,
+                                  data->rx_action.category,
+                                  data->rx_action.data,
+                                  data->rx_action.len, data->rx_action.freq);
+#endif /* CONFIG_P2P */
                break;
+#ifdef CONFIG_P2P
+       case EVENT_REMAIN_ON_CHANNEL:
+               wpas_p2p_remain_on_channel_cb(
+                       wpa_s, data->remain_on_channel.freq,
+                       data->remain_on_channel.duration);
+               break;
+       case EVENT_CANCEL_REMAIN_ON_CHANNEL:
+               wpas_p2p_cancel_remain_on_channel_cb(
+                       wpa_s, data->remain_on_channel.freq);
+               break;
+       case EVENT_RX_PROBE_REQ:
+               wpas_p2p_probe_req_rx(wpa_s, data->rx_probe_req.sa,
+                                     data->rx_probe_req.ie,
+                                     data->rx_probe_req.ie_len);
+               break;
+#endif /* CONFIG_P2P */
 #ifdef CONFIG_CLIENT_MLME
        case EVENT_MLME_RX: {
                struct ieee80211_rx_status rx_status;