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 "crypto/sha2.h"
6
7#include "base/basictypes.h"
8#include "testing/gtest/include/gtest/gtest.h"
9
10TEST(Sha256Test, Test1) {
11  // Example B.1 from FIPS 180-2: one-block message.
12  std::string input1 = "abc";
13  int expected1[] = { 0xba, 0x78, 0x16, 0xbf,
14                      0x8f, 0x01, 0xcf, 0xea,
15                      0x41, 0x41, 0x40, 0xde,
16                      0x5d, 0xae, 0x22, 0x23,
17                      0xb0, 0x03, 0x61, 0xa3,
18                      0x96, 0x17, 0x7a, 0x9c,
19                      0xb4, 0x10, 0xff, 0x61,
20                      0xf2, 0x00, 0x15, 0xad };
21
22  uint8 output1[crypto::SHA256_LENGTH];
23  crypto::SHA256HashString(input1, output1, sizeof(output1));
24  for (size_t i = 0; i < crypto::SHA256_LENGTH; i++)
25    EXPECT_EQ(expected1[i], static_cast<int>(output1[i]));
26
27  uint8 output_truncated1[4];  // 4 bytes == 32 bits
28  crypto::SHA256HashString(input1,
29                           output_truncated1, sizeof(output_truncated1));
30  for (size_t i = 0; i < sizeof(output_truncated1); i++)
31    EXPECT_EQ(expected1[i], static_cast<int>(output_truncated1[i]));
32}
33
34TEST(Sha256Test, Test1_String) {
35  // Same as the above, but using the wrapper that returns a std::string.
36  // Example B.1 from FIPS 180-2: one-block message.
37  std::string input1 = "abc";
38  int expected1[] = { 0xba, 0x78, 0x16, 0xbf,
39                      0x8f, 0x01, 0xcf, 0xea,
40                      0x41, 0x41, 0x40, 0xde,
41                      0x5d, 0xae, 0x22, 0x23,
42                      0xb0, 0x03, 0x61, 0xa3,
43                      0x96, 0x17, 0x7a, 0x9c,
44                      0xb4, 0x10, 0xff, 0x61,
45                      0xf2, 0x00, 0x15, 0xad };
46
47  std::string output1 = crypto::SHA256HashString(input1);
48  ASSERT_EQ(crypto::SHA256_LENGTH, output1.size());
49  for (size_t i = 0; i < crypto::SHA256_LENGTH; i++)
50    EXPECT_EQ(expected1[i], static_cast<uint8>(output1[i]));
51}
52
53TEST(Sha256Test, Test2) {
54  // Example B.2 from FIPS 180-2: multi-block message.
55  std::string input2 =
56      "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";
57  int expected2[] = { 0x24, 0x8d, 0x6a, 0x61,
58                      0xd2, 0x06, 0x38, 0xb8,
59                      0xe5, 0xc0, 0x26, 0x93,
60                      0x0c, 0x3e, 0x60, 0x39,
61                      0xa3, 0x3c, 0xe4, 0x59,
62                      0x64, 0xff, 0x21, 0x67,
63                      0xf6, 0xec, 0xed, 0xd4,
64                      0x19, 0xdb, 0x06, 0xc1 };
65
66  uint8 output2[crypto::SHA256_LENGTH];
67  crypto::SHA256HashString(input2, output2, sizeof(output2));
68  for (size_t i = 0; i < crypto::SHA256_LENGTH; i++)
69    EXPECT_EQ(expected2[i], static_cast<int>(output2[i]));
70
71  uint8 output_truncated2[6];
72  crypto::SHA256HashString(input2,
73                           output_truncated2, sizeof(output_truncated2));
74  for (size_t i = 0; i < sizeof(output_truncated2); i++)
75    EXPECT_EQ(expected2[i], static_cast<int>(output_truncated2[i]));
76}
77
78TEST(Sha256Test, Test3) {
79  // Example B.3 from FIPS 180-2: long message.
80  std::string input3(1000000, 'a');  // 'a' repeated a million times
81  int expected3[] = { 0xcd, 0xc7, 0x6e, 0x5c,
82                      0x99, 0x14, 0xfb, 0x92,
83                      0x81, 0xa1, 0xc7, 0xe2,
84                      0x84, 0xd7, 0x3e, 0x67,
85                      0xf1, 0x80, 0x9a, 0x48,
86                      0xa4, 0x97, 0x20, 0x0e,
87                      0x04, 0x6d, 0x39, 0xcc,
88                      0xc7, 0x11, 0x2c, 0xd0 };
89
90  uint8 output3[crypto::SHA256_LENGTH];
91  crypto::SHA256HashString(input3, output3, sizeof(output3));
92  for (size_t i = 0; i < crypto::SHA256_LENGTH; i++)
93    EXPECT_EQ(expected3[i], static_cast<int>(output3[i]));
94
95  uint8 output_truncated3[12];
96  crypto::SHA256HashString(input3,
97                           output_truncated3, sizeof(output_truncated3));
98  for (size_t i = 0; i < sizeof(output_truncated3); i++)
99    EXPECT_EQ(expected3[i], static_cast<int>(output_truncated3[i]));
100}
101