1 /* 2 * WPA Supplicant - test code for pre-authentication 3 * Copyright (c) 2003-2007, Jouni Malinen <j@w1.fi> 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of the GNU General Public License version 2 as 7 * published by the Free Software Foundation. 8 * 9 * Alternatively, this software may be distributed under the terms of BSD 10 * license. 11 * 12 * See README and COPYING for more details. 13 * 14 * IEEE 802.1X Supplicant test code (to be used in place of wpa_supplicant.c. 15 * Not used in production version. 16 */ 17 18 #include "includes.h" 19 #include <assert.h> 20 21 #include "common.h" 22 #include "config.h" 23 #include "eapol_supp/eapol_supp_sm.h" 24 #include "eloop.h" 25 #include "wpa.h" 26 #include "eap_peer/eap.h" 27 #include "wpa_supplicant_i.h" 28 #include "l2_packet/l2_packet.h" 29 #include "ctrl_iface.h" 30 #include "pcsc_funcs.h" 31 #include "preauth.h" 32 #include "pmksa_cache.h" 33 34 35 extern int wpa_debug_level; 36 extern int wpa_debug_show_keys; 37 38 struct wpa_driver_ops *wpa_supplicant_drivers[] = { NULL }; 39 40 41 struct preauth_test_data { 42 int auth_timed_out; 43 }; 44 45 46 static void _wpa_supplicant_disassociate(void *wpa_s, int reason_code) 47 { 48 wpa_supplicant_disassociate(wpa_s, reason_code); 49 } 50 51 52 static void _wpa_supplicant_deauthenticate(void *wpa_s, int reason_code) 53 { 54 wpa_supplicant_deauthenticate(wpa_s, reason_code); 55 } 56 57 58 static u8 * wpa_alloc_eapol(const struct wpa_supplicant *wpa_s, u8 type, 59 const void *data, u16 data_len, 60 size_t *msg_len, void **data_pos) 61 { 62 struct ieee802_1x_hdr *hdr; 63 64 *msg_len = sizeof(*hdr) + data_len; 65 hdr = os_malloc(*msg_len); 66 if (hdr == NULL) 67 return NULL; 68 69 hdr->version = wpa_s->conf->eapol_version; 70 hdr->type = type; 71 hdr->length = htons(data_len); 72 73 if (data) 74 os_memcpy(hdr + 1, data, data_len); 75 else 76 os_memset(hdr + 1, 0, data_len); 77 78 if (data_pos) 79 *data_pos = hdr + 1; 80 81 return (u8 *) hdr; 82 } 83 84 85 static u8 * _wpa_alloc_eapol(void *wpa_s, u8 type, 86 const void *data, u16 data_len, 87 size_t *msg_len, void **data_pos) 88 { 89 return wpa_alloc_eapol(wpa_s, type, data, data_len, msg_len, data_pos); 90 } 91 92 93 static void _wpa_supplicant_set_state(void *ctx, wpa_states state) 94 { 95 struct wpa_supplicant *wpa_s = ctx; 96 wpa_s->wpa_state = state; 97 } 98 99 100 static wpa_states _wpa_supplicant_get_state(void *ctx) 101 { 102 struct wpa_supplicant *wpa_s = ctx; 103 return wpa_s->wpa_state; 104 } 105 106 107 static int wpa_ether_send(void *wpa_s, const u8 *dest, u16 proto, 108 const u8 *buf, size_t len) 109 { 110 printf("%s - not implemented\n", __func__); 111 return -1; 112 } 113 114 115 static void * wpa_supplicant_get_network_ctx(void *wpa_s) 116 { 117 return wpa_supplicant_get_ssid(wpa_s); 118 } 119 120 121 static void _wpa_supplicant_cancel_auth_timeout(void *wpa_s) 122 { 123 wpa_supplicant_cancel_auth_timeout(wpa_s); 124 } 125 126 127 static int wpa_supplicant_get_beacon_ie(void *wpa_s) 128 { 129 printf("%s - not implemented\n", __func__); 130 return -1; 131 } 132 133 134 static int wpa_supplicant_get_bssid(void *wpa_s, u8 *bssid) 135 { 136 printf("%s - not implemented\n", __func__); 137 return -1; 138 } 139 140 141 static int wpa_supplicant_set_key(void *wpa_s, wpa_alg alg, 142 const u8 *addr, int key_idx, int set_tx, 143 const u8 *seq, size_t seq_len, 144 const u8 *key, size_t key_len) 145 { 146 printf("%s - not implemented\n", __func__); 147 return -1; 148 } 149 150 151 static int wpa_supplicant_mlme_setprotection(void *wpa_s, const u8 *addr, 152 int protection_type, 153 int key_type) 154 { 155 printf("%s - not implemented\n", __func__); 156 return -1; 157 } 158 159 160 static int wpa_supplicant_add_pmkid(void *wpa_s, 161 const u8 *bssid, const u8 *pmkid) 162 { 163 printf("%s - not implemented\n", __func__); 164 return -1; 165 } 166 167 168 static int wpa_supplicant_remove_pmkid(void *wpa_s, 169 const u8 *bssid, const u8 *pmkid) 170 { 171 printf("%s - not implemented\n", __func__); 172 return -1; 173 } 174 175 176 static void wpa_supplicant_set_config_blob(void *ctx, 177 struct wpa_config_blob *blob) 178 { 179 struct wpa_supplicant *wpa_s = ctx; 180 wpa_config_set_blob(wpa_s->conf, blob); 181 } 182 183 184 static const struct wpa_config_blob * 185 wpa_supplicant_get_config_blob(void *ctx, const char *name) 186 { 187 struct wpa_supplicant *wpa_s = ctx; 188 return wpa_config_get_blob(wpa_s->conf, name); 189 } 190 191 192 static void test_eapol_clean(struct wpa_supplicant *wpa_s) 193 { 194 rsn_preauth_deinit(wpa_s->wpa); 195 pmksa_candidate_free(wpa_s->wpa); 196 wpa_sm_deinit(wpa_s->wpa); 197 scard_deinit(wpa_s->scard); 198 if (wpa_s->ctrl_iface) { 199 wpa_supplicant_ctrl_iface_deinit(wpa_s->ctrl_iface); 200 wpa_s->ctrl_iface = NULL; 201 } 202 wpa_config_free(wpa_s->conf); 203 } 204 205 206 static void eapol_test_timeout(void *eloop_ctx, void *timeout_ctx) 207 { 208 struct preauth_test_data *p = eloop_ctx; 209 printf("EAPOL test timed out\n"); 210 p->auth_timed_out = 1; 211 eloop_terminate(); 212 } 213 214 215 static void eapol_test_poll(void *eloop_ctx, void *timeout_ctx) 216 { 217 struct wpa_supplicant *wpa_s = eloop_ctx; 218 if (!rsn_preauth_in_progress(wpa_s->wpa)) 219 eloop_terminate(); 220 else { 221 eloop_register_timeout(0, 100000, eapol_test_poll, eloop_ctx, 222 timeout_ctx); 223 } 224 } 225 226 227 static struct wpa_driver_ops dummy_driver; 228 229 230 static void wpa_init_conf(struct wpa_supplicant *wpa_s, const char *ifname) 231 { 232 struct l2_packet_data *l2; 233 struct wpa_sm_ctx *ctx; 234 235 os_memset(&dummy_driver, 0, sizeof(dummy_driver)); 236 wpa_s->driver = &dummy_driver; 237 238 ctx = os_zalloc(sizeof(*ctx)); 239 assert(ctx != NULL); 240 241 ctx->ctx = wpa_s; 242 ctx->set_state = _wpa_supplicant_set_state; 243 ctx->get_state = _wpa_supplicant_get_state; 244 ctx->deauthenticate = _wpa_supplicant_deauthenticate; 245 ctx->disassociate = _wpa_supplicant_disassociate; 246 ctx->set_key = wpa_supplicant_set_key; 247 ctx->get_network_ctx = wpa_supplicant_get_network_ctx; 248 ctx->get_bssid = wpa_supplicant_get_bssid; 249 ctx->ether_send = wpa_ether_send; 250 ctx->get_beacon_ie = wpa_supplicant_get_beacon_ie; 251 ctx->alloc_eapol = _wpa_alloc_eapol; 252 ctx->cancel_auth_timeout = _wpa_supplicant_cancel_auth_timeout; 253 ctx->add_pmkid = wpa_supplicant_add_pmkid; 254 ctx->remove_pmkid = wpa_supplicant_remove_pmkid; 255 ctx->set_config_blob = wpa_supplicant_set_config_blob; 256 ctx->get_config_blob = wpa_supplicant_get_config_blob; 257 ctx->mlme_setprotection = wpa_supplicant_mlme_setprotection; 258 259 wpa_s->wpa = wpa_sm_init(ctx); 260 assert(wpa_s->wpa != NULL); 261 wpa_sm_set_param(wpa_s->wpa, WPA_PARAM_PROTO, WPA_PROTO_RSN); 262 263 os_strlcpy(wpa_s->ifname, ifname, sizeof(wpa_s->ifname)); 264 wpa_sm_set_ifname(wpa_s->wpa, wpa_s->ifname, NULL); 265 266 l2 = l2_packet_init(wpa_s->ifname, NULL, ETH_P_RSN_PREAUTH, NULL, 267 NULL, 0); 268 assert(l2 != NULL); 269 if (l2_packet_get_own_addr(l2, wpa_s->own_addr)) { 270 wpa_printf(MSG_WARNING, "Failed to get own L2 address\n"); 271 exit(-1); 272 } 273 l2_packet_deinit(l2); 274 wpa_sm_set_own_addr(wpa_s->wpa, wpa_s->own_addr); 275 } 276 277 278 static void eapol_test_terminate(int sig, void *eloop_ctx, 279 void *signal_ctx) 280 { 281 struct wpa_supplicant *wpa_s = eloop_ctx; 282 wpa_msg(wpa_s, MSG_INFO, "Signal %d received - terminating", sig); 283 eloop_terminate(); 284 } 285 286 287 int main(int argc, char *argv[]) 288 { 289 struct wpa_supplicant wpa_s; 290 int ret = 1; 291 u8 bssid[ETH_ALEN]; 292 struct preauth_test_data preauth_test; 293 294 if (os_program_init()) 295 return -1; 296 297 os_memset(&preauth_test, 0, sizeof(preauth_test)); 298 299 wpa_debug_level = 0; 300 wpa_debug_show_keys = 1; 301 302 if (argc != 4) { 303 printf("usage: preauth_test <conf> <target MAC address> " 304 "<ifname>\n"); 305 return -1; 306 } 307 308 if (hwaddr_aton(argv[2], bssid)) { 309 printf("Failed to parse target address '%s'.\n", argv[2]); 310 return -1; 311 } 312 313 if (eap_peer_register_methods()) { 314 wpa_printf(MSG_ERROR, "Failed to register EAP methods"); 315 return -1; 316 } 317 318 if (eloop_init(&wpa_s)) { 319 wpa_printf(MSG_ERROR, "Failed to initialize event loop"); 320 return -1; 321 } 322 323 os_memset(&wpa_s, 0, sizeof(wpa_s)); 324 wpa_s.conf = wpa_config_read(argv[1]); 325 if (wpa_s.conf == NULL) { 326 printf("Failed to parse configuration file '%s'.\n", argv[1]); 327 return -1; 328 } 329 if (wpa_s.conf->ssid == NULL) { 330 printf("No networks defined.\n"); 331 return -1; 332 } 333 334 wpa_init_conf(&wpa_s, argv[3]); 335 wpa_s.ctrl_iface = wpa_supplicant_ctrl_iface_init(&wpa_s); 336 if (wpa_s.ctrl_iface == NULL) { 337 printf("Failed to initialize control interface '%s'.\n" 338 "You may have another preauth_test process already " 339 "running or the file was\n" 340 "left by an unclean termination of preauth_test in " 341 "which case you will need\n" 342 "to manually remove this file before starting " 343 "preauth_test again.\n", 344 wpa_s.conf->ctrl_interface); 345 return -1; 346 } 347 if (wpa_supplicant_scard_init(&wpa_s, wpa_s.conf->ssid)) 348 return -1; 349 350 if (rsn_preauth_init(wpa_s.wpa, bssid, &wpa_s.conf->ssid->eap)) 351 return -1; 352 353 eloop_register_timeout(30, 0, eapol_test_timeout, &preauth_test, NULL); 354 eloop_register_timeout(0, 100000, eapol_test_poll, &wpa_s, NULL); 355 eloop_register_signal_terminate(eapol_test_terminate, NULL); 356 eloop_register_signal_reconfig(eapol_test_terminate, NULL); 357 eloop_run(); 358 359 if (preauth_test.auth_timed_out) 360 ret = -2; 361 else { 362 ret = pmksa_cache_set_current(wpa_s.wpa, NULL, bssid, NULL, 0) 363 ? 0 : -3; 364 } 365 366 test_eapol_clean(&wpa_s); 367 368 eap_peer_unregister_methods(); 369 370 eloop_destroy(); 371 372 os_program_deinit(); 373 374 return ret; 375 } 376