socket-helpers.c (3965B)
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/sockets.h" 23 #include "socket-helpers.h" 24 25 #ifndef AI_ADDRCONFIG 26 # define AI_ADDRCONFIG 0 27 #endif 28 #ifndef EAI_ADDRFAMILY 29 # define EAI_ADDRFAMILY 0 30 #endif 31 32 /* 33 * @hostname: a DNS name or numeric IP address 34 * 35 * Check whether it is possible to bind & connect to ports 36 * on the DNS name or IP address @hostname. If an IP address 37 * is used, it must not be a wildcard address. 38 * 39 * Returns 0 on success, -1 on error with errno set 40 */ 41 static int socket_can_bind_connect(const char *hostname, int family) 42 { 43 int lfd = -1, cfd = -1, afd = -1; 44 struct addrinfo ai, *res = NULL; 45 struct sockaddr_storage ss; 46 socklen_t sslen = sizeof(ss); 47 int soerr; 48 socklen_t soerrlen = sizeof(soerr); 49 bool check_soerr = false; 50 int rc; 51 int ret = -1; 52 53 memset(&ai, 0, sizeof(ai)); 54 ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG; 55 ai.ai_family = family; 56 ai.ai_socktype = SOCK_STREAM; 57 58 /* lookup */ 59 rc = getaddrinfo(hostname, NULL, &ai, &res); 60 if (rc != 0) { 61 if (rc == EAI_ADDRFAMILY || rc == EAI_FAMILY || rc == EAI_NONAME) { 62 errno = EADDRNOTAVAIL; 63 } else { 64 errno = EINVAL; 65 } 66 goto cleanup; 67 } 68 69 lfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); 70 if (lfd < 0) { 71 goto cleanup; 72 } 73 74 cfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); 75 if (cfd < 0) { 76 goto cleanup; 77 } 78 79 if (bind(lfd, res->ai_addr, res->ai_addrlen) < 0) { 80 goto cleanup; 81 } 82 83 if (listen(lfd, 1) < 0) { 84 goto cleanup; 85 } 86 87 if (getsockname(lfd, (struct sockaddr *)&ss, &sslen) < 0) { 88 goto cleanup; 89 } 90 91 qemu_socket_set_nonblock(cfd); 92 if (connect(cfd, (struct sockaddr *)&ss, sslen) < 0) { 93 if (errno == EINPROGRESS) { 94 check_soerr = true; 95 } else { 96 goto cleanup; 97 } 98 } 99 100 sslen = sizeof(ss); 101 afd = accept(lfd, (struct sockaddr *)&ss, &sslen); 102 if (afd < 0) { 103 goto cleanup; 104 } 105 106 if (check_soerr) { 107 if (getsockopt(cfd, SOL_SOCKET, SO_ERROR, &soerr, &soerrlen) < 0) { 108 goto cleanup; 109 } 110 if (soerr) { 111 errno = soerr; 112 goto cleanup; 113 } 114 } 115 116 ret = 0; 117 118 cleanup: 119 if (afd != -1) { 120 close(afd); 121 } 122 if (cfd != -1) { 123 close(cfd); 124 } 125 if (lfd != -1) { 126 close(lfd); 127 } 128 if (res) { 129 freeaddrinfo(res); 130 } 131 return ret; 132 } 133 134 135 int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6) 136 { 137 *has_ipv4 = *has_ipv6 = false; 138 139 if (socket_can_bind_connect("127.0.0.1", PF_INET) < 0) { 140 if (errno != EADDRNOTAVAIL) { 141 return -1; 142 } 143 } else { 144 *has_ipv4 = true; 145 } 146 147 if (socket_can_bind_connect("::1", PF_INET6) < 0) { 148 if (errno != EADDRNOTAVAIL) { 149 return -1; 150 } 151 } else { 152 *has_ipv6 = true; 153 } 154 155 return 0; 156 } 157 158 void socket_check_afunix_support(bool *has_afunix) 159 { 160 int fd; 161 162 fd = socket(PF_UNIX, SOCK_STREAM, 0); 163 closesocket(fd); 164 165 #ifdef _WIN32 166 *has_afunix = (fd != (int)INVALID_SOCKET); 167 #else 168 *has_afunix = (fd >= 0); 169 #endif 170 171 return; 172 }