1 /* 2 * QTest testcase for filter-redirector 3 * 4 * Copyright (c) 2016 FUJITSU LIMITED 5 * Author: Zhang Chen <zhangchen.fnst@cn.fujitsu.com> 6 * 7 * This work is licensed under the terms of the GNU GPL, version 2 or 8 * later. See the COPYING file in the top-level directory. 9 * 10 * Case 1, tx traffic flow: 11 * 12 * qemu side | test side 13 * | 14 * +---------+ | +-------+ 15 * | backend <---------------+ sock0 | 16 * +----+----+ | +-------+ 17 * | | 18 * +----v----+ +-------+ | 19 * | rd0 +->+chardev| | 20 * +---------+ +---+---+ | 21 * | | 22 * +---------+ | | 23 * | rd1 <------+ | 24 * +----+----+ | 25 * | | 26 * +----v----+ | +-------+ 27 * | rd2 +--------------->sock1 | 28 * +---------+ | +-------+ 29 * + 30 * 31 * -------------------------------------- 32 * Case 2, rx traffic flow 33 * qemu side | test side 34 * | 35 * +---------+ | +-------+ 36 * | backend +---------------> sock1 | 37 * +----^----+ | +-------+ 38 * | | 39 * +----+----+ +-------+ | 40 * | rd0 +<-+chardev| | 41 * +---------+ +---+---+ | 42 * ^ | 43 * +---------+ | | 44 * | rd1 +------+ | 45 * +----^----+ | 46 * | | 47 * +----+----+ | +-------+ 48 * | rd2 <---------------+sock0 | 49 * +---------+ | +-------+ 50 * + 51 */ 52 53 #include "qemu/osdep.h" 54 #include "libqtest.h" 55 #include "qapi/qmp/qdict.h" 56 #include "qemu/iov.h" 57 #include "qemu/sockets.h" 58 #include "qemu/error-report.h" 59 #include "qemu/main-loop.h" 60 61 static void test_redirector_tx(void) 62 { 63 int backend_sock[2], recv_sock; 64 uint32_t ret = 0, len = 0; 65 char send_buf[] = "Hello!!"; 66 char sock_path0[] = "filter-redirector0.XXXXXX"; 67 char sock_path1[] = "filter-redirector1.XXXXXX"; 68 char *recv_buf; 69 uint32_t size = sizeof(send_buf); 70 size = htonl(size); 71 QTestState *qts; 72 73 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock); 74 g_assert_cmpint(ret, !=, -1); 75 76 ret = mkstemp(sock_path0); 77 g_assert_cmpint(ret, !=, -1); 78 ret = mkstemp(sock_path1); 79 g_assert_cmpint(ret, !=, -1); 80 81 qts = qtest_initf( 82 "-nic socket,id=qtest-bn0,fd=%d " 83 "-chardev socket,id=redirector0,path=%s,server=on,wait=off " 84 "-chardev socket,id=redirector1,path=%s,server=on,wait=off " 85 "-chardev socket,id=redirector2,path=%s " 86 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0," 87 "queue=tx,outdev=redirector0 " 88 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0," 89 "queue=tx,indev=redirector2 " 90 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0," 91 "queue=tx,outdev=redirector1 ", backend_sock[1], 92 sock_path0, sock_path1, sock_path0); 93 94 recv_sock = unix_connect(sock_path1, NULL); 95 g_assert_cmpint(recv_sock, !=, -1); 96 97 /* send a qmp command to guarantee that 'connected' is setting to true. */ 98 qtest_qmp_assert_success(qts, "{ 'execute' : 'query-status'}"); 99 100 struct iovec iov[] = { 101 { 102 .iov_base = &size, 103 .iov_len = sizeof(size), 104 }, { 105 .iov_base = send_buf, 106 .iov_len = sizeof(send_buf), 107 }, 108 }; 109 110 ret = iov_send(backend_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf)); 111 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size)); 112 close(backend_sock[0]); 113 114 ret = recv(recv_sock, &len, sizeof(len), 0); 115 g_assert_cmpint(ret, ==, sizeof(len)); 116 len = ntohl(len); 117 118 g_assert_cmpint(len, ==, sizeof(send_buf)); 119 recv_buf = g_malloc(len); 120 ret = recv(recv_sock, recv_buf, len, 0); 121 g_assert_cmpint(ret, ==, len); 122 g_assert_cmpstr(recv_buf, ==, send_buf); 123 124 g_free(recv_buf); 125 close(recv_sock); 126 unlink(sock_path0); 127 unlink(sock_path1); 128 qtest_quit(qts); 129 } 130 131 static void test_redirector_rx(void) 132 { 133 int backend_sock[2], send_sock; 134 uint32_t ret = 0, len = 0; 135 char send_buf[] = "Hello!!"; 136 char sock_path0[] = "filter-redirector0.XXXXXX"; 137 char sock_path1[] = "filter-redirector1.XXXXXX"; 138 char *recv_buf; 139 uint32_t size = sizeof(send_buf); 140 size = htonl(size); 141 QTestState *qts; 142 143 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock); 144 g_assert_cmpint(ret, !=, -1); 145 146 ret = mkstemp(sock_path0); 147 g_assert_cmpint(ret, !=, -1); 148 ret = mkstemp(sock_path1); 149 g_assert_cmpint(ret, !=, -1); 150 151 qts = qtest_initf( 152 "-nic socket,id=qtest-bn0,fd=%d " 153 "-chardev socket,id=redirector0,path=%s,server=on,wait=off " 154 "-chardev socket,id=redirector1,path=%s,server=on,wait=off " 155 "-chardev socket,id=redirector2,path=%s " 156 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0," 157 "queue=rx,indev=redirector0 " 158 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0," 159 "queue=rx,outdev=redirector2 " 160 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0," 161 "queue=rx,indev=redirector1 ", backend_sock[1], 162 sock_path0, sock_path1, sock_path0); 163 164 struct iovec iov[] = { 165 { 166 .iov_base = &size, 167 .iov_len = sizeof(size), 168 }, { 169 .iov_base = send_buf, 170 .iov_len = sizeof(send_buf), 171 }, 172 }; 173 174 send_sock = unix_connect(sock_path1, NULL); 175 g_assert_cmpint(send_sock, !=, -1); 176 /* send a qmp command to guarantee that 'connected' is setting to true. */ 177 qtest_qmp_assert_success(qts, "{ 'execute' : 'query-status'}"); 178 179 ret = iov_send(send_sock, iov, 2, 0, sizeof(size) + sizeof(send_buf)); 180 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size)); 181 182 ret = recv(backend_sock[0], &len, sizeof(len), 0); 183 g_assert_cmpint(ret, ==, sizeof(len)); 184 len = ntohl(len); 185 186 g_assert_cmpint(len, ==, sizeof(send_buf)); 187 recv_buf = g_malloc(len); 188 ret = recv(backend_sock[0], recv_buf, len, 0); 189 g_assert_cmpint(ret, ==, len); 190 g_assert_cmpstr(recv_buf, ==, send_buf); 191 192 close(send_sock); 193 g_free(recv_buf); 194 unlink(sock_path0); 195 unlink(sock_path1); 196 qtest_quit(qts); 197 } 198 199 int main(int argc, char **argv) 200 { 201 g_test_init(&argc, &argv, NULL); 202 qtest_add_func("/netfilter/redirector_tx", test_redirector_tx); 203 qtest_add_func("/netfilter/redirector_rx", test_redirector_rx); 204 return g_test_run(); 205 } 206