515af765e99346588f37774114f7ce88a4c9dedb
[libeap.git] / hostapd / driver_nl80211.c
1 /*
2  * hostapd / Kernel driver communication via nl80211
3  * Copyright (c) 2002-2007, Jouni Malinen <j@w1.fi>
4  * Copyright (c) 2003-2004, Instant802 Networks, Inc.
5  * Copyright (c) 2005-2006, Devicescape Software, Inc.
6  * Copyright (c) 2007, Johannes Berg <johannes@sipsolutions.net>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 as
10  * published by the Free Software Foundation.
11  *
12  * Alternatively, this software may be distributed under the terms of BSD
13  * license.
14  *
15  * See README and COPYING for more details.
16  */
17
18 #include "includes.h"
19
20 #include <sys/ioctl.h>
21 #include <netlink/genl/genl.h>
22 #include <netlink/genl/family.h>
23 #include <netlink/genl/ctrl.h>
24 #include <netlink/msg.h>
25 #include <netlink/attr.h>
26 #include <linux/nl80211.h>
27 #include <net/if.h>
28 #include <linux/if_packet.h>
29 #include <linux/if_ether.h>   /* The L2 protocols */
30 #include "wireless_copy.h"
31 #include <net/if_arp.h>
32
33 #include "hostapd.h"
34 #include "driver.h"
35 #include "ieee802_1x.h"
36 #include "eloop.h"
37 #include "ieee802_11.h"
38 #include "sta_info.h"
39 #include "hw_features.h"
40 #include "mlme.h"
41 #include "radiotap.h"
42 #include "radiotap_iter.h"
43
44 enum ieee80211_msg_type {
45         ieee80211_msg_normal = 0,
46         ieee80211_msg_tx_callback_ack = 1,
47         ieee80211_msg_tx_callback_fail = 2,
48 };
49
50 struct i802_driver_data {
51         struct hostapd_data *hapd;
52
53         char iface[IFNAMSIZ + 1];
54         int bridge;
55         int ioctl_sock; /* socket for ioctl() use */
56         int wext_sock; /* socket for wireless events */
57         int eapol_sock; /* socket for EAPOL frames */
58         int monitor_sock; /* socket for monitor */
59         int monitor_ifidx;
60
61         int default_if_indices[16];
62         int *if_indices;
63         int num_if_indices;
64
65         int we_version;
66         struct nl_handle *nl_handle;
67         struct nl_cache *nl_cache;
68         struct nl_cb *nl_cb;
69         struct genl_family *nl80211;
70         int dtim_period, beacon_int;
71         unsigned int beacon_set:1;
72         unsigned int ieee802_1x_active:1;
73 };
74
75
76 static void add_ifidx(struct i802_driver_data *drv, int ifidx)
77 {
78         int i;
79         int *old;
80
81         for (i = 0; i < drv->num_if_indices; i++) {
82                 if (drv->if_indices[i] == 0) {
83                         drv->if_indices[i] = ifidx;
84                         return;
85                 }
86         }
87
88         if (drv->if_indices != drv->default_if_indices)
89                 old = drv->if_indices;
90         else
91                 old = NULL;
92
93         drv->if_indices = realloc(old,
94                                   sizeof(int) * (drv->num_if_indices + 1));
95         if (!drv->if_indices) {
96                 if (!old)
97                         drv->if_indices = drv->default_if_indices;
98                 else
99                         drv->if_indices = old;
100                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
101                            "interfaces");
102                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
103                 return;
104         }
105         drv->if_indices[drv->num_if_indices] = ifidx;
106         drv->num_if_indices++;
107 }
108
109
110 static void del_ifidx(struct i802_driver_data *drv, int ifidx)
111 {
112         int i;
113
114         for (i = 0; i < drv->num_if_indices; i++) {
115                 if (drv->if_indices[i] == ifidx) {
116                         drv->if_indices[i] = 0;
117                         break;
118                 }
119         }
120 }
121
122
123 static int have_ifidx(struct i802_driver_data *drv, int ifidx)
124 {
125         int i;
126
127         if (ifidx == drv->bridge)
128                 return 1;
129
130         for (i = 0; i < drv->num_if_indices; i++)
131                 if (drv->if_indices[i] == ifidx)
132                         return 1;
133
134         return 0;
135 }
136
137
138 /* helper for netlink get routines */
139 static int ack_wait_handler(struct nl_msg *msg, void *arg)
140 {
141         int *finished = arg;
142
143         *finished = 1;
144         return NL_STOP;
145 }
146
147
148 static int hostapd_set_iface_flags(struct i802_driver_data *drv,
149                                    const char *ifname, int dev_up)
150 {
151         struct ifreq ifr;
152
153         if (drv->ioctl_sock < 0)
154                 return -1;
155
156         memset(&ifr, 0, sizeof(ifr));
157         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
158
159         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
160                 perror("ioctl[SIOCGIFFLAGS]");
161                 wpa_printf(MSG_DEBUG, "Could not read interface flags (%s)",
162                            drv->iface);
163                 return -1;
164         }
165
166         if (dev_up)
167                 ifr.ifr_flags |= IFF_UP;
168         else
169                 ifr.ifr_flags &= ~IFF_UP;
170
171         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
172                 perror("ioctl[SIOCSIFFLAGS]");
173                 return -1;
174         }
175
176         return 0;
177 }
178
179
180 static int nl_set_encr(int ifindex, struct i802_driver_data *drv,
181                        const char *alg, const u8 *addr, int idx, const u8 *key,
182                        size_t key_len, int txkey)
183 {
184         struct nl_msg *msg;
185         int ret = -1;
186         int err = 0;
187
188         msg = nlmsg_alloc();
189         if (!msg)
190                 goto out;
191
192         if (strcmp(alg, "none") == 0) {
193                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
194                             0, NL80211_CMD_DEL_KEY, 0);
195         } else {
196                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
197                             0, NL80211_CMD_NEW_KEY, 0);
198                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
199                 if (strcmp(alg, "WEP") == 0) {
200                         if (key_len == 5)
201                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
202                                             0x000FAC01);
203                         else
204                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
205                                             0x000FAC05);
206                 } else if (strcmp(alg, "TKIP") == 0)
207                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
208                 else if (strcmp(alg, "CCMP") == 0)
209                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
210                 else if (strcmp(alg, "IGTK") == 0)
211                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC06);
212                 else {
213                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
214                                    "algorithm '%s'", __func__, alg);
215                         goto out;
216                 }
217         }
218
219         if (addr)
220                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
221         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
222         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
223
224         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
225             (err = nl_wait_for_ack(drv->nl_handle)) < 0) {
226                 if (err != -ENOENT) {
227                         ret = 0;
228                         goto out;
229                 }
230         }
231
232         /*
233          * If we need to set the default TX key we do that below,
234          * otherwise we're done here.
235          */
236         if (!txkey || addr) {
237                 ret = 0;
238                 goto out;
239         }
240
241         nlmsg_free(msg);
242
243         msg = nlmsg_alloc();
244         if (!msg)
245                 goto out;
246
247         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
248                     0, NL80211_CMD_SET_KEY, 0);
249         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
250         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
251 #ifdef NL80211_MFP_PENDING
252         if (strcmp(alg, "IGTK") == 0)
253                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
254         else
255                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
256 #else /* NL80211_MFP_PENDING */
257         NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
258 #endif /* NL80211_MFP_PENDING */
259
260         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
261             (err = nl_wait_for_ack(drv->nl_handle)) < 0) {
262                 if (err != -ENOENT) {
263                         ret = 0;
264                         goto out;
265                 }
266         }
267
268         ret = 0;
269
270  out:
271  nla_put_failure:
272         nlmsg_free(msg);
273         return ret;
274 }
275
276
277 static int i802_set_encryption(const char *iface, void *priv, const char *alg,
278                                const u8 *addr, int idx, const u8 *key,
279                                size_t key_len, int txkey)
280 {
281         struct i802_driver_data *drv = priv;
282         int ret;
283
284         ret = nl_set_encr(if_nametoindex(iface), drv, alg, addr, idx, key,
285                           key_len, txkey);
286         if (ret < 0)
287                 return ret;
288
289         if (strcmp(alg, "IGTK") == 0) {
290                 ret = nl_set_encr(drv->monitor_ifidx, drv, alg, addr, idx, key,
291                                   key_len, txkey);
292         }
293
294         return ret;
295 }
296
297
298 static inline int min_int(int a, int b)
299 {
300         if (a < b)
301                 return a;
302         return b;
303 }
304
305
306 static int get_key_handler(struct nl_msg *msg, void *arg)
307 {
308         struct nlattr *tb[NL80211_ATTR_MAX + 1];
309         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
310
311         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
312                   genlmsg_attrlen(gnlh, 0), NULL);
313
314         /*
315          * TODO: validate the key index and mac address!
316          * Otherwise, there's a race condition as soon as
317          * the kernel starts sending key notifications.
318          */
319
320         if (tb[NL80211_ATTR_KEY_SEQ])
321                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
322                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
323         return NL_SKIP;
324 }
325
326
327 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
328                            int idx, u8 *seq)
329 {
330         struct i802_driver_data *drv = priv;
331         struct nl_msg *msg;
332         struct nl_cb *cb = NULL;
333         int ret = -1;
334         int err = 0;
335         int finished = 0;
336
337         msg = nlmsg_alloc();
338         if (!msg)
339                 goto out;
340
341         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
342                     0, NL80211_CMD_GET_KEY, 0);
343
344         if (addr)
345                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
346         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
347         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
348
349         cb = nl_cb_clone(drv->nl_cb);
350         if (!cb)
351                 goto out;
352
353         memset(seq, 0, 6);
354
355         if (nl_send_auto_complete(drv->nl_handle, msg) < 0)
356                 goto out;
357
358         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, get_key_handler, seq);
359         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_wait_handler, &finished);
360
361         err = nl_recvmsgs(drv->nl_handle, cb);
362
363         if (!finished)
364                 err = nl_wait_for_ack(drv->nl_handle);
365
366         if (err < 0)
367                 goto out;
368
369         ret = 0;
370
371  out:
372         nl_cb_put(cb);
373  nla_put_failure:
374         nlmsg_free(msg);
375         return ret;
376 }
377
378
379 static int i802_set_rate_sets(void *priv, int *supp_rates, int *basic_rates,
380                               int mode)
381 {
382         return -1;
383 }
384
385
386 static int i802_set_ssid(const char *ifname, void *priv, const u8 *buf,
387                          int len)
388 {
389         struct i802_driver_data *drv = priv;
390         struct iwreq iwr;
391
392         memset(&iwr, 0, sizeof(iwr));
393         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
394         iwr.u.essid.flags = 1; /* SSID active */
395         iwr.u.essid.pointer = (caddr_t) buf;
396         iwr.u.essid.length = len;
397
398         if (ioctl(drv->ioctl_sock, SIOCSIWESSID, &iwr) < 0) {
399                 perror("ioctl[SIOCSIWESSID]");
400                 printf("len=%d\n", len);
401                 return -1;
402         }
403
404         return 0;
405 }
406
407
408 static int i802_send_frame(void *priv, const void *data, size_t len,
409                            int encrypt, int flags)
410 {
411         __u8 rtap_hdr[] = {
412                 0x00, 0x00, /* radiotap version */
413                 0x0e, 0x00, /* radiotap length */
414                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
415                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
416                 0x00,       /* padding */
417                 0x00, 0x00, /* RX and TX flags to indicate that */
418                 0x00, 0x00, /* this is the injected frame directly */
419         };
420         struct i802_driver_data *drv = priv;
421         struct iovec iov[2] = {
422                 {
423                         .iov_base = &rtap_hdr,
424                         .iov_len = sizeof(rtap_hdr),
425                 },
426                 {
427                         .iov_base = (void*)data,
428                         .iov_len = len,
429                 }
430         };
431         struct msghdr msg = {
432                 .msg_name = NULL,
433                 .msg_namelen = 0,
434                 .msg_iov = iov,
435                 .msg_iovlen = 2,
436                 .msg_control = NULL,
437                 .msg_controllen = 0,
438                 .msg_flags = 0,
439         };
440
441         if (encrypt)
442                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
443
444         return sendmsg(drv->monitor_sock, &msg, flags);
445 }
446
447 static int i802_send_mgmt_frame(void *priv, const void *data, size_t len,
448                                 int flags)
449 {
450         return i802_send_frame(priv, data, len, 1, flags);
451 }
452
453 /* Set kernel driver on given frequency (MHz) */
454 static int i802_set_freq(void *priv, int mode, int freq)
455 {
456         struct i802_driver_data *drv = priv;
457         struct iwreq iwr;
458
459         memset(&iwr, 0, sizeof(iwr));
460         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
461         iwr.u.freq.m = freq;
462         iwr.u.freq.e = 6;
463
464         if (ioctl(drv->ioctl_sock, SIOCSIWFREQ, &iwr) < 0) {
465                 perror("ioctl[SIOCSIWFREQ]");
466                 return -1;
467         }
468
469         return 0;
470 }
471
472
473 static int i802_set_rts(void *priv, int rts)
474 {
475         struct i802_driver_data *drv = priv;
476         struct iwreq iwr;
477
478         memset(&iwr, 0, sizeof(iwr));
479         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
480         iwr.u.rts.value = rts;
481         iwr.u.rts.fixed = 1;
482
483         if (ioctl(drv->ioctl_sock, SIOCSIWRTS, &iwr) < 0) {
484                 perror("ioctl[SIOCSIWRTS]");
485                 return -1;
486         }
487
488         return 0;
489 }
490
491
492 static int i802_get_rts(void *priv, int *rts)
493 {
494         struct i802_driver_data *drv = priv;
495         struct iwreq iwr;
496
497         memset(&iwr, 0, sizeof(iwr));
498         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
499
500         if (ioctl(drv->ioctl_sock, SIOCGIWRTS, &iwr) < 0) {
501                 perror("ioctl[SIOCGIWRTS]");
502                 return -1;
503         }
504
505         *rts = iwr.u.rts.value;
506
507         return 0;
508 }
509
510
511 static int i802_set_frag(void *priv, int frag)
512 {
513         struct i802_driver_data *drv = priv;
514         struct iwreq iwr;
515
516         memset(&iwr, 0, sizeof(iwr));
517         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
518         iwr.u.frag.value = frag;
519         iwr.u.frag.fixed = 1;
520
521         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
522                 perror("ioctl[SIOCSIWFRAG]");
523                 return -1;
524         }
525
526         return 0;
527 }
528
529
530 static int i802_get_frag(void *priv, int *frag)
531 {
532         struct i802_driver_data *drv = priv;
533         struct iwreq iwr;
534
535         memset(&iwr, 0, sizeof(iwr));
536         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
537
538         if (ioctl(drv->ioctl_sock, SIOCGIWFRAG, &iwr) < 0) {
539                 perror("ioctl[SIOCGIWFRAG]");
540                 return -1;
541         }
542
543         *frag = iwr.u.frag.value;
544
545         return 0;
546 }
547
548
549 static int i802_set_retry(void *priv, int short_retry, int long_retry)
550 {
551         struct i802_driver_data *drv = priv;
552         struct iwreq iwr;
553
554         memset(&iwr, 0, sizeof(iwr));
555         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
556
557         iwr.u.retry.value = short_retry;
558         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
559         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
560                 perror("ioctl[SIOCSIWRETRY(short)]");
561                 return -1;
562         }
563
564         iwr.u.retry.value = long_retry;
565         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
566         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
567                 perror("ioctl[SIOCSIWRETRY(long)]");
568                 return -1;
569         }
570
571         return 0;
572 }
573
574
575 static int i802_get_retry(void *priv, int *short_retry, int *long_retry)
576 {
577         struct i802_driver_data *drv = priv;
578         struct iwreq iwr;
579
580         memset(&iwr, 0, sizeof(iwr));
581         os_strlcpy(iwr.ifr_name, drv->hapd->conf->iface, IFNAMSIZ);
582
583         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
584         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
585                 perror("ioctl[SIOCGIWFRAG(short)]");
586                 return -1;
587         }
588         *short_retry = iwr.u.retry.value;
589
590         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
591         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
592                 perror("ioctl[SIOCGIWFRAG(long)]");
593                 return -1;
594         }
595         *long_retry = iwr.u.retry.value;
596
597         return 0;
598 }
599
600
601 static int i802_flush(void *priv)
602 {
603         struct i802_driver_data *drv = priv;
604         struct nl_msg *msg;
605         int ret = -1;
606
607         msg = nlmsg_alloc();
608         if (!msg)
609                 goto out;
610
611         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
612                     0, NL80211_CMD_DEL_STATION, 0);
613
614         /*
615          * XXX: FIX! this needs to flush all VLANs too
616          */
617         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
618                     if_nametoindex(drv->iface));
619
620         ret = 0;
621
622         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
623             nl_wait_for_ack(drv->nl_handle) < 0) {
624                 ret = -1;
625         }
626
627  nla_put_failure:
628         nlmsg_free(msg);
629
630  out:
631         return ret;
632 }
633
634
635 static int get_sta_handler(struct nl_msg *msg, void *arg)
636 {
637         struct nlattr *tb[NL80211_ATTR_MAX + 1];
638         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
639         struct hostap_sta_driver_data *data = arg;
640         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
641         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
642                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
643                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
644                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
645         };
646
647         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
648                   genlmsg_attrlen(gnlh, 0), NULL);
649
650         /*
651          * TODO: validate the interface and mac address!
652          * Otherwise, there's a race condition as soon as
653          * the kernel starts sending station notifications.
654          */
655
656         if (!tb[NL80211_ATTR_STA_INFO]) {
657                 wpa_printf(MSG_DEBUG, "sta stats missing!");
658                 return NL_SKIP;
659         }
660         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
661                              tb[NL80211_ATTR_STA_INFO],
662                              stats_policy)) {
663                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
664                 return NL_SKIP;
665         }
666
667         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
668                 data->inactive_msec =
669                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
670         if (stats[NL80211_STA_INFO_RX_BYTES])
671                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
672         if (stats[NL80211_STA_INFO_TX_BYTES])
673                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
674
675         return NL_SKIP;
676 }
677
678 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
679                               const u8 *addr)
680 {
681         struct i802_driver_data *drv = priv;
682         struct nl_msg *msg;
683         struct nl_cb *cb = NULL;
684         int ret = -1;
685         int err = 0;
686         int finished = 0;
687
688         msg = nlmsg_alloc();
689         if (!msg)
690                 goto out;
691
692         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
693                     0, NL80211_CMD_GET_STATION, 0);
694
695         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
696         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
697
698         cb = nl_cb_clone(drv->nl_cb);
699         if (!cb)
700                 goto out;
701
702         if (nl_send_auto_complete(drv->nl_handle, msg) < 0)
703                 goto out;
704
705         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, get_sta_handler, data);
706         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_wait_handler, &finished);
707
708         err = nl_recvmsgs(drv->nl_handle, cb);
709
710         if (!finished)
711                 err = nl_wait_for_ack(drv->nl_handle);
712
713         if (err < 0)
714                 goto out;
715
716         ret = 0;
717
718  out:
719         nl_cb_put(cb);
720  nla_put_failure:
721         nlmsg_free(msg);
722         return ret;
723
724 }
725
726
727 static int i802_send_eapol(void *priv, const u8 *addr, const u8 *data,
728                            size_t data_len, int encrypt, const u8 *own_addr)
729 {
730         struct i802_driver_data *drv = priv;
731         struct ieee80211_hdr *hdr;
732         size_t len;
733         u8 *pos;
734         int res;
735 #if 0 /* FIX */
736         int qos = sta->flags & WLAN_STA_WME;
737 #else
738         int qos = 0;
739 #endif
740
741         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
742                 data_len;
743         hdr = os_zalloc(len);
744         if (hdr == NULL) {
745                 printf("malloc() failed for i802_send_data(len=%lu)\n",
746                        (unsigned long) len);
747                 return -1;
748         }
749
750         hdr->frame_control =
751                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
752         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
753         if (encrypt)
754                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
755 #if 0 /* To be enabled if qos determination is added above */
756         if (qos) {
757                 hdr->frame_control |=
758                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
759         }
760 #endif
761
762         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
763         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
764         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
765         pos = (u8 *) (hdr + 1);
766
767 #if 0 /* To be enabled if qos determination is added above */
768         if (qos) {
769                 /* add an empty QoS header if needed */
770                 pos[0] = 0;
771                 pos[1] = 0;
772                 pos += 2;
773         }
774 #endif
775
776         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
777         pos += sizeof(rfc1042_header);
778         WPA_PUT_BE16(pos, ETH_P_PAE);
779         pos += 2;
780         memcpy(pos, data, data_len);
781
782         res = i802_send_frame(drv, (u8 *) hdr, len, encrypt, 0);
783         free(hdr);
784
785         if (res < 0) {
786                 perror("i802_send_eapol: send");
787                 printf("i802_send_eapol - packet len: %lu - failed\n",
788                        (unsigned long) len);
789         }
790
791         return res;
792 }
793
794
795 static int i802_sta_add2(const char *ifname, void *priv,
796                          struct hostapd_sta_add_params *params)
797 {
798         struct i802_driver_data *drv = priv;
799         struct nl_msg *msg;
800         int ret = -1;
801
802         msg = nlmsg_alloc();
803         if (!msg)
804                 goto out;
805
806         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
807                     0, NL80211_CMD_NEW_STATION, 0);
808
809         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
810                     if_nametoindex(drv->iface));
811         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
812         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
813         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
814                 params->supp_rates);
815         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
816                     params->listen_interval);
817
818 #ifdef CONFIG_IEEE80211N
819 #ifdef NL80211_ATTR_HT_CAPABILITY
820         if (params->ht_capabilities) {
821                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
822                         params->ht_capabilities->length,
823                         &params->ht_capabilities->data);
824         }
825 #endif /* NL80211_ATTR_HT_CAPABILITY */
826 #endif /* CONFIG_IEEE80211N */
827
828         ret = nl_send_auto_complete(drv->nl_handle, msg);
829         if (ret < 0)
830                 goto nla_put_failure;
831
832         ret = nl_wait_for_ack(drv->nl_handle);
833         /* ignore EEXIST, this happens if a STA associates while associated */
834         if (ret == -EEXIST || ret >= 0)
835                 ret = 0;
836
837  nla_put_failure:
838         nlmsg_free(msg);
839
840  out:
841         return ret;
842 }
843
844
845 static int i802_sta_remove(void *priv, const u8 *addr)
846 {
847         struct i802_driver_data *drv = priv;
848         struct nl_msg *msg;
849         int ret = -1;
850
851         msg = nlmsg_alloc();
852         if (!msg)
853                 goto out;
854
855         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
856                     0, NL80211_CMD_DEL_STATION, 0);
857
858         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
859                     if_nametoindex(drv->iface));
860         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
861
862         ret = 0;
863
864         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
865             nl_wait_for_ack(drv->nl_handle) < 0) {
866                 ret = -1;
867         }
868
869  nla_put_failure:
870         nlmsg_free(msg);
871
872  out:
873         return ret;
874 }
875
876
877 static int i802_sta_set_flags(void *priv, const u8 *addr,
878                               int total_flags, int flags_or, int flags_and)
879 {
880         struct i802_driver_data *drv = priv;
881         struct nl_msg *msg, *flags = NULL;
882         int ret = -1;
883
884         msg = nlmsg_alloc();
885         if (!msg)
886                 goto out;
887
888         flags = nlmsg_alloc();
889         if (!flags)
890                 goto free_msg;
891
892         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
893                     0, NL80211_CMD_SET_STATION, 0);
894
895         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
896                     if_nametoindex(drv->iface));
897         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
898
899         if (total_flags & WLAN_STA_AUTHORIZED || !drv->ieee802_1x_active)
900                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
901
902         if (total_flags & WLAN_STA_WME)
903                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
904
905         if (total_flags & WLAN_STA_SHORT_PREAMBLE)
906                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
907
908 #ifdef NL80211_MFP_PENDING
909         if (total_flags & WLAN_STA_MFP)
910                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
911 #endif /* NL80211_MFP_PENDING */
912
913         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
914                 goto nla_put_failure;
915
916         ret = 0;
917
918         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
919             nl_wait_for_ack(drv->nl_handle) < 0) {
920                 ret = -1;
921         }
922
923  nla_put_failure:
924         nlmsg_free(flags);
925
926  free_msg:
927         nlmsg_free(msg);
928
929  out:
930         return ret;
931 }
932
933
934 static int i802_set_channel_flag(void *priv, int mode, int chan, int flag,
935                                  unsigned char power_level,
936                                  unsigned char antenna_max)
937 {
938         return -1;
939 }
940
941
942 static int i802_set_regulatory_domain(void *priv, unsigned int rd)
943 {
944         return -1;
945 }
946
947
948 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
949                                     int cw_min, int cw_max, int burst_time)
950 {
951         return -1;
952 }
953
954
955 static void nl80211_remove_iface(struct i802_driver_data *drv, int ifidx)
956 {
957         struct nl_msg *msg;
958
959         /* stop listening for EAPOL on this interface */
960         del_ifidx(drv, ifidx);
961
962         msg = nlmsg_alloc();
963         if (!msg)
964                 goto nla_put_failure;
965
966         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
967                     0, NL80211_CMD_DEL_INTERFACE, 0);
968         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
969         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
970             nl_wait_for_ack(drv->nl_handle) < 0)
971         nla_put_failure:
972                 printf("Failed to remove interface.\n");
973         nlmsg_free(msg);
974 }
975
976
977 static int nl80211_create_iface(struct i802_driver_data *drv,
978                                 const char *ifname,
979                                 enum nl80211_iftype iftype,
980                                 const u8 *addr)
981 {
982         struct nl_msg *msg, *flags = NULL;
983         int ifidx;
984         struct ifreq ifreq;
985         struct iwreq iwr;
986
987         msg = nlmsg_alloc();
988         if (!msg)
989                 return -1;
990
991         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
992                     0, NL80211_CMD_NEW_INTERFACE, 0);
993         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
994                     if_nametoindex(drv->hapd->conf->iface));
995         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
996         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
997
998         if (iftype == NL80211_IFTYPE_MONITOR) {
999                 int err;
1000
1001                 flags = nlmsg_alloc();
1002                 if (!flags)
1003                         goto nla_put_failure;
1004
1005                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1006
1007                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1008
1009                 nlmsg_free(flags);
1010
1011                 if (err)
1012                         goto nla_put_failure;
1013         }
1014
1015         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1016             nl_wait_for_ack(drv->nl_handle) < 0) {
1017  nla_put_failure:
1018                 printf("Failed to create interface %s.\n", ifname);
1019                 nlmsg_free(msg);
1020                 return -1;
1021         }
1022
1023         nlmsg_free(msg);
1024
1025         ifidx = if_nametoindex(ifname);
1026
1027         if (ifidx <= 0)
1028                 return -1;
1029
1030         /* start listening for EAPOL on this interface */
1031         add_ifidx(drv, ifidx);
1032
1033         if (addr) {
1034                 switch (iftype) {
1035                 case NL80211_IFTYPE_AP:
1036                         os_strlcpy(ifreq.ifr_name, ifname, IFNAMSIZ);
1037                         memcpy(ifreq.ifr_hwaddr.sa_data, addr, ETH_ALEN);
1038                         ifreq.ifr_hwaddr.sa_family = ARPHRD_ETHER;
1039
1040                         if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifreq)) {
1041                                 nl80211_remove_iface(drv, ifidx);
1042                                 return -1;
1043                         }
1044                         break;
1045                 case NL80211_IFTYPE_WDS:
1046                         memset(&iwr, 0, sizeof(iwr));
1047                         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1048                         iwr.u.addr.sa_family = ARPHRD_ETHER;
1049                         memcpy(iwr.u.addr.sa_data, addr, ETH_ALEN);
1050                         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr))
1051                                 return -1;
1052                         break;
1053                 default:
1054                         /* nothing */
1055                         break;
1056                 }
1057         }
1058
1059         return ifidx;
1060 }
1061
1062
1063 static int i802_bss_add(void *priv, const char *ifname, const u8 *bssid)
1064 {
1065         int ifidx;
1066
1067         /*
1068          * The kernel supports that when the low-level driver does,
1069          * but we currently don't because we need per-BSS data that
1070          * currently we can't handle easily.
1071          */
1072         return -1;
1073
1074         ifidx = nl80211_create_iface(priv, ifname, NL80211_IFTYPE_AP, bssid);
1075         if (ifidx < 0)
1076                 return -1;
1077         if (hostapd_set_iface_flags(priv, ifname, 1)) {
1078                 nl80211_remove_iface(priv, ifidx);
1079                 return -1;
1080         }
1081         return 0;
1082 }
1083
1084
1085 static int i802_bss_remove(void *priv, const char *ifname)
1086 {
1087         nl80211_remove_iface(priv, if_nametoindex(ifname));
1088         return 0;
1089 }
1090
1091
1092 static int i802_set_beacon(const char *iface, void *priv,
1093                            u8 *head, size_t head_len,
1094                            u8 *tail, size_t tail_len)
1095 {
1096         struct i802_driver_data *drv = priv;
1097         struct nl_msg *msg;
1098         u8 cmd = NL80211_CMD_NEW_BEACON;
1099         int ret = -1;
1100
1101         msg = nlmsg_alloc();
1102         if (!msg)
1103                 goto out;
1104
1105         if (drv->beacon_set)
1106                 cmd = NL80211_CMD_SET_BEACON;
1107
1108         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1109                     0, cmd, 0);
1110         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
1111         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
1112         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
1113         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
1114
1115         if (!drv->dtim_period)
1116                 drv->dtim_period = 2;
1117         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, drv->dtim_period);
1118
1119         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1120             nl_wait_for_ack(drv->nl_handle) < 0)
1121                 goto out;
1122
1123         ret = 0;
1124
1125         drv->beacon_set = 1;
1126
1127  out:
1128  nla_put_failure:
1129         nlmsg_free(msg);
1130         return ret;
1131 }
1132
1133
1134 static int i802_del_beacon(struct i802_driver_data *drv)
1135 {
1136         struct nl_msg *msg;
1137         int ret = -1;
1138
1139         msg = nlmsg_alloc();
1140         if (!msg)
1141                 goto out;
1142
1143         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1144                     0, NL80211_CMD_DEL_BEACON, 0);
1145         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1146
1147         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1148             nl_wait_for_ack(drv->nl_handle) < 0)
1149                 goto out;
1150
1151         ret = 0;
1152
1153  out:
1154  nla_put_failure:
1155         nlmsg_free(msg);
1156         return ret;
1157 }
1158
1159
1160 static int i802_set_ieee8021x(const char *ifname, void *priv, int enabled)
1161 {
1162         struct i802_driver_data *drv = priv;
1163
1164         /*
1165          * FIXME: This needs to be per interface (BSS)
1166          */
1167         drv->ieee802_1x_active = enabled;
1168         return 0;
1169 }
1170
1171
1172 static int i802_set_privacy(const char *ifname, void *priv, int enabled)
1173 {
1174         struct i802_driver_data *drv = priv;
1175         struct iwreq iwr;
1176
1177         memset(&iwr, 0, sizeof(iwr));
1178
1179         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1180         iwr.u.param.flags = IW_AUTH_PRIVACY_INVOKED;
1181         iwr.u.param.value = enabled;
1182
1183         ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr);
1184
1185         /* ignore errors, the kernel/driver might not care */
1186         return 0;
1187 }
1188
1189
1190 static int i802_set_internal_bridge(void *priv, int value)
1191 {
1192         return -1;
1193 }
1194
1195
1196 static int i802_set_beacon_int(void *priv, int value)
1197 {
1198         struct i802_driver_data *drv = priv;
1199         struct nl_msg *msg;
1200         int ret = -1;
1201
1202         drv->beacon_int = value;
1203
1204         if (!drv->beacon_set)
1205                 return 0;
1206
1207         msg = nlmsg_alloc();
1208         if (!msg)
1209                 goto out;
1210
1211         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1212                     0, NL80211_CMD_SET_BEACON, 0);
1213         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1214
1215         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
1216
1217         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1218             nl_wait_for_ack(drv->nl_handle) < 0)
1219                 goto out;
1220
1221         ret = 0;
1222
1223  out:
1224  nla_put_failure:
1225         nlmsg_free(msg);
1226         return ret;
1227 }
1228
1229
1230 static int i802_set_dtim_period(const char *iface, void *priv, int value)
1231 {
1232         struct i802_driver_data *drv = priv;
1233         struct nl_msg *msg;
1234         int ret = -1;
1235
1236         msg = nlmsg_alloc();
1237         if (!msg)
1238                 goto out;
1239
1240         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1241                     0, NL80211_CMD_SET_BEACON, 0);
1242         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
1243
1244         drv->dtim_period = value;
1245         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, drv->dtim_period);
1246
1247         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1248             nl_wait_for_ack(drv->nl_handle) < 0)
1249                 goto out;
1250
1251         ret = 0;
1252
1253  out:
1254  nla_put_failure:
1255         nlmsg_free(msg);
1256         return ret;
1257 }
1258
1259
1260 static int i802_set_bss(void *priv, int cts, int preamble, int slot)
1261 {
1262 #ifdef NL80211_CMD_SET_BSS
1263         struct i802_driver_data *drv = priv;
1264         struct nl_msg *msg;
1265         int ret = -1;
1266
1267         msg = nlmsg_alloc();
1268         if (!msg)
1269                 goto out;
1270
1271         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1272                     NL80211_CMD_SET_BSS, 0);
1273
1274         if (cts >= 0)
1275                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
1276         if (preamble >= 0)
1277                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
1278         if (slot >= 0)
1279                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
1280
1281         ret = 0;
1282
1283         /* TODO: multi-BSS support */
1284         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1285
1286         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1287             nl_wait_for_ack(drv->nl_handle) < 0) {
1288                 ret = -1;
1289         }
1290
1291 nla_put_failure:
1292         nlmsg_free(msg);
1293
1294 out:
1295         return ret;
1296 #else /* NL80211_CMD_SET_BSS */
1297         return -1;
1298 #endif /* NL80211_CMD_SET_BSS */
1299 }
1300
1301
1302 static int i802_set_cts_protect(void *priv, int value)
1303 {
1304         return i802_set_bss(priv, value, -1, -1);
1305 }
1306
1307
1308 static int i802_set_preamble(void *priv, int value)
1309 {
1310         return i802_set_bss(priv, -1, value, -1);
1311 }
1312
1313
1314 static int i802_set_short_slot_time(void *priv, int value)
1315 {
1316         return i802_set_bss(priv, -1, -1, value);
1317 }
1318
1319
1320 static enum nl80211_iftype i802_if_type(enum hostapd_driver_if_type type)
1321 {
1322         switch (type) {
1323         case HOSTAPD_IF_VLAN:
1324                 return NL80211_IFTYPE_AP_VLAN;
1325         case HOSTAPD_IF_WDS:
1326                 return NL80211_IFTYPE_WDS;
1327         }
1328         return -1;
1329 }
1330
1331
1332 static int i802_if_add(const char *iface, void *priv,
1333                        enum hostapd_driver_if_type type, char *ifname,
1334                        const u8 *addr)
1335 {
1336         if (nl80211_create_iface(priv, ifname, i802_if_type(type), addr) < 0)
1337                 return -1;
1338         return 0;
1339 }
1340
1341
1342 static int i802_if_update(void *priv, enum hostapd_driver_if_type type,
1343                           char *ifname, const u8 *addr)
1344 {
1345         /* unused at the moment */
1346         return -1;
1347 }
1348
1349
1350 static int i802_if_remove(void *priv, enum hostapd_driver_if_type type,
1351                           const char *ifname, const u8 *addr)
1352 {
1353         nl80211_remove_iface(priv, if_nametoindex(ifname));
1354         return 0;
1355 }
1356
1357
1358 struct phy_info_arg {
1359         u16 *num_modes;
1360         struct hostapd_hw_modes *modes;
1361         int error;
1362 };
1363
1364 static int phy_info_handler(struct nl_msg *msg, void *arg)
1365 {
1366         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1367         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1368         struct phy_info_arg *phy_info = arg;
1369
1370         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1371
1372         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1373         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1374                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1375                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1376                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1377                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
1378                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1379         };
1380
1381         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1382         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1383                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1384                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
1385         };
1386
1387         struct nlattr *nl_band;
1388         struct nlattr *nl_freq;
1389         struct nlattr *nl_rate;
1390         int rem_band, rem_freq, rem_rate;
1391         struct hostapd_hw_modes *mode;
1392         int idx, mode_is_set;
1393
1394         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1395                   genlmsg_attrlen(gnlh, 0), NULL);
1396
1397         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1398                 return NL_SKIP;
1399
1400         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
1401                 mode = realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
1402                 if (!mode)
1403                         return NL_SKIP;
1404                 phy_info->modes = mode;
1405
1406                 mode_is_set = 0;
1407
1408                 mode = &phy_info->modes[*(phy_info->num_modes)];
1409                 memset(mode, 0, sizeof(*mode));
1410                 *(phy_info->num_modes) += 1;
1411
1412                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1413                           nla_len(nl_band), NULL);
1414
1415                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1416                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1417                                   nla_len(nl_freq), freq_policy);
1418                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1419                                 continue;
1420                         mode->num_channels++;
1421                 }
1422
1423                 mode->channels = calloc(mode->num_channels, sizeof(struct hostapd_channel_data));
1424                 if (!mode->channels)
1425                         return NL_SKIP;
1426
1427                 idx = 0;
1428
1429                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1430                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1431                                   nla_len(nl_freq), freq_policy);
1432                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1433                                 continue;
1434
1435                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1436                         mode->channels[idx].flag |= HOSTAPD_CHAN_W_SCAN |
1437                                                     HOSTAPD_CHAN_W_ACTIVE_SCAN |
1438                                                     HOSTAPD_CHAN_W_IBSS;
1439
1440                         if (!mode_is_set) {
1441                                 /* crude heuristic */
1442                                 if (mode->channels[idx].freq < 4000)
1443                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
1444                                 else
1445                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
1446                                 mode_is_set = 1;
1447                         }
1448
1449                         /* crude heuristic */
1450                         if (mode->channels[idx].freq < 4000)
1451                                 if (mode->channels[idx].freq == 2848)
1452                                         mode->channels[idx].chan = 14;
1453                                 else
1454                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
1455                         else
1456                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
1457
1458                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1459                                 mode->channels[idx].flag &= ~HOSTAPD_CHAN_W_SCAN;
1460                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
1461                                 mode->channels[idx].flag &= ~HOSTAPD_CHAN_W_ACTIVE_SCAN;
1462                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
1463                                 mode->channels[idx].flag &= ~HOSTAPD_CHAN_W_IBSS;
1464                         idx++;
1465                 }
1466
1467                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1468                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1469                                   nla_len(nl_rate), rate_policy);
1470                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1471                                 continue;
1472                         mode->num_rates++;
1473                 }
1474
1475                 mode->rates = calloc(mode->num_rates, sizeof(struct hostapd_rate_data));
1476                 if (!mode->rates)
1477                         return NL_SKIP;
1478
1479                 idx = 0;
1480
1481                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1482                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1483                                   nla_len(nl_rate), rate_policy);
1484                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1485                                 continue;
1486                         mode->rates[idx].rate = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
1487
1488                         /* crude heuristic */
1489                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
1490                             mode->rates[idx].rate > 200)
1491                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
1492
1493                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
1494                                 mode->rates[idx].flags |= HOSTAPD_RATE_PREAMBLE2;
1495
1496                         idx++;
1497                 }
1498         }
1499
1500         phy_info->error = 0;
1501
1502         return NL_SKIP;
1503 }
1504
1505 static struct hostapd_hw_modes *i802_get_hw_feature_data(void *priv,
1506                                                          u16 *num_modes,
1507                                                          u16 *flags)
1508 {
1509         struct i802_driver_data *drv = priv;
1510         struct nl_msg *msg;
1511         int err = -1;
1512         struct nl_cb *cb = NULL;
1513         int finished = 0;
1514         struct phy_info_arg result = {
1515                 .num_modes = num_modes,
1516                 .modes = NULL,
1517                 .error = 1,
1518         };
1519
1520         *num_modes = 0;
1521         *flags = 0;
1522
1523         msg = nlmsg_alloc();
1524         if (!msg)
1525                 return NULL;
1526
1527         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1528                     0, NL80211_CMD_GET_WIPHY, 0);
1529
1530         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1531
1532         cb = nl_cb_clone(drv->nl_cb);
1533         if (!cb)
1534                 goto out;
1535
1536         if (nl_send_auto_complete(drv->nl_handle, msg) < 0)
1537                 goto out;
1538
1539         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, phy_info_handler, &result);
1540         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_wait_handler, &finished);
1541
1542         err = nl_recvmsgs(drv->nl_handle, cb);
1543
1544         if (!finished)
1545                 err = nl_wait_for_ack(drv->nl_handle);
1546
1547         if (err < 0 || result.error) {
1548                 hostapd_free_hw_features(result.modes, *num_modes);
1549                 result.modes = NULL;
1550         }
1551
1552  out:
1553         nl_cb_put(cb);
1554  nla_put_failure:
1555         if (err)
1556                 fprintf(stderr, "failed to get information: %d\n", err);
1557         nlmsg_free(msg);
1558         return result.modes;
1559 }
1560
1561
1562 static int i802_set_sta_vlan(void *priv, const u8 *addr,
1563                              const char *ifname, int vlan_id)
1564 {
1565         struct i802_driver_data *drv = priv;
1566         struct nl_msg *msg;
1567         int ret = -1;
1568
1569         msg = nlmsg_alloc();
1570         if (!msg)
1571                 goto out;
1572
1573         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1574                     0, NL80211_CMD_SET_STATION, 0);
1575
1576         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1577                     if_nametoindex(drv->iface));
1578         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1579         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1580                     if_nametoindex(ifname));
1581
1582         ret = 0;
1583
1584         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1585             (errno = nl_wait_for_ack(drv->nl_handle) < 0)) {
1586                 ret = -1;
1587         }
1588
1589  nla_put_failure:
1590         nlmsg_free(msg);
1591
1592  out:
1593         return ret;
1594 }
1595
1596
1597 static void handle_unknown_sta(struct hostapd_data *hapd, u8 *ta)
1598 {
1599         struct sta_info *sta;
1600
1601         sta = ap_get_sta(hapd, ta);
1602         if (!sta || !(sta->flags & WLAN_STA_ASSOC)) {
1603                 printf("Data/PS-poll frame from not associated STA "
1604                        MACSTR "\n", MAC2STR(ta));
1605                 if (sta && (sta->flags & WLAN_STA_AUTH))
1606                         hostapd_sta_disassoc(
1607                                 hapd, ta,
1608                                 WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1609                 else
1610                         hostapd_sta_deauth(
1611                                 hapd, ta,
1612                                 WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1613         }
1614 }
1615
1616
1617 static void handle_tx_callback(struct hostapd_data *hapd, u8 *buf, size_t len,
1618                                int ok)
1619 {
1620         struct ieee80211_hdr *hdr;
1621         u16 fc, type, stype;
1622         struct sta_info *sta;
1623
1624         hdr = (struct ieee80211_hdr *) buf;
1625         fc = le_to_host16(hdr->frame_control);
1626
1627         type = WLAN_FC_GET_TYPE(fc);
1628         stype = WLAN_FC_GET_STYPE(fc);
1629
1630         switch (type) {
1631         case WLAN_FC_TYPE_MGMT:
1632                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
1633                            ok ? "ACK" : "fail");
1634                 ieee802_11_mgmt_cb(hapd, buf, len, stype, ok);
1635                 break;
1636         case WLAN_FC_TYPE_CTRL:
1637                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
1638                            ok ? "ACK" : "fail");
1639                 break;
1640         case WLAN_FC_TYPE_DATA:
1641                 wpa_printf(MSG_DEBUG, "DATA (TX callback) %s",
1642                            ok ? "ACK" : "fail");
1643                 sta = ap_get_sta(hapd, hdr->addr1);
1644                 if (sta && sta->flags & WLAN_STA_PENDING_POLL) {
1645                         wpa_printf(MSG_DEBUG, "STA " MACSTR " %s pending "
1646                                    "activity poll", MAC2STR(sta->addr),
1647                                    ok ? "ACKed" : "did not ACK");
1648                         if (ok)
1649                                 sta->flags &= ~WLAN_STA_PENDING_POLL;
1650                 }
1651                 if (sta)
1652                         ieee802_1x_tx_status(hapd, sta, buf, len, ok);
1653                 break;
1654         default:
1655                 printf("unknown TX callback frame type %d\n", type);
1656                 break;
1657         }
1658 }
1659
1660
1661 static void handle_frame(struct hostapd_iface *iface, u8 *buf, size_t len,
1662                          struct hostapd_frame_info *hfi,
1663                          enum ieee80211_msg_type msg_type)
1664 {
1665         struct ieee80211_hdr *hdr;
1666         u16 fc, type, stype;
1667         size_t data_len = len;
1668         struct hostapd_data *hapd = NULL;
1669         int broadcast_bssid = 0;
1670         size_t i;
1671         u8 *bssid;
1672
1673         /*
1674          * PS-Poll frames are 16 bytes. All other frames are
1675          * 24 bytes or longer.
1676          */
1677         if (len < 16)
1678                 return;
1679
1680         hdr = (struct ieee80211_hdr *) buf;
1681         fc = le_to_host16(hdr->frame_control);
1682
1683         type = WLAN_FC_GET_TYPE(fc);
1684         stype = WLAN_FC_GET_STYPE(fc);
1685
1686         switch (type) {
1687         case WLAN_FC_TYPE_DATA:
1688                 if (len < 24)
1689                         return;
1690                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
1691                 case WLAN_FC_TODS:
1692                         bssid = hdr->addr1;
1693                         break;
1694                 default:
1695                         /* discard */
1696                         return;
1697                 }
1698                 break;
1699         case WLAN_FC_TYPE_CTRL:
1700                 /* discard non-ps-poll frames */
1701                 if (stype != WLAN_FC_STYPE_PSPOLL)
1702                         return;
1703                 bssid = hdr->addr1;
1704                 break;
1705         case WLAN_FC_TYPE_MGMT:
1706                 bssid = hdr->addr3;
1707                 break;
1708         default:
1709                 /* discard */
1710                 return;
1711         }
1712
1713         /* find interface frame belongs to */
1714         for (i = 0; i < iface->num_bss; i++) {
1715                 if (memcmp(bssid, iface->bss[i]->own_addr, ETH_ALEN) == 0) {
1716                         hapd = iface->bss[i];
1717                         break;
1718                 }
1719         }
1720
1721         if (hapd == NULL) {
1722                 hapd = iface->bss[0];
1723
1724                 if (bssid[0] != 0xff || bssid[1] != 0xff ||
1725                     bssid[2] != 0xff || bssid[3] != 0xff ||
1726                     bssid[4] != 0xff || bssid[5] != 0xff) {
1727                         /*
1728                          * Unknown BSSID - drop frame if this is not from
1729                          * passive scanning or a beacon (at least ProbeReq
1730                          * frames to other APs may be allowed through RX
1731                          * filtering in the wlan hw/driver)
1732                          */
1733                         if ((type != WLAN_FC_TYPE_MGMT ||
1734                              stype != WLAN_FC_STYPE_BEACON))
1735                                 return;
1736                 } else
1737                         broadcast_bssid = 1;
1738         }
1739
1740         switch (msg_type) {
1741         case ieee80211_msg_normal:
1742                 /* continue processing */
1743                 break;
1744         case ieee80211_msg_tx_callback_ack:
1745                 handle_tx_callback(hapd, buf, data_len, 1);
1746                 return;
1747         case ieee80211_msg_tx_callback_fail:
1748                 handle_tx_callback(hapd, buf, data_len, 0);
1749                 return;
1750         }
1751
1752         switch (type) {
1753         case WLAN_FC_TYPE_MGMT:
1754                 if (stype != WLAN_FC_STYPE_BEACON &&
1755                     stype != WLAN_FC_STYPE_PROBE_REQ)
1756                         wpa_printf(MSG_MSGDUMP, "MGMT");
1757                 if (broadcast_bssid) {
1758                         for (i = 0; i < iface->num_bss; i++)
1759                                 ieee802_11_mgmt(iface->bss[i], buf, data_len,
1760                                                 stype, hfi);
1761                 } else
1762                         ieee802_11_mgmt(hapd, buf, data_len, stype, hfi);
1763                 break;
1764         case WLAN_FC_TYPE_CTRL:
1765                 /* can only get here with PS-Poll frames */
1766                 wpa_printf(MSG_DEBUG, "CTRL");
1767                 handle_unknown_sta(hapd, hdr->addr2);
1768                 break;
1769         case WLAN_FC_TYPE_DATA:
1770                 wpa_printf(MSG_DEBUG, "DATA");
1771                 handle_unknown_sta(hapd, hdr->addr2);
1772                 break;
1773         }
1774 }
1775
1776
1777 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
1778 {
1779         struct i802_driver_data *drv = eloop_ctx;
1780         struct hostapd_data *hapd = drv->hapd;
1781         struct sockaddr_ll lladdr;
1782         unsigned char buf[3000];
1783         int len;
1784         socklen_t fromlen = sizeof(lladdr);
1785
1786         len = recvfrom(sock, buf, sizeof(buf), 0,
1787                        (struct sockaddr *)&lladdr, &fromlen);
1788         if (len < 0) {
1789                 perror("recv");
1790                 return;
1791         }
1792
1793         if (have_ifidx(drv, lladdr.sll_ifindex))
1794                 ieee802_1x_receive(hapd, lladdr.sll_addr, buf, len);
1795 }
1796
1797
1798 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1799 {
1800         struct i802_driver_data *drv = eloop_ctx;
1801         int len;
1802         unsigned char buf[3000];
1803         struct hostapd_data *hapd = drv->hapd;
1804         struct ieee80211_radiotap_iterator iter;
1805         int ret;
1806         struct hostapd_frame_info hfi;
1807         int injected = 0, failed = 0, msg_type, rxflags = 0;
1808
1809         len = recv(sock, buf, sizeof(buf), 0);
1810         if (len < 0) {
1811                 perror("recv");
1812                 return;
1813         }
1814
1815         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
1816                 printf("received invalid radiotap frame\n");
1817                 return;
1818         }
1819
1820         memset(&hfi, 0, sizeof(hfi));
1821
1822         while (1) {
1823                 ret = ieee80211_radiotap_iterator_next(&iter);
1824                 if (ret == -ENOENT)
1825                         break;
1826                 if (ret) {
1827                         printf("received invalid radiotap frame (%d)\n", ret);
1828                         return;
1829                 }
1830                 switch (iter.this_arg_index) {
1831                 case IEEE80211_RADIOTAP_FLAGS:
1832                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1833                                 len -= 4;
1834                         break;
1835                 case IEEE80211_RADIOTAP_RX_FLAGS:
1836                         rxflags = 1;
1837                         break;
1838                 case IEEE80211_RADIOTAP_TX_FLAGS:
1839                         injected = 1;
1840                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
1841                                         IEEE80211_RADIOTAP_F_TX_FAIL;
1842                         break;
1843                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1844                         break;
1845                 case IEEE80211_RADIOTAP_CHANNEL:
1846                         /* TODO convert from freq/flags to channel number
1847                         hfi.channel = XXX;
1848                         hfi.phytype = XXX;
1849                          */
1850                         break;
1851                 case IEEE80211_RADIOTAP_RATE:
1852                         hfi.datarate = *iter.this_arg * 5;
1853                         break;
1854                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1855                         hfi.ssi_signal = *iter.this_arg;
1856                         break;
1857                 }
1858         }
1859
1860         if (rxflags && injected)
1861                 return;
1862
1863         if (!injected)
1864                 msg_type = ieee80211_msg_normal;
1865         else if (failed)
1866                 msg_type = ieee80211_msg_tx_callback_fail;
1867         else
1868                 msg_type = ieee80211_msg_tx_callback_ack;
1869
1870         handle_frame(hapd->iface, buf + iter.max_length,
1871                      len - iter.max_length, &hfi, msg_type);
1872 }
1873
1874
1875 static int nl80211_create_monitor_interface(struct i802_driver_data *drv)
1876 {
1877         char buf[IFNAMSIZ];
1878         struct sockaddr_ll ll;
1879         int optval;
1880         socklen_t optlen;
1881
1882         snprintf(buf, IFNAMSIZ, "mon.%s", drv->iface);
1883         buf[IFNAMSIZ - 1] = '\0';
1884
1885         drv->monitor_ifidx =
1886                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL);
1887
1888         if (drv->monitor_ifidx < 0)
1889                 return -1;
1890
1891         if (hostapd_set_iface_flags(drv, buf, 1))
1892                 goto error;
1893
1894         memset(&ll, 0, sizeof(ll));
1895         ll.sll_family = AF_PACKET;
1896         ll.sll_ifindex = drv->monitor_ifidx;
1897         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
1898         if (drv->monitor_sock < 0) {
1899                 perror("socket[PF_PACKET,SOCK_RAW]");
1900                 goto error;
1901         }
1902
1903         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
1904                  sizeof(ll)) < 0) {
1905                 perror("monitor socket bind");
1906                 goto error;
1907         }
1908
1909         optlen = sizeof(optval);
1910         optval = 20;
1911         if (setsockopt
1912             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
1913                 perror("Failed to set socket priority");
1914                 goto error;
1915         }
1916
1917         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
1918                                      drv, NULL)) {
1919                 printf("Could not register monitor read socket\n");
1920                 goto error;
1921         }
1922
1923         return 0;
1924  error:
1925         nl80211_remove_iface(drv, drv->monitor_ifidx);
1926         return -1;
1927 }
1928
1929
1930 static int nl80211_set_master_mode(struct i802_driver_data *drv,
1931                                    const char *ifname)
1932 {
1933         struct nl_msg *msg;
1934
1935         msg = nlmsg_alloc();
1936         if (!msg)
1937                 return -1;
1938
1939         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1940                     0, NL80211_CMD_SET_INTERFACE, 0);
1941         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1942                     if_nametoindex(ifname));
1943         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, NL80211_IFTYPE_AP);
1944
1945         if (nl_send_auto_complete(drv->nl_handle, msg) < 0 ||
1946             nl_wait_for_ack(drv->nl_handle) < 0) {
1947  nla_put_failure:
1948                 wpa_printf(MSG_ERROR, "Failed to set interface %s to master "
1949                            "mode.", ifname);
1950                 nlmsg_free(msg);
1951                 return -1;
1952         }
1953
1954         nlmsg_free(msg);
1955
1956         return 0;
1957 }
1958
1959
1960 static int i802_init_sockets(struct i802_driver_data *drv, const u8 *bssid)
1961 {
1962         struct ifreq ifr;
1963         struct sockaddr_ll addr;
1964
1965         drv->ioctl_sock = -1;
1966
1967         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1968         if (drv->ioctl_sock < 0) {
1969                 perror("socket[PF_INET,SOCK_DGRAM]");
1970                 return -1;
1971         }
1972
1973         /* start listening for EAPOL on the default AP interface */
1974         add_ifidx(drv, if_nametoindex(drv->iface));
1975
1976         if (hostapd_set_iface_flags(drv, drv->iface, 0))
1977                 return -1;
1978
1979         if (bssid) {
1980                 os_strlcpy(ifr.ifr_name, drv->iface, IFNAMSIZ);
1981                 memcpy(ifr.ifr_hwaddr.sa_data, bssid, ETH_ALEN);
1982                 ifr.ifr_hwaddr.sa_family = ARPHRD_ETHER;
1983
1984                 if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifr)) {
1985                         perror("ioctl(SIOCSIFHWADDR)");
1986                         return -1;
1987                 }
1988         }
1989
1990         /*
1991          * initialise generic netlink and nl80211
1992          */
1993         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
1994         if (!drv->nl_cb) {
1995                 printf("Failed to allocate netlink callbacks.\n");
1996                 return -1;
1997         }
1998
1999         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
2000         if (!drv->nl_handle) {
2001                 printf("Failed to allocate netlink handle.\n");
2002                 return -1;
2003         }
2004
2005         if (genl_connect(drv->nl_handle)) {
2006                 printf("Failed to connect to generic netlink.\n");
2007                 return -1;
2008         }
2009
2010         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
2011         if (!drv->nl_cache) {
2012                 printf("Failed to allocate generic netlink cache.\n");
2013                 return -1;
2014         }
2015
2016         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
2017         if (!drv->nl80211) {
2018                 printf("nl80211 not found.\n");
2019                 return -1;
2020         }
2021
2022         /* Initialise a monitor interface */
2023         if (nl80211_create_monitor_interface(drv))
2024                 return -1;
2025
2026         if (nl80211_set_master_mode(drv, drv->iface))
2027                 return -1;
2028
2029         if (hostapd_set_iface_flags(drv, drv->iface, 1))
2030                 return -1;
2031
2032         memset(&addr, 0, sizeof(addr));
2033         addr.sll_family = AF_PACKET;
2034         addr.sll_ifindex = ifr.ifr_ifindex;
2035         wpa_printf(MSG_DEBUG, "Opening raw packet socket for ifindex %d",
2036                    addr.sll_ifindex);
2037
2038         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
2039         if (drv->eapol_sock < 0) {
2040                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
2041                 return -1;
2042         }
2043
2044         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
2045         {
2046                 printf("Could not register read socket for eapol\n");
2047                 return -1;
2048         }
2049
2050         memset(&ifr, 0, sizeof(ifr));
2051         os_strlcpy(ifr.ifr_name, drv->iface, sizeof(ifr.ifr_name));
2052         if (ioctl(drv->ioctl_sock, SIOCGIFHWADDR, &ifr) != 0) {
2053                 perror("ioctl(SIOCGIFHWADDR)");
2054                 return -1;
2055         }
2056
2057         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
2058                 printf("Invalid HW-addr family 0x%04x\n",
2059                        ifr.ifr_hwaddr.sa_family);
2060                 return -1;
2061         }
2062         memcpy(drv->hapd->own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
2063
2064         return 0;
2065 }
2066
2067
2068 static int i802_get_inact_sec(void *priv, const u8 *addr)
2069 {
2070         struct hostap_sta_driver_data data;
2071         int ret;
2072
2073         data.inactive_msec = (unsigned long) -1;
2074         ret = i802_read_sta_data(priv, &data, addr);
2075         if (ret || data.inactive_msec == (unsigned long) -1)
2076                 return -1;
2077         return data.inactive_msec / 1000;
2078 }
2079
2080
2081 static int i802_sta_clear_stats(void *priv, const u8 *addr)
2082 {
2083 #if 0
2084         /* TODO */
2085 #endif
2086         return 0;
2087 }
2088
2089
2090 static void
2091 hostapd_wireless_event_wireless_custom(struct i802_driver_data *drv,
2092                                        char *custom)
2093 {
2094         wpa_printf(MSG_DEBUG, "Custom wireless event: '%s'", custom);
2095
2096         if (strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
2097                 char *pos;
2098                 u8 addr[ETH_ALEN];
2099                 pos = strstr(custom, "addr=");
2100                 if (pos == NULL) {
2101                         wpa_printf(MSG_DEBUG,
2102                                    "MLME-MICHAELMICFAILURE.indication "
2103                                    "without sender address ignored");
2104                         return;
2105                 }
2106                 pos += 5;
2107                 if (hwaddr_aton(pos, addr) == 0) {
2108                         ieee80211_michael_mic_failure(drv->hapd, addr, 1);
2109                 } else {
2110                         wpa_printf(MSG_DEBUG,
2111                                    "MLME-MICHAELMICFAILURE.indication "
2112                                    "with invalid MAC address");
2113                 }
2114         }
2115 }
2116
2117
2118 static void hostapd_wireless_event_wireless(struct i802_driver_data *drv,
2119                                             char *data, int len)
2120 {
2121         struct iw_event iwe_buf, *iwe = &iwe_buf;
2122         char *pos, *end, *custom, *buf;
2123
2124         pos = data;
2125         end = data + len;
2126
2127         while (pos + IW_EV_LCP_LEN <= end) {
2128                 /* Event data may be unaligned, so make a local, aligned copy
2129                  * before processing. */
2130                 memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2131                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
2132                            iwe->cmd, iwe->len);
2133                 if (iwe->len <= IW_EV_LCP_LEN)
2134                         return;
2135
2136                 custom = pos + IW_EV_POINT_LEN;
2137                 if (drv->we_version > 18 &&
2138                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
2139                      iwe->cmd == IWEVCUSTOM)) {
2140                         /* WE-19 removed the pointer from struct iw_point */
2141                         char *dpos = (char *) &iwe_buf.u.data.length;
2142                         int dlen = dpos - (char *) &iwe_buf;
2143                         memcpy(dpos, pos + IW_EV_LCP_LEN,
2144                                sizeof(struct iw_event) - dlen);
2145                 } else {
2146                         memcpy(&iwe_buf, pos, sizeof(struct iw_event));
2147                         custom += IW_EV_POINT_OFF;
2148                 }
2149
2150                 switch (iwe->cmd) {
2151                 case IWEVCUSTOM:
2152                         if (custom + iwe->u.data.length > end)
2153                                 return;
2154                         buf = malloc(iwe->u.data.length + 1);
2155                         if (buf == NULL)
2156                                 return;
2157                         memcpy(buf, custom, iwe->u.data.length);
2158                         buf[iwe->u.data.length] = '\0';
2159                         hostapd_wireless_event_wireless_custom(drv, buf);
2160                         free(buf);
2161                         break;
2162                 }
2163
2164                 pos += iwe->len;
2165         }
2166 }
2167
2168
2169 static void hostapd_wireless_event_rtm_newlink(struct i802_driver_data *drv,
2170                                                struct nlmsghdr *h, int len)
2171 {
2172         struct ifinfomsg *ifi;
2173         int attrlen, nlmsg_len, rta_len;
2174         struct rtattr *attr;
2175
2176         if (len < (int) sizeof(*ifi))
2177                 return;
2178
2179         ifi = NLMSG_DATA(h);
2180
2181         /* TODO: use ifi->ifi_index to filter out wireless events from other
2182          * interfaces */
2183
2184         nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
2185
2186         attrlen = h->nlmsg_len - nlmsg_len;
2187         if (attrlen < 0)
2188                 return;
2189
2190         attr = (struct rtattr *) (((char *) ifi) + nlmsg_len);
2191
2192         rta_len = RTA_ALIGN(sizeof(struct rtattr));
2193         while (RTA_OK(attr, attrlen)) {
2194                 if (attr->rta_type == IFLA_WIRELESS) {
2195                         hostapd_wireless_event_wireless(
2196                                 drv, ((char *) attr) + rta_len,
2197                                 attr->rta_len - rta_len);
2198                 }
2199                 attr = RTA_NEXT(attr, attrlen);
2200         }
2201 }
2202
2203
2204 static void hostapd_wireless_event_receive(int sock, void *eloop_ctx,
2205                                            void *sock_ctx)
2206 {
2207         char buf[256];
2208         int left;
2209         struct sockaddr_nl from;
2210         socklen_t fromlen;
2211         struct nlmsghdr *h;
2212         struct i802_driver_data *drv = eloop_ctx;
2213
2214         fromlen = sizeof(from);
2215         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
2216                         (struct sockaddr *) &from, &fromlen);
2217         if (left < 0) {
2218                 if (errno != EINTR && errno != EAGAIN)
2219                         perror("recvfrom(netlink)");
2220                 return;
2221         }
2222
2223         h = (struct nlmsghdr *) buf;
2224         while (left >= (int) sizeof(*h)) {
2225                 int len, plen;
2226
2227                 len = h->nlmsg_len;
2228                 plen = len - sizeof(*h);
2229                 if (len > left || plen < 0) {
2230                         printf("Malformed netlink message: "
2231                                "len=%d left=%d plen=%d\n",
2232                                len, left, plen);
2233                         break;
2234                 }
2235
2236                 switch (h->nlmsg_type) {
2237                 case RTM_NEWLINK:
2238                         hostapd_wireless_event_rtm_newlink(drv, h, plen);
2239                         break;
2240                 }
2241
2242                 len = NLMSG_ALIGN(len);
2243                 left -= len;
2244                 h = (struct nlmsghdr *) ((char *) h + len);
2245         }
2246
2247         if (left > 0) {
2248                 printf("%d extra bytes in the end of netlink message\n", left);
2249         }
2250 }
2251
2252
2253 static int hostap_get_we_version(struct i802_driver_data *drv)
2254 {
2255         struct iw_range *range;
2256         struct iwreq iwr;
2257         int minlen;
2258         size_t buflen;
2259
2260         drv->we_version = 0;
2261
2262         /*
2263          * Use larger buffer than struct iw_range in order to allow the
2264          * structure to grow in the future.
2265          */
2266         buflen = sizeof(struct iw_range) + 500;
2267         range = os_zalloc(buflen);
2268         if (range == NULL)
2269                 return -1;
2270
2271         memset(&iwr, 0, sizeof(iwr));
2272         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2273         iwr.u.data.pointer = (caddr_t) range;
2274         iwr.u.data.length = buflen;
2275
2276         minlen = ((char *) &range->enc_capa) - (char *) range +
2277                 sizeof(range->enc_capa);
2278
2279         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
2280                 perror("ioctl[SIOCGIWRANGE]");
2281                 free(range);
2282                 return -1;
2283         } else if (iwr.u.data.length >= minlen &&
2284                    range->we_version_compiled >= 18) {
2285                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
2286                            "WE(source)=%d enc_capa=0x%x",
2287                            range->we_version_compiled,
2288                            range->we_version_source,
2289                            range->enc_capa);
2290                 drv->we_version = range->we_version_compiled;
2291         }
2292
2293         free(range);
2294         return 0;
2295 }
2296
2297
2298 static int i802_wireless_event_init(void *priv)
2299 {
2300         struct i802_driver_data *drv = priv;
2301         int s;
2302         struct sockaddr_nl local;
2303
2304         hostap_get_we_version(drv);
2305
2306         drv->wext_sock = -1;
2307
2308         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
2309         if (s < 0) {
2310                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
2311                 return -1;
2312         }
2313
2314         memset(&local, 0, sizeof(local));
2315         local.nl_family = AF_NETLINK;
2316         local.nl_groups = RTMGRP_LINK;
2317         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
2318                 perror("bind(netlink)");
2319                 close(s);
2320                 return -1;
2321         }
2322
2323         eloop_register_read_sock(s, hostapd_wireless_event_receive, drv,
2324                                  NULL);
2325         drv->wext_sock = s;
2326
2327         return 0;
2328 }
2329
2330
2331 static void i802_wireless_event_deinit(void *priv)
2332 {
2333         struct i802_driver_data *drv = priv;
2334         if (drv->wext_sock < 0)
2335                 return;
2336         eloop_unregister_read_sock(drv->wext_sock);
2337         close(drv->wext_sock);
2338 }
2339
2340
2341 static int i802_sta_deauth(void *priv, const u8 *addr, int reason)
2342 {
2343         struct i802_driver_data *drv = priv;
2344         struct ieee80211_mgmt mgmt;
2345
2346         memset(&mgmt, 0, sizeof(mgmt));
2347         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2348                                           WLAN_FC_STYPE_DEAUTH);
2349         memcpy(mgmt.da, addr, ETH_ALEN);
2350         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2351         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2352         mgmt.u.deauth.reason_code = host_to_le16(reason);
2353         return i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2354                                       sizeof(mgmt.u.deauth), 0);
2355 }
2356
2357
2358 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason)
2359 {
2360         struct i802_driver_data *drv = priv;
2361         struct ieee80211_mgmt mgmt;
2362
2363         memset(&mgmt, 0, sizeof(mgmt));
2364         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2365                                           WLAN_FC_STYPE_DISASSOC);
2366         memcpy(mgmt.da, addr, ETH_ALEN);
2367         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2368         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2369         mgmt.u.disassoc.reason_code = host_to_le16(reason);
2370         return  i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2371                                        sizeof(mgmt.u.disassoc), 0);
2372 }
2373
2374
2375 static void *i802_init_bssid(struct hostapd_data *hapd, const u8 *bssid)
2376 {
2377         struct i802_driver_data *drv;
2378
2379         drv = os_zalloc(sizeof(struct i802_driver_data));
2380         if (drv == NULL) {
2381                 printf("Could not allocate memory for i802 driver data\n");
2382                 return NULL;
2383         }
2384
2385         drv->hapd = hapd;
2386         memcpy(drv->iface, hapd->conf->iface, sizeof(drv->iface));
2387
2388         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
2389         drv->if_indices = drv->default_if_indices;
2390         drv->bridge = if_nametoindex(hapd->conf->bridge);
2391
2392         if (i802_init_sockets(drv, bssid))
2393                 goto failed;
2394
2395         return drv;
2396
2397 failed:
2398         free(drv);
2399         return NULL;
2400 }
2401
2402
2403 static void *i802_init(struct hostapd_data *hapd)
2404 {
2405         return i802_init_bssid(hapd, NULL);
2406 }
2407
2408
2409 static void i802_deinit(void *priv)
2410 {
2411         struct i802_driver_data *drv = priv;
2412
2413         i802_del_beacon(drv);
2414
2415         /* remove monitor interface */
2416         nl80211_remove_iface(drv, drv->monitor_ifidx);
2417
2418         (void) hostapd_set_iface_flags(drv, drv->iface, 0);
2419
2420         if (drv->monitor_sock >= 0) {
2421                 eloop_unregister_read_sock(drv->monitor_sock);
2422                 close(drv->monitor_sock);
2423         }
2424         if (drv->ioctl_sock >= 0)
2425                 close(drv->ioctl_sock);
2426         if (drv->eapol_sock >= 0) {
2427                 eloop_unregister_read_sock(drv->eapol_sock);
2428                 close(drv->eapol_sock);
2429         }
2430
2431         genl_family_put(drv->nl80211);
2432         nl_cache_free(drv->nl_cache);
2433         nl_handle_destroy(drv->nl_handle);
2434         nl_cb_put(drv->nl_cb);
2435
2436         if (drv->if_indices != drv->default_if_indices)
2437                 free(drv->if_indices);
2438
2439         free(drv);
2440 }
2441
2442
2443 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
2444         .name = "nl80211",
2445         .init = i802_init,
2446         .init_bssid = i802_init_bssid,
2447         .deinit = i802_deinit,
2448         .wireless_event_init = i802_wireless_event_init,
2449         .wireless_event_deinit = i802_wireless_event_deinit,
2450         .set_ieee8021x = i802_set_ieee8021x,
2451         .set_privacy = i802_set_privacy,
2452         .set_encryption = i802_set_encryption,
2453         .get_seqnum = i802_get_seqnum,
2454         .flush = i802_flush,
2455         .read_sta_data = i802_read_sta_data,
2456         .send_eapol = i802_send_eapol,
2457         .sta_set_flags = i802_sta_set_flags,
2458         .sta_deauth = i802_sta_deauth,
2459         .sta_disassoc = i802_sta_disassoc,
2460         .sta_remove = i802_sta_remove,
2461         .set_ssid = i802_set_ssid,
2462         .send_mgmt_frame = i802_send_mgmt_frame,
2463         .sta_add2 = i802_sta_add2,
2464         .get_inact_sec = i802_get_inact_sec,
2465         .sta_clear_stats = i802_sta_clear_stats,
2466         .set_freq = i802_set_freq,
2467         .set_rts = i802_set_rts,
2468         .get_rts = i802_get_rts,
2469         .set_frag = i802_set_frag,
2470         .get_frag = i802_get_frag,
2471         .set_retry = i802_set_retry,
2472         .get_retry = i802_get_retry,
2473         .set_rate_sets = i802_set_rate_sets,
2474         .set_channel_flag = i802_set_channel_flag,
2475         .set_regulatory_domain = i802_set_regulatory_domain,
2476         .set_beacon = i802_set_beacon,
2477         .set_internal_bridge = i802_set_internal_bridge,
2478         .set_beacon_int = i802_set_beacon_int,
2479         .set_dtim_period = i802_set_dtim_period,
2480         .set_cts_protect = i802_set_cts_protect,
2481         .set_preamble = i802_set_preamble,
2482         .set_short_slot_time = i802_set_short_slot_time,
2483         .set_tx_queue_params = i802_set_tx_queue_params,
2484         .bss_add = i802_bss_add,
2485         .bss_remove = i802_bss_remove,
2486         .if_add = i802_if_add,
2487         .if_update = i802_if_update,
2488         .if_remove = i802_if_remove,
2489         .get_hw_feature_data = i802_get_hw_feature_data,
2490         .set_sta_vlan = i802_set_sta_vlan,
2491 };