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