115 lines
4.0 KiB
C++
115 lines
4.0 KiB
C++
#include <gtest/gtest.h>
|
|
#include "../../../cpp-utils/crypto/symmetric/testutils/FakeAuthenticatedCipher.h"
|
|
#include "blockstore/implementations/encrypted/EncryptedBlockStore.h"
|
|
#include "blockstore/implementations/testfake/FakeBlockStore.h"
|
|
#include "blockstore/utils/BlockStoreUtils.h"
|
|
#include <cpp-utils/data/DataFixture.h>
|
|
|
|
using ::testing::Test;
|
|
|
|
using cpputils::DataFixture;
|
|
using cpputils::Data;
|
|
using cpputils::unique_ref;
|
|
using cpputils::make_unique_ref;
|
|
using cpputils::FakeAuthenticatedCipher;
|
|
|
|
using blockstore::testfake::FakeBlockStore;
|
|
|
|
using namespace blockstore::encrypted;
|
|
|
|
class EncryptedBlockStoreTest: public Test {
|
|
public:
|
|
static constexpr unsigned int BLOCKSIZE = 1024;
|
|
EncryptedBlockStoreTest():
|
|
baseBlockStore(new FakeBlockStore),
|
|
blockStore(make_unique_ref<EncryptedBlockStore<FakeAuthenticatedCipher>>(std::move(cpputils::nullcheck(std::unique_ptr<FakeBlockStore>(baseBlockStore)).value()), FakeAuthenticatedCipher::Key1())),
|
|
data(DataFixture::generate(BLOCKSIZE)) {
|
|
}
|
|
FakeBlockStore *baseBlockStore;
|
|
unique_ref<EncryptedBlockStore<FakeAuthenticatedCipher>> blockStore;
|
|
Data data;
|
|
|
|
blockstore::Key CreateBlockDirectlyWithFixtureAndReturnKey() {
|
|
return blockStore->create(data)->key();
|
|
}
|
|
|
|
blockstore::Key CreateBlockWriteFixtureToItAndReturnKey() {
|
|
auto block = blockStore->create(Data(data.size()));
|
|
block->write(data.data(), 0, data.size());
|
|
return block->key();
|
|
}
|
|
|
|
void ModifyBaseBlock(const blockstore::Key &key) {
|
|
auto block = baseBlockStore->load(key).value();
|
|
uint8_t middle_byte = ((byte*)block->data())[10];
|
|
uint8_t new_middle_byte = middle_byte + 1;
|
|
block->write(&new_middle_byte, 10, 1);
|
|
}
|
|
|
|
blockstore::Key CopyBaseBlock(const blockstore::Key &key) {
|
|
auto source = baseBlockStore->load(key).value();
|
|
return blockstore::utils::copyToNewBlock(baseBlockStore, *source)->key();
|
|
}
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(EncryptedBlockStoreTest);
|
|
};
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithSameKeyWorks_WriteOnCreate) {
|
|
auto key = CreateBlockDirectlyWithFixtureAndReturnKey();
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_NE(boost::none, loaded);
|
|
EXPECT_EQ(data.size(), (*loaded)->size());
|
|
EXPECT_EQ(0, std::memcmp(data.data(), (*loaded)->data(), data.size()));
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithSameKeyWorks_WriteSeparately) {
|
|
auto key = CreateBlockWriteFixtureToItAndReturnKey();
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_NE(boost::none, loaded);
|
|
EXPECT_EQ(data.size(), (*loaded)->size());
|
|
EXPECT_EQ(0, std::memcmp(data.data(), (*loaded)->data(), data.size()));
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithDifferentKeyDoesntWork_WriteOnCreate) {
|
|
auto key = CreateBlockDirectlyWithFixtureAndReturnKey();
|
|
blockStore->__setKey(FakeAuthenticatedCipher::Key2());
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithDifferentKeyDoesntWork_WriteSeparately) {
|
|
auto key = CreateBlockWriteFixtureToItAndReturnKey();
|
|
blockStore->__setKey(FakeAuthenticatedCipher::Key2());
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingModifiedBlockFails_WriteOnCreate) {
|
|
auto key = CreateBlockDirectlyWithFixtureAndReturnKey();
|
|
ModifyBaseBlock(key);
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingModifiedBlockFails_WriteSeparately) {
|
|
auto key = CreateBlockWriteFixtureToItAndReturnKey();
|
|
ModifyBaseBlock(key);
|
|
auto loaded = blockStore->load(key);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithDifferentBlockIdFails_WriteOnCreate) {
|
|
auto key = CreateBlockDirectlyWithFixtureAndReturnKey();
|
|
auto key2 = CopyBaseBlock(key);
|
|
auto loaded = blockStore->load(key2);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|
|
|
|
TEST_F(EncryptedBlockStoreTest, LoadingWithDifferentBlockIdFails_WriteSeparately) {
|
|
auto key = CreateBlockWriteFixtureToItAndReturnKey();
|
|
auto key2 = CopyBaseBlock(key);
|
|
auto loaded = blockStore->load(key2);
|
|
EXPECT_EQ(boost::none, loaded);
|
|
}
|