Increased the image buffer size to avoid invalid address access issue

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2200 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 4 insertions, 5 deletions

diff --git a/Tools/CCode/Source/PeiRebase/PeiRebaseExe.c b/Tools/CCode/Source/PeiRebase/PeiRebaseExe.c
index 5f7dfc2..e153110 100644
--- a/Tools/CCode/Source/PeiRebase/PeiRebaseExe.c
+++ b/Tools/CCode/Source/PeiRebase/PeiRebaseExe.c
@@ -749,14 +749,13 @@ Returns:
     // Allocate a buffer for the image to be loaded into.

     //

     Pe32ImageSize       = GetLength (CurrentPe32Section.Pe32Section->CommonHeader.Size) - sizeof (EFI_PE32_SECTION);

-    MemoryImagePointer  = (UINTN) (malloc (Pe32ImageSize + 0x10000));

+    MemoryImagePointer  = (UINTN) (malloc (Pe32ImageSize + 0x100000));

     if (MemoryImagePointer == 0) {

       Error (NULL, 0, 0, "memory allocation failure", NULL);

       return EFI_OUT_OF_RESOURCES;

     }

-    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x10000);

+    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x100000);

     MemoryImagePointerAligned = (MemoryImagePointer + 0x0FFFF) & (-1 << 16);

-    

 

     ImageContext.ImageAddress = MemoryImagePointerAligned;

 

@@ -1078,13 +1077,13 @@ Returns:
     //

     // Allocate a buffer for the image to be loaded into.

     //

-    MemoryImagePointer = (UINTN) (malloc (Pe32ImageSize + 0x10000));

+    MemoryImagePointer = (UINTN) (malloc (Pe32ImageSize + 0x100000));

     if (MemoryImagePointer == 0) {

       Error (NULL, 0, 0, "memory allocation error on rebase of TE image", FileGuidString);

       free (TEBuffer);

       return EFI_OUT_OF_RESOURCES;

     }

-    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x10000);

+    memset ((void *) MemoryImagePointer, 0, Pe32ImageSize + 0x100000);

     MemoryImagePointerAligned = (MemoryImagePointer + 0x0FFFF) & (-1 << 16);