Correct parameter UINTN to UINT32. Fix UINTN conver to UINT32

[mirror_edk2.git] / MdeModulePkg / Core / Dxe / SectionExtraction / CoreSectionExtraction.c

diff --git a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
index b86f273f33898f5bcd1fa878fc3e97b03b8775c6..de719f3e2e9c1d4036fefc27c1c60fce59816934 100644 (file)
--- a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
+++ b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c
@@ -1474,6 +1474,7 @@ CustomDecompressExtractSection (
 {\r
   EFI_STATUS      Status;\r
   UINT8           *ScratchBuffer;\r
+  UINT32          DestinationSize;\r
   UINT32          ScratchSize;\r
   UINT32          SectionLength;  \r
   \r
@@ -1492,7 +1493,7 @@ CustomDecompressExtractSection (
              (GUID *) ((UINT8 *) InputSection + sizeof (EFI_COMMON_SECTION_HEADER)),\r
              (UINT8 *) InputSection + sizeof (EFI_GUID_DEFINED_SECTION),\r
              SectionLength - sizeof (EFI_GUID_DEFINED_SECTION),\r
-             OutputSize,\r
+             &DestinationSize,\r
              &ScratchSize\r
              );\r
   if (EFI_ERROR (Status)) {\r
@@ -1513,6 +1514,7 @@ CustomDecompressExtractSection (
   //\r
   // Allocate destination buffer\r
   //\r
+  *OutputSize = (UINTN) DestinationSize;\r
   *OutputBuffer = CoreAllocateBootServicesPool (*OutputSize);\r
   if (*OutputBuffer == NULL) {\r
     CoreFreePool (ScratchBuffer);\r