md5_unittest.cc 4.7 KB

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