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