Fix crash in silver-bun (#679)

This commit is contained in:
Jack 2024-03-08 23:20:49 +00:00 committed by GitHub
parent 85a2fb9c56
commit d3ee91c1f5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions

View File

@ -15,11 +15,21 @@
//-----------------------------------------------------------------------------
CModule::CModule(HMODULE hModule)
{
m_pModuleBase = reinterpret_cast<uintptr_t>(hModule);
MODULEINFO mInfo {0};
if (hModule && hModule != INVALID_HANDLE_VALUE)
GetModuleInformation(GetCurrentProcess(), hModule, &mInfo, sizeof(MODULEINFO));
m_nModuleSize = static_cast<size_t>(mInfo.SizeOfImage);
m_pModuleBase = reinterpret_cast<uintptr_t>(mInfo.lpBaseOfDll);
if (!m_nModuleSize || !m_pModuleBase)
return;
CHAR szModuleName[MAX_PATH];
DWORD dwSize = GetModuleFileNameA(hModule, szModuleName, sizeof(szModuleName));
m_ModuleName = strrchr(szModuleName, '\\') + 1;
char* chLast = strrchr(szModuleName, '\\');
m_ModuleName = chLast == nullptr ? szModuleName : chLast + 1;
Init();