diff --git a/Makefile.am b/Makefile.am index b04cee1a..d2f456c0 100644 --- a/Makefile.am +++ b/Makefile.am @@ -46,6 +46,8 @@ UTIL_FILES = \ hash/md2.hpp \ hash/md4.cpp \ hash/md4.hpp \ + hash/md5.cpp \ + hash/md5.hpp \ image.cpp \ image.hpp \ io.cpp \ diff --git a/hash/md5.cpp b/hash/md5.cpp new file mode 100644 index 00000000..973d0c1d --- /dev/null +++ b/hash/md5.cpp @@ -0,0 +1,292 @@ +/* + * This file is part of libgim. + * + * libgim is free software: you can redistribute it and/or modify it under the + * terms of the GNU General Public License as published by the Free Software + * Foundation, either version 3 of the License, or (at your option) any later + * version. + * + * libgim is distributed in the hope that it will be useful, but WITHOUT ANY + * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. + * + * You should have received a copy of the GNU General Public License + * along with libgim. If not, see . + * + * Copyright 2013 Danny Robson + */ + +#include "md5.hpp" + +#include "bitwise.hpp" + +#include + + +using util::hash::MD5; + + +// Per-round mixing functions +static inline uint32_t +F (uint32_t x, uint32_t y, uint32_t z) + //{ return (x & y) | (~x & z); } + { return z ^ (x & (y ^ z)); } + + +static inline uint32_t +G (uint32_t x, uint32_t y, uint32_t z) + //{ return (x & z) | (y & ~z); } + { return F (z, x, y); } + + +static inline uint32_t +H (uint32_t x, uint32_t y, uint32_t z) + { return x ^ y ^ z; } + + +static inline uint32_t +I (uint32_t x, uint32_t y, uint32_t z) + { return y ^ (x | ~z); } + + +// Mixing constants for all rounds. They are defined as 'abs(sin(i)) * max_uint32', but we use the +// literals to avoid any stupid maths issues during compilation. +const std::array T = { { + 0x00000000, + + // Round 1 + 0xd76aa478u, 0xe8c7b756u, 0x242070dbu, 0xc1bdceeeu, + 0xf57c0fafu, 0x4787c62au, 0xa8304613u, 0xfd469501u, + 0x698098d8u, 0x8b44f7afu, 0xffff5bb1u, 0x895cd7beu, + 0x6b901122u, 0xfd987193u, 0xa679438eu, 0x49b40821u, + + // Round 2 + 0xf61e2562u, 0xc040b340u, 0x265e5a51u, 0xe9b6c7aau, + 0xd62f105du, 0x02441453u, 0xd8a1e681u, 0xe7d3fbc8u, + 0x21e1cde6u, 0xc33707d6u, 0xf4d50d87u, 0x455a14edu, + 0xa9e3e905u, 0xfcefa3f8u, 0x676f02d9u, 0x8d2a4c8au, + + // Round 3 + 0xfffa3942u, 0x8771f681u, 0x6d9d6122u, 0xfde5380cu, + 0xa4beea44u, 0x4bdecfa9u, 0xf6bb4b60u, 0xbebfbc70u, + 0x289b7ec6u, 0xeaa127fau, 0xd4ef3085u, 0x04881d05u, + 0xd9d4d039u, 0xe6db99e5u, 0x1fa27cf8u, 0xc4ac5665u, + + // Round 4 + 0xf4292244u, 0x432aff97u, 0xab9423a7u, 0xfc93a039u, + 0x655b59c3u, 0x8f0ccc92u, 0xffeff47du, 0x85845dd1u, + 0x6fa87e4fu, 0xfe2ce6e0u, 0xa3014314u, 0x4e0811a1u, + 0xf7537e82u, 0xbd3af235u, 0x2ad7d2bbu, 0xeb86d391u +} }; + + + +static const uint32_t DEFAULT_A = 0x67452301; +static const uint32_t DEFAULT_B = 0xefcdab89; +static const uint32_t DEFAULT_C = 0x98badcfe; +static const uint32_t DEFAULT_D = 0x10325476; + + +MD5::MD5() +{ + reset (); +} + + +void +MD5::reset (void) { + m_total = 0; + + ABCD[0] = DEFAULT_A; + ABCD[1] = DEFAULT_B; + ABCD[2] = DEFAULT_C; + ABCD[3] = DEFAULT_D; +} + + +void +MD5::update (const void *data, size_t len) + { MD5::update (static_cast (data), len); } + + +void +MD5::update (const uint8_t *data, size_t size) { + size_t offset = m_total % sizeof (Xb); + size_t remain = sizeof (Xb) - offset; + + if (size > remain) { + memcpy (Xb + offset, data, remain); + transform (); + + m_total += remain; + size -= remain; + data += remain; + + while (size >= sizeof (Xb)) { + memcpy (Xb, data, sizeof (Xb)); + transform (); + + m_total += sizeof (Xb); + size -= sizeof (Xb); + data += sizeof (Xb); + } + + offset = 0; + } + + memcpy (Xb + offset, data, size); + m_total += size; +} + + +MD5::digest_t +MD5::digest (void) { + finish (); + + static_assert (sizeof (ABCD) == sizeof (digest_t), + "Hash state must be the same size as the final digest"); + + digest_t d; + memcpy (d.data (), ABCD.data (), sizeof (ABCD)); + return d; +} + + +void +MD5::finish (void) { + uint64_t bits = m_total * 8; + + { + // Pad with the mandatory 1 bit + size_t offset = m_total % sizeof (Xb); + Xb[offset] = 0x80; + } + + { + // Pad the remainder with 0's, until 56 bytes + size_t offset = (m_total + 1) % sizeof (Xb); + size_t remain = (56 - offset % sizeof (Xb)) % sizeof (Xb); + + if (offset > 56) { + memset (Xb + offset, 0, sizeof (Xb) - offset); + transform (); + remain -= sizeof (Xb) - offset; + offset = 0; + } + + memset (Xb + offset, 0, remain); + + // Put in the length (in bits) least significant first + for (size_t i = 0; i < sizeof (bits); ++i) { + Xb[56 + i] = bits & 0xFF; + bits >>= 8; + } + + transform (); + } +} + + +void +MD5::transform (void) { + uint32_t A = ABCD[0], + B = ABCD[1], + C = ABCD[2], + D = ABCD[3]; + + #define ROUNDx(a,b,c,d,k,s,i,f) do { \ + (a) += (f)((b), (c), (d)) + X[k] + T[i]; \ + (a) = rotatel ((a), (s)); \ + (a) += (b); \ + } while (0) + + // Round 1 + ROUNDx(A,B,C,D, 0, 7, 1, F); + ROUNDx(D,A,B,C, 1, 12, 2, F); + ROUNDx(C,D,A,B, 2, 17, 3, F); + ROUNDx(B,C,D,A, 3, 22, 4, F); + + ROUNDx(A,B,C,D, 4, 7, 5, F); + ROUNDx(D,A,B,C, 5, 12, 6, F); + ROUNDx(C,D,A,B, 6, 17, 7, F); + ROUNDx(B,C,D,A, 7, 22, 8, F); + + ROUNDx(A,B,C,D, 8, 7, 9, F); + ROUNDx(D,A,B,C, 9, 12, 10, F); + ROUNDx(C,D,A,B, 10, 17, 11, F); + ROUNDx(B,C,D,A, 11, 22, 12, F); + + ROUNDx(A,B,C,D, 12, 7, 13, F); + ROUNDx(D,A,B,C, 13, 12, 14, F); + ROUNDx(C,D,A,B, 14, 17, 15, F); + ROUNDx(B,C,D,A, 15, 22, 16, F); + + // Round 2 + ROUNDx(A,B,C,D, 1, 5, 17, G); + ROUNDx(D,A,B,C, 6, 9, 18, G); + ROUNDx(C,D,A,B, 11, 14, 19, G); + ROUNDx(B,C,D,A, 0, 20, 20, G); + + ROUNDx(A,B,C,D, 5, 5, 21, G); + ROUNDx(D,A,B,C, 10, 9, 22, G); + ROUNDx(C,D,A,B, 15, 14, 23, G); + ROUNDx(B,C,D,A, 4, 20, 24, G); + + ROUNDx(A,B,C,D, 9, 5, 25, G); + ROUNDx(D,A,B,C, 14, 9, 26, G); + ROUNDx(C,D,A,B, 3, 14, 27, G); + ROUNDx(B,C,D,A, 8, 20, 28, G); + + ROUNDx(A,B,C,D, 13, 5, 29, G); + ROUNDx(D,A,B,C, 2, 9, 30, G); + ROUNDx(C,D,A,B, 7, 14, 31, G); + ROUNDx(B,C,D,A, 12, 20, 32, G); + + // Round 3 + ROUNDx(A,B,C,D, 5, 4, 33, H); + ROUNDx(D,A,B,C, 8, 11, 34, H); + ROUNDx(C,D,A,B, 11, 16, 35, H); + ROUNDx(B,C,D,A, 14, 23, 36, H); + + ROUNDx(A,B,C,D, 1, 4, 37, H); + ROUNDx(D,A,B,C, 4, 11, 38, H); + ROUNDx(C,D,A,B, 7, 16, 39, H); + ROUNDx(B,C,D,A, 10, 23, 40, H); + + ROUNDx(A,B,C,D, 13, 4, 41, H); + ROUNDx(D,A,B,C, 0, 11, 42, H); + ROUNDx(C,D,A,B, 3, 16, 43, H); + ROUNDx(B,C,D,A, 6, 23, 44, H); + + ROUNDx(A,B,C,D, 9, 4, 45, H); + ROUNDx(D,A,B,C, 12, 11, 46, H); + ROUNDx(C,D,A,B, 15, 16, 47, H); + ROUNDx(B,C,D,A, 2, 23, 48, H); + + // Round 4 + ROUNDx(A,B,C,D, 0, 6, 49, I); + ROUNDx(D,A,B,C, 7, 10, 50, I); + ROUNDx(C,D,A,B, 14, 15, 51, I); + ROUNDx(B,C,D,A, 5, 21, 52, I); + + ROUNDx(A,B,C,D, 12, 6, 53, I); + ROUNDx(D,A,B,C, 3, 10, 54, I); + ROUNDx(C,D,A,B, 10, 15, 55, I); + ROUNDx(B,C,D,A, 1, 21, 56, I); + + ROUNDx(A,B,C,D, 8, 6, 57, I); + ROUNDx(D,A,B,C, 15, 10, 58, I); + ROUNDx(C,D,A,B, 6, 15, 59, I); + ROUNDx(B,C,D,A, 13, 21, 60, I); + + ROUNDx(A,B,C,D, 4, 6, 61, I); + ROUNDx(D,A,B,C, 11, 10, 62, I); + ROUNDx(C,D,A,B, 2, 15, 63, I); + ROUNDx(B,C,D,A, 9, 21, 64, I); + + ABCD[0] += A; + ABCD[1] += B; + ABCD[2] += C; + ABCD[3] += D; +} diff --git a/hash/md5.hpp b/hash/md5.hpp new file mode 100644 index 00000000..9dd2debf --- /dev/null +++ b/hash/md5.hpp @@ -0,0 +1,60 @@ +/* + * This file is part of libgim. + * + * libgim is free software: you can redistribute it and/or modify it under the + * terms of the GNU General Public License as published by the Free Software + * Foundation, either version 3 of the License, or (at your option) any later + * version. + * + * libgim is distributed in the hope that it will be useful, but WITHOUT ANY + * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. + * + * You should have received a copy of the GNU General Public License + * along with libgim. If not, see . + * + * Copyright 2013 Danny Robson + */ + +#ifndef __UTIL_MD5_HPP +#define __UTIL_MD5_HPP + +#include +#include +#include + +namespace util { + namespace hash { + class MD5 { + public: + typedef std::array digest_t; + + public: + MD5(); + + void update (const void *data, size_t len); + void update (const uint8_t *data, size_t len); + + digest_t digest (void); + void reset (void); + + private: + void transform (void); + void finish (void); + + uint64_t m_total; + std::array ABCD; + + union { + uint32_t X [16]; + uint8_t Xb[64]; + }; + }; + } + + typedef uint8_t md5_t; + md5_t md5 (const void *restrict data, size_t len); +} + +#endif diff --git a/test/Makefile.am b/test/Makefile.am index 42ddc356..efb80673 100644 --- a/test/Makefile.am +++ b/test/Makefile.am @@ -19,6 +19,7 @@ TEST_BIN = \ matrix \ md2 \ md4 \ + md5 \ option \ pool \ range \ @@ -64,6 +65,9 @@ md2_SOURCES = md2.cpp md4_LDADD = $(builddir)/../libutil.la md4_SOURCES = md4.cpp +md5_LDADD = $(builddir)/../libutil.la +md5_SOURCES = md5.cpp + option_LDADD = $(builddir)/../libutil.la option_SOURCES = options/success.cpp diff --git a/test/md5.cpp b/test/md5.cpp new file mode 100644 index 00000000..abd82ad9 --- /dev/null +++ b/test/md5.cpp @@ -0,0 +1,58 @@ +#include "../hash/md5.hpp" + +#include +#include + +using util::hash::MD5; + +int +main (int, char**) { + static const struct { + const char *input; + MD5::digest_t output; + } TESTS[] = { + { "", + { { 0xd4, 0x1d, 0x8c, 0xd9, 0x8f, 0x00, 0xb2, 0x04, + 0xe9, 0x80, 0x09, 0x98, 0xec, 0xf8, 0x42, 0x7e } } + }, + { "a", + { { 0x0c, 0xc1, 0x75, 0xb9, 0xc0, 0xf1, 0xb6, 0xa8, + 0x31, 0xc3, 0x99, 0xe2, 0x69, 0x77, 0x26, 0x61 } } + }, + { "abc", + { { 0x90, 0x01, 0x50, 0x98, 0x3c, 0xd2, 0x4f, 0xb0, + 0xd6, 0x96, 0x3f, 0x7d, 0x28, 0xe1, 0x7f, 0x72 } } + }, + { "message digest", + { { 0xf9, 0x6b, 0x69, 0x7d, 0x7c, 0xb7, 0x93, 0x8d, + 0x52, 0x5a, 0x2f, 0x31, 0xaa, 0xf1, 0x61, 0xd0 } } + }, + { "abcdefghijklmnopqrstuvwxyz", + { { 0xc3, 0xfc, 0xd3, 0xd7, 0x61, 0x92, 0xe4, 0x00, + 0x7d, 0xfb, 0x49, 0x6c, 0xca, 0x67, 0xe1, 0x3b } } + }, + { "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789", + { { 0xd1, 0x74, 0xab, 0x98, 0xd2, 0x77, 0xd9, 0xf5, + 0xa5, 0x61, 0x1c, 0x2c, 0x9f, 0x41, 0x9d, 0x9f } } + }, + { "12345678901234567890123456789012345678901234567890123456789012345678901234567890", + { { 0x57, 0xed, 0xf4, 0xa2, 0x2b, 0xe3, 0xc9, 0x55, + 0xac, 0x49, 0xda, 0x2e, 0x21, 0x07, 0xb6, 0x7a } } + } + }; + + bool success = true; + + for (auto i: TESTS) { + MD5 h; + h.update (i.input, strlen (i.input)); + auto out = h.digest (); + + if (out != i.output) { + std::cerr << "Failed on '" << i.input << "'\n"; + success = false; + } + } + + return success ? 0 : 1; +}