Refine code to make it more safely.

[mirror_edk2.git] / SourceLevelDebugPkg / DebugAgentDxe / DebugAgentDxe.c

/** @file\r
  Initialize Debug Agent in DXE by invoking Debug Agent Library.\r
\r
Copyright (c) 2013 - 2014, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials                          \r
are licensed and made available under the terms and conditions of the BSD License         \r
which accompanies this distribution.  The full text of the license may be found at        \r
http://opensource.org/licenses/bsd-license.php                                            \r
                                                                                          \r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             \r
\r
**/\r
\r
#include <PiDxe.h>\r
#include <Guid/EventGroup.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/DebugAgentLib.h>\r
\r
EFI_EVENT       mExitBootServiceEvent; \r
\r
/**\r
  One notified function to disable Debug Timer interrupt when gBS->ExitBootServices() called.\r
\r
  @param[in]  Event              Pointer to this event\r
  @param[in]  Context            Event hanlder private data\r
\r
**/\r
VOID\r
EFIAPI\r
DisableDebugTimerExitBootService (\r
  EFI_EVENT                      Event,\r
  VOID                           *Context\r
  )\r
\r
{\r
  SaveAndSetDebugTimerInterrupt (FALSE);\r
}\r
\r
/**\r
  The Entry Point for Debug Agent Dxe driver.\r
\r
  It will invoke Debug Agent Library to enable source debugging feature in DXE phase.\r
\r
  @param[in] ImageHandle    The firmware allocated handle for the EFI image.\r
  @param[in] SystemTable    A pointer to the EFI System Table.\r
\r
  @retval EFI_SUCCESS       The entry point is executed successfully.\r
  @retval other             Some error occurs when initialzed Debug Agent.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DebugAgentDxeInitialize(\r
  IN EFI_HANDLE           ImageHandle,\r
  IN EFI_SYSTEM_TABLE     *SystemTable\r
  )\r
{\r
  EFI_STATUS      Status;\r
\r
  Status = EFI_UNSUPPORTED;\r
  InitializeDebugAgent (DEBUG_AGENT_INIT_DXE_LOAD, &Status, NULL);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  //\r
  // Create event to disable Debug Timer interrupt when exit boot service.\r
  //\r
  Status = gBS->CreateEventEx (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_NOTIFY,\r
                  DisableDebugTimerExitBootService,\r
                  NULL,\r
                  &gEfiEventExitBootServicesGuid,\r
                  &mExitBootServiceEvent\r
                  );\r
  return Status;\r
}\r
\r
/**\r
  This is the unload handle for Debug Agent Dxe driver.\r
\r
  It will invoke Debug Agent Library to disable source debugging feature.\r
\r
  @param[in]  ImageHandle       The drivers' driver image.\r
\r
  @retval EFI_SUCCESS           The image is unloaded.\r
  @retval Others                Failed to unload the image.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
DebugAgentDxeUnload (\r
  IN EFI_HANDLE           ImageHandle\r
  )\r
{\r
  EFI_STATUS          Status;\r
\r
  Status = EFI_UNSUPPORTED;\r
  InitializeDebugAgent (DEBUG_AGENT_INIT_DXE_UNLOAD, &Status, NULL);\r
\r
  return Status;\r
}\r