kernel-netlink: Add a simple send message test querying available links
This commit is contained in:
@@ -29,6 +29,7 @@ check_PROGRAMS = $(TESTS)
|
||||
|
||||
tests_SOURCES = \
|
||||
tests.h tests.c \
|
||||
suites/test_socket.c \
|
||||
kernel_netlink_shared.c
|
||||
|
||||
tests_CFLAGS = \
|
||||
|
||||
@@ -0,0 +1,67 @@
|
||||
/*
|
||||
* Copyright (C) 2014 Martin Willi
|
||||
* Copyright (C) 2014 revosec AG
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the
|
||||
* Free Software Foundation; either version 2 of the License, or (at your
|
||||
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* for more details.
|
||||
*/
|
||||
|
||||
#include <test_suite.h>
|
||||
|
||||
#include "../kernel_netlink_shared.h"
|
||||
|
||||
START_TEST(test_echo)
|
||||
{
|
||||
netlink_socket_t *s;
|
||||
struct nlmsghdr *out, *current;
|
||||
struct rtgenmsg *msg;
|
||||
size_t len;
|
||||
netlink_buf_t request = {
|
||||
.hdr = {
|
||||
.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtgenmsg)),
|
||||
.nlmsg_flags = NLM_F_REQUEST | NLM_F_MATCH | NLM_F_ROOT,
|
||||
.nlmsg_type = RTM_GETLINK,
|
||||
},
|
||||
};
|
||||
|
||||
s = netlink_socket_create(NETLINK_ROUTE, NULL);
|
||||
msg = NLMSG_DATA(&request.hdr);
|
||||
msg->rtgen_family = AF_UNSPEC;
|
||||
|
||||
ck_assert(s->send(s, &request.hdr, &out, &len) == SUCCESS);
|
||||
current = out;
|
||||
while (TRUE)
|
||||
{
|
||||
ck_assert(NLMSG_OK(current, len));
|
||||
if (current->nlmsg_type == NLMSG_DONE)
|
||||
{
|
||||
break;
|
||||
}
|
||||
ck_assert_int_eq(current->nlmsg_type, RTM_NEWLINK);
|
||||
current = NLMSG_NEXT(current, len);
|
||||
}
|
||||
free(out);
|
||||
s->destroy(s);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
Suite *socket_suite_create()
|
||||
{
|
||||
Suite *s;
|
||||
TCase *tc;
|
||||
|
||||
s = suite_create("netlink socket");
|
||||
|
||||
tc = tcase_create("echo");
|
||||
tcase_add_test(tc, test_echo);
|
||||
suite_add_tcase(s, tc);
|
||||
|
||||
return s;
|
||||
}
|
||||
@@ -12,3 +12,5 @@
|
||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* for more details.
|
||||
*/
|
||||
|
||||
TEST_SUITE(socket_suite_create)
|
||||
|
||||
Reference in New Issue
Block a user