tests: Add p2p-fuzzer

This program can be used to run fuzzing tests for areas related to P2P
message parsing and processing. p2p-fuzzer allows data files to be used
to inject Probe Response and Action frames for processing by the P2P
module.

Signed-off-by: Jouni Malinen <j@w1.fi>
This commit is contained in:
Jouni Malinen 2015-04-08 23:12:25 +03:00
parent 632931c3ce
commit a65d7495b5
10 changed files with 347 additions and 9 deletions

View file

@ -1,8 +1,28 @@
all: all: libcommon.a
@echo Nothing to be made.
clean: clean:
rm -f *~ *.o *.d *.gcno *.gcda *.gcov rm -f *~ *.o *.d *.gcno *.gcda *.gcov libcommon.a
install: install:
@echo Nothing to be made. @echo Nothing to be made.
include ../lib.rules
CFLAGS += -DCONFIG_IEEE80211R
CFLAGS += -DCONFIG_IEEE80211W
CFLAGS += -DCONFIG_HS20
CFLAGS += -DCONFIG_SAE
CFLAGS += -DCONFIG_SUITE
CFLAGS += -DCONFIG_SUITEB
LIB_OBJS= \
gas.o \
hw_features_common.o \
ieee802_11_common.o \
sae.o \
wpa_common.o
libcommon.a: $(LIB_OBJS)
$(AR) crT $@ $?
-include $(OBJS:%.o=%.d)

View file

@ -1,8 +1,29 @@
all: all: libp2p.a
@echo Nothing to be made.
clean: clean:
rm -f *~ *.o *.d *.gcno *.gcda *.gcov rm -f *~ *.o *.d *.gcno *.gcda *.gcov libp2p.a
install: install:
@echo Nothing to be made. @echo Nothing to be made.
include ../lib.rules
CFLAGS += -DCONFIG_WIFI_DISPLAY
CFLAGS += -DCONFIG_WPS_NFC
LIB_OBJS= \
p2p_build.o \
p2p.o \
p2p_dev_disc.o \
p2p_go_neg.o \
p2p_group.o \
p2p_invitation.o \
p2p_parse.o \
p2p_pd.o \
p2p_sd.o \
p2p_utils.o
libp2p.a: $(LIB_OBJS)
$(AR) crT $@ $?
-include $(OBJS:%.o=%.d)

View file

@ -1,8 +1,41 @@
all: all: libwps.a
@echo Nothing to be made.
clean: clean:
rm -f *~ *.o *.d *.gcno *.gcda *.gcov rm -f *~ *.o *.d *.gcno *.gcda *.gcov libwps.a
install: install:
@echo Nothing to be made. @echo Nothing to be made.
include ../lib.rules
CFLAGS += -DCONFIG_P2P
CFLAGS += -DCONFIG_WPS_OOB
CFLAGS += -DCONFIG_WPS_NFC
LIB_OBJS= \
http_client.o \
httpread.o \
http_server.o \
ndef.o \
upnp_xml.o \
wps_attr_build.o \
wps_attr_parse.o \
wps_attr_process.o \
wps.o \
wps_common.o \
wps_dev_attr.o \
wps_enrollee.o \
wps_er.o \
wps_er_ssdp.o \
wps_module_tests.o \
wps_registrar.o \
wps_upnp_ap.o \
wps_upnp.o \
wps_upnp_event.o \
wps_upnp_ssdp.o \
wps_upnp_web.o
libwps.a: $(LIB_OBJS)
$(AR) crT $@ $?
-include $(OBJS:%.o=%.d)

51
tests/p2p-fuzzer/Makefile Normal file
View file

@ -0,0 +1,51 @@
all: p2p-fuzzer
ifndef CC
CC=gcc
endif
ifndef LDO
LDO=$(CC)
endif
ifndef CFLAGS
CFLAGS = -MMD -O2 -Wall -g
endif
SRC=../../src
CFLAGS += -I$(SRC)
$(SRC)/utils/libutils.a:
$(MAKE) -C $(SRC)/utils
$(SRC)/common/libcommon.a:
$(MAKE) -C $(SRC)/common
$(SRC)/crypto/libcrypto.a:
$(MAKE) -C $(SRC)/crypto
$(SRC)/tls/libtls.a:
$(MAKE) -C $(SRC)/tls
$(SRC)/p2p/libp2p.a:
$(MAKE) -C $(SRC)/p2p
$(SRC)/wps/libwps.a:
$(MAKE) -C $(SRC)/wps
LIBS += $(SRC)/utils/libutils.a
LIBS += $(SRC)/common/libcommon.a
LIBS += $(SRC)/crypto/libcrypto.a
LIBS += $(SRC)/p2p/libp2p.a
LIBS += $(SRC)/tls/libtls.a
LIBS += $(SRC)/wps/libwps.a
p2p-fuzzer: p2p-fuzzer.o $(LIBS)
$(LDO) $(LDFLAGS) -o $@ $^ $(LIBS)
clean:
$(MAKE) -C $(SRC) clean
rm -f p2p-fuzzer *~ *.o *.d
-include $(OBJS:%.o=%.d)

Binary file not shown.

Binary file not shown.

View file

