You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
955a26ecbf
* Fixed so visualizer can visualize traces coming from UbaCli without needing restart * Added some more logging to cache server maintenance * Fixed scroll wheel zoom in visualizer * Fixed so network server DisconnectClients does not put server in a bad state [CL 34213980 by henrik karlsson in ue5-main branch]
1535 lines
47 KiB
C++
1535 lines
47 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "UbaCacheServer.h"
|
|
#include "UbaCacheEntry.h"
|
|
#include "UbaCompactTables.h"
|
|
#include "UbaBinaryReaderWriter.h"
|
|
#include "UbaDirectoryIterator.h"
|
|
#include "UbaFileAccessor.h"
|
|
#include "UbaNetworkServer.h"
|
|
#include "UbaStorageServer.h"
|
|
//#include <oodle2.h>
|
|
|
|
namespace uba
|
|
{
|
|
static constexpr u32 CacheFileVersion = 5;
|
|
static constexpr u32 CacheFileCompatibilityVersion = 3;
|
|
|
|
bool IsCaseInsensitive(u64 id) { return (id & (1ull << 32)) == 0; }
|
|
|
|
struct CacheServer::ConnectionBucket
|
|
{
|
|
ConnectionBucket(u64 i) : pathTable(CachePathTableMaxSize, CompactPathTable::V1, IsCaseInsensitive(i)), casKeyTable(CacheCasKeyTableMaxSize), id(i) {}
|
|
CompactPathTable pathTable;
|
|
CompactCasKeyTable casKeyTable;
|
|
|
|
ReaderWriterLock cacheEntryLookupLock;
|
|
UnorderedMap<CasKey, CacheEntry> cacheEntryLookup;
|
|
u64 id;
|
|
};
|
|
|
|
struct CacheServer::Connection
|
|
{
|
|
u32 clientVersion;
|
|
UnorderedMap<u64, ConnectionBucket> buckets;
|
|
};
|
|
|
|
struct CacheServer::Bucket
|
|
{
|
|
Bucket(u64 id) : m_pathTable(CachePathTableMaxSize, CompactPathTable::V1, IsCaseInsensitive(id)), m_casKeyTable(CacheCasKeyTableMaxSize) {}
|
|
ReaderWriterLock m_cacheEntryLookupLock;
|
|
UnorderedMap<CasKey, CacheEntries> m_cacheEntryLookup;
|
|
|
|
CompactPathTable m_pathTable;
|
|
CompactCasKeyTable m_casKeyTable;
|
|
|
|
Atomic<u64> totalEntryCount;
|
|
Atomic<u64> totalEntrySize;
|
|
Atomic<bool> hasDeletedEntries;
|
|
Atomic<bool> needsSave;
|
|
|
|
Atomic<u64> lastSavedTime;
|
|
Atomic<u64> lastUsedTime;
|
|
|
|
u32 index = ~0u;
|
|
};
|
|
|
|
const tchar* ToString(CacheMessageType type)
|
|
{
|
|
switch (type)
|
|
{
|
|
#define UBA_CACHE_MESSAGE(x) case CacheMessageType_##x: return TC("")#x;
|
|
UBA_CACHE_MESSAGES
|
|
#undef UBA_CACHE_MESSAGE
|
|
default:
|
|
return TC("Unknown"); // Should never happen
|
|
}
|
|
}
|
|
|
|
CacheServer::CacheServer(const CacheServerCreateInfo& info)
|
|
: m_logger(info.logWriter, TC("UbaCacheServer"))
|
|
, m_server(info.storage.GetServer())
|
|
, m_storage(info.storage)
|
|
{
|
|
m_checkInputsForDeletedCas = info.checkInputsForDeletedCas;
|
|
m_startTime = GetTime();
|
|
|
|
m_expirationTimeSeconds = info.expirationTimeSeconds;
|
|
|
|
m_rootDir.count = GetFullPathNameW(info.rootDir, m_rootDir.capacity, m_rootDir.data, NULL);
|
|
m_rootDir.Replace('/', PathSeparator).EnsureEndsWithSlash();
|
|
|
|
m_server.RegisterService(CacheServiceId,
|
|
[this](const ConnectionInfo& connectionInfo, MessageInfo& messageInfo, BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
return HandleMessage(connectionInfo, messageInfo.type, reader, writer);
|
|
},
|
|
[](u8 messageType)
|
|
{
|
|
return ToString(CacheMessageType(messageType));
|
|
}
|
|
);
|
|
|
|
m_server.RegisterOnClientDisconnected(CacheServiceId, [this](const Guid& clientUid, u32 clientId)
|
|
{
|
|
OnDisconnected(clientId);
|
|
});
|
|
}
|
|
|
|
CacheServer::~CacheServer()
|
|
{
|
|
}
|
|
|
|
struct CacheServer::LoadStats
|
|
{
|
|
Atomic<u32> totalPathTableSize;
|
|
Atomic<u32> totalCasKeyTableSize;
|
|
Atomic<u64> totalCacheEntryCount;
|
|
};
|
|
|
|
bool CacheServer::Load()
|
|
{
|
|
u64 startTime = GetTime();
|
|
|
|
StringBuffer<> fileName(m_rootDir);
|
|
fileName.EnsureEndsWithSlash().Append(TC("cachedb"));
|
|
|
|
FileAccessor file(m_logger, fileName.data);
|
|
if (!file.OpenMemoryRead(0, false))
|
|
{
|
|
m_logger.Detail(TC("No database found. Starting a new one at %s"), fileName.data);
|
|
m_creationTime = GetSystemTimeAsFileTime();
|
|
m_dbfileDirty = true;
|
|
return true;
|
|
}
|
|
BinaryReader reader(file.GetData(), 0, file.GetSize());
|
|
|
|
u32 databaseVersion = reader.ReadU32();
|
|
if (databaseVersion < CacheFileCompatibilityVersion || databaseVersion > CacheFileVersion)
|
|
{
|
|
m_logger.Detail(TC("Can't load database of version %u. Starting a new one at %s"), databaseVersion, fileName.data);
|
|
return true;
|
|
}
|
|
if (databaseVersion == 3)
|
|
m_creationTime = GetSystemTimeAsFileTime() - 1;
|
|
else
|
|
m_creationTime = reader.ReadU64();
|
|
|
|
if (databaseVersion != CacheFileVersion)
|
|
m_dbfileDirty = true;
|
|
|
|
LoadStats stats;
|
|
|
|
if (databaseVersion == 4)
|
|
{
|
|
u32 bucketCount = reader.ReadU32();
|
|
while (bucketCount--)
|
|
{
|
|
Bucket& bucket = GetBucket(reader.ReadU64());
|
|
LoadBucket(bucket, reader, databaseVersion, stats);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
StringBuffer<MaxPath> bucketsDir(m_rootDir);
|
|
bucketsDir.EnsureEndsWithSlash().Append(TC("buckets"));
|
|
TraverseDir(m_logger, bucketsDir.data, [&](const DirectoryEntry& e)
|
|
{
|
|
StringBuffer<128> keyName;
|
|
keyName.Append(e.name, e.nameLen);
|
|
u64 id;
|
|
if (!keyName.Parse(id))
|
|
return;
|
|
GetBucket(id);
|
|
});
|
|
|
|
m_server.ParallelFor(GetBucketWorkerCount(), m_buckets, [&](auto& it)
|
|
{
|
|
u64 key = it->first;
|
|
|
|
StringBuffer<MaxPath> bucketFilename(bucketsDir);
|
|
bucketFilename.EnsureEndsWithSlash().AppendValue(key);
|
|
FileAccessor bucketFile(m_logger, bucketFilename.data);
|
|
if (!bucketFile.OpenMemoryRead(0, false))
|
|
{
|
|
m_logger.Detail(TC("Failed to open bucket file %s"), bucketFilename.data);
|
|
return;
|
|
}
|
|
BinaryReader reader(bucketFile.GetData(), 0, bucketFile.GetSize());
|
|
u32 bucketVersion = reader.ReadU32();
|
|
LoadBucket(it->second, reader, bucketVersion, stats);
|
|
});
|
|
}
|
|
|
|
u64 duration = GetTime() - startTime;
|
|
m_logger.Detail(TC("Database (v%u) loaded from %s in %s (%llu bucket(s) containing %s paths, %s keys, %llu cache entries)"), databaseVersion, fileName.data, TimeToText(duration).str, m_buckets.size(), BytesToText(stats.totalPathTableSize).str, BytesToText(stats.totalCasKeyTableSize).str, stats.totalCacheEntryCount.load());
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::LoadBucket(Bucket& bucket, BinaryReader& reader, u32 databaseVersion, LoadStats& outStats)
|
|
{
|
|
if (databaseVersion != CacheFileVersion)
|
|
bucket.needsSave = true;
|
|
|
|
u32 pathTableSize = reader.ReadU32();
|
|
if (pathTableSize)
|
|
{
|
|
BinaryReader pathTableReader(reader.GetPositionData(), 0, pathTableSize);
|
|
bucket.m_pathTable.ReadMem(pathTableReader, true);
|
|
reader.Skip(pathTableSize);
|
|
}
|
|
outStats.totalPathTableSize += pathTableSize;
|
|
|
|
u32 casKeyTableSize = reader.ReadU32();
|
|
if (casKeyTableSize)
|
|
{
|
|
BinaryReader casKeyTableReader(reader.GetPositionData(), 0, casKeyTableSize);
|
|
bucket.m_casKeyTable.ReadMem(casKeyTableReader, true);
|
|
reader.Skip(casKeyTableSize);
|
|
}
|
|
outStats.totalCasKeyTableSize += casKeyTableSize;
|
|
|
|
u32 entryLookupCount = reader.ReadU32();
|
|
bucket.m_cacheEntryLookup.reserve(entryLookupCount);
|
|
|
|
while (entryLookupCount--)
|
|
{
|
|
auto insres = bucket.m_cacheEntryLookup.try_emplace(reader.ReadCasKey());
|
|
UBA_ASSERT(insres.second);
|
|
auto& cacheEntries = insres.first->second;
|
|
cacheEntries.Read(m_logger, reader, databaseVersion);
|
|
outStats.totalCacheEntryCount += cacheEntries.entries.size();
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::Save()
|
|
{
|
|
for (auto& kv : m_buckets)
|
|
{
|
|
Bucket& bucket = kv.second;
|
|
if (bucket.lastSavedTime < bucket.lastUsedTime)
|
|
bucket.needsSave = true;
|
|
}
|
|
|
|
return SaveNoLock();
|
|
}
|
|
|
|
struct FileWriter
|
|
{
|
|
static constexpr u64 TempBufferSize = 1024*1024;
|
|
|
|
FileWriter(Logger& l, const tchar* fn)
|
|
: logger(l)
|
|
, fileName(fn)
|
|
, tempFileName(StringBuffer<MaxPath>(fn).Append(TC(".tmp")).data)
|
|
, file(logger, tempFileName.c_str())
|
|
{
|
|
tempBuffer = (u8*)malloc(TempBufferSize);
|
|
}
|
|
|
|
~FileWriter()
|
|
{
|
|
free(tempBuffer);
|
|
}
|
|
|
|
void WriteBytes(const void* data, u64 size)
|
|
{
|
|
u8* readPos = (u8*)data;
|
|
u64 left = size;
|
|
while (left)
|
|
{
|
|
if (tempBufferPos != TempBufferSize)
|
|
{
|
|
u64 toWrite = Min(TempBufferSize - tempBufferPos, left);
|
|
memcpy(tempBuffer+tempBufferPos, readPos, toWrite);
|
|
tempBufferPos += toWrite;
|
|
left -= toWrite;
|
|
readPos += toWrite;
|
|
}
|
|
else
|
|
{
|
|
written += tempBufferPos;
|
|
success &= file.Write(tempBuffer, tempBufferPos);
|
|
tempBufferPos = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void Write(const T& v)
|
|
{
|
|
WriteBytes(&v, sizeof(v));
|
|
}
|
|
|
|
bool Create() { return file.CreateWrite(); }
|
|
|
|
bool Close()
|
|
{
|
|
success &= file.Write(tempBuffer, tempBufferPos);
|
|
written += tempBufferPos;
|
|
|
|
if (!success)
|
|
return false;
|
|
|
|
if (!file.Close())
|
|
return false;
|
|
|
|
if (!MoveFileExW(tempFileName.data(), fileName.data(), MOVEFILE_REPLACE_EXISTING))
|
|
return logger.Error(TC("Can't move file from %s to %s (%s)"), tempFileName.data(), fileName.data(), LastErrorToText().data);
|
|
|
|
return true;
|
|
}
|
|
|
|
Logger& logger;
|
|
bool success = true;
|
|
u8* tempBuffer = nullptr;
|
|
u64 tempBufferPos = 0;
|
|
u64 written = 0;
|
|
TString fileName;
|
|
TString tempFileName;
|
|
FileAccessor file;
|
|
};
|
|
|
|
bool CacheServer::SaveBucket(u64 bucketId, Bucket& bucket)
|
|
{
|
|
u64 saveStart = GetTime();
|
|
|
|
StringBuffer<MaxPath> bucketsDir(m_rootDir);
|
|
bucketsDir.EnsureEndsWithSlash().Append(TC("buckets"));
|
|
if (!m_storage.CreateDirectory(bucketsDir.data))
|
|
return false;
|
|
bucketsDir.EnsureEndsWithSlash();
|
|
StringBuffer<MaxPath> bucketsFile(bucketsDir);
|
|
bucketsFile.AppendValue(bucketId);
|
|
|
|
FileWriter file(m_logger, bucketsFile.data);
|
|
|
|
if (!file.Create())
|
|
return false;
|
|
|
|
file.Write(CacheFileVersion);
|
|
|
|
u32 pathTableSize = bucket.m_pathTable.GetSize();
|
|
file.Write(pathTableSize);
|
|
file.WriteBytes(bucket.m_pathTable.GetMemory(), pathTableSize);
|
|
|
|
u32 casKeyTableSize = bucket.m_casKeyTable.GetSize();
|
|
file.Write(casKeyTableSize);
|
|
file.WriteBytes(bucket.m_casKeyTable.GetMemory(), casKeyTableSize);
|
|
|
|
u32 entryLookupCount = u32(bucket.m_cacheEntryLookup.size());
|
|
file.Write(entryLookupCount);
|
|
|
|
Vector<u8> temp;
|
|
|
|
for (auto& kv2 : bucket.m_cacheEntryLookup)
|
|
{
|
|
file.Write(kv2.first);
|
|
|
|
temp.resize(kv2.second.GetTotalSize(true));
|
|
BinaryWriter writer(temp.data(), 0, temp.size());
|
|
kv2.second.Write(writer, CacheNetworkVersion, true);
|
|
UBA_ASSERT(writer.GetPosition() == temp.size());
|
|
file.WriteBytes(temp.data(), temp.size());
|
|
}
|
|
|
|
if (!file.Close())
|
|
return false;
|
|
|
|
bucket.lastSavedTime = GetSystemTimeAsFileTime() - m_creationTime;
|
|
|
|
m_logger.Detail(TC(" Bucket %u saved - %s (%s)"), bucket.index, BytesToText(file.written).str, TimeToText(GetTime() - saveStart).str);
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::SaveNoLock()
|
|
{
|
|
if (m_dbfileDirty)
|
|
{
|
|
StringBuffer<MaxPath> fileName(m_rootDir);
|
|
fileName.EnsureEndsWithSlash().Append(TC("cachedb"));
|
|
|
|
FileWriter file(m_logger, fileName.data);
|
|
|
|
if (!file.Create())
|
|
return false;
|
|
|
|
file.Write(CacheFileVersion);
|
|
|
|
file.Write(m_creationTime);
|
|
|
|
if (!file.Close())
|
|
return false;
|
|
m_dbfileDirty = false;
|
|
}
|
|
|
|
StringBuffer<MaxPath> bucketsDir(m_rootDir);
|
|
bucketsDir.EnsureEndsWithSlash().Append(TC("buckets"));
|
|
if (!m_storage.CreateDirectory(bucketsDir.data))
|
|
return false;
|
|
bucketsDir.EnsureEndsWithSlash();
|
|
|
|
Atomic<bool> success = true;
|
|
|
|
for (auto it=m_buckets.begin(); it!=m_buckets.end();)
|
|
{
|
|
Bucket& bucket = it->second;
|
|
if (!bucket.m_cacheEntryLookup.empty())
|
|
{
|
|
++it;
|
|
continue;
|
|
}
|
|
|
|
StringBuffer<MaxPath> bucketsFile(m_rootDir);
|
|
bucketsFile.EnsureEndsWithSlash().Append(TC("buckets")).EnsureEndsWithSlash().AppendValue(it->first);
|
|
DeleteFileW(bucketsFile.data);
|
|
m_logger.Detail(TC(" Bucket %u was empty. Deleted"), bucket.index);
|
|
it = m_buckets.erase(it);
|
|
}
|
|
|
|
m_server.ParallelFor(GetBucketWorkerCount(), m_buckets, [&, temp = Vector<u8>()](auto& it) mutable
|
|
{
|
|
Bucket& bucket = it->second;
|
|
if (!bucket.needsSave)
|
|
return;
|
|
if (SaveBucket(it->first, bucket))
|
|
bucket.needsSave = false;
|
|
else
|
|
success = false;
|
|
});
|
|
|
|
return success;
|
|
}
|
|
|
|
bool CacheServer::RunMaintenance(bool force, const Function<bool()>& shouldExit)
|
|
{
|
|
if (m_addsSinceMaintenance == 0 && !force)
|
|
return true;
|
|
|
|
SCOPED_WRITE_LOCK(m_connectionsLock, lock2);
|
|
if (!m_connections.empty())
|
|
return true;
|
|
m_isRunningMaintenance = true;
|
|
lock2.Leave();
|
|
|
|
auto g = MakeGuard([&]()
|
|
{
|
|
SCOPED_WRITE_LOCK(m_connectionsLock, lock2);
|
|
m_isRunningMaintenance = false;
|
|
});
|
|
|
|
|
|
//m_forceAllSteps = true;
|
|
bool forceAllSteps = m_forceAllSteps;
|
|
m_forceAllSteps = false;
|
|
|
|
if (m_shouldWipe)
|
|
{
|
|
m_shouldWipe = false;
|
|
m_logger.Info(TC("Obliterating database"));
|
|
m_addsSinceMaintenance = 0;
|
|
m_longestMaintenance = 0;
|
|
m_buckets.clear();
|
|
forceAllSteps = true;
|
|
m_creationTime = GetSystemTimeAsFileTime();
|
|
}
|
|
else
|
|
{
|
|
m_logger.Info(TC("Maintenance started after %u added cache entries"), m_addsSinceMaintenance.load());
|
|
}
|
|
|
|
u64 startTime = GetTime();
|
|
|
|
bool entriesAdded = m_addsSinceMaintenance != 0;
|
|
m_addsSinceMaintenance = 0;
|
|
|
|
UnorderedSet<CasKey> deletedCasFiles;
|
|
m_storage.HandleOverflow(&deletedCasFiles);
|
|
u64 deletedCasCount = deletedCasFiles.size();
|
|
|
|
u64 totalCasSize = 0;
|
|
|
|
struct CasFileInfo { u64 size; u64 refCount; };
|
|
UnorderedMap<CasKey, CasFileInfo> existingCas;
|
|
ReaderWriterLock existingCasLock;
|
|
|
|
m_storage.WaitForActiveWork();
|
|
|
|
u32 removedNonExisting = 0;
|
|
{
|
|
// TODO: Make this cleaner...
|
|
SCOPED_WRITE_LOCK(m_storage.m_casLookupLock, lookupLock);
|
|
|
|
existingCas.reserve(m_storage.m_casLookup.size());
|
|
|
|
for (auto i=m_storage.m_casLookup.begin(), e=m_storage.m_casLookup.end(); i!=e;)
|
|
{
|
|
if (i->second.verified && !i->second.exists)
|
|
{
|
|
m_storage.DetachEntry(i->second);
|
|
++removedNonExisting;
|
|
i = m_storage.m_casLookup.erase(i);
|
|
e = m_storage.m_casLookup.end();
|
|
continue;
|
|
}
|
|
totalCasSize += i->second.size;
|
|
existingCas.try_emplace(i->first, CasFileInfo{i->second.size, 0ull});
|
|
++i;
|
|
}
|
|
}
|
|
|
|
if (removedNonExisting)
|
|
m_logger.Detail(TC(" Removed %u cas entries (marked as not existing)"), removedNonExisting);
|
|
|
|
m_logger.Detail(TC(" Found %llu cas files and %llu deleted by overflow (%s)"), existingCas.size(), deletedCasFiles.size(), BytesToText(totalCasSize).str);
|
|
u64 totalCasCount = existingCas.size() + deletedCasCount;
|
|
|
|
if (shouldExit())
|
|
return true;
|
|
|
|
u64 now = GetSystemTimeAsFileTime();
|
|
u64 oldest = 0;
|
|
u64 longestUnused = 0;
|
|
|
|
u64 lastUseTimeLimit = 0;
|
|
if (m_expirationTimeSeconds && GetFileTimeAsSeconds(now - m_creationTime) > m_expirationTimeSeconds)
|
|
lastUseTimeLimit = (now - m_creationTime) - GetSecondsAsFileTime(m_expirationTimeSeconds);
|
|
|
|
|
|
u32 workerCount = m_server.GetWorkerCount();
|
|
u32 workerCountToUse = workerCount > 0 ? workerCount - 1 : 0;
|
|
u32 workerCountToUseForBuckets = Min(workerCountToUse, u32(m_buckets.size()));
|
|
|
|
Atomic<u64> totalEntryCount;
|
|
Atomic<u64> deleteEntryCount;
|
|
Atomic<u64> expiredEntryCount;
|
|
Atomic<u64> overflowedEntryCount;
|
|
Atomic<u64> missingOutputEntryCount;
|
|
Atomic<u64> missingInputEntryCount;
|
|
|
|
Atomic<u64> activeDropCount;
|
|
auto dropCasGuard = MakeGuard([&]() { while (activeDropCount != 0) Sleep(1); });
|
|
|
|
u64 deleteCacheEntriesStartTime = GetTime();
|
|
do
|
|
{
|
|
bool checkInputsForDeletes = m_checkInputsForDeletedCas && !deletedCasFiles.empty();
|
|
|
|
oldest = 0;
|
|
longestUnused = 0;
|
|
totalEntryCount = 0;
|
|
|
|
m_server.ParallelFor(workerCountToUseForBuckets, m_buckets, [&](auto& it)
|
|
{
|
|
Bucket& bucket = it->second;
|
|
bucket.totalEntryCount = 0;
|
|
bucket.totalEntrySize = 0;
|
|
|
|
ReaderWriterLock keysToEraseLock;
|
|
Vector<CasKey> keysToErase;
|
|
|
|
m_server.ParallelFor(workerCountToUse, bucket.m_cacheEntryLookup, [&, touchedCas = Vector<u64*>()](auto& li) mutable
|
|
{
|
|
CacheEntries& entries = li->second;
|
|
|
|
// There is currently no idea saving more than 256kb worth of entries per lookup key (because that is what fetch max returns).. so let's wipe out
|
|
// all the entries that overflow that number
|
|
u64 capacityLeft = SendMaxSize - 32 - entries.GetSharedSize();
|
|
|
|
auto IsOffsetDeleted = [&](u64 offset) { CasKey casKey; bucket.m_casKeyTable.GetKey(casKey, offset); return deletedCasFiles.find(casKey) != deletedCasFiles.end(); };
|
|
|
|
// Check if any offset has been deleted in shared offsets..
|
|
bool offsetDeletedInShared = false;
|
|
auto& sharedOffsets = entries.sharedInputCasKeyOffsets;
|
|
if (checkInputsForDeletes)
|
|
{
|
|
BinaryReader reader2(sharedOffsets.data(), 0, sharedOffsets.size());
|
|
while (reader2.GetLeft())
|
|
{
|
|
if (!IsOffsetDeleted(reader2.Read7BitEncoded()))
|
|
continue;
|
|
offsetDeletedInShared = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
for (auto i=entries.entries.begin(), e=entries.entries.end(); i!=e;)
|
|
{
|
|
auto& entry = *i;
|
|
bool deleteEntry = false;
|
|
|
|
u64 neededSize = entries.GetEntrySize(entry, false);
|
|
if (neededSize > capacityLeft)
|
|
{
|
|
deleteEntry = true;
|
|
capacityLeft = 0;
|
|
++overflowedEntryCount;
|
|
}
|
|
|
|
if (!deleteEntry && entry.creationTime < lastUseTimeLimit && entry.lastUsedTime < lastUseTimeLimit)
|
|
{
|
|
deleteEntry = true;
|
|
++expiredEntryCount;
|
|
}
|
|
|
|
// This is an attempt at removing entries that has inputs that depends on other entries outputs.
|
|
// and that there is no point keeping them if the other entry is removed
|
|
// Example would be that there is no idea keeping entries that uses a pch if the entry producing the pch is gone
|
|
if (checkInputsForDeletes)
|
|
{
|
|
if (!deleteEntry && offsetDeletedInShared)
|
|
{
|
|
BinaryReader rangeReader(entry.sharedInputCasKeyOffsetRanges.data(), 0, entry.sharedInputCasKeyOffsetRanges.size());
|
|
while (!deleteEntry && rangeReader.GetLeft())
|
|
{
|
|
u64 begin = rangeReader.Read7BitEncoded();
|
|
u64 end = rangeReader.Read7BitEncoded();
|
|
BinaryReader inputReader(sharedOffsets.data() + begin, 0, end - begin);
|
|
while (inputReader.GetLeft())
|
|
{
|
|
if (!IsOffsetDeleted(inputReader.Read7BitEncoded()))
|
|
continue;
|
|
deleteEntry = true;
|
|
++missingInputEntryCount;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!deleteEntry)
|
|
{
|
|
auto& extraInputs = entry.extraInputCasKeyOffsets;
|
|
BinaryReader extraReader(extraInputs.data(), 0, extraInputs.size());
|
|
while (extraReader.GetLeft())
|
|
{
|
|
if (!IsOffsetDeleted(extraReader.Read7BitEncoded()))
|
|
continue;
|
|
deleteEntry = true;
|
|
++missingInputEntryCount;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!deleteEntry)
|
|
{
|
|
// Traverse outputs and check if cas files exists for each output, if not, delete entry.
|
|
touchedCas.clear();
|
|
|
|
auto& outputs = entry.outputCasKeyOffsets;
|
|
BinaryReader outputsReader(outputs.data(), 0, outputs.size());
|
|
while (outputsReader.GetLeft())
|
|
{
|
|
u64 offset = outputsReader.Read7BitEncoded();
|
|
CasKey casKey;
|
|
bucket.m_casKeyTable.GetKey(casKey, offset);
|
|
UBA_ASSERT(IsCompressed(casKey));
|
|
auto findIt = existingCas.find(casKey);
|
|
if (findIt != existingCas.end())
|
|
{
|
|
touchedCas.push_back(&findIt->second.refCount);
|
|
continue;
|
|
}
|
|
deleteEntry = true;
|
|
++missingOutputEntryCount;
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Remove entry from entries list and skip increasing ref count of cas files
|
|
if (deleteEntry)
|
|
{
|
|
if (i->id == entries.primaryId)
|
|
entries.primaryId = ~0u;
|
|
bucket.hasDeletedEntries = true;
|
|
++deleteEntryCount;
|
|
i = entries.entries.erase(i);
|
|
e = entries.entries.end();
|
|
continue;
|
|
}
|
|
|
|
++bucket.totalEntryCount;
|
|
|
|
capacityLeft -= neededSize;
|
|
|
|
SCOPED_WRITE_LOCK(existingCasLock, l);
|
|
if (!oldest || entry.creationTime < oldest)
|
|
oldest = entry.creationTime;
|
|
if (!longestUnused || entry.lastUsedTime < longestUnused)
|
|
longestUnused = entry.lastUsedTime;
|
|
|
|
for (u64* v : touchedCas)
|
|
++(*v);
|
|
l.Leave();
|
|
|
|
++i;
|
|
}
|
|
|
|
|
|
if (entries.entries.empty())
|
|
{
|
|
SCOPED_WRITE_LOCK(keysToEraseLock, lock2);
|
|
keysToErase.push_back(li->first);
|
|
}
|
|
else
|
|
bucket.totalEntrySize += entries.GetTotalSize(false);
|
|
});
|
|
|
|
for (auto& key : keysToErase)
|
|
bucket.m_cacheEntryLookup.erase(key);
|
|
|
|
totalEntryCount += bucket.totalEntryCount;
|
|
}, TC(""), true);
|
|
|
|
// Reset deleted cas files and update it again..
|
|
deletedCasFiles.clear();
|
|
|
|
for (auto i=existingCas.begin(), e=existingCas.end(); i!=e;)
|
|
{
|
|
if (i->second.refCount != 0)
|
|
{
|
|
i->second.refCount = 0;
|
|
++i;
|
|
continue;
|
|
}
|
|
deletedCasFiles.insert(i->first);
|
|
++deletedCasCount;
|
|
totalCasSize -= i->second.size;
|
|
i = existingCas.erase(i);
|
|
e = existingCas.end();
|
|
}
|
|
|
|
// Add drop cas as work so it can run in the background
|
|
for (auto& casKey : deletedCasFiles)
|
|
{
|
|
++activeDropCount;
|
|
m_server.AddWork([&, key = casKey]() { m_storage.DropCasFile(key, true, TC("")); --activeDropCount; }, 1, TC(""));
|
|
}
|
|
}
|
|
while (!deletedCasFiles.empty()); // if cas files are deleted we need to do another loop and check cache entry inputs to see if files were inputs
|
|
|
|
if (overflowedEntryCount)
|
|
m_logger.Detail(TC(" Found %llu overflowed cache entries"), overflowedEntryCount.load());
|
|
if (expiredEntryCount)
|
|
m_logger.Detail(TC(" Found %llu expired cache entries (older than %s)"), expiredEntryCount.load(), TimeToText(MsToTime(m_expirationTimeSeconds*1000), true).str);
|
|
if (missingOutputEntryCount)
|
|
m_logger.Detail(TC(" Found %llu cache entries with missing output cas"), missingOutputEntryCount.load());
|
|
if (missingInputEntryCount)
|
|
m_logger.Detail(TC(" Found %llu cache entries with missing input cas"), missingInputEntryCount.load());
|
|
|
|
m_logger.Detail(TC(" Deleted %llu cas files and %llu cache entries (%s)"), deletedCasCount, deleteEntryCount.load(), TimeToText(GetTime() - deleteCacheEntriesStartTime).str);
|
|
|
|
if (shouldExit())
|
|
return true;
|
|
|
|
m_server.ParallelFor(workerCountToUseForBuckets, m_buckets, [&](auto& it)
|
|
{
|
|
u64 bucketStartTime = GetTime();
|
|
|
|
Bucket& bucket = it->second;
|
|
|
|
if (!bucket.hasDeletedEntries && !forceAllSteps)
|
|
{
|
|
m_logger.Detail(TC(" Bucket %u skipped updating. (%llu entries)"), bucket.index, bucket.totalEntryCount.load());
|
|
return;
|
|
}
|
|
bucket.hasDeletedEntries = false;
|
|
|
|
MemoryBlock memoryBlock(128*1024*1024);
|
|
|
|
GrowingNoLockUnorderedSet<u32> usedCasKeyOffsets(&memoryBlock);
|
|
usedCasKeyOffsets.reserve(bucket.m_casKeyTable.GetKeyCount());
|
|
|
|
u64 collectUsedCasKeysStart = GetTime();
|
|
|
|
// Collect all caskeys that are used by cache entries.
|
|
for (auto& kv2 : bucket.m_cacheEntryLookup)
|
|
{
|
|
auto collectUsedCasKeyOffsets = [&](const Vector<u8>& offsets)
|
|
{
|
|
BinaryReader reader2(offsets.data(), 0, offsets.size());
|
|
while (reader2.GetLeft())
|
|
usedCasKeyOffsets.insert(u32(reader2.Read7BitEncoded()));
|
|
};
|
|
|
|
collectUsedCasKeyOffsets(kv2.second.sharedInputCasKeyOffsets);
|
|
for (auto& entry : kv2.second.entries)
|
|
{
|
|
collectUsedCasKeyOffsets(entry.extraInputCasKeyOffsets);
|
|
collectUsedCasKeyOffsets(entry.outputCasKeyOffsets);
|
|
}
|
|
}
|
|
m_logger.Detail(TC(" Bucket %u Collected %llu used caskeys. (%s)"), bucket.index, usedCasKeyOffsets.size(), TimeToText(GetTime() - collectUsedCasKeysStart).str);
|
|
|
|
u64 recreatePathTableStart = GetTime();
|
|
|
|
// Traverse all caskeys in caskey table and figure out which ones we can delete
|
|
GrowingNoLockUnorderedSet<u32> usedPathOffsets(&memoryBlock);
|
|
usedPathOffsets.reserve(usedCasKeyOffsets.size());
|
|
|
|
for (u32 casKeyOffset : usedCasKeyOffsets)
|
|
{
|
|
BinaryReader reader2(bucket.m_casKeyTable.GetMemory(), casKeyOffset, bucket.m_casKeyTable.GetSize());
|
|
u32 pathOffset = u32(reader2.Read7BitEncoded());
|
|
usedPathOffsets.insert(pathOffset);
|
|
}
|
|
|
|
// Build new path table based on used offsets
|
|
GrowingNoLockUnorderedMap<u32, u32> oldToNewPathOffset(&memoryBlock);
|
|
u32 oldSize = bucket.m_pathTable.GetSize();
|
|
{
|
|
CompactPathTable newPathTable(CachePathTableMaxSize, CompactPathTable::V1, bucket.m_pathTable.GetPathCount(), bucket.m_pathTable.GetSegmentCount());
|
|
oldToNewPathOffset.reserve(usedPathOffsets.size());
|
|
|
|
for (u32 pathOffset : usedPathOffsets)
|
|
{
|
|
StringBuffer<> temp;
|
|
bucket.m_pathTable.GetString(temp, pathOffset);
|
|
u32 newOffset = newPathTable.AddNoLock(temp.data, temp.count);
|
|
|
|
#if 0
|
|
StringBuffer<> test;
|
|
newPathTable.GetString(test, newOffset);
|
|
UBA_ASSERT(test.Equals(temp.data));
|
|
#endif
|
|
|
|
auto res = oldToNewPathOffset.try_emplace(pathOffset, newOffset);
|
|
UBA_ASSERT(res.second);(void)res;
|
|
}
|
|
bucket.m_pathTable.Swap(newPathTable);
|
|
}
|
|
m_logger.Detail(TC(" Bucket %u Recreated path table. %s -> %s (%s)"), bucket.index, BytesToText(oldSize).str, BytesToText(bucket.m_pathTable.GetSize()).str, TimeToText(GetTime() - recreatePathTableStart).str);
|
|
|
|
|
|
// Build new caskey table based on used offsets
|
|
u64 recreateCasKeyTableStart = GetTime();
|
|
GrowingNoLockUnorderedMap<u32, u32> oldToNewCasKeyOffset(&memoryBlock);
|
|
oldSize = bucket.m_casKeyTable.GetSize();
|
|
{
|
|
oldToNewCasKeyOffset.reserve(usedCasKeyOffsets.size());
|
|
CompactCasKeyTable newCasKeyTable(CacheCasKeyTableMaxSize, usedCasKeyOffsets.size());
|
|
for (u32 casKeyOffset : usedCasKeyOffsets)
|
|
{
|
|
BinaryReader reader2(bucket.m_casKeyTable.GetMemory(), casKeyOffset, bucket.m_casKeyTable.GetSize());
|
|
u32 oldPathOffset = u32(reader2.Read7BitEncoded());
|
|
CasKey casKey = reader2.ReadCasKey();
|
|
auto findIt = oldToNewPathOffset.find(oldPathOffset);
|
|
UBA_ASSERT(findIt != oldToNewPathOffset.end());
|
|
u32 newCasKeyOffset = newCasKeyTable.Add(casKey, findIt->second);
|
|
if (casKeyOffset == newCasKeyOffset)
|
|
continue;
|
|
auto res = oldToNewCasKeyOffset.try_emplace(casKeyOffset, newCasKeyOffset);
|
|
UBA_ASSERT(res.second);(void)res;
|
|
}
|
|
bucket.m_casKeyTable.Swap(newCasKeyTable);
|
|
}
|
|
m_logger.Detail(TC(" Bucket %u Recreated caskey table. %s -> %s (%s)"), bucket.index, BytesToText(oldSize).str, BytesToText(bucket.m_casKeyTable.GetSize()).str, TimeToText(GetTime() - recreateCasKeyTableStart).str);
|
|
|
|
|
|
if (!oldToNewCasKeyOffset.empty())
|
|
{
|
|
// Update all casKeyOffsets
|
|
u64 updateEntriesStart = GetTime();
|
|
|
|
m_server.ParallelFor(workerCountToUse, bucket.m_cacheEntryLookup, [&, temp = Vector<u32>(), temp2 = Vector<u8>()](auto& it) mutable
|
|
{
|
|
it->second.UpdateEntries(m_logger, oldToNewCasKeyOffset, temp, temp2);
|
|
});
|
|
|
|
#if 0
|
|
u8* mem = bucket.m_pathTable.GetMemory();
|
|
u64 memLeft = bucket.m_pathTable.GetSize();
|
|
while (memLeft)
|
|
{
|
|
u8 buffer[256*1024];
|
|
auto compressor = OodleLZ_Compressor_Kraken;
|
|
auto compressionLevel = OodleLZ_CompressionLevel_SuperFast;
|
|
u64 toCompress = Min(memLeft, u64(256*1024 - 128));
|
|
auto compressedBlockSize = OodleLZ_Compress(compressor, mem, (OO_SINTa)toCompress, buffer, compressionLevel);
|
|
(void)compressedBlockSize;
|
|
memLeft -= toCompress;
|
|
}
|
|
#endif
|
|
|
|
|
|
m_logger.Detail(TC(" Bucket %u Updated cache entries with new tables (%s)"), bucket.index, TimeToText(GetTime() - updateEntriesStart).str);
|
|
}
|
|
|
|
bucket.needsSave = true;
|
|
|
|
m_logger.Info(TC(" Bucket %u Done (%s). CacheEntries: %llu (%s) PathTable: %s CasTable: %s"), bucket.index, TimeToText(GetTime() - bucketStartTime).str, bucket.totalEntryCount.load(), BytesToText(bucket.totalEntrySize.load()).str, BytesToText(bucket.m_pathTable.GetSize()).str, BytesToText(bucket.m_casKeyTable.GetSize()).str);
|
|
}, TC(""), true);
|
|
|
|
// Need to make sure all cas entries are dropped before saving cas table
|
|
u64 dropStartTime = GetTime();
|
|
dropCasGuard.Execute();
|
|
u64 dropCasDuration = GetTime() - dropStartTime;
|
|
if (TimeToMs(dropCasDuration) > 10)
|
|
m_logger.Detail(TC(" Done deleting cas files (%s)"), TimeToText(dropCasDuration).str);
|
|
|
|
if (entriesAdded || deletedCasCount || deleteEntryCount || forceAllSteps)
|
|
{
|
|
u64 saveStart = GetTime();
|
|
m_logger.Detail(TC(" Saving to disk"));
|
|
Event saveCasEvent(true);
|
|
m_server.AddWork([&]() { m_storage.SaveCasTable(false, false); saveCasEvent.Set(); }, 1, TC("SaveCas"));
|
|
SaveNoLock();
|
|
saveCasEvent.IsSet();
|
|
m_logger.Detail(TC(" Save Done (%s)"), TimeToText(GetTime() - saveStart).str);
|
|
}
|
|
|
|
u64 oldestTime = oldest ? GetFileTimeAsTime(now - (m_creationTime + oldest)) : 0;
|
|
u64 longestUnusedTime = longestUnused ? GetFileTimeAsTime(now - (m_creationTime + longestUnused)) : 0;
|
|
u64 duration = GetTime() - startTime;
|
|
m_logger.Info(TC("Maintenance done! (%s) CasFiles: %llu (%s) Entries: %llu Oldest: %s LongestUnused: %s"), TimeToText(duration).str, totalCasCount - deletedCasCount, BytesToText(totalCasSize).str, totalEntryCount.load(), TimeToText(oldestTime, true).str, TimeToText(longestUnusedTime, true).str);
|
|
|
|
m_longestMaintenance = Max(m_longestMaintenance, duration);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::ShouldShutdown()
|
|
{
|
|
if (!m_shutdownRequested)
|
|
return false;
|
|
SCOPED_READ_LOCK(m_connectionsLock, lock2);
|
|
if (!m_connections.empty() || m_addsSinceMaintenance)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void CacheServer::OnDisconnected(u32 clientId)
|
|
{
|
|
SCOPED_WRITE_LOCK(m_connectionsLock, lock);
|
|
m_connections.erase(clientId);
|
|
lock.Leave();
|
|
}
|
|
|
|
CacheServer::ConnectionBucket& CacheServer::GetConnectionBucket(const ConnectionInfo& connectionInfo, BinaryReader& reader)
|
|
{
|
|
u64 id = reader.Read7BitEncoded();
|
|
SCOPED_WRITE_LOCK(m_connectionsLock, lock);
|
|
auto& connection = m_connections[connectionInfo.GetId()];
|
|
return connection.buckets.try_emplace(id, id).first->second;
|
|
}
|
|
|
|
CacheServer::Bucket& CacheServer::GetBucket(BinaryReader& reader)
|
|
{
|
|
u64 id = reader.Read7BitEncoded();
|
|
return GetBucket(id);
|
|
|
|
}
|
|
|
|
CacheServer::Bucket& CacheServer::GetBucket(u64 id)
|
|
{
|
|
SCOPED_WRITE_LOCK(m_bucketsLock, bucketsLock);
|
|
auto insres = m_buckets.try_emplace(id, id);
|
|
if (insres.second)
|
|
insres.first->second.index = u32(m_buckets.size() - 1);
|
|
return insres.first->second;
|
|
}
|
|
|
|
u32 CacheServer::GetBucketWorkerCount()
|
|
{
|
|
u32 workerCount = m_server.GetWorkerCount();
|
|
u32 workerCountToUse = workerCount > 0 ? workerCount - 1 : 0;
|
|
return Min(workerCountToUse, u32(m_buckets.size()));
|
|
}
|
|
|
|
bool CacheServer::HandleMessage(const ConnectionInfo& connectionInfo, u8 messageType, BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
if (messageType != CacheMessageType_Connect && m_isRunningMaintenance)
|
|
return m_logger.Error(TC("Can't handle network message %s while running maintenance mode"), ToString(CacheMessageType(messageType)));
|
|
|
|
switch (messageType)
|
|
{
|
|
case CacheMessageType_Connect:
|
|
{
|
|
u32 clientVersion = reader.ReadU32();
|
|
if (clientVersion < 3 || clientVersion > CacheNetworkVersion)
|
|
return m_logger.Error(TC("Different network versions. Client: %u, Server: %u. Disconnecting"), clientVersion, CacheNetworkVersion);
|
|
|
|
SCOPED_WRITE_LOCK(m_connectionsLock, lock);
|
|
if (m_isRunningMaintenance)
|
|
{
|
|
writer.WriteBool(false);
|
|
writer.WriteString(TC("Running maintenance..."));
|
|
}
|
|
|
|
writer.WriteBool(true);
|
|
auto insres = m_connections.try_emplace(connectionInfo.GetId());
|
|
auto& connection = insres.first->second;
|
|
connection.clientVersion = clientVersion;
|
|
return true;
|
|
}
|
|
case CacheMessageType_StorePathTable:
|
|
{
|
|
GetConnectionBucket(connectionInfo, reader).pathTable.ReadMem(reader, false);
|
|
return true;
|
|
}
|
|
case CacheMessageType_StoreCasTable:
|
|
{
|
|
GetConnectionBucket(connectionInfo, reader).casKeyTable.ReadMem(reader, false);
|
|
return true;
|
|
}
|
|
case CacheMessageType_StoreEntry:
|
|
{
|
|
auto& bucket = GetConnectionBucket(connectionInfo, reader);
|
|
return HandleStoreEntry(bucket, reader, writer);
|
|
}
|
|
case CacheMessageType_StoreEntryDone:
|
|
{
|
|
auto& connectionBucket = GetConnectionBucket(connectionInfo, reader);
|
|
CasKey cmdKey = reader.ReadCasKey();
|
|
|
|
bool success = true;
|
|
if (reader.GetLeft())
|
|
success = reader.ReadBool();
|
|
|
|
SCOPED_WRITE_LOCK(connectionBucket.cacheEntryLookupLock, lock2);
|
|
auto findIt = connectionBucket.cacheEntryLookup.find(cmdKey);
|
|
if (findIt == connectionBucket.cacheEntryLookup.end())
|
|
return true;
|
|
auto g = MakeGuard([&]() { connectionBucket.cacheEntryLookup.erase(findIt); });
|
|
if (!success)
|
|
return true;
|
|
|
|
u64 id = connectionBucket.id;
|
|
Bucket& bucket = GetBucket(id);
|
|
|
|
SCOPED_WRITE_LOCK(bucket.m_cacheEntryLookupLock, lock3);
|
|
auto insres = bucket.m_cacheEntryLookup.try_emplace(cmdKey);
|
|
auto& cacheEntries = insres.first->second;
|
|
lock3.Leave();
|
|
|
|
SCOPED_WRITE_LOCK(cacheEntries.lock, lock4);
|
|
cacheEntries.entries.emplace_front(std::move(findIt->second));
|
|
return true;
|
|
}
|
|
case CacheMessageType_FetchPathTable:
|
|
return HandleFetchPathTable(reader, writer);
|
|
|
|
case CacheMessageType_FetchCasTable:
|
|
return HandleFetchCasTable(reader, writer);
|
|
|
|
case CacheMessageType_FetchEntries:
|
|
{
|
|
SCOPED_READ_LOCK(m_connectionsLock, lock);
|
|
u32 clientVersion = m_connections[connectionInfo.GetId()].clientVersion;
|
|
lock.Leave();
|
|
return HandleFetchEntries(reader, writer, clientVersion);
|
|
}
|
|
case CacheMessageType_ExecuteCommand:
|
|
return HandleExecuteCommand(reader, writer);
|
|
|
|
case CacheMessageType_ReportUsedEntry:
|
|
return HandleReportUsedEntry(reader, writer);
|
|
|
|
case CacheMessageType_RequestShutdown:
|
|
{
|
|
TString reason = reader.ReadString();
|
|
m_logger.Info(TC("Shutdown requested. Reason: %s"), reason.empty() ? TC("Unknown") : reason.c_str());
|
|
m_shutdownRequested = true;
|
|
writer.WriteBool(true);
|
|
return true;
|
|
}
|
|
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool CacheServer::HandleStoreEntry(ConnectionBucket& connectionBucket, BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
CasKey cmdKey = reader.ReadCasKey();
|
|
|
|
u64 outputCount = reader.Read7BitEncoded();
|
|
u64 index = 0;
|
|
|
|
Set<u32> inputs;
|
|
u64 bytesForInput = 0;
|
|
|
|
u64 outputStartOffset = reader.GetPosition();
|
|
u64 id = connectionBucket.id;
|
|
Bucket& bucket = GetBucket(id);
|
|
|
|
while (reader.GetLeft())
|
|
{
|
|
u32 offset = u32(reader.Read7BitEncoded());
|
|
bool isInput = index++ >= outputCount;
|
|
if (!isInput)
|
|
continue;
|
|
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
connectionBucket.casKeyTable.GetPathAndKey(path, casKey, connectionBucket.pathTable, offset);
|
|
UBA_ASSERT(path.count);
|
|
|
|
u32 pathOffset = bucket.m_pathTable.Add(path.data, path.count);
|
|
|
|
#if 0
|
|
StringBuffer<> test;
|
|
bucket.m_pathTable.GetString(test, pathOffset);
|
|
UBA_ASSERT(test.Equals(path.data));
|
|
#endif
|
|
|
|
u32 casKeyOffset = bucket.m_casKeyTable.Add(casKey, pathOffset);
|
|
auto insres = inputs.insert(casKeyOffset);
|
|
if (!insres.second)
|
|
{
|
|
m_logger.Warning(TC("Input file %s exists more than once in cache entry"), path.data);
|
|
continue;
|
|
}
|
|
bytesForInput += Get7BitEncodedCount(casKeyOffset);
|
|
|
|
//m_logger.Info(TC("%s - %s"), path.data, CasKeyString(casKey).str);
|
|
}
|
|
|
|
Vector<u8> inputCasKeyOffsets;
|
|
{
|
|
inputCasKeyOffsets.resize(bytesForInput);
|
|
BinaryWriter w2(inputCasKeyOffsets.data(), 0, inputCasKeyOffsets.size());
|
|
for (u32 input : inputs)
|
|
w2.Write7BitEncoded(input);
|
|
}
|
|
|
|
SCOPED_WRITE_LOCK(bucket.m_cacheEntryLookupLock, lock);
|
|
auto insres = bucket.m_cacheEntryLookup.try_emplace(cmdKey);
|
|
auto& cacheEntries = insres.first->second;
|
|
lock.Leave();
|
|
|
|
SCOPED_WRITE_LOCK(cacheEntries.lock, lock2);
|
|
|
|
|
|
// Create entry based on existing entry
|
|
CacheEntry newEntry;
|
|
cacheEntries.BuildInputs(newEntry, inputs);
|
|
|
|
List<CacheEntry>::iterator matchingEntry = cacheEntries.entries.end();
|
|
for (auto i=cacheEntries.entries.begin(), e=cacheEntries.entries.end(); i!=e; ++i)
|
|
{
|
|
if (i->sharedInputCasKeyOffsetRanges != newEntry.sharedInputCasKeyOffsetRanges || i->extraInputCasKeyOffsets != newEntry.extraInputCasKeyOffsets)
|
|
continue;
|
|
matchingEntry = i;
|
|
break;
|
|
}
|
|
|
|
#if UBA_USE_OLD
|
|
List<CacheEntry>::iterator matchingEntry2 = cacheEntries.entries.end();
|
|
for (auto i=cacheEntries.entries.begin(), e=cacheEntries.entries.end(); i!=e; ++i)
|
|
{
|
|
if (i->inputCasKeyOffsets != inputCasKeyOffsets)
|
|
continue;
|
|
matchingEntry2 = i;
|
|
break;
|
|
}
|
|
if (matchingEntry2 != matchingEntry)
|
|
m_logger.Warning(L"CODE MISMATCH!!!");
|
|
#endif
|
|
|
|
|
|
// Already exists
|
|
if (matchingEntry != cacheEntries.entries.end())
|
|
{
|
|
bool shouldOverwrite = false;
|
|
Map<TString, CasKey> existing;
|
|
|
|
BinaryReader r2(matchingEntry->outputCasKeyOffsets.data(), 0, matchingEntry->outputCasKeyOffsets.size());
|
|
while (r2.GetLeft())
|
|
{
|
|
u32 existingOffset = u32(r2.Read7BitEncoded());
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
bucket.m_casKeyTable.GetPathAndKey(path, casKey, bucket.m_pathTable, existingOffset);
|
|
existing.try_emplace(path.data, casKey);
|
|
}
|
|
|
|
reader.SetPosition(outputStartOffset);
|
|
u64 left = outputCount;
|
|
while (left--)
|
|
{
|
|
u32 outputOffset = u32(reader.Read7BitEncoded());
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
connectionBucket.casKeyTable.GetPathAndKey(path, casKey, connectionBucket.pathTable, outputOffset);
|
|
|
|
auto findIt = existing.find(path.data);
|
|
if (findIt == existing.end())
|
|
{
|
|
m_logger.Warning(TC("Existing cache entry matches input but does not match output (output file %s did not exist in existing cache entry)"), path.data);
|
|
cacheEntries.entries.erase(matchingEntry);
|
|
shouldOverwrite = true;
|
|
break;
|
|
}
|
|
if (findIt->second != casKey)
|
|
{
|
|
//m_logger.Warning(TC("Existing cache entry matches input but does not match output (%s has different caskey)"), path.data);
|
|
cacheEntries.entries.erase(matchingEntry);
|
|
shouldOverwrite = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!shouldOverwrite)
|
|
return true;
|
|
}
|
|
|
|
#if UBA_USE_OLD
|
|
// Add new entry
|
|
newEntry.inputCasKeyOffsets.swap(inputCasKeyOffsets);
|
|
cacheEntries.ValidateEntry(m_logger, newEntry);
|
|
#endif
|
|
|
|
Set<u32> outputs;
|
|
u64 bytesForOutput = 0;
|
|
|
|
bool hasAllContent = true;
|
|
reader.SetPosition(outputStartOffset);
|
|
u64 left = outputCount;
|
|
while (left--)
|
|
{
|
|
u32 outputOffset = u32(reader.Read7BitEncoded());
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
connectionBucket.casKeyTable.GetPathAndKey(path, casKey, connectionBucket.pathTable, outputOffset);
|
|
u32 pathOffset = bucket.m_pathTable.Add(path.data, path.count);
|
|
|
|
#if 0
|
|
StringBuffer<> test;
|
|
bucket.m_pathTable.GetString(test, pathOffset);
|
|
UBA_ASSERT(test.Equals(path.data));
|
|
#endif
|
|
|
|
u32 casKeyOffset = bucket.m_casKeyTable.Add(casKey, pathOffset);
|
|
outputs.insert(casKeyOffset);
|
|
bytesForOutput += Get7BitEncodedCount(casKeyOffset);
|
|
|
|
if (!m_storage.EnsureCasFile(casKey, nullptr))
|
|
{
|
|
writer.Write7BitEncoded(outputOffset);
|
|
hasAllContent = false;
|
|
}
|
|
}
|
|
|
|
newEntry.outputCasKeyOffsets.resize(bytesForOutput);
|
|
BinaryWriter w2(newEntry.outputCasKeyOffsets.data(), 0, newEntry.outputCasKeyOffsets.size());
|
|
for (u32 output : outputs)
|
|
w2.Write7BitEncoded(output);
|
|
|
|
|
|
newEntry.creationTime = GetSystemTimeAsFileTime() - m_creationTime;
|
|
newEntry.id = cacheEntries.idCounter++;
|
|
|
|
// If cache server has all content we can put the new cache entry directly in the lookup.. otherwise we'll have to wait until client has uploaded content
|
|
if (hasAllContent)
|
|
{
|
|
cacheEntries.entries.emplace_front(std::move(newEntry));
|
|
}
|
|
else
|
|
{
|
|
SCOPED_WRITE_LOCK(connectionBucket.cacheEntryLookupLock, lock3);
|
|
bool res = connectionBucket.cacheEntryLookup.try_emplace(cmdKey, std::move(newEntry)).second;
|
|
UBA_ASSERT(res);(void)res;
|
|
}
|
|
|
|
//m_logger.Info(TC("Added new cache entry (%u inputs and %u outputs)"), u32(inputs.size()), outputCount);
|
|
bucket.needsSave = true;
|
|
|
|
++m_addsSinceMaintenance;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::HandleFetchPathTable(BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
Bucket& bucket = GetBucket(reader);
|
|
u32 haveSize = reader.ReadU32();
|
|
u32 size = bucket.m_pathTable.GetSize();
|
|
writer.WriteU32(size);
|
|
u32 toSend = Min(u32(writer.GetCapacityLeft()), size - haveSize);
|
|
writer.WriteBytes(bucket.m_pathTable.GetMemory() + haveSize, toSend);
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::HandleFetchCasTable(BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
Bucket& bucket = GetBucket(reader);
|
|
u32 haveSize = reader.ReadU32();
|
|
u32 size = bucket.m_casKeyTable.GetSize();
|
|
writer.WriteU32(size);
|
|
u32 toSend = Min(u32(writer.GetCapacityLeft()), size - haveSize);
|
|
writer.WriteBytes(bucket.m_casKeyTable.GetMemory() + haveSize, toSend);
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::HandleFetchEntries(BinaryReader& reader, BinaryWriter& writer, u32 clientVersion)
|
|
{
|
|
Bucket& bucket = GetBucket(reader);
|
|
CasKey cmdKey = reader.ReadCasKey();
|
|
|
|
SCOPED_READ_LOCK(bucket.m_cacheEntryLookupLock, lock);
|
|
auto findIt = bucket.m_cacheEntryLookup.find(cmdKey);
|
|
if (findIt == bucket.m_cacheEntryLookup.end())
|
|
{
|
|
writer.WriteU16(0);
|
|
return true;
|
|
}
|
|
auto& cacheEntries = findIt->second;
|
|
lock.Leave();
|
|
|
|
SCOPED_READ_LOCK(cacheEntries.lock, lock2);
|
|
return cacheEntries.Write(writer, clientVersion, false);
|
|
}
|
|
|
|
bool CacheServer::HandleReportUsedEntry(BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
Bucket& bucket = GetBucket(reader);
|
|
CasKey cmdKey = reader.ReadCasKey();
|
|
u64 entryId = reader.Read7BitEncoded();
|
|
|
|
SCOPED_READ_LOCK(bucket.m_cacheEntryLookupLock, lock);
|
|
auto findIt = bucket.m_cacheEntryLookup.find(cmdKey);
|
|
if (findIt == bucket.m_cacheEntryLookup.end())
|
|
return true;
|
|
auto& cacheEntries = findIt->second;
|
|
lock.Leave();
|
|
|
|
SCOPED_WRITE_LOCK(cacheEntries.lock, lock2);
|
|
for (auto& entry : cacheEntries.entries)
|
|
{
|
|
if (entryId != entry.id)
|
|
continue;
|
|
u64 fileTime = GetSystemTimeAsFileTime() - m_creationTime;
|
|
entry.lastUsedTime = fileTime;
|
|
bucket.lastUsedTime = fileTime;
|
|
break;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool CacheServer::HandleExecuteCommand(BinaryReader& reader, BinaryWriter& writer)
|
|
{
|
|
StringBuffer<> command;
|
|
reader.ReadString(command);
|
|
|
|
StringBuffer<> additionalInfo;
|
|
reader.ReadString(additionalInfo);
|
|
|
|
StringBuffer<> tempFile(m_storage.GetTempPath());
|
|
Guid guid;
|
|
CreateGuid(guid);
|
|
tempFile.Append(GuidToString(guid).str);
|
|
|
|
FileAccessor file(m_logger, tempFile.data);
|
|
if (!file.CreateWrite())
|
|
return false;
|
|
|
|
bool writeSuccess = true;
|
|
auto Write = [&](const void* data, u64 size) { writeSuccess &= file.Write(data, size); };
|
|
|
|
u8 bom[] = {0xEF,0xBB,0xBF};
|
|
Write(bom, sizeof(bom));
|
|
|
|
auto writeLine = [&](const tchar* text)
|
|
{
|
|
u8 buffer[1024];
|
|
BinaryWriter w(buffer, 0, sizeof(buffer));
|
|
w.WriteUtf8String(text, TStrlen(text));
|
|
w.WriteUtf8String(TC("\n"), 1);
|
|
Write(buffer, w.GetPosition());
|
|
};
|
|
|
|
StringBuffer<> line;
|
|
|
|
if (command.Equals(TC("content")))
|
|
{
|
|
writeLine(TC("UbaCache server summary"));
|
|
|
|
StringBufferBase& filterString = additionalInfo;
|
|
|
|
u64 now = GetSystemTimeAsFileTime();
|
|
|
|
Vector<u8> temp;
|
|
|
|
SCOPED_READ_LOCK(m_bucketsLock, bucketsLock);
|
|
for (auto& kv : m_buckets)
|
|
{
|
|
Bucket& bucket = kv.second;
|
|
SCOPED_READ_LOCK(bucket.m_cacheEntryLookupLock, lock2);
|
|
|
|
for (auto& kv2 : bucket.m_cacheEntryLookup)
|
|
{
|
|
CacheEntries& entries = kv2.second;
|
|
SCOPED_READ_LOCK(entries.lock, lock3);
|
|
|
|
Set<u32> visibleIndices;
|
|
if (filterString.count)
|
|
{
|
|
u32 index = 0;
|
|
auto findString = [&](const Vector<u8>& offsets)
|
|
{
|
|
BinaryReader reader2(offsets.data(), 0, offsets.size());
|
|
while (reader2.GetLeft())
|
|
{
|
|
u64 offset = reader2.Read7BitEncoded();
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
bucket.m_casKeyTable.GetPathAndKey(path, casKey, bucket.m_pathTable, offset);
|
|
if (path.Contains(filterString.data))
|
|
return true;
|
|
if (Contains(CasKeyString(casKey).str, filterString.data))
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
|
|
for (auto& entry : entries.entries)
|
|
{
|
|
entries.Flatten(temp, entry);
|
|
if (findString(temp) || findString(entry.outputCasKeyOffsets))
|
|
visibleIndices.insert(index);
|
|
++index;
|
|
}
|
|
if (visibleIndices.empty())
|
|
continue;
|
|
}
|
|
|
|
|
|
writeLine(CasKeyString(kv2.first).str);
|
|
u32 index = 0;
|
|
for (auto& entry : entries.entries)
|
|
{
|
|
if (!visibleIndices.empty() && visibleIndices.find(index) == visibleIndices.end())
|
|
{
|
|
++index;
|
|
continue;
|
|
}
|
|
|
|
u64 age = GetFileTimeAsTime(now - entry.creationTime);
|
|
writeLine(line.Clear().Appendf(TC(" #%u (%s ago)"), index, TimeToText(age, true).str).data);
|
|
|
|
auto writeOffsets = [&](const Vector<u8>& offsets)
|
|
{
|
|
BinaryReader reader2(offsets.data(), 0, offsets.size());
|
|
while (reader2.GetLeft())
|
|
{
|
|
u64 offset = reader2.Read7BitEncoded();
|
|
CasKey casKey;
|
|
StringBuffer<> path;
|
|
bucket.m_casKeyTable.GetPathAndKey(path, casKey, bucket.m_pathTable, offset);
|
|
writeLine(line.Clear().Appendf(TC(" %s - %s"), path.data, CasKeyString(casKey).str).data);
|
|
}
|
|
};
|
|
|
|
writeLine(line.Clear().Append(TC(" Inputs:")).data);
|
|
entries.Flatten(temp, entry);
|
|
writeOffsets(temp);
|
|
writeLine(line.Clear().Append(TC(" Outputs:")).data);
|
|
writeOffsets(entry.outputCasKeyOffsets);
|
|
++index;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else if (command.Equals(TC("status")))
|
|
{
|
|
writeLine(TC("UbaCacheServer status"));
|
|
writeLine(line.Clear().Appendf(TC(" CreationTime: %s ago"), TimeToText(GetFileTimeAsTime(GetSystemTimeAsFileTime() - m_creationTime), true).str).data);
|
|
writeLine(line.Clear().Appendf(TC(" UpTime: %s"), TimeToText(GetTime() - m_startTime, true).str).data);
|
|
writeLine(line.Clear().Appendf(TC(" Longest maintenance: %s"), TimeToText(m_longestMaintenance).str).data);
|
|
writeLine(line.Clear().Appendf(TC(" Buckets:")).data);
|
|
u32 index = 0;
|
|
|
|
{
|
|
SCOPED_READ_LOCK(m_bucketsLock, bucketsLock);
|
|
for (auto& kv : m_buckets)
|
|
{
|
|
Bucket& bucket = kv.second;
|
|
SCOPED_READ_LOCK(bucket.m_cacheEntryLookupLock, lock2);
|
|
u64 mostEntries = 0;
|
|
u64 lastUsed = 0;
|
|
u64 totalEntryCount = 0;
|
|
for (auto& kv2 : bucket.m_cacheEntryLookup)
|
|
{
|
|
CacheEntries& entries = kv2.second;
|
|
SCOPED_READ_LOCK(entries.lock, lock3);
|
|
mostEntries = Max(mostEntries, u64(entries.entries.size()));
|
|
for (auto& entry : entries.entries)
|
|
lastUsed = Max(lastUsed, entry.lastUsedTime);
|
|
totalEntryCount += entries.entries.size();
|
|
}
|
|
lock2.Leave();
|
|
u64 lastUsedTime = 0;
|
|
if (lastUsed)
|
|
lastUsedTime = GetFileTimeAsTime(GetSystemTimeAsFileTime() - (m_creationTime + lastUsed));
|
|
|
|
writeLine(line.Clear().Appendf(TC(" #%u - %llu"), index++, kv.first).data);
|
|
writeLine(line.Clear().Appendf(TC(" PathTable: %lls (%s)"), bucket.m_pathTable.GetPathCount(), BytesToText(bucket.m_pathTable.GetSize()).str).data);
|
|
writeLine(line.Clear().Appendf(TC(" CasKeyTable: %llu (%s)"), bucket.m_cacheEntryLookup.size(), BytesToText(bucket.m_casKeyTable.GetSize()).str).data);
|
|
writeLine(line.Clear().Appendf(TC(" TotalEntries: %llu"), totalEntryCount).data);
|
|
writeLine(line.Clear().Appendf(TC(" KeyMostEntries: %llu"), mostEntries).data);
|
|
writeLine(line.Clear().Appendf(TC(" LastEntryUsed: %s ago"), TimeToText(lastUsedTime, true).str).data);
|
|
}
|
|
}
|
|
u64 totalCasSize = 0;
|
|
u64 totalCasCount = 0;
|
|
m_storage.TraverseAllCasFiles([&](const CasKey& casKey, u64 size) { ++totalCasCount; totalCasSize += size; });
|
|
writeLine(line.Clear().Appendf(TC(" CasDb:")).data);
|
|
writeLine(line.Clear().Appendf(TC(" Count: %llu"), totalCasCount).data);
|
|
writeLine(line.Clear().Appendf(TC(" Size: %s"), BytesToText(totalCasSize).str).data);
|
|
}
|
|
else if (command.Equals(TC("obliterate")))
|
|
{
|
|
m_shouldWipe = true;
|
|
m_addsSinceMaintenance = 1;
|
|
writeLine(line.Clear().Appendf(TC("Cache server database obliteration queued!")).data);
|
|
}
|
|
else if (command.Equals(TC("maintenance")))
|
|
{
|
|
m_forceAllSteps = true;
|
|
m_addsSinceMaintenance = 1;
|
|
writeLine(line.Clear().Appendf(TC("Cache server maintenance queued!")).data);
|
|
}
|
|
else
|
|
{
|
|
writeLine(line.Clear().Appendf(TC("Unknown command: %s"), command.data).data);
|
|
}
|
|
|
|
Write("", 1);
|
|
|
|
if (!writeSuccess || !file.Close())
|
|
return false;
|
|
|
|
CasKey key;
|
|
bool deferCreation = false;
|
|
bool fileIsCompressed = false;
|
|
if (!m_storage.StoreCasFile(key, tempFile.data, CasKeyZero, deferCreation, fileIsCompressed))
|
|
return false;
|
|
|
|
writer.WriteCasKey(key);
|
|
|
|
DeleteFileW(tempFile.data);
|
|
return true;
|
|
}
|
|
}
|