2 * Tests for util/qemu-sockets.c
4 * Copyright 2018 Red Hat, Inc.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
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.
16 * You should have received a copy of the GNU General Public License
17 * along with this library; if not, see <http://www.gnu.org/licenses/>.
21 #include "qemu/osdep.h"
22 #include "qemu-common.h"
23 #include "qemu/sockets.h"
24 #include "qapi/error.h"
25 #include "socket-helpers.h"
26 #include "monitor/monitor.h"
28 static void test_fd_is_socket_bad(void)
30 char *tmp
= g_strdup("qemu-test-util-sockets-XXXXXX");
31 int fd
= mkstemp(tmp
);
39 g_assert(!fd_is_socket(fd
));
43 static void test_fd_is_socket_good(void)
45 int fd
= qemu_socket(PF_INET
, SOCK_STREAM
, 0);
49 g_assert(fd_is_socket(fd
));
53 static int mon_fd
= -1;
54 static const char *mon_fdname
;
55 __thread Monitor
*cur_mon
;
57 int monitor_get_fd(Monitor
*mon
, const char *fdname
, Error
**errp
)
60 g_assert(mon
== cur_mon
);
61 if (mon_fd
== -1 || !g_str_equal(mon_fdname
, fdname
)) {
62 error_setg(errp
, "No fd named %s", fdname
);
69 * Syms of stubs in libqemuutil.a are discarded at .o file
70 * granularity. To replace monitor_get_fd() and monitor_cur(), we
71 * must ensure that we also replace any other symbol that is used in
72 * the binary and would be taken from the same stub object file,
73 * otherwise we get duplicate syms at link time.
75 Monitor
*monitor_cur(void) { return cur_mon
; }
76 int monitor_vprintf(Monitor
*mon
, const char *fmt
, va_list ap
) { abort(); }
79 static void test_socket_fd_pass_name_good(void)
84 cur_mon
= g_malloc(1); /* Fake a monitor */
86 mon_fd
= qemu_socket(AF_INET
, SOCK_STREAM
, 0);
87 g_assert_cmpint(mon_fd
, >, STDERR_FILENO
);
89 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
90 addr
.u
.fd
.str
= g_strdup(mon_fdname
);
92 fd
= socket_connect(&addr
, &error_abort
);
93 g_assert_cmpint(fd
, !=, -1);
94 g_assert_cmpint(fd
, !=, mon_fd
);
97 fd
= socket_listen(&addr
, 1, &error_abort
);
98 g_assert_cmpint(fd
, !=, -1);
99 g_assert_cmpint(fd
, !=, mon_fd
);
102 g_free(addr
.u
.fd
.str
);
110 static void test_socket_fd_pass_name_bad(void)
116 cur_mon
= g_malloc(1); /* Fake a monitor */
118 mon_fd
= dup(STDOUT_FILENO
);
119 g_assert_cmpint(mon_fd
, >, STDERR_FILENO
);
121 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
122 addr
.u
.fd
.str
= g_strdup(mon_fdname
);
124 fd
= socket_connect(&addr
, &err
);
125 g_assert_cmpint(fd
, ==, -1);
126 error_free_or_abort(&err
);
128 fd
= socket_listen(&addr
, 1, &err
);
129 g_assert_cmpint(fd
, ==, -1);
130 error_free_or_abort(&err
);
132 g_free(addr
.u
.fd
.str
);
140 static void test_socket_fd_pass_name_nomon(void)
146 g_assert(cur_mon
== NULL
);
148 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
149 addr
.u
.fd
.str
= g_strdup("myfd");
151 fd
= socket_connect(&addr
, &err
);
152 g_assert_cmpint(fd
, ==, -1);
153 error_free_or_abort(&err
);
155 fd
= socket_listen(&addr
, 1, &err
);
156 g_assert_cmpint(fd
, ==, -1);
157 error_free_or_abort(&err
);
159 g_free(addr
.u
.fd
.str
);
163 static void test_socket_fd_pass_num_good(void)
168 g_assert(cur_mon
== NULL
);
169 sfd
= qemu_socket(AF_INET
, SOCK_STREAM
, 0);
170 g_assert_cmpint(sfd
, >, STDERR_FILENO
);
172 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
173 addr
.u
.fd
.str
= g_strdup_printf("%d", sfd
);
175 fd
= socket_connect(&addr
, &error_abort
);
176 g_assert_cmpint(fd
, ==, sfd
);
178 fd
= socket_listen(&addr
, 1, &error_abort
);
179 g_assert_cmpint(fd
, ==, sfd
);
181 g_free(addr
.u
.fd
.str
);
185 static void test_socket_fd_pass_num_bad(void)
191 g_assert(cur_mon
== NULL
);
192 sfd
= dup(STDOUT_FILENO
);
194 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
195 addr
.u
.fd
.str
= g_strdup_printf("%d", sfd
);
197 fd
= socket_connect(&addr
, &err
);
198 g_assert_cmpint(fd
, ==, -1);
199 error_free_or_abort(&err
);
201 fd
= socket_listen(&addr
, 1, &err
);
202 g_assert_cmpint(fd
, ==, -1);
203 error_free_or_abort(&err
);
205 g_free(addr
.u
.fd
.str
);
209 static void test_socket_fd_pass_num_nocli(void)
215 cur_mon
= g_malloc(1); /* Fake a monitor */
217 addr
.type
= SOCKET_ADDRESS_TYPE_FD
;
218 addr
.u
.fd
.str
= g_strdup_printf("%d", STDOUT_FILENO
);
220 fd
= socket_connect(&addr
, &err
);
221 g_assert_cmpint(fd
, ==, -1);
222 error_free_or_abort(&err
);
224 fd
= socket_listen(&addr
, 1, &err
);
225 g_assert_cmpint(fd
, ==, -1);
226 error_free_or_abort(&err
);
228 g_free(addr
.u
.fd
.str
);
233 static gpointer
unix_server_thread_func(gpointer user_data
)
235 SocketAddress
*addr
= user_data
;
238 struct sockaddr_un un
;
239 socklen_t len
= sizeof(un
);
241 fd
= socket_listen(addr
, 1, &error_abort
);
242 g_assert_cmpint(fd
, >=, 0);
243 g_assert(fd_is_socket(fd
));
245 connfd
= accept(fd
, (struct sockaddr
*)&un
, &len
);
246 g_assert_cmpint(connfd
, !=, -1);
253 static gpointer
unix_client_thread_func(gpointer user_data
)
255 SocketAddress
*addr
= user_data
;
258 fd
= socket_connect(addr
, &error_abort
);
259 g_assert_cmpint(fd
, >=, 0);
264 static void test_socket_unix_abstract_good(void)
268 addr
.type
= SOCKET_ADDRESS_TYPE_UNIX
;
269 addr
.u
.q_unix
.path
= g_strdup_printf("unix-%d-%u",
270 getpid(), g_random_int());
271 addr
.u
.q_unix
.has_abstract
= true;
272 addr
.u
.q_unix
.abstract
= true;
274 /* non tight socklen serv and cli */
276 addr
.u
.q_unix
.has_tight
= false;
277 addr
.u
.q_unix
.tight
= false;
279 GThread
*serv
= g_thread_new("abstract_unix_server",
280 unix_server_thread_func
,
285 GThread
*cli
= g_thread_new("abstract_unix_client",
286 unix_client_thread_func
,
292 /* tight socklen serv and cli */
294 addr
.u
.q_unix
.has_tight
= true;
295 addr
.u
.q_unix
.tight
= true;
297 serv
= g_thread_new("abstract_unix_server",
298 unix_server_thread_func
,
303 cli
= g_thread_new("abstract_unix_client",
304 unix_client_thread_func
,
310 g_free(addr
.u
.q_unix
.path
);
314 int main(int argc
, char **argv
)
316 bool has_ipv4
, has_ipv6
;
318 qemu_init_main_loop(&error_abort
);
321 g_test_init(&argc
, &argv
, NULL
);
323 /* We're creating actual IPv4/6 sockets, so we should
324 * check if the host running tests actually supports
325 * each protocol to avoid breaking tests on machines
326 * with either IPv4 or IPv6 disabled.
328 if (socket_check_protocol_support(&has_ipv4
, &has_ipv6
) < 0) {
329 g_printerr("socket_check_protocol_support() failed\n");
334 g_test_add_func("/util/socket/is-socket/bad",
335 test_fd_is_socket_bad
);
336 g_test_add_func("/util/socket/is-socket/good",
337 test_fd_is_socket_good
);
339 g_test_add_func("/socket/fd-pass/name/good",
340 test_socket_fd_pass_name_good
);
341 g_test_add_func("/socket/fd-pass/name/bad",
342 test_socket_fd_pass_name_bad
);
343 g_test_add_func("/socket/fd-pass/name/nomon",
344 test_socket_fd_pass_name_nomon
);
345 g_test_add_func("/socket/fd-pass/num/good",
346 test_socket_fd_pass_num_good
);
347 g_test_add_func("/socket/fd-pass/num/bad",
348 test_socket_fd_pass_num_bad
);
349 g_test_add_func("/socket/fd-pass/num/nocli",
350 test_socket_fd_pass_num_nocli
);
355 g_test_add_func("/util/socket/unix-abstract/good",
356 test_socket_unix_abstract_good
);