1 /* 2 * QEMU I/O channel command test 3 * 4 * Copyright (c) 2015 Red Hat, Inc. 5 * 6 * This library is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * This library 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 GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with this library; if not, see <http://www.gnu.org/licenses/>. 18 * 19 */ 20 21 #include "qemu/osdep.h" 22 #include <glib/gstdio.h> 23 #include "io/channel-command.h" 24 #include "io-channel-helpers.h" 25 #include "qapi/error.h" 26 #include "qemu/module.h" 27 28 #define TEST_FIFO "test-io-channel-command.fifo" 29 30 static char *socat = NULL; 31 32 #if !defined(_WIN32) && !defined(CONFIG_DARWIN) 33 static void test_io_channel_command_fifo(bool async) 34 { 35 g_autofree gchar *tmpdir = g_dir_make_tmp("qemu-test-io-channel.XXXXXX", NULL); 36 g_autofree gchar *fifo = g_build_filename(tmpdir, TEST_FIFO, NULL); 37 g_autofree gchar *srcargs = g_strdup_printf("%s - PIPE:%s,wronly", socat, fifo); 38 g_autofree gchar *dstargs = g_strdup_printf("%s PIPE:%s,rdonly -", socat, fifo); 39 g_auto(GStrv) srcargv = g_strsplit(srcargs, " ", -1); 40 g_auto(GStrv) dstargv = g_strsplit(dstargs, " ", -1); 41 QIOChannel *src, *dst; 42 QIOChannelTest *test; 43 int err; 44 45 if (mkfifo(fifo, 0600)) { 46 g_error("mkfifo: %s", strerror(errno)); 47 } 48 49 src = QIO_CHANNEL(qio_channel_command_new_spawn((const char **) srcargv, 50 O_WRONLY, 51 &error_abort)); 52 dst = QIO_CHANNEL(qio_channel_command_new_spawn((const char **) dstargv, 53 O_RDONLY, 54 &error_abort)); 55 56 test = qio_channel_test_new(); 57 qio_channel_test_run_threads(test, async, src, dst); 58 qio_channel_test_validate(test); 59 60 object_unref(OBJECT(src)); 61 object_unref(OBJECT(dst)); 62 63 err = g_unlink(fifo); 64 g_assert(err == 0); 65 err = g_rmdir(tmpdir); 66 g_assert(err == 0); 67 } 68 69 static void test_io_channel_command_fifo_async(void) 70 { 71 if (!socat) { 72 g_test_skip("socat is not found in PATH"); 73 return; 74 } 75 76 test_io_channel_command_fifo(true); 77 } 78 79 static void test_io_channel_command_fifo_sync(void) 80 { 81 if (!socat) { 82 g_test_skip("socat is not found in PATH"); 83 return; 84 } 85 86 test_io_channel_command_fifo(false); 87 } 88 #endif 89 90 91 static void test_io_channel_command_echo(bool async) 92 { 93 QIOChannel *ioc; 94 QIOChannelTest *test; 95 const char *socatargv[] = { 96 socat, "-", "-", NULL, 97 }; 98 99 if (!socat) { 100 g_test_skip("socat is not found in PATH"); 101 return; 102 } 103 104 ioc = QIO_CHANNEL(qio_channel_command_new_spawn(socatargv, 105 O_RDWR, 106 &error_abort)); 107 test = qio_channel_test_new(); 108 qio_channel_test_run_threads(test, async, ioc, ioc); 109 qio_channel_test_validate(test); 110 111 object_unref(OBJECT(ioc)); 112 } 113 114 115 static void test_io_channel_command_echo_async(void) 116 { 117 test_io_channel_command_echo(true); 118 } 119 120 static void test_io_channel_command_echo_sync(void) 121 { 122 test_io_channel_command_echo(false); 123 } 124 125 int main(int argc, char **argv) 126 { 127 module_call_init(MODULE_INIT_QOM); 128 129 g_test_init(&argc, &argv, NULL); 130 131 socat = g_find_program_in_path("socat"); 132 133 #if !defined(_WIN32) && !defined(CONFIG_DARWIN) 134 g_test_add_func("/io/channel/command/fifo/sync", 135 test_io_channel_command_fifo_sync); 136 g_test_add_func("/io/channel/command/fifo/async", 137 test_io_channel_command_fifo_async); 138 #endif 139 g_test_add_func("/io/channel/command/echo/sync", 140 test_io_channel_command_echo_sync); 141 g_test_add_func("/io/channel/command/echo/async", 142 test_io_channel_command_echo_async); 143 144 return g_test_run(); 145 } 146