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