]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/Tcp4Dxe/SockImpl.c
Update the copyright notice format
[mirror_edk2.git] / MdeModulePkg / Universal / Network / Tcp4Dxe / SockImpl.c
index 46c6ba492cbd7d3598dfde983a7a3d70d9dd372c..c627a140a89060a80cd6d03221b494e9f39a9952 100644 (file)
@@ -1,40 +1,58 @@
 /** @file\r
+  Implementation of the Socket.\r
 \r
-Copyright (c) 2005 - 2006, Intel Corporation\r
-All rights reserved. This program and the accompanying materials\r
+Copyright (c) 2005 - 2009, Intel Corporation. All rights reserved.<BR>\r
+This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
+http://opensource.org/licenses/bsd-license.php<BR>\r
 \r
 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 \r
-Module Name:\r
-\r
-  SockImpl.c\r
+**/\r
 \r
-Abstract:\r
+#include "SockImpl.h"\r
 \r
+/**\r
+  Get the length of the data that can be retrieved from the socket\r
+  receive buffer.\r
 \r
-**/\r
+  @param  SockBuffer            Pointer to the socket receive buffer.\r
+  @param  IsUrg                 Pointer to a BOOLEAN variable. If TRUE the data is\r
+                                OOB.\r
+  @param  BufLen                The maximum length of the data buffer to store the\r
+                                received data in socket layer.\r
 \r
-#include "SockImpl.h"\r
+  @return The length of the data can be retreived.\r
 \r
-STATIC\r
+**/\r
 UINT32\r
 SockTcpDataToRcv (\r
-  IN  SOCK_BUFFER   *SockBuffer,\r
-  OUT BOOLEAN       *IsOOB,\r
-  IN  UINT32        BufLen\r
+  IN     SOCK_BUFFER    *SockBuffer,\r
+     OUT BOOLEAN        *IsUrg,\r
+  IN     UINT32         BufLen\r
   );\r
+  \r
+/**\r
+  Process the send token.\r
 \r
-STATIC\r
+  @param  Sock                  Pointer to the socket.\r
+\r
+**/\r
 VOID\r
 SockProcessSndToken (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   );\r
 \r
+/**\r
+  Supporting function for both SockImpl and SockInterface.\r
+\r
+  @param Event             The Event this notify function registered to, ignored.\r
+  \r
+**/\r
 VOID\r
+EFIAPI\r
 SockFreeFoo (\r
   IN EFI_EVENT Event\r
   )\r
