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

[mirror_edk2.git] / MdeModulePkg / Universal / Network / Mtftp4Dxe / Mtftp4Option.c

diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Option.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Option.c
index 64932ae8a22634e376e11223adafdd61098febcf..f413c763053015d0b54d1b83e2e81e12c6e17f14 100644 (file)
--- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Option.c
+++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Option.c
@@ -167,7 +167,7 @@ Mtftp4ExtractOptions (
     return EFI_SUCCESS;\r
   }\r
 \r
-  *OptionList = NetAllocatePool (*OptionCount * sizeof (EFI_MTFTP4_OPTION));\r
+  *OptionList = AllocatePool (*OptionCount * sizeof (EFI_MTFTP4_OPTION));\r
 \r
   if (*OptionList == NULL) {\r
     return EFI_OUT_OF_RESOURCES;\r
@@ -537,6 +537,6 @@ Mtftp4ParseOptionOack (
 \r
   Status = Mtftp4ParseOption (OptionList, Count, FALSE, MtftpOption);\r
 \r
-  NetFreePool (OptionList);\r
+  gBS->FreePool (OptionList);\r
   return Status;\r
 }\r