@ -0,0 +1,213 @@
/*
* wpa_supplicant - P2P fuzzer
* Copyright (c) 2015, 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 "p2p/p2p.h"
static void debug_print(void *ctx, int level, const char *msg)
{
wpa_printf(level, "P2P: %s", msg);
}
static void find_stopped(void *ctx)
{
}
static int start_listen(void *ctx, unsigned int freq,
unsigned int duration,
const struct wpabuf *probe_resp_ie)
{
return 0;
}
static void stop_listen(void *ctx)
{
}
static void dev_found(void *ctx, const u8 *addr,
const struct p2p_peer_info *info,
int new_device)
{
}
static void dev_lost(void *ctx, const u8 *dev_addr)
{
}
static int send_action(void *ctx, unsigned int freq, const u8 *dst,
const u8 *src, const u8 *bssid, const u8 *buf,
size_t len, unsigned int wait_time)
{
return 0;
}
static void send_action_done(void *ctx)
{
}
static void go_neg_req_rx(void *ctx, const u8 *src, u16 dev_passwd_id)
{
}
static struct p2p_data * init_p2p(void)
{
struct p2p_config p2p;
os_memset(&p2p, 0, sizeof(p2p));
p2p.max_peers = 100;
p2p.passphrase_len = 8;
p2p.channels.reg_classes = 1;
p2p.channels.reg_class[0].reg_class = 81;
p2p.channels.reg_class[0].channel[0] = 1;
p2p.channels.reg_class[0].channel[1] = 2;
p2p.channels.reg_class[0].channels = 2;
p2p.debug_print = debug_print;
p2p.find_stopped = find_stopped;
p2p.start_listen = start_listen;
p2p.stop_listen = stop_listen;
p2p.dev_found = dev_found;
p2p.dev_lost = dev_lost;
p2p.send_action = send_action;
p2p.send_action_done = send_action_done;
p2p.go_neg_req_rx = go_neg_req_rx;
return p2p_init(&p2p);
}
struct arg_ctx {
struct p2p_data *p2p;
const char *fname;
};
static void test_send_proberesp(void *eloop_data, void *user_ctx)
{
struct arg_ctx *ctx = eloop_data;
char *data;
size_t len;
struct os_reltime rx_time;
wpa_printf(MSG_INFO, "p2p-fuzzer: Send proberesp '%s'", ctx->fname);
data = os_readfile(ctx->fname, &len);
if (!data) {
wpa_printf(MSG_ERROR, "Could not read '%s'", ctx->fname);
return;
}
wpa_hexdump(MSG_MSGDUMP, "fuzzer - IEs", data, len);
os_memset(&rx_time, 0, sizeof(rx_time));
p2p_scan_res_handler(ctx->p2p, (u8 *) "\x02\x00\x00\x00\x01\x00", 2412,
&rx_time, 0, (u8 *) data, len);
p2p_scan_res_handled(ctx->p2p);
os_free(data);
eloop_terminate();
}
static void test_send_action(void *eloop_data, void *user_ctx)
{
struct arg_ctx *ctx = eloop_data;
char *data;
size_t len;
struct os_reltime rx_time;
struct ieee80211_mgmt *mgmt;
wpa_printf(MSG_INFO, "p2p-fuzzer: Send action '%s'", ctx->fname);
data = os_readfile(ctx->fname, &len);
if (!data) {
wpa_printf(MSG_ERROR, "Could not read '%s'", ctx->fname);
return;
}
if (len < IEEE80211_HDRLEN + 1)
goto out;
wpa_hexdump(MSG_MSGDUMP, "fuzzer - action", data, len);
mgmt = (struct ieee80211_mgmt *) data;
os_memset(&rx_time, 0, sizeof(rx_time));
p2p_rx_action(ctx->p2p, mgmt->da, mgmt->sa, mgmt->bssid,
mgmt->u.action.category,
(u8 *) data + IEEE80211_HDRLEN + 1,
len - IEEE80211_HDRLEN - 1, 2412);
out:
os_free(data);
eloop_terminate();
}
int main(int argc, char *argv[])
{
struct p2p_data *p2p;
struct arg_ctx ctx;
/* TODO: probreq and wpas_p2p_probe_req_rx() */
if (argc < 3) {
printf("usage: %s <proberesp|action> <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;
}
p2p = init_p2p();
if (!p2p) {
wpa_printf(MSG_ERROR, "P2P init failed");
return -1;
}
ctx.p2p = p2p;
ctx.fname = argv[2];
if (os_strcmp(argv[1], "proberesp") == 0) {
eloop_register_timeout(0, 0, test_send_proberesp, &ctx, NULL);
} else if (os_strcmp(argv[1], "action") == 0) {
eloop_register_timeout(0, 0, test_send_action, &ctx, NULL);
} else {
wpa_printf(MSG_ERROR, "Unsupported test type '%s'", argv[1]);
return -1;
}
wpa_printf(MSG_DEBUG, "Starting eloop");
eloop_run();
wpa_printf(MSG_DEBUG, "eloop done");
p2p_deinit(p2p);
eloop_destroy();
os_program_deinit();
return 0;
}

Binary file not shown.

Binary file not shown.

Binary file not shown.