driver_test.c

   1 /*
   2  * WPA Supplicant - testing driver interface
   3  * Copyright (c) 2004-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
  15 #include "includes.h"
  16 #include <sys/un.h>
  17 #include <dirent.h>
  18 #include <sys/stat.h>
  19
  20 #include "common.h"
  21 #include "driver.h"
  22 #include "wpa_supplicant.h"
  23 #include "l2_packet.h"
  24 #include "eloop.h"
  25 #include "sha1.h"
  26 #include "wpa.h"
  27 #include "mlme.h"
  28
  29
  30 struct wpa_driver_test_data {
  31         void *ctx;
  32         u8 own_addr[ETH_ALEN];
  33         int test_socket;
  34         struct sockaddr_un hostapd_addr;
  35         int hostapd_addr_set;
  36         char *own_socket_path;
  37         char *test_dir;
  38         u8 bssid[ETH_ALEN];
  39         u8 ssid[32];
  40         size_t ssid_len;
  41 #define MAX_SCAN_RESULTS 30
  42         struct wpa_scan_result scanres[MAX_SCAN_RESULTS];
  43         size_t num_scanres;
  44         int use_associnfo;
  45         u8 assoc_wpa_ie[80];
  46         size_t assoc_wpa_ie_len;
  47         int use_mlme;
  48 };
  49
  50
  51 static int wpa_driver_test_set_wpa(void *priv, int enabled)
  52 {
  53         wpa_printf(MSG_DEBUG, "%s: enabled=%d", __func__, enabled);
  54         return 0;
  55 }
  56
  57
  58 static void wpa_driver_test_scan_timeout(void *eloop_ctx, void *timeout_ctx)
  59 {
  60         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
  61         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
  62 }
  63
  64
  65 static void wpa_driver_scan_dir(struct wpa_driver_test_data *drv,
  66                                 const char *path)
  67 {
  68         struct dirent *dent;
  69         DIR *dir;
  70         struct sockaddr_un addr;
  71
  72         dir = opendir(path);
  73         if (dir == NULL)
  74                 return;
  75
  76         while ((dent = readdir(dir))) {
  77                 if (os_strncmp(dent->d_name, "AP-", 3) != 0)
  78                         continue;
  79                 wpa_printf(MSG_DEBUG, "%s: SCAN %s", __func__, dent->d_name);
  80
  81                 os_memset(&addr, 0, sizeof(addr));
  82                 addr.sun_family = AF_UNIX;
  83                 os_snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s",
  84                             path, dent->d_name);
  85
  86                 if (sendto(drv->test_socket, "SCAN", 4, 0,
  87                            (struct sockaddr *) &addr, sizeof(addr)) < 0) {
  88                         perror("sendto(test_socket)");
  89                 }
  90         }
  91         closedir(dir);
  92 }
  93
  94
  95 static int wpa_driver_test_scan(void *priv, const u8 *ssid, size_t ssid_len)
  96 {
  97         struct wpa_driver_test_data *drv = priv;
  98         wpa_printf(MSG_DEBUG, "%s: priv=%p", __func__, priv);
  99
 100         drv->num_scanres = 0;
 101
 102         if (drv->test_socket >= 0 && drv->test_dir)
 103                 wpa_driver_scan_dir(drv, drv->test_dir);
 104
 105         if (drv->test_socket >= 0 && drv->hostapd_addr_set &&
 106             sendto(drv->test_socket, "SCAN", 4, 0,
 107                    (struct sockaddr *) &drv->hostapd_addr,
 108                    sizeof(drv->hostapd_addr)) < 0) {
 109                 perror("sendto(test_socket)");
 110         }
 111
 112         eloop_cancel_timeout(wpa_driver_test_scan_timeout, drv, drv->ctx);
 113         eloop_register_timeout(1, 0, wpa_driver_test_scan_timeout, drv,
 114                                drv->ctx);
 115         return 0;
 116 }
 117
 118
 119 static int wpa_driver_test_get_scan_results(void *priv,
 120                                             struct wpa_scan_result *results,
 121                                             size_t max_size)
 122 {
 123         struct wpa_driver_test_data *drv = priv;
 124         size_t num = drv->num_scanres;
 125         if (num > max_size)
 126                 num = max_size;
 127         os_memcpy(results, &drv->scanres,
 128                   num * sizeof(struct wpa_scan_result));
 129         return num;
 130 }
 131
 132
 133 static int wpa_driver_test_set_key(void *priv, wpa_alg alg, const u8 *addr,
 134                                    int key_idx, int set_tx,
 135                                    const u8 *seq, size_t seq_len,
 136                                    const u8 *key, size_t key_len)
 137 {
 138         wpa_printf(MSG_DEBUG, "%s: priv=%p alg=%d key_idx=%d set_tx=%d",
 139                    __func__, priv, alg, key_idx, set_tx);
 140         if (addr) {
 141                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
 142         }
 143         if (seq) {
 144                 wpa_hexdump(MSG_DEBUG, "   seq", seq, seq_len);
 145         }
 146         if (key) {
 147                 wpa_hexdump(MSG_DEBUG, "   key", key, key_len);
 148         }
 149         return 0;
 150 }
 151
 152
 153 static int wpa_driver_test_associate(
 154         void *priv, struct wpa_driver_associate_params *params)
 155 {
 156         struct wpa_driver_test_data *drv = priv;
 157         wpa_printf(MSG_DEBUG, "%s: priv=%p freq=%d pairwise_suite=%d "
 158                    "group_suite=%d key_mgmt_suite=%d auth_alg=%d mode=%d",
 159                    __func__, priv, params->freq, params->pairwise_suite,
 160                    params->group_suite, params->key_mgmt_suite,
 161                    params->auth_alg, params->mode);
 162         if (params->bssid) {
 163                 wpa_printf(MSG_DEBUG, "   bssid=" MACSTR,
 164                            MAC2STR(params->bssid));
 165         }
 166         if (params->ssid) {
 167                 wpa_hexdump_ascii(MSG_DEBUG, "   ssid",
 168                                   params->ssid, params->ssid_len);
 169         }
 170         if (params->wpa_ie) {
 171                 wpa_hexdump(MSG_DEBUG, "   wpa_ie",
 172                             params->wpa_ie, params->wpa_ie_len);
 173                 drv->assoc_wpa_ie_len = params->wpa_ie_len;
 174                 if (drv->assoc_wpa_ie_len > sizeof(drv->assoc_wpa_ie))
 175                         drv->assoc_wpa_ie_len = sizeof(drv->assoc_wpa_ie);
 176                 os_memcpy(drv->assoc_wpa_ie, params->wpa_ie,
 177                           drv->assoc_wpa_ie_len);
 178         } else
 179                 drv->assoc_wpa_ie_len = 0;
 180
 181         if (drv->test_dir && params->bssid) {
 182                 os_memset(&drv->hostapd_addr, 0, sizeof(drv->hostapd_addr));
 183                 drv->hostapd_addr.sun_family = AF_UNIX;
 184                 os_snprintf(drv->hostapd_addr.sun_path,
 185                             sizeof(drv->hostapd_addr.sun_path),
 186                             "%s/AP-" MACSTR,
 187                             drv->test_dir, MAC2STR(params->bssid));
 188                 drv->hostapd_addr_set = 1;
 189         }
 190
 191         if (drv->test_socket >= 0 && drv->hostapd_addr_set) {
 192                 char cmd[200], *pos, *end;
 193                 int ret;
 194                 end = cmd + sizeof(cmd);
 195                 pos = cmd;
 196                 ret = os_snprintf(pos, end - pos, "ASSOC " MACSTR " ",
 197                                   MAC2STR(drv->own_addr));
 198                 if (ret >= 0 && ret < end - pos)
 199                         pos += ret;
 200                 pos += wpa_snprintf_hex(pos, end - pos, params->ssid,
 201                                         params->ssid_len);
 202                 ret = os_snprintf(pos, end - pos, " ");
 203                 if (ret >= 0 && ret < end - pos)
 204                         pos += ret;
 205                 pos += wpa_snprintf_hex(pos, end - pos, params->wpa_ie,
 206                                         params->wpa_ie_len);
 207                 end[-1] = '\0';
 208                 if (sendto(drv->test_socket, cmd, os_strlen(cmd), 0,
 209                            (struct sockaddr *) &drv->hostapd_addr,
 210                            sizeof(drv->hostapd_addr)) < 0) {
 211                         perror("sendto(test_socket)");
 212                         return -1;
 213                 }
 214
 215                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
 216                 drv->ssid_len = params->ssid_len;
 217         } else
 218                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
 219
 220         return 0;
 221 }
 222
 223
 224 static int wpa_driver_test_get_bssid(void *priv, u8 *bssid)
 225 {
 226         struct wpa_driver_test_data *drv = priv;
 227         os_memcpy(bssid, drv->bssid, ETH_ALEN);
 228         return 0;
 229 }
 230
 231
 232 static int wpa_driver_test_get_ssid(void *priv, u8 *ssid)
 233 {
 234         struct wpa_driver_test_data *drv = priv;
 235         os_memcpy(ssid, drv->ssid, 32);
 236         return drv->ssid_len;
 237 }
 238
 239
 240 static int wpa_driver_test_send_disassoc(struct wpa_driver_test_data *drv)
 241 {
 242         if (drv->test_socket >= 0 &&
 243             sendto(drv->test_socket, "DISASSOC", 8, 0,
 244                    (struct sockaddr *) &drv->hostapd_addr,
 245                    sizeof(drv->hostapd_addr)) < 0) {
 246                 perror("sendto(test_socket)");
 247                 return -1;
 248         }
 249         return 0;
 250 }
 251
 252
 253 static int wpa_driver_test_deauthenticate(void *priv, const u8 *addr,
 254                                           int reason_code)
 255 {
 256         struct wpa_driver_test_data *drv = priv;
 257         wpa_printf(MSG_DEBUG, "%s addr=" MACSTR " reason_code=%d",
 258                    __func__, MAC2STR(addr), reason_code);
 259         os_memset(drv->bssid, 0, ETH_ALEN);
 260         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
 261         return wpa_driver_test_send_disassoc(drv);
 262 }
 263
 264
 265 static int wpa_driver_test_disassociate(void *priv, const u8 *addr,
 266                                         int reason_code)
 267 {
 268         struct wpa_driver_test_data *drv = priv;
 269         wpa_printf(MSG_DEBUG, "%s addr=" MACSTR " reason_code=%d",
 270                    __func__, MAC2STR(addr), reason_code);
 271         os_memset(drv->bssid, 0, ETH_ALEN);
 272         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
 273         return wpa_driver_test_send_disassoc(drv);
 274 }
 275
 276
 277 static void wpa_driver_test_scanresp(struct wpa_driver_test_data *drv,
 278                                      struct sockaddr_un *from,
 279                                      socklen_t fromlen,
 280                                      const char *data)
 281 {
 282         struct wpa_scan_result *res;
 283         const char *pos, *pos2;
 284         size_t len;
 285         u8 ie[200], *ipos, *end;
 286
 287         wpa_printf(MSG_DEBUG, "test_driver: SCANRESP %s", data);
 288         if (drv->num_scanres >= MAX_SCAN_RESULTS) {
 289                 wpa_printf(MSG_DEBUG, "test_driver: No room for the new scan "
 290                            "result");
 291                 return;
 292         }
 293
 294         /* SCANRESP BSSID SSID IEs */
 295         res = &drv->scanres[drv->num_scanres];
 296
 297         os_memset(res, 0, sizeof(*res));
 298         if (hwaddr_aton(data, res->bssid)) {
 299                 wpa_printf(MSG_DEBUG, "test_driver: invalid BSSID in scanres");
 300                 return;
 301         }
 302
 303         pos = data + 17;
 304         while (*pos == ' ')
 305                 pos++;
 306         pos2 = os_strchr(pos, ' ');
 307         if (pos2 == NULL) {
 308                 wpa_printf(MSG_DEBUG, "test_driver: invalid SSID termination "
 309                            "in scanres");
 310                 return;
 311         }
 312         len = (pos2 - pos) / 2;
 313         if (len > sizeof(res->ssid))
 314                 len = sizeof(res->ssid);
 315         if (hexstr2bin(pos, res->ssid, len) < 0) {
 316                 wpa_printf(MSG_DEBUG, "test_driver: invalid SSID in scanres");
 317                 return;
 318         }
 319         res->ssid_len = len;
 320
 321         pos = pos2 + 1;
 322         pos2 = os_strchr(pos, ' ');
 323         if (pos2 == NULL)
 324                 len = os_strlen(pos) / 2;
 325         else
 326                 len = (pos2 - pos) / 2;
 327         if (len > sizeof(ie))
 328                 len = sizeof(ie);
 329         if (hexstr2bin(pos, ie, len) < 0) {
 330                 wpa_printf(MSG_DEBUG, "test_driver: invalid IEs in scanres");
 331                 return;
 332         }
 333
 334         ipos = ie;
 335         end = ipos + len;
 336         while (ipos + 1 < end && ipos + 2 + ipos[1] <= end) {
 337                 len = 2 + ipos[1];
 338                 if (len > SSID_MAX_WPA_IE_LEN)
 339                         len = SSID_MAX_WPA_IE_LEN;
 340                 if (ipos[0] == RSN_INFO_ELEM) {
 341                         os_memcpy(res->rsn_ie, ipos, len);
 342                         res->rsn_ie_len = len;
 343                 } else if (ipos[0] == GENERIC_INFO_ELEM) {
 344                         os_memcpy(res->wpa_ie, ipos, len);
 345                         res->wpa_ie_len = len;
 346                 }
 347
 348                 ipos += 2 + ipos[1];
 349         }
 350
 351         if (pos2) {
 352                 pos = pos2 + 1;
 353                 while (*pos == ' ')
 354                         pos++;
 355                 if (os_strncmp(pos, "PRIVACY", 7) == 0)
 356                         res->caps |= IEEE80211_CAP_PRIVACY;
 357         }
 358
 359         drv->num_scanres++;
 360 }
 361
 362
 363 static void wpa_driver_test_assocresp(struct wpa_driver_test_data *drv,
 364                                       struct sockaddr_un *from,
 365                                       socklen_t fromlen,
 366                                       const char *data)
 367 {
 368         /* ASSOCRESP BSSID <res> */
 369         if (hwaddr_aton(data, drv->bssid)) {
 370                 wpa_printf(MSG_DEBUG, "test_driver: invalid BSSID in "
 371                            "assocresp");
 372         }
 373         if (drv->use_associnfo) {
 374                 union wpa_event_data event;
 375                 os_memset(&event, 0, sizeof(event));
 376                 event.assoc_info.req_ies = drv->assoc_wpa_ie;
 377                 event.assoc_info.req_ies_len = drv->assoc_wpa_ie_len;
 378                 wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &event);
 379         }
 380         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
 381 }
 382
 383
 384 static void wpa_driver_test_disassoc(struct wpa_driver_test_data *drv,
 385                                      struct sockaddr_un *from,
 386                                      socklen_t fromlen)
 387 {
 388         wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
 389 }
 390
 391
 392 static void wpa_driver_test_eapol(struct wpa_driver_test_data *drv,
 393                                   struct sockaddr_un *from,
 394                                   socklen_t fromlen,
 395                                   const u8 *data, size_t data_len)
 396 {
 397         const u8 *src = drv->bssid;
 398
 399         if (data_len > 14) {
 400                 /* Skip Ethernet header */
 401                 src = data + ETH_ALEN;
 402                 data += 14;
 403                 data_len -= 14;
 404         }
 405         wpa_supplicant_rx_eapol(drv->ctx, src, data, data_len);
 406 }
 407
 408
 409 static void wpa_driver_test_mlme(struct wpa_driver_test_data *drv,
 410                                  struct sockaddr_un *from,
 411                                  socklen_t fromlen,
 412                                  const u8 *data, size_t data_len)
 413 {
 414         struct ieee80211_rx_status rx_status;
 415         os_memset(&rx_status, 0, sizeof(rx_status));
 416         ieee80211_sta_rx(drv->ctx, data, data_len, &rx_status);
 417 }
 418
 419
 420 static void wpa_driver_test_receive_unix(int sock, void *eloop_ctx,
 421                                          void *sock_ctx)
 422 {
 423         struct wpa_driver_test_data *drv = eloop_ctx;
 424         char *buf;
 425         int res;
 426         struct sockaddr_un from;
 427         socklen_t fromlen = sizeof(from);
 428         const size_t buflen = 2000;
 429
 430         buf = os_malloc(buflen);
 431         if (buf == NULL)
 432                 return;
 433         res = recvfrom(sock, buf, buflen - 1, 0,
 434                        (struct sockaddr *) &from, &fromlen);
 435         if (res < 0) {
 436                 perror("recvfrom(test_socket)");
 437                 os_free(buf);
 438                 return;
 439         }
 440         buf[res] = '\0';
 441
 442         wpa_printf(MSG_DEBUG, "test_driver: received %u bytes", res);
 443
 444         if (os_strncmp(buf, "SCANRESP ", 9) == 0) {
 445                 wpa_driver_test_scanresp(drv, &from, fromlen, buf + 9);
 446         } else if (os_strncmp(buf, "ASSOCRESP ", 10) == 0) {
 447                 wpa_driver_test_assocresp(drv, &from, fromlen, buf + 10);
 448         } else if (os_strcmp(buf, "DISASSOC") == 0) {
 449                 wpa_driver_test_disassoc(drv, &from, fromlen);
 450         } else if (os_strcmp(buf, "DEAUTH") == 0) {
 451                 wpa_driver_test_disassoc(drv, &from, fromlen);
 452         } else if (os_strncmp(buf, "EAPOL ", 6) == 0) {
 453                 wpa_driver_test_eapol(drv, &from, fromlen,
 454                                       (const u8 *) buf + 6, res - 6);
 455         } else if (os_strncmp(buf, "MLME ", 5) == 0) {
 456                 wpa_driver_test_mlme(drv, &from, fromlen,
 457                                      (const u8 *) buf + 5, res - 5);
 458         } else {
 459                 wpa_hexdump_ascii(MSG_DEBUG, "Unknown test_socket command",
 460                                   (u8 *) buf, res);
 461         }
 462         os_free(buf);
 463 }
 464
 465
 466 static void * wpa_driver_test_init(void *ctx, const char *ifname)
 467 {
 468         struct wpa_driver_test_data *drv;
 469
 470         drv = os_zalloc(sizeof(*drv));
 471         if (drv == NULL)
 472                 return NULL;
 473         drv->ctx = ctx;
 474         drv->test_socket = -1;
 475
 476         /* Set dummy BSSID and SSID for testing. */
 477         drv->bssid[0] = 0x02;
 478         drv->bssid[1] = 0x00;
 479         drv->bssid[2] = 0x00;
 480         drv->bssid[3] = 0x00;
 481         drv->bssid[4] = 0x00;
 482         drv->bssid[5] = 0x01;
 483         os_memcpy(drv->ssid, "test", 5);
 484         drv->ssid_len = 4;
 485
 486         /* Generate a MAC address to help testing with multiple STAs */
 487         drv->own_addr[0] = 0x02; /* locally administered */
 488         sha1_prf((const u8 *) ifname, os_strlen(ifname),
 489                  "wpa_supplicant test mac addr generation",
 490                  NULL, 0, drv->own_addr + 1, ETH_ALEN - 1);
 491
 492         return drv;
 493 }
 494
 495
 496 static void wpa_driver_test_close_test_socket(struct wpa_driver_test_data *drv)
 497 {
 498         if (drv->test_socket >= 0) {
 499                 eloop_unregister_read_sock(drv->test_socket);
 500                 close(drv->test_socket);
 501                 drv->test_socket = -1;
 502         }
 503
 504         if (drv->own_socket_path) {
 505                 unlink(drv->own_socket_path);
 506                 os_free(drv->own_socket_path);
 507                 drv->own_socket_path = NULL;
 508         }
 509 }
 510
 511
 512 static void wpa_driver_test_deinit(void *priv)
 513 {
 514         struct wpa_driver_test_data *drv = priv;
 515         wpa_driver_test_close_test_socket(drv);
 516         eloop_cancel_timeout(wpa_driver_test_scan_timeout, drv, drv->ctx);
 517         os_free(drv->test_dir);
 518         os_free(drv);
 519 }
 520
 521
 522 static int wpa_driver_test_attach(struct wpa_driver_test_data *drv,
 523                                   const char *dir)
 524 {
 525         static unsigned int counter = 0;
 526         struct sockaddr_un addr;
 527         size_t len;
 528
 529         os_free(drv->own_socket_path);
 530         if (dir) {
 531                 len = os_strlen(dir) + 30;
 532                 drv->own_socket_path = os_malloc(len);
 533                 if (drv->own_socket_path == NULL)
 534                         return -1;
 535                 os_snprintf(drv->own_socket_path, len, "%s/STA-" MACSTR,
 536                             dir, MAC2STR(drv->own_addr));
 537         } else {
 538                 drv->own_socket_path = os_malloc(100);
 539                 if (drv->own_socket_path == NULL)
 540                         return -1;
 541                 os_snprintf(drv->own_socket_path, 100,
 542                             "/tmp/wpa_supplicant_test-%d-%d",
 543                             getpid(), counter++);
 544         }
 545
 546         drv->test_socket = socket(PF_UNIX, SOCK_DGRAM, 0);
 547         if (drv->test_socket < 0) {
 548                 perror("socket(PF_UNIX)");
 549                 os_free(drv->own_socket_path);
 550                 drv->own_socket_path = NULL;
 551                 return -1;
 552         }
 553
 554         os_memset(&addr, 0, sizeof(addr));
 555         addr.sun_family = AF_UNIX;
 556         os_strncpy(addr.sun_path, drv->own_socket_path, sizeof(addr.sun_path));
 557         if (bind(drv->test_socket, (struct sockaddr *) &addr,
 558                  sizeof(addr)) < 0) {
 559                 perror("bind(PF_UNIX)");
 560                 close(drv->test_socket);
 561                 unlink(drv->own_socket_path);
 562                 os_free(drv->own_socket_path);
 563                 drv->own_socket_path = NULL;
 564                 return -1;
 565         }
 566
 567         eloop_register_read_sock(drv->test_socket,
 568                                  wpa_driver_test_receive_unix, drv, NULL);
 569
 570         return 0;
 571 }
 572
 573
 574 static int wpa_driver_test_set_param(void *priv, const char *param)
 575 {
 576         struct wpa_driver_test_data *drv = priv;
 577         const char *pos, *pos2;
 578         size_t len;
 579
 580         wpa_printf(MSG_DEBUG, "%s: param='%s'", __func__, param);
 581         if (param == NULL)
 582                 return 0;
 583
 584         wpa_driver_test_close_test_socket(drv);
 585         pos = os_strstr(param, "test_socket=");
 586         if (pos) {
 587                 pos += 12;
 588                 pos2 = os_strchr(pos, ' ');
 589                 if (pos2)
 590                         len = pos2 - pos;
 591                 else
 592                         len = os_strlen(pos);
 593                 if (len > sizeof(drv->hostapd_addr.sun_path))
 594                         return -1;
 595                 os_memset(&drv->hostapd_addr, 0, sizeof(drv->hostapd_addr));
 596                 drv->hostapd_addr.sun_family = AF_UNIX;
 597                 os_memcpy(drv->hostapd_addr.sun_path, pos, len);
 598                 drv->hostapd_addr_set = 1;
 599         }
 600
 601         pos = os_strstr(param, "test_dir=");
 602         if (pos) {
 603                 char *end;
 604                 os_free(drv->test_dir);
 605                 drv->test_dir = os_strdup(pos + 9);
 606                 if (drv->test_dir == NULL)
 607                         return -1;
 608                 end = os_strchr(drv->test_dir, ' ');
 609                 if (end)
 610                         *end = '\0';
 611                 wpa_driver_test_attach(drv, drv->test_dir);
 612         } else
 613                 wpa_driver_test_attach(drv, NULL);
 614
 615         if (os_strstr(param, "use_associnfo=1")) {
 616                 wpa_printf(MSG_DEBUG, "test_driver: Use AssocInfo events");
 617                 drv->use_associnfo = 1;
 618         }
 619
 620 #ifdef CONFIG_CLIENT_MLME
 621         if (os_strstr(param, "use_mlme=1")) {
 622                 wpa_printf(MSG_DEBUG, "test_driver: Use internal MLME");
 623                 drv->use_mlme = 1;
 624         }
 625 #endif /* CONFIG_CLIENT_MLME */
 626
 627         return 0;
 628 }
 629
 630
 631 static const u8 * wpa_driver_test_get_mac_addr(void *priv)
 632 {
 633         struct wpa_driver_test_data *drv = priv;
 634         wpa_printf(MSG_DEBUG, "%s", __func__);
 635         return drv->own_addr;
 636 }
 637
 638
 639 static int wpa_driver_test_send_eapol(void *priv, const u8 *dest, u16 proto,
 640                                       const u8 *data, size_t data_len)
 641 {
 642         struct wpa_driver_test_data *drv = priv;
 643         struct msghdr msg;
 644         struct iovec io[3];
 645         struct l2_ethhdr eth;
 646         struct sockaddr_un addr;
 647
 648         wpa_hexdump(MSG_MSGDUMP, "test_send_eapol TX frame", data, data_len);
 649
 650         os_memset(&eth, 0, sizeof(eth));
 651         os_memcpy(eth.h_dest, dest, ETH_ALEN);
 652         os_memcpy(eth.h_source, drv->own_addr, ETH_ALEN);
 653         eth.h_proto = host_to_be16(proto);
 654
 655         io[0].iov_base = "EAPOL ";
 656         io[0].iov_len = 6;
 657         io[1].iov_base = (u8 *) &eth;
 658         io[1].iov_len = sizeof(eth);
 659         io[2].iov_base = (u8 *) data;
 660         io[2].iov_len = data_len;
 661
 662         os_memset(&msg, 0, sizeof(msg));
 663         msg.msg_iov = io;
 664         msg.msg_iovlen = 3;
 665         if (os_memcmp(dest, drv->bssid, ETH_ALEN) == 0 ||
 666             drv->test_dir == NULL) {
 667                 msg.msg_name = &drv->hostapd_addr;
 668                 msg.msg_namelen = sizeof(drv->hostapd_addr);
 669         } else {
 670                 struct stat st;
 671                 os_memset(&addr, 0, sizeof(addr));
 672                 addr.sun_family = AF_UNIX;
 673                 os_snprintf(addr.sun_path, sizeof(addr.sun_path),
 674                             "%s/STA-" MACSTR, drv->test_dir, MAC2STR(dest));
 675                 if (stat(addr.sun_path, &st) < 0) {
 676                         os_snprintf(addr.sun_path, sizeof(addr.sun_path),
 677                                     "%s/AP-" MACSTR,
 678                                     drv->test_dir, MAC2STR(dest));
 679                 }
 680                 msg.msg_name = &addr;
 681                 msg.msg_namelen = sizeof(addr);
 682         }
 683
 684         if (sendmsg(drv->test_socket, &msg, 0) < 0) {
 685                 perror("sendmsg(test_socket)");
 686                 return -1;
 687         }
 688
 689         return 0;
 690 }
 691
 692
 693 static int wpa_driver_test_get_capa(void *priv, struct wpa_driver_capa *capa)
 694 {
 695         struct wpa_driver_test_data *drv = priv;
 696         os_memset(capa, 0, sizeof(*capa));
 697         capa->key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
 698                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
 699                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
 700                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK |
 701                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_NONE;
 702         capa->enc = WPA_DRIVER_CAPA_ENC_WEP40 |
 703                 WPA_DRIVER_CAPA_ENC_WEP104 |
 704                 WPA_DRIVER_CAPA_ENC_TKIP |
 705                 WPA_DRIVER_CAPA_ENC_CCMP;
 706         capa->auth = WPA_DRIVER_AUTH_OPEN |
 707                 WPA_DRIVER_AUTH_SHARED |
 708                 WPA_DRIVER_AUTH_LEAP;
 709         if (drv->use_mlme)
 710                 capa->flags |= WPA_DRIVER_FLAGS_USER_SPACE_MLME;
 711
 712         return 0;
 713 }
 714
 715
 716 static int wpa_driver_test_mlme_setprotection(void *priv, const u8 *addr,
 717                                               int protect_type,
 718                                               int key_type)
 719 {
 720         wpa_printf(MSG_DEBUG, "%s: protect_type=%d key_type=%d",
 721                    __func__, protect_type, key_type);
 722
 723         if (addr) {
 724                 wpa_printf(MSG_DEBUG, "%s: addr=" MACSTR,
 725                            __func__, MAC2STR(addr));
 726         }
 727
 728         return 0;
 729 }
 730
 731
 732 #ifdef CONFIG_CLIENT_MLME
 733 static struct wpa_hw_modes *
 734 wpa_driver_test_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
 735 {
 736         struct wpa_hw_modes *modes;
 737
 738         *num_modes = 1;
 739         *flags = 0;
 740         modes = os_zalloc(*num_modes * sizeof(struct wpa_hw_modes));
 741         if (modes == NULL)
 742                 return NULL;
 743         modes[0].mode = WPA_MODE_IEEE80211G;
 744         modes[0].num_channels = 1;
 745         modes[0].num_rates = 1;
 746         modes[0].channels = os_zalloc(sizeof(struct wpa_channel_data));
 747         modes[0].rates = os_zalloc(sizeof(struct wpa_rate_data));
 748         if (modes[0].channels == NULL || modes[0].rates == NULL) {
 749                 ieee80211_sta_free_hw_features(modes, *num_modes);
 750                 return NULL;
 751         }
 752         modes[0].channels[0].chan = 1;
 753         modes[0].channels[0].freq = 2412;
 754         modes[0].channels[0].flag = WPA_CHAN_W_SCAN | WPA_CHAN_W_ACTIVE_SCAN;
 755         modes[0].rates[0].rate = 10;
 756         modes[0].rates[0].flags = WPA_RATE_BASIC | WPA_RATE_SUPPORTED |
 757                 WPA_RATE_CCK | WPA_RATE_MANDATORY;
 758
 759         return modes;
 760 }
 761
 762
 763 int wpa_driver_test_set_channel(void *priv, wpa_hw_mode phymode, int chan,
 764                                 int freq)
 765 {
 766         wpa_printf(MSG_DEBUG, "%s: phymode=%d chan=%d freq=%d",
 767                    __func__, phymode, chan, freq);
 768         return 0;
 769 }
 770
 771
 772 static int wpa_driver_test_send_mlme(void *priv, const u8 *data,
 773                                      size_t data_len)
 774 {
 775         struct wpa_driver_test_data *drv = priv;
 776         struct msghdr msg;
 777         struct iovec io[2];
 778         struct sockaddr_un addr;
 779         const u8 *dest;
 780         struct dirent *dent;
 781         DIR *dir;
 782
 783         wpa_hexdump(MSG_MSGDUMP, "test_send_mlme", data, data_len);
 784         if (data_len < 10)
 785                 return -1;
 786         dest = data + 4;
 787
 788         io[0].iov_base = "MLME ";
 789         io[0].iov_len = 5;
 790         io[1].iov_base = (u8 *) data;
 791         io[1].iov_len = data_len;
 792
 793         os_memset(&msg, 0, sizeof(msg));
 794         msg.msg_iov = io;
 795         msg.msg_iovlen = 2;
 796         if (os_memcmp(dest, drv->bssid, ETH_ALEN) == 0 ||
 797             drv->test_dir == NULL) {
 798                 msg.msg_name = &drv->hostapd_addr;
 799                 msg.msg_namelen = sizeof(drv->hostapd_addr);
 800         } else if (os_memcmp(dest, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) == 0)
 801         {
 802                 dir = opendir(drv->test_dir);
 803                 if (dir == NULL)
 804                         return -1;
 805                 while ((dent = readdir(dir))) {
 806 #ifdef _DIRENT_HAVE_D_TYPE
 807                         /* Skip the file if it is not a socket.
 808                          * Also accept DT_UNKNOWN (0) in case
 809                          * the C library or underlying file
 810                          * system does not support d_type. */
 811                         if (dent->d_type != DT_SOCK &&
 812                             dent->d_type != DT_UNKNOWN)
 813                                 continue;
 814 #endif /* _DIRENT_HAVE_D_TYPE */
 815                         if (os_strcmp(dent->d_name, ".") == 0 ||
 816                             os_strcmp(dent->d_name, "..") == 0)
 817                                 continue;
 818                         wpa_printf(MSG_DEBUG, "%s: Send broadcast MLME to %s",
 819                                    __func__, dent->d_name);
 820                         os_memset(&addr, 0, sizeof(addr));
 821                         addr.sun_family = AF_UNIX;
 822                         os_snprintf(addr.sun_path, sizeof(addr.sun_path),
 823                                     "%s/%s", drv->test_dir, dent->d_name);
 824
 825                         msg.msg_name = &addr;
 826                         msg.msg_namelen = sizeof(addr);
 827
 828                         if (sendmsg(drv->test_socket, &msg, 0) < 0)
 829                                 perror("sendmsg(test_socket)");
 830                 }
 831                 closedir(dir);
 832                 return 0;
 833         } else {
 834                 struct stat st;
 835                 os_memset(&addr, 0, sizeof(addr));
 836                 addr.sun_family = AF_UNIX;
 837                 os_snprintf(addr.sun_path, sizeof(addr.sun_path),
 838                             "%s/AP-" MACSTR, drv->test_dir, MAC2STR(dest));
 839                 if (stat(addr.sun_path, &st) < 0) {
 840                         os_snprintf(addr.sun_path, sizeof(addr.sun_path),
 841                                     "%s/STA-" MACSTR,
 842                                     drv->test_dir, MAC2STR(dest));
 843                 }
 844                 msg.msg_name = &addr;
 845                 msg.msg_namelen = sizeof(addr);
 846         }
 847
 848         if (sendmsg(drv->test_socket, &msg, 0) < 0) {
 849                 perror("sendmsg(test_socket)");
 850                 return -1;
 851         }
 852
 853         return 0;
 854 }
 855
 856
 857 static int wpa_driver_test_mlme_add_sta(void *priv, const u8 *addr,
 858                                         const u8 *supp_rates,
 859                                         size_t supp_rates_len)
 860 {
 861         wpa_printf(MSG_DEBUG, "%s: addr=" MACSTR, __func__, MAC2STR(addr));
 862         return 0;
 863 }
 864
 865
 866 static int wpa_driver_test_mlme_remove_sta(void *priv, const u8 *addr)
 867 {
 868         wpa_printf(MSG_DEBUG, "%s: addr=" MACSTR, __func__, MAC2STR(addr));
 869         return 0;
 870 }
 871
 872
 873 int wpa_driver_test_set_ssid(void *priv, const u8 *ssid, size_t ssid_len)
 874 {
 875         wpa_printf(MSG_DEBUG, "%s", __func__);
 876         return 0;
 877 }
 878
 879
 880 int wpa_driver_test_set_bssid(void *priv, const u8 *bssid)
 881 {
 882         wpa_printf(MSG_DEBUG, "%s: bssid=" MACSTR, __func__, MAC2STR(bssid));
 883         return 0;
 884 }
 885 #endif /* CONFIG_CLIENT_MLME */
 886
 887 int wpa_driver_test_driver_cmd( void *priv, char *cmd, char *buf, size_t buf_len )
 888 {
 889     struct wpa_driver_test_data *drv = (struct wpa_driver_test_data *)priv;
 890     int ret = -1;
 891
 892     wpa_printf(MSG_DEBUG, "%s %s", __func__, cmd);
 893     if( os_strncasecmp(cmd, "start", 5) == 0 ) {
 894         wpa_printf(MSG_DEBUG,"Start command");
 895         ret = 0;
 896     }
 897     else if( os_strncasecmp(cmd, "stop", 4) == 0 ) {
 898         wpa_printf(MSG_DEBUG,"Stop command");
 899         ret = 0;
 900     }
 901     else if( os_strncasecmp(cmd, "macaddr", 7) == 0 ) {
 902         u8 *macaddr = (u8 *)wpa_driver_test_get_mac_addr(priv);
 903         wpa_printf(MSG_DEBUG,"Macaddr command");
 904         wpa_printf(MSG_DEBUG, "   Macaddr = " MACSTR, MAC2STR(macaddr));
 905         ret = os_snprintf(buf, buf_len, "Macaddr = " MACSTR "\n", MAC2STR(macaddr));
 906     }
 907     else if( os_strncasecmp(cmd, "rssi", 4) == 0 ) {
 908         wpa_printf(MSG_DEBUG,"RSSI command");
 909         ret = os_snprintf(buf, buf_len, MACSTR " Rssi %d\n", MAC2STR(drv->bssid), -10);
 910     } else if (os_strncasecmp(cmd, "linkspeed", 9) == 0) {
 911         wpa_printf(MSG_DEBUG, "LinkSpeed command");
 912         ret = os_snprintf(buf, buf_len, "LinkSpeed %u\n", 11);
 913     }
 914     return ret;
 915 }
 916
 917 const struct wpa_driver_ops wpa_driver_test_ops = {
 918         "test",
 919         "wpa_supplicant test driver",
 920         wpa_driver_test_get_bssid,
 921         wpa_driver_test_get_ssid,
 922         wpa_driver_test_set_wpa,
 923         wpa_driver_test_set_key,
 924         wpa_driver_test_init,
 925         wpa_driver_test_deinit,
 926         wpa_driver_test_set_param,
 927         NULL /* set_countermeasures */,
 928         NULL /* set_drop_unencrypted */,
 929         wpa_driver_test_scan,
 930         wpa_driver_test_get_scan_results,
 931         wpa_driver_test_deauthenticate,
 932         wpa_driver_test_disassociate,
 933         wpa_driver_test_associate,
 934         NULL /* set_auth_alg */,
 935         NULL /* add_pmkid */,
 936         NULL /* remove_pmkid */,
 937         NULL /* flush_pmkid */,
 938         wpa_driver_test_get_capa,
 939         NULL /* poll */,
 940         NULL /* get_ifname */,
 941         wpa_driver_test_get_mac_addr,
 942         wpa_driver_test_send_eapol,
 943         NULL /* set_operstate */,
 944         wpa_driver_test_mlme_setprotection,
 945 #ifdef CONFIG_CLIENT_MLME
 946         wpa_driver_test_get_hw_feature_data,
 947         wpa_driver_test_set_channel,
 948         wpa_driver_test_set_ssid,
 949         wpa_driver_test_set_bssid,
 950         wpa_driver_test_send_mlme,
 951         wpa_driver_test_mlme_add_sta,
 952         wpa_driver_test_mlme_remove_sta,
 953 #else /* CONFIG_CLIENT_MLME */
 954         NULL /* get_hw_feature_data */,
 955         NULL /* set_channel */,
 956         NULL /* set_ssid */,
 957         NULL /* set_bssid */,
 958         NULL /* send_mlme */,
 959         NULL /* mlme_add_sta */,
 960         NULL /* mlme_remove_sta */,
 961 #endif /* CONFIG_CLIENT_MLME */
 962     wpa_driver_test_driver_cmd
 963 };