]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c
Code clean up in NetLib:
[mirror_edk2.git] / MdeModulePkg / Universal / Network / Tcp4Dxe / Tcp4Input.c
index a054a4a56d03528144ac0a0fdf50805f475d6c12..9bea6b298b5ab5f1f7dbf3817efa06d6df033082 100644 (file)
@@ -1,36 +1,29 @@
 /** @file\r
+  TCP input process routines.\r
 \r
-Copyright (c) 2005 - 2007, Intel Corporation\r
+Copyright (c) 2005 - 2009, Intel Corporation<BR>\r
 All rights reserved. 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
-  Tcp4Input.c\r
-\r
-Abstract:\r
-\r
-  TCP input process routines.\r
-\r
-\r
 **/\r
 \r
+\r
 #include "Tcp4Main.h"\r
 \r
 \r
 /**\r
-  Check whether the sequence number of the incoming segment\r
-  is acceptable.\r
+  Check whether the sequence number of the incoming segment is acceptable.\r
 \r
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Seg      Pointer to the incoming segment.\r
 \r
-  @return 1 if the sequence number is acceptable, otherwise 0.\r
+  @retval 1       The sequence number is acceptable.\r
+  @retval 0       The sequence number is not acceptable.\r
 \r
 **/\r
 INTN\r
@@ -50,13 +43,11 @@ TcpSeqAcceptable (
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Seg      Segment that triggers the fast recovery.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpFastRecover (\r
-  IN TCP_CB  *Tcb,\r
-  IN TCP_SEG *Seg\r
+  IN OUT TCP_CB  *Tcb,\r
+  IN     TCP_SEG *Seg\r
   )\r
 {\r
   UINT32  FlightSize;\r
@@ -85,7 +76,7 @@ TcpFastRecover (
     Tcb->CWnd = Tcb->Ssthresh + 3 * Tcb->SndMss;\r
 \r
     DEBUG ((EFI_D_INFO, "TcpFastRecover: enter fast retransmission"\r
-      " for TCB %x, recover point is %d\n", Tcb, Tcb->Recover));\r
+      " for TCB %p, recover point is %d\n", Tcb, Tcb->Recover));\r
     return;\r
   }\r
 \r
@@ -104,7 +95,7 @@ TcpFastRecover (
     //\r
     Tcb->CWnd += Tcb->SndMss;\r
     DEBUG ((EFI_D_INFO, "TcpFastRecover: received another"\r
-      " duplicated ACK (%d) for TCB %x\n", Seg->Ack, Tcb));\r
+      " duplicated ACK (%d) for TCB %p\n", Seg->Ack, Tcb));\r
 \r
   } else {\r
 \r
@@ -124,7 +115,7 @@ TcpFastRecover (
 \r
       Tcb->CongestState = TCP_CONGEST_OPEN;\r
       DEBUG ((EFI_D_INFO, "TcpFastRecover: received a full ACK(%d)"\r
-        " for TCB %x, exit fast recovery\n", Seg->Ack, Tcb));\r
+        " for TCB %p, exit fast recovery\n", Seg->Ack, Tcb));\r
 \r
     } else {\r
 \r
@@ -149,7 +140,7 @@ TcpFastRecover (
       Tcb->CWnd -= Acked;\r
 \r
       DEBUG ((EFI_D_INFO, "TcpFastRecover: received a partial"\r
-        " ACK(%d) for TCB %x\n", Seg->Ack, Tcb));\r
+        " ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
 \r
     }\r
   }\r