@@ -56,23 +74,21 @@ SockFreeFoo (
   @return The length of the data can be retreived.\r
 \r
 **/\r
-STATIC\r
 UINT32\r
 SockTcpDataToRcv (\r
-  IN  SOCK_BUFFER    *SockBuffer,\r
-  OUT BOOLEAN        *IsUrg,\r
-  IN  UINT32         BufLen\r
+  IN     SOCK_BUFFER    *SockBuffer,\r
+     OUT BOOLEAN        *IsUrg,\r
+  IN     UINT32         BufLen\r
   )\r
 {\r
   NET_BUF       *RcvBufEntry;\r
   UINT32        DataLen;\r
   TCP_RSV_DATA  *TcpRsvData;\r
   BOOLEAN       Urg;\r
-\r
-  ASSERT (SockBuffer && IsUrg && (BufLen > 0));\r
+  ASSERT ((SockBuffer != NULL) && (IsUrg != NULL) && (BufLen > 0));\r
 \r
   RcvBufEntry = SockBufFirst (SockBuffer);\r
-  ASSERT (RcvBufEntry);\r
+  ASSERT (RcvBufEntry != NULL);\r
 \r
   TcpRsvData  = (TCP_RSV_DATA *) RcvBufEntry->ProtoData;\r
 \r
@@ -134,9 +150,7 @@ SockTcpDataToRcv (
   @param  Sock                  Pointer to the socket.\r
   @param  TcpRxData             Pointer to the application provided receive buffer.\r
   @param  RcvdBytes             The maximum length of the data can be copied.\r
-  @param  IsOOB                 If TURE the data is OOB, else the data is normal.\r
-\r
-  @return None.\r
+  @param  IsOOB                 If TURE the data is OOB, FALSE the data is normal.\r
 \r
 **/\r
 VOID\r
@@ -192,17 +206,17 @@ SockSetTcpRxData (
 **/\r
 UINT32\r
 SockProcessRcvToken (\r
-  IN SOCKET        *Sock,\r
-  IN SOCK_IO_TOKEN *RcvToken\r
+  IN     SOCKET        *Sock,\r
+  IN OUT SOCK_IO_TOKEN *RcvToken\r
   )\r
 {\r
   UINT32                 TokenRcvdBytes;\r
   EFI_TCP4_RECEIVE_DATA  *RxData;\r
   BOOLEAN                IsUrg;\r
 \r
-  ASSERT (Sock);\r
+  ASSERT (Sock != NULL);\r
 \r
-  ASSERT (SOCK_STREAM == Sock->Type);\r
+  ASSERT (SockStream == Sock->Type);\r
 \r
   RxData = RcvToken->Packet.RxData;\r
 \r
@@ -218,7 +232,7 @@ SockProcessRcvToken (
   //\r
   SockSetTcpRxData (Sock, RxData, TokenRcvdBytes, IsUrg);\r
 \r
-  SOCK_TRIM_RCV_BUFF (Sock, TokenRcvdBytes);\r
+  NetbufQueTrim (Sock->RcvBuffer.DataQueue, TokenRcvdBytes);\r
   SIGNAL_TOKEN (&(RcvToken->Token), EFI_SUCCESS);\r
 \r
   return TokenRcvdBytes;\r
@@ -230,7 +244,7 @@ SockProcessRcvToken (
   the buffer to socket send buffer,then try to send it.\r
 \r
   @param  Sock                  Pointer to the socket.\r
-  @param  TcpTxData             Pointer to the tcp txdata.\r
+  @param  TcpTxData             Pointer to the application provided send buffer.\r
 \r
   @retval EFI_SUCCESS           The operation is completed successfully.\r
   @retval EFI_OUT_OF_RESOURCES  Failed due to resource limit.\r
@@ -262,7 +276,7 @@ SockProcessTcpSndData (
               );\r
 \r
   if (NULL == SndData) {\r
-    SOCK_DEBUG_ERROR (("SockKProcessSndData: Failed to"\r
+    DEBUG ((EFI_D_ERROR, "SockKProcessSndData: Failed to"\r
       " call NetBufferFromExt\n"));\r
 \r
     return EFI_OUT_OF_RESOURCES;\r
@@ -308,23 +322,20 @@ SockProcessTcpSndData (
 \r
   @param  Sock                  Pointer to the socket.\r
   @param  PendingTokenList      Pointer to the token list to be flushed.\r
-\r
-  @return None.\r
-\r
+  \r
 **/\r
-STATIC\r
 VOID\r
 SockFlushPendingToken (\r
   IN SOCKET         *Sock,\r
-  IN NET_LIST_ENTRY *PendingTokenList\r
+  IN LIST_ENTRY     *PendingTokenList\r
   )\r
 {\r
   SOCK_TOKEN            *SockToken;\r
   SOCK_COMPLETION_TOKEN *Token;\r
 \r
-  ASSERT (Sock && PendingTokenList);\r
+  ASSERT ((Sock != NULL) && (PendingTokenList != NULL));\r
 \r
-  while (!NetListIsEmpty (PendingTokenList)) {\r
+  while (!IsListEmpty (PendingTokenList)) {\r
     SockToken = NET_LIST_HEAD (\r
                   PendingTokenList,\r
                   SOCK_TOKEN,\r
@@ -334,26 +345,22 @@ SockFlushPendingToken (
     Token = SockToken->Token;\r
     SIGNAL_TOKEN (Token, Sock->SockError);\r
 \r
-    NetListRemoveEntry (&(SockToken->TokenList));\r
-    NetFreePool (SockToken);\r
+    RemoveEntryList (&(SockToken->TokenList));\r
+    FreePool (SockToken);\r
   }\r
 }\r
 \r
 \r
 /**\r
-  Wake up the connection token while the connection is\r
-  successfully established, then try to process any\r
-  pending send token.\r
+  Wake up the connection token while the connection is successfully established, \r
+  then try to process any pending send token.\r
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
-STATIC\r
 VOID\r
 SockWakeConnToken (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   ASSERT (Sock->ConnectionToken != NULL);\r
@@ -370,18 +377,14 @@ SockWakeConnToken (
 \r
 \r
 /**\r
-  Wake up the listen token while the connection is\r
-  established successfully.\r
+  Wake up the listen token while the connection is established successfully.\r
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
-STATIC\r
 VOID\r
 SockWakeListenToken (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   SOCKET                *Parent;\r
@@ -390,9 +393,9 @@ SockWakeListenToken (
 \r
   Parent = Sock->Parent;\r
 \r
-  ASSERT (Parent && SOCK_IS_LISTENING (Parent) && SOCK_IS_CONNECTED (Sock));\r
+  ASSERT ((Parent != NULL) && SOCK_IS_LISTENING (Parent) && SOCK_IS_CONNECTED (Sock));\r
 \r
-  if (!NetListIsEmpty (&Parent->ListenTokenList)) {\r
+  if (!IsListEmpty (&Parent->ListenTokenList)) {\r
     SockToken = NET_LIST_HEAD (\r
                   &Parent->ListenTokenList,\r
                   SOCK_TOKEN,\r
@@ -404,14 +407,13 @@ SockWakeListenToken (
 \r
     SIGNAL_TOKEN (&(ListenToken->CompletionToken), EFI_SUCCESS);\r
 \r
-    NetListRemoveEntry (&SockToken->TokenList);\r
-    NetFreePool (SockToken);\r
+    RemoveEntryList (&SockToken->TokenList);\r
+    FreePool (SockToken);\r
 \r
-    NetListRemoveEntry (&Sock->ConnectionList);\r
+    RemoveEntryList (&Sock->ConnectionList);\r
 \r
     Parent->ConnCnt--;\r
-    SOCK_DEBUG_WARN (("SockWakeListenToken: accept a socket,"\r
-      "now conncnt is %d", Parent->ConnCnt));\r
+    DEBUG ((EFI_D_INFO, "SockWakeListenToken: accept a socket, now conncnt is %d", Parent->ConnCnt));\r
 \r
     Sock->Parent = NULL;\r
   }\r
@@ -423,10 +425,7 @@ SockWakeListenToken (
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
-STATIC\r
 VOID\r
 SockWakeRcvToken (\r
   IN SOCKET *Sock\r
@@ -437,13 +436,13 @@ SockWakeRcvToken (
   SOCK_TOKEN    *SockToken;\r
   SOCK_IO_TOKEN *RcvToken;\r
 \r
-  ASSERT (Sock->RcvBuffer.DataQueue);\r
+  ASSERT (Sock->RcvBuffer.DataQueue != NULL);\r
 \r
   RcvdBytes = (Sock->RcvBuffer.DataQueue)->BufSize;\r
 \r
   ASSERT (RcvdBytes > 0);\r
 \r
-  while (RcvdBytes > 0 && !NetListIsEmpty (&Sock->RcvTokenList)) {\r
+  while (RcvdBytes > 0 && !IsListEmpty (&Sock->RcvTokenList)) {\r
 \r
     SockToken = NET_LIST_HEAD (\r
                   &Sock->RcvTokenList,\r
@@ -458,8 +457,8 @@ SockWakeRcvToken (
       return ;\r
     }\r
 \r
-    NetListRemoveEntry (&(SockToken->TokenList));\r
-    NetFreePool (SockToken);\r
+    RemoveEntryList (&(SockToken->TokenList));\r
+    FreePool (SockToken);\r
     RcvdBytes -= TokenRcvdBytes;\r
   }\r
 }\r
@@ -470,13 +469,10 @@ SockWakeRcvToken (
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
-STATIC\r
 VOID\r
 SockProcessSndToken (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   UINT32                  FreeSpace;\r
@@ -486,7 +482,7 @@ SockProcessSndToken (
   EFI_TCP4_TRANSMIT_DATA  *TxData;\r
   EFI_STATUS              Status;\r
 \r
-  ASSERT (Sock && (SOCK_STREAM == Sock->Type));\r
+  ASSERT ((Sock != NULL) && (SockStream == Sock->Type));\r
 \r
   FreeSpace = SockGetFreeSpace (Sock, SOCK_SND_BUF);\r
 \r
@@ -495,7 +491,7 @@ SockProcessSndToken (
   // socket layer flow control policy\r
   //\r
   while ((FreeSpace >= Sock->SndBuffer.LowWater) &&\r
-         !NetListIsEmpty (&Sock->SndTokenList)) {\r
+         !IsListEmpty (&Sock->SndTokenList)) {\r
 \r
     SockToken = NET_LIST_HEAD (\r
                   &(Sock->SndTokenList),\r
@@ -506,8 +502,8 @@ SockProcessSndToken (
     //\r
     // process this token\r
     //\r
-    NetListRemoveEntry (&(SockToken->TokenList));\r
-    NetListInsertTail (\r
+    RemoveEntryList (&(SockToken->TokenList));\r
+    InsertTailList (\r
       &(Sock->ProcessingSndTokenList),\r
       &(SockToken->TokenList)\r
       );\r
@@ -538,9 +534,9 @@ SockProcessSndToken (
 \r
 OnError:\r
 \r
-  NetListRemoveEntry (&SockToken->TokenList);\r
+  RemoveEntryList (&SockToken->TokenList);\r
   SIGNAL_TOKEN (SockToken->Token, Status);\r
-  NetFreePool (SockToken);\r
+  FreePool (SockToken);\r
 }\r
 \r
 \r
@@ -549,7 +545,7 @@ OnError:
 \r
   @param  SockInitData          Pointer to the initial data of the socket.\r
 \r
-  @return Pointer to the newly created socket.\r
+  @return Pointer to the newly created socket, return NULL when exception occured.\r
 \r
 **/\r
 SOCKET *\r
@@ -561,15 +557,16 @@ SockCreate (
   SOCKET      *Parent;\r
   EFI_STATUS  Status;\r
 \r
-  ASSERT (SockInitData && SockInitData->ProtoHandler);\r
-  ASSERT (SockInitData->Type == SOCK_STREAM);\r
-  ASSERT (SockInitData->ProtoData && (SockInitData->DataSize <= PROTO_RESERVED_LEN));\r
+  ASSERT ((SockInitData != NULL) && (SockInitData->ProtoHandler != NULL));\r
+  ASSERT (SockInitData->Type == SockStream);\r
+  ASSERT ((SockInitData->ProtoData != NULL) && (SockInitData->DataSize <= PROTO_RESERVED_LEN));\r
 \r
   Parent = SockInitData->Parent;\r
 \r
-  if (Parent && (Parent->ConnCnt == Parent->BackLog)) {\r
-    SOCK_DEBUG_ERROR (\r
-      ("SockCreate: Socket parent has "\r
+  if ((Parent != NULL) && (Parent->ConnCnt == Parent->BackLog)) {\r
+    DEBUG (\r
+      (EFI_D_ERROR,\r
+      "SockCreate: Socket parent has "\r
       "reached its connection limit with %d ConnCnt and %d BackLog\n",\r
       Parent->ConnCnt,\r
       Parent->BackLog)\r
@@ -578,25 +575,25 @@ SockCreate (
     return NULL;\r
   }\r
 \r
-  Sock = NetAllocateZeroPool (sizeof (SOCKET));\r
+  Sock = AllocateZeroPool (sizeof (SOCKET));\r
   if (NULL == Sock) {\r
 \r
-    SOCK_DEBUG_ERROR (("SockCreate: No resource to create a new socket\n"));\r
+    DEBUG ((EFI_D_ERROR, "SockCreate: No resource to create a new socket\n"));\r
     return NULL;\r
   }\r
 \r
-  NetListInit (&Sock->Link);\r
-  NetListInit (&Sock->ConnectionList);\r
-  NetListInit (&Sock->ListenTokenList);\r
-  NetListInit (&Sock->RcvTokenList);\r
-  NetListInit (&Sock->SndTokenList);\r
-  NetListInit (&Sock->ProcessingSndTokenList);\r
+  InitializeListHead (&Sock->Link);\r
+  InitializeListHead (&Sock->ConnectionList);\r
+  InitializeListHead (&Sock->ListenTokenList);\r
+  InitializeListHead (&Sock->RcvTokenList);\r
+  InitializeListHead (&Sock->SndTokenList);\r
+  InitializeListHead (&Sock->ProcessingSndTokenList);\r
 \r
-  NET_LOCK_INIT (&(Sock->Lock));\r
+  EfiInitializeLock (&(Sock->Lock), TPL_CALLBACK);\r
 \r
   Sock->SndBuffer.DataQueue = NetbufQueAlloc ();\r
   if (NULL == Sock->SndBuffer.DataQueue) {\r
-    SOCK_DEBUG_ERROR (("SockCreate: No resource to allocate"\r
+    DEBUG ((EFI_D_ERROR, "SockCreate: No resource to allocate"\r
       " SndBuffer for new socket\n"));\r
 \r
     goto OnError;\r
@@ -604,7 +601,7 @@ SockCreate (
 \r
   Sock->RcvBuffer.DataQueue = NetbufQueAlloc ();\r
   if (NULL == Sock->RcvBuffer.DataQueue) {\r
-    SOCK_DEBUG_ERROR (("SockCreate: No resource to allocate "\r
+    DEBUG ((EFI_D_ERROR, "SockCreate: No resource to allocate "\r
       "RcvBuffer for new socket\n"));\r
 \r
     goto OnError;\r
@@ -631,7 +628,7 @@ SockCreate (
   //\r
   // Install protocol on Sock->SockHandle\r
   //\r
-  NetCopyMem (\r
+  CopyMem (\r
     &(Sock->NetProtocol.TcpProtocol),\r
     SockInitData->Protocol,\r
     sizeof (EFI_TCP4_PROTOCOL)\r
@@ -640,7 +637,7 @@ SockCreate (
   //\r
   // copy the protodata into socket\r
   //\r
-  NetCopyMem (Sock->ProtoReserved, SockInitData->ProtoData, SockInitData->DataSize);\r
+  CopyMem (Sock->ProtoReserved, SockInitData->ProtoData, SockInitData->DataSize);\r
 \r
   Status = gBS->InstallMultipleProtocolInterfaces (\r
                   &Sock->SockHandle,\r
@@ -650,7 +647,7 @@ SockCreate (
                   );\r
 \r
   if (EFI_ERROR (Status)) {\r
-    SOCK_DEBUG_ERROR (("SockCreate: Install TCP protocol in "\r
+    DEBUG ((EFI_D_ERROR, "SockCreate: Install TCP protocol in "\r
       "socket failed with %r\n", Status));\r
 \r
     goto OnError;\r
@@ -666,10 +663,13 @@ SockCreate (
     //\r
     Parent->ConnCnt++;\r
 \r
-    SOCK_DEBUG_WARN (("SockCreate: Create a new socket and"\r
-      "add to parent, now conncnt is %d\n", Parent->ConnCnt));\r
+    DEBUG (\r
+      (EFI_D_INFO,\r
+      "SockCreate: Create a new socket and add to parent, now conncnt is %d\n",\r
+      Parent->ConnCnt)\r
+      );\r
 \r
-    NetListInsertTail (&Parent->ConnectionList, &Sock->ConnectionList);\r
+    InsertTailList (&Parent->ConnectionList, &Sock->ConnectionList);\r
   }\r
 \r
   if (Sock->CreateCallback != NULL) {\r
@@ -700,7 +700,7 @@ OnError:
     NetbufQueFree (Sock->RcvBuffer.DataQueue);\r
   }\r
 \r
-  NetFreePool (Sock);\r
+  FreePool (Sock);\r
 \r
   return NULL;\r
 }\r
@@ -711,19 +711,17 @@ OnError:
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockDestroy (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   VOID        *SockProtocol;\r
   EFI_GUID    *ProtocolGuid;\r
   EFI_STATUS  Status;\r
 \r
-  ASSERT (SOCK_STREAM == Sock->Type);\r
+  ASSERT (SockStream == Sock->Type);\r
 \r
   if (Sock->DestroyCallback != NULL) {\r
     Sock->DestroyCallback (Sock, Sock->Context);\r
@@ -749,13 +747,17 @@ SockDestroy (
   //\r
   // Remove it from parent connection list if needed\r
   //\r
-  if (Sock->Parent) {\r
+  if (Sock->Parent != NULL) {\r
 \r
-    NetListRemoveEntry (&(Sock->ConnectionList));\r
+    RemoveEntryList (&(Sock->ConnectionList));\r
     (Sock->Parent->ConnCnt)--;\r
 \r
-    SOCK_DEBUG_WARN (("SockDestory: Delete a unaccepted socket from parent"\r
-      "now conncnt is %d\n", Sock->Parent->ConnCnt));\r
+    DEBUG (\r
+      (EFI_D_INFO,\r
+      "SockDestory: Delete a unaccepted socket from parent"\r
+      "now conncnt is %d\n",\r
+      Sock->Parent->ConnCnt)\r
+      );\r
 \r
     Sock->Parent = NULL;\r
   }\r
@@ -780,7 +782,7 @@ SockDestroy (
 \r
   if (EFI_ERROR (Status)) {\r
 \r
-    SOCK_DEBUG_ERROR (("SockDestroy: Open protocol installed "\r
+    DEBUG ((EFI_D_ERROR, "SockDestroy: Open protocol installed "\r
       "on socket failed with %r\n", Status));\r
 \r
     goto FreeSock;\r
@@ -798,27 +800,25 @@ SockDestroy (
         );\r
 \r
 FreeSock:\r
-  NetFreePool (Sock);\r
+  FreePool (Sock);\r
   return ;\r
 }\r
 \r
 \r
 /**\r
-  Flush the socket.\r
+  Flush the sndBuffer and rcvBuffer of socket.\r
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockConnFlush (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   SOCKET  *Child;\r
 \r
-  ASSERT (Sock);\r
+  ASSERT (Sock != NULL);\r
 \r
   //\r
   // Clear the flag in this socket\r
@@ -853,7 +853,7 @@ SockConnFlush (
   // Destroy the pending connection, if it is a listening socket\r
   //\r
   if (SOCK_IS_LISTENING (Sock)) {\r
-    while (!NetListIsEmpty (&Sock->ConnectionList)) {\r
+    while (!IsListEmpty (&Sock->ConnectionList)) {\r
       Child = NET_LIST_HEAD (\r
                 &Sock->ConnectionList,\r
                 SOCKET,\r
@@ -874,15 +874,13 @@ SockConnFlush (
   Set the state of the socket.\r
 \r
   @param  Sock                  Pointer to the socket.\r
-  @param  State                 The new state to be set.\r
-\r
-  @return None.\r
+  @param  State                 The new socket state to be set.\r
 \r
 **/\r
 VOID\r
 SockSetState (\r
-  IN SOCKET     *Sock,\r
-  IN SOCK_STATE State\r
+  IN OUT SOCKET     *Sock,\r
+  IN     UINT8      State\r
   )\r
 {\r
   Sock->State = State;\r
@@ -894,8 +892,7 @@ SockSetState (
 \r
   @param  Sock                  Pointer to the socket to be cloned.\r
 \r
-  @retval *                     Pointer to the newly cloned socket. If NULL, error\r
-                                condition occurred.\r
+  @return Pointer to the newly cloned socket. If NULL, error condition occurred.\r
 \r
 **/\r
 SOCKET *\r
@@ -906,25 +903,25 @@ SockClone (
   SOCKET          *ClonedSock;\r
   SOCK_INIT_DATA  InitData;\r
 \r
-  InitData.BackLog        = Sock->BackLog;\r
-  InitData.Parent         = Sock;\r
-  InitData.State          = Sock->State;\r
-  InitData.ProtoHandler   = Sock->ProtoHandler;\r
-  InitData.Type           = Sock->Type;\r
-  InitData.RcvBufferSize  = Sock->RcvBuffer.HighWater;\r
-  InitData.SndBufferSize  = Sock->SndBuffer.HighWater;\r
-  InitData.DriverBinding  = Sock->DriverBinding;\r
-  InitData.Protocol       = &(Sock->NetProtocol);\r
+  InitData.BackLog         = Sock->BackLog;\r
+  InitData.Parent          = Sock;\r
+  InitData.State           = Sock->State;\r
+  InitData.ProtoHandler    = Sock->ProtoHandler;\r
+  InitData.Type            = Sock->Type;\r
+  InitData.RcvBufferSize   = Sock->RcvBuffer.HighWater;\r
+  InitData.SndBufferSize   = Sock->SndBuffer.HighWater;\r
+  InitData.DriverBinding   = Sock->DriverBinding;\r
+  InitData.Protocol        = &(Sock->NetProtocol);\r
   InitData.CreateCallback  = Sock->CreateCallback;\r
   InitData.DestroyCallback = Sock->DestroyCallback;\r
   InitData.Context         = Sock->Context;\r
   InitData.ProtoData       = Sock->ProtoReserved;\r
   InitData.DataSize        = sizeof (Sock->ProtoReserved);\r
 \r
-  ClonedSock              = SockCreate (&InitData);\r
+  ClonedSock               = SockCreate (&InitData);\r
 \r
   if (NULL == ClonedSock) {\r
-    SOCK_DEBUG_ERROR (("SockClone: no resource to create a cloned sock\n"));\r
+    DEBUG ((EFI_D_ERROR, "SockClone: no resource to create a cloned sock\n"));\r
     return NULL;\r
   }\r
 \r
@@ -936,16 +933,14 @@ SockClone (
 \r
 \r
 /**\r
-  Called by the low layer protocol to indicate the socket\r
-  a connection is established. This function just changes\r
-  the socket's state to SO_CONNECTED and signals the token\r
-  used for connection establishment.\r
+  Called by the low layer protocol to indicate the socket a connection is \r
+  established. \r
+  \r
+  This function just changes the socket's state to SO_CONNECTED \r
+  and signals the token used for connection establishment.\r
 \r
   @param  Sock                  Pointer to the socket associated with the\r
                                 established connection.\r
-\r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockConnEstablished (\r
@@ -968,22 +963,21 @@ SockConnEstablished (
 \r
 \r
 /**\r
-  Called by the low layer protocol to indicate the connection\r
-  is closed. This function flushes the socket, sets the state\r
-  to SO_CLOSED and signals the close token.\r
+  Called by the low layer protocol to indicate the connection is closed.\r
+  \r
+  This function flushes the socket, sets the state to SO_CLOSED and signals \r
+  the close token.\r
 \r
   @param  Sock                  Pointer to the socket associated with the closed\r
                                 connection.\r
-\r
-  @return None.\r
-\r
+                                \r
 **/\r
 VOID\r
 SockConnClosed (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
-  if (Sock->CloseToken) {\r
+  if (Sock->CloseToken != NULL) {\r
     SIGNAL_TOKEN (Sock->CloseToken, EFI_SUCCESS);\r
     Sock->CloseToken = NULL;\r
   }\r
@@ -999,16 +993,14 @@ SockConnClosed (
 \r
 \r
 /**\r
-  Called by low layer protocol to indicate that some\r
-  data is sent or processed. This function trims the\r
-  sent data in the socket send buffer, signals the\r
-  data token if proper\r
+  Called by low layer protocol to indicate that some data is sent or processed.\r
+   \r
+  This function trims the sent data in the socket send buffer, signals the data \r
+  token if proper.\r
 \r
   @param  Sock                  Pointer to the socket.\r
   @param  Count                 The length of the data processed or sent, in bytes.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockDataSent (\r
@@ -1019,7 +1011,7 @@ SockDataSent (
   SOCK_TOKEN            *SockToken;\r
   SOCK_COMPLETION_TOKEN *SndToken;\r
 \r
-  ASSERT (!NetListIsEmpty (&Sock->ProcessingSndTokenList));\r
+  ASSERT (!IsListEmpty (&Sock->ProcessingSndTokenList));\r
   ASSERT (Count <= (Sock->SndBuffer.DataQueue)->BufSize);\r
 \r
   NetbufQueTrim (Sock->SndBuffer.DataQueue, Count);\r
@@ -1038,10 +1030,10 @@ SockDataSent (
 \r
     if (SockToken->RemainDataLen <= Count) {\r
 \r
-      NetListRemoveEntry (&(SockToken->TokenList));\r
+      RemoveEntryList (&(SockToken->TokenList));\r
       SIGNAL_TOKEN (SndToken, EFI_SUCCESS);\r
       Count -= SockToken->RemainDataLen;\r
-      NetFreePool (SockToken);\r
+      FreePool (SockToken);\r
     } else {\r
 \r
       SockToken->RemainDataLen -= Count;\r
@@ -1079,7 +1071,7 @@ SockGetDataToSend (
   IN UINT8       *Dest\r
   )\r
 {\r
-  ASSERT (Sock && SOCK_STREAM == Sock->Type);\r
+  ASSERT ((Sock != NULL) && SockStream == Sock->Type);\r
 \r
   return NetbufQueCopy (\r
           Sock->SndBuffer.DataQueue,\r
@@ -1091,27 +1083,25 @@ SockGetDataToSend (
 \r
 \r
 /**\r
-  Called by the low layer protocol to deliver received data\r
-  to socket layer. This function will append the data to the\r
-  socket receive buffer, set ther urgent data length and then\r
-  check if any receive token can be signaled.\r
+  Called by the low layer protocol to deliver received data to socket layer.\r
+  \r
+  This function will append the data to the socket receive buffer, set ther \r
+  urgent data length and then check if any receive token can be signaled.\r
 \r
   @param  Sock                  Pointer to the socket.\r
   @param  NetBuffer             Pointer to the buffer that contains the received\r
                                 data.\r
   @param  UrgLen                The length of the urgent data in the received data.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockDataRcvd (\r
-  IN SOCKET    *Sock,\r
-  IN NET_BUF   *NetBuffer,\r
-  IN UINT32    UrgLen\r
+  IN     SOCKET    *Sock,\r
+  IN OUT NET_BUF   *NetBuffer,\r
+  IN     UINT32    UrgLen\r
   )\r
 {\r
-  ASSERT (Sock && Sock->RcvBuffer.DataQueue &&\r
+  ASSERT ((Sock != NULL) && (Sock->RcvBuffer.DataQueue != NULL) &&\r
     UrgLen <= NetBuffer->TotalSize);\r
 \r
   NET_GET_REF (NetBuffer);\r
@@ -1144,7 +1134,7 @@ SockGetFreeSpace (
   UINT32      BufferCC;\r
   SOCK_BUFFER *SockBuffer;\r
 \r
-  ASSERT (Sock && ((SOCK_SND_BUF == Which) || (SOCK_RCV_BUF == Which)));\r
+  ASSERT ((Sock != NULL) && ((SOCK_SND_BUF == Which) || (SOCK_RCV_BUF == Which)));\r
 \r
   if (SOCK_SND_BUF == Which) {\r
     SockBuffer = &(Sock->SndBuffer);\r
@@ -1170,18 +1160,16 @@ SockGetFreeSpace (
   @param  Sock                  Pointer to the socket.\r
   @param  Error                 The error code received.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockRcvdErr (\r
-  IN SOCKET       *Sock,\r
-  IN EFI_STATUS   Error\r
+  IN OUT SOCKET       *Sock,\r
+  IN     EFI_STATUS   Error\r
   )\r
 {\r
   SOCK_TOKEN  *SockToken;\r
 \r
-  if (!NetListIsEmpty (&Sock->RcvTokenList)) {\r
+  if (!IsListEmpty (&Sock->RcvTokenList)) {\r
 \r
     SockToken = NET_LIST_HEAD (\r
                   &Sock->RcvTokenList,\r
@@ -1189,11 +1177,11 @@ SockRcvdErr (
                   TokenList\r
                   );\r
 \r
-    NetListRemoveEntry (&SockToken->TokenList);\r
+    RemoveEntryList (&SockToken->TokenList);\r
 \r
     SIGNAL_TOKEN (SockToken->Token, Error);\r
 \r
-    NetFreePool (SockToken);\r
+    FreePool (SockToken);\r
   } else {\r
 \r
     SOCK_ERROR (Sock, Error);\r
@@ -1202,27 +1190,25 @@ SockRcvdErr (
 \r
 \r
 /**\r
-  Called by the low layer protocol to indicate that there\r
-  will be no more data from the communication peer. This\r
-  function set the socket's state to SO_NO_MORE_DATA and\r
-  signal all queued IO tokens with the error status\r
-  EFI_CONNECTION_FIN.\r
+  Called by the low layer protocol to indicate that there will be no more data \r
+  from the communication peer.\r
+  \r
+  This function set the socket's state to SO_NO_MORE_DATA and signal all queued \r
+  IO tokens with the error status EFI_CONNECTION_FIN.\r
 \r
   @param  Sock                  Pointer to the socket.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 SockNoMoreData (\r
-  IN SOCKET *Sock\r
+  IN OUT SOCKET *Sock\r
   )\r
 {\r
   EFI_STATUS  Err;\r
 \r
   SOCK_NO_MORE_DATA (Sock);\r
 \r
-  if (!NetListIsEmpty (&Sock->RcvTokenList)) {\r
+  if (!IsListEmpty (&Sock->RcvTokenList)) {\r
 \r
     ASSERT (0 == GET_RCV_DATASIZE (Sock));\r
 \r
@@ -1252,11 +1238,11 @@ SockBufFirst (
   IN SOCK_BUFFER *Sockbuf\r
   )\r
 {\r
-  NET_LIST_ENTRY  *NetbufList;\r
+  LIST_ENTRY      *NetbufList;\r
 \r
   NetbufList = &(Sockbuf->DataQueue->BufList);\r
 \r
-  if (NetListIsEmpty (NetbufList)) {\r
+  if (IsListEmpty (NetbufList)) {\r
     return NULL;\r
   }\r
 \r
@@ -1271,7 +1257,8 @@ SockBufFirst (
   @param  SockEntry             Pointer to the buffer block prior to the required\r
                                 one.\r
 \r
-  @return Pointer to the buffer block next to SockEntry. NULL if SockEntry is the tail or head entry.\r
+  @return Pointer to the buffer block next to SockEntry. NULL if SockEntry is \r
+          the tail or head entry.\r
 \r
 **/\r
 NET_BUF *\r
@@ -1280,14 +1267,13 @@ SockBufNext (
   IN NET_BUF     *SockEntry\r
   )\r
 {\r
-  NET_LIST_ENTRY  *NetbufList;\r
+  LIST_ENTRY      *NetbufList;\r
 \r
   NetbufList = &(Sockbuf->DataQueue->BufList);\r
 \r
   if ((SockEntry->List.ForwardLink == NetbufList) ||\r
       (SockEntry->List.BackLink == &SockEntry->List) ||\r
-      (SockEntry->List.ForwardLink == &SockEntry->List)\r
-      ) {\r
+      (SockEntry->List.ForwardLink == &SockEntry->List)) {\r
 \r
     return NULL;\r
   }\r