1 files changed, 50 insertions, 0 deletions
diff --git a/tests/dns.c b/tests/dns.c
new file mode 100644
index 0000000..c100cd4
--- /dev/null
+++ b/tests/dns.c
@@ -0,0 +1,50 @@
+/*
+ * tests/dns.c
+ * (c) 2021 Jonas Gunz <himself@jonasgunz.de>
+ * License: MIT
+ */
+
+#include "tests.h"
+
+#include <dns.h>
+
+START_TEST (dns_qname) {
+	char in[128];
+	char out[128];
+
+	strncpy ( in, "sub.domain.example.com\0", 127);
+
+	ck_assert_int_gt( fqdn_to_qname (in,128,out,128), 0 );
+	ck_assert_int_ge( qname_check(out,128), 0 );
+	ck_assert_int_gt( qname_to_fqdn (out,128,in,128), 0);
+} END_TEST
+
+START_TEST (dns_qname_fuzz) {
+	FILE* urand = fopen ("/dev/urandom", "r");
+	char rand[128];
+	const unsigned long int limit = 1000000;
+	unsigned long int valid_cnt = 0;
+	unsigned long int i;
+
+	if ( !urand )
+		ck_abort_msg("Failed to open /dev/urandom");
+
+	for ( i = 0; i < limit; i++) {
+		if (fread (rand, 128, 1, urand) > 0) {
+			if ( qname_check(rand, 128) > 0 ) {
+				valid_cnt++;
+			}
+		}
+	}
+
+	ck_assert_float_le( (float)valid_cnt / (float)limit * 100, 10);
+}
+
+TCase* test_dns(void) {
+	TCase *tc = tcase_create("DNS");
+
+	tcase_add_test(tc, dns_qname);
+	tcase_add_test(tc, dns_qname_fuzz);
+
+	return tc;
+}