wlantest/wlantest.c

   1 /*
   2  * wlantest - IEEE 802.11 protocol monitoring and testing tool
   3  * Copyright (c) 2010, 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
  15 #include "utils/includes.h"
  16
  17 #include "utils/common.h"
  18 #include "utils/eloop.h"
  19 #include "wlantest.h"
  20
  21
  22 extern int wpa_debug_level;
  23 extern int wpa_debug_show_keys;
  24
  25
  26 static void wlantest_terminate(int sig, void *signal_ctx)
  27 {
  28         eloop_terminate();
  29 }
  30
  31
  32 static void usage(void)
  33 {
  34         printf("wlantest [-cddhqq] [-i<ifname>] [-r<pcap file>] "
  35                "[-p<passphrase>]\n"
  36                 "         [-I<wired ifname>] [-R<wired pcap file>] "
  37                "[-P<RADIUS shared secret>]\n"
  38                 "         [-w<write pcap file>]\n");
  39 }
  40
  41
  42 static void passphrase_deinit(struct wlantest_passphrase *p)
  43 {
  44         dl_list_del(&p->list);
  45         os_free(p);
  46 }
  47
  48
  49 static void secret_deinit(struct wlantest_radius_secret *r)
  50 {
  51         dl_list_del(&r->list);
  52         os_free(r);
  53 }
  54
  55
  56 static void wlantest_init(struct wlantest *wt)
  57 {
  58         int i;
  59         os_memset(wt, 0, sizeof(*wt));
  60         wt->monitor_sock = -1;
  61         wt->ctrl_sock = -1;
  62         for (i = 0; i < MAX_CTRL_CONNECTIONS; i++)
  63                 wt->ctrl_socks[i] = -1;
  64         dl_list_init(&wt->passphrase);
  65         dl_list_init(&wt->bss);
  66         dl_list_init(&wt->secret);
  67         dl_list_init(&wt->radius);
  68         dl_list_init(&wt->pmk);
  69 }
  70
  71
  72 void radius_deinit(struct wlantest_radius *r)
  73 {
  74         dl_list_del(&r->list);
  75         os_free(r);
  76 }
  77
  78
  79 static void wlantest_deinit(struct wlantest *wt)
  80 {
  81         struct wlantest_passphrase *p, *pn;
  82         struct wlantest_radius_secret *s, *sn;
  83         struct wlantest_radius *r, *rn;
  84         struct wlantest_pmk *pmk, *np;
  85
  86         if (wt->ctrl_sock >= 0)
  87                 ctrl_deinit(wt);
  88         if (wt->monitor_sock >= 0)
  89                 monitor_deinit(wt);
  90         bss_flush(wt);
  91         dl_list_for_each_safe(p, pn, &wt->passphrase,
  92                               struct wlantest_passphrase, list)
  93                 passphrase_deinit(p);
  94         dl_list_for_each_safe(s, sn, &wt->secret,
  95                               struct wlantest_radius_secret, list)
  96                 secret_deinit(s);
  97         dl_list_for_each_safe(r, rn, &wt->radius, struct wlantest_radius, list)
  98                 radius_deinit(r);
  99         dl_list_for_each_safe(pmk, np, &wt->pmk, struct wlantest_pmk, list)
 100                 pmk_deinit(pmk);
 101         write_pcap_deinit(wt);
 102 }
 103
 104
 105 static void add_passphrase(struct wlantest *wt, const char *passphrase)
 106 {
 107         struct wlantest_passphrase *p;
 108         size_t len = os_strlen(passphrase);
 109
 110         if (len < 8 || len > 63)
 111                 return;
 112         p = os_zalloc(sizeof(*p));
 113         if (p == NULL)
 114                 return;
 115         os_memcpy(p->passphrase, passphrase, len);
 116         dl_list_add(&wt->passphrase, &p->list);
 117 }
 118
 119
 120 static void add_secret(struct wlantest *wt, const char *secret)
 121 {
 122         struct wlantest_radius_secret *s;
 123         size_t len = os_strlen(secret);
 124
 125         if (len >= MAX_RADIUS_SECRET_LEN)
 126                 return;
 127         s = os_zalloc(sizeof(*s));
 128         if (s == NULL)
 129                 return;
 130         os_memcpy(s->secret, secret, len);
 131         dl_list_add(&wt->secret, &s->list);
 132 }
 133
 134
 135 int main(int argc, char *argv[])
 136 {
 137         int c;
 138         const char *read_file = NULL;
 139         const char *read_wired_file = NULL;
 140         const char *write_file = NULL;
 141         const char *ifname = NULL;
 142         const char *ifname_wired = NULL;
 143         struct wlantest wt;
 144         int ctrl_iface = 0;
 145
 146         wpa_debug_level = MSG_INFO;
 147         wpa_debug_show_keys = 1;
 148
 149         if (os_program_init())
 150                 return -1;
 151
 152         wlantest_init(&wt);
 153
 154         for (;;) {
 155                 c = getopt(argc, argv, "cdhi:I:p:P:qr:R:w:");
 156                 if (c < 0)
 157                         break;
 158                 switch (c) {
 159                 case 'c':
 160                         ctrl_iface = 1;
 161                         break;
 162                 case 'd':
 163                         if (wpa_debug_level > 0)
 164                                 wpa_debug_level--;
 165                         break;
 166                 case 'h':
 167                         usage();
 168                         return 0;
 169                 case 'i':
 170                         ifname = optarg;
 171                         break;
 172                 case 'I':
 173                         ifname_wired = optarg;
 174                         break;
 175                 case 'p':
 176                         add_passphrase(&wt, optarg);
 177                         break;
 178                 case 'P':
 179                         add_secret(&wt, optarg);
 180                         break;
 181                 case 'q':
 182                         wpa_debug_level++;
 183                         break;
 184                 case 'r':
 185                         read_file = optarg;
 186                         break;
 187                 case 'R':
 188                         read_wired_file = optarg;
 189                         break;
 190                 case 'w':
 191                         write_file = optarg;
 192                         break;
 193                 default:
 194                         usage();
 195                         return -1;
 196                 }
 197         }
 198
 199         if (ifname == NULL && ifname_wired == NULL &&
 200             read_file == NULL && read_wired_file == NULL) {
 201                 usage();
 202                 return 0;
 203         }
 204
 205         if (eloop_init())
 206                 return -1;
 207
 208         if (write_file && write_pcap_init(&wt, write_file) < 0)
 209                 return -1;
 210
 211         if (read_wired_file && read_wired_cap_file(&wt, read_wired_file) < 0)
 212                 return -1;
 213
 214         if (read_file && read_cap_file(&wt, read_file) < 0)
 215                 return -1;
 216
 217         if (ifname && monitor_init(&wt, ifname) < 0)
 218                 return -1;
 219
 220         if (ifname_wired && monitor_init_wired(&wt, ifname_wired) < 0)
 221                 return -1;
 222
 223         if (ctrl_iface && ctrl_init(&wt) < 0)
 224                 return -1;
 225
 226         eloop_register_signal_terminate(wlantest_terminate, &wt);
 227
 228         eloop_run();
 229
 230         wpa_printf(MSG_INFO, "Processed: rx_mgmt=%u rx_ctrl=%u rx_data=%u "
 231                    "fcs_error=%u",
 232                    wt.rx_mgmt, wt.rx_ctrl, wt.rx_data, wt.fcs_error);
 233
 234         wlantest_deinit(&wt);
 235
 236         eloop_destroy();
 237         os_program_deinit();
 238
 239         return 0;
 240 }