From bd2057a846ee3b65e76b4398129cc02b93c46fbd Mon Sep 17 00:00:00 2001
From: gikidy <gikidy@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 7 Dec 2009 05:55:07 +0000
Subject: Refine code make callback logic same with frontPage.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@9531 6f19259b-4bc3-4df7-8a09-765794883524
---
 .../Universal/BdsDxe/DeviceMngr/DeviceManager.c              | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'IntelFrameworkModulePkg/Universal/BdsDxe/DeviceMngr/DeviceManager.c')

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