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