Browse Source

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>
Jouni Malinen 10 years ago
parent
commit
f79a5fa006
1 changed files with 7 additions and 0 deletions
  1. 7 0
      tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c

+ 7 - 0
tests/ap-mgmt-fuzzer/ap-mgmt-fuzzer.c

@@ -12,6 +12,7 @@
 #include "utils/eloop.h"
 #include "utils/eloop.h"
 #include "ap/hostapd.h"
 #include "ap/hostapd.h"
 #include "ap/ieee802_11.h"
 #include "ap/ieee802_11.h"
+#include "ap/sta_info.h"
 
 
 
 
 const struct wpa_driver_ops *const wpa_drivers[] =
 const struct wpa_driver_ops *const wpa_drivers[] =
@@ -59,6 +60,7 @@ out:
 static int init_hapd(struct arg_ctx *ctx)
 static int init_hapd(struct arg_ctx *ctx)
 {
 {
 	struct hostapd_data *hapd = &ctx->hapd;
 	struct hostapd_data *hapd = &ctx->hapd;
+	struct sta_info *sta;
 
 
 	hapd->driver = &ctx->driver;
 	hapd->driver = &ctx->driver;
 	os_memcpy(hapd->own_addr, "\x02\x00\x00\x00\x03\x00", ETH_ALEN);
 	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];
 	hapd->conf = hapd->iconf->bss[0];
 	hostapd_config_defaults_bss(hapd->conf);
 	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;
 	return 0;
 }
 }
 
 
@@ -105,6 +111,7 @@ int main(int argc, char *argv[])
 	wpa_printf(MSG_DEBUG, "Starting eloop");
 	wpa_printf(MSG_DEBUG, "Starting eloop");
 	eloop_run();
 	eloop_run();
 	wpa_printf(MSG_DEBUG, "eloop done");
 	wpa_printf(MSG_DEBUG, "eloop done");
+	hostapd_free_stas(&ctx.hapd);
 
 
 	ret = 0;
 	ret = 0;
 fail:
 fail: