tests/qtest/test-filter-mirror.c

   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     QTestState *qts;
  32
  33     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, send_sock);
  34     g_assert_cmpint(ret, !=, -1);
  35
  36     ret = socketpair(PF_UNIX, SOCK_STREAM, 0, recv_sock);
  37     g_assert_cmpint(ret, !=, -1);
  38
  39     qts = qtest_initf(
  40         "-nic socket,id=qtest-bn0,fd=%d "
  41         "-chardev socket,id=mirror0,fd=%d "
  42         "-object filter-mirror,id=qtest-f0,netdev=qtest-bn0,queue=tx,outdev=mirror0 "
  43         , send_sock[1], recv_sock[1]);
  44
  45     struct iovec iov[] = {
  46         {
  47             .iov_base = &size,
  48             .iov_len = sizeof(size),
  49         }, {
  50             .iov_base = send_buf,
  51             .iov_len = sizeof(send_buf),
  52         },
  53     };
  54
  55     /* send a qmp command to guarantee that 'connected' is setting to true. */
  56     qmp_discard_response(qts, "{ 'execute' : 'query-status'}");
  57     ret = iov_send(send_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
  58     g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
  59     close(send_sock[0]);
  60
  61     ret = recv(recv_sock[0], &len, sizeof(len), 0);
  62     g_assert_cmpint(ret, ==, sizeof(len));
  63     len = ntohl(len);
  64
  65     g_assert_cmpint(len, ==, sizeof(send_buf));
  66     recv_buf = g_malloc(len);
  67     ret = recv(recv_sock[0], recv_buf, len, 0);
  68     g_assert_cmpstr(recv_buf, ==, send_buf);
  69
  70     g_free(recv_buf);
  71     close(send_sock[0]);
  72     close(send_sock[1]);
  73     close(recv_sock[0]);
  74     close(recv_sock[1]);
  75     qtest_quit(qts);
  76 }
  77
  78 int main(int argc, char **argv)
  79 {
  80     int ret;
  81
  82     g_test_init(&argc, &argv, NULL);
  83
  84     qtest_add_func("/netfilter/mirror", test_mirror);
  85     ret = g_test_run();
  86
  87     return ret;
  88 }