static void test_base64_decode(void) { const char base64_map[65] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/="; char expstr[] = "This is a message that is multiple of 3 chars"; const char *instrs[] = { "VGhpcyBpcyBhIG1lc3NhZ2UgdGhhdCBpcyBtdWx0aXBsZSBvZiAzIGNoYXJz", "VGhpcyBpcyBhIG1lc3NhZ2UgdGhhdCBpcyBtdWx0aXBsZSBvZiAzIGNoYXI=", "VGhpcyBpcyBhIG1lc3NhZ2UgdGhhdCBpcyBtdWx0aXBsZSBvZiAzIGNoYQ==", "VGhpcyBpcyBhIG1lc3NhZ2UgdGhhdCBpcyBtdWx0aXBsZSBvZiAzIGNo" }; struct sol_str_slice exp, slice; char outstr[sizeof(expstr)]; size_t r, i; exp = sol_str_slice_from_str(expstr); for (i = 0; i < SOL_UTIL_ARRAY_SIZE(instrs); i++) { slice = sol_str_slice_from_str(instrs[i]); memset(outstr, 0xff, sizeof(outstr)); r = sol_util_base64_decode(outstr, sizeof(outstr), slice, base64_map); ASSERT_INT_EQ(r, exp.len); ASSERT_INT_EQ(outstr[r], (char)0xff); outstr[r] = '\0'; ASSERT_STR_EQ(outstr, exp.data); exp.len--; expstr[exp.len] = '\0'; } /* negative test (invalid char) */ slice = sol_str_slice_from_str("****"); memset(outstr, 0xff, sizeof(outstr)); r = sol_util_base64_decode(outstr, sizeof(outstr), slice, base64_map); ASSERT_INT_EQ(r, -EINVAL); /* short sequence (not multiple of 4) */ slice = sol_str_slice_from_str("123"); memset(outstr, 0xff, sizeof(outstr)); r = sol_util_base64_decode(outstr, sizeof(outstr), slice, base64_map); ASSERT_INT_EQ(r, -EINVAL); }
SOL_API int sol_buffer_insert_from_base64(struct sol_buffer *buf, size_t pos, const struct sol_str_slice slice, const char base64_map[SOL_STATIC_ARRAY_SIZE(65)]) { char *p; size_t new_size; ssize_t decoded_size, r; const size_t nul_size = nul_byte_size(buf); int err; SOL_NULL_CHECK(buf, -EINVAL); SOL_INT_CHECK(pos, > buf->used, -EINVAL); if (slice.len == 0) return 0; if (pos == buf->used) return sol_buffer_append_from_base64(buf, slice, base64_map); decoded_size = sol_util_base64_calculate_decoded_len(slice, base64_map); if (decoded_size < 0) return decoded_size; err = sol_util_size_add(buf->used, decoded_size, &new_size); if (err < 0) return err; if (nul_size) { err = sol_util_size_add(new_size, nul_size, &new_size); if (err < 0) return err; } err = sol_buffer_ensure(buf, new_size); if (err < 0) return err; p = sol_buffer_at(buf, pos); memmove(p + decoded_size, p, buf->used - pos); r = sol_util_base64_decode(p, decoded_size, slice, base64_map); if (r != decoded_size) { memmove(p, p + decoded_size, buf->used - pos); if (nul_size) sol_buffer_ensure_nul_byte(buf); if (r < 0) return r; else return -EINVAL; } buf->used += decoded_size; if (nul_size) return sol_buffer_ensure_nul_byte(buf); return 0; }