tests: Add a STA entry for ap-mgmt-fuzzer

This increases the coverage for AP mode management frame fuzzing by
allowing number of additional Action frame code paths to be executed.

Signed-off-by: Jouni Malinen <j@w1.fi>

diff --git a/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c b/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c
index a9e197541449b4cfdb843e2d7475aa44d4a48f62..dd061448ed05f88da6674c3f4216b659b9f91906 100644 (file)
--- a/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c
+++ b/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c
@@ -12,6 +12,7 @@
 #include "utils/eloop.h"
 #include "ap/hostapd.h"
 #include "ap/ieee802_11.h"
+#include "ap/sta_info.h"
 
 
 const struct wpa_driver_ops *const wpa_drivers[] =
@@ -59,6 +60,7 @@ out:
 static int init_hapd(struct arg_ctx *ctx)
 {
        struct hostapd_data *hapd = &ctx->hapd;
+       struct sta_info *sta;
 
        hapd->driver = &ctx->driver;
        os_memcpy(hapd->own_addr, "\x02\x00\x00\x00\x03\x00", ETH_ALEN);
@@ -70,6 +72,10 @@ static int init_hapd(struct arg_ctx *ctx)
        hapd->conf = hapd->iconf->bss[0];
        hostapd_config_defaults_bss(hapd->conf);
 
+       sta = ap_sta_add(hapd, (u8 *) "\x02\x00\x00\x00\x00\x00");
+       if (sta)
+               sta->flags |= WLAN_STA_ASSOC | WLAN_STA_WMM;
+
        return 0;
 }
 
@@ -105,6 +111,7 @@ int main(int argc, char *argv[])
        wpa_printf(MSG_DEBUG, "Starting eloop");
        eloop_run();
        wpa_printf(MSG_DEBUG, "eloop done");
+       hostapd_free_stas(&ctx.hapd);
 
        ret = 0;
 fail: