vpp_echo_proto_tcp.c revision 7c40a383
1/*
2 * Copyright (c) 2019 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#include <stdio.h>
17#include <signal.h>
18
19#include <hs_apps/sapi/vpp_echo_common.h>
20
21static void
22tcp_echo_cleanup_cb (echo_session_t * s, u8 parent_died)
23{
24  echo_main_t *em = &echo_main;
25  echo_session_t *ls;
26  ASSERT (s->session_state < ECHO_SESSION_STATE_CLOSED);
27  if (parent_died)
28    clib_atomic_fetch_add (&em->stats.clean_count.s, 1);
29  else if (s->listener_index != SESSION_INVALID_INDEX)
30    {
31      ls = pool_elt_at_index (em->sessions, s->listener_index);
32      clib_atomic_sub_fetch (&ls->accepted_session_count, 1);
33    }
34
35
36  clib_atomic_sub_fetch (&em->n_clients_connected, 1);
37  s->session_state = ECHO_SESSION_STATE_CLOSED;
38  if (!em->n_clients_connected)
39    em->state = STATE_DATA_DONE;
40}
41
42static void
43tcp_echo_connected_cb (session_connected_bundled_msg_t * mp,
44		       u32 session_index, u8 is_failed)
45{
46  static u32 client_index = 0;
47  echo_main_t *em = &echo_main;
48  echo_session_t *session = pool_elt_at_index (em->sessions, session_index);
49  if (is_failed)
50    {
51      ECHO_FAIL (ECHO_FAIL_TCP_BAPI_CONNECT,
52		 "Bapi connect errored on session %u", session_index);
53      return;			/* Dont handle bapi connect errors for now */
54    }
55
56  ECHO_LOG (1, "Connected session 0x%lx -> URI",
57	    ((session_connected_msg_t *) mp)->handle);
58  session->session_type = ECHO_SESSION_TYPE_STREAM;
59  session->accepted_session_count = 0;
60  clib_atomic_fetch_add (&em->n_clients_connected, 1);
61  session->bytes_to_send = em->bytes_to_send;
62  session->bytes_to_receive = em->bytes_to_receive;
63  session->session_state = ECHO_SESSION_STATE_READY;
64  em->data_thread_args[client_index++] = session->session_index;
65
66  if (em->n_clients_connected == em->n_clients && em->state < STATE_READY)
67    {
68      echo_notify_event (em, ECHO_EVT_LAST_SCONNECTED);
69      em->state = STATE_READY;
70    }
71}
72
73static void
74tcp_echo_accepted_cb (session_accepted_msg_t * mp, echo_session_t * session)
75{
76  static u32 client_index = 0;
77  echo_main_t *em = &echo_main;
78  echo_session_t *ls;
79
80  echo_notify_event (em, ECHO_EVT_FIRST_QCONNECT);
81  ls = pool_elt_at_index (em->sessions, session->listener_index);
82  session->session_type = ECHO_SESSION_TYPE_STREAM;
83  echo_notify_event (em, ECHO_EVT_FIRST_SCONNECT);
84  clib_atomic_fetch_add (&ls->accepted_session_count, 1);
85  clib_atomic_fetch_add (&em->n_clients_connected, 1);
86
87  session->bytes_to_send = em->bytes_to_send;
88  session->bytes_to_receive = em->bytes_to_receive;
89  em->data_thread_args[client_index++] = session->session_index;
90  session->session_state = ECHO_SESSION_STATE_READY;
91
92  if (em->n_clients_connected == em->n_clients && em->state < STATE_READY)
93    {
94      echo_notify_event (em, ECHO_EVT_LAST_SCONNECTED);
95      em->state = STATE_READY;
96    }
97}
98
99static void
100tcp_echo_sent_disconnect_cb (echo_session_t * s)
101{
102  s->session_state = ECHO_SESSION_STATE_CLOSING;
103}
104
105static void
106tcp_echo_disconnected_cb (session_disconnected_msg_t * mp, echo_session_t * s)
107{
108  echo_main_t *em = &echo_main;
109  echo_session_print_stats (em, s);
110  if (s->bytes_to_receive || s->bytes_to_send)
111    s->session_state = ECHO_SESSION_STATE_AWAIT_DATA;
112  else
113    s->session_state = ECHO_SESSION_STATE_CLOSING;
114  clib_atomic_fetch_add (&em->stats.close_count.s, 1);
115}
116
117static void
118tcp_echo_reset_cb (session_reset_msg_t * mp, echo_session_t * s)
119{
120  echo_main_t *em = &echo_main;
121  clib_atomic_fetch_add (&em->stats.reset_count.s, 1);
122  s->session_state = ECHO_SESSION_STATE_CLOSING;
123}
124
125echo_proto_cb_vft_t echo_tcp_proto_cb_vft = {
126  .disconnected_cb = tcp_echo_disconnected_cb,
127  .connected_cb = tcp_echo_connected_cb,
128  .accepted_cb = tcp_echo_accepted_cb,
129  .reset_cb = tcp_echo_reset_cb,
130  .sent_disconnect_cb = tcp_echo_sent_disconnect_cb,
131  .cleanup_cb = tcp_echo_cleanup_cb,
132};
133
134echo_proto_cb_vft_t echo_tls_proto_cb_vft = {
135  .disconnected_cb = tcp_echo_disconnected_cb,
136  .connected_cb = tcp_echo_connected_cb,
137  .accepted_cb = tcp_echo_accepted_cb,
138  .reset_cb = tcp_echo_reset_cb,
139  .sent_disconnect_cb = tcp_echo_sent_disconnect_cb,
140  .cleanup_cb = tcp_echo_cleanup_cb,
141};
142
143ECHO_REGISTER_PROTO (TRANSPORT_PROTO_TCP, echo_tcp_proto_cb_vft);
144ECHO_REGISTER_PROTO (TRANSPORT_PROTO_TLS, echo_tls_proto_cb_vft);
145ECHO_REGISTER_PROTO (TRANSPORT_PROTO_SCTP, echo_tcp_proto_cb_vft);
146
147/*
148 * fd.io coding-style-patch-verification: ON
149 *
150 * Local Variables:
151 * eval: (c-set-style "gnu")
152 * End:
153 */
154