#include "stdafx.h" #include "StringHelpers.h" #include "ConsoleSaveFileOriginal.h" #include "File.h" #include #include "compression.h" #include "..\Minecraft.Client\Minecraft.h" #include "..\Minecraft.Client\MinecraftServer.h" #include "..\Minecraft.Client\ServerLevel.h" #include "..\Minecraft.World\net.minecraft.world.level.h" #include "..\Minecraft.World\LevelData.h" #include "..\Minecraft.Client\Common\GameRules\LevelGenerationOptions.h" #include "..\Minecraft.World\net.minecraft.world.level.chunk.storage.h" #define RESERVE_ALLOCATION MEM_RESERVE #define COMMIT_ALLOCATION MEM_COMMIT unsigned int ConsoleSaveFileOriginal::pagesCommitted = 0; void *ConsoleSaveFileOriginal::pvHeap = NULL; ConsoleSaveFileOriginal::ConsoleSaveFileOriginal(const wstring &fileName, LPVOID pvSaveData /*= NULL*/, DWORD dFileSize /*= 0*/, bool forceCleanSave /*= false*/, ESavePlatform plat /*= SAVE_FILE_PLATFORM_LOCAL*/) { InitializeCriticalSectionAndSpinCount(&m_lock,5120); // One time initialise of static stuff required for our storage if( pvHeap == NULL ) { // Reserve a chunk of 64MB of virtual address space for our saves, using 64KB pages. // We'll only be committing these as required to grow the storage we need, which will // the storage to grow without having to use realloc. // AP - The Vita doesn't have virtual memory so a pretend system has been implemented in PSVitaStubs.cpp. // All access to the memory must be done via the access function as the pointer returned from VirtualAlloc // can't be used directly. pvHeap = VirtualAlloc(NULL, MAX_PAGE_COUNT * CSF_PAGE_SIZE, RESERVE_ALLOCATION, PAGE_READWRITE ); } pvSaveMem = pvHeap; m_fileName = fileName; DWORD fileSize = dFileSize; // Load a save from the game rules bool bLevelGenBaseSave = false; LevelGenerationOptions *levelGen = app.getLevelGenerationOptions(); if( pvSaveData == NULL && levelGen != NULL && levelGen->requiresBaseSave()) { pvSaveData = levelGen->getBaseSaveData(fileSize); if(pvSaveData && fileSize != 0) bLevelGenBaseSave = true; } if( pvSaveData == NULL || fileSize == 0) fileSize = StorageManager.GetSaveSize(); if( forceCleanSave ) fileSize = 0; DWORD heapSize = max( fileSize, (DWORD)(1024 * 1024 * 2)); // 4J Stu - Our files are going to be bigger than 2MB so allocate high to start with // Initially committ enough room to store headSize bytes (using CSF_PAGE_SIZE pages, so rounding up here). We should only ever have one save file at a time, // and the pages should be decommitted in the dtor, so pages committed should always be zero at this point. if( pagesCommitted != 0 ) { #ifndef _CONTENT_PACKAGE __debugbreak(); #endif } unsigned int pagesRequired = ( heapSize + (CSF_PAGE_SIZE - 1 ) ) / CSF_PAGE_SIZE; void *pvRet = VirtualAlloc(pvHeap, pagesRequired * CSF_PAGE_SIZE, COMMIT_ALLOCATION, PAGE_READWRITE); if( pvRet == NULL ) { #ifndef _CONTENT_PACKAGE // Out of physical memory __debugbreak(); #endif } pagesCommitted = pagesRequired; if( fileSize > 0) { bool AllocData = false; if(pvSaveData != NULL) { memcpy(pvSaveMem, pvSaveData, fileSize); if(bLevelGenBaseSave) { levelGen->deleteBaseSaveData(); } } else { unsigned int storageLength; StorageManager.GetSaveData( pvSaveMem, &storageLength ); #ifdef __PS3__ StorageManager.FreeSaveData(); #endif app.DebugPrintf("Filesize - %d, Adjusted size - %d\n",fileSize,storageLength); fileSize = storageLength; } { void* pvSourceData = pvSaveMem; int compressed = *(int*)pvSourceData; if( compressed == 0 ) { unsigned int decompSize = *( (int*)pvSourceData+1 ); if(isLocalEndianDifferent(plat)) System::ReverseULONG(&decompSize); // An invalid save, so clear the memory and start from scratch if(decompSize == 0) { // 4J Stu - Saves created between 2/12/2011 and 7/12/2011 will have this problem app.DebugPrintf("Invalid save data format\n"); ZeroMemory( pvSourceData, fileSize ); // Clear the first 8 bytes that reference the header header.WriteHeader( pvSourceData ); } else { unsigned char *buf = new unsigned char[decompSize]; if(plat == SAVE_FILE_PLATFORM_PSVITA) { Compression::VitaVirtualDecompress(buf, &decompSize, (unsigned char *)pvSourceData+8, fileSize-8 ); } else { Compression::getCompression()->SetDecompressionType(plat); // if this save is from another platform, set the correct decompression type Compression::getCompression()->Decompress(buf, &decompSize, (unsigned char *)pvSourceData+8, fileSize-8 ); Compression::getCompression()->SetDecompressionType(SAVE_FILE_PLATFORM_LOCAL); // and then set the decompression back to the local machine's standard type } // Only ReAlloc if we need to (we might already have enough) and align to 512 byte boundaries DWORD currentHeapSize = pagesCommitted * CSF_PAGE_SIZE; DWORD desiredSize = decompSize; if( desiredSize > currentHeapSize ) { unsigned int pagesRequired = ( desiredSize + (CSF_PAGE_SIZE - 1 ) ) / CSF_PAGE_SIZE; void *pvRet = VirtualAlloc(pvHeap, pagesRequired * CSF_PAGE_SIZE, COMMIT_ALLOCATION, PAGE_READWRITE); if( pvRet == NULL ) { // Out of physical memory __debugbreak(); } pagesCommitted = pagesRequired; } memcpy(pvSaveMem, buf, decompSize); delete[] buf; } } } header.ReadHeader( pvSaveMem, plat ); } else { // Clear the first 8 bytes that reference the header header.WriteHeader( pvSaveMem ); } } ConsoleSaveFileOriginal::~ConsoleSaveFileOriginal() { VirtualFree( pvHeap, MAX_PAGE_COUNT * CSF_PAGE_SIZE, MEM_DECOMMIT ); pagesCommitted = 0; // Make sure we don't have any thumbnail data still waiting round - we can't need it now we've destroyed the save file anyway #if defined __PS3__ app.GetSaveThumbnail(NULL,NULL, NULL,NULL); #endif DeleteCriticalSection(&m_lock); } // Add the file to our table of internal files if not already there // Open our actual save file ready for reading/writing, and the set the file pointer to the start of this file FileEntry *ConsoleSaveFileOriginal::createFile( const ConsoleSavePath &fileName ) { LockSaveAccess(); FileEntry *file = header.AddFile( fileName.getName() ); ReleaseSaveAccess(); return file; } void ConsoleSaveFileOriginal::deleteFile( FileEntry *file ) { if( file == NULL ) return; LockSaveAccess(); DWORD numberOfBytesRead = 0; DWORD numberOfBytesWritten = 0; const int bufferSize = 4096; int amountToRead = bufferSize; byte buffer[bufferSize]; DWORD bufferDataSize = 0; char *readStartOffset = (char *)pvSaveMem + file->data.startOffset + file->getFileSize(); char *writeStartOffset = (char *)pvSaveMem + file->data.startOffset; char *endOfDataOffset = (char *)pvSaveMem + header.GetStartOfNextData(); while(true) { // Fill buffer from file if( readStartOffset + bufferSize > endOfDataOffset ) { amountToRead = (int)(endOfDataOffset - readStartOffset); } else { amountToRead = bufferSize; } if( amountToRead == 0 ) break; memcpy( buffer, readStartOffset, amountToRead ); numberOfBytesRead = amountToRead; bufferDataSize = amountToRead; readStartOffset += numberOfBytesRead; // Write buffer to file memcpy( (void *)writeStartOffset, buffer, bufferDataSize ); numberOfBytesWritten = bufferDataSize; writeStartOffset += numberOfBytesWritten; } header.RemoveFile( file ); finalizeWrite(); ReleaseSaveAccess(); } void ConsoleSaveFileOriginal::setFilePointer(FileEntry *file,LONG lDistanceToMove,PLONG lpDistanceToMoveHigh,DWORD dwMoveMethod) { LockSaveAccess(); file->currentFilePointer = file->data.startOffset + lDistanceToMove; if( dwMoveMethod == FILE_END) { file->currentFilePointer += file->getFileSize(); } ReleaseSaveAccess(); } // If this file needs to grow, move the data after along void ConsoleSaveFileOriginal::PrepareForWrite( FileEntry *file, DWORD nNumberOfBytesToWrite ) { int bytesToGrowBy = ( (file->currentFilePointer - file->data.startOffset) + nNumberOfBytesToWrite) - file->getFileSize(); if( bytesToGrowBy <= 0 ) return; // 4J Stu - Not forcing a minimum size, it is up to the caller to write data in sensible amounts // This lets us keep some of the smaller files small //if( bytesToGrowBy < 1024 ) // bytesToGrowBy = 1024; // Move all the data beyond us MoveDataBeyond(file, bytesToGrowBy); // Update our length if( file->data.length < 0 ) file->data.length = 0; file->data.length += bytesToGrowBy; // Write the header with the updated data finalizeWrite(); } BOOL ConsoleSaveFileOriginal::writeFile(FileEntry *file,LPCVOID lpBuffer, DWORD nNumberOfBytesToWrite, LPDWORD lpNumberOfBytesWritten) { assert( pvSaveMem != NULL ); if( pvSaveMem == NULL ) { return 0; } LockSaveAccess(); PrepareForWrite( file, nNumberOfBytesToWrite ); char *writeStartOffset = (char *)pvSaveMem + file->currentFilePointer; //printf("Write: pvSaveMem = %0xd, currentFilePointer = %d, writeStartOffset = %0xd\n", pvSaveMem, file->currentFilePointer, writeStartOffset); memcpy( (void *)writeStartOffset, lpBuffer, nNumberOfBytesToWrite ); *lpNumberOfBytesWritten = nNumberOfBytesToWrite; if(file->data.length < 0) file->data.length = 0; file->currentFilePointer += *lpNumberOfBytesWritten; //wprintf(L"Wrote %d bytes to %s, new file pointer is %I64d\n", *lpNumberOfBytesWritten, file->data.filename, file->currentFilePointer); file->updateLastModifiedTime(); ReleaseSaveAccess(); return 1; } BOOL ConsoleSaveFileOriginal::zeroFile(FileEntry *file, DWORD nNumberOfBytesToWrite, LPDWORD lpNumberOfBytesWritten) { assert( pvSaveMem != NULL ); if( pvSaveMem == NULL ) { return 0; } LockSaveAccess(); PrepareForWrite( file, nNumberOfBytesToWrite ); char *writeStartOffset = (char *)pvSaveMem + file->currentFilePointer; //printf("Write: pvSaveMem = %0xd, currentFilePointer = %d, writeStartOffset = %0xd\n", pvSaveMem, file->currentFilePointer, writeStartOffset); memset( (void *)writeStartOffset, 0, nNumberOfBytesToWrite ); *lpNumberOfBytesWritten = nNumberOfBytesToWrite; if(file->data.length < 0) file->data.length = 0; file->currentFilePointer += *lpNumberOfBytesWritten; //wprintf(L"Wrote %d bytes to %s, new file pointer is %I64d\n", *lpNumberOfBytesWritten, file->data.filename, file->currentFilePointer); file->updateLastModifiedTime(); ReleaseSaveAccess(); return 1; } BOOL ConsoleSaveFileOriginal::readFile( FileEntry *file, LPVOID lpBuffer, DWORD nNumberOfBytesToRead, LPDWORD lpNumberOfBytesRead) { DWORD actualBytesToRead; assert( pvSaveMem != NULL ); if( pvSaveMem == NULL ) { return 0; } LockSaveAccess(); char *readStartOffset = (char *)pvSaveMem + file->currentFilePointer; //printf("Read: pvSaveMem = %0xd, currentFilePointer = %d, readStartOffset = %0xd\n", pvSaveMem, file->currentFilePointer, readStartOffset); assert( nNumberOfBytesToRead <= file->getFileSize() ); actualBytesToRead = nNumberOfBytesToRead; if( file->currentFilePointer + nNumberOfBytesToRead > file->data.startOffset + file->data.length ) { actualBytesToRead = (file->data.startOffset + file->data.length) - file->currentFilePointer; } memcpy( lpBuffer, readStartOffset, actualBytesToRead ); *lpNumberOfBytesRead = actualBytesToRead; file->currentFilePointer += *lpNumberOfBytesRead; //wprintf(L"Read %d bytes from %s, new file pointer is %I64d\n", *lpNumberOfBytesRead, file->data.filename, file->currentFilePointer); ReleaseSaveAccess(); return 1; } BOOL ConsoleSaveFileOriginal::closeHandle( FileEntry *file ) { LockSaveAccess(); finalizeWrite(); ReleaseSaveAccess(); return TRUE; } void ConsoleSaveFileOriginal::finalizeWrite() { LockSaveAccess(); header.WriteHeader( pvSaveMem ); ReleaseSaveAccess(); } void ConsoleSaveFileOriginal::MoveDataBeyond(FileEntry *file, DWORD nNumberOfBytesToWrite) { DWORD numberOfBytesRead = 0; DWORD numberOfBytesWritten = 0; const DWORD bufferSize = 4096; DWORD amountToRead = bufferSize; //assert( nNumberOfBytesToWrite <= bufferSize ); static byte buffer1[bufferSize]; static byte buffer2[bufferSize]; DWORD buffer1Size = 0; DWORD buffer2Size = 0; // Only ReAlloc if we need to (we might already have enough) and align to 512 byte boundaries DWORD currentHeapSize = pagesCommitted * CSF_PAGE_SIZE; DWORD desiredSize = header.GetFileSize() + nNumberOfBytesToWrite; if( desiredSize > currentHeapSize ) { unsigned int pagesRequired = ( desiredSize + (CSF_PAGE_SIZE - 1 ) ) / CSF_PAGE_SIZE; void *pvRet = VirtualAlloc(pvHeap, pagesRequired * CSF_PAGE_SIZE, COMMIT_ALLOCATION, PAGE_READWRITE); if( pvRet == NULL ) { // Out of physical memory __debugbreak(); } pagesCommitted = pagesRequired; } // This is the start of where we want the space to be, and the start of the data that we need to move char *spaceStartOffset = (char *)pvSaveMem + file->data.startOffset + file->getFileSize(); // This is the end of where we want the space to be char *spaceEndOffset = spaceStartOffset + nNumberOfBytesToWrite; // This is the current end of the data that we want to move char *beginEndOfDataOffset = (char *)pvSaveMem + header.GetStartOfNextData(); // This is where the end of the data is going to be char *finishEndOfDataOffset = beginEndOfDataOffset + nNumberOfBytesToWrite; // This is where we are going to read from (with the amount we want to read subtracted before we read) char *readStartOffset = beginEndOfDataOffset; // This is where we can safely write to (with the amount we want write subtracted before we write) char *writeStartOffset = finishEndOfDataOffset; //printf("\n******* MOVEDATABEYOND *******\n"); //printf("Space start: %d, space end: %d\n", spaceStartOffset - (char *)pvSaveMem, spaceEndOffset - (char *)pvSaveMem); //printf("Current end of data: %d, new end of data: %d\n", beginEndOfDataOffset - (char *)pvSaveMem, finishEndOfDataOffset - (char *)pvSaveMem); // Optimisation for things that are being moved in whole region file sector (4K chunks). We could generalise this a bit more but seems safest at the moment to identify this particular type // of move and code explicitly for this situation if( ( nNumberOfBytesToWrite & 4095 ) == 0 ) { if( nNumberOfBytesToWrite > 0 ) { // Get addresses for start & end of the region we are copying from as uintptr_t, for easier maths uintptr_t uiFromStart = (uintptr_t)spaceStartOffset; uintptr_t uiFromEnd = (uintptr_t)beginEndOfDataOffset; // Round both of these values to get 4096 byte chunks that we will need to at least partially move uintptr_t uiFromStartChunk = uiFromStart & ~((uintptr_t)4095); uintptr_t uiFromEndChunk = (uiFromEnd - 1 ) & ~((uintptr_t)4095); // Loop through all the affected source 4096 chunks, going backwards so we don't overwrite anything we'll need in the future for( uintptr_t uiCurrentChunk = uiFromEndChunk; uiCurrentChunk >= uiFromStartChunk; uiCurrentChunk -= 4096 ) { // Establish chunk we'll need to copy uintptr_t uiCopyStart = uiCurrentChunk; uintptr_t uiCopyEnd = uiCurrentChunk + 4096; // Clamp chunk to the bounds of the full region we are trying to copy if( uiCopyStart < uiFromStart ) { // Needs to be clampged against the start of our region uiCopyStart = uiFromStart; } if ( uiCopyEnd > uiFromEnd ) { // Needs to be clamped to the end of our region uiCopyEnd = uiFromEnd; } XMemCpy( (void *)(uiCopyStart + nNumberOfBytesToWrite), ( void *)uiCopyStart, uiCopyEnd - uiCopyStart ); } } } else { while(true) { // Copy buffer 1 to buffer 2 memcpy( buffer2, buffer1, buffer1Size); buffer2Size = buffer1Size; // Fill buffer 1 from file if( (readStartOffset - bufferSize) < spaceStartOffset ) { amountToRead = (DWORD)(readStartOffset - spaceStartOffset); } else { amountToRead = bufferSize; } // Push the read point back by the amount of bytes that we are going to read readStartOffset -= amountToRead; //printf("About to read %u from %d\n", amountToRead, readStartOffset - (char *)pvSaveMem ); memcpy( buffer1, readStartOffset, amountToRead ); numberOfBytesRead = amountToRead; buffer1Size = amountToRead; // Move back the write pointer by the amount of bytes we are going to write writeStartOffset -= buffer2Size; // Write buffer 2 to file if( (writeStartOffset + buffer2Size) <= finishEndOfDataOffset) { //printf("About to write %u to %d\n", buffer2Size, writeStartOffset - (char *)pvSaveMem ); memcpy( (void *)writeStartOffset, buffer2, buffer2Size ); numberOfBytesWritten = buffer2Size; } else { assert((writeStartOffset + buffer2Size) <= finishEndOfDataOffset); numberOfBytesWritten = 0; } if( numberOfBytesRead == 0 ) { //printf("\n************** MOVE COMPLETED *************** \n\n"); assert( writeStartOffset == spaceEndOffset ); break; } } } header.AdjustStartOffsets( file, nNumberOfBytesToWrite ); } bool ConsoleSaveFileOriginal::doesFileExist(ConsoleSavePath file) { LockSaveAccess(); bool exists = header.fileExists( file.getName() ); ReleaseSaveAccess(); return exists; } void ConsoleSaveFileOriginal::Flush(bool autosave, bool updateThumbnail ) { LockSaveAccess(); finalizeWrite(); // Get the frequency of the timer LARGE_INTEGER qwTicksPerSec, qwTime, qwNewTime, qwDeltaTime; float fElapsedTime = 0.0f; QueryPerformanceFrequency( &qwTicksPerSec ); float fSecsPerTick = 1.0f / (float)qwTicksPerSec.QuadPart; unsigned int fileSize = header.GetFileSize(); // Assume that the compression will make it smaller so initially attempt to allocate the current file size // We add 4 bytes to the start so that we can signal compressed data // And another 4 bytes to store the decompressed data size unsigned int compLength = fileSize+8; // 4J Stu - Added TU-1 interim #ifdef __PS3__ // On PS3, don't compress the data as we can't really afford the extra memory this requires for the output buffer. Instead we'll be writing // directly from the save data. StorageManager.SetSaveData(pvSaveMem,fileSize); byte *compData = (byte *)pvSaveMem; #else // Attempt to allocate the required memory // We do not own this, it belongs to the StorageManager byte *compData = (byte *)StorageManager.AllocateSaveData( compLength ); // If we failed to allocate then compData will be NULL // Pre-calculate the compressed data size so that we can attempt to allocate a smaller buffer if(compData == NULL) { // Length should be 0 here so that the compression call knows that we want to know the length back compLength = 0; // Pre-calculate the buffer size required for the compressed data PIXBeginNamedEvent(0,"Pre-calc save compression"); // Save the start time QueryPerformanceCounter( &qwTime ); Compression::getCompression()->Compress(NULL,&compLength,pvSaveMem,fileSize); QueryPerformanceCounter( &qwNewTime ); qwDeltaTime.QuadPart = qwNewTime.QuadPart - qwTime.QuadPart; fElapsedTime = fSecsPerTick * ((FLOAT)(qwDeltaTime.QuadPart)); app.DebugPrintf("Check buffer size: Elapsed time %f\n", fElapsedTime); PIXEndNamedEvent(); // We add 4 bytes to the start so that we can signal compressed data // And another 4 bytes to store the decompressed data size compLength = compLength+8; // Attempt to allocate the required memory compData = (byte *)StorageManager.AllocateSaveData( compLength ); } #endif if(compData != NULL) { // No compression on PS3 - see comment above #ifndef __PS3__ // Re-compress all save data before we save it to disk PIXBeginNamedEvent(0,"Actual save compression"); // Save the start time QueryPerformanceCounter( &qwTime ); Compression::getCompression()->Compress(compData+8,&compLength,pvSaveMem,fileSize); QueryPerformanceCounter( &qwNewTime ); qwDeltaTime.QuadPart = qwNewTime.QuadPart - qwTime.QuadPart; fElapsedTime = fSecsPerTick * ((FLOAT)(qwDeltaTime.QuadPart)); app.DebugPrintf("Compress: Elapsed time %f\n", fElapsedTime); PIXEndNamedEvent(); ZeroMemory(compData,8); int saveVer = 0; memcpy( compData, &saveVer, sizeof(int) ); memcpy( compData+4, &fileSize, sizeof(int) ); app.DebugPrintf("Save data compressed from %d to %d\n", fileSize, compLength); #endif PBYTE pbThumbnailData=NULL; DWORD dwThumbnailDataSize=0; PBYTE pbDataSaveImage=NULL; DWORD dwDataSizeSaveImage=0; #if defined __PS3__ app.GetSaveThumbnail(&pbThumbnailData,&dwThumbnailDataSize,&pbDataSaveImage,&dwDataSizeSaveImage); #endif BYTE bTextMetadata[88]; ZeroMemory(bTextMetadata,88); __int64 seed = 0; bool hasSeed = false; if(MinecraftServer::getInstance()!= NULL && MinecraftServer::getInstance()->levels[0]!=NULL) { seed = MinecraftServer::getInstance()->levels[0]->getLevelData()->getSeed(); hasSeed = true; } int iTextMetadataBytes = app.CreateImageTextData(bTextMetadata, seed, hasSeed, app.GetGameHostOption(eGameHostOption_All), Minecraft::GetInstance()->getCurrentTexturePackId()); INT saveOrCheckpointId = 0; bool validSave = StorageManager.GetSaveUniqueNumber(&saveOrCheckpointId); TelemetryManager->RecordLevelSaveOrCheckpoint(ProfileManager.GetPrimaryPad(), saveOrCheckpointId, compLength+8); #if defined __PS3__ // set the icon and save image StorageManager.SetSaveImages(pbThumbnailData,dwThumbnailDataSize,pbDataSaveImage,dwDataSizeSaveImage,bTextMetadata,iTextMetadataBytes); app.DebugPrintf("Save thumbnail size %d\n",dwThumbnailDataSize); // save the data StorageManager.SaveSaveData( &ConsoleSaveFileOriginal::SaveSaveDataCallback, this ); #ifndef _CONTENT_PACKAGE if( app.DebugSettingsOn()) { if(app.GetWriteSavesToFolderEnabled() ) { DebugFlushToFile(compData, compLength+8); } } #endif ReleaseSaveAccess(); } #else } ReleaseSaveAccess(); #endif } #if defined __PS3__ int ConsoleSaveFileOriginal::SaveSaveDataCallback(LPVOID lpParam,bool bRes) { ConsoleSaveFile *pClass=(ConsoleSaveFile *)lpParam; return 0; } #endif #ifndef _CONTENT_PACKAGE void ConsoleSaveFileOriginal::DebugFlushToFile(void *compressedData /*= NULL*/, unsigned int compressedDataSize /*= 0*/) { LockSaveAccess(); finalizeWrite(); unsigned int fileSize = header.GetFileSize(); DWORD numberOfBytesWritten = 0; File targetFileDir(L"Saves"); if(!targetFileDir.exists()) targetFileDir.mkdir(); wchar_t *fileName = new wchar_t[XCONTENT_MAX_FILENAME_LENGTH+1]; SYSTEMTIME t; GetSystemTime( &t ); //14 chars for the digits //11 chars for the separators + suffix //25 chars total wstring cutFileName = m_fileName; if(m_fileName.length() > XCONTENT_MAX_FILENAME_LENGTH - 25) { cutFileName = m_fileName.substr(0, XCONTENT_MAX_FILENAME_LENGTH - 25); } swprintf(fileName, XCONTENT_MAX_FILENAME_LENGTH+1, L"\\v%04d-%ls%02d.%02d.%02d.%02d.%02d.mcs",VER_PRODUCTBUILD,cutFileName.c_str(), t.wMonth, t.wDay, t.wHour, t.wMinute, t.wSecond); #ifdef _UNICODE wstring wtemp = targetFileDir.getPath() + wstring(fileName); LPCWSTR lpFileName = wtemp.c_str(); #else LPCSTR lpFileName = wstringtofilename( targetFileDir.getPath() + wstring(fileName) ); #endif HANDLE hSaveFile = CreateFile( lpFileName, GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_FLAG_RANDOM_ACCESS, NULL); if(compressedData != NULL && compressedDataSize > 0) { WriteFile( hSaveFile,compressedData,compressedDataSize,&numberOfBytesWritten,NULL); assert(numberOfBytesWritten == compressedDataSize); } else { WriteFile(hSaveFile,pvSaveMem,fileSize,&numberOfBytesWritten,NULL); assert(numberOfBytesWritten == fileSize); } CloseHandle( hSaveFile ); delete[] fileName; ReleaseSaveAccess(); } #endif unsigned int ConsoleSaveFileOriginal::getSizeOnDisk() { return header.GetFileSize(); } wstring ConsoleSaveFileOriginal::getFilename() { return m_fileName; } vector *ConsoleSaveFileOriginal::getFilesWithPrefix(const wstring &prefix) { return header.getFilesWithPrefix( prefix ); } vector *ConsoleSaveFileOriginal::getRegionFilesByDimension(unsigned int dimensionIndex) { return NULL; } #if defined(__PS3__) wstring ConsoleSaveFileOriginal::getPlayerDataFilenameForLoad(const PlayerUID& pUID) { return header.getPlayerDataFilenameForLoad( pUID ); } wstring ConsoleSaveFileOriginal::getPlayerDataFilenameForSave(const PlayerUID& pUID) { return header.getPlayerDataFilenameForSave( pUID ); } vector *ConsoleSaveFileOriginal::getValidPlayerDatFiles() { return header.getValidPlayerDatFiles(); } #endif int ConsoleSaveFileOriginal::getSaveVersion() { return header.getSaveVersion(); } int ConsoleSaveFileOriginal::getOriginalSaveVersion() { return header.getOriginalSaveVersion(); } void ConsoleSaveFileOriginal::LockSaveAccess() { EnterCriticalSection(&m_lock); } void ConsoleSaveFileOriginal::ReleaseSaveAccess() { LeaveCriticalSection(&m_lock); } ESavePlatform ConsoleSaveFileOriginal::getSavePlatform() { return header.getSavePlatform(); } bool ConsoleSaveFileOriginal::isSaveEndianDifferent() { return header.isSaveEndianDifferent(); } void ConsoleSaveFileOriginal::setLocalPlatform() { header.setLocalPlatform(); } void ConsoleSaveFileOriginal::setPlatform(ESavePlatform plat) { header.setPlatform(plat); } ByteOrder ConsoleSaveFileOriginal::getSaveEndian() { return header.getSaveEndian(); } ByteOrder ConsoleSaveFileOriginal::getLocalEndian() { return header.getLocalEndian(); } void ConsoleSaveFileOriginal::setEndian(ByteOrder endian) { header.setEndian(endian); } bool ConsoleSaveFileOriginal::isLocalEndianDifferent( ESavePlatform plat ) { return getLocalEndian() != header.getEndian(plat); } void ConsoleSaveFileOriginal::ConvertRegionFile(File sourceFile) { DWORD numberOfBytesWritten = 0; DWORD numberOfBytesRead = 0; RegionFile sourceRegionFile(this, &sourceFile); for(unsigned int x = 0; x < 32; ++x) { for(unsigned int z = 0; z < 32; ++z) { DataInputStream *dis = sourceRegionFile.getChunkDataInputStream(x,z); if(dis) { byteArray inData(1024*1024); int read = dis->read(inData); dis->close(); dis->deleteChildStream(); delete dis; DataOutputStream *dos = sourceRegionFile.getChunkDataOutputStream(x,z); dos->write(inData, 0, read); dos->close(); dos->deleteChildStream(); delete dos; delete inData.data; } } } sourceRegionFile.writeAllOffsets(); // saves all the endian swapped offsets back out to the file (not all of these are written in the above processing). } void ConsoleSaveFileOriginal::ConvertToLocalPlatform() { if(getSavePlatform() == SAVE_FILE_PLATFORM_LOCAL) { // already in the correct format return; } // convert each of the region files to the local platform vector *allFilesInSave = getFilesWithPrefix(wstring(L"")); for(AUTO_VAR(it, allFilesInSave->begin()); it < allFilesInSave->end(); ++it) { FileEntry *fe = *it; wstring fName( fe->data.filename ); wstring suffix(L".mcr"); if( fName.compare(fName.length() - suffix.length(), suffix.length(), suffix) == 0 ) { app.DebugPrintf("Processing a region file: %ls\n",fName.c_str()); ConvertRegionFile(File(fe->data.filename) ); } else { app.DebugPrintf("%ls is not a region file, ignoring\n", fName.c_str()); } } setLocalPlatform(); // set the platform of this save to the local platform, now that it's been coverted } void *ConsoleSaveFileOriginal::getWritePointer(FileEntry *file) { return (char *)pvSaveMem + file->currentFilePointer;; }