src/vnet/tcp/tcp.c

   1 /*
   2  * Copyright (c) 2016-2019 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 /**
  17  * @file
  18  * @brief TCP host stack utilities
  19  */
  20
  21 #include <vnet/tcp/tcp.h>
  22 #include <vnet/session/session.h>
  23 #include <vnet/fib/fib.h>
  24 #include <vnet/dpo/load_balance.h>
  25 #include <vnet/dpo/receive_dpo.h>
  26 #include <vnet/ip-neighbor/ip_neighbor.h>
  27 #include <math.h>
  28
  29 tcp_main_t tcp_main;
  30
  31 typedef struct
  32 {
  33   fib_protocol_t nh_proto;
  34   vnet_link_t link_type;
  35   ip46_address_t ip;
  36   u32 sw_if_index;
  37   u8 is_add;
  38 } tcp_add_del_adj_args_t;
  39
  40 static void
  41 tcp_add_del_adj_cb (tcp_add_del_adj_args_t * args)
  42 {
  43   u32 ai;
  44   if (args->is_add)
  45     {
  46       adj_nbr_add_or_lock (args->nh_proto, args->link_type, &args->ip,
  47                            args->sw_if_index);
  48     }
  49   else
  50     {
  51       ai = adj_nbr_find (FIB_PROTOCOL_IP6, VNET_LINK_IP6, &args->ip,
  52                          args->sw_if_index);
  53       if (ai != ADJ_INDEX_INVALID)
  54         adj_unlock (ai);
  55     }
  56 }
  57
  58 static void
  59 tcp_add_del_adjacency (tcp_connection_t * tc, u8 is_add)
  60 {
  61   tcp_add_del_adj_args_t args = {
  62     .nh_proto = FIB_PROTOCOL_IP6,
  63     .link_type = VNET_LINK_IP6,
  64     .ip = tc->c_rmt_ip,
  65     .sw_if_index = tc->sw_if_index,
  66     .is_add = is_add
  67   };
  68   vlib_rpc_call_main_thread (tcp_add_del_adj_cb, (u8 *) & args,
  69                              sizeof (args));
  70 }
  71
  72 static void
  73 tcp_cc_init (tcp_connection_t * tc)
  74 {
  75   tc->cc_algo->init (tc);
  76 }
  77
  78 static void
  79 tcp_cc_cleanup (tcp_connection_t * tc)
  80 {
  81   if (tc->cc_algo->cleanup)
  82     tc->cc_algo->cleanup (tc);
  83 }
  84
  85 void
  86 tcp_cc_algo_register (tcp_cc_algorithm_type_e type,
  87                       const tcp_cc_algorithm_t * vft)
  88 {
  89   tcp_main_t *tm = vnet_get_tcp_main ();
  90   vec_validate (tm->cc_algos, type);
  91
  92   tm->cc_algos[type] = *vft;
  93   hash_set_mem (tm->cc_algo_by_name, vft->name, type);
  94 }
  95
  96 tcp_cc_algorithm_t *
  97 tcp_cc_algo_get (tcp_cc_algorithm_type_e type)
  98 {
  99   tcp_main_t *tm = vnet_get_tcp_main ();
 100   return &tm->cc_algos[type];
 101 }
 102
 103 tcp_cc_algorithm_type_e
 104 tcp_cc_algo_new_type (const tcp_cc_algorithm_t * vft)
 105 {
 106   tcp_main_t *tm = vnet_get_tcp_main ();
 107   tcp_cc_algo_register (++tm->cc_last_type, vft);
 108   return tm->cc_last_type;
 109 }
 110
 111 static u32
 112 tcp_connection_bind (u32 session_index, transport_endpoint_t * lcl)
 113 {
 114   tcp_main_t *tm = &tcp_main;
 115   tcp_connection_t *listener;
 116   void *iface_ip;
 117
 118   pool_get (tm->listener_pool, listener);
 119   clib_memset (listener, 0, sizeof (*listener));
 120
 121   listener->c_c_index = listener - tm->listener_pool;
 122   listener->c_lcl_port = lcl->port;
 123
 124   /* If we are provided a sw_if_index, bind using one of its ips */
 125   if (ip_is_zero (&lcl->ip, 1) && lcl->sw_if_index != ENDPOINT_INVALID_INDEX)
 126     {
 127       if ((iface_ip = ip_interface_get_first_ip (lcl->sw_if_index,
 128                                                  lcl->is_ip4)))
 129         ip_set (&lcl->ip, iface_ip, lcl->is_ip4);
 130     }
 131   ip_copy (&listener->c_lcl_ip, &lcl->ip, lcl->is_ip4);
 132   listener->c_is_ip4 = lcl->is_ip4;
 133   listener->c_proto = TRANSPORT_PROTO_TCP;
 134   listener->c_s_index = session_index;
 135   listener->c_fib_index = lcl->fib_index;
 136   listener->state = TCP_STATE_LISTEN;
 137   listener->cc_algo = tcp_cc_algo_get (tcp_cfg.cc_algo);
 138
 139   tcp_connection_timers_init (listener);
 140
 141   TCP_EVT (TCP_EVT_BIND, listener);
 142
 143   return listener->c_c_index;
 144 }
 145
 146 static u32
 147 tcp_session_bind (u32 session_index, transport_endpoint_t * tep)
 148 {
 149   return tcp_connection_bind (session_index, tep);
 150 }
 151
 152 static void
 153 tcp_connection_unbind (u32 listener_index)
 154 {
 155   tcp_main_t *tm = vnet_get_tcp_main ();
 156   tcp_connection_t *tc;
 157
 158   tc = pool_elt_at_index (tm->listener_pool, listener_index);
 159
 160   TCP_EVT (TCP_EVT_UNBIND, tc);
 161
 162   /* Poison the entry */
 163   if (CLIB_DEBUG > 0)
 164     clib_memset (tc, 0xFA, sizeof (*tc));
 165
 166   pool_put_index (tm->listener_pool, listener_index);
 167 }
 168
 169 static u32
 170 tcp_session_unbind (u32 listener_index)
 171 {
 172   tcp_connection_unbind (listener_index);
 173   return 0;
 174 }
 175
 176 static transport_connection_t *
 177 tcp_session_get_listener (u32 listener_index)
 178 {
 179   tcp_main_t *tm = vnet_get_tcp_main ();
 180   tcp_connection_t *tc;
 181   tc = pool_elt_at_index (tm->listener_pool, listener_index);
 182   return &tc->connection;
 183 }
 184
 185 /**
 186  * Cleanup half-open connection
 187  *
 188  */
 189 static void
 190 tcp_half_open_connection_del (tcp_connection_t * tc)
 191 {
 192   tcp_main_t *tm = vnet_get_tcp_main ();
 193   clib_spinlock_lock_if_init (&tm->half_open_lock);
 194   if (CLIB_DEBUG)
 195     clib_memset (tc, 0xFA, sizeof (*tc));
 196   pool_put (tm->half_open_connections, tc);
 197   clib_spinlock_unlock_if_init (&tm->half_open_lock);
 198 }
 199
 200 /**
 201  * Try to cleanup half-open connection
 202  *
 203  * If called from a thread that doesn't own tc, the call won't have any
 204  * effect.
 205  *
 206  * @param tc - connection to be cleaned up
 207  * @return non-zero if cleanup failed.
 208  */
 209 int
 210 tcp_half_open_connection_cleanup (tcp_connection_t * tc)
 211 {
 212   /* Make sure this is the owning thread */
 213   if (tc->c_thread_index != vlib_get_thread_index ())
 214     return 1;
 215   tcp_timer_reset (tc, TCP_TIMER_RETRANSMIT_SYN);
 216   tcp_half_open_connection_del (tc);
 217   return 0;
 218 }
 219
 220 static tcp_connection_t *
 221 tcp_half_open_connection_new (void)
 222 {
 223   tcp_main_t *tm = vnet_get_tcp_main ();
 224   tcp_connection_t *tc = 0;
 225   ASSERT (vlib_get_thread_index () == 0);
 226   pool_get (tm->half_open_connections, tc);
 227   clib_memset (tc, 0, sizeof (*tc));
 228   tc->c_c_index = tc - tm->half_open_connections;
 229   return tc;
 230 }
 231
 232 /**
 233  * Cleans up connection state.
 234  *
 235  * No notifications.
 236  */
 237 void
 238 tcp_connection_cleanup (tcp_connection_t * tc)
 239 {
 240   TCP_EVT (TCP_EVT_DELETE, tc);
 241
 242   /* Cleanup local endpoint if this was an active connect */
 243   if (!(tc->cfg_flags & TCP_CFG_F_NO_ENDPOINT))
 244     transport_endpoint_cleanup (TRANSPORT_PROTO_TCP, &tc->c_lcl_ip,
 245                                 tc->c_lcl_port);
 246
 247   /* Check if connection is not yet fully established */
 248   if (tc->state == TCP_STATE_SYN_SENT)
 249     {
 250       /* Try to remove the half-open connection. If this is not the owning
 251        * thread, tc won't be removed. Retransmit or establish timers will
 252        * eventually expire and call again cleanup on the right thread. */
 253       if (tcp_half_open_connection_cleanup (tc))
 254         tc->flags |= TCP_CONN_HALF_OPEN_DONE;
 255     }
 256   else
 257     {
 258       /* Make sure all timers are cleared */
 259       tcp_connection_timers_reset (tc);
 260
 261       if (!tc->c_is_ip4 && ip6_address_is_link_local_unicast (&tc->c_rmt_ip6))
 262         tcp_add_del_adjacency (tc, 0);
 263
 264       tcp_cc_cleanup (tc);
 265       vec_free (tc->snd_sacks);
 266       vec_free (tc->snd_sacks_fl);
 267       vec_free (tc->rcv_opts.sacks);
 268       pool_free (tc->sack_sb.holes);
 269
 270       if (tc->cfg_flags & TCP_CFG_F_RATE_SAMPLE)
 271         tcp_bt_cleanup (tc);
 272
 273       tcp_connection_free (tc);
 274     }
 275 }
 276
 277 /**
 278  * Connection removal.
 279  *
 280  * This should be called only once connection enters CLOSED state. Note
 281  * that it notifies the session of the removal event, so if the goal is to
 282  * just remove the connection, call tcp_connection_cleanup instead.
 283  */
 284 void
 285 tcp_connection_del (tcp_connection_t * tc)
 286 {
 287   session_transport_delete_notify (&tc->connection);
 288   tcp_connection_cleanup (tc);
 289 }
 290
 291 tcp_connection_t *
 292 tcp_connection_alloc (u8 thread_index)
 293 {
 294   tcp_worker_ctx_t *wrk = tcp_get_worker (thread_index);
 295   tcp_connection_t *tc;
 296
 297   pool_get (wrk->connections, tc);
 298   clib_memset (tc, 0, sizeof (*tc));
 299   tc->c_c_index = tc - wrk->connections;
 300   tc->c_thread_index = thread_index;
 301   return tc;
 302 }
 303
 304 tcp_connection_t *
 305 tcp_connection_alloc_w_base (u8 thread_index, tcp_connection_t * base)
 306 {
 307   tcp_worker_ctx_t *wrk = tcp_get_worker (thread_index);
 308   tcp_connection_t *tc;
 309
 310   pool_get (wrk->connections, tc);
 311   clib_memcpy_fast (tc, base, sizeof (*tc));
 312   tc->c_c_index = tc - wrk->connections;
 313   tc->c_thread_index = thread_index;
 314   return tc;
 315 }
 316
 317 void
 318 tcp_connection_free (tcp_connection_t * tc)
 319 {
 320   tcp_worker_ctx_t *wrk = tcp_get_worker (tc->c_thread_index);
 321   if (CLIB_DEBUG)
 322     {
 323       clib_memset (tc, 0xFA, sizeof (*tc));
 324       pool_put (wrk->connections, tc);
 325       return;
 326     }
 327   pool_put (wrk->connections, tc);
 328 }
 329
 330 void
 331 tcp_program_cleanup (tcp_worker_ctx_t * wrk, tcp_connection_t * tc)
 332 {
 333   tcp_cleanup_req_t *req;
 334   clib_time_type_t now;
 335
 336   now = transport_time_now (tc->c_thread_index);
 337   clib_fifo_add2 (wrk->pending_cleanups, req);
 338   req->connection_index = tc->c_c_index;
 339   req->free_time = now + tcp_cfg.cleanup_time;
 340 }
 341
 342 /**
 343  * Begin connection closing procedure.
 344  *
 345  * If at the end the connection is not in CLOSED state, it is not removed.
 346  * Instead, we rely on on TCP to advance through state machine to either
 347  * 1) LAST_ACK (passive close) whereby when the last ACK is received
 348  * tcp_connection_del is called. This notifies session of the delete and
 349  * calls cleanup.
 350  * 2) TIME_WAIT (active close) whereby after 2MSL the 2MSL timer triggers
 351  * and cleanup is called.
 352  *
 353  * N.B. Half-close connections are not supported
 354  */
 355 void
 356 tcp_connection_close (tcp_connection_t * tc)
 357 {
 358   TCP_EVT (TCP_EVT_CLOSE, tc);
 359
 360   /* Send/Program FIN if needed and switch state */
 361   switch (tc->state)
 362     {
 363     case TCP_STATE_SYN_SENT:
 364       /* Try to cleanup. If not on the right thread, mark as half-open done.
 365        * Connection will be cleaned up when establish timer pops */
 366       tcp_connection_cleanup (tc);
 367       break;
 368     case TCP_STATE_SYN_RCVD:
 369       tcp_connection_timers_reset (tc);
 370       tcp_send_fin (tc);
 371       tcp_connection_set_state (tc, TCP_STATE_FIN_WAIT_1);
 372       tcp_timer_update (tc, TCP_TIMER_WAITCLOSE, tcp_cfg.finwait1_time);
 373       break;
 374     case TCP_STATE_ESTABLISHED:
 375       /* If closing with unread data, reset the connection */
 376       if (transport_max_rx_dequeue (&tc->connection))
 377         {
 378           tcp_send_reset (tc);
 379           tcp_connection_timers_reset (tc);
 380           tcp_connection_set_state (tc, TCP_STATE_CLOSED);
 381           session_transport_closed_notify (&tc->connection);
 382           tcp_program_cleanup (tcp_get_worker (tc->c_thread_index), tc);
 383           tcp_worker_stats_inc (tc->c_thread_index, rst_unread, 1);
 384           break;
 385         }
 386       if (!transport_max_tx_dequeue (&tc->connection))
 387         tcp_send_fin (tc);
 388       else
 389         tc->flags |= TCP_CONN_FINPNDG;
 390       tcp_connection_set_state (tc, TCP_STATE_FIN_WAIT_1);
 391       /* Set a timer in case the peer stops responding. Otherwise the
 392        * connection will be stuck here forever. */
 393       ASSERT (tc->timers[TCP_TIMER_WAITCLOSE] == TCP_TIMER_HANDLE_INVALID);
 394       tcp_timer_set (tc, TCP_TIMER_WAITCLOSE, tcp_cfg.finwait1_time);
 395       break;
 396     case TCP_STATE_CLOSE_WAIT:
 397       if (!transport_max_tx_dequeue (&tc->connection))
 398         {
 399           tcp_send_fin (tc);
 400           tcp_connection_timers_reset (tc);
 401           tcp_connection_set_state (tc, TCP_STATE_LAST_ACK);
 402           tcp_timer_update (tc, TCP_TIMER_WAITCLOSE, tcp_cfg.lastack_time);
 403         }
 404       else
 405         tc->flags |= TCP_CONN_FINPNDG;
 406       break;
 407     case TCP_STATE_FIN_WAIT_1:
 408       tcp_timer_update (tc, TCP_TIMER_WAITCLOSE, tcp_cfg.finwait1_time);
 409       break;
 410     case TCP_STATE_CLOSED:
 411       /* Cleanup should've been programmed already */
 412       break;
 413     default:
 414       TCP_DBG ("state: %u", tc->state);
 415     }
 416 }
 417
 418 static void
 419 tcp_session_close (u32 conn_index, u32 thread_index)
 420 {
 421   tcp_connection_t *tc;
 422   tc = tcp_connection_get (conn_index, thread_index);
 423   tcp_connection_close (tc);
 424 }
 425
 426 static void
 427 tcp_session_cleanup (u32 conn_index, u32 thread_index)
 428 {
 429   tcp_connection_t *tc;
 430   tc = tcp_connection_get (conn_index, thread_index);
 431   if (!tc)
 432     return;
 433   tcp_connection_set_state (tc, TCP_STATE_CLOSED);
 434   tcp_connection_cleanup (tc);
 435 }
 436
 437 static void
 438 tcp_session_reset (u32 conn_index, u32 thread_index)
 439 {
 440   tcp_connection_t *tc;
 441   tc = tcp_connection_get (conn_index, thread_index);
 442   tcp_send_reset (tc);
 443   tcp_connection_timers_reset (tc);
 444   tcp_cong_recovery_off (tc);
 445   tcp_connection_set_state (tc, TCP_STATE_CLOSED);
 446   session_transport_closed_notify (&tc->connection);
 447   tcp_program_cleanup (tcp_get_worker (thread_index), tc);
 448 }
 449
 450 /**
 451  * Initialize all connection timers as invalid
 452  */
 453 void
 454 tcp_connection_timers_init (tcp_connection_t * tc)
 455 {
 456   int i;
 457
 458   /* Set all to invalid */
 459   for (i = 0; i < TCP_N_TIMERS; i++)
 460     {
 461       tc->timers[i] = TCP_TIMER_HANDLE_INVALID;
 462     }
 463
 464   tc->rto = TCP_RTO_INIT;
 465 }
 466
 467 /**
 468  * Stop all connection timers
 469  */
 470 void
 471 tcp_connection_timers_reset (tcp_connection_t * tc)
 472 {
 473   int i;
 474   for (i = 0; i < TCP_N_TIMERS; i++)
 475     {
 476       tcp_timer_reset (tc, i);
 477     }
 478 }
 479
 480 #if 0
 481 typedef struct ip4_tcp_hdr
 482 {
 483   ip4_header_t ip;
 484   tcp_header_t tcp;
 485 } ip4_tcp_hdr_t;
 486
 487 typedef struct ip6_tcp_hdr
 488 {
 489   ip6_header_t ip;
 490   tcp_header_t tcp;
 491 } ip6_tcp_hdr_t;
 492
 493 static void
 494 tcp_connection_select_lb_bucket (tcp_connection_t * tc, const dpo_id_t * dpo,
 495                                  dpo_id_t * result)
 496 {
 497   const dpo_id_t *choice;
 498   load_balance_t *lb;
 499   int hash;
 500
 501   lb = load_balance_get (dpo->dpoi_index);
 502   if (tc->c_is_ip4)
 503     {
 504       ip4_tcp_hdr_t hdr;
 505       clib_memset (&hdr, 0, sizeof (hdr));
 506       hdr.ip.protocol = IP_PROTOCOL_TCP;
 507       hdr.ip.address_pair.src.as_u32 = tc->c_lcl_ip.ip4.as_u32;
 508       hdr.ip.address_pair.dst.as_u32 = tc->c_rmt_ip.ip4.as_u32;
 509       hdr.tcp.src_port = tc->c_lcl_port;
 510       hdr.tcp.dst_port = tc->c_rmt_port;
 511       hash = ip4_compute_flow_hash (&hdr.ip, lb->lb_hash_config);
 512     }
 513   else
 514     {
 515       ip6_tcp_hdr_t hdr;
 516       clib_memset (&hdr, 0, sizeof (hdr));
 517       hdr.ip.protocol = IP_PROTOCOL_TCP;
 518       clib_memcpy_fast (&hdr.ip.src_address, &tc->c_lcl_ip.ip6,
 519                         sizeof (ip6_address_t));
 520       clib_memcpy_fast (&hdr.ip.dst_address, &tc->c_rmt_ip.ip6,
 521                         sizeof (ip6_address_t));
 522       hdr.tcp.src_port = tc->c_lcl_port;
 523       hdr.tcp.dst_port = tc->c_rmt_port;
 524       hash = ip6_compute_flow_hash (&hdr.ip, lb->lb_hash_config);
 525     }
 526   choice = load_balance_get_bucket_i (lb, hash & lb->lb_n_buckets_minus_1);
 527   dpo_copy (result, choice);
 528 }
 529
 530 fib_node_index_t
 531 tcp_lookup_rmt_in_fib (tcp_connection_t * tc)
 532 {
 533   fib_prefix_t prefix;
 534   u32 fib_index;
 535
 536   clib_memcpy_fast (&prefix.fp_addr, &tc->c_rmt_ip, sizeof (prefix.fp_addr));
 537   prefix.fp_proto = tc->c_is_ip4 ? FIB_PROTOCOL_IP4 : FIB_PROTOCOL_IP6;
 538   prefix.fp_len = tc->c_is_ip4 ? 32 : 128;
 539   fib_index = fib_table_find (prefix.fp_proto, tc->c_fib_index);
 540   return fib_table_lookup (fib_index, &prefix);
 541 }
 542
 543 static int
 544 tcp_connection_stack_on_fib_entry (tcp_connection_t * tc)
 545 {
 546   dpo_id_t choice = DPO_INVALID;
 547   u32 output_node_index;
 548   fib_entry_t *fe;
 549
 550   fe = fib_entry_get (tc->c_rmt_fei);
 551   if (fe->fe_lb.dpoi_type != DPO_LOAD_BALANCE)
 552     return -1;
 553
 554   tcp_connection_select_lb_bucket (tc, &fe->fe_lb, &choice);
 555
 556   output_node_index =
 557     tc->c_is_ip4 ? tcp4_output_node.index : tcp6_output_node.index;
 558   dpo_stack_from_node (output_node_index, &tc->c_rmt_dpo, &choice);
 559   return 0;
 560 }
 561
 562 /** Stack tcp connection on peer's fib entry.
 563  *
 564  * This ultimately populates the dpo the connection will use to send packets.
 565  */
 566 static void
 567 tcp_connection_fib_attach (tcp_connection_t * tc)
 568 {
 569   tc->c_rmt_fei = tcp_lookup_rmt_in_fib (tc);
 570
 571   ASSERT (tc->c_rmt_fei != FIB_NODE_INDEX_INVALID);
 572
 573   tcp_connection_stack_on_fib_entry (tc);
 574 }
 575 #endif /* 0 */
 576
 577 /**
 578  * Generate random iss as per rfc6528
 579  */
 580 static u32
 581 tcp_generate_random_iss (tcp_connection_t * tc)
 582 {
 583   tcp_main_t *tm = &tcp_main;
 584   u64 tmp;
 585
 586   if (tc->c_is_ip4)
 587     tmp = (u64) tc->c_lcl_ip.ip4.as_u32 << 32 | (u64) tc->c_rmt_ip.ip4.as_u32;
 588   else
 589     tmp = tc->c_lcl_ip.ip6.as_u64[0] ^ tc->c_lcl_ip.ip6.as_u64[1]
 590       ^ tc->c_rmt_ip.ip6.as_u64[0] ^ tc->c_rmt_ip.ip6.as_u64[1];
 591
 592   tmp ^= tm->iss_seed.first | ((u64) tc->c_lcl_port << 16 | tc->c_rmt_port);
 593   tmp ^= tm->iss_seed.second;
 594   tmp = clib_xxhash (tmp) + clib_cpu_time_now ();
 595   return ((tmp >> 32) ^ (tmp & 0xffffffff));
 596 }
 597
 598 /**
 599  * Initialize max segment size we're able to process.
 600  *
 601  * The value is constrained by the output interface's MTU and by the size
 602  * of the IP and TCP headers (see RFC6691). It is also what we advertise
 603  * to our peer.
 604  */
 605 static void
 606 tcp_init_rcv_mss (tcp_connection_t * tc)
 607 {
 608   u8 ip_hdr_len;
 609
 610   /* Already provided at connection init time */
 611   if (tc->mss)
 612     return;
 613
 614   ip_hdr_len = tc->c_is_ip4 ? sizeof (ip4_header_t) : sizeof (ip6_header_t);
 615   tc->mss = tcp_cfg.default_mtu - sizeof (tcp_header_t) - ip_hdr_len;
 616 }
 617
 618 static void
 619 tcp_init_mss (tcp_connection_t * tc)
 620 {
 621   u16 default_min_mss = 536;
 622
 623   tcp_init_rcv_mss (tc);
 624
 625   /* TODO consider PMTU discovery */
 626   tc->snd_mss = clib_min (tc->rcv_opts.mss, tc->mss);
 627
 628   if (tc->snd_mss < 45)
 629     {
 630       /* Assume that at least the min default mss works */
 631       tc->snd_mss = default_min_mss;
 632       tc->rcv_opts.mss = default_min_mss;
 633     }
 634
 635   /* We should have enough space for 40 bytes of options */
 636   ASSERT (tc->snd_mss > 45);
 637
 638   /* If we use timestamp option, account for it */
 639   if (tcp_opts_tstamp (&tc->rcv_opts))
 640     tc->snd_mss -= TCP_OPTION_LEN_TIMESTAMP;
 641 }
 642
 643 /**
 644  * Initialize connection send variables.
 645  */
 646 void
 647 tcp_init_snd_vars (tcp_connection_t * tc)
 648 {
 649   /*
 650    * We use the time to randomize iss and for setting up the initial
 651    * timestamp. Make sure it's updated otherwise syn and ack in the
 652    * handshake may make it look as if time has flown in the opposite
 653    * direction for us.
 654    */
 655   tcp_set_time_now (tcp_get_worker (vlib_get_thread_index ()));
 656
 657   tcp_init_rcv_mss (tc);
 658   tc->iss = tcp_generate_random_iss (tc);
 659   tc->snd_una = tc->iss;
 660   tc->snd_nxt = tc->iss + 1;
 661   tc->snd_una_max = tc->snd_nxt;
 662   tc->srtt = 100;               /* 100 ms */
 663
 664   if (!tcp_cfg.csum_offload)
 665     tc->cfg_flags |= TCP_CFG_F_NO_CSUM_OFFLOAD;
 666 }
 667
 668 void
 669 tcp_enable_pacing (tcp_connection_t * tc)
 670 {
 671   u32 byte_rate;
 672   byte_rate = tc->cwnd / (tc->srtt * TCP_TICK);
 673   transport_connection_tx_pacer_init (&tc->connection, byte_rate, tc->cwnd);
 674   tc->mrtt_us = (u32) ~ 0;
 675 }
 676
 677 /** Initialize tcp connection variables
 678  *
 679  * Should be called after having received a msg from the peer, i.e., a SYN or
 680  * a SYNACK, such that connection options have already been exchanged. */
 681 void
 682 tcp_connection_init_vars (tcp_connection_t * tc)
 683 {
 684   tcp_connection_timers_init (tc);
 685   tcp_init_mss (tc);
 686   scoreboard_init (&tc->sack_sb);
 687   if (tc->state == TCP_STATE_SYN_RCVD)
 688     tcp_init_snd_vars (tc);
 689
 690   tcp_cc_init (tc);
 691
 692   if (!tc->c_is_ip4 && ip6_address_is_link_local_unicast (&tc->c_rmt_ip6))
 693     tcp_add_del_adjacency (tc, 1);
 694
 695   /*  tcp_connection_fib_attach (tc); */
 696
 697   if (transport_connection_is_tx_paced (&tc->connection)
 698       || tcp_cfg.enable_tx_pacing)
 699     tcp_enable_pacing (tc);
 700
 701   if (tc->cfg_flags & TCP_CFG_F_RATE_SAMPLE)
 702     tcp_bt_init (tc);
 703
 704   if (!tcp_cfg.allow_tso)
 705     tc->cfg_flags |= TCP_CFG_F_NO_TSO;
 706
 707   tc->start_ts = tcp_time_now_us (tc->c_thread_index);
 708 }
 709
 710 void
 711 tcp_init_w_buffer (tcp_connection_t * tc, vlib_buffer_t * b, u8 is_ip4)
 712 {
 713   tcp_header_t *th = tcp_buffer_hdr (b);
 714
 715   tc->c_lcl_port = th->dst_port;
 716   tc->c_rmt_port = th->src_port;
 717   tc->c_is_ip4 = is_ip4;
 718
 719   if (is_ip4)
 720     {
 721       ip4_header_t *ip4 = vlib_buffer_get_current (b);
 722       tc->c_lcl_ip4.as_u32 = ip4->dst_address.as_u32;
 723       tc->c_rmt_ip4.as_u32 = ip4->src_address.as_u32;
 724     }
 725   else
 726     {
 727       ip6_header_t *ip6 = vlib_buffer_get_current (b);
 728       clib_memcpy_fast (&tc->c_lcl_ip6, &ip6->dst_address,
 729                         sizeof (ip6_address_t));
 730       clib_memcpy_fast (&tc->c_rmt_ip6, &ip6->src_address,
 731                         sizeof (ip6_address_t));
 732     }
 733
 734   tc->irs = vnet_buffer (b)->tcp.seq_number;
 735   tc->rcv_nxt = vnet_buffer (b)->tcp.seq_number + 1;
 736   tc->rcv_las = tc->rcv_nxt;
 737   tc->sw_if_index = vnet_buffer (b)->sw_if_index[VLIB_RX];
 738   tc->snd_wl1 = vnet_buffer (b)->tcp.seq_number;
 739   tc->snd_wl2 = vnet_buffer (b)->tcp.ack_number;
 740
 741   /* RFC1323: TSval timestamps sent on {SYN} and {SYN,ACK}
 742    * segments are used to initialize PAWS. */
 743   if (tcp_opts_tstamp (&tc->rcv_opts))
 744     {
 745       tc->tsval_recent = tc->rcv_opts.tsval;
 746       tc->tsval_recent_age = tcp_time_now ();
 747     }
 748
 749   if (tcp_opts_wscale (&tc->rcv_opts))
 750     tc->snd_wscale = tc->rcv_opts.wscale;
 751
 752   tc->snd_wnd = clib_net_to_host_u16 (th->window) << tc->snd_wscale;
 753 }
 754
 755 static int
 756 tcp_alloc_custom_local_endpoint (tcp_main_t * tm, ip46_address_t * lcl_addr,
 757                                  u16 * lcl_port, u8 is_ip4)
 758 {
 759   int index, port;
 760   if (is_ip4)
 761     {
 762       index = tm->last_v4_addr_rotor++;
 763       if (tm->last_v4_addr_rotor >= vec_len (tcp_cfg.ip4_src_addrs))
 764         tm->last_v4_addr_rotor = 0;
 765       lcl_addr->ip4.as_u32 = tcp_cfg.ip4_src_addrs[index].as_u32;
 766     }
 767   else
 768     {
 769       index = tm->last_v6_addr_rotor++;
 770       if (tm->last_v6_addr_rotor >= vec_len (tcp_cfg.ip6_src_addrs))
 771         tm->last_v6_addr_rotor = 0;
 772       clib_memcpy_fast (&lcl_addr->ip6, &tcp_cfg.ip6_src_addrs[index],
 773                         sizeof (ip6_address_t));
 774     }
 775   port = transport_alloc_local_port (TRANSPORT_PROTO_TCP, lcl_addr);
 776   if (port < 1)
 777     {
 778       clib_warning ("Failed to allocate src port");
 779       return -1;
 780     }
 781   *lcl_port = port;
 782   return 0;
 783 }
 784
 785 static int
 786 tcp_session_open (transport_endpoint_cfg_t * rmt)
 787 {
 788   tcp_main_t *tm = vnet_get_tcp_main ();
 789   tcp_connection_t *tc;
 790   ip46_address_t lcl_addr;
 791   u16 lcl_port;
 792   int rv;
 793
 794   /*
 795    * Allocate local endpoint
 796    */
 797   if ((rmt->is_ip4 && vec_len (tcp_cfg.ip4_src_addrs))
 798       || (!rmt->is_ip4 && vec_len (tcp_cfg.ip6_src_addrs)))
 799     rv = tcp_alloc_custom_local_endpoint (tm, &lcl_addr, &lcl_port,
 800                                           rmt->is_ip4);
 801   else
 802     rv = transport_alloc_local_endpoint (TRANSPORT_PROTO_TCP,
 803                                          rmt, &lcl_addr, &lcl_port);
 804
 805   if (rv)
 806     return -1;
 807
 808   /*
 809    * Create connection and send SYN
 810    */
 811   clib_spinlock_lock_if_init (&tm->half_open_lock);
 812   tc = tcp_half_open_connection_new ();
 813   ip_copy (&tc->c_rmt_ip, &rmt->ip, rmt->is_ip4);
 814   ip_copy (&tc->c_lcl_ip, &lcl_addr, rmt->is_ip4);
 815   tc->c_rmt_port = rmt->port;
 816   tc->c_lcl_port = clib_host_to_net_u16 (lcl_port);
 817   tc->c_is_ip4 = rmt->is_ip4;
 818   tc->c_proto = TRANSPORT_PROTO_TCP;
 819   tc->c_fib_index = rmt->fib_index;
 820   tc->cc_algo = tcp_cc_algo_get (tcp_cfg.cc_algo);
 821   /* The other connection vars will be initialized after SYN ACK */
 822   tcp_connection_timers_init (tc);
 823   tc->mss = rmt->mss;
 824
 825   TCP_EVT (TCP_EVT_OPEN, tc);
 826   tc->state = TCP_STATE_SYN_SENT;
 827   tcp_init_snd_vars (tc);
 828   tcp_send_syn (tc);
 829   clib_spinlock_unlock_if_init (&tm->half_open_lock);
 830
 831   return tc->c_c_index;
 832 }
 833
 834 const char *tcp_fsm_states[] = {
 835 #define _(sym, str) str,
 836   foreach_tcp_fsm_state
 837 #undef _
 838 };
 839
 840 u8 *
 841 format_tcp_state (u8 * s, va_list * args)
 842 {
 843   u32 state = va_arg (*args, u32);
 844
 845   if (state < TCP_N_STATES)
 846     s = format (s, "%s", tcp_fsm_states[state]);
 847   else
 848     s = format (s, "UNKNOWN (%d (0x%x))", state, state);
 849   return s;
 850 }
 851
 852 const char *tcp_cfg_flags_str[] = {
 853 #define _(sym, str) str,
 854   foreach_tcp_cfg_flag
 855 #undef _
 856 };
 857
 858 static u8 *
 859 format_tcp_cfg_flags (u8 * s, va_list * args)
 860 {
 861   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 862   int i, last = -1;
 863
 864   for (i = 0; i < TCP_CFG_N_FLAG_BITS; i++)
 865     if (tc->cfg_flags & (1 << i))
 866       last = i;
 867   for (i = 0; i < last; i++)
 868     {
 869       if (tc->cfg_flags & (1 << i))
 870         s = format (s, "%s, ", tcp_cfg_flags_str[i]);
 871     }
 872   if (last >= 0)
 873     s = format (s, "%s", tcp_cfg_flags_str[last]);
 874   return s;
 875 }
 876
 877 const char *tcp_connection_flags_str[] = {
 878 #define _(sym, str) str,
 879   foreach_tcp_connection_flag
 880 #undef _
 881 };
 882
 883 static u8 *
 884 format_tcp_connection_flags (u8 * s, va_list * args)
 885 {
 886   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 887   int i, last = -1;
 888
 889   for (i = 0; i < TCP_CONN_N_FLAG_BITS; i++)
 890     if (tc->flags & (1 << i))
 891       last = i;
 892   for (i = 0; i < last; i++)
 893     {
 894       if (tc->flags & (1 << i))
 895         s = format (s, "%s, ", tcp_connection_flags_str[i]);
 896     }
 897   if (last >= 0)
 898     s = format (s, "%s", tcp_connection_flags_str[last]);
 899   return s;
 900 }
 901
 902 const char *tcp_conn_timers[] = {
 903 #define _(sym, str) str,
 904   foreach_tcp_timer
 905 #undef _
 906 };
 907
 908 static u8 *
 909 format_tcp_timers (u8 * s, va_list * args)
 910 {
 911   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 912   int i, last = -1;
 913
 914   for (i = 0; i < TCP_N_TIMERS; i++)
 915     if (tc->timers[i] != TCP_TIMER_HANDLE_INVALID)
 916       last = i;
 917
 918   for (i = 0; i < last; i++)
 919     {
 920       if (tc->timers[i] != TCP_TIMER_HANDLE_INVALID)
 921         s = format (s, "%s,", tcp_conn_timers[i]);
 922     }
 923
 924   if (last >= 0)
 925     s = format (s, "%s", tcp_conn_timers[i]);
 926
 927   return s;
 928 }
 929
 930 static u8 *
 931 format_tcp_congestion_status (u8 * s, va_list * args)
 932 {
 933   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 934   if (tcp_in_recovery (tc))
 935     s = format (s, "recovery");
 936   else if (tcp_in_fastrecovery (tc))
 937     s = format (s, "fastrecovery");
 938   else
 939     s = format (s, "none");
 940   return s;
 941 }
 942
 943 static i32
 944 tcp_rcv_wnd_available (tcp_connection_t * tc)
 945 {
 946   return (i32) tc->rcv_wnd - (tc->rcv_nxt - tc->rcv_las);
 947 }
 948
 949 static u8 *
 950 format_tcp_congestion (u8 * s, va_list * args)
 951 {
 952   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 953   u32 indent = format_get_indent (s), prr_space = 0;
 954
 955   s = format (s, "%U ", format_tcp_congestion_status, tc);
 956   s = format (s, "algo %s cwnd %u ssthresh %u bytes_acked %u\n",
 957               tc->cc_algo->name, tc->cwnd, tc->ssthresh, tc->bytes_acked);
 958   s = format (s, "%Ucc space %u prev_cwnd %u prev_ssthresh %u\n",
 959               format_white_space, indent, tcp_available_cc_snd_space (tc),
 960               tc->prev_cwnd, tc->prev_ssthresh);
 961   s = format (s, "%Usnd_cong %u dupack %u limited_tx %u\n",
 962               format_white_space, indent, tc->snd_congestion - tc->iss,
 963               tc->rcv_dupacks, tc->limited_transmit - tc->iss);
 964   s = format (s, "%Urxt_bytes %u rxt_delivered %u rxt_head %u rxt_ts %u\n",
 965               format_white_space, indent, tc->snd_rxt_bytes,
 966               tc->rxt_delivered, tc->rxt_head - tc->iss,
 967               tcp_time_now_w_thread (tc->c_thread_index) - tc->snd_rxt_ts);
 968   if (tcp_in_fastrecovery (tc))
 969     prr_space = tcp_fastrecovery_prr_snd_space (tc);
 970   s = format (s, "%Uprr_start %u prr_delivered %u prr space %u\n",
 971               format_white_space, indent, tc->prr_start - tc->iss,
 972               tc->prr_delivered, prr_space);
 973   return s;
 974 }
 975
 976 static u8 *
 977 format_tcp_stats (u8 * s, va_list * args)
 978 {
 979   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
 980   u32 indent = format_get_indent (s);
 981   s = format (s, "in segs %lu dsegs %lu bytes %lu dupacks %u\n",
 982               tc->segs_in, tc->data_segs_in, tc->bytes_in, tc->dupacks_in);
 983   s = format (s, "%Uout segs %lu dsegs %lu bytes %lu dupacks %u\n",
 984               format_white_space, indent, tc->segs_out,
 985               tc->data_segs_out, tc->bytes_out, tc->dupacks_out);
 986   s = format (s, "%Ufr %u tr %u rxt segs %lu bytes %lu duration %.3f\n",
 987               format_white_space, indent, tc->fr_occurences,
 988               tc->tr_occurences, tc->segs_retrans, tc->bytes_retrans,
 989               tcp_time_now_us (tc->c_thread_index) - tc->start_ts);
 990   s = format (s, "%Uerr wnd data below %u above %u ack below %u above %u",
 991               format_white_space, indent, tc->errors.below_data_wnd,
 992               tc->errors.above_data_wnd, tc->errors.below_ack_wnd,
 993               tc->errors.above_ack_wnd);
 994   return s;
 995 }
 996
 997 static u8 *
 998 format_tcp_vars (u8 * s, va_list * args)
 999 {
1000   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1001   s = format (s, " index: %u cfg: %U flags: %U timers: %U\n", tc->c_c_index,
1002               format_tcp_cfg_flags, tc, format_tcp_connection_flags, tc,
1003               format_tcp_timers, tc);
1004   s = format (s, " snd_una %u snd_nxt %u snd_una_max %u",
1005               tc->snd_una - tc->iss, tc->snd_nxt - tc->iss,
1006               tc->snd_una_max - tc->iss);
1007   s = format (s, " rcv_nxt %u rcv_las %u\n",
1008               tc->rcv_nxt - tc->irs, tc->rcv_las - tc->irs);
1009   s = format (s, " snd_wnd %u rcv_wnd %u rcv_wscale %u ",
1010               tc->snd_wnd, tc->rcv_wnd, tc->rcv_wscale);
1011   s = format (s, "snd_wl1 %u snd_wl2 %u\n", tc->snd_wl1 - tc->irs,
1012               tc->snd_wl2 - tc->iss);
1013   s = format (s, " flight size %u out space %u rcv_wnd_av %u",
1014               tcp_flight_size (tc), tcp_available_output_snd_space (tc),
1015               tcp_rcv_wnd_available (tc));
1016   s = format (s, " tsval_recent %u\n", tc->tsval_recent);
1017   s = format (s, " tsecr %u tsecr_last_ack %u tsval_recent_age %u",
1018               tc->rcv_opts.tsecr, tc->tsecr_last_ack,
1019               tcp_time_now () - tc->tsval_recent_age);
1020   s = format (s, " snd_mss %u\n", tc->snd_mss);
1021   s = format (s, " rto %u rto_boff %u srtt %u us %.3f rttvar %u rtt_ts %.4f",
1022               tc->rto, tc->rto_boff, tc->srtt, tc->mrtt_us * 1000, tc->rttvar,
1023               tc->rtt_ts);
1024   s = format (s, " rtt_seq %u\n", tc->rtt_seq - tc->iss);
1025   s = format (s, " next_node %u opaque 0x%x fib_index %u\n",
1026               tc->next_node_index, tc->next_node_opaque, tc->c_fib_index);
1027   s = format (s, " cong:   %U", format_tcp_congestion, tc);
1028
1029   if (tc->state >= TCP_STATE_ESTABLISHED)
1030     {
1031       s = format (s, " sboard: %U\n", format_tcp_scoreboard, &tc->sack_sb,
1032                   tc);
1033       s = format (s, " stats: %U\n", format_tcp_stats, tc);
1034     }
1035   if (vec_len (tc->snd_sacks))
1036     s = format (s, " sacks tx: %U\n", format_tcp_sacks, tc);
1037
1038   return s;
1039 }
1040
1041 u8 *
1042 format_tcp_connection_id (u8 * s, va_list * args)
1043 {
1044   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1045   if (!tc)
1046     return s;
1047   if (tc->c_is_ip4)
1048     {
1049       s = format (s, "[%d:%d][%s] %U:%d->%U:%d", tc->c_thread_index,
1050                   tc->c_s_index, "T", format_ip4_address, &tc->c_lcl_ip4,
1051                   clib_net_to_host_u16 (tc->c_lcl_port), format_ip4_address,
1052                   &tc->c_rmt_ip4, clib_net_to_host_u16 (tc->c_rmt_port));
1053     }
1054   else
1055     {
1056       s = format (s, "[%d:%d][%s] %U:%d->%U:%d", tc->c_thread_index,
1057                   tc->c_s_index, "T", format_ip6_address, &tc->c_lcl_ip6,
1058                   clib_net_to_host_u16 (tc->c_lcl_port), format_ip6_address,
1059                   &tc->c_rmt_ip6, clib_net_to_host_u16 (tc->c_rmt_port));
1060     }
1061
1062   return s;
1063 }
1064
1065 u8 *
1066 format_tcp_connection (u8 * s, va_list * args)
1067 {
1068   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1069   u32 verbose = va_arg (*args, u32);
1070
1071   if (!tc)
1072     return s;
1073   s = format (s, "%-50U", format_tcp_connection_id, tc);
1074   if (verbose)
1075     {
1076       s = format (s, "%-15U", format_tcp_state, tc->state);
1077       if (verbose > 1)
1078         s = format (s, "\n%U", format_tcp_vars, tc);
1079     }
1080
1081   return s;
1082 }
1083
1084 static u8 *
1085 format_tcp_session (u8 * s, va_list * args)
1086 {
1087   u32 tci = va_arg (*args, u32);
1088   u32 thread_index = va_arg (*args, u32);
1089   u32 verbose = va_arg (*args, u32);
1090   tcp_connection_t *tc;
1091
1092   tc = tcp_connection_get (tci, thread_index);
1093   if (tc)
1094     s = format (s, "%U", format_tcp_connection, tc, verbose);
1095   else
1096     s = format (s, "empty\n");
1097   return s;
1098 }
1099
1100 static u8 *
1101 format_tcp_listener_session (u8 * s, va_list * args)
1102 {
1103   u32 tci = va_arg (*args, u32);
1104   u32 __clib_unused thread_index = va_arg (*args, u32);
1105   u32 verbose = va_arg (*args, u32);
1106   tcp_connection_t *tc = tcp_listener_get (tci);
1107   s = format (s, "%-50U", format_tcp_connection_id, tc);
1108   if (verbose)
1109     s = format (s, "%-15U", format_tcp_state, tc->state);
1110   return s;
1111 }
1112
1113 static u8 *
1114 format_tcp_half_open_session (u8 * s, va_list * args)
1115 {
1116   u32 tci = va_arg (*args, u32);
1117   u32 __clib_unused thread_index = va_arg (*args, u32);
1118   tcp_connection_t *tc = tcp_half_open_connection_get (tci);
1119   return format (s, "%U", format_tcp_connection_id, tc);
1120 }
1121
1122 u8 *
1123 format_tcp_sacks (u8 * s, va_list * args)
1124 {
1125   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1126   sack_block_t *sacks = tc->snd_sacks;
1127   sack_block_t *block;
1128   int i, len = 0;
1129
1130   len = vec_len (sacks);
1131   for (i = 0; i < len - 1; i++)
1132     {
1133       block = &sacks[i];
1134       s = format (s, " start %u end %u\n", block->start - tc->irs,
1135                   block->end - tc->irs);
1136     }
1137   if (len)
1138     {
1139       block = &sacks[len - 1];
1140       s = format (s, " start %u end %u", block->start - tc->irs,
1141                   block->end - tc->irs);
1142     }
1143   return s;
1144 }
1145
1146 u8 *
1147 format_tcp_rcv_sacks (u8 * s, va_list * args)
1148 {
1149   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1150   sack_block_t *sacks = tc->rcv_opts.sacks;
1151   sack_block_t *block;
1152   int i, len = 0;
1153
1154   len = vec_len (sacks);
1155   for (i = 0; i < len - 1; i++)
1156     {
1157       block = &sacks[i];
1158       s = format (s, " start %u end %u\n", block->start - tc->iss,
1159                   block->end - tc->iss);
1160     }
1161   if (len)
1162     {
1163       block = &sacks[len - 1];
1164       s = format (s, " start %u end %u", block->start - tc->iss,
1165                   block->end - tc->iss);
1166     }
1167   return s;
1168 }
1169
1170 static u8 *
1171 format_tcp_sack_hole (u8 * s, va_list * args)
1172 {
1173   sack_scoreboard_hole_t *hole = va_arg (*args, sack_scoreboard_hole_t *);
1174   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1175   if (tc)
1176     s = format (s, "  [%u, %u]", hole->start - tc->iss, hole->end - tc->iss);
1177   else
1178     s = format (s, "  [%u, %u]", hole->start, hole->end);
1179   return s;
1180 }
1181
1182 u8 *
1183 format_tcp_scoreboard (u8 * s, va_list * args)
1184 {
1185   sack_scoreboard_t *sb = va_arg (*args, sack_scoreboard_t *);
1186   tcp_connection_t *tc = va_arg (*args, tcp_connection_t *);
1187   sack_scoreboard_hole_t *hole;
1188   u32 indent = format_get_indent (s);
1189
1190   s = format (s, "sacked %u last_sacked %u lost %u last_lost %u"
1191               " rxt_sacked %u\n",
1192               sb->sacked_bytes, sb->last_sacked_bytes, sb->lost_bytes,
1193               sb->last_lost_bytes, sb->rxt_sacked);
1194   s = format (s, "%Ulast_delivered %u high_sacked %u is_reneging %u\n",
1195               format_white_space, indent, sb->last_bytes_delivered,
1196               sb->high_sacked - tc->iss, sb->is_reneging);
1197   s = format (s, "%Ucur_rxt_hole %u high_rxt %u rescue_rxt %u",
1198               format_white_space, indent, sb->cur_rxt_hole,
1199               sb->high_rxt - tc->iss, sb->rescue_rxt - tc->iss);
1200
1201   hole = scoreboard_first_hole (sb);
1202   if (hole)
1203     s = format (s, "\n%Uhead %u tail %u %u holes:\n%U", format_white_space,
1204                 indent, sb->head, sb->tail, pool_elts (sb->holes),
1205                 format_white_space, indent);
1206
1207   while (hole)
1208     {
1209       s = format (s, "%U", format_tcp_sack_hole, hole, tc);
1210       hole = scoreboard_next_hole (sb, hole);
1211     }
1212
1213   return s;
1214 }
1215
1216 static transport_connection_t *
1217 tcp_session_get_transport (u32 conn_index, u32 thread_index)
1218 {
1219   tcp_connection_t *tc = tcp_connection_get (conn_index, thread_index);
1220   if (PREDICT_FALSE (!tc))
1221     return 0;
1222   return &tc->connection;
1223 }
1224
1225 static transport_connection_t *
1226 tcp_half_open_session_get_transport (u32 conn_index)
1227 {
1228   tcp_connection_t *tc = tcp_half_open_connection_get (conn_index);
1229   return &tc->connection;
1230 }
1231
1232 static u16
1233 tcp_session_cal_goal_size (tcp_connection_t * tc)
1234 {
1235   u16 goal_size = tc->snd_mss;
1236
1237   goal_size = TCP_MAX_GSO_SZ - tc->snd_mss % TCP_MAX_GSO_SZ;
1238   goal_size = clib_min (goal_size, tc->snd_wnd / 2);
1239
1240   return goal_size > tc->snd_mss ? goal_size : tc->snd_mss;
1241 }
1242
1243 always_inline u32
1244 tcp_round_snd_space (tcp_connection_t * tc, u32 snd_space)
1245 {
1246   if (PREDICT_FALSE (tc->snd_wnd < tc->snd_mss))
1247     {
1248       return tc->snd_wnd <= snd_space ? tc->snd_wnd : 0;
1249     }
1250
1251   /* If not snd_wnd constrained and we can't write at least a segment,
1252    * don't try at all */
1253   if (PREDICT_FALSE (snd_space < tc->snd_mss))
1254     return snd_space < tc->cwnd ? 0 : snd_space;
1255
1256   /* round down to mss multiple */
1257   return snd_space - (snd_space % tc->snd_mss);
1258 }
1259
1260 /**
1261  * Compute tx window session is allowed to fill.
1262  *
1263  * Takes into account available send space, snd_mss and the congestion
1264  * state of the connection. If possible, the value returned is a multiple
1265  * of snd_mss.
1266  *
1267  * @param tc tcp connection
1268  * @return number of bytes session is allowed to write
1269  */
1270 static inline u32
1271 tcp_snd_space_inline (tcp_connection_t * tc)
1272 {
1273   int snd_space;
1274
1275   if (PREDICT_FALSE (tcp_in_fastrecovery (tc)
1276                      || tc->state == TCP_STATE_CLOSED))
1277     return 0;
1278
1279   snd_space = tcp_available_output_snd_space (tc);
1280
1281   /* If we got dupacks or sacked bytes but we're not yet in recovery, try
1282    * to force the peer to send enough dupacks to start retransmitting as
1283    * per Limited Transmit (RFC3042)
1284    */
1285   if (PREDICT_FALSE (tc->rcv_dupacks != 0 || tc->sack_sb.sacked_bytes))
1286     {
1287       if (tc->limited_transmit != tc->snd_nxt
1288           && (seq_lt (tc->limited_transmit, tc->snd_nxt - 2 * tc->snd_mss)
1289               || seq_gt (tc->limited_transmit, tc->snd_nxt)))
1290         tc->limited_transmit = tc->snd_nxt;
1291
1292       ASSERT (seq_leq (tc->limited_transmit, tc->snd_nxt));
1293
1294       int snt_limited = tc->snd_nxt - tc->limited_transmit;
1295       snd_space = clib_max ((int) 2 * tc->snd_mss - snt_limited, 0);
1296     }
1297   return tcp_round_snd_space (tc, snd_space);
1298 }
1299
1300 u32
1301 tcp_snd_space (tcp_connection_t * tc)
1302 {
1303   return tcp_snd_space_inline (tc);
1304 }
1305
1306 static int
1307 tcp_session_send_params (transport_connection_t * trans_conn,
1308                          transport_send_params_t * sp)
1309 {
1310   tcp_connection_t *tc = (tcp_connection_t *) trans_conn;
1311
1312   /* Ensure snd_mss does accurately reflect the amount of data we can push
1313    * in a segment. This also makes sure that options are updated according to
1314    * the current state of the connection. */
1315   tcp_update_burst_snd_vars (tc);
1316
1317   if (PREDICT_FALSE (tc->cfg_flags & TCP_CFG_F_TSO))
1318     sp->snd_mss = tcp_session_cal_goal_size (tc);
1319   else
1320     sp->snd_mss = tc->snd_mss;
1321
1322   sp->snd_space = clib_min (tcp_snd_space_inline (tc),
1323                             tc->snd_wnd - (tc->snd_nxt - tc->snd_una));
1324
1325   ASSERT (seq_geq (tc->snd_nxt, tc->snd_una));
1326   /* This still works if fast retransmit is on */
1327   sp->tx_offset = tc->snd_nxt - tc->snd_una;
1328
1329   sp->flags = sp->snd_space ? 0 : TRANSPORT_SND_F_DESCHED;
1330
1331   return 0;
1332 }
1333
1334 static void
1335 tcp_timer_waitclose_handler (tcp_connection_t * tc)
1336 {
1337   tcp_worker_ctx_t *wrk = tcp_get_worker (tc->c_thread_index);
1338
1339   switch (tc->state)
1340     {
1341     case TCP_STATE_CLOSE_WAIT:
1342       tcp_connection_timers_reset (tc);
1343       /* App never returned with a close */
1344       if (!(tc->flags & TCP_CONN_FINPNDG))
1345         {
1346           tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1347           session_transport_closed_notify (&tc->connection);
1348           tcp_program_cleanup (wrk, tc);
1349           tcp_workerp_stats_inc (wrk, to_closewait, 1);
1350           break;
1351         }
1352
1353       /* Send FIN either way and switch to LAST_ACK. */
1354       tcp_cong_recovery_off (tc);
1355       /* Make sure we don't try to send unsent data */
1356       tc->snd_nxt = tc->snd_una;
1357       tcp_send_fin (tc);
1358       tcp_connection_set_state (tc, TCP_STATE_LAST_ACK);
1359       session_transport_closed_notify (&tc->connection);
1360
1361       /* Make sure we don't wait in LAST ACK forever */
1362       tcp_timer_set (tc, TCP_TIMER_WAITCLOSE, tcp_cfg.lastack_time);
1363       tcp_workerp_stats_inc (wrk, to_closewait2, 1);
1364
1365       /* Don't delete the connection yet */
1366       break;
1367     case TCP_STATE_FIN_WAIT_1:
1368       tcp_connection_timers_reset (tc);
1369       if (tc->flags & TCP_CONN_FINPNDG)
1370         {
1371           /* If FIN pending, we haven't sent everything, but we did try.
1372            * Notify session layer that transport is closed. */
1373           tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1374           tcp_send_reset (tc);
1375           tcp_program_cleanup (wrk, tc);
1376         }
1377       else
1378         {
1379           /* We've sent the fin but no progress. Close the connection and
1380            * to make sure everything is flushed, setup a cleanup timer */
1381           tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1382           tcp_program_cleanup (wrk, tc);
1383         }
1384       session_transport_closed_notify (&tc->connection);
1385       tcp_workerp_stats_inc (wrk, to_finwait1, 1);
1386       break;
1387     case TCP_STATE_LAST_ACK:
1388       tcp_connection_timers_reset (tc);
1389       tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1390       session_transport_closed_notify (&tc->connection);
1391       tcp_program_cleanup (wrk, tc);
1392       tcp_workerp_stats_inc (wrk, to_lastack, 1);
1393       break;
1394     case TCP_STATE_CLOSING:
1395       tcp_connection_timers_reset (tc);
1396       tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1397       session_transport_closed_notify (&tc->connection);
1398       tcp_program_cleanup (wrk, tc);
1399       tcp_workerp_stats_inc (wrk, to_closing, 1);
1400       break;
1401     case TCP_STATE_FIN_WAIT_2:
1402       tcp_send_reset (tc);
1403       tcp_connection_timers_reset (tc);
1404       tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1405       session_transport_closed_notify (&tc->connection);
1406       tcp_program_cleanup (wrk, tc);
1407       tcp_workerp_stats_inc (wrk, to_finwait2, 1);
1408       break;
1409     case TCP_STATE_TIME_WAIT:
1410       tcp_connection_set_state (tc, TCP_STATE_CLOSED);
1411       tcp_program_cleanup (wrk, tc);
1412       break;
1413     default:
1414       clib_warning ("waitclose in state: %U", format_tcp_state, tc->state);
1415       break;
1416     }
1417 }
1418
1419 /* *INDENT-OFF* */
1420 static timer_expiration_handler *timer_expiration_handlers[TCP_N_TIMERS] =
1421 {
1422     tcp_timer_retransmit_handler,
1423     tcp_timer_delack_handler,
1424     tcp_timer_persist_handler,
1425     tcp_timer_waitclose_handler,
1426     tcp_timer_retransmit_syn_handler,
1427 };
1428 /* *INDENT-ON* */
1429
1430 static void
1431 tcp_dispatch_pending_timers (tcp_worker_ctx_t * wrk)
1432 {
1433   u32 n_timers, connection_index, timer_id, thread_index, timer_handle;
1434   tcp_connection_t *tc;
1435   int i;
1436
1437   if (!(n_timers = clib_fifo_elts (wrk->pending_timers)))
1438     return;
1439
1440   thread_index = wrk->vm->thread_index;
1441   for (i = 0; i < clib_min (n_timers, wrk->max_timers_per_loop); i++)
1442     {
1443       clib_fifo_sub1 (wrk->pending_timers, timer_handle);
1444       connection_index = timer_handle & 0x0FFFFFFF;
1445       timer_id = timer_handle >> 28;
1446
1447       if (PREDICT_TRUE (timer_id != TCP_TIMER_RETRANSMIT_SYN))
1448         tc = tcp_connection_get (connection_index, thread_index);
1449       else
1450         tc = tcp_half_open_connection_get (connection_index);
1451
1452       if (PREDICT_FALSE (!tc))
1453         continue;
1454
1455       /* Skip timer if it was rearmed while pending dispatch */
1456       if (PREDICT_FALSE (tc->timers[timer_id] != TCP_TIMER_HANDLE_INVALID))
1457         continue;
1458
1459       (*timer_expiration_handlers[timer_id]) (tc);
1460     }
1461
1462   if (thread_index == 0 && clib_fifo_elts (wrk->pending_timers))
1463     session_queue_run_on_main_thread (wrk->vm);
1464 }
1465
1466 static void
1467 tcp_handle_cleanups (tcp_worker_ctx_t * wrk, clib_time_type_t now)
1468 {
1469   u32 thread_index = wrk->vm->thread_index;
1470   tcp_cleanup_req_t *req;
1471   tcp_connection_t *tc;
1472
1473   while (clib_fifo_elts (wrk->pending_cleanups))
1474     {
1475       req = clib_fifo_head (wrk->pending_cleanups);
1476       if (req->free_time > now)
1477         break;
1478       clib_fifo_sub2 (wrk->pending_cleanups, req);
1479       tc = tcp_connection_get (req->connection_index, thread_index);
1480       if (PREDICT_FALSE (!tc))
1481         continue;
1482       session_transport_delete_notify (&tc->connection);
1483       tcp_connection_cleanup (tc);
1484     }
1485 }
1486
1487 static void
1488 tcp_update_time (f64 now, u8 thread_index)
1489 {
1490   tcp_worker_ctx_t *wrk = tcp_get_worker (thread_index);
1491
1492   tcp_set_time_now (wrk);
1493   tcp_handle_cleanups (wrk, now);
1494   tw_timer_expire_timers_16t_2w_512sl (&wrk->timer_wheel, now);
1495   tcp_dispatch_pending_timers (wrk);
1496 }
1497
1498 static void
1499 tcp_session_flush_data (transport_connection_t * tconn)
1500 {
1501   tcp_connection_t *tc = (tcp_connection_t *) tconn;
1502   if (tc->flags & TCP_CONN_PSH_PENDING)
1503     return;
1504   tc->flags |= TCP_CONN_PSH_PENDING;
1505   tc->psh_seq = tc->snd_una + transport_max_tx_dequeue (tconn) - 1;
1506 }
1507
1508 /* *INDENT-OFF* */
1509 const static transport_proto_vft_t tcp_proto = {
1510   .enable = vnet_tcp_enable_disable,
1511   .start_listen = tcp_session_bind,
1512   .stop_listen = tcp_session_unbind,
1513   .push_header = tcp_session_push_header,
1514   .get_connection = tcp_session_get_transport,
1515   .get_listener = tcp_session_get_listener,
1516   .get_half_open = tcp_half_open_session_get_transport,
1517   .connect = tcp_session_open,
1518   .close = tcp_session_close,
1519   .cleanup = tcp_session_cleanup,
1520   .reset = tcp_session_reset,
1521   .send_params = tcp_session_send_params,
1522   .update_time = tcp_update_time,
1523   .flush_data = tcp_session_flush_data,
1524   .custom_tx = tcp_session_custom_tx,
1525   .format_connection = format_tcp_session,
1526   .format_listener = format_tcp_listener_session,
1527   .format_half_open = format_tcp_half_open_session,
1528   .transport_options = {
1529     .name = "tcp",
1530     .short_name = "T",
1531     .tx_type = TRANSPORT_TX_PEEK,
1532     .service_type = TRANSPORT_SERVICE_VC,
1533   },
1534 };
1535 /* *INDENT-ON* */
1536
1537 void
1538 tcp_connection_tx_pacer_update (tcp_connection_t * tc)
1539 {
1540   if (!transport_connection_is_tx_paced (&tc->connection))
1541     return;
1542
1543   f64 srtt = clib_min ((f64) tc->srtt * TCP_TICK, tc->mrtt_us);
1544
1545   transport_connection_tx_pacer_update (&tc->connection,
1546                                         tcp_cc_get_pacing_rate (tc),
1547                                         srtt * CLIB_US_TIME_FREQ);
1548 }
1549
1550 void
1551 tcp_connection_tx_pacer_reset (tcp_connection_t * tc, u32 window,
1552                                u32 start_bucket)
1553 {
1554   f64 srtt = clib_min ((f64) tc->srtt * TCP_TICK, tc->mrtt_us);
1555   transport_connection_tx_pacer_reset (&tc->connection,
1556                                        tcp_cc_get_pacing_rate (tc),
1557                                        start_bucket,
1558                                        srtt * CLIB_US_TIME_FREQ);
1559 }
1560
1561 void
1562 tcp_reschedule (tcp_connection_t * tc)
1563 {
1564   if (tcp_in_cong_recovery (tc) || tcp_snd_space_inline (tc))
1565     transport_connection_reschedule (&tc->connection);
1566 }
1567
1568 static void
1569 tcp_expired_timers_dispatch (u32 * expired_timers)
1570 {
1571   u32 thread_index = vlib_get_thread_index (), n_left, max_per_loop;
1572   u32 connection_index, timer_id, n_expired, max_loops;
1573   tcp_worker_ctx_t *wrk;
1574   tcp_connection_t *tc;
1575   int i;
1576
1577   wrk = tcp_get_worker (thread_index);
1578   n_expired = vec_len (expired_timers);
1579   tcp_workerp_stats_inc (wrk, timer_expirations, n_expired);
1580   n_left = clib_fifo_elts (wrk->pending_timers);
1581
1582   /*
1583    * Invalidate all timer handles before dispatching. This avoids dangling
1584    * index references to timer wheel pool entries that have been freed.
1585    */
1586   for (i = 0; i < n_expired; i++)
1587     {
1588       connection_index = expired_timers[i] & 0x0FFFFFFF;
1589       timer_id = expired_timers[i] >> 28;
1590
1591       if (timer_id != TCP_TIMER_RETRANSMIT_SYN)
1592         tc = tcp_connection_get (connection_index, thread_index);
1593       else
1594         tc = tcp_half_open_connection_get (connection_index);
1595
1596       TCP_EVT (TCP_EVT_TIMER_POP, connection_index, timer_id);
1597
1598       tc->timers[timer_id] = TCP_TIMER_HANDLE_INVALID;
1599     }
1600
1601   clib_fifo_add (wrk->pending_timers, expired_timers, n_expired);
1602
1603   max_loops = clib_max (1, 0.5 * TCP_TIMER_TICK * wrk->vm->loops_per_second);
1604   max_per_loop = clib_max ((n_left + n_expired) / max_loops, 10);
1605   max_per_loop = clib_min (max_per_loop, VLIB_FRAME_SIZE);
1606   wrk->max_timers_per_loop = clib_max (n_left ? wrk->max_timers_per_loop : 0,
1607                                        max_per_loop);
1608
1609   if (thread_index == 0)
1610     session_queue_run_on_main_thread (wrk->vm);
1611 }
1612
1613 static void
1614 tcp_initialize_timer_wheels (tcp_main_t * tm)
1615 {
1616   tw_timer_wheel_16t_2w_512sl_t *tw;
1617   /* *INDENT-OFF* */
1618   foreach_vlib_main (({
1619     tw = &tm->wrk_ctx[ii].timer_wheel;
1620     tw_timer_wheel_init_16t_2w_512sl (tw, tcp_expired_timers_dispatch,
1621                                       TCP_TIMER_TICK, ~0);
1622     tw->last_run_time = vlib_time_now (this_vlib_main);
1623   }));
1624   /* *INDENT-ON* */
1625 }
1626
1627 static void
1628 tcp_initialize_iss_seed (tcp_main_t * tm)
1629 {
1630   u32 default_seed = random_default_seed ();
1631   u64 time_now = clib_cpu_time_now ();
1632
1633   tm->iss_seed.first = (u64) random_u32 (&default_seed) << 32;
1634   tm->iss_seed.second = random_u64 (&time_now);
1635 }
1636
1637 static clib_error_t *
1638 tcp_main_enable (vlib_main_t * vm)
1639 {
1640   vlib_thread_main_t *vtm = vlib_get_thread_main ();
1641   u32 num_threads, n_workers, prealloc_conn_per_wrk;
1642   tcp_connection_t *tc __attribute__ ((unused));
1643   tcp_main_t *tm = vnet_get_tcp_main ();
1644   tcp_worker_ctx_t *wrk;
1645   clib_error_t *error = 0;
1646   int thread;
1647
1648   if ((error = vlib_call_init_function (vm, ip_main_init)))
1649     return error;
1650   if ((error = vlib_call_init_function (vm, ip4_lookup_init)))
1651     return error;
1652   if ((error = vlib_call_init_function (vm, ip6_lookup_init)))
1653     return error;
1654
1655   /*
1656    * Registrations
1657    */
1658
1659   ip4_register_protocol (IP_PROTOCOL_TCP, tcp4_input_node.index);
1660   ip6_register_protocol (IP_PROTOCOL_TCP, tcp6_input_node.index);
1661
1662   /*
1663    * Initialize data structures
1664    */
1665
1666   num_threads = 1 /* main thread */  + vtm->n_threads;
1667   vec_validate (tm->wrk_ctx, num_threads - 1);
1668   n_workers = num_threads == 1 ? 1 : vtm->n_threads;
1669   prealloc_conn_per_wrk = tcp_cfg.preallocated_connections / n_workers;
1670
1671   wrk = &tm->wrk_ctx[0];
1672   wrk->tco_next_node[0] = vlib_node_get_next (vm, session_queue_node.index,
1673                                               tcp4_output_node.index);
1674   wrk->tco_next_node[1] = vlib_node_get_next (vm, session_queue_node.index,
1675                                               tcp6_output_node.index);
1676
1677   for (thread = 0; thread < num_threads; thread++)
1678     {
1679       wrk = &tm->wrk_ctx[thread];
1680
1681       vec_validate (wrk->pending_deq_acked, 255);
1682       vec_validate (wrk->pending_disconnects, 255);
1683       vec_validate (wrk->pending_resets, 255);
1684       vec_reset_length (wrk->pending_deq_acked);
1685       vec_reset_length (wrk->pending_disconnects);
1686       vec_reset_length (wrk->pending_resets);
1687       wrk->vm = vlib_mains[thread];
1688       wrk->max_timers_per_loop = 10;
1689
1690       if (thread > 0)
1691         {
1692           wrk->tco_next_node[0] = tm->wrk_ctx[0].tco_next_node[0];
1693           wrk->tco_next_node[1] = tm->wrk_ctx[0].tco_next_node[1];
1694         }
1695
1696       /*
1697        * Preallocate connections. Assume that thread 0 won't
1698        * use preallocated threads when running multi-core
1699        */
1700       if ((thread > 0 || num_threads == 1) && prealloc_conn_per_wrk)
1701         pool_init_fixed (wrk->connections, prealloc_conn_per_wrk);
1702     }
1703
1704   /*
1705    * Use a preallocated half-open connection pool?
1706    */
1707   if (tcp_cfg.preallocated_half_open_connections)
1708     pool_init_fixed (tm->half_open_connections,
1709                      tcp_cfg.preallocated_half_open_connections);
1710
1711   /* Initialize clocks per tick for TCP timestamp. Used to compute
1712    * monotonically increasing timestamps. */
1713   tm->tstamp_ticks_per_clock = vm->clib_time.seconds_per_clock
1714     / TCP_TSTAMP_RESOLUTION;
1715
1716   if (num_threads > 1)
1717     {
1718       clib_spinlock_init (&tm->half_open_lock);
1719     }
1720
1721   tcp_initialize_timer_wheels (tm);
1722   tcp_initialize_iss_seed (tm);
1723
1724   tm->bytes_per_buffer = vlib_buffer_get_default_data_size (vm);
1725   tm->cc_last_type = TCP_CC_LAST;
1726
1727   tm->ipl_next_node[0] = vlib_node_get_next (vm, session_queue_node.index,
1728                                              ip4_lookup_node.index);
1729   tm->ipl_next_node[1] = vlib_node_get_next (vm, session_queue_node.index,
1730                                              ip6_lookup_node.index);
1731   return error;
1732 }
1733
1734 clib_error_t *
1735 vnet_tcp_enable_disable (vlib_main_t * vm, u8 is_en)
1736 {
1737   if (is_en)
1738     {
1739       if (tcp_main.is_enabled)
1740         return 0;
1741
1742       return tcp_main_enable (vm);
1743     }
1744   else
1745     {
1746       tcp_main.is_enabled = 0;
1747     }
1748
1749   return 0;
1750 }
1751
1752 void
1753 tcp_punt_unknown (vlib_main_t * vm, u8 is_ip4, u8 is_add)
1754 {
1755   tcp_main_t *tm = &tcp_main;
1756   if (is_ip4)
1757     tm->punt_unknown4 = is_add;
1758   else
1759     tm->punt_unknown6 = is_add;
1760 }
1761
1762 /**
1763  * Initialize default values for tcp parameters
1764  */
1765 static void
1766 tcp_configuration_init (void)
1767 {
1768   /* Initial wnd for SYN. Fifos are not allocated at that point so use some
1769    * predefined value. For SYN-ACK we still want the scale to be computed in
1770    * the same way */
1771   tcp_cfg.max_rx_fifo = 32 << 20;
1772   tcp_cfg.min_rx_fifo = 4 << 10;
1773
1774   tcp_cfg.default_mtu = 1500;
1775   tcp_cfg.initial_cwnd_multiplier = 0;
1776   tcp_cfg.enable_tx_pacing = 1;
1777   tcp_cfg.allow_tso = 0;
1778   tcp_cfg.csum_offload = 1;
1779   tcp_cfg.cc_algo = TCP_CC_NEWRENO;
1780   tcp_cfg.rwnd_min_update_ack = 1;
1781
1782   /* Time constants defined as timer tick (100ms) multiples */
1783   tcp_cfg.delack_time = 1;      /* 0.1s */
1784   tcp_cfg.closewait_time = 20;  /* 2s */
1785   tcp_cfg.timewait_time = 100;  /* 10s */
1786   tcp_cfg.finwait1_time = 600;  /* 60s */
1787   tcp_cfg.lastack_time = 300;   /* 30s */
1788   tcp_cfg.finwait2_time = 300;  /* 30s */
1789   tcp_cfg.closing_time = 300;   /* 30s */
1790   tcp_cfg.cleanup_time = 0.1;   /* 100ms */
1791 }
1792
1793 static clib_error_t *
1794 tcp_init (vlib_main_t * vm)
1795 {
1796   tcp_main_t *tm = vnet_get_tcp_main ();
1797   ip_main_t *im = &ip_main;
1798   ip_protocol_info_t *pi;
1799
1800   /* Session layer, and by implication tcp, are disabled by default */
1801   tm->is_enabled = 0;
1802
1803   /* Register with IP for header parsing */
1804   pi = ip_get_protocol_info (im, IP_PROTOCOL_TCP);
1805   if (pi == 0)
1806     return clib_error_return (0, "TCP protocol info AWOL");
1807   pi->format_header = format_tcp_header;
1808   pi->unformat_pg_edit = unformat_pg_tcp_header;
1809
1810   /* Register as transport with session layer */
1811   transport_register_protocol (TRANSPORT_PROTO_TCP, &tcp_proto,
1812                                FIB_PROTOCOL_IP4, tcp4_output_node.index);
1813   transport_register_protocol (TRANSPORT_PROTO_TCP, &tcp_proto,
1814                                FIB_PROTOCOL_IP6, tcp6_output_node.index);
1815
1816   tcp_api_reference ();
1817   tcp_configuration_init ();
1818
1819   tm->cc_algo_by_name = hash_create_string (0, sizeof (uword));
1820
1821   return 0;
1822 }
1823
1824 VLIB_INIT_FUNCTION (tcp_init);
1825
1826 uword
1827 unformat_tcp_cc_algo (unformat_input_t * input, va_list * va)
1828 {
1829   tcp_cc_algorithm_type_e *result = va_arg (*va, tcp_cc_algorithm_type_e *);
1830   tcp_main_t *tm = &tcp_main;
1831   char *cc_algo_name;
1832   u8 found = 0;
1833   uword *p;
1834
1835   if (unformat (input, "%s", &cc_algo_name)
1836       && ((p = hash_get_mem (tm->cc_algo_by_name, cc_algo_name))))
1837     {
1838       *result = *p;
1839       found = 1;
1840     }
1841
1842   vec_free (cc_algo_name);
1843   return found;
1844 }
1845
1846 uword
1847 unformat_tcp_cc_algo_cfg (unformat_input_t * input, va_list * va)
1848 {
1849   tcp_main_t *tm = vnet_get_tcp_main ();
1850   tcp_cc_algorithm_t *cc_alg;
1851   unformat_input_t sub_input;
1852   int found = 0;
1853
1854   vec_foreach (cc_alg, tm->cc_algos)
1855   {
1856     if (!unformat (input, cc_alg->name))
1857       continue;
1858
1859     if (cc_alg->unformat_cfg
1860         && unformat (input, "%U", unformat_vlib_cli_sub_input, &sub_input))
1861       {
1862         if (cc_alg->unformat_cfg (&sub_input))
1863           found = 1;
1864       }
1865   }
1866   return found;
1867 }
1868
1869 static clib_error_t *
1870 tcp_config_fn (vlib_main_t * vm, unformat_input_t * input)
1871 {
1872   u32 cwnd_multiplier, tmp_time;
1873   uword memory_size;
1874
1875   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1876     {
1877       if (unformat (input, "preallocated-connections %d",
1878                     &tcp_cfg.preallocated_connections))
1879         ;
1880       else if (unformat (input, "preallocated-half-open-connections %d",
1881                          &tcp_cfg.preallocated_half_open_connections))
1882         ;
1883       else if (unformat (input, "buffer-fail-fraction %f",
1884                          &tcp_cfg.buffer_fail_fraction))
1885         ;
1886       else if (unformat (input, "max-rx-fifo %U", unformat_memory_size,
1887                          &memory_size))
1888         {
1889           if (memory_size >= 0x100000000)
1890             {
1891               return clib_error_return
1892                 (0, "max-rx-fifo %llu (0x%llx) too large", memory_size,
1893                  memory_size);
1894             }
1895           tcp_cfg.max_rx_fifo = memory_size;
1896         }
1897       else if (unformat (input, "min-rx-fifo %U", unformat_memory_size,
1898                          &memory_size))
1899         {
1900           if (memory_size >= 0x100000000)
1901             {
1902               return clib_error_return
1903                 (0, "min-rx-fifo %llu (0x%llx) too large", memory_size,
1904                  memory_size);
1905             }
1906           tcp_cfg.min_rx_fifo = memory_size;
1907         }
1908       else if (unformat (input, "mtu %u", &tcp_cfg.default_mtu))
1909         ;
1910       else if (unformat (input, "rwnd-min-update-ack %d",
1911                          &tcp_cfg.rwnd_min_update_ack))
1912         ;
1913       else if (unformat (input, "initial-cwnd-multiplier %u",
1914                          &cwnd_multiplier))
1915         tcp_cfg.initial_cwnd_multiplier = cwnd_multiplier;
1916       else if (unformat (input, "no-tx-pacing"))
1917         tcp_cfg.enable_tx_pacing = 0;
1918       else if (unformat (input, "tso"))
1919         tcp_cfg.allow_tso = 1;
1920       else if (unformat (input, "no-csum-offload"))
1921         tcp_cfg.csum_offload = 0;
1922       else if (unformat (input, "cc-algo %U", unformat_tcp_cc_algo,
1923                          &tcp_cfg.cc_algo))
1924         ;
1925       else if (unformat (input, "%U", unformat_tcp_cc_algo_cfg))
1926         ;
1927       else if (unformat (input, "closewait-time %u", &tmp_time))
1928         tcp_cfg.closewait_time = tmp_time / TCP_TIMER_TICK;
1929       else if (unformat (input, "timewait-time %u", &tmp_time))
1930         tcp_cfg.timewait_time = tmp_time / TCP_TIMER_TICK;
1931       else if (unformat (input, "finwait1-time %u", &tmp_time))
1932         tcp_cfg.finwait1_time = tmp_time / TCP_TIMER_TICK;
1933       else if (unformat (input, "finwait2-time %u", &tmp_time))
1934         tcp_cfg.finwait2_time = tmp_time / TCP_TIMER_TICK;
1935       else if (unformat (input, "lastack-time %u", &tmp_time))
1936         tcp_cfg.lastack_time = tmp_time / TCP_TIMER_TICK;
1937       else if (unformat (input, "closing-time %u", &tmp_time))
1938         tcp_cfg.closing_time = tmp_time / TCP_TIMER_TICK;
1939       else if (unformat (input, "cleanup-time %u", &tmp_time))
1940         tcp_cfg.cleanup_time = tmp_time / 1000.0;
1941       else
1942         return clib_error_return (0, "unknown input `%U'",
1943                                   format_unformat_error, input);
1944     }
1945   return 0;
1946 }
1947
1948 VLIB_CONFIG_FUNCTION (tcp_config_fn, "tcp");
1949
1950
1951 /**
1952  * \brief Configure an ipv4 source address range
1953  * @param vm vlib_main_t pointer
1954  * @param start first ipv4 address in the source address range
1955  * @param end last ipv4 address in the source address range
1956  * @param table_id VRF / table ID, 0 for the default FIB
1957  * @return 0 if all OK, else an error indication from api_errno.h
1958  */
1959
1960 int
1961 tcp_configure_v4_source_address_range (vlib_main_t * vm,
1962                                        ip4_address_t * start,
1963                                        ip4_address_t * end, u32 table_id)
1964 {
1965   u32 start_host_byte_order, end_host_byte_order;
1966   fib_prefix_t prefix;
1967   fib_node_index_t fei;
1968   u32 fib_index = 0;
1969   u32 sw_if_index;
1970   int rv;
1971
1972   clib_memset (&prefix, 0, sizeof (prefix));
1973
1974   fib_index = fib_table_find (FIB_PROTOCOL_IP4, table_id);
1975
1976   if (fib_index == ~0)
1977     return VNET_API_ERROR_NO_SUCH_FIB;
1978
1979   start_host_byte_order = clib_net_to_host_u32 (start->as_u32);
1980   end_host_byte_order = clib_net_to_host_u32 (end->as_u32);
1981
1982   /* sanity check for reversed args or some such */
1983   if ((end_host_byte_order - start_host_byte_order) > (10 << 10))
1984     return VNET_API_ERROR_INVALID_ARGUMENT;
1985
1986   /* Lookup the last address, to identify the interface involved */
1987   prefix.fp_len = 32;
1988   prefix.fp_proto = FIB_PROTOCOL_IP4;
1989   memcpy (&prefix.fp_addr.ip4, end, sizeof (ip4_address_t));
1990
1991   fei = fib_table_lookup (fib_index, &prefix);
1992
1993   /* Couldn't find route to destination. Bail out. */
1994   if (fei == FIB_NODE_INDEX_INVALID)
1995     return VNET_API_ERROR_NEXT_HOP_NOT_IN_FIB;
1996
1997   sw_if_index = fib_entry_get_resolving_interface (fei);
1998
1999   /* Configure proxy arp across the range */
2000   rv = ip4_neighbor_proxy_add (fib_index, start, end);
2001
2002   if (rv)
2003     return rv;
2004
2005   rv = ip4_neighbor_proxy_enable (sw_if_index);
2006
2007   if (rv)
2008     return rv;
2009
2010   do
2011     {
2012       dpo_id_t dpo = DPO_INVALID;
2013
2014       vec_add1 (tcp_cfg.ip4_src_addrs, start[0]);
2015
2016       /* Add local adjacencies for the range */
2017
2018       receive_dpo_add_or_lock (DPO_PROTO_IP4, ~0 /* sw_if_index */ ,
2019                                NULL, &dpo);
2020       prefix.fp_len = 32;
2021       prefix.fp_proto = FIB_PROTOCOL_IP4;
2022       prefix.fp_addr.ip4.as_u32 = start->as_u32;
2023
2024       fib_table_entry_special_dpo_update (fib_index,
2025                                           &prefix,
2026                                           FIB_SOURCE_API,
2027                                           FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
2028       dpo_reset (&dpo);
2029
2030       start_host_byte_order++;
2031       start->as_u32 = clib_host_to_net_u32 (start_host_byte_order);
2032     }
2033   while (start_host_byte_order <= end_host_byte_order);
2034
2035   return 0;
2036 }
2037
2038 /**
2039  * \brief Configure an ipv6 source address range
2040  * @param vm vlib_main_t pointer
2041  * @param start first ipv6 address in the source address range
2042  * @param end last ipv6 address in the source address range
2043  * @param table_id VRF / table ID, 0 for the default FIB
2044  * @return 0 if all OK, else an error indication from api_errno.h
2045  */
2046
2047 int
2048 tcp_configure_v6_source_address_range (vlib_main_t * vm,
2049                                        ip6_address_t * start,
2050                                        ip6_address_t * end, u32 table_id)
2051 {
2052   fib_prefix_t prefix;
2053   u32 fib_index = 0;
2054   fib_node_index_t fei;
2055   u32 sw_if_index;
2056
2057   clib_memset (&prefix, 0, sizeof (prefix));
2058
2059   fib_index = fib_table_find (FIB_PROTOCOL_IP6, table_id);
2060
2061   if (fib_index == ~0)
2062     return VNET_API_ERROR_NO_SUCH_FIB;
2063
2064   while (1)
2065     {
2066       int i;
2067       ip6_address_t tmp;
2068       dpo_id_t dpo = DPO_INVALID;
2069
2070       /* Remember this address */
2071       vec_add1 (tcp_cfg.ip6_src_addrs, start[0]);
2072
2073       /* Lookup the prefix, to identify the interface involved */
2074       prefix.fp_len = 128;
2075       prefix.fp_proto = FIB_PROTOCOL_IP6;
2076       memcpy (&prefix.fp_addr.ip6, start, sizeof (ip6_address_t));
2077
2078       fei = fib_table_lookup (fib_index, &prefix);
2079
2080       /* Couldn't find route to destination. Bail out. */
2081       if (fei == FIB_NODE_INDEX_INVALID)
2082         return VNET_API_ERROR_NEXT_HOP_NOT_IN_FIB;
2083
2084       sw_if_index = fib_entry_get_resolving_interface (fei);
2085
2086       if (sw_if_index == (u32) ~ 0)
2087         return VNET_API_ERROR_NO_MATCHING_INTERFACE;
2088
2089       /* Add a proxy neighbor discovery entry for this address */
2090       ip6_neighbor_proxy_add (sw_if_index, start);
2091
2092       /* Add a receive adjacency for this address */
2093       receive_dpo_add_or_lock (DPO_PROTO_IP6, ~0 /* sw_if_index */ ,
2094                                NULL, &dpo);
2095
2096       fib_table_entry_special_dpo_update (fib_index,
2097                                           &prefix,
2098                                           FIB_SOURCE_API,
2099                                           FIB_ENTRY_FLAG_EXCLUSIVE, &dpo);
2100       dpo_reset (&dpo);
2101
2102       /* Done with the entire range? */
2103       if (!memcmp (start, end, sizeof (start[0])))
2104         break;
2105
2106       /* Increment the address. DGMS. */
2107       tmp = start[0];
2108       for (i = 15; i >= 0; i--)
2109         {
2110           tmp.as_u8[i] += 1;
2111           if (tmp.as_u8[i] != 0)
2112             break;
2113         }
2114       start[0] = tmp;
2115     }
2116   return 0;
2117 }
2118
2119 static clib_error_t *
2120 tcp_src_address_fn (vlib_main_t * vm,
2121                     unformat_input_t * input, vlib_cli_command_t * cmd_arg)
2122 {
2123   ip4_address_t v4start, v4end;
2124   ip6_address_t v6start, v6end;
2125   u32 table_id = 0;
2126   int v4set = 0;
2127   int v6set = 0;
2128   int rv;
2129
2130   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2131     {
2132       if (unformat (input, "%U - %U", unformat_ip4_address, &v4start,
2133                     unformat_ip4_address, &v4end))
2134         v4set = 1;
2135       else if (unformat (input, "%U", unformat_ip4_address, &v4start))
2136         {
2137           memcpy (&v4end, &v4start, sizeof (v4start));
2138           v4set = 1;
2139         }
2140       else if (unformat (input, "%U - %U", unformat_ip6_address, &v6start,
2141                          unformat_ip6_address, &v6end))
2142         v6set = 1;
2143       else if (unformat (input, "%U", unformat_ip6_address, &v6start))
2144         {
2145           memcpy (&v6end, &v6start, sizeof (v6start));
2146           v6set = 1;
2147         }
2148       else if (unformat (input, "fib-table %d", &table_id))
2149         ;
2150       else
2151         break;
2152     }
2153
2154   if (!v4set && !v6set)
2155     return clib_error_return (0, "at least one v4 or v6 address required");
2156
2157   if (v4set)
2158     {
2159       rv = tcp_configure_v4_source_address_range (vm, &v4start, &v4end,
2160                                                   table_id);
2161       switch (rv)
2162         {
2163         case 0:
2164           break;
2165
2166         case VNET_API_ERROR_NO_SUCH_FIB:
2167           return clib_error_return (0, "Invalid table-id %d", table_id);
2168
2169         case VNET_API_ERROR_INVALID_ARGUMENT:
2170           return clib_error_return (0, "Invalid address range %U - %U",
2171                                     format_ip4_address, &v4start,
2172                                     format_ip4_address, &v4end);
2173         default:
2174           return clib_error_return (0, "error %d", rv);
2175           break;
2176         }
2177     }
2178   if (v6set)
2179     {
2180       rv = tcp_configure_v6_source_address_range (vm, &v6start, &v6end,
2181                                                   table_id);
2182       switch (rv)
2183         {
2184         case 0:
2185           break;
2186
2187         case VNET_API_ERROR_NO_SUCH_FIB:
2188           return clib_error_return (0, "Invalid table-id %d", table_id);
2189
2190         default:
2191           return clib_error_return (0, "error %d", rv);
2192           break;
2193         }
2194     }
2195   return 0;
2196 }
2197
2198 /* *INDENT-OFF* */
2199 VLIB_CLI_COMMAND (tcp_src_address_command, static) =
2200 {
2201   .path = "tcp src-address",
2202   .short_help = "tcp src-address <ip-addr> [- <ip-addr>] add src address range",
2203   .function = tcp_src_address_fn,
2204 };
2205 /* *INDENT-ON* */
2206
2207 static u8 *
2208 tcp_scoreboard_dump_trace (u8 * s, sack_scoreboard_t * sb)
2209 {
2210 #if TCP_SCOREBOARD_TRACE
2211
2212   scoreboard_trace_elt_t *block;
2213   int i = 0;
2214
2215   if (!sb->trace)
2216     return s;
2217
2218   s = format (s, "scoreboard trace:");
2219   vec_foreach (block, sb->trace)
2220   {
2221     s = format (s, "{%u, %u, %u, %u, %u}, ", block->start, block->end,
2222                 block->ack, block->snd_una_max, block->group);
2223     if ((++i % 3) == 0)
2224       s = format (s, "\n");
2225   }
2226   return s;
2227 #else
2228   return 0;
2229 #endif
2230 }
2231
2232 static clib_error_t *
2233 tcp_show_scoreboard_trace_fn (vlib_main_t * vm, unformat_input_t * input,
2234                               vlib_cli_command_t * cmd_arg)
2235 {
2236   transport_connection_t *tconn = 0;
2237   tcp_connection_t *tc;
2238   u8 *s = 0;
2239   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2240     {
2241       if (unformat (input, "%U", unformat_transport_connection, &tconn,
2242                     TRANSPORT_PROTO_TCP))
2243         ;
2244       else
2245         return clib_error_return (0, "unknown input `%U'",
2246                                   format_unformat_error, input);
2247     }
2248
2249   if (!TCP_SCOREBOARD_TRACE)
2250     {
2251       vlib_cli_output (vm, "scoreboard tracing not enabled");
2252       return 0;
2253     }
2254
2255   tc = tcp_get_connection_from_transport (tconn);
2256   s = tcp_scoreboard_dump_trace (s, &tc->sack_sb);
2257   vlib_cli_output (vm, "%v", s);
2258   return 0;
2259 }
2260
2261 /* *INDENT-OFF* */
2262 VLIB_CLI_COMMAND (tcp_show_scoreboard_trace_command, static) =
2263 {
2264   .path = "show tcp scoreboard trace",
2265   .short_help = "show tcp scoreboard trace <connection>",
2266   .function = tcp_show_scoreboard_trace_fn,
2267 };
2268 /* *INDENT-ON* */
2269
2270 u8 *
2271 tcp_scoreboard_replay (u8 * s, tcp_connection_t * tc, u8 verbose)
2272 {
2273   int i, trace_len;
2274   scoreboard_trace_elt_t *trace;
2275   u32 next_ack, left, group, has_new_ack = 0;
2276   tcp_connection_t _dummy_tc, *dummy_tc = &_dummy_tc;
2277   sack_block_t *block;
2278
2279   if (!TCP_SCOREBOARD_TRACE)
2280     {
2281       s = format (s, "scoreboard tracing not enabled");
2282       return s;
2283     }
2284
2285   if (!tc)
2286     return s;
2287
2288   clib_memset (dummy_tc, 0, sizeof (*dummy_tc));
2289   tcp_connection_timers_init (dummy_tc);
2290   scoreboard_init (&dummy_tc->sack_sb);
2291   dummy_tc->rcv_opts.flags |= TCP_OPTS_FLAG_SACK;
2292
2293 #if TCP_SCOREBOARD_TRACE
2294   trace = tc->sack_sb.trace;
2295   trace_len = vec_len (tc->sack_sb.trace);
2296 #endif
2297
2298   for (i = 0; i < trace_len; i++)
2299     {
2300       if (trace[i].ack != 0)
2301         {
2302           dummy_tc->snd_una = trace[i].ack - 1448;
2303           dummy_tc->snd_una_max = trace[i].ack;
2304         }
2305     }
2306
2307   left = 0;
2308   while (left < trace_len)
2309     {
2310       group = trace[left].group;
2311       vec_reset_length (dummy_tc->rcv_opts.sacks);
2312       has_new_ack = 0;
2313       while (trace[left].group == group)
2314         {
2315           if (trace[left].ack != 0)
2316             {
2317               if (verbose)
2318                 s = format (s, "Adding ack %u, snd_una_max %u, segs: ",
2319                             trace[left].ack, trace[left].snd_una_max);
2320               dummy_tc->snd_una_max = trace[left].snd_una_max;
2321               next_ack = trace[left].ack;
2322               has_new_ack = 1;
2323             }
2324           else
2325             {
2326               if (verbose)
2327                 s = format (s, "[%u, %u], ", trace[left].start,
2328                             trace[left].end);
2329               vec_add2 (dummy_tc->rcv_opts.sacks, block, 1);
2330               block->start = trace[left].start;
2331               block->end = trace[left].end;
2332             }
2333           left++;
2334         }
2335
2336       /* Push segments */
2337       tcp_rcv_sacks (dummy_tc, next_ack);
2338       if (has_new_ack)
2339         dummy_tc->snd_una = next_ack;
2340
2341       if (verbose)
2342         s = format (s, "result: %U", format_tcp_scoreboard,
2343                     &dummy_tc->sack_sb);
2344
2345     }
2346   s = format (s, "result: %U", format_tcp_scoreboard, &dummy_tc->sack_sb);
2347
2348   return s;
2349 }
2350
2351 static clib_error_t *
2352 tcp_scoreboard_trace_fn (vlib_main_t * vm, unformat_input_t * input,
2353                          vlib_cli_command_t * cmd_arg)
2354 {
2355   transport_connection_t *tconn = 0;
2356   tcp_connection_t *tc = 0;
2357   u8 *str = 0;
2358   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2359     {
2360       if (unformat (input, "%U", unformat_transport_connection, &tconn,
2361                     TRANSPORT_PROTO_TCP))
2362         ;
2363       else
2364         return clib_error_return (0, "unknown input `%U'",
2365                                   format_unformat_error, input);
2366     }
2367
2368   if (!TCP_SCOREBOARD_TRACE)
2369     {
2370       vlib_cli_output (vm, "scoreboard tracing not enabled");
2371       return 0;
2372     }
2373
2374   tc = tcp_get_connection_from_transport (tconn);
2375   if (!tc)
2376     {
2377       vlib_cli_output (vm, "connection not found");
2378       return 0;
2379     }
2380   str = tcp_scoreboard_replay (str, tc, 1);
2381   vlib_cli_output (vm, "%v", str);
2382   return 0;
2383 }
2384
2385 /* *INDENT-OFF* */
2386 VLIB_CLI_COMMAND (tcp_replay_scoreboard_command, static) =
2387 {
2388   .path = "tcp replay scoreboard",
2389   .short_help = "tcp replay scoreboard <connection>",
2390   .function = tcp_scoreboard_trace_fn,
2391 };
2392 /* *INDENT-ON* */
2393
2394 static clib_error_t *
2395 show_tcp_punt_fn (vlib_main_t * vm, unformat_input_t * input,
2396                   vlib_cli_command_t * cmd_arg)
2397 {
2398   tcp_main_t *tm = vnet_get_tcp_main ();
2399   if (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2400     return clib_error_return (0, "unknown input `%U'", format_unformat_error,
2401                               input);
2402   vlib_cli_output (vm, "IPv4 TCP punt: %s",
2403                    tm->punt_unknown4 ? "enabled" : "disabled");
2404   vlib_cli_output (vm, "IPv6 TCP punt: %s",
2405                    tm->punt_unknown6 ? "enabled" : "disabled");
2406   return 0;
2407 }
2408 /* *INDENT-OFF* */
2409 VLIB_CLI_COMMAND (show_tcp_punt_command, static) =
2410 {
2411   .path = "show tcp punt",
2412   .short_help = "show tcp punt",
2413   .function = show_tcp_punt_fn,
2414 };
2415 /* *INDENT-ON* */
2416
2417 static clib_error_t *
2418 show_tcp_stats_fn (vlib_main_t * vm, unformat_input_t * input,
2419                    vlib_cli_command_t * cmd)
2420 {
2421   tcp_main_t *tm = vnet_get_tcp_main ();
2422   tcp_worker_ctx_t *wrk;
2423   u32 thread;
2424
2425   if (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2426     return clib_error_return (0, "unknown input `%U'", format_unformat_error,
2427                               input);
2428   for (thread = 0; thread < vec_len (tm->wrk_ctx); thread++)
2429     {
2430       wrk = tcp_get_worker (thread);
2431       vlib_cli_output (vm, "Thread %u:\n", thread);
2432
2433       if (clib_fifo_elts (wrk->pending_timers))
2434         vlib_cli_output (vm, " %lu pending timers",
2435                          clib_fifo_elts (wrk->pending_timers));
2436
2437 #define _(name,type,str)                                        \
2438   if (wrk->stats.name)                                          \
2439     vlib_cli_output (vm, " %lu %s", wrk->stats.name, str);
2440       foreach_tcp_wrk_stat
2441 #undef _
2442     }
2443
2444   return 0;
2445 }
2446
2447 /* *INDENT-OFF* */
2448 VLIB_CLI_COMMAND (show_tcp_stats_command, static) =
2449 {
2450   .path = "show tcp stats",
2451   .short_help = "show tcp stats",
2452   .function = show_tcp_stats_fn,
2453 };
2454 /* *INDENT-ON* */
2455
2456 static clib_error_t *
2457 clear_tcp_stats_fn (vlib_main_t * vm, unformat_input_t * input,
2458                     vlib_cli_command_t * cmd)
2459 {
2460   tcp_main_t *tm = vnet_get_tcp_main ();
2461   tcp_worker_ctx_t *wrk;
2462   u32 thread;
2463
2464   if (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
2465     return clib_error_return (0, "unknown input `%U'", format_unformat_error,
2466                               input);
2467
2468   for (thread = 0; thread < vec_len (tm->wrk_ctx); thread++)
2469     {
2470       wrk = tcp_get_worker (thread);
2471       clib_memset (&wrk->stats, 0, sizeof (wrk->stats));
2472     }
2473
2474   return 0;
2475 }
2476
2477 /* *INDENT-OFF* */
2478 VLIB_CLI_COMMAND (clear_tcp_stats_command, static) =
2479 {
2480   .path = "clear tcp stats",
2481   .short_help = "clear tcp stats",
2482   .function = clear_tcp_stats_fn,
2483 };
2484 /* *INDENT-ON* */
2485
2486 /*
2487  * fd.io coding-style-patch-verification: ON
2488  *
2489  * Local Variables:
2490  * eval: (c-set-style "gnu")
2491  * End:
2492  */