git.maemo.org Git - wpasupplicant/blob - hostapd/driver_nl80211.c

   1 /*
   2  * hostapd / Kernel driver communication via nl80211
   3  * Copyright (c) 2002-2007, Jouni Malinen <j@w1.fi>
   4  * Copyright (c) 2003-2004, Instant802 Networks, Inc.
   5  * Copyright (c) 2005-2006, Devicescape Software, Inc.
   6  * Copyright (c) 2007, Johannes Berg <johannes@sipsolutions.net>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License version 2 as
  10  * published by the Free Software Foundation.
  11  *
  12  * Alternatively, this software may be distributed under the terms of BSD
  13  * license.
  14  *
  15  * See README and COPYING for more details.
  16  */
  17
  18 #include "includes.h"
  19
  20 #include <sys/ioctl.h>
  21 #include <netlink/genl/genl.h>
  22 #include <netlink/genl/family.h>
  23 #include <netlink/genl/ctrl.h>
  24 #include <netlink/msg.h>
  25 #include <netlink/attr.h>
  26 #include "nl80211_copy.h"
  27 #include <net/if.h>
  28 #include <netpacket/packet.h>
  29 #include "wireless_copy.h"
  30 #include <linux/filter.h>
  31 #include <net/if_arp.h>
  32
  33 #include "hostapd.h"
  34 #include "driver.h"
  35 #include "eloop.h"
  36 #include "hw_features.h"
  37 #include "mlme.h"
  38 #include "radiotap.h"
  39 #include "radiotap_iter.h"
  40 #include "ieee802_11_defs.h"
  41 #include "ieee802_11_common.h"
  42
  43 #ifdef CONFIG_LIBNL20
  44 /* libnl 2.0 compatibility code */
  45 #define nl_handle_alloc_cb nl_socket_alloc_cb
  46 #define nl_handle_destroy nl_socket_free
  47 #endif /* CONFIG_LIBNL20 */
  48
  49 static const u8 rfc1042_header[6] = { 0xaa, 0xaa, 0x03, 0x00, 0x00, 0x00 };
  50
  51 enum ieee80211_msg_type {
  52         ieee80211_msg_normal = 0,
  53         ieee80211_msg_tx_callback_ack = 1,
  54         ieee80211_msg_tx_callback_fail = 2,
  55 };
  56
  57 struct i802_driver_data {
  58         struct hostapd_data *hapd;
  59
  60         char iface[IFNAMSIZ + 1];
  61         int bridge;
  62         int ioctl_sock; /* socket for ioctl() use */
  63         int wext_sock; /* socket for wireless events */
  64         int eapol_sock; /* socket for EAPOL frames */
  65         int monitor_sock; /* socket for monitor */
  66         int monitor_ifidx;
  67
  68         int default_if_indices[16];
  69         int *if_indices;
  70         int num_if_indices;
  71
  72         int we_version;
  73         struct nl_handle *nl_handle;
  74         struct nl_cache *nl_cache;
  75         struct nl_cb *nl_cb;
  76         struct genl_family *nl80211;
  77         int dtim_period, beacon_int;
  78         unsigned int beacon_set:1;
  79         unsigned int ieee802_1x_active:1;
  80         unsigned int ht_40mhz_scan:1;
  81
  82         int last_freq;
  83         int last_freq_ht;
  84         struct hostapd_neighbor_bss *neighbors;
  85         size_t num_neighbors;
  86 };
  87
  88
  89 static int i802_sta_deauth(void *priv, const u8 *addr, int reason);
  90 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason);
  91
  92
  93 static void add_ifidx(struct i802_driver_data *drv, int ifidx)
  94 {
  95         int i;
  96         int *old;
  97
  98         for (i = 0; i < drv->num_if_indices; i++) {
  99                 if (drv->if_indices[i] == 0) {
 100                         drv->if_indices[i] = ifidx;
 101                         return;
 102                 }
 103         }
 104
 105         if (drv->if_indices != drv->default_if_indices)
 106                 old = drv->if_indices;
 107         else
 108                 old = NULL;
 109
 110         drv->if_indices = realloc(old,
 111                                   sizeof(int) * (drv->num_if_indices + 1));
 112         if (!drv->if_indices) {
 113                 if (!old)
 114                         drv->if_indices = drv->default_if_indices;
 115                 else
 116                         drv->if_indices = old;
 117                 wpa_printf(MSG_ERROR, "Failed to reallocate memory for "
 118                            "interfaces");
 119                 wpa_printf(MSG_ERROR, "Ignoring EAPOL on interface %d", ifidx);
 120                 return;
 121         }
 122         drv->if_indices[drv->num_if_indices] = ifidx;
 123         drv->num_if_indices++;
 124 }
 125
 126
 127 static void del_ifidx(struct i802_driver_data *drv, int ifidx)
 128 {
 129         int i;
 130
 131         for (i = 0; i < drv->num_if_indices; i++) {
 132                 if (drv->if_indices[i] == ifidx) {
 133                         drv->if_indices[i] = 0;
 134                         break;
 135                 }
 136         }
 137 }
 138
 139
 140 static int have_ifidx(struct i802_driver_data *drv, int ifidx)
 141 {
 142         int i;
 143
 144         if (ifidx == drv->bridge)
 145                 return 1;
 146
 147         for (i = 0; i < drv->num_if_indices; i++)
 148                 if (drv->if_indices[i] == ifidx)
 149                         return 1;
 150
 151         return 0;
 152 }
 153
 154
 155 /* nl80211 code */
 156 static int ack_handler(struct nl_msg *msg, void *arg)
 157 {
 158         int *err = arg;
 159         *err = 0;
 160         return NL_STOP;
 161 }
 162
 163 static int finish_handler(struct nl_msg *msg, void *arg)
 164 {
 165         int *ret = arg;
 166         *ret = 0;
 167         return NL_SKIP;
 168 }
 169
 170 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
 171                          void *arg)
 172 {
 173         int *ret = arg;
 174         *ret = err->error;
 175         return NL_SKIP;
 176 }
 177
 178 static int send_and_recv_msgs(struct i802_driver_data *drv,
 179                               struct nl_msg *msg,
 180                               int (*valid_handler)(struct nl_msg *, void *),
 181                               void *valid_data)
 182 {
 183         struct nl_cb *cb;
 184         int err = -ENOMEM;
 185
 186         cb = nl_cb_clone(drv->nl_cb);
 187         if (!cb)
 188                 goto out;
 189
 190         err = nl_send_auto_complete(drv->nl_handle, msg);
 191         if (err < 0)
 192                 goto out;
 193
 194         err = 1;
 195
 196         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
 197         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
 198         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
 199
 200         if (valid_handler)
 201                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
 202                           valid_handler, valid_data);
 203
 204         while (err > 0)
 205                 nl_recvmsgs(drv->nl_handle, cb);
 206  out:
 207         nl_cb_put(cb);
 208         nlmsg_free(msg);
 209         return err;
 210 }
 211
 212 static int hostapd_set_iface_flags(struct i802_driver_data *drv,
 213                                    const char *ifname, int dev_up)
 214 {
 215         struct ifreq ifr;
 216
 217         if (drv->ioctl_sock < 0)
 218                 return -1;
 219
 220         memset(&ifr, 0, sizeof(ifr));
 221         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 222
 223         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, &ifr) != 0) {
 224                 perror("ioctl[SIOCGIFFLAGS]");
 225                 wpa_printf(MSG_DEBUG, "Could not read interface flags (%s)",
 226                            drv->iface);
 227                 return -1;
 228         }
 229
 230         if (dev_up)
 231                 ifr.ifr_flags |= IFF_UP;
 232         else
 233                 ifr.ifr_flags &= ~IFF_UP;
 234
 235         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, &ifr) != 0) {
 236                 perror("ioctl[SIOCSIFFLAGS]");
 237                 return -1;
 238         }
 239
 240         return 0;
 241 }
 242
 243
 244 static int nl_set_encr(int ifindex, struct i802_driver_data *drv,
 245                        const char *alg, const u8 *addr, int idx, const u8 *key,
 246                        size_t key_len, int txkey)
 247 {
 248         struct nl_msg *msg;
 249         int ret;
 250
 251         msg = nlmsg_alloc();
 252         if (!msg)
 253                 return -ENOMEM;
 254
 255         if (strcmp(alg, "none") == 0) {
 256                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 257                             0, NL80211_CMD_DEL_KEY, 0);
 258         } else {
 259                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 260                             0, NL80211_CMD_NEW_KEY, 0);
 261                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
 262                 if (strcmp(alg, "WEP") == 0) {
 263                         if (key_len == 5)
 264                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
 265                                             0x000FAC01);
 266                         else
 267                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
 268                                             0x000FAC05);
 269                 } else if (strcmp(alg, "TKIP") == 0)
 270                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
 271                 else if (strcmp(alg, "CCMP") == 0)
 272                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
 273                 else if (strcmp(alg, "IGTK") == 0)
 274                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC06);
 275                 else {
 276                         wpa_printf(MSG_ERROR, "%s: Unsupported encryption "
 277                                    "algorithm '%s'", __func__, alg);
 278                         nlmsg_free(msg);
 279                         return -1;
 280                 }
 281         }
 282
 283         if (addr)
 284                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
 285         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
 286         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
 287
 288         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
 289         if (ret == -ENOENT)
 290                 ret = 0;
 291
 292         /*
 293          * If we failed or don't need to set the default TX key (below),
 294          * we're done here.
 295          */
 296         if (ret || !txkey || addr)
 297                 return ret;
 298
 299         msg = nlmsg_alloc();
 300         if (!msg)
 301                 return -ENOMEM;
 302
 303         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 304                     0, NL80211_CMD_SET_KEY, 0);
 305         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
 306         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifindex);
 307         if (strcmp(alg, "IGTK") == 0)
 308                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT_MGMT);
 309         else
 310                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
 311
 312         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
 313         if (ret == -ENOENT)
 314                 ret = 0;
 315         return ret;
 316  nla_put_failure:
 317         return -ENOBUFS;
 318 }
 319
 320
 321 static int i802_set_encryption(const char *iface, void *priv, const char *alg,
 322                                const u8 *addr, int idx, const u8 *key,
 323                                size_t key_len, int txkey)
 324 {
 325         struct i802_driver_data *drv = priv;
 326         int ret;
 327
 328         ret = nl_set_encr(if_nametoindex(iface), drv, alg, addr, idx, key,
 329                           key_len, txkey);
 330         if (ret < 0)
 331                 return ret;
 332
 333         return ret;
 334 }
 335
 336
 337 static inline int min_int(int a, int b)
 338 {
 339         if (a < b)
 340                 return a;
 341         return b;
 342 }
 343
 344
 345 static int get_key_handler(struct nl_msg *msg, void *arg)
 346 {
 347         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 348         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 349
 350         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 351                   genlmsg_attrlen(gnlh, 0), NULL);
 352
 353         /*
 354          * TODO: validate the key index and mac address!
 355          * Otherwise, there's a race condition as soon as
 356          * the kernel starts sending key notifications.
 357          */
 358
 359         if (tb[NL80211_ATTR_KEY_SEQ])
 360                 memcpy(arg, nla_data(tb[NL80211_ATTR_KEY_SEQ]),
 361                        min_int(nla_len(tb[NL80211_ATTR_KEY_SEQ]), 6));
 362         return NL_SKIP;
 363 }
 364
 365
 366 static int i802_get_seqnum(const char *iface, void *priv, const u8 *addr,
 367                            int idx, u8 *seq)
 368 {
 369         struct i802_driver_data *drv = priv;
 370         struct nl_msg *msg;
 371
 372         msg = nlmsg_alloc();
 373         if (!msg)
 374                 return -ENOMEM;
 375
 376         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 377                     0, NL80211_CMD_GET_KEY, 0);
 378
 379         if (addr)
 380                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
 381         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, idx);
 382         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
 383
 384         memset(seq, 0, 6);
 385
 386         return send_and_recv_msgs(drv, msg, get_key_handler, seq);
 387  nla_put_failure:
 388         return -ENOBUFS;
 389 }
 390
 391
 392 static int i802_set_rate_sets(void *priv, int *supp_rates, int *basic_rates,
 393                               int mode)
 394 {
 395         struct i802_driver_data *drv = priv;
 396         struct nl_msg *msg;
 397         u8 rates[NL80211_MAX_SUPP_RATES];
 398         u8 rates_len = 0;
 399         int i;
 400
 401         msg = nlmsg_alloc();
 402         if (!msg)
 403                 return -ENOMEM;
 404
 405         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
 406                     NL80211_CMD_SET_BSS, 0);
 407
 408         for (i = 0; i < NL80211_MAX_SUPP_RATES && basic_rates[i] >= 0; i++)
 409                 rates[rates_len++] = basic_rates[i] / 5;
 410
 411         NLA_PUT(msg, NL80211_ATTR_BSS_BASIC_RATES, rates_len, rates);
 412
 413         /* TODO: multi-BSS support */
 414         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
 415
 416         return send_and_recv_msgs(drv, msg, NULL, NULL);
 417  nla_put_failure:
 418         return -ENOBUFS;
 419 }
 420
 421
 422 static int i802_send_frame(void *priv, const void *data, size_t len,
 423                            int encrypt, int flags)
 424 {
 425         __u8 rtap_hdr[] = {
 426                 0x00, 0x00, /* radiotap version */
 427                 0x0e, 0x00, /* radiotap length */
 428                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
 429                 IEEE80211_RADIOTAP_F_FRAG, /* F_FRAG (fragment if required) */
 430                 0x00,       /* padding */
 431                 0x00, 0x00, /* RX and TX flags to indicate that */
 432                 0x00, 0x00, /* this is the injected frame directly */
 433         };
 434         struct i802_driver_data *drv = priv;
 435         struct iovec iov[2] = {
 436                 {
 437                         .iov_base = &rtap_hdr,
 438                         .iov_len = sizeof(rtap_hdr),
 439                 },
 440                 {
 441                         .iov_base = (void*)data,
 442                         .iov_len = len,
 443                 }
 444         };
 445         struct msghdr msg = {
 446                 .msg_name = NULL,
 447                 .msg_namelen = 0,
 448                 .msg_iov = iov,
 449                 .msg_iovlen = 2,
 450                 .msg_control = NULL,
 451                 .msg_controllen = 0,
 452                 .msg_flags = 0,
 453         };
 454
 455         if (encrypt)
 456                 rtap_hdr[8] |= IEEE80211_RADIOTAP_F_WEP;
 457
 458         return sendmsg(drv->monitor_sock, &msg, flags);
 459 }
 460
 461 static int i802_send_mgmt_frame(void *priv, const void *data, size_t len,
 462                                 int flags)
 463 {
 464         struct ieee80211_mgmt *mgmt;
 465         int do_not_encrypt = 0;
 466         u16 fc;
 467
 468         mgmt = (struct ieee80211_mgmt *) data;
 469         fc = le_to_host16(mgmt->frame_control);
 470
 471         if (WLAN_FC_GET_TYPE(fc) == WLAN_FC_TYPE_MGMT &&
 472             WLAN_FC_GET_STYPE(fc) == WLAN_FC_STYPE_AUTH) {
 473                 /*
 474                  * Only one of the authentication frame types is encrypted.
 475                  * In order for static WEP encryption to work properly (i.e.,
 476                  * to not encrypt the frame), we need to tell mac80211 about
 477                  * the frames that must not be encrypted.
 478                  */
 479                 u16 auth_alg = le_to_host16(mgmt->u.auth.auth_alg);
 480                 u16 auth_trans = le_to_host16(mgmt->u.auth.auth_transaction);
 481                 if (auth_alg == WLAN_AUTH_OPEN ||
 482                     (auth_alg == WLAN_AUTH_SHARED_KEY && auth_trans != 3))
 483                         do_not_encrypt = 1;
 484         }
 485
 486         return i802_send_frame(priv, data, len, !do_not_encrypt, flags);
 487 }
 488
 489 /* Set kernel driver on given frequency (MHz) */
 490 static int i802_set_freq2(void *priv, struct hostapd_freq_params *freq)
 491 {
 492         struct i802_driver_data *drv = priv;
 493         struct nl_msg *msg;
 494
 495         msg = nlmsg_alloc();
 496         if (!msg)
 497                 return -1;
 498
 499         drv->last_freq = freq->freq;
 500         drv->last_freq_ht = freq->ht_enabled;
 501
 502         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
 503                     NL80211_CMD_SET_WIPHY, 0);
 504
 505         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
 506         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_FREQ, freq->freq);
 507         if (freq->ht_enabled) {
 508                 switch (freq->sec_channel_offset) {
 509                 case -1:
 510                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
 511                                     NL80211_CHAN_HT40MINUS);
 512                         break;
 513                 case 1:
 514                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
 515                                     NL80211_CHAN_HT40PLUS);
 516                         break;
 517                 default:
 518                         NLA_PUT_U32(msg, NL80211_ATTR_WIPHY_CHANNEL_TYPE,
 519                                     NL80211_CHAN_HT20);
 520                         break;
 521                 }
 522         }
 523
 524         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
 525                 return 0;
 526  nla_put_failure:
 527         return -1;
 528 }
 529
 530
 531 static int i802_set_rts(void *priv, int rts)
 532 {
 533         struct i802_driver_data *drv = priv;
 534         struct iwreq iwr;
 535
 536         memset(&iwr, 0, sizeof(iwr));
 537         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 538         iwr.u.rts.value = rts;
 539         iwr.u.rts.fixed = 1;
 540
 541         if (ioctl(drv->ioctl_sock, SIOCSIWRTS, &iwr) < 0) {
 542                 perror("ioctl[SIOCSIWRTS]");
 543                 return -1;
 544         }
 545
 546         return 0;
 547 }
 548
 549
 550 static int i802_get_rts(void *priv, int *rts)
 551 {
 552         struct i802_driver_data *drv = priv;
 553         struct iwreq iwr;
 554
 555         memset(&iwr, 0, sizeof(iwr));
 556         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 557
 558         if (ioctl(drv->ioctl_sock, SIOCGIWRTS, &iwr) < 0) {
 559                 perror("ioctl[SIOCGIWRTS]");
 560                 return -1;
 561         }
 562
 563         *rts = iwr.u.rts.value;
 564
 565         return 0;
 566 }
 567
 568
 569 static int i802_set_frag(void *priv, int frag)
 570 {
 571         struct i802_driver_data *drv = priv;
 572         struct iwreq iwr;
 573
 574         memset(&iwr, 0, sizeof(iwr));
 575         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 576         iwr.u.frag.value = frag;
 577         iwr.u.frag.fixed = 1;
 578
 579         if (ioctl(drv->ioctl_sock, SIOCSIWFRAG, &iwr) < 0) {
 580                 perror("ioctl[SIOCSIWFRAG]");
 581                 return -1;
 582         }
 583
 584         return 0;
 585 }
 586
 587
 588 static int i802_get_frag(void *priv, int *frag)
 589 {
 590         struct i802_driver_data *drv = priv;
 591         struct iwreq iwr;
 592
 593         memset(&iwr, 0, sizeof(iwr));
 594         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 595
 596         if (ioctl(drv->ioctl_sock, SIOCGIWFRAG, &iwr) < 0) {
 597                 perror("ioctl[SIOCGIWFRAG]");
 598                 return -1;
 599         }
 600
 601         *frag = iwr.u.frag.value;
 602
 603         return 0;
 604 }
 605
 606
 607 static int i802_set_retry(void *priv, int short_retry, int long_retry)
 608 {
 609         struct i802_driver_data *drv = priv;
 610         struct iwreq iwr;
 611
 612         memset(&iwr, 0, sizeof(iwr));
 613         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 614
 615         iwr.u.retry.value = short_retry;
 616         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
 617         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
 618                 perror("ioctl[SIOCSIWRETRY(short)]");
 619                 return -1;
 620         }
 621
 622         iwr.u.retry.value = long_retry;
 623         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
 624         if (ioctl(drv->ioctl_sock, SIOCSIWRETRY, &iwr) < 0) {
 625                 perror("ioctl[SIOCSIWRETRY(long)]");
 626                 return -1;
 627         }
 628
 629         return 0;
 630 }
 631
 632
 633 static int i802_get_retry(void *priv, int *short_retry, int *long_retry)
 634 {
 635         struct i802_driver_data *drv = priv;
 636         struct iwreq iwr;
 637
 638         memset(&iwr, 0, sizeof(iwr));
 639         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
 640
 641         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MIN;
 642         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
 643                 perror("ioctl[SIOCGIWFRAG(short)]");
 644                 return -1;
 645         }
 646         *short_retry = iwr.u.retry.value;
 647
 648         iwr.u.retry.flags = IW_RETRY_LIMIT | IW_RETRY_MAX;
 649         if (ioctl(drv->ioctl_sock, SIOCGIWRETRY, &iwr) < 0) {
 650                 perror("ioctl[SIOCGIWFRAG(long)]");
 651                 return -1;
 652         }
 653         *long_retry = iwr.u.retry.value;
 654
 655         return 0;
 656 }
 657
 658
 659 static int i802_flush(void *priv)
 660 {
 661         struct i802_driver_data *drv = priv;
 662         struct nl_msg *msg;
 663
 664         msg = nlmsg_alloc();
 665         if (!msg)
 666                 return -1;
 667
 668         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 669                     0, NL80211_CMD_DEL_STATION, 0);
 670
 671         /*
 672          * XXX: FIX! this needs to flush all VLANs too
 673          */
 674         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
 675                     if_nametoindex(drv->iface));
 676
 677         return send_and_recv_msgs(drv, msg, NULL, NULL);
 678  nla_put_failure:
 679         return -ENOBUFS;
 680 }
 681
 682
 683 static int get_sta_handler(struct nl_msg *msg, void *arg)
 684 {
 685         struct nlattr *tb[NL80211_ATTR_MAX + 1];
 686         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 687         struct hostap_sta_driver_data *data = arg;
 688         struct nlattr *stats[NL80211_STA_INFO_MAX + 1];
 689         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
 690                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 },
 691                 [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 },
 692                 [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 },
 693         };
 694
 695         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 696                   genlmsg_attrlen(gnlh, 0), NULL);
 697
 698         /*
 699          * TODO: validate the interface and mac address!
 700          * Otherwise, there's a race condition as soon as
 701          * the kernel starts sending station notifications.
 702          */
 703
 704         if (!tb[NL80211_ATTR_STA_INFO]) {
 705                 wpa_printf(MSG_DEBUG, "sta stats missing!");
 706                 return NL_SKIP;
 707         }
 708         if (nla_parse_nested(stats, NL80211_STA_INFO_MAX,
 709                              tb[NL80211_ATTR_STA_INFO],
 710                              stats_policy)) {
 711                 wpa_printf(MSG_DEBUG, "failed to parse nested attributes!");
 712                 return NL_SKIP;
 713         }
 714
 715         if (stats[NL80211_STA_INFO_INACTIVE_TIME])
 716                 data->inactive_msec =
 717                         nla_get_u32(stats[NL80211_STA_INFO_INACTIVE_TIME]);
 718         if (stats[NL80211_STA_INFO_RX_BYTES])
 719                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_RX_BYTES]);
 720         if (stats[NL80211_STA_INFO_TX_BYTES])
 721                 data->rx_bytes = nla_get_u32(stats[NL80211_STA_INFO_TX_BYTES]);
 722
 723         return NL_SKIP;
 724 }
 725
 726 static int i802_read_sta_data(void *priv, struct hostap_sta_driver_data *data,
 727                               const u8 *addr)
 728 {
 729         struct i802_driver_data *drv = priv;
 730         struct nl_msg *msg;
 731
 732         msg = nlmsg_alloc();
 733         if (!msg)
 734                 return -ENOMEM;
 735
 736         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 737                     0, NL80211_CMD_GET_STATION, 0);
 738
 739         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
 740         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
 741
 742         return send_and_recv_msgs(drv, msg, get_sta_handler, data);
 743  nla_put_failure:
 744         return -ENOBUFS;
 745 }
 746
 747
 748 static int i802_send_eapol(void *priv, const u8 *addr, const u8 *data,
 749                            size_t data_len, int encrypt, const u8 *own_addr)
 750 {
 751         struct i802_driver_data *drv = priv;
 752         struct ieee80211_hdr *hdr;
 753         size_t len;
 754         u8 *pos;
 755         int res;
 756 #if 0 /* FIX */
 757         int qos = sta->flags & WLAN_STA_WME;
 758 #else
 759         int qos = 0;
 760 #endif
 761
 762         len = sizeof(*hdr) + (qos ? 2 : 0) + sizeof(rfc1042_header) + 2 +
 763                 data_len;
 764         hdr = os_zalloc(len);
 765         if (hdr == NULL) {
 766                 printf("malloc() failed for i802_send_data(len=%lu)\n",
 767                        (unsigned long) len);
 768                 return -1;
 769         }
 770
 771         hdr->frame_control =
 772                 IEEE80211_FC(WLAN_FC_TYPE_DATA, WLAN_FC_STYPE_DATA);
 773         hdr->frame_control |= host_to_le16(WLAN_FC_FROMDS);
 774         if (encrypt)
 775                 hdr->frame_control |= host_to_le16(WLAN_FC_ISWEP);
 776 #if 0 /* To be enabled if qos determination is added above */
 777         if (qos) {
 778                 hdr->frame_control |=
 779                         host_to_le16(WLAN_FC_STYPE_QOS_DATA << 4);
 780         }
 781 #endif
 782
 783         memcpy(hdr->IEEE80211_DA_FROMDS, addr, ETH_ALEN);
 784         memcpy(hdr->IEEE80211_BSSID_FROMDS, own_addr, ETH_ALEN);
 785         memcpy(hdr->IEEE80211_SA_FROMDS, own_addr, ETH_ALEN);
 786         pos = (u8 *) (hdr + 1);
 787
 788 #if 0 /* To be enabled if qos determination is added above */
 789         if (qos) {
 790                 /* add an empty QoS header if needed */
 791                 pos[0] = 0;
 792                 pos[1] = 0;
 793                 pos += 2;
 794         }
 795 #endif
 796
 797         memcpy(pos, rfc1042_header, sizeof(rfc1042_header));
 798         pos += sizeof(rfc1042_header);
 799         WPA_PUT_BE16(pos, ETH_P_PAE);
 800         pos += 2;
 801         memcpy(pos, data, data_len);
 802
 803         res = i802_send_frame(drv, (u8 *) hdr, len, encrypt, 0);
 804         free(hdr);
 805
 806         if (res < 0) {
 807                 perror("i802_send_eapol: send");
 808                 printf("i802_send_eapol - packet len: %lu - failed\n",
 809                        (unsigned long) len);
 810         }
 811
 812         return res;
 813 }
 814
 815
 816 static int i802_sta_add2(const char *ifname, void *priv,
 817                          struct hostapd_sta_add_params *params)
 818 {
 819         struct i802_driver_data *drv = priv;
 820         struct nl_msg *msg;
 821         int ret = -ENOBUFS;
 822
 823         msg = nlmsg_alloc();
 824         if (!msg)
 825                 return -ENOMEM;
 826
 827         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 828                     0, NL80211_CMD_NEW_STATION, 0);
 829
 830         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
 831                     if_nametoindex(drv->iface));
 832         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, params->addr);
 833         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, params->aid);
 834         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, params->supp_rates_len,
 835                 params->supp_rates);
 836         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL,
 837                     params->listen_interval);
 838
 839 #ifdef CONFIG_IEEE80211N
 840         if (params->ht_capabilities) {
 841                 NLA_PUT(msg, NL80211_ATTR_HT_CAPABILITY,
 842                         params->ht_capabilities->length,
 843                         &params->ht_capabilities->data);
 844         }
 845 #endif /* CONFIG_IEEE80211N */
 846
 847         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
 848         if (ret == -EEXIST)
 849                 ret = 0;
 850  nla_put_failure:
 851         return ret;
 852 }
 853
 854
 855 static int i802_sta_remove(void *priv, const u8 *addr)
 856 {
 857         struct i802_driver_data *drv = priv;
 858         struct nl_msg *msg;
 859         int ret;
 860
 861         msg = nlmsg_alloc();
 862         if (!msg)
 863                 return -ENOMEM;
 864
 865         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 866                     0, NL80211_CMD_DEL_STATION, 0);
 867
 868         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
 869                     if_nametoindex(drv->iface));
 870         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
 871
 872         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
 873         if (ret == -ENOENT)
 874                 return 0;
 875         return ret;
 876  nla_put_failure:
 877         return -ENOBUFS;
 878 }
 879
 880
 881 static int i802_sta_set_flags(void *priv, const u8 *addr,
 882                               int total_flags, int flags_or, int flags_and)
 883 {
 884         struct i802_driver_data *drv = priv;
 885         struct nl_msg *msg, *flags = NULL;
 886
 887         msg = nlmsg_alloc();
 888         if (!msg)
 889                 return -ENOMEM;
 890
 891         flags = nlmsg_alloc();
 892         if (!flags) {
 893                 nlmsg_free(msg);
 894                 return -ENOMEM;
 895         }
 896
 897         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 898                     0, NL80211_CMD_SET_STATION, 0);
 899
 900         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
 901                     if_nametoindex(drv->iface));
 902         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
 903
 904         if (total_flags & WLAN_STA_AUTHORIZED || !drv->ieee802_1x_active)
 905                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_AUTHORIZED);
 906
 907         if (total_flags & WLAN_STA_WME)
 908                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_WME);
 909
 910         if (total_flags & WLAN_STA_SHORT_PREAMBLE)
 911                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_SHORT_PREAMBLE);
 912
 913         if (total_flags & WLAN_STA_MFP)
 914                 NLA_PUT_FLAG(flags, NL80211_STA_FLAG_MFP);
 915
 916         if (nla_put_nested(msg, NL80211_ATTR_STA_FLAGS, flags))
 917                 goto nla_put_failure;
 918
 919         nlmsg_free(flags);
 920
 921         return send_and_recv_msgs(drv, msg, NULL, NULL);
 922  nla_put_failure:
 923         nlmsg_free(flags);
 924         return -ENOBUFS;
 925 }
 926
 927
 928 static int i802_set_tx_queue_params(void *priv, int queue, int aifs,
 929                                     int cw_min, int cw_max, int burst_time)
 930 {
 931         struct i802_driver_data *drv = priv;
 932         struct nl_msg *msg;
 933         struct nlattr *txq, *params;
 934
 935         msg = nlmsg_alloc();
 936         if (!msg)
 937                 return -1;
 938
 939         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 940                     0, NL80211_CMD_SET_WIPHY, 0);
 941
 942         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
 943
 944         txq = nla_nest_start(msg, NL80211_ATTR_WIPHY_TXQ_PARAMS);
 945         if (!txq)
 946                 goto nla_put_failure;
 947
 948         /* We are only sending parameters for a single TXQ at a time */
 949         params = nla_nest_start(msg, 1);
 950         if (!params)
 951                 goto nla_put_failure;
 952
 953         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_QUEUE, queue);
 954         /* Burst time is configured in units of 0.1 msec and TXOP parameter in
 955          * 32 usec, so need to convert the value here. */
 956         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_TXOP, (burst_time * 100 + 16) / 32);
 957         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMIN, cw_min);
 958         NLA_PUT_U16(msg, NL80211_TXQ_ATTR_CWMAX, cw_max);
 959         NLA_PUT_U8(msg, NL80211_TXQ_ATTR_AIFS, aifs);
 960
 961         nla_nest_end(msg, params);
 962
 963         nla_nest_end(msg, txq);
 964
 965         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
 966                 return 0;
 967  nla_put_failure:
 968         return -1;
 969 }
 970
 971
 972 static void nl80211_remove_iface(struct i802_driver_data *drv, int ifidx)
 973 {
 974         struct nl_msg *msg;
 975
 976         /* stop listening for EAPOL on this interface */
 977         del_ifidx(drv, ifidx);
 978
 979         msg = nlmsg_alloc();
 980         if (!msg)
 981                 goto nla_put_failure;
 982
 983         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 984                     0, NL80211_CMD_DEL_INTERFACE, 0);
 985         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
 986
 987         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
 988                 return;
 989  nla_put_failure:
 990         printf("Failed to remove interface.\n");
 991 }
 992
 993
 994 static int nl80211_create_iface(struct i802_driver_data *drv,
 995                                 const char *ifname,
 996                                 enum nl80211_iftype iftype,
 997                                 const u8 *addr)
 998 {
 999         struct nl_msg *msg, *flags = NULL;
1000         int ifidx;
1001         struct ifreq ifreq;
1002         struct iwreq iwr;
1003         int ret = -ENOBUFS;
1004
1005         msg = nlmsg_alloc();
1006         if (!msg)
1007                 return -1;
1008
1009         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1010                     0, NL80211_CMD_NEW_INTERFACE, 0);
1011         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1012         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
1013         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
1014
1015         if (iftype == NL80211_IFTYPE_MONITOR) {
1016                 int err;
1017
1018                 flags = nlmsg_alloc();
1019                 if (!flags)
1020                         goto nla_put_failure;
1021
1022                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1023
1024                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1025
1026                 nlmsg_free(flags);
1027
1028                 if (err)
1029                         goto nla_put_failure;
1030         }
1031
1032         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1033         if (ret) {
1034  nla_put_failure:
1035                 printf("Failed to create interface %s.\n", ifname);
1036                 return ret;
1037         }
1038
1039         ifidx = if_nametoindex(ifname);
1040
1041         if (ifidx <= 0)
1042                 return -1;
1043
1044         /* start listening for EAPOL on this interface */
1045         add_ifidx(drv, ifidx);
1046
1047         if (addr) {
1048                 switch (iftype) {
1049                 case NL80211_IFTYPE_AP:
1050                         os_strlcpy(ifreq.ifr_name, ifname, IFNAMSIZ);
1051                         memcpy(ifreq.ifr_hwaddr.sa_data, addr, ETH_ALEN);
1052                         ifreq.ifr_hwaddr.sa_family = ARPHRD_ETHER;
1053
1054                         if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifreq)) {
1055                                 nl80211_remove_iface(drv, ifidx);
1056                                 return -1;
1057                         }
1058                         break;
1059                 case NL80211_IFTYPE_WDS:
1060                         memset(&iwr, 0, sizeof(iwr));
1061                         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1062                         iwr.u.addr.sa_family = ARPHRD_ETHER;
1063                         memcpy(iwr.u.addr.sa_data, addr, ETH_ALEN);
1064                         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr))
1065                                 return -1;
1066                         break;
1067                 default:
1068                         /* nothing */
1069                         break;
1070                 }
1071         }
1072
1073         return ifidx;
1074 }
1075
1076
1077 static int i802_bss_add(void *priv, const char *ifname, const u8 *bssid)
1078 {
1079         int ifidx;
1080
1081         /*
1082          * The kernel supports that when the low-level driver does,
1083          * but we currently don't because we need per-BSS data that
1084          * currently we can't handle easily.
1085          */
1086         return -1;
1087
1088         ifidx = nl80211_create_iface(priv, ifname, NL80211_IFTYPE_AP, bssid);
1089         if (ifidx < 0)
1090                 return -1;
1091         if (hostapd_set_iface_flags(priv, ifname, 1)) {
1092                 nl80211_remove_iface(priv, ifidx);
1093                 return -1;
1094         }
1095         return 0;
1096 }
1097
1098
1099 static int i802_bss_remove(void *priv, const char *ifname)
1100 {
1101         nl80211_remove_iface(priv, if_nametoindex(ifname));
1102         return 0;
1103 }
1104
1105
1106 static int i802_set_beacon(const char *iface, void *priv,
1107                            u8 *head, size_t head_len,
1108                            u8 *tail, size_t tail_len)
1109 {
1110         struct i802_driver_data *drv = priv;
1111         struct nl_msg *msg;
1112         u8 cmd = NL80211_CMD_NEW_BEACON;
1113         int ret;
1114
1115         msg = nlmsg_alloc();
1116         if (!msg)
1117                 return -ENOMEM;
1118
1119         if (drv->beacon_set)
1120                 cmd = NL80211_CMD_SET_BEACON;
1121
1122         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1123                     0, cmd, 0);
1124         NLA_PUT(msg, NL80211_ATTR_BEACON_HEAD, head_len, head);
1125         NLA_PUT(msg, NL80211_ATTR_BEACON_TAIL, tail_len, tail);
1126         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
1127         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, drv->beacon_int);
1128
1129         if (!drv->dtim_period)
1130                 drv->dtim_period = 2;
1131         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, drv->dtim_period);
1132
1133         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1134         if (!ret)
1135                 drv->beacon_set = 1;
1136         return ret;
1137  nla_put_failure:
1138         return -ENOBUFS;
1139 }
1140
1141
1142 static int i802_del_beacon(struct i802_driver_data *drv)
1143 {
1144         struct nl_msg *msg;
1145
1146         msg = nlmsg_alloc();
1147         if (!msg)
1148                 return -ENOMEM;
1149
1150         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1151                     0, NL80211_CMD_DEL_BEACON, 0);
1152         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1153
1154         return send_and_recv_msgs(drv, msg, NULL, NULL);
1155  nla_put_failure:
1156         return -ENOBUFS;
1157 }
1158
1159
1160 static int i802_set_ieee8021x(const char *ifname, void *priv, int enabled)
1161 {
1162         struct i802_driver_data *drv = priv;
1163
1164         /*
1165          * FIXME: This needs to be per interface (BSS)
1166          */
1167         drv->ieee802_1x_active = enabled;
1168         return 0;
1169 }
1170
1171
1172 static int i802_set_privacy(const char *ifname, void *priv, int enabled)
1173 {
1174         struct i802_driver_data *drv = priv;
1175         struct iwreq iwr;
1176
1177         memset(&iwr, 0, sizeof(iwr));
1178
1179         os_strlcpy(iwr.ifr_name, ifname, IFNAMSIZ);
1180         iwr.u.param.flags = IW_AUTH_PRIVACY_INVOKED;
1181         iwr.u.param.value = enabled;
1182
1183         ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr);
1184
1185         /* ignore errors, the kernel/driver might not care */
1186         return 0;
1187 }
1188
1189
1190 static int i802_set_internal_bridge(void *priv, int value)
1191 {
1192         return -1;
1193 }
1194
1195
1196 static int i802_set_beacon_int(void *priv, int value)
1197 {
1198         struct i802_driver_data *drv = priv;
1199         struct nl_msg *msg;
1200
1201         drv->beacon_int = value;
1202
1203         if (!drv->beacon_set)
1204                 return 0;
1205
1206         msg = nlmsg_alloc();
1207         if (!msg)
1208                 return -ENOMEM;
1209
1210         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1211                     0, NL80211_CMD_SET_BEACON, 0);
1212         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1213
1214         NLA_PUT_U32(msg, NL80211_ATTR_BEACON_INTERVAL, value);
1215
1216         return send_and_recv_msgs(drv, msg, NULL, NULL);
1217  nla_put_failure:
1218         return -ENOBUFS;
1219 }
1220
1221
1222 static int i802_set_dtim_period(const char *iface, void *priv, int value)
1223 {
1224         struct i802_driver_data *drv = priv;
1225         struct nl_msg *msg;
1226
1227         msg = nlmsg_alloc();
1228         if (!msg)
1229                 return -ENOMEM;
1230
1231         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1232                     0, NL80211_CMD_SET_BEACON, 0);
1233         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(iface));
1234
1235         drv->dtim_period = value;
1236         NLA_PUT_U32(msg, NL80211_ATTR_DTIM_PERIOD, drv->dtim_period);
1237
1238         return send_and_recv_msgs(drv, msg, NULL, NULL);
1239  nla_put_failure:
1240         return -ENOBUFS;
1241 }
1242
1243
1244 static int i802_set_bss(void *priv, int cts, int preamble, int slot)
1245 {
1246         struct i802_driver_data *drv = priv;
1247         struct nl_msg *msg;
1248
1249         msg = nlmsg_alloc();
1250         if (!msg)
1251                 return -ENOMEM;
1252
1253         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1254                     NL80211_CMD_SET_BSS, 0);
1255
1256         if (cts >= 0)
1257                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_CTS_PROT, cts);
1258         if (preamble >= 0)
1259                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_PREAMBLE, preamble);
1260         if (slot >= 0)
1261                 NLA_PUT_U8(msg, NL80211_ATTR_BSS_SHORT_SLOT_TIME, slot);
1262
1263         /* TODO: multi-BSS support */
1264         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1265
1266         return send_and_recv_msgs(drv, msg, NULL, NULL);
1267  nla_put_failure:
1268         return -ENOBUFS;
1269 }
1270
1271
1272 static int i802_set_cts_protect(void *priv, int value)
1273 {
1274         return i802_set_bss(priv, value, -1, -1);
1275 }
1276
1277
1278 static int i802_set_preamble(void *priv, int value)
1279 {
1280         return i802_set_bss(priv, -1, value, -1);
1281 }
1282
1283
1284 static int i802_set_short_slot_time(void *priv, int value)
1285 {
1286         return i802_set_bss(priv, -1, -1, value);
1287 }
1288
1289
1290 static enum nl80211_iftype i802_if_type(enum hostapd_driver_if_type type)
1291 {
1292         switch (type) {
1293         case HOSTAPD_IF_VLAN:
1294                 return NL80211_IFTYPE_AP_VLAN;
1295         case HOSTAPD_IF_WDS:
1296                 return NL80211_IFTYPE_WDS;
1297         }
1298         return -1;
1299 }
1300
1301
1302 static int i802_if_add(const char *iface, void *priv,
1303                        enum hostapd_driver_if_type type, char *ifname,
1304                        const u8 *addr)
1305 {
1306         if (nl80211_create_iface(priv, ifname, i802_if_type(type), addr) < 0)
1307                 return -1;
1308         return 0;
1309 }
1310
1311
1312 static int i802_if_update(void *priv, enum hostapd_driver_if_type type,
1313                           char *ifname, const u8 *addr)
1314 {
1315         /* unused at the moment */
1316         return -1;
1317 }
1318
1319
1320 static int i802_if_remove(void *priv, enum hostapd_driver_if_type type,
1321                           const char *ifname, const u8 *addr)
1322 {
1323         nl80211_remove_iface(priv, if_nametoindex(ifname));
1324         return 0;
1325 }
1326
1327
1328 struct phy_info_arg {
1329         u16 *num_modes;
1330         struct hostapd_hw_modes *modes;
1331 };
1332
1333 static int phy_info_handler(struct nl_msg *msg, void *arg)
1334 {
1335         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
1336         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
1337         struct phy_info_arg *phy_info = arg;
1338
1339         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
1340
1341         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
1342         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1343                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
1344                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
1345                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1346                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
1347                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
1348                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32 },
1349         };
1350
1351         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
1352         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
1353                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
1354                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] = { .type = NLA_FLAG },
1355         };
1356
1357         struct nlattr *nl_band;
1358         struct nlattr *nl_freq;
1359         struct nlattr *nl_rate;
1360         int rem_band, rem_freq, rem_rate;
1361         struct hostapd_hw_modes *mode;
1362         int idx, mode_is_set;
1363
1364         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
1365                   genlmsg_attrlen(gnlh, 0), NULL);
1366
1367         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
1368                 return NL_SKIP;
1369
1370         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS], rem_band) {
1371                 mode = realloc(phy_info->modes, (*phy_info->num_modes + 1) * sizeof(*mode));
1372                 if (!mode)
1373                         return NL_SKIP;
1374                 phy_info->modes = mode;
1375
1376                 mode_is_set = 0;
1377
1378                 mode = &phy_info->modes[*(phy_info->num_modes)];
1379                 memset(mode, 0, sizeof(*mode));
1380                 *(phy_info->num_modes) += 1;
1381
1382                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
1383                           nla_len(nl_band), NULL);
1384
1385                 if (tb_band[NL80211_BAND_ATTR_HT_CAPA]) {
1386                         mode->ht_capab = nla_get_u16(
1387                                 tb_band[NL80211_BAND_ATTR_HT_CAPA]);
1388                 }
1389
1390                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1391                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1392                                   nla_len(nl_freq), freq_policy);
1393                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1394                                 continue;
1395                         mode->num_channels++;
1396                 }
1397
1398                 mode->channels = calloc(mode->num_channels, sizeof(struct hostapd_channel_data));
1399                 if (!mode->channels)
1400                         return NL_SKIP;
1401
1402                 idx = 0;
1403
1404                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS], rem_freq) {
1405                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX, nla_data(nl_freq),
1406                                   nla_len(nl_freq), freq_policy);
1407                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
1408                                 continue;
1409
1410                         mode->channels[idx].freq = nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
1411                         mode->channels[idx].flag = 0;
1412
1413                         if (!mode_is_set) {
1414                                 /* crude heuristic */
1415                                 if (mode->channels[idx].freq < 4000)
1416                                         mode->mode = HOSTAPD_MODE_IEEE80211B;
1417                                 else
1418                                         mode->mode = HOSTAPD_MODE_IEEE80211A;
1419                                 mode_is_set = 1;
1420                         }
1421
1422                         /* crude heuristic */
1423                         if (mode->channels[idx].freq < 4000)
1424                                 if (mode->channels[idx].freq == 2848)
1425                                         mode->channels[idx].chan = 14;
1426                                 else
1427                                         mode->channels[idx].chan = (mode->channels[idx].freq - 2407) / 5;
1428                         else
1429                                 mode->channels[idx].chan = mode->channels[idx].freq/5 - 1000;
1430
1431                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1432                                 mode->channels[idx].flag |=
1433                                         HOSTAPD_CHAN_DISABLED;
1434                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
1435                                 mode->channels[idx].flag |=
1436                                         HOSTAPD_CHAN_PASSIVE_SCAN;
1437                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
1438                                 mode->channels[idx].flag |=
1439                                         HOSTAPD_CHAN_NO_IBSS;
1440                         if (tb_freq[NL80211_FREQUENCY_ATTR_RADAR])
1441                                 mode->channels[idx].flag |=
1442                                         HOSTAPD_CHAN_RADAR;
1443
1444                         if (tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] &&
1445                             !tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
1446                                 mode->channels[idx].max_tx_power =
1447                                         nla_get_u32(tb_freq[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]) / 100;
1448
1449                         idx++;
1450                 }
1451
1452                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1453                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1454                                   nla_len(nl_rate), rate_policy);
1455                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1456                                 continue;
1457                         mode->num_rates++;
1458                 }
1459
1460                 mode->rates = calloc(mode->num_rates, sizeof(struct hostapd_rate_data));
1461                 if (!mode->rates)
1462                         return NL_SKIP;
1463
1464                 idx = 0;
1465
1466                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES], rem_rate) {
1467                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX, nla_data(nl_rate),
1468                                   nla_len(nl_rate), rate_policy);
1469                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
1470                                 continue;
1471                         mode->rates[idx].rate = nla_get_u32(tb_rate[NL80211_BITRATE_ATTR_RATE]);
1472
1473                         /* crude heuristic */
1474                         if (mode->mode == HOSTAPD_MODE_IEEE80211B &&
1475                             mode->rates[idx].rate > 200)
1476                                 mode->mode = HOSTAPD_MODE_IEEE80211G;
1477
1478                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
1479                                 mode->rates[idx].flags |= HOSTAPD_RATE_PREAMBLE2;
1480
1481                         idx++;
1482                 }
1483         }
1484
1485         return NL_SKIP;
1486 }
1487
1488 static struct hostapd_hw_modes *i802_add_11b(struct hostapd_hw_modes *modes,
1489                                              u16 *num_modes)
1490 {
1491         u16 m;
1492         struct hostapd_hw_modes *mode11g = NULL, *nmodes, *mode;
1493         int i, mode11g_idx = -1;
1494
1495         /* If only 802.11g mode is included, use it to construct matching
1496          * 802.11b mode data. */
1497
1498         for (m = 0; m < *num_modes; m++) {
1499                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211B)
1500                         return modes; /* 802.11b already included */
1501                 if (modes[m].mode == HOSTAPD_MODE_IEEE80211G)
1502                         mode11g_idx = m;
1503         }
1504
1505         if (mode11g_idx < 0)
1506                 return modes; /* 2.4 GHz band not supported at all */
1507
1508         nmodes = os_realloc(modes, (*num_modes + 1) * sizeof(*nmodes));
1509         if (nmodes == NULL)
1510                 return modes; /* Could not add 802.11b mode */
1511
1512         mode = &nmodes[*num_modes];
1513         os_memset(mode, 0, sizeof(*mode));
1514         (*num_modes)++;
1515         modes = nmodes;
1516
1517         mode->mode = HOSTAPD_MODE_IEEE80211B;
1518
1519         mode11g = &modes[mode11g_idx];
1520         mode->num_channels = mode11g->num_channels;
1521         mode->channels = os_malloc(mode11g->num_channels *
1522                                    sizeof(struct hostapd_channel_data));
1523         if (mode->channels == NULL) {
1524                 (*num_modes)--;
1525                 return modes; /* Could not add 802.11b mode */
1526         }
1527         os_memcpy(mode->channels, mode11g->channels,
1528                   mode11g->num_channels * sizeof(struct hostapd_channel_data));
1529
1530         mode->num_rates = 0;
1531         mode->rates = os_malloc(4 * sizeof(struct hostapd_rate_data));
1532         if (mode->rates == NULL) {
1533                 os_free(mode->channels);
1534                 (*num_modes)--;
1535                 return modes; /* Could not add 802.11b mode */
1536         }
1537
1538         for (i = 0; i < mode11g->num_rates; i++) {
1539                 if (mode11g->rates[i].rate > 110 ||
1540                     mode11g->rates[i].flags &
1541                     (HOSTAPD_RATE_ERP | HOSTAPD_RATE_OFDM))
1542                         continue;
1543                 mode->rates[mode->num_rates] = mode11g->rates[i];
1544                 mode->num_rates++;
1545                 if (mode->num_rates == 4)
1546                         break;
1547         }
1548
1549         if (mode->num_rates == 0) {
1550                 os_free(mode->channels);
1551                 os_free(mode->rates);
1552                 (*num_modes)--;
1553                 return modes; /* No 802.11b rates */
1554         }
1555
1556         wpa_printf(MSG_DEBUG, "nl80211: Added 802.11b mode based on 802.11g "
1557                    "information");
1558
1559         return modes;
1560 }
1561
1562 static struct hostapd_hw_modes *i802_get_hw_feature_data(void *priv,
1563                                                          u16 *num_modes,
1564                                                          u16 *flags)
1565 {
1566         struct i802_driver_data *drv = priv;
1567         struct nl_msg *msg;
1568         struct phy_info_arg result = {
1569                 .num_modes = num_modes,
1570                 .modes = NULL,
1571         };
1572
1573         *num_modes = 0;
1574         *flags = 0;
1575
1576         msg = nlmsg_alloc();
1577         if (!msg)
1578                 return NULL;
1579
1580         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1581                     0, NL80211_CMD_GET_WIPHY, 0);
1582
1583         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->iface));
1584
1585         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
1586                 return i802_add_11b(result.modes, num_modes);
1587  nla_put_failure:
1588         return NULL;
1589 }
1590
1591
1592 static int i802_set_sta_vlan(void *priv, const u8 *addr,
1593                              const char *ifname, int vlan_id)
1594 {
1595         struct i802_driver_data *drv = priv;
1596         struct nl_msg *msg;
1597
1598         msg = nlmsg_alloc();
1599         if (!msg)
1600                 return -ENOMEM;
1601
1602         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1603                     0, NL80211_CMD_SET_STATION, 0);
1604
1605         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1606                     if_nametoindex(drv->iface));
1607         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
1608         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
1609                     if_nametoindex(ifname));
1610
1611         return send_and_recv_msgs(drv, msg, NULL, NULL);
1612  nla_put_failure:
1613         return -ENOBUFS;
1614 }
1615
1616
1617 static int i802_set_country(void *priv, const char *country)
1618 {
1619         struct i802_driver_data *drv = priv;
1620         struct nl_msg *msg;
1621         char alpha2[3];
1622
1623         msg = nlmsg_alloc();
1624         if (!msg)
1625                 return -ENOMEM;
1626
1627         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1628                     0, NL80211_CMD_REQ_SET_REG, 0);
1629
1630         alpha2[0] = country[0];
1631         alpha2[1] = country[1];
1632         alpha2[2] = '\0';
1633         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
1634
1635         return send_and_recv_msgs(drv, msg, NULL, NULL);
1636  nla_put_failure:
1637         return -ENOBUFS;
1638 }
1639
1640
1641 static void handle_tx_callback(struct hostapd_data *hapd, u8 *buf, size_t len,
1642                                int ok)
1643 {
1644         struct ieee80211_hdr *hdr;
1645         u16 fc, type, stype;
1646
1647         hdr = (struct ieee80211_hdr *) buf;
1648         fc = le_to_host16(hdr->frame_control);
1649
1650         type = WLAN_FC_GET_TYPE(fc);
1651         stype = WLAN_FC_GET_STYPE(fc);
1652
1653         switch (type) {
1654         case WLAN_FC_TYPE_MGMT:
1655                 wpa_printf(MSG_DEBUG, "MGMT (TX callback) %s",
1656                            ok ? "ACK" : "fail");
1657                 hostapd_mgmt_tx_cb(hapd, buf, len, stype, ok);
1658                 break;
1659         case WLAN_FC_TYPE_CTRL:
1660                 wpa_printf(MSG_DEBUG, "CTRL (TX callback) %s",
1661                            ok ? "ACK" : "fail");
1662                 break;
1663         case WLAN_FC_TYPE_DATA:
1664                 hostapd_tx_status(hapd, hdr->addr1, buf, len, ok);
1665                 break;
1666         default:
1667                 printf("unknown TX callback frame type %d\n", type);
1668                 break;
1669         }
1670 }
1671
1672
1673 static void handle_frame(struct i802_driver_data *drv,
1674                          struct hostapd_iface *iface, u8 *buf, size_t len,
1675                          struct hostapd_frame_info *hfi,
1676                          enum ieee80211_msg_type msg_type)
1677 {
1678         struct ieee80211_hdr *hdr;
1679         u16 fc, type, stype;
1680         size_t data_len = len;
1681         struct hostapd_data *hapd = NULL;
1682         int broadcast_bssid = 0;
1683         size_t i;
1684         u8 *bssid;
1685
1686         /*
1687          * PS-Poll frames are 16 bytes. All other frames are
1688          * 24 bytes or longer.
1689          */
1690         if (len < 16)
1691                 return;
1692
1693         hdr = (struct ieee80211_hdr *) buf;
1694         fc = le_to_host16(hdr->frame_control);
1695
1696         type = WLAN_FC_GET_TYPE(fc);
1697         stype = WLAN_FC_GET_STYPE(fc);
1698
1699         switch (type) {
1700         case WLAN_FC_TYPE_DATA:
1701                 if (len < 24)
1702                         return;
1703                 switch (fc & (WLAN_FC_FROMDS | WLAN_FC_TODS)) {
1704                 case WLAN_FC_TODS:
1705                         bssid = hdr->addr1;
1706                         break;
1707                 case WLAN_FC_FROMDS:
1708                         bssid = hdr->addr2;
1709                         break;
1710                 default:
1711                         /* discard */
1712                         return;
1713                 }
1714                 break;
1715         case WLAN_FC_TYPE_CTRL:
1716                 /* discard non-ps-poll frames */
1717                 if (stype != WLAN_FC_STYPE_PSPOLL)
1718                         return;
1719                 bssid = hdr->addr1;
1720                 break;
1721         case WLAN_FC_TYPE_MGMT:
1722                 bssid = hdr->addr3;
1723                 break;
1724         default:
1725                 /* discard */
1726                 return;
1727         }
1728
1729         /* find interface frame belongs to */
1730         for (i = 0; i < iface->num_bss; i++) {
1731                 if (memcmp(bssid, iface->bss[i]->own_addr, ETH_ALEN) == 0) {
1732                         hapd = iface->bss[i];
1733                         break;
1734                 }
1735         }
1736
1737         if (hapd == NULL) {
1738                 hapd = iface->bss[0];
1739
1740                 if (bssid[0] != 0xff || bssid[1] != 0xff ||
1741                     bssid[2] != 0xff || bssid[3] != 0xff ||
1742                     bssid[4] != 0xff || bssid[5] != 0xff) {
1743                         /*
1744                          * Unknown BSSID - drop frame if this is not from
1745                          * passive scanning or a beacon (at least ProbeReq
1746                          * frames to other APs may be allowed through RX
1747                          * filtering in the wlan hw/driver)
1748                          */
1749                         if ((type != WLAN_FC_TYPE_MGMT ||
1750                              stype != WLAN_FC_STYPE_BEACON))
1751                                 return;
1752                 } else
1753                         broadcast_bssid = 1;
1754         }
1755
1756         switch (msg_type) {
1757         case ieee80211_msg_normal:
1758                 /* continue processing */
1759                 break;
1760         case ieee80211_msg_tx_callback_ack:
1761                 handle_tx_callback(hapd, buf, data_len, 1);
1762                 return;
1763         case ieee80211_msg_tx_callback_fail:
1764                 handle_tx_callback(hapd, buf, data_len, 0);
1765                 return;
1766         }
1767
1768         switch (type) {
1769         case WLAN_FC_TYPE_MGMT:
1770                 if (stype != WLAN_FC_STYPE_BEACON &&
1771                     stype != WLAN_FC_STYPE_PROBE_REQ)
1772                         wpa_printf(MSG_MSGDUMP, "MGMT");
1773                 if (broadcast_bssid) {
1774                         for (i = 0; i < iface->num_bss; i++)
1775                                 hostapd_mgmt_rx(iface->bss[i], buf, data_len,
1776                                                 stype, hfi);
1777                 } else
1778                         hostapd_mgmt_rx(hapd, buf, data_len, stype, hfi);
1779                 break;
1780         case WLAN_FC_TYPE_CTRL:
1781                 /* can only get here with PS-Poll frames */
1782                 wpa_printf(MSG_DEBUG, "CTRL");
1783                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
1784                 break;
1785         case WLAN_FC_TYPE_DATA:
1786                 hostapd_rx_from_unknown_sta(drv->hapd, hdr->addr2);
1787                 break;
1788         }
1789 }
1790
1791
1792 static void handle_eapol(int sock, void *eloop_ctx, void *sock_ctx)
1793 {
1794         struct i802_driver_data *drv = eloop_ctx;
1795         struct hostapd_data *hapd = drv->hapd;
1796         struct sockaddr_ll lladdr;
1797         unsigned char buf[3000];
1798         int len;
1799         socklen_t fromlen = sizeof(lladdr);
1800
1801         len = recvfrom(sock, buf, sizeof(buf), 0,
1802                        (struct sockaddr *)&lladdr, &fromlen);
1803         if (len < 0) {
1804                 perror("recv");
1805                 return;
1806         }
1807
1808         if (have_ifidx(drv, lladdr.sll_ifindex))
1809                 hostapd_eapol_receive(hapd, lladdr.sll_addr, buf, len);
1810 }
1811
1812
1813 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1814 {
1815         struct i802_driver_data *drv = eloop_ctx;
1816         int len;
1817         unsigned char buf[3000];
1818         struct hostapd_data *hapd = drv->hapd;
1819         struct ieee80211_radiotap_iterator iter;
1820         int ret;
1821         struct hostapd_frame_info hfi;
1822         int injected = 0, failed = 0, msg_type, rxflags = 0;
1823
1824         len = recv(sock, buf, sizeof(buf), 0);
1825         if (len < 0) {
1826                 perror("recv");
1827                 return;
1828         }
1829
1830         if (ieee80211_radiotap_iterator_init(&iter, (void*)buf, len)) {
1831                 printf("received invalid radiotap frame\n");
1832                 return;
1833         }
1834
1835         memset(&hfi, 0, sizeof(hfi));
1836
1837         while (1) {
1838                 ret = ieee80211_radiotap_iterator_next(&iter);
1839                 if (ret == -ENOENT)
1840                         break;
1841                 if (ret) {
1842                         printf("received invalid radiotap frame (%d)\n", ret);
1843                         return;
1844                 }
1845                 switch (iter.this_arg_index) {
1846                 case IEEE80211_RADIOTAP_FLAGS:
1847                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1848                                 len -= 4;
1849                         break;
1850                 case IEEE80211_RADIOTAP_RX_FLAGS:
1851                         rxflags = 1;
1852                         break;
1853                 case IEEE80211_RADIOTAP_TX_FLAGS:
1854                         injected = 1;
1855                         failed = le_to_host16((*(uint16_t *) iter.this_arg)) &
1856                                         IEEE80211_RADIOTAP_F_TX_FAIL;
1857                         break;
1858                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1859                         break;
1860                 case IEEE80211_RADIOTAP_CHANNEL:
1861                         /* TODO convert from freq/flags to channel number
1862                         hfi.channel = XXX;
1863                         hfi.phytype = XXX;
1864                          */
1865                         break;
1866                 case IEEE80211_RADIOTAP_RATE:
1867                         hfi.datarate = *iter.this_arg * 5;
1868                         break;
1869                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1870                         hfi.ssi_signal = *iter.this_arg;
1871                         break;
1872                 }
1873         }
1874
1875         if (rxflags && injected)
1876                 return;
1877
1878         if (!injected)
1879                 msg_type = ieee80211_msg_normal;
1880         else if (failed)
1881                 msg_type = ieee80211_msg_tx_callback_fail;
1882         else
1883                 msg_type = ieee80211_msg_tx_callback_ack;
1884
1885         handle_frame(drv, hapd->iface, buf + iter.max_length,
1886                      len - iter.max_length, &hfi, msg_type);
1887 }
1888
1889
1890 /*
1891  * we post-process the filter code later and rewrite
1892  * this to the offset to the last instruction
1893  */
1894 #define PASS    0xFF
1895 #define FAIL    0xFE
1896
1897 static struct sock_filter msock_filter_insns[] = {
1898         /*
1899          * do a little-endian load of the radiotap length field
1900          */
1901         /* load lower byte into A */
1902         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 2),
1903         /* put it into X (== index register) */
1904         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1905         /* load upper byte into A */
1906         BPF_STMT(BPF_LD  | BPF_B | BPF_ABS, 3),
1907         /* left-shift it by 8 */
1908         BPF_STMT(BPF_ALU | BPF_LSH | BPF_K, 8),
1909         /* or with X */
1910         BPF_STMT(BPF_ALU | BPF_OR | BPF_X, 0),
1911         /* put result into X */
1912         BPF_STMT(BPF_MISC| BPF_TAX, 0),
1913
1914         /*
1915          * Allow management frames through, this also gives us those
1916          * management frames that we sent ourselves with status
1917          */
1918         /* load the lower byte of the IEEE 802.11 frame control field */
1919         BPF_STMT(BPF_LD  | BPF_B | BPF_IND, 0),
1920         /* mask off frame type and version */
1921         BPF_STMT(BPF_ALU | BPF_AND | BPF_K, 0xF),
1922         /* accept frame if it's both 0, fall through otherwise */
1923         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0, PASS, 0),
1924
1925         /*
1926          * TODO: add a bit to radiotap RX flags that indicates
1927          * that the sending station is not associated, then
1928          * add a filter here that filters on our DA and that flag
1929          * to allow us to deauth frames to that bad station.
1930          *
1931          * Not a regression -- we didn't do it before either.
1932          */
1933
1934 #if 0
1935         /*
1936          * drop non-data frames, WDS frames
1937          */
1938         /* load the lower byte of the frame control field */
1939         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1940         /* mask off QoS bit */
1941         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x0c),
1942         /* drop non-data frames */
1943         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 8, 0, FAIL),
1944         /* load the upper byte of the frame control field */
1945         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1946         /* mask off toDS/fromDS */
1947         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x03),
1948         /* drop WDS frames */
1949         BPF_JUMP(BPF_JMP  | BPF_JEQ | BPF_K, 3, FAIL, 0),
1950 #endif
1951
1952         /*
1953          * add header length to index
1954          */
1955         /* load the lower byte of the frame control field */
1956         BPF_STMT(BPF_LD   | BPF_B | BPF_IND, 0),
1957         /* mask off QoS bit */
1958         BPF_STMT(BPF_ALU  | BPF_AND | BPF_K, 0x80),
1959         /* right shift it by 6 to give 0 or 2 */
1960         BPF_STMT(BPF_ALU  | BPF_RSH | BPF_K, 6),
1961         /* add data frame header length */
1962         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_K, 24),
1963         /* add index, was start of 802.11 header */
1964         BPF_STMT(BPF_ALU  | BPF_ADD | BPF_X, 0),
1965         /* move to index, now start of LL header */
1966         BPF_STMT(BPF_MISC | BPF_TAX, 0),
1967
1968         /*
1969          * Accept empty data frames, we use those for
1970          * polling activity.
1971          */
1972         BPF_STMT(BPF_LD  | BPF_W | BPF_LEN, 0),
1973         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_X, 0, PASS, 0),
1974
1975         /*
1976          * Accept EAPOL frames
1977          */
1978         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 0),
1979         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0xAAAA0300, 0, FAIL),
1980         BPF_STMT(BPF_LD  | BPF_W | BPF_IND, 4),
1981         BPF_JUMP(BPF_JMP | BPF_JEQ | BPF_K, 0x0000888E, PASS, FAIL),
1982
1983         /* keep these last two statements or change the code below */
1984         /* return 0 == "DROP" */
1985         BPF_STMT(BPF_RET | BPF_K, 0),
1986         /* return ~0 == "keep all" */
1987         BPF_STMT(BPF_RET | BPF_K, ~0),
1988 };
1989
1990 static struct sock_fprog msock_filter = {
1991         .len = sizeof(msock_filter_insns)/sizeof(msock_filter_insns[0]),
1992         .filter = msock_filter_insns,
1993 };
1994
1995
1996 static int add_monitor_filter(int s)
1997 {
1998         int idx;
1999
2000         /* rewrite all PASS/FAIL jump offsets */
2001         for (idx = 0; idx < msock_filter.len; idx++) {
2002                 struct sock_filter *insn = &msock_filter_insns[idx];
2003
2004                 if (BPF_CLASS(insn->code) == BPF_JMP) {
2005                         if (insn->code == (BPF_JMP|BPF_JA)) {
2006                                 if (insn->k == PASS)
2007                                         insn->k = msock_filter.len - idx - 2;
2008                                 else if (insn->k == FAIL)
2009                                         insn->k = msock_filter.len - idx - 3;
2010                         }
2011
2012                         if (insn->jt == PASS)
2013                                 insn->jt = msock_filter.len - idx - 2;
2014                         else if (insn->jt == FAIL)
2015                                 insn->jt = msock_filter.len - idx - 3;
2016
2017                         if (insn->jf == PASS)
2018                                 insn->jf = msock_filter.len - idx - 2;
2019                         else if (insn->jf == FAIL)
2020                                 insn->jf = msock_filter.len - idx - 3;
2021                 }
2022         }
2023
2024         if (setsockopt(s, SOL_SOCKET, SO_ATTACH_FILTER,
2025                        &msock_filter, sizeof(msock_filter))) {
2026                 perror("SO_ATTACH_FILTER");
2027                 return -1;
2028         }
2029
2030         return 0;
2031 }
2032
2033
2034 static int nl80211_create_monitor_interface(struct i802_driver_data *drv)
2035 {
2036         char buf[IFNAMSIZ];
2037         struct sockaddr_ll ll;
2038         int optval;
2039         socklen_t optlen;
2040
2041         snprintf(buf, IFNAMSIZ, "mon.%s", drv->iface);
2042         buf[IFNAMSIZ - 1] = '\0';
2043
2044         drv->monitor_ifidx =
2045                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR, NULL);
2046
2047         if (drv->monitor_ifidx < 0)
2048                 return -1;
2049
2050         if (hostapd_set_iface_flags(drv, buf, 1))
2051                 goto error;
2052
2053         memset(&ll, 0, sizeof(ll));
2054         ll.sll_family = AF_PACKET;
2055         ll.sll_ifindex = drv->monitor_ifidx;
2056         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
2057         if (drv->monitor_sock < 0) {
2058                 perror("socket[PF_PACKET,SOCK_RAW]");
2059                 goto error;
2060         }
2061
2062         if (add_monitor_filter(drv->monitor_sock)) {
2063                 wpa_printf(MSG_INFO, "Failed to set socket filter for monitor "
2064                            "interface; do filtering in user space");
2065                 /* This works, but will cost in performance. */
2066         }
2067
2068         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
2069                  sizeof(ll)) < 0) {
2070                 perror("monitor socket bind");
2071                 goto error;
2072         }
2073
2074         optlen = sizeof(optval);
2075         optval = 20;
2076         if (setsockopt
2077             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
2078                 perror("Failed to set socket priority");
2079                 goto error;
2080         }
2081
2082         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
2083                                      drv, NULL)) {
2084                 printf("Could not register monitor read socket\n");
2085                 goto error;
2086         }
2087
2088         return 0;
2089  error:
2090         nl80211_remove_iface(drv, drv->monitor_ifidx);
2091         return -1;
2092 }
2093
2094
2095 static int nl80211_set_mode(struct i802_driver_data *drv, const char *ifname,
2096                             int mode)
2097 {
2098         struct nl_msg *msg;
2099         int ret = -ENOBUFS;
2100
2101         msg = nlmsg_alloc();
2102         if (!msg)
2103                 return -ENOMEM;
2104
2105         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2106                     0, NL80211_CMD_SET_INTERFACE, 0);
2107         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX,
2108                     if_nametoindex(ifname));
2109         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, mode);
2110
2111         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2112         if (!ret)
2113                 return 0;
2114  nla_put_failure:
2115         wpa_printf(MSG_ERROR, "Failed to set interface %s to master "
2116                    "mode.", ifname);
2117         return ret;
2118 }
2119
2120
2121 #ifdef CONFIG_IEEE80211N
2122 static void i802_add_neighbor(struct i802_driver_data *drv, u8 *bssid,
2123                               int freq, u8 *ie, size_t ie_len)
2124 {
2125         struct ieee802_11_elems elems;
2126         int ht, pri_chan = 0, sec_chan = 0;
2127         struct ieee80211_ht_operation *oper;
2128         struct hostapd_neighbor_bss *nnei;
2129
2130         ieee802_11_parse_elems(ie, ie_len, &elems, 0);
2131         ht = elems.ht_capabilities || elems.ht_operation;
2132         if (elems.ht_operation && elems.ht_operation_len >= sizeof(*oper)) {
2133                 oper = (struct ieee80211_ht_operation *) elems.ht_operation;
2134                 pri_chan = oper->control_chan;
2135                 if (oper->ht_param & HT_INFO_HT_PARAM_REC_TRANS_CHNL_WIDTH) {
2136                         if (oper->ht_param &
2137                             HT_INFO_HT_PARAM_SECONDARY_CHNL_ABOVE)
2138                                 sec_chan = pri_chan + 4;
2139                         else if (oper->ht_param &
2140                             HT_INFO_HT_PARAM_SECONDARY_CHNL_BELOW)
2141                                 sec_chan = pri_chan - 4;
2142                 }
2143         }
2144
2145         wpa_printf(MSG_DEBUG, "nl80211: Neighboring BSS - bssid=" MACSTR
2146                    " freq=%d MHz HT=%d pri_chan=%d sec_chan=%d",
2147                    MAC2STR(bssid), freq, ht, pri_chan, sec_chan);
2148
2149         nnei = os_realloc(drv->neighbors, (drv->num_neighbors + 1) *
2150                           sizeof(struct hostapd_neighbor_bss));
2151         if (nnei == NULL)
2152                 return;
2153         drv->neighbors = nnei;
2154         nnei = &nnei[drv->num_neighbors];
2155         os_memcpy(nnei->bssid, bssid, ETH_ALEN);
2156         nnei->freq = freq;
2157         nnei->ht = !!ht;
2158         nnei->pri_chan = pri_chan;
2159         nnei->sec_chan = sec_chan;
2160         drv->num_neighbors++;
2161 }
2162
2163
2164 static int i802_get_scan_freq(struct iw_event *iwe, int *freq)
2165 {
2166         int divi = 1000000, i;
2167
2168         if (iwe->u.freq.e == 0) {
2169                 /*
2170                  * Some drivers do not report frequency, but a channel.
2171                  * Try to map this to frequency by assuming they are using
2172                  * IEEE 802.11b/g.  But don't overwrite a previously parsed
2173                  * frequency if the driver sends both frequency and channel,
2174                  * since the driver may be sending an A-band channel that we
2175                  * don't handle here.
2176                  */
2177
2178                 if (*freq)
2179                         return 0;
2180
2181                 if (iwe->u.freq.m >= 1 && iwe->u.freq.m <= 13) {
2182                         *freq = 2407 + 5 * iwe->u.freq.m;
2183                         return 0;
2184                 } else if (iwe->u.freq.m == 14) {
2185                         *freq = 2484;
2186                         return 0;
2187                 }
2188         }
2189
2190         if (iwe->u.freq.e > 6) {
2191                 wpa_printf(MSG_DEBUG, "Invalid freq in scan results: "
2192                            "m=%d e=%d", iwe->u.freq.m, iwe->u.freq.e);
2193                 return -1;
2194         }
2195
2196         for (i = 0; i < iwe->u.freq.e; i++)
2197                 divi /= 10;
2198         *freq = iwe->u.freq.m / divi;
2199         return 0;
2200 }
2201
2202
2203 static int i802_parse_scan(struct i802_driver_data *drv, u8 *res_buf,
2204                            size_t len)
2205 {
2206         size_t ap_num = 0;
2207         int first;
2208         struct iw_event iwe_buf, *iwe = &iwe_buf;
2209         char *pos, *end, *custom;
2210         u8 bssid[ETH_ALEN];
2211         int freq = 0;
2212         u8 *ie = NULL;
2213         size_t ie_len = 0;
2214
2215         ap_num = 0;
2216         first = 1;
2217
2218         pos = (char *) res_buf;
2219         end = (char *) res_buf + len;
2220
2221         while (pos + IW_EV_LCP_LEN <= end) {
2222                 /* Event data may be unaligned, so make a local, aligned copy
2223                  * before processing. */
2224                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2225                 if (iwe->len <= IW_EV_LCP_LEN)
2226                         break;
2227
2228                 custom = pos + IW_EV_POINT_LEN;
2229                 if (iwe->cmd == IWEVGENIE) {
2230                         /* WE-19 removed the pointer from struct iw_point */
2231                         char *dpos = (char *) &iwe_buf.u.data.length;
2232                         int dlen = dpos - (char *) &iwe_buf;
2233                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
2234                                   sizeof(struct iw_event) - dlen);
2235                 } else {
2236                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
2237                         custom += IW_EV_POINT_OFF;
2238                 }
2239
2240                 switch (iwe->cmd) {
2241                 case SIOCGIWAP:
2242                         if (!first)
2243                                 i802_add_neighbor(drv, bssid, freq, ie,
2244                                                   ie_len);
2245                         first = 0;
2246                         os_memcpy(bssid, iwe->u.ap_addr.sa_data, ETH_ALEN);
2247                         freq = 0;
2248                         ie = NULL;
2249                         ie_len = 0;
2250                         break;
2251                 case SIOCGIWFREQ:
2252                         i802_get_scan_freq(iwe, &freq);
2253                         break;
2254                 case IWEVGENIE:
2255                         if (custom + iwe->u.data.length > end) {
2256                                 wpa_printf(MSG_ERROR, "IWEVGENIE overflow");
2257                                 return -1;
2258                         }
2259                         ie = (u8 *) custom;
2260                         ie_len = iwe->u.data.length;
2261                         break;
2262                 }
2263
2264                 pos += iwe->len;
2265         }
2266
2267         if (!first)
2268                 i802_add_neighbor(drv, bssid, freq, ie, ie_len);
2269
2270         return 0;
2271 }
2272
2273
2274 static int i802_get_ht_scan_res(struct i802_driver_data *drv)
2275 {
2276         struct iwreq iwr;
2277         u8 *res_buf;
2278         size_t res_buf_len;
2279         int res;
2280
2281         res_buf_len = IW_SCAN_MAX_DATA;
2282         for (;;) {
2283                 res_buf = os_malloc(res_buf_len);
2284                 if (res_buf == NULL)
2285                         return -1;
2286                 os_memset(&iwr, 0, sizeof(iwr));
2287                 os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2288                 iwr.u.data.pointer = res_buf;
2289                 iwr.u.data.length = res_buf_len;
2290
2291                 if (ioctl(drv->ioctl_sock, SIOCGIWSCAN, &iwr) == 0)
2292                         break;
2293
2294                 if (errno == E2BIG && res_buf_len < 100000) {
2295                         os_free(res_buf);
2296                         res_buf = NULL;
2297                         res_buf_len *= 2;
2298                         wpa_printf(MSG_DEBUG, "Scan results did not fit - "
2299                                    "trying larger buffer (%lu bytes)",
2300                                    (unsigned long) res_buf_len);
2301                 } else {
2302                         perror("ioctl[SIOCGIWSCAN]");
2303                         os_free(res_buf);
2304                         return -1;
2305                 }
2306         }
2307
2308         if (iwr.u.data.length > res_buf_len) {
2309                 os_free(res_buf);
2310                 return -1;
2311         }
2312
2313         res = i802_parse_scan(drv, res_buf, iwr.u.data.length);
2314         os_free(res_buf);
2315
2316         return res;
2317 }
2318
2319
2320 static int i802_is_event_wireless_scan_complete(char *data, int len)
2321 {
2322         struct iw_event iwe_buf, *iwe = &iwe_buf;
2323         char *pos, *end;
2324
2325         pos = data;
2326         end = data + len;
2327
2328         while (pos + IW_EV_LCP_LEN <= end) {
2329                 /* Event data may be unaligned, so make a local, aligned copy
2330                  * before processing. */
2331                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2332                 if (iwe->cmd == SIOCGIWSCAN)
2333                         return 1;
2334
2335                 pos += iwe->len;
2336         }
2337
2338         return 0;
2339 }
2340
2341
2342 static int i802_is_rtm_scan_complete(int ifindex, struct nlmsghdr *h, int len)
2343 {
2344         struct ifinfomsg *ifi;
2345         int attrlen, _nlmsg_len, rta_len;
2346         struct rtattr *attr;
2347
2348         if (len < (int) sizeof(*ifi))
2349                 return 0;
2350
2351         ifi = NLMSG_DATA(h);
2352
2353         if (ifindex != ifi->ifi_index)
2354                 return 0; /* event for foreign ifindex */
2355
2356         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
2357
2358         attrlen = h->nlmsg_len - _nlmsg_len;
2359         if (attrlen < 0)
2360                 return 0;
2361
2362         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
2363
2364         rta_len = RTA_ALIGN(sizeof(struct rtattr));
2365         while (RTA_OK(attr, attrlen)) {
2366                 if (attr->rta_type == IFLA_WIRELESS &&
2367                     i802_is_event_wireless_scan_complete(
2368                             ((char *) attr) + rta_len,
2369                             attr->rta_len - rta_len))
2370                         return 1;
2371                 attr = RTA_NEXT(attr, attrlen);
2372         }
2373
2374         return 0;
2375 }
2376
2377
2378 static int i802_is_scan_complete(int s, int ifindex)
2379 {
2380         char buf[1024];
2381         int left;
2382         struct nlmsghdr *h;
2383
2384         left = recv(s, buf, sizeof(buf), MSG_DONTWAIT);
2385         if (left < 0) {
2386                 perror("recv(netlink)");
2387                 return 0;
2388         }
2389
2390         h = (struct nlmsghdr *) buf;
2391         while (left >= (int) sizeof(*h)) {
2392                 int len, plen;
2393
2394                 len = h->nlmsg_len;
2395                 plen = len - sizeof(*h);
2396                 if (len > left || plen < 0) {
2397                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
2398                                    "len=%d left=%d plen=%d",
2399                                    len, left, plen);
2400                         break;
2401                 }
2402
2403                 switch (h->nlmsg_type) {
2404                 case RTM_NEWLINK:
2405                         if (i802_is_rtm_scan_complete(ifindex, h, plen))
2406                                 return 1;
2407                         break;
2408                 }
2409
2410                 len = NLMSG_ALIGN(len);
2411                 left -= len;
2412                 h = (struct nlmsghdr *) ((char *) h + len);
2413         }
2414
2415         return 0;
2416 }
2417
2418
2419 static int i802_ht_scan(struct i802_driver_data *drv)
2420 {
2421         struct iwreq iwr;
2422         int s, res, ifindex;
2423         struct sockaddr_nl local;
2424         time_t now, end;
2425         fd_set rfds;
2426         struct timeval tv;
2427
2428         wpa_printf(MSG_DEBUG, "nl80211: Scanning overlapping BSSes before "
2429                    "starting HT 20/40 MHz BSS");
2430
2431         /* Request a new scan */
2432         /* TODO: would be enough to scan the selected band */
2433         os_memset(&iwr, 0, sizeof(iwr));
2434         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2435         if (ioctl(drv->ioctl_sock, SIOCSIWSCAN, &iwr) < 0) {
2436                 perror("ioctl[SIOCSIWSCAN]");
2437                 return -1;
2438         }
2439
2440         ifindex = if_nametoindex(drv->iface);
2441
2442         /* Wait for scan completion event or timeout */
2443         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
2444         if (s < 0) {
2445                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
2446                 return -1;
2447         }
2448
2449         os_memset(&local, 0, sizeof(local));
2450         local.nl_family = AF_NETLINK;
2451         local.nl_groups = RTMGRP_LINK;
2452         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
2453                 perror("bind(netlink)");
2454                 close(s);
2455                 return -1;
2456         }
2457
2458         time(&end);
2459         end += 30; /* Wait at most 30 seconds for scan results */
2460         for (;;) {
2461                 time(&now);
2462                 tv.tv_sec = end > now ? end - now : 0;
2463                 tv.tv_usec = 0;
2464                 FD_ZERO(&rfds);
2465                 FD_SET(s, &rfds);
2466                 res = select(s + 1, &rfds, NULL, NULL, &tv);
2467                 if (res < 0) {
2468                         perror("select");
2469                         /* Assume results are ready after 10 seconds wait */
2470                         os_sleep(10, 0);
2471                         break;
2472                 } else if (res) {
2473                         if (i802_is_scan_complete(s, ifindex)) {
2474                                 wpa_printf(MSG_DEBUG, "nl80211: Scan "
2475                                            "completed");
2476                                 break;
2477                         }
2478                 } else {
2479                         wpa_printf(MSG_DEBUG, "nl80211: Scan timeout");
2480                         /* Assume results are ready to be read now */
2481                         break;
2482                 }
2483         }
2484
2485         close(s);
2486
2487         return i802_get_ht_scan_res(drv);
2488 }
2489 #endif /* CONFIG_IEEE80211N */
2490
2491
2492 static int i802_init_sockets(struct i802_driver_data *drv, const u8 *bssid)
2493 {
2494         struct ifreq ifr;
2495         struct sockaddr_ll addr;
2496
2497         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
2498         if (drv->ioctl_sock < 0) {
2499                 perror("socket[PF_INET,SOCK_DGRAM]");
2500                 return -1;
2501         }
2502
2503         /* start listening for EAPOL on the default AP interface */
2504         add_ifidx(drv, if_nametoindex(drv->iface));
2505
2506         if (hostapd_set_iface_flags(drv, drv->iface, 0))
2507                 return -1;
2508
2509         if (bssid) {
2510                 os_strlcpy(ifr.ifr_name, drv->iface, IFNAMSIZ);
2511                 memcpy(ifr.ifr_hwaddr.sa_data, bssid, ETH_ALEN);
2512                 ifr.ifr_hwaddr.sa_family = ARPHRD_ETHER;
2513
2514                 if (ioctl(drv->ioctl_sock, SIOCSIFHWADDR, &ifr)) {
2515                         perror("ioctl(SIOCSIFHWADDR)");
2516                         return -1;
2517                 }
2518         }
2519
2520         /*
2521          * initialise generic netlink and nl80211
2522          */
2523         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
2524         if (!drv->nl_cb) {
2525                 printf("Failed to allocate netlink callbacks.\n");
2526                 return -1;
2527         }
2528
2529         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
2530         if (!drv->nl_handle) {
2531                 printf("Failed to allocate netlink handle.\n");
2532                 return -1;
2533         }
2534
2535         if (genl_connect(drv->nl_handle)) {
2536                 printf("Failed to connect to generic netlink.\n");
2537                 return -1;
2538         }
2539
2540 #ifdef CONFIG_LIBNL20
2541         if (genl_ctrl_alloc_cache(drv->nl_handle, &drv->nl_cache) < 0) {
2542                 printf("Failed to allocate generic netlink cache.\n");
2543                 return -1;
2544         }
2545 #else /* CONFIG_LIBNL20 */
2546         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
2547         if (!drv->nl_cache) {
2548                 printf("Failed to allocate generic netlink cache.\n");
2549                 return -1;
2550         }
2551 #endif /* CONFIG_LIBNL20 */
2552
2553         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
2554         if (!drv->nl80211) {
2555                 printf("nl80211 not found.\n");
2556                 return -1;
2557         }
2558
2559 #ifdef CONFIG_IEEE80211N
2560         if (drv->ht_40mhz_scan) {
2561                 if (nl80211_set_mode(drv, drv->iface, NL80211_IFTYPE_STATION)
2562                     || hostapd_set_iface_flags(drv, drv->iface, 1) ||
2563                     i802_ht_scan(drv) ||
2564                     hostapd_set_iface_flags(drv, drv->iface, 0)) {
2565                         wpa_printf(MSG_ERROR, "Failed to scan channels for "
2566                                    "HT 40 MHz operations");
2567                         return -1;
2568                 }
2569         }
2570 #endif /* CONFIG_IEEE80211N */
2571
2572         /* Initialise a monitor interface */
2573         if (nl80211_create_monitor_interface(drv))
2574                 return -1;
2575
2576         if (nl80211_set_mode(drv, drv->iface, NL80211_IFTYPE_AP))
2577                 goto fail1;
2578
2579         if (hostapd_set_iface_flags(drv, drv->iface, 1))
2580                 goto fail1;
2581
2582         memset(&addr, 0, sizeof(addr));
2583         addr.sll_family = AF_PACKET;
2584         addr.sll_ifindex = ifr.ifr_ifindex;
2585         wpa_printf(MSG_DEBUG, "Opening raw packet socket for ifindex %d",
2586                    addr.sll_ifindex);
2587
2588         drv->eapol_sock = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_PAE));
2589         if (drv->eapol_sock < 0) {
2590                 perror("socket(PF_PACKET, SOCK_DGRAM, ETH_P_PAE)");
2591                 goto fail1;
2592         }
2593
2594         if (eloop_register_read_sock(drv->eapol_sock, handle_eapol, drv, NULL))
2595         {
2596                 printf("Could not register read socket for eapol\n");
2597                 return -1;
2598         }
2599
2600         memset(&ifr, 0, sizeof(ifr));
2601         os_strlcpy(ifr.ifr_name, drv->iface, sizeof(ifr.ifr_name));
2602         if (ioctl(drv->ioctl_sock, SIOCGIFHWADDR, &ifr) != 0) {
2603                 perror("ioctl(SIOCGIFHWADDR)");
2604                 goto fail1;
2605         }
2606
2607         if (ifr.ifr_hwaddr.sa_family != ARPHRD_ETHER) {
2608                 printf("Invalid HW-addr family 0x%04x\n",
2609                        ifr.ifr_hwaddr.sa_family);
2610                 goto fail1;
2611         }
2612         memcpy(drv->hapd->own_addr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
2613
2614         return 0;
2615
2616 fail1:
2617         nl80211_remove_iface(drv, drv->monitor_ifidx);
2618         return -1;
2619 }
2620
2621
2622 static int i802_get_inact_sec(void *priv, const u8 *addr)
2623 {
2624         struct hostap_sta_driver_data data;
2625         int ret;
2626
2627         data.inactive_msec = (unsigned long) -1;
2628         ret = i802_read_sta_data(priv, &data, addr);
2629         if (ret || data.inactive_msec == (unsigned long) -1)
2630                 return -1;
2631         return data.inactive_msec / 1000;
2632 }
2633
2634
2635 static int i802_sta_clear_stats(void *priv, const u8 *addr)
2636 {
2637 #if 0
2638         /* TODO */
2639 #endif
2640         return 0;
2641 }
2642
2643
2644 static void
2645 hostapd_wireless_event_wireless_custom(struct i802_driver_data *drv,
2646                                        char *custom)
2647 {
2648         wpa_printf(MSG_DEBUG, "Custom wireless event: '%s'", custom);
2649
2650         if (strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
2651                 char *pos;
2652                 u8 addr[ETH_ALEN];
2653                 pos = strstr(custom, "addr=");
2654                 if (pos == NULL) {
2655                         wpa_printf(MSG_DEBUG,
2656                                    "MLME-MICHAELMICFAILURE.indication "
2657                                    "without sender address ignored");
2658                         return;
2659                 }
2660                 pos += 5;
2661                 if (hwaddr_aton(pos, addr) == 0) {
2662                         hostapd_michael_mic_failure(drv->hapd, addr);
2663                 } else {
2664                         wpa_printf(MSG_DEBUG,
2665                                    "MLME-MICHAELMICFAILURE.indication "
2666                                    "with invalid MAC address");
2667                 }
2668         }
2669 }
2670
2671
2672 static void hostapd_wireless_event_wireless(struct i802_driver_data *drv,
2673                                             char *data, int len)
2674 {
2675         struct iw_event iwe_buf, *iwe = &iwe_buf;
2676         char *pos, *end, *custom, *buf;
2677
2678         pos = data;
2679         end = data + len;
2680
2681         while (pos + IW_EV_LCP_LEN <= end) {
2682                 /* Event data may be unaligned, so make a local, aligned copy
2683                  * before processing. */
2684                 memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
2685                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
2686                            iwe->cmd, iwe->len);
2687                 if (iwe->len <= IW_EV_LCP_LEN)
2688                         return;
2689
2690                 custom = pos + IW_EV_POINT_LEN;
2691                 if (drv->we_version > 18 &&
2692                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
2693                      iwe->cmd == IWEVCUSTOM)) {
2694                         /* WE-19 removed the pointer from struct iw_point */
2695                         char *dpos = (char *) &iwe_buf.u.data.length;
2696                         int dlen = dpos - (char *) &iwe_buf;
2697                         memcpy(dpos, pos + IW_EV_LCP_LEN,
2698                                sizeof(struct iw_event) - dlen);
2699                 } else {
2700                         memcpy(&iwe_buf, pos, sizeof(struct iw_event));
2701                         custom += IW_EV_POINT_OFF;
2702                 }
2703
2704                 switch (iwe->cmd) {
2705                 case IWEVCUSTOM:
2706                         if (custom + iwe->u.data.length > end)
2707                                 return;
2708                         buf = malloc(iwe->u.data.length + 1);
2709                         if (buf == NULL)
2710                                 return;
2711                         memcpy(buf, custom, iwe->u.data.length);
2712                         buf[iwe->u.data.length] = '\0';
2713                         hostapd_wireless_event_wireless_custom(drv, buf);
2714                         free(buf);
2715                         break;
2716                 }
2717
2718                 pos += iwe->len;
2719         }
2720 }
2721
2722
2723 static void hostapd_wireless_event_rtm_newlink(struct i802_driver_data *drv,
2724                                                struct nlmsghdr *h, int len)
2725 {
2726         struct ifinfomsg *ifi;
2727         int attrlen, nlmsg_len, rta_len;
2728         struct rtattr *attr;
2729
2730         if (len < (int) sizeof(*ifi))
2731                 return;
2732
2733         ifi = NLMSG_DATA(h);
2734
2735         /* TODO: use ifi->ifi_index to filter out wireless events from other
2736          * interfaces */
2737
2738         nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
2739
2740         attrlen = h->nlmsg_len - nlmsg_len;
2741         if (attrlen < 0)
2742                 return;
2743
2744         attr = (struct rtattr *) (((char *) ifi) + nlmsg_len);
2745
2746         rta_len = RTA_ALIGN(sizeof(struct rtattr));
2747         while (RTA_OK(attr, attrlen)) {
2748                 if (attr->rta_type == IFLA_WIRELESS) {
2749                         hostapd_wireless_event_wireless(
2750                                 drv, ((char *) attr) + rta_len,
2751                                 attr->rta_len - rta_len);
2752                 }
2753                 attr = RTA_NEXT(attr, attrlen);
2754         }
2755 }
2756
2757
2758 static void hostapd_wireless_event_receive(int sock, void *eloop_ctx,
2759                                            void *sock_ctx)
2760 {
2761         char buf[256];
2762         int left;
2763         struct sockaddr_nl from;
2764         socklen_t fromlen;
2765         struct nlmsghdr *h;
2766         struct i802_driver_data *drv = eloop_ctx;
2767
2768         fromlen = sizeof(from);
2769         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
2770                         (struct sockaddr *) &from, &fromlen);
2771         if (left < 0) {
2772                 if (errno != EINTR && errno != EAGAIN)
2773                         perror("recvfrom(netlink)");
2774                 return;
2775         }
2776
2777         h = (struct nlmsghdr *) buf;
2778         while (left >= (int) sizeof(*h)) {
2779                 int len, plen;
2780
2781                 len = h->nlmsg_len;
2782                 plen = len - sizeof(*h);
2783                 if (len > left || plen < 0) {
2784                         printf("Malformed netlink message: "
2785                                "len=%d left=%d plen=%d\n",
2786                                len, left, plen);
2787                         break;
2788                 }
2789
2790                 switch (h->nlmsg_type) {
2791                 case RTM_NEWLINK:
2792                         hostapd_wireless_event_rtm_newlink(drv, h, plen);
2793                         break;
2794                 }
2795
2796                 len = NLMSG_ALIGN(len);
2797                 left -= len;
2798                 h = (struct nlmsghdr *) ((char *) h + len);
2799         }
2800
2801         if (left > 0) {
2802                 printf("%d extra bytes in the end of netlink message\n", left);
2803         }
2804 }
2805
2806
2807 static int hostap_get_we_version(struct i802_driver_data *drv)
2808 {
2809         struct iw_range *range;
2810         struct iwreq iwr;
2811         int minlen;
2812         size_t buflen;
2813
2814         drv->we_version = 0;
2815
2816         /*
2817          * Use larger buffer than struct iw_range in order to allow the
2818          * structure to grow in the future.
2819          */
2820         buflen = sizeof(struct iw_range) + 500;
2821         range = os_zalloc(buflen);
2822         if (range == NULL)
2823                 return -1;
2824
2825         memset(&iwr, 0, sizeof(iwr));
2826         os_strlcpy(iwr.ifr_name, drv->iface, IFNAMSIZ);
2827         iwr.u.data.pointer = (caddr_t) range;
2828         iwr.u.data.length = buflen;
2829
2830         minlen = ((char *) &range->enc_capa) - (char *) range +
2831                 sizeof(range->enc_capa);
2832
2833         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
2834                 perror("ioctl[SIOCGIWRANGE]");
2835                 free(range);
2836                 return -1;
2837         } else if (iwr.u.data.length >= minlen &&
2838                    range->we_version_compiled >= 18) {
2839                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
2840                            "WE(source)=%d enc_capa=0x%x",
2841                            range->we_version_compiled,
2842                            range->we_version_source,
2843                            range->enc_capa);
2844                 drv->we_version = range->we_version_compiled;
2845         }
2846
2847         free(range);
2848         return 0;
2849 }
2850
2851
2852 static int i802_wireless_event_init(void *priv)
2853 {
2854         struct i802_driver_data *drv = priv;
2855         int s;
2856         struct sockaddr_nl local;
2857
2858         hostap_get_we_version(drv);
2859
2860         drv->wext_sock = -1;
2861
2862         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
2863         if (s < 0) {
2864                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
2865                 return -1;
2866         }
2867
2868         memset(&local, 0, sizeof(local));
2869         local.nl_family = AF_NETLINK;
2870         local.nl_groups = RTMGRP_LINK;
2871         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
2872                 perror("bind(netlink)");
2873                 close(s);
2874                 return -1;
2875         }
2876
2877         eloop_register_read_sock(s, hostapd_wireless_event_receive, drv,
2878                                  NULL);
2879         drv->wext_sock = s;
2880
2881         return 0;
2882 }
2883
2884
2885 static void i802_wireless_event_deinit(void *priv)
2886 {
2887         struct i802_driver_data *drv = priv;
2888         if (drv->wext_sock < 0)
2889                 return;
2890         eloop_unregister_read_sock(drv->wext_sock);
2891         close(drv->wext_sock);
2892 }
2893
2894
2895 static int i802_sta_deauth(void *priv, const u8 *addr, int reason)
2896 {
2897         struct i802_driver_data *drv = priv;
2898         struct ieee80211_mgmt mgmt;
2899
2900         memset(&mgmt, 0, sizeof(mgmt));
2901         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2902                                           WLAN_FC_STYPE_DEAUTH);
2903         memcpy(mgmt.da, addr, ETH_ALEN);
2904         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2905         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2906         mgmt.u.deauth.reason_code = host_to_le16(reason);
2907         return i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2908                                       sizeof(mgmt.u.deauth), 0);
2909 }
2910
2911
2912 static int i802_sta_disassoc(void *priv, const u8 *addr, int reason)
2913 {
2914         struct i802_driver_data *drv = priv;
2915         struct ieee80211_mgmt mgmt;
2916
2917         memset(&mgmt, 0, sizeof(mgmt));
2918         mgmt.frame_control = IEEE80211_FC(WLAN_FC_TYPE_MGMT,
2919                                           WLAN_FC_STYPE_DISASSOC);
2920         memcpy(mgmt.da, addr, ETH_ALEN);
2921         memcpy(mgmt.sa, drv->hapd->own_addr, ETH_ALEN);
2922         memcpy(mgmt.bssid, drv->hapd->own_addr, ETH_ALEN);
2923         mgmt.u.disassoc.reason_code = host_to_le16(reason);
2924         return  i802_send_mgmt_frame(drv, &mgmt, IEEE80211_HDRLEN +
2925                                        sizeof(mgmt.u.disassoc), 0);
2926 }
2927
2928
2929 static const struct hostapd_neighbor_bss *
2930 i802_get_neighbor_bss(void *priv, size_t *num)
2931 {
2932         struct i802_driver_data *drv = priv;
2933         *num = drv->num_neighbors;
2934         return drv->neighbors;
2935 }
2936
2937
2938 static void *i802_init_bssid(struct hostapd_data *hapd, const u8 *bssid)
2939 {
2940         struct i802_driver_data *drv;
2941
2942         drv = os_zalloc(sizeof(struct i802_driver_data));
2943         if (drv == NULL) {
2944                 printf("Could not allocate memory for i802 driver data\n");
2945                 return NULL;
2946         }
2947
2948         drv->hapd = hapd;
2949         memcpy(drv->iface, hapd->conf->iface, sizeof(drv->iface));
2950
2951         drv->num_if_indices = sizeof(drv->default_if_indices) / sizeof(int);
2952         drv->if_indices = drv->default_if_indices;
2953         drv->bridge = if_nametoindex(hapd->conf->bridge);
2954         drv->ht_40mhz_scan = hapd->iconf->secondary_channel != 0;
2955
2956         if (i802_init_sockets(drv, bssid))
2957                 goto failed;
2958
2959         return drv;
2960
2961 failed:
2962         free(drv);
2963         return NULL;
2964 }
2965
2966
2967 static void *i802_init(struct hostapd_data *hapd)
2968 {
2969         return i802_init_bssid(hapd, NULL);
2970 }
2971
2972
2973 static void i802_deinit(void *priv)
2974 {
2975         struct i802_driver_data *drv = priv;
2976
2977         if (drv->last_freq_ht) {
2978                 /* Clear HT flags from the driver */
2979                 struct hostapd_freq_params freq;
2980                 os_memset(&freq, 0, sizeof(freq));
2981                 freq.freq = drv->last_freq;
2982                 i802_set_freq2(priv, &freq);
2983         }
2984
2985         i802_del_beacon(drv);
2986
2987         /* remove monitor interface */
2988         nl80211_remove_iface(drv, drv->monitor_ifidx);
2989
2990         (void) hostapd_set_iface_flags(drv, drv->iface, 0);
2991
2992         if (drv->monitor_sock >= 0) {
2993                 eloop_unregister_read_sock(drv->monitor_sock);
2994                 close(drv->monitor_sock);
2995         }
2996         if (drv->ioctl_sock >= 0)
2997                 close(drv->ioctl_sock);
2998         if (drv->eapol_sock >= 0) {
2999                 eloop_unregister_read_sock(drv->eapol_sock);
3000                 close(drv->eapol_sock);
3001         }
3002
3003         genl_family_put(drv->nl80211);
3004         nl_cache_free(drv->nl_cache);
3005         nl_handle_destroy(drv->nl_handle);
3006         nl_cb_put(drv->nl_cb);
3007
3008         if (drv->if_indices != drv->default_if_indices)
3009                 free(drv->if_indices);
3010
3011         os_free(drv->neighbors);
3012         free(drv);
3013 }
3014
3015
3016 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
3017         .name = "nl80211",
3018         .init = i802_init,
3019         .init_bssid = i802_init_bssid,
3020         .deinit = i802_deinit,
3021         .wireless_event_init = i802_wireless_event_init,
3022         .wireless_event_deinit = i802_wireless_event_deinit,
3023         .set_ieee8021x = i802_set_ieee8021x,
3024         .set_privacy = i802_set_privacy,
3025         .set_encryption = i802_set_encryption,
3026         .get_seqnum = i802_get_seqnum,
3027         .flush = i802_flush,
3028         .read_sta_data = i802_read_sta_data,
3029         .send_eapol = i802_send_eapol,
3030         .sta_set_flags = i802_sta_set_flags,
3031         .sta_deauth = i802_sta_deauth,
3032         .sta_disassoc = i802_sta_disassoc,
3033         .sta_remove = i802_sta_remove,
3034         .send_mgmt_frame = i802_send_mgmt_frame,
3035         .sta_add2 = i802_sta_add2,
3036         .get_inact_sec = i802_get_inact_sec,
3037         .sta_clear_stats = i802_sta_clear_stats,
3038         .set_freq2 = i802_set_freq2,
3039         .set_rts = i802_set_rts,
3040         .get_rts = i802_get_rts,
3041         .set_frag = i802_set_frag,
3042         .get_frag = i802_get_frag,
3043         .set_retry = i802_set_retry,
3044         .get_retry = i802_get_retry,
3045         .set_rate_sets = i802_set_rate_sets,
3046         .set_beacon = i802_set_beacon,
3047         .set_internal_bridge = i802_set_internal_bridge,
3048         .set_beacon_int = i802_set_beacon_int,
3049         .set_dtim_period = i802_set_dtim_period,
3050         .set_cts_protect = i802_set_cts_protect,
3051         .set_preamble = i802_set_preamble,
3052         .set_short_slot_time = i802_set_short_slot_time,
3053         .set_tx_queue_params = i802_set_tx_queue_params,
3054         .bss_add = i802_bss_add,
3055         .bss_remove = i802_bss_remove,
3056         .if_add = i802_if_add,
3057         .if_update = i802_if_update,
3058         .if_remove = i802_if_remove,
3059         .get_hw_feature_data = i802_get_hw_feature_data,
3060         .set_sta_vlan = i802_set_sta_vlan,
3061         .set_country = i802_set_country,
3062         .get_neighbor_bss = i802_get_neighbor_bss,
3063 };