- Add base64 decode unit test

diff --git a/testcode/unitldns.c b/testcode/unitldns.c
index 66f75617037de92473463017fab99019643b80a8..22c9ed9455856f003deb77905580793c07138346 100644 (file)
--- a/testcode/unitldns.c
+++ b/testcode/unitldns.c
@@ -44,6 +44,7 @@
 #include "sldns/sbuffer.h"
 #include "sldns/str2wire.h"
 #include "sldns/wire2str.h"
+#include "sldns/parseutil.h"
 
 /** verbose this unit test */
 static int vbmp = 0; 
@@ -220,9 +221,60 @@ rr_tests(void)
                SRCDIRSTR "/testdata/test_ldnsrr.c5");
 }
 
+/** test various base64 decoding options */
+static void
+b64_test(void)
+{
+       /* "normal" b64 alphabet, with padding */
+       char* p1 = "aGVsbG8="; /* "hello" */
+       char* p2 = "aGVsbG8+"; /* "hello>" */
+       char* p3 = "aGVsbG8/IQ=="; /* "hello?!" */
+       char* p4 = "aGVsbG8"; /* "hel" + extra garbage */
+
+       /* base64 url, without padding */
+       char* u1 = "aGVsbG8"; /* "hello" */
+       char* u2 = "aGVsbG8-"; /* "hello>" */
+       char* u3 = "aGVsbG8_IQ"; /* "hello?!" */
+       char* u4 = "aaaaa"; /* garbage */
+
+       char target[128];
+       size_t tarsize = 128;
+       int result;
+
+       memset(target, 0, sizeof(target));
+       result = sldns_b64_pton(p1, (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello") && strcmp(target, "hello") == 0);
+       memset(target, 0, sizeof(target));
+       result = sldns_b64_pton(p2, (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello>") && strcmp(target, "hello>") == 0);
+       memset(target, 0, sizeof(target));
+       result = sldns_b64_pton(p3, (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello?!") && strcmp(target, "hello?!") == 0);
+       memset(target, 0, sizeof(target));
+       result = sldns_b64_pton(p4, (uint8_t*)target, tarsize);
+       /* when padding is used everything that is not a block of 4 will be
+        * ignored */
+       unit_assert(result == strlen("hel") && strcmp(target, "hel") == 0);
+
+       memset(target, 0, sizeof(target));
+       result = sldns_b64url_pton(u1, strlen(u1), (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello") && strcmp(target, "hello") == 0);
+       memset(target, 0, sizeof(target));
+       result = sldns_b64url_pton(u2, strlen(u2), (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello>") && strcmp(target, "hello>") == 0);
+       memset(target, 0, sizeof(target));
+       result = sldns_b64url_pton(u3, strlen(u3), (uint8_t*)target, tarsize);
+       unit_assert(result == strlen("hello+/") && strcmp(target, "hello?!") == 0);
+       /* one item in block of four is not allowed */
+       memset(target, 0, sizeof(target));
+       result = sldns_b64url_pton(u4, strlen(u4), (uint8_t*)target, tarsize);
+       unit_assert(result == -1);
+}
+
 void
 ldns_test(void)
 {
        unit_show_feature("sldns");
        rr_tests();
+       b64_test();
 }