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