Chromium Code Reviews| Index: base/sha1_unittest.cc |
| diff --git a/base/sha1_unittest.cc b/base/sha1_unittest.cc |
| index b29fe4662aef30636ecf0086bca33cc6026887cf..67983d12d1c6e84ce5db35168af8e6871b1f81ac 100644 |
| --- a/base/sha1_unittest.cc |
| +++ b/base/sha1_unittest.cc |
| @@ -7,6 +7,7 @@ |
| #include <string> |
| #include "base/basictypes.h" |
| +#include "base/files/scoped_file.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| TEST(SHA1Test, Test1) { |
| @@ -106,3 +107,64 @@ TEST(SHA1Test, Test3Bytes) { |
| for (size_t i = 0; i < base::kSHA1Length; i++) |
| EXPECT_EQ(expected[i], output[i]); |
| } |
| + |
| +// Helper function to test SHA1HashFile. |
| +void doHashFile(const std::string& input, unsigned char *output) { |
|
Petr Hosek
2015/12/02 22:51:31
Nit: * belongs to type.
Sean Klein
2015/12/02 23:01:22
Done.
|
| + std::string file_name = "sha1_test_data.txt"; |
| + base::ScopedFILE file(fopen(file_name.c_str(), "w+")); |
| + EXPECT_NE(nullptr, file); |
| + EXPECT_EQ(input.length(), |
| + fwrite(input.c_str(), 1, input.length(), file.get())); |
|
Petr Hosek
2015/12/02 22:51:31
Nit: I'd use sizeof(char) instead of 1.
Sean Klein
2015/12/02 23:01:22
Done.
|
| + EXPECT_EQ(0, fflush(file.get())); |
| + EXPECT_EQ(true, base::SHA1HashFile(file_name, output)); |
| + EXPECT_EQ(0, unlink(file_name.c_str())); |
| +} |
| + |
| +TEST(SHA1Test, Test1File) { |
| + // Example A.1 from FIPS 180-2: one-block message. |
| + std::string input = "abc"; |
| + unsigned char output[base::kSHA1Length]; |
| + |
| + unsigned char expected[] = { 0xa9, 0x99, 0x3e, 0x36, |
|
Petr Hosek
2015/12/02 22:51:31
This should be const.
Sean Klein
2015/12/02 23:01:22
Done.
|
| + 0x47, 0x06, 0x81, 0x6a, |
| + 0xba, 0x3e, 0x25, 0x71, |
| + 0x78, 0x50, 0xc2, 0x6c, |
| + 0x9c, 0xd0, 0xd8, 0x9d }; |
| + |
| + doHashFile(input, output); |
| + for (size_t i = 0; i < base::kSHA1Length; i++) |
| + EXPECT_EQ(expected[i], output[i]); |
| +} |
| + |
| +TEST(SHA1Test, Test2File) { |
| + // Example A.2 from FIPS 180-2: multi-block message. |
| + std::string input = |
| + "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"; |
| + unsigned char output[base::kSHA1Length]; |
| + |
| + unsigned char expected[] = { 0x84, 0x98, 0x3e, 0x44, |
|
Petr Hosek
2015/12/02 22:51:31
ditto
Sean Klein
2015/12/02 23:01:21
Done.
|
| + 0x1c, 0x3b, 0xd2, 0x6e, |
| + 0xba, 0xae, 0x4a, 0xa1, |
| + 0xf9, 0x51, 0x29, 0xe5, |
| + 0xe5, 0x46, 0x70, 0xf1 }; |
| + |
| + doHashFile(input, output); |
| + for (size_t i = 0; i < base::kSHA1Length; i++) |
| + EXPECT_EQ(expected[i], output[i]); |
| +} |
| + |
| +TEST(SHA1Test, Test3File) { |
| + // Example A.3 from FIPS 180-2: long message. |
| + std::string input(1000000, 'a'); |
| + unsigned char output[base::kSHA1Length]; |
| + |
| + unsigned char expected[] = { 0x34, 0xaa, 0x97, 0x3c, |
|
Petr Hosek
2015/12/02 22:51:32
ditto
Sean Klein
2015/12/02 23:01:22
Done.
|
| + 0xd4, 0xc4, 0xda, 0xa4, |
| + 0xf6, 0x1e, 0xeb, 0x2b, |
| + 0xdb, 0xad, 0x27, 0x31, |
| + 0x65, 0x34, 0x01, 0x6f }; |
| + |
| + doHashFile(input, output); |
| + for (size_t i = 0; i < base::kSHA1Length; i++) |
| + EXPECT_EQ(expected[i], output[i]); |
| +} |