]>
Commit | Line | Data |
---|---|---|
9b589ffb DB |
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" | |
9b589ffb DB |
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 | ||
71714178 MAL |
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) | |
9b589ffb | 42 | { |
abd983c0 | 43 | int lfd = -1, cfd = -1, afd = -1; |
9b589ffb | 44 | struct addrinfo ai, *res = NULL; |
abd983c0 DB |
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; | |
9b589ffb DB |
50 | int rc; |
51 | int ret = -1; | |
52 | ||
53 | memset(&ai, 0, sizeof(ai)); | |
54 | ai.ai_flags = AI_CANONNAME | AI_ADDRCONFIG; | |
71714178 | 55 | ai.ai_family = family; |
9b589ffb DB |
56 | ai.ai_socktype = SOCK_STREAM; |
57 | ||
58 | /* lookup */ | |
59 | rc = getaddrinfo(hostname, NULL, &ai, &res); | |
60 | if (rc != 0) { | |
b822c05b | 61 | if (rc == EAI_ADDRFAMILY || rc == EAI_FAMILY || rc == EAI_NONAME) { |
9b589ffb DB |
62 | errno = EADDRNOTAVAIL; |
63 | } else { | |
64 | errno = EINVAL; | |
65 | } | |
66 | goto cleanup; | |
67 | } | |
68 | ||
abd983c0 DB |
69 | lfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); |
70 | if (lfd < 0) { | |
9b589ffb DB |
71 | goto cleanup; |
72 | } | |
73 | ||
abd983c0 DB |
74 | cfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol); |
75 | if (cfd < 0) { | |
9b589ffb DB |
76 | goto cleanup; |
77 | } | |
78 | ||
abd983c0 DB |
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 | ||
ff5927ba | 91 | qemu_socket_set_nonblock(cfd); |
abd983c0 DB |
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) { | |
e7b79428 | 107 | if (getsockopt(cfd, SOL_SOCKET, SO_ERROR, &soerr, &soerrlen) < 0) { |
abd983c0 DB |
108 | goto cleanup; |
109 | } | |
110 | if (soerr) { | |
111 | errno = soerr; | |
112 | goto cleanup; | |
113 | } | |
114 | } | |
115 | ||
9b589ffb DB |
116 | ret = 0; |
117 | ||
118 | cleanup: | |
abd983c0 DB |
119 | if (afd != -1) { |
120 | close(afd); | |
121 | } | |
122 | if (cfd != -1) { | |
123 | close(cfd); | |
124 | } | |
125 | if (lfd != -1) { | |
126 | close(lfd); | |
9b589ffb DB |
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 | ||
71714178 | 139 | if (socket_can_bind_connect("127.0.0.1", PF_INET) < 0) { |
9b589ffb DB |
140 | if (errno != EADDRNOTAVAIL) { |
141 | return -1; | |
142 | } | |
143 | } else { | |
144 | *has_ipv4 = true; | |
145 | } | |
146 | ||
71714178 | 147 | if (socket_can_bind_connect("::1", PF_INET6) < 0) { |
9b589ffb DB |
148 | if (errno != EADDRNOTAVAIL) { |
149 | return -1; | |
150 | } | |
151 | } else { | |
152 | *has_ipv6 = true; | |
153 | } | |
154 | ||
155 | return 0; | |
156 | } | |
0370f239 BM |
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 | } |