From 2352474a36adc6bc68bbf73f22630f76dcea178a Mon Sep 17 00:00:00 2001
From: PolynomialDivision <vincent@systemli.org>
Date: Sun, 7 Jan 2018 21:32:19 +0100
Subject: [PATCH] use max station diff

---
 files/dawn.config         | 5 +++--
 src/include/datastorage.h | 1 +
 src/storage/datastorage.c | 2 +-
 src/utils/dawn_uci.c      | 1 +
 4 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/files/dawn.config b/files/dawn.config
index 151f042..9cfa5f8 100644
--- a/files/dawn.config
+++ b/files/dawn.config
@@ -34,11 +34,12 @@ config metric
     option low_rssi_val         '-80'
     option chan_util_val        '140'
     option max_chan_util_val    '170'
-    option min_probe_count      '0'
+    option min_probe_count      '1'
     option bandwith_threshold   '6'
     option use_station_count    '1'
+    option max_station_diff     '1'
     option eval_probe_req       '0'
     option eval_auth_req        '0'     # no real reasoncode...
     option eval_assoc_req       '1'     # just deny assocs...
     option deny_auth_reason     '1'     # unspecified
-    option deny_assoc_reason    '17'    # assoc rejected can't handle new station
\ No newline at end of file
+    option deny_assoc_reason    '17'    # assoc rejected can't handle new station
diff --git a/src/include/datastorage.h b/src/include/datastorage.h
index 96d7f36..cce340e 100644
--- a/src/include/datastorage.h
+++ b/src/include/datastorage.h
@@ -49,6 +49,7 @@ struct probe_metric_s {
     int min_probe_count;
     int bandwith_threshold;
     int use_station_count;
+    int max_station_diff;
     int eval_probe_req;
     int eval_auth_req;
     int eval_assoc_req;
diff --git a/src/storage/datastorage.c b/src/storage/datastorage.c
index 566ff5b..297e039 100644
--- a/src/storage/datastorage.c
+++ b/src/storage/datastorage.c
@@ -284,7 +284,7 @@ int compare_station_count(uint8_t *bssid_addr_own, uint8_t *bssid_addr_to_compar
         }
         printf("AFTER: Comparing own %d to %d\n", sta_count, sta_count_to_compare);
 
-        return sta_count > sta_count_to_compare;
+        return sta_count - sta_count_to_compare < dawn_metric.max_station_diff;
     }
 
     return 0;
diff --git a/src/utils/dawn_uci.c b/src/utils/dawn_uci.c
index 02a0069..fe35b96 100644
--- a/src/utils/dawn_uci.c
+++ b/src/utils/dawn_uci.c
@@ -71,6 +71,7 @@ struct probe_metric_s uci_get_dawn_metric() {
             ret.eval_assoc_req = uci_lookup_option_int(uci_ctx, s, "eval_assoc_req");
             ret.deny_auth_reason = uci_lookup_option_int(uci_ctx, s, "deny_auth_reason");
             ret.deny_assoc_reason = uci_lookup_option_int(uci_ctx, s, "deny_assoc_reason");
+            ret.max_station_diff = uci_lookup_option_int(uci_ctx, s, "max_station_diff");
             return ret;
         }
     }