MdePkg PeiServicesLib: Refine the code to avoid error report.

[mirror_edk2.git] / MdePkg / Library / PeiServicesLib / PeiServicesLib.c

diff --git a/MdePkg/Library/PeiServicesLib/PeiServicesLib.c b/MdePkg/Library/PeiServicesLib/PeiServicesLib.c
index e10dd8800fb9226812005a0df40e6fd6cc2354b2..3428addcc63bff491a10c335db9fd3ff56d45a70 100644 (file)
--- a/MdePkg/Library/PeiServicesLib/PeiServicesLib.c
+++ b/MdePkg/Library/PeiServicesLib/PeiServicesLib.c
@@ -624,12 +624,14 @@ InternalPeiServicesInstallFvInfoPpi (
     // To install FvInfo Ppi.\r
     //\r
     FvInfoPpi = AllocateZeroPool (sizeof (EFI_PEI_FIRMWARE_VOLUME_INFO_PPI));\r
     // To install FvInfo Ppi.\r
     //\r
     FvInfoPpi = AllocateZeroPool (sizeof (EFI_PEI_FIRMWARE_VOLUME_INFO_PPI));\r

+    ASSERT (FvInfoPpi != NULL);\r

     PpiGuid = &gEfiPeiFirmwareVolumeInfoPpiGuid;\r
   } else {\r
     //\r
     // To install FvInfo2 Ppi.\r
     //\r
     FvInfoPpi = AllocateZeroPool (sizeof (EFI_PEI_FIRMWARE_VOLUME_INFO2_PPI));\r
     PpiGuid = &gEfiPeiFirmwareVolumeInfoPpiGuid;\r
   } else {\r
     //\r
     // To install FvInfo2 Ppi.\r
     //\r
     FvInfoPpi = AllocateZeroPool (sizeof (EFI_PEI_FIRMWARE_VOLUME_INFO2_PPI));\r

+    ASSERT (FvInfoPpi != NULL);\r

     ((EFI_PEI_FIRMWARE_VOLUME_INFO2_PPI *) FvInfoPpi)->AuthenticationStatus = AuthenticationStatus;\r
     PpiGuid = &gEfiPeiFirmwareVolumeInfo2PpiGuid;\r
   }\r
     ((EFI_PEI_FIRMWARE_VOLUME_INFO2_PPI *) FvInfoPpi)->AuthenticationStatus = AuthenticationStatus;\r
     PpiGuid = &gEfiPeiFirmwareVolumeInfo2PpiGuid;\r
   }\r