@@ -162,13 +153,11 @@ TcpFastRecover (
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Seg      Segment that triggers the fast loss recovery.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpFastLossRecover (\r
-  IN TCP_CB  *Tcb,\r
-  IN TCP_SEG *Seg\r
+  IN OUT TCP_CB  *Tcb,\r
+  IN     TCP_SEG *Seg\r
   )\r
 {\r
   if (TCP_SEQ_GT (Seg->Ack, Tcb->SndUna)) {\r
@@ -186,7 +175,7 @@ TcpFastLossRecover (
       Tcb->CongestState = TCP_CONGEST_OPEN;\r
 \r
       DEBUG ((EFI_D_INFO, "TcpFastLossRecover: received a "\r
-        "full ACK(%d) for TCB %x\n", Seg->Ack, Tcb));\r
+        "full ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
 \r
     } else {\r
 \r
@@ -196,25 +185,23 @@ TcpFastLossRecover (
       //\r
       TcpRetransmit (Tcb, Seg->Ack);\r
       DEBUG ((EFI_D_INFO, "TcpFastLossRecover: received a "\r
-        "partial ACK(%d) for TCB %x\n", Seg->Ack, Tcb));\r
+        "partial ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
     }\r
   }\r
 }\r
 \r
 \r
 /**\r
-  Compute the RTT as specified in RFC2988\r
+  Compute the RTT as specified in RFC2988.\r
 \r
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Measure  Currently measured RTT in heart beats.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpComputeRtt (\r
-  IN TCP_CB *Tcb,\r
-  IN UINT32 Measure\r
+  IN OUT TCP_CB *Tcb,\r
+  IN     UINT32 Measure\r
   )\r
 {\r
   INT32 Var;\r
@@ -256,7 +243,7 @@ TcpComputeRtt (
 \r
   }\r
 \r
-  DEBUG ((EFI_D_INFO, "TcpComputeRtt: new RTT for TCB %x"\r
+  DEBUG ((EFI_D_INFO, "TcpComputeRtt: new RTT for TCB %p"\r
     " computed SRTT: %d RTTVAR: %d RTO: %d\n",\r
     Tcb, Tcb->SRtt, Tcb->RttVar, Tcb->Rto));\r
 \r
@@ -264,18 +251,14 @@ TcpComputeRtt (
 \r
 \r
 /**\r
-  Trim the data, SYN and FIN to fit into the window defined by\r
-  Left and Right.\r
+  Trim the data, SYN and FIN to fit into the window defined by Left and Right.\r
 \r
   @param  Nbuf     Buffer that contains received TCP segment without IP header.\r
   @param  Left     The sequence number of the window's left edge.\r
   @param  Right    The sequence number of the window's right edge.\r
 \r
-  @return 0, the data is successfully trimmed.\r
-\r
 **/\r
-STATIC\r
-INTN\r
+VOID\r
 TcpTrimSegment (\r
   IN NET_BUF   *Nbuf,\r
   IN TCP_SEQNO Left,\r
@@ -299,7 +282,7 @@ TcpTrimSegment (
 \r
     Seg->Seq = Seg->End;\r
     NetbufTrim (Nbuf, Nbuf->TotalSize, NET_BUF_HEAD);\r
-    return 0;\r
+    return;\r
   }\r
 \r
   //\r
@@ -329,7 +312,7 @@ TcpTrimSegment (
       }\r
     }\r
 \r
-    if (Drop) {\r
+    if (Drop != 0) {\r
       NetbufTrim (Nbuf, Drop, NET_BUF_HEAD);\r
     }\r
   }\r
@@ -347,13 +330,12 @@ TcpTrimSegment (
       Drop--;\r
     }\r
 \r
-    if (Drop) {\r
+    if (Drop != 0) {\r
       NetbufTrim (Nbuf, Drop, NET_BUF_TAIL);\r
     }\r
   }\r
 \r
-  ASSERT (TcpVerifySegment (Nbuf));\r
-  return 0;\r
+  ASSERT (TcpVerifySegment (Nbuf) != 0);\r
 }\r
 \r
 \r
@@ -363,16 +345,14 @@ TcpTrimSegment (
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Nbuf     Pointer to the NET_BUF containing the received tcp segment.\r
 \r
-  @return 0, the data is trimmed.\r
-\r
 **/\r
-INTN\r
+VOID\r
 TcpTrimInWnd (\r
   IN TCP_CB  *Tcb,\r
   IN NET_BUF *Nbuf\r
   )\r
 {\r
-  return TcpTrimSegment (Nbuf, Tcb->RcvNxt, Tcb->RcvWl2 + Tcb->RcvWnd);\r
+  TcpTrimSegment (Nbuf, Tcb->RcvNxt, Tcb->RcvWl2 + Tcb->RcvWnd);\r
 }\r
 \r
 \r
@@ -389,7 +369,7 @@ TcpTrimInWnd (
 **/\r
 INTN\r
 TcpDeliverData (\r
-  IN TCP_CB *Tcb\r
+  IN OUT TCP_CB *Tcb\r
   )\r
 {\r
   LIST_ENTRY      *Entry;\r
@@ -398,7 +378,7 @@ TcpDeliverData (
   TCP_SEG         *Seg;\r
   UINT32          Urgent;\r
 \r
-  ASSERT (Tcb && Tcb->Sk);\r
+  ASSERT ((Tcb != NULL) && (Tcb->Sk != NULL));\r
 \r
   //\r
   // make sure there is some data queued,\r
@@ -419,7 +399,7 @@ TcpDeliverData (
     Nbuf  = NET_LIST_USER_STRUCT (Entry, NET_BUF, List);\r
     Seg   = TCPSEG_NETBUF (Nbuf);\r
 \r
-    ASSERT (TcpVerifySegment (Nbuf));\r
+    ASSERT (TcpVerifySegment (Nbuf) != 0);\r
     ASSERT (Nbuf->Tcp == NULL);\r
 \r
     if (TCP_SEQ_GT (Seg->Seq, Seq)) {\r
@@ -443,14 +423,14 @@ TcpDeliverData (
       //\r
       if (!IsListEmpty (&Tcb->RcvQue)) {\r
         DEBUG ((EFI_D_ERROR, "TcpDeliverData: data received after"\r
-          " FIN from peer of TCB %x, reset connection\n", Tcb));\r
+          " FIN from peer of TCB %p, reset connection\n", Tcb));\r
 \r
         NetbufFree (Nbuf);\r
         return -1;\r
       }\r
 \r
       DEBUG ((EFI_D_INFO, "TcpDeliverData: processing FIN "\r
-        "from peer of TCB %x\n", Tcb));\r
+        "from peer of TCB %p\n", Tcb));\r
 \r
       switch (Tcb->State) {\r
       case TCP_SYN_RCVD:\r
@@ -481,7 +461,7 @@ TcpDeliverData (
         } else {\r
 \r
           DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
-            "because app disables TIME_WAIT timer for %x\n", Tcb));\r
+            "because app disables TIME_WAIT timer for %p\n", Tcb));\r
 \r
           TcpSendAck (Tcb);\r
           TcpClose (Tcb);\r
@@ -499,6 +479,8 @@ TcpDeliverData (
         NetbufFree (Nbuf);\r
         return -1;\r
         break;\r
+      default:\r
+        break;\r
       }\r
 \r
       TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
@@ -514,7 +496,7 @@ TcpDeliverData (
       TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
     }\r
 \r
-    if (Nbuf->TotalSize) {\r
+    if (Nbuf->TotalSize != 0) {\r
       Urgent = 0;\r
 \r
       if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_RCVD_URG) &&\r
@@ -548,13 +530,11 @@ TcpDeliverData (
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Nbuf     Pointer to the buffer containing the data to be queued.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpQueueData (\r
-  IN TCP_CB  *Tcb,\r
-  IN NET_BUF *Nbuf\r
+  IN OUT TCP_CB  *Tcb,\r
+  IN     NET_BUF *Nbuf\r
   )\r
 {\r
   TCP_SEG         *Seg;\r
@@ -563,7 +543,7 @@ TcpQueueData (
   LIST_ENTRY      *Cur;\r
   NET_BUF         *Node;\r
 \r
-  ASSERT (Tcb && Nbuf && (Nbuf->Tcp == NULL));\r
+  ASSERT ((Tcb != NULL) && (Nbuf != NULL) && (Nbuf->Tcp == NULL));\r
 \r
   NET_GET_REF (Nbuf);\r
 \r
@@ -656,8 +636,6 @@ TcpQueueData (
   @param  Tcb      Pointer to the TCP_CB of this TCP instance.\r
   @param  Ack      The acknowledge seuqence number of the received segment.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpAdjustSndQue (\r
@@ -736,7 +714,7 @@ TcpInput (
   Len     = Nbuf->TotalSize - (Head->HeadLen << 2);\r
 \r
   if ((Head->HeadLen < 5) || (Len < 0) ||\r
-      TcpChecksum (Nbuf, NetPseudoHeadChecksum (Src, Dst, 6, 0))) {\r
+      (TcpChecksum (Nbuf, NetPseudoHeadChecksum (Src, Dst, 6, 0)) != 0)) {\r
 \r
     DEBUG ((EFI_D_INFO, "TcpInput: received an mal-formated packet\n"));\r
     goto DISCARD;\r
@@ -773,7 +751,7 @@ TcpInput (
   //\r
   if (TcpParseOption (Nbuf->Tcp, &Option) == -1) {\r
     DEBUG ((EFI_D_ERROR, "TcpInput: reset the peer because"\r
-      " of mal-format option for Tcb %x\n", Tcb));\r
+      " of mal-format option for Tcb %p\n", Tcb));\r
 \r
     goto SEND_RESET;\r
   }\r
@@ -784,10 +762,6 @@ TcpInput (
   NetbufTrim (Nbuf, (Head->HeadLen << 2), NET_BUF_HEAD);\r
   Nbuf->Tcp = NULL;\r
 \r
-  //\r
-  // TODO: add fast path process here\r
-  //\r
-\r
   //\r
   // Process the segment in LISTEN state.\r
   //\r
@@ -797,7 +771,7 @@ TcpInput (
     //\r
     if (TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
       DEBUG ((EFI_D_WARN, "TcpInput: discard a reset segment "\r
-        "for TCB %x in listening\n", Tcb));\r
+        "for TCB %p in listening\n", Tcb));\r
 \r
       goto DISCARD;\r
     }\r
@@ -808,7 +782,7 @@ TcpInput (
     //\r
     if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
       DEBUG ((EFI_D_WARN, "TcpInput: send reset because of"\r
-        " segment with ACK for TCB %x in listening\n", Tcb));\r
+        " segment with ACK for TCB %p in listening\n", Tcb));\r
 \r
       goto SEND_RESET;\r
     }\r
@@ -825,12 +799,12 @@ TcpInput (
       Tcb     = TcpCloneTcb (Parent);\r
       if (Tcb == NULL) {\r
         DEBUG ((EFI_D_ERROR, "TcpInput: discard a segment because"\r
-          "failed to clone a child for TCB%x\n", Tcb));\r
+          " failed to clone a child for TCB%x\n", Tcb));\r
 \r
         goto DISCARD;\r
       }\r
 \r
-      DEBUG ((EFI_D_INFO, "TcpInput: create a child for TCB %x"\r
+      DEBUG ((EFI_D_INFO, "TcpInput: create a child for TCB %p"\r
         " in listening\n", Tcb));\r
 \r
       //\r
@@ -860,7 +834,7 @@ TcpInput (
     if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK) && (Seg->Ack != Tcb->Iss + 1)) {\r
 \r
       DEBUG ((EFI_D_WARN, "TcpInput: send reset because of "\r
-        "wrong ACK received for TCB %x in SYN_SENT\n", Tcb));\r
+        "wrong ACK received for TCB %p in SYN_SENT\n", Tcb));\r
 \r
       goto SEND_RESET;\r
     }\r
@@ -873,14 +847,14 @@ TcpInput (
       if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
 \r
         DEBUG ((EFI_D_WARN, "TcpInput: connection reset by"\r
-          " peer for TCB%x in SYN_SENT\n", Tcb));\r
+          " peer for TCB %p in SYN_SENT\n", Tcb));\r
 \r
         SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
         goto DROP_CONNECTION;\r
       } else {\r
 \r
         DEBUG ((EFI_D_WARN, "TcpInput: discard a reset segment "\r
-          "because of no ACK for TCB%x in SYN_SENT\n", Tcb));\r
+          "because of no ACK for TCB %p in SYN_SENT\n", Tcb));\r
 \r
         goto DISCARD;\r
       }\r
@@ -923,7 +897,7 @@ TcpInput (
         TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
 \r
         DEBUG ((EFI_D_INFO, "TcpInput: connection established"\r
-          " for TCB %x in SYN_SENT\n", Tcb));\r
+          " for TCB %p in SYN_SENT\n", Tcb));\r
 \r
         goto StepSix;\r
       } else {\r
@@ -938,7 +912,7 @@ TcpInput (
         TcpTrimInWnd (Tcb, Nbuf);\r
 \r
         DEBUG ((EFI_D_WARN, "TcpInput: simultanous open "\r
-          "for TCB %x in SYN_SENT\n", Tcb));\r
+          "for TCB %p in SYN_SENT\n", Tcb));\r
 \r
         goto StepSix;\r
       }\r
