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