tests: added unit tests of _gnutls_utf8_to_ucs2 and _gnutls_ucs2_to_utf8

2 files changed, 113 insertions, 1 deletions

diff --git a/tests/Makefile.am b/tests/Makefile.am
index 9aef95b585..daf9d017bf 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -117,7 +117,7 @@ endif
 
 if HAVE_CMOCKA
 LDADD += $(CMOCKA_LIBS)
-ctests += dtls-sliding-window ip-utils name-constraints-ip
+ctests += dtls-sliding-window ip-utils name-constraints-ip conv-utf8
 endif
 
 mini_dtls_pthread_LDADD = $(LDADD) -lpthread
diff --git a/tests/conv-utf8.c b/tests/conv-utf8.c
new file mode 100644
index 0000000000..9bb063dd70
--- /dev/null
+++ b/tests/conv-utf8.c
@@ -0,0 +1,112 @@
+/*
+ * Copyright (C) 2016 Red Hat, Inc.
+ *
+ * Authors: Nikos Mavrogiannopoulos
+ *
+ * This file is part of GnuTLS.
+ *
+ * GnuTLS 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GnuTLS 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.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GnuTLS; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <config.h>
+#include <stdarg.h>
+#include <stddef.h>
+#include <setjmp.h>
+#include <limits.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+#include <gnutls/gnutls.h>
+#include <gnutls/x509.h>
+
+#include <cmocka.h>
+
+int _gnutls_utf8_to_ucs2(const void *data, size_t size,
+			 gnutls_datum_t * output);
+
+int _gnutls_ucs2_to_utf8(const void *data, size_t size,
+                         gnutls_datum_t * output, unsigned be);
+
+#define UTF8_MATCH(fname, utf8, utf16) \
+static void fname(void **glob_state) \
+{ \
+	gnutls_datum_t out; \
+	int ret = _gnutls_utf8_to_ucs2(utf8, strlen(utf8), &out); \
+	assert_int_equal(ret, 0); \
+	assert_int_equal(out.size, sizeof(utf16)-1); \
+	assert_memory_equal(utf16, out.data, out.size); \
+	gnutls_free(out.data); \
+}
+
+#define UTF16_MATCH(fname, utf8, utf16) \
+static void fname(void **glob_state) \
+{ \
+	gnutls_datum_t out; \
+	int ret = _gnutls_ucs2_to_utf8(utf16, sizeof(utf16)-1, &out, 1); \
+	assert_int_equal(ret, 0); \
+	assert_int_equal(out.size, strlen(utf8)); \
+	assert_memory_equal(utf8, out.data, out.size); \
+	gnutls_free(out.data); \
+}
+
+#define UTF8_FAIL(fname, utf8, utf8_size) \
+static void fname(void **glob_state) \
+{ \
+	gnutls_datum_t out; \
+	int ret = _gnutls_utf8_to_ucs2(utf8, utf8_size, &out); \
+	assert_int_not_equal(ret, 0); \
+}
+
+#define UTF16_FAIL(fname, utf16, utf16_size) \
+static void fname(void **glob_state) \
+{ \
+	gnutls_datum_t out; \
+	int ret = _gnutls_ucs2_to_utf8(utf16, utf16_size, &out, 1); \
+	assert_int_not_equal(ret, 0); \
+}
+
+UTF8_MATCH(check_utf8_ok1, "abcd", "\x00\x61\x00\x62\x00\x63\x00\x64");
+UTF8_MATCH(check_utf8_ok2, "ユーザー別サイト", "\x30\xE6\x30\xFC\x30\xB6\x30\xFC\x52\x25\x30\xB5\x30\xA4\x30\xC8");
+UTF8_MATCH(check_utf8_ok3, "简体中文", "\x7B\x80\x4F\x53\x4E\x2D\x65\x87");
+UTF8_MATCH(check_utf8_ok4, "Σὲ γνωρίζω ἀπὸ", "\x03\xA3\x1F\x72\x00\x20\x03\xB3\x03\xBD\x03\xC9\x03\xC1\x03\xAF\x03\xB6\x03\xC9\x00\x20\x1F\x00\x03\xC0\x1F\x78");
+
+UTF16_MATCH(check_utf16_ok1, "abcd", "\x00\x61\x00\x62\x00\x63\x00\x64");
+UTF16_MATCH(check_utf16_ok2, "ユーザー別サイト", "\x30\xE6\x30\xFC\x30\xB6\x30\xFC\x52\x25\x30\xB5\x30\xA4\x30\xC8");
+UTF16_MATCH(check_utf16_ok3, "简体中文", "\x7B\x80\x4F\x53\x4E\x2D\x65\x87");
+UTF16_MATCH(check_utf16_ok4, "Σὲ γνωρίζω ἀπὸ", "\x03\xA3\x1F\x72\x00\x20\x03\xB3\x03\xBD\x03\xC9\x03\xC1\x03\xAF\x03\xB6\x03\xC9\x00\x20\x1F\x00\x03\xC0\x1F\x78");
+
+UTF8_FAIL(check_utf8_fail1, "\xfe\xff\xaa\x80\xff", 5);
+UTF8_FAIL(check_utf8_fail2, "\x64\x00\x62\xf3\x64\x65", 6);
+
+UTF16_FAIL(check_utf16_fail1, "\xd8\x00\xdb\xff\x00\x63\x00\x04", 8);
+
+int main(void)
+{
+	const struct CMUnitTest tests[] = {
+		cmocka_unit_test(check_utf8_fail1),
+		cmocka_unit_test(check_utf8_fail2),
+		cmocka_unit_test(check_utf16_fail1),
+		cmocka_unit_test(check_utf8_ok1),
+		cmocka_unit_test(check_utf8_ok2),
+		cmocka_unit_test(check_utf8_ok3),
+		cmocka_unit_test(check_utf8_ok4),
+		cmocka_unit_test(check_utf16_ok1),
+		cmocka_unit_test(check_utf16_ok2),
+		cmocka_unit_test(check_utf16_ok3),
+		cmocka_unit_test(check_utf16_ok4)
+	};
+	return cmocka_run_group_tests(tests, NULL, NULL);
+}