[NTOS:IO] Fix an ASSERT. Addendum to commit 1fd730b7.

This commit is contained in:
Hermès Bélusca-Maïto 2021-09-06 01:05:14 +02:00
parent fe9ac14aa3
commit 4795d953c0
No known key found for this signature in database
GPG key ID: 3B2539C65E7B93D0

View file

@ -454,7 +454,8 @@ IopInitializeDriverModule(
*/
PIMAGE_NT_HEADERS NtHeaders = RtlImageNtHeader(ModuleObject->DllBase);
ASSERT(NtHeaders);
ASSERT(ModuleObject->SizeOfImage == NtHeaders->OptionalHeader.SizeOfImage);
// NOTE: ModuleObject->SizeOfImage is actually (number of PTEs)*PAGE_SIZE.
ASSERT(ModuleObject->SizeOfImage == ROUND_TO_PAGES(NtHeaders->OptionalHeader.SizeOfImage));
ASSERT(ModuleObject->EntryPoint == RVA(ModuleObject->DllBase, NtHeaders->OptionalHeader.AddressOfEntryPoint));
/* Obtain the registry path for the DriverInit routine */