1 files changed, 38 insertions, 2 deletions
diff --git a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.h b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.h
index 9ca5e87f..68bfbf7c 100644
--- a/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.h
+++ b/SourceLevelDebugPkg/Library/DebugAgent/SecPeiDebugAgent/SecPeiDebugAgentLib.h
@@ -1,7 +1,7 @@
 /** @file
   Header file for Sec Core Debug Agent Library instance.
 
-  Copyright (c) 2010, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2010 - 2013, Intel Corporation. All rights reserved.<BR>
   This program and the accompanying materials
   are licensed and made available under the terms and conditions of the BSD License
   which accompanies this distribution.  The full text of the license may be found at
@@ -16,7 +16,9 @@
 #define _SEC_CORE_DEBUG_AGENT_LIB_H_
 
 #include <PiPei.h>
-
+#include <Ppi/MemoryDiscovered.h>
+#include <Library/PeiServicesLib.h>
+#include <Library/MemoryAllocationLib.h>
 #include "DebugAgent.h"
 
 typedef struct {
@@ -24,5 +26,39 @@ typedef struct {
   DEBUG_AGENT_CONTINUE    Function;
 } DEBUG_AGENT_PHASE2_CONTEXT;
 
+/**
+  Caller provided function to be invoked at the end of DebugPortInitialize().
+
+  Refer to the descrption for DebugPortInitialize() for more details.
+
+  @param[in] Context           The first input argument of DebugPortInitialize().
+  @param[in] DebugPortHandle   Debug port handle created by Debug Communication Libary.
+
+**/
+VOID
+EFIAPI
+InitializeDebugAgentPhase2 (
+  IN VOID                  *Context,
+  IN DEBUG_PORT_HANDLE     DebugPortHandle
+  );
+
+/**
+  Debug Agent provided notify callback function on Memory Discovered PPI.
+
+  @param[in] PeiServices      Indirect reference to the PEI Services Table.
+  @param[in] NotifyDescriptor Address of the notification descriptor data structure.
+  @param[in] Ppi              Address of the PPI that was installed.
+
+  @retval EFI_SUCCESS If the function completed successfully.
+
+**/
+EFI_STATUS
+EFIAPI
+DebugAgentCallbackMemoryDiscoveredPpi (
+  IN EFI_PEI_SERVICES                     **PeiServices,
+  IN EFI_PEI_NOTIFY_DESCRIPTOR            *NotifyDescriptor,
+  IN VOID                                 *Ppi
+  );
+
 #endif