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