src/drivers/driver_nl80211.c

   1 /*
   2  * WPA Supplicant - driver interaction with Linux nl80211/cfg80211
   3  * Copyright (c) 2003-2008, Jouni Malinen <j@w1.fi>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * Alternatively, this software may be distributed under the terms of BSD
  10  * license.
  11  *
  12  * See README and COPYING for more details.
  13  */
  14
  15 #include "includes.h"
  16 #include <sys/ioctl.h>
  17 #include <net/if_arp.h>
  18 #include <netlink/genl/genl.h>
  19 #include <netlink/genl/family.h>
  20 #include <netlink/genl/ctrl.h>
  21 #include "nl80211_copy.h"
  22 #ifdef CONFIG_CLIENT_MLME
  23 #include <netpacket/packet.h>
  24 #include <linux/if_ether.h>
  25 #include "radiotap.h"
  26 #include "radiotap_iter.h"
  27 #endif /* CONFIG_CLIENT_MLME */
  28
  29 #include "wireless_copy.h"
  30 #include "common.h"
  31 #include "driver.h"
  32 #include "eloop.h"
  33 #include "ieee802_11_defs.h"
  34
  35 #ifndef IFF_LOWER_UP
  36 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
  37 #endif
  38 #ifndef IFF_DORMANT
  39 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
  40 #endif
  41
  42 #ifndef IF_OPER_DORMANT
  43 #define IF_OPER_DORMANT 5
  44 #endif
  45 #ifndef IF_OPER_UP
  46 #define IF_OPER_UP 6
  47 #endif
  48
  49
  50 struct wpa_driver_nl80211_data {
  51         void *ctx;
  52         int event_sock;
  53         int ioctl_sock;
  54         char ifname[IFNAMSIZ + 1];
  55         int ifindex;
  56         int if_removed;
  57         u8 *assoc_req_ies;
  58         size_t assoc_req_ies_len;
  59         u8 *assoc_resp_ies;
  60         size_t assoc_resp_ies_len;
  61         struct wpa_driver_capa capa;
  62         int has_capability;
  63         int we_version_compiled;
  64
  65         /* for set_auth_alg fallback */
  66         int use_crypt;
  67         int auth_alg_fallback;
  68
  69         int operstate;
  70
  71         char mlmedev[IFNAMSIZ + 1];
  72
  73         int scan_complete_events;
  74
  75         struct nl_handle *nl_handle;
  76         struct nl_cache *nl_cache;
  77         struct nl_cb *nl_cb;
  78         struct genl_family *nl80211;
  79
  80 #ifdef CONFIG_CLIENT_MLME
  81         int monitor_sock; /* socket for monitor */
  82         int monitor_ifidx;
  83 #endif /* CONFIG_CLIENT_MLME */
  84 };
  85
  86
  87 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
  88                                             void *timeout_ctx);
  89 static int wpa_driver_nl80211_set_mode(void *priv, int mode);
  90 static int wpa_driver_nl80211_flush_pmkid(void *priv);
  91 static int wpa_driver_nl80211_get_range(void *priv);
  92 static void
  93 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
  94
  95
  96 /* nl80211 code */
  97 static int ack_handler(struct nl_msg *msg, void *arg)
  98 {
  99         int *err = arg;
 100         *err = 0;
 101         return NL_STOP;
 102 }
 103
 104 static int finish_handler(struct nl_msg *msg, void *arg)
 105 {
 106         int *ret = arg;
 107         *ret = 0;
 108         return NL_SKIP;
 109 }
 110
 111 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
 112                          void *arg)
 113 {
 114         int *ret = arg;
 115         *ret = err->error;
 116         return NL_SKIP;
 117 }
 118
 119 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
 120                               struct nl_msg *msg,
 121                               int (*valid_handler)(struct nl_msg *, void *),
 122                               void *valid_data)
 123 {
 124         struct nl_cb *cb;
 125         int err = -ENOMEM;
 126
 127         cb = nl_cb_clone(drv->nl_cb);
 128         if (!cb)
 129                 goto out;
 130
 131         err = nl_send_auto_complete(drv->nl_handle, msg);
 132         if (err < 0)
 133                 goto out;
 134
 135         err = 1;
 136
 137         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
 138         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
 139         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
 140
 141         if (valid_handler)
 142                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
 143                           valid_handler, valid_data);
 144
 145         while (err > 0)
 146                 nl_recvmsgs(drv->nl_handle, cb);
 147  out:
 148         nl_cb_put(cb);
 149         nlmsg_free(msg);
 150         return err;
 151 }
 152
 153
 154 static int wpa_driver_nl80211_send_oper_ifla(
 155         struct wpa_driver_nl80211_data *drv,
 156         int linkmode, int operstate)
 157 {
 158         struct {
 159                 struct nlmsghdr hdr;
 160                 struct ifinfomsg ifinfo;
 161                 char opts[16];
 162         } req;
 163         struct rtattr *rta;
 164         static int nl_seq;
 165         ssize_t ret;
 166
 167         os_memset(&req, 0, sizeof(req));
 168
 169         req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
 170         req.hdr.nlmsg_type = RTM_SETLINK;
 171         req.hdr.nlmsg_flags = NLM_F_REQUEST;
 172         req.hdr.nlmsg_seq = ++nl_seq;
 173         req.hdr.nlmsg_pid = 0;
 174
 175         req.ifinfo.ifi_family = AF_UNSPEC;
 176         req.ifinfo.ifi_type = 0;
 177         req.ifinfo.ifi_index = drv->ifindex;
 178         req.ifinfo.ifi_flags = 0;
 179         req.ifinfo.ifi_change = 0;
 180
 181         if (linkmode != -1) {
 182                 rta = (struct rtattr *)
 183                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
 184                 rta->rta_type = IFLA_LINKMODE;
 185                 rta->rta_len = RTA_LENGTH(sizeof(char));
 186                 *((char *) RTA_DATA(rta)) = linkmode;
 187                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
 188                         RTA_LENGTH(sizeof(char));
 189         }
 190         if (operstate != -1) {
 191                 rta = (struct rtattr *)
 192                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
 193                 rta->rta_type = IFLA_OPERSTATE;
 194                 rta->rta_len = RTA_LENGTH(sizeof(char));
 195                 *((char *) RTA_DATA(rta)) = operstate;
 196                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
 197                         RTA_LENGTH(sizeof(char));
 198         }
 199
 200         wpa_printf(MSG_DEBUG, "WEXT: Operstate: linkmode=%d, operstate=%d",
 201                    linkmode, operstate);
 202
 203         ret = send(drv->event_sock, &req, req.hdr.nlmsg_len, 0);
 204         if (ret < 0) {
 205                 wpa_printf(MSG_DEBUG, "WEXT: Sending operstate IFLA failed: "
 206                            "%s (assume operstate is not supported)",
 207                            strerror(errno));
 208         }
 209
 210         return ret < 0 ? -1 : 0;
 211 }
 212
 213
 214 static int wpa_driver_nl80211_set_auth_param(
 215         struct wpa_driver_nl80211_data *drv, int idx, u32 value)
 216 {
 217         struct iwreq iwr;
 218         int ret = 0;
 219
 220         os_memset(&iwr, 0, sizeof(iwr));
 221         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 222         iwr.u.param.flags = idx & IW_AUTH_INDEX;
 223         iwr.u.param.value = value;
 224
 225         if (ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr) < 0) {
 226                 if (errno != EOPNOTSUPP) {
 227                         wpa_printf(MSG_DEBUG, "WEXT: SIOCSIWAUTH(param %d "
 228                                    "value 0x%x) failed: %s)",
 229                                    idx, value, strerror(errno));
 230                 }
 231                 ret = errno == EOPNOTSUPP ? -2 : -1;
 232         }
 233
 234         return ret;
 235 }
 236
 237
 238 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
 239 {
 240         struct wpa_driver_nl80211_data *drv = priv;
 241         struct iwreq iwr;
 242         int ret = 0;
 243
 244         os_memset(&iwr, 0, sizeof(iwr));
 245         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 246
 247         if (ioctl(drv->ioctl_sock, SIOCGIWAP, &iwr) < 0) {
 248                 perror("ioctl[SIOCGIWAP]");
 249                 ret = -1;
 250         }
 251         os_memcpy(bssid, iwr.u.ap_addr.sa_data, ETH_ALEN);
 252
 253         return ret;
 254 }
 255
 256
 257 static int wpa_driver_nl80211_set_bssid(void *priv, const u8 *bssid)
 258 {
 259         struct wpa_driver_nl80211_data *drv = priv;
 260         struct iwreq iwr;
 261         int ret = 0;
 262
 263         os_memset(&iwr, 0, sizeof(iwr));
 264         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 265         iwr.u.ap_addr.sa_family = ARPHRD_ETHER;
 266         if (bssid)
 267                 os_memcpy(iwr.u.ap_addr.sa_data, bssid, ETH_ALEN);
 268         else
 269                 os_memset(iwr.u.ap_addr.sa_data, 0, ETH_ALEN);
 270
 271         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr) < 0) {
 272                 perror("ioctl[SIOCSIWAP]");
 273                 ret = -1;
 274         }
 275
 276         return ret;
 277 }
 278
 279
 280 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
 281 {
 282         struct wpa_driver_nl80211_data *drv = priv;
 283         struct iwreq iwr;
 284         int ret = 0;
 285
 286         os_memset(&iwr, 0, sizeof(iwr));
 287         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 288         iwr.u.essid.pointer = (caddr_t) ssid;
 289         iwr.u.essid.length = 32;
 290
 291         if (ioctl(drv->ioctl_sock, SIOCGIWESSID, &iwr) < 0) {
 292                 perror("ioctl[SIOCGIWESSID]");
 293                 ret = -1;
 294         } else {
 295                 ret = iwr.u.essid.length;
 296                 if (ret > 32)
 297                         ret = 32;
 298                 /* Some drivers include nul termination in the SSID, so let's
 299                  * remove it here before further processing. WE-21 changes this
 300                  * to explicitly require the length _not_ to include nul
 301                  * termination. */
 302                 if (ret > 0 && ssid[ret - 1] == '\0' &&
 303                     drv->we_version_compiled < 21)
 304                         ret--;
 305         }
 306
 307         return ret;
 308 }
 309
 310
 311 static int wpa_driver_nl80211_set_ssid(void *priv, const u8 *ssid,
 312                                        size_t ssid_len)
 313 {
 314         struct wpa_driver_nl80211_data *drv = priv;
 315         struct iwreq iwr;
 316         int ret = 0;
 317         char buf[33];
 318
 319         if (ssid_len > 32)
 320                 return -1;
 321
 322         os_memset(&iwr, 0, sizeof(iwr));
 323         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 324         /* flags: 1 = ESSID is active, 0 = not (promiscuous) */
 325         iwr.u.essid.flags = (ssid_len != 0);
 326         os_memset(buf, 0, sizeof(buf));
 327         os_memcpy(buf, ssid, ssid_len);
 328         iwr.u.essid.pointer = (caddr_t) buf;
 329         if (drv->we_version_compiled < 21) {
 330                 /* For historic reasons, set SSID length to include one extra
 331                  * character, C string nul termination, even though SSID is
 332                  * really an octet string that should not be presented as a C
 333                  * string. Some Linux drivers decrement the length by one and
 334                  * can thus end up missing the last octet of the SSID if the
 335                  * length is not incremented here. WE-21 changes this to
 336                  * explicitly require the length _not_ to include nul
 337                  * termination. */
 338                 if (ssid_len)
 339                         ssid_len++;
 340         }
 341         iwr.u.essid.length = ssid_len;
 342
 343         if (ioctl(drv->ioctl_sock, SIOCSIWESSID, &iwr) < 0) {
 344                 perror("ioctl[SIOCSIWESSID]");
 345                 ret = -1;
 346         }
 347
 348         return ret;
 349 }
 350
 351
 352 static int wpa_driver_nl80211_set_freq(void *priv, int freq)
 353 {
 354         struct wpa_driver_nl80211_data *drv = priv;
 355         struct iwreq iwr;
 356         int ret = 0;
 357
 358         os_memset(&iwr, 0, sizeof(iwr));
 359         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
 360         iwr.u.freq.m = freq * 100000;
 361         iwr.u.freq.e = 1;
 362
 363         if (ioctl(drv->ioctl_sock, SIOCSIWFREQ, &iwr) < 0) {
 364                 perror("ioctl[SIOCSIWFREQ]");
 365                 ret = -1;
 366         }
 367
 368         return ret;
 369 }
 370
 371
 372 static void
 373 wpa_driver_nl80211_event_wireless_custom(void *ctx, char *custom)
 374 {
 375         union wpa_event_data data;
 376
 377         wpa_printf(MSG_MSGDUMP, "WEXT: Custom wireless event: '%s'",
 378                    custom);
 379
 380         os_memset(&data, 0, sizeof(data));
 381         /* Host AP driver */
 382         if (os_strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
 383                 data.michael_mic_failure.unicast =
 384                         os_strstr(custom, " unicast ") != NULL;
 385                 /* TODO: parse parameters(?) */
 386                 wpa_supplicant_event(ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
 387         } else if (os_strncmp(custom, "ASSOCINFO(ReqIEs=", 17) == 0) {
 388                 char *spos;
 389                 int bytes;
 390
 391                 spos = custom + 17;
 392
 393                 bytes = strspn(spos, "0123456789abcdefABCDEF");
 394                 if (!bytes || (bytes & 1))
 395                         return;
 396                 bytes /= 2;
 397
 398                 data.assoc_info.req_ies = os_malloc(bytes);
 399                 if (data.assoc_info.req_ies == NULL)
 400                         return;
 401
 402                 data.assoc_info.req_ies_len = bytes;
 403                 hexstr2bin(spos, data.assoc_info.req_ies, bytes);
 404
 405                 spos += bytes * 2;
 406
 407                 data.assoc_info.resp_ies = NULL;
 408                 data.assoc_info.resp_ies_len = 0;
 409
 410                 if (os_strncmp(spos, " RespIEs=", 9) == 0) {
 411                         spos += 9;
 412
 413                         bytes = strspn(spos, "0123456789abcdefABCDEF");
 414                         if (!bytes || (bytes & 1))
 415                                 goto done;
 416                         bytes /= 2;
 417
 418                         data.assoc_info.resp_ies = os_malloc(bytes);
 419                         if (data.assoc_info.resp_ies == NULL)
 420                                 goto done;
 421
 422                         data.assoc_info.resp_ies_len = bytes;
 423                         hexstr2bin(spos, data.assoc_info.resp_ies, bytes);
 424                 }
 425
 426                 wpa_supplicant_event(ctx, EVENT_ASSOCINFO, &data);
 427
 428         done:
 429                 os_free(data.assoc_info.resp_ies);
 430                 os_free(data.assoc_info.req_ies);
 431 #ifdef CONFIG_PEERKEY
 432         } else if (os_strncmp(custom, "STKSTART.request=", 17) == 0) {
 433                 if (hwaddr_aton(custom + 17, data.stkstart.peer)) {
 434                         wpa_printf(MSG_DEBUG, "WEXT: unrecognized "
 435                                    "STKSTART.request '%s'", custom + 17);
 436                         return;
 437                 }
 438                 wpa_supplicant_event(ctx, EVENT_STKSTART, &data);
 439 #endif /* CONFIG_PEERKEY */
 440         }
 441 }
 442
 443
 444 static int wpa_driver_nl80211_event_wireless_michaelmicfailure(
 445         void *ctx, const char *ev, size_t len)
 446 {
 447         const struct iw_michaelmicfailure *mic;
 448         union wpa_event_data data;
 449
 450         if (len < sizeof(*mic))
 451                 return -1;
 452
 453         mic = (const struct iw_michaelmicfailure *) ev;
 454
 455         wpa_printf(MSG_DEBUG, "Michael MIC failure wireless event: "
 456                    "flags=0x%x src_addr=" MACSTR, mic->flags,
 457                    MAC2STR(mic->src_addr.sa_data));
 458
 459         os_memset(&data, 0, sizeof(data));
 460         data.michael_mic_failure.unicast = !(mic->flags & IW_MICFAILURE_GROUP);
 461         wpa_supplicant_event(ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
 462
 463         return 0;
 464 }
 465
 466
 467 static int wpa_driver_nl80211_event_wireless_pmkidcand(
 468         struct wpa_driver_nl80211_data *drv, const char *ev, size_t len)
 469 {
 470         const struct iw_pmkid_cand *cand;
 471         union wpa_event_data data;
 472         const u8 *addr;
 473
 474         if (len < sizeof(*cand))
 475                 return -1;
 476
 477         cand = (const struct iw_pmkid_cand *) ev;
 478         addr = (const u8 *) cand->bssid.sa_data;
 479
 480         wpa_printf(MSG_DEBUG, "PMKID candidate wireless event: "
 481                    "flags=0x%x index=%d bssid=" MACSTR, cand->flags,
 482                    cand->index, MAC2STR(addr));
 483
 484         os_memset(&data, 0, sizeof(data));
 485         os_memcpy(data.pmkid_candidate.bssid, addr, ETH_ALEN);
 486         data.pmkid_candidate.index = cand->index;
 487         data.pmkid_candidate.preauth = cand->flags & IW_PMKID_CAND_PREAUTH;
 488         wpa_supplicant_event(drv->ctx, EVENT_PMKID_CANDIDATE, &data);
 489
 490         return 0;
 491 }
 492
 493
 494 static int wpa_driver_nl80211_event_wireless_assocreqie(
 495         struct wpa_driver_nl80211_data *drv, const char *ev, int len)
 496 {
 497         if (len < 0)
 498                 return -1;
 499
 500         wpa_hexdump(MSG_DEBUG, "AssocReq IE wireless event", (const u8 *) ev,
 501                     len);
 502         os_free(drv->assoc_req_ies);
 503         drv->assoc_req_ies = os_malloc(len);
 504         if (drv->assoc_req_ies == NULL) {
 505                 drv->assoc_req_ies_len = 0;
 506                 return -1;
 507         }
 508         os_memcpy(drv->assoc_req_ies, ev, len);
 509         drv->assoc_req_ies_len = len;
 510
 511         return 0;
 512 }
 513
 514
 515 static int wpa_driver_nl80211_event_wireless_assocrespie(
 516         struct wpa_driver_nl80211_data *drv, const char *ev, int len)
 517 {
 518         if (len < 0)
 519                 return -1;
 520
 521         wpa_hexdump(MSG_DEBUG, "AssocResp IE wireless event", (const u8 *) ev,
 522                     len);
 523         os_free(drv->assoc_resp_ies);
 524         drv->assoc_resp_ies = os_malloc(len);
 525         if (drv->assoc_resp_ies == NULL) {
 526                 drv->assoc_resp_ies_len = 0;
 527                 return -1;
 528         }
 529         os_memcpy(drv->assoc_resp_ies, ev, len);
 530         drv->assoc_resp_ies_len = len;
 531
 532         return 0;
 533 }
 534
 535
 536 static void wpa_driver_nl80211_event_assoc_ies(struct wpa_driver_nl80211_data *drv)
 537 {
 538         union wpa_event_data data;
 539
 540         if (drv->assoc_req_ies == NULL && drv->assoc_resp_ies == NULL)
 541                 return;
 542
 543         os_memset(&data, 0, sizeof(data));
 544         if (drv->assoc_req_ies) {
 545                 data.assoc_info.req_ies = drv->assoc_req_ies;
 546                 drv->assoc_req_ies = NULL;
 547                 data.assoc_info.req_ies_len = drv->assoc_req_ies_len;
 548         }
 549         if (drv->assoc_resp_ies) {
 550                 data.assoc_info.resp_ies = drv->assoc_resp_ies;
 551                 drv->assoc_resp_ies = NULL;
 552                 data.assoc_info.resp_ies_len = drv->assoc_resp_ies_len;
 553         }
 554
 555         wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &data);
 556
 557         os_free(data.assoc_info.req_ies);
 558         os_free(data.assoc_info.resp_ies);
 559 }
 560
 561
 562 static void wpa_driver_nl80211_event_wireless(struct wpa_driver_nl80211_data *drv,
 563                                            void *ctx, char *data, int len)
 564 {
 565         struct iw_event iwe_buf, *iwe = &iwe_buf;
 566         char *pos, *end, *custom, *buf;
 567
 568         pos = data;
 569         end = data + len;
 570
 571         while (pos + IW_EV_LCP_LEN <= end) {
 572                 /* Event data may be unaligned, so make a local, aligned copy
 573                  * before processing. */
 574                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
 575                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
 576                            iwe->cmd, iwe->len);
 577                 if (iwe->len <= IW_EV_LCP_LEN)
 578                         return;
 579
 580                 custom = pos + IW_EV_POINT_LEN;
 581                 if (drv->we_version_compiled > 18 &&
 582                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
 583                      iwe->cmd == IWEVCUSTOM ||
 584                      iwe->cmd == IWEVASSOCREQIE ||
 585                      iwe->cmd == IWEVASSOCRESPIE ||
 586                      iwe->cmd == IWEVPMKIDCAND)) {
 587                         /* WE-19 removed the pointer from struct iw_point */
 588                         char *dpos = (char *) &iwe_buf.u.data.length;
 589                         int dlen = dpos - (char *) &iwe_buf;
 590                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
 591                                   sizeof(struct iw_event) - dlen);
 592                 } else {
 593                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
 594                         custom += IW_EV_POINT_OFF;
 595                 }
 596
 597                 switch (iwe->cmd) {
 598                 case SIOCGIWAP:
 599                         wpa_printf(MSG_DEBUG, "Wireless event: new AP: "
 600                                    MACSTR,
 601                                    MAC2STR((u8 *) iwe->u.ap_addr.sa_data));
 602                         if (is_zero_ether_addr(
 603                                     (const u8 *) iwe->u.ap_addr.sa_data) ||
 604                             os_memcmp(iwe->u.ap_addr.sa_data,
 605                                       "\x44\x44\x44\x44\x44\x44", ETH_ALEN) ==
 606                             0) {
 607                                 os_free(drv->assoc_req_ies);
 608                                 drv->assoc_req_ies = NULL;
 609                                 os_free(drv->assoc_resp_ies);
 610                                 drv->assoc_resp_ies = NULL;
 611                                 wpa_supplicant_event(ctx, EVENT_DISASSOC,
 612                                                      NULL);
 613
 614                         } else {
 615                                 wpa_driver_nl80211_event_assoc_ies(drv);
 616                                 wpa_supplicant_event(ctx, EVENT_ASSOC, NULL);
 617                         }
 618                         break;
 619                 case IWEVMICHAELMICFAILURE:
 620                         wpa_driver_nl80211_event_wireless_michaelmicfailure(
 621                                 ctx, custom, iwe->u.data.length);
 622                         break;
 623                 case IWEVCUSTOM:
 624                         if (custom + iwe->u.data.length > end)
 625                                 return;
 626                         buf = os_malloc(iwe->u.data.length + 1);
 627                         if (buf == NULL)
 628                                 return;
 629                         os_memcpy(buf, custom, iwe->u.data.length);
 630                         buf[iwe->u.data.length] = '\0';
 631                         wpa_driver_nl80211_event_wireless_custom(ctx, buf);
 632                         os_free(buf);
 633                         break;
 634                 case SIOCGIWSCAN:
 635                         drv->scan_complete_events = 1;
 636                         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout,
 637                                              drv, ctx);
 638                         wpa_supplicant_event(ctx, EVENT_SCAN_RESULTS, NULL);
 639                         break;
 640                 case IWEVASSOCREQIE:
 641                         wpa_driver_nl80211_event_wireless_assocreqie(
 642                                 drv, custom, iwe->u.data.length);
 643                         break;
 644                 case IWEVASSOCRESPIE:
 645                         wpa_driver_nl80211_event_wireless_assocrespie(
 646                                 drv, custom, iwe->u.data.length);
 647                         break;
 648                 case IWEVPMKIDCAND:
 649                         wpa_driver_nl80211_event_wireless_pmkidcand(
 650                                 drv, custom, iwe->u.data.length);
 651                         break;
 652                 }
 653
 654                 pos += iwe->len;
 655         }
 656 }
 657
 658
 659 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
 660                                           void *ctx, char *buf, size_t len,
 661                                           int del)
 662 {
 663         union wpa_event_data event;
 664
 665         os_memset(&event, 0, sizeof(event));
 666         if (len > sizeof(event.interface_status.ifname))
 667                 len = sizeof(event.interface_status.ifname) - 1;
 668         os_memcpy(event.interface_status.ifname, buf, len);
 669         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
 670                 EVENT_INTERFACE_ADDED;
 671
 672         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
 673                    del ? "DEL" : "NEW",
 674                    event.interface_status.ifname,
 675                    del ? "removed" : "added");
 676
 677         if (os_strcmp(drv->ifname, event.interface_status.ifname) == 0) {
 678                 if (del)
 679                         drv->if_removed = 1;
 680                 else
 681                         drv->if_removed = 0;
 682         }
 683
 684         wpa_supplicant_event(ctx, EVENT_INTERFACE_STATUS, &event);
 685 }
 686
 687
 688 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
 689                                          struct nlmsghdr *h)
 690 {
 691         struct ifinfomsg *ifi;
 692         int attrlen, _nlmsg_len, rta_len;
 693         struct rtattr *attr;
 694
 695         ifi = NLMSG_DATA(h);
 696
 697         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
 698
 699         attrlen = h->nlmsg_len - _nlmsg_len;
 700         if (attrlen < 0)
 701                 return 0;
 702
 703         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
 704
 705         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 706         while (RTA_OK(attr, attrlen)) {
 707                 if (attr->rta_type == IFLA_IFNAME) {
 708                         if (os_strcmp(((char *) attr) + rta_len, drv->ifname)
 709                             == 0)
 710                                 return 1;
 711                         else
 712                                 break;
 713                 }
 714                 attr = RTA_NEXT(attr, attrlen);
 715         }
 716
 717         return 0;
 718 }
 719
 720
 721 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
 722                                           int ifindex, struct nlmsghdr *h)
 723 {
 724         if (drv->ifindex == ifindex)
 725                 return 1;
 726
 727         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, h)) {
 728                 drv->ifindex = if_nametoindex(drv->ifname);
 729                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
 730                            "interface");
 731                 wpa_driver_nl80211_finish_drv_init(drv);
 732                 return 1;
 733         }
 734
 735         return 0;
 736 }
 737
 738
 739 static void wpa_driver_nl80211_event_rtm_newlink(struct wpa_driver_nl80211_data *drv,
 740                                               void *ctx, struct nlmsghdr *h,
 741                                               size_t len)
 742 {
 743         struct ifinfomsg *ifi;
 744         int attrlen, _nlmsg_len, rta_len;
 745         struct rtattr * attr;
 746
 747         if (len < sizeof(*ifi))
 748                 return;
 749
 750         ifi = NLMSG_DATA(h);
 751
 752         if (!wpa_driver_nl80211_own_ifindex(drv, ifi->ifi_index, h)) {
 753                 wpa_printf(MSG_DEBUG, "Ignore event for foreign ifindex %d",
 754                            ifi->ifi_index);
 755                 return;
 756         }
 757
 758         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
 759                    "(%s%s%s%s)",
 760                    drv->operstate, ifi->ifi_flags,
 761                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
 762                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
 763                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
 764                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
 765         /*
 766          * Some drivers send the association event before the operup event--in
 767          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
 768          * fails. This will hit us when wpa_supplicant does not need to do
 769          * IEEE 802.1X authentication
 770          */
 771         if (drv->operstate == 1 &&
 772             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
 773             !(ifi->ifi_flags & IFF_RUNNING))
 774                 wpa_driver_nl80211_send_oper_ifla(drv, -1, IF_OPER_UP);
 775
 776         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
 777
 778         attrlen = h->nlmsg_len - _nlmsg_len;
 779         if (attrlen < 0)
 780                 return;
 781
 782         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
 783
 784         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 785         while (RTA_OK(attr, attrlen)) {
 786                 if (attr->rta_type == IFLA_WIRELESS) {
 787                         wpa_driver_nl80211_event_wireless(
 788                                 drv, ctx, ((char *) attr) + rta_len,
 789                                 attr->rta_len - rta_len);
 790                 } else if (attr->rta_type == IFLA_IFNAME) {
 791                         wpa_driver_nl80211_event_link(
 792                                 drv, ctx,
 793                                 ((char *) attr) + rta_len,
 794                                 attr->rta_len - rta_len, 0);
 795                 }
 796                 attr = RTA_NEXT(attr, attrlen);
 797         }
 798 }
 799
 800
 801 static void wpa_driver_nl80211_event_rtm_dellink(struct wpa_driver_nl80211_data *drv,
 802                                               void *ctx, struct nlmsghdr *h,
 803                                               size_t len)
 804 {
 805         struct ifinfomsg *ifi;
 806         int attrlen, _nlmsg_len, rta_len;
 807         struct rtattr * attr;
 808
 809         if (len < sizeof(*ifi))
 810                 return;
 811
 812         ifi = NLMSG_DATA(h);
 813
 814         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
 815
 816         attrlen = h->nlmsg_len - _nlmsg_len;
 817         if (attrlen < 0)
 818                 return;
 819
 820         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
 821
 822         rta_len = RTA_ALIGN(sizeof(struct rtattr));
 823         while (RTA_OK(attr, attrlen)) {
 824                 if (attr->rta_type == IFLA_IFNAME) {
 825                         wpa_driver_nl80211_event_link(
 826                                 drv, ctx,
 827                                 ((char *) attr) + rta_len,
 828                                 attr->rta_len - rta_len, 1);
 829                 }
 830                 attr = RTA_NEXT(attr, attrlen);
 831         }
 832 }
 833
 834
 835 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
 836                                           void *sock_ctx)
 837 {
 838         char buf[8192];
 839         int left;
 840         struct sockaddr_nl from;
 841         socklen_t fromlen;
 842         struct nlmsghdr *h;
 843         int max_events = 10;
 844
 845 try_again:
 846         fromlen = sizeof(from);
 847         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
 848                         (struct sockaddr *) &from, &fromlen);
 849         if (left < 0) {
 850                 if (errno != EINTR && errno != EAGAIN)
 851                         perror("recvfrom(netlink)");
 852                 return;
 853         }
 854
 855         h = (struct nlmsghdr *) buf;
 856         while (left >= (int) sizeof(*h)) {
 857                 int len, plen;
 858
 859                 len = h->nlmsg_len;
 860                 plen = len - sizeof(*h);
 861                 if (len > left || plen < 0) {
 862                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
 863                                    "len=%d left=%d plen=%d",
 864                                    len, left, plen);
 865                         break;
 866                 }
 867
 868                 switch (h->nlmsg_type) {
 869                 case RTM_NEWLINK:
 870                         wpa_driver_nl80211_event_rtm_newlink(eloop_ctx, sock_ctx,
 871                                                           h, plen);
 872                         break;
 873                 case RTM_DELLINK:
 874                         wpa_driver_nl80211_event_rtm_dellink(eloop_ctx, sock_ctx,
 875                                                           h, plen);
 876                         break;
 877                 }
 878
 879                 len = NLMSG_ALIGN(len);
 880                 left -= len;
 881                 h = (struct nlmsghdr *) ((char *) h + len);
 882         }
 883
 884         if (left > 0) {
 885                 wpa_printf(MSG_DEBUG, "%d extra bytes in the end of netlink "
 886                            "message", left);
 887         }
 888
 889         if (--max_events > 0) {
 890                 /*
 891                  * Try to receive all events in one eloop call in order to
 892                  * limit race condition on cases where AssocInfo event, Assoc
 893                  * event, and EAPOL frames are received more or less at the
 894                  * same time. We want to process the event messages first
 895                  * before starting EAPOL processing.
 896                  */
 897                 goto try_again;
 898         }
 899 }
 900
 901
 902 static int wpa_driver_nl80211_get_ifflags_ifname(struct wpa_driver_nl80211_data *drv,
 903                                               const char *ifname, int *flags)
 904 {
 905         struct ifreq ifr;
 906
 907         os_memset(&ifr, 0, sizeof(ifr));
 908         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 909         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, (caddr_t) &ifr) < 0) {
 910                 perror("ioctl[SIOCGIFFLAGS]");
 911                 return -1;
 912         }
 913         *flags = ifr.ifr_flags & 0xffff;
 914         return 0;
 915 }
 916
 917
 918 /**
 919  * wpa_driver_nl80211_get_ifflags - Get interface flags (SIOCGIFFLAGS)
 920  * @drv: driver_nl80211 private data
 921  * @flags: Pointer to returned flags value
 922  * Returns: 0 on success, -1 on failure
 923  */
 924 static int wpa_driver_nl80211_get_ifflags(struct wpa_driver_nl80211_data *drv,
 925                                           int *flags)
 926 {
 927         return wpa_driver_nl80211_get_ifflags_ifname(drv, drv->ifname, flags);
 928 }
 929
 930
 931 static int wpa_driver_nl80211_set_ifflags_ifname(
 932         struct wpa_driver_nl80211_data *drv,
 933         const char *ifname, int flags)
 934 {
 935         struct ifreq ifr;
 936
 937         os_memset(&ifr, 0, sizeof(ifr));
 938         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
 939         ifr.ifr_flags = flags & 0xffff;
 940         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, (caddr_t) &ifr) < 0) {
 941                 perror("SIOCSIFFLAGS");
 942                 return -1;
 943         }
 944         return 0;
 945 }
 946
 947
 948 /**
 949  * wpa_driver_nl80211_set_ifflags - Set interface flags (SIOCSIFFLAGS)
 950  * @drv: driver_nl80211 private data
 951  * @flags: New value for flags
 952  * Returns: 0 on success, -1 on failure
 953  */
 954 static int wpa_driver_nl80211_set_ifflags(struct wpa_driver_nl80211_data *drv,
 955                                           int flags)
 956 {
 957         return wpa_driver_nl80211_set_ifflags_ifname(drv, drv->ifname, flags);
 958 }
 959
 960
 961 /**
 962  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
 963  * @priv: driver_nl80211 private data
 964  * @alpha2_arg: country to which to switch to
 965  * Returns: 0 on success, -1 on failure
 966  *
 967  * This asks nl80211 to set the regulatory domain for given
 968  * country ISO / IEC alpha2.
 969  */
 970 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
 971 {
 972         struct wpa_driver_nl80211_data *drv = priv;
 973         char alpha2[3];
 974         struct nl_msg *msg;
 975
 976         msg = nlmsg_alloc();
 977         if (!msg)
 978                 goto nla_put_failure;
 979
 980         alpha2[0] = alpha2_arg[0];
 981         alpha2[1] = alpha2_arg[1];
 982         alpha2[2] = '\0';
 983
 984         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
 985                     0, NL80211_CMD_REQ_SET_REG, 0);
 986
 987         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
 988         if (send_and_recv_msgs(drv, msg, NULL, NULL))
 989                 return -EINVAL;
 990         return 0;
 991 nla_put_failure:
 992         return -EINVAL;
 993 }
 994
 995
 996 static int wpa_driver_nl80211_set_probe_req_ie(void *priv, const u8 *ies,
 997                                                size_t ies_len)
 998 {
 999         struct wpa_driver_nl80211_data *drv = priv;
1000         struct nl_msg *msg;
1001         int ret = -1;
1002
1003         msg = nlmsg_alloc();
1004         if (!msg)
1005                 return -ENOMEM;
1006
1007         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1008                     NL80211_CMD_SET_MGMT_EXTRA_IE, 0);
1009
1010         NLA_PUT_U8(msg, NL80211_ATTR_MGMT_SUBTYPE, 4 /* ProbeReq */);
1011         if (ies)
1012                 NLA_PUT(msg, NL80211_ATTR_IE, ies_len, ies);
1013
1014         ret = 0;
1015
1016         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1017
1018         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1019         return ret;
1020
1021 nla_put_failure:
1022         return -ENOBUFS;
1023 }
1024
1025
1026 #ifdef CONFIG_CLIENT_MLME
1027
1028 static int nl80211_set_vif(struct wpa_driver_nl80211_data *drv,
1029                            int drop_unencrypted, int userspace_mlme)
1030 {
1031 #ifdef NL80211_CMD_SET_VIF
1032         struct nl_msg *msg;
1033         int ret = -1;
1034
1035         msg = nlmsg_alloc();
1036         if (!msg)
1037                 return -ENOMEM;
1038
1039         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1040                     NL80211_CMD_SET_VIF, 0);
1041
1042         if (drop_unencrypted >= 0)
1043                 NLA_PUT_U8(msg, NL80211_ATTR_VIF_DROP_UNENCRYPTED,
1044                            drop_unencrypted);
1045         if (userspace_mlme >= 0)
1046                 NLA_PUT_U8(msg, NL80211_ATTR_VIF_USERSPACE_MLME,
1047                            userspace_mlme);
1048
1049         ret = 0;
1050
1051         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1052
1053         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1054         return ret;
1055
1056 nla_put_failure:
1057         return -ENOBUFS;
1058 #else /* NL80211_CMD_SET_VIF */
1059         return -1;
1060 #endif /* NL80211_CMD_SET_VIF */
1061 }
1062
1063
1064 static int wpa_driver_nl80211_set_userspace_mlme(
1065         struct wpa_driver_nl80211_data *drv, int enabled)
1066 {
1067         return nl80211_set_vif(drv, -1, enabled);
1068 }
1069
1070
1071 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
1072                                  int ifidx)
1073 {
1074         struct nl_msg *msg;
1075
1076         msg = nlmsg_alloc();
1077         if (!msg)
1078                 goto nla_put_failure;
1079
1080         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1081                     0, NL80211_CMD_DEL_INTERFACE, 0);
1082         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
1083         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
1084                 return;
1085 nla_put_failure:
1086         wpa_printf(MSG_ERROR, "nl80211: Failed to remove interface.");
1087 }
1088
1089
1090 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
1091                                 const char *ifname, enum nl80211_iftype iftype)
1092 {
1093         struct nl_msg *msg, *flags = NULL;
1094         int ifidx, err;
1095         int ret = -ENOBUFS;
1096
1097         msg = nlmsg_alloc();
1098         if (!msg)
1099                 return -1;
1100
1101         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1102                     0, NL80211_CMD_NEW_INTERFACE, 0);
1103         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
1104         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
1105         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
1106
1107         if (iftype == NL80211_IFTYPE_MONITOR) {
1108                 flags = nlmsg_alloc();
1109                 if (!flags)
1110                         goto nla_put_failure;
1111
1112                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1113
1114                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1115
1116                 nlmsg_free(flags);
1117
1118                 if (err)
1119                         goto nla_put_failure;
1120         }
1121
1122         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1123         if (ret) {
1124         nla_put_failure:
1125                 wpa_printf(MSG_ERROR, "nl80211: Failed to create interface %d",
1126                            ret);
1127                 return ret;
1128         }
1129
1130         ifidx = if_nametoindex(ifname);
1131         if (ifidx <= 0)
1132                 return -1;
1133
1134         return ifidx;
1135 }
1136
1137
1138 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1139 {
1140         struct wpa_driver_nl80211_data *drv = eloop_ctx;
1141         int len;
1142         unsigned char buf[3000];
1143         struct ieee80211_radiotap_iterator iter;
1144         int ret;
1145         int injected = 0, failed = 0, rxflags = 0;
1146         struct ieee80211_rx_status rx_status;
1147
1148         len = recv(sock, buf, sizeof(buf), 0);
1149         if (len < 0) {
1150                 perror("recv");
1151                 return;
1152         }
1153
1154         if (ieee80211_radiotap_iterator_init(&iter, (void *) buf, len)) {
1155                 wpa_printf(MSG_DEBUG, "nl80211: received invalid radiotap "
1156                            "frame");
1157                 return;
1158         }
1159
1160         os_memset(&rx_status, 0, sizeof(rx_status));
1161
1162         while (1) {
1163                 ret = ieee80211_radiotap_iterator_next(&iter);
1164                 if (ret == -ENOENT)
1165                         break;
1166                 if (ret) {
1167                         wpa_printf(MSG_DEBUG, "nl80211: received invalid "
1168                                    "radiotap frame (%d)", ret);
1169                         return;
1170                 }
1171                 switch (iter.this_arg_index) {
1172                 case IEEE80211_RADIOTAP_FLAGS:
1173                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1174                                 len -= 4;
1175                         break;
1176                 case IEEE80211_RADIOTAP_RX_FLAGS:
1177                         rxflags = 1;
1178                         break;
1179                 case IEEE80211_RADIOTAP_TX_FLAGS:
1180                         injected = 1;
1181                         failed = le_to_host16((*(u16 *) iter.this_arg)) &
1182                                 IEEE80211_RADIOTAP_F_TX_FAIL;
1183                         break;
1184                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1185                         break;
1186                 case IEEE80211_RADIOTAP_CHANNEL:
1187                         /* TODO convert from freq/flags to channel number
1188                          * rx_status.channel = XXX;
1189                         */
1190                         break;
1191                 case IEEE80211_RADIOTAP_RATE:
1192                         break;
1193                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1194                         rx_status.ssi = *iter.this_arg;
1195                         break;
1196                 }
1197         }
1198
1199         if (rxflags && injected)
1200                 return;
1201
1202         if (!injected) {
1203                 wpa_supplicant_sta_rx(drv->ctx, buf + iter.max_length,
1204                                       len - iter.max_length, &rx_status);
1205         } else if (failed) {
1206                 /* TX failure callback */
1207         } else {
1208                 /* TX success (ACK) callback */
1209         }
1210 }
1211
1212
1213 static int wpa_driver_nl80211_create_monitor_interface(
1214         struct wpa_driver_nl80211_data *drv)
1215 {
1216         char buf[IFNAMSIZ];
1217         struct sockaddr_ll ll;
1218         int optval, flags;
1219         socklen_t optlen;
1220
1221         os_snprintf(buf, IFNAMSIZ, "mon.%s", drv->ifname);
1222         buf[IFNAMSIZ - 1] = '\0';
1223
1224         drv->monitor_ifidx =
1225                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR);
1226
1227         if (drv->monitor_ifidx < 0)
1228                 return -1;
1229
1230         if (wpa_driver_nl80211_get_ifflags_ifname(drv, buf, &flags) != 0 ||
1231             wpa_driver_nl80211_set_ifflags_ifname(drv, buf, flags | IFF_UP) !=
1232             0) {
1233                 wpa_printf(MSG_ERROR, "nl80211: Could not set interface '%s' "
1234                            "UP", buf);
1235                 goto error;
1236         }
1237
1238         os_memset(&ll, 0, sizeof(ll));
1239         ll.sll_family = AF_PACKET;
1240         ll.sll_ifindex = drv->monitor_ifidx;
1241         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
1242         if (drv->monitor_sock < 0) {
1243                 perror("socket[PF_PACKET,SOCK_RAW]");
1244                 goto error;
1245         }
1246
1247         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
1248                  sizeof(ll)) < 0) {
1249                 perror("monitor socket bind");
1250                 goto error;
1251         }
1252
1253         optlen = sizeof(optval);
1254         optval = 20;
1255         if (setsockopt
1256             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
1257                 perror("Failed to set socket priority");
1258                 goto error;
1259         }
1260
1261         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
1262                                      drv, NULL)) {
1263                 wpa_printf(MSG_ERROR, "nl80211: Could not register monitor "
1264                            "read socket");
1265                 goto error;
1266         }
1267
1268         return 0;
1269
1270  error:
1271         nl80211_remove_iface(drv, drv->monitor_ifidx);
1272         return -1;
1273 }
1274
1275 #endif /* CONFIG_CLIENT_MLME */
1276
1277
1278 /**
1279  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
1280  * @ctx: context to be used when calling wpa_supplicant functions,
1281  * e.g., wpa_supplicant_event()
1282  * @ifname: interface name, e.g., wlan0
1283  * Returns: Pointer to private data, %NULL on failure
1284  */
1285 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname)
1286 {
1287         int s;
1288         struct sockaddr_nl local;
1289         struct wpa_driver_nl80211_data *drv;
1290
1291         drv = os_zalloc(sizeof(*drv));
1292         if (drv == NULL)
1293                 return NULL;
1294         drv->ctx = ctx;
1295         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
1296
1297         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
1298         if (drv->nl_cb == NULL) {
1299                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1300                            "callbacks");
1301                 goto err1;
1302         }
1303
1304         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
1305         if (drv->nl_handle == NULL) {
1306                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1307                            "callbacks");
1308                 goto err2;
1309         }
1310
1311         if (genl_connect(drv->nl_handle)) {
1312                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
1313                            "netlink");
1314                 goto err3;
1315         }
1316
1317         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
1318         if (drv->nl_cache == NULL) {
1319                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1320                            "netlink cache");
1321                 goto err3;
1322         }
1323
1324         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
1325         if (drv->nl80211 == NULL) {
1326                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
1327                            "found");
1328                 goto err4;
1329         }
1330
1331         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1332         if (drv->ioctl_sock < 0) {
1333                 perror("socket(PF_INET,SOCK_DGRAM)");
1334                 goto err5;
1335         }
1336
1337         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1338         if (s < 0) {
1339                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1340                 goto err6;
1341         }
1342
1343         os_memset(&local, 0, sizeof(local));
1344         local.nl_family = AF_NETLINK;
1345         local.nl_groups = RTMGRP_LINK;
1346         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1347                 perror("bind(netlink)");
1348                 close(s);
1349                 goto err6;
1350         }
1351
1352         eloop_register_read_sock(s, wpa_driver_nl80211_event_receive, drv,
1353                                  ctx);
1354         drv->event_sock = s;
1355
1356         wpa_driver_nl80211_finish_drv_init(drv);
1357
1358         return drv;
1359
1360 err6:
1361         close(drv->ioctl_sock);
1362 err5:
1363         genl_family_put(drv->nl80211);
1364 err4:
1365         nl_cache_free(drv->nl_cache);
1366 err3:
1367         nl_handle_destroy(drv->nl_handle);
1368 err2:
1369         nl_cb_put(drv->nl_cb);
1370 err1:
1371         os_free(drv);
1372         return NULL;
1373 }
1374
1375
1376 static void
1377 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
1378 {
1379         int flags;
1380
1381         if (wpa_driver_nl80211_get_ifflags(drv, &flags) != 0)
1382                 printf("Could not get interface '%s' flags\n", drv->ifname);
1383         else if (!(flags & IFF_UP)) {
1384                 if (wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP) != 0) {
1385                         printf("Could not set interface '%s' UP\n",
1386                                drv->ifname);
1387                 } else {
1388                         /*
1389                          * Wait some time to allow driver to initialize before
1390                          * starting configuring the driver. This seems to be
1391                          * needed at least some drivers that load firmware etc.
1392                          * when the interface is set up.
1393                          */
1394                         wpa_printf(MSG_DEBUG, "Interface %s set UP - waiting "
1395                                    "a second for the driver to complete "
1396                                    "initialization", drv->ifname);
1397                         sleep(1);
1398                 }
1399         }
1400
1401         /*
1402          * Make sure that the driver does not have any obsolete PMKID entries.
1403          */
1404         wpa_driver_nl80211_flush_pmkid(drv);
1405
1406         if (wpa_driver_nl80211_set_mode(drv, 0) < 0) {
1407                 printf("Could not configure driver to use managed mode\n");
1408         }
1409
1410         wpa_driver_nl80211_get_range(drv);
1411
1412         drv->ifindex = if_nametoindex(drv->ifname);
1413
1414         wpa_driver_nl80211_send_oper_ifla(drv, 1, IF_OPER_DORMANT);
1415 }
1416
1417
1418 /**
1419  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
1420  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1421  *
1422  * Shut down driver interface and processing of driver events. Free
1423  * private data buffer if one was allocated in wpa_driver_nl80211_init().
1424  */
1425 static void wpa_driver_nl80211_deinit(void *priv)
1426 {
1427         struct wpa_driver_nl80211_data *drv = priv;
1428         int flags;
1429
1430 #ifdef CONFIG_CLIENT_MLME
1431         if (drv->monitor_sock >= 0) {
1432                 eloop_unregister_read_sock(drv->monitor_sock);
1433                 close(drv->monitor_sock);
1434         }
1435         if (drv->monitor_ifidx > 0)
1436                 nl80211_remove_iface(drv, drv->monitor_ifidx);
1437         if (drv->capa.flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
1438                 wpa_driver_nl80211_set_userspace_mlme(drv, 0);
1439 #endif /* CONFIG_CLIENT_MLME */
1440
1441         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1442
1443         /*
1444          * Clear possibly configured driver parameters in order to make it
1445          * easier to use the driver after wpa_supplicant has been terminated.
1446          */
1447         (void) wpa_driver_nl80211_set_bssid(drv,
1448                                          (u8 *) "\x00\x00\x00\x00\x00\x00");
1449
1450         wpa_driver_nl80211_send_oper_ifla(priv, 0, IF_OPER_UP);
1451
1452         eloop_unregister_read_sock(drv->event_sock);
1453
1454         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0)
1455                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
1456
1457         close(drv->event_sock);
1458         close(drv->ioctl_sock);
1459         os_free(drv->assoc_req_ies);
1460         os_free(drv->assoc_resp_ies);
1461
1462         genl_family_put(drv->nl80211);
1463         nl_cache_free(drv->nl_cache);
1464         nl_handle_destroy(drv->nl_handle);
1465         nl_cb_put(drv->nl_cb);
1466
1467         os_free(drv);
1468 }
1469
1470
1471 /**
1472  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
1473  * @eloop_ctx: Unused
1474  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
1475  *
1476  * This function can be used as registered timeout when starting a scan to
1477  * generate a scan completed event if the driver does not report this.
1478  */
1479 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1480 {
1481         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
1482         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
1483 }
1484
1485
1486 /**
1487  * wpa_driver_nl80211_scan - Request the driver to initiate scan
1488  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1489  * @ssid: Specific SSID to scan for (ProbeReq) or %NULL to scan for
1490  *      all SSIDs (either active scan with broadcast SSID or passive
1491  *      scan
1492  * @ssid_len: Length of the SSID
1493  * Returns: 0 on success, -1 on failure
1494  */
1495 static int wpa_driver_nl80211_scan(void *priv, const u8 *ssid, size_t ssid_len)
1496 {
1497         struct wpa_driver_nl80211_data *drv = priv;
1498         struct iwreq iwr;
1499         int ret = 0, timeout;
1500         struct iw_scan_req req;
1501
1502         if (ssid_len > IW_ESSID_MAX_SIZE) {
1503                 wpa_printf(MSG_DEBUG, "%s: too long SSID (%lu)",
1504                            __FUNCTION__, (unsigned long) ssid_len);
1505                 return -1;
1506         }
1507
1508         os_memset(&iwr, 0, sizeof(iwr));
1509         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1510
1511         if (ssid && ssid_len) {
1512                 os_memset(&req, 0, sizeof(req));
1513                 req.essid_len = ssid_len;
1514                 req.bssid.sa_family = ARPHRD_ETHER;
1515                 os_memset(req.bssid.sa_data, 0xff, ETH_ALEN);
1516                 os_memcpy(req.essid, ssid, ssid_len);
1517                 iwr.u.data.pointer = (caddr_t) &req;
1518                 iwr.u.data.length = sizeof(req);
1519                 iwr.u.data.flags = IW_SCAN_THIS_ESSID;
1520         }
1521
1522         if (ioctl(drv->ioctl_sock, SIOCSIWSCAN, &iwr) < 0) {
1523                 perror("ioctl[SIOCSIWSCAN]");
1524                 ret = -1;
1525         }
1526
1527         /* Not all drivers generate "scan completed" wireless event, so try to
1528          * read results after a timeout. */
1529         timeout = 5;
1530         if (drv->scan_complete_events) {
1531                 /*
1532                  * The driver seems to deliver SIOCGIWSCAN events to notify
1533                  * when scan is complete, so use longer timeout to avoid race
1534                  * conditions with scanning and following association request.
1535                  */
1536                 timeout = 30;
1537         }
1538         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
1539                    "seconds", ret, timeout);
1540         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1541         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout, drv,
1542                                drv->ctx);
1543
1544         return ret;
1545 }
1546
1547
1548 static u8 * wpa_driver_nl80211_giwscan(struct wpa_driver_nl80211_data *drv,
1549                                     size_t *len)
1550 {
1551         struct iwreq iwr;
1552         u8 *res_buf;
1553         size_t res_buf_len;
1554
1555         res_buf_len = IW_SCAN_MAX_DATA;
1556         for (;;) {
1557                 res_buf = os_malloc(res_buf_len);
1558                 if (res_buf == NULL)
1559                         return NULL;
1560                 os_memset(&iwr, 0, sizeof(iwr));
1561                 os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1562                 iwr.u.data.pointer = res_buf;
1563                 iwr.u.data.length = res_buf_len;
1564
1565                 if (ioctl(drv->ioctl_sock, SIOCGIWSCAN, &iwr) == 0)
1566                         break;
1567
1568                 if (errno == E2BIG && res_buf_len < 100000) {
1569                         os_free(res_buf);
1570                         res_buf = NULL;
1571                         res_buf_len *= 2;
1572                         wpa_printf(MSG_DEBUG, "Scan results did not fit - "
1573                                    "trying larger buffer (%lu bytes)",
1574                                    (unsigned long) res_buf_len);
1575                 } else {
1576                         perror("ioctl[SIOCGIWSCAN]");
1577                         os_free(res_buf);
1578                         return NULL;
1579                 }
1580         }
1581
1582         if (iwr.u.data.length > res_buf_len) {
1583                 os_free(res_buf);
1584                 return NULL;
1585         }
1586         *len = iwr.u.data.length;
1587
1588         return res_buf;
1589 }
1590
1591
1592 /*
1593  * Data structure for collecting WEXT scan results. This is needed to allow
1594  * the various methods of reporting IEs to be combined into a single IE buffer.
1595  */
1596 struct wext_scan_data {
1597         struct wpa_scan_res res;
1598         u8 *ie;
1599         size_t ie_len;
1600         u8 ssid[32];
1601         size_t ssid_len;
1602         int maxrate;
1603 };
1604
1605
1606 static void wext_get_scan_mode(struct iw_event *iwe,
1607                                struct wext_scan_data *res)
1608 {
1609         if (iwe->u.mode == IW_MODE_ADHOC)
1610                 res->res.caps |= IEEE80211_CAP_IBSS;
1611         else if (iwe->u.mode == IW_MODE_MASTER || iwe->u.mode == IW_MODE_INFRA)
1612                 res->res.caps |= IEEE80211_CAP_ESS;
1613 }
1614
1615
1616 static void wext_get_scan_ssid(struct iw_event *iwe,
1617                                struct wext_scan_data *res, char *custom,
1618                                char *end)
1619 {
1620         int ssid_len = iwe->u.essid.length;
1621         if (custom + ssid_len > end)
1622                 return;
1623         if (iwe->u.essid.flags &&
1624             ssid_len > 0 &&
1625             ssid_len <= IW_ESSID_MAX_SIZE) {
1626                 os_memcpy(res->ssid, custom, ssid_len);
1627                 res->ssid_len = ssid_len;
1628         }
1629 }
1630
1631
1632 static void wext_get_scan_freq(struct iw_event *iwe,
1633                                struct wext_scan_data *res)
1634 {
1635         int divi = 1000000, i;
1636
1637         if (iwe->u.freq.e == 0) {
1638                 /*
1639                  * Some drivers do not report frequency, but a channel.
1640                  * Try to map this to frequency by assuming they are using
1641                  * IEEE 802.11b/g.  But don't overwrite a previously parsed
1642                  * frequency if the driver sends both frequency and channel,
1643                  * since the driver may be sending an A-band channel that we
1644                  * don't handle here.
1645                  */
1646
1647                 if (res->res.freq)
1648                         return;
1649
1650                 if (iwe->u.freq.m >= 1 && iwe->u.freq.m <= 13) {
1651                         res->res.freq = 2407 + 5 * iwe->u.freq.m;
1652                         return;
1653                 } else if (iwe->u.freq.m == 14) {
1654                         res->res.freq = 2484;
1655                         return;
1656                 }
1657         }
1658
1659         if (iwe->u.freq.e > 6) {
1660                 wpa_printf(MSG_DEBUG, "Invalid freq in scan results (BSSID="
1661                            MACSTR " m=%d e=%d)",
1662                            MAC2STR(res->res.bssid), iwe->u.freq.m,
1663                            iwe->u.freq.e);
1664                 return;
1665         }
1666
1667         for (i = 0; i < iwe->u.freq.e; i++)
1668                 divi /= 10;
1669         res->res.freq = iwe->u.freq.m / divi;
1670 }
1671
1672
1673 static void wext_get_scan_qual(struct iw_event *iwe,
1674                                struct wext_scan_data *res)
1675 {
1676         res->res.qual = iwe->u.qual.qual;
1677         res->res.noise = iwe->u.qual.noise;
1678         res->res.level = iwe->u.qual.level;
1679 }
1680
1681
1682 static void wext_get_scan_encode(struct iw_event *iwe,
1683                                  struct wext_scan_data *res)
1684 {
1685         if (!(iwe->u.data.flags & IW_ENCODE_DISABLED))
1686                 res->res.caps |= IEEE80211_CAP_PRIVACY;
1687 }
1688
1689
1690 static void wext_get_scan_rate(struct iw_event *iwe,
1691                                struct wext_scan_data *res, char *pos,
1692                                char *end)
1693 {
1694         int maxrate;
1695         char *custom = pos + IW_EV_LCP_LEN;
1696         struct iw_param p;
1697         size_t clen;
1698
1699         clen = iwe->len;
1700         if (custom + clen > end)
1701                 return;
1702         maxrate = 0;
1703         while (((ssize_t) clen) >= (ssize_t) sizeof(struct iw_param)) {
1704                 /* Note: may be misaligned, make a local, aligned copy */
1705                 os_memcpy(&p, custom, sizeof(struct iw_param));
1706                 if (p.value > maxrate)
1707                         maxrate = p.value;
1708                 clen -= sizeof(struct iw_param);
1709                 custom += sizeof(struct iw_param);
1710         }
1711
1712         /* Convert the maxrate from WE-style (b/s units) to
1713          * 802.11 rates (500000 b/s units).
1714          */
1715         res->maxrate = maxrate / 500000;
1716 }
1717
1718
1719 static void wext_get_scan_iwevgenie(struct iw_event *iwe,
1720                                     struct wext_scan_data *res, char *custom,
1721                                     char *end)
1722 {
1723         char *genie, *gpos, *gend;
1724         u8 *tmp;
1725
1726         if (iwe->u.data.length == 0)
1727                 return;
1728
1729         gpos = genie = custom;
1730         gend = genie + iwe->u.data.length;
1731         if (gend > end) {
1732                 wpa_printf(MSG_INFO, "IWEVGENIE overflow");
1733                 return;
1734         }
1735
1736         tmp = os_realloc(res->ie, res->ie_len + gend - gpos);
1737         if (tmp == NULL)
1738                 return;
1739         os_memcpy(tmp + res->ie_len, gpos, gend - gpos);
1740         res->ie = tmp;
1741         res->ie_len += gend - gpos;
1742 }
1743
1744
1745 static void wext_get_scan_custom(struct iw_event *iwe,
1746                                  struct wext_scan_data *res, char *custom,
1747                                  char *end)
1748 {
1749         size_t clen;
1750         u8 *tmp;
1751
1752         clen = iwe->u.data.length;
1753         if (custom + clen > end)
1754                 return;
1755
1756         if (clen > 7 && os_strncmp(custom, "wpa_ie=", 7) == 0) {
1757                 char *spos;
1758                 int bytes;
1759                 spos = custom + 7;
1760                 bytes = custom + clen - spos;
1761                 if (bytes & 1 || bytes == 0)
1762                         return;
1763                 bytes /= 2;
1764                 tmp = os_realloc(res->ie, res->ie_len + bytes);
1765                 if (tmp == NULL)
1766                         return;
1767                 hexstr2bin(spos, tmp + res->ie_len, bytes);
1768                 res->ie = tmp;
1769                 res->ie_len += bytes;
1770         } else if (clen > 7 && os_strncmp(custom, "rsn_ie=", 7) == 0) {
1771                 char *spos;
1772                 int bytes;
1773                 spos = custom + 7;
1774                 bytes = custom + clen - spos;
1775                 if (bytes & 1 || bytes == 0)
1776                         return;
1777                 bytes /= 2;
1778                 tmp = os_realloc(res->ie, res->ie_len + bytes);
1779                 if (tmp == NULL)
1780                         return;
1781                 hexstr2bin(spos, tmp + res->ie_len, bytes);
1782                 res->ie = tmp;
1783                 res->ie_len += bytes;
1784         } else if (clen > 4 && os_strncmp(custom, "tsf=", 4) == 0) {
1785                 char *spos;
1786                 int bytes;
1787                 u8 bin[8];
1788                 spos = custom + 4;
1789                 bytes = custom + clen - spos;
1790                 if (bytes != 16) {
1791                         wpa_printf(MSG_INFO, "Invalid TSF length (%d)", bytes);
1792                         return;
1793                 }
1794                 bytes /= 2;
1795                 hexstr2bin(spos, bin, bytes);
1796                 res->res.tsf += WPA_GET_BE64(bin);
1797         }
1798 }
1799
1800
1801 static int wext_19_iw_point(struct wpa_driver_nl80211_data *drv, u16 cmd)
1802 {
1803         return drv->we_version_compiled > 18 &&
1804                 (cmd == SIOCGIWESSID || cmd == SIOCGIWENCODE ||
1805                  cmd == IWEVGENIE || cmd == IWEVCUSTOM);
1806 }
1807
1808
1809 static void wpa_driver_nl80211_add_scan_entry(struct wpa_scan_results *res,
1810                                            struct wext_scan_data *data)
1811 {
1812         struct wpa_scan_res **tmp;
1813         struct wpa_scan_res *r;
1814         size_t extra_len;
1815         u8 *pos, *end, *ssid_ie = NULL, *rate_ie = NULL;
1816
1817         /* Figure out whether we need to fake any IEs */
1818         pos = data->ie;
1819         end = pos + data->ie_len;
1820         while (pos && pos + 1 < end) {
1821                 if (pos + 2 + pos[1] > end)
1822                         break;
1823                 if (pos[0] == WLAN_EID_SSID)
1824                         ssid_ie = pos;
1825                 else if (pos[0] == WLAN_EID_SUPP_RATES)
1826                         rate_ie = pos;
1827                 else if (pos[0] == WLAN_EID_EXT_SUPP_RATES)
1828                         rate_ie = pos;
1829                 pos += 2 + pos[1];
1830         }
1831
1832         extra_len = 0;
1833         if (ssid_ie == NULL)
1834                 extra_len += 2 + data->ssid_len;
1835         if (rate_ie == NULL && data->maxrate)
1836                 extra_len += 3;
1837
1838         r = os_zalloc(sizeof(*r) + extra_len + data->ie_len);
1839         if (r == NULL)
1840                 return;
1841         os_memcpy(r, &data->res, sizeof(*r));
1842         r->ie_len = extra_len + data->ie_len;
1843         pos = (u8 *) (r + 1);
1844         if (ssid_ie == NULL) {
1845                 /*
1846                  * Generate a fake SSID IE since the driver did not report
1847                  * a full IE list.
1848                  */
1849                 *pos++ = WLAN_EID_SSID;
1850                 *pos++ = data->ssid_len;
1851                 os_memcpy(pos, data->ssid, data->ssid_len);
1852                 pos += data->ssid_len;
1853         }
1854         if (rate_ie == NULL && data->maxrate) {
1855                 /*
1856                  * Generate a fake Supported Rates IE since the driver did not
1857                  * report a full IE list.
1858                  */
1859                 *pos++ = WLAN_EID_SUPP_RATES;
1860                 *pos++ = 1;
1861                 *pos++ = data->maxrate;
1862         }
1863         if (data->ie)
1864                 os_memcpy(pos, data->ie, data->ie_len);
1865
1866         tmp = os_realloc(res->res,
1867                          (res->num + 1) * sizeof(struct wpa_scan_res *));
1868         if (tmp == NULL) {
1869                 os_free(r);
1870                 return;
1871         }
1872         tmp[res->num++] = r;
1873         res->res = tmp;
1874 }
1875
1876
1877 /**
1878  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
1879  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1880  * Returns: Scan results on success, -1 on failure
1881  */
1882 static struct wpa_scan_results *
1883 wpa_driver_nl80211_get_scan_results(void *priv)
1884 {
1885         struct wpa_driver_nl80211_data *drv = priv;
1886         size_t ap_num = 0, len;
1887         int first;
1888         u8 *res_buf;
1889         struct iw_event iwe_buf, *iwe = &iwe_buf;
1890         char *pos, *end, *custom;
1891         struct wpa_scan_results *res;
1892         struct wext_scan_data data;
1893
1894         res_buf = wpa_driver_nl80211_giwscan(drv, &len);
1895         if (res_buf == NULL)
1896                 return NULL;
1897
1898         ap_num = 0;
1899         first = 1;
1900
1901         res = os_zalloc(sizeof(*res));
1902         if (res == NULL) {
1903                 os_free(res_buf);
1904                 return NULL;
1905         }
1906
1907         pos = (char *) res_buf;
1908         end = (char *) res_buf + len;
1909         os_memset(&data, 0, sizeof(data));
1910
1911         while (pos + IW_EV_LCP_LEN <= end) {
1912                 /* Event data may be unaligned, so make a local, aligned copy
1913                  * before processing. */
1914                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
1915                 if (iwe->len <= IW_EV_LCP_LEN)
1916                         break;
1917
1918                 custom = pos + IW_EV_POINT_LEN;
1919                 if (wext_19_iw_point(drv, iwe->cmd)) {
1920                         /* WE-19 removed the pointer from struct iw_point */
1921                         char *dpos = (char *) &iwe_buf.u.data.length;
1922                         int dlen = dpos - (char *) &iwe_buf;
1923                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
1924                                   sizeof(struct iw_event) - dlen);
1925                 } else {
1926                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
1927                         custom += IW_EV_POINT_OFF;
1928                 }
1929
1930                 switch (iwe->cmd) {
1931                 case SIOCGIWAP:
1932                         if (!first)
1933                                 wpa_driver_nl80211_add_scan_entry(res, &data);
1934                         first = 0;
1935                         os_free(data.ie);
1936                         os_memset(&data, 0, sizeof(data));
1937                         os_memcpy(data.res.bssid,
1938                                   iwe->u.ap_addr.sa_data, ETH_ALEN);
1939                         break;
1940                 case SIOCGIWMODE:
1941                         wext_get_scan_mode(iwe, &data);
1942                         break;
1943                 case SIOCGIWESSID:
1944                         wext_get_scan_ssid(iwe, &data, custom, end);
1945                         break;
1946                 case SIOCGIWFREQ:
1947                         wext_get_scan_freq(iwe, &data);
1948                         break;
1949                 case IWEVQUAL:
1950                         wext_get_scan_qual(iwe, &data);
1951                         break;
1952                 case SIOCGIWENCODE:
1953                         wext_get_scan_encode(iwe, &data);
1954                         break;
1955                 case SIOCGIWRATE:
1956                         wext_get_scan_rate(iwe, &data, pos, end);
1957                         break;
1958                 case IWEVGENIE:
1959                         wext_get_scan_iwevgenie(iwe, &data, custom, end);
1960                         break;
1961                 case IWEVCUSTOM:
1962                         wext_get_scan_custom(iwe, &data, custom, end);
1963                         break;
1964                 }
1965
1966                 pos += iwe->len;
1967         }
1968         os_free(res_buf);
1969         res_buf = NULL;
1970         if (!first)
1971                 wpa_driver_nl80211_add_scan_entry(res, &data);
1972         os_free(data.ie);
1973
1974         wpa_printf(MSG_DEBUG, "Received %lu bytes of scan results (%lu BSSes)",
1975                    (unsigned long) len, (unsigned long) res->num);
1976
1977         return res;
1978 }
1979
1980
1981 static int wpa_driver_nl80211_get_range(void *priv)
1982 {
1983         struct wpa_driver_nl80211_data *drv = priv;
1984         struct iw_range *range;
1985         struct iwreq iwr;
1986         int minlen;
1987         size_t buflen;
1988
1989         /*
1990          * Use larger buffer than struct iw_range in order to allow the
1991          * structure to grow in the future.
1992          */
1993         buflen = sizeof(struct iw_range) + 500;
1994         range = os_zalloc(buflen);
1995         if (range == NULL)
1996                 return -1;
1997
1998         os_memset(&iwr, 0, sizeof(iwr));
1999         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2000         iwr.u.data.pointer = (caddr_t) range;
2001         iwr.u.data.length = buflen;
2002
2003         minlen = ((char *) &range->enc_capa) - (char *) range +
2004                 sizeof(range->enc_capa);
2005
2006         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
2007                 perror("ioctl[SIOCGIWRANGE]");
2008                 os_free(range);
2009                 return -1;
2010         } else if (iwr.u.data.length >= minlen &&
2011                    range->we_version_compiled >= 18) {
2012                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
2013                            "WE(source)=%d enc_capa=0x%x",
2014                            range->we_version_compiled,
2015                            range->we_version_source,
2016                            range->enc_capa);
2017                 drv->has_capability = 1;
2018                 drv->we_version_compiled = range->we_version_compiled;
2019                 if (range->enc_capa & IW_ENC_CAPA_WPA) {
2020                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA |
2021                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
2022                 }
2023                 if (range->enc_capa & IW_ENC_CAPA_WPA2) {
2024                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
2025                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
2026                 }
2027                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40 |
2028                         WPA_DRIVER_CAPA_ENC_WEP104;
2029                 if (range->enc_capa & IW_ENC_CAPA_CIPHER_TKIP)
2030                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
2031                 if (range->enc_capa & IW_ENC_CAPA_CIPHER_CCMP)
2032                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
2033                 wpa_printf(MSG_DEBUG, "  capabilities: key_mgmt 0x%x enc 0x%x",
2034                            drv->capa.key_mgmt, drv->capa.enc);
2035         } else {
2036                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: too old (short) data - "
2037                            "assuming WPA is not supported");
2038         }
2039
2040         os_free(range);
2041         return 0;
2042 }
2043
2044
2045 static int wpa_driver_nl80211_set_wpa(void *priv, int enabled)
2046 {
2047         struct wpa_driver_nl80211_data *drv = priv;
2048         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2049
2050         return wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_WPA_ENABLED,
2051                                               enabled);
2052 }
2053
2054
2055 static int wpa_driver_nl80211_set_key(void *priv, wpa_alg alg,
2056                                       const u8 *addr, int key_idx,
2057                                       int set_tx, const u8 *seq,
2058                                       size_t seq_len,
2059                                       const u8 *key, size_t key_len)
2060 {
2061         struct wpa_driver_nl80211_data *drv = priv;
2062         int err;
2063         struct nl_msg *msg;
2064
2065         wpa_printf(MSG_DEBUG, "%s: alg=%d addr=%p key_idx=%d set_tx=%d "
2066                    "seq_len=%lu key_len=%lu",
2067                    __func__, alg, addr, key_idx, set_tx,
2068                    (unsigned long) seq_len, (unsigned long) key_len);
2069
2070         msg = nlmsg_alloc();
2071         if (msg == NULL)
2072                 return -1;
2073
2074         if (alg == WPA_ALG_NONE) {
2075                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2076                             NL80211_CMD_DEL_KEY, 0);
2077         } else {
2078                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2079                             NL80211_CMD_NEW_KEY, 0);
2080                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
2081                 switch (alg) {
2082                 case WPA_ALG_WEP:
2083                         if (key_len == 5)
2084                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2085                                             0x000FAC01);
2086                         else
2087                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2088                                             0x000FAC05);
2089                         break;
2090                 case WPA_ALG_TKIP:
2091                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
2092                         break;
2093                 case WPA_ALG_CCMP:
2094                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
2095                         break;
2096                 default:
2097                         nlmsg_free(msg);
2098                         return -1;
2099                 }
2100         }
2101
2102         if (addr && os_memcmp(addr, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
2103         {
2104                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
2105                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2106         }
2107         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2108         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2109
2110         err = send_and_recv_msgs(drv, msg, NULL, NULL);
2111         if (err) {
2112                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d", err);
2113                 return -1;
2114         }
2115
2116         if (set_tx && alg != WPA_ALG_NONE) {
2117                 msg = nlmsg_alloc();
2118                 if (msg == NULL)
2119                         return -1;
2120
2121                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2122                             0, NL80211_CMD_SET_KEY, 0);
2123                 NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2124                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2125                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
2126
2127                 err = send_and_recv_msgs(drv, msg, NULL, NULL);
2128                 if (err) {
2129                         wpa_printf(MSG_DEBUG, "nl80211: set default key "
2130                                    "failed; err=%d", err);
2131                         return -1;
2132                 }
2133         }
2134
2135         return 0;
2136
2137 nla_put_failure:
2138         return -ENOBUFS;
2139 }
2140
2141
2142 static int wpa_driver_nl80211_set_countermeasures(void *priv,
2143                                                int enabled)
2144 {
2145         struct wpa_driver_nl80211_data *drv = priv;
2146         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2147         return wpa_driver_nl80211_set_auth_param(drv,
2148                                               IW_AUTH_TKIP_COUNTERMEASURES,
2149                                               enabled);
2150 }
2151
2152
2153 static int wpa_driver_nl80211_set_drop_unencrypted(void *priv,
2154                                                 int enabled)
2155 {
2156         struct wpa_driver_nl80211_data *drv = priv;
2157         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2158         drv->use_crypt = enabled;
2159         return wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED,
2160                                               enabled);
2161 }
2162
2163
2164 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
2165                                 const u8 *addr, int cmd, int reason_code)
2166 {
2167         struct iwreq iwr;
2168         struct iw_mlme mlme;
2169         int ret = 0;
2170
2171         os_memset(&iwr, 0, sizeof(iwr));
2172         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2173         os_memset(&mlme, 0, sizeof(mlme));
2174         mlme.cmd = cmd;
2175         mlme.reason_code = reason_code;
2176         mlme.addr.sa_family = ARPHRD_ETHER;
2177         os_memcpy(mlme.addr.sa_data, addr, ETH_ALEN);
2178         iwr.u.data.pointer = (caddr_t) &mlme;
2179         iwr.u.data.length = sizeof(mlme);
2180
2181         if (ioctl(drv->ioctl_sock, SIOCSIWMLME, &iwr) < 0) {
2182                 perror("ioctl[SIOCSIWMLME]");
2183                 ret = -1;
2184         }
2185
2186         return ret;
2187 }
2188
2189
2190 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
2191                                           int reason_code)
2192 {
2193         struct wpa_driver_nl80211_data *drv = priv;
2194         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2195         return wpa_driver_nl80211_mlme(drv, addr, IW_MLME_DEAUTH, reason_code);
2196 }
2197
2198
2199 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
2200                                         int reason_code)
2201 {
2202         struct wpa_driver_nl80211_data *drv = priv;
2203         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2204         return wpa_driver_nl80211_mlme(drv, addr, IW_MLME_DISASSOC,
2205                                     reason_code);
2206 }
2207
2208
2209 static int wpa_driver_nl80211_set_gen_ie(void *priv, const u8 *ie,
2210                                       size_t ie_len)
2211 {
2212         struct wpa_driver_nl80211_data *drv = priv;
2213         struct iwreq iwr;
2214         int ret = 0;
2215
2216         os_memset(&iwr, 0, sizeof(iwr));
2217         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2218         iwr.u.data.pointer = (caddr_t) ie;
2219         iwr.u.data.length = ie_len;
2220
2221         if (ioctl(drv->ioctl_sock, SIOCSIWGENIE, &iwr) < 0) {
2222                 perror("ioctl[SIOCSIWGENIE]");
2223                 ret = -1;
2224         }
2225
2226         return ret;
2227 }
2228
2229
2230 static int wpa_driver_nl80211_cipher2wext(int cipher)
2231 {
2232         switch (cipher) {
2233         case CIPHER_NONE:
2234                 return IW_AUTH_CIPHER_NONE;
2235         case CIPHER_WEP40:
2236                 return IW_AUTH_CIPHER_WEP40;
2237         case CIPHER_TKIP:
2238                 return IW_AUTH_CIPHER_TKIP;
2239         case CIPHER_CCMP:
2240                 return IW_AUTH_CIPHER_CCMP;
2241         case CIPHER_WEP104:
2242                 return IW_AUTH_CIPHER_WEP104;
2243         default:
2244                 return 0;
2245         }
2246 }
2247
2248
2249 static int wpa_driver_nl80211_keymgmt2wext(int keymgmt)
2250 {
2251         switch (keymgmt) {
2252         case KEY_MGMT_802_1X:
2253         case KEY_MGMT_802_1X_NO_WPA:
2254                 return IW_AUTH_KEY_MGMT_802_1X;
2255         case KEY_MGMT_PSK:
2256                 return IW_AUTH_KEY_MGMT_PSK;
2257         default:
2258                 return 0;
2259         }
2260 }
2261
2262
2263 static int
2264 wpa_driver_nl80211_auth_alg_fallback(struct wpa_driver_nl80211_data *drv,
2265                                   struct wpa_driver_associate_params *params)
2266 {
2267         struct iwreq iwr;
2268         int ret = 0;
2269
2270         wpa_printf(MSG_DEBUG, "WEXT: Driver did not support "
2271                    "SIOCSIWAUTH for AUTH_ALG, trying SIOCSIWENCODE");
2272
2273         os_memset(&iwr, 0, sizeof(iwr));
2274         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2275         /* Just changing mode, not actual keys */
2276         iwr.u.encoding.flags = 0;
2277         iwr.u.encoding.pointer = (caddr_t) NULL;
2278         iwr.u.encoding.length = 0;
2279
2280         /*
2281          * Note: IW_ENCODE_{OPEN,RESTRICTED} can be interpreted to mean two
2282          * different things. Here they are used to indicate Open System vs.
2283          * Shared Key authentication algorithm. However, some drivers may use
2284          * them to select between open/restricted WEP encrypted (open = allow
2285          * both unencrypted and encrypted frames; restricted = only allow
2286          * encrypted frames).
2287          */
2288
2289         if (!drv->use_crypt) {
2290                 iwr.u.encoding.flags |= IW_ENCODE_DISABLED;
2291         } else {
2292                 if (params->auth_alg & AUTH_ALG_OPEN_SYSTEM)
2293                         iwr.u.encoding.flags |= IW_ENCODE_OPEN;
2294                 if (params->auth_alg & AUTH_ALG_SHARED_KEY)
2295                         iwr.u.encoding.flags |= IW_ENCODE_RESTRICTED;
2296         }
2297
2298         if (ioctl(drv->ioctl_sock, SIOCSIWENCODE, &iwr) < 0) {
2299                 perror("ioctl[SIOCSIWENCODE]");
2300                 ret = -1;
2301         }
2302
2303         return ret;
2304 }
2305
2306
2307 static int wpa_driver_nl80211_associate(
2308         void *priv, struct wpa_driver_associate_params *params)
2309 {
2310         struct wpa_driver_nl80211_data *drv = priv;
2311         int ret = 0;
2312         int allow_unencrypted_eapol;
2313         int value;
2314
2315         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2316
2317         /*
2318          * If the driver did not support SIOCSIWAUTH, fallback to
2319          * SIOCSIWENCODE here.
2320          */
2321         if (drv->auth_alg_fallback &&
2322             wpa_driver_nl80211_auth_alg_fallback(drv, params) < 0)
2323                 ret = -1;
2324
2325         if (!params->bssid &&
2326             wpa_driver_nl80211_set_bssid(drv, NULL) < 0)
2327                 ret = -1;
2328
2329         /* TODO: should consider getting wpa version and cipher/key_mgmt suites
2330          * from configuration, not from here, where only the selected suite is
2331          * available */
2332         if (wpa_driver_nl80211_set_gen_ie(drv, params->wpa_ie, params->wpa_ie_len)
2333             < 0)
2334                 ret = -1;
2335         if (params->wpa_ie == NULL || params->wpa_ie_len == 0)
2336                 value = IW_AUTH_WPA_VERSION_DISABLED;
2337         else if (params->wpa_ie[0] == WLAN_EID_RSN)
2338                 value = IW_AUTH_WPA_VERSION_WPA2;
2339         else
2340                 value = IW_AUTH_WPA_VERSION_WPA;
2341         if (wpa_driver_nl80211_set_auth_param(drv,
2342                                            IW_AUTH_WPA_VERSION, value) < 0)
2343                 ret = -1;
2344         value = wpa_driver_nl80211_cipher2wext(params->pairwise_suite);
2345         if (wpa_driver_nl80211_set_auth_param(drv,
2346                                            IW_AUTH_CIPHER_PAIRWISE, value) < 0)
2347                 ret = -1;
2348         value = wpa_driver_nl80211_cipher2wext(params->group_suite);
2349         if (wpa_driver_nl80211_set_auth_param(drv,
2350                                            IW_AUTH_CIPHER_GROUP, value) < 0)
2351                 ret = -1;
2352         value = wpa_driver_nl80211_keymgmt2wext(params->key_mgmt_suite);
2353         if (wpa_driver_nl80211_set_auth_param(drv,
2354                                            IW_AUTH_KEY_MGMT, value) < 0)
2355                 ret = -1;
2356         value = params->key_mgmt_suite != KEY_MGMT_NONE ||
2357                 params->pairwise_suite != CIPHER_NONE ||
2358                 params->group_suite != CIPHER_NONE ||
2359                 params->wpa_ie_len;
2360         if (wpa_driver_nl80211_set_auth_param(drv,
2361                                            IW_AUTH_PRIVACY_INVOKED, value) < 0)
2362                 ret = -1;
2363
2364         /* Allow unencrypted EAPOL messages even if pairwise keys are set when
2365          * not using WPA. IEEE 802.1X specifies that these frames are not
2366          * encrypted, but WPA encrypts them when pairwise keys are in use. */
2367         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
2368             params->key_mgmt_suite == KEY_MGMT_PSK)
2369                 allow_unencrypted_eapol = 0;
2370         else
2371                 allow_unencrypted_eapol = 1;
2372
2373         if (wpa_driver_nl80211_set_auth_param(drv,
2374                                            IW_AUTH_RX_UNENCRYPTED_EAPOL,
2375                                            allow_unencrypted_eapol) < 0)
2376                 ret = -1;
2377         if (params->freq && wpa_driver_nl80211_set_freq(drv, params->freq) < 0)
2378                 ret = -1;
2379         if (wpa_driver_nl80211_set_ssid(drv, params->ssid, params->ssid_len) < 0)
2380                 ret = -1;
2381         if (params->bssid &&
2382             wpa_driver_nl80211_set_bssid(drv, params->bssid) < 0)
2383                 ret = -1;
2384
2385         return ret;
2386 }
2387
2388
2389 static int wpa_driver_nl80211_set_auth_alg(void *priv, int auth_alg)
2390 {
2391         struct wpa_driver_nl80211_data *drv = priv;
2392         int algs = 0, res;
2393
2394         if (auth_alg & AUTH_ALG_OPEN_SYSTEM)
2395                 algs |= IW_AUTH_ALG_OPEN_SYSTEM;
2396         if (auth_alg & AUTH_ALG_SHARED_KEY)
2397                 algs |= IW_AUTH_ALG_SHARED_KEY;
2398         if (auth_alg & AUTH_ALG_LEAP)
2399                 algs |= IW_AUTH_ALG_LEAP;
2400         if (algs == 0) {
2401                 /* at least one algorithm should be set */
2402                 algs = IW_AUTH_ALG_OPEN_SYSTEM;
2403         }
2404
2405         res = wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_80211_AUTH_ALG,
2406                                              algs);
2407         drv->auth_alg_fallback = res == -2;
2408         return res;
2409 }
2410
2411
2412 /**
2413  * wpa_driver_nl80211_set_mode - Set wireless mode (infra/adhoc), SIOCSIWMODE
2414  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
2415  * @mode: 0 = infra/BSS (associate with an AP), 1 = adhoc/IBSS
2416  * Returns: 0 on success, -1 on failure
2417  */
2418 static int wpa_driver_nl80211_set_mode(void *priv, int mode)
2419 {
2420         struct wpa_driver_nl80211_data *drv = priv;
2421         int ret = -1, flags;
2422         struct nl_msg *msg;
2423
2424         msg = nlmsg_alloc();
2425         if (!msg)
2426                 return -1;
2427
2428         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2429                     0, NL80211_CMD_SET_INTERFACE, 0);
2430         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2431         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE,
2432                     mode ? NL80211_IFTYPE_ADHOC : NL80211_IFTYPE_STATION);
2433
2434         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2435         if (!ret)
2436                 return 0;
2437         else
2438                 goto try_again;
2439
2440 nla_put_failure:
2441         wpa_printf(MSG_ERROR, "nl80211: Failed to set interface mode");
2442         return -1;
2443
2444 try_again:
2445         /* mac80211 doesn't allow mode changes while the device is up, so
2446          * take the device down, try to set the mode again, and bring the
2447          * device back up.
2448          */
2449         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0) {
2450                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
2451
2452                 /* Try to set the mode again while the interface is down */
2453                 msg = nlmsg_alloc();
2454                 if (!msg)
2455                         return -1;
2456
2457                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2458                             0, NL80211_CMD_SET_INTERFACE, 0);
2459                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2460                 NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE,
2461                             mode ? NL80211_IFTYPE_ADHOC :
2462                             NL80211_IFTYPE_STATION);
2463                 ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2464                 if (ret) {
2465                         wpa_printf(MSG_ERROR, "Failed to set interface %s "
2466                                    "mode", drv->ifname);
2467                 }
2468
2469                 /* Ignore return value of get_ifflags to ensure that the device
2470                  * is always up like it was before this function was called.
2471                  */
2472                 (void) wpa_driver_nl80211_get_ifflags(drv, &flags);
2473                 (void) wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP);
2474         }
2475
2476         return ret;
2477 }
2478
2479
2480 static int wpa_driver_nl80211_pmksa(struct wpa_driver_nl80211_data *drv,
2481                                  u32 cmd, const u8 *bssid, const u8 *pmkid)
2482 {
2483         struct iwreq iwr;
2484         struct iw_pmksa pmksa;
2485         int ret = 0;
2486
2487         os_memset(&iwr, 0, sizeof(iwr));
2488         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2489         os_memset(&pmksa, 0, sizeof(pmksa));
2490         pmksa.cmd = cmd;
2491         pmksa.bssid.sa_family = ARPHRD_ETHER;
2492         if (bssid)
2493                 os_memcpy(pmksa.bssid.sa_data, bssid, ETH_ALEN);
2494         if (pmkid)
2495                 os_memcpy(pmksa.pmkid, pmkid, IW_PMKID_LEN);
2496         iwr.u.data.pointer = (caddr_t) &pmksa;
2497         iwr.u.data.length = sizeof(pmksa);
2498
2499         if (ioctl(drv->ioctl_sock, SIOCSIWPMKSA, &iwr) < 0) {
2500                 if (errno != EOPNOTSUPP)
2501                         perror("ioctl[SIOCSIWPMKSA]");
2502                 ret = -1;
2503         }
2504
2505         return ret;
2506 }
2507
2508
2509 static int wpa_driver_nl80211_add_pmkid(void *priv, const u8 *bssid,
2510                                      const u8 *pmkid)
2511 {
2512         struct wpa_driver_nl80211_data *drv = priv;
2513         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_ADD, bssid, pmkid);
2514 }
2515
2516
2517 static int wpa_driver_nl80211_remove_pmkid(void *priv, const u8 *bssid,
2518                                         const u8 *pmkid)
2519 {
2520         struct wpa_driver_nl80211_data *drv = priv;
2521         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_REMOVE, bssid, pmkid);
2522 }
2523
2524
2525 static int wpa_driver_nl80211_flush_pmkid(void *priv)
2526 {
2527         struct wpa_driver_nl80211_data *drv = priv;
2528         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_FLUSH, NULL, NULL);
2529 }
2530
2531
2532 static int wpa_driver_nl80211_get_capa(void *priv,
2533                                        struct wpa_driver_capa *capa)
2534 {
2535         struct wpa_driver_nl80211_data *drv = priv;
2536         if (!drv->has_capability)
2537                 return -1;
2538         os_memcpy(capa, &drv->capa, sizeof(*capa));
2539         return 0;
2540 }
2541
2542
2543 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
2544 {
2545         struct wpa_driver_nl80211_data *drv = priv;
2546
2547         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
2548                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
2549         drv->operstate = state;
2550         return wpa_driver_nl80211_send_oper_ifla(
2551                 drv, -1, state ? IF_OPER_UP : IF_OPER_DORMANT);
2552 }
2553
2554
2555 #ifdef CONFIG_CLIENT_MLME
2556 static int wpa_driver_nl80211_open_mlme(struct wpa_driver_nl80211_data *drv)
2557 {
2558         if (wpa_driver_nl80211_set_userspace_mlme(drv, 1) < 0) {
2559                 wpa_printf(MSG_ERROR, "nl80211: Failed to enable userspace "
2560                            "MLME");
2561                 return -1;
2562         }
2563         if (wpa_driver_nl80211_create_monitor_interface(drv)) {
2564                 wpa_printf(MSG_ERROR, "nl80211: Failed to create monitor "
2565                            "interface");
2566                 return -1;
2567         }
2568         return 0;
2569 }
2570 #endif /* CONFIG_CLIENT_MLME */
2571
2572
2573 static int wpa_driver_nl80211_set_param(void *priv, const char *param)
2574 {
2575 #ifdef CONFIG_CLIENT_MLME
2576         struct wpa_driver_nl80211_data *drv = priv;
2577
2578         if (param == NULL)
2579                 return 0;
2580
2581         wpa_printf(MSG_DEBUG, "%s: param='%s'", __func__, param);
2582
2583         if (os_strstr(param, "use_mlme=1")) {
2584                 wpa_printf(MSG_DEBUG, "nl80211: Using user space MLME");
2585                 drv->capa.flags |= WPA_DRIVER_FLAGS_USER_SPACE_MLME;
2586
2587                 if (wpa_driver_nl80211_open_mlme(drv))
2588                         return -1;
2589         }
2590 #endif /* CONFIG_CLIENT_MLME */
2591
2592         return 0;
2593 }
2594
2595
2596 #ifdef CONFIG_CLIENT_MLME
2597
2598 struct phy_info_arg {
2599         u16 *num_modes;
2600         struct wpa_hw_modes *modes;
2601 };
2602
2603
2604 static int phy_info_handler(struct nl_msg *msg, void *arg)
2605 {
2606         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
2607         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2608         struct phy_info_arg *phy_info = arg;
2609
2610         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
2611
2612         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
2613         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1]
2614                 = {
2615                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
2616                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
2617                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
2618                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
2619                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
2620         };
2621
2622         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
2623         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
2624                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
2625                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
2626                 { .type = NLA_FLAG },
2627         };
2628
2629         struct nlattr *nl_band;
2630         struct nlattr *nl_freq;
2631         struct nlattr *nl_rate;
2632         int rem_band, rem_freq, rem_rate;
2633         struct wpa_hw_modes *mode;
2634         int idx, mode_is_set;
2635
2636         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2637                   genlmsg_attrlen(gnlh, 0), NULL);
2638
2639         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
2640                 return NL_SKIP;
2641
2642         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS],
2643                             rem_band) {
2644                 mode = os_realloc(phy_info->modes,
2645                                   (*phy_info->num_modes + 1) * sizeof(*mode));
2646                 if (!mode)
2647                         return NL_SKIP;
2648                 phy_info->modes = mode;
2649
2650                 mode_is_set = 0;
2651
2652                 mode = &phy_info->modes[*(phy_info->num_modes)];
2653                 os_memset(mode, 0, sizeof(*mode));
2654                 *(phy_info->num_modes) += 1;
2655
2656                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
2657                           nla_len(nl_band), NULL);
2658
2659                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS],
2660                                     rem_freq) {
2661                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
2662                                   nla_data(nl_freq), nla_len(nl_freq),
2663                                   freq_policy);
2664                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2665                                 continue;
2666                         mode->num_channels++;
2667                 }
2668
2669                 mode->channels = os_zalloc(mode->num_channels *
2670                                            sizeof(struct wpa_channel_data));
2671                 if (!mode->channels)
2672                         return NL_SKIP;
2673
2674                 idx = 0;
2675
2676                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS],
2677                                     rem_freq) {
2678                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
2679                                   nla_data(nl_freq), nla_len(nl_freq),
2680                                   freq_policy);
2681                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2682                                 continue;
2683
2684                         mode->channels[idx].freq = nla_get_u32(
2685                                 tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
2686                         mode->channels[idx].flag |= WPA_CHAN_W_SCAN |
2687                                 WPA_CHAN_W_ACTIVE_SCAN |
2688                                 WPA_CHAN_W_IBSS;
2689
2690                         if (!mode_is_set) {
2691                                 /* crude heuristic */
2692                                 if (mode->channels[idx].freq < 4000)
2693                                         mode->mode = WPA_MODE_IEEE80211B;
2694                                 else
2695                                         mode->mode = WPA_MODE_IEEE80211A;
2696                                 mode_is_set = 1;
2697                         }
2698
2699                         /* crude heuristic */
2700                         if (mode->channels[idx].freq < 4000) {
2701                                 if (mode->channels[idx].freq == 2848)
2702                                         mode->channels[idx].chan = 14;
2703                                 else
2704                                         mode->channels[idx].chan =
2705                                                 (mode->channels[idx].freq -
2706                                                  2407) / 5;
2707                         } else
2708                                 mode->channels[idx].chan =
2709                                         mode->channels[idx].freq / 5 - 1000;
2710
2711                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
2712                                 mode->channels[idx].flag &= ~WPA_CHAN_W_SCAN;
2713                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
2714                                 mode->channels[idx].flag &=
2715                                         ~WPA_CHAN_W_ACTIVE_SCAN;
2716                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
2717                                 mode->channels[idx].flag &= ~WPA_CHAN_W_IBSS;
2718                         idx++;
2719                 }
2720
2721                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES],
2722                                     rem_rate) {
2723                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
2724                                   nla_data(nl_rate), nla_len(nl_rate),
2725                                   rate_policy);
2726                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2727                                 continue;
2728                         mode->num_rates++;
2729                 }
2730
2731                 mode->rates = os_zalloc(mode->num_rates *
2732                                         sizeof(struct wpa_rate_data));
2733                 if (!mode->rates)
2734                         return NL_SKIP;
2735
2736                 idx = 0;
2737
2738                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES],
2739                                     rem_rate) {
2740                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
2741                                   nla_data(nl_rate), nla_len(nl_rate),
2742                                   rate_policy);
2743                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2744                                 continue;
2745                         mode->rates[idx].rate = nla_get_u32(
2746                                 tb_rate[NL80211_BITRATE_ATTR_RATE]);
2747
2748                         /* crude heuristic */
2749                         if (mode->mode == WPA_MODE_IEEE80211B &&
2750                             mode->rates[idx].rate > 200)
2751                                 mode->mode = WPA_MODE_IEEE80211G;
2752
2753                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
2754                                 mode->rates[idx].flags |= WPA_RATE_PREAMBLE2;
2755
2756                         idx++;
2757                 }
2758         }
2759
2760         return NL_SKIP;
2761 }
2762
2763
2764 static struct wpa_hw_modes *
2765 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
2766 {
2767         struct wpa_driver_nl80211_data *drv = priv;
2768         struct nl_msg *msg;
2769         struct phy_info_arg result = {
2770                 .num_modes = num_modes,
2771                 .modes = NULL,
2772         };
2773
2774         *num_modes = 0;
2775         *flags = 0;
2776
2777         msg = nlmsg_alloc();
2778         if (!msg)
2779                 return NULL;
2780
2781         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2782                     0, NL80211_CMD_GET_WIPHY, 0);
2783
2784         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2785
2786         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
2787                 return result.modes;
2788 nla_put_failure:
2789         return NULL;
2790 }
2791
2792
2793 static int wpa_driver_nl80211_set_channel(void *priv, wpa_hw_mode phymode,
2794                                           int chan, int freq)
2795 {
2796         return wpa_driver_nl80211_set_freq(priv, freq);
2797 }
2798
2799
2800 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
2801                                         size_t data_len)
2802 {
2803         struct wpa_driver_nl80211_data *drv = priv;
2804         __u8 rtap_hdr[] = {
2805                 0x00, 0x00, /* radiotap version */
2806                 0x0e, 0x00, /* radiotap length */
2807                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
2808                 0x0c,       /* F_WEP | F_FRAG (encrypt/fragment if required) */
2809                 0x00,       /* padding */
2810                 0x00, 0x00, /* RX and TX flags to indicate that */
2811                 0x00, 0x00, /* this is the injected frame directly */
2812         };
2813         struct iovec iov[2] = {
2814                 {
2815                         .iov_base = &rtap_hdr,
2816                         .iov_len = sizeof(rtap_hdr),
2817                 },
2818                 {
2819                         .iov_base = (void *) data,
2820                         .iov_len = data_len,
2821                 }
2822         };
2823         struct msghdr msg = {
2824                 .msg_name = NULL,
2825                 .msg_namelen = 0,
2826                 .msg_iov = iov,
2827                 .msg_iovlen = 2,
2828                 .msg_control = NULL,
2829                 .msg_controllen = 0,
2830                 .msg_flags = 0,
2831         };
2832
2833         if (sendmsg(drv->monitor_sock, &msg, 0) < 0) {
2834                 perror("send[MLME]");
2835                 return -1;
2836         }
2837
2838         return 0;
2839 }
2840
2841
2842 static int wpa_driver_nl80211_mlme_add_sta(void *priv, const u8 *addr,
2843                                            const u8 *supp_rates,
2844                                            size_t supp_rates_len)
2845 {
2846         struct wpa_driver_nl80211_data *drv = priv;
2847         struct nl_msg *msg;
2848         int ret = -1;
2849
2850         msg = nlmsg_alloc();
2851         if (!msg)
2852                 return -ENOMEM;
2853
2854         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2855                     0, NL80211_CMD_NEW_STATION, 0);
2856
2857         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2858         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2859         /* TODO: Get proper Association ID and listen interval */
2860         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, 1);
2861         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, supp_rates_len,
2862                 supp_rates);
2863         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL, 1);
2864
2865         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2866         /* ignore EEXIST, this happens if a STA associates while associated */
2867         if (ret == -EEXIST || ret >= 0)
2868                 ret = 0;
2869
2870 nla_put_failure:
2871         return ret;
2872 }
2873
2874
2875 static int wpa_driver_nl80211_mlme_remove_sta(void *priv, const u8 *addr)
2876 {
2877         struct wpa_driver_nl80211_data *drv = priv;
2878         struct nl_msg *msg;
2879         int ret = -1;
2880
2881         msg = nlmsg_alloc();
2882         if (!msg)
2883                 return -ENOMEM;
2884
2885         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2886                     0, NL80211_CMD_DEL_STATION, 0);
2887
2888         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2889         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2890
2891         ret = 0;
2892
2893         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2894         return ret;
2895
2896 nla_put_failure:
2897         return -ENOBUFS;
2898 }
2899
2900 #endif /* CONFIG_CLIENT_MLME */
2901
2902
2903 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
2904         .name = "nl80211",
2905         .desc = "Linux nl80211/cfg80211",
2906         .get_bssid = wpa_driver_nl80211_get_bssid,
2907         .get_ssid = wpa_driver_nl80211_get_ssid,
2908         .set_wpa = wpa_driver_nl80211_set_wpa,
2909         .set_key = wpa_driver_nl80211_set_key,
2910         .set_countermeasures = wpa_driver_nl80211_set_countermeasures,
2911         .set_drop_unencrypted = wpa_driver_nl80211_set_drop_unencrypted,
2912         .scan = wpa_driver_nl80211_scan,
2913         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
2914         .deauthenticate = wpa_driver_nl80211_deauthenticate,
2915         .disassociate = wpa_driver_nl80211_disassociate,
2916         .set_mode = wpa_driver_nl80211_set_mode,
2917         .associate = wpa_driver_nl80211_associate,
2918         .set_auth_alg = wpa_driver_nl80211_set_auth_alg,
2919         .init = wpa_driver_nl80211_init,
2920         .deinit = wpa_driver_nl80211_deinit,
2921         .set_param = wpa_driver_nl80211_set_param,
2922         .add_pmkid = wpa_driver_nl80211_add_pmkid,
2923         .remove_pmkid = wpa_driver_nl80211_remove_pmkid,
2924         .flush_pmkid = wpa_driver_nl80211_flush_pmkid,
2925         .get_capa = wpa_driver_nl80211_get_capa,
2926         .set_operstate = wpa_driver_nl80211_set_operstate,
2927         .set_country = wpa_driver_nl80211_set_country,
2928         .set_probe_req_ie = wpa_driver_nl80211_set_probe_req_ie,
2929 #ifdef CONFIG_CLIENT_MLME
2930         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
2931         .set_channel = wpa_driver_nl80211_set_channel,
2932         .set_ssid = wpa_driver_nl80211_set_ssid,
2933         .set_bssid = wpa_driver_nl80211_set_bssid,
2934         .send_mlme = wpa_driver_nl80211_send_mlme,
2935         .mlme_add_sta = wpa_driver_nl80211_mlme_add_sta,
2936         .mlme_remove_sta = wpa_driver_nl80211_mlme_remove_sta,
2937 #endif /* CONFIG_CLIENT_MLME */
2938 };