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

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
+  // Destroy the ARP instance if this is the last IP instance that
-  // has the address. Remove all the system transmitted packets
+  // has the address.
-  // from this interface, cancel the receive request if there is
-  // one, and destroy the ARP requests.
   //
-  Ip4CancelFrames (Interface, EFI_ABORTED, Ip4CancelInstanceFrame, NULL);
+  if (Interface->Arp != NULL && IsListEmpty (&Interface->IpInstances)) {
-  Ip4CancelReceive (Interface);
-
-  ASSERT (IsListEmpty (&Interface->IpInstances));
-  ASSERT (IsListEmpty (&Interface->ArpQues));
-  ASSERT (IsListEmpty (&Interface->SentFrames));
-
-  if (Interface->Arp != NULL) {
     gBS->CloseProtocol (
           Interface->ArpHandle,
           &gEfiArpProtocolGuid,
@@ -771,7 +758,24 @@ 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);

MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Impl.c

@@ -1,6 +1,6 @@
 /** @file
 
-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
@@ -562,6 +562,9 @@ Ip4AutoConfigCallBackDpc (
   IP4_ADDR                  SubnetMask;
   IP4_ADDR                  SubnetAddress;
   IP4_ADDR                  GatewayAddress;
+  IP4_PROTOCOL              *Ip4Instance;
+  EFI_ARP_PROTOCOL          *Arp;
+  LIST_ENTRY                *Entry;
 
   IpSb      = (IP4_SERVICE *) Context;
   NET_CHECK_SIGNATURE (IpSb, IP4_SERVICE_SIGNATURE);
@@ -650,11 +653,33 @@ Ip4AutoConfigCallBackDpc (
   StationAddress = EFI_NTOHL (Data->StationAddress);
   SubnetMask = EFI_NTOHL (Data->SubnetMask);
   Status = Ip4SetAddress (IpIf, StationAddress, SubnetMask);
-
   if (EFI_ERROR (Status)) {
     goto ON_EXIT;
   }
 
+  if (IpIf->Arp != NULL) {
+    //   
+    // A non-NULL IpIf->Arp here means a new ARP child is created when setting default address, 
+    // but some IP children may have referenced the default interface before it is configured,
+    // these IP instances also consume this ARP protocol so they need to open it BY_CHILD_CONTROLLER.
+    //
+    Arp = NULL;
+    NET_LIST_FOR_EACH (Entry, &IpIf->IpInstances) {
+      Ip4Instance = NET_LIST_USER_STRUCT_S (Entry, IP4_PROTOCOL, AddrLink, IP4_PROTOCOL_SIGNATURE);
+      Status = gBS->OpenProtocol (
+                      IpIf->ArpHandle,
+                      &gEfiArpProtocolGuid,
+                      (VOID **) &Arp,
+                      gIp4DriverBinding.DriverBindingHandle,
+                      Ip4Instance->Handle,
+                      EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
+                      );
+      if (EFI_ERROR (Status)) {
+        goto ON_EXIT;
+      }
+    }
+  }
+
   Ip4AddRoute (
     IpSb->DefaultRouteTable,
     StationAddress,