4 files changed, 100 insertions, 1 deletions
diff --git a/src/libstrongswan/tests/Makefile.am b/src/libstrongswan/tests/Makefile.am
index f175aa4a7..585f9c16e 100644
--- a/src/libstrongswan/tests/Makefile.am
+++ b/src/libstrongswan/tests/Makefile.am
@@ -7,7 +7,7 @@ test_runner_SOURCES = \
   test_linked_list.c test_enumerator.c test_linked_list_enumerator.c \
   test_bio_reader.c test_bio_writer.c test_chunk.c test_enum.c test_hashtable.c \
   test_identification.c test_threading.c test_utils.c test_vectors.c \
-  test_array.c test_ecdsa.c test_rsa.c
+  test_array.c test_ecdsa.c test_rsa.c test_host.c
 
 test_runner_CFLAGS = \
   -I$(top_srcdir)/src/libstrongswan \
diff --git a/src/libstrongswan/tests/test_host.c b/src/libstrongswan/tests/test_host.c
new file mode 100644
index 000000000..bcc0301d5
--- /dev/null
+++ b/src/libstrongswan/tests/test_host.c
@@ -0,0 +1,97 @@
+/*
+ * Copyright (C) 2013 Tobias Brunner
+ * Hochschule fuer Technik Rapperswil
+ *
+ * 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 <networking/host.h>
+
+/*******************************************************************************
+ * host_create_netmask
+ */
+
+static void verify_netmask(chunk_t addr, int mask)
+{
+	int byte, bit;
+
+	for (byte = 0; byte < addr.len; byte++)
+	{
+		for (bit = 7; bit >= 0; bit--)
+		{
+			int val = (addr.ptr[byte] >> bit) & 0x01;
+			if (mask-- > 0)
+			{
+				ck_assert_int_eq(val, 1);
+			}
+			else
+			{
+				ck_assert_int_eq(val, 0);
+			}
+		}
+	}
+}
+
+static void test_create_netmask(int family)
+{
+	host_t *netmask;
+	int i, len = (family == AF_INET) ? 32 : 128;
+
+	netmask = host_create_netmask(family, -1);
+	ck_assert(netmask == NULL);
+	for (i = 0; i <= len; i++)
+	{
+		netmask = host_create_netmask(family, i);
+		verify_netmask(netmask->get_address(netmask), i);
+		netmask->destroy(netmask);
+	}
+	netmask = host_create_netmask(family, len + 1);
+	ck_assert(netmask == NULL);
+}
+
+START_TEST(test_create_netmask_v4)
+{
+	test_create_netmask(AF_INET);
+}
+END_TEST
+
+START_TEST(test_create_netmask_v6)
+{
+	test_create_netmask(AF_INET6);
+}
+END_TEST
+
+START_TEST(test_create_netmask_other)
+{
+	host_t *netmask;
+
+	netmask = host_create_netmask(AF_UNSPEC, 0);
+	ck_assert(netmask == NULL);
+}
+END_TEST
+
+Suite *host_suite_create()
+{
+	Suite *s;
+	TCase *tc;
+
+	s = suite_create("host");
+
+	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);
+
+	return s;
+}
diff --git a/src/libstrongswan/tests/test_runner.c b/src/libstrongswan/tests/test_runner.c
index b9e1901b0..e7a04fd9a 100644
--- a/src/libstrongswan/tests/test_runner.c
+++ b/src/libstrongswan/tests/test_runner.c
@@ -82,6 +82,7 @@ int main()
 	srunner_add_suite(sr, identification_suite_create());
 	srunner_add_suite(sr, threading_suite_create());
 	srunner_add_suite(sr, utils_suite_create());
+	srunner_add_suite(sr, host_suite_create());
 	srunner_add_suite(sr, vectors_suite_create());
 	if (lib->plugins->has_feature(lib->plugins,
 								  PLUGIN_DEPENDS(PRIVKEY_GEN, KEY_RSA)))
diff --git a/src/libstrongswan/tests/test_runner.h b/src/libstrongswan/tests/test_runner.h
index 34aa0cf18..e9381756c 100644
--- a/src/libstrongswan/tests/test_runner.h
+++ b/src/libstrongswan/tests/test_runner.h
@@ -33,5 +33,6 @@ Suite *utils_suite_create();
 Suite *vectors_suite_create();
 Suite *ecdsa_suite_create();
 Suite *rsa_suite_create();
+Suite *host_suite_create();
 
 #endif /** TEST_RUNNER_H_ */