1 /* $Id: loader.c,v 1.131 2003/06/01 14:59:02 chorns Exp $
3 * COPYRIGHT: See COPYING in the top level directory
4 * PROJECT: ReactOS kernel
5 * FILE: ntoskrnl/ldr/loader.c
6 * PURPOSE: Loaders for PE executables
7 * PROGRAMMERS: Jean Michault
8 * Rex Jolliff (rex@lvcablemodem.com)
9 * Jason Filby (jasonfilby@yahoo.com)
10 * Casper S. Hornstrup (chorns@users.sourceforge.net)
13 * RJJ 10/12/98 Completed image loader function and added hooks for MZ/PE
14 * RJJ 10/12/98 Built driver loader function and added hooks for PE/COFF
15 * RJJ 10/12/98 Rolled in David's code to load COFF drivers
16 * JM 14/12/98 Built initial PE user module loader
17 * RJJ 06/03/99 Moved user PE loader into NTDLL
18 * JF 26/01/2000 Recoded some parts to retrieve export details correctly
19 * DW 27/06/2000 Removed redundant header files
20 * CSH 11/04/2001 Added automatic loading of module symbols if they exist
21 * KJK 02/04/2003 Nebbet-ized a couple of type names
25 /* INCLUDES *****************************************************************/
28 #include <ddk/ntddk.h>
30 #include <internal/module.h>
31 #include <internal/ntoskrnl.h>
32 #include <internal/kd.h>
33 #include <internal/io.h>
34 #include <internal/mm.h>
35 #include <internal/ps.h>
36 #include <internal/ldr.h>
37 #include <internal/pool.h>
38 #include <internal/kd.h>
39 #include <ntos/minmax.h>
42 #include <internal/ntosdbg.h>
48 #include <internal/debug.h>
50 /* GLOBALS *******************************************************************/
52 LIST_ENTRY ModuleListHead
;
53 KSPIN_LOCK ModuleListLock
;
55 LIST_ENTRY ModuleTextListHead
;
56 STATIC MODULE_TEXT_SECTION NtoskrnlTextSection
;
57 STATIC MODULE_TEXT_SECTION LdrHalTextSection
;
60 #define TAG_DRIVER_MEM TAG('D', 'R', 'V', 'M')
62 /* FORWARD DECLARATIONS ******************************************************/
65 LdrProcessModule(PVOID ModuleLoadBase
,
66 PUNICODE_STRING ModuleName
,
67 PMODULE_OBJECT
*ModuleObject
);
70 LdrGetExportAddress(PMODULE_OBJECT ModuleObject
,
75 LdrpBuildModuleBaseName(PUNICODE_STRING BaseName
,
76 PUNICODE_STRING FullName
);
79 LdrpCompareModuleNames(IN PUNICODE_STRING String1
,
80 IN PUNICODE_STRING String2
);
83 /* PE Driver load support */
84 static NTSTATUS
LdrPEProcessModule(PVOID ModuleLoadBase
,
85 PUNICODE_STRING FileName
,
86 PMODULE_OBJECT
*ModuleObject
);
88 LdrPEGetExportAddress(PMODULE_OBJECT ModuleObject
,
93 LdrSafePEGetExportAddress(PVOID ImportModuleBase
,
98 LdrPEFixupForward(PCHAR ForwardName
);
101 /* FUNCTIONS *****************************************************************/
104 LdrInitDebug(PLOADER_MODULE Module
, PWCH Name
)
106 PLIST_ENTRY current_entry
;
107 MODULE_TEXT_SECTION
* current
;
109 current_entry
= ModuleTextListHead
.Flink
;
110 while (current_entry
!= &ModuleTextListHead
)
113 CONTAINING_RECORD(current_entry
, MODULE_TEXT_SECTION
, ListEntry
);
114 if (wcscmp(current
->Name
, Name
) == 0)
118 current_entry
= current_entry
->Flink
;
121 if (current_entry
== &ModuleTextListHead
)
130 PIMAGE_DOS_HEADER DosHeader
;
131 PIMAGE_FILE_HEADER FileHeader
;
132 PIMAGE_OPTIONAL_HEADER OptionalHeader
;
133 PIMAGE_SECTION_HEADER SectionList
;
135 InitializeListHead(&ModuleTextListHead
);
137 /* Setup ntoskrnl.exe text section */
138 DosHeader
= (PIMAGE_DOS_HEADER
) KERNEL_BASE
;
140 (PIMAGE_FILE_HEADER
) ((DWORD
)KERNEL_BASE
+
141 DosHeader
->e_lfanew
+ sizeof(ULONG
));
142 OptionalHeader
= (PIMAGE_OPTIONAL_HEADER
)
143 ((DWORD
)FileHeader
+ sizeof(IMAGE_FILE_HEADER
));
144 SectionList
= (PIMAGE_SECTION_HEADER
)
145 ((DWORD
)OptionalHeader
+ sizeof(IMAGE_OPTIONAL_HEADER
));
146 NtoskrnlTextSection
.Base
= KERNEL_BASE
;
147 NtoskrnlTextSection
.Length
= SectionList
[0].Misc
.VirtualSize
+
148 SectionList
[0].VirtualAddress
;
149 NtoskrnlTextSection
.Name
= KERNEL_MODULE_NAME
;
150 NtoskrnlTextSection
.OptionalHeader
= OptionalHeader
;
151 InsertTailList(&ModuleTextListHead
, &NtoskrnlTextSection
.ListEntry
);
153 /* Setup hal.dll text section */
154 DosHeader
= (PIMAGE_DOS_HEADER
)LdrHalBase
;
156 (PIMAGE_FILE_HEADER
) ((DWORD
)LdrHalBase
+
157 DosHeader
->e_lfanew
+ sizeof(ULONG
));
158 OptionalHeader
= (PIMAGE_OPTIONAL_HEADER
)
159 ((DWORD
)FileHeader
+ sizeof(IMAGE_FILE_HEADER
));
160 SectionList
= (PIMAGE_SECTION_HEADER
)
161 ((DWORD
)OptionalHeader
+ sizeof(IMAGE_OPTIONAL_HEADER
));
162 LdrHalTextSection
.Base
= LdrHalBase
;
163 LdrHalTextSection
.Length
= SectionList
[0].Misc
.VirtualSize
+
164 SectionList
[0].VirtualAddress
;
165 LdrHalTextSection
.Name
= HAL_MODULE_NAME
;
166 LdrHalTextSection
.OptionalHeader
= OptionalHeader
;
167 InsertTailList(&ModuleTextListHead
, &LdrHalTextSection
.ListEntry
);
169 /* Hook for KDB on initialization of the loader. */
170 KDB_LOADERINIT_HOOK(&NtoskrnlTextSection
, &LdrHalTextSection
);
175 LdrInitModuleManagement(VOID
)
177 PIMAGE_DOS_HEADER DosHeader
;
178 PMODULE_OBJECT ModuleObject
;
180 /* Initialize the module list and spinlock */
181 InitializeListHead(&ModuleListHead
);
182 KeInitializeSpinLock(&ModuleListLock
);
184 /* Create module object for NTOSKRNL */
185 ModuleObject
= ExAllocatePool(NonPagedPool
, sizeof(MODULE_OBJECT
));
186 assert(ModuleObject
!= NULL
);
187 RtlZeroMemory(ModuleObject
, sizeof(MODULE_OBJECT
));
189 /* Initialize ModuleObject data */
190 ModuleObject
->Base
= (PVOID
) KERNEL_BASE
;
191 ModuleObject
->Flags
= MODULE_FLAG_PE
;
192 RtlCreateUnicodeString(&ModuleObject
->FullName
,
194 LdrpBuildModuleBaseName(&ModuleObject
->BaseName
,
195 &ModuleObject
->FullName
);
197 DosHeader
= (PIMAGE_DOS_HEADER
) KERNEL_BASE
;
198 ModuleObject
->Image
.PE
.FileHeader
=
199 (PIMAGE_FILE_HEADER
) ((DWORD
) ModuleObject
->Base
+
200 DosHeader
->e_lfanew
+ sizeof(ULONG
));
201 ModuleObject
->Image
.PE
.OptionalHeader
= (PIMAGE_OPTIONAL_HEADER
)
202 ((DWORD
)ModuleObject
->Image
.PE
.FileHeader
+ sizeof(IMAGE_FILE_HEADER
));
203 ModuleObject
->Image
.PE
.SectionList
= (PIMAGE_SECTION_HEADER
)
204 ((DWORD
)ModuleObject
->Image
.PE
.OptionalHeader
+ sizeof(IMAGE_OPTIONAL_HEADER
));
205 ModuleObject
->EntryPoint
= (PVOID
) ((DWORD
) ModuleObject
->Base
+
206 ModuleObject
->Image
.PE
.OptionalHeader
->AddressOfEntryPoint
);
207 DPRINT("ModuleObject:%08x entrypoint at %x\n", ModuleObject
, ModuleObject
->EntryPoint
);
208 ModuleObject
->Length
= ModuleObject
->Image
.PE
.OptionalHeader
->SizeOfImage
;
209 ModuleObject
->TextSection
= &NtoskrnlTextSection
;
211 InsertTailList(&ModuleListHead
,
212 &ModuleObject
->ListEntry
);
214 /* Create module object for HAL */
215 ModuleObject
= ExAllocatePool(NonPagedPool
, sizeof(MODULE_OBJECT
));
216 assert(ModuleObject
!= NULL
);
217 RtlZeroMemory(ModuleObject
, sizeof(MODULE_OBJECT
));
219 /* Initialize ModuleObject data */
220 ModuleObject
->Base
= (PVOID
) LdrHalBase
;
221 ModuleObject
->Flags
= MODULE_FLAG_PE
;
223 RtlCreateUnicodeString(&ModuleObject
->FullName
,
225 LdrpBuildModuleBaseName(&ModuleObject
->BaseName
,
226 &ModuleObject
->FullName
);
228 DosHeader
= (PIMAGE_DOS_HEADER
) LdrHalBase
;
229 ModuleObject
->Image
.PE
.FileHeader
=
230 (PIMAGE_FILE_HEADER
) ((DWORD
) ModuleObject
->Base
+
231 DosHeader
->e_lfanew
+ sizeof(ULONG
));
232 ModuleObject
->Image
.PE
.OptionalHeader
= (PIMAGE_OPTIONAL_HEADER
)
233 ((DWORD
)ModuleObject
->Image
.PE
.FileHeader
+ sizeof(IMAGE_FILE_HEADER
));
234 ModuleObject
->Image
.PE
.SectionList
= (PIMAGE_SECTION_HEADER
)
235 ((DWORD
)ModuleObject
->Image
.PE
.OptionalHeader
+ sizeof(IMAGE_OPTIONAL_HEADER
));
236 ModuleObject
->EntryPoint
= (PVOID
) ((DWORD
) ModuleObject
->Base
+
237 ModuleObject
->Image
.PE
.OptionalHeader
->AddressOfEntryPoint
);
238 DPRINT("ModuleObject:%08x entrypoint at %x\n", ModuleObject
, ModuleObject
->EntryPoint
);
239 ModuleObject
->Length
= ModuleObject
->Image
.PE
.OptionalHeader
->SizeOfImage
;
240 ModuleObject
->TextSection
= &LdrHalTextSection
;
242 InsertTailList(&ModuleListHead
,
243 &ModuleObject
->ListEntry
);
247 LdrpLoadImage(PUNICODE_STRING DriverName
,
249 PVOID
*SectionPointer
,
251 PVOID
*ExportSectionPointer
)
253 PMODULE_OBJECT ModuleObject
;
256 ModuleObject
= LdrGetModuleObject(DriverName
);
257 if (ModuleObject
== NULL
)
259 Status
= LdrLoadModule(DriverName
, &ModuleObject
);
260 if (!NT_SUCCESS(Status
))
267 *ModuleBase
= ModuleObject
->Base
;
269 // if (SectionPointer)
270 // *SectionPointer = ModuleObject->
273 *EntryPoint
= ModuleObject
->EntryPoint
;
275 // if (ExportSectionPointer)
276 // *ExportSectionPointer = ModuleObject->
278 return(STATUS_SUCCESS
);
283 LdrpUnloadImage(PVOID ModuleBase
)
285 return(STATUS_NOT_IMPLEMENTED
);
290 LdrpLoadAndCallImage(PUNICODE_STRING ModuleName
)
292 PDRIVER_INITIALIZE DriverEntry
;
293 PMODULE_OBJECT ModuleObject
;
296 ModuleObject
= LdrGetModuleObject(ModuleName
);
297 if (ModuleObject
!= NULL
)
299 return(STATUS_IMAGE_ALREADY_LOADED
);
302 Status
= LdrLoadModule(ModuleName
, &ModuleObject
);
303 if (!NT_SUCCESS(Status
))
308 DriverEntry
= (PDRIVER_INITIALIZE
)ModuleObject
->EntryPoint
;
310 Status
= DriverEntry(NULL
, NULL
);
311 if (!NT_SUCCESS(Status
))
313 LdrUnloadModule(ModuleObject
);
321 LdrLoadModule(PUNICODE_STRING Filename
,
322 PMODULE_OBJECT
*ModuleObject
)
324 PVOID ModuleLoadBase
;
327 OBJECT_ATTRIBUTES ObjectAttributes
;
328 PMODULE_OBJECT Module
;
329 FILE_STANDARD_INFORMATION FileStdInfo
;
330 IO_STATUS_BLOCK IoStatusBlock
;
332 *ModuleObject
= NULL
;
334 DPRINT("Loading Module %wZ...\n", Filename
);
336 /* Open the Module */
337 InitializeObjectAttributes(&ObjectAttributes
,
343 Status
= NtOpenFile(&FileHandle
,
348 FILE_SYNCHRONOUS_IO_NONALERT
);
350 if (!NT_SUCCESS(Status
))
352 CPRINT("Could not open module file: %wZ\n", Filename
);
357 /* Get the size of the file */
358 Status
= NtQueryInformationFile(FileHandle
,
362 FileStandardInformation
);
363 if (!NT_SUCCESS(Status
))
365 CPRINT("Could not get file size\n");
371 /* Allocate nonpageable memory for driver */
372 ModuleLoadBase
= ExAllocatePoolWithTag(NonPagedPool
,
373 FileStdInfo
.EndOfFile
.u
.LowPart
,
375 if (ModuleLoadBase
== NULL
)
377 CPRINT("Could not allocate memory for module");
379 return(STATUS_INSUFFICIENT_RESOURCES
);
383 /* Load driver into memory chunk */
384 Status
= NtReadFile(FileHandle
,
388 FileStdInfo
.EndOfFile
.u
.LowPart
,
390 if (!NT_SUCCESS(Status
))
392 CPRINT("Could not read module file into memory");
393 ExFreePool(ModuleLoadBase
);
401 Status
= LdrProcessModule(ModuleLoadBase
,
404 if (!NT_SUCCESS(Status
))
406 CPRINT("Could not process module");
407 ExFreePool(ModuleLoadBase
);
412 ExFreePool(ModuleLoadBase
);
414 *ModuleObject
= Module
;
416 /* Hook for KDB on loading a driver. */
417 KDB_LOADDRIVER_HOOK(Filename
, Module
);
419 return(STATUS_SUCCESS
);
424 LdrUnloadModule(PMODULE_OBJECT ModuleObject
)
428 /* Remove the module from the module list */
429 KeAcquireSpinLock(&ModuleListLock
,&Irql
);
430 RemoveEntryList(&ModuleObject
->ListEntry
);
431 KeReleaseSpinLock(&ModuleListLock
, Irql
);
433 /* Hook for KDB on unloading a driver. */
434 KDB_UNLOADDRIVER_HOOK(ModuleObject
);
436 /* Free text section */
437 if (ModuleObject
->TextSection
!= NULL
)
439 ExFreePool(ModuleObject
->TextSection
->Name
);
440 RemoveEntryList(&ModuleObject
->TextSection
->ListEntry
);
441 ExFreePool(ModuleObject
->TextSection
);
442 ModuleObject
->TextSection
= NULL
;
445 /* Free module section */
446 // MmFreeSection(ModuleObject->Base);
448 ExFreePool(ModuleObject
);
450 return(STATUS_SUCCESS
);
455 LdrInitializeBootStartDriver(PVOID ModuleLoadBase
,
459 PMODULE_OBJECT ModuleObject
;
460 UNICODE_STRING ModuleName
;
461 PDEVICE_NODE DeviceNode
;
464 WCHAR Buffer
[MAX_PATH
];
469 CHAR TextBuffer
[256];
472 HalQueryDisplayParameters(&x
, &y
, &cx
, &cy
);
473 RtlFillMemory(TextBuffer
, x
, ' ');
474 TextBuffer
[x
] = '\0';
475 HalSetDisplayParameters(0, y
-1);
476 HalDisplayString(TextBuffer
);
478 sprintf(TextBuffer
, "Initializing %s...\n", FileName
);
479 HalSetDisplayParameters(0, y
-1);
480 HalDisplayString(TextBuffer
);
481 HalSetDisplayParameters(cx
, cy
);
483 /* Split the filename into base name and extension */
484 FileExt
= strrchr(FileName
, '.');
486 Length
= FileExt
- FileName
;
488 Length
= strlen(FileName
);
490 if ((FileExt
!= NULL
) && (strcmp(FileExt
, ".sym") == 0))
492 KDB_SYMBOLFILE_HOOK(ModuleLoadBase
, FileName
, Length
);
493 return(STATUS_SUCCESS
);
495 else if ((FileExt
!= NULL
) && !(strcmp(FileExt
, ".sys") == 0))
497 CPRINT("Ignoring non-driver file %s\n", FileName
);
498 return STATUS_SUCCESS
;
501 /* Use IopRootDeviceNode for now */
502 Status
= IopCreateDeviceNode(IopRootDeviceNode
, NULL
, &DeviceNode
);
503 if (!NT_SUCCESS(Status
))
505 CPRINT("Driver load failed, status (%x)\n", Status
);
509 RtlCreateUnicodeStringFromAsciiz(&ModuleName
,
511 Status
= LdrProcessModule(ModuleLoadBase
,
514 RtlFreeUnicodeString(&ModuleName
);
515 if (ModuleObject
== NULL
)
517 IopFreeDeviceNode(DeviceNode
);
518 CPRINT("Driver load failed, status (%x)\n", Status
);
519 return(STATUS_UNSUCCESSFUL
);
523 /* Get the service name from the module name */
524 Start
= wcsrchr(ModuleObject
->BaseName
.Buffer
, L
'\\');
526 Start
= ModuleObject
->BaseName
.Buffer
;
530 Ext
= wcsrchr(ModuleObject
->BaseName
.Buffer
, L
'.');
532 Length
= Ext
- Start
;
534 Length
= wcslen(Start
);
536 wcsncpy(Buffer
, Start
, Length
);
538 RtlCreateUnicodeString(&DeviceNode
->ServiceName
, Buffer
);
540 Status
= IopInitializeDriver(ModuleObject
->EntryPoint
,
542 if (!NT_SUCCESS(Status
))
544 IopFreeDeviceNode(DeviceNode
);
545 CPRINT("Driver load failed, status (%x)\n", Status
);
553 LdrProcessModule(PVOID ModuleLoadBase
,
554 PUNICODE_STRING ModuleName
,
555 PMODULE_OBJECT
*ModuleObject
)
557 PIMAGE_DOS_HEADER PEDosHeader
;
559 /* If MZ header exists */
560 PEDosHeader
= (PIMAGE_DOS_HEADER
) ModuleLoadBase
;
561 if (PEDosHeader
->e_magic
== IMAGE_DOS_MAGIC
&& PEDosHeader
->e_lfanew
!= 0L)
563 return LdrPEProcessModule(ModuleLoadBase
,
568 CPRINT("Module wasn't PE\n");
569 return STATUS_UNSUCCESSFUL
;
574 LdrGetExportAddress(PMODULE_OBJECT ModuleObject
,
578 if (ModuleObject
->Flags
& MODULE_FLAG_PE
)
580 return LdrPEGetExportAddress(ModuleObject
, Name
, Hint
);
590 LdrpQueryModuleInformation(PVOID Buffer
,
594 PLIST_ENTRY current_entry
;
595 PMODULE_OBJECT current
;
596 ULONG ModuleCount
= 0;
597 PSYSTEM_MODULE_INFORMATION Smi
;
598 ANSI_STRING AnsiName
;
602 KeAcquireSpinLock(&ModuleListLock
,&Irql
);
604 /* calculate required size */
605 current_entry
= ModuleListHead
.Flink
;
606 while (current_entry
!= (&ModuleListHead
))
609 current_entry
= current_entry
->Flink
;
612 *ReqSize
= sizeof(SYSTEM_MODULE_INFORMATION
)+
613 (ModuleCount
- 1) * sizeof(SYSTEM_MODULE_INFORMATION_ENTRY
);
617 KeReleaseSpinLock(&ModuleListLock
, Irql
);
618 return(STATUS_INFO_LENGTH_MISMATCH
);
621 /* fill the buffer */
622 memset(Buffer
, '=', Size
);
624 Smi
= (PSYSTEM_MODULE_INFORMATION
)Buffer
;
625 Smi
->Count
= ModuleCount
;
628 current_entry
= ModuleListHead
.Flink
;
629 while (current_entry
!= (&ModuleListHead
))
631 current
= CONTAINING_RECORD(current_entry
,MODULE_OBJECT
,ListEntry
);
633 Smi
->Module
[ModuleCount
].Unknown1
= 0; /* Always 0 */
634 Smi
->Module
[ModuleCount
].Unknown2
= 0; /* Always 0 */
635 Smi
->Module
[ModuleCount
].Base
= current
->Base
;
636 Smi
->Module
[ModuleCount
].Size
= current
->Length
;
637 Smi
->Module
[ModuleCount
].Flags
= 0; /* Flags ??? (GN) */
638 Smi
->Module
[ModuleCount
].Index
= ModuleCount
;
639 Smi
->Module
[ModuleCount
].NameLength
= 0;
640 Smi
->Module
[ModuleCount
].LoadCount
= 0; /* FIXME */
643 AnsiName
.MaximumLength
= 256;
644 AnsiName
.Buffer
= Smi
->Module
[ModuleCount
].ImageName
;
645 RtlUnicodeStringToAnsiString(&AnsiName
,
649 p
= strrchr(AnsiName
.Buffer
, '\\');
652 Smi
->Module
[ModuleCount
].PathLength
= 0;
657 Smi
->Module
[ModuleCount
].PathLength
= p
- AnsiName
.Buffer
;
661 current_entry
= current_entry
->Flink
;
664 KeReleaseSpinLock(&ModuleListLock
, Irql
);
666 return(STATUS_SUCCESS
);
671 LdrpBuildModuleBaseName(PUNICODE_STRING BaseName
,
672 PUNICODE_STRING FullName
)
678 DPRINT("LdrpBuildModuleBaseName()\n");
679 DPRINT("FullName %wZ\n", FullName
);
681 p
= wcsrchr(FullName
->Buffer
, L
'\\');
684 p
= FullName
->Buffer
;
693 RtlCreateUnicodeString(&Name
, p
);
695 q
= wcschr(Name
.Buffer
, L
'.');
703 RtlCreateUnicodeString(BaseName
, Name
.Buffer
);
704 RtlFreeUnicodeString(&Name
);
709 LdrpCompareModuleNames(IN PUNICODE_STRING String1
,
710 IN PUNICODE_STRING String2
)
716 if (String1
&& String2
)
718 /* Search String1 for last path component */
719 len1
= String1
->Length
/ sizeof(WCHAR
);
720 s1
= String1
->Buffer
;
721 for (i
= 0, p
= String1
->Buffer
; i
< String1
->Length
; i
= i
+ sizeof(WCHAR
), p
++)
725 if (i
== String1
->Length
- sizeof(WCHAR
))
733 len1
= (String1
->Length
- i
) / sizeof(WCHAR
);
738 /* Search String2 for last path component */
739 len2
= String2
->Length
/ sizeof(WCHAR
);
740 s2
= String2
->Buffer
;
741 for (i
= 0, p
= String2
->Buffer
; i
< String2
->Length
; i
= i
+ sizeof(WCHAR
), p
++)
745 if (i
== String2
->Length
- sizeof(WCHAR
))
753 len2
= (String2
->Length
- i
) / sizeof(WCHAR
);
758 /* Compare last path components */
763 c1
= len1
-- ? RtlUpcaseUnicodeChar (*s1
++) : 0;
764 c2
= len2
-- ? RtlUpcaseUnicodeChar (*s2
++) : 0;
765 if ((c1
== 0 && c2
== L
'.') || (c1
== L
'.' && c2
== 0))
767 if (!c1
|| !c2
|| c1
!= c2
)
778 LdrGetModuleObject(PUNICODE_STRING ModuleName
)
780 PMODULE_OBJECT Module
;
784 DPRINT("LdrpGetModuleObject(%wZ) called\n", ModuleName
);
786 KeAcquireSpinLock(&ModuleListLock
,&Irql
);
788 Entry
= ModuleListHead
.Flink
;
789 while (Entry
!= &ModuleListHead
)
791 Module
= CONTAINING_RECORD(Entry
, MODULE_OBJECT
, ListEntry
);
793 DPRINT("Comparing %wZ and %wZ\n",
797 if (!LdrpCompareModuleNames(&Module
->BaseName
, ModuleName
))
799 DPRINT("Module %wZ\n", &Module
->BaseName
);
800 KeReleaseSpinLock(&ModuleListLock
, Irql
);
804 Entry
= Entry
->Flink
;
807 KeReleaseSpinLock(&ModuleListLock
, Irql
);
809 DPRINT("Could not find module '%wZ'\n", ModuleName
);
815 /* ---------------------------------------------- PE Module support */
818 PageNeedsWriteAccess(PVOID PageStart
,
820 PIMAGE_FILE_HEADER PEFileHeader
,
821 PIMAGE_SECTION_HEADER PESectionHeaders
)
823 BOOL NeedsWriteAccess
;
825 ULONG Characteristics
;
829 NeedsWriteAccess
= FALSE
;
830 /* Set the protections for the various parts of the driver */
831 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
&& ! NeedsWriteAccess
; Idx
++)
833 Characteristics
= PESectionHeaders
[Idx
].Characteristics
;
834 if (!(Characteristics
& IMAGE_SECTION_CHAR_CODE
) ||
835 (Characteristics
& IMAGE_SECTION_CHAR_WRITABLE
||
836 Characteristics
& IMAGE_SECTION_CHAR_DATA
||
837 Characteristics
& IMAGE_SECTION_CHAR_BSS
))
840 max(PESectionHeaders
[Idx
].Misc
.VirtualSize
,
841 PESectionHeaders
[Idx
].SizeOfRawData
);
842 BaseAddress
= PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
;
843 NeedsWriteAccess
= BaseAddress
< PageStart
+ PAGE_SIZE
&&
844 PageStart
< (PVOID
)((PCHAR
) BaseAddress
+ Length
);
848 return(NeedsWriteAccess
);
852 LdrPEProcessModule(PVOID ModuleLoadBase
,
853 PUNICODE_STRING FileName
,
854 PMODULE_OBJECT
*ModuleObject
)
856 unsigned int DriverSize
, Idx
;
857 ULONG RelocDelta
, NumRelocs
;
858 DWORD CurrentSize
, TotalRelocs
;
861 PIMAGE_DOS_HEADER PEDosHeader
;
862 PIMAGE_FILE_HEADER PEFileHeader
;
863 PIMAGE_OPTIONAL_HEADER PEOptionalHeader
;
864 PIMAGE_SECTION_HEADER PESectionHeaders
;
865 PRELOCATION_DIRECTORY RelocDir
;
866 PRELOCATION_ENTRY RelocEntry
;
867 PMODULE_OBJECT LibraryModuleObject
;
868 PMODULE_OBJECT CreatedModuleObject
;
869 PVOID
*ImportAddressList
;
870 PULONG FunctionNameList
;
873 UNICODE_STRING ModuleName
;
874 UNICODE_STRING NameString
;
875 WCHAR NameBuffer
[60];
876 MODULE_TEXT_SECTION
* ModuleTextSection
;
880 DPRINT("Processing PE Module at module base:%08lx\n", ModuleLoadBase
);
882 /* Get header pointers */
883 PEDosHeader
= (PIMAGE_DOS_HEADER
) ModuleLoadBase
;
884 PEMagic
= (PULONG
) ((unsigned int) ModuleLoadBase
+
885 PEDosHeader
->e_lfanew
);
886 PEFileHeader
= (PIMAGE_FILE_HEADER
) ((unsigned int) ModuleLoadBase
+
887 PEDosHeader
->e_lfanew
+ sizeof(ULONG
));
888 PEOptionalHeader
= (PIMAGE_OPTIONAL_HEADER
) ((unsigned int) ModuleLoadBase
+
889 PEDosHeader
->e_lfanew
+ sizeof(ULONG
) + sizeof(IMAGE_FILE_HEADER
));
890 PESectionHeaders
= (PIMAGE_SECTION_HEADER
) ((unsigned int) ModuleLoadBase
+
891 PEDosHeader
->e_lfanew
+ sizeof(ULONG
) + sizeof(IMAGE_FILE_HEADER
) +
892 sizeof(IMAGE_OPTIONAL_HEADER
));
895 /* Check file magic numbers */
896 if (PEDosHeader
->e_magic
!= IMAGE_DOS_MAGIC
)
898 CPRINT("Incorrect MZ magic: %04x\n", PEDosHeader
->e_magic
);
899 return STATUS_UNSUCCESSFUL
;
901 if (PEDosHeader
->e_lfanew
== 0)
903 CPRINT("Invalid lfanew offset: %08x\n", PEDosHeader
->e_lfanew
);
904 return STATUS_UNSUCCESSFUL
;
906 if (*PEMagic
!= IMAGE_PE_MAGIC
)
908 CPRINT("Incorrect PE magic: %08x\n", *PEMagic
);
909 return STATUS_UNSUCCESSFUL
;
911 if (PEFileHeader
->Machine
!= IMAGE_FILE_MACHINE_I386
)
913 CPRINT("Incorrect Architechture: %04x\n", PEFileHeader
->Machine
);
914 return STATUS_UNSUCCESSFUL
;
918 /* FIXME: if image is fixed-address load, then fail */
920 /* FIXME: check/verify OS version number */
922 DPRINT("OptionalHdrMagic:%04x LinkVersion:%d.%d\n",
923 PEOptionalHeader
->Magic
,
924 PEOptionalHeader
->MajorLinkerVersion
,
925 PEOptionalHeader
->MinorLinkerVersion
);
926 DPRINT("Entry Point:%08lx\n", PEOptionalHeader
->AddressOfEntryPoint
);
929 /* Determine the size of the module */
930 DriverSize
= PEOptionalHeader
->SizeOfImage
;
931 DPRINT("DriverSize %x\n",DriverSize
);
933 /* Allocate a virtual section for the module */
934 DriverBase
= MmAllocateSection(DriverSize
);
937 CPRINT("Failed to allocate a virtual section for driver\n");
938 return STATUS_UNSUCCESSFUL
;
940 DbgPrint("DriverBase for %wZ: %x\n", FileName
, DriverBase
);
942 /* Copy headers over */
943 memcpy(DriverBase
, ModuleLoadBase
, PEOptionalHeader
->SizeOfHeaders
);
945 /* Copy image sections into virtual section */
946 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
; Idx
++)
948 // Copy current section into current offset of virtual section
949 if (PESectionHeaders
[Idx
].Characteristics
&
950 (IMAGE_SECTION_CHAR_CODE
| IMAGE_SECTION_CHAR_DATA
))
952 DPRINT("PESectionHeaders[Idx].VirtualAddress + DriverBase %x\n",
953 PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
);
954 memcpy(PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
,
955 (PVOID
)(ModuleLoadBase
+ PESectionHeaders
[Idx
].PointerToRawData
),
956 PESectionHeaders
[Idx
].Misc
.VirtualSize
> PESectionHeaders
[Idx
].SizeOfRawData
957 ? PESectionHeaders
[Idx
].SizeOfRawData
: PESectionHeaders
[Idx
].Misc
.VirtualSize
);
961 DPRINT("PESectionHeaders[Idx].VirtualAddress + DriverBase %x\n",
962 PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
);
963 memset(PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
,
964 '\0', PESectionHeaders
[Idx
].Misc
.VirtualSize
);
967 CurrentSize
+= ROUND_UP(PESectionHeaders
[Idx
].Misc
.VirtualSize
,
968 PEOptionalHeader
->SectionAlignment
);
971 // CurrentBase = (PVOID)((DWORD)CurrentBase +
972 // ROUND_UP(PESectionHeaders[Idx].SizeOfRawData.Misc.VirtualSize,
973 // PEOptionalHeader->SectionAlignment));
976 /* Perform relocation fixups */
977 RelocDelta
= (DWORD
) DriverBase
- PEOptionalHeader
->ImageBase
;
978 RelocDir
= (PRELOCATION_DIRECTORY
)(PEOptionalHeader
->DataDirectory
[
979 IMAGE_DIRECTORY_ENTRY_BASERELOC
].VirtualAddress
);
980 DPRINT("DrvrBase:%08lx ImgBase:%08lx RelocDelta:%08lx\n",
982 PEOptionalHeader
->ImageBase
,
984 DPRINT("RelocDir %x\n",RelocDir
);
986 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
; Idx
++)
988 if (PESectionHeaders
[Idx
].VirtualAddress
== (DWORD
)RelocDir
)
990 DPRINT("Name %.8s PESectionHeader[Idx].PointerToRawData %x\n",
991 PESectionHeaders
[Idx
].Name
,
992 PESectionHeaders
[Idx
].PointerToRawData
);
993 RelocDir
= PESectionHeaders
[Idx
].PointerToRawData
+
995 CurrentSize
= PESectionHeaders
[Idx
].Misc
.VirtualSize
;
1000 RelocDir
= RelocDir
+ (ULONG
)DriverBase
;
1001 CurrentSize
= PEOptionalHeader
->DataDirectory
1002 [IMAGE_DIRECTORY_ENTRY_BASERELOC
].Size
;
1004 DPRINT("RelocDir %08lx CurrentSize %08lx\n", RelocDir
, CurrentSize
);
1006 while (TotalRelocs
< CurrentSize
&& RelocDir
->SizeOfBlock
!= 0)
1008 NumRelocs
= (RelocDir
->SizeOfBlock
- sizeof(RELOCATION_DIRECTORY
)) /
1010 /* DPRINT("RelocDir at %08lx for VA %08lx with %08lx relocs\n",
1012 RelocDir->VirtualAddress,
1014 RelocEntry
= (PRELOCATION_ENTRY
) ((DWORD
)RelocDir
+
1015 sizeof(RELOCATION_DIRECTORY
));
1016 for (Idx
= 0; Idx
< NumRelocs
; Idx
++)
1022 Offset
= RelocEntry
[Idx
].TypeOffset
& 0xfff;
1023 Type
= (RelocEntry
[Idx
].TypeOffset
>> 12) & 0xf;
1024 RelocItem
= (PDWORD
)(DriverBase
+ RelocDir
->VirtualAddress
+
1026 /* DPRINT(" reloc at %08lx %x %s old:%08lx new:%08lx\n",
1029 Type ? "HIGHLOW" : "ABS",
1031 (*RelocItem) + RelocDelta); */
1034 (*RelocItem
) += RelocDelta
;
1038 CPRINT("Unknown relocation type %x at %x\n",Type
, &Type
);
1039 return STATUS_UNSUCCESSFUL
;
1042 TotalRelocs
+= RelocDir
->SizeOfBlock
;
1043 RelocDir
= (PRELOCATION_DIRECTORY
)((DWORD
)RelocDir
+
1044 RelocDir
->SizeOfBlock
);
1045 // DPRINT("TotalRelocs: %08lx CurrentSize: %08lx\n", TotalRelocs, CurrentSize);
1048 DPRINT("PEOptionalHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT] %x\n",
1049 PEOptionalHeader
->DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
]
1051 /* Perform import fixups */
1052 if (PEOptionalHeader
->DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
].VirtualAddress
)
1054 PIMAGE_IMPORT_MODULE_DIRECTORY ImportModuleDirectory
;
1056 /* Process each import module */
1057 ImportModuleDirectory
= (PIMAGE_IMPORT_MODULE_DIRECTORY
)
1058 ((DWORD
)DriverBase
+ PEOptionalHeader
->
1059 DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
].VirtualAddress
);
1060 DPRINT("Processeing import directory at %p\n", ImportModuleDirectory
);
1061 while (ImportModuleDirectory
->dwRVAModuleName
)
1063 /* Check to make sure that import lib is kernel */
1064 pName
= (PCHAR
) DriverBase
+
1065 ImportModuleDirectory
->dwRVAModuleName
;
1067 RtlCreateUnicodeStringFromAsciiz(&ModuleName
, pName
);
1068 DPRINT("Import module: %wZ\n", &ModuleName
);
1070 LibraryModuleObject
= LdrGetModuleObject(&ModuleName
);
1071 if (LibraryModuleObject
== NULL
)
1073 DPRINT("Module '%wZ' not loaded yet\n", &ModuleName
);
1074 wcscpy(NameBuffer
, L
"\\SystemRoot\\system32\\drivers\\");
1075 wcscat(NameBuffer
, ModuleName
.Buffer
);
1076 RtlInitUnicodeString(&NameString
, NameBuffer
);
1077 Status
= LdrLoadModule(&NameString
, &LibraryModuleObject
);
1078 if (!NT_SUCCESS(Status
))
1080 wcscpy(NameBuffer
, L
"\\SystemRoot\\system32\\");
1081 wcscat(NameBuffer
, ModuleName
.Buffer
);
1082 RtlInitUnicodeString(&NameString
, NameBuffer
);
1083 Status
= LdrLoadModule(&NameString
, &LibraryModuleObject
);
1084 if (!NT_SUCCESS(Status
))
1086 DPRINT1("Unknown import module: %wZ (Status %lx)\n", &ModuleName
, Status
);
1091 /* Get the import address list */
1092 ImportAddressList
= (PVOID
*) ((DWORD
)DriverBase
+
1093 ImportModuleDirectory
->dwRVAFunctionAddressList
);
1095 /* Get the list of functions to import */
1096 if (ImportModuleDirectory
->dwRVAFunctionNameList
!= 0)
1098 FunctionNameList
= (PULONG
) ((DWORD
)DriverBase
+
1099 ImportModuleDirectory
->dwRVAFunctionNameList
);
1103 FunctionNameList
= (PULONG
) ((DWORD
)DriverBase
+
1104 ImportModuleDirectory
->dwRVAFunctionAddressList
);
1106 /* Walk through function list and fixup addresses */
1107 while (*FunctionNameList
!= 0L)
1109 if ((*FunctionNameList
) & 0x80000000) // hint
1114 Hint
= (*FunctionNameList
) & 0xffff;
1118 pName
= (PCHAR
)((DWORD
)DriverBase
+
1119 *FunctionNameList
+ 2);
1120 Hint
= *(PWORD
)((DWORD
)DriverBase
+ *FunctionNameList
);
1122 DPRINT(" Hint:%04x Name:%s\n", Hint
, pName
);
1124 /* Fixup the current import symbol */
1125 if (LibraryModuleObject
!= NULL
)
1127 *ImportAddressList
= LdrGetExportAddress(LibraryModuleObject
,
1133 CPRINT("Unresolved kernel symbol: %s\n", pName
);
1134 return STATUS_UNSUCCESSFUL
;
1136 ImportAddressList
++;
1140 RtlFreeUnicodeString(&ModuleName
);
1142 ImportModuleDirectory
++;
1146 /* Set the protections for the various parts of the driver */
1147 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
; Idx
++)
1149 ULONG Characteristics
= PESectionHeaders
[Idx
].Characteristics
;
1153 if (Characteristics
& IMAGE_SECTION_CHAR_CODE
&&
1154 !(Characteristics
& IMAGE_SECTION_CHAR_WRITABLE
||
1155 Characteristics
& IMAGE_SECTION_CHAR_DATA
||
1156 Characteristics
& IMAGE_SECTION_CHAR_BSS
))
1159 max(PESectionHeaders
[Idx
].Misc
.VirtualSize
,
1160 PESectionHeaders
[Idx
].SizeOfRawData
);
1161 BaseAddress
= PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
;
1162 PageAddress
= (PVOID
)PAGE_ROUND_DOWN(BaseAddress
);
1163 if (! PageNeedsWriteAccess(PageAddress
, DriverBase
, PEFileHeader
, PESectionHeaders
))
1165 MmSetPageProtect(NULL
, PageAddress
, PAGE_READONLY
);
1167 PageAddress
= (PVOID
)((PCHAR
) PageAddress
+ PAGE_SIZE
);
1168 while ((PVOID
)((PCHAR
) PageAddress
+ PAGE_SIZE
) <
1169 (PVOID
)((PCHAR
) BaseAddress
+ Length
))
1171 MmSetPageProtect(NULL
, PageAddress
, PAGE_READONLY
);
1172 PageAddress
= (PVOID
)((PCHAR
) PageAddress
+ PAGE_SIZE
);
1174 if (PageAddress
< (PVOID
)((PCHAR
) BaseAddress
+ Length
) &&
1175 ! PageNeedsWriteAccess(PageAddress
, DriverBase
, PEFileHeader
, PESectionHeaders
))
1177 MmSetPageProtect(NULL
, PageAddress
, PAGE_READONLY
);
1182 /* Create the module */
1183 CreatedModuleObject
= ExAllocatePool(NonPagedPool
, sizeof(MODULE_OBJECT
));
1184 if (CreatedModuleObject
== NULL
)
1186 return(STATUS_INSUFFICIENT_RESOURCES
);
1189 RtlZeroMemory(CreatedModuleObject
, sizeof(MODULE_OBJECT
));
1191 /* Initialize ModuleObject data */
1192 CreatedModuleObject
->Base
= DriverBase
;
1193 CreatedModuleObject
->Flags
= MODULE_FLAG_PE
;
1195 RtlCreateUnicodeString(&CreatedModuleObject
->FullName
,
1197 LdrpBuildModuleBaseName(&CreatedModuleObject
->BaseName
,
1198 &CreatedModuleObject
->FullName
);
1200 CreatedModuleObject
->EntryPoint
=
1201 (PVOID
)((DWORD
)DriverBase
+
1202 PEOptionalHeader
->AddressOfEntryPoint
);
1203 CreatedModuleObject
->Length
= DriverSize
;
1204 DPRINT("EntryPoint at %x\n", CreatedModuleObject
->EntryPoint
);
1206 CreatedModuleObject
->Image
.PE
.FileHeader
=
1207 (PIMAGE_FILE_HEADER
) ((unsigned int) DriverBase
+ PEDosHeader
->e_lfanew
+ sizeof(ULONG
));
1209 DPRINT("FileHeader at %x\n", CreatedModuleObject
->Image
.PE
.FileHeader
);
1210 CreatedModuleObject
->Image
.PE
.OptionalHeader
=
1211 (PIMAGE_OPTIONAL_HEADER
) ((unsigned int) DriverBase
+ PEDosHeader
->e_lfanew
+ sizeof(ULONG
) +
1212 sizeof(IMAGE_FILE_HEADER
));
1213 DPRINT("OptionalHeader at %x\n", CreatedModuleObject
->Image
.PE
.OptionalHeader
);
1214 CreatedModuleObject
->Image
.PE
.SectionList
=
1215 (PIMAGE_SECTION_HEADER
) ((unsigned int) DriverBase
+ PEDosHeader
->e_lfanew
+ sizeof(ULONG
) +
1216 sizeof(IMAGE_FILE_HEADER
) + sizeof(IMAGE_OPTIONAL_HEADER
));
1217 DPRINT("SectionList at %x\n", CreatedModuleObject
->Image
.PE
.SectionList
);
1220 KeAcquireSpinLock(&ModuleListLock
, &Irql
);
1221 InsertTailList(&ModuleListHead
,
1222 &CreatedModuleObject
->ListEntry
);
1223 KeReleaseSpinLock(&ModuleListLock
, Irql
);
1226 ModuleTextSection
= ExAllocatePool(NonPagedPool
,
1227 sizeof(MODULE_TEXT_SECTION
));
1228 assert(ModuleTextSection
);
1229 RtlZeroMemory(ModuleTextSection
, sizeof(MODULE_TEXT_SECTION
));
1230 ModuleTextSection
->Base
= (ULONG
)DriverBase
;
1231 ModuleTextSection
->Length
= DriverSize
;
1232 ModuleTextSection
->Name
= ExAllocatePool(NonPagedPool
,
1233 (wcslen(CreatedModuleObject
->BaseName
.Buffer
) + 1) * sizeof(WCHAR
));
1234 wcscpy(ModuleTextSection
->Name
, CreatedModuleObject
->BaseName
.Buffer
);
1235 ModuleTextSection
->OptionalHeader
=
1236 CreatedModuleObject
->Image
.PE
.OptionalHeader
;
1237 InsertTailList(&ModuleTextListHead
, &ModuleTextSection
->ListEntry
);
1239 CreatedModuleObject
->TextSection
= ModuleTextSection
;
1241 *ModuleObject
= CreatedModuleObject
;
1243 DPRINT("Loading Module %wZ...\n", FileName
);
1245 if ((KdDebuggerEnabled
== TRUE
) && (KdDebugState
& KD_DEBUG_GDB
))
1247 DPRINT("Module %wZ loaded at 0x%.08x.\n",
1248 FileName
, CreatedModuleObject
->Base
);
1251 return STATUS_SUCCESS
;
1256 LdrSafePEProcessModule(PVOID ModuleLoadBase
,
1258 PVOID ImportModuleBase
,
1262 ULONG RelocDelta
, NumRelocs
;
1263 ULONG CurrentSize
, TotalRelocs
;
1265 PIMAGE_DOS_HEADER PEDosHeader
;
1266 PIMAGE_FILE_HEADER PEFileHeader
;
1267 PIMAGE_OPTIONAL_HEADER PEOptionalHeader
;
1268 PIMAGE_SECTION_HEADER PESectionHeaders
;
1269 PRELOCATION_DIRECTORY RelocDir
;
1270 PRELOCATION_ENTRY RelocEntry
;
1271 PVOID
*ImportAddressList
;
1272 PULONG FunctionNameList
;
1276 ps("Processing PE Module at module base:%08lx\n", ModuleLoadBase
);
1278 /* Get header pointers */
1279 PEDosHeader
= (PIMAGE_DOS_HEADER
) ModuleLoadBase
;
1280 PEMagic
= (PULONG
) ((unsigned int) ModuleLoadBase
+
1281 PEDosHeader
->e_lfanew
);
1282 PEFileHeader
= (PIMAGE_FILE_HEADER
) ((unsigned int) ModuleLoadBase
+
1283 PEDosHeader
->e_lfanew
+ sizeof(ULONG
));
1284 PEOptionalHeader
= (PIMAGE_OPTIONAL_HEADER
) ((unsigned int) ModuleLoadBase
+
1285 PEDosHeader
->e_lfanew
+ sizeof(ULONG
) + sizeof(IMAGE_FILE_HEADER
));
1286 PESectionHeaders
= (PIMAGE_SECTION_HEADER
) ((unsigned int) ModuleLoadBase
+
1287 PEDosHeader
->e_lfanew
+ sizeof(ULONG
) + sizeof(IMAGE_FILE_HEADER
) +
1288 sizeof(IMAGE_OPTIONAL_HEADER
));
1291 /* Check file magic numbers */
1292 if (PEDosHeader
->e_magic
!= IMAGE_DOS_MAGIC
)
1296 if (PEDosHeader
->e_lfanew
== 0)
1300 if (*PEMagic
!= IMAGE_PE_MAGIC
)
1304 if (PEFileHeader
->Machine
!= IMAGE_FILE_MACHINE_I386
)
1309 ps("OptionalHdrMagic:%04x LinkVersion:%d.%d\n",
1310 PEOptionalHeader
->Magic
,
1311 PEOptionalHeader
->MajorLinkerVersion
,
1312 PEOptionalHeader
->MinorLinkerVersion
);
1313 ps("Entry Point:%08lx\n", PEOptionalHeader
->AddressOfEntryPoint
);
1315 /* Determine the size of the module */
1316 *DriverSize
= PEOptionalHeader
->SizeOfImage
;
1317 ps("DriverSize %x\n",*DriverSize
);
1319 /* Copy headers over */
1320 if (DriverBase
!= ModuleLoadBase
)
1322 memcpy(DriverBase
, ModuleLoadBase
, PEOptionalHeader
->SizeOfHeaders
);
1325 ps("Hdr: 0x%X\n", (ULONG
)PEOptionalHeader
);
1326 ps("Hdr->SizeOfHeaders: 0x%X\n", (ULONG
)PEOptionalHeader
->SizeOfHeaders
);
1327 ps("FileHdr->NumberOfSections: 0x%X\n", (ULONG
)PEFileHeader
->NumberOfSections
);
1329 /* Ntoskrnl.exe need no relocation fixups since it is linked to run at the same
1330 address as it is mapped */
1331 if (DriverBase
!= ModuleLoadBase
)
1335 /* Copy image sections into virtual section */
1336 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
; Idx
++)
1338 // Copy current section into current offset of virtual section
1339 if (PESectionHeaders
[Idx
].Characteristics
&
1340 (IMAGE_SECTION_CHAR_CODE
| IMAGE_SECTION_CHAR_DATA
))
1342 //ps("PESectionHeaders[Idx].VirtualAddress (%X) + DriverBase %x\n",
1343 //PESectionHeaders[Idx].VirtualAddress, PESectionHeaders[Idx].VirtualAddress + DriverBase);
1344 memcpy(PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
,
1345 (PVOID
)(ModuleLoadBase
+ PESectionHeaders
[Idx
].PointerToRawData
),
1346 PESectionHeaders
[Idx
].Misc
.VirtualSize
> PESectionHeaders
[Idx
].SizeOfRawData
?
1347 PESectionHeaders
[Idx
].SizeOfRawData
: PESectionHeaders
[Idx
].Misc
.VirtualSize
);
1351 ps("PESectionHeaders[Idx].VirtualAddress (%X) + DriverBase %x\n",
1352 PESectionHeaders
[Idx
].VirtualAddress
, PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
);
1353 memset(PESectionHeaders
[Idx
].VirtualAddress
+ DriverBase
,
1355 PESectionHeaders
[Idx
].Misc
.VirtualSize
);
1357 CurrentSize
+= ROUND_UP(PESectionHeaders
[Idx
].Misc
.VirtualSize
,
1358 PEOptionalHeader
->SectionAlignment
);
1361 /* Perform relocation fixups */
1362 RelocDelta
= (ULONG
) DriverBase
- PEOptionalHeader
->ImageBase
;
1363 RelocDir
= (PRELOCATION_DIRECTORY
)(PEOptionalHeader
->DataDirectory
[
1364 IMAGE_DIRECTORY_ENTRY_BASERELOC
].VirtualAddress
);
1365 ps("DrvrBase:%08lx ImgBase:%08lx RelocDelta:%08lx\n",
1367 PEOptionalHeader
->ImageBase
,
1369 ps("RelocDir %x\n",RelocDir
);
1371 for (Idx
= 0; Idx
< PEFileHeader
->NumberOfSections
; Idx
++)
1373 if (PESectionHeaders
[Idx
].VirtualAddress
== (ULONG
)RelocDir
)
1375 DPRINT("Name %.8s PESectionHeader[Idx].PointerToRawData %x\n",
1376 PESectionHeaders
[Idx
].Name
,
1377 PESectionHeaders
[Idx
].PointerToRawData
);
1378 RelocDir
= PESectionHeaders
[Idx
].PointerToRawData
+ ModuleLoadBase
;
1379 CurrentSize
= PESectionHeaders
[Idx
].Misc
.VirtualSize
;
1384 ps("RelocDir %08lx CurrentSize %08lx\n", RelocDir
, CurrentSize
);
1387 while (TotalRelocs
< CurrentSize
&& RelocDir
->SizeOfBlock
!= 0)
1389 NumRelocs
= (RelocDir
->SizeOfBlock
- sizeof(RELOCATION_DIRECTORY
)) /
1391 RelocEntry
= (PRELOCATION_ENTRY
)((ULONG
)RelocDir
+
1392 sizeof(RELOCATION_DIRECTORY
));
1393 for (Idx
= 0; Idx
< NumRelocs
; Idx
++)
1399 Offset
= RelocEntry
[Idx
].TypeOffset
& 0xfff;
1400 Type
= (RelocEntry
[Idx
].TypeOffset
>> 12) & 0xf;
1401 RelocItem
= (PULONG
)(DriverBase
+ RelocDir
->VirtualAddress
+ Offset
);
1404 (*RelocItem
) += RelocDelta
;
1408 CPRINT("Unknown relocation type %x at %x\n",Type
, &Type
);
1412 TotalRelocs
+= RelocDir
->SizeOfBlock
;
1413 RelocDir
= (PRELOCATION_DIRECTORY
)((ULONG
)RelocDir
+
1414 RelocDir
->SizeOfBlock
);
1417 ps("PEOptionalHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT] %x\n",
1418 PEOptionalHeader
->DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
]
1422 /* Perform import fixups */
1423 if (PEOptionalHeader
->DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
].VirtualAddress
)
1425 PIMAGE_IMPORT_MODULE_DIRECTORY ImportModuleDirectory
;
1427 /* Process each import module */
1428 ImportModuleDirectory
= (PIMAGE_IMPORT_MODULE_DIRECTORY
)
1429 ((ULONG
)DriverBase
+ PEOptionalHeader
->
1430 DataDirectory
[IMAGE_DIRECTORY_ENTRY_IMPORT
].VirtualAddress
);
1432 ps("Processeing import directory at %p\n", ImportModuleDirectory
);
1434 /* Check to make sure that import lib is kernel */
1435 pName
= (PCHAR
)DriverBase
+ ImportModuleDirectory
->dwRVAModuleName
;
1437 ps("Import module: %s\n", pName
);
1439 /* Get the import address list */
1440 ImportAddressList
= (PVOID
*)((ULONG
)DriverBase
+
1441 ImportModuleDirectory
->dwRVAFunctionAddressList
);
1443 ps(" ImportModuleDirectory->dwRVAFunctionAddressList: 0x%X\n",
1444 ImportModuleDirectory
->dwRVAFunctionAddressList
);
1445 ps(" ImportAddressList: 0x%X\n", ImportAddressList
);
1447 /* Get the list of functions to import */
1448 if (ImportModuleDirectory
->dwRVAFunctionNameList
!= 0)
1450 ps("Using function name list.\n");
1452 FunctionNameList
= (PULONG
)((ULONG
)DriverBase
+
1453 ImportModuleDirectory
->dwRVAFunctionNameList
);
1457 ps("Using function address list.\n");
1459 FunctionNameList
= (PULONG
)((ULONG
)DriverBase
+
1460 ImportModuleDirectory
->dwRVAFunctionAddressList
);
1463 /* Walk through function list and fixup addresses */
1464 while (*FunctionNameList
!= 0L)
1466 if ((*FunctionNameList
) & 0x80000000)
1470 Hint
= (*FunctionNameList
) & 0xffff;
1475 pName
= (PCHAR
)((ULONG
)DriverBase
+ *FunctionNameList
+ 2);
1476 Hint
= *(PWORD
)((ULONG
)DriverBase
+ *FunctionNameList
);
1478 //ps(" Hint:%04x Name:%s(0x%X)(%x)\n", Hint, pName, pName, ImportAddressList);
1480 *ImportAddressList
= LdrSafePEGetExportAddress(ImportModuleBase
,
1484 ImportAddressList
++;
1489 ps("Finished importing.\n");
1496 LdrPEGetExportAddress(PMODULE_OBJECT ModuleObject
,
1500 PIMAGE_EXPORT_DIRECTORY ExportDir
;
1501 ULONG ExportDirSize
;
1503 PVOID ExportAddress
;
1505 PDWORD FunctionList
, NameList
;
1507 ExportDir
= (PIMAGE_EXPORT_DIRECTORY
)
1508 RtlImageDirectoryEntryToData(ModuleObject
->Base
,
1510 IMAGE_DIRECTORY_ENTRY_EXPORT
,
1512 DPRINT("ExportDir %p ExportDirSize %lx\n", ExportDir
, ExportDirSize
);
1513 if (ExportDir
== NULL
)
1518 FunctionList
= (PDWORD
)((DWORD
)ExportDir
->AddressOfFunctions
+ ModuleObject
->Base
);
1519 NameList
= (PDWORD
)((DWORD
)ExportDir
->AddressOfNames
+ ModuleObject
->Base
);
1520 OrdinalList
= (PWORD
)((DWORD
)ExportDir
->AddressOfNameOrdinals
+ ModuleObject
->Base
);
1526 for (Idx
= 0; Idx
< ExportDir
->NumberOfNames
; Idx
++)
1529 DPRINT(" Name:%s NameList[%d]:%s\n",
1532 (DWORD
) ModuleObject
->Base
+ NameList
[Idx
]);
1535 if (!strcmp(Name
, (PCHAR
) ((DWORD
)ModuleObject
->Base
+ NameList
[Idx
])))
1537 ExportAddress
= (PVOID
) ((DWORD
)ModuleObject
->Base
+
1538 FunctionList
[OrdinalList
[Idx
]]);
1539 if (((ULONG
)ExportAddress
>= (ULONG
)ExportDir
) &&
1540 ((ULONG
)ExportAddress
< (ULONG
)ExportDir
+ ExportDirSize
))
1542 DPRINT("Forward: %s\n", (PCHAR
)ExportAddress
);
1543 ExportAddress
= LdrPEFixupForward((PCHAR
)ExportAddress
);
1544 DPRINT("ExportAddress: %p\n", ExportAddress
);
1553 ExportAddress
= (PVOID
) ((DWORD
)ModuleObject
->Base
+
1554 FunctionList
[Hint
- ExportDir
->Base
]);
1557 if (ExportAddress
== NULL
)
1559 CPRINT("Export not found for %d:%s\n",
1561 Name
!= NULL
? Name
: "(Ordinal)");
1565 return(ExportAddress
);
1570 LdrSafePEGetExportAddress(PVOID ImportModuleBase
,
1575 PVOID ExportAddress
;
1577 PDWORD FunctionList
, NameList
;
1578 PIMAGE_EXPORT_DIRECTORY ExportDir
;
1579 ULONG ExportDirSize
;
1581 static BOOLEAN EP
= FALSE
;
1583 ExportDir
= (PIMAGE_EXPORT_DIRECTORY
)
1584 RtlImageDirectoryEntryToData(ImportModuleBase
,
1586 IMAGE_DIRECTORY_ENTRY_EXPORT
,
1591 ps("ExportDir %x\n", ExportDir
);
1594 FunctionList
= (PDWORD
)((DWORD
)ExportDir
->AddressOfFunctions
+ ImportModuleBase
);
1595 NameList
= (PDWORD
)((DWORD
)ExportDir
->AddressOfNames
+ ImportModuleBase
);
1596 OrdinalList
= (PWORD
)((DWORD
)ExportDir
->AddressOfNameOrdinals
+ ImportModuleBase
);
1602 for (Idx
= 0; Idx
< ExportDir
->NumberOfNames
; Idx
++)
1604 if (!strcmp(Name
, (PCHAR
) ((DWORD
)ImportModuleBase
+ NameList
[Idx
])))
1606 ExportAddress
= (PVOID
) ((DWORD
)ImportModuleBase
+
1607 FunctionList
[OrdinalList
[Idx
]]);
1614 ExportAddress
= (PVOID
) ((DWORD
)ImportModuleBase
+
1616 FunctionList
[Hint
- ExportDir
->Base
]);
1619 if (ExportAddress
== 0)
1621 ps("Export not found for %d:%s\n",
1623 Name
!= NULL
? Name
: "(Ordinal)");
1626 return ExportAddress
;
1631 LdrPEFixupForward(PCHAR ForwardName
)
1633 CHAR NameBuffer
[128];
1634 UNICODE_STRING ModuleName
;
1636 PMODULE_OBJECT ModuleObject
;
1638 DPRINT("LdrPEFixupForward (%s)\n", ForwardName
);
1640 strcpy(NameBuffer
, ForwardName
);
1641 p
= strchr(NameBuffer
, '.');
1649 DPRINT("Driver: %s Function: %s\n", NameBuffer
, p
+1);
1651 RtlCreateUnicodeStringFromAsciiz(&ModuleName
,
1653 ModuleObject
= LdrGetModuleObject(&ModuleName
);
1654 RtlFreeUnicodeString(&ModuleName
);
1656 DPRINT("ModuleObject: %p\n", ModuleObject
);
1658 if (ModuleObject
== NULL
)
1660 CPRINT("LdrPEFixupForward: failed to find module %s\n", NameBuffer
);
1664 return(LdrPEGetExportAddress(ModuleObject
, p
+1, 0));