LXR qemu/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
  23static 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
  84int 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