nl80211: Share monitor mode filtering and reading functions
[wpasupplicant] / src / drivers / driver_nl80211.c
1 /*
2  * Driver interaction with Linux nl80211/cfg80211
3  * Copyright (c) 2002-2008, 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  * Copyright (c) 2009, Atheros Communications
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License version 2 as
11  * published by the Free Software Foundation.
12  *
13  * Alternatively, this software may be distributed under the terms of BSD
14  * license.
15  *
16  * See README and COPYING for more details.
17  */
18
19 #include "includes.h"
20 #include <sys/ioctl.h>
21 #include <net/if_arp.h>
22 #include <netlink/genl/genl.h>
23 #include <netlink/genl/family.h>
24 #include <netlink/genl/ctrl.h>
25 #include "nl80211_copy.h"
26 #include "wireless_copy.h"
27
28 #include "common.h"
29 #include "driver.h"
30 #include "eloop.h"
31 #include "ieee802_11_defs.h"
32
33 #if defined(CONFIG_AP) || defined(HOSTAPD)
34 #include <netpacket/packet.h>
35 #include <linux/filter.h>
36 #include "radiotap.h"
37 #include "radiotap_iter.h"
38 #endif /* CONFIG_AP || HOSTAPD */
39
40 #ifdef CONFIG_AP
41
42 #include "../hostapd/hostapd_defs.h"
43
44 #ifndef ETH_P_ALL
45 #define ETH_P_ALL 0x0003
46 #endif
47
48 #endif /* CONFIG_AP */
49
50 #ifdef HOSTAPD
51 #include <netlink/msg.h>
52 #include <netlink/attr.h>
53 #include <net/if.h>
54 #include <net/if_arp.h>
55
56 #include "../../hostapd/hostapd.h"
57 #include "../../hostapd/config.h"
58 #include "../../hostapd/hw_features.h"
59 #include "../../hostapd/mlme.h"
60 #include "../../hostapd/sta_flags.h"
61 #include "ieee802_11_common.h"
62
63 #ifdef CONFIG_LIBNL20
64 /* libnl 2.0 compatibility code */
65 #define nl_handle_alloc_cb nl_socket_alloc_cb
66 #define nl_handle_destroy nl_socket_free
67 #endif /* CONFIG_LIBNL20 */
68
69 #endif /* HOSTAPD */
70
71
72 #ifndef IFF_LOWER_UP
73 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
74 #endif
75 #ifndef IFF_DORMANT
76 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
77 #endif
78
79 #ifndef IF_OPER_DORMANT
80 #define IF_OPER_DORMANT 5
81 #endif
82 #ifndef IF_OPER_UP
83 #define IF_OPER_UP 6
84 #endif
85
86 enum ieee80211_msg_type {
87         ieee80211_msg_normal = 0,
88         ieee80211_msg_tx_callback_ack = 1,
89         ieee80211_msg_tx_callback_fail = 2,
90 };
91
92 struct i802_bss {
93         struct i802_bss *next;
94         char ifname[IFNAMSIZ + 1];
95         unsigned int beacon_set:1;
96 };
97
98 struct wpa_driver_nl80211_data {
99         void *ctx;
100         int link_event_sock;
101         int ioctl_sock; /* socket for ioctl() use */
102         char ifname[IFNAMSIZ + 1];
103         int ifindex;
104         int if_removed;
105         struct wpa_driver_capa capa;
106         int has_capability;
107
108         int operstate;
109
110         int scan_complete_events;
111
112         struct nl_handle *nl_handle;
113         struct nl_cache *nl_cache;
114         struct nl_cb *nl_cb;
115         struct genl_family *nl80211;
116
117         u8 bssid[ETH_ALEN];
118         int associated;
119         u8 ssid[32];
120         size_t ssid_len;
121
122 #ifdef CONFIG_AP
123         int beacon_int;
124         unsigned int beacon_set:1;
125         int monitor_sock;
126         int monitor_ifidx;
127 #endif /* CONFIG_AP */
128
129 #ifdef HOSTAPD
130         struct hostapd_data *hapd;
131
132         int eapol_sock; /* socket for EAPOL frames */
133         int monitor_sock; /* socket for monitor */
134         int monitor_ifidx;
135
136         int default_if_indices[16];
137         int *if_indices;
138         int num_if_indices;
139
140         int beacon_int;
141         struct i802_bss bss;
142         unsigned int ht_40mhz_scan:1;
143
144         int last_freq;
145         int last_freq_ht;
146         struct hostapd_neighbor_bss *neighbors;
147         size_t num_neighbors;
148 #endif /* HOSTAPD */
149 };
150
151
152 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
153                                             void *timeout_ctx);
154 static int wpa_driver_nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
155                                        int mode);
156 static int
157 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
158
159 #ifdef CONFIG_AP
160 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
161                                  int ifidx);
162 #endif /* CONFIG_AP */
163
164 #ifdef HOSTAPD
165 static void handle_frame(struct wpa_driver_nl80211_data *drv,
166                          u8 *buf, size_t len,
167                          struct hostapd_frame_info *hfi,
168                          enum ieee80211_msg_type msg_type);
169 #endif /* HOSTAPD */
170
171
172 /* nl80211 code */
173 static int ack_handler(struct nl_msg *msg, void *arg)
174 {
175         int *err = arg;
176         *err = 0;
177         return NL_STOP;
178 }
179
180 static int finish_handler(struct nl_msg *msg, void *arg)
181 {
182         int *ret = arg;
183         *ret = 0;
184         return NL_SKIP;
185 }
186
187 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
188                          void *arg)
189 {
190         int *ret = arg;
191         *ret = err->error;
192         return NL_SKIP;
193 }
194
195
196 static int no_seq_check(struct nl_msg *msg, void *arg)
197 {
198         return NL_OK;
199 }
200
201
202 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
203                               struct nl_msg *msg,
204                               int (*valid_handler)(struct nl_msg *, void *),
205                               void *valid_data)
206 {
207         struct nl_cb *cb;
208         int err = -ENOMEM;
209
210         cb = nl_cb_clone(drv->nl_cb);
211         if (!cb)
212                 goto out;
213
214         err = nl_send_auto_complete(drv->nl_handle, msg);
215         if (err < 0)
216                 goto out;
217
218         err = 1;
219
220         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
221         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
222         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
223
224         if (valid_handler)
225                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
226                           valid_handler, valid_data);
227
228         while (err > 0)
229                 nl_recvmsgs(drv->nl_handle, cb);
230  out:
231         nl_cb_put(cb);
232         nlmsg_free(msg);
233         return err;
234 }
235
236
237 struct family_data {
238         const char *group;
239         int id;
240 };
241
242
243 static int family_handler(struct nl_msg *msg, void *arg)
244 {
245         struct family_data *res = arg;
246         struct nlattr *tb[CTRL_ATTR_MAX + 1];
247         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
248         struct nlattr *mcgrp;
249         int i;
250
251         nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
252                   genlmsg_attrlen(gnlh, 0), NULL);
253         if (!tb[CTRL_ATTR_MCAST_GROUPS])
254                 return NL_SKIP;
255
256         nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], i) {
257                 struct nlattr *tb2[CTRL_ATTR_MCAST_GRP_MAX + 1];
258                 nla_parse(tb2, CTRL_ATTR_MCAST_GRP_MAX, nla_data(mcgrp),
259                           nla_len(mcgrp), NULL);
260                 if (!tb2[CTRL_ATTR_MCAST_GRP_NAME] ||
261                     !tb2[CTRL_ATTR_MCAST_GRP_ID] ||
262                     os_strncmp(nla_data(tb2[CTRL_ATTR_MCAST_GRP_NAME]),
263                                res->group,
264                                nla_len(tb2[CTRL_ATTR_MCAST_GRP_NAME])) != 0)
265                         continue;
266                 res->id = nla_get_u32(tb2[CTRL_ATTR_MCAST_GRP_ID]);
267                 break;
268         };
269
270         return NL_SKIP;
271 }
272
273
274 static int nl_get_multicast_id(struct wpa_driver_nl80211_data *drv,
275                                const char *family, const char *group)
276 {
277         struct nl_msg *msg;
278         int ret = -1;
279         struct family_data res = { group, -ENOENT };
280
281         msg = nlmsg_alloc();
282         if (!msg)
283                 return -ENOMEM;
284         genlmsg_put(msg, 0, 0, genl_ctrl_resolve(drv->nl_handle, "nlctrl"),
285                     0, 0, CTRL_CMD_GETFAMILY, 0);
286         NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
287
288         ret = send_and_recv_msgs(drv, msg, family_handler, &res);
289         msg = NULL;
290         if (ret == 0)
291                 ret = res.id;
292
293 nla_put_failure:
294         nlmsg_free(msg);
295         return ret;
296 }
297
298
299 static int wpa_driver_nl80211_send_oper_ifla(
300         struct wpa_driver_nl80211_data *drv,
301         int linkmode, int operstate)
302 {
303         struct {
304                 struct nlmsghdr hdr;
305                 struct ifinfomsg ifinfo;
306                 char opts[16];
307         } req;
308         struct rtattr *rta;
309         static int nl_seq;
310         ssize_t ret;
311
312         os_memset(&req, 0, sizeof(req));
313
314         req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
315         req.hdr.nlmsg_type = RTM_SETLINK;
316         req.hdr.nlmsg_flags = NLM_F_REQUEST;
317         req.hdr.nlmsg_seq = ++nl_seq;
318         req.hdr.nlmsg_pid = 0;
319
320         req.ifinfo.ifi_family = AF_UNSPEC;
321         req.ifinfo.ifi_type = 0;
322         req.ifinfo.ifi_index = drv->ifindex;
323         req.ifinfo.ifi_flags = 0;
324         req.ifinfo.ifi_change = 0;
325
326         if (linkmode != -1) {
327                 rta = (struct rtattr *)
328                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
329                 rta->rta_type = IFLA_LINKMODE;
330                 rta->rta_len = RTA_LENGTH(sizeof(char));
331                 *((char *) RTA_DATA(rta)) = linkmode;
332                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
333                         RTA_LENGTH(sizeof(char));
334         }
335         if (operstate != -1) {
336                 rta = (struct rtattr *)
337                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
338                 rta->rta_type = IFLA_OPERSTATE;
339                 rta->rta_len = RTA_LENGTH(sizeof(char));
340                 *((char *) RTA_DATA(rta)) = operstate;
341                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
342                         RTA_LENGTH(sizeof(char));
343         }
344
345         wpa_printf(MSG_DEBUG, "WEXT: Operstate: linkmode=%d, operstate=%d",
346                    linkmode, operstate);
347
348         ret = send(drv->link_event_sock, &req, req.hdr.nlmsg_len, 0);
349         if (ret < 0) {
350                 wpa_printf(MSG_DEBUG, "WEXT: Sending operstate IFLA failed: "
351                            "%s (assume operstate is not supported)",
352                            strerror(errno));
353         }
354
355         return ret < 0 ? -1 : 0;
356 }
357
358
359 static int wpa_driver_nl80211_set_auth_param(
360         struct wpa_driver_nl80211_data *drv, int idx, u32 value)
361 {
362         struct iwreq iwr;
363         int ret = 0;
364
365         os_memset(&iwr, 0, sizeof(iwr));
366         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
367         iwr.u.param.flags = idx & IW_AUTH_INDEX;
368         iwr.u.param.value = value;
369
370         if (ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr) < 0) {
371                 if (errno != EOPNOTSUPP) {
372                         wpa_printf(MSG_DEBUG, "WEXT: SIOCSIWAUTH(param %d "
373                                    "value 0x%x) failed: %s)",
374                                    idx, value, strerror(errno));
375                 }
376                 ret = errno == EOPNOTSUPP ? -2 : -1;
377         }
378
379         return ret;
380 }
381
382
383 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
384 {
385         struct wpa_driver_nl80211_data *drv = priv;
386         if (!drv->associated)
387                 return -1;
388         os_memcpy(bssid, drv->bssid, ETH_ALEN);
389         return 0;
390 }
391
392
393 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
394 {
395         struct wpa_driver_nl80211_data *drv = priv;
396         if (!drv->associated)
397                 return -1;
398         os_memcpy(ssid, drv->ssid, drv->ssid_len);
399         return drv->ssid_len;
400 }
401
402
403 #ifndef HOSTAPD
404 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
405                                           void *ctx, char *buf, size_t len,
406                                           int del)
407 {
408         union wpa_event_data event;
409
410         os_memset(&event, 0, sizeof(event));
411         if (len > sizeof(event.interface_status.ifname))
412                 len = sizeof(event.interface_status.ifname) - 1;
413         os_memcpy(event.interface_status.ifname, buf, len);
414         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
415                 EVENT_INTERFACE_ADDED;
416
417         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
418                    del ? "DEL" : "NEW",
419                    event.interface_status.ifname,
420                    del ? "removed" : "added");
421
422         if (os_strcmp(drv->ifname, event.interface_status.ifname) == 0) {
423                 if (del)
424                         drv->if_removed = 1;
425                 else
426                         drv->if_removed = 0;
427         }
428
429         wpa_supplicant_event(ctx, EVENT_INTERFACE_STATUS, &event);
430 }
431
432
433 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
434                                          struct nlmsghdr *h)
435 {
436         struct ifinfomsg *ifi;
437         int attrlen, _nlmsg_len, rta_len;
438         struct rtattr *attr;
439
440         ifi = NLMSG_DATA(h);
441
442         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
443
444         attrlen = h->nlmsg_len - _nlmsg_len;
445         if (attrlen < 0)
446                 return 0;
447
448         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
449
450         rta_len = RTA_ALIGN(sizeof(struct rtattr));
451         while (RTA_OK(attr, attrlen)) {
452                 if (attr->rta_type == IFLA_IFNAME) {
453                         if (os_strcmp(((char *) attr) + rta_len, drv->ifname)
454                             == 0)
455                                 return 1;
456                         else
457                                 break;
458                 }
459                 attr = RTA_NEXT(attr, attrlen);
460         }
461
462         return 0;
463 }
464
465
466 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
467                                           int ifindex, struct nlmsghdr *h)
468 {
469         if (drv->ifindex == ifindex)
470                 return 1;
471
472         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, h)) {
473                 drv->ifindex = if_nametoindex(drv->ifname);
474                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
475                            "interface");
476                 wpa_driver_nl80211_finish_drv_init(drv);
477                 return 1;
478         }
479
480         return 0;
481 }
482
483
484 static void wpa_driver_nl80211_event_rtm_newlink(struct wpa_driver_nl80211_data *drv,
485                                               void *ctx, struct nlmsghdr *h,
486                                               size_t len)
487 {
488         struct ifinfomsg *ifi;
489         int attrlen, _nlmsg_len, rta_len;
490         struct rtattr * attr;
491
492         if (len < sizeof(*ifi))
493                 return;
494
495         ifi = NLMSG_DATA(h);
496
497         if (!wpa_driver_nl80211_own_ifindex(drv, ifi->ifi_index, h)) {
498                 wpa_printf(MSG_DEBUG, "Ignore event for foreign ifindex %d",
499                            ifi->ifi_index);
500                 return;
501         }
502
503         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
504                    "(%s%s%s%s)",
505                    drv->operstate, ifi->ifi_flags,
506                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
507                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
508                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
509                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
510         /*
511          * Some drivers send the association event before the operup event--in
512          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
513          * fails. This will hit us when wpa_supplicant does not need to do
514          * IEEE 802.1X authentication
515          */
516         if (drv->operstate == 1 &&
517             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
518             !(ifi->ifi_flags & IFF_RUNNING))
519                 wpa_driver_nl80211_send_oper_ifla(drv, -1, IF_OPER_UP);
520
521         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
522
523         attrlen = h->nlmsg_len - _nlmsg_len;
524         if (attrlen < 0)
525                 return;
526
527         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
528
529         rta_len = RTA_ALIGN(sizeof(struct rtattr));
530         while (RTA_OK(attr, attrlen)) {
531                 if (attr->rta_type == IFLA_IFNAME) {
532                         wpa_driver_nl80211_event_link(
533                                 drv, ctx,
534                                 ((char *) attr) + rta_len,
535                                 attr->rta_len - rta_len, 0);
536                 }
537                 attr = RTA_NEXT(attr, attrlen);
538         }
539 }
540
541
542 static void wpa_driver_nl80211_event_rtm_dellink(struct wpa_driver_nl80211_data *drv,
543                                               void *ctx, struct nlmsghdr *h,
544                                               size_t len)
545 {
546         struct ifinfomsg *ifi;
547         int attrlen, _nlmsg_len, rta_len;
548         struct rtattr * attr;
549
550         if (len < sizeof(*ifi))
551                 return;
552
553         ifi = NLMSG_DATA(h);
554
555         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
556
557         attrlen = h->nlmsg_len - _nlmsg_len;
558         if (attrlen < 0)
559                 return;
560
561         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
562
563         rta_len = RTA_ALIGN(sizeof(struct rtattr));
564         while (RTA_OK(attr, attrlen)) {
565                 if (attr->rta_type == IFLA_IFNAME) {
566                         wpa_driver_nl80211_event_link(
567                                 drv, ctx,
568                                 ((char *) attr) + rta_len,
569                                 attr->rta_len - rta_len, 1);
570                 }
571                 attr = RTA_NEXT(attr, attrlen);
572         }
573 }
574
575
576 static void wpa_driver_nl80211_event_receive_link(int sock, void *eloop_ctx,
577                                                   void *sock_ctx)
578 {
579         char buf[8192];
580         int left;
581         struct sockaddr_nl from;
582         socklen_t fromlen;
583         struct nlmsghdr *h;
584         int max_events = 10;
585
586 try_again:
587         fromlen = sizeof(from);
588         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
589                         (struct sockaddr *) &from, &fromlen);
590         if (left < 0) {
591                 if (errno != EINTR && errno != EAGAIN)
592                         perror("recvfrom(netlink)");
593                 return;
594         }
595
596         h = (struct nlmsghdr *) buf;
597         while (left >= (int) sizeof(*h)) {
598                 int len, plen;
599
600                 len = h->nlmsg_len;
601                 plen = len - sizeof(*h);
602                 if (len > left || plen < 0) {
603                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
604                                    "len=%d left=%d plen=%d",
605                                    len, left, plen);
606                         break;
607                 }
608
609                 switch (h->nlmsg_type) {
610                 case RTM_NEWLINK:
611                         wpa_driver_nl80211_event_rtm_newlink(eloop_ctx, sock_ctx,
612                                                           h, plen);
613                         break;
614                 case RTM_DELLINK:
615                         wpa_driver_nl80211_event_rtm_dellink(eloop_ctx, sock_ctx,
616                                                           h, plen);
617                         break;
618                 }
619
620                 len = NLMSG_ALIGN(len);
621                 left -= len;
622                 h = (struct nlmsghdr *) ((char *) h + len);
623         }
624
625         if (left > 0) {
626                 wpa_printf(MSG_DEBUG, "%d extra bytes in the end of netlink "
627                            "message", left);
628         }
629
630         if (--max_events > 0) {
631                 /*
632                  * Try to receive all events in one eloop call in order to
633                  * limit race condition on cases where AssocInfo event, Assoc
634                  * event, and EAPOL frames are received more or less at the
635                  * same time. We want to process the event messages first
636                  * before starting EAPOL processing.
637                  */
638                 goto try_again;
639         }
640 }
641
642
643 static void mlme_event_auth(struct wpa_driver_nl80211_data *drv,
644                             const u8 *frame, size_t len)
645 {
646         const struct ieee80211_mgmt *mgmt;
647         union wpa_event_data event;
648
649         mgmt = (const struct ieee80211_mgmt *) frame;
650         if (len < 24 + sizeof(mgmt->u.auth)) {
651                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
652                            "frame");
653                 return;
654         }
655
656         os_memset(&event, 0, sizeof(event));
657         os_memcpy(event.auth.peer, mgmt->sa, ETH_ALEN);
658         event.auth.auth_type = le_to_host16(mgmt->u.auth.auth_alg);
659         event.auth.status_code = le_to_host16(mgmt->u.auth.status_code);
660         if (len > 24 + sizeof(mgmt->u.auth)) {
661                 event.auth.ies = mgmt->u.auth.variable;
662                 event.auth.ies_len = len - 24 - sizeof(mgmt->u.auth);
663         }
664
665         wpa_supplicant_event(drv->ctx, EVENT_AUTH, &event);
666 }
667
668
669 static void mlme_event_assoc(struct wpa_driver_nl80211_data *drv,
670                             const u8 *frame, size_t len)
671 {
672         const struct ieee80211_mgmt *mgmt;
673         union wpa_event_data event;
674         u16 status;
675
676         mgmt = (const struct ieee80211_mgmt *) frame;
677         if (len < 24 + sizeof(mgmt->u.assoc_resp)) {
678                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
679                            "frame");
680                 return;
681         }
682
683         status = le_to_host16(mgmt->u.assoc_resp.status_code);
684         if (status != WLAN_STATUS_SUCCESS) {
685                 os_memset(&event, 0, sizeof(event));
686                 if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
687                         event.assoc_reject.resp_ies =
688                                 (u8 *) mgmt->u.assoc_resp.variable;
689                         event.assoc_reject.resp_ies_len =
690                                 len - 24 - sizeof(mgmt->u.assoc_resp);
691                 }
692                 event.assoc_reject.status_code = status;
693
694                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
695                 return;
696         }
697
698         drv->associated = 1;
699         os_memcpy(drv->bssid, mgmt->sa, ETH_ALEN);
700
701         os_memset(&event, 0, sizeof(event));
702         if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
703                 event.assoc_info.resp_ies = (u8 *) mgmt->u.assoc_resp.variable;
704                 event.assoc_info.resp_ies_len =
705                         len - 24 - sizeof(mgmt->u.assoc_resp);
706         }
707
708         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
709 }
710
711
712 static void mlme_event(struct wpa_driver_nl80211_data *drv,
713                        enum nl80211_commands cmd, struct nlattr *frame)
714 {
715         if (frame == NULL) {
716                 wpa_printf(MSG_DEBUG, "nl80211: MLME event %d without frame "
717                            "data", cmd);
718                 return;
719         }
720
721         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d", cmd);
722         wpa_hexdump(MSG_MSGDUMP, "nl80211: MLME event frame",
723                     nla_data(frame), nla_len(frame));
724
725         switch (cmd) {
726         case NL80211_CMD_AUTHENTICATE:
727                 mlme_event_auth(drv, nla_data(frame), nla_len(frame));
728                 break;
729         case NL80211_CMD_ASSOCIATE:
730                 mlme_event_assoc(drv, nla_data(frame), nla_len(frame));
731                 break;
732         case NL80211_CMD_DEAUTHENTICATE:
733                 drv->associated = 0;
734                 wpa_supplicant_event(drv->ctx, EVENT_DEAUTH, NULL);
735                 break;
736         case NL80211_CMD_DISASSOCIATE:
737                 drv->associated = 0;
738                 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
739                 break;
740         default:
741                 break;
742         }
743 }
744
745 #endif /* HOSTAPD */
746
747
748 static void mlme_event_michael_mic_failure(struct wpa_driver_nl80211_data *drv,
749                                            struct nlattr *tb[])
750 {
751 #ifdef HOSTAPD
752         if (tb[NL80211_ATTR_MAC])
753                 hostapd_michael_mic_failure(drv->hapd,
754                                             nla_data(tb[NL80211_ATTR_MAC]));
755 #else /* HOSTAPD */
756         union wpa_event_data data;
757
758         wpa_printf(MSG_DEBUG, "nl80211: MLME event Michael MIC failure");
759         os_memset(&data, 0, sizeof(data));
760         if (tb[NL80211_ATTR_MAC]) {
761                 wpa_hexdump(MSG_DEBUG, "nl80211: Source MAC address",
762                             nla_data(tb[NL80211_ATTR_MAC]),
763                             nla_len(tb[NL80211_ATTR_MAC]));
764         }
765         if (tb[NL80211_ATTR_KEY_SEQ]) {
766                 wpa_hexdump(MSG_DEBUG, "nl80211: TSC",
767                             nla_data(tb[NL80211_ATTR_KEY_SEQ]),
768                             nla_len(tb[NL80211_ATTR_KEY_SEQ]));
769         }
770         if (tb[NL80211_ATTR_KEY_TYPE]) {
771                 enum nl80211_key_type key_type =
772                         nla_get_u32(tb[NL80211_ATTR_KEY_TYPE]);
773                 wpa_printf(MSG_DEBUG, "nl80211: Key Type %d", key_type);
774                 if (key_type == NL80211_KEYTYPE_PAIRWISE)
775                         data.michael_mic_failure.unicast = 1;
776         } else
777                 data.michael_mic_failure.unicast = 1;
778
779         if (tb[NL80211_ATTR_KEY_IDX]) {
780                 u8 key_id = nla_get_u8(tb[NL80211_ATTR_KEY_IDX]);
781                 wpa_printf(MSG_DEBUG, "nl80211: Key Id %d", key_id);
782         }
783
784         wpa_supplicant_event(drv->ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
785 #endif /* HOSTAPD */
786 }
787
788
789 static int process_event(struct nl_msg *msg, void *arg)
790 {
791         struct wpa_driver_nl80211_data *drv = arg;
792         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
793         struct nlattr *tb[NL80211_ATTR_MAX + 1];
794
795         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
796                   genlmsg_attrlen(gnlh, 0), NULL);
797
798         if (tb[NL80211_ATTR_IFINDEX]) {
799                 int ifindex = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
800                 if (ifindex != drv->ifindex) {
801                         wpa_printf(MSG_DEBUG, "nl80211: Ignored event (cmd=%d)"
802                                    " for foreign interface (ifindex %d)",
803                                    gnlh->cmd, ifindex);
804                         return NL_SKIP;
805                 }
806         }
807
808         switch (gnlh->cmd) {
809 #ifndef HOSTAPD
810         case NL80211_CMD_NEW_SCAN_RESULTS:
811                 wpa_printf(MSG_DEBUG, "nl80211: New scan results available");
812                 drv->scan_complete_events = 1;
813                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
814                                      drv->ctx);
815                 wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, NULL);
816                 break;
817         case NL80211_CMD_SCAN_ABORTED:
818                 wpa_printf(MSG_DEBUG, "nl80211: Scan aborted");
819                 /*
820                  * Need to indicate that scan results are available in order
821                  * not to make wpa_supplicant stop its scanning.
822                  */
823                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
824                                      drv->ctx);
825                 wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, NULL);
826                 break;
827         case NL80211_CMD_AUTHENTICATE:
828         case NL80211_CMD_ASSOCIATE:
829         case NL80211_CMD_DEAUTHENTICATE:
830         case NL80211_CMD_DISASSOCIATE:
831                 mlme_event(drv, gnlh->cmd, tb[NL80211_ATTR_FRAME]);
832                 break;
833 #endif /* HOSTAPD */
834         case NL80211_CMD_MICHAEL_MIC_FAILURE:
835                 mlme_event_michael_mic_failure(drv, tb);
836                 break;
837         default:
838                 wpa_printf(MSG_DEBUG, "nl80211: Ignored unknown event "
839                            "(cmd=%d)", gnlh->cmd);
840                 break;
841         }
842
843         return NL_SKIP;
844 }
845
846
847 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
848                                              void *sock_ctx)
849 {
850         struct nl_cb *cb;
851         struct wpa_driver_nl80211_data *drv = eloop_ctx;
852
853         wpa_printf(MSG_DEBUG, "nl80211: Event message available");
854
855         cb = nl_cb_clone(drv->nl_cb);
856         if (!cb)
857                 return;
858         nl_cb_set(cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, no_seq_check, NULL);
859         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, process_event, drv);
860         nl_recvmsgs(drv->nl_handle, cb);
861         nl_cb_put(cb);
862 }
863
864
865 static int wpa_driver_nl80211_get_ifflags_ifname(struct wpa_driver_nl80211_data *drv,
866                                               const char *ifname, int *flags)
867 {
868         struct ifreq ifr;
869
870         os_memset(&ifr, 0, sizeof(ifr));
871         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
872         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, (caddr_t) &ifr) < 0) {
873                 perror("ioctl[SIOCGIFFLAGS]");
874                 return -1;
875         }
876         *flags = ifr.ifr_flags & 0xffff;
877         return 0;
878 }
879
880
881 /**
882  * wpa_driver_nl80211_get_ifflags - Get interface flags (SIOCGIFFLAGS)
883  * @drv: driver_nl80211 private data
884  * @flags: Pointer to returned flags value
885  * Returns: 0 on success, -1 on failure
886  */
887 static int wpa_driver_nl80211_get_ifflags(struct wpa_driver_nl80211_data *drv,
888                                           int *flags)
889 {
890         return wpa_driver_nl80211_get_ifflags_ifname(drv, drv->ifname, flags);
891 }
892
893
894 static int wpa_driver_nl80211_set_ifflags_ifname(
895         struct wpa_driver_nl80211_data *drv,
896         const char *ifname, int flags)
897 {
898         struct ifreq ifr;
899
900         os_memset(&ifr, 0, sizeof(ifr));
901         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
902         ifr.ifr_flags = flags & 0xffff;
903         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, (caddr_t) &ifr) < 0) {
904                 perror("SIOCSIFFLAGS");
905                 return -1;
906         }
907         return 0;
908 }
909
910
911 /**
912  * wpa_driver_nl80211_set_ifflags - Set interface flags (SIOCSIFFLAGS)
913  * @drv: driver_nl80211 private data
914  * @flags: New value for flags
915  * Returns: 0 on success, -1 on failure
916  */
917 static int wpa_driver_nl80211_set_ifflags(struct wpa_driver_nl80211_data *drv,
918                                           int flags)
919 {
920         return wpa_driver_nl80211_set_ifflags_ifname(drv, drv->ifname, flags);
921 }
922
923
924 /**
925  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
926  * @priv: driver_nl80211 private data
927  * @alpha2_arg: country to which to switch to
928  * Returns: 0 on success, -1 on failure
929  *
930  * This asks nl80211 to set the regulatory domain for given
931  * country ISO / IEC alpha2.
932  */
933 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
934 {
935         struct wpa_driver_nl80211_data *drv = priv;
936         char alpha2[3];
937         struct nl_msg *msg;
938
939         msg = nlmsg_alloc();
940         if (!msg)
941                 return -ENOMEM;
942
943         alpha2[0] = alpha2_arg[0];
944         alpha2[1] = alpha2_arg[1];
945         alpha2[2] = '\0';
946
947         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
948                     0, NL80211_CMD_REQ_SET_REG, 0);
949
950         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
951         if (send_and_recv_msgs(drv, msg, NULL, NULL))
952                 return -EINVAL;
953         return 0;
954 nla_put_failure:
955         return -EINVAL;
956 }
957
958
959 struct wiphy_info_data {
960         int max_scan_ssids;
961         int ap_supported;
962 };
963
964
965 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
966 {
967         struct nlattr *tb[NL80211_ATTR_MAX + 1];
968         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
969         struct wiphy_info_data *info = arg;
970
971         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
972                   genlmsg_attrlen(gnlh, 0), NULL);
973
974         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
975                 info->max_scan_ssids =
976                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
977
978         if (tb[NL80211_ATTR_SUPPORTED_IFTYPES]) {
979                 struct nlattr *nl_mode;
980                 int i;
981                 nla_for_each_nested(nl_mode,
982                                     tb[NL80211_ATTR_SUPPORTED_IFTYPES], i) {
983                         if (nl_mode->nla_type == NL80211_IFTYPE_AP) {
984                                 info->ap_supported = 1;
985                                 break;
986                         }
987                 }
988         }
989
990         return NL_SKIP;
991 }
992
993
994 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
995                                        struct wiphy_info_data *info)
996 {
997         struct nl_msg *msg;
998
999         os_memset(info, 0, sizeof(*info));
1000         msg = nlmsg_alloc();
1001         if (!msg)
1002                 return -1;
1003
1004         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1005                     0, NL80211_CMD_GET_WIPHY, 0);
1006
1007         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1008
1009         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info) == 0)
1010                 return 0;
1011         msg = NULL;
1012 nla_put_failure:
1013         nlmsg_free(msg);
1014         return -1;
1015 }
1016
1017
1018 static void wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
1019 {
1020         struct wiphy_info_data info;
1021         if (wpa_driver_nl80211_get_info(drv, &info))
1022                 return;
1023         drv->has_capability = 1;
1024         /* For now, assume TKIP, CCMP, WPA, WPA2 are supported */
1025         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1026                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
1027                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1028                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
1029         drv->capa.enc = WPA_DRIVER_CAPA_ENC_WEP40 |
1030                 WPA_DRIVER_CAPA_ENC_WEP104 |
1031                 WPA_DRIVER_CAPA_ENC_TKIP |
1032                 WPA_DRIVER_CAPA_ENC_CCMP;
1033
1034         drv->capa.max_scan_ssids = info.max_scan_ssids;
1035         if (info.ap_supported)
1036                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP;
1037 }
1038
1039
1040 /**
1041  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
1042  * @ctx: context to be used when calling wpa_supplicant functions,
1043  * e.g., wpa_supplicant_event()
1044  * @ifname: interface name, e.g., wlan0
1045  * Returns: Pointer to private data, %NULL on failure
1046  */
1047 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname)
1048 {
1049         int s, ret;
1050         struct sockaddr_nl local;
1051         struct wpa_driver_nl80211_data *drv;
1052
1053         drv = os_zalloc(sizeof(*drv));
1054         if (drv == NULL)
1055                 return NULL;
1056         drv->ctx = ctx;
1057         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
1058 #ifdef CONFIG_AP
1059         drv->monitor_ifidx = -1;
1060         drv->monitor_sock = -1;
1061 #endif /* CONFIG_AP */
1062
1063         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
1064         if (drv->nl_cb == NULL) {
1065                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1066                            "callbacks");
1067                 goto err1;
1068         }
1069
1070         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
1071         if (drv->nl_handle == NULL) {
1072                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1073                            "callbacks");
1074                 goto err2;
1075         }
1076
1077         if (genl_connect(drv->nl_handle)) {
1078                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
1079                            "netlink");
1080                 goto err3;
1081         }
1082
1083         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
1084         if (drv->nl_cache == NULL) {
1085                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1086                            "netlink cache");
1087                 goto err3;
1088         }
1089
1090         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
1091         if (drv->nl80211 == NULL) {
1092                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
1093                            "found");
1094                 goto err4;
1095         }
1096
1097         ret = nl_get_multicast_id(drv, "nl80211", "scan");
1098         if (ret >= 0)
1099                 ret = nl_socket_add_membership(drv->nl_handle, ret);
1100         if (ret < 0) {
1101                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
1102                            "membership for scan events: %d (%s)",
1103                            ret, strerror(-ret));
1104                 goto err4;
1105         }
1106
1107         ret = nl_get_multicast_id(drv, "nl80211", "mlme");
1108         if (ret >= 0)
1109                 ret = nl_socket_add_membership(drv->nl_handle, ret);
1110         if (ret < 0) {
1111                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
1112                            "membership for mlme events: %d (%s)",
1113                            ret, strerror(-ret));
1114                 goto err4;
1115         }
1116         drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
1117
1118         eloop_register_read_sock(nl_socket_get_fd(drv->nl_handle),
1119                                  wpa_driver_nl80211_event_receive, drv, ctx);
1120
1121         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1122         if (drv->ioctl_sock < 0) {
1123                 perror("socket(PF_INET,SOCK_DGRAM)");
1124                 goto err5;
1125         }
1126
1127         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1128         if (s < 0) {
1129                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1130                 goto err6;
1131         }
1132
1133         os_memset(&local, 0, sizeof(local));
1134         local.nl_family = AF_NETLINK;
1135         local.nl_groups = RTMGRP_LINK;
1136         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1137                 perror("bind(netlink)");
1138                 close(s);
1139                 goto err6;
1140         }
1141
1142 #ifndef HOSTAPD
1143         eloop_register_read_sock(s, wpa_driver_nl80211_event_receive_link, drv,
1144                                  ctx);
1145 #endif /* HOSTAPD */
1146         drv->link_event_sock = s;
1147
1148         if (wpa_driver_nl80211_finish_drv_init(drv))
1149                 goto err7;
1150
1151         return drv;
1152
1153 err7:
1154         eloop_unregister_read_sock(drv->link_event_sock);
1155         close(drv->link_event_sock);
1156 err6:
1157         close(drv->ioctl_sock);
1158 err5:
1159         genl_family_put(drv->nl80211);
1160 err4:
1161         nl_cache_free(drv->nl_cache);
1162 err3:
1163         nl_handle_destroy(drv->nl_handle);
1164 err2:
1165         nl_cb_put(drv->nl_cb);
1166 err1:
1167         os_free(drv);
1168         return NULL;
1169 }
1170
1171
1172 static int
1173 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
1174 {
1175         int flags;
1176
1177         drv->ifindex = if_nametoindex(drv->ifname);
1178
1179         if (wpa_driver_nl80211_set_mode(drv, 0) < 0) {
1180                 wpa_printf(MSG_DEBUG, "nl80211: Could not configure driver to "
1181                            "use managed mode");
1182         }
1183
1184         if (wpa_driver_nl80211_get_ifflags(drv, &flags) != 0) {
1185                 wpa_printf(MSG_ERROR, "Could not get interface '%s' flags",
1186                            drv->ifname);
1187                 return -1;
1188         }
1189         if (!(flags & IFF_UP)) {
1190                 if (wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP) != 0) {
1191                         wpa_printf(MSG_ERROR, "Could not set interface '%s' "
1192                                    "UP", drv->ifname);
1193                         return -1;
1194                 }
1195         }
1196
1197         wpa_driver_nl80211_capa(drv);
1198
1199         wpa_driver_nl80211_send_oper_ifla(drv, 1, IF_OPER_DORMANT);
1200
1201         return 0;
1202 }
1203
1204
1205 /**
1206  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
1207  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1208  *
1209  * Shut down driver interface and processing of driver events. Free
1210  * private data buffer if one was allocated in wpa_driver_nl80211_init().
1211  */
1212 static void wpa_driver_nl80211_deinit(void *priv)
1213 {
1214         struct wpa_driver_nl80211_data *drv = priv;
1215         int flags;
1216
1217 #ifdef CONFIG_AP
1218         if (drv->monitor_ifidx >= 0)
1219                 nl80211_remove_iface(drv, drv->monitor_ifidx);
1220         if (drv->monitor_sock >= 0) {
1221                 eloop_unregister_read_sock(drv->monitor_sock);
1222                 close(drv->monitor_sock);
1223         }
1224 #endif /* CONFIG_AP */
1225
1226         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1227
1228         wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED, 0);
1229
1230         wpa_driver_nl80211_send_oper_ifla(priv, 0, IF_OPER_UP);
1231
1232         eloop_unregister_read_sock(drv->link_event_sock);
1233
1234         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0)
1235                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
1236         wpa_driver_nl80211_set_mode(drv, 0);
1237
1238         close(drv->link_event_sock);
1239         close(drv->ioctl_sock);
1240
1241         eloop_unregister_read_sock(nl_socket_get_fd(drv->nl_handle));
1242         genl_family_put(drv->nl80211);
1243         nl_cache_free(drv->nl_cache);
1244         nl_handle_destroy(drv->nl_handle);
1245         nl_cb_put(drv->nl_cb);
1246
1247         os_free(drv);
1248 }
1249
1250
1251 /**
1252  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
1253  * @eloop_ctx: Unused
1254  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
1255  *
1256  * This function can be used as registered timeout when starting a scan to
1257  * generate a scan completed event if the driver does not report this.
1258  */
1259 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1260 {
1261         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
1262 #ifndef HOSTAPD
1263         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
1264 #endif /* HOSTAPD */
1265 }
1266
1267
1268 /**
1269  * wpa_driver_nl80211_scan - Request the driver to initiate scan
1270  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1271  * @params: Scan parameters
1272  * Returns: 0 on success, -1 on failure
1273  */
1274 static int wpa_driver_nl80211_scan(void *priv,
1275                                    struct wpa_driver_scan_params *params)
1276 {
1277         struct wpa_driver_nl80211_data *drv = priv;
1278         int ret = 0, timeout;
1279         struct nl_msg *msg, *ssids, *freqs;
1280         size_t i;
1281
1282         msg = nlmsg_alloc();
1283         ssids = nlmsg_alloc();
1284         freqs = nlmsg_alloc();
1285         if (!msg || !ssids || !freqs) {
1286                 nlmsg_free(msg);
1287                 nlmsg_free(ssids);
1288                 nlmsg_free(freqs);
1289                 return -1;
1290         }
1291
1292         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1293                     NL80211_CMD_TRIGGER_SCAN, 0);
1294
1295         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1296
1297         for (i = 0; i < params->num_ssids; i++) {
1298                 NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
1299                         params->ssids[i].ssid);
1300         }
1301         if (params->num_ssids)
1302                 nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids);
1303
1304         if (params->extra_ies) {
1305                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
1306                         params->extra_ies);
1307         }
1308
1309         if (params->freqs) {
1310                 for (i = 0; params->freqs[i]; i++)
1311                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
1312                 nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs);
1313         }
1314
1315         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1316         msg = NULL;
1317         if (ret) {
1318                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
1319                            "(%s)", ret, strerror(-ret));
1320                 goto nla_put_failure;
1321         }
1322
1323         /* Not all drivers generate "scan completed" wireless event, so try to
1324          * read results after a timeout. */
1325         timeout = 10;
1326         if (drv->scan_complete_events) {
1327                 /*
1328                  * The driver seems to deliver events to notify when scan is
1329                  * complete, so use longer timeout to avoid race conditions
1330                  * with scanning and following association request.
1331                  */
1332                 timeout = 30;
1333         }
1334         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
1335                    "seconds", ret, timeout);
1336         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1337         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
1338                                drv, drv->ctx);
1339
1340 nla_put_failure:
1341         nlmsg_free(ssids);
1342         nlmsg_free(msg);
1343         nlmsg_free(freqs);
1344         return ret;
1345 }
1346
1347
1348 static int bss_info_handler(struct nl_msg *msg, void *arg)
1349 {
1350         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1351         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1352         struct nlattr *bss[NL80211_BSS_MAX + 1];
1353         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
1354                 [NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
1355                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
1356                 [NL80211_BSS_TSF] = { .type = NLA_U64 },
1357                 [NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
1358                 [NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
1359                 [NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
1360                 [NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
1361                 [NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
1362         };
1363         struct wpa_scan_results *res = arg;
1364         struct wpa_scan_res **tmp;
1365         struct wpa_scan_res *r;
1366         const u8 *ie;
1367         size_t ie_len;
1368
1369         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1370                   genlmsg_attrlen(gnlh, 0), NULL);
1371         if (!tb[NL80211_ATTR_BSS])
1372                 return NL_SKIP;
1373         if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
1374                              bss_policy))
1375                 return NL_SKIP;
1376         if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
1377                 ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1378                 ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1379         } else {
1380                 ie = NULL;
1381                 ie_len = 0;
1382         }
1383
1384         r = os_zalloc(sizeof(*r) + ie_len);
1385         if (r == NULL)
1386                 return NL_SKIP;
1387         if (bss[NL80211_BSS_BSSID])
1388                 os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
1389                           ETH_ALEN);
1390         if (bss[NL80211_BSS_FREQUENCY])
1391                 r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
1392         if (bss[NL80211_BSS_BEACON_INTERVAL])
1393                 r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
1394         if (bss[NL80211_BSS_CAPABILITY])
1395                 r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
1396         r->flags |= WPA_SCAN_NOISE_INVALID;
1397         if (bss[NL80211_BSS_SIGNAL_MBM]) {
1398                 r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
1399                 r->level /= 100; /* mBm to dBm */
1400                 r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
1401         } else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
1402                 r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
1403                 r->flags |= WPA_SCAN_LEVEL_INVALID;
1404         } else
1405                 r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
1406         if (bss[NL80211_BSS_TSF])
1407                 r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
1408         r->ie_len = ie_len;
1409         if (ie)
1410                 os_memcpy(r + 1, ie, ie_len);
1411
1412         tmp = os_realloc(res->res,
1413                          (res->num + 1) * sizeof(struct wpa_scan_res *));
1414         if (tmp == NULL) {
1415                 os_free(r);
1416                 return NL_SKIP;
1417         }
1418         tmp[res->num++] = r;
1419         res->res = tmp;
1420
1421         return NL_SKIP;
1422 }
1423
1424
1425 /**
1426  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
1427  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1428  * Returns: Scan results on success, -1 on failure
1429  */
1430 static struct wpa_scan_results *
1431 wpa_driver_nl80211_get_scan_results(void *priv)
1432 {
1433         struct wpa_driver_nl80211_data *drv = priv;
1434         struct nl_msg *msg;
1435         struct wpa_scan_results *res;
1436         int ret;
1437
1438         res = os_zalloc(sizeof(*res));
1439         if (res == NULL)
1440                 return 0;
1441         msg = nlmsg_alloc();
1442         if (!msg)
1443                 goto nla_put_failure;
1444
1445         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, NLM_F_DUMP,
1446                     NL80211_CMD_GET_SCAN, 0);
1447         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1448
1449         ret = send_and_recv_msgs(drv, msg, bss_info_handler, res);
1450         msg = NULL;
1451         if (ret == 0) {
1452                 wpa_printf(MSG_DEBUG, "Received scan results (%lu BSSes)",
1453                            (unsigned long) res->num);
1454                 return res;
1455         }
1456         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
1457                    "(%s)", ret, strerror(-ret));
1458 nla_put_failure:
1459         nlmsg_free(msg);
1460         wpa_scan_results_free(res);
1461         return NULL;
1462 }
1463
1464
1465 static int nl_set_encr(int ifindex, struct wpa_driver_nl80211_data *drv,
1466                        wpa_alg alg, const u8 *addr, int key_idx, int set_tx,
1467                        const u8 *seq, size_t seq_len,
1468                        const u8 *key, size_t key_len)
1469 {
1470         struct nl_msg *msg;
1471         int ret;
1472
1473         wpa_printf(MSG_DEBUG, "%s: ifindex=%d alg=%d addr=%p key_idx=%d "
1474                    "set_tx=%d seq_len=%lu key_len=%lu",
1475                    __func__, ifindex, alg, addr, key_idx, set_tx,
1476                    (unsigned long) seq_len, (unsigned long) key_len);
1477
1478         msg = nlmsg_alloc();
1479         if (!msg)
1480                 return -ENOMEM;
1481
1482         if (alg == WPA_ALG_NONE) {
1483                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1484                             0, NL80211_CMD_DEL_KEY, 0);
1485         } else {
1486                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1487                             0, NL80211_CMD_NEW_KEY, 0);
1488                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
1489                 switch (alg) {
1490                 case WPA_ALG_WEP:
1491                         if (key_len == 5)
1492                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
1493                                             0x000FAC01);
1494                         else
1495                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
1496                                             0x000FAC05);
1497                         break;
1498                 case WPA_ALG_TKIP:
1499                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
1500                         break;
1501                 case WPA_ALG_CCMP:
1502                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
1503                         break;
1504                 case WPA_ALG_IGTK:
1505                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC06);
1506                         break;
1507                 default:
1508                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
1509                                    "algorithm %d", __func__, alg);
1510                         nlmsg_free(msg);
1511                         return -1;
1512                 }
1513         }
1514
1515         if (seq)
1516                 NLA_PUT(msg, NL80211_ATTR_KEY_SEQ, seq_len, seq);
1517
1518         if (addr && os_memcmp(addr, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
1519         {
1520                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
1521                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1522         }
1523         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
1524         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
1525
1526         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1527         if (ret == -ENOENT && alg == WPA_ALG_NONE)
1528                 ret = 0;
1529         if (ret)
1530                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d %s)",
1531                            ret, strerror(-ret));
1532
1533         /*
1534          * If we failed or don't need to set the default TX key (below),
1535          * we're done here.
1536          */
1537         if (ret || !set_tx || alg == WPA_ALG_NONE)
1538                 return ret;
1539 #ifdef HOSTAPD /* FIX: is this needed? */
1540         if (addr)
1541                 return ret;
1542 #endif /* HOSTAPD */
1543
1544         msg = nlmsg_alloc();
1545         if (!msg)
1546                 return -ENOMEM;
1547
1548         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1549                     0, NL80211_CMD_SET_KEY, 0);
1550         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
1551         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
1552         if (alg == WPA_ALG_IGTK)
1553                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
1554         else
1555                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
1556
1557         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1558         if (ret == -ENOENT)
1559                 ret = 0;
1560         if (ret)
1561                 wpa_printf(MSG_DEBUG, "nl80211: set_key default failed; "
1562                            "err=%d %s)", ret, strerror(-ret));
1563         return ret;
1564
1565 nla_put_failure:
1566         return -ENOBUFS;
1567 }
1568
1569
1570 static int wpa_driver_nl80211_set_key(void *priv, wpa_alg alg,
1571                                       const u8 *addr, int key_idx,
1572                                       int set_tx, const u8 *seq,
1573                                       size_t seq_len,
1574                                       const u8 *key, size_t key_len)
1575 {
1576         struct wpa_driver_nl80211_data *drv = priv;
1577         return nl_set_encr(drv->ifindex, drv, alg, addr, key_idx, set_tx, seq,
1578                            seq_len, key, key_len);
1579 }
1580
1581
1582 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
1583                                    const u8 *addr, int cmd, u16 reason_code)
1584 {
1585         int ret = -1;
1586         struct nl_msg *msg;
1587
1588         msg = nlmsg_alloc();
1589         if (!msg)
1590                 return -1;
1591
1592         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0, cmd, 0);
1593
1594         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1595         NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason_code);
1596         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1597
1598         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1599         msg = NULL;
1600         if (ret) {
1601                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
1602                            "(%s)", ret, strerror(-ret));
1603                 goto nla_put_failure;
1604         }
1605         ret = 0;
1606
1607 nla_put_failure:
1608         nlmsg_free(msg);
1609         return ret;
1610 }
1611
1612
1613 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
1614                                              int reason_code)
1615 {
1616         struct wpa_driver_nl80211_data *drv = priv;
1617         wpa_printf(MSG_DEBUG, "%s", __func__);
1618         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DEAUTHENTICATE,
1619                                        reason_code);
1620 }
1621
1622
1623 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
1624                                            int reason_code)
1625 {
1626         struct wpa_driver_nl80211_data *drv = priv;
1627         wpa_printf(MSG_DEBUG, "%s", __func__);
1628         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISASSOCIATE,
1629                                        reason_code);
1630 }
1631
1632
1633 static int wpa_driver_nl80211_authenticate(
1634         void *priv, struct wpa_driver_auth_params *params)
1635 {
1636         struct wpa_driver_nl80211_data *drv = priv;
1637         int ret = -1, i;
1638         struct nl_msg *msg;
1639         enum nl80211_auth_type type;
1640
1641         drv->associated = 0;
1642
1643         msg = nlmsg_alloc();
1644         if (!msg)
1645                 return -1;
1646
1647         wpa_printf(MSG_DEBUG, "nl80211: Authenticate (ifindex=%d)",
1648                    drv->ifindex);
1649
1650         for (i = 0; i < 4; i++) {
1651                 if (!params->wep_key[i])
1652                         continue;
1653                 wpa_driver_nl80211_set_key(drv, WPA_ALG_WEP, NULL, i,
1654                                            i == params->wep_tx_keyidx, NULL, 0,
1655                                            params->wep_key[i],
1656                                            params->wep_key_len[i]);
1657         }
1658
1659         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1660                     NL80211_CMD_AUTHENTICATE, 0);
1661
1662         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1663         if (params->bssid) {
1664                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
1665                            MAC2STR(params->bssid));
1666                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
1667         }
1668         if (params->freq) {
1669                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
1670                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
1671         }
1672         if (params->ssid) {
1673                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
1674                                   params->ssid, params->ssid_len);
1675                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
1676                         params->ssid);
1677         }
1678         wpa_hexdump(MSG_DEBUG, "  * IEs", params->ie, params->ie_len);
1679         if (params->ie)
1680                 NLA_PUT(msg, NL80211_ATTR_IE, params->ie_len, params->ie);
1681         /*
1682          * TODO: if multiple auth_alg options enabled, try them one by one if
1683          * the AP rejects authentication due to unknown auth alg
1684          */
1685         if (params->auth_alg & AUTH_ALG_OPEN_SYSTEM)
1686                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
1687         else if (params->auth_alg & AUTH_ALG_SHARED_KEY)
1688                 type = NL80211_AUTHTYPE_SHARED_KEY;
1689         else if (params->auth_alg & AUTH_ALG_LEAP)
1690                 type = NL80211_AUTHTYPE_NETWORK_EAP;
1691         else if (params->auth_alg & AUTH_ALG_FT)
1692                 type = NL80211_AUTHTYPE_FT;
1693         else
1694                 goto nla_put_failure;
1695         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
1696         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
1697
1698         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1699         msg = NULL;
1700         if (ret) {
1701                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
1702                            "(%s)", ret, strerror(-ret));
1703                 goto nla_put_failure;
1704         }
1705         ret = 0;
1706         wpa_printf(MSG_DEBUG, "nl80211: Authentication request send "
1707                    "successfully");
1708
1709 nla_put_failure:
1710         nlmsg_free(msg);
1711         return ret;
1712 }
1713
1714
1715 #if defined(CONFIG_AP) || defined(HOSTAPD)
1716
1717 struct phy_info_arg {
1718         u16 *num_modes;
1719         struct hostapd_hw_modes *modes;
1720 };
1721
1722 static int phy_info_handler(struct nl_msg *msg, void *arg)
1723 {
1724         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1725         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1726         struct phy_info_arg *phy_info = arg;
1727
1728         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1729
1730         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1731         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1732                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1733                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1734                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1735                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
1736                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1737                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1738         };
1739
1740         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1741         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1742                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1743                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
1744         };
1745
1746         struct nlattr *nl_band;
1747         struct nlattr *nl_freq;
1748         struct nlattr *nl_rate;
1749         int rem_band, rem_freq, rem_rate;
1750         struct hostapd_hw_modes *mode;
1751         int idx, mode_is_set;
1752
1753         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1754                   genlmsg_attrlen(gnlh, 0), NULL);
1755
1756         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1757                 return NL_SKIP;
1758
1759         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
1760                 mode = realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
1761                 if (!mode)
1762                         return NL_SKIP;
1763                 phy_info->modes = mode;
1764
1765                 mode_is_set = 0;
1766
1767                 mode = &phy_info->modes[*(phy_info->num_modes)];
1768                 memset(mode, 0, sizeof(*mode));
1769                 *(phy_info->num_modes) += 1;
1770
1771                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1772                           nla_len(nl_band), NULL);
1773
1774                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
1775                         mode->ht_capab = nla_get_u16(
1776                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
1777                 }
1778
1779                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1780                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1781                                   nla_len(nl_freq), freq_policy);
1782                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1783                                 continue;
1784                         mode->num_channels++;
1785                 }
1786
1787                 mode->channels = calloc(mode->num_channels, sizeof(struct hostapd_channel_data));
1788                 if (!mode->channels)
1789                         return NL_SKIP;
1790
1791                 idx = 0;
1792
1793                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1794                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1795                                   nla_len(nl_freq), freq_policy);
1796                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1797                                 continue;
1798
1799                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1800                         mode->channels[idx].flag = 0;
1801
1802                         if (!mode_is_set) {
1803                                 /* crude heuristic */
1804                                 if (mode->channels[idx].freq < 4000)
1805                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
1806                                 else
1807                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
1808                                 mode_is_set = 1;
1809                         }
1810
1811                         /* crude heuristic */
1812                         if (mode->channels[idx].freq < 4000)
1813                                 if (mode->channels[idx].freq == 2484)
1814                                         mode->channels[idx].chan = 14;
1815                                 else
1816                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
1817                         else
1818                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
1819
1820                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1821                                 mode->channels[idx].flag |=
1822                                         HOSTAPD_CHAN_DISABLED;
1823                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
1824                                 mode->channels[idx].flag |=
1825                                         HOSTAPD_CHAN_PASSIVE_SCAN;
1826                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
1827                                 mode->channels[idx].flag |=
1828                                         HOSTAPD_CHAN_NO_IBSS;
1829                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1830                                 mode->channels[idx].flag |=
1831                                         HOSTAPD_CHAN_RADAR;
1832
1833                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
1834                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1835                                 mode->channels[idx].max_tx_power =
1836                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
1837
1838                         idx++;
1839                 }
1840
1841                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1842                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1843                                   nla_len(nl_rate), rate_policy);
1844                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1845                                 continue;
1846                         mode->num_rates++;
1847                 }
1848
1849                 mode->rates = calloc(mode->num_rates, sizeof(struct hostapd_rate_data));
1850                 if (!mode->rates)
1851                         return NL_SKIP;
1852
1853                 idx = 0;
1854
1855                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1856                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1857                                   nla_len(nl_rate), rate_policy);
1858                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1859                                 continue;
1860                         mode->rates[idx].rate = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
1861
1862                         /* crude heuristic */
1863                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
1864                             mode->rates[idx].rate > 200)
1865                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
1866
1867                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
1868                                 mode->rates[idx].flags |= HOSTAPD_RATE_PREAMBLE2;
1869
1870                         idx++;
1871                 }
1872         }
1873
1874         return NL_SKIP;
1875 }
1876
1877 static struct hostapd_hw_modes *
1878 wpa_driver_nl80211_add_11b(struct hostapd_hw_modes *modes, u16 *num_modes)
1879 {
1880         u16 m;
1881         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1882         int i, mode11g_idx = -1;
1883
1884         /* If only 802.11g mode is included, use it to construct matching
1885          * 802.11b mode data. */
1886
1887         for (m = 0; m < *num_modes; m++) {
1888                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1889                         return modes; /* 802.11b already included */
1890                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1891                         mode11g_idx = m;
1892         }
1893
1894         if (mode11g_idx < 0)
1895                 return modes; /* 2.4 GHz band not supported at all */
1896
1897         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
1898         if (nmodes == NULL)
1899                 return modes; /* Could not add 802.11b mode */
1900
1901         mode = &nmodes[*num_modes];
1902         os_memset(mode, 0, sizeof(*mode));
1903         (*num_modes)++;
1904         modes = nmodes;
1905
1906         mode->mode = HOSTAPD_MODE_IEEE80211B;
1907
1908         mode11g = &modes[mode11g_idx];
1909         mode->num_channels = mode11g->num_channels;
1910         mode->channels = os_malloc(mode11g->num_channels *
1911                                    sizeof(struct hostapd_channel_data));
1912         if (mode->channels == NULL) {
1913                 (*num_modes)--;
1914                 return modes; /* Could not add 802.11b mode */
1915         }
1916         os_memcpy(mode->channels, mode11g->channels,
1917                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
1918
1919         mode->num_rates = 0;
1920         mode->rates = os_malloc(4 * sizeof(struct hostapd_rate_data));
1921         if (mode->rates == NULL) {
1922                 os_free(mode->channels);
1923                 (*num_modes)--;
1924                 return modes; /* Could not add 802.11b mode */
1925         }
1926
1927         for (i = 0; i < mode11g->num_rates; i++) {
1928                 if (mode11g->rates[i].rate > 110 ||
1929                     mode11g->rates[i].flags &
1930                     (HOSTAPD_RATE_ERP | HOSTAPD_RATE_OFDM))
1931                         continue;
1932                 mode->rates[mode->num_rates] = mode11g->rates[i];
1933                 mode->num_rates++;
1934                 if (mode->num_rates == 4)
1935                         break;
1936         }
1937
1938         if (mode->num_rates == 0) {
1939                 os_free(mode->channels);
1940                 os_free(mode->rates);
1941                 (*num_modes)--;
1942                 return modes; /* No 802.11b rates */
1943         }
1944
1945         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1946                    "information");
1947
1948         return modes;
1949 }
1950
1951
1952 static struct hostapd_hw_modes *
1953 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
1954 {
1955         struct wpa_driver_nl80211_data *drv = priv;
1956         struct nl_msg *msg;
1957         struct phy_info_arg result = {
1958                 .num_modes = num_modes,
1959                 .modes = NULL,
1960         };
1961
1962         *num_modes = 0;
1963         *flags = 0;
1964
1965         msg = nlmsg_alloc();
1966         if (!msg)
1967                 return NULL;
1968
1969         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1970                     0, NL80211_CMD_GET_WIPHY, 0);
1971
1972         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1973
1974         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
1975                 return wpa_driver_nl80211_add_11b(result.modes, num_modes);
1976  nla_put_failure:
1977         return NULL;
1978 }
1979
1980
1981 static int wpa_driver_nl80211_send_frame(struct wpa_driver_nl80211_data *drv,
1982                                          const void *data, size_t len,
1983                                          int encrypt)
1984 {
1985         __u8 rtap_hdr[] = {
1986                 0x00, 0x00, /* radiotap version */
1987                 0x0e, 0x00, /* radiotap length */
1988                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
1989                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
1990                 0x00,       /* padding */
1991                 0x00, 0x00, /* RX and TX flags to indicate that */
1992                 0x00, 0x00, /* this is the injected frame directly */
1993         };
1994         struct iovec iov[2] = {
1995                 {
1996                         .iov_base = &rtap_hdr,
1997                         .iov_len = sizeof(rtap_hdr),
1998                 },
1999                 {
2000                         .iov_base = (void *) data,
2001                         .iov_len = len,
2002                 }
2003         };
2004         struct msghdr msg = {
2005                 .msg_name = NULL,
2006                 .msg_namelen = 0,
2007                 .msg_iov = iov,
2008                 .msg_iovlen = 2,
2009                 .msg_control = NULL,
2010                 .msg_controllen = 0,
2011                 .msg_flags = 0,
2012         };
2013
2014         if (encrypt)
2015                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
2016
2017         return sendmsg(drv->monitor_sock, &msg, 0);
2018 }
2019
2020
2021 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
2022                                         size_t data_len)
2023 {
2024         struct wpa_driver_nl80211_data *drv = priv;
2025         struct ieee80211_mgmt *mgmt;
2026         int do_not_encrypt = 0;
2027         u16 fc;
2028
2029         mgmt = (struct ieee80211_mgmt *) data;
2030         fc = le_to_host16(mgmt->frame_control);
2031
2032         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
2033             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
2034                 /*
2035                  * Only one of the authentication frame types is encrypted.
2036                  * In order for static WEP encryption to work properly (i.e.,
2037                  * to not encrypt the frame), we need to tell mac80211 about
2038                  * the frames that must not be encrypted.
2039                  */
2040                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
2041                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
2042                 if (auth_alg == WLAN_AUTH_OPEN ||
2043                     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_trans != 3))
2044                         do_not_encrypt = 1;
2045         }
2046
2047         return wpa_driver_nl80211_send_frame(drv, data, data_len,
2048                                              !do_not_encrypt);
2049 }
2050
2051 #endif /* CONFIG_AP || HOSTAPD */
2052
2053
2054 #ifdef CONFIG_AP
2055
2056 static int wpa_driver_nl80211_set_beacon(void *priv,
2057                                          const u8 *head, size_t head_len,
2058                                          const u8 *tail, size_t tail_len,
2059                                          int dtim_period)
2060 {
2061         struct wpa_driver_nl80211_data *drv = priv;
2062         struct nl_msg *msg;
2063         u8 cmd = NL80211_CMD_NEW_BEACON;
2064         int ret;
2065
2066         msg = nlmsg_alloc();
2067         if (!msg)
2068                 return -ENOMEM;
2069
2070         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (beacon_set=%d)",
2071                    drv->beacon_set);
2072         if (drv->beacon_set)
2073                 cmd = NL80211_CMD_SET_BEACON;
2074
2075         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2076                     0, cmd, 0);
2077         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
2078         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
2079         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2080         if (!drv->beacon_int)
2081                 drv->beacon_int = 100;
2082         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
2083         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, dtim_period);
2084
2085         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2086         if (ret) {
2087                 wpa_printf(MSG_DEBUG, "nl80211: Beacon set failed: %d (%s)",
2088                            ret, strerror(-ret));
2089         } else
2090                 drv->beacon_set = 1;
2091         return ret;
2092  nla_put_failure:
2093         return -ENOBUFS;
2094 }
2095
2096
2097 static int wpa_driver_nl80211_set_beacon_int(void *priv, int value)
2098 {
2099         struct wpa_driver_nl80211_data *drv = priv;
2100         struct nl_msg *msg;
2101
2102         drv->beacon_int = value;
2103
2104         if (!drv->beacon_set)
2105                 return 0;
2106
2107         msg = nlmsg_alloc();
2108         if (!msg)
2109                 return -ENOMEM;
2110
2111         wpa_printf(MSG_DEBUG, "nl80211: Set beacon interval %d "
2112                    "(beacon_set=%d)", value, drv->beacon_set);
2113         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2114                     0, NL80211_CMD_SET_BEACON, 0);
2115         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2116
2117         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
2118
2119         return send_and_recv_msgs(drv, msg, NULL, NULL);
2120  nla_put_failure:
2121         return -ENOBUFS;
2122 }
2123
2124
2125 static int wpa_driver_nl80211_set_freq2(
2126         struct wpa_driver_nl80211_data *drv,
2127         struct wpa_driver_associate_params *params)
2128 {
2129         struct nl_msg *msg;
2130         int ret;
2131
2132         msg = nlmsg_alloc();
2133         if (!msg)
2134                 return -1;
2135
2136         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2137                     NL80211_CMD_SET_WIPHY, 0);
2138
2139         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2140
2141         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
2142
2143         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2144         if (ret == 0)
2145                 return 0;
2146         wpa_printf(MSG_DEBUG, "nl80211: MLME Failed to set channel (freq=%d): "
2147                    "%d (%s)", params->freq, ret, strerror(-ret));
2148 nla_put_failure:
2149         return -1;
2150 }
2151
2152
2153 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
2154                                  int ifidx)
2155 {
2156         struct nl_msg *msg;
2157
2158         msg = nlmsg_alloc();
2159         if (!msg)
2160                 goto nla_put_failure;
2161
2162         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2163                     0, NL80211_CMD_DEL_INTERFACE, 0);
2164         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
2165
2166         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
2167                 return;
2168  nla_put_failure:
2169         wpa_printf(MSG_ERROR, "Failed to remove interface (ifidx=%d).\n",
2170                    ifidx);
2171 }
2172
2173
2174 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
2175                                 const char *ifname, enum nl80211_iftype iftype)
2176 {
2177         struct nl_msg *msg, *flags = NULL;
2178         int ifidx;
2179         int ret = -ENOBUFS;
2180
2181         msg = nlmsg_alloc();
2182         if (!msg)
2183                 return -1;
2184
2185         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2186                     0, NL80211_CMD_NEW_INTERFACE, 0);
2187         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2188         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
2189         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
2190
2191         if (iftype == NL80211_IFTYPE_MONITOR) {
2192                 int err;
2193
2194                 flags = nlmsg_alloc();
2195                 if (!flags)
2196                         goto nla_put_failure;
2197
2198                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
2199
2200                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
2201
2202                 nlmsg_free(flags);
2203
2204                 if (err)
2205                         goto nla_put_failure;
2206         }
2207
2208         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2209         if (ret) {
2210  nla_put_failure:
2211                 wpa_printf(MSG_ERROR, "Failed to create interface %s.",
2212                            ifname);
2213                 return ret;
2214         }
2215
2216         ifidx = if_nametoindex(ifname);
2217
2218         if (ifidx <= 0)
2219                 return -1;
2220
2221         return ifidx;
2222 }
2223
2224
2225 void ap_tx_status(void *ctx, const u8 *addr,
2226                   const u8 *buf, size_t len, int ack);
2227 void ap_rx_from_unknown_sta(void *ctx, const u8 *addr);
2228 void ap_mgmt_rx(void *ctx, u8 *buf, size_t len, u16 stype,
2229                 struct hostapd_frame_info *fi);
2230 void ap_mgmt_tx_cb(void *ctx, u8 *buf, size_t len, u16 stype, int ok);
2231
2232 #endif /* CONFIG_AP */
2233
2234 #if defined(CONFIG_AP) || defined(HOSTAPD)
2235
2236 static void handle_tx_callback(void *ctx, u8 *buf, size_t len, int ok)
2237 {
2238         struct ieee80211_hdr *hdr;
2239         u16 fc, type, stype;
2240
2241         hdr = (struct ieee80211_hdr *) buf;
2242         fc = le_to_host16(hdr->frame_control);
2243
2244         type = WLAN_FC_GET_TYPE(fc);
2245         stype = WLAN_FC_GET_STYPE(fc);
2246
2247         switch (type) {
2248         case WLAN_FC_TYPE_MGMT:
2249                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
2250                            ok ? "ACK" : "fail");
2251 #ifdef HOSTAPD
2252                 hostapd_mgmt_tx_cb(ctx, buf, len, stype, ok);
2253 #else /* HOSTAPD */
2254                 ap_mgmt_tx_cb(ctx, buf, len, stype, ok);
2255 #endif /* HOSTAPD */
2256                 break;
2257         case WLAN_FC_TYPE_CTRL:
2258                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
2259                            ok ? "ACK" : "fail");
2260                 break;
2261         case WLAN_FC_TYPE_DATA:
2262 #ifdef HOSTAPD
2263                 hostapd_tx_status(ctx, hdr->addr1, buf, len, ok);
2264 #else /* HOSTAPD */
2265                 ap_tx_status(ctx, hdr->addr1, buf, len, ok);
2266 #endif /* HOSTAPD */
2267                 break;
2268         default:
2269                 wpa_printf(MSG_DEBUG, "unknown TX callback frame type %d",
2270                            type);
2271                 break;
2272         }
2273 }
2274
2275 #endif /* CONFIG_AP || HOSTAPD */
2276
2277 #ifdef CONFIG_AP
2278
2279 static void handle_frame(struct wpa_driver_nl80211_data *drv,
2280                          u8 *buf, size_t len,
2281                          struct hostapd_frame_info *hfi,
2282                          enum ieee80211_msg_type msg_type)
2283 {
2284         struct ieee80211_hdr *hdr;
2285         u16 fc, type, stype;
2286         size_t data_len = len;
2287
2288         /*
2289          * PS-Poll frames are 16 bytes. All other frames are
2290          * 24 bytes or longer.
2291          */
2292         if (len < 16)
2293                 return;
2294
2295         hdr = (struct ieee80211_hdr *) buf;
2296         fc = le_to_host16(hdr->frame_control);
2297
2298         type = WLAN_FC_GET_TYPE(fc);
2299         stype = WLAN_FC_GET_STYPE(fc);
2300
2301         switch (type) {
2302         case WLAN_FC_TYPE_DATA:
2303                 if (len < 24)
2304                         return;
2305                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
2306                 case WLAN_FC_TODS:
2307                         break;
2308                 case WLAN_FC_FROMDS:
2309                         break;
2310                 default:
2311                         /* discard */
2312                         return;
2313                 }
2314                 break;
2315         case WLAN_FC_TYPE_CTRL:
2316                 /* discard non-ps-poll frames */
2317                 if (stype != WLAN_FC_STYPE_PSPOLL)
2318                         return;
2319                 break;
2320         case WLAN_FC_TYPE_MGMT:
2321                 break;
2322         default:
2323                 /* discard */
2324                 return;
2325         }
2326
2327         switch (msg_type) {
2328         case ieee80211_msg_normal:
2329                 /* continue processing */
2330                 break;
2331         case ieee80211_msg_tx_callback_ack:
2332                 handle_tx_callback(drv->ctx, buf, data_len, 1);
2333                 return;
2334         case ieee80211_msg_tx_callback_fail:
2335                 handle_tx_callback(drv->ctx, buf, data_len, 0);
2336                 return;
2337         }
2338
2339         switch (type) {
2340         case WLAN_FC_TYPE_MGMT:
2341                 if (stype != WLAN_FC_STYPE_BEACON &&
2342                     stype != WLAN_FC_STYPE_PROBE_REQ)
2343                         wpa_printf(MSG_MSGDUMP, "MGMT");
2344                 ap_mgmt_rx(drv->ctx, buf, data_len, stype, hfi);
2345                 break;
2346         case WLAN_FC_TYPE_CTRL:
2347                 /* can only get here with PS-Poll frames */
2348                 wpa_printf(MSG_DEBUG, "CTRL");
2349                 ap_rx_from_unknown_sta(drv->ctx, hdr->addr2);
2350                 break;
2351         case WLAN_FC_TYPE_DATA:
2352                 ap_rx_from_unknown_sta(drv->ctx, hdr->addr2);
2353                 break;
2354         }
2355 }
2356
2357 #endif /* CONFIG_AP */
2358
2359 #if defined(CONFIG_AP) || defined(HOSTAPD)
2360
2361 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
2362 {
2363         struct wpa_driver_nl80211_data *drv = eloop_ctx;
2364         int len;
2365         unsigned char buf[3000];
2366         struct ieee80211_radiotap_iterator iter;
2367         int ret;
2368         struct hostapd_frame_info hfi;
2369         int injected = 0, failed = 0, msg_type, rxflags = 0;
2370
2371         len = recv(sock, buf, sizeof(buf), 0);
2372         if (len < 0) {
2373                 perror("recv");
2374                 return;
2375         }
2376
2377         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
2378                 printf("received invalid radiotap frame\n");
2379                 return;
2380         }
2381
2382         memset(&hfi, 0, sizeof(hfi));
2383
2384         while (1) {
2385                 ret = ieee80211_radiotap_iterator_next(&iter);
2386                 if (ret == -ENOENT)
2387                         break;
2388                 if (ret) {
2389                         printf("received invalid radiotap frame (%d)\n", ret);
2390                         return;
2391                 }
2392                 switch (iter.this_arg_index) {
2393                 case IEEE80211_RADIOTAP_FLAGS:
2394                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
2395                                 len -= 4;
2396                         break;
2397                 case IEEE80211_RADIOTAP_RX_FLAGS:
2398                         rxflags = 1;
2399                         break;
2400                 case IEEE80211_RADIOTAP_TX_FLAGS:
2401                         injected = 1;
2402                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
2403                                         IEEE80211_RADIOTAP_F_TX_FAIL;
2404                         break;
2405                 case IEEE80211_RADIOTAP_DATA_RETRIES:
2406                         break;
2407                 case IEEE80211_RADIOTAP_CHANNEL:
2408                         /* TODO convert from freq/flags to channel number
2409                         hfi.channel = XXX;
2410                         hfi.phytype = XXX;
2411                          */
2412                         break;
2413                 case IEEE80211_RADIOTAP_RATE:
2414                         hfi.datarate = *iter.this_arg * 5;
2415                         break;
2416                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
2417                         hfi.ssi_signal = *iter.this_arg;
2418                         break;
2419                 }
2420         }
2421
2422         if (rxflags && injected)
2423                 return;
2424
2425         if (!injected)
2426                 msg_type = ieee80211_msg_normal;
2427         else if (failed)
2428                 msg_type = ieee80211_msg_tx_callback_fail;
2429         else
2430                 msg_type = ieee80211_msg_tx_callback_ack;
2431
2432         handle_frame(drv, buf + iter.max_length,
2433                      len - iter.max_length, &hfi, msg_type);
2434 }
2435
2436
2437 /*
2438  * we post-process the filter code later and rewrite
2439  * this to the offset to the last instruction
2440  */
2441 #define PASS    0xFF
2442 #define FAIL    0xFE
2443
2444 static struct sock_filter msock_filter_insns[] = {
2445         /*
2446          * do a little-endian load of the radiotap length field
2447          */
2448         /* load lower byte into A */
2449         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
2450         /* put it into X (== index register) */
2451         BPF_STMT(BPF_MISC| BPF_TAX, 0),
2452         /* load upper byte into A */
2453         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
2454         /* left-shift it by 8 */
2455         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
2456         /* or with X */
2457         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
2458         /* put result into X */
2459         BPF_STMT(BPF_MISC| BPF_TAX, 0),
2460
2461         /*
2462          * Allow management frames through, this also gives us those
2463          * management frames that we sent ourselves with status
2464          */
2465         /* load the lower byte of the IEEE 802.11 frame control field */
2466         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
2467         /* mask off frame type and version */
2468         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
2469         /* accept frame if it's both 0, fall through otherwise */
2470         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
2471
2472         /*
2473          * TODO: add a bit to radiotap RX flags that indicates
2474          * that the sending station is not associated, then
2475          * add a filter here that filters on our DA and that flag
2476          * to allow us to deauth frames to that bad station.
2477          *
2478          * Not a regression -- we didn't do it before either.
2479          */
2480
2481 #if 0
2482         /*
2483          * drop non-data frames, WDS frames
2484          */
2485         /* load the lower byte of the frame control field */
2486         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
2487         /* mask off QoS bit */
2488         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
2489         /* drop non-data frames */
2490         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
2491         /* load the upper byte of the frame control field */
2492         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
2493         /* mask off toDS/fromDS */
2494         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
2495         /* drop WDS frames */
2496         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, FAIL, 0),
2497 #endif
2498
2499         /*
2500          * add header length to index
2501          */
2502         /* load the lower byte of the frame control field */
2503         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
2504         /* mask off QoS bit */
2505         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
2506         /* right shift it by 6 to give 0 or 2 */
2507         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
2508         /* add data frame header length */
2509         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
2510         /* add index, was start of 802.11 header */
2511         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
2512         /* move to index, now start of LL header */
2513         BPF_STMT(BPF_MISC | BPF_TAX, 0),
2514
2515         /*
2516          * Accept empty data frames, we use those for
2517          * polling activity.
2518          */
2519         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
2520         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
2521
2522         /*
2523          * Accept EAPOL frames
2524          */
2525         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
2526         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
2527         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
2528         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
2529
2530         /* keep these last two statements or change the code below */
2531         /* return 0 == "DROP" */
2532         BPF_STMT(BPF_RET | BPF_K, 0),
2533         /* return ~0 == "keep all" */
2534         BPF_STMT(BPF_RET | BPF_K, ~0),
2535 };
2536
2537 static struct sock_fprog msock_filter = {
2538         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
2539         .filter = msock_filter_insns,
2540 };
2541
2542
2543 static int add_monitor_filter(int s)
2544 {
2545         int idx;
2546
2547         /* rewrite all PASS/FAIL jump offsets */
2548         for (idx = 0; idx < msock_filter.len; idx++) {
2549                 struct sock_filter *insn = &msock_filter_insns[idx];
2550
2551                 if (BPF_CLASS(insn->code) == BPF_JMP) {
2552                         if (insn->code == (BPF_JMP|BPF_JA)) {
2553                                 if (insn->k == PASS)
2554                                         insn->k = msock_filter.len - idx - 2;
2555                                 else if (insn->k == FAIL)
2556                                         insn->k = msock_filter.len - idx - 3;
2557                         }
2558
2559                         if (insn->jt == PASS)
2560                                 insn->jt = msock_filter.len - idx - 2;
2561                         else if (insn->jt == FAIL)
2562                                 insn->jt = msock_filter.len - idx - 3;
2563
2564                         if (insn->jf == PASS)
2565                                 insn->jf = msock_filter.len - idx - 2;
2566                         else if (insn->jf == FAIL)
2567                                 insn->jf = msock_filter.len - idx - 3;
2568                 }
2569         }
2570
2571         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
2572                        &msock_filter, sizeof(msock_filter))) {
2573                 perror("SO_ATTACH_FILTER");
2574                 return -1;
2575         }
2576
2577         return 0;
2578 }
2579
2580 #endif /* CONFIG_AP || HOSTAPD */
2581
2582 #ifdef CONFIG_AP
2583
2584 static int
2585 nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
2586 {
2587         char buf[IFNAMSIZ];
2588         struct sockaddr_ll ll;
2589         int optval;
2590         socklen_t optlen;
2591         int flags;
2592
2593         snprintf(buf, IFNAMSIZ, "mon.%s", drv->ifname);
2594         buf[IFNAMSIZ - 1] = '\0';
2595
2596         drv->monitor_ifidx =
2597                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR);
2598
2599         if (drv->monitor_ifidx < 0)
2600                 return -1;
2601
2602         if (wpa_driver_nl80211_get_ifflags_ifname(drv, buf, &flags) != 0) {
2603                 wpa_printf(MSG_ERROR, "Could not get interface '%s' flags",
2604                            buf);
2605                 goto error;
2606         }
2607         if (!(flags & IFF_UP)) {
2608                 if (wpa_driver_nl80211_set_ifflags_ifname(drv, buf,
2609                                                           flags | IFF_UP) != 0)
2610                 {
2611                         wpa_printf(MSG_ERROR, "Could not set interface '%s' "
2612                                    "UP", buf);
2613                         goto error;
2614                 }
2615         }
2616
2617         memset(&ll, 0, sizeof(ll));
2618         ll.sll_family = AF_PACKET;
2619         ll.sll_ifindex = drv->monitor_ifidx;
2620         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
2621         if (drv->monitor_sock < 0) {
2622                 perror("socket[PF_PACKET,SOCK_RAW]");
2623                 goto error;
2624         }
2625
2626         if (add_monitor_filter(drv->monitor_sock)) {
2627                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
2628                            "interface; do filtering in user space");
2629                 /* This works, but will cost in performance. */
2630         }
2631
2632         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
2633                  sizeof(ll)) < 0) {
2634                 perror("monitor socket bind");
2635                 goto error;
2636         }
2637
2638         optlen = sizeof(optval);
2639         optval = 20;
2640         if (setsockopt
2641             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
2642                 perror("Failed to set socket priority");
2643                 goto error;
2644         }
2645
2646         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
2647                                      drv, NULL)) {
2648                 printf("Could not register monitor read socket\n");
2649                 goto error;
2650         }
2651
2652         return 0;
2653  error:
2654         nl80211_remove_iface(drv, drv->monitor_ifidx);
2655         return -1;
2656 }
2657
2658
2659 static int wpa_driver_nl80211_ap(struct wpa_driver_nl80211_data *drv,
2660                                  struct wpa_driver_associate_params *params)
2661 {
2662         if (drv->monitor_ifidx < 0 &&
2663             nl80211_create_monitor_interface(drv))
2664                 return -1;
2665
2666         if (wpa_driver_nl80211_set_mode(drv, params->mode) ||
2667             wpa_driver_nl80211_set_freq2(drv, params)) {
2668                 nl80211_remove_iface(drv, drv->monitor_ifidx);
2669                 drv->monitor_ifidx = -1;
2670                 return -1;
2671         }
2672
2673         /* TODO: setup monitor interface (and add code somewhere to remove this
2674          * when AP mode is stopped; associate with mode != 2 or drv_deinit) */
2675
2676         return 0;
2677 }
2678 #endif /* CONFIG_AP */
2679
2680
2681 static int wpa_driver_nl80211_associate(
2682         void *priv, struct wpa_driver_associate_params *params)
2683 {
2684         struct wpa_driver_nl80211_data *drv = priv;
2685         int ret = -1;
2686         struct nl_msg *msg;
2687
2688 #ifdef CONFIG_AP
2689         if (params->mode == 2)
2690                 return wpa_driver_nl80211_ap(drv, params);
2691 #endif /* CONFIG_AP */
2692
2693         wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED,
2694                                           params->drop_unencrypted);
2695
2696         drv->associated = 0;
2697
2698         msg = nlmsg_alloc();
2699         if (!msg)
2700                 return -1;
2701
2702         wpa_printf(MSG_DEBUG, "nl80211: Associate (ifindex=%d)",
2703                    drv->ifindex);
2704         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2705                     NL80211_CMD_ASSOCIATE, 0);
2706
2707         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2708         if (params->bssid) {
2709                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
2710                            MAC2STR(params->bssid));
2711                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
2712         }
2713         if (params->freq) {
2714                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
2715                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
2716         }
2717         if (params->ssid) {
2718                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
2719                                   params->ssid, params->ssid_len);
2720                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
2721                         params->ssid);
2722                 if (params->ssid_len > sizeof(drv->ssid))
2723                         goto nla_put_failure;
2724                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
2725                 drv->ssid_len = params->ssid_len;
2726         }
2727         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
2728         if (params->wpa_ie)
2729                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
2730                         params->wpa_ie);
2731
2732         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2733         msg = NULL;
2734         if (ret) {
2735                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
2736                            "(%s)", ret, strerror(-ret));
2737                 goto nla_put_failure;
2738         }
2739         ret = 0;
2740         wpa_printf(MSG_DEBUG, "nl80211: Association request send "
2741                    "successfully");
2742
2743 nla_put_failure:
2744         nlmsg_free(msg);
2745         return ret;
2746 }
2747
2748
2749 /**
2750  * wpa_driver_nl80211_set_mode - Set wireless mode (infra/adhoc)
2751  * @drv: Pointer to private driver data from wpa_driver_nl80211_init()
2752  * @mode: 0 = infra/BSS (associate with an AP), 1 = adhoc/IBSS
2753  * Returns: 0 on success, -1 on failure
2754  */
2755 static int wpa_driver_nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
2756                                        int mode)
2757 {
2758         int ret = -1, flags;
2759         struct nl_msg *msg;
2760         int nlmode;
2761
2762         switch (mode) {
2763         case 0:
2764                 nlmode = NL80211_IFTYPE_STATION;
2765                 break;
2766         case 1:
2767                 nlmode = NL80211_IFTYPE_ADHOC;
2768                 break;
2769         case 2:
2770                 nlmode = NL80211_IFTYPE_AP;
2771                 break;
2772         default:
2773                 return -1;
2774         }
2775
2776         msg = nlmsg_alloc();
2777         if (!msg)
2778                 return -1;
2779
2780         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2781                     0, NL80211_CMD_SET_INTERFACE, 0);
2782         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2783         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, nlmode);
2784
2785         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2786         if (!ret)
2787                 return 0;
2788         else
2789                 goto try_again;
2790
2791 nla_put_failure:
2792         wpa_printf(MSG_ERROR, "nl80211: Failed to set interface mode: %d (%s)",
2793                    ret, strerror(-ret));
2794         return -1;
2795
2796 try_again:
2797         /* mac80211 doesn't allow mode changes while the device is up, so
2798          * take the device down, try to set the mode again, and bring the
2799          * device back up.
2800          */
2801         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0) {
2802                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
2803
2804                 /* Try to set the mode again while the interface is down */
2805                 msg = nlmsg_alloc();
2806                 if (!msg)
2807                         return -1;
2808
2809                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2810                             0, NL80211_CMD_SET_INTERFACE, 0);
2811                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2812                 NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, nlmode);
2813                 ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2814                 if (ret) {
2815                         wpa_printf(MSG_ERROR, "Failed to set interface %s "
2816                                    "mode(try_again): %d (%s)",
2817                                    drv->ifname, ret, strerror(-ret));
2818                 }
2819
2820                 /* Ignore return value of get_ifflags to ensure that the device
2821                  * is always up like it was before this function was called.
2822                  */
2823                 (void) wpa_driver_nl80211_get_ifflags(drv, &flags);
2824                 (void) wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP);
2825         }
2826
2827         return ret;
2828 }
2829
2830
2831 static int wpa_driver_nl80211_get_capa(void *priv,
2832                                        struct wpa_driver_capa *capa)
2833 {
2834         struct wpa_driver_nl80211_data *drv = priv;
2835         if (!drv->has_capability)
2836                 return -1;
2837         os_memcpy(capa, &drv->capa, sizeof(*capa));
2838         return 0;
2839 }
2840
2841
2842 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
2843 {
2844         struct wpa_driver_nl80211_data *drv = priv;
2845
2846         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
2847                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
2848         drv->operstate = state;
2849         return wpa_driver_nl80211_send_oper_ifla(
2850                 drv, -1, state ? IF_OPER_UP : IF_OPER_DORMANT);
2851 }
2852
2853
2854 #ifdef HOSTAPD
2855
2856 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
2857
2858 static int i802_sta_deauth(void *priv, const u8 *addr, int reason);
2859 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason);
2860
2861
2862 static struct i802_bss * get_bss(struct wpa_driver_nl80211_data *drv,
2863                                  const char *iface)
2864 {
2865         struct i802_bss *bss = &drv->bss;
2866         while (bss) {
2867                 if (os_strncmp(iface, bss->ifname, IFNAMSIZ) == 0)
2868                         return bss;
2869                 bss = bss->next;
2870         }
2871         wpa_printf(MSG_DEBUG, "nl80211: get_bss(%s) failed", iface);
2872         return NULL;
2873 }
2874
2875
2876 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
2877 {
2878         int i;
2879         int *old;
2880
2881         wpa_printf(MSG_DEBUG, "nl80211: Add own interface ifindex %d",
2882                    ifidx);
2883         for (i = 0; i < drv->num_if_indices; i++) {
2884                 if (drv->if_indices[i] == 0) {
2885                         drv->if_indices[i] = ifidx;
2886                         return;
2887                 }
2888         }
2889
2890         if (drv->if_indices != drv->default_if_indices)
2891                 old = drv->if_indices;
2892         else
2893                 old = NULL;
2894
2895         drv->if_indices = realloc(old,
2896                                   sizeof(int) * (drv->num_if_indices + 1));
2897         if (!drv->if_indices) {
2898                 if (!old)
2899                         drv->if_indices = drv->default_if_indices;
2900                 else
2901                         drv->if_indices = old;
2902                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
2903                            "interfaces");
2904                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
2905                 return;
2906         }
2907         drv->if_indices[drv->num_if_indices] = ifidx;
2908         drv->num_if_indices++;
2909 }
2910
2911
2912 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
2913 {
2914         int i;
2915
2916         for (i = 0; i < drv->num_if_indices; i++) {
2917                 if (drv->if_indices[i] == ifidx) {
2918                         drv->if_indices[i] = 0;
2919                         break;
2920                 }
2921         }
2922 }
2923
2924
2925 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
2926 {
2927         int i;
2928
2929         for (i = 0; i < drv->num_if_indices; i++)
2930                 if (drv->if_indices[i] == ifidx)
2931                         return 1;
2932
2933         return 0;
2934 }
2935
2936
2937 static int hostapd_set_iface_flags(struct wpa_driver_nl80211_data *drv,
2938                                    const char *ifname, int dev_up)
2939 {
2940         struct ifreq ifr;
2941
2942         if (drv->ioctl_sock < 0)
2943                 return -1;
2944
2945         memset(&ifr, 0, sizeof(ifr));
2946         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
2947
2948         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
2949                 perror("ioctl[SIOCGIFFLAGS]");
2950                 wpa_printf(MSG_DEBUG, "Could not read interface flags (%s)",
2951                            drv->ifname);
2952                 return -1;
2953         }
2954
2955         if (dev_up)
2956                 ifr.ifr_flags |= IFF_UP;
2957         else
2958                 ifr.ifr_flags &= ~IFF_UP;
2959
2960         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
2961                 perror("ioctl[SIOCSIFFLAGS]");
2962                 return -1;
2963         }
2964
2965         return 0;
2966 }
2967
2968
2969 static int i802_set_key(const char *iface, void *priv, wpa_alg alg,
2970                         const u8 *addr, int key_idx, int set_tx, const u8 *seq,
2971                         size_t seq_len, const u8 *key, size_t key_len)
2972 {
2973         struct wpa_driver_nl80211_data *drv = priv;
2974         return nl_set_encr(if_nametoindex(iface), drv, alg, addr, key_idx,
2975                            set_tx, seq, seq_len, key, key_len);
2976 }
2977
2978
2979 static inline int min_int(int a, int b)
2980 {
2981         if (a < b)
2982                 return a;
2983         return b;
2984 }
2985
2986
2987 static int get_key_handler(struct nl_msg *msg, void *arg)
2988 {
2989         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2990         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2991
2992         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2993                   genlmsg_attrlen(gnlh, 0), NULL);
2994
2995         /*
2996          * TODO: validate the key index and mac address!
2997          * Otherwise, there's a race condition as soon as
2998          * the kernel starts sending key notifications.
2999          */
3000
3001         if (tb[NL80211_ATTR_KEY_SEQ])
3002                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
3003                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
3004         return NL_SKIP;
3005 }
3006
3007
3008 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
3009                            int idx, u8 *seq)
3010 {
3011         struct wpa_driver_nl80211_data *drv = priv;
3012         struct nl_msg *msg;
3013
3014         msg = nlmsg_alloc();
3015         if (!msg)
3016                 return -ENOMEM;
3017
3018         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3019                     0, NL80211_CMD_GET_KEY, 0);
3020
3021         if (addr)
3022                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3023         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
3024         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
3025
3026         memset(seq, 0, 6);
3027
3028         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
3029  nla_put_failure:
3030         return -ENOBUFS;
3031 }
3032
3033
3034 static int i802_set_rate_sets(void *priv, int *supp_rates, int *basic_rates,
3035                               int mode)
3036 {
3037         struct wpa_driver_nl80211_data *drv = priv;
3038         struct nl_msg *msg;
3039         u8 rates[NL80211_MAX_SUPP_RATES];
3040         u8 rates_len = 0;
3041         int i;
3042
3043         msg = nlmsg_alloc();
3044         if (!msg)
3045                 return -ENOMEM;
3046
3047         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3048                     NL80211_CMD_SET_BSS, 0);
3049
3050         for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0; i++)
3051                 rates[rates_len++] = basic_rates[i] / 5;
3052
3053         NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
3054
3055         /* TODO: multi-BSS support */
3056         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3057
3058         return send_and_recv_msgs(drv, msg, NULL, NULL);
3059  nla_put_failure:
3060         return -ENOBUFS;
3061 }
3062
3063
3064 /* Set kernel driver on given frequency (MHz) */
3065 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq)
3066 {
3067         struct wpa_driver_nl80211_data *drv = priv;
3068         struct nl_msg *msg;
3069
3070         msg = nlmsg_alloc();
3071         if (!msg)
3072                 return -1;
3073
3074         drv->last_freq = freq->freq;
3075         drv->last_freq_ht = freq->ht_enabled;
3076
3077         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3078                     NL80211_CMD_SET_WIPHY, 0);
3079
3080         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3081         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq->freq);
3082         if (freq->ht_enabled) {
3083                 switch (freq->sec_channel_offset) {
3084                 case -1:
3085                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3086                                     NL80211_CHAN_HT40MINUS);
3087                         break;
3088                 case 1:
3089                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3090                                     NL80211_CHAN_HT40PLUS);
3091                         break;
3092                 default:
3093                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3094                                     NL80211_CHAN_HT20);
3095                         break;
3096                 }
3097         }
3098
3099         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
3100                 return 0;
3101  nla_put_failure:
3102         return -1;
3103 }
3104
3105
3106 static int i802_set_rts(void *priv, int rts)
3107 {
3108         struct wpa_driver_nl80211_data *drv = priv;
3109         struct iwreq iwr;
3110
3111         memset(&iwr, 0, sizeof(iwr));
3112         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
3113         iwr.u.rts.value = rts;
3114         iwr.u.rts.fixed = 1;
3115
3116         if (ioctl(drv->ioctl_sock, SIOCSIWRTS, &iwr) < 0) {
3117                 perror("ioctl[SIOCSIWRTS]");
3118                 return -1;
3119         }
3120
3121         return 0;
3122 }
3123
3124
3125 static int i802_set_frag(void *priv, int frag)
3126 {
3127         struct wpa_driver_nl80211_data *drv = priv;
3128         struct iwreq iwr;
3129
3130         memset(&iwr, 0, sizeof(iwr));
3131         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
3132         iwr.u.frag.value = frag;
3133         iwr.u.frag.fixed = 1;
3134
3135         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
3136                 perror("ioctl[SIOCSIWFRAG]");
3137                 return -1;
3138         }
3139
3140         return 0;
3141 }
3142
3143
3144 static int i802_set_retry(void *priv, int short_retry, int long_retry)
3145 {
3146         struct wpa_driver_nl80211_data *drv = priv;
3147         struct iwreq iwr;
3148
3149         memset(&iwr, 0, sizeof(iwr));
3150         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
3151
3152         iwr.u.retry.value = short_retry;
3153         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
3154         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
3155                 perror("ioctl[SIOCSIWRETRY(short)]");
3156                 return -1;
3157         }
3158
3159         iwr.u.retry.value = long_retry;
3160         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
3161         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
3162                 perror("ioctl[SIOCSIWRETRY(long)]");
3163                 return -1;
3164         }
3165
3166         return 0;
3167 }
3168
3169
3170 static int i802_flush(void *priv)
3171 {
3172         struct wpa_driver_nl80211_data *drv = priv;
3173         struct nl_msg *msg;
3174
3175         msg = nlmsg_alloc();
3176         if (!msg)
3177                 return -1;
3178
3179         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3180                     0, NL80211_CMD_DEL_STATION, 0);
3181
3182         /*
3183          * XXX: FIX! this needs to flush all VLANs too
3184          */
3185         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3186                     if_nametoindex(drv->ifname));
3187
3188         return send_and_recv_msgs(drv, msg, NULL, NULL);
3189  nla_put_failure:
3190         return -ENOBUFS;
3191 }
3192
3193
3194 static int get_sta_handler(struct nl_msg *msg, void *arg)
3195 {
3196         struct nlattr *tb[NL80211_ATTR_MAX + 1];
3197         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
3198         struct hostap_sta_driver_data *data = arg;
3199         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
3200         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
3201                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
3202                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
3203                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
3204                 [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
3205                 [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
3206         };
3207
3208         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
3209                   genlmsg_attrlen(gnlh, 0), NULL);
3210
3211         /*
3212          * TODO: validate the interface and mac address!
3213          * Otherwise, there's a race condition as soon as
3214          * the kernel starts sending station notifications.
3215          */
3216
3217         if (!tb[NL80211_ATTR_STA_INFO]) {
3218                 wpa_printf(MSG_DEBUG, "sta stats missing!");
3219                 return NL_SKIP;
3220         }
3221         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
3222                              tb[NL80211_ATTR_STA_INFO],
3223                              stats_policy)) {
3224                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
3225                 return NL_SKIP;
3226         }
3227
3228         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
3229                 data->inactive_msec =
3230                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
3231         if (stats[NL80211_STA_INFO_RX_BYTES])
3232                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
3233         if (stats[NL80211_STA_INFO_TX_BYTES])
3234                 data->tx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
3235         if (stats[NL80211_STA_INFO_RX_PACKETS])
3236                 data->rx_packets =
3237                         nla_get_u32(stats[NL80211_STA_INFO_RX_PACKETS]);
3238         if (stats[NL80211_STA_INFO_TX_PACKETS])
3239                 data->tx_packets =
3240                         nla_get_u32(stats[NL80211_STA_INFO_TX_PACKETS]);
3241
3242         return NL_SKIP;
3243 }
3244
3245 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
3246                               const u8 *addr)
3247 {
3248         struct wpa_driver_nl80211_data *drv = priv;
3249         struct nl_msg *msg;
3250
3251         os_memset(data, 0, sizeof(*data));
3252         msg = nlmsg_alloc();
3253         if (!msg)
3254                 return -ENOMEM;
3255
3256         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3257                     0, NL80211_CMD_GET_STATION, 0);
3258
3259         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3260         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3261
3262         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
3263  nla_put_failure:
3264         return -ENOBUFS;
3265 }
3266
3267
3268 static int i802_send_eapol(void *priv, const u8 *addr, const u8 *data,
3269                            size_t data_len, int encrypt, const u8 *own_addr)
3270 {
3271         struct wpa_driver_nl80211_data *drv = priv;
3272         struct ieee80211_hdr *hdr;
3273         size_t len;
3274         u8 *pos;
3275         int res;
3276 #if 0 /* FIX */
3277         int qos = sta->flags & WLAN_STA_WME;
3278 #else
3279         int qos = 0;
3280 #endif
3281
3282         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
3283                 data_len;
3284         hdr = os_zalloc(len);
3285         if (hdr == NULL) {
3286                 printf("malloc() failed for i802_send_data(len=%lu)\n",
3287                        (unsigned long) len);
3288                 return -1;
3289         }
3290
3291         hdr->frame_control =
3292                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
3293         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
3294         if (encrypt)
3295                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
3296 #if 0 /* To be enabled if qos determination is added above */
3297         if (qos) {
3298                 hdr->frame_control |=
3299                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
3300         }
3301 #endif
3302
3303         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
3304         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
3305         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
3306         pos = (u8 *) (hdr + 1);
3307
3308 #if 0 /* To be enabled if qos determination is added above */
3309         if (qos) {
3310                 /* add an empty QoS header if needed */
3311                 pos[0] = 0;
3312                 pos[1] = 0;
3313                 pos += 2;
3314         }
3315 #endif
3316
3317         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
3318         pos += sizeof(rfc1042_header);
3319         WPA_PUT_BE16(pos, ETH_P_PAE);
3320         pos += 2;
3321         memcpy(pos, data, data_len);
3322
3323         res = wpa_driver_nl80211_send_frame(drv, (u8 *) hdr, len, encrypt);
3324         if (res < 0) {
3325                 wpa_printf(MSG_ERROR, "i802_send_eapol - packet len: %lu - "
3326                            "failed: %d (%s)",
3327                            (unsigned long) len, errno, strerror(errno));
3328         }
3329         free(hdr);
3330
3331         return res;
3332 }
3333
3334
3335 static int i802_sta_add(const char *ifname, void *priv,
3336                         struct hostapd_sta_add_params *params)
3337 {
3338         struct wpa_driver_nl80211_data *drv = priv;
3339         struct nl_msg *msg;
3340         int ret = -ENOBUFS;
3341
3342         msg = nlmsg_alloc();
3343         if (!msg)
3344                 return -ENOMEM;
3345
3346         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3347                     0, NL80211_CMD_NEW_STATION, 0);
3348
3349         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3350                     if_nametoindex(drv->ifname));
3351         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
3352         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
3353         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
3354                 params->supp_rates);
3355         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
3356                     params->listen_interval);
3357
3358 #ifdef CONFIG_IEEE80211N
3359         if (params->ht_capabilities) {
3360                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
3361                         params->ht_capabilities->length,
3362                         &params->ht_capabilities->data);
3363         }
3364 #endif /* CONFIG_IEEE80211N */
3365
3366         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3367         if (ret)
3368                 wpa_printf(MSG_DEBUG, "nl80211: NL80211_CMD_NEW_STATION "
3369                            "result: %d (%s)", ret, strerror(-ret));
3370         if (ret == -EEXIST)
3371                 ret = 0;
3372  nla_put_failure:
3373         return ret;
3374 }
3375
3376
3377 static int i802_sta_remove(void *priv, const u8 *addr)
3378 {
3379         struct wpa_driver_nl80211_data *drv = priv;
3380         struct nl_msg *msg;
3381         int ret;
3382
3383         msg = nlmsg_alloc();
3384         if (!msg)
3385                 return -ENOMEM;
3386
3387         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3388                     0, NL80211_CMD_DEL_STATION, 0);
3389
3390         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3391                     if_nametoindex(drv->ifname));
3392         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3393
3394         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3395         if (ret == -ENOENT)
3396                 return 0;
3397         return ret;
3398  nla_put_failure:
3399         return -ENOBUFS;
3400 }
3401
3402
3403 static int i802_sta_set_flags(void *priv, const u8 *addr,
3404                               int total_flags, int flags_or, int flags_and)
3405 {
3406         struct wpa_driver_nl80211_data *drv = priv;
3407         struct nl_msg *msg, *flags = NULL;
3408
3409         msg = nlmsg_alloc();
3410         if (!msg)
3411                 return -ENOMEM;
3412
3413         flags = nlmsg_alloc();
3414         if (!flags) {
3415                 nlmsg_free(msg);
3416                 return -ENOMEM;
3417         }
3418
3419         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3420                     0, NL80211_CMD_SET_STATION, 0);
3421
3422         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3423                     if_nametoindex(drv->ifname));
3424         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3425
3426         if (total_flags & WLAN_STA_AUTHORIZED)
3427                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
3428
3429         if (total_flags & WLAN_STA_WMM)
3430                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
3431
3432         if (total_flags & WLAN_STA_SHORT_PREAMBLE)
3433                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
3434
3435         if (total_flags & WLAN_STA_MFP)
3436                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
3437
3438         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
3439                 goto nla_put_failure;
3440
3441         nlmsg_free(flags);
3442
3443         return send_and_recv_msgs(drv, msg, NULL, NULL);
3444  nla_put_failure:
3445         nlmsg_free(flags);
3446         return -ENOBUFS;
3447 }
3448
3449
3450 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
3451                                     int cw_min, int cw_max, int burst_time)
3452 {
3453         struct wpa_driver_nl80211_data *drv = priv;
3454         struct nl_msg *msg;
3455         struct nlattr *txq, *params;
3456
3457         msg = nlmsg_alloc();
3458         if (!msg)
3459                 return -1;
3460
3461         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3462                     0, NL80211_CMD_SET_WIPHY, 0);
3463
3464         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3465
3466         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
3467         if (!txq)
3468                 goto nla_put_failure;
3469
3470         /* We are only sending parameters for a single TXQ at a time */
3471         params = nla_nest_start(msg, 1);
3472         if (!params)
3473                 goto nla_put_failure;
3474
3475         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, queue);
3476         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
3477          * 32 usec, so need to convert the value here. */
3478         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
3479         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
3480         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
3481         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
3482
3483         nla_nest_end(msg, params);
3484
3485         nla_nest_end(msg, txq);
3486
3487         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
3488                 return 0;
3489  nla_put_failure:
3490         return -1;
3491 }
3492
3493
3494 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv, int ifidx)
3495 {
3496         struct nl_msg *msg;
3497
3498         /* stop listening for EAPOL on this interface */
3499         del_ifidx(drv, ifidx);
3500
3501         msg = nlmsg_alloc();
3502         if (!msg)
3503                 goto nla_put_failure;
3504
3505         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3506                     0, NL80211_CMD_DEL_INTERFACE, 0);
3507         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
3508
3509         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
3510                 return;
3511  nla_put_failure:
3512         printf("Failed to remove interface.\n");
3513 }
3514
3515
3516 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
3517                                 const char *ifname,
3518                                 enum nl80211_iftype iftype,
3519                                 const u8 *addr)
3520 {
3521         struct nl_msg *msg, *flags = NULL;
3522         int ifidx;
3523         struct ifreq ifreq;
3524         struct iwreq iwr;
3525         int ret = -ENOBUFS;
3526
3527         msg = nlmsg_alloc();
3528         if (!msg)
3529                 return -1;
3530
3531         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3532                     0, NL80211_CMD_NEW_INTERFACE, 0);
3533         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3534         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
3535         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
3536
3537         if (iftype == NL80211_IFTYPE_MONITOR) {
3538                 int err;
3539
3540                 flags = nlmsg_alloc();
3541                 if (!flags)
3542                         goto nla_put_failure;
3543
3544                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
3545
3546                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
3547
3548                 nlmsg_free(flags);
3549
3550                 if (err)
3551                         goto nla_put_failure;
3552         }
3553
3554         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3555         if (ret) {
3556  nla_put_failure:
3557                 printf("Failed to create interface %s.\n", ifname);
3558                 return ret;
3559         }
3560
3561         ifidx = if_nametoindex(ifname);
3562
3563         if (ifidx <= 0)
3564                 return -1;
3565
3566         /* start listening for EAPOL on this interface */
3567         add_ifidx(drv, ifidx);
3568
3569         if (addr) {
3570                 switch (iftype) {
3571                 case NL80211_IFTYPE_AP:
3572                         os_strlcpy(ifreq.ifr_name, ifname, IFNAMSIZ);
3573                         memcpy(ifreq.ifr_hwaddr.sa_data, addr, ETH_ALEN);
3574                         ifreq.ifr_hwaddr.sa_family = ARPHRD_ETHER;
3575
3576                         if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifreq)) {
3577                                 nl80211_remove_iface(drv, ifidx);
3578                                 return -1;
3579                         }
3580                         break;
3581                 case NL80211_IFTYPE_WDS:
3582                         memset(&iwr, 0, sizeof(iwr));
3583                         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
3584                         iwr.u.addr.sa_family = ARPHRD_ETHER;
3585                         memcpy(iwr.u.addr.sa_data, addr, ETH_ALEN);
3586                         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr))
3587                                 return -1;
3588                         break;
3589                 default:
3590                         /* nothing */
3591                         break;
3592                 }
3593         }
3594
3595         return ifidx;
3596 }
3597
3598
3599 static int i802_bss_add(void *priv, const char *ifname, const u8 *bssid)
3600 {
3601         struct wpa_driver_nl80211_data *drv = priv;
3602         int ifidx;
3603         struct i802_bss *bss;
3604
3605         bss = os_zalloc(sizeof(*bss));
3606         if (bss == NULL)
3607                 return -1;
3608         os_strlcpy(bss->ifname, ifname, IFNAMSIZ);
3609
3610         ifidx = nl80211_create_iface(priv, ifname, NL80211_IFTYPE_AP, bssid);
3611         if (ifidx < 0) {
3612                 os_free(bss);
3613                 return -1;
3614         }
3615         if (hostapd_set_iface_flags(priv, ifname, 1)) {
3616                 nl80211_remove_iface(priv, ifidx);
3617                 os_free(bss);
3618                 return -1;
3619         }
3620         bss->next = drv->bss.next;
3621         drv->bss.next = bss;
3622         return 0;
3623 }
3624
3625
3626 static int i802_bss_remove(void *priv, const char *ifname)
3627 {
3628         struct wpa_driver_nl80211_data *drv = priv;
3629         struct i802_bss *bss, *prev;
3630         nl80211_remove_iface(priv, if_nametoindex(ifname));
3631         prev = &drv->bss;
3632         bss = drv->bss.next;
3633         while (bss) {
3634                 if (os_strncmp(ifname, bss->ifname, IFNAMSIZ) == 0) {
3635                         prev->next = bss->next;
3636                         os_free(bss);
3637                         break;
3638                 }
3639                 prev = bss;
3640                 bss = bss->next;
3641         }
3642         return 0;
3643 }
3644
3645
3646 static int i802_set_beacon(const char *iface, void *priv,
3647                            const u8 *head, size_t head_len,
3648                            const u8 *tail, size_t tail_len, int dtim_period)
3649 {
3650         struct wpa_driver_nl80211_data *drv = priv;
3651         struct nl_msg *msg;
3652         u8 cmd = NL80211_CMD_NEW_BEACON;
3653         int ret;
3654         struct i802_bss *bss;
3655
3656         bss = get_bss(drv, iface);
3657         if (bss == NULL)
3658                 return -ENOENT;
3659
3660         msg = nlmsg_alloc();
3661         if (!msg)
3662                 return -ENOMEM;
3663
3664         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (iface=%s beacon_set=%d)",
3665                    iface, bss->beacon_set);
3666         if (bss->beacon_set)
3667                 cmd = NL80211_CMD_SET_BEACON;
3668
3669         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3670                     0, cmd, 0);
3671         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
3672         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
3673         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
3674         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
3675         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, dtim_period);
3676
3677         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3678         if (!ret)
3679                 bss->beacon_set = 1;
3680         return ret;
3681  nla_put_failure:
3682         return -ENOBUFS;
3683 }
3684
3685
3686 static int i802_del_beacon(struct wpa_driver_nl80211_data *drv)
3687 {
3688         struct nl_msg *msg;
3689
3690         msg = nlmsg_alloc();
3691         if (!msg)
3692                 return -ENOMEM;
3693
3694         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3695                     0, NL80211_CMD_DEL_BEACON, 0);
3696         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3697
3698         return send_and_recv_msgs(drv, msg, NULL, NULL);
3699  nla_put_failure:
3700         return -ENOBUFS;
3701 }
3702
3703
3704 static int i802_set_beacon_int(void *priv, int value)
3705 {
3706         struct wpa_driver_nl80211_data *drv = priv;
3707         struct nl_msg *msg;
3708
3709         drv->beacon_int = value;
3710
3711         if (!drv->bss.beacon_set)
3712                 return 0;
3713
3714         msg = nlmsg_alloc();
3715         if (!msg)
3716                 return -ENOMEM;
3717
3718         wpa_printf(MSG_DEBUG, "nl80211: Set beacon interval %d "
3719                    "(beacon_set=%d)", value, drv->bss.beacon_set);
3720         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3721                     0, NL80211_CMD_SET_BEACON, 0);
3722         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3723
3724         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
3725
3726         return send_and_recv_msgs(drv, msg, NULL, NULL);
3727  nla_put_failure:
3728         return -ENOBUFS;
3729 }
3730
3731
3732 static int i802_set_bss(void *priv, int cts, int preamble, int slot)
3733 {
3734         struct wpa_driver_nl80211_data *drv = priv;
3735         struct nl_msg *msg;
3736
3737         msg = nlmsg_alloc();
3738         if (!msg)
3739                 return -ENOMEM;
3740
3741         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3742                     NL80211_CMD_SET_BSS, 0);
3743
3744         if (cts >= 0)
3745                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
3746         if (preamble >= 0)
3747                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
3748         if (slot >= 0)
3749                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
3750
3751         /* TODO: multi-BSS support */
3752         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
3753
3754         return send_and_recv_msgs(drv, msg, NULL, NULL);
3755  nla_put_failure:
3756         return -ENOBUFS;
3757 }
3758
3759
3760 static int i802_set_cts_protect(void *priv, int value)
3761 {
3762         return i802_set_bss(priv, value, -1, -1);
3763 }
3764
3765
3766 static int i802_set_preamble(void *priv, int value)
3767 {
3768         return i802_set_bss(priv, -1, value, -1);
3769 }
3770
3771
3772 static int i802_set_short_slot_time(void *priv, int value)
3773 {
3774         return i802_set_bss(priv, -1, -1, value);
3775 }
3776
3777
3778 static enum nl80211_iftype i802_if_type(enum hostapd_driver_if_type type)
3779 {
3780         switch (type) {
3781         case HOSTAPD_IF_VLAN:
3782                 return NL80211_IFTYPE_AP_VLAN;
3783         case HOSTAPD_IF_WDS:
3784                 return NL80211_IFTYPE_WDS;
3785         }
3786         return -1;
3787 }
3788
3789
3790 static int i802_if_add(const char *iface, void *priv,
3791                        enum hostapd_driver_if_type type, char *ifname,
3792                        const u8 *addr)
3793 {
3794         if (nl80211_create_iface(priv, ifname, i802_if_type(type), addr) < 0)
3795                 return -1;
3796         return 0;
3797 }
3798
3799
3800 static int i802_if_update(void *priv, enum hostapd_driver_if_type type,
3801                           char *ifname, const u8 *addr)
3802 {
3803         /* unused at the moment */
3804         return -1;
3805 }
3806
3807
3808 static int i802_if_remove(void *priv, enum hostapd_driver_if_type type,
3809                           const char *ifname, const u8 *addr)
3810 {
3811         nl80211_remove_iface(priv, if_nametoindex(ifname));
3812         return 0;
3813 }
3814
3815
3816 static int i802_set_sta_vlan(void *priv, const u8 *addr,
3817                              const char *ifname, int vlan_id)
3818 {
3819         struct wpa_driver_nl80211_data *drv = priv;
3820         struct nl_msg *msg;
3821
3822         msg = nlmsg_alloc();
3823         if (!msg)
3824                 return -ENOMEM;
3825
3826         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3827                     0, NL80211_CMD_SET_STATION, 0);
3828
3829         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3830                     if_nametoindex(drv->ifname));
3831         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3832         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3833                     if_nametoindex(ifname));
3834
3835         return send_and_recv_msgs(drv, msg, NULL, NULL);
3836  nla_put_failure:
3837         return -ENOBUFS;
3838 }
3839
3840
3841 static void handle_frame(struct wpa_driver_nl80211_data *drv,
3842                          u8 *buf, size_t len,
3843                          struct hostapd_frame_info *hfi,
3844                          enum ieee80211_msg_type msg_type)
3845 {
3846         struct hostapd_iface *iface = drv->hapd->iface;
3847         struct ieee80211_hdr *hdr;
3848         u16 fc, type, stype;
3849         size_t data_len = len;
3850         struct hostapd_data *hapd = NULL;
3851         int broadcast_bssid = 0;
3852         size_t i;
3853         u8 *bssid;
3854
3855         /*
3856          * PS-Poll frames are 16 bytes. All other frames are
3857          * 24 bytes or longer.
3858          */
3859         if (len < 16)
3860                 return;
3861
3862         hdr = (struct ieee80211_hdr *) buf;
3863         fc = le_to_host16(hdr->frame_control);
3864
3865         type = WLAN_FC_GET_TYPE(fc);
3866         stype = WLAN_FC_GET_STYPE(fc);
3867
3868         switch (type) {
3869         case WLAN_FC_TYPE_DATA:
3870                 if (len < 24)
3871                         return;
3872                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
3873                 case WLAN_FC_TODS:
3874                         bssid = hdr->addr1;
3875                         break;
3876                 case WLAN_FC_FROMDS:
3877                         bssid = hdr->addr2;
3878                         break;
3879                 default:
3880                         /* discard */
3881                         return;
3882                 }
3883                 break;
3884         case WLAN_FC_TYPE_CTRL:
3885                 /* discard non-ps-poll frames */
3886                 if (stype != WLAN_FC_STYPE_PSPOLL)
3887                         return;
3888                 bssid = hdr->addr1;
3889                 break;
3890         case WLAN_FC_TYPE_MGMT:
3891                 bssid = hdr->addr3;
3892                 break;
3893         default:
3894                 /* discard */
3895                 return;
3896         }
3897
3898         /* find interface frame belongs to */
3899         for (i = 0; i < iface->num_bss; i++) {
3900                 if (memcmp(bssid, iface->bss[i]->own_addr, ETH_ALEN) == 0) {
3901                         hapd = iface->bss[i];
3902                         break;
3903                 }
3904         }
3905
3906         if (hapd == NULL) {
3907                 hapd = iface->bss[0];
3908
3909                 if (bssid[0] != 0xff || bssid[1] != 0xff ||
3910                     bssid[2] != 0xff || bssid[3] != 0xff ||
3911                     bssid[4] != 0xff || bssid[5] != 0xff) {
3912                         /*
3913                          * Unknown BSSID - drop frame if this is not from
3914                          * passive scanning or a beacon (at least ProbeReq
3915                          * frames to other APs may be allowed through RX
3916                          * filtering in the wlan hw/driver)
3917                          */
3918                         if ((type != WLAN_FC_TYPE_MGMT ||
3919                              stype != WLAN_FC_STYPE_BEACON))
3920                                 return;
3921                 } else
3922                         broadcast_bssid = 1;
3923         }
3924
3925         switch (msg_type) {
3926         case ieee80211_msg_normal:
3927                 /* continue processing */
3928                 break;
3929         case ieee80211_msg_tx_callback_ack:
3930                 handle_tx_callback(hapd, buf, data_len, 1);
3931                 return;
3932         case ieee80211_msg_tx_callback_fail:
3933                 handle_tx_callback(hapd, buf, data_len, 0);
3934                 return;
3935         }
3936
3937         switch (type) {
3938         case WLAN_FC_TYPE_MGMT:
3939                 if (stype != WLAN_FC_STYPE_BEACON &&
3940                     stype != WLAN_FC_STYPE_PROBE_REQ)
3941                         wpa_printf(MSG_MSGDUMP, "MGMT");
3942                 if (broadcast_bssid) {
3943                         for (i = 0; i < iface->num_bss; i++)
3944                                 hostapd_mgmt_rx(iface->bss[i], buf, data_len,
3945                                                 stype, hfi);
3946                 } else
3947                         hostapd_mgmt_rx(hapd, buf, data_len, stype, hfi);
3948                 break;
3949         case WLAN_FC_TYPE_CTRL:
3950                 /* can only get here with PS-Poll frames */
3951                 wpa_printf(MSG_DEBUG, "CTRL");
3952                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
3953                 break;
3954         case WLAN_FC_TYPE_DATA:
3955                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
3956                 break;
3957         }
3958 }
3959
3960
3961 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
3962 {
3963         struct wpa_driver_nl80211_data *drv = eloop_ctx;
3964         struct sockaddr_ll lladdr;
3965         unsigned char buf[3000];
3966         int len;
3967         socklen_t fromlen = sizeof(lladdr);
3968
3969         len = recvfrom(sock, buf, sizeof(buf), 0,
3970                        (struct sockaddr *)&lladdr, &fromlen);
3971         if (len < 0) {
3972                 perror("recv");
3973                 return;
3974         }
3975
3976         if (have_ifidx(drv, lladdr.sll_ifindex)) {
3977                 struct hostapd_data *hapd;
3978                 hapd = hostapd_sta_get_bss(drv->hapd, lladdr.sll_addr);
3979                 if (!hapd)
3980                         return;
3981                 hostapd_eapol_receive(hapd, lladdr.sll_addr, buf, len);
3982         }
3983 }
3984
3985
3986 static int nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
3987 {
3988         char buf[IFNAMSIZ];
3989         struct sockaddr_ll ll;
3990         int optval;
3991         socklen_t optlen;
3992
3993         snprintf(buf, IFNAMSIZ, "mon.%s", drv->ifname);
3994         buf[IFNAMSIZ - 1] = '\0';
3995
3996         drv->monitor_ifidx =
3997                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL);
3998
3999         if (drv->monitor_ifidx < 0)
4000                 return -1;
4001
4002         if (hostapd_set_iface_flags(drv, buf, 1))
4003                 goto error;
4004
4005         memset(&ll, 0, sizeof(ll));
4006         ll.sll_family = AF_PACKET;
4007         ll.sll_ifindex = drv->monitor_ifidx;
4008         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
4009         if (drv->monitor_sock < 0) {
4010                 perror("socket[PF_PACKET,SOCK_RAW]");
4011                 goto error;
4012         }
4013
4014         if (add_monitor_filter(drv->monitor_sock)) {
4015                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
4016                            "interface; do filtering in user space");
4017                 /* This works, but will cost in performance. */
4018         }
4019
4020         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
4021                  sizeof(ll)) < 0) {
4022                 perror("monitor socket bind");
4023                 goto error;
4024         }
4025
4026         optlen = sizeof(optval);
4027         optval = 20;
4028         if (setsockopt
4029             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
4030                 perror("Failed to set socket priority");
4031                 goto error;
4032         }
4033
4034         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
4035                                      drv, NULL)) {
4036                 printf("Could not register monitor read socket\n");
4037                 goto error;
4038         }
4039
4040         return 0;
4041  error:
4042         nl80211_remove_iface(drv, drv->monitor_ifidx);
4043         return -1;
4044 }
4045
4046
4047 static int nl80211_set_mode(struct wpa_driver_nl80211_data *drv, const char *ifname,
4048                             int mode)
4049 {
4050         struct nl_msg *msg;
4051         int ret = -ENOBUFS;
4052
4053         msg = nlmsg_alloc();
4054         if (!msg)
4055                 return -ENOMEM;
4056
4057         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4058                     0, NL80211_CMD_SET_INTERFACE, 0);
4059         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
4060                     if_nametoindex(ifname));
4061         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
4062
4063         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4064         if (!ret)
4065                 return 0;
4066  nla_put_failure:
4067         wpa_printf(MSG_ERROR, "Failed to set interface %s to master "
4068                    "mode.", ifname);
4069         return ret;
4070 }
4071
4072
4073 #ifdef CONFIG_IEEE80211N
4074 static void i802_add_neighbor(struct wpa_driver_nl80211_data *drv, u8 *bssid,
4075                               int freq, u8 *ie, size_t ie_len)
4076 {
4077         struct ieee802_11_elems elems;
4078         int ht, pri_chan = 0, sec_chan = 0;
4079         struct ieee80211_ht_operation *oper;
4080         struct hostapd_neighbor_bss *nnei;
4081
4082         ieee802_11_parse_elems(ie, ie_len, &elems, 0);
4083         ht = elems.ht_capabilities || elems.ht_operation;
4084         if (elems.ht_operation && elems.ht_operation_len >= sizeof(*oper)) {
4085                 oper = (struct ieee80211_ht_operation *) elems.ht_operation;
4086                 pri_chan = oper->control_chan;
4087                 if (oper->ht_param & HT_INFO_HT_PARAM_REC_TRANS_CHNL_WIDTH) {
4088                         if (oper->ht_param &
4089                             HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE)
4090                                 sec_chan = pri_chan + 4;
4091                         else if (oper->ht_param &
4092                             HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW)
4093                                 sec_chan = pri_chan - 4;
4094                 }
4095         }
4096
4097         wpa_printf(MSG_DEBUG, "nl80211: Neighboring BSS - bssid=" MACSTR
4098                    " freq=%d MHz HT=%d pri_chan=%d sec_chan=%d",
4099                    MAC2STR(bssid), freq, ht, pri_chan, sec_chan);
4100
4101         nnei = os_realloc(drv->neighbors, (drv->num_neighbors + 1) *
4102                           sizeof(struct hostapd_neighbor_bss));
4103         if (nnei == NULL)
4104                 return;
4105         drv->neighbors = nnei;
4106         nnei = &nnei[drv->num_neighbors];
4107         os_memcpy(nnei->bssid, bssid, ETH_ALEN);
4108         nnei->freq = freq;
4109         nnei->ht = !!ht;
4110         nnei->pri_chan = pri_chan;
4111         nnei->sec_chan = sec_chan;
4112         drv->num_neighbors++;
4113 }
4114
4115
4116 static int i802_get_scan_freq(struct iw_event *iwe, int *freq)
4117 {
4118         int divi = 1000000, i;
4119
4120         if (iwe->u.freq.e == 0) {
4121                 /*
4122                  * Some drivers do not report frequency, but a channel.
4123                  * Try to map this to frequency by assuming they are using
4124                  * IEEE 802.11b/g.  But don't overwrite a previously parsed
4125                  * frequency if the driver sends both frequency and channel,
4126                  * since the driver may be sending an A-band channel that we
4127                  * don't handle here.
4128                  */
4129
4130                 if (*freq)
4131                         return 0;
4132
4133                 if (iwe->u.freq.m >= 1 && iwe->u.freq.m <= 13) {
4134                         *freq = 2407 + 5 * iwe->u.freq.m;
4135                         return 0;
4136                 } else if (iwe->u.freq.m == 14) {
4137                         *freq = 2484;
4138                         return 0;
4139                 }
4140         }
4141
4142         if (iwe->u.freq.e > 6) {
4143                 wpa_printf(MSG_DEBUG, "Invalid freq in scan results: "
4144                            "m=%d e=%d", iwe->u.freq.m, iwe->u.freq.e);
4145                 return -1;
4146         }
4147
4148         for (i = 0; i < iwe->u.freq.e; i++)
4149                 divi /= 10;
4150         *freq = iwe->u.freq.m / divi;
4151         return 0;
4152 }
4153
4154
4155 static int i802_parse_scan(struct wpa_driver_nl80211_data *drv, u8 *res_buf,
4156                            size_t len)
4157 {
4158         size_t ap_num = 0;
4159         int first;
4160         struct iw_event iwe_buf, *iwe = &iwe_buf;
4161         char *pos, *end, *custom;
4162         u8 bssid[ETH_ALEN];
4163         int freq = 0;
4164         u8 *ie = NULL;
4165         size_t ie_len = 0;
4166
4167         ap_num = 0;
4168         first = 1;
4169
4170         pos = (char *) res_buf;
4171         end = (char *) res_buf + len;
4172
4173         while (pos + IW_EV_LCP_LEN <= end) {
4174                 /* Event data may be unaligned, so make a local, aligned copy
4175                  * before processing. */
4176                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
4177                 if (iwe->len <= IW_EV_LCP_LEN)
4178                         break;
4179
4180                 custom = pos + IW_EV_POINT_LEN;
4181                 if (iwe->cmd == IWEVGENIE) {
4182                         /* WE-19 removed the pointer from struct iw_point */
4183                         char *dpos = (char *) &iwe_buf.u.data.length;
4184                         int dlen = dpos - (char *) &iwe_buf;
4185                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
4186                                   sizeof(struct iw_event) - dlen);
4187                 } else {
4188                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
4189                         custom += IW_EV_POINT_OFF;
4190                 }
4191
4192                 switch (iwe->cmd) {
4193                 case SIOCGIWAP:
4194                         if (!first)
4195                                 i802_add_neighbor(drv, bssid, freq, ie,
4196                                                   ie_len);
4197                         first = 0;
4198                         os_memcpy(bssid, iwe->u.ap_addr.sa_data, ETH_ALEN);
4199                         freq = 0;
4200                         ie = NULL;
4201                         ie_len = 0;
4202                         break;
4203                 case SIOCGIWFREQ:
4204                         i802_get_scan_freq(iwe, &freq);
4205                         break;
4206                 case IWEVGENIE:
4207                         if (custom + iwe->u.data.length > end) {
4208                                 wpa_printf(MSG_ERROR, "IWEVGENIE overflow");
4209                                 return -1;
4210                         }
4211                         ie = (u8 *) custom;
4212                         ie_len = iwe->u.data.length;
4213                         break;
4214                 }
4215
4216                 pos += iwe->len;
4217         }
4218
4219         if (!first)
4220                 i802_add_neighbor(drv, bssid, freq, ie, ie_len);
4221
4222         return 0;
4223 }
4224
4225
4226 static int i802_get_ht_scan_res(struct wpa_driver_nl80211_data *drv)
4227 {
4228         struct iwreq iwr;
4229         u8 *res_buf;
4230         size_t res_buf_len;
4231         int res;
4232
4233         res_buf_len = IW_SCAN_MAX_DATA;
4234         for (;;) {
4235                 res_buf = os_malloc(res_buf_len);
4236                 if (res_buf == NULL)
4237                         return -1;
4238                 os_memset(&iwr, 0, sizeof(iwr));
4239                 os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
4240                 iwr.u.data.pointer = res_buf;
4241                 iwr.u.data.length = res_buf_len;
4242
4243                 if (ioctl(drv->ioctl_sock, SIOCGIWSCAN, &iwr) == 0)
4244                         break;
4245
4246                 if (errno == E2BIG && res_buf_len < 65535) {
4247                         os_free(res_buf);
4248                         res_buf = NULL;
4249                         res_buf_len *= 2;
4250                         if (res_buf_len > 65535)
4251                                 res_buf_len = 65535; /* 16-bit length field */
4252                         wpa_printf(MSG_DEBUG, "Scan results did not fit - "
4253                                    "trying larger buffer (%lu bytes)",
4254                                    (unsigned long) res_buf_len);
4255                 } else {
4256                         perror("ioctl[SIOCGIWSCAN]");
4257                         os_free(res_buf);
4258                         return -1;
4259                 }
4260         }
4261
4262         if (iwr.u.data.length > res_buf_len) {
4263                 os_free(res_buf);
4264                 return -1;
4265         }
4266
4267         res = i802_parse_scan(drv, res_buf, iwr.u.data.length);
4268         os_free(res_buf);
4269
4270         return res;
4271 }
4272
4273
4274 static int i802_is_event_wireless_scan_complete(char *data, int len)
4275 {
4276         struct iw_event iwe_buf, *iwe = &iwe_buf;
4277         char *pos, *end;
4278
4279         pos = data;
4280         end = data + len;
4281
4282         while (pos + IW_EV_LCP_LEN <= end) {
4283                 /* Event data may be unaligned, so make a local, aligned copy
4284                  * before processing. */
4285                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
4286                 if (iwe->cmd == SIOCGIWSCAN)
4287                         return 1;
4288
4289                 pos += iwe->len;
4290         }
4291
4292         return 0;
4293 }
4294
4295
4296 static int i802_is_rtm_scan_complete(int ifindex, struct nlmsghdr *h, int len)
4297 {
4298         struct ifinfomsg *ifi;
4299         int attrlen, _nlmsg_len, rta_len;
4300         struct rtattr *attr;
4301
4302         if (len < (int) sizeof(*ifi))
4303                 return 0;
4304
4305         ifi = NLMSG_DATA(h);
4306
4307         if (ifindex != ifi->ifi_index)
4308                 return 0; /* event for foreign ifindex */
4309
4310         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
4311
4312         attrlen = h->nlmsg_len - _nlmsg_len;
4313         if (attrlen < 0)
4314                 return 0;
4315
4316         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
4317
4318         rta_len = RTA_ALIGN(sizeof(struct rtattr));
4319         while (RTA_OK(attr, attrlen)) {
4320                 if (attr->rta_type == IFLA_WIRELESS &&
4321                     i802_is_event_wireless_scan_complete(
4322                             ((char *) attr) + rta_len,
4323                             attr->rta_len - rta_len))
4324                         return 1;
4325                 attr = RTA_NEXT(attr, attrlen);
4326         }
4327
4328         return 0;
4329 }
4330
4331
4332 static int i802_is_scan_complete(int s, int ifindex)
4333 {
4334         char buf[1024];
4335         int left;
4336         struct nlmsghdr *h;
4337
4338         left = recv(s, buf, sizeof(buf), MSG_DONTWAIT);
4339         if (left < 0) {
4340                 perror("recv(netlink)");
4341                 return 0;
4342         }
4343
4344         h = (struct nlmsghdr *) buf;
4345         while (left >= (int) sizeof(*h)) {
4346                 int len, plen;
4347
4348                 len = h->nlmsg_len;
4349                 plen = len - sizeof(*h);
4350                 if (len > left || plen < 0) {
4351                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
4352                                    "len=%d left=%d plen=%d",
4353                                    len, left, plen);
4354                         break;
4355                 }
4356
4357                 switch (h->nlmsg_type) {
4358                 case RTM_NEWLINK:
4359                         if (i802_is_rtm_scan_complete(ifindex, h, plen))
4360                                 return 1;
4361                         break;
4362                 }
4363
4364                 len = NLMSG_ALIGN(len);
4365                 left -= len;
4366                 h = (struct nlmsghdr *) ((char *) h + len);
4367         }
4368
4369         return 0;
4370 }
4371
4372
4373 static int i802_ht_scan(struct wpa_driver_nl80211_data *drv)
4374 {
4375         struct iwreq iwr;
4376         int s, res, ifindex;
4377         struct sockaddr_nl local;
4378         time_t now, end;
4379         fd_set rfds;
4380         struct timeval tv;
4381
4382         wpa_printf(MSG_DEBUG, "nl80211: Scanning overlapping BSSes before "
4383                    "starting HT 20/40 MHz BSS");
4384
4385         /* Request a new scan */
4386         /* TODO: would be enough to scan the selected band */
4387         os_memset(&iwr, 0, sizeof(iwr));
4388         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
4389         if (ioctl(drv->ioctl_sock, SIOCSIWSCAN, &iwr) < 0) {
4390                 perror("ioctl[SIOCSIWSCAN]");
4391                 return -1;
4392         }
4393
4394         ifindex = if_nametoindex(drv->ifname);
4395
4396         /* Wait for scan completion event or timeout */
4397         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
4398         if (s < 0) {
4399                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
4400                 return -1;
4401         }
4402
4403         os_memset(&local, 0, sizeof(local));
4404         local.nl_family = AF_NETLINK;
4405         local.nl_groups = RTMGRP_LINK;
4406         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
4407                 perror("bind(netlink)");
4408                 close(s);
4409                 return -1;
4410         }
4411
4412         time(&end);
4413         end += 30; /* Wait at most 30 seconds for scan results */
4414         for (;;) {
4415                 time(&now);
4416                 tv.tv_sec = end > now ? end - now : 0;
4417                 tv.tv_usec = 0;
4418                 FD_ZERO(&rfds);
4419                 FD_SET(s, &rfds);
4420                 res = select(s + 1, &rfds, NULL, NULL, &tv);
4421                 if (res < 0) {
4422                         perror("select");
4423                         /* Assume results are ready after 10 seconds wait */
4424                         os_sleep(10, 0);
4425                         break;
4426                 } else if (res) {
4427                         if (i802_is_scan_complete(s, ifindex)) {
4428                                 wpa_printf(MSG_DEBUG, "nl80211: Scan "
4429                                            "completed");
4430                                 break;
4431                         }
4432                 } else {
4433                         wpa_printf(MSG_DEBUG, "nl80211: Scan timeout");
4434                         /* Assume results are ready to be read now */
4435                         break;
4436                 }
4437         }
4438
4439         close(s);
4440
4441         return i802_get_ht_scan_res(drv);
4442 }
4443 #endif /* CONFIG_IEEE80211N */
4444
4445
4446 static int i802_init_sockets(struct wpa_driver_nl80211_data *drv, const u8 *bssid)
4447 {
4448         struct ifreq ifr;
4449         int ret;
4450
4451         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
4452         if (drv->ioctl_sock < 0) {
4453                 perror("socket[PF_INET,SOCK_DGRAM]");
4454                 return -1;
4455         }
4456
4457         /* start listening for EAPOL on the default AP interface */
4458         add_ifidx(drv, if_nametoindex(drv->ifname));
4459
4460         if (hostapd_set_iface_flags(drv, drv->ifname, 0))
4461                 return -1;
4462
4463         if (bssid) {
4464                 os_strlcpy(ifr.ifr_name, drv->ifname, IFNAMSIZ);
4465                 memcpy(ifr.ifr_hwaddr.sa_data, bssid, ETH_ALEN);
4466                 ifr.ifr_hwaddr.sa_family = ARPHRD_ETHER;
4467
4468                 if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifr)) {
4469                         perror("ioctl(SIOCSIFHWADDR)");
4470                         return -1;
4471                 }
4472         }
4473
4474         /*
4475          * initialise generic netlink and nl80211
4476          */
4477         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
4478         if (!drv->nl_cb) {
4479                 printf("Failed to allocate netlink callbacks.\n");
4480                 return -1;
4481         }
4482
4483         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
4484         if (!drv->nl_handle) {
4485                 printf("Failed to allocate netlink handle.\n");
4486                 return -1;
4487         }
4488
4489         if (genl_connect(drv->nl_handle)) {
4490                 printf("Failed to connect to generic netlink.\n");
4491                 return -1;
4492         }
4493
4494 #ifdef CONFIG_LIBNL20
4495         if (genl_ctrl_alloc_cache(drv->nl_handle, &drv->nl_cache) < 0) {
4496                 printf("Failed to allocate generic netlink cache.\n");
4497                 return -1;
4498         }
4499 #else /* CONFIG_LIBNL20 */
4500         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
4501         if (!drv->nl_cache) {
4502                 printf("Failed to allocate generic netlink cache.\n");
4503                 return -1;
4504         }
4505 #endif /* CONFIG_LIBNL20 */
4506
4507         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
4508         if (!drv->nl80211) {
4509                 printf("nl80211 not found.\n");
4510                 return -1;
4511         }
4512
4513         ret = nl_get_multicast_id(drv, "nl80211", "scan");
4514         if (ret >= 0)
4515                 ret = nl_socket_add_membership(drv->nl_handle, ret);
4516         if (ret < 0) {
4517                 wpa_printf(MSG_DEBUG, "nl80211: Could not add multicast "
4518                            "membership for scan events: %d (%s)",
4519                            ret, strerror(-ret));
4520         }
4521
4522         ret = nl_get_multicast_id(drv, "nl80211", "mlme");
4523         if (ret >= 0)
4524                 ret = nl_socket_add_membership(drv->nl_handle, ret);
4525         if (ret < 0) {
4526                 wpa_printf(MSG_DEBUG, "nl80211: Could not add multicast "
4527                            "membership for mlme events: %d (%s)",
4528                            ret, strerror(-ret));
4529         }
4530
4531         eloop_register_read_sock(nl_socket_get_fd(drv->nl_handle),
4532                                  wpa_driver_nl80211_event_receive, drv,
4533                                  drv->hapd);
4534
4535 #ifdef CONFIG_IEEE80211N
4536         if (drv->ht_40mhz_scan) {
4537                 if (nl80211_set_mode(drv, drv->ifname, NL80211_IFTYPE_STATION)
4538                     || hostapd_set_iface_flags(drv, drv->ifname, 1) ||
4539                     i802_ht_scan(drv) ||
4540                     hostapd_set_iface_flags(drv, drv->ifname, 0)) {
4541                         wpa_printf(MSG_ERROR, "Failed to scan channels for "
4542                                    "HT 40 MHz operations");
4543                         return -1;
4544                 }
4545         }
4546 #endif /* CONFIG_IEEE80211N */
4547
4548         /* Initialise a monitor interface */
4549         if (nl80211_create_monitor_interface(drv))
4550                 return -1;
4551
4552         if (nl80211_set_mode(drv, drv->ifname, NL80211_IFTYPE_AP))
4553                 goto fail1;
4554
4555         if (hostapd_set_iface_flags(drv, drv->ifname, 1))
4556                 goto fail1;
4557
4558         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
4559         if (drv->eapol_sock < 0) {
4560                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
4561                 goto fail1;
4562         }
4563
4564         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
4565         {
4566                 printf("Could not register read socket for eapol\n");
4567                 return -1;
4568         }
4569
4570         memset(&ifr, 0, sizeof(ifr));
4571         os_strlcpy(ifr.ifr_name, drv->ifname, sizeof(ifr.ifr_name));
4572         if (ioctl(drv->ioctl_sock, SIOCGIFHWADDR, &ifr) != 0) {
4573                 perror("ioctl(SIOCGIFHWADDR)");
4574                 goto fail1;
4575         }
4576
4577         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
4578                 printf("Invalid HW-addr family 0x%04x\n",
4579                        ifr.ifr_hwaddr.sa_family);
4580                 goto fail1;
4581         }
4582         memcpy(drv->hapd->own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
4583
4584         return 0;
4585
4586 fail1:
4587         nl80211_remove_iface(drv, drv->monitor_ifidx);
4588         return -1;
4589 }
4590
4591
4592 static int i802_get_inact_sec(void *priv, const u8 *addr)
4593 {
4594         struct hostap_sta_driver_data data;
4595         int ret;
4596
4597         data.inactive_msec = (unsigned long) -1;
4598         ret = i802_read_sta_data(priv, &data, addr);
4599         if (ret || data.inactive_msec == (unsigned long) -1)
4600                 return -1;
4601         return data.inactive_msec / 1000;
4602 }
4603
4604
4605 static int i802_sta_clear_stats(void *priv, const u8 *addr)
4606 {
4607 #if 0
4608         /* TODO */
4609 #endif
4610         return 0;
4611 }
4612
4613
4614 static int i802_sta_deauth(void *priv, const u8 *addr, int reason)
4615 {
4616         struct wpa_driver_nl80211_data *drv = priv;
4617         struct ieee80211_mgmt mgmt;
4618
4619         memset(&mgmt, 0, sizeof(mgmt));
4620         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
4621                                           WLAN_FC_STYPE_DEAUTH);
4622         memcpy(mgmt.da, addr, ETH_ALEN);
4623         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
4624         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
4625         mgmt.u.deauth.reason_code = host_to_le16(reason);
4626         return wpa_driver_nl80211_send_mlme(drv, (u8 *) &mgmt,
4627                                             IEEE80211_HDRLEN +
4628                                             sizeof(mgmt.u.deauth));
4629 }
4630
4631
4632 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason)
4633 {
4634         struct wpa_driver_nl80211_data *drv = priv;
4635         struct ieee80211_mgmt mgmt;
4636
4637         memset(&mgmt, 0, sizeof(mgmt));
4638         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
4639                                           WLAN_FC_STYPE_DISASSOC);
4640         memcpy(mgmt.da, addr, ETH_ALEN);
4641         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
4642         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
4643         mgmt.u.disassoc.reason_code = host_to_le16(reason);
4644         return wpa_driver_nl80211_send_mlme(drv, (u8 *) &mgmt,
4645                                             IEEE80211_HDRLEN +
4646                                             sizeof(mgmt.u.disassoc));
4647 }
4648
4649
4650 static const struct hostapd_neighbor_bss *
4651 i802_get_neighbor_bss(void *priv, size_t *num)
4652 {
4653         struct wpa_driver_nl80211_data *drv = priv;
4654         *num = drv->num_neighbors;
4655         return drv->neighbors;
4656 }
4657
4658
4659 static void *i802_init_bssid(struct hostapd_data *hapd, const u8 *bssid)
4660 {
4661         struct wpa_driver_nl80211_data *drv;
4662         size_t i;
4663
4664         drv = os_zalloc(sizeof(struct wpa_driver_nl80211_data));
4665         if (drv == NULL) {
4666                 printf("Could not allocate memory for i802 driver data\n");
4667                 return NULL;
4668         }
4669
4670         drv->hapd = hapd;
4671         memcpy(drv->ifname, hapd->conf->iface, sizeof(drv->ifname));
4672         memcpy(drv->bss.ifname, hapd->conf->iface, sizeof(drv->bss.ifname));
4673         drv->ifindex = if_nametoindex(drv->ifname);
4674
4675         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
4676         drv->if_indices = drv->default_if_indices;
4677         for (i = 0; i < hapd->iface->num_bss; i++) {
4678                 struct hostapd_data *bss = hapd->iface->bss[i];
4679                 if (bss->conf->bridge)
4680                         add_ifidx(drv, if_nametoindex(bss->conf->bridge));
4681         }
4682         drv->ht_40mhz_scan = hapd->iconf->secondary_channel != 0;
4683
4684         if (i802_init_sockets(drv, bssid))
4685                 goto failed;
4686
4687         return drv;
4688
4689 failed:
4690         free(drv);
4691         return NULL;
4692 }
4693
4694
4695 static void *i802_init(struct hostapd_data *hapd)
4696 {
4697         return i802_init_bssid(hapd, NULL);
4698 }
4699
4700
4701 static void i802_deinit(void *priv)
4702 {
4703         struct wpa_driver_nl80211_data *drv = priv;
4704         struct i802_bss *bss, *prev;
4705
4706         if (drv->last_freq_ht) {
4707                 /* Clear HT flags from the driver */
4708                 struct hostapd_freq_params freq;
4709                 os_memset(&freq, 0, sizeof(freq));
4710                 freq.freq = drv->last_freq;
4711                 i802_set_freq(priv, &freq);
4712         }
4713
4714         i802_del_beacon(drv);
4715
4716         /* remove monitor interface */
4717         nl80211_remove_iface(drv, drv->monitor_ifidx);
4718
4719         (void) hostapd_set_iface_flags(drv, drv->ifname, 0);
4720
4721         if (drv->monitor_sock >= 0) {
4722                 eloop_unregister_read_sock(drv->monitor_sock);
4723                 close(drv->monitor_sock);
4724         }
4725         if (drv->ioctl_sock >= 0)
4726                 close(drv->ioctl_sock);
4727         if (drv->eapol_sock >= 0) {
4728                 eloop_unregister_read_sock(drv->eapol_sock);
4729                 close(drv->eapol_sock);
4730         }
4731
4732         eloop_unregister_read_sock(nl_socket_get_fd(drv->nl_handle));
4733         genl_family_put(drv->nl80211);
4734         nl_cache_free(drv->nl_cache);
4735         nl_handle_destroy(drv->nl_handle);
4736         nl_cb_put(drv->nl_cb);
4737
4738         if (drv->if_indices != drv->default_if_indices)
4739                 free(drv->if_indices);
4740
4741         os_free(drv->neighbors);
4742
4743         bss = drv->bss.next;
4744         while (bss) {
4745                 prev = bss;
4746                 bss = bss->next;
4747                 os_free(bss);
4748         }
4749
4750         free(drv);
4751 }
4752
4753 #endif /* HOSTAPD */
4754
4755
4756 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
4757         .name = "nl80211",
4758         .desc = "Linux nl80211/cfg80211",
4759         .get_bssid = wpa_driver_nl80211_get_bssid,
4760         .get_ssid = wpa_driver_nl80211_get_ssid,
4761         .set_key = wpa_driver_nl80211_set_key,
4762         .scan2 = wpa_driver_nl80211_scan,
4763         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
4764         .deauthenticate = wpa_driver_nl80211_deauthenticate,
4765         .disassociate = wpa_driver_nl80211_disassociate,
4766         .authenticate = wpa_driver_nl80211_authenticate,
4767         .associate = wpa_driver_nl80211_associate,
4768         .init = wpa_driver_nl80211_init,
4769         .deinit = wpa_driver_nl80211_deinit,
4770         .get_capa = wpa_driver_nl80211_get_capa,
4771         .set_operstate = wpa_driver_nl80211_set_operstate,
4772         .set_country = wpa_driver_nl80211_set_country,
4773 #ifdef CONFIG_AP
4774         .set_beacon = wpa_driver_nl80211_set_beacon,
4775         .set_beacon_int = wpa_driver_nl80211_set_beacon_int,
4776         .send_mlme = wpa_driver_nl80211_send_mlme,
4777 #endif /* CONFIG_AP */
4778 #if defined(CONFIG_AP) || defined(HOSTAPD)
4779         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
4780 #endif /* CONFIG_AP || HOSTAPD */
4781 #ifdef HOSTAPD
4782         .hapd_init = i802_init,
4783         .init_bssid = i802_init_bssid,
4784         .hapd_deinit = i802_deinit,
4785         .hapd_set_key = i802_set_key,
4786         .get_seqnum = i802_get_seqnum,
4787         .flush = i802_flush,
4788         .read_sta_data = i802_read_sta_data,
4789         .hapd_send_eapol = i802_send_eapol,
4790         .sta_set_flags = i802_sta_set_flags,
4791         .sta_deauth = i802_sta_deauth,
4792         .sta_disassoc = i802_sta_disassoc,
4793         .sta_remove = i802_sta_remove,
4794         .sta_add = i802_sta_add,
4795         .get_inact_sec = i802_get_inact_sec,
4796         .sta_clear_stats = i802_sta_clear_stats,
4797         .set_freq = i802_set_freq,
4798         .set_rts = i802_set_rts,
4799         .set_frag = i802_set_frag,
4800         .set_retry = i802_set_retry,
4801         .set_rate_sets = i802_set_rate_sets,
4802         .hapd_set_beacon = i802_set_beacon,
4803         .hapd_set_beacon_int = i802_set_beacon_int,
4804         .set_cts_protect = i802_set_cts_protect,
4805         .set_preamble = i802_set_preamble,
4806         .set_short_slot_time = i802_set_short_slot_time,
4807         .set_tx_queue_params = i802_set_tx_queue_params,
4808         .bss_add = i802_bss_add,
4809         .bss_remove = i802_bss_remove,
4810         .if_add = i802_if_add,
4811         .if_update = i802_if_update,
4812         .if_remove = i802_if_remove,
4813         .set_sta_vlan = i802_set_sta_vlan,
4814         .get_neighbor_bss = i802_get_neighbor_bss,
4815 #endif /* HOSTAPD */
4816 };