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 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 18 static void check_all_len_0 (const base64_charset enc) { 19 check_base64(0, "", 0, enc); 20 } 21 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 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 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 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); 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