| /* |
| * 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/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) \ |
| _(ACCEPT_SESSION_REPLY, accept_session_reply) \ |
| _(RESET_SESSION_REPLY, reset_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 |
| send_app_cut_through_registration_add (u32 api_client_index, |
| u32 wrk_map_index, u64 mq_addr, |
| u64 peer_mq_addr) |
| { |
| vl_api_app_cut_through_registration_add_t *mp; |
| vl_api_registration_t *reg; |
| svm_msg_q_t *mq, *peer_mq; |
| int fds[2]; |
| |
| 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_APP_CUT_THROUGH_REGISTRATION_ADD); |
| |
| mp->evt_q_address = mq_addr; |
| mp->peer_evt_q_address = peer_mq_addr; |
| mp->wrk_index = wrk_map_index; |
| |
| mq = uword_to_pointer (mq_addr, svm_msg_q_t *); |
| peer_mq = uword_to_pointer (peer_mq_addr, svm_msg_q_t *); |
| |
| if (svm_msg_q_get_producer_eventfd (mq) != -1) |
| { |
| mp->fd_flags |= SESSION_FD_F_MQ_EVENTFD; |
| mp->n_fds = 2; |
| /* app will overwrite exactly the fds we pass here. So |
| * when we swap mq with peer_mq (accept vs connect) the |
| * fds will still be valid */ |
| fds[0] = svm_msg_q_get_consumer_eventfd (mq); |
| fds[1] = svm_msg_q_get_producer_eventfd (peer_mq); |
| } |
| |
| vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); |
| |
| if (mp->n_fds != 0) |
| session_send_fds (reg, fds, mp->n_fds); |
| |
| return 0; |
| } |
| |
| static int |
| send_session_accept_callback (session_t * s) |
| { |
| app_worker_t *server_wrk = app_worker_get (s->app_wrk_index); |
| vl_api_accept_session_t *mp; |
| vl_api_registration_t *reg; |
| transport_connection_t *tc; |
| session_t *listener; |
| svm_msg_q_t *vpp_queue; |
| application_t *server; |
| app_listener_t *al; |
| |
| server = application_get (server_wrk->app_index); |
| reg = |
| vl_mem_api_client_index_to_registration (server_wrk->api_client_index); |
| if (!reg) |
| { |
| clib_warning ("no registration: %u", server_wrk->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_ACCEPT_SESSION); |
| mp->context = server_wrk->wrk_index; |
| mp->server_rx_fifo = pointer_to_uword (s->rx_fifo); |
| mp->server_tx_fifo = pointer_to_uword (s->tx_fifo); |
| |
| if (session_has_transport (s)) |
| { |
| listener = listen_session_get (s->listener_index); |
| al = app_listener_get (server, listener->al_index); |
| mp->listener_handle = app_listener_handle (al); |
| if (application_is_proxy (server)) |
| { |
| listener = |
| app_worker_first_listener (server_wrk, session_get_fib_proto (s), |
| session_get_transport_proto (s)); |
| if (listener) |
| mp->listener_handle = listen_session_get_handle (listener); |
| } |
| vpp_queue = session_manager_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 |
| { |
| local_session_t *ls = (local_session_t *) s; |
| local_session_t *ll; |
| if (application_local_session_listener_has_transport (ls)) |
| { |
| listener = listen_session_get (ls->listener_index); |
| al = app_listener_get (server, listener->al_index); |
| mp->listener_handle = app_listener_handle (al); |
| mp->is_ip4 = session_type_is_ip4 (listener->session_type); |
| } |
| else |
| { |
| ll = application_get_local_listen_session (server, |
| ls->listener_index); |
| al = app_listener_get (server, ll->al_index); |
| mp->listener_handle = app_listener_handle (al); |
| mp->is_ip4 = session_type_is_ip4 (ll->listener_session_type); |
| } |
| mp->handle = application_local_session_handle (ls); |
| mp->port = ls->port; |
| mp->vpp_event_queue_address = ls->client_evt_q; |
| mp->server_event_queue_address = ls->server_evt_q; |
| } |
| vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); |
| |
| return 0; |
| } |
| |
| static void |
| send_session_disconnect_callback (session_t * s) |
| { |
| app_worker_t *app_wrk = app_worker_get (s->app_wrk_index); |
| vl_api_disconnect_session_t *mp; |
| vl_api_registration_t *reg; |
| |
| reg = vl_mem_api_client_index_to_registration (app_wrk->api_client_index); |
| if (!reg) |
| { |
| clib_warning ("no registration: %u", app_wrk->api_client_index); |
| return; |
| } |
| |
| 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_DISCONNECT_SESSION); |
| mp->handle = session_handle (s); |
| mp->context = app_wrk->api_client_index; |
| vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); |
| } |
| |
| static void |
| send_session_reset_callback (session_t * s) |
| { |
| app_worker_t *app_wrk = app_worker_get (s->app_wrk_index); |
| vl_api_registration_t *reg; |
| vl_api_reset_session_t *mp; |
| |
| reg = vl_mem_api_client_index_to_registration (app_wrk->api_client_index); |
| if (!reg) |
| { |
| clib_warning ("no registration: %u", app_wrk->api_client_index); |
| return; |
| } |
| |
| 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_RESET_SESSION); |
| mp->handle = session_handle (s); |
| vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); |
| } |
| |
| int |
| send_session_connected_callback (u32 app_wrk_index, u32 api_context, |
| session_t * s, u8 is_fail) |
| { |
| vl_api_connect_session_reply_t *mp; |
| transport_connection_t *tc; |
| vl_api_registration_t *reg; |
| svm_msg_q_t *vpp_queue; |
| app_worker_t *app_wrk; |
| |
| app_wrk = app_worker_get (app_wrk_index); |
| reg = vl_mem_api_client_index_to_registration (app_wrk->api_client_index); |
| if (!reg) |
| { |
| clib_warning ("no registration: %u", app_wrk->api_client_index); |
| return -1; |
| } |
| |
| mp = vl_mem_api_alloc_as_if_client_w_reg (reg, sizeof (*mp)); |
| mp->_vl_msg_id = clib_host_to_net_u16 (VL_API_CONNECT_SESSION_REPLY); |
| 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_queue = session_manager_get_vpp_event_queue (s->thread_index); |
| mp->handle = session_handle (s); |
| mp->vpp_event_queue_address = pointer_to_uword (vpp_queue); |
| 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); |
| } |
| else |
| { |
| local_session_t *ls = (local_session_t *) s; |
| mp->handle = application_local_session_handle (ls); |
| mp->lcl_port = ls->port; |
| mp->vpp_event_queue_address = ls->server_evt_q; |
| mp->client_event_queue_address = ls->client_evt_q; |
| mp->server_rx_fifo = pointer_to_uword (s->tx_fifo); |
| mp->server_tx_fifo = pointer_to_uword (s->rx_fifo); |
| } |
| |
| done: |
| mp->retval = is_fail ? |
| clib_host_to_net_u32 (VNET_API_ERROR_SESSION_CONNECT) : 0; |
| vl_msg_api_send_shmem (reg->vl_input_queue, (u8 *) & mp); |
| return 0; |
| } |
| |
| static session_cb_vft_t session_cb_vft = { |
| .session_accept_callback = send_session_accept_callback, |
| .session_disconnect_callback = send_session_disconnect_callback, |
| .session_connected_callback = send_session_connected_callback, |
| .session_reset_callback = send_session_reset_callback, |
| .add_segment_callback = send_add_segment_callback, |
| .del_segment_callback = send_del_segment_callback, |
| }; |
| |
| 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_manager_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 |
| { |
| local_session_t *ls = (local_session_t *) s; |
| local_session_t *ll; |
| u8 main_thread = vlib_num_workers ()? 1 : 0; |
| |
| send_app_cut_through_registration_add (app_wrk->api_client_index, |
| app_wrk->wrk_map_index, |
| ls->server_evt_q, |
| ls->client_evt_q); |
| |
| if (application_local_session_listener_has_transport (ls)) |
| { |
| listener = listen_session_get (ls->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); |
| } |
| else |
| { |
| ll = application_get_local_listen_session (app, ls->listener_index); |
| al = app_listener_get (app, ll->al_index); |
| mp->listener_handle = app_listener_handle (al); |
| mp->is_ip4 = session_type_is_ip4 (ll->listener_session_type); |
| } |
| mp->handle = application_local_session_handle (ls); |
| mp->port = ls->port; |
| vpp_queue = session_manager_get_vpp_event_queue (main_thread); |
| mp->vpp_event_queue_address = pointer_to_uword (vpp_queue); |
| mp->client_event_queue_address = ls->client_evt_q; |
| mp->server_event_queue_address = ls->server_evt_q; |
| } |
| 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); |
| } |
| |
| void |
| mq_send_local_session_disconnected_cb (u32 app_wrk_index, |
| local_session_t * ls) |
| { |
| app_worker_t *app_wrk = app_worker_get (app_wrk_index); |
| session_handle_t sh = application_local_session_handle (ls); |
| |
| mq_send_session_close_evt (app_wrk, sh, SESSION_CTRL_EVT_DISCONNECTED); |
| |
| if (svm_fifo_n_subscribers (ls->rx_fifo)) |
| mq_notify_close_subscribers (app_wrk->app_index, sh, ls->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; |
| |
| mp->segment_handle = session_segment_handle (s); |
| |
| if (session_has_transport (s)) |
| { |
| tc = session_get_transport (s); |
| if (!tc) |
| { |
| is_fail = 1; |
| goto done; |
| } |
| |
| vpp_mq = session_manager_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); |
| } |
| else |
| { |
| local_session_t *ls = (local_session_t *) s; |
| u8 main_thread = vlib_num_workers ()? 1 : 0; |
| |
| send_app_cut_through_registration_add (app_wrk->api_client_index, |
| app_wrk->wrk_map_index, |
| ls->client_evt_q, |
| ls->server_evt_q); |
| |
| mp->handle = application_local_session_handle (ls); |
| mp->lcl_port = ls->port; |
| vpp_mq = session_manager_get_vpp_event_queue (main_thread); |
| mp->vpp_event_queue_address = pointer_to_uword (vpp_mq); |
| mp->client_event_queue_address = ls->client_evt_q; |
| mp->server_event_queue_address = ls->server_evt_q; |
| mp->server_rx_fifo = pointer_to_uword (s->tx_fifo); |
| mp->server_tx_fifo = pointer_to_uword (s->rx_fifo); |
| } |
| |
| 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; |
| application_t *app; |
| app_listener_t *al; |
| session_t *ls = 0; |
| |
| app_wrk = app_worker_get (app_wrk_index); |
| app = application_get (app_wrk->app_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; |
| if (application_has_global_scope (app)) |
| { |
| al = app_listener_get_w_handle (handle); |
| ls = app_listener_get_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)); |
| } |
| else |
| { |
| local_session_t *local; |
| local = application_get_local_listener_w_handle (handle); |
| mp->lcl_port = local->port; |
| mp->lcl_is_ip4 = session_type_is_ip4 (local->session_type); |
| } |
| |
| vpp_evt_q = session_manager_get_vpp_event_queue (0); |
| mp->vpp_evt_q = pointer_to_uword (vpp_evt_q); |
| |
| if (ls && 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_manager_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; |
| |
| if (a->options[APP_OPTIONS_FLAGS] & APP_OPTIONS_FLAGS_USE_MQ_FOR_CTRL_MSGS) |
| a->session_cb_vft = &session_mq_cb_vft; |
| else |
| a->session_cb_vft = &session_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_manager_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_manager_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) |
| { |
| transport_connection_t *tc = 0; |
| vnet_listen_args_t _a, *a = &_a; |
| vl_api_bind_uri_reply_t *rmp; |
| session_t *s; |
| application_t *app = 0; |
| svm_msg_q_t *vpp_evt_q; |
| app_worker_t *app_wrk; |
| int rv; |
| |
| if (session_manager_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: |
| |
| /* *INDENT-OFF* */ |
| REPLY_MACRO2 (VL_API_BIND_URI_REPLY, ({ |
| if (!rv) |
| { |
| rmp->handle = a->handle; |
| if (app && application_has_global_scope (app)) |
| { |
| s = listen_session_get_from_handle (a->handle); |
| tc = listen_session_get_transport (s); |
| rmp->lcl_is_ip4 = tc->is_ip4; |
| rmp->lcl_port = tc->lcl_port; |
| clib_memcpy_fast (rmp->lcl_ip, &tc->lcl_ip, sizeof(tc->lcl_ip)); |
| if (session_transport_service_type (s) == TRANSPORT_SERVICE_CL) |
| { |
| rmp->rx_fifo = pointer_to_uword (s->rx_fifo); |
| rmp->tx_fifo = pointer_to_uword (s->tx_fifo); |
| vpp_evt_q = session_manager_get_vpp_event_queue (0); |
| rmp->vpp_evt_q = pointer_to_uword (vpp_evt_q); |
| } |
| } |
| } |
| })); |
| /* *INDENT-ON* */ |
| |
| /* If app uses mq for control messages, send an mq message as well */ |
| if (app && application_use_mq_for_ctrl (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_manager_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_session_reply_t *rmp; |
| vnet_connect_args_t _a, *a = &_a; |
| application_t *app; |
| int rv = 0; |
| |
| if (session_manager_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: |
| /* *INDENT-OFF* */ |
| REPLY_MACRO (VL_API_CONNECT_SESSION_REPLY); |
| /* *INDENT-ON* */ |
| } |
| |
| 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_manager_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_reset_session_reply_t_handler (vl_api_reset_session_reply_t * mp) |
| { |
| vnet_disconnect_args_t _a = { 0 }, *a = &_a; |
| app_worker_t *app_wrk; |
| application_t *app; |
| session_t *s; |
| u32 index, thread_index; |
| |
| app = application_lookup (mp->context); |
| if (!app) |
| return; |
| |
| session_parse_handle (mp->handle, &index, &thread_index); |
| s = session_get_if_valid (index, thread_index); |
| if (!s) |
| { |
| clib_warning ("Invalid session!"); |
| return; |
| } |
| |
| app_wrk = app_worker_get (s->app_wrk_index); |
| if (app_wrk->app_index != app->app_index) |
| { |
| clib_warning ("app %u does not own handle 0x%lx", app->app_index, |
| mp->handle); |
| return; |
| } |
| |
| /* Client objected to resetting the session, log and continue */ |
| if (mp->retval) |
| { |
| clib_warning ("client retval %d", mp->retval); |
| return; |
| } |
| |
| /* This comes as a response to a reset, transport only waiting for |
| * confirmation to remove connection state, no need to disconnect */ |
| a->handle = mp->handle; |
| a->app_index = app->app_index; |
| vnet_disconnect_session (a); |
| } |
| |
| static void |
| vl_api_accept_session_reply_t_handler (vl_api_accept_session_reply_t * mp) |
| { |
| vnet_disconnect_args_t _a = { 0 }, *a = &_a; |
| local_session_t *ls; |
| session_t *s; |
| |
| /* Server isn't interested, kill the session */ |
| if (mp->retval) |
| { |
| a->app_index = mp->context; |
| a->handle = mp->handle; |
| vnet_disconnect_session (a); |
| return; |
| } |
| |
| if (session_handle_is_local (mp->handle)) |
| { |
| ls = app_worker_get_local_session_from_handle (mp->handle); |
| if (!ls || ls->app_wrk_index != mp->context) |
| { |
| clib_warning ("server %u doesn't own local handle %llu", |
| mp->context, mp->handle); |
| return; |
| } |
| if (app_worker_local_session_connect_notify (ls)) |
| return; |
| ls->session_state = SESSION_STATE_READY; |
| } |
| else |
| { |
| s = session_get_from_handle_if_valid (mp->handle); |
| if (!s) |
| { |
| clib_warning ("session doesn't exist"); |
| return; |
| } |
| if (s->app_wrk_index != mp->context) |
| { |
| clib_warning ("app doesn't own session"); |
| return; |
| } |
| s->session_state = SESSION_STATE_READY; |
| } |
| } |
| |
| 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; |
| transport_connection_t *tc = 0; |
| vl_api_bind_sock_reply_t *rmp; |
| svm_msg_q_t *vpp_evt_q; |
| application_t *app = 0; |
| app_worker_t *app_wrk; |
| ip46_address_t *ip46; |
| app_listener_t *al; |
| session_t *s; |
| int rv = 0; |
| |
| if (session_manager_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: |
| /* *INDENT-OFF* */ |
| REPLY_MACRO2 (VL_API_BIND_SOCK_REPLY,({ |
| if (!rv) |
| { |
| rmp->handle = a->handle; |
| rmp->lcl_port = mp->port; |
| rmp->lcl_is_ip4 = mp->is_ip4; |
| if (app && application_has_global_scope (app)) |
| { |
| al = app_listener_get_w_handle (a->handle); |
| s = app_listener_get_session (al); |
| tc = listen_session_get_transport (s); |
| clib_memcpy_fast (rmp->lcl_ip, &tc->lcl_ip, sizeof (tc->lcl_ip)); |
| if (session_transport_service_type (s) == TRANSPORT_SERVICE_CL) |
| { |
| rmp->rx_fifo = pointer_to_uword (s->rx_fifo); |
| rmp->tx_fifo = pointer_to_uword (s->tx_fifo); |
| vpp_evt_q = session_manager_get_vpp_event_queue (0); |
| rmp->vpp_evt_q = pointer_to_uword (vpp_evt_q); |
| } |
| } |
| } |
| })); |
| /* *INDENT-ON* */ |
| |
| /* If app uses mq for control messages, send an mq message as well */ |
| if (app && application_use_mq_for_ctrl (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; |
| application_t *app; |
| int rv = 0; |
| |
| if (session_manager_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); |
| } |
| |
| static void |
| vl_api_connect_sock_t_handler (vl_api_connect_sock_t * mp) |
| { |
| vl_api_connect_session_reply_t *rmp; |
| vnet_connect_args_t _a, *a = &_a; |
| application_t *app = 0; |
| int rv = 0; |
| |
| if (session_manager_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_SESSION_REPLY); |
| |
| if (app && application_use_mq_for_ctrl (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_manager_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_manager_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_manager_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_manager_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: |
| */ |