summaryrefslogtreecommitdiffstats
path: root/src/rocksdb/db/kv_checksum.h
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:54:28 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-21 11:54:28 +0000
commite6918187568dbd01842d8d1d2c808ce16a894239 (patch)
tree64f88b554b444a49f656b6c656111a145cbbaa28 /src/rocksdb/db/kv_checksum.h
parentInitial commit. (diff)
downloadceph-e6918187568dbd01842d8d1d2c808ce16a894239.tar.xz
ceph-e6918187568dbd01842d8d1d2c808ce16a894239.zip
Adding upstream version 18.2.2.upstream/18.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/rocksdb/db/kv_checksum.h')
-rw-r--r--src/rocksdb/db/kv_checksum.h398
1 files changed, 398 insertions, 0 deletions
diff --git a/src/rocksdb/db/kv_checksum.h b/src/rocksdb/db/kv_checksum.h
new file mode 100644
index 000000000..bce507fcf
--- /dev/null
+++ b/src/rocksdb/db/kv_checksum.h
@@ -0,0 +1,398 @@
+// Copyright (c) 2020-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).
+//
+// This file contains classes containing fields to protect individual entries.
+// The classes are named "ProtectionInfo<suffix>", where <suffix> indicates the
+// combination of fields that are covered. Each field has a single letter
+// abbreviation as follows.
+//
+// K = key
+// V = value
+// O = optype aka value type
+// S = seqno
+// C = CF ID
+//
+// Then, for example, a class that protects an entry consisting of key, value,
+// optype, and CF ID (i.e., a `WriteBatch` entry) would be named
+// `ProtectionInfoKVOC`.
+//
+// The `ProtectionInfo.*` classes are templated on the integer type used to hold
+// the XOR of hashes for each field. Only unsigned integer types are supported,
+// and the maximum supported integer width is 64 bits. When the integer type is
+// narrower than the hash values, we lop off the most significant bits to make
+// them fit.
+//
+// The `ProtectionInfo.*` classes are all intended to be non-persistent. We do
+// not currently make the byte order consistent for integer fields before
+// hashing them, so the resulting values are endianness-dependent.
+
+#pragma once
+
+#include <type_traits>
+
+#include "db/dbformat.h"
+#include "rocksdb/types.h"
+#include "util/hash.h"
+
+namespace ROCKSDB_NAMESPACE {
+
+template <typename T>
+class ProtectionInfo;
+template <typename T>
+class ProtectionInfoKVO;
+template <typename T>
+class ProtectionInfoKVOC;
+template <typename T>
+class ProtectionInfoKVOS;
+
+// Aliases for 64-bit protection infos.
+using ProtectionInfo64 = ProtectionInfo<uint64_t>;
+using ProtectionInfoKVO64 = ProtectionInfoKVO<uint64_t>;
+using ProtectionInfoKVOC64 = ProtectionInfoKVOC<uint64_t>;
+using ProtectionInfoKVOS64 = ProtectionInfoKVOS<uint64_t>;
+
+template <typename T>
+class ProtectionInfo {
+ public:
+ ProtectionInfo() = default;
+
+ Status GetStatus() const;
+ ProtectionInfoKVO<T> ProtectKVO(const Slice& key, const Slice& value,
+ ValueType op_type) const;
+ ProtectionInfoKVO<T> ProtectKVO(const SliceParts& key,
+ const SliceParts& value,
+ ValueType op_type) const;
+
+ T GetVal() const { return val_; }
+
+ private:
+ friend class ProtectionInfoKVO<T>;
+ friend class ProtectionInfoKVOS<T>;
+ friend class ProtectionInfoKVOC<T>;
+
+ // Each field is hashed with an independent value so we can catch fields being
+ // swapped. Per the `NPHash64()` docs, using consecutive seeds is a pitfall,
+ // and we should instead vary our seeds by a large odd number. This value by
+ // which we increment (0xD28AAD72F49BD50B) was taken from
+ // `head -c8 /dev/urandom | hexdump`, run repeatedly until it yielded an odd
+ // number. The values are computed manually since the Windows C++ compiler
+ // complains about the overflow when adding constants.
+ static const uint64_t kSeedK = 0;
+ static const uint64_t kSeedV = 0xD28AAD72F49BD50B;
+ static const uint64_t kSeedO = 0xA5155AE5E937AA16;
+ static const uint64_t kSeedS = 0x77A00858DDD37F21;
+ static const uint64_t kSeedC = 0x4A2AB5CBD26F542C;
+
+ ProtectionInfo(T val) : val_(val) {
+ static_assert(sizeof(ProtectionInfo<T>) == sizeof(T), "");
+ }
+
+ void SetVal(T val) { val_ = val; }
+
+ T val_ = 0;
+};
+
+template <typename T>
+class ProtectionInfoKVO {
+ public:
+ ProtectionInfoKVO() = default;
+
+ ProtectionInfo<T> StripKVO(const Slice& key, const Slice& value,
+ ValueType op_type) const;
+ ProtectionInfo<T> StripKVO(const SliceParts& key, const SliceParts& value,
+ ValueType op_type) const;
+
+ ProtectionInfoKVOC<T> ProtectC(ColumnFamilyId column_family_id) const;
+ ProtectionInfoKVOS<T> ProtectS(SequenceNumber sequence_number) const;
+
+ void UpdateK(const Slice& old_key, const Slice& new_key);
+ void UpdateK(const SliceParts& old_key, const SliceParts& new_key);
+ void UpdateV(const Slice& old_value, const Slice& new_value);
+ void UpdateV(const SliceParts& old_value, const SliceParts& new_value);
+ void UpdateO(ValueType old_op_type, ValueType new_op_type);
+
+ T GetVal() const { return info_.GetVal(); }
+
+ private:
+ friend class ProtectionInfo<T>;
+ friend class ProtectionInfoKVOS<T>;
+ friend class ProtectionInfoKVOC<T>;
+
+ explicit ProtectionInfoKVO(T val) : info_(val) {
+ static_assert(sizeof(ProtectionInfoKVO<T>) == sizeof(T), "");
+ }
+
+ void SetVal(T val) { info_.SetVal(val); }
+
+ ProtectionInfo<T> info_;
+};
+
+template <typename T>
+class ProtectionInfoKVOC {
+ public:
+ ProtectionInfoKVOC() = default;
+
+ ProtectionInfoKVO<T> StripC(ColumnFamilyId column_family_id) const;
+
+ void UpdateK(const Slice& old_key, const Slice& new_key) {
+ kvo_.UpdateK(old_key, new_key);
+ }
+ void UpdateK(const SliceParts& old_key, const SliceParts& new_key) {
+ kvo_.UpdateK(old_key, new_key);
+ }
+ void UpdateV(const Slice& old_value, const Slice& new_value) {
+ kvo_.UpdateV(old_value, new_value);
+ }
+ void UpdateV(const SliceParts& old_value, const SliceParts& new_value) {
+ kvo_.UpdateV(old_value, new_value);
+ }
+ void UpdateO(ValueType old_op_type, ValueType new_op_type) {
+ kvo_.UpdateO(old_op_type, new_op_type);
+ }
+ void UpdateC(ColumnFamilyId old_column_family_id,
+ ColumnFamilyId new_column_family_id);
+
+ T GetVal() const { return kvo_.GetVal(); }
+
+ private:
+ friend class ProtectionInfoKVO<T>;
+
+ explicit ProtectionInfoKVOC(T val) : kvo_(val) {
+ static_assert(sizeof(ProtectionInfoKVOC<T>) == sizeof(T), "");
+ }
+
+ void SetVal(T val) { kvo_.SetVal(val); }
+
+ ProtectionInfoKVO<T> kvo_;
+};
+
+template <typename T>
+class ProtectionInfoKVOS {
+ public:
+ ProtectionInfoKVOS() = default;
+
+ ProtectionInfoKVO<T> StripS(SequenceNumber sequence_number) const;
+
+ void UpdateK(const Slice& old_key, const Slice& new_key) {
+ kvo_.UpdateK(old_key, new_key);
+ }
+ void UpdateK(const SliceParts& old_key, const SliceParts& new_key) {
+ kvo_.UpdateK(old_key, new_key);
+ }
+ void UpdateV(const Slice& old_value, const Slice& new_value) {
+ kvo_.UpdateV(old_value, new_value);
+ }
+ void UpdateV(const SliceParts& old_value, const SliceParts& new_value) {
+ kvo_.UpdateV(old_value, new_value);
+ }
+ void UpdateO(ValueType old_op_type, ValueType new_op_type) {
+ kvo_.UpdateO(old_op_type, new_op_type);
+ }
+ void UpdateS(SequenceNumber old_sequence_number,
+ SequenceNumber new_sequence_number);
+
+ T GetVal() const { return kvo_.GetVal(); }
+
+ private:
+ friend class ProtectionInfoKVO<T>;
+
+ explicit ProtectionInfoKVOS(T val) : kvo_(val) {
+ static_assert(sizeof(ProtectionInfoKVOS<T>) == sizeof(T), "");
+ }
+
+ void SetVal(T val) { kvo_.SetVal(val); }
+
+ ProtectionInfoKVO<T> kvo_;
+};
+
+template <typename T>
+Status ProtectionInfo<T>::GetStatus() const {
+ if (val_ != 0) {
+ return Status::Corruption("ProtectionInfo mismatch");
+ }
+ return Status::OK();
+}
+
+template <typename T>
+ProtectionInfoKVO<T> ProtectionInfo<T>::ProtectKVO(const Slice& key,
+ const Slice& value,
+ ValueType op_type) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(GetSliceNPHash64(key, ProtectionInfo<T>::kSeedK));
+ val =
+ val ^ static_cast<T>(GetSliceNPHash64(value, ProtectionInfo<T>::kSeedV));
+ val = val ^
+ static_cast<T>(NPHash64(reinterpret_cast<char*>(&op_type),
+ sizeof(op_type), ProtectionInfo<T>::kSeedO));
+ return ProtectionInfoKVO<T>(val);
+}
+
+template <typename T>
+ProtectionInfoKVO<T> ProtectionInfo<T>::ProtectKVO(const SliceParts& key,
+ const SliceParts& value,
+ ValueType op_type) const {
+ T val = GetVal();
+ val = val ^
+ static_cast<T>(GetSlicePartsNPHash64(key, ProtectionInfo<T>::kSeedK));
+ val = val ^
+ static_cast<T>(GetSlicePartsNPHash64(value, ProtectionInfo<T>::kSeedV));
+ val = val ^
+ static_cast<T>(NPHash64(reinterpret_cast<char*>(&op_type),
+ sizeof(op_type), ProtectionInfo<T>::kSeedO));
+ return ProtectionInfoKVO<T>(val);
+}
+
+template <typename T>
+void ProtectionInfoKVO<T>::UpdateK(const Slice& old_key, const Slice& new_key) {
+ T val = GetVal();
+ val = val ^
+ static_cast<T>(GetSliceNPHash64(old_key, ProtectionInfo<T>::kSeedK));
+ val = val ^
+ static_cast<T>(GetSliceNPHash64(new_key, ProtectionInfo<T>::kSeedK));
+ SetVal(val);
+}
+
+template <typename T>
+void ProtectionInfoKVO<T>::UpdateK(const SliceParts& old_key,
+ const SliceParts& new_key) {
+ T val = GetVal();
+ val = val ^ static_cast<T>(
+ GetSlicePartsNPHash64(old_key, ProtectionInfo<T>::kSeedK));
+ val = val ^ static_cast<T>(
+ GetSlicePartsNPHash64(new_key, ProtectionInfo<T>::kSeedK));
+ SetVal(val);
+}
+
+template <typename T>
+void ProtectionInfoKVO<T>::UpdateV(const Slice& old_value,
+ const Slice& new_value) {
+ T val = GetVal();
+ val = val ^
+ static_cast<T>(GetSliceNPHash64(old_value, ProtectionInfo<T>::kSeedV));
+ val = val ^
+ static_cast<T>(GetSliceNPHash64(new_value, ProtectionInfo<T>::kSeedV));
+ SetVal(val);
+}
+
+template <typename T>
+void ProtectionInfoKVO<T>::UpdateV(const SliceParts& old_value,
+ const SliceParts& new_value) {
+ T val = GetVal();
+ val = val ^ static_cast<T>(
+ GetSlicePartsNPHash64(old_value, ProtectionInfo<T>::kSeedV));
+ val = val ^ static_cast<T>(
+ GetSlicePartsNPHash64(new_value, ProtectionInfo<T>::kSeedV));
+ SetVal(val);
+}
+
+template <typename T>
+void ProtectionInfoKVO<T>::UpdateO(ValueType old_op_type,
+ ValueType new_op_type) {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(reinterpret_cast<char*>(&old_op_type),
+ sizeof(old_op_type),
+ ProtectionInfo<T>::kSeedO));
+ val = val ^ static_cast<T>(NPHash64(reinterpret_cast<char*>(&new_op_type),
+ sizeof(new_op_type),
+ ProtectionInfo<T>::kSeedO));
+ SetVal(val);
+}
+
+template <typename T>
+ProtectionInfo<T> ProtectionInfoKVO<T>::StripKVO(const Slice& key,
+ const Slice& value,
+ ValueType op_type) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(GetSliceNPHash64(key, ProtectionInfo<T>::kSeedK));
+ val =
+ val ^ static_cast<T>(GetSliceNPHash64(value, ProtectionInfo<T>::kSeedV));
+ val = val ^
+ static_cast<T>(NPHash64(reinterpret_cast<char*>(&op_type),
+ sizeof(op_type), ProtectionInfo<T>::kSeedO));
+ return ProtectionInfo<T>(val);
+}
+
+template <typename T>
+ProtectionInfo<T> ProtectionInfoKVO<T>::StripKVO(const SliceParts& key,
+ const SliceParts& value,
+ ValueType op_type) const {
+ T val = GetVal();
+ val = val ^
+ static_cast<T>(GetSlicePartsNPHash64(key, ProtectionInfo<T>::kSeedK));
+ val = val ^
+ static_cast<T>(GetSlicePartsNPHash64(value, ProtectionInfo<T>::kSeedV));
+ val = val ^
+ static_cast<T>(NPHash64(reinterpret_cast<char*>(&op_type),
+ sizeof(op_type), ProtectionInfo<T>::kSeedO));
+ return ProtectionInfo<T>(val);
+}
+
+template <typename T>
+ProtectionInfoKVOC<T> ProtectionInfoKVO<T>::ProtectC(
+ ColumnFamilyId column_family_id) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&column_family_id),
+ sizeof(column_family_id), ProtectionInfo<T>::kSeedC));
+ return ProtectionInfoKVOC<T>(val);
+}
+
+template <typename T>
+ProtectionInfoKVO<T> ProtectionInfoKVOC<T>::StripC(
+ ColumnFamilyId column_family_id) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&column_family_id),
+ sizeof(column_family_id), ProtectionInfo<T>::kSeedC));
+ return ProtectionInfoKVO<T>(val);
+}
+
+template <typename T>
+void ProtectionInfoKVOC<T>::UpdateC(ColumnFamilyId old_column_family_id,
+ ColumnFamilyId new_column_family_id) {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&old_column_family_id),
+ sizeof(old_column_family_id), ProtectionInfo<T>::kSeedC));
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&new_column_family_id),
+ sizeof(new_column_family_id), ProtectionInfo<T>::kSeedC));
+ SetVal(val);
+}
+
+template <typename T>
+ProtectionInfoKVOS<T> ProtectionInfoKVO<T>::ProtectS(
+ SequenceNumber sequence_number) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(reinterpret_cast<char*>(&sequence_number),
+ sizeof(sequence_number),
+ ProtectionInfo<T>::kSeedS));
+ return ProtectionInfoKVOS<T>(val);
+}
+
+template <typename T>
+ProtectionInfoKVO<T> ProtectionInfoKVOS<T>::StripS(
+ SequenceNumber sequence_number) const {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(reinterpret_cast<char*>(&sequence_number),
+ sizeof(sequence_number),
+ ProtectionInfo<T>::kSeedS));
+ return ProtectionInfoKVO<T>(val);
+}
+
+template <typename T>
+void ProtectionInfoKVOS<T>::UpdateS(SequenceNumber old_sequence_number,
+ SequenceNumber new_sequence_number) {
+ T val = GetVal();
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&old_sequence_number),
+ sizeof(old_sequence_number), ProtectionInfo<T>::kSeedS));
+ val = val ^ static_cast<T>(NPHash64(
+ reinterpret_cast<char*>(&new_sequence_number),
+ sizeof(new_sequence_number), ProtectionInfo<T>::kSeedS));
+ SetVal(val);
+}
+
+} // namespace ROCKSDB_NAMESPACE