114 static_assert(ValBytes.size() ==
sizeof(
ID.getOpaqueData()));
124 std::optional<ArrayRef<char>> UpstreamValue;
125 if (
Error E = UpstreamKVDB->get(
Key).moveInto(UpstreamValue))
135 PrimaryGraphDB->getReference(UpstreamGraphDB->getDigest(UpstreamID));
137 return PrimaryID.takeError();
154 DirI.increment(EC)) {
159 FoundDBDirs.
push_back({0, std::string(SubDir)});
167 "unexpected directory " + DirI->path());
168 FoundDBDirs.
push_back({Order, std::string(SubDir)});
174 return LHS.Order <
RHS.Order;
178 for (DBDir &Dir : FoundDBDirs)
186 return DBDirs.takeError();
191 for (
unsigned Keep = 2;
Keep > 0 && !DBDirs->empty(); --
Keep) {
215 Args.push_back(
"-check-hash");
220 "llvm-cas-validate-stderr",
"txt", StdErrFD, StdErrPath,
225 std::optional<llvm::StringRef> Redirects[] = {
241 if (!ErrMsg.empty()) {
246 if (StdErrBuf && !(*StdErrBuf)->getBuffer().empty()) {
248 Err += (*StdErrBuf)->getBuffer();
256 unsigned HashByteSize,
bool CheckHash) {
257 std::shared_ptr<UnifiedOnDiskCache> UniDB;
263 if (
Error E = CAS->validate(CheckHash))
266 if (
Error E = Cache->validate())
273 bool CheckHash,
bool AllowRecovery,
bool ForceValidation,
274 std::optional<StringRef> LLVMCasBinaryPath) {
298 if (!Bytes.empty() &&
299 StringRef(Bytes).trim().getAsInteger(10, ValidationBootTime))
308 if (ValidationBootTime == BootTime && !ForceValidation)
312 bool NeedsRecovery =
false;
320 NeedsRecovery =
true;
337 if (EC == std::errc::no_lock_available)
340 "CAS validation requires exclusive access but CAS was in use");
347 return DBDirs.takeError();
353 int Attempt = 0, MaxAttempts = 100;
355 for (; Attempt < MaxAttempts; ++Attempt) {
364 if (Attempt == MaxAttempts)
366 EC,
"rename " + PathBuf +
367 " failed: too many CAS directories awaiting pruning");
370 " failed: " + EC.message());
374 if (ValidationBootTime != BootTime) {
382 OS << BootTime <<
'\n';
392 StringRef HashName,
unsigned HashByteSize,
410 if (std::error_code EC =
416 return DBDirs.takeError();
426 auto UniDB = std::unique_ptr<UnifiedOnDiskCache>(
new UnifiedOnDiskCache());
427 std::unique_ptr<OnDiskGraphDB> UpstreamGraphDB;
428 std::unique_ptr<OnDiskKeyValueDB> UpstreamKVDB;
429 if (DBDirs->size() > 1) {
430 StringRef UpstreamDir = *(DBDirs->end() - 2);
434 nullptr, FaultInPolicy)
435 .moveInto(UpstreamGraphDB))
440 .moveInto(UpstreamKVDB))
444 StringRef PrimaryDir = *(DBDirs->end() - 1);
447 std::unique_ptr<OnDiskGraphDB> PrimaryGraphDB;
449 UpstreamGraphDB.get(), FaultInPolicy)
450 .moveInto(PrimaryGraphDB))
452 std::unique_ptr<OnDiskKeyValueDB> PrimaryKVDB;
459 .moveInto(PrimaryKVDB))
462 UniDB->RootPath = RootPath;
463 UniDB->SizeLimit = SizeLimit.value_or(0);
464 UniDB->LockFD = LockFD;
465 UniDB->NeedsGarbageCollection = DBDirs->size() > 2;
466 UniDB->PrimaryDBDir = PrimaryDir;
467 UniDB->UpstreamGraphDB = std::move(UpstreamGraphDB);
468 UniDB->PrimaryGraphDB = std::move(PrimaryGraphDB);
469 UniDB->UpstreamKVDB = std::move(UpstreamKVDB);
470 UniDB->PrimaryKVDB = std::move(PrimaryKVDB);
472 return std::move(UniDB);
476 this->SizeLimit = SizeLimit.value_or(0);
480 uint64_t TotalSize = getPrimaryStorageSize();
482 TotalSize += UpstreamGraphDB->getStorageSize();
484 TotalSize += UpstreamKVDB->getStorageSize();
488uint64_t UnifiedOnDiskCache::getPrimaryStorageSize()
const {
489 return PrimaryGraphDB->getStorageSize() + PrimaryKVDB->getStorageSize();
498 unsigned CurrentPercent =
499 std::max(PrimaryGraphDB->getHardStorageLimitUtilization(),
500 PrimaryKVDB->getHardStorageLimitUtilization());
501 if (CurrentPercent > 85)
514 return (CurSizeLimit / 2) < getPrimaryStorageSize();
530 UpstreamKVDB.reset();
532 UpstreamGraphDB.reset();
533 PrimaryGraphDB.reset();
537 if (!ExceededSizeLimit)
566 NeedsGarbageCollection =
true;
570UnifiedOnDiskCache::UnifiedOnDiskCache() =
default;
577 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.
ScopedSetting scopedDisable()
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.