Re-initialize hostapd/wpa_supplicant git repository based on 0.6.3 release
[mech_eap.git] / hostapd / driver_hostap.c
1 /*
2  * hostapd / Kernel driver communication with Linux Host AP driver
3  * Copyright (c) 2002-2007, 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 #ifdef USE_KERNEL_HEADERS
19 #include <asm/types.h>
20 #include <linux/if_packet.h>
21 #include <linux/if_ether.h>   /* The L2 protocols */
22 #include <linux/if_arp.h>
23 #include <linux/wireless.h>
24 #else /* USE_KERNEL_HEADERS */
25 #include <net/if_arp.h>
26 #include <netpacket/packet.h>
27 #include "wireless_copy.h"
28 #endif /* USE_KERNEL_HEADERS */
29
30 #include "hostapd.h"
31 #include "driver.h"
32 #include "ieee802_1x.h"
33 #include "eloop.h"
34 #include "priv_netlink.h"
35 #include "ieee802_11.h"
36 #include "sta_info.h"
37 #include "hostap_common.h"
38 #include "hw_features.h"
39
40
41 struct hostap_driver_data {
42         struct hostapd_data *hapd;
43
44         char iface[IFNAMSIZ + 1];
45         int sock; /* raw packet socket for driver access */
46         int ioctl_sock; /* socket for ioctl() use */
47         int wext_sock; /* socket for wireless events */
48
49         int we_version;
50
51         u8 *generic_ie;
52         size_t generic_ie_len;
53 };
54
55
56 static int hostapd_ioctl(void *priv, struct prism2_hostapd_param *param,
57                          int len);
58 static int hostap_set_iface_flags(void *priv, int dev_up);
59
60 static void handle_data(struct hostapd_data *hapd, u8 *buf, size_t len,
61                         u16 stype)
62 {
63         struct ieee80211_hdr *hdr;
64         u16 fc, ethertype;
65         u8 *pos, *sa;
66         size_t left;
67         struct sta_info *sta;
68
69         if (len < sizeof(struct ieee80211_hdr))
70                 return;
71
72         hdr = (struct ieee80211_hdr *) buf;
73         fc = le_to_host16(hdr->frame_control);
74
75         if ((fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) != WLAN_FC_TODS) {
76                 printf("Not ToDS data frame (fc=0x%04x)\n", fc);
77                 return;
78         }
79
80         sa = hdr->addr2;
81         sta = ap_get_sta(hapd, sa);
82         if (!sta || !(sta->flags & WLAN_STA_ASSOC)) {
83                 printf("Data frame from not associated STA " MACSTR "\n",
84                        MAC2STR(sa));
85                 if (sta && (sta->flags & WLAN_STA_AUTH))
86                         hostapd_sta_disassoc(
87                                 hapd, sa,
88                                 WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
89                 else
90                         hostapd_sta_deauth(
91                                 hapd, sa,
92                                 WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
93                 return;
94         }
95
96         pos = (u8 *) (hdr + 1);
97         left = len - sizeof(*hdr);
98
99         if (left < sizeof(rfc1042_header)) {
100                 printf("Too short data frame\n");
101                 return;
102         }
103
104         if (memcmp(pos, rfc1042_header, sizeof(rfc1042_header)) != 0) {
105                 printf("Data frame with no RFC1042 header\n");
106                 return;
107         }
108         pos += sizeof(rfc1042_header);
109         left -= sizeof(rfc1042_header);
110
111         if (left < 2) {
112                 printf("No ethertype in data frame\n");
113                 return;
114         }
115
116         ethertype = WPA_GET_BE16(pos);
117         pos += 2;
118         left -= 2;
119         switch (ethertype) {
120         case ETH_P_PAE:
121                 ieee802_1x_receive(hapd, sa, pos, left);
122                 break;
123
124         default:
125                 printf("Unknown ethertype 0x%04x in data frame\n", ethertype);
126                 break;
127         }
128 }
129
130
131 static void handle_tx_callback(struct hostapd_data *hapd, u8 *buf, size_t len,
132                                int ok)
133 {
134         struct ieee80211_hdr *hdr;
135         u16 fc, type, stype;
136         struct sta_info *sta;
137
138         hdr = (struct ieee80211_hdr *) buf;
139         fc = le_to_host16(hdr->frame_control);
140
141         type = WLAN_FC_GET_TYPE(fc);
142         stype = WLAN_FC_GET_STYPE(fc);
143
144         switch (type) {
145         case WLAN_FC_TYPE_MGMT:
146                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
147                            ok ? "ACK" : "fail");
148                 ieee802_11_mgmt_cb(hapd, buf, len, stype, ok);
149                 break;
150         case WLAN_FC_TYPE_CTRL:
151                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
152                            ok ? "ACK" : "fail");
153                 break;
154         case WLAN_FC_TYPE_DATA:
155                 wpa_printf(MSG_DEBUG, "DATA (TX callback) %s",
156                            ok ? "ACK" : "fail");
157                 sta = ap_get_sta(hapd, hdr->addr1);
158                 if (sta && sta->flags & WLAN_STA_PENDING_POLL) {
159                         wpa_printf(MSG_DEBUG, "STA " MACSTR
160                                    " %s pending activity poll",
161                                    MAC2STR(sta->addr),
162                                    ok ? "ACKed" : "did not ACK");
163                         if (ok)
164                                 sta->flags &= ~WLAN_STA_PENDING_POLL;
165                 }
166                 if (sta)
167                         ieee802_1x_tx_status(hapd, sta, buf, len, ok);
168                 break;
169         default:
170                 printf("unknown TX callback frame type %d\n", type);
171                 break;
172         }
173 }
174
175
176 static void handle_frame(struct hostapd_data *hapd, u8 *buf, size_t len)
177 {
178         struct ieee80211_hdr *hdr;
179         u16 fc, extra_len, type, stype;
180         unsigned char *extra = NULL;
181         size_t data_len = len;
182         int ver;
183
184         /* PSPOLL is only 16 bytes, but driver does not (at least yet) pass
185          * these to user space */
186         if (len < 24) {
187                 wpa_printf(MSG_MSGDUMP, "handle_frame: too short (%lu)",
188                            (unsigned long) len);
189                 return;
190         }
191
192         hdr = (struct ieee80211_hdr *) buf;
193         fc = le_to_host16(hdr->frame_control);
194         type = WLAN_FC_GET_TYPE(fc);
195         stype = WLAN_FC_GET_STYPE(fc);
196
197         if (type != WLAN_FC_TYPE_MGMT || stype != WLAN_FC_STYPE_BEACON) {
198                 wpa_hexdump(MSG_MSGDUMP, "Received management frame",
199                             buf, len);
200         }
201
202         ver = fc & WLAN_FC_PVER;
203
204         /* protocol version 3 is reserved for indicating extra data after the
205          * payload, version 2 for indicating ACKed frame (TX callbacks), and
206          * version 1 for indicating failed frame (no ACK, TX callbacks) */
207         if (ver == 3) {
208                 u8 *pos = buf + len - 2;
209                 extra_len = WPA_GET_LE16(pos);
210                 printf("extra data in frame (elen=%d)\n", extra_len);
211                 if ((size_t) extra_len + 2 > len) {
212                         printf("  extra data overflow\n");
213                         return;
214                 }
215                 len -= extra_len + 2;
216                 extra = buf + len;
217         } else if (ver == 1 || ver == 2) {
218                 handle_tx_callback(hapd, buf, data_len, ver == 2 ? 1 : 0);
219                 return;
220         } else if (ver != 0) {
221                 printf("unknown protocol version %d\n", ver);
222                 return;
223         }
224
225         switch (type) {
226         case WLAN_FC_TYPE_MGMT:
227                 if (stype != WLAN_FC_STYPE_BEACON)
228                         wpa_printf(MSG_MSGDUMP, "MGMT");
229                 ieee802_11_mgmt(hapd, buf, data_len, stype, NULL);
230                 break;
231         case WLAN_FC_TYPE_CTRL:
232                 wpa_printf(MSG_DEBUG, "CTRL");
233                 break;
234         case WLAN_FC_TYPE_DATA:
235                 wpa_printf(MSG_DEBUG, "DATA");
236                 handle_data(hapd, buf, data_len, stype);
237                 break;
238         default:
239                 wpa_printf(MSG_DEBUG, "unknown frame type %d", type);
240                 break;
241         }
242 }
243
244
245 static void handle_read(int sock, void *eloop_ctx, void *sock_ctx)
246 {
247         struct hostapd_data *hapd = (struct hostapd_data *) eloop_ctx;
248         int len;
249         unsigned char buf[3000];
250
251         len = recv(sock, buf, sizeof(buf), 0);
252         if (len < 0) {
253                 perror("recv");
254                 return;
255         }
256
257         handle_frame(hapd, buf, len);
258 }
259
260
261 static int hostap_init_sockets(struct hostap_driver_data *drv)
262 {
263         struct ifreq ifr;
264         struct sockaddr_ll addr;
265
266         drv->sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
267         if (drv->sock < 0) {
268                 perror("socket[PF_PACKET,SOCK_RAW]");
269                 return -1;
270         }
271
272         if (eloop_register_read_sock(drv->sock, handle_read, drv->hapd, NULL))
273         {
274                 printf("Could not register read socket\n");
275                 return -1;
276         }
277
278         memset(&ifr, 0, sizeof(ifr));
279         snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "%sap", drv->iface);
280         if (ioctl(drv->sock, SIOCGIFINDEX, &ifr) != 0) {
281                 perror("ioctl(SIOCGIFINDEX)");
282                 return -1;
283         }
284
285         if (hostap_set_iface_flags(drv, 1)) {
286                 return -1;
287         }
288
289         memset(&addr, 0, sizeof(addr));
290         addr.sll_family = AF_PACKET;
291         addr.sll_ifindex = ifr.ifr_ifindex;
292         wpa_printf(MSG_DEBUG, "Opening raw packet socket for ifindex %d",
293                    addr.sll_ifindex);
294
295         if (bind(drv->sock, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
296                 perror("bind");
297                 return -1;
298         }
299
300         memset(&ifr, 0, sizeof(ifr));
301         os_strlcpy(ifr.ifr_name, drv->iface, sizeof(ifr.ifr_name));
302         if (ioctl(drv->sock, SIOCGIFHWADDR, &ifr) != 0) {
303                 perror("ioctl(SIOCGIFHWADDR)");
304                 return -1;
305         }
306
307         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
308                 printf("Invalid HW-addr family 0x%04x\n",
309                        ifr.ifr_hwaddr.sa_family);
310                 return -1;
311         }
312         memcpy(drv->hapd->own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
313
314         return 0;
315 }
316
317
318 static int hostap_send_mgmt_frame(void *priv, const void *msg, size_t len,
319                                   int flags)
320 {
321         struct hostap_driver_data *drv = priv;
322         
323         return send(drv->sock, msg, len, flags);
324 }
325
326
327 static int hostap_send_eapol(void *priv, const u8 *addr, const u8 *data,
328                              size_t data_len, int encrypt, const u8 *own_addr)
329 {
330         struct hostap_driver_data *drv = priv;
331         struct ieee80211_hdr *hdr;
332         size_t len;
333         u8 *pos;
334         int res;
335
336         len = sizeof(*hdr) + sizeof(rfc1042_header) + 2 + data_len;
337         hdr = os_zalloc(len);
338         if (hdr == NULL) {
339                 printf("malloc() failed for hostapd_send_data(len=%lu)\n",
340                        (unsigned long) len);
341                 return -1;
342         }
343
344         hdr->frame_control =
345                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
346         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
347         /* Request TX callback */
348         hdr->frame_control |= host_to_le16(BIT(1));
349         if (encrypt)
350                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
351         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
352         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
353         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
354
355         pos = (u8 *) (hdr + 1);
356         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
357         pos += sizeof(rfc1042_header);
358         *((u16 *) pos) = htons(ETH_P_PAE);
359         pos += 2;
360         memcpy(pos, data, data_len);
361
362         res = hostap_send_mgmt_frame(drv, (u8 *) hdr, len, 0);
363         free(hdr);
364
365         if (res < 0) {
366                 perror("hostapd_send_eapol: send");
367                 printf("hostapd_send_eapol - packet len: %lu - failed\n",
368                        (unsigned long) len);
369         }
370
371         return res;
372 }
373
374
375 static int hostap_sta_set_flags(void *priv, const u8 *addr,
376                                 int total_flags, int flags_or, int flags_and)
377 {
378         struct hostap_driver_data *drv = priv;
379         struct prism2_hostapd_param param;
380
381         memset(&param, 0, sizeof(param));
382         param.cmd = PRISM2_HOSTAPD_SET_FLAGS_STA;
383         memcpy(param.sta_addr, addr, ETH_ALEN);
384         param.u.set_flags_sta.flags_or = flags_or;
385         param.u.set_flags_sta.flags_and = flags_and;
386         return hostapd_ioctl(drv, &param, sizeof(param));
387 }
388
389
390 static int hostap_set_iface_flags(void *priv, int dev_up)
391 {
392         struct hostap_driver_data *drv = priv;
393         struct ifreq ifr;
394
395         if (drv->ioctl_sock < 0)
396                 return -1;
397
398         memset(&ifr, 0, sizeof(ifr));
399         snprintf(ifr.ifr_name, IFNAMSIZ, "%sap", drv->iface);
400
401         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
402                 perror("ioctl[SIOCGIFFLAGS]");
403                 return -1;
404         }
405
406         if (dev_up)
407                 ifr.ifr_flags |= IFF_UP;
408         else
409                 ifr.ifr_flags &= ~IFF_UP;
410
411         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
412                 perror("ioctl[SIOCSIFFLAGS]");
413                 return -1;
414         }
415
416         if (dev_up) {
417                 memset(&ifr, 0, sizeof(ifr));
418                 snprintf(ifr.ifr_name, IFNAMSIZ, "%sap", drv->iface);
419                 ifr.ifr_mtu = HOSTAPD_MTU;
420                 if (ioctl(drv->ioctl_sock, SIOCSIFMTU, &ifr) != 0) {
421                         perror("ioctl[SIOCSIFMTU]");
422                         printf("Setting MTU failed - trying to survive with "
423                                "current value\n");
424                 }
425         }
426
427         return 0;
428 }
429
430
431 static int hostapd_ioctl(void *priv, struct prism2_hostapd_param *param,
432                          int len)
433 {
434         struct hostap_driver_data *drv = priv;
435         struct iwreq iwr;
436
437         memset(&iwr, 0, sizeof(iwr));
438         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
439         iwr.u.data.pointer = (caddr_t) param;
440         iwr.u.data.length = len;
441
442         if (ioctl(drv->ioctl_sock, PRISM2_IOCTL_HOSTAPD, &iwr) < 0) {
443                 perror("ioctl[PRISM2_IOCTL_HOSTAPD]");
444                 return -1;
445         }
446
447         return 0;
448 }
449
450
451 static int hostap_set_encryption(const char *ifname, void *priv,
452                                  const char *alg, const u8 *addr,
453                                  int idx, const u8 *key, size_t key_len,
454                                  int txkey)
455 {
456         struct hostap_driver_data *drv = priv;
457         struct prism2_hostapd_param *param;
458         u8 *buf;
459         size_t blen;
460         int ret = 0;
461
462         blen = sizeof(*param) + key_len;
463         buf = os_zalloc(blen);
464         if (buf == NULL)
465                 return -1;
466
467         param = (struct prism2_hostapd_param *) buf;
468         param->cmd = PRISM2_SET_ENCRYPTION;
469         if (addr == NULL)
470                 memset(param->sta_addr, 0xff, ETH_ALEN);
471         else
472                 memcpy(param->sta_addr, addr, ETH_ALEN);
473         os_strlcpy((char *) param->u.crypt.alg, alg,
474                    HOSTAP_CRYPT_ALG_NAME_LEN);
475         param->u.crypt.flags = txkey ? HOSTAP_CRYPT_FLAG_SET_TX_KEY : 0;
476         param->u.crypt.idx = 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(const char *ifname, void *priv, int enabled)
546 {
547         struct hostap_driver_data *drv = priv;
548
549         /* enable kernel driver support for IEEE 802.1X */
550         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_IEEE_802_1X, enabled)) {
551                 printf("Could not setup IEEE 802.1X support in kernel driver."
552                        "\n");
553                 return -1;
554         }
555
556         if (!enabled)
557                 return 0;
558
559         /* use host driver implementation of encryption to allow
560          * individual keys and passing plaintext EAPOL frames */
561         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOST_DECRYPT, 1) ||
562             hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOST_ENCRYPT, 1)) {
563                 printf("Could not setup host-based encryption in kernel "
564                        "driver.\n");
565                 return -1;
566         }
567
568         return 0;
569 }
570
571
572 static int hostap_set_privacy(const char *ifname, void *priv, int enabled)
573 {
574         struct hostap_drvier_data *drv = priv;
575
576         return hostap_ioctl_prism2param(drv, PRISM2_PARAM_PRIVACY_INVOKED,
577                                         enabled);
578 }
579
580  
581 static int hostap_set_ssid(const char *ifname, void *priv, const u8 *buf,
582                            int len)
583 {
584         struct hostap_driver_data *drv = priv;
585         struct iwreq iwr;
586
587         memset(&iwr, 0, sizeof(iwr));
588         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
589         iwr.u.essid.flags = 1; /* SSID active */
590         iwr.u.essid.pointer = (caddr_t) buf;
591         iwr.u.essid.length = len + 1;
592
593         if (ioctl(drv->ioctl_sock, SIOCSIWESSID, &iwr) < 0) {
594                 perror("ioctl[SIOCSIWESSID]");
595                 printf("len=%d\n", len);
596                 return -1;
597         }
598
599         return 0;
600 }
601
602
603 static int hostap_flush(void *priv)
604 {
605         struct hostap_driver_data *drv = priv;
606         struct prism2_hostapd_param param;
607
608         memset(&param, 0, sizeof(param));
609         param.cmd = PRISM2_HOSTAPD_FLUSH;
610         return hostapd_ioctl(drv, &param, sizeof(param));
611 }
612
613
614 static int hostap_read_sta_data(void *priv,
615                                 struct hostap_sta_driver_data *data,
616                                 const u8 *addr)
617 {
618         struct hostap_driver_data *drv = priv;
619         char buf[1024], line[128], *pos;
620         FILE *f;
621         unsigned long val;
622
623         memset(data, 0, sizeof(*data));
624         snprintf(buf, sizeof(buf), "/proc/net/hostap/%s/" MACSTR,
625                  drv->iface, MAC2STR(addr));
626
627         f = fopen(buf, "r");
628         if (!f)
629                 return -1;
630         /* Need to read proc file with in one piece, so use large enough
631          * buffer. */
632         setbuffer(f, buf, sizeof(buf));
633
634         while (fgets(line, sizeof(line), f)) {
635                 pos = strchr(line, '=');
636                 if (!pos)
637                         continue;
638                 *pos++ = '\0';
639                 val = strtoul(pos, NULL, 10);
640                 if (strcmp(line, "rx_packets") == 0)
641                         data->rx_packets = val;
642                 else if (strcmp(line, "tx_packets") == 0)
643                         data->tx_packets = val;
644                 else if (strcmp(line, "rx_bytes") == 0)
645                         data->rx_bytes = val;
646                 else if (strcmp(line, "tx_bytes") == 0)
647                         data->tx_bytes = val;
648         }
649
650         fclose(f);
651
652         return 0;
653 }
654
655
656 static int hostap_sta_add(const char *ifname, void *priv, const u8 *addr,
657                           u16 aid, u16 capability, u8 *supp_rates,
658                           size_t supp_rates_len, int flags)
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 < supp_rates_len; i++) {
671                 if ((supp_rates[i] & 0x7f) == 2)
672                         tx_supp_rates |= WLAN_RATE_1M;
673                 if ((supp_rates[i] & 0x7f) == 4)
674                         tx_supp_rates |= WLAN_RATE_2M;
675                 if ((supp_rates[i] & 0x7f) == 11)
676                         tx_supp_rates |= WLAN_RATE_5M5;
677                 if ((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, addr, ETH_ALEN);
684         param.u.add_sta.aid = aid;
685         param.u.add_sta.capability = 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 hostap_set_assoc_ap(void *priv, const u8 *addr)
742 {
743         struct hostap_driver_data *drv = priv;
744         struct prism2_hostapd_param param;
745
746         memset(&param, 0, sizeof(param));
747         param.cmd = PRISM2_HOSTAPD_SET_ASSOC_AP_ADDR;
748         memcpy(param.sta_addr, addr, ETH_ALEN);
749         if (hostapd_ioctl(drv, &param, sizeof(param)))
750                 return -1;
751
752         return 0;
753 }
754
755
756 static int hostapd_ioctl_set_generic_elem(struct hostap_driver_data *drv)
757 {
758         struct prism2_hostapd_param *param;
759         int res;
760         size_t blen, elem_len;
761
762         elem_len = drv->generic_ie_len;
763         blen = PRISM2_HOSTAPD_GENERIC_ELEMENT_HDR_LEN + elem_len;
764         if (blen < sizeof(*param))
765                 blen = sizeof(*param);
766
767         param = os_zalloc(blen);
768         if (param == NULL)
769                 return -1;
770
771         param->cmd = PRISM2_HOSTAPD_SET_GENERIC_ELEMENT;
772         param->u.generic_elem.len = elem_len;
773         if (drv->generic_ie) {
774                 os_memcpy(param->u.generic_elem.data, drv->generic_ie,
775                           drv->generic_ie_len);
776         }
777         wpa_hexdump(MSG_DEBUG, "hostap: Set generic IE",
778                     param->u.generic_elem.data, elem_len);
779         res = hostapd_ioctl(drv, param, blen);
780
781         os_free(param);
782
783         return res;
784 }
785
786
787 static int hostap_set_generic_elem(const char *ifname, void *priv,
788                                    const u8 *elem, size_t elem_len)
789 {
790         struct hostap_driver_data *drv = priv;
791
792         os_free(drv->generic_ie);
793         drv->generic_ie = NULL;
794         drv->generic_ie_len = 0;
795         if (elem) {
796                 drv->generic_ie = os_malloc(elem_len);
797                 if (drv->generic_ie == NULL)
798                         return -1;
799                 os_memcpy(drv->generic_ie, elem, elem_len);
800                 drv->generic_ie_len = elem_len;
801         }
802
803         return hostapd_ioctl_set_generic_elem(drv);
804 }
805
806
807 static void
808 hostapd_wireless_event_wireless_custom(struct hostap_driver_data *drv,
809                                        char *custom)
810 {
811         wpa_printf(MSG_DEBUG, "Custom wireless event: '%s'", custom);
812
813         if (strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
814                 char *pos;
815                 u8 addr[ETH_ALEN];
816                 pos = strstr(custom, "addr=");
817                 if (pos == NULL) {
818                         wpa_printf(MSG_DEBUG,
819                                    "MLME-MICHAELMICFAILURE.indication "
820                                    "without sender address ignored");
821                         return;
822                 }
823                 pos += 5;
824                 if (hwaddr_aton(pos, addr) == 0) {
825                         ieee80211_michael_mic_failure(drv->hapd, addr, 1);
826                 } else {
827                         wpa_printf(MSG_DEBUG,
828                                    "MLME-MICHAELMICFAILURE.indication "
829                                    "with invalid MAC address");
830                 }
831         }
832 }
833
834
835 static void hostapd_wireless_event_wireless(struct hostap_driver_data *drv,
836                                             char *data, int len)
837 {
838         struct iw_event iwe_buf, *iwe = &iwe_buf;
839         char *pos, *end, *custom, *buf;
840
841         pos = data;
842         end = data + len;
843
844         while (pos + IW_EV_LCP_LEN <= end) {
845                 /* Event data may be unaligned, so make a local, aligned copy
846                  * before processing. */
847                 memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
848                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
849                            iwe->cmd, iwe->len);
850                 if (iwe->len <= IW_EV_LCP_LEN)
851                         return;
852
853                 custom = pos + IW_EV_POINT_LEN;
854                 if (drv->we_version > 18 &&
855                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
856                      iwe->cmd == IWEVCUSTOM)) {
857                         /* WE-19 removed the pointer from struct iw_point */
858                         char *dpos = (char *) &iwe_buf.u.data.length;
859                         int dlen = dpos - (char *) &iwe_buf;
860                         memcpy(dpos, pos + IW_EV_LCP_LEN,
861                                sizeof(struct iw_event) - dlen);
862                 } else {
863                         memcpy(&iwe_buf, pos, sizeof(struct iw_event));
864                         custom += IW_EV_POINT_OFF;
865                 }
866
867                 switch (iwe->cmd) {
868                 case IWEVCUSTOM:
869                         if (custom + iwe->u.data.length > end)
870                                 return;
871                         buf = malloc(iwe->u.data.length + 1);
872                         if (buf == NULL)
873                                 return;
874                         memcpy(buf, custom, iwe->u.data.length);
875                         buf[iwe->u.data.length] = '\0';
876                         hostapd_wireless_event_wireless_custom(drv, buf);
877                         free(buf);
878                         break;
879                 }
880
881                 pos += iwe->len;
882         }
883 }
884
885
886 static void hostapd_wireless_event_rtm_newlink(struct hostap_driver_data *drv,
887                                                struct nlmsghdr *h, int len)
888 {
889         struct ifinfomsg *ifi;
890         int attrlen, nlmsg_len, rta_len;
891         struct rtattr * attr;
892
893         if (len < (int) sizeof(*ifi))
894                 return;
895
896         ifi = NLMSG_DATA(h);
897
898         /* TODO: use ifi->ifi_index to filter out wireless events from other
899          * interfaces */
900
901         nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
902
903         attrlen = h->nlmsg_len - nlmsg_len;
904         if (attrlen < 0)
905                 return;
906
907         attr = (struct rtattr *) (((char *) ifi) + nlmsg_len);
908
909         rta_len = RTA_ALIGN(sizeof(struct rtattr));
910         while (RTA_OK(attr, attrlen)) {
911                 if (attr->rta_type == IFLA_WIRELESS) {
912                         hostapd_wireless_event_wireless(
913                                 drv, ((char *) attr) + rta_len,
914                                 attr->rta_len - rta_len);
915                 }
916                 attr = RTA_NEXT(attr, attrlen);
917         }
918 }
919
920
921 static void hostapd_wireless_event_receive(int sock, void *eloop_ctx,
922                                            void *sock_ctx)
923 {
924         char buf[256];
925         int left;
926         struct sockaddr_nl from;
927         socklen_t fromlen;
928         struct nlmsghdr *h;
929         struct hostap_driver_data *drv = eloop_ctx;
930
931         fromlen = sizeof(from);
932         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
933                         (struct sockaddr *) &from, &fromlen);
934         if (left < 0) {
935                 if (errno != EINTR && errno != EAGAIN)
936                         perror("recvfrom(netlink)");
937                 return;
938         }
939
940         h = (struct nlmsghdr *) buf;
941         while (left >= (int) sizeof(*h)) {
942                 int len, plen;
943
944                 len = h->nlmsg_len;
945                 plen = len - sizeof(*h);
946                 if (len > left || plen < 0) {
947                         printf("Malformed netlink message: "
948                                "len=%d left=%d plen=%d\n",
949                                len, left, plen);
950                         break;
951                 }
952
953                 switch (h->nlmsg_type) {
954                 case RTM_NEWLINK:
955                         hostapd_wireless_event_rtm_newlink(drv, h, plen);
956                         break;
957                 }
958
959                 len = NLMSG_ALIGN(len);
960                 left -= len;
961                 h = (struct nlmsghdr *) ((char *) h + len);
962         }
963
964         if (left > 0) {
965                 printf("%d extra bytes in the end of netlink message\n", left);
966         }
967 }
968
969
970 static int hostap_get_we_version(struct hostap_driver_data *drv)
971 {
972         struct iw_range *range;
973         struct iwreq iwr;
974         int minlen;
975         size_t buflen;
976
977         drv->we_version = 0;
978
979         /*
980          * Use larger buffer than struct iw_range in order to allow the
981          * structure to grow in the future.
982          */
983         buflen = sizeof(struct iw_range) + 500;
984         range = os_zalloc(buflen);
985         if (range == NULL)
986                 return -1;
987
988         memset(&iwr, 0, sizeof(iwr));
989         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
990         iwr.u.data.pointer = (caddr_t) range;
991         iwr.u.data.length = buflen;
992
993         minlen = ((char *) &range->enc_capa) - (char *) range +
994                 sizeof(range->enc_capa);
995
996         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
997                 perror("ioctl[SIOCGIWRANGE]");
998                 free(range);
999                 return -1;
1000         } else if (iwr.u.data.length >= minlen &&
1001                    range->we_version_compiled >= 18) {
1002                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
1003                            "WE(source)=%d enc_capa=0x%x",
1004                            range->we_version_compiled,
1005                            range->we_version_source,
1006                            range->enc_capa);
1007                 drv->we_version = range->we_version_compiled;
1008         }
1009
1010         free(range);
1011         return 0;
1012 }
1013
1014
1015 static int hostap_wireless_event_init(void *priv)
1016 {
1017         struct hostap_driver_data *drv = priv;
1018         int s;
1019         struct sockaddr_nl local;
1020
1021         hostap_get_we_version(drv);
1022
1023         drv->wext_sock = -1;
1024
1025         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1026         if (s < 0) {
1027                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1028                 return -1;
1029         }
1030
1031         memset(&local, 0, sizeof(local));
1032         local.nl_family = AF_NETLINK;
1033         local.nl_groups = RTMGRP_LINK;
1034         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1035                 perror("bind(netlink)");
1036                 close(s);
1037                 return -1;
1038         }
1039
1040         eloop_register_read_sock(s, hostapd_wireless_event_receive, drv,
1041                                  NULL);
1042         drv->wext_sock = s;
1043
1044         return 0;
1045 }
1046
1047
1048 static void hostap_wireless_event_deinit(void *priv)
1049 {
1050         struct hostap_driver_data *drv = priv;
1051         if (drv->wext_sock < 0)
1052                 return;
1053         eloop_unregister_read_sock(drv->wext_sock);
1054         close(drv->wext_sock);
1055 }
1056
1057
1058 static void * hostap_init(struct hostapd_data *hapd)
1059 {
1060         struct hostap_driver_data *drv;
1061
1062         drv = os_zalloc(sizeof(struct hostap_driver_data));
1063         if (drv == NULL) {
1064                 printf("Could not allocate memory for hostapd driver data\n");
1065                 return NULL;
1066         }
1067
1068         drv->hapd = hapd;
1069         drv->ioctl_sock = drv->sock = -1;
1070         memcpy(drv->iface, hapd->conf->iface, sizeof(drv->iface));
1071
1072         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1073         if (drv->ioctl_sock < 0) {
1074                 perror("socket[PF_INET,SOCK_DGRAM]");
1075                 free(drv);
1076                 return NULL;
1077         }
1078
1079         if (hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD, 1)) {
1080                 printf("Could not enable hostapd mode for interface %s\n",
1081                        drv->iface);
1082                 close(drv->ioctl_sock);
1083                 free(drv);
1084                 return NULL;
1085         }
1086
1087         if (hapd->conf->assoc_ap &&
1088             hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD_STA, 1)) {
1089                 printf("Could not enable hostapd STA mode for interface %s\n",
1090                        drv->iface);
1091                 close(drv->ioctl_sock);
1092                 free(drv);
1093                 return NULL;
1094         }
1095
1096         if (hostap_init_sockets(drv)) {
1097                 close(drv->ioctl_sock);
1098                 free(drv);
1099                 return NULL;
1100         }
1101
1102         return drv;
1103 }
1104
1105
1106 static void hostap_driver_deinit(void *priv)
1107 {
1108         struct hostap_driver_data *drv = priv;
1109
1110         (void) hostap_set_iface_flags(drv, 0);
1111         (void) hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD, 0);
1112         (void) hostap_ioctl_prism2param(drv, PRISM2_PARAM_HOSTAPD_STA, 0);
1113
1114         if (drv->ioctl_sock >= 0)
1115                 close(drv->ioctl_sock);
1116
1117         if (drv->sock >= 0)
1118                 close(drv->sock);
1119
1120         os_free(drv->generic_ie);
1121
1122         free(drv);
1123 }
1124
1125
1126 static int hostap_sta_deauth(void *priv, const u8 *addr, int reason)
1127 {
1128         struct hostap_driver_data *drv = priv;
1129         struct ieee80211_mgmt mgmt;
1130
1131         memset(&mgmt, 0, sizeof(mgmt));
1132         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1133                                           WLAN_FC_STYPE_DEAUTH);
1134         memcpy(mgmt.da, addr, ETH_ALEN);
1135         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
1136         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
1137         mgmt.u.deauth.reason_code = host_to_le16(reason);
1138         return hostap_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
1139                                       sizeof(mgmt.u.deauth), 0);
1140 }
1141
1142
1143 static int hostap_sta_disassoc(void *priv, const u8 *addr, int reason)
1144 {
1145         struct hostap_driver_data *drv = priv;
1146         struct ieee80211_mgmt mgmt;
1147
1148         memset(&mgmt, 0, sizeof(mgmt));
1149         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
1150                                           WLAN_FC_STYPE_DISASSOC);
1151         memcpy(mgmt.da, addr, ETH_ALEN);
1152         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
1153         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
1154         mgmt.u.disassoc.reason_code = host_to_le16(reason);
1155         return  hostap_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
1156                                        sizeof(mgmt.u.disassoc), 0);
1157 }
1158
1159
1160 static struct hostapd_hw_modes * hostap_get_hw_feature_data(void *priv,
1161                                                             u16 *num_modes,
1162                                                             u16 *flags)
1163 {
1164         struct hostapd_hw_modes *mode;
1165         int i, clen, rlen;
1166         const short chan2freq[14] = {
1167                 2412, 2417, 2422, 2427, 2432, 2437, 2442,
1168                 2447, 2452, 2457, 2462, 2467, 2472, 2484
1169         };
1170
1171         mode = os_zalloc(sizeof(struct hostapd_hw_modes));
1172         if (mode == NULL)
1173                 return NULL;
1174
1175         *num_modes = 1;
1176         *flags = 0;
1177
1178         mode->mode = HOSTAPD_MODE_IEEE80211B;
1179         mode->num_channels = 14;
1180         mode->num_rates = 4;
1181
1182         clen = mode->num_channels * sizeof(struct hostapd_channel_data);
1183         rlen = mode->num_rates * sizeof(struct hostapd_rate_data);
1184
1185         mode->channels = os_zalloc(clen);
1186         mode->rates = os_zalloc(rlen);
1187         if (mode->channels == NULL || mode->rates == NULL) {
1188                 hostapd_free_hw_features(mode, *num_modes);
1189                 return NULL;
1190         }
1191
1192         for (i = 0; i < 14; i++) {
1193                 mode->channels[i].chan = i + 1;
1194                 mode->channels[i].freq = chan2freq[i];
1195         }
1196
1197         mode->rates[0].rate = 10;
1198         mode->rates[0].flags = HOSTAPD_RATE_CCK;
1199         mode->rates[1].rate = 20;
1200         mode->rates[1].flags = HOSTAPD_RATE_CCK;
1201         mode->rates[2].rate = 55;
1202         mode->rates[2].flags = HOSTAPD_RATE_CCK;
1203         mode->rates[3].rate = 110;
1204         mode->rates[3].flags = HOSTAPD_RATE_CCK;
1205
1206         return mode;
1207 }
1208
1209
1210 const struct wpa_driver_ops wpa_driver_hostap_ops = {
1211         .name = "hostap",
1212         .init = hostap_init,
1213         .deinit = hostap_driver_deinit,
1214         .wireless_event_init = hostap_wireless_event_init,
1215         .wireless_event_deinit = hostap_wireless_event_deinit,
1216         .set_ieee8021x = hostap_set_ieee8021x,
1217         .set_privacy = hostap_set_privacy,
1218         .set_encryption = hostap_set_encryption,
1219         .get_seqnum = hostap_get_seqnum,
1220         .flush = hostap_flush,
1221         .set_generic_elem = hostap_set_generic_elem,
1222         .read_sta_data = hostap_read_sta_data,
1223         .send_eapol = hostap_send_eapol,
1224         .sta_set_flags = hostap_sta_set_flags,
1225         .sta_deauth = hostap_sta_deauth,
1226         .sta_disassoc = hostap_sta_disassoc,
1227         .sta_remove = hostap_sta_remove,
1228         .set_ssid = hostap_set_ssid,
1229         .send_mgmt_frame = hostap_send_mgmt_frame,
1230         .set_assoc_ap = hostap_set_assoc_ap,
1231         .sta_add = hostap_sta_add,
1232         .get_inact_sec = hostap_get_inact_sec,
1233         .sta_clear_stats = hostap_sta_clear_stats,
1234         .get_hw_feature_data = hostap_get_hw_feature_data,
1235 };