preauth_test.c revision 189251 
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
35extern int wpa_debug_level;
36extern int wpa_debug_show_keys;
37
38struct wpa_driver_ops *wpa_supplicant_drivers[] = { NULL };
39
40
41struct preauth_test_data {
42	int auth_timed_out;
43};
44
45
46static void _wpa_supplicant_disassociate(void *wpa_s, int reason_code)
47{
48	wpa_supplicant_disassociate(wpa_s, reason_code);
49}
50
51
52static void _wpa_supplicant_deauthenticate(void *wpa_s, int reason_code)
53{
54	wpa_supplicant_deauthenticate(wpa_s, reason_code);
55}
56
57
58static 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
85static 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
93static 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
100static 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
107static 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
115static void * wpa_supplicant_get_network_ctx(void *wpa_s)
116{
117	return wpa_supplicant_get_ssid(wpa_s);
118}
119
120
121static void _wpa_supplicant_cancel_auth_timeout(void *wpa_s)
122{
123	wpa_supplicant_cancel_auth_timeout(wpa_s);
124}
125
126
127static int wpa_supplicant_get_beacon_ie(void *wpa_s)
128{
129	printf("%s - not implemented\n", __func__);
130	return -1;
131}
132
133
134static int wpa_supplicant_get_bssid(void *wpa_s, u8 *bssid)
135{
136	printf("%s - not implemented\n", __func__);
137	return -1;
138}
139
140
141static 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
151static 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
160static 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
168static 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
176static 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
184static const struct wpa_config_blob *
185wpa_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
192static 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
206static 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
215static 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
227static struct wpa_driver_ops dummy_driver;
228
229
230static 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
278static 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
287int 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