src/drivers/driver_nl80211.c

   1 /*
   2  * Driver interaction with Linux nl80211/cfg80211
   3  * Copyright (c) 2002-2010, 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 program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License version 2 as
  11  * published by the Free Software Foundation.
  12  *
  13  * Alternatively, this software may be distributed under the terms of BSD
  14  * license.
  15  *
  16  * See README and COPYING for more details.
  17  */
  18
  19 #include "includes.h"
  20 #include <sys/ioctl.h>
  21 #include <net/if.h>
  22 #include <netlink/genl/genl.h>
  23 #include <netlink/genl/family.h>
  24 #include <netlink/genl/ctrl.h>
  25 #include <linux/rtnetlink.h>
  26 #include <netpacket/packet.h>
  27 #include <linux/filter.h>
  28 #include "nl80211_copy.h"
  29
  30 #include "common.h"
  31 #include "eloop.h"
  32 #include "common/ieee802_11_defs.h"
  33 #include "netlink.h"
  34 #include "linux_ioctl.h"
  35 #include "radiotap.h"
  36 #include "radiotap_iter.h"
  37 #include "rfkill.h"
  38 #include "driver.h"
  39
  40 #ifdef CONFIG_LIBNL20
  41 /* libnl 2.0 compatibility code */
  42 #define nl_handle nl_sock
  43 #define nl80211_handle_alloc nl_socket_alloc_cb
  44 #define nl80211_handle_destroy nl_socket_free
  45 #else
  46 /*
  47  * libnl 1.1 has a bug, it tries to allocate socket numbers densely
  48  * but when you free a socket again it will mess up its bitmap and
  49  * and use the wrong number the next time it needs a socket ID.
  50  * Therefore, we wrap the handle alloc/destroy and add our own pid
  51  * accounting.
  52  */
  53 static uint32_t port_bitmap[32] = { 0 };
  54
  55 static struct nl_handle *nl80211_handle_alloc(void *cb)
  56 {
  57         struct nl_handle *handle;
  58         uint32_t pid = getpid() & 0x3FFFFF;
  59         int i;
  60
  61         handle = nl_handle_alloc_cb(cb);
  62
  63         for (i = 0; i < 1024; i++) {
  64                 if (port_bitmap[i / 32] & (1 << (i % 32)))
  65                         continue;
  66                 port_bitmap[i / 32] |= 1 << (i % 32);
  67                 pid += i << 22;
  68                 break;
  69         }
  70
  71         nl_socket_set_local_port(handle, pid);
  72
  73         return handle;
  74 }
  75
  76 static void nl80211_handle_destroy(struct nl_handle *handle)
  77 {
  78         uint32_t port = nl_socket_get_local_port(handle);
  79
  80         port >>= 22;
  81         port_bitmap[port / 32] &= ~(1 << (port % 32));
  82
  83         nl_handle_destroy(handle);
  84 }
  85 #endif /* CONFIG_LIBNL20 */
  86
  87
  88 #ifndef IFF_LOWER_UP
  89 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
  90 #endif
  91 #ifndef IFF_DORMANT
  92 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
  93 #endif
  94
  95 #ifndef IF_OPER_DORMANT
  96 #define IF_OPER_DORMANT 5
  97 #endif
  98 #ifndef IF_OPER_UP
  99 #define IF_OPER_UP 6
 100 #endif
 101
 102 struct i802_bss {
 103         struct wpa_driver_nl80211_data *drv;
 104         struct i802_bss *next;
 105         int ifindex;
 106         char ifname[IFNAMSIZ + 1];
 107         unsigned int beacon_set:1;
 108 };
 109
 110 struct wpa_driver_nl80211_data {
 111         void *ctx;
 112         struct netlink_data *netlink;
 113         int ioctl_sock; /* socket for ioctl() use */
 114         char brname[IFNAMSIZ];
 115         int ifindex;
 116         int if_removed;
 117         int if_disabled;
 118         struct rfkill_data *rfkill;
 119         struct wpa_driver_capa capa;
 120         int has_capability;
 121
 122         int operstate;
 123
 124         int scan_complete_events;
 125
 126         struct nl_handle *nl_handle;
 127         struct nl_handle *nl_handle_event;
 128         struct nl_handle *nl_handle_preq;
 129         struct nl_cache *nl_cache;
 130         struct nl_cache *nl_cache_event;
 131         struct nl_cache *nl_cache_preq;
 132         struct nl_cb *nl_cb;
 133         struct genl_family *nl80211;
 134
 135         u8 auth_bssid[ETH_ALEN];
 136         u8 bssid[ETH_ALEN];
 137         int associated;
 138         u8 ssid[32];
 139         size_t ssid_len;
 140         int nlmode;
 141         int ap_scan_as_station;
 142         unsigned int assoc_freq;
 143
 144         int monitor_sock;
 145         int monitor_ifidx;
 146         int disable_11b_rates;
 147
 148         unsigned int pending_remain_on_chan:1;
 149         unsigned int added_bridge:1;
 150         unsigned int added_if_into_bridge:1;
 151
 152         u64 remain_on_chan_cookie;
 153         u64 send_action_cookie;
 154
 155         unsigned int last_mgmt_freq;
 156
 157         struct wpa_driver_scan_filter *filter_ssids;
 158         size_t num_filter_ssids;
 159
 160         struct i802_bss first_bss;
 161
 162 #ifdef HOSTAPD
 163         int eapol_sock; /* socket for EAPOL frames */
 164
 165         int default_if_indices[16];
 166         int *if_indices;
 167         int num_if_indices;
 168
 169         int last_freq;
 170         int last_freq_ht;
 171 #endif /* HOSTAPD */
 172 };
 173
 174
 175 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
 176                                             void *timeout_ctx);
 177 static int wpa_driver_nl80211_set_mode(void *priv, int mode);
 178 static int
 179 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
 180 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
 181                                    const u8 *addr, int cmd, u16 reason_code,
 182                                    int local_state_change);
 183 static void nl80211_remove_monitor_interface(
 184         struct wpa_driver_nl80211_data *drv);
 185 static int nl80211_send_frame_cmd(struct wpa_driver_nl80211_data *drv,
 186                                   unsigned int freq, const u8 *buf,
 187                                   size_t buf_len, u64 *cookie);
 188 static int wpa_driver_nl80211_probe_req_report(void *priv, int report);
 189
 190 #ifdef HOSTAPD
 191 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
 192 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
 193 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx);
 194 static int wpa_driver_nl80211_if_remove(void *priv,
 195                                         enum wpa_driver_if_type type,
 196                                         const char *ifname);
 197 #else /* HOSTAPD */
 198 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
 199 {
 200         return 0;
 201 }
 202 #endif /* HOSTAPD */
 203
 204 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq);
 205 static int nl80211_disable_11b_rates(struct wpa_driver_nl80211_data *drv,
 206                                      int ifindex, int disabled);
 207
 208
 209 /* nl80211 code */
 210 static int ack_handler(struct nl_msg *msg, void *arg)
 211 {
 212         int *err = arg;
 213         *err = 0;
 214         return NL_STOP;
 215 }
 216
 217 static int finish_handler(struct nl_msg *msg, void *arg)
 218 {
 219         int *ret = arg;
 220         *ret = 0;
 221         return NL_SKIP;
 222 }
 223
 224 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
 225                          void *arg)
 226 {
 227         int *ret = arg;
 228         *ret = err->error;
 229         return NL_SKIP;
 230 }
 231
 232
 233 static int no_seq_check(struct nl_msg *msg, void *arg)
 234 {
 235         return NL_OK;
 236 }
 237
 238
 239 static int send_and_recv(struct wpa_driver_nl80211_data *drv,
 240                          struct nl_handle *nl_handle, struct nl_msg *msg,
 241                          int (*valid_handler)(struct nl_msg *, void *),
 242                          void *valid_data)
 243 {
 244         struct nl_cb *cb;
 245         int err = -ENOMEM;
 246
 247         cb = nl_cb_clone(drv->nl_cb);
 248         if (!cb)
 249                 goto out;
 250
 251         err = nl_send_auto_complete(nl_handle, msg);
 252         if (err < 0)
 253                 goto out;
 254
 255         err = 1;
 256
 257         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
 258         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
 259         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
 260
 261         if (valid_handler)
 262                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
 263                           valid_handler, valid_data);
 264
 265         while (err > 0)
 266                 nl_recvmsgs(nl_handle, cb);
 267  out:
 268         nl_cb_put(cb);
 269         nlmsg_free(msg);
 270         return err;
 271 }
 272
 273
 274 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
 275                               struct nl_msg *msg,
 276                               int (*valid_handler)(struct nl_msg *, void *),
 277                               void *valid_data)
 278 {
 279         return send_and_recv(drv, drv->nl_handle, msg, valid_handler,
 280                              valid_data);
 281 }
 282
 283
 284 struct family_data {
 285         const char *group;
 286         int id;
 287 };
 288
 289
 290 static int family_handler(struct nl_msg *msg, void *arg)
 291 {
 292         struct family_data *res = arg;
 293         struct nlattr *tb[CTRL_ATTR_MAX + 1];
 294         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 295         struct nlattr *mcgrp;
 296         int i;
 297
 298         nla_parse(tb, CTRL_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 299                   genlmsg_attrlen(gnlh, 0), NULL);
 300         if (!tb[CTRL_ATTR_MCAST_GROUPS])
 301                 return NL_SKIP;
 302
 303         nla_for_each_nested(mcgrp, tb[CTRL_ATTR_MCAST_GROUPS], i) {
 304                 struct nlattr *tb2[CTRL_ATTR_MCAST_GRP_MAX + 1];
 305                 nla_parse(tb2, CTRL_ATTR_MCAST_GRP_MAX, nla_data(mcgrp),
 306                           nla_len(mcgrp), NULL);
 307                 if (!tb2[CTRL_ATTR_MCAST_GRP_NAME] ||
 308                     !tb2[CTRL_ATTR_MCAST_GRP_ID] ||
 309                     os_strncmp(nla_data(tb2[CTRL_ATTR_MCAST_GRP_NAME]),
 310                                res->group,
 311                                nla_len(tb2[CTRL_ATTR_MCAST_GRP_NAME])) != 0)
 312                         continue;
 313                 res->id = nla_get_u32(tb2[CTRL_ATTR_MCAST_GRP_ID]);
 314                 break;
 315         };
 316
 317         return NL_SKIP;
 318 }
 319
 320
 321 static int nl_get_multicast_id(struct wpa_driver_nl80211_data *drv,
 322                                const char *family, const char *group)
 323 {
 324         struct nl_msg *msg;
 325         int ret = -1;
 326         struct family_data res = { group, -ENOENT };
 327
 328         msg = nlmsg_alloc();
 329         if (!msg)
 330                 return -ENOMEM;
 331         genlmsg_put(msg, 0, 0, genl_ctrl_resolve(drv->nl_handle, "nlctrl"),
 332                     0, 0, CTRL_CMD_GETFAMILY, 0);
 333         NLA_PUT_STRING(msg, CTRL_ATTR_FAMILY_NAME, family);
 334
 335         ret = send_and_recv_msgs(drv, msg, family_handler, &res);
 336         msg = NULL;
 337         if (ret == 0)
 338                 ret = res.id;
 339
 340 nla_put_failure:
 341         nlmsg_free(msg);
 342         return ret;
 343 }
 344
 345
 346 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
 347 {
 348         struct i802_bss *bss = priv;
 349         struct wpa_driver_nl80211_data *drv = bss->drv;
 350         if (!drv->associated)
 351                 return -1;
 352         os_memcpy(bssid, drv->bssid, ETH_ALEN);
 353         return 0;
 354 }
 355
 356
 357 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
 358 {
 359         struct i802_bss *bss = priv;
 360         struct wpa_driver_nl80211_data *drv = bss->drv;
 361         if (!drv->associated)
 362                 return -1;
 363         os_memcpy(ssid, drv->ssid, drv->ssid_len);
 364         return drv->ssid_len;
 365 }
 366
 367
 368 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
 369                                           char *buf, size_t len, int del)
 370 {
 371         union wpa_event_data event;
 372
 373         os_memset(&event, 0, sizeof(event));
 374         if (len > sizeof(event.interface_status.ifname))
 375                 len = sizeof(event.interface_status.ifname) - 1;
 376         os_memcpy(event.interface_status.ifname, buf, len);
 377         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
 378                 EVENT_INTERFACE_ADDED;
 379
 380         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
 381                    del ? "DEL" : "NEW",
 382                    event.interface_status.ifname,
 383                    del ? "removed" : "added");
 384
 385         if (os_strcmp(drv->first_bss.ifname, event.interface_status.ifname) == 0) {
 386                 if (del)
 387                         drv->if_removed = 1;
 388                 else
 389                         drv->if_removed = 0;
 390         }
 391
 392         wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_STATUS, &event);
 393 }
 394
 395
 396 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
 397                                          u8 *buf, size_t len)
 398 {
 399         int attrlen, rta_len;
 400         struct rtattr *attr;
 401
 402         attrlen = len;
 403         attr = (struct rtattr *) buf;
 404
 405         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 406         while (RTA_OK(attr, attrlen)) {
 407                 if (attr->rta_type == IFLA_IFNAME) {
 408                         if (os_strcmp(((char *) attr) + rta_len, drv->first_bss.ifname)
 409                             == 0)
 410                                 return 1;
 411                         else
 412                                 break;
 413                 }
 414                 attr = RTA_NEXT(attr, attrlen);
 415         }
 416
 417         return 0;
 418 }
 419
 420
 421 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
 422                                           int ifindex, u8 *buf, size_t len)
 423 {
 424         if (drv->ifindex == ifindex)
 425                 return 1;
 426
 427         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, buf, len)) {
 428                 drv->first_bss.ifindex = if_nametoindex(drv->first_bss.ifname);
 429                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
 430                            "interface");
 431                 wpa_driver_nl80211_finish_drv_init(drv);
 432                 return 1;
 433         }
 434
 435         return 0;
 436 }
 437
 438
 439 static void wpa_driver_nl80211_event_rtm_newlink(void *ctx,
 440                                                  struct ifinfomsg *ifi,
 441                                                  u8 *buf, size_t len)
 442 {
 443         struct wpa_driver_nl80211_data *drv = ctx;
 444         int attrlen, rta_len;
 445         struct rtattr *attr;
 446         u32 brid = 0;
 447
 448         if (!wpa_driver_nl80211_own_ifindex(drv, ifi->ifi_index, buf, len) &&
 449             !have_ifidx(drv, ifi->ifi_index)) {
 450                 wpa_printf(MSG_DEBUG, "nl80211: Ignore event for foreign "
 451                            "ifindex %d", ifi->ifi_index);
 452                 return;
 453         }
 454
 455         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
 456                    "(%s%s%s%s)",
 457                    drv->operstate, ifi->ifi_flags,
 458                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
 459                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
 460                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
 461                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
 462
 463         if (!drv->if_disabled && !(ifi->ifi_flags & IFF_UP)) {
 464                 wpa_printf(MSG_DEBUG, "nl80211: Interface down");
 465                 drv->if_disabled = 1;
 466                 wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_DISABLED, NULL);
 467         }
 468
 469         if (drv->if_disabled && (ifi->ifi_flags & IFF_UP)) {
 470                 wpa_printf(MSG_DEBUG, "nl80211: Interface up");
 471                 drv->if_disabled = 0;
 472                 wpa_supplicant_event(drv->ctx, EVENT_INTERFACE_ENABLED, NULL);
 473         }
 474
 475         /*
 476          * Some drivers send the association event before the operup event--in
 477          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
 478          * fails. This will hit us when wpa_supplicant does not need to do
 479          * IEEE 802.1X authentication
 480          */
 481         if (drv->operstate == 1 &&
 482             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
 483             !(ifi->ifi_flags & IFF_RUNNING))
 484                 netlink_send_oper_ifla(drv->netlink, drv->ifindex,
 485                                        -1, IF_OPER_UP);
 486
 487         attrlen = len;
 488         attr = (struct rtattr *) buf;
 489         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 490         while (RTA_OK(attr, attrlen)) {
 491                 if (attr->rta_type == IFLA_IFNAME) {
 492                         wpa_driver_nl80211_event_link(
 493                                 drv,
 494                                 ((char *) attr) + rta_len,
 495                                 attr->rta_len - rta_len, 0);
 496                 } else if (attr->rta_type == IFLA_MASTER)
 497                         brid = nla_get_u32((struct nlattr *) attr);
 498                 attr = RTA_NEXT(attr, attrlen);
 499         }
 500
 501 #ifdef HOSTAPD
 502         if (ifi->ifi_family == AF_BRIDGE && brid) {
 503                 /* device has been added to bridge */
 504                 char namebuf[IFNAMSIZ];
 505                 if_indextoname(brid, namebuf);
 506                 wpa_printf(MSG_DEBUG, "nl80211: Add ifindex %u for bridge %s",
 507                            brid, namebuf);
 508                 add_ifidx(drv, brid);
 509         }
 510 #endif /* HOSTAPD */
 511 }
 512
 513
 514 static void wpa_driver_nl80211_event_rtm_dellink(void *ctx,
 515                                                  struct ifinfomsg *ifi,
 516                                                  u8 *buf, size_t len)
 517 {
 518         struct wpa_driver_nl80211_data *drv = ctx;
 519         int attrlen, rta_len;
 520         struct rtattr *attr;
 521         u32 brid = 0;
 522
 523         attrlen = len;
 524         attr = (struct rtattr *) buf;
 525
 526         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 527         while (RTA_OK(attr, attrlen)) {
 528                 if (attr->rta_type == IFLA_IFNAME) {
 529                         wpa_driver_nl80211_event_link(
 530                                 drv,
 531                                 ((char *) attr) + rta_len,
 532                                 attr->rta_len - rta_len, 1);
 533                 } else if (attr->rta_type == IFLA_MASTER)
 534                         brid = nla_get_u32((struct nlattr *) attr);
 535                 attr = RTA_NEXT(attr, attrlen);
 536         }
 537
 538 #ifdef HOSTAPD
 539         if (ifi->ifi_family == AF_BRIDGE && brid) {
 540                 /* device has been removed from bridge */
 541                 char namebuf[IFNAMSIZ];
 542                 if_indextoname(brid, namebuf);
 543                 wpa_printf(MSG_DEBUG, "nl80211: Remove ifindex %u for bridge "
 544                            "%s", brid, namebuf);
 545                 del_ifidx(drv, brid);
 546         }
 547 #endif /* HOSTAPD */
 548 }
 549
 550
 551 static void mlme_event_auth(struct wpa_driver_nl80211_data *drv,
 552                             const u8 *frame, size_t len)
 553 {
 554         const struct ieee80211_mgmt *mgmt;
 555         union wpa_event_data event;
 556
 557         mgmt = (const struct ieee80211_mgmt *) frame;
 558         if (len < 24 + sizeof(mgmt->u.auth)) {
 559                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
 560                            "frame");
 561                 return;
 562         }
 563
 564         os_memcpy(drv->auth_bssid, mgmt->sa, ETH_ALEN);
 565         os_memset(&event, 0, sizeof(event));
 566         os_memcpy(event.auth.peer, mgmt->sa, ETH_ALEN);
 567         event.auth.auth_type = le_to_host16(mgmt->u.auth.auth_alg);
 568         event.auth.status_code = le_to_host16(mgmt->u.auth.status_code);
 569         if (len > 24 + sizeof(mgmt->u.auth)) {
 570                 event.auth.ies = mgmt->u.auth.variable;
 571                 event.auth.ies_len = len - 24 - sizeof(mgmt->u.auth);
 572         }
 573
 574         wpa_supplicant_event(drv->ctx, EVENT_AUTH, &event);
 575 }
 576
 577
 578 static void mlme_event_assoc(struct wpa_driver_nl80211_data *drv,
 579                             const u8 *frame, size_t len)
 580 {
 581         const struct ieee80211_mgmt *mgmt;
 582         union wpa_event_data event;
 583         u16 status;
 584
 585         mgmt = (const struct ieee80211_mgmt *) frame;
 586         if (len < 24 + sizeof(mgmt->u.assoc_resp)) {
 587                 wpa_printf(MSG_DEBUG, "nl80211: Too short association event "
 588                            "frame");
 589                 return;
 590         }
 591
 592         status = le_to_host16(mgmt->u.assoc_resp.status_code);
 593         if (status != WLAN_STATUS_SUCCESS) {
 594                 os_memset(&event, 0, sizeof(event));
 595                 if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
 596                         event.assoc_reject.resp_ies =
 597                                 (u8 *) mgmt->u.assoc_resp.variable;
 598                         event.assoc_reject.resp_ies_len =
 599                                 len - 24 - sizeof(mgmt->u.assoc_resp);
 600                 }
 601                 event.assoc_reject.status_code = status;
 602
 603                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
 604                 return;
 605         }
 606
 607         drv->associated = 1;
 608         os_memcpy(drv->bssid, mgmt->sa, ETH_ALEN);
 609
 610         os_memset(&event, 0, sizeof(event));
 611         if (len > 24 + sizeof(mgmt->u.assoc_resp)) {
 612                 event.assoc_info.resp_ies = (u8 *) mgmt->u.assoc_resp.variable;
 613                 event.assoc_info.resp_ies_len =
 614                         len - 24 - sizeof(mgmt->u.assoc_resp);
 615         }
 616
 617         event.assoc_info.freq = drv->assoc_freq;
 618
 619         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
 620 }
 621
 622
 623 static void mlme_event_connect(struct wpa_driver_nl80211_data *drv,
 624                                enum nl80211_commands cmd, struct nlattr *status,
 625                                struct nlattr *addr, struct nlattr *req_ie,
 626                                struct nlattr *resp_ie)
 627 {
 628         union wpa_event_data event;
 629
 630         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
 631                 /*
 632                  * Avoid reporting two association events that would confuse
 633                  * the core code.
 634                  */
 635                 wpa_printf(MSG_DEBUG, "nl80211: Ignore connect event (cmd=%d) "
 636                            "when using userspace SME", cmd);
 637                 return;
 638         }
 639
 640         os_memset(&event, 0, sizeof(event));
 641         if (cmd == NL80211_CMD_CONNECT &&
 642             nla_get_u16(status) != WLAN_STATUS_SUCCESS) {
 643                 if (resp_ie) {
 644                         event.assoc_reject.resp_ies = nla_data(resp_ie);
 645                         event.assoc_reject.resp_ies_len = nla_len(resp_ie);
 646                 }
 647                 event.assoc_reject.status_code = nla_get_u16(status);
 648                 wpa_supplicant_event(drv->ctx, EVENT_ASSOC_REJECT, &event);
 649                 return;
 650         }
 651
 652         drv->associated = 1;
 653         if (addr)
 654                 os_memcpy(drv->bssid, nla_data(addr), ETH_ALEN);
 655
 656         if (req_ie) {
 657                 event.assoc_info.req_ies = nla_data(req_ie);
 658                 event.assoc_info.req_ies_len = nla_len(req_ie);
 659         }
 660         if (resp_ie) {
 661                 event.assoc_info.resp_ies = nla_data(resp_ie);
 662                 event.assoc_info.resp_ies_len = nla_len(resp_ie);
 663         }
 664
 665         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, &event);
 666 }
 667
 668
 669 static void mlme_timeout_event(struct wpa_driver_nl80211_data *drv,
 670                                enum nl80211_commands cmd, struct nlattr *addr)
 671 {
 672         union wpa_event_data event;
 673         enum wpa_event_type ev;
 674
 675         if (nla_len(addr) != ETH_ALEN)
 676                 return;
 677
 678         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d; timeout with " MACSTR,
 679                    cmd, MAC2STR((u8 *) nla_data(addr)));
 680
 681         if (cmd == NL80211_CMD_AUTHENTICATE)
 682                 ev = EVENT_AUTH_TIMED_OUT;
 683         else if (cmd == NL80211_CMD_ASSOCIATE)
 684                 ev = EVENT_ASSOC_TIMED_OUT;
 685         else
 686                 return;
 687
 688         os_memset(&event, 0, sizeof(event));
 689         os_memcpy(event.timeout_event.addr, nla_data(addr), ETH_ALEN);
 690         wpa_supplicant_event(drv->ctx, ev, &event);
 691 }
 692
 693
 694 static void mlme_event_mgmt(struct wpa_driver_nl80211_data *drv,
 695                             struct nlattr *freq, const u8 *frame, size_t len)
 696 {
 697         const struct ieee80211_mgmt *mgmt;
 698         union wpa_event_data event;
 699         u16 fc, stype;
 700
 701         mgmt = (const struct ieee80211_mgmt *) frame;
 702         if (len < 24) {
 703                 wpa_printf(MSG_DEBUG, "nl80211: Too short action frame");
 704                 return;
 705         }
 706
 707         fc = le_to_host16(mgmt->frame_control);
 708         stype = WLAN_FC_GET_STYPE(fc);
 709
 710         os_memset(&event, 0, sizeof(event));
 711         if (freq) {
 712                 event.rx_action.freq = nla_get_u32(freq);
 713                 drv->last_mgmt_freq = event.rx_action.freq;
 714         }
 715         if (stype == WLAN_FC_STYPE_ACTION) {
 716                 event.rx_action.da = mgmt->da;
 717                 event.rx_action.sa = mgmt->sa;
 718                 event.rx_action.bssid = mgmt->bssid;
 719                 event.rx_action.category = mgmt->u.action.category;
 720                 event.rx_action.data = &mgmt->u.action.category + 1;
 721                 event.rx_action.len = frame + len - event.rx_action.data;
 722                 wpa_supplicant_event(drv->ctx, EVENT_RX_ACTION, &event);
 723         } else {
 724                 event.rx_mgmt.frame = frame;
 725                 event.rx_mgmt.frame_len = len;
 726                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
 727         }
 728 }
 729
 730
 731 static void mlme_event_action_tx_status(struct wpa_driver_nl80211_data *drv,
 732                                         struct nlattr *cookie, const u8 *frame,
 733                                         size_t len, struct nlattr *ack)
 734 {
 735         union wpa_event_data event;
 736         const struct ieee80211_hdr *hdr;
 737         u16 fc;
 738         u64 cookie_val;
 739
 740         if (!cookie)
 741                 return;
 742
 743         cookie_val = nla_get_u64(cookie);
 744         wpa_printf(MSG_DEBUG, "nl80211: Action TX status: cookie=0%llx%s "
 745                    "(ack=%d)",
 746                    (long long unsigned int) cookie_val,
 747                    cookie_val == drv->send_action_cookie ?
 748                    " (match)" : " (unknown)", ack != NULL);
 749         if (cookie_val != drv->send_action_cookie)
 750                 return;
 751
 752         hdr = (const struct ieee80211_hdr *) frame;
 753         fc = le_to_host16(hdr->frame_control);
 754
 755         os_memset(&event, 0, sizeof(event));
 756         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
 757         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
 758         event.tx_status.dst = hdr->addr1;
 759         event.tx_status.data = frame;
 760         event.tx_status.data_len = len;
 761         event.tx_status.ack = ack != NULL;
 762         wpa_supplicant_event(drv->ctx, EVENT_TX_STATUS, &event);
 763 }
 764
 765
 766 static void mlme_event_deauth_disassoc(struct wpa_driver_nl80211_data *drv,
 767                                        enum wpa_event_type type,
 768                                        const u8 *frame, size_t len)
 769 {
 770         const struct ieee80211_mgmt *mgmt;
 771         union wpa_event_data event;
 772         const u8 *bssid = NULL;
 773         u16 reason_code = 0;
 774
 775         mgmt = (const struct ieee80211_mgmt *) frame;
 776         if (len >= 24) {
 777                 bssid = mgmt->bssid;
 778
 779                 if (drv->associated != 0 &&
 780                     os_memcmp(bssid, drv->bssid, ETH_ALEN) != 0 &&
 781                     os_memcmp(bssid, drv->auth_bssid, ETH_ALEN) != 0) {
 782                         /*
 783                          * We have presumably received this deauth as a
 784                          * response to a clear_state_mismatch() outgoing
 785                          * deauth.  Don't let it take us offline!
 786                          */
 787                         wpa_printf(MSG_DEBUG, "nl80211: Deauth received "
 788                                    "from Unknown BSSID " MACSTR " -- ignoring",
 789                                    MAC2STR(bssid));
 790                         return;
 791                 }
 792         }
 793
 794         drv->associated = 0;
 795         os_memset(&event, 0, sizeof(event));
 796
 797         /* Note: Same offset for Reason Code in both frame subtypes */
 798         if (len >= 24 + sizeof(mgmt->u.deauth))
 799                 reason_code = le_to_host16(mgmt->u.deauth.reason_code);
 800
 801         if (type == EVENT_DISASSOC) {
 802                 event.disassoc_info.addr = bssid;
 803                 event.disassoc_info.reason_code = reason_code;
 804                 if (frame + len > mgmt->u.disassoc.variable) {
 805                         event.disassoc_info.ie = mgmt->u.disassoc.variable;
 806                         event.disassoc_info.ie_len = frame + len -
 807                                 mgmt->u.disassoc.variable;
 808                 }
 809         } else {
 810                 event.deauth_info.addr = bssid;
 811                 event.deauth_info.reason_code = reason_code;
 812                 if (frame + len > mgmt->u.deauth.variable) {
 813                         event.deauth_info.ie = mgmt->u.deauth.variable;
 814                         event.deauth_info.ie_len = frame + len -
 815                                 mgmt->u.deauth.variable;
 816                 }
 817         }
 818
 819         wpa_supplicant_event(drv->ctx, type, &event);
 820 }
 821
 822
 823 static void mlme_event(struct wpa_driver_nl80211_data *drv,
 824                        enum nl80211_commands cmd, struct nlattr *frame,
 825                        struct nlattr *addr, struct nlattr *timed_out,
 826                        struct nlattr *freq, struct nlattr *ack,
 827                        struct nlattr *cookie)
 828 {
 829         if (timed_out && addr) {
 830                 mlme_timeout_event(drv, cmd, addr);
 831                 return;
 832         }
 833
 834         if (frame == NULL) {
 835                 wpa_printf(MSG_DEBUG, "nl80211: MLME event %d without frame "
 836                            "data", cmd);
 837                 return;
 838         }
 839
 840         wpa_printf(MSG_DEBUG, "nl80211: MLME event %d", cmd);
 841         wpa_hexdump(MSG_MSGDUMP, "nl80211: MLME event frame",
 842                     nla_data(frame), nla_len(frame));
 843
 844         switch (cmd) {
 845         case NL80211_CMD_AUTHENTICATE:
 846                 mlme_event_auth(drv, nla_data(frame), nla_len(frame));
 847                 break;
 848         case NL80211_CMD_ASSOCIATE:
 849                 mlme_event_assoc(drv, nla_data(frame), nla_len(frame));
 850                 break;
 851         case NL80211_CMD_DEAUTHENTICATE:
 852                 mlme_event_deauth_disassoc(drv, EVENT_DEAUTH,
 853                                            nla_data(frame), nla_len(frame));
 854                 break;
 855         case NL80211_CMD_DISASSOCIATE:
 856                 mlme_event_deauth_disassoc(drv, EVENT_DISASSOC,
 857                                            nla_data(frame), nla_len(frame));
 858                 break;
 859         case NL80211_CMD_FRAME:
 860                 mlme_event_mgmt(drv, freq, nla_data(frame), nla_len(frame));
 861                 break;
 862         case NL80211_CMD_FRAME_TX_STATUS:
 863                 mlme_event_action_tx_status(drv, cookie, nla_data(frame),
 864                                             nla_len(frame), ack);
 865                 break;
 866         default:
 867                 break;
 868         }
 869 }
 870
 871
 872 static void mlme_event_michael_mic_failure(struct wpa_driver_nl80211_data *drv,
 873                                            struct nlattr *tb[])
 874 {
 875         union wpa_event_data data;
 876
 877         wpa_printf(MSG_DEBUG, "nl80211: MLME event Michael MIC failure");
 878         os_memset(&data, 0, sizeof(data));
 879         if (tb[NL80211_ATTR_MAC]) {
 880                 wpa_hexdump(MSG_DEBUG, "nl80211: Source MAC address",
 881                             nla_data(tb[NL80211_ATTR_MAC]),
 882                             nla_len(tb[NL80211_ATTR_MAC]));
 883                 data.michael_mic_failure.src = nla_data(tb[NL80211_ATTR_MAC]);
 884         }
 885         if (tb[NL80211_ATTR_KEY_SEQ]) {
 886                 wpa_hexdump(MSG_DEBUG, "nl80211: TSC",
 887                             nla_data(tb[NL80211_ATTR_KEY_SEQ]),
 888                             nla_len(tb[NL80211_ATTR_KEY_SEQ]));
 889         }
 890         if (tb[NL80211_ATTR_KEY_TYPE]) {
 891                 enum nl80211_key_type key_type =
 892                         nla_get_u32(tb[NL80211_ATTR_KEY_TYPE]);
 893                 wpa_printf(MSG_DEBUG, "nl80211: Key Type %d", key_type);
 894                 if (key_type == NL80211_KEYTYPE_PAIRWISE)
 895                         data.michael_mic_failure.unicast = 1;
 896         } else
 897                 data.michael_mic_failure.unicast = 1;
 898
 899         if (tb[NL80211_ATTR_KEY_IDX]) {
 900                 u8 key_id = nla_get_u8(tb[NL80211_ATTR_KEY_IDX]);
 901                 wpa_printf(MSG_DEBUG, "nl80211: Key Id %d", key_id);
 902         }
 903
 904         wpa_supplicant_event(drv->ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
 905 }
 906
 907
 908 static void mlme_event_join_ibss(struct wpa_driver_nl80211_data *drv,
 909                                  struct nlattr *tb[])
 910 {
 911         if (tb[NL80211_ATTR_MAC] == NULL) {
 912                 wpa_printf(MSG_DEBUG, "nl80211: No address in IBSS joined "
 913                            "event");
 914                 return;
 915         }
 916         os_memcpy(drv->bssid, nla_data(tb[NL80211_ATTR_MAC]), ETH_ALEN);
 917         drv->associated = 1;
 918         wpa_printf(MSG_DEBUG, "nl80211: IBSS " MACSTR " joined",
 919                    MAC2STR(drv->bssid));
 920
 921         wpa_supplicant_event(drv->ctx, EVENT_ASSOC, NULL);
 922 }
 923
 924
 925 static void mlme_event_remain_on_channel(struct wpa_driver_nl80211_data *drv,
 926                                          int cancel_event, struct nlattr *tb[])
 927 {
 928         unsigned int freq, chan_type, duration;
 929         union wpa_event_data data;
 930         u64 cookie;
 931
 932         if (tb[NL80211_ATTR_WIPHY_FREQ])
 933                 freq = nla_get_u32(tb[NL80211_ATTR_WIPHY_FREQ]);
 934         else
 935                 freq = 0;
 936
 937         if (tb[NL80211_ATTR_WIPHY_CHANNEL_TYPE])
 938                 chan_type = nla_get_u32(tb[NL80211_ATTR_WIPHY_CHANNEL_TYPE]);
 939         else
 940                 chan_type = 0;
 941
 942         if (tb[NL80211_ATTR_DURATION])
 943                 duration = nla_get_u32(tb[NL80211_ATTR_DURATION]);
 944         else
 945                 duration = 0;
 946
 947         if (tb[NL80211_ATTR_COOKIE])
 948                 cookie = nla_get_u64(tb[NL80211_ATTR_COOKIE]);
 949         else
 950                 cookie = 0;
 951
 952         wpa_printf(MSG_DEBUG, "nl80211: Remain-on-channel event (cancel=%d "
 953                    "freq=%u channel_type=%u duration=%u cookie=0x%llx (%s))",
 954                    cancel_event, freq, chan_type, duration,
 955                    (long long unsigned int) cookie,
 956                    cookie == drv->remain_on_chan_cookie ? "match" : "unknown");
 957
 958         if (cookie != drv->remain_on_chan_cookie)
 959                 return; /* not for us */
 960
 961         drv->pending_remain_on_chan = !cancel_event;
 962
 963         os_memset(&data, 0, sizeof(data));
 964         data.remain_on_channel.freq = freq;
 965         data.remain_on_channel.duration = duration;
 966         wpa_supplicant_event(drv->ctx, cancel_event ?
 967                              EVENT_CANCEL_REMAIN_ON_CHANNEL :
 968                              EVENT_REMAIN_ON_CHANNEL, &data);
 969 }
 970
 971
 972 static void send_scan_event(struct wpa_driver_nl80211_data *drv, int aborted,
 973                             struct nlattr *tb[])
 974 {
 975         union wpa_event_data event;
 976         struct nlattr *nl;
 977         int rem;
 978         struct scan_info *info;
 979 #define MAX_REPORT_FREQS 50
 980         int freqs[MAX_REPORT_FREQS];
 981         int num_freqs = 0;
 982
 983         os_memset(&event, 0, sizeof(event));
 984         info = &event.scan_info;
 985         info->aborted = aborted;
 986
 987         if (tb[NL80211_ATTR_SCAN_SSIDS]) {
 988                 nla_for_each_nested(nl, tb[NL80211_ATTR_SCAN_SSIDS], rem) {
 989                         struct wpa_driver_scan_ssid *s =
 990                                 &info->ssids[info->num_ssids];
 991                         s->ssid = nla_data(nl);
 992                         s->ssid_len = nla_len(nl);
 993                         info->num_ssids++;
 994                         if (info->num_ssids == WPAS_MAX_SCAN_SSIDS)
 995                                 break;
 996                 }
 997         }
 998         if (tb[NL80211_ATTR_SCAN_FREQUENCIES]) {
 999                 nla_for_each_nested(nl, tb[NL80211_ATTR_SCAN_FREQUENCIES], rem)
1000                 {
1001                         freqs[num_freqs] = nla_get_u32(nl);
1002                         num_freqs++;
1003                         if (num_freqs == MAX_REPORT_FREQS - 1)
1004                                 break;
1005                 }
1006                 info->freqs = freqs;
1007                 info->num_freqs = num_freqs;
1008         }
1009         wpa_supplicant_event(drv->ctx, EVENT_SCAN_RESULTS, &event);
1010 }
1011
1012
1013 static int get_link_signal(struct nl_msg *msg, void *arg)
1014 {
1015         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1016         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1017         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1018         static struct nla_policy policy[NL80211_STA_INFO_MAX + 1] = {
1019                 [NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 },
1020         };
1021         int *sig = arg;
1022
1023         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1024                   genlmsg_attrlen(gnlh, 0), NULL);
1025         if (!tb[NL80211_ATTR_STA_INFO] ||
1026             nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1027                              tb[NL80211_ATTR_STA_INFO], policy))
1028                 return NL_SKIP;
1029         if (!sinfo[NL80211_STA_INFO_SIGNAL])
1030                 return NL_SKIP;
1031
1032         *sig = (s8) nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1033         return NL_SKIP;
1034 }
1035
1036
1037 static int nl80211_get_link_signal(struct wpa_driver_nl80211_data *drv,
1038                                    int *sig)
1039 {
1040         struct nl_msg *msg;
1041
1042         *sig = -9999;
1043
1044         msg = nlmsg_alloc();
1045         if (!msg)
1046                 return -ENOMEM;
1047
1048         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1049                     0, NL80211_CMD_GET_STATION, 0);
1050
1051         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1052         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, drv->bssid);
1053
1054         return send_and_recv_msgs(drv, msg, get_link_signal, sig);
1055  nla_put_failure:
1056         return -ENOBUFS;
1057 }
1058
1059
1060 static void nl80211_cqm_event(struct wpa_driver_nl80211_data *drv,
1061                               struct nlattr *tb[])
1062 {
1063         static struct nla_policy cqm_policy[NL80211_ATTR_CQM_MAX + 1] = {
1064                 [NL80211_ATTR_CQM_RSSI_THOLD] = { .type = NLA_U32 },
1065                 [NL80211_ATTR_CQM_RSSI_HYST] = { .type = NLA_U8 },
1066                 [NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] = { .type = NLA_U32 },
1067         };
1068         struct nlattr *cqm[NL80211_ATTR_CQM_MAX + 1];
1069         enum nl80211_cqm_rssi_threshold_event event;
1070         union wpa_event_data ed;
1071         int sig, res;
1072
1073         if (tb[NL80211_ATTR_CQM] == NULL ||
1074             nla_parse_nested(cqm, NL80211_ATTR_CQM_MAX, tb[NL80211_ATTR_CQM],
1075                              cqm_policy)) {
1076                 wpa_printf(MSG_DEBUG, "nl80211: Ignore invalid CQM event");
1077                 return;
1078         }
1079
1080         if (cqm[NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT] == NULL)
1081                 return;
1082         event = nla_get_u32(cqm[NL80211_ATTR_CQM_RSSI_THRESHOLD_EVENT]);
1083
1084         os_memset(&ed, 0, sizeof(ed));
1085
1086         if (event == NL80211_CQM_RSSI_THRESHOLD_EVENT_HIGH) {
1087                 wpa_printf(MSG_DEBUG, "nl80211: Connection quality monitor "
1088                            "event: RSSI high");
1089                 ed.signal_change.above_threshold = 1;
1090         } else if (event == NL80211_CQM_RSSI_THRESHOLD_EVENT_LOW) {
1091                 wpa_printf(MSG_DEBUG, "nl80211: Connection quality monitor "
1092                            "event: RSSI low");
1093                 ed.signal_change.above_threshold = 0;
1094         } else
1095                 return;
1096
1097         res = nl80211_get_link_signal(drv, &sig);
1098         if (res == 0) {
1099                 ed.signal_change.current_signal = sig;
1100                 wpa_printf(MSG_DEBUG, "nl80211: Signal: %d dBm", sig);
1101         }
1102
1103         wpa_supplicant_event(drv->ctx, EVENT_SIGNAL_CHANGE, &ed);
1104 }
1105
1106
1107 static int process_event(struct nl_msg *msg, void *arg)
1108 {
1109         struct wpa_driver_nl80211_data *drv = arg;
1110         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1111         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1112         union wpa_event_data data;
1113
1114         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1115                   genlmsg_attrlen(gnlh, 0), NULL);
1116
1117         if (tb[NL80211_ATTR_IFINDEX]) {
1118                 int ifindex = nla_get_u32(tb[NL80211_ATTR_IFINDEX]);
1119                 if (ifindex != drv->ifindex && !have_ifidx(drv, ifindex)) {
1120                         wpa_printf(MSG_DEBUG, "nl80211: Ignored event (cmd=%d)"
1121                                    " for foreign interface (ifindex %d)",
1122                                    gnlh->cmd, ifindex);
1123                         return NL_SKIP;
1124                 }
1125         }
1126
1127         if (drv->ap_scan_as_station &&
1128             (gnlh->cmd == NL80211_CMD_NEW_SCAN_RESULTS ||
1129              gnlh->cmd == NL80211_CMD_SCAN_ABORTED)) {
1130                 wpa_driver_nl80211_set_mode(&drv->first_bss,
1131                                             IEEE80211_MODE_AP);
1132                 drv->ap_scan_as_station = 0;
1133         }
1134
1135         switch (gnlh->cmd) {
1136         case NL80211_CMD_TRIGGER_SCAN:
1137                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger");
1138                 break;
1139         case NL80211_CMD_NEW_SCAN_RESULTS:
1140                 wpa_printf(MSG_DEBUG, "nl80211: New scan results available");
1141                 drv->scan_complete_events = 1;
1142                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
1143                                      drv->ctx);
1144                 send_scan_event(drv, 0, tb);
1145                 break;
1146         case NL80211_CMD_SCAN_ABORTED:
1147                 wpa_printf(MSG_DEBUG, "nl80211: Scan aborted");
1148                 /*
1149                  * Need to indicate that scan results are available in order
1150                  * not to make wpa_supplicant stop its scanning.
1151                  */
1152                 eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv,
1153                                      drv->ctx);
1154                 send_scan_event(drv, 1, tb);
1155                 break;
1156         case NL80211_CMD_AUTHENTICATE:
1157         case NL80211_CMD_ASSOCIATE:
1158         case NL80211_CMD_DEAUTHENTICATE:
1159         case NL80211_CMD_DISASSOCIATE:
1160         case NL80211_CMD_FRAME:
1161         case NL80211_CMD_FRAME_TX_STATUS:
1162                 mlme_event(drv, gnlh->cmd, tb[NL80211_ATTR_FRAME],
1163                            tb[NL80211_ATTR_MAC], tb[NL80211_ATTR_TIMED_OUT],
1164                            tb[NL80211_ATTR_WIPHY_FREQ], tb[NL80211_ATTR_ACK],
1165                            tb[NL80211_ATTR_COOKIE]);
1166                 break;
1167         case NL80211_CMD_CONNECT:
1168         case NL80211_CMD_ROAM:
1169                 mlme_event_connect(drv, gnlh->cmd,
1170                                    tb[NL80211_ATTR_STATUS_CODE],
1171                                    tb[NL80211_ATTR_MAC],
1172                                    tb[NL80211_ATTR_REQ_IE],
1173                                    tb[NL80211_ATTR_RESP_IE]);
1174                 break;
1175         case NL80211_CMD_DISCONNECT:
1176                 if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
1177                         /*
1178                          * Avoid reporting two disassociation events that could
1179                          * confuse the core code.
1180                          */
1181                         wpa_printf(MSG_DEBUG, "nl80211: Ignore disconnect "
1182                                    "event when using userspace SME");
1183                         break;
1184                 }
1185                 drv->associated = 0;
1186                 os_memset(&data, 0, sizeof(data));
1187                 if (tb[NL80211_ATTR_REASON_CODE])
1188                         data.disassoc_info.reason_code =
1189                                 nla_get_u16(tb[NL80211_ATTR_REASON_CODE]);
1190                 wpa_supplicant_event(drv->ctx, EVENT_DISASSOC, &data);
1191                 break;
1192         case NL80211_CMD_MICHAEL_MIC_FAILURE:
1193                 mlme_event_michael_mic_failure(drv, tb);
1194                 break;
1195         case NL80211_CMD_JOIN_IBSS:
1196                 mlme_event_join_ibss(drv, tb);
1197                 break;
1198         case NL80211_CMD_REMAIN_ON_CHANNEL:
1199                 mlme_event_remain_on_channel(drv, 0, tb);
1200                 break;
1201         case NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL:
1202                 mlme_event_remain_on_channel(drv, 1, tb);
1203                 break;
1204         case NL80211_CMD_NOTIFY_CQM:
1205                 nl80211_cqm_event(drv, tb);
1206                 break;
1207         default:
1208                 wpa_printf(MSG_DEBUG, "nl80211: Ignored unknown event "
1209                            "(cmd=%d)", gnlh->cmd);
1210                 break;
1211         }
1212
1213         return NL_SKIP;
1214 }
1215
1216
1217 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
1218                                              void *handle)
1219 {
1220         struct nl_cb *cb;
1221         struct wpa_driver_nl80211_data *drv = eloop_ctx;
1222
1223         wpa_printf(MSG_DEBUG, "nl80211: Event message available");
1224
1225         cb = nl_cb_clone(drv->nl_cb);
1226         if (!cb)
1227                 return;
1228         nl_cb_set(cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, no_seq_check, NULL);
1229         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, process_event, drv);
1230         nl_recvmsgs(handle, cb);
1231         nl_cb_put(cb);
1232 }
1233
1234
1235 /**
1236  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
1237  * @priv: driver_nl80211 private data
1238  * @alpha2_arg: country to which to switch to
1239  * Returns: 0 on success, -1 on failure
1240  *
1241  * This asks nl80211 to set the regulatory domain for given
1242  * country ISO / IEC alpha2.
1243  */
1244 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
1245 {
1246         struct i802_bss *bss = priv;
1247         struct wpa_driver_nl80211_data *drv = bss->drv;
1248         char alpha2[3];
1249         struct nl_msg *msg;
1250
1251         msg = nlmsg_alloc();
1252         if (!msg)
1253                 return -ENOMEM;
1254
1255         alpha2[0] = alpha2_arg[0];
1256         alpha2[1] = alpha2_arg[1];
1257         alpha2[2] = '\0';
1258
1259         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1260                     0, NL80211_CMD_REQ_SET_REG, 0);
1261
1262         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
1263         if (send_and_recv_msgs(drv, msg, NULL, NULL))
1264                 return -EINVAL;
1265         return 0;
1266 nla_put_failure:
1267         return -EINVAL;
1268 }
1269
1270
1271 #ifndef HOSTAPD
1272 struct wiphy_info_data {
1273         int max_scan_ssids;
1274         int ap_supported;
1275         int auth_supported;
1276         int connect_supported;
1277 };
1278
1279
1280 static int wiphy_info_handler(struct nl_msg *msg, void *arg)
1281 {
1282         struct nlattr *tb[NL80211_ATTR_MAX + 1];
1283         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1284         struct wiphy_info_data *info = arg;
1285
1286         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1287                   genlmsg_attrlen(gnlh, 0), NULL);
1288
1289         if (tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS])
1290                 info->max_scan_ssids =
1291                         nla_get_u8(tb[NL80211_ATTR_MAX_NUM_SCAN_SSIDS]);
1292
1293         if (tb[NL80211_ATTR_SUPPORTED_IFTYPES]) {
1294                 struct nlattr *nl_mode;
1295                 int i;
1296                 nla_for_each_nested(nl_mode,
1297                                     tb[NL80211_ATTR_SUPPORTED_IFTYPES], i) {
1298                         if (nl_mode->nla_type == NL80211_IFTYPE_AP) {
1299                                 info->ap_supported = 1;
1300                                 break;
1301                         }
1302                 }
1303         }
1304
1305         if (tb[NL80211_ATTR_SUPPORTED_COMMANDS]) {
1306                 struct nlattr *nl_cmd;
1307                 int i;
1308
1309                 nla_for_each_nested(nl_cmd,
1310                                     tb[NL80211_ATTR_SUPPORTED_COMMANDS], i) {
1311                         u32 cmd = nla_get_u32(nl_cmd);
1312                         if (cmd == NL80211_CMD_AUTHENTICATE)
1313                                 info->auth_supported = 1;
1314                         else if (cmd == NL80211_CMD_CONNECT)
1315                                 info->connect_supported = 1;
1316                 }
1317         }
1318
1319         return NL_SKIP;
1320 }
1321
1322
1323 static int wpa_driver_nl80211_get_info(struct wpa_driver_nl80211_data *drv,
1324                                        struct wiphy_info_data *info)
1325 {
1326         struct nl_msg *msg;
1327
1328         os_memset(info, 0, sizeof(*info));
1329         msg = nlmsg_alloc();
1330         if (!msg)
1331                 return -1;
1332
1333         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1334                     0, NL80211_CMD_GET_WIPHY, 0);
1335
1336         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->first_bss.ifindex);
1337
1338         if (send_and_recv_msgs(drv, msg, wiphy_info_handler, info) == 0)
1339                 return 0;
1340         msg = NULL;
1341 nla_put_failure:
1342         nlmsg_free(msg);
1343         return -1;
1344 }
1345
1346
1347 static int wpa_driver_nl80211_capa(struct wpa_driver_nl80211_data *drv)
1348 {
1349         struct wiphy_info_data info;
1350         if (wpa_driver_nl80211_get_info(drv, &info))
1351                 return -1;
1352         drv->has_capability = 1;
1353         /* For now, assume TKIP, CCMP, WPA, WPA2 are supported */
1354         drv->capa.key_mgmt = WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1355                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK |
1356                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1357                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
1358         drv->capa.enc = WPA_DRIVER_CAPA_ENC_WEP40 |
1359                 WPA_DRIVER_CAPA_ENC_WEP104 |
1360                 WPA_DRIVER_CAPA_ENC_TKIP |
1361                 WPA_DRIVER_CAPA_ENC_CCMP;
1362         drv->capa.auth = WPA_DRIVER_AUTH_OPEN |
1363                 WPA_DRIVER_AUTH_SHARED |
1364                 WPA_DRIVER_AUTH_LEAP;
1365
1366         drv->capa.max_scan_ssids = info.max_scan_ssids;
1367         if (info.ap_supported)
1368                 drv->capa.flags |= WPA_DRIVER_FLAGS_AP;
1369
1370         if (info.auth_supported)
1371                 drv->capa.flags |= WPA_DRIVER_FLAGS_SME;
1372         else if (!info.connect_supported) {
1373                 wpa_printf(MSG_INFO, "nl80211: Driver does not support "
1374                            "authentication/association or connect commands");
1375                 return -1;
1376         }
1377
1378         drv->capa.flags |= WPA_DRIVER_FLAGS_SET_KEYS_AFTER_ASSOC_DONE;
1379         drv->capa.flags |= WPA_DRIVER_FLAGS_P2P_CAPABLE;
1380         drv->capa.max_remain_on_chan = 5000;
1381
1382         return 0;
1383 }
1384 #endif /* HOSTAPD */
1385
1386
1387 static int wpa_driver_nl80211_init_nl(struct wpa_driver_nl80211_data *drv)
1388 {
1389         int ret;
1390
1391         /* Initialize generic netlink and nl80211 */
1392
1393         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
1394         if (drv->nl_cb == NULL) {
1395                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1396                            "callbacks");
1397                 goto err1;
1398         }
1399
1400         drv->nl_handle = nl80211_handle_alloc(drv->nl_cb);
1401         if (drv->nl_handle == NULL) {
1402                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1403                            "callbacks");
1404                 goto err2;
1405         }
1406
1407         drv->nl_handle_event = nl80211_handle_alloc(drv->nl_cb);
1408         if (drv->nl_handle_event == NULL) {
1409                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1410                            "callbacks (event)");
1411                 goto err2b;
1412         }
1413
1414         if (genl_connect(drv->nl_handle)) {
1415                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
1416                            "netlink");
1417                 goto err3;
1418         }
1419
1420         if (genl_connect(drv->nl_handle_event)) {
1421                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
1422                            "netlink (event)");
1423                 goto err3;
1424         }
1425
1426 #ifdef CONFIG_LIBNL20
1427         if (genl_ctrl_alloc_cache(drv->nl_handle, &drv->nl_cache) < 0) {
1428                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1429                            "netlink cache");
1430                 goto err3;
1431         }
1432         if (genl_ctrl_alloc_cache(drv->nl_handle_event, &drv->nl_cache_event) <
1433             0) {
1434                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1435                            "netlink cache (event)");
1436                 goto err3b;
1437         }
1438 #else /* CONFIG_LIBNL20 */
1439         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
1440         if (drv->nl_cache == NULL) {
1441                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1442                            "netlink cache");
1443                 goto err3;
1444         }
1445         drv->nl_cache_event = genl_ctrl_alloc_cache(drv->nl_handle_event);
1446         if (drv->nl_cache_event == NULL) {
1447                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1448                            "netlink cache (event)");
1449                 goto err3b;
1450         }
1451 #endif /* CONFIG_LIBNL20 */
1452
1453         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
1454         if (drv->nl80211 == NULL) {
1455                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
1456                            "found");
1457                 goto err4;
1458         }
1459
1460         ret = nl_get_multicast_id(drv, "nl80211", "scan");
1461         if (ret >= 0)
1462                 ret = nl_socket_add_membership(drv->nl_handle_event, ret);
1463         if (ret < 0) {
1464                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
1465                            "membership for scan events: %d (%s)",
1466                            ret, strerror(-ret));
1467                 goto err4;
1468         }
1469
1470         ret = nl_get_multicast_id(drv, "nl80211", "mlme");
1471         if (ret >= 0)
1472                 ret = nl_socket_add_membership(drv->nl_handle_event, ret);
1473         if (ret < 0) {
1474                 wpa_printf(MSG_ERROR, "nl80211: Could not add multicast "
1475                            "membership for mlme events: %d (%s)",
1476                            ret, strerror(-ret));
1477                 goto err4;
1478         }
1479
1480         eloop_register_read_sock(nl_socket_get_fd(drv->nl_handle_event),
1481                                  wpa_driver_nl80211_event_receive, drv,
1482                                  drv->nl_handle_event);
1483
1484         return 0;
1485
1486 err4:
1487         nl_cache_free(drv->nl_cache_event);
1488 err3b:
1489         nl_cache_free(drv->nl_cache);
1490 err3:
1491         nl80211_handle_destroy(drv->nl_handle_event);
1492 err2b:
1493         nl80211_handle_destroy(drv->nl_handle);
1494 err2:
1495         nl_cb_put(drv->nl_cb);
1496 err1:
1497         return -1;
1498 }
1499
1500
1501 static void wpa_driver_nl80211_rfkill_blocked(void *ctx)
1502 {
1503         wpa_printf(MSG_DEBUG, "nl80211: RFKILL blocked");
1504         /*
1505          * This may be for any interface; use ifdown event to disable
1506          * interface.
1507          */
1508 }
1509
1510
1511 static void wpa_driver_nl80211_rfkill_unblocked(void *ctx)
1512 {
1513         struct wpa_driver_nl80211_data *drv = ctx;
1514         wpa_printf(MSG_DEBUG, "nl80211: RFKILL unblocked");
1515         if (linux_set_iface_flags(drv->ioctl_sock, drv->first_bss.ifname, 1)) {
1516                 wpa_printf(MSG_DEBUG, "nl80211: Could not set interface UP "
1517                            "after rfkill unblock");
1518                 return;
1519         }
1520         /* rtnetlink ifup handler will report interface as enabled */
1521 }
1522
1523
1524 /**
1525  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
1526  * @ctx: context to be used when calling wpa_supplicant functions,
1527  * e.g., wpa_supplicant_event()
1528  * @ifname: interface name, e.g., wlan0
1529  * Returns: Pointer to private data, %NULL on failure
1530  */
1531 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname)
1532 {
1533         struct wpa_driver_nl80211_data *drv;
1534         struct netlink_config *cfg;
1535         struct rfkill_config *rcfg;
1536         struct i802_bss *bss;
1537
1538         drv = os_zalloc(sizeof(*drv));
1539         if (drv == NULL)
1540                 return NULL;
1541         drv->ctx = ctx;
1542         bss = &drv->first_bss;
1543         bss->drv = drv;
1544         os_strlcpy(bss->ifname, ifname, sizeof(bss->ifname));
1545         drv->monitor_ifidx = -1;
1546         drv->monitor_sock = -1;
1547         drv->ioctl_sock = -1;
1548
1549         if (wpa_driver_nl80211_init_nl(drv)) {
1550                 os_free(drv);
1551                 return NULL;
1552         }
1553
1554         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1555         if (drv->ioctl_sock < 0) {
1556                 perror("socket(PF_INET,SOCK_DGRAM)");
1557                 goto failed;
1558         }
1559
1560         cfg = os_zalloc(sizeof(*cfg));
1561         if (cfg == NULL)
1562                 goto failed;
1563         cfg->ctx = drv;
1564         cfg->newlink_cb = wpa_driver_nl80211_event_rtm_newlink;
1565         cfg->dellink_cb = wpa_driver_nl80211_event_rtm_dellink;
1566         drv->netlink = netlink_init(cfg);
1567         if (drv->netlink == NULL) {
1568                 os_free(cfg);
1569                 goto failed;
1570         }
1571
1572         rcfg = os_zalloc(sizeof(*rcfg));
1573         if (rcfg == NULL)
1574                 goto failed;
1575         rcfg->ctx = drv;
1576         os_strlcpy(rcfg->ifname, ifname, sizeof(rcfg->ifname));
1577         rcfg->blocked_cb = wpa_driver_nl80211_rfkill_blocked;
1578         rcfg->unblocked_cb = wpa_driver_nl80211_rfkill_unblocked;
1579         drv->rfkill = rfkill_init(rcfg);
1580         if (drv->rfkill == NULL) {
1581                 wpa_printf(MSG_DEBUG, "nl80211: RFKILL status not available");
1582                 os_free(rcfg);
1583         }
1584
1585         if (wpa_driver_nl80211_finish_drv_init(drv))
1586                 goto failed;
1587
1588         return bss;
1589
1590 failed:
1591         rfkill_deinit(drv->rfkill);
1592         netlink_deinit(drv->netlink);
1593         if (drv->ioctl_sock >= 0)
1594                 close(drv->ioctl_sock);
1595
1596         genl_family_put(drv->nl80211);
1597         nl_cache_free(drv->nl_cache);
1598         nl80211_handle_destroy(drv->nl_handle);
1599         nl_cb_put(drv->nl_cb);
1600         eloop_unregister_read_sock(nl_socket_get_fd(drv->nl_handle_event));
1601
1602         os_free(drv);
1603         return NULL;
1604 }
1605
1606
1607 static int nl80211_register_frame(struct wpa_driver_nl80211_data *drv,
1608                                   struct nl_handle *nl_handle,
1609                                   u16 type, const u8 *match, size_t match_len)
1610 {
1611         struct nl_msg *msg;
1612         int ret = -1;
1613
1614         msg = nlmsg_alloc();
1615         if (!msg)
1616                 return -1;
1617
1618         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1619                     NL80211_CMD_REGISTER_ACTION, 0);
1620
1621         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1622         NLA_PUT_U16(msg, NL80211_ATTR_FRAME_TYPE, type);
1623         NLA_PUT(msg, NL80211_ATTR_FRAME_MATCH, match_len, match);
1624
1625         ret = send_and_recv(drv, nl_handle, msg, NULL, NULL);
1626         msg = NULL;
1627         if (ret) {
1628                 wpa_printf(MSG_DEBUG, "nl80211: Register frame command "
1629                            "failed (type=%u): ret=%d (%s)",
1630                            type, ret, strerror(-ret));
1631                 wpa_hexdump(MSG_DEBUG, "nl80211: Register frame match",
1632                             match, match_len);
1633                 goto nla_put_failure;
1634         }
1635         ret = 0;
1636 nla_put_failure:
1637         nlmsg_free(msg);
1638         return ret;
1639 }
1640
1641
1642 static int nl80211_register_action_frame(struct wpa_driver_nl80211_data *drv,
1643                                          const u8 *match, size_t match_len)
1644 {
1645         u16 type = (WLAN_FC_TYPE_MGMT << 2) | (WLAN_FC_STYPE_ACTION << 4);
1646         return nl80211_register_frame(drv, drv->nl_handle_event,
1647                                       type, match, match_len);
1648 }
1649
1650
1651 static int nl80211_register_action_frames(struct wpa_driver_nl80211_data *drv)
1652 {
1653 #ifdef CONFIG_P2P
1654         /* GAS Initial Request */
1655         if (nl80211_register_action_frame(drv, (u8 *) "\x04\x0a", 2) < 0)
1656                 return -1;
1657         /* GAS Initial Response */
1658         if (nl80211_register_action_frame(drv, (u8 *) "\x04\x0b", 2) < 0)
1659                 return -1;
1660         /* GAS Comeback Request */
1661         if (nl80211_register_action_frame(drv, (u8 *) "\x04\x0c", 2) < 0)
1662                 return -1;
1663         /* GAS Comeback Response */
1664         if (nl80211_register_action_frame(drv, (u8 *) "\x04\x0d", 2) < 0)
1665                 return -1;
1666         /* P2P Public Action */
1667         if (nl80211_register_action_frame(drv,
1668                                           (u8 *) "\x04\x09\x50\x6f\x9a\x09",
1669                                           6) < 0)
1670                 return -1;
1671         /* P2P Action */
1672         if (nl80211_register_action_frame(drv,
1673                                           (u8 *) "\x7f\x50\x6f\x9a\x09",
1674                                           5) < 0)
1675                 return -1;
1676 #endif /* CONFIG_P2P */
1677
1678         /* FT Action frames */
1679         if (nl80211_register_action_frame(drv, (u8 *) "\x06", 1) < 0)
1680                 return -1;
1681         else
1682                 drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_FT |
1683                         WPA_DRIVER_CAPA_KEY_MGMT_FT_PSK;
1684
1685         return 0;
1686 }
1687
1688
1689 static void wpa_driver_nl80211_send_rfkill(void *eloop_ctx, void *timeout_ctx)
1690 {
1691         wpa_supplicant_event(timeout_ctx, EVENT_INTERFACE_DISABLED, NULL);
1692 }
1693
1694
1695 static int
1696 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
1697 {
1698         struct i802_bss *bss = &drv->first_bss;
1699         int send_rfkill_event = 0;
1700
1701         drv->ifindex = if_nametoindex(bss->ifname);
1702         drv->first_bss.ifindex = drv->ifindex;
1703
1704 #ifndef HOSTAPD
1705         if (wpa_driver_nl80211_set_mode(bss, IEEE80211_MODE_INFRA) < 0) {
1706                 wpa_printf(MSG_DEBUG, "nl80211: Could not configure driver to "
1707                            "use managed mode");
1708         }
1709
1710         if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 1)) {
1711                 if (rfkill_is_blocked(drv->rfkill)) {
1712                         wpa_printf(MSG_DEBUG, "nl80211: Could not yet enable "
1713                                    "interface '%s' due to rfkill",
1714                                    bss->ifname);
1715                         drv->if_disabled = 1;
1716                         send_rfkill_event = 1;
1717                 } else {
1718                         wpa_printf(MSG_ERROR, "nl80211: Could not set "
1719                                    "interface '%s' UP", bss->ifname);
1720                         return -1;
1721                 }
1722         }
1723
1724         if (wpa_driver_nl80211_capa(drv))
1725                 return -1;
1726
1727         netlink_send_oper_ifla(drv->netlink, drv->ifindex,
1728                                1, IF_OPER_DORMANT);
1729 #endif /* HOSTAPD */
1730
1731         if (nl80211_register_action_frames(drv) < 0) {
1732                 wpa_printf(MSG_DEBUG, "nl80211: Failed to register Action "
1733                            "frame processing - ignore for now");
1734                 /*
1735                  * Older kernel versions did not support this, so ignore the
1736                  * error for now. Some functionality may not be available
1737                  * because of this.
1738                  */
1739         }
1740
1741         if (send_rfkill_event) {
1742                 eloop_register_timeout(0, 0, wpa_driver_nl80211_send_rfkill,
1743                                        drv, drv->ctx);
1744         }
1745
1746         return 0;
1747 }
1748
1749
1750 static int wpa_driver_nl80211_del_beacon(struct wpa_driver_nl80211_data *drv)
1751 {
1752         struct nl_msg *msg;
1753
1754         msg = nlmsg_alloc();
1755         if (!msg)
1756                 return -ENOMEM;
1757
1758         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1759                     0, NL80211_CMD_DEL_BEACON, 0);
1760         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1761
1762         return send_and_recv_msgs(drv, msg, NULL, NULL);
1763  nla_put_failure:
1764         return -ENOBUFS;
1765 }
1766
1767
1768 /**
1769  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
1770  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1771  *
1772  * Shut down driver interface and processing of driver events. Free
1773  * private data buffer if one was allocated in wpa_driver_nl80211_init().
1774  */
1775 static void wpa_driver_nl80211_deinit(void *priv)
1776 {
1777         struct i802_bss *bss = priv;
1778         struct wpa_driver_nl80211_data *drv = bss->drv;
1779
1780         if (drv->nl_handle_preq)
1781                 wpa_driver_nl80211_probe_req_report(bss, 0);
1782         if (drv->added_if_into_bridge) {
1783                 if (linux_br_del_if(drv->ioctl_sock, drv->brname, bss->ifname)
1784                     < 0)
1785                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
1786                                    "interface %s from bridge %s: %s",
1787                                    bss->ifname, drv->brname, strerror(errno));
1788         }
1789         if (drv->added_bridge) {
1790                 if (linux_br_del(drv->ioctl_sock, drv->brname) < 0)
1791                         wpa_printf(MSG_INFO, "nl80211: Failed to remove "
1792                                    "bridge %s: %s",
1793                                    drv->brname, strerror(errno));
1794         }
1795
1796         nl80211_remove_monitor_interface(drv);
1797
1798         if (drv->nlmode == NL80211_IFTYPE_AP)
1799                 wpa_driver_nl80211_del_beacon(drv);
1800
1801 #ifdef HOSTAPD
1802         if (drv->last_freq_ht) {
1803                 /* Clear HT flags from the driver */
1804                 struct hostapd_freq_params freq;
1805                 os_memset(&freq, 0, sizeof(freq));
1806                 freq.freq = drv->last_freq;
1807                 i802_set_freq(priv, &freq);
1808         }
1809
1810         if (drv->eapol_sock >= 0) {
1811                 eloop_unregister_read_sock(drv->eapol_sock);
1812                 close(drv->eapol_sock);
1813         }
1814
1815         if (drv->if_indices != drv->default_if_indices)
1816                 os_free(drv->if_indices);
1817 #endif /* HOSTAPD */
1818
1819         if (drv->disable_11b_rates)
1820                 nl80211_disable_11b_rates(drv, drv->ifindex, 0);
1821
1822         netlink_send_oper_ifla(drv->netlink, drv->ifindex, 0, IF_OPER_UP);
1823         netlink_deinit(drv->netlink);
1824         rfkill_deinit(drv->rfkill);
1825
1826         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1827
1828         (void) linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 0);
1829         wpa_driver_nl80211_set_mode(bss, IEEE80211_MODE_INFRA);
1830
1831         if (drv->ioctl_sock >= 0)
1832                 close(drv->ioctl_sock);
1833
1834         eloop_unregister_read_sock(nl_socket_get_fd(drv->nl_handle_event));
1835         genl_family_put(drv->nl80211);
1836         nl_cache_free(drv->nl_cache);
1837         nl_cache_free(drv->nl_cache_event);
1838         nl80211_handle_destroy(drv->nl_handle);
1839         nl80211_handle_destroy(drv->nl_handle_event);
1840         nl_cb_put(drv->nl_cb);
1841
1842         os_free(drv->filter_ssids);
1843
1844         os_free(drv);
1845 }
1846
1847
1848 /**
1849  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
1850  * @eloop_ctx: Driver private data
1851  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
1852  *
1853  * This function can be used as registered timeout when starting a scan to
1854  * generate a scan completed event if the driver does not report this.
1855  */
1856 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1857 {
1858         struct wpa_driver_nl80211_data *drv = eloop_ctx;
1859         if (drv->ap_scan_as_station) {
1860                 wpa_driver_nl80211_set_mode(&drv->first_bss,
1861                                             IEEE80211_MODE_AP);
1862                 drv->ap_scan_as_station = 0;
1863         }
1864         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
1865         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
1866 }
1867
1868
1869 /**
1870  * wpa_driver_nl80211_scan - Request the driver to initiate scan
1871  * @priv: Pointer to private driver data from wpa_driver_nl80211_init()
1872  * @params: Scan parameters
1873  * Returns: 0 on success, -1 on failure
1874  */
1875 static int wpa_driver_nl80211_scan(void *priv,
1876                                    struct wpa_driver_scan_params *params)
1877 {
1878         struct i802_bss *bss = priv;
1879         struct wpa_driver_nl80211_data *drv = bss->drv;
1880         int ret = 0, timeout;
1881         struct nl_msg *msg, *ssids, *freqs;
1882         size_t i;
1883
1884         msg = nlmsg_alloc();
1885         ssids = nlmsg_alloc();
1886         freqs = nlmsg_alloc();
1887         if (!msg || !ssids || !freqs) {
1888                 nlmsg_free(msg);
1889                 nlmsg_free(ssids);
1890                 nlmsg_free(freqs);
1891                 return -1;
1892         }
1893
1894         os_free(drv->filter_ssids);
1895         drv->filter_ssids = params->filter_ssids;
1896         params->filter_ssids = NULL;
1897         drv->num_filter_ssids = params->num_filter_ssids;
1898
1899         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1900                     NL80211_CMD_TRIGGER_SCAN, 0);
1901
1902         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1903
1904         for (i = 0; i < params->num_ssids; i++) {
1905                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Scan SSID",
1906                                   params->ssids[i].ssid,
1907                                   params->ssids[i].ssid_len);
1908                 NLA_PUT(ssids, i + 1, params->ssids[i].ssid_len,
1909                         params->ssids[i].ssid);
1910         }
1911         if (params->num_ssids)
1912                 nla_put_nested(msg, NL80211_ATTR_SCAN_SSIDS, ssids);
1913
1914         if (params->extra_ies) {
1915                 wpa_hexdump_ascii(MSG_MSGDUMP, "nl80211: Scan extra IEs",
1916                                   params->extra_ies, params->extra_ies_len);
1917                 NLA_PUT(msg, NL80211_ATTR_IE, params->extra_ies_len,
1918                         params->extra_ies);
1919         }
1920
1921         if (params->freqs) {
1922                 for (i = 0; params->freqs[i]; i++) {
1923                         wpa_printf(MSG_MSGDUMP, "nl80211: Scan frequency %u "
1924                                    "MHz", params->freqs[i]);
1925                         NLA_PUT_U32(freqs, i + 1, params->freqs[i]);
1926                 }
1927                 nla_put_nested(msg, NL80211_ATTR_SCAN_FREQUENCIES, freqs);
1928         }
1929
1930         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1931         msg = NULL;
1932         if (ret) {
1933                 wpa_printf(MSG_DEBUG, "nl80211: Scan trigger failed: ret=%d "
1934                            "(%s)", ret, strerror(-ret));
1935 #ifdef HOSTAPD
1936                 if (drv->nlmode == NL80211_IFTYPE_AP) {
1937                         /*
1938                          * mac80211 does not allow scan requests in AP mode, so
1939                          * try to do this in station mode.
1940                          */
1941                         if (wpa_driver_nl80211_set_mode(bss,
1942                                                         IEEE80211_MODE_INFRA))
1943                                 goto nla_put_failure;
1944
1945                         if (wpa_driver_nl80211_scan(drv, params)) {
1946                                 wpa_driver_nl80211_set_mode(bss,
1947                                                             IEEE80211_MODE_AP);
1948                                 goto nla_put_failure;
1949                         }
1950
1951                         /* Restore AP mode when processing scan results */
1952                         drv->ap_scan_as_station = 1;
1953                         ret = 0;
1954                 } else
1955                         goto nla_put_failure;
1956 #else /* HOSTAPD */
1957                 goto nla_put_failure;
1958 #endif /* HOSTAPD */
1959         }
1960
1961         /* Not all drivers generate "scan completed" wireless event, so try to
1962          * read results after a timeout. */
1963         timeout = 10;
1964         if (drv->scan_complete_events) {
1965                 /*
1966                  * The driver seems to deliver events to notify when scan is
1967                  * complete, so use longer timeout to avoid race conditions
1968                  * with scanning and following association request.
1969                  */
1970                 timeout = 30;
1971         }
1972         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
1973                    "seconds", ret, timeout);
1974         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1975         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout,
1976                                drv, drv->ctx);
1977
1978 nla_put_failure:
1979         nlmsg_free(ssids);
1980         nlmsg_free(msg);
1981         nlmsg_free(freqs);
1982         return ret;
1983 }
1984
1985
1986 static const u8 * nl80211_get_ie(const u8 *ies, size_t ies_len, u8 ie)
1987 {
1988         const u8 *end, *pos;
1989
1990         if (ies == NULL)
1991                 return NULL;
1992
1993         pos = ies;
1994         end = ies + ies_len;
1995
1996         while (pos + 1 < end) {
1997                 if (pos + 2 + pos[1] > end)
1998                         break;
1999                 if (pos[0] == ie)
2000                         return pos;
2001                 pos += 2 + pos[1];
2002         }
2003
2004         return NULL;
2005 }
2006
2007
2008 static int nl80211_scan_filtered(struct wpa_driver_nl80211_data *drv,
2009                                  const u8 *ie, size_t ie_len)
2010 {
2011         const u8 *ssid;
2012         size_t i;
2013
2014         if (drv->filter_ssids == NULL)
2015                 return 0;
2016
2017         ssid = nl80211_get_ie(ie, ie_len, WLAN_EID_SSID);
2018         if (ssid == NULL)
2019                 return 1;
2020
2021         for (i = 0; i < drv->num_filter_ssids; i++) {
2022                 if (ssid[1] == drv->filter_ssids[i].ssid_len &&
2023                     os_memcmp(ssid + 2, drv->filter_ssids[i].ssid, ssid[1]) ==
2024                     0)
2025                         return 0;
2026         }
2027
2028         return 1;
2029 }
2030
2031
2032 struct nl80211_bss_info_arg {
2033         struct wpa_driver_nl80211_data *drv;
2034         struct wpa_scan_results *res;
2035 };
2036
2037 static int bss_info_handler(struct nl_msg *msg, void *arg)
2038 {
2039         struct nlattr *tb[NL80211_ATTR_MAX + 1];
2040         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2041         struct nlattr *bss[NL80211_BSS_MAX + 1];
2042         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
2043                 [NL80211_BSS_BSSID] = { .type = NLA_UNSPEC },
2044                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
2045                 [NL80211_BSS_TSF] = { .type = NLA_U64 },
2046                 [NL80211_BSS_BEACON_INTERVAL] = { .type = NLA_U16 },
2047                 [NL80211_BSS_CAPABILITY] = { .type = NLA_U16 },
2048                 [NL80211_BSS_INFORMATION_ELEMENTS] = { .type = NLA_UNSPEC },
2049                 [NL80211_BSS_SIGNAL_MBM] = { .type = NLA_U32 },
2050                 [NL80211_BSS_SIGNAL_UNSPEC] = { .type = NLA_U8 },
2051                 [NL80211_BSS_STATUS] = { .type = NLA_U32 },
2052                 [NL80211_BSS_SEEN_MS_AGO] = { .type = NLA_U32 },
2053                 [NL80211_BSS_BEACON_IES] = { .type = NLA_UNSPEC },
2054         };
2055         struct nl80211_bss_info_arg *_arg = arg;
2056         struct wpa_scan_results *res = _arg->res;
2057         struct wpa_scan_res **tmp;
2058         struct wpa_scan_res *r;
2059         const u8 *ie, *beacon_ie;
2060         size_t ie_len, beacon_ie_len;
2061         u8 *pos;
2062
2063         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2064                   genlmsg_attrlen(gnlh, 0), NULL);
2065         if (!tb[NL80211_ATTR_BSS])
2066                 return NL_SKIP;
2067         if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
2068                              bss_policy))
2069                 return NL_SKIP;
2070         if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
2071                 ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
2072                 ie_len = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
2073         } else {
2074                 ie = NULL;
2075                 ie_len = 0;
2076         }
2077         if (bss[NL80211_BSS_BEACON_IES]) {
2078                 beacon_ie = nla_data(bss[NL80211_BSS_BEACON_IES]);
2079                 beacon_ie_len = nla_len(bss[NL80211_BSS_BEACON_IES]);
2080         } else {
2081                 beacon_ie = NULL;
2082                 beacon_ie_len = 0;
2083         }
2084
2085         if (nl80211_scan_filtered(_arg->drv, ie ? ie : beacon_ie,
2086                                   ie ? ie_len : beacon_ie_len))
2087                 return NL_SKIP;
2088
2089         r = os_zalloc(sizeof(*r) + ie_len + beacon_ie_len);
2090         if (r == NULL)
2091                 return NL_SKIP;
2092         if (bss[NL80211_BSS_BSSID])
2093                 os_memcpy(r->bssid, nla_data(bss[NL80211_BSS_BSSID]),
2094                           ETH_ALEN);
2095         if (bss[NL80211_BSS_FREQUENCY])
2096                 r->freq = nla_get_u32(bss[NL80211_BSS_FREQUENCY]);
2097         if (bss[NL80211_BSS_BEACON_INTERVAL])
2098                 r->beacon_int = nla_get_u16(bss[NL80211_BSS_BEACON_INTERVAL]);
2099         if (bss[NL80211_BSS_CAPABILITY])
2100                 r->caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
2101         r->flags |= WPA_SCAN_NOISE_INVALID;
2102         if (bss[NL80211_BSS_SIGNAL_MBM]) {
2103                 r->level = nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]);
2104                 r->level /= 100; /* mBm to dBm */
2105                 r->flags |= WPA_SCAN_LEVEL_DBM | WPA_SCAN_QUAL_INVALID;
2106         } else if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
2107                 r->level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
2108                 r->flags |= WPA_SCAN_LEVEL_INVALID;
2109         } else
2110                 r->flags |= WPA_SCAN_LEVEL_INVALID | WPA_SCAN_QUAL_INVALID;
2111         if (bss[NL80211_BSS_TSF])
2112                 r->tsf = nla_get_u64(bss[NL80211_BSS_TSF]);
2113         if (bss[NL80211_BSS_SEEN_MS_AGO])
2114                 r->age = nla_get_u32(bss[NL80211_BSS_SEEN_MS_AGO]);
2115         r->ie_len = ie_len;
2116         pos = (u8 *) (r + 1);
2117         if (ie) {
2118                 os_memcpy(pos, ie, ie_len);
2119                 pos += ie_len;
2120         }
2121         r->beacon_ie_len = beacon_ie_len;
2122         if (beacon_ie)
2123                 os_memcpy(pos, beacon_ie, beacon_ie_len);
2124
2125         if (bss[NL80211_BSS_STATUS]) {
2126                 enum nl80211_bss_status status;
2127                 status = nla_get_u32(bss[NL80211_BSS_STATUS]);
2128                 switch (status) {
2129                 case NL80211_BSS_STATUS_AUTHENTICATED:
2130                         r->flags |= WPA_SCAN_AUTHENTICATED;
2131                         break;
2132                 case NL80211_BSS_STATUS_ASSOCIATED:
2133                         r->flags |= WPA_SCAN_ASSOCIATED;
2134                         break;
2135                 default:
2136                         break;
2137                 }
2138         }
2139
2140         tmp = os_realloc(res->res,
2141                          (res->num + 1) * sizeof(struct wpa_scan_res *));
2142         if (tmp == NULL) {
2143                 os_free(r);
2144                 return NL_SKIP;
2145         }
2146         tmp[res->num++] = r;
2147         res->res = tmp;
2148
2149         return NL_SKIP;
2150 }
2151
2152
2153 static void clear_state_mismatch(struct wpa_driver_nl80211_data *drv,
2154                                  const u8 *addr)
2155 {
2156         if (drv->capa.flags & WPA_DRIVER_FLAGS_SME) {
2157                 wpa_printf(MSG_DEBUG, "nl80211: Clear possible state "
2158                            "mismatch (" MACSTR ")", MAC2STR(addr));
2159                 wpa_driver_nl80211_mlme(drv, addr,
2160                                         NL80211_CMD_DEAUTHENTICATE,
2161                                         WLAN_REASON_PREV_AUTH_NOT_VALID, 1);
2162         }
2163 }
2164
2165
2166 static void wpa_driver_nl80211_check_bss_status(
2167         struct wpa_driver_nl80211_data *drv, struct wpa_scan_results *res)
2168 {
2169         size_t i;
2170
2171         for (i = 0; i < res->num; i++) {
2172                 struct wpa_scan_res *r = res->res[i];
2173                 if (r->flags & WPA_SCAN_AUTHENTICATED) {
2174                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
2175                                    "indicates BSS status with " MACSTR
2176                                    " as authenticated",
2177                                    MAC2STR(r->bssid));
2178                         if (drv->nlmode == NL80211_IFTYPE_STATION &&
2179                             os_memcmp(r->bssid, drv->bssid, ETH_ALEN) != 0 &&
2180                             os_memcmp(r->bssid, drv->auth_bssid, ETH_ALEN) !=
2181                             0) {
2182                                 wpa_printf(MSG_DEBUG, "nl80211: Unknown BSSID"
2183                                            " in local state (auth=" MACSTR
2184                                            " assoc=" MACSTR ")",
2185                                            MAC2STR(drv->auth_bssid),
2186                                            MAC2STR(drv->bssid));
2187                                 clear_state_mismatch(drv, r->bssid);
2188                         }
2189                 }
2190
2191                 if (r->flags & WPA_SCAN_ASSOCIATED) {
2192                         wpa_printf(MSG_DEBUG, "nl80211: Scan results "
2193                                    "indicate BSS status with " MACSTR
2194                                    " as associated",
2195                                    MAC2STR(r->bssid));
2196                         if (drv->nlmode == NL80211_IFTYPE_STATION &&
2197                             !drv->associated) {
2198                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
2199                                            "(not associated) does not match "
2200                                            "with BSS state");
2201                                 clear_state_mismatch(drv, r->bssid);
2202                         } else if (drv->nlmode == NL80211_IFTYPE_STATION &&
2203                                    os_memcmp(drv->bssid, r->bssid, ETH_ALEN) !=
2204                                    0) {
2205                                 wpa_printf(MSG_DEBUG, "nl80211: Local state "
2206                                            "(associated with " MACSTR ") does "
2207                                            "not match with BSS state",
2208                                            MAC2STR(drv->bssid));
2209                                 clear_state_mismatch(drv, r->bssid);
2210                                 clear_state_mismatch(drv, drv->bssid);
2211                         }
2212                 }
2213         }
2214 }
2215
2216
2217 static void wpa_scan_results_free(struct wpa_scan_results *res)
2218 {
2219         size_t i;
2220
2221         if (res == NULL)
2222                 return;
2223
2224         for (i = 0; i < res->num; i++)
2225                 os_free(res->res[i]);
2226         os_free(res->res);
2227         os_free(res);
2228 }
2229
2230
2231 static struct wpa_scan_results *
2232 nl80211_get_scan_results(struct wpa_driver_nl80211_data *drv)
2233 {
2234         struct nl_msg *msg;
2235         struct wpa_scan_results *res;
2236         int ret;
2237         struct nl80211_bss_info_arg arg;
2238
2239         res = os_zalloc(sizeof(*res));
2240         if (res == NULL)
2241                 return NULL;
2242         msg = nlmsg_alloc();
2243         if (!msg)
2244                 goto nla_put_failure;
2245
2246         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, NLM_F_DUMP,
2247                     NL80211_CMD_GET_SCAN, 0);
2248         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2249
2250         arg.drv = drv;
2251         arg.res = res;
2252         ret = send_and_recv_msgs(drv, msg, bss_info_handler, &arg);
2253         msg = NULL;
2254         if (ret == 0) {
2255                 wpa_printf(MSG_DEBUG, "Received scan results (%lu BSSes)",
2256                            (unsigned long) res->num);
2257                 return res;
2258         }
2259         wpa_printf(MSG_DEBUG, "nl80211: Scan result fetch failed: ret=%d "
2260                    "(%s)", ret, strerror(-ret));
2261 nla_put_failure:
2262         nlmsg_free(msg);
2263         wpa_scan_results_free(res);
2264         return NULL;
2265 }
2266
2267
2268 /**
2269  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
2270  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
2271  * Returns: Scan results on success, -1 on failure
2272  */
2273 static struct wpa_scan_results *
2274 wpa_driver_nl80211_get_scan_results(void *priv)
2275 {
2276         struct i802_bss *bss = priv;
2277         struct wpa_driver_nl80211_data *drv = bss->drv;
2278         struct wpa_scan_results *res;
2279
2280         res = nl80211_get_scan_results(drv);
2281         if (res)
2282                 wpa_driver_nl80211_check_bss_status(drv, res);
2283         return res;
2284 }
2285
2286
2287 static void nl80211_dump_scan(struct wpa_driver_nl80211_data *drv)
2288 {
2289         struct wpa_scan_results *res;
2290         size_t i;
2291
2292         res = nl80211_get_scan_results(drv);
2293         if (res == NULL) {
2294                 wpa_printf(MSG_DEBUG, "nl80211: Failed to get scan results");
2295                 return;
2296         }
2297
2298         wpa_printf(MSG_DEBUG, "nl80211: Scan result dump");
2299         for (i = 0; i < res->num; i++) {
2300                 struct wpa_scan_res *r = res->res[i];
2301                 wpa_printf(MSG_DEBUG, "nl80211: %d/%d " MACSTR "%s%s",
2302                            (int) i, (int) res->num, MAC2STR(r->bssid),
2303                            r->flags & WPA_SCAN_AUTHENTICATED ? " [auth]" : "",
2304                            r->flags & WPA_SCAN_ASSOCIATED ? " [assoc]" : "");
2305         }
2306
2307         wpa_scan_results_free(res);
2308 }
2309
2310
2311 static int wpa_driver_nl80211_set_key(const char *ifname, void *priv,
2312                                       enum wpa_alg alg, const u8 *addr,
2313                                       int key_idx, int set_tx,
2314                                       const u8 *seq, size_t seq_len,
2315                                       const u8 *key, size_t key_len)
2316 {
2317         struct i802_bss *bss = priv;
2318         struct wpa_driver_nl80211_data *drv = bss->drv;
2319         int ifindex = if_nametoindex(ifname);
2320         struct nl_msg *msg;
2321         int ret;
2322
2323         wpa_printf(MSG_DEBUG, "%s: ifindex=%d alg=%d addr=%p key_idx=%d "
2324                    "set_tx=%d seq_len=%lu key_len=%lu",
2325                    __func__, ifindex, alg, addr, key_idx, set_tx,
2326                    (unsigned long) seq_len, (unsigned long) key_len);
2327
2328         msg = nlmsg_alloc();
2329         if (!msg)
2330                 return -ENOMEM;
2331
2332         if (alg == WPA_ALG_NONE) {
2333                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2334                             0, NL80211_CMD_DEL_KEY, 0);
2335         } else {
2336                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2337                             0, NL80211_CMD_NEW_KEY, 0);
2338                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
2339                 switch (alg) {
2340                 case WPA_ALG_WEP:
2341                         if (key_len == 5)
2342                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2343                                             WLAN_CIPHER_SUITE_WEP40);
2344                         else
2345                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2346                                             WLAN_CIPHER_SUITE_WEP104);
2347                         break;
2348                 case WPA_ALG_TKIP:
2349                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2350                                     WLAN_CIPHER_SUITE_TKIP);
2351                         break;
2352                 case WPA_ALG_CCMP:
2353                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2354                                     WLAN_CIPHER_SUITE_CCMP);
2355                         break;
2356                 case WPA_ALG_IGTK:
2357                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2358                                     WLAN_CIPHER_SUITE_AES_CMAC);
2359                         break;
2360                 default:
2361                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
2362                                    "algorithm %d", __func__, alg);
2363                         nlmsg_free(msg);
2364                         return -1;
2365                 }
2366         }
2367
2368         if (seq && seq_len)
2369                 NLA_PUT(msg, NL80211_ATTR_KEY_SEQ, seq_len, seq);
2370
2371         if (addr && os_memcmp(addr, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
2372         {
2373                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
2374                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2375         }
2376         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2377         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
2378
2379         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2380         if ((ret == -ENOENT || ret == -ENOLINK) && alg == WPA_ALG_NONE)
2381                 ret = 0;
2382         if (ret)
2383                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d %s)",
2384                            ret, strerror(-ret));
2385
2386         /*
2387          * If we failed or don't need to set the default TX key (below),
2388          * we're done here.
2389          */
2390         if (ret || !set_tx || alg == WPA_ALG_NONE)
2391                 return ret;
2392 #ifdef HOSTAPD
2393         if (addr)
2394                 return ret;
2395 #else /* HOSTAPD */
2396         if (drv->nlmode == NL80211_IFTYPE_AP && addr)
2397                 return ret;
2398 #endif /* HOSTAPD */
2399
2400         msg = nlmsg_alloc();
2401         if (!msg)
2402                 return -ENOMEM;
2403
2404         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2405                     0, NL80211_CMD_SET_KEY, 0);
2406         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2407         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
2408         if (alg == WPA_ALG_IGTK)
2409                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
2410         else
2411                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
2412
2413         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2414         if (ret == -ENOENT)
2415                 ret = 0;
2416         if (ret)
2417                 wpa_printf(MSG_DEBUG, "nl80211: set_key default failed; "
2418                            "err=%d %s)", ret, strerror(-ret));
2419         return ret;
2420
2421 nla_put_failure:
2422         return -ENOBUFS;
2423 }
2424
2425
2426 static int nl_add_key(struct nl_msg *msg, enum wpa_alg alg,
2427                       int key_idx, int defkey,
2428                       const u8 *seq, size_t seq_len,
2429                       const u8 *key, size_t key_len)
2430 {
2431         struct nlattr *key_attr = nla_nest_start(msg, NL80211_ATTR_KEY);
2432         if (!key_attr)
2433                 return -1;
2434
2435         if (defkey && alg == WPA_ALG_IGTK)
2436                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT_MGMT);
2437         else if (defkey)
2438                 NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
2439
2440         NLA_PUT_U8(msg, NL80211_KEY_IDX, key_idx);
2441
2442         switch (alg) {
2443         case WPA_ALG_WEP:
2444                 if (key_len == 5)
2445                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
2446                                     WLAN_CIPHER_SUITE_WEP40);
2447                 else
2448                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
2449                                     WLAN_CIPHER_SUITE_WEP104);
2450                 break;
2451         case WPA_ALG_TKIP:
2452                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_TKIP);
2453                 break;
2454         case WPA_ALG_CCMP:
2455                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER, WLAN_CIPHER_SUITE_CCMP);
2456                 break;
2457         case WPA_ALG_IGTK:
2458                 NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
2459                             WLAN_CIPHER_SUITE_AES_CMAC);
2460                 break;
2461         default:
2462                 wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
2463                            "algorithm %d", __func__, alg);
2464                 return -1;
2465         }
2466
2467         if (seq && seq_len)
2468                 NLA_PUT(msg, NL80211_KEY_SEQ, seq_len, seq);
2469
2470         NLA_PUT(msg, NL80211_KEY_DATA, key_len, key);
2471
2472         nla_nest_end(msg, key_attr);
2473
2474         return 0;
2475  nla_put_failure:
2476         return -1;
2477 }
2478
2479
2480 static int nl80211_set_conn_keys(struct wpa_driver_associate_params *params,
2481                                  struct nl_msg *msg)
2482 {
2483         int i, privacy = 0;
2484         struct nlattr *nl_keys, *nl_key;
2485
2486         for (i = 0; i < 4; i++) {
2487                 if (!params->wep_key[i])
2488                         continue;
2489                 privacy = 1;
2490                 break;
2491         }
2492         if (!privacy)
2493                 return 0;
2494
2495         NLA_PUT_FLAG(msg, NL80211_ATTR_PRIVACY);
2496
2497         nl_keys = nla_nest_start(msg, NL80211_ATTR_KEYS);
2498         if (!nl_keys)
2499                 goto nla_put_failure;
2500
2501         for (i = 0; i < 4; i++) {
2502                 if (!params->wep_key[i])
2503                         continue;
2504
2505                 nl_key = nla_nest_start(msg, i);
2506                 if (!nl_key)
2507                         goto nla_put_failure;
2508
2509                 NLA_PUT(msg, NL80211_KEY_DATA, params->wep_key_len[i],
2510                         params->wep_key[i]);
2511                 if (params->wep_key_len[i] == 5)
2512                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
2513                                     WLAN_CIPHER_SUITE_WEP40);
2514                 else
2515                         NLA_PUT_U32(msg, NL80211_KEY_CIPHER,
2516                                     WLAN_CIPHER_SUITE_WEP104);
2517
2518                 NLA_PUT_U8(msg, NL80211_KEY_IDX, i);
2519
2520                 if (i == params->wep_tx_keyidx)
2521                         NLA_PUT_FLAG(msg, NL80211_KEY_DEFAULT);
2522
2523                 nla_nest_end(msg, nl_key);
2524         }
2525         nla_nest_end(msg, nl_keys);
2526
2527         return 0;
2528
2529 nla_put_failure:
2530         return -ENOBUFS;
2531 }
2532
2533
2534 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
2535                                    const u8 *addr, int cmd, u16 reason_code,
2536                                    int local_state_change)
2537 {
2538         int ret = -1;
2539         struct nl_msg *msg;
2540
2541         msg = nlmsg_alloc();
2542         if (!msg)
2543                 return -1;
2544
2545         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0, cmd, 0);
2546
2547         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2548         NLA_PUT_U16(msg, NL80211_ATTR_REASON_CODE, reason_code);
2549         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2550         if (local_state_change)
2551                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
2552
2553         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2554         msg = NULL;
2555         if (ret) {
2556                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
2557                            "(%s)", ret, strerror(-ret));
2558                 goto nla_put_failure;
2559         }
2560         ret = 0;
2561
2562 nla_put_failure:
2563         nlmsg_free(msg);
2564         return ret;
2565 }
2566
2567
2568 static int wpa_driver_nl80211_disconnect(struct wpa_driver_nl80211_data *drv,
2569                                          const u8 *addr, int reason_code)
2570 {
2571         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
2572                    __func__, MAC2STR(addr), reason_code);
2573         drv->associated = 0;
2574         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISCONNECT,
2575                                        reason_code, 0);
2576 }
2577
2578
2579 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
2580                                              int reason_code)
2581 {
2582         struct i802_bss *bss = priv;
2583         struct wpa_driver_nl80211_data *drv = bss->drv;
2584         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
2585                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
2586         wpa_printf(MSG_DEBUG, "%s(addr=" MACSTR " reason_code=%d)",
2587                    __func__, MAC2STR(addr), reason_code);
2588         drv->associated = 0;
2589         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DEAUTHENTICATE,
2590                                        reason_code, 0);
2591 }
2592
2593
2594 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
2595                                            int reason_code)
2596 {
2597         struct i802_bss *bss = priv;
2598         struct wpa_driver_nl80211_data *drv = bss->drv;
2599         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME))
2600                 return wpa_driver_nl80211_disconnect(drv, addr, reason_code);
2601         wpa_printf(MSG_DEBUG, "%s", __func__);
2602         drv->associated = 0;
2603         return wpa_driver_nl80211_mlme(drv, addr, NL80211_CMD_DISASSOCIATE,
2604                                        reason_code, 0);
2605 }
2606
2607
2608 static int wpa_driver_nl80211_authenticate(
2609         void *priv, struct wpa_driver_auth_params *params)
2610 {
2611         struct i802_bss *bss = priv;
2612         struct wpa_driver_nl80211_data *drv = bss->drv;
2613         int ret = -1, i;
2614         struct nl_msg *msg;
2615         enum nl80211_auth_type type;
2616         int count = 0;
2617
2618         drv->associated = 0;
2619         os_memset(drv->auth_bssid, 0, ETH_ALEN);
2620         /* FIX: IBSS mode */
2621         if (drv->nlmode != NL80211_IFTYPE_STATION)
2622                 wpa_driver_nl80211_set_mode(priv, IEEE80211_MODE_INFRA);
2623
2624         if (wpa_driver_nl80211_set_mode(priv, IEEE80211_MODE_INFRA) < 0)
2625                 return -1;
2626
2627 retry:
2628         msg = nlmsg_alloc();
2629         if (!msg)
2630                 return -1;
2631
2632         wpa_printf(MSG_DEBUG, "nl80211: Authenticate (ifindex=%d)",
2633                    drv->ifindex);
2634
2635         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2636                     NL80211_CMD_AUTHENTICATE, 0);
2637
2638         for (i = 0; i < 4; i++) {
2639                 if (!params->wep_key[i])
2640                         continue;
2641                 wpa_driver_nl80211_set_key(bss->ifname, priv, WPA_ALG_WEP,
2642                                            NULL, i,
2643                                            i == params->wep_tx_keyidx, NULL, 0,
2644                                            params->wep_key[i],
2645                                            params->wep_key_len[i]);
2646                 if (params->wep_tx_keyidx != i)
2647                         continue;
2648                 if (nl_add_key(msg, WPA_ALG_WEP, i, 1, NULL, 0,
2649                                params->wep_key[i], params->wep_key_len[i])) {
2650                         nlmsg_free(msg);
2651                         return -1;
2652                 }
2653         }
2654
2655         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2656         if (params->bssid) {
2657                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
2658                            MAC2STR(params->bssid));
2659                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
2660         }
2661         if (params->freq) {
2662                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
2663                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
2664         }
2665         if (params->ssid) {
2666                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
2667                                   params->ssid, params->ssid_len);
2668                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
2669                         params->ssid);
2670         }
2671         wpa_hexdump(MSG_DEBUG, "  * IEs", params->ie, params->ie_len);
2672         if (params->ie)
2673                 NLA_PUT(msg, NL80211_ATTR_IE, params->ie_len, params->ie);
2674         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
2675                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
2676         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
2677                 type = NL80211_AUTHTYPE_SHARED_KEY;
2678         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
2679                 type = NL80211_AUTHTYPE_NETWORK_EAP;
2680         else if (params->auth_alg & WPA_AUTH_ALG_FT)
2681                 type = NL80211_AUTHTYPE_FT;
2682         else
2683                 goto nla_put_failure;
2684         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
2685         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
2686         if (params->local_state_change) {
2687                 wpa_printf(MSG_DEBUG, "  * Local state change only");
2688                 NLA_PUT_FLAG(msg, NL80211_ATTR_LOCAL_STATE_CHANGE);
2689         }
2690
2691         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2692         msg = NULL;
2693         if (ret) {
2694                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
2695                            "(%s)", ret, strerror(-ret));
2696                 count++;
2697                 if (ret == -EALREADY && count == 1 && params->bssid &&
2698                     !params->local_state_change) {
2699                         /*
2700                          * mac80211 does not currently accept new
2701                          * authentication if we are already authenticated. As a
2702                          * workaround, force deauthentication and try again.
2703                          */
2704                         wpa_printf(MSG_DEBUG, "nl80211: Retry authentication "
2705                                    "after forced deauthentication");
2706                         wpa_driver_nl80211_deauthenticate(
2707                                 bss, params->bssid,
2708                                 WLAN_REASON_PREV_AUTH_NOT_VALID);
2709                         nlmsg_free(msg);
2710                         goto retry;
2711                 }
2712                 goto nla_put_failure;
2713         }
2714         ret = 0;
2715         wpa_printf(MSG_DEBUG, "nl80211: Authentication request send "
2716                    "successfully");
2717
2718 nla_put_failure:
2719         nlmsg_free(msg);
2720         return ret;
2721 }
2722
2723
2724 struct phy_info_arg {
2725         u16 *num_modes;
2726         struct hostapd_hw_modes *modes;
2727 };
2728
2729 static int phy_info_handler(struct nl_msg *msg, void *arg)
2730 {
2731         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
2732         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2733         struct phy_info_arg *phy_info = arg;
2734
2735         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
2736
2737         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
2738         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
2739                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
2740                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
2741                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
2742                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
2743                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
2744                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
2745         };
2746
2747         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
2748         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
2749                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
2750                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
2751         };
2752
2753         struct nlattr *nl_band;
2754         struct nlattr *nl_freq;
2755         struct nlattr *nl_rate;
2756         int rem_band, rem_freq, rem_rate;
2757         struct hostapd_hw_modes *mode;
2758         int idx, mode_is_set;
2759
2760         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2761                   genlmsg_attrlen(gnlh, 0), NULL);
2762
2763         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
2764                 return NL_SKIP;
2765
2766         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
2767                 mode = os_realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
2768                 if (!mode)
2769                         return NL_SKIP;
2770                 phy_info->modes = mode;
2771
2772                 mode_is_set = 0;
2773
2774                 mode = &phy_info->modes[*(phy_info->num_modes)];
2775                 memset(mode, 0, sizeof(*mode));
2776                 *(phy_info->num_modes) += 1;
2777
2778                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
2779                           nla_len(nl_band), NULL);
2780
2781                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
2782                         mode->ht_capab = nla_get_u16(
2783                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
2784                 }
2785
2786                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) {
2787                         mode->a_mpdu_params |= nla_get_u8(
2788                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_FACTOR]) &
2789                                 0x03;
2790                 }
2791
2792                 if (tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) {
2793                         mode->a_mpdu_params |= nla_get_u8(
2794                                 tb_band[NL80211_BAND_ATTR_HT_AMPDU_DENSITY]) <<
2795                                 2;
2796                 }
2797
2798                 if (tb_band[NL80211_BAND_ATTR_HT_MCS_SET] &&
2799                     nla_len(tb_band[NL80211_BAND_ATTR_HT_MCS_SET])) {
2800                         u8 *mcs;
2801                         mcs = nla_data(tb_band[NL80211_BAND_ATTR_HT_MCS_SET]);
2802                         os_memcpy(mode->mcs_set, mcs, 16);
2803                 }
2804
2805                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
2806                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
2807                                   nla_len(nl_freq), freq_policy);
2808                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2809                                 continue;
2810                         mode->num_channels++;
2811                 }
2812
2813                 mode->channels = os_zalloc(mode->num_channels * sizeof(struct hostapd_channel_data));
2814                 if (!mode->channels)
2815                         return NL_SKIP;
2816
2817                 idx = 0;
2818
2819                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
2820                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
2821                                   nla_len(nl_freq), freq_policy);
2822                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2823                                 continue;
2824
2825                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
2826                         mode->channels[idx].flag = 0;
2827
2828                         if (!mode_is_set) {
2829                                 /* crude heuristic */
2830                                 if (mode->channels[idx].freq < 4000)
2831                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
2832                                 else
2833                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
2834                                 mode_is_set = 1;
2835                         }
2836
2837                         /* crude heuristic */
2838                         if (mode->channels[idx].freq < 4000)
2839                                 if (mode->channels[idx].freq == 2484)
2840                                         mode->channels[idx].chan = 14;
2841                                 else
2842                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
2843                         else
2844                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
2845
2846                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
2847                                 mode->channels[idx].flag |=
2848                                         HOSTAPD_CHAN_DISABLED;
2849                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
2850                                 mode->channels[idx].flag |=
2851                                         HOSTAPD_CHAN_PASSIVE_SCAN;
2852                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
2853                                 mode->channels[idx].flag |=
2854                                         HOSTAPD_CHAN_NO_IBSS;
2855                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
2856                                 mode->channels[idx].flag |=
2857                                         HOSTAPD_CHAN_RADAR;
2858
2859                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
2860                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
2861                                 mode->channels[idx].max_tx_power =
2862                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
2863
2864                         idx++;
2865                 }
2866
2867                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
2868                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
2869                                   nla_len(nl_rate), rate_policy);
2870                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2871                                 continue;
2872                         mode->num_rates++;
2873                 }
2874
2875                 mode->rates = os_zalloc(mode->num_rates * sizeof(int));
2876                 if (!mode->rates)
2877                         return NL_SKIP;
2878
2879                 idx = 0;
2880
2881                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
2882                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
2883                                   nla_len(nl_rate), rate_policy);
2884                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2885                                 continue;
2886                         mode->rates[idx] = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
2887
2888                         /* crude heuristic */
2889                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
2890                             mode->rates[idx] > 200)
2891                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
2892
2893                         idx++;
2894                 }
2895         }
2896
2897         return NL_SKIP;
2898 }
2899
2900 static struct hostapd_hw_modes *
2901 wpa_driver_nl80211_add_11b(struct hostapd_hw_modes *modes, u16 *num_modes)
2902 {
2903         u16 m;
2904         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
2905         int i, mode11g_idx = -1;
2906
2907         /* If only 802.11g mode is included, use it to construct matching
2908          * 802.11b mode data. */
2909
2910         for (m = 0; m < *num_modes; m++) {
2911                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
2912                         return modes; /* 802.11b already included */
2913                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
2914                         mode11g_idx = m;
2915         }
2916
2917         if (mode11g_idx < 0)
2918                 return modes; /* 2.4 GHz band not supported at all */
2919
2920         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
2921         if (nmodes == NULL)
2922                 return modes; /* Could not add 802.11b mode */
2923
2924         mode = &nmodes[*num_modes];
2925         os_memset(mode, 0, sizeof(*mode));
2926         (*num_modes)++;
2927         modes = nmodes;
2928
2929         mode->mode = HOSTAPD_MODE_IEEE80211B;
2930
2931         mode11g = &modes[mode11g_idx];
2932         mode->num_channels = mode11g->num_channels;
2933         mode->channels = os_malloc(mode11g->num_channels *
2934                                    sizeof(struct hostapd_channel_data));
2935         if (mode->channels == NULL) {
2936                 (*num_modes)--;
2937                 return modes; /* Could not add 802.11b mode */
2938         }
2939         os_memcpy(mode->channels, mode11g->channels,
2940                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
2941
2942         mode->num_rates = 0;
2943         mode->rates = os_malloc(4 * sizeof(int));
2944         if (mode->rates == NULL) {
2945                 os_free(mode->channels);
2946                 (*num_modes)--;
2947                 return modes; /* Could not add 802.11b mode */
2948         }
2949
2950         for (i = 0; i < mode11g->num_rates; i++) {
2951                 if (mode11g->rates[i] != 10 && mode11g->rates[i] != 20 &&
2952                     mode11g->rates[i] != 55 && mode11g->rates[i] != 110)
2953                         continue;
2954                 mode->rates[mode->num_rates] = mode11g->rates[i];
2955                 mode->num_rates++;
2956                 if (mode->num_rates == 4)
2957                         break;
2958         }
2959
2960         if (mode->num_rates == 0) {
2961                 os_free(mode->channels);
2962                 os_free(mode->rates);
2963                 (*num_modes)--;
2964                 return modes; /* No 802.11b rates */
2965         }
2966
2967         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
2968                    "information");
2969
2970         return modes;
2971 }
2972
2973
2974 static struct hostapd_hw_modes *
2975 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
2976 {
2977         struct i802_bss *bss = priv;
2978         struct wpa_driver_nl80211_data *drv = bss->drv;
2979         struct nl_msg *msg;
2980         struct phy_info_arg result = {
2981                 .num_modes = num_modes,
2982                 .modes = NULL,
2983         };
2984
2985         *num_modes = 0;
2986         *flags = 0;
2987
2988         msg = nlmsg_alloc();
2989         if (!msg)
2990                 return NULL;
2991
2992         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2993                     0, NL80211_CMD_GET_WIPHY, 0);
2994
2995         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2996
2997         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
2998                 return wpa_driver_nl80211_add_11b(result.modes, num_modes);
2999  nla_put_failure:
3000         return NULL;
3001 }
3002
3003
3004 static int wpa_driver_nl80211_send_frame(struct wpa_driver_nl80211_data *drv,
3005                                          const void *data, size_t len,
3006                                          int encrypt)
3007 {
3008         __u8 rtap_hdr[] = {
3009                 0x00, 0x00, /* radiotap version */
3010                 0x0e, 0x00, /* radiotap length */
3011                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
3012                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
3013                 0x00,       /* padding */
3014                 0x00, 0x00, /* RX and TX flags to indicate that */
3015                 0x00, 0x00, /* this is the injected frame directly */
3016         };
3017         struct iovec iov[2] = {
3018                 {
3019                         .iov_base = &rtap_hdr,
3020                         .iov_len = sizeof(rtap_hdr),
3021                 },
3022                 {
3023                         .iov_base = (void *) data,
3024                         .iov_len = len,
3025                 }
3026         };
3027         struct msghdr msg = {
3028                 .msg_name = NULL,
3029                 .msg_namelen = 0,
3030                 .msg_iov = iov,
3031                 .msg_iovlen = 2,
3032                 .msg_control = NULL,
3033                 .msg_controllen = 0,
3034                 .msg_flags = 0,
3035         };
3036
3037         if (encrypt)
3038                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
3039
3040         return sendmsg(drv->monitor_sock, &msg, 0);
3041 }
3042
3043
3044 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
3045                                         size_t data_len)
3046 {
3047         struct i802_bss *bss = priv;
3048         struct wpa_driver_nl80211_data *drv = bss->drv;
3049         struct ieee80211_mgmt *mgmt;
3050         int encrypt = 1;
3051         u16 fc;
3052
3053         mgmt = (struct ieee80211_mgmt *) data;
3054         fc = le_to_host16(mgmt->frame_control);
3055
3056         if (drv->nlmode == NL80211_IFTYPE_STATION &&
3057             WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
3058             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_PROBE_RESP) {
3059                 /*
3060                  * The use of last_mgmt_freq is a bit of a hack,
3061                  * but it works due to the single-threaded nature
3062                  * of wpa_supplicant.
3063                  */
3064                 return nl80211_send_frame_cmd(drv, drv->last_mgmt_freq,
3065                                               data, data_len, NULL);
3066         }
3067
3068         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
3069             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
3070                 /*
3071                  * Only one of the authentication frame types is encrypted.
3072                  * In order for static WEP encryption to work properly (i.e.,
3073                  * to not encrypt the frame), we need to tell mac80211 about
3074                  * the frames that must not be encrypted.
3075                  */
3076                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
3077                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
3078                 if (auth_alg != WLAN_AUTH_SHARED_KEY || auth_trans != 3)
3079                         encrypt = 0;
3080         }
3081
3082         return wpa_driver_nl80211_send_frame(drv, data, data_len, encrypt);
3083 }
3084
3085
3086 static int wpa_driver_nl80211_set_beacon(void *priv,
3087                                          const u8 *head, size_t head_len,
3088                                          const u8 *tail, size_t tail_len,
3089                                          int dtim_period, int beacon_int)
3090 {
3091         struct i802_bss *bss = priv;
3092         struct wpa_driver_nl80211_data *drv = bss->drv;
3093         struct nl_msg *msg;
3094         u8 cmd = NL80211_CMD_NEW_BEACON;
3095         int ret;
3096         int beacon_set;
3097         int ifindex = if_nametoindex(bss->ifname);
3098
3099         beacon_set = bss->beacon_set;
3100
3101         msg = nlmsg_alloc();
3102         if (!msg)
3103                 return -ENOMEM;
3104
3105         wpa_printf(MSG_DEBUG, "nl80211: Set beacon (beacon_set=%d)",
3106                    beacon_set);
3107         if (beacon_set)
3108                 cmd = NL80211_CMD_SET_BEACON;
3109
3110         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3111                     0, cmd, 0);
3112         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
3113         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
3114         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
3115         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, beacon_int);
3116         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, dtim_period);
3117
3118         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3119         if (ret) {
3120                 wpa_printf(MSG_DEBUG, "nl80211: Beacon set failed: %d (%s)",
3121                            ret, strerror(-ret));
3122         } else {
3123                 bss->beacon_set = 1;
3124         }
3125         return ret;
3126  nla_put_failure:
3127         return -ENOBUFS;
3128 }
3129
3130
3131 static int wpa_driver_nl80211_set_freq(struct wpa_driver_nl80211_data *drv,
3132                                        int freq, int ht_enabled,
3133                                        int sec_channel_offset)
3134 {
3135         struct nl_msg *msg;
3136         int ret;
3137
3138         msg = nlmsg_alloc();
3139         if (!msg)
3140                 return -1;
3141
3142         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3143                     NL80211_CMD_SET_WIPHY, 0);
3144
3145         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3146         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
3147         if (ht_enabled) {
3148                 switch (sec_channel_offset) {
3149                 case -1:
3150                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3151                                     NL80211_CHAN_HT40MINUS);
3152                         break;
3153                 case 1:
3154                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3155                                     NL80211_CHAN_HT40PLUS);
3156                         break;
3157                 default:
3158                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
3159                                     NL80211_CHAN_HT20);
3160                         break;
3161                 }
3162         }
3163
3164         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3165         if (ret == 0)
3166                 return 0;
3167         wpa_printf(MSG_DEBUG, "nl80211: Failed to set channel (freq=%d): "
3168                    "%d (%s)", freq, ret, strerror(-ret));
3169 nla_put_failure:
3170         return -1;
3171 }
3172
3173
3174 static int wpa_driver_nl80211_sta_add(void *priv,
3175                                       struct hostapd_sta_add_params *params)
3176 {
3177         struct i802_bss *bss = priv;
3178         struct wpa_driver_nl80211_data *drv = bss->drv;
3179         struct nl_msg *msg;
3180         int ret = -ENOBUFS;
3181
3182         msg = nlmsg_alloc();
3183         if (!msg)
3184                 return -ENOMEM;
3185
3186         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3187                     0, NL80211_CMD_NEW_STATION, 0);
3188
3189         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
3190         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
3191         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
3192         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
3193                 params->supp_rates);
3194         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
3195                     params->listen_interval);
3196         if (params->ht_capabilities) {
3197                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
3198                         sizeof(*params->ht_capabilities),
3199                         params->ht_capabilities);
3200         }
3201
3202         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3203         if (ret)
3204                 wpa_printf(MSG_DEBUG, "nl80211: NL80211_CMD_NEW_STATION "
3205                            "result: %d (%s)", ret, strerror(-ret));
3206         if (ret == -EEXIST)
3207                 ret = 0;
3208  nla_put_failure:
3209         return ret;
3210 }
3211
3212
3213 static int wpa_driver_nl80211_sta_remove(void *priv, const u8 *addr)
3214 {
3215         struct i802_bss *bss = priv;
3216         struct wpa_driver_nl80211_data *drv = bss->drv;
3217         struct nl_msg *msg;
3218         int ret;
3219
3220         msg = nlmsg_alloc();
3221         if (!msg)
3222                 return -ENOMEM;
3223
3224         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3225                     0, NL80211_CMD_DEL_STATION, 0);
3226
3227         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3228                     if_nametoindex(bss->ifname));
3229         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3230
3231         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3232         if (ret == -ENOENT)
3233                 return 0;
3234         return ret;
3235  nla_put_failure:
3236         return -ENOBUFS;
3237 }
3238
3239
3240 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
3241                                  int ifidx)
3242 {
3243         struct nl_msg *msg;
3244
3245         wpa_printf(MSG_DEBUG, "nl80211: Remove interface ifindex=%d", ifidx);
3246
3247 #ifdef HOSTAPD
3248         /* stop listening for EAPOL on this interface */
3249         del_ifidx(drv, ifidx);
3250 #endif /* HOSTAPD */
3251
3252         msg = nlmsg_alloc();
3253         if (!msg)
3254                 goto nla_put_failure;
3255
3256         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3257                     0, NL80211_CMD_DEL_INTERFACE, 0);
3258         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
3259
3260         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
3261                 return;
3262  nla_put_failure:
3263         wpa_printf(MSG_ERROR, "Failed to remove interface (ifidx=%d)", ifidx);
3264 }
3265
3266
3267 static int nl80211_create_iface_once(struct wpa_driver_nl80211_data *drv,
3268                                      const char *ifname,
3269                                      enum nl80211_iftype iftype,
3270                                      const u8 *addr, int wds)
3271 {
3272         struct nl_msg *msg, *flags = NULL;
3273         int ifidx;
3274         int ret = -ENOBUFS;
3275
3276         msg = nlmsg_alloc();
3277         if (!msg)
3278                 return -1;
3279
3280         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3281                     0, NL80211_CMD_NEW_INTERFACE, 0);
3282         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3283         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
3284         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
3285
3286         if (iftype == NL80211_IFTYPE_MONITOR) {
3287                 int err;
3288
3289                 flags = nlmsg_alloc();
3290                 if (!flags)
3291                         goto nla_put_failure;
3292
3293                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
3294
3295                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
3296
3297                 nlmsg_free(flags);
3298
3299                 if (err)
3300                         goto nla_put_failure;
3301         } else if (wds) {
3302                 NLA_PUT_U8(msg, NL80211_ATTR_4ADDR, wds);
3303         }
3304
3305         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3306         if (ret) {
3307  nla_put_failure:
3308                 wpa_printf(MSG_ERROR, "Failed to create interface %s: %d (%s)",
3309                            ifname, ret, strerror(-ret));
3310                 return ret;
3311         }
3312
3313         ifidx = if_nametoindex(ifname);
3314         wpa_printf(MSG_DEBUG, "nl80211: New interface %s created: ifindex=%d",
3315                    ifname, ifidx);
3316
3317         if (ifidx <= 0)
3318                 return -1;
3319
3320 #ifdef HOSTAPD
3321         /* start listening for EAPOL on this interface */
3322         add_ifidx(drv, ifidx);
3323 #endif /* HOSTAPD */
3324
3325         if (addr && iftype != NL80211_IFTYPE_MONITOR &&
3326             linux_set_ifhwaddr(drv->ioctl_sock, ifname, addr)) {
3327                 nl80211_remove_iface(drv, ifidx);
3328                 return -1;
3329         }
3330
3331         return ifidx;
3332 }
3333
3334
3335 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
3336                                 const char *ifname, enum nl80211_iftype iftype,
3337                                 const u8 *addr, int wds)
3338 {
3339         int ret;
3340
3341         ret = nl80211_create_iface_once(drv, ifname, iftype, addr, wds);
3342
3343         /* if error occured and interface exists already */
3344         if (ret == -ENFILE && if_nametoindex(ifname)) {
3345                 wpa_printf(MSG_INFO, "Try to remove and re-create %s", ifname);
3346
3347                 /* Try to remove the interface that was already there. */
3348                 nl80211_remove_iface(drv, if_nametoindex(ifname));
3349
3350                 /* Try to create the interface again */
3351                 ret = nl80211_create_iface_once(drv, ifname, iftype, addr,
3352                                                 wds);
3353         }
3354
3355         if (ret >= 0 && drv->disable_11b_rates)
3356                 nl80211_disable_11b_rates(drv, ret, 1);
3357
3358         return ret;
3359 }
3360
3361
3362 static void handle_tx_callback(void *ctx, u8 *buf, size_t len, int ok)
3363 {
3364         struct ieee80211_hdr *hdr;
3365         u16 fc;
3366         union wpa_event_data event;
3367
3368         hdr = (struct ieee80211_hdr *) buf;
3369         fc = le_to_host16(hdr->frame_control);
3370
3371         os_memset(&event, 0, sizeof(event));
3372         event.tx_status.type = WLAN_FC_GET_TYPE(fc);
3373         event.tx_status.stype = WLAN_FC_GET_STYPE(fc);
3374         event.tx_status.dst = hdr->addr1;
3375         event.tx_status.data = buf;
3376         event.tx_status.data_len = len;
3377         event.tx_status.ack = ok;
3378         wpa_supplicant_event(ctx, EVENT_TX_STATUS, &event);
3379 }
3380
3381
3382 static void from_unknown_sta(struct wpa_driver_nl80211_data *drv,
3383                              u8 *buf, size_t len)
3384 {
3385         union wpa_event_data event;
3386         os_memset(&event, 0, sizeof(event));
3387         event.rx_from_unknown.frame = buf;
3388         event.rx_from_unknown.len = len;
3389         wpa_supplicant_event(drv->ctx, EVENT_RX_FROM_UNKNOWN, &event);
3390 }
3391
3392
3393 static void handle_frame(struct wpa_driver_nl80211_data *drv,
3394                          u8 *buf, size_t len, int datarate, int ssi_signal)
3395 {
3396         struct ieee80211_hdr *hdr;
3397         u16 fc;
3398         union wpa_event_data event;
3399
3400         hdr = (struct ieee80211_hdr *) buf;
3401         fc = le_to_host16(hdr->frame_control);
3402
3403         switch (WLAN_FC_GET_TYPE(fc)) {
3404         case WLAN_FC_TYPE_MGMT:
3405                 os_memset(&event, 0, sizeof(event));
3406                 event.rx_mgmt.frame = buf;
3407                 event.rx_mgmt.frame_len = len;
3408                 event.rx_mgmt.datarate = datarate;
3409                 event.rx_mgmt.ssi_signal = ssi_signal;
3410                 wpa_supplicant_event(drv->ctx, EVENT_RX_MGMT, &event);
3411                 break;
3412         case WLAN_FC_TYPE_CTRL:
3413                 /* can only get here with PS-Poll frames */
3414                 wpa_printf(MSG_DEBUG, "CTRL");
3415                 from_unknown_sta(drv, buf, len);
3416                 break;
3417         case WLAN_FC_TYPE_DATA:
3418                 from_unknown_sta(drv, buf, len);
3419                 break;
3420         }
3421 }
3422
3423
3424 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
3425 {
3426         struct wpa_driver_nl80211_data *drv = eloop_ctx;
3427         int len;
3428         unsigned char buf[3000];
3429         struct ieee80211_radiotap_iterator iter;
3430         int ret;
3431         int datarate = 0, ssi_signal = 0;
3432         int injected = 0, failed = 0, rxflags = 0;
3433
3434         len = recv(sock, buf, sizeof(buf), 0);
3435         if (len < 0) {
3436                 perror("recv");
3437                 return;
3438         }
3439
3440         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
3441                 printf("received invalid radiotap frame\n");
3442                 return;
3443         }
3444
3445         while (1) {
3446                 ret = ieee80211_radiotap_iterator_next(&iter);
3447                 if (ret == -ENOENT)
3448                         break;
3449                 if (ret) {
3450                         printf("received invalid radiotap frame (%d)\n", ret);
3451                         return;
3452                 }
3453                 switch (iter.this_arg_index) {
3454                 case IEEE80211_RADIOTAP_FLAGS:
3455                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
3456                                 len -= 4;
3457                         break;
3458                 case IEEE80211_RADIOTAP_RX_FLAGS:
3459                         rxflags = 1;
3460                         break;
3461                 case IEEE80211_RADIOTAP_TX_FLAGS:
3462                         injected = 1;
3463                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
3464                                         IEEE80211_RADIOTAP_F_TX_FAIL;
3465                         break;
3466                 case IEEE80211_RADIOTAP_DATA_RETRIES:
3467                         break;
3468                 case IEEE80211_RADIOTAP_CHANNEL:
3469                         /* TODO: convert from freq/flags to channel number */
3470                         break;
3471                 case IEEE80211_RADIOTAP_RATE:
3472                         datarate = *iter.this_arg * 5;
3473                         break;
3474                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
3475                         ssi_signal = *iter.this_arg;
3476                         break;
3477                 }
3478         }
3479
3480         if (rxflags && injected)
3481                 return;
3482
3483         if (!injected)
3484                 handle_frame(drv, buf + iter.max_length,
3485                              len - iter.max_length, datarate, ssi_signal);
3486         else
3487                 handle_tx_callback(drv->ctx, buf + iter.max_length,
3488                                    len - iter.max_length, !failed);
3489 }
3490
3491
3492 /*
3493  * we post-process the filter code later and rewrite
3494  * this to the offset to the last instruction
3495  */
3496 #define PASS    0xFF
3497 #define FAIL    0xFE
3498
3499 static struct sock_filter msock_filter_insns[] = {
3500         /*
3501          * do a little-endian load of the radiotap length field
3502          */
3503         /* load lower byte into A */
3504         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
3505         /* put it into X (== index register) */
3506         BPF_STMT(BPF_MISC| BPF_TAX, 0),
3507         /* load upper byte into A */
3508         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
3509         /* left-shift it by 8 */
3510         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
3511         /* or with X */
3512         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
3513         /* put result into X */
3514         BPF_STMT(BPF_MISC| BPF_TAX, 0),
3515
3516         /*
3517          * Allow management frames through, this also gives us those
3518          * management frames that we sent ourselves with status
3519          */
3520         /* load the lower byte of the IEEE 802.11 frame control field */
3521         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
3522         /* mask off frame type and version */
3523         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
3524         /* accept frame if it's both 0, fall through otherwise */
3525         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
3526
3527         /*
3528          * TODO: add a bit to radiotap RX flags that indicates
3529          * that the sending station is not associated, then
3530          * add a filter here that filters on our DA and that flag
3531          * to allow us to deauth frames to that bad station.
3532          *
3533          * Not a regression -- we didn't do it before either.
3534          */
3535
3536 #if 0
3537         /*
3538          * drop non-data frames
3539          */
3540         /* load the lower byte of the frame control field */
3541         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
3542         /* mask off QoS bit */
3543         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
3544         /* drop non-data frames */
3545         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
3546 #endif
3547         /* load the upper byte of the frame control field */
3548         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 1),
3549         /* mask off toDS/fromDS */
3550         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
3551         /* accept WDS frames */
3552         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, PASS, 0),
3553
3554         /*
3555          * add header length to index
3556          */
3557         /* load the lower byte of the frame control field */
3558         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
3559         /* mask off QoS bit */
3560         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
3561         /* right shift it by 6 to give 0 or 2 */
3562         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
3563         /* add data frame header length */
3564         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
3565         /* add index, was start of 802.11 header */
3566         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
3567         /* move to index, now start of LL header */
3568         BPF_STMT(BPF_MISC | BPF_TAX, 0),
3569
3570         /*
3571          * Accept empty data frames, we use those for
3572          * polling activity.
3573          */
3574         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
3575         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
3576
3577         /*
3578          * Accept EAPOL frames
3579          */
3580         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
3581         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
3582         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
3583         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
3584
3585         /* keep these last two statements or change the code below */
3586         /* return 0 == "DROP" */
3587         BPF_STMT(BPF_RET | BPF_K, 0),
3588         /* return ~0 == "keep all" */
3589         BPF_STMT(BPF_RET | BPF_K, ~0),
3590 };
3591
3592 static struct sock_fprog msock_filter = {
3593         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
3594         .filter = msock_filter_insns,
3595 };
3596
3597
3598 static int add_monitor_filter(int s)
3599 {
3600         int idx;
3601
3602         /* rewrite all PASS/FAIL jump offsets */
3603         for (idx = 0; idx < msock_filter.len; idx++) {
3604                 struct sock_filter *insn = &msock_filter_insns[idx];
3605
3606                 if (BPF_CLASS(insn->code) == BPF_JMP) {
3607                         if (insn->code == (BPF_JMP|BPF_JA)) {
3608                                 if (insn->k == PASS)
3609                                         insn->k = msock_filter.len - idx - 2;
3610                                 else if (insn->k == FAIL)
3611                                         insn->k = msock_filter.len - idx - 3;
3612                         }
3613
3614                         if (insn->jt == PASS)
3615                                 insn->jt = msock_filter.len - idx - 2;
3616                         else if (insn->jt == FAIL)
3617                                 insn->jt = msock_filter.len - idx - 3;
3618
3619                         if (insn->jf == PASS)
3620                                 insn->jf = msock_filter.len - idx - 2;
3621                         else if (insn->jf == FAIL)
3622                                 insn->jf = msock_filter.len - idx - 3;
3623                 }
3624         }
3625
3626         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
3627                        &msock_filter, sizeof(msock_filter))) {
3628                 perror("SO_ATTACH_FILTER");
3629                 return -1;
3630         }
3631
3632         return 0;
3633 }
3634
3635
3636 static void nl80211_remove_monitor_interface(
3637         struct wpa_driver_nl80211_data *drv)
3638 {
3639         if (drv->monitor_ifidx >= 0) {
3640                 nl80211_remove_iface(drv, drv->monitor_ifidx);
3641                 drv->monitor_ifidx = -1;
3642         }
3643         if (drv->monitor_sock >= 0) {
3644                 eloop_unregister_read_sock(drv->monitor_sock);
3645                 close(drv->monitor_sock);
3646                 drv->monitor_sock = -1;
3647         }
3648 }
3649
3650
3651 static int
3652 nl80211_create_monitor_interface(struct wpa_driver_nl80211_data *drv)
3653 {
3654         char buf[IFNAMSIZ];
3655         struct sockaddr_ll ll;
3656         int optval;
3657         socklen_t optlen;
3658
3659         snprintf(buf, IFNAMSIZ, "mon.%s", drv->first_bss.ifname);
3660         buf[IFNAMSIZ - 1] = '\0';
3661
3662         drv->monitor_ifidx =
3663                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL,
3664                                      0);
3665
3666         if (drv->monitor_ifidx < 0)
3667                 return -1;
3668
3669         if (linux_set_iface_flags(drv->ioctl_sock, buf, 1))
3670                 goto error;
3671
3672         memset(&ll, 0, sizeof(ll));
3673         ll.sll_family = AF_PACKET;
3674         ll.sll_ifindex = drv->monitor_ifidx;
3675         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
3676         if (drv->monitor_sock < 0) {
3677                 perror("socket[PF_PACKET,SOCK_RAW]");
3678                 goto error;
3679         }
3680
3681         if (add_monitor_filter(drv->monitor_sock)) {
3682                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
3683                            "interface; do filtering in user space");
3684                 /* This works, but will cost in performance. */
3685         }
3686
3687         if (bind(drv->monitor_sock, (struct sockaddr *) &ll, sizeof(ll)) < 0) {
3688                 perror("monitor socket bind");
3689                 goto error;
3690         }
3691
3692         optlen = sizeof(optval);
3693         optval = 20;
3694         if (setsockopt
3695             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
3696                 perror("Failed to set socket priority");
3697                 goto error;
3698         }
3699
3700         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
3701                                      drv, NULL)) {
3702                 printf("Could not register monitor read socket\n");
3703                 goto error;
3704         }
3705
3706         return 0;
3707  error:
3708         nl80211_remove_monitor_interface(drv);
3709         return -1;
3710 }
3711
3712
3713 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
3714
3715 static int wpa_driver_nl80211_hapd_send_eapol(
3716         void *priv, const u8 *addr, const u8 *data,
3717         size_t data_len, int encrypt, const u8 *own_addr)
3718 {
3719         struct i802_bss *bss = priv;
3720         struct wpa_driver_nl80211_data *drv = bss->drv;
3721         struct ieee80211_hdr *hdr;
3722         size_t len;
3723         u8 *pos;
3724         int res;
3725 #if 0 /* FIX */
3726         int qos = sta->flags & WPA_STA_WMM;
3727 #else
3728         int qos = 0;
3729 #endif
3730
3731         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
3732                 data_len;
3733         hdr = os_zalloc(len);
3734         if (hdr == NULL) {
3735                 printf("malloc() failed for i802_send_data(len=%lu)\n",
3736                        (unsigned long) len);
3737                 return -1;
3738         }
3739
3740         hdr->frame_control =
3741                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
3742         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
3743         if (encrypt)
3744                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
3745 #if 0 /* To be enabled if qos determination is added above */
3746         if (qos) {
3747                 hdr->frame_control |=
3748                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
3749         }
3750 #endif
3751
3752         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
3753         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
3754         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
3755         pos = (u8 *) (hdr + 1);
3756
3757 #if 0 /* To be enabled if qos determination is added above */
3758         if (qos) {
3759                 /* add an empty QoS header if needed */
3760                 pos[0] = 0;
3761                 pos[1] = 0;
3762                 pos += 2;
3763         }
3764 #endif
3765
3766         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
3767         pos += sizeof(rfc1042_header);
3768         WPA_PUT_BE16(pos, ETH_P_PAE);
3769         pos += 2;
3770         memcpy(pos, data, data_len);
3771
3772         res = wpa_driver_nl80211_send_frame(drv, (u8 *) hdr, len, encrypt);
3773         if (res < 0) {
3774                 wpa_printf(MSG_ERROR, "i802_send_eapol - packet len: %lu - "
3775                            "failed: %d (%s)",
3776                            (unsigned long) len, errno, strerror(errno));
3777         }
3778         os_free(hdr);
3779
3780         return res;
3781 }
3782
3783
3784 static u32 sta_flags_nl80211(int flags)
3785 {
3786         u32 f = 0;
3787
3788         if (flags & WPA_STA_AUTHORIZED)
3789                 f |= BIT(NL80211_STA_FLAG_AUTHORIZED);
3790         if (flags & WPA_STA_WMM)
3791                 f |= BIT(NL80211_STA_FLAG_WME);
3792         if (flags & WPA_STA_SHORT_PREAMBLE)
3793                 f |= BIT(NL80211_STA_FLAG_SHORT_PREAMBLE);
3794         if (flags & WPA_STA_MFP)
3795                 f |= BIT(NL80211_STA_FLAG_MFP);
3796
3797         return f;
3798 }
3799
3800
3801 static int wpa_driver_nl80211_sta_set_flags(void *priv, const u8 *addr,
3802                                             int total_flags,
3803                                             int flags_or, int flags_and)
3804 {
3805         struct i802_bss *bss = priv;
3806         struct wpa_driver_nl80211_data *drv = bss->drv;
3807         struct nl_msg *msg, *flags = NULL;
3808         struct nl80211_sta_flag_update upd;
3809
3810         msg = nlmsg_alloc();
3811         if (!msg)
3812                 return -ENOMEM;
3813
3814         flags = nlmsg_alloc();
3815         if (!flags) {
3816                 nlmsg_free(msg);
3817                 return -ENOMEM;
3818         }
3819
3820         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
3821                     0, NL80211_CMD_SET_STATION, 0);
3822
3823         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
3824                     if_nametoindex(bss->ifname));
3825         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
3826
3827         /*
3828          * Backwards compatibility version using NL80211_ATTR_STA_FLAGS. This
3829          * can be removed eventually.
3830          */
3831         if (total_flags & WPA_STA_AUTHORIZED)
3832                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
3833
3834         if (total_flags & WPA_STA_WMM)
3835                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
3836
3837         if (total_flags & WPA_STA_SHORT_PREAMBLE)
3838                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
3839
3840         if (total_flags & WPA_STA_MFP)
3841                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
3842
3843         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
3844                 goto nla_put_failure;
3845
3846         os_memset(&upd, 0, sizeof(upd));
3847         upd.mask = sta_flags_nl80211(flags_or | ~flags_and);
3848         upd.set = sta_flags_nl80211(flags_or);
3849         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
3850
3851         nlmsg_free(flags);
3852
3853         return send_and_recv_msgs(drv, msg, NULL, NULL);
3854  nla_put_failure:
3855         nlmsg_free(flags);
3856         return -ENOBUFS;
3857 }
3858
3859
3860 static int wpa_driver_nl80211_ap(struct wpa_driver_nl80211_data *drv,
3861                                  struct wpa_driver_associate_params *params)
3862 {
3863         if (params->p2p)
3864                 wpa_printf(MSG_DEBUG, "nl80211: Setup AP operations for P2P "
3865                            "group (GO)");
3866         if (wpa_driver_nl80211_set_mode(&drv->first_bss, params->mode) ||
3867             wpa_driver_nl80211_set_freq(drv, params->freq, 0, 0)) {
3868                 nl80211_remove_monitor_interface(drv);
3869                 return -1;
3870         }
3871
3872         /* TODO: setup monitor interface (and add code somewhere to remove this
3873          * when AP mode is stopped; associate with mode != 2 or drv_deinit) */
3874
3875         return 0;
3876 }
3877
3878
3879 static int nl80211_leave_ibss(struct wpa_driver_nl80211_data *drv)
3880 {
3881         struct nl_msg *msg;
3882         int ret = -1;
3883
3884         msg = nlmsg_alloc();
3885         if (!msg)
3886                 return -1;
3887
3888         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3889                     NL80211_CMD_LEAVE_IBSS, 0);
3890         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3891         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3892         msg = NULL;
3893         if (ret) {
3894                 wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS failed: ret=%d "
3895                            "(%s)", ret, strerror(-ret));
3896                 goto nla_put_failure;
3897         }
3898
3899         ret = 0;
3900         wpa_printf(MSG_DEBUG, "nl80211: Leave IBSS request sent successfully");
3901
3902 nla_put_failure:
3903         nlmsg_free(msg);
3904         return ret;
3905 }
3906
3907
3908 static int wpa_driver_nl80211_ibss(struct wpa_driver_nl80211_data *drv,
3909                                    struct wpa_driver_associate_params *params)
3910 {
3911         struct nl_msg *msg;
3912         int ret = -1;
3913         int count = 0;
3914
3915         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS (ifindex=%d)", drv->ifindex);
3916
3917         if (wpa_driver_nl80211_set_mode(&drv->first_bss, params->mode)) {
3918                 wpa_printf(MSG_INFO, "nl80211: Failed to set interface into "
3919                            "IBSS mode");
3920                 return -1;
3921         }
3922
3923 retry:
3924         msg = nlmsg_alloc();
3925         if (!msg)
3926                 return -1;
3927
3928         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3929                     NL80211_CMD_JOIN_IBSS, 0);
3930         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3931
3932         if (params->ssid == NULL || params->ssid_len > sizeof(drv->ssid))
3933                 goto nla_put_failure;
3934
3935         wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
3936                           params->ssid, params->ssid_len);
3937         NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
3938                 params->ssid);
3939         os_memcpy(drv->ssid, params->ssid, params->ssid_len);
3940         drv->ssid_len = params->ssid_len;
3941
3942         wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
3943         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
3944
3945         ret = nl80211_set_conn_keys(params, msg);
3946         if (ret)
3947                 goto nla_put_failure;
3948
3949         if (params->wpa_ie) {
3950                 wpa_hexdump(MSG_DEBUG,
3951                             "  * Extra IEs for Beacon/Probe Response frames",
3952                             params->wpa_ie, params->wpa_ie_len);
3953                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
3954                         params->wpa_ie);
3955         }
3956
3957         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
3958         msg = NULL;
3959         if (ret) {
3960                 wpa_printf(MSG_DEBUG, "nl80211: Join IBSS failed: ret=%d (%s)",
3961                            ret, strerror(-ret));
3962                 count++;
3963                 if (ret == -EALREADY && count == 1) {
3964                         wpa_printf(MSG_DEBUG, "nl80211: Retry IBSS join after "
3965                                    "forced leave");
3966                         nl80211_leave_ibss(drv);
3967                         nlmsg_free(msg);
3968                         goto retry;
3969                 }
3970
3971                 goto nla_put_failure;
3972         }
3973         ret = 0;
3974         wpa_printf(MSG_DEBUG, "nl80211: Join IBSS request sent successfully");
3975
3976 nla_put_failure:
3977         nlmsg_free(msg);
3978         return ret;
3979 }
3980
3981
3982 static int wpa_driver_nl80211_connect(
3983         struct wpa_driver_nl80211_data *drv,
3984         struct wpa_driver_associate_params *params)
3985 {
3986         struct nl_msg *msg;
3987         enum nl80211_auth_type type;
3988         int ret = 0;
3989
3990         msg = nlmsg_alloc();
3991         if (!msg)
3992                 return -1;
3993
3994         wpa_printf(MSG_DEBUG, "nl80211: Connect (ifindex=%d)", drv->ifindex);
3995         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
3996                     NL80211_CMD_CONNECT, 0);
3997
3998         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
3999         if (params->bssid) {
4000                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
4001                            MAC2STR(params->bssid));
4002                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
4003         }
4004         if (params->freq) {
4005                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
4006                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
4007         }
4008         if (params->ssid) {
4009                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
4010                                   params->ssid, params->ssid_len);
4011                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
4012                         params->ssid);
4013                 if (params->ssid_len > sizeof(drv->ssid))
4014                         goto nla_put_failure;
4015                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
4016                 drv->ssid_len = params->ssid_len;
4017         }
4018         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
4019         if (params->wpa_ie)
4020                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
4021                         params->wpa_ie);
4022
4023         if (params->auth_alg & WPA_AUTH_ALG_OPEN)
4024                 type = NL80211_AUTHTYPE_OPEN_SYSTEM;
4025         else if (params->auth_alg & WPA_AUTH_ALG_SHARED)
4026                 type = NL80211_AUTHTYPE_SHARED_KEY;
4027         else if (params->auth_alg & WPA_AUTH_ALG_LEAP)
4028                 type = NL80211_AUTHTYPE_NETWORK_EAP;
4029         else if (params->auth_alg & WPA_AUTH_ALG_FT)
4030                 type = NL80211_AUTHTYPE_FT;
4031         else
4032                 goto nla_put_failure;
4033
4034         wpa_printf(MSG_DEBUG, "  * Auth Type %d", type);
4035         NLA_PUT_U32(msg, NL80211_ATTR_AUTH_TYPE, type);
4036
4037         if (params->wpa_ie && params->wpa_ie_len) {
4038                 enum nl80211_wpa_versions ver;
4039
4040                 if (params->wpa_ie[0] == WLAN_EID_RSN)
4041                         ver = NL80211_WPA_VERSION_2;
4042                 else
4043                         ver = NL80211_WPA_VERSION_1;
4044
4045                 wpa_printf(MSG_DEBUG, "  * WPA Version %d", ver);
4046                 NLA_PUT_U32(msg, NL80211_ATTR_WPA_VERSIONS, ver);
4047         }
4048
4049         if (params->pairwise_suite != CIPHER_NONE) {
4050                 int cipher;
4051
4052                 switch (params->pairwise_suite) {
4053                 case CIPHER_WEP40:
4054                         cipher = WLAN_CIPHER_SUITE_WEP40;
4055                         break;
4056                 case CIPHER_WEP104:
4057                         cipher = WLAN_CIPHER_SUITE_WEP104;
4058                         break;
4059                 case CIPHER_CCMP:
4060                         cipher = WLAN_CIPHER_SUITE_CCMP;
4061                         break;
4062                 case CIPHER_TKIP:
4063                 default:
4064                         cipher = WLAN_CIPHER_SUITE_TKIP;
4065                         break;
4066                 }
4067                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITES_PAIRWISE, cipher);
4068         }
4069
4070         if (params->group_suite != CIPHER_NONE) {
4071                 int cipher;
4072
4073                 switch (params->group_suite) {
4074                 case CIPHER_WEP40:
4075                         cipher = WLAN_CIPHER_SUITE_WEP40;
4076                         break;
4077                 case CIPHER_WEP104:
4078                         cipher = WLAN_CIPHER_SUITE_WEP104;
4079                         break;
4080                 case CIPHER_CCMP:
4081                         cipher = WLAN_CIPHER_SUITE_CCMP;
4082                         break;
4083                 case CIPHER_TKIP:
4084                 default:
4085                         cipher = WLAN_CIPHER_SUITE_TKIP;
4086                         break;
4087                 }
4088                 NLA_PUT_U32(msg, NL80211_ATTR_CIPHER_SUITE_GROUP, cipher);
4089         }
4090
4091         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
4092             params->key_mgmt_suite == KEY_MGMT_PSK) {
4093                 int mgmt = WLAN_AKM_SUITE_PSK;
4094
4095                 switch (params->key_mgmt_suite) {
4096                 case KEY_MGMT_802_1X:
4097                         mgmt = WLAN_AKM_SUITE_8021X;
4098                         break;
4099                 case KEY_MGMT_PSK:
4100                 default:
4101                         mgmt = WLAN_AKM_SUITE_PSK;
4102                         break;
4103                 }
4104                 NLA_PUT_U32(msg, NL80211_ATTR_AKM_SUITES, mgmt);
4105         }
4106
4107         ret = nl80211_set_conn_keys(params, msg);
4108         if (ret)
4109                 goto nla_put_failure;
4110
4111         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4112         msg = NULL;
4113         if (ret) {
4114                 wpa_printf(MSG_DEBUG, "nl80211: MLME connect failed: ret=%d "
4115                            "(%s)", ret, strerror(-ret));
4116                 goto nla_put_failure;
4117         }
4118         ret = 0;
4119         wpa_printf(MSG_DEBUG, "nl80211: Connect request send successfully");
4120
4121 nla_put_failure:
4122         nlmsg_free(msg);
4123         return ret;
4124
4125 }
4126
4127
4128 static int wpa_driver_nl80211_associate(
4129         void *priv, struct wpa_driver_associate_params *params)
4130 {
4131         struct i802_bss *bss = priv;
4132         struct wpa_driver_nl80211_data *drv = bss->drv;
4133         int ret = -1;
4134         struct nl_msg *msg;
4135
4136         if (params->mode == IEEE80211_MODE_AP)
4137                 return wpa_driver_nl80211_ap(drv, params);
4138
4139         if (params->mode == IEEE80211_MODE_IBSS)
4140                 return wpa_driver_nl80211_ibss(drv, params);
4141
4142         if (!(drv->capa.flags & WPA_DRIVER_FLAGS_SME)) {
4143                 if (wpa_driver_nl80211_set_mode(priv, params->mode) < 0)
4144                         return -1;
4145                 return wpa_driver_nl80211_connect(drv, params);
4146         }
4147
4148         drv->associated = 0;
4149
4150         msg = nlmsg_alloc();
4151         if (!msg)
4152                 return -1;
4153
4154         wpa_printf(MSG_DEBUG, "nl80211: Associate (ifindex=%d)",
4155                    drv->ifindex);
4156         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
4157                     NL80211_CMD_ASSOCIATE, 0);
4158
4159         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4160         if (params->bssid) {
4161                 wpa_printf(MSG_DEBUG, "  * bssid=" MACSTR,
4162                            MAC2STR(params->bssid));
4163                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->bssid);
4164         }
4165         if (params->freq) {
4166                 wpa_printf(MSG_DEBUG, "  * freq=%d", params->freq);
4167                 NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, params->freq);
4168                 drv->assoc_freq = params->freq;
4169         } else
4170                 drv->assoc_freq = 0;
4171         if (params->ssid) {
4172                 wpa_hexdump_ascii(MSG_DEBUG, "  * SSID",
4173                                   params->ssid, params->ssid_len);
4174                 NLA_PUT(msg, NL80211_ATTR_SSID, params->ssid_len,
4175                         params->ssid);
4176                 if (params->ssid_len > sizeof(drv->ssid))
4177                         goto nla_put_failure;
4178                 os_memcpy(drv->ssid, params->ssid, params->ssid_len);
4179                 drv->ssid_len = params->ssid_len;
4180         }
4181         wpa_hexdump(MSG_DEBUG, "  * IEs", params->wpa_ie, params->wpa_ie_len);
4182         if (params->wpa_ie)
4183                 NLA_PUT(msg, NL80211_ATTR_IE, params->wpa_ie_len,
4184                         params->wpa_ie);
4185
4186 #ifdef CONFIG_IEEE80211W
4187         if (params->mgmt_frame_protection == MGMT_FRAME_PROTECTION_REQUIRED)
4188                 NLA_PUT_U32(msg, NL80211_ATTR_USE_MFP, NL80211_MFP_REQUIRED);
4189 #endif /* CONFIG_IEEE80211W */
4190
4191         NLA_PUT_FLAG(msg, NL80211_ATTR_CONTROL_PORT);
4192
4193         if (params->prev_bssid) {
4194                 wpa_printf(MSG_DEBUG, "  * prev_bssid=" MACSTR,
4195                            MAC2STR(params->prev_bssid));
4196                 NLA_PUT(msg, NL80211_ATTR_PREV_BSSID, ETH_ALEN,
4197                         params->prev_bssid);
4198         }
4199
4200         if (params->p2p)
4201                 wpa_printf(MSG_DEBUG, "  * P2P group");
4202
4203         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4204         msg = NULL;
4205         if (ret) {
4206                 wpa_printf(MSG_DEBUG, "nl80211: MLME command failed: ret=%d "
4207                            "(%s)", ret, strerror(-ret));
4208                 nl80211_dump_scan(drv);
4209                 goto nla_put_failure;
4210         }
4211         ret = 0;
4212         wpa_printf(MSG_DEBUG, "nl80211: Association request send "
4213                    "successfully");
4214
4215 nla_put_failure:
4216         nlmsg_free(msg);
4217         return ret;
4218 }
4219
4220
4221 static int nl80211_set_mode(struct wpa_driver_nl80211_data *drv,
4222                             int ifindex, int mode)
4223 {
4224         struct nl_msg *msg;
4225         int ret = -ENOBUFS;
4226
4227         msg = nlmsg_alloc();
4228         if (!msg)
4229                 return -ENOMEM;
4230
4231         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4232                     0, NL80211_CMD_SET_INTERFACE, 0);
4233         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
4234         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
4235
4236         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4237         if (!ret)
4238                 return 0;
4239 nla_put_failure:
4240         wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface %d to mode %d:"
4241                    " %d (%s)", ifindex, mode, ret, strerror(-ret));
4242         return ret;
4243 }
4244
4245
4246 static int wpa_driver_nl80211_set_mode(void *priv, int mode)
4247 {
4248         struct i802_bss *bss = priv;
4249         struct wpa_driver_nl80211_data *drv = bss->drv;
4250         int ret = -1;
4251         int nlmode;
4252         int i;
4253
4254         switch (mode) {
4255         case 0:
4256                 nlmode = NL80211_IFTYPE_STATION;
4257                 break;
4258         case 1:
4259                 nlmode = NL80211_IFTYPE_ADHOC;
4260                 break;
4261         case 2:
4262                 nlmode = NL80211_IFTYPE_AP;
4263                 break;
4264         default:
4265                 return -1;
4266         }
4267
4268         if (nl80211_set_mode(drv, drv->ifindex, nlmode) == 0) {
4269                 drv->nlmode = nlmode;
4270                 ret = 0;
4271                 goto done;
4272         }
4273
4274         if (nlmode == drv->nlmode) {
4275                 wpa_printf(MSG_DEBUG, "nl80211: Interface already in "
4276                            "requested mode - ignore error");
4277                 ret = 0;
4278                 goto done; /* Already in the requested mode */
4279         }
4280
4281         /* mac80211 doesn't allow mode changes while the device is up, so
4282          * take the device down, try to set the mode again, and bring the
4283          * device back up.
4284          */
4285         wpa_printf(MSG_DEBUG, "nl80211: Try mode change after setting "
4286                    "interface down");
4287         for (i = 0; i < 10; i++) {
4288                 if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 0) ==
4289                     0) {
4290                         /* Try to set the mode again while the interface is
4291                          * down */
4292                         ret = nl80211_set_mode(drv, drv->ifindex, nlmode);
4293                         if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname,
4294                                                   1))
4295                                 ret = -1;
4296                         if (!ret)
4297                                 break;
4298                 } else
4299                         wpa_printf(MSG_DEBUG, "nl80211: Failed to set "
4300                                    "interface down");
4301                 os_sleep(0, 100000);
4302         }
4303
4304         if (!ret) {
4305                 wpa_printf(MSG_DEBUG, "nl80211: Mode change succeeded while "
4306                            "interface is down");
4307                 drv->nlmode = nlmode;
4308         }
4309
4310 done:
4311         if (!ret && nlmode == NL80211_IFTYPE_AP) {
4312                 /* Setup additional AP mode functionality if needed */
4313                 if (drv->monitor_ifidx < 0 &&
4314                     nl80211_create_monitor_interface(drv))
4315                         return -1;
4316         } else if (!ret && nlmode != NL80211_IFTYPE_AP) {
4317                 /* Remove additional AP mode functionality */
4318                 nl80211_remove_monitor_interface(drv);
4319                 bss->beacon_set = 0;
4320         }
4321
4322         if (ret)
4323                 wpa_printf(MSG_DEBUG, "nl80211: Interface mode change to %d "
4324                            "from %d failed", nlmode, drv->nlmode);
4325
4326         return ret;
4327 }
4328
4329
4330 static int wpa_driver_nl80211_get_capa(void *priv,
4331                                        struct wpa_driver_capa *capa)
4332 {
4333         struct i802_bss *bss = priv;
4334         struct wpa_driver_nl80211_data *drv = bss->drv;
4335         if (!drv->has_capability)
4336                 return -1;
4337         os_memcpy(capa, &drv->capa, sizeof(*capa));
4338         return 0;
4339 }
4340
4341
4342 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
4343 {
4344         struct i802_bss *bss = priv;
4345         struct wpa_driver_nl80211_data *drv = bss->drv;
4346
4347         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
4348                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
4349         drv->operstate = state;
4350         return netlink_send_oper_ifla(drv->netlink, drv->ifindex, -1,
4351                                       state ? IF_OPER_UP : IF_OPER_DORMANT);
4352 }
4353
4354
4355 static int wpa_driver_nl80211_set_supp_port(void *priv, int authorized)
4356 {
4357         struct i802_bss *bss = priv;
4358         struct wpa_driver_nl80211_data *drv = bss->drv;
4359         struct nl_msg *msg;
4360         struct nl80211_sta_flag_update upd;
4361
4362         msg = nlmsg_alloc();
4363         if (!msg)
4364                 return -ENOMEM;
4365
4366         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4367                     0, NL80211_CMD_SET_STATION, 0);
4368
4369         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
4370                     if_nametoindex(bss->ifname));
4371         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, drv->bssid);
4372
4373         os_memset(&upd, 0, sizeof(upd));
4374         upd.mask = BIT(NL80211_STA_FLAG_AUTHORIZED);
4375         if (authorized)
4376                 upd.set = BIT(NL80211_STA_FLAG_AUTHORIZED);
4377         NLA_PUT(msg, NL80211_ATTR_STA_FLAGS2, sizeof(upd), &upd);
4378
4379         return send_and_recv_msgs(drv, msg, NULL, NULL);
4380  nla_put_failure:
4381         return -ENOBUFS;
4382 }
4383
4384
4385 #ifdef HOSTAPD
4386
4387 static void add_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
4388 {
4389         int i;
4390         int *old;
4391
4392         wpa_printf(MSG_DEBUG, "nl80211: Add own interface ifindex %d",
4393                    ifidx);
4394         for (i = 0; i < drv->num_if_indices; i++) {
4395                 if (drv->if_indices[i] == 0) {
4396                         drv->if_indices[i] = ifidx;
4397                         return;
4398                 }
4399         }
4400
4401         if (drv->if_indices != drv->default_if_indices)
4402                 old = drv->if_indices;
4403         else
4404                 old = NULL;
4405
4406         drv->if_indices = os_realloc(old,
4407                                      sizeof(int) * (drv->num_if_indices + 1));
4408         if (!drv->if_indices) {
4409                 if (!old)
4410                         drv->if_indices = drv->default_if_indices;
4411                 else
4412                         drv->if_indices = old;
4413                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
4414                            "interfaces");
4415                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
4416                 return;
4417         } else if (!old)
4418                 os_memcpy(drv->if_indices, drv->default_if_indices,
4419                           sizeof(drv->default_if_indices));
4420         drv->if_indices[drv->num_if_indices] = ifidx;
4421         drv->num_if_indices++;
4422 }
4423
4424
4425 static void del_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
4426 {
4427         int i;
4428
4429         for (i = 0; i < drv->num_if_indices; i++) {
4430                 if (drv->if_indices[i] == ifidx) {
4431                         drv->if_indices[i] = 0;
4432                         break;
4433                 }
4434         }
4435 }
4436
4437
4438 static int have_ifidx(struct wpa_driver_nl80211_data *drv, int ifidx)
4439 {
4440         int i;
4441
4442         for (i = 0; i < drv->num_if_indices; i++)
4443                 if (drv->if_indices[i] == ifidx)
4444                         return 1;
4445
4446         return 0;
4447 }
4448
4449
4450 static inline int min_int(int a, int b)
4451 {
4452         if (a < b)
4453                 return a;
4454         return b;
4455 }
4456
4457
4458 static int get_key_handler(struct nl_msg *msg, void *arg)
4459 {
4460         struct nlattr *tb[NL80211_ATTR_MAX + 1];
4461         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
4462
4463         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
4464                   genlmsg_attrlen(gnlh, 0), NULL);
4465
4466         /*
4467          * TODO: validate the key index and mac address!
4468          * Otherwise, there's a race condition as soon as
4469          * the kernel starts sending key notifications.
4470          */
4471
4472         if (tb[NL80211_ATTR_KEY_SEQ])
4473                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
4474                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
4475         return NL_SKIP;
4476 }
4477
4478
4479 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
4480                            int idx, u8 *seq)
4481 {
4482         struct i802_bss *bss = priv;
4483         struct wpa_driver_nl80211_data *drv = bss->drv;
4484         struct nl_msg *msg;
4485
4486         msg = nlmsg_alloc();
4487         if (!msg)
4488                 return -ENOMEM;
4489
4490         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4491                     0, NL80211_CMD_GET_KEY, 0);
4492
4493         if (addr)
4494                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4495         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
4496         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
4497
4498         memset(seq, 0, 6);
4499
4500         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
4501  nla_put_failure:
4502         return -ENOBUFS;
4503 }
4504
4505
4506 static int i802_set_rate_sets(void *priv, int *supp_rates, int *basic_rates,
4507                               int mode)
4508 {
4509         struct i802_bss *bss = priv;
4510         struct wpa_driver_nl80211_data *drv = bss->drv;
4511         struct nl_msg *msg;
4512         u8 rates[NL80211_MAX_SUPP_RATES];
4513         u8 rates_len = 0;
4514         int i;
4515
4516         msg = nlmsg_alloc();
4517         if (!msg)
4518                 return -ENOMEM;
4519
4520         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
4521                     NL80211_CMD_SET_BSS, 0);
4522
4523         for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0; i++)
4524                 rates[rates_len++] = basic_rates[i] / 5;
4525
4526         NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
4527
4528         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
4529
4530         return send_and_recv_msgs(drv, msg, NULL, NULL);
4531  nla_put_failure:
4532         return -ENOBUFS;
4533 }
4534
4535 #endif /* HOSTAPD */
4536
4537
4538 /* Set kernel driver on given frequency (MHz) */
4539 static int i802_set_freq(void *priv, struct hostapd_freq_params *freq)
4540 {
4541         struct i802_bss *bss = priv;
4542         struct wpa_driver_nl80211_data *drv = bss->drv;
4543         return wpa_driver_nl80211_set_freq(drv, freq->freq, freq->ht_enabled,
4544                                            freq->sec_channel_offset);
4545 }
4546
4547
4548 #ifdef HOSTAPD
4549
4550 static int i802_set_rts(void *priv, int rts)
4551 {
4552         struct i802_bss *bss = priv;
4553         struct wpa_driver_nl80211_data *drv = bss->drv;
4554         struct nl_msg *msg;
4555         int ret = -ENOBUFS;
4556         u32 val;
4557
4558         msg = nlmsg_alloc();
4559         if (!msg)
4560                 return -ENOMEM;
4561
4562         if (rts >= 2347)
4563                 val = (u32) -1;
4564         else
4565                 val = rts;
4566
4567         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4568                     0, NL80211_CMD_SET_WIPHY, 0);
4569         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4570         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_RTS_THRESHOLD, val);
4571
4572         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4573         if (!ret)
4574                 return 0;
4575 nla_put_failure:
4576         wpa_printf(MSG_DEBUG, "nl80211: Failed to set RTS threshold %d: "
4577                    "%d (%s)", rts, ret, strerror(-ret));
4578         return ret;
4579 }
4580
4581
4582 static int i802_set_frag(void *priv, int frag)
4583 {
4584         struct i802_bss *bss = priv;
4585         struct wpa_driver_nl80211_data *drv = bss->drv;
4586         struct nl_msg *msg;
4587         int ret = -ENOBUFS;
4588         u32 val;
4589
4590         msg = nlmsg_alloc();
4591         if (!msg)
4592                 return -ENOMEM;
4593
4594         if (frag >= 2346)
4595                 val = (u32) -1;
4596         else
4597                 val = frag;
4598
4599         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4600                     0, NL80211_CMD_SET_WIPHY, 0);
4601         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
4602         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FRAG_THRESHOLD, val);
4603
4604         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4605         if (!ret)
4606                 return 0;
4607 nla_put_failure:
4608         wpa_printf(MSG_DEBUG, "nl80211: Failed to set fragmentation threshold "
4609                    "%d: %d (%s)", frag, ret, strerror(-ret));
4610         return ret;
4611 }
4612
4613
4614 static int i802_flush(void *priv)
4615 {
4616         struct i802_bss *bss = priv;
4617         struct wpa_driver_nl80211_data *drv = bss->drv;
4618         struct nl_msg *msg;
4619
4620         msg = nlmsg_alloc();
4621         if (!msg)
4622                 return -1;
4623
4624         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4625                     0, NL80211_CMD_DEL_STATION, 0);
4626
4627         /*
4628          * XXX: FIX! this needs to flush all VLANs too
4629          */
4630         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
4631                     if_nametoindex(bss->ifname));
4632
4633         return send_and_recv_msgs(drv, msg, NULL, NULL);
4634  nla_put_failure:
4635         return -ENOBUFS;
4636 }
4637
4638
4639 static int get_sta_handler(struct nl_msg *msg, void *arg)
4640 {
4641         struct nlattr *tb[NL80211_ATTR_MAX + 1];
4642         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
4643         struct hostap_sta_driver_data *data = arg;
4644         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
4645         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
4646                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
4647                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
4648                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
4649                 [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 },
4650                 [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 },
4651         };
4652
4653         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
4654                   genlmsg_attrlen(gnlh, 0), NULL);
4655
4656         /*
4657          * TODO: validate the interface and mac address!
4658          * Otherwise, there's a race condition as soon as
4659          * the kernel starts sending station notifications.
4660          */
4661
4662         if (!tb[NL80211_ATTR_STA_INFO]) {
4663                 wpa_printf(MSG_DEBUG, "sta stats missing!");
4664                 return NL_SKIP;
4665         }
4666         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
4667                              tb[NL80211_ATTR_STA_INFO],
4668                              stats_policy)) {
4669                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
4670                 return NL_SKIP;
4671         }
4672
4673         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
4674                 data->inactive_msec =
4675                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
4676         if (stats[NL80211_STA_INFO_RX_BYTES])
4677                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
4678         if (stats[NL80211_STA_INFO_TX_BYTES])
4679                 data->tx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
4680         if (stats[NL80211_STA_INFO_RX_PACKETS])
4681                 data->rx_packets =
4682                         nla_get_u32(stats[NL80211_STA_INFO_RX_PACKETS]);
4683         if (stats[NL80211_STA_INFO_TX_PACKETS])
4684                 data->tx_packets =
4685                         nla_get_u32(stats[NL80211_STA_INFO_TX_PACKETS]);
4686
4687         return NL_SKIP;
4688 }
4689
4690 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
4691                               const u8 *addr)
4692 {
4693         struct i802_bss *bss = priv;
4694         struct wpa_driver_nl80211_data *drv = bss->drv;
4695         struct nl_msg *msg;
4696
4697         os_memset(data, 0, sizeof(*data));
4698         msg = nlmsg_alloc();
4699         if (!msg)
4700                 return -ENOMEM;
4701
4702         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4703                     0, NL80211_CMD_GET_STATION, 0);
4704
4705         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4706         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
4707
4708         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
4709  nla_put_failure:
4710         return -ENOBUFS;
4711 }
4712
4713
4714 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
4715                                     int cw_min, int cw_max, int burst_time)
4716 {
4717         struct i802_bss *bss = priv;
4718         struct wpa_driver_nl80211_data *drv = bss->drv;
4719         struct nl_msg *msg;
4720         struct nlattr *txq, *params;
4721
4722         msg = nlmsg_alloc();
4723         if (!msg)
4724                 return -1;
4725
4726         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4727                     0, NL80211_CMD_SET_WIPHY, 0);
4728
4729         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
4730
4731         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
4732         if (!txq)
4733                 goto nla_put_failure;
4734
4735         /* We are only sending parameters for a single TXQ at a time */
4736         params = nla_nest_start(msg, 1);
4737         if (!params)
4738                 goto nla_put_failure;
4739
4740         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, queue);
4741         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
4742          * 32 usec, so need to convert the value here. */
4743         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
4744         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
4745         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
4746         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
4747
4748         nla_nest_end(msg, params);
4749
4750         nla_nest_end(msg, txq);
4751
4752         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
4753                 return 0;
4754  nla_put_failure:
4755         return -1;
4756 }
4757
4758
4759 static int i802_set_bss(void *priv, int cts, int preamble, int slot)
4760 {
4761         struct i802_bss *bss = priv;
4762         struct wpa_driver_nl80211_data *drv = bss->drv;
4763         struct nl_msg *msg;
4764
4765         msg = nlmsg_alloc();
4766         if (!msg)
4767                 return -ENOMEM;
4768
4769         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
4770                     NL80211_CMD_SET_BSS, 0);
4771
4772         if (cts >= 0)
4773                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
4774         if (preamble >= 0)
4775                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
4776         if (slot >= 0)
4777                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
4778
4779         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
4780
4781         return send_and_recv_msgs(drv, msg, NULL, NULL);
4782  nla_put_failure:
4783         return -ENOBUFS;
4784 }
4785
4786
4787 static int i802_set_cts_protect(void *priv, int value)
4788 {
4789         return i802_set_bss(priv, value, -1, -1);
4790 }
4791
4792
4793 static int i802_set_preamble(void *priv, int value)
4794 {
4795         return i802_set_bss(priv, -1, value, -1);
4796 }
4797
4798
4799 static int i802_set_short_slot_time(void *priv, int value)
4800 {
4801         return i802_set_bss(priv, -1, -1, value);
4802 }
4803
4804
4805 static int i802_set_sta_vlan(void *priv, const u8 *addr,
4806                              const char *ifname, int vlan_id)
4807 {
4808         struct i802_bss *bss = priv;
4809         struct wpa_driver_nl80211_data *drv = bss->drv;
4810         struct nl_msg *msg;
4811         int ret = -ENOBUFS;
4812
4813         msg = nlmsg_alloc();
4814         if (!msg)
4815                 return -ENOMEM;
4816
4817         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
4818                     0, NL80211_CMD_SET_STATION, 0);
4819
4820         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
4821                     if_nametoindex(bss->ifname));
4822         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
4823         NLA_PUT_U32(msg, NL80211_ATTR_STA_VLAN,
4824                     if_nametoindex(ifname));
4825
4826         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
4827         if (ret < 0) {
4828                 wpa_printf(MSG_ERROR, "nl80211: NL80211_ATTR_STA_VLAN (addr="
4829                            MACSTR " ifname=%s vlan_id=%d) failed: %d (%s)",
4830                            MAC2STR(addr), ifname, vlan_id, ret,
4831                            strerror(-ret));
4832         }
4833  nla_put_failure:
4834         return ret;
4835 }
4836
4837
4838 static int i802_set_wds_sta(void *priv, const u8 *addr, int aid, int val)
4839 {
4840         struct i802_bss *bss = priv;
4841         struct wpa_driver_nl80211_data *drv = bss->drv;
4842         char name[IFNAMSIZ + 1];
4843
4844         os_snprintf(name, sizeof(name), "%s.sta%d", bss->ifname, aid);
4845         wpa_printf(MSG_DEBUG, "nl80211: Set WDS STA addr=" MACSTR
4846                    " aid=%d val=%d name=%s", MAC2STR(addr), aid, val, name);
4847         if (val) {
4848                 if (nl80211_create_iface(drv, name, NL80211_IFTYPE_AP_VLAN,
4849                                          NULL, 1) < 0)
4850                         return -1;
4851                 linux_set_iface_flags(drv->ioctl_sock, name, 1);
4852                 return i802_set_sta_vlan(priv, addr, name, 0);
4853         } else {
4854                 i802_set_sta_vlan(priv, addr, bss->ifname, 0);
4855                 return wpa_driver_nl80211_if_remove(priv, WPA_IF_AP_VLAN,
4856                                                     name);
4857         }
4858 }
4859
4860
4861 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
4862 {
4863         struct wpa_driver_nl80211_data *drv = eloop_ctx;
4864         struct sockaddr_ll lladdr;
4865         unsigned char buf[3000];
4866         int len;
4867         socklen_t fromlen = sizeof(lladdr);
4868
4869         len = recvfrom(sock, buf, sizeof(buf), 0,
4870                        (struct sockaddr *)&lladdr, &fromlen);
4871         if (len < 0) {
4872                 perror("recv");
4873                 return;
4874         }
4875
4876         if (have_ifidx(drv, lladdr.sll_ifindex))
4877                 drv_event_eapol_rx(drv->ctx, lladdr.sll_addr, buf, len);
4878 }
4879
4880
4881 static int i802_get_inact_sec(void *priv, const u8 *addr)
4882 {
4883         struct hostap_sta_driver_data data;
4884         int ret;
4885
4886         data.inactive_msec = (unsigned long) -1;
4887         ret = i802_read_sta_data(priv, &data, addr);
4888         if (ret || data.inactive_msec == (unsigned long) -1)
4889                 return -1;
4890         return data.inactive_msec / 1000;
4891 }
4892
4893
4894 static int i802_sta_clear_stats(void *priv, const u8 *addr)
4895 {
4896 #if 0
4897         /* TODO */
4898 #endif
4899         return 0;
4900 }
4901
4902
4903 static int i802_sta_deauth(void *priv, const u8 *own_addr, const u8 *addr,
4904                            int reason)
4905 {
4906         struct i802_bss *bss = priv;
4907         struct ieee80211_mgmt mgmt;
4908
4909         memset(&mgmt, 0, sizeof(mgmt));
4910         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
4911                                           WLAN_FC_STYPE_DEAUTH);
4912         memcpy(mgmt.da, addr, ETH_ALEN);
4913         memcpy(mgmt.sa, own_addr, ETH_ALEN);
4914         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
4915         mgmt.u.deauth.reason_code = host_to_le16(reason);
4916         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
4917                                             IEEE80211_HDRLEN +
4918                                             sizeof(mgmt.u.deauth));
4919 }
4920
4921
4922 static int i802_sta_disassoc(void *priv, const u8 *own_addr, const u8 *addr,
4923                              int reason)
4924 {
4925         struct i802_bss *bss = priv;
4926         struct ieee80211_mgmt mgmt;
4927
4928         memset(&mgmt, 0, sizeof(mgmt));
4929         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
4930                                           WLAN_FC_STYPE_DISASSOC);
4931         memcpy(mgmt.da, addr, ETH_ALEN);
4932         memcpy(mgmt.sa, own_addr, ETH_ALEN);
4933         memcpy(mgmt.bssid, own_addr, ETH_ALEN);
4934         mgmt.u.disassoc.reason_code = host_to_le16(reason);
4935         return wpa_driver_nl80211_send_mlme(bss, (u8 *) &mgmt,
4936                                             IEEE80211_HDRLEN +
4937                                             sizeof(mgmt.u.disassoc));
4938 }
4939
4940
4941 static int i802_check_bridge(struct wpa_driver_nl80211_data *drv,
4942                              const char *brname, const char *ifname)
4943 {
4944         int ifindex;
4945         char in_br[IFNAMSIZ];
4946
4947         os_strlcpy(drv->brname, brname, IFNAMSIZ);
4948         ifindex = if_nametoindex(brname);
4949         if (ifindex == 0) {
4950                 /*
4951                  * Bridge was configured, but the bridge device does
4952                  * not exist. Try to add it now.
4953                  */
4954                 if (linux_br_add(drv->ioctl_sock, brname) < 0) {
4955                         wpa_printf(MSG_ERROR, "nl80211: Failed to add the "
4956                                    "bridge interface %s: %s",
4957                                    brname, strerror(errno));
4958                         return -1;
4959                 }
4960                 drv->added_bridge = 1;
4961                 add_ifidx(drv, if_nametoindex(brname));
4962         }
4963
4964         if (linux_br_get(in_br, ifname) == 0) {
4965                 if (os_strcmp(in_br, brname) == 0)
4966                         return 0; /* already in the bridge */
4967
4968                 wpa_printf(MSG_DEBUG, "nl80211: Removing interface %s from "
4969                            "bridge %s", ifname, in_br);
4970                 if (linux_br_del_if(drv->ioctl_sock, in_br, ifname) < 0) {
4971                         wpa_printf(MSG_ERROR, "nl80211: Failed to "
4972                                    "remove interface %s from bridge "
4973                                    "%s: %s",
4974                                    ifname, brname, strerror(errno));
4975                         return -1;
4976                 }
4977         }
4978
4979         wpa_printf(MSG_DEBUG, "nl80211: Adding interface %s into bridge %s",
4980                    ifname, brname);
4981         if (linux_br_add_if(drv->ioctl_sock, brname, ifname) < 0) {
4982                 wpa_printf(MSG_ERROR, "nl80211: Failed to add interface %s "
4983                            "into bridge %s: %s",
4984                            ifname, brname, strerror(errno));
4985                 return -1;
4986         }
4987         drv->added_if_into_bridge = 1;
4988
4989         return 0;
4990 }
4991
4992
4993 static void *i802_init(struct hostapd_data *hapd,
4994                        struct wpa_init_params *params)
4995 {
4996         struct wpa_driver_nl80211_data *drv;
4997         struct i802_bss *bss;
4998         size_t i;
4999         char brname[IFNAMSIZ];
5000         int ifindex, br_ifindex;
5001         int br_added = 0;
5002
5003         bss = wpa_driver_nl80211_init(hapd, params->ifname);
5004         if (bss == NULL)
5005                 return NULL;
5006
5007         drv = bss->drv;
5008         if (linux_br_get(brname, params->ifname) == 0) {
5009                 wpa_printf(MSG_DEBUG, "nl80211: Interface %s is in bridge %s",
5010                            params->ifname, brname);
5011                 br_ifindex = if_nametoindex(brname);
5012         } else {
5013                 brname[0] = '\0';
5014                 br_ifindex = 0;
5015         }
5016
5017         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
5018         drv->if_indices = drv->default_if_indices;
5019         for (i = 0; i < params->num_bridge; i++) {
5020                 if (params->bridge[i]) {
5021                         ifindex = if_nametoindex(params->bridge[i]);
5022                         if (ifindex)
5023                                 add_ifidx(drv, ifindex);
5024                         if (ifindex == br_ifindex)
5025                                 br_added = 1;
5026                 }
5027         }
5028         if (!br_added && br_ifindex &&
5029             (params->num_bridge == 0 || !params->bridge[0]))
5030                 add_ifidx(drv, br_ifindex);
5031
5032         /* start listening for EAPOL on the default AP interface */
5033         add_ifidx(drv, drv->ifindex);
5034
5035         if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 0))
5036                 goto failed;
5037
5038         if (params->bssid) {
5039                 if (linux_set_ifhwaddr(drv->ioctl_sock, bss->ifname,
5040                                        params->bssid))
5041                         goto failed;
5042         }
5043
5044         if (wpa_driver_nl80211_set_mode(bss, IEEE80211_MODE_AP)) {
5045                 wpa_printf(MSG_ERROR, "nl80211: Failed to set interface %s "
5046                            "into AP mode", bss->ifname);
5047                 goto failed;
5048         }
5049
5050         if (params->num_bridge && params->bridge[0] &&
5051             i802_check_bridge(drv, params->bridge[0], params->ifname) < 0)
5052                 goto failed;
5053
5054         if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 1))
5055                 goto failed;
5056
5057         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
5058         if (drv->eapol_sock < 0) {
5059                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
5060                 goto failed;
5061         }
5062
5063         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
5064         {
5065                 printf("Could not register read socket for eapol\n");
5066                 goto failed;
5067         }
5068
5069         if (linux_get_ifhwaddr(drv->ioctl_sock, bss->ifname, params->own_addr))
5070                 goto failed;
5071
5072         return bss;
5073
5074 failed:
5075         nl80211_remove_monitor_interface(drv);
5076         if (drv->ioctl_sock >= 0)
5077                 close(drv->ioctl_sock);
5078
5079         genl_family_put(drv->nl80211);
5080         nl_cache_free(drv->nl_cache);
5081         nl80211_handle_destroy(drv->nl_handle);
5082         nl_cb_put(drv->nl_cb);
5083
5084         os_free(drv);
5085         return NULL;
5086 }
5087
5088
5089 static void i802_deinit(void *priv)
5090 {
5091         wpa_driver_nl80211_deinit(priv);
5092 }
5093
5094 #endif /* HOSTAPD */
5095
5096
5097 static enum nl80211_iftype wpa_driver_nl80211_if_type(
5098         enum wpa_driver_if_type type)
5099 {
5100         switch (type) {
5101         case WPA_IF_STATION:
5102         case WPA_IF_P2P_CLIENT:
5103         case WPA_IF_P2P_GROUP:
5104                 return NL80211_IFTYPE_STATION;
5105         case WPA_IF_AP_VLAN:
5106                 return NL80211_IFTYPE_AP_VLAN;
5107         case WPA_IF_AP_BSS:
5108         case WPA_IF_P2P_GO:
5109                 return NL80211_IFTYPE_AP;
5110         }
5111         return -1;
5112 }
5113
5114
5115 static int wpa_driver_nl80211_if_add(void *priv, enum wpa_driver_if_type type,
5116                                      const char *ifname, const u8 *addr,
5117                                      void *bss_ctx, void **drv_priv,
5118                                      char *force_ifname, u8 *if_addr)
5119 {
5120         struct i802_bss *bss = priv;
5121         struct wpa_driver_nl80211_data *drv = bss->drv;
5122         int ifidx;
5123 #ifdef HOSTAPD
5124         struct i802_bss *new_bss = NULL;
5125
5126         if (type == WPA_IF_AP_BSS) {
5127                 new_bss = os_zalloc(sizeof(*new_bss));
5128                 if (new_bss == NULL)
5129                         return -1;
5130         }
5131 #endif /* HOSTAPD */
5132
5133         if (addr)
5134                 os_memcpy(if_addr, addr, ETH_ALEN);
5135         ifidx = nl80211_create_iface(drv, ifname,
5136                                      wpa_driver_nl80211_if_type(type), addr,
5137                                      0);
5138         if (ifidx < 0) {
5139 #ifdef HOSTAPD
5140                 os_free(new_bss);
5141 #endif /* HOSTAPD */
5142                 return -1;
5143         }
5144
5145         if (!addr &&
5146             linux_get_ifhwaddr(drv->ioctl_sock, bss->ifname, if_addr) < 0)
5147                 return -1;
5148
5149 #ifdef HOSTAPD
5150         if (type == WPA_IF_AP_BSS) {
5151                 if (linux_set_iface_flags(drv->ioctl_sock, ifname, 1)) {
5152                         nl80211_remove_iface(drv, ifidx);
5153                         os_free(new_bss);
5154                         return -1;
5155                 }
5156                 os_strlcpy(new_bss->ifname, ifname, IFNAMSIZ);
5157                 new_bss->ifindex = ifidx;
5158                 new_bss->drv = drv;
5159                 new_bss->next = drv->first_bss.next;
5160                 drv->first_bss.next = new_bss;
5161                 if (drv_priv)
5162                         *drv_priv = new_bss;
5163         }
5164 #endif /* HOSTAPD */
5165
5166         return 0;
5167 }
5168
5169
5170 static int wpa_driver_nl80211_if_remove(void *priv,
5171                                         enum wpa_driver_if_type type,
5172                                         const char *ifname)
5173 {
5174         struct i802_bss *bss = priv;
5175         struct wpa_driver_nl80211_data *drv = bss->drv;
5176         int ifindex = if_nametoindex(ifname);
5177
5178         wpa_printf(MSG_DEBUG, "nl80211: %s(type=%d ifname=%s) ifindex=%d",
5179                    __func__, type, ifname, ifindex);
5180         if (ifindex <= 0)
5181                 return -1;
5182         nl80211_remove_iface(drv, ifindex);
5183
5184 #ifdef HOSTAPD
5185         if (type != WPA_IF_AP_BSS)
5186                 return 0;
5187
5188         if (bss != &drv->first_bss) {
5189                 struct i802_bss *tbss = &drv->first_bss;
5190
5191                 while (tbss) {
5192                         if (tbss->next != bss)
5193                                 continue;
5194
5195                         tbss->next = bss->next;
5196                         os_free(bss);
5197                         break;
5198                 }
5199         }
5200 #endif /* HOSTAPD */
5201
5202         return 0;
5203 }
5204
5205
5206 static int cookie_handler(struct nl_msg *msg, void *arg)
5207 {
5208         struct nlattr *tb[NL80211_ATTR_MAX + 1];
5209         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
5210         u64 *cookie = arg;
5211         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
5212                   genlmsg_attrlen(gnlh, 0), NULL);
5213         if (tb[NL80211_ATTR_COOKIE])
5214                 *cookie = nla_get_u64(tb[NL80211_ATTR_COOKIE]);
5215         return NL_SKIP;
5216 }
5217
5218
5219 static int nl80211_send_frame_cmd(struct wpa_driver_nl80211_data *drv,
5220                                   unsigned int freq, const u8 *buf,
5221                                   size_t buf_len, u64 *cookie_out)
5222 {
5223         struct nl_msg *msg;
5224         u64 cookie;
5225         int ret = -1;
5226
5227         msg = nlmsg_alloc();
5228         if (!msg)
5229                 return -1;
5230
5231         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
5232                     NL80211_CMD_FRAME, 0);
5233
5234         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5235         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
5236         NLA_PUT(msg, NL80211_ATTR_FRAME, buf_len, buf);
5237
5238         cookie = 0;
5239         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
5240         msg = NULL;
5241         if (ret) {
5242                 wpa_printf(MSG_DEBUG, "nl80211: Frame command failed: ret=%d "
5243                            "(%s)", ret, strerror(-ret));
5244                 goto nla_put_failure;
5245         }
5246         wpa_printf(MSG_DEBUG, "nl80211: Frame TX command accepted; "
5247                    "cookie 0x%llx", (long long unsigned int) cookie);
5248
5249         if (cookie_out)
5250                 *cookie_out = cookie;
5251
5252 nla_put_failure:
5253         nlmsg_free(msg);
5254         return ret;
5255 }
5256
5257
5258 static int wpa_driver_nl80211_send_action(void *priv, unsigned int freq,
5259                                           const u8 *dst, const u8 *src,
5260                                           const u8 *bssid,
5261                                           const u8 *data, size_t data_len)
5262 {
5263         struct i802_bss *bss = priv;
5264         struct wpa_driver_nl80211_data *drv = bss->drv;
5265         int ret = -1;
5266         u8 *buf;
5267         struct ieee80211_hdr *hdr;
5268
5269         wpa_printf(MSG_DEBUG, "nl80211: Send Action frame (ifindex=%d)",
5270                    drv->ifindex);
5271
5272         buf = os_zalloc(24 + data_len);
5273         if (buf == NULL)
5274                 return ret;
5275         os_memcpy(buf + 24, data, data_len);
5276         hdr = (struct ieee80211_hdr *) buf;
5277         hdr->frame_control =
5278                 IEEE80211_FC(WLAN_FC_TYPE_MGMT, WLAN_FC_STYPE_ACTION);
5279         os_memcpy(hdr->addr1, dst, ETH_ALEN);
5280         os_memcpy(hdr->addr2, src, ETH_ALEN);
5281         os_memcpy(hdr->addr3, bssid, ETH_ALEN);
5282
5283         if (drv->nlmode == NL80211_IFTYPE_AP)
5284                 ret = wpa_driver_nl80211_send_mlme(priv, buf, 24 + data_len);
5285         else
5286                 ret = nl80211_send_frame_cmd(drv, freq, buf, 24 + data_len,
5287                                              &drv->send_action_cookie);
5288
5289         os_free(buf);
5290         return ret;
5291 }
5292
5293
5294 static int wpa_driver_nl80211_remain_on_channel(void *priv, unsigned int freq,
5295                                                 unsigned int duration)
5296 {
5297         struct i802_bss *bss = priv;
5298         struct wpa_driver_nl80211_data *drv = bss->drv;
5299         struct nl_msg *msg;
5300         int ret;
5301         u64 cookie;
5302
5303         msg = nlmsg_alloc();
5304         if (!msg)
5305                 return -1;
5306
5307         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
5308                     NL80211_CMD_REMAIN_ON_CHANNEL, 0);
5309
5310         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5311         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq);
5312         NLA_PUT_U32(msg, NL80211_ATTR_DURATION, duration);
5313
5314         cookie = 0;
5315         ret = send_and_recv_msgs(drv, msg, cookie_handler, &cookie);
5316         if (ret == 0) {
5317                 wpa_printf(MSG_DEBUG, "nl80211: Remain-on-channel cookie "
5318                            "0x%llx for freq=%u MHz duration=%u",
5319                            (long long unsigned int) cookie, freq, duration);
5320                 drv->remain_on_chan_cookie = cookie;
5321                 return 0;
5322         }
5323         wpa_printf(MSG_DEBUG, "nl80211: Failed to request remain-on-channel "
5324                    "(freq=%d duration=%u): %d (%s)",
5325                    freq, duration, ret, strerror(-ret));
5326 nla_put_failure:
5327         return -1;
5328 }
5329
5330
5331 static int wpa_driver_nl80211_cancel_remain_on_channel(void *priv)
5332 {
5333         struct i802_bss *bss = priv;
5334         struct wpa_driver_nl80211_data *drv = bss->drv;
5335         struct nl_msg *msg;
5336         int ret;
5337
5338         if (!drv->pending_remain_on_chan) {
5339                 wpa_printf(MSG_DEBUG, "nl80211: No pending remain-on-channel "
5340                            "to cancel");
5341                 return -1;
5342         }
5343
5344         wpa_printf(MSG_DEBUG, "nl80211: Cancel remain-on-channel with cookie "
5345                    "0x%llx",
5346                    (long long unsigned int) drv->remain_on_chan_cookie);
5347
5348         msg = nlmsg_alloc();
5349         if (!msg)
5350                 return -1;
5351
5352         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
5353                     NL80211_CMD_CANCEL_REMAIN_ON_CHANNEL, 0);
5354
5355         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
5356         NLA_PUT_U64(msg, NL80211_ATTR_COOKIE, drv->remain_on_chan_cookie);
5357
5358         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5359         if (ret == 0)
5360                 return 0;
5361         wpa_printf(MSG_DEBUG, "nl80211: Failed to cancel remain-on-channel: "
5362                    "%d (%s)", ret, strerror(-ret));
5363 nla_put_failure:
5364         return -1;
5365 }
5366
5367
5368 static int wpa_driver_nl80211_probe_req_report(void *priv, int report)
5369 {
5370         struct i802_bss *bss = priv;
5371         struct wpa_driver_nl80211_data *drv = bss->drv;
5372
5373         if (drv->nlmode != NL80211_IFTYPE_STATION) {
5374                 wpa_printf(MSG_DEBUG, "nl80211: probe_req_report control only "
5375                            "allowed in station mode (iftype=%d)",
5376                            drv->nlmode);
5377                 return -1;
5378         }
5379
5380         if (!report) {
5381                 if (drv->nl_handle_preq) {
5382                         eloop_unregister_read_sock(
5383                                 nl_socket_get_fd(drv->nl_handle_preq));
5384                         nl_cache_free(drv->nl_cache_preq);
5385                         nl80211_handle_destroy(drv->nl_handle_preq);
5386                         drv->nl_handle_preq = NULL;
5387                 }
5388                 return 0;
5389         }
5390
5391         if (drv->nl_handle_preq) {
5392                 wpa_printf(MSG_DEBUG, "nl80211: Probe Request reporting "
5393                            "already on!");
5394                 return 0;
5395         }
5396
5397         drv->nl_handle_preq = nl80211_handle_alloc(drv->nl_cb);
5398         if (drv->nl_handle_preq == NULL) {
5399                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate "
5400                            "netlink callbacks (preq)");
5401                 goto out_err1;
5402         }
5403
5404         if (genl_connect(drv->nl_handle_preq)) {
5405                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to "
5406                            "generic netlink (preq)");
5407                 goto out_err2;
5408                 return -1;
5409         }
5410
5411 #ifdef CONFIG_LIBNL20
5412         if (genl_ctrl_alloc_cache(drv->nl_handle_preq,
5413                                   &drv->nl_cache_preq) < 0) {
5414                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
5415                            "netlink cache (preq)");
5416                 goto out_err2;
5417         }
5418 #else /* CONFIG_LIBNL20 */
5419         drv->nl_cache_preq = genl_ctrl_alloc_cache(drv->nl_handle_preq);
5420         if (drv->nl_cache_preq == NULL) {
5421                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
5422                            "netlink cache (preq)");
5423                 goto out_err2;
5424         }
5425 #endif /* CONFIG_LIBNL20 */
5426
5427         if (nl80211_register_frame(drv, drv->nl_handle_preq,
5428                                    (WLAN_FC_TYPE_MGMT << 2) |
5429                                    (WLAN_FC_STYPE_PROBE_REQ << 4),
5430                                    NULL, 0) < 0) {
5431                 goto out_err3;
5432         }
5433
5434         eloop_register_read_sock(nl_socket_get_fd(drv->nl_handle_preq),
5435                                  wpa_driver_nl80211_event_receive, drv,
5436                                  drv->nl_handle_preq);
5437
5438         return 0;
5439
5440  out_err3:
5441         nl_cache_free(drv->nl_cache_preq);
5442  out_err2:
5443         nl80211_handle_destroy(drv->nl_handle_preq);
5444         drv->nl_handle_preq = NULL;
5445  out_err1:
5446         return -1;
5447 }
5448
5449
5450 static int nl80211_disable_11b_rates(struct wpa_driver_nl80211_data *drv,
5451                                      int ifindex, int disabled)
5452 {
5453         struct nl_msg *msg;
5454         struct nlattr *bands, *band;
5455         int ret;
5456
5457         msg = nlmsg_alloc();
5458         if (!msg)
5459                 return -1;
5460
5461         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
5462                     NL80211_CMD_SET_TX_BITRATE_MASK, 0);
5463         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
5464
5465         bands = nla_nest_start(msg, NL80211_ATTR_TX_RATES);
5466         if (!bands)
5467                 goto nla_put_failure;
5468
5469         /*
5470          * Disable 2 GHz rates 1, 2, 5.5, 11 Mbps by masking out everything
5471          * else apart from 6, 9, 12, 18, 24, 36, 48, 54 Mbps from non-MCS
5472          * rates. All 5 GHz rates are left enabled.
5473          */
5474         band = nla_nest_start(msg, NL80211_BAND_2GHZ);
5475         if (!band)
5476                 goto nla_put_failure;
5477         NLA_PUT(msg, NL80211_TXRATE_LEGACY, 8,
5478                 "\x0c\x12\x18\x24\x30\x48\x60\x6c");
5479         nla_nest_end(msg, band);
5480
5481         nla_nest_end(msg, bands);
5482
5483         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
5484         msg = NULL;
5485         if (ret) {
5486                 wpa_printf(MSG_DEBUG, "nl80211: Set TX rates failed: ret=%d "
5487                            "(%s)", ret, strerror(-ret));
5488         }
5489
5490         return ret;
5491
5492 nla_put_failure:
5493         nlmsg_free(msg);
5494         return -1;
5495 }
5496
5497
5498 static int wpa_driver_nl80211_disable_11b_rates(void *priv, int disabled)
5499 {
5500         struct i802_bss *bss = priv;
5501         struct wpa_driver_nl80211_data *drv = bss->drv;
5502         drv->disable_11b_rates = disabled;
5503         return nl80211_disable_11b_rates(drv, drv->ifindex, disabled);
5504 }
5505
5506
5507 static int wpa_driver_nl80211_deinit_ap(void *priv)
5508 {
5509         struct i802_bss *bss = priv;
5510         struct wpa_driver_nl80211_data *drv = bss->drv;
5511         if (drv->nlmode != NL80211_IFTYPE_AP)
5512                 return -1;
5513         wpa_driver_nl80211_del_beacon(drv);
5514         return wpa_driver_nl80211_set_mode(priv, IEEE80211_MODE_INFRA);
5515 }
5516
5517
5518 static void wpa_driver_nl80211_resume(void *priv)
5519 {
5520         struct i802_bss *bss = priv;
5521         struct wpa_driver_nl80211_data *drv = bss->drv;
5522         if (linux_set_iface_flags(drv->ioctl_sock, bss->ifname, 1)) {
5523                 wpa_printf(MSG_DEBUG, "nl80211: Failed to set interface up on "
5524                            "resume event");
5525         }
5526 }
5527
5528
5529 static int nl80211_send_ft_action(void *priv, u8 action, const u8 *target_ap,
5530                                   const u8 *ies, size_t ies_len)
5531 {
5532         struct i802_bss *bss = priv;
5533         struct wpa_driver_nl80211_data *drv = bss->drv;
5534         int ret;
5535         u8 *data, *pos;
5536         size_t data_len;
5537         u8 own_addr[ETH_ALEN];
5538
5539         if (linux_get_ifhwaddr(drv->ioctl_sock, bss->ifname, own_addr) < 0)
5540                 return -1;
5541
5542         if (action != 1) {
5543                 wpa_printf(MSG_ERROR, "nl80211: Unsupported send_ft_action "
5544                            "action %d", action);
5545                 return -1;
5546         }
5547
5548         /*
5549          * Action frame payload:
5550          * Category[1] = 6 (Fast BSS Transition)
5551          * Action[1] = 1 (Fast BSS Transition Request)
5552          * STA Address
5553          * Target AP Address
5554          * FT IEs
5555          */
5556
5557         data_len = 2 + 2 * ETH_ALEN + ies_len;
5558         data = os_malloc(data_len);
5559         if (data == NULL)
5560                 return -1;
5561         pos = data;
5562         *pos++ = 0x06; /* FT Action category */
5563         *pos++ = action;
5564         os_memcpy(pos, own_addr, ETH_ALEN);
5565         pos += ETH_ALEN;
5566         os_memcpy(pos, target_ap, ETH_ALEN);
5567         pos += ETH_ALEN;
5568         os_memcpy(pos, ies, ies_len);
5569
5570         ret = wpa_driver_nl80211_send_action(bss, drv->assoc_freq, drv->bssid,
5571                                              own_addr, drv->bssid,
5572                                              data, data_len);
5573         os_free(data);
5574
5575         return ret;
5576 }
5577
5578
5579 static int nl80211_signal_monitor(void *priv, int threshold, int hysteresis)
5580 {
5581         struct i802_bss *bss = priv;
5582         struct wpa_driver_nl80211_data *drv = bss->drv;
5583         struct nl_msg *msg, *cqm = NULL;
5584
5585         wpa_printf(MSG_DEBUG, "nl80211: Signal monitor threshold=%d "
5586                    "hysteresis=%d", threshold, hysteresis);
5587
5588         msg = nlmsg_alloc();
5589         if (!msg)
5590                 return -1;
5591
5592         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
5593                     0, NL80211_CMD_SET_CQM, 0);
5594
5595         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, bss->ifindex);
5596
5597         cqm = nlmsg_alloc();
5598         if (cqm == NULL)
5599                 return -1;
5600
5601         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_THOLD, threshold);
5602         NLA_PUT_U32(cqm, NL80211_ATTR_CQM_RSSI_HYST, hysteresis);
5603         nla_put_nested(msg, NL80211_ATTR_CQM, cqm);
5604
5605         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
5606                 return 0;
5607         msg = NULL;
5608
5609 nla_put_failure:
5610         if (cqm)
5611                 nlmsg_free(cqm);
5612         nlmsg_free(msg);
5613         return -1;
5614 }
5615
5616
5617 static int nl80211_send_frame(void *priv, const u8 *data, size_t data_len,
5618                               int encrypt)
5619 {
5620         struct i802_bss *bss = priv;
5621         struct wpa_driver_nl80211_data *drv = bss->drv;
5622         return wpa_driver_nl80211_send_frame(drv, data, data_len, encrypt);
5623 }
5624
5625
5626 static int nl80211_set_intra_bss(void *priv, int enabled)
5627 {
5628         struct i802_bss *bss = priv;
5629         struct wpa_driver_nl80211_data *drv = bss->drv;
5630         struct nl_msg *msg;
5631
5632         msg = nlmsg_alloc();
5633         if (!msg)
5634                 return -ENOMEM;
5635
5636         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
5637                     NL80211_CMD_SET_BSS, 0);
5638
5639         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(bss->ifname));
5640         NLA_PUT_U8(msg, NL80211_ATTR_AP_ISOLATE, !enabled);
5641
5642         return send_and_recv_msgs(drv, msg, NULL, NULL);
5643  nla_put_failure:
5644         return -ENOBUFS;
5645 }
5646
5647
5648 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
5649         .name = "nl80211",
5650         .desc = "Linux nl80211/cfg80211",
5651         .get_bssid = wpa_driver_nl80211_get_bssid,
5652         .get_ssid = wpa_driver_nl80211_get_ssid,
5653         .set_key = wpa_driver_nl80211_set_key,
5654         .scan2 = wpa_driver_nl80211_scan,
5655         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
5656         .deauthenticate = wpa_driver_nl80211_deauthenticate,
5657         .disassociate = wpa_driver_nl80211_disassociate,
5658         .authenticate = wpa_driver_nl80211_authenticate,
5659         .associate = wpa_driver_nl80211_associate,
5660         .init = wpa_driver_nl80211_init,
5661         .deinit = wpa_driver_nl80211_deinit,
5662         .get_capa = wpa_driver_nl80211_get_capa,
5663         .set_operstate = wpa_driver_nl80211_set_operstate,
5664         .set_supp_port = wpa_driver_nl80211_set_supp_port,
5665         .set_country = wpa_driver_nl80211_set_country,
5666         .set_beacon = wpa_driver_nl80211_set_beacon,
5667         .if_add = wpa_driver_nl80211_if_add,
5668         .if_remove = wpa_driver_nl80211_if_remove,
5669         .send_mlme = wpa_driver_nl80211_send_mlme,
5670         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
5671         .sta_add = wpa_driver_nl80211_sta_add,
5672         .sta_remove = wpa_driver_nl80211_sta_remove,
5673         .hapd_send_eapol = wpa_driver_nl80211_hapd_send_eapol,
5674         .sta_set_flags = wpa_driver_nl80211_sta_set_flags,
5675 #ifdef HOSTAPD
5676         .hapd_init = i802_init,
5677         .hapd_deinit = i802_deinit,
5678         .get_seqnum = i802_get_seqnum,
5679         .flush = i802_flush,
5680         .read_sta_data = i802_read_sta_data,
5681         .sta_deauth = i802_sta_deauth,
5682         .sta_disassoc = i802_sta_disassoc,
5683         .get_inact_sec = i802_get_inact_sec,
5684         .sta_clear_stats = i802_sta_clear_stats,
5685         .set_rts = i802_set_rts,
5686         .set_frag = i802_set_frag,
5687         .set_rate_sets = i802_set_rate_sets,
5688         .set_cts_protect = i802_set_cts_protect,
5689         .set_preamble = i802_set_preamble,
5690         .set_short_slot_time = i802_set_short_slot_time,
5691         .set_tx_queue_params = i802_set_tx_queue_params,
5692         .set_sta_vlan = i802_set_sta_vlan,
5693         .set_wds_sta = i802_set_wds_sta,
5694 #endif /* HOSTAPD */
5695         .set_freq = i802_set_freq,
5696         .send_action = wpa_driver_nl80211_send_action,
5697         .remain_on_channel = wpa_driver_nl80211_remain_on_channel,
5698         .cancel_remain_on_channel =
5699         wpa_driver_nl80211_cancel_remain_on_channel,
5700         .probe_req_report = wpa_driver_nl80211_probe_req_report,
5701         .disable_11b_rates = wpa_driver_nl80211_disable_11b_rates,
5702         .deinit_ap = wpa_driver_nl80211_deinit_ap,
5703         .resume = wpa_driver_nl80211_resume,
5704         .send_ft_action = nl80211_send_ft_action,
5705         .signal_monitor = nl80211_signal_monitor,
5706         .send_frame = nl80211_send_frame,
5707         .set_intra_bss = nl80211_set_intra_bss,
5708 };