12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- // Copyright 2015 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.
- #include "crypto/aead.h"
- #include <string>
- #include "testing/gtest/include/gtest/gtest.h"
- namespace {
- const crypto::Aead::AeadAlgorithm kAllAlgorithms[]{
- crypto::Aead::AES_128_CTR_HMAC_SHA256,
- crypto::Aead::AES_256_GCM,
- crypto::Aead::AES_256_GCM_SIV,
- crypto::Aead::CHACHA20_POLY1305,
- };
- class AeadTest : public testing::TestWithParam<crypto::Aead::AeadAlgorithm> {};
- INSTANTIATE_TEST_SUITE_P(All, AeadTest, testing::ValuesIn(kAllAlgorithms));
- TEST_P(AeadTest, SealOpen) {
- crypto::Aead::AeadAlgorithm alg = GetParam();
- crypto::Aead aead(alg);
- std::string key(aead.KeyLength(), 0);
- aead.Init(&key);
- std::string nonce(aead.NonceLength(), 0);
- std::string plaintext("this is the plaintext");
- std::string ad("this is the additional data");
- std::string ciphertext;
- EXPECT_TRUE(aead.Seal(plaintext, nonce, ad, &ciphertext));
- EXPECT_LT(0U, ciphertext.size());
- std::string decrypted;
- EXPECT_TRUE(aead.Open(ciphertext, nonce, ad, &decrypted));
- EXPECT_EQ(plaintext, decrypted);
- }
- TEST_P(AeadTest, SealOpenSpan) {
- crypto::Aead::AeadAlgorithm alg = GetParam();
- crypto::Aead aead(alg);
- std::vector<uint8_t> key(aead.KeyLength(), 0u);
- aead.Init(key);
- std::vector<uint8_t> nonce(aead.NonceLength(), 0u);
- static constexpr uint8_t kPlaintext[] = "plaintext";
- static constexpr uint8_t kAdditionalData[] = "additional data input";
- std::vector<uint8_t> ciphertext =
- aead.Seal(kPlaintext, nonce, kAdditionalData);
- EXPECT_LT(sizeof(kPlaintext), ciphertext.size());
- absl::optional<std::vector<uint8_t>> decrypted =
- aead.Open(ciphertext, nonce, kAdditionalData);
- ASSERT_TRUE(decrypted);
- ASSERT_EQ(decrypted->size(), sizeof(kPlaintext));
- ASSERT_EQ(0, memcmp(decrypted->data(), kPlaintext, sizeof(kPlaintext)));
- std::vector<uint8_t> wrong_key(aead.KeyLength(), 1u);
- crypto::Aead aead_wrong_key(alg);
- aead_wrong_key.Init(wrong_key);
- decrypted = aead_wrong_key.Open(ciphertext, nonce, kAdditionalData);
- EXPECT_FALSE(decrypted);
- }
- TEST_P(AeadTest, SealOpenWrongKey) {
- crypto::Aead::AeadAlgorithm alg = GetParam();
- crypto::Aead aead(alg);
- std::string key(aead.KeyLength(), 0);
- std::string wrong_key(aead.KeyLength(), 1);
- aead.Init(&key);
- crypto::Aead aead_wrong_key(alg);
- aead_wrong_key.Init(&wrong_key);
- std::string nonce(aead.NonceLength(), 0);
- std::string plaintext("this is the plaintext");
- std::string ad("this is the additional data");
- std::string ciphertext;
- EXPECT_TRUE(aead.Seal(plaintext, nonce, ad, &ciphertext));
- EXPECT_LT(0U, ciphertext.size());
- std::string decrypted;
- EXPECT_FALSE(aead_wrong_key.Open(ciphertext, nonce, ad, &decrypted));
- EXPECT_EQ(0U, decrypted.size());
- }
- } // namespace
|