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

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