/*****************************************************************************/ /* StormTest.cpp Copyright (c) Ladislav Zezula 2003 */ /*---------------------------------------------------------------------------*/ /* Test module for StormLib */ /*---------------------------------------------------------------------------*/ /* Date Ver Who Comment */ /* -------- ---- --- ------- */ /* 25.03.03 1.00 Lad The first version of StormTest.cpp */ /*****************************************************************************/ #define _CRT_NON_CONFORMING_SWPRINTFS #define _CRT_SECURE_NO_DEPRECATE #define __INCLUDE_CRYPTOGRAPHY__ #define __STORMLIB_SELF__ // Don't use StormLib.lib #include #ifdef _MSC_VER #include #endif #include "../src/StormLib.h" #include "../src/StormCommon.h" #include "TLogHelper.cpp" // Helper class for showing test results #ifdef _MSC_VER #pragma warning(disable: 4505) // 'XXX' : unreferenced local function has been removed #pragma comment(lib, "winmm.lib") #endif #ifdef PLATFORM_LINUX #include #endif //------------------------------------------------------------------------------ // Defines #ifdef PLATFORM_WINDOWS #define WORK_PATH_ROOT "E:\\Multimedia\\MPQs" #endif #ifdef PLATFORM_LINUX #define WORK_PATH_ROOT "/home/ladik/MPQs" #endif #ifdef PLATFORM_MAC #define WORK_PATH_ROOT "/Users/sam/StormLib/test" #endif // Global for the work MPQ static const char * szMpqSubDir = "1995 - Test MPQs"; static const char * szMpqPatchDir = "1995 - Test MPQs\\patches"; typedef int (*FIND_FILE_CALLBACK)(const char * szFullPath); typedef int (*FIND_PAIR_CALLBACK)(const char * szFullPath1, const char * szFullPath2); //----------------------------------------------------------------------------- // Testing data static DWORD AddFlags[] = { // Compression Encryption Fixed key Single Unit Sector CRC 0 | 0 | 0 | 0 | 0, 0 | MPQ_FILE_ENCRYPTED | 0 | 0 | 0, 0 | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | 0 | 0, 0 | 0 | 0 | MPQ_FILE_SINGLE_UNIT | 0, 0 | MPQ_FILE_ENCRYPTED | 0 | MPQ_FILE_SINGLE_UNIT | 0, 0 | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_IMPLODE | 0 | 0 | 0 | 0, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | 0 | 0 | 0, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | 0 | 0, MPQ_FILE_IMPLODE | 0 | 0 | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | 0 | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_IMPLODE | 0 | 0 | 0 | MPQ_FILE_SECTOR_CRC, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | 0 | 0 | MPQ_FILE_SECTOR_CRC, MPQ_FILE_IMPLODE | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | 0 | MPQ_FILE_SECTOR_CRC, MPQ_FILE_COMPRESS | 0 | 0 | 0 | 0, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | 0 | 0 | 0, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | 0 | 0, MPQ_FILE_COMPRESS | 0 | 0 | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | 0 | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | MPQ_FILE_SINGLE_UNIT | 0, MPQ_FILE_COMPRESS | 0 | 0 | 0 | MPQ_FILE_SECTOR_CRC, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | 0 | 0 | MPQ_FILE_SECTOR_CRC, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | MPQ_FILE_FIX_KEY | 0 | MPQ_FILE_SECTOR_CRC, 0xFFFFFFFF }; static DWORD WaveCompressions[] = { MPQ_COMPRESSION_ADPCM_MONO | MPQ_COMPRESSION_HUFFMANN, MPQ_COMPRESSION_ADPCM_STEREO | MPQ_COMPRESSION_HUFFMANN, MPQ_COMPRESSION_PKWARE, MPQ_COMPRESSION_ZLIB, MPQ_COMPRESSION_BZIP2 }; static const wchar_t szUnicodeName1[] = { // Czech 0x010C, 0x0065, 0x0073, 0x006B, 0x00FD, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const wchar_t szUnicodeName2[] = { // Russian 0x0420, 0x0443, 0x0441, 0x0441, 0x043A, 0x0438, 0x0439, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const wchar_t szUnicodeName3[] = { // Greek 0x03B5, 0x03BB, 0x03BB, 0x03B7, 0x03BD, 0x03B9, 0x03BA, 0x03AC, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const wchar_t szUnicodeName4[] = { // Chinese 0x65E5, 0x672C, 0x8A9E, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const wchar_t szUnicodeName5[] = { // Japanese 0x7B80, 0x4F53, 0x4E2D, 0x6587, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const wchar_t szUnicodeName6[] = { // Arabic 0x0627, 0x0644, 0x0639, 0x0639, 0x0631, 0x0628, 0x064A, 0x0629, _T('.'), _T('m'), _T('p'), _T('q'), 0 }; static const char * PatchList_WoW_OldWorld13286[] = { "MPQ_2012_v4_OldWorld.MPQ", "wow-update-oldworld-13154.MPQ", "wow-update-oldworld-13286.MPQ", NULL }; static const char * PatchList_WoW15050[] = { "MPQ_2013_v4_world.MPQ", "wow-update-13164.MPQ", "wow-update-13205.MPQ", "wow-update-13287.MPQ", "wow-update-13329.MPQ", "wow-update-13596.MPQ", "wow-update-13623.MPQ", "wow-update-base-13914.MPQ", "wow-update-base-14007.MPQ", "wow-update-base-14333.MPQ", "wow-update-base-14480.MPQ", "wow-update-base-14545.MPQ", "wow-update-base-14946.MPQ", "wow-update-base-15005.MPQ", "wow-update-base-15050.MPQ", NULL }; static const char * PatchList_WoW16965[] = { "MPQ_2013_v4_locale-enGB.MPQ", "wow-update-enGB-16016.MPQ", "wow-update-enGB-16048.MPQ", "wow-update-enGB-16057.MPQ", "wow-update-enGB-16309.MPQ", "wow-update-enGB-16357.MPQ", "wow-update-enGB-16516.MPQ", "wow-update-enGB-16650.MPQ", "wow-update-enGB-16844.MPQ", "wow-update-enGB-16965.MPQ", NULL }; //----------------------------------------------------------------------------- // Local file functions // Definition of the path separator #ifdef PLATFORM_WINDOWS #define PATH_SEPARATOR '\\' // Path separator for Windows platforms #else #define PATH_SEPARATOR '/' // Path separator for Windows platforms #endif // This must be the directory where our test MPQs are stored. // We also expect a subdirectory named static char szMpqDirectory[MAX_PATH]; size_t cchMpqDirectory = 0; static bool IsFullPath(const char * szFileName) { #ifdef PLATFORM_WINDOWS if(('A' <= szFileName[0] && szFileName[0] <= 'Z') || ('a' <= szFileName[0] && szFileName[0] <= 'z')) { return (szFileName[1] == ':' && szFileName[2] == PATH_SEPARATOR); } #endif szFileName = szFileName; return false; } static bool IsMpqExtension(const char * szFileName) { const char * szExtension = strrchr(szFileName, '.'); if(szExtension != NULL) { if(!_stricmp(szExtension, ".mpq")) return true; if(!_stricmp(szExtension, ".w3m")) return true; if(!_stricmp(szExtension, ".w3x")) return true; if(!_stricmp(szExtension, ".mpqe")) return true; if(!_stricmp(szExtension, ".part")) return true; if(!_stricmp(szExtension, ".sv")) return true; if(!_stricmp(szExtension, ".s2ma")) return true; if(!_stricmp(szExtension, ".SC2Map")) return true; if(!_stricmp(szExtension, ".0")) // .MPQ.0 return true; // if(!_stricmp(szExtension, ".link")) // return true; } return false; } static bool CompareBlocks(LPBYTE pbBlock1, LPBYTE pbBlock2, DWORD dwLength, DWORD * pdwDifference) { for(DWORD i = 0; i < dwLength; i++) { if(pbBlock1[i] != pbBlock2[i]) { pdwDifference[0] = i; return false; } } return true; } static size_t ConvertSha1ToText(const unsigned char * sha1_digest, char * szSha1Text) { const char * szTable = "0123456789abcdef"; for(size_t i = 0; i < SHA1_DIGEST_SIZE; i++) { *szSha1Text++ = szTable[(sha1_digest[0] >> 0x04)]; *szSha1Text++ = szTable[(sha1_digest[0] & 0x0F)]; sha1_digest++; } *szSha1Text = 0; return (SHA1_DIGEST_SIZE * 2); } static int GetPathSeparatorCount(const char * szPath) { int nSeparatorCount = 0; while(szPath[0] != 0) { if(szPath[0] == '\\' || szPath[0] == '/') nSeparatorCount++; szPath++; } return nSeparatorCount; } static const char * FindNextPathPart(const char * szPath, size_t nPartCount) { const char * szPathPart = szPath; while(szPath[0] != 0 && nPartCount > 0) { // Is there path separator? if(szPath[0] == '\\' || szPath[0] == '/') { szPathPart = szPath + 1; nPartCount--; } // Move to the next letter szPath++; } return szPathPart; } static const char * GetShortPlainName(const char * szFileName) { const char * szPlainName = FindNextPathPart(szFileName, 1000); const char * szPlainEnd = szFileName + strlen(szFileName); // If the name is still too long, cut it if((szPlainEnd - szPlainName) > 50) szPlainName = szPlainEnd - 50; return szPlainName; } static void CopyPathPart(char * szBuffer, const char * szPath) { while(szPath[0] != 0) { szBuffer[0] = (szPath[0] == '\\' || szPath[0] == '/') ? '/' : szPath[0]; szBuffer++; szPath++; } *szBuffer = 0; } static bool CopyStringAndVerifyConversion( const TCHAR * szFoundFile, TCHAR * szBufferT, char * szBufferA) { // Convert the TCHAR name to ANSI name CopyFileName(szBufferA, szFoundFile, _tcslen(szFoundFile)); CopyFileName(szBufferT, szBufferA, strlen(szBufferA)); // Compare both TCHAR strings return (_tcsicmp(szBufferT, szFoundFile) == 0) ? true : false; } static void CalculateRelativePath(const char * szFullPath1, const char * szFullPath2, char * szBuffer) { const char * szPathPart1 = szFullPath1; const char * szPathPart2 = szFullPath2; const char * szNextPart1; const char * szNextPart2; int nEqualParts = 0; int nStepsUp = 0; // Parse both paths and find all path parts that are equal for(;;) { // Find the next part of the first path szNextPart1 = FindNextPathPart(szPathPart1, 1); if(szNextPart1 == szPathPart1) break; szNextPart2 = FindNextPathPart(szPathPart2, 1); if(szNextPart2 == szPathPart2) break; // Are these equal? if((szNextPart2 - szPathPart2) != (szNextPart1 - szPathPart1)) break; if(_strnicmp(szPathPart1, szPathPart2, (szNextPart1 - szPathPart1 - 1))) break; // Increment the number of path parts that are equal szPathPart1 = szNextPart1; szPathPart2 = szNextPart2; nEqualParts++; } // If we found at least one equal part, we can create relative path if(nEqualParts != 0) { // Calculate how many steps up we need to go nStepsUp = GetPathSeparatorCount(szPathPart2); // Append "../" nStepsUp-times for(int i = 0; i < nStepsUp; i++) { *szBuffer++ = '.'; *szBuffer++ = '.'; *szBuffer++ = '/'; } // Append the rest of the path. Also change DOS backslashes to slashes CopyPathPart(szBuffer, szPathPart1); return; } // Failed. Just copy the source path as it is strcpy(szBuffer, szFullPath1); } static TFileStream * FileStream_OpenFileA(const char * szFileName, DWORD dwStreamFlags) { TCHAR szFileNameT[MAX_PATH]; CopyFileName(szFileNameT, szFileName, strlen(szFileName)); return FileStream_OpenFile(szFileNameT, dwStreamFlags); } static TFileStream * FileStream_CreateFileA(const char * szFileName, DWORD dwStreamFlags) { TCHAR szFileNameT[MAX_PATH]; CopyFileName(szFileNameT, szFileName, strlen(szFileName)); return FileStream_CreateFile(szFileNameT, dwStreamFlags); } static size_t FileStream_PrefixA(const char * szFileName, DWORD * pdwProvider) { TCHAR szFileNameT[MAX_PATH]; size_t nPrefixLength = 0; if(szFileName != NULL) { CopyFileName(szFileNameT, szFileName, strlen(szFileName)); nPrefixLength = FileStream_Prefix(szFileNameT, pdwProvider); } return nPrefixLength; } static void CreateFullPathName(char * szBuffer, const char * szSubDir, const char * szNamePart1, const char * szNamePart2 = NULL) { char * szSaveBuffer = szBuffer; size_t nPrefixLength = 0; size_t nLength; DWORD dwProvider = 0; bool bIsFullPath = false; char chSeparator = PATH_SEPARATOR; // Determine the path prefix if(szNamePart1 != NULL) { nPrefixLength = FileStream_PrefixA(szNamePart1, &dwProvider); if((dwProvider & BASE_PROVIDER_MASK) == BASE_PROVIDER_HTTP) { bIsFullPath = true; chSeparator = '/'; } else bIsFullPath = IsFullPath(szNamePart1 + nPrefixLength); } // Copy the MPQ prefix, if any if(nPrefixLength > 0) { memcpy(szBuffer, szNamePart1, nPrefixLength); szSaveBuffer += nPrefixLength; szNamePart1 += nPrefixLength; szBuffer += nPrefixLength; } // If the given name is not a full path, copy the MPQ directory if(bIsFullPath == false) { // Copy the master MPQ directory memcpy(szBuffer, szMpqDirectory, cchMpqDirectory); szBuffer += cchMpqDirectory; // Append the subdirectory, if any if(szSubDir != NULL && (nLength = strlen(szSubDir)) != 0) { // No leading or trailing separator are allowed assert(szSubDir[0] != '/' && szSubDir[0] != '\\'); assert(szSubDir[nLength - 1] != '/' && szSubDir[nLength - 1] != '\\'); // Append file path separator *szBuffer++ = PATH_SEPARATOR; // Append the subdirectory memcpy(szBuffer, szSubDir, nLength); szBuffer += nLength; } } // Copy the file name, if any if(szNamePart1 != NULL && (nLength = strlen(szNamePart1)) != 0) { // Path separators are not allowed in the name part assert(szNamePart1[0] != '\\' && szNamePart1[0] != '/'); assert(szNamePart1[nLength - 1] != '/' && szNamePart1[nLength - 1] != '\\'); // Append file path separator if(bIsFullPath == false) *szBuffer++ = PATH_SEPARATOR; // Copy the file name memcpy(szBuffer, szNamePart1, nLength); szBuffer += nLength; } // Append the second part of the name if(szNamePart2 != NULL && (nLength = strlen(szNamePart2)) != 0) { // Copy the file name memcpy(szBuffer, szNamePart2, nLength); szBuffer += nLength; } // Normalize the path separators while(szSaveBuffer < szBuffer) { szSaveBuffer[0] = (szSaveBuffer[0] != '/' && szSaveBuffer[0] != '\\') ? szSaveBuffer[0] : chSeparator; szSaveBuffer++; } // Terminate the buffer with zero *szBuffer = 0; } static int CalculateFileSha1(TLogHelper * pLogger, const char * szFullPath, char * szFileSha1) { TFileStream * pStream; unsigned char sha1_digest[SHA1_DIGEST_SIZE]; const char * szShortPlainName = GetShortPlainName(szFullPath); hash_state sha1_state; ULONGLONG ByteOffset = 0; ULONGLONG FileSize = 0; BYTE * pbFileBlock; DWORD cbBytesToRead; DWORD cbFileBlock = 0x100000; int nError = ERROR_SUCCESS; // Notify the user pLogger->PrintProgress("Hashing file %s", szShortPlainName); szFileSha1[0] = 0; // Open the file to be verified pStream = FileStream_OpenFileA(szFullPath, STREAM_FLAG_READ_ONLY); if(pStream != NULL) { // Retrieve the size of the file FileStream_GetSize(pStream, &FileSize); // Allocate the buffer for loading file parts pbFileBlock = STORM_ALLOC(BYTE, cbFileBlock); if(pbFileBlock != NULL) { // Initialize SHA1 calculation sha1_init(&sha1_state); // Calculate the SHA1 of the file while(ByteOffset < FileSize) { // Notify the user pLogger->PrintProgress("Hashing file %s (%I64u of %I64u)", szShortPlainName, ByteOffset, FileSize); // Load the file block cbBytesToRead = ((FileSize - ByteOffset) > cbFileBlock) ? cbFileBlock : (DWORD)(FileSize - ByteOffset); if(!FileStream_Read(pStream, &ByteOffset, pbFileBlock, cbBytesToRead)) { nError = GetLastError(); break; } // Add to SHA1 sha1_process(&sha1_state, pbFileBlock, cbBytesToRead); ByteOffset += cbBytesToRead; } // Notify the user pLogger->PrintProgress("Hashing file %s (%I64u of %I64u)", szShortPlainName, ByteOffset, FileSize); // Finalize SHA1 sha1_done(&sha1_state, sha1_digest); // Convert the SHA1 to ANSI text ConvertSha1ToText(sha1_digest, szFileSha1); STORM_FREE(pbFileBlock); } FileStream_Close(pStream); } // If we calculated something, return OK if(nError == ERROR_SUCCESS && szFileSha1[0] == 0) nError = ERROR_CAN_NOT_COMPLETE; return nError; } //----------------------------------------------------------------------------- // Directory search static HANDLE InitDirectorySearch(const char * szDirectory) { #ifdef PLATFORM_WINDOWS WIN32_FIND_DATA wf; HANDLE hFind; TCHAR szSearchMask[MAX_PATH]; // Keep compilers happy CopyFileName(szSearchMask, szDirectory, strlen(szDirectory)); _tcscat(szSearchMask, _T("\\*")); // Construct the directory mask hFind = FindFirstFile(szSearchMask, &wf); return (hFind != INVALID_HANDLE_VALUE) ? hFind : NULL; #endif #ifdef PLATFORM_LINUX // Keep compilers happy return (HANDLE)opendir(szDirectory); #endif } static bool SearchDirectory(HANDLE hFind, char * szDirEntry, bool & IsDirectory) { #ifdef PLATFORM_WINDOWS WIN32_FIND_DATA wf; TCHAR szDirEntryT[MAX_PATH]; char szDirEntryA[MAX_PATH]; __SearchNextEntry: // Search for the hnext entry. if(FindNextFile(hFind, &wf)) { // Verify if the directory entry is an UNICODE name that would be destroyed // by Unicode->ANSI->Unicode conversion if(CopyStringAndVerifyConversion(wf.cFileName, szDirEntryT, szDirEntryA) == false) goto __SearchNextEntry; IsDirectory = (wf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ? true : false; CopyFileName(szDirEntry, wf.cFileName, _tcslen(wf.cFileName)); return true; } return false; #endif #ifdef PLATFORM_LINUX struct dirent * directory_entry; directory_entry = readdir((DIR *)hFind); if(directory_entry != NULL) { IsDirectory = (directory_entry->d_type == DT_DIR) ? true : false; strcpy(szDirEntry, directory_entry->d_name); return true; } return false; #endif } static void FreeDirectorySearch(HANDLE hFind) { #ifdef PLATFORM_WINDOWS FindClose(hFind); #endif #ifdef PLATFORM_LINUX closedir((DIR *)hFind); #endif } static int FindFilesInternal(FIND_FILE_CALLBACK pfnTest, char * szDirectory) { HANDLE hFind; char * szPlainName; size_t nLength; char szDirEntry[MAX_PATH]; bool IsDirectory = false; int nError = ERROR_SUCCESS; if(szDirectory != NULL) { // Initiate directory search hFind = InitDirectorySearch(szDirectory); if(hFind != NULL) { // Append slash at the end of the directory name nLength = strlen(szDirectory); szDirectory[nLength++] = PATH_SEPARATOR; szPlainName = szDirectory + nLength; // Skip the first entry, since it's always "." or ".." while(SearchDirectory(hFind, szDirEntry, IsDirectory) && nError == ERROR_SUCCESS) { // Copy the directory entry name to both names strcpy(szPlainName, szDirEntry); // Found a directory? if(IsDirectory) { if(szDirEntry[0] != '.') { nError = FindFilesInternal(pfnTest, szDirectory); } } else { if(pfnTest != NULL) { nError = pfnTest(szDirectory); } } } FreeDirectorySearch(hFind); } } // Free the path buffer, if any return nError; } static int FindFiles(FIND_FILE_CALLBACK pfnFindFile, const char * szSubDirectory) { char szWorkBuff[MAX_PATH]; CreateFullPathName(szWorkBuff, szSubDirectory, NULL); return FindFilesInternal(pfnFindFile, szWorkBuff); } static int FindFilePairsInternal( FIND_PAIR_CALLBACK pfnFilePair, char * szSource, char * szTarget) { char * szPlainName1; char * szPlainName2; int nError = ERROR_SUCCESS; // Setup the search masks strcat(szSource, "\\*"); szPlainName1 = strrchr(szSource, '*'); strcat(szTarget, "\\*"); szPlainName2 = strrchr(szTarget, '*'); // If both paths are OK, perform the search if(szPlainName1 != NULL && szPlainName2 != NULL) { #ifdef PLATFORM_WINDOWS WIN32_FIND_DATAA wf; HANDLE hFind; // Search the second directory hFind = FindFirstFileA(szTarget, &wf); if(hFind != INVALID_HANDLE_VALUE) { // Skip the first entry, since it's always "." or ".." while(FindNextFileA(hFind, &wf) && nError == ERROR_SUCCESS) { // Found a directory? if(wf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { if(wf.cFileName[0] != '.') { strcpy(szPlainName1, wf.cFileName); strcpy(szPlainName2, wf.cFileName); nError = FindFilePairsInternal(pfnFilePair, szSource, szTarget); } } else { if(pfnFilePair != NULL) { strcpy(szPlainName1, wf.cFileName); strcpy(szPlainName2, wf.cFileName); nError = pfnFilePair(szSource, szTarget); } } } FindClose(hFind); } #endif } return nError; } static int FindFilePairs(FIND_PAIR_CALLBACK pfnFindPair, const char * szSourceSubDir, const char * szTargetSubDir) { char szSource[MAX_PATH]; char szTarget[MAX_PATH]; // Create the source search mask CreateFullPathName(szSource, szSourceSubDir, NULL); CreateFullPathName(szTarget, szTargetSubDir, NULL); return FindFilePairsInternal(pfnFindPair, szSource, szTarget); } static int InitializeMpqDirectory(char * argv[], int argc) { TLogHelper Logger("InitWorkDir"); TFileStream * pStream; const char * szWhereFrom = NULL; const char * szDirName; char szFullPath[MAX_PATH]; // Retrieve the name of the MPQ directory if(argc > 1 && argv[1] != NULL) { szWhereFrom = "command line"; szDirName = argv[1]; } else { szWhereFrom = "default"; szDirName = WORK_PATH_ROOT; } // Copy the name of the MPQ directory. strcpy(szMpqDirectory, szDirName); cchMpqDirectory = strlen(szMpqDirectory); // Cut trailing slashes and/or backslashes while((cchMpqDirectory > 0) && (szMpqDirectory[cchMpqDirectory - 1] == '/' || szMpqDirectory[cchMpqDirectory - 1] == '\\')) cchMpqDirectory--; szMpqDirectory[cchMpqDirectory] = 0; // Print the work directory info Logger.PrintMessage("Work directory %s (%s)", szMpqDirectory, szWhereFrom); // Verify if the work MPQ directory is writable CreateFullPathName(szFullPath, NULL, "TestFile.bin"); pStream = FileStream_CreateFileA(szFullPath, 0); if(pStream == NULL) return Logger.PrintError("MPQ subdirectory doesn't exist or is not writable"); // Close the stream FileStream_Close(pStream); remove(szFullPath); // Verify if the working directory exists and if there is a subdirectory with the file name CreateFullPathName(szFullPath, szMpqSubDir, "ListFile_Blizzard.txt"); pStream = FileStream_OpenFileA(szFullPath, STREAM_FLAG_READ_ONLY); if(pStream == NULL) return Logger.PrintError("The main listfile (%s) was not found. Check your paths", GetShortPlainName(szFullPath)); // Close the stream FileStream_Close(pStream); return ERROR_SUCCESS; } static int GetFilePatchCount(TLogHelper * pLogger, HANDLE hMpq, const char * szFileName) { TCHAR * szPatchName; HANDLE hFile; TCHAR szPatchChain[0x400]; int nPatchCount = 0; int nError = ERROR_SUCCESS; // Open the MPQ file if(SFileOpenFileEx(hMpq, szFileName, 0, &hFile)) { // Notify the user pLogger->PrintProgress("Verifying patch chain for %s ...", GetShortPlainName(szFileName)); // Query the patch chain if(!SFileGetFileInfo(hFile, SFileInfoPatchChain, szPatchChain, sizeof(szPatchChain), NULL)) nError = pLogger->PrintError("Failed to retrieve the patch chain on %s", szFileName); // Is there anything at all in the patch chain? if(nError == ERROR_SUCCESS && szPatchChain[0] == 0) { pLogger->PrintError("The patch chain for %s is empty", szFileName); nError = ERROR_FILE_CORRUPT; } // Now calculate the number of patches if(nError == ERROR_SUCCESS) { // Get the pointer to the patch szPatchName = szPatchChain; // Skip the base name for(;;) { // Skip the current name szPatchName = szPatchName + _tcslen(szPatchName) + 1; if(szPatchName[0] == 0) break; // Increment number of patches nPatchCount++; } } SFileCloseFile(hFile); } else { pLogger->PrintError("Failed to open file %s", szFileName); } return nPatchCount; } static int VerifyFilePatchCount(TLogHelper * pLogger, HANDLE hMpq, const char * szFileName, int nExpectedPatchCount) { int nPatchCount = 0; // Retrieve the patch count pLogger->PrintProgress("Verifying patch count for %s ...", szFileName); nPatchCount = GetFilePatchCount(pLogger, hMpq, szFileName); // Check if there are any patches at all if(nExpectedPatchCount != 0 && nPatchCount == 0) { pLogger->PrintMessage("There are no patches beyond %s", szFileName); return ERROR_FILE_CORRUPT; } // Check if the number of patches fits if(nPatchCount != nExpectedPatchCount) { pLogger->PrintMessage("Unexpected number of patches for %s", szFileName); return ERROR_FILE_CORRUPT; } return ERROR_SUCCESS; } static int CreateEmptyFile(TLogHelper * pLogger, const char * szPlainName, ULONGLONG FileSize, char * szBuffer) { TFileStream * pStream; char szFullPath[MAX_PATH]; // Notify the user pLogger->PrintProgress("Creating empty file %s ...", szPlainName); // Construct the full path and crete the file CreateFullPathName(szFullPath, NULL, szPlainName); pStream = FileStream_CreateFileA(szFullPath, STREAM_PROVIDER_FLAT | BASE_PROVIDER_FILE); if(pStream == NULL) return pLogger->PrintError("Failed to create file %s", szBuffer); // Write the required size FileStream_SetSize(pStream, FileSize); FileStream_Close(pStream); // Give the caller the full file name if(szBuffer != NULL) strcpy(szBuffer, szFullPath); return ERROR_SUCCESS; } static int VerifyFilePosition( TLogHelper * pLogger, TFileStream * pStream, ULONGLONG ExpectedPosition) { ULONGLONG ByteOffset = 0; int nError = ERROR_SUCCESS; // Retrieve the file position if(FileStream_GetPos(pStream, &ByteOffset)) { if(ByteOffset != ExpectedPosition) { pLogger->PrintMessage("The file position is different than expected (expected: " I64u_a ", current: " I64u_a, ExpectedPosition, ByteOffset); nError = ERROR_FILE_CORRUPT; } } else { nError = pLogger->PrintError("Failed to retrieve the file offset"); } return nError; } static int VerifyFileMpqHeader(TLogHelper * pLogger, TFileStream * pStream, ULONGLONG * pByteOffset) { TMPQHeader Header; int nError = ERROR_SUCCESS; memset(&Header, 0xFE, sizeof(TMPQHeader)); if(FileStream_Read(pStream, pByteOffset, &Header, sizeof(TMPQHeader))) { if(Header.dwID != ID_MPQ) { pLogger->PrintMessage("Read error - the data is not a MPQ header"); nError = ERROR_FILE_CORRUPT; } } else { nError = pLogger->PrintError("Failed to read the MPQ header"); } return nError; } static int WriteMpqUserDataHeader( TLogHelper * pLogger, TFileStream * pStream, ULONGLONG ByteOffset, DWORD dwByteCount) { TMPQUserData UserData; int nError = ERROR_SUCCESS; // Notify the user pLogger->PrintProgress("Writing user data header..."); // Fill the user data header UserData.dwID = ID_MPQ_USERDATA; UserData.cbUserDataSize = dwByteCount; UserData.dwHeaderOffs = (dwByteCount + sizeof(TMPQUserData)); UserData.cbUserDataHeader = dwByteCount / 2; if(!FileStream_Write(pStream, &ByteOffset, &UserData, sizeof(TMPQUserData))) nError = GetLastError(); return nError; } static int WriteFileData( TLogHelper * pLogger, TFileStream * pStream, ULONGLONG ByteOffset, ULONGLONG ByteCount) { ULONGLONG SaveByteCount = ByteCount; ULONGLONG BytesWritten = 0; LPBYTE pbDataBuffer; DWORD cbDataBuffer = 0x10000; int nError = ERROR_SUCCESS; // Write some data pbDataBuffer = new BYTE[cbDataBuffer]; if(pbDataBuffer != NULL) { memset(pbDataBuffer, 0, cbDataBuffer); strcpy((char *)pbDataBuffer, "This is a test data written to a file."); // Perform the write while(ByteCount > 0) { DWORD cbToWrite = (ByteCount > cbDataBuffer) ? cbDataBuffer : (DWORD)ByteCount; // Notify the user pLogger->PrintProgress("Writing file data (%I64u of %I64u) ...", BytesWritten, SaveByteCount); // Write the data if(!FileStream_Write(pStream, &ByteOffset, pbDataBuffer, cbToWrite)) { nError = GetLastError(); break; } BytesWritten += cbToWrite; ByteOffset += cbToWrite; ByteCount -= cbToWrite; } delete [] pbDataBuffer; } return nError; } static int CopyFileData( TLogHelper * pLogger, TFileStream * pStream1, TFileStream * pStream2, ULONGLONG ByteOffset, ULONGLONG ByteCount) { ULONGLONG BytesCopied = 0; ULONGLONG EndOffset = ByteOffset + ByteCount; LPBYTE pbCopyBuffer; DWORD BytesToRead; DWORD BlockLength = 0x100000; int nError = ERROR_SUCCESS; // Allocate copy buffer pbCopyBuffer = STORM_ALLOC(BYTE, BlockLength); if(pbCopyBuffer != NULL) { while(ByteOffset < EndOffset) { // Read source BytesToRead = ((EndOffset - ByteOffset) > BlockLength) ? BlockLength : (DWORD)(EndOffset - ByteOffset); if(!FileStream_Read(pStream1, &ByteOffset, pbCopyBuffer, BytesToRead)) { nError = GetLastError(); break; } // Write to the destination file if(!FileStream_Write(pStream2, NULL, pbCopyBuffer, BytesToRead)) { nError = GetLastError(); break; } // Increment the byte counts BytesCopied += BytesToRead; ByteOffset += BytesToRead; // Notify the user pLogger->PrintProgress("Copying (%I64u of %I64u complete) ...", BytesCopied, ByteCount); } STORM_FREE(pbCopyBuffer); } return nError; } // Support function for copying file static int CreateFileCopy( TLogHelper * pLogger, const char * szPlainName, const char * szFileCopy, char * szBuffer, ULONGLONG PreMpqDataSize = 0, ULONGLONG UserDataSize = 0) { TFileStream * pStream1; // Source file TFileStream * pStream2; // Target file ULONGLONG ByteOffset = 0; ULONGLONG FileSize = 0; char szFileName1[MAX_PATH]; char szFileName2[MAX_PATH]; int nError = ERROR_SUCCESS; // Notify the user szPlainName += FileStream_PrefixA(szPlainName, NULL); pLogger->PrintProgress("Creating copy of %s ...", szPlainName); // Construct both file names. Check if they are not the same CreateFullPathName(szFileName1, szMpqSubDir, szPlainName); CreateFullPathName(szFileName2, NULL, szFileCopy + FileStream_PrefixA(szFileCopy, NULL)); if(!_stricmp(szFileName1, szFileName2)) { pLogger->PrintError("Failed to create copy of MPQ (the copy name is the same like the original name)"); return ERROR_CAN_NOT_COMPLETE; } // Open the source file pStream1 = FileStream_OpenFileA(szFileName1, STREAM_FLAG_READ_ONLY); if(pStream1 == NULL) { pLogger->PrintError("Failed to open the source file %s", szFileName1); return ERROR_CAN_NOT_COMPLETE; } // Create the destination file pStream2 = FileStream_CreateFileA(szFileName2, 0); if(pStream2 != NULL) { // If we should write some pre-MPQ data to the target file, do it if(PreMpqDataSize != 0) { nError = WriteFileData(pLogger, pStream2, ByteOffset, PreMpqDataSize); ByteOffset += PreMpqDataSize; } // If we should write some MPQ user data, write the header first if(UserDataSize != 0) { nError = WriteMpqUserDataHeader(pLogger, pStream2, ByteOffset, (DWORD)UserDataSize); ByteOffset += sizeof(TMPQUserData); nError = WriteFileData(pLogger, pStream2, ByteOffset, UserDataSize); ByteOffset += UserDataSize; } // Copy the file data from the source file to the destination file FileStream_GetSize(pStream1, &FileSize); if(FileSize != 0) { nError = CopyFileData(pLogger, pStream1, pStream2, 0, FileSize); ByteOffset += FileSize; } FileStream_Close(pStream2); } // Close the source file FileStream_Close(pStream1); if(szBuffer != NULL) CreateFullPathName(szBuffer, NULL, szFileCopy); if(nError != ERROR_SUCCESS) pLogger->PrintError("Failed to create copy of MPQ"); return nError; } static int CreateMasterAndMirrorPaths( TLogHelper * pLogger, char * szMirrorPath, char * szMasterPath, const char * szMirrorName, const char * szMasterName, bool bCopyMirrorFile) { char szCopyPath[MAX_PATH]; int nError = ERROR_SUCCESS; // Always delete the mirror file CreateFullPathName(szMasterPath, szMpqSubDir, szMasterName); CreateFullPathName(szCopyPath, NULL, szMirrorName); remove(szCopyPath + FileStream_PrefixA(szCopyPath, NULL)); // Copy the mirrored file from the source to the work directory if(bCopyMirrorFile) nError = CreateFileCopy(pLogger, szMirrorName, szMirrorName, NULL); // Create the mirror*master path if(nError == ERROR_SUCCESS) sprintf(szMirrorPath, "%s*%s", szCopyPath, szMasterPath); return nError; } static void WINAPI AddFileCallback(void * pvUserData, DWORD dwBytesWritten, DWORD dwTotalBytes, bool bFinalCall) { TLogHelper * pLogger = (TLogHelper *)pvUserData; // Keep compiler happy bFinalCall = bFinalCall; pLogger->PrintProgress("Adding file (%s) (%u of %u) (%u of %u) ...", pLogger->UserString, pLogger->UserCount, pLogger->UserTotal, dwBytesWritten, dwTotalBytes); } static void WINAPI CompactCallback(void * pvUserData, DWORD dwWork, ULONGLONG BytesDone, ULONGLONG TotalBytes) { TLogHelper * pLogger = (TLogHelper *)pvUserData; const char * szWork = NULL; switch(dwWork) { case CCB_CHECKING_FILES: szWork = "Checking files in archive"; break; case CCB_CHECKING_HASH_TABLE: szWork = "Checking hash table"; break; case CCB_COPYING_NON_MPQ_DATA: szWork = "Copying non-MPQ data"; break; case CCB_COMPACTING_FILES: szWork = "Compacting files"; break; case CCB_CLOSING_ARCHIVE: szWork = "Closing archive"; break; } if(szWork != NULL) { if(pLogger != NULL) pLogger->PrintProgress("%s (%I64u of %I64u) ...", szWork, BytesDone, TotalBytes); else printf("%s (" I64u_a " of " I64u_a ") ... \r", szWork, BytesDone, TotalBytes); } } //----------------------------------------------------------------------------- // MPQ file utilities #define TEST_FLAG_LOAD_FILES 0x00000001 // Test function should load all files in the MPQ #define TEST_FLAG_HASH_FILES 0x00000002 // Test function should load all files in the MPQ #define TEST_FLAG_PLAY_WAVES 0x00000004 // Play extracted WAVE files #define TEST_FLAG_MOST_PATCHED 0x00000008 // Find the most patched file struct TFileData { DWORD dwBlockIndex; DWORD dwFileSize; DWORD dwFlags; DWORD dwReserved; // Alignment BYTE FileData[1]; }; static bool CheckIfFileIsPresent(TLogHelper * pLogger, HANDLE hMpq, const char * szFileName, bool bShouldExist) { HANDLE hFile = NULL; if(SFileOpenFileEx(hMpq, szFileName, 0, &hFile)) { if(bShouldExist == false) pLogger->PrintMessage("The file %s is present, but it should not be", szFileName); SFileCloseFile(hFile); return true; } else { if(bShouldExist) pLogger->PrintMessage("The file %s is not present, but it should be", szFileName); return false; } } static TFileData * LoadLocalFile(TLogHelper * pLogger, const char * szFileName, bool bMustSucceed) { TFileStream * pStream; TFileData * pFileData = NULL; ULONGLONG FileSize = 0; size_t nAllocateBytes; // Notify the user if(pLogger != NULL) pLogger->PrintProgress("Loading local file ..."); // Attempt to open the file pStream = FileStream_OpenFileA(szFileName, STREAM_FLAG_READ_ONLY); if(pStream == NULL) { if(pLogger != NULL && bMustSucceed == true) pLogger->PrintError("Failed to open the file %s", szFileName); return NULL; } // Verify the size FileStream_GetSize(pStream, &FileSize); if((FileSize >> 0x20) == 0) { // Allocate space for the file nAllocateBytes = sizeof(TFileData) + (size_t)FileSize; pFileData = (TFileData *)STORM_ALLOC(BYTE, nAllocateBytes); if(pFileData != NULL) { // Make sure it;s properly zeroed memset(pFileData, 0, nAllocateBytes); pFileData->dwFileSize = (DWORD)FileSize; // Load to memory if(!FileStream_Read(pStream, NULL, pFileData->FileData, pFileData->dwFileSize)) { STORM_FREE(pFileData); pFileData = NULL; } } } FileStream_Close(pStream); return pFileData; } static int CompareTwoLocalFilesRR( TLogHelper * pLogger, TFileStream * pStream1, // Master file TFileStream * pStream2, // Mirror file int nIterations) // Number of iterations { ULONGLONG RandomNumber = 0x12345678; // We need pseudo-random number that will repeat each run of the program ULONGLONG RandomSeed; ULONGLONG ByteOffset; ULONGLONG FileSize1 = 1; ULONGLONG FileSize2 = 2; DWORD BytesToRead; DWORD Difference; LPBYTE pbBuffer1; LPBYTE pbBuffer2; DWORD cbBuffer = 0x100000; int nError = ERROR_SUCCESS; // Compare file sizes FileStream_GetSize(pStream1, &FileSize1); FileStream_GetSize(pStream2, &FileSize2); if(FileSize1 != FileSize2) { pLogger->PrintMessage("The files have different size"); return ERROR_CAN_NOT_COMPLETE; } // Allocate both buffers pbBuffer1 = STORM_ALLOC(BYTE, cbBuffer); pbBuffer2 = STORM_ALLOC(BYTE, cbBuffer); if(pbBuffer1 && pbBuffer2) { // Perform many random reads for(int i = 0; i < nIterations; i++) { // Generate psudo-random offsrt and data size ByteOffset = (RandomNumber % FileSize1); BytesToRead = (DWORD)(RandomNumber % cbBuffer); // Show the progress message pLogger->PrintProgress("Comparing file: Offset: " I64u_a ", Length: %u", ByteOffset, BytesToRead); // Only perform read if the byte offset is below if(ByteOffset < FileSize1) { if((ByteOffset + BytesToRead) > FileSize1) BytesToRead = (DWORD)(FileSize1 - ByteOffset); memset(pbBuffer1, 0xEE, cbBuffer); memset(pbBuffer2, 0xAA, cbBuffer); FileStream_Read(pStream1, &ByteOffset, pbBuffer1, BytesToRead); FileStream_Read(pStream2, &ByteOffset, pbBuffer2, BytesToRead); if(!CompareBlocks(pbBuffer1, pbBuffer2, BytesToRead, &Difference)) { pLogger->PrintMessage("Difference at %u (Offset " I64X_a ", Length %X)", Difference, ByteOffset, BytesToRead); nError = ERROR_FILE_CORRUPT; break; } // Shuffle the random number memcpy(&RandomSeed, pbBuffer1, sizeof(RandomSeed)); RandomNumber = ((RandomNumber >> 0x11) | (RandomNumber << 0x29)) ^ (RandomNumber + RandomSeed); } } } // Free both buffers if(pbBuffer2 != NULL) STORM_FREE(pbBuffer2); if(pbBuffer1 != NULL) STORM_FREE(pbBuffer1); return nError; } static TFileData * LoadMpqFile(TLogHelper * pLogger, HANDLE hMpq, const char * szFileName) { TFileData * pFileData = NULL; HANDLE hFile; DWORD dwFileSizeHi = 0xCCCCCCCC; DWORD dwFileSizeLo = 0; DWORD dwBytesRead; int nError = ERROR_SUCCESS; // Notify the user that we are loading a file from MPQ pLogger->PrintProgress("Loading file %s ...", GetShortPlainName(szFileName)); // Open the file from MPQ if(!SFileOpenFileEx(hMpq, szFileName, 0, &hFile)) nError = pLogger->PrintError("Failed to open the file %s", szFileName); // Get the size of the file if(nError == ERROR_SUCCESS) { dwFileSizeLo = SFileGetFileSize(hFile, &dwFileSizeHi); if(dwFileSizeLo == SFILE_INVALID_SIZE || dwFileSizeHi != 0) nError = pLogger->PrintError("Failed to query the file size"); } // Spazzler protector: Creates fake files with size of 0x7FFFE7CA if(nError == ERROR_SUCCESS) { if(dwFileSizeLo > 0x1FFFFFFF) nError = ERROR_FILE_CORRUPT; } // Allocate buffer for the file content if(nError == ERROR_SUCCESS) { pFileData = (TFileData *)STORM_ALLOC(BYTE, sizeof(TFileData) + dwFileSizeLo); if(pFileData == NULL) { pLogger->PrintError("Failed to allocate buffer for the file content"); nError = ERROR_NOT_ENOUGH_MEMORY; } } // get the file index of the MPQ file if(nError == ERROR_SUCCESS) { // Store the file size memset(pFileData, 0, sizeof(TFileData) + dwFileSizeLo); pFileData->dwFileSize = dwFileSizeLo; // Retrieve the block index and file flags if(!SFileGetFileInfo(hFile, SFileInfoFileIndex, &pFileData->dwBlockIndex, sizeof(DWORD), NULL)) nError = pLogger->PrintError("Failed retrieve the file index of %s", szFileName); if(!SFileGetFileInfo(hFile, SFileInfoFlags, &pFileData->dwFlags, sizeof(DWORD), NULL)) nError = pLogger->PrintError("Failed retrieve the file flags of %s", szFileName); } // Load the entire file if(nError == ERROR_SUCCESS) { // Read the file data SFileReadFile(hFile, pFileData->FileData, dwFileSizeLo, &dwBytesRead, NULL); if(dwBytesRead != dwFileSizeLo) nError = pLogger->PrintError("Failed to read the content of the file %s", szFileName); } // If failed, free the buffer if(nError != ERROR_SUCCESS) { STORM_FREE(pFileData); SetLastError(nError); pFileData = NULL; } // Close the file and return what we got if(hFile != NULL) SFileCloseFile(hFile); return pFileData; } static bool CompareTwoFiles(TLogHelper * pLogger, TFileData * pFileData1, TFileData * pFileData2) { // Compare the file size if(pFileData1->dwFileSize != pFileData2->dwFileSize) { pLogger->PrintErrorVa(_T("The files have different size (%u vs %u)"), pFileData1->dwFileSize, pFileData2->dwFileSize); SetLastError(ERROR_FILE_CORRUPT); return false; } // Compare the files for(DWORD i = 0; i < pFileData1->dwFileSize; i++) { if(pFileData1->FileData[i] != pFileData2->FileData[i]) { pLogger->PrintErrorVa(_T("Files are different at offset %08X"), i); SetLastError(ERROR_FILE_CORRUPT); return false; } } // The files are identical return true; } static int SearchArchive( TLogHelper * pLogger, HANDLE hMpq, DWORD dwTestFlags = 0, DWORD * pdwFileCount = NULL, LPBYTE pbFileHash = NULL) { SFILE_FIND_DATA sf; TFileData * pFileData; HANDLE hFind; DWORD dwFileCount = 0; hash_state md5state; char szMostPatched[MAX_PATH] = ""; char szListFile[MAX_PATH]; bool bFound = true; int nMaxPatchCount = 0; int nPatchCount = 0; int nError = ERROR_SUCCESS; // Construct the full name of the listfile CreateFullPathName(szListFile, szMpqSubDir, "ListFile_Blizzard.txt"); // Prepare hashing md5_init(&md5state); // Initiate the MPQ search pLogger->PrintProgress("Searching the archive ..."); hFind = SFileFindFirstFile(hMpq, "*", &sf, szListFile); if(hFind == NULL) { nError = GetLastError(); nError = (nError == ERROR_NO_MORE_FILES) ? ERROR_SUCCESS : nError; return nError; } // Perform the search while(bFound == true) { // Increment number of files dwFileCount++; // if(!_stricmp(sf.cFileName, "OldWorld\\world\\maps\\Northrend\\Northrend.tex")) // DebugBreak(); if(dwTestFlags & TEST_FLAG_MOST_PATCHED) { // Load the patch count nPatchCount = GetFilePatchCount(pLogger, hMpq, sf.cFileName); // Check if it's greater than maximum if(nPatchCount > nMaxPatchCount) { strcpy(szMostPatched, sf.cFileName); nMaxPatchCount = nPatchCount; } } // Load the file to memory, if required if(dwTestFlags & TEST_FLAG_LOAD_FILES) { // Load the entire file to the MPQ pFileData = LoadMpqFile(pLogger, hMpq, sf.cFileName); if(pFileData != NULL) { // Hash the file data, if needed if((dwTestFlags & TEST_FLAG_HASH_FILES) && !IsInternalMpqFileName(sf.cFileName)) md5_process(&md5state, pFileData->FileData, pFileData->dwFileSize); // Play sound files, if required if((dwTestFlags & TEST_FLAG_PLAY_WAVES) && strstr(sf.cFileName, ".wav") != NULL) { #ifdef _MSC_VER pLogger->PrintProgress("Playing sound %s", sf.cFileName); PlaySound((LPCTSTR)pFileData->FileData, NULL, SND_MEMORY); #endif } STORM_FREE(pFileData); } } bFound = SFileFindNextFile(hFind, &sf); } SFileFindClose(hFind); // Give the file count, if required if(pdwFileCount != NULL) pdwFileCount[0] = dwFileCount; // Give the hash, if required if(pbFileHash != NULL && (dwTestFlags & TEST_FLAG_HASH_FILES)) md5_done(&md5state, pbFileHash); return nError; } static int CreateNewArchive(TLogHelper * pLogger, const char * szPlainName, DWORD dwCreateFlags, DWORD dwMaxFileCount, HANDLE * phMpq) { HANDLE hMpq = NULL; TCHAR szMpqName[MAX_PATH]; char szFullPath[MAX_PATH]; // Make sure that the MPQ is deleted CreateFullPathName(szFullPath, NULL, szPlainName); remove(szFullPath); // Create the new MPQ CopyFileName(szMpqName, szFullPath, strlen(szFullPath)); if(!SFileCreateArchive(szMpqName, dwCreateFlags, dwMaxFileCount, &hMpq)) return pLogger->PrintError(_T("Failed to create archive %s"), szMpqName); // Shall we close it right away? if(phMpq == NULL) SFileCloseArchive(hMpq); else *phMpq = hMpq; return ERROR_SUCCESS; } static int CreateNewArchive_V2(TLogHelper * pLogger, const char * szPlainName, DWORD dwCreateFlags, DWORD dwMaxFileCount, HANDLE * phMpq) { SFILE_CREATE_MPQ CreateInfo; HANDLE hMpq = NULL; TCHAR szMpqName[MAX_PATH]; char szFullPath[MAX_PATH]; // Make sure that the MPQ is deleted CreateFullPathName(szFullPath, NULL, szPlainName); CopyFileName(szMpqName, szFullPath, strlen(szFullPath)); remove(szFullPath); // Fill the create structure memset(&CreateInfo, 0, sizeof(SFILE_CREATE_MPQ)); CreateInfo.cbSize = sizeof(SFILE_CREATE_MPQ); CreateInfo.dwMpqVersion = (dwCreateFlags & MPQ_CREATE_ARCHIVE_VMASK) >> FLAGS_TO_FORMAT_SHIFT; CreateInfo.dwStreamFlags = STREAM_PROVIDER_FLAT | BASE_PROVIDER_FILE; CreateInfo.dwFileFlags1 = (dwCreateFlags & MPQ_CREATE_LISTFILE) ? MPQ_FILE_EXISTS : 0; CreateInfo.dwFileFlags2 = (dwCreateFlags & MPQ_CREATE_ATTRIBUTES) ? MPQ_FILE_EXISTS : 0; CreateInfo.dwFileFlags3 = (dwCreateFlags & MPQ_CREATE_SIGNATURE) ? MPQ_FILE_EXISTS : 0; CreateInfo.dwAttrFlags = (dwCreateFlags & MPQ_CREATE_ATTRIBUTES) ? (MPQ_ATTRIBUTE_CRC32 | MPQ_ATTRIBUTE_FILETIME | MPQ_ATTRIBUTE_MD5) : 0; CreateInfo.dwSectorSize = (CreateInfo.dwMpqVersion >= MPQ_FORMAT_VERSION_3) ? 0x4000 : 0x1000; CreateInfo.dwRawChunkSize = (CreateInfo.dwMpqVersion >= MPQ_FORMAT_VERSION_4) ? 0x4000 : 0; CreateInfo.dwMaxFileCount = dwMaxFileCount; // Create the new MPQ if(!SFileCreateArchive2(szMpqName, &CreateInfo, &hMpq)) return pLogger->PrintError(_T("Failed to create archive %s"), szMpqName); // Shall we close it right away? if(phMpq == NULL) SFileCloseArchive(hMpq); else *phMpq = hMpq; return ERROR_SUCCESS; } // Creates new archive with UNICODE name. Adds prefix to the name static int CreateNewArchiveU(TLogHelper * pLogger, const wchar_t * szPlainName, DWORD dwCreateFlags, DWORD dwMaxFileCount) { #ifdef _UNICODE HANDLE hMpq = NULL; TCHAR szMpqName[MAX_PATH]; char szFullPath[MAX_PATH]; // Construct the full UNICODE name CreateFullPathName(szFullPath, NULL, "StormLibTest_"); CopyFileName(szMpqName, szFullPath, strlen(szFullPath)); wcscat(szMpqName, szPlainName); // Make sure that the MPQ is deleted _tremove(szMpqName); // Create the archive pLogger->PrintProgress("Creating new archive with UNICODE name ..."); if(!SFileCreateArchive(szMpqName, dwCreateFlags, dwMaxFileCount, &hMpq)) return pLogger->PrintError(_T("Failed to create archive %s"), szMpqName); SFileCloseArchive(hMpq); #else pLogger = pLogger; szPlainName = szPlainName; dwCreateFlags = dwCreateFlags; dwMaxFileCount = dwMaxFileCount; #endif return ERROR_SUCCESS; } static int OpenExistingArchive(TLogHelper * pLogger, const char * szFullPath, DWORD dwFlags, HANDLE * phMpq) { HANDLE hMpq = NULL; TCHAR szMpqName[MAX_PATH]; int nError = ERROR_SUCCESS; // Is it an encrypted MPQ ? if(strstr(szFullPath, ".MPQE") != NULL) dwFlags |= STREAM_PROVIDER_MPQE; if(strstr(szFullPath, ".MPQ.part") != NULL) dwFlags |= STREAM_PROVIDER_PARTIAL; if(strstr(szFullPath, ".mpq.part") != NULL) dwFlags |= STREAM_PROVIDER_PARTIAL; if(strstr(szFullPath, ".MPQ.0") != NULL) dwFlags |= STREAM_PROVIDER_BLOCK4; // Open the copied archive pLogger->PrintProgress("Opening archive %s ...", GetShortPlainName(szFullPath)); CopyFileName(szMpqName, szFullPath, strlen(szFullPath)); if(!SFileOpenArchive(szMpqName, 0, dwFlags, &hMpq)) { // Ignore the error if it's an AVI file or if the file is incomplete nError = GetLastError(); if(nError == ERROR_AVI_FILE || nError == ERROR_FILE_INCOMPLETE) return nError; // Show the open error to the user return pLogger->PrintError("Failed to open archive %s", szFullPath); } // Store the archive handle or close the archive if(phMpq == NULL) SFileCloseArchive(hMpq); else *phMpq = hMpq; return nError; } static int OpenPatchArchive(TLogHelper * pLogger, HANDLE hMpq, const char * szFullPath) { TCHAR szPatchName[MAX_PATH]; int nError = ERROR_SUCCESS; pLogger->PrintProgress("Adding patch %s ...", GetShortPlainName(szFullPath)); CopyFileName(szPatchName, szFullPath, strlen(szFullPath)); if(!SFileOpenPatchArchive(hMpq, szPatchName, NULL, 0)) nError = pLogger->PrintError("Failed to add patch %s ...", szFullPath); return nError; } static int OpenExistingArchiveWithCopy(TLogHelper * pLogger, const char * szFileName, const char * szCopyName, HANDLE * phMpq) { DWORD dwFlags = 0; char szFullPath[MAX_PATH]; int nError = ERROR_SUCCESS; // We expect MPQ directory to be already prepared by InitializeMpqDirectory assert(szMpqDirectory[0] != 0); // At least one name must be entered assert(szFileName != NULL || szCopyName != NULL); // If both names entered, create a copy if(szFileName != NULL && szCopyName != NULL) { nError = CreateFileCopy(pLogger, szFileName, szCopyName, szFullPath); if(nError != ERROR_SUCCESS) return nError; } // If only source name entered, open it for read-only access else if(szFileName != NULL && szCopyName == NULL) { CreateFullPathName(szFullPath, szMpqSubDir, szFileName); dwFlags |= MPQ_OPEN_READ_ONLY; } // If only target name entered, open it directly else if(szFileName == NULL && szCopyName != NULL) { CreateFullPathName(szFullPath, NULL, szCopyName); } // Open the archive return OpenExistingArchive(pLogger, szFullPath, dwFlags, phMpq); } static int OpenPatchedArchive(TLogHelper * pLogger, HANDLE * phMpq, const char * PatchList[]) { HANDLE hMpq = NULL; char szFullPath[MAX_PATH]; int nError = ERROR_SUCCESS; // The first file is expected to be valid assert(PatchList[0] != NULL); // Open the primary MPQ CreateFullPathName(szFullPath, szMpqSubDir, PatchList[0]); nError = OpenExistingArchive(pLogger, szFullPath, MPQ_OPEN_READ_ONLY, &hMpq); // Add all patches if(nError == ERROR_SUCCESS) { for(size_t i = 1; PatchList[i] != NULL; i++) { CreateFullPathName(szFullPath, szMpqPatchDir, PatchList[i]); nError = OpenPatchArchive(pLogger, hMpq, szFullPath); if(nError != ERROR_SUCCESS) break; } } // Store the archive handle or close the archive if(phMpq == NULL) SFileCloseArchive(hMpq); else *phMpq = hMpq; return nError; } static int AddFileToMpq( TLogHelper * pLogger, HANDLE hMpq, const char * szFileName, const char * szFileData, DWORD dwFlags = 0, DWORD dwCompression = 0, bool bMustSucceed = false) { HANDLE hFile = NULL; DWORD dwFileSize = (DWORD)strlen(szFileData); int nError = ERROR_SUCCESS; // Notify the user pLogger->PrintProgress("Adding file %s ...", szFileName); // Get the default flags if(dwFlags == 0) dwFlags = MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED; if(dwCompression == 0) dwCompression = MPQ_COMPRESSION_ZLIB; // Create the file within the MPQ if(!SFileCreateFile(hMpq, szFileName, 0, dwFileSize, 0, dwFlags, &hFile)) { // If success is not expected, it is actually a good thing if(bMustSucceed == true) return pLogger->PrintError("Failed to create MPQ file %s", szFileName); return GetLastError(); } // Write the file if(!SFileWriteFile(hFile, szFileData, dwFileSize, dwCompression)) nError = pLogger->PrintError("Failed to write data to the MPQ"); SFileCloseFile(hFile); return nError; } static int AddLocalFileToMpq( TLogHelper * pLogger, HANDLE hMpq, const char * szArchivedName, const char * szLocalFileName, DWORD dwFlags = 0, DWORD dwCompression = 0, bool bMustSucceed = false) { TCHAR szFileName[MAX_PATH]; DWORD dwVerifyResult; // Notify the user pLogger->PrintProgress("Adding file %s (%u of %u)...", GetShortPlainName(szLocalFileName), pLogger->UserCount, pLogger->UserTotal); pLogger->UserString = szArchivedName; // Get the default flags if(dwFlags == 0) dwFlags = MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED; if(dwCompression == 0) dwCompression = MPQ_COMPRESSION_ZLIB; // Set the notification callback SFileSetAddFileCallback(hMpq, AddFileCallback, pLogger); // Add the file to the MPQ CopyFileName(szFileName, szLocalFileName, strlen(szLocalFileName)); if(!SFileAddFileEx(hMpq, szFileName, szArchivedName, dwFlags, dwCompression, MPQ_COMPRESSION_NEXT_SAME)) { if(bMustSucceed) return pLogger->PrintError("Failed to add the file %s", szArchivedName); return GetLastError(); } // Verify the file unless it was lossy compression if((dwCompression & (MPQ_COMPRESSION_ADPCM_MONO | MPQ_COMPRESSION_ADPCM_STEREO)) == 0) { // Notify the user pLogger->PrintProgress("Verifying file %s (%u of %u) ...", szArchivedName, pLogger->UserCount, pLogger->UserTotal); // Perform the verification dwVerifyResult = SFileVerifyFile(hMpq, szArchivedName, MPQ_ATTRIBUTE_CRC32 | MPQ_ATTRIBUTE_MD5); if(dwVerifyResult & (VERIFY_OPEN_ERROR | VERIFY_READ_ERROR | VERIFY_FILE_SECTOR_CRC_ERROR | VERIFY_FILE_CHECKSUM_ERROR | VERIFY_FILE_MD5_ERROR)) return pLogger->PrintError("CRC error on %s", szArchivedName); } return ERROR_SUCCESS; } static int RenameMpqFile(TLogHelper * pLogger, HANDLE hMpq, const char * szOldFileName, const char * szNewFileName, bool bMustSucceed) { // Notify the user pLogger->PrintProgress("Renaming %s to %s ...", szOldFileName, szNewFileName); // Perform the deletion if(!SFileRenameFile(hMpq, szOldFileName, szNewFileName)) { if(bMustSucceed == true) return pLogger->PrintErrorVa("Failed to rename %s to %s", szOldFileName, szNewFileName); return GetLastError(); } return ERROR_SUCCESS; } static int RemoveMpqFile(TLogHelper * pLogger, HANDLE hMpq, const char * szFileName, bool bMustSucceed) { // Notify the user pLogger->PrintProgress("Removing file %s ...", szFileName); // Perform the deletion if(!SFileRemoveFile(hMpq, szFileName, 0)) { if(bMustSucceed == true) return pLogger->PrintError("Failed to remove the file %s from the archive", szFileName); return GetLastError(); } return ERROR_SUCCESS; } //----------------------------------------------------------------------------- // Tests static void TestGetFileInfo( TLogHelper * pLogger, HANDLE hMpqOrFile, SFileInfoClass InfoClass, void * pvFileInfo, DWORD cbFileInfo, DWORD * pcbLengthNeeded, bool bExpectedResult, int nExpectedError) { bool bResult; int nError = ERROR_SUCCESS; // Call the get file info bResult = SFileGetFileInfo(hMpqOrFile, InfoClass, pvFileInfo, cbFileInfo, pcbLengthNeeded); if(!bResult) nError = GetLastError(); if(bResult != bExpectedResult) pLogger->PrintMessage("Different result of SFileGetFileInfo."); if(nError != nExpectedError) pLogger->PrintMessage("Different error from SFileGetFileInfo (expected %u, returned %u)", nExpectedError, nError); } // StormLib is able to open local files (as well as the original Storm.dll) // I want to keep this for occasional use static int TestOpenLocalFile(const char * szPlainName) { TLogHelper Logger("OpenLocalFile", szPlainName); HANDLE hFile; DWORD dwFileSizeHi = 0; DWORD dwFileSizeLo = 0; char szFileName1[MAX_PATH]; char szFileName2[MAX_PATH]; char szFileLine[0x40]; CreateFullPathName(szFileName1, szMpqSubDir, szPlainName); if(SFileOpenFileEx(NULL, szFileName1, SFILE_OPEN_LOCAL_FILE, &hFile)) { // Retrieve the file name. It must match the name under which the file was open SFileGetFileName(hFile, szFileName2); if(strcmp(szFileName2, szFileName1)) Logger.PrintMessage("The retrieved name does not match the open name"); // Retrieve the file size dwFileSizeLo = SFileGetFileSize(hFile, &dwFileSizeHi); if(dwFileSizeHi != 0 || dwFileSizeLo != 3904784) Logger.PrintMessage("Local file size mismatch"); // Read the first line memset(szFileLine, 0, sizeof(szFileLine)); SFileReadFile(hFile, szFileLine, 18, NULL, NULL); if(strcmp(szFileLine, "(1)Enslavers01.scm")) Logger.PrintMessage("Content of the listfile does not match"); SFileCloseFile(hFile); } return ERROR_SUCCESS; } static int TestSearchListFile(const char * szPlainName) { SFILE_FIND_DATA sf; TLogHelper Logger("SearchListFile", szPlainName); HANDLE hFind; char szFullPath[MAX_PATH]; int nFileCount = 0; CreateFullPathName(szFullPath, szMpqSubDir, szPlainName); hFind = SListFileFindFirstFile(NULL, szFullPath, "*", &sf); if(hFind != NULL) { for(;;) { Logger.PrintProgress("Found file (%04u): %s", nFileCount++, GetShortPlainName(sf.cFileName)); if(!SListFileFindNextFile(hFind, &sf)) break; } SListFileFindClose(hFind); } return ERROR_SUCCESS; } static void WINAPI TestReadFile_DownloadCallback( void * UserData, ULONGLONG ByteOffset, DWORD DataLength) { TLogHelper * pLogger = (TLogHelper *)UserData; if(ByteOffset != 0 && DataLength != 0) pLogger->PrintProgress("Downloading data (offset: " I64X_a ", length: %X)", ByteOffset, DataLength); else pLogger->PrintProgress("Download complete."); } // Open a file stream with mirroring a master file static int TestReadFile_MasterMirror(const char * szMirrorName, const char * szMasterName, bool bCopyMirrorFile) { TFileStream * pStream1; // Master file TFileStream * pStream2; // Mirror file TLogHelper Logger("OpenMirrorFile", szMirrorName); DWORD dwProvider = 0; char szMirrorPath[MAX_PATH + MAX_PATH]; char szMasterPath[MAX_PATH]; int nIterations = 0x10000; int nError; // Retrieve the provider FileStream_PrefixA(szMasterName, &dwProvider); #ifndef PLATFORM_WINDOWS if((dwProvider & BASE_PROVIDER_MASK) == BASE_PROVIDER_HTTP) return ERROR_SUCCESS; #endif // Create copy of the file to serve as mirror, keep master there nError = CreateMasterAndMirrorPaths(&Logger, szMirrorPath, szMasterPath, szMirrorName, szMasterName, bCopyMirrorFile); if(nError == ERROR_SUCCESS) { // Open both master and mirror file pStream1 = FileStream_OpenFileA(szMasterPath, STREAM_FLAG_READ_ONLY); pStream2 = FileStream_OpenFileA(szMirrorPath, STREAM_FLAG_READ_ONLY | STREAM_FLAG_USE_BITMAP); if(pStream1 && pStream2) { // For internet based files, we limit the number of operations if((dwProvider & BASE_PROVIDER_MASK) == BASE_PROVIDER_HTTP) nIterations = 0x80; FileStream_SetCallback(pStream2, TestReadFile_DownloadCallback, &Logger); nError = CompareTwoLocalFilesRR(&Logger, pStream1, pStream2, nIterations); } if(pStream2 != NULL) FileStream_Close(pStream2); if(pStream1 != NULL) FileStream_Close(pStream1); } return nError; } // Test of the TFileStream object static int TestFileStreamOperations(const char * szPlainName, DWORD dwStreamFlags) { TFileStream * pStream = NULL; TLogHelper Logger("FileStreamTest", szPlainName); ULONGLONG ByteOffset; ULONGLONG FileSize = 0; DWORD dwRequiredFlags = 0; char szFullPath[MAX_PATH]; BYTE Buffer[0x10]; int nError = ERROR_SUCCESS; // Copy the file so we won't screw up if((dwStreamFlags & STREAM_PROVIDER_MASK) == STREAM_PROVIDER_BLOCK4) CreateFullPathName(szFullPath, szMpqSubDir, szPlainName); else nError = CreateFileCopy(&Logger, szPlainName, szPlainName, szFullPath); // Open the file stream if(nError == ERROR_SUCCESS) { pStream = FileStream_OpenFileA(szFullPath, dwStreamFlags); if(pStream == NULL) nError = Logger.PrintError("Failed to open %s", szFullPath); } // Get the size of the file stream if(nError == ERROR_SUCCESS) { if(!FileStream_GetFlags(pStream, &dwStreamFlags)) nError = Logger.PrintError("Failed to retrieve the stream flags"); if(!FileStream_GetSize(pStream, &FileSize)) nError = Logger.PrintError("Failed to retrieve the file size"); // Any other stream except STREAM_PROVIDER_FLAT | BASE_PROVIDER_FILE should be read-only if((dwStreamFlags & STREAM_PROVIDERS_MASK) != (STREAM_PROVIDER_FLAT | BASE_PROVIDER_FILE)) dwRequiredFlags |= STREAM_FLAG_READ_ONLY; // if(pStream->BlockPresent) // dwRequiredFlags |= STREAM_FLAG_READ_ONLY; // Check the flags there if((dwStreamFlags & dwRequiredFlags) != dwRequiredFlags) { Logger.PrintMessage("The stream should be read-only but it isn't"); nError = ERROR_FILE_CORRUPT; } } // After successful open, the stream position must be zero if(nError == ERROR_SUCCESS) nError = VerifyFilePosition(&Logger, pStream, 0); // Read the MPQ header from the current file offset. if(nError == ERROR_SUCCESS) nError = VerifyFileMpqHeader(&Logger, pStream, NULL); // After successful open, the stream position must sizeof(TMPQHeader) if(nError == ERROR_SUCCESS) nError = VerifyFilePosition(&Logger, pStream, sizeof(TMPQHeader)); // Now try to read the MPQ header from the offset 0 if(nError == ERROR_SUCCESS) { ByteOffset = 0; nError = VerifyFileMpqHeader(&Logger, pStream, &ByteOffset); } // After successful open, the stream position must sizeof(TMPQHeader) if(nError == ERROR_SUCCESS) nError = VerifyFilePosition(&Logger, pStream, sizeof(TMPQHeader)); // Try a write operation if(nError == ERROR_SUCCESS) { bool bExpectedResult = (dwStreamFlags & STREAM_FLAG_READ_ONLY) ? false : true; bool bResult; // Attempt to write to the file ByteOffset = 0; bResult = FileStream_Write(pStream, &ByteOffset, Buffer, sizeof(Buffer)); // If the result is not expected if(bResult != bExpectedResult) { Logger.PrintMessage("FileStream_Write result is different than expected"); nError = ERROR_FILE_CORRUPT; } } // Move the position 9 bytes from the end and try to read 10 bytes. // This must fail, because stream reading functions are "all or nothing" if(nError == ERROR_SUCCESS) { ByteOffset = FileSize - 9; if(FileStream_Read(pStream, &ByteOffset, Buffer, 10)) { Logger.PrintMessage("FileStream_Read succeeded, but it shouldn't"); nError = ERROR_FILE_CORRUPT; } } // Try again with 9 bytes. This must succeed, unless the file block is not available if(nError == ERROR_SUCCESS) { ByteOffset = FileSize - 9; if(!FileStream_Read(pStream, &ByteOffset, Buffer, 9)) { Logger.PrintMessage("FileStream_Read from the end of the file failed"); nError = ERROR_FILE_CORRUPT; } } // Verify file position - it must be at the end of the file if(nError == ERROR_SUCCESS) nError = VerifyFilePosition(&Logger, pStream, FileSize); // Close the stream if(pStream != NULL) FileStream_Close(pStream); return nError; } static int TestOpenFile_OpenById(const char * szPlainName) { TLogHelper Logger("OpenFileById", szPlainName); TFileData * pFileData1 = NULL; TFileData * pFileData2 = NULL; HANDLE hMpq; int nError; // Copy the archive so we won't fuck up the original one nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, NULL, &hMpq); // Now try to open a file without knowing the file name if(nError == ERROR_SUCCESS) { // File00000023.xxx = music\dintro.wav pFileData1 = LoadMpqFile(&Logger, hMpq, "File00000023.xxx"); if(pFileData1 == NULL) nError = Logger.PrintError("Failed to load the file %s", "File00000023.xxx"); } // Now try to open the file again with its original name if(nError == ERROR_SUCCESS) { // File00000023.xxx = music\dintro.wav pFileData2 = LoadMpqFile(&Logger, hMpq, "music\\dintro.wav"); if(pFileData2 == NULL) nError = Logger.PrintError("Failed to load the file %s", "music\\dintro.wav"); } // Now compare both files if(nError == ERROR_SUCCESS) { if(!CompareTwoFiles(&Logger, pFileData1, pFileData1)) nError = Logger.PrintError("The file has different size/content when open without name"); } // Close the archive if(pFileData2 != NULL) STORM_FREE(pFileData2); if(pFileData1 != NULL) STORM_FREE(pFileData1); if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } // Open an empty archive (found in WoW cache - it's just a header) static int TestOpenArchive(const char * szPlainName, const char * szListFile = NULL) { TLogHelper Logger("OpenMpqTest", szPlainName); TFileData * pFileData; HANDLE hMpq; DWORD dwFileCount = 0; DWORD dwTestFlags; char szListFileBuff[MAX_PATH]; bool bIsPartialMpq = false; int nError; // If the file is a partial MPQ, don;t load all files bIsPartialMpq = (strstr(szPlainName, ".MPQ.part") != NULL); // Copy the archive so we won't fuck up the original one nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, NULL, &hMpq); if(nError == ERROR_SUCCESS) { // If the listfile was given, add it to the MPQ if(szListFile != NULL) { Logger.PrintProgress("Adding listfile %s ...", szListFile); CreateFullPathName(szListFileBuff, szMpqSubDir, szListFile); nError = SFileAddListFile(hMpq, szListFileBuff); if(nError != ERROR_SUCCESS) Logger.PrintMessage("Failed to add the listfile to the MPQ"); } // Attempt to open the listfile and attributes if(SFileHasFile(hMpq, LISTFILE_NAME)) { pFileData = LoadMpqFile(&Logger, hMpq, LISTFILE_NAME); if(pFileData != NULL) STORM_FREE(pFileData); } // Attempt to open the listfile and attributes if(SFileHasFile(hMpq, ATTRIBUTES_NAME)) { pFileData = LoadMpqFile(&Logger, hMpq, ATTRIBUTES_NAME); if(pFileData != NULL) STORM_FREE(pFileData); } // Search the archive and load every file dwTestFlags = bIsPartialMpq ? 0 : TEST_FLAG_LOAD_FILES; nError = SearchArchive(&Logger, hMpq, dwTestFlags, &dwFileCount); SFileCloseArchive(hMpq); } return nError; } static int TestOpenArchive_WillFail(const char * szPlainName, const char * szListFile = NULL) { TestOpenArchive(szPlainName, szListFile); return ERROR_SUCCESS; } static int TestOpenArchive_Corrupt(const char * szPlainName) { TLogHelper Logger("OpenCorruptMpqTest", szPlainName); HANDLE hMpq = NULL; TCHAR szFullPathT[MAX_PATH]; char szFullPath[MAX_PATH]; // Copy the archive so we won't fuck up the original one CreateFullPathName(szFullPath, szMpqSubDir, szPlainName); CopyFileName(szFullPathT, szFullPath, strlen(szFullPath)); if(SFileOpenArchive(szFullPathT, 0, STREAM_FLAG_READ_ONLY, &hMpq)) { SFileCloseArchive(hMpq); Logger.PrintMessage("Opening archive %s succeeded, but it shouldn't", szFullPath); return ERROR_CAN_NOT_COMPLETE; } return ERROR_SUCCESS; } // Opens a patched MPQ archive static int TestOpenArchive_Patched(const char * PatchList[], const char * szPatchedFile = NULL, int nExpectedPatchCount = 0) { TLogHelper Logger("OpenPatchedMpqTest", PatchList[0]); HANDLE hMpq; DWORD dwFileCount = 0; int nError; // Open a patched MPQ archive nError = OpenPatchedArchive(&Logger, &hMpq, PatchList); if(nError == ERROR_SUCCESS) { // Check patch count if(szPatchedFile != NULL) nError = VerifyFilePatchCount(&Logger, hMpq, szPatchedFile, nExpectedPatchCount); // Search the archive and load every file if(nError == ERROR_SUCCESS) nError = SearchArchive(&Logger, hMpq, TEST_FLAG_LOAD_FILES, &dwFileCount); // Close the archive SFileCloseArchive(hMpq); } return nError; } // Open an archive for read-only access static int TestOpenArchive_ReadOnly(const char * szPlainName, bool bReadOnly) { const char * szCopyName; TLogHelper Logger("ReadOnlyTest", szPlainName); HANDLE hMpq = NULL; char szFullPathName[MAX_PATH]; DWORD dwFlags = bReadOnly ? MPQ_OPEN_READ_ONLY : 0;; bool bMustSucceed; int nError; // Copy the fiel so we wont screw up something szCopyName = bReadOnly ? "StormLibTest_ReadOnly.mpq" : "StormLibTest_ReadWrite.mpq"; nError = CreateFileCopy(&Logger, szPlainName, szCopyName, szFullPathName); // Now open the archive for read-only access if(nError == ERROR_SUCCESS) nError = OpenExistingArchive(&Logger, szFullPathName, dwFlags, &hMpq); // Now try to add a file. This must fail if the MPQ is read only if(nError == ERROR_SUCCESS) { bMustSucceed = (bReadOnly == false); nError = AddFileToMpq(&Logger, hMpq, "AddedFile.txt", "This is an added file.", MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED, 0, bMustSucceed); if(nError != ERROR_SUCCESS && bMustSucceed == false) nError = ERROR_SUCCESS; } // Now try to rename a file in the MPQ. This must only succeed if the MPQ is not read only if(nError == ERROR_SUCCESS) { bMustSucceed = (bReadOnly == false); nError = RenameMpqFile(&Logger, hMpq, "spawn.mpq", "spawn-renamed.mpq", bMustSucceed); if(nError != ERROR_SUCCESS && bMustSucceed == false) nError = ERROR_SUCCESS; } // Now try to delete a file in the MPQ. This must only succeed if the MPQ is not read only if(nError == ERROR_SUCCESS) { bMustSucceed = (bReadOnly == false); nError = RemoveMpqFile(&Logger, hMpq, "spawn-renamed.mpq", bMustSucceed); if(nError != ERROR_SUCCESS && bMustSucceed == false) nError = ERROR_SUCCESS; } // Close the archive if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestOpenArchive_GetFileInfo(const char * szPlainName1, const char * szPlainName4) { TLogHelper Logger("GetFileInfoTest"); HANDLE hFile; HANDLE hMpq4; HANDLE hMpq1; DWORD cbLength; BYTE DataBuff[0x400]; int nError1; int nError4; // Copy the archive so we won't fuck up the original one nError1 = OpenExistingArchiveWithCopy(&Logger, szPlainName1, NULL, &hMpq1); nError4 = OpenExistingArchiveWithCopy(&Logger, szPlainName4, NULL, &hMpq4); if(nError1 == ERROR_SUCCESS && nError4 == ERROR_SUCCESS) { // Invalid handle - expected (false, ERROR_INVALID_HANDLE) TestGetFileInfo(&Logger, NULL, SFileMpqBetHeader, NULL, 0, NULL, false, ERROR_INVALID_HANDLE); // Valid handle but invalid value of file info class (false, ERROR_INVALID_PARAMETER) TestGetFileInfo(&Logger, NULL, (SFileInfoClass)0xFFF, NULL, 0, NULL, false, ERROR_INVALID_PARAMETER); // Valid archive handle but file info class is for file (false, ERROR_INVALID_HANDLE) TestGetFileInfo(&Logger, NULL, SFileInfoNameHash1, NULL, 0, NULL, false, ERROR_INVALID_HANDLE); // Valid handle and all parameters NULL // Returns (true, ERROR_SUCCESS), if BET table is present, otherwise (false, ERROR_CAN_NOT_COMPLETE) TestGetFileInfo(&Logger, hMpq1, SFileMpqBetHeader, NULL, 0, NULL, false, ERROR_FILE_NOT_FOUND); TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, NULL, 0, NULL, true, ERROR_SUCCESS); // Now try to retrieve the required size of the BET table header TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, NULL, 0, &cbLength, true, ERROR_SUCCESS); // When we call SFileInfo with buffer = NULL and nonzero buffer size, it is ignored TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, NULL, 3, &cbLength, true, ERROR_SUCCESS); // When we call SFileInfo with buffer != NULL and nonzero buffer size, it should return error TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, DataBuff, 3, &cbLength, false, ERROR_INSUFFICIENT_BUFFER); // Request for bet table header should also succeed if we want header only TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, DataBuff, sizeof(TMPQBetHeader), &cbLength, true, ERROR_SUCCESS); // Request for bet table header should also succeed if we want header+flag table only TestGetFileInfo(&Logger, hMpq4, SFileMpqBetHeader, DataBuff, sizeof(DataBuff), &cbLength, true, ERROR_SUCCESS); // Try to retrieve strong signature from the MPQ TestGetFileInfo(&Logger, hMpq1, SFileMpqStrongSignature, NULL, 0, NULL, true, ERROR_SUCCESS); TestGetFileInfo(&Logger, hMpq4, SFileMpqStrongSignature, NULL, 0, NULL, false, ERROR_FILE_NOT_FOUND); // Strong signature is returned including the signature ID TestGetFileInfo(&Logger, hMpq1, SFileMpqStrongSignature, NULL, 0, &cbLength, true, ERROR_SUCCESS); assert(cbLength == MPQ_STRONG_SIGNATURE_SIZE + 4); // Retrieve the signature TestGetFileInfo(&Logger, hMpq1, SFileMpqStrongSignature, DataBuff, sizeof(DataBuff), &cbLength, true, ERROR_SUCCESS); assert(memcmp(DataBuff, "NGIS", 4) == 0); // Check SFileGetFileInfo on if(SFileOpenFileEx(hMpq4, LISTFILE_NAME, 0, &hFile)) { // Valid parameters but the handle should be file handle TestGetFileInfo(&Logger, hMpq4, SFileInfoFileTime, DataBuff, sizeof(DataBuff), &cbLength, false, ERROR_INVALID_HANDLE); // Valid parameters TestGetFileInfo(&Logger, hFile, SFileInfoFileTime, DataBuff, sizeof(DataBuff), &cbLength, true, ERROR_SUCCESS); SFileCloseFile(hFile); } } if(hMpq4 != NULL) SFileCloseArchive(hMpq4); if(hMpq1 != NULL) SFileCloseArchive(hMpq1); return ERROR_SUCCESS; } static int TestOpenArchive_MasterMirror(const char * szMirrorName, const char * szMasterName, const char * szFileToExtract, bool bCopyMirrorFile) { TFileData * pFileData; TLogHelper Logger("OpenServerMirror", szMirrorName); HANDLE hFile = NULL; HANDLE hMpq = NULL; DWORD dwVerifyResult; char szMirrorPath[MAX_PATH + MAX_PATH]; // Combined name char szMasterPath[MAX_PATH]; // Original (server) name int nError; // Create both paths nError = CreateMasterAndMirrorPaths(&Logger, szMirrorPath, szMasterPath, szMirrorName, szMasterName, bCopyMirrorFile); // Now open both archives as local-server pair if(nError == ERROR_SUCCESS) { nError = OpenExistingArchive(&Logger, szMirrorPath, 0, &hMpq); } // The MPQ must be read-only. Writing to mirrored MPQ is not allowed if(nError == ERROR_SUCCESS) { if(SFileCreateFile(hMpq, "AddedFile.bin", 0, 0x10, 0, MPQ_FILE_COMPRESS, &hFile)) { SFileCloseFile(hFile); Logger.PrintMessage("The archive is writable, although it should not be"); nError = ERROR_FILE_CORRUPT; } } // Verify the file if(nError == ERROR_SUCCESS && szFileToExtract != NULL) { dwVerifyResult = SFileVerifyFile(hMpq, szFileToExtract, SFILE_VERIFY_ALL); if(dwVerifyResult & VERIFY_FILE_ERROR_MASK) { Logger.PrintMessage("File verification failed"); nError = ERROR_FILE_CORRUPT; } } // Load the file to memory if(nError == ERROR_SUCCESS && szFileToExtract) { pFileData = LoadMpqFile(&Logger, hMpq, szFileToExtract); if(pFileData != NULL) STORM_FREE(pFileData); } if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestOpenArchive_VerifySignature(const char * szPlainName, const char * szOriginalName) { TLogHelper Logger("VerifySignatureTest", szPlainName); HANDLE hMpq; DWORD dwSignatures = 0; int nVerifyError; int nError = ERROR_SUCCESS; // We need original name for the signature check nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, szOriginalName, &hMpq); if(nError == ERROR_SUCCESS) { // Query the signature types Logger.PrintProgress("Retrieving signatures ..."); TestGetFileInfo(&Logger, hMpq, SFileMpqSignatures, &dwSignatures, sizeof(DWORD), NULL, true, ERROR_SUCCESS); // Verify any of the present signatures Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Verify the result if((dwSignatures & SIGNATURE_TYPE_STRONG) && (nVerifyError != ERROR_STRONG_SIGNATURE_OK)) { Logger.PrintMessage("Strong signature verification error"); nError = ERROR_FILE_CORRUPT; } // Verify the result if((dwSignatures & SIGNATURE_TYPE_WEAK) && (nVerifyError != ERROR_WEAK_SIGNATURE_OK)) { Logger.PrintMessage("Weak signature verification error"); nError = ERROR_FILE_CORRUPT; } SFileCloseArchive(hMpq); } return nError; } static int TestOpenArchive_ModifySigned(const char * szPlainName, const char * szOriginalName) { TLogHelper Logger("ModifySignedTest", szPlainName); HANDLE hMpq = NULL; int nVerifyError; int nError = ERROR_SUCCESS; // We need original name for the signature check nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, szOriginalName, &hMpq); if(nError == ERROR_SUCCESS) { // Verify the weak signature Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Check the result signature if(nVerifyError != ERROR_WEAK_SIGNATURE_OK) { Logger.PrintMessage("Weak signature verification error"); nError = ERROR_FILE_CORRUPT; } } // Add a file and verify the signature again if(nError == ERROR_SUCCESS) { // Verify any of the present signatures Logger.PrintProgress("Modifying signed archive ..."); nError = AddFileToMpq(&Logger, hMpq, "AddedFile01.txt", "This is a file added to signed MPQ", 0, 0, true); } // Verify the signature again if(nError == ERROR_SUCCESS) { // Verify the weak signature Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Check the result signature if(nVerifyError != ERROR_WEAK_SIGNATURE_OK) { Logger.PrintMessage("Weak signature verification error"); nError = ERROR_FILE_CORRUPT; } } // Close the MPQ if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestOpenArchive_SignExisting(const char * szPlainName) { TLogHelper Logger("SignExistingMpq", szPlainName); HANDLE hMpq = NULL; int nVerifyError; int nError = ERROR_SUCCESS; // We need original name for the signature check nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, szPlainName, &hMpq); if(nError == ERROR_SUCCESS) { // Verify the weak signature Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Check the result signature if(nVerifyError != ERROR_NO_SIGNATURE) { Logger.PrintMessage("There already is a signature in the MPQ"); nError = ERROR_FILE_CORRUPT; } } // Add a file and verify the signature again if(nError == ERROR_SUCCESS) { // Verify any of the present signatures Logger.PrintProgress("Signing the MPQ ..."); if(!SFileSignArchive(hMpq, SIGNATURE_TYPE_WEAK)) { Logger.PrintMessage("Failed to create archive signature"); nError = ERROR_FILE_CORRUPT; } } // Verify the signature again if(nError == ERROR_SUCCESS) { // Verify the weak signature Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Check the result signature if(nVerifyError != ERROR_WEAK_SIGNATURE_OK) { Logger.PrintMessage("Weak signature verification error"); nError = ERROR_FILE_CORRUPT; } } // Close the MPQ if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } // Open an empty archive (found in WoW cache - it's just a header) static int TestOpenArchive_CraftedUserData(const char * szPlainName, const char * szCopyName) { TLogHelper Logger("CraftedMpqTest", szPlainName); HANDLE hMpq; DWORD dwFileCount1 = 0; DWORD dwFileCount2 = 0; BYTE FileHash1[MD5_DIGEST_SIZE]; BYTE FileHash2[MD5_DIGEST_SIZE]; char szFullPath[MAX_PATH]; int nError; // Create copy of the archive, with interleaving some user data nError = CreateFileCopy(&Logger, szPlainName, szCopyName, szFullPath, 0x400, 0x531); // Open the archive and load some files if(nError == ERROR_SUCCESS) { // Open the archive nError = OpenExistingArchive(&Logger, szFullPath, 0, &hMpq); if(nError != ERROR_SUCCESS) return nError; // Verify presence of (listfile) and (attributes) CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, true); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, true); // Search the archive and load every file nError = SearchArchive(&Logger, hMpq, TEST_FLAG_LOAD_FILES | TEST_FLAG_HASH_FILES, &dwFileCount1, FileHash1); SFileCloseArchive(hMpq); } // Try to compact the MPQ if(nError == ERROR_SUCCESS) { // Open the archive again nError = OpenExistingArchive(&Logger, szFullPath, 0, &hMpq); if(nError != ERROR_SUCCESS) return nError; // Compact the archive Logger.PrintProgress("Compacting archive %s ...", GetShortPlainName(szFullPath)); if(!SFileSetCompactCallback(hMpq, CompactCallback, &Logger)) nError = Logger.PrintError("Failed to compact archive %s", szFullPath); SFileCompactArchive(hMpq, NULL, false); SFileCloseArchive(hMpq); } // Open the archive and load some files if(nError == ERROR_SUCCESS) { // Open the archive nError = OpenExistingArchive(&Logger, szFullPath, 0, &hMpq); if(nError != ERROR_SUCCESS) return nError; // Verify presence of (listfile) and (attributes) CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, true); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, true); // Search the archive and load every file nError = SearchArchive(&Logger, hMpq, TEST_FLAG_LOAD_FILES | TEST_FLAG_HASH_FILES, &dwFileCount2, FileHash2); SFileCloseArchive(hMpq); } // Compare the file counts and their hashes if(nError == ERROR_SUCCESS) { if(dwFileCount2 != dwFileCount1) Logger.PrintMessage("Different file count after compacting archive: %u vs %u", dwFileCount2, dwFileCount1); if(memcmp(FileHash2, FileHash1, MD5_DIGEST_SIZE)) Logger.PrintMessage("Different file hash after compacting archive"); } return nError; } static int TestOpenArchive_CompactingTest(const char * szPlainName, const char * szListFile) { TLogHelper Logger("CompactingTest", szPlainName); HANDLE hMpq = NULL; char szFullListName[MAX_PATH]; int nError = ERROR_SUCCESS; // Create copy of the listfile if(szListFile != NULL) { nError = CreateFileCopy(&Logger, szListFile, szListFile, szFullListName, 0, 0); szListFile = szFullListName; } // Create copy of the archive if(nError == ERROR_SUCCESS) { nError = OpenExistingArchiveWithCopy(&Logger, szPlainName, szPlainName, &hMpq); } if(nError == ERROR_SUCCESS) { // Compact the archive Logger.PrintProgress("Compacting archive %s ...", szPlainName); if(!SFileSetCompactCallback(hMpq, CompactCallback, &Logger)) nError = Logger.PrintError("Failed to set the compact callback"); if(!SFileCompactArchive(hMpq, szListFile, false)) nError = Logger.PrintError("Failed to compact archive %s", szPlainName); SFileCloseArchive(hMpq); } return nError; } static int ForEachFile_VerifyFileChecksum(const char * szFullPath) { const char * szShortPlainName = GetShortPlainName(szFullPath); TFileData * pFileData; char * szExtension; char szShaFileName[MAX_PATH]; char szSha1Text[0x40]; int nError = ERROR_SUCCESS; // Try to load the file with the SHA extension strcpy(szShaFileName, szFullPath); szExtension = strrchr(szShaFileName, '.'); if(szExtension == NULL) return ERROR_SUCCESS; // Skip .SHA and .TXT files if(!_stricmp(szExtension, ".sha") || !_stricmp(szExtension, ".txt")) return ERROR_SUCCESS; // Load the local file to memory strcpy(szExtension, ".sha"); pFileData = LoadLocalFile(NULL, szShaFileName, false); if(pFileData != NULL) { TLogHelper Logger("VerifyFileHash", szShortPlainName); // Calculate SHA1 of the entire file nError = CalculateFileSha1(&Logger, szFullPath, szSha1Text); if(nError == ERROR_SUCCESS) { // Compare with what we loaded from the file if(pFileData->dwFileSize >= (SHA1_DIGEST_SIZE * 2)) { // Compare the SHA1 if(_strnicmp(szSha1Text, (char *)pFileData->FileData, (SHA1_DIGEST_SIZE * 2))) { Logger.PrintError("File CRC check failed: %s", szFullPath); nError = ERROR_FILE_CORRUPT; } } } STORM_FREE(pFileData); } return nError; } // Opens a found archive static int ForEachFile_OpenArchive(const char * szFullPath) { HANDLE hMpq = NULL; DWORD dwFileCount = 0; int nError = ERROR_SUCCESS; // Check if it's a MPQ file type if(IsMpqExtension(szFullPath)) { TLogHelper Logger("OpenEachMpqTest", GetShortPlainName(szFullPath)); // Open the MPQ name nError = OpenExistingArchive(&Logger, szFullPath, 0, &hMpq); if(nError == ERROR_AVI_FILE || nError == ERROR_FILE_CORRUPT || nError == ERROR_BAD_FORMAT) return ERROR_SUCCESS; // Search the archive and load every file if(nError == ERROR_SUCCESS) { nError = SearchArchive(&Logger, hMpq, 0, &dwFileCount); SFileCloseArchive(hMpq); } } // Correct some errors if(nError == ERROR_FILE_CORRUPT || nError == ERROR_FILE_INCOMPLETE) return ERROR_SUCCESS; return nError; } // Adding a file to MPQ that had no (listfile) and no (attributes). // We expect that neither of these will be present after the archive is closed static int TestAddFile_ListFileTest(const char * szSourceMpq, bool bShouldHaveListFile, bool bShouldHaveAttributes) { TLogHelper Logger("ListFileTest", szSourceMpq); TFileData * pFileData = NULL; const char * szBackupMpq = bShouldHaveListFile ? "StormLibTest_HasListFile.mpq" : "StormLibTest_NoListFile.mpq"; const char * szFileName = "AddedFile001.txt"; const char * szFileData = "0123456789ABCDEF"; HANDLE hMpq = NULL; DWORD dwFileSize = (DWORD)strlen(szFileData); int nError = ERROR_SUCCESS; // Copy the archive so we won't fuck up the original one nError = OpenExistingArchiveWithCopy(&Logger, szSourceMpq, szBackupMpq, &hMpq); // Add a file if(nError == ERROR_SUCCESS) { // Now add a file nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData, MPQ_FILE_IMPLODE, MPQ_COMPRESSION_PKWARE); SFileCloseArchive(hMpq); } // Now reopen the archive if(nError == ERROR_SUCCESS) nError = OpenExistingArchiveWithCopy(&Logger, NULL, szBackupMpq, &hMpq); // Now the file has been written and the MPQ has been saved. // We Reopen the MPQ and check if there is no (listfile) nor (attributes). if(nError == ERROR_SUCCESS) { // Verify presence of (listfile) and (attributes) CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, bShouldHaveListFile); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, bShouldHaveAttributes); // Try to open the file that we recently added pFileData = LoadMpqFile(&Logger, hMpq, szFileName); if(pFileData != NULL) { // Verify if the file size matches if(pFileData->dwFileSize == dwFileSize) { // Verify if the file data match if(memcmp(pFileData->FileData, szFileData, dwFileSize)) { Logger.PrintError("The data of the added file does not match"); nError = ERROR_FILE_CORRUPT; } } else { Logger.PrintError("The size of the added file does not match"); nError = ERROR_FILE_CORRUPT; } // Delete the file data STORM_FREE(pFileData); } else { nError = Logger.PrintError("Failed to open the file previously added"); } } // Close the MPQ archive if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestCreateArchive_EmptyMpq(const char * szPlainName, DWORD dwCreateFlags) { TLogHelper Logger("CreateEmptyMpq", szPlainName); HANDLE hMpq = NULL; DWORD dwFileCount = 0; int nError; // Create the full path name nError = CreateNewArchive(&Logger, szPlainName, dwCreateFlags, 0, &hMpq); if(nError == ERROR_SUCCESS) { SearchArchive(&Logger, hMpq); SFileCloseArchive(hMpq); } // Reopen the empty MPQ if(nError == ERROR_SUCCESS) { nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); if(nError == ERROR_SUCCESS) { SFileGetFileInfo(hMpq, SFileMpqNumberOfFiles, &dwFileCount, sizeof(dwFileCount), NULL); CheckIfFileIsPresent(&Logger, hMpq, "File00000000.xxx", false); CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, false); SearchArchive(&Logger, hMpq); SFileCloseArchive(hMpq); } } return nError; } static int TestCreateArchive_TestGaps(const char * szPlainName) { TLogHelper Logger("CreateGapsTest", szPlainName); ULONGLONG ByteOffset1 = 0xFFFFFFFF; ULONGLONG ByteOffset2 = 0xEEEEEEEE; HANDLE hMpq = NULL; HANDLE hFile = NULL; char szFullPath[MAX_PATH]; int nError = ERROR_SUCCESS; // Create new MPQ nError = CreateNewArchive_V2(&Logger, szPlainName, MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES | MPQ_FORMAT_VERSION_4, 4000, &hMpq); if(nError == ERROR_SUCCESS) { // Add one file and flush the archive nError = AddFileToMpq(&Logger, hMpq, "AddedFile01.txt", "This is the file data.", MPQ_FILE_COMPRESS); SFileCloseArchive(hMpq); hMpq = NULL; } // Reopen the MPQ and add another file. // The new file must be added to the position of the (listfile) if(nError == ERROR_SUCCESS) { CreateFullPathName(szFullPath, NULL, szPlainName); nError = OpenExistingArchive(&Logger, szFullPath, 0, &hMpq); if(nError == ERROR_SUCCESS) { // Retrieve the position of the (listfile) if(SFileOpenFileEx(hMpq, LISTFILE_NAME, 0, &hFile)) { SFileGetFileInfo(hFile, SFileInfoByteOffset, &ByteOffset1, sizeof(ULONGLONG), NULL); SFileCloseFile(hFile); } else nError = GetLastError(); } } // Add another file and check its position. It must be at the position of the former listfile if(nError == ERROR_SUCCESS) { const char * szAddedFile = "AddedFile02.txt"; // Add another file nError = AddFileToMpq(&Logger, hMpq, szAddedFile, "This is the second added file.", MPQ_FILE_COMPRESS); // Retrieve the position of the (listfile) if(SFileOpenFileEx(hMpq, szAddedFile, 0, &hFile)) { SFileGetFileInfo(hFile, SFileInfoByteOffset, &ByteOffset2, sizeof(ULONGLONG), NULL); SFileCloseFile(hFile); } else nError = GetLastError(); } // Now check the positions if(nError == ERROR_SUCCESS) { if(ByteOffset1 != ByteOffset2) { Logger.PrintError("The added file was not written to the position of (listfile)"); nError = ERROR_FILE_CORRUPT; } } // Close the archive if needed if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestCreateArchive_Signed(const char * szPlainName, bool bSignAtCreate) { TLogHelper Logger("CreateSignedMpq", szPlainName); HANDLE hMpq = NULL; DWORD dwCreateFlags = MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES | MPQ_FORMAT_VERSION_1; DWORD dwSignatures = 0; DWORD nVerifyError = 0; int nError = ERROR_SUCCESS; // Method 1: Create the archive as signed if(bSignAtCreate) dwCreateFlags |= MPQ_CREATE_SIGNATURE; // Create new MPQ nError = CreateNewArchive_V2(&Logger, szPlainName, dwCreateFlags, 4000, &hMpq); if(nError == ERROR_SUCCESS) { // Add one file and flush the archive nError = AddFileToMpq(&Logger, hMpq, "AddedFile01.txt", "This is the file data.", MPQ_FILE_COMPRESS); } // Sign the archive with weak signature if(nError == ERROR_SUCCESS) { if(!SFileSignArchive(hMpq, SIGNATURE_TYPE_WEAK)) nError = ERROR_SUCCESS; } // Reopen the MPQ and add another file. // The new file must be added to the position of the (listfile) if(nError == ERROR_SUCCESS) { // Query the signature types Logger.PrintProgress("Retrieving signatures ..."); TestGetFileInfo(&Logger, hMpq, SFileMpqSignatures, &dwSignatures, sizeof(DWORD), NULL, true, ERROR_SUCCESS); // Verify any of the present signatures Logger.PrintProgress("Verifying archive signature ..."); nVerifyError = SFileVerifyArchive(hMpq); // Verify the result if((dwSignatures != SIGNATURE_TYPE_WEAK) && (nVerifyError != ERROR_WEAK_SIGNATURE_OK)) { Logger.PrintMessage("Weak signature verification error"); nError = ERROR_FILE_CORRUPT; } } // Close the archive if(hMpq != NULL) SFileCloseArchive(hMpq); return nError; } static int TestCreateArchive_MpqEditor(const char * szPlainName, const char * szFileName) { TLogHelper Logger("CreateMpqEditor", szPlainName); HANDLE hMpq = NULL; int nError = ERROR_SUCCESS; // Create new MPQ nError = CreateNewArchive_V2(&Logger, szPlainName, MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, 4000, &hMpq); if(nError == ERROR_SUCCESS) { // Flush the archive first SFileFlushArchive(hMpq); // Add one file nError = AddFileToMpq(&Logger, hMpq, szFileName, "This is the file data.", MPQ_FILE_COMPRESS); // Flush the archive again SFileFlushArchive(hMpq); SFileCloseArchive(hMpq); } else { nError = GetLastError(); } return nError; } static int TestCreateArchive_FillArchive(const char * szPlainName, DWORD dwCreateFlags) { TLogHelper Logger("CreateFullMpq", szPlainName); const char * szFileData = "TestCreateArchive_FillArchive: Testing file data"; char szFileName[MAX_PATH]; HANDLE hMpq = NULL; DWORD dwMaxFileCount = 6; DWORD dwCompression = MPQ_COMPRESSION_ZLIB; DWORD dwFlags = MPQ_FILE_ENCRYPTED | MPQ_FILE_COMPRESS; int nError; // Note that StormLib will round the maxfile count // up to hash table size (nearest power of two) if((dwCreateFlags & MPQ_CREATE_LISTFILE) == 0) dwMaxFileCount++; if((dwCreateFlags & MPQ_CREATE_ATTRIBUTES) == 0) dwMaxFileCount++; // Create the new MPQ archive nError = CreateNewArchive_V2(&Logger, szPlainName, dwCreateFlags, dwMaxFileCount, &hMpq); if(nError == ERROR_SUCCESS) { // Flush the archive first SFileFlushArchive(hMpq); // Add all files for(unsigned int i = 0; i < dwMaxFileCount; i++) { sprintf(szFileName, "AddedFile%03u.txt", i); nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData, dwFlags, dwCompression); if(nError != ERROR_SUCCESS) break; } // Flush the archive again SFileFlushArchive(hMpq); } // Now the MPQ should be full. It must not be possible to add another file if(nError == ERROR_SUCCESS) { nError = AddFileToMpq(&Logger, hMpq, "ShouldNotBeHere.txt", szFileData, MPQ_FILE_COMPRESS, MPQ_COMPRESSION_ZLIB, false); assert(nError != ERROR_SUCCESS); nError = ERROR_SUCCESS; } // Close the archive to enforce saving all tables if(hMpq != NULL) SFileCloseArchive(hMpq); hMpq = NULL; // Reopen the archive again if(nError == ERROR_SUCCESS) nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); // The archive should still be full if(nError == ERROR_SUCCESS) { CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, (dwCreateFlags & MPQ_CREATE_LISTFILE) ? true : false); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, (dwCreateFlags & MPQ_CREATE_ATTRIBUTES) ? true : false); nError = AddFileToMpq(&Logger, hMpq, "ShouldNotBeHere.txt", szFileData, MPQ_FILE_COMPRESS, MPQ_COMPRESSION_ZLIB, false); assert(nError != ERROR_SUCCESS); nError = ERROR_SUCCESS; } // The (listfile) must be present if(nError == ERROR_SUCCESS) { CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, (dwCreateFlags & MPQ_CREATE_LISTFILE) ? true : false); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, (dwCreateFlags & MPQ_CREATE_ATTRIBUTES) ? true : false); nError = RemoveMpqFile(&Logger, hMpq, szFileName, true); } // Now add the file again. This time, it should be possible OK if(nError == ERROR_SUCCESS) { nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData, dwFlags, dwCompression, true); assert(nError == ERROR_SUCCESS); } // Now add the file again. This time, it should be fail if(nError == ERROR_SUCCESS) { nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData, dwFlags, dwCompression, false); assert(nError != ERROR_SUCCESS); nError = ERROR_SUCCESS; } // Close the archive and return if(hMpq != NULL) SFileCloseArchive(hMpq); hMpq = NULL; // Reopen the archive for the third time to verify that both internal files are there if(nError == ERROR_SUCCESS) { nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); if(nError == ERROR_SUCCESS) { CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, (dwCreateFlags & MPQ_CREATE_LISTFILE) ? true : false); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, (dwCreateFlags & MPQ_CREATE_ATTRIBUTES) ? true : false); SFileCloseArchive(hMpq); } } return nError; } static int TestCreateArchive_IncMaxFileCount(const char * szPlainName) { TLogHelper Logger("IncMaxFileCount", szPlainName); const char * szFileData = "TestCreateArchive_IncMaxFileCount: Testing file data"; char szFileName[MAX_PATH]; HANDLE hMpq = NULL; DWORD dwMaxFileCount = 1; int nError; // Create the new MPQ nError = CreateNewArchive(&Logger, szPlainName, MPQ_CREATE_ARCHIVE_V4 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, dwMaxFileCount, &hMpq); // Now add exactly one file if(nError == ERROR_SUCCESS) { nError = AddFileToMpq(&Logger, hMpq, "AddFile_base.txt", szFileData); SFileFlushArchive(hMpq); SFileCloseArchive(hMpq); } // Now add 10 files. Each time we cannot add the file due to archive being full, // we increment the max file count if(nError == ERROR_SUCCESS) { for(unsigned int i = 0; i < 10; i++) { // Open the archive again nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); if(nError != ERROR_SUCCESS) break; // Add one file sprintf(szFileName, "AddFile_%04u.txt", i); nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData); if(nError != ERROR_SUCCESS) { // Increment the ma file count by one dwMaxFileCount = SFileGetMaxFileCount(hMpq) + 1; Logger.PrintProgress("Increasing max file count to %u ...", dwMaxFileCount); SFileSetMaxFileCount(hMpq, dwMaxFileCount); // Attempt to create the file again nError = AddFileToMpq(&Logger, hMpq, szFileName, szFileData, 0, 0, true); } // Compact the archive and close it SFileSetCompactCallback(hMpq, CompactCallback, &Logger); SFileCompactArchive(hMpq, NULL, false); SFileCloseArchive(hMpq); if(nError != ERROR_SUCCESS) break; } } return nError; } static int TestCreateArchive_UnicodeNames() { TLogHelper Logger("MpqUnicodeName"); DWORD dwCreateFlags = MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES; int nError = ERROR_SUCCESS; nError = CreateNewArchiveU(&Logger, szUnicodeName1, dwCreateFlags | MPQ_CREATE_ARCHIVE_V1, 15); if(nError != ERROR_SUCCESS) return nError; nError = CreateNewArchiveU(&Logger, szUnicodeName2, dwCreateFlags | MPQ_CREATE_ARCHIVE_V2, 58); if(nError != ERROR_SUCCESS) return nError; nError = CreateNewArchiveU(&Logger, szUnicodeName3, dwCreateFlags | MPQ_CREATE_ARCHIVE_V3, 15874); if(nError != ERROR_SUCCESS) return nError; nError = CreateNewArchiveU(&Logger, szUnicodeName4, dwCreateFlags | MPQ_CREATE_ARCHIVE_V4, 87541); if(nError != ERROR_SUCCESS) return nError; nError = CreateNewArchiveU(&Logger, szUnicodeName5, dwCreateFlags | MPQ_CREATE_ARCHIVE_V3, 87541); if(nError != ERROR_SUCCESS) return nError; nError = CreateNewArchiveU(&Logger, szUnicodeName5, dwCreateFlags | MPQ_CREATE_ARCHIVE_V2, 87541); if(nError != ERROR_SUCCESS) return nError; return nError; } static int TestCreateArchive_FileFlagTest(const char * szPlainName) { TLogHelper Logger("FileFlagTest", szPlainName); HANDLE hMpq = NULL; // Handle of created archive char szFileName1[MAX_PATH]; char szFileName2[MAX_PATH]; char szFullPath[MAX_PATH]; const char * szMiddleFile = "FileTest_10.exe"; LCID LocaleIDs[] = {0x000, 0x405, 0x406, 0x407, 0xFFFF}; char szArchivedName[MAX_PATH]; DWORD dwMaxFileCount = 0; DWORD dwFileCount = 0; size_t i; int nError; // Create paths for local file to be added CreateFullPathName(szFileName1, szMpqSubDir, "AddFile.exe"); CreateFullPathName(szFileName2, szMpqSubDir, "AddFile.bin"); // Create an empty file that will serve as holder for the MPQ nError = CreateEmptyFile(&Logger, szPlainName, 0x100000, szFullPath); // Create new MPQ archive over that file if(nError == ERROR_SUCCESS) nError = CreateNewArchive(&Logger, szPlainName, MPQ_CREATE_ARCHIVE_V1 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, 17, &hMpq); // Add the same file multiple times if(nError == ERROR_SUCCESS) { dwMaxFileCount = SFileGetMaxFileCount(hMpq); for(i = 0; AddFlags[i] != 0xFFFFFFFF; i++) { sprintf(szArchivedName, "FileTest_%02u.exe", (unsigned int)i); nError = AddLocalFileToMpq(&Logger, hMpq, szArchivedName, szFileName1, AddFlags[i], 0); if(nError != ERROR_SUCCESS) break; dwFileCount++; } } // Delete a file in the middle of the file table if(nError == ERROR_SUCCESS) { Logger.PrintProgress("Removing file %s ...", szMiddleFile); nError = RemoveMpqFile(&Logger, hMpq, szMiddleFile, true); dwFileCount--; } // Add one more file if(nError == ERROR_SUCCESS) { nError = AddLocalFileToMpq(&Logger, hMpq, "FileTest_xx.exe", szFileName1); dwFileCount++; } // Try to decrement max file count. This must succeed if(nError == ERROR_SUCCESS) { Logger.PrintProgress("Attempting to decrement max file count ..."); if(SFileSetMaxFileCount(hMpq, 5)) nError = Logger.PrintError("Max file count decremented, even if it should fail"); } // Add ZeroSize.txt several times under a different locale if(nError == ERROR_SUCCESS) { for(i = 0; LocaleIDs[i] != 0xFFFF; i++) { bool bMustSucceed = ((dwFileCount + 2) < dwMaxFileCount); SFileSetLocale(LocaleIDs[i]); nError = AddLocalFileToMpq(&Logger, hMpq, "ZeroSize_1.txt", szFileName2); if(nError != ERROR_SUCCESS) { if(bMustSucceed == false) nError = ERROR_SUCCESS; break; } dwFileCount++; } } // Add ZeroSize.txt again several times under a different locale if(nError == ERROR_SUCCESS) { for(i = 0; LocaleIDs[i] != 0xFFFF; i++) { bool bMustSucceed = ((dwFileCount + 2) < dwMaxFileCount); SFileSetLocale(LocaleIDs[i]); nError = AddLocalFileToMpq(&Logger, hMpq, "ZeroSize_2.txt", szFileName2, 0, 0, bMustSucceed); if(nError != ERROR_SUCCESS) { if(bMustSucceed == false) nError = ERROR_SUCCESS; break; } dwFileCount++; } } // Verify how many files did we add to the MPQ if(nError == ERROR_SUCCESS) { if(dwFileCount + 2 != dwMaxFileCount) { Logger.PrintErrorVa("Number of files added to MPQ was unexpected (expected %u, added %u)", dwFileCount, dwMaxFileCount - 2); nError = ERROR_FILE_CORRUPT; } } // Test rename function if(nError == ERROR_SUCCESS) { Logger.PrintProgress("Testing rename files ..."); SFileSetLocale(LANG_NEUTRAL); if(!SFileRenameFile(hMpq, "FileTest_08.exe", "FileTest_08a.exe")) nError = Logger.PrintError("Failed to rename the file"); } if(nError == ERROR_SUCCESS) { if(!SFileRenameFile(hMpq, "FileTest_08a.exe", "FileTest_08.exe")) nError = Logger.PrintError("Failed to rename the file"); } if(nError == ERROR_SUCCESS) { if(SFileRenameFile(hMpq, "FileTest_10.exe", "FileTest_10a.exe")) { Logger.PrintError("Rename test succeeded even if it shouldn't"); nError = ERROR_FILE_CORRUPT; } } if(nError == ERROR_SUCCESS) { if(SFileRenameFile(hMpq, "FileTest_10a.exe", "FileTest_10.exe")) { Logger.PrintError("Rename test succeeded even if it shouldn't"); nError = ERROR_FILE_CORRUPT; } } // Close the archive if(hMpq != NULL) SFileCloseArchive(hMpq); hMpq = NULL; // Try to reopen the archive if(nError == ERROR_SUCCESS) nError = OpenExistingArchive(&Logger, szFullPath, 0, NULL); return nError; } static int TestCreateArchive_WaveCompressionsTest(const char * szPlainName, const char * szWaveFile) { TLogHelper Logger("CompressionsTest", szPlainName); HANDLE hMpq = NULL; // Handle of created archive char szFileName[MAX_PATH]; // Source file to be added char szArchivedName[MAX_PATH]; DWORD dwCmprCount = sizeof(WaveCompressions) / sizeof(DWORD); DWORD dwAddedFiles = 0; DWORD dwFoundFiles = 0; int nError; // Create paths for local file to be added CreateFullPathName(szFileName, szMpqSubDir, szWaveFile); // Create new archive nError = CreateNewArchive(&Logger, szPlainName, MPQ_CREATE_ARCHIVE_V1 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, 0x40, &hMpq); // Add the same file multiple times if(nError == ERROR_SUCCESS) { Logger.UserTotal = dwCmprCount; for(unsigned int i = 0; i < dwCmprCount; i++) { sprintf(szArchivedName, "WaveFile_%02u.wav", i + 1); nError = AddLocalFileToMpq(&Logger, hMpq, szArchivedName, szFileName, MPQ_FILE_COMPRESS | MPQ_FILE_ENCRYPTED | MPQ_FILE_SECTOR_CRC, WaveCompressions[i]); if(nError != ERROR_SUCCESS) break; Logger.UserCount++; dwAddedFiles++; } SFileCloseArchive(hMpq); } // Reopen the archive extract each WAVE file and try to play it if(nError == ERROR_SUCCESS) { nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); if(nError == ERROR_SUCCESS) { SearchArchive(&Logger, hMpq, TEST_FLAG_LOAD_FILES | TEST_FLAG_PLAY_WAVES, &dwFoundFiles, NULL); SFileCloseArchive(hMpq); } // Check if the number of found files is the same like the number of added files // DOn;t forget that there will be (listfile) and (attributes) if(dwFoundFiles != (dwAddedFiles + 2)) { Logger.PrintError("Number of found files does not match number of added files."); nError = ERROR_FILE_CORRUPT; } } return nError; } static int TestCreateArchive_ListFilePos(const char * szPlainName) { TFileData * pFileData; const char * szReaddedFile = "AddedFile_##.txt"; const char * szFileMask = "AddedFile_%02u.txt"; TLogHelper Logger("ListFilePos", szPlainName); HANDLE hMpq = NULL; // Handle of created archive char szArchivedName[MAX_PATH]; DWORD dwMaxFileCount = 0x1E; DWORD dwAddedCount = 0; size_t i; int nError; // Create a new archive with the limit of 0x20 files nError = CreateNewArchive(&Logger, szPlainName, MPQ_CREATE_ARCHIVE_V4 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, dwMaxFileCount, &hMpq); // Add 0x1E files if(nError == ERROR_SUCCESS) { for(i = 0; i < dwMaxFileCount; i++) { sprintf(szArchivedName, szFileMask, i); nError = AddFileToMpq(&Logger, hMpq, szArchivedName, "This is a text data.", 0, 0, true); if(nError != ERROR_SUCCESS) break; dwAddedCount++; } } // Delete few middle files if(nError == ERROR_SUCCESS) { for(i = 0; i < (dwMaxFileCount / 2); i++) { sprintf(szArchivedName, szFileMask, i); nError = RemoveMpqFile(&Logger, hMpq, szArchivedName, true); if(nError != ERROR_SUCCESS) break; } } // Close the archive if(hMpq != NULL) SFileCloseArchive(hMpq); hMpq = NULL; // Reopen the archive to catch any asserts if(nError == ERROR_SUCCESS) nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); // Check that (listfile) is at the end if(nError == ERROR_SUCCESS) { pFileData = LoadMpqFile(&Logger, hMpq, LISTFILE_NAME); if(pFileData != NULL) { if(pFileData->dwBlockIndex < dwAddedCount) Logger.PrintMessage("Unexpected file index of %s", LISTFILE_NAME); STORM_FREE(pFileData); } pFileData = LoadMpqFile(&Logger, hMpq, ATTRIBUTES_NAME); if(pFileData != NULL) { if(pFileData->dwBlockIndex <= dwAddedCount) Logger.PrintMessage("Unexpected file index of %s", ATTRIBUTES_NAME); STORM_FREE(pFileData); } // Add new file to the archive. It should be added to position 0 // (since position 0 should be free) nError = AddFileToMpq(&Logger, hMpq, szReaddedFile, "This is a re-added file.", 0, 0, true); if(nError == ERROR_SUCCESS) { pFileData = LoadMpqFile(&Logger, hMpq, szReaddedFile); if(pFileData != NULL) { if(pFileData->dwBlockIndex != 0) Logger.PrintMessage("Unexpected file index of %s", szReaddedFile); STORM_FREE(pFileData); } } SFileCloseArchive(hMpq); } return nError; } static int TestCreateArchive_BigArchive(const char * szPlainName) { const char * szFileMask = "AddedFile_%02u.txt"; TLogHelper Logger("BigMpqTest"); HANDLE hMpq = NULL; // Handle of created archive char szLocalFileName[MAX_PATH]; char szArchivedName[MAX_PATH]; DWORD dwMaxFileCount = 0x20; DWORD dwAddedCount = 0; size_t i; int nError; // Create a new archive with the limit of 0x20 files nError = CreateNewArchive(&Logger, szPlainName, MPQ_CREATE_ARCHIVE_V3 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES, dwMaxFileCount, &hMpq); if(nError == ERROR_SUCCESS) { // Now add few really big files CreateFullPathName(szLocalFileName, szMpqSubDir, "MPQ_1997_v1_Diablo1_DIABDAT.MPQ"); Logger.UserTotal = (dwMaxFileCount / 2); for(i = 0; i < dwMaxFileCount / 2; i++) { sprintf(szArchivedName, szFileMask, i + 1); nError = AddLocalFileToMpq(&Logger, hMpq, szArchivedName, szLocalFileName, 0, 0, true); if(nError != ERROR_SUCCESS) break; Logger.UserCount++; dwAddedCount++; } } // Close the archive if(hMpq != NULL) SFileCloseArchive(hMpq); hMpq = NULL; // Reopen the archive to catch any asserts if(nError == ERROR_SUCCESS) nError = OpenExistingArchiveWithCopy(&Logger, NULL, szPlainName, &hMpq); // Check that (listfile) is at the end if(nError == ERROR_SUCCESS) { CheckIfFileIsPresent(&Logger, hMpq, LISTFILE_NAME, true); CheckIfFileIsPresent(&Logger, hMpq, ATTRIBUTES_NAME, true); SFileCloseArchive(hMpq); } return nError; } //----------------------------------------------------------------------------- // Comparing two directories, creating links #define LINK_COMPARE_BLOCK_SIZE 0x200 static int CreateArchiveLinkFile(const char * szFullPath1, const char * szFullPath2, const char * szFileHash) { TFileStream * pStream; char szLinkData[MAX_PATH + 0x80]; char szLinkFile[MAX_PATH]; char szLinkPath[MAX_PATH]; int nLength; // Construct the link file name CalculateRelativePath(szFullPath1, szFullPath2, szLinkPath); sprintf(szLinkFile, "%s.link", szFullPath2); // Format the content of the link file nLength = sprintf(szLinkData, "LINK:%s\x0D\x0ASHA1:%s", szLinkPath, szFileHash); // Create the link file pStream = FileStream_CreateFileA(szLinkFile, 0); if(pStream == NULL) return GetLastError(); // Write the content of the link file FileStream_Write(pStream, NULL, szLinkData, (DWORD)nLength); FileStream_Close(pStream); return ERROR_SUCCESS; } static int ForEachFile_CreateArchiveLink(const char * szFullPath1, const char * szFullPath2) { TLogHelper Logger("CreateMpqLink", GetShortPlainName(szFullPath2)); char szFileHash1[0x40]; char szFileHash2[0x40]; int nError; // Prevent logger from witing any result messages Logger.bDontPrintResult = true; // Create SHA1 of both files nError = CalculateFileSha1(&Logger, szFullPath1, szFileHash1); if(nError == ERROR_SUCCESS) { nError = CalculateFileSha1(&Logger, szFullPath2, szFileHash2); if(nError == ERROR_SUCCESS) { // If the hashes are identical, we can create link if(!strcmp(szFileHash1, szFileHash2)) { nError = CreateArchiveLinkFile(szFullPath1, szFullPath2, szFileHash1); if(nError == ERROR_SUCCESS) { Logger.PrintMessage("Created link to %s", szFullPath2); } } } } return ERROR_SUCCESS; } //----------------------------------------------------------------------------- // Main int main(int argc, char * argv[]) { int nError = ERROR_SUCCESS; #if defined(_MSC_VER) && defined(_DEBUG) _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF); #endif // defined(_MSC_VER) && defined(_DEBUG) // Initialize storage and mix the random number generator printf("==== Test Suite for StormLib version %s ====\n", STORMLIB_VERSION_STRING); nError = InitializeMpqDirectory(argv, argc); // Not a test, but rather a tool for creating links to duplicated files // if(nError == ERROR_SUCCESS) // nError = FindFilePairs(ForEachFile_CreateArchiveLink, "2004 - WoW\\06080", "2004 - WoW\\06299"); // Search all testing archives and verify their SHA1 hash // if(nError == ERROR_SUCCESS) // nError = FindFiles(ForEachFile_VerifyFileChecksum, szMpqSubDir); // Test reading linear file without bitmap if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("MPQ_2013_v4_alternate-original.MPQ", 0); // Test reading linear file without bitmap (read only) if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("MPQ_2013_v4_alternate-original.MPQ", STREAM_FLAG_READ_ONLY); // Test reading linear file with bitmap if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("MPQ_2013_v4_alternate-complete.MPQ", STREAM_FLAG_USE_BITMAP); // Test reading partial file if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("part-file://MPQ_2009_v2_WoW_patch.MPQ.part", 0); // Test reading Block4K file if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("blk4-file://streaming/model.MPQ.0", STREAM_PROVIDER_BLOCK4); // Test reading encrypted file if(nError == ERROR_SUCCESS) nError = TestFileStreamOperations("mpqe-file://MPQ_2011_v2_EncryptedMpq.MPQE", STREAM_PROVIDER_MPQE); // Open a stream, paired with local master. The mirror file is created new if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("part-file://MPQ_2009_v1_patch-created.MPQ.part", "MPQ_2009_v1_patch-original.MPQ", false); // Open a stream, paired with local master. Only part of the mirror exists if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("part-file://MPQ_2009_v1_patch-partial.MPQ.part", "MPQ_2009_v1_patch-original.MPQ", true); // Open a stream, paired with local master. Only part of the mirror exists if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("part-file://MPQ_2009_v1_patch-complete.MPQ.part", "MPQ_2009_v1_patch-original.MPQ", true); // Open a stream, paired with local master if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("MPQ_2013_v4_alternate-created.MPQ", "MPQ_2013_v4_alternate-original.MPQ", false); // Open a stream, paired with local master if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("MPQ_2013_v4_alternate-incomplete.MPQ", "MPQ_2013_v4_alternate-incomplete.MPQ", true); // Open a stream, paired with local master if(nError == ERROR_SUCCESS) nError = TestReadFile_MasterMirror("MPQ_2013_v4_alternate-complete.MPQ", "MPQ_2013_v4_alternate-original.MPQ", true); // Open a stream, paired with remote master (takes hell lot of time!) // if(nError == ERROR_SUCCESS) // nError = TestReadFile_MasterMirror("MPQ_2013_v4_alternate-downloaded.MPQ", "http://www.zezula.net\\mpqs\\alternate.zip", false); // Search in listfile if(nError == ERROR_SUCCESS) nError = TestSearchListFile("ListFile_Blizzard.txt"); // Test opening local file with SFileOpenFileEx if(nError == ERROR_SUCCESS) nError = TestOpenLocalFile("ListFile_Blizzard.txt"); // Test working with an archive that has no listfile if(nError == ERROR_SUCCESS) nError = TestOpenFile_OpenById("MPQ_1997_v1_Diablo1_DIABDAT.MPQ"); // Open an empty archive (found in WoW cache - it's just a header) if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2012_v2_EmptyMpq.MPQ"); // Open an empty archive (created artificially - it's just a header) if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2013_v4_EmptyMpq.MPQ"); // Open an empty archive (found in WoW cache - it's just a header) if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2013_v4_patch-base-16357.MPQ"); // Open an empty archive (A buggy MPQ with invalid HET entry count) if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2011_v4_InvalidHetEntryCount.MPQ"); // Open a truncated archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_BlockTableCut.MPQ"); // Open an Warcraft III map locked by a protector if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_ProtectedMap_HashTable_FakeValid.w3x"); // Open an Warcraft III map locked by a protector if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_ProtectedMap_InvalidUserData.w3x"); // Open an Warcraft III map locked by a protector if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_ProtectedMap_InvalidMpqFormat.w3x"); // Open a MPQ that actually has user data if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2010_v2_HasUserData.s2ma"); // Open an Warcraft III map locked by the Spazzler protector if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_ProtectedMap_Spazzler.w3x"); if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2014_v1_ProtectedMap_Spazzler2.w3x"); if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2014_v1_ProtectedMap_Spazzler3.w3x"); // Open an Warcraft III map locked by the BOBA protector if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2002_v1_ProtectedMap_BOBA.w3m"); // Open an Warcraft III map whose "(attributes)" file has (BlockTableSize-1) entries if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2014_v1_AttributesOneEntryLess.w3x"); // Open a MPQ archive v 3.0 if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPQ_2010_v3_expansion-locale-frFR.MPQ"); // Open an encrypted archive from Starcraft II installer if(nError == ERROR_SUCCESS) nError = TestOpenArchive("mpqe-file://MPQ_2011_v2_EncryptedMpq.MPQE"); // Open a MPK archive from Longwu online if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPx_2013_v1_LongwuOnline.mpk"); // Open a SQP archive from War of the Immortals if(nError == ERROR_SUCCESS) nError = TestOpenArchive("MPx_2013_v1_WarOfTheImmortals.sqp", "ListFile_WarOfTheImmortals.txt"); // Open a partial MPQ with compressed hash table if(nError == ERROR_SUCCESS) nError = TestOpenArchive("part-file://MPQ_2010_v2_HashTableCompressed.MPQ.part"); // Open the multi-file archive with wrong prefix to see how StormLib deals with it if(nError == ERROR_SUCCESS) nError = TestOpenArchive_WillFail("flat-file://streaming/model.MPQ.0"); // Open an archive that is merged with multiple files if(nError == ERROR_SUCCESS) nError = TestOpenArchive("blk4-file://streaming/model.MPQ.0"); // Open every MPQ that we have in the storage if(nError == ERROR_SUCCESS) nError = FindFiles(ForEachFile_OpenArchive, NULL); // Test on an archive that has been invalidated by extending an old valid MPQ if(nError == ERROR_SUCCESS) nError = TestOpenArchive_Corrupt("MPQ_2013_vX_Battle.net.MPQ"); // Open a patched archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_Patched(PatchList_WoW_OldWorld13286, "OldWorld\\World\\Model.blob", 2); // Open a patched archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_Patched(PatchList_WoW15050, "World\\Model.blob", 8); // Open a patched archive. The file is in each patch as full, so there is 0 patches in the chain if(nError == ERROR_SUCCESS) nError = TestOpenArchive_Patched(PatchList_WoW16965, "DBFilesClient\\BattlePetNPCTeamMember.db2", 0); // Check the opening archive for read-only if(nError == ERROR_SUCCESS) nError = TestOpenArchive_ReadOnly("MPQ_1997_v1_Diablo1_DIABDAT.MPQ", true); // Check the opening archive for read-only if(nError == ERROR_SUCCESS) nError = TestOpenArchive_ReadOnly("MPQ_1997_v1_Diablo1_DIABDAT.MPQ", false); // Check the SFileGetFileInfo function if(nError == ERROR_SUCCESS) nError = TestOpenArchive_GetFileInfo("MPQ_2002_v1_StrongSignature.w3m", "MPQ_2013_v4_SC2_EmptyMap.SC2Map"); // Downloadable MPQ archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_MasterMirror("part-file://MPQ_2009_v1_patch-partial.MPQ.part", "MPQ_2009_v1_patch-original.MPQ", "world\\Azeroth\\DEADMINES\\PASSIVEDOODADS\\GOBLINMELTINGPOT\\DUST2.BLP", false); // Downloadable MPQ archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_MasterMirror("MPQ_2013_v4_alternate-downloaded.MPQ", "MPQ_2013_v4_alternate-original.MPQ", "alternate\\DUNGEONS\\TEXTURES\\ICECROWN\\GATE\\jlo_IceC_Floor_Thrown.blp", false); // Check archive signature if(nError == ERROR_SUCCESS) nError = TestOpenArchive_VerifySignature("MPQ_1999_v1_WeakSignature.exe", "War2Patch_202.exe"); // Check archive signature if(nError == ERROR_SUCCESS) nError = TestOpenArchive_VerifySignature("MPQ_2002_v1_StrongSignature.w3m", "(10)DustwallowKeys.w3m"); // Compact the archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_CraftedUserData("MPQ_2010_v3_expansion-locale-frFR.MPQ", "StormLibTest_CraftedMpq1_v3.mpq"); // Open a MPQ (add custom user data to it) if(nError == ERROR_SUCCESS) nError = TestOpenArchive_CraftedUserData("MPQ_2013_v4_SC2_EmptyMap.SC2Map", "StormLibTest_CraftedMpq2_v4.mpq"); // Open a MPQ (add custom user data to it) if(nError == ERROR_SUCCESS) nError = TestOpenArchive_CraftedUserData("MPQ_2013_v4_expansion1.MPQ", "StormLibTest_CraftedMpq3_v4.mpq"); // if(nError == ERROR_SUCCESS) // nError = TestOpenArchive_CompactingTest("MPQ_2014_v1_CompactTest.w3x", "ListFile_Blizzard.txt"); // Test modifying file with no (listfile) and no (attributes) if(nError == ERROR_SUCCESS) nError = TestAddFile_ListFileTest("MPQ_1997_v1_Diablo1_DIABDAT.MPQ", false, false); // Test modifying an archive that contains (listfile) and (attributes) if(nError == ERROR_SUCCESS) nError = TestAddFile_ListFileTest("MPQ_2013_v4_SC2_EmptyMap.SC2Map", true, true); // Test archive compacting // Create an empty archive v2 if(nError == ERROR_SUCCESS) nError = TestCreateArchive_EmptyMpq("StormLibTest_EmptyMpq_v2.mpq", MPQ_CREATE_ARCHIVE_V2 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES); // Create an empty archive v4 if(nError == ERROR_SUCCESS) nError = TestCreateArchive_EmptyMpq("StormLibTest_EmptyMpq_v4.mpq", MPQ_CREATE_ARCHIVE_V4 | MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES); // Test creating of an archive the same way like MPQ Editor does if(nError == ERROR_SUCCESS) nError = TestCreateArchive_TestGaps("StormLibTest_GapsTest.mpq"); // Sign an existing non-signed archive if(nError == ERROR_SUCCESS) nError = TestOpenArchive_SignExisting("MPQ_1998_v1_StarDat.mpq"); // Open a signed archive, add a file and verify the signature if(nError == ERROR_SUCCESS) nError = TestOpenArchive_ModifySigned("MPQ_1999_v1_WeakSignature.exe", "War2Patch_202.exe"); // Create new archive and sign it if(nError == ERROR_SUCCESS) nError = TestCreateArchive_Signed("MPQ_1999_v1_WeakSigned1.mpq", true); if(nError == ERROR_SUCCESS) nError = TestCreateArchive_Signed("MPQ_1999_v1_WeakSigned2.mpq", false); // Test creating of an archive the same way like MPQ Editor does if(nError == ERROR_SUCCESS) nError = TestCreateArchive_MpqEditor("StormLibTest_MpqEditorTest.mpq", "AddedFile.exe"); // Create an archive and fill it with files up to the max file count if(nError == ERROR_SUCCESS) nError = TestCreateArchive_FillArchive("StormLibTest_FileTableFull.mpq", 0); // Create an archive and fill it with files up to the max file count if(nError == ERROR_SUCCESS) nError = TestCreateArchive_FillArchive("StormLibTest_FileTableFull.mpq", MPQ_CREATE_LISTFILE); // Create an archive and fill it with files up to the max file count if(nError == ERROR_SUCCESS) nError = TestCreateArchive_FillArchive("StormLibTest_FileTableFull.mpq", MPQ_CREATE_ATTRIBUTES); // Create an archive and fill it with files up to the max file count if(nError == ERROR_SUCCESS) nError = TestCreateArchive_FillArchive("StormLibTest_FileTableFull.mpq", MPQ_CREATE_ATTRIBUTES | MPQ_CREATE_LISTFILE); // Create an archive, and increment max file count several times if(nError == ERROR_SUCCESS) nError = TestCreateArchive_IncMaxFileCount("StormLibTest_IncMaxFileCount.mpq"); // Create a MPQ archive with UNICODE names if(nError == ERROR_SUCCESS) nError = TestCreateArchive_UnicodeNames(); // Create a MPQ file, add files with various flags if(nError == ERROR_SUCCESS) nError = TestCreateArchive_FileFlagTest("StormLibTest_FileFlagTest.mpq"); // Create a MPQ file, add a mono-WAVE file with various compressions if(nError == ERROR_SUCCESS) nError = TestCreateArchive_WaveCompressionsTest("StormLibTest_AddWaveMonoTest.mpq", "AddFile-Mono.wav"); // Create a MPQ file, add a mono-WAVE with 8 bits per sample file with various compressions if(nError == ERROR_SUCCESS) nError = TestCreateArchive_WaveCompressionsTest("StormLibTest_AddWaveMonoBadTest.mpq", "AddFile-MonoBad.wav"); // Create a MPQ file, add a stereo-WAVE file with various compressions if(nError == ERROR_SUCCESS) nError = TestCreateArchive_WaveCompressionsTest("StormLibTest_AddWaveStereoTest.mpq", "AddFile-Stereo.wav"); // Check if the listfile is always created at the end of the file table in the archive if(nError == ERROR_SUCCESS) nError = TestCreateArchive_ListFilePos("StormLibTest_ListFilePos.mpq"); // Open a MPQ (add custom user data to it) if(nError == ERROR_SUCCESS) nError = TestCreateArchive_BigArchive("StormLibTest_BigArchive_v4.mpq"); return nError; }