Add security check.

Signed-off-by: Jeff Fan <jeff.fan@intel.com>
Reviewed-by: Ruiyu Ni <ruiyu.ni@intel.com>
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@14306 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
index 13745c5aace0e1ceb4d6278c39deb396f767cc48..c253fa59a37a3dad67764bb2422dda37ee1643c3 100644 (file)
--- a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
+++ b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
@@ -281,6 +281,7 @@ DebugAgentCallbackMemoryDiscoveredPpi (
   // Update Mailbox Location pointer in GUIDed HOB and IDT entry with new one\r
   //\r
   MailboxLocationInHob = GetMailboxLocationFromHob ();\r
+  ASSERT (MailboxLocationInHob != NULL);\r
   *MailboxLocationInHob = (UINT64)(UINTN)NewMailbox;\r
   SetLocationSavedMailboxPointerInIdtEntry (MailboxLocationInHob);\r
   //\r