MdeModulePkg UhciPei: Also check TempPtr against NULL to return error

[mirror_edk2.git] / MdeModulePkg / Bus / Pci / UhciPei / UhcPeim.c

diff --git a/MdeModulePkg/Bus/Pci/UhciPei/UhcPeim.c b/MdeModulePkg/Bus/Pci/UhciPei/UhcPeim.c
index b7d60db0c94e090f70a7bd80352b41494257d96c..482c404c0eaeadec2068acaf652e18e10e440c2f 100644 (file)
--- a/MdeModulePkg/Bus/Pci/UhciPei/UhcPeim.c
+++ b/MdeModulePkg/Bus/Pci/UhciPei/UhcPeim.c
@@ -2863,8 +2863,8 @@ CreateMemoryBlock (
              &MappedAddr,\r
              &Mapping\r
              );\r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
+  if (EFI_ERROR (Status) || (TempPtr == NULL)) {\r
+    return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
   Ptr = TempPtr;\r