Fix netlink payload length calculation
[libeap.git] / src / drivers / driver_hostap.c
1 /*
2  * Driver interaction with Linux Host AP driver
3  * Copyright (c) 2003-2005, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "includes.h"
16 #include <sys/ioctl.h>
17
18 #include "wireless_copy.h"
19 #include "common.h"
20 #include "driver.h"
21 #include "driver_wext.h"
22 #include "eloop.h"
23 #include "driver_hostap.h"
24
25
26 #ifdef HOSTAPD
27
28 #include <net/if_arp.h>
29 #include <netpacket/packet.h>
30
31 #include "priv_netlink.h"
32 #include "common/ieee802_11_defs.h"
33
34
35 /* MTU to be set for the wlan#ap device; this is mainly needed for IEEE 802.1X
36  * frames that might be longer than normal default MTU and they are not
37  * fragmented */
38 #define HOSTAPD_MTU 2290
39
40 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
41
42 struct hostap_driver_data {
43         struct hostapd_data *hapd;
44
45         char iface[IFNAMSIZ + 1];
46         int sock; /* raw packet socket for driver access */
47         int ioctl_sock; /* socket for ioctl() use */
48         int wext_sock; /* socket for wireless events */
49
50         int we_version;
51
52         u8 *generic_ie;
53         size_t generic_ie_len;
54         u8 *wps_ie;
55         size_t wps_ie_len;
56 };
57
58
59 static int hostapd_ioctl(void *priv, struct prism2_hostapd_param *param,
60                          int len);
61 static int hostap_set_iface_flags(void *priv, int dev_up);
62
63 static void handle_data(struct hostap_driver_data *drv, u8 *buf, size_t len,
64                         u16 stype)
65 {
66         struct ieee80211_hdr *hdr;
67         u16 fc, ethertype;
68         u8 *pos, *sa;
69         size_t left;
70         union wpa_event_data event;
71
72         if (len < sizeof(struct ieee80211_hdr))
73                 return;
74
75         hdr = (struct ieee80211_hdr *) buf;
76         fc = le_to_host16(hdr->frame_control);
77
78         if ((fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) != WLAN_FC_TODS) {
79                 printf("Not ToDS data frame (fc=0x%04x)\n", fc);
80                 return;
81         }
82
83         sa = hdr->addr2;
84         os_memset(&event, 0, sizeof(event));
85         event.rx_from_unknown.hdr = hdr;
86         event.rx_from_unknown.len = len;
87         wpa_supplicant_event(drv->hapd, EVENT_RX_FROM_UNKNOWN, &event);
88
89         pos = (u8 *) (hdr + 1);
90         left = len - sizeof(*hdr);
91
92         if (left < sizeof(rfc1042_header)) {
93                 printf("Too short data frame\n");
94                 return;
95         }
96
97         if (memcmp(pos, rfc1042_header, sizeof(rfc1042_header)) != 0) {
98                 printf("Data frame with no RFC1042 header\n");
99                 return;
100         }
101         pos += sizeof(rfc1042_header);
102         left -= sizeof(rfc1042_header);
103
104         if (left < 2) {
105                 printf("No ethertype in data frame\n");
106                 return;
107         }
108
109         ethertype = WPA_GET_BE16(pos);
110         pos += 2;
111         left -= 2;
112         switch (ethertype) {
113         case ETH_P_PAE:
114                 hostapd_eapol_receive(drv->hapd, sa, pos, left);
115                 break;
116
117         default:
118                 printf("Unknown ethertype 0x%04x in data frame\n", ethertype);
119                 break;
120         }
121 }
122
123
124 static void handle_tx_callback(struct hostap_driver_data *drv, u8 *buf,
125                                size_t len, int ok)
126 {
127         struct ieee80211_hdr *hdr;
128         u16 fc;
129         union wpa_event_data event;
130
131         hdr = (struct ieee80211_hdr *) buf;
132         fc = le_to_host16(hdr->frame_control);
133
134         os_memset(&event, 0, sizeof(event));
135         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
136         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
137         event.tx_status.dst = hdr->addr1;
138         event.tx_status.data = buf;
139         event.tx_status.data_len = len;
140         event.tx_status.ack = ok;
141         wpa_supplicant_event(drv->hapd, EVENT_TX_STATUS, &event);
142 }
143
144
145 static void handle_frame(struct hostap_driver_data *drv, u8 *buf, size_t len)
146 {
147         struct ieee80211_hdr *hdr;
148         u16 fc, extra_len, type, stype;
149         unsigned char *extra = NULL;
150         size_t data_len = len;
151         int ver;
152         union wpa_event_data event;
153
154         /* PSPOLL is only 16 bytes, but driver does not (at least yet) pass
155          * these to user space */
156         if (len < 24) {
157                 wpa_printf(MSG_MSGDUMP, "handle_frame: too short (%lu)",
158                            (unsigned long) len);
159                 return;
160         }
161
162         hdr = (struct ieee80211_hdr *) buf;
163         fc = le_to_host16(hdr->frame_control);
164         type = WLAN_FC_GET_TYPE(fc);
165         stype = WLAN_FC_GET_STYPE(fc);
166
167         if (type != WLAN_FC_TYPE_MGMT || stype != WLAN_FC_STYPE_BEACON) {
168                 wpa_hexdump(MSG_MSGDUMP, "Received management frame",
169                             buf, len);
170         }
171
172         ver = fc & WLAN_FC_PVER;
173
174         /* protocol version 3 is reserved for indicating extra data after the
175          * payload, version 2 for indicating ACKed frame (TX callbacks), and
176          * version 1 for indicating failed frame (no ACK, TX callbacks) */
177         if (ver == 3) {
178                 u8 *pos = buf + len - 2;
179                 extra_len = WPA_GET_LE16(pos);
180                 printf("extra data in frame (elen=%d)\n", extra_len);
181                 if ((size_t) extra_len + 2 > len) {
182                         printf("  extra data overflow\n");
183                         return;
184                 }
185                 len -= extra_len + 2;
186                 extra = buf + len;
187         } else if (ver == 1 || ver == 2) {
188                 handle_tx_callback(drv, buf, data_len, ver == 2 ? 1 : 0);
189                 return;
190         } else if (ver != 0) {
191                 printf("unknown protocol version %d\n", ver);
192                 return;
193         }
194
195         switch (type) {
196         case WLAN_FC_TYPE_MGMT:
197                 os_memset(&event, 0, sizeof(event));
198                 event.rx_mgmt.frame = buf;
199                 event.rx_mgmt.frame_len = data_len;
200                 wpa_supplicant_event(drv->hapd, EVENT_RX_MGMT, &event);
201                 break;
202         case WLAN_FC_TYPE_CTRL:
203                 wpa_printf(MSG_DEBUG, "CTRL");
204                 break;
205         case WLAN_FC_TYPE_DATA:
206                 wpa_printf(MSG_DEBUG, "DATA");
207                 handle_data(drv, buf, data_len, stype);
208                 break;
209         default:
210                 wpa_printf(MSG_DEBUG, "unknown frame type %d", type);
211                 break;
212         }
213 }
214
215
216 static void handle_read(int sock, void *eloop_ctx, void *sock_ctx)
217 {
218         struct hostap_driver_data *drv = eloop_ctx;
219         int len;
220         unsigned char buf[3000];
221
222         len = recv(sock, buf, sizeof(buf), 0);
223         if (len < 0) {
224                 perror("recv");
225                 return;
226         }
227
228         handle_frame(drv, buf, len);
229 }
230
231
232 static int hostap_init_sockets(struct hostap_driver_data *drv, u8 *own_addr)
233 {
234         struct ifreq ifr;
235         struct sockaddr_ll addr;
236
237         drv->sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
238         if (drv->sock < 0) {
239                 perror("socket[PF_PACKET,SOCK_RAW]");
240                 return -1;
241         }
242
243         if (eloop_register_read_sock(drv->sock, handle_read, drv, NULL)) {
244                 printf("Could not register read socket\n");
245                 return -1;
246         }
247
248         memset(&ifr, 0, sizeof(ifr));
249         snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%sap", drv->iface);
250         if (ioctl(drv->sock, SIOCGIFINDEX, &ifr) != 0) {
251                 perror("ioctl(SIOCGIFINDEX)");
252                 return -1;
253         }
254
255         if (hostap_set_iface_flags(drv, 1)) {
256                 return -1;
257         }
258
259         memset(&addr, 0, sizeof(addr));
260         addr.sll_family = AF_PACKET;
261         addr.sll_ifindex = ifr.ifr_ifindex;
262         wpa_printf(MSG_DEBUG, "Opening raw packet socket for ifindex %d",
263                    addr.sll_ifindex);
264
265         if (bind(drv->sock, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
266                 perror("bind");
267                 return -1;
268         }
269
270         memset(&ifr, 0, sizeof(ifr));
271         os_strlcpy(ifr.ifr_name, drv->iface, sizeof(ifr.ifr_name));
272         if (ioctl(drv->sock, SIOCGIFHWADDR, &ifr) != 0) {
273                 perror("ioctl(SIOCGIFHWADDR)");
274                 return -1;
275         }
276
277         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
278                 printf("Invalid HW-addr family 0x%04x\n",
279                        ifr.ifr_hwaddr.sa_family);
280                 return -1;
281         }
282         os_memcpy(own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
283
284         return 0;
285 }
286
287
288 static int hostap_send_mlme(void *priv, const u8 *msg, size_t len)
289 {
290         struct hostap_driver_data *drv = priv;
291         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) msg;
292         int res;
293
294         /* Request TX callback */
295         hdr->frame_control |= host_to_le16(BIT(1));
296         res = send(drv->sock, msg, len, 0);
297         hdr->frame_control &= ~host_to_le16(BIT(1));
298
299         return res;
300 }
301
302
303 static int hostap_send_eapol(void *priv, const u8 *addr, const u8 *data,
304                              size_t data_len, int encrypt, const u8 *own_addr)
305 {
306         struct hostap_driver_data *drv = priv;
307         struct ieee80211_hdr *hdr;
308         size_t len;
309         u8 *pos;
310         int res;
311
312         len = sizeof(*hdr) + sizeof(rfc1042_header) + 2 + data_len;
313         hdr = os_zalloc(len);
314         if (hdr == NULL) {
315                 printf("malloc() failed for hostapd_send_data(len=%lu)\n",
316                        (unsigned long) len);
317                 return -1;
318         }
319
320         hdr->frame_control =
321                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
322         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
323         if (encrypt)
324                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
325         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
326         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
327         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
328
329         pos = (u8 *) (hdr + 1);
330         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
331         pos += sizeof(rfc1042_header);
332         *((u16 *) pos) = htons(ETH_P_PAE);
333         pos += 2;
334         memcpy(pos, data, data_len);
335
336         res = hostap_send_mlme(drv, (u8 *) hdr, len);
337         if (res < 0) {
338                 wpa_printf(MSG_ERROR, "hostap_send_eapol - packet len: %lu - "
339                            "failed: %d (%s)",
340                            (unsigned long) len, errno, strerror(errno));
341         }
342         free(hdr);
343
344         return res;
345 }
346
347
348 static int hostap_sta_set_flags(void *priv, const u8 *addr,
349                                 int total_flags, int flags_or, int flags_and)
350 {
351         struct hostap_driver_data *drv = priv;
352         struct prism2_hostapd_param param;
353
354         if (flags_or & WPA_STA_AUTHORIZED)
355                 flags_or = BIT(5); /* WLAN_STA_AUTHORIZED */
356         if (!(flags_and & WPA_STA_AUTHORIZED))
357                 flags_and = ~BIT(5);
358         else
359                 flags_and = ~0;
360         memset(&param, 0, sizeof(param));
361         param.cmd = PRISM2_HOSTAPD_SET_FLAGS_STA;
362         memcpy(param.sta_addr, addr, ETH_ALEN);
363         param.u.set_flags_sta.flags_or = flags_or;
364         param.u.set_flags_sta.flags_and = flags_and;
365         return hostapd_ioctl(drv, &param, sizeof(param));
366 }
367
368
369 static int hostap_set_iface_flags(void *priv, int dev_up)
370 {
371         struct hostap_driver_data *drv = priv;
372         struct ifreq ifr;
373
374         if (drv->ioctl_sock < 0)
375                 return -1;
376
377         memset(&ifr, 0, sizeof(ifr));
378         snprintf(ifr.ifr_name, IFNAMSIZ, "%sap", drv->iface);
379
380         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
381                 perror("ioctl[SIOCGIFFLAGS]");
382                 return -1;
383         }
384
385         if (dev_up)
386                 ifr.ifr_flags |= IFF_UP;
387         else
388                 ifr.ifr_flags &= ~IFF_UP;
389
390         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
391                 perror("ioctl[SIOCSIFFLAGS]");
392                 return -1;
393         }
394
395         if (dev_up) {
396                 memset(&ifr, 0, sizeof(ifr));
397                 snprintf(ifr.ifr_name, IFNAMSIZ, "%sap", drv->iface);
398                 ifr.ifr_mtu = HOSTAPD_MTU;
399                 if (ioctl(drv->ioctl_sock, SIOCSIFMTU, &ifr) != 0) {
400                         perror("ioctl[SIOCSIFMTU]");
401                         printf("Setting MTU failed - trying to survive with "
402                                "current value\n");
403                 }
404         }
405
406         return 0;
407 }
408
409
410 static int hostapd_ioctl(void *priv, struct prism2_hostapd_param *param,
411                          int len)
412 {
413         struct hostap_driver_data *drv = priv;
414         struct iwreq iwr;
415
416         memset(&iwr, 0, sizeof(iwr));
417         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
418         iwr.u.data.pointer = (caddr_t) param;
419         iwr.u.data.length = len;
420
421         if (ioctl(drv->ioctl_sock, PRISM2_IOCTL_HOSTAPD, &iwr) < 0) {
422                 perror("ioctl[PRISM2_IOCTL_HOSTAPD]");
423                 return -1;
424         }
425
426         return 0;
427 }
428
429
430 static int wpa_driver_hostap_set_key(const char *ifname, void *priv,
431                                      wpa_alg alg, const u8 *addr, int key_idx,
432                                      int set_tx, const u8 *seq, size_t seq_len,
433                                      const u8 *key, size_t key_len)
434 {
435         struct hostap_driver_data *drv = priv;
436         struct prism2_hostapd_param *param;
437         u8 *buf;
438         size_t blen;
439         int ret = 0;
440
441         blen = sizeof(*param) + key_len;
442         buf = os_zalloc(blen);
443         if (buf == NULL)
444                 return -1;
445
446         param = (struct prism2_hostapd_param *) buf;
447         param->cmd = PRISM2_SET_ENCRYPTION;
448         if (addr == NULL)
449                 memset(param->sta_addr, 0xff, ETH_ALEN);
450         else
451                 memcpy(param->sta_addr, addr, ETH_ALEN);
452         switch (alg) {
453         case WPA_ALG_NONE:
454                 os_strlcpy((char *) param->u.crypt.alg, "NONE",
455                            HOSTAP_CRYPT_ALG_NAME_LEN);
456                 break;
457         case WPA_ALG_WEP:
458                 os_strlcpy((char *) param->u.crypt.alg, "WEP",
459                            HOSTAP_CRYPT_ALG_NAME_LEN);
460                 break;
461         case WPA_ALG_TKIP:
462                 os_strlcpy((char *) param->u.crypt.alg, "TKIP",
463                            HOSTAP_CRYPT_ALG_NAME_LEN);
464                 break;
465         case WPA_ALG_CCMP:
466                 os_strlcpy((char *) param->u.crypt.alg, "CCMP",
467                            HOSTAP_CRYPT_ALG_NAME_LEN);
468                 break;
469         default:
470                 os_free(buf);
471                 return -1;
472         }
473         param->u.crypt.flags = set_tx ? HOSTAP_CRYPT_FLAG_SET_TX_KEY : 0;
474         param->u.crypt.idx = key_idx;
475         param->u.crypt.key_len = key_len;
476         memcpy((u8 *) (param + 1), key, key_len);
477
478         if (hostapd_ioctl(drv, param, blen)) {
479                 printf("Failed to set encryption.\n");
480                 ret = -1;
481         }
482         free(buf);
483
484         return ret;
485 }
486
487
488 static int hostap_get_seqnum(const char *ifname, void *priv, const u8 *addr,
489                              int idx, u8 *seq)
490 {
491         struct hostap_driver_data *drv = priv;
492         struct prism2_hostapd_param *param;
493         u8 *buf;
494         size_t blen;
495         int ret = 0;
496
497         blen = sizeof(*param) + 32;
498         buf = os_zalloc(blen);
499         if (buf == NULL)
500                 return -1;
501
502         param = (struct prism2_hostapd_param *) buf;
503         param->cmd = PRISM2_GET_ENCRYPTION;
504         if (addr == NULL)
505                 memset(param->sta_addr, 0xff, ETH_ALEN);
506         else
507                 memcpy(param->sta_addr, addr, ETH_ALEN);
508         param->u.crypt.idx = idx;
509
510         if (hostapd_ioctl(drv, param, blen)) {
511                 printf("Failed to get encryption.\n");
512                 ret = -1;
513         } else {
514                 memcpy(seq, param->u.crypt.seq, 8);
515         }
516         free(buf);
517
518         return ret;
519 }
520
521
522 static int hostap_ioctl_prism2param(void *priv, int param, int value)
523 {
524         struct hostap_driver_data *drv = priv;
525         struct iwreq iwr;
526         int *i;
527
528         memset(&iwr, 0, sizeof(iwr));
529         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
530         i = (int *) iwr.u.name;
531         *i++ = param;
532         *i++ = value;
533
534         if (ioctl(drv->ioctl_sock, PRISM2_IOCTL_PRISM2_PARAM, &iwr) < 0) {
535                 perror("ioctl[PRISM2_IOCTL_PRISM2_PARAM]");
536                 return -1;
537         }
538
539         return 0;
540 }
541
542
543 static int hostap_set_ieee8021x(void *priv, struct wpa_bss_params *params)
544 {
545         struct hostap_driver_data *drv = priv;
546         int enabled = params->enabled;
547
548         /* enable kernel driver support for IEEE 802.1X */
549         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_IEEE_802_1X, enabled)) {
550                 printf("Could not setup IEEE 802.1X support in kernel driver."
551                        "\n");
552                 return -1;
553         }
554
555         if (!enabled)
556                 return 0;
557
558         /* use host driver implementation of encryption to allow
559          * individual keys and passing plaintext EAPOL frames */
560         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOST_DECRYPT, 1) ||
561             hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOST_ENCRYPT, 1)) {
562                 printf("Could not setup host-based encryption in kernel "
563                        "driver.\n");
564                 return -1;
565         }
566
567         return 0;
568 }
569
570
571 static int hostap_set_privacy(const char *ifname, void *priv, int enabled)
572 {
573         struct hostap_drvier_data *drv = priv;
574
575         return hostap_ioctl_prism2param(drv, PRISM2_PARAM_PRIVACY_INVOKED,
576                                         enabled);
577 }
578
579
580 static int hostap_set_ssid(const char *ifname, void *priv, const u8 *buf,
581                            int len)
582 {
583         struct hostap_driver_data *drv = priv;
584         struct iwreq iwr;
585
586         memset(&iwr, 0, sizeof(iwr));
587         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
588         iwr.u.essid.flags = 1; /* SSID active */
589         iwr.u.essid.pointer = (caddr_t) buf;
590         iwr.u.essid.length = len + 1;
591
592         if (ioctl(drv->ioctl_sock, SIOCSIWESSID, &iwr) < 0) {
593                 perror("ioctl[SIOCSIWESSID]");
594                 printf("len=%d\n", len);
595                 return -1;
596         }
597
598         return 0;
599 }
600
601
602 static int hostap_flush(void *priv)
603 {
604         struct hostap_driver_data *drv = priv;
605         struct prism2_hostapd_param param;
606
607         memset(&param, 0, sizeof(param));
608         param.cmd = PRISM2_HOSTAPD_FLUSH;
609         return hostapd_ioctl(drv, &param, sizeof(param));
610 }
611
612
613 static int hostap_read_sta_data(void *priv,
614                                 struct hostap_sta_driver_data *data,
615                                 const u8 *addr)
616 {
617         struct hostap_driver_data *drv = priv;
618         char buf[1024], line[128], *pos;
619         FILE *f;
620         unsigned long val;
621
622         memset(data, 0, sizeof(*data));
623         snprintf(buf, sizeof(buf), "/proc/net/hostap/%s/" MACSTR,
624                  drv->iface, MAC2STR(addr));
625
626         f = fopen(buf, "r");
627         if (!f)
628                 return -1;
629         /* Need to read proc file with in one piece, so use large enough
630          * buffer. */
631         setbuffer(f, buf, sizeof(buf));
632
633         while (fgets(line, sizeof(line), f)) {
634                 pos = strchr(line, '=');
635                 if (!pos)
636                         continue;
637                 *pos++ = '\0';
638                 val = strtoul(pos, NULL, 10);
639                 if (strcmp(line, "rx_packets") == 0)
640                         data->rx_packets = val;
641                 else if (strcmp(line, "tx_packets") == 0)
642                         data->tx_packets = val;
643                 else if (strcmp(line, "rx_bytes") == 0)
644                         data->rx_bytes = val;
645                 else if (strcmp(line, "tx_bytes") == 0)
646                         data->tx_bytes = val;
647         }
648
649         fclose(f);
650
651         return 0;
652 }
653
654
655 static int hostap_sta_add(const char *ifname, void *priv,
656                           struct hostapd_sta_add_params *params)
657 {
658         struct hostap_driver_data *drv = priv;
659         struct prism2_hostapd_param param;
660         int tx_supp_rates = 0;
661         size_t i;
662
663 #define WLAN_RATE_1M BIT(0)
664 #define WLAN_RATE_2M BIT(1)
665 #define WLAN_RATE_5M5 BIT(2)
666 #define WLAN_RATE_11M BIT(3)
667
668         for (i = 0; i < params->supp_rates_len; i++) {
669                 if ((params->supp_rates[i] & 0x7f) == 2)
670                         tx_supp_rates |= WLAN_RATE_1M;
671                 if ((params->supp_rates[i] & 0x7f) == 4)
672                         tx_supp_rates |= WLAN_RATE_2M;
673                 if ((params->supp_rates[i] & 0x7f) == 11)
674                         tx_supp_rates |= WLAN_RATE_5M5;
675                 if ((params->supp_rates[i] & 0x7f) == 22)
676                         tx_supp_rates |= WLAN_RATE_11M;
677         }
678
679         memset(&param, 0, sizeof(param));
680         param.cmd = PRISM2_HOSTAPD_ADD_STA;
681         memcpy(param.sta_addr, params->addr, ETH_ALEN);
682         param.u.add_sta.aid = params->aid;
683         param.u.add_sta.capability = params->capability;
684         param.u.add_sta.tx_supp_rates = tx_supp_rates;
685         return hostapd_ioctl(drv, &param, sizeof(param));
686 }
687
688
689 static int hostap_sta_remove(void *priv, const u8 *addr)
690 {
691         struct hostap_driver_data *drv = priv;
692         struct prism2_hostapd_param param;
693
694         hostap_sta_set_flags(drv, addr, 0, 0, ~WPA_STA_AUTHORIZED);
695
696         memset(&param, 0, sizeof(param));
697         param.cmd = PRISM2_HOSTAPD_REMOVE_STA;
698         memcpy(param.sta_addr, addr, ETH_ALEN);
699         if (hostapd_ioctl(drv, &param, sizeof(param))) {
700                 printf("Could not remove station from kernel driver.\n");
701                 return -1;
702         }
703         return 0;
704 }
705
706
707 static int hostap_get_inact_sec(void *priv, const u8 *addr)
708 {
709         struct hostap_driver_data *drv = priv;
710         struct prism2_hostapd_param param;
711
712         memset(&param, 0, sizeof(param));
713         param.cmd = PRISM2_HOSTAPD_GET_INFO_STA;
714         memcpy(param.sta_addr, addr, ETH_ALEN);
715         if (hostapd_ioctl(drv, &param, sizeof(param))) {
716                 return -1;
717         }
718
719         return param.u.get_info_sta.inactive_sec;
720 }
721
722
723 static int hostap_sta_clear_stats(void *priv, const u8 *addr)
724 {
725         struct hostap_driver_data *drv = priv;
726         struct prism2_hostapd_param param;
727
728         memset(&param, 0, sizeof(param));
729         param.cmd = PRISM2_HOSTAPD_STA_CLEAR_STATS;
730         memcpy(param.sta_addr, addr, ETH_ALEN);
731         if (hostapd_ioctl(drv, &param, sizeof(param))) {
732                 return -1;
733         }
734
735         return 0;
736 }
737
738
739 static int hostapd_ioctl_set_generic_elem(struct hostap_driver_data *drv)
740 {
741         struct prism2_hostapd_param *param;
742         int res;
743         size_t blen, elem_len;
744
745         elem_len = drv->generic_ie_len + drv->wps_ie_len;
746         blen = PRISM2_HOSTAPD_GENERIC_ELEMENT_HDR_LEN + elem_len;
747         if (blen < sizeof(*param))
748                 blen = sizeof(*param);
749
750         param = os_zalloc(blen);
751         if (param == NULL)
752                 return -1;
753
754         param->cmd = PRISM2_HOSTAPD_SET_GENERIC_ELEMENT;
755         param->u.generic_elem.len = elem_len;
756         if (drv->generic_ie) {
757                 os_memcpy(param->u.generic_elem.data, drv->generic_ie,
758                           drv->generic_ie_len);
759         }
760         if (drv->wps_ie) {
761                 os_memcpy(&param->u.generic_elem.data[drv->generic_ie_len],
762                           drv->wps_ie, drv->wps_ie_len);
763         }
764         wpa_hexdump(MSG_DEBUG, "hostap: Set generic IE",
765                     param->u.generic_elem.data, elem_len);
766         res = hostapd_ioctl(drv, param, blen);
767
768         os_free(param);
769
770         return res;
771 }
772
773
774 static int hostap_set_generic_elem(const char *ifname, void *priv,
775                                    const u8 *elem, size_t elem_len)
776 {
777         struct hostap_driver_data *drv = priv;
778
779         os_free(drv->generic_ie);
780         drv->generic_ie = NULL;
781         drv->generic_ie_len = 0;
782         if (elem) {
783                 drv->generic_ie = os_malloc(elem_len);
784                 if (drv->generic_ie == NULL)
785                         return -1;
786                 os_memcpy(drv->generic_ie, elem, elem_len);
787                 drv->generic_ie_len = elem_len;
788         }
789
790         return hostapd_ioctl_set_generic_elem(drv);
791 }
792
793
794 static int hostap_set_wps_beacon_ie(const char *ifname, void *priv,
795                                     const u8 *ie, size_t len)
796 {
797         /* Host AP driver supports only one set of extra IEs, so we need to
798          * use the ProbeResp IEs also for Beacon frames since they include more
799          * information. */
800         return 0;
801 }
802
803
804 static int hostap_set_wps_probe_resp_ie(const char *ifname, void *priv,
805                                         const u8 *ie, size_t len)
806 {
807         struct hostap_driver_data *drv = priv;
808
809         os_free(drv->wps_ie);
810         drv->wps_ie = NULL;
811         drv->wps_ie_len = 0;
812         if (ie) {
813                 drv->wps_ie = os_malloc(len);
814                 if (drv->wps_ie == NULL)
815                         return -1;
816                 os_memcpy(drv->wps_ie, ie, len);
817                 drv->wps_ie_len = len;
818         }
819
820         return hostapd_ioctl_set_generic_elem(drv);
821 }
822
823
824 static void
825 hostapd_wireless_event_wireless_custom(struct hostap_driver_data *drv,
826                                        char *custom)
827 {
828         wpa_printf(MSG_DEBUG, "Custom wireless event: '%s'", custom);
829
830         if (strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
831                 char *pos;
832                 u8 addr[ETH_ALEN];
833                 pos = strstr(custom, "addr=");
834                 if (pos == NULL) {
835                         wpa_printf(MSG_DEBUG,
836                                    "MLME-MICHAELMICFAILURE.indication "
837                                    "without sender address ignored");
838                         return;
839                 }
840                 pos += 5;
841                 if (hwaddr_aton(pos, addr) == 0) {
842                         union wpa_event_data data;
843                         os_memset(&data, 0, sizeof(data));
844                         data.michael_mic_failure.unicast = 1;
845                         data.michael_mic_failure.src = addr;
846                         wpa_supplicant_event(drv->hapd,
847                                              EVENT_MICHAEL_MIC_FAILURE, &data);
848                 } else {
849                         wpa_printf(MSG_DEBUG,
850                                    "MLME-MICHAELMICFAILURE.indication "
851                                    "with invalid MAC address");
852                 }
853         }
854 }
855
856
857 static void hostapd_wireless_event_wireless(struct hostap_driver_data *drv,
858                                             char *data, int len)
859 {
860         struct iw_event iwe_buf, *iwe = &iwe_buf;
861         char *pos, *end, *custom, *buf;
862
863         pos = data;
864         end = data + len;
865
866         while (pos + IW_EV_LCP_LEN <= end) {
867                 /* Event data may be unaligned, so make a local, aligned copy
868                  * before processing. */
869                 memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
870                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
871                            iwe->cmd, iwe->len);
872                 if (iwe->len <= IW_EV_LCP_LEN)
873                         return;
874
875                 custom = pos + IW_EV_POINT_LEN;
876                 if (drv->we_version > 18 &&
877                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
878                      iwe->cmd == IWEVCUSTOM)) {
879                         /* WE-19 removed the pointer from struct iw_point */
880                         char *dpos = (char *) &iwe_buf.u.data.length;
881                         int dlen = dpos - (char *) &iwe_buf;
882                         memcpy(dpos, pos + IW_EV_LCP_LEN,
883                                sizeof(struct iw_event) - dlen);
884                 } else {
885                         memcpy(&iwe_buf, pos, sizeof(struct iw_event));
886                         custom += IW_EV_POINT_OFF;
887                 }
888
889                 switch (iwe->cmd) {
890                 case IWEVCUSTOM:
891                         if (custom + iwe->u.data.length > end)
892                                 return;
893                         buf = malloc(iwe->u.data.length + 1);
894                         if (buf == NULL)
895                                 return;
896                         memcpy(buf, custom, iwe->u.data.length);
897                         buf[iwe->u.data.length] = '\0';
898                         hostapd_wireless_event_wireless_custom(drv, buf);
899                         free(buf);
900                         break;
901                 }
902
903                 pos += iwe->len;
904         }
905 }
906
907
908 static void hostapd_wireless_event_rtm_newlink(struct hostap_driver_data *drv,
909                                                struct nlmsghdr *h, int len)
910 {
911         struct ifinfomsg *ifi;
912         int attrlen, nlmsg_len, rta_len;
913         struct rtattr * attr;
914
915         if (len < (int) sizeof(*ifi))
916                 return;
917
918         ifi = NLMSG_DATA(h);
919
920         /* TODO: use ifi->ifi_index to filter out wireless events from other
921          * interfaces */
922
923         nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
924
925         attrlen = NLMSG_PAYLOAD(h, sizeof(struct ifinfomsg));
926         if (attrlen < 0)
927                 return;
928
929         attr = (struct rtattr *) (((char *) ifi) + nlmsg_len);
930
931         rta_len = RTA_ALIGN(sizeof(struct rtattr));
932         while (RTA_OK(attr, attrlen)) {
933                 if (attr->rta_type == IFLA_WIRELESS) {
934                         hostapd_wireless_event_wireless(
935                                 drv, ((char *) attr) + rta_len,
936                                 attr->rta_len - rta_len);
937                 }
938                 attr = RTA_NEXT(attr, attrlen);
939         }
940 }
941
942
943 static void hostapd_wireless_event_receive(int sock, void *eloop_ctx,
944                                            void *sock_ctx)
945 {
946         char buf[256];
947         int left;
948         struct sockaddr_nl from;
949         socklen_t fromlen;
950         struct nlmsghdr *h;
951         struct hostap_driver_data *drv = eloop_ctx;
952
953         fromlen = sizeof(from);
954         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
955                         (struct sockaddr *) &from, &fromlen);
956         if (left < 0) {
957                 if (errno != EINTR && errno != EAGAIN)
958                         perror("recvfrom(netlink)");
959                 return;
960         }
961
962         h = (struct nlmsghdr *) buf;
963         while (left >= (int) sizeof(*h)) {
964                 int len, plen;
965
966                 len = h->nlmsg_len;
967                 plen = len - sizeof(*h);
968                 if (len > left || plen < 0) {
969                         printf("Malformed netlink message: "
970                                "len=%d left=%d plen=%d\n",
971                                len, left, plen);
972                         break;
973                 }
974
975                 switch (h->nlmsg_type) {
976                 case RTM_NEWLINK:
977                         hostapd_wireless_event_rtm_newlink(drv, h, plen);
978                         break;
979                 }
980
981                 len = NLMSG_ALIGN(len);
982                 left -= len;
983                 h = (struct nlmsghdr *) ((char *) h + len);
984         }
985
986         if (left > 0) {
987                 printf("%d extra bytes in the end of netlink message\n", left);
988         }
989 }
990
991
992 static int hostap_get_we_version(struct hostap_driver_data *drv)
993 {
994         struct iw_range *range;
995         struct iwreq iwr;
996         int minlen;
997         size_t buflen;
998
999         drv->we_version = 0;
1000
1001         /*
1002          * Use larger buffer than struct iw_range in order to allow the
1003          * structure to grow in the future.
1004          */
1005         buflen = sizeof(struct iw_range) + 500;
1006         range = os_zalloc(buflen);
1007         if (range == NULL)
1008                 return -1;
1009
1010         memset(&iwr, 0, sizeof(iwr));
1011         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
1012         iwr.u.data.pointer = (caddr_t) range;
1013         iwr.u.data.length = buflen;
1014
1015         minlen = ((char *) &range->enc_capa) - (char *) range +
1016                 sizeof(range->enc_capa);
1017
1018         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
1019                 perror("ioctl[SIOCGIWRANGE]");
1020                 free(range);
1021                 return -1;
1022         } else if (iwr.u.data.length >= minlen &&
1023                    range->we_version_compiled >= 18) {
1024                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
1025                            "WE(source)=%d enc_capa=0x%x",
1026                            range->we_version_compiled,
1027                            range->we_version_source,
1028                            range->enc_capa);
1029                 drv->we_version = range->we_version_compiled;
1030         }
1031
1032         free(range);
1033         return 0;
1034 }
1035
1036
1037 static int hostap_wireless_event_init(struct hostap_driver_data *drv)
1038 {
1039         int s;
1040         struct sockaddr_nl local;
1041
1042         hostap_get_we_version(drv);
1043
1044         drv->wext_sock = -1;
1045
1046         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1047         if (s < 0) {
1048                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1049                 return -1;
1050         }
1051
1052         memset(&local, 0, sizeof(local));
1053         local.nl_family = AF_NETLINK;
1054         local.nl_groups = RTMGRP_LINK;
1055         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1056                 perror("bind(netlink)");
1057                 close(s);
1058                 return -1;
1059         }
1060
1061         eloop_register_read_sock(s, hostapd_wireless_event_receive, drv,
1062                                  NULL);
1063         drv->wext_sock = s;
1064
1065         return 0;
1066 }
1067
1068
1069 static void hostap_wireless_event_deinit(struct hostap_driver_data *drv)
1070 {
1071         if (drv->wext_sock < 0)
1072                 return;
1073         eloop_unregister_read_sock(drv->wext_sock);
1074         close(drv->wext_sock);
1075 }
1076
1077
1078 static void * hostap_init(struct hostapd_data *hapd,
1079                           struct wpa_init_params *params)
1080 {
1081         struct hostap_driver_data *drv;
1082
1083         drv = os_zalloc(sizeof(struct hostap_driver_data));
1084         if (drv == NULL) {
1085                 printf("Could not allocate memory for hostapd driver data\n");
1086                 return NULL;
1087         }
1088
1089         drv->hapd = hapd;
1090         drv->ioctl_sock = drv->sock = -1;
1091         memcpy(drv->iface, params->ifname, sizeof(drv->iface));
1092
1093         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1094         if (drv->ioctl_sock < 0) {
1095                 perror("socket[PF_INET,SOCK_DGRAM]");
1096                 free(drv);
1097                 return NULL;
1098         }
1099
1100         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD, 1)) {
1101                 printf("Could not enable hostapd mode for interface %s\n",
1102                        drv->iface);
1103                 close(drv->ioctl_sock);
1104                 free(drv);
1105                 return NULL;
1106         }
1107
1108         if (hostap_init_sockets(drv, params->own_addr) ||
1109             hostap_wireless_event_init(drv)) {
1110                 close(drv->ioctl_sock);
1111                 free(drv);
1112                 return NULL;
1113         }
1114
1115         return drv;
1116 }
1117
1118
1119 static void hostap_driver_deinit(void *priv)
1120 {
1121         struct hostap_driver_data *drv = priv;
1122
1123         hostap_wireless_event_deinit(drv);
1124         (void) hostap_set_iface_flags(drv, 0);
1125         (void) hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD, 0);
1126         (void) hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD_STA, 0);
1127
1128         if (drv->ioctl_sock >= 0)
1129                 close(drv->ioctl_sock);
1130
1131         if (drv->sock >= 0)
1132                 close(drv->sock);
1133
1134         os_free(drv->generic_ie);
1135         os_free(drv->wps_ie);
1136
1137         free(drv);
1138 }
1139
1140
1141 static int hostap_sta_deauth(void *priv, const u8 *own_addr, const u8 *addr,
1142                              int reason)
1143 {
1144         struct hostap_driver_data *drv = priv;
1145         struct ieee80211_mgmt mgmt;
1146
1147         memset(&mgmt, 0, sizeof(mgmt));
1148         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1149                                           WLAN_FC_STYPE_DEAUTH);
1150         memcpy(mgmt.da, addr, ETH_ALEN);
1151         memcpy(mgmt.sa, own_addr, ETH_ALEN);
1152         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
1153         mgmt.u.deauth.reason_code = host_to_le16(reason);
1154         return hostap_send_mlme(drv, (u8 *) &mgmt, IEEE80211_HDRLEN +
1155                                 sizeof(mgmt.u.deauth));
1156 }
1157
1158
1159 static int hostap_sta_disassoc(void *priv, const u8 *own_addr, const u8 *addr,
1160                                int reason)
1161 {
1162         struct hostap_driver_data *drv = priv;
1163         struct ieee80211_mgmt mgmt;
1164
1165         memset(&mgmt, 0, sizeof(mgmt));
1166         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1167                                           WLAN_FC_STYPE_DISASSOC);
1168         memcpy(mgmt.da, addr, ETH_ALEN);
1169         memcpy(mgmt.sa, own_addr, ETH_ALEN);
1170         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
1171         mgmt.u.disassoc.reason_code = host_to_le16(reason);
1172         return  hostap_send_mlme(drv, (u8 *) &mgmt, IEEE80211_HDRLEN +
1173                                  sizeof(mgmt.u.disassoc));
1174 }
1175
1176
1177 static struct hostapd_hw_modes * hostap_get_hw_feature_data(void *priv,
1178                                                             u16 *num_modes,
1179                                                             u16 *flags)
1180 {
1181         struct hostapd_hw_modes *mode;
1182         int i, clen, rlen;
1183         const short chan2freq[14] = {
1184                 2412, 2417, 2422, 2427, 2432, 2437, 2442,
1185                 2447, 2452, 2457, 2462, 2467, 2472, 2484
1186         };
1187
1188         mode = os_zalloc(sizeof(struct hostapd_hw_modes));
1189         if (mode == NULL)
1190                 return NULL;
1191
1192         *num_modes = 1;
1193         *flags = 0;
1194
1195         mode->mode = HOSTAPD_MODE_IEEE80211B;
1196         mode->num_channels = 14;
1197         mode->num_rates = 4;
1198
1199         clen = mode->num_channels * sizeof(struct hostapd_channel_data);
1200         rlen = mode->num_rates * sizeof(int);
1201
1202         mode->channels = os_zalloc(clen);
1203         mode->rates = os_zalloc(rlen);
1204         if (mode->channels == NULL || mode->rates == NULL) {
1205                 os_free(mode->channels);
1206                 os_free(mode->rates);
1207                 os_free(mode);
1208                 return NULL;
1209         }
1210
1211         for (i = 0; i < 14; i++) {
1212                 mode->channels[i].chan = i + 1;
1213                 mode->channels[i].freq = chan2freq[i];
1214                 /* TODO: Get allowed channel list from the driver */
1215                 if (i >= 11)
1216                         mode->channels[i].flag = HOSTAPD_CHAN_DISABLED;
1217         }
1218
1219         mode->rates[0] = 10;
1220         mode->rates[1] = 20;
1221         mode->rates[2] = 55;
1222         mode->rates[3] = 110;
1223
1224         return mode;
1225 }
1226
1227 #else /* HOSTAPD */
1228
1229 struct wpa_driver_hostap_data {
1230         void *wext; /* private data for driver_wext */
1231         void *ctx;
1232         char ifname[IFNAMSIZ + 1];
1233         int sock;
1234         int current_mode; /* infra/adhoc */
1235 };
1236
1237
1238 static int wpa_driver_hostap_set_auth_alg(void *priv, int auth_alg);
1239
1240
1241 static int hostapd_ioctl(struct wpa_driver_hostap_data *drv,
1242                          struct prism2_hostapd_param *param,
1243                          int len, int show_err)
1244 {
1245         struct iwreq iwr;
1246
1247         os_memset(&iwr, 0, sizeof(iwr));
1248         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1249         iwr.u.data.pointer = (caddr_t) param;
1250         iwr.u.data.length = len;
1251
1252         if (ioctl(drv->sock, PRISM2_IOCTL_HOSTAPD, &iwr) < 0) {
1253                 int ret = errno;
1254                 if (show_err)
1255                         perror("ioctl[PRISM2_IOCTL_HOSTAPD]");
1256                 return ret;
1257         }
1258
1259         return 0;
1260 }
1261
1262
1263 static int wpa_driver_hostap_set_wpa_ie(struct wpa_driver_hostap_data *drv,
1264                                         const u8 *wpa_ie, size_t wpa_ie_len)
1265 {
1266         struct prism2_hostapd_param *param;
1267         int res;
1268         size_t blen = PRISM2_HOSTAPD_GENERIC_ELEMENT_HDR_LEN + wpa_ie_len;
1269         if (blen < sizeof(*param))
1270                 blen = sizeof(*param);
1271
1272         param = os_zalloc(blen);
1273         if (param == NULL)
1274                 return -1;
1275
1276         param->cmd = PRISM2_HOSTAPD_SET_GENERIC_ELEMENT;
1277         param->u.generic_elem.len = wpa_ie_len;
1278         os_memcpy(param->u.generic_elem.data, wpa_ie, wpa_ie_len);
1279         res = hostapd_ioctl(drv, param, blen, 1);
1280
1281         os_free(param);
1282
1283         return res;
1284 }
1285
1286
1287 static int prism2param(struct wpa_driver_hostap_data *drv, int param,
1288                        int value)
1289 {
1290         struct iwreq iwr;
1291         int *i, ret = 0;
1292
1293         os_memset(&iwr, 0, sizeof(iwr));
1294         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1295         i = (int *) iwr.u.name;
1296         *i++ = param;
1297         *i++ = value;
1298
1299         if (ioctl(drv->sock, PRISM2_IOCTL_PRISM2_PARAM, &iwr) < 0) {
1300                 perror("ioctl[PRISM2_IOCTL_PRISM2_PARAM]");
1301                 ret = -1;
1302         }
1303         return ret;
1304 }
1305
1306
1307 static int wpa_driver_hostap_set_wpa(void *priv, int enabled)
1308 {
1309         struct wpa_driver_hostap_data *drv = priv;
1310         int ret = 0;
1311
1312         wpa_printf(MSG_DEBUG, "%s: enabled=%d", __FUNCTION__, enabled);
1313
1314         if (!enabled && wpa_driver_hostap_set_wpa_ie(drv, NULL, 0) < 0)
1315                 ret = -1;
1316         if (prism2param(drv, PRISM2_PARAM_HOST_ROAMING, enabled ? 2 : 0) < 0)
1317                 ret = -1;
1318         if (prism2param(drv, PRISM2_PARAM_WPA, enabled) < 0)
1319                 ret = -1;
1320
1321         return ret;
1322 }
1323
1324
1325 static void show_set_key_error(struct prism2_hostapd_param *param)
1326 {
1327         switch (param->u.crypt.err) {
1328         case HOSTAP_CRYPT_ERR_UNKNOWN_ALG:
1329                 wpa_printf(MSG_INFO, "Unknown algorithm '%s'.",
1330                            param->u.crypt.alg);
1331                 wpa_printf(MSG_INFO, "You may need to load kernel module to "
1332                            "register that algorithm.");
1333                 wpa_printf(MSG_INFO, "E.g., 'modprobe hostap_crypt_wep' for "
1334                            "WEP.");
1335                 break;
1336         case HOSTAP_CRYPT_ERR_UNKNOWN_ADDR:
1337                 wpa_printf(MSG_INFO, "Unknown address " MACSTR ".",
1338                            MAC2STR(param->sta_addr));
1339                 break;
1340         case HOSTAP_CRYPT_ERR_CRYPT_INIT_FAILED:
1341                 wpa_printf(MSG_INFO, "Crypt algorithm initialization failed.");
1342                 break;
1343         case HOSTAP_CRYPT_ERR_KEY_SET_FAILED:
1344                 wpa_printf(MSG_INFO, "Key setting failed.");
1345                 break;
1346         case HOSTAP_CRYPT_ERR_TX_KEY_SET_FAILED:
1347                 wpa_printf(MSG_INFO, "TX key index setting failed.");
1348                 break;
1349         case HOSTAP_CRYPT_ERR_CARD_CONF_FAILED:
1350                 wpa_printf(MSG_INFO, "Card configuration failed.");
1351                 break;
1352         }
1353 }
1354
1355
1356 static int wpa_driver_hostap_set_key(const char *ifname, void *priv,
1357                                      wpa_alg alg, const u8 *addr, int key_idx,
1358                                      int set_tx, const u8 *seq, size_t seq_len,
1359                                      const u8 *key, size_t key_len)
1360 {
1361         struct wpa_driver_hostap_data *drv = priv;
1362         struct prism2_hostapd_param *param;
1363         u8 *buf;
1364         size_t blen;
1365         int ret = 0;
1366         char *alg_name;
1367
1368         switch (alg) {
1369         case WPA_ALG_NONE:
1370                 alg_name = "none";
1371                 break;
1372         case WPA_ALG_WEP:
1373                 alg_name = "WEP";
1374                 break;
1375         case WPA_ALG_TKIP:
1376                 alg_name = "TKIP";
1377                 break;
1378         case WPA_ALG_CCMP:
1379                 alg_name = "CCMP";
1380                 break;
1381         default:
1382                 return -1;
1383         }
1384
1385         wpa_printf(MSG_DEBUG, "%s: alg=%s key_idx=%d set_tx=%d seq_len=%lu "
1386                    "key_len=%lu", __FUNCTION__, alg_name, key_idx, set_tx,
1387                    (unsigned long) seq_len, (unsigned long) key_len);
1388
1389         if (seq_len > 8)
1390                 return -2;
1391
1392         blen = sizeof(*param) + key_len;
1393         buf = os_zalloc(blen);
1394         if (buf == NULL)
1395                 return -1;
1396
1397         param = (struct prism2_hostapd_param *) buf;
1398         param->cmd = PRISM2_SET_ENCRYPTION;
1399         /* TODO: In theory, STA in client mode can use five keys; four default
1400          * keys for receiving (with keyidx 0..3) and one individual key for
1401          * both transmitting and receiving (keyidx 0) _unicast_ packets. Now,
1402          * keyidx 0 is reserved for this unicast use and default keys can only
1403          * use keyidx 1..3 (i.e., default key with keyidx 0 is not supported).
1404          * This should be fine for more or less all cases, but for completeness
1405          * sake, the driver could be enhanced to support the missing key. */
1406 #if 0
1407         if (addr == NULL)
1408                 os_memset(param->sta_addr, 0xff, ETH_ALEN);
1409         else
1410                 os_memcpy(param->sta_addr, addr, ETH_ALEN);
1411 #else
1412         os_memset(param->sta_addr, 0xff, ETH_ALEN);
1413 #endif
1414         os_strlcpy((char *) param->u.crypt.alg, alg_name,
1415                    HOSTAP_CRYPT_ALG_NAME_LEN);
1416         param->u.crypt.flags = set_tx ? HOSTAP_CRYPT_FLAG_SET_TX_KEY : 0;
1417         param->u.crypt.idx = key_idx;
1418         os_memcpy(param->u.crypt.seq, seq, seq_len);
1419         param->u.crypt.key_len = key_len;
1420         os_memcpy((u8 *) (param + 1), key, key_len);
1421
1422         if (hostapd_ioctl(drv, param, blen, 1)) {
1423                 wpa_printf(MSG_WARNING, "Failed to set encryption.");
1424                 show_set_key_error(param);
1425                 ret = -1;
1426         }
1427         os_free(buf);
1428
1429         return ret;
1430 }
1431
1432
1433 static int wpa_driver_hostap_set_countermeasures(void *priv, int enabled)
1434 {
1435         struct wpa_driver_hostap_data *drv = priv;
1436         wpa_printf(MSG_DEBUG, "%s: enabled=%d", __FUNCTION__, enabled);
1437         return prism2param(drv, PRISM2_PARAM_TKIP_COUNTERMEASURES, enabled);
1438 }
1439
1440
1441 static int wpa_driver_hostap_reset(struct wpa_driver_hostap_data *drv,
1442                                    int type)
1443 {
1444         struct iwreq iwr;
1445         int *i, ret = 0;
1446
1447         wpa_printf(MSG_DEBUG, "%s: type=%d", __FUNCTION__, type);
1448
1449         os_memset(&iwr, 0, sizeof(iwr));
1450         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1451         i = (int *) iwr.u.name;
1452         *i++ = type;
1453
1454         if (ioctl(drv->sock, PRISM2_IOCTL_RESET, &iwr) < 0) {
1455                 perror("ioctl[PRISM2_IOCTL_RESET]");
1456                 ret = -1;
1457         }
1458         return ret;
1459 }
1460
1461
1462 static int wpa_driver_hostap_mlme(struct wpa_driver_hostap_data *drv,
1463                                   const u8 *addr, int cmd, int reason_code)
1464 {
1465         struct prism2_hostapd_param param;
1466         int ret;
1467
1468         /* There does not seem to be a better way of deauthenticating or
1469          * disassociating with Prism2/2.5/3 than sending the management frame
1470          * and then resetting the Port0 to make sure both the AP and the STA
1471          * end up in disconnected state. */
1472         os_memset(&param, 0, sizeof(param));
1473         param.cmd = PRISM2_HOSTAPD_MLME;
1474         os_memcpy(param.sta_addr, addr, ETH_ALEN);
1475         param.u.mlme.cmd = cmd;
1476         param.u.mlme.reason_code = reason_code;
1477         ret = hostapd_ioctl(drv, &param, sizeof(param), 1);
1478         if (ret == 0) {
1479                 os_sleep(0, 100000);
1480                 ret = wpa_driver_hostap_reset(drv, 2);
1481         }
1482         return ret;
1483 }
1484
1485
1486 static int wpa_driver_hostap_deauthenticate(void *priv, const u8 *addr,
1487                                             int reason_code)
1488 {
1489         struct wpa_driver_hostap_data *drv = priv;
1490         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
1491         return wpa_driver_hostap_mlme(drv, addr, MLME_STA_DEAUTH,
1492                                       reason_code);
1493 }
1494
1495
1496 static int wpa_driver_hostap_disassociate(void *priv, const u8 *addr,
1497                                           int reason_code)
1498 {
1499         struct wpa_driver_hostap_data *drv = priv;
1500         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
1501         return wpa_driver_hostap_mlme(drv, addr, MLME_STA_DISASSOC,
1502                                       reason_code);
1503 }
1504
1505
1506 static int
1507 wpa_driver_hostap_associate(void *priv,
1508                             struct wpa_driver_associate_params *params)
1509 {
1510         struct wpa_driver_hostap_data *drv = priv;
1511         int ret = 0;
1512         int allow_unencrypted_eapol;
1513
1514         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
1515
1516         if (prism2param(drv, PRISM2_PARAM_DROP_UNENCRYPTED,
1517                         params->drop_unencrypted) < 0)
1518                 ret = -1;
1519         if (wpa_driver_hostap_set_auth_alg(drv, params->auth_alg) < 0)
1520                 ret = -1;
1521         if (params->mode != drv->current_mode) {
1522                 /* At the moment, Host AP driver requires host_roaming=2 for
1523                  * infrastructure mode and host_roaming=0 for adhoc. */
1524                 if (prism2param(drv, PRISM2_PARAM_HOST_ROAMING,
1525                                 params->mode == IEEE80211_MODE_IBSS ? 0 : 2) <
1526                     0) {
1527                         wpa_printf(MSG_DEBUG, "%s: failed to set host_roaming",
1528                                    __func__);
1529                 }
1530                 drv->current_mode = params->mode;
1531         }
1532
1533         if (prism2param(drv, PRISM2_PARAM_PRIVACY_INVOKED,
1534                         params->key_mgmt_suite != KEY_MGMT_NONE) < 0)
1535                 ret = -1;
1536         if (wpa_driver_hostap_set_wpa_ie(drv, params->wpa_ie,
1537                                          params->wpa_ie_len) < 0)
1538                 ret = -1;
1539         if (wpa_driver_wext_set_mode(drv->wext, params->mode) < 0)
1540                 ret = -1;
1541         if (params->freq &&
1542             wpa_driver_wext_set_freq(drv->wext, params->freq) < 0)
1543                 ret = -1;
1544         if (wpa_driver_wext_set_ssid(drv->wext, params->ssid, params->ssid_len)
1545             < 0)
1546                 ret = -1;
1547         if (wpa_driver_wext_set_bssid(drv->wext, params->bssid) < 0)
1548                 ret = -1;
1549
1550         /* Allow unencrypted EAPOL messages even if pairwise keys are set when
1551          * not using WPA. IEEE 802.1X specifies that these frames are not
1552          * encrypted, but WPA encrypts them when pairwise keys are in use. */
1553         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
1554             params->key_mgmt_suite == KEY_MGMT_PSK)
1555                 allow_unencrypted_eapol = 0;
1556         else
1557                 allow_unencrypted_eapol = 1;
1558         
1559         if (prism2param(drv, PRISM2_PARAM_IEEE_802_1X,
1560                         allow_unencrypted_eapol) < 0) {
1561                 wpa_printf(MSG_DEBUG, "hostap: Failed to configure "
1562                            "ieee_802_1x param");
1563                 /* Ignore this error.. driver_hostap.c can also be used with
1564                  * other drivers that do not support this prism2_param. */
1565         }
1566
1567         return ret;
1568 }
1569
1570
1571 static int wpa_driver_hostap_scan(void *priv,
1572                                   struct wpa_driver_scan_params *params)
1573 {
1574         struct wpa_driver_hostap_data *drv = priv;
1575         struct prism2_hostapd_param param;
1576         int ret;
1577         const u8 *ssid = params->ssids[0].ssid;
1578         size_t ssid_len = params->ssids[0].ssid_len;
1579
1580         if (ssid == NULL) {
1581                 /* Use standard Linux Wireless Extensions ioctl if possible
1582                  * because some drivers using hostap code in wpa_supplicant
1583                  * might not support Host AP specific scan request (with SSID
1584                  * info). */
1585                 return wpa_driver_wext_scan(drv->wext, params);
1586         }
1587
1588         if (ssid_len > 32)
1589                 ssid_len = 32;
1590
1591         os_memset(&param, 0, sizeof(param));
1592         param.cmd = PRISM2_HOSTAPD_SCAN_REQ;
1593         param.u.scan_req.ssid_len = ssid_len;
1594         os_memcpy(param.u.scan_req.ssid, ssid, ssid_len);
1595         ret = hostapd_ioctl(drv, &param, sizeof(param), 1);
1596
1597         /* Not all drivers generate "scan completed" wireless event, so try to
1598          * read results after a timeout. */
1599         eloop_cancel_timeout(wpa_driver_wext_scan_timeout, drv->wext,
1600                              drv->ctx);
1601         eloop_register_timeout(3, 0, wpa_driver_wext_scan_timeout, drv->wext,
1602                                drv->ctx);
1603
1604         return ret;
1605 }
1606
1607
1608 static int wpa_driver_hostap_set_auth_alg(void *priv, int auth_alg)
1609 {
1610         struct wpa_driver_hostap_data *drv = priv;
1611         int algs = 0;
1612
1613         if (auth_alg & AUTH_ALG_OPEN_SYSTEM)
1614                 algs |= 1;
1615         if (auth_alg & AUTH_ALG_SHARED_KEY)
1616                 algs |= 2;
1617         if (auth_alg & AUTH_ALG_LEAP)
1618                 algs |= 4;
1619         if (algs == 0)
1620                 algs = 1; /* at least one algorithm should be set */
1621
1622         return prism2param(drv, PRISM2_PARAM_AP_AUTH_ALGS, algs);
1623 }
1624
1625
1626 static int wpa_driver_hostap_get_bssid(void *priv, u8 *bssid)
1627 {
1628         struct wpa_driver_hostap_data *drv = priv;
1629         return wpa_driver_wext_get_bssid(drv->wext, bssid);
1630 }
1631
1632
1633 static int wpa_driver_hostap_get_ssid(void *priv, u8 *ssid)
1634 {
1635         struct wpa_driver_hostap_data *drv = priv;
1636         return wpa_driver_wext_get_ssid(drv->wext, ssid);
1637 }
1638
1639
1640 static struct wpa_scan_results * wpa_driver_hostap_get_scan_results(void *priv)
1641 {
1642         struct wpa_driver_hostap_data *drv = priv;
1643         return wpa_driver_wext_get_scan_results(drv->wext);
1644 }
1645
1646
1647 static int wpa_driver_hostap_set_operstate(void *priv, int state)
1648 {
1649         struct wpa_driver_hostap_data *drv = priv;
1650         return wpa_driver_wext_set_operstate(drv->wext, state);
1651 }
1652
1653
1654 static void * wpa_driver_hostap_init(void *ctx, const char *ifname)
1655 {
1656         struct wpa_driver_hostap_data *drv;
1657
1658         drv = os_zalloc(sizeof(*drv));
1659         if (drv == NULL)
1660                 return NULL;
1661         drv->wext = wpa_driver_wext_init(ctx, ifname);
1662         if (drv->wext == NULL) {
1663                 os_free(drv);
1664                 return NULL;
1665         }
1666
1667         drv->ctx = ctx;
1668         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
1669         drv->sock = socket(PF_INET, SOCK_DGRAM, 0);
1670         if (drv->sock < 0) {
1671                 perror("socket");
1672                 wpa_driver_wext_deinit(drv->wext);
1673                 os_free(drv);
1674                 return NULL;
1675         }
1676
1677         if (os_strncmp(ifname, "wlan", 4) == 0) {
1678                 /*
1679                  * Host AP driver may use both wlan# and wifi# interface in
1680                  * wireless events.
1681                  */
1682                 char ifname2[IFNAMSIZ + 1];
1683                 os_strlcpy(ifname2, ifname, sizeof(ifname2));
1684                 os_memcpy(ifname2, "wifi", 4);
1685                 wpa_driver_wext_alternative_ifindex(drv->wext, ifname2);
1686         }
1687
1688         wpa_driver_hostap_set_wpa(drv, 1);
1689
1690         return drv;
1691 }
1692
1693
1694 static void wpa_driver_hostap_deinit(void *priv)
1695 {
1696         struct wpa_driver_hostap_data *drv = priv;
1697         wpa_driver_hostap_set_wpa(drv, 0);
1698         wpa_driver_wext_deinit(drv->wext);
1699         close(drv->sock);
1700         os_free(drv);
1701 }
1702
1703 #endif /* HOSTAPD */
1704
1705
1706 const struct wpa_driver_ops wpa_driver_hostap_ops = {
1707         .name = "hostap",
1708         .desc = "Host AP driver (Intersil Prism2/2.5/3)",
1709         .set_key = wpa_driver_hostap_set_key,
1710 #ifdef HOSTAPD
1711         .hapd_init = hostap_init,
1712         .hapd_deinit = hostap_driver_deinit,
1713         .set_ieee8021x = hostap_set_ieee8021x,
1714         .set_privacy = hostap_set_privacy,
1715         .get_seqnum = hostap_get_seqnum,
1716         .flush = hostap_flush,
1717         .set_generic_elem = hostap_set_generic_elem,
1718         .read_sta_data = hostap_read_sta_data,
1719         .hapd_send_eapol = hostap_send_eapol,
1720         .sta_set_flags = hostap_sta_set_flags,
1721         .sta_deauth = hostap_sta_deauth,
1722         .sta_disassoc = hostap_sta_disassoc,
1723         .sta_remove = hostap_sta_remove,
1724         .hapd_set_ssid = hostap_set_ssid,
1725         .send_mlme = hostap_send_mlme,
1726         .sta_add = hostap_sta_add,
1727         .get_inact_sec = hostap_get_inact_sec,
1728         .sta_clear_stats = hostap_sta_clear_stats,
1729         .get_hw_feature_data = hostap_get_hw_feature_data,
1730         .set_wps_beacon_ie = hostap_set_wps_beacon_ie,
1731         .set_wps_probe_resp_ie = hostap_set_wps_probe_resp_ie,
1732 #else /* HOSTAPD */
1733         .get_bssid = wpa_driver_hostap_get_bssid,
1734         .get_ssid = wpa_driver_hostap_get_ssid,
1735         .set_countermeasures = wpa_driver_hostap_set_countermeasures,
1736         .scan2 = wpa_driver_hostap_scan,
1737         .get_scan_results2 = wpa_driver_hostap_get_scan_results,
1738         .deauthenticate = wpa_driver_hostap_deauthenticate,
1739         .disassociate = wpa_driver_hostap_disassociate,
1740         .associate = wpa_driver_hostap_associate,
1741         .init = wpa_driver_hostap_init,
1742         .deinit = wpa_driver_hostap_deinit,
1743         .set_operstate = wpa_driver_hostap_set_operstate,
1744 #endif /* HOSTAPD */
1745 };