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