tests/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 int socket_can_bind(const char *hostname)
  34 {
  35     int fd = -1;
  36     struct addrinfo ai, *res = NULL;
  37     int rc;
  38     int ret = -1;
  39
  40     memset(&ai, 0, sizeof(ai));
  41     ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG;
  42     ai.ai_family = AF_UNSPEC;
  43     ai.ai_socktype = SOCK_STREAM;
  44
  45     /* lookup */
  46     rc = getaddrinfo(hostname, NULL, &ai, &res);
  47     if (rc != 0) {
  48         if (rc == EAI_ADDRFAMILY ||
  49             rc == EAI_FAMILY) {
  50             errno = EADDRNOTAVAIL;
  51         } else {
  52             errno = EINVAL;
  53         }
  54         goto cleanup;
  55     }
  56
  57     fd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
  58     if (fd < 0) {
  59         goto cleanup;
  60     }
  61
  62     if (bind(fd, res->ai_addr, res->ai_addrlen) < 0) {
  63         goto cleanup;
  64     }
  65
  66     ret = 0;
  67
  68  cleanup:
  69     if (fd != -1) {
  70         close(fd);
  71     }
  72     if (res) {
  73         freeaddrinfo(res);
  74     }
  75     return ret;
  76 }
  77
  78
  79 int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6)
  80 {
  81     *has_ipv4 = *has_ipv6 = false;
  82
  83     if (socket_can_bind("127.0.0.1") < 0) {
  84         if (errno != EADDRNOTAVAIL) {
  85             return -1;
  86         }
  87     } else {
  88         *has_ipv4 = true;
  89     }
  90
  91     if (socket_can_bind("::1") < 0) {
  92         if (errno != EADDRNOTAVAIL) {
  93             return -1;
  94         }
  95     } else {
  96         *has_ipv6 = true;
  97     }
  98
  99     return 0;
 100 }