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