sha1_unittest.cc 3.72 KB
Newer Older
gejun's avatar
gejun committed
1 2 3 4
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

5
#include "butil/sha1.h"
gejun's avatar
gejun committed
6 7 8

#include <string>

9
#include "butil/basictypes.h"
gejun's avatar
gejun committed
10 11 12 13 14 15 16 17 18 19 20 21
#include <gtest/gtest.h>

TEST(SHA1Test, Test1) {
  // Example A.1 from FIPS 180-2: one-block message.
  std::string input = "abc";

  int expected[] = { 0xa9, 0x99, 0x3e, 0x36,
                     0x47, 0x06, 0x81, 0x6a,
                     0xba, 0x3e, 0x25, 0x71,
                     0x78, 0x50, 0xc2, 0x6c,
                     0x9c, 0xd0, 0xd8, 0x9d };

22 23
  std::string output = butil::SHA1HashString(input);
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
24 25 26 27 28 29 30 31 32 33 34 35 36 37
    EXPECT_EQ(expected[i], output[i] & 0xFF);
}

TEST(SHA1Test, Test2) {
  // Example A.2 from FIPS 180-2: multi-block message.
  std::string input =
      "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";

  int expected[] = { 0x84, 0x98, 0x3e, 0x44,
                     0x1c, 0x3b, 0xd2, 0x6e,
                     0xba, 0xae, 0x4a, 0xa1,
                     0xf9, 0x51, 0x29, 0xe5,
                     0xe5, 0x46, 0x70, 0xf1 };

38 39
  std::string output = butil::SHA1HashString(input);
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
40 41 42 43 44 45 46 47 48 49 50 51 52
    EXPECT_EQ(expected[i], output[i] & 0xFF);
}

TEST(SHA1Test, Test3) {
  // Example A.3 from FIPS 180-2: long message.
  std::string input(1000000, 'a');

  int expected[] = { 0x34, 0xaa, 0x97, 0x3c,
                     0xd4, 0xc4, 0xda, 0xa4,
                     0xf6, 0x1e, 0xeb, 0x2b,
                     0xdb, 0xad, 0x27, 0x31,
                     0x65, 0x34, 0x01, 0x6f };

53 54
  std::string output = butil::SHA1HashString(input);
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
55 56 57 58 59 60
    EXPECT_EQ(expected[i], output[i] & 0xFF);
}

TEST(SHA1Test, Test1Bytes) {
  // Example A.1 from FIPS 180-2: one-block message.
  std::string input = "abc";
61
  unsigned char output[butil::kSHA1Length];
gejun's avatar
gejun committed
62 63 64 65 66 67 68

  unsigned char expected[] = { 0xa9, 0x99, 0x3e, 0x36,
                               0x47, 0x06, 0x81, 0x6a,
                               0xba, 0x3e, 0x25, 0x71,
                               0x78, 0x50, 0xc2, 0x6c,
                               0x9c, 0xd0, 0xd8, 0x9d };

69
  butil::SHA1HashBytes(reinterpret_cast<const unsigned char*>(input.c_str()),
gejun's avatar
gejun committed
70
                      input.length(), output);
71
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
72 73 74 75 76 77 78
    EXPECT_EQ(expected[i], output[i]);
}

TEST(SHA1Test, Test2Bytes) {
  // Example A.2 from FIPS 180-2: multi-block message.
  std::string input =
      "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";
79
  unsigned char output[butil::kSHA1Length];
gejun's avatar
gejun committed
80 81 82 83 84 85 86

  unsigned char expected[] = { 0x84, 0x98, 0x3e, 0x44,
                               0x1c, 0x3b, 0xd2, 0x6e,
                               0xba, 0xae, 0x4a, 0xa1,
                               0xf9, 0x51, 0x29, 0xe5,
                               0xe5, 0x46, 0x70, 0xf1 };

87
  butil::SHA1HashBytes(reinterpret_cast<const unsigned char*>(input.c_str()),
gejun's avatar
gejun committed
88
                      input.length(), output);
89
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
90 91 92 93 94 95
    EXPECT_EQ(expected[i], output[i]);
}

TEST(SHA1Test, Test3Bytes) {
  // Example A.3 from FIPS 180-2: long message.
  std::string input(1000000, 'a');
96
  unsigned char output[butil::kSHA1Length];
gejun's avatar
gejun committed
97 98 99 100 101 102 103

  unsigned char expected[] = { 0x34, 0xaa, 0x97, 0x3c,
                               0xd4, 0xc4, 0xda, 0xa4,
                               0xf6, 0x1e, 0xeb, 0x2b,
                               0xdb, 0xad, 0x27, 0x31,
                               0x65, 0x34, 0x01, 0x6f };

104
  butil::SHA1HashBytes(reinterpret_cast<const unsigned char*>(input.c_str()),
gejun's avatar
gejun committed
105
                      input.length(), output);
106
  for (size_t i = 0; i < butil::kSHA1Length; i++)
gejun's avatar
gejun committed
107 108
    EXPECT_EQ(expected[i], output[i]);
}