diff --git a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c index c7aed6dbda..57cf646f6e 100644 --- a/MdePkg/Library/PeiPcdLib/PeiPcdLib.c +++ b/MdePkg/Library/PeiPcdLib/PeiPcdLib.c @@ -869,6 +869,8 @@ LibPcdCallbackOnSet ( { EFI_STATUS Status; + ASSERT (NotificationFunction != NULL); + Status = (GetPcdPpiPointer ())->CallbackOnSet (Guid, TokenNumber, NotificationFunction); ASSERT_EFI_ERROR (Status); @@ -902,6 +904,8 @@ LibPcdCancelCallback ( { EFI_STATUS Status; + ASSERT (NotificationFunction != NULL); + Status = (GetPcdPpiPointer ())->CancelCallback (Guid, TokenNumber, NotificationFunction); ASSERT_EFI_ERROR (Status);