In this unit test we will test the mirror function. start qemu with: -netdev socket,id=qtest-bn0,fd=sockfd -device e1000,netdev=qtest-bn0,id=qtest-e0 -chardev socket,id=mirror0,path=/tmp/filter-mirror-test.sock,server,nowait -object filter-mirror,id=qtest-f0,netdev=qtest-bn0,queue=tx,outdev=mirror0 We inject packet to netdev socket id = qtest-bn0, filter-mirror will copy and mirror the packet to mirror0. we read packet from mirror0 and then compare to what we injected. Signed-off-by: Zhang Chen <zhangchen.fnst@cn.fujitsu.com> Signed-off-by: Wen Congyang <wency@cn.fujitsu.com> Signed-off-by: Jason Wang <jasowang@redhat.com>
94 lines
2.5 KiB
C
94 lines
2.5 KiB
C
/*
|
|
* QTest testcase for filter-mirror
|
|
*
|
|
* Copyright (c) 2016 FUJITSU LIMITED
|
|
* Author: Zhang Chen <zhangchen.fnst@cn.fujitsu.com>
|
|
*
|
|
* This work is licensed under the terms of the GNU GPL, version 2 or
|
|
* later. See the COPYING file in the top-level directory.
|
|
*/
|
|
|
|
#include "qemu/osdep.h"
|
|
#include <glib.h>
|
|
#include "libqtest.h"
|
|
#include "qemu/iov.h"
|
|
#include "qemu/sockets.h"
|
|
#include "qemu/error-report.h"
|
|
#include "qemu/main-loop.h"
|
|
|
|
static void test_mirror(void)
|
|
{
|
|
#ifndef _WIN32
|
|
/* socketpair(PF_UNIX) which does not exist on windows */
|
|
|
|
int send_sock[2], recv_sock;
|
|
char *cmdline;
|
|
uint32_t ret = 0, len = 0;
|
|
char send_buf[] = "Hello! filter-mirror~";
|
|
char sock_path[] = "filter-mirror.XXXXXX";
|
|
char *recv_buf;
|
|
uint32_t size = sizeof(send_buf);
|
|
size = htonl(size);
|
|
|
|
ret = socketpair(PF_UNIX, SOCK_STREAM, 0, send_sock);
|
|
g_assert_cmpint(ret, !=, -1);
|
|
|
|
ret = mkstemp(sock_path);
|
|
g_assert_cmpint(ret, !=, -1);
|
|
|
|
cmdline = g_strdup_printf("-netdev socket,id=qtest-bn0,fd=%d "
|
|
"-device e1000,netdev=qtest-bn0,id=qtest-e0 "
|
|
"-chardev socket,id=mirror0,path=%s,server,nowait "
|
|
"-object filter-mirror,id=qtest-f0,netdev=qtest-bn0,queue=tx,outdev=mirror0 "
|
|
, send_sock[1], sock_path);
|
|
qtest_start(cmdline);
|
|
g_free(cmdline);
|
|
|
|
recv_sock = unix_connect(sock_path, NULL);
|
|
g_assert_cmpint(recv_sock, !=, -1);
|
|
|
|
struct iovec iov[] = {
|
|
{
|
|
.iov_base = &size,
|
|
.iov_len = sizeof(size),
|
|
}, {
|
|
.iov_base = send_buf,
|
|
.iov_len = sizeof(send_buf),
|
|
},
|
|
};
|
|
|
|
/* send a qmp command to guarantee that 'connected' is setting to true. */
|
|
qmp("{ 'execute' : 'query-status'}");
|
|
ret = iov_send(send_sock[0], iov, 2, 0, sizeof(size) + sizeof(send_buf));
|
|
g_assert_cmpint(ret, ==, sizeof(send_buf) + sizeof(size));
|
|
close(send_sock[0]);
|
|
|
|
ret = qemu_recv(recv_sock, &len, sizeof(len), 0);
|
|
g_assert_cmpint(ret, ==, sizeof(len));
|
|
len = ntohl(len);
|
|
|
|
g_assert_cmpint(len, ==, sizeof(send_buf));
|
|
recv_buf = g_malloc(len);
|
|
ret = qemu_recv(recv_sock, recv_buf, len, 0);
|
|
g_assert_cmpstr(recv_buf, ==, send_buf);
|
|
|
|
g_free(recv_buf);
|
|
close(recv_sock);
|
|
unlink(sock_path);
|
|
|
|
#endif
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int ret;
|
|
|
|
g_test_init(&argc, &argv, NULL);
|
|
|
|
qtest_add_func("/netfilter/mirror", test_mirror);
|
|
ret = g_test_run();
|
|
qtest_end();
|
|
|
|
return ret;
|
|
}
|