sharpetronics/strongswan
1
0
Fork 0
mirror of https://github.com/strongswan/strongswan.git synced 2025-10-19 00:01:57 -04:00
Code Issues Projects Releases Wiki Activity

kernel-netlink: Add test cases for successful and timing out retransmissions

Browse Source
This commit is contained in:
Martin Willi 2014-07-15 15:47:03 +02:00
parent 553be051b7
commit adb930ed4f
2 changed files with 90 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/libhydra/plugins/kernel_netlink/Makefile.am
View File

@ -35,6 +35,7 @@ tests_SOURCES = \
tests_CFLAGS = \ tests_CFLAGS = \
-I$(top_srcdir)/src/libstrongswan \ -I$(top_srcdir)/src/libstrongswan \
-I$(top_srcdir)/src/libstrongswan/tests \ -I$(top_srcdir)/src/libstrongswan/tests \
-DNETLINK_MSG_LOSS_HOOK=netlink_msg_loss \
@COVERAGE_CFLAGS@ @COVERAGE_CFLAGS@
tests_LDFLAGS = @COVERAGE_LDFLAGS@ tests_LDFLAGS = @COVERAGE_LDFLAGS@

89
src/libhydra/plugins/kernel_netlink/suites/test_socket.c
View File

@ -19,6 +19,25 @@
#include "../kernel_netlink_shared.h" #include "../kernel_netlink_shared.h"
/**
* Netlink message drop configuration
*/
static int drop_interval = 0;
/**
* Netlink message drop hook
*/
bool netlink_msg_loss(struct nlmsghdr *hdr)
{
static refcount_t i;
if (drop_interval)
{
return ref_get(&i) % drop_interval == drop_interval - 1;
}
return FALSE;
}
START_TEST(test_echo) START_TEST(test_echo)
{ {
netlink_socket_t *s; netlink_socket_t *s;
@ -110,6 +129,71 @@ START_TEST(test_stress)
} }
END_TEST END_TEST
START_TEST(test_retransmit_success)
{
netlink_socket_t *s;
struct nlmsghdr *out;
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,
},
};
drop_interval = 2;
lib->settings->set_int(lib->settings,
"%s.plugins.kernel-netlink.timeout", 100, lib->ns);
lib->settings->set_int(lib->settings,
"%s.plugins.kernel-netlink.retries", 1, lib->ns);
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);
free(out);
s->destroy(s);
drop_interval = 0;
}
END_TEST
START_TEST(test_retransmit_fail)
{
netlink_socket_t *s;
struct nlmsghdr *out;
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,
},
};
drop_interval = 1;
lib->settings->set_int(lib->settings,
"%s.plugins.kernel-netlink.timeout", 50, lib->ns);
lib->settings->set_int(lib->settings,
"%s.plugins.kernel-netlink.retries", 3, lib->ns);
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) == OUT_OF_RES);
s->destroy(s);
drop_interval = 0;
}
END_TEST
Suite *socket_suite_create() Suite *socket_suite_create()
{ {
Suite *s; Suite *s;
@ -125,5 +209,10 @@ Suite *socket_suite_create()
tcase_add_test(tc, test_stress); tcase_add_test(tc, test_stress);
suite_add_tcase(s, tc); suite_add_tcase(s, tc);
tc = tcase_create("retransmit");
tcase_add_test(tc, test_retransmit_success);
tcase_add_test(tc, test_retransmit_fail);
suite_add_tcase(s, tc);
return s; return s;
} }