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