src/plugins/unittest/tcp_test.c

   1 /*
   2  * Copyright (c) 2017 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 #include <vnet/tcp/tcp.h>
  16
  17 #define TCP_TEST_I(_cond, _comment, _args...)                   \
  18 ({                                                              \
  19   int _evald = (_cond);                                         \
  20   if (!(_evald)) {                                              \
  21     fformat(stderr, "FAIL:%d: " _comment "\n",                  \
  22             __LINE__, ##_args);                                 \
  23   } else {                                                      \
  24     fformat(stderr, "PASS:%d: " _comment "\n",                  \
  25             __LINE__, ##_args);                                 \
  26   }                                                             \
  27   _evald;                                                       \
  28 })
  29
  30 #define TCP_TEST(_cond, _comment, _args...)                     \
  31 {                                                               \
  32     if (!TCP_TEST_I(_cond, _comment, ##_args)) {                \
  33         return 1;                                               \
  34     }                                                           \
  35 }
  36
  37 /* *INDENT-OFF* */
  38 scoreboard_trace_elt_t sb_trace[] = {};
  39 /* *INDENT-ON* */
  40
  41 static int
  42 tcp_test_scoreboard_replay (vlib_main_t * vm, unformat_input_t * input)
  43 {
  44   int verbose = 0;
  45   tcp_connection_t _tc, *tc = &_tc;
  46   u8 *s = 0;
  47
  48   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
  49     {
  50       if (unformat (input, "detail"))
  51         verbose = 1;
  52       else
  53         {
  54           clib_error_t *e = clib_error_return
  55             (0, "unknown input `%U'", format_unformat_error, input);
  56           clib_error_report (e);
  57           return -1;
  58         }
  59     }
  60
  61 #if TCP_SCOREBOARD_TRACE
  62   tc->sack_sb.trace = sb_trace;
  63 #endif
  64   s = tcp_scoreboard_replay (s, tc, verbose);
  65   vlib_cli_output (vm, "%v", s);
  66   return 0;
  67 }
  68
  69 static int
  70 tcp_test_sack_rx (vlib_main_t * vm, unformat_input_t * input)
  71 {
  72   tcp_connection_t _tc, *tc = &_tc;
  73   sack_scoreboard_t *sb = &tc->sack_sb;
  74   sack_block_t *sacks = 0, block;
  75   sack_scoreboard_hole_t *hole;
  76   int i, verbose = 0;
  77
  78   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
  79     {
  80       if (unformat (input, "verbose"))
  81         verbose = 1;
  82       else if (unformat (input, "replay"))
  83         return tcp_test_scoreboard_replay (vm, input);
  84     }
  85
  86   clib_memset (tc, 0, sizeof (*tc));
  87
  88   tc->snd_una = 0;
  89   tc->snd_una_max = 1000;
  90   tc->snd_nxt = 1000;
  91   tc->rcv_opts.flags |= TCP_OPTS_FLAG_SACK;
  92   tc->snd_mss = 150;
  93   scoreboard_init (&tc->sack_sb);
  94
  95   for (i = 0; i < 1000 / 100; i++)
  96     {
  97       block.start = i * 100;
  98       block.end = (i + 1) * 100;
  99       vec_add1 (sacks, block);
 100     }
 101
 102   /*
 103    * Inject even blocks
 104    */
 105
 106   for (i = 0; i < 1000 / 200; i++)
 107     {
 108       vec_add1 (tc->rcv_opts.sacks, sacks[i * 2]);
 109     }
 110   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 111   tcp_rcv_sacks (tc, 0);
 112
 113   if (verbose)
 114     vlib_cli_output (vm, "sb after even blocks (mss %u):\n%U",
 115                      tc->snd_mss, format_tcp_scoreboard, sb, tc);
 116
 117   TCP_TEST ((pool_elts (sb->holes) == 5),
 118             "scoreboard has %d elements", pool_elts (sb->holes));
 119
 120   /* First SACK block should be rejected */
 121   hole = scoreboard_first_hole (sb);
 122   TCP_TEST ((hole->start == 0 && hole->end == 200),
 123             "first hole start %u end %u", hole->start, hole->end);
 124   hole = scoreboard_last_hole (sb);
 125   TCP_TEST ((hole->start == 900 && hole->end == 1000),
 126             "last hole start %u end %u", hole->start, hole->end);
 127   TCP_TEST ((sb->sacked_bytes == 400), "sacked bytes %d", sb->sacked_bytes);
 128   TCP_TEST ((sb->snd_una_adv == 0), "snd_una_adv %u", sb->snd_una_adv);
 129   TCP_TEST ((sb->last_sacked_bytes == 400),
 130             "last sacked bytes %d", sb->last_sacked_bytes);
 131   TCP_TEST ((sb->high_sacked == 900), "high sacked %u", sb->high_sacked);
 132   TCP_TEST ((sb->lost_bytes == 300), "lost bytes %u", sb->lost_bytes);
 133
 134   /*
 135    * Inject odd blocks
 136    */
 137
 138   vec_reset_length (tc->rcv_opts.sacks);
 139   for (i = 0; i < 1000 / 200; i++)
 140     {
 141       vec_add1 (tc->rcv_opts.sacks, sacks[i * 2 + 1]);
 142     }
 143   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 144   tcp_rcv_sacks (tc, 0);
 145
 146   if (verbose)
 147     vlib_cli_output (vm, "\nsb after odd blocks:\n%U", format_tcp_scoreboard,
 148                      sb, tc);
 149
 150   hole = scoreboard_first_hole (sb);
 151   TCP_TEST ((pool_elts (sb->holes) == 1),
 152             "scoreboard has %d holes", pool_elts (sb->holes));
 153   TCP_TEST ((hole->start == 0 && hole->end == 100),
 154             "first hole start %u end %u", hole->start, hole->end);
 155   TCP_TEST ((sb->sacked_bytes == 900), "sacked bytes %d", sb->sacked_bytes);
 156   TCP_TEST ((sb->snd_una_adv == 0), "snd_una_adv %u", sb->snd_una_adv);
 157   TCP_TEST ((sb->high_sacked == 1000), "high sacked %u", sb->high_sacked);
 158   TCP_TEST ((sb->last_sacked_bytes == 500),
 159             "last sacked bytes %d", sb->last_sacked_bytes);
 160   TCP_TEST ((sb->lost_bytes == 100), "lost bytes %u", sb->lost_bytes);
 161
 162   /*
 163    *  Ack until byte 100, all bytes are now acked + sacked
 164    */
 165   tcp_rcv_sacks (tc, 100);
 166   if (verbose)
 167     vlib_cli_output (vm, "\nack until byte 100:\n%U", format_tcp_scoreboard,
 168                      sb, tc);
 169
 170   TCP_TEST ((pool_elts (sb->holes) == 0),
 171             "scoreboard has %d elements", pool_elts (sb->holes));
 172   TCP_TEST ((sb->snd_una_adv == 900),
 173             "snd_una_adv after ack %u", sb->snd_una_adv);
 174   TCP_TEST ((sb->high_sacked == 1000), "max sacked byte %u", sb->high_sacked);
 175   TCP_TEST ((sb->sacked_bytes == 0), "sacked bytes %d", sb->sacked_bytes);
 176   TCP_TEST ((sb->last_sacked_bytes == 0),
 177             "last sacked bytes %d", sb->last_sacked_bytes);
 178   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 179
 180   /*
 181    * Add new block
 182    */
 183
 184   vec_reset_length (tc->rcv_opts.sacks);
 185
 186   block.start = 1200;
 187   block.end = 1300;
 188   vec_add1 (tc->rcv_opts.sacks, block);
 189
 190   tc->snd_una_max = 1500;
 191   tc->snd_una = 1000;
 192   tc->snd_nxt = 1500;
 193   tcp_rcv_sacks (tc, 1000);
 194
 195   if (verbose)
 196     vlib_cli_output (vm, "\nadd [1200, 1300] snd_una_max 1500, snd_una 1000:"
 197                      " \n%U", format_tcp_scoreboard, sb, tc);
 198
 199   TCP_TEST ((sb->snd_una_adv == 0),
 200             "snd_una_adv after ack %u", sb->snd_una_adv);
 201   TCP_TEST ((pool_elts (sb->holes) == 2),
 202             "scoreboard has %d holes", pool_elts (sb->holes));
 203   hole = scoreboard_first_hole (sb);
 204   TCP_TEST ((hole->start == 1000 && hole->end == 1200),
 205             "first hole start %u end %u", hole->start, hole->end);
 206   TCP_TEST ((sb->snd_una_adv == 0),
 207             "snd_una_adv after ack %u", sb->snd_una_adv);
 208   TCP_TEST ((sb->high_sacked == 1300), "max sacked byte %u", sb->high_sacked);
 209   hole = scoreboard_last_hole (sb);
 210   TCP_TEST ((hole->start == 1300 && hole->end == 1500),
 211             "last hole start %u end %u", hole->start, hole->end);
 212   TCP_TEST ((sb->sacked_bytes == 100), "sacked bytes %d", sb->sacked_bytes);
 213   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 214
 215   /*
 216    * Ack first hole
 217    */
 218
 219   vec_reset_length (tc->rcv_opts.sacks);
 220   tcp_rcv_sacks (tc, 1200);
 221
 222   if (verbose)
 223     vlib_cli_output (vm, "\nsb ack up to byte 1200:\n%U",
 224                      format_tcp_scoreboard, sb, tc);
 225
 226   TCP_TEST ((sb->snd_una_adv == 100),
 227             "snd_una_adv after ack %u", sb->snd_una_adv);
 228   TCP_TEST ((sb->sacked_bytes == 0), "sacked bytes %d", sb->sacked_bytes);
 229   TCP_TEST ((pool_elts (sb->holes) == 0),
 230             "scoreboard has %d elements", pool_elts (sb->holes));
 231   TCP_TEST ((sb->last_bytes_delivered == 100), "last bytes delivered %d",
 232             sb->last_bytes_delivered);
 233   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 234   TCP_TEST ((sb->head == TCP_INVALID_SACK_HOLE_INDEX), "head %u", sb->head);
 235   TCP_TEST ((sb->tail == TCP_INVALID_SACK_HOLE_INDEX), "tail %u", sb->tail);
 236
 237   /*
 238    * Add some more blocks and then remove all
 239    */
 240   vec_reset_length (tc->rcv_opts.sacks);
 241   tc->snd_una += sb->snd_una_adv;
 242   tc->snd_una_max = 1900;
 243   for (i = 0; i < 5; i++)
 244     {
 245       block.start = i * 100 + 1200;
 246       block.end = (i + 1) * 100 + 1200;
 247       vec_add1 (tc->rcv_opts.sacks, block);
 248     }
 249   tcp_rcv_sacks (tc, 1900);
 250
 251   scoreboard_clear (sb);
 252   if (verbose)
 253     vlib_cli_output (vm, "\nsb cleared all:\n%U", format_tcp_scoreboard, sb,
 254                      tc);
 255
 256   TCP_TEST ((pool_elts (sb->holes) == 0),
 257             "number of holes %d", pool_elts (sb->holes));
 258   TCP_TEST ((sb->head == TCP_INVALID_SACK_HOLE_INDEX), "head %u", sb->head);
 259   TCP_TEST ((sb->tail == TCP_INVALID_SACK_HOLE_INDEX), "tail %u", sb->tail);
 260
 261   /*
 262    * Re-inject odd blocks and ack them all
 263    */
 264
 265   tc->snd_una = 0;
 266   tc->snd_una_max = 1000;
 267   tc->snd_nxt = 1000;
 268   for (i = 0; i < 5; i++)
 269     {
 270       vec_add1 (tc->rcv_opts.sacks, sacks[i * 2 + 1]);
 271     }
 272   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 273   tcp_rcv_sacks (tc, 0);
 274   if (verbose)
 275     vlib_cli_output (vm, "\nsb added odd blocks snd_una 0 snd_una_max 1500:"
 276                      "\n%U", format_tcp_scoreboard, sb, tc);
 277   TCP_TEST ((pool_elts (sb->holes) == 5),
 278             "scoreboard has %d elements", pool_elts (sb->holes));
 279   TCP_TEST ((sb->lost_bytes == 300), "lost bytes %u", sb->lost_bytes);
 280
 281   tcp_rcv_sacks (tc, 950);
 282
 283   if (verbose)
 284     vlib_cli_output (vm, "\nack [0, 950]:\n%U", format_tcp_scoreboard, sb,
 285                      tc);
 286
 287   TCP_TEST ((pool_elts (sb->holes) == 0),
 288             "scoreboard has %d elements", pool_elts (sb->holes));
 289   TCP_TEST ((sb->snd_una_adv == 50), "snd_una_adv %u", sb->snd_una_adv);
 290   TCP_TEST ((sb->sacked_bytes == 0), "sacked bytes %d", sb->sacked_bytes);
 291   TCP_TEST ((sb->last_sacked_bytes == 0),
 292             "last sacked bytes %d", sb->last_sacked_bytes);
 293   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 294
 295   /*
 296    * Inject one block, ack it and overlap hole
 297    */
 298
 299   tc->snd_una = 0;
 300   tc->snd_una_max = 1000;
 301   tc->snd_nxt = 1000;
 302
 303   block.start = 100;
 304   block.end = 500;
 305   vec_add1 (tc->rcv_opts.sacks, block);
 306   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 307
 308   tcp_rcv_sacks (tc, 0);
 309
 310   if (verbose)
 311     vlib_cli_output (vm, "\nsb added [100, 500] snd_una 0 snd_una_max 1000:"
 312                      "\n%U", format_tcp_scoreboard, sb, tc);
 313
 314   tcp_rcv_sacks (tc, 800);
 315
 316   if (verbose)
 317     vlib_cli_output (vm, "\nsb ack [0, 800]:\n%U", format_tcp_scoreboard, sb,
 318                      tc);
 319
 320   TCP_TEST ((pool_elts (sb->holes) == 0),
 321             "scoreboard has %d elements", pool_elts (sb->holes));
 322   TCP_TEST ((sb->snd_una_adv == 0), "snd_una_adv %u", sb->snd_una_adv);
 323   TCP_TEST ((sb->sacked_bytes == 0), "sacked bytes %d", sb->sacked_bytes);
 324   TCP_TEST ((sb->last_sacked_bytes == 0), "last sacked bytes %d",
 325             sb->last_sacked_bytes);
 326   TCP_TEST ((sb->last_bytes_delivered == 400),
 327             "last bytes delivered %d", sb->last_bytes_delivered);
 328   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 329   TCP_TEST ((sb->head == TCP_INVALID_SACK_HOLE_INDEX), "head %u", sb->head);
 330   TCP_TEST ((sb->tail == TCP_INVALID_SACK_HOLE_INDEX), "tail %u", sb->tail);
 331
 332   /*
 333    * One hole close to head, patch head, split in two and start acking
 334    * the lowest part
 335    */
 336   scoreboard_clear (sb);
 337   tc->snd_una = 0;
 338   tc->snd_una_max = 1000;
 339   tc->snd_nxt = 1000;
 340
 341   block.start = 500;
 342   block.end = 1000;
 343   vec_add1 (tc->rcv_opts.sacks, block);
 344   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 345
 346   tcp_rcv_sacks (tc, 0);
 347   if (verbose)
 348     vlib_cli_output (vm, "\nsb added [500, 1000]:\n%U",
 349                      format_tcp_scoreboard, sb, tc);
 350   TCP_TEST ((sb->sacked_bytes == 500), "sacked bytes %d", sb->sacked_bytes);
 351   TCP_TEST ((sb->last_sacked_bytes == 500), "last sacked bytes %d",
 352             sb->last_sacked_bytes);
 353   TCP_TEST ((sb->lost_bytes == 500), "lost bytes %u", sb->lost_bytes);
 354
 355   vec_reset_length (tc->rcv_opts.sacks);
 356   block.start = 300;
 357   block.end = 400;
 358   vec_add1 (tc->rcv_opts.sacks, block);
 359   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 360   tcp_rcv_sacks (tc, 100);
 361   if (verbose)
 362     vlib_cli_output (vm, "\nsb added [0, 100] [300, 400]:\n%U",
 363                      format_tcp_scoreboard, sb, tc);
 364   TCP_TEST ((pool_elts (sb->holes) == 2),
 365             "scoreboard has %d elements", pool_elts (sb->holes));
 366   TCP_TEST ((sb->sacked_bytes == 600), "sacked bytes %d", sb->sacked_bytes);
 367   TCP_TEST ((sb->last_sacked_bytes == 100), "last sacked bytes %d",
 368             sb->last_sacked_bytes);
 369   TCP_TEST ((sb->last_bytes_delivered == 0), "last bytes delivered %d",
 370             sb->last_bytes_delivered);
 371   TCP_TEST ((sb->lost_bytes == 300), "lost bytes %u", sb->lost_bytes);
 372
 373   tc->snd_una = 100;
 374   tcp_rcv_sacks (tc, 200);
 375   tc->snd_una = 200;
 376   tcp_rcv_sacks (tc, 300);
 377   if (verbose)
 378     vlib_cli_output (vm, "\nacked [0, 300] in two steps:\n%U",
 379                      format_tcp_scoreboard, sb, tc);
 380   TCP_TEST ((sb->sacked_bytes == 500), "sacked bytes %d", sb->sacked_bytes);
 381   TCP_TEST ((sb->lost_bytes == 100), "lost bytes %u", sb->lost_bytes);
 382   TCP_TEST ((sb->last_bytes_delivered == 100), "last bytes delivered %d",
 383             sb->last_bytes_delivered);
 384
 385   tc->snd_una = 400;
 386   tcp_rcv_sacks (tc, 500);
 387   if (verbose)
 388     vlib_cli_output (vm, "\nacked [400, 500]:\n%U", format_tcp_scoreboard, sb,
 389                      tc);
 390   TCP_TEST ((pool_elts (sb->holes) == 0),
 391             "scoreboard has %d elements", pool_elts (sb->holes));
 392   TCP_TEST ((sb->sacked_bytes == 0), "sacked bytes %d", sb->sacked_bytes);
 393   TCP_TEST ((sb->last_sacked_bytes == 0), "last sacked bytes %d",
 394             sb->last_sacked_bytes);
 395   TCP_TEST ((sb->last_bytes_delivered == 500), "last bytes delivered %d",
 396             sb->last_bytes_delivered);
 397   TCP_TEST ((sb->lost_bytes == 0), "lost bytes %u", sb->lost_bytes);
 398   TCP_TEST ((sb->snd_una_adv == 500), "snd_una_adv %u", sb->snd_una_adv);
 399   TCP_TEST ((sb->head == TCP_INVALID_SACK_HOLE_INDEX), "head %u", sb->head);
 400   TCP_TEST ((sb->tail == TCP_INVALID_SACK_HOLE_INDEX), "tail %u", sb->tail);
 401
 402   /*
 403    * Re-ack high sacked, to make sure last_bytes_delivered and
 404    * snd_una_adv are 0-ed
 405    */
 406   tcp_rcv_sacks (tc, 1000);
 407   if (verbose)
 408     vlib_cli_output (vm, "\nAck high sacked:\n%U", format_tcp_scoreboard, sb,
 409                      tc);
 410   TCP_TEST ((sb->last_bytes_delivered == 0), "last bytes delivered %d",
 411             sb->last_bytes_delivered);
 412   TCP_TEST ((sb->snd_una_adv == 0), "snd_una_adv %u", sb->snd_una_adv);
 413
 414   /*
 415    * Add [1200, 1500] and test that [1000, 1200] is lost (bytes condition)
 416    * snd_una = 1000 and snd_una_max = 1600
 417    */
 418   tc->snd_una = 1000;
 419   tc->snd_una_max = 1600;
 420   vec_reset_length (tc->rcv_opts.sacks);
 421   block.start = 1200;
 422   block.end = 1500;
 423   vec_add1 (tc->rcv_opts.sacks, block);
 424   tc->rcv_opts.n_sack_blocks = vec_len (tc->rcv_opts.sacks);
 425   tcp_rcv_sacks (tc, 1000);
 426   if (verbose)
 427     vlib_cli_output (vm, "\nacked [1200, 1500] test first hole is lost:\n%U",
 428                      format_tcp_scoreboard, sb, tc);
 429   TCP_TEST ((pool_elts (sb->holes) == 2), "scoreboard has %d elements",
 430             pool_elts (sb->holes));
 431   TCP_TEST ((sb->sacked_bytes == 300), "sacked bytes %d", sb->sacked_bytes);
 432   TCP_TEST ((sb->last_sacked_bytes == 300), "last sacked bytes %d",
 433             sb->last_sacked_bytes);
 434   TCP_TEST ((sb->last_bytes_delivered == 0), "last bytes delivered %d",
 435             sb->last_bytes_delivered);
 436   TCP_TEST ((sb->lost_bytes == 200), "lost bytes %u", sb->lost_bytes);
 437   TCP_TEST ((sb->snd_una_adv == 0), "snd_una_adv %u", sb->snd_una_adv);
 438
 439   return 0;
 440 }
 441
 442 static int
 443 tcp_test_sack_tx (vlib_main_t * vm, unformat_input_t * input)
 444 {
 445   tcp_connection_t _tc, *tc = &_tc;
 446   sack_block_t *sacks;
 447   int i, verbose = 0, expected;
 448
 449   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
 450     {
 451       if (unformat (input, "verbose"))
 452         verbose = 1;
 453       else
 454         {
 455           vlib_cli_output (vm, "parse error: '%U'", format_unformat_error,
 456                            input);
 457           return -1;
 458         }
 459     }
 460
 461   clib_memset (tc, 0, sizeof (*tc));
 462
 463   /*
 464    * Add odd sack block pairs
 465    */
 466   for (i = 1; i < 10; i += 2)
 467     {
 468       tcp_update_sack_list (tc, i * 100, (i + 1) * 100);
 469     }
 470
 471   TCP_TEST ((vec_len (tc->snd_sacks) == 5), "sack blocks %d expected %d",
 472             vec_len (tc->snd_sacks), 5);
 473   TCP_TEST ((tc->snd_sacks[0].start = 900),
 474             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 475             900);
 476
 477   /*
 478    * Try to add one extra
 479    */
 480   sacks = vec_dup (tc->snd_sacks);
 481
 482   tcp_update_sack_list (tc, 1100, 1200);
 483   if (verbose)
 484     vlib_cli_output (vm, "add new segment [1100, 1200]\n%U",
 485                      format_tcp_sacks, tc);
 486   expected = 5 < TCP_MAX_SACK_BLOCKS ? 6 : 5;
 487   TCP_TEST ((vec_len (tc->snd_sacks) == expected),
 488             "sack blocks %d expected %d", vec_len (tc->snd_sacks), expected);
 489   TCP_TEST ((tc->snd_sacks[0].start == 1100),
 490             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 491             1100);
 492
 493   /* restore */
 494   vec_free (tc->snd_sacks);
 495   tc->snd_sacks = sacks;
 496
 497   /*
 498    * Overlap first 2 segment
 499    */
 500   tc->rcv_nxt = 300;
 501   tcp_update_sack_list (tc, 300, 300);
 502   if (verbose)
 503     vlib_cli_output (vm, "overlap first 2 segments:\n%U",
 504                      format_tcp_sacks, tc);
 505   TCP_TEST ((vec_len (tc->snd_sacks) == 3), "sack blocks %d expected %d",
 506             vec_len (tc->snd_sacks), 3);
 507   TCP_TEST ((tc->snd_sacks[0].start == 900),
 508             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 509             500);
 510
 511   /*
 512    * Add a new segment
 513    */
 514   tcp_update_sack_list (tc, 1100, 1200);
 515   if (verbose)
 516     vlib_cli_output (vm, "add new segment [1100, 1200]\n%U",
 517                      format_tcp_sacks, tc);
 518   TCP_TEST ((vec_len (tc->snd_sacks) == 4), "sack blocks %d expected %d",
 519             vec_len (tc->snd_sacks), 4);
 520   TCP_TEST ((tc->snd_sacks[0].start == 1100),
 521             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 522             1100);
 523
 524   /*
 525    * Join middle segments
 526    */
 527   tcp_update_sack_list (tc, 800, 900);
 528   if (verbose)
 529     vlib_cli_output (vm, "join middle segments [800, 900]\n%U",
 530                      format_tcp_sacks, tc);
 531
 532   TCP_TEST ((vec_len (tc->snd_sacks) == 3), "sack blocks %d expected %d",
 533             vec_len (tc->snd_sacks), 3);
 534   TCP_TEST ((tc->snd_sacks[0].start == 700),
 535             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 536             1100);
 537
 538   /*
 539    * Advance rcv_nxt to overlap all
 540    */
 541   tc->rcv_nxt = 1200;
 542   tcp_update_sack_list (tc, 1200, 1200);
 543   if (verbose)
 544     vlib_cli_output (vm, "advance rcv_nxt to 1200\n%U", format_tcp_sacks, tc);
 545   TCP_TEST ((vec_len (tc->snd_sacks) == 0), "sack blocks %d expected %d",
 546             vec_len (tc->snd_sacks), 0);
 547
 548
 549   /*
 550    * Add 2 blocks, overwrite first and update rcv_nxt to also remove it
 551    */
 552
 553   vec_reset_length (tc->snd_sacks);
 554   tc->rcv_nxt = 0;
 555
 556   tcp_update_sack_list (tc, 100, 200);
 557   tcp_update_sack_list (tc, 300, 400);
 558
 559   if (verbose)
 560     vlib_cli_output (vm, "add [100, 200] [300, 400]\n%U",
 561                      format_tcp_sacks, tc);
 562   TCP_TEST ((vec_len (tc->snd_sacks) == 2),
 563             "sack blocks %d expected %d", vec_len (tc->snd_sacks), 2);
 564   TCP_TEST ((tc->snd_sacks[0].start == 300),
 565             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 566             300);
 567
 568   tc->rcv_nxt = 100;
 569   tcp_update_sack_list (tc, 100, 100);
 570   if (verbose)
 571     vlib_cli_output (vm, "add [100, 200] rcv_nxt = 100\n%U",
 572                      format_tcp_sacks, tc);
 573   TCP_TEST ((vec_len (tc->snd_sacks) == 1),
 574             "sack blocks %d expected %d", vec_len (tc->snd_sacks), 1);
 575   TCP_TEST ((tc->snd_sacks[0].start == 300),
 576             "first sack block start %u expected %u", tc->snd_sacks[0].start,
 577             300);
 578   return 0;
 579 }
 580
 581 static int
 582 tcp_test_sack (vlib_main_t * vm, unformat_input_t * input)
 583 {
 584   int res = 0;
 585
 586   /* Run all tests */
 587   if (unformat_check_input (input) == UNFORMAT_END_OF_INPUT)
 588     {
 589       if (tcp_test_sack_tx (vm, input))
 590         {
 591           return -1;
 592         }
 593
 594       if (tcp_test_sack_rx (vm, input))
 595         {
 596           return -1;
 597         }
 598     }
 599   else
 600     {
 601       if (unformat (input, "tx"))
 602         {
 603           res = tcp_test_sack_tx (vm, input);
 604         }
 605       else if (unformat (input, "rx"))
 606         {
 607           res = tcp_test_sack_rx (vm, input);
 608         }
 609     }
 610
 611   return res;
 612 }
 613
 614
 615 typedef struct
 616 {
 617   u32 offset;
 618   u32 len;
 619 } test_pattern_t;
 620
 621 /* *INDENT-OFF* */
 622 test_pattern_t test_pattern[] = {
 623   {380, 8}, {768, 8}, {1156, 8}, {1544, 8}, {1932, 8}, {2320, 8}, {2708, 8},
 624   {2992, 8}, {372, 8}, {760, 8}, {1148, 8}, {1536, 8}, {1924, 8}, {2312, 8},
 625   {2700, 8}, {2984, 8}, {364, 8}, {752, 8}, {1140, 8}, {1528, 8}, {1916, 8},
 626   {2304, 8}, {2692, 8}, {2976, 8}, {356, 8}, {744, 8}, {1132, 8}, {1520, 8},
 627   {1908, 8}, {2296, 8}, {2684, 8}, {2968, 8}, {348, 8}, {736, 8}, {1124, 8},
 628   {1512, 8}, {1900, 8}, {2288, 8}, {2676, 8}, {2960, 8}, {340, 8}, {728, 8},
 629   {1116, 8}, {1504, 8}, {1892, 8}, {2280, 8}, {2668, 8}, {2952, 8}, {332, 8},
 630   {720, 8}, {1108, 8}, {1496, 8}, {1884, 8}, {2272, 8}, {2660, 8}, {2944, 8},
 631   {324, 8}, {712, 8}, {1100, 8}, {1488, 8}, {1876, 8}, {2264, 8}, {2652, 8},
 632   {2936, 8}, {316, 8}, {704, 8}, {1092, 8}, {1480, 8}, {1868, 8}, {2256, 8},
 633   {2644, 8}, {2928, 8}, {308, 8}, {696, 8}, {1084, 8}, {1472, 8}, {1860, 8},
 634   {2248, 8}, {2636, 8}, {2920, 8}, {300, 8}, {688, 8}, {1076, 8}, {1464, 8},
 635   {1852, 8}, {2240, 8}, {2628, 8}, {2912, 8}, {292, 8}, {680, 8}, {1068, 8},
 636   {1456, 8}, {1844, 8}, {2232, 8}, {2620, 8}, {2904, 8}, {284, 8}, {672, 8},
 637   {1060, 8}, {1448, 8}, {1836, 8}, {2224, 8}, {2612, 8}, {2896, 8}, {276, 8},
 638   {664, 8}, {1052, 8}, {1440, 8}, {1828, 8},  {2216, 8}, {2604, 8}, {2888, 8},
 639   {268, 8}, {656, 8}, {1044, 8}, {1432, 8}, {1820, 8}, {2208, 8}, {2596, 8},
 640   {2880, 8}, {260, 8}, {648, 8}, {1036, 8}, {1424, 8}, {1812, 8}, {2200, 8},
 641   {2588, 8}, {2872, 8}, {252, 8}, {640, 8}, {1028, 8}, {1416, 8}, {1804, 8},
 642   {2192, 8}, {2580, 8}, {2864, 8}, {244, 8}, {632, 8}, {1020, 8}, {1408, 8},
 643   {1796, 8}, {2184, 8}, {2572, 8}, {2856, 8}, {236, 8}, {624, 8}, {1012, 8},
 644   {1400, 8}, {1788, 8}, {2176, 8}, {2564, 8}, {2848, 8}, {228, 8}, {616, 8},
 645   {1004, 8}, {1392, 8}, {1780, 8}, {2168, 8}, {2556, 8}, {2840, 8}, {220, 8},
 646   {608, 8}, {996, 8}, {1384, 8}, {1772, 8}, {2160, 8}, {2548, 8}, {2832, 8},
 647   {212, 8}, {600, 8}, {988, 8}, {1376, 8}, {1764, 8}, {2152, 8}, {2540, 8},
 648   {2824, 8}, {204, 8}, {592, 8}, {980, 8}, {1368, 8}, {1756, 8}, {2144, 8},
 649   {2532, 8}, {2816, 8}, {196, 8}, {584, 8}, {972, 8}, {1360, 8}, {1748, 8},
 650   {2136, 8}, {2524, 8}, {2808, 8}, {188, 8}, {576, 8}, {964, 8}, {1352, 8},
 651   {1740, 8}, {2128, 8}, {2516, 8}, {2800, 8}, {180, 8}, {568, 8}, {956, 8},
 652   {1344, 8}, {1732, 8}, {2120, 8}, {2508, 8}, {2792, 8}, {172, 8}, {560, 8},
 653   {948, 8}, {1336, 8}, {1724, 8}, {2112, 8}, {2500, 8}, {2784, 8}, {164, 8},
 654   {552, 8}, {940, 8}, {1328, 8}, {1716, 8}, {2104, 8}, {2492, 8}, {2776, 8},
 655   {156, 8}, {544, 8}, {932, 8}, {1320, 8}, {1708, 8}, {2096, 8}, {2484, 8},
 656   {2768, 8}, {148, 8}, {536, 8}, {924, 8}, {1312, 8}, {1700, 8}, {2088, 8},
 657   {2476, 8}, {2760, 8}, {140, 8}, {528, 8}, {916, 8}, {1304, 8}, {1692, 8},
 658   {2080, 8}, {2468, 8}, {2752, 8}, {132, 8}, {520, 8}, {908, 8}, {1296, 8},
 659   {1684, 8}, {2072, 8}, {2460, 8}, {2744, 8}, {124, 8}, {512, 8}, {900, 8},
 660   {1288, 8}, {1676, 8}, {2064, 8}, {2452, 8}, {2736, 8}, {116, 8}, {504, 8},
 661   {892, 8}, {1280, 8}, {1668, 8}, {2056, 8}, {2444, 8}, {2728, 8}, {108, 8},
 662   {496, 8}, {884, 8}, {1272, 8}, {1660, 8}, {2048, 8}, {2436, 8}, {2720, 8},
 663   {100, 8}, {488, 8}, {876, 8}, {1264, 8}, {1652, 8}, {2040, 8}, {2428, 8},
 664   {2716, 4}, {92, 8}, {480, 8}, {868, 8}, {1256, 8}, {1644, 8}, {2032, 8},
 665   {2420, 8}, {84, 8}, {472, 8}, {860, 8}, {1248, 8}, {1636, 8}, {2024, 8},
 666   {2412, 8}, {76, 8}, {464, 8}, {852, 8}, {1240, 8}, {1628, 8}, {2016, 8},
 667   {2404, 8}, {68, 8}, {456, 8}, {844, 8}, {1232, 8}, {1620, 8}, {2008, 8},
 668   {2396, 8}, {60, 8}, {448, 8}, {836, 8}, {1224, 8}, {1612, 8}, {2000, 8},
 669   {2388, 8}, {52, 8}, {440, 8}, {828, 8}, {1216, 8}, {1604, 8}, {1992, 8},
 670   {2380, 8}, {44, 8}, {432, 8}, {820, 8}, {1208, 8}, {1596, 8}, {1984, 8},
 671   {2372, 8}, {36, 8}, {424, 8}, {812, 8}, {1200, 8}, {1588, 8}, {1976, 8},
 672   {2364, 8}, {28, 8}, {416, 8}, {804, 8}, {1192, 8}, {1580, 8}, {1968, 8},
 673   {2356, 8}, {20, 8}, {408, 8}, {796, 8}, {1184, 8}, {1572, 8}, {1960, 8},
 674   {2348, 8}, {12, 8}, {400, 8}, {788, 8}, {1176, 8}, {1564, 8}, {1952, 8},
 675   {2340, 8}, {4, 8}, {392, 8}, {780, 8}, {1168, 8}, {1556, 8}, {1944, 8},
 676   {2332, 8},
 677   /* missing from original data set */
 678   {388, 4}, {776, 4}, {1164, 4}, {1552, 4}, {1940, 4}, {2328, 4},
 679 };
 680 /* *INDENT-ON* */
 681
 682 int
 683 pattern_cmp (const void *arg1, const void *arg2)
 684 {
 685   test_pattern_t *a1 = (test_pattern_t *) arg1;
 686   test_pattern_t *a2 = (test_pattern_t *) arg2;
 687
 688   if (a1->offset < a2->offset)
 689     return -1;
 690   else if (a1->offset > a2->offset)
 691     return 1;
 692   return 0;
 693 }
 694
 695 static u8
 696 fifo_validate_pattern (vlib_main_t * vm, test_pattern_t * pattern,
 697                        u32 pattern_length)
 698 {
 699   test_pattern_t *tp = pattern;
 700   int i;
 701
 702   /* Go through the pattern and make 100% sure it's sane */
 703   for (i = 0; i < pattern_length - 1; i++)
 704     {
 705       if (tp->offset + tp->len != (tp + 1)->offset)
 706         {
 707           vlib_cli_output (vm, "[%d] missing {%d, %d}", i,
 708                            (tp->offset + tp->len),
 709                            (tp + 1)->offset - (tp->offset + tp->len));
 710           return 0;
 711         }
 712       tp++;
 713     }
 714   return 1;
 715 }
 716
 717 static test_pattern_t *
 718 fifo_get_validate_pattern (vlib_main_t * vm, test_pattern_t * test_data,
 719                            u32 test_data_len)
 720 {
 721   test_pattern_t *validate_pattern = 0;
 722
 723   /* Validate, and try segments in order... */
 724   vec_validate (validate_pattern, test_data_len - 1);
 725   memcpy (validate_pattern, test_data,
 726           test_data_len * sizeof (test_pattern_t));
 727   qsort ((u8 *) validate_pattern, test_data_len, sizeof (test_pattern_t),
 728          pattern_cmp);
 729
 730   if (fifo_validate_pattern (vm, validate_pattern, test_data_len) == 0)
 731     return 0;
 732
 733   return validate_pattern;
 734 }
 735
 736 static svm_fifo_t *
 737 fifo_prepare (u32 fifo_size)
 738 {
 739   svm_fifo_t *f;
 740   f = svm_fifo_create (fifo_size);
 741
 742   /* Paint fifo data vector with -1's */
 743   clib_memset (f->data, 0xFF, fifo_size);
 744
 745   return f;
 746 }
 747
 748 static int
 749 compare_data (u8 * data1, u8 * data2, u32 start, u32 len, u32 * index)
 750 {
 751   int i;
 752
 753   for (i = start; i < len; i++)
 754     {
 755       if (data1[i] != data2[i])
 756         {
 757           *index = i;
 758           return 1;
 759         }
 760     }
 761   return 0;
 762 }
 763
 764 int
 765 tcp_test_fifo1 (vlib_main_t * vm, unformat_input_t * input)
 766 {
 767   svm_fifo_t *f;
 768   u32 fifo_size = 1 << 20;
 769   u32 *test_data = 0;
 770   u32 offset;
 771   int i, rv, verbose = 0;
 772   u32 data_word, test_data_len, j;
 773   ooo_segment_t *ooo_seg;
 774   u8 *data, *s, *data_buf = 0;
 775
 776   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
 777     {
 778       if (unformat (input, "verbose"))
 779         verbose = 1;
 780     }
 781
 782   test_data_len = fifo_size / sizeof (u32);
 783   vec_validate (test_data, test_data_len - 1);
 784
 785   for (i = 0; i < vec_len (test_data); i++)
 786     test_data[i] = i;
 787
 788   f = fifo_prepare (fifo_size);
 789
 790   /*
 791    * Enqueue an initial (un-dequeued) chunk
 792    */
 793   rv = svm_fifo_enqueue_nowait (f, sizeof (u32), (u8 *) test_data);
 794   TCP_TEST ((rv == sizeof (u32)), "enqueued %d", rv);
 795   TCP_TEST ((f->tail == 4), "fifo tail %u", f->tail);
 796
 797   /*
 798    * Create 3 chunks in the future. The offsets are relative
 799    * to the current fifo tail
 800    */
 801   for (i = 0; i < 3; i++)
 802     {
 803       offset = (2 * i + 1) * sizeof (u32) - f->tail;
 804       data = (u8 *) (test_data + (2 * i + 1));
 805       if (i == 0)
 806         {
 807           rv = svm_fifo_enqueue_nowait (f, sizeof (u32), data);
 808           rv = rv > 0 ? 0 : rv;
 809         }
 810       else
 811         rv = svm_fifo_enqueue_with_offset (f, offset, sizeof (u32), data);
 812       if (verbose)
 813         vlib_cli_output (vm, "add [%d] [%d, %d]", 2 * i + 1, offset,
 814                          offset + sizeof (u32));
 815       if (rv)
 816         {
 817           clib_warning ("enqueue returned %d", rv);
 818           goto err;
 819         }
 820     }
 821
 822   if (verbose)
 823     vlib_cli_output (vm, "fifo after odd segs: %U", format_svm_fifo, f, 1);
 824
 825   TCP_TEST ((f->tail == 8), "fifo tail %u", f->tail);
 826   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 2),
 827             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 828
 829   /*
 830    * Try adding a completely overlapped segment
 831    */
 832   offset = 3 * sizeof (u32) - f->tail;
 833   data = (u8 *) (test_data + 3);
 834   rv = svm_fifo_enqueue_with_offset (f, offset, sizeof (u32), data);
 835   if (rv)
 836     {
 837       clib_warning ("enqueue returned %d", rv);
 838       goto err;
 839     }
 840
 841   if (verbose)
 842     vlib_cli_output (vm, "fifo after overlap seg: %U", format_svm_fifo, f, 1);
 843
 844   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 2),
 845             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 846
 847   /*
 848    * Make sure format functions are not buggy
 849    */
 850   s = format (0, "%U", format_svm_fifo, f, 2);
 851   vec_free (s);
 852
 853   /*
 854    * Paint some of missing data backwards
 855    */
 856   for (i = 3; i > 1; i--)
 857     {
 858       offset = (2 * i + 0) * sizeof (u32) - f->tail;
 859       data = (u8 *) (test_data + (2 * i + 0));
 860       rv = svm_fifo_enqueue_with_offset (f, offset, sizeof (u32), data);
 861       if (verbose)
 862         vlib_cli_output (vm, "add [%d] [%d, %d]", 2 * i, offset,
 863                          offset + sizeof (u32));
 864       if (rv)
 865         {
 866           clib_warning ("enqueue returned %d", rv);
 867           goto err;
 868         }
 869     }
 870
 871   if (verbose)
 872     vlib_cli_output (vm, "fifo before missing link: %U", format_svm_fifo, f,
 873                      1);
 874   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
 875             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 876   ooo_seg = svm_fifo_first_ooo_segment (f);
 877   TCP_TEST ((ooo_seg->start == 12),
 878             "first ooo seg position %u", ooo_seg->start);
 879   TCP_TEST ((ooo_seg->length == 16),
 880             "first ooo seg length %u", ooo_seg->length);
 881
 882   /*
 883    * Enqueue the missing u32
 884    */
 885   rv = svm_fifo_enqueue_nowait (f, sizeof (u32), (u8 *) (test_data + 2));
 886   if (verbose)
 887     vlib_cli_output (vm, "fifo after missing link: %U", format_svm_fifo, f,
 888                      1);
 889   TCP_TEST ((rv == 20), "bytes to be enqueued %u", rv);
 890   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 0),
 891             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 892
 893   /*
 894    * Collect results
 895    */
 896   for (i = 0; i < 7; i++)
 897     {
 898       rv = svm_fifo_dequeue_nowait (f, sizeof (u32), (u8 *) & data_word);
 899       if (rv != sizeof (u32))
 900         {
 901           clib_warning ("bytes dequeues %u", rv);
 902           goto err;
 903         }
 904       if (data_word != test_data[i])
 905         {
 906           clib_warning ("recovered [%d] %d not %d", i, data_word,
 907                         test_data[i]);
 908           goto err;
 909         }
 910     }
 911
 912   /*
 913    * Test segment overlaps: last ooo segment overlaps all
 914    */
 915   svm_fifo_free (f);
 916   f = fifo_prepare (fifo_size);
 917
 918   for (i = 0; i < 4; i++)
 919     {
 920       offset = (2 * i + 1) * sizeof (u32) - f->tail;
 921       data = (u8 *) (test_data + (2 * i + 1));
 922       rv = svm_fifo_enqueue_with_offset (f, offset, sizeof (u32), data);
 923       if (verbose)
 924         vlib_cli_output (vm, "add [%d] [%d, %d]", 2 * i + 1, offset,
 925                          offset + sizeof (u32));
 926       if (rv)
 927         {
 928           clib_warning ("enqueue returned %d", rv);
 929           goto err;
 930         }
 931     }
 932
 933   rv = svm_fifo_enqueue_with_offset (f, 8 - f->tail, 21, data);
 934   TCP_TEST ((rv == 0), "ooo enqueued %u", rv);
 935   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
 936             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 937
 938   vec_validate (data_buf, vec_len (data));
 939   svm_fifo_peek (f, 0, vec_len (data), data_buf);
 940   if (compare_data (data_buf, data, 8, vec_len (data), &j))
 941     {
 942       TCP_TEST (0, "[%d] peeked %u expected %u", j, data_buf[j], data[j]);
 943     }
 944   vec_reset_length (data_buf);
 945
 946   /*
 947    * Test segment overlaps: enqueue and overlap ooo segments
 948    */
 949   svm_fifo_free (f);
 950   f = fifo_prepare (fifo_size);
 951
 952   for (i = 0; i < 4; i++)
 953     {
 954       offset = (2 * i + 1) * sizeof (u32) - f->tail;
 955       data = (u8 *) (test_data + (2 * i + 1));
 956       rv = svm_fifo_enqueue_with_offset (f, offset, sizeof (u32), data);
 957       if (verbose)
 958         vlib_cli_output (vm, "add [%d] [%d, %d]", 2 * i + 1, offset,
 959                          offset + sizeof (u32));
 960       if (rv)
 961         {
 962           clib_warning ("enqueue returned %d", rv);
 963           goto err;
 964         }
 965     }
 966
 967   if (verbose)
 968     vlib_cli_output (vm, "fifo after enqueue: %U", format_svm_fifo, f, 1);
 969
 970   rv = svm_fifo_enqueue_nowait (f, 29, data);
 971   if (verbose)
 972     vlib_cli_output (vm, "fifo after enqueueing 29: %U", format_svm_fifo, f,
 973                      1);
 974   TCP_TEST ((rv == 32), "ooo enqueued %u", rv);
 975   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 0),
 976             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
 977
 978   vec_validate (data_buf, vec_len (data));
 979   svm_fifo_peek (f, 0, vec_len (data), data_buf);
 980   if (compare_data (data_buf, data, 0, vec_len (data), &j))
 981     {
 982       TCP_TEST (0, "[%d] peeked %u expected %u", j, data_buf[j], data[j]);
 983     }
 984
 985   /* Try to peek beyond the data */
 986   rv = svm_fifo_peek (f, svm_fifo_max_dequeue (f), vec_len (data), data_buf);
 987   TCP_TEST ((rv == 0), "peeked %u expected 0", rv);
 988
 989   vec_free (data_buf);
 990   svm_fifo_free (f);
 991   vec_free (test_data);
 992
 993   return 0;
 994
 995 err:
 996   svm_fifo_free (f);
 997   vec_free (test_data);
 998   return -1;
 999 }
