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