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;
}
