LXR qemu/tests/unit/socket-helpers.c

   1/*
   2 * Helper functions for tests using sockets
   3 *
   4 * Copyright 2015-2018 Red Hat, Inc.
   5 *
   6 * This program is free software; you can redistribute it and/or
   7 * modify it under the terms of the GNU General Public License as
   8 * published by the Free Software Foundation; either version 2 or
   9 * (at your option) version 3 of the License.
  10 *
  11 * This program is distributed in the hope that it will be useful,
  12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 * GNU General Public License for more details.
  15 *
  16 * You should have received a copy of the GNU General Public License
  17 * along with this program; if not, see <http://www.gnu.org/licenses/>.
  18 *
  19 */
  20
  21#include "qemu/osdep.h"
  22#include "qemu-common.h"
  23#include "qemu/sockets.h"
  24#include "socket-helpers.h"
  25
  26#ifndef AI_ADDRCONFIG
  27# define AI_ADDRCONFIG 0
  28#endif
  29#ifndef EAI_ADDRFAMILY
  30# define EAI_ADDRFAMILY 0
  31#endif
  32
  33/*
  34 * @hostname: a DNS name or numeric IP address
  35 *
  36 * Check whether it is possible to bind & connect to ports
  37 * on the DNS name or IP address @hostname. If an IP address
  38 * is used, it must not be a wildcard address.
  39 *
  40 * Returns 0 on success, -1 on error with errno set
  41 */
  42static int socket_can_bind_connect(const char *hostname, int family)
  43{
  44    int lfd = -1, cfd = -1, afd = -1;
  45    struct addrinfo ai, *res = NULL;
  46    struct sockaddr_storage ss;
  47    socklen_t sslen = sizeof(ss);
  48    int soerr;
  49    socklen_t soerrlen = sizeof(soerr);
  50    bool check_soerr = false;
  51    int rc;
  52    int ret = -1;
  53
  54    memset(&ai, 0, sizeof(ai));
  55    ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
  56    ai.ai_family = family;
  57    ai.ai_socktype = SOCK_STREAM;
  58
  59    /* lookup */
  60    rc = getaddrinfo(hostname, NULL, &ai, &res);
  61    if (rc != 0) {
  62        if (rc == EAI_ADDRFAMILY || rc == EAI_FAMILY || rc == EAI_NONAME) {
  63            errno = EADDRNOTAVAIL;
  64        } else {
  65            errno = EINVAL;
  66        }
  67        goto cleanup;
  68    }
  69
  70    lfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
  71    if (lfd < 0) {
  72        goto cleanup;
  73    }
  74
  75    cfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
  76    if (cfd < 0) {
  77        goto cleanup;
  78    }
  79
  80    if (bind(lfd, res->ai_addr, res->ai_addrlen) < 0) {
  81        goto cleanup;
  82    }
  83
  84    if (listen(lfd, 1) < 0) {
  85        goto cleanup;
  86    }
  87
  88    if (getsockname(lfd, (struct sockaddr *)&ss, &sslen) < 0) {
  89        goto cleanup;
  90    }
  91
  92    qemu_set_nonblock(cfd);
  93    if (connect(cfd, (struct sockaddr *)&ss, sslen) < 0) {
  94        if (errno == EINPROGRESS) {
  95            check_soerr = true;
  96        } else {
  97            goto cleanup;
  98        }
  99    }
 100
 101    sslen = sizeof(ss);
 102    afd = accept(lfd,  (struct sockaddr *)&ss, &sslen);
 103    if (afd < 0) {
 104        goto cleanup;
 105    }
 106
 107    if (check_soerr) {
 108        if (qemu_getsockopt(cfd, SOL_SOCKET, SO_ERROR, &soerr, &soerrlen) < 0) {
 109            goto cleanup;
 110        }
 111        if (soerr) {
 112            errno = soerr;
 113            goto cleanup;
 114        }
 115    }
 116
 117    ret = 0;
 118
 119 cleanup:
 120    if (afd != -1) {
 121        close(afd);
 122    }
 123    if (cfd != -1) {
 124        close(cfd);
 125    }
 126    if (lfd != -1) {
 127        close(lfd);
 128    }
 129    if (res) {
 130        freeaddrinfo(res);
 131    }
 132    return ret;
 133}
 134
 135
 136int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6)
 137{
 138    *has_ipv4 = *has_ipv6 = false;
 139
 140    if (socket_can_bind_connect("127.0.0.1", PF_INET) < 0) {
 141        if (errno != EADDRNOTAVAIL) {
 142            return -1;
 143        }
 144    } else {
 145        *has_ipv4 = true;
 146    }
 147
 148    if (socket_can_bind_connect("::1", PF_INET6) < 0) {
 149        if (errno != EADDRNOTAVAIL) {
 150            return -1;
 151        }
 152    } else {
 153        *has_ipv6 = true;
 154    }
 155
 156    return 0;
 157}
 158