SourceLevelDebugPkg/SecPeiDebugAgentLib: Restore CPU interrupt state

[mirror_edk2.git] / MdeModulePkg / Universal / WatchdogTimerDxe / WatchdogTimer.c

/** @file\r
  Implementation of Watchdog Timer Architectural Protocol using UEFI APIs.\r
  \r
Copyright (c) 2006 - 2010, 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 "WatchdogTimer.h"\r
\r
//\r
// Handle for the Watchdog Timer Architectural Protocol instance produced by this driver\r
//\r
EFI_HANDLE  mWatchdogTimerHandle = NULL;\r
\r
//\r
// The Watchdog Timer Architectural Protocol instance produced by this driver\r
//\r
EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  mWatchdogTimer = {\r
  WatchdogTimerDriverRegisterHandler,\r
  WatchdogTimerDriverSetTimerPeriod,\r
  WatchdogTimerDriverGetTimerPeriod\r
};\r
\r
//\r
// The watchdog timer period in 100 ns units\r
//\r
UINT64  mWatchdogTimerPeriod = 0;\r
\r
//\r
// The notification function to call if the watchdig timer fires\r
//\r
EFI_WATCHDOG_TIMER_NOTIFY  mWatchdogTimerNotifyFunction = NULL;\r
\r
//\r
// The one-shot timer event that is armed when the watchdog timer is enabled\r
//\r
EFI_EVENT  mWatchdogTimerEvent;\r
\r
\r
/**\r
  Notification function that is called if the watchdog timer is fired.\r
\r
  Notification function for the one-shot timer event that was signaled\r
  when the watchdog timer expired. If a handler has been registered with\r
  the Watchdog Timer Architectural Protocol, then that handler is called\r
  passing in the time period that has passed that cause the watchdog timer\r
  to fire.  Then, a call to the Runtime Service ResetSystem() is made to\r
  reset the platform.\r
\r
  @param  Timer     The one-shot timer event that was signaled when the\r
                    watchdog timer expired.\r
  @param  Context   The context that was registered when the event Timer was created.\r
\r
**/\r
VOID\r
EFIAPI\r
WatchdogTimerDriverExpires (\r
  IN EFI_EVENT    Timer,\r
  IN VOID         *Context\r
  )\r
{\r
  REPORT_STATUS_CODE (EFI_ERROR_CODE | EFI_ERROR_MINOR, (EFI_COMPUTING_UNIT_HOST_PROCESSOR | EFI_CU_HP_EC_TIMER_EXPIRED));\r
\r
  //\r
  // If a notification function has been registered, then call it\r
  //\r
  if (mWatchdogTimerNotifyFunction != NULL) {\r
    mWatchdogTimerNotifyFunction (mWatchdogTimerPeriod);\r
  }\r
  \r
  DEBUG ((EFI_D_ERROR, "Watchdog Timer reseting system\n"));\r
\r
  //\r
  // Reset the platform\r
  //\r
  gRT->ResetSystem (EfiResetCold, EFI_TIMEOUT, 0, NULL);\r
}\r
\r
\r
/**\r
  Registers a handler that is to be invoked when the watchdog timer fires.\r
\r
  This function registers a handler that is to be invoked when the watchdog\r
  timer fires.  By default, the EFI_WATCHDOG_TIMER protocol will call the\r
  Runtime Service ResetSystem() when the watchdog timer fires.  If a\r
  NotifyFunction is registered, then the NotifyFunction will be called before\r
  the Runtime Service ResetSystem() is called.  If NotifyFunction is NULL, then\r
  the watchdog handler is unregistered.  If a watchdog handler is registered,\r
  then EFI_SUCCESS is returned.  If an attempt is made to register a handler\r
  when a handler is already registered, then EFI_ALREADY_STARTED is returned.\r
  If an attempt is made to uninstall a handler when a handler is not installed,\r
  then return EFI_INVALID_PARAMETER.\r
\r
  @param  This                  The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.\r
  @param  NotifyFunction        The function to call when the watchdog timer fires.  If this\r
                                is NULL, then the handler will be unregistered.\r
\r
  @retval EFI_SUCCESS           The watchdog timer handler was registered or unregistered.\r
  @retval EFI_ALREADY_STARTED   NotifyFunction is not NULL, and a handler is already registered.\r
  @retval EFI_INVALID_PARAMETER NotifyFunction is NULL, and a handler was not previously registered.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WatchdogTimerDriverRegisterHandler (\r
  IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,\r
  IN EFI_WATCHDOG_TIMER_NOTIFY         NotifyFunction\r
  )\r
{\r
  //\r
  // If NotifyFunction is NULL, and a handler was not previously registered,\r
  // return EFI_INVALID_PARAMETER.\r
  //\r
  if (NotifyFunction == NULL && mWatchdogTimerNotifyFunction == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  //\r
  // If NotifyFunction is not NULL, and a handler is already registered,\r
  // return EFI_ALREADY_STARTED.\r
  //\r
  if (NotifyFunction != NULL && mWatchdogTimerNotifyFunction != NULL) {\r
    return EFI_ALREADY_STARTED;\r
  }\r
\r
  mWatchdogTimerNotifyFunction = NotifyFunction;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Sets the amount of time in the future to fire the watchdog timer.\r
\r
  This function sets the amount of time to wait before firing the watchdog\r
  timer to TimerPeriod 100 ns units.  If TimerPeriod is 0, then the watchdog\r
  timer is disabled.\r
\r
  @param  This              The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.\r
  @param  TimerPeriod       The amount of time in 100 ns units to wait before the watchdog\r
                            timer is fired.  If TimerPeriod is zero, then the watchdog\r
                            timer is disabled.\r
\r
  @retval EFI_SUCCESS       The watchdog timer has been programmed to fire in Time\r
                            100 ns units.\r
  @retval EFI_DEVICE_ERROR  A watchdog timer could not be programmed due to a device\r
                            error.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WatchdogTimerDriverSetTimerPeriod (\r
  IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,\r
  IN UINT64                            TimerPeriod\r
  )\r
{\r
  mWatchdogTimerPeriod = TimerPeriod;\r
\r
  return gBS->SetTimer (\r
                mWatchdogTimerEvent,\r
                (mWatchdogTimerPeriod == 0) ? TimerCancel : TimerRelative,\r
                mWatchdogTimerPeriod\r
                );\r
}\r
\r
/**\r
  Retrieves the amount of time in 100 ns units that the system will wait before firing the watchdog timer.\r
\r
  This function retrieves the amount of time the system will wait before firing\r
  the watchdog timer.  This period is returned in TimerPeriod, and EFI_SUCCESS\r
  is returned.  If TimerPeriod is NULL, then EFI_INVALID_PARAMETER is returned.\r
\r
  @param  This                    The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.\r
  @param  TimerPeriod             A pointer to the amount of time in 100 ns units that the system\r
                                  will wait before the watchdog timer is fired.  If TimerPeriod of\r
                                  zero is returned, then the watchdog timer is disabled.\r
\r
  @retval EFI_SUCCESS             The amount of time that the system will wait before\r
                                  firing the watchdog timer was returned in TimerPeriod.\r
  @retval EFI_INVALID_PARAMETER   TimerPeriod is NULL.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WatchdogTimerDriverGetTimerPeriod (\r
  IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,\r
  IN UINT64                            *TimerPeriod\r
  )\r
{\r
  if (TimerPeriod == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  *TimerPeriod = mWatchdogTimerPeriod;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Entry point of the Watchdog Timer Architectural Protocol driver.\r
\r
  @param  ImageHandle   The image handle of this driver.\r
  @param  SystemTable   The pointer of EFI_SYSTEM_TABLE.\r
\r
  @retval EFI_SUCCESS   Watchdog Timer Architectural Protocol successfully installed.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
WatchdogTimerDriverInitialize (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  //\r
  // Make sure the Watchdog Timer Architectural Protocol has not been installed in the system yet.\r
  //\r
  ASSERT_PROTOCOL_ALREADY_INSTALLED (NULL, &gEfiWatchdogTimerArchProtocolGuid);\r
\r
  //\r
  // Create the timer event to implement a simple watchdog timer\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_TIMER | EVT_NOTIFY_SIGNAL,\r
                  TPL_NOTIFY,\r
                  WatchdogTimerDriverExpires,\r
                  NULL,\r
                  &mWatchdogTimerEvent\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  //\r
  // Install the Watchdog Timer Arch Protocol onto a new handle\r
  //\r
  Status = gBS->InstallMultipleProtocolInterfaces (\r
                  &mWatchdogTimerHandle,\r
                  &gEfiWatchdogTimerArchProtocolGuid,\r
                  &mWatchdogTimer,\r
                  NULL\r
                  );\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return EFI_SUCCESS;\r
}\r