mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
5fc38d5c96
In rare occasions (depending on the installation path), the regex might fail to extract the level name from the VPK file name. If the path has a similar pattern, or certain characters, it will capture the wrong string. This fix has been tested, and confirmed to fix the bug.
493 lines
16 KiB
C++
493 lines
16 KiB
C++
//=============================================================================//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//=============================================================================//
|
|
|
|
#include "core/stdafx.h"
|
|
#include "tier0/memstd.h"
|
|
#include "tier0/jobthread.h"
|
|
#include "tier2/fileutils.h"
|
|
#include "engine/sys_dll2.h"
|
|
#include "engine/host_cmd.h"
|
|
#include "engine/cmodel_bsp.h"
|
|
#include "rtech/rtech_utils.h"
|
|
#include "rtech/rtech_game.h"
|
|
#include "vpc/keyvalues.h"
|
|
#include "datacache/mdlcache.h"
|
|
#include "filesystem/filesystem.h"
|
|
#ifndef DEDICATED
|
|
#include "client/clientstate.h"
|
|
#endif // !DEDICATED
|
|
|
|
vector<string> g_InstalledMaps;
|
|
string s_LevelName;
|
|
|
|
std::regex s_ArchiveRegex{ R"([^_]*_(.*)(.bsp.pak000_dir).*)" };
|
|
|
|
bool s_bLevelResourceInitialized = false;
|
|
bool s_bBasePaksInitialized = false;
|
|
KeyValues* s_pLevelSetKV = nullptr;
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: checks if level has changed
|
|
// Input : *pszLevelName -
|
|
// Output : true if level name deviates from previous level
|
|
//-----------------------------------------------------------------------------
|
|
bool Mod_LevelHasChanged(const char* pszLevelName)
|
|
{
|
|
return (s_LevelName.compare(pszLevelName) != 0);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: gets all installed maps
|
|
//-----------------------------------------------------------------------------
|
|
void Mod_GetAllInstalledMaps()
|
|
{
|
|
CUtlVector<CUtlString> fileList;
|
|
AddFilesToList(fileList, "vpk", "vpk", nullptr, '/');
|
|
|
|
std::cmatch regexMatches;
|
|
std::lock_guard<std::mutex> l(g_InstalledMapsMutex);
|
|
|
|
g_InstalledMaps.clear(); // Clear current list.
|
|
|
|
FOR_EACH_VEC(fileList, i)
|
|
{
|
|
const CUtlString& filePath = fileList[i];
|
|
|
|
const char* pFilePath = filePath.Get();
|
|
const char* pFileName = strrchr(pFilePath, '/')+1;
|
|
|
|
// Should always point right in front of the last
|
|
// slash, as the files are loaded from 'vpk/'.
|
|
Assert(pFileName);
|
|
|
|
std::regex_search(pFileName, regexMatches, s_ArchiveRegex);
|
|
|
|
if (!regexMatches.empty())
|
|
{
|
|
if (regexMatches[1].str().compare("frontend") == 0)
|
|
continue; // Frontend contains no BSP's.
|
|
|
|
else if (regexMatches[1].str().compare("mp_common") == 0)
|
|
{
|
|
if (std::find(g_InstalledMaps.begin(), g_InstalledMaps.end(), "mp_lobby") == g_InstalledMaps.end())
|
|
g_InstalledMaps.push_back("mp_lobby");
|
|
continue; // Common contains mp_lobby.
|
|
}
|
|
|
|
if (std::find(g_InstalledMaps.begin(), g_InstalledMaps.end(), regexMatches[1].str()) == g_InstalledMaps.end())
|
|
g_InstalledMaps.push_back(regexMatches[1].str());
|
|
}
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: gets the queued pak handles
|
|
// Input : *a1 -
|
|
// *a2 -
|
|
// a3 -
|
|
// Output : __int64
|
|
//-----------------------------------------------------------------------------
|
|
__int64 __fastcall Mod_GetQueuedPakHandle(char* a1, char* a2, __int64 a3)
|
|
{
|
|
char v3; // al
|
|
signed int v4; // er11
|
|
__int64 v5; // r10
|
|
char* v6; // r9
|
|
signed __int64 v7; // rdx
|
|
char v8; // al
|
|
char* v10; // r8
|
|
char* v11; // r8
|
|
|
|
v3 = *a2;
|
|
v4 = 0;
|
|
*a1 = *a2;
|
|
v5 = 0i64;
|
|
if (v3)
|
|
{
|
|
v6 = a1;
|
|
v7 = a2 - a1;
|
|
while (1)
|
|
{
|
|
++v5;
|
|
++v6;
|
|
if (v5 == a3)
|
|
break;
|
|
v8 = v6[v7];
|
|
*v6 = v8;
|
|
if (!v8)
|
|
return v5;
|
|
}
|
|
*(v6 - 1) = 0;
|
|
if (--v5)
|
|
{
|
|
v10 = &a1[v5 - 1];
|
|
if ((*v10 & 0xC0) == 0x80)
|
|
{
|
|
do
|
|
++v4;
|
|
while ((v10[-v4] & 0xC0) == 0x80);
|
|
}
|
|
v11 = &v10[-v4];
|
|
if (v4 != (signed int)((0xE5000000 >> (((unsigned __int8)*v11 >> 3) & 0x1E)) & 3))
|
|
{
|
|
*v11 = 0;
|
|
v5 -= v4;
|
|
}
|
|
}
|
|
}
|
|
return v5;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: processes queued pak files
|
|
//-----------------------------------------------------------------------------
|
|
void Mod_ProcessPakQueue()
|
|
{
|
|
char v0; // bl
|
|
char** v1; // r10
|
|
__int64 i; // er9
|
|
char* v3; // rcx
|
|
signed __int64 v4; // r8
|
|
int v5; // eax
|
|
int v6; // edx
|
|
__int64 v7; // eax
|
|
__int64 v8; // rbp
|
|
__int64 v9; // rsi
|
|
char* v10; // rbx
|
|
unsigned int v11; // ecx
|
|
__int64 v12; // rax
|
|
int v13; // edi
|
|
char v14; // al
|
|
char* v15; // rbx
|
|
__int64 v16; // edi
|
|
char* v17; // rsi
|
|
char* v18; // rax
|
|
int v19; // ecx
|
|
int v20; // er8
|
|
int v21; // ecx
|
|
__int64 v22; // rdx
|
|
__int64 v24{}; // rdx
|
|
__int64 v25{}; // rcx
|
|
|
|
v0 = 0;
|
|
#ifndef DEDICATED
|
|
bool bUnconnected = !(*g_pClientState_Shifted)->IsConnected();
|
|
#else // !DEDICATED
|
|
bool bUnconnected = true; // Always true for dedicated.
|
|
#endif
|
|
|
|
if (*(float*)&*dword_14B383420 == 1.0 && *qword_167ED7BB8 && bUnconnected)
|
|
{
|
|
*byte_16709DDDF = 0;
|
|
v0 = 1;
|
|
}
|
|
else if (*byte_16709DDDF)
|
|
{
|
|
return;
|
|
}
|
|
if (FileSystem()->ResetItemCache() && !*dword_1634F445C)
|
|
{
|
|
v1 = &*off_141874660;
|
|
for (i = 0; i < 5; ++i)
|
|
{
|
|
if (*((_BYTE*)v1 - 268))
|
|
break;
|
|
v3 = (char*)&*unk_141874555 + 280 * i;
|
|
v4 = *v1 - v3;
|
|
do
|
|
{
|
|
v5 = (unsigned __int8)v3[v4];
|
|
v6 = (unsigned __int8)*v3 - v5;
|
|
if (v6)
|
|
break;
|
|
++v3;
|
|
} while (v5);
|
|
if (v6)
|
|
break;
|
|
v1 += 35;
|
|
}
|
|
v7 = 0;
|
|
if (!v0)
|
|
v7 = i;
|
|
v8 = v7;
|
|
if (v7 <= 4i64)
|
|
{
|
|
v9 = 4i64;
|
|
v10 = (char*)&*unk_1418749B0;
|
|
do
|
|
{
|
|
if (v10[5])
|
|
{
|
|
v11 = *(_DWORD*)v10;
|
|
v12 = *(_DWORD*)v10 & 0x1FF;
|
|
v10[4] = 1;
|
|
if (*((_DWORD*)&*g_pLoadedPakInfo + 46 * v12) == v11)
|
|
{
|
|
v13 = *((_DWORD*)&*g_pLoadedPakInfo + 46 * v12 + 1);
|
|
v14 = v10[4];
|
|
}
|
|
else
|
|
{
|
|
v13 = 14;
|
|
v14 = 1;
|
|
}
|
|
if (!v14 || v13 == 9)
|
|
{
|
|
// SDK pak files must be unloaded before the engine pak files,
|
|
// as we reference assets within engine pak files.
|
|
const RPakLoadedInfo_t* pLoadedPakInfo = g_pRTech->GetPakLoadedInfo(*(RPakHandle_t*)v10);
|
|
if (pLoadedPakInfo)
|
|
{
|
|
const char* pszLoadedPakName = pLoadedPakInfo->m_pszFileName;
|
|
|
|
if (strcmp(pszLoadedPakName, "common_mp.rpak") == 0 ||
|
|
strcmp(pszLoadedPakName, "common_sp.rpak") == 0 ||
|
|
strcmp(pszLoadedPakName, "common_pve.rpak") == 0)
|
|
{
|
|
const RPakLoadedInfo_t* pLoadedSdkPak = g_pRTech->GetPakLoadedInfo("common_sdk.rpak");
|
|
|
|
if (pLoadedSdkPak) // Only unload if sdk pak file is loaded.
|
|
g_pakLoadApi->UnloadPak(pLoadedSdkPak->m_nHandle);
|
|
|
|
}
|
|
#ifndef DEDICATED
|
|
else if (strcmp(pszLoadedPakName, "ui_mp.rpak") == 0)
|
|
{
|
|
const RPakLoadedInfo_t* pLoadedSdkPak = g_pRTech->GetPakLoadedInfo("ui_sdk.rpak");
|
|
|
|
if (pLoadedSdkPak) // Only unload if sdk pak file is loaded.
|
|
g_pakLoadApi->UnloadPak(pLoadedSdkPak->m_nHandle);
|
|
}
|
|
#endif // !DEDICATED
|
|
}
|
|
|
|
// The old gather props is set if a model couldn't be
|
|
// loaded properly. If we unload level assets, we just
|
|
// enable the new implementation again and re-evaluate
|
|
// on the next level load. If we load a missing/bad
|
|
// model again, we toggle the old implementation as
|
|
// the helper functions for this implementation have
|
|
// been restored in the SDK, and modified to support
|
|
// stubbing missing model assets. See the function
|
|
// 'CMDLCache::GetErrorModel' for more information.
|
|
if (old_gather_props->GetBool())
|
|
old_gather_props->SetValue(false);
|
|
|
|
g_pakLoadApi->UnloadPak(*(RPakHandle_t*)v10);
|
|
Mod_UnloadPakFile(); // Unload mod pak files.
|
|
|
|
if (s_pLevelSetKV)
|
|
{
|
|
// Delete current level settings if we drop all paks..
|
|
s_pLevelSetKV->DeleteThis();
|
|
s_pLevelSetKV = nullptr;
|
|
}
|
|
}
|
|
if (v13 && (unsigned int)(v13 - 13) > 1)
|
|
return;
|
|
*((_WORD*)v10 + 2) = 0;
|
|
*(_DWORD*)v10 = 0xFFFFFFFF;
|
|
}
|
|
--v9;
|
|
v10 -= 280;
|
|
} while (v9 >= v8);
|
|
}
|
|
*byte_16709DDDF = 1;
|
|
v15 = (char*)&*unk_141874550;
|
|
v16 = 0;
|
|
while (1)
|
|
{
|
|
v17 = (char*)&*unk_141874550 + 280 * v16 + 5;
|
|
v18 = v17;
|
|
do
|
|
{
|
|
v19 = (unsigned __int8)v18[*((_QWORD*)v15 + 34) - (_QWORD)v17];
|
|
v20 = (unsigned __int8)*v18 - v19;
|
|
if (v20)
|
|
break;
|
|
++v18;
|
|
} while (v19);
|
|
if (!v20)
|
|
goto LABEL_37;
|
|
Mod_GetQueuedPakHandle(v17, *((char**)v15 + 34), 260i64);
|
|
if (v15[5])
|
|
break;
|
|
*(_DWORD*)v15 = 0xFFFFFFFF;
|
|
LABEL_40:
|
|
++v16;
|
|
v15 += 280;
|
|
if (v16 >= 5)
|
|
{
|
|
if (*byte_16709DDDF)
|
|
{
|
|
if (*g_pMTVFTaskItem)
|
|
{
|
|
if (!*(_BYTE*)(*g_pMTVFTaskItem + 4))
|
|
{
|
|
if (*qword_167ED7BC0 || WORD2(*g_pPakLoadJobID) != HIWORD(*g_pPakLoadJobID))
|
|
{
|
|
if (!JT_AcquireFifoLock(g_pPakFifoLock)
|
|
&& !(unsigned __int8)sub_14045BAC0((__int64(__fastcall*)(__int64, _DWORD*, __int64, _QWORD*))g_pPakFifoLockWrapper, g_pPakFifoLock, -1i64, 0i64))
|
|
{
|
|
sub_14045A1D0((unsigned __int8(__fastcall*)(_QWORD))g_pPakFifoLockWrapper, g_pPakFifoLock, -1i64, 0i64, 0i64, 1);
|
|
}
|
|
|
|
sub_140441220(v25, v24);
|
|
if (ThreadInMainThread())
|
|
{
|
|
if (*g_bPakFifoLockAcquired)
|
|
{
|
|
*g_bPakFifoLockAcquired = 0;
|
|
JT_ReleaseFifoLock(g_pPakFifoLock);
|
|
}
|
|
}
|
|
JT_ReleaseFifoLock(g_pPakFifoLock);
|
|
}
|
|
FileSystem()->ResetItemCacheSize(256);
|
|
FileSystem()->PrecacheTaskItem(*g_pMTVFTaskItem);
|
|
}
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
}
|
|
if (strcmp(v17, "mp_lobby.rpak") == 0)
|
|
s_bBasePaksInitialized = true;
|
|
|
|
if (s_bBasePaksInitialized && !s_bLevelResourceInitialized)
|
|
{
|
|
Mod_PreloadLevelPaks(s_LevelName.c_str());
|
|
s_bLevelResourceInitialized = true;
|
|
}
|
|
*(_DWORD*)v15 = g_pakLoadApi->LoadAsync(v17, AlignedMemAlloc(), 4, 0);
|
|
|
|
if (strcmp(v17, "common_mp.rpak") == 0 || strcmp(v17, "common_sp.rpak") == 0 || strcmp(v17, "common_pve.rpak") == 0)
|
|
g_pakLoadApi->LoadAsync("common_sdk.rpak", AlignedMemAlloc(), 4, 0);
|
|
#ifndef DEDICATED
|
|
if (strcmp(v17, "ui_mp.rpak") == 0)
|
|
g_pakLoadApi->LoadAsync("ui_sdk.rpak", AlignedMemAlloc(), 4, 0);
|
|
#endif // !DEDICATED
|
|
|
|
LABEL_37:
|
|
v21 = *(_DWORD*)v15;
|
|
if (*(_DWORD*)v15 != INVALID_PAK_HANDLE)
|
|
{
|
|
#if defined (GAMEDLL_S0) || defined (GAMEDLL_S1) || defined (GAMEDLL_S2)
|
|
v22 = 232i64 * (v21 & 0x1FF);
|
|
#else
|
|
v22 = 184i64 * (v21 & 0x1FF);
|
|
#endif
|
|
if (*(_DWORD*)((char*)&*g_pLoadedPakInfo + v22) != _DWORD(v21) || ((*(_DWORD*)((char*)&*g_pLoadedPakInfo + v22 + 4) - 9) & 0xFFFFFFFB) != 0)
|
|
{
|
|
*byte_16709DDDF = 0; return;
|
|
}
|
|
}
|
|
goto LABEL_40;
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: load assets for level with fifolock.
|
|
// Input : *szLevelName -
|
|
// Output : true on success, false on failure
|
|
//-----------------------------------------------------------------------------
|
|
void Mod_LoadPakForMap(const char* pszLevelName)
|
|
{
|
|
if (Mod_LevelHasChanged(pszLevelName))
|
|
s_bLevelResourceInitialized = false;
|
|
|
|
s_LevelName = pszLevelName;
|
|
|
|
// Dedicated should not load loadscreens.
|
|
#ifndef DEDICATED
|
|
v_Mod_LoadPakForMap(pszLevelName);
|
|
#endif // !DEDICATED
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: loads the level settings file, returns current if level hasn't changed.
|
|
// Input : *pszLevelName -
|
|
// Output : KeyValues*
|
|
//-----------------------------------------------------------------------------
|
|
KeyValues* Mod_GetLevelSettings(const char* pszLevelName)
|
|
{
|
|
if (s_pLevelSetKV)
|
|
{
|
|
if (s_bLevelResourceInitialized)
|
|
return s_pLevelSetKV;
|
|
|
|
s_pLevelSetKV->DeleteThis();
|
|
}
|
|
|
|
char szPathBuffer[MAX_PATH];
|
|
snprintf(szPathBuffer, sizeof(szPathBuffer), "scripts/levels/settings/%s.kv", pszLevelName);
|
|
|
|
s_pLevelSetKV = FileSystem()->LoadKeyValues(IFileSystem::TYPE_LEVELSETTINGS, szPathBuffer, "GAME");
|
|
return s_pLevelSetKV;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: loads required pakfile assets for specified BSP level
|
|
// Input : &svSetFile -
|
|
//-----------------------------------------------------------------------------
|
|
void Mod_PreloadLevelPaks(const char* pszLevelName)
|
|
{
|
|
KeyValues* pSettingsKV = Mod_GetLevelSettings(pszLevelName);
|
|
|
|
if (!pSettingsKV)
|
|
return;
|
|
|
|
KeyValues* pPakListKV = pSettingsKV->FindKey("PakList");
|
|
|
|
if (!pPakListKV)
|
|
return;
|
|
|
|
char szPathBuffer[MAX_PATH];
|
|
|
|
for (KeyValues* pSubKey = pPakListKV->GetFirstSubKey(); pSubKey != nullptr; pSubKey = pSubKey->GetNextKey())
|
|
{
|
|
if (!pSubKey->GetBool())
|
|
continue;
|
|
|
|
snprintf(szPathBuffer, sizeof(szPathBuffer), "%s.rpak", pSubKey->GetName());
|
|
RPakHandle_t nPakId = g_pakLoadApi->LoadAsync(szPathBuffer, AlignedMemAlloc(), 4, 0);
|
|
|
|
if (nPakId == INVALID_PAK_HANDLE)
|
|
Error(eDLL_T::ENGINE, NO_ERROR, "%s: unable to load pak '%s' results '%d'\n", __FUNCTION__, szPathBuffer, nPakId);
|
|
else
|
|
g_vLoadedPakHandle.AddToTail(nPakId);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Purpose: unloads all pakfiles loaded by the SDK
|
|
//-----------------------------------------------------------------------------
|
|
void Mod_UnloadPakFile(void)
|
|
{
|
|
for (const RPakHandle_t& it : g_vLoadedPakHandle)
|
|
{
|
|
if (it >= 0)
|
|
{
|
|
g_pakLoadApi->UnloadPak(it);
|
|
}
|
|
}
|
|
g_vLoadedPakHandle.Purge();
|
|
g_vBadMDLHandles.clear();
|
|
}
|
|
|
|
void VModel_BSP::Attach() const
|
|
{
|
|
DetourAttach((LPVOID*)&v_Mod_LoadPakForMap, &Mod_LoadPakForMap);
|
|
DetourAttach((LPVOID*)&v_Mod_ProcessPakQueue, &Mod_ProcessPakQueue);
|
|
}
|
|
|
|
void VModel_BSP::Detach() const
|
|
{
|
|
DetourDetach((LPVOID*)&v_Mod_LoadPakForMap, &Mod_LoadPakForMap);
|
|
DetourDetach((LPVOID*)&v_Mod_ProcessPakQueue, &Mod_ProcessPakQueue);
|
|
} |