vcs.maemo.org Git - wpasupplicant/blob - src/drivers/driver_nl80211.c

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