Unreviewed, rolling out r234489.
[WebKit-https.git] / Source / WTF / wtf / MD5.cpp
1 // The original file was copied from sqlite, and was in the public domain.
2 // Modifications Copyright 2006 Google Inc. All Rights Reserved
3 /*
4  * Copyright (C) 2010 Google Inc. All rights reserved.
5  * Copyright (C) 2015 Apple Inc. All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are
9  * met:
10  *
11  *     * Redistributions of source code must retain the above copyright
12  * notice, this list of conditions and the following disclaimer.
13  *     * Redistributions in binary form must reproduce the above
14  * copyright notice, this list of conditions and the following disclaimer
15  * in the documentation and/or other materials provided with the
16  * distribution.
17  *     * Neither the name of Google Inc. nor the names of its
18  * contributors may be used to endorse or promote products derived from
19  * this software without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32  */
33 /*
34  * This code implements the MD5 message-digest algorithm.
35  * The algorithm is due to Ron Rivest.  This code was
36  * written by Colin Plumb in 1993, no copyright is claimed.
37  * This code is in the public domain; do with it what you wish.
38  *
39  * Equivalent code is available from RSA Data Security, Inc.
40  * This code has been tested against that, and is equivalent,
41  * except that you don't need to include two pages of legalese
42  * with every copy.
43  *
44  * To compute the message digest of a chunk of bytes, construct an
45  * MD5 instance, call addBytes as needed on buffers full of bytes,
46  * and then call checksum, which will fill a supplied 16-byte array
47  * with the digest.
48  */
49
50 #include "config.h"
51 #include "MD5.h"
52
53 #include "Assertions.h"
54 #include <wtf/StdLibExtras.h>
55
56 namespace WTF {
57
58 #if PLATFORM(COCOA)
59
60 MD5::MD5()
61 {
62     CC_MD5_Init(&m_context);
63 }
64
65 void MD5::addBytes(const uint8_t* input, size_t length)
66 {
67     CC_MD5_Update(&m_context, input, length);
68 }
69
70 void MD5::checksum(Digest& hash)
71 {
72     CC_MD5_Final(hash.data(), &m_context);
73 }
74     
75 #else
76
77 // Note: this code is harmless on little-endian machines.
78
79 static void toLittleEndian(uint8_t* buf, unsigned longs)
80 {
81     ASSERT(longs > 0);
82     do {
83         uint32_t t = static_cast<uint32_t>(buf[3] << 8 | buf[2]) << 16 | buf[1] << 8 | buf[0];
84         ASSERT_WITH_MESSAGE(!(reinterpret_cast<uintptr_t>(buf) % sizeof(t)), "alignment error of buf");
85         memcpy(buf, &t, sizeof(t));
86         buf += sizeof(t);
87     } while (--longs);
88 }
89
90 // The four core functions.
91 // F1 is originally defined as (x & y | ~x & z), but optimized somewhat: 4 bit ops -> 3 bit ops.
92 #define F1(x, y, z) (z ^ (x & (y ^ z)))
93 #define F2(x, y, z) F1(z, x, y)
94 #define F3(x, y, z) (x ^ y ^ z)
95 #define F4(x, y, z) (y ^ (x | ~z))
96
97 // This is the central step in the MD5 algorithm.
98 #define MD5STEP(f, w, x, y, z, data, s) \
99     (w += f(x, y, z) + data, w = w << s | w >> (32 - s), w += x)
100
101 static void MD5Transform(uint32_t buf[4], const uint32_t in[16])
102 {
103     uint32_t a = buf[0];
104     uint32_t b = buf[1];
105     uint32_t c = buf[2];
106     uint32_t d = buf[3];
107
108     MD5STEP(F1, a, b, c, d, in[ 0]+0xd76aa478,  7);
109     MD5STEP(F1, d, a, b, c, in[ 1]+0xe8c7b756, 12);
110     MD5STEP(F1, c, d, a, b, in[ 2]+0x242070db, 17);
111     MD5STEP(F1, b, c, d, a, in[ 3]+0xc1bdceee, 22);
112     MD5STEP(F1, a, b, c, d, in[ 4]+0xf57c0faf,  7);
113     MD5STEP(F1, d, a, b, c, in[ 5]+0x4787c62a, 12);
114     MD5STEP(F1, c, d, a, b, in[ 6]+0xa8304613, 17);
115     MD5STEP(F1, b, c, d, a, in[ 7]+0xfd469501, 22);
116     MD5STEP(F1, a, b, c, d, in[ 8]+0x698098d8,  7);
117     MD5STEP(F1, d, a, b, c, in[ 9]+0x8b44f7af, 12);
118     MD5STEP(F1, c, d, a, b, in[10]+0xffff5bb1, 17);
119     MD5STEP(F1, b, c, d, a, in[11]+0x895cd7be, 22);
120     MD5STEP(F1, a, b, c, d, in[12]+0x6b901122,  7);
121     MD5STEP(F1, d, a, b, c, in[13]+0xfd987193, 12);
122     MD5STEP(F1, c, d, a, b, in[14]+0xa679438e, 17);
123     MD5STEP(F1, b, c, d, a, in[15]+0x49b40821, 22);
124
125     MD5STEP(F2, a, b, c, d, in[ 1]+0xf61e2562,  5);
126     MD5STEP(F2, d, a, b, c, in[ 6]+0xc040b340,  9);
127     MD5STEP(F2, c, d, a, b, in[11]+0x265e5a51, 14);
128     MD5STEP(F2, b, c, d, a, in[ 0]+0xe9b6c7aa, 20);
129     MD5STEP(F2, a, b, c, d, in[ 5]+0xd62f105d,  5);
130     MD5STEP(F2, d, a, b, c, in[10]+0x02441453,  9);
131     MD5STEP(F2, c, d, a, b, in[15]+0xd8a1e681, 14);
132     MD5STEP(F2, b, c, d, a, in[ 4]+0xe7d3fbc8, 20);
133     MD5STEP(F2, a, b, c, d, in[ 9]+0x21e1cde6,  5);
134     MD5STEP(F2, d, a, b, c, in[14]+0xc33707d6,  9);
135     MD5STEP(F2, c, d, a, b, in[ 3]+0xf4d50d87, 14);
136     MD5STEP(F2, b, c, d, a, in[ 8]+0x455a14ed, 20);
137     MD5STEP(F2, a, b, c, d, in[13]+0xa9e3e905,  5);
138     MD5STEP(F2, d, a, b, c, in[ 2]+0xfcefa3f8,  9);
139     MD5STEP(F2, c, d, a, b, in[ 7]+0x676f02d9, 14);
140     MD5STEP(F2, b, c, d, a, in[12]+0x8d2a4c8a, 20);
141
142     MD5STEP(F3, a, b, c, d, in[ 5]+0xfffa3942,  4);
143     MD5STEP(F3, d, a, b, c, in[ 8]+0x8771f681, 11);
144     MD5STEP(F3, c, d, a, b, in[11]+0x6d9d6122, 16);
145     MD5STEP(F3, b, c, d, a, in[14]+0xfde5380c, 23);
146     MD5STEP(F3, a, b, c, d, in[ 1]+0xa4beea44,  4);
147     MD5STEP(F3, d, a, b, c, in[ 4]+0x4bdecfa9, 11);
148     MD5STEP(F3, c, d, a, b, in[ 7]+0xf6bb4b60, 16);
149     MD5STEP(F3, b, c, d, a, in[10]+0xbebfbc70, 23);
150     MD5STEP(F3, a, b, c, d, in[13]+0x289b7ec6,  4);
151     MD5STEP(F3, d, a, b, c, in[ 0]+0xeaa127fa, 11);
152     MD5STEP(F3, c, d, a, b, in[ 3]+0xd4ef3085, 16);
153     MD5STEP(F3, b, c, d, a, in[ 6]+0x04881d05, 23);
154     MD5STEP(F3, a, b, c, d, in[ 9]+0xd9d4d039,  4);
155     MD5STEP(F3, d, a, b, c, in[12]+0xe6db99e5, 11);
156     MD5STEP(F3, c, d, a, b, in[15]+0x1fa27cf8, 16);
157     MD5STEP(F3, b, c, d, a, in[ 2]+0xc4ac5665, 23);
158
159     MD5STEP(F4, a, b, c, d, in[ 0]+0xf4292244,  6);
160     MD5STEP(F4, d, a, b, c, in[ 7]+0x432aff97, 10);
161     MD5STEP(F4, c, d, a, b, in[14]+0xab9423a7, 15);
162     MD5STEP(F4, b, c, d, a, in[ 5]+0xfc93a039, 21);
163     MD5STEP(F4, a, b, c, d, in[12]+0x655b59c3,  6);
164     MD5STEP(F4, d, a, b, c, in[ 3]+0x8f0ccc92, 10);
165     MD5STEP(F4, c, d, a, b, in[10]+0xffeff47d, 15);
166     MD5STEP(F4, b, c, d, a, in[ 1]+0x85845dd1, 21);
167     MD5STEP(F4, a, b, c, d, in[ 8]+0x6fa87e4f,  6);
168     MD5STEP(F4, d, a, b, c, in[15]+0xfe2ce6e0, 10);
169     MD5STEP(F4, c, d, a, b, in[ 6]+0xa3014314, 15);
170     MD5STEP(F4, b, c, d, a, in[13]+0x4e0811a1, 21);
171     MD5STEP(F4, a, b, c, d, in[ 4]+0xf7537e82,  6);
172     MD5STEP(F4, d, a, b, c, in[11]+0xbd3af235, 10);
173     MD5STEP(F4, c, d, a, b, in[ 2]+0x2ad7d2bb, 15);
174     MD5STEP(F4, b, c, d, a, in[ 9]+0xeb86d391, 21);
175
176     buf[0] += a;
177     buf[1] += b;
178     buf[2] += c;
179     buf[3] += d;
180 }
181
182 MD5::MD5()
183 {
184     m_buf[0] = 0x67452301;
185     m_buf[1] = 0xefcdab89;
186     m_buf[2] = 0x98badcfe;
187     m_buf[3] = 0x10325476;
188     m_bits[0] = 0;
189     m_bits[1] = 0;
190     memset(m_in, 0, sizeof(m_in));
191     ASSERT_WITH_MESSAGE(!(reinterpret_cast<uintptr_t>(m_in) % sizeof(uint32_t)), "alignment error of m_in");
192 }
193
194 void MD5::addBytes(const uint8_t* input, size_t length)
195 {
196     const uint8_t* buf = input;
197
198     // Update bitcount
199     uint32_t t = m_bits[0];
200     m_bits[0] = t + (length << 3);
201     if (m_bits[0] < t)
202         m_bits[1]++; // Carry from low to high
203     m_bits[1] += length >> 29;
204
205     t = (t >> 3) & 0x3f; // Bytes already in shsInfo->data
206
207     // Handle any leading odd-sized chunks
208
209     if (t) {
210         uint8_t* p = m_in + t;
211
212         t = 64 - t;
213         if (length < t) {
214             memcpy(p, buf, length);
215             return;
216         }
217         memcpy(p, buf, t);
218         toLittleEndian(m_in, 16);
219         MD5Transform(m_buf, reinterpret_cast_ptr<uint32_t*>(m_in)); // m_in is 4-byte aligned.
220         buf += t;
221         length -= t;
222     }
223
224     // Process data in 64-byte chunks
225
226     while (length >= 64) {
227         memcpy(m_in, buf, 64);
228         toLittleEndian(m_in, 16);
229         MD5Transform(m_buf, reinterpret_cast_ptr<uint32_t*>(m_in)); // m_in is 4-byte aligned.
230         buf += 64;
231         length -= 64;
232     }
233
234     // Handle any remaining bytes of data.
235     memcpy(m_in, buf, length);
236 }
237
238 void MD5::checksum(Digest& digest)
239 {
240     // Compute number of bytes mod 64
241     unsigned count = (m_bits[0] >> 3) & 0x3F;
242
243     // Set the first char of padding to 0x80.  This is safe since there is
244     // always at least one byte free
245     uint8_t* p = m_in + count;
246     *p++ = 0x80;
247
248     // Bytes of padding needed to make 64 bytes
249     count = 64 - 1 - count;
250
251     // Pad out to 56 mod 64
252     if (count < 8) {
253         // Two lots of padding:  Pad the first block to 64 bytes
254         memset(p, 0, count);
255         toLittleEndian(m_in, 16);
256         MD5Transform(m_buf, reinterpret_cast_ptr<uint32_t *>(m_in)); // m_in is 4-byte aligned.
257
258         // Now fill the next block with 56 bytes
259         memset(m_in, 0, 56);
260     } else {
261         // Pad block to 56 bytes
262         memset(p, 0, count - 8);
263     }
264     toLittleEndian(m_in, 14);
265
266     // Append length in bits and transform
267     memcpy(m_in + 56, m_bits, sizeof(m_bits));
268
269     MD5Transform(m_buf, reinterpret_cast_ptr<uint32_t*>(m_in));
270     toLittleEndian(reinterpret_cast<uint8_t*>(m_buf), 4);
271
272     // Now, m_buf contains checksum result.
273     uint8_t* mBufUInt8 = reinterpret_cast<uint8_t*>(m_buf);
274     for (size_t i = 0; i < hashSize; ++i)
275         digest[i] = mBufUInt8[i];
276
277     // In case it's sensitive
278     memset(m_buf, 0, sizeof(m_buf));
279     memset(m_bits, 0, sizeof(m_bits));
280     memset(m_in, 0, sizeof(m_in));
281 }
282
283 #endif
284
285 } // namespace WTF