crypto/sha2_unittest.cc

   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
  10 TEST(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
  34 TEST(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
  53 TEST(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
  78 TEST(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 }