@@ -954,9 +928,9 @@ TcpInput (
   //\r
   // First step: Check whether SEG.SEQ is acceptable\r
   //\r
-  if (!TcpSeqAcceptable (Tcb, Seg)) {\r
+  if (TcpSeqAcceptable (Tcb, Seg) == 0) {\r
     DEBUG ((EFI_D_WARN, "TcpInput: sequence acceptance"\r
-      " test failed for segment of TCB %x\n", Tcb));\r
+      " test failed for segment of TCB %p\n", Tcb));\r
 \r
     if (!TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
       TcpSendAck (Tcb);\r
@@ -977,7 +951,7 @@ TcpInput (
   //\r
   if (TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
 \r
-    DEBUG ((EFI_D_WARN, "TcpInput: connection reset for TCB %x\n", Tcb));\r
+    DEBUG ((EFI_D_WARN, "TcpInput: connection reset for TCB %p\n", Tcb));\r
 \r
     if (Tcb->State == TCP_SYN_RCVD) {\r
 \r
@@ -990,17 +964,14 @@ TcpInput (
       // if it comes from a LISTEN TCB.\r
       //\r
     } else if ((Tcb->State == TCP_ESTABLISHED) ||\r
-             (Tcb->State == TCP_FIN_WAIT_1) ||\r
-             (Tcb->State == TCP_FIN_WAIT_2) ||\r
-             (Tcb->State == TCP_CLOSE_WAIT)\r
-            ) {\r
+               (Tcb->State == TCP_FIN_WAIT_1) ||\r
+               (Tcb->State == TCP_FIN_WAIT_2) ||\r
+               (Tcb->State == TCP_CLOSE_WAIT)) {\r
 \r
       SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
 \r
     } else {\r
-      //\r
-      // TODO: set socket error to CLOSED\r
-      //\r
+\r
     }\r
 \r
     goto DROP_CONNECTION;\r
@@ -1021,7 +992,7 @@ TcpInput (
   if (TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN)) {\r
 \r
     DEBUG ((EFI_D_WARN, "TcpInput: connection reset "\r
-      "because received extra SYN for TCB %x\n", Tcb));\r
+      "because received extra SYN for TCB %p\n", Tcb));\r
 \r
     SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
     goto RESET_THEN_DROP;\r
@@ -1032,7 +1003,7 @@ TcpInput (
   //\r
   if (!TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
     DEBUG ((EFI_D_WARN, "TcpInput: segment discard because"\r
-      " of no ACK for connected TCB %x\n", Tcb));\r
+      " of no ACK for connected TCB %p\n", Tcb));\r
 \r
     goto DISCARD;\r
 \r
@@ -1053,14 +1024,14 @@ TcpInput (
       TcpDeliverData (Tcb);\r
 \r
       DEBUG ((EFI_D_INFO, "TcpInput: connection established "\r
-        " for TCB %x in SYN_RCVD\n", Tcb));\r
+        " for TCB %p in SYN_RCVD\n", Tcb));\r
 \r
       //\r
       // Continue the process as ESTABLISHED state\r
       //\r
     } else {\r
       DEBUG ((EFI_D_WARN, "TcpInput: send reset because of"\r
-        " wrong ACK for TCB %x in SYN_RCVD\n", Tcb));\r
+        " wrong ACK for TCB %p in SYN_RCVD\n", Tcb));\r
 \r
       goto SEND_RESET;\r
     }\r
