X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FCore%2FDxe%2FSectionExtraction%2FCoreSectionExtraction.c;h=70c71c6db409303147bad1d55713eb453dbc7f2c;hp=b86f273f33898f5bcd1fa878fc3e97b03b8775c6;hb=7a31af6a12e9586fc0d6bed972c00a0aafd728d8;hpb=39a4d286523188429aec6ca9b09187fdc87f7a1f

diff --git a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
index b86f273f33..70c71c6db4 100644
--- a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
+++ b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
@@ -262,7 +262,7 @@ Returns:
 {
   EFI_STATUS                         Status;
   EFI_GUID                           **DecompressGuidList;
-  UINT32                             DecompressMethodNumber;
+  UINTN                              DecompressMethodNumber;
 
   //
   // Install SEP to a new handle
@@ -1474,6 +1474,7 @@ CustomDecompressExtractSection (
 {
   EFI_STATUS      Status;
   UINT8           *ScratchBuffer;
+  UINT32          DestinationSize;
   UINT32          ScratchSize;
   UINT32          SectionLength;  
   
@@ -1492,7 +1493,7 @@ CustomDecompressExtractSection (
              (GUID *) ((UINT8 *) InputSection + sizeof (EFI_COMMON_SECTION_HEADER)),
              (UINT8 *) InputSection + sizeof (EFI_GUID_DEFINED_SECTION),
              SectionLength - sizeof (EFI_GUID_DEFINED_SECTION),
-             OutputSize,
+             &DestinationSize,
              &ScratchSize
              );
   if (EFI_ERROR (Status)) {
@@ -1513,6 +1514,7 @@ CustomDecompressExtractSection (
   //
   // Allocate destination buffer
   //
+  *OutputSize = (UINTN) DestinationSize;
   *OutputBuffer = CoreAllocateBootServicesPool (*OutputSize);
   if (*OutputBuffer == NULL) {
     CoreFreePool (ScratchBuffer);