1 // Copyright 2011 The Chromium Authors
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 <memory>
7 #include <string>
8
9 #include "base/hash/md5.h"
10
11 #include "testing/gtest/include/gtest/gtest.h"
12
13 namespace base {
14
TEST(MD5,DigestToBase16)15 TEST(MD5, DigestToBase16) {
16 MD5Digest digest;
17
18 int data[] = {0xd4, 0x1d, 0x8c, 0xd9, 0x8f, 0x00, 0xb2, 0x04,
19 0xe9, 0x80, 0x09, 0x98, 0xec, 0xf8, 0x42, 0x7e};
20
21 for (int i = 0; i < 16; ++i)
22 digest.a[i] = data[i] & 0xff;
23
24 std::string actual = MD5DigestToBase16(digest);
25 std::string expected = "d41d8cd98f00b204e9800998ecf8427e";
26
27 EXPECT_EQ(expected, actual);
28 }
29
TEST(MD5,MD5SumEmtpyData)30 TEST(MD5, MD5SumEmtpyData) {
31 MD5Digest digest;
32 const char data[] = "";
33
34 MD5Sum(data, strlen(data), &digest);
35
36 int expected[] = {0xd4, 0x1d, 0x8c, 0xd9, 0x8f, 0x00, 0xb2, 0x04,
37 0xe9, 0x80, 0x09, 0x98, 0xec, 0xf8, 0x42, 0x7e};
38
39 for (int i = 0; i < 16; ++i)
40 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
41 }
42
TEST(MD5,MD5SumOneByteData)43 TEST(MD5, MD5SumOneByteData) {
44 MD5Digest digest;
45 const char data[] = "a";
46
47 MD5Sum(data, strlen(data), &digest);
48
49 int expected[] = {0x0c, 0xc1, 0x75, 0xb9, 0xc0, 0xf1, 0xb6, 0xa8,
50 0x31, 0xc3, 0x99, 0xe2, 0x69, 0x77, 0x26, 0x61};
51
52 for (int i = 0; i < 16; ++i)
53 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
54 }
55
TEST(MD5,MD5SumLongData)56 TEST(MD5, MD5SumLongData) {
57 const int length = 10 * 1024 * 1024 + 1;
58 std::unique_ptr<char[]> data(new char[length]);
59
60 for (int i = 0; i < length; ++i)
61 data[i] = i & 0xFF;
62
63 MD5Digest digest;
64 MD5Sum(data.get(), length, &digest);
65
66 int expected[] = {0x90, 0xbd, 0x6a, 0xd9, 0x0a, 0xce, 0xf5, 0xad,
67 0xaa, 0x92, 0x20, 0x3e, 0x21, 0xc7, 0xa1, 0x3e};
68
69 for (int i = 0; i < 16; ++i)
70 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
71 }
72
TEST(MD5,ContextWithEmptyData)73 TEST(MD5, ContextWithEmptyData) {
74 MD5Context ctx;
75 MD5Init(&ctx);
76
77 MD5Digest digest;
78 MD5Final(&digest, &ctx);
79
80 int expected[] = {0xd4, 0x1d, 0x8c, 0xd9, 0x8f, 0x00, 0xb2, 0x04,
81 0xe9, 0x80, 0x09, 0x98, 0xec, 0xf8, 0x42, 0x7e};
82
83 for (int i = 0; i < 16; ++i)
84 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
85 }
86
TEST(MD5,ContextWithLongData)87 TEST(MD5, ContextWithLongData) {
88 MD5Context ctx;
89 MD5Init(&ctx);
90
91 const int length = 10 * 1024 * 1024 + 1;
92 std::unique_ptr<char[]> data(new char[length]);
93
94 for (int i = 0; i < length; ++i)
95 data[i] = i & 0xFF;
96
97 int total = 0;
98 while (total < length) {
99 int len = 4097; // intentionally not 2^k.
100 if (len > length - total)
101 len = length - total;
102
103 MD5Update(&ctx,
104 StringPiece(reinterpret_cast<char*>(data.get() + total), len));
105 total += len;
106 }
107
108 EXPECT_EQ(length, total);
109
110 MD5Digest digest;
111 MD5Final(&digest, &ctx);
112
113 int expected[] = {0x90, 0xbd, 0x6a, 0xd9, 0x0a, 0xce, 0xf5, 0xad,
114 0xaa, 0x92, 0x20, 0x3e, 0x21, 0xc7, 0xa1, 0x3e};
115
116 for (int i = 0; i < 16; ++i)
117 EXPECT_EQ(expected[i], digest.a[i] & 0xFF);
118 }
119
120 // Example data from http://www.ietf.org/rfc/rfc1321.txt A.5 Test Suite
TEST(MD5,MD5StringTestSuite1)121 TEST(MD5, MD5StringTestSuite1) {
122 std::string actual = MD5String("");
123 std::string expected = "d41d8cd98f00b204e9800998ecf8427e";
124 EXPECT_EQ(expected, actual);
125 }
126
TEST(MD5,MD5StringTestSuite2)127 TEST(MD5, MD5StringTestSuite2) {
128 std::string actual = MD5String("a");
129 std::string expected = "0cc175b9c0f1b6a831c399e269772661";
130 EXPECT_EQ(expected, actual);
131 }
132
TEST(MD5,MD5StringTestSuite3)133 TEST(MD5, MD5StringTestSuite3) {
134 std::string actual = MD5String("abc");
135 std::string expected = "900150983cd24fb0d6963f7d28e17f72";
136 EXPECT_EQ(expected, actual);
137 }
138
TEST(MD5,MD5StringTestSuite4)139 TEST(MD5, MD5StringTestSuite4) {
140 std::string actual = MD5String("message digest");
141 std::string expected = "f96b697d7cb7938d525a2f31aaf161d0";
142 EXPECT_EQ(expected, actual);
143 }
144
TEST(MD5,MD5StringTestSuite5)145 TEST(MD5, MD5StringTestSuite5) {
146 std::string actual = MD5String("abcdefghijklmnopqrstuvwxyz");
147 std::string expected = "c3fcd3d76192e4007dfb496cca67e13b";
148 EXPECT_EQ(expected, actual);
149 }
150
TEST(MD5,MD5StringTestSuite6)151 TEST(MD5, MD5StringTestSuite6) {
152 std::string actual = MD5String(
153 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
154 "abcdefghijklmnopqrstuvwxyz"
155 "0123456789");
156 std::string expected = "d174ab98d277d9f5a5611c2c9f419d9f";
157 EXPECT_EQ(expected, actual);
158 }
159
TEST(MD5,MD5StringTestSuite7)160 TEST(MD5, MD5StringTestSuite7) {
161 std::string actual = MD5String(
162 "12345678901234567890"
163 "12345678901234567890"
164 "12345678901234567890"
165 "12345678901234567890");
166 std::string expected = "57edf4a22be3c955ac49da2e2107b67a";
167 EXPECT_EQ(expected, actual);
168 }
169
TEST(MD5,ContextWithStringData)170 TEST(MD5, ContextWithStringData) {
171 MD5Context ctx;
172 MD5Init(&ctx);
173
174 MD5Update(&ctx, "abc");
175
176 MD5Digest digest;
177 MD5Final(&digest, &ctx);
178
179 std::string actual = MD5DigestToBase16(digest);
180 std::string expected = "900150983cd24fb0d6963f7d28e17f72";
181
182 EXPECT_EQ(expected, actual);
183 }
184
185 } // namespace base
186