129 lines
3.9 KiB
C++
129 lines
3.9 KiB
C++
#include <google/gtest/gtest.h>
|
|
#include "../../../implementations/encrypted/ciphers/AES256_CFB.h"
|
|
|
|
#include "../../testutils/DataBlockFixture.h"
|
|
#include "../../../utils/Data.h"
|
|
|
|
using namespace blockstore::encrypted;
|
|
using blockstore::Data;
|
|
|
|
template<class Cipher>
|
|
class CipherTest: public ::testing::Test {
|
|
public:
|
|
typename Cipher::EncryptionKey encKey = createRandomKey();
|
|
|
|
static typename Cipher::EncryptionKey createRandomKey(int seed = 0) {
|
|
DataBlockFixture data(Cipher::EncryptionKey::BINARY_LENGTH, seed);
|
|
return Cipher::EncryptionKey::FromBinary(data.data());
|
|
}
|
|
|
|
void CheckEncryptThenDecryptIsIdentity(const Data &plaintext) {
|
|
Data ciphertext = Encrypt(plaintext);
|
|
Data decrypted = Decrypt(ciphertext);
|
|
EXPECT_EQ(plaintext.size(), decrypted.size());
|
|
EXPECT_EQ(0, std::memcmp(plaintext.data(), decrypted.data(), plaintext.size()));
|
|
}
|
|
|
|
void CheckEncryptIsIndeterministic(const Data &plaintext) {
|
|
Data ciphertext = Encrypt(plaintext);
|
|
Data ciphertext2 = Encrypt(plaintext);
|
|
EXPECT_NE(0, std::memcmp(ciphertext.data(), ciphertext2.data(), ciphertext.size()));
|
|
}
|
|
|
|
Data Encrypt(const Data &plaintext) {
|
|
Data ciphertext(Cipher::ciphertextSize(plaintext.size()));
|
|
Cipher::encrypt((byte*)plaintext.data(), plaintext.size(), (byte*)ciphertext.data(), this->encKey);
|
|
return ciphertext;
|
|
}
|
|
|
|
Data Decrypt(const Data &ciphertext) {
|
|
Data decrypted(Cipher::plaintextSize(ciphertext.size()));
|
|
Cipher::decrypt((byte*)ciphertext.data(), (byte*) decrypted.data(), decrypted.size(), this->encKey);
|
|
return decrypted;
|
|
}
|
|
|
|
Data CreateZeroes(unsigned int size) {
|
|
Data zeroes(size);
|
|
zeroes.FillWithZeroes();
|
|
return zeroes;
|
|
}
|
|
|
|
Data CreateData(unsigned int size, unsigned int seed = 0) {
|
|
DataBlockFixture data(size, seed);
|
|
Data result(size);
|
|
std::memcpy(result.data(), data.data(), size);
|
|
return result;
|
|
}
|
|
};
|
|
|
|
TYPED_TEST_CASE_P(CipherTest);
|
|
|
|
TYPED_TEST_P(CipherTest, Size_0) {
|
|
EXPECT_EQ(0, TypeParam::ciphertextSize(TypeParam::plaintextSize(0)));
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, Size_1) {
|
|
EXPECT_EQ(1, TypeParam::ciphertextSize(TypeParam::plaintextSize(1)));
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, Size_1024) {
|
|
EXPECT_EQ(1024, TypeParam::ciphertextSize(TypeParam::plaintextSize(1024)));
|
|
EXPECT_EQ(1024, TypeParam::plaintextSize(TypeParam::ciphertextSize(1024)));
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, Size_4096) {
|
|
EXPECT_EQ(4096, TypeParam::ciphertextSize(TypeParam::plaintextSize(4096)));
|
|
EXPECT_EQ(4096, TypeParam::plaintextSize(TypeParam::ciphertextSize(4096)));
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, Size_1048576) {
|
|
EXPECT_EQ(1048576, TypeParam::ciphertextSize(TypeParam::plaintextSize(1048576)));
|
|
EXPECT_EQ(1048576, TypeParam::plaintextSize(TypeParam::ciphertextSize(1048576)));
|
|
}
|
|
|
|
constexpr std::initializer_list<unsigned int> SIZES = {0, 1, 100, 1024, 5000, 1048576, 52428800};
|
|
|
|
TYPED_TEST_P(CipherTest, EncryptThenDecrypt_Zeroes) {
|
|
for (auto size: SIZES) {
|
|
Data plaintext = this->CreateZeroes(size);
|
|
this->CheckEncryptThenDecryptIsIdentity(plaintext);
|
|
}
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, EncryptThenDecrypt_Data) {
|
|
for (auto size: SIZES) {
|
|
Data plaintext = this->CreateData(size);
|
|
this->CheckEncryptThenDecryptIsIdentity(plaintext);
|
|
}
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, EncryptIsIndeterministic_Zeroes) {
|
|
for (auto size: SIZES) {
|
|
Data plaintext = this->CreateZeroes(size);
|
|
this->CheckEncryptIsIndeterministic(plaintext);
|
|
}
|
|
}
|
|
|
|
TYPED_TEST_P(CipherTest, EncryptIsIndeterministic_Data) {
|
|
for (auto size: SIZES) {
|
|
Data plaintext = this->CreateData(size);
|
|
this->CheckEncryptIsIndeterministic(plaintext);
|
|
}
|
|
}
|
|
|
|
REGISTER_TYPED_TEST_CASE_P(CipherTest,
|
|
Size_0,
|
|
Size_1,
|
|
Size_1024,
|
|
Size_4096,
|
|
Size_1048576,
|
|
EncryptThenDecrypt_Zeroes,
|
|
EncryptThenDecrypt_Data,
|
|
EncryptIsIndeterministic_Zeroes,
|
|
EncryptIsIndeterministic_Data
|
|
);
|
|
|
|
//TODO For authenticated ciphers, we need test cases checking that authentication fails on manipulations
|
|
|
|
INSTANTIATE_TYPED_TEST_CASE_P(AES256_CFB, CipherTest, AES256_CFB);
|