X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=EdkModulePkg%2FUniversal%2FPCD%2FPei%2FPcd.c;h=031259b22a6e694ecf68a1319a7d88186c363187;hp=1f6db7a0e7a73b045579f563bbbdded3ee0b5c23;hb=4c114006de95489973d33a5415bb451bcd6cd62a;hpb=b560bb65c476d0477667ad820c89d6f8ef3043b1

diff --git a/EdkModulePkg/Universal/PCD/Pei/Pcd.c b/EdkModulePkg/Universal/PCD/Pei/Pcd.c
index 1f6db7a0e7..031259b22a 100644
--- a/EdkModulePkg/Universal/PCD/Pei/Pcd.c
+++ b/EdkModulePkg/Universal/PCD/Pei/Pcd.c
@@ -50,7 +50,7 @@ PCD_PPI mPcdPpiInstance = {
   PeiPcdSetPtrEx,         
   PeiPcdSetBoolEx,
 
-  PcdRegisterCallBackOnSet,
+  PeiRegisterCallBackOnSet,
   PcdUnRegisterCallBackOnSet,
   PeiPcdGetNextToken
 };
@@ -454,12 +454,14 @@ PeiPcdSetBoolEx (
 
 EFI_STATUS
 EFIAPI
-PcdRegisterCallBackOnSet (
+PeiRegisterCallBackOnSet (
   IN  PCD_TOKEN_NUMBER            ExTokenNumber,
   IN  CONST EFI_GUID              *Guid, OPTIONAL
   IN  PCD_PPI_CALLBACK            CallBackFunction
   )
 {
+  ASSERT (CallBackFunction != NULL);
+  
   return PeiRegisterCallBackWorker (ExTokenNumber, Guid, CallBackFunction, TRUE);
 }
 
@@ -473,6 +475,8 @@ PcdUnRegisterCallBackOnSet (
   IN  PCD_PPI_CALLBACK            CallBackFunction
   )
 {
+  ASSERT (CallBackFunction != NULL);
+  
   return PeiRegisterCallBackWorker (ExTokenNumber, Guid, CallBackFunction, FALSE);
 }
 
@@ -482,7 +486,7 @@ EFI_STATUS
 EFIAPI
 PeiPcdGetNextToken (
   IN CONST EFI_GUID               *Guid, OPTIONAL
-  IN OUT  PCD_TOKEN_NUMBER                   *TokenNumber
+  IN OUT  PCD_TOKEN_NUMBER        *TokenNumber
   )
 {
   UINTN               GuidTableIdx;