/*
* 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, unsigned be);
int _gnutls_ucs2_to_utf8(const void *data, size_t size,
gnutls_datum_t * output, unsigned be);
#define DEBUG
#ifdef DEBUG
static void PRINT(const char *str, unsigned char *val, unsigned int size)
{
unsigned i;
printf("%s", str);
for (i=0;i<size;i++) {
printf("%.2x", val[i]);
}
printf("\n");
}
#else
#define PRINT(x, y, x)
#endif
#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, 1); \
assert_int_equal(ret, 0); \
if (out.size != sizeof(utf16)-1 || memcmp(utf16, out.data, out.size) != 0) { PRINT("got: ", out.data, out.size); \
PRINT("expected: ", (unsigned char*)utf16, sizeof(utf16)-1); } \
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); \
if (out.size != strlen(utf8) || memcmp(utf8, out.data, out.size) != 0) { PRINT("got: ", out.data, out.size); \
PRINT("expected: ", (unsigned char*)utf8, strlen(utf8)); } \
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, 1); \
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);
}