Use Mde library and definition instead of some native definitions in NetLib, to simpl...

[mirror_edk2.git] / MdeModulePkg / Universal / Network / Udp4Dxe / Udp4Driver.c

diff --git a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Driver.c b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Driver.c
index 7a7c184498688cbebe59654bb95e41a5dde47059..0c5436412799be1dfbe5770679b4d3ca0897fc5c 100644 (file)
--- a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Driver.c
+++ b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Driver.c
@@ -116,14 +116,14 @@ Udp4DriverBindingStart (
   //\r
   // Allocate Private Context Data Structure.\r
   //\r
-  Udp4Service = NetAllocatePool (sizeof (UDP4_SERVICE_DATA));\r
+  Udp4Service = AllocatePool (sizeof (UDP4_SERVICE_DATA));\r
   if (Udp4Service == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
 \r
   Status = Udp4CreateService (Udp4Service, This->DriverBindingHandle, ControllerHandle);\r
   if (EFI_ERROR (Status)) {\r
-    NetFreePool (Udp4Service);\r
+    gBS->FreePool (Udp4Service);\r
     return Status;\r
   }\r
 \r
@@ -138,7 +138,7 @@ Udp4DriverBindingStart (
                   );\r
   if (EFI_ERROR (Status)) {\r
     Udp4CleanService (Udp4Service);\r
-    NetFreePool (Udp4Service);\r
+    gBS->FreePool (Udp4Service);\r
   } else {\r
     Udp4SetVariableData (Udp4Service);\r
   }\r
@@ -213,10 +213,10 @@ Udp4DriverBindingStop (
 \r
     Udp4CleanService (Udp4Service);\r
 \r
-    NetFreePool (Udp4Service);\r
+    gBS->FreePool (Udp4Service);\r
   } else {\r
 \r
-    while (!NetListIsEmpty (&Udp4Service->ChildrenList)) {\r
+    while (!IsListEmpty (&Udp4Service->ChildrenList)) {\r
       Instance = NET_LIST_HEAD (&Udp4Service->ChildrenList, UDP4_INSTANCE_DATA, Link);\r
 \r
       ServiceBinding->DestroyChild (ServiceBinding, Instance->ChildHandle);\r
@@ -264,7 +264,7 @@ Udp4ServiceBindingCreateChild (
   //\r
   // Allocate the instance private data structure.\r
   //\r
-  Instance = NetAllocateZeroPool (sizeof (UDP4_INSTANCE_DATA));\r
+  Instance = AllocateZeroPool (sizeof (UDP4_INSTANCE_DATA));\r
   if (Instance == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
@@ -310,15 +310,15 @@ Udp4ServiceBindingCreateChild (
     goto ON_ERROR;\r
   }\r
 \r
-  OldTpl = NET_RAISE_TPL (NET_TPL_LOCK);\r
+  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
 \r
   //\r
   // Link this instance into the service context data and increase the ChildrenNumber.\r
   //\r
-  NetListInsertTail (&Udp4Service->ChildrenList, &Instance->Link);\r
+  InsertTailList (&Udp4Service->ChildrenList, &Instance->Link);\r
   Udp4Service->ChildrenNumber++;\r
 \r
-  NET_RESTORE_TPL (OldTpl);\r
+  gBS->RestoreTPL (OldTpl);\r
 \r
   return EFI_SUCCESS;\r
 \r
@@ -339,7 +339,7 @@ ON_ERROR:
 \r
   Udp4CleanInstance (Instance);\r
 \r
-  NetFreePool (Instance);\r
+  gBS->FreePool (Instance);\r
 \r
   return Status;\r
 }\r
@@ -440,12 +440,12 @@ Udp4ServiceBindingDestroyChild (
   //\r
   IpIoRemoveIp (Udp4Service->IpIo, Instance->IpInfo);\r
 \r
-  OldTpl = NET_RAISE_TPL (NET_TPL_LOCK);\r
+  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
 \r
   //\r
   // Remove this instance from the service context data's ChildrenList.\r
   //\r
-  NetListRemoveEntry (&Instance->Link);\r
+  RemoveEntryList (&Instance->Link);\r
   Udp4Service->ChildrenNumber--;\r
 \r
   //\r
@@ -453,9 +453,9 @@ Udp4ServiceBindingDestroyChild (
   //\r
   Udp4CleanInstance (Instance);\r
 \r
-  NET_RESTORE_TPL (OldTpl);\r
+  gBS->RestoreTPL (OldTpl);\r
 \r
-  NetFreePool (Instance);\r
+  gBS->FreePool (Instance);\r
 \r
   return EFI_SUCCESS;\r
 }\r