Merged the hostap_2.6 updates, and the Leap of Faith work, from the hostap_update...

[mech_eap.git] / libeap / tests / eapol-fuzzer / eapol-fuzzer.c

diff --git a/libeap/tests/eapol-fuzzer/eapol-fuzzer.c b/libeap/tests/eapol-fuzzer/eapol-fuzzer.c
new file mode 100644 (file)
index 0000000..7429ee3
--- /dev/null
+++ b/libeap/tests/eapol-fuzzer/eapol-fuzzer.c
@@ -0,0 +1,211 @@
+/*
+ * wpa_supplicant - EAPOL 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 "eapol_supp/eapol_supp_sm.h"
+#include "rsn_supp/wpa.h"
+
+
+struct arg_ctx {
+       const char *fname;
+       struct wpa_sm *wpa;
+       struct eapol_sm *eapol;
+};
+
+
+static void test_send_eapol(void *eloop_data, void *user_ctx)
+{
+       struct arg_ctx *ctx = eloop_data;
+       char *data;
+       size_t len;
+       u8 src[ETH_ALEN] = { 0x02, 0x00, 0x00, 0x00, 0x00, 0x01 };
+       u8 wpa_ie[200];
+       size_t wpa_ie_len;
+
+       wpa_printf(MSG_INFO, "eapol-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 - EAPOL", data, len);
+
+       eapol_sm_notify_portEnabled(ctx->eapol, TRUE);
+
+       wpa_sm_set_param(ctx->wpa, WPA_PARAM_PROTO, WPA_PROTO_RSN);
+       wpa_sm_set_param(ctx->wpa, WPA_PARAM_RSN_ENABLED, 1);
+       wpa_sm_set_param(ctx->wpa, WPA_PARAM_KEY_MGMT, WPA_KEY_MGMT_PSK);
+       wpa_sm_set_param(ctx->wpa, WPA_PARAM_PAIRWISE, WPA_CIPHER_CCMP);
+       wpa_sm_set_param(ctx->wpa, WPA_PARAM_GROUP, WPA_CIPHER_CCMP);
+
+       wpa_ie_len = sizeof(wpa_ie);
+       wpa_sm_set_assoc_wpa_ie_default(ctx->wpa, wpa_ie, &wpa_ie_len);
+
+       if (eapol_sm_rx_eapol(ctx->eapol, src, (u8 *) data, len) <= 0)
+               wpa_sm_rx_eapol(ctx->wpa, src, (u8 *) data, len);
+
+out:
+       os_free(data);
+       eloop_terminate();
+}
+
+
+static void * get_network_ctx(void *arg)
+{
+       return (void *) 1;
+}
+
+
+static void set_state(void *arg, enum wpa_states state)
+{
+}
+
+
+static void deauthenticate(void *arg, int reason_code)
+{
+}
+
+
+static u8 * alloc_eapol(void *arg, u8 type,
+                       const void *data, u16 data_len,
+                       size_t *msg_len, void **data_pos)
+{
+       struct ieee802_1x_hdr *hdr;
+
+       *msg_len = sizeof(*hdr) + data_len;
+       hdr = os_malloc(*msg_len);
+       if (hdr == NULL)
+               return NULL;
+
+       hdr->version = 2;
+       hdr->type = type;
+       hdr->length = host_to_be16(data_len);
+
+       if (data)
+               os_memcpy(hdr + 1, data, data_len);
+       else
+               os_memset(hdr + 1, 0, data_len);
+
+       if (data_pos)
+               *data_pos = hdr + 1;
+
+       return (u8 *) hdr;
+}
+
+
+static int ether_send(void *arg, const u8 *dest, u16 proto,
+                     const u8 *buf, size_t len)
+{
+       return 0;
+}
+
+
+static int get_bssid(void *ctx, u8 *bssid)
+{
+       return -1;
+}
+
+
+static int eapol_send(void *ctx, int type, const u8 *buf, size_t len)
+{
+       return 0;
+}
+
+
+static int init_wpa(struct arg_ctx *arg)
+{
+       struct wpa_sm_ctx *ctx;
+
+       ctx = os_zalloc(sizeof(*ctx));
+       if (ctx == NULL) {
+               wpa_printf(MSG_ERROR, "Failed to allocate WPA context.");
+               return -1;
+       }
+
+       ctx->ctx = arg;
+       ctx->msg_ctx = arg;
+       ctx->get_network_ctx = get_network_ctx;
+       ctx->set_state = set_state;
+       ctx->deauthenticate = deauthenticate;
+       ctx->alloc_eapol = alloc_eapol;
+       ctx->ether_send = ether_send;
+       ctx->get_bssid = get_bssid;
+
+       arg->wpa = wpa_sm_init(ctx);
+       return arg->wpa ? 0 : -1;
+}
+
+
+static int init_eapol(struct arg_ctx *arg)
+{
+       struct eapol_ctx *ctx;
+
+       ctx = os_zalloc(sizeof(*ctx));
+       if (ctx == NULL) {
+               wpa_printf(MSG_ERROR, "Failed to allocate EAPOL context.");
+               return -1;
+       }
+
+       ctx->ctx = arg;
+       ctx->msg_ctx = arg;
+       ctx->eapol_send = eapol_send;
+
+       arg->eapol = eapol_sm_init(ctx);
+       return arg->eapol ? 0 : -1;
+}
+
+
+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) || init_eapol(&ctx))
+               goto fail;
+
+       eloop_register_timeout(0, 0, test_send_eapol, &ctx, NULL);
+
+       wpa_printf(MSG_DEBUG, "Starting eloop");
+       eloop_run();
+       wpa_printf(MSG_DEBUG, "eloop done");
+
+       ret = 0;
+fail:
+       if (ctx.wpa)
+               wpa_sm_deinit(ctx.wpa);
+       if (ctx.eapol)
+               eapol_sm_deinit(ctx.eapol);
+
+       eloop_destroy();
+       os_program_deinit();
+
+       return ret;
+}