91 lines
3.7 KiB
C++
91 lines
3.7 KiB
C++
#include "encrypt.hpp"
|
|
#include <vector>
|
|
|
|
namespace encrypt
|
|
{
|
|
string MD5(const string &info)
|
|
{
|
|
auto leftrotate = [](uint32_t x, uint32_t c) -> uint32_t { return (x << c) | (x >> (32 - c)); };
|
|
|
|
static const uint32_t s[64] = {
|
|
7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22,
|
|
5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20, 5, 9, 14, 20,
|
|
4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23,
|
|
6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21
|
|
};
|
|
static const uint32_t K[64] = {
|
|
0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee, 0xf57c0faf, 0x4787c62a, 0xa8304613, 0xfd469501,
|
|
0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be, 0x6b901122, 0xfd987193, 0xa679438e, 0x49b40821,
|
|
0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa, 0xd62f105d, 0x02441453, 0xd8a1e681, 0xe7d3fbc8,
|
|
0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed, 0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a,
|
|
0xfffa3942, 0x8771f681, 0x6d9d6122, 0xfde5380c, 0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70,
|
|
0x289b7ec6, 0xeaa127fa, 0xd4ef3085, 0x04881d05, 0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665,
|
|
0xf4292244, 0x432aff97, 0xab9423a7, 0xfc93a039, 0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1,
|
|
0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1, 0xf7537e82, 0xbd3af235, 0x2ad7d2bb, 0xeb86d391
|
|
};
|
|
|
|
uint32_t a0 = 0x67452301;
|
|
uint32_t b0 = 0xefcdab89;
|
|
uint32_t c0 = 0x98badcfe;
|
|
uint32_t d0 = 0x10325476;
|
|
|
|
std::vector<uint8_t> msg(info.begin(), info.end());
|
|
uint64_t bit_len = static_cast<uint64_t>(msg.size()) * 8ULL;
|
|
msg.push_back(0x80);
|
|
while ((msg.size() % 64) != 56) msg.push_back(0x00);
|
|
for (int i = 0; i < 8; ++i) msg.push_back(static_cast<uint8_t>((bit_len >> (8 * i)) & 0xff));
|
|
|
|
for (size_t offset = 0; offset < msg.size(); offset += 64)
|
|
{
|
|
uint32_t M[16];
|
|
for (int i = 0; i < 16; ++i)
|
|
{
|
|
size_t j = offset + i * 4;
|
|
M[i] = static_cast<uint32_t>(msg[j]) |
|
|
(static_cast<uint32_t>(msg[j + 1]) << 8) |
|
|
(static_cast<uint32_t>(msg[j + 2]) << 16) |
|
|
(static_cast<uint32_t>(msg[j + 3]) << 24);
|
|
}
|
|
|
|
uint32_t A = a0, B = b0, C = c0, D = d0;
|
|
for (uint32_t i = 0; i < 64; ++i)
|
|
{
|
|
uint32_t F, g;
|
|
if (i < 16) { F = (B & C) | ((~B) & D); g = i; }
|
|
else if (i < 32) { F = (D & B) | ((~D) & C); g = (5 * i + 1) % 16; }
|
|
else if (i < 48) { F = B ^ C ^ D; g = (3 * i + 5) % 16; }
|
|
else { F = C ^ (B | (~D)); g = (7 * i) % 16; }
|
|
|
|
F = F + A + K[i] + M[g];
|
|
A = D;
|
|
D = C;
|
|
C = B;
|
|
B = B + leftrotate(F, s[i]);
|
|
}
|
|
|
|
a0 += A; b0 += B; c0 += C; d0 += D;
|
|
}
|
|
|
|
uint8_t digest[16];
|
|
auto u32_to_le = [](uint32_t v, uint8_t out[4]) {
|
|
out[0] = static_cast<uint8_t>(v & 0xff);
|
|
out[1] = static_cast<uint8_t>((v >> 8) & 0xff);
|
|
out[2] = static_cast<uint8_t>((v >> 16) & 0xff);
|
|
out[3] = static_cast<uint8_t>((v >> 24) & 0xff);
|
|
};
|
|
u32_to_le(a0, digest + 0);
|
|
u32_to_le(b0, digest + 4);
|
|
u32_to_le(c0, digest + 8);
|
|
u32_to_le(d0, digest + 12);
|
|
|
|
static const char *hex = "0123456789abcdef";
|
|
std::string out;
|
|
out.resize(32);
|
|
for (int i = 0; i < 16; ++i)
|
|
{
|
|
out[i * 2] = hex[(digest[i] >> 4) & 0x0f];
|
|
out[i * 2 + 1] = hex[digest[i] & 0x0f];
|
|
}
|
|
return out;
|
|
}
|
|
} |