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   u32 current_length = vec_len (tm->tx_buffers[thread_index]);
 443
 444   vec_validate (tm->tx_buffers[thread_index],
 445                 current_length + n_free_buffers - 1);
 446   _vec_len (tm->tx_buffers[thread_index]) = current_length
 447     + vlib_buffer_alloc (vlib_get_main (),
 448                          &tm->tx_buffers[thread_index][current_length],
 449                          n_free_buffers);
 450   /* buffer shortage, report failure */
 451   if (vec_len (tm->tx_buffers[thread_index]) == 0)
 452     {
 453       clib_warning ("out of buffers");
 454       return -1;
 455     }
 456   return 0;
 457 }
 458
 459 always_inline int
 460 tcp_get_free_buffer_index (tcp_main_t * tm, u32 * bidx)
 461 {
 462   u32 *my_tx_buffers;
 463   u32 thread_index = vlib_get_thread_index ();
 464   if (PREDICT_FALSE (vec_len (tm->tx_buffers[thread_index]) == 0))
 465     {
 466       if (tcp_alloc_tx_buffers (tm, thread_index, VLIB_FRAME_SIZE))
 467         return -1;
 468     }
 469   my_tx_buffers = tm->tx_buffers[thread_index];
 470   *bidx = my_tx_buffers[vec_len (my_tx_buffers) - 1];
 471   _vec_len (my_tx_buffers) -= 1;
 472   return 0;
 473 }
 474
 475 always_inline void
 476 tcp_return_buffer (tcp_main_t * tm)
 477 {
 478   _vec_len (tm->tx_buffers[vlib_get_thread_index ()]) += 1;
 479 }
 480
 481 always_inline void *
 482 tcp_reuse_buffer (vlib_main_t * vm, vlib_buffer_t * b)
 483 {
 484   if (b->flags & VLIB_BUFFER_NEXT_PRESENT)
 485     vlib_buffer_free_one (vm, b->next_buffer);
 486   /* Zero all flags but free list index and trace flag */
 487   b->flags &= VLIB_BUFFER_NEXT_PRESENT - 1;
 488   b->current_data = 0;
 489   b->current_length = 0;
 490   b->total_length_not_including_first_buffer = 0;
 491   vnet_buffer (b)->tcp.flags = 0;
 492
 493   /* Leave enough space for headers */
 494   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
 495 }
 496
 497 always_inline void *
 498 tcp_init_buffer (vlib_main_t * vm, vlib_buffer_t * b)
 499 {
 500   ASSERT ((b->flags & VLIB_BUFFER_NEXT_PRESENT) == 0);
 501   b->flags &= VLIB_BUFFER_FREE_LIST_INDEX_MASK;
 502   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
 503   b->total_length_not_including_first_buffer = 0;
 504   vnet_buffer (b)->tcp.flags = 0;
 505
 506   /* Leave enough space for headers */
 507   return vlib_buffer_make_headroom (b, MAX_HDRS_LEN);
 508 }
 509
 510 /**
 511  * Prepare ACK
 512  */
 513 void
 514 tcp_make_ack_i (tcp_connection_t * tc, vlib_buffer_t * b, tcp_state_t state,
 515                 u8 flags)
 516 {
 517   tcp_options_t _snd_opts, *snd_opts = &_snd_opts;
 518   u8 tcp_opts_len, tcp_hdr_opts_len;
 519   tcp_header_t *th;
 520   u16 wnd;
 521
 522   wnd = tcp_window_to_advertise (tc, state);
 523
 524   /* Make and write options */
 525   tcp_opts_len = tcp_make_established_options (tc, snd_opts);
 526   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
 527
 528   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
 529                              tc->rcv_nxt, tcp_hdr_opts_len, flags, wnd);
 530
 531   tcp_options_write ((u8 *) (th + 1), snd_opts);
 532   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
 533 }
 534
 535 /**
 536  * Convert buffer to ACK
 537  */
 538 void
 539 tcp_make_ack (tcp_connection_t * tc, vlib_buffer_t * b)
 540 {
 541   vlib_main_t *vm = vlib_get_main ();
 542
 543   tcp_reuse_buffer (vm, b);
 544   tcp_make_ack_i (tc, b, TCP_STATE_ESTABLISHED, TCP_FLAG_ACK);
 545   TCP_EVT_DBG (TCP_EVT_ACK_SENT, tc);
 546   vnet_buffer (b)->tcp.flags = TCP_BUF_FLAG_ACK;
 547   tc->rcv_las = tc->rcv_nxt;
 548 }
 549
 550 /**
 551  * Convert buffer to FIN-ACK
 552  */
 553 void
 554 tcp_make_fin (tcp_connection_t * tc, vlib_buffer_t * b)
 555 {
 556   vlib_main_t *vm = vlib_get_main ();
 557   u8 flags = 0;
 558
 559   tcp_reuse_buffer (vm, b);
 560
 561   flags = TCP_FLAG_FIN | TCP_FLAG_ACK;
 562   tcp_make_ack_i (tc, b, TCP_STATE_ESTABLISHED, flags);
 563
 564   /* Reset flags, make sure ack is sent */
 565   vnet_buffer (b)->tcp.flags &= ~TCP_BUF_FLAG_DUPACK;
 566 }
 567
 568 /**
 569  * Convert buffer to SYN
 570  */
 571 void
 572 tcp_make_syn (tcp_connection_t * tc, vlib_buffer_t * b)
 573 {
 574   u8 tcp_hdr_opts_len, tcp_opts_len;
 575   tcp_header_t *th;
 576   u16 initial_wnd;
 577   tcp_options_t snd_opts;
 578
 579   initial_wnd = tcp_initial_window_to_advertise (tc);
 580
 581   /* Make and write options */
 582   memset (&snd_opts, 0, sizeof (snd_opts));
 583   tcp_opts_len = tcp_make_syn_options (&snd_opts, tc->rcv_wscale);
 584   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
 585
 586   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->iss,
 587                              tc->rcv_nxt, tcp_hdr_opts_len, TCP_FLAG_SYN,
 588                              initial_wnd);
 589   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
 590   tcp_options_write ((u8 *) (th + 1), &snd_opts);
 591
 592   tcp_timer_update (tc, TCP_TIMER_RETRANSMIT_SYN,
 593                     tc->rto * TCP_TO_TIMER_TICK);
 594 }
 595
 596 /**
 597  * Convert buffer to SYN-ACK
 598  */
 599 void
 600 tcp_make_synack (tcp_connection_t * tc, vlib_buffer_t * b)
 601 {
 602   vlib_main_t *vm = vlib_get_main ();
 603   tcp_options_t _snd_opts, *snd_opts = &_snd_opts;
 604   u8 tcp_opts_len, tcp_hdr_opts_len;
 605   tcp_header_t *th;
 606   u16 initial_wnd;
 607
 608   memset (snd_opts, 0, sizeof (*snd_opts));
 609   tcp_reuse_buffer (vm, b);
 610
 611   initial_wnd = tcp_initial_window_to_advertise (tc);
 612   tcp_opts_len = tcp_make_synack_options (tc, snd_opts);
 613   tcp_hdr_opts_len = tcp_opts_len + sizeof (tcp_header_t);
 614
 615   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->iss,
 616                              tc->rcv_nxt, tcp_hdr_opts_len,
 617                              TCP_FLAG_SYN | TCP_FLAG_ACK, initial_wnd);
 618   tcp_options_write ((u8 *) (th + 1), snd_opts);
 619
 620   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
 621   vnet_buffer (b)->tcp.flags = TCP_BUF_FLAG_ACK;
 622
 623   /* Init retransmit timer. Use update instead of set because of
 624    * retransmissions */
 625   tcp_retransmit_timer_force_update (tc);
 626   TCP_EVT_DBG (TCP_EVT_SYNACK_SENT, tc);
 627 }
 628
 629 always_inline void
 630 tcp_enqueue_to_ip_lookup (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
 631                           u8 is_ip4)
 632 {
 633   u32 *to_next, next_index;
 634   vlib_frame_t *f;
 635
 636   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
 637   b->error = 0;
 638
 639   /* Default FIB for now */
 640   vnet_buffer (b)->sw_if_index[VLIB_TX] = 0;
 641
 642   /* Send to IP lookup */
 643   next_index = is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
 644   f = vlib_get_frame_to_node (vm, next_index);
 645
 646   /* Enqueue the packet */
 647   to_next = vlib_frame_vector_args (f);
 648   to_next[0] = bi;
 649   f->n_vectors = 1;
 650   vlib_put_frame_to_node (vm, next_index, f);
 651 }
 652
 653 always_inline void
 654 tcp_enqueue_to_output_i (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
 655                          u8 is_ip4, u8 flush)
 656 {
 657   tcp_main_t *tm = vnet_get_tcp_main ();
 658   u32 thread_index = vlib_get_thread_index ();
 659   u32 *to_next, next_index;
 660   vlib_frame_t *f;
 661
 662   b->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
 663   b->error = 0;
 664
 665   /* Decide where to send the packet */
 666   next_index = is_ip4 ? tcp4_output_node.index : tcp6_output_node.index;
 667
 668   /* Initialize the trajectory trace, if configured */
 669   if (VLIB_BUFFER_TRACE_TRAJECTORY > 0)
 670     {
 671       b->pre_data[0] = 1;
 672       b->pre_data[1] = next_index;
 673     }
 674
 675   /* Get frame to v4/6 output node */
 676   f = tm->tx_frames[!is_ip4][thread_index];
 677   if (!f)
 678     {
 679       f = vlib_get_frame_to_node (vm, next_index);
 680       ASSERT (f);
 681       tm->tx_frames[!is_ip4][thread_index] = f;
 682     }
 683   to_next = vlib_frame_vector_args (f);
 684   to_next[f->n_vectors] = bi;
 685   f->n_vectors += 1;
 686   if (flush || f->n_vectors == VLIB_FRAME_SIZE)
 687     {
 688       vlib_put_frame_to_node (vm, next_index, f);
 689       tm->tx_frames[!is_ip4][thread_index] = 0;
 690     }
 691 }
 692
 693 always_inline void
 694 tcp_enqueue_to_output (vlib_main_t * vm, vlib_buffer_t * b, u32 bi, u8 is_ip4)
 695 {
 696   tcp_enqueue_to_output_i (vm, b, bi, is_ip4, 0);
 697 }
 698
 699 always_inline void
 700 tcp_enqueue_to_output_now (vlib_main_t * vm, vlib_buffer_t * b, u32 bi,
 701                            u8 is_ip4)
 702 {
 703   tcp_enqueue_to_output_i (vm, b, bi, is_ip4, 1);
 704 }
 705
 706 int
 707 tcp_make_reset_in_place (vlib_main_t * vm, vlib_buffer_t * b0,
 708                          tcp_state_t state, u8 thread_index, u8 is_ip4)
 709 {
 710   ip4_header_t *ih4;
 711   ip6_header_t *ih6;
 712   tcp_header_t *th0;
 713   ip4_address_t src_ip40, dst_ip40;
 714   ip6_address_t src_ip60, dst_ip60;
 715   u16 src_port, dst_port;
 716   u32 tmp;
 717   u32 seq, ack;
 718   u8 flags;
 719
 720   /* Find IP and TCP headers */
 721   th0 = tcp_buffer_hdr (b0);
 722
 723   /* Save src and dst ip */
 724   if (is_ip4)
 725     {
 726       ih4 = vlib_buffer_get_current (b0);
 727       ASSERT ((ih4->ip_version_and_header_length & 0xF0) == 0x40);
 728       src_ip40.as_u32 = ih4->src_address.as_u32;
 729       dst_ip40.as_u32 = ih4->dst_address.as_u32;
 730     }
 731   else
 732     {
 733       ih6 = vlib_buffer_get_current (b0);
 734       ASSERT ((ih6->ip_version_traffic_class_and_flow_label & 0xF0) == 0x60);
 735       clib_memcpy (&src_ip60, &ih6->src_address, sizeof (ip6_address_t));
 736       clib_memcpy (&dst_ip60, &ih6->dst_address, sizeof (ip6_address_t));
 737     }
 738
 739   src_port = th0->src_port;
 740   dst_port = th0->dst_port;
 741
 742   /* Try to determine what/why we're actually resetting */
 743   if (state == TCP_STATE_CLOSED)
 744     {
 745       if (!tcp_syn (th0))
 746         return -1;
 747
 748       tmp = clib_net_to_host_u32 (th0->seq_number);
 749
 750       /* Got a SYN for no listener. */
 751       flags = TCP_FLAG_RST | TCP_FLAG_ACK;
 752       ack = clib_host_to_net_u32 (tmp + 1);
 753       seq = 0;
 754     }
 755   else
 756     {
 757       flags = TCP_FLAG_RST;
 758       seq = th0->ack_number;
 759       ack = 0;
 760     }
 761
 762   tcp_reuse_buffer (vm, b0);
 763   th0 = vlib_buffer_push_tcp_net_order (b0, dst_port, src_port, seq, ack,
 764                                         sizeof (tcp_header_t), flags, 0);
 765
 766   if (is_ip4)
 767     {
 768       ih4 = vlib_buffer_push_ip4 (vm, b0, &dst_ip40, &src_ip40,
 769                                   IP_PROTOCOL_TCP, 1);
 770       th0->checksum = ip4_tcp_udp_compute_checksum (vm, b0, ih4);
 771     }
 772   else
 773     {
 774       int bogus = ~0;
 775       ih6 = vlib_buffer_push_ip6 (vm, b0, &dst_ip60, &src_ip60,
 776                                   IP_PROTOCOL_TCP);
 777       th0->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b0, ih6, &bogus);
 778       ASSERT (!bogus);
 779     }
 780
 781   return 0;
 782 }
 783
 784 /**
 785  *  Send reset without reusing existing buffer
 786  *
 787  *  It extracts connection info out of original packet
 788  */
 789 void
 790 tcp_send_reset_w_pkt (tcp_connection_t * tc, vlib_buffer_t * pkt, u8 is_ip4)
 791 {
 792   vlib_buffer_t *b;
 793   u32 bi;
 794   tcp_main_t *tm = vnet_get_tcp_main ();
 795   vlib_main_t *vm = vlib_get_main ();
 796   u8 tcp_hdr_len, flags = 0;
 797   tcp_header_t *th, *pkt_th;
 798   u32 seq, ack;
 799   ip4_header_t *ih4, *pkt_ih4;
 800   ip6_header_t *ih6, *pkt_ih6;
 801
 802   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
 803     return;
 804
 805   b = vlib_get_buffer (vm, bi);
 806   tcp_init_buffer (vm, b);
 807
 808   /* Make and write options */
 809   tcp_hdr_len = sizeof (tcp_header_t);
 810
 811   if (is_ip4)
 812     {
 813       pkt_ih4 = vlib_buffer_get_current (pkt);
 814       pkt_th = ip4_next_header (pkt_ih4);
 815     }
 816   else
 817     {
 818       pkt_ih6 = vlib_buffer_get_current (pkt);
 819       pkt_th = ip6_next_header (pkt_ih6);
 820     }
 821
 822   if (tcp_ack (pkt_th))
 823     {
 824       flags = TCP_FLAG_RST;
 825       seq = pkt_th->ack_number;
 826       ack = (tc && tc->state >= TCP_STATE_SYN_RCVD) ? tc->rcv_nxt : 0;
 827     }
 828   else
 829     {
 830       flags = TCP_FLAG_RST | TCP_FLAG_ACK;
 831       seq = 0;
 832       ack = clib_host_to_net_u32 (vnet_buffer (pkt)->tcp.seq_end);
 833     }
 834
 835   th = vlib_buffer_push_tcp_net_order (b, pkt_th->dst_port, pkt_th->src_port,
 836                                        seq, ack, tcp_hdr_len, flags, 0);
 837
 838   /* Swap src and dst ip */
 839   if (is_ip4)
 840     {
 841       ASSERT ((pkt_ih4->ip_version_and_header_length & 0xF0) == 0x40);
 842       ih4 = vlib_buffer_push_ip4 (vm, b, &pkt_ih4->dst_address,
 843                                   &pkt_ih4->src_address, IP_PROTOCOL_TCP, 1);
 844       th->checksum = ip4_tcp_udp_compute_checksum (vm, b, ih4);
 845     }
 846   else
 847     {
 848       int bogus = ~0;
 849       ASSERT ((pkt_ih6->ip_version_traffic_class_and_flow_label & 0xF0) ==
 850               0x60);
 851       ih6 = vlib_buffer_push_ip6 (vm, b, &pkt_ih6->dst_address,
 852                                   &pkt_ih6->src_address, IP_PROTOCOL_TCP);
 853       th->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b, ih6, &bogus);
 854       ASSERT (!bogus);
 855     }
 856
 857   tcp_enqueue_to_ip_lookup (vm, b, bi, is_ip4);
 858   TCP_EVT_DBG (TCP_EVT_RST_SENT, tc);
 859 }
 860
 861 /**
 862  * Build and set reset packet for connection
 863  */
 864 void
 865 tcp_send_reset (tcp_connection_t * tc)
 866 {
 867   vlib_main_t *vm = vlib_get_main ();
 868   tcp_main_t *tm = vnet_get_tcp_main ();
 869   vlib_buffer_t *b;
 870   u32 bi;
 871   tcp_header_t *th;
 872   u16 tcp_hdr_opts_len, advertise_wnd, opts_write_len;
 873   u8 flags;
 874
 875   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
 876     return;
 877   b = vlib_get_buffer (vm, bi);
 878   tcp_init_buffer (vm, b);
 879
 880   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
 881   tcp_hdr_opts_len = tc->snd_opts_len + sizeof (tcp_header_t);
 882   advertise_wnd = tcp_window_to_advertise (tc, TCP_STATE_ESTABLISHED);
 883   flags = TCP_FLAG_RST;
 884   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
 885                              tc->rcv_nxt, tcp_hdr_opts_len, flags,
 886                              advertise_wnd);
 887   opts_write_len = tcp_options_write ((u8 *) (th + 1), &tc->snd_opts);
 888   ASSERT (opts_write_len == tc->snd_opts_len);
 889   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
 890   tcp_enqueue_to_output_now (vm, b, bi, tc->c_is_ip4);
 891 }
 892
 893 void
 894 tcp_push_ip_hdr (tcp_main_t * tm, tcp_connection_t * tc, vlib_buffer_t * b)
 895 {
 896   tcp_header_t *th = vlib_buffer_get_current (b);
 897   vlib_main_t *vm = vlib_get_main ();
 898   if (tc->c_is_ip4)
 899     {
 900       ip4_header_t *ih;
 901       ih = vlib_buffer_push_ip4 (vm, b, &tc->c_lcl_ip4,
 902                                  &tc->c_rmt_ip4, IP_PROTOCOL_TCP, 1);
 903       th->checksum = ip4_tcp_udp_compute_checksum (vm, b, ih);
 904     }
 905   else
 906     {
 907       ip6_header_t *ih;
 908       int bogus = ~0;
 909
 910       ih = vlib_buffer_push_ip6 (vm, b, &tc->c_lcl_ip6,
 911                                  &tc->c_rmt_ip6, IP_PROTOCOL_TCP);
 912       th->checksum = ip6_tcp_udp_icmp_compute_checksum (vm, b, ih, &bogus);
 913       ASSERT (!bogus);
 914     }
 915 }
 916
 917 /**
 918  *  Send SYN
 919  *
 920  *  Builds a SYN packet for a half-open connection and sends it to ipx_lookup.
 921  *  The packet is not forwarded through tcpx_output to avoid doing lookups
 922  *  in the half_open pool.
 923  */
 924 void
 925 tcp_send_syn (tcp_connection_t * tc)
 926 {
 927   vlib_buffer_t *b;
 928   u32 bi;
 929   tcp_main_t *tm = vnet_get_tcp_main ();
 930   vlib_main_t *vm = vlib_get_main ();
 931
 932   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
 933     return;
 934
 935   b = vlib_get_buffer (vm, bi);
 936   tcp_init_buffer (vm, b);
 937   tcp_make_syn (tc, b);
 938
 939   /* Measure RTT with this */
 940   tc->rtt_ts = tcp_time_now ();
 941   tc->rtt_seq = tc->snd_nxt;
 942   tc->rto_boff = 0;
 943
 944   /* Set the connection establishment timer */
 945   tcp_timer_set (tc, TCP_TIMER_ESTABLISH, TCP_ESTABLISH_TIME);
 946
 947   tcp_push_ip_hdr (tm, tc, b);
 948   tcp_enqueue_to_ip_lookup (vm, b, bi, tc->c_is_ip4);
 949   TCP_EVT_DBG (TCP_EVT_SYN_SENT, tc);
 950 }
 951
 952 /**
 953  * Flush tx frame populated by retransmits and timer pops
 954  */
 955 void
 956 tcp_flush_frame_to_output (vlib_main_t * vm, u8 thread_index, u8 is_ip4)
 957 {
 958   if (tcp_main.tx_frames[!is_ip4][thread_index])
 959     {
 960       u32 next_index;
 961       next_index = is_ip4 ? tcp4_output_node.index : tcp6_output_node.index;
 962       vlib_put_frame_to_node (vm, next_index,
 963                               tcp_main.tx_frames[!is_ip4][thread_index]);
 964       tcp_main.tx_frames[!is_ip4][thread_index] = 0;
 965     }
 966 }
 967
 968 /**
 969  * Flush both v4 and v6 tx frames for thread index
 970  */
 971 void
 972 tcp_flush_frames_to_output (u8 thread_index)
 973 {
 974   vlib_main_t *vm = vlib_get_main ();
 975   tcp_flush_frame_to_output (vm, thread_index, 1);
 976   tcp_flush_frame_to_output (vm, thread_index, 0);
 977 }
 978
 979 /**
 980  *  Send FIN
 981  */
 982 void
 983 tcp_send_fin (tcp_connection_t * tc)
 984 {
 985   vlib_buffer_t *b;
 986   u32 bi;
 987   tcp_main_t *tm = vnet_get_tcp_main ();
 988   vlib_main_t *vm = vlib_get_main ();
 989
 990   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
 991     return;
 992   b = vlib_get_buffer (vm, bi);
 993   /* buffer will be initialized by in tcp_make_fin */
 994   tcp_make_fin (tc, b);
 995   tcp_enqueue_to_output_now (vm, b, bi, tc->c_is_ip4);
 996   if (!(tc->flags & TCP_CONN_FINSNT))
 997     {
 998       tc->flags |= TCP_CONN_FINSNT;
 999       tc->flags &= ~TCP_CONN_FINPNDG;
1000       tc->snd_nxt += 1;
1001     }
1002   tcp_retransmit_timer_force_update (tc);
1003   TCP_EVT_DBG (TCP_EVT_FIN_SENT, tc);
1004 }
1005
1006 always_inline u8
1007 tcp_make_state_flags (tcp_connection_t * tc, tcp_state_t next_state)
1008 {
1009   switch (next_state)
1010     {
1011     case TCP_STATE_ESTABLISHED:
1012       return TCP_FLAG_ACK;
1013     case TCP_STATE_SYN_RCVD:
1014       return TCP_FLAG_SYN | TCP_FLAG_ACK;
1015     case TCP_STATE_SYN_SENT:
1016       return TCP_FLAG_SYN;
1017     case TCP_STATE_LAST_ACK:
1018     case TCP_STATE_FIN_WAIT_1:
1019       if (tc->snd_nxt + 1 < tc->snd_una_max)
1020         return TCP_FLAG_ACK;
1021       else
1022         return TCP_FLAG_FIN;
1023     default:
1024       clib_warning ("Shouldn't be here!");
1025     }
1026   return 0;
1027 }
1028
1029 /**
1030  * Push TCP header and update connection variables
1031  */
1032 static void
1033 tcp_push_hdr_i (tcp_connection_t * tc, vlib_buffer_t * b,
1034                 tcp_state_t next_state, u8 compute_opts)
1035 {
1036   u32 advertise_wnd, data_len;
1037   u8 tcp_hdr_opts_len, opts_write_len, flags;
1038   tcp_header_t *th;
1039
1040   data_len = b->current_length + b->total_length_not_including_first_buffer;
1041   ASSERT (!b->total_length_not_including_first_buffer
1042           || (b->flags & VLIB_BUFFER_NEXT_PRESENT));
1043   vnet_buffer (b)->tcp.flags = 0;
1044
1045   if (compute_opts)
1046     tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1047
1048   tcp_hdr_opts_len = tc->snd_opts_len + sizeof (tcp_header_t);
1049   advertise_wnd = tcp_window_to_advertise (tc, next_state);
1050   flags = tcp_make_state_flags (tc, next_state);
1051
1052   /* Push header and options */
1053   th = vlib_buffer_push_tcp (b, tc->c_lcl_port, tc->c_rmt_port, tc->snd_nxt,
1054                              tc->rcv_nxt, tcp_hdr_opts_len, flags,
1055                              advertise_wnd);
1056   opts_write_len = tcp_options_write ((u8 *) (th + 1), &tc->snd_opts);
1057
1058   ASSERT (opts_write_len == tc->snd_opts_len);
1059   vnet_buffer (b)->tcp.connection_index = tc->c_c_index;
1060
1061   /*
1062    * Update connection variables
1063    */
1064
1065   tc->snd_nxt += data_len;
1066   tc->rcv_las = tc->rcv_nxt;
1067
1068   /* TODO this is updated in output as well ... */
1069   if (seq_gt (tc->snd_nxt, tc->snd_una_max))
1070     {
1071       tc->snd_una_max = tc->snd_nxt;
1072       tcp_validate_txf_size (tc, tc->snd_una_max - tc->snd_una);
1073     }
1074
1075   TCP_EVT_DBG (TCP_EVT_PKTIZE, tc);
1076 }
1077
1078 void
1079 tcp_send_ack (tcp_connection_t * tc)
1080 {
1081   tcp_main_t *tm = vnet_get_tcp_main ();
1082   vlib_main_t *vm = vlib_get_main ();
1083
1084   vlib_buffer_t *b;
1085   u32 bi;
1086
1087   /* Get buffer */
1088   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1089     return;
1090   b = vlib_get_buffer (vm, bi);
1091
1092   /* Fill in the ACK */
1093   tcp_make_ack (tc, b);
1094   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1095 }
1096
1097 /**
1098  * Delayed ack timer handler
1099  *
1100  * Sends delayed ACK when timer expires
1101  */
1102 void
1103 tcp_timer_delack_handler (u32 index)
1104 {
1105   u32 thread_index = vlib_get_thread_index ();
1106   tcp_connection_t *tc;
1107
1108   tc = tcp_connection_get (index, thread_index);
1109   tc->timers[TCP_TIMER_DELACK] = TCP_TIMER_HANDLE_INVALID;
1110   tcp_send_ack (tc);
1111 }
1112
1113 /**
1114  * Build a retransmit segment
1115  *
1116  * @return the number of bytes in the segment or 0 if there's nothing to
1117  *         retransmit
1118  */
1119 u32
1120 tcp_prepare_retransmit_segment (tcp_connection_t * tc, u32 offset,
1121                                 u32 max_deq_bytes, vlib_buffer_t ** b)
1122 {
1123   tcp_main_t *tm = vnet_get_tcp_main ();
1124   vlib_main_t *vm = vlib_get_main ();
1125   int n_bytes = 0;
1126   u32 start, bi, available_bytes, seg_size;
1127   u8 *data;
1128
1129   ASSERT (tc->state >= TCP_STATE_ESTABLISHED);
1130   ASSERT (max_deq_bytes != 0);
1131
1132   /*
1133    * Make sure we can retransmit something
1134    */
1135   available_bytes = stream_session_tx_fifo_max_dequeue (&tc->connection);
1136   ASSERT (available_bytes >= offset);
1137   available_bytes -= offset;
1138   if (!available_bytes)
1139     return 0;
1140   max_deq_bytes = clib_min (tc->snd_mss, max_deq_bytes);
1141   max_deq_bytes = clib_min (available_bytes, max_deq_bytes);
1142
1143   /* Start is beyond snd_congestion */
1144   start = tc->snd_una + offset;
1145   if (seq_geq (start, tc->snd_congestion))
1146     goto done;
1147
1148   /* Don't overshoot snd_congestion */
1149   if (seq_gt (start + max_deq_bytes, tc->snd_congestion))
1150     {
1151       max_deq_bytes = tc->snd_congestion - start;
1152       if (max_deq_bytes == 0)
1153         goto done;
1154     }
1155
1156   seg_size = max_deq_bytes + MAX_HDRS_LEN;
1157
1158   /*
1159    * Prepare options
1160    */
1161   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1162
1163   /*
1164    * Allocate and fill in buffer(s)
1165    */
1166
1167   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1168     return 0;
1169   *b = vlib_get_buffer (vm, bi);
1170   data = tcp_init_buffer (vm, *b);
1171
1172   /* Easy case, buffer size greater than mss */
1173   if (PREDICT_TRUE (seg_size <= tm->bytes_per_buffer))
1174     {
1175       n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1176                                            max_deq_bytes);
1177       ASSERT (n_bytes == max_deq_bytes);
1178       b[0]->current_length = n_bytes;
1179       tcp_push_hdr_i (tc, *b, tc->state, 0);
1180     }
1181   /* Split mss into multiple buffers */
1182   else
1183     {
1184       u32 chain_bi = ~0, n_bufs_per_seg;
1185       u32 thread_index = vlib_get_thread_index ();
1186       u16 n_peeked, len_to_deq, available_bufs;
1187       vlib_buffer_t *chain_b, *prev_b;
1188       int i;
1189
1190       n_bufs_per_seg = ceil ((double) seg_size / tm->bytes_per_buffer);
1191
1192       /* Make sure we have enough buffers */
1193       available_bufs = vec_len (tm->tx_buffers[thread_index]);
1194       if (n_bufs_per_seg > available_bufs)
1195         {
1196           if (tcp_alloc_tx_buffers (tm, thread_index,
1197                                     VLIB_FRAME_SIZE - available_bufs))
1198             {
1199               tcp_return_buffer (tm);
1200               *b = 0;
1201               return 0;
1202             }
1203         }
1204
1205       n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1206                                            tm->bytes_per_buffer -
1207                                            MAX_HDRS_LEN);
1208       b[0]->current_length = n_bytes;
1209       b[0]->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
1210       b[0]->total_length_not_including_first_buffer = 0;
1211       max_deq_bytes -= n_bytes;
1212
1213       chain_b = *b;
1214       for (i = 1; i < n_bufs_per_seg; i++)
1215         {
1216           prev_b = chain_b;
1217           len_to_deq = clib_min (max_deq_bytes, tm->bytes_per_buffer);
1218           tcp_get_free_buffer_index (tm, &chain_bi);
1219           ASSERT (chain_bi != (u32) ~ 0);
1220           chain_b = vlib_get_buffer (vm, chain_bi);
1221           chain_b->current_data = 0;
1222           data = vlib_buffer_get_current (chain_b);
1223           n_peeked = stream_session_peek_bytes (&tc->connection, data,
1224                                                 offset + n_bytes, len_to_deq);
1225           ASSERT (n_peeked == len_to_deq);
1226           n_bytes += n_peeked;
1227           chain_b->current_length = n_peeked;
1228           chain_b->flags &= VLIB_BUFFER_FREE_LIST_INDEX_MASK;
1229           chain_b->next_buffer = 0;
1230
1231           /* update previous buffer */
1232           prev_b->next_buffer = chain_bi;
1233           prev_b->flags |= VLIB_BUFFER_NEXT_PRESENT;
1234
1235           max_deq_bytes -= n_peeked;
1236           b[0]->total_length_not_including_first_buffer += n_peeked;
1237         }
1238
1239       tcp_push_hdr_i (tc, *b, tc->state, 0);
1240     }
1241
1242   ASSERT (n_bytes > 0);
1243   ASSERT (((*b)->current_data + (*b)->current_length) <=
1244           tm->bytes_per_buffer);
1245
1246   if (tcp_in_fastrecovery (tc))
1247     tc->snd_rxt_bytes += n_bytes;
1248
1249 done:
1250   TCP_EVT_DBG (TCP_EVT_CC_RTX, tc, offset, n_bytes);
1251   return n_bytes;
1252 }
1253
1254 /**
1255  * Reset congestion control, switch cwnd to loss window and try again.
1256  */
1257 static void
1258 tcp_rtx_timeout_cc (tcp_connection_t * tc)
1259 {
1260   tc->prev_ssthresh = tc->ssthresh;
1261   tc->prev_cwnd = tc->cwnd;
1262
1263   /* Cleanly recover cc (also clears up fast retransmit) */
1264   if (tcp_in_fastrecovery (tc))
1265     tcp_cc_fastrecovery_exit (tc);
1266
1267   /* Start again from the beginning */
1268   tc->ssthresh = clib_max (tcp_flight_size (tc) / 2, 2 * tc->snd_mss);
1269   tc->cwnd = tcp_loss_wnd (tc);
1270   tc->snd_congestion = tc->snd_una_max;
1271
1272   tcp_recovery_on (tc);
1273 }
1274
1275 static void
1276 tcp_timer_retransmit_handler_i (u32 index, u8 is_syn)
1277 {
1278   tcp_main_t *tm = vnet_get_tcp_main ();
1279   vlib_main_t *vm = vlib_get_main ();
1280   u32 thread_index = vlib_get_thread_index ();
1281   tcp_connection_t *tc;
1282   vlib_buffer_t *b = 0;
1283   u32 bi, n_bytes;
1284
1285   if (is_syn)
1286     {
1287       tc = tcp_half_open_connection_get (index);
1288       /* Note: the connection may have transitioned to ESTABLISHED... */
1289       if (PREDICT_FALSE (tc == 0))
1290         return;
1291       tc->timers[TCP_TIMER_RETRANSMIT_SYN] = TCP_TIMER_HANDLE_INVALID;
1292     }
1293   else
1294     {
1295       tc = tcp_connection_get (index, thread_index);
1296       /* Note: the connection may have been closed and pool_put */
1297       if (PREDICT_FALSE (tc == 0))
1298         return;
1299       tc->timers[TCP_TIMER_RETRANSMIT] = TCP_TIMER_HANDLE_INVALID;
1300     }
1301
1302   if (tc->state >= TCP_STATE_ESTABLISHED)
1303     {
1304       /* Lost FIN, retransmit and return */
1305       if (tcp_is_lost_fin (tc))
1306         {
1307           tcp_send_fin (tc);
1308           return;
1309         }
1310
1311       /* We're not in recovery so make sure rto_boff is 0 */
1312       if (!tcp_in_recovery (tc) && tc->rto_boff > 0)
1313         {
1314           tc->rto_boff = 0;
1315           tcp_update_rto (tc);
1316         }
1317
1318       /* Increment RTO backoff (also equal to number of retries) and go back
1319        * to first un-acked byte  */
1320       tc->rto_boff += 1;
1321       tc->snd_nxt = tc->snd_una;
1322
1323       /* First retransmit timeout */
1324       if (tc->rto_boff == 1)
1325         tcp_rtx_timeout_cc (tc);
1326
1327       /* Exponential backoff */
1328       tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1329
1330       TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 1);
1331
1332       /* Send one segment. Note that n_bytes may be zero due to buffer shortfall  */
1333       n_bytes = tcp_prepare_retransmit_segment (tc, 0, tc->snd_mss, &b);
1334
1335       /* TODO be less aggressive about this */
1336       scoreboard_clear (&tc->sack_sb);
1337
1338       if (n_bytes == 0)
1339         {
1340           ASSERT (!b);
1341           if (tc->snd_una == tc->snd_una_max)
1342             return;
1343           ASSERT (tc->rto_boff > 1 && tc->snd_una == tc->snd_congestion);
1344           clib_warning ("retransmit fail: %U", format_tcp_connection, tc, 2);
1345           /* Try again eventually */
1346           tcp_retransmit_timer_set (tc);
1347           return;
1348         }
1349
1350       bi = vlib_get_buffer_index (vm, b);
1351
1352       /* For first retransmit, record timestamp (Eifel detection RFC3522) */
1353       if (tc->rto_boff == 1)
1354         tc->snd_rxt_ts = tcp_time_now ();
1355
1356       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1357       tcp_retransmit_timer_update (tc);
1358     }
1359   /* Retransmit for SYN */
1360   else if (tc->state == TCP_STATE_SYN_SENT)
1361     {
1362       /* Half-open connection actually moved to established but we were
1363        * waiting for syn retransmit to pop to call cleanup from the right
1364        * thread. */
1365       if (tc->flags & TCP_CONN_HALF_OPEN_DONE)
1366         {
1367           if (tcp_half_open_connection_cleanup (tc))
1368             {
1369               clib_warning ("could not remove half-open connection");
1370               ASSERT (0);
1371             }
1372           return;
1373         }
1374
1375       /* Try without increasing RTO a number of times. If this fails,
1376        * start growing RTO exponentially */
1377       tc->rto_boff += 1;
1378       if (tc->rto_boff > TCP_RTO_SYN_RETRIES)
1379         tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1380
1381       if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1382         return;
1383
1384       b = vlib_get_buffer (vm, bi);
1385       tcp_init_buffer (vm, b);
1386       tcp_make_syn (tc, b);
1387
1388       tc->rtt_ts = 0;
1389       TCP_EVT_DBG (TCP_EVT_SYN_RXT, tc, 0);
1390
1391       /* This goes straight to ipx_lookup. Retransmit timer set already */
1392       tcp_push_ip_hdr (tm, tc, b);
1393       tcp_enqueue_to_ip_lookup (vm, b, bi, tc->c_is_ip4);
1394     }
1395   /* Retransmit SYN-ACK */
1396   else if (tc->state == TCP_STATE_SYN_RCVD)
1397     {
1398       tc->rto_boff += 1;
1399       tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1400       tc->rtt_ts = 0;
1401
1402       if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1403         return;
1404
1405       b = vlib_get_buffer (vm, bi);
1406       tcp_make_synack (tc, b);
1407       TCP_EVT_DBG (TCP_EVT_SYN_RXT, tc, 1);
1408
1409       /* Retransmit timer already updated, just enqueue to output */
1410       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1411     }
1412   else
1413     {
1414       ASSERT (tc->state == TCP_STATE_CLOSED);
1415       clib_warning ("connection closed ...");
1416       return;
1417     }
1418 }
1419
1420 void
1421 tcp_timer_retransmit_handler (u32 index)
1422 {
1423   tcp_timer_retransmit_handler_i (index, 0);
1424 }
1425
1426 void
1427 tcp_timer_retransmit_syn_handler (u32 index)
1428 {
1429   tcp_timer_retransmit_handler_i (index, 1);
1430 }
1431
1432 /**
1433  * Got 0 snd_wnd from peer, try to do something about it.
1434  *
1435  */
1436 void
1437 tcp_timer_persist_handler (u32 index)
1438 {
1439   tcp_main_t *tm = vnet_get_tcp_main ();
1440   vlib_main_t *vm = vlib_get_main ();
1441   u32 thread_index = vlib_get_thread_index ();
1442   tcp_connection_t *tc;
1443   vlib_buffer_t *b;
1444   u32 bi, old_snd_nxt, max_snd_bytes, available_bytes, offset;
1445   int n_bytes = 0;
1446   u8 *data;
1447
1448   tc = tcp_connection_get_if_valid (index, thread_index);
1449
1450   if (!tc)
1451     return;
1452
1453   /* Make sure timer handle is set to invalid */
1454   tc->timers[TCP_TIMER_PERSIST] = TCP_TIMER_HANDLE_INVALID;
1455
1456   /* Problem already solved or worse */
1457   if (tc->state == TCP_STATE_CLOSED || tc->state > TCP_STATE_ESTABLISHED
1458       || tc->snd_wnd > tc->snd_mss || tcp_in_recovery (tc))
1459     return;
1460
1461   available_bytes = stream_session_tx_fifo_max_dequeue (&tc->connection);
1462   offset = tc->snd_una_max - tc->snd_una;
1463
1464   /* Reprogram persist if no new bytes available to send. We may have data
1465    * next time */
1466   if (!available_bytes)
1467     {
1468       tcp_persist_timer_set (tc);
1469       return;
1470     }
1471
1472   if (available_bytes <= offset)
1473     {
1474       ASSERT (tcp_timer_is_active (tc, TCP_TIMER_RETRANSMIT));
1475       return;
1476     }
1477
1478   /* Increment RTO backoff */
1479   tc->rto_boff += 1;
1480   tc->rto = clib_min (tc->rto << 1, TCP_RTO_MAX);
1481
1482   /*
1483    * Try to force the first unsent segment (or buffer)
1484    */
1485   if (PREDICT_FALSE (tcp_get_free_buffer_index (tm, &bi)))
1486     return;
1487   b = vlib_get_buffer (vm, bi);
1488   data = tcp_init_buffer (vm, b);
1489
1490   tcp_validate_txf_size (tc, offset);
1491   tc->snd_opts_len = tcp_make_options (tc, &tc->snd_opts, tc->state);
1492   max_snd_bytes = clib_min (tc->snd_mss, tm->bytes_per_buffer - MAX_HDRS_LEN);
1493   n_bytes = stream_session_peek_bytes (&tc->connection, data, offset,
1494                                        max_snd_bytes);
1495   b->current_length = n_bytes;
1496   ASSERT (n_bytes != 0 && (tc->snd_nxt == tc->snd_una_max || tc->rto_boff > 1
1497                            || tcp_timer_is_active (tc,
1498                                                    TCP_TIMER_RETRANSMIT)));
1499
1500   /* Allow updating of snd_una_max but don't update snd_nxt */
1501   old_snd_nxt = tc->snd_nxt;
1502   tcp_push_hdr_i (tc, b, tc->state, 0);
1503   tc->snd_nxt = old_snd_nxt;
1504   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1505
1506   /* Just sent new data, enable retransmit */
1507   tcp_retransmit_timer_update (tc);
1508 }
1509
1510 /**
1511  * Retransmit first unacked segment
1512  */
1513 void
1514 tcp_retransmit_first_unacked (tcp_connection_t * tc)
1515 {
1516   vlib_main_t *vm = vlib_get_main ();
1517   vlib_buffer_t *b;
1518   u32 bi, old_snd_nxt, n_bytes;
1519
1520   old_snd_nxt = tc->snd_nxt;
1521   tc->snd_nxt = tc->snd_una;
1522
1523   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 2);
1524   n_bytes = tcp_prepare_retransmit_segment (tc, 0, tc->snd_mss, &b);
1525   if (!n_bytes)
1526     return;
1527   bi = vlib_get_buffer_index (vm, b);
1528   tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1529
1530   tc->snd_nxt = old_snd_nxt;
1531 }
1532
1533 /**
1534  * Do fast retransmit with SACKs
1535  */
1536 void
1537 tcp_fast_retransmit_sack (tcp_connection_t * tc)
1538 {
1539   vlib_main_t *vm = vlib_get_main ();
1540   u32 n_written = 0, offset, max_bytes;
1541   vlib_buffer_t *b = 0;
1542   sack_scoreboard_hole_t *hole;
1543   sack_scoreboard_t *sb;
1544   u32 bi, old_snd_nxt;
1545   int snd_space;
1546   u8 snd_limited = 0, can_rescue = 0;
1547
1548   ASSERT (tcp_in_fastrecovery (tc));
1549   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 0);
1550
1551   old_snd_nxt = tc->snd_nxt;
1552   sb = &tc->sack_sb;
1553   snd_space = tcp_available_snd_space (tc);
1554
1555   hole = scoreboard_get_hole (sb, sb->cur_rxt_hole);
1556   while (hole && snd_space > 0)
1557     {
1558       hole = scoreboard_next_rxt_hole (sb, hole,
1559                                        tcp_fastrecovery_sent_1_smss (tc),
1560                                        &can_rescue, &snd_limited);
1561       if (!hole)
1562         {
1563           if (!can_rescue || !(seq_lt (sb->rescue_rxt, tc->snd_una)
1564                                || seq_gt (sb->rescue_rxt,
1565                                           tc->snd_congestion)))
1566             break;
1567
1568           /* If rescue rxt undefined or less than snd_una then one segment of
1569            * up to SMSS octets that MUST include the highest outstanding
1570            * unSACKed sequence number SHOULD be returned, and RescueRxt set to
1571            * RecoveryPoint. HighRxt MUST NOT be updated.
1572            */
1573           max_bytes = clib_min (tc->snd_mss,
1574                                 tc->snd_congestion - tc->snd_una);
1575           max_bytes = clib_min (max_bytes, snd_space);
1576           offset = tc->snd_congestion - tc->snd_una - max_bytes;
1577           sb->rescue_rxt = tc->snd_congestion;
1578           tc->snd_nxt = tc->snd_una + offset;
1579           n_written = tcp_prepare_retransmit_segment (tc, offset, max_bytes,
1580                                                       &b);
1581           ASSERT (n_written);
1582           bi = vlib_get_buffer_index (vm, b);
1583           tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1584           break;
1585         }
1586
1587       max_bytes = clib_min (hole->end - sb->high_rxt, snd_space);
1588       max_bytes = snd_limited ? clib_min (max_bytes, tc->snd_mss) : max_bytes;
1589       if (max_bytes == 0)
1590         break;
1591       offset = sb->high_rxt - tc->snd_una;
1592       tc->snd_nxt = sb->high_rxt;
1593       n_written = tcp_prepare_retransmit_segment (tc, offset, max_bytes, &b);
1594
1595       /* Nothing left to retransmit */
1596       if (n_written == 0)
1597         break;
1598
1599       bi = vlib_get_buffer_index (vm, b);
1600       sb->high_rxt += n_written;
1601       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1602       ASSERT (n_written <= snd_space);
1603       snd_space -= n_written;
1604     }
1605
1606   /* If window allows, send 1 SMSS of new data */
1607   tc->snd_nxt = old_snd_nxt;
1608 }
1609
1610 /**
1611  * Fast retransmit without SACK info
1612  */
1613 void
1614 tcp_fast_retransmit_no_sack (tcp_connection_t * tc)
1615 {
1616   vlib_main_t *vm = vlib_get_main ();
1617   u32 n_written = 0, offset = 0, bi, old_snd_nxt;
1618   int snd_space;
1619   vlib_buffer_t *b;
1620
1621   ASSERT (tcp_in_fastrecovery (tc));
1622   TCP_EVT_DBG (TCP_EVT_CC_EVT, tc, 0);
1623
1624   /* Start resending from first un-acked segment */
1625   old_snd_nxt = tc->snd_nxt;
1626   tc->snd_nxt = tc->snd_una;
1627   snd_space = tcp_available_snd_space (tc);
1628
1629   while (snd_space > 0)
1630     {
1631       offset += n_written;
1632       n_written = tcp_prepare_retransmit_segment (tc, offset, snd_space, &b);
1633
1634       /* Nothing left to retransmit */
1635       if (n_written == 0)
1636         break;
1637
1638       bi = vlib_get_buffer_index (vm, b);
1639       tcp_enqueue_to_output (vm, b, bi, tc->c_is_ip4);
1640       snd_space -= n_written;
1641     }
1642
1643   /* Restore snd_nxt. If window allows, send 1 SMSS of new data */
1644   tc->snd_nxt = old_snd_nxt;
1645 }
1646
1647 /**
1648  * Do fast retransmit
1649  */
1650 void
1651 tcp_fast_retransmit (tcp_connection_t * tc)
1652 {
1653   if (tcp_opts_sack_permitted (&tc->rcv_opts)
1654       && scoreboard_first_hole (&tc->sack_sb))
1655     tcp_fast_retransmit_sack (tc);
1656   else
1657     tcp_fast_retransmit_no_sack (tc);
1658 }
1659
1660 always_inline u32
1661 tcp_session_has_ooo_data (tcp_connection_t * tc)
1662 {
1663   stream_session_t *s =
1664     stream_session_get (tc->c_s_index, tc->c_thread_index);
1665   return svm_fifo_has_ooo_data (s->server_rx_fifo);
1666 }
1667
1668 always_inline uword
1669 tcp46_output_inline (vlib_main_t * vm,
1670                      vlib_node_runtime_t * node,
1671                      vlib_frame_t * from_frame, int is_ip4)
1672 {
1673   u32 n_left_from, next_index, *from, *to_next;
1674   u32 my_thread_index = vm->thread_index;
1675
1676   from = vlib_frame_vector_args (from_frame);
1677   n_left_from = from_frame->n_vectors;
1678   next_index = node->cached_next_index;
1679   tcp_set_time_now (my_thread_index);
1680
1681   while (n_left_from > 0)
1682     {
1683       u32 n_left_to_next;
1684
1685       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
1686
1687       while (n_left_from > 0 && n_left_to_next > 0)
1688         {
1689           u32 bi0;
1690           vlib_buffer_t *b0;
1691           tcp_connection_t *tc0;
1692           tcp_tx_trace_t *t0;
1693           tcp_header_t *th0 = 0;
1694           u32 error0 = TCP_ERROR_PKTS_SENT, next0 = TCP_OUTPUT_NEXT_IP_LOOKUP;
1695
1696           bi0 = from[0];
1697           to_next[0] = bi0;
1698           from += 1;
1699           to_next += 1;
1700           n_left_from -= 1;
1701           n_left_to_next -= 1;
1702
1703           b0 = vlib_get_buffer (vm, bi0);
1704           tc0 = tcp_connection_get (vnet_buffer (b0)->tcp.connection_index,
1705                                     my_thread_index);
1706           if (PREDICT_FALSE (tc0 == 0 || tc0->state == TCP_STATE_CLOSED))
1707             {
1708               error0 = TCP_ERROR_INVALID_CONNECTION;
1709               next0 = TCP_OUTPUT_NEXT_DROP;
1710               goto done;
1711             }
1712
1713           th0 = vlib_buffer_get_current (b0);
1714           TCP_EVT_DBG (TCP_EVT_OUTPUT, tc0, th0->flags, b0->current_length);
1715
1716           if (is_ip4)
1717             {
1718               vlib_buffer_push_ip4 (vm, b0, &tc0->c_lcl_ip4, &tc0->c_rmt_ip4,
1719                                     IP_PROTOCOL_TCP, 1);
1720               b0->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
1721               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1722               th0->checksum = 0;
1723             }
1724           else
1725             {
1726               ip6_header_t *ih0;
1727               ih0 = vlib_buffer_push_ip6 (vm, b0, &tc0->c_lcl_ip6,
1728                                           &tc0->c_rmt_ip6, IP_PROTOCOL_TCP);
1729               b0->flags |= VNET_BUFFER_F_OFFLOAD_TCP_CKSUM;
1730               vnet_buffer (b0)->l3_hdr_offset = (u8 *) ih0 - b0->data;
1731               vnet_buffer (b0)->l4_hdr_offset = (u8 *) th0 - b0->data;
1732               th0->checksum = 0;
1733             }
1734
1735           /* Filter out DUPACKs if there are no OOO segments left */
1736           if (PREDICT_FALSE
1737               (vnet_buffer (b0)->tcp.flags & TCP_BUF_FLAG_DUPACK))
1738             {
1739               if (!tcp_session_has_ooo_data (tc0))
1740                 {
1741                   error0 = TCP_ERROR_FILTERED_DUPACKS;
1742                   next0 = TCP_OUTPUT_NEXT_DROP;
1743                   goto done;
1744                 }
1745             }
1746
1747           /* Stop DELACK timer and fix flags */
1748           tc0->flags &= ~(TCP_CONN_SNDACK);
1749           tcp_timer_reset (tc0, TCP_TIMER_DELACK);
1750
1751           /* If not retransmitting
1752            * 1) update snd_una_max (SYN, SYNACK, FIN)
1753            * 2) If we're not tracking an ACK, start tracking */
1754           if (seq_lt (tc0->snd_una_max, tc0->snd_nxt))
1755             {
1756               tc0->snd_una_max = tc0->snd_nxt;
1757               if (tc0->rtt_ts == 0)
1758                 {
1759                   tc0->rtt_ts = tcp_time_now ();
1760                   tc0->rtt_seq = tc0->snd_nxt;
1761                 }
1762             }
1763
1764           /* Set the retransmit timer if not set already and not
1765            * doing a pure ACK */
1766           if (!tcp_timer_is_active (tc0, TCP_TIMER_RETRANSMIT)
1767               && tc0->snd_nxt != tc0->snd_una)
1768             {
1769               tcp_retransmit_timer_set (tc0);
1770               tc0->rto_boff = 0;
1771             }
1772
1773 #if 0
1774           /* Make sure we haven't lost route to our peer */
1775           if (PREDICT_FALSE (tc0->last_fib_check
1776                              < tc0->snd_opts.tsval + TCP_FIB_RECHECK_PERIOD))
1777             {
1778               if (PREDICT_TRUE
1779                   (tc0->c_rmt_fei == tcp_lookup_rmt_in_fib (tc0)))
1780                 {
1781                   tc0->last_fib_check = tc0->snd_opts.tsval;
1782                 }
1783               else
1784                 {
1785                   clib_warning ("lost connection to peer");
1786                   tcp_connection_reset (tc0);
1787                   goto done;
1788                 }
1789             }
1790
1791           /* Use pre-computed dpo to set next node */
1792           next0 = tc0->c_rmt_dpo.dpoi_next_node;
1793           vnet_buffer (b0)->ip.adj_index[VLIB_TX] = tc0->c_rmt_dpo.dpoi_index;
1794 #endif
1795
1796           vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0;
1797           vnet_buffer (b0)->sw_if_index[VLIB_TX] = ~0;
1798
1799           b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
1800         done:
1801           b0->error = node->errors[error0];
1802           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
1803             {
1804               t0 = vlib_add_trace (vm, node, b0, sizeof (*t0));
1805               if (th0)
1806                 {
1807                   clib_memcpy (&t0->tcp_header, th0, sizeof (t0->tcp_header));
1808                 }
1809               else
1810                 {
1811                   memset (&t0->tcp_header, 0, sizeof (t0->tcp_header));
1812                 }
1813               clib_memcpy (&t0->tcp_connection, tc0,
1814                            sizeof (t0->tcp_connection));
1815             }
1816
1817           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
1818                                            n_left_to_next, bi0, next0);
1819         }
1820
1821       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
1822     }
1823
1824   return from_frame->n_vectors;
1825 }
1826
1827 static uword
1828 tcp4_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1829              vlib_frame_t * from_frame)
1830 {
1831   return tcp46_output_inline (vm, node, from_frame, 1 /* is_ip4 */ );
1832 }
1833
1834 static uword
1835 tcp6_output (vlib_main_t * vm, vlib_node_runtime_t * node,
1836              vlib_frame_t * from_frame)
1837 {
1838   return tcp46_output_inline (vm, node, from_frame, 0 /* is_ip4 */ );
1839 }
1840
1841 /* *INDENT-OFF* */
1842 VLIB_REGISTER_NODE (tcp4_output_node) =
1843 {
1844   .function = tcp4_output,.name = "tcp4-output",
1845     /* Takes a vector of packets. */
1846     .vector_size = sizeof (u32),
1847     .n_errors = TCP_N_ERROR,
1848     .error_strings = tcp_error_strings,
1849     .n_next_nodes = TCP_OUTPUT_N_NEXT,
1850     .next_nodes = {
1851 #define _(s,n) [TCP_OUTPUT_NEXT_##s] = n,
1852     foreach_tcp4_output_next
1853 #undef _
1854     },
1855     .format_buffer = format_tcp_header,
1856     .format_trace = format_tcp_tx_trace,
1857 };
1858 /* *INDENT-ON* */
1859
1860 VLIB_NODE_FUNCTION_MULTIARCH (tcp4_output_node, tcp4_output);
1861
1862 /* *INDENT-OFF* */
1863 VLIB_REGISTER_NODE (tcp6_output_node) =
1864 {
1865   .function = tcp6_output,
1866   .name = "tcp6-output",
1867     /* Takes a vector of packets. */
1868   .vector_size = sizeof (u32),
1869   .n_errors = TCP_N_ERROR,
1870   .error_strings = tcp_error_strings,
1871   .n_next_nodes = TCP_OUTPUT_N_NEXT,
1872   .next_nodes = {
1873 #define _(s,n) [TCP_OUTPUT_NEXT_##s] = n,
1874     foreach_tcp6_output_next
1875 #undef _
1876   },
1877   .format_buffer = format_tcp_header,
1878   .format_trace = format_tcp_tx_trace,
1879 };
1880 /* *INDENT-ON* */
1881
1882 VLIB_NODE_FUNCTION_MULTIARCH (tcp6_output_node, tcp6_output);
1883
1884 u32
1885 tcp_push_header (transport_connection_t * tconn, vlib_buffer_t * b)
1886 {
1887   tcp_connection_t *tc;
1888
1889   tc = (tcp_connection_t *) tconn;
1890   tcp_push_hdr_i (tc, b, TCP_STATE_ESTABLISHED, 0);
1891   ASSERT (seq_leq (tc->snd_una_max, tc->snd_una + tc->snd_wnd));
1892
1893   if (tc->rtt_ts == 0 && !tcp_in_cong_recovery (tc))
1894     {
1895       tc->rtt_ts = tcp_time_now ();
1896       tc->rtt_seq = tc->snd_nxt;
1897     }
1898   return 0;
1899 }
1900
1901 typedef enum _tcp_reset_next
1902 {
1903   TCP_RESET_NEXT_DROP,
1904   TCP_RESET_NEXT_IP_LOOKUP,
1905   TCP_RESET_N_NEXT
1906 } tcp_reset_next_t;
1907
1908 #define foreach_tcp4_reset_next         \
1909   _(DROP, "error-drop")                 \
1910   _(IP_LOOKUP, "ip4-lookup")
1911
1912 #define foreach_tcp6_reset_next         \
1913   _(DROP, "error-drop")                 \
1914   _(IP_LOOKUP, "ip6-lookup")
1915
1916 static uword
1917 tcp46_send_reset_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
1918                          vlib_frame_t * from_frame, u8 is_ip4)
1919 {
1920   u32 n_left_from, next_index, *from, *to_next;
1921   u32 my_thread_index = vm->thread_index;
1922
1923   from = vlib_frame_vector_args (from_frame);
1924   n_left_from = from_frame->n_vectors;
1925
1926   next_index = node->cached_next_index;
1927
1928   while (n_left_from > 0)
1929     {
1930       u32 n_left_to_next;
1931
1932       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
1933
1934       while (n_left_from > 0 && n_left_to_next > 0)
1935         {
1936           u32 bi0;
1937           vlib_buffer_t *b0;
1938           tcp_tx_trace_t *t0;
1939           tcp_header_t *th0;
1940           u32 error0 = TCP_ERROR_RST_SENT, next0 = TCP_RESET_NEXT_IP_LOOKUP;
1941
1942           bi0 = from[0];
1943           to_next[0] = bi0;
1944           from += 1;
1945           to_next += 1;
1946           n_left_from -= 1;
1947           n_left_to_next -= 1;
1948
1949           b0 = vlib_get_buffer (vm, bi0);
1950
1951           if (tcp_make_reset_in_place (vm, b0, vnet_buffer (b0)->tcp.flags,
1952                                        my_thread_index, is_ip4))
1953             {
1954               error0 = TCP_ERROR_LOOKUP_DROPS;
1955               next0 = TCP_RESET_NEXT_DROP;
1956               goto done;
1957             }
1958
1959           /* Prepare to send to IP lookup */
1960           vnet_buffer (b0)->sw_if_index[VLIB_TX] = 0;
1961           next0 = TCP_RESET_NEXT_IP_LOOKUP;
1962
1963         done:
1964           b0->error = node->errors[error0];
1965           b0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
1966           if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
1967             {
1968               th0 = vlib_buffer_get_current (b0);
1969               if (is_ip4)
1970                 th0 = ip4_next_header ((ip4_header_t *) th0);
1971               else
1972                 th0 = ip6_next_header ((ip6_header_t *) th0);
1973               t0 = vlib_add_trace (vm, node, b0, sizeof (*t0));
1974               clib_memcpy (&t0->tcp_header, th0, sizeof (t0->tcp_header));
1975             }
1976
1977           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
1978                                            n_left_to_next, bi0, next0);
1979         }
1980       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
1981     }
1982   return from_frame->n_vectors;
1983 }
1984
1985 static uword
1986 tcp4_send_reset (vlib_main_t * vm, vlib_node_runtime_t * node,
1987                  vlib_frame_t * from_frame)
1988 {
1989   return tcp46_send_reset_inline (vm, node, from_frame, 1);
1990 }
1991
1992 static uword
1993 tcp6_send_reset (vlib_main_t * vm, vlib_node_runtime_t * node,
1994                  vlib_frame_t * from_frame)
1995 {
1996   return tcp46_send_reset_inline (vm, node, from_frame, 0);
1997 }
1998
1999 /* *INDENT-OFF* */
2000 VLIB_REGISTER_NODE (tcp4_reset_node) = {
2001   .function = tcp4_send_reset,
2002   .name = "tcp4-reset",
2003   .vector_size = sizeof (u32),
2004   .n_errors = TCP_N_ERROR,
2005   .error_strings = tcp_error_strings,
2006   .n_next_nodes = TCP_RESET_N_NEXT,
2007   .next_nodes = {
2008 #define _(s,n) [TCP_RESET_NEXT_##s] = n,
2009     foreach_tcp4_reset_next
2010 #undef _
2011   },
2012   .format_trace = format_tcp_tx_trace,
2013 };
2014 /* *INDENT-ON* */
2015
2016 VLIB_NODE_FUNCTION_MULTIARCH (tcp4_reset_node, tcp4_send_reset);
2017
2018 /* *INDENT-OFF* */
2019 VLIB_REGISTER_NODE (tcp6_reset_node) = {
2020   .function = tcp6_send_reset,
2021   .name = "tcp6-reset",
2022   .vector_size = sizeof (u32),
2023   .n_errors = TCP_N_ERROR,
2024   .error_strings = tcp_error_strings,
2025   .n_next_nodes = TCP_RESET_N_NEXT,
2026   .next_nodes = {
2027 #define _(s,n) [TCP_RESET_NEXT_##s] = n,
2028     foreach_tcp6_reset_next
2029 #undef _
2030   },
2031   .format_trace = format_tcp_tx_trace,
2032 };
2033 /* *INDENT-ON* */
2034
2035 VLIB_NODE_FUNCTION_MULTIARCH (tcp6_reset_node, tcp6_send_reset);
2036
2037 /*
2038  * fd.io coding-style-patch-verification: ON
2039  *
2040  * Local Variables:
2041  * eval: (c-set-style "gnu")
2042  * End:
2043  */