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