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