nl80211: Check nla_put_nested() return value and handle errors
[mech_eap.git] / src / drivers / driver_nl80211.c
1 /*
2  * Driver interaction with Linux nl80211/cfg80211
3  * Copyright (c) 2002-2012, 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-2010, Atheros Communications
8  *
9  * This software may be distributed under the terms of the BSD license.
10  * See README for more details.
11  */
12
13 #include "includes.h"
14 #include <sys/ioctl.h>
15 #include <sys/types.h>
16 #include <sys/stat.h>
17 #include <fcntl.h>
18 #include <net/if.h>
19 #include <netlink/genl/genl.h>
20 #include <netlink/genl/family.h>
21 #include <netlink/genl/ctrl.h>
22 #include <linux/rtnetlink.h>
23 #include <netpacket/packet.h>
24 #include <linux/filter.h>
25 #include <linux/errqueue.h>
26 #include "nl80211_copy.h"
27
28 #include "common.h"
29 #include "eloop.h"
30 #include "utils/list.h"
31 #include "common/ieee802_11_defs.h"
32 #include "common/ieee802_11_common.h"
33 #include "l2_packet/l2_packet.h"
34 #include "netlink.h"
35 #include "linux_ioctl.h"
36 #include "radiotap.h"
37 #include "radiotap_iter.h"
38 #include "rfkill.h"
39 #include "driver.h"
40
41 #ifndef SO_WIFI_STATUS
42 # if defined(__sparc__)
43 #  define SO_WIFI_STATUS        0x0025
44 # elif defined(__parisc__)
45 #  define SO_WIFI_STATUS        0x4022
46 # else
47 #  define SO_WIFI_STATUS        41
48 # endif
49
50 # define SCM_WIFI_STATUS        SO_WIFI_STATUS
51 #endif
52
53 #ifndef SO_EE_ORIGIN_TXSTATUS
54 #define SO_EE_ORIGIN_TXSTATUS   4
55 #endif
56
57 #ifndef PACKET_TX_TIMESTAMP
58 #define PACKET_TX_TIMESTAMP     16
59 #endif
60
61 #ifdef ANDROID
62 #include "android_drv.h"
63 #endif /* ANDROID */
64 #ifdef CONFIG_LIBNL20
65 /* libnl 2.0 compatibility code */
66 #define nl_handle nl_sock
67 #define nl80211_handle_alloc nl_socket_alloc_cb
68 #define nl80211_handle_destroy nl_socket_free
69 #else
70 /*
71  * libnl 1.1 has a bug, it tries to allocate socket numbers densely
72  * but when you free a socket again it will mess up its bitmap and
73  * and use the wrong number the next time it needs a socket ID.
74  * Therefore, we wrap the handle alloc/destroy and add our own pid
75  * accounting.
76  */
77 static uint32_t port_bitmap[32] = { 0 };
78
79 static struct nl_handle *nl80211_handle_alloc(void *cb)
80 {
81         struct nl_handle *handle;
82         uint32_t pid = getpid() & 0x3FFFFF;
83         int i;
84
85         handle = nl_handle_alloc_cb(cb);
86
87         for (i = 0; i < 1024; i++) {
88                 if (port_bitmap[i / 32] & (1 << (i % 32)))
89                         continue;
90                 port_bitmap[i / 32] |= 1 << (i % 32);
91                 pid += i << 22;
92                 break;
93         }
94
95         nl_socket_set_local_port(handle, pid);
96
97         return handle;
98 }
99
100 static void nl80211_handle_destroy(struct nl_handle *handle)
101 {
102         uint32_t port = nl_socket_get_local_port(handle);
103
104         port >>= 22;
105         port_bitmap[port / 32] &= ~(1 << (port % 32));
106
107         nl_handle_destroy(handle);
108 }
109 #endif /* CONFIG_LIBNL20 */
110
111
112 static struct nl_handle * nl_create_handle(struct nl_cb *cb, const char *dbg)
113 {
114         struct nl_handle *handle;
115
116         handle = nl80211_handle_alloc(cb);
117         if (handle == NULL) {
118                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
119                            "callbacks (%s)", dbg);
120                 return NULL;
121         }
122
123         if (genl_connect(handle)) {
124                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
125                            "netlink (%s)", dbg);
126                 nl80211_handle_destroy(handle);
127                 return NULL;
128         }
129
130         return handle;
131 }
132
133
134 static void nl_destroy_handles(struct nl_handle **handle)
135 {
136         if (*handle == NULL)
137                 return;
138         nl80211_handle_destroy(*handle);
139         *handle = NULL;
140 }
141
142
143 #ifndef IFF_LOWER_UP
144 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
145 #endif
146 #ifndef IFF_DORMANT
147 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
148 #endif
149
150 #ifndef IF_OPER_DORMANT
151 #define IF_OPER_DORMANT 5
152 #endif
153 #ifndef IF_OPER_UP
154 #define IF_OPER_UP 6
155 #endif
156
157 struct nl80211_global {
158         struct dl_list interfaces;
159         int if_add_ifindex;
160         struct netlink_data *netlink;
161         struct nl_cb *nl_cb;
162         struct nl_handle *nl;
163         int nl80211_id;
164         int ioctl_sock; /* socket for ioctl() use */
165
166         struct nl_handle *nl_event;
167 };
168
169 struct nl80211_wiphy_data {
170         struct dl_list list;
171         struct dl_list bsss;
172         struct dl_list drvs;
173
174         struct nl_handle *nl_beacons;
175         struct nl_cb *nl_cb;
176
177         int wiphy_idx;
178 };
179
180 static void nl80211_global_deinit(void *priv);
181 static void wpa_driver_nl80211_deinit(void *priv);
182
183 struct i802_bss {
184         struct wpa_driver_nl80211_data *drv;
185         struct i802_bss *next;
186         int ifindex;
187         char ifname[IFNAMSIZ + 1];
188         char brname[IFNAMSIZ];
189         unsigned int beacon_set:1;
190         unsigned int added_if_into_bridge:1;
191         unsigned int added_bridge:1;
192         unsigned int in_deinit:1;
193
194         u8 addr[ETH_ALEN];
195
196         int freq;
197
198         struct nl_handle *nl_preq, *nl_mgmt;
199         struct nl_cb *nl_cb;
200
201         struct nl80211_wiphy_data *wiphy_data;
202         struct dl_list wiphy_list;
203 };
204
205 struct wpa_driver_nl80211_data {
206         struct nl80211_global *global;
207         struct dl_list list;
208         struct dl_list wiphy_list;
209         char phyname[32];
210         void *ctx;
211         int ifindex;
212         int if_removed;
213         int if_disabled;
214         int ignore_if_down_event;
215         struct rfkill_data *rfkill;
216         struct wpa_driver_capa capa;
217         int has_capability;
218
219         int operstate;
220
221         int scan_complete_events;
222
223         struct nl_cb *nl_cb;
224
225         u8 auth_bssid[ETH_ALEN];
226         u8 bssid[ETH_ALEN];
227         int associated;
228         u8 ssid[32];
229         size_t ssid_len;
230         enum nl80211_iftype nlmode;
231         enum nl80211_iftype ap_scan_as_station;
232         unsigned int assoc_freq;
233
234         int monitor_sock;
235         int monitor_ifidx;
236         int monitor_refcount;
237
238         unsigned int disabled_11b_rates:1;
239         unsigned int pending_remain_on_chan:1;
240         unsigned int in_interface_list:1;
241         unsigned int device_ap_sme:1;
242         unsigned int poll_command_supported:1;
243         unsigned int data_tx_status:1;
244         unsigned int scan_for_auth:1;
245         unsigned int retry_auth:1;
246         unsigned int use_monitor:1;
247
248         u64 remain_on_chan_cookie;
249         u64 send_action_cookie;
250
251         unsigned int last_mgmt_freq;
252
253         struct wpa_driver_scan_filter *filter_ssids;
254         size_t num_filter_ssids;
255
256         struct i802_bss first_bss;
257
258         int eapol_tx_sock;
259
260 #ifdef HOSTAPD
261         int eapol_sock; /* socket for EAPOL frames */
262
263         int default_if_indices[16];
264         int *if_indices;
265         int num_if_indices;
266
267         int last_freq;
268         int last_freq_ht;
269 #endif /* HOSTAPD */
270
271         /* From failed authentication command */
272         int auth_freq;
273         u8 auth_bssid_[ETH_ALEN];
274         u8 auth_ssid[32];
275         size_t auth_ssid_len;
276         int auth_alg;
277         u8 *auth_ie;
278         size_t auth_ie_len;
279         u8 auth_wep_key[4][16];
280         size_t auth_wep_key_len[4];
281         int auth_wep_tx_keyidx;
282         int auth_local_state_change;
283         int auth_p2p;
284 };
285
286
287 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
288                                             void *timeout_ctx);
289 static int wpa_driver_nl80211_set_mode(struct i802_bss *bss,
290                                        enum nl80211_iftype nlmode);
291 static int
292 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
293 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
294                                    const u8 *addr, int cmd, u16 reason_code,
295                                    int local_state_change);
296 static void nl80211_remove_monitor_interface(
297         struct wpa_driver_nl80211_data *drv);
298 static int nl80211_send_frame_cmd(struct i802_bss *bss,
299                                   unsigned int freq, unsigned int wait,
300                                   const u8 *buf, size_t buf_len, u64 *cookie,
301                                   int no_cck, int no_ack, int offchanok);
302 static int wpa_driver_nl80211_probe_req_report(void *priv, int report);
303 #ifdef ANDROID
304 static int android_pno_start(struct i802_bss *bss,
305                              struct wpa_driver_scan_params *params);
306 static int android_pno_stop(struct i802_bss *bss);
307 #endif /* ANDROID */
308
309 #ifdef HOSTAPD
310 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
311 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
312 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
313 static int wpa_driver_nl80211_if_remove(void *priv,
314                                         enum wpa_driver_if_type type,
315                                         const char *ifname);
316 #else /* HOSTAPD */
317 static inline void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
318 {
319 }
320
321 static inline void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
322 {
323 }
324
325 static inline int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
326 {
327         return 0;
328 }
329 #endif /* HOSTAPD */
330
331 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq);
332 static int nl80211_disable_11b_rates(struct wpa_driver_nl80211_data *drv,
333                                      int ifindex, int disabled);
334
335 static int nl80211_leave_ibss(struct wpa_driver_nl80211_data *drv);
336 static int wpa_driver_nl80211_authenticate_retry(
337         struct wpa_driver_nl80211_data *drv);
338
339
340 static int is_ap_interface(enum nl80211_iftype nlmode)
341 {
342         return (nlmode == NL80211_IFTYPE_AP ||
343                 nlmode == NL80211_IFTYPE_P2P_GO);
344 }
345
346
347 static int is_sta_interface(enum nl80211_iftype nlmode)
348 {
349         return (nlmode == NL80211_IFTYPE_STATION ||
350                 nlmode == NL80211_IFTYPE_P2P_CLIENT);
351 }
352
353
354 static int is_p2p_interface(enum nl80211_iftype nlmode)
355 {
356         return (nlmode == NL80211_IFTYPE_P2P_CLIENT ||
357                 nlmode == NL80211_IFTYPE_P2P_GO);
358 }
359
360
361 struct nl80211_bss_info_arg {
362         struct wpa_driver_nl80211_data *drv;
363         struct wpa_scan_results *res;
364         unsigned int assoc_freq;
365         u8 assoc_bssid[ETH_ALEN];
366 };
367
368 static int bss_info_handler(struct nl_msg *msg, void *arg);
369
370
371 /* nl80211 code */
372 static int ack_handler(struct nl_msg *msg, void *arg)
373 {
374         int *err = arg;
375         *err = 0;
376         return NL_STOP;
377 }
378
379 static int finish_handler(struct nl_msg *msg, void *arg)
380 {
381         int *ret = arg;
382         *ret = 0;
383         return NL_SKIP;
384 }
385
386 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
387                          void *arg)
388 {
389         int *ret = arg;
390         *ret = err->error;
391         return NL_SKIP;
392 }
393
394
395 static int no_seq_check(struct nl_msg *msg, void *arg)
396 {
397         return NL_OK;
398 }
399
400
401 static int send_and_recv(struct nl80211_global *global,
402                          struct nl_handle *nl_handle, struct nl_msg *msg,
403                          int (*valid_handler)(struct nl_msg *, void *),
404                          void *valid_data)
405 {
406         struct nl_cb *cb;
407         int err = -ENOMEM;
408
409         cb = nl_cb_clone(global->nl_cb);
410         if (!cb)
411                 goto out;
412
413         err = nl_send_auto_complete(nl_handle, msg);
414         if (err < 0)
415                 goto out;
416
417         err = 1;
418
419         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
420         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
421         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
422
423         if (valid_handler)
424                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
425                           valid_handler, valid_data);
426
427         while (err > 0)
428                 nl_recvmsgs(nl_handle, cb);
429  out:
430         nl_cb_put(cb);
431         nlmsg_free(msg);
432         return err;
433 }
434
435
436 static int send_and_recv_msgs_global(struct nl80211_global *global,
437                                      struct nl_msg *msg,
438                                      int (*valid_handler)(struct nl_msg *, void *),
439                                      void *valid_data)
440 {
441         return send_and_recv(global, global->nl, msg, valid_handler,
442                              valid_data);
443 }
444
445
446 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
447                               struct nl_msg *msg,
448                               int (*valid_handler)(struct nl_msg *, void *),
449                               void *valid_data)
450 {
451         return send_and_recv(drv->global, drv->global->nl, msg,
452                              valid_handler, valid_data);
453 }
454
455
456 struct family_data {
457         const char *group;
458         int id;
459 };
460
461
462 static int family_handler(struct nl_msg *msg, void *arg)
463 {
464         struct family_data *res = arg;
465         struct nlattr *tb[CTRL_ATTR_MAX + 1];
466         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
467         struct nlattr *mcgrp;
468         int i;
469
470         nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
471                   genlmsg_attrlen(gnlh, 0), NULL);
472         if (!tb[CTRL_ATTR_MCAST_GROUPS])
473                 return NL_SKIP;
474
475         nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], i) {
476                 struct nlattr *tb2[CTRL_ATTR_MCAST_GRP_MAX + 1];
477                 nla_parse(tb2, CTRL_ATTR_MCAST_GRP_MAX, nla_data(mcgrp),
478                           nla_len(mcgrp), NULL);
479                 if (!tb2[CTRL_ATTR_MCAST_GRP_NAME] ||
480                     !tb2[CTRL_ATTR_MCAST_GRP_ID] ||
481                     os_strncmp(nla_data(tb2[CTRL_ATTR_MCAST_GRP_NAME]),
482                                res->group,
483                                nla_len(tb2[CTRL_ATTR_MCAST_GRP_NAME])) != 0)
484                         continue;
485                 res->id = nla_get_u32(tb2[CTRL_ATTR_MCAST_GRP_ID]);
486                 break;
487         };
488
489         return NL_SKIP;
490 }
491
492
493 static int nl_get_multicast_id(struct nl80211_global *global,
494                                const char *family, const char *group)
495 {
496         struct nl_msg *msg;
497         int ret = -1;
498         struct family_data res = { group, -ENOENT };
499
500         msg = nlmsg_alloc();
501         if (!msg)
502                 return -ENOMEM;
503         genlmsg_put(msg, 0, 0, genl_ctrl_resolve(global->nl, "nlctrl"),
504                     0, 0, CTRL_CMD_GETFAMILY, 0);
505         NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
506
507         ret = send_and_recv_msgs_global(global, msg, family_handler, &res);
508         msg = NULL;
509         if (ret == 0)
510                 ret = res.id;
511
512 nla_put_failure:
513         nlmsg_free(msg);
514         return ret;
515 }
516
517
518 static void * nl80211_cmd(struct wpa_driver_nl80211_data *drv,
519                           struct nl_msg *msg, int flags, uint8_t cmd)
520 {
521         return genlmsg_put(msg, 0, 0, drv->global->nl80211_id,
522                            0, flags, cmd, 0);
523 }
524
525
526 struct wiphy_idx_data {
527         int wiphy_idx;
528 };
529
530
531 static int netdev_info_handler(struct nl_msg *msg, void *arg)
532 {
533         struct nlattr *tb[NL80211_ATTR_MAX + 1];
534         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
535         struct wiphy_idx_data *info = arg;
536
537         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
538                   genlmsg_attrlen(gnlh, 0), NULL);
539
540         if (tb[NL80211_ATTR_WIPHY])
541                 info->wiphy_idx = nla_get_u32(tb[NL80211_ATTR_WIPHY]);
542
543         return NL_SKIP;
544 }
545
546
547 static int nl80211_get_wiphy_index(struct i802_bss *bss)
548 {
549         struct nl_msg *msg;
550         struct wiphy_idx_data data = {
551                 .wiphy_idx = -1,
552         };
553
554         msg = nlmsg_alloc();
555         if (!msg)
556                 return -1;
557
558         nl80211_cmd(bss->drv, msg, 0, NL80211_CMD_GET_INTERFACE);
559
560         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
561
562         if (send_and_recv_msgs(bss->drv, msg, netdev_info_handler, &data) == 0)
563                 return data.wiphy_idx;
564         msg = NULL;
565 nla_put_failure:
566         nlmsg_free(msg);
567         return -1;
568 }
569
570
571 static int nl80211_register_beacons(struct wpa_driver_nl80211_data *drv,
572                                     struct nl80211_wiphy_data *w)
573 {
574         struct nl_msg *msg;
575         int ret = -1;
576
577         msg = nlmsg_alloc();
578         if (!msg)
579                 return -1;
580
581         nl80211_cmd(drv, msg, 0, NL80211_CMD_REGISTER_BEACONS);
582
583         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, w->wiphy_idx);
584
585         ret = send_and_recv(drv->global, w->nl_beacons, msg, NULL, NULL);
586         msg = NULL;
587         if (ret) {
588                 wpa_printf(MSG_DEBUG, "nl80211: Register beacons command "
589                            "failed: ret=%d (%s)",
590                            ret, strerror(-ret));
591                 goto nla_put_failure;
592         }
593         ret = 0;
594 nla_put_failure:
595         nlmsg_free(msg);
596         return ret;
597 }
598
599
600 static void nl80211_recv_beacons(int sock, void *eloop_ctx, void *handle)
601 {
602         struct nl80211_wiphy_data *w = eloop_ctx;
603
604         wpa_printf(MSG_EXCESSIVE, "nl80211: Beacon event message available");
605
606         nl_recvmsgs(handle, w->nl_cb);
607 }
608
609
610 static int process_beacon_event(struct nl_msg *msg, void *arg)
611 {
612         struct nl80211_wiphy_data *w = arg;
613         struct wpa_driver_nl80211_data *drv;
614         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
615         struct nlattr *tb[NL80211_ATTR_MAX + 1];
616         union wpa_event_data event;
617
618         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
619                   genlmsg_attrlen(gnlh, 0), NULL);
620
621         if (gnlh->cmd != NL80211_CMD_FRAME) {
622                 wpa_printf(MSG_DEBUG, "nl80211: Unexpected beacon event? (%d)",
623                            gnlh->cmd);
624                 return NL_SKIP;
625         }
626
627         if (!tb[NL80211_ATTR_FRAME])
628                 return NL_SKIP;
629
630         dl_list_for_each(drv, &w->drvs, struct wpa_driver_nl80211_data,
631                          wiphy_list) {
632                 os_memset(&event, 0, sizeof(event));
633                 event.rx_mgmt.frame = nla_data(tb[NL80211_ATTR_FRAME]);
634                 event.rx_mgmt.frame_len = nla_len(tb[NL80211_ATTR_FRAME]);
635                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
636         }
637
638         return NL_SKIP;
639 }
640
641
642 static struct nl80211_wiphy_data *
643 nl80211_get_wiphy_data_ap(struct i802_bss *bss)
644 {
645         static DEFINE_DL_LIST(nl80211_wiphys);
646         struct nl80211_wiphy_data *w;
647         int wiphy_idx, found = 0;
648         struct i802_bss *tmp_bss;
649
650         if (bss->wiphy_data != NULL)
651                 return bss->wiphy_data;
652
653         wiphy_idx = nl80211_get_wiphy_index(bss);
654
655         dl_list_for_each(w, &nl80211_wiphys, struct nl80211_wiphy_data, list) {
656                 if (w->wiphy_idx == wiphy_idx)
657                         goto add;
658         }
659
660         /* alloc new one */
661         w = os_zalloc(sizeof(*w));
662         if (w == NULL)
663                 return NULL;
664         w->wiphy_idx = wiphy_idx;
665         dl_list_init(&w->bsss);
666         dl_list_init(&w->drvs);
667
668         w->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
669         if (!w->nl_cb) {
670                 os_free(w);
671                 return NULL;
672         }
673         nl_cb_set(w->nl_cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, no_seq_check, NULL);
674         nl_cb_set(w->nl_cb, NL_CB_VALID, NL_CB_CUSTOM, process_beacon_event,
675                   w);
676
677         w->nl_beacons = nl_create_handle(bss->drv->global->nl_cb,
678                                          "wiphy beacons");
679         if (w->nl_beacons == NULL) {
680                 os_free(w);
681                 return NULL;
682         }
683
684         if (nl80211_register_beacons(bss->drv, w)) {
685                 nl_destroy_handles(&w->nl_beacons);
686                 os_free(w);
687                 return NULL;
688         }
689
690         eloop_register_read_sock(nl_socket_get_fd(w->nl_beacons),
691                                  nl80211_recv_beacons, w, w->nl_beacons);
692
693         dl_list_add(&nl80211_wiphys, &w->list);
694
695 add:
696         /* drv entry for this bss already there? */
697         dl_list_for_each(tmp_bss, &w->bsss, struct i802_bss, wiphy_list) {
698                 if (tmp_bss->drv == bss->drv) {
699                         found = 1;
700                         break;
701                 }
702         }
703         /* if not add it */
704         if (!found)
705                 dl_list_add(&w->drvs, &bss->drv->wiphy_list);
706
707         dl_list_add(&w->bsss, &bss->wiphy_list);
708         bss->wiphy_data = w;
709         return w;
710 }
711
712
713 static void nl80211_put_wiphy_data_ap(struct i802_bss *bss)
714 {
715         struct nl80211_wiphy_data *w = bss->wiphy_data;
716         struct i802_bss *tmp_bss;
717         int found = 0;
718
719         if (w == NULL)
720                 return;
721         bss->wiphy_data = NULL;
722         dl_list_del(&bss->wiphy_list);
723
724         /* still any for this drv present? */
725         dl_list_for_each(tmp_bss, &w->bsss, struct i802_bss, wiphy_list) {
726                 if (tmp_bss->drv == bss->drv) {
727                         found = 1;
728                         break;
729                 }
730         }
731         /* if not remove it */
732         if (!found)
733                 dl_list_del(&bss->drv->wiphy_list);
734
735         if (!dl_list_empty(&w->bsss))
736                 return;
737
738         eloop_unregister_read_sock(nl_socket_get_fd(w->nl_beacons));
739
740         nl_cb_put(w->nl_cb);
741         nl_destroy_handles(&w->nl_beacons);
742         dl_list_del(&w->list);
743         os_free(w);
744 }
745
746
747 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
748 {
749         struct i802_bss *bss = priv;
750         struct wpa_driver_nl80211_data *drv = bss->drv;
751         if (!drv->associated)
752                 return -1;
753         os_memcpy(bssid, drv->bssid, ETH_ALEN);
754         return 0;
755 }
756
757
758 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
759 {
760         struct i802_bss *bss = priv;
761         struct wpa_driver_nl80211_data *drv = bss->drv;
762         if (!drv->associated)
763                 return -1;
764         os_memcpy(ssid, drv->ssid, drv->ssid_len);
765         return drv->ssid_len;
766 }
767
768
769 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
770                                           char *buf, size_t len, int del)
771 {
772         union wpa_event_data event;
773
774         os_memset(&event, 0, sizeof(event));
775         if (len > sizeof(event.interface_status.ifname))
776                 len = sizeof(event.interface_status.ifname) - 1;
777         os_memcpy(event.interface_status.ifname, buf, len);
778         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
779                 EVENT_INTERFACE_ADDED;
780
781         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
782                    del ? "DEL" : "NEW",
783                    event.interface_status.ifname,
784                    del ? "removed" : "added");
785
786         if (os_strcmp(drv->first_bss.ifname, event.interface_status.ifname) == 0) {
787                 if (del) {
788                         if (drv->if_removed) {
789                                 wpa_printf(MSG_DEBUG, "nl80211: if_removed "
790                                            "already set - ignore event");
791                                 return;
792                         }
793                         drv->if_removed = 1;
794                 } else {
795                         if (if_nametoindex(drv->first_bss.ifname) == 0) {
796                                 wpa_printf(MSG_DEBUG, "nl80211: Interface %s "
797                                            "does not exist - ignore "
798                                            "RTM_NEWLINK",
799                                            drv->first_bss.ifname);
800                                 return;
801                         }
802                         if (!drv->if_removed) {
803                                 wpa_printf(MSG_DEBUG, "nl80211: if_removed "
804                                            "already cleared - ignore event");
805                                 return;
806                         }
807                         drv->if_removed = 0;
808                 }
809         }
810
811         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
812 }
813
814
815 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
816                                          u8 *buf, size_t len)
817 {
818         int attrlen, rta_len;
819         struct rtattr *attr;
820
821         attrlen = len;
822         attr = (struct rtattr *) buf;
823
824         rta_len = RTA_ALIGN(sizeof(struct rtattr));
825         while (RTA_OK(attr, attrlen)) {
826                 if (attr->rta_type == IFLA_IFNAME) {
827                         if (os_strcmp(((char *) attr) + rta_len, drv->first_bss.ifname)
828                             == 0)
829                                 return 1;
830                         else
831                                 break;
832                 }
833                 attr = RTA_NEXT(attr, attrlen);
834         }
835
836         return 0;
837 }
838
839
840 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
841                                           int ifindex, u8 *buf, size_t len)
842 {
843         if (drv->ifindex == ifindex)
844                 return 1;
845
846         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, buf, len)) {
847                 drv->first_bss.ifindex = if_nametoindex(drv->first_bss.ifname);
848                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
849                            "interface");
850                 wpa_driver_nl80211_finish_drv_init(drv);
851                 return 1;
852         }
853
854         return 0;
855 }
856
857
858 static struct wpa_driver_nl80211_data *
859 nl80211_find_drv(struct nl80211_global *global, int idx, u8 *buf, size_t len)
860 {
861         struct wpa_driver_nl80211_data *drv;
862         dl_list_for_each(drv, &global->interfaces,
863                          struct wpa_driver_nl80211_data, list) {
864                 if (wpa_driver_nl80211_own_ifindex(drv, idx, buf, len) ||
865                     have_ifidx(drv, idx))
866                         return drv;
867         }
868         return NULL;
869 }
870
871
872 static void wpa_driver_nl80211_event_rtm_newlink(void *ctx,
873                                                  struct ifinfomsg *ifi,
874                                                  u8 *buf, size_t len)
875 {
876         struct nl80211_global *global = ctx;
877         struct wpa_driver_nl80211_data *drv;
878         int attrlen, rta_len;
879         struct rtattr *attr;
880         u32 brid = 0;
881         char namebuf[IFNAMSIZ];
882
883         drv = nl80211_find_drv(global, ifi->ifi_index, buf, len);
884         if (!drv) {
885                 wpa_printf(MSG_DEBUG, "nl80211: Ignore event for foreign "
886                            "ifindex %d", ifi->ifi_index);
887                 return;
888         }
889
890         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
891                    "(%s%s%s%s)",
892                    drv->operstate, ifi->ifi_flags,
893                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
894                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
895                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
896                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
897
898         if (!drv->if_disabled && !(ifi->ifi_flags & IFF_UP)) {
899                 if (if_indextoname(ifi->ifi_index, namebuf) &&
900                     linux_iface_up(drv->global->ioctl_sock,
901                                    drv->first_bss.ifname) > 0) {
902                         wpa_printf(MSG_DEBUG, "nl80211: Ignore interface down "
903                                    "event since interface %s is up", namebuf);
904                         return;
905                 }
906                 wpa_printf(MSG_DEBUG, "nl80211: Interface down");
907                 if (drv->ignore_if_down_event) {
908                         wpa_printf(MSG_DEBUG, "nl80211: Ignore interface down "
909                                    "event generated by mode change");
910                         drv->ignore_if_down_event = 0;
911                 } else {
912                         drv->if_disabled = 1;
913                         wpa_supplicant_event(drv->ctx,
914                                              EVENT_INTERFACE_DISABLED, NULL);
915                 }
916         }
917
918         if (drv->if_disabled && (ifi->ifi_flags & IFF_UP)) {
919                 if (if_indextoname(ifi->ifi_index, namebuf) &&
920                     linux_iface_up(drv->global->ioctl_sock,
921                                    drv->first_bss.ifname) == 0) {
922                         wpa_printf(MSG_DEBUG, "nl80211: Ignore interface up "
923                                    "event since interface %s is down",
924                                    namebuf);
925                 } else if (if_nametoindex(drv->first_bss.ifname) == 0) {
926                         wpa_printf(MSG_DEBUG, "nl80211: Ignore interface up "
927                                    "event since interface %s does not exist",
928                                    drv->first_bss.ifname);
929                 } else if (drv->if_removed) {
930                         wpa_printf(MSG_DEBUG, "nl80211: Ignore interface up "
931                                    "event since interface %s is marked "
932                                    "removed", drv->first_bss.ifname);
933                 } else {
934                         wpa_printf(MSG_DEBUG, "nl80211: Interface up");
935                         drv->if_disabled = 0;
936                         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_ENABLED,
937                                              NULL);
938                 }
939         }
940
941         /*
942          * Some drivers send the association event before the operup event--in
943          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
944          * fails. This will hit us when wpa_supplicant does not need to do
945          * IEEE 802.1X authentication
946          */
947         if (drv->operstate == 1 &&
948             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
949             !(ifi->ifi_flags & IFF_RUNNING))
950                 netlink_send_oper_ifla(drv->global->netlink, drv->ifindex,
951                                        -1, IF_OPER_UP);
952
953         attrlen = len;
954         attr = (struct rtattr *) buf;
955         rta_len = RTA_ALIGN(sizeof(struct rtattr));
956         while (RTA_OK(attr, attrlen)) {
957                 if (attr->rta_type == IFLA_IFNAME) {
958                         wpa_driver_nl80211_event_link(
959                                 drv,
960                                 ((char *) attr) + rta_len,
961                                 attr->rta_len - rta_len, 0);
962                 } else if (attr->rta_type == IFLA_MASTER)
963                         brid = nla_get_u32((struct nlattr *) attr);
964                 attr = RTA_NEXT(attr, attrlen);
965         }
966
967         if (ifi->ifi_family == AF_BRIDGE && brid) {
968                 /* device has been added to bridge */
969                 if_indextoname(brid, namebuf);
970                 wpa_printf(MSG_DEBUG, "nl80211: Add ifindex %u for bridge %s",
971                            brid, namebuf);
972                 add_ifidx(drv, brid);
973         }
974 }
975
976
977 static void wpa_driver_nl80211_event_rtm_dellink(void *ctx,
978                                                  struct ifinfomsg *ifi,
979                                                  u8 *buf, size_t len)
980 {
981         struct nl80211_global *global = ctx;
982         struct wpa_driver_nl80211_data *drv;
983         int attrlen, rta_len;
984         struct rtattr *attr;
985         u32 brid = 0;
986
987         drv = nl80211_find_drv(global, ifi->ifi_index, buf, len);
988         if (!drv) {
989                 wpa_printf(MSG_DEBUG, "nl80211: Ignore dellink event for "
990                            "foreign ifindex %d", ifi->ifi_index);
991                 return;
992         }
993
994         attrlen = len;
995         attr = (struct rtattr *) buf;
996
997         rta_len = RTA_ALIGN(sizeof(struct rtattr));
998         while (RTA_OK(attr, attrlen)) {
999                 if (attr->rta_type == IFLA_IFNAME) {
1000                         wpa_driver_nl80211_event_link(
1001                                 drv,
1002                                 ((char *) attr) + rta_len,
1003                                 attr->rta_len - rta_len, 1);
1004                 } else if (attr->rta_type == IFLA_MASTER)
1005                         brid = nla_get_u32((struct nlattr *) attr);
1006                 attr = RTA_NEXT(attr, attrlen);
1007         }
1008
1009         if (ifi->ifi_family == AF_BRIDGE && brid) {
1010                 /* device has been removed from bridge */
1011                 char namebuf[IFNAMSIZ];
1012                 if_indextoname(brid, namebuf);
1013                 wpa_printf(MSG_DEBUG, "nl80211: Remove ifindex %u for bridge "
1014                            "%s", brid, namebuf);
1015                 del_ifidx(drv, brid);
1016         }
1017 }
1018
1019
1020 static void mlme_event_auth(struct wpa_driver_nl80211_data *drv,
1021                             const u8 *frame, size_t len)
1022 {
1023         const struct ieee80211_mgmt *mgmt;
1024         union wpa_event_data event;
1025
1026         mgmt = (const struct ieee80211_mgmt *) frame;
1027         if (len < 24 + sizeof(mgmt->u.auth)) {
1028                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
1029                            "frame");
1030                 return;
1031         }
1032
1033         os_memcpy(drv->auth_bssid, mgmt->sa, ETH_ALEN);
1034         os_memset(&event, 0, sizeof(event));
1035         os_memcpy(event.auth.peer, mgmt->sa, ETH_ALEN);
1036         event.auth.auth_type = le_to_host16(mgmt->u.auth.auth_alg);
1037         event.auth.status_code = le_to_host16(mgmt->u.auth.status_code);
1038         if (len > 24 + sizeof(mgmt->u.auth)) {
1039                 event.auth.ies = mgmt->u.auth.variable;
1040                 event.auth.ies_len = len - 24 - sizeof(mgmt->u.auth);
1041         }
1042
1043         wpa_supplicant_event(drv->ctx, EVENT_AUTH, &event);
1044 }
1045
1046
1047 static unsigned int nl80211_get_assoc_freq(struct wpa_driver_nl80211_data *drv)
1048 {
1049         struct nl_msg *msg;
1050         int ret;
1051         struct nl80211_bss_info_arg arg;
1052
1053         os_memset(&arg, 0, sizeof(arg));
1054         msg = nlmsg_alloc();
1055         if (!msg)
1056                 goto nla_put_failure;
1057
1058         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
1059         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1060
1061         arg.drv = drv;
1062         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
1063         msg = NULL;
1064         if (ret == 0) {
1065                 wpa_printf(MSG_DEBUG, "nl80211: Operating frequency for the "
1066                            "associated BSS from scan results: %u MHz",
1067                            arg.assoc_freq);
1068                 return arg.assoc_freq ? arg.assoc_freq : drv->assoc_freq;
1069         }
1070         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
1071                    "(%s)", ret, strerror(-ret));
1072 nla_put_failure:
1073         nlmsg_free(msg);
1074         return drv->assoc_freq;
1075 }
1076
1077
1078 static void mlme_event_assoc(struct wpa_driver_nl80211_data *drv,
1079                             const u8 *frame, size_t len)
1080 {
1081         const struct ieee80211_mgmt *mgmt;
1082         union wpa_event_data event;
1083         u16 status;
1084
1085         mgmt = (const struct ieee80211_mgmt *) frame;
1086         if (len < 24 + sizeof(mgmt->u.assoc_resp)) {
1087                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
1088                            "frame");
1089                 return;
1090         }
1091
1092         status = le_to_host16(mgmt->u.assoc_resp.status_code);
1093         if (status != WLAN_STATUS_SUCCESS) {
1094                 os_memset(&event, 0, sizeof(event));
1095                 event.assoc_reject.bssid = mgmt->bssid;
1096                 if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
1097                         event.assoc_reject.resp_ies =
1098                                 (u8 *) mgmt->u.assoc_resp.variable;
1099                         event.assoc_reject.resp_ies_len =
1100                                 len - 24 - sizeof(mgmt->u.assoc_resp);
1101                 }
1102                 event.assoc_reject.status_code = status;
1103
1104                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
1105                 return;
1106         }
1107
1108         drv->associated = 1;
1109         os_memcpy(drv->bssid, mgmt->sa, ETH_ALEN);
1110
1111         os_memset(&event, 0, sizeof(event));
1112         if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
1113                 event.assoc_info.resp_ies = (u8 *) mgmt->u.assoc_resp.variable;
1114                 event.assoc_info.resp_ies_len =
1115                         len - 24 - sizeof(mgmt->u.assoc_resp);
1116         }
1117
1118         event.assoc_info.freq = drv->assoc_freq;
1119
1120         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
1121 }
1122
1123
1124 static void mlme_event_connect(struct wpa_driver_nl80211_data *drv,
1125                                enum nl80211_commands cmd, struct nlattr *status,
1126                                struct nlattr *addr, struct nlattr *req_ie,
1127                                struct nlattr *resp_ie)
1128 {
1129         union wpa_event_data event;
1130
1131         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
1132                 /*
1133                  * Avoid reporting two association events that would confuse
1134                  * the core code.
1135                  */
1136                 wpa_printf(MSG_DEBUG, "nl80211: Ignore connect event (cmd=%d) "
1137                            "when using userspace SME", cmd);
1138                 return;
1139         }
1140
1141         os_memset(&event, 0, sizeof(event));
1142         if (cmd == NL80211_CMD_CONNECT &&
1143             nla_get_u16(status) != WLAN_STATUS_SUCCESS) {
1144                 if (addr)
1145                         event.assoc_reject.bssid = nla_data(addr);
1146                 if (resp_ie) {
1147                         event.assoc_reject.resp_ies = nla_data(resp_ie);
1148                         event.assoc_reject.resp_ies_len = nla_len(resp_ie);
1149                 }
1150                 event.assoc_reject.status_code = nla_get_u16(status);
1151                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
1152                 return;
1153         }
1154
1155         drv->associated = 1;
1156         if (addr)
1157                 os_memcpy(drv->bssid, nla_data(addr), ETH_ALEN);
1158
1159         if (req_ie) {
1160                 event.assoc_info.req_ies = nla_data(req_ie);
1161                 event.assoc_info.req_ies_len = nla_len(req_ie);
1162         }
1163         if (resp_ie) {
1164                 event.assoc_info.resp_ies = nla_data(resp_ie);
1165                 event.assoc_info.resp_ies_len = nla_len(resp_ie);
1166         }
1167
1168         event.assoc_info.freq = nl80211_get_assoc_freq(drv);
1169
1170         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
1171 }
1172
1173
1174 static void mlme_event_disconnect(struct wpa_driver_nl80211_data *drv,
1175                                   struct nlattr *reason, struct nlattr *addr,
1176                                   struct nlattr *by_ap)
1177 {
1178         union wpa_event_data data;
1179
1180         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
1181                 /*
1182                  * Avoid reporting two disassociation events that could
1183                  * confuse the core code.
1184                  */
1185                 wpa_printf(MSG_DEBUG, "nl80211: Ignore disconnect "
1186                            "event when using userspace SME");
1187                 return;
1188         }
1189
1190         drv->associated = 0;
1191         os_memset(&data, 0, sizeof(data));
1192         if (reason)
1193                 data.deauth_info.reason_code = nla_get_u16(reason);
1194         data.deauth_info.locally_generated = by_ap == NULL;
1195         wpa_supplicant_event(drv->ctx, EVENT_DEAUTH, &data);
1196 }
1197
1198
1199 static void mlme_event_ch_switch(struct wpa_driver_nl80211_data *drv,
1200                                  struct nlattr *freq, struct nlattr *type)
1201 {
1202         union wpa_event_data data;
1203         int ht_enabled = 1;
1204         int chan_offset = 0;
1205
1206         wpa_printf(MSG_DEBUG, "nl80211: Channel switch event");
1207
1208         if (!freq || !type)
1209                 return;
1210
1211         switch (nla_get_u32(type)) {
1212         case NL80211_CHAN_NO_HT:
1213                 ht_enabled = 0;
1214                 break;
1215         case NL80211_CHAN_HT20:
1216                 break;
1217         case NL80211_CHAN_HT40PLUS:
1218                 chan_offset = 1;
1219                 break;
1220         case NL80211_CHAN_HT40MINUS:
1221                 chan_offset = -1;
1222                 break;
1223         }
1224
1225         data.ch_switch.freq = nla_get_u32(freq);
1226         data.ch_switch.ht_enabled = ht_enabled;
1227         data.ch_switch.ch_offset = chan_offset;
1228
1229         wpa_supplicant_event(drv->ctx, EVENT_CH_SWITCH, &data);
1230 }
1231
1232
1233 static void mlme_timeout_event(struct wpa_driver_nl80211_data *drv,
1234                                enum nl80211_commands cmd, struct nlattr *addr)
1235 {
1236         union wpa_event_data event;
1237         enum wpa_event_type ev;
1238
1239         if (nla_len(addr) != ETH_ALEN)
1240                 return;
1241
1242         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d; timeout with " MACSTR,
1243                    cmd, MAC2STR((u8 *) nla_data(addr)));
1244
1245         if (cmd == NL80211_CMD_AUTHENTICATE)
1246                 ev = EVENT_AUTH_TIMED_OUT;
1247         else if (cmd == NL80211_CMD_ASSOCIATE)
1248                 ev = EVENT_ASSOC_TIMED_OUT;
1249         else
1250                 return;
1251
1252         os_memset(&event, 0, sizeof(event));
1253         os_memcpy(event.timeout_event.addr, nla_data(addr), ETH_ALEN);
1254         wpa_supplicant_event(drv->ctx, ev, &event);
1255 }
1256
1257
1258 static void mlme_event_mgmt(struct wpa_driver_nl80211_data *drv,
1259                             struct nlattr *freq, struct nlattr *sig,
1260                             const u8 *frame, size_t len)
1261 {
1262         const struct ieee80211_mgmt *mgmt;
1263         union wpa_event_data event;
1264         u16 fc, stype;
1265         int ssi_signal = 0;
1266
1267         mgmt = (const struct ieee80211_mgmt *) frame;
1268         if (len < 24) {
1269                 wpa_printf(MSG_DEBUG, "nl80211: Too short action frame");
1270                 return;
1271         }
1272
1273         fc = le_to_host16(mgmt->frame_control);
1274         stype = WLAN_FC_GET_STYPE(fc);
1275
1276         if (sig)
1277                 ssi_signal = (s32) nla_get_u32(sig);
1278
1279         os_memset(&event, 0, sizeof(event));
1280         if (freq) {
1281                 event.rx_action.freq = nla_get_u32(freq);
1282                 drv->last_mgmt_freq = event.rx_action.freq;
1283         }
1284         if (stype == WLAN_FC_STYPE_ACTION) {
1285                 event.rx_action.da = mgmt->da;
1286                 event.rx_action.sa = mgmt->sa;
1287                 event.rx_action.bssid = mgmt->bssid;
1288                 event.rx_action.category = mgmt->u.action.category;
1289                 event.rx_action.data = &mgmt->u.action.category + 1;
1290                 event.rx_action.len = frame + len - event.rx_action.data;
1291                 wpa_supplicant_event(drv->ctx, EVENT_RX_ACTION, &event);
1292         } else {
1293                 event.rx_mgmt.frame = frame;
1294                 event.rx_mgmt.frame_len = len;
1295                 event.rx_mgmt.ssi_signal = ssi_signal;
1296                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
1297         }
1298 }
1299
1300
1301 static void mlme_event_mgmt_tx_status(struct wpa_driver_nl80211_data *drv,
1302                                       struct nlattr *cookie, const u8 *frame,
1303                                       size_t len, struct nlattr *ack)
1304 {
1305         union wpa_event_data event;
1306         const struct ieee80211_hdr *hdr;
1307         u16 fc;
1308
1309         if (!is_ap_interface(drv->nlmode)) {
1310                 u64 cookie_val;
1311
1312                 if (!cookie)
1313                         return;
1314
1315                 cookie_val = nla_get_u64(cookie);
1316                 wpa_printf(MSG_DEBUG, "nl80211: Action TX status:"
1317                            " cookie=0%llx%s (ack=%d)",
1318                            (long long unsigned int) cookie_val,
1319                            cookie_val == drv->send_action_cookie ?
1320                            " (match)" : " (unknown)", ack != NULL);
1321                 if (cookie_val != drv->send_action_cookie)
1322                         return;
1323         }
1324
1325         hdr = (const struct ieee80211_hdr *) frame;
1326         fc = le_to_host16(hdr->frame_control);
1327
1328         os_memset(&event, 0, sizeof(event));
1329         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
1330         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
1331         event.tx_status.dst = hdr->addr1;
1332         event.tx_status.data = frame;
1333         event.tx_status.data_len = len;
1334         event.tx_status.ack = ack != NULL;
1335         wpa_supplicant_event(drv->ctx, EVENT_TX_STATUS, &event);
1336 }
1337
1338
1339 static void mlme_event_deauth_disassoc(struct wpa_driver_nl80211_data *drv,
1340                                        enum wpa_event_type type,
1341                                        const u8 *frame, size_t len)
1342 {
1343         const struct ieee80211_mgmt *mgmt;
1344         union wpa_event_data event;
1345         const u8 *bssid = NULL;
1346         u16 reason_code = 0;
1347
1348         mgmt = (const struct ieee80211_mgmt *) frame;
1349         if (len >= 24) {
1350                 bssid = mgmt->bssid;
1351
1352                 if (drv->associated != 0 &&
1353                     os_memcmp(bssid, drv->bssid, ETH_ALEN) != 0 &&
1354                     os_memcmp(bssid, drv->auth_bssid, ETH_ALEN) != 0) {
1355                         /*
1356                          * We have presumably received this deauth as a
1357                          * response to a clear_state_mismatch() outgoing
1358                          * deauth.  Don't let it take us offline!
1359                          */
1360                         wpa_printf(MSG_DEBUG, "nl80211: Deauth received "
1361                                    "from Unknown BSSID " MACSTR " -- ignoring",
1362                                    MAC2STR(bssid));
1363                         return;
1364                 }
1365         }
1366
1367         drv->associated = 0;
1368         os_memset(&event, 0, sizeof(event));
1369
1370         /* Note: Same offset for Reason Code in both frame subtypes */
1371         if (len >= 24 + sizeof(mgmt->u.deauth))
1372                 reason_code = le_to_host16(mgmt->u.deauth.reason_code);
1373
1374         if (type == EVENT_DISASSOC) {
1375                 event.disassoc_info.locally_generated =
1376                         !os_memcmp(mgmt->sa, drv->first_bss.addr, ETH_ALEN);
1377                 event.disassoc_info.addr = bssid;
1378                 event.disassoc_info.reason_code = reason_code;
1379                 if (frame + len > mgmt->u.disassoc.variable) {
1380                         event.disassoc_info.ie = mgmt->u.disassoc.variable;
1381                         event.disassoc_info.ie_len = frame + len -
1382                                 mgmt->u.disassoc.variable;
1383                 }
1384         } else {
1385                 event.deauth_info.locally_generated =
1386                         !os_memcmp(mgmt->sa, drv->first_bss.addr, ETH_ALEN);
1387                 event.deauth_info.addr = bssid;
1388                 event.deauth_info.reason_code = reason_code;
1389                 if (frame + len > mgmt->u.deauth.variable) {
1390                         event.deauth_info.ie = mgmt->u.deauth.variable;
1391                         event.deauth_info.ie_len = frame + len -
1392                                 mgmt->u.deauth.variable;
1393                 }
1394         }
1395
1396         wpa_supplicant_event(drv->ctx, type, &event);
1397 }
1398
1399
1400 static void mlme_event_unprot_disconnect(struct wpa_driver_nl80211_data *drv,
1401                                          enum wpa_event_type type,
1402                                          const u8 *frame, size_t len)
1403 {
1404         const struct ieee80211_mgmt *mgmt;
1405         union wpa_event_data event;
1406         u16 reason_code = 0;
1407
1408         if (len < 24)
1409                 return;
1410
1411         mgmt = (const struct ieee80211_mgmt *) frame;
1412
1413         os_memset(&event, 0, sizeof(event));
1414         /* Note: Same offset for Reason Code in both frame subtypes */
1415         if (len >= 24 + sizeof(mgmt->u.deauth))
1416                 reason_code = le_to_host16(mgmt->u.deauth.reason_code);
1417
1418         if (type == EVENT_UNPROT_DISASSOC) {
1419                 event.unprot_disassoc.sa = mgmt->sa;
1420                 event.unprot_disassoc.da = mgmt->da;
1421                 event.unprot_disassoc.reason_code = reason_code;
1422         } else {
1423                 event.unprot_deauth.sa = mgmt->sa;
1424                 event.unprot_deauth.da = mgmt->da;
1425                 event.unprot_deauth.reason_code = reason_code;
1426         }
1427
1428         wpa_supplicant_event(drv->ctx, type, &event);
1429 }
1430
1431
1432 static void mlme_event(struct wpa_driver_nl80211_data *drv,
1433                        enum nl80211_commands cmd, struct nlattr *frame,
1434                        struct nlattr *addr, struct nlattr *timed_out,
1435                        struct nlattr *freq, struct nlattr *ack,
1436                        struct nlattr *cookie, struct nlattr *sig)
1437 {
1438         if (timed_out && addr) {
1439                 mlme_timeout_event(drv, cmd, addr);
1440                 return;
1441         }
1442
1443         if (frame == NULL) {
1444                 wpa_printf(MSG_DEBUG, "nl80211: MLME event %d without frame "
1445                            "data", cmd);
1446                 return;
1447         }
1448
1449         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d", cmd);
1450         wpa_hexdump(MSG_MSGDUMP, "nl80211: MLME event frame",
1451                     nla_data(frame), nla_len(frame));
1452
1453         switch (cmd) {
1454         case NL80211_CMD_AUTHENTICATE:
1455                 mlme_event_auth(drv, nla_data(frame), nla_len(frame));
1456                 break;
1457         case NL80211_CMD_ASSOCIATE:
1458                 mlme_event_assoc(drv, nla_data(frame), nla_len(frame));
1459                 break;
1460         case NL80211_CMD_DEAUTHENTICATE:
1461                 mlme_event_deauth_disassoc(drv, EVENT_DEAUTH,
1462                                            nla_data(frame), nla_len(frame));
1463                 break;
1464         case NL80211_CMD_DISASSOCIATE:
1465                 mlme_event_deauth_disassoc(drv, EVENT_DISASSOC,
1466                                            nla_data(frame), nla_len(frame));
1467                 break;
1468         case NL80211_CMD_FRAME:
1469                 mlme_event_mgmt(drv, freq, sig, nla_data(frame),
1470                                 nla_len(frame));
1471                 break;
1472         case NL80211_CMD_FRAME_TX_STATUS:
1473                 mlme_event_mgmt_tx_status(drv, cookie, nla_data(frame),
1474                                           nla_len(frame), ack);
1475                 break;
1476         case NL80211_CMD_UNPROT_DEAUTHENTICATE:
1477                 mlme_event_unprot_disconnect(drv, EVENT_UNPROT_DEAUTH,
1478                                              nla_data(frame), nla_len(frame));
1479                 break;
1480         case NL80211_CMD_UNPROT_DISASSOCIATE:
1481                 mlme_event_unprot_disconnect(drv, EVENT_UNPROT_DISASSOC,
1482                                              nla_data(frame), nla_len(frame));
1483                 break;
1484         default:
1485                 break;
1486         }
1487 }
1488
1489
1490 static void mlme_event_michael_mic_failure(struct wpa_driver_nl80211_data *drv,
1491                                            struct nlattr *tb[])
1492 {
1493         union wpa_event_data data;
1494
1495         wpa_printf(MSG_DEBUG, "nl80211: MLME event Michael MIC failure");
1496         os_memset(&data, 0, sizeof(data));
1497         if (tb[NL80211_ATTR_MAC]) {
1498                 wpa_hexdump(MSG_DEBUG, "nl80211: Source MAC address",
1499                             nla_data(tb[NL80211_ATTR_MAC]),
1500                             nla_len(tb[NL80211_ATTR_MAC]));
1501                 data.michael_mic_failure.src = nla_data(tb[NL80211_ATTR_MAC]);
1502         }
1503         if (tb[NL80211_ATTR_KEY_SEQ]) {
1504                 wpa_hexdump(MSG_DEBUG, "nl80211: TSC",
1505                             nla_data(tb[NL80211_ATTR_KEY_SEQ]),
1506                             nla_len(tb[NL80211_ATTR_KEY_SEQ]));
1507         }
1508         if (tb[NL80211_ATTR_KEY_TYPE]) {
1509                 enum nl80211_key_type key_type =
1510                         nla_get_u32(tb[NL80211_ATTR_KEY_TYPE]);
1511                 wpa_printf(MSG_DEBUG, "nl80211: Key Type %d", key_type);
1512                 if (key_type == NL80211_KEYTYPE_PAIRWISE)
1513                         data.michael_mic_failure.unicast = 1;
1514         } else
1515                 data.michael_mic_failure.unicast = 1;
1516
1517         if (tb[NL80211_ATTR_KEY_IDX]) {
1518                 u8 key_id = nla_get_u8(tb[NL80211_ATTR_KEY_IDX]);
1519                 wpa_printf(MSG_DEBUG, "nl80211: Key Id %d", key_id);
1520         }
1521
1522         wpa_supplicant_event(drv->ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
1523 }
1524
1525
1526 static void mlme_event_join_ibss(struct wpa_driver_nl80211_data *drv,
1527                                  struct nlattr *tb[])
1528 {
1529         if (tb[NL80211_ATTR_MAC] == NULL) {
1530                 wpa_printf(MSG_DEBUG, "nl80211: No address in IBSS joined "
1531                            "event");
1532                 return;
1533         }
1534         os_memcpy(drv->bssid, nla_data(tb[NL80211_ATTR_MAC]), ETH_ALEN);
1535         drv->associated = 1;
1536         wpa_printf(MSG_DEBUG, "nl80211: IBSS " MACSTR " joined",
1537                    MAC2STR(drv->bssid));
1538
1539         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
1540 }
1541
1542
1543 static void mlme_event_remain_on_channel(struct wpa_driver_nl80211_data *drv,
1544                                          int cancel_event, struct nlattr *tb[])
1545 {
1546         unsigned int freq, chan_type, duration;
1547         union wpa_event_data data;
1548         u64 cookie;
1549
1550         if (tb[NL80211_ATTR_WIPHY_FREQ])
1551                 freq = nla_get_u32(tb[NL80211_ATTR_WIPHY_FREQ]);
1552         else
1553                 freq = 0;
1554
1555         if (tb[NL80211_ATTR_WIPHY_CHANNEL_TYPE])
1556                 chan_type = nla_get_u32(tb[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
1557         else
1558                 chan_type = 0;
1559
1560         if (tb[NL80211_ATTR_DURATION])
1561                 duration = nla_get_u32(tb[NL80211_ATTR_DURATION]);
1562         else
1563                 duration = 0;
1564
1565         if (tb[NL80211_ATTR_COOKIE])
1566                 cookie = nla_get_u64(tb[NL80211_ATTR_COOKIE]);
1567         else
1568                 cookie = 0;
1569
1570         wpa_printf(MSG_DEBUG, "nl80211: Remain-on-channel event (cancel=%d "
1571                    "freq=%u channel_type=%u duration=%u cookie=0x%llx (%s))",
1572                    cancel_event, freq, chan_type, duration,
1573                    (long long unsigned int) cookie,
1574                    cookie == drv->remain_on_chan_cookie ? "match" : "unknown");
1575
1576         if (cookie != drv->remain_on_chan_cookie)
1577                 return; /* not for us */
1578
1579         if (cancel_event)
1580                 drv->pending_remain_on_chan = 0;
1581
1582         os_memset(&data, 0, sizeof(data));
1583         data.remain_on_channel.freq = freq;
1584         data.remain_on_channel.duration = duration;
1585         wpa_supplicant_event(drv->ctx, cancel_event ?
1586                              EVENT_CANCEL_REMAIN_ON_CHANNEL :
1587                              EVENT_REMAIN_ON_CHANNEL, &data);
1588 }
1589
1590
1591 static void send_scan_event(struct wpa_driver_nl80211_data *drv, int aborted,
1592                             struct nlattr *tb[])
1593 {
1594         union wpa_event_data event;
1595         struct nlattr *nl;
1596         int rem;
1597         struct scan_info *info;
1598 #define MAX_REPORT_FREQS 50
1599         int freqs[MAX_REPORT_FREQS];
1600         int num_freqs = 0;
1601
1602         if (drv->scan_for_auth) {
1603                 drv->scan_for_auth = 0;
1604                 wpa_printf(MSG_DEBUG, "nl80211: Scan results for missing "
1605                            "cfg80211 BSS entry");
1606                 wpa_driver_nl80211_authenticate_retry(drv);
1607                 return;
1608         }
1609
1610         os_memset(&event, 0, sizeof(event));
1611         info = &event.scan_info;
1612         info->aborted = aborted;
1613
1614         if (tb[NL80211_ATTR_SCAN_SSIDS]) {
1615                 nla_for_each_nested(nl, tb[NL80211_ATTR_SCAN_SSIDS], rem) {
1616                         struct wpa_driver_scan_ssid *s =
1617                                 &info->ssids[info->num_ssids];
1618                         s->ssid = nla_data(nl);
1619                         s->ssid_len = nla_len(nl);
1620                         info->num_ssids++;
1621                         if (info->num_ssids == WPAS_MAX_SCAN_SSIDS)
1622                                 break;
1623                 }
1624         }
1625         if (tb[NL80211_ATTR_SCAN_FREQUENCIES]) {
1626                 nla_for_each_nested(nl, tb[NL80211_ATTR_SCAN_FREQUENCIES], rem)
1627                 {
1628                         freqs[num_freqs] = nla_get_u32(nl);
1629                         num_freqs++;
1630                         if (num_freqs == MAX_REPORT_FREQS - 1)
1631                                 break;
1632                 }
1633                 info->freqs = freqs;
1634                 info->num_freqs = num_freqs;
1635         }
1636         wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, &event);
1637 }
1638
1639
1640 static int get_link_signal(struct nl_msg *msg, void *arg)
1641 {
1642         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1643         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1644         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1645         static struct nla_policy policy[NL80211_STA_INFO_MAX + 1] = {
1646                 [NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 },
1647         };
1648         struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
1649         static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
1650                 [NL80211_RATE_INFO_BITRATE] = { .type = NLA_U16 },
1651                 [NL80211_RATE_INFO_MCS] = { .type = NLA_U8 },
1652                 [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
1653                 [NL80211_RATE_INFO_SHORT_GI] = { .type = NLA_FLAG },
1654         };
1655         struct wpa_signal_info *sig_change = arg;
1656
1657         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1658                   genlmsg_attrlen(gnlh, 0), NULL);
1659         if (!tb[NL80211_ATTR_STA_INFO] ||
1660             nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1661                              tb[NL80211_ATTR_STA_INFO], policy))
1662                 return NL_SKIP;
1663         if (!sinfo[NL80211_STA_INFO_SIGNAL])
1664                 return NL_SKIP;
1665
1666         sig_change->current_signal =
1667                 (s8) nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1668
1669         if (sinfo[NL80211_STA_INFO_TX_BITRATE]) {
1670                 if (nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
1671                                      sinfo[NL80211_STA_INFO_TX_BITRATE],
1672                                      rate_policy)) {
1673                         sig_change->current_txrate = 0;
1674                 } else {
1675                         if (rinfo[NL80211_RATE_INFO_BITRATE]) {
1676                                 sig_change->current_txrate =
1677                                         nla_get_u16(rinfo[
1678                                              NL80211_RATE_INFO_BITRATE]) * 100;
1679                         }
1680                 }
1681         }
1682
1683         return NL_SKIP;
1684 }
1685
1686
1687 static int nl80211_get_link_signal(struct wpa_driver_nl80211_data *drv,
1688                                    struct wpa_signal_info *sig)
1689 {
1690         struct nl_msg *msg;
1691
1692         sig->current_signal = -9999;
1693         sig->current_txrate = 0;
1694
1695         msg = nlmsg_alloc();
1696         if (!msg)
1697                 return -ENOMEM;
1698
1699         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_STATION);
1700
1701         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1702         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, drv->bssid);
1703
1704         return send_and_recv_msgs(drv, msg, get_link_signal, sig);
1705  nla_put_failure:
1706         nlmsg_free(msg);
1707         return -ENOBUFS;
1708 }
1709
1710
1711 static int get_link_noise(struct nl_msg *msg, void *arg)
1712 {
1713         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1714         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1715         struct nlattr *sinfo[NL80211_SURVEY_INFO_MAX + 1];
1716         static struct nla_policy survey_policy[NL80211_SURVEY_INFO_MAX + 1] = {
1717                 [NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
1718                 [NL80211_SURVEY_INFO_NOISE] = { .type = NLA_U8 },
1719         };
1720         struct wpa_signal_info *sig_change = arg;
1721
1722         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1723                   genlmsg_attrlen(gnlh, 0), NULL);
1724
1725         if (!tb[NL80211_ATTR_SURVEY_INFO]) {
1726                 wpa_printf(MSG_DEBUG, "nl80211: survey data missing!");
1727                 return NL_SKIP;
1728         }
1729
1730         if (nla_parse_nested(sinfo, NL80211_SURVEY_INFO_MAX,
1731                              tb[NL80211_ATTR_SURVEY_INFO],
1732                              survey_policy)) {
1733                 wpa_printf(MSG_DEBUG, "nl80211: failed to parse nested "
1734                            "attributes!");
1735                 return NL_SKIP;
1736         }
1737
1738         if (!sinfo[NL80211_SURVEY_INFO_FREQUENCY])
1739                 return NL_SKIP;
1740
1741         if (nla_get_u32(sinfo[NL80211_SURVEY_INFO_FREQUENCY]) !=
1742             sig_change->frequency)
1743                 return NL_SKIP;
1744
1745         if (!sinfo[NL80211_SURVEY_INFO_NOISE])
1746                 return NL_SKIP;
1747
1748         sig_change->current_noise =
1749                 (s8) nla_get_u8(sinfo[NL80211_SURVEY_INFO_NOISE]);
1750
1751         return NL_SKIP;
1752 }
1753
1754
1755 static int nl80211_get_link_noise(struct wpa_driver_nl80211_data *drv,
1756                                   struct wpa_signal_info *sig_change)
1757 {
1758         struct nl_msg *msg;
1759
1760         sig_change->current_noise = 9999;
1761         sig_change->frequency = drv->assoc_freq;
1762
1763         msg = nlmsg_alloc();
1764         if (!msg)
1765                 return -ENOMEM;
1766
1767         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SURVEY);
1768
1769         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1770
1771         return send_and_recv_msgs(drv, msg, get_link_noise, sig_change);
1772  nla_put_failure:
1773         nlmsg_free(msg);
1774         return -ENOBUFS;
1775 }
1776
1777
1778 static int get_noise_for_scan_results(struct nl_msg *msg, void *arg)
1779 {
1780         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1781         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1782         struct nlattr *sinfo[NL80211_SURVEY_INFO_MAX + 1];
1783         static struct nla_policy survey_policy[NL80211_SURVEY_INFO_MAX + 1] = {
1784                 [NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
1785                 [NL80211_SURVEY_INFO_NOISE] = { .type = NLA_U8 },
1786         };
1787         struct wpa_scan_results *scan_results = arg;
1788         struct wpa_scan_res *scan_res;
1789         size_t i;
1790
1791         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1792                   genlmsg_attrlen(gnlh, 0), NULL);
1793
1794         if (!tb[NL80211_ATTR_SURVEY_INFO]) {
1795                 wpa_printf(MSG_DEBUG, "nl80211: Survey data missing");
1796                 return NL_SKIP;
1797         }
1798
1799         if (nla_parse_nested(sinfo, NL80211_SURVEY_INFO_MAX,
1800                              tb[NL80211_ATTR_SURVEY_INFO],
1801                              survey_policy)) {
1802                 wpa_printf(MSG_DEBUG, "nl80211: Failed to parse nested "
1803                            "attributes");
1804                 return NL_SKIP;
1805         }
1806
1807         if (!sinfo[NL80211_SURVEY_INFO_NOISE])
1808                 return NL_SKIP;
1809
1810         if (!sinfo[NL80211_SURVEY_INFO_FREQUENCY])
1811                 return NL_SKIP;
1812
1813         for (i = 0; i < scan_results->num; ++i) {
1814                 scan_res = scan_results->res[i];
1815                 if (!scan_res)
1816                         continue;
1817                 if ((int) nla_get_u32(sinfo[NL80211_SURVEY_INFO_FREQUENCY]) !=
1818                     scan_res->freq)
1819                         continue;
1820                 if (!(scan_res->flags & WPA_SCAN_NOISE_INVALID))
1821                         continue;
1822                 scan_res->noise = (s8)
1823                         nla_get_u8(sinfo[NL80211_SURVEY_INFO_NOISE]);
1824                 scan_res->flags &= ~WPA_SCAN_NOISE_INVALID;
1825         }
1826
1827         return NL_SKIP;
1828 }
1829
1830
1831 static int nl80211_get_noise_for_scan_results(
1832         struct wpa_driver_nl80211_data *drv,
1833         struct wpa_scan_results *scan_res)
1834 {
1835         struct nl_msg *msg;
1836
1837         msg = nlmsg_alloc();
1838         if (!msg)
1839                 return -ENOMEM;
1840
1841         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SURVEY);
1842
1843         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1844
1845         return send_and_recv_msgs(drv, msg, get_noise_for_scan_results,
1846                                   scan_res);
1847  nla_put_failure:
1848         nlmsg_free(msg);
1849         return -ENOBUFS;
1850 }
1851
1852
1853 static void nl80211_cqm_event(struct wpa_driver_nl80211_data *drv,
1854                               struct nlattr *tb[])
1855 {
1856         static struct nla_policy cqm_policy[NL80211_ATTR_CQM_MAX + 1] = {
1857                 [NL80211_ATTR_CQM_RSSI_THOLD] = { .type = NLA_U32 },
1858                 [NL80211_ATTR_CQM_RSSI_HYST] = { .type = NLA_U8 },
1859                 [NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] = { .type = NLA_U32 },
1860                 [NL80211_ATTR_CQM_PKT_LOSS_EVENT] = { .type = NLA_U32 },
1861         };
1862         struct nlattr *cqm[NL80211_ATTR_CQM_MAX + 1];
1863         enum nl80211_cqm_rssi_threshold_event event;
1864         union wpa_event_data ed;
1865         struct wpa_signal_info sig;
1866         int res;
1867
1868         if (tb[NL80211_ATTR_CQM] == NULL ||
1869             nla_parse_nested(cqm, NL80211_ATTR_CQM_MAX, tb[NL80211_ATTR_CQM],
1870                              cqm_policy)) {
1871                 wpa_printf(MSG_DEBUG, "nl80211: Ignore invalid CQM event");
1872                 return;
1873         }
1874
1875         os_memset(&ed, 0, sizeof(ed));
1876
1877         if (cqm[NL80211_ATTR_CQM_PKT_LOSS_EVENT]) {
1878                 if (!tb[NL80211_ATTR_MAC])
1879                         return;
1880                 os_memcpy(ed.low_ack.addr, nla_data(tb[NL80211_ATTR_MAC]),
1881                           ETH_ALEN);
1882                 wpa_supplicant_event(drv->ctx, EVENT_STATION_LOW_ACK, &ed);
1883                 return;
1884         }
1885
1886         if (cqm[NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] == NULL)
1887                 return;
1888         event = nla_get_u32(cqm[NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT]);
1889
1890         if (event == NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH) {
1891                 wpa_printf(MSG_DEBUG, "nl80211: Connection quality monitor "
1892                            "event: RSSI high");
1893                 ed.signal_change.above_threshold = 1;
1894         } else if (event == NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW) {
1895                 wpa_printf(MSG_DEBUG, "nl80211: Connection quality monitor "
1896                            "event: RSSI low");
1897                 ed.signal_change.above_threshold = 0;
1898         } else
1899                 return;
1900
1901         res = nl80211_get_link_signal(drv, &sig);
1902         if (res == 0) {
1903                 ed.signal_change.current_signal = sig.current_signal;
1904                 ed.signal_change.current_txrate = sig.current_txrate;
1905                 wpa_printf(MSG_DEBUG, "nl80211: Signal: %d dBm  txrate: %d",
1906                            sig.current_signal, sig.current_txrate);
1907         }
1908
1909         res = nl80211_get_link_noise(drv, &sig);
1910         if (res == 0) {
1911                 ed.signal_change.current_noise = sig.current_noise;
1912                 wpa_printf(MSG_DEBUG, "nl80211: Noise: %d dBm",
1913                            sig.current_noise);
1914         }
1915
1916         wpa_supplicant_event(drv->ctx, EVENT_SIGNAL_CHANGE, &ed);
1917 }
1918
1919
1920 static void nl80211_new_station_event(struct wpa_driver_nl80211_data *drv,
1921                                       struct nlattr **tb)
1922 {
1923         u8 *addr;
1924         union wpa_event_data data;
1925
1926         if (tb[NL80211_ATTR_MAC] == NULL)
1927                 return;
1928         addr = nla_data(tb[NL80211_ATTR_MAC]);
1929         wpa_printf(MSG_DEBUG, "nl80211: New station " MACSTR, MAC2STR(addr));
1930
1931         if (is_ap_interface(drv->nlmode) && drv->device_ap_sme) {
1932                 u8 *ies = NULL;
1933                 size_t ies_len = 0;
1934                 if (tb[NL80211_ATTR_IE]) {
1935                         ies = nla_data(tb[NL80211_ATTR_IE]);
1936                         ies_len = nla_len(tb[NL80211_ATTR_IE]);
1937                 }
1938                 wpa_hexdump(MSG_DEBUG, "nl80211: Assoc Req IEs", ies, ies_len);
1939                 drv_event_assoc(drv->ctx, addr, ies, ies_len, 0);
1940                 return;
1941         }
1942
1943         if (drv->nlmode != NL80211_IFTYPE_ADHOC)
1944                 return;
1945
1946         os_memset(&data, 0, sizeof(data));
1947         os_memcpy(data.ibss_rsn_start.peer, addr, ETH_ALEN);
1948         wpa_supplicant_event(drv->ctx, EVENT_IBSS_RSN_START, &data);
1949 }
1950
1951
1952 static void nl80211_del_station_event(struct wpa_driver_nl80211_data *drv,
1953                                       struct nlattr **tb)
1954 {
1955         u8 *addr;
1956         union wpa_event_data data;
1957
1958         if (tb[NL80211_ATTR_MAC] == NULL)
1959                 return;
1960         addr = nla_data(tb[NL80211_ATTR_MAC]);
1961         wpa_printf(MSG_DEBUG, "nl80211: Delete station " MACSTR,
1962                    MAC2STR(addr));
1963
1964         if (is_ap_interface(drv->nlmode) && drv->device_ap_sme) {
1965                 drv_event_disassoc(drv->ctx, addr);
1966                 return;
1967         }
1968
1969         if (drv->nlmode != NL80211_IFTYPE_ADHOC)
1970                 return;
1971
1972         os_memset(&data, 0, sizeof(data));
1973         os_memcpy(data.ibss_peer_lost.peer, addr, ETH_ALEN);
1974         wpa_supplicant_event(drv->ctx, EVENT_IBSS_PEER_LOST, &data);
1975 }
1976
1977
1978 static void nl80211_rekey_offload_event(struct wpa_driver_nl80211_data *drv,
1979                                         struct nlattr **tb)
1980 {
1981         struct nlattr *rekey_info[NUM_NL80211_REKEY_DATA];
1982         static struct nla_policy rekey_policy[NUM_NL80211_REKEY_DATA] = {
1983                 [NL80211_REKEY_DATA_KEK] = {
1984                         .minlen = NL80211_KEK_LEN,
1985                         .maxlen = NL80211_KEK_LEN,
1986                 },
1987                 [NL80211_REKEY_DATA_KCK] = {
1988                         .minlen = NL80211_KCK_LEN,
1989                         .maxlen = NL80211_KCK_LEN,
1990                 },
1991                 [NL80211_REKEY_DATA_REPLAY_CTR] = {
1992                         .minlen = NL80211_REPLAY_CTR_LEN,
1993                         .maxlen = NL80211_REPLAY_CTR_LEN,
1994                 },
1995         };
1996         union wpa_event_data data;
1997
1998         if (!tb[NL80211_ATTR_MAC])
1999                 return;
2000         if (!tb[NL80211_ATTR_REKEY_DATA])
2001                 return;
2002         if (nla_parse_nested(rekey_info, MAX_NL80211_REKEY_DATA,
2003                              tb[NL80211_ATTR_REKEY_DATA], rekey_policy))
2004                 return;
2005         if (!rekey_info[NL80211_REKEY_DATA_REPLAY_CTR])
2006                 return;
2007
2008         os_memset(&data, 0, sizeof(data));
2009         data.driver_gtk_rekey.bssid = nla_data(tb[NL80211_ATTR_MAC]);
2010         wpa_printf(MSG_DEBUG, "nl80211: Rekey offload event for BSSID " MACSTR,
2011                    MAC2STR(data.driver_gtk_rekey.bssid));
2012         data.driver_gtk_rekey.replay_ctr =
2013                 nla_data(rekey_info[NL80211_REKEY_DATA_REPLAY_CTR]);
2014         wpa_hexdump(MSG_DEBUG, "nl80211: Rekey offload - Replay Counter",
2015                     data.driver_gtk_rekey.replay_ctr, NL80211_REPLAY_CTR_LEN);
2016         wpa_supplicant_event(drv->ctx, EVENT_DRIVER_GTK_REKEY, &data);
2017 }
2018
2019
2020 static void nl80211_pmksa_candidate_event(struct wpa_driver_nl80211_data *drv,
2021                                           struct nlattr **tb)
2022 {
2023         struct nlattr *cand[NUM_NL80211_PMKSA_CANDIDATE];
2024         static struct nla_policy cand_policy[NUM_NL80211_PMKSA_CANDIDATE] = {
2025                 [NL80211_PMKSA_CANDIDATE_INDEX] = { .type = NLA_U32 },
2026                 [NL80211_PMKSA_CANDIDATE_BSSID] = {
2027                         .minlen = ETH_ALEN,
2028                         .maxlen = ETH_ALEN,
2029                 },
2030                 [NL80211_PMKSA_CANDIDATE_PREAUTH] = { .type = NLA_FLAG },
2031         };
2032         union wpa_event_data data;
2033
2034         if (!tb[NL80211_ATTR_PMKSA_CANDIDATE])
2035                 return;
2036         if (nla_parse_nested(cand, MAX_NL80211_PMKSA_CANDIDATE,
2037                              tb[NL80211_ATTR_PMKSA_CANDIDATE], cand_policy))
2038                 return;
2039         if (!cand[NL80211_PMKSA_CANDIDATE_INDEX] ||
2040             !cand[NL80211_PMKSA_CANDIDATE_BSSID])
2041                 return;
2042
2043         os_memset(&data, 0, sizeof(data));
2044         os_memcpy(data.pmkid_candidate.bssid,
2045                   nla_data(cand[NL80211_PMKSA_CANDIDATE_BSSID]), ETH_ALEN);
2046         data.pmkid_candidate.index =
2047                 nla_get_u32(cand[NL80211_PMKSA_CANDIDATE_INDEX]);
2048         data.pmkid_candidate.preauth =
2049                 cand[NL80211_PMKSA_CANDIDATE_PREAUTH] != NULL;
2050         wpa_supplicant_event(drv->ctx, EVENT_PMKID_CANDIDATE, &data);
2051 }
2052
2053
2054 static void nl80211_client_probe_event(struct wpa_driver_nl80211_data *drv,
2055                                        struct nlattr **tb)
2056 {
2057         union wpa_event_data data;
2058
2059         if (!tb[NL80211_ATTR_MAC] || !tb[NL80211_ATTR_ACK])
2060                 return;
2061
2062         os_memset(&data, 0, sizeof(data));
2063         os_memcpy(data.client_poll.addr,
2064                   nla_data(tb[NL80211_ATTR_MAC]), ETH_ALEN);
2065
2066         wpa_supplicant_event(drv->ctx, EVENT_DRIVER_CLIENT_POLL_OK, &data);
2067 }
2068
2069
2070 static void nl80211_spurious_frame(struct i802_bss *bss, struct nlattr **tb,
2071                                    int wds)
2072 {
2073         struct wpa_driver_nl80211_data *drv = bss->drv;
2074         union wpa_event_data event;
2075
2076         if (!tb[NL80211_ATTR_MAC])
2077                 return;
2078
2079         os_memset(&event, 0, sizeof(event));
2080         event.rx_from_unknown.bssid = bss->addr;
2081         event.rx_from_unknown.addr = nla_data(tb[NL80211_ATTR_MAC]);
2082         event.rx_from_unknown.wds = wds;
2083
2084         wpa_supplicant_event(drv->ctx, EVENT_RX_FROM_UNKNOWN, &event);
2085 }
2086
2087
2088 static void do_process_drv_event(struct wpa_driver_nl80211_data *drv,
2089                                  int cmd, struct nlattr **tb)
2090 {
2091         if (drv->ap_scan_as_station != NL80211_IFTYPE_UNSPECIFIED &&
2092             (cmd == NL80211_CMD_NEW_SCAN_RESULTS ||
2093              cmd == NL80211_CMD_SCAN_ABORTED)) {
2094                 wpa_driver_nl80211_set_mode(&drv->first_bss,
2095                                             drv->ap_scan_as_station);
2096                 drv->ap_scan_as_station = NL80211_IFTYPE_UNSPECIFIED;
2097         }
2098
2099         switch (cmd) {
2100         case NL80211_CMD_TRIGGER_SCAN:
2101                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger");
2102                 break;
2103         case NL80211_CMD_START_SCHED_SCAN:
2104                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan started");
2105                 break;
2106         case NL80211_CMD_SCHED_SCAN_STOPPED:
2107                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan stopped");
2108                 wpa_supplicant_event(drv->ctx, EVENT_SCHED_SCAN_STOPPED, NULL);
2109                 break;
2110         case NL80211_CMD_NEW_SCAN_RESULTS:
2111                 wpa_printf(MSG_DEBUG, "nl80211: New scan results available");
2112                 drv->scan_complete_events = 1;
2113                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
2114                                      drv->ctx);
2115                 send_scan_event(drv, 0, tb);
2116                 break;
2117         case NL80211_CMD_SCHED_SCAN_RESULTS:
2118                 wpa_printf(MSG_DEBUG,
2119                            "nl80211: New sched scan results available");
2120                 send_scan_event(drv, 0, tb);
2121                 break;
2122         case NL80211_CMD_SCAN_ABORTED:
2123                 wpa_printf(MSG_DEBUG, "nl80211: Scan aborted");
2124                 /*
2125                  * Need to indicate that scan results are available in order
2126                  * not to make wpa_supplicant stop its scanning.
2127                  */
2128                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
2129                                      drv->ctx);
2130                 send_scan_event(drv, 1, tb);
2131                 break;
2132         case NL80211_CMD_AUTHENTICATE:
2133         case NL80211_CMD_ASSOCIATE:
2134         case NL80211_CMD_DEAUTHENTICATE:
2135         case NL80211_CMD_DISASSOCIATE:
2136         case NL80211_CMD_FRAME_TX_STATUS:
2137         case NL80211_CMD_UNPROT_DEAUTHENTICATE:
2138         case NL80211_CMD_UNPROT_DISASSOCIATE:
2139                 mlme_event(drv, cmd, tb[NL80211_ATTR_FRAME],
2140                            tb[NL80211_ATTR_MAC], tb[NL80211_ATTR_TIMED_OUT],
2141                            tb[NL80211_ATTR_WIPHY_FREQ], tb[NL80211_ATTR_ACK],
2142                            tb[NL80211_ATTR_COOKIE],
2143                            tb[NL80211_ATTR_RX_SIGNAL_DBM]);
2144                 break;
2145         case NL80211_CMD_CONNECT:
2146         case NL80211_CMD_ROAM:
2147                 mlme_event_connect(drv, cmd,
2148                                    tb[NL80211_ATTR_STATUS_CODE],
2149                                    tb[NL80211_ATTR_MAC],
2150                                    tb[NL80211_ATTR_REQ_IE],
2151                                    tb[NL80211_ATTR_RESP_IE]);
2152                 break;
2153         case NL80211_CMD_CH_SWITCH_NOTIFY:
2154                 mlme_event_ch_switch(drv, tb[NL80211_ATTR_WIPHY_FREQ],
2155                                      tb[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
2156                 break;
2157         case NL80211_CMD_DISCONNECT:
2158                 mlme_event_disconnect(drv, tb[NL80211_ATTR_REASON_CODE],
2159                                       tb[NL80211_ATTR_MAC],
2160                                       tb[NL80211_ATTR_DISCONNECTED_BY_AP]);
2161                 break;
2162         case NL80211_CMD_MICHAEL_MIC_FAILURE:
2163                 mlme_event_michael_mic_failure(drv, tb);
2164                 break;
2165         case NL80211_CMD_JOIN_IBSS:
2166                 mlme_event_join_ibss(drv, tb);
2167                 break;
2168         case NL80211_CMD_REMAIN_ON_CHANNEL:
2169                 mlme_event_remain_on_channel(drv, 0, tb);
2170                 break;
2171         case NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL:
2172                 mlme_event_remain_on_channel(drv, 1, tb);
2173                 break;
2174         case NL80211_CMD_NOTIFY_CQM:
2175                 nl80211_cqm_event(drv, tb);
2176                 break;
2177         case NL80211_CMD_REG_CHANGE:
2178                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory domain change");
2179                 wpa_supplicant_event(drv->ctx, EVENT_CHANNEL_LIST_CHANGED,
2180                                      NULL);
2181                 break;
2182         case NL80211_CMD_REG_BEACON_HINT:
2183                 wpa_printf(MSG_DEBUG, "nl80211: Regulatory beacon hint");
2184                 wpa_supplicant_event(drv->ctx, EVENT_CHANNEL_LIST_CHANGED,
2185                                      NULL);
2186                 break;
2187         case NL80211_CMD_NEW_STATION:
2188                 nl80211_new_station_event(drv, tb);
2189                 break;
2190         case NL80211_CMD_DEL_STATION:
2191                 nl80211_del_station_event(drv, tb);
2192                 break;
2193         case NL80211_CMD_SET_REKEY_OFFLOAD:
2194                 nl80211_rekey_offload_event(drv, tb);
2195                 break;
2196         case NL80211_CMD_PMKSA_CANDIDATE:
2197                 nl80211_pmksa_candidate_event(drv, tb);
2198                 break;
2199         case NL80211_CMD_PROBE_CLIENT:
2200                 nl80211_client_probe_event(drv, tb);
2201                 break;
2202         default:
2203                 wpa_printf(MSG_DEBUG, "nl80211: Ignored unknown event "
2204                            "(cmd=%d)", cmd);
2205                 break;
2206         }
2207 }
2208
2209
2210 static int process_drv_event(struct nl_msg *msg, void *arg)
2211 {
2212         struct wpa_driver_nl80211_data *drv = arg;
2213         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2214         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2215
2216         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2217                   genlmsg_attrlen(gnlh, 0), NULL);
2218
2219         if (tb[NL80211_ATTR_IFINDEX]) {
2220                 int ifindex = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
2221                 if (ifindex != drv->ifindex && !have_ifidx(drv, ifindex)) {
2222                         wpa_printf(MSG_DEBUG, "nl80211: Ignored event (cmd=%d)"
2223                                    " for foreign interface (ifindex %d)",
2224                                    gnlh->cmd, ifindex);
2225                         return NL_SKIP;
2226                 }
2227         }
2228
2229         do_process_drv_event(drv, gnlh->cmd, tb);
2230         return NL_SKIP;
2231 }
2232
2233
2234 static int process_global_event(struct nl_msg *msg, void *arg)
2235 {
2236         struct nl80211_global *global = arg;
2237         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2238         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2239         struct wpa_driver_nl80211_data *drv, *tmp;
2240         int ifidx = -1;
2241
2242         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2243                   genlmsg_attrlen(gnlh, 0), NULL);
2244
2245         if (tb[NL80211_ATTR_IFINDEX])
2246                 ifidx = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
2247
2248         dl_list_for_each_safe(drv, tmp, &global->interfaces,
2249                               struct wpa_driver_nl80211_data, list) {
2250                 if (ifidx == -1 || ifidx == drv->ifindex ||
2251                     have_ifidx(drv, ifidx))
2252                         do_process_drv_event(drv, gnlh->cmd, tb);
2253         }
2254
2255         return NL_SKIP;
2256 }
2257
2258
2259 static int process_bss_event(struct nl_msg *msg, void *arg)
2260 {
2261         struct i802_bss *bss = arg;
2262         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2263         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2264
2265         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2266                   genlmsg_attrlen(gnlh, 0), NULL);
2267
2268         switch (gnlh->cmd) {
2269         case NL80211_CMD_FRAME:
2270         case NL80211_CMD_FRAME_TX_STATUS:
2271                 mlme_event(bss->drv, gnlh->cmd, tb[NL80211_ATTR_FRAME],
2272                            tb[NL80211_ATTR_MAC], tb[NL80211_ATTR_TIMED_OUT],
2273                            tb[NL80211_ATTR_WIPHY_FREQ], tb[NL80211_ATTR_ACK],
2274                            tb[NL80211_ATTR_COOKIE],
2275                            tb[NL80211_ATTR_RX_SIGNAL_DBM]);
2276                 break;
2277         case NL80211_CMD_UNEXPECTED_FRAME:
2278                 nl80211_spurious_frame(bss, tb, 0);
2279                 break;
2280         case NL80211_CMD_UNEXPECTED_4ADDR_FRAME:
2281                 nl80211_spurious_frame(bss, tb, 1);
2282                 break;
2283         default:
2284                 wpa_printf(MSG_DEBUG, "nl80211: Ignored unknown event "
2285                            "(cmd=%d)", gnlh->cmd);
2286                 break;
2287         }
2288
2289         return NL_SKIP;
2290 }
2291
2292
2293 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
2294                                              void *handle)
2295 {
2296         struct nl_cb *cb = eloop_ctx;
2297
2298         wpa_printf(MSG_DEBUG, "nl80211: Event message available");
2299
2300         nl_recvmsgs(handle, cb);
2301 }
2302
2303
2304 /**
2305  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
2306  * @priv: driver_nl80211 private data
2307  * @alpha2_arg: country to which to switch to
2308  * Returns: 0 on success, -1 on failure
2309  *
2310  * This asks nl80211 to set the regulatory domain for given
2311  * country ISO / IEC alpha2.
2312  */
2313 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
2314 {
2315         struct i802_bss *bss = priv;
2316         struct wpa_driver_nl80211_data *drv = bss->drv;
2317         char alpha2[3];
2318         struct nl_msg *msg;
2319
2320         msg = nlmsg_alloc();
2321         if (!msg)
2322                 return -ENOMEM;
2323
2324         alpha2[0] = alpha2_arg[0];
2325         alpha2[1] = alpha2_arg[1];
2326         alpha2[2] = '\0';
2327
2328         nl80211_cmd(drv, msg, 0, NL80211_CMD_REQ_SET_REG);
2329
2330         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
2331         if (send_and_recv_msgs(drv, msg, NULL, NULL))
2332                 return -EINVAL;
2333         return 0;
2334 nla_put_failure:
2335         nlmsg_free(msg);
2336         return -EINVAL;
2337 }
2338
2339
2340 struct wiphy_info_data {
2341         struct wpa_driver_capa *capa;
2342
2343         unsigned int error:1;
2344         unsigned int device_ap_sme:1;
2345         unsigned int poll_command_supported:1;
2346         unsigned int data_tx_status:1;
2347         unsigned int monitor_supported:1;
2348 };
2349
2350
2351 static unsigned int probe_resp_offload_support(int supp_protocols)
2352 {
2353         unsigned int prot = 0;
2354
2355         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS)
2356                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS;
2357         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_WPS2)
2358                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_WPS2;
2359         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_P2P)
2360                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_P2P;
2361         if (supp_protocols & NL80211_PROBE_RESP_OFFLOAD_SUPPORT_80211U)
2362                 prot |= WPA_DRIVER_PROBE_RESP_OFFLOAD_INTERWORKING;
2363
2364         return prot;
2365 }
2366
2367
2368 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
2369 {
2370         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2371         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2372         struct wiphy_info_data *info = arg;
2373         int p2p_go_supported = 0, p2p_client_supported = 0;
2374         int p2p_concurrent = 0;
2375         int auth_supported = 0, connect_supported = 0;
2376         struct wpa_driver_capa *capa = info->capa;
2377         static struct nla_policy
2378         iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
2379                 [NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
2380                 [NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
2381                 [NL80211_IFACE_COMB_STA_AP_BI_MATCH] = { .type = NLA_FLAG },
2382                 [NL80211_IFACE_COMB_NUM_CHANNELS] = { .type = NLA_U32 },
2383         },
2384         iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
2385                 [NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
2386                 [NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
2387         };
2388
2389         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2390                   genlmsg_attrlen(gnlh, 0), NULL);
2391
2392         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
2393                 capa->max_scan_ssids =
2394                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
2395
2396         if (tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS])
2397                 capa->max_sched_scan_ssids =
2398                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCHED_SCAN_SSIDS]);
2399
2400         if (tb[NL80211_ATTR_MAX_MATCH_SETS])
2401                 capa->max_match_sets =
2402                         nla_get_u8(tb[NL80211_ATTR_MAX_MATCH_SETS]);
2403
2404         if (tb[NL80211_ATTR_SUPPORTED_IFTYPES]) {
2405                 struct nlattr *nl_mode;
2406                 int i;
2407                 nla_for_each_nested(nl_mode,
2408                                     tb[NL80211_ATTR_SUPPORTED_IFTYPES], i) {
2409                         switch (nla_type(nl_mode)) {
2410                         case NL80211_IFTYPE_AP:
2411                                 capa->flags |= WPA_DRIVER_FLAGS_AP;
2412                                 break;
2413                         case NL80211_IFTYPE_P2P_GO:
2414                                 p2p_go_supported = 1;
2415                                 break;
2416                         case NL80211_IFTYPE_P2P_CLIENT:
2417                                 p2p_client_supported = 1;
2418                                 break;
2419                         case NL80211_IFTYPE_MONITOR:
2420                                 info->monitor_supported = 1;
2421                                 break;
2422                         }
2423                 }
2424         }
2425
2426         if (tb[NL80211_ATTR_INTERFACE_COMBINATIONS]) {
2427                 struct nlattr *nl_combi;
2428                 int rem_combi;
2429
2430                 nla_for_each_nested(nl_combi,
2431                                     tb[NL80211_ATTR_INTERFACE_COMBINATIONS],
2432                                     rem_combi) {
2433                         struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB];
2434                         struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT];
2435                         struct nlattr *nl_limit, *nl_mode;
2436                         int err, rem_limit, rem_mode;
2437                         int combination_has_p2p = 0, combination_has_mgd = 0;
2438
2439                         err = nla_parse_nested(tb_comb, MAX_NL80211_IFACE_COMB,
2440                                                nl_combi,
2441                                                iface_combination_policy);
2442                         if (err || !tb_comb[NL80211_IFACE_COMB_LIMITS] ||
2443                             !tb_comb[NL80211_IFACE_COMB_MAXNUM] ||
2444                             !tb_comb[NL80211_IFACE_COMB_NUM_CHANNELS])
2445                                 goto broken_combination;
2446
2447                         nla_for_each_nested(nl_limit,
2448                                             tb_comb[NL80211_IFACE_COMB_LIMITS],
2449                                             rem_limit) {
2450                                 err = nla_parse_nested(tb_limit,
2451                                                        MAX_NL80211_IFACE_LIMIT,
2452                                                        nl_limit,
2453                                                        iface_limit_policy);
2454                                 if (err ||
2455                                     !tb_limit[NL80211_IFACE_LIMIT_TYPES])
2456                                         goto broken_combination;
2457
2458                                 nla_for_each_nested(
2459                                         nl_mode,
2460                                         tb_limit[NL80211_IFACE_LIMIT_TYPES],
2461                                         rem_mode) {
2462                                         int ift = nla_type(nl_mode);
2463                                         if (ift == NL80211_IFTYPE_P2P_GO ||
2464                                             ift == NL80211_IFTYPE_P2P_CLIENT)
2465                                                 combination_has_p2p = 1;
2466                                         if (ift == NL80211_IFTYPE_STATION)
2467                                                 combination_has_mgd = 1;
2468                                 }
2469                                 if (combination_has_p2p && combination_has_mgd)
2470                                         break;
2471                         }
2472
2473                         if (combination_has_p2p && combination_has_mgd) {
2474                                 p2p_concurrent = 1;
2475                                 break;
2476                         }
2477
2478 broken_combination:
2479                         ;
2480                 }
2481         }
2482
2483         if (tb[NL80211_ATTR_SUPPORTED_COMMANDS]) {
2484                 struct nlattr *nl_cmd;
2485                 int i;
2486
2487                 nla_for_each_nested(nl_cmd,
2488                                     tb[NL80211_ATTR_SUPPORTED_COMMANDS], i) {
2489                         switch (nla_get_u32(nl_cmd)) {
2490                         case NL80211_CMD_AUTHENTICATE:
2491                                 auth_supported = 1;
2492                                 break;
2493                         case NL80211_CMD_CONNECT:
2494                                 connect_supported = 1;
2495                                 break;
2496                         case NL80211_CMD_START_SCHED_SCAN:
2497                                 capa->sched_scan_supported = 1;
2498                                 break;
2499                         case NL80211_CMD_PROBE_CLIENT:
2500                                 info->poll_command_supported = 1;
2501                                 break;
2502                         }
2503                 }
2504         }
2505
2506         if (tb[NL80211_ATTR_OFFCHANNEL_TX_OK]) {
2507                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based "
2508                            "off-channel TX");
2509                 capa->flags |= WPA_DRIVER_FLAGS_OFFCHANNEL_TX;
2510         }
2511
2512         if (tb[NL80211_ATTR_ROAM_SUPPORT]) {
2513                 wpa_printf(MSG_DEBUG, "nl80211: Using driver-based roaming");
2514                 capa->flags |= WPA_DRIVER_FLAGS_BSS_SELECTION;
2515         }
2516
2517         /* default to 5000 since early versions of mac80211 don't set it */
2518         capa->max_remain_on_chan = 5000;
2519
2520         if (tb[NL80211_ATTR_SUPPORT_AP_UAPSD])
2521                 capa->flags |= WPA_DRIVER_FLAGS_AP_UAPSD;
2522
2523         if (tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION])
2524                 capa->max_remain_on_chan =
2525                         nla_get_u32(tb[NL80211_ATTR_MAX_REMAIN_ON_CHANNEL_DURATION]);
2526
2527         if (auth_supported)
2528                 capa->flags |= WPA_DRIVER_FLAGS_SME;
2529         else if (!connect_supported) {
2530                 wpa_printf(MSG_INFO, "nl80211: Driver does not support "
2531                            "authentication/association or connect commands");
2532                 info->error = 1;
2533         }
2534
2535         if (p2p_go_supported && p2p_client_supported)
2536                 capa->flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
2537         if (p2p_concurrent) {
2538                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
2539                            "interface (driver advertised support)");
2540                 capa->flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
2541                 capa->flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
2542         }
2543
2544         if (tb[NL80211_ATTR_TDLS_SUPPORT]) {
2545                 wpa_printf(MSG_DEBUG, "nl80211: TDLS supported");
2546                 capa->flags |= WPA_DRIVER_FLAGS_TDLS_SUPPORT;
2547
2548                 if (tb[NL80211_ATTR_TDLS_EXTERNAL_SETUP]) {
2549                         wpa_printf(MSG_DEBUG, "nl80211: TDLS external setup");
2550                         capa->flags |=
2551                                 WPA_DRIVER_FLAGS_TDLS_EXTERNAL_SETUP;
2552                 }
2553         }
2554
2555         if (tb[NL80211_ATTR_DEVICE_AP_SME])
2556                 info->device_ap_sme = 1;
2557
2558         if (tb[NL80211_ATTR_FEATURE_FLAGS]) {
2559                 u32 flags = nla_get_u32(tb[NL80211_ATTR_FEATURE_FLAGS]);
2560
2561                 if (flags & NL80211_FEATURE_SK_TX_STATUS)
2562                         info->data_tx_status = 1;
2563
2564                 if (flags & NL80211_FEATURE_INACTIVITY_TIMER)
2565                         capa->flags |= WPA_DRIVER_FLAGS_INACTIVITY_TIMER;
2566         }
2567
2568         if (tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]) {
2569                 int protocols =
2570                         nla_get_u32(tb[NL80211_ATTR_PROBE_RESP_OFFLOAD]);
2571                 wpa_printf(MSG_DEBUG, "nl80211: Supports Probe Response "
2572                            "offload in AP mode");
2573                 capa->flags |= WPA_DRIVER_FLAGS_PROBE_RESP_OFFLOAD;
2574                 capa->probe_resp_offloads =
2575                         probe_resp_offload_support(protocols);
2576         }
2577
2578         return NL_SKIP;
2579 }
2580
2581
2582 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
2583                                        struct wiphy_info_data *info)
2584 {
2585         struct nl_msg *msg;
2586
2587         os_memset(info, 0, sizeof(*info));
2588         info->capa = &drv->capa;
2589
2590         msg = nlmsg_alloc();
2591         if (!msg)
2592                 return -1;
2593
2594         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_WIPHY);
2595
2596         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->first_bss.ifindex);
2597
2598         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info) == 0)
2599                 return 0;
2600         msg = NULL;
2601 nla_put_failure:
2602         nlmsg_free(msg);
2603         return -1;
2604 }
2605
2606
2607 static int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
2608 {
2609         struct wiphy_info_data info;
2610         if (wpa_driver_nl80211_get_info(drv, &info))
2611                 return -1;
2612
2613         if (info.error)
2614                 return -1;
2615
2616         drv->has_capability = 1;
2617         /* For now, assume TKIP, CCMP, WPA, WPA2 are supported */
2618         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
2619                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
2620                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
2621                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
2622         drv->capa.enc = WPA_DRIVER_CAPA_ENC_WEP40 |
2623                 WPA_DRIVER_CAPA_ENC_WEP104 |
2624                 WPA_DRIVER_CAPA_ENC_TKIP |
2625                 WPA_DRIVER_CAPA_ENC_CCMP;
2626         drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
2627                 WPA_DRIVER_AUTH_SHARED |
2628                 WPA_DRIVER_AUTH_LEAP;
2629
2630         drv->capa.flags |= WPA_DRIVER_FLAGS_SANE_ERROR_CODES;
2631         drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
2632         drv->capa.flags |= WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
2633
2634         if (!info.device_ap_sme) {
2635                 drv->capa.flags |= WPA_DRIVER_FLAGS_DEAUTH_TX_STATUS;
2636
2637                 /*
2638                  * No AP SME is currently assumed to also indicate no AP MLME
2639                  * in the driver/firmware.
2640                  */
2641                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP_MLME;
2642         }
2643
2644         drv->device_ap_sme = info.device_ap_sme;
2645         drv->poll_command_supported = info.poll_command_supported;
2646         drv->data_tx_status = info.data_tx_status;
2647
2648         /*
2649          * If poll command is supported mac80211 is new enough to
2650          * have everything we need to not need monitor interfaces.
2651          */
2652         drv->use_monitor = !info.poll_command_supported;
2653
2654         if (drv->device_ap_sme && drv->use_monitor) {
2655                 /*
2656                  * Non-mac80211 drivers may not support monitor interface.
2657                  * Make sure we do not get stuck with incorrect capability here
2658                  * by explicitly testing this.
2659                  */
2660                 if (!info.monitor_supported) {
2661                         wpa_printf(MSG_DEBUG, "nl80211: Disable use_monitor "
2662                                    "with device_ap_sme since no monitor mode "
2663                                    "support detected");
2664                         drv->use_monitor = 0;
2665                 }
2666         }
2667
2668         /*
2669          * If we aren't going to use monitor interfaces, but the
2670          * driver doesn't support data TX status, we won't get TX
2671          * status for EAPOL frames.
2672          */
2673         if (!drv->use_monitor && !info.data_tx_status)
2674                 drv->capa.flags &= ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
2675
2676         return 0;
2677 }
2678
2679
2680 #ifdef ANDROID
2681 static int android_genl_ctrl_resolve(struct nl_handle *handle,
2682                                      const char *name)
2683 {
2684         /*
2685          * Android ICS has very minimal genl_ctrl_resolve() implementation, so
2686          * need to work around that.
2687          */
2688         struct nl_cache *cache = NULL;
2689         struct genl_family *nl80211 = NULL;
2690         int id = -1;
2691
2692         if (genl_ctrl_alloc_cache(handle, &cache) < 0) {
2693                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
2694                            "netlink cache");
2695                 goto fail;
2696         }
2697
2698         nl80211 = genl_ctrl_search_by_name(cache, name);
2699         if (nl80211 == NULL)
2700                 goto fail;
2701
2702         id = genl_family_get_id(nl80211);
2703
2704 fail:
2705         if (nl80211)
2706                 genl_family_put(nl80211);
2707         if (cache)
2708                 nl_cache_free(cache);
2709
2710         return id;
2711 }
2712 #define genl_ctrl_resolve android_genl_ctrl_resolve
2713 #endif /* ANDROID */
2714
2715
2716 static int wpa_driver_nl80211_init_nl_global(struct nl80211_global *global)
2717 {
2718         int ret;
2719
2720         global->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
2721         if (global->nl_cb == NULL) {
2722                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
2723                            "callbacks");
2724                 return -1;
2725         }
2726
2727         global->nl = nl_create_handle(global->nl_cb, "nl");
2728         if (global->nl == NULL)
2729                 goto err;
2730
2731         global->nl80211_id = genl_ctrl_resolve(global->nl, "nl80211");
2732         if (global->nl80211_id < 0) {
2733                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
2734                            "found");
2735                 goto err;
2736         }
2737
2738         global->nl_event = nl_create_handle(global->nl_cb, "event");
2739         if (global->nl_event == NULL)
2740                 goto err;
2741
2742         ret = nl_get_multicast_id(global, "nl80211", "scan");
2743         if (ret >= 0)
2744                 ret = nl_socket_add_membership(global->nl_event, ret);
2745         if (ret < 0) {
2746                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
2747                            "membership for scan events: %d (%s)",
2748                            ret, strerror(-ret));
2749                 goto err;
2750         }
2751
2752         ret = nl_get_multicast_id(global, "nl80211", "mlme");
2753         if (ret >= 0)
2754                 ret = nl_socket_add_membership(global->nl_event, ret);
2755         if (ret < 0) {
2756                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
2757                            "membership for mlme events: %d (%s)",
2758                            ret, strerror(-ret));
2759                 goto err;
2760         }
2761
2762         ret = nl_get_multicast_id(global, "nl80211", "regulatory");
2763         if (ret >= 0)
2764                 ret = nl_socket_add_membership(global->nl_event, ret);
2765         if (ret < 0) {
2766                 wpa_printf(MSG_DEBUG, "nl80211: Could not add multicast "
2767                            "membership for regulatory events: %d (%s)",
2768                            ret, strerror(-ret));
2769                 /* Continue without regulatory events */
2770         }
2771
2772         nl_cb_set(global->nl_cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM,
2773                   no_seq_check, NULL);
2774         nl_cb_set(global->nl_cb, NL_CB_VALID, NL_CB_CUSTOM,
2775                   process_global_event, global);
2776
2777         eloop_register_read_sock(nl_socket_get_fd(global->nl_event),
2778                                  wpa_driver_nl80211_event_receive,
2779                                  global->nl_cb, global->nl_event);
2780
2781         return 0;
2782
2783 err:
2784         nl_destroy_handles(&global->nl_event);
2785         nl_destroy_handles(&global->nl);
2786         nl_cb_put(global->nl_cb);
2787         global->nl_cb = NULL;
2788         return -1;
2789 }
2790
2791
2792 static int wpa_driver_nl80211_init_nl(struct wpa_driver_nl80211_data *drv)
2793 {
2794         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
2795         if (!drv->nl_cb) {
2796                 wpa_printf(MSG_ERROR, "nl80211: Failed to alloc cb struct");
2797                 return -1;
2798         }
2799
2800         nl_cb_set(drv->nl_cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM,
2801                   no_seq_check, NULL);
2802         nl_cb_set(drv->nl_cb, NL_CB_VALID, NL_CB_CUSTOM,
2803                   process_drv_event, drv);
2804
2805         return 0;
2806 }
2807
2808
2809 static void wpa_driver_nl80211_rfkill_blocked(void *ctx)
2810 {
2811         wpa_printf(MSG_DEBUG, "nl80211: RFKILL blocked");
2812         /*
2813          * This may be for any interface; use ifdown event to disable
2814          * interface.
2815          */
2816 }
2817
2818
2819 static void wpa_driver_nl80211_rfkill_unblocked(void *ctx)
2820 {
2821         struct wpa_driver_nl80211_data *drv = ctx;
2822         wpa_printf(MSG_DEBUG, "nl80211: RFKILL unblocked");
2823         if (linux_set_iface_flags(drv->global->ioctl_sock,
2824                                   drv->first_bss.ifname, 1)) {
2825                 wpa_printf(MSG_DEBUG, "nl80211: Could not set interface UP "
2826                            "after rfkill unblock");
2827                 return;
2828         }
2829         /* rtnetlink ifup handler will report interface as enabled */
2830 }
2831
2832
2833 static void nl80211_get_phy_name(struct wpa_driver_nl80211_data *drv)
2834 {
2835         /* Find phy (radio) to which this interface belongs */
2836         char buf[90], *pos;
2837         int f, rv;
2838
2839         drv->phyname[0] = '\0';
2840         snprintf(buf, sizeof(buf) - 1, "/sys/class/net/%s/phy80211/name",
2841                  drv->first_bss.ifname);
2842         f = open(buf, O_RDONLY);
2843         if (f < 0) {
2844                 wpa_printf(MSG_DEBUG, "Could not open file %s: %s",
2845                            buf, strerror(errno));
2846                 return;
2847         }
2848
2849         rv = read(f, drv->phyname, sizeof(drv->phyname) - 1);
2850         close(f);
2851         if (rv < 0) {
2852                 wpa_printf(MSG_DEBUG, "Could not read file %s: %s",
2853                            buf, strerror(errno));
2854                 return;
2855         }
2856
2857         drv->phyname[rv] = '\0';
2858         pos = os_strchr(drv->phyname, '\n');
2859         if (pos)
2860                 *pos = '\0';
2861         wpa_printf(MSG_DEBUG, "nl80211: interface %s in phy %s",
2862                    drv->first_bss.ifname, drv->phyname);
2863 }
2864
2865
2866 static void wpa_driver_nl80211_handle_eapol_tx_status(int sock,
2867                                                       void *eloop_ctx,
2868                                                       void *handle)
2869 {
2870         struct wpa_driver_nl80211_data *drv = eloop_ctx;
2871         u8 data[2048];
2872         struct msghdr msg;
2873         struct iovec entry;
2874         u8 control[512];
2875         struct cmsghdr *cmsg;
2876         int res, found_ee = 0, found_wifi = 0, acked = 0;
2877         union wpa_event_data event;
2878
2879         memset(&msg, 0, sizeof(msg));
2880         msg.msg_iov = &entry;
2881         msg.msg_iovlen = 1;
2882         entry.iov_base = data;
2883         entry.iov_len = sizeof(data);
2884         msg.msg_control = &control;
2885         msg.msg_controllen = sizeof(control);
2886
2887         res = recvmsg(sock, &msg, MSG_ERRQUEUE);
2888         /* if error or not fitting 802.3 header, return */
2889         if (res < 14)
2890                 return;
2891
2892         for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg))
2893         {
2894                 if (cmsg->cmsg_level == SOL_SOCKET &&
2895                     cmsg->cmsg_type == SCM_WIFI_STATUS) {
2896                         int *ack;
2897
2898                         found_wifi = 1;
2899                         ack = (void *)CMSG_DATA(cmsg);
2900                         acked = *ack;
2901                 }
2902
2903                 if (cmsg->cmsg_level == SOL_PACKET &&
2904                     cmsg->cmsg_type == PACKET_TX_TIMESTAMP) {
2905                         struct sock_extended_err *err =
2906                                 (struct sock_extended_err *)CMSG_DATA(cmsg);
2907
2908                         if (err->ee_origin == SO_EE_ORIGIN_TXSTATUS)
2909                                 found_ee = 1;
2910                 }
2911         }
2912
2913         if (!found_ee || !found_wifi)
2914                 return;
2915
2916         memset(&event, 0, sizeof(event));
2917         event.eapol_tx_status.dst = data;
2918         event.eapol_tx_status.data = data + 14;
2919         event.eapol_tx_status.data_len = res - 14;
2920         event.eapol_tx_status.ack = acked;
2921         wpa_supplicant_event(drv->ctx, EVENT_EAPOL_TX_STATUS, &event);
2922 }
2923
2924
2925 static int nl80211_init_bss(struct i802_bss *bss)
2926 {
2927         bss->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
2928         if (!bss->nl_cb)
2929                 return -1;
2930
2931         nl_cb_set(bss->nl_cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM,
2932                   no_seq_check, NULL);
2933         nl_cb_set(bss->nl_cb, NL_CB_VALID, NL_CB_CUSTOM,
2934                   process_bss_event, bss);
2935
2936         return 0;
2937 }
2938
2939
2940 static void nl80211_destroy_bss(struct i802_bss *bss)
2941 {
2942         nl_cb_put(bss->nl_cb);
2943         bss->nl_cb = NULL;
2944 }
2945
2946
2947 /**
2948  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
2949  * @ctx: context to be used when calling wpa_supplicant functions,
2950  * e.g., wpa_supplicant_event()
2951  * @ifname: interface name, e.g., wlan0
2952  * @global_priv: private driver global data from global_init()
2953  * Returns: Pointer to private data, %NULL on failure
2954  */
2955 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname,
2956                                       void *global_priv)
2957 {
2958         struct wpa_driver_nl80211_data *drv;
2959         struct rfkill_config *rcfg;
2960         struct i802_bss *bss;
2961
2962         if (global_priv == NULL)
2963                 return NULL;
2964         drv = os_zalloc(sizeof(*drv));
2965         if (drv == NULL)
2966                 return NULL;
2967         drv->global = global_priv;
2968         drv->ctx = ctx;
2969         bss = &drv->first_bss;
2970         bss->drv = drv;
2971         os_strlcpy(bss->ifname, ifname, sizeof(bss->ifname));
2972         drv->monitor_ifidx = -1;
2973         drv->monitor_sock = -1;
2974         drv->eapol_tx_sock = -1;
2975         drv->ap_scan_as_station = NL80211_IFTYPE_UNSPECIFIED;
2976
2977         if (wpa_driver_nl80211_init_nl(drv)) {
2978                 os_free(drv);
2979                 return NULL;
2980         }
2981
2982         if (nl80211_init_bss(bss))
2983                 goto failed;
2984
2985         nl80211_get_phy_name(drv);
2986
2987         rcfg = os_zalloc(sizeof(*rcfg));
2988         if (rcfg == NULL)
2989                 goto failed;
2990         rcfg->ctx = drv;
2991         os_strlcpy(rcfg->ifname, ifname, sizeof(rcfg->ifname));
2992         rcfg->blocked_cb = wpa_driver_nl80211_rfkill_blocked;
2993         rcfg->unblocked_cb = wpa_driver_nl80211_rfkill_unblocked;
2994         drv->rfkill = rfkill_init(rcfg);
2995         if (drv->rfkill == NULL) {
2996                 wpa_printf(MSG_DEBUG, "nl80211: RFKILL status not available");
2997                 os_free(rcfg);
2998         }
2999
3000         if (wpa_driver_nl80211_finish_drv_init(drv))
3001                 goto failed;
3002
3003         drv->eapol_tx_sock = socket(PF_PACKET, SOCK_DGRAM, 0);
3004         if (drv->eapol_tx_sock < 0)
3005                 goto failed;
3006
3007         if (drv->data_tx_status) {
3008                 int enabled = 1;
3009
3010                 if (setsockopt(drv->eapol_tx_sock, SOL_SOCKET, SO_WIFI_STATUS,
3011                                &enabled, sizeof(enabled)) < 0) {
3012                         wpa_printf(MSG_DEBUG,
3013                                 "nl80211: wifi status sockopt failed\n");
3014                         drv->data_tx_status = 0;
3015                         if (!drv->use_monitor)
3016                                 drv->capa.flags &=
3017                                         ~WPA_DRIVER_FLAGS_EAPOL_TX_STATUS;
3018                 } else {
3019                         eloop_register_read_sock(drv->eapol_tx_sock,
3020                                 wpa_driver_nl80211_handle_eapol_tx_status,
3021                                 drv, NULL);
3022                 }
3023         }
3024
3025         if (drv->global) {
3026                 dl_list_add(&drv->global->interfaces, &drv->list);
3027                 drv->in_interface_list = 1;
3028         }
3029
3030         return bss;
3031
3032 failed:
3033         wpa_driver_nl80211_deinit(bss);
3034         return NULL;
3035 }
3036
3037
3038 static int nl80211_register_frame(struct i802_bss *bss,
3039                                   struct nl_handle *nl_handle,
3040                                   u16 type, const u8 *match, size_t match_len)
3041 {
3042         struct wpa_driver_nl80211_data *drv = bss->drv;
3043         struct nl_msg *msg;
3044         int ret = -1;
3045
3046         msg = nlmsg_alloc();
3047         if (!msg)
3048                 return -1;
3049
3050         wpa_printf(MSG_DEBUG, "nl80211: Register frame type=0x%x nl_handle=%p",
3051                    type, nl_handle);
3052         wpa_hexdump(MSG_DEBUG, "nl80211: Register frame match",
3053                     match, match_len);
3054
3055         nl80211_cmd(drv, msg, 0, NL80211_CMD_REGISTER_ACTION);
3056
3057         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
3058         NLA_PUT_U16(msg, NL80211_ATTR_FRAME_TYPE, type);
3059         NLA_PUT(msg, NL80211_ATTR_FRAME_MATCH, match_len, match);
3060
3061         ret = send_and_recv(drv->global, nl_handle, msg, NULL, NULL);
3062         msg = NULL;
3063         if (ret) {
3064                 wpa_printf(MSG_DEBUG, "nl80211: Register frame command "
3065                            "failed (type=%u): ret=%d (%s)",
3066                            type, ret, strerror(-ret));
3067                 wpa_hexdump(MSG_DEBUG, "nl80211: Register frame match",
3068                             match, match_len);
3069                 goto nla_put_failure;
3070         }
3071         ret = 0;
3072 nla_put_failure:
3073         nlmsg_free(msg);
3074         return ret;
3075 }
3076
3077
3078 static int nl80211_alloc_mgmt_handle(struct i802_bss *bss)
3079 {
3080         struct wpa_driver_nl80211_data *drv = bss->drv;
3081
3082         if (bss->nl_mgmt) {
3083                 wpa_printf(MSG_DEBUG, "nl80211: Mgmt reporting "
3084                            "already on! (nl_mgmt=%p)", bss->nl_mgmt);
3085                 return -1;
3086         }
3087
3088         bss->nl_mgmt = nl_create_handle(drv->nl_cb, "mgmt");
3089         if (bss->nl_mgmt == NULL)
3090                 return -1;
3091
3092         eloop_register_read_sock(nl_socket_get_fd(bss->nl_mgmt),
3093                                  wpa_driver_nl80211_event_receive, bss->nl_cb,
3094                                  bss->nl_mgmt);
3095
3096         return 0;
3097 }
3098
3099
3100 static int nl80211_register_action_frame(struct i802_bss *bss,
3101                                          const u8 *match, size_t match_len)
3102 {
3103         u16 type = (WLAN_FC_TYPE_MGMT << 2) | (WLAN_FC_STYPE_ACTION << 4);
3104         return nl80211_register_frame(bss, bss->nl_mgmt,
3105                                       type, match, match_len);
3106 }
3107
3108
3109 static int nl80211_mgmt_subscribe_non_ap(struct i802_bss *bss)
3110 {
3111         struct wpa_driver_nl80211_data *drv = bss->drv;
3112
3113         if (nl80211_alloc_mgmt_handle(bss))
3114                 return -1;
3115         wpa_printf(MSG_DEBUG, "nl80211: Subscribe to mgmt frames with non-AP "
3116                    "handle %p", bss->nl_mgmt);
3117
3118 #if defined(CONFIG_P2P) || defined(CONFIG_INTERWORKING)
3119         /* GAS Initial Request */
3120         if (nl80211_register_action_frame(bss, (u8 *) "\x04\x0a", 2) < 0)
3121                 return -1;
3122         /* GAS Initial Response */
3123         if (nl80211_register_action_frame(bss, (u8 *) "\x04\x0b", 2) < 0)
3124                 return -1;
3125         /* GAS Comeback Request */
3126         if (nl80211_register_action_frame(bss, (u8 *) "\x04\x0c", 2) < 0)
3127                 return -1;
3128         /* GAS Comeback Response */
3129         if (nl80211_register_action_frame(bss, (u8 *) "\x04\x0d", 2) < 0)
3130                 return -1;
3131 #endif /* CONFIG_P2P || CONFIG_INTERWORKING */
3132 #ifdef CONFIG_P2P
3133         /* P2P Public Action */
3134         if (nl80211_register_action_frame(bss,
3135                                           (u8 *) "\x04\x09\x50\x6f\x9a\x09",
3136                                           6) < 0)
3137                 return -1;
3138         /* P2P Action */
3139         if (nl80211_register_action_frame(bss,
3140                                           (u8 *) "\x7f\x50\x6f\x9a\x09",
3141                                           5) < 0)
3142                 return -1;
3143 #endif /* CONFIG_P2P */
3144 #ifdef CONFIG_IEEE80211W
3145         /* SA Query Response */
3146         if (nl80211_register_action_frame(bss, (u8 *) "\x08\x01", 2) < 0)
3147                 return -1;
3148 #endif /* CONFIG_IEEE80211W */
3149 #ifdef CONFIG_TDLS
3150         if ((drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT)) {
3151                 /* TDLS Discovery Response */
3152                 if (nl80211_register_action_frame(bss, (u8 *) "\x04\x0e", 2) <
3153                     0)
3154                         return -1;
3155         }
3156 #endif /* CONFIG_TDLS */
3157
3158         /* FT Action frames */
3159         if (nl80211_register_action_frame(bss, (u8 *) "\x06", 1) < 0)
3160                 return -1;
3161         else
3162                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT |
3163                         WPA_DRIVER_CAPA_KEY_MGMT_FT_PSK;
3164
3165         /* WNM - BSS Transition Management Request */
3166         if (nl80211_register_action_frame(bss, (u8 *) "\x0a\x07", 2) < 0)
3167                 return -1;
3168
3169         return 0;
3170 }
3171
3172
3173 static int nl80211_register_spurious_class3(struct i802_bss *bss)
3174 {
3175         struct wpa_driver_nl80211_data *drv = bss->drv;
3176         struct nl_msg *msg;
3177         int ret = -1;
3178
3179         msg = nlmsg_alloc();
3180         if (!msg)
3181                 return -1;
3182
3183         nl80211_cmd(drv, msg, 0, NL80211_CMD_UNEXPECTED_FRAME);
3184
3185         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
3186
3187         ret = send_and_recv(drv->global, bss->nl_mgmt, msg, NULL, NULL);
3188         msg = NULL;
3189         if (ret) {
3190                 wpa_printf(MSG_DEBUG, "nl80211: Register spurious class3 "
3191                            "failed: ret=%d (%s)",
3192                            ret, strerror(-ret));
3193                 goto nla_put_failure;
3194         }
3195         ret = 0;
3196 nla_put_failure:
3197         nlmsg_free(msg);
3198         return ret;
3199 }
3200
3201
3202 static int nl80211_mgmt_subscribe_ap(struct i802_bss *bss)
3203 {
3204         static const int stypes[] = {
3205                 WLAN_FC_STYPE_AUTH,
3206                 WLAN_FC_STYPE_ASSOC_REQ,
3207                 WLAN_FC_STYPE_REASSOC_REQ,
3208                 WLAN_FC_STYPE_DISASSOC,
3209                 WLAN_FC_STYPE_DEAUTH,
3210                 WLAN_FC_STYPE_ACTION,
3211                 WLAN_FC_STYPE_PROBE_REQ,
3212 /* Beacon doesn't work as mac80211 doesn't currently allow
3213  * it, but it wouldn't really be the right thing anyway as
3214  * it isn't per interface ... maybe just dump the scan
3215  * results periodically for OLBC?
3216  */
3217 //              WLAN_FC_STYPE_BEACON,
3218         };
3219         unsigned int i;
3220
3221         if (nl80211_alloc_mgmt_handle(bss))
3222                 return -1;
3223         wpa_printf(MSG_DEBUG, "nl80211: Subscribe to mgmt frames with AP "
3224                    "handle %p", bss->nl_mgmt);
3225
3226         for (i = 0; i < sizeof(stypes) / sizeof(stypes[0]); i++) {
3227                 if (nl80211_register_frame(bss, bss->nl_mgmt,
3228                                            (WLAN_FC_TYPE_MGMT << 2) |
3229                                            (stypes[i] << 4),
3230                                            NULL, 0) < 0) {
3231                         goto out_err;
3232                 }
3233         }
3234
3235         if (nl80211_register_spurious_class3(bss))
3236                 goto out_err;
3237
3238         if (nl80211_get_wiphy_data_ap(bss) == NULL)
3239                 goto out_err;
3240
3241         return 0;
3242
3243 out_err:
3244         eloop_unregister_read_sock(nl_socket_get_fd(bss->nl_mgmt));
3245         nl_destroy_handles(&bss->nl_mgmt);
3246         return -1;
3247 }
3248
3249
3250 static int nl80211_mgmt_subscribe_ap_dev_sme(struct i802_bss *bss)
3251 {
3252         if (nl80211_alloc_mgmt_handle(bss))
3253                 return -1;
3254         wpa_printf(MSG_DEBUG, "nl80211: Subscribe to mgmt frames with AP "
3255                    "handle %p (device SME)", bss->nl_mgmt);
3256
3257         if (nl80211_register_frame(bss, bss->nl_mgmt,
3258                                    (WLAN_FC_TYPE_MGMT << 2) |
3259                                    (WLAN_FC_STYPE_ACTION << 4),
3260                                    NULL, 0) < 0)
3261                 goto out_err;
3262
3263         return 0;
3264
3265 out_err:
3266         eloop_unregister_read_sock(nl_socket_get_fd(bss->nl_mgmt));
3267         nl_destroy_handles(&bss->nl_mgmt);
3268         return -1;
3269 }
3270
3271
3272 static void nl80211_mgmt_unsubscribe(struct i802_bss *bss, const char *reason)
3273 {
3274         if (bss->nl_mgmt == NULL)
3275                 return;
3276         wpa_printf(MSG_DEBUG, "nl80211: Unsubscribe mgmt frames handle %p "
3277                    "(%s)", bss->nl_mgmt, reason);
3278         eloop_unregister_read_sock(nl_socket_get_fd(bss->nl_mgmt));
3279         nl_destroy_handles(&bss->nl_mgmt);
3280
3281         nl80211_put_wiphy_data_ap(bss);
3282 }
3283
3284
3285 static void wpa_driver_nl80211_send_rfkill(void *eloop_ctx, void *timeout_ctx)
3286 {
3287         wpa_supplicant_event(timeout_ctx, EVENT_INTERFACE_DISABLED, NULL);
3288 }
3289
3290
3291 static int
3292 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
3293 {
3294         struct i802_bss *bss = &drv->first_bss;
3295         int send_rfkill_event = 0;
3296
3297         drv->ifindex = if_nametoindex(bss->ifname);
3298         drv->first_bss.ifindex = drv->ifindex;
3299
3300 #ifndef HOSTAPD
3301         /*
3302          * Make sure the interface starts up in station mode unless this is a
3303          * dynamically added interface (e.g., P2P) that was already configured
3304          * with proper iftype.
3305          */
3306         if (drv->ifindex != drv->global->if_add_ifindex &&
3307             wpa_driver_nl80211_set_mode(bss, NL80211_IFTYPE_STATION) < 0) {
3308                 wpa_printf(MSG_ERROR, "nl80211: Could not configure driver to "
3309                            "use managed mode");
3310                 return -1;
3311         }
3312
3313         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 1)) {
3314                 if (rfkill_is_blocked(drv->rfkill)) {
3315                         wpa_printf(MSG_DEBUG, "nl80211: Could not yet enable "
3316                                    "interface '%s' due to rfkill",
3317                                    bss->ifname);
3318                         drv->if_disabled = 1;
3319                         send_rfkill_event = 1;
3320                 } else {
3321                         wpa_printf(MSG_ERROR, "nl80211: Could not set "
3322                                    "interface '%s' UP", bss->ifname);
3323                         return -1;
3324                 }
3325         }
3326
3327         netlink_send_oper_ifla(drv->global->netlink, drv->ifindex,
3328                                1, IF_OPER_DORMANT);
3329 #endif /* HOSTAPD */
3330
3331         if (wpa_driver_nl80211_capa(drv))
3332                 return -1;
3333
3334         if (linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
3335                                bss->addr))
3336                 return -1;
3337
3338         if (send_rfkill_event) {
3339                 eloop_register_timeout(0, 0, wpa_driver_nl80211_send_rfkill,
3340                                        drv, drv->ctx);
3341         }
3342
3343         return 0;
3344 }
3345
3346
3347 static int wpa_driver_nl80211_del_beacon(struct wpa_driver_nl80211_data *drv)
3348 {
3349         struct nl_msg *msg;
3350
3351         msg = nlmsg_alloc();
3352         if (!msg)
3353                 return -ENOMEM;
3354
3355         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_BEACON);
3356         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3357
3358         return send_and_recv_msgs(drv, msg, NULL, NULL);
3359  nla_put_failure:
3360         nlmsg_free(msg);
3361         return -ENOBUFS;
3362 }
3363
3364
3365 /**
3366  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
3367  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
3368  *
3369  * Shut down driver interface and processing of driver events. Free
3370  * private data buffer if one was allocated in wpa_driver_nl80211_init().
3371  */
3372 static void wpa_driver_nl80211_deinit(void *priv)
3373 {
3374         struct i802_bss *bss = priv;
3375         struct wpa_driver_nl80211_data *drv = bss->drv;
3376
3377         bss->in_deinit = 1;
3378         if (drv->data_tx_status)
3379                 eloop_unregister_read_sock(drv->eapol_tx_sock);
3380         if (drv->eapol_tx_sock >= 0)
3381                 close(drv->eapol_tx_sock);
3382
3383         if (bss->nl_preq)
3384                 wpa_driver_nl80211_probe_req_report(bss, 0);
3385         if (bss->added_if_into_bridge) {
3386                 if (linux_br_del_if(drv->global->ioctl_sock, bss->brname,
3387                                     bss->ifname) < 0)
3388                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
3389                                    "interface %s from bridge %s: %s",
3390                                    bss->ifname, bss->brname, strerror(errno));
3391         }
3392         if (bss->added_bridge) {
3393                 if (linux_br_del(drv->global->ioctl_sock, bss->brname) < 0)
3394                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
3395                                    "bridge %s: %s",
3396                                    bss->brname, strerror(errno));
3397         }
3398
3399         nl80211_remove_monitor_interface(drv);
3400
3401         if (is_ap_interface(drv->nlmode))
3402                 wpa_driver_nl80211_del_beacon(drv);
3403
3404 #ifdef HOSTAPD
3405         if (drv->last_freq_ht) {
3406                 /* Clear HT flags from the driver */
3407                 struct hostapd_freq_params freq;
3408                 os_memset(&freq, 0, sizeof(freq));
3409                 freq.freq = drv->last_freq;
3410                 i802_set_freq(priv, &freq);
3411         }
3412
3413         if (drv->eapol_sock >= 0) {
3414                 eloop_unregister_read_sock(drv->eapol_sock);
3415                 close(drv->eapol_sock);
3416         }
3417
3418         if (drv->if_indices != drv->default_if_indices)
3419                 os_free(drv->if_indices);
3420 #endif /* HOSTAPD */
3421
3422         if (drv->disabled_11b_rates)
3423                 nl80211_disable_11b_rates(drv, drv->ifindex, 0);
3424
3425         netlink_send_oper_ifla(drv->global->netlink, drv->ifindex, 0,
3426                                IF_OPER_UP);
3427         rfkill_deinit(drv->rfkill);
3428
3429         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
3430
3431         (void) linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 0);
3432         wpa_driver_nl80211_set_mode(bss, NL80211_IFTYPE_STATION);
3433         nl80211_mgmt_unsubscribe(bss, "deinit");
3434
3435         nl_cb_put(drv->nl_cb);
3436
3437         nl80211_destroy_bss(&drv->first_bss);
3438
3439         os_free(drv->filter_ssids);
3440
3441         os_free(drv->auth_ie);
3442
3443         if (drv->in_interface_list)
3444                 dl_list_del(&drv->list);
3445
3446         os_free(drv);
3447 }
3448
3449
3450 /**
3451  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
3452  * @eloop_ctx: Driver private data
3453  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
3454  *
3455  * This function can be used as registered timeout when starting a scan to
3456  * generate a scan completed event if the driver does not report this.
3457  */
3458 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
3459 {
3460         struct wpa_driver_nl80211_data *drv = eloop_ctx;
3461         if (drv->ap_scan_as_station != NL80211_IFTYPE_UNSPECIFIED) {
3462                 wpa_driver_nl80211_set_mode(&drv->first_bss,
3463                                             drv->ap_scan_as_station);
3464                 drv->ap_scan_as_station = NL80211_IFTYPE_UNSPECIFIED;
3465         }
3466         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
3467         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
3468 }
3469
3470
3471 /**
3472  * wpa_driver_nl80211_scan - Request the driver to initiate scan
3473  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
3474  * @params: Scan parameters
3475  * Returns: 0 on success, -1 on failure
3476  */
3477 static int wpa_driver_nl80211_scan(void *priv,
3478                                    struct wpa_driver_scan_params *params)
3479 {
3480         struct i802_bss *bss = priv;
3481         struct wpa_driver_nl80211_data *drv = bss->drv;
3482         int ret = -1, timeout;
3483         struct nl_msg *msg, *ssids = NULL, *freqs = NULL, *rates = NULL;
3484         size_t i;
3485
3486         drv->scan_for_auth = 0;
3487
3488         msg = nlmsg_alloc();
3489         if (!msg)
3490                 return -1;
3491
3492         os_free(drv->filter_ssids);
3493         drv->filter_ssids = params->filter_ssids;
3494         params->filter_ssids = NULL;
3495         drv->num_filter_ssids = params->num_filter_ssids;
3496
3497         nl80211_cmd(drv, msg, 0, NL80211_CMD_TRIGGER_SCAN);
3498
3499         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3500
3501         if (params->num_ssids) {
3502                 ssids = nlmsg_alloc();
3503                 if (ssids == NULL)
3504                         goto nla_put_failure;
3505                 for (i = 0; i < params->num_ssids; i++) {
3506                         wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Scan SSID",
3507                                           params->ssids[i].ssid,
3508                                           params->ssids[i].ssid_len);
3509                         NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
3510                                 params->ssids[i].ssid);
3511                 }
3512                 if (nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids) < 0)
3513                         goto nla_put_failure;
3514         }
3515
3516         if (params->extra_ies) {
3517                 wpa_hexdump(MSG_MSGDUMP, "nl80211: Scan extra IEs",
3518                             params->extra_ies, params->extra_ies_len);
3519                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
3520                         params->extra_ies);
3521         }
3522
3523         if (params->freqs) {
3524                 freqs = nlmsg_alloc();
3525                 if (freqs == NULL)
3526                         goto nla_put_failure;
3527                 for (i = 0; params->freqs[i]; i++) {
3528                         wpa_printf(MSG_MSGDUMP, "nl80211: Scan frequency %u "
3529                                    "MHz", params->freqs[i]);
3530                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
3531                 }
3532                 if (nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs) <
3533                     0)
3534                         goto nla_put_failure;
3535         }
3536
3537         if (params->p2p_probe) {
3538                 wpa_printf(MSG_DEBUG, "nl80211: P2P probe - mask SuppRates");
3539
3540                 rates = nlmsg_alloc();
3541                 if (rates == NULL)
3542                         goto nla_put_failure;
3543
3544                 /*
3545                  * Remove 2.4 GHz rates 1, 2, 5.5, 11 Mbps from supported rates
3546                  * by masking out everything else apart from the OFDM rates 6,
3547                  * 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS rates. All 5 GHz
3548                  * rates are left enabled.
3549                  */
3550                 NLA_PUT(rates, NL80211_BAND_2GHZ, 8,
3551                         "\x0c\x12\x18\x24\x30\x48\x60\x6c");
3552                 if (nla_put_nested(msg, NL80211_ATTR_SCAN_SUPP_RATES, rates) <
3553                     0)
3554                         goto nla_put_failure;
3555
3556                 NLA_PUT_FLAG(msg, NL80211_ATTR_TX_NO_CCK_RATE);
3557         }
3558
3559         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3560         msg = NULL;
3561         if (ret) {
3562                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
3563                            "(%s)", ret, strerror(-ret));
3564 #ifdef HOSTAPD
3565                 if (is_ap_interface(drv->nlmode)) {
3566                         /*
3567                          * mac80211 does not allow scan requests in AP mode, so
3568                          * try to do this in station mode.
3569                          */
3570                         if (wpa_driver_nl80211_set_mode(
3571                                     bss, NL80211_IFTYPE_STATION))
3572                                 goto nla_put_failure;
3573
3574                         if (wpa_driver_nl80211_scan(drv, params)) {
3575                                 wpa_driver_nl80211_set_mode(bss, drv->nlmode);
3576                                 goto nla_put_failure;
3577                         }
3578
3579                         /* Restore AP mode when processing scan results */
3580                         drv->ap_scan_as_station = drv->nlmode;
3581                         ret = 0;
3582                 } else
3583                         goto nla_put_failure;
3584 #else /* HOSTAPD */
3585                 goto nla_put_failure;
3586 #endif /* HOSTAPD */
3587         }
3588
3589         /* Not all drivers generate "scan completed" wireless event, so try to
3590          * read results after a timeout. */
3591         timeout = 10;
3592         if (drv->scan_complete_events) {
3593                 /*
3594                  * The driver seems to deliver events to notify when scan is
3595                  * complete, so use longer timeout to avoid race conditions
3596                  * with scanning and following association request.
3597                  */
3598                 timeout = 30;
3599         }
3600         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
3601                    "seconds", ret, timeout);
3602         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
3603         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
3604                                drv, drv->ctx);
3605
3606 nla_put_failure:
3607         nlmsg_free(ssids);
3608         nlmsg_free(msg);
3609         nlmsg_free(freqs);
3610         nlmsg_free(rates);
3611         return ret;
3612 }
3613
3614
3615 /**
3616  * wpa_driver_nl80211_sched_scan - Initiate a scheduled scan
3617  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
3618  * @params: Scan parameters
3619  * @interval: Interval between scan cycles in milliseconds
3620  * Returns: 0 on success, -1 on failure or if not supported
3621  */
3622 static int wpa_driver_nl80211_sched_scan(void *priv,
3623                                          struct wpa_driver_scan_params *params,
3624                                          u32 interval)
3625 {
3626         struct i802_bss *bss = priv;
3627         struct wpa_driver_nl80211_data *drv = bss->drv;
3628         int ret = -1;
3629         struct nl_msg *msg, *ssids = NULL, *freqs = NULL;
3630         struct nl_msg *match_set_ssid = NULL, *match_sets = NULL;
3631         struct nl_msg *match_set_rssi = NULL;
3632         size_t i;
3633
3634 #ifdef ANDROID
3635         if (!drv->capa.sched_scan_supported)
3636                 return android_pno_start(bss, params);
3637 #endif /* ANDROID */
3638
3639         msg = nlmsg_alloc();
3640         if (!msg)
3641                 goto nla_put_failure;
3642
3643         os_free(drv->filter_ssids);
3644         drv->filter_ssids = params->filter_ssids;
3645         params->filter_ssids = NULL;
3646         drv->num_filter_ssids = params->num_filter_ssids;
3647
3648         nl80211_cmd(drv, msg, 0, NL80211_CMD_START_SCHED_SCAN);
3649
3650         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3651
3652         NLA_PUT_U32(msg, NL80211_ATTR_SCHED_SCAN_INTERVAL, interval);
3653
3654         if ((drv->num_filter_ssids &&
3655             (int) drv->num_filter_ssids <= drv->capa.max_match_sets) ||
3656             params->filter_rssi) {
3657                 match_sets = nlmsg_alloc();
3658                 if (match_sets == NULL)
3659                         goto nla_put_failure;
3660
3661                 for (i = 0; i < drv->num_filter_ssids; i++) {
3662                         wpa_hexdump_ascii(MSG_MSGDUMP,
3663                                           "nl80211: Sched scan filter SSID",
3664                                           drv->filter_ssids[i].ssid,
3665                                           drv->filter_ssids[i].ssid_len);
3666
3667                         match_set_ssid = nlmsg_alloc();
3668                         if (match_set_ssid == NULL)
3669                                 goto nla_put_failure;
3670                         NLA_PUT(match_set_ssid,
3671                                 NL80211_ATTR_SCHED_SCAN_MATCH_SSID,
3672                                 drv->filter_ssids[i].ssid_len,
3673                                 drv->filter_ssids[i].ssid);
3674
3675                         if (nla_put_nested(match_sets, i + 1, match_set_ssid) <
3676                             0)
3677                                 goto nla_put_failure;
3678                 }
3679
3680                 if (params->filter_rssi) {
3681                         match_set_rssi = nlmsg_alloc();
3682                         if (match_set_rssi == NULL)
3683                                 goto nla_put_failure;
3684                         NLA_PUT_U32(match_set_rssi,
3685                                     NL80211_SCHED_SCAN_MATCH_ATTR_RSSI,
3686                                     params->filter_rssi);
3687                         wpa_printf(MSG_MSGDUMP,
3688                                    "nl80211: Sched scan RSSI filter %d dBm",
3689                                    params->filter_rssi);
3690                         if (nla_put_nested(match_sets, 0, match_set_rssi) < 0)
3691                                 goto nla_put_failure;
3692                 }
3693
3694                 if (nla_put_nested(msg, NL80211_ATTR_SCHED_SCAN_MATCH,
3695                                    match_sets) < 0)
3696                         goto nla_put_failure;
3697         }
3698
3699         if (params->num_ssids) {
3700                 ssids = nlmsg_alloc();
3701                 if (ssids == NULL)
3702                         goto nla_put_failure;
3703                 for (i = 0; i < params->num_ssids; i++) {
3704                         wpa_hexdump_ascii(MSG_MSGDUMP,
3705                                           "nl80211: Sched scan SSID",
3706                                           params->ssids[i].ssid,
3707                                           params->ssids[i].ssid_len);
3708                         NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
3709                                 params->ssids[i].ssid);
3710                 }
3711                 if (nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids) < 0)
3712                         goto nla_put_failure;
3713         }
3714
3715         if (params->extra_ies) {
3716                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Sched scan extra IEs",
3717                                   params->extra_ies, params->extra_ies_len);
3718                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
3719                         params->extra_ies);
3720         }
3721
3722         if (params->freqs) {
3723                 freqs = nlmsg_alloc();
3724                 if (freqs == NULL)
3725                         goto nla_put_failure;
3726                 for (i = 0; params->freqs[i]; i++) {
3727                         wpa_printf(MSG_MSGDUMP, "nl80211: Scan frequency %u "
3728                                    "MHz", params->freqs[i]);
3729                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
3730                 }
3731                 if (nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs) <
3732                     0)
3733                         goto nla_put_failure;
3734         }
3735
3736         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3737
3738         /* TODO: if we get an error here, we should fall back to normal scan */
3739
3740         msg = NULL;
3741         if (ret) {
3742                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan start failed: "
3743                            "ret=%d (%s)", ret, strerror(-ret));
3744                 goto nla_put_failure;
3745         }
3746
3747         wpa_printf(MSG_DEBUG, "nl80211: Sched scan requested (ret=%d) - "
3748                    "scan interval %d msec", ret, interval);
3749
3750 nla_put_failure:
3751         nlmsg_free(match_set_ssid);
3752         nlmsg_free(match_sets);
3753         nlmsg_free(match_set_rssi);
3754         nlmsg_free(ssids);
3755         nlmsg_free(msg);
3756         nlmsg_free(freqs);
3757         return ret;
3758 }
3759
3760
3761 /**
3762  * wpa_driver_nl80211_stop_sched_scan - Stop a scheduled scan
3763  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
3764  * Returns: 0 on success, -1 on failure or if not supported
3765  */
3766 static int wpa_driver_nl80211_stop_sched_scan(void *priv)
3767 {
3768         struct i802_bss *bss = priv;
3769         struct wpa_driver_nl80211_data *drv = bss->drv;
3770         int ret = 0;
3771         struct nl_msg *msg;
3772
3773 #ifdef ANDROID
3774         if (!drv->capa.sched_scan_supported)
3775                 return android_pno_stop(bss);
3776 #endif /* ANDROID */
3777
3778         msg = nlmsg_alloc();
3779         if (!msg)
3780                 return -1;
3781
3782         nl80211_cmd(drv, msg, 0, NL80211_CMD_STOP_SCHED_SCAN);
3783
3784         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3785
3786         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3787         msg = NULL;
3788         if (ret) {
3789                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan stop failed: "
3790                            "ret=%d (%s)", ret, strerror(-ret));
3791                 goto nla_put_failure;
3792         }
3793
3794         wpa_printf(MSG_DEBUG, "nl80211: Sched scan stop sent (ret=%d)", ret);
3795
3796 nla_put_failure:
3797         nlmsg_free(msg);
3798         return ret;
3799 }
3800
3801
3802 static const u8 * nl80211_get_ie(const u8 *ies, size_t ies_len, u8 ie)
3803 {
3804         const u8 *end, *pos;
3805
3806         if (ies == NULL)
3807                 return NULL;
3808
3809         pos = ies;
3810         end = ies + ies_len;
3811
3812         while (pos + 1 < end) {
3813                 if (pos + 2 + pos[1] > end)
3814                         break;
3815                 if (pos[0] == ie)
3816                         return pos;
3817                 pos += 2 + pos[1];
3818         }
3819
3820         return NULL;
3821 }
3822
3823
3824 static int nl80211_scan_filtered(struct wpa_driver_nl80211_data *drv,
3825                                  const u8 *ie, size_t ie_len)
3826 {
3827         const u8 *ssid;
3828         size_t i;
3829
3830         if (drv->filter_ssids == NULL)
3831                 return 0;
3832
3833         ssid = nl80211_get_ie(ie, ie_len, WLAN_EID_SSID);
3834         if (ssid == NULL)
3835                 return 1;
3836
3837         for (i = 0; i < drv->num_filter_ssids; i++) {
3838                 if (ssid[1] == drv->filter_ssids[i].ssid_len &&
3839                     os_memcmp(ssid + 2, drv->filter_ssids[i].ssid, ssid[1]) ==
3840                     0)
3841                         return 0;
3842         }
3843
3844         return 1;
3845 }
3846
3847
3848 static int bss_info_handler(struct nl_msg *msg, void *arg)
3849 {
3850         struct nlattr *tb[NL80211_ATTR_MAX + 1];
3851         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
3852         struct nlattr *bss[NL80211_BSS_MAX + 1];
3853         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
3854                 [NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
3855                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
3856                 [NL80211_BSS_TSF] = { .type = NLA_U64 },
3857                 [NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
3858                 [NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
3859                 [NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
3860                 [NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
3861                 [NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
3862                 [NL80211_BSS_STATUS] = { .type = NLA_U32 },
3863                 [NL80211_BSS_SEEN_MS_AGO] = { .type = NLA_U32 },
3864                 [NL80211_BSS_BEACON_IES] = { .type = NLA_UNSPEC },
3865         };
3866         struct nl80211_bss_info_arg *_arg = arg;
3867         struct wpa_scan_results *res = _arg->res;
3868         struct wpa_scan_res **tmp;
3869         struct wpa_scan_res *r;
3870         const u8 *ie, *beacon_ie;
3871         size_t ie_len, beacon_ie_len;
3872         u8 *pos;
3873         size_t i;
3874
3875         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
3876                   genlmsg_attrlen(gnlh, 0), NULL);
3877         if (!tb[NL80211_ATTR_BSS])
3878                 return NL_SKIP;
3879         if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
3880                              bss_policy))
3881                 return NL_SKIP;
3882         if (bss[NL80211_BSS_STATUS]) {
3883                 enum nl80211_bss_status status;
3884                 status = nla_get_u32(bss[NL80211_BSS_STATUS]);
3885                 if (status == NL80211_BSS_STATUS_ASSOCIATED &&
3886                     bss[NL80211_BSS_FREQUENCY]) {
3887                         _arg->assoc_freq =
3888                                 nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
3889                         wpa_printf(MSG_DEBUG, "nl80211: Associated on %u MHz",
3890                                    _arg->assoc_freq);
3891                 }
3892                 if (status == NL80211_BSS_STATUS_ASSOCIATED &&
3893                     bss[NL80211_BSS_BSSID]) {
3894                         os_memcpy(_arg->assoc_bssid,
3895                                   nla_data(bss[NL80211_BSS_BSSID]), ETH_ALEN);
3896                         wpa_printf(MSG_DEBUG, "nl80211: Associated with "
3897                                    MACSTR, MAC2STR(_arg->assoc_bssid));
3898                 }
3899         }
3900         if (!res)
3901                 return NL_SKIP;
3902         if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
3903                 ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
3904                 ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
3905         } else {
3906                 ie = NULL;
3907                 ie_len = 0;
3908         }
3909         if (bss[NL80211_BSS_BEACON_IES]) {
3910                 beacon_ie = nla_data(bss[NL80211_BSS_BEACON_IES]);
3911                 beacon_ie_len = nla_len(bss[NL80211_BSS_BEACON_IES]);
3912         } else {
3913                 beacon_ie = NULL;
3914                 beacon_ie_len = 0;
3915         }
3916
3917         if (nl80211_scan_filtered(_arg->drv, ie ? ie : beacon_ie,
3918                                   ie ? ie_len : beacon_ie_len))
3919                 return NL_SKIP;
3920
3921         r = os_zalloc(sizeof(*r) + ie_len + beacon_ie_len);
3922         if (r == NULL)
3923                 return NL_SKIP;
3924         if (bss[NL80211_BSS_BSSID])
3925                 os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
3926                           ETH_ALEN);
3927         if (bss[NL80211_BSS_FREQUENCY])
3928                 r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
3929         if (bss[NL80211_BSS_BEACON_INTERVAL])
3930                 r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
3931         if (bss[NL80211_BSS_CAPABILITY])
3932                 r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
3933         r->flags |= WPA_SCAN_NOISE_INVALID;
3934         if (bss[NL80211_BSS_SIGNAL_MBM]) {
3935                 r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
3936                 r->level /= 100; /* mBm to dBm */
3937                 r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
3938         } else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
3939                 r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
3940                 r->flags |= WPA_SCAN_QUAL_INVALID;
3941         } else
3942                 r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
3943         if (bss[NL80211_BSS_TSF])
3944                 r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
3945         if (bss[NL80211_BSS_SEEN_MS_AGO])
3946                 r->age = nla_get_u32(bss[NL80211_BSS_SEEN_MS_AGO]);
3947         r->ie_len = ie_len;
3948         pos = (u8 *) (r + 1);
3949         if (ie) {
3950                 os_memcpy(pos, ie, ie_len);
3951                 pos += ie_len;
3952         }
3953         r->beacon_ie_len = beacon_ie_len;
3954         if (beacon_ie)
3955                 os_memcpy(pos, beacon_ie, beacon_ie_len);
3956
3957         if (bss[NL80211_BSS_STATUS]) {
3958                 enum nl80211_bss_status status;
3959                 status = nla_get_u32(bss[NL80211_BSS_STATUS]);
3960                 switch (status) {
3961                 case NL80211_BSS_STATUS_AUTHENTICATED:
3962                         r->flags |= WPA_SCAN_AUTHENTICATED;
3963                         break;
3964                 case NL80211_BSS_STATUS_ASSOCIATED:
3965                         r->flags |= WPA_SCAN_ASSOCIATED;
3966                         break;
3967                 default:
3968                         break;
3969                 }
3970         }
3971
3972         /*
3973          * cfg80211 maintains separate BSS table entries for APs if the same
3974          * BSSID,SSID pair is seen on multiple channels. wpa_supplicant does
3975          * not use frequency as a separate key in the BSS table, so filter out
3976          * duplicated entries. Prefer associated BSS entry in such a case in
3977          * order to get the correct frequency into the BSS table.
3978          */
3979         for (i = 0; i < res->num; i++) {
3980                 const u8 *s1, *s2;
3981                 if (os_memcmp(res->res[i]->bssid, r->bssid, ETH_ALEN) != 0)
3982                         continue;
3983
3984                 s1 = nl80211_get_ie((u8 *) (res->res[i] + 1),
3985                                     res->res[i]->ie_len, WLAN_EID_SSID);
3986                 s2 = nl80211_get_ie((u8 *) (r + 1), r->ie_len, WLAN_EID_SSID);
3987                 if (s1 == NULL || s2 == NULL || s1[1] != s2[1] ||
3988                     os_memcmp(s1, s2, 2 + s1[1]) != 0)
3989                         continue;
3990
3991                 /* Same BSSID,SSID was already included in scan results */
3992                 wpa_printf(MSG_DEBUG, "nl80211: Remove duplicated scan result "
3993                            "for " MACSTR, MAC2STR(r->bssid));
3994
3995                 if ((r->flags & WPA_SCAN_ASSOCIATED) &&
3996                     !(res->res[i]->flags & WPA_SCAN_ASSOCIATED)) {
3997                         os_free(res->res[i]);
3998                         res->res[i] = r;
3999                 } else
4000                         os_free(r);
4001                 return NL_SKIP;
4002         }
4003
4004         tmp = os_realloc(res->res,
4005                          (res->num + 1) * sizeof(struct wpa_scan_res *));
4006         if (tmp == NULL) {
4007                 os_free(r);
4008                 return NL_SKIP;
4009         }
4010         tmp[res->num++] = r;
4011         res->res = tmp;
4012
4013         return NL_SKIP;
4014 }
4015
4016
4017 static void clear_state_mismatch(struct wpa_driver_nl80211_data *drv,
4018                                  const u8 *addr)
4019 {
4020         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
4021                 wpa_printf(MSG_DEBUG, "nl80211: Clear possible state "
4022                            "mismatch (" MACSTR ")", MAC2STR(addr));
4023                 wpa_driver_nl80211_mlme(drv, addr,
4024                                         NL80211_CMD_DEAUTHENTICATE,
4025                                         WLAN_REASON_PREV_AUTH_NOT_VALID, 1);
4026         }
4027 }
4028
4029
4030 static void wpa_driver_nl80211_check_bss_status(
4031         struct wpa_driver_nl80211_data *drv, struct wpa_scan_results *res)
4032 {
4033         size_t i;
4034
4035         for (i = 0; i < res->num; i++) {
4036                 struct wpa_scan_res *r = res->res[i];
4037                 if (r->flags & WPA_SCAN_AUTHENTICATED) {
4038                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
4039                                    "indicates BSS status with " MACSTR
4040                                    " as authenticated",
4041                                    MAC2STR(r->bssid));
4042                         if (is_sta_interface(drv->nlmode) &&
4043                             os_memcmp(r->bssid, drv->bssid, ETH_ALEN) != 0 &&
4044                             os_memcmp(r->bssid, drv->auth_bssid, ETH_ALEN) !=
4045                             0) {
4046                                 wpa_printf(MSG_DEBUG, "nl80211: Unknown BSSID"
4047                                            " in local state (auth=" MACSTR
4048                                            " assoc=" MACSTR ")",
4049                                            MAC2STR(drv->auth_bssid),
4050                                            MAC2STR(drv->bssid));
4051                                 clear_state_mismatch(drv, r->bssid);
4052                         }
4053                 }
4054
4055                 if (r->flags & WPA_SCAN_ASSOCIATED) {
4056                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
4057                                    "indicate BSS status with " MACSTR
4058                                    " as associated",
4059                                    MAC2STR(r->bssid));
4060                         if (is_sta_interface(drv->nlmode) &&
4061                             !drv->associated) {
4062                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
4063                                            "(not associated) does not match "
4064                                            "with BSS state");
4065                                 clear_state_mismatch(drv, r->bssid);
4066                         } else if (is_sta_interface(drv->nlmode) &&
4067                                    os_memcmp(drv->bssid, r->bssid, ETH_ALEN) !=
4068                                    0) {
4069                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
4070                                            "(associated with " MACSTR ") does "
4071                                            "not match with BSS state",
4072                                            MAC2STR(drv->bssid));
4073                                 clear_state_mismatch(drv, r->bssid);
4074                                 clear_state_mismatch(drv, drv->bssid);
4075                         }
4076                 }
4077         }
4078 }
4079
4080
4081 static struct wpa_scan_results *
4082 nl80211_get_scan_results(struct wpa_driver_nl80211_data *drv)
4083 {
4084         struct nl_msg *msg;
4085         struct wpa_scan_results *res;
4086         int ret;
4087         struct nl80211_bss_info_arg arg;
4088
4089         res = os_zalloc(sizeof(*res));
4090         if (res == NULL)
4091                 return NULL;
4092         msg = nlmsg_alloc();
4093         if (!msg)
4094                 goto nla_put_failure;
4095
4096         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
4097         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4098
4099         arg.drv = drv;
4100         arg.res = res;
4101         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
4102         msg = NULL;
4103         if (ret == 0) {
4104                 wpa_printf(MSG_DEBUG, "nl80211: Received scan results (%lu "
4105                            "BSSes)", (unsigned long) res->num);
4106                 nl80211_get_noise_for_scan_results(drv, res);
4107                 return res;
4108         }
4109         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
4110                    "(%s)", ret, strerror(-ret));
4111 nla_put_failure:
4112         nlmsg_free(msg);
4113         wpa_scan_results_free(res);
4114         return NULL;
4115 }
4116
4117
4118 /**
4119  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
4120  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
4121  * Returns: Scan results on success, -1 on failure
4122  */
4123 static struct wpa_scan_results *
4124 wpa_driver_nl80211_get_scan_results(void *priv)
4125 {
4126         struct i802_bss *bss = priv;
4127         struct wpa_driver_nl80211_data *drv = bss->drv;
4128         struct wpa_scan_results *res;
4129
4130         res = nl80211_get_scan_results(drv);
4131         if (res)
4132                 wpa_driver_nl80211_check_bss_status(drv, res);
4133         return res;
4134 }
4135
4136
4137 static void nl80211_dump_scan(struct wpa_driver_nl80211_data *drv)
4138 {
4139         struct wpa_scan_results *res;
4140         size_t i;
4141
4142         res = nl80211_get_scan_results(drv);
4143         if (res == NULL) {
4144                 wpa_printf(MSG_DEBUG, "nl80211: Failed to get scan results");
4145                 return;
4146         }
4147
4148         wpa_printf(MSG_DEBUG, "nl80211: Scan result dump");
4149         for (i = 0; i < res->num; i++) {
4150                 struct wpa_scan_res *r = res->res[i];
4151                 wpa_printf(MSG_DEBUG, "nl80211: %d/%d " MACSTR "%s%s",
4152                            (int) i, (int) res->num, MAC2STR(r->bssid),
4153                            r->flags & WPA_SCAN_AUTHENTICATED ? " [auth]" : "",
4154                            r->flags & WPA_SCAN_ASSOCIATED ? " [assoc]" : "");
4155         }
4156
4157         wpa_scan_results_free(res);
4158 }
4159
4160
4161 static int wpa_driver_nl80211_set_key(const char *ifname, void *priv,
4162                                       enum wpa_alg alg, const u8 *addr,
4163                                       int key_idx, int set_tx,
4164                                       const u8 *seq, size_t seq_len,
4165                                       const u8 *key, size_t key_len)
4166 {
4167         struct i802_bss *bss = priv;
4168         struct wpa_driver_nl80211_data *drv = bss->drv;
4169         int ifindex = if_nametoindex(ifname);
4170         struct nl_msg *msg;
4171         int ret;
4172
4173         wpa_printf(MSG_DEBUG, "%s: ifindex=%d alg=%d addr=%p key_idx=%d "
4174                    "set_tx=%d seq_len=%lu key_len=%lu",
4175                    __func__, ifindex, alg, addr, key_idx, set_tx,
4176                    (unsigned long) seq_len, (unsigned long) key_len);
4177 #ifdef CONFIG_TDLS
4178         if (key_idx == -1)
4179                 key_idx = 0;
4180 #endif /* CONFIG_TDLS */
4181
4182         msg = nlmsg_alloc();
4183         if (!msg)
4184                 return -ENOMEM;
4185
4186         if (alg == WPA_ALG_NONE) {
4187                 nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_KEY);
4188         } else {
4189                 nl80211_cmd(drv, msg, 0, NL80211_CMD_NEW_KEY);
4190                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
4191                 switch (alg) {
4192                 case WPA_ALG_WEP:
4193                         if (key_len == 5)
4194                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4195                                             WLAN_CIPHER_SUITE_WEP40);
4196                         else
4197                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4198                                             WLAN_CIPHER_SUITE_WEP104);
4199                         break;
4200                 case WPA_ALG_TKIP:
4201                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4202                                     WLAN_CIPHER_SUITE_TKIP);
4203                         break;
4204                 case WPA_ALG_CCMP:
4205                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4206                                     WLAN_CIPHER_SUITE_CCMP);
4207                         break;
4208                 case WPA_ALG_IGTK:
4209                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4210                                     WLAN_CIPHER_SUITE_AES_CMAC);
4211                         break;
4212                 default:
4213                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
4214                                    "algorithm %d", __func__, alg);
4215                         nlmsg_free(msg);
4216                         return -1;
4217                 }
4218         }
4219
4220         if (seq && seq_len)
4221                 NLA_PUT(msg, NL80211_ATTR_KEY_SEQ, seq_len, seq);
4222
4223         if (addr && !is_broadcast_ether_addr(addr)) {
4224                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
4225                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4226
4227                 if (alg != WPA_ALG_WEP && key_idx && !set_tx) {
4228                         wpa_printf(MSG_DEBUG, "   RSN IBSS RX GTK");
4229                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_TYPE,
4230                                     NL80211_KEYTYPE_GROUP);
4231                 }
4232         } else if (addr && is_broadcast_ether_addr(addr)) {
4233                 struct nl_msg *types;
4234                 int err;
4235                 wpa_printf(MSG_DEBUG, "   broadcast key");
4236                 types = nlmsg_alloc();
4237                 if (!types)
4238                         goto nla_put_failure;
4239                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_MULTICAST);
4240                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4241                                      types);
4242                 nlmsg_free(types);
4243                 if (err)
4244                         goto nla_put_failure;
4245         }
4246         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
4247         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
4248
4249         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4250         if ((ret == -ENOENT || ret == -ENOLINK) && alg == WPA_ALG_NONE)
4251                 ret = 0;
4252         if (ret)
4253                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d %s)",
4254                            ret, strerror(-ret));
4255
4256         /*
4257          * If we failed or don't need to set the default TX key (below),
4258          * we're done here.
4259          */
4260         if (ret || !set_tx || alg == WPA_ALG_NONE)
4261                 return ret;
4262         if (is_ap_interface(drv->nlmode) && addr &&
4263             !is_broadcast_ether_addr(addr))
4264                 return ret;
4265
4266         msg = nlmsg_alloc();
4267         if (!msg)
4268                 return -ENOMEM;
4269
4270         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_KEY);
4271         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
4272         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
4273         if (alg == WPA_ALG_IGTK)
4274                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
4275         else
4276                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
4277         if (addr && is_broadcast_ether_addr(addr)) {
4278                 struct nl_msg *types;
4279                 int err;
4280                 types = nlmsg_alloc();
4281                 if (!types)
4282                         goto nla_put_failure;
4283                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_MULTICAST);
4284                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4285                                      types);
4286                 nlmsg_free(types);
4287                 if (err)
4288                         goto nla_put_failure;
4289         } else if (addr) {
4290                 struct nl_msg *types;
4291                 int err;
4292                 types = nlmsg_alloc();
4293                 if (!types)
4294                         goto nla_put_failure;
4295                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_UNICAST);
4296                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4297                                      types);
4298                 nlmsg_free(types);
4299                 if (err)
4300                         goto nla_put_failure;
4301         }
4302
4303         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4304         if (ret == -ENOENT)
4305                 ret = 0;
4306         if (ret)
4307                 wpa_printf(MSG_DEBUG, "nl80211: set_key default failed; "
4308                            "err=%d %s)", ret, strerror(-ret));
4309         return ret;
4310
4311 nla_put_failure:
4312         nlmsg_free(msg);
4313         return -ENOBUFS;
4314 }
4315
4316
4317 static int nl_add_key(struct nl_msg *msg, enum wpa_alg alg,
4318                       int key_idx, int defkey,
4319                       const u8 *seq, size_t seq_len,
4320                       const u8 *key, size_t key_len)
4321 {
4322         struct nlattr *key_attr = nla_nest_start(msg, NL80211_ATTR_KEY);
4323         if (!key_attr)
4324                 return -1;
4325
4326         if (defkey && alg == WPA_ALG_IGTK)
4327                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT_MGMT);
4328         else if (defkey)
4329                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
4330
4331         NLA_PUT_U8(msg, NL80211_KEY_IDX, key_idx);
4332
4333         switch (alg) {
4334         case WPA_ALG_WEP:
4335                 if (key_len == 5)
4336                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4337                                     WLAN_CIPHER_SUITE_WEP40);
4338                 else
4339                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4340                                     WLAN_CIPHER_SUITE_WEP104);
4341                 break;
4342         case WPA_ALG_TKIP:
4343                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_TKIP);
4344                 break;
4345         case WPA_ALG_CCMP:
4346                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_CCMP);
4347                 break;
4348         case WPA_ALG_IGTK:
4349                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4350                             WLAN_CIPHER_SUITE_AES_CMAC);
4351                 break;
4352         default:
4353                 wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
4354                            "algorithm %d", __func__, alg);
4355                 return -1;
4356         }
4357
4358         if (seq && seq_len)
4359                 NLA_PUT(msg, NL80211_KEY_SEQ, seq_len, seq);
4360
4361         NLA_PUT(msg, NL80211_KEY_DATA, key_len, key);
4362
4363         nla_nest_end(msg, key_attr);
4364
4365         return 0;
4366  nla_put_failure:
4367         return -1;
4368 }
4369
4370
4371 static int nl80211_set_conn_keys(struct wpa_driver_associate_params *params,
4372                                  struct nl_msg *msg)
4373 {
4374         int i, privacy = 0;
4375         struct nlattr *nl_keys, *nl_key;
4376
4377         for (i = 0; i < 4; i++) {
4378                 if (!params->wep_key[i])
4379                         continue;
4380                 privacy = 1;
4381                 break;
4382         }
4383         if (params->wps == WPS_MODE_PRIVACY)
4384                 privacy = 1;
4385         if (params->pairwise_suite &&
4386             params->pairwise_suite != WPA_CIPHER_NONE)
4387                 privacy = 1;
4388
4389         if (!privacy)
4390                 return 0;
4391
4392         NLA_PUT_FLAG(msg, NL80211_ATTR_PRIVACY);
4393
4394         nl_keys = nla_nest_start(msg, NL80211_ATTR_KEYS);
4395         if (!nl_keys)
4396                 goto nla_put_failure;
4397
4398         for (i = 0; i < 4; i++) {
4399                 if (!params->wep_key[i])
4400                         continue;
4401
4402                 nl_key = nla_nest_start(msg, i);
4403                 if (!nl_key)
4404                         goto nla_put_failure;
4405
4406                 NLA_PUT(msg, NL80211_KEY_DATA, params->wep_key_len[i],
4407                         params->wep_key[i]);
4408                 if (params->wep_key_len[i] == 5)
4409                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4410                                     WLAN_CIPHER_SUITE_WEP40);
4411                 else
4412                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4413                                     WLAN_CIPHER_SUITE_WEP104);
4414
4415                 NLA_PUT_U8(msg, NL80211_KEY_IDX, i);
4416
4417                 if (i == params->wep_tx_keyidx)
4418                         NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
4419
4420                 nla_nest_end(msg, nl_key);
4421         }
4422         nla_nest_end(msg, nl_keys);
4423
4424         return 0;
4425
4426 nla_put_failure:
4427         return -ENOBUFS;
4428 }
4429
4430
4431 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
4432                                    const u8 *addr, int cmd, u16 reason_code,
4433                                    int local_state_change)
4434 {
4435         int ret = -1;
4436         struct nl_msg *msg;
4437
4438         msg = nlmsg_alloc();
4439         if (!msg)
4440                 return -1;
4441
4442         nl80211_cmd(drv, msg, 0, cmd);
4443
4444         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4445         NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason_code);
4446         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4447         if (local_state_change)
4448                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
4449
4450         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4451         msg = NULL;
4452         if (ret) {
4453                 wpa_dbg(drv->ctx, MSG_DEBUG,
4454                         "nl80211: MLME command failed: reason=%u ret=%d (%s)",
4455                         reason_code, ret, strerror(-ret));
4456                 goto nla_put_failure;
4457         }
4458         ret = 0;
4459
4460 nla_put_failure:
4461         nlmsg_free(msg);
4462         return ret;
4463 }
4464
4465
4466 static int wpa_driver_nl80211_disconnect(struct wpa_driver_nl80211_data *drv,
4467                                          const u8 *addr, int reason_code)
4468 {
4469         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
4470                    __func__, MAC2STR(addr), reason_code);
4471         drv->associated = 0;
4472         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISCONNECT,
4473                                        reason_code, 0);
4474 }
4475
4476
4477 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
4478                                              int reason_code)
4479 {
4480         struct i802_bss *bss = priv;
4481         struct wpa_driver_nl80211_data *drv = bss->drv;
4482         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
4483                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
4484         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
4485                    __func__, MAC2STR(addr), reason_code);
4486         drv->associated = 0;
4487         if (drv->nlmode == NL80211_IFTYPE_ADHOC)
4488                 return nl80211_leave_ibss(drv);
4489         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DEAUTHENTICATE,
4490                                        reason_code, 0);
4491 }
4492
4493
4494 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
4495                                            int reason_code)
4496 {
4497         struct i802_bss *bss = priv;
4498         struct wpa_driver_nl80211_data *drv = bss->drv;
4499         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
4500                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
4501         wpa_printf(MSG_DEBUG, "%s", __func__);
4502         drv->associated = 0;
4503         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISASSOCIATE,
4504                                        reason_code, 0);
4505 }
4506
4507
4508 static void nl80211_copy_auth_params(struct wpa_driver_nl80211_data *drv,
4509                                      struct wpa_driver_auth_params *params)
4510 {
4511         int i;
4512
4513         drv->auth_freq = params->freq;
4514         drv->auth_alg = params->auth_alg;
4515         drv->auth_wep_tx_keyidx = params->wep_tx_keyidx;
4516         drv->auth_local_state_change = params->local_state_change;
4517         drv->auth_p2p = params->p2p;
4518
4519         if (params->bssid)
4520                 os_memcpy(drv->auth_bssid_, params->bssid, ETH_ALEN);
4521         else
4522                 os_memset(drv->auth_bssid_, 0, ETH_ALEN);
4523
4524         if (params->ssid) {
4525                 os_memcpy(drv->auth_ssid, params->ssid, params->ssid_len);
4526                 drv->auth_ssid_len = params->ssid_len;
4527         } else
4528                 drv->auth_ssid_len = 0;
4529
4530
4531         os_free(drv->auth_ie);
4532         drv->auth_ie = NULL;
4533         drv->auth_ie_len = 0;
4534         if (params->ie) {
4535                 drv->auth_ie = os_malloc(params->ie_len);
4536                 if (drv->auth_ie) {
4537                         os_memcpy(drv->auth_ie, params->ie, params->ie_len);
4538                         drv->auth_ie_len = params->ie_len;
4539                 }
4540         }
4541
4542         for (i = 0; i < 4; i++) {
4543                 if (params->wep_key[i] && params->wep_key_len[i] &&
4544                     params->wep_key_len[i] <= 16) {
4545                         os_memcpy(drv->auth_wep_key[i], params->wep_key[i],
4546                                   params->wep_key_len[i]);
4547                         drv->auth_wep_key_len[i] = params->wep_key_len[i];
4548                 } else
4549                         drv->auth_wep_key_len[i] = 0;
4550         }
4551 }
4552
4553
4554 static int wpa_driver_nl80211_authenticate(
4555         void *priv, struct wpa_driver_auth_params *params)
4556 {
4557         struct i802_bss *bss = priv;
4558         struct wpa_driver_nl80211_data *drv = bss->drv;
4559         int ret = -1, i;
4560         struct nl_msg *msg;
4561         enum nl80211_auth_type type;
4562         enum nl80211_iftype nlmode;
4563         int count = 0;
4564         int is_retry;
4565
4566         is_retry = drv->retry_auth;
4567         drv->retry_auth = 0;
4568
4569         drv->associated = 0;
4570         os_memset(drv->auth_bssid, 0, ETH_ALEN);
4571         /* FIX: IBSS mode */
4572         nlmode = params->p2p ?
4573                 NL80211_IFTYPE_P2P_CLIENT : NL80211_IFTYPE_STATION;
4574         if (drv->nlmode != nlmode &&
4575             wpa_driver_nl80211_set_mode(priv, nlmode) < 0)
4576                 return -1;
4577
4578 retry:
4579         msg = nlmsg_alloc();
4580         if (!msg)
4581                 return -1;
4582
4583         wpa_printf(MSG_DEBUG, "nl80211: Authenticate (ifindex=%d)",
4584                    drv->ifindex);
4585
4586         nl80211_cmd(drv, msg, 0, NL80211_CMD_AUTHENTICATE);
4587
4588         for (i = 0; i < 4; i++) {
4589                 if (!params->wep_key[i])
4590                         continue;
4591                 wpa_driver_nl80211_set_key(bss->ifname, priv, WPA_ALG_WEP,
4592                                            NULL, i,
4593                                            i == params->wep_tx_keyidx, NULL, 0,
4594                                            params->wep_key[i],
4595                                            params->wep_key_len[i]);
4596                 if (params->wep_tx_keyidx != i)
4597                         continue;
4598                 if (nl_add_key(msg, WPA_ALG_WEP, i, 1, NULL, 0,
4599                                params->wep_key[i], params->wep_key_len[i])) {
4600                         nlmsg_free(msg);
4601                         return -1;
4602                 }
4603         }
4604
4605         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4606         if (params->bssid) {
4607                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
4608                            MAC2STR(params->bssid));
4609                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
4610         }
4611         if (params->freq) {
4612                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
4613                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
4614         }
4615         if (params->ssid) {
4616                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
4617                                   params->ssid, params->ssid_len);
4618                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
4619                         params->ssid);
4620         }
4621         wpa_hexdump(MSG_DEBUG, "  * IEs", params->ie, params->ie_len);
4622         if (params->ie)
4623                 NLA_PUT(msg, NL80211_ATTR_IE, params->ie_len, params->ie);
4624         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
4625                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
4626         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
4627                 type = NL80211_AUTHTYPE_SHARED_KEY;
4628         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
4629                 type = NL80211_AUTHTYPE_NETWORK_EAP;
4630         else if (params->auth_alg & WPA_AUTH_ALG_FT)
4631                 type = NL80211_AUTHTYPE_FT;
4632         else
4633                 goto nla_put_failure;
4634         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
4635         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
4636         if (params->local_state_change) {
4637                 wpa_printf(MSG_DEBUG, "  * Local state change only");
4638                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
4639         }
4640
4641         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4642         msg = NULL;
4643         if (ret) {
4644                 wpa_dbg(drv->ctx, MSG_DEBUG,
4645                         "nl80211: MLME command failed (auth): ret=%d (%s)",
4646                         ret, strerror(-ret));
4647                 count++;
4648                 if (ret == -EALREADY && count == 1 && params->bssid &&
4649                     !params->local_state_change) {
4650                         /*
4651                          * mac80211 does not currently accept new
4652                          * authentication if we are already authenticated. As a
4653                          * workaround, force deauthentication and try again.
4654                          */
4655                         wpa_printf(MSG_DEBUG, "nl80211: Retry authentication "
4656                                    "after forced deauthentication");
4657                         wpa_driver_nl80211_deauthenticate(
4658                                 bss, params->bssid,
4659                                 WLAN_REASON_PREV_AUTH_NOT_VALID);
4660                         nlmsg_free(msg);
4661                         goto retry;
4662                 }
4663
4664                 if (ret == -ENOENT && params->freq && !is_retry) {
4665                         /*
4666                          * cfg80211 has likely expired the BSS entry even
4667                          * though it was previously available in our internal
4668                          * BSS table. To recover quickly, start a single
4669                          * channel scan on the specified channel.
4670                          */
4671                         struct wpa_driver_scan_params scan;
4672                         int freqs[2];
4673
4674                         os_memset(&scan, 0, sizeof(scan));
4675                         scan.num_ssids = 1;
4676                         if (params->ssid) {
4677                                 scan.ssids[0].ssid = params->ssid;
4678                                 scan.ssids[0].ssid_len = params->ssid_len;
4679                         }
4680                         freqs[0] = params->freq;
4681                         freqs[1] = 0;
4682                         scan.freqs = freqs;
4683                         wpa_printf(MSG_DEBUG, "nl80211: Trigger single "
4684                                    "channel scan to refresh cfg80211 BSS "
4685                                    "entry");
4686                         ret = wpa_driver_nl80211_scan(bss, &scan);
4687                         if (ret == 0) {
4688                                 nl80211_copy_auth_params(drv, params);
4689                                 drv->scan_for_auth = 1;
4690                         }
4691                 } else if (is_retry) {
4692                         /*
4693                          * Need to indicate this with an event since the return
4694                          * value from the retry is not delivered to core code.
4695                          */
4696                         union wpa_event_data event;
4697                         wpa_printf(MSG_DEBUG, "nl80211: Authentication retry "
4698                                    "failed");
4699                         os_memset(&event, 0, sizeof(event));
4700                         os_memcpy(event.timeout_event.addr, drv->auth_bssid_,
4701                                   ETH_ALEN);
4702                         wpa_supplicant_event(drv->ctx, EVENT_AUTH_TIMED_OUT,
4703                                              &event);
4704                 }
4705
4706                 goto nla_put_failure;
4707         }
4708         ret = 0;
4709         wpa_printf(MSG_DEBUG, "nl80211: Authentication request send "
4710                    "successfully");
4711
4712 nla_put_failure:
4713         nlmsg_free(msg);
4714         return ret;
4715 }
4716
4717
4718 static int wpa_driver_nl80211_authenticate_retry(
4719         struct wpa_driver_nl80211_data *drv)
4720 {
4721         struct wpa_driver_auth_params params;
4722         struct i802_bss *bss = &drv->first_bss;
4723         int i;
4724
4725         wpa_printf(MSG_DEBUG, "nl80211: Try to authenticate again");
4726
4727         os_memset(&params, 0, sizeof(params));
4728         params.freq = drv->auth_freq;
4729         params.auth_alg = drv->auth_alg;
4730         params.wep_tx_keyidx = drv->auth_wep_tx_keyidx;
4731         params.local_state_change = drv->auth_local_state_change;
4732         params.p2p = drv->auth_p2p;
4733
4734         if (!is_zero_ether_addr(drv->auth_bssid_))
4735                 params.bssid = drv->auth_bssid_;
4736
4737         if (drv->auth_ssid_len) {
4738                 params.ssid = drv->auth_ssid;
4739                 params.ssid_len = drv->auth_ssid_len;
4740         }
4741
4742         params.ie = drv->auth_ie;
4743         params.ie_len = drv->auth_ie_len;
4744
4745         for (i = 0; i < 4; i++) {
4746                 if (drv->auth_wep_key_len[i]) {
4747                         params.wep_key[i] = drv->auth_wep_key[i];
4748                         params.wep_key_len[i] = drv->auth_wep_key_len[i];
4749                 }
4750         }
4751
4752         drv->retry_auth = 1;
4753         return wpa_driver_nl80211_authenticate(bss, &params);
4754 }
4755
4756
4757 struct phy_info_arg {
4758         u16 *num_modes;
4759         struct hostapd_hw_modes *modes;
4760 };
4761
4762 static int phy_info_handler(struct nl_msg *msg, void *arg)
4763 {
4764         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
4765         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
4766         struct phy_info_arg *phy_info = arg;
4767
4768         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
4769
4770         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
4771         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
4772                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
4773                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
4774                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
4775                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
4776                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
4777                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
4778         };
4779
4780         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
4781         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
4782                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
4783                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
4784         };
4785
4786         struct nlattr *nl_band;
4787         struct nlattr *nl_freq;
4788         struct nlattr *nl_rate;
4789         int rem_band, rem_freq, rem_rate;
4790         struct hostapd_hw_modes *mode;
4791         int idx, mode_is_set;
4792
4793         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
4794                   genlmsg_attrlen(gnlh, 0), NULL);
4795
4796         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
4797                 return NL_SKIP;
4798
4799         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
4800                 mode = os_realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
4801                 if (!mode)
4802                         return NL_SKIP;
4803                 phy_info->modes = mode;
4804
4805                 mode_is_set = 0;
4806
4807                 mode = &phy_info->modes[*(phy_info->num_modes)];
4808                 memset(mode, 0, sizeof(*mode));
4809                 mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN;
4810                 *(phy_info->num_modes) += 1;
4811
4812                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
4813                           nla_len(nl_band), NULL);
4814
4815                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
4816                         mode->ht_capab = nla_get_u16(
4817                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
4818                 }
4819
4820                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) {
4821                         mode->a_mpdu_params |= nla_get_u8(
4822                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) &
4823                                 0x03;
4824                 }
4825
4826                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) {
4827                         mode->a_mpdu_params |= nla_get_u8(
4828                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) <<
4829                                 2;
4830                 }
4831
4832                 if (tb_band[NL80211_BAND_ATTR_HT_MCS_SET] &&
4833                     nla_len(tb_band[NL80211_BAND_ATTR_HT_MCS_SET])) {
4834                         u8 *mcs;
4835                         mcs = nla_data(tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
4836                         os_memcpy(mode->mcs_set, mcs, 16);
4837                 }
4838
4839                 if (tb_band[NL80211_BAND_ATTR_VHT_CAPA]) {
4840                         mode->vht_capab = nla_get_u32(
4841                                 tb_band[NL80211_BAND_ATTR_VHT_CAPA]);
4842                 }
4843
4844                 if (tb_band[NL80211_BAND_ATTR_VHT_MCS_SET] &&
4845                     nla_len(tb_band[NL80211_BAND_ATTR_VHT_MCS_SET])) {
4846                         u8 *mcs;
4847                         mcs = nla_data(tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
4848                         os_memcpy(mode->vht_mcs_set, mcs, 8);
4849                 }
4850
4851                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
4852                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
4853                                   nla_len(nl_freq), freq_policy);
4854                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
4855                                 continue;
4856                         mode->num_channels++;
4857                 }
4858
4859                 mode->channels = os_zalloc(mode->num_channels * sizeof(struct hostapd_channel_data));
4860                 if (!mode->channels)
4861                         return NL_SKIP;
4862
4863                 idx = 0;
4864
4865                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
4866                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
4867                                   nla_len(nl_freq), freq_policy);
4868                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
4869                                 continue;
4870
4871                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
4872                         mode->channels[idx].flag = 0;
4873
4874                         if (!mode_is_set) {
4875                                 /* crude heuristic */
4876                                 if (mode->channels[idx].freq < 4000)
4877                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
4878                                 else
4879                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
4880                                 mode_is_set = 1;
4881                         }
4882
4883                         /* crude heuristic */
4884                         if (mode->channels[idx].freq < 4000)
4885                                 if (mode->channels[idx].freq == 2484)
4886                                         mode->channels[idx].chan = 14;
4887                                 else
4888                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
4889                         else
4890                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
4891
4892                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
4893                                 mode->channels[idx].flag |=
4894                                         HOSTAPD_CHAN_DISABLED;
4895                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
4896                                 mode->channels[idx].flag |=
4897                                         HOSTAPD_CHAN_PASSIVE_SCAN;
4898                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
4899                                 mode->channels[idx].flag |=
4900                                         HOSTAPD_CHAN_NO_IBSS;
4901                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
4902                                 mode->channels[idx].flag |=
4903                                         HOSTAPD_CHAN_RADAR;
4904
4905                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
4906                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
4907                                 mode->channels[idx].max_tx_power =
4908                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
4909
4910                         idx++;
4911                 }
4912
4913                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
4914                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
4915                                   nla_len(nl_rate), rate_policy);
4916                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
4917                                 continue;
4918                         mode->num_rates++;
4919                 }
4920
4921                 mode->rates = os_zalloc(mode->num_rates * sizeof(int));
4922                 if (!mode->rates)
4923                         return NL_SKIP;
4924
4925                 idx = 0;
4926
4927                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
4928                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
4929                                   nla_len(nl_rate), rate_policy);
4930                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
4931                                 continue;
4932                         mode->rates[idx] = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
4933
4934                         /* crude heuristic */
4935                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
4936                             mode->rates[idx] > 200)
4937                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
4938
4939                         idx++;
4940                 }
4941         }
4942
4943         return NL_SKIP;
4944 }
4945
4946 static struct hostapd_hw_modes *
4947 wpa_driver_nl80211_add_11b(struct hostapd_hw_modes *modes, u16 *num_modes)
4948 {
4949         u16 m;
4950         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
4951         int i, mode11g_idx = -1;
4952
4953         /* If only 802.11g mode is included, use it to construct matching
4954          * 802.11b mode data. */
4955
4956         for (m = 0; m < *num_modes; m++) {
4957                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
4958                         return modes; /* 802.11b already included */
4959                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
4960                         mode11g_idx = m;
4961         }
4962
4963         if (mode11g_idx < 0)
4964                 return modes; /* 2.4 GHz band not supported at all */
4965
4966         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
4967         if (nmodes == NULL)
4968                 return modes; /* Could not add 802.11b mode */
4969
4970         mode = &nmodes[*num_modes];
4971         os_memset(mode, 0, sizeof(*mode));
4972         (*num_modes)++;
4973         modes = nmodes;
4974
4975         mode->mode = HOSTAPD_MODE_IEEE80211B;
4976
4977         mode11g = &modes[mode11g_idx];
4978         mode->num_channels = mode11g->num_channels;
4979         mode->channels = os_malloc(mode11g->num_channels *
4980                                    sizeof(struct hostapd_channel_data));
4981         if (mode->channels == NULL) {
4982                 (*num_modes)--;
4983                 return modes; /* Could not add 802.11b mode */
4984         }
4985         os_memcpy(mode->channels, mode11g->channels,
4986                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
4987
4988         mode->num_rates = 0;
4989         mode->rates = os_malloc(4 * sizeof(int));
4990         if (mode->rates == NULL) {
4991                 os_free(mode->channels);
4992                 (*num_modes)--;
4993                 return modes; /* Could not add 802.11b mode */
4994         }
4995
4996         for (i = 0; i < mode11g->num_rates; i++) {
4997                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
4998                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
4999                         continue;
5000                 mode->rates[mode->num_rates] = mode11g->rates[i];
5001                 mode->num_rates++;
5002                 if (mode->num_rates == 4)
5003                         break;
5004         }
5005
5006         if (mode->num_rates == 0) {
5007                 os_free(mode->channels);
5008                 os_free(mode->rates);
5009                 (*num_modes)--;
5010                 return modes; /* No 802.11b rates */
5011         }
5012
5013         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
5014                    "information");
5015
5016         return modes;
5017 }
5018
5019
5020 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
5021                                   int end)
5022 {
5023         int c;
5024
5025         for (c = 0; c < mode->num_channels; c++) {
5026                 struct hostapd_channel_data *chan = &mode->channels[c];
5027                 if (chan->freq - 10 >= start && chan->freq + 10 <= end)
5028                         chan->flag |= HOSTAPD_CHAN_HT40;
5029         }
5030 }
5031
5032
5033 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
5034                                       int end)
5035 {
5036         int c;
5037
5038         for (c = 0; c < mode->num_channels; c++) {
5039                 struct hostapd_channel_data *chan = &mode->channels[c];
5040                 if (!(chan->flag & HOSTAPD_CHAN_HT40))
5041                         continue;
5042                 if (chan->freq - 30 >= start && chan->freq - 10 <= end)
5043                         chan->flag |= HOSTAPD_CHAN_HT40MINUS;
5044                 if (chan->freq + 10 >= start && chan->freq + 30 <= end)
5045                         chan->flag |= HOSTAPD_CHAN_HT40PLUS;
5046         }
5047 }
5048
5049
5050 static void nl80211_reg_rule_ht40(struct nlattr *tb[],
5051                                   struct phy_info_arg *results)
5052 {
5053         u32 start, end, max_bw;
5054         u16 m;
5055
5056         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
5057             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
5058             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
5059                 return;
5060
5061         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
5062         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
5063         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
5064
5065         wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz",
5066                    start, end, max_bw);
5067         if (max_bw < 40)
5068                 return;
5069
5070         for (m = 0; m < *results->num_modes; m++) {
5071                 if (!(results->modes[m].ht_capab &
5072                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
5073                         continue;
5074                 nl80211_set_ht40_mode(&results->modes[m], start, end);
5075         }
5076 }
5077
5078
5079 static void nl80211_reg_rule_sec(struct nlattr *tb[],
5080                                  struct phy_info_arg *results)
5081 {
5082         u32 start, end, max_bw;
5083         u16 m;
5084
5085         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
5086             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
5087             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
5088                 return;
5089
5090         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
5091         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
5092         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
5093
5094         if (max_bw < 20)
5095                 return;
5096
5097         for (m = 0; m < *results->num_modes; m++) {
5098                 if (!(results->modes[m].ht_capab &
5099                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
5100                         continue;
5101                 nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
5102         }
5103 }
5104
5105
5106 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
5107 {
5108         struct phy_info_arg *results = arg;
5109         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
5110         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
5111         struct nlattr *nl_rule;
5112         struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
5113         int rem_rule;
5114         static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
5115                 [NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
5116                 [NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
5117                 [NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
5118                 [NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
5119                 [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
5120                 [NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
5121         };
5122
5123         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
5124                   genlmsg_attrlen(gnlh, 0), NULL);
5125         if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
5126             !tb_msg[NL80211_ATTR_REG_RULES]) {
5127                 wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
5128                            "available");
5129                 return NL_SKIP;
5130         }
5131
5132         wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
5133                    (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
5134
5135         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
5136         {
5137                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
5138                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
5139                 nl80211_reg_rule_ht40(tb_rule, results);
5140         }
5141
5142         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
5143         {
5144                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
5145                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
5146                 nl80211_reg_rule_sec(tb_rule, results);
5147         }
5148
5149         return NL_SKIP;
5150 }
5151
5152
5153 static int nl80211_set_ht40_flags(struct wpa_driver_nl80211_data *drv,
5154                                   struct phy_info_arg *results)
5155 {
5156         struct nl_msg *msg;
5157
5158         msg = nlmsg_alloc();
5159         if (!msg)
5160                 return -ENOMEM;
5161
5162         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
5163         return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
5164 }
5165
5166
5167 static struct hostapd_hw_modes *
5168 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
5169 {
5170         struct i802_bss *bss = priv;
5171         struct wpa_driver_nl80211_data *drv = bss->drv;
5172         struct nl_msg *msg;
5173         struct phy_info_arg result = {
5174                 .num_modes = num_modes,
5175                 .modes = NULL,
5176         };
5177
5178         *num_modes = 0;
5179         *flags = 0;
5180
5181         msg = nlmsg_alloc();
5182         if (!msg)
5183                 return NULL;
5184
5185         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_WIPHY);
5186
5187         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5188
5189         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
5190                 nl80211_set_ht40_flags(drv, &result);
5191                 return wpa_driver_nl80211_add_11b(result.modes, num_modes);
5192         }
5193         msg = NULL;
5194  nla_put_failure:
5195         nlmsg_free(msg);
5196         return NULL;
5197 }
5198
5199
5200 static int wpa_driver_nl80211_send_mntr(struct wpa_driver_nl80211_data *drv,
5201                                         const void *data, size_t len,
5202                                         int encrypt, int noack)
5203 {
5204         __u8 rtap_hdr[] = {
5205                 0x00, 0x00, /* radiotap version */
5206                 0x0e, 0x00, /* radiotap length */
5207                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
5208                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
5209                 0x00,       /* padding */
5210                 0x00, 0x00, /* RX and TX flags to indicate that */
5211                 0x00, 0x00, /* this is the injected frame directly */
5212         };
5213         struct iovec iov[2] = {
5214                 {
5215                         .iov_base = &rtap_hdr,
5216                         .iov_len = sizeof(rtap_hdr),
5217                 },
5218                 {
5219                         .iov_base = (void *) data,
5220                         .iov_len = len,
5221                 }
5222         };
5223         struct msghdr msg = {
5224                 .msg_name = NULL,
5225                 .msg_namelen = 0,
5226                 .msg_iov = iov,
5227                 .msg_iovlen = 2,
5228                 .msg_control = NULL,
5229                 .msg_controllen = 0,
5230                 .msg_flags = 0,
5231         };
5232         int res;
5233         u16 txflags = 0;
5234
5235         if (encrypt)
5236                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
5237
5238         if (drv->monitor_sock < 0) {
5239                 wpa_printf(MSG_DEBUG, "nl80211: No monitor socket available "
5240                            "for %s", __func__);
5241                 return -1;
5242         }
5243
5244         if (noack)
5245                 txflags |= IEEE80211_RADIOTAP_F_TX_NOACK;
5246         *(le16 *) &rtap_hdr[12] = host_to_le16(txflags);
5247
5248         res = sendmsg(drv->monitor_sock, &msg, 0);
5249         if (res < 0) {
5250                 wpa_printf(MSG_INFO, "nl80211: sendmsg: %s", strerror(errno));
5251                 return -1;
5252         }
5253         return 0;
5254 }
5255
5256
5257 static int wpa_driver_nl80211_send_frame(struct i802_bss *bss,
5258                                          const void *data, size_t len,
5259                                          int encrypt, int noack,
5260                                          unsigned int freq, int no_cck,
5261                                          int offchanok, unsigned int wait_time)
5262 {
5263         struct wpa_driver_nl80211_data *drv = bss->drv;
5264         u64 cookie;
5265
5266         if (freq == 0)
5267                 freq = bss->freq;
5268
5269         if (drv->use_monitor)
5270                 return wpa_driver_nl80211_send_mntr(drv, data, len,
5271                                                     encrypt, noack);
5272
5273         return nl80211_send_frame_cmd(bss, freq, wait_time, data, len,
5274                                       &cookie, no_cck, noack, offchanok);
5275 }
5276
5277
5278 static int wpa_driver_nl80211_send_mlme_freq(struct i802_bss *bss,
5279                                              const u8 *data,
5280                                              size_t data_len, int noack,
5281                                              unsigned int freq, int no_cck,
5282                                              int offchanok,
5283                                              unsigned int wait_time)
5284 {
5285         struct wpa_driver_nl80211_data *drv = bss->drv;
5286         struct ieee80211_mgmt *mgmt;
5287         int encrypt = 1;
5288         u16 fc;
5289
5290         mgmt = (struct ieee80211_mgmt *) data;
5291         fc = le_to_host16(mgmt->frame_control);
5292
5293         if (is_sta_interface(drv->nlmode) &&
5294             WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
5295             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_PROBE_RESP) {
5296                 /*
5297                  * The use of last_mgmt_freq is a bit of a hack,
5298                  * but it works due to the single-threaded nature
5299                  * of wpa_supplicant.
5300                  */
5301                 if (freq == 0)
5302                         freq = drv->last_mgmt_freq;
5303                 return nl80211_send_frame_cmd(bss, freq, 0,
5304                                               data, data_len, NULL, 1, noack,
5305                                               1);
5306         }
5307
5308         if (drv->device_ap_sme && is_ap_interface(drv->nlmode)) {
5309                 if (freq == 0)
5310                         freq = bss->freq;
5311                 return nl80211_send_frame_cmd(bss, freq,
5312                                               (int) freq == bss->freq ? 0 :
5313                                               wait_time,
5314                                               data, data_len,
5315                                               &drv->send_action_cookie,
5316                                               no_cck, noack, offchanok);
5317         }
5318
5319         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
5320             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
5321                 /*
5322                  * Only one of the authentication frame types is encrypted.
5323                  * In order for static WEP encryption to work properly (i.e.,
5324                  * to not encrypt the frame), we need to tell mac80211 about
5325                  * the frames that must not be encrypted.
5326                  */
5327                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
5328                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
5329                 if (auth_alg != WLAN_AUTH_SHARED_KEY || auth_trans != 3)
5330                         encrypt = 0;
5331         }
5332
5333         return wpa_driver_nl80211_send_frame(bss, data, data_len, encrypt,
5334                                              noack, freq, no_cck, offchanok,
5335                                              wait_time);
5336 }
5337
5338
5339 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
5340                                         size_t data_len, int noack)
5341 {
5342         struct i802_bss *bss = priv;
5343         return wpa_driver_nl80211_send_mlme_freq(bss, data, data_len, noack,
5344                                                  0, 0, 0, 0);
5345 }
5346
5347
5348 static int nl80211_set_bss(struct i802_bss *bss, int cts, int preamble,
5349                            int slot, int ht_opmode, int ap_isolate,
5350                            int *basic_rates)
5351 {
5352         struct wpa_driver_nl80211_data *drv = bss->drv;
5353         struct nl_msg *msg;
5354
5355         msg = nlmsg_alloc();
5356         if (!msg)
5357                 return -ENOMEM;
5358
5359         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_BSS);
5360
5361         if (cts >= 0)
5362                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
5363         if (preamble >= 0)
5364                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
5365         if (slot >= 0)
5366                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
5367         if (ht_opmode >= 0)
5368                 NLA_PUT_U16(msg, NL80211_ATTR_BSS_HT_OPMODE, ht_opmode);
5369         if (ap_isolate >= 0)
5370                 NLA_PUT_U8(msg, NL80211_ATTR_AP_ISOLATE, ap_isolate);
5371
5372         if (basic_rates) {
5373                 u8 rates[NL80211_MAX_SUPP_RATES];
5374                 u8 rates_len = 0;
5375                 int i;
5376
5377                 for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0;
5378                      i++)
5379                         rates[rates_len++] = basic_rates[i] / 5;
5380
5381                 NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
5382         }
5383
5384         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
5385
5386         return send_and_recv_msgs(drv, msg, NULL, NULL);
5387  nla_put_failure:
5388         nlmsg_free(msg);
5389         return -ENOBUFS;
5390 }
5391
5392
5393 static int wpa_driver_nl80211_set_ap(void *priv,
5394                                      struct wpa_driver_ap_params *params)
5395 {
5396         struct i802_bss *bss = priv;
5397         struct wpa_driver_nl80211_data *drv = bss->drv;
5398         struct nl_msg *msg;
5399         u8 cmd = NL80211_CMD_NEW_BEACON;
5400         int ret;
5401         int beacon_set;
5402         int ifindex = if_nametoindex(bss->ifname);
5403         int num_suites;
5404         u32 suites[10];
5405         u32 ver;
5406
5407         beacon_set = bss->beacon_set;
5408
5409         msg = nlmsg_alloc();
5410         if (!msg)
5411                 return -ENOMEM;
5412
5413         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (beacon_set=%d)",
5414                    beacon_set);
5415         if (beacon_set)
5416                 cmd = NL80211_CMD_SET_BEACON;
5417
5418         nl80211_cmd(drv, msg, 0, cmd);
5419         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, params->head_len, params->head);
5420         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, params->tail_len, params->tail);
5421         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
5422         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, params->beacon_int);
5423         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, params->dtim_period);
5424         NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
5425                 params->ssid);
5426         if (params->proberesp && params->proberesp_len)
5427                 NLA_PUT(msg, NL80211_ATTR_PROBE_RESP, params->proberesp_len,
5428                         params->proberesp);
5429         switch (params->hide_ssid) {
5430         case NO_SSID_HIDING:
5431                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5432                             NL80211_HIDDEN_SSID_NOT_IN_USE);
5433                 break;
5434         case HIDDEN_SSID_ZERO_LEN:
5435                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5436                             NL80211_HIDDEN_SSID_ZERO_LEN);
5437                 break;
5438         case HIDDEN_SSID_ZERO_CONTENTS:
5439                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5440                             NL80211_HIDDEN_SSID_ZERO_CONTENTS);
5441                 break;
5442         }
5443         if (params->privacy)
5444                 NLA_PUT_FLAG(msg, NL80211_ATTR_PRIVACY);
5445         if ((params->auth_algs & (WPA_AUTH_ALG_OPEN | WPA_AUTH_ALG_SHARED)) ==
5446             (WPA_AUTH_ALG_OPEN | WPA_AUTH_ALG_SHARED)) {
5447                 /* Leave out the attribute */
5448         } else if (params->auth_algs & WPA_AUTH_ALG_SHARED)
5449                 NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
5450                             NL80211_AUTHTYPE_SHARED_KEY);
5451         else
5452                 NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
5453                             NL80211_AUTHTYPE_OPEN_SYSTEM);
5454
5455         ver = 0;
5456         if (params->wpa_version & WPA_PROTO_WPA)
5457                 ver |= NL80211_WPA_VERSION_1;
5458         if (params->wpa_version & WPA_PROTO_RSN)
5459                 ver |= NL80211_WPA_VERSION_2;
5460         if (ver)
5461                 NLA_PUT_U32(msg, NL80211_ATTR_WPA_VERSIONS, ver);
5462
5463         num_suites = 0;
5464         if (params->key_mgmt_suites & WPA_KEY_MGMT_IEEE8021X)
5465                 suites[num_suites++] = WLAN_AKM_SUITE_8021X;
5466         if (params->key_mgmt_suites & WPA_KEY_MGMT_PSK)
5467                 suites[num_suites++] = WLAN_AKM_SUITE_PSK;
5468         if (num_suites) {
5469                 NLA_PUT(msg, NL80211_ATTR_AKM_SUITES,
5470                         num_suites * sizeof(u32), suites);
5471         }
5472
5473         if (params->key_mgmt_suites & WPA_KEY_MGMT_IEEE8021X &&
5474             params->pairwise_ciphers & (WPA_CIPHER_WEP104 | WPA_CIPHER_WEP40))
5475                 NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT);
5476
5477         num_suites = 0;
5478         if (params->pairwise_ciphers & WPA_CIPHER_CCMP)
5479                 suites[num_suites++] = WLAN_CIPHER_SUITE_CCMP;
5480         if (params->pairwise_ciphers & WPA_CIPHER_TKIP)
5481                 suites[num_suites++] = WLAN_CIPHER_SUITE_TKIP;
5482         if (params->pairwise_ciphers & WPA_CIPHER_WEP104)
5483                 suites[num_suites++] = WLAN_CIPHER_SUITE_WEP104;
5484         if (params->pairwise_ciphers & WPA_CIPHER_WEP40)
5485                 suites[num_suites++] = WLAN_CIPHER_SUITE_WEP40;
5486         if (num_suites) {
5487                 NLA_PUT(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE,
5488                         num_suites * sizeof(u32), suites);
5489         }
5490
5491         switch (params->group_cipher) {
5492         case WPA_CIPHER_CCMP:
5493                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5494                             WLAN_CIPHER_SUITE_CCMP);
5495                 break;
5496         case WPA_CIPHER_TKIP:
5497                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5498                             WLAN_CIPHER_SUITE_TKIP);
5499                 break;
5500         case WPA_CIPHER_WEP104:
5501                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5502                             WLAN_CIPHER_SUITE_WEP104);
5503                 break;
5504         case WPA_CIPHER_WEP40:
5505                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5506                             WLAN_CIPHER_SUITE_WEP40);
5507                 break;
5508         }
5509
5510         if (params->beacon_ies) {
5511                 NLA_PUT(msg, NL80211_ATTR_IE, wpabuf_len(params->beacon_ies),
5512                         wpabuf_head(params->beacon_ies));
5513         }
5514         if (params->proberesp_ies) {
5515                 NLA_PUT(msg, NL80211_ATTR_IE_PROBE_RESP,
5516                         wpabuf_len(params->proberesp_ies),
5517                         wpabuf_head(params->proberesp_ies));
5518         }
5519         if (params->assocresp_ies) {
5520                 NLA_PUT(msg, NL80211_ATTR_IE_ASSOC_RESP,
5521                         wpabuf_len(params->assocresp_ies),
5522                         wpabuf_head(params->assocresp_ies));
5523         }
5524
5525         if (drv->capa.flags & WPA_DRIVER_FLAGS_INACTIVITY_TIMER)  {
5526                 NLA_PUT_U16(msg, NL80211_ATTR_INACTIVITY_TIMEOUT,
5527                             params->ap_max_inactivity);
5528         }
5529
5530         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5531         if (ret) {
5532                 wpa_printf(MSG_DEBUG, "nl80211: Beacon set failed: %d (%s)",
5533                            ret, strerror(-ret));
5534         } else {
5535                 bss->beacon_set = 1;
5536                 nl80211_set_bss(bss, params->cts_protect, params->preamble,
5537                                 params->short_slot_time, params->ht_opmode,
5538                                 params->isolate, params->basic_rates);
5539         }
5540         return ret;
5541  nla_put_failure:
5542         nlmsg_free(msg);
5543         return -ENOBUFS;
5544 }
5545
5546
5547 static int wpa_driver_nl80211_set_freq(struct i802_bss *bss,
5548                                        int freq, int ht_enabled,
5549                                        int sec_channel_offset)
5550 {
5551         struct wpa_driver_nl80211_data *drv = bss->drv;
5552         struct nl_msg *msg;
5553         int ret;
5554
5555         wpa_printf(MSG_DEBUG, "nl80211: Set freq %d (ht_enabled=%d "
5556                    "sec_channel_offset=%d)",
5557                    freq, ht_enabled, sec_channel_offset);
5558         msg = nlmsg_alloc();
5559         if (!msg)
5560                 return -1;
5561
5562         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
5563
5564         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5565         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
5566         if (ht_enabled) {
5567                 switch (sec_channel_offset) {
5568                 case -1:
5569                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5570                                     NL80211_CHAN_HT40MINUS);
5571                         break;
5572                 case 1:
5573                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5574                                     NL80211_CHAN_HT40PLUS);
5575                         break;
5576                 default:
5577                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5578                                     NL80211_CHAN_HT20);
5579                         break;
5580                 }
5581         }
5582
5583         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5584         msg = NULL;
5585         if (ret == 0) {
5586                 bss->freq = freq;
5587                 return 0;
5588         }
5589         wpa_printf(MSG_DEBUG, "nl80211: Failed to set channel (freq=%d): "
5590                    "%d (%s)", freq, ret, strerror(-ret));
5591 nla_put_failure:
5592         nlmsg_free(msg);
5593         return -1;
5594 }
5595
5596
5597 static u32 sta_flags_nl80211(int flags)
5598 {
5599         u32 f = 0;
5600
5601         if (flags & WPA_STA_AUTHORIZED)
5602                 f |= BIT(NL80211_STA_FLAG_AUTHORIZED);
5603         if (flags & WPA_STA_WMM)
5604                 f |= BIT(NL80211_STA_FLAG_WME);
5605         if (flags & WPA_STA_SHORT_PREAMBLE)
5606                 f |= BIT(NL80211_STA_FLAG_SHORT_PREAMBLE);
5607         if (flags & WPA_STA_MFP)
5608                 f |= BIT(NL80211_STA_FLAG_MFP);
5609         if (flags & WPA_STA_TDLS_PEER)
5610                 f |= BIT(NL80211_STA_FLAG_TDLS_PEER);
5611
5612         return f;
5613 }
5614
5615
5616 static int wpa_driver_nl80211_sta_add(void *priv,
5617                                       struct hostapd_sta_add_params *params)
5618 {
5619         struct i802_bss *bss = priv;
5620         struct wpa_driver_nl80211_data *drv = bss->drv;
5621         struct nl_msg *msg, *wme = NULL;
5622         struct nl80211_sta_flag_update upd;
5623         int ret = -ENOBUFS;
5624
5625         if ((params->flags & WPA_STA_TDLS_PEER) &&
5626             !(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
5627                 return -EOPNOTSUPP;
5628
5629         msg = nlmsg_alloc();
5630         if (!msg)
5631                 return -ENOMEM;
5632
5633         nl80211_cmd(drv, msg, 0, params->set ? NL80211_CMD_SET_STATION :
5634                     NL80211_CMD_NEW_STATION);
5635
5636         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
5637         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
5638         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
5639                 params->supp_rates);
5640         if (!params->set) {
5641                 NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
5642                 NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
5643                             params->listen_interval);
5644         }
5645         if (params->ht_capabilities) {
5646                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
5647                         sizeof(*params->ht_capabilities),
5648                         params->ht_capabilities);
5649         }
5650
5651         os_memset(&upd, 0, sizeof(upd));
5652         upd.mask = sta_flags_nl80211(params->flags);
5653         upd.set = upd.mask;
5654         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
5655
5656         if (params->flags & WPA_STA_WMM) {
5657                 wme = nlmsg_alloc();
5658                 if (!wme)
5659                         goto nla_put_failure;
5660
5661                 NLA_PUT_U8(wme, NL80211_STA_WME_UAPSD_QUEUES,
5662                                 params->qosinfo & WMM_QOSINFO_STA_AC_MASK);
5663                 NLA_PUT_U8(wme, NL80211_STA_WME_MAX_SP,
5664                                 (params->qosinfo > WMM_QOSINFO_STA_SP_SHIFT) &
5665                                 WMM_QOSINFO_STA_SP_MASK);
5666                 if (nla_put_nested(msg, NL80211_ATTR_STA_WME, wme) < 0)
5667                         goto nla_put_failure;
5668         }
5669
5670         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5671         msg = NULL;
5672         if (ret)
5673                 wpa_printf(MSG_DEBUG, "nl80211: NL80211_CMD_%s_STATION "
5674                            "result: %d (%s)", params->set ? "SET" : "NEW", ret,
5675                            strerror(-ret));
5676         if (ret == -EEXIST)
5677                 ret = 0;
5678  nla_put_failure:
5679         nlmsg_free(wme);
5680         nlmsg_free(msg);
5681         return ret;
5682 }
5683
5684
5685 static int wpa_driver_nl80211_sta_remove(void *priv, const u8 *addr)
5686 {
5687         struct i802_bss *bss = priv;
5688         struct wpa_driver_nl80211_data *drv = bss->drv;
5689         struct nl_msg *msg;
5690         int ret;
5691
5692         msg = nlmsg_alloc();
5693         if (!msg)
5694                 return -ENOMEM;
5695
5696         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_STATION);
5697
5698         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
5699                     if_nametoindex(bss->ifname));
5700         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
5701
5702         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5703         if (ret == -ENOENT)
5704                 return 0;
5705         return ret;
5706  nla_put_failure:
5707         nlmsg_free(msg);
5708         return -ENOBUFS;
5709 }
5710
5711
5712 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
5713                                  int ifidx)
5714 {
5715         struct nl_msg *msg;
5716
5717         wpa_printf(MSG_DEBUG, "nl80211: Remove interface ifindex=%d", ifidx);
5718
5719         /* stop listening for EAPOL on this interface */
5720         del_ifidx(drv, ifidx);
5721
5722         msg = nlmsg_alloc();
5723         if (!msg)
5724                 goto nla_put_failure;
5725
5726         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_INTERFACE);
5727         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
5728
5729         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
5730                 return;
5731         msg = NULL;
5732  nla_put_failure:
5733         nlmsg_free(msg);
5734         wpa_printf(MSG_ERROR, "Failed to remove interface (ifidx=%d)", ifidx);
5735 }
5736
5737
5738 static const char * nl80211_iftype_str(enum nl80211_iftype mode)
5739 {
5740         switch (mode) {
5741         case NL80211_IFTYPE_ADHOC:
5742                 return "ADHOC";
5743         case NL80211_IFTYPE_STATION:
5744                 return "STATION";
5745         case NL80211_IFTYPE_AP:
5746                 return "AP";
5747         case NL80211_IFTYPE_MONITOR:
5748                 return "MONITOR";
5749         case NL80211_IFTYPE_P2P_CLIENT:
5750                 return "P2P_CLIENT";
5751         case NL80211_IFTYPE_P2P_GO:
5752                 return "P2P_GO";
5753         default:
5754                 return "unknown";
5755         }
5756 }
5757
5758
5759 static int nl80211_create_iface_once(struct wpa_driver_nl80211_data *drv,
5760                                      const char *ifname,
5761                                      enum nl80211_iftype iftype,
5762                                      const u8 *addr, int wds)
5763 {
5764         struct nl_msg *msg, *flags = NULL;
5765         int ifidx;
5766         int ret = -ENOBUFS;
5767
5768         wpa_printf(MSG_DEBUG, "nl80211: Create interface iftype %d (%s)",
5769                    iftype, nl80211_iftype_str(iftype));
5770
5771         msg = nlmsg_alloc();
5772         if (!msg)
5773                 return -1;
5774
5775         nl80211_cmd(drv, msg, 0, NL80211_CMD_NEW_INTERFACE);
5776         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5777         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
5778         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
5779
5780         if (iftype == NL80211_IFTYPE_MONITOR) {
5781                 int err;
5782
5783                 flags = nlmsg_alloc();
5784                 if (!flags)
5785                         goto nla_put_failure;
5786
5787                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
5788
5789                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
5790
5791                 nlmsg_free(flags);
5792
5793                 if (err)
5794                         goto nla_put_failure;
5795         } else if (wds) {
5796                 NLA_PUT_U8(msg, NL80211_ATTR_4ADDR, wds);
5797         }
5798
5799         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5800         msg = NULL;
5801         if (ret) {
5802  nla_put_failure:
5803                 nlmsg_free(msg);
5804                 wpa_printf(MSG_ERROR, "Failed to create interface %s: %d (%s)",
5805                            ifname, ret, strerror(-ret));
5806                 return ret;
5807         }
5808
5809         ifidx = if_nametoindex(ifname);
5810         wpa_printf(MSG_DEBUG, "nl80211: New interface %s created: ifindex=%d",
5811                    ifname, ifidx);
5812
5813         if (ifidx <= 0)
5814                 return -1;
5815
5816         /* start listening for EAPOL on this interface */
5817         add_ifidx(drv, ifidx);
5818
5819         if (addr && iftype != NL80211_IFTYPE_MONITOR &&
5820             linux_set_ifhwaddr(drv->global->ioctl_sock, ifname, addr)) {
5821                 nl80211_remove_iface(drv, ifidx);
5822                 return -1;
5823         }
5824
5825         return ifidx;
5826 }
5827
5828
5829 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
5830                                 const char *ifname, enum nl80211_iftype iftype,
5831                                 const u8 *addr, int wds)
5832 {
5833         int ret;
5834
5835         ret = nl80211_create_iface_once(drv, ifname, iftype, addr, wds);
5836
5837         /* if error occurred and interface exists already */
5838         if (ret == -ENFILE && if_nametoindex(ifname)) {
5839                 wpa_printf(MSG_INFO, "Try to remove and re-create %s", ifname);
5840
5841                 /* Try to remove the interface that was already there. */
5842                 nl80211_remove_iface(drv, if_nametoindex(ifname));
5843
5844                 /* Try to create the interface again */
5845                 ret = nl80211_create_iface_once(drv, ifname, iftype, addr,
5846                                                 wds);
5847         }
5848
5849         if (ret >= 0 && is_p2p_interface(iftype))
5850                 nl80211_disable_11b_rates(drv, ret, 1);
5851
5852         return ret;
5853 }
5854
5855
5856 static void handle_tx_callback(void *ctx, u8 *buf, size_t len, int ok)
5857 {
5858         struct ieee80211_hdr *hdr;
5859         u16 fc;
5860         union wpa_event_data event;
5861
5862         hdr = (struct ieee80211_hdr *) buf;
5863         fc = le_to_host16(hdr->frame_control);
5864
5865         os_memset(&event, 0, sizeof(event));
5866         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
5867         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
5868         event.tx_status.dst = hdr->addr1;
5869         event.tx_status.data = buf;
5870         event.tx_status.data_len = len;
5871         event.tx_status.ack = ok;
5872         wpa_supplicant_event(ctx, EVENT_TX_STATUS, &event);
5873 }
5874
5875
5876 static void from_unknown_sta(struct wpa_driver_nl80211_data *drv,
5877                              u8 *buf, size_t len)
5878 {
5879         struct ieee80211_hdr *hdr = (void *)buf;
5880         u16 fc;
5881         union wpa_event_data event;
5882
5883         if (len < sizeof(*hdr))
5884                 return;
5885
5886         fc = le_to_host16(hdr->frame_control);
5887
5888         os_memset(&event, 0, sizeof(event));
5889         event.rx_from_unknown.bssid = get_hdr_bssid(hdr, len);
5890         event.rx_from_unknown.addr = hdr->addr2;
5891         event.rx_from_unknown.wds = (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) ==
5892                 (WLAN_FC_FROMDS | WLAN_FC_TODS);
5893         wpa_supplicant_event(drv->ctx, EVENT_RX_FROM_UNKNOWN, &event);
5894 }
5895
5896
5897 static void handle_frame(struct wpa_driver_nl80211_data *drv,
5898                          u8 *buf, size_t len, int datarate, int ssi_signal)
5899 {
5900         struct ieee80211_hdr *hdr;
5901         u16 fc;
5902         union wpa_event_data event;
5903
5904         hdr = (struct ieee80211_hdr *) buf;
5905         fc = le_to_host16(hdr->frame_control);
5906
5907         switch (WLAN_FC_GET_TYPE(fc)) {
5908         case WLAN_FC_TYPE_MGMT:
5909                 os_memset(&event, 0, sizeof(event));
5910                 event.rx_mgmt.frame = buf;
5911                 event.rx_mgmt.frame_len = len;
5912                 event.rx_mgmt.datarate = datarate;
5913                 event.rx_mgmt.ssi_signal = ssi_signal;
5914                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
5915                 break;
5916         case WLAN_FC_TYPE_CTRL:
5917                 /* can only get here with PS-Poll frames */
5918                 wpa_printf(MSG_DEBUG, "CTRL");
5919                 from_unknown_sta(drv, buf, len);
5920                 break;
5921         case WLAN_FC_TYPE_DATA:
5922                 from_unknown_sta(drv, buf, len);
5923                 break;
5924         }
5925 }
5926
5927
5928 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
5929 {
5930         struct wpa_driver_nl80211_data *drv = eloop_ctx;
5931         int len;
5932         unsigned char buf[3000];
5933         struct ieee80211_radiotap_iterator iter;
5934         int ret;
5935         int datarate = 0, ssi_signal = 0;
5936         int injected = 0, failed = 0, rxflags = 0;
5937
5938         len = recv(sock, buf, sizeof(buf), 0);
5939         if (len < 0) {
5940                 perror("recv");
5941                 return;
5942         }
5943
5944         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
5945                 printf("received invalid radiotap frame\n");
5946                 return;
5947         }
5948
5949         while (1) {
5950                 ret = ieee80211_radiotap_iterator_next(&iter);
5951                 if (ret == -ENOENT)
5952                         break;
5953                 if (ret) {
5954                         printf("received invalid radiotap frame (%d)\n", ret);
5955                         return;
5956                 }
5957                 switch (iter.this_arg_index) {
5958                 case IEEE80211_RADIOTAP_FLAGS:
5959                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
5960                                 len -= 4;
5961                         break;
5962                 case IEEE80211_RADIOTAP_RX_FLAGS:
5963                         rxflags = 1;
5964                         break;
5965                 case IEEE80211_RADIOTAP_TX_FLAGS:
5966                         injected = 1;
5967                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
5968                                         IEEE80211_RADIOTAP_F_TX_FAIL;
5969                         break;
5970                 case IEEE80211_RADIOTAP_DATA_RETRIES:
5971                         break;
5972                 case IEEE80211_RADIOTAP_CHANNEL:
5973                         /* TODO: convert from freq/flags to channel number */
5974                         break;
5975                 case IEEE80211_RADIOTAP_RATE:
5976                         datarate = *iter.this_arg * 5;
5977                         break;
5978                 case IEEE80211_RADIOTAP_DBM_ANTSIGNAL:
5979                         ssi_signal = (s8) *iter.this_arg;
5980                         break;
5981                 }
5982         }
5983
5984         if (rxflags && injected)
5985                 return;
5986
5987         if (!injected)
5988                 handle_frame(drv, buf + iter.max_length,
5989                              len - iter.max_length, datarate, ssi_signal);
5990         else
5991                 handle_tx_callback(drv->ctx, buf + iter.max_length,
5992                                    len - iter.max_length, !failed);
5993 }
5994
5995
5996 /*
5997  * we post-process the filter code later and rewrite
5998  * this to the offset to the last instruction
5999  */
6000 #define PASS    0xFF
6001 #define FAIL    0xFE
6002
6003 static struct sock_filter msock_filter_insns[] = {
6004         /*
6005          * do a little-endian load of the radiotap length field
6006          */
6007         /* load lower byte into A */
6008         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
6009         /* put it into X (== index register) */
6010         BPF_STMT(BPF_MISC| BPF_TAX, 0),
6011         /* load upper byte into A */
6012         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
6013         /* left-shift it by 8 */
6014         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
6015         /* or with X */
6016         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
6017         /* put result into X */
6018         BPF_STMT(BPF_MISC| BPF_TAX, 0),
6019
6020         /*
6021          * Allow management frames through, this also gives us those
6022          * management frames that we sent ourselves with status
6023          */
6024         /* load the lower byte of the IEEE 802.11 frame control field */
6025         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
6026         /* mask off frame type and version */
6027         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
6028         /* accept frame if it's both 0, fall through otherwise */
6029         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
6030
6031         /*
6032          * TODO: add a bit to radiotap RX flags that indicates
6033          * that the sending station is not associated, then
6034          * add a filter here that filters on our DA and that flag
6035          * to allow us to deauth frames to that bad station.
6036          *
6037          * For now allow all To DS data frames through.
6038          */
6039         /* load the IEEE 802.11 frame control field */
6040         BPF_STMT(BPF_LD  | BPF_H | BPF_IND, 0),
6041         /* mask off frame type, version and DS status */
6042         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0x0F03),
6043         /* accept frame if version 0, type 2 and To DS, fall through otherwise
6044          */
6045         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0801, PASS, 0),
6046
6047 #if 0
6048         /*
6049          * drop non-data frames
6050          */
6051         /* load the lower byte of the frame control field */
6052         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
6053         /* mask off QoS bit */
6054         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
6055         /* drop non-data frames */
6056         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
6057 #endif
6058         /* load the upper byte of the frame control field */
6059         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 1),
6060         /* mask off toDS/fromDS */
6061         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
6062         /* accept WDS frames */
6063         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, PASS, 0),
6064
6065         /*
6066          * add header length to index
6067          */
6068         /* load the lower byte of the frame control field */
6069         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
6070         /* mask off QoS bit */
6071         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
6072         /* right shift it by 6 to give 0 or 2 */
6073         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
6074         /* add data frame header length */
6075         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
6076         /* add index, was start of 802.11 header */
6077         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
6078         /* move to index, now start of LL header */
6079         BPF_STMT(BPF_MISC | BPF_TAX, 0),
6080
6081         /*
6082          * Accept empty data frames, we use those for
6083          * polling activity.
6084          */
6085         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
6086         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
6087
6088         /*
6089          * Accept EAPOL frames
6090          */
6091         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
6092         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
6093         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
6094         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
6095
6096         /* keep these last two statements or change the code below */
6097         /* return 0 == "DROP" */
6098         BPF_STMT(BPF_RET | BPF_K, 0),
6099         /* return ~0 == "keep all" */
6100         BPF_STMT(BPF_RET | BPF_K, ~0),
6101 };
6102
6103 static struct sock_fprog msock_filter = {
6104         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
6105         .filter = msock_filter_insns,
6106 };
6107
6108
6109 static int add_monitor_filter(int s)
6110 {
6111         int idx;
6112
6113         /* rewrite all PASS/FAIL jump offsets */
6114         for (idx = 0; idx < msock_filter.len; idx++) {
6115                 struct sock_filter *insn = &msock_filter_insns[idx];
6116
6117                 if (BPF_CLASS(insn->code) == BPF_JMP) {
6118                         if (insn->code == (BPF_JMP|BPF_JA)) {
6119                                 if (insn->k == PASS)
6120                                         insn->k = msock_filter.len - idx - 2;
6121                                 else if (insn->k == FAIL)
6122                                         insn->k = msock_filter.len - idx - 3;
6123                         }
6124
6125                         if (insn->jt == PASS)
6126                                 insn->jt = msock_filter.len - idx - 2;
6127                         else if (insn->jt == FAIL)
6128                                 insn->jt = msock_filter.len - idx - 3;
6129
6130                         if (insn->jf == PASS)
6131                                 insn->jf = msock_filter.len - idx - 2;
6132                         else if (insn->jf == FAIL)
6133                                 insn->jf = msock_filter.len - idx - 3;
6134                 }
6135         }
6136
6137         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
6138                        &msock_filter, sizeof(msock_filter))) {
6139                 perror("SO_ATTACH_FILTER");
6140                 return -1;
6141         }
6142
6143         return 0;
6144 }
6145
6146
6147 static void nl80211_remove_monitor_interface(
6148         struct wpa_driver_nl80211_data *drv)
6149 {
6150         drv->monitor_refcount--;
6151         if (drv->monitor_refcount > 0)
6152                 return;
6153
6154         if (drv->monitor_ifidx >= 0) {
6155                 nl80211_remove_iface(drv, drv->monitor_ifidx);
6156                 drv->monitor_ifidx = -1;
6157         }
6158         if (drv->monitor_sock >= 0) {
6159                 eloop_unregister_read_sock(drv->monitor_sock);
6160                 close(drv->monitor_sock);
6161                 drv->monitor_sock = -1;
6162         }
6163 }
6164
6165
6166 static int
6167 nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
6168 {
6169         char buf[IFNAMSIZ];
6170         struct sockaddr_ll ll;
6171         int optval;
6172         socklen_t optlen;
6173
6174         if (drv->monitor_ifidx >= 0) {
6175                 drv->monitor_refcount++;
6176                 return 0;
6177         }
6178
6179         if (os_strncmp(drv->first_bss.ifname, "p2p-", 4) == 0) {
6180                 /*
6181                  * P2P interface name is of the format p2p-%s-%d. For monitor
6182                  * interface name corresponding to P2P GO, replace "p2p-" with
6183                  * "mon-" to retain the same interface name length and to
6184                  * indicate that it is a monitor interface.
6185                  */
6186                 snprintf(buf, IFNAMSIZ, "mon-%s", drv->first_bss.ifname + 4);
6187         } else {
6188                 /* Non-P2P interface with AP functionality. */
6189                 snprintf(buf, IFNAMSIZ, "mon.%s", drv->first_bss.ifname);
6190         }
6191
6192         buf[IFNAMSIZ - 1] = '\0';
6193
6194         drv->monitor_ifidx =
6195                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL,
6196                                      0);
6197
6198         if (drv->monitor_ifidx == -EOPNOTSUPP) {
6199                 /*
6200                  * This is backward compatibility for a few versions of
6201                  * the kernel only that didn't advertise the right
6202                  * attributes for the only driver that then supported
6203                  * AP mode w/o monitor -- ath6kl.
6204                  */
6205                 wpa_printf(MSG_DEBUG, "nl80211: Driver does not support "
6206                            "monitor interface type - try to run without it");
6207                 drv->device_ap_sme = 1;
6208         }
6209
6210         if (drv->monitor_ifidx < 0)
6211                 return -1;
6212
6213         if (linux_set_iface_flags(drv->global->ioctl_sock, buf, 1))
6214                 goto error;
6215
6216         memset(&ll, 0, sizeof(ll));
6217         ll.sll_family = AF_PACKET;
6218         ll.sll_ifindex = drv->monitor_ifidx;
6219         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
6220         if (drv->monitor_sock < 0) {
6221                 perror("socket[PF_PACKET,SOCK_RAW]");
6222                 goto error;
6223         }
6224
6225         if (add_monitor_filter(drv->monitor_sock)) {
6226                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
6227                            "interface; do filtering in user space");
6228                 /* This works, but will cost in performance. */
6229         }
6230
6231         if (bind(drv->monitor_sock, (struct sockaddr *) &ll, sizeof(ll)) < 0) {
6232                 perror("monitor socket bind");
6233                 goto error;
6234         }
6235
6236         optlen = sizeof(optval);
6237         optval = 20;
6238         if (setsockopt
6239             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
6240                 perror("Failed to set socket priority");
6241                 goto error;
6242         }
6243
6244         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
6245                                      drv, NULL)) {
6246                 printf("Could not register monitor read socket\n");
6247                 goto error;
6248         }
6249
6250         return 0;
6251  error:
6252         nl80211_remove_monitor_interface(drv);
6253         return -1;
6254 }
6255
6256
6257 static int nl80211_setup_ap(struct i802_bss *bss)
6258 {
6259         struct wpa_driver_nl80211_data *drv = bss->drv;
6260
6261         wpa_printf(MSG_DEBUG, "nl80211: Setup AP - device_ap_sme=%d "
6262                    "use_monitor=%d", drv->device_ap_sme, drv->use_monitor);
6263
6264         /*
6265          * Disable Probe Request reporting unless we need it in this way for
6266          * devices that include the AP SME, in the other case (unless using
6267          * monitor iface) we'll get it through the nl_mgmt socket instead.
6268          */
6269         if (!drv->device_ap_sme)
6270                 wpa_driver_nl80211_probe_req_report(bss, 0);
6271
6272         if (!drv->device_ap_sme && !drv->use_monitor)
6273                 if (nl80211_mgmt_subscribe_ap(bss))
6274                         return -1;
6275
6276         if (drv->device_ap_sme && !drv->use_monitor)
6277                 if (nl80211_mgmt_subscribe_ap_dev_sme(bss))
6278                         return -1;
6279
6280         if (!drv->device_ap_sme && drv->use_monitor &&
6281             nl80211_create_monitor_interface(drv) &&
6282             !drv->device_ap_sme)
6283                 return -1;
6284
6285         if (drv->device_ap_sme &&
6286             wpa_driver_nl80211_probe_req_report(bss, 1) < 0) {
6287                 wpa_printf(MSG_DEBUG, "nl80211: Failed to enable "
6288                            "Probe Request frame reporting in AP mode");
6289                 /* Try to survive without this */
6290         }
6291
6292         return 0;
6293 }
6294
6295
6296 static void nl80211_teardown_ap(struct i802_bss *bss)
6297 {
6298         struct wpa_driver_nl80211_data *drv = bss->drv;
6299
6300         if (drv->device_ap_sme) {
6301                 wpa_driver_nl80211_probe_req_report(bss, 0);
6302                 if (!drv->use_monitor)
6303                         nl80211_mgmt_unsubscribe(bss, "AP teardown (dev SME)");
6304         } else if (drv->use_monitor)
6305                 nl80211_remove_monitor_interface(drv);
6306         else
6307                 nl80211_mgmt_unsubscribe(bss, "AP teardown");
6308
6309         bss->beacon_set = 0;
6310 }
6311
6312
6313 static int nl80211_send_eapol_data(struct i802_bss *bss,
6314                                    const u8 *addr, const u8 *data,
6315                                    size_t data_len)
6316 {
6317         struct sockaddr_ll ll;
6318         int ret;
6319
6320         if (bss->drv->eapol_tx_sock < 0) {
6321                 wpa_printf(MSG_DEBUG, "nl80211: No socket to send EAPOL");
6322                 return -1;
6323         }
6324
6325         os_memset(&ll, 0, sizeof(ll));
6326         ll.sll_family = AF_PACKET;
6327         ll.sll_ifindex = bss->ifindex;
6328         ll.sll_protocol = htons(ETH_P_PAE);
6329         ll.sll_halen = ETH_ALEN;
6330         os_memcpy(ll.sll_addr, addr, ETH_ALEN);
6331         ret = sendto(bss->drv->eapol_tx_sock, data, data_len, 0,
6332                      (struct sockaddr *) &ll, sizeof(ll));
6333         if (ret < 0)
6334                 wpa_printf(MSG_ERROR, "nl80211: EAPOL TX: %s",
6335                            strerror(errno));
6336
6337         return ret;
6338 }
6339
6340
6341 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
6342
6343 static int wpa_driver_nl80211_hapd_send_eapol(
6344         void *priv, const u8 *addr, const u8 *data,
6345         size_t data_len, int encrypt, const u8 *own_addr, u32 flags)
6346 {
6347         struct i802_bss *bss = priv;
6348         struct wpa_driver_nl80211_data *drv = bss->drv;
6349         struct ieee80211_hdr *hdr;
6350         size_t len;
6351         u8 *pos;
6352         int res;
6353         int qos = flags & WPA_STA_WMM;
6354
6355         if (drv->device_ap_sme || !drv->use_monitor)
6356                 return nl80211_send_eapol_data(bss, addr, data, data_len);
6357
6358         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
6359                 data_len;
6360         hdr = os_zalloc(len);
6361         if (hdr == NULL) {
6362                 printf("malloc() failed for i802_send_data(len=%lu)\n",
6363                        (unsigned long) len);
6364                 return -1;
6365         }
6366
6367         hdr->frame_control =
6368                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
6369         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
6370         if (encrypt)
6371                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
6372         if (qos) {
6373                 hdr->frame_control |=
6374                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
6375         }
6376
6377         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
6378         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
6379         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
6380         pos = (u8 *) (hdr + 1);
6381
6382         if (qos) {
6383                 /* add an empty QoS header if needed */
6384                 pos[0] = 0;
6385                 pos[1] = 0;
6386                 pos += 2;
6387         }
6388
6389         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
6390         pos += sizeof(rfc1042_header);
6391         WPA_PUT_BE16(pos, ETH_P_PAE);
6392         pos += 2;
6393         memcpy(pos, data, data_len);
6394
6395         res = wpa_driver_nl80211_send_frame(bss, (u8 *) hdr, len, encrypt, 0,
6396                                             0, 0, 0, 0);
6397         if (res < 0) {
6398                 wpa_printf(MSG_ERROR, "i802_send_eapol - packet len: %lu - "
6399                            "failed: %d (%s)",
6400                            (unsigned long) len, errno, strerror(errno));
6401         }
6402         os_free(hdr);
6403
6404         return res;
6405 }
6406
6407
6408 static int wpa_driver_nl80211_sta_set_flags(void *priv, const u8 *addr,
6409                                             int total_flags,
6410                                             int flags_or, int flags_and)
6411 {
6412         struct i802_bss *bss = priv;
6413         struct wpa_driver_nl80211_data *drv = bss->drv;
6414         struct nl_msg *msg, *flags = NULL;
6415         struct nl80211_sta_flag_update upd;
6416
6417         msg = nlmsg_alloc();
6418         if (!msg)
6419                 return -ENOMEM;
6420
6421         flags = nlmsg_alloc();
6422         if (!flags) {
6423                 nlmsg_free(msg);
6424                 return -ENOMEM;
6425         }
6426
6427         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
6428
6429         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
6430                     if_nametoindex(bss->ifname));
6431         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
6432
6433         /*
6434          * Backwards compatibility version using NL80211_ATTR_STA_FLAGS. This
6435          * can be removed eventually.
6436          */
6437         if (total_flags & WPA_STA_AUTHORIZED)
6438                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
6439
6440         if (total_flags & WPA_STA_WMM)
6441                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
6442
6443         if (total_flags & WPA_STA_SHORT_PREAMBLE)
6444                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
6445
6446         if (total_flags & WPA_STA_MFP)
6447                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
6448
6449         if (total_flags & WPA_STA_TDLS_PEER)
6450                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_TDLS_PEER);
6451
6452         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
6453                 goto nla_put_failure;
6454
6455         os_memset(&upd, 0, sizeof(upd));
6456         upd.mask = sta_flags_nl80211(flags_or | ~flags_and);
6457         upd.set = sta_flags_nl80211(flags_or);
6458         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
6459
6460         nlmsg_free(flags);
6461
6462         return send_and_recv_msgs(drv, msg, NULL, NULL);
6463  nla_put_failure:
6464         nlmsg_free(msg);
6465         nlmsg_free(flags);
6466         return -ENOBUFS;
6467 }
6468
6469
6470 static int wpa_driver_nl80211_ap(struct wpa_driver_nl80211_data *drv,
6471                                  struct wpa_driver_associate_params *params)
6472 {
6473         enum nl80211_iftype nlmode;
6474
6475         if (params->p2p) {
6476                 wpa_printf(MSG_DEBUG, "nl80211: Setup AP operations for P2P "
6477                            "group (GO)");
6478                 nlmode = NL80211_IFTYPE_P2P_GO;
6479         } else
6480                 nlmode = NL80211_IFTYPE_AP;
6481
6482         if (wpa_driver_nl80211_set_mode(&drv->first_bss, nlmode) ||
6483             wpa_driver_nl80211_set_freq(&drv->first_bss, params->freq, 0, 0)) {
6484                 nl80211_remove_monitor_interface(drv);
6485                 return -1;
6486         }
6487
6488         return 0;
6489 }
6490
6491
6492 static int nl80211_leave_ibss(struct wpa_driver_nl80211_data *drv)
6493 {
6494         struct nl_msg *msg;
6495         int ret = -1;
6496
6497         msg = nlmsg_alloc();
6498         if (!msg)
6499                 return -1;
6500
6501         nl80211_cmd(drv, msg, 0, NL80211_CMD_LEAVE_IBSS);
6502         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6503         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6504         msg = NULL;
6505         if (ret) {
6506                 wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS failed: ret=%d "
6507                            "(%s)", ret, strerror(-ret));
6508                 goto nla_put_failure;
6509         }
6510
6511         ret = 0;
6512         wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS request sent successfully");
6513
6514 nla_put_failure:
6515         nlmsg_free(msg);
6516         return ret;
6517 }
6518
6519
6520 static int wpa_driver_nl80211_ibss(struct wpa_driver_nl80211_data *drv,
6521                                    struct wpa_driver_associate_params *params)
6522 {
6523         struct nl_msg *msg;
6524         int ret = -1;
6525         int count = 0;
6526
6527         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS (ifindex=%d)", drv->ifindex);
6528
6529         if (wpa_driver_nl80211_set_mode(&drv->first_bss,
6530                                         NL80211_IFTYPE_ADHOC)) {
6531                 wpa_printf(MSG_INFO, "nl80211: Failed to set interface into "
6532                            "IBSS mode");
6533                 return -1;
6534         }
6535
6536 retry:
6537         msg = nlmsg_alloc();
6538         if (!msg)
6539                 return -1;
6540
6541         nl80211_cmd(drv, msg, 0, NL80211_CMD_JOIN_IBSS);
6542         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6543
6544         if (params->ssid == NULL || params->ssid_len > sizeof(drv->ssid))
6545                 goto nla_put_failure;
6546
6547         wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6548                           params->ssid, params->ssid_len);
6549         NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6550                 params->ssid);
6551         os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6552         drv->ssid_len = params->ssid_len;
6553
6554         wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6555         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6556
6557         ret = nl80211_set_conn_keys(params, msg);
6558         if (ret)
6559                 goto nla_put_failure;
6560
6561         if (params->bssid && params->fixed_bssid) {
6562                 wpa_printf(MSG_DEBUG, "  * BSSID=" MACSTR,
6563                            MAC2STR(params->bssid));
6564                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6565         }
6566
6567         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
6568             params->key_mgmt_suite == KEY_MGMT_PSK ||
6569             params->key_mgmt_suite == KEY_MGMT_802_1X_SHA256 ||
6570             params->key_mgmt_suite == KEY_MGMT_PSK_SHA256) {
6571                 wpa_printf(MSG_DEBUG, "  * control port");
6572                 NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT);
6573         }
6574
6575         if (params->wpa_ie) {
6576                 wpa_hexdump(MSG_DEBUG,
6577                             "  * Extra IEs for Beacon/Probe Response frames",
6578                             params->wpa_ie, params->wpa_ie_len);
6579                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6580                         params->wpa_ie);
6581         }
6582
6583         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6584         msg = NULL;
6585         if (ret) {
6586                 wpa_printf(MSG_DEBUG, "nl80211: Join IBSS failed: ret=%d (%s)",
6587                            ret, strerror(-ret));
6588                 count++;
6589                 if (ret == -EALREADY && count == 1) {
6590                         wpa_printf(MSG_DEBUG, "nl80211: Retry IBSS join after "
6591                                    "forced leave");
6592                         nl80211_leave_ibss(drv);
6593                         nlmsg_free(msg);
6594                         goto retry;
6595                 }
6596
6597                 goto nla_put_failure;
6598         }
6599         ret = 0;
6600         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS request sent successfully");
6601
6602 nla_put_failure:
6603         nlmsg_free(msg);
6604         return ret;
6605 }
6606
6607
6608 static unsigned int nl80211_get_assoc_bssid(struct wpa_driver_nl80211_data *drv,
6609                                             u8 *bssid)
6610 {
6611         struct nl_msg *msg;
6612         int ret;
6613         struct nl80211_bss_info_arg arg;
6614
6615         os_memset(&arg, 0, sizeof(arg));
6616         msg = nlmsg_alloc();
6617         if (!msg)
6618                 goto nla_put_failure;
6619
6620         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
6621         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6622
6623         arg.drv = drv;
6624         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
6625         msg = NULL;
6626         if (ret == 0) {
6627                 if (is_zero_ether_addr(arg.assoc_bssid))
6628                         return -ENOTCONN;
6629                 os_memcpy(bssid, arg.assoc_bssid, ETH_ALEN);
6630                 return 0;
6631         }
6632         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
6633                    "(%s)", ret, strerror(-ret));
6634 nla_put_failure:
6635         nlmsg_free(msg);
6636         return drv->assoc_freq;
6637 }
6638
6639
6640 static int nl80211_disconnect(struct wpa_driver_nl80211_data *drv,
6641                               const u8 *bssid)
6642 {
6643         u8 addr[ETH_ALEN];
6644
6645         if (bssid == NULL) {
6646                 int res = nl80211_get_assoc_bssid(drv, addr);
6647                 if (res)
6648                         return res;
6649                 bssid = addr;
6650         }
6651
6652         return wpa_driver_nl80211_disconnect(drv, bssid,
6653                                              WLAN_REASON_PREV_AUTH_NOT_VALID);
6654 }
6655
6656
6657 static int wpa_driver_nl80211_connect(
6658         struct wpa_driver_nl80211_data *drv,
6659         struct wpa_driver_associate_params *params)
6660 {
6661         struct nl_msg *msg;
6662         enum nl80211_auth_type type;
6663         int ret = 0;
6664         int algs;
6665
6666         msg = nlmsg_alloc();
6667         if (!msg)
6668                 return -1;
6669
6670         wpa_printf(MSG_DEBUG, "nl80211: Connect (ifindex=%d)", drv->ifindex);
6671         nl80211_cmd(drv, msg, 0, NL80211_CMD_CONNECT);
6672
6673         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6674         if (params->bssid) {
6675                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
6676                            MAC2STR(params->bssid));
6677                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6678         }
6679         if (params->freq) {
6680                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6681                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6682         }
6683         if (params->bg_scan_period >= 0) {
6684                 wpa_printf(MSG_DEBUG, "  * bg scan period=%d",
6685                            params->bg_scan_period);
6686                 NLA_PUT_U16(msg, NL80211_ATTR_BG_SCAN_PERIOD,
6687                             params->bg_scan_period);
6688         }
6689         if (params->ssid) {
6690                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6691                                   params->ssid, params->ssid_len);
6692                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6693                         params->ssid);
6694                 if (params->ssid_len > sizeof(drv->ssid))
6695                         goto nla_put_failure;
6696                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6697                 drv->ssid_len = params->ssid_len;
6698         }
6699         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
6700         if (params->wpa_ie)
6701                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6702                         params->wpa_ie);
6703
6704         algs = 0;
6705         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
6706                 algs++;
6707         if (params->auth_alg & WPA_AUTH_ALG_SHARED)
6708                 algs++;
6709         if (params->auth_alg & WPA_AUTH_ALG_LEAP)
6710                 algs++;
6711         if (algs > 1) {
6712                 wpa_printf(MSG_DEBUG, "  * Leave out Auth Type for automatic "
6713                            "selection");
6714                 goto skip_auth_type;
6715         }
6716
6717         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
6718                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
6719         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
6720                 type = NL80211_AUTHTYPE_SHARED_KEY;
6721         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
6722                 type = NL80211_AUTHTYPE_NETWORK_EAP;
6723         else if (params->auth_alg & WPA_AUTH_ALG_FT)
6724                 type = NL80211_AUTHTYPE_FT;
6725         else
6726                 goto nla_put_failure;
6727
6728         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
6729         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
6730
6731 skip_auth_type:
6732         if (params->wpa_proto) {
6733                 enum nl80211_wpa_versions ver = 0;
6734
6735                 if (params->wpa_proto & WPA_PROTO_WPA)
6736                         ver |= NL80211_WPA_VERSION_1;
6737                 if (params->wpa_proto & WPA_PROTO_RSN)
6738                         ver |= NL80211_WPA_VERSION_2;
6739
6740                 wpa_printf(MSG_DEBUG, "  * WPA Versions 0x%x", ver);
6741                 NLA_PUT_U32(msg, NL80211_ATTR_WPA_VERSIONS, ver);
6742         }
6743
6744         if (params->pairwise_suite != CIPHER_NONE) {
6745                 int cipher;
6746
6747                 switch (params->pairwise_suite) {
6748                 case CIPHER_WEP40:
6749                         cipher = WLAN_CIPHER_SUITE_WEP40;
6750                         break;
6751                 case CIPHER_WEP104:
6752                         cipher = WLAN_CIPHER_SUITE_WEP104;
6753                         break;
6754                 case CIPHER_CCMP:
6755                         cipher = WLAN_CIPHER_SUITE_CCMP;
6756                         break;
6757                 case CIPHER_TKIP:
6758                 default:
6759                         cipher = WLAN_CIPHER_SUITE_TKIP;
6760                         break;
6761                 }
6762                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE, cipher);
6763         }
6764
6765         if (params->group_suite != CIPHER_NONE) {
6766                 int cipher;
6767
6768                 switch (params->group_suite) {
6769                 case CIPHER_WEP40:
6770                         cipher = WLAN_CIPHER_SUITE_WEP40;
6771                         break;
6772                 case CIPHER_WEP104:
6773                         cipher = WLAN_CIPHER_SUITE_WEP104;
6774                         break;
6775                 case CIPHER_CCMP:
6776                         cipher = WLAN_CIPHER_SUITE_CCMP;
6777                         break;
6778                 case CIPHER_TKIP:
6779                 default:
6780                         cipher = WLAN_CIPHER_SUITE_TKIP;
6781                         break;
6782                 }
6783                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP, cipher);
6784         }
6785
6786         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
6787             params->key_mgmt_suite == KEY_MGMT_PSK) {
6788                 int mgmt = WLAN_AKM_SUITE_PSK;
6789
6790                 switch (params->key_mgmt_suite) {
6791                 case KEY_MGMT_802_1X:
6792                         mgmt = WLAN_AKM_SUITE_8021X;
6793                         break;
6794                 case KEY_MGMT_PSK:
6795                 default:
6796                         mgmt = WLAN_AKM_SUITE_PSK;
6797                         break;
6798                 }
6799                 NLA_PUT_U32(msg, NL80211_ATTR_AKM_SUITES, mgmt);
6800         }
6801
6802         if (params->disable_ht)
6803                 NLA_PUT_FLAG(msg, NL80211_ATTR_DISABLE_HT);
6804
6805         if (params->htcaps && params->htcaps_mask) {
6806                 int sz = sizeof(struct ieee80211_ht_capabilities);
6807                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY, sz, params->htcaps);
6808                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY_MASK, sz,
6809                         params->htcaps_mask);
6810         }
6811
6812         ret = nl80211_set_conn_keys(params, msg);
6813         if (ret)
6814                 goto nla_put_failure;
6815
6816         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6817         msg = NULL;
6818         if (ret) {
6819                 wpa_printf(MSG_DEBUG, "nl80211: MLME connect failed: ret=%d "
6820                            "(%s)", ret, strerror(-ret));
6821                 /*
6822                  * cfg80211 does not currently accept new connection if we are
6823                  * already connected. As a workaround, force disconnection and
6824                  * try again once the driver indicates it completed
6825                  * disconnection.
6826                  */
6827                 if (ret == -EALREADY)
6828                         nl80211_disconnect(drv, params->bssid);
6829                 goto nla_put_failure;
6830         }
6831         ret = 0;
6832         wpa_printf(MSG_DEBUG, "nl80211: Connect request send successfully");
6833
6834 nla_put_failure:
6835         nlmsg_free(msg);
6836         return ret;
6837
6838 }
6839
6840
6841 static int wpa_driver_nl80211_associate(
6842         void *priv, struct wpa_driver_associate_params *params)
6843 {
6844         struct i802_bss *bss = priv;
6845         struct wpa_driver_nl80211_data *drv = bss->drv;
6846         int ret = -1;
6847         struct nl_msg *msg;
6848
6849         if (params->mode == IEEE80211_MODE_AP)
6850                 return wpa_driver_nl80211_ap(drv, params);
6851
6852         if (params->mode == IEEE80211_MODE_IBSS)
6853                 return wpa_driver_nl80211_ibss(drv, params);
6854
6855         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME)) {
6856                 enum nl80211_iftype nlmode = params->p2p ?
6857                         NL80211_IFTYPE_P2P_CLIENT : NL80211_IFTYPE_STATION;
6858
6859                 if (wpa_driver_nl80211_set_mode(priv, nlmode) < 0)
6860                         return -1;
6861                 return wpa_driver_nl80211_connect(drv, params);
6862         }
6863
6864         drv->associated = 0;
6865
6866         msg = nlmsg_alloc();
6867         if (!msg)
6868                 return -1;
6869
6870         wpa_printf(MSG_DEBUG, "nl80211: Associate (ifindex=%d)",
6871                    drv->ifindex);
6872         nl80211_cmd(drv, msg, 0, NL80211_CMD_ASSOCIATE);
6873
6874         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6875         if (params->bssid) {
6876                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
6877                            MAC2STR(params->bssid));
6878                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6879         }
6880         if (params->freq) {
6881                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6882                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6883                 drv->assoc_freq = params->freq;
6884         } else
6885                 drv->assoc_freq = 0;
6886         if (params->bg_scan_period >= 0) {
6887                 wpa_printf(MSG_DEBUG, "  * bg scan period=%d",
6888                            params->bg_scan_period);
6889                 NLA_PUT_U16(msg, NL80211_ATTR_BG_SCAN_PERIOD,
6890                             params->bg_scan_period);
6891         }
6892         if (params->ssid) {
6893                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6894                                   params->ssid, params->ssid_len);
6895                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6896                         params->ssid);
6897                 if (params->ssid_len > sizeof(drv->ssid))
6898                         goto nla_put_failure;
6899                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6900                 drv->ssid_len = params->ssid_len;
6901         }
6902         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
6903         if (params->wpa_ie)
6904                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6905                         params->wpa_ie);
6906
6907         if (params->pairwise_suite != CIPHER_NONE) {
6908                 int cipher;
6909
6910                 switch (params->pairwise_suite) {
6911                 case CIPHER_WEP40:
6912                         cipher = WLAN_CIPHER_SUITE_WEP40;
6913                         break;
6914                 case CIPHER_WEP104:
6915                         cipher = WLAN_CIPHER_SUITE_WEP104;
6916                         break;
6917                 case CIPHER_CCMP:
6918                         cipher = WLAN_CIPHER_SUITE_CCMP;
6919                         break;
6920                 case CIPHER_TKIP:
6921                 default:
6922                         cipher = WLAN_CIPHER_SUITE_TKIP;
6923                         break;
6924                 }
6925                 wpa_printf(MSG_DEBUG, "  * pairwise=0x%x", cipher);
6926                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE, cipher);
6927         }
6928
6929         if (params->group_suite != CIPHER_NONE) {
6930                 int cipher;
6931
6932                 switch (params->group_suite) {
6933                 case CIPHER_WEP40:
6934                         cipher = WLAN_CIPHER_SUITE_WEP40;
6935                         break;
6936                 case CIPHER_WEP104:
6937                         cipher = WLAN_CIPHER_SUITE_WEP104;
6938                         break;
6939                 case CIPHER_CCMP:
6940                         cipher = WLAN_CIPHER_SUITE_CCMP;
6941                         break;
6942                 case CIPHER_TKIP:
6943                 default:
6944                         cipher = WLAN_CIPHER_SUITE_TKIP;
6945                         break;
6946                 }
6947                 wpa_printf(MSG_DEBUG, "  * group=0x%x", cipher);
6948                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP, cipher);
6949         }
6950
6951 #ifdef CONFIG_IEEE80211W
6952         if (params->mgmt_frame_protection == MGMT_FRAME_PROTECTION_REQUIRED)
6953                 NLA_PUT_U32(msg, NL80211_ATTR_USE_MFP, NL80211_MFP_REQUIRED);
6954 #endif /* CONFIG_IEEE80211W */
6955
6956         NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT);
6957
6958         if (params->prev_bssid) {
6959                 wpa_printf(MSG_DEBUG, "  * prev_bssid=" MACSTR,
6960                            MAC2STR(params->prev_bssid));
6961                 NLA_PUT(msg, NL80211_ATTR_PREV_BSSID, ETH_ALEN,
6962                         params->prev_bssid);
6963         }
6964
6965         if (params->disable_ht)
6966                 NLA_PUT_FLAG(msg, NL80211_ATTR_DISABLE_HT);
6967
6968         if (params->htcaps && params->htcaps_mask) {
6969                 int sz = sizeof(struct ieee80211_ht_capabilities);
6970                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY, sz, params->htcaps);
6971                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY_MASK, sz,
6972                         params->htcaps_mask);
6973         }
6974
6975         if (params->p2p)
6976                 wpa_printf(MSG_DEBUG, "  * P2P group");
6977
6978         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6979         msg = NULL;
6980         if (ret) {
6981                 wpa_dbg(drv->ctx, MSG_DEBUG,
6982                         "nl80211: MLME command failed (assoc): ret=%d (%s)",
6983                         ret, strerror(-ret));
6984                 nl80211_dump_scan(drv);
6985                 goto nla_put_failure;
6986         }
6987         ret = 0;
6988         wpa_printf(MSG_DEBUG, "nl80211: Association request send "
6989                    "successfully");
6990
6991 nla_put_failure:
6992         nlmsg_free(msg);
6993         return ret;
6994 }
6995
6996
6997 static int nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
6998                             int ifindex, enum nl80211_iftype mode)
6999 {
7000         struct nl_msg *msg;
7001         int ret = -ENOBUFS;
7002
7003         wpa_printf(MSG_DEBUG, "nl80211: Set mode ifindex %d iftype %d (%s)",
7004                    ifindex, mode, nl80211_iftype_str(mode));
7005
7006         msg = nlmsg_alloc();
7007         if (!msg)
7008                 return -ENOMEM;
7009
7010         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_INTERFACE);
7011         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
7012         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
7013
7014         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7015         msg = NULL;
7016         if (!ret)
7017                 return 0;
7018 nla_put_failure:
7019         nlmsg_free(msg);
7020         wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface %d to mode %d:"
7021                    " %d (%s)", ifindex, mode, ret, strerror(-ret));
7022         return ret;
7023 }
7024
7025
7026 static int wpa_driver_nl80211_set_mode(struct i802_bss *bss,
7027                                        enum nl80211_iftype nlmode)
7028 {
7029         struct wpa_driver_nl80211_data *drv = bss->drv;
7030         int ret = -1;
7031         int i;
7032         int was_ap = is_ap_interface(drv->nlmode);
7033         int res;
7034
7035         res = nl80211_set_mode(drv, drv->ifindex, nlmode);
7036         if (res == 0) {
7037                 drv->nlmode = nlmode;
7038                 ret = 0;
7039                 goto done;
7040         }
7041
7042         if (res == -ENODEV)
7043                 return -1;
7044
7045         if (nlmode == drv->nlmode) {
7046                 wpa_printf(MSG_DEBUG, "nl80211: Interface already in "
7047                            "requested mode - ignore error");
7048                 ret = 0;
7049                 goto done; /* Already in the requested mode */
7050         }
7051
7052         /* mac80211 doesn't allow mode changes while the device is up, so
7053          * take the device down, try to set the mode again, and bring the
7054          * device back up.
7055          */
7056         wpa_printf(MSG_DEBUG, "nl80211: Try mode change after setting "
7057                    "interface down");
7058         for (i = 0; i < 10; i++) {
7059                 res = linux_set_iface_flags(drv->global->ioctl_sock,
7060                                             bss->ifname, 0);
7061                 if (res == -EACCES || res == -ENODEV)
7062                         break;
7063                 if (res == 0) {
7064                         /* Try to set the mode again while the interface is
7065                          * down */
7066                         ret = nl80211_set_mode(drv, drv->ifindex, nlmode);
7067                         if (ret == -EACCES)
7068                                 break;
7069                         res = linux_set_iface_flags(drv->global->ioctl_sock,
7070                                                     bss->ifname, 1);
7071                         if (res && !ret)
7072                                 ret = -1;
7073                         else if (ret != -EBUSY)
7074                                 break;
7075                 } else
7076                         wpa_printf(MSG_DEBUG, "nl80211: Failed to set "
7077                                    "interface down");
7078                 os_sleep(0, 100000);
7079         }
7080
7081         if (!ret) {
7082                 wpa_printf(MSG_DEBUG, "nl80211: Mode change succeeded while "
7083                            "interface is down");
7084                 drv->nlmode = nlmode;
7085                 drv->ignore_if_down_event = 1;
7086         }
7087
7088 done:
7089         if (ret) {
7090                 wpa_printf(MSG_DEBUG, "nl80211: Interface mode change to %d "
7091                            "from %d failed", nlmode, drv->nlmode);
7092                 return ret;
7093         }
7094
7095         if (is_ap_interface(nlmode)) {
7096                 nl80211_mgmt_unsubscribe(bss, "start AP");
7097                 /* Setup additional AP mode functionality if needed */
7098                 if (nl80211_setup_ap(bss))
7099                         return -1;
7100         } else if (was_ap) {
7101                 /* Remove additional AP mode functionality */
7102                 nl80211_teardown_ap(bss);
7103         } else {
7104                 nl80211_mgmt_unsubscribe(bss, "mode change");
7105         }
7106
7107         if (!bss->in_deinit && !is_ap_interface(nlmode) &&
7108             nl80211_mgmt_subscribe_non_ap(bss) < 0)
7109                 wpa_printf(MSG_DEBUG, "nl80211: Failed to register Action "
7110                            "frame processing - ignore for now");
7111
7112         return 0;
7113 }
7114
7115
7116 static int wpa_driver_nl80211_get_capa(void *priv,
7117                                        struct wpa_driver_capa *capa)
7118 {
7119         struct i802_bss *bss = priv;
7120         struct wpa_driver_nl80211_data *drv = bss->drv;
7121         if (!drv->has_capability)
7122                 return -1;
7123         os_memcpy(capa, &drv->capa, sizeof(*capa));
7124         return 0;
7125 }
7126
7127
7128 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
7129 {
7130         struct i802_bss *bss = priv;
7131         struct wpa_driver_nl80211_data *drv = bss->drv;
7132
7133         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
7134                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
7135         drv->operstate = state;
7136         return netlink_send_oper_ifla(drv->global->netlink, drv->ifindex, -1,
7137                                       state ? IF_OPER_UP : IF_OPER_DORMANT);
7138 }
7139
7140
7141 static int wpa_driver_nl80211_set_supp_port(void *priv, int authorized)
7142 {
7143         struct i802_bss *bss = priv;
7144         struct wpa_driver_nl80211_data *drv = bss->drv;
7145         struct nl_msg *msg;
7146         struct nl80211_sta_flag_update upd;
7147
7148         msg = nlmsg_alloc();
7149         if (!msg)
7150                 return -ENOMEM;
7151
7152         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
7153
7154         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7155                     if_nametoindex(bss->ifname));
7156         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, drv->bssid);
7157
7158         os_memset(&upd, 0, sizeof(upd));
7159         upd.mask = BIT(NL80211_STA_FLAG_AUTHORIZED);
7160         if (authorized)
7161                 upd.set = BIT(NL80211_STA_FLAG_AUTHORIZED);
7162         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
7163
7164         return send_and_recv_msgs(drv, msg, NULL, NULL);
7165  nla_put_failure:
7166         nlmsg_free(msg);
7167         return -ENOBUFS;
7168 }
7169
7170
7171 /* Set kernel driver on given frequency (MHz) */
7172 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq)
7173 {
7174         struct i802_bss *bss = priv;
7175         return wpa_driver_nl80211_set_freq(bss, freq->freq, freq->ht_enabled,
7176                                            freq->sec_channel_offset);
7177 }
7178
7179
7180 #if defined(HOSTAPD) || defined(CONFIG_AP)
7181
7182 static inline int min_int(int a, int b)
7183 {
7184         if (a < b)
7185                 return a;
7186         return b;
7187 }
7188
7189
7190 static int get_key_handler(struct nl_msg *msg, void *arg)
7191 {
7192         struct nlattr *tb[NL80211_ATTR_MAX + 1];
7193         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
7194
7195         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
7196                   genlmsg_attrlen(gnlh, 0), NULL);
7197
7198         /*
7199          * TODO: validate the key index and mac address!
7200          * Otherwise, there's a race condition as soon as
7201          * the kernel starts sending key notifications.
7202          */
7203
7204         if (tb[NL80211_ATTR_KEY_SEQ])
7205                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
7206                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
7207         return NL_SKIP;
7208 }
7209
7210
7211 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
7212                            int idx, u8 *seq)
7213 {
7214         struct i802_bss *bss = priv;
7215         struct wpa_driver_nl80211_data *drv = bss->drv;
7216         struct nl_msg *msg;
7217
7218         msg = nlmsg_alloc();
7219         if (!msg)
7220                 return -ENOMEM;
7221
7222         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_KEY);
7223
7224         if (addr)
7225                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7226         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
7227         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
7228
7229         memset(seq, 0, 6);
7230
7231         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
7232  nla_put_failure:
7233         nlmsg_free(msg);
7234         return -ENOBUFS;
7235 }
7236
7237
7238 static int i802_set_rts(void *priv, int rts)
7239 {
7240         struct i802_bss *bss = priv;
7241         struct wpa_driver_nl80211_data *drv = bss->drv;
7242         struct nl_msg *msg;
7243         int ret = -ENOBUFS;
7244         u32 val;
7245
7246         msg = nlmsg_alloc();
7247         if (!msg)
7248                 return -ENOMEM;
7249
7250         if (rts >= 2347)
7251                 val = (u32) -1;
7252         else
7253                 val = rts;
7254
7255         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7256         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
7257         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD, val);
7258
7259         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7260         msg = NULL;
7261         if (!ret)
7262                 return 0;
7263 nla_put_failure:
7264         nlmsg_free(msg);
7265         wpa_printf(MSG_DEBUG, "nl80211: Failed to set RTS threshold %d: "
7266                    "%d (%s)", rts, ret, strerror(-ret));
7267         return ret;
7268 }
7269
7270
7271 static int i802_set_frag(void *priv, int frag)
7272 {
7273         struct i802_bss *bss = priv;
7274         struct wpa_driver_nl80211_data *drv = bss->drv;
7275         struct nl_msg *msg;
7276         int ret = -ENOBUFS;
7277         u32 val;
7278
7279         msg = nlmsg_alloc();
7280         if (!msg)
7281                 return -ENOMEM;
7282
7283         if (frag >= 2346)
7284                 val = (u32) -1;
7285         else
7286                 val = frag;
7287
7288         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7289         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
7290         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD, val);
7291
7292         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7293         msg = NULL;
7294         if (!ret)
7295                 return 0;
7296 nla_put_failure:
7297         nlmsg_free(msg);
7298         wpa_printf(MSG_DEBUG, "nl80211: Failed to set fragmentation threshold "
7299                    "%d: %d (%s)", frag, ret, strerror(-ret));
7300         return ret;
7301 }
7302
7303
7304 static int i802_flush(void *priv)
7305 {
7306         struct i802_bss *bss = priv;
7307         struct wpa_driver_nl80211_data *drv = bss->drv;
7308         struct nl_msg *msg;
7309         int res;
7310
7311         msg = nlmsg_alloc();
7312         if (!msg)
7313                 return -1;
7314
7315         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_STATION);
7316
7317         /*
7318          * XXX: FIX! this needs to flush all VLANs too
7319          */
7320         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7321                     if_nametoindex(bss->ifname));
7322
7323         res = send_and_recv_msgs(drv, msg, NULL, NULL);
7324         if (res) {
7325                 wpa_printf(MSG_DEBUG, "nl80211: Station flush failed: ret=%d "
7326                            "(%s)", res, strerror(-res));
7327         }
7328         return res;
7329  nla_put_failure:
7330         nlmsg_free(msg);
7331         return -ENOBUFS;
7332 }
7333
7334
7335 static int get_sta_handler(struct nl_msg *msg, void *arg)
7336 {
7337         struct nlattr *tb[NL80211_ATTR_MAX + 1];
7338         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
7339         struct hostap_sta_driver_data *data = arg;
7340         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
7341         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
7342                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
7343                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
7344                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
7345                 [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
7346                 [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
7347         };
7348
7349         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
7350                   genlmsg_attrlen(gnlh, 0), NULL);
7351
7352         /*
7353          * TODO: validate the interface and mac address!
7354          * Otherwise, there's a race condition as soon as
7355          * the kernel starts sending station notifications.
7356          */
7357
7358         if (!tb[NL80211_ATTR_STA_INFO]) {
7359                 wpa_printf(MSG_DEBUG, "sta stats missing!");
7360                 return NL_SKIP;
7361         }
7362         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
7363                              tb[NL80211_ATTR_STA_INFO],
7364                              stats_policy)) {
7365                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
7366                 return NL_SKIP;
7367         }
7368
7369         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
7370                 data->inactive_msec =
7371                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
7372         if (stats[NL80211_STA_INFO_RX_BYTES])
7373                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
7374         if (stats[NL80211_STA_INFO_TX_BYTES])
7375                 data->tx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
7376         if (stats[NL80211_STA_INFO_RX_PACKETS])
7377                 data->rx_packets =
7378                         nla_get_u32(stats[NL80211_STA_INFO_RX_PACKETS]);
7379         if (stats[NL80211_STA_INFO_TX_PACKETS])
7380                 data->tx_packets =
7381                         nla_get_u32(stats[NL80211_STA_INFO_TX_PACKETS]);
7382
7383         return NL_SKIP;
7384 }
7385
7386 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
7387                               const u8 *addr)
7388 {
7389         struct i802_bss *bss = priv;
7390         struct wpa_driver_nl80211_data *drv = bss->drv;
7391         struct nl_msg *msg;
7392
7393         os_memset(data, 0, sizeof(*data));
7394         msg = nlmsg_alloc();
7395         if (!msg)
7396                 return -ENOMEM;
7397
7398         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_STATION);
7399
7400         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7401         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
7402
7403         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
7404  nla_put_failure:
7405         nlmsg_free(msg);
7406         return -ENOBUFS;
7407 }
7408
7409
7410 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
7411                                     int cw_min, int cw_max, int burst_time)
7412 {
7413         struct i802_bss *bss = priv;
7414         struct wpa_driver_nl80211_data *drv = bss->drv;
7415         struct nl_msg *msg;
7416         struct nlattr *txq, *params;
7417
7418         msg = nlmsg_alloc();
7419         if (!msg)
7420                 return -1;
7421
7422         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7423
7424         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
7425
7426         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
7427         if (!txq)
7428                 goto nla_put_failure;
7429
7430         /* We are only sending parameters for a single TXQ at a time */
7431         params = nla_nest_start(msg, 1);
7432         if (!params)
7433                 goto nla_put_failure;
7434
7435         switch (queue) {
7436         case 0:
7437                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_VO);
7438                 break;
7439         case 1:
7440                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_VI);
7441                 break;
7442         case 2:
7443                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_BE);
7444                 break;
7445         case 3:
7446                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_BK);
7447                 break;
7448         }
7449         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
7450          * 32 usec, so need to convert the value here. */
7451         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
7452         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
7453         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
7454         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
7455
7456         nla_nest_end(msg, params);
7457
7458         nla_nest_end(msg, txq);
7459
7460         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
7461                 return 0;
7462         msg = NULL;
7463  nla_put_failure:
7464         nlmsg_free(msg);
7465         return -1;
7466 }
7467
7468
7469 static int i802_set_sta_vlan(void *priv, const u8 *addr,
7470                              const char *ifname, int vlan_id)
7471 {
7472         struct i802_bss *bss = priv;
7473         struct wpa_driver_nl80211_data *drv = bss->drv;
7474         struct nl_msg *msg;
7475         int ret = -ENOBUFS;
7476
7477         msg = nlmsg_alloc();
7478         if (!msg)
7479                 return -ENOMEM;
7480
7481         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
7482
7483         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7484                     if_nametoindex(bss->ifname));
7485         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7486         NLA_PUT_U32(msg, NL80211_ATTR_STA_VLAN,
7487                     if_nametoindex(ifname));
7488
7489         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7490         msg = NULL;
7491         if (ret < 0) {
7492                 wpa_printf(MSG_ERROR, "nl80211: NL80211_ATTR_STA_VLAN (addr="
7493                            MACSTR " ifname=%s vlan_id=%d) failed: %d (%s)",
7494                            MAC2STR(addr), ifname, vlan_id, ret,
7495                            strerror(-ret));
7496         }
7497  nla_put_failure:
7498         nlmsg_free(msg);
7499         return ret;
7500 }
7501
7502
7503 static int i802_get_inact_sec(void *priv, const u8 *addr)
7504 {
7505         struct hostap_sta_driver_data data;
7506         int ret;
7507
7508         data.inactive_msec = (unsigned long) -1;
7509         ret = i802_read_sta_data(priv, &data, addr);
7510         if (ret || data.inactive_msec == (unsigned long) -1)
7511                 return -1;
7512         return data.inactive_msec / 1000;
7513 }
7514
7515
7516 static int i802_sta_clear_stats(void *priv, const u8 *addr)
7517 {
7518 #if 0
7519         /* TODO */
7520 #endif
7521         return 0;
7522 }
7523
7524
7525 static int i802_sta_deauth(void *priv, const u8 *own_addr, const u8 *addr,
7526                            int reason)
7527 {
7528         struct i802_bss *bss = priv;
7529         struct wpa_driver_nl80211_data *drv = bss->drv;
7530         struct ieee80211_mgmt mgmt;
7531
7532         if (drv->device_ap_sme)
7533                 return wpa_driver_nl80211_sta_remove(bss, addr);
7534
7535         memset(&mgmt, 0, sizeof(mgmt));
7536         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
7537                                           WLAN_FC_STYPE_DEAUTH);
7538         memcpy(mgmt.da, addr, ETH_ALEN);
7539         memcpy(mgmt.sa, own_addr, ETH_ALEN);
7540         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
7541         mgmt.u.deauth.reason_code = host_to_le16(reason);
7542         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
7543                                             IEEE80211_HDRLEN +
7544                                             sizeof(mgmt.u.deauth), 0);
7545 }
7546
7547
7548 static int i802_sta_disassoc(void *priv, const u8 *own_addr, const u8 *addr,
7549                              int reason)
7550 {
7551         struct i802_bss *bss = priv;
7552         struct wpa_driver_nl80211_data *drv = bss->drv;
7553         struct ieee80211_mgmt mgmt;
7554
7555         if (drv->device_ap_sme)
7556                 return wpa_driver_nl80211_sta_remove(bss, addr);
7557
7558         memset(&mgmt, 0, sizeof(mgmt));
7559         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
7560                                           WLAN_FC_STYPE_DISASSOC);
7561         memcpy(mgmt.da, addr, ETH_ALEN);
7562         memcpy(mgmt.sa, own_addr, ETH_ALEN);
7563         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
7564         mgmt.u.disassoc.reason_code = host_to_le16(reason);
7565         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
7566                                             IEEE80211_HDRLEN +
7567                                             sizeof(mgmt.u.disassoc), 0);
7568 }
7569
7570 #endif /* HOSTAPD || CONFIG_AP */
7571
7572 #ifdef HOSTAPD
7573
7574 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7575 {
7576         int i;
7577         int *old;
7578
7579         wpa_printf(MSG_DEBUG, "nl80211: Add own interface ifindex %d",
7580                    ifidx);
7581         for (i = 0; i < drv->num_if_indices; i++) {
7582                 if (drv->if_indices[i] == 0) {
7583                         drv->if_indices[i] = ifidx;
7584                         return;
7585                 }
7586         }
7587
7588         if (drv->if_indices != drv->default_if_indices)
7589                 old = drv->if_indices;
7590         else
7591                 old = NULL;
7592
7593         drv->if_indices = os_realloc(old,
7594                                      sizeof(int) * (drv->num_if_indices + 1));
7595         if (!drv->if_indices) {
7596                 if (!old)
7597                         drv->if_indices = drv->default_if_indices;
7598                 else
7599                         drv->if_indices = old;
7600                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
7601                            "interfaces");
7602                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
7603                 return;
7604         } else if (!old)
7605                 os_memcpy(drv->if_indices, drv->default_if_indices,
7606                           sizeof(drv->default_if_indices));
7607         drv->if_indices[drv->num_if_indices] = ifidx;
7608         drv->num_if_indices++;
7609 }
7610
7611
7612 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7613 {
7614         int i;
7615
7616         for (i = 0; i < drv->num_if_indices; i++) {
7617                 if (drv->if_indices[i] == ifidx) {
7618                         drv->if_indices[i] = 0;
7619                         break;
7620                 }
7621         }
7622 }
7623
7624
7625 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7626 {
7627         int i;
7628
7629         for (i = 0; i < drv->num_if_indices; i++)
7630                 if (drv->if_indices[i] == ifidx)
7631                         return 1;
7632
7633         return 0;
7634 }
7635
7636
7637 static int i802_set_wds_sta(void *priv, const u8 *addr, int aid, int val,
7638                             const char *bridge_ifname)
7639 {
7640         struct i802_bss *bss = priv;
7641         struct wpa_driver_nl80211_data *drv = bss->drv;
7642         char name[IFNAMSIZ + 1];
7643
7644         os_snprintf(name, sizeof(name), "%s.sta%d", bss->ifname, aid);
7645         wpa_printf(MSG_DEBUG, "nl80211: Set WDS STA addr=" MACSTR
7646                    " aid=%d val=%d name=%s", MAC2STR(addr), aid, val, name);
7647         if (val) {
7648                 if (!if_nametoindex(name)) {
7649                         if (nl80211_create_iface(drv, name,
7650                                                  NL80211_IFTYPE_AP_VLAN,
7651                                                  NULL, 1) < 0)
7652                                 return -1;
7653                         if (bridge_ifname &&
7654                             linux_br_add_if(drv->global->ioctl_sock,
7655                                             bridge_ifname, name) < 0)
7656                                 return -1;
7657                 }
7658                 linux_set_iface_flags(drv->global->ioctl_sock, name, 1);
7659                 return i802_set_sta_vlan(priv, addr, name, 0);
7660         } else {
7661                 i802_set_sta_vlan(priv, addr, bss->ifname, 0);
7662                 return wpa_driver_nl80211_if_remove(priv, WPA_IF_AP_VLAN,
7663                                                     name);
7664         }
7665 }
7666
7667
7668 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
7669 {
7670         struct wpa_driver_nl80211_data *drv = eloop_ctx;
7671         struct sockaddr_ll lladdr;
7672         unsigned char buf[3000];
7673         int len;
7674         socklen_t fromlen = sizeof(lladdr);
7675
7676         len = recvfrom(sock, buf, sizeof(buf), 0,
7677                        (struct sockaddr *)&lladdr, &fromlen);
7678         if (len < 0) {
7679                 perror("recv");
7680                 return;
7681         }
7682
7683         if (have_ifidx(drv, lladdr.sll_ifindex))
7684                 drv_event_eapol_rx(drv->ctx, lladdr.sll_addr, buf, len);
7685 }
7686
7687
7688 static int i802_check_bridge(struct wpa_driver_nl80211_data *drv,
7689                              struct i802_bss *bss,
7690                              const char *brname, const char *ifname)
7691 {
7692         int ifindex;
7693         char in_br[IFNAMSIZ];
7694
7695         os_strlcpy(bss->brname, brname, IFNAMSIZ);
7696         ifindex = if_nametoindex(brname);
7697         if (ifindex == 0) {
7698                 /*
7699                  * Bridge was configured, but the bridge device does
7700                  * not exist. Try to add it now.
7701                  */
7702                 if (linux_br_add(drv->global->ioctl_sock, brname) < 0) {
7703                         wpa_printf(MSG_ERROR, "nl80211: Failed to add the "
7704                                    "bridge interface %s: %s",
7705                                    brname, strerror(errno));
7706                         return -1;
7707                 }
7708                 bss->added_bridge = 1;
7709                 add_ifidx(drv, if_nametoindex(brname));
7710         }
7711
7712         if (linux_br_get(in_br, ifname) == 0) {
7713                 if (os_strcmp(in_br, brname) == 0)
7714                         return 0; /* already in the bridge */
7715
7716                 wpa_printf(MSG_DEBUG, "nl80211: Removing interface %s from "
7717                            "bridge %s", ifname, in_br);
7718                 if (linux_br_del_if(drv->global->ioctl_sock, in_br, ifname) <
7719                     0) {
7720                         wpa_printf(MSG_ERROR, "nl80211: Failed to "
7721                                    "remove interface %s from bridge "
7722                                    "%s: %s",
7723                                    ifname, brname, strerror(errno));
7724                         return -1;
7725                 }
7726         }
7727
7728         wpa_printf(MSG_DEBUG, "nl80211: Adding interface %s into bridge %s",
7729                    ifname, brname);
7730         if (linux_br_add_if(drv->global->ioctl_sock, brname, ifname) < 0) {
7731                 wpa_printf(MSG_ERROR, "nl80211: Failed to add interface %s "
7732                            "into bridge %s: %s",
7733                            ifname, brname, strerror(errno));
7734                 return -1;
7735         }
7736         bss->added_if_into_bridge = 1;
7737
7738         return 0;
7739 }
7740
7741
7742 static void *i802_init(struct hostapd_data *hapd,
7743                        struct wpa_init_params *params)
7744 {
7745         struct wpa_driver_nl80211_data *drv;
7746         struct i802_bss *bss;
7747         size_t i;
7748         char brname[IFNAMSIZ];
7749         int ifindex, br_ifindex;
7750         int br_added = 0;
7751
7752         bss = wpa_driver_nl80211_init(hapd, params->ifname,
7753                                       params->global_priv);
7754         if (bss == NULL)
7755                 return NULL;
7756
7757         drv = bss->drv;
7758         drv->nlmode = NL80211_IFTYPE_AP;
7759         drv->eapol_sock = -1;
7760
7761         if (linux_br_get(brname, params->ifname) == 0) {
7762                 wpa_printf(MSG_DEBUG, "nl80211: Interface %s is in bridge %s",
7763                            params->ifname, brname);
7764                 br_ifindex = if_nametoindex(brname);
7765         } else {
7766                 brname[0] = '\0';
7767                 br_ifindex = 0;
7768         }
7769
7770         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
7771         drv->if_indices = drv->default_if_indices;
7772         for (i = 0; i < params->num_bridge; i++) {
7773                 if (params->bridge[i]) {
7774                         ifindex = if_nametoindex(params->bridge[i]);
7775                         if (ifindex)
7776                                 add_ifidx(drv, ifindex);
7777                         if (ifindex == br_ifindex)
7778                                 br_added = 1;
7779                 }
7780         }
7781         if (!br_added && br_ifindex &&
7782             (params->num_bridge == 0 || !params->bridge[0]))
7783                 add_ifidx(drv, br_ifindex);
7784
7785         /* start listening for EAPOL on the default AP interface */
7786         add_ifidx(drv, drv->ifindex);
7787
7788         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 0))
7789                 goto failed;
7790
7791         if (params->bssid) {
7792                 if (linux_set_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7793                                        params->bssid))
7794                         goto failed;
7795         }
7796
7797         if (wpa_driver_nl80211_set_mode(bss, drv->nlmode)) {
7798                 wpa_printf(MSG_ERROR, "nl80211: Failed to set interface %s "
7799                            "into AP mode", bss->ifname);
7800                 goto failed;
7801         }
7802
7803         if (params->num_bridge && params->bridge[0] &&
7804             i802_check_bridge(drv, bss, params->bridge[0], params->ifname) < 0)
7805                 goto failed;
7806
7807         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 1))
7808                 goto failed;
7809
7810         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
7811         if (drv->eapol_sock < 0) {
7812                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
7813                 goto failed;
7814         }
7815
7816         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
7817         {
7818                 printf("Could not register read socket for eapol\n");
7819                 goto failed;
7820         }
7821
7822         if (linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7823                                params->own_addr))
7824                 goto failed;
7825
7826         memcpy(bss->addr, params->own_addr, ETH_ALEN);
7827
7828         return bss;
7829
7830 failed:
7831         wpa_driver_nl80211_deinit(bss);
7832         return NULL;
7833 }
7834
7835
7836 static void i802_deinit(void *priv)
7837 {
7838         wpa_driver_nl80211_deinit(priv);
7839 }
7840
7841 #endif /* HOSTAPD */
7842
7843
7844 static enum nl80211_iftype wpa_driver_nl80211_if_type(
7845         enum wpa_driver_if_type type)
7846 {
7847         switch (type) {
7848         case WPA_IF_STATION:
7849                 return NL80211_IFTYPE_STATION;
7850         case WPA_IF_P2P_CLIENT:
7851         case WPA_IF_P2P_GROUP:
7852                 return NL80211_IFTYPE_P2P_CLIENT;
7853         case WPA_IF_AP_VLAN:
7854                 return NL80211_IFTYPE_AP_VLAN;
7855         case WPA_IF_AP_BSS:
7856                 return NL80211_IFTYPE_AP;
7857         case WPA_IF_P2P_GO:
7858                 return NL80211_IFTYPE_P2P_GO;
7859         }
7860         return -1;
7861 }
7862
7863
7864 #ifdef CONFIG_P2P
7865
7866 static int nl80211_addr_in_use(struct nl80211_global *global, const u8 *addr)
7867 {
7868         struct wpa_driver_nl80211_data *drv;
7869         dl_list_for_each(drv, &global->interfaces,
7870                          struct wpa_driver_nl80211_data, list) {
7871                 if (os_memcmp(addr, drv->first_bss.addr, ETH_ALEN) == 0)
7872                         return 1;
7873         }
7874         return 0;
7875 }
7876
7877
7878 static int nl80211_p2p_interface_addr(struct wpa_driver_nl80211_data *drv,
7879                                       u8 *new_addr)
7880 {
7881         unsigned int idx;
7882
7883         if (!drv->global)
7884                 return -1;
7885
7886         os_memcpy(new_addr, drv->first_bss.addr, ETH_ALEN);
7887         for (idx = 0; idx < 64; idx++) {
7888                 new_addr[0] = drv->first_bss.addr[0] | 0x02;
7889                 new_addr[0] ^= idx << 2;
7890                 if (!nl80211_addr_in_use(drv->global, new_addr))
7891                         break;
7892         }
7893         if (idx == 64)
7894                 return -1;
7895
7896         wpa_printf(MSG_DEBUG, "nl80211: Assigned new P2P Interface Address "
7897                    MACSTR, MAC2STR(new_addr));
7898
7899         return 0;
7900 }
7901
7902 #endif /* CONFIG_P2P */
7903
7904
7905 static int wpa_driver_nl80211_if_add(void *priv, enum wpa_driver_if_type type,
7906                                      const char *ifname, const u8 *addr,
7907                                      void *bss_ctx, void **drv_priv,
7908                                      char *force_ifname, u8 *if_addr,
7909                                      const char *bridge)
7910 {
7911         struct i802_bss *bss = priv;
7912         struct wpa_driver_nl80211_data *drv = bss->drv;
7913         int ifidx;
7914 #ifdef HOSTAPD
7915         struct i802_bss *new_bss = NULL;
7916
7917         if (type == WPA_IF_AP_BSS) {
7918                 new_bss = os_zalloc(sizeof(*new_bss));
7919                 if (new_bss == NULL)
7920                         return -1;
7921         }
7922 #endif /* HOSTAPD */
7923
7924         if (addr)
7925                 os_memcpy(if_addr, addr, ETH_ALEN);
7926         ifidx = nl80211_create_iface(drv, ifname,
7927                                      wpa_driver_nl80211_if_type(type), addr,
7928                                      0);
7929         if (ifidx < 0) {
7930 #ifdef HOSTAPD
7931                 os_free(new_bss);
7932 #endif /* HOSTAPD */
7933                 return -1;
7934         }
7935
7936         if (!addr &&
7937             linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7938                                if_addr) < 0) {
7939                 nl80211_remove_iface(drv, ifidx);
7940                 return -1;
7941         }
7942
7943 #ifdef CONFIG_P2P
7944         if (!addr &&
7945             (type == WPA_IF_P2P_CLIENT || type == WPA_IF_P2P_GROUP ||
7946              type == WPA_IF_P2P_GO)) {
7947                 /* Enforce unique P2P Interface Address */
7948                 u8 new_addr[ETH_ALEN], own_addr[ETH_ALEN];
7949
7950                 if (linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7951                                        own_addr) < 0 ||
7952                     linux_get_ifhwaddr(drv->global->ioctl_sock, ifname,
7953                                        new_addr) < 0) {
7954                         nl80211_remove_iface(drv, ifidx);
7955                         return -1;
7956                 }
7957                 if (os_memcmp(own_addr, new_addr, ETH_ALEN) == 0) {
7958                         wpa_printf(MSG_DEBUG, "nl80211: Allocate new address "
7959                                    "for P2P group interface");
7960                         if (nl80211_p2p_interface_addr(drv, new_addr) < 0) {
7961                                 nl80211_remove_iface(drv, ifidx);
7962                                 return -1;
7963                         }
7964                         if (linux_set_ifhwaddr(drv->global->ioctl_sock, ifname,
7965                                                new_addr) < 0) {
7966                                 nl80211_remove_iface(drv, ifidx);
7967                                 return -1;
7968                         }
7969                 }
7970                 os_memcpy(if_addr, new_addr, ETH_ALEN);
7971         }
7972 #endif /* CONFIG_P2P */
7973
7974 #ifdef HOSTAPD
7975         if (bridge &&
7976             i802_check_bridge(drv, new_bss, bridge, ifname) < 0) {
7977                 wpa_printf(MSG_ERROR, "nl80211: Failed to add the new "
7978                            "interface %s to a bridge %s", ifname, bridge);
7979                 nl80211_remove_iface(drv, ifidx);
7980                 os_free(new_bss);
7981                 return -1;
7982         }
7983
7984         if (type == WPA_IF_AP_BSS) {
7985                 if (linux_set_iface_flags(drv->global->ioctl_sock, ifname, 1))
7986                 {
7987                         nl80211_remove_iface(drv, ifidx);
7988                         os_free(new_bss);
7989                         return -1;
7990                 }
7991                 os_strlcpy(new_bss->ifname, ifname, IFNAMSIZ);
7992                 os_memcpy(new_bss->addr, if_addr, ETH_ALEN);
7993                 new_bss->ifindex = ifidx;
7994                 new_bss->drv = drv;
7995                 new_bss->next = drv->first_bss.next;
7996                 new_bss->freq = drv->first_bss.freq;
7997                 drv->first_bss.next = new_bss;
7998                 if (drv_priv)
7999                         *drv_priv = new_bss;
8000                 nl80211_init_bss(new_bss);
8001
8002                 /* Subscribe management frames for this WPA_IF_AP_BSS */
8003                 if (nl80211_setup_ap(new_bss))
8004                         return -1;
8005         }
8006 #endif /* HOSTAPD */
8007
8008         if (drv->global)
8009                 drv->global->if_add_ifindex = ifidx;
8010
8011         return 0;
8012 }
8013
8014
8015 static int wpa_driver_nl80211_if_remove(void *priv,
8016                                         enum wpa_driver_if_type type,
8017                                         const char *ifname)
8018 {
8019         struct i802_bss *bss = priv;
8020         struct wpa_driver_nl80211_data *drv = bss->drv;
8021         int ifindex = if_nametoindex(ifname);
8022
8023         wpa_printf(MSG_DEBUG, "nl80211: %s(type=%d ifname=%s) ifindex=%d",
8024                    __func__, type, ifname, ifindex);
8025         if (ifindex <= 0)
8026                 return -1;
8027
8028 #ifdef HOSTAPD
8029         if (bss->added_if_into_bridge) {
8030                 if (linux_br_del_if(drv->global->ioctl_sock, bss->brname,
8031                                     bss->ifname) < 0)
8032                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
8033                                    "interface %s from bridge %s: %s",
8034                                    bss->ifname, bss->brname, strerror(errno));
8035         }
8036         if (bss->added_bridge) {
8037                 if (linux_br_del(drv->global->ioctl_sock, bss->brname) < 0)
8038                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
8039                                    "bridge %s: %s",
8040                                    bss->brname, strerror(errno));
8041         }
8042 #endif /* HOSTAPD */
8043
8044         nl80211_remove_iface(drv, ifindex);
8045
8046 #ifdef HOSTAPD
8047         if (type != WPA_IF_AP_BSS)
8048                 return 0;
8049
8050         if (bss != &drv->first_bss) {
8051                 struct i802_bss *tbss;
8052
8053                 for (tbss = &drv->first_bss; tbss; tbss = tbss->next) {
8054                         if (tbss->next == bss) {
8055                                 tbss->next = bss->next;
8056                                 /* Unsubscribe management frames */
8057                                 nl80211_teardown_ap(bss);
8058                                 nl80211_destroy_bss(bss);
8059                                 os_free(bss);
8060                                 bss = NULL;
8061                                 break;
8062                         }
8063                 }
8064                 if (bss)
8065                         wpa_printf(MSG_INFO, "nl80211: %s - could not find "
8066                                    "BSS %p in the list", __func__, bss);
8067         }
8068 #endif /* HOSTAPD */
8069
8070         return 0;
8071 }
8072
8073
8074 static int cookie_handler(struct nl_msg *msg, void *arg)
8075 {
8076         struct nlattr *tb[NL80211_ATTR_MAX + 1];
8077         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
8078         u64 *cookie = arg;
8079         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
8080                   genlmsg_attrlen(gnlh, 0), NULL);
8081         if (tb[NL80211_ATTR_COOKIE])
8082                 *cookie = nla_get_u64(tb[NL80211_ATTR_COOKIE]);
8083         return NL_SKIP;
8084 }
8085
8086
8087 static int nl80211_send_frame_cmd(struct i802_bss *bss,
8088                                   unsigned int freq, unsigned int wait,
8089                                   const u8 *buf, size_t buf_len,
8090                                   u64 *cookie_out, int no_cck, int no_ack,
8091                                   int offchanok)
8092 {
8093         struct wpa_driver_nl80211_data *drv = bss->drv;
8094         struct nl_msg *msg;
8095         u64 cookie;
8096         int ret = -1;
8097
8098         msg = nlmsg_alloc();
8099         if (!msg)
8100                 return -1;
8101
8102         wpa_printf(MSG_DEBUG, "nl80211: CMD_FRAME freq=%u wait=%u no_cck=%d "
8103                    "no_ack=%d offchanok=%d",
8104                    freq, wait, no_cck, no_ack, offchanok);
8105         nl80211_cmd(drv, msg, 0, NL80211_CMD_FRAME);
8106
8107         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8108         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
8109         if (wait)
8110                 NLA_PUT_U32(msg, NL80211_ATTR_DURATION, wait);
8111         if (offchanok && (drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
8112                 NLA_PUT_FLAG(msg, NL80211_ATTR_OFFCHANNEL_TX_OK);
8113         if (no_cck)
8114                 NLA_PUT_FLAG(msg, NL80211_ATTR_TX_NO_CCK_RATE);
8115         if (no_ack)
8116                 NLA_PUT_FLAG(msg, NL80211_ATTR_DONT_WAIT_FOR_ACK);
8117
8118         NLA_PUT(msg, NL80211_ATTR_FRAME, buf_len, buf);
8119
8120         cookie = 0;
8121         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
8122         msg = NULL;
8123         if (ret) {
8124                 wpa_printf(MSG_DEBUG, "nl80211: Frame command failed: ret=%d "
8125                            "(%s) (freq=%u wait=%u)", ret, strerror(-ret),
8126                            freq, wait);
8127                 goto nla_put_failure;
8128         }
8129         wpa_printf(MSG_DEBUG, "nl80211: Frame TX command accepted%s; "
8130                    "cookie 0x%llx", no_ack ? " (no ACK)" : "",
8131                    (long long unsigned int) cookie);
8132
8133         if (cookie_out)
8134                 *cookie_out = no_ack ? (u64) -1 : cookie;
8135
8136 nla_put_failure:
8137         nlmsg_free(msg);
8138         return ret;
8139 }
8140
8141
8142 static int wpa_driver_nl80211_send_action(void *priv, unsigned int freq,
8143                                           unsigned int wait_time,
8144                                           const u8 *dst, const u8 *src,
8145                                           const u8 *bssid,
8146                                           const u8 *data, size_t data_len,
8147                                           int no_cck)
8148 {
8149         struct i802_bss *bss = priv;
8150         struct wpa_driver_nl80211_data *drv = bss->drv;
8151         int ret = -1;
8152         u8 *buf;
8153         struct ieee80211_hdr *hdr;
8154
8155         wpa_printf(MSG_DEBUG, "nl80211: Send Action frame (ifindex=%d, "
8156                    "freq=%u MHz wait=%d ms no_cck=%d)",
8157                    drv->ifindex, freq, wait_time, no_cck);
8158
8159         buf = os_zalloc(24 + data_len);
8160         if (buf == NULL)
8161                 return ret;
8162         os_memcpy(buf + 24, data, data_len);
8163         hdr = (struct ieee80211_hdr *) buf;
8164         hdr->frame_control =
8165                 IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_ACTION);
8166         os_memcpy(hdr->addr1, dst, ETH_ALEN);
8167         os_memcpy(hdr->addr2, src, ETH_ALEN);
8168         os_memcpy(hdr->addr3, bssid, ETH_ALEN);
8169
8170         if (is_ap_interface(drv->nlmode))
8171                 ret = wpa_driver_nl80211_send_mlme_freq(priv, buf,
8172                                                         24 + data_len,
8173                                                         0, freq, no_cck, 1,
8174                                                         wait_time);
8175         else
8176                 ret = nl80211_send_frame_cmd(bss, freq, wait_time, buf,
8177                                              24 + data_len,
8178                                              &drv->send_action_cookie,
8179                                              no_cck, 0, 1);
8180
8181         os_free(buf);
8182         return ret;
8183 }
8184
8185
8186 static void wpa_driver_nl80211_send_action_cancel_wait(void *priv)
8187 {
8188         struct i802_bss *bss = priv;
8189         struct wpa_driver_nl80211_data *drv = bss->drv;
8190         struct nl_msg *msg;
8191         int ret;
8192
8193         msg = nlmsg_alloc();
8194         if (!msg)
8195                 return;
8196
8197         nl80211_cmd(drv, msg, 0, NL80211_CMD_FRAME_WAIT_CANCEL);
8198
8199         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8200         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, drv->send_action_cookie);
8201
8202         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8203         msg = NULL;
8204         if (ret)
8205                 wpa_printf(MSG_DEBUG, "nl80211: wait cancel failed: ret=%d "
8206                            "(%s)", ret, strerror(-ret));
8207
8208  nla_put_failure:
8209         nlmsg_free(msg);
8210 }
8211
8212
8213 static int wpa_driver_nl80211_remain_on_channel(void *priv, unsigned int freq,
8214                                                 unsigned int duration)
8215 {
8216         struct i802_bss *bss = priv;
8217         struct wpa_driver_nl80211_data *drv = bss->drv;
8218         struct nl_msg *msg;
8219         int ret;
8220         u64 cookie;
8221
8222         msg = nlmsg_alloc();
8223         if (!msg)
8224                 return -1;
8225
8226         nl80211_cmd(drv, msg, 0, NL80211_CMD_REMAIN_ON_CHANNEL);
8227
8228         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8229         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
8230         NLA_PUT_U32(msg, NL80211_ATTR_DURATION, duration);
8231
8232         cookie = 0;
8233         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
8234         msg = NULL;
8235         if (ret == 0) {
8236                 wpa_printf(MSG_DEBUG, "nl80211: Remain-on-channel cookie "
8237                            "0x%llx for freq=%u MHz duration=%u",
8238                            (long long unsigned int) cookie, freq, duration);
8239                 drv->remain_on_chan_cookie = cookie;
8240                 drv->pending_remain_on_chan = 1;
8241                 return 0;
8242         }
8243         wpa_printf(MSG_DEBUG, "nl80211: Failed to request remain-on-channel "
8244                    "(freq=%d duration=%u): %d (%s)",
8245                    freq, duration, ret, strerror(-ret));
8246 nla_put_failure:
8247         nlmsg_free(msg);
8248         return -1;
8249 }
8250
8251
8252 static int wpa_driver_nl80211_cancel_remain_on_channel(void *priv)
8253 {
8254         struct i802_bss *bss = priv;
8255         struct wpa_driver_nl80211_data *drv = bss->drv;
8256         struct nl_msg *msg;
8257         int ret;
8258
8259         if (!drv->pending_remain_on_chan) {
8260                 wpa_printf(MSG_DEBUG, "nl80211: No pending remain-on-channel "
8261                            "to cancel");
8262                 return -1;
8263         }
8264
8265         wpa_printf(MSG_DEBUG, "nl80211: Cancel remain-on-channel with cookie "
8266                    "0x%llx",
8267                    (long long unsigned int) drv->remain_on_chan_cookie);
8268
8269         msg = nlmsg_alloc();
8270         if (!msg)
8271                 return -1;
8272
8273         nl80211_cmd(drv, msg, 0, NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL);
8274
8275         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8276         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, drv->remain_on_chan_cookie);
8277
8278         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8279         msg = NULL;
8280         if (ret == 0)
8281                 return 0;
8282         wpa_printf(MSG_DEBUG, "nl80211: Failed to cancel remain-on-channel: "
8283                    "%d (%s)", ret, strerror(-ret));
8284 nla_put_failure:
8285         nlmsg_free(msg);
8286         return -1;
8287 }
8288
8289
8290 static int wpa_driver_nl80211_probe_req_report(void *priv, int report)
8291 {
8292         struct i802_bss *bss = priv;
8293         struct wpa_driver_nl80211_data *drv = bss->drv;
8294
8295         if (!report) {
8296                 if (bss->nl_preq && drv->device_ap_sme &&
8297                     is_ap_interface(drv->nlmode)) {
8298                         /*
8299                          * Do not disable Probe Request reporting that was
8300                          * enabled in nl80211_setup_ap().
8301                          */
8302                         wpa_printf(MSG_DEBUG, "nl80211: Skip disabling of "
8303                                    "Probe Request reporting nl_preq=%p while "
8304                                    "in AP mode", bss->nl_preq);
8305                 } else if (bss->nl_preq) {
8306                         wpa_printf(MSG_DEBUG, "nl80211: Disable Probe Request "
8307                                    "reporting nl_preq=%p", bss->nl_preq);
8308                         eloop_unregister_read_sock(
8309                                 nl_socket_get_fd(bss->nl_preq));
8310                         nl_destroy_handles(&bss->nl_preq);
8311                 }
8312                 return 0;
8313         }
8314
8315         if (bss->nl_preq) {
8316                 wpa_printf(MSG_DEBUG, "nl80211: Probe Request reporting "
8317                            "already on! nl_preq=%p", bss->nl_preq);
8318                 return 0;
8319         }
8320
8321         bss->nl_preq = nl_create_handle(drv->global->nl_cb, "preq");
8322         if (bss->nl_preq == NULL)
8323                 return -1;
8324         wpa_printf(MSG_DEBUG, "nl80211: Enable Probe Request "
8325                    "reporting nl_preq=%p", bss->nl_preq);
8326
8327         if (nl80211_register_frame(bss, bss->nl_preq,
8328                                    (WLAN_FC_TYPE_MGMT << 2) |
8329                                    (WLAN_FC_STYPE_PROBE_REQ << 4),
8330                                    NULL, 0) < 0)
8331                 goto out_err;
8332
8333         eloop_register_read_sock(nl_socket_get_fd(bss->nl_preq),
8334                                  wpa_driver_nl80211_event_receive, bss->nl_cb,
8335                                  bss->nl_preq);
8336
8337         return 0;
8338
8339  out_err:
8340         nl_destroy_handles(&bss->nl_preq);
8341         return -1;
8342 }
8343
8344
8345 static int nl80211_disable_11b_rates(struct wpa_driver_nl80211_data *drv,
8346                                      int ifindex, int disabled)
8347 {
8348         struct nl_msg *msg;
8349         struct nlattr *bands, *band;
8350         int ret;
8351
8352         msg = nlmsg_alloc();
8353         if (!msg)
8354                 return -1;
8355
8356         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_TX_BITRATE_MASK);
8357         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
8358
8359         bands = nla_nest_start(msg, NL80211_ATTR_TX_RATES);
8360         if (!bands)
8361                 goto nla_put_failure;
8362
8363         /*
8364          * Disable 2 GHz rates 1, 2, 5.5, 11 Mbps by masking out everything
8365          * else apart from 6, 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS
8366          * rates. All 5 GHz rates are left enabled.
8367          */
8368         band = nla_nest_start(msg, NL80211_BAND_2GHZ);
8369         if (!band)
8370                 goto nla_put_failure;
8371         if (disabled) {
8372                 NLA_PUT(msg, NL80211_TXRATE_LEGACY, 8,
8373                         "\x0c\x12\x18\x24\x30\x48\x60\x6c");
8374         }
8375         nla_nest_end(msg, band);
8376
8377         nla_nest_end(msg, bands);
8378
8379         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8380         msg = NULL;
8381         if (ret) {
8382                 wpa_printf(MSG_DEBUG, "nl80211: Set TX rates failed: ret=%d "
8383                            "(%s)", ret, strerror(-ret));
8384         }
8385
8386         return ret;
8387
8388 nla_put_failure:
8389         nlmsg_free(msg);
8390         return -1;
8391 }
8392
8393
8394 static int wpa_driver_nl80211_deinit_ap(void *priv)
8395 {
8396         struct i802_bss *bss = priv;
8397         struct wpa_driver_nl80211_data *drv = bss->drv;
8398         if (!is_ap_interface(drv->nlmode))
8399                 return -1;
8400         wpa_driver_nl80211_del_beacon(drv);
8401         return wpa_driver_nl80211_set_mode(priv, NL80211_IFTYPE_STATION);
8402 }
8403
8404
8405 static int wpa_driver_nl80211_deinit_p2p_cli(void *priv)
8406 {
8407         struct i802_bss *bss = priv;
8408         struct wpa_driver_nl80211_data *drv = bss->drv;
8409         if (drv->nlmode != NL80211_IFTYPE_P2P_CLIENT)
8410                 return -1;
8411         return wpa_driver_nl80211_set_mode(priv, NL80211_IFTYPE_STATION);
8412 }
8413
8414
8415 static void wpa_driver_nl80211_resume(void *priv)
8416 {
8417         struct i802_bss *bss = priv;
8418         struct wpa_driver_nl80211_data *drv = bss->drv;
8419         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 1)) {
8420                 wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface up on "
8421                            "resume event");
8422         }
8423 }
8424
8425
8426 static int nl80211_send_ft_action(void *priv, u8 action, const u8 *target_ap,
8427                                   const u8 *ies, size_t ies_len)
8428 {
8429         struct i802_bss *bss = priv;
8430         struct wpa_driver_nl80211_data *drv = bss->drv;
8431         int ret;
8432         u8 *data, *pos;
8433         size_t data_len;
8434         const u8 *own_addr = bss->addr;
8435
8436         if (action != 1) {
8437                 wpa_printf(MSG_ERROR, "nl80211: Unsupported send_ft_action "
8438                            "action %d", action);
8439                 return -1;
8440         }
8441
8442         /*
8443          * Action frame payload:
8444          * Category[1] = 6 (Fast BSS Transition)
8445          * Action[1] = 1 (Fast BSS Transition Request)
8446          * STA Address
8447          * Target AP Address
8448          * FT IEs
8449          */
8450
8451         data_len = 2 + 2 * ETH_ALEN + ies_len;
8452         data = os_malloc(data_len);
8453         if (data == NULL)
8454                 return -1;
8455         pos = data;
8456         *pos++ = 0x06; /* FT Action category */
8457         *pos++ = action;
8458         os_memcpy(pos, own_addr, ETH_ALEN);
8459         pos += ETH_ALEN;
8460         os_memcpy(pos, target_ap, ETH_ALEN);
8461         pos += ETH_ALEN;
8462         os_memcpy(pos, ies, ies_len);
8463
8464         ret = wpa_driver_nl80211_send_action(bss, drv->assoc_freq, 0,
8465                                              drv->bssid, own_addr, drv->bssid,
8466                                              data, data_len, 0);
8467         os_free(data);
8468
8469         return ret;
8470 }
8471
8472
8473 static int nl80211_signal_monitor(void *priv, int threshold, int hysteresis)
8474 {
8475         struct i802_bss *bss = priv;
8476         struct wpa_driver_nl80211_data *drv = bss->drv;
8477         struct nl_msg *msg, *cqm = NULL;
8478         int ret = -1;
8479
8480         wpa_printf(MSG_DEBUG, "nl80211: Signal monitor threshold=%d "
8481                    "hysteresis=%d", threshold, hysteresis);
8482
8483         msg = nlmsg_alloc();
8484         if (!msg)
8485                 return -1;
8486
8487         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_CQM);
8488
8489         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8490
8491         cqm = nlmsg_alloc();
8492         if (cqm == NULL)
8493                 goto nla_put_failure;
8494
8495         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_THOLD, threshold);
8496         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_HYST, hysteresis);
8497         if (nla_put_nested(msg, NL80211_ATTR_CQM, cqm) < 0)
8498                 goto nla_put_failure;
8499
8500         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8501         msg = NULL;
8502
8503 nla_put_failure:
8504         nlmsg_free(cqm);
8505         nlmsg_free(msg);
8506         return ret;
8507 }
8508
8509
8510 static int nl80211_signal_poll(void *priv, struct wpa_signal_info *si)
8511 {
8512         struct i802_bss *bss = priv;
8513         struct wpa_driver_nl80211_data *drv = bss->drv;
8514         int res;
8515
8516         os_memset(si, 0, sizeof(*si));
8517         res = nl80211_get_link_signal(drv, si);
8518         if (res != 0)
8519                 return res;
8520
8521         return nl80211_get_link_noise(drv, si);
8522 }
8523
8524
8525 static int wpa_driver_nl80211_shared_freq(void *priv)
8526 {
8527         struct i802_bss *bss = priv;
8528         struct wpa_driver_nl80211_data *drv = bss->drv;
8529         struct wpa_driver_nl80211_data *driver;
8530         int freq = 0;
8531
8532         /*
8533          * If the same PHY is in connected state with some other interface,
8534          * then retrieve the assoc freq.
8535          */
8536         wpa_printf(MSG_DEBUG, "nl80211: Get shared freq for PHY %s",
8537                    drv->phyname);
8538
8539         dl_list_for_each(driver, &drv->global->interfaces,
8540                          struct wpa_driver_nl80211_data, list) {
8541                 if (drv == driver ||
8542                     os_strcmp(drv->phyname, driver->phyname) != 0 ||
8543                     !driver->associated)
8544                         continue;
8545
8546                 wpa_printf(MSG_DEBUG, "nl80211: Found a match for PHY %s - %s "
8547                            MACSTR,
8548                            driver->phyname, driver->first_bss.ifname,
8549                            MAC2STR(driver->first_bss.addr));
8550                 if (is_ap_interface(driver->nlmode))
8551                         freq = driver->first_bss.freq;
8552                 else
8553                         freq = nl80211_get_assoc_freq(driver);
8554                 wpa_printf(MSG_DEBUG, "nl80211: Shared freq for PHY %s: %d",
8555                            drv->phyname, freq);
8556         }
8557
8558         if (!freq)
8559                 wpa_printf(MSG_DEBUG, "nl80211: No shared interface for "
8560                            "PHY (%s) in associated state", drv->phyname);
8561
8562         return freq;
8563 }
8564
8565
8566 static int nl80211_send_frame(void *priv, const u8 *data, size_t data_len,
8567                               int encrypt)
8568 {
8569         struct i802_bss *bss = priv;
8570         return wpa_driver_nl80211_send_frame(bss, data, data_len, encrypt, 0,
8571                                              0, 0, 0, 0);
8572 }
8573
8574
8575 static int nl80211_set_param(void *priv, const char *param)
8576 {
8577         wpa_printf(MSG_DEBUG, "nl80211: driver param='%s'", param);
8578         if (param == NULL)
8579                 return 0;
8580
8581 #ifdef CONFIG_P2P
8582         if (os_strstr(param, "use_p2p_group_interface=1")) {
8583                 struct i802_bss *bss = priv;
8584                 struct wpa_driver_nl80211_data *drv = bss->drv;
8585
8586                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
8587                            "interface");
8588                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
8589                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
8590         }
8591 #endif /* CONFIG_P2P */
8592
8593         return 0;
8594 }
8595
8596
8597 static void * nl80211_global_init(void)
8598 {
8599         struct nl80211_global *global;
8600         struct netlink_config *cfg;
8601
8602         global = os_zalloc(sizeof(*global));
8603         if (global == NULL)
8604                 return NULL;
8605         global->ioctl_sock = -1;
8606         dl_list_init(&global->interfaces);
8607         global->if_add_ifindex = -1;
8608
8609         cfg = os_zalloc(sizeof(*cfg));
8610         if (cfg == NULL)
8611                 goto err;
8612
8613         cfg->ctx = global;
8614         cfg->newlink_cb = wpa_driver_nl80211_event_rtm_newlink;
8615         cfg->dellink_cb = wpa_driver_nl80211_event_rtm_dellink;
8616         global->netlink = netlink_init(cfg);
8617         if (global->netlink == NULL) {
8618                 os_free(cfg);
8619                 goto err;
8620         }
8621
8622         if (wpa_driver_nl80211_init_nl_global(global) < 0)
8623                 goto err;
8624
8625         global->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
8626         if (global->ioctl_sock < 0) {
8627                 perror("socket(PF_INET,SOCK_DGRAM)");
8628                 goto err;
8629         }
8630
8631         return global;
8632
8633 err:
8634         nl80211_global_deinit(global);
8635         return NULL;
8636 }
8637
8638
8639 static void nl80211_global_deinit(void *priv)
8640 {
8641         struct nl80211_global *global = priv;
8642         if (global == NULL)
8643                 return;
8644         if (!dl_list_empty(&global->interfaces)) {
8645                 wpa_printf(MSG_ERROR, "nl80211: %u interface(s) remain at "
8646                            "nl80211_global_deinit",
8647                            dl_list_len(&global->interfaces));
8648         }
8649
8650         if (global->netlink)
8651                 netlink_deinit(global->netlink);
8652
8653         nl_destroy_handles(&global->nl);
8654
8655         if (global->nl_event) {
8656                 eloop_unregister_read_sock(
8657                         nl_socket_get_fd(global->nl_event));
8658                 nl_destroy_handles(&global->nl_event);
8659         }
8660
8661         nl_cb_put(global->nl_cb);
8662
8663         if (global->ioctl_sock >= 0)
8664                 close(global->ioctl_sock);
8665
8666         os_free(global);
8667 }
8668
8669
8670 static const char * nl80211_get_radio_name(void *priv)
8671 {
8672         struct i802_bss *bss = priv;
8673         struct wpa_driver_nl80211_data *drv = bss->drv;
8674         return drv->phyname;
8675 }
8676
8677
8678 static int nl80211_pmkid(struct i802_bss *bss, int cmd, const u8 *bssid,
8679                          const u8 *pmkid)
8680 {
8681         struct nl_msg *msg;
8682
8683         msg = nlmsg_alloc();
8684         if (!msg)
8685                 return -ENOMEM;
8686
8687         nl80211_cmd(bss->drv, msg, 0, cmd);
8688
8689         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
8690         if (pmkid)
8691                 NLA_PUT(msg, NL80211_ATTR_PMKID, 16, pmkid);
8692         if (bssid)
8693                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid);
8694
8695         return send_and_recv_msgs(bss->drv, msg, NULL, NULL);
8696  nla_put_failure:
8697         nlmsg_free(msg);
8698         return -ENOBUFS;
8699 }
8700
8701
8702 static int nl80211_add_pmkid(void *priv, const u8 *bssid, const u8 *pmkid)
8703 {
8704         struct i802_bss *bss = priv;
8705         wpa_printf(MSG_DEBUG, "nl80211: Add PMKID for " MACSTR, MAC2STR(bssid));
8706         return nl80211_pmkid(bss, NL80211_CMD_SET_PMKSA, bssid, pmkid);
8707 }
8708
8709
8710 static int nl80211_remove_pmkid(void *priv, const u8 *bssid, const u8 *pmkid)
8711 {
8712         struct i802_bss *bss = priv;
8713         wpa_printf(MSG_DEBUG, "nl80211: Delete PMKID for " MACSTR,
8714                    MAC2STR(bssid));
8715         return nl80211_pmkid(bss, NL80211_CMD_DEL_PMKSA, bssid, pmkid);
8716 }
8717
8718
8719 static int nl80211_flush_pmkid(void *priv)
8720 {
8721         struct i802_bss *bss = priv;
8722         wpa_printf(MSG_DEBUG, "nl80211: Flush PMKIDs");
8723         return nl80211_pmkid(bss, NL80211_CMD_FLUSH_PMKSA, NULL, NULL);
8724 }
8725
8726
8727 static void nl80211_set_rekey_info(void *priv, const u8 *kek, const u8 *kck,
8728                                    const u8 *replay_ctr)
8729 {
8730         struct i802_bss *bss = priv;
8731         struct wpa_driver_nl80211_data *drv = bss->drv;
8732         struct nlattr *replay_nested;
8733         struct nl_msg *msg;
8734
8735         msg = nlmsg_alloc();
8736         if (!msg)
8737                 return;
8738
8739         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_REKEY_OFFLOAD);
8740
8741         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8742
8743         replay_nested = nla_nest_start(msg, NL80211_ATTR_REKEY_DATA);
8744         if (!replay_nested)
8745                 goto nla_put_failure;
8746
8747         NLA_PUT(msg, NL80211_REKEY_DATA_KEK, NL80211_KEK_LEN, kek);
8748         NLA_PUT(msg, NL80211_REKEY_DATA_KCK, NL80211_KCK_LEN, kck);
8749         NLA_PUT(msg, NL80211_REKEY_DATA_REPLAY_CTR, NL80211_REPLAY_CTR_LEN,
8750                 replay_ctr);
8751
8752         nla_nest_end(msg, replay_nested);
8753
8754         send_and_recv_msgs(drv, msg, NULL, NULL);
8755         return;
8756  nla_put_failure:
8757         nlmsg_free(msg);
8758 }
8759
8760
8761 static void nl80211_send_null_frame(struct i802_bss *bss, const u8 *own_addr,
8762                                     const u8 *addr, int qos)
8763 {
8764         /* send data frame to poll STA and check whether
8765          * this frame is ACKed */
8766         struct {
8767                 struct ieee80211_hdr hdr;
8768                 u16 qos_ctl;
8769         } STRUCT_PACKED nulldata;
8770         size_t size;
8771
8772         /* Send data frame to poll STA and check whether this frame is ACKed */
8773
8774         os_memset(&nulldata, 0, sizeof(nulldata));
8775
8776         if (qos) {
8777                 nulldata.hdr.frame_control =
8778                         IEEE80211_FC(WLAN_FC_TYPE_DATA,
8779                                      WLAN_FC_STYPE_QOS_NULL);
8780                 size = sizeof(nulldata);
8781         } else {
8782                 nulldata.hdr.frame_control =
8783                         IEEE80211_FC(WLAN_FC_TYPE_DATA,
8784                                      WLAN_FC_STYPE_NULLFUNC);
8785                 size = sizeof(struct ieee80211_hdr);
8786         }
8787
8788         nulldata.hdr.frame_control |= host_to_le16(WLAN_FC_FROMDS);
8789         os_memcpy(nulldata.hdr.IEEE80211_DA_FROMDS, addr, ETH_ALEN);
8790         os_memcpy(nulldata.hdr.IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
8791         os_memcpy(nulldata.hdr.IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
8792
8793         if (wpa_driver_nl80211_send_mlme(bss, (u8 *) &nulldata, size, 0) < 0)
8794                 wpa_printf(MSG_DEBUG, "nl80211_send_null_frame: Failed to "
8795                            "send poll frame");
8796 }
8797
8798 static void nl80211_poll_client(void *priv, const u8 *own_addr, const u8 *addr,
8799                                 int qos)
8800 {
8801         struct i802_bss *bss = priv;
8802         struct wpa_driver_nl80211_data *drv = bss->drv;
8803         struct nl_msg *msg;
8804
8805         if (!drv->poll_command_supported) {
8806                 nl80211_send_null_frame(bss, own_addr, addr, qos);
8807                 return;
8808         }
8809
8810         msg = nlmsg_alloc();
8811         if (!msg)
8812                 return;
8813
8814         nl80211_cmd(drv, msg, 0, NL80211_CMD_PROBE_CLIENT);
8815
8816         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8817         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
8818
8819         send_and_recv_msgs(drv, msg, NULL, NULL);
8820         return;
8821  nla_put_failure:
8822         nlmsg_free(msg);
8823 }
8824
8825
8826 static int nl80211_set_power_save(struct i802_bss *bss, int enabled)
8827 {
8828         struct nl_msg *msg;
8829
8830         msg = nlmsg_alloc();
8831         if (!msg)
8832                 return -ENOMEM;
8833
8834         nl80211_cmd(bss->drv, msg, 0, NL80211_CMD_SET_POWER_SAVE);
8835         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8836         NLA_PUT_U32(msg, NL80211_ATTR_PS_STATE,
8837                     enabled ? NL80211_PS_ENABLED : NL80211_PS_DISABLED);
8838         return send_and_recv_msgs(bss->drv, msg, NULL, NULL);
8839 nla_put_failure:
8840         nlmsg_free(msg);
8841         return -ENOBUFS;
8842 }
8843
8844
8845 static int nl80211_set_p2p_powersave(void *priv, int legacy_ps, int opp_ps,
8846                                      int ctwindow)
8847 {
8848         struct i802_bss *bss = priv;
8849
8850         wpa_printf(MSG_DEBUG, "nl80211: set_p2p_powersave (legacy_ps=%d "
8851                    "opp_ps=%d ctwindow=%d)", legacy_ps, opp_ps, ctwindow);
8852
8853         if (opp_ps != -1 || ctwindow != -1)
8854                 return -1; /* Not yet supported */
8855
8856         if (legacy_ps == -1)
8857                 return 0;
8858         if (legacy_ps != 0 && legacy_ps != 1)
8859                 return -1; /* Not yet supported */
8860
8861         return nl80211_set_power_save(bss, legacy_ps);
8862 }
8863
8864
8865 #ifdef CONFIG_TDLS
8866
8867 static int nl80211_send_tdls_mgmt(void *priv, const u8 *dst, u8 action_code,
8868                                   u8 dialog_token, u16 status_code,
8869                                   const u8 *buf, size_t len)
8870 {
8871         struct i802_bss *bss = priv;
8872         struct wpa_driver_nl80211_data *drv = bss->drv;
8873         struct nl_msg *msg;
8874
8875         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
8876                 return -EOPNOTSUPP;
8877
8878         if (!dst)
8879                 return -EINVAL;
8880
8881         msg = nlmsg_alloc();
8882         if (!msg)
8883                 return -ENOMEM;
8884
8885         nl80211_cmd(drv, msg, 0, NL80211_CMD_TDLS_MGMT);
8886         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8887         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
8888         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_ACTION, action_code);
8889         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_DIALOG_TOKEN, dialog_token);
8890         NLA_PUT_U16(msg, NL80211_ATTR_STATUS_CODE, status_code);
8891         NLA_PUT(msg, NL80211_ATTR_IE, len, buf);
8892
8893         return send_and_recv_msgs(drv, msg, NULL, NULL);
8894
8895 nla_put_failure:
8896         nlmsg_free(msg);
8897         return -ENOBUFS;
8898 }
8899
8900
8901 static int nl80211_tdls_oper(void *priv, enum tdls_oper oper, const u8 *peer)
8902 {
8903         struct i802_bss *bss = priv;
8904         struct wpa_driver_nl80211_data *drv = bss->drv;
8905         struct nl_msg *msg;
8906         enum nl80211_tdls_operation nl80211_oper;
8907
8908         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
8909                 return -EOPNOTSUPP;
8910
8911         switch (oper) {
8912         case TDLS_DISCOVERY_REQ:
8913                 nl80211_oper = NL80211_TDLS_DISCOVERY_REQ;
8914                 break;
8915         case TDLS_SETUP:
8916                 nl80211_oper = NL80211_TDLS_SETUP;
8917                 break;
8918         case TDLS_TEARDOWN:
8919                 nl80211_oper = NL80211_TDLS_TEARDOWN;
8920                 break;
8921         case TDLS_ENABLE_LINK:
8922                 nl80211_oper = NL80211_TDLS_ENABLE_LINK;
8923                 break;
8924         case TDLS_DISABLE_LINK:
8925                 nl80211_oper = NL80211_TDLS_DISABLE_LINK;
8926                 break;
8927         case TDLS_ENABLE:
8928                 return 0;
8929         case TDLS_DISABLE:
8930                 return 0;
8931         default:
8932                 return -EINVAL;
8933         }
8934
8935         msg = nlmsg_alloc();
8936         if (!msg)
8937                 return -ENOMEM;
8938
8939         nl80211_cmd(drv, msg, 0, NL80211_CMD_TDLS_OPER);
8940         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_OPERATION, nl80211_oper);
8941         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8942         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, peer);
8943
8944         return send_and_recv_msgs(drv, msg, NULL, NULL);
8945
8946 nla_put_failure:
8947         nlmsg_free(msg);
8948         return -ENOBUFS;
8949 }
8950
8951 #endif /* CONFIG TDLS */
8952
8953
8954 #ifdef ANDROID
8955
8956 typedef struct android_wifi_priv_cmd {
8957         char *buf;
8958         int used_len;
8959         int total_len;
8960 } android_wifi_priv_cmd;
8961
8962 static int drv_errors = 0;
8963
8964 static void wpa_driver_send_hang_msg(struct wpa_driver_nl80211_data *drv)
8965 {
8966         drv_errors++;
8967         if (drv_errors > DRV_NUMBER_SEQUENTIAL_ERRORS) {
8968                 drv_errors = 0;
8969                 wpa_msg(drv->ctx, MSG_INFO, WPA_EVENT_DRIVER_STATE "HANGED");
8970         }
8971 }
8972
8973
8974 static int android_priv_cmd(struct i802_bss *bss, const char *cmd)
8975 {
8976         struct wpa_driver_nl80211_data *drv = bss->drv;
8977         struct ifreq ifr;
8978         android_wifi_priv_cmd priv_cmd;
8979         char buf[MAX_DRV_CMD_SIZE];
8980         int ret;
8981
8982         os_memset(&ifr, 0, sizeof(ifr));
8983         os_memset(&priv_cmd, 0, sizeof(priv_cmd));
8984         os_strlcpy(ifr.ifr_name, bss->ifname, IFNAMSIZ);
8985
8986         os_memset(buf, 0, sizeof(buf));
8987         os_strlcpy(buf, cmd, sizeof(buf));
8988
8989         priv_cmd.buf = buf;
8990         priv_cmd.used_len = sizeof(buf);
8991         priv_cmd.total_len = sizeof(buf);
8992         ifr.ifr_data = &priv_cmd;
8993
8994         ret = ioctl(drv->global->ioctl_sock, SIOCDEVPRIVATE + 1, &ifr);
8995         if (ret < 0) {
8996                 wpa_printf(MSG_ERROR, "%s: failed to issue private commands",
8997                            __func__);
8998                 wpa_driver_send_hang_msg(drv);
8999                 return ret;
9000         }
9001
9002         drv_errors = 0;
9003         return 0;
9004 }
9005
9006
9007 static int android_pno_start(struct i802_bss *bss,
9008                              struct wpa_driver_scan_params *params)
9009 {
9010         struct wpa_driver_nl80211_data *drv = bss->drv;
9011         struct ifreq ifr;
9012         android_wifi_priv_cmd priv_cmd;
9013         int ret = 0, i = 0, bp;
9014         char buf[WEXT_PNO_MAX_COMMAND_SIZE];
9015
9016         bp = WEXT_PNOSETUP_HEADER_SIZE;
9017         os_memcpy(buf, WEXT_PNOSETUP_HEADER, bp);
9018         buf[bp++] = WEXT_PNO_TLV_PREFIX;
9019         buf[bp++] = WEXT_PNO_TLV_VERSION;
9020         buf[bp++] = WEXT_PNO_TLV_SUBVERSION;
9021         buf[bp++] = WEXT_PNO_TLV_RESERVED;
9022
9023         while (i < WEXT_PNO_AMOUNT && (size_t) i < params->num_ssids) {
9024                 /* Check that there is enough space needed for 1 more SSID, the
9025                  * other sections and null termination */
9026                 if ((bp + WEXT_PNO_SSID_HEADER_SIZE + MAX_SSID_LEN +
9027                      WEXT_PNO_NONSSID_SECTIONS_SIZE + 1) >= (int) sizeof(buf))
9028                         break;
9029                 wpa_hexdump_ascii(MSG_DEBUG, "For PNO Scan",
9030                                   params->ssids[i].ssid,
9031                                   params->ssids[i].ssid_len);
9032                 buf[bp++] = WEXT_PNO_SSID_SECTION;
9033                 buf[bp++] = params->ssids[i].ssid_len;
9034                 os_memcpy(&buf[bp], params->ssids[i].ssid,
9035                           params->ssids[i].ssid_len);
9036                 bp += params->ssids[i].ssid_len;
9037                 i++;
9038         }
9039
9040         buf[bp++] = WEXT_PNO_SCAN_INTERVAL_SECTION;
9041         os_snprintf(&buf[bp], WEXT_PNO_SCAN_INTERVAL_LENGTH + 1, "%x",
9042                     WEXT_PNO_SCAN_INTERVAL);
9043         bp += WEXT_PNO_SCAN_INTERVAL_LENGTH;
9044
9045         buf[bp++] = WEXT_PNO_REPEAT_SECTION;
9046         os_snprintf(&buf[bp], WEXT_PNO_REPEAT_LENGTH + 1, "%x",
9047                     WEXT_PNO_REPEAT);
9048         bp += WEXT_PNO_REPEAT_LENGTH;
9049
9050         buf[bp++] = WEXT_PNO_MAX_REPEAT_SECTION;
9051         os_snprintf(&buf[bp], WEXT_PNO_MAX_REPEAT_LENGTH + 1, "%x",
9052                     WEXT_PNO_MAX_REPEAT);
9053         bp += WEXT_PNO_MAX_REPEAT_LENGTH + 1;
9054
9055         memset(&ifr, 0, sizeof(ifr));
9056         memset(&priv_cmd, 0, sizeof(priv_cmd));
9057         os_strncpy(ifr.ifr_name, bss->ifname, IFNAMSIZ);
9058
9059         priv_cmd.buf = buf;
9060         priv_cmd.used_len = bp;
9061         priv_cmd.total_len = bp;
9062         ifr.ifr_data = &priv_cmd;
9063
9064         ret = ioctl(drv->global->ioctl_sock, SIOCDEVPRIVATE + 1, &ifr);
9065
9066         if (ret < 0) {
9067                 wpa_printf(MSG_ERROR, "ioctl[SIOCSIWPRIV] (pnosetup): %d",
9068                            ret);
9069                 wpa_driver_send_hang_msg(drv);
9070                 return ret;
9071         }
9072
9073         drv_errors = 0;
9074
9075         return android_priv_cmd(bss, "PNOFORCE 1");
9076 }
9077
9078
9079 static int android_pno_stop(struct i802_bss *bss)
9080 {
9081         return android_priv_cmd(bss, "PNOFORCE 0");
9082 }
9083
9084 #endif /* ANDROID */
9085
9086
9087 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
9088         .name = "nl80211",
9089         .desc = "Linux nl80211/cfg80211",
9090         .get_bssid = wpa_driver_nl80211_get_bssid,
9091         .get_ssid = wpa_driver_nl80211_get_ssid,
9092         .set_key = wpa_driver_nl80211_set_key,
9093         .scan2 = wpa_driver_nl80211_scan,
9094         .sched_scan = wpa_driver_nl80211_sched_scan,
9095         .stop_sched_scan = wpa_driver_nl80211_stop_sched_scan,
9096         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
9097         .deauthenticate = wpa_driver_nl80211_deauthenticate,
9098         .disassociate = wpa_driver_nl80211_disassociate,
9099         .authenticate = wpa_driver_nl80211_authenticate,
9100         .associate = wpa_driver_nl80211_associate,
9101         .global_init = nl80211_global_init,
9102         .global_deinit = nl80211_global_deinit,
9103         .init2 = wpa_driver_nl80211_init,
9104         .deinit = wpa_driver_nl80211_deinit,
9105         .get_capa = wpa_driver_nl80211_get_capa,
9106         .set_operstate = wpa_driver_nl80211_set_operstate,
9107         .set_supp_port = wpa_driver_nl80211_set_supp_port,
9108         .set_country = wpa_driver_nl80211_set_country,
9109         .set_ap = wpa_driver_nl80211_set_ap,
9110         .if_add = wpa_driver_nl80211_if_add,
9111         .if_remove = wpa_driver_nl80211_if_remove,
9112         .send_mlme = wpa_driver_nl80211_send_mlme,
9113         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
9114         .sta_add = wpa_driver_nl80211_sta_add,
9115         .sta_remove = wpa_driver_nl80211_sta_remove,
9116         .hapd_send_eapol = wpa_driver_nl80211_hapd_send_eapol,
9117         .sta_set_flags = wpa_driver_nl80211_sta_set_flags,
9118 #ifdef HOSTAPD
9119         .hapd_init = i802_init,
9120         .hapd_deinit = i802_deinit,
9121         .set_wds_sta = i802_set_wds_sta,
9122 #endif /* HOSTAPD */
9123 #if defined(HOSTAPD) || defined(CONFIG_AP)
9124         .get_seqnum = i802_get_seqnum,
9125         .flush = i802_flush,
9126         .read_sta_data = i802_read_sta_data,
9127         .get_inact_sec = i802_get_inact_sec,
9128         .sta_clear_stats = i802_sta_clear_stats,
9129         .set_rts = i802_set_rts,
9130         .set_frag = i802_set_frag,
9131         .set_tx_queue_params = i802_set_tx_queue_params,
9132         .set_sta_vlan = i802_set_sta_vlan,
9133         .sta_deauth = i802_sta_deauth,
9134         .sta_disassoc = i802_sta_disassoc,
9135 #endif /* HOSTAPD || CONFIG_AP */
9136         .set_freq = i802_set_freq,
9137         .send_action = wpa_driver_nl80211_send_action,
9138         .send_action_cancel_wait = wpa_driver_nl80211_send_action_cancel_wait,
9139         .remain_on_channel = wpa_driver_nl80211_remain_on_channel,
9140         .cancel_remain_on_channel =
9141         wpa_driver_nl80211_cancel_remain_on_channel,
9142         .probe_req_report = wpa_driver_nl80211_probe_req_report,
9143         .deinit_ap = wpa_driver_nl80211_deinit_ap,
9144         .deinit_p2p_cli = wpa_driver_nl80211_deinit_p2p_cli,
9145         .resume = wpa_driver_nl80211_resume,
9146         .send_ft_action = nl80211_send_ft_action,
9147         .signal_monitor = nl80211_signal_monitor,
9148         .signal_poll = nl80211_signal_poll,
9149         .send_frame = nl80211_send_frame,
9150         .shared_freq = wpa_driver_nl80211_shared_freq,
9151         .set_param = nl80211_set_param,
9152         .get_radio_name = nl80211_get_radio_name,
9153         .add_pmkid = nl80211_add_pmkid,
9154         .remove_pmkid = nl80211_remove_pmkid,
9155         .flush_pmkid = nl80211_flush_pmkid,
9156         .set_rekey_info = nl80211_set_rekey_info,
9157         .poll_client = nl80211_poll_client,
9158         .set_p2p_powersave = nl80211_set_p2p_powersave,
9159 #ifdef CONFIG_TDLS
9160         .send_tdls_mgmt = nl80211_send_tdls_mgmt,
9161         .tdls_oper = nl80211_tdls_oper,
9162 #endif /* CONFIG_TDLS */
9163 };