1. Add EFI_COMPONENT_NAME2_PROTOCOL.GetControllerName() support.

[mirror_edk2.git] / NetworkPkg / Mtftp6Dxe / Mtftp6Impl.c

diff --git a/NetworkPkg/Mtftp6Dxe/Mtftp6Impl.c b/NetworkPkg/Mtftp6Dxe/Mtftp6Impl.c
index 4a4e5b192c27f533c6c4ba36b4b233fbe7186109..9b08455ef3ac0ff35db53a0e853beede8e1f3f91 100644 (file)
--- a/NetworkPkg/Mtftp6Dxe/Mtftp6Impl.c
+++ b/NetworkPkg/Mtftp6Dxe/Mtftp6Impl.c
@@ -197,6 +197,19 @@ EfiMtftp6Configure (
                           UDP_IO_UDP6_VERSION,\r
                           NULL\r
                           );\r
+      if (Instance->UdpIo != NULL) {\r
+        Status = gBS->OpenProtocol (\r
+                        Instance->UdpIo->UdpHandle,\r
+                        &gEfiUdp6ProtocolGuid,\r
+                        (VOID **) &Udp6,\r
+                        Service->Image,\r
+                        Instance->Handle,\r
+                        EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER\r
+                        );\r
+        if (EFI_ERROR (Status)) {\r
+          goto ON_EXIT;\r
+        }\r
+      }\r
     }\r
 \r
     if (Instance->UdpIo == NULL) {\r
@@ -626,8 +639,6 @@ EfiMtftp6Poll (
   //\r
   if (Instance->Config == NULL) {\r
     return EFI_NOT_STARTED;\r
-  } else if (Instance->InDestroy) {\r
-    return EFI_DEVICE_ERROR;\r
   }\r
 \r
   Udp6 = Instance->UdpIo->Protocol.Udp6;\r