source-engine/utils/xbox/MakeGameData/MakeZip.cpp
FluorescentCIAAfricanAmerican 3bf9df6b27 1
2020-04-22 12:56:21 -04:00

594 lines
18 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================
#include "MakeGameData.h"
static CUtlSymbolTable g_CriticalPreloadTable( 0, 32, true );
//-----------------------------------------------------------------------------
// Purpose: Compute preload data by file type. Calls into appropriate libraries
// to get the preload info. Libraries would use filename to generate
// the preload if there is a compilation step, otherwise the file buffer
// is a buffer loaded filename.
//-----------------------------------------------------------------------------
static bool GetPreloadBuffer( const char *pFilename, CUtlBuffer &fileBuffer, CUtlBuffer &preloadBuffer )
{
char fileExtension[MAX_PATH];
Q_ExtractFileExtension( pFilename, fileExtension, sizeof( fileExtension ) );
// adding an entire file IS ONLY for files that are expected to be read by the game as a single read
// NOT for files that have any seek pattern
bool bAddEntireFile = false;
// trivial small files, always add
if ( !Q_stricmp( fileExtension, "txt" ) ||
!Q_stricmp( fileExtension, "dat" ) ||
!Q_stricmp( fileExtension, "lst" ) ||
!Q_stricmp( fileExtension, "res" ) ||
!Q_stricmp( fileExtension, "vmt" ) ||
!Q_stricmp( fileExtension, "cfg" ) ||
!Q_stricmp( fileExtension, "bnf" ) ||
!Q_stricmp( fileExtension, "rc" ) ||
!Q_stricmp( fileExtension, "vbf" ) ||
!Q_stricmp( fileExtension, "vfe" ) ||
!Q_stricmp( fileExtension, "pcf" ) ||
!Q_stricmp( fileExtension, "inf" ) )
{
bAddEntireFile = true;
}
// critical resources get blindly added to the preload
if ( !bAddEntireFile && ( g_CriticalPreloadTable.Find( pFilename ) != UTL_INVAL_SYMBOL ) )
{
bAddEntireFile = true;
}
if ( bAddEntireFile && LZMA_IsCompressed( (unsigned char *)fileBuffer.Base() ) )
{
// sorry, not allowed to add entirely to preload if already compressed
// breaks the run-time filesystem due to inability to deliver file as-is
bAddEntireFile = false;
}
if ( bAddEntireFile )
{
if ( fileBuffer.TellMaxPut() >= 1*1024 )
{
// only compress preload files of reasonable size
unsigned int compressedSize = 0;
unsigned char *pCompressedOutput = LZMA_OpportunisticCompress( (unsigned char *)fileBuffer.Base(),
fileBuffer.TellMaxPut(), &compressedSize );
if ( pCompressedOutput )
{
// add as compressed
preloadBuffer.EnsureCapacity( compressedSize );
preloadBuffer.Put( pCompressedOutput, compressedSize );
free( pCompressedOutput );
return true;
}
}
// add entire file to preload section
preloadBuffer.EnsureCapacity( fileBuffer.TellMaxPut() );
preloadBuffer.Put( fileBuffer.Base(), fileBuffer.TellMaxPut() );
return true;
}
// Each library will fetch the optional preload data into caller's buffer
if ( !Q_stricmp( fileExtension, "wav" ) )
{
return GetPreloadData_WAV( pFilename, fileBuffer, preloadBuffer );
}
else if ( !Q_stricmp( fileExtension, "vtf" ) )
{
return GetPreloadData_VTF( pFilename, fileBuffer, preloadBuffer );
}
else if ( !Q_stricmp( fileExtension, "vcs" ) )
{
return GetPreloadData_VCS( pFilename, fileBuffer, preloadBuffer );
}
else if ( !Q_stricmp( fileExtension, "vhv" ) )
{
return GetPreloadData_VHV( pFilename, fileBuffer, preloadBuffer );
}
else if ( !Q_stricmp( fileExtension, "vtx" ) )
{
return GetPreloadData_VTX( pFilename, fileBuffer, preloadBuffer );
}
else if ( !Q_stricmp( fileExtension, "vvd" ) )
{
return GetPreloadData_VVD( pFilename, fileBuffer, preloadBuffer );
}
// others...
return false;
}
void SetupCriticalPreloadScript( const char *pModPath )
{
characterset_t breakSet;
CharacterSetBuild( &breakSet, "" );
// purge any prior entries
g_CriticalPreloadTable.RemoveAll();
// populate table
char szCriticaList[MAX_PATH];
char szToken[MAX_PATH];
V_ComposeFileName( pModPath, "scripts\\preload_xbox.xsc", szCriticaList, sizeof( szCriticaList ) );
CUtlBuffer criticalListBuffer;
if ( ReadFileToBuffer( szCriticaList, criticalListBuffer, true, true ) )
{
for ( ;; )
{
int nTokenSize = criticalListBuffer.ParseToken( &breakSet, szToken, sizeof( szToken ) );
if ( nTokenSize <= 0 )
{
break;
}
V_strlower( szToken );
V_FixSlashes( szToken, CORRECT_PATH_SEPARATOR );
if ( UTL_INVAL_SYMBOL == g_CriticalPreloadTable.Find( szToken ) )
{
g_CriticalPreloadTable.AddString( szToken );
}
}
}
}
CXZipTool::CXZipTool()
{
m_pZip = NULL;
m_hPreloadFile = INVALID_HANDLE_VALUE;
m_hOutputZipFile = INVALID_HANDLE_VALUE;
m_PreloadFilename[0] = '\0';
}
CXZipTool::~CXZipTool()
{
Reset();
}
void CXZipTool::Reset()
{
if ( m_hOutputZipFile != INVALID_HANDLE_VALUE )
{
CloseHandle( m_hOutputZipFile );
m_hOutputZipFile = INVALID_HANDLE_VALUE;
}
if ( m_hPreloadFile != INVALID_HANDLE_VALUE )
{
CloseHandle( m_hPreloadFile );
m_hPreloadFile = INVALID_HANDLE_VALUE;
}
if ( m_PreloadFilename[0] )
{
DeleteFile( m_PreloadFilename );
m_PreloadFilename[0] = '\0';
}
if ( m_pZip )
{
IZip::ReleaseZip( m_pZip );
m_pZip = NULL;
}
m_ZipPreloadDirectoryEntries.Purge();
m_ZipCRCList.Purge();
m_ZipPreloadRemapEntries.Purge();
}
//-----------------------------------------------------------------------------
// Purpose: Add a file buffer to the zip
//-----------------------------------------------------------------------------
bool CXZipTool::AddBuffer( const char *pFilename, CUtlBuffer &fileBuffer, bool bDoPreload )
{
if ( !m_pZip )
{
return false;
}
// safely strip unecessary prefix, otherise pollutes CRC
if ( !strnicmp( pFilename, ".\\", 2 ) )
{
pFilename += 2;
}
// scan for CRC collision now, not at runtime
CRCEntry_t crcEntry;
crcEntry.fileNameCRC = HashStringCaselessConventional( pFilename );
crcEntry.filename = pFilename;
int idx = m_ZipCRCList.Find( crcEntry );
if ( -1 != idx )
{
if ( !V_stricmp( pFilename, m_ZipCRCList[idx].filename.String() ) )
{
// file has already been added, ignore as succesful
return true;
}
Msg( "ERROR: CRC Collision: '%s' with '%s'\n", pFilename, m_ZipCRCList[idx].filename.String() );
return false;
}
else
{
// add unique entry to lists
// must track filenames in a non-sort manner
m_ZipCRCList.Insert( crcEntry );
}
// default, no preload entry
unsigned short preloadDir = INVALID_PRELOAD_ENTRY;
if ( bDoPreload )
{
CUtlBuffer preloadBuffer;
bool bHasPreload = GetPreloadBuffer( pFilename, fileBuffer, preloadBuffer );
int preloadSize = preloadBuffer.TellMaxPut();
if ( bHasPreload && preloadSize > 0 )
{
if ( m_ZipPreloadDirectoryEntries.Count() >= 65534 )
{
Msg( "ERROR: Preload section FULL!, skipping %s\n", pFilename );
return FALSE;
}
// Initialize the entry header
ZIP_PreloadDirectoryEntry entry;
memset( &entry, 0, sizeof( entry ) );
entry.Length = preloadSize;
entry.DataOffset = SetFilePointer( m_hPreloadFile, 0, NULL, FILE_CURRENT );
// Add the directory entry to the preload table
preloadDir = m_ZipPreloadDirectoryEntries.AddToTail( entry );
// Append the preload data to the preload file
DWORD numBytesWritten;
BOOL bOK = WriteFile( m_hPreloadFile, preloadBuffer.Base(), preloadSize, &numBytesWritten, NULL );
if ( !bOK || preloadSize != numBytesWritten )
{
Msg( "ERROR: writing %d preload bytes of '%s'\n", preloadSize, pFilename );
return false;
}
if ( !g_bQuiet )
{
// Spew it
if ( LZMA_IsCompressed( (unsigned char *)preloadBuffer.Base() ) )
{
unsigned int actualSize = LZMA_GetActualSize( (unsigned char *)preloadBuffer.Base() );
Msg( "Preload: '%s': Compressed:%u Actual:%u\n", pFilename, preloadSize, actualSize );
}
else
{
Msg( "Preload: '%s': Length:%u\n", pFilename, preloadSize );
}
}
}
}
unsigned int fileSize = fileBuffer.TellMaxPut();
if ( fileSize > 0 )
{
// order in zip is sorted, not sequential
m_pZip->AddBufferToZip( pFilename, fileBuffer.Base(), fileSize, false );
// track the file in the zip directory to it's preload entry
// order in preload is sequential as buffers are added
preloadRemap_t remap;
remap.filename = pFilename;
remap.preloadDirIndex = preloadDir;
m_ZipPreloadRemapEntries.AddToTail( remap );
if ( !g_bQuiet )
{
Msg( "File: '%s': Length:%u\n", pFilename, fileSize );
}
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Load a file and add it to the zip
//-----------------------------------------------------------------------------
bool CXZipTool::AddFile( const char *pFilename, bool bDoPreload )
{
if ( !m_pZip )
{
return false;
}
FILE* pFile = fopen( pFilename, "rb" );
if( !pFile )
{
Msg( "ERROR: failed to open file: '%s'\n", pFilename );
return false;
}
// Get the length of the file
fseek( pFile, 0, SEEK_END );
unsigned fileSize = ftell( pFile );
fseek( pFile, 0, SEEK_SET);
// read file to buffer
CUtlBuffer fileBuffer;
fileBuffer.EnsureCapacity( fileSize );
fread( fileBuffer.Base(), fileSize, 1, pFile );
fclose( pFile );
fileBuffer.SeekPut( CUtlBuffer::SEEK_HEAD, fileSize );
return AddBuffer( pFilename, fileBuffer, bDoPreload );
}
//-----------------------------------------------------------------------------
// Purpose: Add the preload section and save out the zip file
//-----------------------------------------------------------------------------
bool CXZipTool::End()
{
if ( !m_pZip )
{
return false;
}
// Add the preload section to the zip
if ( m_ZipPreloadDirectoryEntries.Count() )
{
CUtlBuffer sectionBuffer;
CByteswap byteSwap;
// pc tools write 360 native data
byteSwap.ActivateByteSwapping( IsPC() );
// determine the preload data footprint
unsigned int preloadDataSize = SetFilePointer( m_hPreloadFile, 0, NULL, FILE_CURRENT );
// finalize header
m_ZipPreloadHeader.DirectoryEntries = m_ZipPreloadRemapEntries.Count();
m_ZipPreloadHeader.PreloadDirectoryEntries = m_ZipPreloadDirectoryEntries.Count();
// determine the total section size ( treated as a single file inside zip )
unsigned int sectionSize = sizeof( ZIP_PreloadHeader ) +
m_ZipPreloadHeader.PreloadDirectoryEntries * sizeof( ZIP_PreloadDirectoryEntry ) +
m_ZipPreloadHeader.DirectoryEntries * sizeof( unsigned short ) +
preloadDataSize;
sectionSize = AlignValue( sectionSize, m_ZipPreloadHeader.Alignment );
sectionBuffer.EnsureCapacity( sectionSize );
// save data in order
// save the header
byteSwap.SwapFieldsToTargetEndian( &m_ZipPreloadHeader );
sectionBuffer.Put( &m_ZipPreloadHeader, sizeof( ZIP_PreloadHeader ) );
// fixup and save the preload directory
for ( int i=0; i<m_ZipPreloadDirectoryEntries.Count(); i++ )
{
ZIP_PreloadDirectoryEntry entry = m_ZipPreloadDirectoryEntries[i];
byteSwap.SwapFieldsToTargetEndian( &entry );
sectionBuffer.Put( &entry, sizeof( ZIP_PreloadDirectoryEntry ) );
}
// generate remap table
char fileName[MAX_PATH];
int fileSize;
int zipIndex = -1;
unsigned short *pRemapTable = (unsigned short *)malloc( m_ZipPreloadRemapEntries.Count() * sizeof( unsigned short ) );
for ( int i=0; i<m_ZipPreloadRemapEntries.Count(); i++ )
{
// zip files get iterated in the same order they are serialized to disk
fileName[0] = '\0';
fileSize = 0;
zipIndex = m_pZip->GetNextFilename( zipIndex, fileName, sizeof( fileName ), fileSize );
// find the file in the preload remap table
bool bFound = false;
int j;
for ( j=0; j<m_ZipPreloadRemapEntries.Count(); j++ )
{
if ( !Q_stricmp( fileName, m_ZipPreloadRemapEntries[j].filename.String() ) )
{
bFound = true;
break;
}
}
if ( !bFound )
{
// shouldn't happen, every file in the zip has a matching preload remap entry, that is valid or marked invalid
Msg( "ERROR: file '%s' was expected to have an entry in preload table\n", fileName );
}
// the remap table is used to go find a file's preload data (if available)
pRemapTable[i] = m_ZipPreloadRemapEntries[j].preloadDirIndex;
}
for ( int i=0; i<m_ZipPreloadRemapEntries.Count(); i++ )
{
unsigned short s = pRemapTable[i];
sectionBuffer.PutShort( BigShort( s ) );
}
free( pRemapTable );
// get and save preload data
void *pPreloadData = malloc( preloadDataSize );
SetFilePointer( m_hPreloadFile, 0, NULL, FILE_BEGIN );
DWORD numBytesRead;
BOOL bOK = ReadFile( m_hPreloadFile, pPreloadData, preloadDataSize, &numBytesRead, NULL );
if ( !bOK || numBytesRead != preloadDataSize )
{
Msg( "ERROR: failed to read %d bytes from temporary preload file\n", preloadDataSize );
}
CloseHandle( m_hPreloadFile );
m_hPreloadFile = INVALID_HANDLE_VALUE;
sectionBuffer.Put( pPreloadData, preloadDataSize );
free( pPreloadData );
// cannot have written more than was pre-calced, unless code was broken
Assert( (unsigned int)sectionBuffer.TellMaxPut() <= sectionSize );
while( (unsigned int)sectionBuffer.TellMaxPut() < sectionSize )
{
// pad to final alignment
sectionBuffer.PutChar( 0 );
}
m_pZip->AddBufferToZip( PRELOAD_SECTION_NAME, sectionBuffer.Base(), sectionBuffer.TellMaxPut(), false );
}
else
{
// Clear the preload section placeholder
m_pZip->RemoveFileFromZip( PRELOAD_SECTION_NAME );
}
m_pZip->SaveToDisk( m_hOutputZipFile );
Reset();
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Create the zip file
//-----------------------------------------------------------------------------
bool CXZipTool::Begin( const char *pZipFileName, unsigned int alignment )
{
// get the volume of the target zip
char drivePath[MAX_PATH];
_splitpath( pZipFileName, drivePath, NULL, NULL, NULL );
m_pZip = IZip::CreateZip( drivePath, true );
if ( alignment )
{
// making an aligned zip that uses an optimized (but incompatible) format
m_pZip->ForceAlignment( true, false, alignment );
}
// Open the output file
m_hOutputZipFile = CreateFile( pZipFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
if ( m_hOutputZipFile == INVALID_HANDLE_VALUE )
{
Msg( "ERROR: failed to create zip file '%s'\n", pZipFileName );
return false;
}
// Create a temporary file for storing the preloaded data
scriptlib->MakeTemporaryFilename( g_szModPath, m_PreloadFilename, sizeof( m_PreloadFilename ) );
m_hPreloadFile = CreateFile( m_PreloadFilename, GENERIC_READ|GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
if ( m_hPreloadFile == INVALID_HANDLE_VALUE )
{
Msg( "ERROR: failed to create temporary file '%s' for preload data\n", m_PreloadFilename );
CloseHandle( m_hOutputZipFile );
m_hOutputZipFile = INVALID_HANDLE_VALUE;
return false;
}
memset( &m_ZipPreloadHeader, 0, sizeof( ZIP_PreloadHeader ) );
m_ZipPreloadHeader.Version = PRELOAD_HDR_VERSION;
m_ZipPreloadHeader.Alignment = alignment;
// Add a placeholder for the preload section
m_pZip->AddBufferToZip( PRELOAD_SECTION_NAME, NULL, 0, false );
preloadRemap_t remap;
remap.filename = PRELOAD_SECTION_NAME;
remap.preloadDirIndex = INVALID_PRELOAD_ENTRY;
m_ZipPreloadRemapEntries.AddToTail( remap );
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Dump the preload contents
//-----------------------------------------------------------------------------
void CXZipTool::SpewPreloadInfo( const char *pZipName )
{
IZip *pZip = IZip::CreateZip( NULL, true );
HANDLE hZipFile = pZip->ParseFromDisk( pZipName );
if ( !hZipFile )
{
Msg( "Bad or missing zip file, failed to open '%s'\n", pZipName );
return;
}
CUtlBuffer preloadBuffer;
if ( !pZip->ReadFileFromZip( hZipFile, PRELOAD_SECTION_NAME, false, preloadBuffer ) )
{
Msg( "No preload info for '%s'\n", pZipName );
return;
}
preloadBuffer.ActivateByteSwapping( IsPC() );
ZIP_PreloadHeader header;
preloadBuffer.GetObjects( &header );
// get the dir table
ZIP_PreloadDirectoryEntry *pDir = (ZIP_PreloadDirectoryEntry *)malloc( header.PreloadDirectoryEntries * sizeof( ZIP_PreloadDirectoryEntry ) );
preloadBuffer.GetObjects( pDir, header.PreloadDirectoryEntries );
// get the remap table
unsigned short *pRemap = (unsigned short *)malloc( header.DirectoryEntries * sizeof( unsigned short ) );
for ( unsigned int i = 0; i < header.DirectoryEntries; i++ )
{
pRemap[i] = preloadBuffer.GetShort();
}
int zipIndex = -1;
int fileSize;
char fileName[MAX_PATH];
// iterate preload entries sequentially
CUtlDict< unsigned int, int > sizes( true );
for ( unsigned int i = 0; i < header.DirectoryEntries; i++ )
{
fileName[0] = '\0';
fileSize = 0;
zipIndex = pZip->GetNextFilename( zipIndex, fileName, sizeof( fileName ), fileSize );
unsigned short zipPreloadDirIndex = pRemap[i];
if ( zipPreloadDirIndex == INVALID_PRELOAD_ENTRY )
{
continue;
}
Msg( "Offset: 0x%8.8x Length: %5d %s (%d)\n", pDir[zipPreloadDirIndex].DataOffset, pDir[zipPreloadDirIndex].Length, fileName, fileSize );
// total preload sizes by extension
const char *pExt = V_GetFileExtension( fileName );
if ( !pExt )
{
pExt = "???";
}
int iIndex = sizes.Find( pExt );
if ( iIndex == sizes.InvalidIndex() )
{
iIndex = sizes.Insert( pExt );
sizes[iIndex] = 0;
}
sizes[iIndex] += pDir[zipPreloadDirIndex].Length;
}
Msg( "\n" );
Msg( "Preload Size: %.2f MB\n", (float)preloadBuffer.TellMaxPut()/(1024.0f * 1024.0f) );
Msg( "Zip Entries: %d\n", header.DirectoryEntries );
Msg( "Preload Entries: %d\n", header.PreloadDirectoryEntries );
// dump each extension's total size, necessary for debugging who is the largest contributor
for ( int i = 0; i < sizes.Count(); i++ )
{
Msg( "Extension: '%3s' %d bytes (%.2f%s)\n", sizes.GetElementName( i ), sizes[i], (float)sizes[i]/(float)preloadBuffer.TellMaxPut() * 100.0f, "%%" );
}
Msg( "\n" );
free( pRemap );
free( pDir );
IZip::ReleaseZip( pZip );
}