/* * Copyright (c) 2015-2019 Cisco and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include <vnet/vnet.h> #include <vlibmemory/api.h> #include <vnet/session/application.h> #include <vnet/session/application_interface.h> #include <vnet/session/application_local.h> #include <vnet/session/session_rules_table.h> #include <vnet/session/session_table.h> #include <vnet/session/session.h> #include <vnet/vnet_msg_enum.h> #define vl_typedefs /* define message structures */ #include <vnet/vnet_all_api_h.h> #undef vl_typedefs #define vl_endianfun /* define message structures */ #include <vnet/vnet_all_api_h.h> #undef vl_endianfun /* instantiate all the print functions we know about */ #define vl_print(handle, ...) vlib_cli_output (handle, __VA_ARGS__) #define vl_printfun #include <vnet/vnet_all_api_h.h> #undef vl_printfun #include <vlibapi/api_helper_macros.h> #define foreach_session_api_msg \ _(MAP_ANOTHER_SEGMENT_REPLY, map_another_segment_reply) \ _(APPLICATION_ATTACH, application_attach) \ _(APPLICATION_DETACH, application_detach) \ _(BIND_URI, bind_uri) \ _(UNBIND_URI, unbind_uri) \ _(CONNECT_URI, connect_uri) \ _(DISCONNECT_SESSION, disconnect_session) \ _(DISCONNECT_SESSION_REPLY, disconnect_session_reply) \ _(BIND_SOCK, bind_sock) \ _(UNBIND_SOCK, unbind_sock) \ _(CONNECT_SOCK, connect_sock) \ _(SESSION_ENABLE_DISABLE, session_enable_disable) \ _(APP_NAMESPACE_ADD_DEL, app_namespace_add_del) \ _(SESSION_RULE_ADD_DEL, session_rule_add_del) \ _(SESSION_RULES_DUMP, session_rules_dump) \ _(APPLICATION_TLS_CERT_ADD, application_tls_cert_add) \ _(APPLICATION_TLS_KEY_ADD, application_tls_key_add) \ _(APP_WORKER_ADD_DEL, app_worker_add_del) \ static int session_send_fds (vl_api_registration_t * reg, int fds[], int n_fds) { clib_error_t *error; if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI) { clib_warning ("can't send memfd fd"); return -1; } error = vl_api_send_fd_msg (reg, fds, n_fds); if (error) { clib_error_report (error); return -1; } return 0; } static int send_add_segment_callback (u32 api_client_index, u64 segment_handle) { int fds[SESSION_N_FD_TYPE], n_fds = 0; vl_api_map_another_segment_t *mp; svm_fifo_segment_private_t *fs; vl_api_registration_t *reg; ssvm_private_t *sp; u8 fd_flags = 0; reg = vl_mem_api_client_index_to_registration (api_client_index); if (!reg) { clib_warning ("no api registration for client: %u", api_client_index); return -1; } fs = segment_manager_get_segment_w_handle (segment_handle); sp = &fs->ssvm; if (ssvm_type (sp) == SSVM_SEGMENT_MEMFD) { if (vl_api_registration_file_index (reg) == VL_API_INVALID_FI) { clib_warning ("can't send memfd fd"); return -1; } fd_flags |= SESSION_FD_F_MEMFD_SEGMENT; fds[n_fds] = sp->fd; n_fds += 1; } mp = vl_mem_api_alloc_as_if_client_w_reg (reg, sizeof (*mp)); clib_memset (mp, 0, sizeof (*mp)); mp->_vl_msg_id = clib_host_to_net_u16 (VL_API_MAP_ANOTHER_SEGMENT); mp->segment_size = sp->ssvm_size; mp->fd_flags = fd_flags; mp->segment_handle = clib_host_to_net_u64 (segment_handle); strncpy ((char *) mp->segment_name, (char *) sp->name, sizeof (mp->segment_name) - 1); vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); if (n_fds) return session_send_fds (reg, fds, n_fds); return 0; } static int send_del_segment_callback (u32 api_client_index, u64 segment_handle) { vl_api_unmap_segment_t *mp; vl_api_registration_t *reg; reg = vl_mem_api_client_index_to_registration (api_client_index); if (!reg) { clib_warning ("no registration: %u", api_client_index); return -1; } mp = vl_mem_api_alloc_as_if_client_w_reg (reg, sizeof (*mp)); clib_memset (mp, 0, sizeof (*mp)); mp->_vl_msg_id = clib_host_to_net_u16 (VL_API_UNMAP_SEGMENT); mp->segment_handle = clib_host_to_net_u64 (segment_handle); vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); return 0; } static int mq_try_lock_and_alloc_msg (svm_msg_q_t * app_mq, svm_msg_q_msg_t * msg) { int rv; u8 try = 0; while (try < 100) { rv = svm_msg_q_lock_and_alloc_msg_w_ring (app_mq, SESSION_MQ_CTRL_EVT_RING, SVM_Q_NOWAIT, msg); if (!rv) return 0; try++; usleep (1); } clib_warning ("failed to alloc msg"); return -1; } static int mq_send_session_accepted_cb (session_t * s) { app_worker_t *app_wrk = app_worker_get (s->app_wrk_index); svm_msg_q_msg_t _msg, *msg = &_msg; svm_msg_q_t *vpp_queue, *app_mq; transport_connection_t *tc; session_t *listener; session_accepted_msg_t *mp; session_event_t *evt; application_t *app; app_listener_t *al; app = application_get (app_wrk->app_index); app_mq = app_wrk->event_queue; if (mq_try_lock_and_alloc_msg (app_mq, msg)) return -1; evt = svm_msg_q_msg_data (app_mq, msg); clib_memset (evt, 0, sizeof (*evt)); evt->event_type = SESSION_CTRL_EVT_ACCEPTED; mp = (session_accepted_msg_t *) evt->data; clib_memset (mp, 0, sizeof (*mp)); mp->context = app->app_index; mp->server_rx_fifo = pointer_to_uword (s->rx_fifo); mp->server_tx_fifo = pointer_to_uword (s->tx_fifo); mp->segment_handle = session_segment_handle (s); if (session_has_transport (s)) { listener = listen_session_get (s->listener_index); al = app_listener_get (app, listener->al_index); mp->listener_handle = app_listener_handle (al); if (application_is_proxy (app)) { listener = app_worker_first_listener (app_wrk, session_get_fib_proto (s), session_get_transport_proto (s)); if (listener) mp->listener_handle = listen_session_get_handle (listener); } vpp_queue = session_main_get_vpp_event_queue (s->thread_index); mp->vpp_event_queue_address = pointer_to_uword (vpp_queue); mp->handle = session_handle (s); tc = transport_get_connection (session_get_transport_proto (s), s->connection_index, s->thread_index); mp->port = tc->rmt_port; mp->is_ip4 = tc->is_ip4; clib_memcpy_fast (&mp->ip, &tc->rmt_ip, sizeof (tc->rmt_ip)); } else { ct_connection_t *ct; ct = (ct_connection_t *) session_get_transport (s); listener = listen_session_get (s->listener_index); al = app_listener_get (app, listener->al_index); mp->listener_handle = app_listener_handle (al); mp->is_ip4 = session_type_is_ip4 (listener->session_type); mp->handle = session_handle (s); mp->port = ct->c_rmt_port; vpp_queue = session_main_get_vpp_event_queue (0); mp->vpp_event_queue_address = pointer_to_uword (vpp_queue); } svm_msg_q_add_and_unlock (app_mq, msg); return 0; } static inline void mq_send_session_close_evt (app_worker_t * app_wrk, session_handle_t sh, session_evt_type_t evt_type) { svm_msg_q_msg_t _msg, *msg = &_msg; session_disconnected_msg_t *mp; svm_msg_q_t *app_mq; session_event_t *evt; app_mq = app_wrk->event_queue; if (mq_try_lock_and_alloc_msg (app_mq, msg)) return; evt = svm_msg_q_msg_data (app_mq, msg); clib_memset (evt, 0, sizeof (*evt)); evt->event_type = evt_type; mp = (session_disconnected_msg_t *) evt->data; mp->handle = sh; mp->context = app_wrk->api_client_index; svm_msg_q_add_and_unlock (app_mq, msg); } static inline void mq_notify_close_subscribers (u32 app_index, session_handle_t sh, svm_fifo_t * f, session_evt_type_t evt_type) { app_worker_t *app_wrk; application_t *app; int i; app = application_get (app_index); if (!app) return; for (i = 0; i < f->n_subscribers; i++) { if (!(app_wrk = application_get_worker (app, f->subscribers[i]))) continue; mq_send_session_close_evt (app_wrk, sh, SESSION_CTRL_EVT_DISCONNECTED); } } static void mq_send_session_disconnected_cb (session_t * s) { app_worker_t *app_wrk = app_worker_get (s->app_wrk_index); session_handle_t sh = session_handle (s); mq_send_session_close_evt (app_wrk, session_handle (s), SESSION_CTRL_EVT_DISCONNECTED); if (svm_fifo_n_subscribers (s->rx_fifo)) mq_notify_close_subscribers (app_wrk->app_index, sh, s->rx_fifo, SESSION_CTRL_EVT_DISCONNECTED); } static void mq_send_session_reset_cb (session_t * s) { app_worker_t *app_wrk = app_worker_get (s->app_wrk_index); session_handle_t sh = session_handle (s); mq_send_session_close_evt (app_wrk, sh, SESSION_CTRL_EVT_RESET); if (svm_fifo_n_subscribers (s->rx_fifo)) mq_notify_close_subscribers (app_wrk->app_index, sh, s->rx_fifo, SESSION_CTRL_EVT_RESET); } static int mq_send_session_connected_cb (u32 app_wrk_index, u32 api_context, session_t * s, u8 is_fail) { svm_msg_q_msg_t _msg, *msg = &_msg; session_connected_msg_t *mp; svm_msg_q_t *vpp_mq, *app_mq; transport_connection_t *tc; app_worker_t *app_wrk; session_event_t *evt; app_wrk = app_worker_get (app_wrk_index); app_mq = app_wrk->event_queue; if (!app_mq) { clib_warning ("app %u with api index: %u not attached", app_wrk->app_index, app_wrk->api_client_index); return -1; } if (mq_try_lock_and_alloc_msg (app_mq, msg)) return -1; evt = svm_msg_q_msg_data (app_mq, msg); clib_memset (evt, 0, sizeof (*evt)); evt->event_type = SESSION_CTRL_EVT_CONNECTED; mp = (session_connected_msg_t *) evt->data; clib_memset (mp, 0, sizeof (*mp)); mp->context = api_context; if (is_fail) goto done; if (session_has_transport (s)) { tc = session_get_transport (s); if (!tc) { is_fail = 1; goto done; } vpp_mq = session_main_get_vpp_event_queue (s->thread_index); mp->handle = session_handle (s); mp->vpp_event_queue_address = pointer_to_uword (vpp_mq); clib_memcpy_fast (mp->lcl_ip, &tc->lcl_ip, sizeof (tc->lcl_ip)); mp->is_ip4 = tc->is_ip4; mp->lcl_port = tc->lcl_port; mp->server_rx_fifo = pointer_to_uword (s->rx_fifo); mp->server_tx_fifo = pointer_to_uword (s->tx_fifo); mp->segment_handle = session_segment_handle (s); } else { ct_connection_t *cct; session_t *ss; cct = (ct_connection_t *) session_get_transport (s); mp->handle = session_handle (s); mp->lcl_port = cct->c_lcl_port; mp->is_ip4 = cct->c_is_ip4; vpp_mq = session_main_get_vpp_event_queue (0); mp->vpp_event_queue_address = pointer_to_uword (vpp_mq); mp->server_rx_fifo = pointer_to_uword (s->rx_fifo); mp->server_tx_fifo = pointer_to_uword (s->tx_fifo); mp->segment_handle = session_segment_handle (s); ss = ct_session_get_peer (s); mp->ct_rx_fifo = pointer_to_uword (ss->tx_fifo); mp->ct_tx_fifo = pointer_to_uword (ss->rx_fifo); mp->ct_segment_handle = session_segment_handle (ss); } done: mp->retval = is_fail ? clib_host_to_net_u32 (VNET_API_ERROR_SESSION_CONNECT) : 0; svm_msg_q_add_and_unlock (app_mq, msg); return 0; } static int mq_send_session_bound_cb (u32 app_wrk_index, u32 api_context, session_handle_t handle, int rv) { svm_msg_q_msg_t _msg, *msg = &_msg; svm_msg_q_t *app_mq, *vpp_evt_q; transport_connection_t *tc; session_bound_msg_t *mp; app_worker_t *app_wrk; session_event_t *evt; app_listener_t *al; session_t *ls = 0; app_wrk = app_worker_get (app_wrk_index); app_mq = app_wrk->event_queue; if (!app_mq) { clib_warning ("app %u with api index: %u not attached", app_wrk->app_index, app_wrk->api_client_index); return -1; } if (mq_try_lock_and_alloc_msg (app_mq, msg)) return -1; evt = svm_msg_q_msg_data (app_mq, msg); clib_memset (evt, 0, sizeof (*evt)); evt->event_type = SESSION_CTRL_EVT_BOUND; mp = (session_bound_msg_t *) evt->data; mp->context = api_context; if (rv) goto done; mp->handle = handle; al = app_listener_get_w_handle (handle); if (al->session_index != SESSION_INVALID_INDEX) ls = app_listener_get_session (al); else ls = app_listener_get_local_session (al); tc = listen_session_get_transport (ls); mp->lcl_port = tc->lcl_port; mp->lcl_is_ip4 = tc->is_ip4; clib_memcpy_fast (mp->lcl_ip, &tc->lcl_ip, sizeof (tc->lcl_ip)); vpp_evt_q = session_main_get_vpp_event_queue (0); mp->vpp_evt_q = pointer_to_uword (vpp_evt_q); if (session_transport_service_type (ls) == TRANSPORT_SERVICE_CL) { mp->rx_fifo = pointer_to_uword (ls->rx_fifo); mp->tx_fifo = pointer_to_uword (ls->tx_fifo); } done: mp->retval = rv; svm_msg_q_add_and_unlock (app_mq, msg); return 0; } static session_cb_vft_t session_mq_cb_vft = { .session_accept_callback = mq_send_session_accepted_cb, .session_disconnect_callback = mq_send_session_disconnected_cb, .session_connected_callback = mq_send_session_connected_cb, .session_reset_callback = mq_send_session_reset_cb, .add_segment_callback = send_add_segment_callback, .del_segment_callback = send_del_segment_callback, }; static void vl_api_session_enable_disable_t_handler (vl_api_session_enable_disable_t * mp) { vl_api_session_enable_disable_reply_t *rmp; vlib_main_t *vm = vlib_get_main (); int rv = 0; vnet_session_enable_disable (vm, mp->is_enable); REPLY_MACRO (VL_API_SESSION_ENABLE_DISABLE_REPLY); } static void vl_api_application_attach_t_handler (vl_api_application_attach_t * mp) { int rv = 0, fds[SESSION_N_FD_TYPE], n_fds = 0; vl_api_application_attach_reply_t *rmp; ssvm_private_t *segp, *evt_q_segment; vnet_app_attach_args_t _a, *a = &_a; vl_api_registration_t *reg; u8 fd_flags = 0; reg = vl_api_client_index_to_registration (mp->client_index); if (!reg) return; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } STATIC_ASSERT (sizeof (u64) * APP_OPTIONS_N_OPTIONS <= sizeof (mp->options), "Out of options, fix api message definition"); clib_memset (a, 0, sizeof (*a)); a->api_client_index = mp->client_index; a->options = mp->options; a->session_cb_vft = &session_mq_cb_vft; if (mp->namespace_id_len > 64) { rv = VNET_API_ERROR_INVALID_VALUE; goto done; } if (mp->namespace_id_len) { vec_validate (a->namespace_id, mp->namespace_id_len - 1); clib_memcpy_fast (a->namespace_id, mp->namespace_id, mp->namespace_id_len); } if ((rv = vnet_application_attach (a))) { clib_warning ("attach returned: %d", rv); vec_free (a->namespace_id); goto done; } vec_free (a->namespace_id); /* Send event queues segment */ if ((evt_q_segment = session_main_get_evt_q_segment ())) { fd_flags |= SESSION_FD_F_VPP_MQ_SEGMENT; fds[n_fds] = evt_q_segment->fd; n_fds += 1; } /* Send fifo segment fd if needed */ if (ssvm_type (a->segment) == SSVM_SEGMENT_MEMFD) { fd_flags |= SESSION_FD_F_MEMFD_SEGMENT; fds[n_fds] = a->segment->fd; n_fds += 1; } if (a->options[APP_OPTIONS_FLAGS] & APP_OPTIONS_FLAGS_EVT_MQ_USE_EVENTFD) { fd_flags |= SESSION_FD_F_MQ_EVENTFD; fds[n_fds] = svm_msg_q_get_producer_eventfd (a->app_evt_q); n_fds += 1; } done: /* *INDENT-OFF* */ REPLY_MACRO2 (VL_API_APPLICATION_ATTACH_REPLY, ({ if (!rv) { segp = a->segment; rmp->app_index = clib_host_to_net_u32 (a->app_index); rmp->segment_name_length = 0; rmp->segment_size = segp->ssvm_size; if (vec_len (segp->name)) { memcpy (rmp->segment_name, segp->name, vec_len (segp->name)); rmp->segment_name_length = vec_len (segp->name); } rmp->app_event_queue_address = pointer_to_uword (a->app_evt_q); rmp->n_fds = n_fds; rmp->fd_flags = fd_flags; rmp->segment_handle = clib_host_to_net_u64 (a->segment_handle); } })); /* *INDENT-ON* */ if (n_fds) session_send_fds (reg, fds, n_fds); } static void vl_api_application_detach_t_handler (vl_api_application_detach_t * mp) { vl_api_application_detach_reply_t *rmp; int rv = VNET_API_ERROR_INVALID_VALUE_2; vnet_app_detach_args_t _a, *a = &_a; application_t *app; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { a->app_index = app->app_index; a->api_client_index = mp->client_index; rv = vnet_application_detach (a); } done: REPLY_MACRO (VL_API_APPLICATION_DETACH_REPLY); } static void vl_api_bind_uri_t_handler (vl_api_bind_uri_t * mp) { vl_api_bind_uri_reply_t *rmp; vnet_listen_args_t _a, *a = &_a; application_t *app = 0; app_worker_t *app_wrk; int rv; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { clib_memset (a, 0, sizeof (*a)); a->uri = (char *) mp->uri; a->app_index = app->app_index; rv = vnet_bind_uri (a); } else { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; } done: REPLY_MACRO (VL_API_BIND_URI_REPLY); if (app) { app_wrk = application_get_worker (app, 0); mq_send_session_bound_cb (app_wrk->wrk_index, mp->context, a->handle, rv); } } static void vl_api_unbind_uri_t_handler (vl_api_unbind_uri_t * mp) { vl_api_unbind_uri_reply_t *rmp; application_t *app; vnet_unlisten_args_t _a, *a = &_a; int rv; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { a->uri = (char *) mp->uri; a->app_index = app->app_index; rv = vnet_unbind_uri (a); } else { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; } done: REPLY_MACRO (VL_API_UNBIND_URI_REPLY); } static void vl_api_connect_uri_t_handler (vl_api_connect_uri_t * mp) { vl_api_connect_uri_reply_t *rmp; vnet_connect_args_t _a, *a = &_a; application_t *app; int rv = 0; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { clib_memset (a, 0, sizeof (*a)); a->uri = (char *) mp->uri; a->api_context = mp->context; a->app_index = app->app_index; if ((rv = vnet_connect_uri (a))) clib_warning ("connect_uri returned: %d", rv); } else { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; } /* * Don't reply to stream (tcp) connects. The reply will come once * the connection is established. In case of the redirects, the reply * will come from the server app. */ if (rv == 0) return; done: REPLY_MACRO (VL_API_CONNECT_URI_REPLY); } static void vl_api_disconnect_session_t_handler (vl_api_disconnect_session_t * mp) { vl_api_disconnect_session_reply_t *rmp; vnet_disconnect_args_t _a, *a = &_a; application_t *app; int rv = 0; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { a->handle = mp->handle; a->app_index = app->app_index; rv = vnet_disconnect_session (a); } else { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; } done: REPLY_MACRO2 (VL_API_DISCONNECT_SESSION_REPLY, rmp->handle = mp->handle); } static void vl_api_disconnect_session_reply_t_handler (vl_api_disconnect_session_reply_t * mp) { vnet_disconnect_args_t _a, *a = &_a; application_t *app; /* Client objected to disconnecting the session, log and continue */ if (mp->retval) { clib_warning ("client retval %d", mp->retval); return; } /* Disconnect has been confirmed. Confirm close to transport */ app = application_lookup (mp->context); if (app) { a->handle = mp->handle; a->app_index = app->app_index; vnet_disconnect_session (a); } } static void vl_api_map_another_segment_reply_t_handler (vl_api_map_another_segment_reply_t * mp) { clib_warning ("not implemented"); } static void vl_api_bind_sock_t_handler (vl_api_bind_sock_t * mp) { vnet_listen_args_t _a, *a = &_a; vl_api_bind_sock_reply_t *rmp; application_t *app = 0; app_worker_t *app_wrk; ip46_address_t *ip46; int rv = 0; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (!app) { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; goto done; } ip46 = (ip46_address_t *) mp->ip; clib_memset (a, 0, sizeof (*a)); a->sep.is_ip4 = mp->is_ip4; a->sep.ip = *ip46; a->sep.port = mp->port; a->sep.fib_index = mp->vrf; a->sep.sw_if_index = ENDPOINT_INVALID_INDEX; a->sep.transport_proto = mp->proto; a->app_index = app->app_index; a->wrk_map_index = mp->wrk_index; if ((rv = vnet_listen (a))) clib_warning ("listen returned: %d", rv); done: /* Actual reply sent only over mq */ REPLY_MACRO (VL_API_BIND_SOCK_REPLY); if (app) { app_wrk = application_get_worker (app, mp->wrk_index); mq_send_session_bound_cb (app_wrk->wrk_index, mp->context, a->handle, rv); } } static void vl_api_unbind_sock_t_handler (vl_api_unbind_sock_t * mp) { vl_api_unbind_sock_reply_t *rmp; vnet_unlisten_args_t _a, *a = &_a; app_worker_t *app_wrk; application_t *app = 0; int rv = 0; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { a->app_index = app->app_index; a->handle = mp->handle; a->wrk_map_index = mp->wrk_index; if ((rv = vnet_unlisten (a))) clib_warning ("unlisten returned: %d", rv); } done: REPLY_MACRO (VL_API_UNBIND_SOCK_REPLY); /* * Send reply over msg queue */ svm_msg_q_msg_t _msg, *msg = &_msg; session_unlisten_reply_msg_t *ump; svm_msg_q_t *app_mq; session_event_t *evt; if (!app) return; app_wrk = application_get_worker (app, a->wrk_map_index); if (!app_wrk) return; app_mq = app_wrk->event_queue; if (mq_try_lock_and_alloc_msg (app_mq, msg)) return; evt = svm_msg_q_msg_data (app_mq, msg); clib_memset (evt, 0, sizeof (*evt)); evt->event_type = SESSION_CTRL_EVT_UNLISTEN_REPLY; ump = (session_unlisten_reply_msg_t *) evt->data; ump->context = mp->context; ump->handle = mp->handle; ump->retval = rv; svm_msg_q_add_and_unlock (app_mq, msg); } static void vl_api_connect_sock_t_handler (vl_api_connect_sock_t * mp) { vl_api_connect_sock_reply_t *rmp; vnet_connect_args_t _a, *a = &_a; application_t *app = 0; int rv = 0; if (session_main_is_enabled () == 0) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } app = application_lookup (mp->client_index); if (app) { svm_queue_t *client_q; ip46_address_t *ip46 = (ip46_address_t *) mp->ip; clib_memset (a, 0, sizeof (*a)); client_q = vl_api_client_index_to_input_queue (mp->client_index); mp->client_queue_address = pointer_to_uword (client_q); a->sep.is_ip4 = mp->is_ip4; a->sep.ip = *ip46; a->sep.port = mp->port; a->sep.transport_proto = mp->proto; a->sep.peer.fib_index = mp->vrf; a->sep.peer.sw_if_index = ENDPOINT_INVALID_INDEX; if (mp->hostname_len) { vec_validate (a->sep_ext.hostname, mp->hostname_len - 1); clib_memcpy_fast (a->sep_ext.hostname, mp->hostname, mp->hostname_len); } a->api_context = mp->context; a->app_index = app->app_index; a->wrk_map_index = mp->wrk_index; if ((rv = vnet_connect (a))) clib_warning ("connect returned: %u", rv); vec_free (a->sep_ext.hostname); } else { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; } if (rv == 0) return; /* Got some error, relay it */ done: REPLY_MACRO (VL_API_CONNECT_SOCK_REPLY); if (app) { app_worker_t *app_wrk = application_get_worker (app, mp->wrk_index); mq_send_session_connected_cb (app_wrk->wrk_index, mp->context, 0, 1); } } static void vl_api_app_worker_add_del_t_handler (vl_api_app_worker_add_del_t * mp) { int rv = 0, fds[SESSION_N_FD_TYPE], n_fds = 0; vl_api_app_worker_add_del_reply_t *rmp; vl_api_registration_t *reg; application_t *app; u8 fd_flags = 0; if (!session_main_is_enabled ()) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } reg = vl_api_client_index_to_registration (mp->client_index); if (!reg) return; app = application_get_if_valid (clib_net_to_host_u32 (mp->app_index)); if (!app) { rv = VNET_API_ERROR_INVALID_VALUE; goto done; } vnet_app_worker_add_del_args_t args = { .app_index = app->app_index, .wrk_map_index = clib_net_to_host_u32 (mp->wrk_index), .api_client_index = mp->client_index, .is_add = mp->is_add }; rv = vnet_app_worker_add_del (&args); if (rv) { clib_warning ("app worker add/del returned: %d", rv); goto done; } if (!mp->is_add) goto done; /* Send fifo segment fd if needed */ if (ssvm_type (args.segment) == SSVM_SEGMENT_MEMFD) { fd_flags |= SESSION_FD_F_MEMFD_SEGMENT; fds[n_fds] = args.segment->fd; n_fds += 1; } if (application_segment_manager_properties (app)->use_mq_eventfd) { fd_flags |= SESSION_FD_F_MQ_EVENTFD; fds[n_fds] = svm_msg_q_get_producer_eventfd (args.evt_q); n_fds += 1; } /* *INDENT-OFF* */ done: REPLY_MACRO2 (VL_API_APP_WORKER_ADD_DEL_REPLY, ({ rmp->is_add = mp->is_add; rmp->wrk_index = clib_host_to_net_u32 (args.wrk_map_index); rmp->segment_handle = clib_host_to_net_u64 (args.segment_handle); if (!rv && mp->is_add) { if (vec_len (args.segment->name)) { memcpy (rmp->segment_name, args.segment->name, vec_len (args.segment->name)); rmp->segment_name_length = vec_len (args.segment->name); } rmp->app_event_queue_address = pointer_to_uword (args.evt_q); rmp->n_fds = n_fds; rmp->fd_flags = fd_flags; } })); /* *INDENT-ON* */ if (n_fds) session_send_fds (reg, fds, n_fds); } static void vl_api_app_namespace_add_del_t_handler (vl_api_app_namespace_add_del_t * mp) { vl_api_app_namespace_add_del_reply_t *rmp; u32 appns_index = 0; u8 *ns_id = 0; int rv = 0; if (!session_main_is_enabled ()) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } if (mp->namespace_id_len > ARRAY_LEN (mp->namespace_id)) { rv = VNET_API_ERROR_INVALID_VALUE; goto done; } vec_validate (ns_id, mp->namespace_id_len - 1); clib_memcpy_fast (ns_id, mp->namespace_id, mp->namespace_id_len); vnet_app_namespace_add_del_args_t args = { .ns_id = ns_id, .secret = clib_net_to_host_u64 (mp->secret), .sw_if_index = clib_net_to_host_u32 (mp->sw_if_index), .ip4_fib_id = clib_net_to_host_u32 (mp->ip4_fib_id), .ip6_fib_id = clib_net_to_host_u32 (mp->ip6_fib_id), .is_add = 1 }; rv = vnet_app_namespace_add_del (&args); if (!rv) { appns_index = app_namespace_index_from_id (ns_id); if (appns_index == APP_NAMESPACE_INVALID_INDEX) { clib_warning ("app ns lookup failed"); rv = VNET_API_ERROR_UNSPECIFIED; } } vec_free (ns_id); /* *INDENT-OFF* */ done: REPLY_MACRO2 (VL_API_APP_NAMESPACE_ADD_DEL_REPLY, ({ if (!rv) rmp->appns_index = clib_host_to_net_u32 (appns_index); })); /* *INDENT-ON* */ } static void vl_api_session_rule_add_del_t_handler (vl_api_session_rule_add_del_t * mp) { vl_api_session_rule_add_del_reply_t *rmp; session_rule_add_del_args_t args; session_rule_table_add_del_args_t *table_args = &args.table_args; u8 fib_proto; int rv = 0; clib_memset (&args, 0, sizeof (args)); fib_proto = mp->is_ip4 ? FIB_PROTOCOL_IP4 : FIB_PROTOCOL_IP6; table_args->lcl.fp_len = mp->lcl_plen; table_args->lcl.fp_proto = fib_proto; table_args->rmt.fp_len = mp->rmt_plen; table_args->rmt.fp_proto = fib_proto; table_args->lcl_port = mp->lcl_port; table_args->rmt_port = mp->rmt_port; table_args->action_index = clib_net_to_host_u32 (mp->action_index); table_args->is_add = mp->is_add; mp->tag[sizeof (mp->tag) - 1] = 0; table_args->tag = format (0, "%s", mp->tag); args.appns_index = clib_net_to_host_u32 (mp->appns_index); args.scope = mp->scope; args.transport_proto = mp->transport_proto; clib_memset (&table_args->lcl.fp_addr, 0, sizeof (table_args->lcl.fp_addr)); clib_memset (&table_args->rmt.fp_addr, 0, sizeof (table_args->rmt.fp_addr)); ip_set (&table_args->lcl.fp_addr, mp->lcl_ip, mp->is_ip4); ip_set (&table_args->rmt.fp_addr, mp->rmt_ip, mp->is_ip4); rv = vnet_session_rule_add_del (&args); if (rv) clib_warning ("rule add del returned: %d", rv); vec_free (table_args->tag); REPLY_MACRO (VL_API_SESSION_RULE_ADD_DEL_REPLY); } static void send_session_rule_details4 (mma_rule_16_t * rule, u8 is_local, u8 transport_proto, u32 appns_index, u8 * tag, vl_api_registration_t * reg, u32 context) { vl_api_session_rules_details_t *rmp = 0; session_mask_or_match_4_t *match = (session_mask_or_match_4_t *) & rule->match; session_mask_or_match_4_t *mask = (session_mask_or_match_4_t *) & rule->mask; rmp = vl_msg_api_alloc (sizeof (*rmp)); clib_memset (rmp, 0, sizeof (*rmp)); rmp->_vl_msg_id = ntohs (VL_API_SESSION_RULES_DETAILS); rmp->context = context; rmp->is_ip4 = 1; clib_memcpy_fast (rmp->lcl_ip, &match->lcl_ip, sizeof (match->lcl_ip)); clib_memcpy_fast (rmp->rmt_ip, &match->rmt_ip, sizeof (match->rmt_ip)); rmp->lcl_plen = ip4_mask_to_preflen (&mask->lcl_ip); rmp->rmt_plen = ip4_mask_to_preflen (&mask->rmt_ip); rmp->lcl_port = match->lcl_port; rmp->rmt_port = match->rmt_port; rmp->action_index = clib_host_to_net_u32 (rule->action_index); rmp->scope = is_local ? SESSION_RULE_SCOPE_LOCAL : SESSION_RULE_SCOPE_GLOBAL; rmp->transport_proto = transport_proto; rmp->appns_index = clib_host_to_net_u32 (appns_index); if (tag) { clib_memcpy_fast (rmp->tag, tag, vec_len (tag)); rmp->tag[vec_len (tag)] = 0; } vl_api_send_msg (reg, (u8 *) rmp); } static void send_session_rule_details6 (mma_rule_40_t * rule, u8 is_local, u8 transport_proto, u32 appns_index, u8 * tag, vl_api_registration_t * reg, u32 context) { vl_api_session_rules_details_t *rmp = 0; session_mask_or_match_6_t *match = (session_mask_or_match_6_t *) & rule->match; session_mask_or_match_6_t *mask = (session_mask_or_match_6_t *) & rule->mask; rmp = vl_msg_api_alloc (sizeof (*rmp)); clib_memset (rmp, 0, sizeof (*rmp)); rmp->_vl_msg_id = ntohs (VL_API_SESSION_RULES_DETAILS); rmp->context = context; rmp->is_ip4 = 0; clib_memcpy_fast (rmp->lcl_ip, &match->lcl_ip, sizeof (match->lcl_ip)); clib_memcpy_fast (rmp->rmt_ip, &match->rmt_ip, sizeof (match->rmt_ip)); rmp->lcl_plen = ip6_mask_to_preflen (&mask->lcl_ip); rmp->rmt_plen = ip6_mask_to_preflen (&mask->rmt_ip); rmp->lcl_port = match->lcl_port; rmp->rmt_port = match->rmt_port; rmp->action_index = clib_host_to_net_u32 (rule->action_index); rmp->scope = is_local ? SESSION_RULE_SCOPE_LOCAL : SESSION_RULE_SCOPE_GLOBAL; rmp->transport_proto = transport_proto; rmp->appns_index = clib_host_to_net_u32 (appns_index); if (tag) { clib_memcpy_fast (rmp->tag, tag, vec_len (tag)); rmp->tag[vec_len (tag)] = 0; } vl_api_send_msg (reg, (u8 *) rmp); } static void send_session_rules_table_details (session_rules_table_t * srt, u8 fib_proto, u8 tp, u8 is_local, u32 appns_index, vl_api_registration_t * reg, u32 context) { mma_rule_16_t *rule16; mma_rule_40_t *rule40; mma_rules_table_16_t *srt16; mma_rules_table_40_t *srt40; u32 ri; if (is_local || fib_proto == FIB_PROTOCOL_IP4) { u8 *tag = 0; /* *INDENT-OFF* */ srt16 = &srt->session_rules_tables_16; pool_foreach (rule16, srt16->rules, ({ ri = mma_rules_table_rule_index_16 (srt16, rule16); tag = session_rules_table_rule_tag (srt, ri, 1); send_session_rule_details4 (rule16, is_local, tp, appns_index, tag, reg, context); })); /* *INDENT-ON* */ } if (is_local || fib_proto == FIB_PROTOCOL_IP6) { u8 *tag = 0; /* *INDENT-OFF* */ srt40 = &srt->session_rules_tables_40; pool_foreach (rule40, srt40->rules, ({ ri = mma_rules_table_rule_index_40 (srt40, rule40); tag = session_rules_table_rule_tag (srt, ri, 1); send_session_rule_details6 (rule40, is_local, tp, appns_index, tag, reg, context); })); /* *INDENT-ON* */ } } static void vl_api_session_rules_dump_t_handler (vl_api_one_map_server_dump_t * mp) { vl_api_registration_t *reg; session_table_t *st; u8 tp; reg = vl_api_client_index_to_registration (mp->client_index); if (!reg) return; /* *INDENT-OFF* */ session_table_foreach (st, ({ for (tp = 0; tp < TRANSPORT_N_PROTO; tp++) { send_session_rules_table_details (&st->session_rules[tp], st->active_fib_proto, tp, st->is_local, st->appns_index, reg, mp->context); } })); /* *INDENT-ON* */ } static void vl_api_application_tls_cert_add_t_handler (vl_api_application_tls_cert_add_t * mp) { vl_api_app_namespace_add_del_reply_t *rmp; vnet_app_add_tls_cert_args_t _a, *a = &_a; clib_error_t *error; application_t *app; u32 cert_len; int rv = 0; if (!session_main_is_enabled ()) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } if (!(app = application_lookup (mp->client_index))) { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; goto done; } clib_memset (a, 0, sizeof (*a)); a->app_index = app->app_index; cert_len = clib_net_to_host_u16 (mp->cert_len); if (cert_len > 10000) { rv = VNET_API_ERROR_INVALID_VALUE; goto done; } vec_validate (a->cert, cert_len); clib_memcpy_fast (a->cert, mp->cert, cert_len); if ((error = vnet_app_add_tls_cert (a))) { rv = clib_error_get_code (error); clib_error_report (error); } vec_free (a->cert); done: REPLY_MACRO (VL_API_APPLICATION_TLS_CERT_ADD_REPLY); } static void vl_api_application_tls_key_add_t_handler (vl_api_application_tls_key_add_t * mp) { vl_api_app_namespace_add_del_reply_t *rmp; vnet_app_add_tls_key_args_t _a, *a = &_a; clib_error_t *error; application_t *app; u32 key_len; int rv = 0; if (!session_main_is_enabled ()) { rv = VNET_API_ERROR_FEATURE_DISABLED; goto done; } if (!(app = application_lookup (mp->client_index))) { rv = VNET_API_ERROR_APPLICATION_NOT_ATTACHED; goto done; } clib_memset (a, 0, sizeof (*a)); a->app_index = app->app_index; key_len = clib_net_to_host_u16 (mp->key_len); if (key_len > 10000) { rv = VNET_API_ERROR_INVALID_VALUE; goto done; } vec_validate (a->key, key_len); clib_memcpy_fast (a->key, mp->key, key_len); if ((error = vnet_app_add_tls_key (a))) { rv = clib_error_get_code (error); clib_error_report (error); } vec_free (a->key); done: REPLY_MACRO (VL_API_APPLICATION_TLS_KEY_ADD_REPLY); } static clib_error_t * application_reaper_cb (u32 client_index) { application_t *app = application_lookup (client_index); vnet_app_detach_args_t _a, *a = &_a; if (app) { a->app_index = app->app_index; a->api_client_index = client_index; vnet_application_detach (a); } return 0; } VL_MSG_API_REAPER_FUNCTION (application_reaper_cb); #define vl_msg_name_crc_list #include <vnet/vnet_all_api_h.h> #undef vl_msg_name_crc_list static void setup_message_id_table (api_main_t * am) { #define _(id,n,crc) vl_msg_api_add_msg_name_crc (am, #n "_" #crc, id); foreach_vl_msg_name_crc_session; #undef _ } /* * session_api_hookup * Add uri's API message handlers to the table. * vlib has alread mapped shared memory and * added the client registration handlers. * See .../open-repo/vlib/memclnt_vlib.c:memclnt_process() */ static clib_error_t * session_api_hookup (vlib_main_t * vm) { api_main_t *am = &api_main; #define _(N,n) \ vl_msg_api_set_handlers(VL_API_##N, #n, \ vl_api_##n##_t_handler, \ vl_noop_handler, \ vl_api_##n##_t_endian, \ vl_api_##n##_t_print, \ sizeof(vl_api_##n##_t), 1); foreach_session_api_msg; #undef _ /* * Messages which bounce off the data-plane to * an API client. Simply tells the message handling infra not * to free the message. * * Bounced message handlers MUST NOT block the data plane */ am->message_bounce[VL_API_CONNECT_URI] = 1; am->message_bounce[VL_API_CONNECT_SOCK] = 1; /* * Set up the (msg_name, crc, message-id) table */ setup_message_id_table (am); return 0; } VLIB_API_INIT_FUNCTION (session_api_hookup); /* * fd.io coding-style-patch-verification: ON * * Local Variables: * eval: (c-set-style "gnu") * End: */