13ded69484
Summary: Env::GenerateUniqueId() works fine on Windows and on POSIX where /proc/sys/kernel/random/uuid exists. Our other implementation is flawed and easily produces collision in a new multi-threaded test. As we rely more heavily on DB session ID uniqueness, this becomes a serious issue. This change combines several individually suitable entropy sources for reliable generation of random unique IDs, with goal of uniqueness and portability, not cryptographic strength nor maximum speed. Specifically: * Moves code for getting UUIDs from the OS to port::GenerateRfcUuid rather than in Env implementation details. Callers are now told whether the operation fails or succeeds. * Adds an internal API GenerateRawUniqueId for generating high-quality 128-bit unique identifiers, by combining entropy from three "tracks": * Lots of info from default Env like time, process id, and hostname. * std::random_device * port::GenerateRfcUuid (when working) * Built-in implementations of Env::GenerateUniqueId() will now always produce an RFC 4122 UUID string, either from platform-specific API or by converting the output of GenerateRawUniqueId. DB session IDs now use GenerateRawUniqueId while DB IDs (not as critical) try to use port::GenerateRfcUuid but fall back on GenerateRawUniqueId with conversion to an RFC 4122 UUID. GenerateRawUniqueId is declared and defined under env/ rather than util/ or even port/ because of the Env dependency. Likely follow-up: enhance GenerateRawUniqueId to be faster after the first call and to guarantee uniqueness within the lifetime of a single process (imparting the same property onto DB session IDs). Pull Request resolved: https://github.com/facebook/rocksdb/pull/8708 Test Plan: A new mini-stress test in env_test checks the various public and internal APIs for uniqueness, including each track of GenerateRawUniqueId individually. We can't hope to verify anywhere close to 128 bits of entropy, but it can at least detect flaws as bad as the old code. Serial execution of the new tests takes about 350 ms on my machine. Reviewed By: zhichao-cao, mrambacher Differential Revision: D30563780 Pulled By: pdillinger fbshipit-source-id: de4c9ff4b2f581cf784fcedb5f39f16e5185c364
164 lines
5.4 KiB
C++
164 lines
5.4 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).
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "rocksdb/rocksdb_namespace.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
class Slice;
|
|
|
|
extern std::vector<std::string> StringSplit(const std::string& arg, char delim);
|
|
|
|
template <typename T>
|
|
inline std::string ToString(T value) {
|
|
#if !(defined OS_ANDROID) && !(defined CYGWIN) && !(defined OS_FREEBSD)
|
|
return std::to_string(value);
|
|
#else
|
|
// Andorid or cygwin doesn't support all of C++11, std::to_string() being
|
|
// one of the not supported features.
|
|
std::ostringstream os;
|
|
os << value;
|
|
return os.str();
|
|
#endif
|
|
}
|
|
|
|
// Append a human-readable printout of "num" to *str
|
|
extern void AppendNumberTo(std::string* str, uint64_t num);
|
|
|
|
// Append a human-readable printout of "value" to *str.
|
|
// Escapes any non-printable characters found in "value".
|
|
extern void AppendEscapedStringTo(std::string* str, const Slice& value);
|
|
|
|
// Return a string printout of "num"
|
|
extern std::string NumberToString(uint64_t num);
|
|
|
|
// Put n digits from v in base kBase to (*buf)[0] to (*buf)[n-1] and
|
|
// advance *buf to the position after what was written.
|
|
template <size_t kBase>
|
|
inline void PutBaseChars(char** buf, size_t n, uint64_t v, bool uppercase) {
|
|
const char* digitChars = uppercase ? "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
: "0123456789abcdefghijklmnopqrstuvwxyz";
|
|
for (size_t i = n; i > 0; --i) {
|
|
(*buf)[i - 1] = digitChars[v % kBase];
|
|
v /= kBase;
|
|
}
|
|
*buf += n;
|
|
}
|
|
|
|
// Return a human-readable version of num.
|
|
// for num >= 10.000, prints "xxK"
|
|
// for num >= 10.000.000, prints "xxM"
|
|
// for num >= 10.000.000.000, prints "xxG"
|
|
extern std::string NumberToHumanString(int64_t num);
|
|
|
|
// Return a human-readable version of bytes
|
|
// ex: 1048576 -> 1.00 GB
|
|
extern std::string BytesToHumanString(uint64_t bytes);
|
|
|
|
// Return a human-readable version of unix time
|
|
// ex: 1562116015 -> "Tue Jul 2 18:06:55 2019"
|
|
extern std::string TimeToHumanString(int unixtime);
|
|
|
|
// Append a human-readable time in micros.
|
|
int AppendHumanMicros(uint64_t micros, char* output, int len,
|
|
bool fixed_format);
|
|
|
|
// Append a human-readable size in bytes
|
|
int AppendHumanBytes(uint64_t bytes, char* output, int len);
|
|
|
|
// Return a human-readable version of "value".
|
|
// Escapes any non-printable characters found in "value".
|
|
extern std::string EscapeString(const Slice& value);
|
|
|
|
// Parse a human-readable number from "*in" into *value. On success,
|
|
// advances "*in" past the consumed number and sets "*val" to the
|
|
// numeric value. Otherwise, returns false and leaves *in in an
|
|
// unspecified state.
|
|
extern bool ConsumeDecimalNumber(Slice* in, uint64_t* val);
|
|
|
|
// Returns true if the input char "c" is considered as a special character
|
|
// that will be escaped when EscapeOptionString() is called.
|
|
//
|
|
// @param c the input char
|
|
// @return true if the input char "c" is considered as a special character.
|
|
// @see EscapeOptionString
|
|
bool isSpecialChar(const char c);
|
|
|
|
// If the input char is an escaped char, it will return the its
|
|
// associated raw-char. Otherwise, the function will simply return
|
|
// the original input char.
|
|
char UnescapeChar(const char c);
|
|
|
|
// If the input char is a control char, it will return the its
|
|
// associated escaped char. Otherwise, the function will simply return
|
|
// the original input char.
|
|
char EscapeChar(const char c);
|
|
|
|
// Converts a raw string to an escaped string. Escaped-characters are
|
|
// defined via the isSpecialChar() function. When a char in the input
|
|
// string "raw_string" is classified as a special characters, then it
|
|
// will be prefixed by '\' in the output.
|
|
//
|
|
// It's inverse function is UnescapeOptionString().
|
|
// @param raw_string the input string
|
|
// @return the '\' escaped string of the input "raw_string"
|
|
// @see isSpecialChar, UnescapeOptionString
|
|
std::string EscapeOptionString(const std::string& raw_string);
|
|
|
|
// The inverse function of EscapeOptionString. It converts
|
|
// an '\' escaped string back to a raw string.
|
|
//
|
|
// @param escaped_string the input '\' escaped string
|
|
// @return the raw string of the input "escaped_string"
|
|
std::string UnescapeOptionString(const std::string& escaped_string);
|
|
|
|
std::string trim(const std::string& str);
|
|
|
|
// Returns true if "string" ends with "pattern"
|
|
bool EndsWith(const std::string& string, const std::string& pattern);
|
|
|
|
// Returns true if "string" starts with "pattern"
|
|
bool StartsWith(const std::string& string, const std::string& pattern);
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
bool ParseBoolean(const std::string& type, const std::string& value);
|
|
|
|
uint8_t ParseUint8(const std::string& value);
|
|
|
|
uint32_t ParseUint32(const std::string& value);
|
|
|
|
int32_t ParseInt32(const std::string& value);
|
|
#endif
|
|
|
|
uint64_t ParseUint64(const std::string& value);
|
|
|
|
int ParseInt(const std::string& value);
|
|
|
|
int64_t ParseInt64(const std::string& value);
|
|
|
|
double ParseDouble(const std::string& value);
|
|
|
|
size_t ParseSizeT(const std::string& value);
|
|
|
|
std::vector<int> ParseVectorInt(const std::string& value);
|
|
|
|
bool SerializeIntVector(const std::vector<int>& vec, std::string* value);
|
|
|
|
extern const std::string kNullptrString;
|
|
|
|
// errnoStr() function returns a string that describes the error code passed in
|
|
// the argument err
|
|
extern std::string errnoStr(int err);
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|