FD.io VPP  v19.08-27-gf4dcae4
Vector Packet Processing
echo_client.c
Go to the documentation of this file.
1 /*
2  * echo_client.c - vpp built-in echo client code
3  *
4  * Copyright (c) 2017-2019 by Cisco and/or its affiliates.
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 
18 #include <vnet/vnet.h>
19 #include <vlibapi/api.h>
20 #include <vlibmemory/api.h>
21 #include <hs_apps/echo_client.h>
22 
24 
25 #define ECHO_CLIENT_DBG (0)
26 #define DBG(_fmt, _args...) \
27  if (ECHO_CLIENT_DBG) \
28  clib_warning (_fmt, ##_args)
29 
30 static void
32 {
34  ASSERT (vlib_get_thread_index () == 0);
35  vlib_process_signal_event (ecm->vlib_main, ecm->cli_node_index, *code, 0);
36 }
37 
38 static void
40 {
41  if (vlib_get_thread_index () != 0)
43  sizeof (code));
44  else
45  signal_evt_to_cli_i (&code);
46 }
47 
48 static void
50 {
51  u8 *test_data = ecm->connect_test_data;
52  int test_buf_len, test_buf_offset, rv;
53  u32 bytes_this_chunk;
54 
55  test_buf_len = vec_len (test_data);
56  ASSERT (test_buf_len > 0);
57  test_buf_offset = s->bytes_sent % test_buf_len;
58  bytes_this_chunk = clib_min (test_buf_len - test_buf_offset,
59  s->bytes_to_send);
60 
61  if (!ecm->is_dgram)
62  {
63  if (ecm->no_copy)
64  {
65  svm_fifo_t *f = s->data.tx_fifo;
66  rv = clib_min (svm_fifo_max_enqueue_prod (f), bytes_this_chunk);
68  session_send_io_evt_to_thread_custom (&f->master_session_index,
69  s->thread_index,
71  }
72  else
73  rv = app_send_stream (&s->data, test_data + test_buf_offset,
74  bytes_this_chunk, 0);
75  }
76  else
77  {
78  if (ecm->no_copy)
79  {
81  svm_fifo_t *f = s->data.tx_fifo;
82  app_session_transport_t *at = &s->data.transport;
83  u32 max_enqueue = svm_fifo_max_enqueue_prod (f);
84 
85  if (max_enqueue <= sizeof (session_dgram_hdr_t))
86  return;
87 
88  max_enqueue -= sizeof (session_dgram_hdr_t);
89  rv = clib_min (max_enqueue, bytes_this_chunk);
90 
91  hdr.data_length = rv;
92  hdr.data_offset = 0;
93  clib_memcpy_fast (&hdr.rmt_ip, &at->rmt_ip,
94  sizeof (ip46_address_t));
95  hdr.is_ip4 = at->is_ip4;
96  hdr.rmt_port = at->rmt_port;
97  clib_memcpy_fast (&hdr.lcl_ip, &at->lcl_ip,
98  sizeof (ip46_address_t));
99  hdr.lcl_port = at->lcl_port;
100  svm_fifo_enqueue (f, sizeof (hdr), (u8 *) & hdr);
101  svm_fifo_enqueue_nocopy (f, rv);
102  session_send_io_evt_to_thread_custom (&f->master_session_index,
103  s->thread_index,
105  }
106  else
107  rv = app_send_dgram (&s->data, test_data + test_buf_offset,
108  bytes_this_chunk, 0);
109  }
110 
111  /* If we managed to enqueue data... */
112  if (rv > 0)
113  {
114  /* Account for it... */
115  s->bytes_to_send -= rv;
116  s->bytes_sent += rv;
117 
118  if (ECHO_CLIENT_DBG)
119  {
120  /* *INDENT-OFF* */
121  ELOG_TYPE_DECLARE (e) =
122  {
123  .format = "tx-enq: xfer %d bytes, sent %u remain %u",
124  .format_args = "i4i4i4",
125  };
126  /* *INDENT-ON* */
127  struct
128  {
129  u32 data[3];
130  } *ed;
132  ed->data[0] = rv;
133  ed->data[1] = s->bytes_sent;
134  ed->data[2] = s->bytes_to_send;
135  }
136  }
137 }
138 
139 static void
141 {
142  svm_fifo_t *rx_fifo = s->data.rx_fifo;
143  u32 thread_index = vlib_get_thread_index ();
144  int n_read, i;
145 
146  if (ecm->test_bytes)
147  {
148  if (!ecm->is_dgram)
149  n_read = app_recv_stream (&s->data, ecm->rx_buf[thread_index],
150  vec_len (ecm->rx_buf[thread_index]));
151  else
152  n_read = app_recv_dgram (&s->data, ecm->rx_buf[thread_index],
153  vec_len (ecm->rx_buf[thread_index]));
154  }
155  else
156  {
157  n_read = svm_fifo_max_dequeue_cons (rx_fifo);
158  svm_fifo_dequeue_drop (rx_fifo, n_read);
159  }
160 
161  if (n_read > 0)
162  {
163  if (ECHO_CLIENT_DBG)
164  {
165  /* *INDENT-OFF* */
166  ELOG_TYPE_DECLARE (e) =
167  {
168  .format = "rx-deq: %d bytes",
169  .format_args = "i4",
170  };
171  /* *INDENT-ON* */
172  struct
173  {
174  u32 data[1];
175  } *ed;
177  ed->data[0] = n_read;
178  }
179 
180  if (ecm->test_bytes)
181  {
182  for (i = 0; i < n_read; i++)
183  {
184  if (ecm->rx_buf[thread_index][i]
185  != ((s->bytes_received + i) & 0xff))
186  {
187  clib_warning ("read %d error at byte %lld, 0x%x not 0x%x",
188  n_read, s->bytes_received + i,
189  ecm->rx_buf[thread_index][i],
190  ((s->bytes_received + i) & 0xff));
191  ecm->test_failed = 1;
192  }
193  }
194  }
195  ASSERT (n_read <= s->bytes_to_receive);
196  s->bytes_to_receive -= n_read;
197  s->bytes_received += n_read;
198  }
199 }
200 
201 static uword
203  vlib_frame_t * frame)
204 {
206  int my_thread_index = vlib_get_thread_index ();
207  eclient_session_t *sp;
208  int i;
209  int delete_session;
210  u32 *connection_indices;
211  u32 *connections_this_batch;
212  u32 nconnections_this_batch;
213 
214  connection_indices = ecm->connection_index_by_thread[my_thread_index];
215  connections_this_batch =
216  ecm->connections_this_batch_by_thread[my_thread_index];
217 
218  if ((ecm->run_test != ECHO_CLIENTS_RUNNING) ||
219  ((vec_len (connection_indices) == 0)
220  && vec_len (connections_this_batch) == 0))
221  return 0;
222 
223  /* Grab another pile of connections */
224  if (PREDICT_FALSE (vec_len (connections_this_batch) == 0))
225  {
226  nconnections_this_batch =
227  clib_min (ecm->connections_per_batch, vec_len (connection_indices));
228 
229  ASSERT (nconnections_this_batch > 0);
230  vec_validate (connections_this_batch, nconnections_this_batch - 1);
231  clib_memcpy_fast (connections_this_batch,
232  connection_indices + vec_len (connection_indices)
233  - nconnections_this_batch,
234  nconnections_this_batch * sizeof (u32));
235  _vec_len (connection_indices) -= nconnections_this_batch;
236  }
237 
239  && ecm->prev_conns == vec_len (connections_this_batch)))
240  {
241  ecm->repeats++;
242  ecm->prev_conns = vec_len (connections_this_batch);
243  if (ecm->repeats == 500000)
244  {
245  clib_warning ("stuck clients");
246  }
247  }
248  else
249  {
250  ecm->prev_conns = vec_len (connections_this_batch);
251  ecm->repeats = 0;
252  }
253 
254  for (i = 0; i < vec_len (connections_this_batch); i++)
255  {
256  delete_session = 1;
257 
258  sp = pool_elt_at_index (ecm->sessions, connections_this_batch[i]);
259 
260  if (sp->bytes_to_send > 0)
261  {
262  send_data_chunk (ecm, sp);
263  delete_session = 0;
264  }
265  if (sp->bytes_to_receive > 0)
266  {
267  delete_session = 0;
268  }
269  if (PREDICT_FALSE (delete_session == 1))
270  {
271  session_t *s;
272 
276 
277  if (s)
278  {
279  vnet_disconnect_args_t _a, *a = &_a;
280  a->handle = session_handle (s);
281  a->app_index = ecm->app_index;
283 
284  vec_delete (connections_this_batch, 1, i);
285  i--;
287  }
288  else
289  {
290  clib_warning ("session AWOL?");
291  vec_delete (connections_this_batch, 1, i);
292  }
293 
294  /* Kick the debug CLI process */
295  if (ecm->ready_connections == 0)
296  {
297  signal_evt_to_cli (2);
298  }
299  }
300  }
301 
302  ecm->connection_index_by_thread[my_thread_index] = connection_indices;
303  ecm->connections_this_batch_by_thread[my_thread_index] =
304  connections_this_batch;
305  return 0;
306 }
307 
308 /* *INDENT-OFF* */
310 {
311  .function = echo_client_node_fn,
312  .name = "echo-clients",
313  .type = VLIB_NODE_TYPE_INPUT,
314  .state = VLIB_NODE_STATE_DISABLED,
315 };
316 /* *INDENT-ON* */
317 
318 static int
320 {
321  api_main_t *am = &api_main;
322  vl_shmem_hdr_t *shmem_hdr;
323 
324  shmem_hdr = am->shmem_hdr;
325  ecm->vl_input_queue = shmem_hdr->vl_input_queue;
326  ecm->my_client_index = vl_api_memclnt_create_internal ("echo_client",
327  ecm->vl_input_queue);
328  return 0;
329 }
330 
331 static int
333 {
336  u32 num_threads;
337  int i;
338 
339  if (create_api_loopback (ecm))
340  return -1;
341 
342  num_threads = 1 /* main thread */ + vtm->n_threads;
343 
344  /* Init test data. Big buffer */
345  vec_validate (ecm->connect_test_data, 4 * 1024 * 1024 - 1);
346  for (i = 0; i < vec_len (ecm->connect_test_data); i++)
347  ecm->connect_test_data[i] = i & 0xff;
348 
349  vec_validate (ecm->rx_buf, num_threads - 1);
350  for (i = 0; i < num_threads; i++)
351  vec_validate (ecm->rx_buf[i], vec_len (ecm->connect_test_data) - 1);
352 
353  ecm->is_init = 1;
354 
359 
360  return 0;
361 }
362 
363 static int
365  session_t * s, u8 is_fail)
366 {
368  vnet_connect_args_t *a = 0;
369  int rv;
370  u8 thread_index = vlib_get_thread_index ();
372  u32 stream_n;
373 
374  DBG ("QUIC Connection handle %d", session_handle (s));
375 
376  vec_validate (a, 1);
377  a->uri = (char *) ecm->connect_uri;
378  if (parse_uri (a->uri, &sep))
379  return -1;
380  sep.parent_handle = session_handle (s);
381 
382  for (stream_n = 0; stream_n < ecm->quic_streams; stream_n++)
383  {
384  clib_memset (a, 0, sizeof (*a));
385  a->app_index = ecm->app_index;
386  a->api_context = -1 - api_context;
387  clib_memcpy (&a->sep_ext, &sep, sizeof (sep));
388 
389  DBG ("QUIC opening stream %d", stream_n);
390  if ((rv = vnet_connect (a)))
391  {
392  clib_error ("Stream session %d opening failed: %d", stream_n, rv);
393  return -1;
394  }
395  DBG ("QUIC stream %d connected", stream_n);
396  }
397  vec_add1 (ecm->quic_session_index_by_thread[thread_index],
398  session_handle (s));
399  vec_free (a);
400  return 0;
401 }
402 
403 static int
405  session_t * s, u8 is_fail)
406 {
408  eclient_session_t *session;
409  u32 session_index;
410  u8 thread_index;
411 
413  return -1;
414 
415  if (is_fail)
416  {
417  clib_warning ("connection %d failed!", api_context);
419  signal_evt_to_cli (-1);
420  return 0;
421  }
422 
425  api_context, s,
426  is_fail);
427  DBG ("STREAM Connection callback %d", api_context);
428 
429  thread_index = s->thread_index;
430  ASSERT (thread_index == vlib_get_thread_index ()
432 
433  if (!ecm->vpp_event_queue[thread_index])
434  ecm->vpp_event_queue[thread_index] =
435  session_main_get_vpp_event_queue (thread_index);
436 
437  /*
438  * Setup session
439  */
441  pool_get (ecm->sessions, session);
443 
444  clib_memset (session, 0, sizeof (*session));
445  session_index = session - ecm->sessions;
446  session->bytes_to_send = ecm->bytes_to_send;
447  session->bytes_to_receive = ecm->no_return ? 0ULL : ecm->bytes_to_send;
448  session->data.rx_fifo = s->rx_fifo;
449  session->data.rx_fifo->client_session_index = session_index;
450  session->data.tx_fifo = s->tx_fifo;
451  session->data.tx_fifo->client_session_index = session_index;
452  session->data.vpp_evt_q = ecm->vpp_event_queue[thread_index];
453  session->vpp_session_handle = session_handle (s);
454 
455  if (ecm->is_dgram)
456  {
458  tc = session_get_transport (s);
459  clib_memcpy_fast (&session->data.transport, tc,
460  sizeof (session->data.transport));
461  session->data.is_dgram = 1;
462  }
463 
464  vec_add1 (ecm->connection_index_by_thread[thread_index], session_index);
466  if (ecm->ready_connections == ecm->expected_connections)
467  {
469  /* Signal the CLI process that the action is starting... */
470  signal_evt_to_cli (1);
471  }
472 
473  return 0;
474 }
475 
476 static int
478  session_t * s, u8 is_fail)
479 {
481  eclient_session_t *session;
482  u32 session_index;
483  u8 thread_index;
484 
486  return -1;
487 
488  if (is_fail)
489  {
490  clib_warning ("connection %d failed!", api_context);
492  signal_evt_to_cli (-1);
493  return 0;
494  }
495 
496  thread_index = s->thread_index;
497  ASSERT (thread_index == vlib_get_thread_index ()
499 
500  if (!ecm->vpp_event_queue[thread_index])
501  ecm->vpp_event_queue[thread_index] =
502  session_main_get_vpp_event_queue (thread_index);
503 
504  /*
505  * Setup session
506  */
508  pool_get (ecm->sessions, session);
510 
511  clib_memset (session, 0, sizeof (*session));
512  session_index = session - ecm->sessions;
513  session->bytes_to_send = ecm->bytes_to_send;
514  session->bytes_to_receive = ecm->no_return ? 0ULL : ecm->bytes_to_send;
515  session->data.rx_fifo = s->rx_fifo;
516  session->data.rx_fifo->client_session_index = session_index;
517  session->data.tx_fifo = s->tx_fifo;
518  session->data.tx_fifo->client_session_index = session_index;
519  session->data.vpp_evt_q = ecm->vpp_event_queue[thread_index];
520  session->vpp_session_handle = session_handle (s);
521 
522  if (ecm->is_dgram)
523  {
525  tc = session_get_transport (s);
526  clib_memcpy_fast (&session->data.transport, tc,
527  sizeof (session->data.transport));
528  session->data.is_dgram = 1;
529  }
530 
531  vec_add1 (ecm->connection_index_by_thread[thread_index], session_index);
533  if (ecm->ready_connections == ecm->expected_connections)
534  {
536  /* Signal the CLI process that the action is starting... */
537  signal_evt_to_cli (1);
538  }
539 
540  return 0;
541 }
542 
543 static void
545 {
547  vnet_disconnect_args_t _a = { 0 }, *a = &_a;
548 
550  clib_warning ("Reset active connection %U", format_session, s, 2);
551 
552  a->handle = session_handle (s);
553  a->app_index = ecm->app_index;
555  return;
556 }
557 
558 static int
560 {
561  return 0;
562 }
563 
564 static void
566 {
568  vnet_disconnect_args_t _a = { 0 }, *a = &_a;
569  a->handle = session_handle (s);
570  a->app_index = ecm->app_index;
572  return;
573 }
574 
575 void
577 {
579  vnet_disconnect_args_t _a = { 0 }, *a = &_a;
580  a->handle = session_handle (s);
581  a->app_index = ecm->app_index;
583 }
584 
585 static int
587 {
589  eclient_session_t *sp;
590 
592  {
594  return -1;
595  }
596 
597  sp = pool_elt_at_index (ecm->sessions, s->rx_fifo->client_session_index);
598  receive_data_chunk (ecm, sp);
599 
601  {
602  if (svm_fifo_set_event (s->rx_fifo))
604  }
605  return 0;
606 }
607 
608 int
609 echo_client_add_segment_callback (u32 client_index, u64 segment_handle)
610 {
611  /* New heaps may be added */
612  return 0;
613 }
614 
615 /* *INDENT-OFF* */
617  .session_reset_callback = echo_clients_session_reset_callback,
618  .session_connected_callback = echo_clients_session_connected_callback,
619  .session_accept_callback = echo_clients_session_create_callback,
620  .session_disconnect_callback = echo_clients_session_disconnect_callback,
621  .builtin_app_rx_callback = echo_clients_rx_callback,
622  .add_segment_callback = echo_client_add_segment_callback
623 };
624 /* *INDENT-ON* */
625 
626 static clib_error_t *
627 echo_clients_attach (u8 * appns_id, u64 appns_flags, u64 appns_secret)
628 {
629  u32 prealloc_fifos, segment_size = 256 << 20;
631  vnet_app_attach_args_t _a, *a = &_a;
632  u64 options[16];
633  int rv;
634 
635  clib_memset (a, 0, sizeof (*a));
636  clib_memset (options, 0, sizeof (options));
637 
638  a->api_client_index = ecm->my_client_index;
640  echo_clients.session_connected_callback =
642  a->session_cb_vft = &echo_clients;
643 
644  prealloc_fifos = ecm->prealloc_fifos ? ecm->expected_connections : 1;
645 
646  if (ecm->private_segment_size)
647  segment_size = ecm->private_segment_size;
648 
649  options[APP_OPTIONS_ACCEPT_COOKIE] = 0x12345678;
650  options[APP_OPTIONS_SEGMENT_SIZE] = segment_size;
651  options[APP_OPTIONS_ADD_SEGMENT_SIZE] = segment_size;
652  options[APP_OPTIONS_RX_FIFO_SIZE] = ecm->fifo_size;
653  options[APP_OPTIONS_TX_FIFO_SIZE] = ecm->fifo_size;
655  options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = prealloc_fifos;
656  options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN;
657  options[APP_OPTIONS_TLS_ENGINE] = ecm->tls_engine;
658  if (appns_id)
659  {
660  options[APP_OPTIONS_FLAGS] |= appns_flags;
661  options[APP_OPTIONS_NAMESPACE_SECRET] = appns_secret;
662  }
663  a->options = options;
664  a->namespace_id = appns_id;
665 
666  if ((rv = vnet_application_attach (a)))
667  return clib_error_return (0, "attach returned %d", rv);
668 
669  ecm->app_index = a->app_index;
670  return 0;
671 }
672 
673 static int
675 {
677  vnet_app_detach_args_t _da, *da = &_da;
678  int rv;
679 
680  da->app_index = ecm->app_index;
681  da->api_client_index = ~0;
682  rv = vnet_application_detach (da);
683  ecm->test_client_attached = 0;
684  ecm->app_index = ~0;
685  return rv;
686 }
687 
688 static void *
690 {
691  return 0;
692 }
693 
694 /** Start a transmit thread */
695 int
697 {
698  if (ecm->client_thread_handle == 0)
699  {
700  int rv = pthread_create (&ecm->client_thread_handle,
701  NULL /*attr */ ,
703  if (rv)
704  {
705  ecm->client_thread_handle = 0;
706  return -1;
707  }
708  }
709  return 0;
710 }
711 
712 clib_error_t *
714 {
716  vnet_connect_args_t _a, *a = &_a;
717  int i, rv;
718 
719  clib_memset (a, 0, sizeof (*a));
720 
721  for (i = 0; i < n_clients; i++)
722  {
723  a->uri = (char *) ecm->connect_uri;
724  a->api_context = i;
725  a->app_index = ecm->app_index;
726  if ((rv = vnet_connect_uri (a)))
727  return clib_error_return (0, "connect returned: %d", rv);
728 
729  /* Crude pacing for call setups */
730  if ((i % 16) == 0)
731  vlib_process_suspend (vm, 100e-6);
732  ASSERT (i + 1 >= ecm->ready_connections);
733  while (i + 1 - ecm->ready_connections > 128)
734  vlib_process_suspend (vm, 1e-3);
735  }
736  return 0;
737 }
738 
739 #define ec_cli_output(_fmt, _args...) \
740  if (!ecm->no_output) \
741  vlib_cli_output(vm, _fmt, ##_args)
742 
743 static clib_error_t *
745  unformat_input_t * input, vlib_cli_command_t * cmd)
746 {
748  vlib_thread_main_t *thread_main = vlib_get_thread_main ();
749  u64 tmp, total_bytes, appns_flags = 0, appns_secret = 0;
750  f64 test_timeout = 20.0, syn_timeout = 20.0, delta;
751  char *default_uri = "tcp://6.0.1.1/1234";
752  uword *event_data = 0, event_type;
753  f64 time_before_connects;
754  u32 n_clients = 1;
755  int preallocate_sessions = 0;
756  char *transfer_type;
757  clib_error_t *error = 0;
758  u8 *appns_id = 0;
759  int i;
761  int rv;
762 
763  ecm->quic_streams = 1;
764  ecm->bytes_to_send = 8192;
765  ecm->no_return = 0;
766  ecm->fifo_size = 64 << 10;
767  ecm->connections_per_batch = 1000;
768  ecm->private_segment_count = 0;
769  ecm->private_segment_size = 0;
770  ecm->no_output = 0;
771  ecm->test_bytes = 0;
772  ecm->test_failed = 0;
773  ecm->vlib_main = vm;
775  ecm->no_copy = 0;
777 
778  if (thread_main->n_vlib_mains > 1)
780  vec_free (ecm->connect_uri);
781 
783  {
784  if (unformat (input, "uri %s", &ecm->connect_uri))
785  ;
786  else if (unformat (input, "nclients %d", &n_clients))
787  ;
788  else if (unformat (input, "quic-streams %d", &ecm->quic_streams))
789  ;
790  else if (unformat (input, "mbytes %lld", &tmp))
791  ecm->bytes_to_send = tmp << 20;
792  else if (unformat (input, "gbytes %lld", &tmp))
793  ecm->bytes_to_send = tmp << 30;
794  else if (unformat (input, "bytes %lld", &ecm->bytes_to_send))
795  ;
796  else if (unformat (input, "test-timeout %f", &test_timeout))
797  ;
798  else if (unformat (input, "syn-timeout %f", &syn_timeout))
799  ;
800  else if (unformat (input, "no-return"))
801  ecm->no_return = 1;
802  else if (unformat (input, "fifo-size %d", &ecm->fifo_size))
803  ecm->fifo_size <<= 10;
804  else if (unformat (input, "private-segment-count %d",
805  &ecm->private_segment_count))
806  ;
807  else if (unformat (input, "private-segment-size %U",
808  unformat_memory_size, &tmp))
809  {
810  if (tmp >= 0x100000000ULL)
811  return clib_error_return
812  (0, "private segment size %lld (%llu) too large", tmp, tmp);
813  ecm->private_segment_size = tmp;
814  }
815  else if (unformat (input, "preallocate-fifos"))
816  ecm->prealloc_fifos = 1;
817  else if (unformat (input, "preallocate-sessions"))
818  preallocate_sessions = 1;
819  else
820  if (unformat (input, "client-batch %d", &ecm->connections_per_batch))
821  ;
822  else if (unformat (input, "appns %_%v%_", &appns_id))
823  ;
824  else if (unformat (input, "all-scope"))
825  appns_flags |= (APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE
826  | APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE);
827  else if (unformat (input, "local-scope"))
828  appns_flags = APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE;
829  else if (unformat (input, "global-scope"))
830  appns_flags = APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE;
831  else if (unformat (input, "secret %lu", &appns_secret))
832  ;
833  else if (unformat (input, "no-output"))
834  ecm->no_output = 1;
835  else if (unformat (input, "test-bytes"))
836  ecm->test_bytes = 1;
837  else if (unformat (input, "tls-engine %d", &ecm->tls_engine))
838  ;
839  else
840  return clib_error_return (0, "failed: unknown input `%U'",
841  format_unformat_error, input);
842  }
843 
844  /* Store cli process node index for signalling */
845  ecm->cli_node_index =
847 
848  if (ecm->is_init == 0)
849  {
850  if (echo_clients_init (vm))
851  return clib_error_return (0, "failed init");
852  }
853 
854 
855  ecm->ready_connections = 0;
856  ecm->expected_connections = n_clients * ecm->quic_streams;
857  ecm->rx_total = 0;
858  ecm->tx_total = 0;
859 
860  if (!ecm->connect_uri)
861  {
862  clib_warning ("No uri provided. Using default: %s", default_uri);
863  ecm->connect_uri = format (0, "%s%c", default_uri, 0);
864  }
865 
866  if ((rv = parse_uri ((char *) ecm->connect_uri, &sep)))
867  return clib_error_return (0, "Uri parse error: %d", rv);
868  ecm->transport_proto = sep.transport_proto;
869  ecm->is_dgram = (sep.transport_proto == TRANSPORT_PROTO_UDP);
870 
871 #if ECHO_CLIENT_PTHREAD
873 #endif
874 
876  vnet_session_enable_disable (vm, 1 /* turn on session and transports */ );
878 
879  if (ecm->test_client_attached == 0)
880  {
881  if ((error = echo_clients_attach (appns_id, appns_flags, appns_secret)))
882  {
883  vec_free (appns_id);
884  clib_error_report (error);
885  return error;
886  }
887  vec_free (appns_id);
888  }
889  ecm->test_client_attached = 1;
890 
891  /* Turn on the builtin client input nodes */
892  for (i = 0; i < thread_main->n_vlib_mains; i++)
894  VLIB_NODE_STATE_POLLING);
895 
896  if (preallocate_sessions)
897  pool_init_fixed (ecm->sessions, 1.1 * n_clients);
898 
899  /* Fire off connect requests */
900  time_before_connects = vlib_time_now (vm);
901  if ((error = echo_clients_connect (vm, n_clients)))
902  goto cleanup;
903 
904  /* Park until the sessions come up, or ten seconds elapse... */
905  vlib_process_wait_for_event_or_clock (vm, syn_timeout);
906  event_type = vlib_process_get_events (vm, &event_data);
907  switch (event_type)
908  {
909  case ~0:
910  ec_cli_output ("Timeout with only %d sessions active...",
911  ecm->ready_connections);
912  error = clib_error_return (0, "failed: syn timeout with %d sessions",
913  ecm->ready_connections);
914  goto cleanup;
915 
916  case 1:
917  delta = vlib_time_now (vm) - time_before_connects;
918  if (delta != 0.0)
919  ec_cli_output ("%d three-way handshakes in %.2f seconds %.2f/s",
920  n_clients, delta, ((f64) n_clients) / delta);
921 
923  ec_cli_output ("Test started at %.6f", ecm->test_start_time);
924  break;
925 
926  default:
927  ec_cli_output ("unexpected event(1): %d", event_type);
928  error = clib_error_return (0, "failed: unexpected event(1): %d",
929  event_type);
930  goto cleanup;
931  }
932 
933  /* Now wait for the sessions to finish... */
934  vlib_process_wait_for_event_or_clock (vm, test_timeout);
935  event_type = vlib_process_get_events (vm, &event_data);
936  switch (event_type)
937  {
938  case ~0:
939  ec_cli_output ("Timeout with %d sessions still active...",
940  ecm->ready_connections);
941  error = clib_error_return (0, "failed: timeout with %d sessions",
942  ecm->ready_connections);
943  goto cleanup;
944 
945  case 2:
946  ecm->test_end_time = vlib_time_now (vm);
947  ec_cli_output ("Test finished at %.6f", ecm->test_end_time);
948  break;
949 
950  default:
951  ec_cli_output ("unexpected event(2): %d", event_type);
952  error = clib_error_return (0, "failed: unexpected event(2): %d",
953  event_type);
954  goto cleanup;
955  }
956 
957  delta = ecm->test_end_time - ecm->test_start_time;
958  if (delta != 0.0)
959  {
960  total_bytes = (ecm->no_return ? ecm->tx_total : ecm->rx_total);
961  transfer_type = ecm->no_return ? "half-duplex" : "full-duplex";
962  ec_cli_output ("%lld bytes (%lld mbytes, %lld gbytes) in %.2f seconds",
963  total_bytes, total_bytes / (1ULL << 20),
964  total_bytes / (1ULL << 30), delta);
965  ec_cli_output ("%.2f bytes/second %s", ((f64) total_bytes) / (delta),
966  transfer_type);
967  ec_cli_output ("%.4f gbit/second %s",
968  (((f64) total_bytes * 8.0) / delta / 1e9),
969  transfer_type);
970  }
971  else
972  {
973  ec_cli_output ("zero delta-t?");
974  error = clib_error_return (0, "failed: zero delta-t");
975  goto cleanup;
976  }
977 
978  if (ecm->test_bytes && ecm->test_failed)
979  error = clib_error_return (0, "failed: test bytes");
980 
981 cleanup:
984  for (i = 0; i < vec_len (ecm->connection_index_by_thread); i++)
985  {
989  }
990 
991  pool_free (ecm->sessions);
992 
993  /* Detach the application, so we can use different fifo sizes next time */
994  if (ecm->test_client_attached)
995  {
996  if (echo_clients_detach ())
997  {
998  error = clib_error_return (0, "failed: app detach");
999  ec_cli_output ("WARNING: app detach failed...");
1000  }
1001  }
1002  if (error)
1003  ec_cli_output ("test failed");
1004  vec_free (ecm->connect_uri);
1005  return error;
1006 }
1007 
1008 /* *INDENT-OFF* */
1009 VLIB_CLI_COMMAND (echo_clients_command, static) =
1010 {
1011  .path = "test echo clients",
1012  .short_help = "test echo clients [nclients %d][[m|g]bytes <bytes>]"
1013  "[test-timeout <time>][syn-timeout <time>][no-return][fifo-size <size>]"
1014  "[private-segment-count <count>][private-segment-size <bytes>[m|g]]"
1015  "[preallocate-fifos][preallocate-sessions][client-batch <batch-size>]"
1016  "[uri <tcp://ip/port>][test-bytes][no-output]",
1017  .function = echo_clients_command_fn,
1018  .is_mp_safe = 1,
1019 };
1020 /* *INDENT-ON* */
1021 
1022 clib_error_t *
1024 {
1026  ecm->is_init = 0;
1027  return 0;
1028 }
1029 
1031 
1032 /*
1033  * fd.io coding-style-patch-verification: ON
1034  *
1035  * Local Variables:
1036  * eval: (c-set-style "gnu")
1037  * End:
1038  */
#define vec_validate(V, I)
Make sure vector is long enough for given index (no header, unspecified alignment) ...
Definition: vec.h:439
int echo_client_add_segment_callback(u32 client_index, u64 segment_handle)
Definition: echo_client.c:609
vlib_main_t vlib_global_main
Definition: main.c:1937
vlib_main_t * vlib_main
Definition: echo_client.h:107
clib_error_t * echo_clients_connect(vlib_main_t *vm, u32 n_clients)
Definition: echo_client.c:713
#define clib_min(x, y)
Definition: clib.h:295
static int echo_clients_session_create_callback(session_t *s)
Definition: echo_client.c:559
u32 vl_api_memclnt_create_internal(char *name, svm_queue_t *q)
Definition: memory_api.c:120
static f64 vlib_process_wait_for_event_or_clock(vlib_main_t *vm, f64 dt)
Suspend a cooperative multi-tasking thread Waits for an event, or for the indicated number of seconds...
Definition: node_funcs.h:673
u8 is_ip4
set if uses ip4 networking
static u32 svm_fifo_max_enqueue_prod(svm_fifo_t *f)
Maximum number of bytes that can be enqueued into fifo.
Definition: svm_fifo.h:609
vlib_node_runtime_t node_runtime
Definition: node.h:552
u8 * format_session(u8 *s, va_list *args)
Format stream session as per the following format.
Definition: session_cli.c:55
a
Definition: bitmap.h:538
svm_fifo_t * tx_fifo
struct _vnet_connect_args vnet_connect_args_t
static session_cb_vft_t echo_clients
Definition: echo_client.c:616
#define clib_error(format, args...)
Definition: error.h:62
unsigned long u64
Definition: types.h:89
static svm_msg_q_t * session_main_get_vpp_event_queue(u32 thread_index)
Definition: session.h:554
u32 tls_engine
TLS engine mbedtls/openssl.
Definition: echo_client.h:65
static int echo_clients_detach()
Definition: echo_client.c:674
#define clib_memcpy_fast(a, b, c)
Definition: string.h:81
#define NULL
Definition: clib.h:58
transport_connection_t * session_get_transport(session_t *s)
Definition: session.c:1398
static f64 vlib_time_now(vlib_main_t *vm)
Definition: main.h:258
static uword echo_client_node_fn(vlib_main_t *vm, vlib_node_runtime_t *node, vlib_frame_t *frame)
Definition: echo_client.c:202
svm_fifo_t * rx_fifo
Pointers to rx/tx buffers.
u32 expected_connections
Number of clients/connections.
Definition: echo_client.h:61
static_always_inline void clib_spinlock_unlock_if_init(clib_spinlock_t *p)
Definition: lock.h:108
#define vec_add1(V, E)
Add 1 element to end of vector (unspecified alignment).
Definition: vec.h:522
eclient_session_t * sessions
Session pool, shared.
Definition: echo_client.h:73
static int app_send_stream(app_session_t *s, u8 *data, u32 len, u8 noblock)
u64 bytes_to_send
Bytes to send.
Definition: echo_client.h:58
void svm_fifo_enqueue_nocopy(svm_fifo_t *f, u32 len)
Advance tail.
Definition: svm_fifo.c:883
int i
static void receive_data_chunk(echo_client_main_t *ecm, eclient_session_t *s)
Definition: echo_client.c:140
clib_memset(h->entries, 0, sizeof(h->entries[0])*entries)
volatile int run_test
Signal start of test.
Definition: echo_client.h:86
u32 quic_streams
QUIC streams per connection.
Definition: echo_client.h:68
u8 * format(u8 *s, const char *fmt,...)
Definition: format.c:424
u8 data[128]
Definition: ipsec.api:249
#define pool_get(P, E)
Allocate an object E from a pool P (unspecified alignment).
Definition: pool.h:236
vlib_main_t ** vlib_mains
Definition: buffer.c:321
void echo_clients_session_disconnect(session_t *s)
Definition: echo_client.c:576
unsigned char u8
Definition: types.h:56
app_session_t data
Definition: echo_client.h:33
u8 * connect_test_data
Pre-computed test data.
Definition: echo_client.h:76
#define vec_reset_length(v)
Reset vector length to zero NULL-pointer tolerant.
double f64
Definition: types.h:142
#define vlib_worker_thread_barrier_sync(X)
Definition: threads.h:204
static session_handle_t session_handle(session_t *s)
struct _svm_fifo svm_fifo_t
#define clib_memcpy(d, s, n)
Definition: string.h:180
static uword vlib_process_suspend(vlib_main_t *vm, f64 dt)
Suspend a vlib cooperative multi-tasking thread for a period of time.
Definition: node_funcs.h:422
clib_error_t * echo_clients_main_init(vlib_main_t *vm)
Definition: echo_client.c:1023
#define VLIB_INIT_FUNCTION(x)
Definition: init.h:173
static uword vlib_process_get_events(vlib_main_t *vm, uword **data_vector)
Return the first event type which has occurred and a vector of per-event data of that type...
Definition: node_funcs.h:516
struct _vnet_disconnect_args_t vnet_disconnect_args_t
int echo_clients_start_tx_pthread(echo_client_main_t *ecm)
Start a transmit thread.
Definition: echo_client.c:696
static u32 svm_fifo_max_dequeue_cons(svm_fifo_t *f)
Fifo max bytes to dequeue optimized for consumer.
Definition: svm_fifo.h:489
#define DBG(_fmt, _args...)
Definition: echo_client.c:26
#define clib_error_return(e, args...)
Definition: error.h:99
void vl_api_rpc_call_main_thread(void *fp, u8 *data, u32 data_length)
Definition: vlib_api.c:600
u32 app_index
app index after attach
Definition: echo_client.h:52
static int quic_echo_clients_qsession_connected_callback(u32 app_index, u32 api_context, session_t *s, u8 is_fail)
Definition: echo_client.c:364
svm_queue_t * vl_input_queue
vpe input queue
Definition: echo_client.h:47
struct vl_shmem_hdr_ * shmem_hdr
Binary API shared-memory segment header pointer.
Definition: api_common.h:266
ip46_address_t lcl_ip
svm_msg_q_t ** vpp_event_queue
Definition: echo_client.h:48
unsigned int u32
Definition: types.h:88
int session_send_io_evt_to_thread(svm_fifo_t *f, session_evt_type_t evt_type)
Definition: session.c:79
#define SESSION_INVALID_HANDLE
Definition: session_types.h:23
struct _vnet_app_attach_args_t vnet_app_attach_args_t
static void echo_clients_session_disconnect_callback(session_t *s)
Definition: echo_client.c:565
static void clib_spinlock_init(clib_spinlock_t *p)
Definition: lock.h:61
struct _session_endpoint_cfg session_endpoint_cfg_t
static int create_api_loopback(echo_client_main_t *ecm)
Definition: echo_client.c:319
static void signal_evt_to_cli(int code)
Definition: echo_client.c:39
static clib_error_t * echo_clients_command_fn(vlib_main_t *vm, unformat_input_t *input, vlib_cli_command_t *cmd)
Definition: echo_client.c:744
#define pool_elt_at_index(p, i)
Returns pointer to element at given index.
Definition: pool.h:514
volatile u32 ready_connections
Definition: echo_client.h:82
static void vlib_process_signal_event(vlib_main_t *vm, uword node_index, uword type_opaque, uword data)
Definition: node_funcs.h:934
struct _unformat_input_t unformat_input_t
static void cleanup(void)
Definition: client.c:131
int svm_fifo_enqueue(svm_fifo_t *f, u32 len, const u8 *src)
Enqueue data to fifo.
Definition: svm_fifo.c:807
#define ELOG_DATA(em, f)
Definition: elog.h:484
#define PREDICT_FALSE(x)
Definition: clib.h:111
ip46_address_t rmt_ip
remote ip
u32 ** connection_index_by_thread
Definition: echo_client.h:78
u32 private_segment_count
Number of private fifo segs.
Definition: echo_client.h:63
u32 ** connections_this_batch_by_thread
active connection batch
Definition: echo_client.h:79
u32 node_index
Node index.
Definition: node.h:494
clib_error_t * vnet_session_enable_disable(vlib_main_t *vm, u8 is_en)
Definition: session.c:1543
API main structure, used by both vpp and binary API clients.
Definition: api_common.h:203
int vnet_application_attach(vnet_app_attach_args_t *a)
Attach application to vpp.
Definition: application.c:809
static session_t * session_get_from_handle_if_valid(session_handle_t handle)
Definition: session.h:287
static void echo_clients_session_reset_callback(session_t *s)
Definition: echo_client.c:544
#define pool_free(p)
Free a pool.
Definition: pool.h:407
static u8 svm_fifo_set_event(svm_fifo_t *f)
Set fifo event flag.
Definition: svm_fifo.h:739
u32 no_copy
Don&#39;t memcpy data to tx fifo.
Definition: echo_client.h:67
u32 connections_per_batch
Connections to rx/tx at once.
Definition: echo_client.h:62
static int echo_clients_rx_callback(session_t *s)
Definition: echo_client.c:586
#define VLIB_REGISTER_NODE(x,...)
Definition: node.h:169
#define UNFORMAT_END_OF_INPUT
Definition: format.h:145
session_handle_t listener_handle
Parent listener session index if the result of an accept.
static_always_inline uword vlib_get_thread_index(void)
Definition: threads.h:213
static clib_error_t * echo_clients_attach(u8 *appns_id, u64 appns_flags, u64 appns_secret)
Definition: echo_client.c:627
static vlib_process_t * vlib_get_current_process(vlib_main_t *vm)
Definition: node_funcs.h:391
vlib_main_t * vm
Definition: buffer.c:312
vlib_node_registration_t echo_clients_node
(constructor) VLIB_REGISTER_NODE (echo_clients_node)
Definition: echo_client.c:309
#define vec_free(V)
Free vector&#39;s memory (no header).
Definition: vec.h:341
static int echo_clients_session_connected_callback(u32 app_index, u32 api_context, session_t *s, u8 is_fail)
Definition: echo_client.c:477
#define clib_warning(format, args...)
Definition: error.h:59
struct _stream_session_cb_vft session_cb_vft_t
echo_client_main_t echo_client_main
Definition: echo_client.c:23
elog_main_t elog_main
Definition: main.h:172
struct _transport_connection transport_connection_t
#define ELOG_TYPE_DECLARE(f)
Definition: elog.h:442
u32 private_segment_size
size of private fifo segs
Definition: echo_client.h:64
static int app_recv_stream(app_session_t *s, u8 *buf, u32 len)
svm_queue_t * vl_input_queue
Definition: memory_shared.h:84
static void send_data_chunk(echo_client_main_t *ecm, eclient_session_t *s)
Definition: echo_client.c:49
#define pool_init_fixed(pool, max_elts)
initialize a fixed-size, preallocated pool
Definition: pool.h:86
#define VLIB_CLI_COMMAND(x,...)
Definition: cli.h:155
#define ASSERT(truth)
u32 cli_node_index
cli process node index
Definition: echo_client.h:50
#define vec_delete(V, N, M)
Delete N elements starting at element M.
Definition: vec.h:784
u32 my_client_index
loopback API client handle
Definition: echo_client.h:51
int vnet_application_detach(vnet_app_detach_args_t *a)
Detach application from vpp.
Definition: application.c:869
static int app_send_dgram(app_session_t *s, u8 *data, u32 len, u8 noblock)
#define clib_error_report(e)
Definition: error.h:113
static void vlib_node_set_state(vlib_main_t *vm, u32 node_index, vlib_node_state_t new_state)
Set node dispatch state.
Definition: node_funcs.h:148
#define ECHO_CLIENT_DBG
Definition: echo_client.c:25
ip46_address_t rmt_ip
u16 lcl_port
local port (network order)
struct _vnet_app_detach_args_t vnet_app_detach_args_t
u8 ** rx_buf
intermediate rx buffers
Definition: echo_client.h:75
int vnet_connect(vnet_connect_args_t *a)
Definition: application.c:999
u8 thread_index
Index of the thread that allocated the session.
#define clib_atomic_fetch_add(a, b)
Definition: atomics.h:23
ip46_address_t lcl_ip
local ip
u8 prealloc_fifos
Request fifo preallocation.
Definition: echo_client.h:101
u32 ** quic_session_index_by_thread
Definition: echo_client.h:77
volatile u64 tx_total
Definition: echo_client.h:85
int parse_uri(char *uri, session_endpoint_cfg_t *sep)
volatile u64 rx_total
Definition: echo_client.h:84
#define vec_len(v)
Number of elements in vector (rvalue-only, NULL tolerant)
volatile u8 session_state
State in session layer state machine.
u8 * connect_uri
URI for slave&#39;s connect.
Definition: echo_client.h:57
#define ec_cli_output(_fmt, _args...)
Definition: echo_client.c:739
u64 uword
Definition: types.h:112
clib_spinlock_t sessions_lock
Definition: echo_client.h:74
static void * echo_client_thread_fn(void *arg)
Definition: echo_client.c:689
connectionless service
int vnet_disconnect_session(vnet_disconnect_args_t *a)
Definition: application.c:1064
static int quic_echo_clients_session_connected_callback(u32 app_index, u32 api_context, session_t *s, u8 is_fail)
Definition: echo_client.c:404
#define SESSION_ENDPOINT_CFG_NULL
Definition: session_types.h:74
int session_send_io_evt_to_thread_custom(void *data, u32 thread_index, session_evt_type_t evt_type)
Definition: session.c:86
unformat_function_t unformat_memory_size
Definition: format.h:296
static void signal_evt_to_cli_i(int *code)
Definition: echo_client.c:31
u8 * format_unformat_error(u8 *s, va_list *va)
Definition: unformat.c:91
void vlib_worker_thread_barrier_release(vlib_main_t *vm)
Definition: threads.c:1487
static vlib_thread_main_t * vlib_get_thread_main()
Definition: global_funcs.h:32
int vnet_connect_uri(vnet_connect_args_t *a)
pthread_t client_thread_handle
Definition: echo_client.h:80
int svm_fifo_dequeue_drop(svm_fifo_t *f, u32 len)
Dequeue and drop bytes from fifo.
Definition: svm_fifo.c:948
static transport_service_type_t session_transport_service_type(session_t *s)
api_main_t api_main
Definition: api_shared.c:35
static_always_inline void clib_spinlock_lock_if_init(clib_spinlock_t *p)
Definition: lock.h:93
struct session_dgram_header_ session_dgram_hdr_t
static int echo_clients_init(vlib_main_t *vm)
Definition: echo_client.c:332
uword unformat(unformat_input_t *i, const char *fmt,...)
Definition: unformat.c:978
static int app_recv_dgram(app_session_t *s, u8 *buf, u32 len)
static uword unformat_check_input(unformat_input_t *i)
Definition: format.h:171
u16 rmt_port
remote port (network order)