blob: f1c58c4f2bd70bb85d75207cb99a99d7b5aa2102 [file] [log] [blame]
/*
* Copyright (c) 2017 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this
* 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 <stdio.h>
#include <stdlib.h>
#include <svm/svm_fifo_segment.h>
#include <vcl/vppcom.h>
#include <vcl/vcl_debug.h>
#include <vcl/vcl_private.h>
__thread uword __vcl_worker_index = ~0;
static int
vcl_wait_for_segment (u64 segment_handle)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
u32 wait_for_seconds = 10, segment_index;
f64 timeout;
if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
return 1;
timeout = clib_time_now (&wrk->clib_time) + wait_for_seconds;
while (clib_time_now (&wrk->clib_time) < timeout)
{
segment_index = vcl_segment_table_lookup (segment_handle);
if (segment_index != VCL_INVALID_SEGMENT_INDEX)
return 0;
usleep (10);
}
return 1;
}
const char *
vppcom_session_state_str (session_state_t state)
{
char *st;
switch (state)
{
case STATE_START:
st = "STATE_START";
break;
case STATE_CONNECT:
st = "STATE_CONNECT";
break;
case STATE_LISTEN:
st = "STATE_LISTEN";
break;
case STATE_ACCEPT:
st = "STATE_ACCEPT";
break;
case STATE_VPP_CLOSING:
st = "STATE_VPP_CLOSING";
break;
case STATE_DISCONNECT:
st = "STATE_DISCONNECT";
break;
case STATE_FAILED:
st = "STATE_FAILED";
break;
default:
st = "UNKNOWN_STATE";
break;
}
return st;
}
u8 *
format_ip4_address (u8 * s, va_list * args)
{
u8 *a = va_arg (*args, u8 *);
return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
}
u8 *
format_ip6_address (u8 * s, va_list * args)
{
ip6_address_t *a = va_arg (*args, ip6_address_t *);
u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
i_max_n_zero = ARRAY_LEN (a->as_u16);
max_n_zeros = 0;
i_first_zero = i_max_n_zero;
n_zeros = 0;
for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
{
u32 is_zero = a->as_u16[i] == 0;
if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
{
i_first_zero = i;
n_zeros = 0;
}
n_zeros += is_zero;
if ((!is_zero && n_zeros > max_n_zeros)
|| (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
{
i_max_n_zero = i_first_zero;
max_n_zeros = n_zeros;
i_first_zero = ARRAY_LEN (a->as_u16);
n_zeros = 0;
}
}
last_double_colon = 0;
for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
{
if (i == i_max_n_zero && max_n_zeros > 1)
{
s = format (s, "::");
i += max_n_zeros - 1;
last_double_colon = 1;
}
else
{
s = format (s, "%s%x",
(last_double_colon || i == 0) ? "" : ":",
clib_net_to_host_u16 (a->as_u16[i]));
last_double_colon = 0;
}
}
return s;
}
/* Format an IP46 address. */
u8 *
format_ip46_address (u8 * s, va_list * args)
{
ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
ip46_type_t type = va_arg (*args, ip46_type_t);
int is_ip4 = 1;
switch (type)
{
case IP46_TYPE_ANY:
is_ip4 = ip46_address_is_ip4 (ip46);
break;
case IP46_TYPE_IP4:
is_ip4 = 1;
break;
case IP46_TYPE_IP6:
is_ip4 = 0;
break;
}
return is_ip4 ?
format (s, "%U", format_ip4_address, &ip46->ip4) :
format (s, "%U", format_ip6_address, &ip46->ip6);
}
/*
* VPPCOM Utility Functions
*/
static svm_msg_q_t *
vcl_session_vpp_evt_q (vcl_worker_t * wrk, vcl_session_t * s)
{
if (vcl_session_is_ct (s))
return wrk->vpp_event_queues[0];
else
return wrk->vpp_event_queues[s->vpp_thread_index];
}
static void
vcl_send_session_accepted_reply (svm_msg_q_t * mq, u32 context,
session_handle_t handle, int retval)
{
app_session_evt_t _app_evt, *app_evt = &_app_evt;
session_accepted_reply_msg_t *rmp;
app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_ACCEPTED_REPLY);
rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
rmp->handle = handle;
rmp->context = context;
rmp->retval = retval;
app_send_ctrl_evt_to_vpp (mq, app_evt);
}
static void
vcl_send_session_disconnected_reply (svm_msg_q_t * mq, u32 context,
session_handle_t handle, int retval)
{
app_session_evt_t _app_evt, *app_evt = &_app_evt;
session_disconnected_reply_msg_t *rmp;
app_alloc_ctrl_evt_to_vpp (mq, app_evt,
SESSION_CTRL_EVT_DISCONNECTED_REPLY);
rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
rmp->handle = handle;
rmp->context = context;
rmp->retval = retval;
app_send_ctrl_evt_to_vpp (mq, app_evt);
}
static void
vcl_send_session_reset_reply (svm_msg_q_t * mq, u32 context,
session_handle_t handle, int retval)
{
app_session_evt_t _app_evt, *app_evt = &_app_evt;
session_reset_reply_msg_t *rmp;
app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_RESET_REPLY);
rmp = (session_reset_reply_msg_t *) app_evt->evt->data;
rmp->handle = handle;
rmp->context = context;
rmp->retval = retval;
app_send_ctrl_evt_to_vpp (mq, app_evt);
}
static u32
vcl_session_accepted_handler (vcl_worker_t * wrk, session_accepted_msg_t * mp)
{
vcl_session_t *session, *listen_session;
svm_fifo_t *rx_fifo, *tx_fifo;
u32 vpp_wrk_index;
svm_msg_q_t *evt_q;
session = vcl_session_alloc (wrk);
listen_session = vcl_session_table_lookup_listener (wrk,
mp->listener_handle);
if (!listen_session)
{
svm_msg_q_t *evt_q;
evt_q = uword_to_pointer (mp->vpp_event_queue_address, svm_msg_q_t *);
clib_warning ("VCL<%d>: ERROR: couldn't find listen session: "
"unknown vpp listener handle %llx",
getpid (), mp->listener_handle);
vcl_send_session_accepted_reply (evt_q, mp->context, mp->handle,
VNET_API_ERROR_INVALID_ARGUMENT);
vcl_session_free (wrk, session);
return VCL_INVALID_SESSION_INDEX;
}
rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
if (mp->server_event_queue_address)
{
session->vpp_evt_q = uword_to_pointer (mp->client_event_queue_address,
svm_msg_q_t *);
session->our_evt_q = uword_to_pointer (mp->server_event_queue_address,
svm_msg_q_t *);
if (vcl_wait_for_segment (mp->segment_handle))
{
clib_warning ("segment for session %u couldn't be mounted!",
session->session_index);
return VCL_INVALID_SESSION_INDEX;
}
rx_fifo->master_session_index = session->session_index;
tx_fifo->master_session_index = session->session_index;
rx_fifo->master_thread_index = vcl_get_worker_index ();
tx_fifo->master_thread_index = vcl_get_worker_index ();
vec_validate (wrk->vpp_event_queues, 0);
evt_q = uword_to_pointer (mp->vpp_event_queue_address, svm_msg_q_t *);
wrk->vpp_event_queues[0] = evt_q;
}
else
{
session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
svm_msg_q_t *);
rx_fifo->client_session_index = session->session_index;
tx_fifo->client_session_index = session->session_index;
rx_fifo->client_thread_index = vcl_get_worker_index ();
tx_fifo->client_thread_index = vcl_get_worker_index ();
vpp_wrk_index = tx_fifo->master_thread_index;
vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
}
session->vpp_handle = mp->handle;
session->vpp_thread_index = rx_fifo->master_thread_index;
session->client_context = mp->context;
session->rx_fifo = rx_fifo;
session->tx_fifo = tx_fifo;
session->session_state = STATE_ACCEPT;
session->transport.rmt_port = mp->port;
session->transport.is_ip4 = mp->is_ip4;
clib_memcpy_fast (&session->transport.rmt_ip, mp->ip,
sizeof (ip46_address_t));
vcl_session_table_add_vpp_handle (wrk, mp->handle, session->session_index);
session->transport.lcl_port = listen_session->transport.lcl_port;
session->transport.lcl_ip = listen_session->transport.lcl_ip;
session->session_type = listen_session->session_type;
session->is_dgram = session->session_type == VPPCOM_PROTO_UDP;
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: client accept request from %s"
" address %U port %d queue %p!", getpid (), mp->handle,
session->session_index,
mp->is_ip4 ? "IPv4" : "IPv6", format_ip46_address, &mp->ip,
mp->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (mp->port), session->vpp_evt_q);
vcl_evt (VCL_EVT_ACCEPT, session, listen_session, session_index);
return session->session_index;
}
static u32
vcl_session_connected_handler (vcl_worker_t * wrk,
session_connected_msg_t * mp)
{
u32 session_index, vpp_wrk_index;
svm_fifo_t *rx_fifo, *tx_fifo;
vcl_session_t *session = 0;
svm_msg_q_t *evt_q;
session_index = mp->context;
session = vcl_session_get (wrk, session_index);
if (!session)
{
clib_warning ("[%s] ERROR: vpp handle 0x%llx, sid %u: "
"Invalid session index (%u)!",
getpid (), mp->handle, session_index);
return VCL_INVALID_SESSION_INDEX;
}
if (mp->retval)
{
clib_warning ("VCL<%d>: ERROR: sid %u: connect failed! %U", getpid (),
session_index, format_api_error, ntohl (mp->retval));
session->session_state = STATE_FAILED;
session->vpp_handle = mp->handle;
return session_index;
}
rx_fifo = uword_to_pointer (mp->server_rx_fifo, svm_fifo_t *);
tx_fifo = uword_to_pointer (mp->server_tx_fifo, svm_fifo_t *);
if (vcl_wait_for_segment (mp->segment_handle))
{
clib_warning ("segment for session %u couldn't be mounted!",
session->session_index);
return VCL_INVALID_SESSION_INDEX;
}
rx_fifo->client_session_index = session_index;
tx_fifo->client_session_index = session_index;
rx_fifo->client_thread_index = vcl_get_worker_index ();
tx_fifo->client_thread_index = vcl_get_worker_index ();
if (mp->client_event_queue_address)
{
session->vpp_evt_q = uword_to_pointer (mp->server_event_queue_address,
svm_msg_q_t *);
session->our_evt_q = uword_to_pointer (mp->client_event_queue_address,
svm_msg_q_t *);
vec_validate (wrk->vpp_event_queues, 0);
evt_q = uword_to_pointer (mp->vpp_event_queue_address, svm_msg_q_t *);
wrk->vpp_event_queues[0] = evt_q;
}
else
{
session->vpp_evt_q = uword_to_pointer (mp->vpp_event_queue_address,
svm_msg_q_t *);
vpp_wrk_index = tx_fifo->master_thread_index;
vec_validate (wrk->vpp_event_queues, vpp_wrk_index);
wrk->vpp_event_queues[vpp_wrk_index] = session->vpp_evt_q;
}
session->rx_fifo = rx_fifo;
session->tx_fifo = tx_fifo;
session->vpp_handle = mp->handle;
session->vpp_thread_index = rx_fifo->master_thread_index;
session->transport.is_ip4 = mp->is_ip4;
clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
sizeof (session->transport.lcl_ip));
session->transport.lcl_port = mp->lcl_port;
session->session_state = STATE_CONNECT;
/* Add it to lookup table */
vcl_session_table_add_vpp_handle (wrk, mp->handle, session_index);
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: connect succeeded! "
"session_rx_fifo %p, refcnt %d, session_tx_fifo %p, refcnt %d",
getpid (), mp->handle, session_index, session->rx_fifo,
session->rx_fifo->refcnt, session->tx_fifo, session->tx_fifo->refcnt);
return session_index;
}
static int
vcl_flag_accepted_session (vcl_session_t * session, u64 handle, u32 flags)
{
vcl_session_msg_t *accepted_msg;
int i;
for (i = 0; i < vec_len (session->accept_evts_fifo); i++)
{
accepted_msg = &session->accept_evts_fifo[i];
if (accepted_msg->accepted_msg.handle == handle)
{
accepted_msg->flags |= flags;
return 1;
}
}
return 0;
}
static u32
vcl_session_reset_handler (vcl_worker_t * wrk,
session_reset_msg_t * reset_msg)
{
vcl_session_t *session;
u32 sid;
sid = vcl_session_index_from_vpp_handle (wrk, reset_msg->handle);
session = vcl_session_get (wrk, sid);
if (!session)
{
VDBG (0, "request to reset unknown handle 0x%llx", reset_msg->handle);
return VCL_INVALID_SESSION_INDEX;
}
/* Caught a reset before actually accepting the session */
if (session->session_state == STATE_LISTEN)
{
if (!vcl_flag_accepted_session (session, reset_msg->handle,
VCL_ACCEPTED_F_RESET))
VDBG (0, "session was not accepted!");
return VCL_INVALID_SESSION_INDEX;
}
session->session_state = STATE_DISCONNECT;
VDBG (0, "reset session %u [0x%llx]", sid, reset_msg->handle);
return sid;
}
static u32
vcl_session_bound_handler (vcl_worker_t * wrk, session_bound_msg_t * mp)
{
vcl_session_t *session;
u32 sid = mp->context;
session = vcl_session_get (wrk, sid);
if (mp->retval)
{
VERR ("vpp handle 0x%llx, sid %u: bind failed: %U", mp->handle, sid,
format_api_error, mp->retval);
if (session)
{
session->session_state = STATE_FAILED;
session->vpp_handle = mp->handle;
return sid;
}
else
{
clib_warning ("[%s] ERROR: vpp handle 0x%llx, sid %u: "
"Invalid session index (%u)!",
getpid (), mp->handle, sid);
return VCL_INVALID_SESSION_INDEX;
}
}
session->vpp_handle = mp->handle;
session->transport.is_ip4 = mp->lcl_is_ip4;
clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
sizeof (ip46_address_t));
session->transport.lcl_port = mp->lcl_port;
vcl_session_table_add_listener (wrk, mp->handle, sid);
session->session_state = STATE_LISTEN;
if (session->is_dgram)
{
svm_fifo_t *rx_fifo, *tx_fifo;
session->vpp_evt_q = uword_to_pointer (mp->vpp_evt_q, svm_msg_q_t *);
rx_fifo = uword_to_pointer (mp->rx_fifo, svm_fifo_t *);
rx_fifo->client_session_index = sid;
tx_fifo = uword_to_pointer (mp->tx_fifo, svm_fifo_t *);
tx_fifo->client_session_index = sid;
session->rx_fifo = rx_fifo;
session->tx_fifo = tx_fifo;
}
VDBG (0, "session %u [0x%llx]: listen succeeded!", sid, mp->handle);
return sid;
}
static vcl_session_t *
vcl_session_accepted (vcl_worker_t * wrk, session_accepted_msg_t * msg)
{
vcl_session_msg_t *vcl_msg;
vcl_session_t *session;
session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
if (PREDICT_FALSE (session != 0))
VWRN ("session overlap handle %lu state %u!", msg->handle,
session->session_state);
session = vcl_session_table_lookup_listener (wrk, msg->listener_handle);
if (!session)
{
VERR ("couldn't find listen session: listener handle %llx",
msg->listener_handle);
return 0;
}
clib_fifo_add2 (session->accept_evts_fifo, vcl_msg);
vcl_msg->accepted_msg = *msg;
/* Session handle points to listener until fully accepted by app */
vcl_session_table_add_vpp_handle (wrk, msg->handle, session->session_index);
return session;
}
static vcl_session_t *
vcl_session_disconnected_handler (vcl_worker_t * wrk,
session_disconnected_msg_t * msg)
{
vcl_session_t *session;
session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
if (!session)
{
VDBG (0, "request to disconnect unknown handle 0x%llx", msg->handle);
return 0;
}
/* Caught a disconnect before actually accepting the session */
if (session->session_state == STATE_LISTEN)
{
if (!vcl_flag_accepted_session (session, msg->handle,
VCL_ACCEPTED_F_CLOSED))
VDBG (0, "session was not accepted!");
return 0;
}
session->session_state = STATE_VPP_CLOSING;
return session;
}
static int
vcl_handle_mq_event (vcl_worker_t * wrk, session_event_t * e)
{
session_disconnected_msg_t *disconnected_msg;
vcl_session_t *session;
switch (e->event_type)
{
case FIFO_EVENT_APP_RX:
case FIFO_EVENT_APP_TX:
case SESSION_IO_EVT_CT_RX:
case SESSION_IO_EVT_CT_TX:
vec_add1 (wrk->unhandled_evts_vector, *e);
break;
case SESSION_CTRL_EVT_ACCEPTED:
vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
break;
case SESSION_CTRL_EVT_CONNECTED:
vcl_session_connected_handler (wrk,
(session_connected_msg_t *) e->data);
break;
case SESSION_CTRL_EVT_DISCONNECTED:
disconnected_msg = (session_disconnected_msg_t *) e->data;
session = vcl_session_disconnected_handler (wrk, disconnected_msg);
if (!session)
break;
session->session_state = STATE_DISCONNECT;
VDBG (0, "disconnected session %u [0x%llx]", session->session_index,
session->vpp_handle);
break;
case SESSION_CTRL_EVT_RESET:
vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
break;
case SESSION_CTRL_EVT_BOUND:
vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
break;
default:
clib_warning ("unhandled %u", e->event_type);
}
return VPPCOM_OK;
}
static inline int
vppcom_wait_for_session_state_change (u32 session_index,
session_state_t state,
f64 wait_for_time)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
vcl_session_t *volatile session;
svm_msg_q_msg_t msg;
session_event_t *e;
do
{
session = vcl_session_get (wrk, session_index);
if (PREDICT_FALSE (!session))
{
return VPPCOM_EBADFD;
}
if (session->session_state & state)
{
return VPPCOM_OK;
}
if (session->session_state & STATE_FAILED)
{
return VPPCOM_ECONNREFUSED;
}
if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0))
{
usleep (100);
continue;
}
e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
vcl_handle_mq_event (wrk, e);
svm_msg_q_free_msg (wrk->app_event_queue, &msg);
}
while (clib_time_now (&wrk->clib_time) < timeout);
VDBG (0, "timeout waiting for state 0x%x (%s)", state,
vppcom_session_state_str (state));
vcl_evt (VCL_EVT_SESSION_TIMEOUT, session, session_state);
return VPPCOM_ETIMEDOUT;
}
static int
vppcom_app_session_enable (void)
{
int rv;
if (vcm->app_state != STATE_APP_ENABLED)
{
vppcom_send_session_enable_disable (1 /* is_enabled == TRUE */ );
rv = vcl_wait_for_app_state_change (STATE_APP_ENABLED);
if (PREDICT_FALSE (rv))
{
VDBG (0, "VCL<%d>: application session enable timed out! "
"returning %d (%s)", getpid (), rv, vppcom_retval_str (rv));
return rv;
}
}
return VPPCOM_OK;
}
static int
vppcom_app_attach (void)
{
int rv;
vppcom_app_send_attach ();
rv = vcl_wait_for_app_state_change (STATE_APP_ATTACHED);
if (PREDICT_FALSE (rv))
{
VDBG (0, "VCL<%d>: application attach timed out! returning %d (%s)",
getpid (), rv, vppcom_retval_str (rv));
return rv;
}
return VPPCOM_OK;
}
static int
vppcom_session_unbind (u32 session_handle)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session = 0;
u64 vpp_handle;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
vpp_handle = session->vpp_handle;
vcl_session_table_del_listener (wrk, vpp_handle);
session->vpp_handle = ~0;
session->session_state = STATE_DISCONNECT;
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: sending unbind msg! new state"
" 0x%x (%s)", getpid (), vpp_handle, session_handle, STATE_DISCONNECT,
vppcom_session_state_str (STATE_DISCONNECT));
vcl_evt (VCL_EVT_UNBIND, session);
vppcom_send_unbind_sock (vpp_handle);
return VPPCOM_OK;
}
static int
vppcom_session_disconnect (u32 session_handle)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
svm_msg_q_t *vpp_evt_q;
vcl_session_t *session;
session_state_t state;
u64 vpp_handle;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
vpp_handle = session->vpp_handle;
state = session->session_state;
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u state 0x%x (%s)", getpid (),
vpp_handle, session_handle, state, vppcom_session_state_str (state));
if (PREDICT_FALSE (state & STATE_LISTEN))
{
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"Cannot disconnect a listen socket!",
getpid (), vpp_handle, session_handle);
return VPPCOM_EBADFD;
}
if (state & STATE_VPP_CLOSING)
{
vpp_evt_q = vcl_session_vpp_evt_q (wrk, session);
vcl_send_session_disconnected_reply (vpp_evt_q, wrk->my_client_index,
vpp_handle, 0);
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: sending disconnect "
"REPLY...", getpid (), vpp_handle, session_handle);
}
else
{
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: sending disconnect...",
getpid (), vpp_handle, session_handle);
vppcom_send_disconnect_session (vpp_handle);
}
return VPPCOM_OK;
}
static void
vcl_cleanup_bapi (void)
{
socket_client_main_t *scm = &socket_client_main;
api_main_t *am = &api_main;
am->my_client_index = ~0;
am->my_registration = 0;
am->vl_input_queue = 0;
am->msg_index_by_name_and_crc = 0;
scm->socket_fd = 0;
vl_client_api_unmap ();
}
static void
vcl_cleanup_forked_child (vcl_worker_t * wrk, vcl_worker_t * child_wrk)
{
vcl_worker_t *sub_child;
int tries = 0;
if (child_wrk->forked_child != ~0)
{
sub_child = vcl_worker_get_if_valid (child_wrk->forked_child);
if (sub_child)
{
/* Wait a bit, maybe the process is going away */
while (kill (sub_child->current_pid, 0) >= 0 && tries++ < 50)
usleep (1e3);
if (kill (sub_child->current_pid, 0) < 0)
vcl_cleanup_forked_child (child_wrk, sub_child);
}
}
vcl_worker_cleanup (child_wrk, 1 /* notify vpp */ );
VDBG (0, "Cleaned up wrk %u", child_wrk->wrk_index);
wrk->forked_child = ~0;
}
static struct sigaction old_sa;
static void
vcl_intercept_sigchld_handler (int signum, siginfo_t * si, void *uc)
{
vcl_worker_t *wrk, *child_wrk;
if (vcl_get_worker_index () == ~0)
return;
if (sigaction (SIGCHLD, &old_sa, 0))
{
VERR ("couldn't restore sigchld");
exit (-1);
}
wrk = vcl_worker_get_current ();
if (wrk->forked_child == ~0)
return;
child_wrk = vcl_worker_get_if_valid (wrk->forked_child);
if (!child_wrk)
goto done;
if (si && si->si_pid != child_wrk->current_pid)
{
VDBG (0, "unexpected child pid %u", si->si_pid);
goto done;
}
vcl_cleanup_forked_child (wrk, child_wrk);
done:
if (old_sa.sa_flags & SA_SIGINFO)
{
void (*fn) (int, siginfo_t *, void *) = old_sa.sa_sigaction;
fn (signum, si, uc);
}
else
{
void (*fn) (int) = old_sa.sa_handler;
if (fn)
fn (signum);
}
}
static void
vcl_incercept_sigchld ()
{
struct sigaction sa;
clib_memset (&sa, 0, sizeof (sa));
sa.sa_sigaction = vcl_intercept_sigchld_handler;
sa.sa_flags = SA_SIGINFO;
if (sigaction (SIGCHLD, &sa, &old_sa))
{
VERR ("couldn't intercept sigchld");
exit (-1);
}
}
static void
vcl_app_pre_fork (void)
{
vcl_incercept_sigchld ();
}
static void
vcl_app_fork_child_handler (void)
{
int rv, parent_wrk_index;
vcl_worker_t *parent_wrk;
u8 *child_name;
parent_wrk_index = vcl_get_worker_index ();
VDBG (0, "initializing forked child with parent wrk %u", parent_wrk_index);
/*
* Allocate worker
*/
vcl_set_worker_index (~0);
if (!vcl_worker_alloc_and_init ())
VERR ("couldn't allocate new worker");
/*
* Attach to binary api
*/
child_name = format (0, "%v-child-%u%c", vcm->app_name, getpid (), 0);
vcl_cleanup_bapi ();
vppcom_api_hookup ();
vcm->app_state = STATE_APP_START;
rv = vppcom_connect_to_vpp ((char *) child_name);
vec_free (child_name);
if (rv)
{
VERR ("couldn't connect to VPP!");
return;
}
/*
* Register worker with vpp and share sessions
*/
vcl_worker_register_with_vpp ();
parent_wrk = vcl_worker_get (parent_wrk_index);
vcl_worker_share_sessions (parent_wrk);
parent_wrk->forked_child = vcl_get_worker_index ();
VDBG (0, "forked child main worker initialized");
vcm->forking = 0;
}
static void
vcl_app_fork_parent_handler (void)
{
vcm->forking = 1;
while (vcm->forking)
;
}
/**
* Handle app exit
*
* Notify vpp of the disconnect and mark the worker as free. If we're the
* last worker, do a full cleanup otherwise, since we're probably a forked
* child, avoid syscalls as much as possible. We might've lost privileges.
*/
void
vppcom_app_exit (void)
{
if (!pool_elts (vcm->workers))
return;
vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
vcl_set_worker_index (~0);
vcl_elog_stop (vcm);
if (vec_len (vcm->workers) == 1)
vl_client_disconnect_from_vlib ();
else
vl_client_send_disconnect (1 /* vpp should cleanup */ );
}
/*
* VPPCOM Public API functions
*/
int
vppcom_app_create (char *app_name)
{
vppcom_cfg_t *vcl_cfg = &vcm->cfg;
int rv;
if (vcm->is_init)
{
VDBG (1, "already initialized");
return VPPCOM_EEXIST;
}
vcm->is_init = 1;
vppcom_cfg (&vcm->cfg);
vcl_cfg = &vcm->cfg;
vcm->main_cpu = pthread_self ();
vcm->main_pid = getpid ();
vcm->app_name = format (0, "%s", app_name);
vppcom_init_error_string_table ();
svm_fifo_segment_main_init (&vcm->segment_main, vcl_cfg->segment_baseva,
20 /* timeout in secs */ );
pool_alloc (vcm->workers, vcl_cfg->max_workers);
clib_spinlock_init (&vcm->workers_lock);
clib_rwlock_init (&vcm->segment_table_lock);
pthread_atfork (vcl_app_pre_fork, vcl_app_fork_parent_handler,
vcl_app_fork_child_handler);
atexit (vppcom_app_exit);
/* Allocate default worker */
vcl_worker_alloc_and_init ();
/* API hookup and connect to VPP */
vppcom_api_hookup ();
vcl_elog_init (vcm);
vcm->app_state = STATE_APP_START;
rv = vppcom_connect_to_vpp (app_name);
if (rv)
{
VERR ("couldn't connect to VPP!");
return rv;
}
VDBG (0, "sending session enable");
rv = vppcom_app_session_enable ();
if (rv)
{
VERR ("vppcom_app_session_enable() failed!");
return rv;
}
VDBG (0, "sending app attach");
rv = vppcom_app_attach ();
if (rv)
{
VERR ("vppcom_app_attach() failed!");
return rv;
}
VDBG (0, "app_name '%s', my_client_index %d (0x%x)", app_name,
vcm->workers[0].my_client_index, vcm->workers[0].my_client_index);
return VPPCOM_OK;
}
void
vppcom_app_destroy (void)
{
int rv;
f64 orig_app_timeout;
if (!pool_elts (vcm->workers))
return;
vcl_evt (VCL_EVT_DETACH, vcm);
if (pool_elts (vcm->workers) == 1)
{
vppcom_app_send_detach ();
orig_app_timeout = vcm->cfg.app_timeout;
vcm->cfg.app_timeout = 2.0;
rv = vcl_wait_for_app_state_change (STATE_APP_ENABLED);
vcm->cfg.app_timeout = orig_app_timeout;
if (PREDICT_FALSE (rv))
VDBG (0, "application detach timed out! returning %d (%s)", rv,
vppcom_retval_str (rv));
vec_free (vcm->app_name);
vcl_worker_cleanup (vcl_worker_get_current (), 0 /* notify vpp */ );
}
else
{
vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
}
vcl_set_worker_index (~0);
vcl_elog_stop (vcm);
vl_client_disconnect_from_vlib ();
}
int
vppcom_session_create (u8 proto, u8 is_nonblocking)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session;
session = vcl_session_alloc (wrk);
session->session_type = proto;
session->session_state = STATE_START;
session->vpp_handle = ~0;
session->is_dgram = proto == VPPCOM_PROTO_UDP;
if (is_nonblocking)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
vcl_evt (VCL_EVT_CREATE, session, session_type, session->session_state,
is_nonblocking, session_index);
VDBG (0, "created sid %u", session->session_index);
return vcl_session_handle (session);
}
int
vppcom_session_close (uint32_t session_handle)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
u8 is_vep, do_disconnect = 1;
vcl_session_t *session = 0;
session_state_t state;
u32 next_sh, vep_sh;
int rv = VPPCOM_OK;
u64 vpp_handle;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
if (session->shared_index != ~0)
do_disconnect = vcl_worker_unshare_session (wrk, session);
is_vep = session->is_vep;
next_sh = session->vep.next_sh;
vep_sh = session->vep.vep_sh;
state = session->session_state;
vpp_handle = session->vpp_handle;
VDBG (1, "closing session handle %u vpp handle %u", session_handle,
vpp_handle);
if (is_vep)
{
while (next_sh != ~0)
{
rv = vppcom_epoll_ctl (session_handle, EPOLL_CTL_DEL, next_sh, 0);
if (PREDICT_FALSE (rv < 0))
VDBG (0, "vpp handle 0x%llx, sid %u: EPOLL_CTL_DEL vep_idx %u"
" failed! rv %d (%s)", vpp_handle, next_sh, vep_sh, rv,
vppcom_retval_str (rv));
next_sh = session->vep.next_sh;
}
}
else
{
if (session->is_vep_session)
{
rv = vppcom_epoll_ctl (vep_sh, EPOLL_CTL_DEL, session_handle, 0);
if (rv < 0)
VDBG (0, "vpp handle 0x%llx, sid %u: EPOLL_CTL_DEL vep_idx %u "
"failed! rv %d (%s)", vpp_handle, session_handle, vep_sh,
rv, vppcom_retval_str (rv));
}
if (!do_disconnect)
goto cleanup;
if (state & STATE_LISTEN)
{
rv = vppcom_session_unbind (session_handle);
if (PREDICT_FALSE (rv < 0))
VDBG (0, "vpp handle 0x%llx, sid %u: listener unbind failed! "
"rv %d (%s)", vpp_handle, session_handle, rv,
vppcom_retval_str (rv));
}
else if (state & STATE_OPEN)
{
rv = vppcom_session_disconnect (session_handle);
if (PREDICT_FALSE (rv < 0))
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"session disconnect failed! rv %d (%s)",
getpid (), vpp_handle, session_handle,
rv, vppcom_retval_str (rv));
}
else if (state == STATE_DISCONNECT)
{
svm_msg_q_t *mq = vcl_session_vpp_evt_q (wrk, session);
vcl_send_session_reset_reply (mq, wrk->my_client_index,
session->vpp_handle, 0);
}
}
cleanup:
if (vcl_session_is_ct (session))
{
vcl_cut_through_registration_t *ctr;
uword mq_addr;
mq_addr = pointer_to_uword (session->our_evt_q);
ctr = vcl_ct_registration_lock_and_lookup (wrk, mq_addr);
ASSERT (ctr);
if (ctr->epoll_evt_conn_index != ~0)
vcl_mq_epoll_del_evfd (wrk, ctr->epoll_evt_conn_index);
VDBG (0, "Removing ct registration %u",
vcl_ct_registration_index (wrk, ctr));
vcl_ct_registration_del (wrk, ctr);
vcl_ct_registration_lookup_del (wrk, mq_addr);
vcl_ct_registration_unlock (wrk);
}
if (vpp_handle != ~0)
{
vcl_session_table_del_vpp_handle (wrk, vpp_handle);
}
vcl_session_free (wrk, session);
VDBG (0, "session handle %u [0x%llx] removed", session_handle, vpp_handle);
vcl_evt (VCL_EVT_CLOSE, session, rv);
return rv;
}
int
vppcom_session_bind (uint32_t session_handle, vppcom_endpt_t * ep)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session = 0;
if (!ep || !ep->ip)
return VPPCOM_EINVAL;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
if (session->is_vep)
{
clib_warning ("VCL<%d>: ERROR: sid %u: cannot "
"bind to an epoll session!", getpid (), session_handle);
return VPPCOM_EBADFD;
}
session->transport.is_ip4 = ep->is_ip4;
if (ep->is_ip4)
clib_memcpy_fast (&session->transport.lcl_ip.ip4, ep->ip,
sizeof (ip4_address_t));
else
clib_memcpy_fast (&session->transport.lcl_ip.ip6, ep->ip,
sizeof (ip6_address_t));
session->transport.lcl_port = ep->port;
VDBG (0, "VCL<%d>: sid %u: binding to local %s address %U port %u, "
"proto %s", getpid (), session_handle,
session->transport.is_ip4 ? "IPv4" : "IPv6",
format_ip46_address, &session->transport.lcl_ip,
session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (session->transport.lcl_port),
session->session_type ? "UDP" : "TCP");
vcl_evt (VCL_EVT_BIND, session);
if (session->session_type == VPPCOM_PROTO_UDP)
vppcom_session_listen (session_handle, 10);
return VPPCOM_OK;
}
int
vppcom_session_listen (uint32_t listen_sh, uint32_t q_len)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *listen_session = 0;
u64 listen_vpp_handle;
int rv;
listen_session = vcl_session_get_w_handle (wrk, listen_sh);
if (!listen_session || listen_session->is_vep)
return VPPCOM_EBADFD;
if (q_len == 0 || q_len == ~0)
q_len = vcm->cfg.listen_queue_size;
listen_vpp_handle = listen_session->vpp_handle;
if (listen_session->session_state & STATE_LISTEN)
{
VDBG (0, "session %u [0x%llx]: already in listen state!",
listen_sh, listen_vpp_handle);
return VPPCOM_OK;
}
VDBG (0, "session %u [0x%llx]: sending vpp listen request...",
listen_sh, listen_vpp_handle);
/*
* Send listen request to vpp and wait for reply
*/
vppcom_send_bind_sock (listen_session);
rv = vppcom_wait_for_session_state_change (listen_session->session_index,
STATE_LISTEN,
vcm->cfg.session_timeout);
if (PREDICT_FALSE (rv))
{
listen_session = vcl_session_get_w_handle (wrk, listen_sh);
VDBG (0, "session %u [0x%llx]: listen failed! returning %d (%s)",
listen_sh, listen_session->vpp_handle, rv,
vppcom_retval_str (rv));
return rv;
}
return VPPCOM_OK;
}
static int
validate_args_session_accept_ (vcl_worker_t * wrk,
vcl_session_t * listen_session)
{
/* Input validation - expects spinlock on sessions_lockp */
if (listen_session->is_vep)
{
clib_warning ("VCL<%d>: ERROR: sid %u: cannot accept on an "
"epoll session!", getpid (),
listen_session->session_index);
return VPPCOM_EBADFD;
}
if (listen_session->session_state != STATE_LISTEN)
{
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"not in listen state! state 0x%x (%s)", getpid (),
listen_session->vpp_handle, listen_session->session_index,
listen_session->session_state,
vppcom_session_state_str (listen_session->session_state));
return VPPCOM_EBADFD;
}
return VPPCOM_OK;
}
int
vppcom_session_accept (uint32_t listen_session_handle, vppcom_endpt_t * ep,
uint32_t flags)
{
u32 client_session_index = ~0, listen_session_index, accept_flags = 0;
vcl_worker_t *wrk = vcl_worker_get_current ();
session_accepted_msg_t accepted_msg;
vcl_session_t *listen_session = 0;
vcl_session_t *client_session = 0;
svm_msg_q_t *vpp_evt_q;
vcl_session_msg_t *evt;
u64 listen_vpp_handle;
svm_msg_q_msg_t msg;
session_event_t *e;
u8 is_nonblocking;
int rv;
listen_session = vcl_session_get_w_handle (wrk, listen_session_handle);
if (!listen_session)
return VPPCOM_EBADFD;
listen_session_index = listen_session->session_index;
if ((rv = validate_args_session_accept_ (wrk, listen_session)))
return rv;
if (clib_fifo_elts (listen_session->accept_evts_fifo))
{
clib_fifo_sub2 (listen_session->accept_evts_fifo, evt);
accept_flags = evt->flags;
accepted_msg = evt->accepted_msg;
goto handle;
}
is_nonblocking = VCL_SESS_ATTR_TEST (listen_session->attr,
VCL_SESS_ATTR_NONBLOCK);
if (svm_msg_q_is_empty (wrk->app_event_queue) && is_nonblocking)
return VPPCOM_EAGAIN;
while (1)
{
if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_WAIT, 0))
return VPPCOM_EAGAIN;
e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
if (e->event_type != SESSION_CTRL_EVT_ACCEPTED)
{
clib_warning ("discarded event: %u", e->event_type);
svm_msg_q_free_msg (wrk->app_event_queue, &msg);
continue;
}
clib_memcpy_fast (&accepted_msg, e->data, sizeof (accepted_msg));
svm_msg_q_free_msg (wrk->app_event_queue, &msg);
break;
}
handle:
client_session_index = vcl_session_accepted_handler (wrk, &accepted_msg);
listen_session = vcl_session_get (wrk, listen_session_index);
client_session = vcl_session_get (wrk, client_session_index);
if (flags & O_NONBLOCK)
VCL_SESS_ATTR_SET (client_session->attr, VCL_SESS_ATTR_NONBLOCK);
listen_vpp_handle = listen_session->vpp_handle;
VDBG (1, "vpp handle 0x%llx, sid %u: Got a client request! "
"vpp handle 0x%llx, sid %u, flags %d, is_nonblocking %u",
listen_vpp_handle, listen_session_handle,
client_session->vpp_handle, client_session_index,
flags, VCL_SESS_ATTR_TEST (client_session->attr,
VCL_SESS_ATTR_NONBLOCK));
if (ep)
{
ep->is_ip4 = client_session->transport.is_ip4;
ep->port = client_session->transport.rmt_port;
if (client_session->transport.is_ip4)
clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip4,
sizeof (ip4_address_t));
else
clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip6,
sizeof (ip6_address_t));
}
if (accepted_msg.server_event_queue_address)
vpp_evt_q = uword_to_pointer (accepted_msg.vpp_event_queue_address,
svm_msg_q_t *);
else
vpp_evt_q = client_session->vpp_evt_q;
vcl_send_session_accepted_reply (vpp_evt_q, client_session->client_context,
client_session->vpp_handle, 0);
VDBG (0, "listener %u [0x%llx] accepted %u [0x%llx] peer: %U:%u "
"local: %U:%u", listen_session_handle, listen_vpp_handle,
client_session_index, client_session->vpp_handle,
format_ip46_address, &client_session->transport.rmt_ip,
client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (client_session->transport.rmt_port),
format_ip46_address, &client_session->transport.lcl_ip,
client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (client_session->transport.lcl_port));
vcl_evt (VCL_EVT_ACCEPT, client_session, listen_session,
client_session_index);
/*
* Session might have been closed already
*/
if (accept_flags)
{
if (accept_flags & VCL_ACCEPTED_F_CLOSED)
client_session->session_state = STATE_VPP_CLOSING;
else if (accept_flags & VCL_ACCEPTED_F_RESET)
client_session->session_state = STATE_DISCONNECT;
}
return vcl_session_handle (client_session);
}
int
vppcom_session_connect (uint32_t session_handle, vppcom_endpt_t * server_ep)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session = 0;
u32 session_index;
int rv;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
session_index = session->session_index;
if (PREDICT_FALSE (session->is_vep))
{
clib_warning ("VCL<%d>: ERROR: sid %u: cannot "
"connect on an epoll session!", getpid (),
session_handle);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (session->session_state & CLIENT_STATE_OPEN))
{
VDBG (0, "VCL<%d>: vpp handle 0x%llx, sid %u: session already "
"connected to %s %U port %d proto %s, state 0x%x (%s)",
getpid (), session->vpp_handle, session_handle,
session->transport.is_ip4 ? "IPv4" : "IPv6",
format_ip46_address,
&session->transport.rmt_ip, session->transport.is_ip4 ?
IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (session->transport.rmt_port),
session->session_type ? "UDP" : "TCP", session->session_state,
vppcom_session_state_str (session->session_state));
return VPPCOM_OK;
}
session->transport.is_ip4 = server_ep->is_ip4;
if (session->transport.is_ip4)
clib_memcpy_fast (&session->transport.rmt_ip.ip4, server_ep->ip,
sizeof (ip4_address_t));
else
clib_memcpy_fast (&session->transport.rmt_ip.ip6, server_ep->ip,
sizeof (ip6_address_t));
session->transport.rmt_port = server_ep->port;
VDBG (0, "VCL<%d>: vpp handle 0x%llx, sid %u: connecting to server %s %U "
"port %d proto %s",
getpid (), session->vpp_handle, session_handle,
session->transport.is_ip4 ? "IPv4" : "IPv6",
format_ip46_address,
&session->transport.rmt_ip, session->transport.is_ip4 ?
IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (session->transport.rmt_port),
session->session_type ? "UDP" : "TCP");
/*
* Send connect request and wait for reply from vpp
*/
vppcom_send_connect_sock (session);
rv = vppcom_wait_for_session_state_change (session_index, STATE_CONNECT,
vcm->cfg.session_timeout);
session = vcl_session_get (wrk, session_index);
if (PREDICT_FALSE (rv))
{
if (VPPCOM_DEBUG > 0)
{
if (session)
clib_warning ("VCL<%d>: vpp handle 0x%llx, sid %u: connect "
"failed! returning %d (%s)", getpid (),
session->vpp_handle, session_handle, rv,
vppcom_retval_str (rv));
else
clib_warning ("VCL<%d>: no session for sid %u: connect failed! "
"returning %d (%s)", getpid (),
session_handle, rv, vppcom_retval_str (rv));
}
}
else
VDBG (0, "VCL<%d>: vpp handle 0x%llx, sid %u: connected!",
getpid (), session->vpp_handle, session_handle);
return rv;
}
static u8
vcl_is_rx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
{
if (!is_ct)
return (e->event_type == FIFO_EVENT_APP_RX
&& e->fifo->client_session_index == sid);
else
return (e->event_type == SESSION_IO_EVT_CT_TX);
}
static inline u8
vcl_session_is_readable (vcl_session_t * s)
{
return ((s->session_state & STATE_OPEN)
|| (s->session_state == STATE_LISTEN
&& s->session_type == VPPCOM_PROTO_UDP));
}
static inline int
vppcom_session_read_internal (uint32_t session_handle, void *buf, int n,
u8 peek)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
int n_read = 0, rv, is_nonblocking;
vcl_session_t *s = 0;
svm_fifo_t *rx_fifo;
svm_msg_q_msg_t msg;
session_event_t *e;
svm_msg_q_t *mq;
u8 is_ct;
if (PREDICT_FALSE (!buf))
return VPPCOM_EINVAL;
s = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!s || s->is_vep))
return VPPCOM_EBADFD;
if (PREDICT_FALSE (!vcl_session_is_readable (s)))
{
session_state_t state = s->session_state;
rv = ((state & STATE_DISCONNECT) ? VPPCOM_ECONNRESET : VPPCOM_ENOTCONN);
VDBG (0, "session handle %u[0x%llx] is not open! state 0x%x (%s),"
" returning %d (%s)", session_handle, s->vpp_handle, state,
vppcom_session_state_str (state), rv, vppcom_retval_str (rv));
return rv;
}
is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
is_ct = vcl_session_is_ct (s);
mq = is_ct ? s->our_evt_q : wrk->app_event_queue;
rx_fifo = s->rx_fifo;
s->has_rx_evt = 0;
if (svm_fifo_is_empty (rx_fifo))
{
if (is_nonblocking)
{
svm_fifo_unset_event (rx_fifo);
return VPPCOM_EWOULDBLOCK;
}
while (svm_fifo_is_empty (rx_fifo))
{
svm_fifo_unset_event (rx_fifo);
svm_msg_q_lock (mq);
if (svm_msg_q_is_empty (mq))
svm_msg_q_wait (mq);
svm_msg_q_sub_w_lock (mq, &msg);
e = svm_msg_q_msg_data (mq, &msg);
svm_msg_q_unlock (mq);
if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
vcl_handle_mq_event (wrk, e);
svm_msg_q_free_msg (mq, &msg);
if (PREDICT_FALSE (s->session_state == STATE_DISCONNECT))
return VPPCOM_ECONNRESET;
}
}
if (s->is_dgram)
n_read = app_recv_dgram_raw (rx_fifo, buf, n, &s->transport, 0, peek);
else
n_read = app_recv_stream_raw (rx_fifo, buf, n, 0, peek);
if (svm_fifo_is_empty (rx_fifo))
svm_fifo_unset_event (rx_fifo);
if (is_ct && svm_fifo_want_tx_evt (rx_fifo))
{
svm_fifo_set_want_tx_evt (s->rx_fifo, 0);
app_send_io_evt_to_vpp (s->vpp_evt_q, s->rx_fifo, SESSION_IO_EVT_CT_RX,
SVM_Q_WAIT);
}
VDBG (2, "vpp handle 0x%llx, sid %u: read %d bytes from (%p)",
s->vpp_handle, session_handle, n_read, rx_fifo);
return n_read;
}
int
vppcom_session_read (uint32_t session_handle, void *buf, size_t n)
{
return (vppcom_session_read_internal (session_handle, buf, n, 0));
}
static int
vppcom_session_peek (uint32_t session_handle, void *buf, int n)
{
return (vppcom_session_read_internal (session_handle, buf, n, 1));
}
int
vppcom_session_read_segments (uint32_t session_handle,
vppcom_data_segments_t ds)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
int n_read = 0, rv, is_nonblocking;
vcl_session_t *s = 0;
svm_fifo_t *rx_fifo;
svm_msg_q_msg_t msg;
session_event_t *e;
svm_msg_q_t *mq;
u8 is_ct;
s = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!s || s->is_vep))
return VPPCOM_EBADFD;
if (PREDICT_FALSE (!vcl_session_is_readable (s)))
{
session_state_t state = s->session_state;
rv = ((state & STATE_DISCONNECT) ? VPPCOM_ECONNRESET : VPPCOM_ENOTCONN);
return rv;
}
is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
is_ct = vcl_session_is_ct (s);
mq = is_ct ? s->our_evt_q : wrk->app_event_queue;
rx_fifo = s->rx_fifo;
s->has_rx_evt = 0;
if (svm_fifo_is_empty (rx_fifo))
{
if (is_nonblocking)
{
svm_fifo_unset_event (rx_fifo);
return VPPCOM_EWOULDBLOCK;
}
while (svm_fifo_is_empty (rx_fifo))
{
svm_fifo_unset_event (rx_fifo);
svm_msg_q_lock (mq);
if (svm_msg_q_is_empty (mq))
svm_msg_q_wait (mq);
svm_msg_q_sub_w_lock (mq, &msg);
e = svm_msg_q_msg_data (mq, &msg);
svm_msg_q_unlock (mq);
if (!vcl_is_rx_evt_for_session (e, s->session_index, is_ct))
vcl_handle_mq_event (wrk, e);
svm_msg_q_free_msg (mq, &msg);
if (PREDICT_FALSE (s->session_state == STATE_DISCONNECT))
return VPPCOM_ECONNRESET;
}
}
n_read = svm_fifo_segments (rx_fifo, (svm_fifo_segment_t *) ds);
svm_fifo_unset_event (rx_fifo);
if (is_ct && n_read + svm_fifo_max_dequeue (rx_fifo) == rx_fifo->nitems)
{
/* If the peer is not polling send notification */
if (!svm_fifo_has_event (s->rx_fifo))
app_send_io_evt_to_vpp (s->vpp_evt_q, s->rx_fifo,
SESSION_IO_EVT_CT_RX, SVM_Q_WAIT);
}
return n_read;
}
void
vppcom_session_free_segments (uint32_t session_handle,
vppcom_data_segments_t ds)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *s;
s = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!s || s->is_vep))
return;
svm_fifo_segments_free (s->rx_fifo, (svm_fifo_segment_t *) ds);
}
static inline int
vppcom_session_read_ready (vcl_session_t * session)
{
/* Assumes caller has acquired spinlock: vcm->sessions_lockp */
if (PREDICT_FALSE (session->is_vep))
{
clib_warning ("VCL<%d>: ERROR: sid %u: cannot read from an "
"epoll session!", getpid (), session->session_index);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (!(session->session_state & (STATE_OPEN | STATE_LISTEN))))
{
session_state_t state = session->session_state;
int rv;
rv = ((state & STATE_DISCONNECT) ? VPPCOM_ECONNRESET : VPPCOM_ENOTCONN);
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: session is not open!"
" state 0x%x (%s), returning %d (%s)", getpid (),
session->vpp_handle, session->session_index, state,
vppcom_session_state_str (state), rv, vppcom_retval_str (rv));
return rv;
}
if (session->session_state & STATE_LISTEN)
return clib_fifo_elts (session->accept_evts_fifo);
return svm_fifo_max_dequeue (session->rx_fifo);
}
int
vppcom_data_segment_copy (void *buf, vppcom_data_segments_t ds, u32 max_bytes)
{
u32 first_copy = clib_min (ds[0].len, max_bytes);
clib_memcpy_fast (buf, ds[0].data, first_copy);
if (first_copy < max_bytes)
{
clib_memcpy_fast (buf + first_copy, ds[1].data,
clib_min (ds[1].len, max_bytes - first_copy));
}
return 0;
}
static u8
vcl_is_tx_evt_for_session (session_event_t * e, u32 sid, u8 is_ct)
{
if (!is_ct)
return (e->event_type == FIFO_EVENT_APP_TX
&& e->fifo->client_session_index == sid);
else
return (e->event_type == SESSION_IO_EVT_CT_RX);
}
static inline int
vppcom_session_write_inline (uint32_t session_handle, void *buf, size_t n,
u8 is_flush)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
int rv, n_write, is_nonblocking;
vcl_session_t *s = 0;
svm_fifo_t *tx_fifo = 0;
session_evt_type_t et;
svm_msg_q_msg_t msg;
session_event_t *e;
svm_msg_q_t *mq;
u8 is_ct;
if (PREDICT_FALSE (!buf))
return VPPCOM_EINVAL;
s = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!s))
return VPPCOM_EBADFD;
if (PREDICT_FALSE (s->is_vep))
{
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"cannot write to an epoll session!",
getpid (), s->vpp_handle, session_handle);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (!(s->session_state & STATE_OPEN)))
{
session_state_t state = s->session_state;
rv = ((state & STATE_DISCONNECT) ? VPPCOM_ECONNRESET : VPPCOM_ENOTCONN);
VDBG (1, "VCL<%d>: vpp handle 0x%llx, sid %u: session is not open! "
"state 0x%x (%s)", getpid (), s->vpp_handle, session_handle,
state, vppcom_session_state_str (state));
return rv;
}
tx_fifo = s->tx_fifo;
is_ct = vcl_session_is_ct (s);
is_nonblocking = VCL_SESS_ATTR_TEST (s->attr, VCL_SESS_ATTR_NONBLOCK);
mq = is_ct ? s->our_evt_q : wrk->app_event_queue;
if (svm_fifo_is_full (tx_fifo))
{
if (is_nonblocking)
{
return VPPCOM_EWOULDBLOCK;
}
while (svm_fifo_is_full (tx_fifo))
{
svm_fifo_set_want_tx_evt (tx_fifo, 1);
svm_msg_q_lock (mq);
if (svm_msg_q_is_empty (mq))
svm_msg_q_wait (mq);
svm_msg_q_sub_w_lock (mq, &msg);
e = svm_msg_q_msg_data (mq, &msg);
svm_msg_q_unlock (mq);
if (!vcl_is_tx_evt_for_session (e, s->session_index, is_ct))
vcl_handle_mq_event (wrk, e);
svm_msg_q_free_msg (mq, &msg);
if (PREDICT_FALSE (!(s->session_state & STATE_OPEN)))
return VPPCOM_ECONNRESET;
}
}
ASSERT (FIFO_EVENT_APP_TX + 1 == SESSION_IO_EVT_CT_TX);
et = FIFO_EVENT_APP_TX + vcl_session_is_ct (s);
if (is_flush && !vcl_session_is_ct (s))
et = SESSION_IO_EVT_TX_FLUSH;
if (s->is_dgram)
n_write = app_send_dgram_raw (tx_fifo, &s->transport,
s->vpp_evt_q, buf, n, et, SVM_Q_WAIT);
else
n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
SVM_Q_WAIT);
ASSERT (n_write > 0);
VDBG (2, "VCL<%d>: vpp handle 0x%llx, sid %u: wrote %d bytes", getpid (),
s->vpp_handle, session_handle, n_write);
return n_write;
}
int
vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
{
return vppcom_session_write_inline (session_handle, buf, n,
0 /* is_flush */ );
}
int
vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
{
return vppcom_session_write_inline (session_handle, buf, n,
1 /* is_flush */ );
}
static vcl_session_t *
vcl_ct_session_get_from_fifo (vcl_worker_t * wrk, svm_fifo_t * f, u8 type)
{
vcl_session_t *s;
s = vcl_session_get (wrk, f->client_session_index);
if (s)
{
/* rx fifo */
if (type == 0 && s->rx_fifo == f)
return s;
/* tx fifo */
if (type == 1 && s->tx_fifo == f)
return s;
}
s = vcl_session_get (wrk, f->master_session_index);
if (s)
{
if (type == 0 && s->rx_fifo == f)
return s;
if (type == 1 && s->tx_fifo == f)
return s;
}
return 0;
}
static inline int
vppcom_session_write_ready (vcl_session_t * session)
{
/* Assumes caller has acquired spinlock: vcm->sessions_lockp */
if (PREDICT_FALSE (session->is_vep))
{
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"cannot write to an epoll session!",
getpid (), session->vpp_handle, session->session_index);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (session->session_state & STATE_LISTEN))
{
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"cannot write to a listen session!",
getpid (), session->vpp_handle, session->session_index);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (!(session->session_state & STATE_OPEN)))
{
session_state_t state = session->session_state;
int rv;
rv = ((state & STATE_DISCONNECT) ? VPPCOM_ECONNRESET : VPPCOM_ENOTCONN);
clib_warning ("VCL<%d>: ERROR: vpp handle 0x%llx, sid %u: "
"session is not open! state 0x%x (%s), "
"returning %d (%s)", getpid (), session->vpp_handle,
session->session_index,
state, vppcom_session_state_str (state),
rv, vppcom_retval_str (rv));
return rv;
}
VDBG (3, "VCL<%d>: vpp handle 0x%llx, sid %u: peek %s (%p), ready = %d",
getpid (), session->vpp_handle, session->session_index,
session->tx_fifo, svm_fifo_max_enqueue (session->tx_fifo));
return svm_fifo_max_enqueue (session->tx_fifo);
}
static inline int
vcl_mq_dequeue_batch (vcl_worker_t * wrk, svm_msg_q_t * mq)
{
svm_msg_q_msg_t *msg;
u32 n_msgs;
int i;
n_msgs = svm_msg_q_size (mq);
for (i = 0; i < n_msgs; i++)
{
vec_add2 (wrk->mq_msg_vector, msg, 1);
svm_msg_q_sub_w_lock (mq, msg);
}
return n_msgs;
}
#define vcl_fifo_rx_evt_valid_or_break(_fifo) \
if (PREDICT_FALSE (svm_fifo_is_empty (_fifo))) \
{ \
svm_fifo_unset_event (_fifo); \
if (svm_fifo_is_empty (_fifo)) \
break; \
} \
static void
vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
unsigned long n_bits, unsigned long *read_map,
unsigned long *write_map,
unsigned long *except_map, u32 * bits_set)
{
session_disconnected_msg_t *disconnected_msg;
session_connected_msg_t *connected_msg;
vcl_session_t *session;
u32 sid;
switch (e->event_type)
{
case FIFO_EVENT_APP_RX:
vcl_fifo_rx_evt_valid_or_break (e->fifo);
sid = e->fifo->client_session_index;
session = vcl_session_get (wrk, sid);
if (!session)
break;
if (sid < n_bits && read_map)
{
clib_bitmap_set_no_check (read_map, sid, 1);
*bits_set += 1;
}
break;
case FIFO_EVENT_APP_TX:
sid = e->fifo->client_session_index;
session = vcl_session_get (wrk, sid);
if (!session)
break;
if (sid < n_bits && write_map)
{
clib_bitmap_set_no_check (write_map, sid, 1);
*bits_set += 1;
}
break;
case SESSION_IO_EVT_CT_TX:
vcl_fifo_rx_evt_valid_or_break (e->fifo);
session = vcl_ct_session_get_from_fifo (wrk, e->fifo, 0);
if (!session)
break;
sid = session->session_index;
if (sid < n_bits && read_map)
{
clib_bitmap_set_no_check (read_map, sid, 1);
*bits_set += 1;
}
break;
case SESSION_IO_EVT_CT_RX:
session = vcl_ct_session_get_from_fifo (wrk, e->fifo, 1);
if (!session)
break;
sid = session->session_index;
if (sid < n_bits && write_map)
{
clib_bitmap_set_no_check (write_map, sid, 1);
*bits_set += 1;
}
break;
case SESSION_CTRL_EVT_ACCEPTED:
session = vcl_session_accepted (wrk,
(session_accepted_msg_t *) e->data);
if (!session)
break;
sid = session->session_index;
if (sid < n_bits && read_map)
{
clib_bitmap_set_no_check (read_map, sid, 1);
*bits_set += 1;
}
break;
case SESSION_CTRL_EVT_CONNECTED:
connected_msg = (session_connected_msg_t *) e->data;
vcl_session_connected_handler (wrk, connected_msg);
break;
case SESSION_CTRL_EVT_DISCONNECTED:
disconnected_msg = (session_disconnected_msg_t *) e->data;
session = vcl_session_disconnected_handler (wrk, disconnected_msg);
if (!session)
break;
sid = session->session_index;
if (sid < n_bits && except_map)
{
clib_bitmap_set_no_check (except_map, sid, 1);
*bits_set += 1;
}
break;
case SESSION_CTRL_EVT_RESET:
sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
if (sid < n_bits && except_map)
{
clib_bitmap_set_no_check (except_map, sid, 1);
*bits_set += 1;
}
break;
default:
clib_warning ("unhandled: %u", e->event_type);
break;
}
}
static int
vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
unsigned long n_bits, unsigned long *read_map,
unsigned long *write_map, unsigned long *except_map,
double time_to_wait, u32 * bits_set)
{
svm_msg_q_msg_t *msg;
session_event_t *e;
u32 i;
svm_msg_q_lock (mq);
if (svm_msg_q_is_empty (mq))
{
if (*bits_set)
{
svm_msg_q_unlock (mq);
return 0;
}
if (!time_to_wait)
{
svm_msg_q_unlock (mq);
return 0;
}
else if (time_to_wait < 0)
{
svm_msg_q_wait (mq);
}
else
{
if (svm_msg_q_timedwait (mq, time_to_wait))
{
svm_msg_q_unlock (mq);
return 0;
}
}
}
vcl_mq_dequeue_batch (wrk, mq);
svm_msg_q_unlock (mq);
for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
{
msg = vec_elt_at_index (wrk->mq_msg_vector, i);
e = svm_msg_q_msg_data (mq, msg);
vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
except_map, bits_set);
svm_msg_q_free_msg (mq, msg);
}
vec_reset_length (wrk->mq_msg_vector);
return *bits_set;
}
static int
vppcom_select_condvar (vcl_worker_t * wrk, unsigned long n_bits,
unsigned long *read_map, unsigned long *write_map,
unsigned long *except_map, double time_to_wait,
u32 * bits_set)
{
double total_wait = 0, wait_slice;
vcl_cut_through_registration_t *cr;
time_to_wait = (time_to_wait == -1) ? 10e9 : time_to_wait;
wait_slice = wrk->cut_through_registrations ? 10e-6 : time_to_wait;
do
{
vcl_ct_registration_lock (wrk);
/* *INDENT-OFF* */
pool_foreach (cr, wrk->cut_through_registrations, ({
vcl_select_handle_mq (wrk, cr->mq, n_bits, read_map, write_map, except_map,
0, bits_set);
}));
/* *INDENT-ON* */
vcl_ct_registration_unlock (wrk);
vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
write_map, except_map, time_to_wait, bits_set);
total_wait += wait_slice;
if (*bits_set)
return *bits_set;
}
while (total_wait < time_to_wait);
return 0;
}
static int
vppcom_select_eventfd (vcl_worker_t * wrk, unsigned long n_bits,
unsigned long *read_map, unsigned long *write_map,
unsigned long *except_map, double time_to_wait,
u32 * bits_set)
{
vcl_mq_evt_conn_t *mqc;
int __clib_unused n_read;
int n_mq_evts, i;
u64 buf;
vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
vec_len (wrk->mq_events), time_to_wait);
for (i = 0; i < n_mq_evts; i++)
{
mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
n_read = read (mqc->mq_fd, &buf, sizeof (buf));
vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
except_map, 0, bits_set);
}
return (n_mq_evts > 0 ? (int) *bits_set : 0);
}
int
vppcom_select (unsigned long n_bits, unsigned long *read_map,
unsigned long *write_map, unsigned long *except_map,
double time_to_wait)
{
u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session = 0;
int rv, i;
ASSERT (sizeof (clib_bitmap_t) == sizeof (long int));
if (n_bits && read_map)
{
clib_bitmap_validate (wrk->rd_bitmap, minbits);
clib_memcpy_fast (wrk->rd_bitmap, read_map,
vec_len (wrk->rd_bitmap) * sizeof (clib_bitmap_t));
memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (clib_bitmap_t));
}
if (n_bits && write_map)
{
clib_bitmap_validate (wrk->wr_bitmap, minbits);
clib_memcpy_fast (wrk->wr_bitmap, write_map,
vec_len (wrk->wr_bitmap) * sizeof (clib_bitmap_t));
memset (write_map, 0,
vec_len (wrk->wr_bitmap) * sizeof (clib_bitmap_t));
}
if (n_bits && except_map)
{
clib_bitmap_validate (wrk->ex_bitmap, minbits);
clib_memcpy_fast (wrk->ex_bitmap, except_map,
vec_len (wrk->ex_bitmap) * sizeof (clib_bitmap_t));
memset (except_map, 0,
vec_len (wrk->ex_bitmap) * sizeof (clib_bitmap_t));
}
if (!n_bits)
return 0;
if (!write_map)
goto check_rd;
/* *INDENT-OFF* */
clib_bitmap_foreach (sid, wrk->wr_bitmap, ({
if (!(session = vcl_session_get (wrk, sid)))
{
if (except_map && sid < minbits)
clib_bitmap_set_no_check (except_map, sid, 1);
continue;
}
rv = svm_fifo_is_full (session->tx_fifo);
if (!rv)
{
clib_bitmap_set_no_check (write_map, sid, 1);
bits_set++;
}
}));
check_rd:
if (!read_map)
goto check_mq;
clib_bitmap_foreach (sid, wrk->rd_bitmap, ({
if (!(session = vcl_session_get (wrk, sid)))
{
if (except_map && sid < minbits)
clib_bitmap_set_no_check (except_map, sid, 1);
continue;
}
rv = vppcom_session_read_ready (session);
if (rv)
{
clib_bitmap_set_no_check (read_map, sid, 1);
bits_set++;
}
}));
/* *INDENT-ON* */
check_mq:
for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
{
vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
read_map, write_map, except_map, &bits_set);
}
vec_reset_length (wrk->unhandled_evts_vector);
if (vcm->cfg.use_mq_eventfd)
vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
time_to_wait, &bits_set);
else
vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
time_to_wait, &bits_set);
return (bits_set);
}
static inline void
vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_idx)
{
vcl_session_t *session;
vppcom_epoll_t *vep;
u32 sid = vep_idx;
if (VPPCOM_DEBUG <= 1)
return;
/* Assumes caller has acquired spinlock: vcm->sessions_lockp */
session = vcl_session_get (wrk, vep_idx);
if (PREDICT_FALSE (!session))
{
clib_warning ("VCL<%d>: ERROR: Invalid vep_idx (%u)!",
getpid (), vep_idx);
goto done;
}
if (PREDICT_FALSE (!session->is_vep))
{
clib_warning ("VCL<%d>: ERROR: vep_idx (%u) is not a vep!",
getpid (), vep_idx);
goto done;
}
vep = &session->vep;
clib_warning ("VCL<%d>: vep_idx (%u): Dumping epoll chain\n"
"{\n"
" is_vep = %u\n"
" is_vep_session = %u\n"
" next_sid = 0x%x (%u)\n"
" wait_cont_idx = 0x%x (%u)\n"
"}\n", getpid (), vep_idx,
session->is_vep, session->is_vep_session,
vep->next_sh, vep->next_sh,
session->wait_cont_idx, session->wait_cont_idx);
for (sid = vep->next_sh; sid != ~0; sid = vep->next_sh)
{
session = vcl_session_get (wrk, sid);
if (PREDICT_FALSE (!session))
{
clib_warning ("VCL<%d>: ERROR: Invalid sid (%u)!", getpid (), sid);
goto done;
}
if (PREDICT_FALSE (session->is_vep))
clib_warning ("VCL<%d>: ERROR: sid (%u) is a vep!",
getpid (), vep_idx);
else if (PREDICT_FALSE (!session->is_vep_session))
{
clib_warning ("VCL<%d>: ERROR: session (%u) "
"is not a vep session!", getpid (), sid);
goto done;
}
vep = &session->vep;
if (PREDICT_FALSE (vep->vep_sh != vep_idx))
clib_warning ("VCL<%d>: ERROR: session (%u) vep_idx (%u) != "
"vep_idx (%u)!", getpid (),
sid, session->vep.vep_sh, vep_idx);
if (session->is_vep_session)
{
clib_warning ("vep_idx[%u]: sid 0x%x (%u)\n"
"{\n"
" next_sid = 0x%x (%u)\n"
" prev_sid = 0x%x (%u)\n"
" vep_idx = 0x%x (%u)\n"
" ev.events = 0x%x\n"
" ev.data.u64 = 0x%llx\n"
" et_mask = 0x%x\n"
"}\n",
vep_idx, sid, sid,
vep->next_sh, vep->next_sh,
vep->prev_sh, vep->prev_sh,
vep->vep_sh, vep->vep_sh,
vep->ev.events, vep->ev.data.u64, vep->et_mask);
}
}
done:
clib_warning ("VCL<%d>: vep_idx (%u): Dump complete!\n",
getpid (), vep_idx);
}
int
vppcom_epoll_create (void)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *vep_session;
vep_session = vcl_session_alloc (wrk);
vep_session->is_vep = 1;
vep_session->vep.vep_sh = ~0;
vep_session->vep.next_sh = ~0;
vep_session->vep.prev_sh = ~0;
vep_session->wait_cont_idx = ~0;
vep_session->vpp_handle = ~0;
vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
VDBG (0, "Created vep_idx %u", vep_session->session_index);
return vcl_session_handle (vep_session);
}
int
vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
struct epoll_event *event)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *vep_session;
vcl_session_t *session;
int rv = VPPCOM_OK;
if (vep_handle == session_handle)
{
VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
return VPPCOM_EINVAL;
}
vep_session = vcl_session_get_w_handle (wrk, vep_handle);
if (PREDICT_FALSE (!vep_session))
{
VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (!vep_session->is_vep))
{
VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
return VPPCOM_EINVAL;
}
ASSERT (vep_session->vep.vep_sh == ~0);
ASSERT (vep_session->vep.prev_sh == ~0);
session = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!session))
{
VDBG (0, "Invalid session_handle (%u)!", session_handle);
return VPPCOM_EBADFD;
}
if (PREDICT_FALSE (session->is_vep))
{
VDBG (0, "session_handle (%u) is a vep!", vep_handle);
return VPPCOM_EINVAL;
}
switch (op)
{
case EPOLL_CTL_ADD:
if (PREDICT_FALSE (!event))
{
VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
return VPPCOM_EINVAL;
}
if (vep_session->vep.next_sh != ~0)
{
vcl_session_t *next_session;
next_session = vcl_session_get_w_handle (wrk,
vep_session->vep.next_sh);
if (PREDICT_FALSE (!next_session))
{
VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sid (%u) on "
"vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
return VPPCOM_EBADFD;
}
ASSERT (next_session->vep.prev_sh == vep_handle);
next_session->vep.prev_sh = session_handle;
}
session->vep.next_sh = vep_session->vep.next_sh;
session->vep.prev_sh = vep_handle;
session->vep.vep_sh = vep_handle;
session->vep.et_mask = VEP_DEFAULT_ET_MASK;
session->vep.ev = *event;
session->is_vep = 0;
session->is_vep_session = 1;
vep_session->vep.next_sh = session_handle;
VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
vep_handle, session_handle, event->events, event->data.u64);
vcl_evt (VCL_EVT_EPOLL_CTLADD, session, event->events, event->data.u64);
break;
case EPOLL_CTL_MOD:
if (PREDICT_FALSE (!event))
{
VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
rv = VPPCOM_EINVAL;
goto done;
}
else if (PREDICT_FALSE (!session->is_vep_session))
{
VDBG (0, "sid %u EPOLL_CTL_MOD: not a vep session!",
session_handle);
rv = VPPCOM_EINVAL;
goto done;
}
else if (PREDICT_FALSE (session->vep.vep_sh != vep_handle))
{
VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
session_handle, session->vep.vep_sh, vep_handle);
rv = VPPCOM_EINVAL;
goto done;
}
session->vep.et_mask = VEP_DEFAULT_ET_MASK;
session->vep.ev = *event;
VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
vep_handle, session_handle, event->events, event->data.u64);
break;
case EPOLL_CTL_DEL:
if (PREDICT_FALSE (!session->is_vep_session))
{
VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
rv = VPPCOM_EINVAL;
goto done;
}
else if (PREDICT_FALSE (session->vep.vep_sh != vep_handle))
{
VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
session_handle, session->vep.vep_sh, vep_handle);
rv = VPPCOM_EINVAL;
goto done;
}
vep_session->wait_cont_idx =
(vep_session->wait_cont_idx == session_handle) ?
session->vep.next_sh : vep_session->wait_cont_idx;
if (session->vep.prev_sh == vep_handle)
vep_session->vep.next_sh = session->vep.next_sh;
else
{
vcl_session_t *prev_session;
prev_session = vcl_session_get_w_handle (wrk, session->vep.prev_sh);
if (PREDICT_FALSE (!prev_session))
{
VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sid (%u) on sid (%u)!",
session->vep.prev_sh, session_handle);
return VPPCOM_EBADFD;
}
ASSERT (prev_session->vep.next_sh == session_handle);
prev_session->vep.next_sh = session->vep.next_sh;
}
if (session->vep.next_sh != ~0)
{
vcl_session_t *next_session;
next_session = vcl_session_get_w_handle (wrk, session->vep.next_sh);
if (PREDICT_FALSE (!next_session))
{
VDBG (0, "EPOLL_CTL_DEL: Invalid next_sid (%u) on sid (%u)!",
session->vep.next_sh, session_handle);
return VPPCOM_EBADFD;
}
ASSERT (next_session->vep.prev_sh == session_handle);
next_session->vep.prev_sh = session->vep.prev_sh;
}
memset (&session->vep, 0, sizeof (session->vep));
session->vep.next_sh = ~0;
session->vep.prev_sh = ~0;
session->vep.vep_sh = ~0;
session->is_vep_session = 0;
VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sid %u!", vep_handle,
session_handle);
vcl_evt (VCL_EVT_EPOLL_CTLDEL, session, vep_sh);
break;
default:
VDBG (0, "Invalid operation (%d)!", op);
rv = VPPCOM_EINVAL;
}
vep_verify_epoll_chain (wrk, vep_handle);
done:
return rv;
}
static inline void
vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
struct epoll_event *events, u32 * num_ev)
{
session_disconnected_msg_t *disconnected_msg;
session_connected_msg_t *connected_msg;
u32 sid = ~0, session_events;
u64 session_evt_data = ~0;
vcl_session_t *session;
u8 add_event = 0;
switch (e->event_type)
{
case FIFO_EVENT_APP_RX:
ASSERT (e->fifo->client_thread_index == vcl_get_worker_index ());
vcl_fifo_rx_evt_valid_or_break (e->fifo);
sid = e->fifo->client_session_index;
if (!(session = vcl_session_get (wrk, sid)))
break;
session_events = session->vep.ev.events;
if (!(EPOLLIN & session->vep.ev.events) || session->has_rx_evt)
break;
add_event = 1;
events[*num_ev].events |= EPOLLIN;
session_evt_data = session->vep.ev.data.u64;
session->has_rx_evt = 1;
break;
case FIFO_EVENT_APP_TX:
sid = e->fifo->client_session_index;
if (!(session = vcl_session_get (wrk, sid)))
break;
session_events = session->vep.ev.events;
if (!(EPOLLOUT & session_events))
break;
add_event = 1;
events[*num_ev].events |= EPOLLOUT;
session_evt_data = session->vep.ev.data.u64;
break;
case SESSION_IO_EVT_CT_TX:
vcl_fifo_rx_evt_valid_or_break (e->fifo);
session = vcl_ct_session_get_from_fifo (wrk, e->fifo, 0);
if (PREDICT_FALSE (!session))
break;
sid = session->session_index;
session_events = session->vep.ev.events;
if (!(EPOLLIN & session->vep.ev.events) || session->has_rx_evt)
break;
add_event = 1;
events[*num_ev].events |= EPOLLIN;
session_evt_data = session->vep.ev.data.u64;
session->has_rx_evt = 1;
break;
case SESSION_IO_EVT_CT_RX:
session = vcl_ct_session_get_from_fifo (wrk, e->fifo, 1);
if (PREDICT_FALSE (!session))
break;
sid = session->session_index;
session_events = session->vep.ev.events;
if (!(EPOLLOUT & session_events))
break;
add_event = 1;
events[*num_ev].events |= EPOLLOUT;
session_evt_data = session->vep.ev.data.u64;
break;
case SESSION_CTRL_EVT_ACCEPTED:
session = vcl_session_accepted (wrk,
(session_accepted_msg_t *) e->data);
if (!session)
break;
session_events = session->vep.ev.events;
if (!(EPOLLIN & session_events))
break;
add_event = 1;
events[*num_ev].events |= EPOLLIN;
session_evt_data = session->vep.ev.data.u64;
break;
case SESSION_CTRL_EVT_CONNECTED:
connected_msg = (session_connected_msg_t *) e->data;
vcl_session_connected_handler (wrk, connected_msg);
/* Generate EPOLLOUT because there's no connected event */
sid = vcl_session_index_from_vpp_handle (wrk, connected_msg->handle);
if (!(session = vcl_session_get (wrk, sid)))
break;
session_events = session->vep.ev.events;
if (EPOLLOUT & session_events)
{
add_event = 1;
events[*num_ev].events |= EPOLLOUT;
session_evt_data = session->vep.ev.data.u64;
}
break;
case SESSION_CTRL_EVT_DISCONNECTED:
disconnected_msg = (session_disconnected_msg_t *) e->data;
session = vcl_session_disconnected_handler (wrk, disconnected_msg);
if (!session)
break;
add_event = 1;
events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
session_evt_data = session->vep.ev.data.u64;
session_events = session->vep.ev.events;
break;
case SESSION_CTRL_EVT_RESET:
sid = vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
if (!(session = vcl_session_get (wrk, sid)))
break;
add_event = 1;
events[*num_ev].events |= EPOLLHUP | EPOLLRDHUP;
session_evt_data = session->vep.ev.data.u64;
session_events = session->vep.ev.events;
break;
default:
VDBG (0, "unhandled: %u", e->event_type);
break;
}
if (add_event)
{
events[*num_ev].data.u64 = session_evt_data;
if (EPOLLONESHOT & session_events)
{
session = vcl_session_get (wrk, sid);
session->vep.ev.events = 0;
}
*num_ev += 1;
}
}
static int
vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
struct epoll_event *events, u32 maxevents,
double wait_for_time, u32 * num_ev)
{
svm_msg_q_msg_t *msg;
session_event_t *e;
int i;
if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
goto handle_dequeued;
svm_msg_q_lock (mq);
if (svm_msg_q_is_empty (mq))
{
if (!wait_for_time)
{
svm_msg_q_unlock (mq);
return 0;
}
else if (wait_for_time < 0)
{
svm_msg_q_wait (mq);
}
else
{
if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
{
svm_msg_q_unlock (mq);
return 0;
}
}
}
vcl_mq_dequeue_batch (wrk, mq);
svm_msg_q_unlock (mq);
handle_dequeued:
for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
{
msg = vec_elt_at_index (wrk->mq_msg_vector, i);
e = svm_msg_q_msg_data (mq, msg);
if (*num_ev < maxevents)
vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
else
vec_add1 (wrk->unhandled_evts_vector, *e);
svm_msg_q_free_msg (mq, msg);
}
vec_reset_length (wrk->mq_msg_vector);
return *num_ev;
}
static int
vppcom_epoll_wait_condvar (vcl_worker_t * wrk, struct epoll_event *events,
int maxevents, u32 n_evts, double wait_for_time)
{
vcl_cut_through_registration_t *cr;
double total_wait = 0, wait_slice;
int rv;
wait_for_time = (wait_for_time == -1) ? (double) 10e9 : wait_for_time;
wait_slice = wrk->cut_through_registrations ? 10e-6 : wait_for_time;
do
{
vcl_ct_registration_lock (wrk);
/* *INDENT-OFF* */
pool_foreach (cr, wrk->cut_through_registrations, ({
vcl_epoll_wait_handle_mq (wrk, cr->mq, events, maxevents, 0, &n_evts);
}));
/* *INDENT-ON* */
vcl_ct_registration_unlock (wrk);
rv = vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events,
maxevents, n_evts ? 0 : wait_slice,
&n_evts);
if (rv)
total_wait += wait_slice;
if (n_evts)
return n_evts;
}
while (total_wait < wait_for_time);
return n_evts;
}
static int
vppcom_epoll_wait_eventfd (vcl_worker_t * wrk, struct epoll_event *events,
int maxevents, u32 n_evts, double wait_for_time)
{
vcl_mq_evt_conn_t *mqc;
int __clib_unused n_read;
int n_mq_evts, i;
u64 buf;
vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
again:
n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
vec_len (wrk->mq_events), wait_for_time);
for (i = 0; i < n_mq_evts; i++)
{
mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
n_read = read (mqc->mq_fd, &buf, sizeof (buf));
vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0, &n_evts);
}
if (!n_evts && n_mq_evts > 0)
goto again;
return (int) n_evts;
}
int
vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
int maxevents, double wait_for_time)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *vep_session;
u32 n_evts = 0;
int i;
if (PREDICT_FALSE (maxevents <= 0))
{
clib_warning ("VCL<%d>: ERROR: Invalid maxevents (%d)!",
getpid (), maxevents);
return VPPCOM_EINVAL;
}
vep_session = vcl_session_get_w_handle (wrk, vep_handle);
if (!vep_session)
return VPPCOM_EBADFD;
if (PREDICT_FALSE (!vep_session->is_vep))
{
clib_warning ("VCL<%d>: ERROR: vep_idx (%u) is not a vep!",
getpid (), vep_handle);
return VPPCOM_EINVAL;
}
memset (events, 0, sizeof (*events) * maxevents);
if (vec_len (wrk->unhandled_evts_vector))
{
for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
{
vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
events, &n_evts);
if (n_evts == maxevents)
{
i += 1;
break;
}
}
vec_delete (wrk->unhandled_evts_vector, i, 0);
}
if (vcm->cfg.use_mq_eventfd)
return vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
wait_for_time);
return vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
wait_for_time);
}
int
vppcom_session_attr (uint32_t session_handle, uint32_t op,
void *buffer, uint32_t * buflen)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
vcl_session_t *session;
int rv = VPPCOM_OK;
u32 *flags = buffer;
vppcom_endpt_t *ep = buffer;
session = vcl_session_get_w_handle (wrk, session_handle);
if (!session)
return VPPCOM_EBADFD;
switch (op)
{
case VPPCOM_ATTR_GET_NREAD:
rv = vppcom_session_read_ready (session);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_NREAD: sid %u, nread = %d",
getpid (), rv);
break;
case VPPCOM_ATTR_GET_NWRITE:
rv = vppcom_session_write_ready (session);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_NWRITE: sid %u, nwrite = %d",
getpid (), session_handle, rv);
break;
case VPPCOM_ATTR_GET_FLAGS:
if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
{
*flags = O_RDWR | (VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_NONBLOCK));
*buflen = sizeof (*flags);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_FLAGS: sid %u, flags = 0x%08x, "
"is_nonblocking = %u", getpid (),
session_handle, *flags,
VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_FLAGS:
if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
{
if (*flags & O_NONBLOCK)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_NONBLOCK);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_NONBLOCK);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_FLAGS: sid %u, flags = 0x%08x,"
" is_nonblocking = %u",
getpid (), session_handle, *flags,
VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_NONBLOCK));
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_PEER_ADDR:
if (PREDICT_TRUE (buffer && buflen &&
(*buflen >= sizeof (*ep)) && ep->ip))
{
ep->is_ip4 = session->transport.is_ip4;
ep->port = session->transport.rmt_port;
if (session->transport.is_ip4)
clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
sizeof (ip4_address_t));
else
clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
sizeof (ip6_address_t));
*buflen = sizeof (*ep);
VDBG (1, "VCL<%d>: VPPCOM_ATTR_GET_PEER_ADDR: sid %u, is_ip4 = %u, "
"addr = %U, port %u", getpid (),
session_handle, ep->is_ip4, format_ip46_address,
&session->transport.rmt_ip,
ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (ep->port));
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_LCL_ADDR:
if (PREDICT_TRUE (buffer && buflen &&
(*buflen >= sizeof (*ep)) && ep->ip))
{
ep->is_ip4 = session->transport.is_ip4;
ep->port = session->transport.lcl_port;
if (session->transport.is_ip4)
clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
sizeof (ip4_address_t));
else
clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
sizeof (ip6_address_t));
*buflen = sizeof (*ep);
VDBG (1, "VCL<%d>: VPPCOM_ATTR_GET_LCL_ADDR: sid %u, is_ip4 = %u,"
" addr = %U port %d", getpid (),
session_handle, ep->is_ip4, format_ip46_address,
&session->transport.lcl_ip,
ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
clib_net_to_host_u16 (ep->port));
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_LIBC_EPFD:
rv = session->libc_epfd;
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d",
getpid (), rv);
break;
case VPPCOM_ATTR_SET_LIBC_EPFD:
if (PREDICT_TRUE (buffer && buflen &&
(*buflen == sizeof (session->libc_epfd))))
{
session->libc_epfd = *(int *) buffer;
*buflen = sizeof (session->libc_epfd);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, "
"buflen %d", getpid (), session->libc_epfd, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_PROTOCOL:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
*(int *) buffer = session->session_type;
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
getpid (), *(int *) buffer, *(int *) buffer ? "UDP" : "TCP",
*buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_LISTEN:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_LISTEN);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_LISTEN: %d, buflen %d",
getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_ERROR:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
*(int *) buffer = 0;
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_TX_FIFO_LEN:
if (buffer && buflen && (*buflen >= sizeof (u32)))
{
/* VPP-TBD */
*(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
session->tx_fifo ? session->tx_fifo->nitems :
vcm->cfg.tx_fifo_size);
*buflen = sizeof (u32);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), "
"buflen %d, #VPP-TBD#", getpid (),
*(size_t *) buffer, *(size_t *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_TX_FIFO_LEN:
if (buffer && buflen && (*buflen == sizeof (u32)))
{
/* VPP-TBD */
session->sndbuf_size = *(u32 *) buffer;
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), "
"buflen %d, #VPP-TBD#", getpid (),
session->sndbuf_size, session->sndbuf_size, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_RX_FIFO_LEN:
if (buffer && buflen && (*buflen >= sizeof (u32)))
{
/* VPP-TBD */
*(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
session->rx_fifo ? session->rx_fifo->nitems :
vcm->cfg.rx_fifo_size);
*buflen = sizeof (u32);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), "
"buflen %d, #VPP-TBD#", getpid (),
*(size_t *) buffer, *(size_t *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_RX_FIFO_LEN:
if (buffer && buflen && (*buflen == sizeof (u32)))
{
/* VPP-TBD */
session->rcvbuf_size = *(u32 *) buffer;
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), "
"buflen %d, #VPP-TBD#", getpid (),
session->sndbuf_size, session->sndbuf_size, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_REUSEADDR:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_REUSEADDR);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_REUSEADDR: %d, "
"buflen %d, #VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_REUSEADDR:
if (buffer && buflen && (*buflen == sizeof (int)) &&
!VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEADDR);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEADDR);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d,"
" #VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_REUSEADDR), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_REUSEPORT:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_REUSEPORT);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d,"
" #VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_REUSEPORT:
if (buffer && buflen && (*buflen == sizeof (int)) &&
!VCL_SESS_ATTR_TEST (session->attr, VCL_SESS_ATTR_LISTEN))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_REUSEPORT);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_REUSEPORT);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d,"
" #VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_REUSEPORT), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_BROADCAST:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_BROADCAST);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d,"
" #VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_BROADCAST:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_BROADCAST);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_BROADCAST);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_BROADCAST), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_V6ONLY:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_V6ONLY);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, "
"#VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_V6ONLY:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_V6ONLY);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_V6ONLY);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_V6ONLY), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_KEEPALIVE:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_KEEPALIVE);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, "
"#VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_KEEPALIVE:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_KEEPALIVE);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_KEEPALIVE);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_KEEPALIVE), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_TCP_NODELAY:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_NODELAY);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, "
"#VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_TCP_NODELAY:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_NODELAY);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_NODELAY), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_KEEPIDLE);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, "
"#VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPIDLE);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
if (buffer && buflen && (*buflen >= sizeof (int)))
{
/* VPP-TBD */
*(int *) buffer = VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_KEEPINTVL);
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, "
"#VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
if (buffer && buflen && (*buflen == sizeof (int)))
{
/* VPP-TBD */
if (*(int *) buffer)
VCL_SESS_ATTR_SET (session->attr, VCL_SESS_ATTR_TCP_KEEPINTVL);
else
VCL_SESS_ATTR_CLR (session->attr, VCL_SESS_ATTR_TCP_KEEPINTVL);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, "
"#VPP-TBD#", getpid (),
VCL_SESS_ATTR_TEST (session->attr,
VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_TCP_USER_MSS:
if (buffer && buflen && (*buflen >= sizeof (u32)))
{
/* VPP-TBD */
*(u32 *) buffer = session->user_mss;
*buflen = sizeof (int);
VDBG (2, "VCL<%d>: VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d,"
" #VPP-TBD#", getpid (), *(int *) buffer, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_SET_TCP_USER_MSS:
if (buffer && buflen && (*buflen == sizeof (u32)))
{
/* VPP-TBD */
session->user_mss = *(u32 *) buffer;
VDBG (2, "VCL<%d>: VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d, "
"#VPP-TBD#", getpid (), session->user_mss, *buflen);
}
else
rv = VPPCOM_EINVAL;
break;
case VPPCOM_ATTR_GET_REFCNT:
rv = vcl_session_get_refcnt (session);
break;
default:
rv = VPPCOM_EINVAL;
break;
}
return rv;
}
int
vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
uint32_t buflen, int flags, vppcom_endpt_t * ep)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
int rv = VPPCOM_OK;
vcl_session_t *session = 0;
if (ep)
{
session = vcl_session_get_w_handle (wrk, session_handle);
if (PREDICT_FALSE (!session))
{
VDBG (0, "VCL<%d>: invalid session, sid (%u) has been closed!",
getpid (), session_handle);
return VPPCOM_EBADFD;
}
ep->is_ip4 = session->transport.is_ip4;
ep->port = session->transport.rmt_port;
}
if (flags == 0)
rv = vppcom_session_read (session_handle, buffer, buflen);
else if (flags & MSG_PEEK)
rv = vppcom_session_peek (session_handle, buffer, buflen);
else
{
VDBG (0, "Unsupport flags for recvfrom %d", flags);
return VPPCOM_EAFNOSUPPORT;
}
if (ep)
{
if (session->transport.is_ip4)
clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
sizeof (ip4_address_t));
else
clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
sizeof (ip6_address_t));
}
return rv;
}
int
vppcom_session_sendto (uint32_t session_handle, void *buffer,
uint32_t buflen, int flags, vppcom_endpt_t * ep)
{
if (!buffer)
return VPPCOM_EINVAL;
if (ep)
{
// TBD
return VPPCOM_EINVAL;
}
if (flags)
{
// TBD check the flags and do the right thing
VDBG (2, "VCL<%d>: handling flags 0x%u (%d) not implemented yet.",
getpid (), flags, flags);
}
return (vppcom_session_write_inline (session_handle, buffer, buflen, 1));
}
int
vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
u32 i, keep_trying = 1;
svm_msg_q_msg_t msg;
session_event_t *e;
int rv, num_ev = 0;
VDBG (3, "VCL<%d>: vp %p, nsids %u, wait_for_time %f",
getpid (), vp, n_sids, wait_for_time);
if (!vp)
return VPPCOM_EFAULT;
do
{
vcl_session_t *session;
/* Dequeue all events and drop all unhandled io events */
while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
{
e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
vcl_handle_mq_event (wrk, e);
svm_msg_q_free_msg (wrk->app_event_queue, &msg);
}
vec_reset_length (wrk->unhandled_evts_vector);
for (i = 0; i < n_sids; i++)
{
session = vcl_session_get (wrk, vp[i].sid);
if (!session)
{
vp[i].revents = POLLHUP;
num_ev++;
continue;
}
vp[i].revents = 0;
if (POLLIN & vp[i].events)
{
rv = vppcom_session_read_ready (session);
if (rv > 0)
{
vp[i].revents |= POLLIN;
num_ev++;
}
else if (rv < 0)
{
switch (rv)
{
case VPPCOM_ECONNRESET:
vp[i].revents = POLLHUP;
break;
default:
vp[i].revents = POLLERR;
break;
}
num_ev++;
}
}
if (POLLOUT & vp[i].events)
{
rv = vppcom_session_write_ready (session);
if (rv > 0)
{
vp[i].revents |= POLLOUT;
num_ev++;
}
else if (rv < 0)
{
switch (rv)
{
case VPPCOM_ECONNRESET:
vp[i].revents = POLLHUP;
break;
default:
vp[i].revents = POLLERR;
break;
}
num_ev++;
}
}
if (0) // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
{
vp[i].revents = POLLNVAL;
num_ev++;
}
}
if (wait_for_time != -1)
keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
}
while ((num_ev == 0) && keep_trying);
if (VPPCOM_DEBUG > 3)
{
clib_warning ("VCL<%d>: returning %d", getpid (), num_ev);
for (i = 0; i < n_sids; i++)
{
clib_warning ("VCL<%d>: vp[%d].sid %d (0x%x), .events 0x%x, "
".revents 0x%x", getpid (), i, vp[i].sid, vp[i].sid,
vp[i].events, vp[i].revents);
}
}
return num_ev;
}
int
vppcom_mq_epoll_fd (void)
{
vcl_worker_t *wrk = vcl_worker_get_current ();
return wrk->mqs_epfd;
}
int
vppcom_session_index (uint32_t session_handle)
{
return session_handle & 0xFFFFFF;
}
int
vppcom_session_handle (uint32_t session_index)
{
return (vcl_get_worker_index () << 24) | session_index;
}
int
vppcom_worker_register (void)
{
if (!vcl_worker_alloc_and_init ())
return VPPCOM_EEXIST;
if (vcl_worker_set_bapi ())
return VPPCOM_EEXIST;
if (vcl_worker_register_with_vpp ())
return VPPCOM_EEXIST;
return VPPCOM_OK;
}
int
vppcom_worker_index (void)
{
return vcl_get_worker_index ();
}
/*
* fd.io coding-style-patch-verification: ON
*
* Local Variables:
* eval: (c-set-style "gnu")
* End:
*/