MdePkg PeiServicesLib: Refine the code to avoid error report.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Star Zeng <star.zeng@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@14769 6f19259b-4bc3-4df7-8a09-765794883524

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