1) correct the assert condition for LibPcdSetPtr and LibPcdSetExPtr

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

diff --git a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
index a73d804797836441a2df70cb6906f4956fe846df..4018e44f702b45514164bbbeb195bcf6dc5711f8 100644 (file)
--- a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
+++ b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c
@@ -586,10 +586,12 @@ LibPcdSetPtr (
   PCD_PPI    *PcdPpi;\r
   UINTN      Size;\r
 \r
+  if (*SizeOfBuffer > 0) {\r
+    ASSERT (Buffer != NULL);\r
+  }\r
+\r
   PcdPpi = GetPcdPpiPtr ();\r
   \r
-  ASSERT ((*SizeOfBuffer > 0) && Buffer == NULL);\r
-\r
   Size = LibPcdGetSize (TokenNumber);\r
   \r
   if (*SizeOfBuffer > Size) {\r
@@ -813,6 +815,10 @@ LibPcdSetExPtr (
   PCD_PPI         *PcdPpi;\r
   UINTN           Size;\r
 \r
+  if (*SizeOfBuffer > 0) {\r
+    ASSERT (Buffer != NULL);\r
+  }\r
+\r
   PcdPpi = GetPcdPpiPtr ();\r
 \r
   Size = LibPcdGetExSize (Guid, TokenNumber);\r