wpa_supplicant AP: Add management frame RX for nl80211
[wpasupplicant] / src / drivers / driver_nl80211.c
1 /*
2  * WPA Supplicant - driver interaction with Linux nl80211/cfg80211
3  * Copyright (c) 2003-2008, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "includes.h"
16 #include <sys/ioctl.h>
17 #include <net/if_arp.h>
18 #include <netlink/genl/genl.h>
19 #include <netlink/genl/family.h>
20 #include <netlink/genl/ctrl.h>
21 #include "nl80211_copy.h"
22 #include "wireless_copy.h"
23
24 #include "common.h"
25 #include "driver.h"
26 #include "eloop.h"
27 #include "ieee802_11_defs.h"
28
29 #ifdef CONFIG_AP
30
31 #include <netpacket/packet.h>
32 #include <linux/filter.h>
33 #include "radiotap.h"
34 #include "radiotap_iter.h"
35 #include "../hostapd/driver.h"
36 #include "../hostapd/hostapd_defs.h"
37
38 #ifndef ETH_P_ALL
39 #define ETH_P_ALL 0x0003
40 #endif
41
42 #endif /* CONFIG_AP */
43
44 #ifndef IFF_LOWER_UP
45 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
46 #endif
47 #ifndef IFF_DORMANT
48 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
49 #endif
50
51 #ifndef IF_OPER_DORMANT
52 #define IF_OPER_DORMANT 5
53 #endif
54 #ifndef IF_OPER_UP
55 #define IF_OPER_UP 6
56 #endif
57
58
59 struct wpa_driver_nl80211_data {
60         void *ctx;
61         int link_event_sock;
62         int ioctl_sock;
63         char ifname[IFNAMSIZ + 1];
64         int ifindex;
65         int if_removed;
66         struct wpa_driver_capa capa;
67         int has_capability;
68
69         int operstate;
70
71         int scan_complete_events;
72
73         struct nl_handle *nl_handle;
74         struct nl_cache *nl_cache;
75         struct nl_cb *nl_cb;
76         struct genl_family *nl80211;
77
78         u8 bssid[ETH_ALEN];
79         int associated;
80         u8 ssid[32];
81         size_t ssid_len;
82
83 #ifdef CONFIG_AP
84         int beacon_int;
85         unsigned int beacon_set:1;
86         int monitor_sock;
87         int monitor_ifidx;
88 #endif /* CONFIG_AP */
89 };
90
91
92 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
93                                             void *timeout_ctx);
94 static int wpa_driver_nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
95                                        int mode);
96 static int
97 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
98
99 #ifdef CONFIG_AP
100 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
101                                  int ifidx);
102 #endif /* CONFIG_AP */
103
104
105 /* nl80211 code */
106 static int ack_handler(struct nl_msg *msg, void *arg)
107 {
108         int *err = arg;
109         *err = 0;
110         return NL_STOP;
111 }
112
113 static int finish_handler(struct nl_msg *msg, void *arg)
114 {
115         int *ret = arg;
116         *ret = 0;
117         return NL_SKIP;
118 }
119
120 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
121                          void *arg)
122 {
123         int *ret = arg;
124         *ret = err->error;
125         return NL_SKIP;
126 }
127
128 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
129                               struct nl_msg *msg,
130                               int (*valid_handler)(struct nl_msg *, void *),
131                               void *valid_data)
132 {
133         struct nl_cb *cb;
134         int err = -ENOMEM;
135
136         cb = nl_cb_clone(drv->nl_cb);
137         if (!cb)
138                 goto out;
139
140         err = nl_send_auto_complete(drv->nl_handle, msg);
141         if (err < 0)
142                 goto out;
143
144         err = 1;
145
146         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
147         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
148         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
149
150         if (valid_handler)
151                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
152                           valid_handler, valid_data);
153
154         while (err > 0)
155                 nl_recvmsgs(drv->nl_handle, cb);
156  out:
157         nl_cb_put(cb);
158         nlmsg_free(msg);
159         return err;
160 }
161
162
163 struct family_data {
164         const char *group;
165         int id;
166 };
167
168
169 static int family_handler(struct nl_msg *msg, void *arg)
170 {
171         struct family_data *res = arg;
172         struct nlattr *tb[CTRL_ATTR_MAX + 1];
173         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
174         struct nlattr *mcgrp;
175         int i;
176
177         nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
178                   genlmsg_attrlen(gnlh, 0), NULL);
179         if (!tb[CTRL_ATTR_MCAST_GROUPS])
180                 return NL_SKIP;
181
182         nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], i) {
183                 struct nlattr *tb2[CTRL_ATTR_MCAST_GRP_MAX + 1];
184                 nla_parse(tb2, CTRL_ATTR_MCAST_GRP_MAX, nla_data(mcgrp),
185                           nla_len(mcgrp), NULL);
186                 if (!tb2[CTRL_ATTR_MCAST_GRP_NAME] ||
187                     !tb2[CTRL_ATTR_MCAST_GRP_ID] ||
188                     os_strncmp(nla_data(tb2[CTRL_ATTR_MCAST_GRP_NAME]),
189                                res->group,
190                                nla_len(tb2[CTRL_ATTR_MCAST_GRP_NAME])) != 0)
191                         continue;
192                 res->id = nla_get_u32(tb2[CTRL_ATTR_MCAST_GRP_ID]);
193                 break;
194         };
195
196         return NL_SKIP;
197 }
198
199
200 static int nl_get_multicast_id(struct wpa_driver_nl80211_data *drv,
201                                const char *family, const char *group)
202 {
203         struct nl_msg *msg;
204         int ret = -1;
205         struct family_data res = { group, -ENOENT };
206
207         msg = nlmsg_alloc();
208         if (!msg)
209                 return -ENOMEM;
210         genlmsg_put(msg, 0, 0, genl_ctrl_resolve(drv->nl_handle, "nlctrl"),
211                     0, 0, CTRL_CMD_GETFAMILY, 0);
212         NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
213
214         ret = send_and_recv_msgs(drv, msg, family_handler, &res);
215         msg = NULL;
216         if (ret == 0)
217                 ret = res.id;
218
219 nla_put_failure:
220         nlmsg_free(msg);
221         return ret;
222 }
223
224
225 static int wpa_driver_nl80211_send_oper_ifla(
226         struct wpa_driver_nl80211_data *drv,
227         int linkmode, int operstate)
228 {
229         struct {
230                 struct nlmsghdr hdr;
231                 struct ifinfomsg ifinfo;
232                 char opts[16];
233         } req;
234         struct rtattr *rta;
235         static int nl_seq;
236         ssize_t ret;
237
238         os_memset(&req, 0, sizeof(req));
239
240         req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
241         req.hdr.nlmsg_type = RTM_SETLINK;
242         req.hdr.nlmsg_flags = NLM_F_REQUEST;
243         req.hdr.nlmsg_seq = ++nl_seq;
244         req.hdr.nlmsg_pid = 0;
245
246         req.ifinfo.ifi_family = AF_UNSPEC;
247         req.ifinfo.ifi_type = 0;
248         req.ifinfo.ifi_index = drv->ifindex;
249         req.ifinfo.ifi_flags = 0;
250         req.ifinfo.ifi_change = 0;
251
252         if (linkmode != -1) {
253                 rta = (struct rtattr *)
254                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
255                 rta->rta_type = IFLA_LINKMODE;
256                 rta->rta_len = RTA_LENGTH(sizeof(char));
257                 *((char *) RTA_DATA(rta)) = linkmode;
258                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
259                         RTA_LENGTH(sizeof(char));
260         }
261         if (operstate != -1) {
262                 rta = (struct rtattr *)
263                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
264                 rta->rta_type = IFLA_OPERSTATE;
265                 rta->rta_len = RTA_LENGTH(sizeof(char));
266                 *((char *) RTA_DATA(rta)) = operstate;
267                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
268                         RTA_LENGTH(sizeof(char));
269         }
270
271         wpa_printf(MSG_DEBUG, "WEXT: Operstate: linkmode=%d, operstate=%d",
272                    linkmode, operstate);
273
274         ret = send(drv->link_event_sock, &req, req.hdr.nlmsg_len, 0);
275         if (ret < 0) {
276                 wpa_printf(MSG_DEBUG, "WEXT: Sending operstate IFLA failed: "
277                            "%s (assume operstate is not supported)",
278                            strerror(errno));
279         }
280
281         return ret < 0 ? -1 : 0;
282 }
283
284
285 static int wpa_driver_nl80211_set_auth_param(
286         struct wpa_driver_nl80211_data *drv, int idx, u32 value)
287 {
288         struct iwreq iwr;
289         int ret = 0;
290
291         os_memset(&iwr, 0, sizeof(iwr));
292         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
293         iwr.u.param.flags = idx & IW_AUTH_INDEX;
294         iwr.u.param.value = value;
295
296         if (ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr) < 0) {
297                 if (errno != EOPNOTSUPP) {
298                         wpa_printf(MSG_DEBUG, "WEXT: SIOCSIWAUTH(param %d "
299                                    "value 0x%x) failed: %s)",
300                                    idx, value, strerror(errno));
301                 }
302                 ret = errno == EOPNOTSUPP ? -2 : -1;
303         }
304
305         return ret;
306 }
307
308
309 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
310 {
311         struct wpa_driver_nl80211_data *drv = priv;
312         if (!drv->associated)
313                 return -1;
314         os_memcpy(bssid, drv->bssid, ETH_ALEN);
315         return 0;
316 }
317
318
319 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
320 {
321         struct wpa_driver_nl80211_data *drv = priv;
322         if (!drv->associated)
323                 return -1;
324         os_memcpy(ssid, drv->ssid, drv->ssid_len);
325         return drv->ssid_len;
326 }
327
328
329 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
330                                           void *ctx, char *buf, size_t len,
331                                           int del)
332 {
333         union wpa_event_data event;
334
335         os_memset(&event, 0, sizeof(event));
336         if (len > sizeof(event.interface_status.ifname))
337                 len = sizeof(event.interface_status.ifname) - 1;
338         os_memcpy(event.interface_status.ifname, buf, len);
339         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
340                 EVENT_INTERFACE_ADDED;
341
342         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
343                    del ? "DEL" : "NEW",
344                    event.interface_status.ifname,
345                    del ? "removed" : "added");
346
347         if (os_strcmp(drv->ifname, event.interface_status.ifname) == 0) {
348                 if (del)
349                         drv->if_removed = 1;
350                 else
351                         drv->if_removed = 0;
352         }
353
354         wpa_supplicant_event(ctx, EVENT_INTERFACE_STATUS, &event);
355 }
356
357
358 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
359                                          struct nlmsghdr *h)
360 {
361         struct ifinfomsg *ifi;
362         int attrlen, _nlmsg_len, rta_len;
363         struct rtattr *attr;
364
365         ifi = NLMSG_DATA(h);
366
367         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
368
369         attrlen = h->nlmsg_len - _nlmsg_len;
370         if (attrlen < 0)
371                 return 0;
372
373         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
374
375         rta_len = RTA_ALIGN(sizeof(struct rtattr));
376         while (RTA_OK(attr, attrlen)) {
377                 if (attr->rta_type == IFLA_IFNAME) {
378                         if (os_strcmp(((char *) attr) + rta_len, drv->ifname)
379                             == 0)
380                                 return 1;
381                         else
382                                 break;
383                 }
384                 attr = RTA_NEXT(attr, attrlen);
385         }
386
387         return 0;
388 }
389
390
391 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
392                                           int ifindex, struct nlmsghdr *h)
393 {
394         if (drv->ifindex == ifindex)
395                 return 1;
396
397         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, h)) {
398                 drv->ifindex = if_nametoindex(drv->ifname);
399                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
400                            "interface");
401                 wpa_driver_nl80211_finish_drv_init(drv);
402                 return 1;
403         }
404
405         return 0;
406 }
407
408
409 static void wpa_driver_nl80211_event_rtm_newlink(struct wpa_driver_nl80211_data *drv,
410                                               void *ctx, struct nlmsghdr *h,
411                                               size_t len)
412 {
413         struct ifinfomsg *ifi;
414         int attrlen, _nlmsg_len, rta_len;
415         struct rtattr * attr;
416
417         if (len < sizeof(*ifi))
418                 return;
419
420         ifi = NLMSG_DATA(h);
421
422         if (!wpa_driver_nl80211_own_ifindex(drv, ifi->ifi_index, h)) {
423                 wpa_printf(MSG_DEBUG, "Ignore event for foreign ifindex %d",
424                            ifi->ifi_index);
425                 return;
426         }
427
428         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
429                    "(%s%s%s%s)",
430                    drv->operstate, ifi->ifi_flags,
431                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
432                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
433                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
434                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
435         /*
436          * Some drivers send the association event before the operup event--in
437          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
438          * fails. This will hit us when wpa_supplicant does not need to do
439          * IEEE 802.1X authentication
440          */
441         if (drv->operstate == 1 &&
442             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
443             !(ifi->ifi_flags & IFF_RUNNING))
444                 wpa_driver_nl80211_send_oper_ifla(drv, -1, IF_OPER_UP);
445
446         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
447
448         attrlen = h->nlmsg_len - _nlmsg_len;
449         if (attrlen < 0)
450                 return;
451
452         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
453
454         rta_len = RTA_ALIGN(sizeof(struct rtattr));
455         while (RTA_OK(attr, attrlen)) {
456                 if (attr->rta_type == IFLA_IFNAME) {
457                         wpa_driver_nl80211_event_link(
458                                 drv, ctx,
459                                 ((char *) attr) + rta_len,
460                                 attr->rta_len - rta_len, 0);
461                 }
462                 attr = RTA_NEXT(attr, attrlen);
463         }
464 }
465
466
467 static void wpa_driver_nl80211_event_rtm_dellink(struct wpa_driver_nl80211_data *drv,
468                                               void *ctx, struct nlmsghdr *h,
469                                               size_t len)
470 {
471         struct ifinfomsg *ifi;
472         int attrlen, _nlmsg_len, rta_len;
473         struct rtattr * attr;
474
475         if (len < sizeof(*ifi))
476                 return;
477
478         ifi = NLMSG_DATA(h);
479
480         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
481
482         attrlen = h->nlmsg_len - _nlmsg_len;
483         if (attrlen < 0)
484                 return;
485
486         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
487
488         rta_len = RTA_ALIGN(sizeof(struct rtattr));
489         while (RTA_OK(attr, attrlen)) {
490                 if (attr->rta_type == IFLA_IFNAME) {
491                         wpa_driver_nl80211_event_link(
492                                 drv, ctx,
493                                 ((char *) attr) + rta_len,
494                                 attr->rta_len - rta_len, 1);
495                 }
496                 attr = RTA_NEXT(attr, attrlen);
497         }
498 }
499
500
501 static void wpa_driver_nl80211_event_receive_link(int sock, void *eloop_ctx,
502                                                   void *sock_ctx)
503 {
504         char buf[8192];
505         int left;
506         struct sockaddr_nl from;
507         socklen_t fromlen;
508         struct nlmsghdr *h;
509         int max_events = 10;
510
511 try_again:
512         fromlen = sizeof(from);
513         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
514                         (struct sockaddr *) &from, &fromlen);
515         if (left < 0) {
516                 if (errno != EINTR && errno != EAGAIN)
517                         perror("recvfrom(netlink)");
518                 return;
519         }
520
521         h = (struct nlmsghdr *) buf;
522         while (left >= (int) sizeof(*h)) {
523                 int len, plen;
524
525                 len = h->nlmsg_len;
526                 plen = len - sizeof(*h);
527                 if (len > left || plen < 0) {
528                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
529                                    "len=%d left=%d plen=%d",
530                                    len, left, plen);
531                         break;
532                 }
533
534                 switch (h->nlmsg_type) {
535                 case RTM_NEWLINK:
536                         wpa_driver_nl80211_event_rtm_newlink(eloop_ctx, sock_ctx,
537                                                           h, plen);
538                         break;
539                 case RTM_DELLINK:
540                         wpa_driver_nl80211_event_rtm_dellink(eloop_ctx, sock_ctx,
541                                                           h, plen);
542                         break;
543                 }
544
545                 len = NLMSG_ALIGN(len);
546                 left -= len;
547                 h = (struct nlmsghdr *) ((char *) h + len);
548         }
549
550         if (left > 0) {
551                 wpa_printf(MSG_DEBUG, "%d extra bytes in the end of netlink "
552                            "message", left);
553         }
554
555         if (--max_events > 0) {
556                 /*
557                  * Try to receive all events in one eloop call in order to
558                  * limit race condition on cases where AssocInfo event, Assoc
559                  * event, and EAPOL frames are received more or less at the
560                  * same time. We want to process the event messages first
561                  * before starting EAPOL processing.
562                  */
563                 goto try_again;
564         }
565 }
566
567
568 static int no_seq_check(struct nl_msg *msg, void *arg)
569 {
570         return NL_OK;
571 }
572
573
574 static void mlme_event_auth(struct wpa_driver_nl80211_data *drv,
575                             const u8 *frame, size_t len)
576 {
577         const struct ieee80211_mgmt *mgmt;
578         union wpa_event_data event;
579
580         mgmt = (const struct ieee80211_mgmt *) frame;
581         if (len < 24 + sizeof(mgmt->u.auth)) {
582                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
583                            "frame");
584                 return;
585         }
586
587         os_memset(&event, 0, sizeof(event));
588         os_memcpy(event.auth.peer, mgmt->sa, ETH_ALEN);
589         event.auth.auth_type = le_to_host16(mgmt->u.auth.auth_alg);
590         event.auth.status_code = le_to_host16(mgmt->u.auth.status_code);
591         if (len > 24 + sizeof(mgmt->u.auth)) {
592                 event.auth.ies = mgmt->u.auth.variable;
593                 event.auth.ies_len = len - 24 - sizeof(mgmt->u.auth);
594         }
595
596         wpa_supplicant_event(drv->ctx, EVENT_AUTH, &event);
597 }
598
599
600 static void mlme_event_assoc(struct wpa_driver_nl80211_data *drv,
601                             const u8 *frame, size_t len)
602 {
603         const struct ieee80211_mgmt *mgmt;
604         union wpa_event_data event;
605         u16 status;
606
607         mgmt = (const struct ieee80211_mgmt *) frame;
608         if (len < 24 + sizeof(mgmt->u.assoc_resp)) {
609                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
610                            "frame");
611                 return;
612         }
613
614         status = le_to_host16(mgmt->u.assoc_resp.status_code);
615         if (status != WLAN_STATUS_SUCCESS) {
616                 os_memset(&event, 0, sizeof(event));
617                 if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
618                         event.assoc_reject.resp_ies =
619                                 (u8 *) mgmt->u.assoc_resp.variable;
620                         event.assoc_reject.resp_ies_len =
621                                 len - 24 - sizeof(mgmt->u.assoc_resp);
622                 }
623                 event.assoc_reject.status_code = status;
624
625                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
626                 return;
627         }
628
629         drv->associated = 1;
630         os_memcpy(drv->bssid, mgmt->sa, ETH_ALEN);
631
632         os_memset(&event, 0, sizeof(event));
633         if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
634                 event.assoc_info.resp_ies = (u8 *) mgmt->u.assoc_resp.variable;
635                 event.assoc_info.resp_ies_len =
636                         len - 24 - sizeof(mgmt->u.assoc_resp);
637         }
638
639         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
640 }
641
642
643 static void mlme_event(struct wpa_driver_nl80211_data *drv,
644                        enum nl80211_commands cmd, struct nlattr *frame)
645 {
646         if (frame == NULL) {
647                 wpa_printf(MSG_DEBUG, "nl80211: MLME event %d without frame "
648                            "data", cmd);
649                 return;
650         }
651
652         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d", cmd);
653         wpa_hexdump(MSG_MSGDUMP, "nl80211: MLME event frame",
654                     nla_data(frame), nla_len(frame));
655
656         switch (cmd) {
657         case NL80211_CMD_AUTHENTICATE:
658                 mlme_event_auth(drv, nla_data(frame), nla_len(frame));
659                 break;
660         case NL80211_CMD_ASSOCIATE:
661                 mlme_event_assoc(drv, nla_data(frame), nla_len(frame));
662                 break;
663         case NL80211_CMD_DEAUTHENTICATE:
664                 drv->associated = 0;
665                 wpa_supplicant_event(drv->ctx, EVENT_DEAUTH, NULL);
666                 break;
667         case NL80211_CMD_DISASSOCIATE:
668                 drv->associated = 0;
669                 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, NULL);
670                 break;
671         default:
672                 break;
673         }
674 }
675
676
677 static int process_event(struct nl_msg *msg, void *arg)
678 {
679         struct wpa_driver_nl80211_data *drv = arg;
680         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
681         struct nlattr *tb[NL80211_ATTR_MAX + 1];
682
683         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
684                   genlmsg_attrlen(gnlh, 0), NULL);
685
686         if (tb[NL80211_ATTR_IFINDEX]) {
687                 int ifindex = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
688                 if (ifindex != drv->ifindex) {
689                         wpa_printf(MSG_DEBUG, "nl80211: Ignored event (cmd=%d)"
690                                    " for foreign interface (ifindex %d)",
691                                    gnlh->cmd, ifindex);
692                         return NL_SKIP;
693                 }
694         }
695
696         switch (gnlh->cmd) {
697         case NL80211_CMD_NEW_SCAN_RESULTS:
698                 wpa_printf(MSG_DEBUG, "nl80211: New scan results available");
699                 drv->scan_complete_events = 1;
700                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
701                                      drv->ctx);
702                 wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, NULL);
703                 break;
704         case NL80211_CMD_SCAN_ABORTED:
705                 wpa_printf(MSG_DEBUG, "nl80211: Scan aborted");
706                 /*
707                  * Need to indicate that scan results are available in order
708                  * not to make wpa_supplicant stop its scanning.
709                  */
710                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
711                                      drv->ctx);
712                 wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, NULL);
713                 break;
714         case NL80211_CMD_AUTHENTICATE:
715         case NL80211_CMD_ASSOCIATE:
716         case NL80211_CMD_DEAUTHENTICATE:
717         case NL80211_CMD_DISASSOCIATE:
718                 mlme_event(drv, gnlh->cmd, tb[NL80211_ATTR_FRAME]);
719                 break;
720         default:
721                 wpa_printf(MSG_DEBUG, "nl80211: Ignored unknown event "
722                            "(cmd=%d)", gnlh->cmd);
723                 break;
724         }
725
726         return NL_SKIP;
727 }
728
729
730 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
731                                              void *sock_ctx)
732 {
733         struct nl_cb *cb;
734         struct wpa_driver_nl80211_data *drv = eloop_ctx;
735
736         wpa_printf(MSG_DEBUG, "nl80211: Event message available");
737
738         cb = nl_cb_clone(drv->nl_cb);
739         if (!cb)
740                 return;
741         nl_cb_set(cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, no_seq_check, NULL);
742         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, process_event, drv);
743         nl_recvmsgs(drv->nl_handle, cb);
744         nl_cb_put(cb);
745 }
746
747
748 static int wpa_driver_nl80211_get_ifflags_ifname(struct wpa_driver_nl80211_data *drv,
749                                               const char *ifname, int *flags)
750 {
751         struct ifreq ifr;
752
753         os_memset(&ifr, 0, sizeof(ifr));
754         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
755         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, (caddr_t) &ifr) < 0) {
756                 perror("ioctl[SIOCGIFFLAGS]");
757                 return -1;
758         }
759         *flags = ifr.ifr_flags & 0xffff;
760         return 0;
761 }
762
763
764 /**
765  * wpa_driver_nl80211_get_ifflags - Get interface flags (SIOCGIFFLAGS)
766  * @drv: driver_nl80211 private data
767  * @flags: Pointer to returned flags value
768  * Returns: 0 on success, -1 on failure
769  */
770 static int wpa_driver_nl80211_get_ifflags(struct wpa_driver_nl80211_data *drv,
771                                           int *flags)
772 {
773         return wpa_driver_nl80211_get_ifflags_ifname(drv, drv->ifname, flags);
774 }
775
776
777 static int wpa_driver_nl80211_set_ifflags_ifname(
778         struct wpa_driver_nl80211_data *drv,
779         const char *ifname, int flags)
780 {
781         struct ifreq ifr;
782
783         os_memset(&ifr, 0, sizeof(ifr));
784         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
785         ifr.ifr_flags = flags & 0xffff;
786         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, (caddr_t) &ifr) < 0) {
787                 perror("SIOCSIFFLAGS");
788                 return -1;
789         }
790         return 0;
791 }
792
793
794 /**
795  * wpa_driver_nl80211_set_ifflags - Set interface flags (SIOCSIFFLAGS)
796  * @drv: driver_nl80211 private data
797  * @flags: New value for flags
798  * Returns: 0 on success, -1 on failure
799  */
800 static int wpa_driver_nl80211_set_ifflags(struct wpa_driver_nl80211_data *drv,
801                                           int flags)
802 {
803         return wpa_driver_nl80211_set_ifflags_ifname(drv, drv->ifname, flags);
804 }
805
806
807 /**
808  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
809  * @priv: driver_nl80211 private data
810  * @alpha2_arg: country to which to switch to
811  * Returns: 0 on success, -1 on failure
812  *
813  * This asks nl80211 to set the regulatory domain for given
814  * country ISO / IEC alpha2.
815  */
816 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
817 {
818         struct wpa_driver_nl80211_data *drv = priv;
819         char alpha2[3];
820         struct nl_msg *msg;
821
822         msg = nlmsg_alloc();
823         if (!msg)
824                 goto nla_put_failure;
825
826         alpha2[0] = alpha2_arg[0];
827         alpha2[1] = alpha2_arg[1];
828         alpha2[2] = '\0';
829
830         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
831                     0, NL80211_CMD_REQ_SET_REG, 0);
832
833         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
834         if (send_and_recv_msgs(drv, msg, NULL, NULL))
835                 return -EINVAL;
836         return 0;
837 nla_put_failure:
838         return -EINVAL;
839 }
840
841
842 struct wiphy_info_data {
843         int max_scan_ssids;
844         int ap_supported;
845 };
846
847
848 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
849 {
850         struct nlattr *tb[NL80211_ATTR_MAX + 1];
851         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
852         struct wiphy_info_data *info = arg;
853
854         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
855                   genlmsg_attrlen(gnlh, 0), NULL);
856
857         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
858                 info->max_scan_ssids =
859                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
860
861         if (tb[NL80211_ATTR_SUPPORTED_IFTYPES]) {
862                 struct nlattr *nl_mode;
863                 int i;
864                 nla_for_each_nested(nl_mode,
865                                     tb[NL80211_ATTR_SUPPORTED_IFTYPES], i) {
866                         if (nl_mode->nla_type == NL80211_IFTYPE_AP) {
867                                 info->ap_supported = 1;
868                                 break;
869                         }
870                 }
871         }
872
873         return NL_SKIP;
874 }
875
876
877 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
878                                        struct wiphy_info_data *info)
879 {
880         struct nl_msg *msg;
881
882         os_memset(info, 0, sizeof(*info));
883         msg = nlmsg_alloc();
884         if (!msg)
885                 return -1;
886
887         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
888                     0, NL80211_CMD_GET_WIPHY, 0);
889
890         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
891
892         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info) == 0)
893                 return 0;
894         msg = NULL;
895 nla_put_failure:
896         nlmsg_free(msg);
897         return -1;
898 }
899
900
901 static void wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
902 {
903         struct wiphy_info_data info;
904         if (wpa_driver_nl80211_get_info(drv, &info))
905                 return;
906         drv->has_capability = 1;
907         /* For now, assume TKIP, CCMP, WPA, WPA2 are supported */
908         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
909                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
910                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
911                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
912         drv->capa.enc = WPA_DRIVER_CAPA_ENC_WEP40 |
913                 WPA_DRIVER_CAPA_ENC_WEP104 |
914                 WPA_DRIVER_CAPA_ENC_TKIP |
915                 WPA_DRIVER_CAPA_ENC_CCMP;
916
917         drv->capa.max_scan_ssids = info.max_scan_ssids;
918         if (info.ap_supported)
919                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP;
920 }
921
922
923 /**
924  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
925  * @ctx: context to be used when calling wpa_supplicant functions,
926  * e.g., wpa_supplicant_event()
927  * @ifname: interface name, e.g., wlan0
928  * Returns: Pointer to private data, %NULL on failure
929  */
930 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname)
931 {
932         int s, ret;
933         struct sockaddr_nl local;
934         struct wpa_driver_nl80211_data *drv;
935
936         drv = os_zalloc(sizeof(*drv));
937         if (drv == NULL)
938                 return NULL;
939         drv->ctx = ctx;
940         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
941 #ifdef CONFIG_AP
942         drv->monitor_ifidx = -1;
943         drv->monitor_sock = -1;
944 #endif /* CONFIG_AP */
945
946         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
947         if (drv->nl_cb == NULL) {
948                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
949                            "callbacks");
950                 goto err1;
951         }
952
953         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
954         if (drv->nl_handle == NULL) {
955                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
956                            "callbacks");
957                 goto err2;
958         }
959
960         if (genl_connect(drv->nl_handle)) {
961                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
962                            "netlink");
963                 goto err3;
964         }
965
966         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
967         if (drv->nl_cache == NULL) {
968                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
969                            "netlink cache");
970                 goto err3;
971         }
972
973         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
974         if (drv->nl80211 == NULL) {
975                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
976                            "found");
977                 goto err4;
978         }
979
980         ret = nl_get_multicast_id(drv, "nl80211", "scan");
981         if (ret >= 0)
982                 ret = nl_socket_add_membership(drv->nl_handle, ret);
983         if (ret < 0) {
984                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
985                            "membership for scan events: %d (%s)",
986                            ret, strerror(-ret));
987                 goto err4;
988         }
989
990         ret = nl_get_multicast_id(drv, "nl80211", "mlme");
991         if (ret >= 0)
992                 ret = nl_socket_add_membership(drv->nl_handle, ret);
993         if (ret < 0) {
994                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
995                            "membership for mlme events: %d (%s)",
996                            ret, strerror(-ret));
997                 goto err4;
998         }
999         drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
1000
1001         eloop_register_read_sock(nl_socket_get_fd(drv->nl_handle),
1002                                  wpa_driver_nl80211_event_receive, drv, ctx);
1003
1004         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1005         if (drv->ioctl_sock < 0) {
1006                 perror("socket(PF_INET,SOCK_DGRAM)");
1007                 goto err5;
1008         }
1009
1010         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1011         if (s < 0) {
1012                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1013                 goto err6;
1014         }
1015
1016         os_memset(&local, 0, sizeof(local));
1017         local.nl_family = AF_NETLINK;
1018         local.nl_groups = RTMGRP_LINK;
1019         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1020                 perror("bind(netlink)");
1021                 close(s);
1022                 goto err6;
1023         }
1024
1025         eloop_register_read_sock(s, wpa_driver_nl80211_event_receive_link, drv,
1026                                  ctx);
1027         drv->link_event_sock = s;
1028
1029         if (wpa_driver_nl80211_finish_drv_init(drv))
1030                 goto err7;
1031
1032         return drv;
1033
1034 err7:
1035         eloop_unregister_read_sock(drv->link_event_sock);
1036         close(drv->link_event_sock);
1037 err6:
1038         close(drv->ioctl_sock);
1039 err5:
1040         genl_family_put(drv->nl80211);
1041 err4:
1042         nl_cache_free(drv->nl_cache);
1043 err3:
1044         nl_handle_destroy(drv->nl_handle);
1045 err2:
1046         nl_cb_put(drv->nl_cb);
1047 err1:
1048         os_free(drv);
1049         return NULL;
1050 }
1051
1052
1053 static int
1054 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
1055 {
1056         int flags;
1057
1058         drv->ifindex = if_nametoindex(drv->ifname);
1059
1060         if (wpa_driver_nl80211_set_mode(drv, 0) < 0) {
1061                 wpa_printf(MSG_DEBUG, "nl80211: Could not configure driver to "
1062                            "use managed mode");
1063         }
1064
1065         if (wpa_driver_nl80211_get_ifflags(drv, &flags) != 0) {
1066                 wpa_printf(MSG_ERROR, "Could not get interface '%s' flags",
1067                            drv->ifname);
1068                 return -1;
1069         }
1070         if (!(flags & IFF_UP)) {
1071                 if (wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP) != 0) {
1072                         wpa_printf(MSG_ERROR, "Could not set interface '%s' "
1073                                    "UP", drv->ifname);
1074                         return -1;
1075                 }
1076         }
1077
1078         wpa_driver_nl80211_capa(drv);
1079
1080         wpa_driver_nl80211_send_oper_ifla(drv, 1, IF_OPER_DORMANT);
1081
1082         return 0;
1083 }
1084
1085
1086 /**
1087  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
1088  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1089  *
1090  * Shut down driver interface and processing of driver events. Free
1091  * private data buffer if one was allocated in wpa_driver_nl80211_init().
1092  */
1093 static void wpa_driver_nl80211_deinit(void *priv)
1094 {
1095         struct wpa_driver_nl80211_data *drv = priv;
1096         int flags;
1097
1098 #ifdef CONFIG_AP
1099         if (drv->monitor_ifidx >= 0)
1100                 nl80211_remove_iface(drv, drv->monitor_ifidx);
1101         if (drv->monitor_sock >= 0) {
1102                 eloop_unregister_read_sock(drv->monitor_sock);
1103                 close(drv->monitor_sock);
1104         }
1105 #endif /* CONFIG_AP */
1106
1107         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1108
1109         wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED, 0);
1110
1111         wpa_driver_nl80211_send_oper_ifla(priv, 0, IF_OPER_UP);
1112
1113         eloop_unregister_read_sock(drv->link_event_sock);
1114
1115         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0)
1116                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
1117         wpa_driver_nl80211_set_mode(drv, 0);
1118
1119         close(drv->link_event_sock);
1120         close(drv->ioctl_sock);
1121
1122         eloop_unregister_read_sock(nl_socket_get_fd(drv->nl_handle));
1123         genl_family_put(drv->nl80211);
1124         nl_cache_free(drv->nl_cache);
1125         nl_handle_destroy(drv->nl_handle);
1126         nl_cb_put(drv->nl_cb);
1127
1128         os_free(drv);
1129 }
1130
1131
1132 /**
1133  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
1134  * @eloop_ctx: Unused
1135  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
1136  *
1137  * This function can be used as registered timeout when starting a scan to
1138  * generate a scan completed event if the driver does not report this.
1139  */
1140 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1141 {
1142         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
1143         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
1144 }
1145
1146
1147 /**
1148  * wpa_driver_nl80211_scan - Request the driver to initiate scan
1149  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1150  * @params: Scan parameters
1151  * Returns: 0 on success, -1 on failure
1152  */
1153 static int wpa_driver_nl80211_scan(void *priv,
1154                                    struct wpa_driver_scan_params *params)
1155 {
1156         struct wpa_driver_nl80211_data *drv = priv;
1157         int ret = 0, timeout;
1158         struct nl_msg *msg, *ssids, *freqs;
1159         size_t i;
1160
1161         msg = nlmsg_alloc();
1162         ssids = nlmsg_alloc();
1163         freqs = nlmsg_alloc();
1164         if (!msg || !ssids || !freqs) {
1165                 nlmsg_free(msg);
1166                 nlmsg_free(ssids);
1167                 nlmsg_free(freqs);
1168                 return -1;
1169         }
1170
1171         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1172                     NL80211_CMD_TRIGGER_SCAN, 0);
1173
1174         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1175
1176         for (i = 0; i < params->num_ssids; i++) {
1177                 NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
1178                         params->ssids[i].ssid);
1179         }
1180         if (params->num_ssids)
1181                 nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids);
1182
1183         if (params->extra_ies) {
1184                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
1185                         params->extra_ies);
1186         }
1187
1188         if (params->freqs) {
1189                 for (i = 0; params->freqs[i]; i++)
1190                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
1191                 nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs);
1192         }
1193
1194         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1195         msg = NULL;
1196         if (ret) {
1197                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
1198                            "(%s)", ret, strerror(-ret));
1199                 goto nla_put_failure;
1200         }
1201
1202         /* Not all drivers generate "scan completed" wireless event, so try to
1203          * read results after a timeout. */
1204         timeout = 10;
1205         if (drv->scan_complete_events) {
1206                 /*
1207                  * The driver seems to deliver events to notify when scan is
1208                  * complete, so use longer timeout to avoid race conditions
1209                  * with scanning and following association request.
1210                  */
1211                 timeout = 30;
1212         }
1213         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
1214                    "seconds", ret, timeout);
1215         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1216         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
1217                                drv, drv->ctx);
1218
1219 nla_put_failure:
1220         nlmsg_free(ssids);
1221         nlmsg_free(msg);
1222         nlmsg_free(freqs);
1223         return ret;
1224 }
1225
1226
1227 static int bss_info_handler(struct nl_msg *msg, void *arg)
1228 {
1229         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1230         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1231         struct nlattr *bss[NL80211_BSS_MAX + 1];
1232         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
1233                 [NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
1234                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
1235                 [NL80211_BSS_TSF] = { .type = NLA_U64 },
1236                 [NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
1237                 [NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
1238                 [NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
1239                 [NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
1240                 [NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
1241         };
1242         struct wpa_scan_results *res = arg;
1243         struct wpa_scan_res **tmp;
1244         struct wpa_scan_res *r;
1245         const u8 *ie;
1246         size_t ie_len;
1247
1248         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1249                   genlmsg_attrlen(gnlh, 0), NULL);
1250         if (!tb[NL80211_ATTR_BSS])
1251                 return NL_SKIP;
1252         if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
1253                              bss_policy))
1254                 return NL_SKIP;
1255         if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
1256                 ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1257                 ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1258         } else {
1259                 ie = NULL;
1260                 ie_len = 0;
1261         }
1262
1263         r = os_zalloc(sizeof(*r) + ie_len);
1264         if (r == NULL)
1265                 return NL_SKIP;
1266         if (bss[NL80211_BSS_BSSID])
1267                 os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
1268                           ETH_ALEN);
1269         if (bss[NL80211_BSS_FREQUENCY])
1270                 r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
1271         if (bss[NL80211_BSS_BEACON_INTERVAL])
1272                 r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
1273         if (bss[NL80211_BSS_CAPABILITY])
1274                 r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
1275         r->flags |= WPA_SCAN_NOISE_INVALID;
1276         if (bss[NL80211_BSS_SIGNAL_MBM]) {
1277                 r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
1278                 r->level /= 100; /* mBm to dBm */
1279                 r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
1280         } else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
1281                 r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
1282                 r->flags |= WPA_SCAN_LEVEL_INVALID;
1283         } else
1284                 r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
1285         if (bss[NL80211_BSS_TSF])
1286                 r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
1287         r->ie_len = ie_len;
1288         if (ie)
1289                 os_memcpy(r + 1, ie, ie_len);
1290
1291         tmp = os_realloc(res->res,
1292                          (res->num + 1) * sizeof(struct wpa_scan_res *));
1293         if (tmp == NULL) {
1294                 os_free(r);
1295                 return NL_SKIP;
1296         }
1297         tmp[res->num++] = r;
1298         res->res = tmp;
1299
1300         return NL_SKIP;
1301 }
1302
1303
1304 /**
1305  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
1306  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1307  * Returns: Scan results on success, -1 on failure
1308  */
1309 static struct wpa_scan_results *
1310 wpa_driver_nl80211_get_scan_results(void *priv)
1311 {
1312         struct wpa_driver_nl80211_data *drv = priv;
1313         struct nl_msg *msg;
1314         struct wpa_scan_results *res;
1315         int ret;
1316
1317         res = os_zalloc(sizeof(*res));
1318         if (res == NULL)
1319                 return 0;
1320         msg = nlmsg_alloc();
1321         if (!msg)
1322                 goto nla_put_failure;
1323
1324         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, NLM_F_DUMP,
1325                     NL80211_CMD_GET_SCAN, 0);
1326         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1327
1328         ret = send_and_recv_msgs(drv, msg, bss_info_handler, res);
1329         msg = NULL;
1330         if (ret == 0) {
1331                 wpa_printf(MSG_DEBUG, "Received scan results (%lu BSSes)",
1332                            (unsigned long) res->num);
1333                 return res;
1334         }
1335         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
1336                    "(%s)", ret, strerror(-ret));
1337 nla_put_failure:
1338         nlmsg_free(msg);
1339         wpa_scan_results_free(res);
1340         return NULL;
1341 }
1342
1343
1344 static int wpa_driver_nl80211_set_key(void *priv, wpa_alg alg,
1345                                       const u8 *addr, int key_idx,
1346                                       int set_tx, const u8 *seq,
1347                                       size_t seq_len,
1348                                       const u8 *key, size_t key_len)
1349 {
1350         struct wpa_driver_nl80211_data *drv = priv;
1351         int err;
1352         struct nl_msg *msg;
1353
1354         wpa_printf(MSG_DEBUG, "%s: alg=%d addr=%p key_idx=%d set_tx=%d "
1355                    "seq_len=%lu key_len=%lu",
1356                    __func__, alg, addr, key_idx, set_tx,
1357                    (unsigned long) seq_len, (unsigned long) key_len);
1358
1359         msg = nlmsg_alloc();
1360         if (msg == NULL)
1361                 return -1;
1362
1363         if (alg == WPA_ALG_NONE) {
1364                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1365                             NL80211_CMD_DEL_KEY, 0);
1366         } else {
1367                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1368                             NL80211_CMD_NEW_KEY, 0);
1369                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
1370                 switch (alg) {
1371                 case WPA_ALG_WEP:
1372                         if (key_len == 5)
1373                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
1374                                             0x000FAC01);
1375                         else
1376                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
1377                                             0x000FAC05);
1378                         break;
1379                 case WPA_ALG_TKIP:
1380                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
1381                         break;
1382                 case WPA_ALG_CCMP:
1383                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
1384                         break;
1385 #ifdef CONFIG_IEEE80211W
1386                 case WPA_ALG_IGTK:
1387                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC06);
1388                         break;
1389 #endif /* CONFIG_IEEE80211W */
1390                 default:
1391                         nlmsg_free(msg);
1392                         return -1;
1393                 }
1394         }
1395
1396         if (addr && os_memcmp(addr, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
1397         {
1398                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
1399                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1400         }
1401         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
1402         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1403
1404         err = send_and_recv_msgs(drv, msg, NULL, NULL);
1405         if (err) {
1406                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d", err);
1407                 return -1;
1408         }
1409
1410         if (set_tx && alg != WPA_ALG_NONE) {
1411                 msg = nlmsg_alloc();
1412                 if (msg == NULL)
1413                         return -1;
1414
1415                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1416                             0, NL80211_CMD_SET_KEY, 0);
1417                 NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
1418                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1419                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
1420
1421                 err = send_and_recv_msgs(drv, msg, NULL, NULL);
1422                 if (err) {
1423                         wpa_printf(MSG_DEBUG, "nl80211: set default key "
1424                                    "failed; err=%d", err);
1425                         return -1;
1426                 }
1427         }
1428
1429         return 0;
1430
1431 nla_put_failure:
1432         return -ENOBUFS;
1433 }
1434
1435
1436 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
1437                                    const u8 *addr, int cmd, u16 reason_code)
1438 {
1439         int ret = -1;
1440         struct nl_msg *msg;
1441
1442         msg = nlmsg_alloc();
1443         if (!msg)
1444                 return -1;
1445
1446         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0, cmd, 0);
1447
1448         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1449         NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason_code);
1450         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1451
1452         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1453         msg = NULL;
1454         if (ret) {
1455                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
1456                            "(%s)", ret, strerror(-ret));
1457                 goto nla_put_failure;
1458         }
1459         ret = 0;
1460
1461 nla_put_failure:
1462         nlmsg_free(msg);
1463         return ret;
1464 }
1465
1466
1467 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
1468                                              int reason_code)
1469 {
1470         struct wpa_driver_nl80211_data *drv = priv;
1471         wpa_printf(MSG_DEBUG, "%s", __func__);
1472         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DEAUTHENTICATE,
1473                                        reason_code);
1474 }
1475
1476
1477 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
1478                                            int reason_code)
1479 {
1480         struct wpa_driver_nl80211_data *drv = priv;
1481         wpa_printf(MSG_DEBUG, "%s", __func__);
1482         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISASSOCIATE,
1483                                        reason_code);
1484 }
1485
1486
1487 static int wpa_driver_nl80211_authenticate(
1488         void *priv, struct wpa_driver_auth_params *params)
1489 {
1490         struct wpa_driver_nl80211_data *drv = priv;
1491         int ret = -1;
1492         struct nl_msg *msg;
1493         enum nl80211_auth_type type;
1494
1495         drv->associated = 0;
1496
1497         msg = nlmsg_alloc();
1498         if (!msg)
1499                 return -1;
1500
1501         wpa_printf(MSG_DEBUG, "nl80211: Authenticate (ifindex=%d)",
1502                    drv->ifindex);
1503         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1504                     NL80211_CMD_AUTHENTICATE, 0);
1505
1506         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1507         if (params->bssid) {
1508                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
1509                            MAC2STR(params->bssid));
1510                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
1511         }
1512         if (params->freq) {
1513                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
1514                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
1515         }
1516         if (params->ssid) {
1517                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
1518                                   params->ssid, params->ssid_len);
1519                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
1520                         params->ssid);
1521         }
1522         wpa_hexdump(MSG_DEBUG, "  * IEs", params->ie, params->ie_len);
1523         if (params->ie)
1524                 NLA_PUT(msg, NL80211_ATTR_IE, params->ie_len, params->ie);
1525         /*
1526          * TODO: if multiple auth_alg options enabled, try them one by one if
1527          * the AP rejects authentication due to unknown auth alg
1528          */
1529         if (params->auth_alg & AUTH_ALG_OPEN_SYSTEM)
1530                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
1531         else if (params->auth_alg & AUTH_ALG_SHARED_KEY)
1532                 type = NL80211_AUTHTYPE_SHARED_KEY;
1533         else if (params->auth_alg & AUTH_ALG_LEAP)
1534                 type = NL80211_AUTHTYPE_NETWORK_EAP;
1535         else if (params->auth_alg & AUTH_ALG_FT)
1536                 type = NL80211_AUTHTYPE_FT;
1537         else
1538                 goto nla_put_failure;
1539         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
1540         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
1541
1542         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1543         msg = NULL;
1544         if (ret) {
1545                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
1546                            "(%s)", ret, strerror(-ret));
1547                 goto nla_put_failure;
1548         }
1549         ret = 0;
1550         wpa_printf(MSG_DEBUG, "nl80211: Authentication request send "
1551                    "successfully");
1552
1553 nla_put_failure:
1554         nlmsg_free(msg);
1555         return ret;
1556 }
1557
1558
1559 #ifdef CONFIG_AP
1560 static int wpa_driver_nl80211_set_beacon(void *priv,
1561                                          const u8 *head, size_t head_len,
1562                                          const u8 *tail, size_t tail_len,
1563                                          int dtim_period)
1564 {
1565         struct wpa_driver_nl80211_data *drv = priv;
1566         struct nl_msg *msg;
1567         u8 cmd = NL80211_CMD_NEW_BEACON;
1568         int ret;
1569
1570         msg = nlmsg_alloc();
1571         if (!msg)
1572                 return -ENOMEM;
1573
1574         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (beacon_set=%d)",
1575                    drv->beacon_set);
1576         if (drv->beacon_set)
1577                 cmd = NL80211_CMD_SET_BEACON;
1578
1579         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1580                     0, cmd, 0);
1581         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
1582         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
1583         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1584         if (!drv->beacon_int)
1585                 drv->beacon_int = 100;
1586         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
1587         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, dtim_period);
1588
1589         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1590         if (ret) {
1591                 wpa_printf(MSG_DEBUG, "nl80211: Beacon set failed: %d (%s)",
1592                            ret, strerror(-ret));
1593         } else
1594                 drv->beacon_set = 1;
1595         return ret;
1596  nla_put_failure:
1597         return -ENOBUFS;
1598 }
1599
1600
1601 static int wpa_driver_nl80211_set_beacon_int(void *priv, int value)
1602 {
1603         struct wpa_driver_nl80211_data *drv = priv;
1604         struct nl_msg *msg;
1605
1606         drv->beacon_int = value;
1607
1608         if (!drv->beacon_set)
1609                 return 0;
1610
1611         msg = nlmsg_alloc();
1612         if (!msg)
1613                 return -ENOMEM;
1614
1615         wpa_printf(MSG_DEBUG, "nl80211: Set beacon interval %d "
1616                    "(beacon_set=%d)", value, drv->beacon_set);
1617         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1618                     0, NL80211_CMD_SET_BEACON, 0);
1619         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1620
1621         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
1622
1623         return send_and_recv_msgs(drv, msg, NULL, NULL);
1624  nla_put_failure:
1625         return -ENOBUFS;
1626 }
1627
1628
1629 static int wpa_driver_nl80211_set_freq2(
1630         struct wpa_driver_nl80211_data *drv,
1631         struct wpa_driver_associate_params *params)
1632 {
1633         struct nl_msg *msg;
1634         int ret;
1635
1636         msg = nlmsg_alloc();
1637         if (!msg)
1638                 return -1;
1639
1640         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1641                     NL80211_CMD_SET_WIPHY, 0);
1642
1643         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1644
1645         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
1646
1647         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1648         if (ret == 0)
1649                 return 0;
1650         wpa_printf(MSG_DEBUG, "nl80211: MLME Failed to set channel (freq=%d): "
1651                    "%d (%s)", params->freq, ret, strerror(-ret));
1652 nla_put_failure:
1653         return -1;
1654 }
1655
1656
1657 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
1658                                  int ifidx)
1659 {
1660         struct nl_msg *msg;
1661
1662         msg = nlmsg_alloc();
1663         if (!msg)
1664                 goto nla_put_failure;
1665
1666         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1667                     0, NL80211_CMD_DEL_INTERFACE, 0);
1668         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
1669
1670         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
1671                 return;
1672  nla_put_failure:
1673         wpa_printf(MSG_ERROR, "Failed to remove interface (ifidx=%d).\n",
1674                    ifidx);
1675 }
1676
1677
1678 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
1679                                 const char *ifname, enum nl80211_iftype iftype)
1680 {
1681         struct nl_msg *msg, *flags = NULL;
1682         int ifidx;
1683         int ret = -ENOBUFS;
1684
1685         msg = nlmsg_alloc();
1686         if (!msg)
1687                 return -1;
1688
1689         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1690                     0, NL80211_CMD_NEW_INTERFACE, 0);
1691         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1692         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
1693         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
1694
1695         if (iftype == NL80211_IFTYPE_MONITOR) {
1696                 int err;
1697
1698                 flags = nlmsg_alloc();
1699                 if (!flags)
1700                         goto nla_put_failure;
1701
1702                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1703
1704                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1705
1706                 nlmsg_free(flags);
1707
1708                 if (err)
1709                         goto nla_put_failure;
1710         }
1711
1712         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1713         if (ret) {
1714  nla_put_failure:
1715                 wpa_printf(MSG_ERROR, "Failed to create interface %s.",
1716                            ifname);
1717                 return ret;
1718         }
1719
1720         ifidx = if_nametoindex(ifname);
1721
1722         if (ifidx <= 0)
1723                 return -1;
1724
1725         return ifidx;
1726 }
1727
1728
1729 enum ieee80211_msg_type {
1730         ieee80211_msg_normal = 0,
1731         ieee80211_msg_tx_callback_ack = 1,
1732         ieee80211_msg_tx_callback_fail = 2,
1733 };
1734
1735
1736 void ap_tx_status(void *ctx, const u8 *addr,
1737                   const u8 *buf, size_t len, int ack);
1738 void ap_rx_from_unknown_sta(void *ctx, const u8 *addr);
1739 void ap_mgmt_rx(void *ctx, u8 *buf, size_t len, u16 stype,
1740                 struct hostapd_frame_info *fi);
1741 void ap_mgmt_tx_cb(void *ctx, u8 *buf, size_t len, u16 stype, int ok);
1742
1743
1744 static void handle_tx_callback(void *ctx, u8 *buf, size_t len, int ok)
1745 {
1746         struct ieee80211_hdr *hdr;
1747         u16 fc, type, stype;
1748
1749         hdr = (struct ieee80211_hdr *) buf;
1750         fc = le_to_host16(hdr->frame_control);
1751
1752         type = WLAN_FC_GET_TYPE(fc);
1753         stype = WLAN_FC_GET_STYPE(fc);
1754
1755         switch (type) {
1756         case WLAN_FC_TYPE_MGMT:
1757                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
1758                            ok ? "ACK" : "fail");
1759                 ap_mgmt_tx_cb(ctx, buf, len, stype, ok);
1760                 break;
1761         case WLAN_FC_TYPE_CTRL:
1762                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
1763                            ok ? "ACK" : "fail");
1764                 break;
1765         case WLAN_FC_TYPE_DATA:
1766                 ap_tx_status(ctx, hdr->addr1, buf, len, ok);
1767                 break;
1768         default:
1769                 wpa_printf(MSG_DEBUG, "unknown TX callback frame type %d",
1770                            type);
1771                 break;
1772         }
1773 }
1774
1775
1776 static void handle_frame(struct wpa_driver_nl80211_data *drv,
1777                          u8 *buf, size_t len,
1778                          struct hostapd_frame_info *hfi,
1779                          enum ieee80211_msg_type msg_type)
1780 {
1781         struct ieee80211_hdr *hdr;
1782         u16 fc, type, stype;
1783         size_t data_len = len;
1784
1785         /*
1786          * PS-Poll frames are 16 bytes. All other frames are
1787          * 24 bytes or longer.
1788          */
1789         if (len < 16)
1790                 return;
1791
1792         hdr = (struct ieee80211_hdr *) buf;
1793         fc = le_to_host16(hdr->frame_control);
1794
1795         type = WLAN_FC_GET_TYPE(fc);
1796         stype = WLAN_FC_GET_STYPE(fc);
1797
1798         switch (type) {
1799         case WLAN_FC_TYPE_DATA:
1800                 if (len < 24)
1801                         return;
1802                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
1803                 case WLAN_FC_TODS:
1804                         break;
1805                 case WLAN_FC_FROMDS:
1806                         break;
1807                 default:
1808                         /* discard */
1809                         return;
1810                 }
1811                 break;
1812         case WLAN_FC_TYPE_CTRL:
1813                 /* discard non-ps-poll frames */
1814                 if (stype != WLAN_FC_STYPE_PSPOLL)
1815                         return;
1816                 break;
1817         case WLAN_FC_TYPE_MGMT:
1818                 break;
1819         default:
1820                 /* discard */
1821                 return;
1822         }
1823
1824         switch (msg_type) {
1825         case ieee80211_msg_normal:
1826                 /* continue processing */
1827                 break;
1828         case ieee80211_msg_tx_callback_ack:
1829                 handle_tx_callback(drv->ctx, buf, data_len, 1);
1830                 return;
1831         case ieee80211_msg_tx_callback_fail:
1832                 handle_tx_callback(drv->ctx, buf, data_len, 0);
1833                 return;
1834         }
1835
1836         switch (type) {
1837         case WLAN_FC_TYPE_MGMT:
1838                 if (stype != WLAN_FC_STYPE_BEACON &&
1839                     stype != WLAN_FC_STYPE_PROBE_REQ)
1840                         wpa_printf(MSG_MSGDUMP, "MGMT");
1841                 ap_mgmt_rx(drv->ctx, buf, data_len, stype, hfi);
1842                 break;
1843         case WLAN_FC_TYPE_CTRL:
1844                 /* can only get here with PS-Poll frames */
1845                 wpa_printf(MSG_DEBUG, "CTRL");
1846                 ap_rx_from_unknown_sta(drv->ctx, hdr->addr2);
1847                 break;
1848         case WLAN_FC_TYPE_DATA:
1849                 ap_rx_from_unknown_sta(drv->ctx, hdr->addr2);
1850                 break;
1851         }
1852 }
1853
1854
1855 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1856 {
1857         struct wpa_driver_nl80211_data *drv = eloop_ctx;
1858         int len;
1859         unsigned char buf[3000];
1860         struct ieee80211_radiotap_iterator iter;
1861         int ret;
1862         struct hostapd_frame_info hfi;
1863         int injected = 0, failed = 0, msg_type, rxflags = 0;
1864
1865         len = recv(sock, buf, sizeof(buf), 0);
1866         if (len < 0) {
1867                 perror("recv");
1868                 return;
1869         }
1870
1871         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
1872                 printf("received invalid radiotap frame\n");
1873                 return;
1874         }
1875
1876         memset(&hfi, 0, sizeof(hfi));
1877
1878         while (1) {
1879                 ret = ieee80211_radiotap_iterator_next(&iter);
1880                 if (ret == -ENOENT)
1881                         break;
1882                 if (ret) {
1883                         printf("received invalid radiotap frame (%d)\n", ret);
1884                         return;
1885                 }
1886                 switch (iter.this_arg_index) {
1887                 case IEEE80211_RADIOTAP_FLAGS:
1888                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1889                                 len -= 4;
1890                         break;
1891                 case IEEE80211_RADIOTAP_RX_FLAGS:
1892                         rxflags = 1;
1893                         break;
1894                 case IEEE80211_RADIOTAP_TX_FLAGS:
1895                         injected = 1;
1896                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
1897                                         IEEE80211_RADIOTAP_F_TX_FAIL;
1898                         break;
1899                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1900                         break;
1901                 case IEEE80211_RADIOTAP_CHANNEL:
1902                         /* TODO convert from freq/flags to channel number
1903                         hfi.channel = XXX;
1904                         hfi.phytype = XXX;
1905                          */
1906                         break;
1907                 case IEEE80211_RADIOTAP_RATE:
1908                         hfi.datarate = *iter.this_arg * 5;
1909                         break;
1910                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1911                         hfi.ssi_signal = *iter.this_arg;
1912                         break;
1913                 }
1914         }
1915
1916         if (rxflags && injected)
1917                 return;
1918
1919         if (!injected)
1920                 msg_type = ieee80211_msg_normal;
1921         else if (failed)
1922                 msg_type = ieee80211_msg_tx_callback_fail;
1923         else
1924                 msg_type = ieee80211_msg_tx_callback_ack;
1925
1926         handle_frame(drv, buf + iter.max_length,
1927                      len - iter.max_length, &hfi, msg_type);
1928 }
1929
1930
1931 /*
1932  * we post-process the filter code later and rewrite
1933  * this to the offset to the last instruction
1934  */
1935 #define PASS    0xFF
1936 #define FAIL    0xFE
1937
1938 static struct sock_filter msock_filter_insns[] = {
1939         /*
1940          * do a little-endian load of the radiotap length field
1941          */
1942         /* load lower byte into A */
1943         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
1944         /* put it into X (== index register) */
1945         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1946         /* load upper byte into A */
1947         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
1948         /* left-shift it by 8 */
1949         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
1950         /* or with X */
1951         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
1952         /* put result into X */
1953         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1954
1955         /*
1956          * Allow management frames through, this also gives us those
1957          * management frames that we sent ourselves with status
1958          */
1959         /* load the lower byte of the IEEE 802.11 frame control field */
1960         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
1961         /* mask off frame type and version */
1962         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
1963         /* accept frame if it's both 0, fall through otherwise */
1964         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
1965
1966         /*
1967          * TODO: add a bit to radiotap RX flags that indicates
1968          * that the sending station is not associated, then
1969          * add a filter here that filters on our DA and that flag
1970          * to allow us to deauth frames to that bad station.
1971          *
1972          * Not a regression -- we didn't do it before either.
1973          */
1974
1975 #if 0
1976         /*
1977          * drop non-data frames, WDS frames
1978          */
1979         /* load the lower byte of the frame control field */
1980         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1981         /* mask off QoS bit */
1982         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
1983         /* drop non-data frames */
1984         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
1985         /* load the upper byte of the frame control field */
1986         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1987         /* mask off toDS/fromDS */
1988         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
1989         /* drop WDS frames */
1990         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, FAIL, 0),
1991 #endif
1992
1993         /*
1994          * add header length to index
1995          */
1996         /* load the lower byte of the frame control field */
1997         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1998         /* mask off QoS bit */
1999         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
2000         /* right shift it by 6 to give 0 or 2 */
2001         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
2002         /* add data frame header length */
2003         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
2004         /* add index, was start of 802.11 header */
2005         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
2006         /* move to index, now start of LL header */
2007         BPF_STMT(BPF_MISC | BPF_TAX, 0),
2008
2009         /*
2010          * Accept empty data frames, we use those for
2011          * polling activity.
2012          */
2013         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
2014         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
2015
2016         /*
2017          * Accept EAPOL frames
2018          */
2019         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
2020         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
2021         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
2022         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
2023
2024         /* keep these last two statements or change the code below */
2025         /* return 0 == "DROP" */
2026         BPF_STMT(BPF_RET | BPF_K, 0),
2027         /* return ~0 == "keep all" */
2028         BPF_STMT(BPF_RET | BPF_K, ~0),
2029 };
2030
2031 static struct sock_fprog msock_filter = {
2032         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
2033         .filter = msock_filter_insns,
2034 };
2035
2036
2037 static int add_monitor_filter(int s)
2038 {
2039         int idx;
2040
2041         /* rewrite all PASS/FAIL jump offsets */
2042         for (idx = 0; idx < msock_filter.len; idx++) {
2043                 struct sock_filter *insn = &msock_filter_insns[idx];
2044
2045                 if (BPF_CLASS(insn->code) == BPF_JMP) {
2046                         if (insn->code == (BPF_JMP|BPF_JA)) {
2047                                 if (insn->k == PASS)
2048                                         insn->k = msock_filter.len - idx - 2;
2049                                 else if (insn->k == FAIL)
2050                                         insn->k = msock_filter.len - idx - 3;
2051                         }
2052
2053                         if (insn->jt == PASS)
2054                                 insn->jt = msock_filter.len - idx - 2;
2055                         else if (insn->jt == FAIL)
2056                                 insn->jt = msock_filter.len - idx - 3;
2057
2058                         if (insn->jf == PASS)
2059                                 insn->jf = msock_filter.len - idx - 2;
2060                         else if (insn->jf == FAIL)
2061                                 insn->jf = msock_filter.len - idx - 3;
2062                 }
2063         }
2064
2065         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
2066                        &msock_filter, sizeof(msock_filter))) {
2067                 perror("SO_ATTACH_FILTER");
2068                 return -1;
2069         }
2070
2071         return 0;
2072 }
2073
2074
2075 static int
2076 nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
2077 {
2078         char buf[IFNAMSIZ];
2079         struct sockaddr_ll ll;
2080         int optval;
2081         socklen_t optlen;
2082         int flags;
2083
2084         snprintf(buf, IFNAMSIZ, "mon.%s", drv->ifname);
2085         buf[IFNAMSIZ - 1] = '\0';
2086
2087         drv->monitor_ifidx =
2088                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR);
2089
2090         if (drv->monitor_ifidx < 0)
2091                 return -1;
2092
2093         if (wpa_driver_nl80211_get_ifflags_ifname(drv, buf, &flags) != 0) {
2094                 wpa_printf(MSG_ERROR, "Could not get interface '%s' flags",
2095                            buf);
2096                 goto error;
2097         }
2098         if (!(flags & IFF_UP)) {
2099                 if (wpa_driver_nl80211_set_ifflags_ifname(drv, buf,
2100                                                           flags | IFF_UP) != 0)
2101                 {
2102                         wpa_printf(MSG_ERROR, "Could not set interface '%s' "
2103                                    "UP", buf);
2104                         goto error;
2105                 }
2106         }
2107
2108         memset(&ll, 0, sizeof(ll));
2109         ll.sll_family = AF_PACKET;
2110         ll.sll_ifindex = drv->monitor_ifidx;
2111         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
2112         if (drv->monitor_sock < 0) {
2113                 perror("socket[PF_PACKET,SOCK_RAW]");
2114                 goto error;
2115         }
2116
2117         if (add_monitor_filter(drv->monitor_sock)) {
2118                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
2119                            "interface; do filtering in user space");
2120                 /* This works, but will cost in performance. */
2121         }
2122
2123         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
2124                  sizeof(ll)) < 0) {
2125                 perror("monitor socket bind");
2126                 goto error;
2127         }
2128
2129         optlen = sizeof(optval);
2130         optval = 20;
2131         if (setsockopt
2132             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
2133                 perror("Failed to set socket priority");
2134                 goto error;
2135         }
2136
2137         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
2138                                      drv, NULL)) {
2139                 printf("Could not register monitor read socket\n");
2140                 goto error;
2141         }
2142
2143         return 0;
2144  error:
2145         nl80211_remove_iface(drv, drv->monitor_ifidx);
2146         return -1;
2147 }
2148
2149
2150 static int wpa_driver_nl80211_ap(struct wpa_driver_nl80211_data *drv,
2151                                  struct wpa_driver_associate_params *params)
2152 {
2153         if (drv->monitor_ifidx < 0 &&
2154             nl80211_create_monitor_interface(drv))
2155                 return -1;
2156
2157         if (wpa_driver_nl80211_set_mode(drv, params->mode) ||
2158             wpa_driver_nl80211_set_freq2(drv, params)) {
2159                 nl80211_remove_iface(drv, drv->monitor_ifidx);
2160                 drv->monitor_ifidx = -1;
2161                 return -1;
2162         }
2163
2164         /* TODO: setup monitor interface (and add code somewhere to remove this
2165          * when AP mode is stopped; associate with mode != 2 or drv_deinit) */
2166
2167         return 0;
2168 }
2169 #endif /* CONFIG_AP */
2170
2171
2172 static int wpa_driver_nl80211_associate(
2173         void *priv, struct wpa_driver_associate_params *params)
2174 {
2175         struct wpa_driver_nl80211_data *drv = priv;
2176         int ret = -1;
2177         struct nl_msg *msg;
2178
2179 #ifdef CONFIG_AP
2180         if (params->mode == 2)
2181                 return wpa_driver_nl80211_ap(drv, params);
2182 #endif /* CONFIG_AP */
2183
2184         wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED,
2185                                           params->drop_unencrypted);
2186
2187         drv->associated = 0;
2188
2189         msg = nlmsg_alloc();
2190         if (!msg)
2191                 return -1;
2192
2193         wpa_printf(MSG_DEBUG, "nl80211: Associate (ifindex=%d)",
2194                    drv->ifindex);
2195         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2196                     NL80211_CMD_ASSOCIATE, 0);
2197
2198         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2199         if (params->bssid) {
2200                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
2201                            MAC2STR(params->bssid));
2202                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
2203         }
2204         if (params->freq) {
2205                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
2206                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
2207         }
2208         if (params->ssid) {
2209                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
2210                                   params->ssid, params->ssid_len);
2211                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
2212                         params->ssid);
2213                 if (params->ssid_len > sizeof(drv->ssid))
2214                         goto nla_put_failure;
2215                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
2216                 drv->ssid_len = params->ssid_len;
2217         }
2218         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
2219         if (params->wpa_ie)
2220                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
2221                         params->wpa_ie);
2222
2223         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2224         msg = NULL;
2225         if (ret) {
2226                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
2227                            "(%s)", ret, strerror(-ret));
2228                 goto nla_put_failure;
2229         }
2230         ret = 0;
2231         wpa_printf(MSG_DEBUG, "nl80211: Association request send "
2232                    "successfully");
2233
2234 nla_put_failure:
2235         nlmsg_free(msg);
2236         return ret;
2237 }
2238
2239
2240 /**
2241  * wpa_driver_nl80211_set_mode - Set wireless mode (infra/adhoc)
2242  * @drv: Pointer to private driver data from wpa_driver_nl80211_init()
2243  * @mode: 0 = infra/BSS (associate with an AP), 1 = adhoc/IBSS
2244  * Returns: 0 on success, -1 on failure
2245  */
2246 static int wpa_driver_nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
2247                                        int mode)
2248 {
2249         int ret = -1, flags;
2250         struct nl_msg *msg;
2251         int nlmode;
2252
2253         switch (mode) {
2254         case 0:
2255                 nlmode = NL80211_IFTYPE_STATION;
2256                 break;
2257         case 1:
2258                 nlmode = NL80211_IFTYPE_ADHOC;
2259                 break;
2260         case 2:
2261                 nlmode = NL80211_IFTYPE_AP;
2262                 break;
2263         default:
2264                 return -1;
2265         }
2266
2267         msg = nlmsg_alloc();
2268         if (!msg)
2269                 return -1;
2270
2271         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2272                     0, NL80211_CMD_SET_INTERFACE, 0);
2273         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2274         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, nlmode);
2275
2276         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2277         if (!ret)
2278                 return 0;
2279         else
2280                 goto try_again;
2281
2282 nla_put_failure:
2283         wpa_printf(MSG_ERROR, "nl80211: Failed to set interface mode: %d (%s)",
2284                    ret, strerror(-ret));
2285         return -1;
2286
2287 try_again:
2288         /* mac80211 doesn't allow mode changes while the device is up, so
2289          * take the device down, try to set the mode again, and bring the
2290          * device back up.
2291          */
2292         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0) {
2293                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
2294
2295                 /* Try to set the mode again while the interface is down */
2296                 msg = nlmsg_alloc();
2297                 if (!msg)
2298                         return -1;
2299
2300                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2301                             0, NL80211_CMD_SET_INTERFACE, 0);
2302                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2303                 NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, nlmode);
2304                 ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2305                 if (ret) {
2306                         wpa_printf(MSG_ERROR, "Failed to set interface %s "
2307                                    "mode(try_again): %d (%s)",
2308                                    drv->ifname, ret, strerror(-ret));
2309                 }
2310
2311                 /* Ignore return value of get_ifflags to ensure that the device
2312                  * is always up like it was before this function was called.
2313                  */
2314                 (void) wpa_driver_nl80211_get_ifflags(drv, &flags);
2315                 (void) wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP);
2316         }
2317
2318         return ret;
2319 }
2320
2321
2322 static int wpa_driver_nl80211_get_capa(void *priv,
2323                                        struct wpa_driver_capa *capa)
2324 {
2325         struct wpa_driver_nl80211_data *drv = priv;
2326         if (!drv->has_capability)
2327                 return -1;
2328         os_memcpy(capa, &drv->capa, sizeof(*capa));
2329         return 0;
2330 }
2331
2332
2333 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
2334 {
2335         struct wpa_driver_nl80211_data *drv = priv;
2336
2337         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
2338                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
2339         drv->operstate = state;
2340         return wpa_driver_nl80211_send_oper_ifla(
2341                 drv, -1, state ? IF_OPER_UP : IF_OPER_DORMANT);
2342 }
2343
2344
2345 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
2346         .name = "nl80211",
2347         .desc = "Linux nl80211/cfg80211",
2348         .get_bssid = wpa_driver_nl80211_get_bssid,
2349         .get_ssid = wpa_driver_nl80211_get_ssid,
2350         .set_key = wpa_driver_nl80211_set_key,
2351         .scan2 = wpa_driver_nl80211_scan,
2352         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
2353         .deauthenticate = wpa_driver_nl80211_deauthenticate,
2354         .disassociate = wpa_driver_nl80211_disassociate,
2355         .authenticate = wpa_driver_nl80211_authenticate,
2356         .associate = wpa_driver_nl80211_associate,
2357         .init = wpa_driver_nl80211_init,
2358         .deinit = wpa_driver_nl80211_deinit,
2359         .get_capa = wpa_driver_nl80211_get_capa,
2360         .set_operstate = wpa_driver_nl80211_set_operstate,
2361         .set_country = wpa_driver_nl80211_set_country,
2362 #ifdef CONFIG_AP
2363         .set_beacon = wpa_driver_nl80211_set_beacon,
2364         .set_beacon_int = wpa_driver_nl80211_set_beacon_int,
2365 #endif /* CONFIG_AP */
2366 };