Refine code make callback logic same with frontPage.

[mirror_edk2.git] / IntelFrameworkModulePkg / Universal / BdsDxe / DeviceMngr / DeviceManager.c

diff --git a/IntelFrameworkModulePkg/Universal/BdsDxe/DeviceMngr/DeviceManager.c b/IntelFrameworkModulePkg/Universal/BdsDxe/DeviceMngr/DeviceManager.c
index 1d4b0f525653dc74d6c9be8caf1753f8b2ba2daf..3fa24ad6d527eaaa85a9a1cdea993d2731a9479d 100644 (file)
--- a/IntelFrameworkModulePkg/Universal/BdsDxe/DeviceMngr/DeviceManager.c
+++ b/IntelFrameworkModulePkg/Universal/BdsDxe/DeviceMngr/DeviceManager.c
@@ -505,6 +505,10 @@ CallDeviceManager (
   //\r
   if (gCallbackKey == DEVICE_MANAGER_KEY_DRIVER_HEALTH) {\r
     CallDriverHealth ();\r
+    //\r
+    // Force return to Device Manager\r
+    //\r
+    gCallbackKey = FRONT_PAGE_KEY_DEVICE_MANAGER;\r
   }\r
 \r
   //\r
@@ -849,7 +853,7 @@ CallDriverHealth (
   // a target to display.\r
   // Process the diver health status states here.\r
   // \r
-  if (gCallbackKey >= DRIVER_HEALTH_KEY_OFFSET && gCallbackKey != DRIVER_HEALTH_REPAIR_ALL_KEY && gCallbackKey != FRONT_PAGE_KEY_DEVICE_MANAGER) {\r
+  if (gCallbackKey >= DRIVER_HEALTH_KEY_OFFSET && gCallbackKey != DRIVER_HEALTH_REPAIR_ALL_KEY) {\r
     ActionRequest = EFI_BROWSER_ACTION_REQUEST_NONE;\r
 \r
     Link = GetFirstNode (&DriverHealthList);\r
@@ -938,12 +942,12 @@ CallDriverHealth (
   HiiFreeOpCodeHandle (EndOpCodeHandleRepair); \r
 \r
   if (gCallbackKey == DRIVER_HEALTH_RETURN_KEY) {\r
+    //\r
+    // Force return to Driver Health Form\r
+    //\r
     gCallbackKey = DEVICE_MANAGER_KEY_DRIVER_HEALTH;\r
     CallDriverHealth ();\r
   }\r
-  if (gCallbackKey == DEVICE_MANAGER_KEY_DRIVER_HEALTH) {\r
-    CallDeviceManager();\r
-  }\r
 }\r
 \r
 \r