@@ -1069,14 +1040,14 @@ TcpInput (
   if (TCP_SEQ_LT (Seg->Ack, Tcb->SndUna)) {\r
 \r
     DEBUG ((EFI_D_WARN, "TcpInput: ignore the out-of-data"\r
-      " ACK for connected TCB %x\n", Tcb));\r
+      " ACK for connected TCB %p\n", Tcb));\r
 \r
     goto StepSix;\r
 \r
   } else if (TCP_SEQ_GT (Seg->Ack, Tcb->SndNxt)) {\r
 \r
     DEBUG ((EFI_D_WARN, "TcpInput: discard segment for "\r
-      "future ACK for connected TCB %x\n", Tcb));\r
+      "future ACK for connected TCB %p\n", Tcb));\r
 \r
     TcpSendAck (Tcb);\r
     goto DISCARD;\r
@@ -1163,7 +1134,7 @@ TcpInput (
     Tcb->SndUna = Seg->Ack;\r
 \r
     if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_SND_URG) &&\r
-        (TCP_SEQ_LT (Tcb->SndUp, Seg->Ack))) {\r
+        TCP_SEQ_LT (Tcb->SndUp, Seg->Ack)) {\r
 \r
       TCP_CLEAR_FLG (Tcb->CtrlFlag, TCP_CTRL_SND_URG);\r
     }\r
