71 lines
2.6 KiB
C++
71 lines
2.6 KiB
C++
|
#include <google/gtest/gtest.h>
|
||
|
|
#include "../../../../src/config/crypto/outer/OuterEncryptor.h"
|
||
|
|
#include <messmer/cpp-utils/crypto/symmetric/ciphers.h>
|
||
|
|
#include <messmer/cpp-utils/data/DataFixture.h>
|
||
|
|
#include <boost/optional/optional_io.hpp>
|
||
|
|
|
||
|
|
using std::ostream;
|
||
|
|
using boost::none;
|
||
|
|
using cpputils::Data;
|
||
|
|
using cpputils::DataFixture;
|
||
|
|
using cpputils::unique_ref;
|
||
|
|
using cpputils::make_unique_ref;
|
||
|
|
using cpputils::DerivedKeyConfig;
|
||
|
|
using namespace cryfs;
|
||
|
|
|
||
|
|
// This is needed for google test
|
||
|
|
namespace boost {
|
||
|
|
inline ostream &operator<<(ostream &stream, const Data &) {
|
||
|
|
return stream << "cpputils::Data()";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class OuterEncryptorTest : public ::testing::Test {
|
||
|
|
public:
|
||
|
|
unique_ref<OuterEncryptor> makeOuterEncryptor() {
|
||
|
|
auto key = DataFixture::generateFixedSize<OuterEncryptor::Cipher::EncryptionKey::BINARY_LENGTH>();
|
||
|
|
auto salt = DataFixture::generate(128);
|
||
|
|
return make_unique_ref<OuterEncryptor>(key, DerivedKeyConfig(std::move(salt), 1024, 1, 2));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(OuterEncryptorTest, EncryptAndDecrypt) {
|
||
|
|
auto encryptor = makeOuterEncryptor();
|
||
|
|
OuterConfig encrypted = encryptor->encrypt(DataFixture::generate(200));
|
||
|
|
Data decrypted = encryptor->decrypt(encrypted).value();
|
||
|
|
EXPECT_EQ(DataFixture::generate(200), decrypted);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(OuterEncryptorTest, EncryptAndDecrypt_EmptyData) {
|
||
|
|
auto encryptor = makeOuterEncryptor();
|
||
|
|
OuterConfig encrypted = encryptor->encrypt(Data(0));
|
||
|
|
Data decrypted = encryptor->decrypt(encrypted).value();
|
||
|
|
EXPECT_EQ(Data(0), decrypted);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(OuterEncryptorTest, InvalidCiphertext) {
|
||
|
|
auto encryptor = makeOuterEncryptor();
|
||
|
|
OuterConfig encrypted = encryptor->encrypt(DataFixture::generate(200));
|
||
|
|
*(char*)encrypted.encryptedInnerConfig.data() = *(char*)encrypted.encryptedInnerConfig.data()+1; //Modify ciphertext
|
||
|
|
auto decrypted = encryptor->decrypt(encrypted);
|
||
|
|
EXPECT_EQ(none, decrypted);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(OuterEncryptorTest, DoesntEncryptWhenTooLarge) {
|
||
|
|
auto encryptor = makeOuterEncryptor();
|
||
|
|
EXPECT_THROW(
|
||
|
|
encryptor->encrypt(DataFixture::generate(2000)),
|
||
|
|
std::runtime_error
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(OuterEncryptorTest, EncryptionIsFixedSize) {
|
||
|
|
auto encryptor = makeOuterEncryptor();
|
||
|
|
OuterConfig encrypted1 = encryptor->encrypt(DataFixture::generate(200));
|
||
|
|
OuterConfig encrypted2 = encryptor->encrypt(DataFixture::generate(700));
|
||
|
|
OuterConfig encrypted3 = encryptor->encrypt(Data(0));
|
||
|
|
|
||
|
|
EXPECT_EQ(encrypted1.encryptedInnerConfig.size(), encrypted2.encryptedInnerConfig.size());
|
||
|
|
EXPECT_EQ(encrypted1.encryptedInnerConfig.size(), encrypted3.encryptedInnerConfig.size());
|
||
|
|
}
|