Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-04-05 13:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

[SHIMENG] Mark modified dll's as visited, and apply shims to newly loaded dlls

Browse source
This commit is contained in:
Mark Jansen 2017-08-29 22:21:58 +02:00
parent 0123f4745b
commit 06c1e134e1
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
dll/appcompat/apphelp/shimeng.c
View file

@ -833,6 +833,12 @@ VOID SeiHookImports(PLDR_DATA_TABLE_ENTRY LdrEntry)
return;
}
if (LdrEntry->Flags & LDRP_COMPAT_DATABASE_PROCESSED)
{
SHIMENG_INFO("Skipping module 0x%p \"%wZ\" because it was already processed\n", LdrEntry->DllBase, &LdrEntry->BaseDllName);
return;
}
ImportDescriptor = RtlImageDirectoryEntryToData(DllBase, TRUE, IMAGE_DIRECTORY_ENTRY_IMPORT, &Size);
if (!ImportDescriptor)
{
@ -902,6 +908,9 @@ VOID SeiHookImports(PLDR_DATA_TABLE_ENTRY LdrEntry)
}
}
}
/* Mark this module as processed. */
LdrEntry->Flags |= LDRP_COMPAT_DATABASE_PROCESSED;
}
@ -1185,12 +1194,18 @@ VOID NTAPI SE_ProcessDying(VOID)
VOID WINAPI SE_DllLoaded(PLDR_DATA_TABLE_ENTRY LdrEntry)
{
SHIMENG_INFO("%sINIT. loading DLL \"%wZ\"\n", g_bShimDuringInit ? "" : "AFTER ", &LdrEntry->BaseDllName);
SeiHookImports(LdrEntry);
NotifyShims(SHIM_REASON_DLL_LOAD, LdrEntry);
}
VOID WINAPI SE_DllUnloaded(PLDR_DATA_TABLE_ENTRY LdrEntry)
{
SHIMENG_INFO("(%p)\n", LdrEntry);
/* Should we unhook here? */
NotifyShims(SHIM_REASON_DLL_UNLOAD, LdrEntry);
}