]> git.proxmox.com Git - mirror_edk2.git/blame - MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c
MdeModulePkg: Fix issue about current Ip4Dxe implementation for DHCP DORA process
[mirror_edk2.git] / MdeModulePkg / Universal / Network / Tcp4Dxe / Tcp4Input.c
CommitLineData
8a67d61d 1/** @file\r
dfc1f033 2 TCP input process routines.\r
8a67d61d 3\r
49789216 4Copyright (c) 2005 - 2015, Intel Corporation. All rights reserved.<BR>\r
e5eed7d3 5This program and the accompanying materials\r
8a67d61d 6are licensed and made available under the terms and conditions of the BSD License\r
7which accompanies this distribution. The full text of the license may be found at\r
dfc1f033 8http://opensource.org/licenses/bsd-license.php<BR>\r
8a67d61d 9\r
10THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12\r
8a67d61d 13**/\r
14\r
dfc1f033 15\r
8a67d61d 16#include "Tcp4Main.h"\r
17\r
18\r
19/**\r
120db52c 20 Check whether the sequence number of the incoming segment is acceptable.\r
8a67d61d 21\r
22 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
23 @param Seg Pointer to the incoming segment.\r
24\r
120db52c 25 @retval 1 The sequence number is acceptable.\r
26 @retval 0 The sequence number is not acceptable.\r
8a67d61d 27\r
28**/\r
29INTN\r
30TcpSeqAcceptable (\r
31 IN TCP_CB *Tcb,\r
32 IN TCP_SEG *Seg\r
33 )\r
34{\r
35 return (TCP_SEQ_LEQ (Tcb->RcvWl2, Seg->End) &&\r
ac8cca2a 36 TCP_SEQ_LT (Seg->Seq, Tcb->RcvWl2 + Tcb->RcvWnd));\r
8a67d61d 37}\r
38\r
39\r
40/**\r
41 NewReno fast recovery, RFC3782.\r
42\r
43 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
44 @param Seg Segment that triggers the fast recovery.\r
45\r
8a67d61d 46**/\r
47VOID\r
48TcpFastRecover (\r
77f00155 49 IN OUT TCP_CB *Tcb,\r
50 IN TCP_SEG *Seg\r
8a67d61d 51 )\r
52{\r
53 UINT32 FlightSize;\r
54 UINT32 Acked;\r
55\r
56 //\r
57 // Step 1: Three duplicate ACKs and not in fast recovery\r
58 //\r
59 if (Tcb->CongestState != TCP_CONGEST_RECOVER) {\r
60\r
61 //\r
62 // Step 1A: Invoking fast retransmission.\r
63 //\r
49789216 64 FlightSize = TCP_SUB_SEQ (Tcb->SndNxt, Tcb->SndUna);\r
8a67d61d 65\r
36ee91ca 66 Tcb->Ssthresh = MAX (FlightSize >> 1, (UINT32) (2 * Tcb->SndMss));\r
8a67d61d 67 Tcb->Recover = Tcb->SndNxt;\r
68\r
69 Tcb->CongestState = TCP_CONGEST_RECOVER;\r
70 TCP_CLEAR_FLG (Tcb->CtrlFlag, TCP_CTRL_RTT_ON);\r
71\r
72 //\r
73 // Step 2: Entering fast retransmission\r
74 //\r
75 TcpRetransmit (Tcb, Tcb->SndUna);\r
76 Tcb->CWnd = Tcb->Ssthresh + 3 * Tcb->SndMss;\r
77\r
e48e37fc 78 DEBUG ((EFI_D_INFO, "TcpFastRecover: enter fast retransmission"\r
0e549d5b 79 " for TCB %p, recover point is %d\n", Tcb, Tcb->Recover));\r
8a67d61d 80 return;\r
81 }\r
82\r
83 //\r
84 // During fast recovery, execute Step 3, 4, 5 of RFC3782\r
85 //\r
86 if (Seg->Ack == Tcb->SndUna) {\r
87\r
88 //\r
89 // Step 3: Fast Recovery,\r
90 // If this is a duplicated ACK, increse Cwnd by SMSS.\r
91 //\r
92\r
93 // Step 4 is skipped here only to be executed later\r
94 // by TcpToSendData\r
95 //\r
96 Tcb->CWnd += Tcb->SndMss;\r
e48e37fc 97 DEBUG ((EFI_D_INFO, "TcpFastRecover: received another"\r
0e549d5b 98 " duplicated ACK (%d) for TCB %p\n", Seg->Ack, Tcb));\r
8a67d61d 99\r
100 } else {\r
101\r
102 //\r
103 // New data is ACKed, check whether it is a\r
104 // full ACK or partial ACK\r
105 //\r
106 if (TCP_SEQ_GEQ (Seg->Ack, Tcb->Recover)) {\r
107\r
108 //\r
109 // Step 5 - Full ACK:\r
110 // deflate the congestion window, and exit fast recovery\r
111 //\r
49789216 112 FlightSize = TCP_SUB_SEQ (Tcb->SndNxt, Tcb->SndUna);\r
8a67d61d 113\r
36ee91ca 114 Tcb->CWnd = MIN (Tcb->Ssthresh, FlightSize + Tcb->SndMss);\r
8a67d61d 115\r
116 Tcb->CongestState = TCP_CONGEST_OPEN;\r
e48e37fc 117 DEBUG ((EFI_D_INFO, "TcpFastRecover: received a full ACK(%d)"\r
0e549d5b 118 " for TCB %p, exit fast recovery\n", Seg->Ack, Tcb));\r
8a67d61d 119\r
120 } else {\r
121\r
122 //\r
123 // Step 5 - Partial ACK:\r
124 // fast retransmit the first unacknowledge field\r
125 // , then deflate the CWnd\r
126 //\r
127 TcpRetransmit (Tcb, Seg->Ack);\r
128 Acked = TCP_SUB_SEQ (Seg->Ack, Tcb->SndUna);\r
129\r
130 //\r
131 // Deflate the CWnd by the amount of new data\r
132 // ACKed by SEG.ACK. If more than one SMSS data\r
133 // is ACKed, add back SMSS byte to CWnd after\r
134 //\r
135 if (Acked >= Tcb->SndMss) {\r
136 Acked -= Tcb->SndMss;\r
137\r
138 }\r
139\r
140 Tcb->CWnd -= Acked;\r
141\r
e48e37fc 142 DEBUG ((EFI_D_INFO, "TcpFastRecover: received a partial"\r
0e549d5b 143 " ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
8a67d61d 144\r
145 }\r
146 }\r
147}\r
148\r
149\r
150/**\r
151 NewReno fast loss recovery, RFC3792.\r
152\r
153 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
154 @param Seg Segment that triggers the fast loss recovery.\r
155\r
8a67d61d 156**/\r
157VOID\r
158TcpFastLossRecover (\r
77f00155 159 IN OUT TCP_CB *Tcb,\r
160 IN TCP_SEG *Seg\r
8a67d61d 161 )\r
162{\r
163 if (TCP_SEQ_GT (Seg->Ack, Tcb->SndUna)) {\r
164\r
165 //\r
166 // New data is ACKed, check whether it is a\r
167 // full ACK or partial ACK\r
168 //\r
169 if (TCP_SEQ_GEQ (Seg->Ack, Tcb->LossRecover)) {\r
170\r
171 //\r
172 // Full ACK: exit the loss recovery.\r
173 //\r
174 Tcb->LossTimes = 0;\r
175 Tcb->CongestState = TCP_CONGEST_OPEN;\r
176\r
e48e37fc 177 DEBUG ((EFI_D_INFO, "TcpFastLossRecover: received a "\r
0e549d5b 178 "full ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
8a67d61d 179\r
180 } else {\r
181\r
182 //\r
183 // Partial ACK:\r
184 // fast retransmit the first unacknowledge field.\r
185 //\r
186 TcpRetransmit (Tcb, Seg->Ack);\r
e48e37fc 187 DEBUG ((EFI_D_INFO, "TcpFastLossRecover: received a "\r
0e549d5b 188 "partial ACK(%d) for TCB %p\n", Seg->Ack, Tcb));\r
8a67d61d 189 }\r
190 }\r
191}\r
192\r
193\r
194/**\r
120db52c 195 Compute the RTT as specified in RFC2988.\r
8a67d61d 196\r
197 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
198 @param Measure Currently measured RTT in heart beats.\r
199\r
8a67d61d 200**/\r
201VOID\r
202TcpComputeRtt (\r
77f00155 203 IN OUT TCP_CB *Tcb,\r
204 IN UINT32 Measure\r
8a67d61d 205 )\r
206{\r
207 INT32 Var;\r
208\r
209 //\r
210 // Step 2.3: Compute the RTO for subsequent RTT measurement.\r
211 //\r
212 if (Tcb->SRtt != 0) {\r
213\r
214 Var = Tcb->SRtt - (Measure << TCP_RTT_SHIFT);\r
215\r
216 if (Var < 0) {\r
217 Var = -Var;\r
218 }\r
219\r
220 Tcb->RttVar = (3 * Tcb->RttVar + Var) >> 2;\r
221 Tcb->SRtt = 7 * (Tcb->SRtt >> 3) + Measure;\r
222\r
223 } else {\r
224 //\r
225 // Step 2.2: compute the first RTT measure\r
226 //\r
227 Tcb->SRtt = Measure << TCP_RTT_SHIFT;\r
228 Tcb->RttVar = Measure << (TCP_RTT_SHIFT - 1);\r
229 }\r
230\r
36ee91ca 231 Tcb->Rto = (Tcb->SRtt + MAX (8, 4 * Tcb->RttVar)) >> TCP_RTT_SHIFT;\r
8a67d61d 232\r
233 //\r
234 // Step 2.4: Limit the RTO to at least 1 second\r
235 // Step 2.5: Limit the RTO to a maxium value that\r
236 // is at least 60 second\r
237 //\r
238 if (Tcb->Rto < TCP_RTO_MIN) {\r
239 Tcb->Rto = TCP_RTO_MIN;\r
240\r
241 } else if (Tcb->Rto > TCP_RTO_MAX) {\r
242 Tcb->Rto = TCP_RTO_MAX;\r
243\r
244 }\r
245\r
0e549d5b 246 DEBUG ((EFI_D_INFO, "TcpComputeRtt: new RTT for TCB %p"\r
8a67d61d 247 " computed SRTT: %d RTTVAR: %d RTO: %d\n",\r
248 Tcb, Tcb->SRtt, Tcb->RttVar, Tcb->Rto));\r
249\r
250}\r
251\r
252\r
253/**\r
120db52c 254 Trim the data, SYN and FIN to fit into the window defined by Left and Right.\r
8a67d61d 255\r
256 @param Nbuf Buffer that contains received TCP segment without IP header.\r
257 @param Left The sequence number of the window's left edge.\r
258 @param Right The sequence number of the window's right edge.\r
259\r
8a67d61d 260**/\r
276dcc1b 261VOID\r
8a67d61d 262TcpTrimSegment (\r
263 IN NET_BUF *Nbuf,\r
264 IN TCP_SEQNO Left,\r
265 IN TCP_SEQNO Right\r
266 )\r
267{\r
268 TCP_SEG *Seg;\r
269 TCP_SEQNO Urg;\r
270 UINT32 Drop;\r
271\r
272 Seg = TCPSEG_NETBUF (Nbuf);\r
273\r
274 //\r
275 // If the segment is completely out of window,\r
276 // truncate every thing, include SYN and FIN.\r
277 //\r
278 if (TCP_SEQ_LEQ (Seg->End, Left) || TCP_SEQ_LEQ (Right, Seg->Seq)) {\r
279\r
280 TCP_CLEAR_FLG (Seg->Flag, TCP_FLG_SYN);\r
281 TCP_CLEAR_FLG (Seg->Flag, TCP_FLG_FIN);\r
282\r
283 Seg->Seq = Seg->End;\r
284 NetbufTrim (Nbuf, Nbuf->TotalSize, NET_BUF_HEAD);\r
c191cdd1 285 return;\r
8a67d61d 286 }\r
287\r
288 //\r
289 // Adjust the buffer header\r
290 //\r
291 if (TCP_SEQ_LT (Seg->Seq, Left)) {\r
292\r
293 Drop = TCP_SUB_SEQ (Left, Seg->Seq);\r
294 Urg = Seg->Seq + Seg->Urg;\r
295 Seg->Seq = Left;\r
296\r
297 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN)) {\r
298 TCP_CLEAR_FLG (Seg->Flag, TCP_FLG_SYN);\r
299 Drop--;\r
300 }\r
301\r
302 //\r
303 // Adjust the urgent point\r
304 //\r
305 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_URG)) {\r
306\r
307 if (TCP_SEQ_LT (Urg, Seg->Seq)) {\r
308\r
309 TCP_CLEAR_FLG (Seg->Flag, TCP_FLG_URG);\r
310 } else {\r
311 Seg->Urg = (UINT16) TCP_SUB_SEQ (Urg, Seg->Seq);\r
312 }\r
313 }\r
314\r
85511ddf 315 if (Drop != 0) {\r
8a67d61d 316 NetbufTrim (Nbuf, Drop, NET_BUF_HEAD);\r
317 }\r
318 }\r
319\r
320 //\r
321 // Adjust the buffer tail\r
322 //\r
323 if (TCP_SEQ_GT (Seg->End, Right)) {\r
324\r
325 Drop = TCP_SUB_SEQ (Seg->End, Right);\r
326 Seg->End = Right;\r
327\r
328 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_FIN)) {\r
329 TCP_CLEAR_FLG (Seg->Flag, TCP_FLG_FIN);\r
330 Drop--;\r
331 }\r
332\r
85511ddf 333 if (Drop != 0) {\r
8a67d61d 334 NetbufTrim (Nbuf, Drop, NET_BUF_TAIL);\r
335 }\r
336 }\r
337\r
120db52c 338 ASSERT (TcpVerifySegment (Nbuf) != 0);\r
8a67d61d 339}\r
340\r
341\r
342/**\r
343 Trim off the data outside the tcb's receive window.\r
344\r
345 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
346 @param Nbuf Pointer to the NET_BUF containing the received tcp segment.\r
347\r
8a67d61d 348**/\r
276dcc1b 349VOID\r
8a67d61d 350TcpTrimInWnd (\r
351 IN TCP_CB *Tcb,\r
352 IN NET_BUF *Nbuf\r
353 )\r
354{\r
276dcc1b 355 TcpTrimSegment (Nbuf, Tcb->RcvNxt, Tcb->RcvWl2 + Tcb->RcvWnd);\r
8a67d61d 356}\r
357\r
358\r
359/**\r
360 Process the data and FIN flag, check whether to deliver\r
361 data to the socket layer.\r
362\r
363 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
364\r
365 @retval 0 No error occurred to deliver data.\r
366 @retval -1 Error condition occurred. Proper response is to reset the\r
367 connection.\r
368\r
369**/\r
370INTN\r
371TcpDeliverData (\r
77f00155 372 IN OUT TCP_CB *Tcb\r
8a67d61d 373 )\r
374{\r
e48e37fc 375 LIST_ENTRY *Entry;\r
8a67d61d 376 NET_BUF *Nbuf;\r
377 TCP_SEQNO Seq;\r
378 TCP_SEG *Seg;\r
379 UINT32 Urgent;\r
380\r
120db52c 381 ASSERT ((Tcb != NULL) && (Tcb->Sk != NULL));\r
8a67d61d 382\r
383 //\r
384 // make sure there is some data queued,\r
385 // and TCP is in a proper state\r
386 //\r
e48e37fc 387 if (IsListEmpty (&Tcb->RcvQue) || !TCP_CONNECTED (Tcb->State)) {\r
8a67d61d 388\r
389 return 0;\r
390 }\r
391\r
392 //\r
393 // Deliver data to the socket layer\r
394 //\r
395 Entry = Tcb->RcvQue.ForwardLink;\r
396 Seq = Tcb->RcvNxt;\r
397\r
398 while (Entry != &Tcb->RcvQue) {\r
399 Nbuf = NET_LIST_USER_STRUCT (Entry, NET_BUF, List);\r
400 Seg = TCPSEG_NETBUF (Nbuf);\r
401\r
120db52c 402 ASSERT (TcpVerifySegment (Nbuf) != 0);\r
8a67d61d 403 ASSERT (Nbuf->Tcp == NULL);\r
404\r
405 if (TCP_SEQ_GT (Seg->Seq, Seq)) {\r
406 break;\r
407 }\r
408\r
409 Entry = Entry->ForwardLink;\r
410 Seq = Seg->End;\r
411 Tcb->RcvNxt = Seq;\r
412\r
e48e37fc 413 RemoveEntryList (&Nbuf->List);\r
8a67d61d 414\r
415 //\r
416 // RFC793 Eighth step: process FIN in sequence\r
417 //\r
418 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_FIN)) {\r
419\r
420 //\r
421 // The peer sends to us junky data after FIN,\r
422 // reset the connection.\r
423 //\r
e48e37fc 424 if (!IsListEmpty (&Tcb->RcvQue)) {\r
425 DEBUG ((EFI_D_ERROR, "TcpDeliverData: data received after"\r
0e549d5b 426 " FIN from peer of TCB %p, reset connection\n", Tcb));\r
8a67d61d 427\r
428 NetbufFree (Nbuf);\r
429 return -1;\r
430 }\r
431\r
e48e37fc 432 DEBUG ((EFI_D_INFO, "TcpDeliverData: processing FIN "\r
0e549d5b 433 "from peer of TCB %p\n", Tcb));\r
8a67d61d 434\r
435 switch (Tcb->State) {\r
436 case TCP_SYN_RCVD:\r
437 case TCP_ESTABLISHED:\r
438\r
439 TcpSetState (Tcb, TCP_CLOSE_WAIT);\r
440 break;\r
441\r
442 case TCP_FIN_WAIT_1:\r
443\r
444 if (!TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED)) {\r
445\r
446 TcpSetState (Tcb, TCP_CLOSING);\r
447 break;\r
448 }\r
449\r
450 //\r
451 // fall through\r
452 //\r
453 case TCP_FIN_WAIT_2:\r
454\r
455 TcpSetState (Tcb, TCP_TIME_WAIT);\r
456 TcpClearAllTimer (Tcb);\r
457\r
458 if (Tcb->TimeWaitTimeout != 0) {\r
459\r
460 TcpSetTimer (Tcb, TCP_TIMER_2MSL, Tcb->TimeWaitTimeout);\r
461 } else {\r
462\r
e48e37fc 463 DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
0e549d5b 464 "because app disables TIME_WAIT timer for %p\n", Tcb));\r
8a67d61d 465\r
466 TcpSendAck (Tcb);\r
467 TcpClose (Tcb);\r
468 }\r
469 break;\r
470\r
471 case TCP_CLOSE_WAIT:\r
472 case TCP_CLOSING:\r
473 case TCP_LAST_ACK:\r
474 case TCP_TIME_WAIT:\r
475 //\r
476 // The peer sends to us junk FIN byte. Discard\r
477 // the buffer then reset the connection\r
478 //\r
479 NetbufFree (Nbuf);\r
480 return -1;\r
dfa596b8 481 default:\r
dfc1f033 482 break;\r
8a67d61d 483 }\r
484\r
485 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
486\r
487 Seg->End--;\r
488 }\r
489\r
490 //\r
491 // Don't delay the ack if PUSH flag is on.\r
492 //\r
493 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_PSH)) {\r
494\r
495 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
496 }\r
497\r
52b2a308 498 if (Nbuf->TotalSize != 0) {\r
8a67d61d 499 Urgent = 0;\r
500\r
501 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_RCVD_URG) &&\r
502 TCP_SEQ_LEQ (Seg->Seq, Tcb->RcvUp)) {\r
503\r
504 if (TCP_SEQ_LEQ (Seg->End, Tcb->RcvUp)) {\r
505 Urgent = Nbuf->TotalSize;\r
506 } else {\r
507 Urgent = TCP_SUB_SEQ (Tcb->RcvUp, Seg->Seq) + 1;\r
508 }\r
509 }\r
510\r
511 SockDataRcvd (Tcb->Sk, Nbuf, Urgent);\r
512 }\r
513\r
514 if (TCP_FIN_RCVD (Tcb->State)) {\r
515\r
516 SockNoMoreData (Tcb->Sk);\r
517 }\r
518\r
519 NetbufFree (Nbuf);\r
520 }\r
521\r
522 return 0;\r
523}\r
524\r
525\r
526/**\r
527 Store the data into the reassemble queue.\r
528\r
529 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
530 @param Nbuf Pointer to the buffer containing the data to be queued.\r
531\r
8a67d61d 532**/\r
533VOID\r
534TcpQueueData (\r
77f00155 535 IN OUT TCP_CB *Tcb,\r
536 IN NET_BUF *Nbuf\r
8a67d61d 537 )\r
538{\r
539 TCP_SEG *Seg;\r
e48e37fc 540 LIST_ENTRY *Head;\r
541 LIST_ENTRY *Prev;\r
542 LIST_ENTRY *Cur;\r
8a67d61d 543 NET_BUF *Node;\r
544\r
120db52c 545 ASSERT ((Tcb != NULL) && (Nbuf != NULL) && (Nbuf->Tcp == NULL));\r
8a67d61d 546\r
547 NET_GET_REF (Nbuf);\r
548\r
549 Seg = TCPSEG_NETBUF (Nbuf);\r
550 Head = &Tcb->RcvQue;\r
551\r
552 //\r
553 // Fast path to process normal case. That is,\r
554 // no out-of-order segments are received.\r
555 //\r
e48e37fc 556 if (IsListEmpty (Head)) {\r
8a67d61d 557\r
e48e37fc 558 InsertTailList (Head, &Nbuf->List);\r
49789216 559 return;\r
8a67d61d 560 }\r
561\r
562 //\r
563 // Find the point to insert the buffer\r
564 //\r
565 for (Prev = Head, Cur = Head->ForwardLink;\r
49789216
BZ
566 Cur != Head;\r
567 Prev = Cur, Cur = Cur->ForwardLink) {\r
8a67d61d 568\r
569 Node = NET_LIST_USER_STRUCT (Cur, NET_BUF, List);\r
570\r
571 if (TCP_SEQ_LT (Seg->Seq, TCPSEG_NETBUF (Node)->Seq)) {\r
572 break;\r
573 }\r
574 }\r
575\r
576 //\r
577 // Check whether the current segment overlaps with the\r
578 // previous segment.\r
579 //\r
580 if (Prev != Head) {\r
581 Node = NET_LIST_USER_STRUCT (Prev, NET_BUF, List);\r
582\r
583 if (TCP_SEQ_LT (Seg->Seq, TCPSEG_NETBUF (Node)->End)) {\r
584\r
585 if (TCP_SEQ_LEQ (Seg->End, TCPSEG_NETBUF (Node)->End)) {\r
586\r
587 NetbufFree (Nbuf);\r
49789216 588 return;\r
8a67d61d 589 }\r
590\r
591 TcpTrimSegment (Nbuf, TCPSEG_NETBUF (Node)->End, Seg->End);\r
592 }\r
593 }\r
594\r
e48e37fc 595 InsertHeadList (Prev, &Nbuf->List);\r
8a67d61d 596\r
597 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
598\r
599 //\r
600 // Check the segments after the insert point.\r
601 //\r
602 while (Cur != Head) {\r
603 Node = NET_LIST_USER_STRUCT (Cur, NET_BUF, List);\r
604\r
605 if (TCP_SEQ_LEQ (TCPSEG_NETBUF (Node)->End, Seg->End)) {\r
606\r
607 Cur = Cur->ForwardLink;\r
608\r
e48e37fc 609 RemoveEntryList (&Node->List);\r
8a67d61d 610 NetbufFree (Node);\r
611 continue;\r
612 }\r
613\r
614 if (TCP_SEQ_LT (TCPSEG_NETBUF (Node)->Seq, Seg->End)) {\r
615\r
616 if (TCP_SEQ_LEQ (TCPSEG_NETBUF (Node)->Seq, Seg->Seq)) {\r
617\r
e48e37fc 618 RemoveEntryList (&Nbuf->List);\r
8a67d61d 619 NetbufFree (Nbuf);\r
620 return ;\r
621 }\r
622\r
623 TcpTrimSegment (Nbuf, Seg->Seq, TCPSEG_NETBUF (Node)->Seq);\r
624 break;\r
625 }\r
626\r
627 Cur = Cur->ForwardLink;\r
628 }\r
629}\r
630\r
631\r
632/**\r
633 Ajust the send queue or the retransmit queue.\r
634\r
635 @param Tcb Pointer to the TCP_CB of this TCP instance.\r
636 @param Ack The acknowledge seuqence number of the received segment.\r
637\r
8a67d61d 638**/\r
639VOID\r
640TcpAdjustSndQue (\r
641 IN TCP_CB *Tcb,\r
642 IN TCP_SEQNO Ack\r
643 )\r
644{\r
e48e37fc 645 LIST_ENTRY *Head;\r
646 LIST_ENTRY *Cur;\r
8a67d61d 647 NET_BUF *Node;\r
648 TCP_SEG *Seg;\r
649\r
650 Head = &Tcb->SndQue;\r
651 Cur = Head->ForwardLink;\r
652\r
653 while (Cur != Head) {\r
654 Node = NET_LIST_USER_STRUCT (Cur, NET_BUF, List);\r
655 Seg = TCPSEG_NETBUF (Node);\r
656\r
657 if (TCP_SEQ_GEQ (Seg->Seq, Ack)) {\r
658 break;\r
659 }\r
660\r
661 //\r
662 // Remove completely ACKed segments\r
663 //\r
664 if (TCP_SEQ_LEQ (Seg->End, Ack)) {\r
665 Cur = Cur->ForwardLink;\r
666\r
e48e37fc 667 RemoveEntryList (&Node->List);\r
8a67d61d 668 NetbufFree (Node);\r
669 continue;\r
670 }\r
671\r
672 TcpTrimSegment (Node, Ack, Seg->End);\r
673 break;\r
674 }\r
675}\r
676\r
677\r
678/**\r
679 Process the received TCP segments.\r
680\r
681 @param Nbuf Buffer that contains received TCP segment without IP header.\r
682 @param Src Source address of the segment, or the peer's IP address.\r
683 @param Dst Destination address of the segment, or the local end's IP\r
684 address.\r
685\r
686 @retval 0 Segment is processed successfully. It is either accepted or\r
687 discarded. But no connection is reset by the segment.\r
688 @retval -1 A connection is reset by the segment.\r
689\r
690**/\r
691INTN\r
692TcpInput (\r
693 IN NET_BUF *Nbuf,\r
694 IN UINT32 Src,\r
695 IN UINT32 Dst\r
696 )\r
697{\r
698 TCP_CB *Tcb;\r
699 TCP_CB *Parent;\r
700 TCP_OPTION Option;\r
701 TCP_HEAD *Head;\r
702 INT32 Len;\r
703 TCP_SEG *Seg;\r
704 TCP_SEQNO Right;\r
705 TCP_SEQNO Urg;\r
706\r
707 NET_CHECK_SIGNATURE (Nbuf, NET_BUF_SIGNATURE);\r
708\r
709 Parent = NULL;\r
710 Tcb = NULL;\r
711\r
712 Head = (TCP_HEAD *) NetbufGetByte (Nbuf, 0, NULL);\r
a56b6e03 713 ASSERT (Head != NULL);\r
8a67d61d 714 Len = Nbuf->TotalSize - (Head->HeadLen << 2);\r
715\r
716 if ((Head->HeadLen < 5) || (Len < 0) ||\r
85511ddf 717 (TcpChecksum (Nbuf, NetPseudoHeadChecksum (Src, Dst, 6, 0)) != 0)) {\r
8a67d61d 718\r
e48e37fc 719 DEBUG ((EFI_D_INFO, "TcpInput: received an mal-formated packet\n"));\r
8a67d61d 720 goto DISCARD;\r
721 }\r
722\r
723 if (TCP_FLG_ON (Head->Flag, TCP_FLG_SYN)) {\r
724 Len++;\r
725 }\r
726\r
727 if (TCP_FLG_ON (Head->Flag, TCP_FLG_FIN)) {\r
728 Len++;\r
729 }\r
730\r
731 Tcb = TcpLocateTcb (\r
732 Head->DstPort,\r
733 Dst,\r
734 Head->SrcPort,\r
735 Src,\r
736 (BOOLEAN) TCP_FLG_ON (Head->Flag, TCP_FLG_SYN)\r
737 );\r
738\r
739 if ((Tcb == NULL) || (Tcb->State == TCP_CLOSED)) {\r
49789216 740 DEBUG ((EFI_D_INFO, "TcpInput: send reset because no TCB found\n"));\r
8a67d61d 741\r
742 Tcb = NULL;\r
743 goto SEND_RESET;\r
744 }\r
745\r
746 Seg = TcpFormatNetbuf (Tcb, Nbuf);\r
747\r
748 //\r
749 // RFC1122 recommended reaction to illegal option\r
750 // (in fact, an illegal option length) is reset.\r
751 //\r
752 if (TcpParseOption (Nbuf->Tcp, &Option) == -1) {\r
e48e37fc 753 DEBUG ((EFI_D_ERROR, "TcpInput: reset the peer because"\r
49789216 754 " of malformed option for TCB %p\n", Tcb));\r
8a67d61d 755\r
756 goto SEND_RESET;\r
757 }\r
758\r
759 //\r
760 // From now on, the segment is headless\r
761 //\r
762 NetbufTrim (Nbuf, (Head->HeadLen << 2), NET_BUF_HEAD);\r
763 Nbuf->Tcp = NULL;\r
764\r
8a67d61d 765 //\r
766 // Process the segment in LISTEN state.\r
767 //\r
768 if (Tcb->State == TCP_LISTEN) {\r
769 //\r
770 // First step: Check RST\r
771 //\r
772 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
e48e37fc 773 DEBUG ((EFI_D_WARN, "TcpInput: discard a reset segment "\r
0e549d5b 774 "for TCB %p in listening\n", Tcb));\r
8a67d61d 775\r
776 goto DISCARD;\r
777 }\r
778\r
779 //\r
780 // Second step: Check ACK.\r
781 // Any ACK sent to TCP in LISTEN is reseted.\r
782 //\r
783 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
e48e37fc 784 DEBUG ((EFI_D_WARN, "TcpInput: send reset because of"\r
0e549d5b 785 " segment with ACK for TCB %p in listening\n", Tcb));\r
8a67d61d 786\r
787 goto SEND_RESET;\r
788 }\r
789\r
790 //\r
791 // Third step: Check SYN\r
792 //\r
793 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN)) {\r
794 //\r
795 // create a child TCB to handle the data\r
796 //\r
797 Parent = Tcb;\r
798\r
799 Tcb = TcpCloneTcb (Parent);\r
800 if (Tcb == NULL) {\r
e48e37fc 801 DEBUG ((EFI_D_ERROR, "TcpInput: discard a segment because"\r
49789216 802 " failed to clone a child for TCB %p\n", Tcb));\r
8a67d61d 803\r
804 goto DISCARD;\r
805 }\r
806\r
0e549d5b 807 DEBUG ((EFI_D_INFO, "TcpInput: create a child for TCB %p"\r
8a67d61d 808 " in listening\n", Tcb));\r
809\r
810 //\r
811 // init the TCB structure\r
812 //\r
813 Tcb->LocalEnd.Ip = Dst;\r
814 Tcb->LocalEnd.Port = Head->DstPort;\r
815 Tcb->RemoteEnd.Ip = Src;\r
816 Tcb->RemoteEnd.Port = Head->SrcPort;\r
817\r
818 TcpInitTcbLocal (Tcb);\r
819 TcpInitTcbPeer (Tcb, Seg, &Option);\r
820\r
821 TcpSetState (Tcb, TCP_SYN_RCVD);\r
822 TcpSetTimer (Tcb, TCP_TIMER_CONNECT, Tcb->ConnectTimeout);\r
823 TcpTrimInWnd (Tcb, Nbuf);\r
824\r
825 goto StepSix;\r
826 }\r
827\r
828 goto DISCARD;\r
829\r
830 } else if (Tcb->State == TCP_SYN_SENT) {\r
831 //\r
832 // First step: Check ACK bit\r
833 //\r
834 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK) && (Seg->Ack != Tcb->Iss + 1)) {\r
835\r
e48e37fc 836 DEBUG ((EFI_D_WARN, "TcpInput: send reset because of "\r
0e549d5b 837 "wrong ACK received for TCB %p in SYN_SENT\n", Tcb));\r
8a67d61d 838\r
839 goto SEND_RESET;\r
840 }\r
841\r
842 //\r
843 // Second step: Check RST bit\r
844 //\r
845 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
846\r
847 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
848\r
e48e37fc 849 DEBUG ((EFI_D_WARN, "TcpInput: connection reset by"\r
dfc1f033 850 " peer for TCB %p in SYN_SENT\n", Tcb));\r
8a67d61d 851\r
852 SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
853 goto DROP_CONNECTION;\r
854 } else {\r
855\r
e48e37fc 856 DEBUG ((EFI_D_WARN, "TcpInput: discard a reset segment "\r
dfc1f033 857 "because of no ACK for TCB %p in SYN_SENT\n", Tcb));\r
8a67d61d 858\r
859 goto DISCARD;\r
860 }\r
861 }\r
862\r
863 //\r
864 // Third step: Check security and precedence. Skipped\r
865 //\r
866\r
867 //\r
49789216 868 // Fourth step: Check SYN. Pay attention to simultaneous open\r
8a67d61d 869 //\r
870 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN)) {\r
871\r
872 TcpInitTcbPeer (Tcb, Seg, &Option);\r
873\r
874 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
875\r
876 Tcb->SndUna = Seg->Ack;\r
877 }\r
878\r
879 TcpClearTimer (Tcb, TCP_TIMER_REXMIT);\r
880\r
881 if (TCP_SEQ_GT (Tcb->SndUna, Tcb->Iss)) {\r
882\r
883 TcpSetState (Tcb, TCP_ESTABLISHED);\r
884\r
885 TcpClearTimer (Tcb, TCP_TIMER_CONNECT);\r
886 TcpDeliverData (Tcb);\r
887\r
888 if ((Tcb->CongestState == TCP_CONGEST_OPEN) &&\r
889 TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_RTT_ON)) {\r
890\r
891 TcpComputeRtt (Tcb, Tcb->RttMeasure);\r
892 TCP_CLEAR_FLG (Tcb->CtrlFlag, TCP_CTRL_RTT_ON);\r
893 }\r
894\r
895 TcpTrimInWnd (Tcb, Nbuf);\r
896\r
897 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
898\r
e48e37fc 899 DEBUG ((EFI_D_INFO, "TcpInput: connection established"\r
0e549d5b 900 " for TCB %p in SYN_SENT\n", Tcb));\r
8a67d61d 901\r
902 goto StepSix;\r
903 } else {\r
904 //\r
49789216 905 // Received a SYN segment without ACK, simultaneous open.\r
8a67d61d 906 //\r
907 TcpSetState (Tcb, TCP_SYN_RCVD);\r
908\r
909 ASSERT (Tcb->SndNxt == Tcb->Iss + 1);\r
910 TcpAdjustSndQue (Tcb, Tcb->SndNxt);\r
911\r
912 TcpTrimInWnd (Tcb, Nbuf);\r
913\r
49789216 914 DEBUG ((EFI_D_WARN, "TcpInput: simultaneous open "\r
0e549d5b 915 "for TCB %p in SYN_SENT\n", Tcb));\r
8a67d61d 916\r
917 goto StepSix;\r
918 }\r
919 }\r
920\r
921 goto DISCARD;\r
922 }\r
923\r
924 //\r
925 // Process segment in SYN_RCVD or TCP_CONNECTED states\r
926 //\r
927\r
ac8cca2a 928 //\r
929 // Clear probe timer since the RecvWindow is opened.\r
930 //\r
931 if (Tcb->ProbeTimerOn && (Seg->Wnd != 0)) {\r
932 TcpClearTimer (Tcb, TCP_TIMER_PROBE);\r
933 Tcb->ProbeTimerOn = FALSE;\r
934 }\r
935\r
8a67d61d 936 //\r
937 // First step: Check whether SEG.SEQ is acceptable\r
938 //\r
85511ddf 939 if (TcpSeqAcceptable (Tcb, Seg) == 0) {\r
e48e37fc 940 DEBUG ((EFI_D_WARN, "TcpInput: sequence acceptance"\r
0e549d5b 941 " test failed for segment of TCB %p\n", Tcb));\r
8a67d61d 942\r
943 if (!TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
944 TcpSendAck (Tcb);\r
945 }\r
946\r
947 goto DISCARD;\r
948 }\r
949\r
950 if ((TCP_SEQ_LT (Seg->Seq, Tcb->RcvWl2)) &&\r
951 (Tcb->RcvWl2 == Seg->End) &&\r
952 !TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN | TCP_FLG_FIN)) {\r
953\r
954 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
955 }\r
956\r
957 //\r
958 // Second step: Check the RST\r
959 //\r
960 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_RST)) {\r
961\r
0e549d5b 962 DEBUG ((EFI_D_WARN, "TcpInput: connection reset for TCB %p\n", Tcb));\r
8a67d61d 963\r
964 if (Tcb->State == TCP_SYN_RCVD) {\r
965\r
966 SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_REFUSED);\r
967\r
968 //\r
969 // This TCB comes from either a LISTEN TCB,\r
970 // or active open TCB with simultanous open.\r
971 // Do NOT signal user CONNECTION refused\r
972 // if it comes from a LISTEN TCB.\r
973 //\r
974 } else if ((Tcb->State == TCP_ESTABLISHED) ||\r
120db52c 975 (Tcb->State == TCP_FIN_WAIT_1) ||\r
976 (Tcb->State == TCP_FIN_WAIT_2) ||\r
977 (Tcb->State == TCP_CLOSE_WAIT)) {\r
8a67d61d 978\r
979 SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
980\r
981 } else {\r
dfc1f033 982\r
8a67d61d 983 }\r
984\r
985 goto DROP_CONNECTION;\r
986 }\r
987\r
988 //\r
989 // Trim the data and flags.\r
990 //\r
991 TcpTrimInWnd (Tcb, Nbuf);\r
992\r
993 //\r
994 // Third step: Check security and precedence, Ignored\r
995 //\r
996\r
997 //\r
998 // Fourth step: Check the SYN bit.\r
999 //\r
1000 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_SYN)) {\r
1001\r
e48e37fc 1002 DEBUG ((EFI_D_WARN, "TcpInput: connection reset "\r
0e549d5b 1003 "because received extra SYN for TCB %p\n", Tcb));\r
8a67d61d 1004\r
1005 SOCK_ERROR (Tcb->Sk, EFI_CONNECTION_RESET);\r
1006 goto RESET_THEN_DROP;\r
1007 }\r
1008\r
1009 //\r
1010 // Fifth step: Check the ACK\r
1011 //\r
1012 if (!TCP_FLG_ON (Seg->Flag, TCP_FLG_ACK)) {\r
e48e37fc 1013 DEBUG ((EFI_D_WARN, "TcpInput: segment discard because"\r
0e549d5b 1014 " of no ACK for connected TCB %p\n", Tcb));\r
8a67d61d 1015\r
1016 goto DISCARD;\r
1017\r
1018 }\r
1019\r
1020 if (Tcb->State == TCP_SYN_RCVD) {\r
1021\r
1022 if (TCP_SEQ_LT (Tcb->SndUna, Seg->Ack) &&\r
1023 TCP_SEQ_LEQ (Seg->Ack, Tcb->SndNxt)) {\r
1024\r
1025 Tcb->SndWnd = Seg->Wnd;\r
36ee91ca 1026 Tcb->SndWndMax = MAX (Tcb->SndWnd, Tcb->SndWndMax);\r
8a67d61d 1027 Tcb->SndWl1 = Seg->Seq;\r
1028 Tcb->SndWl2 = Seg->Ack;\r
1029 TcpSetState (Tcb, TCP_ESTABLISHED);\r
1030\r
1031 TcpClearTimer (Tcb, TCP_TIMER_CONNECT);\r
1032 TcpDeliverData (Tcb);\r
1033\r
e48e37fc 1034 DEBUG ((EFI_D_INFO, "TcpInput: connection established "\r
0e549d5b 1035 " for TCB %p in SYN_RCVD\n", Tcb));\r
8a67d61d 1036\r
1037 //\r
1038 // Continue the process as ESTABLISHED state\r
1039 //\r
1040 } else {\r
e48e37fc 1041 DEBUG ((EFI_D_WARN, "TcpInput: send reset because of"\r
0e549d5b 1042 " wrong ACK for TCB %p in SYN_RCVD\n", Tcb));\r
8a67d61d 1043\r
1044 goto SEND_RESET;\r
1045 }\r
1046 }\r
1047\r
1048 if (TCP_SEQ_LT (Seg->Ack, Tcb->SndUna)) {\r
1049\r
e48e37fc 1050 DEBUG ((EFI_D_WARN, "TcpInput: ignore the out-of-data"\r
0e549d5b 1051 " ACK for connected TCB %p\n", Tcb));\r
8a67d61d 1052\r
1053 goto StepSix;\r
1054\r
1055 } else if (TCP_SEQ_GT (Seg->Ack, Tcb->SndNxt)) {\r
1056\r
e48e37fc 1057 DEBUG ((EFI_D_WARN, "TcpInput: discard segment for "\r
0e549d5b 1058 "future ACK for connected TCB %p\n", Tcb));\r
8a67d61d 1059\r
1060 TcpSendAck (Tcb);\r
1061 goto DISCARD;\r
1062 }\r
1063\r
1064 //\r
1065 // From now on: SND.UNA <= SEG.ACK <= SND.NXT.\r
1066 //\r
1067 if (TCP_FLG_ON (Option.Flag, TCP_OPTION_RCVD_TS)) {\r
1068 //\r
1069 // update TsRecent as specified in page 16 RFC1323.\r
1070 // RcvWl2 equals to the variable "LastAckSent"\r
1071 // defined there.\r
1072 //\r
1073 if (TCP_SEQ_LEQ (Seg->Seq, Tcb->RcvWl2) &&\r
1074 TCP_SEQ_LT (Tcb->RcvWl2, Seg->End)) {\r
1075\r
1076 Tcb->TsRecent = Option.TSVal;\r
1077 Tcb->TsRecentAge = mTcpTick;\r
1078 }\r
1079\r
1080 TcpComputeRtt (Tcb, TCP_SUB_TIME (mTcpTick, Option.TSEcr));\r
1081\r
1082 } else if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_RTT_ON)) {\r
1083\r
1084 ASSERT (Tcb->CongestState == TCP_CONGEST_OPEN);\r
1085\r
1086 TcpComputeRtt (Tcb, Tcb->RttMeasure);\r
1087 TCP_CLEAR_FLG (Tcb->CtrlFlag, TCP_CTRL_RTT_ON);\r
1088 }\r
1089\r
1090 if (Seg->Ack == Tcb->SndNxt) {\r
1091\r
1092 TcpClearTimer (Tcb, TCP_TIMER_REXMIT);\r
1093 } else {\r
1094\r
1095 TcpSetTimer (Tcb, TCP_TIMER_REXMIT, Tcb->Rto);\r
1096 }\r
1097\r
1098 //\r
1099 // Count duplicate acks.\r
1100 //\r
1101 if ((Seg->Ack == Tcb->SndUna) &&\r
1102 (Tcb->SndUna != Tcb->SndNxt) &&\r
1103 (Seg->Wnd == Tcb->SndWnd) &&\r
1104 (0 == Len)) {\r
1105\r
1106 Tcb->DupAck++;\r
1107 } else {\r
1108\r
1109 Tcb->DupAck = 0;\r
1110 }\r
1111\r
1112 //\r
1113 // Congestion avoidance, fast recovery and fast retransmission.\r
1114 //\r
1115 if (((Tcb->CongestState == TCP_CONGEST_OPEN) && (Tcb->DupAck < 3)) ||\r
1116 (Tcb->CongestState == TCP_CONGEST_LOSS)) {\r
1117\r
1118 if (TCP_SEQ_GT (Seg->Ack, Tcb->SndUna)) {\r
1119\r
1120 if (Tcb->CWnd < Tcb->Ssthresh) {\r
1121\r
1122 Tcb->CWnd += Tcb->SndMss;\r
1123 } else {\r
1124\r
36ee91ca 1125 Tcb->CWnd += MAX (Tcb->SndMss * Tcb->SndMss / Tcb->CWnd, 1);\r
8a67d61d 1126 }\r
1127\r
36ee91ca 1128 Tcb->CWnd = MIN (Tcb->CWnd, TCP_MAX_WIN << Tcb->SndWndScale);\r
8a67d61d 1129 }\r
1130\r
1131 if (Tcb->CongestState == TCP_CONGEST_LOSS) {\r
1132 TcpFastLossRecover (Tcb, Seg);\r
1133 }\r
1134 } else {\r
1135\r
1136 TcpFastRecover (Tcb, Seg);\r
1137 }\r
1138\r
1139 if (TCP_SEQ_GT (Seg->Ack, Tcb->SndUna)) {\r
1140\r
1141 TcpAdjustSndQue (Tcb, Seg->Ack);\r
1142 Tcb->SndUna = Seg->Ack;\r
1143\r
1144 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_SND_URG) &&\r
120db52c 1145 TCP_SEQ_LT (Tcb->SndUp, Seg->Ack)) {\r
8a67d61d 1146\r
1147 TCP_CLEAR_FLG (Tcb->CtrlFlag, TCP_CTRL_SND_URG);\r
1148 }\r
1149 }\r
1150\r
1151 //\r
1152 // Update window info\r
1153 //\r
1154 if (TCP_SEQ_LT (Tcb->SndWl1, Seg->Seq) ||\r
1155 ((Tcb->SndWl1 == Seg->Seq) && TCP_SEQ_LEQ (Tcb->SndWl2, Seg->Ack))) {\r
1156\r
1157 Right = Seg->Ack + Seg->Wnd;\r
1158\r
1159 if (TCP_SEQ_LT (Right, Tcb->SndWl2 + Tcb->SndWnd)) {\r
1160\r
1161 if ((Tcb->SndWl1 == Seg->Seq) &&\r
1162 (Tcb->SndWl2 == Seg->Ack) &&\r
1163 (Len == 0)) {\r
1164\r
1165 goto NO_UPDATE;\r
1166 }\r
1167\r
e48e37fc 1168 DEBUG ((EFI_D_WARN, "TcpInput: peer shrinks the"\r
dfc1f033 1169 " window for connected TCB %p\n", Tcb));\r
8a67d61d 1170\r
1171 if ((Tcb->CongestState == TCP_CONGEST_RECOVER) &&\r
1172 (TCP_SEQ_LT (Right, Tcb->Recover))) {\r
1173\r
1174 Tcb->Recover = Right;\r
1175 }\r
1176\r
1177 if ((Tcb->CongestState == TCP_CONGEST_LOSS) &&\r
1178 (TCP_SEQ_LT (Right, Tcb->LossRecover))) {\r
1179\r
1180 Tcb->LossRecover = Right;\r
1181 }\r
1182\r
1183 if (TCP_SEQ_LT (Right, Tcb->SndNxt)) {\r
1184\r
1185 Tcb->SndNxt = Right;\r
1186\r
1187 if (Right == Tcb->SndUna) {\r
1188\r
1189 TcpClearTimer (Tcb, TCP_TIMER_REXMIT);\r
1190 TcpSetProbeTimer (Tcb);\r
1191 }\r
1192 }\r
1193 }\r
1194\r
1195 Tcb->SndWnd = Seg->Wnd;\r
36ee91ca 1196 Tcb->SndWndMax = MAX (Tcb->SndWnd, Tcb->SndWndMax);\r
8a67d61d 1197 Tcb->SndWl1 = Seg->Seq;\r
1198 Tcb->SndWl2 = Seg->Ack;\r
1199 }\r
1200\r
1201NO_UPDATE:\r
1202\r
1203 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_FIN_SENT) &&\r
1204 (Tcb->SndUna == Tcb->SndNxt)) {\r
1205\r
e48e37fc 1206 DEBUG ((EFI_D_INFO, "TcpInput: local FIN is ACKed by"\r
0e549d5b 1207 " peer for connected TCB %p\n", Tcb));\r
8a67d61d 1208\r
1209 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED);\r
1210 }\r
1211\r
1212 //\r
1213 // Transit the state if proper.\r
1214 //\r
1215 switch (Tcb->State) {\r
1216 case TCP_FIN_WAIT_1:\r
1217\r
1218 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED)) {\r
1219\r
1220 TcpSetState (Tcb, TCP_FIN_WAIT_2);\r
1221\r
1222 TcpClearAllTimer (Tcb);\r
1223 TcpSetTimer (Tcb, TCP_TIMER_FINWAIT2, Tcb->FinWait2Timeout);\r
1224 }\r
1225\r
1226 case TCP_FIN_WAIT_2:\r
1227\r
1228 break;\r
1229\r
1230 case TCP_CLOSE_WAIT:\r
1231 break;\r
1232\r
1233 case TCP_CLOSING:\r
1234\r
1235 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED)) {\r
1236\r
1237 TcpSetState (Tcb, TCP_TIME_WAIT);\r
1238\r
1239 TcpClearAllTimer (Tcb);\r
1240\r
1241 if (Tcb->TimeWaitTimeout != 0) {\r
1242\r
1243 TcpSetTimer (Tcb, TCP_TIMER_2MSL, Tcb->TimeWaitTimeout);\r
1244 } else {\r
1245\r
e48e37fc 1246 DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
0e549d5b 1247 "because app disables TIME_WAIT timer for %p\n", Tcb));\r
8a67d61d 1248\r
1249 TcpClose (Tcb);\r
1250 }\r
1251 }\r
1252 break;\r
1253\r
1254 case TCP_LAST_ACK:\r
1255\r
1256 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_FIN_ACKED)) {\r
1257\r
1258 TcpSetState (Tcb, TCP_CLOSED);\r
1259 }\r
1260\r
1261 break;\r
1262\r
1263 case TCP_TIME_WAIT:\r
1264\r
1265 TcpSendAck (Tcb);\r
1266\r
1267 if (Tcb->TimeWaitTimeout != 0) {\r
1268\r
1269 TcpSetTimer (Tcb, TCP_TIMER_2MSL, Tcb->TimeWaitTimeout);\r
1270 } else {\r
1271\r
e48e37fc 1272 DEBUG ((EFI_D_WARN, "Connection closed immediately "\r
0e549d5b 1273 "because app disables TIME_WAIT timer for %p\n", Tcb));\r
8a67d61d 1274\r
1275 TcpClose (Tcb);\r
1276 }\r
1277 break;\r
c191cdd1 1278\r
1279 default:\r
1280 break;\r
8a67d61d 1281 }\r
1282\r
1283 //\r
1284 // Sixth step: Check the URG bit.update the Urg point\r
1285 // if in TCP_CAN_RECV, otherwise, leave the RcvUp intact.\r
1286 //\r
1287StepSix:\r
1288\r
1289 Tcb->Idle = 0;\r
1290 TcpSetKeepaliveTimer (Tcb);\r
1291\r
8a67d61d 1292 if (TCP_FLG_ON (Seg->Flag, TCP_FLG_URG) &&\r
49789216
BZ
1293 !TCP_FIN_RCVD (Tcb->State)) \r
1294 {\r
8a67d61d 1295\r
e48e37fc 1296 DEBUG ((EFI_D_INFO, "TcpInput: received urgent data "\r
0e549d5b 1297 "from peer for connected TCB %p\n", Tcb));\r
8a67d61d 1298\r
1299 Urg = Seg->Seq + Seg->Urg;\r
1300\r
1301 if (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_RCVD_URG) &&\r
1302 TCP_SEQ_GT (Urg, Tcb->RcvUp)) {\r
1303\r
1304 Tcb->RcvUp = Urg;\r
1305 } else {\r
1306\r
1307 Tcb->RcvUp = Urg;\r
1308 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_RCVD_URG);\r
1309 }\r
1310 }\r
1311\r
1312 //\r
1313 // Seventh step: Process the segment data\r
1314 //\r
1315 if (Seg->End != Seg->Seq) {\r
1316\r
1317 if (TCP_FIN_RCVD (Tcb->State)) {\r
1318\r
e48e37fc 1319 DEBUG ((EFI_D_WARN, "TcpInput: connection reset because"\r
0e549d5b 1320 " data is lost for connected TCB %p\n", Tcb));\r
8a67d61d 1321\r
1322 goto RESET_THEN_DROP;\r
1323 }\r
1324\r
1325 if (TCP_LOCAL_CLOSED (Tcb->State) && (Nbuf->TotalSize != 0)) {\r
e48e37fc 1326 DEBUG ((EFI_D_WARN, "TcpInput: connection reset because"\r
0e549d5b 1327 " data is lost for connected TCB %p\n", Tcb));\r
8a67d61d 1328\r
1329 goto RESET_THEN_DROP;\r
1330 }\r
1331\r
1332 TcpQueueData (Tcb, Nbuf);\r
1333 if (TcpDeliverData (Tcb) == -1) {\r
1334 goto RESET_THEN_DROP;\r
1335 }\r
1336\r
e48e37fc 1337 if (!IsListEmpty (&Tcb->RcvQue)) {\r
8a67d61d 1338 TCP_SET_FLG (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW);\r
1339 }\r
1340 }\r
1341\r
1342 //\r
1343 // Eighth step: check the FIN.\r
1344 // This step is moved to TcpDeliverData. FIN will be\r
1345 // processed in sequence there. Check the comments in\r
1346 // the beginning of the file header for information.\r
1347 //\r
1348\r
1349 //\r
1350 // Tcb is a new child of the listening Parent,\r
1351 // commit it.\r
1352 //\r
85511ddf 1353 if (Parent != NULL) {\r
8a67d61d 1354 Tcb->Parent = Parent;\r
1355 TcpInsertTcb (Tcb);\r
1356 }\r
1357\r
1358 if ((Tcb->State != TCP_CLOSED) &&\r
85511ddf 1359 (TcpToSendData (Tcb, 0) == 0) &&\r
52b2a308 1360 (TCP_FLG_ON (Tcb->CtrlFlag, TCP_CTRL_ACK_NOW) || (Nbuf->TotalSize != 0))) {\r
8a67d61d 1361\r
1362 TcpToSendAck (Tcb);\r
1363 }\r
1364\r
1365 NetbufFree (Nbuf);\r
1366 return 0;\r
1367\r
1368RESET_THEN_DROP:\r
1369 TcpSendReset (Tcb, Head, Len, Dst, Src);\r
1370\r
1371DROP_CONNECTION:\r
120db52c 1372 ASSERT ((Tcb != NULL) && (Tcb->Sk != NULL));\r
8a67d61d 1373\r
1374 NetbufFree (Nbuf);\r
1375 TcpClose (Tcb);\r
1376\r
1377 return -1;\r
1378\r
1379SEND_RESET:\r
1380\r
1381 TcpSendReset (Tcb, Head, Len, Dst, Src);\r
1382\r
1383DISCARD:\r
1384\r
1385 //\r
1386 // Tcb is a child of Parent, and it doesn't survive\r
1387 //\r
e48e37fc 1388 DEBUG ((EFI_D_WARN, "Tcp4Input: Discard a packet\n"));\r
8a67d61d 1389 NetbufFree (Nbuf);\r
1390\r
85511ddf 1391 if ((Parent != NULL) && (Tcb != NULL)) {\r
8a67d61d 1392\r
120db52c 1393 ASSERT (Tcb->Sk != NULL);\r
8a67d61d 1394 TcpClose (Tcb);\r
1395 }\r
1396\r
1397 return 0;\r
1398}\r
1399\r
1400\r
1401/**\r
1402 Process the received ICMP error messages for TCP.\r
1403\r
1404 @param Nbuf Buffer that contains part of the TCP segment without IP header\r
1405 truncated from the ICMP error packet.\r
1406 @param IcmpErr The ICMP error code interpreted from ICMP error packet.\r
1407 @param Src Source address of the ICMP error message.\r
1408 @param Dst Destination address of the ICMP error message.\r
1409\r
8a67d61d 1410**/\r
1411VOID\r
1412TcpIcmpInput (\r
1413 IN NET_BUF *Nbuf,\r
b45b45b2 1414 IN UINT8 IcmpErr,\r
8a67d61d 1415 IN UINT32 Src,\r
1416 IN UINT32 Dst\r
1417 )\r
1418{\r
1419 TCP_HEAD *Head;\r
1420 TCP_CB *Tcb;\r
1421 TCP_SEQNO Seq;\r
1422 EFI_STATUS IcmpErrStatus;\r
1423 BOOLEAN IcmpErrIsHard;\r
1424 BOOLEAN IcmpErrNotify;\r
1425\r
1426 Head = (TCP_HEAD *) NetbufGetByte (Nbuf, 0, NULL);\r
a56b6e03 1427 ASSERT (Head != NULL);\r
8a67d61d 1428 Tcb = TcpLocateTcb (\r
1429 Head->DstPort,\r
1430 Dst,\r
1431 Head->SrcPort,\r
1432 Src,\r
1433 FALSE\r
1434 );\r
1435 if (Tcb == NULL || Tcb->State == TCP_CLOSED) {\r
1436\r
1437 goto CLEAN_EXIT;\r
1438 }\r
1439\r
1440 //\r
1441 // Validate the sequence number.\r
1442 //\r
1443 Seq = NTOHL (Head->Seq);\r
1444 if (!(TCP_SEQ_LEQ (Tcb->SndUna, Seq) && TCP_SEQ_LT (Seq, Tcb->SndNxt))) {\r
1445\r
1446 goto CLEAN_EXIT;\r
1447 }\r
1448\r
fb115c61 1449 IcmpErrStatus = IpIoGetIcmpErrStatus (\r
1450 IcmpErr,\r
1451 IP_VERSION_4,\r
1452 &IcmpErrIsHard,\r
1453 &IcmpErrNotify\r
1454 );\r
8a67d61d 1455\r
1456 if (IcmpErrNotify) {\r
1457\r
1458 SOCK_ERROR (Tcb->Sk, IcmpErrStatus);\r
1459 }\r
1460\r
1461 if (IcmpErrIsHard) {\r
1462\r
1463 TcpClose (Tcb);\r
1464 }\r
1465\r
1466CLEAN_EXIT:\r
1467 NetbufFree (Nbuf);\r
1468}\r