hostapd/driver_test.c

   1 /*
   2  * hostapd / Driver interface for development testing
   3  * Copyright (c) 2004-2008, 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
  19 #include "hostapd.h"
  20 #include "driver.h"
  21 #include "sha1.h"
  22 #include "eloop.h"
  23 #include "ieee802_1x.h"
  24 #include "sta_info.h"
  25 #include "wpa.h"
  26 #include "accounting.h"
  27 #include "radius/radius.h"
  28 #include "l2_packet/l2_packet.h"
  29 #include "ieee802_11.h"
  30 #include "hw_features.h"
  31
  32
  33 struct test_client_socket {
  34         struct test_client_socket *next;
  35         u8 addr[ETH_ALEN];
  36         struct sockaddr_un un;
  37         socklen_t unlen;
  38         struct test_driver_bss *bss;
  39 };
  40
  41 struct test_driver_bss {
  42         struct test_driver_bss *next;
  43         char ifname[IFNAMSIZ + 1];
  44         u8 bssid[ETH_ALEN];
  45         u8 *ie;
  46         size_t ielen;
  47         u8 ssid[32];
  48         size_t ssid_len;
  49         int privacy;
  50 };
  51
  52 struct test_driver_data {
  53         struct hostapd_data *hapd;
  54         struct test_client_socket *cli;
  55         int test_socket;
  56         struct test_driver_bss *bss;
  57         char *socket_dir;
  58         char *own_socket_path;
  59 };
  60
  61
  62 static void test_driver_free_bss(struct test_driver_bss *bss)
  63 {
  64         free(bss->ie);
  65         free(bss);
  66 }
  67
  68
  69 static void test_driver_free_priv(struct test_driver_data *drv)
  70 {
  71         struct test_driver_bss *bss, *prev;
  72
  73         if (drv == NULL)
  74                 return;
  75
  76         bss = drv->bss;
  77         while (bss) {
  78                 prev = bss;
  79                 bss = bss->next;
  80                 test_driver_free_bss(prev);
  81         }
  82         free(drv->own_socket_path);
  83         free(drv->socket_dir);
  84         free(drv);
  85 }
  86
  87
  88 static struct test_client_socket *
  89 test_driver_get_cli(struct test_driver_data *drv, struct sockaddr_un *from,
  90                     socklen_t fromlen)
  91 {
  92         struct test_client_socket *cli = drv->cli;
  93
  94         while (cli) {
  95                 if (cli->unlen == fromlen &&
  96                     strncmp(cli->un.sun_path, from->sun_path,
  97                             fromlen - sizeof(cli->un.sun_family)) == 0)
  98                         return cli;
  99                 cli = cli->next;
 100         }
 101
 102         return NULL;
 103 }
 104
 105
 106 static int test_driver_send_eapol(void *priv, const u8 *addr, const u8 *data,
 107                                   size_t data_len, int encrypt,
 108                                   const u8 *own_addr)
 109 {
 110         struct test_driver_data *drv = priv;
 111         struct test_client_socket *cli;
 112         struct msghdr msg;
 113         struct iovec io[3];
 114         struct l2_ethhdr eth;
 115
 116         if (drv->test_socket < 0)
 117                 return -1;
 118
 119         cli = drv->cli;
 120         while (cli) {
 121                 if (memcmp(cli->addr, addr, ETH_ALEN) == 0)
 122                         break;
 123                 cli = cli->next;
 124         }
 125
 126         if (!cli) {
 127                 wpa_printf(MSG_DEBUG, "%s: no destination client entry",
 128                            __func__);
 129                 return -1;
 130         }
 131
 132         memcpy(eth.h_dest, addr, ETH_ALEN);
 133         memcpy(eth.h_source, own_addr, ETH_ALEN);
 134         eth.h_proto = htons(ETH_P_EAPOL);
 135
 136         io[0].iov_base = "EAPOL ";
 137         io[0].iov_len = 6;
 138         io[1].iov_base = &eth;
 139         io[1].iov_len = sizeof(eth);
 140         io[2].iov_base = (u8 *) data;
 141         io[2].iov_len = data_len;
 142
 143         memset(&msg, 0, sizeof(msg));
 144         msg.msg_iov = io;
 145         msg.msg_iovlen = 3;
 146         msg.msg_name = &cli->un;
 147         msg.msg_namelen = cli->unlen;
 148         return sendmsg(drv->test_socket, &msg, 0);
 149 }
 150
 151
 152 static int test_driver_send_ether(void *priv, const u8 *dst, const u8 *src,
 153                                   u16 proto, const u8 *data, size_t data_len)
 154 {
 155         struct test_driver_data *drv = priv;
 156         struct msghdr msg;
 157         struct iovec io[3];
 158         struct l2_ethhdr eth;
 159         char desttxt[30];
 160         struct sockaddr_un addr;
 161         struct dirent *dent;
 162         DIR *dir;
 163         int ret = 0, broadcast = 0, count = 0;
 164
 165         if (drv->test_socket < 0 || drv->socket_dir == NULL) {
 166                 wpa_printf(MSG_DEBUG, "%s: invalid parameters (sock=%d "
 167                            "socket_dir=%p)",
 168                            __func__, drv->test_socket, drv->socket_dir);
 169                 return -1;
 170         }
 171
 172         broadcast = memcmp(dst, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) == 0;
 173         snprintf(desttxt, sizeof(desttxt), MACSTR, MAC2STR(dst));
 174
 175         memcpy(eth.h_dest, dst, ETH_ALEN);
 176         memcpy(eth.h_source, src, ETH_ALEN);
 177         eth.h_proto = htons(proto);
 178
 179         io[0].iov_base = "ETHER ";
 180         io[0].iov_len = 6;
 181         io[1].iov_base = &eth;
 182         io[1].iov_len = sizeof(eth);
 183         io[2].iov_base = (u8 *) data;
 184         io[2].iov_len = data_len;
 185
 186         memset(&msg, 0, sizeof(msg));
 187         msg.msg_iov = io;
 188         msg.msg_iovlen = 3;
 189
 190         dir = opendir(drv->socket_dir);
 191         if (dir == NULL) {
 192                 perror("test_driver: opendir");
 193                 return -1;
 194         }
 195         while ((dent = readdir(dir))) {
 196 #ifdef _DIRENT_HAVE_D_TYPE
 197                 /* Skip the file if it is not a socket. Also accept
 198                  * DT_UNKNOWN (0) in case the C library or underlying file
 199                  * system does not support d_type. */
 200                 if (dent->d_type != DT_SOCK && dent->d_type != DT_UNKNOWN)
 201                         continue;
 202 #endif /* _DIRENT_HAVE_D_TYPE */
 203                 if (strcmp(dent->d_name, ".") == 0 ||
 204                     strcmp(dent->d_name, "..") == 0)
 205                         continue;
 206
 207                 memset(&addr, 0, sizeof(addr));
 208                 addr.sun_family = AF_UNIX;
 209                 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s",
 210                          drv->socket_dir, dent->d_name);
 211
 212                 if (strcmp(addr.sun_path, drv->own_socket_path) == 0)
 213                         continue;
 214                 if (!broadcast && strstr(dent->d_name, desttxt) == NULL)
 215                         continue;
 216
 217                 wpa_printf(MSG_DEBUG, "%s: Send ether frame to %s",
 218                            __func__, dent->d_name);
 219
 220                 msg.msg_name = &addr;
 221                 msg.msg_namelen = sizeof(addr);
 222                 ret = sendmsg(drv->test_socket, &msg, 0);
 223                 if (ret < 0)
 224                         perror("driver_test: sendmsg");
 225                 count++;
 226         }
 227         closedir(dir);
 228
 229         if (!broadcast && count == 0) {
 230                 wpa_printf(MSG_DEBUG, "%s: Destination " MACSTR " not found",
 231                            __func__, MAC2STR(dst));
 232                 return -1;
 233         }
 234
 235         return ret;
 236 }
 237
 238
 239 static int test_driver_send_mgmt_frame(void *priv, const void *buf,
 240                                        size_t len, int flags)
 241 {
 242         struct test_driver_data *drv = priv;
 243         struct msghdr msg;
 244         struct iovec io[2];
 245         const u8 *dest;
 246         int ret = 0, broadcast = 0;
 247         char desttxt[30];
 248         struct sockaddr_un addr;
 249         struct dirent *dent;
 250         DIR *dir;
 251         struct ieee80211_hdr *hdr;
 252         u16 fc;
 253
 254         if (drv->test_socket < 0 || len < 10 || drv->socket_dir == NULL) {
 255                 wpa_printf(MSG_DEBUG, "%s: invalid parameters (sock=%d len=%lu"
 256                            " socket_dir=%p)",
 257                            __func__, drv->test_socket, (unsigned long) len,
 258                            drv->socket_dir);
 259                 return -1;
 260         }
 261
 262         dest = buf;
 263         dest += 4;
 264         broadcast = memcmp(dest, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) == 0;
 265         snprintf(desttxt, sizeof(desttxt), MACSTR, MAC2STR(dest));
 266
 267         io[0].iov_base = "MLME ";
 268         io[0].iov_len = 5;
 269         io[1].iov_base = (void *) buf;
 270         io[1].iov_len = len;
 271
 272         memset(&msg, 0, sizeof(msg));
 273         msg.msg_iov = io;
 274         msg.msg_iovlen = 2;
 275
 276         dir = opendir(drv->socket_dir);
 277         if (dir == NULL) {
 278                 perror("test_driver: opendir");
 279                 return -1;
 280         }
 281         while ((dent = readdir(dir))) {
 282 #ifdef _DIRENT_HAVE_D_TYPE
 283                 /* Skip the file if it is not a socket. Also accept
 284                  * DT_UNKNOWN (0) in case the C library or underlying file
 285                  * system does not support d_type. */
 286                 if (dent->d_type != DT_SOCK && dent->d_type != DT_UNKNOWN)
 287                         continue;
 288 #endif /* _DIRENT_HAVE_D_TYPE */
 289                 if (strcmp(dent->d_name, ".") == 0 ||
 290                     strcmp(dent->d_name, "..") == 0)
 291                         continue;
 292
 293                 memset(&addr, 0, sizeof(addr));
 294                 addr.sun_family = AF_UNIX;
 295                 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s",
 296                          drv->socket_dir, dent->d_name);
 297
 298                 if (strcmp(addr.sun_path, drv->own_socket_path) == 0)
 299                         continue;
 300                 if (!broadcast && strstr(dent->d_name, desttxt) == NULL)
 301                         continue;
 302
 303                 wpa_printf(MSG_DEBUG, "%s: Send management frame to %s",
 304                            __func__, dent->d_name);
 305
 306                 msg.msg_name = &addr;
 307                 msg.msg_namelen = sizeof(addr);
 308                 ret = sendmsg(drv->test_socket, &msg, 0);
 309                 if (ret < 0)
 310                         perror("driver_test: sendmsg");
 311         }
 312         closedir(dir);
 313
 314         hdr = (struct ieee80211_hdr *) buf;
 315         fc = le_to_host16(hdr->frame_control);
 316         ieee802_11_mgmt_cb(drv->hapd, (u8 *) buf, len, WLAN_FC_GET_STYPE(fc),
 317                            ret >= 0);
 318
 319         return ret;
 320 }
 321
 322
 323 static void test_driver_scan(struct test_driver_data *drv,
 324                              struct sockaddr_un *from, socklen_t fromlen)
 325 {
 326         char buf[512], *pos, *end;
 327         int ret;
 328         struct test_driver_bss *bss;
 329
 330         wpa_printf(MSG_DEBUG, "test_driver: SCAN");
 331
 332         for (bss = drv->bss; bss; bss = bss->next) {
 333                 pos = buf;
 334                 end = buf + sizeof(buf);
 335
 336                 /* reply: SCANRESP BSSID SSID IEs */
 337                 ret = snprintf(pos, end - pos, "SCANRESP " MACSTR " ",
 338                                MAC2STR(bss->bssid));
 339                 if (ret < 0 || ret >= end - pos)
 340                         return;
 341                 pos += ret;
 342                 pos += wpa_snprintf_hex(pos, end - pos,
 343                                         bss->ssid, bss->ssid_len);
 344                 ret = snprintf(pos, end - pos, " ");
 345                 if (ret < 0 || ret >= end - pos)
 346                         return;
 347                 pos += ret;
 348                 pos += wpa_snprintf_hex(pos, end - pos, bss->ie, bss->ielen);
 349
 350                 if (bss->privacy) {
 351                         ret = snprintf(pos, end - pos, " PRIVACY");
 352                         if (ret < 0 || ret >= end - pos)
 353                                 return;
 354                         pos += ret;
 355                 }
 356
 357                 sendto(drv->test_socket, buf, pos - buf, 0,
 358                        (struct sockaddr *) from, fromlen);
 359         }
 360 }
 361
 362
 363 static struct hostapd_data * test_driver_get_hapd(struct test_driver_data *drv,
 364                                                   struct test_driver_bss *bss)
 365 {
 366         struct hostapd_iface *iface = drv->hapd->iface;
 367         struct hostapd_data *hapd = NULL;
 368         size_t i;
 369
 370         if (bss == NULL) {
 371                 wpa_printf(MSG_DEBUG, "%s: bss == NULL", __func__);
 372                 return NULL;
 373         }
 374
 375         for (i = 0; i < iface->num_bss; i++) {
 376                 hapd = iface->bss[i];
 377                 if (memcmp(hapd->own_addr, bss->bssid, ETH_ALEN) == 0)
 378                         break;
 379         }
 380         if (i == iface->num_bss) {
 381                 wpa_printf(MSG_DEBUG, "%s: no matching interface entry found "
 382                            "for BSSID " MACSTR, __func__, MAC2STR(bss->bssid));
 383                 return NULL;
 384         }
 385
 386         return hapd;
 387 }
 388
 389
 390 static int test_driver_new_sta(struct test_driver_data *drv,
 391                                struct test_driver_bss *bss, const u8 *addr,
 392                                const u8 *ie, size_t ielen)
 393 {
 394         struct hostapd_data *hapd;
 395         struct sta_info *sta;
 396         int new_assoc, res;
 397
 398         hapd = test_driver_get_hapd(drv, bss);
 399         if (hapd == NULL)
 400                 return -1;
 401
 402         hostapd_logger(hapd, addr, HOSTAPD_MODULE_IEEE80211,
 403                 HOSTAPD_LEVEL_INFO, "associated");
 404
 405         sta = ap_get_sta(hapd, addr);
 406         if (sta) {
 407                 accounting_sta_stop(hapd, sta);
 408         } else {
 409                 sta = ap_sta_add(hapd, addr);
 410                 if (sta == NULL)
 411                         return -1;
 412         }
 413         accounting_sta_get_id(hapd, sta);
 414
 415         if (hapd->conf->wpa) {
 416                 if (ie == NULL || ielen == 0) {
 417                         printf("test_driver: no IE from STA\n");
 418                         return -1;
 419                 }
 420                 if (sta->wpa_sm == NULL)
 421                         sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
 422                                                         sta->addr);
 423                 if (sta->wpa_sm == NULL) {
 424                         printf("test_driver: Failed to initialize WPA state "
 425                                "machine\n");
 426                         return -1;
 427                 }
 428                 res = wpa_validate_wpa_ie(hapd->wpa_auth, sta->wpa_sm,
 429                                           ie, ielen, NULL, 0);
 430                 if (res != WPA_IE_OK) {
 431                         printf("WPA/RSN information element rejected? "
 432                                "(res %u)\n", res);
 433                         return -1;
 434                 }
 435         }
 436
 437         new_assoc = (sta->flags & WLAN_STA_ASSOC) == 0;
 438         sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC;
 439         wpa_auth_sm_event(sta->wpa_sm, WPA_ASSOC);
 440
 441         hostapd_new_assoc_sta(hapd, sta, !new_assoc);
 442
 443         ieee802_1x_notify_port_enabled(sta->eapol_sm, 1);
 444
 445         return 0;
 446 }
 447
 448
 449 static void test_driver_assoc(struct test_driver_data *drv,
 450                               struct sockaddr_un *from, socklen_t fromlen,
 451                               char *data)
 452 {
 453         struct test_client_socket *cli;
 454         u8 ie[256], ssid[32];
 455         size_t ielen, ssid_len = 0;
 456         char *pos, *pos2, cmd[50];
 457         struct test_driver_bss *bss;
 458
 459         /* data: STA-addr SSID(hex) IEs(hex) */
 460
 461         cli = os_zalloc(sizeof(*cli));
 462         if (cli == NULL)
 463                 return;
 464
 465         if (hwaddr_aton(data, cli->addr)) {
 466                 printf("test_socket: Invalid MAC address '%s' in ASSOC\n",
 467                        data);
 468                 free(cli);
 469                 return;
 470         }
 471         pos = data + 17;
 472         while (*pos == ' ')
 473                 pos++;
 474         pos2 = strchr(pos, ' ');
 475         ielen = 0;
 476         if (pos2) {
 477                 ssid_len = (pos2 - pos) / 2;
 478                 if (hexstr2bin(pos, ssid, ssid_len) < 0) {
 479                         wpa_printf(MSG_DEBUG, "%s: Invalid SSID", __func__);
 480                         free(cli);
 481                         return;
 482                 }
 483                 wpa_hexdump_ascii(MSG_DEBUG, "test_driver_assoc: SSID",
 484                                   ssid, ssid_len);
 485
 486                 pos = pos2 + 1;
 487                 ielen = strlen(pos) / 2;
 488                 if (ielen > sizeof(ie))
 489                         ielen = sizeof(ie);
 490                 if (hexstr2bin(pos, ie, ielen) < 0)
 491                         ielen = 0;
 492         }
 493
 494         for (bss = drv->bss; bss; bss = bss->next) {
 495                 if (bss->ssid_len == ssid_len &&
 496                     memcmp(bss->ssid, ssid, ssid_len) == 0)
 497                         break;
 498         }
 499         if (bss == NULL) {
 500                 wpa_printf(MSG_DEBUG, "%s: No matching SSID found from "
 501                            "configured BSSes", __func__);
 502                 free(cli);
 503                 return;
 504         }
 505
 506         cli->bss = bss;
 507         memcpy(&cli->un, from, sizeof(cli->un));
 508         cli->unlen = fromlen;
 509         cli->next = drv->cli;
 510         drv->cli = cli;
 511         wpa_hexdump_ascii(MSG_DEBUG, "test_socket: ASSOC sun_path",
 512                           (const u8 *) cli->un.sun_path,
 513                           cli->unlen - sizeof(cli->un.sun_family));
 514
 515         snprintf(cmd, sizeof(cmd), "ASSOCRESP " MACSTR " 0",
 516                  MAC2STR(bss->bssid));
 517         sendto(drv->test_socket, cmd, strlen(cmd), 0,
 518                (struct sockaddr *) from, fromlen);
 519
 520         if (test_driver_new_sta(drv, bss, cli->addr, ie, ielen) < 0) {
 521                 wpa_printf(MSG_DEBUG, "test_driver: failed to add new STA");
 522         }
 523 }
 524
 525
 526 static void test_driver_disassoc(struct test_driver_data *drv,
 527                                  struct sockaddr_un *from, socklen_t fromlen)
 528 {
 529         struct test_client_socket *cli;
 530         struct sta_info *sta;
 531
 532         cli = test_driver_get_cli(drv, from, fromlen);
 533         if (!cli)
 534                 return;
 535
 536         hostapd_logger(drv->hapd, cli->addr, HOSTAPD_MODULE_IEEE80211,
 537                        HOSTAPD_LEVEL_INFO, "disassociated");
 538
 539         sta = ap_get_sta(drv->hapd, cli->addr);
 540         if (sta != NULL) {
 541                 sta->flags &= ~WLAN_STA_ASSOC;
 542                 wpa_auth_sm_event(sta->wpa_sm, WPA_DISASSOC);
 543                 sta->acct_terminate_cause =
 544                         RADIUS_ACCT_TERMINATE_CAUSE_USER_REQUEST;
 545                 ieee802_1x_notify_port_enabled(sta->eapol_sm, 0);
 546                 ap_free_sta(drv->hapd, sta);
 547         }
 548 }
 549
 550
 551 static void test_driver_eapol(struct test_driver_data *drv,
 552                               struct sockaddr_un *from, socklen_t fromlen,
 553                               u8 *data, size_t datalen)
 554 {
 555         struct test_client_socket *cli;
 556         if (datalen > 14) {
 557                 u8 *proto = data + 2 * ETH_ALEN;
 558                 /* Skip Ethernet header */
 559                 wpa_printf(MSG_DEBUG, "test_driver: dst=" MACSTR " src="
 560                            MACSTR " proto=%04x",
 561                            MAC2STR(data), MAC2STR(data + ETH_ALEN),
 562                            WPA_GET_BE16(proto));
 563                 data += 14;
 564                 datalen -= 14;
 565         }
 566         cli = test_driver_get_cli(drv, from, fromlen);
 567         if (cli) {
 568                 struct hostapd_data *hapd;
 569                 hapd = test_driver_get_hapd(drv, cli->bss);
 570                 if (hapd == NULL)
 571                         return;
 572                 ieee802_1x_receive(hapd, cli->addr, data, datalen);
 573         } else {
 574                 wpa_printf(MSG_DEBUG, "test_socket: EAPOL from unknown "
 575                            "client");
 576         }
 577 }
 578
 579
 580 static void test_driver_ether(struct test_driver_data *drv,
 581                               struct sockaddr_un *from, socklen_t fromlen,
 582                               u8 *data, size_t datalen)
 583 {
 584         struct l2_ethhdr *eth;
 585
 586         if (datalen < sizeof(*eth))
 587                 return;
 588
 589         eth = (struct l2_ethhdr *) data;
 590         wpa_printf(MSG_DEBUG, "test_driver: RX ETHER dst=" MACSTR " src="
 591                    MACSTR " proto=%04x",
 592                    MAC2STR(eth->h_dest), MAC2STR(eth->h_source),
 593                    be_to_host16(eth->h_proto));
 594
 595 #ifdef CONFIG_IEEE80211R
 596         if (be_to_host16(eth->h_proto) == ETH_P_RRB) {
 597                 wpa_ft_rrb_rx(drv->hapd->wpa_auth, eth->h_source,
 598                               data + sizeof(*eth), datalen - sizeof(*eth));
 599         }
 600 #endif /* CONFIG_IEEE80211R */
 601 }
 602
 603
 604 static void test_driver_mlme(struct test_driver_data *drv,
 605                              struct sockaddr_un *from, socklen_t fromlen,
 606                              u8 *data, size_t datalen)
 607 {
 608         struct ieee80211_hdr *hdr;
 609         u16 fc;
 610
 611         hdr = (struct ieee80211_hdr *) data;
 612
 613         if (test_driver_get_cli(drv, from, fromlen) == NULL && datalen >= 16) {
 614                 struct test_client_socket *cli;
 615                 cli = os_zalloc(sizeof(*cli));
 616                 if (cli == NULL)
 617                         return;
 618                 wpa_printf(MSG_DEBUG, "Adding client entry for " MACSTR,
 619                            MAC2STR(hdr->addr2));
 620                 memcpy(cli->addr, hdr->addr2, ETH_ALEN);
 621                 memcpy(&cli->un, from, sizeof(cli->un));
 622                 cli->unlen = fromlen;
 623                 cli->next = drv->cli;
 624                 drv->cli = cli;
 625         }
 626
 627         wpa_hexdump(MSG_MSGDUMP, "test_driver_mlme: received frame",
 628                     data, datalen);
 629         fc = le_to_host16(hdr->frame_control);
 630         if (WLAN_FC_GET_TYPE(fc) != WLAN_FC_TYPE_MGMT) {
 631                 wpa_printf(MSG_ERROR, "%s: received non-mgmt frame",
 632                            __func__);
 633                 return;
 634         }
 635         ieee802_11_mgmt(drv->hapd, data, datalen, WLAN_FC_GET_STYPE(fc), NULL);
 636 }
 637
 638
 639 static void test_driver_receive_unix(int sock, void *eloop_ctx, void *sock_ctx)
 640 {
 641         struct test_driver_data *drv = eloop_ctx;
 642         char buf[2000];
 643         int res;
 644         struct sockaddr_un from;
 645         socklen_t fromlen = sizeof(from);
 646
 647         res = recvfrom(sock, buf, sizeof(buf) - 1, 0,
 648                        (struct sockaddr *) &from, &fromlen);
 649         if (res < 0) {
 650                 perror("recvfrom(test_socket)");
 651                 return;
 652         }
 653         buf[res] = '\0';
 654
 655         wpa_printf(MSG_DEBUG, "test_driver: received %u bytes", res);
 656
 657         if (strcmp(buf, "SCAN") == 0) {
 658                 test_driver_scan(drv, &from, fromlen);
 659         } else if (strncmp(buf, "ASSOC ", 6) == 0) {
 660                 test_driver_assoc(drv, &from, fromlen, buf + 6);
 661         } else if (strcmp(buf, "DISASSOC") == 0) {
 662                 test_driver_disassoc(drv, &from, fromlen);
 663         } else if (strncmp(buf, "EAPOL ", 6) == 0) {
 664                 test_driver_eapol(drv, &from, fromlen, (u8 *) buf + 6,
 665                                   res - 6);
 666         } else if (strncmp(buf, "ETHER ", 6) == 0) {
 667                 test_driver_ether(drv, &from, fromlen, (u8 *) buf + 6,
 668                                   res - 6);
 669         } else if (strncmp(buf, "MLME ", 5) == 0) {
 670                 test_driver_mlme(drv, &from, fromlen, (u8 *) buf + 5, res - 5);
 671         } else {
 672                 wpa_hexdump_ascii(MSG_DEBUG, "Unknown test_socket command",
 673                                   (u8 *) buf, res);
 674         }
 675 }
 676
 677
 678 static struct test_driver_bss *
 679 test_driver_get_bss(struct test_driver_data *drv, const char *ifname)
 680 {
 681         struct test_driver_bss *bss;
 682
 683         for (bss = drv->bss; bss; bss = bss->next) {
 684                 if (strcmp(bss->ifname, ifname) == 0)
 685                         return bss;
 686         }
 687         return NULL;
 688 }
 689
 690
 691 static int test_driver_set_generic_elem(const char *ifname, void *priv,
 692                                         const u8 *elem, size_t elem_len)
 693 {
 694         struct test_driver_data *drv = priv;
 695         struct test_driver_bss *bss;
 696
 697         bss = test_driver_get_bss(drv, ifname);
 698         if (bss == NULL)
 699                 return -1;
 700
 701         free(bss->ie);
 702
 703         if (elem == NULL) {
 704                 bss->ie = NULL;
 705                 bss->ielen = 0;
 706                 return 0;
 707         }
 708
 709         bss->ie = malloc(elem_len);
 710         if (bss->ie == NULL) {
 711                 bss->ielen = 0;
 712                 return -1;
 713         }
 714
 715         memcpy(bss->ie, elem, elem_len);
 716         bss->ielen = elem_len;
 717         return 0;
 718 }
 719
 720
 721 static int test_driver_sta_deauth(void *priv, const u8 *addr, int reason)
 722 {
 723         struct test_driver_data *drv = priv;
 724         struct test_client_socket *cli;
 725
 726         if (drv->test_socket < 0)
 727                 return -1;
 728
 729         cli = drv->cli;
 730         while (cli) {
 731                 if (memcmp(cli->addr, addr, ETH_ALEN) == 0)
 732                         break;
 733                 cli = cli->next;
 734         }
 735
 736         if (!cli)
 737                 return -1;
 738
 739         return sendto(drv->test_socket, "DEAUTH", 6, 0,
 740                       (struct sockaddr *) &cli->un, cli->unlen);
 741 }
 742
 743
 744 static int test_driver_sta_disassoc(void *priv, const u8 *addr, int reason)
 745 {
 746         struct test_driver_data *drv = priv;
 747         struct test_client_socket *cli;
 748
 749         if (drv->test_socket < 0)
 750                 return -1;
 751
 752         cli = drv->cli;
 753         while (cli) {
 754                 if (memcmp(cli->addr, addr, ETH_ALEN) == 0)
 755                         break;
 756                 cli = cli->next;
 757         }
 758
 759         if (!cli)
 760                 return -1;
 761
 762         return sendto(drv->test_socket, "DISASSOC", 8, 0,
 763                       (struct sockaddr *) &cli->un, cli->unlen);
 764 }
 765
 766
 767 static struct hostapd_hw_modes *
 768 test_driver_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
 769 {
 770         struct hostapd_hw_modes *modes;
 771
 772         *num_modes = 3;
 773         *flags = 0;
 774         modes = os_zalloc(*num_modes * sizeof(struct hostapd_hw_modes));
 775         if (modes == NULL)
 776                 return NULL;
 777         modes[0].mode = HOSTAPD_MODE_IEEE80211G;
 778         modes[0].num_channels = 1;
 779         modes[0].num_rates = 1;
 780         modes[0].channels = os_zalloc(sizeof(struct hostapd_channel_data));
 781         modes[0].rates = os_zalloc(sizeof(struct hostapd_rate_data));
 782         if (modes[0].channels == NULL || modes[0].rates == NULL) {
 783                 hostapd_free_hw_features(modes, *num_modes);
 784                 return NULL;
 785         }
 786         modes[0].channels[0].chan = 1;
 787         modes[0].channels[0].freq = 2412;
 788         modes[0].channels[0].flag = HOSTAPD_CHAN_W_SCAN |
 789                 HOSTAPD_CHAN_W_ACTIVE_SCAN;
 790         modes[0].rates[0].rate = 10;
 791         modes[0].rates[0].flags = HOSTAPD_RATE_BASIC | HOSTAPD_RATE_SUPPORTED |
 792                 HOSTAPD_RATE_CCK | HOSTAPD_RATE_MANDATORY;
 793
 794         modes[1].mode = HOSTAPD_MODE_IEEE80211B;
 795         modes[1].num_channels = 1;
 796         modes[1].num_rates = 1;
 797         modes[1].channels = os_zalloc(sizeof(struct hostapd_channel_data));
 798         modes[1].rates = os_zalloc(sizeof(struct hostapd_rate_data));
 799         if (modes[1].channels == NULL || modes[1].rates == NULL) {
 800                 hostapd_free_hw_features(modes, *num_modes);
 801                 return NULL;
 802         }
 803         modes[1].channels[0].chan = 1;
 804         modes[1].channels[0].freq = 2412;
 805         modes[1].channels[0].flag = HOSTAPD_CHAN_W_SCAN |
 806                 HOSTAPD_CHAN_W_ACTIVE_SCAN;
 807         modes[1].rates[0].rate = 10;
 808         modes[1].rates[0].flags = HOSTAPD_RATE_BASIC | HOSTAPD_RATE_SUPPORTED |
 809                 HOSTAPD_RATE_CCK | HOSTAPD_RATE_MANDATORY;
 810
 811         modes[2].mode = HOSTAPD_MODE_IEEE80211A;
 812         modes[2].num_channels = 1;
 813         modes[2].num_rates = 1;
 814         modes[2].channels = os_zalloc(sizeof(struct hostapd_channel_data));
 815         modes[2].rates = os_zalloc(sizeof(struct hostapd_rate_data));
 816         if (modes[2].channels == NULL || modes[2].rates == NULL) {
 817                 hostapd_free_hw_features(modes, *num_modes);
 818                 return NULL;
 819         }
 820         modes[2].channels[0].chan = 60;
 821         modes[2].channels[0].freq = 5300;
 822         modes[2].channels[0].flag = HOSTAPD_CHAN_W_SCAN |
 823                 HOSTAPD_CHAN_W_ACTIVE_SCAN;
 824         modes[2].rates[0].rate = 60;
 825         modes[2].rates[0].flags = HOSTAPD_RATE_BASIC | HOSTAPD_RATE_SUPPORTED |
 826                 HOSTAPD_RATE_MANDATORY;
 827
 828         return modes;
 829 }
 830
 831
 832 static int test_driver_bss_add(void *priv, const char *ifname, const u8 *bssid)
 833 {
 834         struct test_driver_data *drv = priv;
 835         struct test_driver_bss *bss;
 836
 837         wpa_printf(MSG_DEBUG, "%s(ifname=%s bssid=" MACSTR ")",
 838                    __func__, ifname, MAC2STR(bssid));
 839
 840         bss = os_zalloc(sizeof(*bss));
 841         if (bss == NULL)
 842                 return -1;
 843
 844         os_strlcpy(bss->ifname, ifname, IFNAMSIZ);
 845         memcpy(bss->bssid, bssid, ETH_ALEN);
 846
 847         bss->next = drv->bss;
 848         drv->bss = bss;
 849
 850         return 0;
 851 }
 852
 853
 854 static int test_driver_bss_remove(void *priv, const char *ifname)
 855 {
 856         struct test_driver_data *drv = priv;
 857         struct test_driver_bss *bss, *prev;
 858         struct test_client_socket *cli, *prev_c;
 859
 860         wpa_printf(MSG_DEBUG, "%s(ifname=%s)", __func__, ifname);
 861
 862         for (prev = NULL, bss = drv->bss; bss; prev = bss, bss = bss->next) {
 863                 if (strcmp(bss->ifname, ifname) != 0)
 864                         continue;
 865
 866                 if (prev)
 867                         prev->next = bss->next;
 868                 else
 869                         drv->bss = bss->next;
 870
 871                 for (prev_c = NULL, cli = drv->cli; cli;
 872                      prev_c = cli, cli = cli->next) {
 873                         if (cli->bss != bss)
 874                                 continue;
 875                         if (prev_c)
 876                                 prev_c->next = cli->next;
 877                         else
 878                                 drv->cli = cli->next;
 879                         free(cli);
 880                         break;
 881                 }
 882
 883                 test_driver_free_bss(bss);
 884                 return 0;
 885         }
 886
 887         return -1;
 888 }
 889
 890
 891 static int test_driver_if_add(const char *iface, void *priv,
 892                               enum hostapd_driver_if_type type, char *ifname,
 893                               const u8 *addr)
 894 {
 895         wpa_printf(MSG_DEBUG, "%s(iface=%s type=%d ifname=%s)",
 896                    __func__, iface, type, ifname);
 897         return 0;
 898 }
 899
 900
 901 static int test_driver_if_update(void *priv, enum hostapd_driver_if_type type,
 902                                  char *ifname, const u8 *addr)
 903 {
 904         wpa_printf(MSG_DEBUG, "%s(type=%d ifname=%s)", __func__, type, ifname);
 905         return 0;
 906 }
 907
 908
 909 static int test_driver_if_remove(void *priv, enum hostapd_driver_if_type type,
 910                                  const char *ifname, const u8 *addr)
 911 {
 912         wpa_printf(MSG_DEBUG, "%s(type=%d ifname=%s)", __func__, type, ifname);
 913         return 0;
 914 }
 915
 916
 917 static int test_driver_valid_bss_mask(void *priv, const u8 *addr,
 918                                       const u8 *mask)
 919 {
 920         return 0;
 921 }
 922
 923
 924 static int test_driver_set_ssid(const char *ifname, void *priv, const u8 *buf,
 925                                 int len)
 926 {
 927         struct test_driver_data *drv = priv;
 928         struct test_driver_bss *bss;
 929
 930         wpa_printf(MSG_DEBUG, "%s(ifname=%s)", __func__, ifname);
 931         wpa_hexdump_ascii(MSG_DEBUG, "test_driver_set_ssid: SSID", buf, len);
 932
 933         for (bss = drv->bss; bss; bss = bss->next) {
 934                 if (strcmp(bss->ifname, ifname) != 0)
 935                         continue;
 936
 937                 if (len < 0 || (size_t) len > sizeof(bss->ssid))
 938                         return -1;
 939
 940                 memcpy(bss->ssid, buf, len);
 941                 bss->ssid_len = len;
 942
 943                 return 0;
 944         }
 945
 946         return -1;
 947 }
 948
 949
 950 static int test_driver_set_privacy(const char *ifname, void *priv, int enabled)
 951 {
 952         struct test_driver_data *drv = priv;
 953         struct test_driver_bss *bss;
 954
 955         wpa_printf(MSG_DEBUG, "%s(ifname=%s enabled=%d)",
 956                    __func__, ifname, enabled);
 957
 958         for (bss = drv->bss; bss; bss = bss->next) {
 959                 if (strcmp(bss->ifname, ifname) != 0)
 960                         continue;
 961
 962                 bss->privacy = enabled;
 963
 964                 return 0;
 965         }
 966
 967         return -1;
 968 }
 969
 970
 971 static int test_driver_set_encryption(const char *iface, void *priv,
 972                                       const char *alg, const u8 *addr, int idx,
 973                                       const u8 *key, size_t key_len, int txkey)
 974 {
 975         wpa_printf(MSG_DEBUG, "%s(iface=%s alg=%s idx=%d txkey=%d)",
 976                    __func__, iface, alg, idx, txkey);
 977         if (addr)
 978                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
 979         if (key)
 980                 wpa_hexdump_key(MSG_DEBUG, "   key", key, key_len);
 981         return 0;
 982 }
 983
 984
 985 static int test_driver_set_sta_vlan(void *priv, const u8 *addr,
 986                                     const char *ifname, int vlan_id)
 987 {
 988         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " ifname=%s vlan_id=%d)",
 989                    __func__, MAC2STR(addr), ifname, vlan_id);
 990         return 0;
 991 }
 992
 993
 994 static int test_driver_sta_add(const char *ifname, void *priv, const u8 *addr,
 995                                u16 aid, u16 capability, u8 *supp_rates,
 996                                size_t supp_rates_len, int flags,
 997                                u16 listen_interval)
 998 {
 999         struct test_driver_data *drv = priv;
1000         struct test_client_socket *cli;
1001         struct test_driver_bss *bss;
1002
1003         wpa_printf(MSG_DEBUG, "%s(ifname=%s addr=" MACSTR " aid=%d "
1004                    "capability=0x%x flags=0x%x listen_interval=%d)",
1005                    __func__, ifname, MAC2STR(addr), aid, capability, flags,
1006                    listen_interval);
1007         wpa_hexdump(MSG_DEBUG, "test_driver_sta_add - supp_rates",
1008                     supp_rates, supp_rates_len);
1009
1010         cli = drv->cli;
1011         while (cli) {
1012                 if (memcmp(cli->addr, addr, ETH_ALEN) == 0)
1013                         break;
1014                 cli = cli->next;
1015         }
1016         if (!cli) {
1017                 wpa_printf(MSG_DEBUG, "%s: no matching client entry",
1018                            __func__);
1019                 return -1;
1020         }
1021
1022         for (bss = drv->bss; bss; bss = bss->next) {
1023                 if (strcmp(ifname, bss->ifname) == 0)
1024                         break;
1025         }
1026         if (bss == NULL) {
1027                 wpa_printf(MSG_DEBUG, "%s: No matching interface found from "
1028                            "configured BSSes", __func__);
1029                 return -1;
1030         }
1031
1032         cli->bss = bss;
1033
1034         return 0;
1035 }
1036
1037
1038 static void * test_driver_init(struct hostapd_data *hapd)
1039 {
1040         struct test_driver_data *drv;
1041         struct sockaddr_un addr;
1042
1043         drv = os_zalloc(sizeof(struct test_driver_data));
1044         if (drv == NULL) {
1045                 printf("Could not allocate memory for test driver data\n");
1046                 return NULL;
1047         }
1048         drv->bss = os_zalloc(sizeof(*drv->bss));
1049         if (drv->bss == NULL) {
1050                 printf("Could not allocate memory for test driver BSS data\n");
1051                 free(drv);
1052                 return NULL;
1053         }
1054
1055         drv->hapd = hapd;
1056
1057         /* Generate a MAC address to help testing with multiple APs */
1058         hapd->own_addr[0] = 0x02; /* locally administered */
1059         sha1_prf((const u8 *) hapd->conf->iface, strlen(hapd->conf->iface),
1060                  "hostapd test bssid generation",
1061                  (const u8 *) hapd->conf->ssid.ssid, hapd->conf->ssid.ssid_len,
1062                  hapd->own_addr + 1, ETH_ALEN - 1);
1063
1064         os_strlcpy(drv->bss->ifname, hapd->conf->iface, IFNAMSIZ);
1065         memcpy(drv->bss->bssid, hapd->own_addr, ETH_ALEN);
1066
1067         if (hapd->conf->test_socket) {
1068                 if (strlen(hapd->conf->test_socket) >= sizeof(addr.sun_path)) {
1069                         printf("Too long test_socket path\n");
1070                         test_driver_free_priv(drv);
1071                         return NULL;
1072                 }
1073                 if (strncmp(hapd->conf->test_socket, "DIR:", 4) == 0) {
1074                         size_t len = strlen(hapd->conf->test_socket) + 30;
1075                         drv->socket_dir = strdup(hapd->conf->test_socket + 4);
1076                         drv->own_socket_path = malloc(len);
1077                         if (drv->own_socket_path) {
1078                                 snprintf(drv->own_socket_path, len,
1079                                          "%s/AP-" MACSTR,
1080                                          hapd->conf->test_socket + 4,
1081                                          MAC2STR(hapd->own_addr));
1082                         }
1083                 } else {
1084                         drv->own_socket_path = strdup(hapd->conf->test_socket);
1085                 }
1086                 if (drv->own_socket_path == NULL) {
1087                         test_driver_free_priv(drv);
1088                         return NULL;
1089                 }
1090
1091                 drv->test_socket = socket(PF_UNIX, SOCK_DGRAM, 0);
1092                 if (drv->test_socket < 0) {
1093                         perror("socket(PF_UNIX)");
1094                         test_driver_free_priv(drv);
1095                         return NULL;
1096                 }
1097
1098                 memset(&addr, 0, sizeof(addr));
1099                 addr.sun_family = AF_UNIX;
1100                 os_strlcpy(addr.sun_path, drv->own_socket_path,
1101                            sizeof(addr.sun_path));
1102                 if (bind(drv->test_socket, (struct sockaddr *) &addr,
1103                          sizeof(addr)) < 0) {
1104                         perror("bind(PF_UNIX)");
1105                         close(drv->test_socket);
1106                         unlink(drv->own_socket_path);
1107                         test_driver_free_priv(drv);
1108                         return NULL;
1109                 }
1110                 eloop_register_read_sock(drv->test_socket,
1111                                          test_driver_receive_unix, drv, NULL);
1112         } else
1113                 drv->test_socket = -1;
1114
1115         return drv;
1116 }
1117
1118
1119 static void test_driver_deinit(void *priv)
1120 {
1121         struct test_driver_data *drv = priv;
1122         struct test_client_socket *cli, *prev;
1123
1124         cli = drv->cli;
1125         while (cli) {
1126                 prev = cli;
1127                 cli = cli->next;
1128                 free(prev);
1129         }
1130
1131         if (drv->test_socket >= 0) {
1132                 eloop_unregister_read_sock(drv->test_socket);
1133                 close(drv->test_socket);
1134                 unlink(drv->own_socket_path);
1135         }
1136
1137         /* There should be only one BSS remaining at this point. */
1138         if (drv->bss == NULL)
1139                 wpa_printf(MSG_ERROR, "%s: drv->bss == NULL", __func__);
1140         else if (drv->bss->next)
1141                 wpa_printf(MSG_ERROR, "%s: drv->bss->next != NULL", __func__);
1142
1143         test_driver_free_priv(drv);
1144 }
1145
1146
1147 const struct wpa_driver_ops wpa_driver_test_ops = {
1148         .name = "test",
1149         .init = test_driver_init,
1150         .deinit = test_driver_deinit,
1151         .send_eapol = test_driver_send_eapol,
1152         .send_mgmt_frame = test_driver_send_mgmt_frame,
1153         .set_generic_elem = test_driver_set_generic_elem,
1154         .sta_deauth = test_driver_sta_deauth,
1155         .sta_disassoc = test_driver_sta_disassoc,
1156         .get_hw_feature_data = test_driver_get_hw_feature_data,
1157         .bss_add = test_driver_bss_add,
1158         .bss_remove = test_driver_bss_remove,
1159         .if_add = test_driver_if_add,
1160         .if_update = test_driver_if_update,
1161         .if_remove = test_driver_if_remove,
1162         .valid_bss_mask = test_driver_valid_bss_mask,
1163         .set_ssid = test_driver_set_ssid,
1164         .set_privacy = test_driver_set_privacy,
1165         .set_encryption = test_driver_set_encryption,
1166         .set_sta_vlan = test_driver_set_sta_vlan,
1167         .sta_add = test_driver_sta_add,
1168         .send_ether = test_driver_send_ether,
1169 };