/* * Tests for util/qemu-sockets.c * * Copyright 2018 Red Hat, Inc. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this library; if not, see <http://www.gnu.org/licenses/>. * */ #include "qemu/osdep.h" #include "qemu-common.h" #include "qemu/sockets.h" #include "qapi/error.h" #include "socket-helpers.h" #include "monitor/monitor.h" static void test_fd_is_socket_bad(void) { char *tmp = g_strdup("qemu-test-util-sockets-XXXXXX"); int fd = mkstemp(tmp); if (fd != 0) { unlink(tmp); } g_free(tmp); g_assert(fd >= 0); g_assert(!fd_is_socket(fd)); close(fd); } static void test_fd_is_socket_good(void) { int fd = qemu_socket(PF_INET, SOCK_STREAM, 0); g_assert(fd >= 0); g_assert(fd_is_socket(fd)); close(fd); } static int mon_fd = -1; static const char *mon_fdname; __thread Monitor *cur_mon; int monitor_get_fd(Monitor *mon, const char *fdname, Error **errp) { g_assert(cur_mon); g_assert(mon == cur_mon); if (mon_fd == -1 || !g_str_equal(mon_fdname, fdname)) { error_setg(errp, "No fd named %s", fdname); return -1; } return dup(mon_fd); } /* * Syms of stubs in libqemuutil.a are discarded at .o file * granularity. To replace monitor_get_fd() and monitor_cur(), we * must ensure that we also replace any other symbol that is used in * the binary and would be taken from the same stub object file, * otherwise we get duplicate syms at link time. */ Monitor *monitor_cur(void) { return cur_mon; } int monitor_vprintf(Monitor *mon, const char *fmt, va_list ap) { abort(); } #ifndef _WIN32 static void test_socket_fd_pass_name_good(void) { SocketAddress addr; int fd; cur_mon = g_malloc(1); /* Fake a monitor */ mon_fdname = "myfd"; mon_fd = qemu_socket(AF_INET, SOCK_STREAM, 0); g_assert_cmpint(mon_fd, >, STDERR_FILENO); addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup(mon_fdname); fd = socket_connect(&addr, &error_abort); g_assert_cmpint(fd, !=, -1); g_assert_cmpint(fd, !=, mon_fd); close(fd); fd = socket_listen(&addr, 1, &error_abort); g_assert_cmpint(fd, !=, -1); g_assert_cmpint(fd, !=, mon_fd); close(fd); g_free(addr.u.fd.str); mon_fdname = NULL; close(mon_fd); mon_fd = -1; g_free(cur_mon); cur_mon = NULL; } static void test_socket_fd_pass_name_bad(void) { SocketAddress addr; Error *err = NULL; int fd; cur_mon = g_malloc(1); /* Fake a monitor */ mon_fdname = "myfd"; mon_fd = dup(STDOUT_FILENO); g_assert_cmpint(mon_fd, >, STDERR_FILENO); addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup(mon_fdname); fd = socket_connect(&addr, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); fd = socket_listen(&addr, 1, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); g_free(addr.u.fd.str); mon_fdname = NULL; close(mon_fd); mon_fd = -1; g_free(cur_mon); cur_mon = NULL; } static void test_socket_fd_pass_name_nomon(void) { SocketAddress addr; Error *err = NULL; int fd; g_assert(cur_mon == NULL); addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup("myfd"); fd = socket_connect(&addr, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); fd = socket_listen(&addr, 1, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); g_free(addr.u.fd.str); } static void test_socket_fd_pass_num_good(void) { SocketAddress addr; int fd, sfd; g_assert(cur_mon == NULL); sfd = qemu_socket(AF_INET, SOCK_STREAM, 0); g_assert_cmpint(sfd, >, STDERR_FILENO); addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup_printf("%d", sfd); fd = socket_connect(&addr, &error_abort); g_assert_cmpint(fd, ==, sfd); fd = socket_listen(&addr, 1, &error_abort); g_assert_cmpint(fd, ==, sfd); g_free(addr.u.fd.str); close(sfd); } static void test_socket_fd_pass_num_bad(void) { SocketAddress addr; Error *err = NULL; int fd, sfd; g_assert(cur_mon == NULL); sfd = dup(STDOUT_FILENO); addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup_printf("%d", sfd); fd = socket_connect(&addr, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); fd = socket_listen(&addr, 1, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); g_free(addr.u.fd.str); close(sfd); } static void test_socket_fd_pass_num_nocli(void) { SocketAddress addr; Error *err = NULL; int fd; cur_mon = g_malloc(1); /* Fake a monitor */ addr.type = SOCKET_ADDRESS_TYPE_FD; addr.u.fd.str = g_strdup_printf("%d", STDOUT_FILENO); fd = socket_connect(&addr, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); fd = socket_listen(&addr, 1, &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); g_free(addr.u.fd.str); } #endif #ifdef CONFIG_LINUX #define ABSTRACT_SOCKET_VARIANTS 3 typedef struct { SocketAddress *server, *client[ABSTRACT_SOCKET_VARIANTS]; bool expect_connect[ABSTRACT_SOCKET_VARIANTS]; } abstract_socket_matrix_row; static gpointer unix_client_thread_func(gpointer user_data) { abstract_socket_matrix_row *row = user_data; Error *err = NULL; int i, fd; for (i = 0; i < ABSTRACT_SOCKET_VARIANTS; i++) { if (row->expect_connect[i]) { fd = socket_connect(row->client[i], &error_abort); g_assert_cmpint(fd, >=, 0); } else { fd = socket_connect(row->client[i], &err); g_assert_cmpint(fd, ==, -1); error_free_or_abort(&err); } close(fd); } return NULL; } static void test_socket_unix_abstract_row(abstract_socket_matrix_row *test) { int fd, connfd, i; GThread *cli; struct sockaddr_un un; socklen_t len = sizeof(un); /* Last one must connect, or else accept() below hangs */ assert(test->expect_connect[ABSTRACT_SOCKET_VARIANTS - 1]); fd = socket_listen(test->server, 1, &error_abort); g_assert_cmpint(fd, >=, 0); g_assert(fd_is_socket(fd)); cli = g_thread_new("abstract_unix_client", unix_client_thread_func, test); for (i = 0; i < ABSTRACT_SOCKET_VARIANTS; i++) { if (test->expect_connect[i]) { connfd = accept(fd, (struct sockaddr *)&un, &len); g_assert_cmpint(connfd, !=, -1); close(connfd); } } close(fd); g_thread_join(cli); } static void test_socket_unix_abstract(void) { SocketAddress addr, addr_tight, addr_padded; abstract_socket_matrix_row matrix[ABSTRACT_SOCKET_VARIANTS] = { { &addr, { &addr_tight, &addr_padded, &addr }, { true, false, true } }, { &addr_tight, { &addr_padded, &addr, &addr_tight }, { false, true, true } }, { &addr_padded, { &addr, &addr_tight, &addr_padded }, { false, false, true } } }; int i; addr.type = SOCKET_ADDRESS_TYPE_UNIX; addr.u.q_unix.path = g_strdup_printf("unix-%d-%u", getpid(), g_random_int()); addr.u.q_unix.has_abstract = true; addr.u.q_unix.abstract = true; addr.u.q_unix.has_tight = false; addr.u.q_unix.tight = false; addr_tight = addr; addr_tight.u.q_unix.has_tight = true; addr_tight.u.q_unix.tight = true; addr_padded = addr; addr_padded.u.q_unix.has_tight = true; addr_padded.u.q_unix.tight = false; for (i = 0; i < ABSTRACT_SOCKET_VARIANTS; i++) { test_socket_unix_abstract_row(&matrix[i]); } g_free(addr.u.q_unix.path); } #endif /* CONFIG_LINUX */ int main(int argc, char **argv) { bool has_ipv4, has_ipv6; qemu_init_main_loop(&error_abort); socket_init(); g_test_init(&argc, &argv, NULL); /* We're creating actual IPv4/6 sockets, so we should * check if the host running tests actually supports * each protocol to avoid breaking tests on machines * with either IPv4 or IPv6 disabled. */ if (socket_check_protocol_support(&has_ipv4, &has_ipv6) < 0) { g_printerr("socket_check_protocol_support() failed\n"); goto end; } if (has_ipv4) { g_test_add_func("/util/socket/is-socket/bad", test_fd_is_socket_bad); g_test_add_func("/util/socket/is-socket/good", test_fd_is_socket_good); #ifndef _WIN32 g_test_add_func("/socket/fd-pass/name/good", test_socket_fd_pass_name_good); g_test_add_func("/socket/fd-pass/name/bad", test_socket_fd_pass_name_bad); g_test_add_func("/socket/fd-pass/name/nomon", test_socket_fd_pass_name_nomon); g_test_add_func("/socket/fd-pass/num/good", test_socket_fd_pass_num_good); g_test_add_func("/socket/fd-pass/num/bad", test_socket_fd_pass_num_bad); g_test_add_func("/socket/fd-pass/num/nocli", test_socket_fd_pass_num_nocli); #endif } #ifdef CONFIG_LINUX g_test_add_func("/util/socket/unix-abstract", test_socket_unix_abstract); #endif end: return g_test_run(); }