417a701636
- Fixed numBlocks() in OnDiskBlockStore, FakeBlockStore, CachingBlockStore, ... - CachingBlockStore caches created blocks and doesn't directly create them in the underlying blockstore
47 lines
1.2 KiB
C++
47 lines
1.2 KiB
C++
#include "EncryptedBlockStore.h"
|
|
#include "EncryptedBlock.h"
|
|
#include <messmer/cpp-utils/pointer.h>
|
|
#include "../../utils/BlockStoreUtils.h"
|
|
|
|
using std::unique_ptr;
|
|
using std::make_unique;
|
|
|
|
namespace blockstore {
|
|
namespace encrypted {
|
|
|
|
EncryptedBlockStore::EncryptedBlockStore(unique_ptr<BlockStore> baseBlockStore, const EncryptionKey &encKey)
|
|
: _baseBlockStore(std::move(baseBlockStore)), _encKey(encKey) {
|
|
}
|
|
|
|
Key EncryptedBlockStore::createKey() {
|
|
return _baseBlockStore->createKey();
|
|
}
|
|
|
|
unique_ptr<Block> EncryptedBlockStore::tryCreate(const Key &key, Data data) {
|
|
return EncryptedBlock::TryCreateNew(_baseBlockStore.get(), key, std::move(data), _encKey);
|
|
}
|
|
|
|
unique_ptr<Block> EncryptedBlockStore::load(const Key &key) {
|
|
auto block = _baseBlockStore->load(key);
|
|
if (block.get() == nullptr) {
|
|
return nullptr;
|
|
}
|
|
return make_unique<EncryptedBlock>(std::move(block), _encKey);
|
|
}
|
|
|
|
void EncryptedBlockStore::remove(unique_ptr<Block> block) {
|
|
return _baseBlockStore->remove(std::move(block));
|
|
}
|
|
|
|
uint64_t EncryptedBlockStore::numBlocks() const {
|
|
return _baseBlockStore->numBlocks();
|
|
}
|
|
|
|
unique_ptr<Block> EncryptedBlockStore::tryCreateInBaseStore(const Key &key, Data encryptedData) {
|
|
return _baseBlockStore->tryCreate(key, std::move(encryptedData));
|
|
}
|
|
|
|
}
|
|
}
|
|
|