4 files changed, 181 insertions, 0 deletions
diff --git a/test/test-endswith.c b/test/test-endswith.c
new file mode 100644
index 0000000..7ad61a7
--- /dev/null
+++ b/test/test-endswith.c
@@ -0,0 +1,17 @@
+#include <stdio.h>
+#include "string_common.h"
+
+int main() {
+	int r, result1, result2;
+	char *foo = "this is a string";
+	char *bar = "string";
+
+	result1 = endswith(foo, bar);
+	result2 = endswith("string", "asdf");
+
+	r = (result1 == 1) && (result2 == 0);
+
+	printf("%s: endswith\n", r ? "PASS" : "FAIL");
+
+	return r;
+}
diff --git a/test/test-parse_dns_udp-multi.c b/test/test-parse_dns_udp-multi.c
new file mode 100644
index 0000000..854b61a
--- /dev/null
+++ b/test/test-parse_dns_udp-multi.c
@@ -0,0 +1,47 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdint.h>
+#include "dns.h"
+
+int main() {
+	uint8_t pkt[512] = {0};
+
+	// DNS header
+	pkt[4] = 0x00;
+	pkt[5] = 0x02; // QDCOUNT = 2
+
+	uint8_t *ptr = pkt + 12;
+
+	// Question 1: foo.com A
+	*ptr++ = 3; memcpy(ptr, "foo", 3); ptr += 3;
+	*ptr++ = 3; memcpy(ptr, "com", 3); ptr += 3;
+	*ptr++ = 0; // null terminator
+	*ptr++ = 0x00; *ptr++ = 0x01; // QTYPE A
+	*ptr++ = 0x00; *ptr++ = 0x01; // QCLASS IN
+
+	// Question 2: bar.org AAAA
+	*ptr++ = 3; memcpy(ptr, "bar", 3); ptr += 3;
+	*ptr++ = 3; memcpy(ptr, "org", 3); ptr += 3;
+	*ptr++ = 0;
+	*ptr++ = 0x00; *ptr++ = 0x1c; // QTYPE AAAA
+	*ptr++ = 0x00; *ptr++ = 0x01;
+
+	size_t pkt_len = ptr - pkt;
+
+	struct dns_question out[4];
+	size_t count = parse_dns_udp(pkt, pkt_len, out, 4);
+
+	//printf("Got %zu questions:\n", count);
+	//for (size_t i = 0; i < count; i++) {
+	//	printf("  %s (%s)\n", out[i].name, dns_type_to_string(out[i].qtype));
+	//}
+
+	int res = (count == 2 &&
+			   strcmp(out[0].name, "foo.com") == 0 &&
+			   out[0].qtype == 1 &&
+			   strcmp(out[1].name, "bar.org") == 0 &&
+			   out[1].qtype == 28);
+
+	printf("%s: parse_dns_udp-multi\n", res ? "PASS" : "FAIL");
+	return res;
+}
diff --git a/test/test-parse_dns_udp.c b/test/test-parse_dns_udp.c
new file mode 100644
index 0000000..7fec577
--- /dev/null
+++ b/test/test-parse_dns_udp.c
@@ -0,0 +1,42 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include "../include/dns.h"
+
+#define ASSERT(x) do { \
+    if (!(x)) { \
+        fprintf(stderr, "FAIL: %s:%d: %s\n", __FILE__, __LINE__, #x); \
+        return EXIT_FAILURE; \
+    } \
+} while (0)
+
+int main(void) {
+    // Simple DNS query for "example.com" A record
+    // Manually constructed:
+    // - Header: 12 bytes
+    // - Question: example.com (as labels) + QTYPE A + QCLASS IN
+    uint8_t pkt[] = {
+        0x12, 0x34, // Transaction ID
+        0x01, 0x00, // Flags: standard query
+        0x00, 0x01, // QDCOUNT: 1
+        0x00, 0x00, // ANCOUNT
+        0x00, 0x00, // NSCOUNT
+        0x00, 0x00, // ARCOUNT
+        0x07, 'e','x','a','m','p','l','e',
+        0x03, 'c','o','m',
+        0x00,             // null terminator
+        0x00, 0x01,       // QTYPE A
+        0x00, 0x01        // QCLASS IN
+    };
+
+    struct dns_question qs[MAX_DNS_QUESTIONS];
+    size_t count = parse_dns_udp(pkt, sizeof(pkt), qs, MAX_DNS_QUESTIONS);
+
+    ASSERT(count == 1);
+    ASSERT(strcmp(qs[0].name, "example.com") == 0);
+    ASSERT(qs[0].qtype == 1);
+
+    printf("PASS: parse_dns_udp\n");
+    return EXIT_SUCCESS;
+}
diff --git a/test/test-sha256.c b/test/test-sha256.c
new file mode 100644
index 0000000..a00d22c
--- /dev/null
+++ b/test/test-sha256.c
@@ -0,0 +1,75 @@
+/*********************************************************************
+* Filename:   sha256.c
+* Author:     Brad Conte (brad AT bradconte.com)
+* Copyright:
+* Disclaimer: This code is presented "as is" without any guarantees.
+* Details:    Performs known-answer tests on the corresponding SHA1
+	          implementation. These tests do not encompass the full
+	          range of available test vectors, however, if the tests
+	          pass it is very, very likely that the code is correct
+	          and was compiled properly. This code also serves as
+	          example usage of the functions.
+*********************************************************************/
+
+/*************************** HEADER FILES ***************************/
+#include <stdio.h>
+#include <memory.h>
+#include <string.h>
+#include "sha256.h"
+
+/*********************** FUNCTION DEFINITIONS ***********************/
+int sha256_test()
+{
+	BYTE text1[] = {"abc"};
+	BYTE text2[] = {"abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"};
+	BYTE text3[] = {"aaaaaaaaaa"};
+	BYTE hash1[SHA256_BLOCK_SIZE] = {
+		0xba, 0x78, 0x16, 0xbf, 0x8f, 0x01, 0xcf, 0xea,
+		0x41, 0x41, 0x40, 0xde, 0x5d, 0xae, 0x22, 0x23,
+		0xb0, 0x03, 0x61, 0xa3, 0x96, 0x17, 0x7a, 0x9c,
+		0xb4, 0x10, 0xff, 0x61, 0xf2, 0x00, 0x15, 0xad
+	};
+	BYTE hash2[SHA256_BLOCK_SIZE] = {
+		0x24, 0x8d, 0x6a, 0x61, 0xd2, 0x06, 0x38, 0xb8,
+		0xe5, 0xc0, 0x26, 0x93, 0x0c, 0x3e, 0x60, 0x39,
+		0xa3, 0x3c, 0xe4, 0x59, 0x64, 0xff, 0x21, 0x67,
+		0xf6, 0xec, 0xed, 0xd4, 0x19, 0xdb, 0x06, 0xc1
+	};
+	BYTE hash3[SHA256_BLOCK_SIZE] = {
+		0xcd, 0xc7, 0x6e, 0x5c, 0x99, 0x14, 0xfb, 0x92,
+		0x81, 0xa1, 0xc7, 0xe2, 0x84, 0xd7, 0x3e, 0x67,
+		0xf1, 0x80, 0x9a, 0x48, 0xa4, 0x97, 0x20, 0x0e,
+		0x04, 0x6d, 0x39, 0xcc, 0xc7, 0x11, 0x2c, 0xd0
+	};
+	BYTE buf[SHA256_BLOCK_SIZE];
+	SHA256_CTX ctx;
+	int idx;
+	int pass = 1;
+
+	sha256_init(&ctx);
+	sha256_update(&ctx, text1, strlen((const char *)text1));
+	sha256_final(&ctx, buf);
+	pass = pass && !memcmp(hash1, buf, SHA256_BLOCK_SIZE);
+
+	sha256_init(&ctx);
+	sha256_update(&ctx, text2, strlen((const char *)text2));
+	sha256_final(&ctx, buf);
+	pass = pass && !memcmp(hash2, buf, SHA256_BLOCK_SIZE);
+
+	sha256_init(&ctx);
+	for (idx = 0; idx < 100000; ++idx) {
+		sha256_update(&ctx, text3, strlen((const char *)text3));
+	}
+	sha256_final(&ctx, buf);
+	pass = pass && !memcmp(hash3, buf, SHA256_BLOCK_SIZE);
+
+	return pass;
+}
+
+int main()
+{
+	int result = sha256_test();
+	printf("%s: SHA-256\n", result ? "PASS" : "FAIL");
+
+	return result ? 0 : 1;
+}