[ros-diffs] [fireball] 36300: - Save SizeOfImage field from the image header during PE loading.

Author: fireball
Date: Thu Sep 18 06:19:29 2008
New Revision: 36300

URL: http://svn.reactos.org/svn/reactos?rev=36300&view=rev
Log:
- Save SizeOfImage field from the image header during PE loading.

Modified:
    trunk/reactos/ntoskrnl/mm/pe.c

Modified: trunk/reactos/ntoskrnl/mm/pe.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/ntoskrnl/mm/pe.c?rev=36300…
==============================================================================
--- trunk/reactos/ntoskrnl/mm/pe.c [iso-8859-1] (original)
+++ trunk/reactos/ntoskrnl/mm/pe.c [iso-8859-1] Thu Sep 18 06:19:29 2008
@@ -362,6 +362,9 @@
 	    if(RTL_CONTAINS_FIELD(piohOptHeader, cbOptHeaderSize, ImageBase))
 		ImageSectionObject->ImageBase = piohOptHeader->ImageBase;
 
+	    if(RTL_CONTAINS_FIELD(piohOptHeader, cbOptHeaderSize, SizeOfImage))
+		ImageSectionObject->ImageSize = piohOptHeader->SizeOfImage;
+
 	    if(RTL_CONTAINS_FIELD(piohOptHeader, cbOptHeaderSize, SizeOfStackReserve))
 		ImageSectionObject->StackReserve = piohOptHeader->SizeOfStackReserve;
 
@@ -384,6 +387,14 @@
 		    DIE(("ImageBase exceeds the address space\n"));
 
 		ImageSectionObject->ImageBase = pioh64OptHeader->ImageBase;
+	    }
+
+	    if(RTL_CONTAINS_FIELD(pioh64OptHeader, cbOptHeaderSize, SizeOfImage))
+	    {
+		if(pioh64OptHeader->SizeOfImage > MAXULONG_PTR)
+		    DIE(("SizeOfImage exceeds the address space\n"));
+
+		ImageSectionObject->ImageSize = pioh64OptHeader->SizeOfImage;
 	    }
 
 	    if(RTL_CONTAINS_FIELD(pioh64OptHeader, cbOptHeaderSize, SizeOfStackReserve))