113 static_assert(ValBytes.size() ==
sizeof(
ID.getOpaqueData()));
123 std::optional<ArrayRef<char>> UpstreamValue;
124 if (
Error E = UpstreamKVDB->get(
Key).moveInto(UpstreamValue))
134 PrimaryGraphDB->getReference(UpstreamGraphDB->getDigest(UpstreamID));
136 return PrimaryID.takeError();
153 DirI.increment(EC)) {
158 FoundDBDirs.
push_back({0, std::string(SubDir)});
166 "unexpected directory " + DirI->path());
167 FoundDBDirs.
push_back({Order, std::string(SubDir)});
173 return LHS.Order <
RHS.Order;
177 for (DBDir &Dir : FoundDBDirs)
185 return DBDirs.takeError();
190 for (
unsigned Keep = 2;
Keep > 0 && !DBDirs->empty(); --
Keep) {
214 Args.push_back(
"-check-hash");
219 "llvm-cas-validate-stderr",
"txt", StdErrFD, StdErrPath,
224 std::optional<llvm::StringRef> Redirects[] = {
240 if (!ErrMsg.empty()) {
245 if (StdErrBuf && !(*StdErrBuf)->getBuffer().empty()) {
247 Err += (*StdErrBuf)->getBuffer();
255 unsigned HashByteSize,
bool CheckHash) {
256 std::shared_ptr<UnifiedOnDiskCache> UniDB;
262 if (
Error E = CAS->validate(CheckHash))
265 if (
Error E = Cache->validate())
272 bool CheckHash,
bool AllowRecovery,
bool ForceValidation,
273 std::optional<StringRef> LLVMCasBinaryPath) {
297 if (!Bytes.empty() &&
298 StringRef(Bytes).trim().getAsInteger(10, ValidationBootTime))
307 if (ValidationBootTime == BootTime && !ForceValidation)
311 bool NeedsRecovery =
false;
319 NeedsRecovery =
true;
336 if (EC == std::errc::no_lock_available)
339 "CAS validation requires exclusive access but CAS was in use");
346 return DBDirs.takeError();
352 int Attempt = 0, MaxAttempts = 100;
354 for (; Attempt < MaxAttempts; ++Attempt) {
363 if (Attempt == MaxAttempts)
365 EC,
"rename " + PathBuf +
366 " failed: too many CAS directories awaiting pruning");
369 " failed: " + EC.message());
373 if (ValidationBootTime != BootTime) {
381 OS << BootTime <<
'\n';
391 StringRef HashName,
unsigned HashByteSize,
407 if (std::error_code EC =
413 return DBDirs.takeError();
423 auto UniDB = std::unique_ptr<UnifiedOnDiskCache>(
new UnifiedOnDiskCache());
424 std::unique_ptr<OnDiskGraphDB> UpstreamGraphDB;
425 std::unique_ptr<OnDiskKeyValueDB> UpstreamKVDB;
426 if (DBDirs->size() > 1) {
427 StringRef UpstreamDir = *(DBDirs->end() - 2);
431 nullptr, FaultInPolicy)
432 .moveInto(UpstreamGraphDB))
437 .moveInto(UpstreamKVDB))
441 StringRef PrimaryDir = *(DBDirs->end() - 1);
444 std::unique_ptr<OnDiskGraphDB> PrimaryGraphDB;
446 UpstreamGraphDB.get(), FaultInPolicy)
447 .moveInto(PrimaryGraphDB))
449 std::unique_ptr<OnDiskKeyValueDB> PrimaryKVDB;
456 .moveInto(PrimaryKVDB))
459 UniDB->RootPath = RootPath;
460 UniDB->SizeLimit = SizeLimit.value_or(0);
461 UniDB->LockFD = LockFD;
462 UniDB->NeedsGarbageCollection = DBDirs->size() > 2;
463 UniDB->PrimaryDBDir = PrimaryDir;
464 UniDB->UpstreamGraphDB = std::move(UpstreamGraphDB);
465 UniDB->PrimaryGraphDB = std::move(PrimaryGraphDB);
466 UniDB->UpstreamKVDB = std::move(UpstreamKVDB);
467 UniDB->PrimaryKVDB = std::move(PrimaryKVDB);
469 return std::move(UniDB);
473 this->SizeLimit = SizeLimit.value_or(0);
477 uint64_t TotalSize = getPrimaryStorageSize();
479 TotalSize += UpstreamGraphDB->getStorageSize();
481 TotalSize += UpstreamKVDB->getStorageSize();
485uint64_t UnifiedOnDiskCache::getPrimaryStorageSize()
const {
486 return PrimaryGraphDB->getStorageSize() + PrimaryKVDB->getStorageSize();
495 unsigned CurrentPercent =
496 std::max(PrimaryGraphDB->getHardStorageLimitUtilization(),
497 PrimaryKVDB->getHardStorageLimitUtilization());
498 if (CurrentPercent > 85)
511 return (CurSizeLimit / 2) < getPrimaryStorageSize();
525 UpstreamKVDB.reset();
527 UpstreamGraphDB.reset();
528 PrimaryGraphDB.reset();
532 if (!ExceededSizeLimit)
561 NeedsGarbageCollection =
true;
565UnifiedOnDiskCache::UnifiedOnDiskCache() =
default;
572 return DBDirs.takeError();
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
This file contains the declaration of the ActionCache class, which is the base class for ActionCache ...
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
#define LLVM_UNLIKELY(EXPR)
This declares OnDiskGraphDB, an ondisk CAS database with a fixed length hash.
This declares OnDiskKeyValueDB, a key value storage database of fixed size key and value.
This file defines the make_scope_exit function, which executes user-defined cleanup logic at scope ex...
This file defines the SmallString class.
This file defines the SmallVector class.
static constexpr StringLiteral DBDirPrefix
FIXME: When the version of DBDirPrefix is bumped up we need to figure out how to handle the leftover ...
static Error validateInProcess(StringRef RootPath, StringRef HashName, unsigned HashByteSize, bool CheckHash)
static Expected< SmallVector< std::string, 4 > > getAllGarbageDirs(StringRef Path)
static constexpr StringLiteral ValidationFilename
static constexpr StringLiteral CorruptPrefix
static void getNextDBDirName(StringRef DBDir, llvm::raw_ostream &OS)
static Error validateOutOfProcess(StringRef LLVMCasBinary, StringRef RootPath, bool CheckHash)
static Expected< SmallVector< std::string, 4 > > getAllDBDirs(StringRef Path, bool IncludeCorrupt=false)
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory),...
Lightweight error class with error context and mandatory checking.
static ErrorSuccess success()
Create a success value.
Tagged union holding either a T or a Error.
FileRemover - This class is a simple object meant to be stack allocated.
static ErrorOr< std::unique_ptr< MemoryBuffer > > getFile(const Twine &Filename, bool IsText=false, bool RequiresNullTerminator=true, bool IsVolatile=false, std::optional< Align > Alignment=std::nullopt)
Open the specified file as a MemoryBuffer, returning a new MemoryBuffer if successful,...
SmallString - A SmallString is just a SmallVector with methods and accessors that make it work better...
void assign(StringRef RHS)
Assign from a StringRef.
StringRef str() const
Explicit conversion to StringRef.
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
A wrapper around a string literal that serves as a proxy for constructing global tables of StringRefs...
StringRef - Represent a constant reference to a string, i.e.
bool getAsInteger(unsigned Radix, T &Result) const
Parse the current string as an integer of the specified radix.
constexpr StringRef substr(size_t Start, size_t N=npos) const
Return a reference to the substring from [Start, Start + N).
bool starts_with(StringRef Prefix) const
Check if this string starts with the given Prefix.
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
LLVM Value Representation.
static ObjectID fromOpaqueData(uint64_t Opaque)
FaultInPolicy
How to fault-in nodes if an upstream database is used.
static LLVM_ABI_FOR_TEST Expected< std::unique_ptr< OnDiskGraphDB > > open(StringRef Path, StringRef HashName, unsigned HashByteSize, OnDiskGraphDB *UpstreamDB=nullptr, FaultInPolicy Policy=FaultInPolicy::FullTree)
Open the on-disk store from a directory.
static LLVM_ABI_FOR_TEST Expected< std::unique_ptr< OnDiskKeyValueDB > > open(StringRef Path, StringRef HashName, unsigned KeySize, StringRef ValueName, size_t ValueSize, UnifiedOnDiskCache *UnifiedCache=nullptr)
Open the on-disk store from a directory.
LLVM_ABI_FOR_TEST uint64_t getStorageSize() const
static LLVM_ABI_FOR_TEST ValueBytes getValueFromObjectID(ObjectID ID)
static LLVM_ABI_FOR_TEST Expected< std::unique_ptr< UnifiedOnDiskCache > > open(StringRef Path, std::optional< uint64_t > SizeLimit, StringRef HashName, unsigned HashByteSize, OnDiskGraphDB::FaultInPolicy FaultInPolicy=OnDiskGraphDB::FaultInPolicy::FullTree)
Open a UnifiedOnDiskCache instance for a directory.
LLVM_ABI_FOR_TEST Error close(bool CheckSizeLimit=true)
This is called implicitly at destruction time, so it is not required for a client to call this.
static LLVM_ABI_FOR_TEST ObjectID getObjectIDFromValue(ArrayRef< char > Value)
Helper function to convert the value stored in KeyValueDB and ObjectID.
static Expected< ValidationResult > validateIfNeeded(StringRef Path, StringRef HashName, unsigned HashByteSize, bool CheckHash, bool AllowRecovery, bool ForceValidation, std::optional< StringRef > LLVMCasBinary)
Validate the data in Path, if needed to ensure correctness.
LLVM_ABI_FOR_TEST bool hasExceededSizeLimit() const
LLVM_ABI_FOR_TEST ~UnifiedOnDiskCache()
std::array< char, sizeof(uint64_t)> ValueBytes
Error collectGarbage()
Remove unused data from the current UnifiedOnDiskCache.
LLVM_ABI_FOR_TEST void setSizeLimit(std::optional< uint64_t > SizeLimit)
Set the size for limiting growth.
A raw_ostream that writes to a file descriptor.
bool has_error() const
Return the value of the flag in this raw_fd_ostream indicating whether an output error has been encou...
std::error_code error() const
uint64_t seek(uint64_t off)
Flushes the stream and repositions the underlying file descriptor position to the offset specified fr...
This class implements an extremely fast bulk output stream that can only output to a stream.
A raw_ostream that writes to an SmallVector or SmallString.
directory_iterator - Iterates through the entries in path.
unsigned ID
LLVM IR allows to use arbitrary numbers as calling convention identifiers.
std::unique_ptr< ObjectStore > createObjectStoreFromUnifiedOnDiskCache(std::shared_ptr< ondisk::UnifiedOnDiskCache > UniDB)
std::unique_ptr< ActionCache > createActionCacheFromUnifiedOnDiskCache(std::shared_ptr< ondisk::UnifiedOnDiskCache > UniDB)
std::error_code lockFileThreadSafe(int FD, llvm::sys::fs::LockKind Kind)
Thread-safe alternative to sys::fs::lockFile.
std::error_code unlockFileThreadSafe(int FD)
Thread-safe alternative to sys::fs::unlockFile.
std::error_code tryLockFileThreadSafe(int FD, std::chrono::milliseconds Timeout=std::chrono::milliseconds(0), llvm::sys::fs::LockKind Kind=llvm::sys::fs::LockKind::Exclusive)
Thread-safe alternative to sys::fs::tryLockFile.
Expected< uint64_t > getBootTime()
Get boot time for the OS.
@ Valid
The data is already valid.
@ Recovered
The data was invalid, but was recovered.
@ Skipped
Validation was skipped, as it was not needed.
uint64_t read64le(const void *P)
void write64le(void *P, uint64_t V)
LLVM_ABI std::error_code closeFile(file_t &F)
Close the file object.
std::error_code openFileForReadWrite(const Twine &Name, int &ResultFD, CreationDisposition Disp, OpenFlags Flags, unsigned Mode=0666)
Opens the file with the given name in a write-only or read-write mode, returning its open file descri...
LLVM_ABI std::error_code rename(const Twine &from, const Twine &to)
Rename from to to.
LLVM_ABI Error readNativeFileToEOF(file_t FileHandle, SmallVectorImpl< char > &Buffer, ssize_t ChunkSize=DefaultReadChunkSize)
Reads from FileHandle until EOF, appending to Buffer in chunks of size ChunkSize.
@ OF_Text
The file should be opened in text mode on platforms like z/OS that make this distinction.
@ CD_OpenAlways
CD_OpenAlways - When opening a file:
LLVM_ABI std::error_code create_directories(const Twine &path, bool IgnoreExisting=true, perms Perms=owner_all|group_all)
Create all the non-existent directories in path.
LLVM_ABI std::error_code createTemporaryFile(const Twine &Prefix, StringRef Suffix, int &ResultFD, SmallVectorImpl< char > &ResultPath, OpenFlags Flags=OF_None)
Create a file in the system temporary directory.
LLVM_ABI std::error_code resize_file(int FD, uint64_t Size)
Resize path to size.
LLVM_ABI file_t convertFDToNativeFile(int FD)
Converts from a Posix file descriptor number to a native file handle.
LLVM_ABI std::error_code create_directory(const Twine &path, bool IgnoreExisting=true, perms Perms=owner_all|group_all)
Create the directory in path.
LLVM_ABI std::error_code remove_directories(const Twine &path, bool IgnoreErrors=true)
Recursively delete a directory.
LLVM_ABI StringRef get_separator(Style style=Style::native)
Return the preferred separator for this platform.
LLVM_ABI void remove_filename(SmallVectorImpl< char > &path, Style style=Style::native)
Remove the last component from path unless it is the root dir.
LLVM_ABI StringRef filename(StringRef path LLVM_LIFETIME_BOUND, Style style=Style::native)
Get filename.
LLVM_ABI void append(SmallVectorImpl< char > &path, const Twine &a, const Twine &b="", const Twine &c="", const Twine &d="")
Append to path.
LLVM_ABI int ExecuteAndWait(StringRef Program, ArrayRef< StringRef > Args, std::optional< ArrayRef< StringRef > > Env=std::nullopt, ArrayRef< std::optional< StringRef > > Redirects={}, unsigned SecondsToWait=0, unsigned MemoryLimit=0, std::string *ErrMsg=nullptr, bool *ExecutionFailed=nullptr, std::optional< ProcessStatistics > *ProcStat=nullptr, BitVector *AffinityMask=nullptr)
This function executes the program using the arguments provided.
This is an optimization pass for GlobalISel generic memory operations.
Error createFileError(const Twine &F, Error E)
Concatenate a source file path and/or name with an Error.
detail::scope_exit< std::decay_t< Callable > > make_scope_exit(Callable &&F)
LLVM_ABI std::error_code inconvertibleErrorCode()
The value returned by this function can be returned from convertToErrorCode for Error values where no...
testing::Matcher< const detail::ErrorHolder & > Failed()
Error createStringError(std::error_code EC, char const *Fmt, const Ts &... Vals)
Create formatted StringError object.
void sort(IteratorTy Start, IteratorTy End)
FunctionAddr VTableAddr Count
LLVM_ATTRIBUTE_VISIBILITY_DEFAULT AnalysisKey InnerAnalysisManagerProxy< AnalysisManagerT, IRUnitT, ExtraArgTs... >::Key
std::string join(IteratorT Begin, IteratorT End, StringRef Separator)
Joins the strings in the range [Begin, End), adding Separator between the elements.
void consumeError(Error Err)
Consume a Error without doing anything.
@ Keep
No function return thunk.