jonny_bro/SilentPatch
1
0
Fork 0
mirror of https://github.com/CookiePLMonster/SilentPatch.git synced 2025-01-01 16:53:01 +05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Updated ModuleList.hpp

Browse source
This commit is contained in:
Silent 2017-11-03 22:22:22 +01:00
parent a76b06a400
commit 614354ec8a
1 changed files with 34 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
SilentPatch/ModuleList.hpp
View file

@ -107,12 +107,32 @@ public:
private:
void EnumerateInternal( HMODULE* modules, size_t numModules )
{
size_t moduleNameLength = MAX_PATH;
wchar_t* moduleName = static_cast<wchar_t*>( malloc( moduleNameLength * sizeof(moduleName[0]) ) );
if ( moduleName != nullptr )
{
m_moduleList.reserve( numModules );
for ( size_t i = 0; i < numModules; i++ )
{
wchar_t moduleName[MAX_PATH];
if ( GetModuleFileNameW( *modules, moduleName, MAX_PATH ) != 0 )
// Obtain module name, with resizing if necessary
DWORD size;
while ( size = GetModuleFileNameW( *modules, moduleName, moduleNameLength ), size == moduleNameLength )
{
wchar_t* newName = static_cast<wchar_t*>( realloc( moduleName, 2 * moduleNameLength * sizeof(moduleName[0]) ) );
if ( newName != nullptr )
{
moduleName = newName;
moduleNameLength *= 2;
}
else
{
size = 0;
break;
}
}
if ( size != 0 )
{
const wchar_t* nameBegin = wcsrchr( moduleName, '\\' ) + 1;
const wchar_t* dotPos = wcsrchr( nameBegin, '.' );
@ -127,6 +147,9 @@ private:
}
modules++;
}
free( moduleName );
}
}
std::vector< std::pair<HMODULE, std::wstring> > m_moduleList;