DragonNest/Common/EtFileSystem/EtFileSystem.h

<EFBFBD><EFBFBD>#pragma once

#include <list>
#include <vector>
#include <map>
#include "Singleton.h"
#include "CriticalSection.h"
#include "StringUtil.h"

#define PACKING_FILE_STRING	"EyedentityGames Packing File 0.1"
#define PACKING_FILE_VERSION	11 //RLKT_pak was 11
#define PACKING_FILE_NOCHECKSUM_VERSION 10
#define PACKING_HEADER_RESERVED	( 1024 - sizeof( SPackingFileHeader ) )
#define DEFAULT_PACKING_FILE_SIZE	512 * 1024 * 1024

struct SPackingFileHeader
{
	char szHeaderString[ 256 ];
	int nVersion;
	int nFileCount;
	DWORD dwFileInfoOffset;
	bool bRequireHeaderWrite;
	char cPadding[ 3 ];
};

//----------------------------------------------------------------------------------------
struct SPackingFileInfo
{
	char szFileName[ _MAX_FNAME ];
	DWORD dwCompressSize;
	DWORD dwOriginalSize;
	DWORD dwAllocSize;
	DWORD dwOffset;
	DWORD dwSeedValue;
	DWORD dwChecksum;
	BOOL dwCrypted; //rlkt_CRYPT
	char cReserved[ 32 ]; //rlkt_CRYPT was 36!
};

class CEtFileHandle
{
	friend class CEtPackingFile;
	friend class CEtFileSystem;
public:
	CEtFileHandle();
	virtual ~CEtFileHandle();
	SPackingFileInfo *GetFileContext() { return &m_FileInfo; }

protected:
	SPackingFileInfo m_FileInfo;
	char *m_pData;
	DWORD m_dwCurPos;

public:
	DWORD		Read(void* pBuff, DWORD nSize);
	int			Seek(int lOffset, int iOrigin);
	bool		ExportFile();

};


//------------------------------------------------------------------------------------

// H<EFBFBD><EFBFBD><EFBFBD><EFBFBD> <EFBFBD>Ȍ<EFBFBD> <EFBFBD><EFBFBD><EFBFBD>
class CPatchCallback
{
public:
	enum
	{
		None = 0,
		SrcPackOpenFailed,		// (<EFBFBD>X<EFBFBD>`<EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD>)<EFBFBD>D<EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>.
		SrcPackFileSizeZero,	// <EFBFBD><EFBFBD><EFBFBD><EFBFBD>)<EFBFBD>X<EFBFBD> <EFBFBD>|<EFBFBD>t<EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD>. 
		SrcPackFileNotFound,	// <EFBFBD><EFBFBD><EFBFBD><EFBFBD>)<EFBFBD><EFBFBD><EFBFBD><EFBFBD> <EFBFBD>|<EFBFBD>D<EFBFBD> ><EFBFBD>D<EFBFBD><EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD>. 
		DstPackAddFailed,		// <EFBFBD><EFBFBD><EFBFBD>)<EFBFBD><EFBFBD><EFBFBD> <EFBFBD>|<EFBFBD> <EFBFBD><EFBFBD><EFBFBD><EFBFBD> <EFBFBD><EFBFBD>(ӈ<EFBFBD><EFBFBD><EFBFBD>.
		PatchFailed,			// (<EFBFBD>X<EFBFBD>ij<EFBFBD> <EFBFBD><EFBFBD>(ӈ<EFBFBD><EFBFBD><EFBFBD>.

	};
	virtual void OnPatch(int nCurrentCnt, int nMaxCount, CFileNameString szFilename) { }
	virtual void OnError(int nErrorCode, CFileNameString szMsg) { }
};

class CEtFileSystemLog
{
public:
	virtual void Log(const CHAR* fmt, ...) { }
};


class CEtPackingFile
{
public:
	CEtPackingFile();
	virtual ~CEtPackingFile();

public:
	const char *GetCurDir() { return m_strCurDir; }
	void SetRootPath( const char *szPath );

	DWORD CalcChecksum( int nFileIndex );
	void GenerateChecksum();

	void	CloseFileSystem();
	bool	OpenFileSystem(const char* strPath, bool bReadOnly = false );
	bool	NewFileSystem(const char* strPath);
	bool	OptimizeFileSystem(const char* strPath);
	DWORD	GetFileSystemSize();

	int		GetFileCount() { return ( int )m_vecPackingFileInfo.size(); }
	SPackingFileInfo *GetFileInfo( int nIndex ) { return ( SPackingFileInfo * )&m_vecPackingFileInfo[ nIndex ]; }
	void GetFileInfoListCurDir( std::vector< SPackingFileInfo * > &vecFileInfo );
	void GetDirListCurDir( std::vector< std::string > &vecDirList );
	// ֬<EFBFBD> (ӹ<EFBFBD> <EFBFBD>|<EFBFBD><EFBFBD><EFBFBD> (<EFBFBD>X<EFBFBD> \<EFBFBD><EFBFBD><EFBFBD>. 
	bool	Patch(const char* strPatchPath, CPatchCallback* pCallpack, std::vector< CFileNameString > *pvecSkipList );