@@ -1187,7 +1158,7 @@ TcpInput (
       }\r
 \r
       DEBUG ((EFI_D_WARN, "TcpInput: peer shrinks the"\r
-        " window  for connected TCB %x\n", Tcb));\r
+        " window for connected TCB %p\n", Tcb));\r
 \r
       if ((Tcb->CongestState == TCP_CONGEST_RECOVER) &&\r
           (TCP_SEQ_LT (Right, Tcb->Recover))) {\r
@@ -1225,7 +1196,7 @@ NO_UPDATE:
       (Tcb->SndUna == Tcb->SndNxt)) {\r
 \r
     DEBUG ((EFI_D_INFO, "TcpInput: local FIN is ACKed by"\r
-      " peer for connected TCB %x\n", Tcb));\r
+      " peer for connected TCB %p\n", Tcb));\r
 \r
     TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED);\r
   }\r
@@ -1265,7 +1236,7 @@ NO_UPDATE:
       } else {\r
 \r
         DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
-          "because app disables TIME_WAIT timer for %x\n", Tcb));\r
+          "because app disables TIME_WAIT timer for %p\n", Tcb));\r
 \r
         TcpClose (Tcb);\r
       }\r
@@ -1291,11 +1262,14 @@ NO_UPDATE:
     } else {\r
 \r
       DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
-        "because app disables TIME_WAIT timer for %x\n", Tcb));\r
+        "because app disables TIME_WAIT timer for %p\n", Tcb));\r
 \r
       TcpClose (Tcb);\r
     }\r
     break;\r
