Check Context before use it.

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

diff --git a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
index a3f91d3358085b667037cae5e5579c08026e7c90..7a7ca423c7a4caaa3c113c850c450e0788af45b0 100644 (file)
--- a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
+++ b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.c
@@ -613,7 +613,7 @@ InitializeDebugAgent (
   //\r
   // Set return status for DEBUG_AGENT_INIT_PEI\r
   //\r
-  if (InitFlag == DEBUG_AGENT_INIT_PEI) {\r
+  if (InitFlag == DEBUG_AGENT_INIT_PEI && Context != NULL) {\r
     *(EFI_STATUS *)Context = EFI_SUCCESS;\r
   }\r
 }\r