MdeModulePkg/SmiHandlerProfile: Add Context support in Unregister

The reason is that we observe that a platform may use same Handler
for different context.

In order to support Unregister such handler, we have to input
context information as well.

Cc: Jeff Fan <jeff.fan@intel.com>
Cc: Feng Tian <feng.tian@intel.com>
Cc: Star Zeng <star.zeng@intel.com>
Cc: Bret Barkelew <Bret.Barkelew@microsoft.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jiewen Yao <jiewen.yao@intel.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>

1 files changed, 8 insertions, 2 deletions

diff --git a/MdeModulePkg/Library/SmmSmiHandlerProfileLib/SmmSmiHandlerProfileLib.c b/MdeModulePkg/Library/SmmSmiHandlerProfileLib/SmmSmiHandlerProfileLib.c
index 2edc71be3e..2911619971 100644
--- a/MdeModulePkg/Library/SmmSmiHandlerProfileLib/SmmSmiHandlerProfileLib.c
+++ b/MdeModulePkg/Library/SmmSmiHandlerProfileLib/SmmSmiHandlerProfileLib.c
@@ -63,6 +63,10 @@ SmiHandlerProfileRegisterHandler (
                          For the SmmChildDispatch protocol, the HandlerGuid

                          must be the GUID of SmmChildDispatch protocol.

   @param Handler         The SMI handler.

+  @param Context         The context of the SMI handler.

+                         If it is NOT NULL, it will be used to check what is registered.

+  @param ContextSize     The size of the context in bytes.

+                         If Context is NOT NULL, it will be used to check what is registered.

 

   @retval EFI_SUCCESS           The original record is removed.

   @retval EFI_UNSUPPORTED       The feature is unsupported.

@@ -72,11 +76,13 @@ EFI_STATUS
 EFIAPI

 SmiHandlerProfileUnregisterHandler (

   IN EFI_GUID                       *HandlerGuid,

-  IN EFI_SMM_HANDLER_ENTRY_POINT2   Handler

+  IN EFI_SMM_HANDLER_ENTRY_POINT2   Handler,

+  IN VOID                           *Context, OPTIONAL

+  IN UINTN                          ContextSize OPTIONAL

   )

 {

   if (mSmiHandlerProfile != NULL) {

-    return mSmiHandlerProfile->UnregisterHandler (mSmiHandlerProfile, HandlerGuid, Handler);

+    return mSmiHandlerProfile->UnregisterHandler (mSmiHandlerProfile, HandlerGuid, Handler, Context, ContextSize);

   }

   return EFI_UNSUPPORTED;

 }