diff options
author | Nathan Skrzypczak <nathan.skrzypczak@gmail.com> | 2019-08-12 12:02:25 +0200 |
---|---|---|
committer | Florin Coras <florin.coras@gmail.com> | 2019-08-19 20:50:24 +0000 |
commit | a8462fbca3bdfa5da602f7787f83bbdf8da64d5d (patch) | |
tree | 1bfcb12b58289df0ca10bbb60233d1d9d4525ee5 /src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c | |
parent | 1682b51c416561b083d84194d49ff617e070eb47 (diff) |
hsa: move tcp_echo to vpp_echo
Type: refactor
Change-Id: Ic23859a1686eff2ae4d08a86b69ff4813498459e
Signed-off-by: Nathan Skrzypczak <nathan.skrzypczak@gmail.com>
Diffstat (limited to 'src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c')
-rw-r--r-- | src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c | 143 |
1 files changed, 143 insertions, 0 deletions
diff --git a/src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c b/src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c new file mode 100644 index 00000000000..51144d34889 --- /dev/null +++ b/src/plugins/hs_apps/sapi/vpp_echo_proto_tcp.c @@ -0,0 +1,143 @@ +/* + * Copyright (c) 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 <stdio.h> +#include <signal.h> + +#include <hs_apps/sapi/vpp_echo_common.h> + +static void +tcp_echo_cleanup_cb (echo_session_t * s, u8 parent_died) +{ + echo_main_t *em = &echo_main; + echo_session_t *ls; + ASSERT (s->session_state < ECHO_SESSION_STATE_CLOSED); + if (parent_died) + clib_atomic_fetch_add (&em->stats.clean_count.s, 1); + else if (s->listener_index != SESSION_INVALID_INDEX) + { + ls = pool_elt_at_index (em->sessions, s->listener_index); + clib_atomic_sub_fetch (&ls->accepted_session_count, 1); + } + + + clib_atomic_sub_fetch (&em->n_clients_connected, 1); + s->session_state = ECHO_SESSION_STATE_CLOSED; + if (!em->n_clients_connected) + em->state = STATE_DATA_DONE; +} + +static void +tcp_echo_connected_cb (session_connected_bundled_msg_t * mp, + u32 session_index, u8 is_failed) +{ + static u32 client_index = 0; + echo_main_t *em = &echo_main; + echo_session_t *session = pool_elt_at_index (em->sessions, session_index); + if (is_failed) + { + ECHO_FAIL ("Bapi connect errored"); + return; /* Dont handle bapi connect errors for now */ + } + + ECHO_LOG (1, "Connected session 0x%lx -> URI", + ((session_connected_msg_t *) mp)->handle); + session->session_type = ECHO_SESSION_TYPE_STREAM; + session->accepted_session_count = 0; + clib_atomic_fetch_add (&em->n_clients_connected, 1); + session->bytes_to_send = em->bytes_to_send; + session->bytes_to_receive = em->bytes_to_receive; + session->session_state = ECHO_SESSION_STATE_READY; + em->data_thread_args[client_index++] = session->session_index; + + if (em->n_clients_connected == em->n_clients && em->state < STATE_READY) + { + echo_notify_event (em, ECHO_EVT_LAST_SCONNECTED); + em->state = STATE_READY; + } +} + +static void +tcp_echo_accepted_cb (session_accepted_msg_t * mp, echo_session_t * session) +{ + static u32 client_index = 0; + echo_main_t *em = &echo_main; + echo_session_t *ls; + + echo_notify_event (em, ECHO_EVT_FIRST_QCONNECT); + ls = pool_elt_at_index (em->sessions, session->listener_index); + session->session_type = ECHO_SESSION_TYPE_STREAM; + echo_notify_event (em, ECHO_EVT_FIRST_SCONNECT); + clib_atomic_fetch_add (&ls->accepted_session_count, 1); + clib_atomic_fetch_add (&em->n_clients_connected, 1); + + session->bytes_to_send = em->bytes_to_send; + session->bytes_to_receive = em->bytes_to_receive; + em->data_thread_args[client_index++] = session->session_index; + session->session_state = ECHO_SESSION_STATE_READY; + + if (em->n_clients_connected == em->n_clients && em->state < STATE_READY) + { + echo_notify_event (em, ECHO_EVT_LAST_SCONNECTED); + em->state = STATE_READY; + } +} + +static void +tcp_echo_disconnected_reply_cb (echo_session_t * s) +{ + s->session_state = ECHO_SESSION_STATE_CLOSING; +} + +static void +tcp_echo_disconnected_cb (session_disconnected_msg_t * mp, echo_session_t * s) +{ + echo_main_t *em = &echo_main; + echo_session_print_stats (em, s); + if (s->bytes_to_receive || s->bytes_to_send) + s->session_state = ECHO_SESSION_STATE_AWAIT_DATA; + else + s->session_state = ECHO_SESSION_STATE_CLOSING; + clib_atomic_fetch_add (&em->stats.close_count.s, 1); +} + +static void +tcp_echo_reset_cb (session_reset_msg_t * mp, echo_session_t * s) +{ + echo_main_t *em = &echo_main; + clib_atomic_fetch_add (&em->stats.reset_count.s, 1); + s->session_state = ECHO_SESSION_STATE_CLOSING; +} + +echo_proto_cb_vft_t echo_tcp_proto_cb_vft = { + .disconnected_cb = tcp_echo_disconnected_cb, + .connected_cb = tcp_echo_connected_cb, + .accepted_cb = tcp_echo_accepted_cb, + .reset_cb = tcp_echo_reset_cb, + .disconnected_reply_cb = tcp_echo_disconnected_reply_cb, + .cleanup_cb = tcp_echo_cleanup_cb, +}; + +ECHO_REGISTER_PROTO (TRANSPORT_PROTO_TCP, echo_tcp_proto_cb_vft); +ECHO_REGISTER_PROTO (TRANSPORT_PROTO_TLS, echo_tcp_proto_cb_vft); +ECHO_REGISTER_PROTO (TRANSPORT_PROTO_SCTP, echo_tcp_proto_cb_vft); + +/* + * fd.io coding-style-patch-verification: ON + * + * Local Variables: + * eval: (c-set-style "gnu") + * End: + */ |