1000
1001 static int
1002 tcp_test_fifo2 (vlib_main_t * vm)
1003 {
1004   svm_fifo_t *f;
1005   u32 fifo_size = 1 << 20;
1006   int i, rv, test_data_len;
1007   u64 data64;
1008   test_pattern_t *tp, *vp, *test_data;
1009   ooo_segment_t *ooo_seg;
1010
1011   test_data = test_pattern;
1012   test_data_len = ARRAY_LEN (test_pattern);
1013
1014   vp = fifo_get_validate_pattern (vm, test_data, test_data_len);
1015
1016   /* Create a fifo */
1017   f = fifo_prepare (fifo_size);
1018
1019   /*
1020    * Try with sorted data
1021    */
1022   for (i = 0; i < test_data_len; i++)
1023     {
1024       tp = vp + i;
1025       data64 = tp->offset;
1026       svm_fifo_enqueue_with_offset (f, tp->offset - f->tail, tp->len,
1027                                     (u8 *) & data64);
1028     }
1029
1030   /* Expected result: one big fat chunk at offset 4 */
1031   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
1032             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1033   ooo_seg = svm_fifo_first_ooo_segment (f);
1034   TCP_TEST ((ooo_seg->start == 4),
1035             "first ooo seg position %u", ooo_seg->start);
1036   TCP_TEST ((ooo_seg->length == 2996),
1037             "first ooo seg length %u", ooo_seg->length);
1038
1039   data64 = 0;
1040   rv = svm_fifo_enqueue_nowait (f, sizeof (u32), (u8 *) & data64);
1041   TCP_TEST ((rv == 3000), "bytes to be enqueued %u", rv);
1042
1043   svm_fifo_free (f);
1044   vec_free (vp);
1045
1046   /*
1047    * Now try it again w/ unsorted data...
1048    */
1049
1050   f = fifo_prepare (fifo_size);
1051
1052   for (i = 0; i < test_data_len; i++)
1053     {
1054       tp = &test_data[i];
1055       data64 = tp->offset;
1056       rv = svm_fifo_enqueue_with_offset (f, tp->offset - f->tail, tp->len,
1057                                          (u8 *) & data64);
1058       if (rv)
1059         {
1060           clib_warning ("enqueue returned %d", rv);
1061         }
1062     }
1063
1064   /* Expecting the same result: one big fat chunk at offset 4 */
1065   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
1066             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1067   ooo_seg = svm_fifo_first_ooo_segment (f);
1068   TCP_TEST ((ooo_seg->start == 4),
1069             "first ooo seg position %u", ooo_seg->start);
1070   TCP_TEST ((ooo_seg->length == 2996),
1071             "first ooo seg length %u", ooo_seg->length);
1072
1073   data64 = 0;
1074   rv = svm_fifo_enqueue_nowait (f, sizeof (u32), (u8 *) & data64);
1075
1076   TCP_TEST ((rv == 3000), "bytes to be enqueued %u", rv);
1077
1078   svm_fifo_free (f);
1079
1080   return 0;
1081 }
1082
1083 static int
1084 tcp_test_fifo3 (vlib_main_t * vm, unformat_input_t * input)
1085 {
1086   svm_fifo_t *f;
1087   u32 fifo_size = 4 << 10;
1088   u32 fifo_initial_offset = 0;
1089   u32 total_size = 2 << 10;
1090   int overlap = 0, verbose = 0, randomize = 1, drop = 0, in_seq_all = 0;
1091   u8 *data_pattern = 0, *data_buf = 0;
1092   test_pattern_t *tp, *generate = 0;
1093   u32 nsegs = 2, seg_size, length_so_far;
1094   u32 current_offset, offset_increment, len_this_chunk;
1095   u32 seed = 0xdeaddabe, j;
1096   int i, rv;
1097
1098   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1099     {
1100       if (unformat (input, "fifo-size %d", &fifo_size))
1101         ;
1102       else if (unformat (input, "total-size %d", &total_size))
1103         ;
1104       else if (unformat (input, "verbose"))
1105         verbose = 1;
1106       else if (unformat (input, "overlap"))
1107         overlap = 1;
1108       else if (unformat (input, "initial-offset %d", &fifo_initial_offset))
1109         ;
1110       else if (unformat (input, "seed %d", &seed))
1111         ;
1112       else if (unformat (input, "nsegs %d", &nsegs))
1113         ;
1114       else if (unformat (input, "no-randomize"))
1115         randomize = 0;
1116       else if (unformat (input, "in-seq-all"))
1117         in_seq_all = 1;
1118       else if (unformat (input, "drop"))
1119         drop = 1;
1120       else
1121         {
1122           clib_error_t *e = clib_error_return
1123             (0, "unknown input `%U'", format_unformat_error, input);
1124           clib_error_report (e);
1125           return -1;
1126         }
1127     }
1128
1129   if (total_size > fifo_size)
1130     {
1131       clib_warning ("total_size %d greater than fifo size %d", total_size,
1132                     fifo_size);
1133       return -1;
1134     }
1135   if (overlap && randomize == 0)
1136     {
1137       clib_warning ("Can't enqueue in-order with overlap");
1138       return -1;
1139     }
1140
1141   /*
1142    * Generate data
1143    */
1144   vec_validate (data_pattern, total_size - 1);
1145   for (i = 0; i < vec_len (data_pattern); i++)
1146     data_pattern[i] = i & 0xff;
1147
1148   /*
1149    * Generate segments
1150    */
1151   seg_size = total_size / nsegs;
1152   length_so_far = 0;
1153   current_offset = randomize;
1154   while (length_so_far < total_size)
1155     {
1156       vec_add2 (generate, tp, 1);
1157       len_this_chunk = clib_min (seg_size, total_size - length_so_far);
1158       tp->offset = current_offset;
1159       tp->len = len_this_chunk;
1160
1161       if (overlap && (len_this_chunk == seg_size))
1162         do
1163           {
1164             offset_increment = len_this_chunk
1165               % (1 + (random_u32 (&seed) % len_this_chunk));
1166           }
1167         while (offset_increment == 0);
1168       else
1169         offset_increment = len_this_chunk;
1170
1171       current_offset += offset_increment;
1172       length_so_far = tp->offset + tp->len;
1173     }
1174
1175   /*
1176    * Validate segment list. Only valid for non-overlap cases.
1177    */
1178   if (overlap == 0)
1179     fifo_validate_pattern (vm, generate, vec_len (generate));
1180
1181   if (verbose)
1182     {
1183       vlib_cli_output (vm, "raw data pattern:");
1184       for (i = 0; i < vec_len (generate); i++)
1185         {
1186           vlib_cli_output (vm, "[%d] offset %u len %u", i,
1187                            generate[i].offset, generate[i].len);
1188         }
1189     }
1190
1191   /* Randomize data pattern */
1192   if (randomize)
1193     {
1194       for (i = 0; i < vec_len (generate) / 2; i++)
1195         {
1196           u32 src_index, dst_index;
1197           test_pattern_t _tmp, *tmp = &_tmp;
1198
1199           src_index = random_u32 (&seed) % vec_len (generate);
1200           dst_index = random_u32 (&seed) % vec_len (generate);
1201
1202           tmp[0] = generate[dst_index];
1203           generate[dst_index] = generate[src_index];
1204           generate[src_index] = tmp[0];
1205         }
1206       if (verbose)
1207         {
1208           vlib_cli_output (vm, "randomized data pattern:");
1209           for (i = 0; i < vec_len (generate); i++)
1210             {
1211               vlib_cli_output (vm, "[%d] offset %u len %u", i,
1212                                generate[i].offset, generate[i].len);
1213             }
1214         }
1215     }
1216
1217   /*
1218    * Create a fifo and add segments
1219    */
1220   f = fifo_prepare (fifo_size);
1221
1222   /* manually set head and tail pointers to validate modular arithmetic */
1223   fifo_initial_offset = fifo_initial_offset % fifo_size;
1224   f->head = fifo_initial_offset;
1225   f->tail = fifo_initial_offset;
1226
1227   for (i = !randomize; i < vec_len (generate); i++)
1228     {
1229       tp = generate + i;
1230       svm_fifo_enqueue_with_offset (f,
1231                                     fifo_initial_offset + tp->offset -
1232                                     f->tail, tp->len,
1233                                     (u8 *) data_pattern + tp->offset);
1234     }
1235
1236   /* Add the first segment in order for non random data */
1237   if (!randomize)
1238     svm_fifo_enqueue_nowait (f, generate[0].len, (u8 *) data_pattern);
1239
1240   /*
1241    * Expected result: one big fat chunk at offset 1 if randomize == 1
1242    */
1243
1244   if (verbose)
1245     vlib_cli_output (vm, "fifo before missing link: %U",
1246                      format_svm_fifo, f, 1 /* verbose */ );
1247
1248   /*
1249    * Add the missing byte if segments were randomized
1250    */
1251   if (randomize)
1252     {
1253       u32 bytes_to_enq = 1;
1254       if (in_seq_all)
1255         bytes_to_enq = total_size;
1256       rv = svm_fifo_enqueue_nowait (f, bytes_to_enq, data_pattern + 0);
1257
1258       if (verbose)
1259         vlib_cli_output (vm, "in-order enqueue returned %d", rv);
1260
1261       TCP_TEST ((rv == total_size), "enqueued %u expected %u", rv,
1262                 total_size);
1263
1264     }
1265
1266   TCP_TEST ((svm_fifo_has_ooo_data (f) == 0), "number of ooo segments %u",
1267             svm_fifo_number_ooo_segments (f));
1268
1269   /*
1270    * Test if peeked data is the same as original data
1271    */
1272   vec_validate (data_buf, vec_len (data_pattern));
1273   svm_fifo_peek (f, 0, vec_len (data_pattern), data_buf);
1274   if (compare_data (data_buf, data_pattern, 0, vec_len (data_pattern), &j))
1275     {
1276       TCP_TEST (0, "[%d] peeked %u expected %u", j, data_buf[j],
1277                 data_pattern[j]);
1278     }
1279   vec_reset_length (data_buf);
1280
1281   /*
1282    * Dequeue or drop all data
1283    */
1284   if (drop)
1285     {
1286       svm_fifo_dequeue_drop (f, vec_len (data_pattern));
1287     }
1288   else
1289     {
1290       svm_fifo_dequeue_nowait (f, vec_len (data_pattern), data_buf);
1291       if (compare_data
1292           (data_buf, data_pattern, 0, vec_len (data_pattern), &j))
1293         {
1294           TCP_TEST (0, "[%d] dequeued %u expected %u", j, data_buf[j],
1295                     data_pattern[j]);
1296         }
1297     }
1298
1299   TCP_TEST ((svm_fifo_max_dequeue (f) == 0), "fifo has %d bytes",
1300             svm_fifo_max_dequeue (f));
1301
1302   svm_fifo_free (f);
1303   vec_free (data_pattern);
1304   vec_free (data_buf);
1305
1306   return 0;
1307 }
1308
1309 static int
1310 tcp_test_fifo4 (vlib_main_t * vm, unformat_input_t * input)
1311 {
1312   svm_fifo_t *f;
1313   u32 fifo_size = 6 << 10;
1314   u32 fifo_initial_offset = 1000000000;
1315   u32 test_n_bytes = 5000, j;
1316   u8 *test_data = 0, *data_buf = 0;
1317   int i, rv, verbose = 0;
1318
1319   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1320     {
1321       if (unformat (input, "verbose"))
1322         verbose = 1;
1323       else
1324         {
1325           clib_error_t *e = clib_error_return
1326             (0, "unknown input `%U'", format_unformat_error, input);
1327           clib_error_report (e);
1328           return -1;
1329         }
1330     }
1331
1332   /*
1333    * Create a fifo and add segments
1334    */
1335   f = fifo_prepare (fifo_size);
1336
1337   /* Set head and tail pointers */
1338   fifo_initial_offset = fifo_initial_offset % fifo_size;
1339   svm_fifo_init_pointers (f, fifo_initial_offset);
1340
1341   vec_validate (test_data, test_n_bytes - 1);
1342   for (i = 0; i < vec_len (test_data); i++)
1343     test_data[i] = i;
1344
1345   for (i = test_n_bytes - 1; i > 0; i--)
1346     {
1347       rv = svm_fifo_enqueue_with_offset (f, fifo_initial_offset + i - f->tail,
1348                                          sizeof (u8), &test_data[i]);
1349       if (verbose)
1350         vlib_cli_output (vm, "add [%d] [%d, %d]", i, i, i + sizeof (u8));
1351       if (rv)
1352         {
1353           clib_warning ("enqueue returned %d", rv);
1354           svm_fifo_free (f);
1355           vec_free (test_data);
1356           return -1;
1357         }
1358     }
1359
1360   svm_fifo_enqueue_nowait (f, sizeof (u8), &test_data[0]);
1361
1362   vec_validate (data_buf, vec_len (test_data));
1363
1364   svm_fifo_dequeue_nowait (f, vec_len (test_data), data_buf);
1365   rv = compare_data (data_buf, test_data, 0, vec_len (test_data), &j);
1366   if (rv)
1367     vlib_cli_output (vm, "[%d] dequeued %u expected %u", j, data_buf[j],
1368                      test_data[j]);
1369   TCP_TEST ((rv == 0), "dequeued compared to original returned %d", rv);
1370
1371   svm_fifo_free (f);
1372   vec_free (test_data);
1373   return 0;
1374 }
1375
1376 static u32
1377 fifo_pos (svm_fifo_t * f, u32 pos)
1378 {
1379   return pos % f->nitems;
1380 }
1381
1382 static int
1383 tcp_test_fifo5 (vlib_main_t * vm, unformat_input_t * input)
1384 {
1385   svm_fifo_t *f;
1386   u32 fifo_size = 400, j = 0, offset = 200;
1387   int i, rv, verbose = 0;
1388   u8 *test_data = 0, *data_buf = 0;
1389   ooo_segment_t *ooo_seg;
1390
1391   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1392     {
1393       if (unformat (input, "verbose"))
1394         verbose = 1;
1395       else
1396         {
1397           clib_error_t *e = clib_error_return
1398             (0, "unknown input `%U'", format_unformat_error, input);
1399           clib_error_report (e);
1400           return -1;
1401         }
1402     }
1403
1404   f = fifo_prepare (fifo_size);
1405   svm_fifo_init_pointers (f, offset);
1406
1407   vec_validate (test_data, 399);
1408   for (i = 0; i < vec_len (test_data); i++)
1409     test_data[i] = i % 0xff;
1410
1411   /*
1412    * Start with [100, 200] and [300, 400]
1413    */
1414   svm_fifo_enqueue_with_offset (f, 100, 100, &test_data[100]);
1415   svm_fifo_enqueue_with_offset (f, 300, 100, &test_data[300]);
1416
1417   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 2),
1418             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1419   TCP_TEST ((f->ooos_newest == 1), "newest %u", f->ooos_newest);
1420   if (verbose)
1421     vlib_cli_output (vm, "fifo after [100, 200] and [300, 400] : %U",
1422                      format_svm_fifo, f, 2 /* verbose */ );
1423
1424   /*
1425    * Add [225, 275]
1426    */
1427
1428   rv = svm_fifo_enqueue_with_offset (f, 225, 50, &test_data[200]);
1429   if (verbose)
1430     vlib_cli_output (vm, "fifo after [225, 275] : %U",
1431                      format_svm_fifo, f, 2 /* verbose */ );
1432   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 3),
1433             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1434   ooo_seg = svm_fifo_first_ooo_segment (f);
1435   TCP_TEST ((ooo_seg->start == fifo_pos (f, 100 + offset)),
1436             "first seg start %u expected %u", ooo_seg->start,
1437             fifo_pos (f, 100 + offset));
1438   TCP_TEST ((ooo_seg->length == 100), "first seg length %u expected %u",
1439             ooo_seg->length, 100);
1440   ooo_seg = ooo_segment_next (f, ooo_seg);
1441   TCP_TEST ((ooo_seg->start == fifo_pos (f, 225 + offset)),
1442             "second seg start %u expected %u",
1443             ooo_seg->start, fifo_pos (f, 225 + offset));
1444   TCP_TEST ((ooo_seg->length == 50), "second seg length %u expected %u",
1445             ooo_seg->length, 50);
1446   ooo_seg = ooo_segment_next (f, ooo_seg);
1447   TCP_TEST ((ooo_seg->start == fifo_pos (f, 300 + offset)),
1448             "third seg start %u expected %u",
1449             ooo_seg->start, fifo_pos (f, 300 + offset));
1450   TCP_TEST ((ooo_seg->length == 100), "third seg length %u expected %u",
1451             ooo_seg->length, 100);
1452   TCP_TEST ((f->ooos_newest == 2), "newest %u", f->ooos_newest);
1453   /*
1454    * Add [190, 310]
1455    */
1456   rv = svm_fifo_enqueue_with_offset (f, 190, 120, &test_data[190]);
1457   if (verbose)
1458     vlib_cli_output (vm, "fifo after [190, 310] : %U",
1459                      format_svm_fifo, f, 1 /* verbose */ );
1460   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
1461             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1462   ooo_seg = svm_fifo_first_ooo_segment (f);
1463   TCP_TEST ((ooo_seg->start == fifo_pos (f, offset + 100)),
1464             "first seg start %u expected %u",
1465             ooo_seg->start, fifo_pos (f, offset + 100));
1466   TCP_TEST ((ooo_seg->length == 300), "first seg length %u expected %u",
1467             ooo_seg->length, 300);
1468
1469   /*
1470    * Add [0, 150]
1471    */
1472   rv = svm_fifo_enqueue_nowait (f, 150, test_data);
1473
1474   if (verbose)
1475     vlib_cli_output (vm, "fifo after [0 150] : %U", format_svm_fifo, f,
1476                      2 /* verbose */ );
1477
1478   TCP_TEST ((rv == 400), "managed to enqueue %u expected %u", rv, 400);
1479   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 0),
1480             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1481
1482   vec_validate (data_buf, 399);
1483   svm_fifo_peek (f, 0, 400, data_buf);
1484   if (compare_data (data_buf, test_data, 0, 400, &j))
1485     {
1486       TCP_TEST (0, "[%d] peeked %u expected %u", j, data_buf[j],
1487                 test_data[j]);
1488     }
1489
1490   /*
1491    * Add [100 200] and overlap it with [50 250]
1492    */
1493   svm_fifo_free (f);
1494   f = fifo_prepare (fifo_size);
1495
1496   svm_fifo_enqueue_with_offset (f, 100, 100, &test_data[100]);
1497   svm_fifo_enqueue_with_offset (f, 50, 200, &test_data[50]);
1498   TCP_TEST ((svm_fifo_number_ooo_segments (f) == 1),
1499             "number of ooo segments %u", svm_fifo_number_ooo_segments (f));
1500   ooo_seg = svm_fifo_first_ooo_segment (f);
1501   TCP_TEST ((ooo_seg->start == 50), "first seg start %u expected %u",
1502             ooo_seg->start, 50);
1503   TCP_TEST ((ooo_seg->length == 200), "first seg length %u expected %u",
1504             ooo_seg->length, 200);
1505
1506   svm_fifo_free (f);
1507   vec_free (test_data);
1508   return 0;
1509 }
1510
1511 /* *INDENT-OFF* */
1512 svm_fifo_trace_elem_t fifo_trace[] = {};
1513 /* *INDENT-ON* */
1514
1515 static int
1516 tcp_test_fifo_replay (vlib_main_t * vm, unformat_input_t * input)
1517 {
1518   svm_fifo_t f;
1519   int verbose = 0;
1520   u8 no_read = 0, *str = 0;
1521
1522   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1523     {
1524       if (unformat (input, "verbose"))
1525         verbose = 1;
1526       else if (unformat (input, "no-read"))
1527         no_read = 1;
1528       else
1529         {
1530           clib_error_t *e = clib_error_return
1531             (0, "unknown input `%U'", format_unformat_error, input);
1532           clib_error_report (e);
1533           return -1;
1534         }
1535     }
1536
1537 #if SVMF_FIFO_TRACE
1538   f.trace = fifo_trace;
1539 #endif
1540
1541   str = svm_fifo_replay (str, &f, no_read, verbose);
1542   vlib_cli_output (vm, "%v", str);
1543   return 0;
1544 }
1545
1546 static int
1547 tcp_test_fifo (vlib_main_t * vm, unformat_input_t * input)
1548 {
1549   int res = 0;
1550   char *str;
1551
1552   /* Run all tests */
1553   if (unformat_check_input (input) == UNFORMAT_END_OF_INPUT)
1554     {
1555       res = tcp_test_fifo1 (vm, input);
1556       if (res)
1557         return res;
1558
1559       res = tcp_test_fifo2 (vm);
1560       if (res)
1561         return res;
1562
1563       /*
1564        * Run a number of fifo3 configs
1565        */
1566       str = "nsegs 10 overlap seed 123";
1567       unformat_init_cstring (input, str);
1568       if (tcp_test_fifo3 (vm, input))
1569         return -1;
1570       unformat_free (input);
1571
1572       str = "nsegs 10 overlap seed 123 in-seq-all";
1573       unformat_init_cstring (input, str);
1574       if (tcp_test_fifo3 (vm, input))
1575         return -1;
1576       unformat_free (input);
1577
1578       str = "nsegs 10 overlap seed 123 initial-offset 3917";
1579       unformat_init_cstring (input, str);
1580       if (tcp_test_fifo3 (vm, input))
1581         return -1;
1582       unformat_free (input);
1583
1584       str = "nsegs 10 overlap seed 123 initial-offset 3917 drop";
1585       unformat_init_cstring (input, str);
1586       if (tcp_test_fifo3 (vm, input))
1587         return -1;
1588       unformat_free (input);
1589
1590       str = "nsegs 10 seed 123 initial-offset 3917 drop no-randomize";
1591       unformat_init_cstring (input, str);
1592       if (tcp_test_fifo3 (vm, input))
1593         return -1;
1594       unformat_free (input);
1595
1596       res = tcp_test_fifo4 (vm, input);
1597       if (res)
1598         return res;
1599
1600       res = tcp_test_fifo5 (vm, input);
1601       if (res)
1602         return res;
1603     }
1604   else
1605     {
1606       if (unformat (input, "fifo3"))
1607         {
1608           res = tcp_test_fifo3 (vm, input);
1609         }
1610       else if (unformat (input, "fifo2"))
1611         {
1612           res = tcp_test_fifo2 (vm);
1613         }
1614       else if (unformat (input, "fifo1"))
1615         {
1616           res = tcp_test_fifo1 (vm, input);
1617         }
1618       else if (unformat (input, "fifo4"))
1619         {
1620           res = tcp_test_fifo4 (vm, input);
1621         }
1622       else if (unformat (input, "fifo5"))
1623         {
1624           res = tcp_test_fifo5 (vm, input);
1625         }
1626       else if (unformat (input, "replay"))
1627         {
1628           res = tcp_test_fifo_replay (vm, input);
1629         }
1630     }
1631
1632   return res;
1633 }
1634
1635 static int
1636 tcp_test_lookup (vlib_main_t * vm, unformat_input_t * input)
1637 {
1638   session_manager_main_t *smm = &session_manager_main;
1639   tcp_main_t *tm = &tcp_main;
1640   transport_connection_t _tc1, *tc1 = &_tc1, _tc2, *tc2 = &_tc2, *tconn;
1641   tcp_connection_t *tc;
1642   stream_session_t *s, *s1;
1643   u8 cmp = 0, is_filtered = 0;
1644   u32 sidx;
1645
1646   /*
1647    * Allocate fake session and connection 1
1648    */
1649   pool_get (smm->sessions[0], s);
1650   clib_memset (s, 0, sizeof (*s));
1651   s->session_index = sidx = s - smm->sessions[0];
1652
1653   pool_get (tm->connections[0], tc);
1654   clib_memset (tc, 0, sizeof (*tc));
1655   tc->connection.c_index = tc - tm->connections[0];
1656   tc->connection.s_index = s->session_index;
1657   s->connection_index = tc->connection.c_index;
1658
1659   tc->connection.lcl_ip.ip4.as_u32 = clib_host_to_net_u32 (0x06000101);
1660   tc->connection.rmt_ip.ip4.as_u32 = clib_host_to_net_u32 (0x06000103);
1661   tc->connection.lcl_port = 35051;
1662   tc->connection.rmt_port = 53764;
1663   tc->connection.proto = TRANSPORT_PROTO_TCP;
1664   tc->connection.is_ip4 = 1;
1665   clib_memcpy (tc1, &tc->connection, sizeof (*tc1));
1666
1667   /*
1668    * Allocate fake session and connection 2
1669    */
1670   pool_get (session_manager_main.sessions[0], s);
1671   clib_memset (s, 0, sizeof (*s));
1672   s->session_index = s - smm->sessions[0];
1673
1674   pool_get (tm->connections[0], tc);
1675   clib_memset (tc, 0, sizeof (*tc));
1676   tc->connection.c_index = tc - tm->connections[0];
1677   tc->connection.s_index = s->session_index;
1678   s->connection_index = tc->connection.c_index;
1679
1680   tc->connection.lcl_ip.ip4.as_u32 = clib_host_to_net_u32 (0x06000101);
1681   tc->connection.rmt_ip.ip4.as_u32 = clib_host_to_net_u32 (0x06000102);
1682   tc->connection.lcl_port = 38225;
1683   tc->connection.rmt_port = 53764;
1684   tc->connection.proto = TRANSPORT_PROTO_TCP;
1685   tc->connection.is_ip4 = 1;
1686   clib_memcpy (tc2, &tc->connection, sizeof (*tc2));
1687
1688   /*
1689    * Confirm that connection lookup works
1690    */
1691
1692   s1 = pool_elt_at_index (smm->sessions[0], sidx);
1693   session_lookup_add_connection (tc1, session_handle (s1));
1694   tconn = session_lookup_connection_wt4 (0, &tc1->lcl_ip.ip4,
1695                                          &tc1->rmt_ip.ip4,
1696                                          tc1->lcl_port, tc1->rmt_port,
1697                                          tc1->proto, 0, &is_filtered);
1698
1699   TCP_TEST ((tconn != 0), "connection exists");
1700   cmp = (memcmp (&tconn->rmt_ip, &tc1->rmt_ip, sizeof (tc1->rmt_ip)) == 0);
1701   TCP_TEST ((cmp), "rmt ip is identical %d", cmp);
1702   TCP_TEST ((tconn->lcl_port == tc1->lcl_port),
1703             "rmt port is identical %d", tconn->lcl_port == tc1->lcl_port);
1704
1705   /*
1706    * Non-existing connection lookup should not work
1707    */
1708
1709   tconn = session_lookup_connection_wt4 (0, &tc2->lcl_ip.ip4,
1710                                          &tc2->rmt_ip.ip4,
1711                                          tc2->lcl_port, tc2->rmt_port,
1712                                          tc2->proto, 0, &is_filtered);
1713   TCP_TEST ((tconn == 0), "lookup result should be null");
1714
1715   /*
1716    * Delete and lookup again
1717    */
1718   session_lookup_del_connection (tc1);
1719   tconn = session_lookup_connection_wt4 (0, &tc1->lcl_ip.ip4,
1720                                          &tc1->rmt_ip.ip4,
1721                                          tc1->lcl_port, tc1->rmt_port,
1722                                          tc1->proto, 0, &is_filtered);
1723   TCP_TEST ((tconn == 0), "lookup result should be null");
1724   tconn = session_lookup_connection_wt4 (0, &tc2->lcl_ip.ip4,
1725                                          &tc2->rmt_ip.ip4,
1726                                          tc2->lcl_port, tc2->rmt_port,
1727                                          tc2->proto, 0, &is_filtered);
1728   TCP_TEST ((tconn == 0), "lookup result should be null");
1729
1730   /*
1731    * Re-add and lookup tc2
1732    */
1733   session_lookup_add_connection (tc1, tc1->s_index);
1734   tconn = session_lookup_connection_wt4 (0, &tc2->lcl_ip.ip4,
1735                                          &tc2->rmt_ip.ip4,
1736                                          tc2->lcl_port, tc2->rmt_port,
1737                                          tc2->proto, 0, &is_filtered);
1738   TCP_TEST ((tconn == 0), "lookup result should be null");
1739
1740   return 0;
1741 }
1742
1743 static int
1744 tcp_test_session (vlib_main_t * vm, unformat_input_t * input)
1745 {
1746   int rv = 0;
1747   tcp_connection_t *tc0;
1748   ip4_address_t local, remote;
1749   u16 local_port, remote_port;
1750   tcp_main_t *tm = vnet_get_tcp_main ();
1751   int is_add = 1;
1752
1753
1754   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1755     {
1756       if (unformat (input, "del"))
1757         is_add = 0;
1758       else if (unformat (input, "add"))
1759         is_add = 1;
1760       else
1761         break;
1762     }
1763
1764   if (is_add)
1765     {
1766       local.as_u32 = clib_host_to_net_u32 (0x06000101);
1767       remote.as_u32 = clib_host_to_net_u32 (0x06000102);
1768       local_port = clib_host_to_net_u16 (1234);
1769       remote_port = clib_host_to_net_u16 (11234);
1770
1771       pool_get (tm->connections[0], tc0);
1772       clib_memset (tc0, 0, sizeof (*tc0));
1773
1774       tc0->state = TCP_STATE_ESTABLISHED;
1775       tc0->rcv_las = 1;
1776       tc0->c_c_index = tc0 - tm->connections[0];
1777       tc0->c_lcl_port = local_port;
1778       tc0->c_rmt_port = remote_port;
1779       tc0->c_is_ip4 = 1;
1780       tc0->c_thread_index = 0;
1781       tc0->c_lcl_ip4.as_u32 = local.as_u32;
1782       tc0->c_rmt_ip4.as_u32 = remote.as_u32;
1783       tc0->rcv_opts.mss = 1450;
1784       tcp_connection_init_vars (tc0);
1785
1786       TCP_EVT_DBG (TCP_EVT_OPEN, tc0);
1787
1788       if (stream_session_accept (&tc0->connection, 0 /* listener index */ ,
1789                                  0 /* notify */ ))
1790         clib_warning ("stream_session_accept failed");
1791
1792       stream_session_accept_notify (&tc0->connection);
1793     }
1794   else
1795     {
1796       tc0 = tcp_connection_get (0 /* connection index */ , 0 /* thread */ );
1797       tc0->state = TCP_STATE_CLOSED;
1798       stream_session_disconnect_notify (&tc0->connection);
1799     }
1800
1801   return rv;
1802 }
1803
1804 static clib_error_t *
1805 tcp_test (vlib_main_t * vm,
1806           unformat_input_t * input, vlib_cli_command_t * cmd_arg)
1807 {
1808   int res = 0;
1809
1810   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1811     {
1812       if (unformat (input, "sack"))
1813         {
1814           res = tcp_test_sack (vm, input);
1815         }
1816       else if (unformat (input, "fifo"))
1817         {
1818           res = tcp_test_fifo (vm, input);
1819         }
1820       else if (unformat (input, "session"))
1821         {
1822           res = tcp_test_session (vm, input);
1823         }
1824       else if (unformat (input, "lookup"))
1825         {
1826           res = tcp_test_lookup (vm, input);
1827         }
1828       else if (unformat (input, "all"))
1829         {
1830           if ((res = tcp_test_sack (vm, input)))
1831             goto done;
1832           if ((res = tcp_test_fifo (vm, input)))
1833             goto done;
1834           if ((res = tcp_test_lookup (vm, input)))
1835             goto done;
1836         }
1837       else
1838         break;
1839     }
1840
1841 done:
1842   if (res)
1843     return clib_error_return (0, "TCP unit test failed");
1844   return 0;
1845 }
1846
1847 /* *INDENT-OFF* */
1848 VLIB_CLI_COMMAND (tcp_test_command, static) =
1849 {
1850   .path = "test tcp",
1851   .short_help = "internal tcp unit tests",
1852   .function = tcp_test,
1853 };
1854 /* *INDENT-ON* */
1855
1856 /*
1857  * fd.io coding-style-patch-verification: ON
1858  *
1859  * Local Variables:
1860  * eval: (c-set-style "gnu")
1861  * End:
1862  */