tests/qtest/test-filter-redirector.c

   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 "qemu-common.h"
  55 #include "libqos/libqtest.h"
  56 #include "qapi/qmp/qdict.h"
  57 #include "qemu/iov.h"
  58 #include "qemu/sockets.h"
  59 #include "qemu/error-report.h"
  60 #include "qemu/main-loop.h"
  61
  62 /* TODO actually test the results and get rid of this */
  63 #define qmp_discard_response(qs, ...) qobject_unref(qtest_qmp(qs, __VA_ARGS__))
  64
  65 static void test_redirector_tx(void)
  66 {
  67     int backend_sock[2], recv_sock;
  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     QTestState *qts;
  76
  77     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
  78     g_assert_cmpint(ret, !=, -1);
  79
  80     ret = mkstemp(sock_path0);
  81     g_assert_cmpint(ret, !=, -1);
  82     ret = mkstemp(sock_path1);
  83     g_assert_cmpint(ret, !=, -1);
  84
  85     qts = qtest_initf(
  86         "-nic socket,id=qtest-bn0,fd=%d "
  87         "-chardev socket,id=redirector0,path=%s,server=on,wait=off "
  88         "-chardev socket,id=redirector1,path=%s,server=on,wait=off "
  89         "-chardev socket,id=redirector2,path=%s "
  90         "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
  91         "queue=tx,outdev=redirector0 "
  92         "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
  93         "queue=tx,indev=redirector2 "
  94         "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
  95         "queue=tx,outdev=redirector1 ", backend_sock[1],
  96         sock_path0, sock_path1, sock_path0);
  97
  98     recv_sock = unix_connect(sock_path1, NULL);
  99     g_assert_cmpint(recv_sock, !=, -1);
 100
 101     /* send a qmp command to guarantee that 'connected' is setting to true. */
 102     qmp_discard_response(qts, "{ 'execute' : 'query-status'}");
 103
 104     struct iovec iov[] = {
 105         {
 106             .iov_base = &size,
 107             .iov_len = sizeof(size),
 108         }, {
 109             .iov_base = send_buf,
 110             .iov_len = sizeof(send_buf),
 111         },
 112     };
 113
 114     ret = iov_send(backend_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
 115     g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
 116     close(backend_sock[0]);
 117
 118     ret = qemu_recv(recv_sock, &len, sizeof(len), 0);
 119     g_assert_cmpint(ret, ==, sizeof(len));
 120     len = ntohl(len);
 121
 122     g_assert_cmpint(len, ==, sizeof(send_buf));
 123     recv_buf = g_malloc(len);
 124     ret = qemu_recv(recv_sock, recv_buf, len, 0);
 125     g_assert_cmpstr(recv_buf, ==, send_buf);
 126
 127     g_free(recv_buf);
 128     close(recv_sock);
 129     unlink(sock_path0);
 130     unlink(sock_path1);
 131     qtest_quit(qts);
 132 }
 133
 134 static void test_redirector_rx(void)
 135 {
 136     int backend_sock[2], send_sock;
 137     uint32_t ret = 0, len = 0;
 138     char send_buf[] = "Hello!!";
 139     char sock_path0[] = "filter-redirector0.XXXXXX";
 140     char sock_path1[] = "filter-redirector1.XXXXXX";
 141     char *recv_buf;
 142     uint32_t size = sizeof(send_buf);
 143     size = htonl(size);
 144     QTestState *qts;
 145
 146     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, backend_sock);
 147     g_assert_cmpint(ret, !=, -1);
 148
 149     ret = mkstemp(sock_path0);
 150     g_assert_cmpint(ret, !=, -1);
 151     ret = mkstemp(sock_path1);
 152     g_assert_cmpint(ret, !=, -1);
 153
 154     qts = qtest_initf(
 155         "-nic socket,id=qtest-bn0,fd=%d "
 156         "-chardev socket,id=redirector0,path=%s,server=on,wait=off "
 157         "-chardev socket,id=redirector1,path=%s,server=on,wait=off "
 158         "-chardev socket,id=redirector2,path=%s "
 159         "-object filter-redirector,id=qtest-f0,netdev=qtest-bn0,"
 160         "queue=rx,indev=redirector0 "
 161         "-object filter-redirector,id=qtest-f1,netdev=qtest-bn0,"
 162         "queue=rx,outdev=redirector2 "
 163         "-object filter-redirector,id=qtest-f2,netdev=qtest-bn0,"
 164         "queue=rx,indev=redirector1 ", backend_sock[1],
 165         sock_path0, sock_path1, sock_path0);
 166
 167     struct iovec iov[] = {
 168         {
 169             .iov_base = &size,
 170             .iov_len = sizeof(size),
 171         }, {
 172             .iov_base = send_buf,
 173             .iov_len = sizeof(send_buf),
 174         },
 175     };
 176
 177     send_sock = unix_connect(sock_path1, NULL);
 178     g_assert_cmpint(send_sock, !=, -1);
 179     /* send a qmp command to guarantee that 'connected' is setting to true. */
 180     qmp_discard_response(qts, "{ 'execute' : 'query-status'}");
 181
 182     ret = iov_send(send_sock, iov, 2, 0, sizeof(size) + sizeof(send_buf));
 183     g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
 184
 185     ret = qemu_recv(backend_sock[0], &len, sizeof(len), 0);
 186     g_assert_cmpint(ret, ==, sizeof(len));
 187     len = ntohl(len);
 188
 189     g_assert_cmpint(len, ==, sizeof(send_buf));
 190     recv_buf = g_malloc(len);
 191     ret = qemu_recv(backend_sock[0], recv_buf, len, 0);
 192     g_assert_cmpstr(recv_buf, ==, send_buf);
 193
 194     close(send_sock);
 195     g_free(recv_buf);
 196     unlink(sock_path0);
 197     unlink(sock_path1);
 198     qtest_quit(qts);
 199 }
 200
 201 int main(int argc, char **argv)
 202 {
 203     g_test_init(&argc, &argv, NULL);
 204     qtest_add_func("/netfilter/redirector_tx", test_redirector_tx);
 205     qtest_add_func("/netfilter/redirector_rx", test_redirector_rx);
 206     return g_test_run();
 207 }