xref: /qemu/tests/qtest/test-filter-mirror.c (revision abff1abf)
1 /*
2  * QTest testcase for filter-mirror
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 
11 #include "qemu/osdep.h"
12 #include "qemu-common.h"
13 #include "libqos/libqtest.h"
14 #include "qapi/qmp/qdict.h"
15 #include "qemu/iov.h"
16 #include "qemu/sockets.h"
17 #include "qemu/error-report.h"
18 #include "qemu/main-loop.h"
19 
20 /* TODO actually test the results and get rid of this */
21 #define qmp_discard_response(qs, ...) qobject_unref(qtest_qmp(qs, __VA_ARGS__))
22 
23 static void test_mirror(void)
24 {
25     int send_sock[2], recv_sock[2];
26     uint32_t ret = 0, len = 0;
27     char send_buf[] = "Hello! filter-mirror~";
28     char *recv_buf;
29     uint32_t size = sizeof(send_buf);
30     size = htonl(size);
31     const char *devstr = "e1000";
32     QTestState *qts;
33 
34     if (g_str_equal(qtest_get_arch(), "s390x")) {
35         devstr = "virtio-net-ccw";
36     }
37 
38     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, send_sock);
39     g_assert_cmpint(ret, !=, -1);
40 
41     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, recv_sock);
42     g_assert_cmpint(ret, !=, -1);
43 
44     qts = qtest_initf(
45         "-netdev socket,id=qtest-bn0,fd=%d "
46         "-device %s,netdev=qtest-bn0,id=qtest-e0 "
47         "-chardev socket,id=mirror0,fd=%d "
48         "-object filter-mirror,id=qtest-f0,netdev=qtest-bn0,queue=tx,outdev=mirror0 "
49         , send_sock[1], devstr, recv_sock[1]);
50 
51     struct iovec iov[] = {
52         {
53             .iov_base = &size,
54             .iov_len = sizeof(size),
55         }, {
56             .iov_base = send_buf,
57             .iov_len = sizeof(send_buf),
58         },
59     };
60 
61     /* send a qmp command to guarantee that 'connected' is setting to true. */
62     qmp_discard_response(qts, "{ 'execute' : 'query-status'}");
63     ret = iov_send(send_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
64     g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
65     close(send_sock[0]);
66 
67     ret = qemu_recv(recv_sock[0], &len, sizeof(len), 0);
68     g_assert_cmpint(ret, ==, sizeof(len));
69     len = ntohl(len);
70 
71     g_assert_cmpint(len, ==, sizeof(send_buf));
72     recv_buf = g_malloc(len);
73     ret = qemu_recv(recv_sock[0], recv_buf, len, 0);
74     g_assert_cmpstr(recv_buf, ==, send_buf);
75 
76     g_free(recv_buf);
77     close(send_sock[0]);
78     close(send_sock[1]);
79     close(recv_sock[0]);
80     close(recv_sock[1]);
81     qtest_quit(qts);
82 }
83 
84 int main(int argc, char **argv)
85 {
86     int ret;
87 
88     g_test_init(&argc, &argv, NULL);
89 
90     qtest_add_func("/netfilter/mirror", test_mirror);
91     ret = g_test_run();
92 
93     return ret;
94 }
95