]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/ResetSystemRuntimeDxe/ResetSystem.c
MdeModulePkg/ResetSystemRuntimeDxe: Add platform filter and handler
[mirror_edk2.git] / MdeModulePkg / Universal / ResetSystemRuntimeDxe / ResetSystem.c
index 64f2da5ce9def5ef4bc59141998490e03e2d54f2..fed527fac2dbca7f9dfd1e50fbe82b90c504cbcc 100644 (file)
@@ -1,5 +1,5 @@
 /** @file\r
-  Reset Architectural Protocol implementation\r
+  Reset Architectural and Reset Notification protocols implementation.\r
 \r
   Copyright (c) 2006 - 2017, Intel Corporation. All rights reserved.<BR>\r
 \r
 \r
 #include "ResetSystem.h"\r
 \r
+//\r
+// The current ResetSystem() notification recursion depth\r
+//\r
+UINTN  mResetNotifyDepth = 0;\r
+\r
+/**\r
+  Register a notification function to be called when ResetSystem() is called.\r
+\r
+  The RegisterResetNotify() function registers a notification function that is called when\r
+  ResetSystem()is called and prior to completing the reset of the platform.\r
+  The registered functions must not perform a platform reset themselves. These\r
+  notifications are intended only for the notification of components which may need some\r
+  special-purpose maintenance prior to the platform resetting.\r
+  The list of registered reset notification functions are processed if ResetSystem()is called\r
+  before ExitBootServices(). The list of registered reset notification functions is ignored if\r
+  ResetSystem()is called after ExitBootServices().\r
+\r
+  @param[in]  This              A pointer to the EFI_RESET_NOTIFICATION_PROTOCOL instance.\r
+  @param[in]  ResetFunction     Points to the function to be called when a ResetSystem() is executed.\r
+\r
+  @retval EFI_SUCCESS           The reset notification function was successfully registered.\r
+  @retval EFI_INVALID_PARAMETER ResetFunction is NULL.\r
+  @retval EFI_OUT_OF_RESOURCES  There are not enough resources available to register the reset notification function.\r
+  @retval EFI_ALREADY_STARTED   The reset notification function specified by ResetFunction has already been registered.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+RegisterResetNotify (\r
+  IN EFI_RESET_NOTIFICATION_PROTOCOL *This,\r
+  IN EFI_RESET_SYSTEM                ResetFunction\r
+  )\r
+{\r
+  RESET_NOTIFICATION_INSTANCE        *Instance;\r
+  LIST_ENTRY                         *Link;\r
+  RESET_NOTIFY_ENTRY                 *Entry;\r
+\r
+  if (ResetFunction == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  Instance = RESET_NOTIFICATION_INSTANCE_FROM_THIS (This);\r
+\r
+  for ( Link = GetFirstNode (&Instance->ResetNotifies)\r
+      ; !IsNull (&Instance->ResetNotifies, Link)\r
+      ; Link = GetNextNode (&Instance->ResetNotifies, Link)\r
+      ) {\r
+    Entry = RESET_NOTIFY_ENTRY_FROM_LINK (Link);\r
+    if (Entry->ResetNotify == ResetFunction) {\r
+      return EFI_ALREADY_STARTED;\r
+    }\r
+  }\r
+\r
+  ASSERT (IsNull (&Instance->ResetNotifies, Link));\r
+  Entry = AllocatePool (sizeof (*Entry));\r
+  if (Entry == NULL) {\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+  Entry->Signature   = RESET_NOTIFY_ENTRY_SIGNATURE;\r
+  Entry->ResetNotify = ResetFunction;\r
+  InsertTailList (&Instance->ResetNotifies, &Entry->Link);\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Unregister a notification function.\r
+\r
+  The UnregisterResetNotify() function removes the previously registered\r
+  notification using RegisterResetNotify().\r
+\r
+  @param[in]  This              A pointer to the EFI_RESET_NOTIFICATION_PROTOCOL instance.\r
+  @param[in]  ResetFunction     The pointer to the ResetFunction being unregistered.\r
+\r
+  @retval EFI_SUCCESS           The reset notification function was unregistered.\r
+  @retval EFI_INVALID_PARAMETER ResetFunction is NULL.\r
+  @retval EFI_INVALID_PARAMETER The reset notification function specified by ResetFunction was not previously\r
+                                registered using RegisterResetNotify().\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+UnregisterResetNotify (\r
+  IN EFI_RESET_NOTIFICATION_PROTOCOL *This,\r
+  IN EFI_RESET_SYSTEM                ResetFunction\r
+  )\r
+{\r
+  RESET_NOTIFICATION_INSTANCE        *Instance;\r
+  LIST_ENTRY                         *Link;\r
+  RESET_NOTIFY_ENTRY                 *Entry;\r
+\r
+  if (ResetFunction == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  Instance = RESET_NOTIFICATION_INSTANCE_FROM_THIS (This);\r
+\r
+  for ( Link = GetFirstNode (&Instance->ResetNotifies)\r
+      ; !IsNull (&Instance->ResetNotifies, Link)\r
+      ; Link = GetNextNode (&Instance->ResetNotifies, Link)\r
+      ) {\r
+    Entry = RESET_NOTIFY_ENTRY_FROM_LINK (Link);\r
+    if (Entry->ResetNotify == ResetFunction) {\r
+      RemoveEntryList (&Entry->Link);\r
+      FreePool (Entry);\r
+      return EFI_SUCCESS;\r
+    }\r
+  }\r
+\r
+  return EFI_INVALID_PARAMETER;\r
+}\r
+\r
+RESET_NOTIFICATION_INSTANCE mResetNotification = {\r
+  RESET_NOTIFICATION_INSTANCE_SIGNATURE,\r
+  {\r
+    RegisterResetNotify,\r
+    UnregisterResetNotify\r
+  },\r
+  INITIALIZE_LIST_HEAD_VARIABLE (mResetNotification.ResetNotifies)\r
+};\r
+\r
+RESET_NOTIFICATION_INSTANCE mPlatformSpecificResetFilter = {\r
+  RESET_NOTIFICATION_INSTANCE_SIGNATURE,\r
+  {\r
+    RegisterResetNotify,\r
+    UnregisterResetNotify\r
+  },\r
+  INITIALIZE_LIST_HEAD_VARIABLE (mPlatformSpecificResetFilter.ResetNotifies)\r
+};\r
+\r
+RESET_NOTIFICATION_INSTANCE mPlatformSpecificResetHandler = {\r
+  RESET_NOTIFICATION_INSTANCE_SIGNATURE,\r
+  {\r
+    RegisterResetNotify,\r
+    UnregisterResetNotify\r
+  },\r
+  INITIALIZE_LIST_HEAD_VARIABLE (mPlatformSpecificResetHandler.ResetNotifies)\r
+};\r
+\r
 /**\r
   The driver's entry point.\r
 \r
@@ -53,8 +191,10 @@ InitializeResetSystem (
   Handle = NULL;\r
   Status = gBS->InstallMultipleProtocolInterfaces (\r
                   &Handle,\r
-                  &gEfiResetArchProtocolGuid,\r
-                  NULL,\r
+                  &gEfiResetArchProtocolGuid,         NULL,\r
+                  &gEfiResetNotificationProtocolGuid, &mResetNotification.ResetNotification,\r
+                  &gEdkiiPlatformSpecificResetFilterProtocolGuid, &mPlatformSpecificResetFilter.ResetNotification,\r
+                  &gEdkiiPlatformSpecificResetHandlerProtocolGuid, &mPlatformSpecificResetHandler.ResetNotification,\r
                   NULL\r
                   );\r
   ASSERT_EFI_ERROR (Status);\r
@@ -92,6 +232,9 @@ DoS3 (
                                 valid if ResetStatus is something other than EFI_SUCCESS\r
                                 unless the ResetType is EfiResetPlatformSpecific\r
                                 where a minimum amount of ResetData is always required.\r
+                                For a ResetType of EfiResetPlatformSpecific the data buffer\r
+                                also starts with a Null-terminated string that is followed\r
+                                by an EFI_GUID that describes the specific type of reset to perform.\r
 **/\r
 VOID\r
 EFIAPI\r
