be29f05f65c39b879936aeb4511a9410f061072c
[vpp.git] / src / vnet / tcp / tcp_output.c
1 /*
2  * Copyright (c) 2016 Cisco and/or its affiliates.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at:
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15
16 #include <vnet/tcp/tcp.h>
17 #include <vnet/lisp-cp/packets.h>
18 #include <math.h>
19
20 vlib_node_registration_t tcp4_output_node;
21 vlib_node_registration_t tcp6_output_node;
22
23 typedef enum _tcp_output_next
24 {
25   TCP_OUTPUT_NEXT_DROP,
26   TCP_OUTPUT_NEXT_IP_LOOKUP,
27   TCP_OUTPUT_N_NEXT
28 } tcp_output_next_t;
29
30 #define foreach_tcp4_output_next                \
31   _ (DROP, "error-drop")                        \
32   _ (IP_LOOKUP, "ip4-lookup")
33
34 #define foreach_tcp6_output_next                \
35   _ (DROP, "error-drop")                        \
36   _ (IP_LOOKUP, "ip6-lookup")
37
38 static char *tcp_error_strings[] = {
39 #define tcp_error(n,s) s,
40 #include <vnet/tcp/tcp_error.def>
41 #undef tcp_error
42 };
43
44 typedef struct
45 {
46   tcp_header_t tcp_header;
47   tcp_connection_t tcp_connection;
48 } tcp_tx_trace_t;
49
50 u16 dummy_mtu = 1460;
51
52 u8 *
53 format_tcp_tx_trace (u8 * s, va_list * args)
54 {
55   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
56   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
57   tcp_tx_trace_t *t = va_arg (*args, tcp_tx_trace_t *);
58   uword indent = format_get_indent (s);
59
60   s = format (s, "%U\n%U%U",
61               format_tcp_header, &t->tcp_header, 128,
62               format_white_space, indent,
63               format_tcp_connection, &t->tcp_connection, 1);
64
65   return s;
66 }
67
68 static u8
69 tcp_window_compute_scale (u32 window)
70 {
71   u8 wnd_scale = 0;
72   while (wnd_scale < TCP_MAX_WND_SCALE && (window >> wnd_scale) > TCP_WND_MAX)
73     wnd_scale++;
74   return wnd_scale;
75 }
76
77 /**
78  * Update max segment size we're able to process.
79  *
80  * The value is constrained by our interface's MTU and IP options. It is
81  * also what we advertise to our peer.
82  */
83 void
84 tcp_update_rcv_mss (tcp_connection_t * tc)
85 {
86   /* TODO find our iface MTU */
87   tc->mss = dummy_mtu - sizeof (tcp_header_t);
88 }
89
90 /**
91  * TCP's initial window
92  */
93 always_inline u32
94 tcp_initial_wnd_unscaled (tcp_connection_t * tc)
95 {
96   /* RFC 6928 recommends the value lower. However at the time our connections
97    * are initialized, fifos may not be allocated. Therefore, advertise the
98    * smallest possible unscaled window size and update once fifos are
99    * assigned to the session.
100    */
101   /*
102      tcp_update_rcv_mss (tc);
103      TCP_IW_N_SEGMENTS * tc->mss;
104    */
105   return TCP_MIN_RX_FIFO_SIZE;
106 }
107
108 /**
109  * Compute initial window and scale factor. As per RFC1323, window field in
110  * SYN and SYN-ACK segments is never scaled.
111  */
112 u32
113 tcp_initial_window_to_advertise (tcp_connection_t * tc)
114 {
115   u32 max_fifo;
116
117   /* Initial wnd for SYN. Fifos are not allocated yet.
118    * Use some predefined value. For SYN-ACK we still want the
119    * scale to be computed in the same way */
120   max_fifo = TCP_MAX_RX_FIFO_SIZE;
121
122   tc->rcv_wscale = tcp_window_compute_scale (max_fifo);
123   tc->rcv_wnd = tcp_initial_wnd_unscaled (tc);
124
125   return clib_min (tc->rcv_wnd, TCP_WND_MAX);
126 }
127
128 /**
129  * Compute and return window to advertise, scaled as per RFC1323
130  */
131 u32
132 tcp_window_to_advertise (tcp_connection_t * tc, tcp_state_t state)
133 {
134   if (state < TCP_STATE_ESTABLISHED)
135     return tcp_initial_window_to_advertise (tc);
136
137   tcp_update_rcv_wnd (tc);
138
139   if (tc->rcv_wnd == 0)
140     {
141       tc->flags |= TCP_CONN_SENT_RCV_WND0;
142     }
143   else
144     {
145       tc->flags &= ~TCP_CONN_SENT_RCV_WND0;
146     }
147
148   return tc->rcv_wnd >> tc->rcv_wscale;
149 }
150
151 void
152 tcp_update_rcv_wnd (tcp_connection_t * tc)
153 {
154   i32 observed_wnd;
155   u32 available_space, max_fifo, wnd;
156
157   /*
158    * Figure out how much space we have available
159    */
160   available_space = stream_session_max_rx_enqueue (&tc->connection);
161   max_fifo = stream_session_rx_fifo_size (&tc->connection);
162
163   ASSERT (tc->rcv_opts.mss < max_fifo);
164   if (available_space < tc->rcv_opts.mss && available_space < max_fifo >> 3)
165     available_space = 0;
166
167   /*
168    * Use the above and what we know about what we've previously advertised
169    * to compute the new window
170    */
171   observed_wnd = (i32) tc->rcv_wnd - (tc->rcv_nxt - tc->rcv_las);
172   if (observed_wnd < 0)
173     observed_wnd = 0;
174
175   /* Bad. Thou shalt not shrink */
176   if (available_space < observed_wnd)
177     {
178       wnd = observed_wnd;
179       TCP_EVT_DBG (TCP_EVT_RCV_WND_SHRUNK, tc, observed_wnd, available_space);
180     }
181   else
182     {
183       wnd = available_space;
184     }
185
186   /* Make sure we have a multiple of rcv_wscale */
187   if (wnd && tc->rcv_wscale)
188     {
189       wnd &= ~(1 << tc->rcv_wscale);
190       if (wnd == 0)
191         wnd = 1 << tc->rcv_wscale;
192     }
193
194   tc->rcv_wnd = clib_min (wnd, TCP_WND_MAX << tc->rcv_wscale);
195 }
196
197 /**
198  * Write TCP options to segment.
199  */
200 u32
201 tcp_options_write (u8 * data, tcp_options_t * opts)
202 {
203   u32 opts_len = 0;
204   u32 buf, seq_len = 4;
205
206   if (tcp_opts_mss (opts))
207     {
208       *data++ = TCP_OPTION_MSS;
209       *data++ = TCP_OPTION_LEN_MSS;
210       buf = clib_host_to_net_u16 (opts->mss);
211       clib_memcpy (data, &buf, sizeof (opts->mss));
212       data += sizeof (opts->mss);
213       opts_len += TCP_OPTION_LEN_MSS;
214     }
215
216   if (tcp_opts_wscale (opts))
217     {
218       *data++ = TCP_OPTION_WINDOW_SCALE;
219       *data++ = TCP_OPTION_LEN_WINDOW_SCALE;
220       *data++ = opts->wscale;
221       opts_len += TCP_OPTION_LEN_WINDOW_SCALE;
222     }
223
224   if (tcp_opts_sack_permitted (opts))
225     {
226       *data++ = TCP_OPTION_SACK_PERMITTED;
227       *data++ = TCP_OPTION_LEN_SACK_PERMITTED;
228       opts_len += TCP_OPTION_LEN_SACK_PERMITTED;
229     }
230
231   if (tcp_opts_tstamp (opts))
232     {
233       *data++ = TCP_OPTION_TIMESTAMP;
234       *data++ = TCP_OPTION_LEN_TIMESTAMP;
235       buf = clib_host_to_net_u32 (opts->tsval);
236       clib_memcpy (data, &buf, sizeof (opts->tsval));
237       data += sizeof (opts->tsval);
238       buf = clib_host_to_net_u32 (opts->tsecr);
239       clib_memcpy (data, &buf, sizeof (opts->tsecr));
240       data += sizeof (opts->tsecr);
241       opts_len += TCP_OPTION_LEN_TIMESTAMP;
242     }
243
244   if (tcp_opts_sack (opts))
245     {
246       int i;
247       u32 n_sack_blocks = clib_min (vec_len (opts->sacks),
248                                     TCP_OPTS_MAX_SACK_BLOCKS);
249
250       if (n_sack_blocks != 0)
251         {
252           *data++ = TCP_OPTION_SACK_BLOCK;
253           *data++ = 2 + n_sack_blocks * TCP_OPTION_LEN_SACK_BLOCK;
254           for (i = 0; i < n_sack_blocks; i++)
255             {
256               buf = clib_host_to_net_u32 (opts->sacks[i].start);
257               clib_memcpy (data, &buf, seq_len);
258               data += seq_len;
259               buf = clib_host_to_net_u32 (opts->sacks[i].end);
260               clib_memcpy (data, &buf, seq_len);
261               data += seq_len;
262             }
263           opts_len += 2 + n_sack_blocks * TCP_OPTION_LEN_SACK_BLOCK;
264         }
265     }
266
267   /* Terminate TCP options */
268   if (opts_len % 4)
269     {
270       *data++ = TCP_OPTION_EOL;
271       opts_len += TCP_OPTION_LEN_EOL;
272     }
273
274   /* Pad with zeroes to a u32 boundary */
275   while (opts_len % 4)
276     {
277       *data++ = TCP_OPTION_NOOP;
278       opts_len += TCP_OPTION_LEN_NOOP;
279     }
280   return opts_len;
281 }
282
283 always_inline int
284 tcp_make_syn_options (tcp_options_t * opts, u8 wnd_scale)
285 {
286   u8 len = 0;
287
288   opts->flags |= TCP_OPTS_FLAG_MSS;
289   opts->mss = dummy_mtu;        /*XXX discover that */
290   len += TCP_OPTION_LEN_MSS;
291
292   opts->flags |= TCP_OPTS_FLAG_WSCALE;
293   opts->wscale = wnd_scale;
294   len += TCP_OPTION_LEN_WINDOW_SCALE;
295
296   opts->flags |= TCP_OPTS_FLAG_TSTAMP;
297   opts->tsval = tcp_time_now ();
298   opts->tsecr = 0;
299   len += TCP_OPTION_LEN_TIMESTAMP;
300
301   if (TCP_USE_SACKS)
302     {
303       opts->flags |= TCP_OPTS_FLAG_SACK_PERMITTED;
304       len += TCP_OPTION_LEN_SACK_PERMITTED;
305     }
306
307   /* Align to needed boundary */
308   len += (TCP_OPTS_ALIGN - len % TCP_OPTS_ALIGN) % TCP_OPTS_ALIGN;
309   return len;
310 }
311
312 always_inline int
313 tcp_make_synack_options (tcp_connection_t * tc, tcp_options_t * opts)
314 {
315   u8 len = 0;
316
317   opts->flags |= TCP_OPTS_FLAG_MSS;
318   opts->mss = tc->mss;
319   len += TCP_OPTION_LEN_MSS;
320
321   if (tcp_opts_wscale (&tc->rcv_opts))
322     {
323       opts->flags |= TCP_OPTS_FLAG_WSCALE;
324       opts->wscale = tc->rcv_wscale;
325       len += TCP_OPTION_LEN_WINDOW_SCALE;
326     }
327
328   if (tcp_opts_tstamp (&tc->rcv_opts))
329     {
330       opts->flags |= TCP_OPTS_FLAG_TSTAMP;
331       opts->tsval = tcp_time_now ();
332       opts->tsecr = tc->tsval_recent;
333       len += TCP_OPTION_LEN_TIMESTAMP;
334     }
335
336   if (tcp_opts_sack_permitted (&tc->rcv_opts))
337     {
338       opts->flags |= TCP_OPTS_FLAG_SACK_PERMITTED;
339       len += TCP_OPTION_LEN_SACK_PERMITTED;
340     }
341
342   /* Align to needed boundary */
343   len += (TCP_OPTS_ALIGN - len % TCP_OPTS_ALIGN) % TCP_OPTS_ALIGN;
344   return len;
345 }
346
347 always_inline int
348 tcp_make_established_options (tcp_connection_t * tc, tcp_options_t * opts)
349 {
350   u8 len = 0;
351
352   opts->flags = 0;
353
354   if (tcp_opts_tstamp (&tc->rcv_opts))
355     {
356       opts->flags |= TCP_OPTS_FLAG_TSTAMP;
357       opts->tsval = tcp_time_now ();
358       opts->tsecr = tc->tsval_recent;
359       len += TCP_OPTION_LEN_TIMESTAMP;
360     }
361   if (tcp_opts_sack_permitted (&tc->rcv_opts))
362     {
363       if (vec_len (tc->snd_sacks))
364         {
365           opts->flags |= TCP_OPTS_FLAG_SACK;
366           opts->sacks = tc->snd_sacks;
367           opts->n_sack_blocks = clib_min (vec_len (tc->snd_sacks),
368                                           TCP_OPTS_MAX_SACK_BLOCKS);
369           len += 2 + TCP_OPTION_LEN_SACK_BLOCK * opts->n_sack_blocks;
370         }
371     }
372
373   /* Align to needed boundary */
374   len += (TCP_OPTS_ALIGN - len % TCP_OPTS_ALIGN) % TCP_OPTS_ALIGN;
375   return len;
376 }
377
378 always_inline int
379 tcp_make_options (tcp_connection_t * tc, tcp_options_t * opts,
380                   tcp_state_t state)
381 {
382   switch (state)
383     {
384     case TCP_STATE_ESTABLISHED:
385     case TCP_STATE_FIN_WAIT_1:
386       return tcp_make_established_options (tc, opts);
387     case TCP_STATE_SYN_RCVD:
388       return tcp_make_synack_options (tc, opts);
389     case TCP_STATE_SYN_SENT:
390       return tcp_make_syn_options (opts, tc->rcv_wscale);
391     default:
392       clib_warning ("Not handled!");
393       return 0;
394     }
395 }
396
397 /**
398  * Update snd_mss to reflect the effective segment size that we can send
399  * by taking into account all TCP options, including SACKs
400  */
401 void
402 tcp_update_snd_mss (tcp_connection_t * tc)
403 {
404   /* Compute options to be used for connection. These may be reused when
405    * sending data or to compute the effective mss (snd_mss) */
406   tc->snd_opts_len =
407     tcp_make_options (tc, &tc->snd_opts, TCP_STATE_ESTABLISHED);
408
409   /* XXX check if MTU has been updated */
410   tc->snd_mss = clib_min (tc->mss, tc->rcv_opts.mss) - tc->snd_opts_len;
411   ASSERT (tc->snd_mss > 0);
412 }
413
414 void
415 tcp_init_mss (tcp_connection_t * tc)
416 {
417   u16 default_min_mss = 536;
418   tcp_update_rcv_mss (tc);
419
420   /* TODO cache mss and consider PMTU discovery */
421   tc->snd_mss = clib_min (tc->rcv_opts.mss, tc->mss);
422
423   if (tc->snd_mss < 45)
424     {
425       clib_warning ("snd mss is 0");
426       /* Assume that at least the min default mss works */
427       tc->snd_mss = default_min_mss;
428       tc->rcv_opts.mss = default_min_mss;
429     }
430
431   /* We should have enough space for 40 bytes of options */
432   ASSERT (tc->snd_mss > 45);
433
434   /* If we use timestamp option, account for it */
435   if (tcp_opts_tstamp (&tc->rcv_opts))
436     tc->snd_mss -= TCP_OPTION_LEN_TIMESTAMP;
437 }
438
439 always_inline int
440 tcp_alloc_tx_buffers (tcp_main_t * tm, u8 thread_index, u32 n_free_buffers)
441 {
442   vlib_main_t *vm = vlib_get_main ();
443   u32 current_length = vec_len (tm->tx_buffers[thread_index]);
444   u32 n_allocated;
445
446   vec_validate (tm->tx_buffers[thread_index],
447                 current_length + n_free_buffers - 1);
448   n_allocated =
449     vlib_buffer_alloc (vm, &tm->tx_buffers[thread_index][current_length],
450                        n_free_buffers);
451   _vec_len (tm->tx_buffers[thread_index]) = current_length + n_allocated;
452   /* buffer shortage, report failure */
453   if (vec_len (tm->tx_buffers[thread_index]) == 0)
454     {
455       clib_warning ("out of buffers");
456       return -1;
457     }
458   return 0;
459 }
460
461 always_inline int
462 tcp_get_free_buffer_index (tcp_main_t * tm, u32 * bidx)
463 {
464   u32 *my_tx_buffers;
465   u32 thread_index = vlib_get_thread_index ();
466   if (PREDICT_FALSE (vec_len (tm->tx_buffers[thread_index]) == 0))
467     {
468       if (tcp_alloc_tx_buffers (tm, thread_index, VLIB_FRAME_SIZE))
469         return -1;
470     }
471   my_tx_buffers = tm->tx_buffers[thread_index];
472   *bidx = my_tx_buffers[vec_len (my_tx_buffers) - 1];
473   _vec_len (my_tx_buffers) -= 1;
474   return 0;
475 }
476
477 always_inline void
478 tcp_return_buffer (tcp_main_t * tm)
479 {
480   _vec_len (tm->tx_buffers[vlib_get_thread_index ()]) += 1;
481 }
482
483 always_inline void *
484 tcp_reuse_buffer (vlib_main_t * vm, vlib_buffer_t * b)
485 {
486   if (b->flags & VLIB_BUFFER_NEXT_PRESENT)
487     vlib_buffer_free_one (vm, b->next_buffer);
488   /* Zero all flags but free list index and trace flag */
489   b->flags &= VLIB_BUFFER_NEXT_PRESENT - 1;
490   b->current_data = 0;
491   b->current_length = 0;
492   b->total_length_not_including_first_buffer = 0;
493   vnet_buffer (b)->tcp.flags = 0;
494
495   /* Leave enough space for headers */
496   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
497 }
498
499 always_inline void *
500 tcp_init_buffer (vlib_main_t * vm, vlib_buffer_t * b)
501 {
502   ASSERT ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0);
503   b->flags &= VLIB_BUFFER_FREE_LIST_INDEX_MASK;
504   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
505   b->total_length_not_including_first_buffer = 0;
506   vnet_buffer (b)->tcp.flags = 0;
507
508   /* Leave enough space for headers */
509   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
510 }
511
512 /**
513  * Prepare ACK
514  */
515 void
516 tcp_make_ack_i (tcp_connection_t * tc, vlib_buffer_t * b, tcp_state_t state,
517                 u8 flags)
518 {
519   tcp_options_t _snd_opts, *snd_opts = &_snd_opts;
520   u8 tcp_opts_len, tcp_hdr_opts_len;
521   tcp_header_t *th;
522   u16 wnd;
523
524   wnd = tcp_window_to_advertise (tc, state);
525
526   /* Make and write options */
527   tcp_opts_len = tcp_make_established_options (tc, snd_opts);
528   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
529
530   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
531                              tc->rcv_nxt, tcp_hdr_opts_len, flags, wnd);
532
533   tcp_options_write ((u8 *) (th + 1), snd_opts);
534   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
535 }
536
537 /**
538  * Convert buffer to ACK
539  */
540 void
541 tcp_make_ack (tcp_connection_t * tc, vlib_buffer_t * b)
542 {
543   vlib_main_t *vm = vlib_get_main ();
544
545   tcp_reuse_buffer (vm, b);
546   tcp_make_ack_i (tc, b, TCP_STATE_ESTABLISHED, TCP_FLAG_ACK);
547   TCP_EVT_DBG (TCP_EVT_ACK_SENT, tc);
548   vnet_buffer (b)->tcp.flags = TCP_BUF_FLAG_ACK;
549   tc->rcv_las = tc->rcv_nxt;
550 }
551
552 /**
553  * Convert buffer to FIN-ACK
554  */
555 void
556 tcp_make_fin (tcp_connection_t * tc, vlib_buffer_t * b)
557 {
558   vlib_main_t *vm = vlib_get_main ();
559   u8 flags = 0;
560
561   tcp_reuse_buffer (vm, b);
562
563   flags = TCP_FLAG_FIN | TCP_FLAG_ACK;
564   tcp_make_ack_i (tc, b, TCP_STATE_ESTABLISHED, flags);
565
566   /* Reset flags, make sure ack is sent */
567   vnet_buffer (b)->tcp.flags &= ~TCP_BUF_FLAG_DUPACK;
568 }
569
570 /**
571  * Convert buffer to SYN
572  */
573 void
574 tcp_make_syn (tcp_connection_t * tc, vlib_buffer_t * b)
575 {
576   u8 tcp_hdr_opts_len, tcp_opts_len;
577   tcp_header_t *th;
578   u16 initial_wnd;
579   tcp_options_t snd_opts;
580
581   initial_wnd = tcp_initial_window_to_advertise (tc);
582
583   /* Make and write options */
584   memset (&snd_opts, 0, sizeof (snd_opts));
585   tcp_opts_len = tcp_make_syn_options (&snd_opts, tc->rcv_wscale);
586   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
587
588   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->iss,
589                              tc->rcv_nxt, tcp_hdr_opts_len, TCP_FLAG_SYN,
590                              initial_wnd);
591   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
592   tcp_options_write ((u8 *) (th + 1), &snd_opts);
593
594   tcp_timer_update (tc, TCP_TIMER_RETRANSMIT_SYN,
595                     tc->rto * TCP_TO_TIMER_TICK);
596 }
597
598 /**
599  * Convert buffer to SYN-ACK
600  */
601 void
602 tcp_make_synack (tcp_connection_t * tc, vlib_buffer_t * b)
603 {
604   vlib_main_t *vm = vlib_get_main ();
605   tcp_options_t _snd_opts, *snd_opts = &_snd_opts;
606   u8 tcp_opts_len, tcp_hdr_opts_len;
607   tcp_header_t *th;
608   u16 initial_wnd;
609
610   memset (snd_opts, 0, sizeof (*snd_opts));
611   tcp_reuse_buffer (vm, b);
612
613   initial_wnd = tcp_initial_window_to_advertise (tc);
614   tcp_opts_len = tcp_make_synack_options (tc, snd_opts);
615   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
616
617   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->iss,
618                              tc->rcv_nxt, tcp_hdr_opts_len,
619                              TCP_FLAG_SYN | TCP_FLAG_ACK, initial_wnd);
620   tcp_options_write ((u8 *) (th + 1), snd_opts);
621
622   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
623   vnet_buffer (b)->tcp.flags = TCP_BUF_FLAG_ACK;
624
625   /* Init retransmit timer. Use update instead of set because of
626    * retransmissions */
627   tcp_retransmit_timer_force_update (tc);
628   TCP_EVT_DBG (TCP_EVT_SYNACK_SENT, tc);
629 }
630
631 always_inline void
632 tcp_enqueue_to_ip_lookup_i (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
633                             u8 is_ip4, u8 flush)
634 {
635   tcp_main_t *tm = vnet_get_tcp_main ();
636   u32 thread_index = vlib_get_thread_index ();
637   u32 *to_next, next_index;
638   vlib_frame_t *f;
639
640   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
641   b->error = 0;
642
643   /* Default FIB for now */
644   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
645
646   /* Send to IP lookup */
647   next_index = is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
648   if (VLIB_BUFFER_TRACE_TRAJECTORY > 0)
649     {
650       b->pre_data[0] = 2;
651       b->pre_data[1] = next_index;
652     }
653
654   f = tm->ip_lookup_tx_frames[!is_ip4][thread_index];
655   if (!f)
656     {
657       f = vlib_get_frame_to_node (vm, next_index);
658       ASSERT (f);
659       tm->ip_lookup_tx_frames[!is_ip4][thread_index] = f;
660     }
661
662   to_next = vlib_frame_vector_args (f);
663   to_next[f->n_vectors] = bi;
664   f->n_vectors += 1;
665   if (flush || f->n_vectors == VLIB_FRAME_SIZE)
666     {
667       vlib_put_frame_to_node (vm, next_index, f);
668       tm->ip_lookup_tx_frames[!is_ip4][thread_index] = 0;
669     }
670 }
671
672 always_inline void
673 tcp_enqueue_to_ip_lookup_now (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
674                               u8 is_ip4)
675 {
676   tcp_enqueue_to_ip_lookup_i (vm, b, bi, is_ip4, 1);
677 }
678
679 always_inline void
680 tcp_enqueue_to_ip_lookup (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
681                           u8 is_ip4)
682 {
683   tcp_enqueue_to_ip_lookup_i (vm, b, bi, is_ip4, 0);
684 }
685
686 always_inline void
687 tcp_enqueue_to_output_i (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
688                          u8 is_ip4, u8 flush)
689 {
690   tcp_main_t *tm = vnet_get_tcp_main ();
691   u32 thread_index = vlib_get_thread_index ();
692   u32 *to_next, next_index;
693   vlib_frame_t *f;
694
695   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
696   b->error = 0;
697
698   /* Decide where to send the packet */
699   next_index = is_ip4 ? tcp4_output_node.index : tcp6_output_node.index;
700   if (VLIB_BUFFER_TRACE_TRAJECTORY > 0)
701     {
702       b->pre_data[0] = 1;
703       b->pre_data[1] = next_index;
704     }
705
706   /* Get frame to v4/6 output node */
707   f = tm->tx_frames[!is_ip4][thread_index];
708   if (!f)
709     {
710       f = vlib_get_frame_to_node (vm, next_index);
711       ASSERT (f);
712       tm->tx_frames[!is_ip4][thread_index] = f;
713     }
714   to_next = vlib_frame_vector_args (f);
715   to_next[f->n_vectors] = bi;
716   f->n_vectors += 1;
717   if (flush || f->n_vectors == VLIB_FRAME_SIZE)
718     {
719       vlib_put_frame_to_node (vm, next_index, f);
720       tm->tx_frames[!is_ip4][thread_index] = 0;
721     }
722 }
723
724 always_inline void
725 tcp_enqueue_to_output (vlib_main_t * vm, vlib_buffer_t * b, u32 bi, u8 is_ip4)
726 {
727   tcp_enqueue_to_output_i (vm, b, bi, is_ip4, 0);
728 }
729
730 always_inline void
731 tcp_enqueue_to_output_now (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
732                            u8 is_ip4)
733 {
734   tcp_enqueue_to_output_i (vm, b, bi, is_ip4, 1);
735 }
736
737 int
738 tcp_make_reset_in_place (vlib_main_t * vm, vlib_buffer_t * b0,
739                          tcp_state_t state, u8 thread_index, u8 is_ip4)
740 {
741   ip4_header_t *ih4;
742   ip6_header_t *ih6;
743   tcp_header_t *th0;
744   ip4_address_t src_ip40, dst_ip40;
745   ip6_address_t src_ip60, dst_ip60;
746   u16 src_port, dst_port;
747   u32 tmp;
748   u32 seq, ack;
749   u8 flags;
750
751   /* Find IP and TCP headers */
752   th0 = tcp_buffer_hdr (b0);
753
754   /* Save src and dst ip */
755   if (is_ip4)
756     {
757       ih4 = vlib_buffer_get_current (b0);
758       ASSERT ((ih4->ip_version_and_header_length & 0xF0) == 0x40);
759       src_ip40.as_u32 = ih4->src_address.as_u32;
760       dst_ip40.as_u32 = ih4->dst_address.as_u32;
761     }
762   else
763     {
764       ih6 = vlib_buffer_get_current (b0);
765       ASSERT ((ih6->ip_version_traffic_class_and_flow_label & 0xF0) == 0x60);
766       clib_memcpy (&src_ip60, &ih6->src_address, sizeof (ip6_address_t));
767       clib_memcpy (&dst_ip60, &ih6->dst_address, sizeof (ip6_address_t));
768     }
769
770   src_port = th0->src_port;
771   dst_port = th0->dst_port;
772
773   /* Try to determine what/why we're actually resetting */
774   if (state == TCP_STATE_CLOSED)
775     {
776       if (!tcp_syn (th0))
777         return -1;
778
779       tmp = clib_net_to_host_u32 (th0->seq_number);
780
781       /* Got a SYN for no listener. */
782       flags = TCP_FLAG_RST | TCP_FLAG_ACK;
783       ack = clib_host_to_net_u32 (tmp + 1);
784       seq = 0;
785     }
786   else
787     {
788       flags = TCP_FLAG_RST;
789       seq = th0->ack_number;
790       ack = 0;
791     }
792
793   tcp_reuse_buffer (vm, b0);
794   th0 = vlib_buffer_push_tcp_net_order (b0, dst_port, src_port, seq, ack,
795                                         sizeof (tcp_header_t), flags, 0);
796
797   if (is_ip4)
798     {
799       ih4 = vlib_buffer_push_ip4 (vm, b0, &dst_ip40, &src_ip40,
800                                   IP_PROTOCOL_TCP, 1);
801       th0->checksum = ip4_tcp_udp_compute_checksum (vm, b0, ih4);
802     }
803   else
804     {
805       int bogus = ~0;
806       ih6 = vlib_buffer_push_ip6 (vm, b0, &dst_ip60, &src_ip60,
807                                   IP_PROTOCOL_TCP);
808       th0->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b0, ih6, &bogus);
809       ASSERT (!bogus);
810     }
811
812   return 0;
813 }
814
815 /**
816  *  Send reset without reusing existing buffer
817  *
818  *  It extracts connection info out of original packet
819  */
820 void
821 tcp_send_reset_w_pkt (tcp_connection_t * tc, vlib_buffer_t * pkt, u8 is_ip4)
822 {
823   vlib_buffer_t *b;
824   u32 bi;
825   tcp_main_t *tm = vnet_get_tcp_main ();
826   vlib_main_t *vm = vlib_get_main ();
827   u8 tcp_hdr_len, flags = 0;
828   tcp_header_t *th, *pkt_th;
829   u32 seq, ack;
830   ip4_header_t *ih4, *pkt_ih4;
831   ip6_header_t *ih6, *pkt_ih6;
832
833   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
834     return;
835
836   b = vlib_get_buffer (vm, bi);
837   tcp_init_buffer (vm, b);
838
839   /* Make and write options */
840   tcp_hdr_len = sizeof (tcp_header_t);
841
842   if (is_ip4)
843     {
844       pkt_ih4 = vlib_buffer_get_current (pkt);
845       pkt_th = ip4_next_header (pkt_ih4);
846     }
847   else
848     {
849       pkt_ih6 = vlib_buffer_get_current (pkt);
850       pkt_th = ip6_next_header (pkt_ih6);
851     }
852
853   if (tcp_ack (pkt_th))
854     {
855       flags = TCP_FLAG_RST;
856       seq = pkt_th->ack_number;
857       ack = (tc && tc->state >= TCP_STATE_SYN_RCVD) ? tc->rcv_nxt : 0;
858     }
859   else
860     {
861       flags = TCP_FLAG_RST | TCP_FLAG_ACK;
862       seq = 0;
863       ack = clib_host_to_net_u32 (vnet_buffer (pkt)->tcp.seq_end);
864     }
865
866   th = vlib_buffer_push_tcp_net_order (b, pkt_th->dst_port, pkt_th->src_port,
867                                        seq, ack, tcp_hdr_len, flags, 0);
868
869   /* Swap src and dst ip */
870   if (is_ip4)
871     {
872       ASSERT ((pkt_ih4->ip_version_and_header_length & 0xF0) == 0x40);
873       ih4 = vlib_buffer_push_ip4 (vm, b, &pkt_ih4->dst_address,
874                                   &pkt_ih4->src_address, IP_PROTOCOL_TCP, 1);
875       th->checksum = ip4_tcp_udp_compute_checksum (vm, b, ih4);
876     }
877   else
878     {
879       int bogus = ~0;
880       ASSERT ((pkt_ih6->ip_version_traffic_class_and_flow_label & 0xF0) ==
881               0x60);
882       ih6 = vlib_buffer_push_ip6 (vm, b, &pkt_ih6->dst_address,
883                                   &pkt_ih6->src_address, IP_PROTOCOL_TCP);
884       th->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b, ih6, &bogus);
885       ASSERT (!bogus);
886     }
887
888   tcp_enqueue_to_ip_lookup_now (vm, b, bi, is_ip4);
889   TCP_EVT_DBG (TCP_EVT_RST_SENT, tc);
890 }
891
892 /**
893  * Build and set reset packet for connection
894  */
895 void
896 tcp_send_reset (tcp_connection_t * tc)
897 {
898   vlib_main_t *vm = vlib_get_main ();
899   tcp_main_t *tm = vnet_get_tcp_main ();
900   vlib_buffer_t *b;
901   u32 bi;
902   tcp_header_t *th;
903   u16 tcp_hdr_opts_len, advertise_wnd, opts_write_len;
904   u8 flags;
905
906   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
907     return;
908   b = vlib_get_buffer (vm, bi);
909   tcp_init_buffer (vm, b);
910
911   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
912   tcp_hdr_opts_len = tc->snd_opts_len + sizeof (tcp_header_t);
913   advertise_wnd = tcp_window_to_advertise (tc, TCP_STATE_ESTABLISHED);
914   flags = TCP_FLAG_RST;
915   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
916                              tc->rcv_nxt, tcp_hdr_opts_len, flags,
917                              advertise_wnd);
918   opts_write_len = tcp_options_write ((u8 *) (th + 1), &tc->snd_opts);
919   ASSERT (opts_write_len == tc->snd_opts_len);
920   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
921   tcp_enqueue_to_output_now (vm, b, bi, tc->c_is_ip4);
922 }
923
924 void
925 tcp_push_ip_hdr (tcp_main_t * tm, tcp_connection_t * tc, vlib_buffer_t * b)
926 {
927   tcp_header_t *th = vlib_buffer_get_current (b);
928   vlib_main_t *vm = vlib_get_main ();
929   if (tc->c_is_ip4)
930     {
931       ip4_header_t *ih;
932       ih = vlib_buffer_push_ip4 (vm, b, &tc->c_lcl_ip4,
933                                  &tc->c_rmt_ip4, IP_PROTOCOL_TCP, 1);
934       th->checksum = ip4_tcp_udp_compute_checksum (vm, b, ih);
935     }
936   else
937     {
938       ip6_header_t *ih;
939       int bogus = ~0;
940
941       ih = vlib_buffer_push_ip6 (vm, b, &tc->c_lcl_ip6,
942                                  &tc->c_rmt_ip6, IP_PROTOCOL_TCP);
943       th->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b, ih, &bogus);
944       ASSERT (!bogus);
945     }
946 }
947
948 /**
949  *  Send SYN
950  *
951  *  Builds a SYN packet for a half-open connection and sends it to ipx_lookup.
952  *  The packet is not forwarded through tcpx_output to avoid doing lookups
953  *  in the half_open pool.
954  */
955 void
956 tcp_send_syn (tcp_connection_t * tc)
957 {
958   vlib_buffer_t *b;
959   u32 bi;
960   tcp_main_t *tm = vnet_get_tcp_main ();
961   vlib_main_t *vm = vlib_get_main ();
962
963   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
964     return;
965
966   b = vlib_get_buffer (vm, bi);
967   tcp_init_buffer (vm, b);
968   tcp_make_syn (tc, b);
969
970   /* Measure RTT with this */
971   tc->rtt_ts = tcp_time_now ();
972   tc->rtt_seq = tc->snd_nxt;
973   tc->rto_boff = 0;
974
975   /* Set the connection establishment timer */
976   tcp_timer_set (tc, TCP_TIMER_ESTABLISH, TCP_ESTABLISH_TIME);
977
978   tcp_push_ip_hdr (tm, tc, b);
979   tcp_enqueue_to_ip_lookup (vm, b, bi, tc->c_is_ip4);
980   TCP_EVT_DBG (TCP_EVT_SYN_SENT, tc);
981 }
982
983 /**
984  * Flush tx frame populated by retransmits and timer pops
985  */
986 void
987 tcp_flush_frame_to_output (vlib_main_t * vm, u8 thread_index, u8 is_ip4)
988 {
989   if (tcp_main.tx_frames[!is_ip4][thread_index])
990     {
991       u32 next_index;
992       next_index = is_ip4 ? tcp4_output_node.index : tcp6_output_node.index;
993       vlib_put_frame_to_node (vm, next_index,
994                               tcp_main.tx_frames[!is_ip4][thread_index]);
995       tcp_main.tx_frames[!is_ip4][thread_index] = 0;
996     }
997 }
998
999 /**
1000  * Flush ip lookup tx frames populated by timer pops
1001  */
1002 always_inline void
1003 tcp_flush_frame_to_ip_lookup (vlib_main_t * vm, u8 thread_index, u8 is_ip4)
1004 {
1005   if (tcp_main.ip_lookup_tx_frames[!is_ip4][thread_index])
1006     {
1007       u32 next_index;
1008       next_index = is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
1009       vlib_put_frame_to_node (vm, next_index,
1010                               tcp_main.ip_lookup_tx_frames[!is_ip4]
1011                               [thread_index]);
1012       tcp_main.ip_lookup_tx_frames[!is_ip4][thread_index] = 0;
1013     }
1014 }
1015
1016 /**
1017  * Flush v4 and v6 tcp and ip-lookup tx frames for thread index
1018  */
1019 void
1020 tcp_flush_frames_to_output (u8 thread_index)
1021 {
1022   vlib_main_t *vm = vlib_get_main ();
1023   tcp_flush_frame_to_output (vm, thread_index, 1);
1024   tcp_flush_frame_to_output (vm, thread_index, 0);
1025   tcp_flush_frame_to_ip_lookup (vm, thread_index, 1);
1026   tcp_flush_frame_to_ip_lookup (vm, thread_index, 0);
1027 }
1028
1029 /**
1030  *  Send FIN
1031  */
1032 void
1033 tcp_send_fin (tcp_connection_t * tc)
1034 {
1035   tcp_main_t *tm = vnet_get_tcp_main ();
1036   vlib_main_t *vm = vlib_get_main ();
1037   vlib_buffer_t *b;
1038   u32 bi;
1039   u8 fin_snt = 0;
1040
1041
1042   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1043     return;
1044   b = vlib_get_buffer (vm, bi);
1045   fin_snt = tc->flags & TCP_CONN_FINSNT;
1046   if (fin_snt)
1047     tc->snd_nxt = tc->snd_una;
1048   tcp_make_fin (tc, b);
1049   tcp_enqueue_to_output_now (vm, b, bi, tc->c_is_ip4);
1050   if (!fin_snt)
1051     {
1052       tc->flags |= TCP_CONN_FINSNT;
1053       tc->flags &= ~TCP_CONN_FINPNDG;
1054       /* Account for the FIN */
1055       tc->snd_una_max += 1;
1056       tc->snd_nxt = tc->snd_una_max;
1057     }
1058   tcp_retransmit_timer_force_update (tc);
1059   TCP_EVT_DBG (TCP_EVT_FIN_SENT, tc);
1060 }
1061
1062 always_inline u8
1063 tcp_make_state_flags (tcp_connection_t * tc, tcp_state_t next_state)
1064 {
1065   switch (next_state)
1066     {
1067     case TCP_STATE_ESTABLISHED:
1068       return TCP_FLAG_ACK;
1069     case TCP_STATE_SYN_RCVD:
1070       return TCP_FLAG_SYN | TCP_FLAG_ACK;
1071     case TCP_STATE_SYN_SENT:
1072       return TCP_FLAG_SYN;
1073     case TCP_STATE_LAST_ACK:
1074     case TCP_STATE_FIN_WAIT_1:
1075       if (tc->snd_nxt + 1 < tc->snd_una_max)
1076         return TCP_FLAG_ACK;
1077       else
1078         return TCP_FLAG_FIN;
1079     default:
1080       clib_warning ("Shouldn't be here!");
1081     }
1082   return 0;
1083 }
1084
1085 /**
1086  * Push TCP header and update connection variables
1087  */
1088 static void
1089 tcp_push_hdr_i (tcp_connection_t * tc, vlib_buffer_t * b,
1090                 tcp_state_t next_state, u8 compute_opts)
1091 {
1092   u32 advertise_wnd, data_len;
1093   u8 tcp_hdr_opts_len, opts_write_len, flags;
1094   tcp_header_t *th;
1095
1096   data_len = b->current_length + b->total_length_not_including_first_buffer;
1097   ASSERT (!b->total_length_not_including_first_buffer
1098           || (b->flags & VLIB_BUFFER_NEXT_PRESENT));
1099   vnet_buffer (b)->tcp.flags = 0;
1100
1101   if (compute_opts)
1102     tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1103
1104   tcp_hdr_opts_len = tc->snd_opts_len + sizeof (tcp_header_t);
1105   advertise_wnd = tcp_window_to_advertise (tc, next_state);
1106   flags = tcp_make_state_flags (tc, next_state);
1107
1108   /* Push header and options */
1109   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
1110                              tc->rcv_nxt, tcp_hdr_opts_len, flags,
1111                              advertise_wnd);
1112   opts_write_len = tcp_options_write ((u8 *) (th + 1), &tc->snd_opts);
1113
1114   ASSERT (opts_write_len == tc->snd_opts_len);
1115   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
1116
1117   /*
1118    * Update connection variables
1119    */
1120
1121   tc->snd_nxt += data_len;
1122   tc->rcv_las = tc->rcv_nxt;
1123
1124   /* TODO this is updated in output as well ... */
1125   if (seq_gt (tc->snd_nxt, tc->snd_una_max))
1126     {
1127       tc->snd_una_max = tc->snd_nxt;
1128       tcp_validate_txf_size (tc, tc->snd_una_max - tc->snd_una);
1129     }
1130
1131   TCP_EVT_DBG (TCP_EVT_PKTIZE, tc);
1132 }
1133
1134 void
1135 tcp_send_ack (tcp_connection_t * tc)
1136 {
1137   tcp_main_t *tm = vnet_get_tcp_main ();
1138   vlib_main_t *vm = vlib_get_main ();
1139
1140   vlib_buffer_t *b;
1141   u32 bi;
1142
1143   /* Get buffer */
1144   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1145     return;
1146   b = vlib_get_buffer (vm, bi);
1147
1148   /* Fill in the ACK */
1149   tcp_make_ack (tc, b);
1150   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1151 }
1152
1153 /**
1154  * Delayed ack timer handler
1155  *
1156  * Sends delayed ACK when timer expires
1157  */
1158 void
1159 tcp_timer_delack_handler (u32 index)
1160 {
1161   u32 thread_index = vlib_get_thread_index ();
1162   tcp_connection_t *tc;
1163
1164   tc = tcp_connection_get (index, thread_index);
1165   tc->timers[TCP_TIMER_DELACK] = TCP_TIMER_HANDLE_INVALID;
1166   tcp_send_ack (tc);
1167 }
1168
1169 /**
1170  * Build a retransmit segment
1171  *
1172  * @return the number of bytes in the segment or 0 if there's nothing to
1173  *         retransmit
1174  */
1175 u32
1176 tcp_prepare_retransmit_segment (tcp_connection_t * tc, u32 offset,
1177                                 u32 max_deq_bytes, vlib_buffer_t ** b)
1178 {
1179   tcp_main_t *tm = vnet_get_tcp_main ();
1180   vlib_main_t *vm = vlib_get_main ();
1181   int n_bytes = 0;
1182   u32 start, bi, available_bytes, seg_size;
1183   u8 *data;
1184
1185   ASSERT (tc->state >= TCP_STATE_ESTABLISHED);
1186   ASSERT (max_deq_bytes != 0);
1187
1188   /*
1189    * Make sure we can retransmit something
1190    */
1191   available_bytes = stream_session_tx_fifo_max_dequeue (&tc->connection);
1192   ASSERT (available_bytes >= offset);
1193   available_bytes -= offset;
1194   if (!available_bytes)
1195     return 0;
1196   max_deq_bytes = clib_min (tc->snd_mss, max_deq_bytes);
1197   max_deq_bytes = clib_min (available_bytes, max_deq_bytes);
1198
1199   /* Start is beyond snd_congestion */
1200   start = tc->snd_una + offset;
1201   if (seq_geq (start, tc->snd_congestion))
1202     goto done;
1203
1204   /* Don't overshoot snd_congestion */
1205   if (seq_gt (start + max_deq_bytes, tc->snd_congestion))
1206     {
1207       max_deq_bytes = tc->snd_congestion - start;
1208       if (max_deq_bytes == 0)
1209         goto done;
1210     }
1211
1212   seg_size = max_deq_bytes + MAX_HDRS_LEN;
1213
1214   /*
1215    * Prepare options
1216    */
1217   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1218
1219   /*
1220    * Allocate and fill in buffer(s)
1221    */
1222
1223   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1224     return 0;
1225   *b = vlib_get_buffer (vm, bi);
1226   data = tcp_init_buffer (vm, *b);
1227
1228   /* Easy case, buffer size greater than mss */
1229   if (PREDICT_TRUE (seg_size <= tm->bytes_per_buffer))
1230     {
1231       n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1232                                            max_deq_bytes);
1233       ASSERT (n_bytes == max_deq_bytes);
1234       b[0]->current_length = n_bytes;
1235       tcp_push_hdr_i (tc, *b, tc->state, 0);
1236     }
1237   /* Split mss into multiple buffers */
1238   else
1239     {
1240       u32 chain_bi = ~0, n_bufs_per_seg;
1241       u32 thread_index = vlib_get_thread_index ();
1242       u16 n_peeked, len_to_deq, available_bufs;
1243       vlib_buffer_t *chain_b, *prev_b;
1244       int i;
1245
1246       n_bufs_per_seg = ceil ((double) seg_size / tm->bytes_per_buffer);
1247
1248       /* Make sure we have enough buffers */
1249       available_bufs = vec_len (tm->tx_buffers[thread_index]);
1250       if (n_bufs_per_seg > available_bufs)
1251         {
1252           if (tcp_alloc_tx_buffers (tm, thread_index,
1253                                     VLIB_FRAME_SIZE - available_bufs))
1254             {
1255               tcp_return_buffer (tm);
1256               *b = 0;
1257               return 0;
1258             }
1259         }
1260
1261       n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1262                                            tm->bytes_per_buffer -
1263                                            MAX_HDRS_LEN);
1264       b[0]->current_length = n_bytes;
1265       b[0]->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
1266       b[0]->total_length_not_including_first_buffer = 0;
1267       max_deq_bytes -= n_bytes;
1268
1269       chain_b = *b;
1270       for (i = 1; i < n_bufs_per_seg; i++)
1271         {
1272           prev_b = chain_b;
1273           len_to_deq = clib_min (max_deq_bytes, tm->bytes_per_buffer);
1274           tcp_get_free_buffer_index (tm, &chain_bi);
1275           ASSERT (chain_bi != (u32) ~ 0);
1276           chain_b = vlib_get_buffer (vm, chain_bi);
1277           chain_b->current_data = 0;
1278           data = vlib_buffer_get_current (chain_b);
1279           n_peeked = stream_session_peek_bytes (&tc->connection, data,
1280                                                 offset + n_bytes, len_to_deq);
1281           ASSERT (n_peeked == len_to_deq);
1282           n_bytes += n_peeked;
1283           chain_b->current_length = n_peeked;
1284           chain_b->flags &= VLIB_BUFFER_FREE_LIST_INDEX_MASK;
1285           chain_b->next_buffer = 0;
1286
1287           /* update previous buffer */
1288           prev_b->next_buffer = chain_bi;
1289           prev_b->flags |= VLIB_BUFFER_NEXT_PRESENT;
1290
1291           max_deq_bytes -= n_peeked;
1292           b[0]->total_length_not_including_first_buffer += n_peeked;
1293         }
1294
1295       tcp_push_hdr_i (tc, *b, tc->state, 0);
1296     }
1297
1298   ASSERT (n_bytes > 0);
1299   ASSERT (((*b)->current_data + (*b)->current_length) <=
1300           tm->bytes_per_buffer);
1301
1302   if (tcp_in_fastrecovery (tc))
1303     tc->snd_rxt_bytes += n_bytes;
1304
1305 done:
1306   TCP_EVT_DBG (TCP_EVT_CC_RTX, tc, offset, n_bytes);
1307   return n_bytes;
1308 }
1309
1310 /**
1311  * Reset congestion control, switch cwnd to loss window and try again.
1312  */
1313 static void
1314 tcp_rtx_timeout_cc (tcp_connection_t * tc)
1315 {
1316   tc->prev_ssthresh = tc->ssthresh;
1317   tc->prev_cwnd = tc->cwnd;
1318
1319   /* Cleanly recover cc (also clears up fast retransmit) */
1320   if (tcp_in_fastrecovery (tc))
1321     tcp_cc_fastrecovery_exit (tc);
1322
1323   /* Start again from the beginning */
1324   tc->ssthresh = clib_max (tcp_flight_size (tc) / 2, 2 * tc->snd_mss);
1325   tc->cwnd = tcp_loss_wnd (tc);
1326   tc->snd_congestion = tc->snd_una_max;
1327
1328   tcp_recovery_on (tc);
1329 }
1330
1331 static void
1332 tcp_timer_retransmit_handler_i (u32 index, u8 is_syn)
1333 {
1334   tcp_main_t *tm = vnet_get_tcp_main ();
1335   vlib_main_t *vm = vlib_get_main ();
1336   u32 thread_index = vlib_get_thread_index ();
1337   tcp_connection_t *tc;
1338   vlib_buffer_t *b = 0;
1339   u32 bi, n_bytes;
1340
1341   if (is_syn)
1342     {
1343       tc = tcp_half_open_connection_get (index);
1344       /* Note: the connection may have transitioned to ESTABLISHED... */
1345       if (PREDICT_FALSE (tc == 0))
1346         return;
1347       tc->timers[TCP_TIMER_RETRANSMIT_SYN] = TCP_TIMER_HANDLE_INVALID;
1348     }
1349   else
1350     {
1351       tc = tcp_connection_get (index, thread_index);
1352       /* Note: the connection may have been closed and pool_put */
1353       if (PREDICT_FALSE (tc == 0))
1354         return;
1355       tc->timers[TCP_TIMER_RETRANSMIT] = TCP_TIMER_HANDLE_INVALID;
1356     }
1357
1358   if (tc->state >= TCP_STATE_ESTABLISHED)
1359     {
1360       /* Lost FIN, retransmit and return */
1361       if (tcp_is_lost_fin (tc))
1362         {
1363           tcp_send_fin (tc);
1364           return;
1365         }
1366
1367       /* We're not in recovery so make sure rto_boff is 0 */
1368       if (!tcp_in_recovery (tc) && tc->rto_boff > 0)
1369         {
1370           tc->rto_boff = 0;
1371           tcp_update_rto (tc);
1372         }
1373
1374       /* Increment RTO backoff (also equal to number of retries) and go back
1375        * to first un-acked byte  */
1376       tc->rto_boff += 1;
1377       tc->snd_nxt = tc->snd_una;
1378
1379       /* First retransmit timeout */
1380       if (tc->rto_boff == 1)
1381         tcp_rtx_timeout_cc (tc);
1382
1383       /* Exponential backoff */
1384       tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1385
1386       TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 1);
1387
1388       /* Send one segment. Note that n_bytes may be zero due to buffer shortfall  */
1389       n_bytes = tcp_prepare_retransmit_segment (tc, 0, tc->snd_mss, &b);
1390
1391       /* TODO be less aggressive about this */
1392       scoreboard_clear (&tc->sack_sb);
1393
1394       if (n_bytes == 0)
1395         {
1396           ASSERT (!b);
1397           if (tc->snd_una == tc->snd_una_max)
1398             return;
1399           ASSERT (tc->rto_boff > 1 && tc->snd_una == tc->snd_congestion);
1400           clib_warning ("retransmit fail: %U", format_tcp_connection, tc, 2);
1401           /* Try again eventually */
1402           tcp_retransmit_timer_set (tc);
1403           return;
1404         }
1405
1406       bi = vlib_get_buffer_index (vm, b);
1407
1408       /* For first retransmit, record timestamp (Eifel detection RFC3522) */
1409       if (tc->rto_boff == 1)
1410         tc->snd_rxt_ts = tcp_time_now ();
1411
1412       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1413       tcp_retransmit_timer_update (tc);
1414     }
1415   /* Retransmit for SYN */
1416   else if (tc->state == TCP_STATE_SYN_SENT)
1417     {
1418       /* Half-open connection actually moved to established but we were
1419        * waiting for syn retransmit to pop to call cleanup from the right
1420        * thread. */
1421       if (tc->flags & TCP_CONN_HALF_OPEN_DONE)
1422         {
1423           if (tcp_half_open_connection_cleanup (tc))
1424             {
1425               clib_warning ("could not remove half-open connection");
1426               ASSERT (0);
1427             }
1428           return;
1429         }
1430
1431       /* Try without increasing RTO a number of times. If this fails,
1432        * start growing RTO exponentially */
1433       tc->rto_boff += 1;
1434       if (tc->rto_boff > TCP_RTO_SYN_RETRIES)
1435         tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1436
1437       if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1438         return;
1439
1440       b = vlib_get_buffer (vm, bi);
1441       tcp_init_buffer (vm, b);
1442       tcp_make_syn (tc, b);
1443
1444       tc->rtt_ts = 0;
1445       TCP_EVT_DBG (TCP_EVT_SYN_RXT, tc, 0);
1446
1447       /* This goes straight to ipx_lookup. Retransmit timer set already */
1448       tcp_push_ip_hdr (tm, tc, b);
1449       tcp_enqueue_to_ip_lookup (vm, b, bi, tc->c_is_ip4);
1450     }
1451   /* Retransmit SYN-ACK */
1452   else if (tc->state == TCP_STATE_SYN_RCVD)
1453     {
1454       tc->rto_boff += 1;
1455       if (tc->rto_boff > TCP_RTO_SYN_RETRIES)
1456         tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1457       tc->rtt_ts = 0;
1458
1459       if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1460         return;
1461
1462       b = vlib_get_buffer (vm, bi);
1463       tcp_make_synack (tc, b);
1464       TCP_EVT_DBG (TCP_EVT_SYN_RXT, tc, 1);
1465
1466       /* Retransmit timer already updated, just enqueue to output */
1467       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1468     }
1469   else
1470     {
1471       ASSERT (tc->state == TCP_STATE_CLOSED);
1472       TCP_DBG ("connection state: %d", tc->state);
1473       return;
1474     }
1475 }
1476
1477 void
1478 tcp_timer_retransmit_handler (u32 index)
1479 {
1480   tcp_timer_retransmit_handler_i (index, 0);
1481 }
1482
1483 void
1484 tcp_timer_retransmit_syn_handler (u32 index)
1485 {
1486   tcp_timer_retransmit_handler_i (index, 1);
1487 }
1488
1489 /**
1490  * Got 0 snd_wnd from peer, try to do something about it.
1491  *
1492  */
1493 void
1494 tcp_timer_persist_handler (u32 index)
1495 {
1496   tcp_main_t *tm = vnet_get_tcp_main ();
1497   vlib_main_t *vm = vlib_get_main ();
1498   u32 thread_index = vlib_get_thread_index ();
1499   tcp_connection_t *tc;
1500   vlib_buffer_t *b;
1501   u32 bi, old_snd_nxt, max_snd_bytes, available_bytes, offset;
1502   int n_bytes = 0;
1503   u8 *data;
1504
1505   tc = tcp_connection_get_if_valid (index, thread_index);
1506
1507   if (!tc)
1508     return;
1509
1510   /* Make sure timer handle is set to invalid */
1511   tc->timers[TCP_TIMER_PERSIST] = TCP_TIMER_HANDLE_INVALID;
1512
1513   /* Problem already solved or worse */
1514   if (tc->state == TCP_STATE_CLOSED || tc->state > TCP_STATE_ESTABLISHED
1515       || tc->snd_wnd > tc->snd_mss || tcp_in_recovery (tc))
1516     return;
1517
1518   available_bytes = stream_session_tx_fifo_max_dequeue (&tc->connection);
1519   offset = tc->snd_una_max - tc->snd_una;
1520
1521   /* Reprogram persist if no new bytes available to send. We may have data
1522    * next time */
1523   if (!available_bytes)
1524     {
1525       tcp_persist_timer_set (tc);
1526       return;
1527     }
1528
1529   if (available_bytes <= offset)
1530     {
1531       ASSERT (tcp_timer_is_active (tc, TCP_TIMER_RETRANSMIT));
1532       return;
1533     }
1534
1535   /* Increment RTO backoff */
1536   tc->rto_boff += 1;
1537   tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1538
1539   /*
1540    * Try to force the first unsent segment (or buffer)
1541    */
1542   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1543     return;
1544   b = vlib_get_buffer (vm, bi);
1545   data = tcp_init_buffer (vm, b);
1546
1547   tcp_validate_txf_size (tc, offset);
1548   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1549   max_snd_bytes = clib_min (tc->snd_mss, tm->bytes_per_buffer - MAX_HDRS_LEN);
1550   n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1551                                        max_snd_bytes);
1552   b->current_length = n_bytes;
1553   ASSERT (n_bytes != 0 && (tc->snd_nxt == tc->snd_una_max || tc->rto_boff > 1
1554                            || tcp_timer_is_active (tc,
1555                                                    TCP_TIMER_RETRANSMIT)));
1556
1557   /* Allow updating of snd_una_max but don't update snd_nxt */
1558   old_snd_nxt = tc->snd_nxt;
1559   tcp_push_hdr_i (tc, b, tc->state, 0);
1560   tc->snd_nxt = old_snd_nxt;
1561   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1562
1563   /* Just sent new data, enable retransmit */
1564   tcp_retransmit_timer_update (tc);
1565 }
1566
1567 /**
1568  * Retransmit first unacked segment
1569  */
1570 void
1571 tcp_retransmit_first_unacked (tcp_connection_t * tc)
1572 {
1573   vlib_main_t *vm = vlib_get_main ();
1574   vlib_buffer_t *b;
1575   u32 bi, old_snd_nxt, n_bytes;
1576
1577   old_snd_nxt = tc->snd_nxt;
1578   tc->snd_nxt = tc->snd_una;
1579
1580   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 2);
1581   n_bytes = tcp_prepare_retransmit_segment (tc, 0, tc->snd_mss, &b);
1582   if (!n_bytes)
1583     return;
1584   bi = vlib_get_buffer_index (vm, b);
1585   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1586
1587   tc->snd_nxt = old_snd_nxt;
1588 }
1589
1590 /**
1591  * Do fast retransmit with SACKs
1592  */
1593 void
1594 tcp_fast_retransmit_sack (tcp_connection_t * tc)
1595 {
1596   vlib_main_t *vm = vlib_get_main ();
1597   u32 n_written = 0, offset, max_bytes;
1598   vlib_buffer_t *b = 0;
1599   sack_scoreboard_hole_t *hole;
1600   sack_scoreboard_t *sb;
1601   u32 bi, old_snd_nxt;
1602   int snd_space;
1603   u8 snd_limited = 0, can_rescue = 0;
1604
1605   ASSERT (tcp_in_fastrecovery (tc));
1606   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 0);
1607
1608   old_snd_nxt = tc->snd_nxt;
1609   sb = &tc->sack_sb;
1610   snd_space = tcp_available_snd_space (tc);
1611
1612   hole = scoreboard_get_hole (sb, sb->cur_rxt_hole);
1613   while (hole && snd_space > 0)
1614     {
1615       hole = scoreboard_next_rxt_hole (sb, hole,
1616                                        tcp_fastrecovery_sent_1_smss (tc),
1617                                        &can_rescue, &snd_limited);
1618       if (!hole)
1619         {
1620           if (!can_rescue || !(seq_lt (sb->rescue_rxt, tc->snd_una)
1621                                || seq_gt (sb->rescue_rxt,
1622                                           tc->snd_congestion)))
1623             break;
1624
1625           /* If rescue rxt undefined or less than snd_una then one segment of
1626            * up to SMSS octets that MUST include the highest outstanding
1627            * unSACKed sequence number SHOULD be returned, and RescueRxt set to
1628            * RecoveryPoint. HighRxt MUST NOT be updated.
1629            */
1630           max_bytes = clib_min (tc->snd_mss,
1631                                 tc->snd_congestion - tc->snd_una);
1632           max_bytes = clib_min (max_bytes, snd_space);
1633           offset = tc->snd_congestion - tc->snd_una - max_bytes;
1634           sb->rescue_rxt = tc->snd_congestion;
1635           tc->snd_nxt = tc->snd_una + offset;
1636           n_written = tcp_prepare_retransmit_segment (tc, offset, max_bytes,
1637                                                       &b);
1638           ASSERT (n_written);
1639           bi = vlib_get_buffer_index (vm, b);
1640           tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1641           break;
1642         }
1643
1644       max_bytes = clib_min (hole->end - sb->high_rxt, snd_space);
1645       max_bytes = snd_limited ? clib_min (max_bytes, tc->snd_mss) : max_bytes;
1646       if (max_bytes == 0)
1647         break;
1648       offset = sb->high_rxt - tc->snd_una;
1649       tc->snd_nxt = sb->high_rxt;
1650       n_written = tcp_prepare_retransmit_segment (tc, offset, max_bytes, &b);
1651
1652       /* Nothing left to retransmit */
1653       if (n_written == 0)
1654         break;
1655
1656       bi = vlib_get_buffer_index (vm, b);
1657       sb->high_rxt += n_written;
1658       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1659       ASSERT (n_written <= snd_space);
1660       snd_space -= n_written;
1661     }
1662
1663   /* If window allows, send 1 SMSS of new data */
1664   tc->snd_nxt = old_snd_nxt;
1665 }
1666
1667 /**
1668  * Fast retransmit without SACK info
1669  */
1670 void
1671 tcp_fast_retransmit_no_sack (tcp_connection_t * tc)
1672 {
1673   vlib_main_t *vm = vlib_get_main ();
1674   u32 n_written = 0, offset = 0, bi, old_snd_nxt;
1675   int snd_space;
1676   vlib_buffer_t *b;
1677
1678   ASSERT (tcp_in_fastrecovery (tc));
1679   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 0);
1680
1681   /* Start resending from first un-acked segment */
1682   old_snd_nxt = tc->snd_nxt;
1683   tc->snd_nxt = tc->snd_una;
1684   snd_space = tcp_available_snd_space (tc);
1685
1686   while (snd_space > 0)
1687     {
1688       offset += n_written;
1689       n_written = tcp_prepare_retransmit_segment (tc, offset, snd_space, &b);
1690
1691       /* Nothing left to retransmit */
1692       if (n_written == 0)
1693         break;
1694
1695       bi = vlib_get_buffer_index (vm, b);
1696       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1697       snd_space -= n_written;
1698     }
1699
1700   /* Restore snd_nxt. If window allows, send 1 SMSS of new data */
1701   tc->snd_nxt = old_snd_nxt;
1702 }
1703
1704 /**
1705  * Do fast retransmit
1706  */
1707 void
1708 tcp_fast_retransmit (tcp_connection_t * tc)
1709 {
1710   if (tcp_opts_sack_permitted (&tc->rcv_opts)
1711       && scoreboard_first_hole (&tc->sack_sb))
1712     tcp_fast_retransmit_sack (tc);
1713   else
1714     tcp_fast_retransmit_no_sack (tc);
1715 }
1716
1717 always_inline u32
1718 tcp_session_has_ooo_data (tcp_connection_t * tc)
1719 {
1720   stream_session_t *s =
1721     stream_session_get (tc->c_s_index, tc->c_thread_index);
1722   return svm_fifo_has_ooo_data (s->server_rx_fifo);
1723 }
1724
1725 always_inline uword
1726 tcp46_output_inline (vlib_main_t * vm,
1727                      vlib_node_runtime_t * node,
1728                      vlib_frame_t * from_frame, int is_ip4)
1729 {
1730   u32 n_left_from, next_index, *from, *to_next;
1731   u32 my_thread_index = vm->thread_index;
1732
1733   from = vlib_frame_vector_args (from_frame);
1734   n_left_from = from_frame->n_vectors;
1735   next_index = node->cached_next_index;
1736   tcp_set_time_now (my_thread_index);
1737
1738   while (n_left_from > 0)
1739     {
1740       u32 n_left_to_next;
1741
1742       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
1743
1744       while (n_left_from > 0 && n_left_to_next > 0)
1745         {
1746           u32 bi0;
1747           vlib_buffer_t *b0;
1748           tcp_connection_t *tc0;
1749           tcp_tx_trace_t *t0;
1750           tcp_header_t *th0 = 0;
1751           u32 error0 = TCP_ERROR_PKTS_SENT, next0 = TCP_OUTPUT_NEXT_IP_LOOKUP;
1752
1753           bi0 = from[0];
1754           to_next[0] = bi0;
1755           from += 1;
1756           to_next += 1;
1757           n_left_from -= 1;
1758           n_left_to_next -= 1;
1759
1760           b0 = vlib_get_buffer (vm, bi0);
1761           tc0 = tcp_connection_get (vnet_buffer (b0)->tcp.connection_index,
1762                                     my_thread_index);
1763           if (PREDICT_FALSE (tc0 == 0 || tc0->state == TCP_STATE_CLOSED))
1764             {
1765               error0 = TCP_ERROR_INVALID_CONNECTION;
1766               next0 = TCP_OUTPUT_NEXT_DROP;
1767               goto done;
1768             }
1769
1770           th0 = vlib_buffer_get_current (b0);
1771           TCP_EVT_DBG (TCP_EVT_OUTPUT, tc0, th0->flags, b0->current_length);
1772
1773           if (is_ip4)
1774             {
1775               vlib_buffer_push_ip4 (vm, b0, &tc0->c_lcl_ip4, &tc0->c_rmt_ip4,
1776                                     IP_PROTOCOL_TCP, 1);
1777               b0->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
1778               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1779               th0->checksum = 0;
1780             }
1781           else
1782             {
1783               ip6_header_t *ih0;
1784               ih0 = vlib_buffer_push_ip6 (vm, b0, &tc0->c_lcl_ip6,
1785                                           &tc0->c_rmt_ip6, IP_PROTOCOL_TCP);
1786               b0->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
1787               vnet_buffer (b0)->l3_hdr_offset = (u8 *) ih0 - b0->data;
1788               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1789               th0->checksum = 0;
1790             }
1791
1792           /* Filter out DUPACKs if there are no OOO segments left */
1793           if (PREDICT_FALSE
1794               (vnet_buffer (b0)->tcp.flags & TCP_BUF_FLAG_DUPACK))
1795             {
1796               if (!tcp_session_has_ooo_data (tc0))
1797                 {
1798                   error0 = TCP_ERROR_FILTERED_DUPACKS;
1799                   next0 = TCP_OUTPUT_NEXT_DROP;
1800                   goto done;
1801                 }
1802             }
1803
1804           /* Stop DELACK timer and fix flags */
1805           tc0->flags &= ~(TCP_CONN_SNDACK);
1806           tcp_timer_reset (tc0, TCP_TIMER_DELACK);
1807
1808           /* If not retransmitting
1809            * 1) update snd_una_max (SYN, SYNACK, FIN)
1810            * 2) If we're not tracking an ACK, start tracking */
1811           if (seq_lt (tc0->snd_una_max, tc0->snd_nxt))
1812             {
1813               tc0->snd_una_max = tc0->snd_nxt;
1814               if (tc0->rtt_ts == 0)
1815                 {
1816                   tc0->rtt_ts = tcp_time_now ();
1817                   tc0->rtt_seq = tc0->snd_nxt;
1818                 }
1819             }
1820
1821           /* Set the retransmit timer if not set already and not
1822            * doing a pure ACK */
1823           if (!tcp_timer_is_active (tc0, TCP_TIMER_RETRANSMIT)
1824               && tc0->snd_nxt != tc0->snd_una)
1825             {
1826               tcp_retransmit_timer_set (tc0);
1827               tc0->rto_boff = 0;
1828             }
1829
1830 #if 0
1831           /* Make sure we haven't lost route to our peer */
1832           if (PREDICT_FALSE (tc0->last_fib_check
1833                              < tc0->snd_opts.tsval + TCP_FIB_RECHECK_PERIOD))
1834             {
1835               if (PREDICT_TRUE
1836                   (tc0->c_rmt_fei == tcp_lookup_rmt_in_fib (tc0)))
1837                 {
1838                   tc0->last_fib_check = tc0->snd_opts.tsval;
1839                 }
1840               else
1841                 {
1842                   clib_warning ("lost connection to peer");
1843                   tcp_connection_reset (tc0);
1844                   goto done;
1845                 }
1846             }
1847
1848           /* Use pre-computed dpo to set next node */
1849           next0 = tc0->c_rmt_dpo.dpoi_next_node;
1850           vnet_buffer (b0)->ip.adj_index[VLIB_TX] = tc0->c_rmt_dpo.dpoi_index;
1851 #endif
1852
1853           vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0;
1854           vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0;
1855
1856           b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
1857         done:
1858           b0->error = node->errors[error0];
1859           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
1860             {
1861               t0 = vlib_add_trace (vm, node, b0, sizeof (*t0));
1862               if (th0)
1863                 {
1864                   clib_memcpy (&t0->tcp_header, th0, sizeof (t0->tcp_header));
1865                 }
1866               else
1867                 {
1868                   memset (&t0->tcp_header, 0, sizeof (t0->tcp_header));
1869                 }
1870               clib_memcpy (&t0->tcp_connection, tc0,
1871                            sizeof (t0->tcp_connection));
1872             }
1873
1874           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
1875                                            n_left_to_next, bi0, next0);
1876         }
1877
1878       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
1879     }
1880
1881   return from_frame->n_vectors;
1882 }
1883
1884 static uword
1885 tcp4_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1886              vlib_frame_t * from_frame)
1887 {
1888   return tcp46_output_inline (vm, node, from_frame, 1 /* is_ip4 */ );
1889 }
1890
1891 static uword
1892 tcp6_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1893              vlib_frame_t * from_frame)
1894 {
1895   return tcp46_output_inline (vm, node, from_frame, 0 /* is_ip4 */ );
1896 }
1897
1898 /* *INDENT-OFF* */
1899 VLIB_REGISTER_NODE (tcp4_output_node) =
1900 {
1901   .function = tcp4_output,.name = "tcp4-output",
1902     /* Takes a vector of packets. */
1903     .vector_size = sizeof (u32),
1904     .n_errors = TCP_N_ERROR,
1905     .error_strings = tcp_error_strings,
1906     .n_next_nodes = TCP_OUTPUT_N_NEXT,
1907     .next_nodes = {
1908 #define _(s,n) [TCP_OUTPUT_NEXT_##s] = n,
1909     foreach_tcp4_output_next
1910 #undef _
1911     },
1912     .format_buffer = format_tcp_header,
1913     .format_trace = format_tcp_tx_trace,
1914 };
1915 /* *INDENT-ON* */
1916
1917 VLIB_NODE_FUNCTION_MULTIARCH (tcp4_output_node, tcp4_output);
1918
1919 /* *INDENT-OFF* */
1920 VLIB_REGISTER_NODE (tcp6_output_node) =
1921 {
1922   .function = tcp6_output,
1923   .name = "tcp6-output",
1924     /* Takes a vector of packets. */
1925   .vector_size = sizeof (u32),
1926   .n_errors = TCP_N_ERROR,
1927   .error_strings = tcp_error_strings,
1928   .n_next_nodes = TCP_OUTPUT_N_NEXT,
1929   .next_nodes = {
1930 #define _(s,n) [TCP_OUTPUT_NEXT_##s] = n,
1931     foreach_tcp6_output_next
1932 #undef _
1933   },
1934   .format_buffer = format_tcp_header,
1935   .format_trace = format_tcp_tx_trace,
1936 };
1937 /* *INDENT-ON* */
1938
1939 VLIB_NODE_FUNCTION_MULTIARCH (tcp6_output_node, tcp6_output);
1940
1941 u32
1942 tcp_push_header (transport_connection_t * tconn, vlib_buffer_t * b)
1943 {
1944   tcp_connection_t *tc;
1945
1946   tc = (tcp_connection_t *) tconn;
1947   tcp_push_hdr_i (tc, b, TCP_STATE_ESTABLISHED, 0);
1948   ASSERT (seq_leq (tc->snd_una_max, tc->snd_una + tc->snd_wnd));
1949
1950   if (tc->rtt_ts == 0 && !tcp_in_cong_recovery (tc))
1951     {
1952       tc->rtt_ts = tcp_time_now ();
1953       tc->rtt_seq = tc->snd_nxt;
1954     }
1955   return 0;
1956 }
1957
1958 typedef enum _tcp_reset_next
1959 {
1960   TCP_RESET_NEXT_DROP,
1961   TCP_RESET_NEXT_IP_LOOKUP,
1962   TCP_RESET_N_NEXT
1963 } tcp_reset_next_t;
1964
1965 #define foreach_tcp4_reset_next         \
1966   _(DROP, "error-drop")                 \
1967   _(IP_LOOKUP, "ip4-lookup")
1968
1969 #define foreach_tcp6_reset_next         \
1970   _(DROP, "error-drop")                 \
1971   _(IP_LOOKUP, "ip6-lookup")
1972
1973 static uword
1974 tcp46_send_reset_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
1975                          vlib_frame_t * from_frame, u8 is_ip4)
1976 {
1977   u32 n_left_from, next_index, *from, *to_next;
1978   u32 my_thread_index = vm->thread_index;
1979
1980   from = vlib_frame_vector_args (from_frame);
1981   n_left_from = from_frame->n_vectors;
1982
1983   next_index = node->cached_next_index;
1984
1985   while (n_left_from > 0)
1986     {
1987       u32 n_left_to_next;
1988
1989       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
1990
1991       while (n_left_from > 0 && n_left_to_next > 0)
1992         {
1993           u32 bi0;
1994           vlib_buffer_t *b0;
1995           tcp_tx_trace_t *t0;
1996           tcp_header_t *th0;
1997           u32 error0 = TCP_ERROR_RST_SENT, next0 = TCP_RESET_NEXT_IP_LOOKUP;
1998
1999           bi0 = from[0];
2000           to_next[0] = bi0;
2001           from += 1;
2002           to_next += 1;
2003           n_left_from -= 1;
2004           n_left_to_next -= 1;
2005
2006           b0 = vlib_get_buffer (vm, bi0);
2007
2008           if (tcp_make_reset_in_place (vm, b0, vnet_buffer (b0)->tcp.flags,
2009                                        my_thread_index, is_ip4))
2010             {
2011               error0 = TCP_ERROR_LOOKUP_DROPS;
2012               next0 = TCP_RESET_NEXT_DROP;
2013               goto done;
2014             }
2015
2016           /* Prepare to send to IP lookup */
2017           vnet_buffer (b0)->sw_if_index[VLIB_TX] = 0;
2018           next0 = TCP_RESET_NEXT_IP_LOOKUP;
2019
2020         done:
2021           b0->error = node->errors[error0];
2022           b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
2023           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
2024             {
2025               th0 = vlib_buffer_get_current (b0);
2026               if (is_ip4)
2027                 th0 = ip4_next_header ((ip4_header_t *) th0);
2028               else
2029                 th0 = ip6_next_header ((ip6_header_t *) th0);
2030               t0 = vlib_add_trace (vm, node, b0, sizeof (*t0));
2031               clib_memcpy (&t0->tcp_header, th0, sizeof (t0->tcp_header));
2032             }
2033
2034           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
2035                                            n_left_to_next, bi0, next0);
2036         }
2037       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
2038     }
2039   return from_frame->n_vectors;
2040 }
2041
2042 static uword
2043 tcp4_send_reset (vlib_main_t * vm, vlib_node_runtime_t * node,
2044                  vlib_frame_t * from_frame)
2045 {
2046   return tcp46_send_reset_inline (vm, node, from_frame, 1);
2047 }
2048
2049 static uword
2050 tcp6_send_reset (vlib_main_t * vm, vlib_node_runtime_t * node,
2051                  vlib_frame_t * from_frame)
2052 {
2053   return tcp46_send_reset_inline (vm, node, from_frame, 0);
2054 }
2055
2056 /* *INDENT-OFF* */
2057 VLIB_REGISTER_NODE (tcp4_reset_node) = {
2058   .function = tcp4_send_reset,
2059   .name = "tcp4-reset",
2060   .vector_size = sizeof (u32),
2061   .n_errors = TCP_N_ERROR,
2062   .error_strings = tcp_error_strings,
2063   .n_next_nodes = TCP_RESET_N_NEXT,
2064   .next_nodes = {
2065 #define _(s,n) [TCP_RESET_NEXT_##s] = n,
2066     foreach_tcp4_reset_next
2067 #undef _
2068   },
2069   .format_trace = format_tcp_tx_trace,
2070 };
2071 /* *INDENT-ON* */
2072
2073 VLIB_NODE_FUNCTION_MULTIARCH (tcp4_reset_node, tcp4_send_reset);
2074
2075 /* *INDENT-OFF* */
2076 VLIB_REGISTER_NODE (tcp6_reset_node) = {
2077   .function = tcp6_send_reset,
2078   .name = "tcp6-reset",
2079   .vector_size = sizeof (u32),
2080   .n_errors = TCP_N_ERROR,
2081   .error_strings = tcp_error_strings,
2082   .n_next_nodes = TCP_RESET_N_NEXT,
2083   .next_nodes = {
2084 #define _(s,n) [TCP_RESET_NEXT_##s] = n,
2085     foreach_tcp6_reset_next
2086 #undef _
2087   },
2088   .format_trace = format_tcp_tx_trace,
2089 };
2090 /* *INDENT-ON* */
2091
2092 VLIB_NODE_FUNCTION_MULTIARCH (tcp6_reset_node, tcp6_send_reset);
2093
2094 /*
2095  * fd.io coding-style-patch-verification: ON
2096  *
2097  * Local Variables:
2098  * eval: (c-set-style "gnu")
2099  * End:
2100  */