Browse Source

refactor: pure constexpr functions

master
Dnomd343 8 months ago
parent
commit
2570730938
  1. 47
      src/impl/constexpr.inc

47
src/impl/constexpr.inc

@ -2,23 +2,25 @@
namespace md5::impl { namespace md5::impl {
struct md5_ctx_ce { struct md5_ce_ctx {
uint32_t A = value::kA; uint32_t A = value::kA;
uint32_t B = value::kB; uint32_t B = value::kB;
uint32_t C = value::kC; uint32_t C = value::kC;
uint32_t D = value::kD; uint32_t D = value::kD;
};
struct md5_ce {
const char *data; const char *data;
uint64_t data_len, padded_len; uint64_t data_len, padded_len;
constexpr md5_ctx_ce(const char *data, const uint64_t len) constexpr md5_ce(const char *data, const uint64_t len)
: data(data), data_len(len), padded_len((len + 64 + 8) & ~0b111111ULL) {} : data(data), data_len(len), padded_len((len + 64 + 8) & ~0b111111ULL) {}
}; };
using Block = std::array<uint32_t, 16>; // single md5 block with 64 bytes using Block = std::array<uint32_t, 16>; // single md5 block with 64 bytes
/// Get the data and padding byte of the specified index. /// Get the data and padding byte of the specified index.
constexpr uint8_t GetByte(const md5_ctx_ce *ctx, const uint64_t index) { constexpr uint8_t GetByte(const md5_ce *ctx, const uint64_t index) {
if (index < ctx->data_len) // message data if (index < ctx->data_len) // message data
return ctx->data[index]; return ctx->data[index];
if (index == ctx->data_len) // padding flag if (index == ctx->data_len) // padding flag
@ -30,7 +32,7 @@ constexpr uint8_t GetByte(const md5_ctx_ce *ctx, const uint64_t index) {
} }
/// Get the MD5 block content at the specified index. /// Get the MD5 block content at the specified index.
constexpr Block GetBlock(const md5_ctx_ce *ctx, const uint64_t index) { constexpr Block GetBlock(const md5_ce *ctx, const uint64_t index) {
Block block {}; Block block {};
for (int i = 0; i < 16; ++i) { for (int i = 0; i < 16; ++i) {
const auto offset = index + i * 4; const auto offset = index + i * 4;
@ -54,15 +56,32 @@ constexpr std::array<char, 32> DigestCE(const std::array<uint32_t, 4> &ctx) {
return result; return result;
} }
constexpr std::array<char, 32> MD5::HashCE(const char *data, uint64_t len) { constexpr uint32_t Calc(const md5_ce_ctx &ctx, const int i) {
md5_ctx_ce ctx(data, len); if (i < 0x10)
for (uint32_t index = 0; index < ctx.padded_len; index += 64) { return ctx.D ^ (ctx.B & (ctx.C ^ ctx.D));
auto block = GetBlock(&ctx, index); if (i < 0x20)
auto A = ctx.A; return ctx.C ^ (ctx.D & (ctx.B ^ ctx.C));
auto B = ctx.B; if (i < 0x30)
auto C = ctx.C; return ctx.B ^ ctx.C ^ ctx.D;
auto D = ctx.D; return ctx.C ^ (ctx.B | ~ctx.D);
MD5_UPDATE }
constexpr md5_ce_ctx Round(const Block &block, md5_ce_ctx ctx) {
for (int i = 0; i < 64; ++i) {
const auto a = ctx.A + Calc(ctx, i) + block[K(i)] + T(i);
ctx.A = ctx.D;
ctx.D = ctx.C;
ctx.C = ctx.B;
ctx.B += a << S(i) | a >> (32 - S(i));
}
return ctx;
}
constexpr std::array<char, 32> MD5::HashCE(const char *data, const uint64_t len) {
md5_ce_ctx ctx;
const md5_ce md5(data, len);
for (uint32_t index = 0; index < md5.padded_len; index += 64) {
const auto [A, B, C, D] = Round(GetBlock(&md5, index), ctx);
ctx.A += A; ctx.A += A;
ctx.B += B; ctx.B += B;
ctx.C += C; ctx.C += C;
@ -71,6 +90,6 @@ constexpr std::array<char, 32> MD5::HashCE(const char *data, uint64_t len) {
return DigestCE({ctx.A, ctx.B, ctx.C, ctx.D}); return DigestCE({ctx.A, ctx.B, ctx.C, ctx.D});
} }
static_assert(MD5::HashCE("")[0] == 'd', "invalid constexpr"); static_assert(MD5::HashCE("")[0] == 'd');
} // namespace md5::impl } // namespace md5::impl

Loading…
Cancel
Save