Add more check to avoid access violation.

[mirror_edk2.git] / EdkCompatibilityPkg / Foundation / Library / Dxe / EfiIfrSupportLib / IfrCommon.c

diff --git a/EdkCompatibilityPkg/Foundation/Library/Dxe/EfiIfrSupportLib/IfrCommon.c b/EdkCompatibilityPkg/Foundation/Library/Dxe/EfiIfrSupportLib/IfrCommon.c
index 400b367136355f287821da738a93bff3d982e458..c0ce1b23f8c725f0910693009d6e1528a3edde5b 100644 (file)
--- a/EdkCompatibilityPkg/Foundation/Library/Dxe/EfiIfrSupportLib/IfrCommon.c
+++ b/EdkCompatibilityPkg/Foundation/Library/Dxe/EfiIfrSupportLib/IfrCommon.c
@@ -582,7 +582,7 @@ Returns:
   //\r
   // Return an error if buffer is too small\r
   //\r
-  if (SizeOfNvStore > *ImageLength) {\r
+  if (SizeOfNvStore > *ImageLength || DefaultImage == NULL) {\r
     gBS->FreePool (OldData);\r
     *ImageLength = (UINT16) SizeOfNvStore;\r
     return EFI_BUFFER_TOO_SMALL;\r