179 lines
8.2 KiB
C
179 lines
8.2 KiB
C
|
/* LibTomCrypt, modular cryptographic library -- Tom St Denis */
|
||
|
|
/* SPDX-License-Identifier: Unlicense */
|
||
|
|
#include <tomcrypt_test.h>
|
||
|
|
|
||
|
|
#if defined(LTC_BASE64) || defined(LTC_BASE64_URL)
|
||
|
|
enum { insane = 0, strict = 1, relaxed = 2, invalid = 666 };
|
||
|
|
|
||
|
|
int base64_test(void)
|
||
|
|
{
|
||
|
|
unsigned char in[64], tmp[64];
|
||
|
|
char out[256];
|
||
|
|
unsigned long x, l1, l2, slen1;
|
||
|
|
|
||
|
|
const unsigned char special_case[] = {
|
||
|
|
0xbe, 0xe8, 0x92, 0x3c, 0xa2, 0x25, 0xf0, 0xf8,
|
||
|
|
0x91, 0xe4, 0xef, 0xab, 0x0b, 0x8c, 0xfd, 0xff,
|
||
|
|
0x14, 0xd0, 0x29, 0x9d, 0x00 };
|
||
|
|
/* 3 A's are encoded as QUFB */
|
||
|
|
const char *As_lf = "QUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFB\nQUFB";
|
||
|
|
const char *As_crlf = "QUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFB\r\nQUFB";
|
||
|
|
const char *As_ssh_lf = "QUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFB\nQUFB";
|
||
|
|
const char *As_ssh_crlf = "QUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFB\r\nQUFB";
|
||
|
|
|
||
|
|
#if defined(LTC_BASE64)
|
||
|
|
/*
|
||
|
|
TEST CASES SOURCE:
|
||
|
|
|
||
|
|
Network Working Group S. Josefsson
|
||
|
|
Request for Comments: 4648 SJD
|
||
|
|
Obsoletes: 3548 October 2006
|
||
|
|
Category: Standards Track
|
||
|
|
*/
|
||
|
|
const struct {
|
||
|
|
const char* s;
|
||
|
|
const char* b64;
|
||
|
|
} cases[] = {
|
||
|
|
{"", "" },
|
||
|
|
{"f", "Zg==" },
|
||
|
|
{"fo", "Zm8=" },
|
||
|
|
{"foo", "Zm9v" },
|
||
|
|
{"foob", "Zm9vYg==" },
|
||
|
|
{"fooba", "Zm9vYmE=" },
|
||
|
|
{"foobar", "Zm9vYmFy"},
|
||
|
|
{(char*)special_case,"vuiSPKIl8PiR5O+rC4z9/xTQKZ0="}
|
||
|
|
};
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#ifdef LTC_BASE64_URL
|
||
|
|
const struct {
|
||
|
|
const char* s;
|
||
|
|
int flag;
|
||
|
|
} url_cases[] = {
|
||
|
|
{"vuiSPKIl8PiR5O-rC4z9_xTQKZ0", strict}, /* 0 */
|
||
|
|
{"vuiSPKIl8PiR5O-rC4z9_xTQKZ0=", strict},
|
||
|
|
{"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0", insane},
|
||
|
|
{"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0=", insane},
|
||
|
|
{"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0==", insane},
|
||
|
|
{"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0===", insane}, /* 5 */
|
||
|
|
{"vuiS*PKIl8P*iR5O-rC4*z9_xTQKZ0====", insane},
|
||
|
|
{"vuiS*=PKIl8P*iR5O-rC4*z9_xTQKZ0=", insane},
|
||
|
|
{"vuiS*==PKIl8P*iR5O-rC4*z9_xTQKZ0=", insane},
|
||
|
|
{"vuiS*==\xffPKIl8P*iR5O-rC4*z9_xTQKZ0=", insane},
|
||
|
|
{"vuiS PKIl8P\niR5O-rC4\tz9_xTQKZ0", relaxed}, /* 10 */
|
||
|
|
{"vuiS PKIl8P\niR5O-rC4\tz9_xTQKZ0=", relaxed},
|
||
|
|
{"vuiS PKIl8P\niR5O-rC4\tz9_xTQKZ0==", relaxed},
|
||
|
|
{"vuiS PKIl8P\niR5O-rC4\tz9_xTQKZ0===", relaxed},
|
||
|
|
{"vuiS PKIl8P\niR5O-rC4\tz9_xTQKZ0====", relaxed},
|
||
|
|
{"vuiS\rPKIl8P\niR5O-rC4\tz9_xTQKZ0=", relaxed}, /* 15 */
|
||
|
|
{"vuiS\rPKIl8P\niR5O-rC4\tz9_xTQKZ0= = =\x00", relaxed},
|
||
|
|
{"\nvuiS\rPKIl8P\niR5O-rC4\tz9_xTQKZ0=\n", relaxed},
|
||
|
|
{"vuiSPKIl8PiR5O-rC4z9_xTQK", invalid},
|
||
|
|
};
|
||
|
|
|
||
|
|
for (x = 0; x < sizeof(url_cases)/sizeof(url_cases[0]); ++x) {
|
||
|
|
slen1 = XSTRLEN(url_cases[x].s);
|
||
|
|
l1 = sizeof(tmp);
|
||
|
|
if(url_cases[x].flag == strict) {
|
||
|
|
DO(base64url_strict_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_strict_decode", x));
|
||
|
|
DO(base64url_sane_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_sane_decode/strict", x));
|
||
|
|
DO(base64url_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_decode/strict", x));
|
||
|
|
}
|
||
|
|
else if(url_cases[x].flag == relaxed) {
|
||
|
|
DO(base64url_strict_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
DO(base64url_sane_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_sane_decode/relaxed", x));
|
||
|
|
DO(base64url_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_decode/relaxed", x));
|
||
|
|
}
|
||
|
|
else if(url_cases[x].flag == insane) {
|
||
|
|
DO(base64url_strict_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
DO(base64url_sane_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
DO(base64url_decode(url_cases[x].s, slen1, tmp, &l1));
|
||
|
|
DO(do_compare_testvector(tmp, l1, special_case, sizeof(special_case) - 1, "base64url_decode/insane", x));
|
||
|
|
}
|
||
|
|
else { /* invalid */
|
||
|
|
DO(base64url_strict_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
DO(base64url_sane_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
DO(base64url_decode(url_cases[x].s, slen1, tmp, &l1) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_FAIL_TESTVECTOR);
|
||
|
|
}
|
||
|
|
l2 = sizeof(out);
|
||
|
|
if(x == 0) {
|
||
|
|
DO(base64url_encode(tmp, l1, out, &l2));
|
||
|
|
DO(do_compare_testvector(out, l2, url_cases[x].s, XSTRLEN(url_cases[x].s), "base64url_encode", x));
|
||
|
|
}
|
||
|
|
if(x == 1) {
|
||
|
|
DO(base64url_strict_encode(tmp, l1, out, &l2));
|
||
|
|
DO(do_compare_testvector(out, l2, url_cases[x].s, XSTRLEN(url_cases[x].s), "base64url_strict_encode", x));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#if defined(LTC_BASE64)
|
||
|
|
for (x = 0; x < sizeof(cases)/sizeof(cases[0]); ++x) {
|
||
|
|
memset(out, 0, sizeof(out));
|
||
|
|
memset(tmp, 0, sizeof(tmp));
|
||
|
|
slen1 = XSTRLEN(cases[x].s);
|
||
|
|
l1 = sizeof(out);
|
||
|
|
DO(base64_encode((unsigned char*)cases[x].s, slen1, out, &l1));
|
||
|
|
DO(do_compare_testvector(out, l1, cases[x].b64, XSTRLEN(cases[x].b64), "base64_encode", x));
|
||
|
|
l2 = sizeof(tmp);
|
||
|
|
DO(base64_strict_decode(out, l1, tmp, &l2));
|
||
|
|
DO(do_compare_testvector(tmp, l2, cases[x].s, slen1, "base64_strict_decode", x));
|
||
|
|
DO(base64_sane_decode(out, l1, tmp, &l2));
|
||
|
|
DO(do_compare_testvector(tmp, l2, cases[x].s, slen1, "base64_sane_decode", x));
|
||
|
|
DO(base64_decode(out, l1, tmp, &l2));
|
||
|
|
DO(do_compare_testvector(tmp, l2, cases[x].s, slen1, "base64_decode", x));
|
||
|
|
}
|
||
|
|
|
||
|
|
for (x = 0; x < 64; x++) {
|
||
|
|
ENSURE(yarrow_read(in, x, &yarrow_prng) == x);
|
||
|
|
l1 = sizeof(out);
|
||
|
|
DO(base64_encode(in, x, out, &l1));
|
||
|
|
l2 = sizeof(tmp);
|
||
|
|
DO(base64_decode(out, l1, tmp, &l2));
|
||
|
|
DO(do_compare_testvector(tmp, x, in, x, "random base64", x));
|
||
|
|
}
|
||
|
|
|
||
|
|
x--;
|
||
|
|
memmove(&out[11], &out[10], l1 - 10);
|
||
|
|
l1++;
|
||
|
|
l2 = sizeof(tmp);
|
||
|
|
|
||
|
|
out[10] = 0;
|
||
|
|
DO(base64_decode(out, l1, tmp, &l2));
|
||
|
|
DO(compare_testvector(tmp, l2, in, l2, "insane base64 decoding (NUL)", 0));
|
||
|
|
DO(base64_sane_decode(out, l1, tmp, &l2) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_INVALID_PACKET);
|
||
|
|
DO(base64_strict_decode(out, l1, tmp, &l2) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_INVALID_PACKET);
|
||
|
|
|
||
|
|
out[10] = 9; /* tab */
|
||
|
|
DO(base64_decode(out, l1, tmp, &l2));
|
||
|
|
DO(compare_testvector(tmp, l2, in, l2, "insane base64 decoding (TAB)", 0));
|
||
|
|
DO(base64_sane_decode(out, l1, tmp, &l2));
|
||
|
|
DO(compare_testvector(tmp, l2, in, l2, "relaxed base64 decoding (TAB)", 0));
|
||
|
|
DO(base64_strict_decode(out, l1, tmp, &l2) == CRYPT_INVALID_PACKET ? CRYPT_OK : CRYPT_INVALID_PACKET);
|
||
|
|
|
||
|
|
memset(in, 'A', sizeof(in));
|
||
|
|
l1 = strlen(As_lf);
|
||
|
|
SHOULD_FAIL(base64_encode_pem(in, 51, out, &l1, 0));
|
||
|
|
l1++;
|
||
|
|
DO(base64_encode_pem(in, 51, out, &l1, 0));
|
||
|
|
DO(compare_testvector(out, l1, As_lf, strlen(As_lf), "PEM output with \\n", 0));
|
||
|
|
l1 = strlen(As_crlf) + 1;
|
||
|
|
DO(base64_encode_pem(in, 51, out, &l1, BASE64_PEM_CRLF));
|
||
|
|
DO(compare_testvector(out, l1, As_crlf, strlen(As_crlf), "PEM output with \\r\\n", 0));
|
||
|
|
l1 = strlen(As_ssh_lf) + 1;
|
||
|
|
DO(base64_encode_pem(in, 57, out, &l1, BASE64_PEM_SSH));
|
||
|
|
DO(compare_testvector(out, l1, As_ssh_lf, strlen(As_ssh_lf), "PEM SSH-style output with \\n", 0));
|
||
|
|
l1 = strlen(As_ssh_crlf) + 1;
|
||
|
|
DO(base64_encode_pem(in, 57, out, &l1, BASE64_PEM_SSH | BASE64_PEM_CRLF));
|
||
|
|
DO(compare_testvector(out, l1, As_ssh_crlf, strlen(As_ssh_crlf), "PEM SSH-style output with \\r\\n", 0));
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
#endif
|