	void	SetLogger(CEtFileSystemLog* pLog) { m_pLog = pLog; }

public:
	void	CloseFile( CEtFileHandle* pFileHandle ) { SAFE_DELETE( pFileHandle ); }
	CEtFileHandle *OpenFile(const char *strPath);
	CEtFileHandle *OpenOnly( const char *strPath );

	int 	FindFile( const char *pFileName );
	void	AddPackingMap( int nFileInfoIndex );
	void	RemovePackingMap( int nFileInfoIndex );
	void	GeneratePackingMap();
	int		FindSuitableEmptySpace( DWORD dwSize );
	bool	AddFile(const char* strPath);
	bool	AddFile(const char* strName, char *pData, DWORD nSize);
	bool	PatchFile( CEtFileHandle *pFileHandle );
	bool	PatchFileIndex( CEtFileHandle *pFileHandle, int nFileIndex, BOOL bSaveFileInfo = TRUE );
	int		ReservePatchFile( SPackingFileInfo *pFileInfo );
	void	WriteReserveFileInfo();
	void	RemoveFileInfo( int nFileIndex );

	bool	Remove(const char* strName);
	bool	Rename(const char* strName, const char* strRename);

public:
	bool	ChangeDir(const char* strPath);
	void	GetNameOnly(char* strDst, const char* strPath);
	void	GetPathOnly(char* strDst, const char* strPath);
private:
	void	SetCurDir(const char* pTok);
	void	GetSafeName(char* strDst, const char* strName);

private:
	void		_Encode(IN BYTE* pBuffer, int nBufSize, DWORD nCode);
	void		_Decode(IN BYTE* pBuffer, int nBufSize, DWORD nCode);

	// t<EFBFBD> h<EFBFBD><EFBFBD><EFBFBD><EFBFBD>@<EFBFBD> Win32 API <EFBFBD> 8<EFBFBD><EFBFBD><EFBFBD><EFBFBD> <EFBFBD><EFBFBD>t<EFBFBD>Q<EFBFBD> X<EFBFBD>0<EFBFBD><EFBFBD> <EFBFBD><EFBFBD>.<EFBFBD>D<EFBFBD><EFBFBD>.. 
	bool		_Read(HANDLE hFile, IN void* pBuffer, DWORD ToReadSize);
	bool		_Write(HANDLE hFile, IN void* pBuffer, DWORD ToWriteSize);
	LONGLONG	_Seek(HANDLE hFile, LONGLONG nDistance, DWORD SeekType);
	
	// string <EFBFBD>(<EFBFBD>
	void		_Tokenize( const std::string& str, std::vector<std::string>& tokens, const std::string& delimiters /* = " " */ );
	void		_AllReplace( std::string& str, std::string& szOld, std::string& szNew );


protected:

	bool				m_bIsReadOnly;
	HANDLE				m_hFile;
	SPackingFileHeader	m_PackingFileHeader;
	char				m_strCurDir[_MAX_PATH];
	CFileNameString			m_strPackingFileName;
	CFileNameString			m_szRootPath;

	std::vector< SPackingFileInfo > m_vecPackingFileInfo;
	std::map< INT64, int > m_mapPackingFileIndex;

	CEtFileSystemLog*	m_pLog;
};


class CEtFileSystem : public CSingleton<CEtFileSystem>
{
public:
	CEtFileSystem( bool bDynamicPath = true );
	virtual ~CEtFileSystem();
	
	void	AddPackingFolder(const char* szPackingFileName, const char* szExt = "pak" );
	void	AddPackingFolder(const WCHAR* szPackingFileName, const char* szExt = "pak" );	// <EFBFBD><EFBFBD>L<EFBFBD> 踸<EFBFBD><EFBFBD><EFBFBD>T<EFBFBD> <EFBFBD>̬<EFBFBD>|<EFBFBD> <EFBFBD>t<EFBFBD> <EFBFBD><EFBFBD><EFBFBD>.
	bool	AddPackingFile(const char* szPackingFileName, const char *szRootPath);
	void	RemoveAll();
	
	void FindFileListInFolder( const char *szFolderName, const char *szExt, std::vector<CFileNameString> &szVecResult);

	CEtFileHandle *OpenFile( const char* strPath, bool bOpenOnly = false );
	void	CloseFile(CEtFileHandle* pFH);
	CEtPackingFile *FindFile( const char *pFileName, int &nFindIndex );
	CFileNameString GetFullName( const char *szFileName, bool *bFind = NULL );
	CFileNameString GetFullPath( const char *szPath );
	void FindFullPathList( const char *szPath, std::vector<CFileNameString> &szVecResult );

	void AddResourcePath( const char *szPath, bool bIncludeSubDir = false );
	void RemoveResourcePath( const char *szPath, bool bIncludeSubDir = false );

	bool IsEmpty() {return m_PackingFileMap.empty() ;}

protected:
	void CacheFileList( int nIndex, bool bAdd );

private:
	std::map< CFileNameString, CEtPackingFile * >		m_PackingFileMap;

	struct OffsetStruct {
		std::vector<CFileNameString> szVecFileList;
	};

	std::map< std::string, std::vector<std::string> >	m_szMapCacheFileListInFolder;

	CSyncLock m_SyncLock;
	bool m_bDynamicPath;
};