preauth_test.c

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