BOOL WINAPI OnFreeLibraryWork(FARPROC lpfn, HookItem *ph, BOOL bMainThread, HMODULE hModule)
{
typedef BOOL (WINAPI* OnFreeLibrary_t)(HMODULE hModule);
BOOL lbRc = FALSE;
BOOL lbResource = LDR_IS_RESOURCE(hModule);
// lbResource получается TRUE например при вызовах из version.dll
UnprepareModule(hModule, NULL, 0);
#ifdef _DEBUG
BOOL lbModulePre = IsModuleValid(hModule); // GetModuleFileName(hModule, szModule, countof(szModule));
#endif
// Section locking is inadmissible. One FreeLibrary may cause another FreeLibrary in _different_ thread.
lbRc = ((OnFreeLibrary_t)lpfn)(hModule);
DWORD dwFreeErrCode = GetLastError();
// Далее только если !LDR_IS_RESOURCE
if (lbRc && !lbResource)
UnprepareModule(hModule, NULL, 1);
SetLastError(dwFreeErrCode);
return lbRc;
}
void UnitModuleTest()
{
wchar_t* pszConEmuCD = lstrmerge(gpConEmu->ms_ConEmuBaseDir, WIN3264TEST(L"\\ConEmuCD.dll",L"\\ConEmuCD64.dll"));
HMODULE hMod, hGetMod;
bool bTest;
_ASSERTE(!IsModuleValid((HMODULE)NULL));
_ASSERTE(!IsModuleValid((HMODULE)INVALID_HANDLE_VALUE));
hMod = GetModuleHandle(L"kernel32.dll");
if (hMod)
{
bTest = IsModuleValid(hMod);
_ASSERTE(bTest);
}
else
{
_ASSERTE(FALSE && "GetModuleHandle(kernel32) failed");
}
hMod = LoadLibrary(pszConEmuCD);
if (hMod)
{
bTest = IsModuleValid(hMod);
_ASSERTE(bTest);
FreeLibrary(hMod);
bTest = IsModuleValid(hMod);
// Due to unknown reason (KIS?) FreeLibrary was not able to release hMod sometimes
hGetMod = GetModuleHandle(pszConEmuCD);
if (!hGetMod)
bTest = IsModuleValid(hMod);
_ASSERTE(!bTest || (hGetMod!=NULL));
}
else
{
_ASSERTE(FALSE && "LoadLibrary(pszConEmuCD) failed");
}
}
bool CheckCallbackPtr(HMODULE hModule, size_t ProcCount, FARPROC* CallBack, BOOL abCheckModuleInfo, BOOL abAllowNTDLL, BOOL abTestVirtual /*= TRUE*/)
{
if ((hModule == NULL) || (hModule == INVALID_HANDLE_VALUE) || LDR_IS_RESOURCE(hModule))
{
_ASSERTE((hModule != NULL) && (hModule != INVALID_HANDLE_VALUE) && !LDR_IS_RESOURCE(hModule));
return false;
}
if (!CallBack || !ProcCount)
{
_ASSERTE(CallBack && ProcCount);
return false;
}
DWORD_PTR nModulePtr = (DWORD_PTR)hModule;
DWORD_PTR nModuleSize = (4<<20);
//BOOL lbModuleInformation = FALSE;
DWORD_PTR nModulePtr2 = 0;
DWORD_PTR nModuleSize2 = 0;
if (abAllowNTDLL)
{
nModulePtr2 = (DWORD_PTR)GetModuleHandle(L"ntdll.dll");
nModuleSize2 = (4<<20);
}
// Если разрешили - попробовать определить размер модуля, чтобы CallBack не выпал из его тела
if (abCheckModuleInfo)
{
if (!IsModuleValid(hModule, abTestVirtual))
{
_ASSERTE("!IsModuleValid(hModule)" && 0);
return false;
}
IMAGE_NT_HEADERS* nt_header = (IMAGE_NT_HEADERS*)((char*)hModule + ((IMAGE_DOS_HEADER*)hModule)->e_lfanew);
// Получить размер модуля из OptionalHeader
nModuleSize = nt_header->OptionalHeader.SizeOfImage;
if (nModulePtr2)
{
nt_header = (IMAGE_NT_HEADERS*)((char*)nModulePtr2 + ((IMAGE_DOS_HEADER*)nModulePtr2)->e_lfanew);
nModuleSize2 = nt_header->OptionalHeader.SizeOfImage;
}
}
for (size_t i = 0; i < ProcCount; i++)
{
if (!(CallBack[i]))
{
_ASSERTE((CallBack[i])!=NULL);
return false;
}
if ((((DWORD_PTR)(CallBack[i])) < nModulePtr)
&& (!nModulePtr2 || (((DWORD_PTR)(CallBack[i])) < nModulePtr2)))
{
_ASSERTE(((DWORD_PTR)(CallBack[i])) >= nModulePtr);
return false;
}
if ((((DWORD_PTR)(CallBack[i])) > (nModuleSize + nModulePtr))
&& (!nModulePtr2 || (((DWORD_PTR)(CallBack[i])) > (nModuleSize2 + nModulePtr2))))
{
_ASSERTE(((DWORD_PTR)(CallBack[i])) <= (nModuleSize + nModulePtr));
return false;
}
}
return true;
}