2dfe2ad6cb
Avoid a memory leak in fuzzer tests by calling wnm_deallocate_memory() before exiting since this fuzzing wrapper does not use full wpa_supplicant_cleanup(). Signed-off-by: Jouni Malinen <jouni@codeaurora.org>
116 lines
2.2 KiB
C
116 lines
2.2 KiB
C
/*
|
|
* wpa_supplicant - WNM fuzzer
|
|
* Copyright (c) 2015-2019, Jouni Malinen <j@w1.fi>
|
|
*
|
|
* This software may be distributed under the terms of the BSD license.
|
|
* See README for more details.
|
|
*/
|
|
|
|
#include "utils/includes.h"
|
|
|
|
#include "utils/common.h"
|
|
#include "utils/eloop.h"
|
|
#include "common/ieee802_11_defs.h"
|
|
#include "rsn_supp/wpa.h"
|
|
#include "rsn_supp/wpa_i.h"
|
|
#include "../../wpa_supplicant/wpa_supplicant_i.h"
|
|
#include "../../wpa_supplicant/bss.h"
|
|
#include "../../wpa_supplicant/wnm_sta.h"
|
|
#include "../../wpa_supplicant/config.h"
|
|
|
|
|
|
struct arg_ctx {
|
|
const char *fname;
|
|
struct wpa_supplicant wpa_s;
|
|
struct wpa_bss bss;
|
|
struct wpa_driver_ops driver;
|
|
struct wpa_sm wpa;
|
|
struct wpa_config conf;
|
|
};
|
|
|
|
|
|
static void test_send_wnm(void *eloop_data, void *user_ctx)
|
|
{
|
|
struct arg_ctx *ctx = eloop_data;
|
|
char *data;
|
|
size_t len;
|
|
struct ieee80211_mgmt *mgmt;
|
|
|
|
wpa_printf(MSG_INFO, "wnm-fuzzer: Send '%s'", ctx->fname);
|
|
|
|
data = os_readfile(ctx->fname, &len);
|
|
if (!data) {
|
|
wpa_printf(MSG_ERROR, "Could not read '%s'", ctx->fname);
|
|
goto out;
|
|
}
|
|
|
|
wpa_hexdump(MSG_MSGDUMP, "fuzzer - WNM", data, len);
|
|
|
|
mgmt = (struct ieee80211_mgmt *) data;
|
|
ieee802_11_rx_wnm_action(&ctx->wpa_s, mgmt, len);
|
|
|
|
out:
|
|
os_free(data);
|
|
eloop_terminate();
|
|
}
|
|
|
|
|
|
static int init_wpa(struct arg_ctx *ctx)
|
|
{
|
|
ctx->wpa_s.wpa_state = WPA_COMPLETED;
|
|
os_memcpy(ctx->wpa_s.bssid, "\x02\x00\x00\x00\x03\x00", ETH_ALEN);
|
|
ctx->wpa_s.current_bss = &ctx->bss;
|
|
ctx->wpa_s.driver = &ctx->driver;
|
|
ctx->wpa_s.wpa = &ctx->wpa;
|
|
ctx->wpa_s.conf = &ctx->conf;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
static void deinit_wpa(struct arg_ctx *ctx)
|
|
{
|
|
wnm_deallocate_memory(&ctx->wpa_s);
|
|
}
|
|
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
struct arg_ctx ctx;
|
|
int ret = -1;
|
|
|
|
if (argc < 2) {
|
|
printf("usage: %s <file>\n", argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
if (os_program_init())
|
|
return -1;
|
|
|
|
wpa_debug_level = 0;
|
|
wpa_debug_show_keys = 1;
|
|
|
|
if (eloop_init()) {
|
|
wpa_printf(MSG_ERROR, "Failed to initialize event loop");
|
|
return -1;
|
|
}
|
|
|
|
os_memset(&ctx, 0, sizeof(ctx));
|
|
ctx.fname = argv[1];
|
|
if (init_wpa(&ctx))
|
|
goto fail;
|
|
|
|
eloop_register_timeout(0, 0, test_send_wnm, &ctx, NULL);
|
|
|
|
wpa_printf(MSG_DEBUG, "Starting eloop");
|
|
eloop_run();
|
|
wpa_printf(MSG_DEBUG, "eloop done");
|
|
deinit_wpa(&ctx);
|
|
|
|
ret = 0;
|
|
fail:
|
|
eloop_destroy();
|
|
os_program_deinit();
|
|
|
|
return ret;
|
|
}
|