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 */ 42 static 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 (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 136 int 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