From 4795d953c01b2e4bf21c2843f347e3f6c201db0e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herm=C3=A8s=20B=C3=A9lusca-Ma=C3=AFto?= Date: Mon, 6 Sep 2021 01:05:14 +0200 Subject: [PATCH] [NTOS:IO] Fix an ASSERT. Addendum to commit 1fd730b7. --- ntoskrnl/io/iomgr/driver.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ntoskrnl/io/iomgr/driver.c b/ntoskrnl/io/iomgr/driver.c index eb3817b98ec..8be45e3459a 100644 --- a/ntoskrnl/io/iomgr/driver.c +++ b/ntoskrnl/io/iomgr/driver.c @@ -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 */