wpa_supplicant/wpa_priv.c

   1 /*
   2  * WPA Supplicant / privileged helper program
   3  * Copyright (c) 2007-2009, Jouni Malinen <j@w1.fi>
   4  *
   5  * This software may be distributed under the terms of the BSD license.
   6  * See README for more details.
   7  */
   8
   9 #include "includes.h"
  10 #ifdef __linux__
  11 #include <fcntl.h>
  12 #endif /* __linux__ */
  13 #include <sys/un.h>
  14 #include <sys/stat.h>
  15
  16 #include "common.h"
  17 #include "eloop.h"
  18 #include "common/version.h"
  19 #include "drivers/driver.h"
  20 #include "l2_packet/l2_packet.h"
  21 #include "common/privsep_commands.h"
  22 #include "common/ieee802_11_defs.h"
  23
  24
  25 struct wpa_priv_interface {
  26         struct wpa_priv_interface *next;
  27         char *driver_name;
  28         char *ifname;
  29         char *sock_name;
  30         int fd;
  31
  32         struct wpa_driver_ops *driver;
  33         void *drv_priv;
  34         struct sockaddr_un drv_addr;
  35         int wpas_registered;
  36
  37         /* TODO: add support for multiple l2 connections */
  38         struct l2_packet_data *l2;
  39         struct sockaddr_un l2_addr;
  40 };
  41
  42
  43 static void wpa_priv_cmd_register(struct wpa_priv_interface *iface,
  44                                   struct sockaddr_un *from)
  45 {
  46         if (iface->drv_priv) {
  47                 wpa_printf(MSG_DEBUG, "Cleaning up forgotten driver instance");
  48                 if (iface->driver->deinit)
  49                         iface->driver->deinit(iface->drv_priv);
  50                 iface->drv_priv = NULL;
  51                 iface->wpas_registered = 0;
  52         }
  53
  54         if (iface->l2) {
  55                 wpa_printf(MSG_DEBUG, "Cleaning up forgotten l2_packet "
  56                            "instance");
  57                 l2_packet_deinit(iface->l2);
  58                 iface->l2 = NULL;
  59         }
  60
  61         if (iface->driver->init == NULL)
  62                 return;
  63
  64         iface->drv_priv = iface->driver->init(iface, iface->ifname);
  65         if (iface->drv_priv == NULL) {
  66                 wpa_printf(MSG_DEBUG, "Failed to initialize driver wrapper");
  67                 return;
  68         }
  69
  70         wpa_printf(MSG_DEBUG, "Driver wrapper '%s' initialized for interface "
  71                    "'%s'", iface->driver_name, iface->ifname);
  72
  73         os_memcpy(&iface->drv_addr, from, sizeof(iface->drv_addr));
  74         iface->wpas_registered = 1;
  75
  76         if (iface->driver->set_param &&
  77             iface->driver->set_param(iface->drv_priv, NULL) < 0) {
  78                 wpa_printf(MSG_ERROR, "Driver interface rejected param");
  79         }
  80 }
  81
  82
  83 static void wpa_priv_cmd_unregister(struct wpa_priv_interface *iface,
  84                                     struct sockaddr_un *from)
  85 {
  86         if (iface->drv_priv) {
  87                 if (iface->driver->deinit)
  88                         iface->driver->deinit(iface->drv_priv);
  89                 iface->drv_priv = NULL;
  90                 iface->wpas_registered = 0;
  91         }
  92 }
  93
  94
  95 static void wpa_priv_cmd_scan(struct wpa_priv_interface *iface,
  96                               char *buf, size_t len)
  97 {
  98         struct wpa_driver_scan_params params;
  99
 100         if (iface->drv_priv == NULL)
 101                 return;
 102
 103         os_memset(&params, 0, sizeof(params));
 104         if (len) {
 105                 params.ssids[0].ssid = (u8 *) buf;
 106                 params.ssids[0].ssid_len = len;
 107                 params.num_ssids = 1;
 108         }
 109
 110         if (iface->driver->scan2)
 111                 iface->driver->scan2(iface->drv_priv, &params);
 112 }
 113
 114
 115 static void wpa_priv_get_scan_results2(struct wpa_priv_interface *iface,
 116                                        struct sockaddr_un *from)
 117 {
 118         struct wpa_scan_results *res;
 119         u8 *buf = NULL, *pos, *end;
 120         int val;
 121         size_t i;
 122
 123         res = iface->driver->get_scan_results2(iface->drv_priv);
 124         if (res == NULL)
 125                 goto fail;
 126
 127         buf = os_malloc(60000);
 128         if (buf == NULL)
 129                 goto fail;
 130         pos = buf;
 131         end = buf + 60000;
 132         val = res->num;
 133         os_memcpy(pos, &val, sizeof(int));
 134         pos += sizeof(int);
 135
 136         for (i = 0; i < res->num; i++) {
 137                 struct wpa_scan_res *r = res->res[i];
 138                 val = sizeof(*r) + r->ie_len;
 139                 if (end - pos < (int) sizeof(int) + val)
 140                         break;
 141                 os_memcpy(pos, &val, sizeof(int));
 142                 pos += sizeof(int);
 143                 os_memcpy(pos, r, val);
 144                 pos += val;
 145         }
 146
 147         sendto(iface->fd, buf, pos - buf, 0, (struct sockaddr *) from,
 148                sizeof(*from));
 149
 150         os_free(buf);
 151         wpa_scan_results_free(res);
 152         return;
 153
 154 fail:
 155         os_free(buf);
 156         wpa_scan_results_free(res);
 157         sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
 158 }
 159
 160
 161 static void wpa_priv_cmd_get_scan_results(struct wpa_priv_interface *iface,
 162                                           struct sockaddr_un *from)
 163 {
 164         if (iface->drv_priv == NULL)
 165                 return;
 166
 167         if (iface->driver->get_scan_results2)
 168                 wpa_priv_get_scan_results2(iface, from);
 169         else
 170                 sendto(iface->fd, "", 0, 0, (struct sockaddr *) from,
 171                        sizeof(*from));
 172 }
 173
 174
 175 static void wpa_priv_cmd_associate(struct wpa_priv_interface *iface,
 176                                    void *buf, size_t len)
 177 {
 178         struct wpa_driver_associate_params params;
 179         struct privsep_cmd_associate *assoc;
 180         u8 *bssid;
 181         int res;
 182
 183         if (iface->drv_priv == NULL || iface->driver->associate == NULL)
 184                 return;
 185
 186         if (len < sizeof(*assoc)) {
 187                 wpa_printf(MSG_DEBUG, "Invalid association request");
 188                 return;
 189         }
 190
 191         assoc = buf;
 192         if (sizeof(*assoc) + assoc->wpa_ie_len > len) {
 193                 wpa_printf(MSG_DEBUG, "Association request overflow");
 194                 return;
 195         }
 196
 197         os_memset(&params, 0, sizeof(params));
 198         bssid = assoc->bssid;
 199         if (bssid[0] | bssid[1] | bssid[2] | bssid[3] | bssid[4] | bssid[5])
 200                 params.bssid = bssid;
 201         params.ssid = assoc->ssid;
 202         if (assoc->ssid_len > 32)
 203                 return;
 204         params.ssid_len = assoc->ssid_len;
 205         params.freq = assoc->freq;
 206         if (assoc->wpa_ie_len) {
 207                 params.wpa_ie = (u8 *) (assoc + 1);
 208                 params.wpa_ie_len = assoc->wpa_ie_len;
 209         }
 210         params.pairwise_suite = assoc->pairwise_suite;
 211         params.group_suite = assoc->group_suite;
 212         params.key_mgmt_suite = assoc->key_mgmt_suite;
 213         params.auth_alg = assoc->auth_alg;
 214         params.mode = assoc->mode;
 215
 216         res = iface->driver->associate(iface->drv_priv, &params);
 217         wpa_printf(MSG_DEBUG, "drv->associate: res=%d", res);
 218 }
 219
 220
 221 static void wpa_priv_cmd_get_bssid(struct wpa_priv_interface *iface,
 222                                    struct sockaddr_un *from)
 223 {
 224         u8 bssid[ETH_ALEN];
 225
 226         if (iface->drv_priv == NULL)
 227                 goto fail;
 228
 229         if (iface->driver->get_bssid == NULL ||
 230             iface->driver->get_bssid(iface->drv_priv, bssid) < 0)
 231                 goto fail;
 232
 233         sendto(iface->fd, bssid, ETH_ALEN, 0, (struct sockaddr *) from,
 234                sizeof(*from));
 235         return;
 236
 237 fail:
 238         sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
 239 }
 240
 241
 242 static void wpa_priv_cmd_get_ssid(struct wpa_priv_interface *iface,
 243                                   struct sockaddr_un *from)
 244 {
 245         u8 ssid[sizeof(int) + 32];
 246         int res;
 247
 248         if (iface->drv_priv == NULL)
 249                 goto fail;
 250
 251         if (iface->driver->get_ssid == NULL)
 252                 goto fail;
 253
 254         res = iface->driver->get_ssid(iface->drv_priv, &ssid[sizeof(int)]);
 255         if (res < 0 || res > 32)
 256                 goto fail;
 257         os_memcpy(ssid, &res, sizeof(int));
 258
 259         sendto(iface->fd, ssid, sizeof(ssid), 0, (struct sockaddr *) from,
 260                sizeof(*from));
 261         return;
 262
 263 fail:
 264         sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
 265 }
 266
 267
 268 static void wpa_priv_cmd_set_key(struct wpa_priv_interface *iface,
 269                                  void *buf, size_t len)
 270 {
 271         struct privsep_cmd_set_key *params;
 272         int res;
 273
 274         if (iface->drv_priv == NULL || iface->driver->set_key == NULL)
 275                 return;
 276
 277         if (len != sizeof(*params)) {
 278                 wpa_printf(MSG_DEBUG, "Invalid set_key request");
 279                 return;
 280         }
 281
 282         params = buf;
 283
 284         res = iface->driver->set_key(iface->ifname, iface->drv_priv,
 285                                      params->alg,
 286                                      params->addr, params->key_idx,
 287                                      params->set_tx,
 288                                      params->seq_len ? params->seq : NULL,
 289                                      params->seq_len,
 290                                      params->key_len ? params->key : NULL,
 291                                      params->key_len);
 292         wpa_printf(MSG_DEBUG, "drv->set_key: res=%d", res);
 293 }
 294
 295
 296 static void wpa_priv_cmd_get_capa(struct wpa_priv_interface *iface,
 297                                   struct sockaddr_un *from)
 298 {
 299         struct wpa_driver_capa capa;
 300
 301         if (iface->drv_priv == NULL)
 302                 goto fail;
 303
 304         if (iface->driver->get_capa == NULL ||
 305             iface->driver->get_capa(iface->drv_priv, &capa) < 0)
 306                 goto fail;
 307
 308         sendto(iface->fd, &capa, sizeof(capa), 0, (struct sockaddr *) from,
 309                sizeof(*from));
 310         return;
 311
 312 fail:
 313         sendto(iface->fd, "", 0, 0, (struct sockaddr *) from, sizeof(*from));
 314 }
 315
 316
 317 static void wpa_priv_l2_rx(void *ctx, const u8 *src_addr, const u8 *buf,
 318                            size_t len)
 319 {
 320         struct wpa_priv_interface *iface = ctx;
 321         struct msghdr msg;
 322         struct iovec io[2];
 323
 324         io[0].iov_base = (u8 *) src_addr;
 325         io[0].iov_len = ETH_ALEN;
 326         io[1].iov_base = (u8 *) buf;
 327         io[1].iov_len = len;
 328
 329         os_memset(&msg, 0, sizeof(msg));
 330         msg.msg_iov = io;
 331         msg.msg_iovlen = 2;
 332         msg.msg_name = &iface->l2_addr;
 333         msg.msg_namelen = sizeof(iface->l2_addr);
 334
 335         if (sendmsg(iface->fd, &msg, 0) < 0) {
 336                 perror("sendmsg(l2 rx)");
 337         }
 338 }
 339
 340
 341 static void wpa_priv_cmd_l2_register(struct wpa_priv_interface *iface,
 342                                      struct sockaddr_un *from,
 343                                      void *buf, size_t len)
 344 {
 345         int *reg_cmd = buf;
 346         u8 own_addr[ETH_ALEN];
 347         int res;
 348         u16 proto;
 349
 350         if (len != 2 * sizeof(int)) {
 351                 wpa_printf(MSG_DEBUG, "Invalid l2_register length %lu",
 352                            (unsigned long) len);
 353                 return;
 354         }
 355
 356         proto = reg_cmd[0];
 357         if (proto != ETH_P_EAPOL && proto != ETH_P_RSN_PREAUTH) {
 358                 wpa_printf(MSG_DEBUG, "Refused l2_packet connection for "
 359                            "ethertype 0x%x", proto);
 360                 return;
 361         }
 362
 363         if (iface->l2) {
 364                 wpa_printf(MSG_DEBUG, "Cleaning up forgotten l2_packet "
 365                            "instance");
 366                 l2_packet_deinit(iface->l2);
 367                 iface->l2 = NULL;
 368         }
 369
 370         os_memcpy(&iface->l2_addr, from, sizeof(iface->l2_addr));
 371
 372         iface->l2 = l2_packet_init(iface->ifname, NULL, proto,
 373                                    wpa_priv_l2_rx, iface, reg_cmd[1]);
 374         if (iface->l2 == NULL) {
 375                 wpa_printf(MSG_DEBUG, "Failed to initialize l2_packet "
 376                            "instance for protocol %d", proto);
 377                 return;
 378         }
 379
 380         if (l2_packet_get_own_addr(iface->l2, own_addr) < 0) {
 381                 wpa_printf(MSG_DEBUG, "Failed to get own address from "
 382                            "l2_packet");
 383                 l2_packet_deinit(iface->l2);
 384                 iface->l2 = NULL;
 385                 return;
 386         }
 387
 388         res = sendto(iface->fd, own_addr, ETH_ALEN, 0,
 389                      (struct sockaddr *) from, sizeof(*from));
 390         wpa_printf(MSG_DEBUG, "L2 registration: res=%d", res);
 391 }
 392
 393
 394 static void wpa_priv_cmd_l2_unregister(struct wpa_priv_interface *iface,
 395                                        struct sockaddr_un *from)
 396 {
 397         if (iface->l2) {
 398                 l2_packet_deinit(iface->l2);
 399                 iface->l2 = NULL;
 400         }
 401 }
 402
 403
 404 static void wpa_priv_cmd_l2_notify_auth_start(struct wpa_priv_interface *iface,
 405                                               struct sockaddr_un *from)
 406 {
 407         if (iface->l2)
 408                 l2_packet_notify_auth_start(iface->l2);
 409 }
 410
 411
 412 static void wpa_priv_cmd_l2_send(struct wpa_priv_interface *iface,
 413                                  struct sockaddr_un *from,
 414                                  void *buf, size_t len)
 415 {
 416         u8 *dst_addr;
 417         u16 proto;
 418         int res;
 419
 420         if (iface->l2 == NULL)
 421                 return;
 422
 423         if (len < ETH_ALEN + 2) {
 424                 wpa_printf(MSG_DEBUG, "Too short L2 send packet (len=%lu)",
 425                            (unsigned long) len);
 426                 return;
 427         }
 428
 429         dst_addr = buf;
 430         os_memcpy(&proto, buf + ETH_ALEN, 2);
 431
 432         if (proto != ETH_P_EAPOL && proto != ETH_P_RSN_PREAUTH) {
 433                 wpa_printf(MSG_DEBUG, "Refused l2_packet send for ethertype "
 434                            "0x%x", proto);
 435                 return;
 436         }
 437
 438         res = l2_packet_send(iface->l2, dst_addr, proto, buf + ETH_ALEN + 2,
 439                              len - ETH_ALEN - 2);
 440         wpa_printf(MSG_DEBUG, "L2 send: res=%d", res);
 441 }
 442
 443
 444 static void wpa_priv_cmd_set_country(struct wpa_priv_interface *iface,
 445                                      char *buf)
 446 {
 447         if (iface->drv_priv == NULL || iface->driver->set_country == NULL ||
 448             *buf == '\0')
 449                 return;
 450
 451         iface->driver->set_country(iface->drv_priv, buf);
 452 }
 453
 454
 455 static void wpa_priv_receive(int sock, void *eloop_ctx, void *sock_ctx)
 456 {
 457         struct wpa_priv_interface *iface = eloop_ctx;
 458         char buf[2000], *pos;
 459         void *cmd_buf;
 460         size_t cmd_len;
 461         int res, cmd;
 462         struct sockaddr_un from;
 463         socklen_t fromlen = sizeof(from);
 464
 465         res = recvfrom(sock, buf, sizeof(buf), 0, (struct sockaddr *) &from,
 466                        &fromlen);
 467         if (res < 0) {
 468                 perror("recvfrom");
 469                 return;
 470         }
 471
 472         if (res < (int) sizeof(int)) {
 473                 wpa_printf(MSG_DEBUG, "Too short command (len=%d)", res);
 474                 return;
 475         }
 476
 477         os_memcpy(&cmd, buf, sizeof(int));
 478         wpa_printf(MSG_DEBUG, "Command %d for interface %s",
 479                    cmd, iface->ifname);
 480         cmd_buf = &buf[sizeof(int)];
 481         cmd_len = res - sizeof(int);
 482
 483         switch (cmd) {
 484         case PRIVSEP_CMD_REGISTER:
 485                 wpa_priv_cmd_register(iface, &from);
 486                 break;
 487         case PRIVSEP_CMD_UNREGISTER:
 488                 wpa_priv_cmd_unregister(iface, &from);
 489                 break;
 490         case PRIVSEP_CMD_SCAN:
 491                 wpa_priv_cmd_scan(iface, cmd_buf, cmd_len);
 492                 break;
 493         case PRIVSEP_CMD_GET_SCAN_RESULTS:
 494                 wpa_priv_cmd_get_scan_results(iface, &from);
 495                 break;
 496         case PRIVSEP_CMD_ASSOCIATE:
 497                 wpa_priv_cmd_associate(iface, cmd_buf, cmd_len);
 498                 break;
 499         case PRIVSEP_CMD_GET_BSSID:
 500                 wpa_priv_cmd_get_bssid(iface, &from);
 501                 break;
 502         case PRIVSEP_CMD_GET_SSID:
 503                 wpa_priv_cmd_get_ssid(iface, &from);
 504                 break;
 505         case PRIVSEP_CMD_SET_KEY:
 506                 wpa_priv_cmd_set_key(iface, cmd_buf, cmd_len);
 507                 break;
 508         case PRIVSEP_CMD_GET_CAPA:
 509                 wpa_priv_cmd_get_capa(iface, &from);
 510                 break;
 511         case PRIVSEP_CMD_L2_REGISTER:
 512                 wpa_priv_cmd_l2_register(iface, &from, cmd_buf, cmd_len);
 513                 break;
 514         case PRIVSEP_CMD_L2_UNREGISTER:
 515                 wpa_priv_cmd_l2_unregister(iface, &from);
 516                 break;
 517         case PRIVSEP_CMD_L2_NOTIFY_AUTH_START:
 518                 wpa_priv_cmd_l2_notify_auth_start(iface, &from);
 519                 break;
 520         case PRIVSEP_CMD_L2_SEND:
 521                 wpa_priv_cmd_l2_send(iface, &from, cmd_buf, cmd_len);
 522                 break;
 523         case PRIVSEP_CMD_SET_COUNTRY:
 524                 pos = cmd_buf;
 525                 if (pos + cmd_len >= buf + sizeof(buf))
 526                         break;
 527                 pos[cmd_len] = '\0';
 528                 wpa_priv_cmd_set_country(iface, pos);
 529                 break;
 530         }
 531 }
 532
 533
 534 static void wpa_priv_interface_deinit(struct wpa_priv_interface *iface)
 535 {
 536         if (iface->drv_priv && iface->driver->deinit)
 537                 iface->driver->deinit(iface->drv_priv);
 538
 539         if (iface->fd >= 0) {
 540                 eloop_unregister_read_sock(iface->fd);
 541                 close(iface->fd);
 542                 unlink(iface->sock_name);
 543         }
 544
 545         if (iface->l2)
 546                 l2_packet_deinit(iface->l2);
 547
 548         os_free(iface->ifname);
 549         os_free(iface->driver_name);
 550         os_free(iface->sock_name);
 551         os_free(iface);
 552 }
 553
 554
 555 extern struct wpa_driver_ops *wpa_drivers[];
 556
 557 static struct wpa_priv_interface *
 558 wpa_priv_interface_init(const char *dir, const char *params)
 559 {
 560         struct wpa_priv_interface *iface;
 561         char *pos;
 562         size_t len;
 563         struct sockaddr_un addr;
 564         int i;
 565
 566         pos = os_strchr(params, ':');
 567         if (pos == NULL)
 568                 return NULL;
 569
 570         iface = os_zalloc(sizeof(*iface));
 571         if (iface == NULL)
 572                 return NULL;
 573         iface->fd = -1;
 574
 575         len = pos - params;
 576         iface->driver_name = dup_binstr(params, len);
 577         if (iface->driver_name == NULL) {
 578                 wpa_priv_interface_deinit(iface);
 579                 return NULL;
 580         }
 581
 582         for (i = 0; wpa_drivers[i]; i++) {
 583                 if (os_strcmp(iface->driver_name,
 584                               wpa_drivers[i]->name) == 0) {
 585                         iface->driver = wpa_drivers[i];
 586                         break;
 587                 }
 588         }
 589         if (iface->driver == NULL) {
 590                 wpa_printf(MSG_ERROR, "Unsupported driver '%s'",
 591                            iface->driver_name);
 592                 wpa_priv_interface_deinit(iface);
 593                 return NULL;
 594         }
 595
 596         pos++;
 597         iface->ifname = os_strdup(pos);
 598         if (iface->ifname == NULL) {
 599                 wpa_priv_interface_deinit(iface);
 600                 return NULL;
 601         }
 602
 603         len = os_strlen(dir) + 1 + os_strlen(iface->ifname);
 604         iface->sock_name = os_malloc(len + 1);
 605         if (iface->sock_name == NULL) {
 606                 wpa_priv_interface_deinit(iface);
 607                 return NULL;
 608         }
 609
 610         os_snprintf(iface->sock_name, len + 1, "%s/%s", dir, iface->ifname);
 611         if (os_strlen(iface->sock_name) >= sizeof(addr.sun_path)) {
 612                 wpa_priv_interface_deinit(iface);
 613                 return NULL;
 614         }
 615
 616         iface->fd = socket(PF_UNIX, SOCK_DGRAM, 0);
 617         if (iface->fd < 0) {
 618                 perror("socket(PF_UNIX)");
 619                 wpa_priv_interface_deinit(iface);
 620                 return NULL;
 621         }
 622
 623         os_memset(&addr, 0, sizeof(addr));
 624         addr.sun_family = AF_UNIX;
 625         os_strlcpy(addr.sun_path, iface->sock_name, sizeof(addr.sun_path));
 626
 627         if (bind(iface->fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
 628                 wpa_printf(MSG_DEBUG, "bind(PF_UNIX) failed: %s",
 629                            strerror(errno));
 630                 if (connect(iface->fd, (struct sockaddr *) &addr,
 631                             sizeof(addr)) < 0) {
 632                         wpa_printf(MSG_DEBUG, "Socket exists, but does not "
 633                                    "allow connections - assuming it was "
 634                                    "leftover from forced program termination");
 635                         if (unlink(iface->sock_name) < 0) {
 636                                 perror("unlink[ctrl_iface]");
 637                                 wpa_printf(MSG_ERROR, "Could not unlink "
 638                                            "existing ctrl_iface socket '%s'",
 639                                            iface->sock_name);
 640                                 goto fail;
 641                         }
 642                         if (bind(iface->fd, (struct sockaddr *) &addr,
 643                                  sizeof(addr)) < 0) {
 644                                 perror("wpa-priv-iface-init: bind(PF_UNIX)");
 645                                 goto fail;
 646                         }
 647                         wpa_printf(MSG_DEBUG, "Successfully replaced leftover "
 648                                    "socket '%s'", iface->sock_name);
 649                 } else {
 650                         wpa_printf(MSG_INFO, "Socket exists and seems to be "
 651                                    "in use - cannot override it");
 652                         wpa_printf(MSG_INFO, "Delete '%s' manually if it is "
 653                                    "not used anymore", iface->sock_name);
 654                         goto fail;
 655                 }
 656         }
 657
 658         if (chmod(iface->sock_name, S_IRWXU | S_IRWXG | S_IRWXO) < 0) {
 659                 perror("chmod");
 660                 goto fail;
 661         }
 662
 663         eloop_register_read_sock(iface->fd, wpa_priv_receive, iface, NULL);
 664
 665         return iface;
 666
 667 fail:
 668         wpa_priv_interface_deinit(iface);
 669         return NULL;
 670 }
 671
 672
 673 static int wpa_priv_send_event(struct wpa_priv_interface *iface, int event,
 674                                const void *data, size_t data_len)
 675 {
 676         struct msghdr msg;
 677         struct iovec io[2];
 678
 679         io[0].iov_base = &event;
 680         io[0].iov_len = sizeof(event);
 681         io[1].iov_base = (u8 *) data;
 682         io[1].iov_len = data_len;
 683
 684         os_memset(&msg, 0, sizeof(msg));
 685         msg.msg_iov = io;
 686         msg.msg_iovlen = data ? 2 : 1;
 687         msg.msg_name = &iface->drv_addr;
 688         msg.msg_namelen = sizeof(iface->drv_addr);
 689
 690         if (sendmsg(iface->fd, &msg, 0) < 0) {
 691                 perror("sendmsg(wpas_socket)");
 692                 return -1;
 693         }
 694
 695         return 0;
 696 }
 697
 698
 699 static void wpa_priv_send_assoc(struct wpa_priv_interface *iface, int event,
 700                                 union wpa_event_data *data)
 701 {
 702         size_t buflen = 3 * sizeof(int);
 703         u8 *buf, *pos;
 704         int len;
 705
 706         if (data) {
 707                 buflen += data->assoc_info.req_ies_len +
 708                         data->assoc_info.resp_ies_len +
 709                         data->assoc_info.beacon_ies_len;
 710         }
 711
 712         buf = os_malloc(buflen);
 713         if (buf == NULL)
 714                 return;
 715
 716         pos = buf;
 717
 718         if (data && data->assoc_info.req_ies) {
 719                 len = data->assoc_info.req_ies_len;
 720                 os_memcpy(pos, &len, sizeof(int));
 721                 pos += sizeof(int);
 722                 os_memcpy(pos, data->assoc_info.req_ies, len);
 723                 pos += len;
 724         } else {
 725                 len = 0;
 726                 os_memcpy(pos, &len, sizeof(int));
 727                 pos += sizeof(int);
 728         }
 729
 730         if (data && data->assoc_info.resp_ies) {
 731                 len = data->assoc_info.resp_ies_len;
 732                 os_memcpy(pos, &len, sizeof(int));
 733                 pos += sizeof(int);
 734                 os_memcpy(pos, data->assoc_info.resp_ies, len);
 735                 pos += len;
 736         } else {
 737                 len = 0;
 738                 os_memcpy(pos, &len, sizeof(int));
 739                 pos += sizeof(int);
 740         }
 741
 742         if (data && data->assoc_info.beacon_ies) {
 743                 len = data->assoc_info.beacon_ies_len;
 744                 os_memcpy(pos, &len, sizeof(int));
 745                 pos += sizeof(int);
 746                 os_memcpy(pos, data->assoc_info.beacon_ies, len);
 747                 pos += len;
 748         } else {
 749                 len = 0;
 750                 os_memcpy(pos, &len, sizeof(int));
 751                 pos += sizeof(int);
 752         }
 753
 754         wpa_priv_send_event(iface, event, buf, buflen);
 755
 756         os_free(buf);
 757 }
 758
 759
 760 static void wpa_priv_send_interface_status(struct wpa_priv_interface *iface,
 761                                            union wpa_event_data *data)
 762 {
 763         int ievent;
 764         size_t len, maxlen;
 765         u8 *buf;
 766         char *ifname;
 767
 768         if (data == NULL)
 769                 return;
 770
 771         ievent = data->interface_status.ievent;
 772         maxlen = sizeof(data->interface_status.ifname);
 773         ifname = data->interface_status.ifname;
 774         for (len = 0; len < maxlen && ifname[len]; len++)
 775                 ;
 776
 777         buf = os_malloc(sizeof(int) + len);
 778         if (buf == NULL)
 779                 return;
 780
 781         os_memcpy(buf, &ievent, sizeof(int));
 782         os_memcpy(buf + sizeof(int), ifname, len);
 783
 784         wpa_priv_send_event(iface, PRIVSEP_EVENT_INTERFACE_STATUS,
 785                             buf, sizeof(int) + len);
 786
 787         os_free(buf);
 788
 789 }
 790
 791
 792 static void wpa_priv_send_ft_response(struct wpa_priv_interface *iface,
 793                                       union wpa_event_data *data)
 794 {
 795         size_t len;
 796         u8 *buf, *pos;
 797
 798         if (data == NULL || data->ft_ies.ies == NULL)
 799                 return;
 800
 801         len = sizeof(int) + ETH_ALEN + data->ft_ies.ies_len;
 802         buf = os_malloc(len);
 803         if (buf == NULL)
 804                 return;
 805
 806         pos = buf;
 807         os_memcpy(pos, &data->ft_ies.ft_action, sizeof(int));
 808         pos += sizeof(int);
 809         os_memcpy(pos, data->ft_ies.target_ap, ETH_ALEN);
 810         pos += ETH_ALEN;
 811         os_memcpy(pos, data->ft_ies.ies, data->ft_ies.ies_len);
 812
 813         wpa_priv_send_event(iface, PRIVSEP_EVENT_FT_RESPONSE, buf, len);
 814
 815         os_free(buf);
 816
 817 }
 818
 819
 820 void wpa_supplicant_event(void *ctx, enum wpa_event_type event,
 821                           union wpa_event_data *data)
 822 {
 823         struct wpa_priv_interface *iface = ctx;
 824
 825         wpa_printf(MSG_DEBUG, "%s - event=%d", __func__, event);
 826
 827         if (!iface->wpas_registered) {
 828                 wpa_printf(MSG_DEBUG, "Driver event received, but "
 829                            "wpa_supplicant not registered");
 830                 return;
 831         }
 832
 833         switch (event) {
 834         case EVENT_ASSOC:
 835                 wpa_priv_send_assoc(iface, PRIVSEP_EVENT_ASSOC, data);
 836                 break;
 837         case EVENT_DISASSOC:
 838                 wpa_priv_send_event(iface, PRIVSEP_EVENT_DISASSOC, NULL, 0);
 839                 break;
 840         case EVENT_ASSOCINFO:
 841                 if (data == NULL)
 842                         return;
 843                 wpa_priv_send_assoc(iface, PRIVSEP_EVENT_ASSOCINFO, data);
 844                 break;
 845         case EVENT_MICHAEL_MIC_FAILURE:
 846                 if (data == NULL)
 847                         return;
 848                 wpa_priv_send_event(iface, PRIVSEP_EVENT_MICHAEL_MIC_FAILURE,
 849                                     &data->michael_mic_failure.unicast,
 850                                     sizeof(int));
 851                 break;
 852         case EVENT_SCAN_RESULTS:
 853                 wpa_priv_send_event(iface, PRIVSEP_EVENT_SCAN_RESULTS, NULL,
 854                                     0);
 855                 break;
 856         case EVENT_INTERFACE_STATUS:
 857                 wpa_priv_send_interface_status(iface, data);
 858                 break;
 859         case EVENT_PMKID_CANDIDATE:
 860                 if (data == NULL)
 861                         return;
 862                 wpa_priv_send_event(iface, PRIVSEP_EVENT_PMKID_CANDIDATE,
 863                                     &data->pmkid_candidate,
 864                                     sizeof(struct pmkid_candidate));
 865                 break;
 866         case EVENT_STKSTART:
 867                 if (data == NULL)
 868                         return;
 869                 wpa_priv_send_event(iface, PRIVSEP_EVENT_STKSTART,
 870                                     &data->stkstart.peer, ETH_ALEN);
 871                 break;
 872         case EVENT_FT_RESPONSE:
 873                 wpa_priv_send_ft_response(iface, data);
 874                 break;
 875         default:
 876                 wpa_printf(MSG_DEBUG, "Unsupported driver event %d - TODO",
 877                            event);
 878                 break;
 879         }
 880 }
 881
 882
 883 void wpa_supplicant_rx_eapol(void *ctx, const u8 *src_addr,
 884                              const u8 *buf, size_t len)
 885 {
 886         struct wpa_priv_interface *iface = ctx;
 887         struct msghdr msg;
 888         struct iovec io[3];
 889         int event = PRIVSEP_EVENT_RX_EAPOL;
 890
 891         wpa_printf(MSG_DEBUG, "RX EAPOL from driver");
 892         io[0].iov_base = &event;
 893         io[0].iov_len = sizeof(event);
 894         io[1].iov_base = (u8 *) src_addr;
 895         io[1].iov_len = ETH_ALEN;
 896         io[2].iov_base = (u8 *) buf;
 897         io[2].iov_len = len;
 898
 899         os_memset(&msg, 0, sizeof(msg));
 900         msg.msg_iov = io;
 901         msg.msg_iovlen = 3;
 902         msg.msg_name = &iface->drv_addr;
 903         msg.msg_namelen = sizeof(iface->drv_addr);
 904
 905         if (sendmsg(iface->fd, &msg, 0) < 0)
 906                 perror("sendmsg(wpas_socket)");
 907 }
 908
 909
 910 static void wpa_priv_terminate(int sig, void *signal_ctx)
 911 {
 912         wpa_printf(MSG_DEBUG, "wpa_priv termination requested");
 913         eloop_terminate();
 914 }
 915
 916
 917 static void wpa_priv_fd_workaround(void)
 918 {
 919 #ifdef __linux__
 920         int s, i;
 921         /* When started from pcmcia-cs scripts, wpa_supplicant might start with
 922          * fd 0, 1, and 2 closed. This will cause some issues because many
 923          * places in wpa_supplicant are still printing out to stdout. As a
 924          * workaround, make sure that fd's 0, 1, and 2 are not used for other
 925          * sockets. */
 926         for (i = 0; i < 3; i++) {
 927                 s = open("/dev/null", O_RDWR);
 928                 if (s > 2) {
 929                         close(s);
 930                         break;
 931                 }
 932         }
 933 #endif /* __linux__ */
 934 }
 935
 936
 937 static void usage(void)
 938 {
 939         printf("wpa_priv v" VERSION_STR "\n"
 940                "Copyright (c) 2007-2009, Jouni Malinen <j@w1.fi> and "
 941                "contributors\n"
 942                "\n"
 943                "usage:\n"
 944                "  wpa_priv [-Bdd] [-P<pid file>] <driver:ifname> "
 945                "[driver:ifname ...]\n");
 946 }
 947
 948
 949 extern int wpa_debug_level;
 950
 951 int main(int argc, char *argv[])
 952 {
 953         int c, i;
 954         int ret = -1;
 955         char *pid_file = NULL;
 956         int daemonize = 0;
 957         char *ctrl_dir = "/var/run/wpa_priv";
 958         struct wpa_priv_interface *interfaces = NULL, *iface;
 959
 960         if (os_program_init())
 961                 return -1;
 962
 963         wpa_priv_fd_workaround();
 964
 965         for (;;) {
 966                 c = getopt(argc, argv, "Bc:dP:");
 967                 if (c < 0)
 968                         break;
 969                 switch (c) {
 970                 case 'B':
 971                         daemonize++;
 972                         break;
 973                 case 'c':
 974                         ctrl_dir = optarg;
 975                         break;
 976                 case 'd':
 977                         wpa_debug_level--;
 978                         break;
 979                 case 'P':
 980                         pid_file = os_rel2abs_path(optarg);
 981                         break;
 982                 default:
 983                         usage();
 984                         goto out;
 985                 }
 986         }
 987
 988         if (optind >= argc) {
 989                 usage();
 990                 goto out;
 991         }
 992
 993         wpa_printf(MSG_DEBUG, "wpa_priv control directory: '%s'", ctrl_dir);
 994
 995         if (eloop_init()) {
 996                 wpa_printf(MSG_ERROR, "Failed to initialize event loop");
 997                 goto out;
 998         }
 999
1000         for (i = optind; i < argc; i++) {
1001                 wpa_printf(MSG_DEBUG, "Adding driver:interface %s", argv[i]);
1002                 iface = wpa_priv_interface_init(ctrl_dir, argv[i]);
1003                 if (iface == NULL)
1004                         goto out;
1005                 iface->next = interfaces;
1006                 interfaces = iface;
1007         }
1008
1009         if (daemonize && os_daemonize(pid_file))
1010                 goto out;
1011
1012         eloop_register_signal_terminate(wpa_priv_terminate, NULL);
1013         eloop_run();
1014
1015         ret = 0;
1016
1017 out:
1018         iface = interfaces;
1019         while (iface) {
1020                 struct wpa_priv_interface *prev = iface;
1021                 iface = iface->next;
1022                 wpa_priv_interface_deinit(prev);
1023         }
1024
1025         eloop_destroy();
1026
1027         os_daemonize_terminate(pid_file);
1028         os_free(pid_file);
1029         os_program_deinit();
1030
1031         return ret;
1032 }