unit-tests: Add additional tests for host_t

This commit is contained in:
Tobias Brunner
2013-07-25 11:28:26 +02:00
parent 7a192c57a3
commit ebb4ad1baa
+551 -3
View File
@@ -17,10 +17,10 @@
#include <networking/host.h>
/*******************************************************************************
* host_create_netmask
/**
* Verify a netmask (a number of set bits starting at byte 0)
* Can also be used to check for %any (mask == 0)
*/
static void verify_netmask(chunk_t addr, int mask)
{
int byte, bit;
@@ -42,6 +42,361 @@ static void verify_netmask(chunk_t addr, int mask)
}
}
/*******************************************************************************
* host_create_any
*/
static void verify_any(host_t *host, int family, u_int16_t port)
{
verify_netmask(host->get_address(host), 0);
ck_assert(host->is_anyaddr(host));
ck_assert_int_eq(host->get_port(host), port);
ck_assert_int_eq(host->get_family(host), family);
}
static void test_create_any(int family)
{
host_t *host;
host = host_create_any(family);
verify_any(host, family, 0);
host->destroy(host);
}
START_TEST(test_create_any_v4)
{
test_create_any(AF_INET);
}
END_TEST
START_TEST(test_create_any_v6)
{
test_create_any(AF_INET6);
}
END_TEST
START_TEST(test_create_any_other)
{
host_t *host;
host = host_create_any(AF_UNSPEC);
ck_assert(host == NULL);
}
END_TEST
/*******************************************************************************
* host_create_from_string
*/
static void verify_address(host_t *host, chunk_t addr, int family, u_int16_t port)
{
ck_assert(chunk_equals(host->get_address(host), addr));
ck_assert(!host->is_anyaddr(host));
ck_assert_int_eq(host->get_port(host), port);
ck_assert_int_eq(host->get_family(host), family);
}
static const chunk_t addr_v4 = chunk_from_chars(0xc0, 0xa8, 0x00, 0x01);
static const chunk_t addr_v6 = chunk_from_chars(0xfe, 0xc1, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01);
START_TEST(test_create_from_string_v4)
{
host_t *host;
host = host_create_from_string("%any", 500);
verify_any(host, AF_INET, 500);
host->destroy(host);
host = host_create_from_string("%any4", 500);
verify_any(host, AF_INET, 500);
host->destroy(host);
host = host_create_from_string("0.0.0.0", 500);
verify_any(host, AF_INET, 500);
host->destroy(host);
host = host_create_from_string("192.168.0.1", 500);
verify_address(host, addr_v4, AF_INET, 500);
host->destroy(host);
host = host_create_from_string("192.168.0.1::500", 500);
ck_assert(host == NULL);
host = host_create_from_string("123.456.789.012", 500);
ck_assert(host == NULL);
host = host_create_from_string("1.1.1.1.1.1.1.1", 500);
ck_assert(host == NULL);
host = host_create_from_string("foo.b.a.r", 500);
ck_assert(host == NULL);
}
END_TEST
START_TEST(test_create_from_string_any_v6)
{
host_t *host;
host = host_create_from_string("%any6", 500);
verify_any(host, AF_INET6, 500);
host->destroy(host);
host = host_create_from_string("::", 500);
verify_any(host, AF_INET6, 500);
host->destroy(host);
host = host_create_from_string("fec1::1", 500);
verify_address(host, addr_v6, AF_INET6, 500);
host->destroy(host);
host = host_create_from_string("fec1::1.500", 500);
ck_assert(host == NULL);
host = host_create_from_string("f::e::c::1::1", 500);
ck_assert(host == NULL);
host = host_create_from_string("foo::bar", 500);
ck_assert(host == NULL);
}
END_TEST
/*******************************************************************************
* host_create_from_string_and_family
*/
static void test_create_from_string_and_family_any(char *string, int family,
int expected)
{
host_t *host;
host = host_create_from_string_and_family(string, family, 500);
if (expected == AF_UNSPEC)
{
ck_assert(host == NULL);
}
else
{
verify_any(host, expected, 500);
host->destroy(host);
}
}
static void test_create_from_string_and_family_addr(char *string, chunk_t addr,
int family, int expected)
{
host_t *host;
host = host_create_from_string_and_family(string, family, 500);
if (expected == AF_UNSPEC)
{
ck_assert(host == NULL);
}
else
{
verify_address(host, addr, expected, 500);
host->destroy(host);
}
}
START_TEST(test_create_from_string_and_family_v4)
{
test_create_from_string_and_family_any("%any", AF_INET, AF_INET);
test_create_from_string_and_family_any("%any4", AF_INET, AF_INET);
test_create_from_string_and_family_any("0.0.0.0", AF_INET, AF_INET);
test_create_from_string_and_family_any("%any4", AF_INET6, AF_UNSPEC);
test_create_from_string_and_family_any("0.0.0.0", AF_INET6, AF_UNSPEC);
test_create_from_string_and_family_addr("192.168.0.1", addr_v4, AF_INET, AF_INET);
test_create_from_string_and_family_addr("192.168.0.1", addr_v4, AF_INET6, AF_UNSPEC);
}
END_TEST
START_TEST(test_create_from_string_and_family_v6)
{
test_create_from_string_and_family_any("%any", AF_INET6, AF_INET6);
test_create_from_string_and_family_any("%any6", AF_INET6, AF_INET6);
test_create_from_string_and_family_any("::", AF_INET6, AF_INET6);
test_create_from_string_and_family_any("%any6", AF_INET, AF_UNSPEC);
test_create_from_string_and_family_any("::", AF_INET, AF_UNSPEC);
test_create_from_string_and_family_addr("fec1::1", addr_v6, AF_INET6, AF_INET6);
test_create_from_string_and_family_addr("fec1::1", addr_v6, AF_INET, AF_UNSPEC);
}
END_TEST
START_TEST(test_create_from_string_and_family_other)
{
test_create_from_string_and_family_any("%any", AF_UNSPEC, AF_INET);
test_create_from_string_and_family_any("%any4", AF_UNSPEC, AF_INET);
test_create_from_string_and_family_any("0.0.0.0", AF_UNSPEC, AF_INET);
test_create_from_string_and_family_any("%any6", AF_UNSPEC, AF_INET6);
test_create_from_string_and_family_any("::", AF_UNSPEC, AF_INET6);
test_create_from_string_and_family_addr("192.168.0.1", addr_v4, AF_UNSPEC, AF_INET);
test_create_from_string_and_family_addr("fec1::1", addr_v6, AF_UNSPEC, AF_INET6);
}
END_TEST
/*******************************************************************************
* host_create_from_sockaddr
*/
START_TEST(test_create_from_sockaddr_v4)
{
struct sockaddr_in addr = {
.sin_family = AF_INET,
.sin_port = htons(500),
}, *val;
socklen_t *socklen;
host_t *host;
host = host_create_from_sockaddr((sockaddr_t*)&addr);
verify_any(host, AF_INET, 500);
val = (struct sockaddr_in*)host->get_sockaddr(host);
ck_assert(memeq(&addr, val, sizeof(addr)));
socklen = host->get_sockaddr_len(host);
ck_assert(*socklen == sizeof(addr));
host->destroy(host);
}
END_TEST
START_TEST(test_create_from_sockaddr_v6)
{
struct sockaddr_in6 addr = {
.sin6_family = AF_INET6,
.sin6_port = htons(500),
}, *val;
socklen_t *socklen;
host_t *host;
host = host_create_from_sockaddr((sockaddr_t*)&addr);
verify_any(host, AF_INET6, 500);
val = (struct sockaddr_in6*)host->get_sockaddr(host);
ck_assert(memeq(&addr, val, sizeof(addr)));
socklen = host->get_sockaddr_len(host);
ck_assert(*socklen == sizeof(addr));
host->destroy(host);
}
END_TEST
START_TEST(test_create_from_sockaddr_other)
{
struct sockaddr_un addr = {
.sun_family = AF_UNIX,
};
host_t *host;
host = host_create_from_sockaddr((sockaddr_t*)&addr);
ck_assert(host == NULL);
}
END_TEST
/*******************************************************************************
* host_create_from_chunk
*/
START_TEST(test_create_from_chunk_v4)
{
host_t *host;
host = host_create_from_chunk(AF_INET, addr_v4, 500);
verify_address(host, addr_v4, AF_INET, 500);
host->destroy(host);
host = host_create_from_chunk(AF_UNSPEC, addr_v4, 500);
verify_address(host, addr_v4, AF_INET, 500);
host->destroy(host);
host = host_create_from_chunk(AF_INET, chunk_empty, 500);
ck_assert(host == NULL);
host = host_create_from_chunk(AF_UNSPEC, chunk_empty, 500);
ck_assert(host == NULL);
}
END_TEST
START_TEST(test_create_from_chunk_v6)
{
host_t *host;
host = host_create_from_chunk(AF_INET6, addr_v6, 500);
verify_address(host, addr_v6, AF_INET6, 500);
host->destroy(host);
host = host_create_from_chunk(AF_UNSPEC, addr_v6, 500);
verify_address(host, addr_v6, AF_INET6, 500);
host->destroy(host);
host = host_create_from_chunk(AF_INET6, chunk_empty, 500);
ck_assert(host == NULL);
}
END_TEST
START_TEST(test_create_from_chunk_other)
{
host_t *host;
host = host_create_from_chunk(AF_UNIX, addr_v6, 500);
ck_assert(host == NULL);
}
END_TEST
/*******************************************************************************
* host_create_from_subnet
*/
START_TEST(test_create_from_subnet_v4)
{
host_t *host;
int bits = -1;
host = host_create_from_subnet("0.0.0.0/0", &bits);
verify_any(host, AF_INET, 0);
ck_assert_int_eq(bits, 0);
host->destroy(host);
host = host_create_from_subnet("192.168.0.1", &bits);
verify_address(host, addr_v4, AF_INET, 0);
ck_assert_int_eq(bits, 32);
host->destroy(host);
host = host_create_from_subnet("192.168.0.1/24", &bits);
verify_address(host, addr_v4, AF_INET, 0);
ck_assert_int_eq(bits, 24);
host->destroy(host);
host = host_create_from_subnet("foo.b.a.r", &bits);
ck_assert(host == NULL);
}
END_TEST
START_TEST(test_create_from_subnet_v6)
{
host_t *host;
int bits = -1;
host = host_create_from_subnet("::/0", &bits);
verify_any(host, AF_INET6, 0);
ck_assert_int_eq(bits, 0);
host->destroy(host);
host = host_create_from_subnet("fec1::1", &bits);
verify_address(host, addr_v6, AF_INET6, 0);
ck_assert_int_eq(bits, 128);
host->destroy(host);
host = host_create_from_subnet("fec1::1/64", &bits);
verify_address(host, addr_v6, AF_INET6, 0);
ck_assert_int_eq(bits, 64);
host->destroy(host);
host = host_create_from_subnet("foo::bar", &bits);
ck_assert(host == NULL);
}
END_TEST
/*******************************************************************************
* host_create_netmask
*/
static void test_create_netmask(int family)
{
host_t *netmask;
@@ -80,6 +435,151 @@ START_TEST(test_create_netmask_other)
}
END_TEST
/*******************************************************************************
* equals, ip_equals
*/
START_TEST(test_equals)
{
host_t *a, *b;
a = host_create_from_string("192.168.0.1", 500);
b = host_create_from_string("192.168.0.1", 0);
ck_assert(!a->equals(a, b));
ck_assert(!b->equals(b, a));
ck_assert(a->ip_equals(a, b));
ck_assert(b->ip_equals(b, a));
b->set_port(b, 500);
ck_assert(a->equals(a, b));
ck_assert(b->equals(b, a));
ck_assert(a->ip_equals(a, b));
ck_assert(b->ip_equals(b, a));
b->destroy(b);
b = host_create_from_string("192.168.0.2", 500);
ck_assert(!a->ip_equals(a, b));
ck_assert(!a->equals(a, b));
b->destroy(b);
b = host_create_from_string("fec1::1", 500);
ck_assert(!a->ip_equals(a, b));
ck_assert(!a->equals(a, b));
a->destroy(a);
a = host_create_from_string("fec1::1", 500);
ck_assert(a->equals(a, b));
ck_assert(a->ip_equals(a, b));
a->destroy(a);
b->destroy(b);
}
END_TEST
START_TEST(test_equals_any)
{
host_t *a, *b;
a = host_create_from_string("%any", 500);
b = host_create_from_string("%any", 0);
ck_assert(!a->equals(a, b));
ck_assert(a->ip_equals(a, b));
b->set_port(b, 500);
ck_assert(a->equals(a, b));
ck_assert(a->ip_equals(a, b));
b->destroy(b);
b = host_create_from_string("%any6", 0);
ck_assert(a->ip_equals(a, b));
ck_assert(!a->equals(a, b));
b->set_port(b, 500);
ck_assert(a->ip_equals(a, b));
ck_assert(a->equals(a, b));
a->destroy(a);
b->destroy(b);
}
END_TEST
/*******************************************************************************
* clone
*/
START_TEST(test_clone)
{
host_t *a, *b;
a = host_create_from_string("192.168.0.1", 500);
b = a->clone(a);
ck_assert(a != b);
ck_assert(a->equals(a, b));
a->destroy(a);
b->destroy(b);
}
END_TEST
/*******************************************************************************
* printf hook
*/
static struct {
char *addr;
u_int16_t port;
/* results for %H, %+H, %#H (falls back to [0]) */
char *result[3];
} printf_data[] = {
{NULL, 0, { "(null)" }},
{NULL, 500, { "(null)" }},
{"%any", 0, { "%any", "0.0.0.0", "0.0.0.0[0]" }},
{"%any", 500, { "%any", "0.0.0.0", "0.0.0.0[500]" }},
{"%any6", 0, { "%any6", "::", "::[0]" }},
{"%any6", 500, { "%any6", "::", "::[500]" }},
{"192.168.0.1", 0, { "192.168.0.1", "192.168.0.1", "192.168.0.1[0]" }},
{"192.168.0.1", 500, { "192.168.0.1", "192.168.0.1", "192.168.0.1[500]" }},
{"fec1::1", 0, { "fec1::1", "fec1::1", "fec1::1[0]" }},
{"fec1::1", 500, { "fec1::1", "fec1::1", "fec1::1[500]" }},
};
static void verify_printf(host_t *host, const char *format, char *expected)
{
char buf[64];
snprintf(buf, sizeof(buf), format, host);
ck_assert_str_eq(expected, buf);
}
START_TEST(test_printf_hook)
{
static const char *formats[] = { "%H", "%+H", "%#H" };
host_t *host = NULL;
char *expected;
int i;
if (printf_data[_i].addr)
{
host = host_create_from_string(printf_data[_i].addr,
printf_data[_i].port);
}
for (i = 0; i < countof(formats); i++)
{
expected = printf_data[_i].result[i];
expected = expected ?: printf_data[_i].result[0];
verify_printf(host, formats[i], expected);
}
DESTROY_IF(host);
}
END_TEST
START_TEST(test_printf_hook_align)
{
host_t *host;
verify_printf(NULL, "%14H", " (null)");
verify_printf(NULL, "%-14H", "(null) ");
host = host_create_from_string("192.168.0.1", 0);
verify_printf(host, "%14H", " 192.168.0.1");
verify_printf(host, "%-14H", "192.168.0.1 ");
verify_printf(host, "%4H", "192.168.0.1");
verify_printf(host, "%-4H", "192.168.0.1");
host->destroy(host);
}
END_TEST
Suite *host_suite_create()
{
Suite *s;
@@ -87,11 +587,59 @@ Suite *host_suite_create()
s = suite_create("host");
tc = tcase_create("host_create_any");
tcase_add_test(tc, test_create_any_v4);
tcase_add_test(tc, test_create_any_v6);
tcase_add_test(tc, test_create_any_other);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_from_string");
tcase_add_test(tc, test_create_from_string_v4);
tcase_add_test(tc, test_create_from_string_any_v6);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_from_string_and_family");
tcase_add_test(tc, test_create_from_string_and_family_v4);
tcase_add_test(tc, test_create_from_string_and_family_v6);
tcase_add_test(tc, test_create_from_string_and_family_other);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_from_sockaddr");
tcase_add_test(tc, test_create_from_sockaddr_v4);
tcase_add_test(tc, test_create_from_sockaddr_v6);
tcase_add_test(tc, test_create_from_sockaddr_other);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_from_chunk");
tcase_add_test(tc, test_create_from_chunk_v4);
tcase_add_test(tc, test_create_from_chunk_v6);
tcase_add_test(tc, test_create_from_chunk_other);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_from_subnet");
tcase_add_test(tc, test_create_from_subnet_v4);
tcase_add_test(tc, test_create_from_subnet_v6);
suite_add_tcase(s, tc);
tc = tcase_create("host_create_netmask");
tcase_add_test(tc, test_create_netmask_v4);
tcase_add_test(tc, test_create_netmask_v6);
tcase_add_test(tc, test_create_netmask_other);
suite_add_tcase(s, tc);
tc = tcase_create("equals, ip_equals");
tcase_add_test(tc, test_equals);
tcase_add_test(tc, test_equals_any);
suite_add_tcase(s, tc);
tc = tcase_create("clone");
tcase_add_test(tc, test_clone);
suite_add_tcase(s, tc);
tc = tcase_create("printf hook");
tcase_add_loop_test(tc, test_printf_hook, 0, countof(printf_data));
tcase_add_test(tc, test_printf_hook_align);
suite_add_tcase(s, tc);
return s;
}