+\r
+  default:\r
+    break;\r
   }\r
 \r
   //\r
@@ -1316,7 +1290,7 @@ StepSix:
       !TCP_FIN_RCVD (Tcb->State)) {\r
 \r
     DEBUG ((EFI_D_INFO, "TcpInput: received urgent data "\r
-      "from peer for connected TCB %x\n", Tcb));\r
+      "from peer for connected TCB %p\n", Tcb));\r
 \r
     Urg = Seg->Seq + Seg->Urg;\r
 \r
@@ -1339,14 +1313,14 @@ StepSix:
     if (TCP_FIN_RCVD (Tcb->State)) {\r
 \r
       DEBUG ((EFI_D_WARN, "TcpInput: connection reset because"\r
-        " data is lost for connected TCB %x\n", Tcb));\r
+        " data is lost for connected TCB %p\n", Tcb));\r
 \r
       goto RESET_THEN_DROP;\r
     }\r
 \r
     if (TCP_LOCAL_CLOSED (Tcb->State) && (Nbuf->TotalSize != 0)) {\r
       DEBUG ((EFI_D_WARN, "TcpInput: connection reset because"\r
-        " data is lost for connected TCB %x\n", Tcb));\r
+        " data is lost for connected TCB %p\n", Tcb));\r
 \r
       goto RESET_THEN_DROP;\r
     }\r
