From ce074eaf7ace51fdb642f491e217d4f50e30150e Mon Sep 17 00:00:00 2001
From: Ladislav Zezula <ladislav.zezula@avast.com>
Date: Fri, 2 Jul 2021 23:53:06 +0200
Subject: Fixed infinite loop in BZIP decompression

---
 test/StormTest.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'test')

diff --git a/test/StormTest.cpp b/test/StormTest.cpp
index 5b32dc1..8786b06 100644
--- a/test/StormTest.cpp
+++ b/test/StormTest.cpp
@@ -342,6 +342,8 @@ static bool IsMpqExtension(LPCTSTR szFileName)
             return true;
         if(!_tcsicmp(szExtension, _T(".SC2Mod")))
             return true;
+        if(!_tcsicmp(szExtension, _T(".SC2Replay")))
+            return true;
         if(!_tcsicmp(szExtension, _T(".0")))        // .MPQ.0
             return true;
 //      if(!_tcsicmp(szExtension, ".link"))
@@ -1522,6 +1524,9 @@ static TFileData * LoadMpqFile(TLogHelper * pLogger, HANDLE hMpq, LPCSTR szFileN
         // Load the entire file
         if(dwErrCode == ERROR_SUCCESS)
         {
+			if(!stricmp(szFileName, "replay.game.events"))
+				__debugbreak();
+
             // Read the file data
             SFileReadFile(hFile, pFileData->FileData, dwFileSizeLo, &dwBytesRead, NULL);
             if(dwBytesRead != dwFileSizeLo)
@@ -4218,6 +4223,9 @@ static const TEST_INFO TestList_MasterMirror[] =
 
 static const TEST_INFO Test_Mpqs[] =
 {
+
+    {_T("ProblemMpqArchive.SC2Replay"),        NULL, 0, "replay.game.events"},
+
     // Correct or damaged archives
     {_T("MPQ_1997_v1_Diablo1_DIABDAT.MPQ"),    NULL, 0, "music\\dintro.wav", "File00000023.xxx"},
     {_T("MPQ_2016_v1_D2XP_IX86_1xx_114a.mpq"), NULL, 0, "waitingroombkgd.dc6"},                   // Update MPQ from Diablo II (patch 2016)
@@ -4296,6 +4304,7 @@ int _tmain(int argc, TCHAR * argv[])
     // Open all files from the command line
     //
 
+	TestArchive(_T("ProblemMpqArchive.SC2Replay"), NULL, 0, "replay.game.events", NULL);
     for(int i = 1; i < argc; i++)
     {
         ForEachFile_OpenArchive(argv[i]);
-- 
cgit v1.2.3