summaryrefslogtreecommitdiffstats
path: root/src/rocksdb/db/db_encryption_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/rocksdb/db/db_encryption_test.cc')
-rw-r--r--src/rocksdb/db/db_encryption_test.cc130
1 files changed, 130 insertions, 0 deletions
diff --git a/src/rocksdb/db/db_encryption_test.cc b/src/rocksdb/db/db_encryption_test.cc
new file mode 100644
index 000000000..73e89d158
--- /dev/null
+++ b/src/rocksdb/db/db_encryption_test.cc
@@ -0,0 +1,130 @@
+// 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();
+}