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