blob: 8aeb9dafa6da4ae40c5b325b1db4594ecf8da20d [file] [log] [blame]
Dave Wallace543852a2017-08-03 02:11:34 -04001/*
Dave Wallace33e002b2017-09-06 01:20:02 -04002 * Copyright (c) 2017 Cisco and/or its affiliates.
Dave Wallace543852a2017-08-03 02:11:34 -04003 * 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;
Dave Wallace33e002b2017-09-06 01:20:02 -040079 unix_shared_memory_queue_t *vpp_event_queue;
Dave Wallace543852a2017-08-03 02:11:34 -040080
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
Dave Wallace33e002b2017-09-06 01:20:02 -0400685vl_api_connect_session_reply_t_handler (vl_api_connect_session_reply_t * mp)
Dave Wallace543852a2017-08-03 02:11:34 -0400686{
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
Dave Wallace33e002b2017-09-06 01:20:02 -0400701 session_index = mp->context;
Dave Wallace543852a2017-08-03 02:11:34 -0400702 if (VPPCOM_DEBUG > 1)
Dave Wallace33e002b2017-09-06 01:20:02 -0400703 clib_warning ("[%d] session_index = %d 0x%08x", vcm->my_pid,
Dave Wallace543852a2017-08-03 02:11:34 -0400704 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;
Dave Wallace33e002b2017-09-06 01:20:02 -0400746 session->vpp_event_queue = uword_to_pointer (mp->vpp_event_queue_address,
747 unix_shared_memory_queue_t *);
Dave Wallace543852a2017-08-03 02:11:34 -0400748
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;
Dave Wallace33e002b2017-09-06 01:20:02 -0400776 cmp->context = session_index;
Dave Wallace543852a2017-08-03 02:11:34 -0400777
778 if (VPPCOM_DEBUG > 1)
Dave Wallace33e002b2017-09-06 01:20:02 -0400779 clib_warning ("[%d] session_index = %d 0x%08x",
780 vcm->my_pid, session_index, session_index);
Dave Wallace543852a2017-08-03 02:11:34 -0400781
782 cmp->vrf = session->vrf;
783 cmp->is_ip4 = session->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -0400784 clib_memcpy (cmp->ip, session->ip, sizeof (cmp->ip));
Dave Wallace543852a2017-08-03 02:11:34 -0400785 cmp->port = session->port;
786 cmp->proto = session->proto;
787 clib_memcpy (cmp->options, session->options, sizeof (cmp->options));
788 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & cmp);
789}
790
791static int
792vppcom_send_disconnect (u32 session_index)
793{
794 vppcom_main_t *vcm = &vppcom_main;
795 vl_api_disconnect_session_t *dmp;
796 session_t *session = 0;
797 int rv;
798
799 clib_spinlock_lock (&vcm->sessions_lockp);
800 rv = vppcom_session_at_index (session_index, &session);
801 if (PREDICT_FALSE (rv))
802 {
803 clib_spinlock_unlock (&vcm->sessions_lockp);
804 if (VPPCOM_DEBUG > 1)
805 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
806 vcm->my_pid, session_index);
807 return rv;
808 }
809
810 dmp = vl_msg_api_alloc (sizeof (*dmp));
811 memset (dmp, 0, sizeof (*dmp));
812 dmp->_vl_msg_id = ntohs (VL_API_DISCONNECT_SESSION);
813 dmp->client_index = vcm->my_client_index;
814 dmp->handle = session->vpp_session_handle;
815 clib_spinlock_unlock (&vcm->sessions_lockp);
816 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & dmp);
817 return VPPCOM_OK;
818}
819
820static void
821vl_api_bind_sock_reply_t_handler (vl_api_bind_sock_reply_t * mp)
822{
823 vppcom_main_t *vcm = &vppcom_main;
824 session_t *session = 0;
825 int rv;
826
827 if (mp->retval)
828 clib_warning ("[%d] bind failed: %U", vcm->my_pid, format_api_error,
829 ntohl (mp->retval));
830
831 ASSERT (vcm->bind_session_index != ~0);
832
833 clib_spinlock_lock (&vcm->sessions_lockp);
834 rv = vppcom_session_at_index (vcm->bind_session_index, &session);
835 if (rv == VPPCOM_OK)
836 {
837 session->vpp_session_handle = mp->handle;
838 hash_set (vcm->session_index_by_vpp_handles, mp->handle,
839 vcm->bind_session_index);
840 session->state = mp->retval ? STATE_FAILED : STATE_LISTEN;
841 vcm->bind_session_index = ~0;
842 }
843 clib_spinlock_unlock (&vcm->sessions_lockp);
844}
845
846static void
847vl_api_unbind_sock_reply_t_handler (vl_api_unbind_sock_reply_t * mp)
848{
849 vppcom_main_t *vcm = &vppcom_main;
850 session_t *session = 0;
851 int rv;
852
853 clib_spinlock_lock (&vcm->sessions_lockp);
854 rv = vppcom_session_at_index (vcm->bind_session_index, &session);
Dave Wallace19481612017-09-15 18:47:44 -0400855 if (rv == VPPCOM_OK)
Dave Wallace543852a2017-08-03 02:11:34 -0400856 {
Dave Wallace19481612017-09-15 18:47:44 -0400857 if ((VPPCOM_DEBUG > 1) && (mp->retval))
858 clib_warning ("[%d] unbind failed: %U", vcm->my_pid, format_api_error,
859 ntohl (mp->retval));
860
861 vcm->bind_session_index = ~0;
862 session->state = STATE_START;
Dave Wallace543852a2017-08-03 02:11:34 -0400863 }
Dave Wallace543852a2017-08-03 02:11:34 -0400864 clib_spinlock_unlock (&vcm->sessions_lockp);
865}
866
867u8 *
868format_ip4_address (u8 * s, va_list * args)
869{
870 u8 *a = va_arg (*args, u8 *);
871 return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
872}
873
874u8 *
875format_ip6_address (u8 * s, va_list * args)
876{
877 ip6_address_t *a = va_arg (*args, ip6_address_t *);
878 u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
879
880 i_max_n_zero = ARRAY_LEN (a->as_u16);
881 max_n_zeros = 0;
882 i_first_zero = i_max_n_zero;
883 n_zeros = 0;
884 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
885 {
886 u32 is_zero = a->as_u16[i] == 0;
887 if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
888 {
889 i_first_zero = i;
890 n_zeros = 0;
891 }
892 n_zeros += is_zero;
893 if ((!is_zero && n_zeros > max_n_zeros)
894 || (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
895 {
896 i_max_n_zero = i_first_zero;
897 max_n_zeros = n_zeros;
898 i_first_zero = ARRAY_LEN (a->as_u16);
899 n_zeros = 0;
900 }
901 }
902
903 last_double_colon = 0;
904 for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
905 {
906 if (i == i_max_n_zero && max_n_zeros > 1)
907 {
908 s = format (s, "::");
909 i += max_n_zeros - 1;
910 last_double_colon = 1;
911 }
912 else
913 {
914 s = format (s, "%s%x",
915 (last_double_colon || i == 0) ? "" : ":",
916 clib_net_to_host_u16 (a->as_u16[i]));
917 last_double_colon = 0;
918 }
919 }
920
921 return s;
922}
923
924/* Format an IP46 address. */
925u8 *
926format_ip46_address (u8 * s, va_list * args)
927{
928 ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
929 ip46_type_t type = va_arg (*args, ip46_type_t);
930 int is_ip4 = 1;
931
932 switch (type)
933 {
934 case IP46_TYPE_ANY:
935 is_ip4 = ip46_address_is_ip4 (ip46);
936 break;
937 case IP46_TYPE_IP4:
938 is_ip4 = 1;
939 break;
940 case IP46_TYPE_IP6:
941 is_ip4 = 0;
942 break;
943 }
944
945 return is_ip4 ?
946 format (s, "%U", format_ip4_address, &ip46->ip4) :
947 format (s, "%U", format_ip6_address, &ip46->ip6);
948}
949
950static void
951vl_api_accept_session_t_handler (vl_api_accept_session_t * mp)
952{
953 vppcom_main_t *vcm = &vppcom_main;
954 vl_api_accept_session_reply_t *rmp;
955 svm_fifo_t *rx_fifo, *tx_fifo;
956 session_t *session;
957 u32 session_index;
958 int rv = 0;
959
960 if (!clib_fifo_free_elts (vcm->client_session_index_fifo))
961 {
962 clib_warning ("[%d] client session queue is full!", vcm->my_pid);
963 rv = VNET_API_ERROR_QUEUE_FULL;
964 goto send_reply;
965 }
966
967 if (VPPCOM_DEBUG > 1)
968 {
969 u8 *ip_str = format (0, "%U", format_ip46_address, &mp->ip, mp->is_ip4);
970 clib_warning ("[%d] accepted session from: %s:%d", vcm->my_pid, ip_str,
971 clib_net_to_host_u16 (mp->port));
972 vec_free (ip_str);
973 }
974
975 clib_spinlock_lock (&vcm->sessions_lockp);
976 /* Allocate local session and set it up */
977 pool_get (vcm->sessions, session);
978 memset (session, 0, sizeof (*session));
979 session_index = session - vcm->sessions;
980
981 rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
982 rx_fifo->client_session_index = session_index;
983 tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
984 tx_fifo->client_session_index = session_index;
985
986 session->server_rx_fifo = rx_fifo;
987 session->server_tx_fifo = tx_fifo;
Dave Wallace33e002b2017-09-06 01:20:02 -0400988 session->vpp_event_queue = uword_to_pointer (mp->vpp_event_queue_address,
989 unix_shared_memory_queue_t *);
Dave Wallace543852a2017-08-03 02:11:34 -0400990 session->state = STATE_ACCEPT;
991 session->is_cut_thru = 0;
Dave Wallace19481612017-09-15 18:47:44 -0400992 session->is_server = 1;
Dave Wallace543852a2017-08-03 02:11:34 -0400993 session->port = ntohs (mp->port);
994 session->is_ip4 = mp->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -0400995 clib_memcpy (session->ip, mp->ip, sizeof (session->ip));
Dave Wallace543852a2017-08-03 02:11:34 -0400996 clib_spinlock_unlock (&vcm->sessions_lockp);
997
998 /* Add it to lookup table */
999 hash_set (vcm->session_index_by_vpp_handles, mp->handle, session_index);
1000
1001 clib_fifo_add1 (vcm->client_session_index_fifo, session_index);
1002
1003 /*
1004 * Send accept reply to vpp
1005 */
1006send_reply:
1007 rmp = vl_msg_api_alloc (sizeof (*rmp));
1008 memset (rmp, 0, sizeof (*rmp));
1009 rmp->_vl_msg_id = ntohs (VL_API_ACCEPT_SESSION_REPLY);
1010 rmp->retval = htonl (rv);
1011 rmp->handle = mp->handle;
1012 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & rmp);
1013}
1014
1015/*
1016 * Acting as server for redirected connect requests
1017 */
1018static void
1019vl_api_connect_sock_t_handler (vl_api_connect_sock_t * mp)
1020{
1021 static svm_fifo_segment_create_args_t _a;
1022 svm_fifo_segment_create_args_t *a = &_a;
1023 vppcom_main_t *vcm = &vppcom_main;
1024 u32 session_index;
1025 svm_fifo_segment_private_t *seg;
1026 unix_shared_memory_queue_t *client_q;
Dave Wallace33e002b2017-09-06 01:20:02 -04001027 vl_api_connect_session_reply_t *rmp;
Dave Wallace543852a2017-08-03 02:11:34 -04001028 session_t *session = 0;
1029 int rv = 0;
1030 svm_fifo_t *rx_fifo;
1031 svm_fifo_t *tx_fifo;
1032 unix_shared_memory_queue_t *event_q = 0;
1033
1034 if (!clib_fifo_free_elts (vcm->client_session_index_fifo))
1035 {
1036 if (VPPCOM_DEBUG > 1)
1037 clib_warning ("[%d] client session queue is full!", vcm->my_pid);
1038 rv = VNET_API_ERROR_QUEUE_FULL;
1039 goto send_reply;
1040 }
1041
1042 /* Create the segment */
1043 memset (a, 0, sizeof (*a));
1044 a->segment_name = (char *) format ((u8 *) a->segment_name, "%d:segment%d%c",
1045 vcm->my_pid, vcm->unique_segment_index++,
1046 0);
1047 a->segment_size = vcm->cfg.segment_size;
1048 a->preallocated_fifo_pairs = vcm->cfg.preallocated_fifo_pairs;
1049 a->rx_fifo_size = vcm->cfg.rx_fifo_size;
1050 a->tx_fifo_size = vcm->cfg.tx_fifo_size;
1051
1052 rv = svm_fifo_segment_create (a);
1053 if (PREDICT_FALSE (rv))
1054 {
1055 if (VPPCOM_DEBUG > 1)
1056 clib_warning ("[%d] svm_fifo_segment_create ('%s') failed",
1057 vcm->my_pid, a->segment_name);
1058 vec_reset_length (a->new_segment_indices);
1059 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1060 goto send_reply;
1061 }
1062
1063 if (VPPCOM_DEBUG > 1)
1064 clib_warning ("[%d] created segment '%s'", vcm->my_pid, a->segment_name);
1065
1066 clib_spinlock_lock (&vcm->sessions_lockp);
1067 pool_get (vcm->sessions, session);
1068 memset (session, 0, sizeof (*session));
1069 session_index = session - vcm->sessions;
1070
1071 session->sm_seg_index = a->new_segment_indices[0];
1072 vec_reset_length (a->new_segment_indices);
1073
Florin Corasc87c91d2017-08-16 19:55:49 -07001074 seg = svm_fifo_segment_get_segment (session->sm_seg_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001075 rx_fifo = session->server_rx_fifo =
1076 svm_fifo_segment_alloc_fifo (seg, vcm->cfg.rx_fifo_size,
1077 FIFO_SEGMENT_RX_FREELIST);
1078 if (PREDICT_FALSE (!session->server_rx_fifo))
1079 {
1080 svm_fifo_segment_delete (seg);
1081 clib_warning ("[%d] rx fifo alloc failed, size %ld (0x%lx)",
1082 vcm->my_pid, vcm->cfg.rx_fifo_size,
1083 vcm->cfg.rx_fifo_size);
1084 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1085 clib_spinlock_unlock (&vcm->sessions_lockp);
1086 goto send_reply;
1087 }
1088
1089 tx_fifo = session->server_tx_fifo =
1090 svm_fifo_segment_alloc_fifo (seg, vcm->cfg.tx_fifo_size,
1091 FIFO_SEGMENT_TX_FREELIST);
1092 if (PREDICT_FALSE (!session->server_tx_fifo))
1093 {
1094 svm_fifo_segment_delete (seg);
1095 if (VPPCOM_DEBUG > 1)
1096 clib_warning ("[%d] tx fifo alloc failed, size %ld (0x%lx)",
1097 vcm->my_pid, vcm->cfg.tx_fifo_size,
1098 vcm->cfg.tx_fifo_size);
1099 rv = VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
1100 clib_spinlock_unlock (&vcm->sessions_lockp);
1101 goto send_reply;
1102 }
1103
1104 session->server_rx_fifo->master_session_index = session_index;
1105 session->server_tx_fifo->master_session_index = session_index;
1106 session->client_queue_address = mp->client_queue_address;
1107 session->is_cut_thru = 1;
1108 session->is_server = 1;
1109 session->is_ip4 = mp->is_ip4;
1110 session->port = mp->port;
1111 {
1112 void *oldheap;
1113 ssvm_shared_header_t *sh = seg->ssvm.sh;
1114
1115 ssvm_lock_non_recursive (sh, 1);
1116 oldheap = ssvm_push_heap (sh);
Dave Wallace33e002b2017-09-06 01:20:02 -04001117 event_q = session->vpp_event_queue =
Dave Wallace543852a2017-08-03 02:11:34 -04001118 unix_shared_memory_queue_init (vcm->cfg.event_queue_size,
1119 sizeof (session_fifo_event_t),
1120 vcm->my_pid, 0 /* signal not sent */ );
1121 ssvm_pop_heap (oldheap);
1122 ssvm_unlock_non_recursive (sh);
1123 }
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001124 clib_memcpy (session->ip, mp->ip, sizeof (session->ip));
1125
Dave Wallace543852a2017-08-03 02:11:34 -04001126 session->state = STATE_ACCEPT;
1127 if (VPPCOM_DEBUG > 1)
1128 clib_warning ("[%d] Connected cut-thru to client: sid %d",
1129 vcm->my_pid, session_index);
1130 clib_spinlock_unlock (&vcm->sessions_lockp);
1131 clib_fifo_add1 (vcm->client_session_index_fifo, session_index);
1132
1133send_reply:
1134 rmp = vl_msg_api_alloc (sizeof (*rmp));
1135 memset (rmp, 0, sizeof (*rmp));
1136
Dave Wallace33e002b2017-09-06 01:20:02 -04001137 rmp->_vl_msg_id = ntohs (VL_API_CONNECT_SESSION_REPLY);
Dave Wallace543852a2017-08-03 02:11:34 -04001138 rmp->context = mp->context;
Dave Wallace543852a2017-08-03 02:11:34 -04001139 rmp->retval = htonl (rv);
1140 rmp->segment_name_length = vec_len (a->segment_name);
1141 clib_memcpy (rmp->segment_name, a->segment_name, vec_len (a->segment_name));
1142 vec_reset_length (a->segment_name);
1143
1144 if (event_q)
1145 {
1146 rmp->vpp_event_queue_address = pointer_to_uword (event_q);
1147 rmp->server_rx_fifo = pointer_to_uword (rx_fifo);
1148 rmp->server_tx_fifo = pointer_to_uword (tx_fifo);
1149 }
1150 client_q =
1151 uword_to_pointer (mp->client_queue_address, unix_shared_memory_queue_t *);
1152
1153 ASSERT (client_q);
1154 vl_msg_api_send_shmem (client_q, (u8 *) & rmp);
1155}
1156
1157static void
1158vppcom_send_bind_sock (session_t * session)
1159{
1160 vppcom_main_t *vcm = &vppcom_main;
1161 vl_api_bind_sock_t *bmp;
1162
1163 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
1164 session->is_server = 1;
1165 bmp = vl_msg_api_alloc (sizeof (*bmp));
1166 memset (bmp, 0, sizeof (*bmp));
1167
1168 bmp->_vl_msg_id = ntohs (VL_API_BIND_SOCK);
1169 bmp->client_index = vcm->my_client_index;
1170 bmp->context = htonl (0xfeedface);
1171 bmp->vrf = session->vrf;
1172 bmp->is_ip4 = session->is_ip4;
Dave Wallace6d5c4cd2017-08-15 16:56:29 -04001173 clib_memcpy (bmp->ip, session->ip, sizeof (bmp->ip));
Dave Wallace543852a2017-08-03 02:11:34 -04001174 bmp->port = session->port;
1175 bmp->proto = session->proto;
1176 clib_memcpy (bmp->options, session->options, sizeof (bmp->options));
1177 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & bmp);
1178}
1179
1180static void
1181vppcom_send_unbind_sock (u32 session_index)
1182{
1183 vppcom_main_t *vcm = &vppcom_main;
1184 vl_api_unbind_sock_t *ump;
1185 session_t *session = 0;
1186 int rv;
1187
1188 clib_spinlock_lock (&vcm->sessions_lockp);
1189 rv = vppcom_session_at_index (session_index, &session);
1190 if (PREDICT_FALSE (rv))
1191 {
1192 clib_spinlock_unlock (&vcm->sessions_lockp);
1193 if (VPPCOM_DEBUG > 0)
1194 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1195 vcm->my_pid, session_index);
1196 return;
1197 }
1198
1199 ump = vl_msg_api_alloc (sizeof (*ump));
1200 memset (ump, 0, sizeof (*ump));
1201
1202 ump->_vl_msg_id = ntohs (VL_API_UNBIND_SOCK);
1203 ump->client_index = vcm->my_client_index;
1204 ump->handle = session->vpp_session_handle;
1205 clib_spinlock_unlock (&vcm->sessions_lockp);
1206 vl_msg_api_send_shmem (vcm->vl_input_queue, (u8 *) & ump);
1207}
1208
1209static int
1210vppcom_session_unbind_cut_thru (session_t * session)
1211{
1212 svm_fifo_segment_main_t *sm = &svm_fifo_segment_main;
1213 svm_fifo_segment_private_t *seg;
1214 int rv = VPPCOM_OK;
1215
1216 seg = vec_elt_at_index (sm->segments, session->sm_seg_index);
1217 svm_fifo_segment_free_fifo (seg, session->server_rx_fifo,
1218 FIFO_SEGMENT_RX_FREELIST);
1219 svm_fifo_segment_free_fifo (seg, session->server_tx_fifo,
1220 FIFO_SEGMENT_TX_FREELIST);
1221 svm_fifo_segment_delete (seg);
1222
1223 return rv;
1224}
1225
1226static int
1227vppcom_session_unbind (u32 session_index)
1228{
1229 vppcom_main_t *vcm = &vppcom_main;
1230 int rv;
1231
1232 clib_spinlock_lock (&vcm->sessions_lockp);
1233 if (PREDICT_FALSE (pool_is_free_index (vcm->sessions, session_index)))
1234 {
1235 clib_spinlock_unlock (&vcm->sessions_lockp);
1236 if (VPPCOM_DEBUG > 1)
1237 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1238 vcm->my_pid, session_index);
1239 return VPPCOM_EBADFD;
1240 }
1241 clib_spinlock_unlock (&vcm->sessions_lockp);
1242
1243 vcm->bind_session_index = session_index;
1244 vppcom_send_unbind_sock (session_index);
1245 rv = vppcom_wait_for_session_state_change (session_index, STATE_START,
1246 vcm->cfg.session_timeout);
1247 if (PREDICT_FALSE (rv))
1248 {
1249 vcm->bind_session_index = ~0;
1250 if (VPPCOM_DEBUG > 0)
1251 clib_warning ("[%d] server unbind timed out, rv = %s (%d)",
1252 vcm->my_pid, vppcom_retval_str (rv), rv);
1253 return rv;
1254 }
1255 return VPPCOM_OK;
1256}
1257
1258static int
1259vppcom_session_disconnect (u32 session_index)
1260{
1261 vppcom_main_t *vcm = &vppcom_main;
1262 int rv;
1263
1264 rv = vppcom_send_disconnect (session_index);
1265 if (PREDICT_FALSE (rv))
1266 return rv;
1267
1268 rv = vppcom_wait_for_session_state_change (session_index, STATE_DISCONNECT,
1269 vcm->cfg.session_timeout);
1270 if (PREDICT_FALSE (rv))
1271 {
1272 if (VPPCOM_DEBUG > 0)
1273 clib_warning ("[%d] client disconnect timed out, rv = %s (%d)",
1274 vcm->my_pid, vppcom_retval_str (rv), rv);
1275 return rv;
1276 }
Dave Wallace543852a2017-08-03 02:11:34 -04001277 return VPPCOM_OK;
1278}
1279
1280#define foreach_sock_msg \
1281_(SESSION_ENABLE_DISABLE_REPLY, session_enable_disable_reply) \
1282_(BIND_SOCK_REPLY, bind_sock_reply) \
1283_(UNBIND_SOCK_REPLY, unbind_sock_reply) \
1284_(ACCEPT_SESSION, accept_session) \
1285_(CONNECT_SOCK, connect_sock) \
Dave Wallace33e002b2017-09-06 01:20:02 -04001286_(CONNECT_SESSION_REPLY, connect_session_reply) \
Dave Wallace543852a2017-08-03 02:11:34 -04001287_(DISCONNECT_SESSION, disconnect_session) \
1288_(DISCONNECT_SESSION_REPLY, disconnect_session_reply) \
1289_(RESET_SESSION, reset_session) \
1290_(APPLICATION_ATTACH_REPLY, application_attach_reply) \
1291_(APPLICATION_DETACH_REPLY, application_detach_reply) \
1292_(MAP_ANOTHER_SEGMENT, map_another_segment)
1293
1294static void
1295vppcom_api_hookup (void)
1296{
1297#define _(N,n) \
1298 vl_msg_api_set_handlers(VL_API_##N, #n, \
1299 vl_api_##n##_t_handler, \
1300 vl_noop_handler, \
1301 vl_api_##n##_t_endian, \
1302 vl_api_##n##_t_print, \
1303 sizeof(vl_api_##n##_t), 1);
1304 foreach_sock_msg;
1305#undef _
1306}
1307
1308static void
1309vppcom_cfg_init (vppcom_cfg_t * vcl_cfg)
1310{
1311 ASSERT (vcl_cfg);
1312
1313 vcl_cfg->heapsize = (256ULL << 20);
1314 vcl_cfg->segment_baseva = 0x200000000ULL;
1315 vcl_cfg->segment_size = (256 << 20);
1316 vcl_cfg->add_segment_size = (128 << 20);
1317 vcl_cfg->preallocated_fifo_pairs = 8;
1318 vcl_cfg->rx_fifo_size = (1 << 20);
1319 vcl_cfg->tx_fifo_size = (1 << 20);
1320 vcl_cfg->event_queue_size = 2048;
1321 vcl_cfg->listen_queue_size = CLIB_CACHE_LINE_BYTES / sizeof (u32);
1322 vcl_cfg->app_timeout = 10 * 60.0;
1323 vcl_cfg->session_timeout = 10 * 60.0;
1324 vcl_cfg->accept_timeout = 60.0;
1325}
1326
1327static void
1328vppcom_cfg_heapsize (char *conf_fname)
1329{
1330 vppcom_main_t *vcm = &vppcom_main;
1331 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1332 FILE *fp;
1333 char inbuf[4096];
1334 int argc = 1;
1335 char **argv = NULL;
1336 char *arg = NULL;
1337 char *p;
1338 int i;
1339 u8 *sizep;
1340 u32 size;
1341
1342 fp = fopen (conf_fname, "r");
1343 if (fp == NULL)
1344 {
1345 if (VPPCOM_DEBUG > 0)
1346 fprintf (stderr, "open configuration file '%s' failed\n", conf_fname);
1347 goto defaulted;
1348 }
1349 argv = calloc (1, sizeof (char *));
1350 if (argv == NULL)
1351 goto defaulted;
1352
1353 while (1)
1354 {
1355 if (fgets (inbuf, 4096, fp) == 0)
1356 break;
1357 p = strtok (inbuf, " \t\n");
1358 while (p != NULL)
1359 {
1360 if (*p == '#')
1361 break;
1362 argc++;
1363 char **tmp = realloc (argv, argc * sizeof (char *));
1364 if (tmp == NULL)
Chris Lukeab7b8d92017-09-07 07:40:13 -04001365 goto defaulted;
Dave Wallace543852a2017-08-03 02:11:34 -04001366 argv = tmp;
1367 arg = strndup (p, 1024);
1368 if (arg == NULL)
Chris Lukeab7b8d92017-09-07 07:40:13 -04001369 goto defaulted;
Dave Wallace543852a2017-08-03 02:11:34 -04001370 argv[argc - 1] = arg;
1371 p = strtok (NULL, " \t\n");
1372 }
1373 }
1374
1375 fclose (fp);
Chris Lukeab7b8d92017-09-07 07:40:13 -04001376 fp = NULL;
Dave Wallace543852a2017-08-03 02:11:34 -04001377
1378 char **tmp = realloc (argv, (argc + 1) * sizeof (char *));
1379 if (tmp == NULL)
1380 goto defaulted;
1381 argv = tmp;
1382 argv[argc] = NULL;
1383
1384 /*
1385 * Look for and parse the "heapsize" config parameter.
1386 * Manual since none of the clib infra has been bootstrapped yet.
1387 *
1388 * Format: heapsize <nn>[mM][gG]
1389 */
1390
1391 for (i = 1; i < (argc - 1); i++)
1392 {
1393 if (!strncmp (argv[i], "heapsize", 8))
1394 {
1395 sizep = (u8 *) argv[i + 1];
1396 size = 0;
1397 while (*sizep >= '0' && *sizep <= '9')
1398 {
1399 size *= 10;
1400 size += *sizep++ - '0';
1401 }
1402 if (size == 0)
1403 {
1404 if (VPPCOM_DEBUG > 0)
1405 clib_warning ("[%d] parse error '%s %s', "
1406 "using default heapsize %lld (0x%llx)",
1407 vcm->my_pid, argv[i], argv[i + 1],
1408 vcl_cfg->heapsize, vcl_cfg->heapsize);
1409 goto defaulted;
1410 }
1411
1412 if (*sizep == 'g' || *sizep == 'G')
1413 vcl_cfg->heapsize = size << 30;
1414 else if (*sizep == 'm' || *sizep == 'M')
1415 vcl_cfg->heapsize = size << 20;
1416 else
1417 {
1418 if (VPPCOM_DEBUG > 0)
1419 clib_warning ("[%d] parse error '%s %s', "
1420 "using default heapsize %lld (0x%llx)",
1421 vcm->my_pid, argv[i], argv[i + 1],
1422 vcl_cfg->heapsize, vcl_cfg->heapsize);
1423 goto defaulted;
1424 }
1425 }
1426 }
1427
1428defaulted:
Chris Lukeab7b8d92017-09-07 07:40:13 -04001429 if (fp != NULL)
1430 fclose (fp);
1431 if (argv != NULL)
1432 free (argv);
Dave Wallace543852a2017-08-03 02:11:34 -04001433 if (!clib_mem_init (0, vcl_cfg->heapsize))
1434 clib_warning ("[%d] vppcom heap allocation failure!", vcm->my_pid);
1435 else if (VPPCOM_DEBUG > 0)
1436 clib_warning ("[%d] allocated vppcom heapsize %lld (0x%llx)",
1437 vcm->my_pid, vcl_cfg->heapsize, vcl_cfg->heapsize);
1438}
1439
1440static void
1441vppcom_cfg_read (char *conf_fname)
1442{
1443 vppcom_main_t *vcm = &vppcom_main;
1444 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1445 int fd;
1446 unformat_input_t _input, *input = &_input;
1447 unformat_input_t _line_input, *line_input = &_line_input;
1448 u8 vc_cfg_input = 0;
1449 u8 *chroot_path;
1450 struct stat s;
1451 u32 uid, gid;
1452
1453 fd = open (conf_fname, O_RDONLY);
1454 if (fd < 0)
1455 {
1456 if (VPPCOM_DEBUG > 0)
1457 clib_warning ("[%d] open configuration file '%s' failed!",
1458 vcm->my_pid, conf_fname);
1459 goto file_done;
1460 }
1461
1462 if (fstat (fd, &s) < 0)
1463 {
1464 if (VPPCOM_DEBUG > 0)
1465 clib_warning ("[%d] failed to stat `%s'", vcm->my_pid, conf_fname);
1466 goto file_done;
1467 }
1468
1469 if (!(S_ISREG (s.st_mode) || S_ISLNK (s.st_mode)))
1470 {
1471 if (VPPCOM_DEBUG > 0)
1472 clib_warning ("[%d] not a regular file `%s'", vcm->my_pid,
1473 conf_fname);
1474 goto file_done;
1475 }
1476
Dave Barach59b25652017-09-10 15:04:27 -04001477 unformat_init_clib_file (input, fd);
Dave Wallace543852a2017-08-03 02:11:34 -04001478
1479 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1480 {
Chris Lukeb2bcad62017-09-18 08:51:22 -04001481 (void) unformat_user (input, unformat_line_input, line_input);
Dave Wallace543852a2017-08-03 02:11:34 -04001482 unformat_skip_white_space (line_input);
1483
1484 if (unformat (line_input, "vppcom {"))
1485 {
1486 vc_cfg_input = 1;
1487 continue;
1488 }
1489
1490 if (vc_cfg_input)
1491 {
1492 if (unformat (line_input, "heapsize %s", &chroot_path))
1493 {
1494 vec_terminate_c_string (chroot_path);
1495 if (VPPCOM_DEBUG > 0)
1496 clib_warning ("[%d] configured heapsize %s, "
1497 "actual heapsize %lld (0x%llx)",
1498 vcm->my_pid, chroot_path, vcl_cfg->heapsize,
1499 vcl_cfg->heapsize);
1500 vec_free (chroot_path);
1501 }
1502 else if (unformat (line_input, "api-prefix %s", &chroot_path))
1503 {
1504 vec_terminate_c_string (chroot_path);
1505 vl_set_memory_root_path ((char *) chroot_path);
1506 if (VPPCOM_DEBUG > 0)
1507 clib_warning ("[%d] configured api-prefix %s",
1508 vcm->my_pid, chroot_path);
1509 chroot_path = 0; /* Don't vec_free() it! */
1510 }
1511 else if (unformat (line_input, "uid %d", &uid))
1512 {
1513 vl_set_memory_uid (uid);
1514 if (VPPCOM_DEBUG > 0)
1515 clib_warning ("[%d] configured uid %d", vcm->my_pid, uid);
1516 }
1517 else if (unformat (line_input, "gid %d", &gid))
1518 {
1519 vl_set_memory_gid (gid);
1520 if (VPPCOM_DEBUG > 0)
1521 clib_warning ("[%d] configured gid %d", vcm->my_pid, gid);
1522 }
1523 else if (unformat (line_input, "segment-baseva 0x%llx",
1524 &vcl_cfg->segment_baseva))
1525 {
1526 if (VPPCOM_DEBUG > 0)
1527 clib_warning ("[%d] configured segment_baseva 0x%llx",
1528 vcm->my_pid, vcl_cfg->segment_baseva);
1529 }
1530 else if (unformat (line_input, "segment-size 0x%lx",
1531 &vcl_cfg->segment_size))
1532 {
1533 if (VPPCOM_DEBUG > 0)
1534 clib_warning ("[%d] configured segment_size 0x%lx (%ld)",
1535 vcm->my_pid, vcl_cfg->segment_size,
1536 vcl_cfg->segment_size);
1537 }
1538 else if (unformat (line_input, "segment-size %ld",
1539 &vcl_cfg->segment_size))
1540 {
1541 if (VPPCOM_DEBUG > 0)
1542 clib_warning ("[%d] configured segment_size %ld (0x%lx)",
1543 vcm->my_pid, vcl_cfg->segment_size,
1544 vcl_cfg->segment_size);
1545 }
1546 else if (unformat (line_input, "add-segment-size 0x%lx",
1547 &vcl_cfg->add_segment_size))
1548 {
1549 if (VPPCOM_DEBUG > 0)
1550 clib_warning
1551 ("[%d] configured add_segment_size 0x%lx (%ld)",
1552 vcm->my_pid, vcl_cfg->add_segment_size,
1553 vcl_cfg->add_segment_size);
1554 }
1555 else if (unformat (line_input, "add-segment-size %ld",
1556 &vcl_cfg->add_segment_size))
1557 {
1558 if (VPPCOM_DEBUG > 0)
1559 clib_warning
1560 ("[%d] configured add_segment_size %ld (0x%lx)",
1561 vcm->my_pid, vcl_cfg->add_segment_size,
1562 vcl_cfg->add_segment_size);
1563 }
1564 else if (unformat (line_input, "preallocated-fifo-pairs %d",
1565 &vcl_cfg->preallocated_fifo_pairs))
1566 {
1567 if (VPPCOM_DEBUG > 0)
1568 clib_warning ("[%d] configured preallocated_fifo_pairs "
1569 "%d (0x%x)", vcm->my_pid,
1570 vcl_cfg->preallocated_fifo_pairs,
1571 vcl_cfg->preallocated_fifo_pairs);
1572 }
1573 else if (unformat (line_input, "rx-fifo-size 0x%lx",
1574 &vcl_cfg->rx_fifo_size))
1575 {
1576 if (VPPCOM_DEBUG > 0)
1577 clib_warning ("[%d] configured rx_fifo_size 0x%lx (%ld)",
1578 vcm->my_pid, vcl_cfg->rx_fifo_size,
1579 vcl_cfg->rx_fifo_size);
1580 }
1581 else if (unformat (line_input, "rx-fifo-size %ld",
1582 &vcl_cfg->rx_fifo_size))
1583 {
1584 if (VPPCOM_DEBUG > 0)
1585 clib_warning ("[%d] configured rx_fifo_size %ld (0x%lx)",
1586 vcm->my_pid, vcl_cfg->rx_fifo_size,
1587 vcl_cfg->rx_fifo_size);
1588 }
1589 else if (unformat (line_input, "tx-fifo-size 0x%lx",
1590 &vcl_cfg->tx_fifo_size))
1591 {
1592 if (VPPCOM_DEBUG > 0)
1593 clib_warning ("[%d] configured tx_fifo_size 0x%lx (%ld)",
1594 vcm->my_pid, vcl_cfg->tx_fifo_size,
1595 vcl_cfg->tx_fifo_size);
1596 }
1597 else if (unformat (line_input, "tx-fifo-size %ld",
1598 &vcl_cfg->tx_fifo_size))
1599 {
1600 if (VPPCOM_DEBUG > 0)
1601 clib_warning ("[%d] configured tx_fifo_size %ld (0x%lx)",
1602 vcm->my_pid, vcl_cfg->tx_fifo_size,
1603 vcl_cfg->tx_fifo_size);
1604 }
1605 else if (unformat (line_input, "event-queue-size 0x%lx",
1606 &vcl_cfg->event_queue_size))
1607 {
1608 if (VPPCOM_DEBUG > 0)
1609 clib_warning ("[%d] configured event_queue_size 0x%lx (%ld)",
1610 vcm->my_pid, vcl_cfg->event_queue_size,
1611 vcl_cfg->event_queue_size);
1612 }
1613 else if (unformat (line_input, "event-queue-size %ld",
1614 &vcl_cfg->event_queue_size))
1615 {
1616 if (VPPCOM_DEBUG > 0)
1617 clib_warning ("[%d] configured event_queue_size %ld (0x%lx)",
1618 vcm->my_pid, vcl_cfg->event_queue_size,
1619 vcl_cfg->event_queue_size);
1620 }
1621 else if (unformat (line_input, "listen-queue-size 0x%lx",
1622 &vcl_cfg->listen_queue_size))
1623 {
1624 if (VPPCOM_DEBUG > 0)
1625 clib_warning ("[%d] configured listen_queue_size 0x%lx (%ld)",
1626 vcm->my_pid, vcl_cfg->listen_queue_size,
1627 vcl_cfg->listen_queue_size);
1628 }
1629 else if (unformat (line_input, "listen-queue-size %ld",
1630 &vcl_cfg->listen_queue_size))
1631 {
1632 if (VPPCOM_DEBUG > 0)
1633 clib_warning ("[%d] configured listen_queue_size %ld (0x%lx)",
1634 vcm->my_pid, vcl_cfg->listen_queue_size,
1635 vcl_cfg->listen_queue_size);
1636 }
1637 else if (unformat (line_input, "app-timeout %f",
1638 &vcl_cfg->app_timeout))
1639 {
1640 if (VPPCOM_DEBUG > 0)
1641 clib_warning ("[%d] configured app_timeout %f",
1642 vcm->my_pid, vcl_cfg->app_timeout);
1643 }
1644 else if (unformat (line_input, "session-timeout %f",
1645 &vcl_cfg->session_timeout))
1646 {
1647 if (VPPCOM_DEBUG > 0)
1648 clib_warning ("[%d] configured session_timeout %f",
1649 vcm->my_pid, vcl_cfg->session_timeout);
1650 }
1651 else if (unformat (line_input, "accept-timeout %f",
1652 &vcl_cfg->accept_timeout))
1653 {
1654 if (VPPCOM_DEBUG > 0)
1655 clib_warning ("[%d] configured accept_timeout %f",
1656 vcm->my_pid, vcl_cfg->accept_timeout);
1657 }
1658 else if (unformat (line_input, "}"))
1659 {
1660 vc_cfg_input = 0;
1661 if (VPPCOM_DEBUG > 0)
1662 clib_warning ("[%d] completed parsing vppcom config!",
1663 vcm->my_pid);
1664 goto input_done;
1665 }
1666 else
1667 {
1668 if (line_input->buffer[line_input->index] != '#')
1669 {
1670 clib_warning ("[%d] Unknown vppcom config option: '%s'",
1671 vcm->my_pid, (char *)
1672 &line_input->buffer[line_input->index]);
1673 }
1674 }
1675 }
1676 }
1677
1678input_done:
1679 unformat_free (input);
1680
1681file_done:
Chris Lukeab7b8d92017-09-07 07:40:13 -04001682 if (fd >= 0)
Dave Wallace543852a2017-08-03 02:11:34 -04001683 close (fd);
1684}
1685
1686/*
1687 * VPPCOM Public API functions
1688 */
1689int
1690vppcom_app_create (char *app_name)
1691{
1692 vppcom_main_t *vcm = &vppcom_main;
1693 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
1694 u8 *heap;
1695 mheap_t *h;
1696 int rv;
1697
1698 if (!vcm->init)
1699 {
1700 char *conf_fname;
1701
1702 vcm->init = 1;
1703 vcm->my_pid = getpid ();
1704 clib_fifo_validate (vcm->client_session_index_fifo,
1705 vcm->cfg.listen_queue_size);
1706 vppcom_cfg_init (vcl_cfg);
1707 conf_fname = getenv (VPPCOM_CONF_ENV);
1708 if (!conf_fname)
1709 {
1710 conf_fname = VPPCOM_CONF_DEFAULT;
1711 if (VPPCOM_DEBUG > 0)
1712 clib_warning ("[%d] getenv '%s' failed!", vcm->my_pid,
1713 VPPCOM_CONF_ENV);
1714 }
1715 vppcom_cfg_heapsize (conf_fname);
1716 vppcom_cfg_read (conf_fname);
1717 vcm->bind_session_index = ~0;
1718 vcm->main_cpu = os_get_thread_index ();
1719 heap = clib_mem_get_per_cpu_heap ();
1720 h = mheap_header (heap);
1721
1722 /* make the main heap thread-safe */
1723 h->flags |= MHEAP_FLAG_THREAD_SAFE;
1724
1725 vcm->session_index_by_vpp_handles = hash_create (0, sizeof (uword));
1726
1727 clib_time_init (&vcm->clib_time);
1728 vppcom_init_error_string_table ();
1729 svm_fifo_segment_init (vcl_cfg->segment_baseva,
1730 20 /* timeout in secs */ );
1731 clib_spinlock_init (&vcm->sessions_lockp);
1732 vppcom_api_hookup ();
1733 }
1734
1735 if (vcm->my_client_index == ~0)
1736 {
1737 vcm->app_state = STATE_APP_START;
1738 rv = vppcom_connect_to_vpp (app_name);
1739 if (rv)
1740 {
1741 clib_warning ("[%s] couldn't connect to VPP.", vcm->my_pid);
1742 return rv;
1743 }
1744
1745 if (VPPCOM_DEBUG > 0)
1746 clib_warning ("[%d] sending session enable", vcm->my_pid);
1747
1748 rv = vppcom_app_session_enable ();
1749 if (rv)
1750 {
1751 clib_warning ("[%d] vppcom_app_session_enable() failed!",
1752 vcm->my_pid);
1753 return rv;
1754 }
1755
1756 if (VPPCOM_DEBUG > 0)
1757 clib_warning ("[%d] sending app attach", vcm->my_pid);
1758
1759 rv = vppcom_app_attach ();
1760 if (rv)
1761 {
1762 clib_warning ("[%d] vppcom_app_attach() failed!", vcm->my_pid);
1763 return rv;
1764 }
1765 }
1766
1767 if (VPPCOM_DEBUG > 0)
1768 clib_warning ("[%d] app_name '%s', my_client_index %d (0x%x)",
1769 vcm->my_pid, app_name, vcm->my_client_index,
1770 vcm->my_client_index);
1771
1772 return VPPCOM_OK;
1773}
1774
1775void
1776vppcom_app_destroy (void)
1777{
1778 vppcom_main_t *vcm = &vppcom_main;
1779 int rv;
1780
1781 if (vcm->my_client_index == ~0)
1782 return;
1783
1784 if (VPPCOM_DEBUG > 0)
1785 clib_warning ("[%d] detaching from VPP, my_client_index %d (0x%x)",
1786 vcm->my_pid, vcm->my_client_index, vcm->my_client_index);
1787
1788 vppcom_app_detach ();
1789 rv = vppcom_wait_for_app_state_change (STATE_APP_ENABLED);
1790 if (PREDICT_FALSE (rv))
1791 {
1792 if (VPPCOM_DEBUG > 0)
1793 clib_warning ("[%d] application detach timed out, rv = %s (%d)",
1794 vcm->my_pid, vppcom_retval_str (rv), rv);
1795 }
1796 vl_client_disconnect_from_vlib ();
1797 vcm->my_client_index = ~0;
1798 vcm->app_state = STATE_APP_START;
1799}
1800
1801int
1802vppcom_session_create (u32 vrf, u8 proto, u8 is_nonblocking)
1803{
1804 vppcom_main_t *vcm = &vppcom_main;
1805 session_t *session;
1806 u32 session_index;
1807
1808 clib_spinlock_lock (&vcm->sessions_lockp);
1809 pool_get (vcm->sessions, session);
1810 session_index = session - vcm->sessions;
1811
1812 session->vrf = vrf;
1813 session->proto = proto;
1814 session->state = STATE_START;
1815 session->is_nonblocking = is_nonblocking;
1816 clib_spinlock_unlock (&vcm->sessions_lockp);
1817
1818 if (VPPCOM_DEBUG > 0)
1819 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1820
1821 return (int) session_index;
1822}
1823
1824int
1825vppcom_session_close (uint32_t session_index)
1826{
1827 vppcom_main_t *vcm = &vppcom_main;
1828 session_t *session = 0;
1829 int rv;
1830
1831 clib_spinlock_lock (&vcm->sessions_lockp);
1832 rv = vppcom_session_at_index (session_index, &session);
1833 if (PREDICT_FALSE (rv))
1834 {
1835 clib_spinlock_unlock (&vcm->sessions_lockp);
1836 if (VPPCOM_DEBUG > 0)
1837 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1838 vcm->my_pid, session_index);
1839 return rv;
1840 }
1841 clib_spinlock_unlock (&vcm->sessions_lockp);
1842
1843 if (VPPCOM_DEBUG > 0)
1844 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1845
1846 if (session->is_cut_thru)
1847 {
1848 if (session->is_server)
Dave Wallace19481612017-09-15 18:47:44 -04001849 {
1850 rv = vppcom_session_unbind_cut_thru (session);
1851 if ((VPPCOM_DEBUG > 0) && (rv < 0))
1852 clib_warning ("[%d] unbind cut-thru (session %d) failed, "
1853 "rv = %s (%d)",
1854 vcm->my_pid, session_index,
1855 vppcom_retval_str (rv), rv);
1856 }
1857 }
1858 else if (session->is_server)
1859 {
1860 rv = vppcom_session_unbind (session_index);
1861 if ((VPPCOM_DEBUG > 0) && (rv < 0))
1862 clib_warning ("[%d] unbind (session %d) failed, rv = %s (%d)",
1863 vcm->my_pid, session_index, vppcom_retval_str (rv), rv);
Dave Wallace543852a2017-08-03 02:11:34 -04001864 }
1865 else
1866 {
Dave Wallace19481612017-09-15 18:47:44 -04001867 rv = vppcom_session_disconnect (session_index);
1868 if ((VPPCOM_DEBUG > 0) && (rv < 0))
1869 clib_warning ("[%d] disconnect (session %d) failed, rv = %s (%d)",
1870 vcm->my_pid, session_index, vppcom_retval_str (rv), rv);
Dave Wallace543852a2017-08-03 02:11:34 -04001871 }
Dave Wallace19481612017-09-15 18:47:44 -04001872 if (rv < 0)
1873 return rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001874
1875 clib_spinlock_lock (&vcm->sessions_lockp);
1876 pool_put_index (vcm->sessions, session_index);
1877 clib_spinlock_unlock (&vcm->sessions_lockp);
1878 return rv;
1879}
1880
1881int
1882vppcom_session_bind (uint32_t session_index, vppcom_endpt_t * ep)
1883{
1884 vppcom_main_t *vcm = &vppcom_main;
1885 session_t *session = 0;
1886 int rv;
Dave Wallace33e002b2017-09-06 01:20:02 -04001887 ip46_address_t *ip46;
Dave Wallace543852a2017-08-03 02:11:34 -04001888
1889 if (!ep || !ep->ip)
1890 return VPPCOM_EINVAL;
1891
1892 clib_spinlock_lock (&vcm->sessions_lockp);
1893 rv = vppcom_session_at_index (session_index, &session);
1894 if (PREDICT_FALSE (rv))
1895 {
1896 clib_spinlock_unlock (&vcm->sessions_lockp);
1897 if (VPPCOM_DEBUG > 0)
1898 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1899 vcm->my_pid, session_index);
1900 return rv;
1901 }
1902
1903 if (VPPCOM_DEBUG > 0)
1904 clib_warning ("[%d] sid %d", vcm->my_pid, session_index);
1905
1906 session->vrf = ep->vrf;
1907 session->is_ip4 = ep->is_ip4;
Dave Wallace33e002b2017-09-06 01:20:02 -04001908 memset (session->ip, 0, sizeof (session->ip));
1909 ip46 = (ip46_address_t *) session->ip;
1910 *ip46 = to_ip46 (!ep->is_ip4, ep->ip);
Dave Wallace543852a2017-08-03 02:11:34 -04001911 session->port = ep->port;
1912
1913 clib_spinlock_unlock (&vcm->sessions_lockp);
1914 return VPPCOM_OK;
1915}
1916
1917int
Dave Wallace33e002b2017-09-06 01:20:02 -04001918vppcom_session_listen (uint32_t listen_session_index, uint32_t q_len)
Dave Wallace543852a2017-08-03 02:11:34 -04001919{
1920 vppcom_main_t *vcm = &vppcom_main;
Dave Wallace33e002b2017-09-06 01:20:02 -04001921 session_t *listen_session = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001922 int rv;
1923
1924 clib_spinlock_lock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04001925 rv = vppcom_session_at_index (listen_session_index, &listen_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001926 if (PREDICT_FALSE (rv))
1927 {
1928 clib_spinlock_unlock (&vcm->sessions_lockp);
1929 if (VPPCOM_DEBUG > 0)
1930 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
Dave Wallace33e002b2017-09-06 01:20:02 -04001931 vcm->my_pid, listen_session_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001932 return rv;
1933 }
1934
1935 if (VPPCOM_DEBUG > 0)
Dave Wallace33e002b2017-09-06 01:20:02 -04001936 clib_warning ("[%d] sid %d", vcm->my_pid, listen_session_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001937
1938 ASSERT (vcm->bind_session_index == ~0);
Dave Wallace33e002b2017-09-06 01:20:02 -04001939 vcm->bind_session_index = listen_session_index;
1940 vppcom_send_bind_sock (listen_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001941 clib_spinlock_unlock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04001942 rv =
1943 vppcom_wait_for_session_state_change (listen_session_index, STATE_LISTEN,
1944 vcm->cfg.session_timeout);
Dave Wallace543852a2017-08-03 02:11:34 -04001945 if (PREDICT_FALSE (rv))
1946 {
1947 vcm->bind_session_index = ~0;
1948 if (VPPCOM_DEBUG > 0)
1949 clib_warning ("[%d] server listen timed out, rv = %d (%d)",
1950 vcm->my_pid, vppcom_retval_str (rv), rv);
1951 return rv;
1952 }
1953
1954 clib_spinlock_lock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04001955 rv = vppcom_session_at_index (listen_session_index, &listen_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001956 if (PREDICT_FALSE (rv))
1957 {
1958 clib_spinlock_unlock (&vcm->sessions_lockp);
1959 if (VPPCOM_DEBUG > 0)
1960 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
Dave Wallace33e002b2017-09-06 01:20:02 -04001961 vcm->my_pid, listen_session_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001962 return rv;
1963 }
Dave Wallace33e002b2017-09-06 01:20:02 -04001964 listen_session->is_listen = 1;
Dave Wallace543852a2017-08-03 02:11:34 -04001965 clib_spinlock_unlock (&vcm->sessions_lockp);
1966 clib_fifo_validate (vcm->client_session_index_fifo, q_len);
1967
1968 return VPPCOM_OK;
1969}
1970
1971int
1972vppcom_session_accept (uint32_t listen_session_index, vppcom_endpt_t * ep,
1973 double wait_for_time)
1974{
1975 vppcom_main_t *vcm = &vppcom_main;
Dave Wallace33e002b2017-09-06 01:20:02 -04001976 session_t *listen_session = 0;
1977 session_t *client_session = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001978 u32 client_session_index;
1979 int rv;
1980 f64 wait_for;
1981
1982 clib_spinlock_lock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04001983 rv = vppcom_session_at_index (listen_session_index, &listen_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001984 if (PREDICT_FALSE (rv))
1985 {
1986 clib_spinlock_unlock (&vcm->sessions_lockp);
1987 if (VPPCOM_DEBUG > 0)
1988 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
1989 vcm->my_pid, listen_session_index);
1990 return rv;
1991 }
1992
Dave Wallace33e002b2017-09-06 01:20:02 -04001993 if (listen_session->state != STATE_LISTEN)
Dave Wallace543852a2017-08-03 02:11:34 -04001994 {
1995 clib_spinlock_unlock (&vcm->sessions_lockp);
1996 if (VPPCOM_DEBUG > 0)
1997 clib_warning ("[%d] session not in listen state, state = %s",
Dave Wallace33e002b2017-09-06 01:20:02 -04001998 vcm->my_pid,
1999 vppcom_session_state_str (listen_session->state));
Dave Wallace543852a2017-08-03 02:11:34 -04002000 return VPPCOM_EBADFD;
2001 }
Dave Wallace33e002b2017-09-06 01:20:02 -04002002 wait_for = listen_session->is_nonblocking ? 0 :
Dave Wallace543852a2017-08-03 02:11:34 -04002003 (wait_for_time < 0) ? vcm->cfg.accept_timeout : wait_for_time;
2004
2005 if (VPPCOM_DEBUG > 0)
Dave Wallace33e002b2017-09-06 01:20:02 -04002006 clib_warning ("[%d] sid %d: %s (%d)", vcm->my_pid,
Dave Wallace543852a2017-08-03 02:11:34 -04002007 listen_session_index,
Dave Wallace33e002b2017-09-06 01:20:02 -04002008 vppcom_session_state_str (listen_session->state),
2009 listen_session->state);
Dave Wallace543852a2017-08-03 02:11:34 -04002010 clib_spinlock_unlock (&vcm->sessions_lockp);
2011
2012 while (1)
2013 {
2014 rv = vppcom_wait_for_client_session_index (wait_for);
2015 if (rv)
2016 {
2017 if ((VPPCOM_DEBUG > 0))
2018 clib_warning ("[%d] sid %d, accept timed out, rv = %s (%d)",
2019 vcm->my_pid, listen_session_index,
2020 vppcom_retval_str (rv), rv);
2021 if ((wait_for == 0) || (wait_for_time > 0))
2022 return rv;
2023 }
2024 else
2025 break;
2026 }
2027
2028 clib_fifo_sub1 (vcm->client_session_index_fifo, client_session_index);
2029
Dave Wallace543852a2017-08-03 02:11:34 -04002030 clib_spinlock_lock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04002031 rv = vppcom_session_at_index (client_session_index, &client_session);
Dave Wallace543852a2017-08-03 02:11:34 -04002032 ASSERT (rv == VPPCOM_OK);
Dave Wallace33e002b2017-09-06 01:20:02 -04002033 ASSERT (client_session->is_ip4 == listen_session->is_ip4);
Dave Wallace543852a2017-08-03 02:11:34 -04002034
2035 if (VPPCOM_DEBUG > 0)
2036 clib_warning ("[%d] Got a request: client sid %d", vcm->my_pid,
2037 client_session_index);
2038
Dave Wallace33e002b2017-09-06 01:20:02 -04002039 ep->vrf = client_session->vrf;
2040 ep->is_cut_thru = client_session->is_cut_thru;
2041 ep->is_ip4 = client_session->is_ip4;
2042 ep->port = client_session->port;
2043 if (client_session->is_ip4)
2044 clib_memcpy (ep->ip, client_session->ip, sizeof (ip4_address_t));
2045 else
2046 clib_memcpy (ep->ip, client_session->ip, sizeof (ip6_address_t));
Dave Wallace543852a2017-08-03 02:11:34 -04002047 clib_spinlock_unlock (&vcm->sessions_lockp);
2048 return (int) client_session_index;
2049}
2050
2051int
2052vppcom_session_connect (uint32_t session_index, vppcom_endpt_t * server_ep)
2053{
2054 vppcom_main_t *vcm = &vppcom_main;
2055 session_t *session = 0;
2056 int rv;
2057 ip46_address_t *ip46;
2058
2059 clib_spinlock_lock (&vcm->sessions_lockp);
2060 rv = vppcom_session_at_index (session_index, &session);
2061 if (PREDICT_FALSE (rv))
2062 {
2063 clib_spinlock_unlock (&vcm->sessions_lockp);
2064 if (VPPCOM_DEBUG > 0)
2065 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2066 vcm->my_pid, session_index);
2067 return rv;
2068 }
2069
2070 if (session->state == STATE_CONNECT)
2071 {
2072 clib_spinlock_unlock (&vcm->sessions_lockp);
2073 if (VPPCOM_DEBUG > 0)
2074 clib_warning ("[%d] session, sid (%d) already connected!",
2075 vcm->my_pid, session_index);
2076 return VPPCOM_OK;
2077 }
2078
2079 session->vrf = server_ep->vrf;
2080 session->is_ip4 = server_ep->is_ip4;
Dave Wallace33e002b2017-09-06 01:20:02 -04002081 memset (session->ip, 0, sizeof (session->ip));
Dave Wallace543852a2017-08-03 02:11:34 -04002082 ip46 = (ip46_address_t *) session->ip;
2083 *ip46 = to_ip46 (!server_ep->is_ip4, server_ep->ip);
2084 session->port = server_ep->port;
2085
2086 if (VPPCOM_DEBUG > 0)
2087 {
2088 u8 *ip_str = format (0, "%U", format_ip46_address,
2089 &session->ip, session->is_ip4);
2090 clib_warning ("[%d] connect sid %d to %s server port %d",
2091 vcm->my_pid, session_index, ip_str,
2092 clib_net_to_host_u16 (session->port));
2093 vec_free (ip_str);
2094 }
2095
2096 vppcom_send_connect_sock (session, session_index);
2097 clib_spinlock_unlock (&vcm->sessions_lockp);
2098 rv = vppcom_wait_for_session_state_change (session_index, STATE_CONNECT,
2099 vcm->cfg.session_timeout);
2100 if (PREDICT_FALSE (rv))
2101 {
2102 if (VPPCOM_DEBUG > 0)
2103 clib_warning ("[%d] connect timed out, rv = %s (%d)",
2104 vcm->my_pid, vppcom_retval_str (rv), rv);
2105 return rv;
2106 }
2107 return VPPCOM_OK;
2108}
2109
2110int
2111vppcom_session_read (uint32_t session_index, void *buf, int n)
2112{
Dave Wallace543852a2017-08-03 02:11:34 -04002113 vppcom_main_t *vcm = &vppcom_main;
2114 session_t *session = 0;
2115 svm_fifo_t *rx_fifo;
2116 int n_read = 0;
2117 int rv;
Dave Wallace33e002b2017-09-06 01:20:02 -04002118 int max_dequeue;
Dave Wallace543852a2017-08-03 02:11:34 -04002119 char *fifo_str;
2120
2121 ASSERT (buf);
2122
2123 clib_spinlock_lock (&vcm->sessions_lockp);
2124 rv = vppcom_session_at_index (session_index, &session);
2125 if (PREDICT_FALSE (rv))
2126 {
2127 clib_spinlock_unlock (&vcm->sessions_lockp);
2128 if (VPPCOM_DEBUG > 0)
2129 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2130 vcm->my_pid, session_index);
2131 return rv;
2132 }
2133
Dave Wallace33e002b2017-09-06 01:20:02 -04002134 if (session->state == STATE_DISCONNECT)
Dave Wallace543852a2017-08-03 02:11:34 -04002135 {
Dave Wallace543852a2017-08-03 02:11:34 -04002136 clib_spinlock_unlock (&vcm->sessions_lockp);
Dave Wallace33e002b2017-09-06 01:20:02 -04002137 if (VPPCOM_DEBUG > 0)
2138 clib_warning ("[%d] sid (%d) has been closed by remote peer!",
2139 vcm->my_pid, session_index);
2140 return VPPCOM_ECONNRESET;
Dave Wallace543852a2017-08-03 02:11:34 -04002141 }
Dave Wallace543852a2017-08-03 02:11:34 -04002142
Dave Wallace33e002b2017-09-06 01:20:02 -04002143 rx_fifo = ((!session->is_cut_thru || session->is_server) ?
2144 session->server_rx_fifo : session->server_tx_fifo);
2145 fifo_str = ((!session->is_cut_thru || session->is_server) ?
2146 "server_rx_fifo" : "server_tx_fifo");
2147 clib_spinlock_unlock (&vcm->sessions_lockp);
Dave Wallace543852a2017-08-03 02:11:34 -04002148
Dave Wallace33e002b2017-09-06 01:20:02 -04002149 max_dequeue = (int) svm_fifo_max_dequeue (rx_fifo);
2150 n_read = svm_fifo_dequeue_nowait (rx_fifo, clib_min (n, max_dequeue), buf);
Dave Wallace543852a2017-08-03 02:11:34 -04002151
2152 if (VPPCOM_DEBUG > 2)
2153 clib_warning ("[%d] sid %d, read %d bytes from %s (%p)", vcm->my_pid,
2154 session_index, n_read, fifo_str, rx_fifo);
Dave Wallace33e002b2017-09-06 01:20:02 -04002155
2156 return (n_read <= 0) ? VPPCOM_EAGAIN : n_read;
Dave Wallace543852a2017-08-03 02:11:34 -04002157}
2158
2159static inline int
2160vppcom_session_read_ready (session_t * session, u32 session_index)
2161{
Dave Wallace543852a2017-08-03 02:11:34 -04002162 vppcom_main_t *vcm = &vppcom_main;
2163 svm_fifo_t *rx_fifo;
Dave Wallace543852a2017-08-03 02:11:34 -04002164 int ready = 0;
2165
2166 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
Dave Wallace33e002b2017-09-06 01:20:02 -04002167 if (session->state == STATE_DISCONNECT)
Dave Wallace543852a2017-08-03 02:11:34 -04002168 {
Dave Wallace33e002b2017-09-06 01:20:02 -04002169 if (VPPCOM_DEBUG > 0)
2170 clib_warning ("[%d] sid (%d) has been closed by remote peer!",
2171 vcm->my_pid, session_index);
2172 return VPPCOM_ECONNRESET;
Dave Wallace543852a2017-08-03 02:11:34 -04002173 }
Dave Wallace33e002b2017-09-06 01:20:02 -04002174
2175 if (session->is_listen)
Dave Wallace543852a2017-08-03 02:11:34 -04002176 ready = clib_fifo_elts (vcm->client_session_index_fifo);
2177 else
2178 {
Dave Wallace33e002b2017-09-06 01:20:02 -04002179 rx_fifo = ((!session->is_cut_thru || session->is_server) ?
2180 session->server_rx_fifo : session->server_tx_fifo);
Dave Wallace543852a2017-08-03 02:11:34 -04002181
Dave Wallace33e002b2017-09-06 01:20:02 -04002182 ready = svm_fifo_max_dequeue (rx_fifo);
Dave Wallace543852a2017-08-03 02:11:34 -04002183 }
2184
Dave Wallace33e002b2017-09-06 01:20:02 -04002185 if (VPPCOM_DEBUG > 3)
Dave Wallace543852a2017-08-03 02:11:34 -04002186 clib_warning ("[%d] sid %d, peek %s (%p), ready = %d", vcm->my_pid,
2187 session_index,
2188 session->is_server ? "server_rx_fifo" : "server_tx_fifo",
2189 rx_fifo, ready);
Dave Wallace543852a2017-08-03 02:11:34 -04002190 return ready;
2191}
2192
2193int
2194vppcom_session_write (uint32_t session_index, void *buf, int n)
2195{
2196 vppcom_main_t *vcm = &vppcom_main;
2197 session_t *session = 0;
2198 svm_fifo_t *tx_fifo;
2199 unix_shared_memory_queue_t *q;
2200 session_fifo_event_t evt;
2201 int rv;
2202 char *fifo_str;
2203 u8 is_nonblocking;
2204
2205 ASSERT (buf);
2206
2207 clib_spinlock_lock (&vcm->sessions_lockp);
2208 rv = vppcom_session_at_index (session_index, &session);
2209 if (PREDICT_FALSE (rv))
2210 {
2211 clib_spinlock_unlock (&vcm->sessions_lockp);
2212 if (VPPCOM_DEBUG > 0)
2213 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2214 vcm->my_pid, session_index);
2215 return rv;
2216 }
2217
Dave Wallace33e002b2017-09-06 01:20:02 -04002218 if (session->state == STATE_DISCONNECT)
2219 {
2220 clib_spinlock_unlock (&vcm->sessions_lockp);
2221 if (VPPCOM_DEBUG > 0)
2222 clib_warning ("[%d] sid (%d) has been closed by remote peer!",
2223 vcm->my_pid, session_index);
2224 return VPPCOM_ECONNRESET;
2225 }
2226
Dave Wallace543852a2017-08-03 02:11:34 -04002227 tx_fifo = ((!session->is_cut_thru || session->is_server) ?
2228 session->server_tx_fifo : session->server_rx_fifo);
2229 fifo_str = ((!session->is_cut_thru || session->is_server) ?
2230 "server_tx_fifo" : "server_rx_fifo");
Dave Wallace33e002b2017-09-06 01:20:02 -04002231
Dave Wallace543852a2017-08-03 02:11:34 -04002232 is_nonblocking = session->is_nonblocking;
2233 clib_spinlock_unlock (&vcm->sessions_lockp);
2234
2235 do
2236 {
2237 rv = svm_fifo_enqueue_nowait (tx_fifo, n, buf);
2238 }
2239 while (!is_nonblocking && (rv <= 0));
2240
2241 /* If event wasn't set, add one */
Dave Wallace33e002b2017-09-06 01:20:02 -04002242 if (!session->is_cut_thru && (rv > 0) && svm_fifo_set_event (tx_fifo))
Dave Wallace543852a2017-08-03 02:11:34 -04002243 {
2244 int rval;
2245
2246 /* Fabricate TX event, send to vpp */
2247 evt.fifo = tx_fifo;
2248 evt.event_type = FIFO_EVENT_APP_TX;
2249 evt.event_id = vcm->tx_event_id++;
2250
2251 clib_spinlock_lock (&vcm->sessions_lockp);
2252 rval = vppcom_session_at_index (session_index, &session);
2253 if (PREDICT_FALSE (rval))
2254 {
2255 clib_spinlock_unlock (&vcm->sessions_lockp);
2256 if (VPPCOM_DEBUG > 1)
2257 clib_warning ("[%d] invalid session, sid (%d) has been closed!",
2258 vcm->my_pid, session_index);
2259 return rval;
2260 }
Dave Wallace33e002b2017-09-06 01:20:02 -04002261 q = session->vpp_event_queue;
Dave Wallace543852a2017-08-03 02:11:34 -04002262 clib_spinlock_unlock (&vcm->sessions_lockp);
2263 ASSERT (q);
2264 unix_shared_memory_queue_add (q, (u8 *) & evt,
2265 0 /* do wait for mutex */ );
2266 }
2267
2268 if (VPPCOM_DEBUG > 2)
2269 clib_warning ("[%d] sid %d, wrote %d bytes to %s (%p)", vcm->my_pid,
2270 session_index, rv, fifo_str, tx_fifo);
2271
2272 return rv;
2273}
2274
2275static inline int
2276vppcom_session_write_ready (session_t * session, u32 session_index)
2277{
2278 vppcom_main_t *vcm = &vppcom_main;
2279 svm_fifo_t *tx_fifo;
Dave Wallace33e002b2017-09-06 01:20:02 -04002280 char *fifo_str;
Dave Wallace543852a2017-08-03 02:11:34 -04002281 int rv;
2282
2283 /* Assumes caller has acquired spinlock: vcm->sessions_lockp */
Dave Wallace33e002b2017-09-06 01:20:02 -04002284 if (session->state == STATE_DISCONNECT)
2285 {
2286 if (VPPCOM_DEBUG > 0)
2287 clib_warning ("[%d] sid (%d) has been closed by remote peer!",
2288 vcm->my_pid, session_index);
2289 return VPPCOM_ECONNRESET;
2290 }
2291
Dave Wallace543852a2017-08-03 02:11:34 -04002292 tx_fifo = ((!session->is_cut_thru || session->is_server) ?
2293 session->server_tx_fifo : session->server_rx_fifo);
Dave Wallace33e002b2017-09-06 01:20:02 -04002294 fifo_str = ((!session->is_cut_thru || session->is_server) ?
2295 "server_tx_fifo" : "server_rx_fifo");
Dave Wallace543852a2017-08-03 02:11:34 -04002296
2297 rv = svm_fifo_max_enqueue (tx_fifo);
2298
Dave Wallace33e002b2017-09-06 01:20:02 -04002299 if (VPPCOM_DEBUG > 3)
Dave Wallace543852a2017-08-03 02:11:34 -04002300 clib_warning ("[%d] sid %d, peek %s (%p), ready = %d", vcm->my_pid,
Dave Wallace33e002b2017-09-06 01:20:02 -04002301 session_index, fifo_str, tx_fifo, rv);
Dave Wallace543852a2017-08-03 02:11:34 -04002302 return rv;
2303}
2304
2305int
2306vppcom_select (unsigned long n_bits, unsigned long *read_map,
2307 unsigned long *write_map, unsigned long *except_map,
2308 double time_to_wait)
2309{
2310 vppcom_main_t *vcm = &vppcom_main;
2311 u32 session_index;
2312 session_t *session = 0;
2313 int rv, bits_set = 0;
2314 f64 timeout = clib_time_now (&vcm->clib_time) + time_to_wait;
2315 u32 minbits = clib_max (n_bits, BITS (uword));
2316
2317 ASSERT (sizeof (clib_bitmap_t) == sizeof (long int));
2318
2319 if (read_map)
2320 {
2321 clib_bitmap_validate (vcm->rd_bitmap, minbits);
2322 clib_memcpy (vcm->rd_bitmap, read_map, vec_len (vcm->rd_bitmap));
2323 memset (read_map, 0, vec_len (vcm->rd_bitmap));
2324 }
2325 if (write_map)
2326 {
2327 clib_bitmap_validate (vcm->wr_bitmap, minbits);
2328 clib_memcpy (vcm->wr_bitmap, write_map, vec_len (vcm->wr_bitmap));
2329 memset (write_map, 0, vec_len (vcm->wr_bitmap));
2330 }
2331 if (except_map)
2332 {
2333 clib_bitmap_validate (vcm->ex_bitmap, minbits);
2334 clib_memcpy (vcm->ex_bitmap, except_map, vec_len (vcm->ex_bitmap));
2335 memset (except_map, 0, vec_len (vcm->ex_bitmap));
2336 }
2337
2338 do
2339 {
2340 /* *INDENT-OFF* */
2341 clib_bitmap_foreach (session_index, vcm->rd_bitmap,
2342 ({
2343 clib_spinlock_lock (&vcm->sessions_lockp);
2344 rv = vppcom_session_at_index (session_index, &session);
2345 if (rv < 0)
2346 {
2347 clib_spinlock_unlock (&vcm->sessions_lockp);
2348 if (VPPCOM_DEBUG > 1)
2349 clib_warning ("[%d] session %d specified in "
2350 "read_map is closed.", vcm->my_pid,
2351 session_index);
2352 bits_set = VPPCOM_EBADFD;
2353 goto select_done;
2354 }
2355
2356 rv = vppcom_session_read_ready (session, session_index);
2357 clib_spinlock_unlock (&vcm->sessions_lockp);
2358 if (vcm->ex_bitmap &&
2359 clib_bitmap_get (vcm->ex_bitmap, session_index) && (rv < 0))
2360 {
2361 // TBD: clib_warning
Chris Lukeb2bcad62017-09-18 08:51:22 -04002362 /* coverity[FORWARD_NULL] */
Dave Wallace543852a2017-08-03 02:11:34 -04002363 clib_bitmap_set_no_check (except_map, session_index, 1);
2364 bits_set++;
2365 }
2366 else if (rv > 0)
2367 {
2368 // TBD: clib_warning
Chris Lukeb2bcad62017-09-18 08:51:22 -04002369 /* coverity[FORWARD_NULL] */
Dave Wallace543852a2017-08-03 02:11:34 -04002370 clib_bitmap_set_no_check (read_map, session_index, 1);
2371 bits_set++;
2372 }
2373 }));
2374
2375 clib_bitmap_foreach (session_index, vcm->wr_bitmap,
2376 ({
2377 clib_spinlock_lock (&vcm->sessions_lockp);
2378 rv = vppcom_session_at_index (session_index, &session);
2379 if (rv < 0)
2380 {
2381 clib_spinlock_unlock (&vcm->sessions_lockp);
2382 if (VPPCOM_DEBUG > 0)
2383 clib_warning ("[%d] session %d specified in "
2384 "write_map is closed.", vcm->my_pid,
2385 session_index);
2386 bits_set = VPPCOM_EBADFD;
2387 goto select_done;
2388 }
2389
2390 rv = vppcom_session_write_ready (session, session_index);
2391 clib_spinlock_unlock (&vcm->sessions_lockp);
Chris Lukeb2bcad62017-09-18 08:51:22 -04002392 if (rv > 0 )
Dave Wallace543852a2017-08-03 02:11:34 -04002393 {
2394 // TBD: clib_warning
Chris Lukeb2bcad62017-09-18 08:51:22 -04002395 /* coverity[FORWARD_NULL] */
Dave Wallace543852a2017-08-03 02:11:34 -04002396 clib_bitmap_set_no_check (write_map, session_index, 1);
2397 bits_set++;
2398 }
2399 }));
2400
2401 clib_bitmap_foreach (session_index, vcm->ex_bitmap,
2402 ({
2403 clib_spinlock_lock (&vcm->sessions_lockp);
2404 rv = vppcom_session_at_index (session_index, &session);
2405 if (rv < 0)
2406 {
2407 clib_spinlock_unlock (&vcm->sessions_lockp);
2408 if (VPPCOM_DEBUG > 1)
2409 clib_warning ("[%d] session %d specified in "
2410 "except_map is closed.", vcm->my_pid,
2411 session_index);
2412 bits_set = VPPCOM_EBADFD;
2413 goto select_done;
2414 }
2415
2416 rv = vppcom_session_read_ready (session, session_index);
2417 clib_spinlock_unlock (&vcm->sessions_lockp);
2418 if (rv < 0)
2419 {
2420 // TBD: clib_warning
Chris Lukeb2bcad62017-09-18 08:51:22 -04002421 /* coverity[FORWARD_NULL] */
Dave Wallace543852a2017-08-03 02:11:34 -04002422 clib_bitmap_set_no_check (except_map, session_index, 1);
2423 bits_set++;
2424 }
2425 }));
2426 /* *INDENT-ON* */
2427 }
2428 while (clib_time_now (&vcm->clib_time) < timeout);
2429
2430select_done:
2431 return (bits_set);
2432}
2433
2434/*
2435 * fd.io coding-style-patch-verification: ON
2436 *
2437 * Local Variables:
2438 * eval: (c-set-style "gnu")
2439 * End:
2440 */