Mtftp6Srv->Signature = MTFTP6_SERVICE_SIGNATURE;\r
Mtftp6Srv->Controller = Controller;\r
Mtftp6Srv->Image = Image;\r
- Mtftp6Srv->InDestroy = FALSE;\r
Mtftp6Srv->ChildrenNum = 0;\r
\r
CopyMem (\r
}\r
\r
\r
+/**\r
+ Callback function which provided by user to remove one node in NetDestroyLinkList process.\r
+ \r
+ @param[in] Entry The entry to be removed.\r
+ @param[in] Context Pointer to the callback context corresponds to the Context in NetDestroyLinkList.\r
+\r
+ @retval EFI_SUCCESS The entry has been removed successfully.\r
+ @retval Others Fail to remove the entry.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+Mtftp6DestroyChildEntryInHandleBuffer (\r
+ IN LIST_ENTRY *Entry,\r
+ IN VOID *Context\r
+ )\r
+{\r
+ MTFTP6_INSTANCE *Instance;\r
+ EFI_SERVICE_BINDING_PROTOCOL *ServiceBinding;\r
+ UINTN NumberOfChildren;\r
+ EFI_HANDLE *ChildHandleBuffer;\r
+\r
+ if (Entry == NULL || Context == NULL) {\r
+ return EFI_INVALID_PARAMETER;\r
+ }\r
+\r
+ Instance = NET_LIST_USER_STRUCT_S (Entry, MTFTP6_INSTANCE, Link, MTFTP6_INSTANCE_SIGNATURE);\r
+ ServiceBinding = ((MTFTP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *) Context)->ServiceBinding;\r
+ NumberOfChildren = ((MTFTP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *) Context)->NumberOfChildren;\r
+ ChildHandleBuffer = ((MTFTP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *) Context)->ChildHandleBuffer;\r
+\r
+ if (!NetIsInHandleBuffer (Instance->Handle, NumberOfChildren, ChildHandleBuffer)) {\r
+ return EFI_SUCCESS;\r
+ }\r
+\r
+ return ServiceBinding->DestroyChild (ServiceBinding, Instance->Handle);\r
+}\r
+\r
+\r
/**\r
This is the declaration of an EFI image entry point. This entry point is\r
the same for UEFI Applications, UEFI OS Loaders, and UEFI Drivers, including\r
IN EFI_HANDLE *ChildHandleBuffer\r
)\r
{\r
- EFI_SERVICE_BINDING_PROTOCOL *ServiceBinding;\r
- MTFTP6_SERVICE *Service;\r
- MTFTP6_INSTANCE *Instance;\r
- EFI_HANDLE NicHandle;\r
- EFI_STATUS Status;\r
- EFI_TPL OldTpl;\r
-\r
+ EFI_SERVICE_BINDING_PROTOCOL *ServiceBinding;\r
+ MTFTP6_SERVICE *Service;\r
+ EFI_HANDLE NicHandle;\r
+ EFI_STATUS Status;\r
+ LIST_ENTRY *List;\r
+ MTFTP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT Context;\r
+ \r
//\r
// Locate the Nic handle to retrieve the Mtftp6 private data.\r
//\r
NicHandle = NetLibGetNicHandle (Controller, &gEfiUdp6ProtocolGuid);\r
\r
if (NicHandle == NULL) {\r
- return EFI_DEVICE_ERROR;\r
+ return EFI_SUCCESS;\r
}\r
\r
Status = gBS->OpenProtocol (\r
\r
Service = MTFTP6_SERVICE_FROM_THIS (ServiceBinding);\r
\r
- if (Service->InDestroy) {\r
- return EFI_SUCCESS;\r
+ if (!IsListEmpty (&Service->Children)) {\r
+ //\r
+ // Destroy the Mtftp6 child instance in ChildHandleBuffer.\r
+ //\r
+ List = &Service->Children;\r
+ Context.ServiceBinding = ServiceBinding;\r
+ Context.NumberOfChildren = NumberOfChildren;\r
+ Context.ChildHandleBuffer = ChildHandleBuffer;\r
+ Status = NetDestroyLinkList (\r
+ List,\r
+ Mtftp6DestroyChildEntryInHandleBuffer,\r
+ &Context,\r
+ NULL\r
+ );\r
}\r
\r
- OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
-\r
- if (NumberOfChildren == 0) {\r
+ if (NumberOfChildren == 0 && IsListEmpty (&Service->Children)) {\r
//\r
// Destroy the Mtftp6 service if there is no Mtftp6 child instance left.\r
//\r
- Service->InDestroy = TRUE;\r
-\r
gBS->UninstallProtocolInterface (\r
NicHandle,\r
&gEfiMtftp6ServiceBindingProtocolGuid,\r
);\r
\r
Mtftp6DestroyService (Service);\r
-\r
- } else {\r
- //\r
- // Destroy the Mtftp6 child instance one by one.\r
- //\r
- while (!IsListEmpty (&Service->Children)) {\r
- Instance = NET_LIST_HEAD (&Service->Children, MTFTP6_INSTANCE, Link);\r
- Mtftp6ServiceBindingDestroyChild (ServiceBinding, Instance->Handle);\r
- }\r
-\r
- if (Service->ChildrenNum != 0) {\r
- Status = EFI_DEVICE_ERROR;\r
- }\r
+ Status = EFI_SUCCESS;\r
}\r
\r
- gBS->RestoreTPL (OldTpl);\r
return Status;\r
}\r
\r
ChildHandle\r
);\r
\r
+ if (Instance->UdpIo != NULL) {\r
+ gBS->CloseProtocol (\r
+ Instance->UdpIo->UdpHandle,\r
+ &gEfiUdp6ProtocolGuid,\r
+ gMtftp6DriverBinding.DriverBindingHandle,\r
+ Instance->Handle\r
+ );\r
+ }\r
+\r
+ if (Instance->McastUdpIo != NULL) {\r
+ gBS->CloseProtocol (\r
+ Instance->McastUdpIo->UdpHandle,\r
+ &gEfiUdp6ProtocolGuid,\r
+ gMtftp6DriverBinding.DriverBindingHandle,\r
+ Instance->Handle\r
+ );\r
+ }\r
+\r
//\r
// Uninstall the MTFTP6 protocol first to enable a top down destruction.\r
//\r
+ gBS->RestoreTPL (OldTpl);\r
Status = gBS->UninstallProtocolInterface (\r
ChildHandle,\r
&gEfiMtftp6ProtocolGuid,\r
Mtftp6\r
);\r
-\r
+ OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
if (EFI_ERROR (Status)) {\r
Instance->InDestroy = FALSE;\r
gBS->RestoreTPL (OldTpl);\r
RemoveEntryList (&Instance->Link);\r
Service->ChildrenNum --;\r
\r
- Mtftp6DestroyInstance (Instance);\r
-\r
gBS->RestoreTPL (OldTpl);\r
\r
+ Mtftp6DestroyInstance (Instance);\r
+\r
return EFI_SUCCESS;\r
}\r