qemu

FORK: QEMU emulator
git clone https://git.neptards.moe/neptards/qemu.git
Log | Files | Refs | Submodules | LICENSE

test-filter-mirror.c (2404B)

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