P2P: Extend P2P manager functionality to work with driver MLME
[libeap.git] / wpa_supplicant / bgscan_learn.c
index 0ffe383..492ce38 100644 (file)
@@ -29,6 +29,8 @@ struct bgscan_learn_bss {
        struct dl_list list;
        u8 bssid[ETH_ALEN];
        int freq;
+       u8 *neigh; /* num_neigh * ETH_ALEN buffer */
+       size_t num_neigh;
 };
 
 struct bgscan_learn_data {
@@ -46,6 +48,49 @@ struct bgscan_learn_data {
 };
 
 
+static void bss_free(struct bgscan_learn_bss *bss)
+{
+       os_free(bss->neigh);
+       os_free(bss);
+}
+
+
+static int bssid_in_array(u8 *array, size_t array_len, const u8 *bssid)
+{
+       size_t i;
+
+       if (array == NULL || array_len == 0)
+               return 0;
+
+       for (i = 0; i < array_len; i++) {
+               if (os_memcmp(array + i * ETH_ALEN, bssid, ETH_ALEN) == 0)
+                       return 1;
+       }
+
+       return 0;
+}
+
+
+static void bgscan_learn_add_neighbor(struct bgscan_learn_bss *bss,
+                                     const u8 *bssid)
+{
+       u8 *n;
+
+       if (os_memcmp(bss->bssid, bssid, ETH_ALEN) == 0)
+               return;
+       if (bssid_in_array(bss->neigh, bss->num_neigh, bssid))
+               return;
+
+       n = os_realloc(bss->neigh, (bss->num_neigh + 1) * ETH_ALEN);
+       if (n == NULL)
+               return;
+
+       os_memcpy(n + bss->num_neigh * ETH_ALEN, bssid, ETH_ALEN);
+       bss->neigh = n;
+       bss->num_neigh++;
+}
+
+
 static struct bgscan_learn_bss * bgscan_learn_get_bss(
        struct bgscan_learn_data *data, const u8 *bssid)
 {
@@ -63,6 +108,7 @@ static int bgscan_learn_load(struct bgscan_learn_data *data)
 {
        FILE *f;
        char buf[128];
+       struct bgscan_learn_bss *bss;
 
        if (data->fname == NULL)
                return 0;
@@ -84,12 +130,11 @@ static int bgscan_learn_load(struct bgscan_learn_data *data)
 
        while (fgets(buf, sizeof(buf), f)) {
                if (os_strncmp(buf, "BSS ", 4) == 0) {
-                       struct bgscan_learn_bss *bss;
                        bss = os_zalloc(sizeof(*bss));
                        if (!bss)
                                continue;
                        if (hwaddr_aton(buf + 4, bss->bssid) < 0) {
-                               os_free(bss);
+                               bss_free(bss);
                                continue;
                        }
                        bss->freq = atoi(buf + 4 + 18);
@@ -98,6 +143,20 @@ static int bgscan_learn_load(struct bgscan_learn_data *data)
                                   "entry: " MACSTR " freq=%d",
                                   MAC2STR(bss->bssid), bss->freq);
                }
+
+               if (os_strncmp(buf, "NEIGHBOR ", 9) == 0) {
+                       u8 addr[ETH_ALEN];
+
+                       if (hwaddr_aton(buf + 9, addr) < 0)
+                               continue;
+                       bss = bgscan_learn_get_bss(data, addr);
+                       if (bss == NULL)
+                               continue;
+                       if (hwaddr_aton(buf + 9 + 18, addr) < 0)
+                               continue;
+
+                       bgscan_learn_add_neighbor(bss, addr);
+               }
        }
 
        fclose(f);
@@ -126,6 +185,15 @@ static void bgscan_learn_save(struct bgscan_learn_data *data)
                        MAC2STR(bss->bssid), bss->freq);
        }
 
+       dl_list_for_each(bss, &data->bss, struct bgscan_learn_bss, list) {
+               size_t i;
+               for (i = 0; i < bss->num_neigh; i++) {
+                       fprintf(f, "NEIGHBOR " MACSTR " " MACSTR "\n",
+                               MAC2STR(bss->bssid),
+                               MAC2STR(bss->neigh + i * ETH_ALEN));
+               }
+       }
+
        fclose(f);
 }
 
@@ -360,6 +428,15 @@ static void * bgscan_learn_init(struct wpa_supplicant *wpa_s,
        data->scan_interval = data->short_interval;
        eloop_register_timeout(data->scan_interval, 0, bgscan_learn_timeout,
                               data, NULL);
+
+       /*
+        * This function is called immediately after an association, so it is
+        * reasonable to assume that a scan was completed recently. This makes
+        * us skip an immediate new scan in cases where the current signal
+        * level is below the bgscan threshold.
+        */
+       os_get_time(&data->last_bgscan);
+
        return data;
 }
 
