]> git.proxmox.com Git - mirror_qemu.git/blob - tests/socket-helpers.c
qemu-option: restrict qemu_opts_set to merge-lists QemuOpts
[mirror_qemu.git] / 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 /*
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 (qemu_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 }