Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-07-23 05:43:39 +00:00
Code Issues Projects Releases Packages Wiki Activity

use tags for pool allocations, and reformat code

Browse source 
svn path=/trunk/; revision=14545
This commit is contained in:
Royce Mitchell III 2005-04-08 12:54:37 +00:00
parent d57bccaddf
commit dac8f53fe5
1 changed files with 1270 additions and 1238 deletions
Download patch file Download diff file Expand all files Collapse all files

96
reactos/ntoskrnl/ldr/loader.c
View file

@ -41,7 +41,10 @@ STATIC MODULE_TEXT_SECTION NtoskrnlTextSection;
STATIC MODULE_TEXT_SECTION LdrHalTextSection; STATIC MODULE_TEXT_SECTION LdrHalTextSection;
ULONG_PTR LdrHalBase; ULONG_PTR LdrHalBase;
#define TAG_DRIVER_MEM TAG('D', 'R', 'V', 'M') #define TAG_DRIVER_MEM TAG('D', 'R', 'V', 'M') /* drvm */
#define TAG_MODULE_OBJECT TAG('k', 'l', 'm', 'o') /* klmo - kernel ldr module object */
#define TAG_LDR_WSTR TAG('k', 'l', 'w', 's') /* klws - kernel ldr wide string */
#define TAG_MODULE_TEXT_SECTION TAG('k', 'l', 'm', 't') /* klmt - kernel ldr module text */
#ifndef HIWORD #ifndef HIWORD
#define HIWORD(X) ((WORD) (((DWORD) (X) >> 16) & 0xFFFF)) #define HIWORD(X) ((WORD) (((DWORD) (X) >> 16) & 0xFFFF))
@ -53,25 +56,32 @@ ULONG_PTR LdrHalBase;
/* FORWARD DECLARATIONS ******************************************************/ /* FORWARD DECLARATIONS ******************************************************/
NTSTATUS NTSTATUS
LdrProcessModule(PVOID ModuleLoadBase, LdrProcessModule (
PVOID ModuleLoadBase,
PUNICODE_STRING ModuleName, PUNICODE_STRING ModuleName,
PMODULE_OBJECT *ModuleObject ); PMODULE_OBJECT *ModuleObject );
static VOID static VOID
LdrpBuildModuleBaseName(PUNICODE_STRING BaseName, LdrpBuildModuleBaseName (
PUNICODE_STRING BaseName,
PUNICODE_STRING FullName ); PUNICODE_STRING FullName );
static LONG static LONG
LdrpCompareModuleNames(IN PUNICODE_STRING String1, LdrpCompareModuleNames (
IN PUNICODE_STRING String1,
IN PUNICODE_STRING String2 ); IN PUNICODE_STRING String2 );
/* PE Driver load support */ /* PE Driver load support */
static NTSTATUS LdrPEProcessModule(PVOID ModuleLoadBase, static NTSTATUS
LdrPEProcessModule (
PVOID ModuleLoadBase,
PUNICODE_STRING FileName, PUNICODE_STRING FileName,
PMODULE_OBJECT *ModuleObject ); PMODULE_OBJECT *ModuleObject );
static PVOID static PVOID
LdrPEGetExportByName(PVOID BaseAddress, LdrPEGetExportByName (
PVOID BaseAddress,
PUCHAR SymbolName, PUCHAR SymbolName,
WORD Hint ); WORD Hint );
@ -79,7 +89,8 @@ static PVOID
LdrPEFixupForward ( PCHAR ForwardName ); LdrPEFixupForward ( PCHAR ForwardName );
static NTSTATUS static NTSTATUS
LdrPEPerformRelocations(PVOID DriverBase, LdrPEPerformRelocations (
PVOID DriverBase,
ULONG DriverSize ); ULONG DriverSize );
static NTSTATUS static NTSTATUS
@ -130,8 +141,8 @@ LdrInit1(VOID)
NtHeader = RtlImageNtHeader((PVOID)KERNEL_BASE); NtHeader = RtlImageNtHeader((PVOID)KERNEL_BASE);
SectionList = IMAGE_FIRST_SECTION(NtHeader); SectionList = IMAGE_FIRST_SECTION(NtHeader);
NtoskrnlTextSection.Base = KERNEL_BASE; NtoskrnlTextSection.Base = KERNEL_BASE;
NtoskrnlTextSection.Length = SectionList[0].Misc.VirtualSize + NtoskrnlTextSection.Length = SectionList[0].Misc.VirtualSize
SectionList[0].VirtualAddress; + SectionList[0].VirtualAddress;
NtoskrnlTextSection.Name = KERNEL_MODULE_NAME; NtoskrnlTextSection.Name = KERNEL_MODULE_NAME;
NtoskrnlTextSection.OptionalHeader = OPTHDROFFSET(KERNEL_BASE); NtoskrnlTextSection.OptionalHeader = OPTHDROFFSET(KERNEL_BASE);
InsertTailList(&ModuleTextListHead, &NtoskrnlTextSection.ListEntry); InsertTailList(&ModuleTextListHead, &NtoskrnlTextSection.ListEntry);
@ -141,8 +152,8 @@ LdrInit1(VOID)
NtHeader = RtlImageNtHeader((PVOID)LdrHalBase); NtHeader = RtlImageNtHeader((PVOID)LdrHalBase);
SectionList = IMAGE_FIRST_SECTION(NtHeader); SectionList = IMAGE_FIRST_SECTION(NtHeader);
LdrHalTextSection.Base = LdrHalBase; LdrHalTextSection.Base = LdrHalBase;
LdrHalTextSection.Length = SectionList[0].Misc.VirtualSize + LdrHalTextSection.Length = SectionList[0].Misc.VirtualSize
SectionList[0].VirtualAddress; + SectionList[0].VirtualAddress;
LdrHalTextSection.Name = HAL_MODULE_NAME; LdrHalTextSection.Name = HAL_MODULE_NAME;
LdrHalTextSection.OptionalHeader = OPTHDROFFSET(LdrHalBase); LdrHalTextSection.OptionalHeader = OPTHDROFFSET(LdrHalBase);
InsertTailList(&ModuleTextListHead, &LdrHalTextSection.ListEntry); InsertTailList(&ModuleTextListHead, &LdrHalTextSection.ListEntry);
@ -201,7 +212,8 @@ LdrInitModuleManagement(VOID)
} }
NTSTATUS NTSTATUS
LdrpLoadImage(PUNICODE_STRING DriverName, LdrpLoadImage (
PUNICODE_STRING DriverName,
PVOID *ModuleBase, PVOID *ModuleBase,
PVOID *SectionPointer, PVOID *SectionPointer,
PVOID *EntryPoint, PVOID *EntryPoint,
@ -275,7 +287,8 @@ LdrpLoadAndCallImage(PUNICODE_STRING ModuleName)
NTSTATUS NTSTATUS
LdrLoadModule(PUNICODE_STRING Filename, LdrLoadModule(
PUNICODE_STRING Filename,
PMODULE_OBJECT *ModuleObject ) PMODULE_OBJECT *ModuleObject )
{ {
PVOID ModuleLoadBase; PVOID ModuleLoadBase;
@ -410,7 +423,8 @@ LdrUnloadModule(PMODULE_OBJECT ModuleObject)
NTSTATUS NTSTATUS
LdrProcessModule(PVOID ModuleLoadBase, LdrProcessModule(
PVOID ModuleLoadBase,
PUNICODE_STRING ModuleName, PUNICODE_STRING ModuleName,
PMODULE_OBJECT *ModuleObject ) PMODULE_OBJECT *ModuleObject )
{ {
@ -430,7 +444,8 @@ LdrProcessModule(PVOID ModuleLoadBase,
} }
NTSTATUS NTSTATUS
LdrpQueryModuleInformation(PVOID Buffer, LdrpQueryModuleInformation (
PVOID Buffer,
ULONG Size, ULONG Size,
PULONG ReqSize ) PULONG ReqSize )
{ {
@ -511,7 +526,8 @@ LdrpQueryModuleInformation(PVOID Buffer,
static VOID static VOID
LdrpBuildModuleBaseName(PUNICODE_STRING BaseName, LdrpBuildModuleBaseName (
PUNICODE_STRING BaseName,
PUNICODE_STRING FullName ) PUNICODE_STRING FullName )
{ {
PWCHAR p; PWCHAR p;
@ -536,7 +552,8 @@ LdrpBuildModuleBaseName(PUNICODE_STRING BaseName,
static LONG static LONG
LdrpCompareModuleNames(IN PUNICODE_STRING String1, LdrpCompareModuleNames (
IN PUNICODE_STRING String1,
IN PUNICODE_STRING String2 ) IN PUNICODE_STRING String2 )
{ {
ULONG len1, len2, i; ULONG len1, len2, i;
@ -644,7 +661,8 @@ LdrGetModuleObject(PUNICODE_STRING ModuleName)
/* ---------------------------------------------- PE Module support */ /* ---------------------------------------------- PE Module support */
static ULONG static ULONG
LdrLookupPageProtection(PVOID PageStart, LdrLookupPageProtection (
PVOID PageStart,
PVOID DriverBase, PVOID DriverBase,
PIMAGE_FILE_HEADER PEFileHeader, PIMAGE_FILE_HEADER PEFileHeader,
PIMAGE_SECTION_HEADER PESectionHeaders ) PIMAGE_SECTION_HEADER PESectionHeaders )
@ -696,7 +714,8 @@ LdrLookupPageProtection(PVOID PageStart,
} }
static NTSTATUS static NTSTATUS
LdrPEProcessModule(PVOID ModuleLoadBase, LdrPEProcessModule(
PVOID ModuleLoadBase,
PUNICODE_STRING FileName, PUNICODE_STRING FileName,
PMODULE_OBJECT *ModuleObject ) PMODULE_OBJECT *ModuleObject )
{ {
@ -804,7 +823,8 @@ LdrPEProcessModule(PVOID ModuleLoadBase,
} }
/* Create the module */ /* Create the module */
CreatedModuleObject = ExAllocatePool(NonPagedPool, sizeof(MODULE_OBJECT)); CreatedModuleObject = ExAllocatePoolWithTag (
NonPagedPool, sizeof(MODULE_OBJECT), TAG_MODULE_OBJECT );
if (CreatedModuleObject == NULL) if (CreatedModuleObject == NULL)
{ {
// MmFreeSection(DriverBase); // MmFreeSection(DriverBase);
@ -819,7 +839,8 @@ LdrPEProcessModule(PVOID ModuleLoadBase,
CreatedModuleObject->FullName.Length = 0; CreatedModuleObject->FullName.Length = 0;
CreatedModuleObject->FullName.MaximumLength = FileName->Length + sizeof(UNICODE_NULL); CreatedModuleObject->FullName.MaximumLength = FileName->Length + sizeof(UNICODE_NULL);
CreatedModuleObject->FullName.Buffer = ExAllocatePool(PagedPool, CreatedModuleObject->FullName.MaximumLength); CreatedModuleObject->FullName.Buffer =
ExAllocatePoolWithTag(PagedPool, CreatedModuleObject->FullName.MaximumLength, TAG_LDR_WSTR);
if (CreatedModuleObject->FullName.Buffer == NULL) if (CreatedModuleObject->FullName.Buffer == NULL)
{ {
ExFreePool(CreatedModuleObject); ExFreePool(CreatedModuleObject);
@ -932,14 +953,18 @@ LdrPEProcessModule(PVOID ModuleLoadBase,
KeReleaseSpinLock(&ModuleListLock, Irql); KeReleaseSpinLock(&ModuleListLock, Irql);
ModuleTextSection = ExAllocatePool(NonPagedPool, ModuleTextSection = ExAllocatePoolWithTag (
sizeof(MODULE_TEXT_SECTION)); NonPagedPool,
sizeof(MODULE_TEXT_SECTION),
TAG_MODULE_TEXT_SECTION );
ASSERT(ModuleTextSection); ASSERT(ModuleTextSection);
RtlZeroMemory(ModuleTextSection, sizeof(MODULE_TEXT_SECTION)); RtlZeroMemory(ModuleTextSection, sizeof(MODULE_TEXT_SECTION));
ModuleTextSection->Base = (ULONG)DriverBase; ModuleTextSection->Base = (ULONG)DriverBase;
ModuleTextSection->Length = DriverSize; ModuleTextSection->Length = DriverSize;
ModuleTextSection->Name = ExAllocatePool(NonPagedPool, ModuleTextSection->Name = ExAllocatePoolWithTag (
(CreatedModuleObject->BaseName.Length + 1) * sizeof(WCHAR)); NonPagedPool,
(CreatedModuleObject->BaseName.Length + 1) * sizeof(WCHAR),
TAG_LDR_WSTR );
RtlCopyMemory(ModuleTextSection->Name, RtlCopyMemory(ModuleTextSection->Name,
CreatedModuleObject->BaseName.Buffer, CreatedModuleObject->BaseName.Buffer,
CreatedModuleObject->BaseName.Length); CreatedModuleObject->BaseName.Length);
@ -965,7 +990,8 @@ LdrPEProcessModule(PVOID ModuleLoadBase,
PVOID INIT_FUNCTION PVOID INIT_FUNCTION
LdrSafePEProcessModule(PVOID ModuleLoadBase, LdrSafePEProcessModule (
PVOID ModuleLoadBase,
PVOID DriverBase, PVOID DriverBase,
PVOID ImportModuleBase, PVOID ImportModuleBase,
PULONG DriverSize) PULONG DriverSize)
@ -1153,7 +1179,8 @@ LdrPEFixupForward(PCHAR ForwardName)
} }
static NTSTATUS static NTSTATUS
LdrPEPerformRelocations(PVOID DriverBase, LdrPEPerformRelocations (
PVOID DriverBase,
ULONG DriverSize) ULONG DriverSize)
{ {
PIMAGE_NT_HEADERS NtHeaders; PIMAGE_NT_HEADERS NtHeaders;
@ -1249,7 +1276,8 @@ LdrPEPerformRelocations(PVOID DriverBase,
} }
static NTSTATUS static NTSTATUS
LdrPEGetOrLoadModule(PMODULE_OBJECT Module, LdrPEGetOrLoadModule (
PMODULE_OBJECT Module,
PCHAR ImportedName, PCHAR ImportedName,
PMODULE_OBJECT* ImportedModule) PMODULE_OBJECT* ImportedModule)
{ {
@ -1321,7 +1349,8 @@ LdrPEGetOrLoadModule(PMODULE_OBJECT Module,
} }
static PVOID static PVOID
LdrPEGetExportByName(PVOID BaseAddress, LdrPEGetExportByName (
PVOID BaseAddress,
PUCHAR SymbolName, PUCHAR SymbolName,
WORD Hint ) WORD Hint )
{ {
@ -1470,7 +1499,8 @@ LdrPEGetExportByName(PVOID BaseAddress,
} }
static PVOID static PVOID
LdrPEGetExportByOrdinal (PVOID BaseAddress, LdrPEGetExportByOrdinal (
PVOID BaseAddress,
ULONG Ordinal ) ULONG Ordinal )
{ {
PIMAGE_EXPORT_DIRECTORY ExportDir; PIMAGE_EXPORT_DIRECTORY ExportDir;
@ -1478,7 +1508,8 @@ LdrPEGetExportByOrdinal (PVOID BaseAddress,
PDWORD * ExFunctions; PDWORD * ExFunctions;
PVOID Function; PVOID Function;
ExportDir = (PIMAGE_EXPORT_DIRECTORY)RtlImageDirectoryEntryToData (BaseAddress, ExportDir = (PIMAGE_EXPORT_DIRECTORY)RtlImageDirectoryEntryToData (
BaseAddress,
TRUE, TRUE,
IMAGE_DIRECTORY_ENTRY_EXPORT, IMAGE_DIRECTORY_ENTRY_EXPORT,
&ExportDirSize); &ExportDirSize);
@ -1504,7 +1535,8 @@ LdrPEGetExportByOrdinal (PVOID BaseAddress,
} }
static NTSTATUS static NTSTATUS
LdrPEProcessImportDirectoryEntry(PVOID DriverBase, LdrPEProcessImportDirectoryEntry(
PVOID DriverBase,
PMODULE_OBJECT ImportedModule, PMODULE_OBJECT ImportedModule,
PIMAGE_IMPORT_DESCRIPTOR ImportModuleDirectory ) PIMAGE_IMPORT_DESCRIPTOR ImportModuleDirectory )
{ {