mirror of https://github.com/facebook/rocksdb.git
131 lines
3.6 KiB
C++
131 lines
3.6 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
//
|
|
#include "db/db_test_util.h"
|
|
#include "port/stack_trace.h"
|
|
#include "rocksdb/perf_context.h"
|
|
#if !defined(ROCKSDB_LITE)
|
|
#include "test_util/sync_point.h"
|
|
#endif
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
class DBEncryptionTest : public DBTestBase {
|
|
public:
|
|
DBEncryptionTest()
|
|
: DBTestBase("db_encryption_test", /*env_do_fsync=*/true) {}
|
|
Env* GetTargetEnv() {
|
|
if (encrypted_env_ != nullptr) {
|
|
return (static_cast<EnvWrapper*>(encrypted_env_))->target();
|
|
} else {
|
|
return env_;
|
|
}
|
|
}
|
|
};
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
|
|
TEST_F(DBEncryptionTest, CheckEncrypted) {
|
|
ASSERT_OK(Put("foo567", "v1.fetdq"));
|
|
ASSERT_OK(Put("bar123", "v2.dfgkjdfghsd"));
|
|
Close();
|
|
|
|
// Open all files and look for the values we've put in there.
|
|
// They should not be found if encrypted, otherwise
|
|
// they should be found.
|
|
std::vector<std::string> fileNames;
|
|
auto status = env_->GetChildren(dbname_, &fileNames);
|
|
ASSERT_OK(status);
|
|
|
|
Env* target = GetTargetEnv();
|
|
int hits = 0;
|
|
for (auto it = fileNames.begin() ; it != fileNames.end(); ++it) {
|
|
if (*it == "LOCK") {
|
|
continue;
|
|
}
|
|
auto filePath = dbname_ + "/" + *it;
|
|
std::unique_ptr<SequentialFile> seqFile;
|
|
auto envOptions = EnvOptions(CurrentOptions());
|
|
status = target->NewSequentialFile(filePath, &seqFile, envOptions);
|
|
ASSERT_OK(status);
|
|
|
|
uint64_t fileSize;
|
|
status = target->GetFileSize(filePath, &fileSize);
|
|
ASSERT_OK(status);
|
|
|
|
std::string scratch;
|
|
scratch.reserve(fileSize);
|
|
Slice data;
|
|
status = seqFile->Read(fileSize, &data, (char*)scratch.data());
|
|
ASSERT_OK(status);
|
|
|
|
if (data.ToString().find("foo567") != std::string::npos) {
|
|
hits++;
|
|
//std::cout << "Hit in " << filePath << "\n";
|
|
}
|
|
if (data.ToString().find("v1.fetdq") != std::string::npos) {
|
|
hits++;
|
|
//std::cout << "Hit in " << filePath << "\n";
|
|
}
|
|
if (data.ToString().find("bar123") != std::string::npos) {
|
|
hits++;
|
|
//std::cout << "Hit in " << filePath << "\n";
|
|
}
|
|
if (data.ToString().find("v2.dfgkjdfghsd") != std::string::npos) {
|
|
hits++;
|
|
//std::cout << "Hit in " << filePath << "\n";
|
|
}
|
|
if (data.ToString().find("dfgk") != std::string::npos) {
|
|
hits++;
|
|
//std::cout << "Hit in " << filePath << "\n";
|
|
}
|
|
}
|
|
if (encrypted_env_) {
|
|
ASSERT_EQ(hits, 0);
|
|
} else {
|
|
ASSERT_GE(hits, 4);
|
|
}
|
|
}
|
|
|
|
TEST_F(DBEncryptionTest, ReadEmptyFile) {
|
|
auto defaultEnv = GetTargetEnv();
|
|
|
|
// create empty file for reading it back in later
|
|
auto envOptions = EnvOptions(CurrentOptions());
|
|
auto filePath = dbname_ + "/empty.empty";
|
|
|
|
Status status;
|
|
{
|
|
std::unique_ptr<WritableFile> writableFile;
|
|
status = defaultEnv->NewWritableFile(filePath, &writableFile, envOptions);
|
|
ASSERT_OK(status);
|
|
}
|
|
|
|
std::unique_ptr<SequentialFile> seqFile;
|
|
status = defaultEnv->NewSequentialFile(filePath, &seqFile, envOptions);
|
|
ASSERT_OK(status);
|
|
|
|
std::string scratch;
|
|
Slice data;
|
|
// reading back 16 bytes from the empty file shouldn't trigger an assertion.
|
|
// it should just work and return an empty string
|
|
status = seqFile->Read(16, &data, (char*)scratch.data());
|
|
ASSERT_OK(status);
|
|
|
|
ASSERT_TRUE(data.empty());
|
|
}
|
|
|
|
#endif // ROCKSDB_LITE
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|
|
|
|
int main(int argc, char** argv) {
|
|
ROCKSDB_NAMESPACE::port::InstallStackTraceHandler();
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|