2 *------------------------------------------------------------------
3 * Copyright (c) 2017 Cisco and/or its affiliates.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *------------------------------------------------------------------
21 #include <arpa/inet.h>
25 #include <vpp-api/vapi/vapi_dbg.h>
26 #include <vpp-api/vapi/vapi.h>
27 #include <vpp-api/vapi/vapi_internal.h>
28 #include <vppinfra/types.h>
29 #include <vppinfra/pool.h>
30 #include <vlib/vlib.h>
31 #include <vlibapi/api_common.h>
32 #include <vlibmemory/memory_client.h>
34 /* we need to use control pings for some stuff and because we're forced to put
35 * the code in headers, we need a way to be able to grab the ids of these
36 * messages - so declare them here as extern */
37 vapi_msg_id_t vapi_msg_id_control_ping = 0;
38 vapi_msg_id_t vapi_msg_id_control_ping_reply = 0;
43 vapi_message_desc_t **msgs;
44 size_t max_len_name_with_crc;
55 static const u32 context_counter_mask = (1 << 31);
59 vapi_error_e (*cb) (vapi_ctx_t ctx, void *callback_ctx, vapi_msg_id_t id,
62 } vapi_generic_cb_with_ctx;
66 vapi_error_e (*cb) (vapi_ctx_t ctx, void *callback_ctx, void *payload);
68 } vapi_event_cb_with_ctx;
73 int requests_size; /* size of the requests array (circular queue) */
74 int requests_start; /* index of first request */
75 int requests_count; /* number of used slots */
78 vapi_generic_cb_with_ctx generic_cb;
79 vapi_event_cb_with_ctx *event_cbs;
80 u16 *vapi_msg_id_t_to_vl_msg_id;
82 vapi_msg_id_t *vl_msg_id_to_vapi_msg_t;
84 pthread_mutex_t requests_mutex;
88 vapi_gen_req_context (vapi_ctx_t ctx)
90 ++ctx->context_counter;
91 ctx->context_counter %= context_counter_mask;
92 return ctx->context_counter | context_counter_mask;
96 vapi_get_request_count (vapi_ctx_t ctx)
98 return ctx->requests_count;
102 vapi_requests_full (vapi_ctx_t ctx)
104 return (ctx->requests_count == ctx->requests_size);
108 vapi_requests_empty (vapi_ctx_t ctx)
110 return (0 == ctx->requests_count);
114 vapi_requests_end (vapi_ctx_t ctx)
116 return (ctx->requests_start + ctx->requests_count) % ctx->requests_size;
120 vapi_store_request (vapi_ctx_t ctx, u32 context, bool is_dump,
121 vapi_cb_t callback, void *callback_ctx)
123 assert (!vapi_requests_full (ctx));
124 /* if the mutex is not held, bad things will happen */
125 assert (0 != pthread_mutex_trylock (&ctx->requests_mutex));
126 const int requests_end = vapi_requests_end (ctx);
127 vapi_req_t *slot = &ctx->requests[requests_end];
128 slot->is_dump = is_dump;
129 slot->context = context;
130 slot->callback = callback;
131 slot->callback_ctx = callback_ctx;
132 VAPI_DBG ("stored@%d: context:%x (start is @%d)", requests_end, context,
133 ctx->requests_start);
134 ++ctx->requests_count;
135 assert (!vapi_requests_empty (ctx));
139 struct to_be_freed_s;
143 struct to_be_freed_s *next;
146 static struct to_be_freed_s *to_be_freed = NULL;
149 vapi_add_to_be_freed (void *v)
151 struct to_be_freed_s *prev = NULL;
152 struct to_be_freed_s *tmp;
154 while (tmp && tmp->v)
163 tmp = to_be_freed = calloc (1, sizeof (*to_be_freed));
167 tmp = prev->next = calloc (1, sizeof (*to_be_freed));
170 VAPI_DBG ("To be freed %p", v);
175 vapi_trace_free (void *v)
177 struct to_be_freed_s *tmp = to_be_freed;
178 while (tmp && tmp->v != v)
182 if (tmp && tmp->v == v)
184 VAPI_DBG ("Freed %p", v);
189 VAPI_ERR ("Trying to free untracked pointer %p", v);
195 vapi_to_be_freed_validate ()
197 struct to_be_freed_s *tmp = to_be_freed;
202 VAPI_ERR ("Unfreed msg %p!", tmp->v);
211 vapi_msg_alloc (vapi_ctx_t ctx, size_t size)
217 void *rv = vl_msg_api_alloc_or_null (size);
222 vapi_msg_free (vapi_ctx_t ctx, void *msg)
229 vapi_trace_free (msg);
231 vl_msg_api_free (msg);
235 vapi_lookup_vapi_msg_id_t (vapi_ctx_t ctx, u16 vl_msg_id)
237 if (vl_msg_id <= ctx->vl_msg_id_max)
239 return ctx->vl_msg_id_to_vapi_msg_t[vl_msg_id];
241 return INVALID_MSG_ID;
245 vapi_ctx_alloc (vapi_ctx_t * result)
247 vapi_ctx_t ctx = calloc (1, sizeof (struct vapi_ctx_s));
252 ctx->context_counter = 0;
253 ctx->vapi_msg_id_t_to_vl_msg_id =
254 malloc (__vapi_metadata.count *
255 sizeof (*ctx->vapi_msg_id_t_to_vl_msg_id));
256 if (!ctx->vapi_msg_id_t_to_vl_msg_id)
260 ctx->event_cbs = calloc (__vapi_metadata.count, sizeof (*ctx->event_cbs));
265 pthread_mutex_init (&ctx->requests_mutex, NULL);
274 vapi_ctx_free (vapi_ctx_t ctx)
276 assert (!ctx->connected);
277 free (ctx->requests);
278 free (ctx->vapi_msg_id_t_to_vl_msg_id);
279 free (ctx->event_cbs);
280 free (ctx->vl_msg_id_to_vapi_msg_t);
281 pthread_mutex_destroy (&ctx->requests_mutex);
286 vapi_is_msg_available (vapi_ctx_t ctx, vapi_msg_id_t id)
288 return vapi_lookup_vl_msg_id (ctx, id) != UINT16_MAX;
292 vapi_connect (vapi_ctx_t ctx, const char *name,
293 const char *chroot_prefix,
294 int max_outstanding_requests,
295 int response_queue_size, vapi_mode_e mode)
297 if (response_queue_size <= 0 || max_outstanding_requests <= 0)
301 ctx->requests_size = max_outstanding_requests;
302 const size_t size = ctx->requests_size * sizeof (*ctx->requests);
303 void *tmp = realloc (ctx->requests, size);
309 memset (ctx->requests, 0, size);
310 /* coverity[MISSING_LOCK] - 177211 requests_mutex is not needed here */
311 ctx->requests_start = ctx->requests_count = 0;
314 VAPI_DBG ("set memory root path `%s'", chroot_prefix);
315 vl_set_memory_root_path ((char *) chroot_prefix);
317 static char api_map[] = "/vpe-api";
318 VAPI_DBG ("client api map `%s'", api_map);
319 if ((vl_client_api_map (api_map)) < 0)
321 return VAPI_EMAP_FAIL;
323 VAPI_DBG ("connect client `%s'", name);
324 if (vl_client_connect ((char *) name, 0, response_queue_size) < 0)
326 vl_client_api_unmap ();
327 return VAPI_ECON_FAIL;
329 #if VAPI_DEBUG_CONNECT
330 VAPI_DBG ("start probing messages");
334 for (i = 0; i < __vapi_metadata.count; ++i)
336 vapi_message_desc_t *m = __vapi_metadata.msgs[i];
337 u8 scratch[m->name_with_crc_len + 1];
338 memcpy (scratch, m->name_with_crc, m->name_with_crc_len + 1);
339 u32 id = vl_msg_api_get_msg_index (scratch);
340 if (INVALID_MSG_ID != id)
344 VAPI_ERR ("Returned vl_msg_id `%u' > UINT16MAX `%u'!", id,
349 if (id > ctx->vl_msg_id_max)
351 vapi_msg_id_t *tmp = realloc (ctx->vl_msg_id_to_vapi_msg_t,
353 (*ctx->vl_msg_id_to_vapi_msg_t) *
360 ctx->vl_msg_id_to_vapi_msg_t = tmp;
361 ctx->vl_msg_id_max = id;
363 ctx->vl_msg_id_to_vapi_msg_t[id] = m->id;
364 ctx->vapi_msg_id_t_to_vl_msg_id[m->id] = id;
365 #if VAPI_DEBUG_CONNECT
366 VAPI_DBG ("Message `%s' has vl_msg_id `%u'", m->name_with_crc,
372 ctx->vapi_msg_id_t_to_vl_msg_id[m->id] = UINT16_MAX;
373 VAPI_DBG ("Message `%s' not available", m->name_with_crc);
376 #if VAPI_DEBUG_CONNECT
377 VAPI_DBG ("finished probing messages");
379 if (!vapi_is_msg_available (ctx, vapi_msg_id_control_ping) ||
380 !vapi_is_msg_available (ctx, vapi_msg_id_control_ping_reply))
383 ("control ping or control ping reply not available, cannot connect");
384 rv = VAPI_EINCOMPATIBLE;
388 ctx->connected = true;
391 vl_client_disconnect ();
392 vl_client_api_unmap ();
397 vapi_disconnect (vapi_ctx_t ctx)
403 vl_client_disconnect ();
404 vl_client_api_unmap ();
406 vapi_to_be_freed_validate ();
408 ctx->connected = false;
413 vapi_get_fd (vapi_ctx_t ctx, int *fd)
419 vapi_send (vapi_ctx_t ctx, void *msg)
421 vapi_error_e rv = VAPI_OK;
422 if (!ctx || !msg || !ctx->connected)
428 svm_queue_t *q = api_main.shmem_hdr->vl_input_queue;
430 unsigned msgid = be16toh (*(u16 *) msg);
431 if (msgid <= ctx->vl_msg_id_max)
433 vapi_msg_id_t id = ctx->vl_msg_id_to_vapi_msg_t[msgid];
434 if (id < __vapi_metadata.count)
436 VAPI_DBG ("send msg@%p:%u[%s]", msg, msgid,
437 __vapi_metadata.msgs[id]->name);
441 VAPI_DBG ("send msg@%p:%u[UNKNOWN]", msg, msgid);
446 VAPI_DBG ("send msg@%p:%u[UNKNOWN]", msg, msgid);
449 tmp = svm_queue_add (q, (u8 *) & msg,
450 VAPI_MODE_BLOCKING == ctx->mode ? 0 : 1);
456 VAPI_DBG ("vapi_send() rv = %d", rv);
461 vapi_send2 (vapi_ctx_t ctx, void *msg1, void *msg2)
463 vapi_error_e rv = VAPI_OK;
464 if (!ctx || !msg1 || !msg2 || !ctx->connected)
469 svm_queue_t *q = api_main.shmem_hdr->vl_input_queue;
471 unsigned msgid1 = be16toh (*(u16 *) msg1);
472 unsigned msgid2 = be16toh (*(u16 *) msg2);
473 const char *name1 = "UNKNOWN";
474 const char *name2 = "UNKNOWN";
475 if (msgid1 <= ctx->vl_msg_id_max)
477 vapi_msg_id_t id = ctx->vl_msg_id_to_vapi_msg_t[msgid1];
478 if (id < __vapi_metadata.count)
480 name1 = __vapi_metadata.msgs[id]->name;
483 if (msgid2 <= ctx->vl_msg_id_max)
485 vapi_msg_id_t id = ctx->vl_msg_id_to_vapi_msg_t[msgid2];
486 if (id < __vapi_metadata.count)
488 name2 = __vapi_metadata.msgs[id]->name;
491 VAPI_DBG ("send two: %u[%s], %u[%s]", msgid1, name1, msgid2, name2);
493 int tmp = svm_queue_add2 (q, (u8 *) & msg1, (u8 *) & msg2,
494 VAPI_MODE_BLOCKING == ctx->mode ? 0 : 1);
500 VAPI_DBG ("vapi_send() rv = %d", rv);
505 vapi_recv (vapi_ctx_t ctx, void **msg, size_t * msg_size,
506 svm_q_conditional_wait_t cond, u32 time)
508 if (!ctx || !ctx->connected || !msg || !msg_size)
512 vapi_error_e rv = VAPI_OK;
513 api_main_t *am = &api_main;
516 if (am->our_pid == 0)
521 svm_queue_t *q = am->vl_input_queue;
522 VAPI_DBG ("doing shm queue sub");
524 int tmp = svm_queue_sub (q, (u8 *) & data, cond, time);
529 vapi_add_to_be_freed ((void *) data);
532 (msgbuf_t *) ((u8 *) data - offsetof (msgbuf_t, data));
533 if (!msgbuf->data_len)
535 vapi_msg_free (ctx, (u8 *) data);
539 *msg_size = ntohl (msgbuf->data_len);
541 unsigned msgid = be16toh (*(u16 *) * msg);
542 if (msgid <= ctx->vl_msg_id_max)
544 vapi_msg_id_t id = ctx->vl_msg_id_to_vapi_msg_t[msgid];
545 if (id < __vapi_metadata.count)
547 VAPI_DBG ("recv msg@%p:%u[%s]", *msg, msgid,
548 __vapi_metadata.msgs[id]->name);
552 VAPI_DBG ("recv msg@%p:%u[UNKNOWN]", *msg, msgid);
557 VAPI_DBG ("recv msg@%p:%u[UNKNOWN]", *msg, msgid);
569 vapi_wait (vapi_ctx_t ctx, vapi_wait_mode_e mode)
575 vapi_dispatch_response (vapi_ctx_t ctx, vapi_msg_id_t id,
576 u32 context, void *msg)
579 if (0 != (mrv = pthread_mutex_lock (&ctx->requests_mutex)))
581 VAPI_DBG ("pthread_mutex_lock() failed, rv=%d:%s", mrv, strerror (mrv));
582 return VAPI_MUTEX_FAILURE;
584 int tmp = ctx->requests_start;
585 const int requests_end = vapi_requests_end (ctx);
586 while (ctx->requests[tmp].context != context && tmp != requests_end)
589 if (tmp == ctx->requests_size)
594 VAPI_DBG ("dispatch, search from %d, %s at %d", ctx->requests_start,
595 ctx->requests[tmp].context == context ? "matched" : "stopped",
597 vapi_error_e rv = VAPI_OK;
598 if (ctx->requests[tmp].context == context)
600 while (ctx->requests_start != tmp)
602 VAPI_ERR ("No response to req with context=%u",
603 (unsigned) ctx->requests[tmp].context);
604 ctx->requests[ctx->requests_start].callback (ctx,
607 requests_start].callback_ctx,
610 memset (&ctx->requests[ctx->requests_start], 0,
611 sizeof (ctx->requests[ctx->requests_start]));
612 ++ctx->requests_start;
613 --ctx->requests_count;
614 if (ctx->requests_start == ctx->requests_size)
616 ctx->requests_start = 0;
619 // now ctx->requests_start == tmp
620 int payload_offset = vapi_get_payload_offset (id);
621 void *payload = ((u8 *) msg) + payload_offset;
623 if (ctx->requests[tmp].is_dump)
625 if (vapi_msg_id_control_ping_reply == id)
634 if (payload_offset != -1)
637 ctx->requests[tmp].callback (ctx, ctx->requests[tmp].callback_ctx,
638 VAPI_OK, is_last, payload);
642 /* this is a message without payload, so bend the callback a little
645 ((vapi_error_e (*)(vapi_ctx_t, void *, vapi_error_e, bool))
646 ctx->requests[tmp].callback) (ctx,
647 ctx->requests[tmp].callback_ctx,
652 memset (&ctx->requests[ctx->requests_start], 0,
653 sizeof (ctx->requests[ctx->requests_start]));
654 ++ctx->requests_start;
655 --ctx->requests_count;
656 if (ctx->requests_start == ctx->requests_size)
658 ctx->requests_start = 0;
661 VAPI_DBG ("after dispatch, req start = %d, end = %d, count = %d",
662 ctx->requests_start, requests_end, ctx->requests_count);
664 if (0 != (mrv = pthread_mutex_unlock (&ctx->requests_mutex)))
666 VAPI_DBG ("pthread_mutex_unlock() failed, rv=%d:%s", mrv,
668 abort (); /* this really shouldn't happen */
674 vapi_dispatch_event (vapi_ctx_t ctx, vapi_msg_id_t id, void *msg)
676 if (ctx->event_cbs[id].cb)
678 return ctx->event_cbs[id].cb (ctx, ctx->event_cbs[id].ctx, msg);
680 else if (ctx->generic_cb.cb)
682 return ctx->generic_cb.cb (ctx, ctx->generic_cb.ctx, id, msg);
687 ("No handler/generic handler for msg id %u[%s], message ignored",
688 (unsigned) id, __vapi_metadata.msgs[id]->name);
694 vapi_msg_is_with_context (vapi_msg_id_t id)
696 assert (id <= __vapi_metadata.count);
697 return __vapi_metadata.msgs[id]->has_context;
701 vapi_dispatch_one (vapi_ctx_t ctx)
703 VAPI_DBG ("vapi_dispatch_one()");
706 vapi_error_e rv = vapi_recv (ctx, &msg, &size, SVM_Q_WAIT, 0);
709 VAPI_DBG ("vapi_recv failed with rv=%d", rv);
712 u16 vpp_id = be16toh (*(u16 *) msg);
713 if (vpp_id > ctx->vl_msg_id_max)
715 VAPI_ERR ("Unknown msg ID received, id `%u', out of range <0,%u>",
716 (unsigned) vpp_id, (unsigned) ctx->vl_msg_id_max);
717 vapi_msg_free (ctx, msg);
720 if (INVALID_MSG_ID == (unsigned) ctx->vl_msg_id_to_vapi_msg_t[vpp_id])
722 VAPI_ERR ("Unknown msg ID received, id `%u' marked as not supported",
724 vapi_msg_free (ctx, msg);
727 const vapi_msg_id_t id = ctx->vl_msg_id_to_vapi_msg_t[vpp_id];
728 const size_t expect_size = vapi_get_message_size (id);
729 if (size < expect_size)
732 ("Invalid msg received, unexpected size `%zu' < expected min `%zu'",
734 vapi_msg_free (ctx, msg);
738 vapi_get_swap_to_host_func (id) (msg);
739 if (vapi_msg_is_with_context (id))
741 context = *(u32 *) (((u8 *) msg) + vapi_get_context_offset (id));
742 /* is this a message originating from VAPI? */
743 VAPI_DBG ("dispatch, context is %x", context);
744 if (context & context_counter_mask)
746 rv = vapi_dispatch_response (ctx, id, context, msg);
750 rv = vapi_dispatch_event (ctx, id, msg);
753 vapi_msg_free (ctx, msg);
758 vapi_dispatch (vapi_ctx_t ctx)
760 vapi_error_e rv = VAPI_OK;
761 while (!vapi_requests_empty (ctx))
763 rv = vapi_dispatch_one (ctx);
773 vapi_set_event_cb (vapi_ctx_t ctx, vapi_msg_id_t id,
774 vapi_event_cb callback, void *callback_ctx)
776 vapi_event_cb_with_ctx *c = &ctx->event_cbs[id];
778 c->ctx = callback_ctx;
782 vapi_clear_event_cb (vapi_ctx_t ctx, vapi_msg_id_t id)
784 vapi_set_event_cb (ctx, id, NULL, NULL);
788 vapi_set_generic_event_cb (vapi_ctx_t ctx, vapi_generic_event_cb callback,
791 ctx->generic_cb.cb = callback;
792 ctx->generic_cb.ctx = callback_ctx;
796 vapi_clear_generic_event_cb (vapi_ctx_t ctx)
798 ctx->generic_cb.cb = NULL;
799 ctx->generic_cb.ctx = NULL;
803 vapi_lookup_vl_msg_id (vapi_ctx_t ctx, vapi_msg_id_t id)
805 assert (id < __vapi_metadata.count);
806 return ctx->vapi_msg_id_t_to_vl_msg_id[id];
810 vapi_get_client_index (vapi_ctx_t ctx)
812 return api_main.my_client_index;
816 vapi_is_nonblocking (vapi_ctx_t ctx)
818 return (VAPI_MODE_NONBLOCKING == ctx->mode);
822 vapi_get_max_request_count (vapi_ctx_t ctx)
824 return ctx->requests_size - 1;
828 vapi_get_payload_offset (vapi_msg_id_t id)
830 assert (id < __vapi_metadata.count);
831 return __vapi_metadata.msgs[id]->payload_offset;
834 void (*vapi_get_swap_to_host_func (vapi_msg_id_t id)) (void *msg)
836 assert (id < __vapi_metadata.count);
837 return __vapi_metadata.msgs[id]->swap_to_host;
840 void (*vapi_get_swap_to_be_func (vapi_msg_id_t id)) (void *msg)
842 assert (id < __vapi_metadata.count);
843 return __vapi_metadata.msgs[id]->swap_to_be;
847 vapi_get_message_size (vapi_msg_id_t id)
849 assert (id < __vapi_metadata.count);
850 return __vapi_metadata.msgs[id]->size;
854 vapi_get_context_offset (vapi_msg_id_t id)
856 assert (id < __vapi_metadata.count);
857 return __vapi_metadata.msgs[id]->context_offset;
861 vapi_register_msg (vapi_message_desc_t * msg)
864 for (i = 0; i < __vapi_metadata.count; ++i)
867 (msg->name_with_crc, __vapi_metadata.msgs[i]->name_with_crc))
869 /* this happens if somebody is linking together several objects while
870 * using the static inline headers, just fill in the already
871 * assigned id here so that all the objects are in sync */
872 msg->id = __vapi_metadata.msgs[i]->id;
876 vapi_msg_id_t id = __vapi_metadata.count;
877 ++__vapi_metadata.count;
878 __vapi_metadata.msgs =
879 realloc (__vapi_metadata.msgs,
880 sizeof (*__vapi_metadata.msgs) * __vapi_metadata.count);
881 __vapi_metadata.msgs[id] = msg;
882 size_t s = strlen (msg->name_with_crc);
883 if (s > __vapi_metadata.max_len_name_with_crc)
885 __vapi_metadata.max_len_name_with_crc = s;
892 vapi_producer_lock (vapi_ctx_t ctx)
895 if (0 != (mrv = pthread_mutex_lock (&ctx->requests_mutex)))
897 VAPI_DBG ("pthread_mutex_lock() failed, rv=%d:%s", mrv, strerror (mrv));
898 (void) mrv; /* avoid warning if the above debug is not enabled */
899 return VAPI_MUTEX_FAILURE;
905 vapi_producer_unlock (vapi_ctx_t ctx)
908 if (0 != (mrv = pthread_mutex_unlock (&ctx->requests_mutex)))
910 VAPI_DBG ("pthread_mutex_unlock() failed, rv=%d:%s", mrv,
912 (void) mrv; /* avoid warning if the above debug is not enabled */
913 return VAPI_MUTEX_FAILURE;
919 vapi_get_message_count ()
921 return __vapi_metadata.count;
925 vapi_get_msg_name (vapi_msg_id_t id)
927 return __vapi_metadata.msgs[id]->name;
931 * fd.io coding-style-patch-verification: ON
934 * eval: (c-set-style "gnu")