target/mips: Simplify decode_opc_mxu() ifdef'ry
[qemu/ar7.git] / tests / socket-helpers.c
blobf704fd1a69cec3d5c081c5ba9876bfc3587c8a64
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  */
21 #include "qemu/osdep.h"
22 #include "qemu-common.h"
23 #include "qemu/sockets.h"
24 #include "socket-helpers.h"
26 #ifndef AI_ADDRCONFIG
27 # define AI_ADDRCONFIG 0
28 #endif
29 #ifndef EAI_ADDRFAMILY
30 # define EAI_ADDRFAMILY 0
31 #endif
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)
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;
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;
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     }
70     lfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
71     if (lfd < 0) {
72         goto cleanup;
73     }
75     cfd = qemu_socket(res->ai_family, res->ai_socktype, res->ai_protocol);
76     if (cfd < 0) {
77         goto cleanup;
78     }
80     if (bind(lfd, res->ai_addr, res->ai_addrlen) < 0) {
81         goto cleanup;
82     }
84     if (listen(lfd, 1) < 0) {
85         goto cleanup;
86     }
88     if (getsockname(lfd, (struct sockaddr *)&ss, &sslen) < 0) {
89         goto cleanup;
90     }
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     }
101     sslen = sizeof(ss);
102     afd = accept(lfd,  (struct sockaddr *)&ss, &sslen);
103     if (afd < 0) {
104         goto cleanup;
105     }
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     }
117     ret = 0;
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;
136 int socket_check_protocol_support(bool *has_ipv4, bool *has_ipv6)
138     *has_ipv4 = *has_ipv6 = false;
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     }
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     }
156     return 0;