blob: 9a4bea8fe5173be33f6783065e9aa34db49617d4 [file] [log] [blame]
Dave Wallace543852a2017-08-03 02:11:34 -04001/*
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
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 <stdlib.h>
18#include <signal.h>
19#include <svm/svm_fifo_segment.h>
20#include <vlibmemory/api.h>
21#include <vpp/api/vpe_msg_enum.h>
22#include <vnet/session/application_interface.h>
23#include <uri/vppcom.h>
24#include <vlib/unix/unix.h>
25#include <vppinfra/vec_bootstrap.h>
26
27#define vl_typedefs /* define message structures */
28#include <vpp/api/vpe_all_api_h.h>
29#undef vl_typedefs
30
31/* declare message handlers for each api */
32
33#define vl_endianfun /* define message structures */
34#include <vpp/api/vpe_all_api_h.h>
35#undef vl_endianfun
36
37/* instantiate all the print functions we know about */
38#define vl_print(handle, ...)
39#define vl_printfun
40#include <vpp/api/vpe_all_api_h.h>
41#undef vl_printfun
42
43#if (CLIB_DEBUG > 0)
44/* Set VPPCOM_DEBUG 2 for connection debug, 3 for read/write debug output */
45#define VPPCOM_DEBUG 1
46#else
47#define VPPCOM_DEBUG 0
48#endif
49
50/*
51 * VPPCOM Private definitions and functions.
52 */
53typedef enum
54{
55 STATE_APP_START,
56 STATE_APP_CONN_VPP,
57 STATE_APP_ENABLED,
58 STATE_APP_ATTACHED,
59} app_state_t;
60
61typedef enum
62{
63 STATE_START,
64 STATE_CONNECT,
65 STATE_LISTEN,
66 STATE_ACCEPT,
67 STATE_DISCONNECT,
68 STATE_FAILED
69} session_state_t;
70
71typedef struct
72{
73 volatile session_state_t state;
74
75 svm_fifo_t *server_rx_fifo;
76 svm_fifo_t *server_tx_fifo;
77 u32 sm_seg_index;
78 u64 vpp_session_handle;
79 unix_shared_memory_queue_t *event_queue;
80
81 /* Socket configuration state */
82 u8 is_server;
83 u8 is_listen;
84 u8 is_cut_thru;
85 u8 is_nonblocking;
86 u32 vrf;
87 u8 is_ip4;
88 u8 ip[16];
89 u16 port;
90 u8 proto;
91 u64 client_queue_address;
92 u64 options[16];
93} session_t;
94
95typedef struct vppcom_cfg_t_
96{
97 u64 heapsize;
98 u64 segment_baseva;
99 u32 segment_size;
100 u32 add_segment_size;
101 u32 preallocated_fifo_pairs;
102 u32 rx_fifo_size;
103 u32 tx_fifo_size;
104 u32 event_queue_size;
105 u32 listen_queue_size;
106 f64 app_timeout;
107 f64 session_timeout;
108 f64 accept_timeout;
109} vppcom_cfg_t;
110
111typedef struct vppcom_main_t_
112{
113 u8 init;
114 u32 *client_session_index_fifo;
115 volatile u32 bind_session_index;
116 u32 tx_event_id;
117 int main_cpu;
118
119 /* vpe input queue */
120 unix_shared_memory_queue_t *vl_input_queue;
121
122 /* API client handle */
123 u32 my_client_index;
124
125 /* Session pool */
126 clib_spinlock_t sessions_lockp;
127 session_t *sessions;
128
129 /* Hash table for disconnect processing */
130 uword *session_index_by_vpp_handles;
131
132 /* Select bitmaps */
133 clib_bitmap_t *rd_bitmap;
134 clib_bitmap_t *wr_bitmap;
135 clib_bitmap_t *ex_bitmap;
136
137 /* Our event queue */
138 unix_shared_memory_queue_t *app_event_queue;
139
140 /* unique segment name counter */
141 u32 unique_segment_index;
142
143 pid_t my_pid;
144
145 /* For deadman timers */
146 clib_time_t clib_time;
147
148 /* State of the connection, shared between msg RX thread and main thread */
149 volatile app_state_t app_state;
150
151 vppcom_cfg_t cfg;
152
153 /* VNET_API_ERROR_FOO -> "Foo" hash table */
154 uword *error_string_by_error_number;
155} vppcom_main_t;
156
157vppcom_main_t vppcom_main = {.my_client_index = ~0 };
158
159static const char *
160vppcom_app_state_str (app_state_t state)
161{
162 char *st;
163
164 switch (state)
165 {
166 case STATE_APP_START:
167 st = "STATE_APP_START";
168 break;
169
170 case STATE_APP_CONN_VPP:
171 st = "STATE_APP_CONN_VPP";
172 break;
173
174 case STATE_APP_ENABLED:
175 st = "STATE_APP_ENABLED";
176 break;
177
178 case STATE_APP_ATTACHED:
179 st = "STATE_APP_ATTACHED";
180 break;
181
182 default:
183 st = "UNKNOWN_APP_STATE";
184 break;
185 }
186
187 return st;
188}
189
190static const char *
191vppcom_session_state_str (session_state_t state)
192{
193 char *st;
194
195 switch (state)
196 {
197 case STATE_START:
198 st = "STATE_START";
199 break;
200
201 case STATE_CONNECT:
202 st = "STATE_CONNECT";
203 break;
204
205 case STATE_LISTEN:
206 st = "STATE_LISTEN";
207 break;
208
209 case STATE_ACCEPT:
210 st = "STATE_ACCEPT";
211 break;
212
213 case STATE_DISCONNECT:
214 st = "STATE_DISCONNECT";
215 break;
216
217 case STATE_FAILED:
218 st = "STATE_FAILED";
219 break;
220
221 default:
222 st = "UNKNOWN_STATE";
223 break;
224 }
225
226 return st;
227}
228
229/*
230 * VPPCOM Utility Functions
231 */
232static inline int
233vppcom_session_at_index (u32 session_index, session_t * volatile *sess)
234{
235 vppcom_main_t *vcm = &vppcom_main;
236
237 /* Assumes that caller has acquired spinlock: vcm->sessions_lockp */
238 if (PREDICT_FALSE ((session_index == ~0) ||
239 pool_is_free_index (vcm->sessions, session_index)))
240 {
241 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
242 vcm->my_pid, session_index);
243 return VPPCOM_EBADFD;
244 }
245 *sess = pool_elt_at_index (vcm->sessions, session_index);
246 return VPPCOM_OK;
247}
248
249static int
250vppcom_connect_to_vpp (char *app_name)
251{
252 api_main_t *am = &api_main;
253 vppcom_main_t *vcm = &vppcom_main;
254
255 if (VPPCOM_DEBUG > 0)
256 printf ("\nConnecting to VPP api...");
257 if (vl_client_connect_to_vlib ("/vpe-api", app_name, 32) < 0)
258 {
259 clib_warning ("[%d] connect to vpp (%s) failed!",
260 vcm->my_pid, app_name);
261 return VPPCOM_ECONNREFUSED;
262 }
263
264 vcm->vl_input_queue = am->shmem_hdr->vl_input_queue;
265 vcm->my_client_index = am->my_client_index;
266 if (VPPCOM_DEBUG > 0)
267 printf (" connected!\n");
268
269 vcm->app_state = STATE_APP_CONN_VPP;
270 return VPPCOM_OK;
271}
272
273static u8 *
274format_api_error (u8 * s, va_list * args)
275{
276 vppcom_main_t *vcm = &vppcom_main;
277 i32 error = va_arg (*args, u32);
278 uword *p;
279
280 p = hash_get (vcm->error_string_by_error_number, -error);
281
282 if (p)
283 s = format (s, "%s (%d)", p[0], error);
284 else
285 s = format (s, "%d", error);
286 return s;
287}
288
289static void
290vppcom_init_error_string_table (void)
291{
292 vppcom_main_t *vcm = &vppcom_main;
293
294 vcm->error_string_by_error_number = hash_create (0, sizeof (uword));
295
296#define _(n,v,s) hash_set (vcm->error_string_by_error_number, -v, s);
297 foreach_vnet_api_error;
298#undef _
299
300 hash_set (vcm->error_string_by_error_number, 99, "Misc");
301}
302
303static inline int
304vppcom_wait_for_app_state_change (app_state_t app_state)
305{
306 vppcom_main_t *vcm = &vppcom_main;
307 f64 timeout = clib_time_now (&vcm->clib_time) + vcm->cfg.app_timeout;
308
309 while (clib_time_now (&vcm->clib_time) < timeout)
310 {
311 if (vcm->app_state == app_state)
312 return VPPCOM_OK;
313 }
314 if (VPPCOM_DEBUG > 0)
315 clib_warning ("[%d] timeout waiting for state %s (%d)", vcm->my_pid,
316 vppcom_app_state_str (app_state), app_state);
317 return VPPCOM_ETIMEDOUT;
318}
319
320static inline int
321vppcom_wait_for_session_state_change (u32 session_index,
322 session_state_t state,
323 f64 wait_for_time)
324{
325 vppcom_main_t *vcm = &vppcom_main;
326 f64 timeout = clib_time_now (&vcm->clib_time) + wait_for_time;
327 session_t *volatile session;
328 int rv;
329
330 do
331 {
332 clib_spinlock_lock (&vcm->sessions_lockp);
333 rv = vppcom_session_at_index (session_index, &session);
334 if (PREDICT_FALSE (rv))
335 {
336 clib_spinlock_unlock (&vcm->sessions_lockp);
337 return rv;
338 }
339 if (session->state == state)
340 {
341 clib_spinlock_unlock (&vcm->sessions_lockp);
342 return VPPCOM_OK;
343 }
344 clib_spinlock_unlock (&vcm->sessions_lockp);
345 }
346 while (clib_time_now (&vcm->clib_time) < timeout);
347
348 if (VPPCOM_DEBUG > 0)
349 clib_warning ("[%d] timeout waiting for state %s (%d)", vcm->my_pid,
350 vppcom_session_state_str (state), state);
351 return VPPCOM_ETIMEDOUT;
352}
353
354static inline int
355vppcom_wait_for_client_session_index (f64 wait_for_time)
356{
357 vppcom_main_t *vcm = &vppcom_main;
358 f64 timeout = clib_time_now (&vcm->clib_time) + wait_for_time;
359
360 do
361 {
362 if (clib_fifo_elts (vcm->client_session_index_fifo))
363 return VPPCOM_OK;
364 }
365 while (clib_time_now (&vcm->clib_time) < timeout);
366
367 if (wait_for_time == 0)
368 return VPPCOM_EAGAIN;
369
370 if (VPPCOM_DEBUG > 0)
371 clib_warning ("[%d] timeout waiting for client_session_index",
372 vcm->my_pid);
373 return VPPCOM_ETIMEDOUT;
374}
375
376/*
377 * VPP-API message functions
378 */
379static void
380vppcom_send_session_enable_disable (u8 is_enable)
381{
382 vppcom_main_t *vcm = &vppcom_main;
383 vl_api_session_enable_disable_t *bmp;
384 bmp = vl_msg_api_alloc (sizeof (*bmp));
385 memset (bmp, 0, sizeof (*bmp));
386
387 bmp->_vl_msg_id = ntohs (VL_API_SESSION_ENABLE_DISABLE);
388 bmp->client_index = vcm->my_client_index;
389 bmp->context = htonl (0xfeedface);
390 bmp->is_enable = is_enable;
391 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & bmp);
392}
393
394static int
395vppcom_app_session_enable (void)
396{
397 vppcom_main_t *vcm = &vppcom_main;
398 int rv;
399
400 if (vcm->app_state != STATE_APP_ENABLED)
401 {
402 vppcom_send_session_enable_disable (1 /* is_enabled == TRUE */ );
403 rv = vppcom_wait_for_app_state_change (STATE_APP_ENABLED);
404 if (PREDICT_FALSE (rv))
405 {
406 if (VPPCOM_DEBUG > 0)
407 clib_warning ("[%d] Session enable timed out, rv = %s (%d)",
408 vcm->my_pid, vppcom_retval_str (rv), rv);
409 return rv;
410 }
411 }
412 return VPPCOM_OK;
413}
414
415static void
416 vl_api_session_enable_disable_reply_t_handler
417 (vl_api_session_enable_disable_reply_t * mp)
418{
419 vppcom_main_t *vcm = &vppcom_main;
420
421 if (mp->retval)
422 {
423 clib_warning ("[%d] session_enable_disable failed: %U", vcm->my_pid,
424 format_api_error, ntohl (mp->retval));
425 }
426 else
427 vcm->app_state = STATE_APP_ENABLED;
428}
429
430static void
431vppcom_app_send_attach (void)
432{
433 vppcom_main_t *vcm = &vppcom_main;
434 vl_api_application_attach_t *bmp;
435 bmp = vl_msg_api_alloc (sizeof (*bmp));
436 memset (bmp, 0, sizeof (*bmp));
437
438 bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_ATTACH);
439 bmp->client_index = vcm->my_client_index;
440 bmp->context = htonl (0xfeedface);
441 bmp->options[APP_OPTIONS_FLAGS] =
442 APP_OPTIONS_FLAGS_USE_FIFO | APP_OPTIONS_FLAGS_ADD_SEGMENT;
443 bmp->options[SESSION_OPTIONS_SEGMENT_SIZE] = vcm->cfg.segment_size;
444 bmp->options[SESSION_OPTIONS_ADD_SEGMENT_SIZE] = vcm->cfg.add_segment_size;
445 bmp->options[SESSION_OPTIONS_RX_FIFO_SIZE] = vcm->cfg.rx_fifo_size;
446 bmp->options[SESSION_OPTIONS_TX_FIFO_SIZE] = vcm->cfg.tx_fifo_size;
447 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & bmp);
448}
449
450static int
451vppcom_app_attach (void)
452{
453 vppcom_main_t *vcm = &vppcom_main;
454 int rv;
455
456 vppcom_app_send_attach ();
457 rv = vppcom_wait_for_app_state_change (STATE_APP_ATTACHED);
458 if (PREDICT_FALSE (rv))
459 {
460 if (VPPCOM_DEBUG > 0)
461 clib_warning ("[%d] application attach timed out, rv = %s (%d)",
462 vcm->my_pid, vppcom_retval_str (rv), rv);
463 return rv;
464 }
465 return VPPCOM_OK;
466}
467
468static void
469vppcom_app_detach (void)
470{
471 vppcom_main_t *vcm = &vppcom_main;
472 vl_api_application_detach_t *bmp;
473 bmp = vl_msg_api_alloc (sizeof (*bmp));
474 memset (bmp, 0, sizeof (*bmp));
475
476 bmp->_vl_msg_id = ntohs (VL_API_APPLICATION_DETACH);
477 bmp->client_index = vcm->my_client_index;
478 bmp->context = htonl (0xfeedface);
479 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & bmp);
480}
481
482static void
483vl_api_application_attach_reply_t_handler (vl_api_application_attach_reply_t *
484 mp)
485{
486 vppcom_main_t *vcm = &vppcom_main;
487 static svm_fifo_segment_create_args_t _a;
488 svm_fifo_segment_create_args_t *a = &_a;
489 int rv;
490
491 memset (a, 0, sizeof (*a));
492 if (mp->retval)
493 {
494 clib_warning ("[%d] attach failed: %U", vcm->my_pid,
495 format_api_error, ntohl (mp->retval));
496 return;
497 }
498
499 if (mp->segment_name_length == 0)
500 {
501 clib_warning ("[%d] segment_name_length zero", vcm->my_pid);
502 return;
503 }
504
505 a->segment_name = (char *) mp->segment_name;
506 a->segment_size = mp->segment_size;
507
508 ASSERT (mp->app_event_queue_address);
509
510 /* Attach to the segment vpp created */
511 rv = svm_fifo_segment_attach (a);
512 vec_reset_length (a->new_segment_indices);
513 if (PREDICT_FALSE (rv))
514 {
515 clib_warning ("[%d] svm_fifo_segment_attach ('%s') failed", vcm->my_pid,
516 mp->segment_name);
517 return;
518 }
519
520 vcm->app_event_queue =
521 uword_to_pointer (mp->app_event_queue_address,
522 unix_shared_memory_queue_t *);
523
524 vcm->app_state = STATE_APP_ATTACHED;
525}
526
527static void
528vl_api_application_detach_reply_t_handler (vl_api_application_detach_reply_t *
529 mp)
530{
531 vppcom_main_t *vcm = &vppcom_main;
532
533 if (mp->retval)
534 clib_warning ("[%d] detach failed: %U", vcm->my_pid, format_api_error,
535 ntohl (mp->retval));
536
537 vcm->app_state = STATE_APP_ENABLED;
538}
539
540static void
541vl_api_disconnect_session_reply_t_handler (vl_api_disconnect_session_reply_t *
542 mp)
543{
544 vppcom_main_t *vcm = &vppcom_main;
545 uword *p;
546
547 p = hash_get (vcm->session_index_by_vpp_handles, mp->handle);
548 if (p)
549 {
550 session_t *session = 0;
551 int rv;
552 clib_spinlock_lock (&vcm->sessions_lockp);
553 rv = vppcom_session_at_index (p[0], &session);
554 if (PREDICT_FALSE (rv))
555 {
556 if (VPPCOM_DEBUG > 1)
557 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
558 vcm->my_pid, p[0]);
559 }
560 hash_unset (vcm->session_index_by_vpp_handles, mp->handle);
561 session->state = STATE_DISCONNECT;
562 clib_spinlock_unlock (&vcm->sessions_lockp);
563 }
564 else
565 {
566 if (VPPCOM_DEBUG > 1)
567 clib_warning ("[%d] couldn't find session key %llx", vcm->my_pid,
568 mp->handle);
569 }
570
571 if (mp->retval)
572 clib_warning ("[%d] disconnect_session failed: %U", vcm->my_pid,
573 format_api_error, ntohl (mp->retval));
574}
575
576static void
577vl_api_map_another_segment_t_handler (vl_api_map_another_segment_t * mp)
578{
579 vppcom_main_t *vcm = &vppcom_main;
580 static svm_fifo_segment_create_args_t _a;
581 svm_fifo_segment_create_args_t *a = &_a;
582 int rv;
583
584 memset (a, 0, sizeof (*a));
585 a->segment_name = (char *) mp->segment_name;
586 a->segment_size = mp->segment_size;
587 /* Attach to the segment vpp created */
588 rv = svm_fifo_segment_attach (a);
589 vec_reset_length (a->new_segment_indices);
590 if (PREDICT_FALSE (rv))
591 {
592 clib_warning ("[%d] svm_fifo_segment_attach ('%s') failed",
593 vcm->my_pid, mp->segment_name);
594 return;
595 }
596 if (VPPCOM_DEBUG > 1)
597 clib_warning ("[%d] mapped new segment '%s' size %d", vcm->my_pid,
598 mp->segment_name, mp->segment_size);
599}
600
601static void
602vl_api_disconnect_session_t_handler (vl_api_disconnect_session_t * mp)
603{
604 vppcom_main_t *vcm = &vppcom_main;
605 session_t *session = 0;
606 vl_api_disconnect_session_reply_t *rmp;
607 uword *p;
608 int rv = 0;
609
610 p = hash_get (vcm->session_index_by_vpp_handles, mp->handle);
611 if (p)
612 {
613 int rval;
614 clib_spinlock_lock (&vcm->sessions_lockp);
615 rval = vppcom_session_at_index (p[0], &session);
616 if (PREDICT_FALSE (rval))
617 {
618 if (VPPCOM_DEBUG > 1)
619 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
620 vcm->my_pid, p[0]);
621 }
622 else
623 pool_put (vcm->sessions, session);
624 clib_spinlock_unlock (&vcm->sessions_lockp);
625 hash_unset (vcm->session_index_by_vpp_handles, mp->handle);
626 }
627 else
628 {
629 clib_warning ("[%d] couldn't find session key %llx", vcm->my_pid,
630 mp->handle);
631 rv = -11;
632 }
633
634 rmp = vl_msg_api_alloc (sizeof (*rmp));
635 memset (rmp, 0, sizeof (*rmp));
636
637 rmp->_vl_msg_id = ntohs (VL_API_DISCONNECT_SESSION_REPLY);
638 rmp->retval = htonl (rv);
639 rmp->handle = mp->handle;
640 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & rmp);
641}
642
643static void
644vl_api_reset_session_t_handler (vl_api_reset_session_t * mp)
645{
646 vppcom_main_t *vcm = &vppcom_main;
647 session_t *session = 0;
648 vl_api_reset_session_reply_t *rmp;
649 uword *p;
650 int rv = 0;
651
652 p = hash_get (vcm->session_index_by_vpp_handles, mp->handle);
653 if (p)
654 {
655 int rval;
656 clib_spinlock_lock (&vcm->sessions_lockp);
657 rval = vppcom_session_at_index (p[0], &session);
658 if (PREDICT_FALSE (rval))
659 {
660 if (VPPCOM_DEBUG > 1)
661 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
662 vcm->my_pid, p[0]);
663 }
664 else
665 pool_put (vcm->sessions, session);
666 clib_spinlock_unlock (&vcm->sessions_lockp);
667 hash_unset (vcm->session_index_by_vpp_handles, mp->handle);
668 }
669 else
670 {
671 clib_warning ("[%d] couldn't find session key %llx", vcm->my_pid,
672 mp->handle);
673 rv = -11;
674 }
675
676 rmp = vl_msg_api_alloc (sizeof (*rmp));
677 memset (rmp, 0, sizeof (*rmp));
678 rmp->_vl_msg_id = ntohs (VL_API_RESET_SESSION_REPLY);
679 rmp->retval = htonl (rv);
680 rmp->handle = mp->handle;
681 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & rmp);
682}
683
684static void
685vl_api_connect_sock_reply_t_handler (vl_api_connect_sock_reply_t * mp)
686{
687 vppcom_main_t *vcm = &vppcom_main;
688 session_t *session;
689 u32 session_index;
690 svm_fifo_t *rx_fifo, *tx_fifo;
691 u8 is_cut_thru = 0;
692 int rv;
693
694 if (mp->retval)
695 {
696 clib_warning ("[%d] connect failed: %U", vcm->my_pid, format_api_error,
697 ntohl (mp->retval));
698 return;
699 }
700
701 session_index = ntohl (mp->app_connect);
702 if (VPPCOM_DEBUG > 1)
703 clib_warning ("[%d] app_connect = %d 0x%08x", vcm->my_pid,
704 session_index, session_index);
705
706 clib_spinlock_lock (&vcm->sessions_lockp);
707 if (pool_is_free_index (vcm->sessions, session_index))
708 {
709 clib_spinlock_unlock (&vcm->sessions_lockp);
710 if (VPPCOM_DEBUG > 1)
711 clib_warning ("[%d] invalid session, sid %d is closed!",
712 vcm->my_pid, session_index);
713 return;
714 }
715
716 /* We've been redirected */
717 if (mp->segment_name_length > 0)
718 {
719 static svm_fifo_segment_create_args_t _a;
720 svm_fifo_segment_create_args_t *a = &_a;
721
722 is_cut_thru = 1;
723 memset (a, 0, sizeof (*a));
724 a->segment_name = (char *) mp->segment_name;
725 if (VPPCOM_DEBUG > 1)
726 clib_warning ("[%d] cut-thru segment: %s", vcm->my_pid,
727 a->segment_name);
728 rv = svm_fifo_segment_attach (a);
729 vec_reset_length (a->new_segment_indices);
730 if (PREDICT_FALSE (rv))
731 {
732 clib_warning ("[%d] sm_fifo_segment_attach ('%s') failed",
733 vcm->my_pid, a->segment_name);
734 return;
735 }
736 }
737
738 /*
739 * Setup session
740 */
741 if (VPPCOM_DEBUG > 1)
742 clib_warning ("[%d] client sid %d", vcm->my_pid, session_index);
743
744 session = pool_elt_at_index (vcm->sessions, session_index);
745 session->is_cut_thru = is_cut_thru;
746 session->event_queue = uword_to_pointer (mp->vpp_event_queue_address,
747 unix_shared_memory_queue_t *);
748
749 rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
750 rx_fifo->client_session_index = session_index;
751 tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
752 tx_fifo->client_session_index = session_index;
753
754 session->server_rx_fifo = rx_fifo;
755 session->server_tx_fifo = tx_fifo;
756 session->vpp_session_handle = mp->handle;
757 session->state = STATE_CONNECT;
758
759 /* Add it to lookup table */
760 hash_set (vcm->session_index_by_vpp_handles, mp->handle, session_index);
761 clib_spinlock_unlock (&vcm->sessions_lockp);
762}
763
764static void
765vppcom_send_connect_sock (session_t * session, u32 session_index)
766{
767 vppcom_main_t *vcm = &vppcom_main;
768 vl_api_connect_sock_t *cmp;
769
770 /* Assumes caller as acquired the spinlock: vcm->sessions_lockp */
771 session->is_server = 0;
772 cmp = vl_msg_api_alloc (sizeof (*cmp));
773 memset (cmp, 0, sizeof (*cmp));
774 cmp->_vl_msg_id = ntohs (VL_API_CONNECT_SOCK);
775 cmp->client_index = vcm->my_client_index;
776 cmp->context = htonl (0xfeedface);
777 cmp->app_connect = session_index;
778
779 if (VPPCOM_DEBUG > 1)
780 clib_warning ("[%d] session_index = %d 0x%08x, app_connect = %d 0x%08x",
781 vcm->my_pid, session_index, session_index,
782 cmp->app_connect, cmp->app_connect);
783
784 cmp->vrf = session->vrf;
785 cmp->is_ip4 = session->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -0400786 clib_memcpy (cmp->ip, session->ip, sizeof (cmp->ip));
Dave Wallace543852a2017-08-03 02:11:34 -0400787 cmp->port = session->port;
788 cmp->proto = session->proto;
789 clib_memcpy (cmp->options, session->options, sizeof (cmp->options));
790 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & cmp);
791}
792
793static int
794vppcom_send_disconnect (u32 session_index)
795{
796 vppcom_main_t *vcm = &vppcom_main;
797 vl_api_disconnect_session_t *dmp;
798 session_t *session = 0;
799 int rv;
800
801 clib_spinlock_lock (&vcm->sessions_lockp);
802 rv = vppcom_session_at_index (session_index, &session);
803 if (PREDICT_FALSE (rv))
804 {
805 clib_spinlock_unlock (&vcm->sessions_lockp);
806 if (VPPCOM_DEBUG > 1)
807 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
808 vcm->my_pid, session_index);
809 return rv;
810 }
811
812 dmp = vl_msg_api_alloc (sizeof (*dmp));
813 memset (dmp, 0, sizeof (*dmp));
814 dmp->_vl_msg_id = ntohs (VL_API_DISCONNECT_SESSION);
815 dmp->client_index = vcm->my_client_index;
816 dmp->handle = session->vpp_session_handle;
817 clib_spinlock_unlock (&vcm->sessions_lockp);
818 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & dmp);
819 return VPPCOM_OK;
820}
821
822static void
823vl_api_bind_sock_reply_t_handler (vl_api_bind_sock_reply_t * mp)
824{
825 vppcom_main_t *vcm = &vppcom_main;
826 session_t *session = 0;
827 int rv;
828
829 if (mp->retval)
830 clib_warning ("[%d] bind failed: %U", vcm->my_pid, format_api_error,
831 ntohl (mp->retval));
832
833 ASSERT (vcm->bind_session_index != ~0);
834
835 clib_spinlock_lock (&vcm->sessions_lockp);
836 rv = vppcom_session_at_index (vcm->bind_session_index, &session);
837 if (rv == VPPCOM_OK)
838 {
839 session->vpp_session_handle = mp->handle;
840 hash_set (vcm->session_index_by_vpp_handles, mp->handle,
841 vcm->bind_session_index);
842 session->state = mp->retval ? STATE_FAILED : STATE_LISTEN;
843 vcm->bind_session_index = ~0;
844 }
845 clib_spinlock_unlock (&vcm->sessions_lockp);
846}
847
848static void
849vl_api_unbind_sock_reply_t_handler (vl_api_unbind_sock_reply_t * mp)
850{
851 vppcom_main_t *vcm = &vppcom_main;
852 session_t *session = 0;
853 int rv;
854
855 clib_spinlock_lock (&vcm->sessions_lockp);
856 rv = vppcom_session_at_index (vcm->bind_session_index, &session);
857 if (PREDICT_FALSE (rv))
858 {
859 if (VPPCOM_DEBUG > 1)
860 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
861 vcm->my_pid, vcm->bind_session_index);
862 }
863
864 if (mp->retval)
865 clib_warning ("[%d] unbind failed: %U", vcm->my_pid, format_api_error,
866 ntohl (mp->retval));
867
868 vcm->bind_session_index = ~0;
869 session->state = STATE_START;
870 clib_spinlock_unlock (&vcm->sessions_lockp);
871}
872
873u8 *
874format_ip4_address (u8 * s, va_list * args)
875{
876 u8 *a = va_arg (*args, u8 *);
877 return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
878}
879
880u8 *
881format_ip6_address (u8 * s, va_list * args)
882{
883 ip6_address_t *a = va_arg (*args, ip6_address_t *);
884 u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
885
886 i_max_n_zero = ARRAY_LEN (a->as_u16);
887 max_n_zeros = 0;
888 i_first_zero = i_max_n_zero;
889 n_zeros = 0;
890 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
891 {
892 u32 is_zero = a->as_u16[i] == 0;
893 if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
894 {
895 i_first_zero = i;
896 n_zeros = 0;
897 }
898 n_zeros += is_zero;
899 if ((!is_zero && n_zeros > max_n_zeros)
900 || (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
901 {
902 i_max_n_zero = i_first_zero;
903 max_n_zeros = n_zeros;
904 i_first_zero = ARRAY_LEN (a->as_u16);
905 n_zeros = 0;
906 }
907 }
908
909 last_double_colon = 0;
910 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
911 {
912 if (i == i_max_n_zero && max_n_zeros > 1)
913 {
914 s = format (s, "::");
915 i += max_n_zeros - 1;
916 last_double_colon = 1;
917 }
918 else
919 {
920 s = format (s, "%s%x",
921 (last_double_colon || i == 0) ? "" : ":",
922 clib_net_to_host_u16 (a->as_u16[i]));
923 last_double_colon = 0;
924 }
925 }
926
927 return s;
928}
929
930/* Format an IP46 address. */
931u8 *
932format_ip46_address (u8 * s, va_list * args)
933{
934 ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
935 ip46_type_t type = va_arg (*args, ip46_type_t);
936 int is_ip4 = 1;
937
938 switch (type)
939 {
940 case IP46_TYPE_ANY:
941 is_ip4 = ip46_address_is_ip4 (ip46);
942 break;
943 case IP46_TYPE_IP4:
944 is_ip4 = 1;
945 break;
946 case IP46_TYPE_IP6:
947 is_ip4 = 0;
948 break;
949 }
950
951 return is_ip4 ?
952 format (s, "%U", format_ip4_address, &ip46->ip4) :
953 format (s, "%U", format_ip6_address, &ip46->ip6);
954}
955
956static void
957vl_api_accept_session_t_handler (vl_api_accept_session_t * mp)
958{
959 vppcom_main_t *vcm = &vppcom_main;
960 vl_api_accept_session_reply_t *rmp;
961 svm_fifo_t *rx_fifo, *tx_fifo;
962 session_t *session;
963 u32 session_index;
964 int rv = 0;
965
966 if (!clib_fifo_free_elts (vcm->client_session_index_fifo))
967 {
968 clib_warning ("[%d] client session queue is full!", vcm->my_pid);
969 rv = VNET_API_ERROR_QUEUE_FULL;
970 goto send_reply;
971 }
972
973 if (VPPCOM_DEBUG > 1)
974 {
975 u8 *ip_str = format (0, "%U", format_ip46_address, &mp->ip, mp->is_ip4);
976 clib_warning ("[%d] accepted session from: %s:%d", vcm->my_pid, ip_str,
977 clib_net_to_host_u16 (mp->port));
978 vec_free (ip_str);
979 }
980
981 clib_spinlock_lock (&vcm->sessions_lockp);
982 /* Allocate local session and set it up */
983 pool_get (vcm->sessions, session);
984 memset (session, 0, sizeof (*session));
985 session_index = session - vcm->sessions;
986
987 rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
988 rx_fifo->client_session_index = session_index;
989 tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
990 tx_fifo->client_session_index = session_index;
991
992 session->server_rx_fifo = rx_fifo;
993 session->server_tx_fifo = tx_fifo;
994 session->event_queue = uword_to_pointer (mp->vpp_event_queue_address,
995 unix_shared_memory_queue_t *);
996 session->state = STATE_ACCEPT;
997 session->is_cut_thru = 0;
998 session->port = ntohs (mp->port);
999 session->is_ip4 = mp->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001000 clib_memcpy (session->ip, mp->ip, sizeof (session->ip));
Dave Wallace543852a2017-08-03 02:11:34 -04001001 clib_spinlock_unlock (&vcm->sessions_lockp);
1002
1003 /* Add it to lookup table */
1004 hash_set (vcm->session_index_by_vpp_handles, mp->handle, session_index);
1005
1006 clib_fifo_add1 (vcm->client_session_index_fifo, session_index);
1007
1008 /*
1009 * Send accept reply to vpp
1010 */
1011send_reply:
1012 rmp = vl_msg_api_alloc (sizeof (*rmp));
1013 memset (rmp, 0, sizeof (*rmp));
1014 rmp->_vl_msg_id = ntohs (VL_API_ACCEPT_SESSION_REPLY);
1015 rmp->retval = htonl (rv);
1016 rmp->handle = mp->handle;
1017 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & rmp);
1018}
1019
1020/*
1021 * Acting as server for redirected connect requests
1022 */
1023static void
1024vl_api_connect_sock_t_handler (vl_api_connect_sock_t * mp)
1025{
1026 static svm_fifo_segment_create_args_t _a;
1027 svm_fifo_segment_create_args_t *a = &_a;
1028 vppcom_main_t *vcm = &vppcom_main;
1029 u32 session_index;
1030 svm_fifo_segment_private_t *seg;
1031 unix_shared_memory_queue_t *client_q;
1032 vl_api_connect_sock_reply_t *rmp;
1033 session_t *session = 0;
1034 int rv = 0;
1035 svm_fifo_t *rx_fifo;
1036 svm_fifo_t *tx_fifo;
1037 unix_shared_memory_queue_t *event_q = 0;
1038
1039 if (!clib_fifo_free_elts (vcm->client_session_index_fifo))
1040 {
1041 if (VPPCOM_DEBUG > 1)
1042 clib_warning ("[%d] client session queue is full!", vcm->my_pid);
1043 rv = VNET_API_ERROR_QUEUE_FULL;
1044 goto send_reply;
1045 }
1046
1047 /* Create the segment */
1048 memset (a, 0, sizeof (*a));
1049 a->segment_name = (char *) format ((u8 *) a->segment_name, "%d:segment%d%c",
1050 vcm->my_pid, vcm->unique_segment_index++,
1051 0);
1052 a->segment_size = vcm->cfg.segment_size;
1053 a->preallocated_fifo_pairs = vcm->cfg.preallocated_fifo_pairs;
1054 a->rx_fifo_size = vcm->cfg.rx_fifo_size;
1055 a->tx_fifo_size = vcm->cfg.tx_fifo_size;
1056
1057 rv = svm_fifo_segment_create (a);
1058 if (PREDICT_FALSE (rv))
1059 {
1060 if (VPPCOM_DEBUG > 1)
1061 clib_warning ("[%d] svm_fifo_segment_create ('%s') failed",
1062 vcm->my_pid, a->segment_name);
1063 vec_reset_length (a->new_segment_indices);
1064 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1065 goto send_reply;
1066 }
1067
1068 if (VPPCOM_DEBUG > 1)
1069 clib_warning ("[%d] created segment '%s'", vcm->my_pid, a->segment_name);
1070
1071 clib_spinlock_lock (&vcm->sessions_lockp);
1072 pool_get (vcm->sessions, session);
1073 memset (session, 0, sizeof (*session));
1074 session_index = session - vcm->sessions;
1075
1076 session->sm_seg_index = a->new_segment_indices[0];
1077 vec_reset_length (a->new_segment_indices);
1078
1079 seg = svm_fifo_get_segment (session->sm_seg_index);
1080 rx_fifo = session->server_rx_fifo =
1081 svm_fifo_segment_alloc_fifo (seg, vcm->cfg.rx_fifo_size,
1082 FIFO_SEGMENT_RX_FREELIST);
1083 if (PREDICT_FALSE (!session->server_rx_fifo))
1084 {
1085 svm_fifo_segment_delete (seg);
1086 clib_warning ("[%d] rx fifo alloc failed, size %ld (0x%lx)",
1087 vcm->my_pid, vcm->cfg.rx_fifo_size,
1088 vcm->cfg.rx_fifo_size);
1089 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1090 clib_spinlock_unlock (&vcm->sessions_lockp);
1091 goto send_reply;
1092 }
1093
1094 tx_fifo = session->server_tx_fifo =
1095 svm_fifo_segment_alloc_fifo (seg, vcm->cfg.tx_fifo_size,
1096 FIFO_SEGMENT_TX_FREELIST);
1097 if (PREDICT_FALSE (!session->server_tx_fifo))
1098 {
1099 svm_fifo_segment_delete (seg);
1100 if (VPPCOM_DEBUG > 1)
1101 clib_warning ("[%d] tx fifo alloc failed, size %ld (0x%lx)",
1102 vcm->my_pid, vcm->cfg.tx_fifo_size,
1103 vcm->cfg.tx_fifo_size);
1104 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1105 clib_spinlock_unlock (&vcm->sessions_lockp);
1106 goto send_reply;
1107 }
1108
1109 session->server_rx_fifo->master_session_index = session_index;
1110 session->server_tx_fifo->master_session_index = session_index;
1111 session->client_queue_address = mp->client_queue_address;
1112 session->is_cut_thru = 1;
1113 session->is_server = 1;
1114 session->is_ip4 = mp->is_ip4;
1115 session->port = mp->port;
1116 {
1117 void *oldheap;
1118 ssvm_shared_header_t *sh = seg->ssvm.sh;
1119
1120 ssvm_lock_non_recursive (sh, 1);
1121 oldheap = ssvm_push_heap (sh);
1122 event_q = session->event_queue =
1123 unix_shared_memory_queue_init (vcm->cfg.event_queue_size,
1124 sizeof (session_fifo_event_t),
1125 vcm->my_pid, 0 /* signal not sent */ );
1126 ssvm_pop_heap (oldheap);
1127 ssvm_unlock_non_recursive (sh);
1128 }
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001129 clib_memcpy (session->ip, mp->ip, sizeof (session->ip));
1130
Dave Wallace543852a2017-08-03 02:11:34 -04001131 session->state = STATE_ACCEPT;
1132 if (VPPCOM_DEBUG > 1)
1133 clib_warning ("[%d] Connected cut-thru to client: sid %d",
1134 vcm->my_pid, session_index);
1135 clib_spinlock_unlock (&vcm->sessions_lockp);
1136 clib_fifo_add1 (vcm->client_session_index_fifo, session_index);
1137
1138send_reply:
1139 rmp = vl_msg_api_alloc (sizeof (*rmp));
1140 memset (rmp, 0, sizeof (*rmp));
1141
1142 rmp->_vl_msg_id = ntohs (VL_API_CONNECT_SOCK_REPLY);
1143 rmp->context = mp->context;
1144 rmp->app_connect = htonl (mp->app_connect);
1145 rmp->retval = htonl (rv);
1146 rmp->segment_name_length = vec_len (a->segment_name);
1147 clib_memcpy (rmp->segment_name, a->segment_name, vec_len (a->segment_name));
1148 vec_reset_length (a->segment_name);
1149
1150 if (event_q)
1151 {
1152 rmp->vpp_event_queue_address = pointer_to_uword (event_q);
1153 rmp->server_rx_fifo = pointer_to_uword (rx_fifo);
1154 rmp->server_tx_fifo = pointer_to_uword (tx_fifo);
1155 }
1156 client_q =
1157 uword_to_pointer (mp->client_queue_address, unix_shared_memory_queue_t *);
1158
1159 ASSERT (client_q);
1160 vl_msg_api_send_shmem (client_q, (u8 *) & rmp);
1161}
1162
1163static void
1164vppcom_send_bind_sock (session_t * session)
1165{
1166 vppcom_main_t *vcm = &vppcom_main;
1167 vl_api_bind_sock_t *bmp;
1168
1169 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
1170 session->is_server = 1;
1171 bmp = vl_msg_api_alloc (sizeof (*bmp));
1172 memset (bmp, 0, sizeof (*bmp));
1173
1174 bmp->_vl_msg_id = ntohs (VL_API_BIND_SOCK);
1175 bmp->client_index = vcm->my_client_index;
1176 bmp->context = htonl (0xfeedface);
1177 bmp->vrf = session->vrf;
1178 bmp->is_ip4 = session->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001179 clib_memcpy (bmp->ip, session->ip, sizeof (bmp->ip));
Dave Wallace543852a2017-08-03 02:11:34 -04001180 bmp->port = session->port;
1181 bmp->proto = session->proto;
1182 clib_memcpy (bmp->options, session->options, sizeof (bmp->options));
1183 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & bmp);
1184}
1185
1186static void
1187vppcom_send_unbind_sock (u32 session_index)
1188{
1189 vppcom_main_t *vcm = &vppcom_main;
1190 vl_api_unbind_sock_t *ump;
1191 session_t *session = 0;
1192 int rv;
1193
1194 clib_spinlock_lock (&vcm->sessions_lockp);
1195 rv = vppcom_session_at_index (session_index, &session);
1196 if (PREDICT_FALSE (rv))
1197 {
1198 clib_spinlock_unlock (&vcm->sessions_lockp);
1199 if (VPPCOM_DEBUG > 0)
1200 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1201 vcm->my_pid, session_index);
1202 return;
1203 }
1204
1205 ump = vl_msg_api_alloc (sizeof (*ump));
1206 memset (ump, 0, sizeof (*ump));
1207
1208 ump->_vl_msg_id = ntohs (VL_API_UNBIND_SOCK);
1209 ump->client_index = vcm->my_client_index;
1210 ump->handle = session->vpp_session_handle;
1211 clib_spinlock_unlock (&vcm->sessions_lockp);
1212 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & ump);
1213}
1214
1215static int
1216vppcom_session_unbind_cut_thru (session_t * session)
1217{
1218 svm_fifo_segment_main_t *sm = &svm_fifo_segment_main;
1219 svm_fifo_segment_private_t *seg;
1220 int rv = VPPCOM_OK;
1221
1222 seg = vec_elt_at_index (sm->segments, session->sm_seg_index);
1223 svm_fifo_segment_free_fifo (seg, session->server_rx_fifo,
1224 FIFO_SEGMENT_RX_FREELIST);
1225 svm_fifo_segment_free_fifo (seg, session->server_tx_fifo,
1226 FIFO_SEGMENT_TX_FREELIST);
1227 svm_fifo_segment_delete (seg);
1228
1229 return rv;
1230}
1231
1232static int
1233vppcom_session_unbind (u32 session_index)
1234{
1235 vppcom_main_t *vcm = &vppcom_main;
1236 int rv;
1237
1238 clib_spinlock_lock (&vcm->sessions_lockp);
1239 if (PREDICT_FALSE (pool_is_free_index (vcm->sessions, session_index)))
1240 {
1241 clib_spinlock_unlock (&vcm->sessions_lockp);
1242 if (VPPCOM_DEBUG > 1)
1243 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1244 vcm->my_pid, session_index);
1245 return VPPCOM_EBADFD;
1246 }
1247 clib_spinlock_unlock (&vcm->sessions_lockp);
1248
1249 vcm->bind_session_index = session_index;
1250 vppcom_send_unbind_sock (session_index);
1251 rv = vppcom_wait_for_session_state_change (session_index, STATE_START,
1252 vcm->cfg.session_timeout);
1253 if (PREDICT_FALSE (rv))
1254 {
1255 vcm->bind_session_index = ~0;
1256 if (VPPCOM_DEBUG > 0)
1257 clib_warning ("[%d] server unbind timed out, rv = %s (%d)",
1258 vcm->my_pid, vppcom_retval_str (rv), rv);
1259 return rv;
1260 }
1261 return VPPCOM_OK;
1262}
1263
1264static int
1265vppcom_session_disconnect (u32 session_index)
1266{
1267 vppcom_main_t *vcm = &vppcom_main;
1268 int rv;
1269
1270 rv = vppcom_send_disconnect (session_index);
1271 if (PREDICT_FALSE (rv))
1272 return rv;
1273
1274 rv = vppcom_wait_for_session_state_change (session_index, STATE_DISCONNECT,
1275 vcm->cfg.session_timeout);
1276 if (PREDICT_FALSE (rv))
1277 {
1278 if (VPPCOM_DEBUG > 0)
1279 clib_warning ("[%d] client disconnect timed out, rv = %s (%d)",
1280 vcm->my_pid, vppcom_retval_str (rv), rv);
1281 return rv;
1282 }
1283
1284 clib_spinlock_lock (&vcm->sessions_lockp);
1285 pool_put_index (vcm->sessions, session_index);
1286 clib_spinlock_unlock (&vcm->sessions_lockp);
1287 return VPPCOM_OK;
1288}
1289
1290#define foreach_sock_msg \
1291_(SESSION_ENABLE_DISABLE_REPLY, session_enable_disable_reply) \
1292_(BIND_SOCK_REPLY, bind_sock_reply) \
1293_(UNBIND_SOCK_REPLY, unbind_sock_reply) \
1294_(ACCEPT_SESSION, accept_session) \
1295_(CONNECT_SOCK, connect_sock) \
1296_(CONNECT_SOCK_REPLY, connect_sock_reply) \
1297_(DISCONNECT_SESSION, disconnect_session) \
1298_(DISCONNECT_SESSION_REPLY, disconnect_session_reply) \
1299_(RESET_SESSION, reset_session) \
1300_(APPLICATION_ATTACH_REPLY, application_attach_reply) \
1301_(APPLICATION_DETACH_REPLY, application_detach_reply) \
1302_(MAP_ANOTHER_SEGMENT, map_another_segment)
1303
1304static void
1305vppcom_api_hookup (void)
1306{
1307#define _(N,n) \
1308 vl_msg_api_set_handlers(VL_API_##N, #n, \
1309 vl_api_##n##_t_handler, \
1310 vl_noop_handler, \
1311 vl_api_##n##_t_endian, \
1312 vl_api_##n##_t_print, \
1313 sizeof(vl_api_##n##_t), 1);
1314 foreach_sock_msg;
1315#undef _
1316}
1317
1318static void
1319vppcom_cfg_init (vppcom_cfg_t * vcl_cfg)
1320{
1321 ASSERT (vcl_cfg);
1322
1323 vcl_cfg->heapsize = (256ULL << 20);
1324 vcl_cfg->segment_baseva = 0x200000000ULL;
1325 vcl_cfg->segment_size = (256 << 20);
1326 vcl_cfg->add_segment_size = (128 << 20);
1327 vcl_cfg->preallocated_fifo_pairs = 8;
1328 vcl_cfg->rx_fifo_size = (1 << 20);
1329 vcl_cfg->tx_fifo_size = (1 << 20);
1330 vcl_cfg->event_queue_size = 2048;
1331 vcl_cfg->listen_queue_size = CLIB_CACHE_LINE_BYTES / sizeof (u32);
1332 vcl_cfg->app_timeout = 10 * 60.0;
1333 vcl_cfg->session_timeout = 10 * 60.0;
1334 vcl_cfg->accept_timeout = 60.0;
1335}
1336
1337static void
1338vppcom_cfg_heapsize (char *conf_fname)
1339{
1340 vppcom_main_t *vcm = &vppcom_main;
1341 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1342 FILE *fp;
1343 char inbuf[4096];
1344 int argc = 1;
1345 char **argv = NULL;
1346 char *arg = NULL;
1347 char *p;
1348 int i;
1349 u8 *sizep;
1350 u32 size;
1351
1352 fp = fopen (conf_fname, "r");
1353 if (fp == NULL)
1354 {
1355 if (VPPCOM_DEBUG > 0)
1356 fprintf (stderr, "open configuration file '%s' failed\n", conf_fname);
1357 goto defaulted;
1358 }
1359 argv = calloc (1, sizeof (char *));
1360 if (argv == NULL)
1361 goto defaulted;
1362
1363 while (1)
1364 {
1365 if (fgets (inbuf, 4096, fp) == 0)
1366 break;
1367 p = strtok (inbuf, " \t\n");
1368 while (p != NULL)
1369 {
1370 if (*p == '#')
1371 break;
1372 argc++;
1373 char **tmp = realloc (argv, argc * sizeof (char *));
1374 if (tmp == NULL)
1375 {
1376 fclose (fp);
1377 goto defaulted;
1378 }
1379 argv = tmp;
1380 arg = strndup (p, 1024);
1381 if (arg == NULL)
1382 {
1383 fclose (fp);
1384 goto defaulted;
1385 }
1386 argv[argc - 1] = arg;
1387 p = strtok (NULL, " \t\n");
1388 }
1389 }
1390
1391 fclose (fp);
1392
1393 char **tmp = realloc (argv, (argc + 1) * sizeof (char *));
1394 if (tmp == NULL)
1395 goto defaulted;
1396 argv = tmp;
1397 argv[argc] = NULL;
1398
1399 /*
1400 * Look for and parse the "heapsize" config parameter.
1401 * Manual since none of the clib infra has been bootstrapped yet.
1402 *
1403 * Format: heapsize <nn>[mM][gG]
1404 */
1405
1406 for (i = 1; i < (argc - 1); i++)
1407 {
1408 if (!strncmp (argv[i], "heapsize", 8))
1409 {
1410 sizep = (u8 *) argv[i + 1];
1411 size = 0;
1412 while (*sizep >= '0' && *sizep <= '9')
1413 {
1414 size *= 10;
1415 size += *sizep++ - '0';
1416 }
1417 if (size == 0)
1418 {
1419 if (VPPCOM_DEBUG > 0)
1420 clib_warning ("[%d] parse error '%s %s', "
1421 "using default heapsize %lld (0x%llx)",
1422 vcm->my_pid, argv[i], argv[i + 1],
1423 vcl_cfg->heapsize, vcl_cfg->heapsize);
1424 goto defaulted;
1425 }
1426
1427 if (*sizep == 'g' || *sizep == 'G')
1428 vcl_cfg->heapsize = size << 30;
1429 else if (*sizep == 'm' || *sizep == 'M')
1430 vcl_cfg->heapsize = size << 20;
1431 else
1432 {
1433 if (VPPCOM_DEBUG > 0)
1434 clib_warning ("[%d] parse error '%s %s', "
1435 "using default heapsize %lld (0x%llx)",
1436 vcm->my_pid, argv[i], argv[i + 1],
1437 vcl_cfg->heapsize, vcl_cfg->heapsize);
1438 goto defaulted;
1439 }
1440 }
1441 }
1442
1443defaulted:
1444 if (!clib_mem_init (0, vcl_cfg->heapsize))
1445 clib_warning ("[%d] vppcom heap allocation failure!", vcm->my_pid);
1446 else if (VPPCOM_DEBUG > 0)
1447 clib_warning ("[%d] allocated vppcom heapsize %lld (0x%llx)",
1448 vcm->my_pid, vcl_cfg->heapsize, vcl_cfg->heapsize);
1449}
1450
1451static void
1452vppcom_cfg_read (char *conf_fname)
1453{
1454 vppcom_main_t *vcm = &vppcom_main;
1455 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1456 int fd;
1457 unformat_input_t _input, *input = &_input;
1458 unformat_input_t _line_input, *line_input = &_line_input;
1459 u8 vc_cfg_input = 0;
1460 u8 *chroot_path;
1461 struct stat s;
1462 u32 uid, gid;
1463
1464 fd = open (conf_fname, O_RDONLY);
1465 if (fd < 0)
1466 {
1467 if (VPPCOM_DEBUG > 0)
1468 clib_warning ("[%d] open configuration file '%s' failed!",
1469 vcm->my_pid, conf_fname);
1470 goto file_done;
1471 }
1472
1473 if (fstat (fd, &s) < 0)
1474 {
1475 if (VPPCOM_DEBUG > 0)
1476 clib_warning ("[%d] failed to stat `%s'", vcm->my_pid, conf_fname);
1477 goto file_done;
1478 }
1479
1480 if (!(S_ISREG (s.st_mode) || S_ISLNK (s.st_mode)))
1481 {
1482 if (VPPCOM_DEBUG > 0)
1483 clib_warning ("[%d] not a regular file `%s'", vcm->my_pid,
1484 conf_fname);
1485 goto file_done;
1486 }
1487
1488 unformat_init_unix_file (input, fd);
1489
1490 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1491 {
1492 unformat_user (input, unformat_line_input, line_input);
1493 unformat_skip_white_space (line_input);
1494
1495 if (unformat (line_input, "vppcom {"))
1496 {
1497 vc_cfg_input = 1;
1498 continue;
1499 }
1500
1501 if (vc_cfg_input)
1502 {
1503 if (unformat (line_input, "heapsize %s", &chroot_path))
1504 {
1505 vec_terminate_c_string (chroot_path);
1506 if (VPPCOM_DEBUG > 0)
1507 clib_warning ("[%d] configured heapsize %s, "
1508 "actual heapsize %lld (0x%llx)",
1509 vcm->my_pid, chroot_path, vcl_cfg->heapsize,
1510 vcl_cfg->heapsize);
1511 vec_free (chroot_path);
1512 }
1513 else if (unformat (line_input, "api-prefix %s", &chroot_path))
1514 {
1515 vec_terminate_c_string (chroot_path);
1516 vl_set_memory_root_path ((char *) chroot_path);
1517 if (VPPCOM_DEBUG > 0)
1518 clib_warning ("[%d] configured api-prefix %s",
1519 vcm->my_pid, chroot_path);
1520 chroot_path = 0; /* Don't vec_free() it! */
1521 }
1522 else if (unformat (line_input, "uid %d", &uid))
1523 {
1524 vl_set_memory_uid (uid);
1525 if (VPPCOM_DEBUG > 0)
1526 clib_warning ("[%d] configured uid %d", vcm->my_pid, uid);
1527 }
1528 else if (unformat (line_input, "gid %d", &gid))
1529 {
1530 vl_set_memory_gid (gid);
1531 if (VPPCOM_DEBUG > 0)
1532 clib_warning ("[%d] configured gid %d", vcm->my_pid, gid);
1533 }
1534 else if (unformat (line_input, "segment-baseva 0x%llx",
1535 &vcl_cfg->segment_baseva))
1536 {
1537 if (VPPCOM_DEBUG > 0)
1538 clib_warning ("[%d] configured segment_baseva 0x%llx",
1539 vcm->my_pid, vcl_cfg->segment_baseva);
1540 }
1541 else if (unformat (line_input, "segment-size 0x%lx",
1542 &vcl_cfg->segment_size))
1543 {
1544 if (VPPCOM_DEBUG > 0)
1545 clib_warning ("[%d] configured segment_size 0x%lx (%ld)",
1546 vcm->my_pid, vcl_cfg->segment_size,
1547 vcl_cfg->segment_size);
1548 }
1549 else if (unformat (line_input, "segment-size %ld",
1550 &vcl_cfg->segment_size))
1551 {
1552 if (VPPCOM_DEBUG > 0)
1553 clib_warning ("[%d] configured segment_size %ld (0x%lx)",
1554 vcm->my_pid, vcl_cfg->segment_size,
1555 vcl_cfg->segment_size);
1556 }
1557 else if (unformat (line_input, "add-segment-size 0x%lx",
1558 &vcl_cfg->add_segment_size))
1559 {
1560 if (VPPCOM_DEBUG > 0)
1561 clib_warning
1562 ("[%d] configured add_segment_size 0x%lx (%ld)",
1563 vcm->my_pid, vcl_cfg->add_segment_size,
1564 vcl_cfg->add_segment_size);
1565 }
1566 else if (unformat (line_input, "add-segment-size %ld",
1567 &vcl_cfg->add_segment_size))
1568 {
1569 if (VPPCOM_DEBUG > 0)
1570 clib_warning
1571 ("[%d] configured add_segment_size %ld (0x%lx)",
1572 vcm->my_pid, vcl_cfg->add_segment_size,
1573 vcl_cfg->add_segment_size);
1574 }
1575 else if (unformat (line_input, "preallocated-fifo-pairs %d",
1576 &vcl_cfg->preallocated_fifo_pairs))
1577 {
1578 if (VPPCOM_DEBUG > 0)
1579 clib_warning ("[%d] configured preallocated_fifo_pairs "
1580 "%d (0x%x)", vcm->my_pid,
1581 vcl_cfg->preallocated_fifo_pairs,
1582 vcl_cfg->preallocated_fifo_pairs);
1583 }
1584 else if (unformat (line_input, "rx-fifo-size 0x%lx",
1585 &vcl_cfg->rx_fifo_size))
1586 {
1587 if (VPPCOM_DEBUG > 0)
1588 clib_warning ("[%d] configured rx_fifo_size 0x%lx (%ld)",
1589 vcm->my_pid, vcl_cfg->rx_fifo_size,
1590 vcl_cfg->rx_fifo_size);
1591 }
1592 else if (unformat (line_input, "rx-fifo-size %ld",
1593 &vcl_cfg->rx_fifo_size))
1594 {
1595 if (VPPCOM_DEBUG > 0)
1596 clib_warning ("[%d] configured rx_fifo_size %ld (0x%lx)",
1597 vcm->my_pid, vcl_cfg->rx_fifo_size,
1598 vcl_cfg->rx_fifo_size);
1599 }
1600 else if (unformat (line_input, "tx-fifo-size 0x%lx",
1601 &vcl_cfg->tx_fifo_size))
1602 {
1603 if (VPPCOM_DEBUG > 0)
1604 clib_warning ("[%d] configured tx_fifo_size 0x%lx (%ld)",
1605 vcm->my_pid, vcl_cfg->tx_fifo_size,
1606 vcl_cfg->tx_fifo_size);
1607 }
1608 else if (unformat (line_input, "tx-fifo-size %ld",
1609 &vcl_cfg->tx_fifo_size))
1610 {
1611 if (VPPCOM_DEBUG > 0)
1612 clib_warning ("[%d] configured tx_fifo_size %ld (0x%lx)",
1613 vcm->my_pid, vcl_cfg->tx_fifo_size,
1614 vcl_cfg->tx_fifo_size);
1615 }
1616 else if (unformat (line_input, "event-queue-size 0x%lx",
1617 &vcl_cfg->event_queue_size))
1618 {
1619 if (VPPCOM_DEBUG > 0)
1620 clib_warning ("[%d] configured event_queue_size 0x%lx (%ld)",
1621 vcm->my_pid, vcl_cfg->event_queue_size,
1622 vcl_cfg->event_queue_size);
1623 }
1624 else if (unformat (line_input, "event-queue-size %ld",
1625 &vcl_cfg->event_queue_size))
1626 {
1627 if (VPPCOM_DEBUG > 0)
1628 clib_warning ("[%d] configured event_queue_size %ld (0x%lx)",
1629 vcm->my_pid, vcl_cfg->event_queue_size,
1630 vcl_cfg->event_queue_size);
1631 }
1632 else if (unformat (line_input, "listen-queue-size 0x%lx",
1633 &vcl_cfg->listen_queue_size))
1634 {
1635 if (VPPCOM_DEBUG > 0)
1636 clib_warning ("[%d] configured listen_queue_size 0x%lx (%ld)",
1637 vcm->my_pid, vcl_cfg->listen_queue_size,
1638 vcl_cfg->listen_queue_size);
1639 }
1640 else if (unformat (line_input, "listen-queue-size %ld",
1641 &vcl_cfg->listen_queue_size))
1642 {
1643 if (VPPCOM_DEBUG > 0)
1644 clib_warning ("[%d] configured listen_queue_size %ld (0x%lx)",
1645 vcm->my_pid, vcl_cfg->listen_queue_size,
1646 vcl_cfg->listen_queue_size);
1647 }
1648 else if (unformat (line_input, "app-timeout %f",
1649 &vcl_cfg->app_timeout))
1650 {
1651 if (VPPCOM_DEBUG > 0)
1652 clib_warning ("[%d] configured app_timeout %f",
1653 vcm->my_pid, vcl_cfg->app_timeout);
1654 }
1655 else if (unformat (line_input, "session-timeout %f",
1656 &vcl_cfg->session_timeout))
1657 {
1658 if (VPPCOM_DEBUG > 0)
1659 clib_warning ("[%d] configured session_timeout %f",
1660 vcm->my_pid, vcl_cfg->session_timeout);
1661 }
1662 else if (unformat (line_input, "accept-timeout %f",
1663 &vcl_cfg->accept_timeout))
1664 {
1665 if (VPPCOM_DEBUG > 0)
1666 clib_warning ("[%d] configured accept_timeout %f",
1667 vcm->my_pid, vcl_cfg->accept_timeout);
1668 }
1669 else if (unformat (line_input, "}"))
1670 {
1671 vc_cfg_input = 0;
1672 if (VPPCOM_DEBUG > 0)
1673 clib_warning ("[%d] completed parsing vppcom config!",
1674 vcm->my_pid);
1675 goto input_done;
1676 }
1677 else
1678 {
1679 if (line_input->buffer[line_input->index] != '#')
1680 {
1681 clib_warning ("[%d] Unknown vppcom config option: '%s'",
1682 vcm->my_pid, (char *)
1683 &line_input->buffer[line_input->index]);
1684 }
1685 }
1686 }
1687 }
1688
1689input_done:
1690 unformat_free (input);
1691
1692file_done:
1693 if (fd > 0)
1694 close (fd);
1695}
1696
1697/*
1698 * VPPCOM Public API functions
1699 */
1700int
1701vppcom_app_create (char *app_name)
1702{
1703 vppcom_main_t *vcm = &vppcom_main;
1704 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1705 u8 *heap;
1706 mheap_t *h;
1707 int rv;
1708
1709 if (!vcm->init)
1710 {
1711 char *conf_fname;
1712
1713 vcm->init = 1;
1714 vcm->my_pid = getpid ();
1715 clib_fifo_validate (vcm->client_session_index_fifo,
1716 vcm->cfg.listen_queue_size);
1717 vppcom_cfg_init (vcl_cfg);
1718 conf_fname = getenv (VPPCOM_CONF_ENV);
1719 if (!conf_fname)
1720 {
1721 conf_fname = VPPCOM_CONF_DEFAULT;
1722 if (VPPCOM_DEBUG > 0)
1723 clib_warning ("[%d] getenv '%s' failed!", vcm->my_pid,
1724 VPPCOM_CONF_ENV);
1725 }
1726 vppcom_cfg_heapsize (conf_fname);
1727 vppcom_cfg_read (conf_fname);
1728 vcm->bind_session_index = ~0;
1729 vcm->main_cpu = os_get_thread_index ();
1730 heap = clib_mem_get_per_cpu_heap ();
1731 h = mheap_header (heap);
1732
1733 /* make the main heap thread-safe */
1734 h->flags |= MHEAP_FLAG_THREAD_SAFE;
1735
1736 vcm->session_index_by_vpp_handles = hash_create (0, sizeof (uword));
1737
1738 clib_time_init (&vcm->clib_time);
1739 vppcom_init_error_string_table ();
1740 svm_fifo_segment_init (vcl_cfg->segment_baseva,
1741 20 /* timeout in secs */ );
1742 clib_spinlock_init (&vcm->sessions_lockp);
1743 vppcom_api_hookup ();
1744 }
1745
1746 if (vcm->my_client_index == ~0)
1747 {
1748 vcm->app_state = STATE_APP_START;
1749 rv = vppcom_connect_to_vpp (app_name);
1750 if (rv)
1751 {
1752 clib_warning ("[%s] couldn't connect to VPP.", vcm->my_pid);
1753 return rv;
1754 }
1755
1756 if (VPPCOM_DEBUG > 0)
1757 clib_warning ("[%d] sending session enable", vcm->my_pid);
1758
1759 rv = vppcom_app_session_enable ();
1760 if (rv)
1761 {
1762 clib_warning ("[%d] vppcom_app_session_enable() failed!",
1763 vcm->my_pid);
1764 return rv;
1765 }
1766
1767 if (VPPCOM_DEBUG > 0)
1768 clib_warning ("[%d] sending app attach", vcm->my_pid);
1769
1770 rv = vppcom_app_attach ();
1771 if (rv)
1772 {
1773 clib_warning ("[%d] vppcom_app_attach() failed!", vcm->my_pid);
1774 return rv;
1775 }
1776 }
1777
1778 if (VPPCOM_DEBUG > 0)
1779 clib_warning ("[%d] app_name '%s', my_client_index %d (0x%x)",
1780 vcm->my_pid, app_name, vcm->my_client_index,
1781 vcm->my_client_index);
1782
1783 return VPPCOM_OK;
1784}
1785
1786void
1787vppcom_app_destroy (void)
1788{
1789 vppcom_main_t *vcm = &vppcom_main;
1790 int rv;
1791
1792 if (vcm->my_client_index == ~0)
1793 return;
1794
1795 if (VPPCOM_DEBUG > 0)
1796 clib_warning ("[%d] detaching from VPP, my_client_index %d (0x%x)",
1797 vcm->my_pid, vcm->my_client_index, vcm->my_client_index);
1798
1799 vppcom_app_detach ();
1800 rv = vppcom_wait_for_app_state_change (STATE_APP_ENABLED);
1801 if (PREDICT_FALSE (rv))
1802 {
1803 if (VPPCOM_DEBUG > 0)
1804 clib_warning ("[%d] application detach timed out, rv = %s (%d)",
1805 vcm->my_pid, vppcom_retval_str (rv), rv);
1806 }
1807 vl_client_disconnect_from_vlib ();
1808 vcm->my_client_index = ~0;
1809 vcm->app_state = STATE_APP_START;
1810}
1811
1812int
1813vppcom_session_create (u32 vrf, u8 proto, u8 is_nonblocking)
1814{
1815 vppcom_main_t *vcm = &vppcom_main;
1816 session_t *session;
1817 u32 session_index;
1818
1819 clib_spinlock_lock (&vcm->sessions_lockp);
1820 pool_get (vcm->sessions, session);
1821 session_index = session - vcm->sessions;
1822
1823 session->vrf = vrf;
1824 session->proto = proto;
1825 session->state = STATE_START;
1826 session->is_nonblocking = is_nonblocking;
1827 clib_spinlock_unlock (&vcm->sessions_lockp);
1828
1829 if (VPPCOM_DEBUG > 0)
1830 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1831
1832 return (int) session_index;
1833}
1834
1835int
1836vppcom_session_close (uint32_t session_index)
1837{
1838 vppcom_main_t *vcm = &vppcom_main;
1839 session_t *session = 0;
1840 int rv;
1841
1842 clib_spinlock_lock (&vcm->sessions_lockp);
1843 rv = vppcom_session_at_index (session_index, &session);
1844 if (PREDICT_FALSE (rv))
1845 {
1846 clib_spinlock_unlock (&vcm->sessions_lockp);
1847 if (VPPCOM_DEBUG > 0)
1848 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1849 vcm->my_pid, session_index);
1850 return rv;
1851 }
1852 clib_spinlock_unlock (&vcm->sessions_lockp);
1853
1854 if (VPPCOM_DEBUG > 0)
1855 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1856
1857 if (session->is_cut_thru)
1858 {
1859 if (session->is_server)
1860 rv = vppcom_session_unbind_cut_thru (session);
1861 }
1862 else
1863 {
1864 rv = (session->is_server) ?
1865 vppcom_session_unbind (session_index) :
1866 vppcom_session_disconnect (session_index);
1867 }
1868
1869 clib_spinlock_lock (&vcm->sessions_lockp);
1870 pool_put_index (vcm->sessions, session_index);
1871 clib_spinlock_unlock (&vcm->sessions_lockp);
1872 return rv;
1873}
1874
1875int
1876vppcom_session_bind (uint32_t session_index, vppcom_endpt_t * ep)
1877{
1878 vppcom_main_t *vcm = &vppcom_main;
1879 session_t *session = 0;
1880 int rv;
1881
1882 if (!ep || !ep->ip)
1883 return VPPCOM_EINVAL;
1884
1885 clib_spinlock_lock (&vcm->sessions_lockp);
1886 rv = vppcom_session_at_index (session_index, &session);
1887 if (PREDICT_FALSE (rv))
1888 {
1889 clib_spinlock_unlock (&vcm->sessions_lockp);
1890 if (VPPCOM_DEBUG > 0)
1891 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1892 vcm->my_pid, session_index);
1893 return rv;
1894 }
1895
1896 if (VPPCOM_DEBUG > 0)
1897 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1898
1899 session->vrf = ep->vrf;
1900 session->is_ip4 = ep->is_ip4;
1901 memset (session->ip, 0, sizeof (*session->ip));
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001902 clib_memcpy (session->ip, ep->ip, sizeof (session->ip));
Dave Wallace543852a2017-08-03 02:11:34 -04001903 session->port = ep->port;
1904
1905 clib_spinlock_unlock (&vcm->sessions_lockp);
1906 return VPPCOM_OK;
1907}
1908
1909int
1910vppcom_session_listen (uint32_t session_index, uint32_t q_len)
1911{
1912 vppcom_main_t *vcm = &vppcom_main;
1913 session_t *session = 0;
1914 int rv;
1915
1916 clib_spinlock_lock (&vcm->sessions_lockp);
1917 rv = vppcom_session_at_index (session_index, &session);
1918 if (PREDICT_FALSE (rv))
1919 {
1920 clib_spinlock_unlock (&vcm->sessions_lockp);
1921 if (VPPCOM_DEBUG > 0)
1922 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1923 vcm->my_pid, session_index);
1924 return rv;
1925 }
1926
1927 if (VPPCOM_DEBUG > 0)
1928 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1929
1930 ASSERT (vcm->bind_session_index == ~0);
1931 vcm->bind_session_index = session_index;
1932 vppcom_send_bind_sock (session);
1933 clib_spinlock_unlock (&vcm->sessions_lockp);
1934 rv = vppcom_wait_for_session_state_change (session_index, STATE_LISTEN,
1935 vcm->cfg.session_timeout);
1936 if (PREDICT_FALSE (rv))
1937 {
1938 vcm->bind_session_index = ~0;
1939 if (VPPCOM_DEBUG > 0)
1940 clib_warning ("[%d] server listen timed out, rv = %d (%d)",
1941 vcm->my_pid, vppcom_retval_str (rv), rv);
1942 return rv;
1943 }
1944
1945 clib_spinlock_lock (&vcm->sessions_lockp);
1946 rv = vppcom_session_at_index (session_index, &session);
1947 if (PREDICT_FALSE (rv))
1948 {
1949 clib_spinlock_unlock (&vcm->sessions_lockp);
1950 if (VPPCOM_DEBUG > 0)
1951 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1952 vcm->my_pid, session_index);
1953 return rv;
1954 }
1955 session->is_listen = 1;
1956 clib_spinlock_unlock (&vcm->sessions_lockp);
1957 clib_fifo_validate (vcm->client_session_index_fifo, q_len);
1958
1959 return VPPCOM_OK;
1960}
1961
1962int
1963vppcom_session_accept (uint32_t listen_session_index, vppcom_endpt_t * ep,
1964 double wait_for_time)
1965{
1966 vppcom_main_t *vcm = &vppcom_main;
1967 session_t *session = 0;
1968 u32 client_session_index;
1969 int rv;
1970 f64 wait_for;
1971
1972 clib_spinlock_lock (&vcm->sessions_lockp);
1973 rv = vppcom_session_at_index (listen_session_index, &session);
1974 if (PREDICT_FALSE (rv))
1975 {
1976 clib_spinlock_unlock (&vcm->sessions_lockp);
1977 if (VPPCOM_DEBUG > 0)
1978 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1979 vcm->my_pid, listen_session_index);
1980 return rv;
1981 }
1982
1983 if (session->state != STATE_LISTEN)
1984 {
1985 clib_spinlock_unlock (&vcm->sessions_lockp);
1986 if (VPPCOM_DEBUG > 0)
1987 clib_warning ("[%d] session not in listen state, state = %s",
1988 vcm->my_pid, vppcom_session_state_str (session->state));
1989 return VPPCOM_EBADFD;
1990 }
1991 wait_for = session->is_nonblocking ? 0 :
1992 (wait_for_time < 0) ? vcm->cfg.accept_timeout : wait_for_time;
1993
1994 if (VPPCOM_DEBUG > 0)
1995 clib_warning ("[%d] sid %d, state %s (%d)", vcm->my_pid,
1996 listen_session_index,
1997 vppcom_session_state_str (session->state), session->state);
1998 clib_spinlock_unlock (&vcm->sessions_lockp);
1999
2000 while (1)
2001 {
2002 rv = vppcom_wait_for_client_session_index (wait_for);
2003 if (rv)
2004 {
2005 if ((VPPCOM_DEBUG > 0))
2006 clib_warning ("[%d] sid %d, accept timed out, rv = %s (%d)",
2007 vcm->my_pid, listen_session_index,
2008 vppcom_retval_str (rv), rv);
2009 if ((wait_for == 0) || (wait_for_time > 0))
2010 return rv;
2011 }
2012 else
2013 break;
2014 }
2015
2016 clib_fifo_sub1 (vcm->client_session_index_fifo, client_session_index);
2017
2018 session = 0;
2019 clib_spinlock_lock (&vcm->sessions_lockp);
2020 rv = vppcom_session_at_index (client_session_index, &session);
2021 ASSERT (rv == VPPCOM_OK);
2022 ASSERT (session->is_server);
2023
2024 if (VPPCOM_DEBUG > 0)
2025 clib_warning ("[%d] Got a request: client sid %d", vcm->my_pid,
2026 client_session_index);
2027
2028 ep->vrf = session->vrf;
2029 ep->is_cut_thru = session->is_cut_thru;
2030 ep->is_ip4 = session->is_ip4;
2031 ep->port = session->port;
2032 memset (ep->ip, 0, sizeof (ip6_address_t));
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04002033 clib_memcpy (ep->ip, session->ip, sizeof (ip6_address_t));
Dave Wallace543852a2017-08-03 02:11:34 -04002034 session->state = STATE_LISTEN;
2035 clib_spinlock_unlock (&vcm->sessions_lockp);
2036 return (int) client_session_index;
2037}
2038
2039int
2040vppcom_session_connect (uint32_t session_index, vppcom_endpt_t * server_ep)
2041{
2042 vppcom_main_t *vcm = &vppcom_main;
2043 session_t *session = 0;
2044 int rv;
2045 ip46_address_t *ip46;
2046
2047 clib_spinlock_lock (&vcm->sessions_lockp);
2048 rv = vppcom_session_at_index (session_index, &session);
2049 if (PREDICT_FALSE (rv))
2050 {
2051 clib_spinlock_unlock (&vcm->sessions_lockp);
2052 if (VPPCOM_DEBUG > 0)
2053 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2054 vcm->my_pid, session_index);
2055 return rv;
2056 }
2057
2058 if (session->state == STATE_CONNECT)
2059 {
2060 clib_spinlock_unlock (&vcm->sessions_lockp);
2061 if (VPPCOM_DEBUG > 0)
2062 clib_warning ("[%d] session, sid (%d) already connected!",
2063 vcm->my_pid, session_index);
2064 return VPPCOM_OK;
2065 }
2066
2067 session->vrf = server_ep->vrf;
2068 session->is_ip4 = server_ep->is_ip4;
2069 ip46 = (ip46_address_t *) session->ip;
2070 *ip46 = to_ip46 (!server_ep->is_ip4, server_ep->ip);
2071 session->port = server_ep->port;
2072
2073 if (VPPCOM_DEBUG > 0)
2074 {
2075 u8 *ip_str = format (0, "%U", format_ip46_address,
2076 &session->ip, session->is_ip4);
2077 clib_warning ("[%d] connect sid %d to %s server port %d",
2078 vcm->my_pid, session_index, ip_str,
2079 clib_net_to_host_u16 (session->port));
2080 vec_free (ip_str);
2081 }
2082
2083 vppcom_send_connect_sock (session, session_index);
2084 clib_spinlock_unlock (&vcm->sessions_lockp);
2085 rv = vppcom_wait_for_session_state_change (session_index, STATE_CONNECT,
2086 vcm->cfg.session_timeout);
2087 if (PREDICT_FALSE (rv))
2088 {
2089 if (VPPCOM_DEBUG > 0)
2090 clib_warning ("[%d] connect timed out, rv = %s (%d)",
2091 vcm->my_pid, vppcom_retval_str (rv), rv);
2092 return rv;
2093 }
2094 return VPPCOM_OK;
2095}
2096
2097int
2098vppcom_session_read (uint32_t session_index, void *buf, int n)
2099{
2100 session_fifo_event_t _e, *e = &_e;
2101 vppcom_main_t *vcm = &vppcom_main;
2102 session_t *session = 0;
2103 svm_fifo_t *rx_fifo;
2104 int n_read = 0;
2105 int rv;
2106 char *fifo_str;
2107
2108 ASSERT (buf);
2109
2110 clib_spinlock_lock (&vcm->sessions_lockp);
2111 rv = vppcom_session_at_index (session_index, &session);
2112 if (PREDICT_FALSE (rv))
2113 {
2114 clib_spinlock_unlock (&vcm->sessions_lockp);
2115 if (VPPCOM_DEBUG > 0)
2116 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2117 vcm->my_pid, session_index);
2118 return rv;
2119 }
2120
2121 if (session->is_cut_thru)
2122 {
2123 rx_fifo = session->is_server ? session->server_rx_fifo :
2124 session->server_tx_fifo;
2125 fifo_str = session->is_server ? "server_rx_fifo" : "server_tx_fifo";
2126 clib_spinlock_unlock (&vcm->sessions_lockp);
2127
2128 n_read = svm_fifo_dequeue_nowait (rx_fifo, n, buf);
2129
2130 if (n_read <= 0)
2131 return VPPCOM_EAGAIN;
2132
2133 }
2134 else
2135 {
2136 rv = unix_shared_memory_queue_sub (session->event_queue, (u8 *) e,
2137 1 /* nowait */ );
2138 clib_spinlock_unlock (&vcm->sessions_lockp);
2139 if (rv < 0)
2140 return VPPCOM_EAGAIN;
2141
2142 switch (e->event_type)
2143 {
2144 case FIFO_EVENT_APP_RX:
2145 rx_fifo = e->fifo;
2146 fifo_str = "app_rx_fifo";
2147 n_read = svm_fifo_dequeue_nowait (rx_fifo, n, buf);
2148 break;
2149
2150 case FIFO_EVENT_DISCONNECT:
2151 return VPPCOM_ECONNRESET;
2152
2153 default:
2154 if (VPPCOM_DEBUG > 0)
2155 clib_warning ("[%d] unknown event type %d", vcm->my_pid,
2156 e->event_type);
2157 return VPPCOM_EAGAIN;
2158 }
2159 }
2160
2161 if (VPPCOM_DEBUG > 2)
2162 clib_warning ("[%d] sid %d, read %d bytes from %s (%p)", vcm->my_pid,
2163 session_index, n_read, fifo_str, rx_fifo);
2164 return n_read;
2165}
2166
2167static inline int
2168vppcom_session_read_ready (session_t * session, u32 session_index)
2169{
2170 session_fifo_event_t _e, *e = &_e;
2171 vppcom_main_t *vcm = &vppcom_main;
2172 svm_fifo_t *rx_fifo;
2173 int rv;
2174 int ready = 0;
2175
2176 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
2177 if (session->is_cut_thru)
2178 {
2179 rx_fifo = session->is_server ? session->server_rx_fifo :
2180 session->server_tx_fifo;
2181
2182 ready = svm_fifo_max_dequeue (rx_fifo);
2183 }
2184 else if (session->is_listen)
2185 ready = clib_fifo_elts (vcm->client_session_index_fifo);
2186 else
2187 {
2188 rv = unix_shared_memory_queue_sub (vcm->app_event_queue, (u8 *) e,
2189 1 /* nowait */ );
2190 if (rv >= 0)
2191 {
2192 switch (e->event_type)
2193 {
2194 case FIFO_EVENT_APP_RX:
2195 rx_fifo = e->fifo;
2196 ready = svm_fifo_max_dequeue (rx_fifo);
2197 break;
2198
2199 case FIFO_EVENT_DISCONNECT:
2200 return VPPCOM_ECONNRESET;
2201
2202 default:
2203 clib_warning ("[%d] unknown event type %d", vcm->my_pid,
2204 e->event_type);
2205 }
2206 }
2207 }
2208
2209 if (VPPCOM_DEBUG > 2)
2210 clib_warning ("[%d] sid %d, peek %s (%p), ready = %d", vcm->my_pid,
2211 session_index,
2212 session->is_server ? "server_rx_fifo" : "server_tx_fifo",
2213 rx_fifo, ready);
2214
2215 return ready;
2216}
2217
2218int
2219vppcom_session_write (uint32_t session_index, void *buf, int n)
2220{
2221 vppcom_main_t *vcm = &vppcom_main;
2222 session_t *session = 0;
2223 svm_fifo_t *tx_fifo;
2224 unix_shared_memory_queue_t *q;
2225 session_fifo_event_t evt;
2226 int rv;
2227 char *fifo_str;
2228 u8 is_nonblocking;
2229
2230 ASSERT (buf);
2231
2232 clib_spinlock_lock (&vcm->sessions_lockp);
2233 rv = vppcom_session_at_index (session_index, &session);
2234 if (PREDICT_FALSE (rv))
2235 {
2236 clib_spinlock_unlock (&vcm->sessions_lockp);
2237 if (VPPCOM_DEBUG > 0)
2238 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2239 vcm->my_pid, session_index);
2240 return rv;
2241 }
2242
2243 tx_fifo = ((!session->is_cut_thru || session->is_server) ?
2244 session->server_tx_fifo : session->server_rx_fifo);
2245 fifo_str = ((!session->is_cut_thru || session->is_server) ?
2246 "server_tx_fifo" : "server_rx_fifo");
2247 is_nonblocking = session->is_nonblocking;
2248 clib_spinlock_unlock (&vcm->sessions_lockp);
2249
2250 do
2251 {
2252 rv = svm_fifo_enqueue_nowait (tx_fifo, n, buf);
2253 }
2254 while (!is_nonblocking && (rv <= 0));
2255
2256 /* If event wasn't set, add one */
2257 if ((rv > 0) && svm_fifo_set_event (tx_fifo))
2258 {
2259 int rval;
2260
2261 /* Fabricate TX event, send to vpp */
2262 evt.fifo = tx_fifo;
2263 evt.event_type = FIFO_EVENT_APP_TX;
2264 evt.event_id = vcm->tx_event_id++;
2265
2266 clib_spinlock_lock (&vcm->sessions_lockp);
2267 rval = vppcom_session_at_index (session_index, &session);
2268 if (PREDICT_FALSE (rval))
2269 {
2270 clib_spinlock_unlock (&vcm->sessions_lockp);
2271 if (VPPCOM_DEBUG > 1)
2272 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2273 vcm->my_pid, session_index);
2274 return rval;
2275 }
2276 q = session->event_queue;
2277 clib_spinlock_unlock (&vcm->sessions_lockp);
2278 ASSERT (q);
2279 unix_shared_memory_queue_add (q, (u8 *) & evt,
2280 0 /* do wait for mutex */ );
2281 }
2282
2283 if (VPPCOM_DEBUG > 2)
2284 clib_warning ("[%d] sid %d, wrote %d bytes to %s (%p)", vcm->my_pid,
2285 session_index, rv, fifo_str, tx_fifo);
2286
2287 return rv;
2288}
2289
2290static inline int
2291vppcom_session_write_ready (session_t * session, u32 session_index)
2292{
2293 vppcom_main_t *vcm = &vppcom_main;
2294 svm_fifo_t *tx_fifo;
2295 int rv;
2296
2297 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
2298 tx_fifo = ((!session->is_cut_thru || session->is_server) ?
2299 session->server_tx_fifo : session->server_rx_fifo);
2300
2301 rv = svm_fifo_max_enqueue (tx_fifo);
2302
2303 if (VPPCOM_DEBUG > 2)
2304 clib_warning ("[%d] sid %d, peek %s (%p), ready = %d", vcm->my_pid,
2305 session_index,
2306 session->is_server ? "server_tx_fifo" : "server_rx_fifo",
2307 tx_fifo, rv);
2308 return rv;
2309}
2310
2311int
2312vppcom_select (unsigned long n_bits, unsigned long *read_map,
2313 unsigned long *write_map, unsigned long *except_map,
2314 double time_to_wait)
2315{
2316 vppcom_main_t *vcm = &vppcom_main;
2317 u32 session_index;
2318 session_t *session = 0;
2319 int rv, bits_set = 0;
2320 f64 timeout = clib_time_now (&vcm->clib_time) + time_to_wait;
2321 u32 minbits = clib_max (n_bits, BITS (uword));
2322
2323 ASSERT (sizeof (clib_bitmap_t) == sizeof (long int));
2324
2325 if (read_map)
2326 {
2327 clib_bitmap_validate (vcm->rd_bitmap, minbits);
2328 clib_memcpy (vcm->rd_bitmap, read_map, vec_len (vcm->rd_bitmap));
2329 memset (read_map, 0, vec_len (vcm->rd_bitmap));
2330 }
2331 if (write_map)
2332 {
2333 clib_bitmap_validate (vcm->wr_bitmap, minbits);
2334 clib_memcpy (vcm->wr_bitmap, write_map, vec_len (vcm->wr_bitmap));
2335 memset (write_map, 0, vec_len (vcm->wr_bitmap));
2336 }
2337 if (except_map)
2338 {
2339 clib_bitmap_validate (vcm->ex_bitmap, minbits);
2340 clib_memcpy (vcm->ex_bitmap, except_map, vec_len (vcm->ex_bitmap));
2341 memset (except_map, 0, vec_len (vcm->ex_bitmap));
2342 }
2343
2344 do
2345 {
2346 /* *INDENT-OFF* */
2347 clib_bitmap_foreach (session_index, vcm->rd_bitmap,
2348 ({
2349 clib_spinlock_lock (&vcm->sessions_lockp);
2350 rv = vppcom_session_at_index (session_index, &session);
2351 if (rv < 0)
2352 {
2353 clib_spinlock_unlock (&vcm->sessions_lockp);
2354 if (VPPCOM_DEBUG > 1)
2355 clib_warning ("[%d] session %d specified in "
2356 "read_map is closed.", vcm->my_pid,
2357 session_index);
2358 bits_set = VPPCOM_EBADFD;
2359 goto select_done;
2360 }
2361
2362 rv = vppcom_session_read_ready (session, session_index);
2363 clib_spinlock_unlock (&vcm->sessions_lockp);
2364 if (vcm->ex_bitmap &&
2365 clib_bitmap_get (vcm->ex_bitmap, session_index) && (rv < 0))
2366 {
2367 // TBD: clib_warning
2368 clib_bitmap_set_no_check (except_map, session_index, 1);
2369 bits_set++;
2370 }
2371 else if (rv > 0)
2372 {
2373 // TBD: clib_warning
2374 clib_bitmap_set_no_check (read_map, session_index, 1);
2375 bits_set++;
2376 }
2377 }));
2378
2379 clib_bitmap_foreach (session_index, vcm->wr_bitmap,
2380 ({
2381 clib_spinlock_lock (&vcm->sessions_lockp);
2382 rv = vppcom_session_at_index (session_index, &session);
2383 if (rv < 0)
2384 {
2385 clib_spinlock_unlock (&vcm->sessions_lockp);
2386 if (VPPCOM_DEBUG > 0)
2387 clib_warning ("[%d] session %d specified in "
2388 "write_map is closed.", vcm->my_pid,
2389 session_index);
2390 bits_set = VPPCOM_EBADFD;
2391 goto select_done;
2392 }
2393
2394 rv = vppcom_session_write_ready (session, session_index);
2395 clib_spinlock_unlock (&vcm->sessions_lockp);
2396 if (rv > 0)
2397 {
2398 // TBD: clib_warning
2399 clib_bitmap_set_no_check (write_map, session_index, 1);
2400 bits_set++;
2401 }
2402 }));
2403
2404 clib_bitmap_foreach (session_index, vcm->ex_bitmap,
2405 ({
2406 clib_spinlock_lock (&vcm->sessions_lockp);
2407 rv = vppcom_session_at_index (session_index, &session);
2408 if (rv < 0)
2409 {
2410 clib_spinlock_unlock (&vcm->sessions_lockp);
2411 if (VPPCOM_DEBUG > 1)
2412 clib_warning ("[%d] session %d specified in "
2413 "except_map is closed.", vcm->my_pid,
2414 session_index);
2415 bits_set = VPPCOM_EBADFD;
2416 goto select_done;
2417 }
2418
2419 rv = vppcom_session_read_ready (session, session_index);
2420 clib_spinlock_unlock (&vcm->sessions_lockp);
2421 if (rv < 0)
2422 {
2423 // TBD: clib_warning
2424 clib_bitmap_set_no_check (except_map, session_index, 1);
2425 bits_set++;
2426 }
2427 }));
2428 /* *INDENT-ON* */
2429 }
2430 while (clib_time_now (&vcm->clib_time) < timeout);
2431
2432select_done:
2433 return (bits_set);
2434}
2435
2436/*
2437 * fd.io coding-style-patch-verification: ON
2438 *
2439 * Local Variables:
2440 * eval: (c-set-style "gnu")
2441 * End:
2442 */