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