[Description]

solve the x64 exception issue when unloading iSCSI driver
[Impaction]
add some handler for failure branch and prohibit the additional ISCSI INITIATOR NAME PROTOCOL from being installed.
[Reference Info]
this bug is reported by DELL. it causes an exception in x64 on unloading iscsi driver.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@4941 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdeModulePkg/Universal/Network/IScsiDxe/IScsiConfig.c b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiConfig.c
index 92169cb9b284043cedf0d8d692537e2aa88ea9ce..494415211166af6abff29cdfd03ecd8342f07eb5 100644 (file)
--- a/MdeModulePkg/Universal/Network/IScsiDxe/IScsiConfig.c
+++ b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiConfig.c
@@ -1,6 +1,6 @@
 /*++\r
 \r
-Copyright (c) 2004 - 2007, Intel Corporation\r
+Copyright (c) 2004 - 2008, Intel Corporation\r
 All rights reserved. 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
@@ -19,10 +19,10 @@ Abstract:
 \r
 #include "IScsiImpl.h"\r
 \r
-EFI_GUID        mVendorGuid             = ISCSI_CONFIG_GUID;\r
-BOOLEAN         mIScsiDeviceListUpdated = FALSE;\r
-UINTN           mNumberOfIScsiDevices   = 0;\r
-ISCSI_FORM_CALLBACK_INFO  *mCallbackInfo;\r
+EFI_GUID        mVendorGuid              = ISCSI_CONFIG_GUID;\r
+BOOLEAN         mIScsiDeviceListUpdated  = FALSE;\r
+UINTN           mNumberOfIScsiDevices    = 0;\r
+ISCSI_FORM_CALLBACK_INFO  *mCallbackInfo = NULL;\r
 \r
 LIST_ENTRY      mIScsiConfigFormList = {\r
   &mIScsiConfigFormList,\r
@@ -923,9 +923,9 @@ Returns:
     return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
-  CallbackInfo->Signature             = ISCSI_FORM_CALLBACK_INFO_SIGNATURE;\r
+  CallbackInfo->Signature   = ISCSI_FORM_CALLBACK_INFO_SIGNATURE;\r
   CallbackInfo->HiiDatabase = HiiDatabase;\r
-  CallbackInfo->Current               = NULL;\r
+  CallbackInfo->Current     = NULL;\r
 \r
   CallbackInfo->ConfigAccess.ExtractConfig = IScsiFormExtractConfig;\r
   CallbackInfo->ConfigAccess.RouteConfig = IScsiFormRouteConfig;\r
@@ -933,6 +933,7 @@ Returns:
 \r
   Status = gBS->LocateProtocol (&gEfiHiiConfigRoutingProtocolGuid, NULL, (VOID **)&CallbackInfo->ConfigRouting);\r
   if (EFI_ERROR (Status)) {\r
+    FreePool(CallbackInfo);\r
     return Status;\r
   }\r
 \r
@@ -941,6 +942,7 @@ Returns:
   //\r
   Status = HiiLibCreateHiiDriverHandle (&CallbackInfo->DriverHandle);\r
   if (EFI_ERROR (Status)) {\r
+    FreePool(CallbackInfo);\r
     return Status;\r
   }\r
   \r
@@ -967,8 +969,9 @@ Returns:
                            CallbackInfo->DriverHandle,\r
                            &CallbackInfo->RegisteredHandle\r
                            );\r
-  gBS->FreePool (PackageList);\r
+  FreePool (PackageList);\r
   if (EFI_ERROR (Status)) {\r
+    FreePool(CallbackInfo);\r
     return Status;\r
   }\r
 \r

diff --git a/MdeModulePkg/Universal/Network/IScsiDxe/IScsiDriver.c b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiDriver.c
index 8cc2a5050f9c793f78848b1a9f536d7eabff3d5d..bc777e7785f42517b4eaab1e7f365e08889c562e 100644 (file)
--- a/MdeModulePkg/Universal/Network/IScsiDxe/IScsiDriver.c
+++ b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiDriver.c
@@ -1,6 +1,6 @@
 /*++\r
 \r
-Copyright (c) 2004 - 2007, Intel Corporation\r
+Copyright (c) 2004 - 2008, Intel Corporation\r
 All rights reserved. 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
@@ -413,8 +413,21 @@ Returns:
 \r
 --*/\r
 {\r
-  EFI_STATUS  Status;\r
-  //EFI_LOADED_IMAGE_PROTOCOL *LoadedImage;\r
+  EFI_STATUS                         Status;\r
+  EFI_ISCSI_INITIATOR_NAME_PROTOCOL  *IScsiInitiatorName;\r
+\r
+  //\r
+  // There should be only one EFI_ISCSI_INITIATOR_NAME_PROTOCOL.\r
+  //\r
+  Status = gBS->LocateProtocol (\r
+                   &gEfiIScsiInitiatorNameProtocolGuid,\r
+                   NULL,\r
+                   &IScsiInitiatorName\r
+                   );\r
+\r
+  if (!EFI_ERROR (Status)) {\r
+    return EFI_ACCESS_DENIED;\r
+  }\r
 \r
   //\r
   // Initialize the EFI Driver Library\r
@@ -427,12 +440,11 @@ Returns:
              &gIScsiComponentName,\r
              &gIScsiComponentName2\r
            );\r
-  \r
-  if (EFI_ERROR (Status)) {\r
-    return Status;\r
-  }\r
 \r
   if (!EFI_ERROR (Status)) {\r
+    //\r
+    // Install the iSCSI Initiator Name Protocol.\r
+    //\r
     Status = gBS->InstallProtocolInterface (\r
                     &ImageHandle,\r
                     &gEfiIScsiInitiatorNameProtocolGuid,\r
@@ -450,13 +462,28 @@ Returns:
             &gIScsiComponentName,\r
             NULL\r
             );\r
+      return Status;\r
+    }\r
+  \r
+    //\r
+    // Initialize the configuration form of iSCSI.\r
+    //\r
+    Status = IScsiConfigFormInit (gIScsiDriverBinding.DriverBindingHandle);\r
+    if (EFI_ERROR (Status)) {\r
+      gBS->UninstallMultipleProtocolInterfaces (\r
+            ImageHandle,\r
+            &gEfiDriverBindingProtocolGuid,\r
+            &gIScsiDriverBinding,\r
+            &gEfiComponentName2ProtocolGuid,\r
+            &gIScsiComponentName2,\r
+            &gEfiComponentNameProtocolGuid,\r
+            &gIScsiComponentName,\r
+            &gEfiIScsiInitiatorNameProtocolGuid,\r
+            &gIScsiInitiatorName,\r
+            NULL\r
+            );\r
     }\r
   }\r
-  //\r
-  // Initialize the configuration form of iSCSI.\r
-  //\r
-  IScsiConfigFormInit (gIScsiDriverBinding.DriverBindingHandle);\r
-\r
   return Status;\r
 }\r
 \r