@@ -377,7 +454,7 @@ static void bgscan_learn_deinit(void *priv)
        dl_list_for_each_safe(bss, n, &data->bss, struct bgscan_learn_bss,
                              list) {
                dl_list_del(&bss->list);
-               os_free(bss);
+               bss_free(bss);
        }
        os_free(data->supp_freqs);
        os_free(data);
@@ -405,7 +482,10 @@ static int bgscan_learn_notify_scan(void *priv,
                                    struct wpa_scan_results *scan_res)
 {
        struct bgscan_learn_data *data = priv;
-       size_t i;
+       size_t i, j;
+#define MAX_BSS 50
+       u8 bssid[MAX_BSS * ETH_ALEN];
+       size_t num_bssid = 0;
 
        wpa_printf(MSG_DEBUG, "bgscan learn: scan result notification");
 
@@ -415,10 +495,25 @@ static int bgscan_learn_notify_scan(void *priv,
 
        for (i = 0; i < scan_res->num; i++) {
                struct wpa_scan_res *res = scan_res->res[i];
+               if (!bgscan_learn_bss_match(data, res))
+                       continue;
+
+               if (num_bssid < MAX_BSS) {
+                       os_memcpy(bssid + num_bssid * ETH_ALEN, res->bssid,
+                                 ETH_ALEN);
+                       num_bssid++;
+               }
+       }
+       wpa_printf(MSG_DEBUG, "bgscan learn: %u matching BSSes in scan "
+                  "results", (unsigned int) num_bssid);
+
+       for (i = 0; i < scan_res->num; i++) {
+               struct wpa_scan_res *res = scan_res->res[i];
                struct bgscan_learn_bss *bss;
 
                if (!bgscan_learn_bss_match(data, res))
                        continue;
+
                bss = bgscan_learn_get_bss(data, res->bssid);
                if (bss && bss->freq != res->freq) {
                        wpa_printf(MSG_DEBUG, "bgscan learn: Update BSS "
@@ -435,6 +530,11 @@ static int bgscan_learn_notify_scan(void *priv,
                        bss->freq = res->freq;
                        dl_list_add(&data->bss, &bss->list);
                }
+
+               for (j = 0; j < num_bssid; j++) {
+                       u8 *addr = bssid + j * ETH_ALEN;
+                       bgscan_learn_add_neighbor(bss, addr);
+               }
        }
 
        /*
@@ -455,23 +555,26 @@ static void bgscan_learn_notify_beacon_loss(void *priv)
 }
 
 
-static void bgscan_learn_notify_signal_change(void *priv, int above)
+static void bgscan_learn_notify_signal_change(void *priv, int above,
+                                             int current_signal)
 {
        struct bgscan_learn_data *data = priv;
+       int scan = 0;
+       struct os_time now;
 
        if (data->short_interval == data->long_interval ||
            data->signal_threshold == 0)
                return;
 
        wpa_printf(MSG_DEBUG, "bgscan learn: signal level changed "
-                  "(above=%d)", above);
+                  "(above=%d current_signal=%d)", above, current_signal);
        if (data->scan_interval == data->long_interval && !above) {
-               wpa_printf(MSG_DEBUG, "bgscan learn: Trigger immediate scan "
-                          "and start using short bgscan interval");
+               wpa_printf(MSG_DEBUG, "bgscan learn: Start using short bgscan "
+                          "interval");
                data->scan_interval = data->short_interval;
-               eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
-               eloop_register_timeout(0, 0, bgscan_learn_timeout, data,
-                                      NULL);
+               os_get_time(&now);
+               if (now.sec > data->last_bgscan.sec + 1)
+                       scan = 1;
        } else if (data->scan_interval == data->short_interval && above) {
                wpa_printf(MSG_DEBUG, "bgscan learn: Start using long bgscan "
                           "interval");
@@ -480,20 +583,19 @@ static void bgscan_learn_notify_signal_change(void *priv, int above)
                eloop_register_timeout(data->scan_interval, 0,
                                       bgscan_learn_timeout, data, NULL);
        } else if (!above) {
-               struct os_time now;
                /*
                 * Signal dropped further 4 dB. Request a new scan if we have
                 * not yet scanned in a while.
                 */
                os_get_time(&now);
-               if (now.sec > data->last_bgscan.sec + 10) {
-                       wpa_printf(MSG_DEBUG, "bgscan learn: Trigger "
-                                  "immediate scan");
-                       eloop_cancel_timeout(bgscan_learn_timeout, data,
-                                            NULL);
-                       eloop_register_timeout(0, 0, bgscan_learn_timeout,
-                                              data, NULL);
-               }
+               if (now.sec > data->last_bgscan.sec + 10)
+                       scan = 1;
+       }
+
+       if (scan) {
+               wpa_printf(MSG_DEBUG, "bgscan learn: Trigger immediate scan");
+               eloop_cancel_timeout(bgscan_learn_timeout, data, NULL);
+               eloop_register_timeout(0, 0, bgscan_learn_timeout, data, NULL);
        }
 }