LinuxGUI/Ventoy2Disk/Core/ventoy_md5.c

   1 /******************************************************************************
   2  * ventoy_md5.c  ---- ventoy md5
   3  *
   4  * Copyright (c) 2021, longpanda <admin@ventoy.net>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 3 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  *
  19  */
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <stdint.h>
  23 #include <string.h>
  24 #include <errno.h>
  25 #include <unistd.h>
  26
  27 const static uint32_t k[64] =
  28 {
  29     0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee,
  30     0xf57c0faf, 0x4787c62a, 0xa8304613, 0xfd469501,
  31     0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be,
  32     0x6b901122, 0xfd987193, 0xa679438e, 0x49b40821,
  33     0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa,
  34     0xd62f105d, 0x02441453, 0xd8a1e681, 0xe7d3fbc8,
  35     0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed,
  36     0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a,
  37     0xfffa3942, 0x8771f681, 0x6d9d6122, 0xfde5380c,
  38     0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70,
  39     0x289b7ec6, 0xeaa127fa, 0xd4ef3085, 0x04881d05,
  40     0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665,
  41     0xf4292244, 0x432aff97, 0xab9423a7, 0xfc93a039,
  42     0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1,
  43     0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1,
  44     0xf7537e82, 0xbd3af235, 0x2ad7d2bb, 0xeb86d391
  45 };
  46
  47 const static uint32_t r[] =
  48 {
  49     7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22, 7, 12, 17, 22,
  50     5,  9, 14, 20, 5,  9, 14, 20, 5,  9, 14, 20, 5,  9, 14, 20,
  51     4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23, 4, 11, 16, 23,
  52     6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21, 6, 10, 15, 21
  53 };
  54
  55 #define LEFTROTATE(x, c) (((x) << (c)) | ((x) >> (32 - (c))))
  56 #define to_bytes(val, bytes) *((uint32_t *)(bytes)) = (val)
  57
  58 #define ROTATE_CALC() \
  59 {\
  60     temp = d; \
  61     d = c; \
  62     c = b; \
  63     b = b + LEFTROTATE((a + f + k[i] + w[g]), r[i]); \
  64     a = temp; \
  65 }
  66
  67 void ventoy_md5(const void *data, uint32_t len, uint8_t *md5)
  68 {
  69     uint32_t h0, h1, h2, h3;
  70     uint32_t w[16];
  71     uint32_t a, b, c, d, i, f, g, temp;
  72     uint32_t offset, mod, delta;
  73     uint8_t postbuf[128] = {0};
  74
  75     // Initialize variables - simple count in nibbles:
  76     h0 = 0x67452301;
  77     h1 = 0xefcdab89;
  78     h2 = 0x98badcfe;
  79     h3 = 0x10325476;
  80
  81     //Pre-processing:
  82     //append "1" bit to message
  83     //append "0" bits until message length in bits ≡ 448 (mod 512)
  84     //append length mod (2^64) to message
  85
  86     mod = len % 64;
  87     if (mod)
  88     {
  89         memcpy(postbuf, (const uint8_t *)data + len - mod, mod);
  90     }
  91
  92     postbuf[mod] = 0x80;
  93     if (mod < 56)
  94     {
  95         to_bytes(len * 8, postbuf + 56);
  96         to_bytes(len >> 29, postbuf + 60);
  97         delta = 64;
  98     }
  99     else
 100     {
 101         to_bytes(len * 8, postbuf + 120);
 102         to_bytes(len >> 29, postbuf + 124);
 103         delta = 128;
 104     }
 105
 106     len -= mod;
 107
 108     for (offset = 0; offset < len + delta; offset += 64)
 109     {
 110         if (offset < len)
 111         {
 112             memcpy(w, (const uint8_t *)data + offset, 64);
 113         }
 114         else
 115         {
 116             memcpy(w, postbuf + offset - len, 64);
 117         }
 118
 119         // Initialize hash value for this chunk:
 120         a = h0;
 121         b = h1;
 122         c = h2;
 123         d = h3;
 124
 125         // Main loop:
 126         for (i = 0; i < 16; i++)
 127         {
 128             f = (b & c) | ((~b) & d);
 129             g = i;
 130             ROTATE_CALC();
 131         }
 132
 133         for (i = 16; i < 32; i++)
 134         {
 135             f = (d & b) | ((~d) & c);
 136             g = (5 * i + 1) % 16;
 137             ROTATE_CALC();
 138         }
 139
 140         for (i = 32; i < 48; i++)
 141         {
 142             f = b ^ c ^ d;
 143             g = (3 * i + 5) % 16;
 144             ROTATE_CALC();
 145         }
 146
 147         for (i = 48; i < 64; i++)
 148         {
 149             f = c ^ (b | (~d));
 150             g = (7 * i) % 16;
 151             ROTATE_CALC();
 152         }
 153
 154         // Add this chunk's hash to result so far:
 155         h0 += a;
 156         h1 += b;
 157         h2 += c;
 158         h3 += d;
 159     }
 160
 161     //var char md5[16] := h0 append h1 append h2 append h3 //(Output is in little-endian)
 162     to_bytes(h0, md5);
 163     to_bytes(h1, md5 + 4);
 164     to_bytes(h2, md5 + 8);
 165     to_bytes(h3, md5 + 12);
 166 }
 167