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 "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
62#define qmp_discard_response(...) qobject_unref(qmp(__VA_ARGS__))
63
64static const char *get_devstr(void)
65{
66 if (g_str_equal(qtest_get_arch(), "s390x")) {
67 return "virtio-net-ccw";
68 }
69
70 return "rtl8139";
71}
72
73
74static void test_redirector_tx(void)
75{
76 int backend_sock[2], recv_sock;
77 uint32_t ret = 0, len = 0;
78 char send_buf[] = "Hello!!";
79 char sock_path0[] = "filter-redirector0.XXXXXX";
80 char sock_path1[] = "filter-redirector1.XXXXXX";
81 char *recv_buf;
82 uint32_t size = sizeof(send_buf);
83 size = htonl(size);
84
85 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
86 g_assert_cmpint(ret, !=, -1);
87
88 ret = mkstemp(sock_path0);
89 g_assert_cmpint(ret, !=, -1);
90 ret = mkstemp(sock_path1);
91 g_assert_cmpint(ret, !=, -1);
92
93 global_qtest = qtest_initf(
94 "-netdev socket,id=qtest-bn0,fd=%d "
95 "-device %s,netdev=qtest-bn0,id=qtest-e0 "
96 "-chardev socket,id=redirector0,path=%s,server,nowait "
97 "-chardev socket,id=redirector1,path=%s,server,nowait "
98 "-chardev socket,id=redirector2,path=%s,nowait "
99 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
100 "queue=tx,outdev=redirector0 "
101 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
102 "queue=tx,indev=redirector2 "
103 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
104 "queue=tx,outdev=redirector1 ", backend_sock[1], get_devstr(),
105 sock_path0, sock_path1, sock_path0);
106
107 recv_sock = unix_connect(sock_path1, NULL);
108 g_assert_cmpint(recv_sock, !=, -1);
109
110
111 qmp_discard_response("{ 'execute' : 'query-status'}");
112
113 struct iovec iov[] = {
114 {
115 .iov_base = &size,
116 .iov_len = sizeof(size),
117 }, {
118 .iov_base = send_buf,
119 .iov_len = sizeof(send_buf),
120 },
121 };
122
123 ret = iov_send(backend_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
124 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
125 close(backend_sock[0]);
126
127 ret = qemu_recv(recv_sock, &len, sizeof(len), 0);
128 g_assert_cmpint(ret, ==, sizeof(len));
129 len = ntohl(len);
130
131 g_assert_cmpint(len, ==, sizeof(send_buf));
132 recv_buf = g_malloc(len);
133 ret = qemu_recv(recv_sock, recv_buf, len, 0);
134 g_assert_cmpstr(recv_buf, ==, send_buf);
135
136 g_free(recv_buf);
137 close(recv_sock);
138 unlink(sock_path0);
139 unlink(sock_path1);
140 qtest_end();
141}
142
143static void test_redirector_rx(void)
144{
145 int backend_sock[2], send_sock;
146 uint32_t ret = 0, len = 0;
147 char send_buf[] = "Hello!!";
148 char sock_path0[] = "filter-redirector0.XXXXXX";
149 char sock_path1[] = "filter-redirector1.XXXXXX";
150 char *recv_buf;
151 uint32_t size = sizeof(send_buf);
152 size = htonl(size);
153
154 ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
155 g_assert_cmpint(ret, !=, -1);
156
157 ret = mkstemp(sock_path0);
158 g_assert_cmpint(ret, !=, -1);
159 ret = mkstemp(sock_path1);
160 g_assert_cmpint(ret, !=, -1);
161
162 global_qtest = qtest_initf(
163 "-netdev socket,id=qtest-bn0,fd=%d "
164 "-device %s,netdev=qtest-bn0,id=qtest-e0 "
165 "-chardev socket,id=redirector0,path=%s,server,nowait "
166 "-chardev socket,id=redirector1,path=%s,server,nowait "
167 "-chardev socket,id=redirector2,path=%s,nowait "
168 "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
169 "queue=rx,indev=redirector0 "
170 "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
171 "queue=rx,outdev=redirector2 "
172 "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
173 "queue=rx,indev=redirector1 ", backend_sock[1], get_devstr(),
174 sock_path0, sock_path1, sock_path0);
175
176 struct iovec iov[] = {
177 {
178 .iov_base = &size,
179 .iov_len = sizeof(size),
180 }, {
181 .iov_base = send_buf,
182 .iov_len = sizeof(send_buf),
183 },
184 };
185
186 send_sock = unix_connect(sock_path1, NULL);
187 g_assert_cmpint(send_sock, !=, -1);
188
189 qmp_discard_response("{ 'execute' : 'query-status'}");
190
191 ret = iov_send(send_sock, iov, 2, 0, sizeof(size) + sizeof(send_buf));
192 g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
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 close(send_sock);
204 g_free(recv_buf);
205 unlink(sock_path0);
206 unlink(sock_path1);
207 qtest_end();
208}
209
210int main(int argc, char **argv)
211{
212 g_test_init(&argc, &argv, NULL);
213 qtest_add_func("/netfilter/redirector_tx", test_redirector_tx);
214 qtest_add_func("/netfilter/redirector_rx", test_redirector_rx);
215 return g_test_run();
216}
217