mirror of
https://github.com/reactos/reactos.git
synced 2024-11-06 22:52:46 +00:00
527f2f9057
* Create a branch for some evul shell experiments. svn path=/branches/shell-experiments/; revision=61927
1640 lines
48 KiB
C
1640 lines
48 KiB
C
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS NT User Mode Library
|
|
* FILE: dll/ntdll/ldr/ldrapi.c
|
|
* PURPOSE: PE Loader Public APIs
|
|
* PROGRAMMERS: Alex Ionescu (alex@relsoft.net)
|
|
* Aleksey Bragin (aleksey@reactos.org)
|
|
*/
|
|
|
|
/* INCLUDES *****************************************************************/
|
|
|
|
#include <ntdll.h>
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
/* GLOBALS *******************************************************************/
|
|
|
|
LIST_ENTRY LdrpUnloadHead;
|
|
LONG LdrpLoaderLockAcquisitonCount;
|
|
BOOLEAN LdrpShowRecursiveLoads, LdrpBreakOnRecursiveDllLoads;
|
|
UNICODE_STRING LdrApiDefaultExtension = RTL_CONSTANT_STRING(L".DLL");
|
|
ULONG AlternateResourceModuleCount;
|
|
|
|
/* FUNCTIONS *****************************************************************/
|
|
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrFindCreateProcessManifest(IN ULONG Flags,
|
|
IN PVOID Image,
|
|
IN PVOID IdPath,
|
|
IN ULONG IdPathLength,
|
|
IN PVOID OutDataEntry)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrDestroyOutOfProcessImage(IN PVOID Image)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrCreateOutOfProcessImage(IN ULONG Flags,
|
|
IN HANDLE ProcessHandle,
|
|
IN HANDLE DllHandle,
|
|
IN PVOID Unknown3)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrAccessOutOfProcessResource(IN PVOID Unknown,
|
|
IN PVOID Image,
|
|
IN PVOID Unknown1,
|
|
IN PVOID Unknown2,
|
|
IN PVOID Unknown3)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
VOID
|
|
NTAPI
|
|
LdrSetDllManifestProber(IN PVOID ProberFunction)
|
|
{
|
|
UNIMPLEMENTED;
|
|
}
|
|
|
|
BOOLEAN
|
|
NTAPI
|
|
LdrAlternateResourcesEnabled(VOID)
|
|
{
|
|
/* ReactOS does not support this */
|
|
return FALSE;
|
|
}
|
|
|
|
ULONG_PTR
|
|
FORCEINLINE
|
|
LdrpMakeCookie(VOID)
|
|
{
|
|
/* Generate a cookie */
|
|
return (((ULONG_PTR)NtCurrentTeb()->RealClientId.UniqueThread & 0xFFF) << 16) |
|
|
(_InterlockedIncrement(&LdrpLoaderLockAcquisitonCount) & 0xFFFF);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrUnlockLoaderLock(IN ULONG Flags,
|
|
IN ULONG Cookie OPTIONAL)
|
|
{
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
|
|
DPRINT("LdrUnlockLoaderLock(%x %x)\n", Flags, Cookie);
|
|
|
|
/* Check for valid flags */
|
|
if (Flags & ~LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* Flags are invalid, check how to fail */
|
|
if (Flags & LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* The caller wants us to raise status */
|
|
RtlRaiseStatus(STATUS_INVALID_PARAMETER_1);
|
|
}
|
|
|
|
/* A normal failure */
|
|
return STATUS_INVALID_PARAMETER_1;
|
|
}
|
|
|
|
/* If we don't have a cookie, just return */
|
|
if (!Cookie) return STATUS_SUCCESS;
|
|
|
|
/* Validate the cookie */
|
|
if ((Cookie & 0xF0000000) ||
|
|
((Cookie >> 16) ^ ((ULONG)(NtCurrentTeb()->RealClientId.UniqueThread) & 0xFFF)))
|
|
{
|
|
DPRINT1("LdrUnlockLoaderLock() called with an invalid cookie!\n");
|
|
|
|
/* Invalid cookie, check how to fail */
|
|
if (Flags & LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* The caller wants us to raise status */
|
|
RtlRaiseStatus(STATUS_INVALID_PARAMETER_2);
|
|
}
|
|
|
|
/* A normal failure */
|
|
return STATUS_INVALID_PARAMETER_2;
|
|
}
|
|
|
|
/* Ready to release the lock */
|
|
if (Flags & LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* Do a direct leave */
|
|
RtlLeaveCriticalSection(&LdrpLoaderLock);
|
|
}
|
|
else
|
|
{
|
|
/* Wrap this in SEH, since we're not supposed to raise */
|
|
_SEH2_TRY
|
|
{
|
|
/* Leave the lock */
|
|
RtlLeaveCriticalSection(&LdrpLoaderLock);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* We should use the LDR Filter instead */
|
|
Status = _SEH2_GetExceptionCode();
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/* All done */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrLockLoaderLock(IN ULONG Flags,
|
|
OUT PULONG Disposition OPTIONAL,
|
|
OUT PULONG_PTR Cookie OPTIONAL)
|
|
{
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
BOOLEAN InInit = LdrpInLdrInit;
|
|
|
|
DPRINT("LdrLockLoaderLock(%x %p %p)\n", Flags, Disposition, Cookie);
|
|
|
|
/* Zero out the outputs */
|
|
if (Disposition) *Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_INVALID;
|
|
if (Cookie) *Cookie = 0;
|
|
|
|
/* Validate the flags */
|
|
if (Flags & ~(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS |
|
|
LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY))
|
|
{
|
|
/* Flags are invalid, check how to fail */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* The caller wants us to raise status */
|
|
RtlRaiseStatus(STATUS_INVALID_PARAMETER_1);
|
|
}
|
|
|
|
/* A normal failure */
|
|
return STATUS_INVALID_PARAMETER_1;
|
|
}
|
|
|
|
/* Make sure we got a cookie */
|
|
if (!Cookie)
|
|
{
|
|
/* No cookie check how to fail */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* The caller wants us to raise status */
|
|
RtlRaiseStatus(STATUS_INVALID_PARAMETER_3);
|
|
}
|
|
|
|
/* A normal failure */
|
|
return STATUS_INVALID_PARAMETER_3;
|
|
}
|
|
|
|
/* Do or Do Not. There is no Try */
|
|
ASSERT((Disposition != NULL) || !(Flags & LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY));
|
|
|
|
/* If the flag is set, make sure we have a valid pointer to use */
|
|
if ((Flags & LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY) && !(Disposition))
|
|
{
|
|
/* No pointer to return the data to */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* The caller wants us to raise status */
|
|
RtlRaiseStatus(STATUS_INVALID_PARAMETER_2);
|
|
}
|
|
|
|
/* Fail */
|
|
return STATUS_INVALID_PARAMETER_2;
|
|
}
|
|
|
|
/* Return now if we are in the init phase */
|
|
if (InInit) return STATUS_SUCCESS;
|
|
|
|
/* Check what locking semantic to use */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS)
|
|
{
|
|
/* Check if we should enter or simply try */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY)
|
|
{
|
|
/* Do a try */
|
|
if (!RtlTryEnterCriticalSection(&LdrpLoaderLock))
|
|
{
|
|
/* It's locked */
|
|
*Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_NOT_ACQUIRED;
|
|
}
|
|
else
|
|
{
|
|
/* It worked */
|
|
*Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_ACQUIRED;
|
|
*Cookie = LdrpMakeCookie();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Do a enter */
|
|
RtlEnterCriticalSection(&LdrpLoaderLock);
|
|
|
|
/* See if result was requested */
|
|
if (Disposition) *Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_ACQUIRED;
|
|
*Cookie = LdrpMakeCookie();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Wrap this in SEH, since we're not supposed to raise */
|
|
_SEH2_TRY
|
|
{
|
|
/* Check if we should enter or simply try */
|
|
if (Flags & LDR_LOCK_LOADER_LOCK_FLAG_TRY_ONLY)
|
|
{
|
|
/* Do a try */
|
|
if (!RtlTryEnterCriticalSection(&LdrpLoaderLock))
|
|
{
|
|
/* It's locked */
|
|
*Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_NOT_ACQUIRED;
|
|
}
|
|
else
|
|
{
|
|
/* It worked */
|
|
*Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_ACQUIRED;
|
|
*Cookie = LdrpMakeCookie();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Do an enter */
|
|
RtlEnterCriticalSection(&LdrpLoaderLock);
|
|
|
|
/* See if result was requested */
|
|
if (Disposition) *Disposition = LDR_LOCK_LOADER_LOCK_DISPOSITION_LOCK_ACQUIRED;
|
|
*Cookie = LdrpMakeCookie();
|
|
}
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* We should use the LDR Filter instead */
|
|
Status = _SEH2_GetExceptionCode();
|
|
}
|
|
_SEH2_END;
|
|
}
|
|
|
|
/* Return status */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrLoadDll(IN PWSTR SearchPath OPTIONAL,
|
|
IN PULONG DllCharacteristics OPTIONAL,
|
|
IN PUNICODE_STRING DllName,
|
|
OUT PVOID *BaseAddress)
|
|
{
|
|
WCHAR StringBuffer[MAX_PATH];
|
|
UNICODE_STRING DllString1, DllString2;
|
|
BOOLEAN RedirectedDll = FALSE;
|
|
NTSTATUS Status;
|
|
ULONG Cookie;
|
|
PUNICODE_STRING OldTldDll;
|
|
PTEB Teb = NtCurrentTeb();
|
|
|
|
/* Initialize the strings */
|
|
RtlInitEmptyUnicodeString(&DllString2, NULL, 0);
|
|
DllString1.Buffer = StringBuffer;
|
|
DllString1.Length = 0;
|
|
DllString1.MaximumLength = sizeof(StringBuffer);
|
|
|
|
/* Check if the SxS Assemblies specify another file */
|
|
Status = RtlDosApplyFileIsolationRedirection_Ustr(TRUE,
|
|
DllName,
|
|
&LdrApiDefaultExtension,
|
|
&DllString1,
|
|
&DllString2,
|
|
&DllName,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
|
|
/* Check success */
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
/* Let Ldrp know */
|
|
RedirectedDll = TRUE;
|
|
}
|
|
else if (Status != STATUS_SXS_KEY_NOT_FOUND)
|
|
{
|
|
/* Unrecoverable SxS failure; did we get a string? */
|
|
if (DllString2.Buffer) RtlFreeUnicodeString(&DllString2);
|
|
return Status;
|
|
}
|
|
|
|
/* Lock the loader lock */
|
|
LdrLockLoaderLock(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, NULL, &Cookie);
|
|
|
|
/* Check if there's a TLD DLL being loaded */
|
|
OldTldDll = LdrpTopLevelDllBeingLoaded;
|
|
if (OldTldDll)
|
|
{
|
|
/* This is a recursive load, do something about it? */
|
|
if ((ShowSnaps) || (LdrpShowRecursiveLoads) || (LdrpBreakOnRecursiveDllLoads))
|
|
{
|
|
/* Print out debug messages */
|
|
DPRINT1("[%p, %p] LDR: Recursive DLL Load\n",
|
|
Teb->RealClientId.UniqueProcess,
|
|
Teb->RealClientId.UniqueThread);
|
|
DPRINT1("[%p, %p] Previous DLL being loaded \"%wZ\"\n",
|
|
Teb->RealClientId.UniqueProcess,
|
|
Teb->RealClientId.UniqueThread,
|
|
OldTldDll);
|
|
DPRINT1("[%p, %p] DLL being requested \"%wZ\"\n",
|
|
Teb->RealClientId.UniqueProcess,
|
|
Teb->RealClientId.UniqueThread,
|
|
DllName);
|
|
|
|
/* Was it initializing too? */
|
|
if (!LdrpCurrentDllInitializer)
|
|
{
|
|
DPRINT1("[%p, %p] LDR: No DLL Initializer was running\n",
|
|
Teb->RealClientId.UniqueProcess,
|
|
Teb->RealClientId.UniqueThread);
|
|
}
|
|
else
|
|
{
|
|
DPRINT1("[%p, %p] DLL whose initializer was currently running \"%wZ\"\n",
|
|
Teb->ClientId.UniqueProcess,
|
|
Teb->ClientId.UniqueThread,
|
|
&LdrpCurrentDllInitializer->BaseDllName);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Set this one as the TLD DLL being loaded*/
|
|
LdrpTopLevelDllBeingLoaded = DllName;
|
|
|
|
/* Load the DLL */
|
|
Status = LdrpLoadDll(RedirectedDll,
|
|
SearchPath,
|
|
DllCharacteristics,
|
|
DllName,
|
|
BaseAddress,
|
|
TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
else if ((Status != STATUS_NO_SUCH_FILE) &&
|
|
(Status != STATUS_DLL_NOT_FOUND) &&
|
|
(Status != STATUS_OBJECT_NAME_NOT_FOUND) &&
|
|
(Status != STATUS_DLL_INIT_FAILED))
|
|
{
|
|
// 85 == DPFLTR_LDR_ID;
|
|
DbgPrintEx(85,
|
|
DPFLTR_WARNING_LEVEL,
|
|
"LDR: %s - failing because LdrpLoadDll(%wZ) returned status %x\n",
|
|
__FUNCTION__,
|
|
DllName,
|
|
Status);
|
|
}
|
|
|
|
/* Restore the old TLD DLL */
|
|
LdrpTopLevelDllBeingLoaded = OldTldDll;
|
|
|
|
/* Release the lock */
|
|
LdrUnlockLoaderLock(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, Cookie);
|
|
|
|
/* Do we have a redirect string? */
|
|
if (DllString2.Buffer) RtlFreeUnicodeString(&DllString2);
|
|
|
|
/* Return */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrFindEntryForAddress(PVOID Address,
|
|
PLDR_DATA_TABLE_ENTRY *Module)
|
|
{
|
|
PLIST_ENTRY ListHead, NextEntry;
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry;
|
|
PIMAGE_NT_HEADERS NtHeader;
|
|
PPEB_LDR_DATA Ldr = NtCurrentPeb()->Ldr;
|
|
ULONG_PTR DllBase, DllEnd;
|
|
|
|
DPRINT("LdrFindEntryForAddress(Address %p)\n", Address);
|
|
|
|
/* Nothing to do */
|
|
if (!Ldr) return STATUS_NO_MORE_ENTRIES;
|
|
|
|
/* Get the current entry */
|
|
LdrEntry = Ldr->EntryInProgress;
|
|
if (LdrEntry)
|
|
{
|
|
/* Get the NT Headers */
|
|
NtHeader = RtlImageNtHeader(LdrEntry->DllBase);
|
|
if (NtHeader)
|
|
{
|
|
/* Get the Image Base */
|
|
DllBase = (ULONG_PTR)LdrEntry->DllBase;
|
|
DllEnd = DllBase + NtHeader->OptionalHeader.SizeOfImage;
|
|
|
|
/* Check if they match */
|
|
if (((ULONG_PTR)Address >= DllBase) &&
|
|
((ULONG_PTR)Address < DllEnd))
|
|
{
|
|
/* Return it */
|
|
*Module = LdrEntry;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Loop the module list */
|
|
ListHead = &Ldr->InMemoryOrderModuleList;
|
|
NextEntry = ListHead->Flink;
|
|
while (NextEntry != ListHead)
|
|
{
|
|
/* Get the entry and NT Headers */
|
|
LdrEntry = CONTAINING_RECORD(NextEntry, LDR_DATA_TABLE_ENTRY, InMemoryOrderModuleList);
|
|
NtHeader = RtlImageNtHeader(LdrEntry->DllBase);
|
|
if (NtHeader)
|
|
{
|
|
/* Get the Image Base */
|
|
DllBase = (ULONG_PTR)LdrEntry->DllBase;
|
|
DllEnd = DllBase + NtHeader->OptionalHeader.SizeOfImage;
|
|
|
|
/* Check if they match */
|
|
if (((ULONG_PTR)Address >= DllBase) &&
|
|
((ULONG_PTR)Address < DllEnd))
|
|
{
|
|
/* Return it */
|
|
*Module = LdrEntry;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
/* Next Entry */
|
|
NextEntry = NextEntry->Flink;
|
|
}
|
|
}
|
|
|
|
/* Nothing found */
|
|
// 85 == DPFLTR_LDR_ID;
|
|
DbgPrintEx(85, DPFLTR_WARNING_LEVEL, "LDR: %s() exiting 0x%08lx\n", __FUNCTION__, STATUS_NO_MORE_ENTRIES);
|
|
return STATUS_NO_MORE_ENTRIES;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrGetDllHandleEx(IN ULONG Flags,
|
|
IN PWSTR DllPath OPTIONAL,
|
|
IN PULONG DllCharacteristics OPTIONAL,
|
|
IN PUNICODE_STRING DllName,
|
|
OUT PVOID *DllHandle OPTIONAL)
|
|
{
|
|
NTSTATUS Status;
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry;
|
|
UNICODE_STRING RedirectName, DllString1, RawDllName;
|
|
PUNICODE_STRING pRedirectName, CompareName;
|
|
PWCHAR p1, p2, p3;
|
|
BOOLEAN Locked, RedirectedDll;
|
|
ULONG_PTR Cookie;
|
|
ULONG LoadFlag, Length;
|
|
|
|
/* Initialize the strings */
|
|
RtlInitEmptyUnicodeString(&DllString1, NULL, 0);
|
|
RtlInitEmptyUnicodeString(&RawDllName, NULL, 0);
|
|
RedirectName = *DllName;
|
|
pRedirectName = &RedirectName;
|
|
|
|
/* Initialize state */
|
|
RedirectedDll = Locked = FALSE;
|
|
LdrEntry = NULL;
|
|
Cookie = 0;
|
|
|
|
/* Clear the handle */
|
|
if (DllHandle) *DllHandle = NULL;
|
|
|
|
/* Check for a valid flag combination */
|
|
if ((Flags & ~(LDR_GET_DLL_HANDLE_EX_PIN | LDR_GET_DLL_HANDLE_EX_UNCHANGED_REFCOUNT)) ||
|
|
(!DllHandle && !(Flags & LDR_GET_DLL_HANDLE_EX_PIN)))
|
|
{
|
|
DPRINT1("Flags are invalid or no DllHandle given\n");
|
|
Status = STATUS_INVALID_PARAMETER;
|
|
goto Quickie;
|
|
}
|
|
|
|
/* If not initializing */
|
|
if (!LdrpInLdrInit)
|
|
{
|
|
/* Acquire the lock */
|
|
Status = LdrLockLoaderLock(0, NULL, &Cookie);
|
|
if (!NT_SUCCESS(Status)) goto Quickie;
|
|
|
|
/* Remember we own it */
|
|
Locked = TRUE;
|
|
}
|
|
|
|
/* Check if the SxS Assemblies specify another file */
|
|
Status = RtlDosApplyFileIsolationRedirection_Ustr(TRUE,
|
|
pRedirectName,
|
|
&LdrApiDefaultExtension,
|
|
NULL,
|
|
&DllString1,
|
|
&pRedirectName,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
|
|
/* Check success */
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
/* Let Ldrp know */
|
|
RedirectedDll = TRUE;
|
|
}
|
|
else if (Status != STATUS_SXS_KEY_NOT_FOUND)
|
|
{
|
|
/* Unrecoverable SxS failure; */
|
|
goto Quickie;
|
|
}
|
|
else
|
|
{
|
|
ASSERT(pRedirectName == &RedirectName);
|
|
}
|
|
|
|
/* Set default failure code */
|
|
Status = STATUS_DLL_NOT_FOUND;
|
|
|
|
/* Use the cache if we can */
|
|
if (LdrpGetModuleHandleCache)
|
|
{
|
|
/* Check if we were redirected */
|
|
if (RedirectedDll)
|
|
{
|
|
/* Check the flag */
|
|
if (!(LdrpGetModuleHandleCache->Flags & LDRP_REDIRECTED))
|
|
{
|
|
goto DontCompare;
|
|
}
|
|
|
|
/* Use the right name */
|
|
CompareName = &LdrpGetModuleHandleCache->FullDllName;
|
|
}
|
|
else
|
|
{
|
|
/* Check the flag */
|
|
if (LdrpGetModuleHandleCache->Flags & LDRP_REDIRECTED)
|
|
{
|
|
goto DontCompare;
|
|
}
|
|
|
|
/* Use the right name */
|
|
CompareName = &LdrpGetModuleHandleCache->BaseDllName;
|
|
}
|
|
|
|
/* Check if the name matches */
|
|
if (RtlEqualUnicodeString(pRedirectName,
|
|
CompareName,
|
|
TRUE))
|
|
{
|
|
/* Skip the rest */
|
|
LdrEntry = LdrpGetModuleHandleCache;
|
|
|
|
/* Return success */
|
|
Status = STATUS_SUCCESS;
|
|
goto Quickie;
|
|
}
|
|
}
|
|
|
|
DontCompare:
|
|
/* Find the name without the extension */
|
|
p1 = pRedirectName->Buffer;
|
|
p2 = NULL;
|
|
p3 = &p1[pRedirectName->Length / sizeof(WCHAR)];
|
|
while (p1 != p3)
|
|
{
|
|
if (*p1++ == L'.')
|
|
{
|
|
p2 = p1;
|
|
}
|
|
else if (*p1 == L'\\')
|
|
{
|
|
p2 = NULL;
|
|
}
|
|
}
|
|
|
|
/* Check if no extension was found or if we got a slash */
|
|
if (!(p2) || (*p2 == L'\\') || (*p2 == L'/'))
|
|
{
|
|
/* Check that we have space to add one */
|
|
Length = pRedirectName->Length +
|
|
LdrApiDefaultExtension.Length + sizeof(UNICODE_NULL);
|
|
if (Length >= UNICODE_STRING_MAX_BYTES)
|
|
{
|
|
/* No space to add the extension */
|
|
Status = STATUS_NAME_TOO_LONG;
|
|
goto Quickie;
|
|
}
|
|
|
|
/* Setup the string */
|
|
RawDllName.MaximumLength = Length;
|
|
ASSERT(Length >= sizeof(UNICODE_NULL));
|
|
RawDllName.Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
|
|
0,
|
|
RawDllName.MaximumLength);
|
|
if (!RawDllName.Buffer)
|
|
{
|
|
Status = STATUS_NO_MEMORY;
|
|
goto Quickie;
|
|
}
|
|
|
|
/* Copy the string and add extension */
|
|
RtlCopyUnicodeString(&RawDllName, pRedirectName);
|
|
RtlAppendUnicodeStringToString(&RawDllName, &LdrApiDefaultExtension);
|
|
}
|
|
else
|
|
{
|
|
/* Check if there's something in the name */
|
|
Length = pRedirectName->Length;
|
|
if (Length)
|
|
{
|
|
/* Check and remove trailing period */
|
|
if (pRedirectName->Buffer[Length / sizeof(WCHAR) - sizeof(ANSI_NULL)] == '.')
|
|
{
|
|
/* Decrease the size */
|
|
pRedirectName->Length -= sizeof(WCHAR);
|
|
}
|
|
}
|
|
|
|
/* Setup the string */
|
|
RawDllName.MaximumLength = pRedirectName->Length + sizeof(WCHAR);
|
|
RawDllName.Buffer = RtlAllocateHeap(RtlGetProcessHeap(),
|
|
0,
|
|
RawDllName.MaximumLength);
|
|
if (!RawDllName.Buffer)
|
|
{
|
|
Status = STATUS_NO_MEMORY;
|
|
goto Quickie;
|
|
}
|
|
|
|
/* Copy the string */
|
|
RtlCopyUnicodeString(&RawDllName, pRedirectName);
|
|
}
|
|
|
|
/* Display debug string */
|
|
if (ShowSnaps)
|
|
{
|
|
DPRINT1("LDR: LdrGetDllHandleEx, searching for %wZ from %ws\n",
|
|
&RawDllName,
|
|
DllPath ? ((ULONG_PTR)DllPath == 1 ? L"" : DllPath) : L"");
|
|
}
|
|
|
|
/* Do the lookup */
|
|
if (LdrpCheckForLoadedDll(DllPath,
|
|
&RawDllName,
|
|
((ULONG_PTR)DllPath == 1) ? TRUE : FALSE,
|
|
RedirectedDll,
|
|
&LdrEntry))
|
|
{
|
|
/* Update cached entry */
|
|
LdrpGetModuleHandleCache = LdrEntry;
|
|
|
|
/* Return success */
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
/* Make sure to NULL this */
|
|
LdrEntry = NULL;
|
|
}
|
|
Quickie:
|
|
/* The success path must have a valid loader entry */
|
|
ASSERT((LdrEntry != NULL) == NT_SUCCESS(Status));
|
|
|
|
/* Check if we got an entry and success */
|
|
DPRINT("Got LdrEntry->BaseDllName %wZ\n", LdrEntry ? &LdrEntry->BaseDllName : NULL);
|
|
if ((LdrEntry) && (NT_SUCCESS(Status)))
|
|
{
|
|
/* Check if the DLL is locked */
|
|
if ((LdrEntry->LoadCount != 0xFFFF) &&
|
|
!(Flags & LDR_GET_DLL_HANDLE_EX_UNCHANGED_REFCOUNT))
|
|
{
|
|
/* Check what to do with the load count */
|
|
if (Flags & LDR_GET_DLL_HANDLE_EX_PIN)
|
|
{
|
|
/* Pin it */
|
|
LdrEntry->LoadCount = 0xFFFF;
|
|
LoadFlag = LDRP_UPDATE_PIN;
|
|
}
|
|
else
|
|
{
|
|
/* Increase the load count */
|
|
LdrEntry->LoadCount++;
|
|
LoadFlag = LDRP_UPDATE_REFCOUNT;
|
|
}
|
|
|
|
/* Update the load count now */
|
|
LdrpUpdateLoadCount2(LdrEntry, LoadFlag);
|
|
LdrpClearLoadInProgress();
|
|
}
|
|
|
|
/* Check if the caller is requesting the handle */
|
|
if (DllHandle) *DllHandle = LdrEntry->DllBase;
|
|
}
|
|
|
|
/* Free string if needed */
|
|
if (DllString1.Buffer) RtlFreeUnicodeString(&DllString1);
|
|
|
|
/* Free the raw DLL Name if needed */
|
|
if (RawDllName.Buffer)
|
|
{
|
|
/* Free the heap-allocated buffer */
|
|
RtlFreeHeap(RtlGetProcessHeap(), 0, RawDllName.Buffer);
|
|
RawDllName.Buffer = NULL;
|
|
}
|
|
|
|
/* Release lock */
|
|
if (Locked)
|
|
{
|
|
LdrUnlockLoaderLock(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS,
|
|
Cookie);
|
|
}
|
|
|
|
/* Return */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrGetDllHandle(IN PWSTR DllPath OPTIONAL,
|
|
IN PULONG DllCharacteristics OPTIONAL,
|
|
IN PUNICODE_STRING DllName,
|
|
OUT PVOID *DllHandle)
|
|
{
|
|
/* Call the newer API */
|
|
return LdrGetDllHandleEx(LDR_GET_DLL_HANDLE_EX_UNCHANGED_REFCOUNT,
|
|
DllPath,
|
|
DllCharacteristics,
|
|
DllName,
|
|
DllHandle);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrGetProcedureAddress(IN PVOID BaseAddress,
|
|
IN PANSI_STRING Name,
|
|
IN ULONG Ordinal,
|
|
OUT PVOID *ProcedureAddress)
|
|
{
|
|
/* Call the internal routine and tell it to execute DllInit */
|
|
return LdrpGetProcedureAddress(BaseAddress, Name, Ordinal, ProcedureAddress, TRUE);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrVerifyImageMatchesChecksum(IN HANDLE FileHandle,
|
|
IN PLDR_CALLBACK Callback,
|
|
IN PVOID CallbackContext,
|
|
OUT PUSHORT ImageCharacteristics)
|
|
{
|
|
FILE_STANDARD_INFORMATION FileStandardInfo;
|
|
PIMAGE_IMPORT_DESCRIPTOR ImportData;
|
|
PIMAGE_SECTION_HEADER LastSection = NULL;
|
|
IO_STATUS_BLOCK IoStatusBlock;
|
|
PIMAGE_NT_HEADERS NtHeader;
|
|
HANDLE SectionHandle;
|
|
SIZE_T ViewSize;
|
|
PVOID ViewBase;
|
|
BOOLEAN Result, NoActualCheck;
|
|
NTSTATUS Status;
|
|
PVOID ImportName;
|
|
ULONG Size;
|
|
DPRINT("LdrVerifyImageMatchesChecksum() called\n");
|
|
|
|
/* If the handle has the magic KnownDll flag, skip actual checksums */
|
|
NoActualCheck = ((ULONG_PTR)FileHandle & 1);
|
|
|
|
/* Create the section */
|
|
Status = NtCreateSection(&SectionHandle,
|
|
SECTION_MAP_EXECUTE,
|
|
NULL,
|
|
NULL,
|
|
PAGE_EXECUTE,
|
|
SEC_COMMIT,
|
|
FileHandle);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT1 ("NtCreateSection() failed (Status 0x%x)\n", Status);
|
|
return Status;
|
|
}
|
|
|
|
/* Map the section */
|
|
ViewSize = 0;
|
|
ViewBase = NULL;
|
|
Status = NtMapViewOfSection(SectionHandle,
|
|
NtCurrentProcess(),
|
|
&ViewBase,
|
|
0,
|
|
0,
|
|
NULL,
|
|
&ViewSize,
|
|
ViewShare,
|
|
0,
|
|
PAGE_EXECUTE);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT1("NtMapViewOfSection() failed (Status 0x%x)\n", Status);
|
|
NtClose(SectionHandle);
|
|
return Status;
|
|
}
|
|
|
|
/* Get the file information */
|
|
Status = NtQueryInformationFile(FileHandle,
|
|
&IoStatusBlock,
|
|
&FileStandardInfo,
|
|
sizeof(FILE_STANDARD_INFORMATION),
|
|
FileStandardInformation);
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
DPRINT1("NtMapViewOfSection() failed (Status 0x%x)\n", Status);
|
|
NtUnmapViewOfSection(NtCurrentProcess(), ViewBase);
|
|
NtClose(SectionHandle);
|
|
return Status;
|
|
}
|
|
|
|
/* Protect with SEH */
|
|
_SEH2_TRY
|
|
{
|
|
/* Check if this is the KnownDll hack */
|
|
if (NoActualCheck)
|
|
{
|
|
/* Don't actually do it */
|
|
Result = TRUE;
|
|
}
|
|
else
|
|
{
|
|
/* Verify the checksum */
|
|
Result = LdrVerifyMappedImageMatchesChecksum(ViewBase,
|
|
FileStandardInfo.EndOfFile.LowPart,
|
|
FileStandardInfo.EndOfFile.LowPart);
|
|
}
|
|
|
|
/* Check if a callback was supplied */
|
|
if ((Result) && (Callback))
|
|
{
|
|
/* Get the NT Header */
|
|
NtHeader = RtlImageNtHeader(ViewBase);
|
|
|
|
/* Check if caller requested this back */
|
|
if (ImageCharacteristics)
|
|
{
|
|
/* Return to caller */
|
|
*ImageCharacteristics = NtHeader->FileHeader.Characteristics;
|
|
}
|
|
|
|
/* Get the Import Directory Data */
|
|
ImportData = RtlImageDirectoryEntryToData(ViewBase,
|
|
FALSE,
|
|
IMAGE_DIRECTORY_ENTRY_IMPORT,
|
|
&Size);
|
|
|
|
/* Make sure there is one */
|
|
if (ImportData)
|
|
{
|
|
/* Loop the imports */
|
|
while (ImportData->Name)
|
|
{
|
|
/* Get the name */
|
|
ImportName = RtlImageRvaToVa(NtHeader,
|
|
ViewBase,
|
|
ImportData->Name,
|
|
&LastSection);
|
|
|
|
/* Notify the callback */
|
|
Callback(CallbackContext, ImportName);
|
|
ImportData++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* Fail the request returning STATUS_IMAGE_CHECKSUM_MISMATCH */
|
|
Result = FALSE;
|
|
}
|
|
_SEH2_END;
|
|
|
|
/* Unmap file and close handle */
|
|
NtUnmapViewOfSection(NtCurrentProcess(), ViewBase);
|
|
NtClose(SectionHandle);
|
|
|
|
/* Return status */
|
|
return Result ? Status : STATUS_IMAGE_CHECKSUM_MISMATCH;
|
|
}
|
|
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrQueryProcessModuleInformationEx(IN ULONG ProcessId,
|
|
IN ULONG Reserved,
|
|
OUT PRTL_PROCESS_MODULES ModuleInformation,
|
|
IN ULONG Size,
|
|
OUT PULONG ReturnedSize OPTIONAL)
|
|
{
|
|
PLIST_ENTRY ModuleListHead, InitListHead;
|
|
PLIST_ENTRY Entry, InitEntry;
|
|
PLDR_DATA_TABLE_ENTRY Module, InitModule;
|
|
PRTL_PROCESS_MODULE_INFORMATION ModulePtr = NULL;
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
ULONG UsedSize = sizeof(ULONG);
|
|
ANSI_STRING AnsiString;
|
|
PCHAR p;
|
|
|
|
DPRINT("LdrQueryProcessModuleInformation() called\n");
|
|
|
|
/* Acquire loader lock */
|
|
RtlEnterCriticalSection(NtCurrentPeb()->LoaderLock);
|
|
|
|
_SEH2_TRY
|
|
{
|
|
/* Check if we were given enough space */
|
|
if (Size < UsedSize)
|
|
{
|
|
Status = STATUS_INFO_LENGTH_MISMATCH;
|
|
}
|
|
else
|
|
{
|
|
ModuleInformation->NumberOfModules = 0;
|
|
ModulePtr = &ModuleInformation->Modules[0];
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
|
|
/* Traverse the list of modules */
|
|
ModuleListHead = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
|
|
Entry = ModuleListHead->Flink;
|
|
|
|
while (Entry != ModuleListHead)
|
|
{
|
|
Module = CONTAINING_RECORD(Entry, LDR_DATA_TABLE_ENTRY, InLoadOrderLinks);
|
|
|
|
DPRINT(" Module %wZ\n", &Module->FullDllName);
|
|
|
|
/* Increase the used size */
|
|
UsedSize += sizeof(RTL_PROCESS_MODULE_INFORMATION);
|
|
|
|
if (UsedSize > Size)
|
|
{
|
|
Status = STATUS_INFO_LENGTH_MISMATCH;
|
|
}
|
|
else
|
|
{
|
|
ModulePtr->ImageBase = Module->DllBase;
|
|
ModulePtr->ImageSize = Module->SizeOfImage;
|
|
ModulePtr->Flags = Module->Flags;
|
|
ModulePtr->LoadCount = Module->LoadCount;
|
|
ModulePtr->MappedBase = NULL;
|
|
ModulePtr->InitOrderIndex = 0;
|
|
ModulePtr->LoadOrderIndex = ModuleInformation->NumberOfModules;
|
|
|
|
/* Now get init order index by traversing init list */
|
|
InitListHead = &NtCurrentPeb()->Ldr->InInitializationOrderModuleList;
|
|
InitEntry = InitListHead->Flink;
|
|
|
|
while (InitEntry != InitListHead)
|
|
{
|
|
InitModule = CONTAINING_RECORD(InitEntry, LDR_DATA_TABLE_ENTRY, InInitializationOrderModuleList);
|
|
|
|
/* Increase the index */
|
|
ModulePtr->InitOrderIndex++;
|
|
|
|
/* Quit the loop if our module is found */
|
|
if (InitModule == Module) break;
|
|
|
|
/* Advance to the next entry */
|
|
InitEntry = InitEntry->Flink;
|
|
}
|
|
|
|
/* Prepare ANSI string with the module's name */
|
|
AnsiString.Length = 0;
|
|
AnsiString.MaximumLength = sizeof(ModulePtr->FullPathName);
|
|
AnsiString.Buffer = ModulePtr->FullPathName;
|
|
RtlUnicodeStringToAnsiString(&AnsiString,
|
|
&Module->FullDllName,
|
|
FALSE);
|
|
|
|
/* Calculate OffsetToFileName field */
|
|
p = strrchr(ModulePtr->FullPathName, '\\');
|
|
if (p != NULL)
|
|
ModulePtr->OffsetToFileName = p - ModulePtr->FullPathName + 1;
|
|
else
|
|
ModulePtr->OffsetToFileName = 0;
|
|
|
|
/* Advance to the next module in the output list */
|
|
ModulePtr++;
|
|
|
|
/* Increase number of modules */
|
|
if (ModuleInformation)
|
|
ModuleInformation->NumberOfModules++;
|
|
}
|
|
|
|
/* Go to the next entry in the modules list */
|
|
Entry = Entry->Flink;
|
|
}
|
|
|
|
/* Set returned size if it was provided */
|
|
if (ReturnedSize)
|
|
*ReturnedSize = UsedSize;
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* Ignoring the exception */
|
|
} _SEH2_END;
|
|
|
|
/* Release the lock */
|
|
RtlLeaveCriticalSection(NtCurrentPeb()->LoaderLock);
|
|
|
|
DPRINT("LdrQueryProcessModuleInformation() done\n");
|
|
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrQueryProcessModuleInformation(IN PRTL_PROCESS_MODULES ModuleInformation,
|
|
IN ULONG Size,
|
|
OUT PULONG ReturnedSize OPTIONAL)
|
|
{
|
|
/* Call Ex version of the API */
|
|
return LdrQueryProcessModuleInformationEx(0, 0, ModuleInformation, Size, ReturnedSize);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrEnumerateLoadedModules(IN BOOLEAN ReservedFlag,
|
|
IN PLDR_ENUM_CALLBACK EnumProc,
|
|
IN PVOID Context)
|
|
{
|
|
PLIST_ENTRY ListHead, ListEntry;
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry;
|
|
NTSTATUS Status;
|
|
ULONG Cookie;
|
|
BOOLEAN Stop = FALSE;
|
|
|
|
/* Check parameters */
|
|
if ((ReservedFlag) || !(EnumProc)) return STATUS_INVALID_PARAMETER;
|
|
|
|
/* Acquire the loader lock */
|
|
Status = LdrLockLoaderLock(0, NULL, &Cookie);
|
|
if (!NT_SUCCESS(Status)) return Status;
|
|
|
|
/* Loop all the modules and call enum proc */
|
|
ListHead = &NtCurrentPeb()->Ldr->InLoadOrderModuleList;
|
|
ListEntry = ListHead->Flink;
|
|
while (ListHead != ListEntry)
|
|
{
|
|
/* Get the entry */
|
|
LdrEntry = CONTAINING_RECORD(ListEntry, LDR_DATA_TABLE_ENTRY, InLoadOrderLinks);
|
|
|
|
/* Call the enumeration proc inside SEH */
|
|
_SEH2_TRY
|
|
{
|
|
EnumProc(LdrEntry, Context, &Stop);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
/* Ignoring the exception */
|
|
} _SEH2_END;
|
|
|
|
/* Break if we were asked to stop enumeration */
|
|
if (Stop)
|
|
{
|
|
/* Release loader lock */
|
|
Status = LdrUnlockLoaderLock(0, Cookie);
|
|
|
|
/* Reset any successful status to STATUS_SUCCESS, but leave
|
|
failure to the caller */
|
|
if (NT_SUCCESS(Status))
|
|
Status = STATUS_SUCCESS;
|
|
|
|
/* Return any possible failure status */
|
|
return Status;
|
|
}
|
|
|
|
/* Advance to the next module */
|
|
ListEntry = ListEntry->Flink;
|
|
}
|
|
|
|
/* Release loader lock, it must succeed this time */
|
|
Status = LdrUnlockLoaderLock(0, Cookie);
|
|
ASSERT(NT_SUCCESS(Status));
|
|
|
|
/* Return success */
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrDisableThreadCalloutsForDll(IN PVOID BaseAddress)
|
|
{
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry;
|
|
NTSTATUS Status;
|
|
BOOLEAN LockHeld;
|
|
ULONG_PTR Cookie;
|
|
DPRINT("LdrDisableThreadCalloutsForDll (BaseAddress %p)\n", BaseAddress);
|
|
|
|
/* Don't do it during shutdown */
|
|
if (LdrpShutdownInProgress) return STATUS_SUCCESS;
|
|
|
|
/* Check if we should grab the lock */
|
|
LockHeld = FALSE;
|
|
if (!LdrpInLdrInit)
|
|
{
|
|
/* Grab the lock */
|
|
Status = LdrLockLoaderLock(0, NULL, &Cookie);
|
|
if (!NT_SUCCESS(Status)) return Status;
|
|
LockHeld = TRUE;
|
|
}
|
|
|
|
/* Make sure the DLL is valid and get its entry */
|
|
Status = STATUS_DLL_NOT_FOUND;
|
|
if (LdrpCheckForLoadedDllHandle(BaseAddress, &LdrEntry))
|
|
{
|
|
/* Get if it has a TLS slot */
|
|
if (!LdrEntry->TlsIndex)
|
|
{
|
|
/* It doesn't, so you're allowed to call this */
|
|
LdrEntry->Flags |= LDRP_DONT_CALL_FOR_THREADS;
|
|
Status = STATUS_SUCCESS;
|
|
}
|
|
}
|
|
|
|
/* Check if the lock was held */
|
|
if (LockHeld)
|
|
{
|
|
/* Release it */
|
|
LdrUnlockLoaderLock(LDR_UNLOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, Cookie);
|
|
}
|
|
|
|
/* Return the status */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrAddRefDll(IN ULONG Flags,
|
|
IN PVOID BaseAddress)
|
|
{
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry;
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
ULONG Cookie;
|
|
BOOLEAN Locked = FALSE;
|
|
|
|
/* Check for invalid flags */
|
|
if (Flags & ~(LDR_ADDREF_DLL_PIN))
|
|
{
|
|
/* Fail with invalid parameter status if so */
|
|
Status = STATUS_INVALID_PARAMETER;
|
|
goto quickie;
|
|
}
|
|
|
|
/* Acquire the loader lock if not in init phase */
|
|
if (!LdrpInLdrInit)
|
|
{
|
|
/* Acquire the lock */
|
|
Status = LdrLockLoaderLock(0, NULL, &Cookie);
|
|
if (!NT_SUCCESS(Status)) goto quickie;
|
|
Locked = TRUE;
|
|
}
|
|
|
|
/* Get this module's data table entry */
|
|
if (LdrpCheckForLoadedDllHandle(BaseAddress, &LdrEntry))
|
|
{
|
|
if (!LdrEntry)
|
|
{
|
|
/* Shouldn't happen */
|
|
Status = STATUS_INTERNAL_ERROR;
|
|
goto quickie;
|
|
}
|
|
|
|
/* If this is not a pinned module */
|
|
if (LdrEntry->LoadCount != 0xFFFF)
|
|
{
|
|
/* Update its load count */
|
|
if (Flags & LDR_ADDREF_DLL_PIN)
|
|
{
|
|
/* Pin it by setting load count to -1 */
|
|
LdrEntry->LoadCount = 0xFFFF;
|
|
LdrpUpdateLoadCount2(LdrEntry, LDRP_UPDATE_PIN);
|
|
}
|
|
else
|
|
{
|
|
/* Increase its load count by one */
|
|
LdrEntry->LoadCount++;
|
|
LdrpUpdateLoadCount2(LdrEntry, LDRP_UPDATE_REFCOUNT);
|
|
}
|
|
|
|
/* Clear load in progress */
|
|
LdrpClearLoadInProgress();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* There was an error getting this module's handle, return invalid param status */
|
|
Status = STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
quickie:
|
|
/* Check for error case */
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
/* Print debug information */
|
|
if ((ShowSnaps) || ((Status != STATUS_NO_SUCH_FILE) &&
|
|
(Status != STATUS_DLL_NOT_FOUND) &&
|
|
(Status != STATUS_OBJECT_NAME_NOT_FOUND)))
|
|
{
|
|
DPRINT1("LDR: LdrAddRefDll(%p) 0x%08lx\n", BaseAddress, Status);
|
|
}
|
|
}
|
|
|
|
/* Release the lock if needed */
|
|
if (Locked) LdrUnlockLoaderLock(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, Cookie);
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrUnloadDll(IN PVOID BaseAddress)
|
|
{
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
PPEB Peb = NtCurrentPeb();
|
|
PLDR_DATA_TABLE_ENTRY LdrEntry, CurrentEntry;
|
|
PVOID EntryPoint;
|
|
PLIST_ENTRY NextEntry;
|
|
LIST_ENTRY UnloadList;
|
|
RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_EXTENDED ActCtx;
|
|
PVOID CorImageData;
|
|
ULONG ComSectionSize;
|
|
|
|
/* Get the LDR Lock */
|
|
if (!LdrpInLdrInit) RtlEnterCriticalSection(Peb->LoaderLock);
|
|
|
|
/* Increase the unload count */
|
|
LdrpActiveUnloadCount++;
|
|
|
|
/* Skip unload */
|
|
if (LdrpShutdownInProgress) goto Quickie;
|
|
|
|
/* Make sure the DLL is valid and get its entry */
|
|
if (!LdrpCheckForLoadedDllHandle(BaseAddress, &LdrEntry))
|
|
{
|
|
Status = STATUS_DLL_NOT_FOUND;
|
|
goto Quickie;
|
|
}
|
|
|
|
/* Check the current Load Count */
|
|
if (LdrEntry->LoadCount != 0xFFFF)
|
|
{
|
|
/* Decrease it */
|
|
LdrEntry->LoadCount--;
|
|
|
|
/* If it's a dll */
|
|
if (LdrEntry->Flags & LDRP_IMAGE_DLL)
|
|
{
|
|
/* Set up the Act Ctx */
|
|
ActCtx.Size = sizeof(ActCtx);
|
|
ActCtx.Format = RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_FORMAT_WHISTLER;
|
|
RtlZeroMemory(&ActCtx.Frame, sizeof(RTL_ACTIVATION_CONTEXT_STACK_FRAME));
|
|
|
|
/* Activate the ActCtx */
|
|
RtlActivateActivationContextUnsafeFast(&ActCtx,
|
|
LdrEntry->EntryPointActivationContext);
|
|
|
|
/* Update the load count */
|
|
LdrpUpdateLoadCount2(LdrEntry, LDRP_UPDATE_DEREFCOUNT);
|
|
|
|
/* Release the context */
|
|
RtlDeactivateActivationContextUnsafeFast(&ActCtx);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* The DLL is locked */
|
|
goto Quickie;
|
|
}
|
|
|
|
/* Show debug message */
|
|
if (ShowSnaps) DPRINT1("LDR: UNINIT LIST\n");
|
|
|
|
/* Check if this is our only unload and initialize the list if so */
|
|
if (LdrpActiveUnloadCount == 1) InitializeListHead(&LdrpUnloadHead);
|
|
|
|
/* Loop the modules to build the list */
|
|
NextEntry = Peb->Ldr->InInitializationOrderModuleList.Blink;
|
|
while (NextEntry != &Peb->Ldr->InInitializationOrderModuleList)
|
|
{
|
|
/* Get the entry */
|
|
LdrEntry = CONTAINING_RECORD(NextEntry,
|
|
LDR_DATA_TABLE_ENTRY,
|
|
InInitializationOrderModuleList);
|
|
NextEntry = NextEntry->Blink;
|
|
|
|
/* Remove flag */
|
|
LdrEntry->Flags &= ~LDRP_UNLOAD_IN_PROGRESS;
|
|
|
|
/* If the load count is now 0 */
|
|
if (!LdrEntry->LoadCount)
|
|
{
|
|
/* Show message */
|
|
if (ShowSnaps)
|
|
{
|
|
DPRINT1("(%lu) [%ws] %ws (%lx) deinit %p\n",
|
|
LdrpActiveUnloadCount,
|
|
LdrEntry->BaseDllName.Buffer,
|
|
LdrEntry->FullDllName.Buffer,
|
|
(ULONG)LdrEntry->LoadCount,
|
|
LdrEntry->EntryPoint);
|
|
}
|
|
|
|
/* FIXME: Call Shim Engine and notify */
|
|
|
|
/* Unlink it */
|
|
CurrentEntry = LdrEntry;
|
|
RemoveEntryList(&CurrentEntry->InInitializationOrderModuleList);
|
|
RemoveEntryList(&CurrentEntry->InMemoryOrderModuleList);
|
|
RemoveEntryList(&CurrentEntry->HashLinks);
|
|
|
|
/* If there's more then one active unload */
|
|
if (LdrpActiveUnloadCount > 1)
|
|
{
|
|
/* Flush the cached DLL handle and clear the list */
|
|
LdrpLoadedDllHandleCache = NULL;
|
|
CurrentEntry->InMemoryOrderModuleList.Flink = NULL;
|
|
}
|
|
|
|
/* Add the entry on the unload list */
|
|
InsertTailList(&LdrpUnloadHead, &CurrentEntry->HashLinks);
|
|
}
|
|
}
|
|
|
|
/* Only call the entrypoints once */
|
|
if (LdrpActiveUnloadCount > 1) goto Quickie;
|
|
|
|
/* Now loop the unload list and create our own */
|
|
InitializeListHead(&UnloadList);
|
|
CurrentEntry = NULL;
|
|
NextEntry = LdrpUnloadHead.Flink;
|
|
while (NextEntry != &LdrpUnloadHead)
|
|
{
|
|
/* Get the current entry */
|
|
LdrEntry = CONTAINING_RECORD(NextEntry, LDR_DATA_TABLE_ENTRY, HashLinks);
|
|
|
|
/* FIXME: Log the Unload Event */
|
|
//LdrpRecordUnloadEvent(LdrEntry);
|
|
|
|
/* Set the entry and clear it from the list */
|
|
CurrentEntry = LdrEntry;
|
|
LdrpLoadedDllHandleCache = NULL;
|
|
CurrentEntry->InMemoryOrderModuleList.Flink = NULL;
|
|
|
|
/* Move it from the global to the local list */
|
|
RemoveEntryList(&CurrentEntry->HashLinks);
|
|
InsertTailList(&UnloadList, &CurrentEntry->HashLinks);
|
|
|
|
/* Get the entrypoint */
|
|
EntryPoint = LdrEntry->EntryPoint;
|
|
|
|
/* Check if we should call it */
|
|
if ((EntryPoint) && (LdrEntry->Flags & LDRP_PROCESS_ATTACH_CALLED))
|
|
{
|
|
/* Show message */
|
|
if (ShowSnaps)
|
|
{
|
|
DPRINT1("LDR: Calling deinit %p\n", EntryPoint);
|
|
}
|
|
|
|
/* Set up the Act Ctx */
|
|
ActCtx.Size = sizeof(ActCtx);
|
|
ActCtx.Format = RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_FORMAT_WHISTLER;
|
|
RtlZeroMemory(&ActCtx.Frame, sizeof(RTL_ACTIVATION_CONTEXT_STACK_FRAME));
|
|
|
|
/* Activate the ActCtx */
|
|
RtlActivateActivationContextUnsafeFast(&ActCtx,
|
|
LdrEntry->EntryPointActivationContext);
|
|
|
|
/* Call the entrypoint */
|
|
LdrpCallInitRoutine(LdrEntry->EntryPoint,
|
|
LdrEntry->DllBase,
|
|
DLL_PROCESS_DETACH,
|
|
NULL);
|
|
|
|
/* Release the context */
|
|
RtlDeactivateActivationContextUnsafeFast(&ActCtx);
|
|
}
|
|
|
|
/* Remove it from the list */
|
|
RemoveEntryList(&CurrentEntry->InLoadOrderLinks);
|
|
CurrentEntry = NULL;
|
|
NextEntry = LdrpUnloadHead.Flink;
|
|
}
|
|
|
|
/* Now loop our local list */
|
|
NextEntry = UnloadList.Flink;
|
|
while (NextEntry != &UnloadList)
|
|
{
|
|
/* Get the entry */
|
|
LdrEntry = CONTAINING_RECORD(NextEntry, LDR_DATA_TABLE_ENTRY, HashLinks);
|
|
NextEntry = NextEntry->Flink;
|
|
CurrentEntry = LdrEntry;
|
|
|
|
/* Notify Application Verifier */
|
|
if (Peb->NtGlobalFlag & FLG_HEAP_ENABLE_TAIL_CHECK)
|
|
{
|
|
DPRINT1("We don't support Application Verifier yet\n");
|
|
}
|
|
|
|
/* Show message */
|
|
if (ShowSnaps)
|
|
{
|
|
DPRINT1("LDR: Unmapping [%ws]\n", LdrEntry->BaseDllName.Buffer);
|
|
}
|
|
|
|
/* Check if this is a .NET executable */
|
|
CorImageData = RtlImageDirectoryEntryToData(LdrEntry->DllBase,
|
|
TRUE,
|
|
IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR,
|
|
&ComSectionSize);
|
|
if (CorImageData)
|
|
{
|
|
/* FIXME */
|
|
DPRINT1(".NET Images are not supported yet\n");
|
|
}
|
|
|
|
/* Check if we should unmap*/
|
|
if (!(CurrentEntry->Flags & LDR_COR_OWNS_UNMAP))
|
|
{
|
|
/* Unmap the DLL */
|
|
Status = NtUnmapViewOfSection(NtCurrentProcess(),
|
|
CurrentEntry->DllBase);
|
|
ASSERT(NT_SUCCESS(Status));
|
|
}
|
|
|
|
/* Unload the alternate resource module, if any */
|
|
LdrUnloadAlternateResourceModule(CurrentEntry->DllBase);
|
|
|
|
/* FIXME: Send shutdown notification */
|
|
//LdrpSendDllNotifications(CurrentEntry, 2, LdrpShutdownInProgress);
|
|
|
|
/* Check if a Hotpatch is active */
|
|
if (LdrEntry->PatchInformation)
|
|
{
|
|
/* FIXME */
|
|
DPRINT1("We don't support Hotpatching yet\n");
|
|
}
|
|
|
|
/* Deallocate the Entry */
|
|
LdrpFinalizeAndDeallocateDataTableEntry(CurrentEntry);
|
|
|
|
/* If this is the cached entry, invalidate it */
|
|
if (LdrpGetModuleHandleCache == CurrentEntry)
|
|
{
|
|
LdrpGetModuleHandleCache = NULL;
|
|
}
|
|
}
|
|
|
|
Quickie:
|
|
/* Decrease unload count */
|
|
LdrpActiveUnloadCount--;
|
|
if (!LdrpInLdrInit) RtlLeaveCriticalSection(Peb->LoaderLock);
|
|
|
|
/* Return to caller */
|
|
return Status;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
BOOLEAN
|
|
NTAPI
|
|
RtlDllShutdownInProgress(VOID)
|
|
{
|
|
/* Return the internal global */
|
|
return LdrpShutdownInProgress;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
PIMAGE_BASE_RELOCATION
|
|
NTAPI
|
|
LdrProcessRelocationBlock(IN ULONG_PTR Address,
|
|
IN ULONG Count,
|
|
IN PUSHORT TypeOffset,
|
|
IN LONG_PTR Delta)
|
|
{
|
|
return LdrProcessRelocationBlockLongLong(Address, Count, TypeOffset, Delta);
|
|
}
|
|
|
|
/* FIXME: Add to ntstatus.mc */
|
|
#define STATUS_MUI_FILE_NOT_FOUND ((NTSTATUS)0xC00B0001L)
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
NTSTATUS
|
|
NTAPI
|
|
LdrLoadAlternateResourceModule(IN PVOID Module,
|
|
IN PWSTR Buffer)
|
|
{
|
|
/* Is MUI Support enabled? */
|
|
if (!LdrAlternateResourcesEnabled()) return STATUS_SUCCESS;
|
|
|
|
UNIMPLEMENTED;
|
|
return STATUS_MUI_FILE_NOT_FOUND;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
BOOLEAN
|
|
NTAPI
|
|
LdrUnloadAlternateResourceModule(IN PVOID BaseAddress)
|
|
{
|
|
ULONG_PTR Cookie;
|
|
|
|
/* Acquire the loader lock */
|
|
LdrLockLoaderLock(LDR_LOCK_LOADER_LOCK_FLAG_RAISE_ON_ERRORS, NULL, &Cookie);
|
|
|
|
/* Check if there's any alternate resources loaded */
|
|
if (AlternateResourceModuleCount)
|
|
{
|
|
UNIMPLEMENTED;
|
|
}
|
|
|
|
/* Release the loader lock */
|
|
LdrUnlockLoaderLock(1, Cookie);
|
|
|
|
/* All done */
|
|
return TRUE;
|
|
}
|
|
|
|
/*
|
|
* @unimplemented
|
|
*/
|
|
BOOLEAN
|
|
NTAPI
|
|
LdrFlushAlternateResourceModules(VOID)
|
|
{
|
|
UNIMPLEMENTED;
|
|
return FALSE;
|
|
}
|
|
|
|
/* EOF */
|