util/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 "qemu/osdep.h"
  19
  20 #include "monitor/monitor.h"
  21 #include "qemu/sockets.h"
  22 #include "qemu/main-loop.h"
  23 #include "qapi/qmp-input-visitor.h"
  24 #include "qapi/qmp-output-visitor.h"
  25 #include "qapi-visit.h"
  26
  27 #ifndef AI_ADDRCONFIG
  28 # define AI_ADDRCONFIG 0
  29 #endif
  30 #ifndef AI_V4MAPPED
  31 # define AI_V4MAPPED 0
  32 #endif
  33
  34
  35 static int inet_getport(struct addrinfo *e)
  36 {
  37     struct sockaddr_in *i4;
  38     struct sockaddr_in6 *i6;
  39
  40     switch (e->ai_family) {
  41     case PF_INET6:
  42         i6 = (void*)e->ai_addr;
  43         return ntohs(i6->sin6_port);
  44     case PF_INET:
  45         i4 = (void*)e->ai_addr;
  46         return ntohs(i4->sin_port);
  47     default:
  48         return 0;
  49     }
  50 }
  51
  52 static void inet_setport(struct addrinfo *e, int port)
  53 {
  54     struct sockaddr_in *i4;
  55     struct sockaddr_in6 *i6;
  56
  57     switch (e->ai_family) {
  58     case PF_INET6:
  59         i6 = (void*)e->ai_addr;
  60         i6->sin6_port = htons(port);
  61         break;
  62     case PF_INET:
  63         i4 = (void*)e->ai_addr;
  64         i4->sin_port = htons(port);
  65         break;
  66     }
  67 }
  68
  69 NetworkAddressFamily inet_netfamily(int family)
  70 {
  71     switch (family) {
  72     case PF_INET6: return NETWORK_ADDRESS_FAMILY_IPV6;
  73     case PF_INET:  return NETWORK_ADDRESS_FAMILY_IPV4;
  74     case PF_UNIX:  return NETWORK_ADDRESS_FAMILY_UNIX;
  75     }
  76     return NETWORK_ADDRESS_FAMILY_UNKNOWN;
  77 }
  78
  79 /*
  80  * Matrix we're trying to apply
  81  *
  82  *  ipv4  ipv6   family
  83  *   -     -       PF_UNSPEC
  84  *   -     f       PF_INET
  85  *   -     t       PF_INET6
  86  *   f     -       PF_INET6
  87  *   f     f       <error>
  88  *   f     t       PF_INET6
  89  *   t     -       PF_INET
  90  *   t     f       PF_INET
  91  *   t     t       PF_INET6
  92  *
  93  * NB, this matrix is only about getting the neccessary results
  94  * from getaddrinfo(). Some of the cases require further work
  95  * after reading results from getaddrinfo in order to fully
  96  * apply the logic the end user wants. eg with the last case
  97  * ipv4=t + ipv6=t + PF_INET6, getaddrinfo alone can only
  98  * guarantee the ipv6=t part of the request - we need more
  99  * checks to provide ipv4=t part of the guarantee. This is
 100  * outside scope of this method and not currently handled by
 101  * callers at all.
 102  */
 103 static int inet_ai_family_from_address(InetSocketAddress *addr,
 104                                        Error **errp)
 105 {
 106     if (addr->has_ipv6 && addr->has_ipv4 &&
 107         !addr->ipv6 && !addr->ipv4) {
 108         error_setg(errp, "Cannot disable IPv4 and IPv6 at same time");
 109         return PF_UNSPEC;
 110     }
 111     if ((addr->has_ipv6 && addr->ipv6) || (addr->has_ipv4 && !addr->ipv4)) {
 112         return PF_INET6;
 113     }
 114     if ((addr->has_ipv4 && addr->ipv4) || (addr->has_ipv6 && !addr->ipv6)) {
 115         return PF_INET;
 116     }
 117     return PF_UNSPEC;
 118 }
 119
 120 static int inet_listen_saddr(InetSocketAddress *saddr,
 121                              int port_offset,
 122                              bool update_addr,
 123                              Error **errp)
 124 {
 125     struct addrinfo ai,*res,*e;
 126     char port[33];
 127     char uaddr[INET6_ADDRSTRLEN+1];
 128     char uport[33];
 129     int slisten, rc, port_min, port_max, p;
 130     Error *err = NULL;
 131
 132     memset(&ai,0, sizeof(ai));
 133     ai.ai_flags = AI_PASSIVE;
 134     ai.ai_family = inet_ai_family_from_address(saddr, &err);
 135     ai.ai_socktype = SOCK_STREAM;
 136
 137     if (err) {
 138         error_propagate(errp, err);
 139         return -1;
 140     }
 141
 142     if (saddr->host == NULL) {
 143         error_setg(errp, "host not specified");
 144         return -1;
 145     }
 146     if (saddr->port != NULL) {
 147         pstrcpy(port, sizeof(port), saddr->port);
 148     } else {
 149         port[0] = '\0';
 150     }
 151
 152     /* lookup */
 153     if (port_offset) {
 154         unsigned long long baseport;
 155         if (strlen(port) == 0) {
 156             error_setg(errp, "port not specified");
 157             return -1;
 158         }
 159         if (parse_uint_full(port, &baseport, 10) < 0) {
 160             error_setg(errp, "can't convert to a number: %s", port);
 161             return -1;
 162         }
 163         if (baseport > 65535 ||
 164             baseport + port_offset > 65535) {
 165             error_setg(errp, "port %s out of range", port);
 166             return -1;
 167         }
 168         snprintf(port, sizeof(port), "%d", (int)baseport + port_offset);
 169     }
 170     rc = getaddrinfo(strlen(saddr->host) ? saddr->host : NULL,
 171                      strlen(port) ? port : NULL, &ai, &res);
 172     if (rc != 0) {
 173         error_setg(errp, "address resolution failed for %s:%s: %s",
 174                    saddr->host, port, gai_strerror(rc));
 175         return -1;
 176     }
 177
 178     /* create socket + bind */
 179     for (e = res; e != NULL; e = e->ai_next) {
 180         getnameinfo((struct sockaddr*)e->ai_addr,e->ai_addrlen,
 181                         uaddr,INET6_ADDRSTRLEN,uport,32,
 182                         NI_NUMERICHOST | NI_NUMERICSERV);
 183         slisten = qemu_socket(e->ai_family, e->ai_socktype, e->ai_protocol);
 184         if (slisten < 0) {
 185             if (!e->ai_next) {
 186                 error_setg_errno(errp, errno, "Failed to create socket");
 187             }
 188             continue;
 189         }
 190
 191         socket_set_fast_reuse(slisten);
 192 #ifdef IPV6_V6ONLY
 193         if (e->ai_family == PF_INET6) {
 194             /* listen on both ipv4 and ipv6 */
 195             const int off = 0;
 196             qemu_setsockopt(slisten, IPPROTO_IPV6, IPV6_V6ONLY, &off,
 197                             sizeof(off));
 198         }
 199 #endif
 200
 201         port_min = inet_getport(e);
 202         port_max = saddr->has_to ? saddr->to + port_offset : port_min;
 203         for (p = port_min; p <= port_max; p++) {
 204             inet_setport(e, p);
 205             if (bind(slisten, e->ai_addr, e->ai_addrlen) == 0) {
 206                 goto listen;
 207             }
 208             if (p == port_max) {
 209                 if (!e->ai_next) {
 210                     error_setg_errno(errp, errno, "Failed to bind socket");
 211                 }
 212             }
 213         }
 214         closesocket(slisten);
 215     }
 216     freeaddrinfo(res);
 217     return -1;
 218
 219 listen:
 220     if (listen(slisten,1) != 0) {
 221         error_setg_errno(errp, errno, "Failed to listen on socket");
 222         closesocket(slisten);
 223         freeaddrinfo(res);
 224         return -1;
 225     }
 226     if (update_addr) {
 227         g_free(saddr->host);
 228         saddr->host = g_strdup(uaddr);
 229         g_free(saddr->port);
 230         saddr->port = g_strdup_printf("%d",
 231                                       inet_getport(e) - port_offset);
 232         saddr->has_ipv6 = saddr->ipv6 = e->ai_family == PF_INET6;
 233         saddr->has_ipv4 = saddr->ipv4 = e->ai_family != PF_INET6;
 234     }
 235     freeaddrinfo(res);
 236     return slisten;
 237 }
 238
 239 #ifdef _WIN32
 240 #define QEMU_SOCKET_RC_INPROGRESS(rc) \
 241     ((rc) == -EINPROGRESS || (rc) == -EWOULDBLOCK || (rc) == -WSAEALREADY)
 242 #else
 243 #define QEMU_SOCKET_RC_INPROGRESS(rc) \
 244     ((rc) == -EINPROGRESS)
 245 #endif
 246
 247 /* Struct to store connect state for non blocking connect */
 248 typedef struct ConnectState {
 249     int fd;
 250     struct addrinfo *addr_list;
 251     struct addrinfo *current_addr;
 252     NonBlockingConnectHandler *callback;
 253     void *opaque;
 254 } ConnectState;
 255
 256 static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
 257                              ConnectState *connect_state, Error **errp);
 258
 259 static void wait_for_connect(void *opaque)
 260 {
 261     ConnectState *s = opaque;
 262     int val = 0, rc = 0;
 263     socklen_t valsize = sizeof(val);
 264     bool in_progress;
 265     Error *err = NULL;
 266
 267     qemu_set_fd_handler(s->fd, NULL, NULL, NULL);
 268
 269     do {
 270         rc = qemu_getsockopt(s->fd, SOL_SOCKET, SO_ERROR, &val, &valsize);
 271     } while (rc == -1 && socket_error() == EINTR);
 272
 273     /* update rc to contain error */
 274     if (!rc && val) {
 275         rc = -1;
 276         errno = val;
 277     }
 278
 279     /* connect error */
 280     if (rc < 0) {
 281         error_setg_errno(&err, errno, "Error connecting to socket");
 282         closesocket(s->fd);
 283         s->fd = rc;
 284     }
 285
 286     /* try to connect to the next address on the list */
 287     if (s->current_addr) {
 288         while (s->current_addr->ai_next != NULL && s->fd < 0) {
 289             s->current_addr = s->current_addr->ai_next;
 290             s->fd = inet_connect_addr(s->current_addr, &in_progress, s, NULL);
 291             if (s->fd < 0) {
 292                 error_free(err);
 293                 err = NULL;
 294                 error_setg_errno(&err, errno, "Unable to start socket connect");
 295             }
 296             /* connect in progress */
 297             if (in_progress) {
 298                 goto out;
 299             }
 300         }
 301
 302         freeaddrinfo(s->addr_list);
 303     }
 304
 305     if (s->callback) {
 306         s->callback(s->fd, err, s->opaque);
 307     }
 308     g_free(s);
 309 out:
 310     error_free(err);
 311 }
 312
 313 static int inet_connect_addr(struct addrinfo *addr, bool *in_progress,
 314                              ConnectState *connect_state, Error **errp)
 315 {
 316     int sock, rc;
 317
 318     *in_progress = false;
 319
 320     sock = qemu_socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol);
 321     if (sock < 0) {
 322         error_setg_errno(errp, errno, "Failed to create socket");
 323         return -1;
 324     }
 325     socket_set_fast_reuse(sock);
 326     if (connect_state != NULL) {
 327         qemu_set_nonblock(sock);
 328     }
 329     /* connect to peer */
 330     do {
 331         rc = 0;
 332         if (connect(sock, addr->ai_addr, addr->ai_addrlen) < 0) {
 333             rc = -socket_error();
 334         }
 335     } while (rc == -EINTR);
 336
 337     if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
 338         connect_state->fd = sock;
 339         qemu_set_fd_handler(sock, NULL, wait_for_connect, connect_state);
 340         *in_progress = true;
 341     } else if (rc < 0) {
 342         error_setg_errno(errp, errno, "Failed to connect socket");
 343         closesocket(sock);
 344         return -1;
 345     }
 346     return sock;
 347 }
 348
 349 static struct addrinfo *inet_parse_connect_saddr(InetSocketAddress *saddr,
 350                                                  Error **errp)
 351 {
 352     struct addrinfo ai, *res;
 353     int rc;
 354     Error *err = NULL;
 355
 356     memset(&ai, 0, sizeof(ai));
 357
 358     ai.ai_flags = AI_CANONNAME | AI_V4MAPPED | AI_ADDRCONFIG;
 359     ai.ai_family = inet_ai_family_from_address(saddr, &err);
 360     ai.ai_socktype = SOCK_STREAM;
 361
 362     if (err) {
 363         error_propagate(errp, err);
 364         return NULL;
 365     }
 366
 367     if (saddr->host == NULL || saddr->port == NULL) {
 368         error_setg(errp, "host and/or port not specified");
 369         return NULL;
 370     }
 371
 372     /* lookup */
 373     rc = getaddrinfo(saddr->host, saddr->port, &ai, &res);
 374     if (rc != 0) {
 375         error_setg(errp, "address resolution failed for %s:%s: %s",
 376                    saddr->host, saddr->port, gai_strerror(rc));
 377         return NULL;
 378     }
 379     return res;
 380 }
 381
 382 /**
 383  * Create a socket and connect it to an address.
 384  *
 385  * @saddr: Inet socket address specification
 386  * @errp: set on error
 387  * @callback: callback function for non-blocking connect
 388  * @opaque: opaque for callback function
 389  *
 390  * Returns: -1 on error, file descriptor on success.
 391  *
 392  * If @callback is non-null, the connect is non-blocking.  If this
 393  * function succeeds, callback will be called when the connection
 394  * completes, with the file descriptor on success, or -1 on error.
 395  */
 396 static int inet_connect_saddr(InetSocketAddress *saddr, Error **errp,
 397                               NonBlockingConnectHandler *callback, void *opaque)
 398 {
 399     Error *local_err = NULL;
 400     struct addrinfo *res, *e;
 401     int sock = -1;
 402     bool in_progress;
 403     ConnectState *connect_state = NULL;
 404
 405     res = inet_parse_connect_saddr(saddr, errp);
 406     if (!res) {
 407         return -1;
 408     }
 409
 410     if (callback != NULL) {
 411         connect_state = g_malloc0(sizeof(*connect_state));
 412         connect_state->addr_list = res;
 413         connect_state->callback = callback;
 414         connect_state->opaque = opaque;
 415     }
 416
 417     for (e = res; e != NULL; e = e->ai_next) {
 418         error_free(local_err);
 419         local_err = NULL;
 420         if (connect_state != NULL) {
 421             connect_state->current_addr = e;
 422         }
 423         sock = inet_connect_addr(e, &in_progress, connect_state, &local_err);
 424         if (sock >= 0) {
 425             break;
 426         }
 427     }
 428
 429     if (sock < 0) {
 430         error_propagate(errp, local_err);
 431     } else if (in_progress) {
 432         /* wait_for_connect() will do the rest */
 433         return sock;
 434     } else {
 435         if (callback) {
 436             callback(sock, NULL, opaque);
 437         }
 438     }
 439     g_free(connect_state);
 440     freeaddrinfo(res);
 441     return sock;
 442 }
 443
 444 static int inet_dgram_saddr(InetSocketAddress *sraddr,
 445                             InetSocketAddress *sladdr,
 446                             Error **errp)
 447 {
 448     struct addrinfo ai, *peer = NULL, *local = NULL;
 449     const char *addr;
 450     const char *port;
 451     int sock = -1, rc;
 452     Error *err = NULL;
 453
 454     /* lookup peer addr */
 455     memset(&ai,0, sizeof(ai));
 456     ai.ai_flags = AI_CANONNAME | AI_V4MAPPED | AI_ADDRCONFIG;
 457     ai.ai_family = inet_ai_family_from_address(sraddr, &err);
 458     ai.ai_socktype = SOCK_DGRAM;
 459
 460     if (err) {
 461         error_propagate(errp, err);
 462         goto err;
 463     }
 464
 465     addr = sraddr->host;
 466     port = sraddr->port;
 467     if (addr == NULL || strlen(addr) == 0) {
 468         addr = "localhost";
 469     }
 470     if (port == NULL || strlen(port) == 0) {
 471         error_setg(errp, "remote port not specified");
 472         goto err;
 473     }
 474
 475     if (0 != (rc = getaddrinfo(addr, port, &ai, &peer))) {
 476         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 477                    gai_strerror(rc));
 478         goto err;
 479     }
 480
 481     /* lookup local addr */
 482     memset(&ai,0, sizeof(ai));
 483     ai.ai_flags = AI_PASSIVE;
 484     ai.ai_family = peer->ai_family;
 485     ai.ai_socktype = SOCK_DGRAM;
 486
 487     if (sladdr) {
 488         addr = sladdr->host;
 489         port = sladdr->port;
 490         if (addr == NULL || strlen(addr) == 0) {
 491             addr = NULL;
 492         }
 493         if (!port || strlen(port) == 0) {
 494             port = "0";
 495         }
 496     } else {
 497         addr = NULL;
 498         port = "0";
 499     }
 500
 501     if (0 != (rc = getaddrinfo(addr, port, &ai, &local))) {
 502         error_setg(errp, "address resolution failed for %s:%s: %s", addr, port,
 503                    gai_strerror(rc));
 504         goto err;
 505     }
 506
 507     /* create socket */
 508     sock = qemu_socket(peer->ai_family, peer->ai_socktype, peer->ai_protocol);
 509     if (sock < 0) {
 510         error_setg_errno(errp, errno, "Failed to create socket");
 511         goto err;
 512     }
 513     socket_set_fast_reuse(sock);
 514
 515     /* bind socket */
 516     if (bind(sock, local->ai_addr, local->ai_addrlen) < 0) {
 517         error_setg_errno(errp, errno, "Failed to bind socket");
 518         goto err;
 519     }
 520
 521     /* connect to peer */
 522     if (connect(sock,peer->ai_addr,peer->ai_addrlen) < 0) {
 523         error_setg_errno(errp, errno, "Failed to connect socket");
 524         goto err;
 525     }
 526
 527     freeaddrinfo(local);
 528     freeaddrinfo(peer);
 529     return sock;
 530
 531 err:
 532     if (-1 != sock)
 533         closesocket(sock);
 534     if (local)
 535         freeaddrinfo(local);
 536     if (peer)
 537         freeaddrinfo(peer);
 538     return -1;
 539 }
 540
 541 /* compatibility wrapper */
 542 InetSocketAddress *inet_parse(const char *str, Error **errp)
 543 {
 544     InetSocketAddress *addr;
 545     const char *optstr, *h;
 546     char host[65];
 547     char port[33];
 548     int to;
 549     int pos;
 550
 551     addr = g_new0(InetSocketAddress, 1);
 552
 553     /* parse address */
 554     if (str[0] == ':') {
 555         /* no host given */
 556         host[0] = '\0';
 557         if (1 != sscanf(str, ":%32[^,]%n", port, &pos)) {
 558             error_setg(errp, "error parsing port in address '%s'", str);
 559             goto fail;
 560         }
 561     } else if (str[0] == '[') {
 562         /* IPv6 addr */
 563         if (2 != sscanf(str, "[%64[^]]]:%32[^,]%n", host, port, &pos)) {
 564             error_setg(errp, "error parsing IPv6 address '%s'", str);
 565             goto fail;
 566         }
 567         addr->ipv6 = addr->has_ipv6 = true;
 568     } else {
 569         /* hostname or IPv4 addr */
 570         if (2 != sscanf(str, "%64[^:]:%32[^,]%n", host, port, &pos)) {
 571             error_setg(errp, "error parsing address '%s'", str);
 572             goto fail;
 573         }
 574         if (host[strspn(host, "0123456789.")] == '\0') {
 575             addr->ipv4 = addr->has_ipv4 = true;
 576         }
 577     }
 578
 579     addr->host = g_strdup(host);
 580     addr->port = g_strdup(port);
 581
 582     /* parse options */
 583     optstr = str + pos;
 584     h = strstr(optstr, ",to=");
 585     if (h) {
 586         h += 4;
 587         if (sscanf(h, "%d%n", &to, &pos) != 1 ||
 588             (h[pos] != '\0' && h[pos] != ',')) {
 589             error_setg(errp, "error parsing to= argument");
 590             goto fail;
 591         }
 592         addr->has_to = true;
 593         addr->to = to;
 594     }
 595     if (strstr(optstr, ",ipv4")) {
 596         addr->ipv4 = addr->has_ipv4 = true;
 597     }
 598     if (strstr(optstr, ",ipv6")) {
 599         addr->ipv6 = addr->has_ipv6 = true;
 600     }
 601     return addr;
 602
 603 fail:
 604     qapi_free_InetSocketAddress(addr);
 605     return NULL;
 606 }
 607
 608 int inet_listen(const char *str, char *ostr, int olen,
 609                 int socktype, int port_offset, Error **errp)
 610 {
 611     char *optstr;
 612     int sock = -1;
 613     InetSocketAddress *addr;
 614
 615     addr = inet_parse(str, errp);
 616     if (addr != NULL) {
 617         sock = inet_listen_saddr(addr, port_offset, true, errp);
 618         if (sock != -1 && ostr) {
 619             optstr = strchr(str, ',');
 620             if (addr->ipv6) {
 621                 snprintf(ostr, olen, "[%s]:%s%s",
 622                          addr->host,
 623                          addr->port,
 624                          optstr ? optstr : "");
 625             } else {
 626                 snprintf(ostr, olen, "%s:%s%s",
 627                          addr->host,
 628                          addr->port,
 629                          optstr ? optstr : "");
 630             }
 631         }
 632         qapi_free_InetSocketAddress(addr);
 633     }
 634     return sock;
 635 }
 636
 637 /**
 638  * Create a blocking socket and connect it to an address.
 639  *
 640  * @str: address string
 641  * @errp: set in case of an error
 642  *
 643  * Returns -1 in case of error, file descriptor on success
 644  **/
 645 int inet_connect(const char *str, Error **errp)
 646 {
 647     int sock = -1;
 648     InetSocketAddress *addr;
 649
 650     addr = inet_parse(str, errp);
 651     if (addr != NULL) {
 652         sock = inet_connect_saddr(addr, errp, NULL, NULL);
 653         qapi_free_InetSocketAddress(addr);
 654     }
 655     return sock;
 656 }
 657
 658 /**
 659  * Create a non-blocking socket and connect it to an address.
 660  * Calls the callback function with fd in case of success or -1 in case of
 661  * error.
 662  *
 663  * @str: address string
 664  * @callback: callback function that is called when connect completes,
 665  *            cannot be NULL.
 666  * @opaque: opaque for callback function
 667  * @errp: set in case of an error
 668  *
 669  * Returns: -1 on immediate error, file descriptor on success.
 670  **/
 671 int inet_nonblocking_connect(const char *str,
 672                              NonBlockingConnectHandler *callback,
 673                              void *opaque, Error **errp)
 674 {
 675     int sock = -1;
 676     InetSocketAddress *addr;
 677
 678     g_assert(callback != NULL);
 679
 680     addr = inet_parse(str, errp);
 681     if (addr != NULL) {
 682         sock = inet_connect_saddr(addr, errp, callback, opaque);
 683         qapi_free_InetSocketAddress(addr);
 684     }
 685     return sock;
 686 }
 687
 688 #ifndef _WIN32
 689
 690 static int unix_listen_saddr(UnixSocketAddress *saddr,
 691                              bool update_addr,
 692                              Error **errp)
 693 {
 694     struct sockaddr_un un;
 695     int sock, fd;
 696
 697     sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
 698     if (sock < 0) {
 699         error_setg_errno(errp, errno, "Failed to create Unix socket");
 700         return -1;
 701     }
 702
 703     memset(&un, 0, sizeof(un));
 704     un.sun_family = AF_UNIX;
 705     if (saddr->path && strlen(saddr->path)) {
 706         snprintf(un.sun_path, sizeof(un.sun_path), "%s", saddr->path);
 707     } else {
 708         const char *tmpdir = getenv("TMPDIR");
 709         tmpdir = tmpdir ? tmpdir : "/tmp";
 710         if (snprintf(un.sun_path, sizeof(un.sun_path), "%s/qemu-socket-XXXXXX",
 711                      tmpdir) >= sizeof(un.sun_path)) {
 712             error_setg_errno(errp, errno,
 713                              "TMPDIR environment variable (%s) too large", tmpdir);
 714             goto err;
 715         }
 716
 717         /*
 718          * This dummy fd usage silences the mktemp() unsecure warning.
 719          * Using mkstemp() doesn't make things more secure here
 720          * though.  bind() complains about existing files, so we have
 721          * to unlink first and thus re-open the race window.  The
 722          * worst case possible is bind() failing, i.e. a DoS attack.
 723          */
 724         fd = mkstemp(un.sun_path);
 725         if (fd < 0) {
 726             error_setg_errno(errp, errno,
 727                              "Failed to make a temporary socket name in %s", tmpdir);
 728             goto err;
 729         }
 730         close(fd);
 731         if (update_addr) {
 732             g_free(saddr->path);
 733             saddr->path = g_strdup(un.sun_path);
 734         }
 735     }
 736
 737     if (unlink(un.sun_path) < 0 && errno != ENOENT) {
 738         error_setg_errno(errp, errno,
 739                          "Failed to unlink socket %s", un.sun_path);
 740         goto err;
 741     }
 742     if (bind(sock, (struct sockaddr*) &un, sizeof(un)) < 0) {
 743         error_setg_errno(errp, errno, "Failed to bind socket to %s", un.sun_path);
 744         goto err;
 745     }
 746     if (listen(sock, 1) < 0) {
 747         error_setg_errno(errp, errno, "Failed to listen on socket");
 748         goto err;
 749     }
 750
 751     return sock;
 752
 753 err:
 754     closesocket(sock);
 755     return -1;
 756 }
 757
 758 static int unix_connect_saddr(UnixSocketAddress *saddr, Error **errp,
 759                               NonBlockingConnectHandler *callback, void *opaque)
 760 {
 761     struct sockaddr_un un;
 762     ConnectState *connect_state = NULL;
 763     int sock, rc;
 764
 765     if (saddr->path == NULL) {
 766         error_setg(errp, "unix connect: no path specified");
 767         return -1;
 768     }
 769
 770     sock = qemu_socket(PF_UNIX, SOCK_STREAM, 0);
 771     if (sock < 0) {
 772         error_setg_errno(errp, errno, "Failed to create socket");
 773         return -1;
 774     }
 775     if (callback != NULL) {
 776         connect_state = g_malloc0(sizeof(*connect_state));
 777         connect_state->callback = callback;
 778         connect_state->opaque = opaque;
 779         qemu_set_nonblock(sock);
 780     }
 781
 782     memset(&un, 0, sizeof(un));
 783     un.sun_family = AF_UNIX;
 784     snprintf(un.sun_path, sizeof(un.sun_path), "%s", saddr->path);
 785
 786     /* connect to peer */
 787     do {
 788         rc = 0;
 789         if (connect(sock, (struct sockaddr *) &un, sizeof(un)) < 0) {
 790             rc = -socket_error();
 791         }
 792     } while (rc == -EINTR);
 793
 794     if (connect_state != NULL && QEMU_SOCKET_RC_INPROGRESS(rc)) {
 795         connect_state->fd = sock;
 796         qemu_set_fd_handler(sock, NULL, wait_for_connect, connect_state);
 797         return sock;
 798     } else if (rc >= 0) {
 799         /* non blocking socket immediate success, call callback */
 800         if (callback != NULL) {
 801             callback(sock, NULL, opaque);
 802         }
 803     }
 804
 805     if (rc < 0) {
 806         error_setg_errno(errp, -rc, "Failed to connect socket");
 807         close(sock);
 808         sock = -1;
 809     }
 810
 811     g_free(connect_state);
 812     return sock;
 813 }
 814
 815 #else
 816
 817 static int unix_listen_saddr(UnixSocketAddress *saddr,
 818                              bool update_addr,
 819                              Error **errp)
 820 {
 821     error_setg(errp, "unix sockets are not available on windows");
 822     errno = ENOTSUP;
 823     return -1;
 824 }
 825
 826 static int unix_connect_saddr(UnixSocketAddress *saddr, Error **errp,
 827                               NonBlockingConnectHandler *callback, void *opaque)
 828 {
 829     error_setg(errp, "unix sockets are not available on windows");
 830     errno = ENOTSUP;
 831     return -1;
 832 }
 833 #endif
 834
 835 /* compatibility wrapper */
 836 int unix_listen(const char *str, char *ostr, int olen, Error **errp)
 837 {
 838     char *path, *optstr;
 839     int sock, len;
 840     UnixSocketAddress *saddr;
 841
 842     saddr = g_new0(UnixSocketAddress, 1);
 843
 844     optstr = strchr(str, ',');
 845     if (optstr) {
 846         len = optstr - str;
 847         if (len) {
 848             path = g_malloc(len+1);
 849             snprintf(path, len+1, "%.*s", len, str);
 850             saddr->path = path;
 851         }
 852     } else {
 853         saddr->path = g_strdup(str);
 854     }
 855
 856     sock = unix_listen_saddr(saddr, true, errp);
 857
 858     if (sock != -1 && ostr)
 859         snprintf(ostr, olen, "%s%s", saddr->path, optstr ? optstr : "");
 860     qapi_free_UnixSocketAddress(saddr);
 861     return sock;
 862 }
 863
 864 int unix_connect(const char *path, Error **errp)
 865 {
 866     UnixSocketAddress *saddr;
 867     int sock;
 868
 869     saddr = g_new0(UnixSocketAddress, 1);
 870     saddr->path = g_strdup(path);
 871     sock = unix_connect_saddr(saddr, errp, NULL, NULL);
 872     qapi_free_UnixSocketAddress(saddr);
 873     return sock;
 874 }
 875
 876
 877 int unix_nonblocking_connect(const char *path,
 878                              NonBlockingConnectHandler *callback,
 879                              void *opaque, Error **errp)
 880 {
 881     UnixSocketAddress *saddr;
 882     int sock = -1;
 883
 884     g_assert(callback != NULL);
 885
 886     saddr = g_new0(UnixSocketAddress, 1);
 887     saddr->path = g_strdup(path);
 888     sock = unix_connect_saddr(saddr, errp, callback, opaque);
 889     qapi_free_UnixSocketAddress(saddr);
 890     return sock;
 891 }
 892
 893 SocketAddress *socket_parse(const char *str, Error **errp)
 894 {
 895     SocketAddress *addr;
 896
 897     addr = g_new0(SocketAddress, 1);
 898     if (strstart(str, "unix:", NULL)) {
 899         if (str[5] == '\0') {
 900             error_setg(errp, "invalid Unix socket address");
 901             goto fail;
 902         } else {
 903             addr->type = SOCKET_ADDRESS_KIND_UNIX;
 904             addr->u.q_unix = g_new(UnixSocketAddress, 1);
 905             addr->u.q_unix->path = g_strdup(str + 5);
 906         }
 907     } else if (strstart(str, "fd:", NULL)) {
 908         if (str[3] == '\0') {
 909             error_setg(errp, "invalid file descriptor address");
 910             goto fail;
 911         } else {
 912             addr->type = SOCKET_ADDRESS_KIND_FD;
 913             addr->u.fd = g_new(String, 1);
 914             addr->u.fd->str = g_strdup(str + 3);
 915         }
 916     } else {
 917         addr->type = SOCKET_ADDRESS_KIND_INET;
 918         addr->u.inet = inet_parse(str, errp);
 919         if (addr->u.inet == NULL) {
 920             goto fail;
 921         }
 922     }
 923     return addr;
 924
 925 fail:
 926     qapi_free_SocketAddress(addr);
 927     return NULL;
 928 }
 929
 930 int socket_connect(SocketAddress *addr, Error **errp,
 931                    NonBlockingConnectHandler *callback, void *opaque)
 932 {
 933     int fd;
 934
 935     switch (addr->type) {
 936     case SOCKET_ADDRESS_KIND_INET:
 937         fd = inet_connect_saddr(addr->u.inet, errp, callback, opaque);
 938         break;
 939
 940     case SOCKET_ADDRESS_KIND_UNIX:
 941         fd = unix_connect_saddr(addr->u.q_unix, errp, callback, opaque);
 942         break;
 943
 944     case SOCKET_ADDRESS_KIND_FD:
 945         fd = monitor_get_fd(cur_mon, addr->u.fd->str, errp);
 946         if (fd >= 0 && callback) {
 947             qemu_set_nonblock(fd);
 948             callback(fd, NULL, opaque);
 949         }
 950         break;
 951
 952     default:
 953         abort();
 954     }
 955     return fd;
 956 }
 957
 958 int socket_listen(SocketAddress *addr, Error **errp)
 959 {
 960     int fd;
 961
 962     switch (addr->type) {
 963     case SOCKET_ADDRESS_KIND_INET:
 964         fd = inet_listen_saddr(addr->u.inet, 0, false, errp);
 965         break;
 966
 967     case SOCKET_ADDRESS_KIND_UNIX:
 968         fd = unix_listen_saddr(addr->u.q_unix, false, errp);
 969         break;
 970
 971     case SOCKET_ADDRESS_KIND_FD:
 972         fd = monitor_get_fd(cur_mon, addr->u.fd->str, errp);
 973         break;
 974
 975     default:
 976         abort();
 977     }
 978     return fd;
 979 }
 980
 981 int socket_dgram(SocketAddress *remote, SocketAddress *local, Error **errp)
 982 {
 983     int fd;
 984
 985     switch (remote->type) {
 986     case SOCKET_ADDRESS_KIND_INET:
 987         fd = inet_dgram_saddr(remote->u.inet, local ? local->u.inet : NULL, errp);
 988         break;
 989
 990     default:
 991         error_setg(errp, "socket type unsupported for datagram");
 992         fd = -1;
 993     }
 994     return fd;
 995 }
 996
 997
 998 static SocketAddress *
 999 socket_sockaddr_to_address_inet(struct sockaddr_storage *sa,
1000                                 socklen_t salen,
1001                                 Error **errp)
1002 {
1003     char host[NI_MAXHOST];
1004     char serv[NI_MAXSERV];
1005     SocketAddress *addr;
1006     InetSocketAddress *inet;
1007     int ret;
1008
1009     ret = getnameinfo((struct sockaddr *)sa, salen,
1010                       host, sizeof(host),
1011                       serv, sizeof(serv),
1012                       NI_NUMERICHOST | NI_NUMERICSERV);
1013     if (ret != 0) {
1014         error_setg(errp, "Cannot format numeric socket address: %s",
1015                    gai_strerror(ret));
1016         return NULL;
1017     }
1018
1019     addr = g_new0(SocketAddress, 1);
1020     addr->type = SOCKET_ADDRESS_KIND_INET;
1021     inet = addr->u.inet = g_new0(InetSocketAddress, 1);
1022     inet->host = g_strdup(host);
1023     inet->port = g_strdup(serv);
1024     if (sa->ss_family == AF_INET) {
1025         inet->has_ipv4 = inet->ipv4 = true;
1026     } else {
1027         inet->has_ipv6 = inet->ipv6 = true;
1028     }
1029
1030     return addr;
1031 }
1032
1033
1034 #ifndef WIN32
1035 static SocketAddress *
1036 socket_sockaddr_to_address_unix(struct sockaddr_storage *sa,
1037                                 socklen_t salen,
1038                                 Error **errp)
1039 {
1040     SocketAddress *addr;
1041     struct sockaddr_un *su = (struct sockaddr_un *)sa;
1042
1043     addr = g_new0(SocketAddress, 1);
1044     addr->type = SOCKET_ADDRESS_KIND_UNIX;
1045     addr->u.q_unix = g_new0(UnixSocketAddress, 1);
1046     if (su->sun_path[0]) {
1047         addr->u.q_unix->path = g_strndup(su->sun_path,
1048                                          sizeof(su->sun_path));
1049     }
1050
1051     return addr;
1052 }
1053 #endif /* WIN32 */
1054
1055 SocketAddress *
1056 socket_sockaddr_to_address(struct sockaddr_storage *sa,
1057                            socklen_t salen,
1058                            Error **errp)
1059 {
1060     switch (sa->ss_family) {
1061     case AF_INET:
1062     case AF_INET6:
1063         return socket_sockaddr_to_address_inet(sa, salen, errp);
1064
1065 #ifndef WIN32
1066     case AF_UNIX:
1067         return socket_sockaddr_to_address_unix(sa, salen, errp);
1068 #endif /* WIN32 */
1069
1070     default:
1071         error_setg(errp, "socket family %d unsupported",
1072                    sa->ss_family);
1073         return NULL;
1074     }
1075     return 0;
1076 }
1077
1078
1079 SocketAddress *socket_local_address(int fd, Error **errp)
1080 {
1081     struct sockaddr_storage ss;
1082     socklen_t sslen = sizeof(ss);
1083
1084     if (getsockname(fd, (struct sockaddr *)&ss, &sslen) < 0) {
1085         error_setg_errno(errp, socket_error(), "%s",
1086                          "Unable to query local socket address");
1087         return NULL;
1088     }
1089
1090     return socket_sockaddr_to_address(&ss, sslen, errp);
1091 }
1092
1093
1094 SocketAddress *socket_remote_address(int fd, Error **errp)
1095 {
1096     struct sockaddr_storage ss;
1097     socklen_t sslen = sizeof(ss);
1098
1099     if (getpeername(fd, (struct sockaddr *)&ss, &sslen) < 0) {
1100         error_setg_errno(errp, socket_error(), "%s",
1101                          "Unable to query remote socket address");
1102         return NULL;
1103     }
1104
1105     return socket_sockaddr_to_address(&ss, sslen, errp);
1106 }
1107
1108
1109 void qapi_copy_SocketAddress(SocketAddress **p_dest,
1110                              SocketAddress *src)
1111 {
1112     QmpOutputVisitor *qov;
1113     QmpInputVisitor *qiv;
1114     Visitor *ov, *iv;
1115     QObject *obj;
1116
1117     *p_dest = NULL;
1118
1119     qov = qmp_output_visitor_new();
1120     ov = qmp_output_get_visitor(qov);
1121     visit_type_SocketAddress(ov, NULL, &src, &error_abort);
1122     obj = qmp_output_get_qobject(qov);
1123     qmp_output_visitor_cleanup(qov);
1124     if (!obj) {
1125         return;
1126     }
1127
1128     qiv = qmp_input_visitor_new(obj);
1129     iv = qmp_input_get_visitor(qiv);
1130     visit_type_SocketAddress(iv, NULL, p_dest, &error_abort);
1131     qmp_input_visitor_cleanup(qiv);
1132     qobject_decref(obj);
1133 }