130 lines
3.1 KiB
C
130 lines
3.1 KiB
C
#include "base64.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
|
|
static bool test_encode_0(void)
|
|
{
|
|
char buf[1] = {};
|
|
base64_encode(buf, (const uint8_t*)"", 0);
|
|
return strlen(buf) == 0;
|
|
}
|
|
|
|
static bool test_encode_1(void)
|
|
{
|
|
static const char input[] = "a";
|
|
char buf[BASE64_ENCODED_SIZE(sizeof(input) - 1)] = {};
|
|
base64_encode(buf, (const uint8_t*)input, sizeof(input) - 1);
|
|
return strcmp(buf, "YQ==") == 0;
|
|
}
|
|
|
|
static bool test_encode_2(void)
|
|
{
|
|
static const char input[] = "ab";
|
|
char buf[BASE64_ENCODED_SIZE(sizeof(input) - 1)] = {};
|
|
base64_encode(buf, (const uint8_t*)input, sizeof(input) - 1);
|
|
return strcmp(buf, "YWI=") == 0;
|
|
}
|
|
|
|
static bool test_encode_3(void)
|
|
{
|
|
static const char input[] = "abc";
|
|
char buf[BASE64_ENCODED_SIZE(sizeof(input) - 1)] = {};
|
|
base64_encode(buf, (const uint8_t*)input, sizeof(input) - 1);
|
|
return strcmp(buf, "YWJj") == 0;
|
|
}
|
|
|
|
static bool test_encode_4(void)
|
|
{
|
|
static const char input[] = "abcd";
|
|
char buf[BASE64_ENCODED_SIZE(sizeof(input) - 1)] = {};
|
|
base64_encode(buf, (const uint8_t*)input, sizeof(input) - 1);
|
|
return strcmp(buf, "YWJjZA==") == 0;
|
|
}
|
|
|
|
static bool test_encode_5(void)
|
|
{
|
|
static const char input[] = "abcde";
|
|
char buf[BASE64_ENCODED_SIZE(sizeof(input) - 1)] = {};
|
|
base64_encode(buf, (const uint8_t*)input, sizeof(input) - 1);
|
|
return strcmp(buf, "YWJjZGU=") == 0;
|
|
}
|
|
|
|
static bool test_decode_0(void)
|
|
{
|
|
uint8_t buf[1] = {};
|
|
ssize_t r = base64_decode(buf, "");
|
|
return r == 0 && buf[0] == 0;
|
|
}
|
|
|
|
static bool test_decode_1(void)
|
|
{
|
|
static const char input[] = "YQ==";
|
|
uint8_t buf[BASE64_DECODED_MAX_SIZE(sizeof(input) - 1)] = {};
|
|
ssize_t r = base64_decode(buf, input);
|
|
return r == 1 && memcmp(buf, "a", r) == 0;
|
|
}
|
|
|
|
static bool test_decode_2(void)
|
|
{
|
|
static const char input[] = "YWI=";
|
|
uint8_t buf[BASE64_DECODED_MAX_SIZE(sizeof(input) - 1)] = {};
|
|
ssize_t r = base64_decode(buf, input);
|
|
return r == 2 && memcmp(buf, "ab", r) == 0;
|
|
}
|
|
|
|
static bool test_decode_3(void)
|
|
{
|
|
static const char input[] = "YWJj";
|
|
uint8_t buf[BASE64_DECODED_MAX_SIZE(sizeof(input) - 1)] = {};
|
|
ssize_t r = base64_decode(buf, input);
|
|
return r == 3 && memcmp(buf, "abc", r) == 0;
|
|
}
|
|
|
|
static bool test_decode_4(void)
|
|
{
|
|
static const char input[] = "YWJjZA==";
|
|
uint8_t buf[BASE64_DECODED_MAX_SIZE(sizeof(input) - 1)] = {};
|
|
ssize_t r = base64_decode(buf, input);
|
|
return r == 4 && memcmp(buf, "abcd", r) == 0;
|
|
}
|
|
|
|
static bool test_decode_5(void)
|
|
{
|
|
static const char input[] = "YWJjZGU=";
|
|
uint8_t buf[BASE64_DECODED_MAX_SIZE(sizeof(input) - 1)] = {};
|
|
ssize_t r = base64_decode(buf, input);
|
|
return r == 5 && memcmp(buf, "abcde", r) == 0;
|
|
}
|
|
|
|
#define XSTR(s) STR(s)
|
|
#define STR(s) #s
|
|
|
|
#define RUN_TEST(name) ({ \
|
|
bool ok = test_ ## name(); \
|
|
printf("[%s] %s\n", ok ? " OK " : "FAIL", XSTR(name)); \
|
|
ok; \
|
|
})
|
|
|
|
int main()
|
|
{
|
|
bool ok = true;
|
|
|
|
ok &= RUN_TEST(encode_0);
|
|
ok &= RUN_TEST(encode_1);
|
|
ok &= RUN_TEST(encode_2);
|
|
ok &= RUN_TEST(encode_3);
|
|
ok &= RUN_TEST(encode_4);
|
|
ok &= RUN_TEST(encode_5);
|
|
|
|
ok &= RUN_TEST(decode_0);
|
|
ok &= RUN_TEST(decode_1);
|
|
ok &= RUN_TEST(decode_2);
|
|
ok &= RUN_TEST(decode_3);
|
|
ok &= RUN_TEST(decode_4);
|
|
ok &= RUN_TEST(decode_5);
|
|
|
|
return ok ? 0 : 1;
|
|
}
|