1 #include "first.h"
2
3 #include "base64.c"
4
5 static const base64_charset encs[] = { BASE64_STANDARD, BASE64_URL };
6 static buffer *check;
7
8 inline
check_base64(size_t out_exp,const char * in,const size_t in_len,const base64_charset enc)9 static void check_base64 (size_t out_exp, const char *in, const size_t in_len, const base64_charset enc) {
10 char out[4] = { 0, 0, 0, 0 };
11 force_assert(out_exp == li_to_base64_no_padding(out, sizeof(out), (const unsigned char *)in, in_len, enc));
12
13 buffer_clear(check);
14 force_assert(NULL != buffer_append_base64_decode(check, out, out_exp, enc));
15 force_assert(buffer_eq_slen(check, in, in_len));
16 }
17
check_all_len_0(const base64_charset enc)18 static void check_all_len_0 (const base64_charset enc) {
19 check_base64(0, "", 0, enc);
20 }
21
check_all_len_1(const base64_charset enc)22 static void check_all_len_1 (const base64_charset enc) {
23 unsigned int c1;
24 for (c1 = 0; c1 < 256; ++c1) {
25 unsigned char in[] = { c1 };
26 check_base64(2, (char *)in, sizeof(in), enc);
27 }
28 }
29
check_all_len_2(const base64_charset enc)30 static void check_all_len_2 (const base64_charset enc) {
31 unsigned int c1, c2;
32 for (c1 = 0; c1 < 256; ++c1) for (c2 = 0; c2 < 256; ++c2) {
33 unsigned char in[] = { c1, c2 };
34 check_base64(3, (char *)in, sizeof(in), enc);
35 }
36 }
37
check_all_len_3(const base64_charset enc)38 static void check_all_len_3 (const base64_charset enc) {
39 unsigned int c1, c2, c3;
40 for (c1 = 0; c1 < 256; c1+=255) for (c2 = 0; c2 < 256; ++c2) for (c3 = 0; c3 < 256; ++c3) {
41 unsigned char in[] = { c1, c2, c3 };
42 check_base64(4, (char *)in, sizeof(in), enc);
43 }
44 }
45
check_decode_ws_3(const base64_charset enc)46 static void check_decode_ws_3 (const base64_charset enc) {
47 buffer_clear(check);
48 force_assert(NULL != buffer_append_base64_decode(check, CONST_STR_LEN("YWJj"), enc));
49 force_assert(buffer_eq_slen(check, CONST_STR_LEN("abc")));
50
51 buffer_clear(check);
52 force_assert(NULL != buffer_append_base64_decode(check, CONST_STR_LEN("YWJj\r\n"), enc));
53 force_assert(buffer_eq_slen(check, CONST_STR_LEN("abc")));
54
55 buffer_clear(check);
56 force_assert(NULL != buffer_append_base64_decode(check, CONST_STR_LEN("YWJj\r\nYW\nJj\r\n"), enc));
57 force_assert(buffer_eq_slen(check, CONST_STR_LEN("abcabc")));
58 }
59
60 void test_base64 (void);
test_base64(void)61 void test_base64 (void)
62 {
63 check = buffer_init();
64
65 for (unsigned int enc = 0; enc < sizeof(encs)/sizeof(*encs); ++enc) {
66 check_all_len_0(encs[enc]);
67 check_all_len_1(encs[enc]);
68 check_all_len_2(encs[enc]);
69 check_all_len_3(encs[enc]);
70 check_decode_ws_3(encs[enc]);
71 }
72
73 buffer_free(check);
74 }
75