Use common driver code for Linux hwaddr get/set
[hostap-gosc2009.git] / wpa_supplicant / preauth_test.c
blobd38a6bb2d90c4a21915ec2e2c6542c08ce1381ac
1 /*
2 * WPA Supplicant - test code for pre-authentication
3 * Copyright (c) 2003-2007, Jouni Malinen <j@w1.fi>
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.
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
12 * See README and COPYING for more details.
14 * IEEE 802.1X Supplicant test code (to be used in place of wpa_supplicant.c.
15 * Not used in production version.
18 #include "includes.h"
19 #include <assert.h>
21 #include "common.h"
22 #include "config.h"
23 #include "eapol_supp/eapol_supp_sm.h"
24 #include "eloop.h"
25 #include "rsn_supp/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 "rsn_supp/preauth.h"
32 #include "rsn_supp/pmksa_cache.h"
33 #include "drivers/driver.h"
36 extern int wpa_debug_level;
37 extern int wpa_debug_show_keys;
39 struct wpa_driver_ops *wpa_drivers[] = { NULL };
42 struct preauth_test_data {
43 int auth_timed_out;
47 static void _wpa_supplicant_disassociate(void *wpa_s, int reason_code)
49 wpa_supplicant_disassociate(wpa_s, reason_code);
53 static void _wpa_supplicant_deauthenticate(void *wpa_s, int reason_code)
55 wpa_supplicant_deauthenticate(wpa_s, reason_code);
59 static u8 * wpa_alloc_eapol(const struct wpa_supplicant *wpa_s, u8 type,
60 const void *data, u16 data_len,
61 size_t *msg_len, void **data_pos)
63 struct ieee802_1x_hdr *hdr;
65 *msg_len = sizeof(*hdr) + data_len;
66 hdr = os_malloc(*msg_len);
67 if (hdr == NULL)
68 return NULL;
70 hdr->version = wpa_s->conf->eapol_version;
71 hdr->type = type;
72 hdr->length = htons(data_len);
74 if (data)
75 os_memcpy(hdr + 1, data, data_len);
76 else
77 os_memset(hdr + 1, 0, data_len);
79 if (data_pos)
80 *data_pos = hdr + 1;
82 return (u8 *) hdr;
86 static u8 * _wpa_alloc_eapol(void *wpa_s, u8 type,
87 const void *data, u16 data_len,
88 size_t *msg_len, void **data_pos)
90 return wpa_alloc_eapol(wpa_s, type, data, data_len, msg_len, data_pos);
94 static void _wpa_supplicant_set_state(void *ctx, enum wpa_states state)
96 struct wpa_supplicant *wpa_s = ctx;
97 wpa_s->wpa_state = state;
101 static enum wpa_states _wpa_supplicant_get_state(void *ctx)
103 struct wpa_supplicant *wpa_s = ctx;
104 return wpa_s->wpa_state;
108 static int wpa_ether_send(void *wpa_s, const u8 *dest, u16 proto,
109 const u8 *buf, size_t len)
111 printf("%s - not implemented\n", __func__);
112 return -1;
116 static void * wpa_supplicant_get_network_ctx(void *wpa_s)
118 return wpa_supplicant_get_ssid(wpa_s);
122 static void _wpa_supplicant_cancel_auth_timeout(void *wpa_s)
124 wpa_supplicant_cancel_auth_timeout(wpa_s);
128 static int wpa_supplicant_get_beacon_ie(void *wpa_s)
130 printf("%s - not implemented\n", __func__);
131 return -1;
135 static int wpa_supplicant_get_bssid(void *wpa_s, u8 *bssid)
137 printf("%s - not implemented\n", __func__);
138 return -1;
142 static int wpa_supplicant_set_key(void *wpa_s, enum wpa_alg alg,
143 const u8 *addr, int key_idx, int set_tx,
144 const u8 *seq, size_t seq_len,
145 const u8 *key, size_t key_len)
147 printf("%s - not implemented\n", __func__);
148 return -1;
152 static int wpa_supplicant_mlme_setprotection(void *wpa_s, const u8 *addr,
153 int protection_type,
154 int key_type)
156 printf("%s - not implemented\n", __func__);
157 return -1;
161 static int wpa_supplicant_add_pmkid(void *wpa_s,
162 const u8 *bssid, const u8 *pmkid)
164 printf("%s - not implemented\n", __func__);
165 return -1;
169 static int wpa_supplicant_remove_pmkid(void *wpa_s,
170 const u8 *bssid, const u8 *pmkid)
172 printf("%s - not implemented\n", __func__);
173 return -1;
177 static void wpa_supplicant_set_config_blob(void *ctx,
178 struct wpa_config_blob *blob)
180 struct wpa_supplicant *wpa_s = ctx;
181 wpa_config_set_blob(wpa_s->conf, blob);
185 static const struct wpa_config_blob *
186 wpa_supplicant_get_config_blob(void *ctx, const char *name)
188 struct wpa_supplicant *wpa_s = ctx;
189 return wpa_config_get_blob(wpa_s->conf, name);
193 static void test_eapol_clean(struct wpa_supplicant *wpa_s)
195 rsn_preauth_deinit(wpa_s->wpa);
196 pmksa_candidate_free(wpa_s->wpa);
197 wpa_sm_deinit(wpa_s->wpa);
198 scard_deinit(wpa_s->scard);
199 if (wpa_s->ctrl_iface) {
200 wpa_supplicant_ctrl_iface_deinit(wpa_s->ctrl_iface);
201 wpa_s->ctrl_iface = NULL;
203 wpa_config_free(wpa_s->conf);
207 static void eapol_test_timeout(void *eloop_ctx, void *timeout_ctx)
209 struct preauth_test_data *p = eloop_ctx;
210 printf("EAPOL test timed out\n");
211 p->auth_timed_out = 1;
212 eloop_terminate();
216 static void eapol_test_poll(void *eloop_ctx, void *timeout_ctx)
218 struct wpa_supplicant *wpa_s = eloop_ctx;
219 if (!rsn_preauth_in_progress(wpa_s->wpa))
220 eloop_terminate();
221 else {
222 eloop_register_timeout(0, 100000, eapol_test_poll, eloop_ctx,
223 timeout_ctx);
228 static struct wpa_driver_ops dummy_driver;
231 static void wpa_init_conf(struct wpa_supplicant *wpa_s, const char *ifname)
233 struct l2_packet_data *l2;
234 struct wpa_sm_ctx *ctx;
236 os_memset(&dummy_driver, 0, sizeof(dummy_driver));
237 wpa_s->driver = &dummy_driver;
239 ctx = os_zalloc(sizeof(*ctx));
240 assert(ctx != NULL);
242 ctx->ctx = wpa_s;
243 ctx->msg_ctx = wpa_s;
244 ctx->set_state = _wpa_supplicant_set_state;
245 ctx->get_state = _wpa_supplicant_get_state;
246 ctx->deauthenticate = _wpa_supplicant_deauthenticate;
247 ctx->disassociate = _wpa_supplicant_disassociate;
248 ctx->set_key = wpa_supplicant_set_key;
249 ctx->get_network_ctx = wpa_supplicant_get_network_ctx;
250 ctx->get_bssid = wpa_supplicant_get_bssid;
251 ctx->ether_send = wpa_ether_send;
252 ctx->get_beacon_ie = wpa_supplicant_get_beacon_ie;
253 ctx->alloc_eapol = _wpa_alloc_eapol;
254 ctx->cancel_auth_timeout = _wpa_supplicant_cancel_auth_timeout;
255 ctx->add_pmkid = wpa_supplicant_add_pmkid;
256 ctx->remove_pmkid = wpa_supplicant_remove_pmkid;
257 ctx->set_config_blob = wpa_supplicant_set_config_blob;
258 ctx->get_config_blob = wpa_supplicant_get_config_blob;
259 ctx->mlme_setprotection = wpa_supplicant_mlme_setprotection;
261 wpa_s->wpa = wpa_sm_init(ctx);
262 assert(wpa_s->wpa != NULL);
263 wpa_sm_set_param(wpa_s->wpa, WPA_PARAM_PROTO, WPA_PROTO_RSN);
265 os_strlcpy(wpa_s->ifname, ifname, sizeof(wpa_s->ifname));
266 wpa_sm_set_ifname(wpa_s->wpa, wpa_s->ifname, NULL);
268 l2 = l2_packet_init(wpa_s->ifname, NULL, ETH_P_RSN_PREAUTH, NULL,
269 NULL, 0);
270 assert(l2 != NULL);
271 if (l2_packet_get_own_addr(l2, wpa_s->own_addr)) {
272 wpa_printf(MSG_WARNING, "Failed to get own L2 address\n");
273 exit(-1);
275 l2_packet_deinit(l2);
276 wpa_sm_set_own_addr(wpa_s->wpa, wpa_s->own_addr);
280 static void eapol_test_terminate(int sig, void *signal_ctx)
282 struct wpa_supplicant *wpa_s = signal_ctx;
283 wpa_msg(wpa_s, MSG_INFO, "Signal %d received - terminating", sig);
284 eloop_terminate();
288 int main(int argc, char *argv[])
290 struct wpa_supplicant wpa_s;
291 int ret = 1;
292 u8 bssid[ETH_ALEN];
293 struct preauth_test_data preauth_test;
295 if (os_program_init())
296 return -1;
298 os_memset(&preauth_test, 0, sizeof(preauth_test));
300 wpa_debug_level = 0;
301 wpa_debug_show_keys = 1;
303 if (argc != 4) {
304 printf("usage: preauth_test <conf> <target MAC address> "
305 "<ifname>\n");
306 return -1;
309 if (hwaddr_aton(argv[2], bssid)) {
310 printf("Failed to parse target address '%s'.\n", argv[2]);
311 return -1;
314 if (eap_register_methods()) {
315 wpa_printf(MSG_ERROR, "Failed to register EAP methods");
316 return -1;
319 if (eloop_init()) {
320 wpa_printf(MSG_ERROR, "Failed to initialize event loop");
321 return -1;
324 os_memset(&wpa_s, 0, sizeof(wpa_s));
325 wpa_s.conf = wpa_config_read(argv[1]);
326 if (wpa_s.conf == NULL) {
327 printf("Failed to parse configuration file '%s'.\n", argv[1]);
328 return -1;
330 if (wpa_s.conf->ssid == NULL) {
331 printf("No networks defined.\n");
332 return -1;
335 wpa_init_conf(&wpa_s, argv[3]);
336 wpa_s.ctrl_iface = wpa_supplicant_ctrl_iface_init(&wpa_s);
337 if (wpa_s.ctrl_iface == NULL) {
338 printf("Failed to initialize control interface '%s'.\n"
339 "You may have another preauth_test process already "
340 "running or the file was\n"
341 "left by an unclean termination of preauth_test in "
342 "which case you will need\n"
343 "to manually remove this file before starting "
344 "preauth_test again.\n",
345 wpa_s.conf->ctrl_interface);
346 return -1;
348 if (wpa_supplicant_scard_init(&wpa_s, wpa_s.conf->ssid))
349 return -1;
351 if (rsn_preauth_init(wpa_s.wpa, bssid, &wpa_s.conf->ssid->eap))
352 return -1;
354 eloop_register_timeout(30, 0, eapol_test_timeout, &preauth_test, NULL);
355 eloop_register_timeout(0, 100000, eapol_test_poll, &wpa_s, NULL);
356 eloop_register_signal_terminate(eapol_test_terminate, &wpa_s);
357 eloop_register_signal_reconfig(eapol_test_terminate, &wpa_s);
358 eloop_run();
360 if (preauth_test.auth_timed_out)
361 ret = -2;
362 else {
363 ret = pmksa_cache_set_current(wpa_s.wpa, NULL, bssid, NULL, 0)
364 ? 0 : -3;
367 test_eapol_clean(&wpa_s);
369 eap_peer_unregister_methods();
371 eloop_destroy();
373 os_program_deinit();
375 return ret;