From: Jouni Malinen <j@w1.fi>
Date: Tue, 28 Apr 2015 23:10:24 +0000 (+0300)
Subject: tests: Add a STA entry for ap-mgmt-fuzzer
X-Git-Tag: hostap_2_5~749
X-Git-Url: http://www.project-moonshot.org/gitweb/?a=commitdiff_plain;h=f79a5fa0068d7c770b2631e3030bca07892f99fe;p=mech_eap.git

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 a9e1975..dd06144 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: