Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

232 lines
7.3 KiB

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================
#ifndef FILETRACKER_H
#define FILETRACKER_H
#ifdef _WIN32
#pragma once
#endif
class CBaseFileSystem;
class CPackedStoreFileHandle;
#if !defined( DEDICATED )
// Comments from Fletcher:
//
// TF isn’t sending any hashes to the server. (That’s probably what CSGO is doing,
// but not TF.) Comparing hashes doesn’t work when there are optional updates. (We
// release a new client without updating the server.) Also on TF, we don’t ship
// any textures or audio to the dedicated server.
//
// On TF, the client just confirms that the files were loaded from a trusted source.
//
// When a client connects to a “pure” server, the client is supposed to limit
// which modified files are allowed.)
#include "ifilelist.h"
#include "tier1/utldict.h"
#include "tier0/tslist.h"
#include "tier1/stringpool.h"
struct TrackedFile_t
{
TrackedFile_t()
{
m_nFileFraction = 0;
m_PackFileID = 0;
m_nPackFileNumber = 0;
m_bPackOrVPKFile = false;
m_bFileInVPK = false;
m_iLoadedSearchPathStoreId = 0;
m_bIgnoredForPureServer = false;
}
FileHash_t m_filehashFinal;
const char *m_filename;
const char *m_path;
int m_nFileFraction;
int m_iLoadedSearchPathStoreId; // ID of the search path that we loaded from. Zero if we are not currently loaded.
int m_PackFileID;
int m_nPackFileNumber;
bool m_bPackOrVPKFile;
bool m_bFileInVPK;
bool m_bIgnoredForPureServer; // Did we ignore a file by this name as a result of pure server rules, the last time it was opened?
// The crcIdentifier is a CRC32 of the filename and path. It could be used for quick comparisons of
// path+filename, but since we don't do that at the moment we don't need this.
// CRC32_t m_crcIdentifier;
void RebuildFileName( CStringPool &stringPool, const char *pFilename, const char *pPathID, int nFileFraction );
static bool Less( const TrackedFile_t& lhs, const TrackedFile_t& rhs )
{
int nCmp = Q_strcmp( lhs.m_path, rhs.m_path );
if ( nCmp < 0 )
return true;
else if ( nCmp > 0 )
return false;
nCmp = Q_strcmp( lhs.m_filename, rhs.m_filename );
if ( nCmp < 0 )
return true;
return false;
}
};
struct TrackedVPKFile_t
{
TrackedVPKFile_t()
{
m_PackFileID = 0;
m_nPackFileNumber = 0;
m_nFileFraction = 0;
}
int m_PackFileID;
int m_nPackFileNumber;
int m_nFileFraction;
int m_idxAllOpenedFiles; // Index into m_treeAllOpenedFiles
static bool Less( const TrackedVPKFile_t& lhs, const TrackedVPKFile_t& rhs )
{
if ( lhs.m_nPackFileNumber < rhs.m_nPackFileNumber )
return true;
else if ( lhs.m_nPackFileNumber > rhs.m_nPackFileNumber )
return false;
if ( lhs.m_nFileFraction < rhs.m_nFileFraction )
return true;
else if ( lhs.m_nFileFraction > rhs.m_nFileFraction )
return false;
if ( lhs.m_PackFileID < rhs.m_PackFileID )
return true;
return false;
}
};
class StuffToMD5_t
{
public:
uint8 *m_pubBuffer;
int m_cubBuffer;
MD5Value_t m_md5Value;
int m_idxTrackedVPKFile;
int m_idxListSubmittedJobs;
};
class SubmittedMd5Job_t
{
public:
bool m_bFinished;
MD5Value_t m_md5Value;
};
class CFileTracker2
#ifdef SUPPORT_PACKED_STORE
: IThreadedFileMD5Processor
#endif
{
public:
CFileTracker2( CBaseFileSystem *pFileSystem );
~CFileTracker2();
void ShutdownAsync();
void MarkAllCRCsUnverified();
int GetUnverifiedFileHashes( CUnverifiedFileHash *pFiles, int nMaxFiles );
EFileCRCStatus CheckCachedFileHash( const char *pPathID, const char *pRelativeFilename, int nFileFraction, FileHash_t *pFileHash );
#ifdef SUPPORT_PACKED_STORE
unsigned ThreadedProcessMD5Requests();
virtual int SubmitThreadedMD5Request( uint8 *pubBuffer, int cubBuffer, int PackFileID, int nPackFileNumber, int nPackFileFraction );
virtual bool BlockUntilMD5RequestComplete( int iRequest, MD5Value_t *pMd5ValueOut );
virtual bool IsMD5RequestComplete( int iRequest, MD5Value_t *pMd5ValueOut );
int NotePackFileOpened( const char *pVPKAbsPath, const char *pPathID, int64 nLength );
void NotePackFileAccess( const char *pFilename, const char *pPathID, int iSearchPathStoreId, CPackedStoreFileHandle &VPKHandle );
void AddFileHashForVPKFile( int nPackFileNumber, int nFileFraction, int cbFileLen, MD5Value_t &md5, CPackedStoreFileHandle &fhandle );
#endif
void NoteFileIgnoredForPureServer( const char *pFilename, const char *pPathID, int iSearchPathStoreId );
void NoteFileLoadedFromDisk( const char *pFilename, const char *pPathID, int iSearchPathStoreId, FILE *fp, int64 nLength );
void NoteFileUnloaded( const char *pFilename, const char *pPathID );
int ListOpenedFiles( bool bAllOpened, const char *pchFilenameFind );
IFileList *GetFilesToUnloadForWhitelistChange( IPureServerWhitelist *pNewWhiteList );
private:
int IdxFileFromName( const char *pFilename, const char *pPathID, int nFileFraction, bool bPackOrVPKFile );
CStringPool m_stringPool;
CUtlRBTree< TrackedFile_t, int > m_treeAllOpenedFiles;
CUtlRBTree< TrackedVPKFile_t, int > m_treeTrackedVPKFiles;
CBaseFileSystem *m_pFileSystem;
CThreadMutex m_Mutex; // Threads call into here, so we need to be safe.
#ifdef SUPPORT_PACKED_STORE
CThreadEvent m_threadEventWorkToDo;
CThreadEvent m_threadEventWorkCompleted;
volatile bool m_bThreadShouldRun;
ThreadHandle_t m_hWorkThread;
CTSQueue< StuffToMD5_t > m_PendingJobs;
CTSQueue< StuffToMD5_t > m_CompletedJobs;
CUtlLinkedList< SubmittedMd5Job_t > m_SubmittedJobs;
#endif // SUPPORT_PACKED_STORE
// Stats
int m_cThreadBlocks;
int m_cDupMD5s;
};
#else
//
// Dedicated server NULL filetracker. Pretty much does nothing.
//
class CFileTracker2
#ifdef SUPPORT_PACKED_STORE
: IThreadedFileMD5Processor
#endif
{
public:
CFileTracker2( CBaseFileSystem *pFileSystem ) {}
~CFileTracker2() {}
void ShutdownAsync() {}
void MarkAllCRCsUnverified() {}
int GetUnverifiedFileHashes( CUnverifiedFileHash *pFiles, int nMaxFiles ) { return 0; }
EFileCRCStatus CheckCachedFileHash( const char *pPathID, const char *pRelativeFilename, int nFileFraction, FileHash_t *pFileHash )
{ return k_eFileCRCStatus_CantOpenFile; }
#ifdef SUPPORT_PACKED_STORE
virtual int SubmitThreadedMD5Request( uint8 *pubBuffer, int cubBuffer, int PackFileID, int nPackFileNumber, int nPackFileFraction )
{ return 0; }
virtual bool BlockUntilMD5RequestComplete( int iRequest, MD5Value_t *pMd5ValueOut ) { Assert(0); return true; }
virtual bool IsMD5RequestComplete( int iRequest, MD5Value_t *pMd5ValueOut ) { Assert(0); return true; }
int NotePackFileOpened( const char *pVPKAbsPath, const char *pPathID, int64 nLength ) { return 0; }
void NotePackFileAccess( const char *pFilename, const char *pPathID, int iSearchPathStoreId, CPackedStoreFileHandle &VPKHandle ) {}
void AddFileHashForVPKFile( int nPackFileNumber, int nFileFraction, int cbFileLen, MD5Value_t &md5, CPackedStoreFileHandle &fhandle ) {}
#endif
void NoteFileIgnoredForPureServer( const char *pFilename, const char *pPathID, int iSearchPathStoreId ) {}
void NoteFileLoadedFromDisk( const char *pFilename, const char *pPathID, int iSearchPathStoreId, FILE *fp, int64 nLength ) {}
void NoteFileUnloaded( const char *pFilename, const char *pPathID ) {}
IFileList *GetFilesToUnloadForWhitelistChange( IPureServerWhitelist *pNewWhiteList ) { return NULL; }
};
#endif // DEDICATED
#endif // FILETRACKER_H