@@ -1372,14 +1346,14 @@ StepSix:
   // Tcb is a new child of the listening Parent,\r
   // commit it.\r
   //\r
-  if (Parent) {\r
+  if (Parent != NULL) {\r
     Tcb->Parent = Parent;\r
     TcpInsertTcb (Tcb);\r
   }\r
 \r
   if ((Tcb->State != TCP_CLOSED) &&\r
-      (!TcpToSendData (Tcb, 0)) &&\r
-      (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW) || Nbuf->TotalSize)) {\r
+      (TcpToSendData (Tcb, 0) == 0) &&\r
+      (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW) || (Nbuf->TotalSize != 0))) {\r
 \r
     TcpToSendAck (Tcb);\r
   }\r
@@ -1391,7 +1365,7 @@ RESET_THEN_DROP:
   TcpSendReset (Tcb, Head, Len, Dst, Src);\r
 \r
 DROP_CONNECTION:\r
-  ASSERT (Tcb && Tcb->Sk);\r
+  ASSERT ((Tcb != NULL) && (Tcb->Sk != NULL));\r
 \r
   NetbufFree (Nbuf);\r
   TcpClose (Tcb);\r
@@ -1410,9 +1384,9 @@ DISCARD:
   DEBUG ((EFI_D_WARN, "Tcp4Input: Discard a packet\n"));\r
   NetbufFree (Nbuf);\r
 \r
-  if (Parent && Tcb) {\r
+  if ((Parent != NULL) && (Tcb != NULL)) {\r
 \r
-    ASSERT (Tcb->Sk);\r
+    ASSERT (Tcb->Sk != NULL);\r
     TcpClose (Tcb);\r
   }\r
 \r
@@ -1429,13 +1403,11 @@ DISCARD:
   @param  Src      Source address of the ICMP error message.\r
   @param  Dst      Destination address of the ICMP error message.\r
 \r
-  @return None.\r
-\r
 **/\r
 VOID\r
 TcpIcmpInput (\r
   IN NET_BUF     *Nbuf,\r
-  IN ICMP_ERROR  IcmpErr,\r
+  IN UINT8       IcmpErr,\r
   IN UINT32      Src,\r
   IN UINT32      Dst\r
   )\r
@@ -1469,7 +1441,12 @@ TcpIcmpInput (
     goto CLEAN_EXIT;\r
   }\r
 \r
-  IcmpErrStatus = IpIoGetIcmpErrStatus (IcmpErr, &IcmpErrIsHard, &IcmpErrNotify);\r
+  IcmpErrStatus = IpIoGetIcmpErrStatus (\r
+                    IcmpErr,\r
+                    IP_VERSION_4,\r
+                    &IcmpErrIsHard,\r
+                    &IcmpErrNotify\r
+                    );\r
 \r
   if (IcmpErrNotify) {\r
 \r