f76dd9465c3f5b6f071bd27f961e5a29d26be416
[wpasupplicant] / 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 "nl80211_copy.h"
27 #include <net/if.h>
28 #include <netpacket/packet.h>
29 #include "wireless_copy.h"
30 #include <linux/filter.h>
31 #include <net/if_arp.h>
32
33 #include "hostapd.h"
34 #include "config.h"
35 #include "driver.h"
36 #include "eloop.h"
37 #include "hw_features.h"
38 #include "mlme.h"
39 #include "radiotap.h"
40 #include "radiotap_iter.h"
41 #include "ieee802_11_defs.h"
42 #include "ieee802_11_common.h"
43
44 #ifdef CONFIG_LIBNL20
45 /* libnl 2.0 compatibility code */
46 #define nl_handle_alloc_cb nl_socket_alloc_cb
47 #define nl_handle_destroy nl_socket_free
48 #endif /* CONFIG_LIBNL20 */
49
50 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
51
52 enum ieee80211_msg_type {
53         ieee80211_msg_normal = 0,
54         ieee80211_msg_tx_callback_ack = 1,
55         ieee80211_msg_tx_callback_fail = 2,
56 };
57
58 struct i802_bss {
59         struct i802_bss *next;
60         char iface[IFNAMSIZ + 1];
61         unsigned int beacon_set:1;
62 };
63
64 struct i802_driver_data {
65         struct hostapd_data *hapd;
66
67         char iface[IFNAMSIZ + 1];
68         int bridge;
69         int ioctl_sock; /* socket for ioctl() use */
70         int wext_sock; /* socket for wireless events */
71         int eapol_sock; /* socket for EAPOL frames */
72         int monitor_sock; /* socket for monitor */
73         int monitor_ifidx;
74
75         int default_if_indices[16];
76         int *if_indices;
77         int num_if_indices;
78
79         int we_version;
80         struct nl_handle *nl_handle;
81         struct nl_cache *nl_cache;
82         struct nl_cb *nl_cb;
83         struct genl_family *nl80211;
84         int beacon_int;
85         struct i802_bss bss;
86         unsigned int ieee802_1x_active:1;
87         unsigned int ht_40mhz_scan:1;
88
89         int last_freq;
90         int last_freq_ht;
91         struct hostapd_neighbor_bss *neighbors;
92         size_t num_neighbors;
93 };
94
95
96 static int i802_sta_deauth(void *priv, const u8 *addr, int reason);
97 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason);
98
99
100 static struct i802_bss * get_bss(struct i802_driver_data *drv,
101                                  const char *iface)
102 {
103         struct i802_bss *bss = &drv->bss;
104         while (bss) {
105                 if (os_strncmp(iface, bss->iface, IFNAMSIZ) == 0)
106                         return bss;
107                 bss = bss->next;
108         }
109         wpa_printf(MSG_DEBUG, "nl80211: get_bss(%s) failed", iface);
110         return NULL;
111 }
112
113
114 static void add_ifidx(struct i802_driver_data *drv, int ifidx)
115 {
116         int i;
117         int *old;
118
119         for (i = 0; i < drv->num_if_indices; i++) {
120                 if (drv->if_indices[i] == 0) {
121                         drv->if_indices[i] = ifidx;
122                         return;
123                 }
124         }
125
126         if (drv->if_indices != drv->default_if_indices)
127                 old = drv->if_indices;
128         else
129                 old = NULL;
130
131         drv->if_indices = realloc(old,
132                                   sizeof(int) * (drv->num_if_indices + 1));
133         if (!drv->if_indices) {
134                 if (!old)
135                         drv->if_indices = drv->default_if_indices;
136                 else
137                         drv->if_indices = old;
138                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
139                            "interfaces");
140                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
141                 return;
142         }
143         drv->if_indices[drv->num_if_indices] = ifidx;
144         drv->num_if_indices++;
145 }
146
147
148 static void del_ifidx(struct i802_driver_data *drv, int ifidx)
149 {
150         int i;
151
152         for (i = 0; i < drv->num_if_indices; i++) {
153                 if (drv->if_indices[i] == ifidx) {
154                         drv->if_indices[i] = 0;
155                         break;
156                 }
157         }
158 }
159
160
161 static int have_ifidx(struct i802_driver_data *drv, int ifidx)
162 {
163         int i;
164
165         if (ifidx == drv->bridge)
166                 return 1;
167
168         for (i = 0; i < drv->num_if_indices; i++)
169                 if (drv->if_indices[i] == ifidx)
170                         return 1;
171
172         return 0;
173 }
174
175
176 /* nl80211 code */
177 static int ack_handler(struct nl_msg *msg, void *arg)
178 {
179         int *err = arg;
180         *err = 0;
181         return NL_STOP;
182 }
183
184 static int finish_handler(struct nl_msg *msg, void *arg)
185 {
186         int *ret = arg;
187         *ret = 0;
188         return NL_SKIP;
189 }
190
191 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
192                          void *arg)
193 {
194         int *ret = arg;
195         *ret = err->error;
196         return NL_SKIP;
197 }
198
199 static int send_and_recv_msgs(struct i802_driver_data *drv,
200                               struct nl_msg *msg,
201                               int (*valid_handler)(struct nl_msg *, void *),
202                               void *valid_data)
203 {
204         struct nl_cb *cb;
205         int err = -ENOMEM;
206
207         cb = nl_cb_clone(drv->nl_cb);
208         if (!cb)
209                 goto out;
210
211         err = nl_send_auto_complete(drv->nl_handle, msg);
212         if (err < 0)
213                 goto out;
214
215         err = 1;
216
217         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
218         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
219         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
220
221         if (valid_handler)
222                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
223                           valid_handler, valid_data);
224
225         while (err > 0)
226                 nl_recvmsgs(drv->nl_handle, cb);
227  out:
228         nl_cb_put(cb);
229         nlmsg_free(msg);
230         return err;
231 }
232
233 static int hostapd_set_iface_flags(struct i802_driver_data *drv,
234                                    const char *ifname, int dev_up)
235 {
236         struct ifreq ifr;
237
238         if (drv->ioctl_sock < 0)
239                 return -1;
240
241         memset(&ifr, 0, sizeof(ifr));
242         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
243
244         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
245                 perror("ioctl[SIOCGIFFLAGS]");
246                 wpa_printf(MSG_DEBUG, "Could not read interface flags (%s)",
247                            drv->iface);
248                 return -1;
249         }
250
251         if (dev_up)
252                 ifr.ifr_flags |= IFF_UP;
253         else
254                 ifr.ifr_flags &= ~IFF_UP;
255
256         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
257                 perror("ioctl[SIOCSIFFLAGS]");
258                 return -1;
259         }
260
261         return 0;
262 }
263
264
265 static int nl_set_encr(int ifindex, struct i802_driver_data *drv,
266                        wpa_alg alg, const u8 *addr, int idx, const u8 *key,
267                        size_t key_len, int txkey)
268 {
269         struct nl_msg *msg;
270         int ret;
271
272         msg = nlmsg_alloc();
273         if (!msg)
274                 return -ENOMEM;
275
276         if (alg == WPA_ALG_NONE) {
277                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
278                             0, NL80211_CMD_DEL_KEY, 0);
279         } else {
280                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
281                             0, NL80211_CMD_NEW_KEY, 0);
282                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
283                 switch (alg) {
284                 case WPA_ALG_WEP:
285                         if (key_len == 5)
286                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
287                                             0x000FAC01);
288                         else
289                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
290                                             0x000FAC05);
291                         break;
292                 case WPA_ALG_TKIP:
293                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
294                         break;
295                 case WPA_ALG_CCMP:
296                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
297                         break;
298                 case WPA_ALG_IGTK:
299                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC06);
300                         break;
301                 default:
302                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
303                                    "algorithm %d", __func__, alg);
304                         nlmsg_free(msg);
305                         return -1;
306                 }
307         }
308
309         if (addr)
310                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
311         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
312         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
313
314         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
315         if (ret == -ENOENT)
316                 ret = 0;
317
318         /*
319          * If we failed or don't need to set the default TX key (below),
320          * we're done here.
321          */
322         if (ret || !txkey || addr)
323                 return ret;
324
325         msg = nlmsg_alloc();
326         if (!msg)
327                 return -ENOMEM;
328
329         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
330                     0, NL80211_CMD_SET_KEY, 0);
331         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
332         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
333         if (alg == WPA_ALG_IGTK)
334                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
335         else
336                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
337
338         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
339         if (ret == -ENOENT)
340                 ret = 0;
341         return ret;
342  nla_put_failure:
343         return -ENOBUFS;
344 }
345
346
347 static int i802_set_key(const char *iface, void *priv, wpa_alg alg,
348                         const u8 *addr, int key_idx, int set_tx, const u8 *seq,
349                         size_t seq_len, const u8 *key, size_t key_len)
350 {
351         struct i802_driver_data *drv = priv;
352         int ret;
353
354         ret = nl_set_encr(if_nametoindex(iface), drv, alg, addr, key_idx, key,
355                           key_len, set_tx);
356         if (ret < 0)
357                 return ret;
358
359         return ret;
360 }
361
362
363 static inline int min_int(int a, int b)
364 {
365         if (a < b)
366                 return a;
367         return b;
368 }
369
370
371 static int get_key_handler(struct nl_msg *msg, void *arg)
372 {
373         struct nlattr *tb[NL80211_ATTR_MAX + 1];
374         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
375
376         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
377                   genlmsg_attrlen(gnlh, 0), NULL);
378
379         /*
380          * TODO: validate the key index and mac address!
381          * Otherwise, there's a race condition as soon as
382          * the kernel starts sending key notifications.
383          */
384
385         if (tb[NL80211_ATTR_KEY_SEQ])
386                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
387                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
388         return NL_SKIP;
389 }
390
391
392 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
393                            int idx, u8 *seq)
394 {
395         struct i802_driver_data *drv = priv;
396         struct nl_msg *msg;
397
398         msg = nlmsg_alloc();
399         if (!msg)
400                 return -ENOMEM;
401
402         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
403                     0, NL80211_CMD_GET_KEY, 0);
404
405         if (addr)
406                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
407         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
408         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
409
410         memset(seq, 0, 6);
411
412         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
413  nla_put_failure:
414         return -ENOBUFS;
415 }
416
417
418 static int i802_set_rate_sets(void *priv, int *supp_rates, int *basic_rates,
419                               int mode)
420 {
421         struct i802_driver_data *drv = priv;
422         struct nl_msg *msg;
423         u8 rates[NL80211_MAX_SUPP_RATES];
424         u8 rates_len = 0;
425         int i;
426
427         msg = nlmsg_alloc();
428         if (!msg)
429                 return -ENOMEM;
430
431         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
432                     NL80211_CMD_SET_BSS, 0);
433
434         for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0; i++)
435                 rates[rates_len++] = basic_rates[i] / 5;
436
437         NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
438
439         /* TODO: multi-BSS support */
440         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
441
442         return send_and_recv_msgs(drv, msg, NULL, NULL);
443  nla_put_failure:
444         return -ENOBUFS;
445 }
446
447
448 static int i802_send_frame(void *priv, const void *data, size_t len,
449                            int encrypt, int flags)
450 {
451         __u8 rtap_hdr[] = {
452                 0x00, 0x00, /* radiotap version */
453                 0x0e, 0x00, /* radiotap length */
454                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
455                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
456                 0x00,       /* padding */
457                 0x00, 0x00, /* RX and TX flags to indicate that */
458                 0x00, 0x00, /* this is the injected frame directly */
459         };
460         struct i802_driver_data *drv = priv;
461         struct iovec iov[2] = {
462                 {
463                         .iov_base = &rtap_hdr,
464                         .iov_len = sizeof(rtap_hdr),
465                 },
466                 {
467                         .iov_base = (void*)data,
468                         .iov_len = len,
469                 }
470         };
471         struct msghdr msg = {
472                 .msg_name = NULL,
473                 .msg_namelen = 0,
474                 .msg_iov = iov,
475                 .msg_iovlen = 2,
476                 .msg_control = NULL,
477                 .msg_controllen = 0,
478                 .msg_flags = 0,
479         };
480
481         if (encrypt)
482                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
483
484         return sendmsg(drv->monitor_sock, &msg, flags);
485 }
486
487 static int i802_send_mgmt_frame(void *priv, const void *data, size_t len,
488                                 int flags)
489 {
490         struct ieee80211_mgmt *mgmt;
491         int do_not_encrypt = 0;
492         u16 fc;
493
494         mgmt = (struct ieee80211_mgmt *) data;
495         fc = le_to_host16(mgmt->frame_control);
496
497         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
498             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
499                 /*
500                  * Only one of the authentication frame types is encrypted.
501                  * In order for static WEP encryption to work properly (i.e.,
502                  * to not encrypt the frame), we need to tell mac80211 about
503                  * the frames that must not be encrypted.
504                  */
505                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
506                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
507                 if (auth_alg == WLAN_AUTH_OPEN ||
508                     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_trans != 3))
509                         do_not_encrypt = 1;
510         }
511
512         return i802_send_frame(priv, data, len, !do_not_encrypt, flags);
513 }
514
515 /* Set kernel driver on given frequency (MHz) */
516 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq)
517 {
518         struct i802_driver_data *drv = priv;
519         struct nl_msg *msg;
520
521         msg = nlmsg_alloc();
522         if (!msg)
523                 return -1;
524
525         drv->last_freq = freq->freq;
526         drv->last_freq_ht = freq->ht_enabled;
527
528         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
529                     NL80211_CMD_SET_WIPHY, 0);
530
531         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
532         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq->freq);
533         if (freq->ht_enabled) {
534                 switch (freq->sec_channel_offset) {
535                 case -1:
536                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
537                                     NL80211_CHAN_HT40MINUS);
538                         break;
539                 case 1:
540                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
541                                     NL80211_CHAN_HT40PLUS);
542                         break;
543                 default:
544                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
545                                     NL80211_CHAN_HT20);
546                         break;
547                 }
548         }
549
550         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
551                 return 0;
552  nla_put_failure:
553         return -1;
554 }
555
556
557 static int i802_set_rts(void *priv, int rts)
558 {
559         struct i802_driver_data *drv = priv;
560         struct iwreq iwr;
561
562         memset(&iwr, 0, sizeof(iwr));
563         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
564         iwr.u.rts.value = rts;
565         iwr.u.rts.fixed = 1;
566
567         if (ioctl(drv->ioctl_sock, SIOCSIWRTS, &iwr) < 0) {
568                 perror("ioctl[SIOCSIWRTS]");
569                 return -1;
570         }
571
572         return 0;
573 }
574
575
576 static int i802_get_rts(void *priv, int *rts)
577 {
578         struct i802_driver_data *drv = priv;
579         struct iwreq iwr;
580
581         memset(&iwr, 0, sizeof(iwr));
582         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
583
584         if (ioctl(drv->ioctl_sock, SIOCGIWRTS, &iwr) < 0) {
585                 perror("ioctl[SIOCGIWRTS]");
586                 return -1;
587         }
588
589         *rts = iwr.u.rts.value;
590
591         return 0;
592 }
593
594
595 static int i802_set_frag(void *priv, int frag)
596 {
597         struct i802_driver_data *drv = priv;
598         struct iwreq iwr;
599
600         memset(&iwr, 0, sizeof(iwr));
601         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
602         iwr.u.frag.value = frag;
603         iwr.u.frag.fixed = 1;
604
605         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
606                 perror("ioctl[SIOCSIWFRAG]");
607                 return -1;
608         }
609
610         return 0;
611 }
612
613
614 static int i802_get_frag(void *priv, int *frag)
615 {
616         struct i802_driver_data *drv = priv;
617         struct iwreq iwr;
618
619         memset(&iwr, 0, sizeof(iwr));
620         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
621
622         if (ioctl(drv->ioctl_sock, SIOCGIWFRAG, &iwr) < 0) {
623                 perror("ioctl[SIOCGIWFRAG]");
624                 return -1;
625         }
626
627         *frag = iwr.u.frag.value;
628
629         return 0;
630 }
631
632
633 static int i802_set_retry(void *priv, int short_retry, int long_retry)
634 {
635         struct i802_driver_data *drv = priv;
636         struct iwreq iwr;
637
638         memset(&iwr, 0, sizeof(iwr));
639         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
640
641         iwr.u.retry.value = short_retry;
642         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
643         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
644                 perror("ioctl[SIOCSIWRETRY(short)]");
645                 return -1;
646         }
647
648         iwr.u.retry.value = long_retry;
649         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
650         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
651                 perror("ioctl[SIOCSIWRETRY(long)]");
652                 return -1;
653         }
654
655         return 0;
656 }
657
658
659 static int i802_get_retry(void *priv, int *short_retry, int *long_retry)
660 {
661         struct i802_driver_data *drv = priv;
662         struct iwreq iwr;
663
664         memset(&iwr, 0, sizeof(iwr));
665         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
666
667         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
668         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
669                 perror("ioctl[SIOCGIWFRAG(short)]");
670                 return -1;
671         }
672         *short_retry = iwr.u.retry.value;
673
674         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
675         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
676                 perror("ioctl[SIOCGIWFRAG(long)]");
677                 return -1;
678         }
679         *long_retry = iwr.u.retry.value;
680
681         return 0;
682 }
683
684
685 static int i802_flush(void *priv)
686 {
687         struct i802_driver_data *drv = priv;
688         struct nl_msg *msg;
689
690         msg = nlmsg_alloc();
691         if (!msg)
692                 return -1;
693
694         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
695                     0, NL80211_CMD_DEL_STATION, 0);
696
697         /*
698          * XXX: FIX! this needs to flush all VLANs too
699          */
700         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
701                     if_nametoindex(drv->iface));
702
703         return send_and_recv_msgs(drv, msg, NULL, NULL);
704  nla_put_failure:
705         return -ENOBUFS;
706 }
707
708
709 static int get_sta_handler(struct nl_msg *msg, void *arg)
710 {
711         struct nlattr *tb[NL80211_ATTR_MAX + 1];
712         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
713         struct hostap_sta_driver_data *data = arg;
714         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
715         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
716                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
717                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
718                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
719                 [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
720                 [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
721         };
722
723         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
724                   genlmsg_attrlen(gnlh, 0), NULL);
725
726         /*
727          * TODO: validate the interface and mac address!
728          * Otherwise, there's a race condition as soon as
729          * the kernel starts sending station notifications.
730          */
731
732         if (!tb[NL80211_ATTR_STA_INFO]) {
733                 wpa_printf(MSG_DEBUG, "sta stats missing!");
734                 return NL_SKIP;
735         }
736         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
737                              tb[NL80211_ATTR_STA_INFO],
738                              stats_policy)) {
739                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
740                 return NL_SKIP;
741         }
742
743         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
744                 data->inactive_msec =
745                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
746         if (stats[NL80211_STA_INFO_RX_BYTES])
747                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
748         if (stats[NL80211_STA_INFO_TX_BYTES])
749                 data->tx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
750         if (stats[NL80211_STA_INFO_RX_PACKETS])
751                 data->rx_packets =
752                         nla_get_u32(stats[NL80211_STA_INFO_RX_PACKETS]);
753         if (stats[NL80211_STA_INFO_TX_PACKETS])
754                 data->tx_packets =
755                         nla_get_u32(stats[NL80211_STA_INFO_TX_PACKETS]);
756
757         return NL_SKIP;
758 }
759
760 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
761                               const u8 *addr)
762 {
763         struct i802_driver_data *drv = priv;
764         struct nl_msg *msg;
765
766         os_memset(data, 0, sizeof(*data));
767         msg = nlmsg_alloc();
768         if (!msg)
769                 return -ENOMEM;
770
771         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
772                     0, NL80211_CMD_GET_STATION, 0);
773
774         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
775         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
776
777         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
778  nla_put_failure:
779         return -ENOBUFS;
780 }
781
782
783 static int i802_send_eapol(void *priv, const u8 *addr, const u8 *data,
784                            size_t data_len, int encrypt, const u8 *own_addr)
785 {
786         struct i802_driver_data *drv = priv;
787         struct ieee80211_hdr *hdr;
788         size_t len;
789         u8 *pos;
790         int res;
791 #if 0 /* FIX */
792         int qos = sta->flags & WLAN_STA_WME;
793 #else
794         int qos = 0;
795 #endif
796
797         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
798                 data_len;
799         hdr = os_zalloc(len);
800         if (hdr == NULL) {
801                 printf("malloc() failed for i802_send_data(len=%lu)\n",
802                        (unsigned long) len);
803                 return -1;
804         }
805
806         hdr->frame_control =
807                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
808         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
809         if (encrypt)
810                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
811 #if 0 /* To be enabled if qos determination is added above */
812         if (qos) {
813                 hdr->frame_control |=
814                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
815         }
816 #endif
817
818         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
819         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
820         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
821         pos = (u8 *) (hdr + 1);
822
823 #if 0 /* To be enabled if qos determination is added above */
824         if (qos) {
825                 /* add an empty QoS header if needed */
826                 pos[0] = 0;
827                 pos[1] = 0;
828                 pos += 2;
829         }
830 #endif
831
832         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
833         pos += sizeof(rfc1042_header);
834         WPA_PUT_BE16(pos, ETH_P_PAE);
835         pos += 2;
836         memcpy(pos, data, data_len);
837
838         res = i802_send_frame(drv, (u8 *) hdr, len, encrypt, 0);
839         free(hdr);
840
841         if (res < 0) {
842                 perror("i802_send_eapol: send");
843                 printf("i802_send_eapol - packet len: %lu - failed\n",
844                        (unsigned long) len);
845         }
846
847         return res;
848 }
849
850
851 static int i802_sta_add(const char *ifname, void *priv,
852                         struct hostapd_sta_add_params *params)
853 {
854         struct i802_driver_data *drv = priv;
855         struct nl_msg *msg;
856         int ret = -ENOBUFS;
857
858         msg = nlmsg_alloc();
859         if (!msg)
860                 return -ENOMEM;
861
862         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
863                     0, NL80211_CMD_NEW_STATION, 0);
864
865         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
866                     if_nametoindex(drv->iface));
867         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
868         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
869         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
870                 params->supp_rates);
871         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
872                     params->listen_interval);
873
874 #ifdef CONFIG_IEEE80211N
875         if (params->ht_capabilities) {
876                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
877                         params->ht_capabilities->length,
878                         &params->ht_capabilities->data);
879         }
880 #endif /* CONFIG_IEEE80211N */
881
882         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
883         if (ret)
884                 wpa_printf(MSG_DEBUG, "nl80211: NL80211_CMD_NEW_STATION "
885                            "result: %d (%s)", ret, strerror(-ret));
886         if (ret == -EEXIST)
887                 ret = 0;
888  nla_put_failure:
889         return ret;
890 }
891
892
893 static int i802_sta_remove(void *priv, const u8 *addr)
894 {
895         struct i802_driver_data *drv = priv;
896         struct nl_msg *msg;
897         int ret;
898
899         msg = nlmsg_alloc();
900         if (!msg)
901                 return -ENOMEM;
902
903         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
904                     0, NL80211_CMD_DEL_STATION, 0);
905
906         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
907                     if_nametoindex(drv->iface));
908         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
909
910         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
911         if (ret == -ENOENT)
912                 return 0;
913         return ret;
914  nla_put_failure:
915         return -ENOBUFS;
916 }
917
918
919 static int i802_sta_set_flags(void *priv, const u8 *addr,
920                               int total_flags, int flags_or, int flags_and)
921 {
922         struct i802_driver_data *drv = priv;
923         struct nl_msg *msg, *flags = NULL;
924
925         msg = nlmsg_alloc();
926         if (!msg)
927                 return -ENOMEM;
928
929         flags = nlmsg_alloc();
930         if (!flags) {
931                 nlmsg_free(msg);
932                 return -ENOMEM;
933         }
934
935         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
936                     0, NL80211_CMD_SET_STATION, 0);
937
938         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
939                     if_nametoindex(drv->iface));
940         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
941
942         if (total_flags & WLAN_STA_AUTHORIZED || !drv->ieee802_1x_active)
943                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
944
945         if (total_flags & WLAN_STA_WMM)
946                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
947
948         if (total_flags & WLAN_STA_SHORT_PREAMBLE)
949                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
950
951         if (total_flags & WLAN_STA_MFP)
952                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
953
954         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
955                 goto nla_put_failure;
956
957         nlmsg_free(flags);
958
959         return send_and_recv_msgs(drv, msg, NULL, NULL);
960  nla_put_failure:
961         nlmsg_free(flags);
962         return -ENOBUFS;
963 }
964
965
966 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
967                                     int cw_min, int cw_max, int burst_time)
968 {
969         struct i802_driver_data *drv = priv;
970         struct nl_msg *msg;
971         struct nlattr *txq, *params;
972
973         msg = nlmsg_alloc();
974         if (!msg)
975                 return -1;
976
977         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
978                     0, NL80211_CMD_SET_WIPHY, 0);
979
980         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
981
982         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
983         if (!txq)
984                 goto nla_put_failure;
985
986         /* We are only sending parameters for a single TXQ at a time */
987         params = nla_nest_start(msg, 1);
988         if (!params)
989                 goto nla_put_failure;
990
991         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, queue);
992         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
993          * 32 usec, so need to convert the value here. */
994         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
995         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
996         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
997         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
998
999         nla_nest_end(msg, params);
1000
1001         nla_nest_end(msg, txq);
1002
1003         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
1004                 return 0;
1005  nla_put_failure:
1006         return -1;
1007 }
1008
1009
1010 static void nl80211_remove_iface(struct i802_driver_data *drv, int ifidx)
1011 {
1012         struct nl_msg *msg;
1013
1014         /* stop listening for EAPOL on this interface */
1015         del_ifidx(drv, ifidx);
1016
1017         msg = nlmsg_alloc();
1018         if (!msg)
1019                 goto nla_put_failure;
1020
1021         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1022                     0, NL80211_CMD_DEL_INTERFACE, 0);
1023         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
1024
1025         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
1026                 return;
1027  nla_put_failure:
1028         printf("Failed to remove interface.\n");
1029 }
1030
1031
1032 static int nl80211_create_iface(struct i802_driver_data *drv,
1033                                 const char *ifname,
1034                                 enum nl80211_iftype iftype,
1035                                 const u8 *addr)
1036 {
1037         struct nl_msg *msg, *flags = NULL;
1038         int ifidx;
1039         struct ifreq ifreq;
1040         struct iwreq iwr;
1041         int ret = -ENOBUFS;
1042
1043         msg = nlmsg_alloc();
1044         if (!msg)
1045                 return -1;
1046
1047         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1048                     0, NL80211_CMD_NEW_INTERFACE, 0);
1049         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1050         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
1051         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
1052
1053         if (iftype == NL80211_IFTYPE_MONITOR) {
1054                 int err;
1055
1056                 flags = nlmsg_alloc();
1057                 if (!flags)
1058                         goto nla_put_failure;
1059
1060                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1061
1062                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1063
1064                 nlmsg_free(flags);
1065
1066                 if (err)
1067                         goto nla_put_failure;
1068         }
1069
1070         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1071         if (ret) {
1072  nla_put_failure:
1073                 printf("Failed to create interface %s.\n", ifname);
1074                 return ret;
1075         }
1076
1077         ifidx = if_nametoindex(ifname);
1078
1079         if (ifidx <= 0)
1080                 return -1;
1081
1082         /* start listening for EAPOL on this interface */
1083         add_ifidx(drv, ifidx);
1084
1085         if (addr) {
1086                 switch (iftype) {
1087                 case NL80211_IFTYPE_AP:
1088                         os_strlcpy(ifreq.ifr_name, ifname, IFNAMSIZ);
1089                         memcpy(ifreq.ifr_hwaddr.sa_data, addr, ETH_ALEN);
1090                         ifreq.ifr_hwaddr.sa_family = ARPHRD_ETHER;
1091
1092                         if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifreq)) {
1093                                 nl80211_remove_iface(drv, ifidx);
1094                                 return -1;
1095                         }
1096                         break;
1097                 case NL80211_IFTYPE_WDS:
1098                         memset(&iwr, 0, sizeof(iwr));
1099                         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1100                         iwr.u.addr.sa_family = ARPHRD_ETHER;
1101                         memcpy(iwr.u.addr.sa_data, addr, ETH_ALEN);
1102                         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr))
1103                                 return -1;
1104                         break;
1105                 default:
1106                         /* nothing */
1107                         break;
1108                 }
1109         }
1110
1111         return ifidx;
1112 }
1113
1114
1115 static int i802_bss_add(void *priv, const char *ifname, const u8 *bssid)
1116 {
1117         struct i802_driver_data *drv = priv;
1118         int ifidx;
1119         struct i802_bss *bss;
1120
1121         bss = os_zalloc(sizeof(*bss));
1122         if (bss == NULL)
1123                 return -1;
1124         os_strlcpy(bss->iface, ifname, IFNAMSIZ);
1125
1126         ifidx = nl80211_create_iface(priv, ifname, NL80211_IFTYPE_AP, bssid);
1127         if (ifidx < 0) {
1128                 os_free(bss);
1129                 return -1;
1130         }
1131         if (hostapd_set_iface_flags(priv, ifname, 1)) {
1132                 nl80211_remove_iface(priv, ifidx);
1133                 os_free(bss);
1134                 return -1;
1135         }
1136         bss->next = drv->bss.next;
1137         drv->bss.next = bss;
1138         return 0;
1139 }
1140
1141
1142 static int i802_bss_remove(void *priv, const char *ifname)
1143 {
1144         struct i802_driver_data *drv = priv;
1145         struct i802_bss *bss, *prev;
1146         nl80211_remove_iface(priv, if_nametoindex(ifname));
1147         prev = &drv->bss;
1148         bss = drv->bss.next;
1149         while (bss) {
1150                 if (os_strncmp(ifname, bss->iface, IFNAMSIZ) == 0) {
1151                         prev->next = bss->next;
1152                         os_free(bss);
1153                         break;
1154                 }
1155                 prev = bss;
1156                 bss = bss->next;
1157         }
1158         return 0;
1159 }
1160
1161
1162 static int i802_set_beacon(const char *iface, void *priv,
1163                            const u8 *head, size_t head_len,
1164                            const u8 *tail, size_t tail_len, int dtim_period)
1165 {
1166         struct i802_driver_data *drv = priv;
1167         struct nl_msg *msg;
1168         u8 cmd = NL80211_CMD_NEW_BEACON;
1169         int ret;
1170         struct i802_bss *bss;
1171
1172         bss = get_bss(drv, iface);
1173         if (bss == NULL)
1174                 return -ENOENT;
1175
1176         msg = nlmsg_alloc();
1177         if (!msg)
1178                 return -ENOMEM;
1179
1180         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (iface=%s beacon_set=%d)",
1181                    iface, bss->beacon_set);
1182         if (bss->beacon_set)
1183                 cmd = NL80211_CMD_SET_BEACON;
1184
1185         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1186                     0, cmd, 0);
1187         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
1188         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
1189         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
1190         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
1191         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, dtim_period);
1192
1193         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1194         if (!ret)
1195                 bss->beacon_set = 1;
1196         return ret;
1197  nla_put_failure:
1198         return -ENOBUFS;
1199 }
1200
1201
1202 static int i802_del_beacon(struct i802_driver_data *drv)
1203 {
1204         struct nl_msg *msg;
1205
1206         msg = nlmsg_alloc();
1207         if (!msg)
1208                 return -ENOMEM;
1209
1210         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1211                     0, NL80211_CMD_DEL_BEACON, 0);
1212         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1213
1214         return send_and_recv_msgs(drv, msg, NULL, NULL);
1215  nla_put_failure:
1216         return -ENOBUFS;
1217 }
1218
1219
1220 static int i802_set_ieee8021x(const char *ifname, void *priv, int enabled)
1221 {
1222         struct i802_driver_data *drv = priv;
1223
1224         /*
1225          * FIXME: This needs to be per interface (BSS)
1226          */
1227         drv->ieee802_1x_active = enabled;
1228         return 0;
1229 }
1230
1231
1232 static int i802_set_privacy(const char *ifname, void *priv, int enabled)
1233 {
1234         struct i802_driver_data *drv = priv;
1235         struct iwreq iwr;
1236
1237         memset(&iwr, 0, sizeof(iwr));
1238
1239         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1240         iwr.u.param.flags = IW_AUTH_PRIVACY_INVOKED;
1241         iwr.u.param.value = enabled;
1242
1243         ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr);
1244
1245         /* ignore errors, the kernel/driver might not care */
1246         return 0;
1247 }
1248
1249
1250 static int i802_set_internal_bridge(void *priv, int value)
1251 {
1252         return -1;
1253 }
1254
1255
1256 static int i802_set_beacon_int(void *priv, int value)
1257 {
1258         struct i802_driver_data *drv = priv;
1259         struct nl_msg *msg;
1260
1261         drv->beacon_int = value;
1262
1263         if (!drv->bss.beacon_set)
1264                 return 0;
1265
1266         msg = nlmsg_alloc();
1267         if (!msg)
1268                 return -ENOMEM;
1269
1270         wpa_printf(MSG_DEBUG, "nl80211: Set beacon interval %d "
1271                    "(beacon_set=%d)", value, drv->bss.beacon_set);
1272         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1273                     0, NL80211_CMD_SET_BEACON, 0);
1274         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1275
1276         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
1277
1278         return send_and_recv_msgs(drv, msg, NULL, NULL);
1279  nla_put_failure:
1280         return -ENOBUFS;
1281 }
1282
1283
1284 static int i802_set_bss(void *priv, int cts, int preamble, int slot)
1285 {
1286         struct i802_driver_data *drv = priv;
1287         struct nl_msg *msg;
1288
1289         msg = nlmsg_alloc();
1290         if (!msg)
1291                 return -ENOMEM;
1292
1293         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1294                     NL80211_CMD_SET_BSS, 0);
1295
1296         if (cts >= 0)
1297                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
1298         if (preamble >= 0)
1299                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
1300         if (slot >= 0)
1301                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
1302
1303         /* TODO: multi-BSS support */
1304         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1305
1306         return send_and_recv_msgs(drv, msg, NULL, NULL);
1307  nla_put_failure:
1308         return -ENOBUFS;
1309 }
1310
1311
1312 static int i802_set_cts_protect(void *priv, int value)
1313 {
1314         return i802_set_bss(priv, value, -1, -1);
1315 }
1316
1317
1318 static int i802_set_preamble(void *priv, int value)
1319 {
1320         return i802_set_bss(priv, -1, value, -1);
1321 }
1322
1323
1324 static int i802_set_short_slot_time(void *priv, int value)
1325 {
1326         return i802_set_bss(priv, -1, -1, value);
1327 }
1328
1329
1330 static enum nl80211_iftype i802_if_type(enum hostapd_driver_if_type type)
1331 {
1332         switch (type) {
1333         case HOSTAPD_IF_VLAN:
1334                 return NL80211_IFTYPE_AP_VLAN;
1335         case HOSTAPD_IF_WDS:
1336                 return NL80211_IFTYPE_WDS;
1337         }
1338         return -1;
1339 }
1340
1341
1342 static int i802_if_add(const char *iface, void *priv,
1343                        enum hostapd_driver_if_type type, char *ifname,
1344                        const u8 *addr)
1345 {
1346         if (nl80211_create_iface(priv, ifname, i802_if_type(type), addr) < 0)
1347                 return -1;
1348         return 0;
1349 }
1350
1351
1352 static int i802_if_update(void *priv, enum hostapd_driver_if_type type,
1353                           char *ifname, const u8 *addr)
1354 {
1355         /* unused at the moment */
1356         return -1;
1357 }
1358
1359
1360 static int i802_if_remove(void *priv, enum hostapd_driver_if_type type,
1361                           const char *ifname, const u8 *addr)
1362 {
1363         nl80211_remove_iface(priv, if_nametoindex(ifname));
1364         return 0;
1365 }
1366
1367
1368 struct phy_info_arg {
1369         u16 *num_modes;
1370         struct hostapd_hw_modes *modes;
1371 };
1372
1373 static int phy_info_handler(struct nl_msg *msg, void *arg)
1374 {
1375         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1376         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1377         struct phy_info_arg *phy_info = arg;
1378
1379         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1380
1381         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1382         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1383                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1384                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1385                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1386                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
1387                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1388                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1389         };
1390
1391         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1392         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1393                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1394                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
1395         };
1396
1397         struct nlattr *nl_band;
1398         struct nlattr *nl_freq;
1399         struct nlattr *nl_rate;
1400         int rem_band, rem_freq, rem_rate;
1401         struct hostapd_hw_modes *mode;
1402         int idx, mode_is_set;
1403
1404         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1405                   genlmsg_attrlen(gnlh, 0), NULL);
1406
1407         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1408                 return NL_SKIP;
1409
1410         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
1411                 mode = realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
1412                 if (!mode)
1413                         return NL_SKIP;
1414                 phy_info->modes = mode;
1415
1416                 mode_is_set = 0;
1417
1418                 mode = &phy_info->modes[*(phy_info->num_modes)];
1419                 memset(mode, 0, sizeof(*mode));
1420                 *(phy_info->num_modes) += 1;
1421
1422                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1423                           nla_len(nl_band), NULL);
1424
1425                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
1426                         mode->ht_capab = nla_get_u16(
1427                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
1428                 }
1429
1430                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1431                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1432                                   nla_len(nl_freq), freq_policy);
1433                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1434                                 continue;
1435                         mode->num_channels++;
1436                 }
1437
1438                 mode->channels = calloc(mode->num_channels, sizeof(struct hostapd_channel_data));
1439                 if (!mode->channels)
1440                         return NL_SKIP;
1441
1442                 idx = 0;
1443
1444                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1445                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1446                                   nla_len(nl_freq), freq_policy);
1447                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1448                                 continue;
1449
1450                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1451                         mode->channels[idx].flag = 0;
1452
1453                         if (!mode_is_set) {
1454                                 /* crude heuristic */
1455                                 if (mode->channels[idx].freq < 4000)
1456                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
1457                                 else
1458                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
1459                                 mode_is_set = 1;
1460                         }
1461
1462                         /* crude heuristic */
1463                         if (mode->channels[idx].freq < 4000)
1464                                 if (mode->channels[idx].freq == 2848)
1465                                         mode->channels[idx].chan = 14;
1466                                 else
1467                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
1468                         else
1469                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
1470
1471                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1472                                 mode->channels[idx].flag |=
1473                                         HOSTAPD_CHAN_DISABLED;
1474                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
1475                                 mode->channels[idx].flag |=
1476                                         HOSTAPD_CHAN_PASSIVE_SCAN;
1477                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
1478                                 mode->channels[idx].flag |=
1479                                         HOSTAPD_CHAN_NO_IBSS;
1480                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1481                                 mode->channels[idx].flag |=
1482                                         HOSTAPD_CHAN_RADAR;
1483
1484                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
1485                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1486                                 mode->channels[idx].max_tx_power =
1487                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
1488
1489                         idx++;
1490                 }
1491
1492                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1493                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1494                                   nla_len(nl_rate), rate_policy);
1495                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1496                                 continue;
1497                         mode->num_rates++;
1498                 }
1499
1500                 mode->rates = calloc(mode->num_rates, sizeof(struct hostapd_rate_data));
1501                 if (!mode->rates)
1502                         return NL_SKIP;
1503
1504                 idx = 0;
1505
1506                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1507                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1508                                   nla_len(nl_rate), rate_policy);
1509                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1510                                 continue;
1511                         mode->rates[idx].rate = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
1512
1513                         /* crude heuristic */
1514                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
1515                             mode->rates[idx].rate > 200)
1516                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
1517
1518                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
1519                                 mode->rates[idx].flags |= HOSTAPD_RATE_PREAMBLE2;
1520
1521                         idx++;
1522                 }
1523         }
1524
1525         return NL_SKIP;
1526 }
1527
1528 static struct hostapd_hw_modes *i802_add_11b(struct hostapd_hw_modes *modes,
1529                                              u16 *num_modes)
1530 {
1531         u16 m;
1532         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1533         int i, mode11g_idx = -1;
1534
1535         /* If only 802.11g mode is included, use it to construct matching
1536          * 802.11b mode data. */
1537
1538         for (m = 0; m < *num_modes; m++) {
1539                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1540                         return modes; /* 802.11b already included */
1541                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1542                         mode11g_idx = m;
1543         }
1544
1545         if (mode11g_idx < 0)
1546                 return modes; /* 2.4 GHz band not supported at all */
1547
1548         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
1549         if (nmodes == NULL)
1550                 return modes; /* Could not add 802.11b mode */
1551
1552         mode = &nmodes[*num_modes];
1553         os_memset(mode, 0, sizeof(*mode));
1554         (*num_modes)++;
1555         modes = nmodes;
1556
1557         mode->mode = HOSTAPD_MODE_IEEE80211B;
1558
1559         mode11g = &modes[mode11g_idx];
1560         mode->num_channels = mode11g->num_channels;
1561         mode->channels = os_malloc(mode11g->num_channels *
1562                                    sizeof(struct hostapd_channel_data));
1563         if (mode->channels == NULL) {
1564                 (*num_modes)--;
1565                 return modes; /* Could not add 802.11b mode */
1566         }
1567         os_memcpy(mode->channels, mode11g->channels,
1568                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
1569
1570         mode->num_rates = 0;
1571         mode->rates = os_malloc(4 * sizeof(struct hostapd_rate_data));
1572         if (mode->rates == NULL) {
1573                 os_free(mode->channels);
1574                 (*num_modes)--;
1575                 return modes; /* Could not add 802.11b mode */
1576         }
1577
1578         for (i = 0; i < mode11g->num_rates; i++) {
1579                 if (mode11g->rates[i].rate > 110 ||
1580                     mode11g->rates[i].flags &
1581                     (HOSTAPD_RATE_ERP | HOSTAPD_RATE_OFDM))
1582                         continue;
1583                 mode->rates[mode->num_rates] = mode11g->rates[i];
1584                 mode->num_rates++;
1585                 if (mode->num_rates == 4)
1586                         break;
1587         }
1588
1589         if (mode->num_rates == 0) {
1590                 os_free(mode->channels);
1591                 os_free(mode->rates);
1592                 (*num_modes)--;
1593                 return modes; /* No 802.11b rates */
1594         }
1595
1596         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1597                    "information");
1598
1599         return modes;
1600 }
1601
1602 static struct hostapd_hw_modes *i802_get_hw_feature_data(void *priv,
1603                                                          u16 *num_modes,
1604                                                          u16 *flags)
1605 {
1606         struct i802_driver_data *drv = priv;
1607         struct nl_msg *msg;
1608         struct phy_info_arg result = {
1609                 .num_modes = num_modes,
1610                 .modes = NULL,
1611         };
1612
1613         *num_modes = 0;
1614         *flags = 0;
1615
1616         msg = nlmsg_alloc();
1617         if (!msg)
1618                 return NULL;
1619
1620         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1621                     0, NL80211_CMD_GET_WIPHY, 0);
1622
1623         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1624
1625         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
1626                 return i802_add_11b(result.modes, num_modes);
1627  nla_put_failure:
1628         return NULL;
1629 }
1630
1631
1632 static int i802_set_sta_vlan(void *priv, const u8 *addr,
1633                              const char *ifname, int vlan_id)
1634 {
1635         struct i802_driver_data *drv = priv;
1636         struct nl_msg *msg;
1637
1638         msg = nlmsg_alloc();
1639         if (!msg)
1640                 return -ENOMEM;
1641
1642         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1643                     0, NL80211_CMD_SET_STATION, 0);
1644
1645         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1646                     if_nametoindex(drv->iface));
1647         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1648         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1649                     if_nametoindex(ifname));
1650
1651         return send_and_recv_msgs(drv, msg, NULL, NULL);
1652  nla_put_failure:
1653         return -ENOBUFS;
1654 }
1655
1656
1657 static int i802_set_country(void *priv, const char *country)
1658 {
1659         struct i802_driver_data *drv = priv;
1660         struct nl_msg *msg;
1661         char alpha2[3];
1662
1663         msg = nlmsg_alloc();
1664         if (!msg)
1665                 return -ENOMEM;
1666
1667         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1668                     0, NL80211_CMD_REQ_SET_REG, 0);
1669
1670         alpha2[0] = country[0];
1671         alpha2[1] = country[1];
1672         alpha2[2] = '\0';
1673         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
1674
1675         return send_and_recv_msgs(drv, msg, NULL, NULL);
1676  nla_put_failure:
1677         return -ENOBUFS;
1678 }
1679
1680
1681 static void handle_tx_callback(struct hostapd_data *hapd, u8 *buf, size_t len,
1682                                int ok)
1683 {
1684         struct ieee80211_hdr *hdr;
1685         u16 fc, type, stype;
1686
1687         hdr = (struct ieee80211_hdr *) buf;
1688         fc = le_to_host16(hdr->frame_control);
1689
1690         type = WLAN_FC_GET_TYPE(fc);
1691         stype = WLAN_FC_GET_STYPE(fc);
1692
1693         switch (type) {
1694         case WLAN_FC_TYPE_MGMT:
1695                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
1696                            ok ? "ACK" : "fail");
1697                 hostapd_mgmt_tx_cb(hapd, buf, len, stype, ok);
1698                 break;
1699         case WLAN_FC_TYPE_CTRL:
1700                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
1701                            ok ? "ACK" : "fail");
1702                 break;
1703         case WLAN_FC_TYPE_DATA:
1704                 hostapd_tx_status(hapd, hdr->addr1, buf, len, ok);
1705                 break;
1706         default:
1707                 printf("unknown TX callback frame type %d\n", type);
1708                 break;
1709         }
1710 }
1711
1712
1713 static void handle_frame(struct i802_driver_data *drv,
1714                          struct hostapd_iface *iface, u8 *buf, size_t len,
1715                          struct hostapd_frame_info *hfi,
1716                          enum ieee80211_msg_type msg_type)
1717 {
1718         struct ieee80211_hdr *hdr;
1719         u16 fc, type, stype;
1720         size_t data_len = len;
1721         struct hostapd_data *hapd = NULL;
1722         int broadcast_bssid = 0;
1723         size_t i;
1724         u8 *bssid;
1725
1726         /*
1727          * PS-Poll frames are 16 bytes. All other frames are
1728          * 24 bytes or longer.
1729          */
1730         if (len < 16)
1731                 return;
1732
1733         hdr = (struct ieee80211_hdr *) buf;
1734         fc = le_to_host16(hdr->frame_control);
1735
1736         type = WLAN_FC_GET_TYPE(fc);
1737         stype = WLAN_FC_GET_STYPE(fc);
1738
1739         switch (type) {
1740         case WLAN_FC_TYPE_DATA:
1741                 if (len < 24)
1742                         return;
1743                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
1744                 case WLAN_FC_TODS:
1745                         bssid = hdr->addr1;
1746                         break;
1747                 case WLAN_FC_FROMDS:
1748                         bssid = hdr->addr2;
1749                         break;
1750                 default:
1751                         /* discard */
1752                         return;
1753                 }
1754                 break;
1755         case WLAN_FC_TYPE_CTRL:
1756                 /* discard non-ps-poll frames */
1757                 if (stype != WLAN_FC_STYPE_PSPOLL)
1758                         return;
1759                 bssid = hdr->addr1;
1760                 break;
1761         case WLAN_FC_TYPE_MGMT:
1762                 bssid = hdr->addr3;
1763                 break;
1764         default:
1765                 /* discard */
1766                 return;
1767         }
1768
1769         /* find interface frame belongs to */
1770         for (i = 0; i < iface->num_bss; i++) {
1771                 if (memcmp(bssid, iface->bss[i]->own_addr, ETH_ALEN) == 0) {
1772                         hapd = iface->bss[i];
1773                         break;
1774                 }
1775         }
1776
1777         if (hapd == NULL) {
1778                 hapd = iface->bss[0];
1779
1780                 if (bssid[0] != 0xff || bssid[1] != 0xff ||
1781                     bssid[2] != 0xff || bssid[3] != 0xff ||
1782                     bssid[4] != 0xff || bssid[5] != 0xff) {
1783                         /*
1784                          * Unknown BSSID - drop frame if this is not from
1785                          * passive scanning or a beacon (at least ProbeReq
1786                          * frames to other APs may be allowed through RX
1787                          * filtering in the wlan hw/driver)
1788                          */
1789                         if ((type != WLAN_FC_TYPE_MGMT ||
1790                              stype != WLAN_FC_STYPE_BEACON))
1791                                 return;
1792                 } else
1793                         broadcast_bssid = 1;
1794         }
1795
1796         switch (msg_type) {
1797         case ieee80211_msg_normal:
1798                 /* continue processing */
1799                 break;
1800         case ieee80211_msg_tx_callback_ack:
1801                 handle_tx_callback(hapd, buf, data_len, 1);
1802                 return;
1803         case ieee80211_msg_tx_callback_fail:
1804                 handle_tx_callback(hapd, buf, data_len, 0);
1805                 return;
1806         }
1807
1808         switch (type) {
1809         case WLAN_FC_TYPE_MGMT:
1810                 if (stype != WLAN_FC_STYPE_BEACON &&
1811                     stype != WLAN_FC_STYPE_PROBE_REQ)
1812                         wpa_printf(MSG_MSGDUMP, "MGMT");
1813                 if (broadcast_bssid) {
1814                         for (i = 0; i < iface->num_bss; i++)
1815                                 hostapd_mgmt_rx(iface->bss[i], buf, data_len,
1816                                                 stype, hfi);
1817                 } else
1818                         hostapd_mgmt_rx(hapd, buf, data_len, stype, hfi);
1819                 break;
1820         case WLAN_FC_TYPE_CTRL:
1821                 /* can only get here with PS-Poll frames */
1822                 wpa_printf(MSG_DEBUG, "CTRL");
1823                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
1824                 break;
1825         case WLAN_FC_TYPE_DATA:
1826                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
1827                 break;
1828         }
1829 }
1830
1831
1832 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
1833 {
1834         struct i802_driver_data *drv = eloop_ctx;
1835         struct hostapd_data *hapd = drv->hapd;
1836         struct sockaddr_ll lladdr;
1837         unsigned char buf[3000];
1838         int len;
1839         socklen_t fromlen = sizeof(lladdr);
1840
1841         len = recvfrom(sock, buf, sizeof(buf), 0,
1842                        (struct sockaddr *)&lladdr, &fromlen);
1843         if (len < 0) {
1844                 perror("recv");
1845                 return;
1846         }
1847
1848         if (have_ifidx(drv, lladdr.sll_ifindex))
1849                 hostapd_eapol_receive(hapd, lladdr.sll_addr, buf, len);
1850 }
1851
1852
1853 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1854 {
1855         struct i802_driver_data *drv = eloop_ctx;
1856         int len;
1857         unsigned char buf[3000];
1858         struct hostapd_data *hapd = drv->hapd;
1859         struct ieee80211_radiotap_iterator iter;
1860         int ret;
1861         struct hostapd_frame_info hfi;
1862         int injected = 0, failed = 0, msg_type, rxflags = 0;
1863
1864         len = recv(sock, buf, sizeof(buf), 0);
1865         if (len < 0) {
1866                 perror("recv");
1867                 return;
1868         }
1869
1870         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
1871                 printf("received invalid radiotap frame\n");
1872                 return;
1873         }
1874
1875         memset(&hfi, 0, sizeof(hfi));
1876
1877         while (1) {
1878                 ret = ieee80211_radiotap_iterator_next(&iter);
1879                 if (ret == -ENOENT)
1880                         break;
1881                 if (ret) {
1882                         printf("received invalid radiotap frame (%d)\n", ret);
1883                         return;
1884                 }
1885                 switch (iter.this_arg_index) {
1886                 case IEEE80211_RADIOTAP_FLAGS:
1887                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1888                                 len -= 4;
1889                         break;
1890                 case IEEE80211_RADIOTAP_RX_FLAGS:
1891                         rxflags = 1;
1892                         break;
1893                 case IEEE80211_RADIOTAP_TX_FLAGS:
1894                         injected = 1;
1895                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
1896                                         IEEE80211_RADIOTAP_F_TX_FAIL;
1897                         break;
1898                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1899                         break;
1900                 case IEEE80211_RADIOTAP_CHANNEL:
1901                         /* TODO convert from freq/flags to channel number
1902                         hfi.channel = XXX;
1903                         hfi.phytype = XXX;
1904                          */
1905                         break;
1906                 case IEEE80211_RADIOTAP_RATE:
1907                         hfi.datarate = *iter.this_arg * 5;
1908                         break;
1909                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1910                         hfi.ssi_signal = *iter.this_arg;
1911                         break;
1912                 }
1913         }
1914
1915         if (rxflags && injected)
1916                 return;
1917
1918         if (!injected)
1919                 msg_type = ieee80211_msg_normal;
1920         else if (failed)
1921                 msg_type = ieee80211_msg_tx_callback_fail;
1922         else
1923                 msg_type = ieee80211_msg_tx_callback_ack;
1924
1925         handle_frame(drv, hapd->iface, buf + iter.max_length,
1926                      len - iter.max_length, &hfi, msg_type);
1927 }
1928
1929
1930 /*
1931  * we post-process the filter code later and rewrite
1932  * this to the offset to the last instruction
1933  */
1934 #define PASS    0xFF
1935 #define FAIL    0xFE
1936
1937 static struct sock_filter msock_filter_insns[] = {
1938         /*
1939          * do a little-endian load of the radiotap length field
1940          */
1941         /* load lower byte into A */
1942         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
1943         /* put it into X (== index register) */
1944         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1945         /* load upper byte into A */
1946         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
1947         /* left-shift it by 8 */
1948         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
1949         /* or with X */
1950         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
1951         /* put result into X */
1952         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1953
1954         /*
1955          * Allow management frames through, this also gives us those
1956          * management frames that we sent ourselves with status
1957          */
1958         /* load the lower byte of the IEEE 802.11 frame control field */
1959         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
1960         /* mask off frame type and version */
1961         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
1962         /* accept frame if it's both 0, fall through otherwise */
1963         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
1964
1965         /*
1966          * TODO: add a bit to radiotap RX flags that indicates
1967          * that the sending station is not associated, then
1968          * add a filter here that filters on our DA and that flag
1969          * to allow us to deauth frames to that bad station.
1970          *
1971          * Not a regression -- we didn't do it before either.
1972          */
1973
1974 #if 0
1975         /*
1976          * drop non-data frames, WDS frames
1977          */
1978         /* load the lower byte of the frame control field */
1979         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1980         /* mask off QoS bit */
1981         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
1982         /* drop non-data frames */
1983         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
1984         /* load the upper byte of the frame control field */
1985         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1986         /* mask off toDS/fromDS */
1987         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
1988         /* drop WDS frames */
1989         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, FAIL, 0),
1990 #endif
1991
1992         /*
1993          * add header length to index
1994          */
1995         /* load the lower byte of the frame control field */
1996         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1997         /* mask off QoS bit */
1998         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
1999         /* right shift it by 6 to give 0 or 2 */
2000         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
2001         /* add data frame header length */
2002         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
2003         /* add index, was start of 802.11 header */
2004         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
2005         /* move to index, now start of LL header */
2006         BPF_STMT(BPF_MISC | BPF_TAX, 0),
2007
2008         /*
2009          * Accept empty data frames, we use those for
2010          * polling activity.
2011          */
2012         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
2013         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
2014
2015         /*
2016          * Accept EAPOL frames
2017          */
2018         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
2019         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
2020         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
2021         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
2022
2023         /* keep these last two statements or change the code below */
2024         /* return 0 == "DROP" */
2025         BPF_STMT(BPF_RET | BPF_K, 0),
2026         /* return ~0 == "keep all" */
2027         BPF_STMT(BPF_RET | BPF_K, ~0),
2028 };
2029
2030 static struct sock_fprog msock_filter = {
2031         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
2032         .filter = msock_filter_insns,
2033 };
2034
2035
2036 static int add_monitor_filter(int s)
2037 {
2038         int idx;
2039
2040         /* rewrite all PASS/FAIL jump offsets */
2041         for (idx = 0; idx < msock_filter.len; idx++) {
2042                 struct sock_filter *insn = &msock_filter_insns[idx];
2043
2044                 if (BPF_CLASS(insn->code) == BPF_JMP) {
2045                         if (insn->code == (BPF_JMP|BPF_JA)) {
2046                                 if (insn->k == PASS)
2047                                         insn->k = msock_filter.len - idx - 2;
2048                                 else if (insn->k == FAIL)
2049                                         insn->k = msock_filter.len - idx - 3;
2050                         }
2051
2052                         if (insn->jt == PASS)
2053                                 insn->jt = msock_filter.len - idx - 2;
2054                         else if (insn->jt == FAIL)
2055                                 insn->jt = msock_filter.len - idx - 3;
2056
2057                         if (insn->jf == PASS)
2058                                 insn->jf = msock_filter.len - idx - 2;
2059                         else if (insn->jf == FAIL)
2060                                 insn->jf = msock_filter.len - idx - 3;
2061                 }
2062         }
2063
2064         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
2065                        &msock_filter, sizeof(msock_filter))) {
2066                 perror("SO_ATTACH_FILTER");
2067                 return -1;
2068         }
2069
2070         return 0;
2071 }
2072
2073
2074 static int nl80211_create_monitor_interface(struct i802_driver_data *drv)
2075 {
2076         char buf[IFNAMSIZ];
2077         struct sockaddr_ll ll;
2078         int optval;
2079         socklen_t optlen;
2080
2081         snprintf(buf, IFNAMSIZ, "mon.%s", drv->iface);
2082         buf[IFNAMSIZ - 1] = '\0';
2083
2084         drv->monitor_ifidx =
2085                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL);
2086
2087         if (drv->monitor_ifidx < 0)
2088                 return -1;
2089
2090         if (hostapd_set_iface_flags(drv, buf, 1))
2091                 goto error;
2092
2093         memset(&ll, 0, sizeof(ll));
2094         ll.sll_family = AF_PACKET;
2095         ll.sll_ifindex = drv->monitor_ifidx;
2096         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
2097         if (drv->monitor_sock < 0) {
2098                 perror("socket[PF_PACKET,SOCK_RAW]");
2099                 goto error;
2100         }
2101
2102         if (add_monitor_filter(drv->monitor_sock)) {
2103                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
2104                            "interface; do filtering in user space");
2105                 /* This works, but will cost in performance. */
2106         }
2107
2108         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
2109                  sizeof(ll)) < 0) {
2110                 perror("monitor socket bind");
2111                 goto error;
2112         }
2113
2114         optlen = sizeof(optval);
2115         optval = 20;
2116         if (setsockopt
2117             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
2118                 perror("Failed to set socket priority");
2119                 goto error;
2120         }
2121
2122         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
2123                                      drv, NULL)) {
2124                 printf("Could not register monitor read socket\n");
2125                 goto error;
2126         }
2127
2128         return 0;
2129  error:
2130         nl80211_remove_iface(drv, drv->monitor_ifidx);
2131         return -1;
2132 }
2133
2134
2135 static int nl80211_set_mode(struct i802_driver_data *drv, const char *ifname,
2136                             int mode)
2137 {
2138         struct nl_msg *msg;
2139         int ret = -ENOBUFS;
2140
2141         msg = nlmsg_alloc();
2142         if (!msg)
2143                 return -ENOMEM;
2144
2145         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2146                     0, NL80211_CMD_SET_INTERFACE, 0);
2147         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
2148                     if_nametoindex(ifname));
2149         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
2150
2151         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2152         if (!ret)
2153                 return 0;
2154  nla_put_failure:
2155         wpa_printf(MSG_ERROR, "Failed to set interface %s to master "
2156                    "mode.", ifname);
2157         return ret;
2158 }
2159
2160
2161 #ifdef CONFIG_IEEE80211N
2162 static void i802_add_neighbor(struct i802_driver_data *drv, u8 *bssid,
2163                               int freq, u8 *ie, size_t ie_len)
2164 {
2165         struct ieee802_11_elems elems;
2166         int ht, pri_chan = 0, sec_chan = 0;
2167         struct ieee80211_ht_operation *oper;
2168         struct hostapd_neighbor_bss *nnei;
2169
2170         ieee802_11_parse_elems(ie, ie_len, &elems, 0);
2171         ht = elems.ht_capabilities || elems.ht_operation;
2172         if (elems.ht_operation && elems.ht_operation_len >= sizeof(*oper)) {
2173                 oper = (struct ieee80211_ht_operation *) elems.ht_operation;
2174                 pri_chan = oper->control_chan;
2175                 if (oper->ht_param & HT_INFO_HT_PARAM_REC_TRANS_CHNL_WIDTH) {
2176                         if (oper->ht_param &
2177                             HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE)
2178                                 sec_chan = pri_chan + 4;
2179                         else if (oper->ht_param &
2180                             HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW)
2181                                 sec_chan = pri_chan - 4;
2182                 }
2183         }
2184
2185         wpa_printf(MSG_DEBUG, "nl80211: Neighboring BSS - bssid=" MACSTR
2186                    " freq=%d MHz HT=%d pri_chan=%d sec_chan=%d",
2187                    MAC2STR(bssid), freq, ht, pri_chan, sec_chan);
2188
2189         nnei = os_realloc(drv->neighbors, (drv->num_neighbors + 1) *
2190                           sizeof(struct hostapd_neighbor_bss));
2191         if (nnei == NULL)
2192                 return;
2193         drv->neighbors = nnei;
2194         nnei = &nnei[drv->num_neighbors];
2195         os_memcpy(nnei->bssid, bssid, ETH_ALEN);
2196         nnei->freq = freq;
2197         nnei->ht = !!ht;
2198         nnei->pri_chan = pri_chan;
2199         nnei->sec_chan = sec_chan;
2200         drv->num_neighbors++;
2201 }
2202
2203
2204 static int i802_get_scan_freq(struct iw_event *iwe, int *freq)
2205 {
2206         int divi = 1000000, i;
2207
2208         if (iwe->u.freq.e == 0) {
2209                 /*
2210                  * Some drivers do not report frequency, but a channel.
2211                  * Try to map this to frequency by assuming they are using
2212                  * IEEE 802.11b/g.  But don't overwrite a previously parsed
2213                  * frequency if the driver sends both frequency and channel,
2214                  * since the driver may be sending an A-band channel that we
2215                  * don't handle here.
2216                  */
2217
2218                 if (*freq)
2219                         return 0;
2220
2221                 if (iwe->u.freq.m >= 1 && iwe->u.freq.m <= 13) {
2222                         *freq = 2407 + 5 * iwe->u.freq.m;
2223                         return 0;
2224                 } else if (iwe->u.freq.m == 14) {
2225                         *freq = 2484;
2226                         return 0;
2227                 }
2228         }
2229
2230         if (iwe->u.freq.e > 6) {
2231                 wpa_printf(MSG_DEBUG, "Invalid freq in scan results: "
2232                            "m=%d e=%d", iwe->u.freq.m, iwe->u.freq.e);
2233                 return -1;
2234         }
2235
2236         for (i = 0; i < iwe->u.freq.e; i++)
2237                 divi /= 10;
2238         *freq = iwe->u.freq.m / divi;
2239         return 0;
2240 }
2241
2242
2243 static int i802_parse_scan(struct i802_driver_data *drv, u8 *res_buf,
2244                            size_t len)
2245 {
2246         size_t ap_num = 0;
2247         int first;
2248         struct iw_event iwe_buf, *iwe = &iwe_buf;
2249         char *pos, *end, *custom;
2250         u8 bssid[ETH_ALEN];
2251         int freq = 0;
2252         u8 *ie = NULL;
2253         size_t ie_len = 0;
2254
2255         ap_num = 0;
2256         first = 1;
2257
2258         pos = (char *) res_buf;
2259         end = (char *) res_buf + len;
2260
2261         while (pos + IW_EV_LCP_LEN <= end) {
2262                 /* Event data may be unaligned, so make a local, aligned copy
2263                  * before processing. */
2264                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2265                 if (iwe->len <= IW_EV_LCP_LEN)
2266                         break;
2267
2268                 custom = pos + IW_EV_POINT_LEN;
2269                 if (iwe->cmd == IWEVGENIE) {
2270                         /* WE-19 removed the pointer from struct iw_point */
2271                         char *dpos = (char *) &iwe_buf.u.data.length;
2272                         int dlen = dpos - (char *) &iwe_buf;
2273                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
2274                                   sizeof(struct iw_event) - dlen);
2275                 } else {
2276                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
2277                         custom += IW_EV_POINT_OFF;
2278                 }
2279
2280                 switch (iwe->cmd) {
2281                 case SIOCGIWAP:
2282                         if (!first)
2283                                 i802_add_neighbor(drv, bssid, freq, ie,
2284                                                   ie_len);
2285                         first = 0;
2286                         os_memcpy(bssid, iwe->u.ap_addr.sa_data, ETH_ALEN);
2287                         freq = 0;
2288                         ie = NULL;
2289                         ie_len = 0;
2290                         break;
2291                 case SIOCGIWFREQ:
2292                         i802_get_scan_freq(iwe, &freq);
2293                         break;
2294                 case IWEVGENIE:
2295                         if (custom + iwe->u.data.length > end) {
2296                                 wpa_printf(MSG_ERROR, "IWEVGENIE overflow");
2297                                 return -1;
2298                         }
2299                         ie = (u8 *) custom;
2300                         ie_len = iwe->u.data.length;
2301                         break;
2302                 }
2303
2304                 pos += iwe->len;
2305         }
2306
2307         if (!first)
2308                 i802_add_neighbor(drv, bssid, freq, ie, ie_len);
2309
2310         return 0;
2311 }
2312
2313
2314 static int i802_get_ht_scan_res(struct i802_driver_data *drv)
2315 {
2316         struct iwreq iwr;
2317         u8 *res_buf;
2318         size_t res_buf_len;
2319         int res;
2320
2321         res_buf_len = IW_SCAN_MAX_DATA;
2322         for (;;) {
2323                 res_buf = os_malloc(res_buf_len);
2324                 if (res_buf == NULL)
2325                         return -1;
2326                 os_memset(&iwr, 0, sizeof(iwr));
2327                 os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2328                 iwr.u.data.pointer = res_buf;
2329                 iwr.u.data.length = res_buf_len;
2330
2331                 if (ioctl(drv->ioctl_sock, SIOCGIWSCAN, &iwr) == 0)
2332                         break;
2333
2334                 if (errno == E2BIG && res_buf_len < 65535) {
2335                         os_free(res_buf);
2336                         res_buf = NULL;
2337                         res_buf_len *= 2;
2338                         if (res_buf_len > 65535)
2339                                 res_buf_len = 65535; /* 16-bit length field */
2340                         wpa_printf(MSG_DEBUG, "Scan results did not fit - "
2341                                    "trying larger buffer (%lu bytes)",
2342                                    (unsigned long) res_buf_len);
2343                 } else {
2344                         perror("ioctl[SIOCGIWSCAN]");
2345                         os_free(res_buf);
2346                         return -1;
2347                 }
2348         }
2349
2350         if (iwr.u.data.length > res_buf_len) {
2351                 os_free(res_buf);
2352                 return -1;
2353         }
2354
2355         res = i802_parse_scan(drv, res_buf, iwr.u.data.length);
2356         os_free(res_buf);
2357
2358         return res;
2359 }
2360
2361
2362 static int i802_is_event_wireless_scan_complete(char *data, int len)
2363 {
2364         struct iw_event iwe_buf, *iwe = &iwe_buf;
2365         char *pos, *end;
2366
2367         pos = data;
2368         end = data + len;
2369
2370         while (pos + IW_EV_LCP_LEN <= end) {
2371                 /* Event data may be unaligned, so make a local, aligned copy
2372                  * before processing. */
2373                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2374                 if (iwe->cmd == SIOCGIWSCAN)
2375                         return 1;
2376
2377                 pos += iwe->len;
2378         }
2379
2380         return 0;
2381 }
2382
2383
2384 static int i802_is_rtm_scan_complete(int ifindex, struct nlmsghdr *h, int len)
2385 {
2386         struct ifinfomsg *ifi;
2387         int attrlen, _nlmsg_len, rta_len;
2388         struct rtattr *attr;
2389
2390         if (len < (int) sizeof(*ifi))
2391                 return 0;
2392
2393         ifi = NLMSG_DATA(h);
2394
2395         if (ifindex != ifi->ifi_index)
2396                 return 0; /* event for foreign ifindex */
2397
2398         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
2399
2400         attrlen = h->nlmsg_len - _nlmsg_len;
2401         if (attrlen < 0)
2402                 return 0;
2403
2404         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
2405
2406         rta_len = RTA_ALIGN(sizeof(struct rtattr));
2407         while (RTA_OK(attr, attrlen)) {
2408                 if (attr->rta_type == IFLA_WIRELESS &&
2409                     i802_is_event_wireless_scan_complete(
2410                             ((char *) attr) + rta_len,
2411                             attr->rta_len - rta_len))
2412                         return 1;
2413                 attr = RTA_NEXT(attr, attrlen);
2414         }
2415
2416         return 0;
2417 }
2418
2419
2420 static int i802_is_scan_complete(int s, int ifindex)
2421 {
2422         char buf[1024];
2423         int left;
2424         struct nlmsghdr *h;
2425
2426         left = recv(s, buf, sizeof(buf), MSG_DONTWAIT);
2427         if (left < 0) {
2428                 perror("recv(netlink)");
2429                 return 0;
2430         }
2431
2432         h = (struct nlmsghdr *) buf;
2433         while (left >= (int) sizeof(*h)) {
2434                 int len, plen;
2435
2436                 len = h->nlmsg_len;
2437                 plen = len - sizeof(*h);
2438                 if (len > left || plen < 0) {
2439                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
2440                                    "len=%d left=%d plen=%d",
2441                                    len, left, plen);
2442                         break;
2443                 }
2444
2445                 switch (h->nlmsg_type) {
2446                 case RTM_NEWLINK:
2447                         if (i802_is_rtm_scan_complete(ifindex, h, plen))
2448                                 return 1;
2449                         break;
2450                 }
2451
2452                 len = NLMSG_ALIGN(len);
2453                 left -= len;
2454                 h = (struct nlmsghdr *) ((char *) h + len);
2455         }
2456
2457         return 0;
2458 }
2459
2460
2461 static int i802_ht_scan(struct i802_driver_data *drv)
2462 {
2463         struct iwreq iwr;
2464         int s, res, ifindex;
2465         struct sockaddr_nl local;
2466         time_t now, end;
2467         fd_set rfds;
2468         struct timeval tv;
2469
2470         wpa_printf(MSG_DEBUG, "nl80211: Scanning overlapping BSSes before "
2471                    "starting HT 20/40 MHz BSS");
2472
2473         /* Request a new scan */
2474         /* TODO: would be enough to scan the selected band */
2475         os_memset(&iwr, 0, sizeof(iwr));
2476         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2477         if (ioctl(drv->ioctl_sock, SIOCSIWSCAN, &iwr) < 0) {
2478                 perror("ioctl[SIOCSIWSCAN]");
2479                 return -1;
2480         }
2481
2482         ifindex = if_nametoindex(drv->iface);
2483
2484         /* Wait for scan completion event or timeout */
2485         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
2486         if (s < 0) {
2487                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
2488                 return -1;
2489         }
2490
2491         os_memset(&local, 0, sizeof(local));
2492         local.nl_family = AF_NETLINK;
2493         local.nl_groups = RTMGRP_LINK;
2494         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
2495                 perror("bind(netlink)");
2496                 close(s);
2497                 return -1;
2498         }
2499
2500         time(&end);
2501         end += 30; /* Wait at most 30 seconds for scan results */
2502         for (;;) {
2503                 time(&now);
2504                 tv.tv_sec = end > now ? end - now : 0;
2505                 tv.tv_usec = 0;
2506                 FD_ZERO(&rfds);
2507                 FD_SET(s, &rfds);
2508                 res = select(s + 1, &rfds, NULL, NULL, &tv);
2509                 if (res < 0) {
2510                         perror("select");
2511                         /* Assume results are ready after 10 seconds wait */
2512                         os_sleep(10, 0);
2513                         break;
2514                 } else if (res) {
2515                         if (i802_is_scan_complete(s, ifindex)) {
2516                                 wpa_printf(MSG_DEBUG, "nl80211: Scan "
2517                                            "completed");
2518                                 break;
2519                         }
2520                 } else {
2521                         wpa_printf(MSG_DEBUG, "nl80211: Scan timeout");
2522                         /* Assume results are ready to be read now */
2523                         break;
2524                 }
2525         }
2526
2527         close(s);
2528
2529         return i802_get_ht_scan_res(drv);
2530 }
2531 #endif /* CONFIG_IEEE80211N */
2532
2533
2534 static int i802_init_sockets(struct i802_driver_data *drv, const u8 *bssid)
2535 {
2536         struct ifreq ifr;
2537         struct sockaddr_ll addr;
2538
2539         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
2540         if (drv->ioctl_sock < 0) {
2541                 perror("socket[PF_INET,SOCK_DGRAM]");
2542                 return -1;
2543         }
2544
2545         /* start listening for EAPOL on the default AP interface */
2546         add_ifidx(drv, if_nametoindex(drv->iface));
2547
2548         if (hostapd_set_iface_flags(drv, drv->iface, 0))
2549                 return -1;
2550
2551         if (bssid) {
2552                 os_strlcpy(ifr.ifr_name, drv->iface, IFNAMSIZ);
2553                 memcpy(ifr.ifr_hwaddr.sa_data, bssid, ETH_ALEN);
2554                 ifr.ifr_hwaddr.sa_family = ARPHRD_ETHER;
2555
2556                 if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifr)) {
2557                         perror("ioctl(SIOCSIFHWADDR)");
2558                         return -1;
2559                 }
2560         }
2561
2562         /*
2563          * initialise generic netlink and nl80211
2564          */
2565         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
2566         if (!drv->nl_cb) {
2567                 printf("Failed to allocate netlink callbacks.\n");
2568                 return -1;
2569         }
2570
2571         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
2572         if (!drv->nl_handle) {
2573                 printf("Failed to allocate netlink handle.\n");
2574                 return -1;
2575         }
2576
2577         if (genl_connect(drv->nl_handle)) {
2578                 printf("Failed to connect to generic netlink.\n");
2579                 return -1;
2580         }
2581
2582 #ifdef CONFIG_LIBNL20
2583         if (genl_ctrl_alloc_cache(drv->nl_handle, &drv->nl_cache) < 0) {
2584                 printf("Failed to allocate generic netlink cache.\n");
2585                 return -1;
2586         }
2587 #else /* CONFIG_LIBNL20 */
2588         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
2589         if (!drv->nl_cache) {
2590                 printf("Failed to allocate generic netlink cache.\n");
2591                 return -1;
2592         }
2593 #endif /* CONFIG_LIBNL20 */
2594
2595         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
2596         if (!drv->nl80211) {
2597                 printf("nl80211 not found.\n");
2598                 return -1;
2599         }
2600
2601 #ifdef CONFIG_IEEE80211N
2602         if (drv->ht_40mhz_scan) {
2603                 if (nl80211_set_mode(drv, drv->iface, NL80211_IFTYPE_STATION)
2604                     || hostapd_set_iface_flags(drv, drv->iface, 1) ||
2605                     i802_ht_scan(drv) ||
2606                     hostapd_set_iface_flags(drv, drv->iface, 0)) {
2607                         wpa_printf(MSG_ERROR, "Failed to scan channels for "
2608                                    "HT 40 MHz operations");
2609                         return -1;
2610                 }
2611         }
2612 #endif /* CONFIG_IEEE80211N */
2613
2614         /* Initialise a monitor interface */
2615         if (nl80211_create_monitor_interface(drv))
2616                 return -1;
2617
2618         if (nl80211_set_mode(drv, drv->iface, NL80211_IFTYPE_AP))
2619                 goto fail1;
2620
2621         if (hostapd_set_iface_flags(drv, drv->iface, 1))
2622                 goto fail1;
2623
2624         memset(&addr, 0, sizeof(addr));
2625         addr.sll_family = AF_PACKET;
2626         addr.sll_ifindex = ifr.ifr_ifindex;
2627         wpa_printf(MSG_DEBUG, "Opening raw packet socket for ifindex %d",
2628                    addr.sll_ifindex);
2629
2630         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
2631         if (drv->eapol_sock < 0) {
2632                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
2633                 goto fail1;
2634         }
2635
2636         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
2637         {
2638                 printf("Could not register read socket for eapol\n");
2639                 return -1;
2640         }
2641
2642         memset(&ifr, 0, sizeof(ifr));
2643         os_strlcpy(ifr.ifr_name, drv->iface, sizeof(ifr.ifr_name));
2644         if (ioctl(drv->ioctl_sock, SIOCGIFHWADDR, &ifr) != 0) {
2645                 perror("ioctl(SIOCGIFHWADDR)");
2646                 goto fail1;
2647         }
2648
2649         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
2650                 printf("Invalid HW-addr family 0x%04x\n",
2651                        ifr.ifr_hwaddr.sa_family);
2652                 goto fail1;
2653         }
2654         memcpy(drv->hapd->own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
2655
2656         return 0;
2657
2658 fail1:
2659         nl80211_remove_iface(drv, drv->monitor_ifidx);
2660         return -1;
2661 }
2662
2663
2664 static int i802_get_inact_sec(void *priv, const u8 *addr)
2665 {
2666         struct hostap_sta_driver_data data;
2667         int ret;
2668
2669         data.inactive_msec = (unsigned long) -1;
2670         ret = i802_read_sta_data(priv, &data, addr);
2671         if (ret || data.inactive_msec == (unsigned long) -1)
2672                 return -1;
2673         return data.inactive_msec / 1000;
2674 }
2675
2676
2677 static int i802_sta_clear_stats(void *priv, const u8 *addr)
2678 {
2679 #if 0
2680         /* TODO */
2681 #endif
2682         return 0;
2683 }
2684
2685
2686 static void
2687 hostapd_wireless_event_wireless_custom(struct i802_driver_data *drv,
2688                                        char *custom)
2689 {
2690         wpa_printf(MSG_DEBUG, "Custom wireless event: '%s'", custom);
2691
2692         if (strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
2693                 char *pos;
2694                 u8 addr[ETH_ALEN];
2695                 pos = strstr(custom, "addr=");
2696                 if (pos == NULL) {
2697                         wpa_printf(MSG_DEBUG,
2698                                    "MLME-MICHAELMICFAILURE.indication "
2699                                    "without sender address ignored");
2700                         return;
2701                 }
2702                 pos += 5;
2703                 if (hwaddr_aton(pos, addr) == 0) {
2704                         hostapd_michael_mic_failure(drv->hapd, addr);
2705                 } else {
2706                         wpa_printf(MSG_DEBUG,
2707                                    "MLME-MICHAELMICFAILURE.indication "
2708                                    "with invalid MAC address");
2709                 }
2710         }
2711 }
2712
2713
2714 static void hostapd_wireless_event_wireless(struct i802_driver_data *drv,
2715                                             char *data, int len)
2716 {
2717         struct iw_event iwe_buf, *iwe = &iwe_buf;
2718         char *pos, *end, *custom, *buf;
2719
2720         pos = data;
2721         end = data + len;
2722
2723         while (pos + IW_EV_LCP_LEN <= end) {
2724                 /* Event data may be unaligned, so make a local, aligned copy
2725                  * before processing. */
2726                 memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2727                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
2728                            iwe->cmd, iwe->len);
2729                 if (iwe->len <= IW_EV_LCP_LEN)
2730                         return;
2731
2732                 custom = pos + IW_EV_POINT_LEN;
2733                 if (drv->we_version > 18 &&
2734                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
2735                      iwe->cmd == IWEVCUSTOM)) {
2736                         /* WE-19 removed the pointer from struct iw_point */
2737                         char *dpos = (char *) &iwe_buf.u.data.length;
2738                         int dlen = dpos - (char *) &iwe_buf;
2739                         memcpy(dpos, pos + IW_EV_LCP_LEN,
2740                                sizeof(struct iw_event) - dlen);
2741                 } else {
2742                         memcpy(&iwe_buf, pos, sizeof(struct iw_event));
2743                         custom += IW_EV_POINT_OFF;
2744                 }
2745
2746                 switch (iwe->cmd) {
2747                 case IWEVCUSTOM:
2748                         if (custom + iwe->u.data.length > end)
2749                                 return;
2750                         buf = malloc(iwe->u.data.length + 1);
2751                         if (buf == NULL)
2752                                 return;
2753                         memcpy(buf, custom, iwe->u.data.length);
2754                         buf[iwe->u.data.length] = '\0';
2755                         hostapd_wireless_event_wireless_custom(drv, buf);
2756                         free(buf);
2757                         break;
2758                 }
2759
2760                 pos += iwe->len;
2761         }
2762 }
2763
2764
2765 static void hostapd_wireless_event_rtm_newlink(struct i802_driver_data *drv,
2766                                                struct nlmsghdr *h, int len)
2767 {
2768         struct ifinfomsg *ifi;
2769         int attrlen, _nlmsg_len, rta_len;
2770         struct rtattr *attr;
2771
2772         if (len < (int) sizeof(*ifi))
2773                 return;
2774
2775         ifi = NLMSG_DATA(h);
2776
2777         /* TODO: use ifi->ifi_index to filter out wireless events from other
2778          * interfaces */
2779
2780         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
2781
2782         attrlen = h->nlmsg_len - _nlmsg_len;
2783         if (attrlen < 0)
2784                 return;
2785
2786         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
2787
2788         rta_len = RTA_ALIGN(sizeof(struct rtattr));
2789         while (RTA_OK(attr, attrlen)) {
2790                 if (attr->rta_type == IFLA_WIRELESS) {
2791                         hostapd_wireless_event_wireless(
2792                                 drv, ((char *) attr) + rta_len,
2793                                 attr->rta_len - rta_len);
2794                 }
2795                 attr = RTA_NEXT(attr, attrlen);
2796         }
2797 }
2798
2799
2800 static void hostapd_wireless_event_receive(int sock, void *eloop_ctx,
2801                                            void *sock_ctx)
2802 {
2803         char buf[256];
2804         int left;
2805         struct sockaddr_nl from;
2806         socklen_t fromlen;
2807         struct nlmsghdr *h;
2808         struct i802_driver_data *drv = eloop_ctx;
2809
2810         fromlen = sizeof(from);
2811         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
2812                         (struct sockaddr *) &from, &fromlen);
2813         if (left < 0) {
2814                 if (errno != EINTR && errno != EAGAIN)
2815                         perror("recvfrom(netlink)");
2816                 return;
2817         }
2818
2819         h = (struct nlmsghdr *) buf;
2820         while (left >= (int) sizeof(*h)) {
2821                 int len, plen;
2822
2823                 len = h->nlmsg_len;
2824                 plen = len - sizeof(*h);
2825                 if (len > left || plen < 0) {
2826                         printf("Malformed netlink message: "
2827                                "len=%d left=%d plen=%d\n",
2828                                len, left, plen);
2829                         break;
2830                 }
2831
2832                 switch (h->nlmsg_type) {
2833                 case RTM_NEWLINK:
2834                         hostapd_wireless_event_rtm_newlink(drv, h, plen);
2835                         break;
2836                 }
2837
2838                 len = NLMSG_ALIGN(len);
2839                 left -= len;
2840                 h = (struct nlmsghdr *) ((char *) h + len);
2841         }
2842
2843         if (left > 0) {
2844                 printf("%d extra bytes in the end of netlink message\n", left);
2845         }
2846 }
2847
2848
2849 static int hostap_get_we_version(struct i802_driver_data *drv)
2850 {
2851         struct iw_range *range;
2852         struct iwreq iwr;
2853         int minlen;
2854         size_t buflen;
2855
2856         drv->we_version = 0;
2857
2858         /*
2859          * Use larger buffer than struct iw_range in order to allow the
2860          * structure to grow in the future.
2861          */
2862         buflen = sizeof(struct iw_range) + 500;
2863         range = os_zalloc(buflen);
2864         if (range == NULL)
2865                 return -1;
2866
2867         memset(&iwr, 0, sizeof(iwr));
2868         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2869         iwr.u.data.pointer = (caddr_t) range;
2870         iwr.u.data.length = buflen;
2871
2872         minlen = ((char *) &range->enc_capa) - (char *) range +
2873                 sizeof(range->enc_capa);
2874
2875         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
2876                 perror("ioctl[SIOCGIWRANGE]");
2877                 free(range);
2878                 return -1;
2879         } else if (iwr.u.data.length >= minlen &&
2880                    range->we_version_compiled >= 18) {
2881                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
2882                            "WE(source)=%d enc_capa=0x%x",
2883                            range->we_version_compiled,
2884                            range->we_version_source,
2885                            range->enc_capa);
2886                 drv->we_version = range->we_version_compiled;
2887         }
2888
2889         free(range);
2890         return 0;
2891 }
2892
2893
2894 static int i802_wireless_event_init(struct i802_driver_data *drv)
2895 {
2896         int s;
2897         struct sockaddr_nl local;
2898
2899         hostap_get_we_version(drv);
2900
2901         drv->wext_sock = -1;
2902
2903         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
2904         if (s < 0) {
2905                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
2906                 return -1;
2907         }
2908
2909         memset(&local, 0, sizeof(local));
2910         local.nl_family = AF_NETLINK;
2911         local.nl_groups = RTMGRP_LINK;
2912         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
2913                 perror("bind(netlink)");
2914                 close(s);
2915                 return -1;
2916         }
2917
2918         eloop_register_read_sock(s, hostapd_wireless_event_receive, drv,
2919                                  NULL);
2920         drv->wext_sock = s;
2921
2922         return 0;
2923 }
2924
2925
2926 static void i802_wireless_event_deinit(struct i802_driver_data *drv)
2927 {
2928         if (drv->wext_sock < 0)
2929                 return;
2930         eloop_unregister_read_sock(drv->wext_sock);
2931         close(drv->wext_sock);
2932 }
2933
2934
2935 static int i802_sta_deauth(void *priv, const u8 *addr, int reason)
2936 {
2937         struct i802_driver_data *drv = priv;
2938         struct ieee80211_mgmt mgmt;
2939
2940         memset(&mgmt, 0, sizeof(mgmt));
2941         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2942                                           WLAN_FC_STYPE_DEAUTH);
2943         memcpy(mgmt.da, addr, ETH_ALEN);
2944         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2945         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2946         mgmt.u.deauth.reason_code = host_to_le16(reason);
2947         return i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2948                                       sizeof(mgmt.u.deauth), 0);
2949 }
2950
2951
2952 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason)
2953 {
2954         struct i802_driver_data *drv = priv;
2955         struct ieee80211_mgmt mgmt;
2956
2957         memset(&mgmt, 0, sizeof(mgmt));
2958         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2959                                           WLAN_FC_STYPE_DISASSOC);
2960         memcpy(mgmt.da, addr, ETH_ALEN);
2961         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2962         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2963         mgmt.u.disassoc.reason_code = host_to_le16(reason);
2964         return  i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2965                                        sizeof(mgmt.u.disassoc), 0);
2966 }
2967
2968
2969 static const struct hostapd_neighbor_bss *
2970 i802_get_neighbor_bss(void *priv, size_t *num)
2971 {
2972         struct i802_driver_data *drv = priv;
2973         *num = drv->num_neighbors;
2974         return drv->neighbors;
2975 }
2976
2977
2978 static void *i802_init_bssid(struct hostapd_data *hapd, const u8 *bssid)
2979 {
2980         struct i802_driver_data *drv;
2981
2982         drv = os_zalloc(sizeof(struct i802_driver_data));
2983         if (drv == NULL) {
2984                 printf("Could not allocate memory for i802 driver data\n");
2985                 return NULL;
2986         }
2987
2988         drv->hapd = hapd;
2989         memcpy(drv->iface, hapd->conf->iface, sizeof(drv->iface));
2990         memcpy(drv->bss.iface, hapd->conf->iface, sizeof(drv->iface));
2991
2992         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
2993         drv->if_indices = drv->default_if_indices;
2994         drv->bridge = if_nametoindex(hapd->conf->bridge);
2995         drv->ht_40mhz_scan = hapd->iconf->secondary_channel != 0;
2996
2997         if (i802_init_sockets(drv, bssid))
2998                 goto failed;
2999
3000         if (i802_wireless_event_init(drv))
3001                 goto failed;
3002
3003         return drv;
3004
3005 failed:
3006         free(drv);
3007         return NULL;
3008 }
3009
3010
3011 static void *i802_init(struct hostapd_data *hapd)
3012 {
3013         return i802_init_bssid(hapd, NULL);
3014 }
3015
3016
3017 static void i802_deinit(void *priv)
3018 {
3019         struct i802_driver_data *drv = priv;
3020         struct i802_bss *bss, *prev;
3021
3022         i802_wireless_event_deinit(drv);
3023
3024         if (drv->last_freq_ht) {
3025                 /* Clear HT flags from the driver */
3026                 struct hostapd_freq_params freq;
3027                 os_memset(&freq, 0, sizeof(freq));
3028                 freq.freq = drv->last_freq;
3029                 i802_set_freq(priv, &freq);
3030         }
3031
3032         i802_del_beacon(drv);
3033
3034         /* remove monitor interface */
3035         nl80211_remove_iface(drv, drv->monitor_ifidx);
3036
3037         (void) hostapd_set_iface_flags(drv, drv->iface, 0);
3038
3039         if (drv->monitor_sock >= 0) {
3040                 eloop_unregister_read_sock(drv->monitor_sock);
3041                 close(drv->monitor_sock);
3042         }
3043         if (drv->ioctl_sock >= 0)
3044                 close(drv->ioctl_sock);
3045         if (drv->eapol_sock >= 0) {
3046                 eloop_unregister_read_sock(drv->eapol_sock);
3047                 close(drv->eapol_sock);
3048         }
3049
3050         genl_family_put(drv->nl80211);
3051         nl_cache_free(drv->nl_cache);
3052         nl_handle_destroy(drv->nl_handle);
3053         nl_cb_put(drv->nl_cb);
3054
3055         if (drv->if_indices != drv->default_if_indices)
3056                 free(drv->if_indices);
3057
3058         os_free(drv->neighbors);
3059
3060         bss = drv->bss.next;
3061         while (bss) {
3062                 prev = bss;
3063                 bss = bss->next;
3064                 os_free(bss);
3065         }
3066
3067         free(drv);
3068 }
3069
3070
3071 const struct hapd_driver_ops wpa_driver_nl80211_ops = {
3072         .name = "nl80211",
3073         .init = i802_init,
3074         .init_bssid = i802_init_bssid,
3075         .deinit = i802_deinit,
3076         .set_ieee8021x = i802_set_ieee8021x,
3077         .set_privacy = i802_set_privacy,
3078         .set_key = i802_set_key,
3079         .get_seqnum = i802_get_seqnum,
3080         .flush = i802_flush,
3081         .read_sta_data = i802_read_sta_data,
3082         .send_eapol = i802_send_eapol,
3083         .sta_set_flags = i802_sta_set_flags,
3084         .sta_deauth = i802_sta_deauth,
3085         .sta_disassoc = i802_sta_disassoc,
3086         .sta_remove = i802_sta_remove,
3087         .send_mgmt_frame = i802_send_mgmt_frame,
3088         .sta_add = i802_sta_add,
3089         .get_inact_sec = i802_get_inact_sec,
3090         .sta_clear_stats = i802_sta_clear_stats,
3091         .set_freq = i802_set_freq,
3092         .set_rts = i802_set_rts,
3093         .get_rts = i802_get_rts,
3094         .set_frag = i802_set_frag,
3095         .get_frag = i802_get_frag,
3096         .set_retry = i802_set_retry,
3097         .get_retry = i802_get_retry,
3098         .set_rate_sets = i802_set_rate_sets,
3099         .set_beacon = i802_set_beacon,
3100         .set_internal_bridge = i802_set_internal_bridge,
3101         .set_beacon_int = i802_set_beacon_int,
3102         .set_cts_protect = i802_set_cts_protect,
3103         .set_preamble = i802_set_preamble,
3104         .set_short_slot_time = i802_set_short_slot_time,
3105         .set_tx_queue_params = i802_set_tx_queue_params,
3106         .bss_add = i802_bss_add,
3107         .bss_remove = i802_bss_remove,
3108         .if_add = i802_if_add,
3109         .if_update = i802_if_update,
3110         .if_remove = i802_if_remove,
3111         .get_hw_feature_data = i802_get_hw_feature_data,
3112         .set_sta_vlan = i802_set_sta_vlan,
3113         .set_country = i802_set_country,
3114         .get_neighbor_bss = i802_get_neighbor_bss,
3115 };