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