diff options
| -rw-r--r-- | tests/base64-raw.c | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/tests/base64-raw.c b/tests/base64-raw.c index 9bfc59e03a..7ad8bf65f3 100644 --- a/tests/base64-raw.c +++ b/tests/base64-raw.c @@ -59,6 +59,24 @@ static void encode(const char *test_name, const gnutls_datum_t *raw, const char gnutls_free(out.data); + ret = gnutls_base64_encode2(raw, &out); + if (ret < 0) { + fail("%s: gnutls_base64_encode2: %s\n", test_name, gnutls_strerror(ret)); + exit(1); + } + + if (strlen(expected)!=out.size) { + fail("%s: gnutls_base64_encode2: output has incorrect size (%d, expected %d)\n", test_name, (int)out.size, (int)strlen(expected)); + exit(1); + } + + if (strncasecmp(expected, (char*)out.data, out.size) != 0) { + fail("%s: gnutls_base64_encode2: output does not match the expected\n", test_name); + exit(1); + } + + gnutls_free(out.data); + in.data = (void*)expected; in.size = strlen(expected); ret = gnutls_pem_base64_decode2("", &in, &out); @@ -82,6 +100,70 @@ static void encode(const char *test_name, const gnutls_datum_t *raw, const char return; } +static void encode_new(const char *test_name, const gnutls_datum_t *raw, const char *expected) +{ + int ret; + gnutls_datum_t out, in; + + ret = gnutls_base64_encode2(raw, &out); + if (ret < 0) { + fail("%s: gnutls_base64_encode2: %s\n", test_name, gnutls_strerror(ret)); + exit(1); + } + + if (strlen(expected)!=out.size) { + fail("%s: gnutls_base64_encode2: output has incorrect size (%d, expected %d)\n", test_name, (int)out.size, (int)strlen(expected)); + exit(1); + } + + if (strncasecmp(expected, (char*)out.data, out.size) != 0) { + fail("%s: gnutls_base64_encode2: output does not match the expected\n", test_name); + exit(1); + } + + gnutls_free(out.data); + + ret = gnutls_base64_encode2(raw, &out); + if (ret < 0) { + fail("%s: gnutls_base64_encode2: %s\n", test_name, gnutls_strerror(ret)); + exit(1); + } + + if (strlen(expected)!=out.size) { + fail("%s: gnutls_base64_encode2: output has incorrect size (%d, expected %d)\n", test_name, (int)out.size, (int)strlen(expected)); + exit(1); + } + + if (strncasecmp(expected, (char*)out.data, out.size) != 0) { + fail("%s: gnutls_base64_encode2: output does not match the expected\n", test_name); + exit(1); + } + + gnutls_free(out.data); + + in.data = (void*)expected; + in.size = strlen(expected); + ret = gnutls_base64_decode2(&in, &out); + if (ret < 0) { + fail("%s: gnutls_base64_decode2: %s\n", test_name, gnutls_strerror(ret)); + exit(1); + } + + if (raw->size!=out.size) { + fail("%s: gnutls_base64_decode2: output has incorrect size (%d, expected %d)\n", test_name, out.size, raw->size); + exit(1); + } + + if (memcmp(raw->data, out.data, out.size) != 0) { + fail("%s: gnutls_base64_decode2: output does not match the expected\n", test_name); + exit(1); + } + + gnutls_free(out.data); + + return; +} + static void decode(const char *test_name, const gnutls_datum_t *raw, const char *hex, int res) { int ret; @@ -117,6 +199,41 @@ static void decode(const char *test_name, const gnutls_datum_t *raw, const char return; } +static void decode_new(const char *test_name, const gnutls_datum_t *raw, const char *hex, int res) +{ + int ret; + gnutls_datum_t out, in; + + in.data = (void*)hex; + in.size = strlen(hex); + ret = gnutls_base64_decode2(&in, &out); + if (ret < 0) { + if (res == ret) /* expected */ + return; + fail("%s: gnutls_base64_decode2: %d/%s\n", test_name, ret, gnutls_strerror(ret)); + exit(1); + } + + if (res != 0) { + fail("%s: gnutls_base64_decode2: expected failure, but succeeded!\n", test_name); + exit(1); + } + + if (raw->size!=out.size) { + fail("%s: gnutls_base64_decode2: output has incorrect size (%d, expected %d)\n", test_name, out.size, raw->size); + exit(1); + } + + if (memcmp(raw->data, out.data, out.size) != 0) { + fail("%s: gnutls_base64_decode2: output does not match the expected\n", test_name); + exit(1); + } + + gnutls_free(out.data); + + return; +} + struct encode_tests_st { const char *name; gnutls_datum_t raw; @@ -169,6 +286,12 @@ struct decode_tests_st decode_tests[] = { .res = GNUTLS_E_BASE64_DECODING_ERROR }, { + .name = "dec-empty", + .pem = "", + .raw = {(void*)"", 1}, + .res = GNUTLS_E_BASE64_DECODING_ERROR + }, + { .name = "dec-invalid-suffix", .pem = "LJ/7hUZ3TtPIz2dlc5+YvELe+Q==XXX", .raw = {(void*)"\x2c\x9f\xfb\x85\x46\x77\x4e\xd3\xc8\xcf\x67\x65\x73\x9f\x98\xbc\x42\xde\xf9", 19}, @@ -182,10 +305,12 @@ void doit(void) for (i=0;i<sizeof(encode_tests)/sizeof(encode_tests[0]);i++) { encode(encode_tests[i].name, &encode_tests[i].raw, encode_tests[i].pem); + encode_new(encode_tests[i].name, &encode_tests[i].raw, encode_tests[i].pem); } for (i=0;i<sizeof(decode_tests)/sizeof(decode_tests[0]);i++) { decode(decode_tests[i].name, &decode_tests[i].raw, decode_tests[i].pem, decode_tests[i].res); + decode_new(decode_tests[i].name, &decode_tests[i].raw, decode_tests[i].pem, decode_tests[i].res); } } |