1189251Ssam/* 2189251Ssam * WPA Supplicant - test code for pre-authentication 3189251Ssam * Copyright (c) 2003-2007, Jouni Malinen <j@w1.fi> 4189251Ssam * 5252726Srpaulo * This software may be distributed under the terms of the BSD license. 6252726Srpaulo * See README for more details. 7189251Ssam * 8189251Ssam * IEEE 802.1X Supplicant test code (to be used in place of wpa_supplicant.c. 9189251Ssam * Not used in production version. 10189251Ssam */ 11189251Ssam 12189251Ssam#include "includes.h" 13189251Ssam#include <assert.h> 14189251Ssam 15189251Ssam#include "common.h" 16189251Ssam#include "config.h" 17189251Ssam#include "eapol_supp/eapol_supp_sm.h" 18189251Ssam#include "eloop.h" 19214734Srpaulo#include "rsn_supp/wpa.h" 20189251Ssam#include "eap_peer/eap.h" 21189251Ssam#include "wpa_supplicant_i.h" 22189251Ssam#include "l2_packet/l2_packet.h" 23189251Ssam#include "ctrl_iface.h" 24189251Ssam#include "pcsc_funcs.h" 25214734Srpaulo#include "rsn_supp/preauth.h" 26214734Srpaulo#include "rsn_supp/pmksa_cache.h" 27214734Srpaulo#include "drivers/driver.h" 28189251Ssam 29189251Ssam 30289549Srpauloconst struct wpa_driver_ops *const wpa_drivers[] = { NULL }; 31189251Ssam 32189251Ssam 33189251Ssamstruct preauth_test_data { 34189251Ssam int auth_timed_out; 35189251Ssam}; 36189251Ssam 37189251Ssam 38351611Scystatic void _wpa_supplicant_deauthenticate(void *wpa_s, u16 reason_code) 39189251Ssam{ 40189251Ssam wpa_supplicant_deauthenticate(wpa_s, reason_code); 41189251Ssam} 42189251Ssam 43189251Ssam 44189251Ssamstatic u8 * wpa_alloc_eapol(const struct wpa_supplicant *wpa_s, u8 type, 45189251Ssam const void *data, u16 data_len, 46189251Ssam size_t *msg_len, void **data_pos) 47189251Ssam{ 48189251Ssam struct ieee802_1x_hdr *hdr; 49189251Ssam 50189251Ssam *msg_len = sizeof(*hdr) + data_len; 51189251Ssam hdr = os_malloc(*msg_len); 52189251Ssam if (hdr == NULL) 53189251Ssam return NULL; 54189251Ssam 55189251Ssam hdr->version = wpa_s->conf->eapol_version; 56189251Ssam hdr->type = type; 57189251Ssam hdr->length = htons(data_len); 58189251Ssam 59189251Ssam if (data) 60189251Ssam os_memcpy(hdr + 1, data, data_len); 61189251Ssam else 62189251Ssam os_memset(hdr + 1, 0, data_len); 63189251Ssam 64189251Ssam if (data_pos) 65189251Ssam *data_pos = hdr + 1; 66189251Ssam 67189251Ssam return (u8 *) hdr; 68189251Ssam} 69189251Ssam 70189251Ssam 71189251Ssamstatic u8 * _wpa_alloc_eapol(void *wpa_s, u8 type, 72189251Ssam const void *data, u16 data_len, 73189251Ssam size_t *msg_len, void **data_pos) 74189251Ssam{ 75189251Ssam return wpa_alloc_eapol(wpa_s, type, data, data_len, msg_len, data_pos); 76189251Ssam} 77189251Ssam 78189251Ssam 79214734Srpaulostatic void _wpa_supplicant_set_state(void *ctx, enum wpa_states state) 80189251Ssam{ 81189251Ssam struct wpa_supplicant *wpa_s = ctx; 82189251Ssam wpa_s->wpa_state = state; 83189251Ssam} 84189251Ssam 85189251Ssam 86214734Srpaulostatic enum wpa_states _wpa_supplicant_get_state(void *ctx) 87189251Ssam{ 88189251Ssam struct wpa_supplicant *wpa_s = ctx; 89189251Ssam return wpa_s->wpa_state; 90189251Ssam} 91189251Ssam 92189251Ssam 93189251Ssamstatic int wpa_ether_send(void *wpa_s, const u8 *dest, u16 proto, 94189251Ssam const u8 *buf, size_t len) 95189251Ssam{ 96189251Ssam printf("%s - not implemented\n", __func__); 97189251Ssam return -1; 98189251Ssam} 99189251Ssam 100189251Ssam 101189251Ssamstatic void * wpa_supplicant_get_network_ctx(void *wpa_s) 102189251Ssam{ 103189251Ssam return wpa_supplicant_get_ssid(wpa_s); 104189251Ssam} 105189251Ssam 106189251Ssam 107189251Ssamstatic void _wpa_supplicant_cancel_auth_timeout(void *wpa_s) 108189251Ssam{ 109189251Ssam wpa_supplicant_cancel_auth_timeout(wpa_s); 110189251Ssam} 111189251Ssam 112189251Ssam 113189251Ssamstatic int wpa_supplicant_get_beacon_ie(void *wpa_s) 114189251Ssam{ 115189251Ssam printf("%s - not implemented\n", __func__); 116189251Ssam return -1; 117189251Ssam} 118189251Ssam 119189251Ssam 120189251Ssamstatic int wpa_supplicant_get_bssid(void *wpa_s, u8 *bssid) 121189251Ssam{ 122189251Ssam printf("%s - not implemented\n", __func__); 123189251Ssam return -1; 124189251Ssam} 125189251Ssam 126189251Ssam 127214734Srpaulostatic int wpa_supplicant_set_key(void *wpa_s, enum wpa_alg alg, 128189251Ssam const u8 *addr, int key_idx, int set_tx, 129189251Ssam const u8 *seq, size_t seq_len, 130189251Ssam const u8 *key, size_t key_len) 131189251Ssam{ 132189251Ssam printf("%s - not implemented\n", __func__); 133189251Ssam return -1; 134189251Ssam} 135189251Ssam 136189251Ssam 137189251Ssamstatic int wpa_supplicant_mlme_setprotection(void *wpa_s, const u8 *addr, 138189251Ssam int protection_type, 139189251Ssam int key_type) 140189251Ssam{ 141189251Ssam printf("%s - not implemented\n", __func__); 142189251Ssam return -1; 143189251Ssam} 144189251Ssam 145189251Ssam 146346981Scystatic int wpa_supplicant_add_pmkid(void *wpa_s, void *network_ctx, 147346981Scy const u8 *bssid, const u8 *pmkid, 148346981Scy const u8 *fils_cache_id, 149346981Scy const u8 *pmk, size_t pmk_len) 150189251Ssam{ 151189251Ssam printf("%s - not implemented\n", __func__); 152189251Ssam return -1; 153189251Ssam} 154189251Ssam 155189251Ssam 156346981Scystatic int wpa_supplicant_remove_pmkid(void *wpa_s, void *network_ctx, 157346981Scy const u8 *bssid, const u8 *pmkid, 158346981Scy const u8 *fils_cache_id) 159189251Ssam{ 160189251Ssam printf("%s - not implemented\n", __func__); 161189251Ssam return -1; 162189251Ssam} 163189251Ssam 164189251Ssam 165189251Ssamstatic void wpa_supplicant_set_config_blob(void *ctx, 166189251Ssam struct wpa_config_blob *blob) 167189251Ssam{ 168189251Ssam struct wpa_supplicant *wpa_s = ctx; 169189251Ssam wpa_config_set_blob(wpa_s->conf, blob); 170189251Ssam} 171189251Ssam 172189251Ssam 173189251Ssamstatic const struct wpa_config_blob * 174189251Ssamwpa_supplicant_get_config_blob(void *ctx, const char *name) 175189251Ssam{ 176189251Ssam struct wpa_supplicant *wpa_s = ctx; 177189251Ssam return wpa_config_get_blob(wpa_s->conf, name); 178189251Ssam} 179189251Ssam 180189251Ssam 181189251Ssamstatic void test_eapol_clean(struct wpa_supplicant *wpa_s) 182189251Ssam{ 183189251Ssam rsn_preauth_deinit(wpa_s->wpa); 184189251Ssam pmksa_candidate_free(wpa_s->wpa); 185189251Ssam wpa_sm_deinit(wpa_s->wpa); 186189251Ssam scard_deinit(wpa_s->scard); 187189251Ssam if (wpa_s->ctrl_iface) { 188189251Ssam wpa_supplicant_ctrl_iface_deinit(wpa_s->ctrl_iface); 189189251Ssam wpa_s->ctrl_iface = NULL; 190189251Ssam } 191189251Ssam wpa_config_free(wpa_s->conf); 192189251Ssam} 193189251Ssam 194189251Ssam 195189251Ssamstatic void eapol_test_timeout(void *eloop_ctx, void *timeout_ctx) 196189251Ssam{ 197189251Ssam struct preauth_test_data *p = eloop_ctx; 198189251Ssam printf("EAPOL test timed out\n"); 199189251Ssam p->auth_timed_out = 1; 200189251Ssam eloop_terminate(); 201189251Ssam} 202189251Ssam 203189251Ssam 204189251Ssamstatic void eapol_test_poll(void *eloop_ctx, void *timeout_ctx) 205189251Ssam{ 206189251Ssam struct wpa_supplicant *wpa_s = eloop_ctx; 207189251Ssam if (!rsn_preauth_in_progress(wpa_s->wpa)) 208189251Ssam eloop_terminate(); 209189251Ssam else { 210189251Ssam eloop_register_timeout(0, 100000, eapol_test_poll, eloop_ctx, 211189251Ssam timeout_ctx); 212189251Ssam } 213189251Ssam} 214189251Ssam 215189251Ssam 216189251Ssamstatic struct wpa_driver_ops dummy_driver; 217189251Ssam 218189251Ssam 219189251Ssamstatic void wpa_init_conf(struct wpa_supplicant *wpa_s, const char *ifname) 220189251Ssam{ 221189251Ssam struct l2_packet_data *l2; 222189251Ssam struct wpa_sm_ctx *ctx; 223189251Ssam 224189251Ssam os_memset(&dummy_driver, 0, sizeof(dummy_driver)); 225189251Ssam wpa_s->driver = &dummy_driver; 226189251Ssam 227189251Ssam ctx = os_zalloc(sizeof(*ctx)); 228189251Ssam assert(ctx != NULL); 229189251Ssam 230189251Ssam ctx->ctx = wpa_s; 231214734Srpaulo ctx->msg_ctx = wpa_s; 232189251Ssam ctx->set_state = _wpa_supplicant_set_state; 233189251Ssam ctx->get_state = _wpa_supplicant_get_state; 234189251Ssam ctx->deauthenticate = _wpa_supplicant_deauthenticate; 235189251Ssam ctx->set_key = wpa_supplicant_set_key; 236189251Ssam ctx->get_network_ctx = wpa_supplicant_get_network_ctx; 237189251Ssam ctx->get_bssid = wpa_supplicant_get_bssid; 238189251Ssam ctx->ether_send = wpa_ether_send; 239189251Ssam ctx->get_beacon_ie = wpa_supplicant_get_beacon_ie; 240189251Ssam ctx->alloc_eapol = _wpa_alloc_eapol; 241189251Ssam ctx->cancel_auth_timeout = _wpa_supplicant_cancel_auth_timeout; 242189251Ssam ctx->add_pmkid = wpa_supplicant_add_pmkid; 243189251Ssam ctx->remove_pmkid = wpa_supplicant_remove_pmkid; 244189251Ssam ctx->set_config_blob = wpa_supplicant_set_config_blob; 245189251Ssam ctx->get_config_blob = wpa_supplicant_get_config_blob; 246189251Ssam ctx->mlme_setprotection = wpa_supplicant_mlme_setprotection; 247189251Ssam 248189251Ssam wpa_s->wpa = wpa_sm_init(ctx); 249189251Ssam assert(wpa_s->wpa != NULL); 250189251Ssam wpa_sm_set_param(wpa_s->wpa, WPA_PARAM_PROTO, WPA_PROTO_RSN); 251189251Ssam 252189251Ssam os_strlcpy(wpa_s->ifname, ifname, sizeof(wpa_s->ifname)); 253189251Ssam wpa_sm_set_ifname(wpa_s->wpa, wpa_s->ifname, NULL); 254189251Ssam 255189251Ssam l2 = l2_packet_init(wpa_s->ifname, NULL, ETH_P_RSN_PREAUTH, NULL, 256189251Ssam NULL, 0); 257189251Ssam assert(l2 != NULL); 258189251Ssam if (l2_packet_get_own_addr(l2, wpa_s->own_addr)) { 259189251Ssam wpa_printf(MSG_WARNING, "Failed to get own L2 address\n"); 260189251Ssam exit(-1); 261189251Ssam } 262189251Ssam l2_packet_deinit(l2); 263189251Ssam wpa_sm_set_own_addr(wpa_s->wpa, wpa_s->own_addr); 264189251Ssam} 265189251Ssam 266189251Ssam 267214734Srpaulostatic void eapol_test_terminate(int sig, void *signal_ctx) 268189251Ssam{ 269214734Srpaulo struct wpa_supplicant *wpa_s = signal_ctx; 270189251Ssam wpa_msg(wpa_s, MSG_INFO, "Signal %d received - terminating", sig); 271189251Ssam eloop_terminate(); 272189251Ssam} 273189251Ssam 274189251Ssam 275189251Ssamint main(int argc, char *argv[]) 276189251Ssam{ 277189251Ssam struct wpa_supplicant wpa_s; 278189251Ssam int ret = 1; 279189251Ssam u8 bssid[ETH_ALEN]; 280189251Ssam struct preauth_test_data preauth_test; 281189251Ssam 282189251Ssam if (os_program_init()) 283189251Ssam return -1; 284189251Ssam 285189251Ssam os_memset(&preauth_test, 0, sizeof(preauth_test)); 286189251Ssam 287189251Ssam wpa_debug_level = 0; 288189251Ssam wpa_debug_show_keys = 1; 289189251Ssam 290189251Ssam if (argc != 4) { 291189251Ssam printf("usage: preauth_test <conf> <target MAC address> " 292189251Ssam "<ifname>\n"); 293189251Ssam return -1; 294189251Ssam } 295189251Ssam 296189251Ssam if (hwaddr_aton(argv[2], bssid)) { 297189251Ssam printf("Failed to parse target address '%s'.\n", argv[2]); 298189251Ssam return -1; 299189251Ssam } 300189251Ssam 301214734Srpaulo if (eap_register_methods()) { 302189251Ssam wpa_printf(MSG_ERROR, "Failed to register EAP methods"); 303189251Ssam return -1; 304189251Ssam } 305189251Ssam 306214734Srpaulo if (eloop_init()) { 307189251Ssam wpa_printf(MSG_ERROR, "Failed to initialize event loop"); 308189251Ssam return -1; 309189251Ssam } 310189251Ssam 311189251Ssam os_memset(&wpa_s, 0, sizeof(wpa_s)); 312281806Srpaulo wpa_s.conf = wpa_config_read(argv[1], NULL); 313189251Ssam if (wpa_s.conf == NULL) { 314189251Ssam printf("Failed to parse configuration file '%s'.\n", argv[1]); 315189251Ssam return -1; 316189251Ssam } 317189251Ssam if (wpa_s.conf->ssid == NULL) { 318189251Ssam printf("No networks defined.\n"); 319189251Ssam return -1; 320189251Ssam } 321189251Ssam 322189251Ssam wpa_init_conf(&wpa_s, argv[3]); 323189251Ssam wpa_s.ctrl_iface = wpa_supplicant_ctrl_iface_init(&wpa_s); 324189251Ssam if (wpa_s.ctrl_iface == NULL) { 325189251Ssam printf("Failed to initialize control interface '%s'.\n" 326189251Ssam "You may have another preauth_test process already " 327189251Ssam "running or the file was\n" 328189251Ssam "left by an unclean termination of preauth_test in " 329189251Ssam "which case you will need\n" 330189251Ssam "to manually remove this file before starting " 331189251Ssam "preauth_test again.\n", 332189251Ssam wpa_s.conf->ctrl_interface); 333189251Ssam return -1; 334189251Ssam } 335189251Ssam if (wpa_supplicant_scard_init(&wpa_s, wpa_s.conf->ssid)) 336189251Ssam return -1; 337189251Ssam 338189251Ssam if (rsn_preauth_init(wpa_s.wpa, bssid, &wpa_s.conf->ssid->eap)) 339189251Ssam return -1; 340189251Ssam 341189251Ssam eloop_register_timeout(30, 0, eapol_test_timeout, &preauth_test, NULL); 342189251Ssam eloop_register_timeout(0, 100000, eapol_test_poll, &wpa_s, NULL); 343214734Srpaulo eloop_register_signal_terminate(eapol_test_terminate, &wpa_s); 344214734Srpaulo eloop_register_signal_reconfig(eapol_test_terminate, &wpa_s); 345189251Ssam eloop_run(); 346189251Ssam 347189251Ssam if (preauth_test.auth_timed_out) 348189251Ssam ret = -2; 349189251Ssam else { 350346981Scy ret = pmksa_cache_set_current(wpa_s.wpa, NULL, bssid, NULL, 0, 351346981Scy NULL, 0) ? 0 : -3; 352189251Ssam } 353189251Ssam 354189251Ssam test_eapol_clean(&wpa_s); 355189251Ssam 356189251Ssam eap_peer_unregister_methods(); 357189251Ssam 358189251Ssam eloop_destroy(); 359189251Ssam 360189251Ssam os_program_deinit(); 361189251Ssam 362189251Ssam return ret; 363189251Ssam} 364