Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-15 17:11:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

[APPHELP] Implement SE_DynamicShim

Browse source
This commit is contained in:
Mark Jansen 2018-11-18 23:18:13 +01:00
parent 18d40ebe81
commit 93e88edbb6
No known key found for this signature in database
GPG key ID: B39240EE84BEAE8B
3 changed files with 39 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
dll/appcompat/apphelp/apphelp.c
View file

@ -185,7 +185,7 @@ ApphelpCheckRunAppEx(
_In_ HANDLE FileHandle,
_In_opt_ PVOID Unk1,
_In_opt_ PVOID Unk2,
_In_opt_z_ PWCHAR ApplicationName,
_In_opt_z_ PCWSTR ApplicationName,
_In_opt_ PVOID Environment,
_In_opt_ USHORT ExeType,
_Inout_opt_ PULONG Reason,

2
dll/appcompat/apphelp/apphelp.spec
View file

@ -160,7 +160,7 @@
@ stdcall SdbWriteWORDTag(ptr long long)
@ stdcall SE_DllLoaded(ptr)
@ stdcall SE_DllUnloaded(ptr)
@ stub SE_DynamicShim
@ stdcall SE_DynamicShim(wstr ptr ptr str ptr)
@ stub SE_DynamicUnshim
@ stdcall SE_InstallAfterInit(ptr ptr)
@ stdcall SE_InstallBeforeInit(ptr ptr)

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

@ -878,7 +878,13 @@ VOID SeiAppendInExclude(PARRAY dest, PCWSTR ModuleName, BOOL IsInclude)
}
}
/* Read the INEXCLUD tags from a given parent tag */
/* Read the INEXCLUD tags from a given parent tag
FIXME:
Some observed tags:
'*' with include
'$' with include, followed by '*' without include
Include list logging, referring to: (MODE: EA)
*/
VOID SeiReadInExclude(PDB pdb, TAGID parent, PARRAY dest)
{
TAGID InExcludeTag;
@ -1175,7 +1181,7 @@ VOID SeiResetEntryProcessed(PPEB Peb)
}
}
VOID SeiInit(PUNICODE_STRING ProcessImage, HSDB hsdb, SDBQUERYRESULT* pQuery)
VOID SeiInit(LPCWSTR ProcessImage, HSDB hsdb, SDBQUERYRESULT* pQuery, BOOLEAN ProcessInit)
{
DWORD n;
ARRAY ShimRefArray;
@ -1197,15 +1203,18 @@ VOID SeiInit(PUNICODE_STRING ProcessImage, HSDB hsdb, SDBQUERYRESULT* pQuery)
SeiCheckComPlusImage(Peb->ImageBaseAddress);
/* Mark all modules loaded until now as 'LDRP_ENTRY_PROCESSED' so that their entrypoint is not called while we are loading shims */
SeiSetEntryProcessed(Peb);
if (ProcessInit)
{
/* Mark all modules loaded until now as 'LDRP_ENTRY_PROCESSED' so that their entrypoint is not called while we are loading shims */
SeiSetEntryProcessed(Peb);
}
/* TODO:
if (pQuery->trApphelp)
SeiDisplayAppHelp(?pQuery->trApphelp?);
*/
SeiDbgPrint(SEI_MSG, NULL, "ShimInfo(ExePath(%wZ))\n", ProcessImage);
SeiDbgPrint(SEI_MSG, NULL, "ShimInfo(ExePath(%S))\n", ProcessImage);
SeiBuildShimRefArray(hsdb, pQuery, &ShimRefArray, &ShimFlags);
if (ShimFlags.AppCompatFlags.QuadPart)
{
@ -1344,8 +1353,11 @@ VOID SeiInit(PUNICODE_STRING ProcessImage, HSDB hsdb, SDBQUERYRESULT* pQuery)
SeiResolveAPIs();
PatchNewModules(Peb);
/* Remove the 'LDRP_ENTRY_PROCESSED' flag from entries we modified, so that the loader can continue to process them */
SeiResetEntryProcessed(Peb);
if (ProcessInit)
{
/* Remove the 'LDRP_ENTRY_PROCESSED' flag from entries we modified, so that the loader can continue to process them */
SeiResetEntryProcessed(Peb);
}
}
@ -1413,7 +1425,7 @@ VOID NTAPI SE_InstallBeforeInit(PUNICODE_STRING ProcessImage, PVOID pShimData)
}
g_bShimDuringInit = TRUE;
SeiInit(ProcessImage, hsdb, &QueryResult);
SeiInit(ProcessImage->Buffer, hsdb, &QueryResult, TRUE);
g_bShimDuringInit = FALSE;
SdbReleaseDatabase(hsdb);
@ -1464,3 +1476,20 @@ BOOL WINAPI SE_IsShimDll(PVOID BaseAddress)
return SeiGetShimModuleInfo(BaseAddress) != NULL;
}
/* 'Private' ntdll function */
BOOLEAN
NTAPI
LdrInitShimEngineDynamic(IN PVOID BaseAddress);
BOOL WINAPI SE_DynamicShim(LPCWSTR ProcessImage, HSDB hsdb, PVOID pQueryResult, LPCSTR Module, LPDWORD lpdwDynamicToken)
{
g_bShimDuringInit = TRUE;
SeiInit(ProcessImage, hsdb, pQueryResult, FALSE);
g_bShimDuringInit = FALSE;
LdrInitShimEngineDynamic(g_hInstance);
return TRUE;
}