Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-12-28 10:04:49 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NTOS:MM] Reformat MmCallDllInitialize and MiCallDllUnloadAndUnloadDll.

Browse source
This commit is contained in:
Hermès Bélusca-Maïto 2023-08-19 19:39:56 +02:00
parent aa719b9989
commit e8b048a282
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0
2 changed files with 35 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ntoskrnl/include/internal/mm.h
View file

@ -1653,9 +1653,8 @@ MmCheckSystemImage(
NTSTATUS
NTAPI
MmCallDllInitialize(
IN PLDR_DATA_TABLE_ENTRY LdrEntry,
IN PLIST_ENTRY ListHead
);
_In_ PLDR_DATA_TABLE_ENTRY LdrEntry,
_In_ PLIST_ENTRY ModuleListHead);
VOID
NTAPI

60
ntoskrnl/mm/ARM3/sysldr.c
View file

@ -292,24 +292,26 @@ MiLocateExportName(IN PVOID DllBase,
NTSTATUS
NTAPI
MmCallDllInitialize(IN PLDR_DATA_TABLE_ENTRY LdrEntry,
IN PLIST_ENTRY ListHead)
MmCallDllInitialize(
_In_ PLDR_DATA_TABLE_ENTRY LdrEntry,
_In_ PLIST_ENTRY ModuleListHead)
{
UNICODE_STRING ServicesKeyName = RTL_CONSTANT_STRING(
L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\");
PMM_DLL_INITIALIZE DllInit;
UNICODE_STRING RegPath, ImportName;
PCWCH Extension;
NTSTATUS Status;
PAGED_CODE();
/* Try to see if the image exports a DllInitialize routine */
DllInit = (PMM_DLL_INITIALIZE)MiLocateExportName(LdrEntry->DllBase,
"DllInitialize");
if (!DllInit) return STATUS_SUCCESS;
if (!DllInit)
return STATUS_SUCCESS;
/*
* Do a temporary copy of BaseDllName called ImportName
* because we'll alter the length of the string.
*/
/* Make a temporary copy of BaseDllName because we will alter its length */
ImportName.Length = LdrEntry->BaseDllName.Length;
ImportName.MaximumLength = LdrEntry->BaseDllName.MaximumLength;
ImportName.Buffer = LdrEntry->BaseDllName.Buffer;
@ -322,7 +324,8 @@ MmCallDllInitialize(IN PLDR_DATA_TABLE_ENTRY LdrEntry,
TAG_LDR_WSTR);
/* Check if this allocation was unsuccessful */
if (!RegPath.Buffer) return STATUS_INSUFFICIENT_RESOURCES;
if (!RegPath.Buffer)
return STATUS_INSUFFICIENT_RESOURCES;
/* Build and append the service name itself */
RegPath.Length = ServicesKeyName.Length;
@ -330,49 +333,52 @@ MmCallDllInitialize(IN PLDR_DATA_TABLE_ENTRY LdrEntry,
ServicesKeyName.Buffer,
ServicesKeyName.Length);
/* Check if there is a dot in the filename */
if (wcschr(ImportName.Buffer, L'.'))
{
/* Remove the extension */
ImportName.Length = (USHORT)(wcschr(ImportName.Buffer, L'.') -
ImportName.Buffer) * sizeof(WCHAR);
}
/* If the filename has an extension, remove it */
Extension = wcschr(ImportName.Buffer, L'.');
if (Extension)
ImportName.Length = (USHORT)(Extension - ImportName.Buffer) * sizeof(WCHAR);
/* Append service name (the basename without extension) */
/* Append the service name (base name without extension) */
RtlAppendUnicodeStringToString(&RegPath, &ImportName);
/* Now call the DllInit func */
/* Now call DllInitialize */
DPRINT("Calling DllInit(%wZ)\n", &RegPath);
Status = DllInit(&RegPath);
/* Clean up */
ExFreePoolWithTag(RegPath.Buffer, TAG_LDR_WSTR);
/* Return status value which DllInitialize returned */
// TODO: This is for Driver Verifier support.
UNREFERENCED_PARAMETER(ModuleListHead);
/* Return the DllInitialize status value */
return Status;
}
BOOLEAN
NTAPI
MiCallDllUnloadAndUnloadDll(IN PLDR_DATA_TABLE_ENTRY LdrEntry)
MiCallDllUnloadAndUnloadDll(
_In_ PLDR_DATA_TABLE_ENTRY LdrEntry)
{
NTSTATUS Status;
PMM_DLL_UNLOAD Func;
PMM_DLL_UNLOAD DllUnload;
PAGED_CODE();
/* Get the unload routine */
Func = (PMM_DLL_UNLOAD)MiLocateExportName(LdrEntry->DllBase, "DllUnload");
if (!Func) return FALSE;
/* Retrieve the DllUnload routine */
DllUnload = (PMM_DLL_UNLOAD)MiLocateExportName(LdrEntry->DllBase, "DllUnload");
if (!DllUnload)
return FALSE;
/* Call it and check for success */
Status = Func();
if (!NT_SUCCESS(Status)) return FALSE;
Status = DllUnload();
if (!NT_SUCCESS(Status))
return FALSE;
/* Lie about the load count so we can unload the image */
ASSERT(LdrEntry->LoadCount == 0);
LdrEntry->LoadCount = 1;
/* Unload it and return true */
/* Unload it */
MmUnloadSystemImage(LdrEntry);
return TRUE;
}