1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53#include "qemu/osdep.h"
54#include <glib.h>
55#include "libqtest.h"
56#include "qemu/iov.h"
57#include "qemu/sockets.h"
58#include "qemu/error-report.h"
59#include "qemu/main-loop.h"
60
61static void test_redirector_tx(void)
62{
63#ifndef _WIN32
64
65
66 int backend_sock[2], recv_sock;
67 char *cmdline;
68 uint32_t ret = 0, len = 0;
69 char send_buf[] = "Hello!!";
70 char sock_path0[] = "filter-redirector0.XXXXXX";
71 char sock_path1[] = "filter-redirector1.XXXXXX";
72 char *recv_buf;
73 uint32_t size = sizeof(send_buf);
74 size = htonl(size);
75
76 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
77 g_assert_cmpint(ret, !=, -1);
78
79 ret = mkstemp(sock_path0);
80 g_assert_cmpint(ret, !=, -1);
81 ret = mkstemp(sock_path1);
82 g_assert_cmpint(ret, !=, -1);
83
84 cmdline = g_strdup_printf("-netdev socket,id=qtest-bn0,fd=%d "
85 "-device rtl8139,netdev=qtest-bn0,id=qtest-e0 "
86 "-chardev socket,id=redirector0,path=%s,server,nowait "
87 "-chardev socket,id=redirector1,path=%s,server,nowait "
88 "-chardev socket,id=redirector2,path=%s,nowait "
89 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
90 "queue=tx,outdev=redirector0 "
91 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
92 "queue=tx,indev=redirector2 "
93 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
94 "queue=tx,outdev=redirector1 "
95 , backend_sock[1], sock_path0, sock_path1, sock_path0);
96 qtest_start(cmdline);
97 g_free(cmdline);
98
99 recv_sock = unix_connect(sock_path1, NULL);
100 g_assert_cmpint(recv_sock, !=, -1);
101
102
103 qmp("{ 'execute' : 'query-status'}");
104
105 struct iovec iov[] = {
106 {
107 .iov_base = &size,
108 .iov_len = sizeof(size),
109 }, {
110 .iov_base = send_buf,
111 .iov_len = sizeof(send_buf),
112 },
113 };
114
115 ret = iov_send(backend_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
116 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
117 close(backend_sock[0]);
118
119 ret = qemu_recv(recv_sock, &len, sizeof(len), 0);
120 g_assert_cmpint(ret, ==, sizeof(len));
121 len = ntohl(len);
122
123 g_assert_cmpint(len, ==, sizeof(send_buf));
124 recv_buf = g_malloc(len);
125 ret = qemu_recv(recv_sock, recv_buf, len, 0);
126 g_assert_cmpstr(recv_buf, ==, send_buf);
127
128 g_free(recv_buf);
129 close(recv_sock);
130 unlink(sock_path0);
131 unlink(sock_path1);
132 qtest_end();
133
134#endif
135}
136
137static void test_redirector_rx(void)
138{
139#ifndef _WIN32
140
141
142 int backend_sock[2], send_sock;
143 char *cmdline;
144 uint32_t ret = 0, len = 0;
145 char send_buf[] = "Hello!!";
146 char sock_path0[] = "filter-redirector0.XXXXXX";
147 char sock_path1[] = "filter-redirector1.XXXXXX";
148 char *recv_buf;
149 uint32_t size = sizeof(send_buf);
150 size = htonl(size);
151
152 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
153 g_assert_cmpint(ret, !=, -1);
154
155 ret = mkstemp(sock_path0);
156 g_assert_cmpint(ret, !=, -1);
157 ret = mkstemp(sock_path1);
158 g_assert_cmpint(ret, !=, -1);
159
160 cmdline = g_strdup_printf("-netdev socket,id=qtest-bn0,fd=%d "
161 "-device rtl8139,netdev=qtest-bn0,id=qtest-e0 "
162 "-chardev socket,id=redirector0,path=%s,server,nowait "
163 "-chardev socket,id=redirector1,path=%s,server,nowait "
164 "-chardev socket,id=redirector2,path=%s,nowait "
165 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
166 "queue=rx,indev=redirector0 "
167 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
168 "queue=rx,outdev=redirector2 "
169 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
170 "queue=rx,indev=redirector1 "
171 , backend_sock[1], sock_path0, sock_path1, sock_path0);
172 qtest_start(cmdline);
173 g_free(cmdline);
174
175 struct iovec iov[] = {
176 {
177 .iov_base = &size,
178 .iov_len = sizeof(size),
179 }, {
180 .iov_base = send_buf,
181 .iov_len = sizeof(send_buf),
182 },
183 };
184
185 send_sock = unix_connect(sock_path1, NULL);
186 g_assert_cmpint(send_sock, !=, -1);
187
188 qmp("{ 'execute' : 'query-status'}");
189
190 ret = iov_send(send_sock, iov, 2, 0, sizeof(size) + sizeof(send_buf));
191 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
192 close(send_sock);
193
194 ret = qemu_recv(backend_sock[0], &len, sizeof(len), 0);
195 g_assert_cmpint(ret, ==, sizeof(len));
196 len = ntohl(len);
197
198 g_assert_cmpint(len, ==, sizeof(send_buf));
199 recv_buf = g_malloc(len);
200 ret = qemu_recv(backend_sock[0], recv_buf, len, 0);
201 g_assert_cmpstr(recv_buf, ==, send_buf);
202
203 g_free(recv_buf);
204 unlink(sock_path0);
205 unlink(sock_path1);
206 qtest_end();
207
208#endif
209}
210
211int main(int argc, char **argv)
212{
213 int ret;
214
215 g_test_init(&argc, &argv, NULL);
216 qtest_add_func("/netfilter/redirector_tx", test_redirector_tx);
217 qtest_add_func("/netfilter/redirector_rx", test_redirector_rx);
218 ret = g_test_run();
219
220 return ret;
221}
222
