mirror of https://github.com/vxunderground/VX-API
78 lines
1.9 KiB
C++
78 lines
1.9 KiB
C++
#include "Win32Helper.h"
|
|
|
|
HMODULE GetModuleHandleEx2A(_In_ LPCSTR lpModuleName)
|
|
{
|
|
PPEB Peb = GetPeb();
|
|
PLDR_MODULE Module = NULL;
|
|
CHAR wDllName[64] = { 0 };
|
|
PLIST_ENTRY Head = &Peb->LoaderData->InMemoryOrderModuleList;
|
|
PLIST_ENTRY Next = Head->Flink;
|
|
Module = (PLDR_MODULE)((PBYTE)Next - 16);
|
|
|
|
while (Next != Head)
|
|
{
|
|
Module = (PLDR_MODULE)((PBYTE)Next - 16);
|
|
if (Module->BaseDllName.Buffer != NULL)
|
|
{
|
|
ZeroMemoryEx(wDllName, sizeof(wDllName));
|
|
WCharStringToCharString(wDllName, Module->BaseDllName.Buffer, 64);
|
|
|
|
CHAR InitialModuleName[256] = { 0 };
|
|
CHAR IdentifiedModuleName[256] = { 0 };
|
|
|
|
if (StringCopyA(InitialModuleName, (PCHAR)lpModuleName) == NULL)
|
|
return NULL;
|
|
|
|
if (StringCopyA(IdentifiedModuleName, wDllName) == NULL)
|
|
return NULL;
|
|
|
|
PCHAR ComparisonObject1 = CaplockStringA(InitialModuleName);
|
|
PCHAR ComparisonObject2 = CaplockStringA(IdentifiedModuleName);
|
|
|
|
|
|
if (StringCompareA(ComparisonObject1, ComparisonObject2) == 0)
|
|
return (HMODULE)Module->BaseAddress;
|
|
}
|
|
|
|
Next = Next->Flink;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
HMODULE GetModuleHandleEx2W(_In_ LPCWSTR lpModuleName)
|
|
{
|
|
PPEB Peb = GetPeb();
|
|
PLDR_MODULE Module = NULL;
|
|
|
|
PLIST_ENTRY Head = &Peb->LoaderData->InMemoryOrderModuleList;
|
|
PLIST_ENTRY Next = Head->Flink;
|
|
Module = (PLDR_MODULE)((PBYTE)Next - 16);
|
|
|
|
while (Next != Head)
|
|
{
|
|
Module = (PLDR_MODULE)((PBYTE)Next - 16);
|
|
if (Module->BaseDllName.Buffer != NULL)
|
|
{
|
|
WCHAR InitialModuleName[256] = { 0 };
|
|
WCHAR IdentifiedModuleName[256] = { 0 };
|
|
|
|
if (StringCopyW(InitialModuleName, (PWCHAR)lpModuleName) == NULL)
|
|
return NULL;
|
|
|
|
if (StringCopyW(IdentifiedModuleName, Module->BaseDllName.Buffer) == NULL)
|
|
return NULL;
|
|
|
|
PWCHAR ComparisonObject1 = CaplockStringW(InitialModuleName);
|
|
PWCHAR ComparisonObject2 = CaplockStringW(IdentifiedModuleName);
|
|
|
|
if (StringCompareW(ComparisonObject1, ComparisonObject2) == 0)
|
|
return (HMODULE)Module->BaseAddress;
|
|
}
|
|
|
|
Next = Next->Flink;
|
|
}
|
|
|
|
return NULL;
|
|
}
|