@@ -102,14 +245,68 @@ ResetSystem (
   IN VOID             *ResetData OPTIONAL\r
   )\r
 {\r
-  EFI_STATUS    Status;\r
-  UINTN         Size;\r
-  UINTN         CapsuleDataPtr;\r
-  \r
+  EFI_STATUS          Status;\r
+  UINTN               Size;\r
+  UINTN               CapsuleDataPtr;\r
+  LIST_ENTRY          *Link;\r
+  RESET_NOTIFY_ENTRY  *Entry;\r
+\r
+  //\r
+  // Above the maximum recursion depth, so do the smallest amount of\r
+  // work to perform a cold reset.\r
+  //\r
+  if (mResetNotifyDepth >= MAX_RESET_NOTIFY_DEPTH) {\r
+    ResetCold ();\r
+    ASSERT (FALSE);\r
+    return;\r
+  }\r
+\r
   //\r
-  // Indicate reset system runtime service is called.\r
+  // Only do REPORT_STATUS_CODE() on first call to ResetSystem()\r
   //\r
-  REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_EFI_RUNTIME_SERVICE | EFI_SW_RS_PC_RESET_SYSTEM));\r
+  if (mResetNotifyDepth == 0) {\r
+    //\r
+    // Indicate reset system runtime service is called.\r
+    //\r
+    REPORT_STATUS_CODE (EFI_PROGRESS_CODE, (EFI_SOFTWARE_EFI_RUNTIME_SERVICE | EFI_SW_RS_PC_RESET_SYSTEM));\r
+  }\r
+\r
+  mResetNotifyDepth++;\r
+  if (!EfiAtRuntime () && mResetNotifyDepth < MAX_RESET_NOTIFY_DEPTH) {\r
+    //\r
+    // Call reset notification functions registered through the\r
+    // EDKII_PLATFORM_SPECIFIC_RESET_FILTER_PROTOCOL.\r
+    //\r
+    for ( Link = GetFirstNode (&mPlatformSpecificResetFilter.ResetNotifies)\r
+        ; !IsNull (&mPlatformSpecificResetFilter.ResetNotifies, Link)\r
+        ; Link = GetNextNode (&mPlatformSpecificResetFilter.ResetNotifies, Link)\r
+        ) {\r
+      Entry = RESET_NOTIFY_ENTRY_FROM_LINK (Link);\r
+      Entry->ResetNotify (ResetType, ResetStatus, DataSize, ResetData);\r
+    }\r
+    //\r
+    // Call reset notification functions registered through the\r
+    // EFI_RESET_NOTIFICATION_PROTOCOL.\r
+    //\r
+    for ( Link = GetFirstNode (&mResetNotification.ResetNotifies)\r
+        ; !IsNull (&mResetNotification.ResetNotifies, Link)\r
+        ; Link = GetNextNode (&mResetNotification.ResetNotifies, Link)\r
+        ) {\r
+      Entry = RESET_NOTIFY_ENTRY_FROM_LINK (Link);\r
+      Entry->ResetNotify (ResetType, ResetStatus, DataSize, ResetData);\r
+    }\r
+    //\r
+    // call reset notification functions registered through the \r
+    // EDKII_PLATFORM_SPECIFIC_RESET_HANDLER_PROTOCOL.\r
+    //\r
+    for ( Link = GetFirstNode (&mPlatformSpecificResetHandler.ResetNotifies)\r
+        ; !IsNull (&mPlatformSpecificResetHandler.ResetNotifies, Link)\r
+        ; Link = GetNextNode (&mPlatformSpecificResetHandler.ResetNotifies, Link)\r
+        ) {\r
+      Entry = RESET_NOTIFY_ENTRY_FROM_LINK (Link);\r
+      Entry->ResetNotify (ResetType, ResetStatus, DataSize, ResetData);\r
+    }\r
+  }\r
 \r
   switch (ResetType) {\r
   case EfiResetWarm:\r