From f79a5fa0068d7c770b2631e3030bca07892f99fe Mon Sep 17 00:00:00 2001 From: Jouni Malinen Date: Wed, 29 Apr 2015 02:10:24 +0300 Subject: [PATCH] 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 --- tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c b/tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c index a9e197541..dd061448e 100644 --- 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: -- 2.39.2