1/* 2 * Testing driver interface for a simulated network driver 3 * Copyright (c) 2004-2010, Jouni Malinen <j@w1.fi> 4 * 5 * This software may be distributed under the terms of the BSD license. 6 * See README for more details. 7 */ 8 9/* Make sure we get winsock2.h for Windows build to get sockaddr_storage */ 10#include "build_config.h" 11#ifdef CONFIG_NATIVE_WINDOWS 12#include <winsock2.h> 13#endif /* CONFIG_NATIVE_WINDOWS */ 14 15#include "utils/includes.h" 16 17#ifndef CONFIG_NATIVE_WINDOWS 18#include <sys/un.h> 19#include <dirent.h> 20#include <sys/stat.h> 21#define DRIVER_TEST_UNIX 22#endif /* CONFIG_NATIVE_WINDOWS */ 23 24#include "utils/common.h" 25#include "utils/eloop.h" 26#include "utils/list.h" 27#include "utils/trace.h" 28#include "common/ieee802_11_defs.h" 29#include "crypto/sha1.h" 30#include "l2_packet/l2_packet.h" 31#include "wps/wps.h" 32#include "driver.h" 33 34 35struct test_client_socket { 36 struct test_client_socket *next; 37 u8 addr[ETH_ALEN]; 38 struct sockaddr_un un; 39 socklen_t unlen; 40 struct test_driver_bss *bss; 41}; 42 43struct test_driver_bss { 44 struct wpa_driver_test_data *drv; 45 struct dl_list list; 46 void *bss_ctx; 47 char ifname[IFNAMSIZ]; 48 u8 bssid[ETH_ALEN]; 49 u8 *ie; 50 size_t ielen; 51 u8 *wps_beacon_ie; 52 size_t wps_beacon_ie_len; 53 u8 *wps_probe_resp_ie; 54 size_t wps_probe_resp_ie_len; 55 u8 ssid[32]; 56 size_t ssid_len; 57 int privacy; 58}; 59 60struct wpa_driver_test_global { 61 int bss_add_used; 62 u8 req_addr[ETH_ALEN]; 63}; 64 65struct wpa_driver_test_data { 66 struct wpa_driver_test_global *global; 67 void *ctx; 68 WPA_TRACE_REF(ctx); 69 u8 own_addr[ETH_ALEN]; 70 int test_socket; 71#ifdef DRIVER_TEST_UNIX 72 struct sockaddr_un hostapd_addr; 73#endif /* DRIVER_TEST_UNIX */ 74 int hostapd_addr_set; 75 struct sockaddr_in hostapd_addr_udp; 76 int hostapd_addr_udp_set; 77 char *own_socket_path; 78 char *test_dir; 79#define MAX_SCAN_RESULTS 30 80 struct wpa_scan_res *scanres[MAX_SCAN_RESULTS]; 81 size_t num_scanres; 82 int use_associnfo; 83 u8 assoc_wpa_ie[80]; 84 size_t assoc_wpa_ie_len; 85 int associated; 86 u8 *probe_req_ie; 87 size_t probe_req_ie_len; 88 u8 probe_req_ssid[32]; 89 size_t probe_req_ssid_len; 90 int ibss; 91 int ap; 92 93 struct test_client_socket *cli; 94 struct dl_list bss; 95 int udp_port; 96 97 int alloc_iface_idx; 98 99 int probe_req_report; 100 unsigned int remain_on_channel_freq; 101 unsigned int remain_on_channel_duration; 102 103 int current_freq; 104}; 105 106 107static void wpa_driver_test_deinit(void *priv); 108static int wpa_driver_test_attach(struct wpa_driver_test_data *drv, 109 const char *dir, int ap); 110static void wpa_driver_test_close_test_socket( 111 struct wpa_driver_test_data *drv); 112static void test_remain_on_channel_timeout(void *eloop_ctx, void *timeout_ctx); 113 114 115static void test_driver_free_bss(struct test_driver_bss *bss) 116{ 117 os_free(bss->ie); 118 os_free(bss->wps_beacon_ie); 119 os_free(bss->wps_probe_resp_ie); 120 os_free(bss); 121} 122 123 124static void test_driver_free_bsses(struct wpa_driver_test_data *drv) 125{ 126 struct test_driver_bss *bss, *tmp; 127 128 dl_list_for_each_safe(bss, tmp, &drv->bss, struct test_driver_bss, 129 list) { 130 dl_list_del(&bss->list); 131 test_driver_free_bss(bss); 132 } 133} 134 135 136static struct test_client_socket * 137test_driver_get_cli(struct wpa_driver_test_data *drv, struct sockaddr_un *from, 138 socklen_t fromlen) 139{ 140 struct test_client_socket *cli = drv->cli; 141 142 while (cli) { 143 if (cli->unlen == fromlen && 144 strncmp(cli->un.sun_path, from->sun_path, 145 fromlen - sizeof(cli->un.sun_family)) == 0) 146 return cli; 147 cli = cli->next; 148 } 149 150 return NULL; 151} 152 153 154static int test_driver_send_eapol(void *priv, const u8 *addr, const u8 *data, 155 size_t data_len, int encrypt, 156 const u8 *own_addr, u32 flags) 157{ 158 struct test_driver_bss *dbss = priv; 159 struct wpa_driver_test_data *drv = dbss->drv; 160 struct test_client_socket *cli; 161 struct msghdr msg; 162 struct iovec io[3]; 163 struct l2_ethhdr eth; 164 165 if (drv->test_socket < 0) 166 return -1; 167 168 cli = drv->cli; 169 while (cli) { 170 if (memcmp(cli->addr, addr, ETH_ALEN) == 0) 171 break; 172 cli = cli->next; 173 } 174 175 if (!cli) { 176 wpa_printf(MSG_DEBUG, "%s: no destination client entry", 177 __func__); 178 return -1; 179 } 180 181 memcpy(eth.h_dest, addr, ETH_ALEN); 182 memcpy(eth.h_source, own_addr, ETH_ALEN); 183 eth.h_proto = host_to_be16(ETH_P_EAPOL); 184 185 io[0].iov_base = "EAPOL "; 186 io[0].iov_len = 6; 187 io[1].iov_base = ð 188 io[1].iov_len = sizeof(eth); 189 io[2].iov_base = (u8 *) data; 190 io[2].iov_len = data_len; 191 192 memset(&msg, 0, sizeof(msg)); 193 msg.msg_iov = io; 194 msg.msg_iovlen = 3; 195 msg.msg_name = &cli->un; 196 msg.msg_namelen = cli->unlen; 197 return sendmsg(drv->test_socket, &msg, 0); 198} 199 200 201static int test_driver_send_ether(void *priv, const u8 *dst, const u8 *src, 202 u16 proto, const u8 *data, size_t data_len) 203{ 204 struct test_driver_bss *dbss = priv; 205 struct wpa_driver_test_data *drv = dbss->drv; 206 struct msghdr msg; 207 struct iovec io[3]; 208 struct l2_ethhdr eth; 209 char desttxt[30]; 210 struct sockaddr_un addr; 211 struct dirent *dent; 212 DIR *dir; 213 int ret = 0, broadcast = 0, count = 0; 214 215 if (drv->test_socket < 0 || drv->test_dir == NULL) { 216 wpa_printf(MSG_DEBUG, "%s: invalid parameters (sock=%d " 217 "test_dir=%p)", 218 __func__, drv->test_socket, drv->test_dir); 219 return -1; 220 } 221 222 broadcast = memcmp(dst, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) == 0; 223 snprintf(desttxt, sizeof(desttxt), MACSTR, MAC2STR(dst)); 224 225 memcpy(eth.h_dest, dst, ETH_ALEN); 226 memcpy(eth.h_source, src, ETH_ALEN); 227 eth.h_proto = host_to_be16(proto); 228 229 io[0].iov_base = "ETHER "; 230 io[0].iov_len = 6; 231 io[1].iov_base = ð 232 io[1].iov_len = sizeof(eth); 233 io[2].iov_base = (u8 *) data; 234 io[2].iov_len = data_len; 235 236 memset(&msg, 0, sizeof(msg)); 237 msg.msg_iov = io; 238 msg.msg_iovlen = 3; 239 240 dir = opendir(drv->test_dir); 241 if (dir == NULL) { 242 perror("test_driver: opendir"); 243 return -1; 244 } 245 while ((dent = readdir(dir))) { 246#ifdef _DIRENT_HAVE_D_TYPE 247 /* Skip the file if it is not a socket. Also accept 248 * DT_UNKNOWN (0) in case the C library or underlying file 249 * system does not support d_type. */ 250 if (dent->d_type != DT_SOCK && dent->d_type != DT_UNKNOWN) 251 continue; 252#endif /* _DIRENT_HAVE_D_TYPE */ 253 if (strcmp(dent->d_name, ".") == 0 || 254 strcmp(dent->d_name, "..") == 0) 255 continue; 256 257 memset(&addr, 0, sizeof(addr)); 258 addr.sun_family = AF_UNIX; 259 snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s", 260 drv->test_dir, dent->d_name); 261 262 if (strcmp(addr.sun_path, drv->own_socket_path) == 0) 263 continue; 264 if (!broadcast && strstr(dent->d_name, desttxt) == NULL) 265 continue; 266 267 wpa_printf(MSG_DEBUG, "%s: Send ether frame to %s", 268 __func__, dent->d_name); 269 270 msg.msg_name = &addr; 271 msg.msg_namelen = sizeof(addr); 272 ret = sendmsg(drv->test_socket, &msg, 0); 273 if (ret < 0) 274 perror("driver_test: sendmsg"); 275 count++; 276 } 277 closedir(dir); 278 279 if (!broadcast && count == 0) { 280 wpa_printf(MSG_DEBUG, "%s: Destination " MACSTR " not found", 281 __func__, MAC2STR(dst)); 282 return -1; 283 } 284 285 return ret; 286} 287 288 289static int wpa_driver_test_send_mlme(void *priv, const u8 *data, 290 size_t data_len, int noack) 291{ 292 struct test_driver_bss *dbss = priv; 293 struct wpa_driver_test_data *drv = dbss->drv; 294 struct msghdr msg; 295 struct iovec io[2]; 296 const u8 *dest; 297 struct sockaddr_un addr; 298 struct dirent *dent; 299 DIR *dir; 300 int broadcast; 301 int ret = 0; 302 struct ieee80211_hdr *hdr; 303 u16 fc; 304 char cmd[50]; 305 int freq; 306#ifdef HOSTAPD 307 char desttxt[30]; 308#endif /* HOSTAPD */ 309 union wpa_event_data event; 310 311 wpa_hexdump(MSG_MSGDUMP, "test_send_mlme", data, data_len); 312 if (drv->test_socket < 0 || data_len < 10) { 313 wpa_printf(MSG_DEBUG, "%s: invalid parameters (sock=%d len=%lu" 314 " test_dir=%p)", 315 __func__, drv->test_socket, 316 (unsigned long) data_len, 317 drv->test_dir); 318 return -1; 319 } 320 321 dest = data + 4; 322 broadcast = os_memcmp(dest, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) == 0; 323 324#ifdef HOSTAPD 325 snprintf(desttxt, sizeof(desttxt), MACSTR, MAC2STR(dest)); 326#endif /* HOSTAPD */ 327 328 if (drv->remain_on_channel_freq) 329 freq = drv->remain_on_channel_freq; 330 else 331 freq = drv->current_freq; 332 wpa_printf(MSG_DEBUG, "test_driver(%s): MLME TX on freq %d MHz", 333 dbss->ifname, freq); 334 os_snprintf(cmd, sizeof(cmd), "MLME freq=%d ", freq); 335 io[0].iov_base = cmd; 336 io[0].iov_len = os_strlen(cmd); 337 io[1].iov_base = (void *) data; 338 io[1].iov_len = data_len; 339 340 os_memset(&msg, 0, sizeof(msg)); 341 msg.msg_iov = io; 342 msg.msg_iovlen = 2; 343 344#ifdef HOSTAPD 345 if (drv->test_dir == NULL) { 346 wpa_printf(MSG_DEBUG, "%s: test_dir == NULL", __func__); 347 return -1; 348 } 349 350 dir = opendir(drv->test_dir); 351 if (dir == NULL) { 352 perror("test_driver: opendir"); 353 return -1; 354 } 355 while ((dent = readdir(dir))) { 356#ifdef _DIRENT_HAVE_D_TYPE 357 /* Skip the file if it is not a socket. Also accept 358 * DT_UNKNOWN (0) in case the C library or underlying file 359 * system does not support d_type. */ 360 if (dent->d_type != DT_SOCK && dent->d_type != DT_UNKNOWN) 361 continue; 362#endif /* _DIRENT_HAVE_D_TYPE */ 363 if (os_strcmp(dent->d_name, ".") == 0 || 364 os_strcmp(dent->d_name, "..") == 0) 365 continue; 366 367 os_memset(&addr, 0, sizeof(addr)); 368 addr.sun_family = AF_UNIX; 369 os_snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s", 370 drv->test_dir, dent->d_name); 371 372 if (os_strcmp(addr.sun_path, drv->own_socket_path) == 0) 373 continue; 374 if (!broadcast && os_strstr(dent->d_name, desttxt) == NULL) 375 continue; 376 377 wpa_printf(MSG_DEBUG, "%s: Send management frame to %s", 378 __func__, dent->d_name); 379 380 msg.msg_name = &addr; 381 msg.msg_namelen = sizeof(addr); 382 ret = sendmsg(drv->test_socket, &msg, 0); 383 if (ret < 0) 384 perror("driver_test: sendmsg(test_socket)"); 385 } 386 closedir(dir); 387#else /* HOSTAPD */ 388 389 if (os_memcmp(dest, dbss->bssid, ETH_ALEN) == 0 || 390 drv->test_dir == NULL) { 391 if (drv->hostapd_addr_udp_set) { 392 msg.msg_name = &drv->hostapd_addr_udp; 393 msg.msg_namelen = sizeof(drv->hostapd_addr_udp); 394 } else { 395#ifdef DRIVER_TEST_UNIX 396 msg.msg_name = &drv->hostapd_addr; 397 msg.msg_namelen = sizeof(drv->hostapd_addr); 398#endif /* DRIVER_TEST_UNIX */ 399 } 400 } else if (broadcast) { 401 dir = opendir(drv->test_dir); 402 if (dir == NULL) 403 return -1; 404 while ((dent = readdir(dir))) { 405#ifdef _DIRENT_HAVE_D_TYPE 406 /* Skip the file if it is not a socket. 407 * Also accept DT_UNKNOWN (0) in case 408 * the C library or underlying file 409 * system does not support d_type. */ 410 if (dent->d_type != DT_SOCK && 411 dent->d_type != DT_UNKNOWN) 412 continue; 413#endif /* _DIRENT_HAVE_D_TYPE */ 414 if (os_strcmp(dent->d_name, ".") == 0 || 415 os_strcmp(dent->d_name, "..") == 0) 416 continue; 417 wpa_printf(MSG_DEBUG, "%s: Send broadcast MLME to %s", 418 __func__, dent->d_name); 419 os_memset(&addr, 0, sizeof(addr)); 420 addr.sun_family = AF_UNIX; 421 os_snprintf(addr.sun_path, sizeof(addr.sun_path), 422 "%s/%s", drv->test_dir, dent->d_name); 423 424 msg.msg_name = &addr; 425 msg.msg_namelen = sizeof(addr); 426 427 ret = sendmsg(drv->test_socket, &msg, 0); 428 if (ret < 0) 429 perror("driver_test: sendmsg(test_socket)"); 430 } 431 closedir(dir); 432 return ret; 433 } else { 434 struct stat st; 435 os_memset(&addr, 0, sizeof(addr)); 436 addr.sun_family = AF_UNIX; 437 os_snprintf(addr.sun_path, sizeof(addr.sun_path), 438 "%s/AP-" MACSTR, drv->test_dir, MAC2STR(dest)); 439 if (stat(addr.sun_path, &st) < 0) { 440 os_snprintf(addr.sun_path, sizeof(addr.sun_path), 441 "%s/STA-" MACSTR, 442 drv->test_dir, MAC2STR(dest)); 443 } 444 msg.msg_name = &addr; 445 msg.msg_namelen = sizeof(addr); 446 } 447 448 if (sendmsg(drv->test_socket, &msg, 0) < 0) { 449 perror("sendmsg(test_socket)"); 450 return -1; 451 } 452#endif /* HOSTAPD */ 453 454 hdr = (struct ieee80211_hdr *) data; 455 fc = le_to_host16(hdr->frame_control); 456 457 os_memset(&event, 0, sizeof(event)); 458 event.tx_status.type = WLAN_FC_GET_TYPE(fc); 459 event.tx_status.stype = WLAN_FC_GET_STYPE(fc); 460 event.tx_status.dst = hdr->addr1; 461 event.tx_status.data = data; 462 event.tx_status.data_len = data_len; 463 event.tx_status.ack = ret >= 0; 464 wpa_supplicant_event(drv->ctx, EVENT_TX_STATUS, &event); 465 466 return ret; 467} 468 469 470static void test_driver_scan(struct wpa_driver_test_data *drv, 471 struct sockaddr_un *from, socklen_t fromlen, 472 char *data) 473{ 474 char buf[512], *pos, *end; 475 int ret; 476 struct test_driver_bss *bss; 477 u8 sa[ETH_ALEN]; 478 u8 ie[512]; 479 size_t ielen; 480 union wpa_event_data event; 481 482 /* data: optional [ ' ' | STA-addr | ' ' | IEs(hex) ] */ 483 484 wpa_printf(MSG_DEBUG, "test_driver: SCAN"); 485 486 if (*data) { 487 if (*data != ' ' || 488 hwaddr_aton(data + 1, sa)) { 489 wpa_printf(MSG_DEBUG, "test_driver: Unexpected SCAN " 490 "command format"); 491 return; 492 } 493 494 data += 18; 495 while (*data == ' ') 496 data++; 497 ielen = os_strlen(data) / 2; 498 if (ielen > sizeof(ie)) 499 ielen = sizeof(ie); 500 if (hexstr2bin(data, ie, ielen) < 0) 501 ielen = 0; 502 503 wpa_printf(MSG_DEBUG, "test_driver: Scan from " MACSTR, 504 MAC2STR(sa)); 505 wpa_hexdump(MSG_MSGDUMP, "test_driver: scan IEs", ie, ielen); 506 507 os_memset(&event, 0, sizeof(event)); 508 event.rx_probe_req.sa = sa; 509 event.rx_probe_req.ie = ie; 510 event.rx_probe_req.ie_len = ielen; 511 wpa_supplicant_event(drv->ctx, EVENT_RX_PROBE_REQ, &event); 512 } 513 514 dl_list_for_each(bss, &drv->bss, struct test_driver_bss, list) { 515 pos = buf; 516 end = buf + sizeof(buf); 517 518 /* reply: SCANRESP BSSID SSID IEs */ 519 ret = snprintf(pos, end - pos, "SCANRESP " MACSTR " ", 520 MAC2STR(bss->bssid)); 521 if (ret < 0 || ret >= end - pos) 522 return; 523 pos += ret; 524 pos += wpa_snprintf_hex(pos, end - pos, 525 bss->ssid, bss->ssid_len); 526 ret = snprintf(pos, end - pos, " "); 527 if (ret < 0 || ret >= end - pos) 528 return; 529 pos += ret; 530 pos += wpa_snprintf_hex(pos, end - pos, bss->ie, bss->ielen); 531 pos += wpa_snprintf_hex(pos, end - pos, bss->wps_probe_resp_ie, 532 bss->wps_probe_resp_ie_len); 533 534 if (bss->privacy) { 535 ret = snprintf(pos, end - pos, " PRIVACY"); 536 if (ret < 0 || ret >= end - pos) 537 return; 538 pos += ret; 539 } 540 541 sendto(drv->test_socket, buf, pos - buf, 0, 542 (struct sockaddr *) from, fromlen); 543 } 544} 545 546 547static void test_driver_assoc(struct wpa_driver_test_data *drv, 548 struct sockaddr_un *from, socklen_t fromlen, 549 char *data) 550{ 551 struct test_client_socket *cli; 552 u8 ie[256], ssid[32]; 553 size_t ielen, ssid_len = 0; 554 char *pos, *pos2, cmd[50]; 555 struct test_driver_bss *bss, *tmp; 556 557 /* data: STA-addr SSID(hex) IEs(hex) */ 558 559 cli = os_zalloc(sizeof(*cli)); 560 if (cli == NULL) 561 return; 562 563 if (hwaddr_aton(data, cli->addr)) { 564 printf("test_socket: Invalid MAC address '%s' in ASSOC\n", 565 data); 566 os_free(cli); 567 return; 568 } 569 pos = data + 17; 570 while (*pos == ' ') 571 pos++; 572 pos2 = strchr(pos, ' '); 573 ielen = 0; 574 if (pos2) { 575 ssid_len = (pos2 - pos) / 2; 576 if (hexstr2bin(pos, ssid, ssid_len) < 0) { 577 wpa_printf(MSG_DEBUG, "%s: Invalid SSID", __func__); 578 os_free(cli); 579 return; 580 } 581 wpa_hexdump_ascii(MSG_DEBUG, "test_driver_assoc: SSID", 582 ssid, ssid_len); 583 584 pos = pos2 + 1; 585 ielen = strlen(pos) / 2; 586 if (ielen > sizeof(ie)) 587 ielen = sizeof(ie); 588 if (hexstr2bin(pos, ie, ielen) < 0) 589 ielen = 0; 590 } 591 592 bss = NULL; 593 dl_list_for_each(tmp, &drv->bss, struct test_driver_bss, list) { 594 if (tmp->ssid_len == ssid_len && 595 os_memcmp(tmp->ssid, ssid, ssid_len) == 0) { 596 bss = tmp; 597 break; 598 } 599 } 600 if (bss == NULL) { 601 wpa_printf(MSG_DEBUG, "%s: No matching SSID found from " 602 "configured BSSes", __func__); 603 os_free(cli); 604 return; 605 } 606 607 cli->bss = bss; 608 memcpy(&cli->un, from, sizeof(cli->un)); 609 cli->unlen = fromlen; 610 cli->next = drv->cli; 611 drv->cli = cli; 612 wpa_hexdump_ascii(MSG_DEBUG, "test_socket: ASSOC sun_path", 613 (const u8 *) cli->un.sun_path, 614 cli->unlen - sizeof(cli->un.sun_family)); 615 616 snprintf(cmd, sizeof(cmd), "ASSOCRESP " MACSTR " 0", 617 MAC2STR(bss->bssid)); 618 sendto(drv->test_socket, cmd, strlen(cmd), 0, 619 (struct sockaddr *) from, fromlen); 620 621 drv_event_assoc(bss->bss_ctx, cli->addr, ie, ielen, 0); 622} 623 624 625static void test_driver_disassoc(struct wpa_driver_test_data *drv, 626 struct sockaddr_un *from, socklen_t fromlen) 627{ 628 struct test_client_socket *cli; 629 630 cli = test_driver_get_cli(drv, from, fromlen); 631 if (!cli) 632 return; 633 634 drv_event_disassoc(drv->ctx, cli->addr); 635} 636 637 638static void test_driver_eapol(struct wpa_driver_test_data *drv, 639 struct sockaddr_un *from, socklen_t fromlen, 640 u8 *data, size_t datalen) 641{ 642#ifdef HOSTAPD 643 struct test_client_socket *cli; 644#endif /* HOSTAPD */ 645 const u8 *src = NULL; 646 647 if (datalen > 14) { 648 /* Skip Ethernet header */ 649 src = data + ETH_ALEN; 650 wpa_printf(MSG_DEBUG, "test_driver: dst=" MACSTR " src=" 651 MACSTR " proto=%04x", 652 MAC2STR(data), MAC2STR(src), 653 WPA_GET_BE16(data + 2 * ETH_ALEN)); 654 data += 14; 655 datalen -= 14; 656 } 657 658#ifdef HOSTAPD 659 cli = test_driver_get_cli(drv, from, fromlen); 660 if (cli) { 661 drv_event_eapol_rx(cli->bss->bss_ctx, cli->addr, data, 662 datalen); 663 } else { 664 wpa_printf(MSG_DEBUG, "test_socket: EAPOL from unknown " 665 "client"); 666 } 667#else /* HOSTAPD */ 668 if (src) 669 drv_event_eapol_rx(drv->ctx, src, data, datalen); 670#endif /* HOSTAPD */ 671} 672 673 674static void test_driver_ether(struct wpa_driver_test_data *drv, 675 struct sockaddr_un *from, socklen_t fromlen, 676 u8 *data, size_t datalen) 677{ 678 struct l2_ethhdr *eth; 679 680 if (datalen < sizeof(*eth)) 681 return; 682 683 eth = (struct l2_ethhdr *) data; 684 wpa_printf(MSG_DEBUG, "test_driver: RX ETHER dst=" MACSTR " src=" 685 MACSTR " proto=%04x", 686 MAC2STR(eth->h_dest), MAC2STR(eth->h_source), 687 be_to_host16(eth->h_proto)); 688 689#ifdef CONFIG_IEEE80211R 690 if (be_to_host16(eth->h_proto) == ETH_P_RRB) { 691 union wpa_event_data ev; 692 os_memset(&ev, 0, sizeof(ev)); 693 ev.ft_rrb_rx.src = eth->h_source; 694 ev.ft_rrb_rx.data = data + sizeof(*eth); 695 ev.ft_rrb_rx.data_len = datalen - sizeof(*eth); 696 } 697#endif /* CONFIG_IEEE80211R */ 698} 699 700 701static void test_driver_mlme(struct wpa_driver_test_data *drv, 702 struct sockaddr_un *from, socklen_t fromlen, 703 u8 *data, size_t datalen) 704{ 705 struct ieee80211_hdr *hdr; 706 u16 fc; 707 union wpa_event_data event; 708 int freq = 0, own_freq; 709 struct test_driver_bss *bss; 710 711 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 712 713 if (datalen > 6 && os_memcmp(data, "freq=", 5) == 0) { 714 size_t pos; 715 for (pos = 5; pos < datalen; pos++) { 716 if (data[pos] == ' ') 717 break; 718 } 719 if (pos < datalen) { 720 freq = atoi((const char *) &data[5]); 721 wpa_printf(MSG_DEBUG, "test_driver(%s): MLME RX on " 722 "freq %d MHz", bss->ifname, freq); 723 pos++; 724 data += pos; 725 datalen -= pos; 726 } 727 } 728 729 if (drv->remain_on_channel_freq) 730 own_freq = drv->remain_on_channel_freq; 731 else 732 own_freq = drv->current_freq; 733 734 if (freq && own_freq && freq != own_freq) { 735 wpa_printf(MSG_DEBUG, "test_driver(%s): Ignore MLME RX on " 736 "another frequency %d MHz (own %d MHz)", 737 bss->ifname, freq, own_freq); 738 return; 739 } 740 741 hdr = (struct ieee80211_hdr *) data; 742 743 if (test_driver_get_cli(drv, from, fromlen) == NULL && datalen >= 16) { 744 struct test_client_socket *cli; 745 cli = os_zalloc(sizeof(*cli)); 746 if (cli == NULL) 747 return; 748 wpa_printf(MSG_DEBUG, "Adding client entry for " MACSTR, 749 MAC2STR(hdr->addr2)); 750 memcpy(cli->addr, hdr->addr2, ETH_ALEN); 751 memcpy(&cli->un, from, sizeof(cli->un)); 752 cli->unlen = fromlen; 753 cli->next = drv->cli; 754 drv->cli = cli; 755 } 756 757 wpa_hexdump(MSG_MSGDUMP, "test_driver_mlme: received frame", 758 data, datalen); 759 fc = le_to_host16(hdr->frame_control); 760 if (WLAN_FC_GET_TYPE(fc) != WLAN_FC_TYPE_MGMT) { 761 wpa_printf(MSG_ERROR, "%s: received non-mgmt frame", 762 __func__); 763 return; 764 } 765 766 os_memset(&event, 0, sizeof(event)); 767 event.rx_mgmt.frame = data; 768 event.rx_mgmt.frame_len = datalen; 769 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event); 770} 771 772 773static void test_driver_receive_unix(int sock, void *eloop_ctx, void *sock_ctx) 774{ 775 struct wpa_driver_test_data *drv = eloop_ctx; 776 char buf[2000]; 777 int res; 778 struct sockaddr_un from; 779 socklen_t fromlen = sizeof(from); 780 781 res = recvfrom(sock, buf, sizeof(buf) - 1, 0, 782 (struct sockaddr *) &from, &fromlen); 783 if (res < 0) { 784 perror("recvfrom(test_socket)"); 785 return; 786 } 787 buf[res] = '\0'; 788 789 wpa_printf(MSG_DEBUG, "test_driver: received %u bytes", res); 790 791 if (strncmp(buf, "SCAN", 4) == 0) { 792 test_driver_scan(drv, &from, fromlen, buf + 4); 793 } else if (strncmp(buf, "ASSOC ", 6) == 0) { 794 test_driver_assoc(drv, &from, fromlen, buf + 6); 795 } else if (strcmp(buf, "DISASSOC") == 0) { 796 test_driver_disassoc(drv, &from, fromlen); 797 } else if (strncmp(buf, "EAPOL ", 6) == 0) { 798 test_driver_eapol(drv, &from, fromlen, (u8 *) buf + 6, 799 res - 6); 800 } else if (strncmp(buf, "ETHER ", 6) == 0) { 801 test_driver_ether(drv, &from, fromlen, (u8 *) buf + 6, 802 res - 6); 803 } else if (strncmp(buf, "MLME ", 5) == 0) { 804 test_driver_mlme(drv, &from, fromlen, (u8 *) buf + 5, res - 5); 805 } else { 806 wpa_hexdump_ascii(MSG_DEBUG, "Unknown test_socket command", 807 (u8 *) buf, res); 808 } 809} 810 811 812static int test_driver_set_generic_elem(void *priv, 813 const u8 *elem, size_t elem_len) 814{ 815 struct test_driver_bss *bss = priv; 816 817 os_free(bss->ie); 818 819 if (elem == NULL) { 820 bss->ie = NULL; 821 bss->ielen = 0; 822 return 0; 823 } 824 825 bss->ie = os_malloc(elem_len); 826 if (bss->ie == NULL) { 827 bss->ielen = 0; 828 return -1; 829 } 830 831 memcpy(bss->ie, elem, elem_len); 832 bss->ielen = elem_len; 833 return 0; 834} 835 836 837static int test_driver_set_ap_wps_ie(void *priv, const struct wpabuf *beacon, 838 const struct wpabuf *proberesp, 839 const struct wpabuf *assocresp) 840{ 841 struct test_driver_bss *bss = priv; 842 843 if (beacon == NULL) 844 wpa_printf(MSG_DEBUG, "test_driver: Clear Beacon WPS IE"); 845 else 846 wpa_hexdump_buf(MSG_DEBUG, "test_driver: Beacon WPS IE", 847 beacon); 848 849 os_free(bss->wps_beacon_ie); 850 851 if (beacon == NULL) { 852 bss->wps_beacon_ie = NULL; 853 bss->wps_beacon_ie_len = 0; 854 } else { 855 bss->wps_beacon_ie = os_malloc(wpabuf_len(beacon)); 856 if (bss->wps_beacon_ie == NULL) { 857 bss->wps_beacon_ie_len = 0; 858 return -1; 859 } 860 861 os_memcpy(bss->wps_beacon_ie, wpabuf_head(beacon), 862 wpabuf_len(beacon)); 863 bss->wps_beacon_ie_len = wpabuf_len(beacon); 864 } 865 866 if (proberesp == NULL) 867 wpa_printf(MSG_DEBUG, "test_driver: Clear Probe Response WPS " 868 "IE"); 869 else 870 wpa_hexdump_buf(MSG_DEBUG, "test_driver: Probe Response WPS " 871 "IE", proberesp); 872 873 os_free(bss->wps_probe_resp_ie); 874 875 if (proberesp == NULL) { 876 bss->wps_probe_resp_ie = NULL; 877 bss->wps_probe_resp_ie_len = 0; 878 } else { 879 bss->wps_probe_resp_ie = os_malloc(wpabuf_len(proberesp)); 880 if (bss->wps_probe_resp_ie == NULL) { 881 bss->wps_probe_resp_ie_len = 0; 882 return -1; 883 } 884 885 os_memcpy(bss->wps_probe_resp_ie, wpabuf_head(proberesp), 886 wpabuf_len(proberesp)); 887 bss->wps_probe_resp_ie_len = wpabuf_len(proberesp); 888 } 889 890 return 0; 891} 892 893 894static int test_driver_sta_deauth(void *priv, const u8 *own_addr, 895 const u8 *addr, int reason) 896{ 897 struct test_driver_bss *dbss = priv; 898 struct wpa_driver_test_data *drv = dbss->drv; 899 struct test_client_socket *cli; 900 901 if (drv->test_socket < 0) 902 return -1; 903 904 cli = drv->cli; 905 while (cli) { 906 if (memcmp(cli->addr, addr, ETH_ALEN) == 0) 907 break; 908 cli = cli->next; 909 } 910 911 if (!cli) 912 return -1; 913 914 return sendto(drv->test_socket, "DEAUTH", 6, 0, 915 (struct sockaddr *) &cli->un, cli->unlen); 916} 917 918 919static int test_driver_sta_disassoc(void *priv, const u8 *own_addr, 920 const u8 *addr, int reason) 921{ 922 struct test_driver_bss *dbss = priv; 923 struct wpa_driver_test_data *drv = dbss->drv; 924 struct test_client_socket *cli; 925 926 if (drv->test_socket < 0) 927 return -1; 928 929 cli = drv->cli; 930 while (cli) { 931 if (memcmp(cli->addr, addr, ETH_ALEN) == 0) 932 break; 933 cli = cli->next; 934 } 935 936 if (!cli) 937 return -1; 938 939 return sendto(drv->test_socket, "DISASSOC", 8, 0, 940 (struct sockaddr *) &cli->un, cli->unlen); 941} 942 943 944static int test_driver_bss_add(void *priv, const char *ifname, const u8 *bssid, 945 void *bss_ctx, void **drv_priv) 946{ 947 struct test_driver_bss *dbss = priv; 948 struct wpa_driver_test_data *drv = dbss->drv; 949 struct test_driver_bss *bss; 950 951 wpa_printf(MSG_DEBUG, "%s(ifname=%s bssid=" MACSTR ")", 952 __func__, ifname, MAC2STR(bssid)); 953 954 bss = os_zalloc(sizeof(*bss)); 955 if (bss == NULL) 956 return -1; 957 958 bss->bss_ctx = bss_ctx; 959 bss->drv = drv; 960 os_strlcpy(bss->ifname, ifname, IFNAMSIZ); 961 os_memcpy(bss->bssid, bssid, ETH_ALEN); 962 963 dl_list_add(&drv->bss, &bss->list); 964 if (drv->global) { 965 drv->global->bss_add_used = 1; 966 os_memcpy(drv->global->req_addr, bssid, ETH_ALEN); 967 } 968 969 if (drv_priv) 970 *drv_priv = bss; 971 972 return 0; 973} 974 975 976static int test_driver_bss_remove(void *priv, const char *ifname) 977{ 978 struct test_driver_bss *dbss = priv; 979 struct wpa_driver_test_data *drv = dbss->drv; 980 struct test_driver_bss *bss; 981 struct test_client_socket *cli, *prev_c; 982 983 wpa_printf(MSG_DEBUG, "%s(ifname=%s)", __func__, ifname); 984 985 dl_list_for_each(bss, &drv->bss, struct test_driver_bss, list) { 986 if (strcmp(bss->ifname, ifname) != 0) 987 continue; 988 989 for (prev_c = NULL, cli = drv->cli; cli; 990 prev_c = cli, cli = cli->next) { 991 if (cli->bss != bss) 992 continue; 993 if (prev_c) 994 prev_c->next = cli->next; 995 else 996 drv->cli = cli->next; 997 os_free(cli); 998 break; 999 } 1000 1001 dl_list_del(&bss->list); 1002 test_driver_free_bss(bss); 1003 return 0; 1004 } 1005 1006 return -1; 1007} 1008 1009 1010static int test_driver_if_add(void *priv, enum wpa_driver_if_type type, 1011 const char *ifname, const u8 *addr, 1012 void *bss_ctx, void **drv_priv, 1013 char *force_ifname, u8 *if_addr, 1014 const char *bridge, int use_existing) 1015{ 1016 struct test_driver_bss *dbss = priv; 1017 struct wpa_driver_test_data *drv = dbss->drv; 1018 1019 wpa_printf(MSG_DEBUG, "%s(type=%d ifname=%s bss_ctx=%p)", 1020 __func__, type, ifname, bss_ctx); 1021 if (addr) 1022 os_memcpy(if_addr, addr, ETH_ALEN); 1023 else { 1024 drv->alloc_iface_idx++; 1025 if_addr[0] = 0x02; /* locally administered */ 1026 sha1_prf(drv->own_addr, ETH_ALEN, 1027 "hostapd test addr generation", 1028 (const u8 *) &drv->alloc_iface_idx, 1029 sizeof(drv->alloc_iface_idx), 1030 if_addr + 1, ETH_ALEN - 1); 1031 } 1032 if (type == WPA_IF_AP_BSS || type == WPA_IF_P2P_GO || 1033 type == WPA_IF_P2P_CLIENT || type == WPA_IF_P2P_GROUP) 1034 return test_driver_bss_add(priv, ifname, if_addr, bss_ctx, 1035 drv_priv); 1036 return 0; 1037} 1038 1039 1040static int test_driver_if_remove(void *priv, enum wpa_driver_if_type type, 1041 const char *ifname) 1042{ 1043 wpa_printf(MSG_DEBUG, "%s(type=%d ifname=%s)", __func__, type, ifname); 1044 if (type == WPA_IF_AP_BSS || type == WPA_IF_P2P_GO || 1045 type == WPA_IF_P2P_CLIENT || type == WPA_IF_P2P_GROUP) 1046 return test_driver_bss_remove(priv, ifname); 1047 return 0; 1048} 1049 1050 1051static int test_driver_set_ssid(void *priv, const u8 *buf, int len) 1052{ 1053 struct test_driver_bss *bss = priv; 1054 1055 wpa_printf(MSG_DEBUG, "%s(ifname=%s)", __func__, bss->ifname); 1056 if (len < 0) 1057 return -1; 1058 wpa_hexdump_ascii(MSG_DEBUG, "test_driver_set_ssid: SSID", buf, len); 1059 1060 if ((size_t) len > sizeof(bss->ssid)) 1061 return -1; 1062 1063 os_memcpy(bss->ssid, buf, len); 1064 bss->ssid_len = len; 1065 1066 return 0; 1067} 1068 1069 1070static int test_driver_set_privacy(void *priv, int enabled) 1071{ 1072 struct test_driver_bss *dbss = priv; 1073 1074 wpa_printf(MSG_DEBUG, "%s(enabled=%d)", __func__, enabled); 1075 dbss->privacy = enabled; 1076 1077 return 0; 1078} 1079 1080 1081static int test_driver_set_sta_vlan(void *priv, const u8 *addr, 1082 const char *ifname, int vlan_id) 1083{ 1084 wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " ifname=%s vlan_id=%d)", 1085 __func__, MAC2STR(addr), ifname, vlan_id); 1086 return 0; 1087} 1088 1089 1090static int test_driver_sta_add(void *priv, 1091 struct hostapd_sta_add_params *params) 1092{ 1093 struct test_driver_bss *bss = priv; 1094 struct wpa_driver_test_data *drv = bss->drv; 1095 struct test_client_socket *cli; 1096 1097 wpa_printf(MSG_DEBUG, "%s(ifname=%s addr=" MACSTR " aid=%d " 1098 "capability=0x%x listen_interval=%d)", 1099 __func__, bss->ifname, MAC2STR(params->addr), params->aid, 1100 params->capability, params->listen_interval); 1101 wpa_hexdump(MSG_DEBUG, "test_driver_sta_add - supp_rates", 1102 params->supp_rates, params->supp_rates_len); 1103 1104 cli = drv->cli; 1105 while (cli) { 1106 if (os_memcmp(cli->addr, params->addr, ETH_ALEN) == 0) 1107 break; 1108 cli = cli->next; 1109 } 1110 if (!cli) { 1111 wpa_printf(MSG_DEBUG, "%s: no matching client entry", 1112 __func__); 1113 return -1; 1114 } 1115 1116 cli->bss = bss; 1117 1118 return 0; 1119} 1120 1121 1122static struct wpa_driver_test_data * test_alloc_data(void *ctx, 1123 const char *ifname) 1124{ 1125 struct wpa_driver_test_data *drv; 1126 struct test_driver_bss *bss; 1127 1128 drv = os_zalloc(sizeof(struct wpa_driver_test_data)); 1129 if (drv == NULL) { 1130 wpa_printf(MSG_ERROR, "Could not allocate memory for test " 1131 "driver data"); 1132 return NULL; 1133 } 1134 1135 bss = os_zalloc(sizeof(struct test_driver_bss)); 1136 if (bss == NULL) { 1137 os_free(drv); 1138 return NULL; 1139 } 1140 1141 drv->ctx = ctx; 1142 wpa_trace_add_ref(drv, ctx, ctx); 1143 dl_list_init(&drv->bss); 1144 dl_list_add(&drv->bss, &bss->list); 1145 os_strlcpy(bss->ifname, ifname, IFNAMSIZ); 1146 bss->bss_ctx = ctx; 1147 bss->drv = drv; 1148 1149 /* Generate a MAC address to help testing with multiple STAs */ 1150 drv->own_addr[0] = 0x02; /* locally administered */ 1151 sha1_prf((const u8 *) ifname, os_strlen(ifname), 1152 "test mac addr generation", 1153 NULL, 0, drv->own_addr + 1, ETH_ALEN - 1); 1154 1155 return drv; 1156} 1157 1158 1159static void * test_driver_init(struct hostapd_data *hapd, 1160 struct wpa_init_params *params) 1161{ 1162 struct wpa_driver_test_data *drv; 1163 struct sockaddr_un addr_un; 1164 struct sockaddr_in addr_in; 1165 struct sockaddr *addr; 1166 socklen_t alen; 1167 struct test_driver_bss *bss; 1168 1169 drv = test_alloc_data(hapd, params->ifname); 1170 if (drv == NULL) 1171 return NULL; 1172 drv->ap = 1; 1173 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 1174 drv->global = params->global_priv; 1175 1176 bss->bss_ctx = hapd; 1177 os_memcpy(bss->bssid, drv->own_addr, ETH_ALEN); 1178 os_memcpy(params->own_addr, drv->own_addr, ETH_ALEN); 1179 1180 if (params->test_socket) { 1181 if (os_strlen(params->test_socket) >= 1182 sizeof(addr_un.sun_path)) { 1183 printf("Too long test_socket path\n"); 1184 wpa_driver_test_deinit(bss); 1185 return NULL; 1186 } 1187 if (strncmp(params->test_socket, "DIR:", 4) == 0) { 1188 size_t len = strlen(params->test_socket) + 30; 1189 drv->test_dir = os_strdup(params->test_socket + 4); 1190 drv->own_socket_path = os_malloc(len); 1191 if (drv->own_socket_path) { 1192 snprintf(drv->own_socket_path, len, 1193 "%s/AP-" MACSTR, 1194 params->test_socket + 4, 1195 MAC2STR(params->own_addr)); 1196 } 1197 } else if (strncmp(params->test_socket, "UDP:", 4) == 0) { 1198 drv->udp_port = atoi(params->test_socket + 4); 1199 } else { 1200 drv->own_socket_path = os_strdup(params->test_socket); 1201 } 1202 if (drv->own_socket_path == NULL && drv->udp_port == 0) { 1203 wpa_driver_test_deinit(bss); 1204 return NULL; 1205 } 1206 1207 drv->test_socket = socket(drv->udp_port ? PF_INET : PF_UNIX, 1208 SOCK_DGRAM, 0); 1209 if (drv->test_socket < 0) { 1210 perror("socket"); 1211 wpa_driver_test_deinit(bss); 1212 return NULL; 1213 } 1214 1215 if (drv->udp_port) { 1216 os_memset(&addr_in, 0, sizeof(addr_in)); 1217 addr_in.sin_family = AF_INET; 1218 addr_in.sin_port = htons(drv->udp_port); 1219 addr = (struct sockaddr *) &addr_in; 1220 alen = sizeof(addr_in); 1221 } else { 1222 os_memset(&addr_un, 0, sizeof(addr_un)); 1223 addr_un.sun_family = AF_UNIX; 1224 os_strlcpy(addr_un.sun_path, drv->own_socket_path, 1225 sizeof(addr_un.sun_path)); 1226 addr = (struct sockaddr *) &addr_un; 1227 alen = sizeof(addr_un); 1228 } 1229 if (bind(drv->test_socket, addr, alen) < 0) { 1230 perror("test-driver-init: bind(PF_UNIX)"); 1231 close(drv->test_socket); 1232 if (drv->own_socket_path) 1233 unlink(drv->own_socket_path); 1234 wpa_driver_test_deinit(bss); 1235 return NULL; 1236 } 1237 eloop_register_read_sock(drv->test_socket, 1238 test_driver_receive_unix, drv, NULL); 1239 } else 1240 drv->test_socket = -1; 1241 1242 return bss; 1243} 1244 1245 1246static void wpa_driver_test_poll(void *eloop_ctx, void *timeout_ctx) 1247{ 1248 struct wpa_driver_test_data *drv = eloop_ctx; 1249 1250#ifdef DRIVER_TEST_UNIX 1251 if (drv->associated && drv->hostapd_addr_set) { 1252 struct stat st; 1253 if (stat(drv->hostapd_addr.sun_path, &st) < 0) { 1254 wpa_printf(MSG_DEBUG, "%s: lost connection to AP: %s", 1255 __func__, strerror(errno)); 1256 drv->associated = 0; 1257 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL); 1258 } 1259 } 1260#endif /* DRIVER_TEST_UNIX */ 1261 1262 eloop_register_timeout(1, 0, wpa_driver_test_poll, drv, NULL); 1263} 1264 1265 1266static void wpa_driver_test_scan_timeout(void *eloop_ctx, void *timeout_ctx) 1267{ 1268 wpa_printf(MSG_DEBUG, "Scan timeout - try to get results"); 1269 wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL); 1270} 1271 1272 1273#ifdef DRIVER_TEST_UNIX 1274static void wpa_driver_scan_dir(struct wpa_driver_test_data *drv, 1275 const char *path) 1276{ 1277 struct dirent *dent; 1278 DIR *dir; 1279 struct sockaddr_un addr; 1280 char cmd[512], *pos, *end; 1281 int ret; 1282 1283 dir = opendir(path); 1284 if (dir == NULL) 1285 return; 1286 1287 end = cmd + sizeof(cmd); 1288 pos = cmd; 1289 ret = os_snprintf(pos, end - pos, "SCAN " MACSTR, 1290 MAC2STR(drv->own_addr)); 1291 if (ret >= 0 && ret < end - pos) 1292 pos += ret; 1293 if (drv->probe_req_ie) { 1294 ret = os_snprintf(pos, end - pos, " "); 1295 if (ret >= 0 && ret < end - pos) 1296 pos += ret; 1297 pos += wpa_snprintf_hex(pos, end - pos, drv->probe_req_ie, 1298 drv->probe_req_ie_len); 1299 } 1300 if (drv->probe_req_ssid_len) { 1301 /* Add SSID IE */ 1302 ret = os_snprintf(pos, end - pos, "%02x%02x", 1303 WLAN_EID_SSID, 1304 (unsigned int) drv->probe_req_ssid_len); 1305 if (ret >= 0 && ret < end - pos) 1306 pos += ret; 1307 pos += wpa_snprintf_hex(pos, end - pos, drv->probe_req_ssid, 1308 drv->probe_req_ssid_len); 1309 } 1310 end[-1] = '\0'; 1311 1312 while ((dent = readdir(dir))) { 1313 if (os_strncmp(dent->d_name, "AP-", 3) != 0 && 1314 os_strncmp(dent->d_name, "STA-", 4) != 0) 1315 continue; 1316 if (drv->own_socket_path) { 1317 size_t olen, dlen; 1318 olen = os_strlen(drv->own_socket_path); 1319 dlen = os_strlen(dent->d_name); 1320 if (olen >= dlen && 1321 os_strcmp(dent->d_name, 1322 drv->own_socket_path + olen - dlen) == 0) 1323 continue; 1324 } 1325 wpa_printf(MSG_DEBUG, "%s: SCAN %s", __func__, dent->d_name); 1326 1327 os_memset(&addr, 0, sizeof(addr)); 1328 addr.sun_family = AF_UNIX; 1329 os_snprintf(addr.sun_path, sizeof(addr.sun_path), "%s/%s", 1330 path, dent->d_name); 1331 1332 if (sendto(drv->test_socket, cmd, os_strlen(cmd), 0, 1333 (struct sockaddr *) &addr, sizeof(addr)) < 0) { 1334 perror("sendto(test_socket)"); 1335 } 1336 } 1337 closedir(dir); 1338} 1339#endif /* DRIVER_TEST_UNIX */ 1340 1341 1342static int wpa_driver_test_scan(void *priv, 1343 struct wpa_driver_scan_params *params) 1344{ 1345 struct test_driver_bss *dbss = priv; 1346 struct wpa_driver_test_data *drv = dbss->drv; 1347 size_t i; 1348 1349 wpa_printf(MSG_DEBUG, "%s: priv=%p", __func__, priv); 1350 1351 os_free(drv->probe_req_ie); 1352 if (params->extra_ies) { 1353 drv->probe_req_ie = os_malloc(params->extra_ies_len); 1354 if (drv->probe_req_ie == NULL) { 1355 drv->probe_req_ie_len = 0; 1356 return -1; 1357 } 1358 os_memcpy(drv->probe_req_ie, params->extra_ies, 1359 params->extra_ies_len); 1360 drv->probe_req_ie_len = params->extra_ies_len; 1361 } else { 1362 drv->probe_req_ie = NULL; 1363 drv->probe_req_ie_len = 0; 1364 } 1365 1366 for (i = 0; i < params->num_ssids; i++) 1367 wpa_hexdump(MSG_DEBUG, "Scan SSID", 1368 params->ssids[i].ssid, params->ssids[i].ssid_len); 1369 drv->probe_req_ssid_len = 0; 1370 if (params->num_ssids) { 1371 os_memcpy(drv->probe_req_ssid, params->ssids[0].ssid, 1372 params->ssids[0].ssid_len); 1373 drv->probe_req_ssid_len = params->ssids[0].ssid_len; 1374 } 1375 wpa_hexdump(MSG_DEBUG, "Scan extra IE(s)", 1376 params->extra_ies, params->extra_ies_len); 1377 1378 drv->num_scanres = 0; 1379 1380#ifdef DRIVER_TEST_UNIX 1381 if (drv->test_socket >= 0 && drv->test_dir) 1382 wpa_driver_scan_dir(drv, drv->test_dir); 1383 1384 if (drv->test_socket >= 0 && drv->hostapd_addr_set && 1385 sendto(drv->test_socket, "SCAN", 4, 0, 1386 (struct sockaddr *) &drv->hostapd_addr, 1387 sizeof(drv->hostapd_addr)) < 0) { 1388 perror("sendto(test_socket)"); 1389 } 1390#endif /* DRIVER_TEST_UNIX */ 1391 1392 if (drv->test_socket >= 0 && drv->hostapd_addr_udp_set && 1393 sendto(drv->test_socket, "SCAN", 4, 0, 1394 (struct sockaddr *) &drv->hostapd_addr_udp, 1395 sizeof(drv->hostapd_addr_udp)) < 0) { 1396 perror("sendto(test_socket)"); 1397 } 1398 1399 eloop_cancel_timeout(wpa_driver_test_scan_timeout, drv, drv->ctx); 1400 eloop_register_timeout(1, 0, wpa_driver_test_scan_timeout, drv, 1401 drv->ctx); 1402 return 0; 1403} 1404 1405 1406static struct wpa_scan_results * wpa_driver_test_get_scan_results2(void *priv) 1407{ 1408 struct test_driver_bss *dbss = priv; 1409 struct wpa_driver_test_data *drv = dbss->drv; 1410 struct wpa_scan_results *res; 1411 size_t i; 1412 1413 res = os_zalloc(sizeof(*res)); 1414 if (res == NULL) 1415 return NULL; 1416 1417 res->res = os_calloc(drv->num_scanres, sizeof(struct wpa_scan_res *)); 1418 if (res->res == NULL) { 1419 os_free(res); 1420 return NULL; 1421 } 1422 1423 for (i = 0; i < drv->num_scanres; i++) { 1424 struct wpa_scan_res *r; 1425 if (drv->scanres[i] == NULL) 1426 continue; 1427 r = os_malloc(sizeof(*r) + drv->scanres[i]->ie_len); 1428 if (r == NULL) 1429 break; 1430 os_memcpy(r, drv->scanres[i], 1431 sizeof(*r) + drv->scanres[i]->ie_len); 1432 res->res[res->num++] = r; 1433 } 1434 1435 return res; 1436} 1437 1438 1439static int wpa_driver_test_set_key(const char *ifname, void *priv, 1440 enum wpa_alg alg, const u8 *addr, 1441 int key_idx, int set_tx, 1442 const u8 *seq, size_t seq_len, 1443 const u8 *key, size_t key_len) 1444{ 1445 wpa_printf(MSG_DEBUG, "%s: ifname=%s priv=%p alg=%d key_idx=%d " 1446 "set_tx=%d", 1447 __func__, ifname, priv, alg, key_idx, set_tx); 1448 if (addr) 1449 wpa_printf(MSG_DEBUG, " addr=" MACSTR, MAC2STR(addr)); 1450 if (seq) 1451 wpa_hexdump(MSG_DEBUG, " seq", seq, seq_len); 1452 if (key) 1453 wpa_hexdump_key(MSG_DEBUG, " key", key, key_len); 1454 return 0; 1455} 1456 1457 1458static int wpa_driver_update_mode(struct wpa_driver_test_data *drv, int ap) 1459{ 1460 if (ap && !drv->ap) { 1461 wpa_driver_test_close_test_socket(drv); 1462 wpa_driver_test_attach(drv, drv->test_dir, 1); 1463 drv->ap = 1; 1464 } else if (!ap && drv->ap) { 1465 wpa_driver_test_close_test_socket(drv); 1466 wpa_driver_test_attach(drv, drv->test_dir, 0); 1467 drv->ap = 0; 1468 } 1469 1470 return 0; 1471} 1472 1473 1474static int wpa_driver_test_associate( 1475 void *priv, struct wpa_driver_associate_params *params) 1476{ 1477 struct test_driver_bss *dbss = priv; 1478 struct wpa_driver_test_data *drv = dbss->drv; 1479 wpa_printf(MSG_DEBUG, "%s: priv=%p freq=%d pairwise_suite=%d " 1480 "group_suite=%d key_mgmt_suite=%d auth_alg=%d mode=%d", 1481 __func__, priv, params->freq.freq, params->pairwise_suite, 1482 params->group_suite, params->key_mgmt_suite, 1483 params->auth_alg, params->mode); 1484 wpa_driver_update_mode(drv, params->mode == IEEE80211_MODE_AP); 1485 if (params->bssid) { 1486 wpa_printf(MSG_DEBUG, " bssid=" MACSTR, 1487 MAC2STR(params->bssid)); 1488 } 1489 if (params->ssid) { 1490 wpa_hexdump_ascii(MSG_DEBUG, " ssid", 1491 params->ssid, params->ssid_len); 1492 } 1493 if (params->wpa_ie) { 1494 wpa_hexdump(MSG_DEBUG, " wpa_ie", 1495 params->wpa_ie, params->wpa_ie_len); 1496 drv->assoc_wpa_ie_len = params->wpa_ie_len; 1497 if (drv->assoc_wpa_ie_len > sizeof(drv->assoc_wpa_ie)) 1498 drv->assoc_wpa_ie_len = sizeof(drv->assoc_wpa_ie); 1499 os_memcpy(drv->assoc_wpa_ie, params->wpa_ie, 1500 drv->assoc_wpa_ie_len); 1501 } else 1502 drv->assoc_wpa_ie_len = 0; 1503 1504 wpa_driver_update_mode(drv, params->mode == IEEE80211_MODE_AP); 1505 1506 drv->ibss = params->mode == IEEE80211_MODE_IBSS; 1507 dbss->privacy = params->key_mgmt_suite & 1508 (WPA_KEY_MGMT_IEEE8021X | 1509 WPA_KEY_MGMT_PSK | 1510 WPA_KEY_MGMT_WPA_NONE | 1511 WPA_KEY_MGMT_FT_IEEE8021X | 1512 WPA_KEY_MGMT_FT_PSK | 1513 WPA_KEY_MGMT_IEEE8021X_SHA256 | 1514 WPA_KEY_MGMT_PSK_SHA256); 1515 if (params->wep_key_len[params->wep_tx_keyidx]) 1516 dbss->privacy = 1; 1517 1518#ifdef DRIVER_TEST_UNIX 1519 if (drv->test_dir && params->bssid && 1520 params->mode != IEEE80211_MODE_IBSS) { 1521 os_memset(&drv->hostapd_addr, 0, sizeof(drv->hostapd_addr)); 1522 drv->hostapd_addr.sun_family = AF_UNIX; 1523 os_snprintf(drv->hostapd_addr.sun_path, 1524 sizeof(drv->hostapd_addr.sun_path), 1525 "%s/AP-" MACSTR, 1526 drv->test_dir, MAC2STR(params->bssid)); 1527 drv->hostapd_addr_set = 1; 1528 } 1529#endif /* DRIVER_TEST_UNIX */ 1530 1531 if (params->mode == IEEE80211_MODE_AP) { 1532 if (params->ssid) 1533 os_memcpy(dbss->ssid, params->ssid, params->ssid_len); 1534 dbss->ssid_len = params->ssid_len; 1535 os_memcpy(dbss->bssid, drv->own_addr, ETH_ALEN); 1536 if (params->wpa_ie && params->wpa_ie_len) { 1537 dbss->ie = os_malloc(params->wpa_ie_len); 1538 if (dbss->ie) { 1539 os_memcpy(dbss->ie, params->wpa_ie, 1540 params->wpa_ie_len); 1541 dbss->ielen = params->wpa_ie_len; 1542 } 1543 } 1544 } else if (drv->test_socket >= 0 && 1545 (drv->hostapd_addr_set || drv->hostapd_addr_udp_set)) { 1546 char cmd[200], *pos, *end; 1547 int ret; 1548 end = cmd + sizeof(cmd); 1549 pos = cmd; 1550 ret = os_snprintf(pos, end - pos, "ASSOC " MACSTR " ", 1551 MAC2STR(drv->own_addr)); 1552 if (ret >= 0 && ret < end - pos) 1553 pos += ret; 1554 if (params->ssid) 1555 pos += wpa_snprintf_hex(pos, end - pos, params->ssid, 1556 params->ssid_len); 1557 ret = os_snprintf(pos, end - pos, " "); 1558 if (ret >= 0 && ret < end - pos) 1559 pos += ret; 1560 pos += wpa_snprintf_hex(pos, end - pos, params->wpa_ie, 1561 params->wpa_ie_len); 1562 end[-1] = '\0'; 1563#ifdef DRIVER_TEST_UNIX 1564 if (drv->hostapd_addr_set && 1565 sendto(drv->test_socket, cmd, os_strlen(cmd), 0, 1566 (struct sockaddr *) &drv->hostapd_addr, 1567 sizeof(drv->hostapd_addr)) < 0) { 1568 perror("sendto(test_socket)"); 1569 return -1; 1570 } 1571#endif /* DRIVER_TEST_UNIX */ 1572 if (drv->hostapd_addr_udp_set && 1573 sendto(drv->test_socket, cmd, os_strlen(cmd), 0, 1574 (struct sockaddr *) &drv->hostapd_addr_udp, 1575 sizeof(drv->hostapd_addr_udp)) < 0) { 1576 perror("sendto(test_socket)"); 1577 return -1; 1578 } 1579 1580 if (params->ssid) 1581 os_memcpy(dbss->ssid, params->ssid, params->ssid_len); 1582 dbss->ssid_len = params->ssid_len; 1583 } else { 1584 drv->associated = 1; 1585 if (params->mode == IEEE80211_MODE_IBSS) { 1586 if (params->ssid) 1587 os_memcpy(dbss->ssid, params->ssid, 1588 params->ssid_len); 1589 dbss->ssid_len = params->ssid_len; 1590 if (params->bssid) 1591 os_memcpy(dbss->bssid, params->bssid, 1592 ETH_ALEN); 1593 else { 1594 os_get_random(dbss->bssid, ETH_ALEN); 1595 dbss->bssid[0] &= ~0x01; 1596 dbss->bssid[0] |= 0x02; 1597 } 1598 } 1599 wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL); 1600 } 1601 1602 return 0; 1603} 1604 1605 1606static int wpa_driver_test_get_bssid(void *priv, u8 *bssid) 1607{ 1608 struct test_driver_bss *dbss = priv; 1609 os_memcpy(bssid, dbss->bssid, ETH_ALEN); 1610 return 0; 1611} 1612 1613 1614static int wpa_driver_test_get_ssid(void *priv, u8 *ssid) 1615{ 1616 struct test_driver_bss *dbss = priv; 1617 os_memcpy(ssid, dbss->ssid, 32); 1618 return dbss->ssid_len; 1619} 1620 1621 1622static int wpa_driver_test_send_disassoc(struct wpa_driver_test_data *drv) 1623{ 1624#ifdef DRIVER_TEST_UNIX 1625 if (drv->test_socket >= 0 && 1626 sendto(drv->test_socket, "DISASSOC", 8, 0, 1627 (struct sockaddr *) &drv->hostapd_addr, 1628 sizeof(drv->hostapd_addr)) < 0) { 1629 perror("sendto(test_socket)"); 1630 return -1; 1631 } 1632#endif /* DRIVER_TEST_UNIX */ 1633 if (drv->test_socket >= 0 && drv->hostapd_addr_udp_set && 1634 sendto(drv->test_socket, "DISASSOC", 8, 0, 1635 (struct sockaddr *) &drv->hostapd_addr_udp, 1636 sizeof(drv->hostapd_addr_udp)) < 0) { 1637 perror("sendto(test_socket)"); 1638 return -1; 1639 } 1640 return 0; 1641} 1642 1643 1644static int wpa_driver_test_deauthenticate(void *priv, const u8 *addr, 1645 int reason_code) 1646{ 1647 struct test_driver_bss *dbss = priv; 1648 struct wpa_driver_test_data *drv = dbss->drv; 1649 wpa_printf(MSG_DEBUG, "%s addr=" MACSTR " reason_code=%d", 1650 __func__, MAC2STR(addr), reason_code); 1651 os_memset(dbss->bssid, 0, ETH_ALEN); 1652 drv->associated = 0; 1653 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL); 1654 return wpa_driver_test_send_disassoc(drv); 1655} 1656 1657 1658static const u8 * wpa_scan_get_ie(const struct wpa_scan_res *res, u8 ie) 1659{ 1660 const u8 *end, *pos; 1661 1662 pos = (const u8 *) (res + 1); 1663 end = pos + res->ie_len; 1664 1665 while (pos + 1 < end) { 1666 if (pos + 2 + pos[1] > end) 1667 break; 1668 if (pos[0] == ie) 1669 return pos; 1670 pos += 2 + pos[1]; 1671 } 1672 1673 return NULL; 1674} 1675 1676 1677static void wpa_driver_test_scanresp(struct wpa_driver_test_data *drv, 1678 struct sockaddr *from, 1679 socklen_t fromlen, 1680 const char *data) 1681{ 1682 struct wpa_scan_res *res; 1683 const char *pos, *pos2; 1684 size_t len; 1685 u8 *ie_pos, *ie_start, *ie_end; 1686#define MAX_IE_LEN 1000 1687 const u8 *ds_params; 1688 1689 wpa_printf(MSG_DEBUG, "test_driver: SCANRESP %s", data); 1690 if (drv->num_scanres >= MAX_SCAN_RESULTS) { 1691 wpa_printf(MSG_DEBUG, "test_driver: No room for the new scan " 1692 "result"); 1693 return; 1694 } 1695 1696 /* SCANRESP BSSID SSID IEs */ 1697 1698 res = os_zalloc(sizeof(*res) + MAX_IE_LEN); 1699 if (res == NULL) 1700 return; 1701 ie_start = ie_pos = (u8 *) (res + 1); 1702 ie_end = ie_pos + MAX_IE_LEN; 1703 1704 if (hwaddr_aton(data, res->bssid)) { 1705 wpa_printf(MSG_DEBUG, "test_driver: invalid BSSID in scanres"); 1706 os_free(res); 1707 return; 1708 } 1709 1710 pos = data + 17; 1711 while (*pos == ' ') 1712 pos++; 1713 pos2 = os_strchr(pos, ' '); 1714 if (pos2 == NULL) { 1715 wpa_printf(MSG_DEBUG, "test_driver: invalid SSID termination " 1716 "in scanres"); 1717 os_free(res); 1718 return; 1719 } 1720 len = (pos2 - pos) / 2; 1721 if (len > 32) 1722 len = 32; 1723 /* 1724 * Generate SSID IE from the SSID field since this IE is not included 1725 * in the main IE field. 1726 */ 1727 *ie_pos++ = WLAN_EID_SSID; 1728 *ie_pos++ = len; 1729 if (hexstr2bin(pos, ie_pos, len) < 0) { 1730 wpa_printf(MSG_DEBUG, "test_driver: invalid SSID in scanres"); 1731 os_free(res); 1732 return; 1733 } 1734 ie_pos += len; 1735 1736 pos = pos2 + 1; 1737 pos2 = os_strchr(pos, ' '); 1738 if (pos2 == NULL) 1739 len = os_strlen(pos) / 2; 1740 else 1741 len = (pos2 - pos) / 2; 1742 if ((int) len > ie_end - ie_pos) 1743 len = ie_end - ie_pos; 1744 if (hexstr2bin(pos, ie_pos, len) < 0) { 1745 wpa_printf(MSG_DEBUG, "test_driver: invalid IEs in scanres"); 1746 os_free(res); 1747 return; 1748 } 1749 ie_pos += len; 1750 res->ie_len = ie_pos - ie_start; 1751 1752 if (pos2) { 1753 pos = pos2 + 1; 1754 while (*pos == ' ') 1755 pos++; 1756 if (os_strstr(pos, "PRIVACY")) 1757 res->caps |= IEEE80211_CAP_PRIVACY; 1758 if (os_strstr(pos, "IBSS")) 1759 res->caps |= IEEE80211_CAP_IBSS; 1760 } 1761 1762 ds_params = wpa_scan_get_ie(res, WLAN_EID_DS_PARAMS); 1763 if (ds_params && ds_params[1] > 0) { 1764 if (ds_params[2] >= 1 && ds_params[2] <= 13) 1765 res->freq = 2407 + ds_params[2] * 5; 1766 } 1767 1768 os_free(drv->scanres[drv->num_scanres]); 1769 drv->scanres[drv->num_scanres++] = res; 1770} 1771 1772 1773static void wpa_driver_test_assocresp(struct wpa_driver_test_data *drv, 1774 struct sockaddr *from, 1775 socklen_t fromlen, 1776 const char *data) 1777{ 1778 struct test_driver_bss *bss; 1779 1780 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 1781 1782 /* ASSOCRESP BSSID <res> */ 1783 if (hwaddr_aton(data, bss->bssid)) { 1784 wpa_printf(MSG_DEBUG, "test_driver: invalid BSSID in " 1785 "assocresp"); 1786 } 1787 if (drv->use_associnfo) { 1788 union wpa_event_data event; 1789 os_memset(&event, 0, sizeof(event)); 1790 event.assoc_info.req_ies = drv->assoc_wpa_ie; 1791 event.assoc_info.req_ies_len = drv->assoc_wpa_ie_len; 1792 wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &event); 1793 } 1794 drv->associated = 1; 1795 wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL); 1796} 1797 1798 1799static void wpa_driver_test_disassoc(struct wpa_driver_test_data *drv, 1800 struct sockaddr *from, 1801 socklen_t fromlen) 1802{ 1803 drv->associated = 0; 1804 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL); 1805} 1806 1807 1808static void wpa_driver_test_eapol(struct wpa_driver_test_data *drv, 1809 struct sockaddr *from, 1810 socklen_t fromlen, 1811 const u8 *data, size_t data_len) 1812{ 1813 const u8 *src; 1814 struct test_driver_bss *bss; 1815 1816 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 1817 1818 if (data_len > 14) { 1819 /* Skip Ethernet header */ 1820 src = data + ETH_ALEN; 1821 data += 14; 1822 data_len -= 14; 1823 } else 1824 src = bss->bssid; 1825 1826 drv_event_eapol_rx(drv->ctx, src, data, data_len); 1827} 1828 1829 1830static void wpa_driver_test_mlme(struct wpa_driver_test_data *drv, 1831 struct sockaddr *from, 1832 socklen_t fromlen, 1833 const u8 *data, size_t data_len) 1834{ 1835 int freq = 0, own_freq; 1836 union wpa_event_data event; 1837 const struct ieee80211_mgmt *mgmt; 1838 u16 fc; 1839 struct test_driver_bss *bss; 1840 1841 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 1842 if (data_len > 6 && os_memcmp(data, "freq=", 5) == 0) { 1843 size_t pos; 1844 for (pos = 5; pos < data_len; pos++) { 1845 if (data[pos] == ' ') 1846 break; 1847 } 1848 if (pos < data_len) { 1849 freq = atoi((const char *) &data[5]); 1850 wpa_printf(MSG_DEBUG, "test_driver(%s): MLME RX on " 1851 "freq %d MHz", bss->ifname, freq); 1852 pos++; 1853 data += pos; 1854 data_len -= pos; 1855 } 1856 } 1857 1858 if (drv->remain_on_channel_freq) 1859 own_freq = drv->remain_on_channel_freq; 1860 else 1861 own_freq = drv->current_freq; 1862 1863 if (freq && own_freq && freq != own_freq) { 1864 wpa_printf(MSG_DEBUG, "test_driver(%s): Ignore MLME RX on " 1865 "another frequency %d MHz (own %d MHz)", 1866 bss->ifname, freq, own_freq); 1867 return; 1868 } 1869 1870 os_memset(&event, 0, sizeof(event)); 1871 event.mlme_rx.buf = data; 1872 event.mlme_rx.len = data_len; 1873 event.mlme_rx.freq = freq; 1874 wpa_supplicant_event(drv->ctx, EVENT_MLME_RX, &event); 1875 1876 mgmt = (const struct ieee80211_mgmt *) data; 1877 fc = le_to_host16(mgmt->frame_control); 1878 1879 if (drv->probe_req_report && data_len >= 24) { 1880 if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT && 1881 WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_PROBE_REQ) { 1882 os_memset(&event, 0, sizeof(event)); 1883 event.rx_probe_req.sa = mgmt->sa; 1884 event.rx_probe_req.da = mgmt->da; 1885 event.rx_probe_req.bssid = mgmt->bssid; 1886 event.rx_probe_req.ie = mgmt->u.probe_req.variable; 1887 event.rx_probe_req.ie_len = 1888 data_len - (mgmt->u.probe_req.variable - data); 1889 wpa_supplicant_event(drv->ctx, EVENT_RX_PROBE_REQ, 1890 &event); 1891 } 1892 } 1893} 1894 1895 1896static void wpa_driver_test_scan_cmd(struct wpa_driver_test_data *drv, 1897 struct sockaddr *from, 1898 socklen_t fromlen, 1899 const u8 *data, size_t data_len) 1900{ 1901 char buf[512], *pos, *end; 1902 int ret; 1903 struct test_driver_bss *bss; 1904 1905 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 1906 1907 /* data: optional [ STA-addr | ' ' | IEs(hex) ] */ 1908 1909 if (bss == NULL || !drv->ibss) 1910 return; 1911 1912 pos = buf; 1913 end = buf + sizeof(buf); 1914 1915 /* reply: SCANRESP BSSID SSID IEs */ 1916 ret = snprintf(pos, end - pos, "SCANRESP " MACSTR " ", 1917 MAC2STR(bss->bssid)); 1918 if (ret < 0 || ret >= end - pos) 1919 return; 1920 pos += ret; 1921 pos += wpa_snprintf_hex(pos, end - pos, 1922 bss->ssid, bss->ssid_len); 1923 ret = snprintf(pos, end - pos, " "); 1924 if (ret < 0 || ret >= end - pos) 1925 return; 1926 pos += ret; 1927 pos += wpa_snprintf_hex(pos, end - pos, drv->assoc_wpa_ie, 1928 drv->assoc_wpa_ie_len); 1929 1930 if (bss->privacy) { 1931 ret = snprintf(pos, end - pos, " PRIVACY"); 1932 if (ret < 0 || ret >= end - pos) 1933 return; 1934 pos += ret; 1935 } 1936 1937 ret = snprintf(pos, end - pos, " IBSS"); 1938 if (ret < 0 || ret >= end - pos) 1939 return; 1940 pos += ret; 1941 1942 sendto(drv->test_socket, buf, pos - buf, 0, 1943 (struct sockaddr *) from, fromlen); 1944} 1945 1946 1947static void wpa_driver_test_receive_unix(int sock, void *eloop_ctx, 1948 void *sock_ctx) 1949{ 1950 struct wpa_driver_test_data *drv = eloop_ctx; 1951 char *buf; 1952 int res; 1953 struct sockaddr_storage from; 1954 socklen_t fromlen = sizeof(from); 1955 const size_t buflen = 2000; 1956 1957 if (drv->ap) { 1958 test_driver_receive_unix(sock, eloop_ctx, sock_ctx); 1959 return; 1960 } 1961 1962 buf = os_malloc(buflen); 1963 if (buf == NULL) 1964 return; 1965 res = recvfrom(sock, buf, buflen - 1, 0, 1966 (struct sockaddr *) &from, &fromlen); 1967 if (res < 0) { 1968 perror("recvfrom(test_socket)"); 1969 os_free(buf); 1970 return; 1971 } 1972 buf[res] = '\0'; 1973 1974 wpa_printf(MSG_DEBUG, "test_driver: received %u bytes", res); 1975 1976 if (os_strncmp(buf, "SCANRESP ", 9) == 0) { 1977 wpa_driver_test_scanresp(drv, (struct sockaddr *) &from, 1978 fromlen, buf + 9); 1979 } else if (os_strncmp(buf, "ASSOCRESP ", 10) == 0) { 1980 wpa_driver_test_assocresp(drv, (struct sockaddr *) &from, 1981 fromlen, buf + 10); 1982 } else if (os_strcmp(buf, "DISASSOC") == 0) { 1983 wpa_driver_test_disassoc(drv, (struct sockaddr *) &from, 1984 fromlen); 1985 } else if (os_strcmp(buf, "DEAUTH") == 0) { 1986 wpa_driver_test_disassoc(drv, (struct sockaddr *) &from, 1987 fromlen); 1988 } else if (os_strncmp(buf, "EAPOL ", 6) == 0) { 1989 wpa_driver_test_eapol(drv, (struct sockaddr *) &from, fromlen, 1990 (const u8 *) buf + 6, res - 6); 1991 } else if (os_strncmp(buf, "MLME ", 5) == 0) { 1992 wpa_driver_test_mlme(drv, (struct sockaddr *) &from, fromlen, 1993 (const u8 *) buf + 5, res - 5); 1994 } else if (os_strncmp(buf, "SCAN ", 5) == 0) { 1995 wpa_driver_test_scan_cmd(drv, (struct sockaddr *) &from, 1996 fromlen, 1997 (const u8 *) buf + 5, res - 5); 1998 } else { 1999 wpa_hexdump_ascii(MSG_DEBUG, "Unknown test_socket command", 2000 (u8 *) buf, res); 2001 } 2002 os_free(buf); 2003} 2004 2005 2006static void * wpa_driver_test_init2(void *ctx, const char *ifname, 2007 void *global_priv) 2008{ 2009 struct wpa_driver_test_data *drv; 2010 struct wpa_driver_test_global *global = global_priv; 2011 struct test_driver_bss *bss; 2012 2013 drv = test_alloc_data(ctx, ifname); 2014 if (drv == NULL) 2015 return NULL; 2016 bss = dl_list_first(&drv->bss, struct test_driver_bss, list); 2017 drv->global = global_priv; 2018 drv->test_socket = -1; 2019 2020 /* Set dummy BSSID and SSID for testing. */ 2021 bss->bssid[0] = 0x02; 2022 bss->bssid[1] = 0x00; 2023 bss->bssid[2] = 0x00; 2024 bss->bssid[3] = 0x00; 2025 bss->bssid[4] = 0x00; 2026 bss->bssid[5] = 0x01; 2027 os_memcpy(bss->ssid, "test", 5); 2028 bss->ssid_len = 4; 2029 2030 if (global->bss_add_used) { 2031 os_memcpy(drv->own_addr, global->req_addr, ETH_ALEN); 2032 global->bss_add_used = 0; 2033 } 2034 2035 eloop_register_timeout(1, 0, wpa_driver_test_poll, drv, NULL); 2036 2037 return bss; 2038} 2039 2040 2041static void wpa_driver_test_close_test_socket(struct wpa_driver_test_data *drv) 2042{ 2043 if (drv->test_socket >= 0) { 2044 eloop_unregister_read_sock(drv->test_socket); 2045 close(drv->test_socket); 2046 drv->test_socket = -1; 2047 } 2048 2049 if (drv->own_socket_path) { 2050 unlink(drv->own_socket_path); 2051 os_free(drv->own_socket_path); 2052 drv->own_socket_path = NULL; 2053 } 2054} 2055 2056 2057static void wpa_driver_test_deinit(void *priv) 2058{ 2059 struct test_driver_bss *dbss = priv; 2060 struct wpa_driver_test_data *drv = dbss->drv; 2061 struct test_client_socket *cli, *prev; 2062 int i; 2063 2064 cli = drv->cli; 2065 while (cli) { 2066 prev = cli; 2067 cli = cli->next; 2068 os_free(prev); 2069 } 2070 2071#ifdef HOSTAPD 2072 /* There should be only one BSS remaining at this point. */ 2073 if (dl_list_len(&drv->bss) != 1) 2074 wpa_printf(MSG_ERROR, "%s: %u remaining BSS entries", 2075 __func__, dl_list_len(&drv->bss)); 2076#endif /* HOSTAPD */ 2077 2078 test_driver_free_bsses(drv); 2079 2080 wpa_driver_test_close_test_socket(drv); 2081 eloop_cancel_timeout(wpa_driver_test_scan_timeout, drv, drv->ctx); 2082 eloop_cancel_timeout(wpa_driver_test_poll, drv, NULL); 2083 eloop_cancel_timeout(test_remain_on_channel_timeout, drv, NULL); 2084 os_free(drv->test_dir); 2085 for (i = 0; i < MAX_SCAN_RESULTS; i++) 2086 os_free(drv->scanres[i]); 2087 os_free(drv->probe_req_ie); 2088 wpa_trace_remove_ref(drv, ctx, drv->ctx); 2089 os_free(drv); 2090} 2091 2092 2093static int wpa_driver_test_attach(struct wpa_driver_test_data *drv, 2094 const char *dir, int ap) 2095{ 2096#ifdef DRIVER_TEST_UNIX 2097 static unsigned int counter = 0; 2098 struct sockaddr_un addr; 2099 size_t len; 2100 2101 os_free(drv->own_socket_path); 2102 if (dir) { 2103 len = os_strlen(dir) + 30; 2104 drv->own_socket_path = os_malloc(len); 2105 if (drv->own_socket_path == NULL) 2106 return -1; 2107 os_snprintf(drv->own_socket_path, len, "%s/%s-" MACSTR, 2108 dir, ap ? "AP" : "STA", MAC2STR(drv->own_addr)); 2109 } else { 2110 drv->own_socket_path = os_malloc(100); 2111 if (drv->own_socket_path == NULL) 2112 return -1; 2113 os_snprintf(drv->own_socket_path, 100, 2114 "/tmp/wpa_supplicant_test-%d-%d", 2115 getpid(), counter++); 2116 } 2117 2118 drv->test_socket = socket(PF_UNIX, SOCK_DGRAM, 0); 2119 if (drv->test_socket < 0) { 2120 perror("socket(PF_UNIX)"); 2121 os_free(drv->own_socket_path); 2122 drv->own_socket_path = NULL; 2123 return -1; 2124 } 2125 2126 os_memset(&addr, 0, sizeof(addr)); 2127 addr.sun_family = AF_UNIX; 2128 os_strlcpy(addr.sun_path, drv->own_socket_path, sizeof(addr.sun_path)); 2129 if (bind(drv->test_socket, (struct sockaddr *) &addr, 2130 sizeof(addr)) < 0) { 2131 perror("test-driver-attach: bind(PF_UNIX)"); 2132 close(drv->test_socket); 2133 unlink(drv->own_socket_path); 2134 os_free(drv->own_socket_path); 2135 drv->own_socket_path = NULL; 2136 return -1; 2137 } 2138 2139 eloop_register_read_sock(drv->test_socket, 2140 wpa_driver_test_receive_unix, drv, NULL); 2141 2142 return 0; 2143#else /* DRIVER_TEST_UNIX */ 2144 return -1; 2145#endif /* DRIVER_TEST_UNIX */ 2146} 2147 2148 2149static int wpa_driver_test_attach_udp(struct wpa_driver_test_data *drv, 2150 char *dst) 2151{ 2152 char *pos; 2153 2154 pos = os_strchr(dst, ':'); 2155 if (pos == NULL) 2156 return -1; 2157 *pos++ = '\0'; 2158 wpa_printf(MSG_DEBUG, "%s: addr=%s port=%s", __func__, dst, pos); 2159 2160 drv->test_socket = socket(PF_INET, SOCK_DGRAM, 0); 2161 if (drv->test_socket < 0) { 2162 perror("socket(PF_INET)"); 2163 return -1; 2164 } 2165 2166 os_memset(&drv->hostapd_addr_udp, 0, sizeof(drv->hostapd_addr_udp)); 2167 drv->hostapd_addr_udp.sin_family = AF_INET; 2168#if defined(CONFIG_NATIVE_WINDOWS) || defined(CONFIG_ANSI_C_EXTRA) 2169 { 2170 int a[4]; 2171 u8 *pos; 2172 sscanf(dst, "%d.%d.%d.%d", &a[0], &a[1], &a[2], &a[3]); 2173 pos = (u8 *) &drv->hostapd_addr_udp.sin_addr; 2174 *pos++ = a[0]; 2175 *pos++ = a[1]; 2176 *pos++ = a[2]; 2177 *pos++ = a[3]; 2178 } 2179#else /* CONFIG_NATIVE_WINDOWS or CONFIG_ANSI_C_EXTRA */ 2180 inet_aton(dst, &drv->hostapd_addr_udp.sin_addr); 2181#endif /* CONFIG_NATIVE_WINDOWS or CONFIG_ANSI_C_EXTRA */ 2182 drv->hostapd_addr_udp.sin_port = htons(atoi(pos)); 2183 2184 drv->hostapd_addr_udp_set = 1; 2185 2186 eloop_register_read_sock(drv->test_socket, 2187 wpa_driver_test_receive_unix, drv, NULL); 2188 2189 return 0; 2190} 2191 2192 2193static int wpa_driver_test_set_param(void *priv, const char *param) 2194{ 2195 struct test_driver_bss *dbss = priv; 2196 struct wpa_driver_test_data *drv = dbss->drv; 2197 const char *pos; 2198 2199 wpa_printf(MSG_DEBUG, "%s: param='%s'", __func__, param); 2200 if (param == NULL) 2201 return 0; 2202 2203 wpa_driver_test_close_test_socket(drv); 2204 2205#ifdef DRIVER_TEST_UNIX 2206 pos = os_strstr(param, "test_socket="); 2207 if (pos) { 2208 const char *pos2; 2209 size_t len; 2210 2211 pos += 12; 2212 pos2 = os_strchr(pos, ' '); 2213 if (pos2) 2214 len = pos2 - pos; 2215 else 2216 len = os_strlen(pos); 2217 if (len > sizeof(drv->hostapd_addr.sun_path)) 2218 return -1; 2219 os_memset(&drv->hostapd_addr, 0, sizeof(drv->hostapd_addr)); 2220 drv->hostapd_addr.sun_family = AF_UNIX; 2221 os_memcpy(drv->hostapd_addr.sun_path, pos, len); 2222 drv->hostapd_addr_set = 1; 2223 } 2224#endif /* DRIVER_TEST_UNIX */ 2225 2226 pos = os_strstr(param, "test_dir="); 2227 if (pos) { 2228 char *end; 2229 os_free(drv->test_dir); 2230 drv->test_dir = os_strdup(pos + 9); 2231 if (drv->test_dir == NULL) 2232 return -1; 2233 end = os_strchr(drv->test_dir, ' '); 2234 if (end) 2235 *end = '\0'; 2236 if (wpa_driver_test_attach(drv, drv->test_dir, 0)) 2237 return -1; 2238 } else { 2239 pos = os_strstr(param, "test_udp="); 2240 if (pos) { 2241 char *dst, *epos; 2242 dst = os_strdup(pos + 9); 2243 if (dst == NULL) 2244 return -1; 2245 epos = os_strchr(dst, ' '); 2246 if (epos) 2247 *epos = '\0'; 2248 if (wpa_driver_test_attach_udp(drv, dst)) 2249 return -1; 2250 os_free(dst); 2251 } else if (wpa_driver_test_attach(drv, NULL, 0)) 2252 return -1; 2253 } 2254 2255 if (os_strstr(param, "use_associnfo=1")) { 2256 wpa_printf(MSG_DEBUG, "test_driver: Use AssocInfo events"); 2257 drv->use_associnfo = 1; 2258 } 2259 2260 return 0; 2261} 2262 2263 2264static const u8 * wpa_driver_test_get_mac_addr(void *priv) 2265{ 2266 struct test_driver_bss *dbss = priv; 2267 struct wpa_driver_test_data *drv = dbss->drv; 2268 wpa_printf(MSG_DEBUG, "%s", __func__); 2269 return drv->own_addr; 2270} 2271 2272 2273static int wpa_driver_test_send_eapol(void *priv, const u8 *dest, u16 proto, 2274 const u8 *data, size_t data_len) 2275{ 2276 struct test_driver_bss *dbss = priv; 2277 struct wpa_driver_test_data *drv = dbss->drv; 2278 char *msg; 2279 size_t msg_len; 2280 struct l2_ethhdr eth; 2281 struct sockaddr *addr; 2282 socklen_t alen; 2283#ifdef DRIVER_TEST_UNIX 2284 struct sockaddr_un addr_un; 2285#endif /* DRIVER_TEST_UNIX */ 2286 2287 wpa_hexdump(MSG_MSGDUMP, "test_send_eapol TX frame", data, data_len); 2288 2289 os_memset(ð, 0, sizeof(eth)); 2290 os_memcpy(eth.h_dest, dest, ETH_ALEN); 2291 os_memcpy(eth.h_source, drv->own_addr, ETH_ALEN); 2292 eth.h_proto = host_to_be16(proto); 2293 2294 msg_len = 6 + sizeof(eth) + data_len; 2295 msg = os_malloc(msg_len); 2296 if (msg == NULL) 2297 return -1; 2298 os_memcpy(msg, "EAPOL ", 6); 2299 os_memcpy(msg + 6, ð, sizeof(eth)); 2300 os_memcpy(msg + 6 + sizeof(eth), data, data_len); 2301 2302 if (os_memcmp(dest, dbss->bssid, ETH_ALEN) == 0 || 2303 drv->test_dir == NULL) { 2304 if (drv->hostapd_addr_udp_set) { 2305 addr = (struct sockaddr *) &drv->hostapd_addr_udp; 2306 alen = sizeof(drv->hostapd_addr_udp); 2307 } else { 2308#ifdef DRIVER_TEST_UNIX 2309 addr = (struct sockaddr *) &drv->hostapd_addr; 2310 alen = sizeof(drv->hostapd_addr); 2311#else /* DRIVER_TEST_UNIX */ 2312 os_free(msg); 2313 return -1; 2314#endif /* DRIVER_TEST_UNIX */ 2315 } 2316 } else { 2317#ifdef DRIVER_TEST_UNIX 2318 struct stat st; 2319 os_memset(&addr_un, 0, sizeof(addr_un)); 2320 addr_un.sun_family = AF_UNIX; 2321 os_snprintf(addr_un.sun_path, sizeof(addr_un.sun_path), 2322 "%s/STA-" MACSTR, drv->test_dir, MAC2STR(dest)); 2323 if (stat(addr_un.sun_path, &st) < 0) { 2324 os_snprintf(addr_un.sun_path, sizeof(addr_un.sun_path), 2325 "%s/AP-" MACSTR, 2326 drv->test_dir, MAC2STR(dest)); 2327 } 2328 addr = (struct sockaddr *) &addr_un; 2329 alen = sizeof(addr_un); 2330#else /* DRIVER_TEST_UNIX */ 2331 os_free(msg); 2332 return -1; 2333#endif /* DRIVER_TEST_UNIX */ 2334 } 2335 2336 if (sendto(drv->test_socket, msg, msg_len, 0, addr, alen) < 0) { 2337 perror("sendmsg(test_socket)"); 2338 os_free(msg); 2339 return -1; 2340 } 2341 2342 os_free(msg); 2343 return 0; 2344} 2345 2346 2347static int wpa_driver_test_get_capa(void *priv, struct wpa_driver_capa *capa) 2348{ 2349 os_memset(capa, 0, sizeof(*capa)); 2350 capa->key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA | 2351 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 | 2352 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK | 2353 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK | 2354 WPA_DRIVER_CAPA_KEY_MGMT_WPA_NONE | 2355 WPA_DRIVER_CAPA_KEY_MGMT_FT | 2356 WPA_DRIVER_CAPA_KEY_MGMT_FT_PSK; 2357 capa->enc = WPA_DRIVER_CAPA_ENC_WEP40 | 2358 WPA_DRIVER_CAPA_ENC_WEP104 | 2359 WPA_DRIVER_CAPA_ENC_TKIP | 2360 WPA_DRIVER_CAPA_ENC_CCMP; 2361 capa->auth = WPA_DRIVER_AUTH_OPEN | 2362 WPA_DRIVER_AUTH_SHARED | 2363 WPA_DRIVER_AUTH_LEAP; 2364 capa->flags |= WPA_DRIVER_FLAGS_AP; 2365 capa->flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT; 2366 capa->flags |= WPA_DRIVER_FLAGS_P2P_DEDICATED_INTERFACE; 2367 capa->flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE; 2368 capa->max_scan_ssids = 2; 2369 capa->max_remain_on_chan = 60000; 2370 2371 return 0; 2372} 2373 2374 2375static int wpa_driver_test_mlme_setprotection(void *priv, const u8 *addr, 2376 int protect_type, 2377 int key_type) 2378{ 2379 wpa_printf(MSG_DEBUG, "%s: protect_type=%d key_type=%d", 2380 __func__, protect_type, key_type); 2381 2382 if (addr) { 2383 wpa_printf(MSG_DEBUG, "%s: addr=" MACSTR, 2384 __func__, MAC2STR(addr)); 2385 } 2386 2387 return 0; 2388} 2389 2390 2391static void * wpa_driver_test_global_init(void) 2392{ 2393 struct wpa_driver_test_global *global; 2394 2395 global = os_zalloc(sizeof(*global)); 2396 return global; 2397} 2398 2399 2400static void wpa_driver_test_global_deinit(void *priv) 2401{ 2402 struct wpa_driver_test_global *global = priv; 2403 os_free(global); 2404} 2405 2406 2407static struct wpa_interface_info * 2408wpa_driver_test_get_interfaces(void *global_priv) 2409{ 2410 /* struct wpa_driver_test_global *global = priv; */ 2411 struct wpa_interface_info *iface; 2412 2413 iface = os_zalloc(sizeof(*iface)); 2414 if (iface == NULL) 2415 return iface; 2416 iface->ifname = os_strdup("sta0"); 2417 iface->desc = os_strdup("test interface 0"); 2418 iface->drv_name = "test"; 2419 iface->next = os_zalloc(sizeof(*iface)); 2420 if (iface->next) { 2421 iface->next->ifname = os_strdup("sta1"); 2422 iface->next->desc = os_strdup("test interface 1"); 2423 iface->next->drv_name = "test"; 2424 } 2425 2426 return iface; 2427} 2428 2429 2430static struct hostapd_hw_modes * 2431wpa_driver_test_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags) 2432{ 2433 struct hostapd_hw_modes *modes; 2434 size_t i; 2435 2436 *num_modes = 3; 2437 *flags = 0; 2438 modes = os_calloc(*num_modes, sizeof(struct hostapd_hw_modes)); 2439 if (modes == NULL) 2440 return NULL; 2441 modes[0].mode = HOSTAPD_MODE_IEEE80211G; 2442 modes[0].num_channels = 11; 2443 modes[0].num_rates = 12; 2444 modes[0].channels = os_calloc(11, sizeof(struct hostapd_channel_data)); 2445 modes[0].rates = os_calloc(modes[0].num_rates, sizeof(int)); 2446 if (modes[0].channels == NULL || modes[0].rates == NULL) 2447 goto fail; 2448 for (i = 0; i < 11; i++) { 2449 modes[0].channels[i].chan = i + 1; 2450 modes[0].channels[i].freq = 2412 + 5 * i; 2451 modes[0].channels[i].flag = 0; 2452 } 2453 modes[0].rates[0] = 10; 2454 modes[0].rates[1] = 20; 2455 modes[0].rates[2] = 55; 2456 modes[0].rates[3] = 110; 2457 modes[0].rates[4] = 60; 2458 modes[0].rates[5] = 90; 2459 modes[0].rates[6] = 120; 2460 modes[0].rates[7] = 180; 2461 modes[0].rates[8] = 240; 2462 modes[0].rates[9] = 360; 2463 modes[0].rates[10] = 480; 2464 modes[0].rates[11] = 540; 2465 2466 modes[1].mode = HOSTAPD_MODE_IEEE80211B; 2467 modes[1].num_channels = 11; 2468 modes[1].num_rates = 4; 2469 modes[1].channels = os_calloc(11, sizeof(struct hostapd_channel_data)); 2470 modes[1].rates = os_calloc(modes[1].num_rates, sizeof(int)); 2471 if (modes[1].channels == NULL || modes[1].rates == NULL) 2472 goto fail; 2473 for (i = 0; i < 11; i++) { 2474 modes[1].channels[i].chan = i + 1; 2475 modes[1].channels[i].freq = 2412 + 5 * i; 2476 modes[1].channels[i].flag = 0; 2477 } 2478 modes[1].rates[0] = 10; 2479 modes[1].rates[1] = 20; 2480 modes[1].rates[2] = 55; 2481 modes[1].rates[3] = 110; 2482 2483 modes[2].mode = HOSTAPD_MODE_IEEE80211A; 2484 modes[2].num_channels = 1; 2485 modes[2].num_rates = 8; 2486 modes[2].channels = os_calloc(1, sizeof(struct hostapd_channel_data)); 2487 modes[2].rates = os_calloc(modes[2].num_rates, sizeof(int)); 2488 if (modes[2].channels == NULL || modes[2].rates == NULL) 2489 goto fail; 2490 modes[2].channels[0].chan = 60; 2491 modes[2].channels[0].freq = 5300; 2492 modes[2].channels[0].flag = 0; 2493 modes[2].rates[0] = 60; 2494 modes[2].rates[1] = 90; 2495 modes[2].rates[2] = 120; 2496 modes[2].rates[3] = 180; 2497 modes[2].rates[4] = 240; 2498 modes[2].rates[5] = 360; 2499 modes[2].rates[6] = 480; 2500 modes[2].rates[7] = 540; 2501 2502 return modes; 2503 2504fail: 2505 if (modes) { 2506 for (i = 0; i < *num_modes; i++) { 2507 os_free(modes[i].channels); 2508 os_free(modes[i].rates); 2509 } 2510 os_free(modes); 2511 } 2512 return NULL; 2513} 2514 2515 2516static int wpa_driver_test_set_freq(void *priv, 2517 struct hostapd_freq_params *freq) 2518{ 2519 struct test_driver_bss *dbss = priv; 2520 struct wpa_driver_test_data *drv = dbss->drv; 2521 wpa_printf(MSG_DEBUG, "test: set_freq %u MHz", freq->freq); 2522 drv->current_freq = freq->freq; 2523 return 0; 2524} 2525 2526 2527static int wpa_driver_test_send_action(void *priv, unsigned int freq, 2528 unsigned int wait, 2529 const u8 *dst, const u8 *src, 2530 const u8 *bssid, 2531 const u8 *data, size_t data_len, 2532 int no_cck) 2533{ 2534 struct test_driver_bss *dbss = priv; 2535 struct wpa_driver_test_data *drv = dbss->drv; 2536 int ret = -1; 2537 u8 *buf; 2538 struct ieee80211_hdr *hdr; 2539 2540 wpa_printf(MSG_DEBUG, "test: Send Action frame"); 2541 2542 if ((drv->remain_on_channel_freq && 2543 freq != drv->remain_on_channel_freq) || 2544 (drv->remain_on_channel_freq == 0 && 2545 freq != (unsigned int) drv->current_freq)) { 2546 wpa_printf(MSG_DEBUG, "test: Reject Action frame TX on " 2547 "unexpected channel: freq=%u MHz (current_freq=%u " 2548 "MHz, remain-on-channel freq=%u MHz)", 2549 freq, drv->current_freq, 2550 drv->remain_on_channel_freq); 2551 return -1; 2552 } 2553 2554 buf = os_zalloc(24 + data_len); 2555 if (buf == NULL) 2556 return ret; 2557 os_memcpy(buf + 24, data, data_len); 2558 hdr = (struct ieee80211_hdr *) buf; 2559 hdr->frame_control = 2560 IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_ACTION); 2561 os_memcpy(hdr->addr1, dst, ETH_ALEN); 2562 os_memcpy(hdr->addr2, src, ETH_ALEN); 2563 os_memcpy(hdr->addr3, bssid, ETH_ALEN); 2564 2565 ret = wpa_driver_test_send_mlme(priv, buf, 24 + data_len, 0); 2566 os_free(buf); 2567 return ret; 2568} 2569 2570 2571static void test_remain_on_channel_timeout(void *eloop_ctx, void *timeout_ctx) 2572{ 2573 struct wpa_driver_test_data *drv = eloop_ctx; 2574 union wpa_event_data data; 2575 2576 wpa_printf(MSG_DEBUG, "test: Remain-on-channel timeout"); 2577 2578 os_memset(&data, 0, sizeof(data)); 2579 data.remain_on_channel.freq = drv->remain_on_channel_freq; 2580 data.remain_on_channel.duration = drv->remain_on_channel_duration; 2581 2582 drv->remain_on_channel_freq = 0; 2583 2584 wpa_supplicant_event(drv->ctx, EVENT_CANCEL_REMAIN_ON_CHANNEL, &data); 2585} 2586 2587 2588static int wpa_driver_test_remain_on_channel(void *priv, unsigned int freq, 2589 unsigned int duration) 2590{ 2591 struct test_driver_bss *dbss = priv; 2592 struct wpa_driver_test_data *drv = dbss->drv; 2593 union wpa_event_data data; 2594 2595 wpa_printf(MSG_DEBUG, "%s(freq=%u, duration=%u)", 2596 __func__, freq, duration); 2597 if (drv->remain_on_channel_freq && 2598 drv->remain_on_channel_freq != freq) { 2599 wpa_printf(MSG_DEBUG, "test: Refuse concurrent " 2600 "remain_on_channel request"); 2601 return -1; 2602 } 2603 2604 drv->remain_on_channel_freq = freq; 2605 drv->remain_on_channel_duration = duration; 2606 eloop_cancel_timeout(test_remain_on_channel_timeout, drv, NULL); 2607 eloop_register_timeout(duration / 1000, (duration % 1000) * 1000, 2608 test_remain_on_channel_timeout, drv, NULL); 2609 2610 os_memset(&data, 0, sizeof(data)); 2611 data.remain_on_channel.freq = freq; 2612 data.remain_on_channel.duration = duration; 2613 wpa_supplicant_event(drv->ctx, EVENT_REMAIN_ON_CHANNEL, &data); 2614 2615 return 0; 2616} 2617 2618 2619static int wpa_driver_test_cancel_remain_on_channel(void *priv) 2620{ 2621 struct test_driver_bss *dbss = priv; 2622 struct wpa_driver_test_data *drv = dbss->drv; 2623 wpa_printf(MSG_DEBUG, "%s", __func__); 2624 if (!drv->remain_on_channel_freq) 2625 return -1; 2626 drv->remain_on_channel_freq = 0; 2627 eloop_cancel_timeout(test_remain_on_channel_timeout, drv, NULL); 2628 return 0; 2629} 2630 2631 2632static int wpa_driver_test_probe_req_report(void *priv, int report) 2633{ 2634 struct test_driver_bss *dbss = priv; 2635 struct wpa_driver_test_data *drv = dbss->drv; 2636 wpa_printf(MSG_DEBUG, "%s(report=%d)", __func__, report); 2637 drv->probe_req_report = report; 2638 return 0; 2639} 2640 2641 2642const struct wpa_driver_ops wpa_driver_test_ops = { 2643 "test", 2644 "wpa_supplicant test driver", 2645 .hapd_init = test_driver_init, 2646 .hapd_deinit = wpa_driver_test_deinit, 2647 .hapd_send_eapol = test_driver_send_eapol, 2648 .send_mlme = wpa_driver_test_send_mlme, 2649 .set_generic_elem = test_driver_set_generic_elem, 2650 .sta_deauth = test_driver_sta_deauth, 2651 .sta_disassoc = test_driver_sta_disassoc, 2652 .get_hw_feature_data = wpa_driver_test_get_hw_feature_data, 2653 .if_add = test_driver_if_add, 2654 .if_remove = test_driver_if_remove, 2655 .hapd_set_ssid = test_driver_set_ssid, 2656 .set_privacy = test_driver_set_privacy, 2657 .set_sta_vlan = test_driver_set_sta_vlan, 2658 .sta_add = test_driver_sta_add, 2659 .send_ether = test_driver_send_ether, 2660 .set_ap_wps_ie = test_driver_set_ap_wps_ie, 2661 .get_bssid = wpa_driver_test_get_bssid, 2662 .get_ssid = wpa_driver_test_get_ssid, 2663 .set_key = wpa_driver_test_set_key, 2664 .deinit = wpa_driver_test_deinit, 2665 .set_param = wpa_driver_test_set_param, 2666 .deauthenticate = wpa_driver_test_deauthenticate, 2667 .associate = wpa_driver_test_associate, 2668 .get_capa = wpa_driver_test_get_capa, 2669 .get_mac_addr = wpa_driver_test_get_mac_addr, 2670 .send_eapol = wpa_driver_test_send_eapol, 2671 .mlme_setprotection = wpa_driver_test_mlme_setprotection, 2672 .get_scan_results2 = wpa_driver_test_get_scan_results2, 2673 .global_init = wpa_driver_test_global_init, 2674 .global_deinit = wpa_driver_test_global_deinit, 2675 .init2 = wpa_driver_test_init2, 2676 .get_interfaces = wpa_driver_test_get_interfaces, 2677 .scan2 = wpa_driver_test_scan, 2678 .set_freq = wpa_driver_test_set_freq, 2679 .send_action = wpa_driver_test_send_action, 2680 .remain_on_channel = wpa_driver_test_remain_on_channel, 2681 .cancel_remain_on_channel = wpa_driver_test_cancel_remain_on_channel, 2682 .probe_req_report = wpa_driver_test_probe_req_report, 2683}; 2684