使用Base64字符串字面量编译期初始化uint8_t数组的实现问询

核心思路 #

利用编译期常量表达式能力，直接将Base64字符串字面量解码为std::array<uint8_t, N>，无需显式传入大小参数：

• 编译期推导输入字符串字面量的长度
• 编译期计算解码后的字节数（处理Base64填充符=）
• 编译期完成字符到索引的映射、分组解码逻辑，填充目标数组

完整实现代码 #

1. 基础辅助函数（兼容所有C++版本） #

#include <array>
#include <cstdint>

// 编译期将Base64字符转换为对应索引
constexpr uint8_t base64_char_to_idx(char c) {
    if (c >= 'A' && c <= 'Z') return c - 'A';
    if (c >= 'a' && c <= 'z') return c - 'a' + 26;
    if (c >= '0' && c <= '9') return c - '0' + 52;
    if (c == '+') return 62;
    if (c == '/') return 63;
    return 0; // 假设输入为合法Base64，非法字符可按需添加错误处理
}

// 编译期计算解码后的字节长度
template <size_t SrcStrLen>
constexpr size_t calc_decoded_size(const char (&src)[SrcStrLen]) {
    size_t padding_count = 0;
    // 跳过字符串末尾的'\0'，检查填充符
    if (src[SrcStrLen - 1] == '=') padding_count++;
    if (src[SrcStrLen - 2] == '=') padding_count++;
    // 有效Base64字符数 = 总长度 - 1（去掉'\0'） - 填充数
    return ((SrcStrLen - 1 - padding_count) * 3) / 4;
}

2. 主解码函数（兼容C++14/17/20） #

// 接收Base64字符串字面量，返回编译期初始化的std::array
template <size_t SrcStrLen>
constexpr std::array<uint8_t, calc_decoded_size(std::declval<const char(&)[SrcStrLen]>())>
base64_decode(const char (&src)[SrcStrLen]) {
    using ResultArray = std::array<uint8_t, calc_decoded_size(src)>;
    ResultArray result{};

    size_t out_pos = 0;
    uint32_t bit_accumulator = 0;
    size_t accumulated_bits = 0;
    size_t in_pos = 0;

    // 遍历所有有效Base64字符（跳过末尾的'\0'）
    while (in_pos < SrcStrLen - 1) {
        const char c = src[in_pos++];
        if (c == '=') break; // 遇到填充符停止解码

        // 累积6位数据
        bit_accumulator = (bit_accumulator << 6) | base64_char_to_idx(c);
        accumulated_bits += 6;

        // 当累积足够8位时，输出一个字节
        if (accumulated_bits >= 8) {
            accumulated_bits -= 8;
            result[out_pos++] = static_cast<uint8_t>((bit_accumulator >> accumulated_bits) & 0xFF);
        }
    }

    return result;
}

使用示例 #

// 示例：解码"Hello World"的Base64字符串
constexpr auto decoded_img_data = base64_decode("SGVsbG8gV29ybGQ=");

// 编译期验证结果长度（应为11）
static_assert(decoded_img_data.size() == 11);
// 编译期验证内容
static_assert(decoded_img_data[0] == 'H');
static_assert(decoded_img_data[10] == 'd');

关键说明 #

• 自动大小推导：通过模板参数推导字符串字面量的长度，结合编译期计算的解码后长度，自动生成对应大小的std::array，无需手动传入尺寸参数
• C++14兼容性：所有逻辑均使用C14允许的constexpr语法（支持简单循环、变量赋值），C17/20可直接复用，也可添加constexpr if等语法做进一步优化
• 输入合法性：当前实现假设输入为合法Base64字符串，若需处理非法字符，可在base64_char_to_idx中添加编译期断言（如static_assert）抛出错误

内容的提问来源于stack exchange，提问作者elechris