diff options
-rw-r--r-- | MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c | 2 | ||||
-rw-r--r-- | MdeModulePkg/Core/DxeIplPeim/DxeLoad.c | 6 |
2 files changed, 5 insertions, 3 deletions
diff --git a/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c b/MdeModulePkg/Core/Dxe/SectionExtraction/CoreSectionExtraction.c index 70c71c6db4..de719f3e2e 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;
- UINTN DecompressMethodNumber;
+ UINT32 DecompressMethodNumber;
//
// Install SEP to a new handle
diff --git a/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c b/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c index e7359e5684..381d60cb2a 100644 --- a/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c +++ b/MdeModulePkg/Core/DxeIplPeim/DxeLoad.c @@ -958,7 +958,8 @@ CustomDecompressExtractSection ( EFI_STATUS Status;
UINT8 *ScratchBuffer;
UINT32 ScratchSize;
- UINT32 SectionLength;
+ UINT32 SectionLength;
+ UINT32 DestinationSize;
//
// Set authentic value to zero.
@@ -975,7 +976,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)) {
@@ -996,6 +997,7 @@ CustomDecompressExtractSection ( //
// Allocate destination buffer
//
+ *OutputSize = (UINTN) DestinationSize;
*OutputBuffer = AllocatePages (EFI_SIZE_TO_PAGES (*OutputSize));
if (*OutputBuffer == NULL) {
return EFI_OUT_OF_RESOURCES;
|