MdePkg PcdLib: Follow the comment of LibPcdGetNextToken to ASSERT when an invalid...

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

diff --git a/MdePkg/Library/DxePcdLib/DxePcdLib.c b/MdePkg/Library/DxePcdLib/DxePcdLib.c
index bc87c4e6a153fd0aafc990ef283787aba277fe82..f14a74652978bb38f6c1af867ab4294c7faa6e77 100644 (file)
--- a/MdePkg/Library/DxePcdLib/DxePcdLib.c
+++ b/MdePkg/Library/DxePcdLib/DxePcdLib.c
@@ -1002,7 +1002,10 @@ LibPcdGetNextToken (
   IN UINTN                    TokenNumber\r
   )\r
 {\r
   IN UINTN                    TokenNumber\r
   )\r
 {\r

-  GetPiPcdProtocol()->GetNextToken (Guid, &TokenNumber);\r
+  EFI_STATUS    Status;\r
+\r
+  Status = GetPiPcdProtocol()->GetNextToken (Guid, &TokenNumber);\r
+  ASSERT (!EFI_ERROR (Status) || TokenNumber == 0);\r

 \r
   return TokenNumber;\r
 }\r
 \r
   return TokenNumber;\r
 }\r