1 /** @file
2   Implementation of Watchdog Timer Architectural Protocol using UEFI APIs.
3 
4 Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution.  The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9 
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12 
13 **/
14 
15 #include "WatchdogTimer.h"
16 
17 //
18 // Handle for the Watchdog Timer Architectural Protocol instance produced by this driver
19 //
20 EFI_HANDLE  mWatchdogTimerHandle = NULL;
21 
22 //
23 // The Watchdog Timer Architectural Protocol instance produced by this driver
24 //
25 EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  mWatchdogTimer = {
26   WatchdogTimerDriverRegisterHandler,
27   WatchdogTimerDriverSetTimerPeriod,
28   WatchdogTimerDriverGetTimerPeriod
29 };
30 
31 //
32 // The watchdog timer period in 100 ns units
33 //
34 UINT64  mWatchdogTimerPeriod = 0;
35 
36 //
37 // The notification function to call if the watchdig timer fires
38 //
39 EFI_WATCHDOG_TIMER_NOTIFY  mWatchdogTimerNotifyFunction = NULL;
40 
41 //
42 // The one-shot timer event that is armed when the watchdog timer is enabled
43 //
44 EFI_EVENT  mWatchdogTimerEvent;
45 
46 
47 /**
48   Notification function that is called if the watchdog timer is fired.
49 
50   Notification function for the one-shot timer event that was signaled
51   when the watchdog timer expired. If a handler has been registered with
52   the Watchdog Timer Architectural Protocol, then that handler is called
53   passing in the time period that has passed that cause the watchdog timer
54   to fire.  Then, a call to the Runtime Service ResetSystem() is made to
55   reset the platform.
56 
57   @param  Timer     The one-shot timer event that was signaled when the
58                     watchdog timer expired.
59   @param  Context   The context that was registered when the event Timer was created.
60 
61 **/
62 VOID
63 EFIAPI
WatchdogTimerDriverExpires(IN EFI_EVENT Timer,IN VOID * Context)64 WatchdogTimerDriverExpires (
65   IN EFI_EVENT    Timer,
66   IN VOID         *Context
67   )
68 {
69   REPORT_STATUS_CODE (EFI_ERROR_CODE | EFI_ERROR_MINOR, (EFI_COMPUTING_UNIT_HOST_PROCESSOR | EFI_CU_HP_EC_TIMER_EXPIRED));
70 
71   //
72   // If a notification function has been registered, then call it
73   //
74   if (mWatchdogTimerNotifyFunction != NULL) {
75     mWatchdogTimerNotifyFunction (mWatchdogTimerPeriod);
76   }
77 
78   DEBUG ((EFI_D_ERROR, "Watchdog Timer reseting system\n"));
79 
80   //
81   // Reset the platform
82   //
83   gRT->ResetSystem (EfiResetCold, EFI_TIMEOUT, 0, NULL);
84 }
85 
86 
87 /**
88   Registers a handler that is to be invoked when the watchdog timer fires.
89 
90   This function registers a handler that is to be invoked when the watchdog
91   timer fires.  By default, the EFI_WATCHDOG_TIMER protocol will call the
92   Runtime Service ResetSystem() when the watchdog timer fires.  If a
93   NotifyFunction is registered, then the NotifyFunction will be called before
94   the Runtime Service ResetSystem() is called.  If NotifyFunction is NULL, then
95   the watchdog handler is unregistered.  If a watchdog handler is registered,
96   then EFI_SUCCESS is returned.  If an attempt is made to register a handler
97   when a handler is already registered, then EFI_ALREADY_STARTED is returned.
98   If an attempt is made to uninstall a handler when a handler is not installed,
99   then return EFI_INVALID_PARAMETER.
100 
101   @param  This                  The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.
102   @param  NotifyFunction        The function to call when the watchdog timer fires.  If this
103                                 is NULL, then the handler will be unregistered.
104 
105   @retval EFI_SUCCESS           The watchdog timer handler was registered or unregistered.
106   @retval EFI_ALREADY_STARTED   NotifyFunction is not NULL, and a handler is already registered.
107   @retval EFI_INVALID_PARAMETER NotifyFunction is NULL, and a handler was not previously registered.
108 
109 **/
110 EFI_STATUS
111 EFIAPI
WatchdogTimerDriverRegisterHandler(IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL * This,IN EFI_WATCHDOG_TIMER_NOTIFY NotifyFunction)112 WatchdogTimerDriverRegisterHandler (
113   IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,
114   IN EFI_WATCHDOG_TIMER_NOTIFY         NotifyFunction
115   )
116 {
117   //
118   // If NotifyFunction is NULL, and a handler was not previously registered,
119   // return EFI_INVALID_PARAMETER.
120   //
121   if (NotifyFunction == NULL && mWatchdogTimerNotifyFunction == NULL) {
122     return EFI_INVALID_PARAMETER;
123   }
124   //
125   // If NotifyFunction is not NULL, and a handler is already registered,
126   // return EFI_ALREADY_STARTED.
127   //
128   if (NotifyFunction != NULL && mWatchdogTimerNotifyFunction != NULL) {
129     return EFI_ALREADY_STARTED;
130   }
131 
132   mWatchdogTimerNotifyFunction = NotifyFunction;
133 
134   return EFI_SUCCESS;
135 }
136 
137 /**
138   Sets the amount of time in the future to fire the watchdog timer.
139 
140   This function sets the amount of time to wait before firing the watchdog
141   timer to TimerPeriod 100 ns units.  If TimerPeriod is 0, then the watchdog
142   timer is disabled.
143 
144   @param  This              The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.
145   @param  TimerPeriod       The amount of time in 100 ns units to wait before the watchdog
146                             timer is fired.  If TimerPeriod is zero, then the watchdog
147                             timer is disabled.
148 
149   @retval EFI_SUCCESS       The watchdog timer has been programmed to fire in Time
150                             100 ns units.
151   @retval EFI_DEVICE_ERROR  A watchdog timer could not be programmed due to a device
152                             error.
153 
154 **/
155 EFI_STATUS
156 EFIAPI
WatchdogTimerDriverSetTimerPeriod(IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL * This,IN UINT64 TimerPeriod)157 WatchdogTimerDriverSetTimerPeriod (
158   IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,
159   IN UINT64                            TimerPeriod
160   )
161 {
162   mWatchdogTimerPeriod = TimerPeriod;
163 
164   return gBS->SetTimer (
165                 mWatchdogTimerEvent,
166                 (mWatchdogTimerPeriod == 0) ? TimerCancel : TimerRelative,
167                 mWatchdogTimerPeriod
168                 );
169 }
170 
171 /**
172   Retrieves the amount of time in 100 ns units that the system will wait before firing the watchdog timer.
173 
174   This function retrieves the amount of time the system will wait before firing
175   the watchdog timer.  This period is returned in TimerPeriod, and EFI_SUCCESS
176   is returned.  If TimerPeriod is NULL, then EFI_INVALID_PARAMETER is returned.
177 
178   @param  This                    The EFI_WATCHDOG_TIMER_ARCH_PROTOCOL instance.
179   @param  TimerPeriod             A pointer to the amount of time in 100 ns units that the system
180                                   will wait before the watchdog timer is fired.  If TimerPeriod of
181                                   zero is returned, then the watchdog timer is disabled.
182 
183   @retval EFI_SUCCESS             The amount of time that the system will wait before
184                                   firing the watchdog timer was returned in TimerPeriod.
185   @retval EFI_INVALID_PARAMETER   TimerPeriod is NULL.
186 
187 **/
188 EFI_STATUS
189 EFIAPI
WatchdogTimerDriverGetTimerPeriod(IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL * This,IN UINT64 * TimerPeriod)190 WatchdogTimerDriverGetTimerPeriod (
191   IN EFI_WATCHDOG_TIMER_ARCH_PROTOCOL  *This,
192   IN UINT64                            *TimerPeriod
193   )
194 {
195   if (TimerPeriod == NULL) {
196     return EFI_INVALID_PARAMETER;
197   }
198 
199   *TimerPeriod = mWatchdogTimerPeriod;
200 
201   return EFI_SUCCESS;
202 }
203 
204 /**
205   Entry point of the Watchdog Timer Architectural Protocol driver.
206 
207   @param  ImageHandle   The image handle of this driver.
208   @param  SystemTable   The pointer of EFI_SYSTEM_TABLE.
209 
210   @retval EFI_SUCCESS   Watchdog Timer Architectural Protocol successfully installed.
211 
212 **/
213 EFI_STATUS
214 EFIAPI
WatchdogTimerDriverInitialize(IN EFI_HANDLE ImageHandle,IN EFI_SYSTEM_TABLE * SystemTable)215 WatchdogTimerDriverInitialize (
216   IN EFI_HANDLE        ImageHandle,
217   IN EFI_SYSTEM_TABLE  *SystemTable
218   )
219 {
220   EFI_STATUS  Status;
221 
222   //
223   // Make sure the Watchdog Timer Architectural Protocol has not been installed in the system yet.
224   //
225   ASSERT_PROTOCOL_ALREADY_INSTALLED (NULL, &gEfiWatchdogTimerArchProtocolGuid);
226 
227   //
228   // Create the timer event to implement a simple watchdog timer
229   //
230   Status = gBS->CreateEvent (
231                   EVT_TIMER | EVT_NOTIFY_SIGNAL,
232                   TPL_NOTIFY,
233                   WatchdogTimerDriverExpires,
234                   NULL,
235                   &mWatchdogTimerEvent
236                   );
237   ASSERT_EFI_ERROR (Status);
238 
239   //
240   // Install the Watchdog Timer Arch Protocol onto a new handle
241   //
242   Status = gBS->InstallMultipleProtocolInterfaces (
243                   &mWatchdogTimerHandle,
244                   &gEfiWatchdogTimerArchProtocolGuid,
245                   &mWatchdogTimer,
246                   NULL
247                   );
248   ASSERT_EFI_ERROR (Status);
249 
250   return EFI_SUCCESS;
251 }
252