src/tests/vnet/session/udp_echo.c

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <stdio.h>
  17 #include <setjmp.h>
  18 #include <signal.h>
  19 #include <vppinfra/clib.h>
  20 #include <vppinfra/format.h>
  21 #include <vppinfra/error.h>
  22 #include <vppinfra/time.h>
  23 #include <vppinfra/macros.h>
  24 #include <vnet/vnet.h>
  25 #include <vlib/vlib.h>
  26 #include <vlib/unix/unix.h>
  27 #include <vlibapi/api.h>
  28 #include <vlibmemory/api.h>
  29 #include <vpp/api/vpe_msg_enum.h>
  30 #include <svm/svm_fifo_segment.h>
  31 #include <pthread.h>
  32 #include <vnet/session/application_interface.h>
  33
  34 #define vl_typedefs             /* define message structures */
  35 #include <vpp/api/vpe_all_api_h.h>
  36 #undef vl_typedefs
  37
  38 /* declare message handlers for each api */
  39
  40 #define vl_endianfun            /* define message structures */
  41 #include <vpp/api/vpe_all_api_h.h>
  42 #undef vl_endianfun
  43
  44 /* instantiate all the print functions we know about */
  45 #define vl_print(handle, ...)
  46 #define vl_printfun
  47 #include <vpp/api/vpe_all_api_h.h>
  48 #undef vl_printfun
  49
  50 typedef enum
  51 {
  52   STATE_START,
  53   STATE_ATTACHED,
  54   STATE_BOUND,
  55   STATE_READY,
  56   STATE_FAILED,
  57   STATE_DISCONNECTING,
  58   STATE_DETACHED
  59 } connection_state_t;
  60
  61 typedef struct
  62 {
  63   /* vpe input queue */
  64   svm_queue_t *vl_input_queue;
  65
  66   /* API client handle */
  67   u32 my_client_index;
  68
  69   /* The URI we're playing with */
  70   u8 *listen_uri;
  71
  72   /* URI for connect */
  73   u8 *connect_uri;
  74
  75   /* Session pool */
  76   app_session_t *sessions;
  77
  78   /* Hash table for disconnect processing */
  79   uword *session_index_by_vpp_handles;
  80
  81   /* fifo segment */
  82   svm_fifo_segment_private_t *seg;
  83
  84   /* intermediate rx buffer */
  85   u8 *rx_buf;
  86
  87   u32 fifo_size;
  88   int i_am_server;
  89   u8 is_connected;
  90
  91   /* Our event queue */
  92   svm_msg_q_t *our_event_queue;
  93   svm_msg_q_t *ct_event_queue;
  94
  95   /* $$$ single thread only for the moment */
  96   svm_msg_q_t *vpp_event_queue;
  97
  98   /* $$$$ hack: cut-through session index */
  99   volatile u32 cut_through_session_index;
 100   volatile u32 connected_session;
 101
 102   /* unique segment name counter */
 103   u32 unique_segment_index;
 104
 105   pid_t my_pid;
 106
 107   /* pthread handle */
 108   pthread_t cut_through_thread_handle;
 109
 110   /* For deadman timers */
 111   clib_time_t clib_time;
 112
 113   /* State of the connection, shared between msg RX thread and main thread */
 114   volatile connection_state_t state;
 115
 116   volatile int time_to_stop;
 117   volatile int time_to_print_stats;
 118
 119   u32 configured_segment_size;
 120
 121   /* VNET_API_ERROR_FOO -> "Foo" hash table */
 122   uword *error_string_by_error_number;
 123
 124   svm_fifo_segment_main_t segment_main;
 125
 126   u8 *connect_test_data;
 127
 128   uword *segments_table;
 129   u8 do_echo;
 130   u8 have_return;
 131   u64 total_to_send;
 132   u64 bytes_to_send;
 133   u64 bytes_sent;
 134 } udp_echo_main_t;
 135
 136 udp_echo_main_t udp_echo_main;
 137
 138 static void
 139 stop_signal (int signum)
 140 {
 141   udp_echo_main_t *um = &udp_echo_main;
 142
 143   um->time_to_stop = 1;
 144 }
 145
 146 static void
 147 stats_signal (int signum)
 148 {
 149   udp_echo_main_t *um = &udp_echo_main;
 150   um->time_to_print_stats = 1;
 151 }
 152
 153 static clib_error_t *
 154 setup_signal_handlers (void)
 155 {
 156   signal (SIGINT, stats_signal);
 157   signal (SIGQUIT, stop_signal);
 158   signal (SIGTERM, stop_signal);
 159
 160   return 0;
 161 }
 162
 163 uword
 164 unformat_ip4_address (unformat_input_t * input, va_list * args)
 165 {
 166   u8 *result = va_arg (*args, u8 *);
 167   unsigned a[4];
 168
 169   if (!unformat (input, "%d.%d.%d.%d", &a[0], &a[1], &a[2], &a[3]))
 170     return 0;
 171
 172   if (a[0] >= 256 || a[1] >= 256 || a[2] >= 256 || a[3] >= 256)
 173     return 0;
 174
 175   result[0] = a[0];
 176   result[1] = a[1];
 177   result[2] = a[2];
 178   result[3] = a[3];
 179
 180   return 1;
 181 }
 182
 183 uword
 184 unformat_ip6_address (unformat_input_t * input, va_list * args)
 185 {
 186   ip6_address_t *result = va_arg (*args, ip6_address_t *);
 187   u16 hex_quads[8];
 188   uword hex_quad, n_hex_quads, hex_digit, n_hex_digits;
 189   uword c, n_colon, double_colon_index;
 190
 191   n_hex_quads = hex_quad = n_hex_digits = n_colon = 0;
 192   double_colon_index = ARRAY_LEN (hex_quads);
 193   while ((c = unformat_get_input (input)) != UNFORMAT_END_OF_INPUT)
 194     {
 195       hex_digit = 16;
 196       if (c >= '0' && c <= '9')
 197         hex_digit = c - '0';
 198       else if (c >= 'a' && c <= 'f')
 199         hex_digit = c + 10 - 'a';
 200       else if (c >= 'A' && c <= 'F')
 201         hex_digit = c + 10 - 'A';
 202       else if (c == ':' && n_colon < 2)
 203         n_colon++;
 204       else
 205         {
 206           unformat_put_input (input);
 207           break;
 208         }
 209
 210       /* Too many hex quads. */
 211       if (n_hex_quads >= ARRAY_LEN (hex_quads))
 212         return 0;
 213
 214       if (hex_digit < 16)
 215         {
 216           hex_quad = (hex_quad << 4) | hex_digit;
 217
 218           /* Hex quad must fit in 16 bits. */
 219           if (n_hex_digits >= 4)
 220             return 0;
 221
 222           n_colon = 0;
 223           n_hex_digits++;
 224         }
 225
 226       /* Save position of :: */
 227       if (n_colon == 2)
 228         {
 229           /* More than one :: ? */
 230           if (double_colon_index < ARRAY_LEN (hex_quads))
 231             return 0;
 232           double_colon_index = n_hex_quads;
 233         }
 234
 235       if (n_colon > 0 && n_hex_digits > 0)
 236         {
 237           hex_quads[n_hex_quads++] = hex_quad;
 238           hex_quad = 0;
 239           n_hex_digits = 0;
 240         }
 241     }
 242
 243   if (n_hex_digits > 0)
 244     hex_quads[n_hex_quads++] = hex_quad;
 245
 246   {
 247     word i;
 248
 249     /* Expand :: to appropriate number of zero hex quads. */
 250     if (double_colon_index < ARRAY_LEN (hex_quads))
 251       {
 252         word n_zero = ARRAY_LEN (hex_quads) - n_hex_quads;
 253
 254         for (i = n_hex_quads - 1; i >= (signed) double_colon_index; i--)
 255           hex_quads[n_zero + i] = hex_quads[i];
 256
 257         for (i = 0; i < n_zero; i++)
 258           hex_quads[double_colon_index + i] = 0;
 259
 260         n_hex_quads = ARRAY_LEN (hex_quads);
 261       }
 262
 263     /* Too few hex quads given. */
 264     if (n_hex_quads < ARRAY_LEN (hex_quads))
 265       return 0;
 266
 267     for (i = 0; i < ARRAY_LEN (hex_quads); i++)
 268       result->as_u16[i] = clib_host_to_net_u16 (hex_quads[i]);
 269
 270     return 1;
 271   }
 272 }
 273
 274 uword
 275 unformat_uri (unformat_input_t * input, va_list * args)
 276 {
 277   session_endpoint_cfg_t *sep = va_arg (*args, session_endpoint_cfg_t *);
 278   u32 port;
 279   char *tmp;
 280
 281   if (unformat (input, "%s://%U/%d", &tmp, unformat_ip4_address, &sep->ip.ip4,
 282                 &port))
 283     {
 284       sep->port = clib_host_to_net_u16 (port);
 285       sep->is_ip4 = 1;
 286       return 1;
 287     }
 288   else if (unformat (input, "%s://%U/%d", &tmp, unformat_ip6_address,
 289                      &sep->ip.ip6, &port))
 290     {
 291       sep->port = clib_host_to_net_u16 (port);
 292       sep->is_ip4 = 0;
 293       return 1;
 294     }
 295   return 0;
 296 }
 297
 298 static void
 299 application_send_attach (udp_echo_main_t * utm)
 300 {
 301   vl_api_application_attach_t *bmp;
 302   bmp = vl_msg_api_alloc (sizeof (*bmp));
 303   clib_memset (bmp, 0, sizeof (*bmp));
 304
 305   bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_ATTACH);
 306   bmp->client_index = utm->my_client_index;
 307   bmp->context = ntohl (0xfeedface);
 308   bmp->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_ADD_SEGMENT;
 309   bmp->options[APP_OPTIONS_FLAGS] |= APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE;
 310   bmp->options[APP_OPTIONS_FLAGS] |= APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE;
 311   bmp->options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = 2;
 312   bmp->options[APP_OPTIONS_RX_FIFO_SIZE] = utm->fifo_size;
 313   bmp->options[APP_OPTIONS_TX_FIFO_SIZE] = utm->fifo_size;
 314   bmp->options[APP_OPTIONS_ADD_SEGMENT_SIZE] = 128 << 20;
 315   bmp->options[APP_OPTIONS_SEGMENT_SIZE] = 256 << 20;
 316   bmp->options[APP_OPTIONS_EVT_QUEUE_SIZE] = 16768;
 317   vl_msg_api_send_shmem (utm->vl_input_queue, (u8 *) & bmp);
 318 }
 319
 320 void
 321 application_detach (udp_echo_main_t * utm)
 322 {
 323   vl_api_application_detach_t *bmp;
 324   bmp = vl_msg_api_alloc (sizeof (*bmp));
 325   clib_memset (bmp, 0, sizeof (*bmp));
 326
 327   bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_DETACH);
 328   bmp->client_index = utm->my_client_index;
 329   bmp->context = ntohl (0xfeedface);
 330   vl_msg_api_send_shmem (utm->vl_input_queue, (u8 *) & bmp);
 331 }
 332
 333 static void
 334 vl_api_application_attach_reply_t_handler (vl_api_application_attach_reply_t *
 335                                            mp)
 336 {
 337   svm_fifo_segment_create_args_t _a = { 0 }, *a = &_a;
 338   udp_echo_main_t *utm = &udp_echo_main;
 339   svm_fifo_segment_main_t *sm = &utm->segment_main;
 340   int rv;
 341
 342   if (mp->retval)
 343     {
 344       clib_warning ("attach failed: %d", mp->retval);
 345       utm->state = STATE_FAILED;
 346       return;
 347     }
 348
 349   if (mp->segment_name_length == 0)
 350     {
 351       clib_warning ("segment_name_length zero");
 352       return;
 353     }
 354
 355   a->segment_name = (char *) mp->segment_name;
 356   a->segment_size = mp->segment_size;
 357
 358   ASSERT (mp->app_event_queue_address);
 359
 360   /* Attach to the segment vpp created */
 361   rv = svm_fifo_segment_attach (sm, a);
 362   if (rv)
 363     {
 364       clib_warning ("svm_fifo_segment_attach ('%s') failed",
 365                     mp->segment_name);
 366       return;
 367     }
 368
 369   utm->our_event_queue = uword_to_pointer (mp->app_event_queue_address,
 370                                            svm_msg_q_t *);
 371   utm->state = STATE_ATTACHED;
 372 }
 373
 374 static void
 375 vl_api_application_detach_reply_t_handler (vl_api_application_detach_reply_t *
 376                                            mp)
 377 {
 378   if (mp->retval)
 379     clib_warning ("detach returned with err: %d", mp->retval);
 380   udp_echo_main.state = STATE_DETACHED;
 381 }
 382
 383 u8 *
 384 format_api_error (u8 * s, va_list * args)
 385 {
 386   udp_echo_main_t *utm = va_arg (*args, udp_echo_main_t *);
 387   i32 error = va_arg (*args, u32);
 388   uword *p;
 389
 390   p = hash_get (utm->error_string_by_error_number, -error);
 391
 392   if (p)
 393     s = format (s, "%s", p[0]);
 394   else
 395     s = format (s, "%d", error);
 396   return s;
 397 }
 398
 399 int
 400 wait_for_state_change (udp_echo_main_t * utm, connection_state_t state)
 401 {
 402 #if CLIB_DEBUG > 0
 403 #define TIMEOUT 600.0
 404 #else
 405 #define TIMEOUT 600.0
 406 #endif
 407
 408   f64 timeout = clib_time_now (&utm->clib_time) + TIMEOUT;
 409
 410   while (clib_time_now (&utm->clib_time) < timeout)
 411     {
 412       if (utm->state == state)
 413         return 0;
 414       if (utm->state == STATE_FAILED)
 415         return -1;
 416     }
 417   return -1;
 418 }
 419
 420 u64 server_bytes_received, server_bytes_sent;
 421
 422 static void *
 423 cut_through_thread_fn (void *arg)
 424 {
 425   app_session_t *s;
 426   svm_fifo_t *rx_fifo;
 427   svm_fifo_t *tx_fifo;
 428   u8 *my_copy_buffer = 0;
 429   udp_echo_main_t *utm = &udp_echo_main;
 430   i32 actual_transfer;
 431   int rv, do_dequeue = 0;
 432   u32 buffer_offset;
 433
 434   while (utm->cut_through_session_index == ~0)
 435     ;
 436
 437   s = pool_elt_at_index (utm->sessions, utm->cut_through_session_index);
 438
 439   rx_fifo = s->rx_fifo;
 440   tx_fifo = s->tx_fifo;
 441
 442   vec_validate (my_copy_buffer, 64 * 1024 - 1);
 443
 444   while (1)
 445     {
 446       do
 447         {
 448           /* We read from the tx fifo and write to the rx fifo */
 449           if (utm->have_return || do_dequeue)
 450             actual_transfer = svm_fifo_dequeue_nowait (rx_fifo,
 451                                                        vec_len
 452                                                        (my_copy_buffer),
 453                                                        my_copy_buffer);
 454           else
 455             {
 456               /* We don't do anything with the data, drop it */
 457               actual_transfer = svm_fifo_max_dequeue_cons (rx_fifo);
 458               svm_fifo_dequeue_drop (rx_fifo, actual_transfer);
 459             }
 460         }
 461       while (actual_transfer <= 0);
 462
 463       server_bytes_received += actual_transfer;
 464
 465       if (utm->have_return)
 466         {
 467           buffer_offset = 0;
 468           while (actual_transfer > 0)
 469             {
 470               rv = svm_fifo_enqueue_nowait (tx_fifo, actual_transfer,
 471                                             my_copy_buffer + buffer_offset);
 472               if (rv > 0)
 473                 {
 474                   actual_transfer -= rv;
 475                   buffer_offset += rv;
 476                   server_bytes_sent += rv;
 477                 }
 478
 479             }
 480         }
 481       if (PREDICT_FALSE (utm->time_to_stop))
 482         break;
 483     }
 484
 485   pthread_exit (0);
 486 }
 487
 488 static void
 489 session_accepted_handler (session_accepted_msg_t * mp)
 490 {
 491   app_session_evt_t _app_evt, *app_evt = &_app_evt;
 492   udp_echo_main_t *utm = &udp_echo_main;
 493   session_accepted_reply_msg_t *rmp;
 494   svm_fifo_t *rx_fifo, *tx_fifo;
 495   app_session_t *session;
 496   static f64 start_time;
 497   u32 session_index;
 498   int rv = 0;
 499
 500   if (start_time == 0.0)
 501     start_time = clib_time_now (&utm->clib_time);
 502
 503   utm->vpp_event_queue = uword_to_pointer (mp->vpp_event_queue_address,
 504                                            svm_msg_q_t *);
 505   rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
 506   tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
 507
 508   pool_get (utm->sessions, session);
 509   clib_memset (session, 0, sizeof (*session));
 510   session_index = session - utm->sessions;
 511   session->session_index = session_index;
 512
 513   rx_fifo->client_session_index = session_index;
 514   tx_fifo->client_session_index = session_index;
 515   session->rx_fifo = rx_fifo;
 516   session->tx_fifo = tx_fifo;
 517   clib_memcpy_fast (&session->transport.rmt_ip, &mp->rmt.ip,
 518                     sizeof (ip46_address_t));
 519   session->transport.is_ip4 = mp->rmt.is_ip4;
 520   session->transport.rmt_port = mp->rmt.port;
 521
 522   hash_set (utm->session_index_by_vpp_handles, mp->handle, session_index);
 523   if (pool_elts (utm->sessions) && (pool_elts (utm->sessions) % 20000) == 0)
 524     {
 525       f64 now = clib_time_now (&utm->clib_time);
 526       fformat (stdout, "%d active sessions in %.2f seconds, %.2f/sec...\n",
 527                pool_elts (utm->sessions), now - start_time,
 528                (f64) pool_elts (utm->sessions) / (now - start_time));
 529     }
 530
 531   app_alloc_ctrl_evt_to_vpp (utm->vpp_event_queue, app_evt,
 532                              SESSION_CTRL_EVT_ACCEPTED_REPLY);
 533   rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
 534   rmp->handle = mp->handle;
 535   rmp->context = mp->context;
 536   rmp->retval = rv;
 537   app_send_ctrl_evt_to_vpp (utm->vpp_event_queue, app_evt);
 538
 539   CLIB_MEMORY_BARRIER ();
 540   utm->state = STATE_READY;
 541 }
 542
 543 static void
 544 session_disconnected_handler (session_disconnected_msg_t * mp)
 545 {
 546   app_session_evt_t _app_evt, *app_evt = &_app_evt;
 547   udp_echo_main_t *utm = &udp_echo_main;
 548   session_disconnected_reply_msg_t *rmp;
 549   app_session_t *session;
 550   uword *p;
 551   int rv = 0;
 552
 553   p = hash_get (utm->session_index_by_vpp_handles, mp->handle);
 554
 555   if (p)
 556     {
 557       session = pool_elt_at_index (utm->sessions, p[0]);
 558       hash_unset (utm->session_index_by_vpp_handles, mp->handle);
 559       clib_warning ("disconnecting %u", session->session_index);
 560       pool_put (utm->sessions, session);
 561     }
 562   else
 563     {
 564       clib_warning ("couldn't find session key %llx", mp->handle);
 565       return;
 566     }
 567
 568   app_alloc_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt,
 569                              SESSION_CTRL_EVT_DISCONNECTED_REPLY);
 570   rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
 571   rmp->retval = rv;
 572   rmp->handle = mp->handle;
 573   rmp->context = mp->context;
 574   app_send_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt);
 575 }
 576
 577 static void
 578 session_connected_handler (session_connected_msg_t * mp)
 579 {
 580   udp_echo_main_t *utm = &udp_echo_main;
 581   unformat_input_t _input, *input = &_input;
 582   session_endpoint_cfg_t _sep, *sep = &_sep;
 583   app_session_t *session;
 584
 585   ASSERT (utm->i_am_server == 0);
 586
 587   if (mp->retval)
 588     {
 589       clib_warning ("failed connect");
 590       return;
 591     }
 592
 593   ASSERT (mp->server_rx_fifo && mp->server_tx_fifo);
 594
 595   pool_get (utm->sessions, session);
 596   session->session_index = session - utm->sessions;
 597   session->rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
 598   session->tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
 599
 600   /* Cut-through case */
 601   if (mp->ct_rx_fifo)
 602     {
 603       clib_warning ("cut-through session");
 604       session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
 605                                              svm_msg_q_t *);
 606       utm->cut_through_session_index = session->session_index;
 607       session->is_dgram = 0;
 608       sleep (1);
 609       session->rx_fifo->client_session_index = session->session_index;
 610       session->tx_fifo->client_session_index = session->session_index;
 611       /* TODO use ct fifos */
 612     }
 613   else
 614     {
 615       utm->connected_session = session->session_index;
 616       utm->vpp_event_queue = uword_to_pointer (mp->vpp_event_queue_address,
 617                                                svm_msg_q_t *);
 618
 619       session->rx_fifo->client_session_index = session->session_index;
 620       session->tx_fifo->client_session_index = session->session_index;
 621       clib_memcpy_fast (&session->transport.lcl_ip, &mp->lcl.ip,
 622                         sizeof (ip46_address_t));
 623       session->transport.is_ip4 = mp->lcl.is_ip4;
 624       session->transport.lcl_port = mp->lcl.port;
 625
 626       unformat_init_vector (input, utm->connect_uri);
 627       if (!unformat (input, "%U", unformat_uri, sep))
 628         {
 629           clib_warning ("can't figure out remote ip and port");
 630           utm->state = STATE_FAILED;
 631           unformat_free (input);
 632           return;
 633         }
 634       unformat_free (input);
 635       clib_memcpy_fast (&session->transport.rmt_ip, &sep->ip,
 636                         sizeof (ip46_address_t));
 637       session->transport.rmt_port = sep->port;
 638       session->is_dgram = !utm->is_connected;
 639     }
 640   utm->state = STATE_READY;
 641 }
 642
 643 static void
 644 session_bound_handler (session_bound_msg_t * mp)
 645 {
 646   udp_echo_main_t *utm = &udp_echo_main;
 647   svm_fifo_t *rx_fifo, *tx_fifo;
 648   app_session_t *session;
 649   u32 session_index;
 650
 651   if (mp->retval)
 652     {
 653       clib_warning ("bind failed: %d", mp->retval);
 654       utm->state = STATE_FAILED;
 655       return;
 656     }
 657
 658   rx_fifo = uword_to_pointer (mp->rx_fifo, svm_fifo_t *);
 659   tx_fifo = uword_to_pointer (mp->tx_fifo, svm_fifo_t *);
 660
 661   pool_get (utm->sessions, session);
 662   clib_memset (session, 0, sizeof (*session));
 663   session_index = session - utm->sessions;
 664
 665   rx_fifo->client_session_index = session_index;
 666   tx_fifo->client_session_index = session_index;
 667   session->rx_fifo = rx_fifo;
 668   session->tx_fifo = tx_fifo;
 669   clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
 670                     sizeof (ip46_address_t));
 671   session->transport.is_ip4 = mp->lcl_is_ip4;
 672   session->transport.lcl_port = mp->lcl_port;
 673   session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
 674
 675   utm->state = utm->is_connected ? STATE_BOUND : STATE_READY;
 676 }
 677
 678 static void
 679 handle_mq_event (session_event_t * e)
 680 {
 681   switch (e->event_type)
 682     {
 683     case SESSION_CTRL_EVT_BOUND:
 684       session_bound_handler ((session_bound_msg_t *) e->data);
 685       break;
 686     case SESSION_CTRL_EVT_ACCEPTED:
 687       session_accepted_handler ((session_accepted_msg_t *) e->data);
 688       break;
 689     case SESSION_CTRL_EVT_CONNECTED:
 690       session_connected_handler ((session_connected_msg_t *) e->data);
 691       break;
 692     case SESSION_CTRL_EVT_DISCONNECTED:
 693       session_disconnected_handler ((session_disconnected_msg_t *) e->data);
 694       break;
 695     default:
 696       clib_warning ("unhandled %u", e->event_type);
 697     }
 698 }
 699
 700 static void
 701 udp_client_send_connect (udp_echo_main_t * utm)
 702 {
 703   vl_api_connect_uri_t *cmp;
 704   cmp = vl_msg_api_alloc (sizeof (*cmp));
 705   clib_memset (cmp, 0, sizeof (*cmp));
 706
 707   cmp->_vl_msg_id = ntohs (VL_API_CONNECT_URI);
 708   cmp->client_index = utm->my_client_index;
 709   cmp->context = ntohl (0xfeedface);
 710   memcpy (cmp->uri, utm->connect_uri, vec_len (utm->connect_uri));
 711   vl_msg_api_send_shmem (utm->vl_input_queue, (u8 *) & cmp);
 712 }
 713
 714 static void
 715 send_test_chunk (udp_echo_main_t * utm, app_session_t * s, u32 bytes)
 716 {
 717   u64 test_buf_len, bytes_this_chunk, test_buf_offset;
 718
 719   u8 *test_data = utm->connect_test_data;
 720   u32 bytes_to_snd, enq_space, min_chunk;
 721   int written;
 722
 723   test_buf_len = vec_len (test_data);
 724   test_buf_offset = utm->bytes_sent % test_buf_len;
 725   bytes_this_chunk = clib_min (test_buf_len - test_buf_offset,
 726                                utm->bytes_to_send);
 727   enq_space = svm_fifo_max_enqueue_prod (s->tx_fifo);
 728   bytes_this_chunk = clib_min (bytes_this_chunk, enq_space);
 729
 730   written = app_send (s, test_data + test_buf_offset, bytes_this_chunk,
 731                       SVM_Q_WAIT);
 732   if (written > 0)
 733     {
 734       utm->bytes_to_send -= written;
 735       utm->bytes_sent += written;
 736     }
 737 }
 738
 739 static void
 740 recv_test_chunk (udp_echo_main_t * utm, app_session_t * s)
 741 {
 742   app_recv (s, utm->rx_buf, vec_len (utm->rx_buf));
 743 }
 744
 745 void
 746 client_send_data (udp_echo_main_t * utm, u32 session_index)
 747 {
 748   f64 start_time, end_time, delta;
 749   app_session_t *session;
 750   char *transfer_type;
 751   u8 *test_data;
 752   int i;
 753
 754   vec_validate_aligned (utm->connect_test_data, 1024 * 1024 - 1,
 755                         CLIB_CACHE_LINE_BYTES);
 756   for (i = 0; i < vec_len (utm->connect_test_data); i++)
 757     utm->connect_test_data[i] = i & 0xff;
 758
 759   test_data = utm->connect_test_data;
 760   session = pool_elt_at_index (utm->sessions, session_index);
 761   ASSERT (vec_len (test_data) > 0);
 762
 763   utm->total_to_send = utm->bytes_to_send;
 764   vec_validate (utm->rx_buf, vec_len (test_data) - 1);
 765   start_time = clib_time_now (&utm->clib_time);
 766   while (!utm->time_to_stop && utm->bytes_to_send)
 767     {
 768       send_test_chunk (utm, session, 0);
 769       if (utm->have_return)
 770         recv_test_chunk (utm, session);
 771       if (utm->time_to_stop)
 772         break;
 773     }
 774
 775   if (utm->have_return)
 776     {
 777       f64 timeout = clib_time_now (&utm->clib_time) + 5;
 778       while (clib_time_now (&utm->clib_time) < timeout)
 779         recv_test_chunk (utm, session);
 780     }
 781
 782   end_time = clib_time_now (&utm->clib_time);
 783   delta = end_time - start_time;
 784   transfer_type = utm->have_return ? "full-duplex" : "half-duplex";
 785   clib_warning ("%lld bytes (%lld mbytes, %lld gbytes) in %.2f seconds",
 786                 utm->total_to_send, utm->total_to_send / (1ULL << 20),
 787                 utm->total_to_send / (1ULL << 30), delta);
 788   clib_warning ("%.2f bytes/second %s", ((f64) utm->total_to_send) / (delta),
 789                 transfer_type);
 790   clib_warning ("%.4f gbit/second %s",
 791                 (((f64) utm->total_to_send * 8.0) / delta / 1e9),
 792                 transfer_type);
 793 }
 794
 795 static int
 796 application_attach (udp_echo_main_t * utm)
 797 {
 798   application_send_attach (utm);
 799   if (wait_for_state_change (utm, STATE_ATTACHED))
 800     {
 801       clib_warning ("timeout waiting for STATE_ATTACHED");
 802       return -1;
 803     }
 804   return 0;
 805 }
 806
 807 static void
 808 client_test (udp_echo_main_t * utm)
 809 {
 810   f64 start_time, timeout = 100.0;
 811   app_session_t *session;
 812   svm_msg_q_msg_t msg;
 813   session_event_t *e;
 814
 815   if (application_attach (utm))
 816     return;
 817
 818   udp_client_send_connect (utm);
 819
 820   start_time = clib_time_now (&utm->clib_time);
 821   while (pool_elts (utm->sessions) != 1 && utm->state != STATE_FAILED)
 822     {
 823       svm_msg_q_sub (utm->our_event_queue, &msg, SVM_Q_WAIT, 0);
 824       e = svm_msg_q_msg_data (utm->our_event_queue, &msg);
 825       handle_mq_event (e);
 826       svm_msg_q_free_msg (utm->our_event_queue, &msg);
 827
 828       if (clib_time_now (&utm->clib_time) - start_time >= timeout)
 829         break;
 830     }
 831
 832   if (utm->cut_through_session_index != ~0)
 833     client_send_data (utm, utm->cut_through_session_index);
 834   else
 835     client_send_data (utm, utm->connected_session);
 836
 837   application_detach (utm);
 838   wait_for_state_change (utm, STATE_DETACHED);
 839 }
 840
 841 static void
 842 vl_api_map_another_segment_t_handler (vl_api_map_another_segment_t * mp)
 843 {
 844   svm_fifo_segment_create_args_t _a, *a = &_a;
 845   udp_echo_main_t *utm = &udp_echo_main;
 846   svm_fifo_segment_main_t *sm = &utm->segment_main;
 847   svm_fifo_segment_private_t *seg;
 848   int rv;
 849
 850   clib_memset (a, 0, sizeof (*a));
 851   a->segment_name = (char *) mp->segment_name;
 852   a->segment_size = mp->segment_size;
 853   /* Attach to the segment vpp created */
 854   rv = svm_fifo_segment_attach (sm, a);
 855   if (rv)
 856     {
 857       clib_warning ("svm_fifo_segment_attach ('%s') failed",
 858                     mp->segment_name);
 859       return;
 860     }
 861   seg = svm_fifo_segment_get_segment (sm, a->new_segment_indices[0]);
 862   clib_warning ("Mapped new segment '%s' size %d", seg->ssvm.name,
 863                 seg->ssvm.ssvm_size);
 864   hash_set (utm->segments_table, clib_net_to_host_u64 (mp->segment_handle),
 865             a->new_segment_indices[0]);
 866 }
 867
 868 static void
 869 vl_api_unmap_segment_t_handler (vl_api_unmap_segment_t * mp)
 870 {
 871   udp_echo_main_t *utm = &udp_echo_main;
 872   svm_fifo_segment_main_t *sm = &utm->segment_main;
 873   svm_fifo_segment_private_t *seg;
 874   uword *seg_indexp;
 875   u64 segment_handle;
 876
 877   segment_handle = clib_net_to_host_u64 (mp->segment_handle);
 878   seg_indexp = hash_get (utm->segments_table, segment_handle);
 879   if (!seg_indexp)
 880     {
 881       clib_warning ("segment not mapped: %s", segment_handle);
 882       return;
 883     }
 884   hash_unset (utm->segments_table, segment_handle);
 885   seg = svm_fifo_segment_get_segment (sm, (u32) seg_indexp[0]);
 886   svm_fifo_segment_delete (sm, seg);
 887   clib_warning ("Unmapped segment '%s'", segment_handle);
 888 }
 889
 890 static void
 891 vl_api_unbind_uri_reply_t_handler (vl_api_unbind_uri_reply_t * mp)
 892 {
 893   udp_echo_main_t *utm = &udp_echo_main;
 894
 895   if (mp->retval != 0)
 896     clib_warning ("returned %d", ntohl (mp->retval));
 897
 898   utm->state = STATE_START;
 899 }
 900
 901 static void
 902   vl_api_app_cut_through_registration_add_t_handler
 903   (vl_api_app_cut_through_registration_add_t * mp)
 904 {
 905
 906 }
 907
 908 #define foreach_tcp_echo_msg                                            \
 909 _(UNBIND_URI_REPLY, unbind_uri_reply)                                   \
 910 _(MAP_ANOTHER_SEGMENT, map_another_segment)                             \
 911 _(UNMAP_SEGMENT, unmap_segment)                                         \
 912 _(APPLICATION_ATTACH_REPLY, application_attach_reply)                   \
 913 _(APPLICATION_DETACH_REPLY, application_detach_reply)                   \
 914 _(APP_CUT_THROUGH_REGISTRATION_ADD, app_cut_through_registration_add)   \
 915
 916 void
 917 tcp_echo_api_hookup (udp_echo_main_t * utm)
 918 {
 919 #define _(N,n)                                                  \
 920     vl_msg_api_set_handlers(VL_API_##N, #n,                     \
 921                            vl_api_##n##_t_handler,              \
 922                            vl_noop_handler,                     \
 923                            vl_api_##n##_t_endian,               \
 924                            vl_api_##n##_t_print,                \
 925                            sizeof(vl_api_##n##_t), 1);
 926   foreach_tcp_echo_msg;
 927 #undef _
 928
 929 }
 930
 931 int
 932 connect_to_vpp (char *name)
 933 {
 934   udp_echo_main_t *utm = &udp_echo_main;
 935   api_main_t *am = &api_main;
 936
 937   if (vl_client_connect_to_vlib ("/vpe-api", name, 32) < 0)
 938     return -1;
 939
 940   utm->vl_input_queue = am->shmem_hdr->vl_input_queue;
 941   utm->my_client_index = am->my_client_index;
 942
 943   return 0;
 944 }
 945
 946 void
 947 vlib_cli_output (struct vlib_main_t *vm, char *fmt, ...)
 948 {
 949   clib_warning ("BUG");
 950 }
 951
 952 static void
 953 init_error_string_table (udp_echo_main_t * utm)
 954 {
 955   utm->error_string_by_error_number = hash_create (0, sizeof (uword));
 956
 957 #define _(n,v,s) hash_set (utm->error_string_by_error_number, -v, s);
 958   foreach_vnet_api_error;
 959 #undef _
 960
 961   hash_set (utm->error_string_by_error_number, 99, "Misc");
 962 }
 963
 964 void
 965 server_handle_fifo_event_rx (udp_echo_main_t * utm, u32 session_index)
 966 {
 967   svm_fifo_t *rx_fifo, *tx_fifo;
 968   int n_read;
 969   app_session_t *session;
 970   int rv;
 971   u32 max_dequeue, offset, max_transfer, rx_buf_len;
 972
 973   session = pool_elt_at_index (utm->sessions, session_index);
 974   rx_buf_len = vec_len (utm->rx_buf);
 975   rx_fifo = session->rx_fifo;
 976   tx_fifo = session->tx_fifo;
 977
 978
 979   max_dequeue = svm_fifo_max_dequeue_cons (rx_fifo);
 980   /* Allow enqueuing of a new event */
 981   svm_fifo_unset_event (rx_fifo);
 982
 983   if (PREDICT_FALSE (!max_dequeue))
 984     return;
 985
 986   /* Read the max_dequeue */
 987   do
 988     {
 989       max_transfer = clib_min (rx_buf_len, max_dequeue);
 990       if (session->is_dgram)
 991         n_read = app_recv_dgram_raw (rx_fifo, utm->rx_buf, max_transfer,
 992                                      &session->transport, 0, 0);
 993       else
 994         n_read = app_recv_stream_raw (rx_fifo, utm->rx_buf, max_transfer, 0,
 995                                       0);
 996
 997       if (n_read > 0)
 998         max_dequeue -= n_read;
 999
1000       /* Reflect if a non-drop session */
1001       if (utm->have_return && n_read > 0)
1002         {
1003           offset = 0;
1004           do
1005             {
1006               rv = app_send (session, &utm->rx_buf[offset], n_read,
1007                              SVM_Q_WAIT);
1008               if (rv > 0)
1009                 {
1010                   n_read -= rv;
1011                   offset += rv;
1012                 }
1013             }
1014           while ((rv <= 0 || n_read > 0) && !utm->time_to_stop);
1015
1016           /* If event wasn't set, add one */
1017           if (svm_fifo_set_event (tx_fifo))
1018             app_send_io_evt_to_vpp (session->vpp_evt_q,
1019                                     tx_fifo->master_session_index,
1020                                     SESSION_IO_EVT_TX, SVM_Q_WAIT);
1021         }
1022     }
1023   while ((n_read < 0 || max_dequeue > 0) && !utm->time_to_stop);
1024 }
1025
1026 static void
1027 server_handle_event_queue (udp_echo_main_t * utm)
1028 {
1029   session_event_t *e;
1030   svm_msg_q_msg_t msg;
1031   svm_msg_q_t *mq = utm->our_event_queue;
1032   int i;
1033
1034   while (utm->state != STATE_READY)
1035     sleep (5);
1036
1037   while (1)
1038     {
1039       if (svm_msg_q_sub (mq, &msg, SVM_Q_WAIT, 0))
1040         {
1041           clib_warning ("svm msg q returned");
1042           continue;
1043         }
1044       e = svm_msg_q_msg_data (mq, &msg);
1045       switch (e->event_type)
1046         {
1047         case SESSION_IO_EVT_RX:
1048           server_handle_fifo_event_rx (utm, e->session_index);
1049           break;
1050
1051         default:
1052           handle_mq_event (e);
1053           break;
1054         }
1055       svm_msg_q_free_msg (mq, &msg);
1056       if (PREDICT_FALSE (utm->time_to_stop == 1))
1057         return;
1058       if (PREDICT_FALSE (utm->time_to_print_stats == 1))
1059         {
1060           utm->time_to_print_stats = 0;
1061           fformat (stdout, "%d connections\n", pool_elts (utm->sessions));
1062         }
1063     }
1064 }
1065
1066 static void
1067 server_unbind (udp_echo_main_t * utm)
1068 {
1069   vl_api_unbind_uri_t *ump;
1070
1071   ump = vl_msg_api_alloc (sizeof (*ump));
1072   clib_memset (ump, 0, sizeof (*ump));
1073
1074   ump->_vl_msg_id = ntohs (VL_API_UNBIND_URI);
1075   ump->client_index = utm->my_client_index;
1076   memcpy (ump->uri, utm->listen_uri, vec_len (utm->listen_uri));
1077   vl_msg_api_send_shmem (utm->vl_input_queue, (u8 *) & ump);
1078 }
1079
1080 static void
1081 server_bind (udp_echo_main_t * utm)
1082 {
1083   vl_api_bind_uri_t *bmp;
1084
1085   bmp = vl_msg_api_alloc (sizeof (*bmp));
1086   clib_memset (bmp, 0, sizeof (*bmp));
1087
1088   bmp->_vl_msg_id = ntohs (VL_API_BIND_URI);
1089   bmp->client_index = utm->my_client_index;
1090   bmp->context = ntohl (0xfeedface);
1091   memcpy (bmp->uri, utm->listen_uri, vec_len (utm->listen_uri));
1092   vl_msg_api_send_shmem (utm->vl_input_queue, (u8 *) & bmp);
1093 }
1094
1095 void
1096 udp_server_test (udp_echo_main_t * utm)
1097 {
1098   u8 wait_for_state = utm->is_connected ? STATE_BOUND : STATE_READY;
1099   application_send_attach (utm);
1100
1101   /* Bind to uri */
1102   server_bind (utm);
1103
1104   if (wait_for_state_change (utm, wait_for_state))
1105     {
1106       clib_warning ("timeout waiting for state change");
1107       return;
1108     }
1109
1110   server_handle_event_queue (utm);
1111
1112   /* Cleanup */
1113   server_unbind (utm);
1114
1115   if (wait_for_state_change (utm, STATE_START))
1116     {
1117       clib_warning ("timeout waiting for STATE_START");
1118       return;
1119     }
1120
1121   application_detach (utm);
1122
1123   fformat (stdout, "Test complete...\n");
1124 }
1125
1126 int
1127 main (int argc, char **argv)
1128 {
1129   udp_echo_main_t *utm = &udp_echo_main;
1130   svm_fifo_segment_main_t *sm = &utm->segment_main;
1131   u8 *uri = (u8 *) "udp://6.0.1.1/1234";
1132   unformat_input_t _argv, *a = &_argv;
1133   int i_am_server = 1;
1134   app_session_t *session;
1135   u8 *chroot_prefix;
1136   char *app_name;
1137   u32 tmp;
1138   int i;
1139
1140   clib_mem_init_thread_safe (0, 256 << 20);
1141
1142   svm_fifo_segment_main_init (sm, HIGH_SEGMENT_BASEVA, 20);
1143
1144   vec_validate (utm->rx_buf, 128 << 10);
1145   utm->session_index_by_vpp_handles = hash_create (0, sizeof (uword));
1146   utm->my_pid = getpid ();
1147   utm->configured_segment_size = 1 << 20;
1148   utm->have_return = 1;
1149   utm->bytes_to_send = 1024;
1150   utm->fifo_size = 128 << 10;
1151   utm->cut_through_session_index = ~0;
1152   clib_time_init (&utm->clib_time);
1153
1154   init_error_string_table (utm);
1155   unformat_init_command_line (a, argv);
1156
1157   while (unformat_check_input (a) != UNFORMAT_END_OF_INPUT)
1158     {
1159       if (unformat (a, "chroot prefix %s", &chroot_prefix))
1160         {
1161           vl_set_memory_root_path ((char *) chroot_prefix);
1162         }
1163       else if (unformat (a, "uri %s", &uri))
1164         ;
1165       else if (unformat (a, "segment-size %dM", &tmp))
1166         utm->configured_segment_size = tmp << 20;
1167       else if (unformat (a, "segment-size %dG", &tmp))
1168         utm->configured_segment_size = tmp << 30;
1169       else if (unformat (a, "server"))
1170         i_am_server = 1;
1171       else if (unformat (a, "client"))
1172         i_am_server = 0;
1173       else if (unformat (a, "no-return"))
1174         utm->have_return = 0;
1175       else if (unformat (a, "mbytes %d", &tmp))
1176         utm->bytes_to_send = (u64) tmp << 20;
1177       else if (unformat (a, "fifo-size %d", &tmp))
1178         utm->fifo_size = tmp << 10;
1179       else
1180         {
1181           fformat (stderr, "%s: usage [server|client]\n");
1182           exit (1);
1183         }
1184     }
1185
1186   utm->i_am_server = i_am_server;
1187
1188   setup_signal_handlers ();
1189   tcp_echo_api_hookup (utm);
1190
1191   if (i_am_server)
1192     {
1193       utm->listen_uri = format (0, "%s%c", uri, 0);
1194       utm->is_connected = (utm->listen_uri[4] == 'c');
1195       app_name = "udp_echo_server";
1196     }
1197   else
1198     {
1199       app_name = "udp_echo_client";
1200       utm->connect_uri = format (0, "%s%c", uri, 0);
1201       utm->is_connected = (utm->connect_uri[4] == 'c');
1202     }
1203   if (connect_to_vpp (app_name) < 0)
1204     {
1205       svm_region_exit ();
1206       fformat (stderr, "Couldn't connect to vpe, exiting...\n");
1207       exit (1);
1208     }
1209
1210   if (i_am_server == 0)
1211     {
1212       client_test (utm);
1213       goto done;
1214     }
1215
1216   /* $$$$ hack preallocation */
1217   for (i = 0; i < 200000; i++)
1218     {
1219       pool_get (utm->sessions, session);
1220       clib_memset (session, 0, sizeof (*session));
1221     }
1222   for (i = 0; i < 200000; i++)
1223     pool_put_index (utm->sessions, i);
1224
1225   udp_server_test (utm);
1226
1227 done:
1228   vl_client_disconnect_from_vlib ();
1229   exit (0);
1230 }
1231
1232 #undef vl_api_version
1233 #define vl_api_version(n,v) static u32 vpe_api_version = v;
1234 #include <vpp/api/vpe.api.h>
1235 #undef vl_api_version
1236
1237 void
1238 vl_client_add_api_signatures (vl_api_memclnt_create_t * mp)
1239 {
1240   /*
1241    * Send the main API signature in slot 0. This bit of code must
1242    * match the checks in ../vpe/api/api.c: vl_msg_api_version_check().
1243    */
1244   mp->api_versions[0] = clib_host_to_net_u32 (vpe_api_version);
1245 }
1246
1247 u32
1248 vl (void *p)
1249 {
1250   return vec_len (p);
1251 }
1252
1253 /*
1254  * fd.io coding-style-patch-verification: ON
1255  *
1256  * Local Variables:
1257  * eval: (c-set-style "gnu")
1258  * End:
1259  */