qemu-sockets.c

   1 /*
   2  *  inet and unix socket functions for qemu
   3  *
   4  *  (c) 2008 Gerd Hoffmann <kraxel@redhat.com>
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; under version 2 of the License.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  * Contributions after 2012-01-13 are licensed under the terms of the
  16  * GNU GPL, version 2 or (at your option) any later version.
  17  */
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <ctype.h>
  22 #include <errno.h>
  23 #include <unistd.h>
  24
  25 #include "qemu_socket.h"
  26 #include "qemu-common.h" /* for qemu_isdigit */
  27 #include "main-loop.h"
  28
  29 #ifndef AI_ADDRCONFIG
  30 # define AI_ADDRCONFIG 0
  31 #endif
  32
  33 static const int on=1, off=0;
  34
  35 /* used temporarely until all users are converted to QemuOpts */
  36 static QemuOptsList dummy_opts = {
  37     .name = "dummy",
  38     .head = QTAILQ_HEAD_INITIALIZER(dummy_opts.head),
  39     .desc = {
  40         {
  41             .name = "path",
  42             .type = QEMU_OPT_STRING,
  43         },{
  44             .name = "host",
  45             .type = QEMU_OPT_STRING,
  46         },{
  47             .name = "port",
  48             .type = QEMU_OPT_STRING,
  49         },{
  50             .name = "to",
  51             .type = QEMU_OPT_NUMBER,
  52         },{
  53             .name = "ipv4",
  54             .type = QEMU_OPT_BOOL,
  55         },{
  56             .name = "ipv6",
  57             .type = QEMU_OPT_BOOL,
  58         },
  59         { /* end if list */ }
  60     },
  61 };
  62
  63 static int inet_getport(struct addrinfo *e)
  64 {
  65     struct sockaddr_in *i4;
  66     struct sockaddr_in6 *i6;
  67
  68     switch (e->ai_family) {
  69     case PF_INET6:
  70         i6 = (void*)e->ai_addr;
  71         return ntohs(i6->sin6_port);
  72     case PF_INET:
  73         i4 = (void*)e->ai_addr;
  74         return ntohs(i4->sin_port);
  75     default:
  76         return 0;
  77     }
  78 }
  79
  80 static void inet_setport(struct addrinfo *e, int port)
  81 {
  82     struct sockaddr_in *i4;
  83     struct sockaddr_in6 *i6;
  84
  85     switch (e->ai_family) {
  86     case PF_INET6:
  87         i6 = (void*)e->ai_addr;
  88         i6->sin6_port = htons(port);
  89         break;
  90     case PF_INET:
  91         i4 = (void*)e->ai_addr;
  92         i4->sin_port = htons(port);
  93         break;
  94     }
  95 }
  96
  97 const char *inet_strfamily(int family)
  98 {
  99     switch (family) {
 100     case PF_INET6: return "ipv6";
 101     case PF_INET:  return "ipv4";
 102     case PF_UNIX:  return "unix";
 103     }
 104     return "unknown";
 105 }
 106
 107 int inet_listen_opts(QemuOpts *opts, int port_offset, Error **errp)
 108 {
 109     struct addrinfo ai,*res,*e;
 110     const char *addr;
 111     char port[33];
 112     char uaddr[INET6_ADDRSTRLEN+1];
 113     char uport[33];
 114     int slisten, rc, to, port_min, port_max, p;
 115
 116     memset(&ai,0, sizeof(ai));
 117     ai.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
 118     ai.ai_family = PF_UNSPEC;
 119     ai.ai_socktype = SOCK_STREAM;
 120
 121     if ((qemu_opt_get(opts, "host") == NULL) ||
 122         (qemu_opt_get(opts, "port") == NULL)) {
 123         error_setg(errp, "host and/or port not specified");
 124         return -1;
 125     }
 126     pstrcpy(port, sizeof(port), qemu_opt_get(opts, "port"));
 127     addr = qemu_opt_get(opts, "host");
 128
 129     to = qemu_opt_get_number(opts, "to", 0);
 130     if (qemu_opt_get_bool(opts, "ipv4", 0))
 131         ai.ai_family = PF_INET;
 132     if (qemu_opt_get_bool(opts, "ipv6", 0))
 133         ai.ai_family = PF_INET6;
 134
 135     /* lookup */
 136     if (port_offset)
 137         snprintf(port, sizeof(port), "%d", atoi(port) + port_offset);
 138     rc = getaddrinfo(strlen(addr) ? addr : NULL, port, &ai, &res);
 139     if (rc != 0) {
 140         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 141                    gai_strerror(rc));
 142         return -1;
 143     }
 144
 145     /* create socket + bind */
 146     for (e = res; e != NULL; e = e->ai_next) {
 147         getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
 148                         uaddr,INET6_ADDRSTRLEN,uport,32,
 149                         NI_NUMERICHOST | NI_NUMERICSERV);
 150         slisten = qemu_socket(e->ai_family, e->ai_socktype, e->ai_protocol);
 151         if (slisten < 0) {
 152             if (!e->ai_next) {
 153                 error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
 154             }
 155             continue;
 156         }
 157
 158         setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
 159 #ifdef IPV6_V6ONLY
 160         if (e->ai_family == PF_INET6) {
 161             /* listen on both ipv4 and ipv6 */
 162             setsockopt(slisten,IPPROTO_IPV6,IPV6_V6ONLY,(void*)&off,
 163                 sizeof(off));
 164         }
 165 #endif
 166
 167         port_min = inet_getport(e);
 168         port_max = to ? to + port_offset : port_min;
 169         for (p = port_min; p <= port_max; p++) {
 170             inet_setport(e, p);
 171             if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
 172                 goto listen;
 173             }
 174             if (p == port_max) {
 175                 if (!e->ai_next) {
 176                     error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
 177                 }
 178             }
 179         }
 180         closesocket(slisten);
 181     }
 182     freeaddrinfo(res);
 183     return -1;
 184
 185 listen:
 186     if (listen(slisten,1) != 0) {
 187         error_set_errno(errp, errno, QERR_SOCKET_LISTEN_FAILED);
 188         closesocket(slisten);
 189         freeaddrinfo(res);
 190         return -1;
 191     }
 192     snprintf(uport, sizeof(uport), "%d", inet_getport(e) - port_offset);
 193     qemu_opt_set(opts, "host", uaddr);
 194     qemu_opt_set(opts, "port", uport);
 195     qemu_opt_set(opts, "ipv6", (e->ai_family == PF_INET6) ? "on" : "off");
 196     qemu_opt_set(opts, "ipv4", (e->ai_family != PF_INET6) ? "on" : "off");
 197     freeaddrinfo(res);
 198     return slisten;
 199 }
 200
 201 #ifdef _WIN32
 202 #define QEMU_SOCKET_RC_INPROGRESS(rc) \
 203     ((rc) == -EINPROGRESS || (rc) == -EWOULDBLOCK || (rc) == -WSAEALREADY)
 204 #else
 205 #define QEMU_SOCKET_RC_INPROGRESS(rc) \
 206     ((rc) == -EINPROGRESS)
 207 #endif
 208
 209 /* Struct to store connect state for non blocking connect */
 210 typedef struct ConnectState {
 211     int fd;
 212     struct addrinfo *addr_list;
 213     struct addrinfo *current_addr;
 214     NonBlockingConnectHandler *callback;
 215     void *opaque;
 216 } ConnectState;
 217
 218 static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
 219                              ConnectState *connect_state, Error **errp);
 220
 221 static void wait_for_connect(void *opaque)
 222 {
 223     ConnectState *s = opaque;
 224     int val = 0, rc = 0;
 225     socklen_t valsize = sizeof(val);
 226     bool in_progress;
 227
 228     qemu_set_fd_handler2(s->fd, NULL, NULL, NULL, NULL);
 229
 230     do {
 231         rc = getsockopt(s->fd, SOL_SOCKET, SO_ERROR, (void *) &val, &valsize);
 232     } while (rc == -1 && socket_error() == EINTR);
 233
 234     /* update rc to contain error */
 235     if (!rc && val) {
 236         rc = -1;
 237     }
 238
 239     /* connect error */
 240     if (rc < 0) {
 241         closesocket(s->fd);
 242         s->fd = rc;
 243     }
 244
 245     /* try to connect to the next address on the list */
 246     if (s->current_addr) {
 247         while (s->current_addr->ai_next != NULL && s->fd < 0) {
 248             s->current_addr = s->current_addr->ai_next;
 249             s->fd = inet_connect_addr(s->current_addr, &in_progress, s, NULL);
 250             /* connect in progress */
 251             if (in_progress) {
 252                 return;
 253             }
 254         }
 255
 256         freeaddrinfo(s->addr_list);
 257     }
 258
 259     if (s->callback) {
 260         s->callback(s->fd, s->opaque);
 261     }
 262     g_free(s);
 263 }
 264
 265 static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
 266                              ConnectState *connect_state, Error **errp)
 267 {
 268     int sock, rc;
 269
 270     *in_progress = false;
 271
 272     sock = qemu_socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
 273     if (sock < 0) {
 274         error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
 275         return -1;
 276     }
 277     qemu_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
 278     if (connect_state != NULL) {
 279         socket_set_nonblock(sock);
 280     }
 281     /* connect to peer */
 282     do {
 283         rc = 0;
 284         if (connect(sock, addr->ai_addr, addr->ai_addrlen) < 0) {
 285             rc = -socket_error();
 286         }
 287     } while (rc == -EINTR);
 288
 289     if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
 290         connect_state->fd = sock;
 291         qemu_set_fd_handler2(sock, NULL, NULL, wait_for_connect,
 292                              connect_state);
 293         *in_progress = true;
 294     } else if (rc < 0) {
 295         error_set_errno(errp, errno, QERR_SOCKET_CONNECT_FAILED);
 296         closesocket(sock);
 297         return -1;
 298     }
 299     return sock;
 300 }
 301
 302 static struct addrinfo *inet_parse_connect_opts(QemuOpts *opts, Error **errp)
 303 {
 304     struct addrinfo ai, *res;
 305     int rc;
 306     const char *addr;
 307     const char *port;
 308
 309     memset(&ai, 0, sizeof(ai));
 310
 311     ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
 312     ai.ai_family = PF_UNSPEC;
 313     ai.ai_socktype = SOCK_STREAM;
 314
 315     addr = qemu_opt_get(opts, "host");
 316     port = qemu_opt_get(opts, "port");
 317     if (addr == NULL || port == NULL) {
 318         error_setg(errp, "host and/or port not specified");
 319         return NULL;
 320     }
 321
 322     if (qemu_opt_get_bool(opts, "ipv4", 0)) {
 323         ai.ai_family = PF_INET;
 324     }
 325     if (qemu_opt_get_bool(opts, "ipv6", 0)) {
 326         ai.ai_family = PF_INET6;
 327     }
 328
 329     /* lookup */
 330     rc = getaddrinfo(addr, port, &ai, &res);
 331     if (rc != 0) {
 332         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 333                    gai_strerror(rc));
 334         return NULL;
 335     }
 336     return res;
 337 }
 338
 339 /**
 340  * Create a socket and connect it to an address.
 341  *
 342  * @opts: QEMU options, recognized parameters strings "host" and "port",
 343  *        bools "ipv4" and "ipv6".
 344  * @errp: set on error
 345  * @callback: callback function for non-blocking connect
 346  * @opaque: opaque for callback function
 347  *
 348  * Returns: -1 on error, file descriptor on success.
 349  *
 350  * If @callback is non-null, the connect is non-blocking.  If this
 351  * function succeeds, callback will be called when the connection
 352  * completes, with the file descriptor on success, or -1 on error.
 353  */
 354 int inet_connect_opts(QemuOpts *opts, Error **errp,
 355                       NonBlockingConnectHandler *callback, void *opaque)
 356 {
 357     struct addrinfo *res, *e;
 358     int sock = -1;
 359     bool in_progress;
 360     ConnectState *connect_state = NULL;
 361
 362     res = inet_parse_connect_opts(opts, errp);
 363     if (!res) {
 364         return -1;
 365     }
 366
 367     if (callback != NULL) {
 368         connect_state = g_malloc0(sizeof(*connect_state));
 369         connect_state->addr_list = res;
 370         connect_state->callback = callback;
 371         connect_state->opaque = opaque;
 372     }
 373
 374     for (e = res; e != NULL; e = e->ai_next) {
 375         if (connect_state != NULL) {
 376             connect_state->current_addr = e;
 377         }
 378         sock = inet_connect_addr(e, &in_progress, connect_state, errp);
 379         if (in_progress) {
 380             return sock;
 381         } else if (sock >= 0) {
 382             /* non blocking socket immediate success, call callback */
 383             if (callback != NULL) {
 384                 callback(sock, opaque);
 385             }
 386             break;
 387         }
 388     }
 389     g_free(connect_state);
 390     freeaddrinfo(res);
 391     return sock;
 392 }
 393
 394 int inet_dgram_opts(QemuOpts *opts, Error **errp)
 395 {
 396     struct addrinfo ai, *peer = NULL, *local = NULL;
 397     const char *addr;
 398     const char *port;
 399     int sock = -1, rc;
 400
 401     /* lookup peer addr */
 402     memset(&ai,0, sizeof(ai));
 403     ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
 404     ai.ai_family = PF_UNSPEC;
 405     ai.ai_socktype = SOCK_DGRAM;
 406
 407     addr = qemu_opt_get(opts, "host");
 408     port = qemu_opt_get(opts, "port");
 409     if (addr == NULL || strlen(addr) == 0) {
 410         addr = "localhost";
 411     }
 412     if (port == NULL || strlen(port) == 0) {
 413         error_setg(errp, "remote port not specified");
 414         return -1;
 415     }
 416
 417     if (qemu_opt_get_bool(opts, "ipv4", 0))
 418         ai.ai_family = PF_INET;
 419     if (qemu_opt_get_bool(opts, "ipv6", 0))
 420         ai.ai_family = PF_INET6;
 421
 422     if (0 != (rc = getaddrinfo(addr, port, &ai, &peer))) {
 423         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 424                    gai_strerror(rc));
 425         return -1;
 426     }
 427
 428     /* lookup local addr */
 429     memset(&ai,0, sizeof(ai));
 430     ai.ai_flags = AI_PASSIVE;
 431     ai.ai_family = peer->ai_family;
 432     ai.ai_socktype = SOCK_DGRAM;
 433
 434     addr = qemu_opt_get(opts, "localaddr");
 435     port = qemu_opt_get(opts, "localport");
 436     if (addr == NULL || strlen(addr) == 0) {
 437         addr = NULL;
 438     }
 439     if (!port || strlen(port) == 0)
 440         port = "0";
 441
 442     if (0 != (rc = getaddrinfo(addr, port, &ai, &local))) {
 443         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 444                    gai_strerror(rc));
 445         goto err;
 446     }
 447
 448     /* create socket */
 449     sock = qemu_socket(peer->ai_family, peer->ai_socktype, peer->ai_protocol);
 450     if (sock < 0) {
 451         error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
 452         goto err;
 453     }
 454     setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,(void*)&on,sizeof(on));
 455
 456     /* bind socket */
 457     if (bind(sock, local->ai_addr, local->ai_addrlen) < 0) {
 458         error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
 459         goto err;
 460     }
 461
 462     /* connect to peer */
 463     if (connect(sock,peer->ai_addr,peer->ai_addrlen) < 0) {
 464         error_set_errno(errp, errno, QERR_SOCKET_CONNECT_FAILED);
 465         goto err;
 466     }
 467
 468     freeaddrinfo(local);
 469     freeaddrinfo(peer);
 470     return sock;
 471
 472 err:
 473     if (-1 != sock)
 474         closesocket(sock);
 475     if (local)
 476         freeaddrinfo(local);
 477     if (peer)
 478         freeaddrinfo(peer);
 479     return -1;
 480 }
 481
 482 /* compatibility wrapper */
 483 static InetSocketAddress *inet_parse(const char *str, Error **errp)
 484 {
 485     InetSocketAddress *addr;
 486     const char *optstr, *h;
 487     char host[64];
 488     char port[33];
 489     int to;
 490     int pos;
 491
 492     addr = g_new0(InetSocketAddress, 1);
 493
 494     /* parse address */
 495     if (str[0] == ':') {
 496         /* no host given */
 497         host[0] = '\0';
 498         if (1 != sscanf(str, ":%32[^,]%n", port, &pos)) {
 499             error_setg(errp, "error parsing port in address '%s'", str);
 500             goto fail;
 501         }
 502     } else if (str[0] == '[') {
 503         /* IPv6 addr */
 504         if (2 != sscanf(str, "[%64[^]]]:%32[^,]%n", host, port, &pos)) {
 505             error_setg(errp, "error parsing IPv6 address '%s'", str);
 506             goto fail;
 507         }
 508         addr->ipv6 = addr->has_ipv6 = true;
 509     } else if (qemu_isdigit(str[0])) {
 510         /* IPv4 addr */
 511         if (2 != sscanf(str, "%64[0-9.]:%32[^,]%n", host, port, &pos)) {
 512             error_setg(errp, "error parsing IPv4 address '%s'", str);
 513             goto fail;
 514         }
 515         addr->ipv4 = addr->has_ipv4 = true;
 516     } else {
 517         /* hostname */
 518         if (2 != sscanf(str, "%64[^:]:%32[^,]%n", host, port, &pos)) {
 519             error_setg(errp, "error parsing address '%s'", str);
 520             goto fail;
 521         }
 522     }
 523
 524     addr->host = g_strdup(host);
 525     addr->port = g_strdup(port);
 526
 527     /* parse options */
 528     optstr = str + pos;
 529     h = strstr(optstr, ",to=");
 530     if (h) {
 531         if (1 != sscanf(str, "%d%n", &to, &pos) ||
 532             (str[pos] != '\0' && str[pos] != ',')) {
 533             error_setg(errp, "error parsing to= argument");
 534             goto fail;
 535         }
 536         addr->has_to = true;
 537         addr->to = to;
 538     }
 539     if (strstr(optstr, ",ipv4")) {
 540         addr->ipv4 = addr->has_ipv4 = true;
 541     }
 542     if (strstr(optstr, ",ipv6")) {
 543         addr->ipv6 = addr->has_ipv6 = true;
 544     }
 545     return addr;
 546
 547 fail:
 548     qapi_free_InetSocketAddress(addr);
 549     return NULL;
 550 }
 551
 552 static void inet_addr_to_opts(QemuOpts *opts, InetSocketAddress *addr)
 553 {
 554     bool ipv4 = addr->ipv4 || !addr->has_ipv4;
 555     bool ipv6 = addr->ipv6 || !addr->has_ipv6;
 556
 557     if (!ipv4 || !ipv6) {
 558         qemu_opt_set_bool(opts, "ipv4", ipv4);
 559         qemu_opt_set_bool(opts, "ipv6", ipv6);
 560     }
 561     if (addr->has_to) {
 562         char to[20];
 563         snprintf(to, sizeof(to), "%d", addr->to);
 564         qemu_opt_set(opts, "to", to);
 565     }
 566     qemu_opt_set(opts, "host", addr->host);
 567     qemu_opt_set(opts, "port", addr->port);
 568 }
 569
 570 int inet_listen(const char *str, char *ostr, int olen,
 571                 int socktype, int port_offset, Error **errp)
 572 {
 573     QemuOpts *opts;
 574     char *optstr;
 575     int sock = -1;
 576     InetSocketAddress *addr;
 577
 578     addr = inet_parse(str, errp);
 579     if (addr != NULL) {
 580         opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 581         inet_addr_to_opts(opts, addr);
 582         qapi_free_InetSocketAddress(addr);
 583         sock = inet_listen_opts(opts, port_offset, errp);
 584         if (sock != -1 && ostr) {
 585             optstr = strchr(str, ',');
 586             if (qemu_opt_get_bool(opts, "ipv6", 0)) {
 587                 snprintf(ostr, olen, "[%s]:%s%s",
 588                          qemu_opt_get(opts, "host"),
 589                          qemu_opt_get(opts, "port"),
 590                          optstr ? optstr : "");
 591             } else {
 592                 snprintf(ostr, olen, "%s:%s%s",
 593                          qemu_opt_get(opts, "host"),
 594                          qemu_opt_get(opts, "port"),
 595                          optstr ? optstr : "");
 596             }
 597         }
 598         qemu_opts_del(opts);
 599     }
 600     return sock;
 601 }
 602
 603 /**
 604  * Create a blocking socket and connect it to an address.
 605  *
 606  * @str: address string
 607  * @errp: set in case of an error
 608  *
 609  * Returns -1 in case of error, file descriptor on success
 610  **/
 611 int inet_connect(const char *str, Error **errp)
 612 {
 613     QemuOpts *opts;
 614     int sock = -1;
 615     InetSocketAddress *addr;
 616
 617     addr = inet_parse(str, errp);
 618     if (addr != NULL) {
 619         opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 620         inet_addr_to_opts(opts, addr);
 621         qapi_free_InetSocketAddress(addr);
 622         sock = inet_connect_opts(opts, errp, NULL, NULL);
 623         qemu_opts_del(opts);
 624     }
 625     return sock;
 626 }
 627
 628 /**
 629  * Create a non-blocking socket and connect it to an address.
 630  * Calls the callback function with fd in case of success or -1 in case of
 631  * error.
 632  *
 633  * @str: address string
 634  * @callback: callback function that is called when connect completes,
 635  *            cannot be NULL.
 636  * @opaque: opaque for callback function
 637  * @errp: set in case of an error
 638  *
 639  * Returns: -1 on immediate error, file descriptor on success.
 640  **/
 641 int inet_nonblocking_connect(const char *str,
 642                              NonBlockingConnectHandler *callback,
 643                              void *opaque, Error **errp)
 644 {
 645     QemuOpts *opts;
 646     int sock = -1;
 647     InetSocketAddress *addr;
 648
 649     g_assert(callback != NULL);
 650
 651     addr = inet_parse(str, errp);
 652     if (addr != NULL) {
 653         opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 654         inet_addr_to_opts(opts, addr);
 655         qapi_free_InetSocketAddress(addr);
 656         sock = inet_connect_opts(opts, errp, callback, opaque);
 657         qemu_opts_del(opts);
 658     }
 659     return sock;
 660 }
 661
 662 #ifndef _WIN32
 663
 664 int unix_listen_opts(QemuOpts *opts, Error **errp)
 665 {
 666     struct sockaddr_un un;
 667     const char *path = qemu_opt_get(opts, "path");
 668     int sock, fd;
 669
 670     sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
 671     if (sock < 0) {
 672         error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
 673         return -1;
 674     }
 675
 676     memset(&un, 0, sizeof(un));
 677     un.sun_family = AF_UNIX;
 678     if (path && strlen(path)) {
 679         snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
 680     } else {
 681         char *tmpdir = getenv("TMPDIR");
 682         snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
 683                  tmpdir ? tmpdir : "/tmp");
 684         /*
 685          * This dummy fd usage silences the mktemp() unsecure warning.
 686          * Using mkstemp() doesn't make things more secure here
 687          * though.  bind() complains about existing files, so we have
 688          * to unlink first and thus re-open the race window.  The
 689          * worst case possible is bind() failing, i.e. a DoS attack.
 690          */
 691         fd = mkstemp(un.sun_path); close(fd);
 692         qemu_opt_set(opts, "path", un.sun_path);
 693     }
 694
 695     unlink(un.sun_path);
 696     if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
 697         error_set_errno(errp, errno, QERR_SOCKET_BIND_FAILED);
 698         goto err;
 699     }
 700     if (listen(sock, 1) < 0) {
 701         error_set_errno(errp, errno, QERR_SOCKET_LISTEN_FAILED);
 702         goto err;
 703     }
 704
 705     return sock;
 706
 707 err:
 708     closesocket(sock);
 709     return -1;
 710 }
 711
 712 int unix_connect_opts(QemuOpts *opts, Error **errp,
 713                       NonBlockingConnectHandler *callback, void *opaque)
 714 {
 715     struct sockaddr_un un;
 716     const char *path = qemu_opt_get(opts, "path");
 717     ConnectState *connect_state = NULL;
 718     int sock, rc;
 719
 720     if (NULL == path) {
 721         error_setg(errp, "unix connect: no path specified\n");
 722         return -1;
 723     }
 724
 725     sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
 726     if (sock < 0) {
 727         error_set_errno(errp, errno, QERR_SOCKET_CREATE_FAILED);
 728         return -1;
 729     }
 730     if (callback != NULL) {
 731         connect_state = g_malloc0(sizeof(*connect_state));
 732         connect_state->callback = callback;
 733         connect_state->opaque = opaque;
 734         socket_set_nonblock(sock);
 735     }
 736
 737     memset(&un, 0, sizeof(un));
 738     un.sun_family = AF_UNIX;
 739     snprintf(un.sun_path, sizeof(un.sun_path), "%s", path);
 740
 741     /* connect to peer */
 742     do {
 743         rc = 0;
 744         if (connect(sock, (struct sockaddr *) &un, sizeof(un)) < 0) {
 745             rc = -socket_error();
 746         }
 747     } while (rc == -EINTR);
 748
 749     if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
 750         connect_state->fd = sock;
 751         qemu_set_fd_handler2(sock, NULL, NULL, wait_for_connect,
 752                              connect_state);
 753         return sock;
 754     } else if (rc >= 0) {
 755         /* non blocking socket immediate success, call callback */
 756         if (callback != NULL) {
 757             callback(sock, opaque);
 758         }
 759     }
 760
 761     if (rc < 0) {
 762         error_set_errno(errp, -rc, QERR_SOCKET_CONNECT_FAILED);
 763         close(sock);
 764         sock = -1;
 765     }
 766
 767     g_free(connect_state);
 768     return sock;
 769 }
 770
 771 #else
 772
 773 int unix_listen_opts(QemuOpts *opts, Error **errp)
 774 {
 775     error_setg(errp, "unix sockets are not available on windows");
 776     errno = ENOTSUP;
 777     return -1;
 778 }
 779
 780 int unix_connect_opts(QemuOpts *opts, Error **errp,
 781                       NonBlockingConnectHandler *callback, void *opaque)
 782 {
 783     error_setg(errp, "unix sockets are not available on windows");
 784     errno = ENOTSUP;
 785     return -1;
 786 }
 787 #endif
 788
 789 /* compatibility wrapper */
 790 int unix_listen(const char *str, char *ostr, int olen, Error **errp)
 791 {
 792     QemuOpts *opts;
 793     char *path, *optstr;
 794     int sock, len;
 795
 796     opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 797
 798     optstr = strchr(str, ',');
 799     if (optstr) {
 800         len = optstr - str;
 801         if (len) {
 802             path = g_malloc(len+1);
 803             snprintf(path, len+1, "%.*s", len, str);
 804             qemu_opt_set(opts, "path", path);
 805             g_free(path);
 806         }
 807     } else {
 808         qemu_opt_set(opts, "path", str);
 809     }
 810
 811     sock = unix_listen_opts(opts, errp);
 812
 813     if (sock != -1 && ostr)
 814         snprintf(ostr, olen, "%s%s", qemu_opt_get(opts, "path"), optstr ? optstr : "");
 815     qemu_opts_del(opts);
 816     return sock;
 817 }
 818
 819 int unix_connect(const char *path, Error **errp)
 820 {
 821     QemuOpts *opts;
 822     int sock;
 823
 824     opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 825     qemu_opt_set(opts, "path", path);
 826     sock = unix_connect_opts(opts, errp, NULL, NULL);
 827     qemu_opts_del(opts);
 828     return sock;
 829 }
 830
 831
 832 int unix_nonblocking_connect(const char *path,
 833                              NonBlockingConnectHandler *callback,
 834                              void *opaque, Error **errp)
 835 {
 836     QemuOpts *opts;
 837     int sock = -1;
 838
 839     g_assert(callback != NULL);
 840
 841     opts = qemu_opts_create(&dummy_opts, NULL, 0, NULL);
 842     qemu_opt_set(opts, "path", path);
 843     sock = unix_connect_opts(opts, errp, callback, opaque);
 844     qemu_opts_del(opts);
 845     return sock;
 846 }
 847
 848 #ifdef _WIN32
 849 static void socket_cleanup(void)
 850 {
 851     WSACleanup();
 852 }
 853 #endif
 854
 855 int socket_init(void)
 856 {
 857 #ifdef _WIN32
 858     WSADATA Data;
 859     int ret, err;
 860
 861     ret = WSAStartup(MAKEWORD(2,2), &Data);
 862     if (ret != 0) {
 863         err = WSAGetLastError();
 864         fprintf(stderr, "WSAStartup: %d\n", err);
 865         return -1;
 866     }
 867     atexit(socket_cleanup);
 868 #endif
 869     return 0;
 870 }