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