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