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 = 0, timeout;
3483         struct nl_msg *msg, *ssids, *freqs, *rates;
3484         size_t i;
3485
3486         drv->scan_for_auth = 0;
3487
3488         msg = nlmsg_alloc();
3489         ssids = nlmsg_alloc();
3490         freqs = nlmsg_alloc();
3491         rates = nlmsg_alloc();
3492         if (!msg || !ssids || !freqs || !rates) {
3493                 nlmsg_free(msg);
3494                 nlmsg_free(ssids);
3495                 nlmsg_free(freqs);
3496                 nlmsg_free(rates);
3497                 return -1;
3498         }
3499
3500         os_free(drv->filter_ssids);
3501         drv->filter_ssids = params->filter_ssids;
3502         params->filter_ssids = NULL;
3503         drv->num_filter_ssids = params->num_filter_ssids;
3504
3505         nl80211_cmd(drv, msg, 0, NL80211_CMD_TRIGGER_SCAN);
3506
3507         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3508
3509         for (i = 0; i < params->num_ssids; i++) {
3510                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Scan SSID",
3511                                   params->ssids[i].ssid,
3512                                   params->ssids[i].ssid_len);
3513                 NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
3514                         params->ssids[i].ssid);
3515         }
3516         if (params->num_ssids)
3517                 nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids);
3518
3519         if (params->extra_ies) {
3520                 wpa_hexdump(MSG_MSGDUMP, "nl80211: Scan extra IEs",
3521                             params->extra_ies, params->extra_ies_len);
3522                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
3523                         params->extra_ies);
3524         }
3525
3526         if (params->freqs) {
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                 nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs);
3533         }
3534
3535         if (params->p2p_probe) {
3536                 wpa_printf(MSG_DEBUG, "nl80211: P2P probe - mask SuppRates");
3537
3538                 /*
3539                  * Remove 2.4 GHz rates 1, 2, 5.5, 11 Mbps from supported rates
3540                  * by masking out everything else apart from the OFDM rates 6,
3541                  * 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS rates. All 5 GHz
3542                  * rates are left enabled.
3543                  */
3544                 NLA_PUT(rates, NL80211_BAND_2GHZ, 8,
3545                         "\x0c\x12\x18\x24\x30\x48\x60\x6c");
3546                 nla_put_nested(msg, NL80211_ATTR_SCAN_SUPP_RATES, rates);
3547
3548                 NLA_PUT_FLAG(msg, NL80211_ATTR_TX_NO_CCK_RATE);
3549         }
3550
3551         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3552         msg = NULL;
3553         if (ret) {
3554                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
3555                            "(%s)", ret, strerror(-ret));
3556 #ifdef HOSTAPD
3557                 if (is_ap_interface(drv->nlmode)) {
3558                         /*
3559                          * mac80211 does not allow scan requests in AP mode, so
3560                          * try to do this in station mode.
3561                          */
3562                         if (wpa_driver_nl80211_set_mode(
3563                                     bss, NL80211_IFTYPE_STATION))
3564                                 goto nla_put_failure;
3565
3566                         if (wpa_driver_nl80211_scan(drv, params)) {
3567                                 wpa_driver_nl80211_set_mode(bss, drv->nlmode);
3568                                 goto nla_put_failure;
3569                         }
3570
3571                         /* Restore AP mode when processing scan results */
3572                         drv->ap_scan_as_station = drv->nlmode;
3573                         ret = 0;
3574                 } else
3575                         goto nla_put_failure;
3576 #else /* HOSTAPD */
3577                 goto nla_put_failure;
3578 #endif /* HOSTAPD */
3579         }
3580
3581         /* Not all drivers generate "scan completed" wireless event, so try to
3582          * read results after a timeout. */
3583         timeout = 10;
3584         if (drv->scan_complete_events) {
3585                 /*
3586                  * The driver seems to deliver events to notify when scan is
3587                  * complete, so use longer timeout to avoid race conditions
3588                  * with scanning and following association request.
3589                  */
3590                 timeout = 30;
3591         }
3592         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
3593                    "seconds", ret, timeout);
3594         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
3595         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
3596                                drv, drv->ctx);
3597
3598 nla_put_failure:
3599         nlmsg_free(ssids);
3600         nlmsg_free(msg);
3601         nlmsg_free(freqs);
3602         nlmsg_free(rates);
3603         return ret;
3604 }
3605
3606
3607 /**
3608  * wpa_driver_nl80211_sched_scan - Initiate a scheduled scan
3609  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
3610  * @params: Scan parameters
3611  * @interval: Interval between scan cycles in milliseconds
3612  * Returns: 0 on success, -1 on failure or if not supported
3613  */
3614 static int wpa_driver_nl80211_sched_scan(void *priv,
3615                                          struct wpa_driver_scan_params *params,
3616                                          u32 interval)
3617 {
3618         struct i802_bss *bss = priv;
3619         struct wpa_driver_nl80211_data *drv = bss->drv;
3620         int ret = 0;
3621         struct nl_msg *msg, *ssids, *freqs, *match_set_ssid, *match_sets;
3622         struct nl_msg *match_set_rssi;
3623         size_t i;
3624
3625 #ifdef ANDROID
3626         if (!drv->capa.sched_scan_supported)
3627                 return android_pno_start(bss, params);
3628 #endif /* ANDROID */
3629
3630         msg = nlmsg_alloc();
3631         ssids = nlmsg_alloc();
3632         freqs = nlmsg_alloc();
3633         if (!msg || !ssids || !freqs) {
3634                 nlmsg_free(msg);
3635                 nlmsg_free(ssids);
3636                 nlmsg_free(freqs);
3637                 return -1;
3638         }
3639
3640         os_free(drv->filter_ssids);
3641         drv->filter_ssids = params->filter_ssids;
3642         params->filter_ssids = NULL;
3643         drv->num_filter_ssids = params->num_filter_ssids;
3644
3645         nl80211_cmd(drv, msg, 0, NL80211_CMD_START_SCHED_SCAN);
3646
3647         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3648
3649         NLA_PUT_U32(msg, NL80211_ATTR_SCHED_SCAN_INTERVAL, interval);
3650
3651         if ((drv->num_filter_ssids &&
3652             (int) drv->num_filter_ssids <= drv->capa.max_match_sets) ||
3653             params->filter_rssi) {
3654                 match_sets = nlmsg_alloc();
3655
3656                 for (i = 0; i < drv->num_filter_ssids; i++) {
3657                         wpa_hexdump_ascii(MSG_MSGDUMP,
3658                                           "nl80211: Sched scan filter SSID",
3659                                           drv->filter_ssids[i].ssid,
3660                                           drv->filter_ssids[i].ssid_len);
3661
3662                         match_set_ssid = nlmsg_alloc();
3663                         nla_put(match_set_ssid,
3664                                 NL80211_ATTR_SCHED_SCAN_MATCH_SSID,
3665                                 drv->filter_ssids[i].ssid_len,
3666                                 drv->filter_ssids[i].ssid);
3667
3668                         nla_put_nested(match_sets, i + 1, match_set_ssid);
3669
3670                         nlmsg_free(match_set_ssid);
3671                 }
3672
3673                 if (params->filter_rssi) {
3674                         match_set_rssi = nlmsg_alloc();
3675                         NLA_PUT_U32(match_set_rssi,
3676                                     NL80211_SCHED_SCAN_MATCH_ATTR_RSSI,
3677                                     params->filter_rssi);
3678                         wpa_printf(MSG_MSGDUMP,
3679                                    "nl80211: Sched scan RSSI filter %d dBm",
3680                                    params->filter_rssi);
3681                         nla_put_nested(match_sets, 0, match_set_rssi);
3682                         nlmsg_free(match_set_rssi);
3683                 }
3684
3685                 nla_put_nested(msg, NL80211_ATTR_SCHED_SCAN_MATCH,
3686                                match_sets);
3687                 nlmsg_free(match_sets);
3688         }
3689
3690         for (i = 0; i < params->num_ssids; i++) {
3691                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Sched scan SSID",
3692                                   params->ssids[i].ssid,
3693                                   params->ssids[i].ssid_len);
3694                 NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
3695                         params->ssids[i].ssid);
3696         }
3697         if (params->num_ssids)
3698                 nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids);
3699
3700         if (params->extra_ies) {
3701                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Sched scan extra IEs",
3702                                   params->extra_ies, params->extra_ies_len);
3703                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
3704                         params->extra_ies);
3705         }
3706
3707         if (params->freqs) {
3708                 for (i = 0; params->freqs[i]; i++) {
3709                         wpa_printf(MSG_MSGDUMP, "nl80211: Scan frequency %u "
3710                                    "MHz", params->freqs[i]);
3711                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
3712                 }
3713                 nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs);
3714         }
3715
3716         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3717
3718         /* TODO: if we get an error here, we should fall back to normal scan */
3719
3720         msg = NULL;
3721         if (ret) {
3722                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan start failed: "
3723                            "ret=%d (%s)", ret, strerror(-ret));
3724                 goto nla_put_failure;
3725         }
3726
3727         wpa_printf(MSG_DEBUG, "nl80211: Sched scan requested (ret=%d) - "
3728                    "scan interval %d msec", ret, interval);
3729
3730 nla_put_failure:
3731         nlmsg_free(ssids);
3732         nlmsg_free(msg);
3733         nlmsg_free(freqs);
3734         return ret;
3735 }
3736
3737
3738 /**
3739  * wpa_driver_nl80211_stop_sched_scan - Stop a scheduled scan
3740  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
3741  * Returns: 0 on success, -1 on failure or if not supported
3742  */
3743 static int wpa_driver_nl80211_stop_sched_scan(void *priv)
3744 {
3745         struct i802_bss *bss = priv;
3746         struct wpa_driver_nl80211_data *drv = bss->drv;
3747         int ret = 0;
3748         struct nl_msg *msg;
3749
3750 #ifdef ANDROID
3751         if (!drv->capa.sched_scan_supported)
3752                 return android_pno_stop(bss);
3753 #endif /* ANDROID */
3754
3755         msg = nlmsg_alloc();
3756         if (!msg)
3757                 return -1;
3758
3759         nl80211_cmd(drv, msg, 0, NL80211_CMD_STOP_SCHED_SCAN);
3760
3761         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3762
3763         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3764         msg = NULL;
3765         if (ret) {
3766                 wpa_printf(MSG_DEBUG, "nl80211: Sched scan stop failed: "
3767                            "ret=%d (%s)", ret, strerror(-ret));
3768                 goto nla_put_failure;
3769         }
3770
3771         wpa_printf(MSG_DEBUG, "nl80211: Sched scan stop sent (ret=%d)", ret);
3772
3773 nla_put_failure:
3774         nlmsg_free(msg);
3775         return ret;
3776 }
3777
3778
3779 static const u8 * nl80211_get_ie(const u8 *ies, size_t ies_len, u8 ie)
3780 {
3781         const u8 *end, *pos;
3782
3783         if (ies == NULL)
3784                 return NULL;
3785
3786         pos = ies;
3787         end = ies + ies_len;
3788
3789         while (pos + 1 < end) {
3790                 if (pos + 2 + pos[1] > end)
3791                         break;
3792                 if (pos[0] == ie)
3793                         return pos;
3794                 pos += 2 + pos[1];
3795         }
3796
3797         return NULL;
3798 }
3799
3800
3801 static int nl80211_scan_filtered(struct wpa_driver_nl80211_data *drv,
3802                                  const u8 *ie, size_t ie_len)
3803 {
3804         const u8 *ssid;
3805         size_t i;
3806
3807         if (drv->filter_ssids == NULL)
3808                 return 0;
3809
3810         ssid = nl80211_get_ie(ie, ie_len, WLAN_EID_SSID);
3811         if (ssid == NULL)
3812                 return 1;
3813
3814         for (i = 0; i < drv->num_filter_ssids; i++) {
3815                 if (ssid[1] == drv->filter_ssids[i].ssid_len &&
3816                     os_memcmp(ssid + 2, drv->filter_ssids[i].ssid, ssid[1]) ==
3817                     0)
3818                         return 0;
3819         }
3820
3821         return 1;
3822 }
3823
3824
3825 static int bss_info_handler(struct nl_msg *msg, void *arg)
3826 {
3827         struct nlattr *tb[NL80211_ATTR_MAX + 1];
3828         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
3829         struct nlattr *bss[NL80211_BSS_MAX + 1];
3830         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
3831                 [NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
3832                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
3833                 [NL80211_BSS_TSF] = { .type = NLA_U64 },
3834                 [NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
3835                 [NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
3836                 [NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
3837                 [NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
3838                 [NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
3839                 [NL80211_BSS_STATUS] = { .type = NLA_U32 },
3840                 [NL80211_BSS_SEEN_MS_AGO] = { .type = NLA_U32 },
3841                 [NL80211_BSS_BEACON_IES] = { .type = NLA_UNSPEC },
3842         };
3843         struct nl80211_bss_info_arg *_arg = arg;
3844         struct wpa_scan_results *res = _arg->res;
3845         struct wpa_scan_res **tmp;
3846         struct wpa_scan_res *r;
3847         const u8 *ie, *beacon_ie;
3848         size_t ie_len, beacon_ie_len;
3849         u8 *pos;
3850         size_t i;
3851
3852         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
3853                   genlmsg_attrlen(gnlh, 0), NULL);
3854         if (!tb[NL80211_ATTR_BSS])
3855                 return NL_SKIP;
3856         if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
3857                              bss_policy))
3858                 return NL_SKIP;
3859         if (bss[NL80211_BSS_STATUS]) {
3860                 enum nl80211_bss_status status;
3861                 status = nla_get_u32(bss[NL80211_BSS_STATUS]);
3862                 if (status == NL80211_BSS_STATUS_ASSOCIATED &&
3863                     bss[NL80211_BSS_FREQUENCY]) {
3864                         _arg->assoc_freq =
3865                                 nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
3866                         wpa_printf(MSG_DEBUG, "nl80211: Associated on %u MHz",
3867                                    _arg->assoc_freq);
3868                 }
3869                 if (status == NL80211_BSS_STATUS_ASSOCIATED &&
3870                     bss[NL80211_BSS_BSSID]) {
3871                         os_memcpy(_arg->assoc_bssid,
3872                                   nla_data(bss[NL80211_BSS_BSSID]), ETH_ALEN);
3873                         wpa_printf(MSG_DEBUG, "nl80211: Associated with "
3874                                    MACSTR, MAC2STR(_arg->assoc_bssid));
3875                 }
3876         }
3877         if (!res)
3878                 return NL_SKIP;
3879         if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
3880                 ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
3881                 ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
3882         } else {
3883                 ie = NULL;
3884                 ie_len = 0;
3885         }
3886         if (bss[NL80211_BSS_BEACON_IES]) {
3887                 beacon_ie = nla_data(bss[NL80211_BSS_BEACON_IES]);
3888                 beacon_ie_len = nla_len(bss[NL80211_BSS_BEACON_IES]);
3889         } else {
3890                 beacon_ie = NULL;
3891                 beacon_ie_len = 0;
3892         }
3893
3894         if (nl80211_scan_filtered(_arg->drv, ie ? ie : beacon_ie,
3895                                   ie ? ie_len : beacon_ie_len))
3896                 return NL_SKIP;
3897
3898         r = os_zalloc(sizeof(*r) + ie_len + beacon_ie_len);
3899         if (r == NULL)
3900                 return NL_SKIP;
3901         if (bss[NL80211_BSS_BSSID])
3902                 os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
3903                           ETH_ALEN);
3904         if (bss[NL80211_BSS_FREQUENCY])
3905                 r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
3906         if (bss[NL80211_BSS_BEACON_INTERVAL])
3907                 r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
3908         if (bss[NL80211_BSS_CAPABILITY])
3909                 r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
3910         r->flags |= WPA_SCAN_NOISE_INVALID;
3911         if (bss[NL80211_BSS_SIGNAL_MBM]) {
3912                 r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
3913                 r->level /= 100; /* mBm to dBm */
3914                 r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
3915         } else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
3916                 r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
3917                 r->flags |= WPA_SCAN_QUAL_INVALID;
3918         } else
3919                 r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
3920         if (bss[NL80211_BSS_TSF])
3921                 r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
3922         if (bss[NL80211_BSS_SEEN_MS_AGO])
3923                 r->age = nla_get_u32(bss[NL80211_BSS_SEEN_MS_AGO]);
3924         r->ie_len = ie_len;
3925         pos = (u8 *) (r + 1);
3926         if (ie) {
3927                 os_memcpy(pos, ie, ie_len);
3928                 pos += ie_len;
3929         }
3930         r->beacon_ie_len = beacon_ie_len;
3931         if (beacon_ie)
3932                 os_memcpy(pos, beacon_ie, beacon_ie_len);
3933
3934         if (bss[NL80211_BSS_STATUS]) {
3935                 enum nl80211_bss_status status;
3936                 status = nla_get_u32(bss[NL80211_BSS_STATUS]);
3937                 switch (status) {
3938                 case NL80211_BSS_STATUS_AUTHENTICATED:
3939                         r->flags |= WPA_SCAN_AUTHENTICATED;
3940                         break;
3941                 case NL80211_BSS_STATUS_ASSOCIATED:
3942                         r->flags |= WPA_SCAN_ASSOCIATED;
3943                         break;
3944                 default:
3945                         break;
3946                 }
3947         }
3948
3949         /*
3950          * cfg80211 maintains separate BSS table entries for APs if the same
3951          * BSSID,SSID pair is seen on multiple channels. wpa_supplicant does
3952          * not use frequency as a separate key in the BSS table, so filter out
3953          * duplicated entries. Prefer associated BSS entry in such a case in
3954          * order to get the correct frequency into the BSS table.
3955          */
3956         for (i = 0; i < res->num; i++) {
3957                 const u8 *s1, *s2;
3958                 if (os_memcmp(res->res[i]->bssid, r->bssid, ETH_ALEN) != 0)
3959                         continue;
3960
3961                 s1 = nl80211_get_ie((u8 *) (res->res[i] + 1),
3962                                     res->res[i]->ie_len, WLAN_EID_SSID);
3963                 s2 = nl80211_get_ie((u8 *) (r + 1), r->ie_len, WLAN_EID_SSID);
3964                 if (s1 == NULL || s2 == NULL || s1[1] != s2[1] ||
3965                     os_memcmp(s1, s2, 2 + s1[1]) != 0)
3966                         continue;
3967
3968                 /* Same BSSID,SSID was already included in scan results */
3969                 wpa_printf(MSG_DEBUG, "nl80211: Remove duplicated scan result "
3970                            "for " MACSTR, MAC2STR(r->bssid));
3971
3972                 if ((r->flags & WPA_SCAN_ASSOCIATED) &&
3973                     !(res->res[i]->flags & WPA_SCAN_ASSOCIATED)) {
3974                         os_free(res->res[i]);
3975                         res->res[i] = r;
3976                 } else
3977                         os_free(r);
3978                 return NL_SKIP;
3979         }
3980
3981         tmp = os_realloc(res->res,
3982                          (res->num + 1) * sizeof(struct wpa_scan_res *));
3983         if (tmp == NULL) {
3984                 os_free(r);
3985                 return NL_SKIP;
3986         }
3987         tmp[res->num++] = r;
3988         res->res = tmp;
3989
3990         return NL_SKIP;
3991 }
3992
3993
3994 static void clear_state_mismatch(struct wpa_driver_nl80211_data *drv,
3995                                  const u8 *addr)
3996 {
3997         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
3998                 wpa_printf(MSG_DEBUG, "nl80211: Clear possible state "
3999                            "mismatch (" MACSTR ")", MAC2STR(addr));
4000                 wpa_driver_nl80211_mlme(drv, addr,
4001                                         NL80211_CMD_DEAUTHENTICATE,
4002                                         WLAN_REASON_PREV_AUTH_NOT_VALID, 1);
4003         }
4004 }
4005
4006
4007 static void wpa_driver_nl80211_check_bss_status(
4008         struct wpa_driver_nl80211_data *drv, struct wpa_scan_results *res)
4009 {
4010         size_t i;
4011
4012         for (i = 0; i < res->num; i++) {
4013                 struct wpa_scan_res *r = res->res[i];
4014                 if (r->flags & WPA_SCAN_AUTHENTICATED) {
4015                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
4016                                    "indicates BSS status with " MACSTR
4017                                    " as authenticated",
4018                                    MAC2STR(r->bssid));
4019                         if (is_sta_interface(drv->nlmode) &&
4020                             os_memcmp(r->bssid, drv->bssid, ETH_ALEN) != 0 &&
4021                             os_memcmp(r->bssid, drv->auth_bssid, ETH_ALEN) !=
4022                             0) {
4023                                 wpa_printf(MSG_DEBUG, "nl80211: Unknown BSSID"
4024                                            " in local state (auth=" MACSTR
4025                                            " assoc=" MACSTR ")",
4026                                            MAC2STR(drv->auth_bssid),
4027                                            MAC2STR(drv->bssid));
4028                                 clear_state_mismatch(drv, r->bssid);
4029                         }
4030                 }
4031
4032                 if (r->flags & WPA_SCAN_ASSOCIATED) {
4033                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
4034                                    "indicate BSS status with " MACSTR
4035                                    " as associated",
4036                                    MAC2STR(r->bssid));
4037                         if (is_sta_interface(drv->nlmode) &&
4038                             !drv->associated) {
4039                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
4040                                            "(not associated) does not match "
4041                                            "with BSS state");
4042                                 clear_state_mismatch(drv, r->bssid);
4043                         } else if (is_sta_interface(drv->nlmode) &&
4044                                    os_memcmp(drv->bssid, r->bssid, ETH_ALEN) !=
4045                                    0) {
4046                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
4047                                            "(associated with " MACSTR ") does "
4048                                            "not match with BSS state",
4049                                            MAC2STR(drv->bssid));
4050                                 clear_state_mismatch(drv, r->bssid);
4051                                 clear_state_mismatch(drv, drv->bssid);
4052                         }
4053                 }
4054         }
4055 }
4056
4057
4058 static struct wpa_scan_results *
4059 nl80211_get_scan_results(struct wpa_driver_nl80211_data *drv)
4060 {
4061         struct nl_msg *msg;
4062         struct wpa_scan_results *res;
4063         int ret;
4064         struct nl80211_bss_info_arg arg;
4065
4066         res = os_zalloc(sizeof(*res));
4067         if (res == NULL)
4068                 return NULL;
4069         msg = nlmsg_alloc();
4070         if (!msg)
4071                 goto nla_put_failure;
4072
4073         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
4074         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4075
4076         arg.drv = drv;
4077         arg.res = res;
4078         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
4079         msg = NULL;
4080         if (ret == 0) {
4081                 wpa_printf(MSG_DEBUG, "nl80211: Received scan results (%lu "
4082                            "BSSes)", (unsigned long) res->num);
4083                 nl80211_get_noise_for_scan_results(drv, res);
4084                 return res;
4085         }
4086         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
4087                    "(%s)", ret, strerror(-ret));
4088 nla_put_failure:
4089         nlmsg_free(msg);
4090         wpa_scan_results_free(res);
4091         return NULL;
4092 }
4093
4094
4095 /**
4096  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
4097  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
4098  * Returns: Scan results on success, -1 on failure
4099  */
4100 static struct wpa_scan_results *
4101 wpa_driver_nl80211_get_scan_results(void *priv)
4102 {
4103         struct i802_bss *bss = priv;
4104         struct wpa_driver_nl80211_data *drv = bss->drv;
4105         struct wpa_scan_results *res;
4106
4107         res = nl80211_get_scan_results(drv);
4108         if (res)
4109                 wpa_driver_nl80211_check_bss_status(drv, res);
4110         return res;
4111 }
4112
4113
4114 static void nl80211_dump_scan(struct wpa_driver_nl80211_data *drv)
4115 {
4116         struct wpa_scan_results *res;
4117         size_t i;
4118
4119         res = nl80211_get_scan_results(drv);
4120         if (res == NULL) {
4121                 wpa_printf(MSG_DEBUG, "nl80211: Failed to get scan results");
4122                 return;
4123         }
4124
4125         wpa_printf(MSG_DEBUG, "nl80211: Scan result dump");
4126         for (i = 0; i < res->num; i++) {
4127                 struct wpa_scan_res *r = res->res[i];
4128                 wpa_printf(MSG_DEBUG, "nl80211: %d/%d " MACSTR "%s%s",
4129                            (int) i, (int) res->num, MAC2STR(r->bssid),
4130                            r->flags & WPA_SCAN_AUTHENTICATED ? " [auth]" : "",
4131                            r->flags & WPA_SCAN_ASSOCIATED ? " [assoc]" : "");
4132         }
4133
4134         wpa_scan_results_free(res);
4135 }
4136
4137
4138 static int wpa_driver_nl80211_set_key(const char *ifname, void *priv,
4139                                       enum wpa_alg alg, const u8 *addr,
4140                                       int key_idx, int set_tx,
4141                                       const u8 *seq, size_t seq_len,
4142                                       const u8 *key, size_t key_len)
4143 {
4144         struct i802_bss *bss = priv;
4145         struct wpa_driver_nl80211_data *drv = bss->drv;
4146         int ifindex = if_nametoindex(ifname);
4147         struct nl_msg *msg;
4148         int ret;
4149
4150         wpa_printf(MSG_DEBUG, "%s: ifindex=%d alg=%d addr=%p key_idx=%d "
4151                    "set_tx=%d seq_len=%lu key_len=%lu",
4152                    __func__, ifindex, alg, addr, key_idx, set_tx,
4153                    (unsigned long) seq_len, (unsigned long) key_len);
4154 #ifdef CONFIG_TDLS
4155         if (key_idx == -1)
4156                 key_idx = 0;
4157 #endif /* CONFIG_TDLS */
4158
4159         msg = nlmsg_alloc();
4160         if (!msg)
4161                 return -ENOMEM;
4162
4163         if (alg == WPA_ALG_NONE) {
4164                 nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_KEY);
4165         } else {
4166                 nl80211_cmd(drv, msg, 0, NL80211_CMD_NEW_KEY);
4167                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
4168                 switch (alg) {
4169                 case WPA_ALG_WEP:
4170                         if (key_len == 5)
4171                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4172                                             WLAN_CIPHER_SUITE_WEP40);
4173                         else
4174                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4175                                             WLAN_CIPHER_SUITE_WEP104);
4176                         break;
4177                 case WPA_ALG_TKIP:
4178                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4179                                     WLAN_CIPHER_SUITE_TKIP);
4180                         break;
4181                 case WPA_ALG_CCMP:
4182                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4183                                     WLAN_CIPHER_SUITE_CCMP);
4184                         break;
4185                 case WPA_ALG_IGTK:
4186                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
4187                                     WLAN_CIPHER_SUITE_AES_CMAC);
4188                         break;
4189                 default:
4190                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
4191                                    "algorithm %d", __func__, alg);
4192                         nlmsg_free(msg);
4193                         return -1;
4194                 }
4195         }
4196
4197         if (seq && seq_len)
4198                 NLA_PUT(msg, NL80211_ATTR_KEY_SEQ, seq_len, seq);
4199
4200         if (addr && !is_broadcast_ether_addr(addr)) {
4201                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
4202                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4203
4204                 if (alg != WPA_ALG_WEP && key_idx && !set_tx) {
4205                         wpa_printf(MSG_DEBUG, "   RSN IBSS RX GTK");
4206                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_TYPE,
4207                                     NL80211_KEYTYPE_GROUP);
4208                 }
4209         } else if (addr && is_broadcast_ether_addr(addr)) {
4210                 struct nl_msg *types;
4211                 int err;
4212                 wpa_printf(MSG_DEBUG, "   broadcast key");
4213                 types = nlmsg_alloc();
4214                 if (!types)
4215                         goto nla_put_failure;
4216                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_MULTICAST);
4217                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4218                                      types);
4219                 nlmsg_free(types);
4220                 if (err)
4221                         goto nla_put_failure;
4222         }
4223         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
4224         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
4225
4226         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4227         if ((ret == -ENOENT || ret == -ENOLINK) && alg == WPA_ALG_NONE)
4228                 ret = 0;
4229         if (ret)
4230                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d %s)",
4231                            ret, strerror(-ret));
4232
4233         /*
4234          * If we failed or don't need to set the default TX key (below),
4235          * we're done here.
4236          */
4237         if (ret || !set_tx || alg == WPA_ALG_NONE)
4238                 return ret;
4239         if (is_ap_interface(drv->nlmode) && addr &&
4240             !is_broadcast_ether_addr(addr))
4241                 return ret;
4242
4243         msg = nlmsg_alloc();
4244         if (!msg)
4245                 return -ENOMEM;
4246
4247         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_KEY);
4248         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
4249         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
4250         if (alg == WPA_ALG_IGTK)
4251                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
4252         else
4253                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
4254         if (addr && is_broadcast_ether_addr(addr)) {
4255                 struct nl_msg *types;
4256                 int err;
4257                 types = nlmsg_alloc();
4258                 if (!types)
4259                         goto nla_put_failure;
4260                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_MULTICAST);
4261                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4262                                      types);
4263                 nlmsg_free(types);
4264                 if (err)
4265                         goto nla_put_failure;
4266         } else if (addr) {
4267                 struct nl_msg *types;
4268                 int err;
4269                 types = nlmsg_alloc();
4270                 if (!types)
4271                         goto nla_put_failure;
4272                 NLA_PUT_FLAG(types, NL80211_KEY_DEFAULT_TYPE_UNICAST);
4273                 err = nla_put_nested(msg, NL80211_ATTR_KEY_DEFAULT_TYPES,
4274                                      types);
4275                 nlmsg_free(types);
4276                 if (err)
4277                         goto nla_put_failure;
4278         }
4279
4280         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4281         if (ret == -ENOENT)
4282                 ret = 0;
4283         if (ret)
4284                 wpa_printf(MSG_DEBUG, "nl80211: set_key default failed; "
4285                            "err=%d %s)", ret, strerror(-ret));
4286         return ret;
4287
4288 nla_put_failure:
4289         nlmsg_free(msg);
4290         return -ENOBUFS;
4291 }
4292
4293
4294 static int nl_add_key(struct nl_msg *msg, enum wpa_alg alg,
4295                       int key_idx, int defkey,
4296                       const u8 *seq, size_t seq_len,
4297                       const u8 *key, size_t key_len)
4298 {
4299         struct nlattr *key_attr = nla_nest_start(msg, NL80211_ATTR_KEY);
4300         if (!key_attr)
4301                 return -1;
4302
4303         if (defkey && alg == WPA_ALG_IGTK)
4304                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT_MGMT);
4305         else if (defkey)
4306                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
4307
4308         NLA_PUT_U8(msg, NL80211_KEY_IDX, key_idx);
4309
4310         switch (alg) {
4311         case WPA_ALG_WEP:
4312                 if (key_len == 5)
4313                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4314                                     WLAN_CIPHER_SUITE_WEP40);
4315                 else
4316                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4317                                     WLAN_CIPHER_SUITE_WEP104);
4318                 break;
4319         case WPA_ALG_TKIP:
4320                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_TKIP);
4321                 break;
4322         case WPA_ALG_CCMP:
4323                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_CCMP);
4324                 break;
4325         case WPA_ALG_IGTK:
4326                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4327                             WLAN_CIPHER_SUITE_AES_CMAC);
4328                 break;
4329         default:
4330                 wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
4331                            "algorithm %d", __func__, alg);
4332                 return -1;
4333         }
4334
4335         if (seq && seq_len)
4336                 NLA_PUT(msg, NL80211_KEY_SEQ, seq_len, seq);
4337
4338         NLA_PUT(msg, NL80211_KEY_DATA, key_len, key);
4339
4340         nla_nest_end(msg, key_attr);
4341
4342         return 0;
4343  nla_put_failure:
4344         return -1;
4345 }
4346
4347
4348 static int nl80211_set_conn_keys(struct wpa_driver_associate_params *params,
4349                                  struct nl_msg *msg)
4350 {
4351         int i, privacy = 0;
4352         struct nlattr *nl_keys, *nl_key;
4353
4354         for (i = 0; i < 4; i++) {
4355                 if (!params->wep_key[i])
4356                         continue;
4357                 privacy = 1;
4358                 break;
4359         }
4360         if (params->wps == WPS_MODE_PRIVACY)
4361                 privacy = 1;
4362         if (params->pairwise_suite &&
4363             params->pairwise_suite != WPA_CIPHER_NONE)
4364                 privacy = 1;
4365
4366         if (!privacy)
4367                 return 0;
4368
4369         NLA_PUT_FLAG(msg, NL80211_ATTR_PRIVACY);
4370
4371         nl_keys = nla_nest_start(msg, NL80211_ATTR_KEYS);
4372         if (!nl_keys)
4373                 goto nla_put_failure;
4374
4375         for (i = 0; i < 4; i++) {
4376                 if (!params->wep_key[i])
4377                         continue;
4378
4379                 nl_key = nla_nest_start(msg, i);
4380                 if (!nl_key)
4381                         goto nla_put_failure;
4382
4383                 NLA_PUT(msg, NL80211_KEY_DATA, params->wep_key_len[i],
4384                         params->wep_key[i]);
4385                 if (params->wep_key_len[i] == 5)
4386                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4387                                     WLAN_CIPHER_SUITE_WEP40);
4388                 else
4389                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
4390                                     WLAN_CIPHER_SUITE_WEP104);
4391
4392                 NLA_PUT_U8(msg, NL80211_KEY_IDX, i);
4393
4394                 if (i == params->wep_tx_keyidx)
4395                         NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
4396
4397                 nla_nest_end(msg, nl_key);
4398         }
4399         nla_nest_end(msg, nl_keys);
4400
4401         return 0;
4402
4403 nla_put_failure:
4404         return -ENOBUFS;
4405 }
4406
4407
4408 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
4409                                    const u8 *addr, int cmd, u16 reason_code,
4410                                    int local_state_change)
4411 {
4412         int ret = -1;
4413         struct nl_msg *msg;
4414
4415         msg = nlmsg_alloc();
4416         if (!msg)
4417                 return -1;
4418
4419         nl80211_cmd(drv, msg, 0, cmd);
4420
4421         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4422         NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason_code);
4423         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4424         if (local_state_change)
4425                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
4426
4427         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4428         msg = NULL;
4429         if (ret) {
4430                 wpa_dbg(drv->ctx, MSG_DEBUG,
4431                         "nl80211: MLME command failed: reason=%u ret=%d (%s)",
4432                         reason_code, ret, strerror(-ret));
4433                 goto nla_put_failure;
4434         }
4435         ret = 0;
4436
4437 nla_put_failure:
4438         nlmsg_free(msg);
4439         return ret;
4440 }
4441
4442
4443 static int wpa_driver_nl80211_disconnect(struct wpa_driver_nl80211_data *drv,
4444                                          const u8 *addr, int reason_code)
4445 {
4446         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
4447                    __func__, MAC2STR(addr), reason_code);
4448         drv->associated = 0;
4449         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISCONNECT,
4450                                        reason_code, 0);
4451 }
4452
4453
4454 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
4455                                              int reason_code)
4456 {
4457         struct i802_bss *bss = priv;
4458         struct wpa_driver_nl80211_data *drv = bss->drv;
4459         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
4460                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
4461         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
4462                    __func__, MAC2STR(addr), reason_code);
4463         drv->associated = 0;
4464         if (drv->nlmode == NL80211_IFTYPE_ADHOC)
4465                 return nl80211_leave_ibss(drv);
4466         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DEAUTHENTICATE,
4467                                        reason_code, 0);
4468 }
4469
4470
4471 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
4472                                            int reason_code)
4473 {
4474         struct i802_bss *bss = priv;
4475         struct wpa_driver_nl80211_data *drv = bss->drv;
4476         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
4477                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
4478         wpa_printf(MSG_DEBUG, "%s", __func__);
4479         drv->associated = 0;
4480         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISASSOCIATE,
4481                                        reason_code, 0);
4482 }
4483
4484
4485 static void nl80211_copy_auth_params(struct wpa_driver_nl80211_data *drv,
4486                                      struct wpa_driver_auth_params *params)
4487 {
4488         int i;
4489
4490         drv->auth_freq = params->freq;
4491         drv->auth_alg = params->auth_alg;
4492         drv->auth_wep_tx_keyidx = params->wep_tx_keyidx;
4493         drv->auth_local_state_change = params->local_state_change;
4494         drv->auth_p2p = params->p2p;
4495
4496         if (params->bssid)
4497                 os_memcpy(drv->auth_bssid_, params->bssid, ETH_ALEN);
4498         else
4499                 os_memset(drv->auth_bssid_, 0, ETH_ALEN);
4500
4501         if (params->ssid) {
4502                 os_memcpy(drv->auth_ssid, params->ssid, params->ssid_len);
4503                 drv->auth_ssid_len = params->ssid_len;
4504         } else
4505                 drv->auth_ssid_len = 0;
4506
4507
4508         os_free(drv->auth_ie);
4509         drv->auth_ie = NULL;
4510         drv->auth_ie_len = 0;
4511         if (params->ie) {
4512                 drv->auth_ie = os_malloc(params->ie_len);
4513                 if (drv->auth_ie) {
4514                         os_memcpy(drv->auth_ie, params->ie, params->ie_len);
4515                         drv->auth_ie_len = params->ie_len;
4516                 }
4517         }
4518
4519         for (i = 0; i < 4; i++) {
4520                 if (params->wep_key[i] && params->wep_key_len[i] &&
4521                     params->wep_key_len[i] <= 16) {
4522                         os_memcpy(drv->auth_wep_key[i], params->wep_key[i],
4523                                   params->wep_key_len[i]);
4524                         drv->auth_wep_key_len[i] = params->wep_key_len[i];
4525                 } else
4526                         drv->auth_wep_key_len[i] = 0;
4527         }
4528 }
4529
4530
4531 static int wpa_driver_nl80211_authenticate(
4532         void *priv, struct wpa_driver_auth_params *params)
4533 {
4534         struct i802_bss *bss = priv;
4535         struct wpa_driver_nl80211_data *drv = bss->drv;
4536         int ret = -1, i;
4537         struct nl_msg *msg;
4538         enum nl80211_auth_type type;
4539         enum nl80211_iftype nlmode;
4540         int count = 0;
4541         int is_retry;
4542
4543         is_retry = drv->retry_auth;
4544         drv->retry_auth = 0;
4545
4546         drv->associated = 0;
4547         os_memset(drv->auth_bssid, 0, ETH_ALEN);
4548         /* FIX: IBSS mode */
4549         nlmode = params->p2p ?
4550                 NL80211_IFTYPE_P2P_CLIENT : NL80211_IFTYPE_STATION;
4551         if (drv->nlmode != nlmode &&
4552             wpa_driver_nl80211_set_mode(priv, nlmode) < 0)
4553                 return -1;
4554
4555 retry:
4556         msg = nlmsg_alloc();
4557         if (!msg)
4558                 return -1;
4559
4560         wpa_printf(MSG_DEBUG, "nl80211: Authenticate (ifindex=%d)",
4561                    drv->ifindex);
4562
4563         nl80211_cmd(drv, msg, 0, NL80211_CMD_AUTHENTICATE);
4564
4565         for (i = 0; i < 4; i++) {
4566                 if (!params->wep_key[i])
4567                         continue;
4568                 wpa_driver_nl80211_set_key(bss->ifname, priv, WPA_ALG_WEP,
4569                                            NULL, i,
4570                                            i == params->wep_tx_keyidx, NULL, 0,
4571                                            params->wep_key[i],
4572                                            params->wep_key_len[i]);
4573                 if (params->wep_tx_keyidx != i)
4574                         continue;
4575                 if (nl_add_key(msg, WPA_ALG_WEP, i, 1, NULL, 0,
4576                                params->wep_key[i], params->wep_key_len[i])) {
4577                         nlmsg_free(msg);
4578                         return -1;
4579                 }
4580         }
4581
4582         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4583         if (params->bssid) {
4584                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
4585                            MAC2STR(params->bssid));
4586                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
4587         }
4588         if (params->freq) {
4589                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
4590                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
4591         }
4592         if (params->ssid) {
4593                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
4594                                   params->ssid, params->ssid_len);
4595                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
4596                         params->ssid);
4597         }
4598         wpa_hexdump(MSG_DEBUG, "  * IEs", params->ie, params->ie_len);
4599         if (params->ie)
4600                 NLA_PUT(msg, NL80211_ATTR_IE, params->ie_len, params->ie);
4601         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
4602                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
4603         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
4604                 type = NL80211_AUTHTYPE_SHARED_KEY;
4605         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
4606                 type = NL80211_AUTHTYPE_NETWORK_EAP;
4607         else if (params->auth_alg & WPA_AUTH_ALG_FT)
4608                 type = NL80211_AUTHTYPE_FT;
4609         else
4610                 goto nla_put_failure;
4611         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
4612         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
4613         if (params->local_state_change) {
4614                 wpa_printf(MSG_DEBUG, "  * Local state change only");
4615                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
4616         }
4617
4618         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4619         msg = NULL;
4620         if (ret) {
4621                 wpa_dbg(drv->ctx, MSG_DEBUG,
4622                         "nl80211: MLME command failed (auth): ret=%d (%s)",
4623                         ret, strerror(-ret));
4624                 count++;
4625                 if (ret == -EALREADY && count == 1 && params->bssid &&
4626                     !params->local_state_change) {
4627                         /*
4628                          * mac80211 does not currently accept new
4629                          * authentication if we are already authenticated. As a
4630                          * workaround, force deauthentication and try again.
4631                          */
4632                         wpa_printf(MSG_DEBUG, "nl80211: Retry authentication "
4633                                    "after forced deauthentication");
4634                         wpa_driver_nl80211_deauthenticate(
4635                                 bss, params->bssid,
4636                                 WLAN_REASON_PREV_AUTH_NOT_VALID);
4637                         nlmsg_free(msg);
4638                         goto retry;
4639                 }
4640
4641                 if (ret == -ENOENT && params->freq && !is_retry) {
4642                         /*
4643                          * cfg80211 has likely expired the BSS entry even
4644                          * though it was previously available in our internal
4645                          * BSS table. To recover quickly, start a single
4646                          * channel scan on the specified channel.
4647                          */
4648                         struct wpa_driver_scan_params scan;
4649                         int freqs[2];
4650
4651                         os_memset(&scan, 0, sizeof(scan));
4652                         scan.num_ssids = 1;
4653                         if (params->ssid) {
4654                                 scan.ssids[0].ssid = params->ssid;
4655                                 scan.ssids[0].ssid_len = params->ssid_len;
4656                         }
4657                         freqs[0] = params->freq;
4658                         freqs[1] = 0;
4659                         scan.freqs = freqs;
4660                         wpa_printf(MSG_DEBUG, "nl80211: Trigger single "
4661                                    "channel scan to refresh cfg80211 BSS "
4662                                    "entry");
4663                         ret = wpa_driver_nl80211_scan(bss, &scan);
4664                         if (ret == 0) {
4665                                 nl80211_copy_auth_params(drv, params);
4666                                 drv->scan_for_auth = 1;
4667                         }
4668                 } else if (is_retry) {
4669                         /*
4670                          * Need to indicate this with an event since the return
4671                          * value from the retry is not delivered to core code.
4672                          */
4673                         union wpa_event_data event;
4674                         wpa_printf(MSG_DEBUG, "nl80211: Authentication retry "
4675                                    "failed");
4676                         os_memset(&event, 0, sizeof(event));
4677                         os_memcpy(event.timeout_event.addr, drv->auth_bssid_,
4678                                   ETH_ALEN);
4679                         wpa_supplicant_event(drv->ctx, EVENT_AUTH_TIMED_OUT,
4680                                              &event);
4681                 }
4682
4683                 goto nla_put_failure;
4684         }
4685         ret = 0;
4686         wpa_printf(MSG_DEBUG, "nl80211: Authentication request send "
4687                    "successfully");
4688
4689 nla_put_failure:
4690         nlmsg_free(msg);
4691         return ret;
4692 }
4693
4694
4695 static int wpa_driver_nl80211_authenticate_retry(
4696         struct wpa_driver_nl80211_data *drv)
4697 {
4698         struct wpa_driver_auth_params params;
4699         struct i802_bss *bss = &drv->first_bss;
4700         int i;
4701
4702         wpa_printf(MSG_DEBUG, "nl80211: Try to authenticate again");
4703
4704         os_memset(&params, 0, sizeof(params));
4705         params.freq = drv->auth_freq;
4706         params.auth_alg = drv->auth_alg;
4707         params.wep_tx_keyidx = drv->auth_wep_tx_keyidx;
4708         params.local_state_change = drv->auth_local_state_change;
4709         params.p2p = drv->auth_p2p;
4710
4711         if (!is_zero_ether_addr(drv->auth_bssid_))
4712                 params.bssid = drv->auth_bssid_;
4713
4714         if (drv->auth_ssid_len) {
4715                 params.ssid = drv->auth_ssid;
4716                 params.ssid_len = drv->auth_ssid_len;
4717         }
4718
4719         params.ie = drv->auth_ie;
4720         params.ie_len = drv->auth_ie_len;
4721
4722         for (i = 0; i < 4; i++) {
4723                 if (drv->auth_wep_key_len[i]) {
4724                         params.wep_key[i] = drv->auth_wep_key[i];
4725                         params.wep_key_len[i] = drv->auth_wep_key_len[i];
4726                 }
4727         }
4728
4729         drv->retry_auth = 1;
4730         return wpa_driver_nl80211_authenticate(bss, &params);
4731 }
4732
4733
4734 struct phy_info_arg {
4735         u16 *num_modes;
4736         struct hostapd_hw_modes *modes;
4737 };
4738
4739 static int phy_info_handler(struct nl_msg *msg, void *arg)
4740 {
4741         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
4742         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
4743         struct phy_info_arg *phy_info = arg;
4744
4745         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
4746
4747         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
4748         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
4749                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
4750                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
4751                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
4752                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
4753                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
4754                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
4755         };
4756
4757         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
4758         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
4759                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
4760                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
4761         };
4762
4763         struct nlattr *nl_band;
4764         struct nlattr *nl_freq;
4765         struct nlattr *nl_rate;
4766         int rem_band, rem_freq, rem_rate;
4767         struct hostapd_hw_modes *mode;
4768         int idx, mode_is_set;
4769
4770         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
4771                   genlmsg_attrlen(gnlh, 0), NULL);
4772
4773         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
4774                 return NL_SKIP;
4775
4776         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
4777                 mode = os_realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
4778                 if (!mode)
4779                         return NL_SKIP;
4780                 phy_info->modes = mode;
4781
4782                 mode_is_set = 0;
4783
4784                 mode = &phy_info->modes[*(phy_info->num_modes)];
4785                 memset(mode, 0, sizeof(*mode));
4786                 mode->flags = HOSTAPD_MODE_FLAG_HT_INFO_KNOWN;
4787                 *(phy_info->num_modes) += 1;
4788
4789                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
4790                           nla_len(nl_band), NULL);
4791
4792                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
4793                         mode->ht_capab = nla_get_u16(
4794                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
4795                 }
4796
4797                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) {
4798                         mode->a_mpdu_params |= nla_get_u8(
4799                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) &
4800                                 0x03;
4801                 }
4802
4803                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) {
4804                         mode->a_mpdu_params |= nla_get_u8(
4805                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) <<
4806                                 2;
4807                 }
4808
4809                 if (tb_band[NL80211_BAND_ATTR_HT_MCS_SET] &&
4810                     nla_len(tb_band[NL80211_BAND_ATTR_HT_MCS_SET])) {
4811                         u8 *mcs;
4812                         mcs = nla_data(tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
4813                         os_memcpy(mode->mcs_set, mcs, 16);
4814                 }
4815
4816                 if (tb_band[NL80211_BAND_ATTR_VHT_CAPA]) {
4817                         mode->vht_capab = nla_get_u32(
4818                                 tb_band[NL80211_BAND_ATTR_VHT_CAPA]);
4819                 }
4820
4821                 if (tb_band[NL80211_BAND_ATTR_VHT_MCS_SET] &&
4822                     nla_len(tb_band[NL80211_BAND_ATTR_VHT_MCS_SET])) {
4823                         u8 *mcs;
4824                         mcs = nla_data(tb_band[NL80211_BAND_ATTR_VHT_MCS_SET]);
4825                         os_memcpy(mode->vht_mcs_set, mcs, 8);
4826                 }
4827
4828                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
4829                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
4830                                   nla_len(nl_freq), freq_policy);
4831                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
4832                                 continue;
4833                         mode->num_channels++;
4834                 }
4835
4836                 mode->channels = os_zalloc(mode->num_channels * sizeof(struct hostapd_channel_data));
4837                 if (!mode->channels)
4838                         return NL_SKIP;
4839
4840                 idx = 0;
4841
4842                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
4843                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
4844                                   nla_len(nl_freq), freq_policy);
4845                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
4846                                 continue;
4847
4848                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
4849                         mode->channels[idx].flag = 0;
4850
4851                         if (!mode_is_set) {
4852                                 /* crude heuristic */
4853                                 if (mode->channels[idx].freq < 4000)
4854                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
4855                                 else
4856                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
4857                                 mode_is_set = 1;
4858                         }
4859
4860                         /* crude heuristic */
4861                         if (mode->channels[idx].freq < 4000)
4862                                 if (mode->channels[idx].freq == 2484)
4863                                         mode->channels[idx].chan = 14;
4864                                 else
4865                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
4866                         else
4867                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
4868
4869                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
4870                                 mode->channels[idx].flag |=
4871                                         HOSTAPD_CHAN_DISABLED;
4872                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
4873                                 mode->channels[idx].flag |=
4874                                         HOSTAPD_CHAN_PASSIVE_SCAN;
4875                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
4876                                 mode->channels[idx].flag |=
4877                                         HOSTAPD_CHAN_NO_IBSS;
4878                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
4879                                 mode->channels[idx].flag |=
4880                                         HOSTAPD_CHAN_RADAR;
4881
4882                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
4883                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
4884                                 mode->channels[idx].max_tx_power =
4885                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
4886
4887                         idx++;
4888                 }
4889
4890                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
4891                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
4892                                   nla_len(nl_rate), rate_policy);
4893                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
4894                                 continue;
4895                         mode->num_rates++;
4896                 }
4897
4898                 mode->rates = os_zalloc(mode->num_rates * sizeof(int));
4899                 if (!mode->rates)
4900                         return NL_SKIP;
4901
4902                 idx = 0;
4903
4904                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
4905                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
4906                                   nla_len(nl_rate), rate_policy);
4907                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
4908                                 continue;
4909                         mode->rates[idx] = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
4910
4911                         /* crude heuristic */
4912                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
4913                             mode->rates[idx] > 200)
4914                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
4915
4916                         idx++;
4917                 }
4918         }
4919
4920         return NL_SKIP;
4921 }
4922
4923 static struct hostapd_hw_modes *
4924 wpa_driver_nl80211_add_11b(struct hostapd_hw_modes *modes, u16 *num_modes)
4925 {
4926         u16 m;
4927         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
4928         int i, mode11g_idx = -1;
4929
4930         /* If only 802.11g mode is included, use it to construct matching
4931          * 802.11b mode data. */
4932
4933         for (m = 0; m < *num_modes; m++) {
4934                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
4935                         return modes; /* 802.11b already included */
4936                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
4937                         mode11g_idx = m;
4938         }
4939
4940         if (mode11g_idx < 0)
4941                 return modes; /* 2.4 GHz band not supported at all */
4942
4943         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
4944         if (nmodes == NULL)
4945                 return modes; /* Could not add 802.11b mode */
4946
4947         mode = &nmodes[*num_modes];
4948         os_memset(mode, 0, sizeof(*mode));
4949         (*num_modes)++;
4950         modes = nmodes;
4951
4952         mode->mode = HOSTAPD_MODE_IEEE80211B;
4953
4954         mode11g = &modes[mode11g_idx];
4955         mode->num_channels = mode11g->num_channels;
4956         mode->channels = os_malloc(mode11g->num_channels *
4957                                    sizeof(struct hostapd_channel_data));
4958         if (mode->channels == NULL) {
4959                 (*num_modes)--;
4960                 return modes; /* Could not add 802.11b mode */
4961         }
4962         os_memcpy(mode->channels, mode11g->channels,
4963                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
4964
4965         mode->num_rates = 0;
4966         mode->rates = os_malloc(4 * sizeof(int));
4967         if (mode->rates == NULL) {
4968                 os_free(mode->channels);
4969                 (*num_modes)--;
4970                 return modes; /* Could not add 802.11b mode */
4971         }
4972
4973         for (i = 0; i < mode11g->num_rates; i++) {
4974                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
4975                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
4976                         continue;
4977                 mode->rates[mode->num_rates] = mode11g->rates[i];
4978                 mode->num_rates++;
4979                 if (mode->num_rates == 4)
4980                         break;
4981         }
4982
4983         if (mode->num_rates == 0) {
4984                 os_free(mode->channels);
4985                 os_free(mode->rates);
4986                 (*num_modes)--;
4987                 return modes; /* No 802.11b rates */
4988         }
4989
4990         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
4991                    "information");
4992
4993         return modes;
4994 }
4995
4996
4997 static void nl80211_set_ht40_mode(struct hostapd_hw_modes *mode, int start,
4998                                   int end)
4999 {
5000         int c;
5001
5002         for (c = 0; c < mode->num_channels; c++) {
5003                 struct hostapd_channel_data *chan = &mode->channels[c];
5004                 if (chan->freq - 10 >= start && chan->freq + 10 <= end)
5005                         chan->flag |= HOSTAPD_CHAN_HT40;
5006         }
5007 }
5008
5009
5010 static void nl80211_set_ht40_mode_sec(struct hostapd_hw_modes *mode, int start,
5011                                       int end)
5012 {
5013         int c;
5014
5015         for (c = 0; c < mode->num_channels; c++) {
5016                 struct hostapd_channel_data *chan = &mode->channels[c];
5017                 if (!(chan->flag & HOSTAPD_CHAN_HT40))
5018                         continue;
5019                 if (chan->freq - 30 >= start && chan->freq - 10 <= end)
5020                         chan->flag |= HOSTAPD_CHAN_HT40MINUS;
5021                 if (chan->freq + 10 >= start && chan->freq + 30 <= end)
5022                         chan->flag |= HOSTAPD_CHAN_HT40PLUS;
5023         }
5024 }
5025
5026
5027 static void nl80211_reg_rule_ht40(struct nlattr *tb[],
5028                                   struct phy_info_arg *results)
5029 {
5030         u32 start, end, max_bw;
5031         u16 m;
5032
5033         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
5034             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
5035             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
5036                 return;
5037
5038         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
5039         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
5040         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
5041
5042         wpa_printf(MSG_DEBUG, "nl80211: %u-%u @ %u MHz",
5043                    start, end, max_bw);
5044         if (max_bw < 40)
5045                 return;
5046
5047         for (m = 0; m < *results->num_modes; m++) {
5048                 if (!(results->modes[m].ht_capab &
5049                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
5050                         continue;
5051                 nl80211_set_ht40_mode(&results->modes[m], start, end);
5052         }
5053 }
5054
5055
5056 static void nl80211_reg_rule_sec(struct nlattr *tb[],
5057                                  struct phy_info_arg *results)
5058 {
5059         u32 start, end, max_bw;
5060         u16 m;
5061
5062         if (tb[NL80211_ATTR_FREQ_RANGE_START] == NULL ||
5063             tb[NL80211_ATTR_FREQ_RANGE_END] == NULL ||
5064             tb[NL80211_ATTR_FREQ_RANGE_MAX_BW] == NULL)
5065                 return;
5066
5067         start = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_START]) / 1000;
5068         end = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_END]) / 1000;
5069         max_bw = nla_get_u32(tb[NL80211_ATTR_FREQ_RANGE_MAX_BW]) / 1000;
5070
5071         if (max_bw < 20)
5072                 return;
5073
5074         for (m = 0; m < *results->num_modes; m++) {
5075                 if (!(results->modes[m].ht_capab &
5076                       HT_CAP_INFO_SUPP_CHANNEL_WIDTH_SET))
5077                         continue;
5078                 nl80211_set_ht40_mode_sec(&results->modes[m], start, end);
5079         }
5080 }
5081
5082
5083 static int nl80211_get_reg(struct nl_msg *msg, void *arg)
5084 {
5085         struct phy_info_arg *results = arg;
5086         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
5087         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
5088         struct nlattr *nl_rule;
5089         struct nlattr *tb_rule[NL80211_FREQUENCY_ATTR_MAX + 1];
5090         int rem_rule;
5091         static struct nla_policy reg_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
5092                 [NL80211_ATTR_REG_RULE_FLAGS] = { .type = NLA_U32 },
5093                 [NL80211_ATTR_FREQ_RANGE_START] = { .type = NLA_U32 },
5094                 [NL80211_ATTR_FREQ_RANGE_END] = { .type = NLA_U32 },
5095                 [NL80211_ATTR_FREQ_RANGE_MAX_BW] = { .type = NLA_U32 },
5096                 [NL80211_ATTR_POWER_RULE_MAX_ANT_GAIN] = { .type = NLA_U32 },
5097                 [NL80211_ATTR_POWER_RULE_MAX_EIRP] = { .type = NLA_U32 },
5098         };
5099
5100         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
5101                   genlmsg_attrlen(gnlh, 0), NULL);
5102         if (!tb_msg[NL80211_ATTR_REG_ALPHA2] ||
5103             !tb_msg[NL80211_ATTR_REG_RULES]) {
5104                 wpa_printf(MSG_DEBUG, "nl80211: No regulatory information "
5105                            "available");
5106                 return NL_SKIP;
5107         }
5108
5109         wpa_printf(MSG_DEBUG, "nl80211: Regulatory information - country=%s",
5110                    (char *) nla_data(tb_msg[NL80211_ATTR_REG_ALPHA2]));
5111
5112         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
5113         {
5114                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
5115                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
5116                 nl80211_reg_rule_ht40(tb_rule, results);
5117         }
5118
5119         nla_for_each_nested(nl_rule, tb_msg[NL80211_ATTR_REG_RULES], rem_rule)
5120         {
5121                 nla_parse(tb_rule, NL80211_FREQUENCY_ATTR_MAX,
5122                           nla_data(nl_rule), nla_len(nl_rule), reg_policy);
5123                 nl80211_reg_rule_sec(tb_rule, results);
5124         }
5125
5126         return NL_SKIP;
5127 }
5128
5129
5130 static int nl80211_set_ht40_flags(struct wpa_driver_nl80211_data *drv,
5131                                   struct phy_info_arg *results)
5132 {
5133         struct nl_msg *msg;
5134
5135         msg = nlmsg_alloc();
5136         if (!msg)
5137                 return -ENOMEM;
5138
5139         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_REG);
5140         return send_and_recv_msgs(drv, msg, nl80211_get_reg, results);
5141 }
5142
5143
5144 static struct hostapd_hw_modes *
5145 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
5146 {
5147         struct i802_bss *bss = priv;
5148         struct wpa_driver_nl80211_data *drv = bss->drv;
5149         struct nl_msg *msg;
5150         struct phy_info_arg result = {
5151                 .num_modes = num_modes,
5152                 .modes = NULL,
5153         };
5154
5155         *num_modes = 0;
5156         *flags = 0;
5157
5158         msg = nlmsg_alloc();
5159         if (!msg)
5160                 return NULL;
5161
5162         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_WIPHY);
5163
5164         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5165
5166         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0) {
5167                 nl80211_set_ht40_flags(drv, &result);
5168                 return wpa_driver_nl80211_add_11b(result.modes, num_modes);
5169         }
5170         msg = NULL;
5171  nla_put_failure:
5172         nlmsg_free(msg);
5173         return NULL;
5174 }
5175
5176
5177 static int wpa_driver_nl80211_send_mntr(struct wpa_driver_nl80211_data *drv,
5178                                         const void *data, size_t len,
5179                                         int encrypt, int noack)
5180 {
5181         __u8 rtap_hdr[] = {
5182                 0x00, 0x00, /* radiotap version */
5183                 0x0e, 0x00, /* radiotap length */
5184                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
5185                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
5186                 0x00,       /* padding */
5187                 0x00, 0x00, /* RX and TX flags to indicate that */
5188                 0x00, 0x00, /* this is the injected frame directly */
5189         };
5190         struct iovec iov[2] = {
5191                 {
5192                         .iov_base = &rtap_hdr,
5193                         .iov_len = sizeof(rtap_hdr),
5194                 },
5195                 {
5196                         .iov_base = (void *) data,
5197                         .iov_len = len,
5198                 }
5199         };
5200         struct msghdr msg = {
5201                 .msg_name = NULL,
5202                 .msg_namelen = 0,
5203                 .msg_iov = iov,
5204                 .msg_iovlen = 2,
5205                 .msg_control = NULL,
5206                 .msg_controllen = 0,
5207                 .msg_flags = 0,
5208         };
5209         int res;
5210         u16 txflags = 0;
5211
5212         if (encrypt)
5213                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
5214
5215         if (drv->monitor_sock < 0) {
5216                 wpa_printf(MSG_DEBUG, "nl80211: No monitor socket available "
5217                            "for %s", __func__);
5218                 return -1;
5219         }
5220
5221         if (noack)
5222                 txflags |= IEEE80211_RADIOTAP_F_TX_NOACK;
5223         *(le16 *) &rtap_hdr[12] = host_to_le16(txflags);
5224
5225         res = sendmsg(drv->monitor_sock, &msg, 0);
5226         if (res < 0) {
5227                 wpa_printf(MSG_INFO, "nl80211: sendmsg: %s", strerror(errno));
5228                 return -1;
5229         }
5230         return 0;
5231 }
5232
5233
5234 static int wpa_driver_nl80211_send_frame(struct i802_bss *bss,
5235                                          const void *data, size_t len,
5236                                          int encrypt, int noack,
5237                                          unsigned int freq, int no_cck,
5238                                          int offchanok, unsigned int wait_time)
5239 {
5240         struct wpa_driver_nl80211_data *drv = bss->drv;
5241         u64 cookie;
5242
5243         if (freq == 0)
5244                 freq = bss->freq;
5245
5246         if (drv->use_monitor)
5247                 return wpa_driver_nl80211_send_mntr(drv, data, len,
5248                                                     encrypt, noack);
5249
5250         return nl80211_send_frame_cmd(bss, freq, wait_time, data, len,
5251                                       &cookie, no_cck, noack, offchanok);
5252 }
5253
5254
5255 static int wpa_driver_nl80211_send_mlme_freq(struct i802_bss *bss,
5256                                              const u8 *data,
5257                                              size_t data_len, int noack,
5258                                              unsigned int freq, int no_cck,
5259                                              int offchanok,
5260                                              unsigned int wait_time)
5261 {
5262         struct wpa_driver_nl80211_data *drv = bss->drv;
5263         struct ieee80211_mgmt *mgmt;
5264         int encrypt = 1;
5265         u16 fc;
5266
5267         mgmt = (struct ieee80211_mgmt *) data;
5268         fc = le_to_host16(mgmt->frame_control);
5269
5270         if (is_sta_interface(drv->nlmode) &&
5271             WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
5272             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_PROBE_RESP) {
5273                 /*
5274                  * The use of last_mgmt_freq is a bit of a hack,
5275                  * but it works due to the single-threaded nature
5276                  * of wpa_supplicant.
5277                  */
5278                 if (freq == 0)
5279                         freq = drv->last_mgmt_freq;
5280                 return nl80211_send_frame_cmd(bss, freq, 0,
5281                                               data, data_len, NULL, 1, noack,
5282                                               1);
5283         }
5284
5285         if (drv->device_ap_sme && is_ap_interface(drv->nlmode)) {
5286                 if (freq == 0)
5287                         freq = bss->freq;
5288                 return nl80211_send_frame_cmd(bss, freq,
5289                                               (int) freq == bss->freq ? 0 :
5290                                               wait_time,
5291                                               data, data_len,
5292                                               &drv->send_action_cookie,
5293                                               no_cck, noack, offchanok);
5294         }
5295
5296         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
5297             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
5298                 /*
5299                  * Only one of the authentication frame types is encrypted.
5300                  * In order for static WEP encryption to work properly (i.e.,
5301                  * to not encrypt the frame), we need to tell mac80211 about
5302                  * the frames that must not be encrypted.
5303                  */
5304                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
5305                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
5306                 if (auth_alg != WLAN_AUTH_SHARED_KEY || auth_trans != 3)
5307                         encrypt = 0;
5308         }
5309
5310         return wpa_driver_nl80211_send_frame(bss, data, data_len, encrypt,
5311                                              noack, freq, no_cck, offchanok,
5312                                              wait_time);
5313 }
5314
5315
5316 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
5317                                         size_t data_len, int noack)
5318 {
5319         struct i802_bss *bss = priv;
5320         return wpa_driver_nl80211_send_mlme_freq(bss, data, data_len, noack,
5321                                                  0, 0, 0, 0);
5322 }
5323
5324
5325 static int nl80211_set_bss(struct i802_bss *bss, int cts, int preamble,
5326                            int slot, int ht_opmode, int ap_isolate,
5327                            int *basic_rates)
5328 {
5329         struct wpa_driver_nl80211_data *drv = bss->drv;
5330         struct nl_msg *msg;
5331
5332         msg = nlmsg_alloc();
5333         if (!msg)
5334                 return -ENOMEM;
5335
5336         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_BSS);
5337
5338         if (cts >= 0)
5339                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
5340         if (preamble >= 0)
5341                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
5342         if (slot >= 0)
5343                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
5344         if (ht_opmode >= 0)
5345                 NLA_PUT_U16(msg, NL80211_ATTR_BSS_HT_OPMODE, ht_opmode);
5346         if (ap_isolate >= 0)
5347                 NLA_PUT_U8(msg, NL80211_ATTR_AP_ISOLATE, ap_isolate);
5348
5349         if (basic_rates) {
5350                 u8 rates[NL80211_MAX_SUPP_RATES];
5351                 u8 rates_len = 0;
5352                 int i;
5353
5354                 for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0;
5355                      i++)
5356                         rates[rates_len++] = basic_rates[i] / 5;
5357
5358                 NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
5359         }
5360
5361         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
5362
5363         return send_and_recv_msgs(drv, msg, NULL, NULL);
5364  nla_put_failure:
5365         nlmsg_free(msg);
5366         return -ENOBUFS;
5367 }
5368
5369
5370 static int wpa_driver_nl80211_set_ap(void *priv,
5371                                      struct wpa_driver_ap_params *params)
5372 {
5373         struct i802_bss *bss = priv;
5374         struct wpa_driver_nl80211_data *drv = bss->drv;
5375         struct nl_msg *msg;
5376         u8 cmd = NL80211_CMD_NEW_BEACON;
5377         int ret;
5378         int beacon_set;
5379         int ifindex = if_nametoindex(bss->ifname);
5380         int num_suites;
5381         u32 suites[10];
5382         u32 ver;
5383
5384         beacon_set = bss->beacon_set;
5385
5386         msg = nlmsg_alloc();
5387         if (!msg)
5388                 return -ENOMEM;
5389
5390         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (beacon_set=%d)",
5391                    beacon_set);
5392         if (beacon_set)
5393                 cmd = NL80211_CMD_SET_BEACON;
5394
5395         nl80211_cmd(drv, msg, 0, cmd);
5396         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, params->head_len, params->head);
5397         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, params->tail_len, params->tail);
5398         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
5399         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, params->beacon_int);
5400         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, params->dtim_period);
5401         NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
5402                 params->ssid);
5403         if (params->proberesp && params->proberesp_len)
5404                 NLA_PUT(msg, NL80211_ATTR_PROBE_RESP, params->proberesp_len,
5405                         params->proberesp);
5406         switch (params->hide_ssid) {
5407         case NO_SSID_HIDING:
5408                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5409                             NL80211_HIDDEN_SSID_NOT_IN_USE);
5410                 break;
5411         case HIDDEN_SSID_ZERO_LEN:
5412                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5413                             NL80211_HIDDEN_SSID_ZERO_LEN);
5414                 break;
5415         case HIDDEN_SSID_ZERO_CONTENTS:
5416                 NLA_PUT_U32(msg, NL80211_ATTR_HIDDEN_SSID,
5417                             NL80211_HIDDEN_SSID_ZERO_CONTENTS);
5418                 break;
5419         }
5420         if (params->privacy)
5421                 NLA_PUT_FLAG(msg, NL80211_ATTR_PRIVACY);
5422         if ((params->auth_algs & (WPA_AUTH_ALG_OPEN | WPA_AUTH_ALG_SHARED)) ==
5423             (WPA_AUTH_ALG_OPEN | WPA_AUTH_ALG_SHARED)) {
5424                 /* Leave out the attribute */
5425         } else if (params->auth_algs & WPA_AUTH_ALG_SHARED)
5426                 NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
5427                             NL80211_AUTHTYPE_SHARED_KEY);
5428         else
5429                 NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE,
5430                             NL80211_AUTHTYPE_OPEN_SYSTEM);
5431
5432         ver = 0;
5433         if (params->wpa_version & WPA_PROTO_WPA)
5434                 ver |= NL80211_WPA_VERSION_1;
5435         if (params->wpa_version & WPA_PROTO_RSN)
5436                 ver |= NL80211_WPA_VERSION_2;
5437         if (ver)
5438                 NLA_PUT_U32(msg, NL80211_ATTR_WPA_VERSIONS, ver);
5439
5440         num_suites = 0;
5441         if (params->key_mgmt_suites & WPA_KEY_MGMT_IEEE8021X)
5442                 suites[num_suites++] = WLAN_AKM_SUITE_8021X;
5443         if (params->key_mgmt_suites & WPA_KEY_MGMT_PSK)
5444                 suites[num_suites++] = WLAN_AKM_SUITE_PSK;
5445         if (num_suites) {
5446                 NLA_PUT(msg, NL80211_ATTR_AKM_SUITES,
5447                         num_suites * sizeof(u32), suites);
5448         }
5449
5450         if (params->key_mgmt_suites & WPA_KEY_MGMT_IEEE8021X &&
5451             params->pairwise_ciphers & (WPA_CIPHER_WEP104 | WPA_CIPHER_WEP40))
5452                 NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT_NO_ENCRYPT);
5453
5454         num_suites = 0;
5455         if (params->pairwise_ciphers & WPA_CIPHER_CCMP)
5456                 suites[num_suites++] = WLAN_CIPHER_SUITE_CCMP;
5457         if (params->pairwise_ciphers & WPA_CIPHER_TKIP)
5458                 suites[num_suites++] = WLAN_CIPHER_SUITE_TKIP;
5459         if (params->pairwise_ciphers & WPA_CIPHER_WEP104)
5460                 suites[num_suites++] = WLAN_CIPHER_SUITE_WEP104;
5461         if (params->pairwise_ciphers & WPA_CIPHER_WEP40)
5462                 suites[num_suites++] = WLAN_CIPHER_SUITE_WEP40;
5463         if (num_suites) {
5464                 NLA_PUT(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE,
5465                         num_suites * sizeof(u32), suites);
5466         }
5467
5468         switch (params->group_cipher) {
5469         case WPA_CIPHER_CCMP:
5470                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5471                             WLAN_CIPHER_SUITE_CCMP);
5472                 break;
5473         case WPA_CIPHER_TKIP:
5474                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5475                             WLAN_CIPHER_SUITE_TKIP);
5476                 break;
5477         case WPA_CIPHER_WEP104:
5478                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5479                             WLAN_CIPHER_SUITE_WEP104);
5480                 break;
5481         case WPA_CIPHER_WEP40:
5482                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP,
5483                             WLAN_CIPHER_SUITE_WEP40);
5484                 break;
5485         }
5486
5487         if (params->beacon_ies) {
5488                 NLA_PUT(msg, NL80211_ATTR_IE, wpabuf_len(params->beacon_ies),
5489                         wpabuf_head(params->beacon_ies));
5490         }
5491         if (params->proberesp_ies) {
5492                 NLA_PUT(msg, NL80211_ATTR_IE_PROBE_RESP,
5493                         wpabuf_len(params->proberesp_ies),
5494                         wpabuf_head(params->proberesp_ies));
5495         }
5496         if (params->assocresp_ies) {
5497                 NLA_PUT(msg, NL80211_ATTR_IE_ASSOC_RESP,
5498                         wpabuf_len(params->assocresp_ies),
5499                         wpabuf_head(params->assocresp_ies));
5500         }
5501
5502         if (drv->capa.flags & WPA_DRIVER_FLAGS_INACTIVITY_TIMER)  {
5503                 NLA_PUT_U16(msg, NL80211_ATTR_INACTIVITY_TIMEOUT,
5504                             params->ap_max_inactivity);
5505         }
5506
5507         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5508         if (ret) {
5509                 wpa_printf(MSG_DEBUG, "nl80211: Beacon set failed: %d (%s)",
5510                            ret, strerror(-ret));
5511         } else {
5512                 bss->beacon_set = 1;
5513                 nl80211_set_bss(bss, params->cts_protect, params->preamble,
5514                                 params->short_slot_time, params->ht_opmode,
5515                                 params->isolate, params->basic_rates);
5516         }
5517         return ret;
5518  nla_put_failure:
5519         nlmsg_free(msg);
5520         return -ENOBUFS;
5521 }
5522
5523
5524 static int wpa_driver_nl80211_set_freq(struct i802_bss *bss,
5525                                        int freq, int ht_enabled,
5526                                        int sec_channel_offset)
5527 {
5528         struct wpa_driver_nl80211_data *drv = bss->drv;
5529         struct nl_msg *msg;
5530         int ret;
5531
5532         wpa_printf(MSG_DEBUG, "nl80211: Set freq %d (ht_enabled=%d "
5533                    "sec_channel_offset=%d)",
5534                    freq, ht_enabled, sec_channel_offset);
5535         msg = nlmsg_alloc();
5536         if (!msg)
5537                 return -1;
5538
5539         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
5540
5541         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5542         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
5543         if (ht_enabled) {
5544                 switch (sec_channel_offset) {
5545                 case -1:
5546                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5547                                     NL80211_CHAN_HT40MINUS);
5548                         break;
5549                 case 1:
5550                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5551                                     NL80211_CHAN_HT40PLUS);
5552                         break;
5553                 default:
5554                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
5555                                     NL80211_CHAN_HT20);
5556                         break;
5557                 }
5558         }
5559
5560         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5561         msg = NULL;
5562         if (ret == 0) {
5563                 bss->freq = freq;
5564                 return 0;
5565         }
5566         wpa_printf(MSG_DEBUG, "nl80211: Failed to set channel (freq=%d): "
5567                    "%d (%s)", freq, ret, strerror(-ret));
5568 nla_put_failure:
5569         nlmsg_free(msg);
5570         return -1;
5571 }
5572
5573
5574 static u32 sta_flags_nl80211(int flags)
5575 {
5576         u32 f = 0;
5577
5578         if (flags & WPA_STA_AUTHORIZED)
5579                 f |= BIT(NL80211_STA_FLAG_AUTHORIZED);
5580         if (flags & WPA_STA_WMM)
5581                 f |= BIT(NL80211_STA_FLAG_WME);
5582         if (flags & WPA_STA_SHORT_PREAMBLE)
5583                 f |= BIT(NL80211_STA_FLAG_SHORT_PREAMBLE);
5584         if (flags & WPA_STA_MFP)
5585                 f |= BIT(NL80211_STA_FLAG_MFP);
5586         if (flags & WPA_STA_TDLS_PEER)
5587                 f |= BIT(NL80211_STA_FLAG_TDLS_PEER);
5588
5589         return f;
5590 }
5591
5592
5593 static int wpa_driver_nl80211_sta_add(void *priv,
5594                                       struct hostapd_sta_add_params *params)
5595 {
5596         struct i802_bss *bss = priv;
5597         struct wpa_driver_nl80211_data *drv = bss->drv;
5598         struct nl_msg *msg, *wme = NULL;
5599         struct nl80211_sta_flag_update upd;
5600         int ret = -ENOBUFS;
5601
5602         if ((params->flags & WPA_STA_TDLS_PEER) &&
5603             !(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
5604                 return -EOPNOTSUPP;
5605
5606         msg = nlmsg_alloc();
5607         if (!msg)
5608                 return -ENOMEM;
5609
5610         nl80211_cmd(drv, msg, 0, params->set ? NL80211_CMD_SET_STATION :
5611                     NL80211_CMD_NEW_STATION);
5612
5613         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
5614         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
5615         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
5616                 params->supp_rates);
5617         if (!params->set) {
5618                 NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
5619                 NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
5620                             params->listen_interval);
5621         }
5622         if (params->ht_capabilities) {
5623                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
5624                         sizeof(*params->ht_capabilities),
5625                         params->ht_capabilities);
5626         }
5627
5628         os_memset(&upd, 0, sizeof(upd));
5629         upd.mask = sta_flags_nl80211(params->flags);
5630         upd.set = upd.mask;
5631         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
5632
5633         if (params->flags & WPA_STA_WMM) {
5634                 wme = nlmsg_alloc();
5635                 if (!wme)
5636                         goto nla_put_failure;
5637
5638                 NLA_PUT_U8(wme, NL80211_STA_WME_UAPSD_QUEUES,
5639                                 params->qosinfo & WMM_QOSINFO_STA_AC_MASK);
5640                 NLA_PUT_U8(wme, NL80211_STA_WME_MAX_SP,
5641                                 (params->qosinfo > WMM_QOSINFO_STA_SP_SHIFT) &
5642                                 WMM_QOSINFO_STA_SP_MASK);
5643                 nla_put_nested(msg, NL80211_ATTR_STA_WME, wme);
5644         }
5645
5646         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5647         msg = NULL;
5648         if (ret)
5649                 wpa_printf(MSG_DEBUG, "nl80211: NL80211_CMD_%s_STATION "
5650                            "result: %d (%s)", params->set ? "SET" : "NEW", ret,
5651                            strerror(-ret));
5652         if (ret == -EEXIST)
5653                 ret = 0;
5654  nla_put_failure:
5655         nlmsg_free(wme);
5656         nlmsg_free(msg);
5657         return ret;
5658 }
5659
5660
5661 static int wpa_driver_nl80211_sta_remove(void *priv, const u8 *addr)
5662 {
5663         struct i802_bss *bss = priv;
5664         struct wpa_driver_nl80211_data *drv = bss->drv;
5665         struct nl_msg *msg;
5666         int ret;
5667
5668         msg = nlmsg_alloc();
5669         if (!msg)
5670                 return -ENOMEM;
5671
5672         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_STATION);
5673
5674         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
5675                     if_nametoindex(bss->ifname));
5676         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
5677
5678         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5679         if (ret == -ENOENT)
5680                 return 0;
5681         return ret;
5682  nla_put_failure:
5683         nlmsg_free(msg);
5684         return -ENOBUFS;
5685 }
5686
5687
5688 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
5689                                  int ifidx)
5690 {
5691         struct nl_msg *msg;
5692
5693         wpa_printf(MSG_DEBUG, "nl80211: Remove interface ifindex=%d", ifidx);
5694
5695         /* stop listening for EAPOL on this interface */
5696         del_ifidx(drv, ifidx);
5697
5698         msg = nlmsg_alloc();
5699         if (!msg)
5700                 goto nla_put_failure;
5701
5702         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_INTERFACE);
5703         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
5704
5705         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
5706                 return;
5707         msg = NULL;
5708  nla_put_failure:
5709         nlmsg_free(msg);
5710         wpa_printf(MSG_ERROR, "Failed to remove interface (ifidx=%d)", ifidx);
5711 }
5712
5713
5714 static const char * nl80211_iftype_str(enum nl80211_iftype mode)
5715 {
5716         switch (mode) {
5717         case NL80211_IFTYPE_ADHOC:
5718                 return "ADHOC";
5719         case NL80211_IFTYPE_STATION:
5720                 return "STATION";
5721         case NL80211_IFTYPE_AP:
5722                 return "AP";
5723         case NL80211_IFTYPE_MONITOR:
5724                 return "MONITOR";
5725         case NL80211_IFTYPE_P2P_CLIENT:
5726                 return "P2P_CLIENT";
5727         case NL80211_IFTYPE_P2P_GO:
5728                 return "P2P_GO";
5729         default:
5730                 return "unknown";
5731         }
5732 }
5733
5734
5735 static int nl80211_create_iface_once(struct wpa_driver_nl80211_data *drv,
5736                                      const char *ifname,
5737                                      enum nl80211_iftype iftype,
5738                                      const u8 *addr, int wds)
5739 {
5740         struct nl_msg *msg, *flags = NULL;
5741         int ifidx;
5742         int ret = -ENOBUFS;
5743
5744         wpa_printf(MSG_DEBUG, "nl80211: Create interface iftype %d (%s)",
5745                    iftype, nl80211_iftype_str(iftype));
5746
5747         msg = nlmsg_alloc();
5748         if (!msg)
5749                 return -1;
5750
5751         nl80211_cmd(drv, msg, 0, NL80211_CMD_NEW_INTERFACE);
5752         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5753         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
5754         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
5755
5756         if (iftype == NL80211_IFTYPE_MONITOR) {
5757                 int err;
5758
5759                 flags = nlmsg_alloc();
5760                 if (!flags)
5761                         goto nla_put_failure;
5762
5763                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
5764
5765                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
5766
5767                 nlmsg_free(flags);
5768
5769                 if (err)
5770                         goto nla_put_failure;
5771         } else if (wds) {
5772                 NLA_PUT_U8(msg, NL80211_ATTR_4ADDR, wds);
5773         }
5774
5775         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5776         msg = NULL;
5777         if (ret) {
5778  nla_put_failure:
5779                 nlmsg_free(msg);
5780                 wpa_printf(MSG_ERROR, "Failed to create interface %s: %d (%s)",
5781                            ifname, ret, strerror(-ret));
5782                 return ret;
5783         }
5784
5785         ifidx = if_nametoindex(ifname);
5786         wpa_printf(MSG_DEBUG, "nl80211: New interface %s created: ifindex=%d",
5787                    ifname, ifidx);
5788
5789         if (ifidx <= 0)
5790                 return -1;
5791
5792         /* start listening for EAPOL on this interface */
5793         add_ifidx(drv, ifidx);
5794
5795         if (addr && iftype != NL80211_IFTYPE_MONITOR &&
5796             linux_set_ifhwaddr(drv->global->ioctl_sock, ifname, addr)) {
5797                 nl80211_remove_iface(drv, ifidx);
5798                 return -1;
5799         }
5800
5801         return ifidx;
5802 }
5803
5804
5805 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
5806                                 const char *ifname, enum nl80211_iftype iftype,
5807                                 const u8 *addr, int wds)
5808 {
5809         int ret;
5810
5811         ret = nl80211_create_iface_once(drv, ifname, iftype, addr, wds);
5812
5813         /* if error occurred and interface exists already */
5814         if (ret == -ENFILE && if_nametoindex(ifname)) {
5815                 wpa_printf(MSG_INFO, "Try to remove and re-create %s", ifname);
5816
5817                 /* Try to remove the interface that was already there. */
5818                 nl80211_remove_iface(drv, if_nametoindex(ifname));
5819
5820                 /* Try to create the interface again */
5821                 ret = nl80211_create_iface_once(drv, ifname, iftype, addr,
5822                                                 wds);
5823         }
5824
5825         if (ret >= 0 && is_p2p_interface(iftype))
5826                 nl80211_disable_11b_rates(drv, ret, 1);
5827
5828         return ret;
5829 }
5830
5831
5832 static void handle_tx_callback(void *ctx, u8 *buf, size_t len, int ok)
5833 {
5834         struct ieee80211_hdr *hdr;
5835         u16 fc;
5836         union wpa_event_data event;
5837
5838         hdr = (struct ieee80211_hdr *) buf;
5839         fc = le_to_host16(hdr->frame_control);
5840
5841         os_memset(&event, 0, sizeof(event));
5842         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
5843         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
5844         event.tx_status.dst = hdr->addr1;
5845         event.tx_status.data = buf;
5846         event.tx_status.data_len = len;
5847         event.tx_status.ack = ok;
5848         wpa_supplicant_event(ctx, EVENT_TX_STATUS, &event);
5849 }
5850
5851
5852 static void from_unknown_sta(struct wpa_driver_nl80211_data *drv,
5853                              u8 *buf, size_t len)
5854 {
5855         struct ieee80211_hdr *hdr = (void *)buf;
5856         u16 fc;
5857         union wpa_event_data event;
5858
5859         if (len < sizeof(*hdr))
5860                 return;
5861
5862         fc = le_to_host16(hdr->frame_control);
5863
5864         os_memset(&event, 0, sizeof(event));
5865         event.rx_from_unknown.bssid = get_hdr_bssid(hdr, len);
5866         event.rx_from_unknown.addr = hdr->addr2;
5867         event.rx_from_unknown.wds = (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) ==
5868                 (WLAN_FC_FROMDS | WLAN_FC_TODS);
5869         wpa_supplicant_event(drv->ctx, EVENT_RX_FROM_UNKNOWN, &event);
5870 }
5871
5872
5873 static void handle_frame(struct wpa_driver_nl80211_data *drv,
5874                          u8 *buf, size_t len, int datarate, int ssi_signal)
5875 {
5876         struct ieee80211_hdr *hdr;
5877         u16 fc;
5878         union wpa_event_data event;
5879
5880         hdr = (struct ieee80211_hdr *) buf;
5881         fc = le_to_host16(hdr->frame_control);
5882
5883         switch (WLAN_FC_GET_TYPE(fc)) {
5884         case WLAN_FC_TYPE_MGMT:
5885                 os_memset(&event, 0, sizeof(event));
5886                 event.rx_mgmt.frame = buf;
5887                 event.rx_mgmt.frame_len = len;
5888                 event.rx_mgmt.datarate = datarate;
5889                 event.rx_mgmt.ssi_signal = ssi_signal;
5890                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
5891                 break;
5892         case WLAN_FC_TYPE_CTRL:
5893                 /* can only get here with PS-Poll frames */
5894                 wpa_printf(MSG_DEBUG, "CTRL");
5895                 from_unknown_sta(drv, buf, len);
5896                 break;
5897         case WLAN_FC_TYPE_DATA:
5898                 from_unknown_sta(drv, buf, len);
5899                 break;
5900         }
5901 }
5902
5903
5904 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
5905 {
5906         struct wpa_driver_nl80211_data *drv = eloop_ctx;
5907         int len;
5908         unsigned char buf[3000];
5909         struct ieee80211_radiotap_iterator iter;
5910         int ret;
5911         int datarate = 0, ssi_signal = 0;
5912         int injected = 0, failed = 0, rxflags = 0;
5913
5914         len = recv(sock, buf, sizeof(buf), 0);
5915         if (len < 0) {
5916                 perror("recv");
5917                 return;
5918         }
5919
5920         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
5921                 printf("received invalid radiotap frame\n");
5922                 return;
5923         }
5924
5925         while (1) {
5926                 ret = ieee80211_radiotap_iterator_next(&iter);
5927                 if (ret == -ENOENT)
5928                         break;
5929                 if (ret) {
5930                         printf("received invalid radiotap frame (%d)\n", ret);
5931                         return;
5932                 }
5933                 switch (iter.this_arg_index) {
5934                 case IEEE80211_RADIOTAP_FLAGS:
5935                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
5936                                 len -= 4;
5937                         break;
5938                 case IEEE80211_RADIOTAP_RX_FLAGS:
5939                         rxflags = 1;
5940                         break;
5941                 case IEEE80211_RADIOTAP_TX_FLAGS:
5942                         injected = 1;
5943                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
5944                                         IEEE80211_RADIOTAP_F_TX_FAIL;
5945                         break;
5946                 case IEEE80211_RADIOTAP_DATA_RETRIES:
5947                         break;
5948                 case IEEE80211_RADIOTAP_CHANNEL:
5949                         /* TODO: convert from freq/flags to channel number */
5950                         break;
5951                 case IEEE80211_RADIOTAP_RATE:
5952                         datarate = *iter.this_arg * 5;
5953                         break;
5954                 case IEEE80211_RADIOTAP_DBM_ANTSIGNAL:
5955                         ssi_signal = (s8) *iter.this_arg;
5956                         break;
5957                 }
5958         }
5959
5960         if (rxflags && injected)
5961                 return;
5962
5963         if (!injected)
5964                 handle_frame(drv, buf + iter.max_length,
5965                              len - iter.max_length, datarate, ssi_signal);
5966         else
5967                 handle_tx_callback(drv->ctx, buf + iter.max_length,
5968                                    len - iter.max_length, !failed);
5969 }
5970
5971
5972 /*
5973  * we post-process the filter code later and rewrite
5974  * this to the offset to the last instruction
5975  */
5976 #define PASS    0xFF
5977 #define FAIL    0xFE
5978
5979 static struct sock_filter msock_filter_insns[] = {
5980         /*
5981          * do a little-endian load of the radiotap length field
5982          */
5983         /* load lower byte into A */
5984         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
5985         /* put it into X (== index register) */
5986         BPF_STMT(BPF_MISC| BPF_TAX, 0),
5987         /* load upper byte into A */
5988         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
5989         /* left-shift it by 8 */
5990         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
5991         /* or with X */
5992         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
5993         /* put result into X */
5994         BPF_STMT(BPF_MISC| BPF_TAX, 0),
5995
5996         /*
5997          * Allow management frames through, this also gives us those
5998          * management frames that we sent ourselves with status
5999          */
6000         /* load the lower byte of the IEEE 802.11 frame control field */
6001         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
6002         /* mask off frame type and version */
6003         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
6004         /* accept frame if it's both 0, fall through otherwise */
6005         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
6006
6007         /*
6008          * TODO: add a bit to radiotap RX flags that indicates
6009          * that the sending station is not associated, then
6010          * add a filter here that filters on our DA and that flag
6011          * to allow us to deauth frames to that bad station.
6012          *
6013          * For now allow all To DS data frames through.
6014          */
6015         /* load the IEEE 802.11 frame control field */
6016         BPF_STMT(BPF_LD  | BPF_H | BPF_IND, 0),
6017         /* mask off frame type, version and DS status */
6018         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0x0F03),
6019         /* accept frame if version 0, type 2 and To DS, fall through otherwise
6020          */
6021         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0801, PASS, 0),
6022
6023 #if 0
6024         /*
6025          * drop non-data frames
6026          */
6027         /* load the lower byte of the frame control field */
6028         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
6029         /* mask off QoS bit */
6030         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
6031         /* drop non-data frames */
6032         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
6033 #endif
6034         /* load the upper byte of the frame control field */
6035         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 1),
6036         /* mask off toDS/fromDS */
6037         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
6038         /* accept WDS frames */
6039         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, PASS, 0),
6040
6041         /*
6042          * add header length to index
6043          */
6044         /* load the lower byte of the frame control field */
6045         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
6046         /* mask off QoS bit */
6047         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
6048         /* right shift it by 6 to give 0 or 2 */
6049         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
6050         /* add data frame header length */
6051         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
6052         /* add index, was start of 802.11 header */
6053         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
6054         /* move to index, now start of LL header */
6055         BPF_STMT(BPF_MISC | BPF_TAX, 0),
6056
6057         /*
6058          * Accept empty data frames, we use those for
6059          * polling activity.
6060          */
6061         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
6062         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
6063
6064         /*
6065          * Accept EAPOL frames
6066          */
6067         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
6068         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
6069         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
6070         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
6071
6072         /* keep these last two statements or change the code below */
6073         /* return 0 == "DROP" */
6074         BPF_STMT(BPF_RET | BPF_K, 0),
6075         /* return ~0 == "keep all" */
6076         BPF_STMT(BPF_RET | BPF_K, ~0),
6077 };
6078
6079 static struct sock_fprog msock_filter = {
6080         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
6081         .filter = msock_filter_insns,
6082 };
6083
6084
6085 static int add_monitor_filter(int s)
6086 {
6087         int idx;
6088
6089         /* rewrite all PASS/FAIL jump offsets */
6090         for (idx = 0; idx < msock_filter.len; idx++) {
6091                 struct sock_filter *insn = &msock_filter_insns[idx];
6092
6093                 if (BPF_CLASS(insn->code) == BPF_JMP) {
6094                         if (insn->code == (BPF_JMP|BPF_JA)) {
6095                                 if (insn->k == PASS)
6096                                         insn->k = msock_filter.len - idx - 2;
6097                                 else if (insn->k == FAIL)
6098                                         insn->k = msock_filter.len - idx - 3;
6099                         }
6100
6101                         if (insn->jt == PASS)
6102                                 insn->jt = msock_filter.len - idx - 2;
6103                         else if (insn->jt == FAIL)
6104                                 insn->jt = msock_filter.len - idx - 3;
6105
6106                         if (insn->jf == PASS)
6107                                 insn->jf = msock_filter.len - idx - 2;
6108                         else if (insn->jf == FAIL)
6109                                 insn->jf = msock_filter.len - idx - 3;
6110                 }
6111         }
6112
6113         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
6114                        &msock_filter, sizeof(msock_filter))) {
6115                 perror("SO_ATTACH_FILTER");
6116                 return -1;
6117         }
6118
6119         return 0;
6120 }
6121
6122
6123 static void nl80211_remove_monitor_interface(
6124         struct wpa_driver_nl80211_data *drv)
6125 {
6126         drv->monitor_refcount--;
6127         if (drv->monitor_refcount > 0)
6128                 return;
6129
6130         if (drv->monitor_ifidx >= 0) {
6131                 nl80211_remove_iface(drv, drv->monitor_ifidx);
6132                 drv->monitor_ifidx = -1;
6133         }
6134         if (drv->monitor_sock >= 0) {
6135                 eloop_unregister_read_sock(drv->monitor_sock);
6136                 close(drv->monitor_sock);
6137                 drv->monitor_sock = -1;
6138         }
6139 }
6140
6141
6142 static int
6143 nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
6144 {
6145         char buf[IFNAMSIZ];
6146         struct sockaddr_ll ll;
6147         int optval;
6148         socklen_t optlen;
6149
6150         if (drv->monitor_ifidx >= 0) {
6151                 drv->monitor_refcount++;
6152                 return 0;
6153         }
6154
6155         if (os_strncmp(drv->first_bss.ifname, "p2p-", 4) == 0) {
6156                 /*
6157                  * P2P interface name is of the format p2p-%s-%d. For monitor
6158                  * interface name corresponding to P2P GO, replace "p2p-" with
6159                  * "mon-" to retain the same interface name length and to
6160                  * indicate that it is a monitor interface.
6161                  */
6162                 snprintf(buf, IFNAMSIZ, "mon-%s", drv->first_bss.ifname + 4);
6163         } else {
6164                 /* Non-P2P interface with AP functionality. */
6165                 snprintf(buf, IFNAMSIZ, "mon.%s", drv->first_bss.ifname);
6166         }
6167
6168         buf[IFNAMSIZ - 1] = '\0';
6169
6170         drv->monitor_ifidx =
6171                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL,
6172                                      0);
6173
6174         if (drv->monitor_ifidx == -EOPNOTSUPP) {
6175                 /*
6176                  * This is backward compatibility for a few versions of
6177                  * the kernel only that didn't advertise the right
6178                  * attributes for the only driver that then supported
6179                  * AP mode w/o monitor -- ath6kl.
6180                  */
6181                 wpa_printf(MSG_DEBUG, "nl80211: Driver does not support "
6182                            "monitor interface type - try to run without it");
6183                 drv->device_ap_sme = 1;
6184         }
6185
6186         if (drv->monitor_ifidx < 0)
6187                 return -1;
6188
6189         if (linux_set_iface_flags(drv->global->ioctl_sock, buf, 1))
6190                 goto error;
6191
6192         memset(&ll, 0, sizeof(ll));
6193         ll.sll_family = AF_PACKET;
6194         ll.sll_ifindex = drv->monitor_ifidx;
6195         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
6196         if (drv->monitor_sock < 0) {
6197                 perror("socket[PF_PACKET,SOCK_RAW]");
6198                 goto error;
6199         }
6200
6201         if (add_monitor_filter(drv->monitor_sock)) {
6202                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
6203                            "interface; do filtering in user space");
6204                 /* This works, but will cost in performance. */
6205         }
6206
6207         if (bind(drv->monitor_sock, (struct sockaddr *) &ll, sizeof(ll)) < 0) {
6208                 perror("monitor socket bind");
6209                 goto error;
6210         }
6211
6212         optlen = sizeof(optval);
6213         optval = 20;
6214         if (setsockopt
6215             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
6216                 perror("Failed to set socket priority");
6217                 goto error;
6218         }
6219
6220         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
6221                                      drv, NULL)) {
6222                 printf("Could not register monitor read socket\n");
6223                 goto error;
6224         }
6225
6226         return 0;
6227  error:
6228         nl80211_remove_monitor_interface(drv);
6229         return -1;
6230 }
6231
6232
6233 static int nl80211_setup_ap(struct i802_bss *bss)
6234 {
6235         struct wpa_driver_nl80211_data *drv = bss->drv;
6236
6237         wpa_printf(MSG_DEBUG, "nl80211: Setup AP - device_ap_sme=%d "
6238                    "use_monitor=%d", drv->device_ap_sme, drv->use_monitor);
6239
6240         /*
6241          * Disable Probe Request reporting unless we need it in this way for
6242          * devices that include the AP SME, in the other case (unless using
6243          * monitor iface) we'll get it through the nl_mgmt socket instead.
6244          */
6245         if (!drv->device_ap_sme)
6246                 wpa_driver_nl80211_probe_req_report(bss, 0);
6247
6248         if (!drv->device_ap_sme && !drv->use_monitor)
6249                 if (nl80211_mgmt_subscribe_ap(bss))
6250                         return -1;
6251
6252         if (drv->device_ap_sme && !drv->use_monitor)
6253                 if (nl80211_mgmt_subscribe_ap_dev_sme(bss))
6254                         return -1;
6255
6256         if (!drv->device_ap_sme && drv->use_monitor &&
6257             nl80211_create_monitor_interface(drv) &&
6258             !drv->device_ap_sme)
6259                 return -1;
6260
6261         if (drv->device_ap_sme &&
6262             wpa_driver_nl80211_probe_req_report(bss, 1) < 0) {
6263                 wpa_printf(MSG_DEBUG, "nl80211: Failed to enable "
6264                            "Probe Request frame reporting in AP mode");
6265                 /* Try to survive without this */
6266         }
6267
6268         return 0;
6269 }
6270
6271
6272 static void nl80211_teardown_ap(struct i802_bss *bss)
6273 {
6274         struct wpa_driver_nl80211_data *drv = bss->drv;
6275
6276         if (drv->device_ap_sme) {
6277                 wpa_driver_nl80211_probe_req_report(bss, 0);
6278                 if (!drv->use_monitor)
6279                         nl80211_mgmt_unsubscribe(bss, "AP teardown (dev SME)");
6280         } else if (drv->use_monitor)
6281                 nl80211_remove_monitor_interface(drv);
6282         else
6283                 nl80211_mgmt_unsubscribe(bss, "AP teardown");
6284
6285         bss->beacon_set = 0;
6286 }
6287
6288
6289 static int nl80211_send_eapol_data(struct i802_bss *bss,
6290                                    const u8 *addr, const u8 *data,
6291                                    size_t data_len)
6292 {
6293         struct sockaddr_ll ll;
6294         int ret;
6295
6296         if (bss->drv->eapol_tx_sock < 0) {
6297                 wpa_printf(MSG_DEBUG, "nl80211: No socket to send EAPOL");
6298                 return -1;
6299         }
6300
6301         os_memset(&ll, 0, sizeof(ll));
6302         ll.sll_family = AF_PACKET;
6303         ll.sll_ifindex = bss->ifindex;
6304         ll.sll_protocol = htons(ETH_P_PAE);
6305         ll.sll_halen = ETH_ALEN;
6306         os_memcpy(ll.sll_addr, addr, ETH_ALEN);
6307         ret = sendto(bss->drv->eapol_tx_sock, data, data_len, 0,
6308                      (struct sockaddr *) &ll, sizeof(ll));
6309         if (ret < 0)
6310                 wpa_printf(MSG_ERROR, "nl80211: EAPOL TX: %s",
6311                            strerror(errno));
6312
6313         return ret;
6314 }
6315
6316
6317 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
6318
6319 static int wpa_driver_nl80211_hapd_send_eapol(
6320         void *priv, const u8 *addr, const u8 *data,
6321         size_t data_len, int encrypt, const u8 *own_addr, u32 flags)
6322 {
6323         struct i802_bss *bss = priv;
6324         struct wpa_driver_nl80211_data *drv = bss->drv;
6325         struct ieee80211_hdr *hdr;
6326         size_t len;
6327         u8 *pos;
6328         int res;
6329         int qos = flags & WPA_STA_WMM;
6330
6331         if (drv->device_ap_sme || !drv->use_monitor)
6332                 return nl80211_send_eapol_data(bss, addr, data, data_len);
6333
6334         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
6335                 data_len;
6336         hdr = os_zalloc(len);
6337         if (hdr == NULL) {
6338                 printf("malloc() failed for i802_send_data(len=%lu)\n",
6339                        (unsigned long) len);
6340                 return -1;
6341         }
6342
6343         hdr->frame_control =
6344                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
6345         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
6346         if (encrypt)
6347                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
6348         if (qos) {
6349                 hdr->frame_control |=
6350                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
6351         }
6352
6353         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
6354         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
6355         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
6356         pos = (u8 *) (hdr + 1);
6357
6358         if (qos) {
6359                 /* add an empty QoS header if needed */
6360                 pos[0] = 0;
6361                 pos[1] = 0;
6362                 pos += 2;
6363         }
6364
6365         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
6366         pos += sizeof(rfc1042_header);
6367         WPA_PUT_BE16(pos, ETH_P_PAE);
6368         pos += 2;
6369         memcpy(pos, data, data_len);
6370
6371         res = wpa_driver_nl80211_send_frame(bss, (u8 *) hdr, len, encrypt, 0,
6372                                             0, 0, 0, 0);
6373         if (res < 0) {
6374                 wpa_printf(MSG_ERROR, "i802_send_eapol - packet len: %lu - "
6375                            "failed: %d (%s)",
6376                            (unsigned long) len, errno, strerror(errno));
6377         }
6378         os_free(hdr);
6379
6380         return res;
6381 }
6382
6383
6384 static int wpa_driver_nl80211_sta_set_flags(void *priv, const u8 *addr,
6385                                             int total_flags,
6386                                             int flags_or, int flags_and)
6387 {
6388         struct i802_bss *bss = priv;
6389         struct wpa_driver_nl80211_data *drv = bss->drv;
6390         struct nl_msg *msg, *flags = NULL;
6391         struct nl80211_sta_flag_update upd;
6392
6393         msg = nlmsg_alloc();
6394         if (!msg)
6395                 return -ENOMEM;
6396
6397         flags = nlmsg_alloc();
6398         if (!flags) {
6399                 nlmsg_free(msg);
6400                 return -ENOMEM;
6401         }
6402
6403         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
6404
6405         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
6406                     if_nametoindex(bss->ifname));
6407         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
6408
6409         /*
6410          * Backwards compatibility version using NL80211_ATTR_STA_FLAGS. This
6411          * can be removed eventually.
6412          */
6413         if (total_flags & WPA_STA_AUTHORIZED)
6414                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
6415
6416         if (total_flags & WPA_STA_WMM)
6417                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
6418
6419         if (total_flags & WPA_STA_SHORT_PREAMBLE)
6420                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
6421
6422         if (total_flags & WPA_STA_MFP)
6423                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
6424
6425         if (total_flags & WPA_STA_TDLS_PEER)
6426                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_TDLS_PEER);
6427
6428         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
6429                 goto nla_put_failure;
6430
6431         os_memset(&upd, 0, sizeof(upd));
6432         upd.mask = sta_flags_nl80211(flags_or | ~flags_and);
6433         upd.set = sta_flags_nl80211(flags_or);
6434         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
6435
6436         nlmsg_free(flags);
6437
6438         return send_and_recv_msgs(drv, msg, NULL, NULL);
6439  nla_put_failure:
6440         nlmsg_free(msg);
6441         nlmsg_free(flags);
6442         return -ENOBUFS;
6443 }
6444
6445
6446 static int wpa_driver_nl80211_ap(struct wpa_driver_nl80211_data *drv,
6447                                  struct wpa_driver_associate_params *params)
6448 {
6449         enum nl80211_iftype nlmode;
6450
6451         if (params->p2p) {
6452                 wpa_printf(MSG_DEBUG, "nl80211: Setup AP operations for P2P "
6453                            "group (GO)");
6454                 nlmode = NL80211_IFTYPE_P2P_GO;
6455         } else
6456                 nlmode = NL80211_IFTYPE_AP;
6457
6458         if (wpa_driver_nl80211_set_mode(&drv->first_bss, nlmode) ||
6459             wpa_driver_nl80211_set_freq(&drv->first_bss, params->freq, 0, 0)) {
6460                 nl80211_remove_monitor_interface(drv);
6461                 return -1;
6462         }
6463
6464         return 0;
6465 }
6466
6467
6468 static int nl80211_leave_ibss(struct wpa_driver_nl80211_data *drv)
6469 {
6470         struct nl_msg *msg;
6471         int ret = -1;
6472
6473         msg = nlmsg_alloc();
6474         if (!msg)
6475                 return -1;
6476
6477         nl80211_cmd(drv, msg, 0, NL80211_CMD_LEAVE_IBSS);
6478         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6479         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6480         msg = NULL;
6481         if (ret) {
6482                 wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS failed: ret=%d "
6483                            "(%s)", ret, strerror(-ret));
6484                 goto nla_put_failure;
6485         }
6486
6487         ret = 0;
6488         wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS request sent successfully");
6489
6490 nla_put_failure:
6491         nlmsg_free(msg);
6492         return ret;
6493 }
6494
6495
6496 static int wpa_driver_nl80211_ibss(struct wpa_driver_nl80211_data *drv,
6497                                    struct wpa_driver_associate_params *params)
6498 {
6499         struct nl_msg *msg;
6500         int ret = -1;
6501         int count = 0;
6502
6503         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS (ifindex=%d)", drv->ifindex);
6504
6505         if (wpa_driver_nl80211_set_mode(&drv->first_bss,
6506                                         NL80211_IFTYPE_ADHOC)) {
6507                 wpa_printf(MSG_INFO, "nl80211: Failed to set interface into "
6508                            "IBSS mode");
6509                 return -1;
6510         }
6511
6512 retry:
6513         msg = nlmsg_alloc();
6514         if (!msg)
6515                 return -1;
6516
6517         nl80211_cmd(drv, msg, 0, NL80211_CMD_JOIN_IBSS);
6518         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6519
6520         if (params->ssid == NULL || params->ssid_len > sizeof(drv->ssid))
6521                 goto nla_put_failure;
6522
6523         wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6524                           params->ssid, params->ssid_len);
6525         NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6526                 params->ssid);
6527         os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6528         drv->ssid_len = params->ssid_len;
6529
6530         wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6531         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6532
6533         ret = nl80211_set_conn_keys(params, msg);
6534         if (ret)
6535                 goto nla_put_failure;
6536
6537         if (params->bssid && params->fixed_bssid) {
6538                 wpa_printf(MSG_DEBUG, "  * BSSID=" MACSTR,
6539                            MAC2STR(params->bssid));
6540                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6541         }
6542
6543         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
6544             params->key_mgmt_suite == KEY_MGMT_PSK ||
6545             params->key_mgmt_suite == KEY_MGMT_802_1X_SHA256 ||
6546             params->key_mgmt_suite == KEY_MGMT_PSK_SHA256) {
6547                 wpa_printf(MSG_DEBUG, "  * control port");
6548                 NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT);
6549         }
6550
6551         if (params->wpa_ie) {
6552                 wpa_hexdump(MSG_DEBUG,
6553                             "  * Extra IEs for Beacon/Probe Response frames",
6554                             params->wpa_ie, params->wpa_ie_len);
6555                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6556                         params->wpa_ie);
6557         }
6558
6559         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6560         msg = NULL;
6561         if (ret) {
6562                 wpa_printf(MSG_DEBUG, "nl80211: Join IBSS failed: ret=%d (%s)",
6563                            ret, strerror(-ret));
6564                 count++;
6565                 if (ret == -EALREADY && count == 1) {
6566                         wpa_printf(MSG_DEBUG, "nl80211: Retry IBSS join after "
6567                                    "forced leave");
6568                         nl80211_leave_ibss(drv);
6569                         nlmsg_free(msg);
6570                         goto retry;
6571                 }
6572
6573                 goto nla_put_failure;
6574         }
6575         ret = 0;
6576         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS request sent successfully");
6577
6578 nla_put_failure:
6579         nlmsg_free(msg);
6580         return ret;
6581 }
6582
6583
6584 static unsigned int nl80211_get_assoc_bssid(struct wpa_driver_nl80211_data *drv,
6585                                             u8 *bssid)
6586 {
6587         struct nl_msg *msg;
6588         int ret;
6589         struct nl80211_bss_info_arg arg;
6590
6591         os_memset(&arg, 0, sizeof(arg));
6592         msg = nlmsg_alloc();
6593         if (!msg)
6594                 goto nla_put_failure;
6595
6596         nl80211_cmd(drv, msg, NLM_F_DUMP, NL80211_CMD_GET_SCAN);
6597         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6598
6599         arg.drv = drv;
6600         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
6601         msg = NULL;
6602         if (ret == 0) {
6603                 if (is_zero_ether_addr(arg.assoc_bssid))
6604                         return -ENOTCONN;
6605                 os_memcpy(bssid, arg.assoc_bssid, ETH_ALEN);
6606                 return 0;
6607         }
6608         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
6609                    "(%s)", ret, strerror(-ret));
6610 nla_put_failure:
6611         nlmsg_free(msg);
6612         return drv->assoc_freq;
6613 }
6614
6615
6616 static int nl80211_disconnect(struct wpa_driver_nl80211_data *drv,
6617                               const u8 *bssid)
6618 {
6619         u8 addr[ETH_ALEN];
6620
6621         if (bssid == NULL) {
6622                 int res = nl80211_get_assoc_bssid(drv, addr);
6623                 if (res)
6624                         return res;
6625                 bssid = addr;
6626         }
6627
6628         return wpa_driver_nl80211_disconnect(drv, bssid,
6629                                              WLAN_REASON_PREV_AUTH_NOT_VALID);
6630 }
6631
6632
6633 static int wpa_driver_nl80211_connect(
6634         struct wpa_driver_nl80211_data *drv,
6635         struct wpa_driver_associate_params *params)
6636 {
6637         struct nl_msg *msg;
6638         enum nl80211_auth_type type;
6639         int ret = 0;
6640         int algs;
6641
6642         msg = nlmsg_alloc();
6643         if (!msg)
6644                 return -1;
6645
6646         wpa_printf(MSG_DEBUG, "nl80211: Connect (ifindex=%d)", drv->ifindex);
6647         nl80211_cmd(drv, msg, 0, NL80211_CMD_CONNECT);
6648
6649         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6650         if (params->bssid) {
6651                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
6652                            MAC2STR(params->bssid));
6653                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6654         }
6655         if (params->freq) {
6656                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6657                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6658         }
6659         if (params->bg_scan_period >= 0) {
6660                 wpa_printf(MSG_DEBUG, "  * bg scan period=%d",
6661                            params->bg_scan_period);
6662                 NLA_PUT_U16(msg, NL80211_ATTR_BG_SCAN_PERIOD,
6663                             params->bg_scan_period);
6664         }
6665         if (params->ssid) {
6666                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6667                                   params->ssid, params->ssid_len);
6668                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6669                         params->ssid);
6670                 if (params->ssid_len > sizeof(drv->ssid))
6671                         goto nla_put_failure;
6672                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6673                 drv->ssid_len = params->ssid_len;
6674         }
6675         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
6676         if (params->wpa_ie)
6677                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6678                         params->wpa_ie);
6679
6680         algs = 0;
6681         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
6682                 algs++;
6683         if (params->auth_alg & WPA_AUTH_ALG_SHARED)
6684                 algs++;
6685         if (params->auth_alg & WPA_AUTH_ALG_LEAP)
6686                 algs++;
6687         if (algs > 1) {
6688                 wpa_printf(MSG_DEBUG, "  * Leave out Auth Type for automatic "
6689                            "selection");
6690                 goto skip_auth_type;
6691         }
6692
6693         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
6694                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
6695         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
6696                 type = NL80211_AUTHTYPE_SHARED_KEY;
6697         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
6698                 type = NL80211_AUTHTYPE_NETWORK_EAP;
6699         else if (params->auth_alg & WPA_AUTH_ALG_FT)
6700                 type = NL80211_AUTHTYPE_FT;
6701         else
6702                 goto nla_put_failure;
6703
6704         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
6705         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
6706
6707 skip_auth_type:
6708         if (params->wpa_proto) {
6709                 enum nl80211_wpa_versions ver = 0;
6710
6711                 if (params->wpa_proto & WPA_PROTO_WPA)
6712                         ver |= NL80211_WPA_VERSION_1;
6713                 if (params->wpa_proto & WPA_PROTO_RSN)
6714                         ver |= NL80211_WPA_VERSION_2;
6715
6716                 wpa_printf(MSG_DEBUG, "  * WPA Versions 0x%x", ver);
6717                 NLA_PUT_U32(msg, NL80211_ATTR_WPA_VERSIONS, ver);
6718         }
6719
6720         if (params->pairwise_suite != CIPHER_NONE) {
6721                 int cipher;
6722
6723                 switch (params->pairwise_suite) {
6724                 case CIPHER_WEP40:
6725                         cipher = WLAN_CIPHER_SUITE_WEP40;
6726                         break;
6727                 case CIPHER_WEP104:
6728                         cipher = WLAN_CIPHER_SUITE_WEP104;
6729                         break;
6730                 case CIPHER_CCMP:
6731                         cipher = WLAN_CIPHER_SUITE_CCMP;
6732                         break;
6733                 case CIPHER_TKIP:
6734                 default:
6735                         cipher = WLAN_CIPHER_SUITE_TKIP;
6736                         break;
6737                 }
6738                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE, cipher);
6739         }
6740
6741         if (params->group_suite != CIPHER_NONE) {
6742                 int cipher;
6743
6744                 switch (params->group_suite) {
6745                 case CIPHER_WEP40:
6746                         cipher = WLAN_CIPHER_SUITE_WEP40;
6747                         break;
6748                 case CIPHER_WEP104:
6749                         cipher = WLAN_CIPHER_SUITE_WEP104;
6750                         break;
6751                 case CIPHER_CCMP:
6752                         cipher = WLAN_CIPHER_SUITE_CCMP;
6753                         break;
6754                 case CIPHER_TKIP:
6755                 default:
6756                         cipher = WLAN_CIPHER_SUITE_TKIP;
6757                         break;
6758                 }
6759                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP, cipher);
6760         }
6761
6762         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
6763             params->key_mgmt_suite == KEY_MGMT_PSK) {
6764                 int mgmt = WLAN_AKM_SUITE_PSK;
6765
6766                 switch (params->key_mgmt_suite) {
6767                 case KEY_MGMT_802_1X:
6768                         mgmt = WLAN_AKM_SUITE_8021X;
6769                         break;
6770                 case KEY_MGMT_PSK:
6771                 default:
6772                         mgmt = WLAN_AKM_SUITE_PSK;
6773                         break;
6774                 }
6775                 NLA_PUT_U32(msg, NL80211_ATTR_AKM_SUITES, mgmt);
6776         }
6777
6778         if (params->disable_ht)
6779                 NLA_PUT_FLAG(msg, NL80211_ATTR_DISABLE_HT);
6780
6781         if (params->htcaps && params->htcaps_mask) {
6782                 int sz = sizeof(struct ieee80211_ht_capabilities);
6783                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY, sz, params->htcaps);
6784                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY_MASK, sz,
6785                         params->htcaps_mask);
6786         }
6787
6788         ret = nl80211_set_conn_keys(params, msg);
6789         if (ret)
6790                 goto nla_put_failure;
6791
6792         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6793         msg = NULL;
6794         if (ret) {
6795                 wpa_printf(MSG_DEBUG, "nl80211: MLME connect failed: ret=%d "
6796                            "(%s)", ret, strerror(-ret));
6797                 /*
6798                  * cfg80211 does not currently accept new connection if we are
6799                  * already connected. As a workaround, force disconnection and
6800                  * try again once the driver indicates it completed
6801                  * disconnection.
6802                  */
6803                 if (ret == -EALREADY)
6804                         nl80211_disconnect(drv, params->bssid);
6805                 goto nla_put_failure;
6806         }
6807         ret = 0;
6808         wpa_printf(MSG_DEBUG, "nl80211: Connect request send successfully");
6809
6810 nla_put_failure:
6811         nlmsg_free(msg);
6812         return ret;
6813
6814 }
6815
6816
6817 static int wpa_driver_nl80211_associate(
6818         void *priv, struct wpa_driver_associate_params *params)
6819 {
6820         struct i802_bss *bss = priv;
6821         struct wpa_driver_nl80211_data *drv = bss->drv;
6822         int ret = -1;
6823         struct nl_msg *msg;
6824
6825         if (params->mode == IEEE80211_MODE_AP)
6826                 return wpa_driver_nl80211_ap(drv, params);
6827
6828         if (params->mode == IEEE80211_MODE_IBSS)
6829                 return wpa_driver_nl80211_ibss(drv, params);
6830
6831         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME)) {
6832                 enum nl80211_iftype nlmode = params->p2p ?
6833                         NL80211_IFTYPE_P2P_CLIENT : NL80211_IFTYPE_STATION;
6834
6835                 if (wpa_driver_nl80211_set_mode(priv, nlmode) < 0)
6836                         return -1;
6837                 return wpa_driver_nl80211_connect(drv, params);
6838         }
6839
6840         drv->associated = 0;
6841
6842         msg = nlmsg_alloc();
6843         if (!msg)
6844                 return -1;
6845
6846         wpa_printf(MSG_DEBUG, "nl80211: Associate (ifindex=%d)",
6847                    drv->ifindex);
6848         nl80211_cmd(drv, msg, 0, NL80211_CMD_ASSOCIATE);
6849
6850         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
6851         if (params->bssid) {
6852                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
6853                            MAC2STR(params->bssid));
6854                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
6855         }
6856         if (params->freq) {
6857                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
6858                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
6859                 drv->assoc_freq = params->freq;
6860         } else
6861                 drv->assoc_freq = 0;
6862         if (params->bg_scan_period >= 0) {
6863                 wpa_printf(MSG_DEBUG, "  * bg scan period=%d",
6864                            params->bg_scan_period);
6865                 NLA_PUT_U16(msg, NL80211_ATTR_BG_SCAN_PERIOD,
6866                             params->bg_scan_period);
6867         }
6868         if (params->ssid) {
6869                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
6870                                   params->ssid, params->ssid_len);
6871                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
6872                         params->ssid);
6873                 if (params->ssid_len > sizeof(drv->ssid))
6874                         goto nla_put_failure;
6875                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
6876                 drv->ssid_len = params->ssid_len;
6877         }
6878         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
6879         if (params->wpa_ie)
6880                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
6881                         params->wpa_ie);
6882
6883         if (params->pairwise_suite != CIPHER_NONE) {
6884                 int cipher;
6885
6886                 switch (params->pairwise_suite) {
6887                 case CIPHER_WEP40:
6888                         cipher = WLAN_CIPHER_SUITE_WEP40;
6889                         break;
6890                 case CIPHER_WEP104:
6891                         cipher = WLAN_CIPHER_SUITE_WEP104;
6892                         break;
6893                 case CIPHER_CCMP:
6894                         cipher = WLAN_CIPHER_SUITE_CCMP;
6895                         break;
6896                 case CIPHER_TKIP:
6897                 default:
6898                         cipher = WLAN_CIPHER_SUITE_TKIP;
6899                         break;
6900                 }
6901                 wpa_printf(MSG_DEBUG, "  * pairwise=0x%x", cipher);
6902                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE, cipher);
6903         }
6904
6905         if (params->group_suite != CIPHER_NONE) {
6906                 int cipher;
6907
6908                 switch (params->group_suite) {
6909                 case CIPHER_WEP40:
6910                         cipher = WLAN_CIPHER_SUITE_WEP40;
6911                         break;
6912                 case CIPHER_WEP104:
6913                         cipher = WLAN_CIPHER_SUITE_WEP104;
6914                         break;
6915                 case CIPHER_CCMP:
6916                         cipher = WLAN_CIPHER_SUITE_CCMP;
6917                         break;
6918                 case CIPHER_TKIP:
6919                 default:
6920                         cipher = WLAN_CIPHER_SUITE_TKIP;
6921                         break;
6922                 }
6923                 wpa_printf(MSG_DEBUG, "  * group=0x%x", cipher);
6924                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP, cipher);
6925         }
6926
6927 #ifdef CONFIG_IEEE80211W
6928         if (params->mgmt_frame_protection == MGMT_FRAME_PROTECTION_REQUIRED)
6929                 NLA_PUT_U32(msg, NL80211_ATTR_USE_MFP, NL80211_MFP_REQUIRED);
6930 #endif /* CONFIG_IEEE80211W */
6931
6932         NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT);
6933
6934         if (params->prev_bssid) {
6935                 wpa_printf(MSG_DEBUG, "  * prev_bssid=" MACSTR,
6936                            MAC2STR(params->prev_bssid));
6937                 NLA_PUT(msg, NL80211_ATTR_PREV_BSSID, ETH_ALEN,
6938                         params->prev_bssid);
6939         }
6940
6941         if (params->disable_ht)
6942                 NLA_PUT_FLAG(msg, NL80211_ATTR_DISABLE_HT);
6943
6944         if (params->htcaps && params->htcaps_mask) {
6945                 int sz = sizeof(struct ieee80211_ht_capabilities);
6946                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY, sz, params->htcaps);
6947                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY_MASK, sz,
6948                         params->htcaps_mask);
6949         }
6950
6951         if (params->p2p)
6952                 wpa_printf(MSG_DEBUG, "  * P2P group");
6953
6954         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6955         msg = NULL;
6956         if (ret) {
6957                 wpa_dbg(drv->ctx, MSG_DEBUG,
6958                         "nl80211: MLME command failed (assoc): ret=%d (%s)",
6959                         ret, strerror(-ret));
6960                 nl80211_dump_scan(drv);
6961                 goto nla_put_failure;
6962         }
6963         ret = 0;
6964         wpa_printf(MSG_DEBUG, "nl80211: Association request send "
6965                    "successfully");
6966
6967 nla_put_failure:
6968         nlmsg_free(msg);
6969         return ret;
6970 }
6971
6972
6973 static int nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
6974                             int ifindex, enum nl80211_iftype mode)
6975 {
6976         struct nl_msg *msg;
6977         int ret = -ENOBUFS;
6978
6979         wpa_printf(MSG_DEBUG, "nl80211: Set mode ifindex %d iftype %d (%s)",
6980                    ifindex, mode, nl80211_iftype_str(mode));
6981
6982         msg = nlmsg_alloc();
6983         if (!msg)
6984                 return -ENOMEM;
6985
6986         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_INTERFACE);
6987         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
6988         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
6989
6990         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
6991         msg = NULL;
6992         if (!ret)
6993                 return 0;
6994 nla_put_failure:
6995         nlmsg_free(msg);
6996         wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface %d to mode %d:"
6997                    " %d (%s)", ifindex, mode, ret, strerror(-ret));
6998         return ret;
6999 }
7000
7001
7002 static int wpa_driver_nl80211_set_mode(struct i802_bss *bss,
7003                                        enum nl80211_iftype nlmode)
7004 {
7005         struct wpa_driver_nl80211_data *drv = bss->drv;
7006         int ret = -1;
7007         int i;
7008         int was_ap = is_ap_interface(drv->nlmode);
7009         int res;
7010
7011         res = nl80211_set_mode(drv, drv->ifindex, nlmode);
7012         if (res == 0) {
7013                 drv->nlmode = nlmode;
7014                 ret = 0;
7015                 goto done;
7016         }
7017
7018         if (res == -ENODEV)
7019                 return -1;
7020
7021         if (nlmode == drv->nlmode) {
7022                 wpa_printf(MSG_DEBUG, "nl80211: Interface already in "
7023                            "requested mode - ignore error");
7024                 ret = 0;
7025                 goto done; /* Already in the requested mode */
7026         }
7027
7028         /* mac80211 doesn't allow mode changes while the device is up, so
7029          * take the device down, try to set the mode again, and bring the
7030          * device back up.
7031          */
7032         wpa_printf(MSG_DEBUG, "nl80211: Try mode change after setting "
7033                    "interface down");
7034         for (i = 0; i < 10; i++) {
7035                 res = linux_set_iface_flags(drv->global->ioctl_sock,
7036                                             bss->ifname, 0);
7037                 if (res == -EACCES || res == -ENODEV)
7038                         break;
7039                 if (res == 0) {
7040                         /* Try to set the mode again while the interface is
7041                          * down */
7042                         ret = nl80211_set_mode(drv, drv->ifindex, nlmode);
7043                         if (ret == -EACCES)
7044                                 break;
7045                         res = linux_set_iface_flags(drv->global->ioctl_sock,
7046                                                     bss->ifname, 1);
7047                         if (res && !ret)
7048                                 ret = -1;
7049                         else if (ret != -EBUSY)
7050                                 break;
7051                 } else
7052                         wpa_printf(MSG_DEBUG, "nl80211: Failed to set "
7053                                    "interface down");
7054                 os_sleep(0, 100000);
7055         }
7056
7057         if (!ret) {
7058                 wpa_printf(MSG_DEBUG, "nl80211: Mode change succeeded while "
7059                            "interface is down");
7060                 drv->nlmode = nlmode;
7061                 drv->ignore_if_down_event = 1;
7062         }
7063
7064 done:
7065         if (ret) {
7066                 wpa_printf(MSG_DEBUG, "nl80211: Interface mode change to %d "
7067                            "from %d failed", nlmode, drv->nlmode);
7068                 return ret;
7069         }
7070
7071         if (is_ap_interface(nlmode)) {
7072                 nl80211_mgmt_unsubscribe(bss, "start AP");
7073                 /* Setup additional AP mode functionality if needed */
7074                 if (nl80211_setup_ap(bss))
7075                         return -1;
7076         } else if (was_ap) {
7077                 /* Remove additional AP mode functionality */
7078                 nl80211_teardown_ap(bss);
7079         } else {
7080                 nl80211_mgmt_unsubscribe(bss, "mode change");
7081         }
7082
7083         if (!bss->in_deinit && !is_ap_interface(nlmode) &&
7084             nl80211_mgmt_subscribe_non_ap(bss) < 0)
7085                 wpa_printf(MSG_DEBUG, "nl80211: Failed to register Action "
7086                            "frame processing - ignore for now");
7087
7088         return 0;
7089 }
7090
7091
7092 static int wpa_driver_nl80211_get_capa(void *priv,
7093                                        struct wpa_driver_capa *capa)
7094 {
7095         struct i802_bss *bss = priv;
7096         struct wpa_driver_nl80211_data *drv = bss->drv;
7097         if (!drv->has_capability)
7098                 return -1;
7099         os_memcpy(capa, &drv->capa, sizeof(*capa));
7100         return 0;
7101 }
7102
7103
7104 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
7105 {
7106         struct i802_bss *bss = priv;
7107         struct wpa_driver_nl80211_data *drv = bss->drv;
7108
7109         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
7110                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
7111         drv->operstate = state;
7112         return netlink_send_oper_ifla(drv->global->netlink, drv->ifindex, -1,
7113                                       state ? IF_OPER_UP : IF_OPER_DORMANT);
7114 }
7115
7116
7117 static int wpa_driver_nl80211_set_supp_port(void *priv, int authorized)
7118 {
7119         struct i802_bss *bss = priv;
7120         struct wpa_driver_nl80211_data *drv = bss->drv;
7121         struct nl_msg *msg;
7122         struct nl80211_sta_flag_update upd;
7123
7124         msg = nlmsg_alloc();
7125         if (!msg)
7126                 return -ENOMEM;
7127
7128         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
7129
7130         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7131                     if_nametoindex(bss->ifname));
7132         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, drv->bssid);
7133
7134         os_memset(&upd, 0, sizeof(upd));
7135         upd.mask = BIT(NL80211_STA_FLAG_AUTHORIZED);
7136         if (authorized)
7137                 upd.set = BIT(NL80211_STA_FLAG_AUTHORIZED);
7138         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
7139
7140         return send_and_recv_msgs(drv, msg, NULL, NULL);
7141  nla_put_failure:
7142         nlmsg_free(msg);
7143         return -ENOBUFS;
7144 }
7145
7146
7147 /* Set kernel driver on given frequency (MHz) */
7148 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq)
7149 {
7150         struct i802_bss *bss = priv;
7151         return wpa_driver_nl80211_set_freq(bss, freq->freq, freq->ht_enabled,
7152                                            freq->sec_channel_offset);
7153 }
7154
7155
7156 #if defined(HOSTAPD) || defined(CONFIG_AP)
7157
7158 static inline int min_int(int a, int b)
7159 {
7160         if (a < b)
7161                 return a;
7162         return b;
7163 }
7164
7165
7166 static int get_key_handler(struct nl_msg *msg, void *arg)
7167 {
7168         struct nlattr *tb[NL80211_ATTR_MAX + 1];
7169         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
7170
7171         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
7172                   genlmsg_attrlen(gnlh, 0), NULL);
7173
7174         /*
7175          * TODO: validate the key index and mac address!
7176          * Otherwise, there's a race condition as soon as
7177          * the kernel starts sending key notifications.
7178          */
7179
7180         if (tb[NL80211_ATTR_KEY_SEQ])
7181                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
7182                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
7183         return NL_SKIP;
7184 }
7185
7186
7187 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
7188                            int idx, u8 *seq)
7189 {
7190         struct i802_bss *bss = priv;
7191         struct wpa_driver_nl80211_data *drv = bss->drv;
7192         struct nl_msg *msg;
7193
7194         msg = nlmsg_alloc();
7195         if (!msg)
7196                 return -ENOMEM;
7197
7198         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_KEY);
7199
7200         if (addr)
7201                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7202         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
7203         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
7204
7205         memset(seq, 0, 6);
7206
7207         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
7208  nla_put_failure:
7209         nlmsg_free(msg);
7210         return -ENOBUFS;
7211 }
7212
7213
7214 static int i802_set_rts(void *priv, int rts)
7215 {
7216         struct i802_bss *bss = priv;
7217         struct wpa_driver_nl80211_data *drv = bss->drv;
7218         struct nl_msg *msg;
7219         int ret = -ENOBUFS;
7220         u32 val;
7221
7222         msg = nlmsg_alloc();
7223         if (!msg)
7224                 return -ENOMEM;
7225
7226         if (rts >= 2347)
7227                 val = (u32) -1;
7228         else
7229                 val = rts;
7230
7231         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7232         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
7233         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD, val);
7234
7235         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7236         msg = NULL;
7237         if (!ret)
7238                 return 0;
7239 nla_put_failure:
7240         nlmsg_free(msg);
7241         wpa_printf(MSG_DEBUG, "nl80211: Failed to set RTS threshold %d: "
7242                    "%d (%s)", rts, ret, strerror(-ret));
7243         return ret;
7244 }
7245
7246
7247 static int i802_set_frag(void *priv, int frag)
7248 {
7249         struct i802_bss *bss = priv;
7250         struct wpa_driver_nl80211_data *drv = bss->drv;
7251         struct nl_msg *msg;
7252         int ret = -ENOBUFS;
7253         u32 val;
7254
7255         msg = nlmsg_alloc();
7256         if (!msg)
7257                 return -ENOMEM;
7258
7259         if (frag >= 2346)
7260                 val = (u32) -1;
7261         else
7262                 val = frag;
7263
7264         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7265         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
7266         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD, val);
7267
7268         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7269         msg = NULL;
7270         if (!ret)
7271                 return 0;
7272 nla_put_failure:
7273         nlmsg_free(msg);
7274         wpa_printf(MSG_DEBUG, "nl80211: Failed to set fragmentation threshold "
7275                    "%d: %d (%s)", frag, ret, strerror(-ret));
7276         return ret;
7277 }
7278
7279
7280 static int i802_flush(void *priv)
7281 {
7282         struct i802_bss *bss = priv;
7283         struct wpa_driver_nl80211_data *drv = bss->drv;
7284         struct nl_msg *msg;
7285         int res;
7286
7287         msg = nlmsg_alloc();
7288         if (!msg)
7289                 return -1;
7290
7291         nl80211_cmd(drv, msg, 0, NL80211_CMD_DEL_STATION);
7292
7293         /*
7294          * XXX: FIX! this needs to flush all VLANs too
7295          */
7296         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7297                     if_nametoindex(bss->ifname));
7298
7299         res = send_and_recv_msgs(drv, msg, NULL, NULL);
7300         if (res) {
7301                 wpa_printf(MSG_DEBUG, "nl80211: Station flush failed: ret=%d "
7302                            "(%s)", res, strerror(-res));
7303         }
7304         return res;
7305  nla_put_failure:
7306         nlmsg_free(msg);
7307         return -ENOBUFS;
7308 }
7309
7310
7311 static int get_sta_handler(struct nl_msg *msg, void *arg)
7312 {
7313         struct nlattr *tb[NL80211_ATTR_MAX + 1];
7314         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
7315         struct hostap_sta_driver_data *data = arg;
7316         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
7317         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
7318                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
7319                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
7320                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
7321                 [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
7322                 [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
7323         };
7324
7325         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
7326                   genlmsg_attrlen(gnlh, 0), NULL);
7327
7328         /*
7329          * TODO: validate the interface and mac address!
7330          * Otherwise, there's a race condition as soon as
7331          * the kernel starts sending station notifications.
7332          */
7333
7334         if (!tb[NL80211_ATTR_STA_INFO]) {
7335                 wpa_printf(MSG_DEBUG, "sta stats missing!");
7336                 return NL_SKIP;
7337         }
7338         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
7339                              tb[NL80211_ATTR_STA_INFO],
7340                              stats_policy)) {
7341                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
7342                 return NL_SKIP;
7343         }
7344
7345         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
7346                 data->inactive_msec =
7347                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
7348         if (stats[NL80211_STA_INFO_RX_BYTES])
7349                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
7350         if (stats[NL80211_STA_INFO_TX_BYTES])
7351                 data->tx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
7352         if (stats[NL80211_STA_INFO_RX_PACKETS])
7353                 data->rx_packets =
7354                         nla_get_u32(stats[NL80211_STA_INFO_RX_PACKETS]);
7355         if (stats[NL80211_STA_INFO_TX_PACKETS])
7356                 data->tx_packets =
7357                         nla_get_u32(stats[NL80211_STA_INFO_TX_PACKETS]);
7358
7359         return NL_SKIP;
7360 }
7361
7362 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
7363                               const u8 *addr)
7364 {
7365         struct i802_bss *bss = priv;
7366         struct wpa_driver_nl80211_data *drv = bss->drv;
7367         struct nl_msg *msg;
7368
7369         os_memset(data, 0, sizeof(*data));
7370         msg = nlmsg_alloc();
7371         if (!msg)
7372                 return -ENOMEM;
7373
7374         nl80211_cmd(drv, msg, 0, NL80211_CMD_GET_STATION);
7375
7376         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7377         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
7378
7379         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
7380  nla_put_failure:
7381         nlmsg_free(msg);
7382         return -ENOBUFS;
7383 }
7384
7385
7386 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
7387                                     int cw_min, int cw_max, int burst_time)
7388 {
7389         struct i802_bss *bss = priv;
7390         struct wpa_driver_nl80211_data *drv = bss->drv;
7391         struct nl_msg *msg;
7392         struct nlattr *txq, *params;
7393
7394         msg = nlmsg_alloc();
7395         if (!msg)
7396                 return -1;
7397
7398         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_WIPHY);
7399
7400         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
7401
7402         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
7403         if (!txq)
7404                 goto nla_put_failure;
7405
7406         /* We are only sending parameters for a single TXQ at a time */
7407         params = nla_nest_start(msg, 1);
7408         if (!params)
7409                 goto nla_put_failure;
7410
7411         switch (queue) {
7412         case 0:
7413                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_VO);
7414                 break;
7415         case 1:
7416                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_VI);
7417                 break;
7418         case 2:
7419                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_BE);
7420                 break;
7421         case 3:
7422                 NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, NL80211_TXQ_Q_BK);
7423                 break;
7424         }
7425         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
7426          * 32 usec, so need to convert the value here. */
7427         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
7428         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
7429         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
7430         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
7431
7432         nla_nest_end(msg, params);
7433
7434         nla_nest_end(msg, txq);
7435
7436         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
7437                 return 0;
7438         msg = NULL;
7439  nla_put_failure:
7440         nlmsg_free(msg);
7441         return -1;
7442 }
7443
7444
7445 static int i802_set_sta_vlan(void *priv, const u8 *addr,
7446                              const char *ifname, int vlan_id)
7447 {
7448         struct i802_bss *bss = priv;
7449         struct wpa_driver_nl80211_data *drv = bss->drv;
7450         struct nl_msg *msg;
7451         int ret = -ENOBUFS;
7452
7453         msg = nlmsg_alloc();
7454         if (!msg)
7455                 return -ENOMEM;
7456
7457         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_STATION);
7458
7459         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
7460                     if_nametoindex(bss->ifname));
7461         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
7462         NLA_PUT_U32(msg, NL80211_ATTR_STA_VLAN,
7463                     if_nametoindex(ifname));
7464
7465         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
7466         msg = NULL;
7467         if (ret < 0) {
7468                 wpa_printf(MSG_ERROR, "nl80211: NL80211_ATTR_STA_VLAN (addr="
7469                            MACSTR " ifname=%s vlan_id=%d) failed: %d (%s)",
7470                            MAC2STR(addr), ifname, vlan_id, ret,
7471                            strerror(-ret));
7472         }
7473  nla_put_failure:
7474         nlmsg_free(msg);
7475         return ret;
7476 }
7477
7478
7479 static int i802_get_inact_sec(void *priv, const u8 *addr)
7480 {
7481         struct hostap_sta_driver_data data;
7482         int ret;
7483
7484         data.inactive_msec = (unsigned long) -1;
7485         ret = i802_read_sta_data(priv, &data, addr);
7486         if (ret || data.inactive_msec == (unsigned long) -1)
7487                 return -1;
7488         return data.inactive_msec / 1000;
7489 }
7490
7491
7492 static int i802_sta_clear_stats(void *priv, const u8 *addr)
7493 {
7494 #if 0
7495         /* TODO */
7496 #endif
7497         return 0;
7498 }
7499
7500
7501 static int i802_sta_deauth(void *priv, const u8 *own_addr, const u8 *addr,
7502                            int reason)
7503 {
7504         struct i802_bss *bss = priv;
7505         struct wpa_driver_nl80211_data *drv = bss->drv;
7506         struct ieee80211_mgmt mgmt;
7507
7508         if (drv->device_ap_sme)
7509                 return wpa_driver_nl80211_sta_remove(bss, addr);
7510
7511         memset(&mgmt, 0, sizeof(mgmt));
7512         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
7513                                           WLAN_FC_STYPE_DEAUTH);
7514         memcpy(mgmt.da, addr, ETH_ALEN);
7515         memcpy(mgmt.sa, own_addr, ETH_ALEN);
7516         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
7517         mgmt.u.deauth.reason_code = host_to_le16(reason);
7518         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
7519                                             IEEE80211_HDRLEN +
7520                                             sizeof(mgmt.u.deauth), 0);
7521 }
7522
7523
7524 static int i802_sta_disassoc(void *priv, const u8 *own_addr, const u8 *addr,
7525                              int reason)
7526 {
7527         struct i802_bss *bss = priv;
7528         struct wpa_driver_nl80211_data *drv = bss->drv;
7529         struct ieee80211_mgmt mgmt;
7530
7531         if (drv->device_ap_sme)
7532                 return wpa_driver_nl80211_sta_remove(bss, addr);
7533
7534         memset(&mgmt, 0, sizeof(mgmt));
7535         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
7536                                           WLAN_FC_STYPE_DISASSOC);
7537         memcpy(mgmt.da, addr, ETH_ALEN);
7538         memcpy(mgmt.sa, own_addr, ETH_ALEN);
7539         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
7540         mgmt.u.disassoc.reason_code = host_to_le16(reason);
7541         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
7542                                             IEEE80211_HDRLEN +
7543                                             sizeof(mgmt.u.disassoc), 0);
7544 }
7545
7546 #endif /* HOSTAPD || CONFIG_AP */
7547
7548 #ifdef HOSTAPD
7549
7550 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7551 {
7552         int i;
7553         int *old;
7554
7555         wpa_printf(MSG_DEBUG, "nl80211: Add own interface ifindex %d",
7556                    ifidx);
7557         for (i = 0; i < drv->num_if_indices; i++) {
7558                 if (drv->if_indices[i] == 0) {
7559                         drv->if_indices[i] = ifidx;
7560                         return;
7561                 }
7562         }
7563
7564         if (drv->if_indices != drv->default_if_indices)
7565                 old = drv->if_indices;
7566         else
7567                 old = NULL;
7568
7569         drv->if_indices = os_realloc(old,
7570                                      sizeof(int) * (drv->num_if_indices + 1));
7571         if (!drv->if_indices) {
7572                 if (!old)
7573                         drv->if_indices = drv->default_if_indices;
7574                 else
7575                         drv->if_indices = old;
7576                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
7577                            "interfaces");
7578                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
7579                 return;
7580         } else if (!old)
7581                 os_memcpy(drv->if_indices, drv->default_if_indices,
7582                           sizeof(drv->default_if_indices));
7583         drv->if_indices[drv->num_if_indices] = ifidx;
7584         drv->num_if_indices++;
7585 }
7586
7587
7588 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7589 {
7590         int i;
7591
7592         for (i = 0; i < drv->num_if_indices; i++) {
7593                 if (drv->if_indices[i] == ifidx) {
7594                         drv->if_indices[i] = 0;
7595                         break;
7596                 }
7597         }
7598 }
7599
7600
7601 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
7602 {
7603         int i;
7604
7605         for (i = 0; i < drv->num_if_indices; i++)
7606                 if (drv->if_indices[i] == ifidx)
7607                         return 1;
7608
7609         return 0;
7610 }
7611
7612
7613 static int i802_set_wds_sta(void *priv, const u8 *addr, int aid, int val,
7614                             const char *bridge_ifname)
7615 {
7616         struct i802_bss *bss = priv;
7617         struct wpa_driver_nl80211_data *drv = bss->drv;
7618         char name[IFNAMSIZ + 1];
7619
7620         os_snprintf(name, sizeof(name), "%s.sta%d", bss->ifname, aid);
7621         wpa_printf(MSG_DEBUG, "nl80211: Set WDS STA addr=" MACSTR
7622                    " aid=%d val=%d name=%s", MAC2STR(addr), aid, val, name);
7623         if (val) {
7624                 if (!if_nametoindex(name)) {
7625                         if (nl80211_create_iface(drv, name,
7626                                                  NL80211_IFTYPE_AP_VLAN,
7627                                                  NULL, 1) < 0)
7628                                 return -1;
7629                         if (bridge_ifname &&
7630                             linux_br_add_if(drv->global->ioctl_sock,
7631                                             bridge_ifname, name) < 0)
7632                                 return -1;
7633                 }
7634                 linux_set_iface_flags(drv->global->ioctl_sock, name, 1);
7635                 return i802_set_sta_vlan(priv, addr, name, 0);
7636         } else {
7637                 i802_set_sta_vlan(priv, addr, bss->ifname, 0);
7638                 return wpa_driver_nl80211_if_remove(priv, WPA_IF_AP_VLAN,
7639                                                     name);
7640         }
7641 }
7642
7643
7644 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
7645 {
7646         struct wpa_driver_nl80211_data *drv = eloop_ctx;
7647         struct sockaddr_ll lladdr;
7648         unsigned char buf[3000];
7649         int len;
7650         socklen_t fromlen = sizeof(lladdr);
7651
7652         len = recvfrom(sock, buf, sizeof(buf), 0,
7653                        (struct sockaddr *)&lladdr, &fromlen);
7654         if (len < 0) {
7655                 perror("recv");
7656                 return;
7657         }
7658
7659         if (have_ifidx(drv, lladdr.sll_ifindex))
7660                 drv_event_eapol_rx(drv->ctx, lladdr.sll_addr, buf, len);
7661 }
7662
7663
7664 static int i802_check_bridge(struct wpa_driver_nl80211_data *drv,
7665                              struct i802_bss *bss,
7666                              const char *brname, const char *ifname)
7667 {
7668         int ifindex;
7669         char in_br[IFNAMSIZ];
7670
7671         os_strlcpy(bss->brname, brname, IFNAMSIZ);
7672         ifindex = if_nametoindex(brname);
7673         if (ifindex == 0) {
7674                 /*
7675                  * Bridge was configured, but the bridge device does
7676                  * not exist. Try to add it now.
7677                  */
7678                 if (linux_br_add(drv->global->ioctl_sock, brname) < 0) {
7679                         wpa_printf(MSG_ERROR, "nl80211: Failed to add the "
7680                                    "bridge interface %s: %s",
7681                                    brname, strerror(errno));
7682                         return -1;
7683                 }
7684                 bss->added_bridge = 1;
7685                 add_ifidx(drv, if_nametoindex(brname));
7686         }
7687
7688         if (linux_br_get(in_br, ifname) == 0) {
7689                 if (os_strcmp(in_br, brname) == 0)
7690                         return 0; /* already in the bridge */
7691
7692                 wpa_printf(MSG_DEBUG, "nl80211: Removing interface %s from "
7693                            "bridge %s", ifname, in_br);
7694                 if (linux_br_del_if(drv->global->ioctl_sock, in_br, ifname) <
7695                     0) {
7696                         wpa_printf(MSG_ERROR, "nl80211: Failed to "
7697                                    "remove interface %s from bridge "
7698                                    "%s: %s",
7699                                    ifname, brname, strerror(errno));
7700                         return -1;
7701                 }
7702         }
7703
7704         wpa_printf(MSG_DEBUG, "nl80211: Adding interface %s into bridge %s",
7705                    ifname, brname);
7706         if (linux_br_add_if(drv->global->ioctl_sock, brname, ifname) < 0) {
7707                 wpa_printf(MSG_ERROR, "nl80211: Failed to add interface %s "
7708                            "into bridge %s: %s",
7709                            ifname, brname, strerror(errno));
7710                 return -1;
7711         }
7712         bss->added_if_into_bridge = 1;
7713
7714         return 0;
7715 }
7716
7717
7718 static void *i802_init(struct hostapd_data *hapd,
7719                        struct wpa_init_params *params)
7720 {
7721         struct wpa_driver_nl80211_data *drv;
7722         struct i802_bss *bss;
7723         size_t i;
7724         char brname[IFNAMSIZ];
7725         int ifindex, br_ifindex;
7726         int br_added = 0;
7727
7728         bss = wpa_driver_nl80211_init(hapd, params->ifname,
7729                                       params->global_priv);
7730         if (bss == NULL)
7731                 return NULL;
7732
7733         drv = bss->drv;
7734         drv->nlmode = NL80211_IFTYPE_AP;
7735         drv->eapol_sock = -1;
7736
7737         if (linux_br_get(brname, params->ifname) == 0) {
7738                 wpa_printf(MSG_DEBUG, "nl80211: Interface %s is in bridge %s",
7739                            params->ifname, brname);
7740                 br_ifindex = if_nametoindex(brname);
7741         } else {
7742                 brname[0] = '\0';
7743                 br_ifindex = 0;
7744         }
7745
7746         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
7747         drv->if_indices = drv->default_if_indices;
7748         for (i = 0; i < params->num_bridge; i++) {
7749                 if (params->bridge[i]) {
7750                         ifindex = if_nametoindex(params->bridge[i]);
7751                         if (ifindex)
7752                                 add_ifidx(drv, ifindex);
7753                         if (ifindex == br_ifindex)
7754                                 br_added = 1;
7755                 }
7756         }
7757         if (!br_added && br_ifindex &&
7758             (params->num_bridge == 0 || !params->bridge[0]))
7759                 add_ifidx(drv, br_ifindex);
7760
7761         /* start listening for EAPOL on the default AP interface */
7762         add_ifidx(drv, drv->ifindex);
7763
7764         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 0))
7765                 goto failed;
7766
7767         if (params->bssid) {
7768                 if (linux_set_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7769                                        params->bssid))
7770                         goto failed;
7771         }
7772
7773         if (wpa_driver_nl80211_set_mode(bss, drv->nlmode)) {
7774                 wpa_printf(MSG_ERROR, "nl80211: Failed to set interface %s "
7775                            "into AP mode", bss->ifname);
7776                 goto failed;
7777         }
7778
7779         if (params->num_bridge && params->bridge[0] &&
7780             i802_check_bridge(drv, bss, params->bridge[0], params->ifname) < 0)
7781                 goto failed;
7782
7783         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 1))
7784                 goto failed;
7785
7786         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
7787         if (drv->eapol_sock < 0) {
7788                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
7789                 goto failed;
7790         }
7791
7792         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
7793         {
7794                 printf("Could not register read socket for eapol\n");
7795                 goto failed;
7796         }
7797
7798         if (linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7799                                params->own_addr))
7800                 goto failed;
7801
7802         memcpy(bss->addr, params->own_addr, ETH_ALEN);
7803
7804         return bss;
7805
7806 failed:
7807         wpa_driver_nl80211_deinit(bss);
7808         return NULL;
7809 }
7810
7811
7812 static void i802_deinit(void *priv)
7813 {
7814         wpa_driver_nl80211_deinit(priv);
7815 }
7816
7817 #endif /* HOSTAPD */
7818
7819
7820 static enum nl80211_iftype wpa_driver_nl80211_if_type(
7821         enum wpa_driver_if_type type)
7822 {
7823         switch (type) {
7824         case WPA_IF_STATION:
7825                 return NL80211_IFTYPE_STATION;
7826         case WPA_IF_P2P_CLIENT:
7827         case WPA_IF_P2P_GROUP:
7828                 return NL80211_IFTYPE_P2P_CLIENT;
7829         case WPA_IF_AP_VLAN:
7830                 return NL80211_IFTYPE_AP_VLAN;
7831         case WPA_IF_AP_BSS:
7832                 return NL80211_IFTYPE_AP;
7833         case WPA_IF_P2P_GO:
7834                 return NL80211_IFTYPE_P2P_GO;
7835         }
7836         return -1;
7837 }
7838
7839
7840 #ifdef CONFIG_P2P
7841
7842 static int nl80211_addr_in_use(struct nl80211_global *global, const u8 *addr)
7843 {
7844         struct wpa_driver_nl80211_data *drv;
7845         dl_list_for_each(drv, &global->interfaces,
7846                          struct wpa_driver_nl80211_data, list) {
7847                 if (os_memcmp(addr, drv->first_bss.addr, ETH_ALEN) == 0)
7848                         return 1;
7849         }
7850         return 0;
7851 }
7852
7853
7854 static int nl80211_p2p_interface_addr(struct wpa_driver_nl80211_data *drv,
7855                                       u8 *new_addr)
7856 {
7857         unsigned int idx;
7858
7859         if (!drv->global)
7860                 return -1;
7861
7862         os_memcpy(new_addr, drv->first_bss.addr, ETH_ALEN);
7863         for (idx = 0; idx < 64; idx++) {
7864                 new_addr[0] = drv->first_bss.addr[0] | 0x02;
7865                 new_addr[0] ^= idx << 2;
7866                 if (!nl80211_addr_in_use(drv->global, new_addr))
7867                         break;
7868         }
7869         if (idx == 64)
7870                 return -1;
7871
7872         wpa_printf(MSG_DEBUG, "nl80211: Assigned new P2P Interface Address "
7873                    MACSTR, MAC2STR(new_addr));
7874
7875         return 0;
7876 }
7877
7878 #endif /* CONFIG_P2P */
7879
7880
7881 static int wpa_driver_nl80211_if_add(void *priv, enum wpa_driver_if_type type,
7882                                      const char *ifname, const u8 *addr,
7883                                      void *bss_ctx, void **drv_priv,
7884                                      char *force_ifname, u8 *if_addr,
7885                                      const char *bridge)
7886 {
7887         struct i802_bss *bss = priv;
7888         struct wpa_driver_nl80211_data *drv = bss->drv;
7889         int ifidx;
7890 #ifdef HOSTAPD
7891         struct i802_bss *new_bss = NULL;
7892
7893         if (type == WPA_IF_AP_BSS) {
7894                 new_bss = os_zalloc(sizeof(*new_bss));
7895                 if (new_bss == NULL)
7896                         return -1;
7897         }
7898 #endif /* HOSTAPD */
7899
7900         if (addr)
7901                 os_memcpy(if_addr, addr, ETH_ALEN);
7902         ifidx = nl80211_create_iface(drv, ifname,
7903                                      wpa_driver_nl80211_if_type(type), addr,
7904                                      0);
7905         if (ifidx < 0) {
7906 #ifdef HOSTAPD
7907                 os_free(new_bss);
7908 #endif /* HOSTAPD */
7909                 return -1;
7910         }
7911
7912         if (!addr &&
7913             linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7914                                if_addr) < 0) {
7915                 nl80211_remove_iface(drv, ifidx);
7916                 return -1;
7917         }
7918
7919 #ifdef CONFIG_P2P
7920         if (!addr &&
7921             (type == WPA_IF_P2P_CLIENT || type == WPA_IF_P2P_GROUP ||
7922              type == WPA_IF_P2P_GO)) {
7923                 /* Enforce unique P2P Interface Address */
7924                 u8 new_addr[ETH_ALEN], own_addr[ETH_ALEN];
7925
7926                 if (linux_get_ifhwaddr(drv->global->ioctl_sock, bss->ifname,
7927                                        own_addr) < 0 ||
7928                     linux_get_ifhwaddr(drv->global->ioctl_sock, ifname,
7929                                        new_addr) < 0) {
7930                         nl80211_remove_iface(drv, ifidx);
7931                         return -1;
7932                 }
7933                 if (os_memcmp(own_addr, new_addr, ETH_ALEN) == 0) {
7934                         wpa_printf(MSG_DEBUG, "nl80211: Allocate new address "
7935                                    "for P2P group interface");
7936                         if (nl80211_p2p_interface_addr(drv, new_addr) < 0) {
7937                                 nl80211_remove_iface(drv, ifidx);
7938                                 return -1;
7939                         }
7940                         if (linux_set_ifhwaddr(drv->global->ioctl_sock, ifname,
7941                                                new_addr) < 0) {
7942                                 nl80211_remove_iface(drv, ifidx);
7943                                 return -1;
7944                         }
7945                 }
7946                 os_memcpy(if_addr, new_addr, ETH_ALEN);
7947         }
7948 #endif /* CONFIG_P2P */
7949
7950 #ifdef HOSTAPD
7951         if (bridge &&
7952             i802_check_bridge(drv, new_bss, bridge, ifname) < 0) {
7953                 wpa_printf(MSG_ERROR, "nl80211: Failed to add the new "
7954                            "interface %s to a bridge %s", ifname, bridge);
7955                 nl80211_remove_iface(drv, ifidx);
7956                 os_free(new_bss);
7957                 return -1;
7958         }
7959
7960         if (type == WPA_IF_AP_BSS) {
7961                 if (linux_set_iface_flags(drv->global->ioctl_sock, ifname, 1))
7962                 {
7963                         nl80211_remove_iface(drv, ifidx);
7964                         os_free(new_bss);
7965                         return -1;
7966                 }
7967                 os_strlcpy(new_bss->ifname, ifname, IFNAMSIZ);
7968                 os_memcpy(new_bss->addr, if_addr, ETH_ALEN);
7969                 new_bss->ifindex = ifidx;
7970                 new_bss->drv = drv;
7971                 new_bss->next = drv->first_bss.next;
7972                 new_bss->freq = drv->first_bss.freq;
7973                 drv->first_bss.next = new_bss;
7974                 if (drv_priv)
7975                         *drv_priv = new_bss;
7976                 nl80211_init_bss(new_bss);
7977
7978                 /* Subscribe management frames for this WPA_IF_AP_BSS */
7979                 if (nl80211_setup_ap(new_bss))
7980                         return -1;
7981         }
7982 #endif /* HOSTAPD */
7983
7984         if (drv->global)
7985                 drv->global->if_add_ifindex = ifidx;
7986
7987         return 0;
7988 }
7989
7990
7991 static int wpa_driver_nl80211_if_remove(void *priv,
7992                                         enum wpa_driver_if_type type,
7993                                         const char *ifname)
7994 {
7995         struct i802_bss *bss = priv;
7996         struct wpa_driver_nl80211_data *drv = bss->drv;
7997         int ifindex = if_nametoindex(ifname);
7998
7999         wpa_printf(MSG_DEBUG, "nl80211: %s(type=%d ifname=%s) ifindex=%d",
8000                    __func__, type, ifname, ifindex);
8001         if (ifindex <= 0)
8002                 return -1;
8003
8004 #ifdef HOSTAPD
8005         if (bss->added_if_into_bridge) {
8006                 if (linux_br_del_if(drv->global->ioctl_sock, bss->brname,
8007                                     bss->ifname) < 0)
8008                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
8009                                    "interface %s from bridge %s: %s",
8010                                    bss->ifname, bss->brname, strerror(errno));
8011         }
8012         if (bss->added_bridge) {
8013                 if (linux_br_del(drv->global->ioctl_sock, bss->brname) < 0)
8014                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
8015                                    "bridge %s: %s",
8016                                    bss->brname, strerror(errno));
8017         }
8018 #endif /* HOSTAPD */
8019
8020         nl80211_remove_iface(drv, ifindex);
8021
8022 #ifdef HOSTAPD
8023         if (type != WPA_IF_AP_BSS)
8024                 return 0;
8025
8026         if (bss != &drv->first_bss) {
8027                 struct i802_bss *tbss;
8028
8029                 for (tbss = &drv->first_bss; tbss; tbss = tbss->next) {
8030                         if (tbss->next == bss) {
8031                                 tbss->next = bss->next;
8032                                 /* Unsubscribe management frames */
8033                                 nl80211_teardown_ap(bss);
8034                                 nl80211_destroy_bss(bss);
8035                                 os_free(bss);
8036                                 bss = NULL;
8037                                 break;
8038                         }
8039                 }
8040                 if (bss)
8041                         wpa_printf(MSG_INFO, "nl80211: %s - could not find "
8042                                    "BSS %p in the list", __func__, bss);
8043         }
8044 #endif /* HOSTAPD */
8045
8046         return 0;
8047 }
8048
8049
8050 static int cookie_handler(struct nl_msg *msg, void *arg)
8051 {
8052         struct nlattr *tb[NL80211_ATTR_MAX + 1];
8053         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
8054         u64 *cookie = arg;
8055         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
8056                   genlmsg_attrlen(gnlh, 0), NULL);
8057         if (tb[NL80211_ATTR_COOKIE])
8058                 *cookie = nla_get_u64(tb[NL80211_ATTR_COOKIE]);
8059         return NL_SKIP;
8060 }
8061
8062
8063 static int nl80211_send_frame_cmd(struct i802_bss *bss,
8064                                   unsigned int freq, unsigned int wait,
8065                                   const u8 *buf, size_t buf_len,
8066                                   u64 *cookie_out, int no_cck, int no_ack,
8067                                   int offchanok)
8068 {
8069         struct wpa_driver_nl80211_data *drv = bss->drv;
8070         struct nl_msg *msg;
8071         u64 cookie;
8072         int ret = -1;
8073
8074         msg = nlmsg_alloc();
8075         if (!msg)
8076                 return -1;
8077
8078         wpa_printf(MSG_DEBUG, "nl80211: CMD_FRAME freq=%u wait=%u no_cck=%d "
8079                    "no_ack=%d offchanok=%d",
8080                    freq, wait, no_cck, no_ack, offchanok);
8081         nl80211_cmd(drv, msg, 0, NL80211_CMD_FRAME);
8082
8083         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8084         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
8085         if (wait)
8086                 NLA_PUT_U32(msg, NL80211_ATTR_DURATION, wait);
8087         if (offchanok && (drv->capa.flags & WPA_DRIVER_FLAGS_OFFCHANNEL_TX))
8088                 NLA_PUT_FLAG(msg, NL80211_ATTR_OFFCHANNEL_TX_OK);
8089         if (no_cck)
8090                 NLA_PUT_FLAG(msg, NL80211_ATTR_TX_NO_CCK_RATE);
8091         if (no_ack)
8092                 NLA_PUT_FLAG(msg, NL80211_ATTR_DONT_WAIT_FOR_ACK);
8093
8094         NLA_PUT(msg, NL80211_ATTR_FRAME, buf_len, buf);
8095
8096         cookie = 0;
8097         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
8098         msg = NULL;
8099         if (ret) {
8100                 wpa_printf(MSG_DEBUG, "nl80211: Frame command failed: ret=%d "
8101                            "(%s) (freq=%u wait=%u)", ret, strerror(-ret),
8102                            freq, wait);
8103                 goto nla_put_failure;
8104         }
8105         wpa_printf(MSG_DEBUG, "nl80211: Frame TX command accepted%s; "
8106                    "cookie 0x%llx", no_ack ? " (no ACK)" : "",
8107                    (long long unsigned int) cookie);
8108
8109         if (cookie_out)
8110                 *cookie_out = no_ack ? (u64) -1 : cookie;
8111
8112 nla_put_failure:
8113         nlmsg_free(msg);
8114         return ret;
8115 }
8116
8117
8118 static int wpa_driver_nl80211_send_action(void *priv, unsigned int freq,
8119                                           unsigned int wait_time,
8120                                           const u8 *dst, const u8 *src,
8121                                           const u8 *bssid,
8122                                           const u8 *data, size_t data_len,
8123                                           int no_cck)
8124 {
8125         struct i802_bss *bss = priv;
8126         struct wpa_driver_nl80211_data *drv = bss->drv;
8127         int ret = -1;
8128         u8 *buf;
8129         struct ieee80211_hdr *hdr;
8130
8131         wpa_printf(MSG_DEBUG, "nl80211: Send Action frame (ifindex=%d, "
8132                    "freq=%u MHz wait=%d ms no_cck=%d)",
8133                    drv->ifindex, freq, wait_time, no_cck);
8134
8135         buf = os_zalloc(24 + data_len);
8136         if (buf == NULL)
8137                 return ret;
8138         os_memcpy(buf + 24, data, data_len);
8139         hdr = (struct ieee80211_hdr *) buf;
8140         hdr->frame_control =
8141                 IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_ACTION);
8142         os_memcpy(hdr->addr1, dst, ETH_ALEN);
8143         os_memcpy(hdr->addr2, src, ETH_ALEN);
8144         os_memcpy(hdr->addr3, bssid, ETH_ALEN);
8145
8146         if (is_ap_interface(drv->nlmode))
8147                 ret = wpa_driver_nl80211_send_mlme_freq(priv, buf,
8148                                                         24 + data_len,
8149                                                         0, freq, no_cck, 1,
8150                                                         wait_time);
8151         else
8152                 ret = nl80211_send_frame_cmd(bss, freq, wait_time, buf,
8153                                              24 + data_len,
8154                                              &drv->send_action_cookie,
8155                                              no_cck, 0, 1);
8156
8157         os_free(buf);
8158         return ret;
8159 }
8160
8161
8162 static void wpa_driver_nl80211_send_action_cancel_wait(void *priv)
8163 {
8164         struct i802_bss *bss = priv;
8165         struct wpa_driver_nl80211_data *drv = bss->drv;
8166         struct nl_msg *msg;
8167         int ret;
8168
8169         msg = nlmsg_alloc();
8170         if (!msg)
8171                 return;
8172
8173         nl80211_cmd(drv, msg, 0, NL80211_CMD_FRAME_WAIT_CANCEL);
8174
8175         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8176         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, drv->send_action_cookie);
8177
8178         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8179         msg = NULL;
8180         if (ret)
8181                 wpa_printf(MSG_DEBUG, "nl80211: wait cancel failed: ret=%d "
8182                            "(%s)", ret, strerror(-ret));
8183
8184  nla_put_failure:
8185         nlmsg_free(msg);
8186 }
8187
8188
8189 static int wpa_driver_nl80211_remain_on_channel(void *priv, unsigned int freq,
8190                                                 unsigned int duration)
8191 {
8192         struct i802_bss *bss = priv;
8193         struct wpa_driver_nl80211_data *drv = bss->drv;
8194         struct nl_msg *msg;
8195         int ret;
8196         u64 cookie;
8197
8198         msg = nlmsg_alloc();
8199         if (!msg)
8200                 return -1;
8201
8202         nl80211_cmd(drv, msg, 0, NL80211_CMD_REMAIN_ON_CHANNEL);
8203
8204         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8205         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
8206         NLA_PUT_U32(msg, NL80211_ATTR_DURATION, duration);
8207
8208         cookie = 0;
8209         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
8210         msg = NULL;
8211         if (ret == 0) {
8212                 wpa_printf(MSG_DEBUG, "nl80211: Remain-on-channel cookie "
8213                            "0x%llx for freq=%u MHz duration=%u",
8214                            (long long unsigned int) cookie, freq, duration);
8215                 drv->remain_on_chan_cookie = cookie;
8216                 drv->pending_remain_on_chan = 1;
8217                 return 0;
8218         }
8219         wpa_printf(MSG_DEBUG, "nl80211: Failed to request remain-on-channel "
8220                    "(freq=%d duration=%u): %d (%s)",
8221                    freq, duration, ret, strerror(-ret));
8222 nla_put_failure:
8223         nlmsg_free(msg);
8224         return -1;
8225 }
8226
8227
8228 static int wpa_driver_nl80211_cancel_remain_on_channel(void *priv)
8229 {
8230         struct i802_bss *bss = priv;
8231         struct wpa_driver_nl80211_data *drv = bss->drv;
8232         struct nl_msg *msg;
8233         int ret;
8234
8235         if (!drv->pending_remain_on_chan) {
8236                 wpa_printf(MSG_DEBUG, "nl80211: No pending remain-on-channel "
8237                            "to cancel");
8238                 return -1;
8239         }
8240
8241         wpa_printf(MSG_DEBUG, "nl80211: Cancel remain-on-channel with cookie "
8242                    "0x%llx",
8243                    (long long unsigned int) drv->remain_on_chan_cookie);
8244
8245         msg = nlmsg_alloc();
8246         if (!msg)
8247                 return -1;
8248
8249         nl80211_cmd(drv, msg, 0, NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL);
8250
8251         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8252         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, drv->remain_on_chan_cookie);
8253
8254         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8255         msg = NULL;
8256         if (ret == 0)
8257                 return 0;
8258         wpa_printf(MSG_DEBUG, "nl80211: Failed to cancel remain-on-channel: "
8259                    "%d (%s)", ret, strerror(-ret));
8260 nla_put_failure:
8261         nlmsg_free(msg);
8262         return -1;
8263 }
8264
8265
8266 static int wpa_driver_nl80211_probe_req_report(void *priv, int report)
8267 {
8268         struct i802_bss *bss = priv;
8269         struct wpa_driver_nl80211_data *drv = bss->drv;
8270
8271         if (!report) {
8272                 if (bss->nl_preq && drv->device_ap_sme &&
8273                     is_ap_interface(drv->nlmode)) {
8274                         /*
8275                          * Do not disable Probe Request reporting that was
8276                          * enabled in nl80211_setup_ap().
8277                          */
8278                         wpa_printf(MSG_DEBUG, "nl80211: Skip disabling of "
8279                                    "Probe Request reporting nl_preq=%p while "
8280                                    "in AP mode", bss->nl_preq);
8281                 } else if (bss->nl_preq) {
8282                         wpa_printf(MSG_DEBUG, "nl80211: Disable Probe Request "
8283                                    "reporting nl_preq=%p", bss->nl_preq);
8284                         eloop_unregister_read_sock(
8285                                 nl_socket_get_fd(bss->nl_preq));
8286                         nl_destroy_handles(&bss->nl_preq);
8287                 }
8288                 return 0;
8289         }
8290
8291         if (bss->nl_preq) {
8292                 wpa_printf(MSG_DEBUG, "nl80211: Probe Request reporting "
8293                            "already on! nl_preq=%p", bss->nl_preq);
8294                 return 0;
8295         }
8296
8297         bss->nl_preq = nl_create_handle(drv->global->nl_cb, "preq");
8298         if (bss->nl_preq == NULL)
8299                 return -1;
8300         wpa_printf(MSG_DEBUG, "nl80211: Enable Probe Request "
8301                    "reporting nl_preq=%p", bss->nl_preq);
8302
8303         if (nl80211_register_frame(bss, bss->nl_preq,
8304                                    (WLAN_FC_TYPE_MGMT << 2) |
8305                                    (WLAN_FC_STYPE_PROBE_REQ << 4),
8306                                    NULL, 0) < 0)
8307                 goto out_err;
8308
8309         eloop_register_read_sock(nl_socket_get_fd(bss->nl_preq),
8310                                  wpa_driver_nl80211_event_receive, bss->nl_cb,
8311                                  bss->nl_preq);
8312
8313         return 0;
8314
8315  out_err:
8316         nl_destroy_handles(&bss->nl_preq);
8317         return -1;
8318 }
8319
8320
8321 static int nl80211_disable_11b_rates(struct wpa_driver_nl80211_data *drv,
8322                                      int ifindex, int disabled)
8323 {
8324         struct nl_msg *msg;
8325         struct nlattr *bands, *band;
8326         int ret;
8327
8328         msg = nlmsg_alloc();
8329         if (!msg)
8330                 return -1;
8331
8332         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_TX_BITRATE_MASK);
8333         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
8334
8335         bands = nla_nest_start(msg, NL80211_ATTR_TX_RATES);
8336         if (!bands)
8337                 goto nla_put_failure;
8338
8339         /*
8340          * Disable 2 GHz rates 1, 2, 5.5, 11 Mbps by masking out everything
8341          * else apart from 6, 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS
8342          * rates. All 5 GHz rates are left enabled.
8343          */
8344         band = nla_nest_start(msg, NL80211_BAND_2GHZ);
8345         if (!band)
8346                 goto nla_put_failure;
8347         if (disabled) {
8348                 NLA_PUT(msg, NL80211_TXRATE_LEGACY, 8,
8349                         "\x0c\x12\x18\x24\x30\x48\x60\x6c");
8350         }
8351         nla_nest_end(msg, band);
8352
8353         nla_nest_end(msg, bands);
8354
8355         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
8356         msg = NULL;
8357         if (ret) {
8358                 wpa_printf(MSG_DEBUG, "nl80211: Set TX rates failed: ret=%d "
8359                            "(%s)", ret, strerror(-ret));
8360         }
8361
8362         return ret;
8363
8364 nla_put_failure:
8365         nlmsg_free(msg);
8366         return -1;
8367 }
8368
8369
8370 static int wpa_driver_nl80211_deinit_ap(void *priv)
8371 {
8372         struct i802_bss *bss = priv;
8373         struct wpa_driver_nl80211_data *drv = bss->drv;
8374         if (!is_ap_interface(drv->nlmode))
8375                 return -1;
8376         wpa_driver_nl80211_del_beacon(drv);
8377         return wpa_driver_nl80211_set_mode(priv, NL80211_IFTYPE_STATION);
8378 }
8379
8380
8381 static int wpa_driver_nl80211_deinit_p2p_cli(void *priv)
8382 {
8383         struct i802_bss *bss = priv;
8384         struct wpa_driver_nl80211_data *drv = bss->drv;
8385         if (drv->nlmode != NL80211_IFTYPE_P2P_CLIENT)
8386                 return -1;
8387         return wpa_driver_nl80211_set_mode(priv, NL80211_IFTYPE_STATION);
8388 }
8389
8390
8391 static void wpa_driver_nl80211_resume(void *priv)
8392 {
8393         struct i802_bss *bss = priv;
8394         struct wpa_driver_nl80211_data *drv = bss->drv;
8395         if (linux_set_iface_flags(drv->global->ioctl_sock, bss->ifname, 1)) {
8396                 wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface up on "
8397                            "resume event");
8398         }
8399 }
8400
8401
8402 static int nl80211_send_ft_action(void *priv, u8 action, const u8 *target_ap,
8403                                   const u8 *ies, size_t ies_len)
8404 {
8405         struct i802_bss *bss = priv;
8406         struct wpa_driver_nl80211_data *drv = bss->drv;
8407         int ret;
8408         u8 *data, *pos;
8409         size_t data_len;
8410         const u8 *own_addr = bss->addr;
8411
8412         if (action != 1) {
8413                 wpa_printf(MSG_ERROR, "nl80211: Unsupported send_ft_action "
8414                            "action %d", action);
8415                 return -1;
8416         }
8417
8418         /*
8419          * Action frame payload:
8420          * Category[1] = 6 (Fast BSS Transition)
8421          * Action[1] = 1 (Fast BSS Transition Request)
8422          * STA Address
8423          * Target AP Address
8424          * FT IEs
8425          */
8426
8427         data_len = 2 + 2 * ETH_ALEN + ies_len;
8428         data = os_malloc(data_len);
8429         if (data == NULL)
8430                 return -1;
8431         pos = data;
8432         *pos++ = 0x06; /* FT Action category */
8433         *pos++ = action;
8434         os_memcpy(pos, own_addr, ETH_ALEN);
8435         pos += ETH_ALEN;
8436         os_memcpy(pos, target_ap, ETH_ALEN);
8437         pos += ETH_ALEN;
8438         os_memcpy(pos, ies, ies_len);
8439
8440         ret = wpa_driver_nl80211_send_action(bss, drv->assoc_freq, 0,
8441                                              drv->bssid, own_addr, drv->bssid,
8442                                              data, data_len, 0);
8443         os_free(data);
8444
8445         return ret;
8446 }
8447
8448
8449 static int nl80211_signal_monitor(void *priv, int threshold, int hysteresis)
8450 {
8451         struct i802_bss *bss = priv;
8452         struct wpa_driver_nl80211_data *drv = bss->drv;
8453         struct nl_msg *msg, *cqm = NULL;
8454
8455         wpa_printf(MSG_DEBUG, "nl80211: Signal monitor threshold=%d "
8456                    "hysteresis=%d", threshold, hysteresis);
8457
8458         msg = nlmsg_alloc();
8459         if (!msg)
8460                 return -1;
8461
8462         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_CQM);
8463
8464         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8465
8466         cqm = nlmsg_alloc();
8467         if (cqm == NULL)
8468                 goto nla_put_failure;
8469
8470         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_THOLD, threshold);
8471         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_HYST, hysteresis);
8472         nla_put_nested(msg, NL80211_ATTR_CQM, cqm);
8473
8474         nlmsg_free(cqm);
8475         cqm = NULL;
8476
8477         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
8478                 return 0;
8479         msg = NULL;
8480
8481 nla_put_failure:
8482         nlmsg_free(cqm);
8483         nlmsg_free(msg);
8484         return -1;
8485 }
8486
8487
8488 static int nl80211_signal_poll(void *priv, struct wpa_signal_info *si)
8489 {
8490         struct i802_bss *bss = priv;
8491         struct wpa_driver_nl80211_data *drv = bss->drv;
8492         int res;
8493
8494         os_memset(si, 0, sizeof(*si));
8495         res = nl80211_get_link_signal(drv, si);
8496         if (res != 0)
8497                 return res;
8498
8499         return nl80211_get_link_noise(drv, si);
8500 }
8501
8502
8503 static int wpa_driver_nl80211_shared_freq(void *priv)
8504 {
8505         struct i802_bss *bss = priv;
8506         struct wpa_driver_nl80211_data *drv = bss->drv;
8507         struct wpa_driver_nl80211_data *driver;
8508         int freq = 0;
8509
8510         /*
8511          * If the same PHY is in connected state with some other interface,
8512          * then retrieve the assoc freq.
8513          */
8514         wpa_printf(MSG_DEBUG, "nl80211: Get shared freq for PHY %s",
8515                    drv->phyname);
8516
8517         dl_list_for_each(driver, &drv->global->interfaces,
8518                          struct wpa_driver_nl80211_data, list) {
8519                 if (drv == driver ||
8520                     os_strcmp(drv->phyname, driver->phyname) != 0 ||
8521                     !driver->associated)
8522                         continue;
8523
8524                 wpa_printf(MSG_DEBUG, "nl80211: Found a match for PHY %s - %s "
8525                            MACSTR,
8526                            driver->phyname, driver->first_bss.ifname,
8527                            MAC2STR(driver->first_bss.addr));
8528                 if (is_ap_interface(driver->nlmode))
8529                         freq = driver->first_bss.freq;
8530                 else
8531                         freq = nl80211_get_assoc_freq(driver);
8532                 wpa_printf(MSG_DEBUG, "nl80211: Shared freq for PHY %s: %d",
8533                            drv->phyname, freq);
8534         }
8535
8536         if (!freq)
8537                 wpa_printf(MSG_DEBUG, "nl80211: No shared interface for "
8538                            "PHY (%s) in associated state", drv->phyname);
8539
8540         return freq;
8541 }
8542
8543
8544 static int nl80211_send_frame(void *priv, const u8 *data, size_t data_len,
8545                               int encrypt)
8546 {
8547         struct i802_bss *bss = priv;
8548         return wpa_driver_nl80211_send_frame(bss, data, data_len, encrypt, 0,
8549                                              0, 0, 0, 0);
8550 }
8551
8552
8553 static int nl80211_set_param(void *priv, const char *param)
8554 {
8555         wpa_printf(MSG_DEBUG, "nl80211: driver param='%s'", param);
8556         if (param == NULL)
8557                 return 0;
8558
8559 #ifdef CONFIG_P2P
8560         if (os_strstr(param, "use_p2p_group_interface=1")) {
8561                 struct i802_bss *bss = priv;
8562                 struct wpa_driver_nl80211_data *drv = bss->drv;
8563
8564                 wpa_printf(MSG_DEBUG, "nl80211: Use separate P2P group "
8565                            "interface");
8566                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CONCURRENT;
8567                 drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_MGMT_AND_NON_P2P;
8568         }
8569 #endif /* CONFIG_P2P */
8570
8571         return 0;
8572 }
8573
8574
8575 static void * nl80211_global_init(void)
8576 {
8577         struct nl80211_global *global;
8578         struct netlink_config *cfg;
8579
8580         global = os_zalloc(sizeof(*global));
8581         if (global == NULL)
8582                 return NULL;
8583         global->ioctl_sock = -1;
8584         dl_list_init(&global->interfaces);
8585         global->if_add_ifindex = -1;
8586
8587         cfg = os_zalloc(sizeof(*cfg));
8588         if (cfg == NULL)
8589                 goto err;
8590
8591         cfg->ctx = global;
8592         cfg->newlink_cb = wpa_driver_nl80211_event_rtm_newlink;
8593         cfg->dellink_cb = wpa_driver_nl80211_event_rtm_dellink;
8594         global->netlink = netlink_init(cfg);
8595         if (global->netlink == NULL) {
8596                 os_free(cfg);
8597                 goto err;
8598         }
8599
8600         if (wpa_driver_nl80211_init_nl_global(global) < 0)
8601                 goto err;
8602
8603         global->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
8604         if (global->ioctl_sock < 0) {
8605                 perror("socket(PF_INET,SOCK_DGRAM)");
8606                 goto err;
8607         }
8608
8609         return global;
8610
8611 err:
8612         nl80211_global_deinit(global);
8613         return NULL;
8614 }
8615
8616
8617 static void nl80211_global_deinit(void *priv)
8618 {
8619         struct nl80211_global *global = priv;
8620         if (global == NULL)
8621                 return;
8622         if (!dl_list_empty(&global->interfaces)) {
8623                 wpa_printf(MSG_ERROR, "nl80211: %u interface(s) remain at "
8624                            "nl80211_global_deinit",
8625                            dl_list_len(&global->interfaces));
8626         }
8627
8628         if (global->netlink)
8629                 netlink_deinit(global->netlink);
8630
8631         nl_destroy_handles(&global->nl);
8632
8633         if (global->nl_event) {
8634                 eloop_unregister_read_sock(
8635                         nl_socket_get_fd(global->nl_event));
8636                 nl_destroy_handles(&global->nl_event);
8637         }
8638
8639         nl_cb_put(global->nl_cb);
8640
8641         if (global->ioctl_sock >= 0)
8642                 close(global->ioctl_sock);
8643
8644         os_free(global);
8645 }
8646
8647
8648 static const char * nl80211_get_radio_name(void *priv)
8649 {
8650         struct i802_bss *bss = priv;
8651         struct wpa_driver_nl80211_data *drv = bss->drv;
8652         return drv->phyname;
8653 }
8654
8655
8656 static int nl80211_pmkid(struct i802_bss *bss, int cmd, const u8 *bssid,
8657                          const u8 *pmkid)
8658 {
8659         struct nl_msg *msg;
8660
8661         msg = nlmsg_alloc();
8662         if (!msg)
8663                 return -ENOMEM;
8664
8665         nl80211_cmd(bss->drv, msg, 0, cmd);
8666
8667         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
8668         if (pmkid)
8669                 NLA_PUT(msg, NL80211_ATTR_PMKID, 16, pmkid);
8670         if (bssid)
8671                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, bssid);
8672
8673         return send_and_recv_msgs(bss->drv, msg, NULL, NULL);
8674  nla_put_failure:
8675         nlmsg_free(msg);
8676         return -ENOBUFS;
8677 }
8678
8679
8680 static int nl80211_add_pmkid(void *priv, const u8 *bssid, const u8 *pmkid)
8681 {
8682         struct i802_bss *bss = priv;
8683         wpa_printf(MSG_DEBUG, "nl80211: Add PMKID for " MACSTR, MAC2STR(bssid));
8684         return nl80211_pmkid(bss, NL80211_CMD_SET_PMKSA, bssid, pmkid);
8685 }
8686
8687
8688 static int nl80211_remove_pmkid(void *priv, const u8 *bssid, const u8 *pmkid)
8689 {
8690         struct i802_bss *bss = priv;
8691         wpa_printf(MSG_DEBUG, "nl80211: Delete PMKID for " MACSTR,
8692                    MAC2STR(bssid));
8693         return nl80211_pmkid(bss, NL80211_CMD_DEL_PMKSA, bssid, pmkid);
8694 }
8695
8696
8697 static int nl80211_flush_pmkid(void *priv)
8698 {
8699         struct i802_bss *bss = priv;
8700         wpa_printf(MSG_DEBUG, "nl80211: Flush PMKIDs");
8701         return nl80211_pmkid(bss, NL80211_CMD_FLUSH_PMKSA, NULL, NULL);
8702 }
8703
8704
8705 static void nl80211_set_rekey_info(void *priv, const u8 *kek, const u8 *kck,
8706                                    const u8 *replay_ctr)
8707 {
8708         struct i802_bss *bss = priv;
8709         struct wpa_driver_nl80211_data *drv = bss->drv;
8710         struct nlattr *replay_nested;
8711         struct nl_msg *msg;
8712
8713         msg = nlmsg_alloc();
8714         if (!msg)
8715                 return;
8716
8717         nl80211_cmd(drv, msg, 0, NL80211_CMD_SET_REKEY_OFFLOAD);
8718
8719         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8720
8721         replay_nested = nla_nest_start(msg, NL80211_ATTR_REKEY_DATA);
8722         if (!replay_nested)
8723                 goto nla_put_failure;
8724
8725         NLA_PUT(msg, NL80211_REKEY_DATA_KEK, NL80211_KEK_LEN, kek);
8726         NLA_PUT(msg, NL80211_REKEY_DATA_KCK, NL80211_KCK_LEN, kck);
8727         NLA_PUT(msg, NL80211_REKEY_DATA_REPLAY_CTR, NL80211_REPLAY_CTR_LEN,
8728                 replay_ctr);
8729
8730         nla_nest_end(msg, replay_nested);
8731
8732         send_and_recv_msgs(drv, msg, NULL, NULL);
8733         return;
8734  nla_put_failure:
8735         nlmsg_free(msg);
8736 }
8737
8738
8739 static void nl80211_send_null_frame(struct i802_bss *bss, const u8 *own_addr,
8740                                     const u8 *addr, int qos)
8741 {
8742         /* send data frame to poll STA and check whether
8743          * this frame is ACKed */
8744         struct {
8745                 struct ieee80211_hdr hdr;
8746                 u16 qos_ctl;
8747         } STRUCT_PACKED nulldata;
8748         size_t size;
8749
8750         /* Send data frame to poll STA and check whether this frame is ACKed */
8751
8752         os_memset(&nulldata, 0, sizeof(nulldata));
8753
8754         if (qos) {
8755                 nulldata.hdr.frame_control =
8756                         IEEE80211_FC(WLAN_FC_TYPE_DATA,
8757                                      WLAN_FC_STYPE_QOS_NULL);
8758                 size = sizeof(nulldata);
8759         } else {
8760                 nulldata.hdr.frame_control =
8761                         IEEE80211_FC(WLAN_FC_TYPE_DATA,
8762                                      WLAN_FC_STYPE_NULLFUNC);
8763                 size = sizeof(struct ieee80211_hdr);
8764         }
8765
8766         nulldata.hdr.frame_control |= host_to_le16(WLAN_FC_FROMDS);
8767         os_memcpy(nulldata.hdr.IEEE80211_DA_FROMDS, addr, ETH_ALEN);
8768         os_memcpy(nulldata.hdr.IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
8769         os_memcpy(nulldata.hdr.IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
8770
8771         if (wpa_driver_nl80211_send_mlme(bss, (u8 *) &nulldata, size, 0) < 0)
8772                 wpa_printf(MSG_DEBUG, "nl80211_send_null_frame: Failed to "
8773                            "send poll frame");
8774 }
8775
8776 static void nl80211_poll_client(void *priv, const u8 *own_addr, const u8 *addr,
8777                                 int qos)
8778 {
8779         struct i802_bss *bss = priv;
8780         struct wpa_driver_nl80211_data *drv = bss->drv;
8781         struct nl_msg *msg;
8782
8783         if (!drv->poll_command_supported) {
8784                 nl80211_send_null_frame(bss, own_addr, addr, qos);
8785                 return;
8786         }
8787
8788         msg = nlmsg_alloc();
8789         if (!msg)
8790                 return;
8791
8792         nl80211_cmd(drv, msg, 0, NL80211_CMD_PROBE_CLIENT);
8793
8794         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8795         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
8796
8797         send_and_recv_msgs(drv, msg, NULL, NULL);
8798         return;
8799  nla_put_failure:
8800         nlmsg_free(msg);
8801 }
8802
8803
8804 static int nl80211_set_power_save(struct i802_bss *bss, int enabled)
8805 {
8806         struct nl_msg *msg;
8807
8808         msg = nlmsg_alloc();
8809         if (!msg)
8810                 return -ENOMEM;
8811
8812         nl80211_cmd(bss->drv, msg, 0, NL80211_CMD_SET_POWER_SAVE);
8813         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
8814         NLA_PUT_U32(msg, NL80211_ATTR_PS_STATE,
8815                     enabled ? NL80211_PS_ENABLED : NL80211_PS_DISABLED);
8816         return send_and_recv_msgs(bss->drv, msg, NULL, NULL);
8817 nla_put_failure:
8818         nlmsg_free(msg);
8819         return -ENOBUFS;
8820 }
8821
8822
8823 static int nl80211_set_p2p_powersave(void *priv, int legacy_ps, int opp_ps,
8824                                      int ctwindow)
8825 {
8826         struct i802_bss *bss = priv;
8827
8828         wpa_printf(MSG_DEBUG, "nl80211: set_p2p_powersave (legacy_ps=%d "
8829                    "opp_ps=%d ctwindow=%d)", legacy_ps, opp_ps, ctwindow);
8830
8831         if (opp_ps != -1 || ctwindow != -1)
8832                 return -1; /* Not yet supported */
8833
8834         if (legacy_ps == -1)
8835                 return 0;
8836         if (legacy_ps != 0 && legacy_ps != 1)
8837                 return -1; /* Not yet supported */
8838
8839         return nl80211_set_power_save(bss, legacy_ps);
8840 }
8841
8842
8843 #ifdef CONFIG_TDLS
8844
8845 static int nl80211_send_tdls_mgmt(void *priv, const u8 *dst, u8 action_code,
8846                                   u8 dialog_token, u16 status_code,
8847                                   const u8 *buf, size_t len)
8848 {
8849         struct i802_bss *bss = priv;
8850         struct wpa_driver_nl80211_data *drv = bss->drv;
8851         struct nl_msg *msg;
8852
8853         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
8854                 return -EOPNOTSUPP;
8855
8856         if (!dst)
8857                 return -EINVAL;
8858
8859         msg = nlmsg_alloc();
8860         if (!msg)
8861                 return -ENOMEM;
8862
8863         nl80211_cmd(drv, msg, 0, NL80211_CMD_TDLS_MGMT);
8864         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8865         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
8866         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_ACTION, action_code);
8867         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_DIALOG_TOKEN, dialog_token);
8868         NLA_PUT_U16(msg, NL80211_ATTR_STATUS_CODE, status_code);
8869         NLA_PUT(msg, NL80211_ATTR_IE, len, buf);
8870
8871         return send_and_recv_msgs(drv, msg, NULL, NULL);
8872
8873 nla_put_failure:
8874         nlmsg_free(msg);
8875         return -ENOBUFS;
8876 }
8877
8878
8879 static int nl80211_tdls_oper(void *priv, enum tdls_oper oper, const u8 *peer)
8880 {
8881         struct i802_bss *bss = priv;
8882         struct wpa_driver_nl80211_data *drv = bss->drv;
8883         struct nl_msg *msg;
8884         enum nl80211_tdls_operation nl80211_oper;
8885
8886         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_TDLS_SUPPORT))
8887                 return -EOPNOTSUPP;
8888
8889         switch (oper) {
8890         case TDLS_DISCOVERY_REQ:
8891                 nl80211_oper = NL80211_TDLS_DISCOVERY_REQ;
8892                 break;
8893         case TDLS_SETUP:
8894                 nl80211_oper = NL80211_TDLS_SETUP;
8895                 break;
8896         case TDLS_TEARDOWN:
8897                 nl80211_oper = NL80211_TDLS_TEARDOWN;
8898                 break;
8899         case TDLS_ENABLE_LINK:
8900                 nl80211_oper = NL80211_TDLS_ENABLE_LINK;
8901                 break;
8902         case TDLS_DISABLE_LINK:
8903                 nl80211_oper = NL80211_TDLS_DISABLE_LINK;
8904                 break;
8905         case TDLS_ENABLE:
8906                 return 0;
8907         case TDLS_DISABLE:
8908                 return 0;
8909         default:
8910                 return -EINVAL;
8911         }
8912
8913         msg = nlmsg_alloc();
8914         if (!msg)
8915                 return -ENOMEM;
8916
8917         nl80211_cmd(drv, msg, 0, NL80211_CMD_TDLS_OPER);
8918         NLA_PUT_U8(msg, NL80211_ATTR_TDLS_OPERATION, nl80211_oper);
8919         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
8920         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, peer);
8921
8922         return send_and_recv_msgs(drv, msg, NULL, NULL);
8923
8924 nla_put_failure:
8925         nlmsg_free(msg);
8926         return -ENOBUFS;
8927 }
8928
8929 #endif /* CONFIG TDLS */
8930
8931
8932 #ifdef ANDROID
8933
8934 typedef struct android_wifi_priv_cmd {
8935         char *buf;
8936         int used_len;
8937         int total_len;
8938 } android_wifi_priv_cmd;
8939
8940 static int drv_errors = 0;
8941
8942 static void wpa_driver_send_hang_msg(struct wpa_driver_nl80211_data *drv)
8943 {
8944         drv_errors++;
8945         if (drv_errors > DRV_NUMBER_SEQUENTIAL_ERRORS) {
8946                 drv_errors = 0;
8947                 wpa_msg(drv->ctx, MSG_INFO, WPA_EVENT_DRIVER_STATE "HANGED");
8948         }
8949 }
8950
8951
8952 static int android_priv_cmd(struct i802_bss *bss, const char *cmd)
8953 {
8954         struct wpa_driver_nl80211_data *drv = bss->drv;
8955         struct ifreq ifr;
8956         android_wifi_priv_cmd priv_cmd;
8957         char buf[MAX_DRV_CMD_SIZE];
8958         int ret;
8959
8960         os_memset(&ifr, 0, sizeof(ifr));
8961         os_memset(&priv_cmd, 0, sizeof(priv_cmd));
8962         os_strlcpy(ifr.ifr_name, bss->ifname, IFNAMSIZ);
8963
8964         os_memset(buf, 0, sizeof(buf));
8965         os_strlcpy(buf, cmd, sizeof(buf));
8966
8967         priv_cmd.buf = buf;
8968         priv_cmd.used_len = sizeof(buf);
8969         priv_cmd.total_len = sizeof(buf);
8970         ifr.ifr_data = &priv_cmd;
8971
8972         ret = ioctl(drv->global->ioctl_sock, SIOCDEVPRIVATE + 1, &ifr);
8973         if (ret < 0) {
8974                 wpa_printf(MSG_ERROR, "%s: failed to issue private commands",
8975                            __func__);
8976                 wpa_driver_send_hang_msg(drv);
8977                 return ret;
8978         }
8979
8980         drv_errors = 0;
8981         return 0;
8982 }
8983
8984
8985 static int android_pno_start(struct i802_bss *bss,
8986                              struct wpa_driver_scan_params *params)
8987 {
8988         struct wpa_driver_nl80211_data *drv = bss->drv;
8989         struct ifreq ifr;
8990         android_wifi_priv_cmd priv_cmd;
8991         int ret = 0, i = 0, bp;
8992         char buf[WEXT_PNO_MAX_COMMAND_SIZE];
8993
8994         bp = WEXT_PNOSETUP_HEADER_SIZE;
8995         os_memcpy(buf, WEXT_PNOSETUP_HEADER, bp);
8996         buf[bp++] = WEXT_PNO_TLV_PREFIX;
8997         buf[bp++] = WEXT_PNO_TLV_VERSION;
8998         buf[bp++] = WEXT_PNO_TLV_SUBVERSION;
8999         buf[bp++] = WEXT_PNO_TLV_RESERVED;
9000
9001         while (i < WEXT_PNO_AMOUNT && (size_t) i < params->num_ssids) {
9002                 /* Check that there is enough space needed for 1 more SSID, the
9003                  * other sections and null termination */
9004                 if ((bp + WEXT_PNO_SSID_HEADER_SIZE + MAX_SSID_LEN +
9005                      WEXT_PNO_NONSSID_SECTIONS_SIZE + 1) >= (int) sizeof(buf))
9006                         break;
9007                 wpa_hexdump_ascii(MSG_DEBUG, "For PNO Scan",
9008                                   params->ssids[i].ssid,
9009                                   params->ssids[i].ssid_len);
9010                 buf[bp++] = WEXT_PNO_SSID_SECTION;
9011                 buf[bp++] = params->ssids[i].ssid_len;
9012                 os_memcpy(&buf[bp], params->ssids[i].ssid,
9013                           params->ssids[i].ssid_len);
9014                 bp += params->ssids[i].ssid_len;
9015                 i++;
9016         }
9017
9018         buf[bp++] = WEXT_PNO_SCAN_INTERVAL_SECTION;
9019         os_snprintf(&buf[bp], WEXT_PNO_SCAN_INTERVAL_LENGTH + 1, "%x",
9020                     WEXT_PNO_SCAN_INTERVAL);
9021         bp += WEXT_PNO_SCAN_INTERVAL_LENGTH;
9022
9023         buf[bp++] = WEXT_PNO_REPEAT_SECTION;
9024         os_snprintf(&buf[bp], WEXT_PNO_REPEAT_LENGTH + 1, "%x",
9025                     WEXT_PNO_REPEAT);
9026         bp += WEXT_PNO_REPEAT_LENGTH;
9027
9028         buf[bp++] = WEXT_PNO_MAX_REPEAT_SECTION;
9029         os_snprintf(&buf[bp], WEXT_PNO_MAX_REPEAT_LENGTH + 1, "%x",
9030                     WEXT_PNO_MAX_REPEAT);
9031         bp += WEXT_PNO_MAX_REPEAT_LENGTH + 1;
9032
9033         memset(&ifr, 0, sizeof(ifr));
9034         memset(&priv_cmd, 0, sizeof(priv_cmd));
9035         os_strncpy(ifr.ifr_name, bss->ifname, IFNAMSIZ);
9036
9037         priv_cmd.buf = buf;
9038         priv_cmd.used_len = bp;
9039         priv_cmd.total_len = bp;
9040         ifr.ifr_data = &priv_cmd;
9041
9042         ret = ioctl(drv->global->ioctl_sock, SIOCDEVPRIVATE + 1, &ifr);
9043
9044         if (ret < 0) {
9045                 wpa_printf(MSG_ERROR, "ioctl[SIOCSIWPRIV] (pnosetup): %d",
9046                            ret);
9047                 wpa_driver_send_hang_msg(drv);
9048                 return ret;
9049         }
9050
9051         drv_errors = 0;
9052
9053         return android_priv_cmd(bss, "PNOFORCE 1");
9054 }
9055
9056
9057 static int android_pno_stop(struct i802_bss *bss)
9058 {
9059         return android_priv_cmd(bss, "PNOFORCE 0");
9060 }
9061
9062 #endif /* ANDROID */
9063
9064
9065 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
9066         .name = "nl80211",
9067         .desc = "Linux nl80211/cfg80211",
9068         .get_bssid = wpa_driver_nl80211_get_bssid,
9069         .get_ssid = wpa_driver_nl80211_get_ssid,
9070         .set_key = wpa_driver_nl80211_set_key,
9071         .scan2 = wpa_driver_nl80211_scan,
9072         .sched_scan = wpa_driver_nl80211_sched_scan,
9073         .stop_sched_scan = wpa_driver_nl80211_stop_sched_scan,
9074         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
9075         .deauthenticate = wpa_driver_nl80211_deauthenticate,
9076         .disassociate = wpa_driver_nl80211_disassociate,
9077         .authenticate = wpa_driver_nl80211_authenticate,
9078         .associate = wpa_driver_nl80211_associate,
9079         .global_init = nl80211_global_init,
9080         .global_deinit = nl80211_global_deinit,
9081         .init2 = wpa_driver_nl80211_init,
9082         .deinit = wpa_driver_nl80211_deinit,
9083         .get_capa = wpa_driver_nl80211_get_capa,
9084         .set_operstate = wpa_driver_nl80211_set_operstate,
9085         .set_supp_port = wpa_driver_nl80211_set_supp_port,
9086         .set_country = wpa_driver_nl80211_set_country,
9087         .set_ap = wpa_driver_nl80211_set_ap,
9088         .if_add = wpa_driver_nl80211_if_add,
9089         .if_remove = wpa_driver_nl80211_if_remove,
9090         .send_mlme = wpa_driver_nl80211_send_mlme,
9091         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
9092         .sta_add = wpa_driver_nl80211_sta_add,
9093         .sta_remove = wpa_driver_nl80211_sta_remove,
9094         .hapd_send_eapol = wpa_driver_nl80211_hapd_send_eapol,
9095         .sta_set_flags = wpa_driver_nl80211_sta_set_flags,
9096 #ifdef HOSTAPD
9097         .hapd_init = i802_init,
9098         .hapd_deinit = i802_deinit,
9099         .set_wds_sta = i802_set_wds_sta,
9100 #endif /* HOSTAPD */
9101 #if defined(HOSTAPD) || defined(CONFIG_AP)
9102         .get_seqnum = i802_get_seqnum,
9103         .flush = i802_flush,
9104         .read_sta_data = i802_read_sta_data,
9105         .get_inact_sec = i802_get_inact_sec,
9106         .sta_clear_stats = i802_sta_clear_stats,
9107         .set_rts = i802_set_rts,
9108         .set_frag = i802_set_frag,
9109         .set_tx_queue_params = i802_set_tx_queue_params,
9110         .set_sta_vlan = i802_set_sta_vlan,
9111         .sta_deauth = i802_sta_deauth,
9112         .sta_disassoc = i802_sta_disassoc,
9113 #endif /* HOSTAPD || CONFIG_AP */
9114         .set_freq = i802_set_freq,
9115         .send_action = wpa_driver_nl80211_send_action,
9116         .send_action_cancel_wait = wpa_driver_nl80211_send_action_cancel_wait,
9117         .remain_on_channel = wpa_driver_nl80211_remain_on_channel,
9118         .cancel_remain_on_channel =
9119         wpa_driver_nl80211_cancel_remain_on_channel,
9120         .probe_req_report = wpa_driver_nl80211_probe_req_report,
9121         .deinit_ap = wpa_driver_nl80211_deinit_ap,
9122         .deinit_p2p_cli = wpa_driver_nl80211_deinit_p2p_cli,
9123         .resume = wpa_driver_nl80211_resume,
9124         .send_ft_action = nl80211_send_ft_action,
9125         .signal_monitor = nl80211_signal_monitor,
9126         .signal_poll = nl80211_signal_poll,
9127         .send_frame = nl80211_send_frame,
9128         .shared_freq = wpa_driver_nl80211_shared_freq,
9129         .set_param = nl80211_set_param,
9130         .get_radio_name = nl80211_get_radio_name,
9131         .add_pmkid = nl80211_add_pmkid,
9132         .remove_pmkid = nl80211_remove_pmkid,
9133         .flush_pmkid = nl80211_flush_pmkid,
9134         .set_rekey_info = nl80211_set_rekey_info,
9135         .poll_client = nl80211_poll_client,
9136         .set_p2p_powersave = nl80211_set_p2p_powersave,
9137 #ifdef CONFIG_TDLS
9138         .send_tdls_mgmt = nl80211_send_tdls_mgmt,
9139         .tdls_oper = nl80211_tdls_oper,
9140 #endif /* CONFIG_TDLS */
9141 };