| // Copyright 2014 The BoringSSL Authors | 
 | // | 
 | // Licensed under the Apache License, Version 2.0 (the "License"); | 
 | // you may not use this file except in compliance with the License. | 
 | // You may obtain a copy of the License at | 
 | // | 
 | //     https://www.apache.org/licenses/LICENSE-2.0 | 
 | // | 
 | // Unless required by applicable law or agreed to in writing, software | 
 | // distributed under the License is distributed on an "AS IS" BASIS, | 
 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
 | // See the License for the specific language governing permissions and | 
 | // limitations under the License. | 
 |  | 
 | #include <stdint.h> | 
 | #include <stdio.h> | 
 | #include <string.h> | 
 |  | 
 | #include <memory> | 
 | #include <vector> | 
 |  | 
 | #include <gtest/gtest.h> | 
 |  | 
 | #include <openssl/asn1.h> | 
 | #include <openssl/bytestring.h> | 
 | #include <openssl/crypto.h> | 
 | #include <openssl/digest.h> | 
 | #include <openssl/err.h> | 
 | #include <openssl/md4.h> | 
 | #include <openssl/md5.h> | 
 | #include <openssl/nid.h> | 
 | #include <openssl/obj.h> | 
 | #include <openssl/sha.h> | 
 |  | 
 | #include "../internal.h" | 
 | #include "../test/test_util.h" | 
 |  | 
 |  | 
 | namespace { | 
 |  | 
 | struct MD { | 
 |   // name is the name of the digest. | 
 |   const char *name; | 
 |   // md_func is the digest to test. | 
 |   const EVP_MD *(*func)(void); | 
 |   // one_shot_func is the convenience one-shot version of the | 
 |   // digest. | 
 |   uint8_t *(*one_shot_func)(const uint8_t *, size_t, uint8_t *); | 
 | }; | 
 |  | 
 | static const MD md4 = {"MD4", &EVP_md4, nullptr}; | 
 | static const MD md5 = {"MD5", &EVP_md5, &MD5}; | 
 | static const MD sha1 = {"SHA1", &EVP_sha1, &SHA1}; | 
 | static const MD sha224 = {"SHA224", &EVP_sha224, &SHA224}; | 
 | static const MD sha256 = {"SHA256", &EVP_sha256, &SHA256}; | 
 | static const MD sha384 = {"SHA384", &EVP_sha384, &SHA384}; | 
 | static const MD sha512 = {"SHA512", &EVP_sha512, &SHA512}; | 
 | static const MD sha512_256 = {"SHA512-256", &EVP_sha512_256, &SHA512_256}; | 
 | static const MD md5_sha1 = {"MD5-SHA1", &EVP_md5_sha1, nullptr}; | 
 | static const MD blake2b256 = {"BLAKE2b-256", &EVP_blake2b256, nullptr}; | 
 |  | 
 | struct DigestTestVector { | 
 |   // md is the digest to test. | 
 |   const MD &md; | 
 |   // input is a NUL-terminated string to hash. | 
 |   const char *input; | 
 |   // repeat is the number of times to repeat input. | 
 |   size_t repeat; | 
 |   // expected_hex is the expected digest in hexadecimal. | 
 |   const char *expected_hex; | 
 | }; | 
 |  | 
 | static const DigestTestVector kTestVectors[] = { | 
 |     // MD4 tests, from RFC 1320. (crypto/md4 does not provide a | 
 |     // one-shot MD4 function.) | 
 |     {md4, "", 1, "31d6cfe0d16ae931b73c59d7e0c089c0"}, | 
 |     {md4, "a", 1, "bde52cb31de33e46245e05fbdbd6fb24"}, | 
 |     {md4, "abc", 1, "a448017aaf21d8525fc10ae87aa6729d"}, | 
 |     {md4, "message digest", 1, "d9130a8164549fe818874806e1c7014b"}, | 
 |     {md4, "abcdefghijklmnopqrstuvwxyz", 1, "d79e1c308aa5bbcdeea8ed63df412da9"}, | 
 |     {md4, "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789", 1, | 
 |      "043f8582f241db351ce627e153e7f0e4"}, | 
 |     {md4, "1234567890", 8, "e33b4ddc9c38f2199c3e7b164fcc0536"}, | 
 |  | 
 |     // MD5 tests, from RFC 1321. | 
 |     {md5, "", 1, "d41d8cd98f00b204e9800998ecf8427e"}, | 
 |     {md5, "a", 1, "0cc175b9c0f1b6a831c399e269772661"}, | 
 |     {md5, "abc", 1, "900150983cd24fb0d6963f7d28e17f72"}, | 
 |     {md5, "message digest", 1, "f96b697d7cb7938d525a2f31aaf161d0"}, | 
 |     {md5, "abcdefghijklmnopqrstuvwxyz", 1, "c3fcd3d76192e4007dfb496cca67e13b"}, | 
 |     {md5, "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789", 1, | 
 |      "d174ab98d277d9f5a5611c2c9f419d9f"}, | 
 |     {md5, "1234567890", 8, "57edf4a22be3c955ac49da2e2107b67a"}, | 
 |  | 
 |     // SHA-1 tests, from RFC 3174. | 
 |     {sha1, "abc", 1, "a9993e364706816aba3e25717850c26c9cd0d89d"}, | 
 |     {sha1, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 1, | 
 |      "84983e441c3bd26ebaae4aa1f95129e5e54670f1"}, | 
 |     {sha1, "a", 1000000, "34aa973cd4c4daa4f61eeb2bdbad27316534016f"}, | 
 |     {sha1, "0123456701234567012345670123456701234567012345670123456701234567", | 
 |      10, "dea356a2cddd90c7a7ecedc5ebb563934f460452"}, | 
 |  | 
 |     // SHA-224 tests, from RFC 3874. | 
 |     {sha224, "abc", 1, | 
 |      "23097d223405d8228642a477bda255b32aadbce4bda0b3f7e36c9da7"}, | 
 |     {sha224, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 1, | 
 |      "75388b16512776cc5dba5da1fd890150b0c6455cb4f58b1952522525"}, | 
 |     {sha224, "a", 1000000, | 
 |      "20794655980c91d8bbb4c1ea97618a4bf03f42581948b2ee4ee7ad67"}, | 
 |  | 
 |     // SHA-256 tests, from NIST. | 
 |     {sha256, "abc", 1, | 
 |      "ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad"}, | 
 |     {sha256, "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq", 1, | 
 |      "248d6a61d20638b8e5c026930c3e6039a33ce45964ff2167f6ecedd419db06c1"}, | 
 |  | 
 |     // SHA-384 tests, from NIST. | 
 |     {sha384, "abc", 1, | 
 |      "cb00753f45a35e8bb5a03d699ac65007272c32ab0eded1631a8b605a43ff5bed" | 
 |      "8086072ba1e7cc2358baeca134c825a7"}, | 
 |     {sha384, | 
 |      "abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmn" | 
 |      "hijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu", | 
 |      1, | 
 |      "09330c33f71147e83d192fc782cd1b4753111b173b3b05d22fa08086e3b0f712" | 
 |      "fcc7c71a557e2db966c3e9fa91746039"}, | 
 |  | 
 |     // SHA-512 tests, from NIST. | 
 |     {sha512, "abc", 1, | 
 |      "ddaf35a193617abacc417349ae20413112e6fa4e89a97ea20a9eeee64b55d39a" | 
 |      "2192992a274fc1a836ba3c23a3feebbd454d4423643ce80e2a9ac94fa54ca49f"}, | 
 |     {sha512, | 
 |      "abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmn" | 
 |      "hijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu", | 
 |      1, | 
 |      "8e959b75dae313da8cf4f72814fc143f8f7779c6eb9f7fa17299aeadb6889018" | 
 |      "501d289e4900f7e4331b99dec4b5433ac7d329eeb6dd26545e96e55b874be909"}, | 
 |  | 
 |     // SHA-512-256 tests, from | 
 |     // https://csrc.nist.gov/csrc/media/projects/cryptographic-standards-and-guidelines/documents/examples/sha512_256.pdf | 
 |     {sha512_256, "abc", 1, | 
 |      "53048e2681941ef99b2e29b76b4c7dabe4c2d0c634fc6d46e0e2f13107e7af23"}, | 
 |     {sha512_256, | 
 |      "abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopj" | 
 |      "klmnopqklmnopqrlmnopqrsmnopqrstnopqrstu", | 
 |      1, "3928e184fb8690f840da3988121d31be65cb9d3ef83ee6146feac861e19b563a"}, | 
 |  | 
 |     // MD5-SHA1 tests. | 
 |     {md5_sha1, "abc", 1, | 
 |      "900150983cd24fb0d6963f7d28e17f72a9993e364706816aba3e25717850c26c9cd0d89" | 
 |      "d"}, | 
 |  | 
 |     // BLAKE2b-256 tests. | 
 |     {blake2b256, "abc", 1, | 
 |      "bddd813c634239723171ef3fee98579b94964e3bb1cb3e427262c8c068d52319"}, | 
 | }; | 
 |  | 
 | static void CompareDigest(const DigestTestVector *test, const uint8_t *digest, | 
 |                           size_t digest_len) { | 
 |   EXPECT_EQ(test->expected_hex, EncodeHex(bssl::Span(digest, digest_len))); | 
 | } | 
 |  | 
 | static void TestDigest(const DigestTestVector *test) { | 
 |   bssl::ScopedEVP_MD_CTX ctx; | 
 |  | 
 |   // Test the input provided. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   for (size_t i = 0; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), test->input, strlen(test->input))); | 
 |   } | 
 |   auto digest = std::make_unique<uint8_t[]>(EVP_MD_size(test->md.func())); | 
 |   unsigned digest_len; | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(ctx.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Test the input one character at a time. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), nullptr, 0)); | 
 |   for (size_t i = 0; i < test->repeat; i++) { | 
 |     for (const char *p = test->input; *p; p++) { | 
 |       ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), p, 1)); | 
 |     } | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(ctx.get(), digest.get(), &digest_len)); | 
 |   EXPECT_EQ(EVP_MD_size(test->md.func()), digest_len); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Test with unaligned input. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   std::vector<char> unaligned(strlen(test->input) + 1); | 
 |   char *ptr = unaligned.data(); | 
 |   if ((reinterpret_cast<uintptr_t>(ptr) & 1) == 0) { | 
 |     ptr++; | 
 |   } | 
 |   OPENSSL_memcpy(ptr, test->input, strlen(test->input)); | 
 |   for (size_t i = 0; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), ptr, strlen(test->input))); | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(ctx.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Make a copy of the digest in the initial state. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   bssl::ScopedEVP_MD_CTX copy; | 
 |   ASSERT_TRUE(EVP_MD_CTX_copy_ex(copy.get(), ctx.get())); | 
 |   for (size_t i = 0; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input, strlen(test->input))); | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(copy.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Make a copy of the digest with half the input provided. | 
 |   size_t half = strlen(test->input) / 2; | 
 |   ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), test->input, half)); | 
 |   ASSERT_TRUE(EVP_MD_CTX_copy_ex(copy.get(), ctx.get())); | 
 |   ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input + half, | 
 |                                strlen(test->input) - half)); | 
 |   for (size_t i = 1; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input, strlen(test->input))); | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(copy.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Move the digest from the initial state. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   copy = std::move(ctx); | 
 |   for (size_t i = 0; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input, strlen(test->input))); | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(copy.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Move the digest with half the input provided. | 
 |   ASSERT_TRUE(EVP_DigestInit_ex(ctx.get(), test->md.func(), nullptr)); | 
 |   ASSERT_TRUE(EVP_DigestUpdate(ctx.get(), test->input, half)); | 
 |   copy = std::move(ctx); | 
 |   ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input + half, | 
 |                                strlen(test->input) - half)); | 
 |   for (size_t i = 1; i < test->repeat; i++) { | 
 |     ASSERT_TRUE(EVP_DigestUpdate(copy.get(), test->input, strlen(test->input))); | 
 |   } | 
 |   ASSERT_TRUE(EVP_DigestFinal_ex(copy.get(), digest.get(), &digest_len)); | 
 |   CompareDigest(test, digest.get(), digest_len); | 
 |  | 
 |   // Test the one-shot function. | 
 |   if (test->md.one_shot_func && test->repeat == 1) { | 
 |     uint8_t *out = test->md.one_shot_func((const uint8_t *)test->input, | 
 |                                           strlen(test->input), digest.get()); | 
 |     // One-shot functions return their supplied buffers. | 
 |     EXPECT_EQ(digest.get(), out); | 
 |     CompareDigest(test, digest.get(), EVP_MD_size(test->md.func())); | 
 |   } | 
 | } | 
 |  | 
 | TEST(DigestTest, TestVectors) { | 
 |   for (size_t i = 0; i < OPENSSL_ARRAY_SIZE(kTestVectors); i++) { | 
 |     SCOPED_TRACE(i); | 
 |     TestDigest(&kTestVectors[i]); | 
 |   } | 
 | } | 
 |  | 
 | TEST(DigestTest, Getters) { | 
 |   EXPECT_EQ(EVP_sha512(), EVP_get_digestbyname("RSA-SHA512")); | 
 |   EXPECT_EQ(EVP_sha512(), EVP_get_digestbyname("sha512WithRSAEncryption")); | 
 |   EXPECT_EQ(nullptr, EVP_get_digestbyname("nonsense")); | 
 |   EXPECT_EQ(EVP_sha512(), EVP_get_digestbyname("SHA512")); | 
 |   EXPECT_EQ(EVP_sha512(), EVP_get_digestbyname("sha512")); | 
 |  | 
 |   EXPECT_EQ(EVP_sha512(), EVP_get_digestbynid(NID_sha512)); | 
 |   EXPECT_EQ(nullptr, EVP_get_digestbynid(NID_sha512WithRSAEncryption)); | 
 |   EXPECT_EQ(nullptr, EVP_get_digestbynid(NID_undef)); | 
 |  | 
 |   bssl::UniquePtr<ASN1_OBJECT> obj(OBJ_txt2obj("1.3.14.3.2.26", 0)); | 
 |   ASSERT_TRUE(obj); | 
 |   EXPECT_EQ(EVP_sha1(), EVP_get_digestbyobj(obj.get())); | 
 |   EXPECT_EQ(EVP_md5_sha1(), EVP_get_digestbyobj(OBJ_nid2obj(NID_md5_sha1))); | 
 |   EXPECT_EQ(EVP_sha1(), EVP_get_digestbyobj(OBJ_nid2obj(NID_sha1))); | 
 | } | 
 |  | 
 | TEST(DigestTest, ASN1) { | 
 |   bssl::ScopedCBB cbb; | 
 |   ASSERT_TRUE(CBB_init(cbb.get(), 0)); | 
 |   EXPECT_FALSE(EVP_marshal_digest_algorithm(cbb.get(), EVP_md5_sha1())); | 
 |  | 
 |   static const uint8_t kSHA256NullParam[] = {0x30, 0x0d, 0x06, 0x09, 0x60, | 
 |                                              0x86, 0x48, 0x01, 0x65, 0x03, | 
 |                                              0x04, 0x02, 0x01, 0x05, 0x00}; | 
 |   static const uint8_t kSHA256NoParam[] = {0x30, 0x0b, 0x06, 0x09, 0x60, | 
 |                                            0x86, 0x48, 0x01, 0x65, 0x03, | 
 |                                            0x04, 0x02, 0x01}; | 
 |   static const uint8_t kSHA256GarbageParam[] = { | 
 |       0x30, 0x0e, 0x06, 0x09, 0x60, 0x86, 0x48, 0x01, | 
 |       0x65, 0x03, 0x04, 0x02, 0x01, 0x02, 0x01, 0x2a}; | 
 |  | 
 |   // Serialize SHA-256, with and without NULL. | 
 |   cbb.Reset(); | 
 |   ASSERT_TRUE(CBB_init(cbb.get(), 0)); | 
 |   ASSERT_TRUE(EVP_marshal_digest_algorithm(cbb.get(), EVP_sha256())); | 
 |   EXPECT_EQ(Bytes(kSHA256NullParam), | 
 |             Bytes(CBB_data(cbb.get()), CBB_len(cbb.get()))); | 
 |   cbb.Reset(); | 
 |   ASSERT_TRUE(CBB_init(cbb.get(), 0)); | 
 |   ASSERT_TRUE(EVP_marshal_digest_algorithm_no_params(cbb.get(), EVP_sha256())); | 
 |   EXPECT_EQ(Bytes(kSHA256NoParam), | 
 |             Bytes(CBB_data(cbb.get()), CBB_len(cbb.get()))); | 
 |  | 
 |   // Parse SHA-256. Either absent or NULL parameters are tolerated for | 
 |   // compatibility. | 
 |   CBS cbs; | 
 |   CBS_init(&cbs, kSHA256NullParam, sizeof(kSHA256NullParam)); | 
 |   EXPECT_EQ(EVP_sha256(), EVP_parse_digest_algorithm(&cbs)); | 
 |   EXPECT_EQ(0u, CBS_len(&cbs)); | 
 |   CBS_init(&cbs, kSHA256NoParam, sizeof(kSHA256NoParam)); | 
 |   EXPECT_EQ(EVP_sha256(), EVP_parse_digest_algorithm(&cbs)); | 
 |   EXPECT_EQ(0u, CBS_len(&cbs)); | 
 |  | 
 |   // Garbage parameters are not. | 
 |   CBS_init(&cbs, kSHA256GarbageParam, sizeof(kSHA256GarbageParam)); | 
 |   EXPECT_FALSE(EVP_parse_digest_algorithm(&cbs)); | 
 | } | 
 |  | 
 | TEST(DigestTest, TransformBlocks) { | 
 |   uint8_t blocks[SHA256_CBLOCK * 10]; | 
 |   for (size_t i = 0; i < sizeof(blocks); i++) { | 
 |     blocks[i] = i * 3; | 
 |   } | 
 |  | 
 |   SHA256_CTX ctx1; | 
 |   SHA256_Init(&ctx1); | 
 |   SHA256_Update(&ctx1, blocks, sizeof(blocks)); | 
 |  | 
 |   SHA256_CTX ctx2; | 
 |   SHA256_Init(&ctx2); | 
 |   SHA256_TransformBlocks(ctx2.h, blocks, sizeof(blocks) / SHA256_CBLOCK); | 
 |  | 
 |   EXPECT_TRUE(0 == OPENSSL_memcmp(ctx1.h, ctx2.h, sizeof(ctx1.h))); | 
 | } | 
 |  | 
 | }  // namespace |