1 //===-- tsan_md5.cc -------------------------------------------------------===// 2 // 3 // This file is distributed under the University of Illinois Open Source 4 // License. See LICENSE.TXT for details. 5 // 6 //===----------------------------------------------------------------------===// 7 // 8 // This file is a part of ThreadSanitizer (TSan), a race detector. 9 // 10 //===----------------------------------------------------------------------===// 11 #include "tsan_defs.h" 12 13 namespace __tsan { 14 15 #define F(x, y, z) ((z) ^ ((x) & ((y) ^ (z)))) 16 #define G(x, y, z) ((y) ^ ((z) & ((x) ^ (y)))) 17 #define H(x, y, z) ((x) ^ (y) ^ (z)) 18 #define I(x, y, z) ((y) ^ ((x) | ~(z))) 19 20 #define STEP(f, a, b, c, d, x, t, s) \ 21 (a) += f((b), (c), (d)) + (x) + (t); \ 22 (a) = (((a) << (s)) | (((a) & 0xffffffff) >> (32 - (s)))); \ 23 (a) += (b); 24 25 #define SET(n) \ 26 (*(MD5_u32plus *)&ptr[(n) * 4]) 27 #define GET(n) \ 28 SET(n) 29 30 typedef unsigned int MD5_u32plus; 31 typedef unsigned long ulong_t; // NOLINT 32 33 typedef struct { 34 MD5_u32plus lo, hi; 35 MD5_u32plus a, b, c, d; 36 unsigned char buffer[64]; 37 MD5_u32plus block[16]; 38 } MD5_CTX; 39 40 static void *body(MD5_CTX *ctx, void *data, ulong_t size) { 41 unsigned char *ptr; 42 MD5_u32plus a, b, c, d; 43 MD5_u32plus saved_a, saved_b, saved_c, saved_d; 44 45 ptr = (unsigned char*)data; 46 47 a = ctx->a; 48 b = ctx->b; 49 c = ctx->c; 50 d = ctx->d; 51 52 do { 53 saved_a = a; 54 saved_b = b; 55 saved_c = c; 56 saved_d = d; 57 58 STEP(F, a, b, c, d, SET(0), 0xd76aa478, 7) 59 STEP(F, d, a, b, c, SET(1), 0xe8c7b756, 12) 60 STEP(F, c, d, a, b, SET(2), 0x242070db, 17) 61 STEP(F, b, c, d, a, SET(3), 0xc1bdceee, 22) 62 STEP(F, a, b, c, d, SET(4), 0xf57c0faf, 7) 63 STEP(F, d, a, b, c, SET(5), 0x4787c62a, 12) 64 STEP(F, c, d, a, b, SET(6), 0xa8304613, 17) 65 STEP(F, b, c, d, a, SET(7), 0xfd469501, 22) 66 STEP(F, a, b, c, d, SET(8), 0x698098d8, 7) 67 STEP(F, d, a, b, c, SET(9), 0x8b44f7af, 12) 68 STEP(F, c, d, a, b, SET(10), 0xffff5bb1, 17) 69 STEP(F, b, c, d, a, SET(11), 0x895cd7be, 22) 70 STEP(F, a, b, c, d, SET(12), 0x6b901122, 7) 71 STEP(F, d, a, b, c, SET(13), 0xfd987193, 12) 72 STEP(F, c, d, a, b, SET(14), 0xa679438e, 17) 73 STEP(F, b, c, d, a, SET(15), 0x49b40821, 22) 74 75 STEP(G, a, b, c, d, GET(1), 0xf61e2562, 5) 76 STEP(G, d, a, b, c, GET(6), 0xc040b340, 9) 77 STEP(G, c, d, a, b, GET(11), 0x265e5a51, 14) 78 STEP(G, b, c, d, a, GET(0), 0xe9b6c7aa, 20) 79 STEP(G, a, b, c, d, GET(5), 0xd62f105d, 5) 80 STEP(G, d, a, b, c, GET(10), 0x02441453, 9) 81 STEP(G, c, d, a, b, GET(15), 0xd8a1e681, 14) 82 STEP(G, b, c, d, a, GET(4), 0xe7d3fbc8, 20) 83 STEP(G, a, b, c, d, GET(9), 0x21e1cde6, 5) 84 STEP(G, d, a, b, c, GET(14), 0xc33707d6, 9) 85 STEP(G, c, d, a, b, GET(3), 0xf4d50d87, 14) 86 STEP(G, b, c, d, a, GET(8), 0x455a14ed, 20) 87 STEP(G, a, b, c, d, GET(13), 0xa9e3e905, 5) 88 STEP(G, d, a, b, c, GET(2), 0xfcefa3f8, 9) 89 STEP(G, c, d, a, b, GET(7), 0x676f02d9, 14) 90 STEP(G, b, c, d, a, GET(12), 0x8d2a4c8a, 20) 91 92 STEP(H, a, b, c, d, GET(5), 0xfffa3942, 4) 93 STEP(H, d, a, b, c, GET(8), 0x8771f681, 11) 94 STEP(H, c, d, a, b, GET(11), 0x6d9d6122, 16) 95 STEP(H, b, c, d, a, GET(14), 0xfde5380c, 23) 96 STEP(H, a, b, c, d, GET(1), 0xa4beea44, 4) 97 STEP(H, d, a, b, c, GET(4), 0x4bdecfa9, 11) 98 STEP(H, c, d, a, b, GET(7), 0xf6bb4b60, 16) 99 STEP(H, b, c, d, a, GET(10), 0xbebfbc70, 23) 100 STEP(H, a, b, c, d, GET(13), 0x289b7ec6, 4) 101 STEP(H, d, a, b, c, GET(0), 0xeaa127fa, 11) 102 STEP(H, c, d, a, b, GET(3), 0xd4ef3085, 16) 103 STEP(H, b, c, d, a, GET(6), 0x04881d05, 23) 104 STEP(H, a, b, c, d, GET(9), 0xd9d4d039, 4) 105 STEP(H, d, a, b, c, GET(12), 0xe6db99e5, 11) 106 STEP(H, c, d, a, b, GET(15), 0x1fa27cf8, 16) 107 STEP(H, b, c, d, a, GET(2), 0xc4ac5665, 23) 108 109 STEP(I, a, b, c, d, GET(0), 0xf4292244, 6) 110 STEP(I, d, a, b, c, GET(7), 0x432aff97, 10) 111 STEP(I, c, d, a, b, GET(14), 0xab9423a7, 15) 112 STEP(I, b, c, d, a, GET(5), 0xfc93a039, 21) 113 STEP(I, a, b, c, d, GET(12), 0x655b59c3, 6) 114 STEP(I, d, a, b, c, GET(3), 0x8f0ccc92, 10) 115 STEP(I, c, d, a, b, GET(10), 0xffeff47d, 15) 116 STEP(I, b, c, d, a, GET(1), 0x85845dd1, 21) 117 STEP(I, a, b, c, d, GET(8), 0x6fa87e4f, 6) 118 STEP(I, d, a, b, c, GET(15), 0xfe2ce6e0, 10) 119 STEP(I, c, d, a, b, GET(6), 0xa3014314, 15) 120 STEP(I, b, c, d, a, GET(13), 0x4e0811a1, 21) 121 STEP(I, a, b, c, d, GET(4), 0xf7537e82, 6) 122 STEP(I, d, a, b, c, GET(11), 0xbd3af235, 10) 123 STEP(I, c, d, a, b, GET(2), 0x2ad7d2bb, 15) 124 STEP(I, b, c, d, a, GET(9), 0xeb86d391, 21) 125 126 a += saved_a; 127 b += saved_b; 128 c += saved_c; 129 d += saved_d; 130 131 ptr += 64; 132 } while (size -= 64); 133 134 ctx->a = a; 135 ctx->b = b; 136 ctx->c = c; 137 ctx->d = d; 138 139 return ptr; 140 } 141 142 void MD5_Init(MD5_CTX *ctx) { 143 ctx->a = 0x67452301; 144 ctx->b = 0xefcdab89; 145 ctx->c = 0x98badcfe; 146 ctx->d = 0x10325476; 147 148 ctx->lo = 0; 149 ctx->hi = 0; 150 } 151 152 void MD5_Update(MD5_CTX *ctx, void *data, ulong_t size) { 153 MD5_u32plus saved_lo; 154 ulong_t used, free; 155 156 saved_lo = ctx->lo; 157 if ((ctx->lo = (saved_lo + size) & 0x1fffffff) < saved_lo) 158 ctx->hi++; 159 ctx->hi += size >> 29; 160 161 used = saved_lo & 0x3f; 162 163 if (used) { 164 free = 64 - used; 165 166 if (size < free) { 167 internal_memcpy(&ctx->buffer[used], data, size); 168 return; 169 } 170 171 internal_memcpy(&ctx->buffer[used], data, free); 172 data = (unsigned char *)data + free; 173 size -= free; 174 body(ctx, ctx->buffer, 64); 175 } 176 177 if (size >= 64) { 178 data = body(ctx, data, size & ~(ulong_t)0x3f); 179 size &= 0x3f; 180 } 181 182 internal_memcpy(ctx->buffer, data, size); 183 } 184 185 void MD5_Final(unsigned char *result, MD5_CTX *ctx) { 186 ulong_t used, free; 187 188 used = ctx->lo & 0x3f; 189 190 ctx->buffer[used++] = 0x80; 191 192 free = 64 - used; 193 194 if (free < 8) { 195 internal_memset(&ctx->buffer[used], 0, free); 196 body(ctx, ctx->buffer, 64); 197 used = 0; 198 free = 64; 199 } 200 201 internal_memset(&ctx->buffer[used], 0, free - 8); 202 203 ctx->lo <<= 3; 204 ctx->buffer[56] = ctx->lo; 205 ctx->buffer[57] = ctx->lo >> 8; 206 ctx->buffer[58] = ctx->lo >> 16; 207 ctx->buffer[59] = ctx->lo >> 24; 208 ctx->buffer[60] = ctx->hi; 209 ctx->buffer[61] = ctx->hi >> 8; 210 ctx->buffer[62] = ctx->hi >> 16; 211 ctx->buffer[63] = ctx->hi >> 24; 212 213 body(ctx, ctx->buffer, 64); 214 215 result[0] = ctx->a; 216 result[1] = ctx->a >> 8; 217 result[2] = ctx->a >> 16; 218 result[3] = ctx->a >> 24; 219 result[4] = ctx->b; 220 result[5] = ctx->b >> 8; 221 result[6] = ctx->b >> 16; 222 result[7] = ctx->b >> 24; 223 result[8] = ctx->c; 224 result[9] = ctx->c >> 8; 225 result[10] = ctx->c >> 16; 226 result[11] = ctx->c >> 24; 227 result[12] = ctx->d; 228 result[13] = ctx->d >> 8; 229 result[14] = ctx->d >> 16; 230 result[15] = ctx->d >> 24; 231 232 internal_memset(ctx, 0, sizeof(*ctx)); 233 } 234 235 MD5Hash md5_hash(const void *data, uptr size) { 236 MD5Hash res; 237 MD5_CTX ctx; 238 MD5_Init(&ctx); 239 MD5_Update(&ctx, (void*)data, size); 240 MD5_Final((unsigned char*)&res.hash[0], &ctx); 241 return res; 242 } 243 } // namespace __tsan 244