Add get_endpoint in transport vft
[vpp.git] / src / tests / vnet / session / quic_echo.c
1 /*
2  * Copyright (c) 2019 Cisco and/or its affiliates.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at:
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15
16 #include <stdio.h>
17 #include <signal.h>
18
19 #include <vnet/session/application_interface.h>
20 #include <svm/svm_fifo_segment.h>
21 #include <vlibmemory/api.h>
22
23 #include <vpp/api/vpe_msg_enum.h>
24
25 #define vl_typedefs             /* define message structures */
26 #include <vpp/api/vpe_all_api_h.h>
27 #undef vl_typedefs
28
29 /* declare message handlers for each api */
30
31 #define vl_endianfun            /* define message structures */
32 #include <vpp/api/vpe_all_api_h.h>
33 #undef vl_endianfun
34
35 /* instantiate all the print functions we know about */
36 #define vl_print(handle, ...)
37 #define vl_printfun
38 #include <vpp/api/vpe_all_api_h.h>
39 #undef vl_printfun
40
41 #define TCP_ECHO_DBG 0
42 #define DBG(_fmt,_args...)                      \
43     if (TCP_ECHO_DBG)                           \
44       clib_warning (_fmt, _args)
45
46 typedef struct
47 {
48   CLIB_CACHE_LINE_ALIGN_MARK (cacheline0);
49 #define _(type, name) type name;
50   foreach_app_session_field
51 #undef _
52   u64 vpp_session_handle;
53   u64 bytes_sent;
54   u64 bytes_to_send;
55   volatile u64 bytes_received;
56   volatile u64 bytes_to_receive;
57   f64 start;
58 } echo_session_t;
59
60 typedef enum
61 {
62   STATE_START,
63   STATE_ATTACHED,
64   STATE_LISTEN,
65   STATE_READY,
66   STATE_DISCONNECTING,
67   STATE_FAILED,
68   STATE_DETACHED
69 } connection_state_t;
70
71 typedef struct
72 {
73   /* vpe input queue */
74   svm_queue_t *vl_input_queue;
75
76   /* API client handle */
77   u32 my_client_index;
78
79   /* The URI we're playing with */
80   u8 *uri;
81
82   /* Session pool */
83   echo_session_t *sessions;
84
85   /* Hash table for disconnect processing */
86   uword *session_index_by_vpp_handles;
87
88   /* intermediate rx buffer */
89   u8 *rx_buf;
90
91   /* URI for slave's connect */
92   u8 *connect_uri;
93
94   u32 connected_session_index;
95
96   int i_am_master;
97
98   /* drop all packets */
99   int no_return;
100
101   /* Our event queue */
102   svm_msg_q_t *our_event_queue;
103
104   u8 *socket_name;
105
106   pid_t my_pid;
107
108   /* For deadman timers */
109   clib_time_t clib_time;
110
111   /* State of the connection, shared between msg RX thread and main thread */
112   volatile connection_state_t state;
113
114   /* Signal variables */
115   volatile int time_to_stop;
116   volatile int time_to_print_stats;
117
118   u32 configured_segment_size;
119
120   /* VNET_API_ERROR_FOO -> "Foo" hash table */
121   uword *error_string_by_error_number;
122
123   u8 *connect_test_data;
124   pthread_t *client_thread_handles;
125   u32 *thread_args;
126   u32 client_bytes_received;
127   u8 test_return_packets;
128   u64 bytes_to_send;
129   u32 fifo_size;
130
131   u32 n_clients;
132   u64 tx_total;
133   u64 rx_total;
134
135   volatile u32 n_clients_connected;
136   volatile u32 n_active_clients;
137
138
139   /** Flag that decides if socket, instead of svm, api is used to connect to
140    * vpp. If sock api is used, shm binary api is subsequently bootstrapped
141    * and all other messages are exchanged using shm IPC. */
142   u8 use_sock_api;
143
144   svm_fifo_segment_main_t segment_main;
145 } echo_main_t;
146
147 echo_main_t echo_main;
148
149 #if CLIB_DEBUG > 0
150 #define NITER 10000
151 #else
152 #define NITER 4000000
153 #endif
154
155 static u8 *
156 format_api_error (u8 * s, va_list * args)
157 {
158   echo_main_t *em = &echo_main;
159   i32 error = va_arg (*args, u32);
160   uword *p;
161
162   p = hash_get (em->error_string_by_error_number, -error);
163
164   if (p)
165     s = format (s, "%s", p[0]);
166   else
167     s = format (s, "%d", error);
168   return s;
169 }
170
171 static void
172 init_error_string_table (echo_main_t * em)
173 {
174   em->error_string_by_error_number = hash_create (0, sizeof (uword));
175
176 #define _(n,v,s) hash_set (em->error_string_by_error_number, -v, s);
177   foreach_vnet_api_error;
178 #undef _
179
180   hash_set (em->error_string_by_error_number, 99, "Misc");
181 }
182
183 static void handle_mq_event (session_event_t * e);
184
185 static int
186 wait_for_state_change (echo_main_t * em, connection_state_t state)
187 {
188   svm_msg_q_msg_t msg;
189   session_event_t *e;
190   f64 timeout;
191
192 #if CLIB_DEBUG > 0
193 #define TIMEOUT 600.0
194 #else
195 #define TIMEOUT 600.0
196 #endif
197
198   timeout = clib_time_now (&em->clib_time) + TIMEOUT;
199
200   while (clib_time_now (&em->clib_time) < timeout)
201     {
202       if (em->state == state)
203         return 0;
204       if (em->state == STATE_FAILED)
205         return -1;
206       if (em->time_to_stop == 1)
207         return 0;
208       if (!em->our_event_queue || em->state < STATE_ATTACHED)
209         continue;
210
211       if (svm_msg_q_sub (em->our_event_queue, &msg, SVM_Q_NOWAIT, 0))
212         continue;
213       e = svm_msg_q_msg_data (em->our_event_queue, &msg);
214       handle_mq_event (e);
215       svm_msg_q_free_msg (em->our_event_queue, &msg);
216     }
217   clib_warning ("timeout waiting for state %d", state);
218   return -1;
219 }
220
221 void
222 application_send_attach (echo_main_t * em)
223 {
224   vl_api_application_attach_t *bmp;
225   vl_api_application_tls_cert_add_t *cert_mp;
226   vl_api_application_tls_key_add_t *key_mp;
227
228   bmp = vl_msg_api_alloc (sizeof (*bmp));
229   clib_memset (bmp, 0, sizeof (*bmp));
230
231   bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_ATTACH);
232   bmp->client_index = em->my_client_index;
233   bmp->context = ntohl (0xfeedface);
234   bmp->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_ACCEPT_REDIRECT;
235   bmp->options[APP_OPTIONS_FLAGS] |= APP_OPTIONS_FLAGS_ADD_SEGMENT;
236   bmp->options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = 16;
237   bmp->options[APP_OPTIONS_RX_FIFO_SIZE] = em->fifo_size;
238   bmp->options[APP_OPTIONS_TX_FIFO_SIZE] = em->fifo_size;
239   bmp->options[APP_OPTIONS_ADD_SEGMENT_SIZE] = 128 << 20;
240   bmp->options[APP_OPTIONS_SEGMENT_SIZE] = 256 << 20;
241   bmp->options[APP_OPTIONS_EVT_QUEUE_SIZE] = 256;
242   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & bmp);
243
244   cert_mp = vl_msg_api_alloc (sizeof (*cert_mp) + test_srv_crt_rsa_len);
245   clib_memset (cert_mp, 0, sizeof (*cert_mp));
246   cert_mp->_vl_msg_id = ntohs (VL_API_APPLICATION_TLS_CERT_ADD);
247   cert_mp->client_index = em->my_client_index;
248   cert_mp->context = ntohl (0xfeedface);
249   cert_mp->cert_len = clib_host_to_net_u16 (test_srv_crt_rsa_len);
250   clib_memcpy_fast (cert_mp->cert, test_srv_crt_rsa, test_srv_crt_rsa_len);
251   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & cert_mp);
252
253   key_mp = vl_msg_api_alloc (sizeof (*key_mp) + test_srv_key_rsa_len);
254   clib_memset (key_mp, 0, sizeof (*key_mp) + test_srv_key_rsa_len);
255   key_mp->_vl_msg_id = ntohs (VL_API_APPLICATION_TLS_KEY_ADD);
256   key_mp->client_index = em->my_client_index;
257   key_mp->context = ntohl (0xfeedface);
258   key_mp->key_len = clib_host_to_net_u16 (test_srv_key_rsa_len);
259   clib_memcpy_fast (key_mp->key, test_srv_key_rsa, test_srv_key_rsa_len);
260   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & key_mp);
261 }
262
263 static int
264 application_attach (echo_main_t * em)
265 {
266   application_send_attach (em);
267   if (wait_for_state_change (em, STATE_ATTACHED))
268     {
269       clib_warning ("timeout waiting for STATE_ATTACHED");
270       return -1;
271     }
272   return 0;
273 }
274
275 void
276 application_detach (echo_main_t * em)
277 {
278   vl_api_application_detach_t *bmp;
279   bmp = vl_msg_api_alloc (sizeof (*bmp));
280   clib_memset (bmp, 0, sizeof (*bmp));
281
282   bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_DETACH);
283   bmp->client_index = em->my_client_index;
284   bmp->context = ntohl (0xfeedface);
285   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & bmp);
286
287   DBG ("%s", "Sent detach");
288 }
289
290 static int
291 ssvm_segment_attach (char *name, ssvm_segment_type_t type, int fd)
292 {
293   svm_fifo_segment_create_args_t _a, *a = &_a;
294   svm_fifo_segment_main_t *sm = &echo_main.segment_main;
295   int rv;
296
297   clib_memset (a, 0, sizeof (*a));
298   a->segment_name = (char *) name;
299   a->segment_type = type;
300
301   if (type == SSVM_SEGMENT_MEMFD)
302     a->memfd_fd = fd;
303
304   if ((rv = svm_fifo_segment_attach (sm, a)))
305     {
306       clib_warning ("svm_fifo_segment_attach ('%s') failed", name);
307       return rv;
308     }
309
310   vec_reset_length (a->new_segment_indices);
311   return 0;
312 }
313
314 static void
315 vl_api_application_attach_reply_t_handler (vl_api_application_attach_reply_t *
316                                            mp)
317 {
318   echo_main_t *em = &echo_main;
319   int *fds = 0;
320   u32 n_fds = 0;
321
322   if (mp->retval)
323     {
324       clib_warning ("attach failed: %U", format_api_error,
325                     clib_net_to_host_u32 (mp->retval));
326       goto failed;
327     }
328
329   if (mp->segment_name_length == 0)
330     {
331       clib_warning ("segment_name_length zero");
332       goto failed;
333     }
334
335   ASSERT (mp->app_event_queue_address);
336   em->our_event_queue = uword_to_pointer (mp->app_event_queue_address,
337                                           svm_msg_q_t *);
338
339   if (mp->n_fds)
340     {
341       vec_validate (fds, mp->n_fds);
342       vl_socket_client_recv_fd_msg (fds, mp->n_fds, 5);
343
344       if (mp->fd_flags & SESSION_FD_F_VPP_MQ_SEGMENT)
345         if (ssvm_segment_attach (0, SSVM_SEGMENT_MEMFD, fds[n_fds++]))
346           goto failed;
347
348       if (mp->fd_flags & SESSION_FD_F_MEMFD_SEGMENT)
349         if (ssvm_segment_attach ((char *) mp->segment_name,
350                                  SSVM_SEGMENT_MEMFD, fds[n_fds++]))
351           goto failed;
352
353       if (mp->fd_flags & SESSION_FD_F_MQ_EVENTFD)
354         svm_msg_q_set_consumer_eventfd (em->our_event_queue, fds[n_fds++]);
355
356       vec_free (fds);
357     }
358   else
359     {
360       if (ssvm_segment_attach ((char *) mp->segment_name, SSVM_SEGMENT_SHM,
361                                -1))
362         goto failed;
363     }
364
365   em->state = STATE_ATTACHED;
366   return;
367 failed:
368   em->state = STATE_FAILED;
369   return;
370 }
371
372 static void
373 vl_api_application_detach_reply_t_handler (vl_api_application_detach_reply_t *
374                                            mp)
375 {
376   if (mp->retval)
377     clib_warning ("detach returned with err: %d", mp->retval);
378   echo_main.state = STATE_DETACHED;
379 }
380
381 static void
382 stop_signal (int signum)
383 {
384   echo_main_t *um = &echo_main;
385   um->time_to_stop = 1;
386 }
387
388 static void
389 stats_signal (int signum)
390 {
391   echo_main_t *um = &echo_main;
392   um->time_to_print_stats = 1;
393 }
394
395 static clib_error_t *
396 setup_signal_handlers (void)
397 {
398   signal (SIGUSR2, stats_signal);
399   signal (SIGINT, stop_signal);
400   signal (SIGQUIT, stop_signal);
401   signal (SIGTERM, stop_signal);
402   return 0;
403 }
404
405 void
406 vlib_cli_output (struct vlib_main_t *vm, char *fmt, ...)
407 {
408   clib_warning ("BUG");
409 }
410
411 int
412 connect_to_vpp (char *name)
413 {
414   echo_main_t *em = &echo_main;
415   api_main_t *am = &api_main;
416
417   if (em->use_sock_api)
418     {
419       if (vl_socket_client_connect ((char *) em->socket_name, name,
420                                     0 /* default rx, tx buffer */ ))
421         {
422           clib_warning ("socket connect failed");
423           return -1;
424         }
425
426       if (vl_socket_client_init_shm (0, 1 /* want_pthread */ ))
427         {
428           clib_warning ("init shm api failed");
429           return -1;
430         }
431     }
432   else
433     {
434       if (vl_client_connect_to_vlib ("/vpe-api", name, 32) < 0)
435         {
436           clib_warning ("shmem connect failed");
437           return -1;
438         }
439     }
440   em->vl_input_queue = am->shmem_hdr->vl_input_queue;
441   em->my_client_index = am->my_client_index;
442   return 0;
443 }
444
445 void
446 disconnect_from_vpp (echo_main_t * em)
447 {
448   if (em->use_sock_api)
449     vl_socket_client_disconnect ();
450   else
451     vl_client_disconnect_from_vlib ();
452 }
453
454 static void
455 vl_api_map_another_segment_t_handler (vl_api_map_another_segment_t * mp)
456 {
457   svm_fifo_segment_main_t *sm = &echo_main.segment_main;
458   svm_fifo_segment_create_args_t _a, *a = &_a;
459   int rv;
460
461   clib_memset (a, 0, sizeof (*a));
462   a->segment_name = (char *) mp->segment_name;
463   a->segment_size = mp->segment_size;
464   /* Attach to the segment vpp created */
465   rv = svm_fifo_segment_attach (sm, a);
466   if (rv)
467     {
468       clib_warning ("svm_fifo_segment_attach ('%s') failed",
469                     mp->segment_name);
470       return;
471     }
472   clib_warning ("Mapped new segment '%s' size %d", mp->segment_name,
473                 mp->segment_size);
474 }
475
476 static void
477 session_print_stats (echo_main_t * em, echo_session_t * session)
478 {
479   f64 deltat;
480   u64 bytes;
481
482   deltat = clib_time_now (&em->clib_time) - session->start;
483   bytes = em->i_am_master ? session->bytes_received : em->bytes_to_send;
484   fformat (stdout, "Finished in %.6f\n", deltat);
485   fformat (stdout, "%.4f Gbit/second\n", (bytes * 8.0) / deltat / 1e9);
486 }
487
488 static void
489 test_recv_bytes (echo_session_t * s, u8 * rx_buf, u32 n_read)
490 {
491   int i;
492   for (i = 0; i < n_read; i++)
493     {
494       if (rx_buf[i] != ((s->bytes_received + i) & 0xff))
495         {
496           clib_warning ("error at byte %lld, 0x%x not 0x%x",
497                         s->bytes_received + i, rx_buf[i],
498                         ((s->bytes_received + i) & 0xff));
499         }
500     }
501 }
502
503 static void
504 recv_data_chunk (echo_main_t * em, echo_session_t * s, u8 * rx_buf)
505 {
506   int n_to_read, n_read;
507
508   n_to_read = svm_fifo_max_dequeue (s->rx_fifo);
509   if (!n_to_read)
510     return;
511
512   do
513     {
514       n_read = app_recv_stream ((app_session_t *) s, rx_buf,
515                                 vec_len (rx_buf));
516
517       if (n_read > 0)
518         {
519           if (em->test_return_packets)
520             test_recv_bytes (s, rx_buf, n_read);
521
522           n_to_read -= n_read;
523
524           s->bytes_received += n_read;
525           s->bytes_to_receive -= n_read;
526         }
527       else
528         break;
529     }
530   while (n_to_read > 0);
531 }
532
533 static void
534 send_data_chunk (echo_main_t * em, echo_session_t * s)
535 {
536   u64 test_buf_len, bytes_this_chunk, test_buf_offset;
537   u8 *test_data = em->connect_test_data;
538   int n_sent;
539
540   test_buf_len = vec_len (test_data);
541   test_buf_offset = s->bytes_sent % test_buf_len;
542   bytes_this_chunk = clib_min (test_buf_len - test_buf_offset,
543                                s->bytes_to_send);
544
545   n_sent = app_send_stream ((app_session_t *) s, test_data + test_buf_offset,
546                             bytes_this_chunk, 0);
547
548   if (n_sent > 0)
549     {
550       s->bytes_to_send -= n_sent;
551       s->bytes_sent += n_sent;
552     }
553 }
554
555 /*
556  * Rx/Tx polling thread per connection
557  */
558 static void *
559 client_thread_fn (void *arg)
560 {
561   echo_main_t *em = &echo_main;
562   static u8 *rx_buf = 0;
563   u32 session_index = *(u32 *) arg;
564   echo_session_t *s;
565
566   vec_validate (rx_buf, 1 << 20);
567
568   while (!em->time_to_stop && em->state != STATE_READY)
569     ;
570
571   s = pool_elt_at_index (em->sessions, session_index);
572   while (!em->time_to_stop)
573     {
574       send_data_chunk (em, s);
575       recv_data_chunk (em, s, rx_buf);
576       if (!s->bytes_to_send && !s->bytes_to_receive)
577         break;
578     }
579
580   clib_warning ("GOT OUT");
581   DBG ("session %d done", session_index);
582   em->tx_total += s->bytes_sent;
583   em->rx_total += s->bytes_received;
584   em->n_active_clients--;
585
586   pthread_exit (0);
587 }
588
589 void
590 client_send_connect (echo_main_t * em)
591 {
592   vl_api_connect_uri_t *cmp;
593   cmp = vl_msg_api_alloc (sizeof (*cmp));
594   clib_memset (cmp, 0, sizeof (*cmp));
595
596   cmp->_vl_msg_id = ntohs (VL_API_CONNECT_URI);
597   cmp->client_index = em->my_client_index;
598   cmp->context = ntohl (0xfeedface);
599   memcpy (cmp->uri, em->connect_uri, vec_len (em->connect_uri));
600   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & cmp);
601 }
602
603 void
604 client_send_disconnect (echo_main_t * em, echo_session_t * s)
605 {
606   vl_api_disconnect_session_t *dmp;
607   dmp = vl_msg_api_alloc (sizeof (*dmp));
608   clib_memset (dmp, 0, sizeof (*dmp));
609   dmp->_vl_msg_id = ntohs (VL_API_DISCONNECT_SESSION);
610   dmp->client_index = em->my_client_index;
611   dmp->handle = s->vpp_session_handle;
612   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & dmp);
613 }
614
615 int
616 client_disconnect (echo_main_t * em, echo_session_t * s)
617 {
618   client_send_disconnect (em, s);
619   pool_put (em->sessions, s);
620   clib_memset (s, 0xfe, sizeof (*s));
621   return 0;
622 }
623
624 static void
625 session_bound_handler (session_bound_msg_t * mp)
626 {
627   echo_main_t *em = &echo_main;
628
629   if (mp->retval)
630     {
631       clib_warning ("bind failed: %U", format_api_error,
632                     clib_net_to_host_u32 (mp->retval));
633       em->state = STATE_FAILED;
634       return;
635     }
636
637   clib_warning ("listening on %U:%u", format_ip46_address, mp->lcl_ip,
638                 mp->lcl_is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
639                 clib_net_to_host_u16 (mp->lcl_port));
640   em->state = STATE_READY;
641 }
642
643 static void
644 session_accepted_handler (session_accepted_msg_t * mp)
645 {
646   app_session_evt_t _app_evt, *app_evt = &_app_evt;
647   session_accepted_reply_msg_t *rmp;
648   svm_fifo_t *rx_fifo, *tx_fifo;
649   echo_main_t *em = &echo_main;
650   echo_session_t *session;
651   static f64 start_time;
652   u32 session_index;
653   u8 *ip_str;
654
655   if (start_time == 0.0)
656     start_time = clib_time_now (&em->clib_time);
657
658   ip_str = format (0, "%U", format_ip46_address, &mp->ip, mp->is_ip4);
659   clib_warning ("Accepted session from: %s:%d", ip_str,
660                 clib_net_to_host_u16 (mp->port));
661
662   /* Allocate local session and set it up */
663   pool_get (em->sessions, session);
664   session_index = session - em->sessions;
665
666   rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
667   rx_fifo->client_session_index = session_index;
668   tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
669   tx_fifo->client_session_index = session_index;
670
671   session->rx_fifo = rx_fifo;
672   session->tx_fifo = tx_fifo;
673   session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
674                                          svm_msg_q_t *);
675
676   /* Add it to lookup table */
677   hash_set (em->session_index_by_vpp_handles, mp->handle, session_index);
678
679   em->state = STATE_READY;
680
681   /* Stats printing */
682   if (pool_elts (em->sessions) && (pool_elts (em->sessions) % 20000) == 0)
683     {
684       f64 now = clib_time_now (&em->clib_time);
685       fformat (stdout, "%d active sessions in %.2f seconds, %.2f/sec...\n",
686                pool_elts (em->sessions), now - start_time,
687                (f64) pool_elts (em->sessions) / (now - start_time));
688     }
689
690   /*
691    * Send accept reply to vpp
692    */
693   app_alloc_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt,
694                              SESSION_CTRL_EVT_ACCEPTED_REPLY);
695   rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
696   rmp->handle = mp->handle;
697   rmp->context = mp->context;
698   app_send_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt);
699
700   session->bytes_received = 0;
701   session->start = clib_time_now (&em->clib_time);
702 }
703
704 static void
705 session_connected_handler (session_connected_msg_t * mp)
706 {
707   echo_main_t *em = &echo_main;
708   echo_session_t *session;
709   u32 session_index;
710   svm_fifo_t *rx_fifo, *tx_fifo;
711   int rv;
712
713   if (mp->retval)
714     {
715       clib_warning ("connection failed with code: %U", format_api_error,
716                     clib_net_to_host_u32 (mp->retval));
717       em->state = STATE_FAILED;
718       return;
719     }
720
721   /*
722    * Setup session
723    */
724
725   pool_get (em->sessions, session);
726   clib_memset (session, 0, sizeof (*session));
727   session_index = session - em->sessions;
728
729   rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
730   rx_fifo->client_session_index = session_index;
731   tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
732   tx_fifo->client_session_index = session_index;
733
734   session->rx_fifo = rx_fifo;
735   session->tx_fifo = tx_fifo;
736   session->vpp_session_handle = mp->handle;
737   session->start = clib_time_now (&em->clib_time);
738   session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
739                                          svm_msg_q_t *);
740
741   hash_set (em->session_index_by_vpp_handles, mp->handle, session_index);
742
743   /*
744    * Start RX thread
745    */
746   em->thread_args[em->n_clients_connected] = session_index;
747   rv = pthread_create (&em->client_thread_handles[em->n_clients_connected],
748                        NULL /*attr */ , client_thread_fn,
749                        (void *) &em->thread_args[em->n_clients_connected]);
750   if (rv)
751     {
752       clib_warning ("pthread_create returned %d", rv);
753       return;
754     }
755
756   em->n_clients_connected += 1;
757   clib_warning ("session %u (0x%llx) connected with local ip %U port %d",
758                 session_index, mp->handle, format_ip46_address, mp->lcl_ip,
759                 mp->is_ip4, clib_net_to_host_u16 (mp->lcl_port));
760 }
761
762 static void
763 session_disconnected_handler (session_disconnected_msg_t * mp)
764 {
765   app_session_evt_t _app_evt, *app_evt = &_app_evt;
766   session_disconnected_reply_msg_t *rmp;
767   echo_main_t *em = &echo_main;
768   echo_session_t *session = 0;
769   uword *p;
770   int rv = 0;
771
772   p = hash_get (em->session_index_by_vpp_handles, mp->handle);
773   if (!p)
774     {
775       clib_warning ("couldn't find session key %llx", mp->handle);
776       return;
777     }
778
779   session = pool_elt_at_index (em->sessions, p[0]);
780   hash_unset (em->session_index_by_vpp_handles, mp->handle);
781   pool_put (em->sessions, session);
782
783   app_alloc_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt,
784                              SESSION_CTRL_EVT_DISCONNECTED_REPLY);
785   rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
786   rmp->retval = rv;
787   rmp->handle = mp->handle;
788   rmp->context = mp->context;
789   app_send_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt);
790
791   session_print_stats (em, session);
792 }
793
794 static void
795 session_reset_handler (session_reset_msg_t * mp)
796 {
797   app_session_evt_t _app_evt, *app_evt = &_app_evt;
798   echo_main_t *em = &echo_main;
799   session_reset_reply_msg_t *rmp;
800   echo_session_t *session = 0;
801   uword *p;
802   int rv = 0;
803
804   p = hash_get (em->session_index_by_vpp_handles, mp->handle);
805
806   if (p)
807     {
808       session = pool_elt_at_index (em->sessions, p[0]);
809       clib_warning ("got reset");
810       /* Cleanup later */
811       em->time_to_stop = 1;
812     }
813   else
814     {
815       clib_warning ("couldn't find session key %llx", mp->handle);
816       return;
817     }
818
819   app_alloc_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt,
820                              SESSION_CTRL_EVT_RESET_REPLY);
821   rmp = (session_reset_reply_msg_t *) app_evt->evt->data;
822   rmp->retval = rv;
823   rmp->handle = mp->handle;
824   app_send_ctrl_evt_to_vpp (session->vpp_evt_q, app_evt);
825 }
826
827 static void
828 handle_mq_event (session_event_t * e)
829 {
830   switch (e->event_type)
831     {
832     case SESSION_CTRL_EVT_BOUND:
833       session_bound_handler ((session_bound_msg_t *) e->data);
834       break;
835     case SESSION_CTRL_EVT_ACCEPTED:
836       session_accepted_handler ((session_accepted_msg_t *) e->data);
837       break;
838     case SESSION_CTRL_EVT_CONNECTED:
839       session_connected_handler ((session_connected_msg_t *) e->data);
840       break;
841     case SESSION_CTRL_EVT_DISCONNECTED:
842       session_disconnected_handler ((session_disconnected_msg_t *) e->data);
843       break;
844     case SESSION_CTRL_EVT_RESET:
845       session_reset_handler ((session_reset_msg_t *) e->data);
846       break;
847     default:
848       clib_warning ("unhandled %u", e->event_type);
849     }
850 }
851
852 static void
853 clients_run (echo_main_t * em)
854 {
855   f64 start_time, deltat, timeout = 100.0;
856   svm_msg_q_msg_t msg;
857   session_event_t *e;
858   echo_session_t *s;
859   int i;
860
861   /* Init test data */
862   vec_validate (em->connect_test_data, 1024 * 1024 - 1);
863   for (i = 0; i < vec_len (em->connect_test_data); i++)
864     em->connect_test_data[i] = i & 0xff;
865
866   /*
867    * Attach and connect the clients
868    */
869   if (application_attach (em))
870     return;
871
872   for (i = 0; i < em->n_clients; i++)
873     client_send_connect (em);
874
875   start_time = clib_time_now (&em->clib_time);
876   while (em->n_clients_connected < em->n_clients
877          && (clib_time_now (&em->clib_time) - start_time < timeout)
878          && em->state != STATE_FAILED && em->time_to_stop != 1)
879
880     {
881       int rc = svm_msg_q_sub (em->our_event_queue, &msg, SVM_Q_TIMEDWAIT, 1);
882       if (rc == ETIMEDOUT && em->time_to_stop)
883         break;
884       if (rc == ETIMEDOUT)
885         continue;
886       e = svm_msg_q_msg_data (em->our_event_queue, &msg);
887       handle_mq_event (e);
888       svm_msg_q_free_msg (em->our_event_queue, &msg);
889     }
890
891   if (em->n_clients_connected != em->n_clients)
892     {
893       clib_warning ("failed to initialize all connections");
894       return;
895     }
896
897   /*
898    * Initialize connections
899    */
900   for (i = 0; i < em->n_clients; i++)
901     {
902       s = pool_elt_at_index (em->sessions, i);
903       s->bytes_to_send = em->bytes_to_send;
904       if (!em->no_return)
905         s->bytes_to_receive = em->bytes_to_send;
906     }
907   em->n_active_clients = em->n_clients_connected;
908
909   /*
910    * Wait for client threads to send the data
911    */
912   start_time = clib_time_now (&em->clib_time);
913   em->state = STATE_READY;
914   while (em->n_active_clients)
915     if (!svm_msg_q_is_empty (em->our_event_queue))
916       {
917         if (svm_msg_q_sub (em->our_event_queue, &msg, SVM_Q_TIMEDWAIT, 0))
918           {
919             clib_warning ("svm msg q returned");
920             continue;
921           }
922         e = svm_msg_q_msg_data (em->our_event_queue, &msg);
923         if (e->event_type != FIFO_EVENT_APP_RX)
924           handle_mq_event (e);
925         svm_msg_q_free_msg (em->our_event_queue, &msg);
926       }
927
928   for (i = 0; i < em->n_clients; i++)
929     {
930       s = pool_elt_at_index (em->sessions, i);
931       client_disconnect (em, s);
932     }
933
934   /*
935    * Stats and detach
936    */
937   deltat = clib_time_now (&em->clib_time) - start_time;
938   fformat (stdout, "%lld bytes (%lld mbytes, %lld gbytes) in %.2f seconds\n",
939            em->tx_total, em->tx_total / (1ULL << 20),
940            em->tx_total / (1ULL << 30), deltat);
941   fformat (stdout, "%.4f Gbit/second\n", (em->tx_total * 8.0) / deltat / 1e9);
942
943   application_detach (em);
944 }
945
946 static void
947 vl_api_bind_uri_reply_t_handler (vl_api_bind_uri_reply_t * mp)
948 {
949   echo_main_t *em = &echo_main;
950
951   if (mp->retval)
952     {
953       clib_warning ("bind failed: %U", format_api_error,
954                     clib_net_to_host_u32 (mp->retval));
955       em->state = STATE_FAILED;
956       return;
957     }
958
959   em->state = STATE_READY;
960 }
961
962 static void
963 vl_api_unbind_uri_reply_t_handler (vl_api_unbind_uri_reply_t * mp)
964 {
965   echo_main_t *em = &echo_main;
966
967   if (mp->retval != 0)
968     clib_warning ("returned %d", ntohl (mp->retval));
969
970   em->state = STATE_START;
971 }
972
973 u8 *
974 format_ip4_address (u8 * s, va_list * args)
975 {
976   u8 *a = va_arg (*args, u8 *);
977   return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
978 }
979
980 u8 *
981 format_ip6_address (u8 * s, va_list * args)
982 {
983   ip6_address_t *a = va_arg (*args, ip6_address_t *);
984   u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
985
986   i_max_n_zero = ARRAY_LEN (a->as_u16);
987   max_n_zeros = 0;
988   i_first_zero = i_max_n_zero;
989   n_zeros = 0;
990   for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
991     {
992       u32 is_zero = a->as_u16[i] == 0;
993       if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
994         {
995           i_first_zero = i;
996           n_zeros = 0;
997         }
998       n_zeros += is_zero;
999       if ((!is_zero && n_zeros > max_n_zeros)
1000           || (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
1001         {
1002           i_max_n_zero = i_first_zero;
1003           max_n_zeros = n_zeros;
1004           i_first_zero = ARRAY_LEN (a->as_u16);
1005           n_zeros = 0;
1006         }
1007     }
1008
1009   last_double_colon = 0;
1010   for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
1011     {
1012       if (i == i_max_n_zero && max_n_zeros > 1)
1013         {
1014           s = format (s, "::");
1015           i += max_n_zeros - 1;
1016           last_double_colon = 1;
1017         }
1018       else
1019         {
1020           s = format (s, "%s%x",
1021                       (last_double_colon || i == 0) ? "" : ":",
1022                       clib_net_to_host_u16 (a->as_u16[i]));
1023           last_double_colon = 0;
1024         }
1025     }
1026
1027   return s;
1028 }
1029
1030 /* Format an IP46 address. */
1031 u8 *
1032 format_ip46_address (u8 * s, va_list * args)
1033 {
1034   ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
1035   ip46_type_t type = va_arg (*args, ip46_type_t);
1036   int is_ip4 = 1;
1037
1038   switch (type)
1039     {
1040     case IP46_TYPE_ANY:
1041       is_ip4 = ip46_address_is_ip4 (ip46);
1042       break;
1043     case IP46_TYPE_IP4:
1044       is_ip4 = 1;
1045       break;
1046     case IP46_TYPE_IP6:
1047       is_ip4 = 0;
1048       break;
1049     }
1050
1051   return is_ip4 ?
1052     format (s, "%U", format_ip4_address, &ip46->ip4) :
1053     format (s, "%U", format_ip6_address, &ip46->ip6);
1054 }
1055
1056 static void
1057 server_handle_rx (echo_main_t * em, session_event_t * e)
1058 {
1059   int n_read, max_dequeue, n_sent;
1060   u32 offset, to_dequeue;
1061   echo_session_t *s;
1062
1063   s = pool_elt_at_index (em->sessions, e->session_index);
1064
1065   /* Clear event only once. Otherwise, if we do it in the loop by calling
1066    * app_recv_stream, we may end up with a lot of unhandled rx events on the
1067    * message queue */
1068   svm_fifo_unset_event (s->rx_fifo);
1069
1070   max_dequeue = svm_fifo_max_dequeue (s->rx_fifo);
1071   if (PREDICT_FALSE (!max_dequeue))
1072     return;
1073
1074   do
1075     {
1076       /* The options here are to limit ourselves to max_dequeue or read
1077        * even the data that was enqueued while we were dequeueing and which
1078        * now has an rx event in the mq. Either of the two work. */
1079       to_dequeue = clib_min (max_dequeue, vec_len (em->rx_buf));
1080       n_read = app_recv_stream_raw (s->rx_fifo, em->rx_buf, to_dequeue,
1081                                     0 /* clear evt */ , 0 /* peek */ );
1082       if (n_read > 0)
1083         {
1084           max_dequeue -= n_read;
1085           s->bytes_received += n_read;
1086         }
1087       else
1088         break;
1089
1090       /* Reflect if a non-drop session */
1091       if (!em->no_return && n_read > 0)
1092         {
1093           offset = 0;
1094           do
1095             {
1096               n_sent = app_send_stream ((app_session_t *) s,
1097                                         &em->rx_buf[offset],
1098                                         n_read, SVM_Q_WAIT);
1099               if (n_sent > 0)
1100                 {
1101                   n_read -= n_sent;
1102                   offset += n_sent;
1103                 }
1104             }
1105           while ((n_sent <= 0 || n_read > 0) && !em->time_to_stop);
1106         }
1107     }
1108   while (max_dequeue > 0 && !em->time_to_stop);
1109 }
1110
1111 static void
1112 server_handle_mq (echo_main_t * em)
1113 {
1114   svm_msg_q_msg_t msg;
1115   session_event_t *e;
1116
1117   while (1)
1118     {
1119       int rc = svm_msg_q_sub (em->our_event_queue, &msg, SVM_Q_TIMEDWAIT, 1);
1120       if (PREDICT_FALSE (rc == ETIMEDOUT && em->time_to_stop))
1121         break;
1122       if (PREDICT_FALSE (em->time_to_print_stats == 1))
1123         {
1124           em->time_to_print_stats = 0;
1125           fformat (stdout, "%d connections\n", pool_elts (em->sessions));
1126         }
1127       if (rc == ETIMEDOUT)
1128         continue;
1129       e = svm_msg_q_msg_data (em->our_event_queue, &msg);
1130       clib_warning ("Event %d", e->event_type);
1131       switch (e->event_type)
1132         {
1133         case FIFO_EVENT_APP_RX:
1134           server_handle_rx (em, e);
1135           break;
1136         default:
1137           handle_mq_event (e);
1138           break;
1139         }
1140       svm_msg_q_free_msg (em->our_event_queue, &msg);
1141     }
1142 }
1143
1144 void
1145 server_send_listen (echo_main_t * em)
1146 {
1147   vl_api_bind_uri_t *bmp;
1148   bmp = vl_msg_api_alloc (sizeof (*bmp));
1149   clib_memset (bmp, 0, sizeof (*bmp));
1150
1151   bmp->_vl_msg_id = ntohs (VL_API_BIND_URI);
1152   bmp->client_index = em->my_client_index;
1153   bmp->context = ntohl (0xfeedface);
1154   memcpy (bmp->uri, em->uri, vec_len (em->uri));
1155   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & bmp);
1156 }
1157
1158 int
1159 server_listen (echo_main_t * em)
1160 {
1161   server_send_listen (em);
1162   if (wait_for_state_change (em, STATE_READY))
1163     {
1164       clib_warning ("timeout waiting for STATE_READY");
1165       return -1;
1166     }
1167   return 0;
1168 }
1169
1170 void
1171 server_send_unbind (echo_main_t * em)
1172 {
1173   vl_api_unbind_uri_t *ump;
1174
1175   ump = vl_msg_api_alloc (sizeof (*ump));
1176   clib_memset (ump, 0, sizeof (*ump));
1177
1178   ump->_vl_msg_id = ntohs (VL_API_UNBIND_URI);
1179   ump->client_index = em->my_client_index;
1180   memcpy (ump->uri, em->uri, vec_len (em->uri));
1181   vl_msg_api_send_shmem (em->vl_input_queue, (u8 *) & ump);
1182 }
1183
1184 void
1185 server_run (echo_main_t * em)
1186 {
1187   echo_session_t *session;
1188   int i;
1189
1190   /* $$$$ hack preallocation */
1191   for (i = 0; i < 200000; i++)
1192     {
1193       pool_get (em->sessions, session);
1194       clib_memset (session, 0, sizeof (*session));
1195     }
1196   for (i = 0; i < 200000; i++)
1197     pool_put_index (em->sessions, i);
1198
1199   if (application_attach (em))
1200     return;
1201
1202   /* Bind to uri */
1203   if (server_listen (em))
1204     return;
1205
1206   /* Enter handle event loop */
1207   server_handle_mq (em);
1208
1209   /* Cleanup */
1210   server_send_unbind (em);
1211
1212   application_detach (em);
1213
1214   fformat (stdout, "Test complete...\n");
1215 }
1216
1217 static void
1218 vl_api_disconnect_session_reply_t_handler (vl_api_disconnect_session_reply_t *
1219                                            mp)
1220 {
1221   echo_main_t *em = &echo_main;
1222   uword *p;
1223
1224   if (mp->retval)
1225     {
1226       clib_warning ("vpp complained about disconnect: %d",
1227                     ntohl (mp->retval));
1228       return;
1229     }
1230
1231   em->state = STATE_START;
1232
1233   p = hash_get (em->session_index_by_vpp_handles, mp->handle);
1234   if (p)
1235     {
1236       hash_unset (em->session_index_by_vpp_handles, mp->handle);
1237     }
1238   else
1239     {
1240       clib_warning ("couldn't find session key %llx", mp->handle);
1241     }
1242 }
1243
1244 static void
1245   vl_api_application_tls_cert_add_reply_t_handler
1246   (vl_api_application_tls_cert_add_reply_t * mp)
1247 {
1248   if (mp->retval)
1249     clib_warning ("failed to add tls cert");
1250 }
1251
1252 static void
1253   vl_api_application_tls_key_add_reply_t_handler
1254   (vl_api_application_tls_key_add_reply_t * mp)
1255 {
1256   if (mp->retval)
1257     clib_warning ("failed to add tls key");
1258 }
1259
1260 #define foreach_quic_echo_msg                                           \
1261 _(BIND_URI_REPLY, bind_uri_reply)                                       \
1262 _(UNBIND_URI_REPLY, unbind_uri_reply)                                   \
1263 _(DISCONNECT_SESSION_REPLY, disconnect_session_reply)                   \
1264 _(APPLICATION_ATTACH_REPLY, application_attach_reply)                   \
1265 _(APPLICATION_DETACH_REPLY, application_detach_reply)                   \
1266 _(MAP_ANOTHER_SEGMENT, map_another_segment)                             \
1267 _(APPLICATION_TLS_CERT_ADD_REPLY, application_tls_cert_add_reply)       \
1268 _(APPLICATION_TLS_KEY_ADD_REPLY, application_tls_key_add_reply)         \
1269
1270 void
1271 quic_echo_api_hookup (echo_main_t * em)
1272 {
1273 #define _(N,n)                                                  \
1274     vl_msg_api_set_handlers(VL_API_##N, #n,                     \
1275                            vl_api_##n##_t_handler,              \
1276                            vl_noop_handler,                     \
1277                            vl_api_##n##_t_endian,               \
1278                            vl_api_##n##_t_print,                \
1279                            sizeof(vl_api_##n##_t), 1);
1280   foreach_quic_echo_msg;
1281 #undef _
1282 }
1283
1284 int
1285 main (int argc, char **argv)
1286 {
1287   int i_am_server = 1, test_return_packets = 0;
1288   echo_main_t *em = &echo_main;
1289   svm_fifo_segment_main_t *sm = &em->segment_main;
1290   unformat_input_t _argv, *a = &_argv;
1291   u8 *chroot_prefix;
1292   u8 *uri = 0;
1293   u8 *bind_uri = (u8 *) "quic://0.0.0.0/1234";
1294   u8 *connect_uri = (u8 *) "quic://6.0.1.1/1234";
1295   u64 bytes_to_send = 64 << 10, mbytes;
1296   char *app_name;
1297   u32 tmp;
1298
1299   clib_mem_init_thread_safe (0, 256 << 20);
1300
1301   clib_memset (em, 0, sizeof (*em));
1302   em->session_index_by_vpp_handles = hash_create (0, sizeof (uword));
1303   em->my_pid = getpid ();
1304   em->configured_segment_size = 1 << 20;
1305   em->socket_name = 0;
1306   em->use_sock_api = 1;
1307   em->fifo_size = 64 << 10;
1308   em->n_clients = 1;
1309
1310   clib_time_init (&em->clib_time);
1311   init_error_string_table (em);
1312   svm_fifo_segment_main_init (sm, HIGH_SEGMENT_BASEVA, 20);
1313   unformat_init_command_line (a, argv);
1314
1315   while (unformat_check_input (a) != UNFORMAT_END_OF_INPUT)
1316     {
1317       if (unformat (a, "chroot prefix %s", &chroot_prefix))
1318         {
1319           vl_set_memory_root_path ((char *) chroot_prefix);
1320         }
1321       else if (unformat (a, "uri %s", &uri))
1322         ;
1323       else if (unformat (a, "segment-size %dM", &tmp))
1324         em->configured_segment_size = tmp << 20;
1325       else if (unformat (a, "segment-size %dG", &tmp))
1326         em->configured_segment_size = tmp << 30;
1327       else if (unformat (a, "server"))
1328         i_am_server = 1;
1329       else if (unformat (a, "client"))
1330         i_am_server = 0;
1331       else if (unformat (a, "no-return"))
1332         em->no_return = 1;
1333       else if (unformat (a, "test"))
1334         test_return_packets = 1;
1335       else if (unformat (a, "bytes %lld", &mbytes))
1336         {
1337           bytes_to_send = mbytes;
1338         }
1339       else if (unformat (a, "mbytes %lld", &mbytes))
1340         {
1341           bytes_to_send = mbytes << 20;
1342         }
1343       else if (unformat (a, "gbytes %lld", &mbytes))
1344         {
1345           bytes_to_send = mbytes << 30;
1346         }
1347       else if (unformat (a, "socket-name %s", &em->socket_name))
1348         ;
1349       else if (unformat (a, "use-svm-api"))
1350         em->use_sock_api = 0;
1351       else if (unformat (a, "fifo-size %d", &tmp))
1352         em->fifo_size = tmp << 10;
1353       else if (unformat (a, "nclients %d", &em->n_clients))
1354         ;
1355       else
1356         {
1357           fformat (stderr, "%s: usage [master|slave]\n", argv[0]);
1358           exit (1);
1359         }
1360     }
1361
1362   if (!em->socket_name)
1363     em->socket_name = format (0, "%s%c", API_SOCKET_FILE, 0);
1364
1365   if (uri)
1366     {
1367       em->uri = format (0, "%s%c", uri, 0);
1368       em->connect_uri = format (0, "%s%c", uri, 0);
1369     }
1370   else
1371     {
1372       em->uri = format (0, "%s%c", bind_uri, 0);
1373       em->connect_uri = format (0, "%s%c", connect_uri, 0);
1374     }
1375
1376   em->i_am_master = i_am_server;
1377   em->test_return_packets = test_return_packets;
1378   em->bytes_to_send = bytes_to_send;
1379   em->time_to_stop = 0;
1380   vec_validate (em->rx_buf, 4 << 20);
1381   vec_validate (em->client_thread_handles, em->n_clients - 1);
1382   vec_validate (em->thread_args, em->n_clients - 1);
1383
1384   setup_signal_handlers ();
1385   quic_echo_api_hookup (em);
1386
1387   app_name = i_am_server ? "quic_echo_server" : "quic_echo_client";
1388   if (connect_to_vpp (app_name) < 0)
1389     {
1390       svm_region_exit ();
1391       fformat (stderr, "Couldn't connect to vpe, exiting...\n");
1392       exit (1);
1393     }
1394
1395   if (i_am_server == 0)
1396     clients_run (em);
1397   else
1398     server_run (em);
1399
1400   /* Make sure detach finishes */
1401   wait_for_state_change (em, STATE_DETACHED);
1402
1403   disconnect_from_vpp (em);
1404   exit (0);
1405 }
1406
1407 /*
1408  * fd.io coding-style-patch-verification: ON
1409  *
1410  * Local Variables:
1411  * eval: (c-set-style "gnu")
1412  * End:
1413  */