Fix a bug in IP4 driver which may cause “reconnect –r” hang.

Signed-off-by: Fu Siyuan <siyuan.fu@intel.com>
Reviewed-by: Ye Ting <ting.ye@intel.com>
Reviewed-by: Ouyang Qian <qian.ouyang@intel.com>


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

1 files changed, 21 insertions, 17 deletions

diff --git a/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4If.c b/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4If.c
index 224244164e..6711636333 100644
--- a/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4If.c
+++ b/MdeModulePkg/Universal/Network/Ip4Dxe/Ip4If.c
@@ -1,7 +1,7 @@
 /** @file

   Implement IP4 pesudo interface.

   

-Copyright (c) 2005 - 2012, Intel Corporation. All rights reserved.<BR>

+Copyright (c) 2005 - 2013, Intel Corporation. All rights reserved.<BR>

 This program and the accompanying materials

 are licensed and made available under the terms and conditions of the BSD License

 which accompanies this distribution.  The full text of the license may be found at

@@ -740,24 +740,11 @@ Ip4FreeInterface (
   //

   Ip4CancelFrames (Interface, EFI_ABORTED, Ip4CancelInstanceFrame, IpInstance);

 

-  if (--Interface->RefCnt > 0) {

-    return EFI_SUCCESS;

-  }

-

   //

-  // Destroy the interface if this is the last IP instance that

-  // has the address. Remove all the system transmitted packets

-  // from this interface, cancel the receive request if there is

-  // one, and destroy the ARP requests.

+  // Destroy the ARP instance if this is the last IP instance that

+  // has the address.

   //

-  Ip4CancelFrames (Interface, EFI_ABORTED, Ip4CancelInstanceFrame, NULL);

-  Ip4CancelReceive (Interface);

-

-  ASSERT (IsListEmpty (&Interface->IpInstances));

-  ASSERT (IsListEmpty (&Interface->ArpQues));

-  ASSERT (IsListEmpty (&Interface->SentFrames));

-

-  if (Interface->Arp != NULL) {

+  if (Interface->Arp != NULL && IsListEmpty (&Interface->IpInstances)) {

     gBS->CloseProtocol (

           Interface->ArpHandle,

           &gEfiArpProtocolGuid,

@@ -771,8 +758,25 @@ Ip4FreeInterface (
       &gEfiArpServiceBindingProtocolGuid,

       Interface->ArpHandle

       );

+    Interface->Arp = NULL;

+  }

+  

+  if (--Interface->RefCnt > 0) {

+    return EFI_SUCCESS;

   }

 

+  //

+  // Destroy the interface if it is not referenced by any IP instance (for common Interface)

+  // or the IP service (for the DefaultInterface). Remove all the system transmitted packets

+  // from this interface, cancel the receive request if there is one.

+  //

+  Ip4CancelFrames (Interface, EFI_ABORTED, Ip4CancelInstanceFrame, NULL);

+  Ip4CancelReceive (Interface);

+

+  ASSERT (IsListEmpty (&Interface->IpInstances));

+  ASSERT (IsListEmpty (&Interface->ArpQues));

+  ASSERT (IsListEmpty (&Interface->SentFrames));

+

   RemoveEntryList (&Interface->Link);

   FreePool (Interface);