mirror of
https://github.com/hasherezade/pe-sieve
synced 2026-06-08 14:34:52 +00:00
hasherezade
31 lines
897 B
C++
31 lines
897 B
C++
#include "modules_enum.h"
|
|
|
|
#include <psapi.h>
|
|
#pragma comment(lib,"psapi.lib")
|
|
|
|
size_t pesieve::util::enum_modules(IN HANDLE hProcess, IN OUT HMODULE hMods[], IN const DWORD hModsMax, IN DWORD filters) //throws exceptions
|
|
{
|
|
if (hProcess == nullptr) {
|
|
return 0;
|
|
}
|
|
const char err_msg[] = "Could not enumerate modules. ";
|
|
DWORD cbNeeded;
|
|
#ifdef _WIN64
|
|
if (!EnumProcessModulesEx(hProcess, hMods, hModsMax, &cbNeeded, filters)) {
|
|
throw std::runtime_error(err_msg);
|
|
return 0;
|
|
}
|
|
#else
|
|
/*
|
|
Some old, 32-bit versions of Windows do not have EnumProcessModulesEx,
|
|
but we can use EnumProcessModules for the 32-bit version: it will work the same and prevent the compatibility issues.
|
|
*/
|
|
if (!EnumProcessModules(hProcess, hMods, hModsMax, &cbNeeded)) {
|
|
throw std::runtime_error(err_msg);
|
|
return 0;
|
|
}
|
|
#endif
|
|
const size_t modules_count = cbNeeded / sizeof(HMODULE);
|
|
return modules_count;
|
|
}
|