1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <string.h>
6 #include <string>
7
8 #include "base/memory/scoped_ptr.h"
9 #include "base/md5.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace base {
13
TEST(MD5,DigestToBase16)14 TEST(MD5, DigestToBase16) {
15 MD5Digest digest;
16
17 int data[] = {
18 0xd4, 0x1d, 0x8c, 0xd9,
19 0x8f, 0x00, 0xb2, 0x04,
20 0xe9, 0x80, 0x09, 0x98,
21 0xec, 0xf8, 0x42, 0x7e
22 };
23
24 for (int i = 0; i < 16; ++i)
25 digest.a[i] = data[i] & 0xff;
26
27 std::string actual = MD5DigestToBase16(digest);
28 std::string expected = "d41d8cd98f00b204e9800998ecf8427e";
29
30 EXPECT_EQ(expected, actual);
31 }
32
TEST(MD5,MD5SumEmtpyData)33 TEST(MD5, MD5SumEmtpyData) {
34 MD5Digest digest;
35 const char data[] = "";
36
37 MD5Sum(data, strlen(data), &digest);
38
39 int expected[] = {
40 0xd4, 0x1d, 0x8c, 0xd9,
41 0x8f, 0x00, 0xb2, 0x04,
42 0xe9, 0x80, 0x09, 0x98,
43 0xec, 0xf8, 0x42, 0x7e
44 };
45
46 for (int i = 0; i < 16; ++i)
47 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
48 }
49
TEST(MD5,MD5SumOneByteData)50 TEST(MD5, MD5SumOneByteData) {
51 MD5Digest digest;
52 const char data[] = "a";
53
54 MD5Sum(data, strlen(data), &digest);
55
56 int expected[] = {
57 0x0c, 0xc1, 0x75, 0xb9,
58 0xc0, 0xf1, 0xb6, 0xa8,
59 0x31, 0xc3, 0x99, 0xe2,
60 0x69, 0x77, 0x26, 0x61
61 };
62
63 for (int i = 0; i < 16; ++i)
64 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
65 }
66
TEST(MD5,MD5SumLongData)67 TEST(MD5, MD5SumLongData) {
68 const int length = 10 * 1024 * 1024 + 1;
69 scoped_ptr<char[]> data(new char[length]);
70
71 for (int i = 0; i < length; ++i)
72 data[i] = i & 0xFF;
73
74 MD5Digest digest;
75 MD5Sum(data.get(), length, &digest);
76
77 int expected[] = {
78 0x90, 0xbd, 0x6a, 0xd9,
79 0x0a, 0xce, 0xf5, 0xad,
80 0xaa, 0x92, 0x20, 0x3e,
81 0x21, 0xc7, 0xa1, 0x3e
82 };
83
84 for (int i = 0; i < 16; ++i)
85 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
86 }
87
TEST(MD5,ContextWithEmptyData)88 TEST(MD5, ContextWithEmptyData) {
89 MD5Context ctx;
90 MD5Init(&ctx);
91
92 MD5Digest digest;
93 MD5Final(&digest, &ctx);
94
95 int expected[] = {
96 0xd4, 0x1d, 0x8c, 0xd9,
97 0x8f, 0x00, 0xb2, 0x04,
98 0xe9, 0x80, 0x09, 0x98,
99 0xec, 0xf8, 0x42, 0x7e
100 };
101
102 for (int i = 0; i < 16; ++i)
103 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
104 }
105
TEST(MD5,ContextWithLongData)106 TEST(MD5, ContextWithLongData) {
107 MD5Context ctx;
108 MD5Init(&ctx);
109
110 const int length = 10 * 1024 * 1024 + 1;
111 scoped_ptr<char[]> data(new char[length]);
112
113 for (int i = 0; i < length; ++i)
114 data[i] = i & 0xFF;
115
116 int total = 0;
117 while (total < length) {
118 int len = 4097; // intentionally not 2^k.
119 if (len > length - total)
120 len = length - total;
121
122 MD5Update(&ctx,
123 StringPiece(reinterpret_cast<char*>(data.get() + total), len));
124 total += len;
125 }
126
127 EXPECT_EQ(length, total);
128
129 MD5Digest digest;
130 MD5Final(&digest, &ctx);
131
132 int expected[] = {
133 0x90, 0xbd, 0x6a, 0xd9,
134 0x0a, 0xce, 0xf5, 0xad,
135 0xaa, 0x92, 0x20, 0x3e,
136 0x21, 0xc7, 0xa1, 0x3e
137 };
138
139 for (int i = 0; i < 16; ++i)
140 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
141 }
142
143 // Example data from http://www.ietf.org/rfc/rfc1321.txt A.5 Test Suite
TEST(MD5,MD5StringTestSuite1)144 TEST(MD5, MD5StringTestSuite1) {
145 std::string actual = MD5String("");
146 std::string expected = "d41d8cd98f00b204e9800998ecf8427e";
147 EXPECT_EQ(expected, actual);
148 }
149
TEST(MD5,MD5StringTestSuite2)150 TEST(MD5, MD5StringTestSuite2) {
151 std::string actual = MD5String("a");
152 std::string expected = "0cc175b9c0f1b6a831c399e269772661";
153 EXPECT_EQ(expected, actual);
154 }
155
TEST(MD5,MD5StringTestSuite3)156 TEST(MD5, MD5StringTestSuite3) {
157 std::string actual = MD5String("abc");
158 std::string expected = "900150983cd24fb0d6963f7d28e17f72";
159 EXPECT_EQ(expected, actual);
160 }
161
TEST(MD5,MD5StringTestSuite4)162 TEST(MD5, MD5StringTestSuite4) {
163 std::string actual = MD5String("message digest");
164 std::string expected = "f96b697d7cb7938d525a2f31aaf161d0";
165 EXPECT_EQ(expected, actual);
166 }
167
TEST(MD5,MD5StringTestSuite5)168 TEST(MD5, MD5StringTestSuite5) {
169 std::string actual = MD5String("abcdefghijklmnopqrstuvwxyz");
170 std::string expected = "c3fcd3d76192e4007dfb496cca67e13b";
171 EXPECT_EQ(expected, actual);
172 }
173
TEST(MD5,MD5StringTestSuite6)174 TEST(MD5, MD5StringTestSuite6) {
175 std::string actual = MD5String("ABCDEFGHIJKLMNOPQRSTUVWXYZ"
176 "abcdefghijklmnopqrstuvwxyz"
177 "0123456789");
178 std::string expected = "d174ab98d277d9f5a5611c2c9f419d9f";
179 EXPECT_EQ(expected, actual);
180 }
181
TEST(MD5,MD5StringTestSuite7)182 TEST(MD5, MD5StringTestSuite7) {
183 std::string actual = MD5String("12345678901234567890"
184 "12345678901234567890"
185 "12345678901234567890"
186 "12345678901234567890");
187 std::string expected = "57edf4a22be3c955ac49da2e2107b67a";
188 EXPECT_EQ(expected, actual);
189 }
190
TEST(MD5,ContextWithStringData)191 TEST(MD5, ContextWithStringData) {
192 MD5Context ctx;
193 MD5Init(&ctx);
194
195 MD5Update(&ctx, "abc");
196
197 MD5Digest digest;
198 MD5Final(&digest, &ctx);
199
200 std::string actual = MD5DigestToBase16(digest);
201 std::string expected = "900150983cd24fb0d6963f7d28e17f72";
202
203 EXPECT_EQ(expected, actual);
204 }
205
206 // Test that a digest generated by MD5IntermediateFinal() gives the same results
207 // as an independently-calculated digest, and also does not modify the context.
TEST(MD5,IntermediateFinal)208 TEST(MD5, IntermediateFinal) {
209 // Independent context over the header.
210 MD5Context check_header_context;
211 MD5Init(&check_header_context);
212
213 // Independent context over entire input.
214 MD5Context check_full_context;
215 MD5Init(&check_full_context);
216
217 // Context intermediate digest will be calculated from.
218 MD5Context context;
219 MD5Init(&context);
220
221 static const char kHeader[] = "header data";
222 static const char kBody[] = "payload data";
223
224 MD5Update(&context, kHeader);
225 MD5Update(&check_header_context, kHeader);
226 MD5Update(&check_full_context, kHeader);
227
228 MD5Digest check_header_digest;
229 MD5Final(&check_header_digest, &check_header_context);
230
231 MD5Digest header_digest;
232 MD5IntermediateFinal(&header_digest, &context);
233
234 MD5Update(&context, kBody);
235 MD5Update(&check_full_context, kBody);
236
237 MD5Digest check_full_digest;
238 MD5Final(&check_full_digest, &check_full_context);
239
240 MD5Digest digest;
241 MD5Final(&digest, &context);
242
243 // The header and full digest pairs are the same, and they aren't the same as
244 // each other.
245 EXPECT_TRUE(!memcmp(&header_digest, &check_header_digest,
246 sizeof(header_digest)));
247 EXPECT_TRUE(!memcmp(&digest, &check_full_digest, sizeof(digest)));
248 EXPECT_TRUE(memcmp(&digest, &header_digest, sizeof(digest)));
249 }
250
251 } // namespace base
252