1 files changed, 106 insertions, 0 deletions
diff --git a/src/supplemental/base64/base64_test.c b/src/supplemental/base64/base64_test.c
new file mode 100644
index 00000000..2c79a243
--- /dev/null
+++ b/src/supplemental/base64/base64_test.c
@@ -0,0 +1,106 @@
+//
+// Copyright 2019 Staysail Systems, Inc. <info@staysail.tech>
+// Copyright 2018 Capitar IT Group BV <info@capitar.com>
+//
+// This software is supplied under the terms of the MIT License, a
+// copy of which should be located in the distribution where this
+// file was obtained (LICENSE.txt).  A copy of the license may also be
+// found online at https://opensource.org/licenses/MIT.
+//
+
+#include <string.h>
+
+#include <nng/nng.h>
+
+#include "base64.h"
+
+#include <acutest.h>
+
+typedef struct {
+	char *decoded;
+	char *encoded;
+} test_case;
+
+static test_case cases[] = {
+	{ "", "" },
+	{ "f", "Zg==" },
+	{ "fo", "Zm8=" },
+	{ "foo", "Zm9v" },
+	{ "foob", "Zm9vYg==" },
+	{ "fooba", "Zm9vYmE=" },
+	{ "foobar", "Zm9vYmFy" },
+	{ NULL, NULL },
+};
+
+void
+test_encode(void)
+{
+	int   i;
+	void *dec;
+
+	for (i = 0; (dec = cases[i].decoded) != NULL; i++) {
+		char buf[1024];
+		char name[8];
+		int  rv;
+
+		(void) snprintf(name, sizeof(name), "%d", i);
+		TEST_CASE(name);
+		rv = nni_base64_encode(dec, strlen(dec), buf, 1024);
+		TEST_CHECK(rv >= 0);
+		TEST_CHECK(rv == (int) strlen(cases[i].encoded));
+		buf[rv] = 0;
+		TEST_CHECK(strcmp(buf, cases[i].encoded) == 0);
+	}
+}
+
+void
+test_decode(void)
+{
+	int   i;
+	void *enc;
+
+	for (i = 0; (enc = cases[i].encoded) != NULL; i++) {
+		char buf[1024];
+		char name[8];
+		int  rv;
+
+		(void) snprintf(name, sizeof(name), "%d", i);
+		TEST_CASE(name);
+
+		rv = nni_base64_decode(enc, strlen(enc), (void *) buf, 1024);
+		TEST_CHECK(rv >= 0);
+		TEST_CHECK(rv == (int) strlen(cases[i].decoded));
+		buf[rv] = 0;
+		TEST_CHECK(strcmp(buf, cases[i].decoded) == 0);
+	}
+}
+
+void
+test_overflow(void)
+{
+	char tmp[1024];
+	for (int i = 1; cases[i].decoded != NULL; i++) {
+		void *enc = cases[i].encoded;
+		void *dec = cases[i].decoded;
+		void *buf = tmp;
+
+		char name[8];
+		(void) snprintf(name, sizeof(name), "%d", i);
+		TEST_CASE(name);
+
+		TEST_CHECK(nni_base64_encode(
+		               dec, strlen(dec), buf, strlen(enc) - 1) == -1);
+		TEST_CHECK(nni_base64_encode(dec, strlen(dec), buf, 0) == -1);
+
+		TEST_CHECK(nni_base64_decode(
+		               enc, strlen(enc), buf, strlen(dec) - 1) == -1);
+		TEST_CHECK(nni_base64_encode(enc, strlen(enc), buf, 0) == -1);
+	}
+}
+
+TEST_LIST = {
+	{ "encode", test_encode },
+	{ "decode", test_decode },
+	{ "overflow", test_overflow },
+	{ NULL, NULL },
+};