refine the code and add more security check.

[mirror_edk2.git] / MdeModulePkg / Universal / Network / Tcp4Dxe / Tcp4Driver.c

diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
index cffd53e9a8e2030664629388b5c5859e3fc3d5b7..f6897e9fb6438105022d3a37d8f35b18971e1276 100644 (file)
--- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
+++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
@@ -38,8 +38,8 @@ EFI_TCP4_PROTOCOL mTcp4ProtocolTemplate = {
 };\r
 \r
 SOCK_INIT_DATA mTcp4DefaultSockData = {\r
-  SOCK_STREAM,\r
-  (SOCK_STATE) 0,\r
+  SockStream,\r
+  0,\r
   NULL,\r
   TCP_BACKLOG,\r
   TCP_SND_BUF_SIZE,\r
@@ -389,7 +389,7 @@ ON_ERROR:
     IpIoDestroy (TcpServiceData->IpIo);\r
   }\r
 \r
-  gBS->FreePool (TcpServiceData);\r
+  FreePool (TcpServiceData);\r
 \r
   return Status;\r
 }\r
@@ -494,7 +494,7 @@ Tcp4DriverBindingStop (
     //\r
     // Release the TCP service data\r
     //\r
-    gBS->FreePool (TcpServiceData);\r
+    FreePool (TcpServiceData);\r
   } else {\r
 \r
     while (!IsListEmpty (&TcpServiceData->SocketList)) {\r