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 "qapi/error.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_blocking(cfd, false, &error_abort); 93 94 if (connect(cfd, (struct sockaddr *)&ss, sslen) < 0) { 95 if (errno == EINPROGRESS) { 96 check_soerr = true; 97 } else { 98 goto cleanup; 99 } 100 } 101 102 sslen = sizeof(ss); 103 afd = accept(lfd, (struct sockaddr *)&ss, &sslen); 104 if (afd < 0) { 105 goto cleanup; 106 } 107 108 if (check_soerr) { 109 if (getsockopt(cfd, SOL_SOCKET, SO_ERROR, &soerr, &soerrlen) < 0) { 110 goto cleanup; 111 } 112 if (soerr) { 113 errno = soerr; 114 goto cleanup; 115 } 116 } 117 118 ret = 0; 119 120 cleanup: 121 if (afd != -1) { 122 close(afd); 123 } 124 if (cfd != -1) { 125 close(cfd); 126 } 127 if (lfd != -1) { 128 close(lfd); 129 } 130 if (res) { 131 freeaddrinfo(res); 132 } 133 return ret; 134 } 135 136 137 int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6) 138 { 139 *has_ipv4 = *has_ipv6 = false; 140 141 if (socket_can_bind_connect("127.0.0.1", PF_INET) < 0) { 142 if (errno != EADDRNOTAVAIL) { 143 return -1; 144 } 145 } else { 146 *has_ipv4 = true; 147 } 148 149 if (socket_can_bind_connect("::1", PF_INET6) < 0) { 150 if (errno != EADDRNOTAVAIL) { 151 return -1; 152 } 153 } else { 154 *has_ipv6 = true; 155 } 156 157 return 0; 158 } 159 160 void socket_check_afunix_support(bool *has_afunix) 161 { 162 int fd; 163 164 fd = socket(PF_UNIX, SOCK_STREAM, 0); 165 166 #ifdef _WIN32 167 *has_afunix = (fd != (int)INVALID_SOCKET); 168 #else 169 *has_afunix = (fd >= 0); 170 #endif 171 172 if (*has_afunix) { 173 close(fd); 174 } 175 } 176