blob: 58f6ac0a13446b6085f3408e449e36a7bc99ca73 [file] [log] [blame]
Dave Wallace543852a2017-08-03 02:11:34 -04001/*
Florin Coras5e062572019-03-14 19:07:51 -07002 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
Dave Wallace543852a2017-08-03 02:11:34 -04003 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this
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 <stdlib.h>
Dave Wallace5c7cf1c2017-10-24 04:12:18 -040018#include <vcl/vppcom.h>
Florin Coras0d427d82018-06-27 03:24:07 -070019#include <vcl/vcl_debug.h>
Florin Coras697faea2018-06-27 17:10:49 -070020#include <vcl/vcl_private.h>
Florin Coras88001c62019-04-24 14:44:46 -070021#include <svm/fifo_segment.h>
Dave Wallace7e607a72018-06-18 18:41:32 -040022
Florin Coras134a9962018-08-28 11:32:04 -070023__thread uword __vcl_worker_index = ~0;
24
Florin Coras30e79c22019-01-02 19:31:22 -080025static inline int
Florin Corase003a1b2019-06-05 10:47:16 -070026vcl_mq_dequeue_batch (vcl_worker_t * wrk, svm_msg_q_t * mq, u32 n_max_msg)
Florin Coras30e79c22019-01-02 19:31:22 -080027{
Florin Coras5398dfb2021-01-25 20:31:27 -080028 u32 n_msgs = 0, sz, len;
Florin Coras30e79c22019-01-02 19:31:22 -080029
Florin Coras5398dfb2021-01-25 20:31:27 -080030 while ((sz = svm_msg_q_size (mq)))
Florin Coras30e79c22019-01-02 19:31:22 -080031 {
Florin Coras5398dfb2021-01-25 20:31:27 -080032 len = vec_len (wrk->mq_msg_vector);
33 vec_validate (wrk->mq_msg_vector, len + sz - 1);
34 svm_msg_q_sub_raw_batch (mq, wrk->mq_msg_vector + len, sz);
35 n_msgs += sz;
Florin Coras30e79c22019-01-02 19:31:22 -080036 }
37 return n_msgs;
38}
39
Dave Wallace543852a2017-08-03 02:11:34 -040040
Florin Coras697faea2018-06-27 17:10:49 -070041
Florin Coras458089b2019-08-21 16:20:44 -070042static void
Florin Coras4ac25842021-04-19 17:34:54 -070043vcl_msg_add_ext_config (vcl_session_t *s, uword *offset)
44{
45 svm_fifo_chunk_t *c;
46
47 c = vcl_segment_alloc_chunk (vcl_vpp_worker_segment_handle (0),
48 0 /* one slice only */, s->ext_config->len,
49 offset);
50 if (c)
51 clib_memcpy_fast (c->data, s->ext_config, s->ext_config->len);
52}
53
Florin Coras32881932023-02-06 13:30:13 -080054void
55vcl_send_session_listen (vcl_worker_t *wrk, vcl_session_t *s)
Florin Coras458089b2019-08-21 16:20:44 -070056{
57 app_session_evt_t _app_evt, *app_evt = &_app_evt;
58 session_listen_msg_t *mp;
59 svm_msg_q_t *mq;
60
61 mq = vcl_worker_ctrl_mq (wrk);
62 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_LISTEN);
63 mp = (session_listen_msg_t *) app_evt->evt->data;
64 memset (mp, 0, sizeof (*mp));
Florin Corascc7c88e2020-09-15 15:56:51 -070065 mp->client_index = wrk->api_client_handle;
Florin Coras458089b2019-08-21 16:20:44 -070066 mp->context = s->session_index;
67 mp->wrk_index = wrk->vpp_wrk_index;
68 mp->is_ip4 = s->transport.is_ip4;
69 clib_memcpy_fast (&mp->ip, &s->transport.lcl_ip, sizeof (mp->ip));
70 mp->port = s->transport.lcl_port;
71 mp->proto = s->session_type;
Florin Coras6a6555a2021-01-28 11:39:27 -080072 mp->vrf = s->vrf;
Florin Coras1e966172020-05-16 18:18:14 +000073 if (s->flags & VCL_SESSION_F_CONNECTED)
74 mp->flags = TRANSPORT_CFG_F_CONNECTED;
Florin Coras4ac25842021-04-19 17:34:54 -070075 if (s->ext_config)
76 vcl_msg_add_ext_config (s, &mp->ext_config);
Florin Coras458089b2019-08-21 16:20:44 -070077 app_send_ctrl_evt_to_vpp (mq, app_evt);
Florin Coras4ac25842021-04-19 17:34:54 -070078 if (s->ext_config)
79 {
80 clib_mem_free (s->ext_config);
81 s->ext_config = 0;
82 }
Florin Coras32881932023-02-06 13:30:13 -080083 s->flags |= VCL_SESSION_F_PENDING_LISTEN;
Florin Coras458089b2019-08-21 16:20:44 -070084}
85
86static void
87vcl_send_session_connect (vcl_worker_t * wrk, vcl_session_t * s)
88{
89 app_session_evt_t _app_evt, *app_evt = &_app_evt;
90 session_connect_msg_t *mp;
91 svm_msg_q_t *mq;
92
93 mq = vcl_worker_ctrl_mq (wrk);
94 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_CONNECT);
95 mp = (session_connect_msg_t *) app_evt->evt->data;
96 memset (mp, 0, sizeof (*mp));
Florin Corascc7c88e2020-09-15 15:56:51 -070097 mp->client_index = wrk->api_client_handle;
Florin Coras458089b2019-08-21 16:20:44 -070098 mp->context = s->session_index;
Filip Tehlar2f09bfc2021-11-15 10:26:56 +000099 mp->dscp = s->dscp;
Florin Coras458089b2019-08-21 16:20:44 -0700100 mp->wrk_index = wrk->vpp_wrk_index;
101 mp->is_ip4 = s->transport.is_ip4;
102 mp->parent_handle = s->parent_handle;
103 clib_memcpy_fast (&mp->ip, &s->transport.rmt_ip, sizeof (mp->ip));
Florin Corasef7cbf62019-10-17 09:56:27 -0700104 clib_memcpy_fast (&mp->lcl_ip, &s->transport.lcl_ip, sizeof (mp->lcl_ip));
Florin Coras458089b2019-08-21 16:20:44 -0700105 mp->port = s->transport.rmt_port;
Florin Coras0a1e1832020-03-29 18:54:04 +0000106 mp->lcl_port = s->transport.lcl_port;
Florin Coras458089b2019-08-21 16:20:44 -0700107 mp->proto = s->session_type;
Florin Coras6a6555a2021-01-28 11:39:27 -0800108 mp->vrf = s->vrf;
Florin Coras0a1e1832020-03-29 18:54:04 +0000109 if (s->flags & VCL_SESSION_F_CONNECTED)
110 mp->flags |= TRANSPORT_CFG_F_CONNECTED;
Florin Coras4ac25842021-04-19 17:34:54 -0700111 if (s->ext_config)
112 vcl_msg_add_ext_config (s, &mp->ext_config);
Florin Coras458089b2019-08-21 16:20:44 -0700113 app_send_ctrl_evt_to_vpp (mq, app_evt);
Florin Coras4ac25842021-04-19 17:34:54 -0700114
115 if (s->ext_config)
116 {
117 clib_mem_free (s->ext_config);
118 s->ext_config = 0;
119 }
Florin Coras458089b2019-08-21 16:20:44 -0700120}
121
122void
123vcl_send_session_unlisten (vcl_worker_t * wrk, vcl_session_t * s)
124{
125 app_session_evt_t _app_evt, *app_evt = &_app_evt;
126 session_unlisten_msg_t *mp;
127 svm_msg_q_t *mq;
128
129 mq = vcl_worker_ctrl_mq (wrk);
130 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_UNLISTEN);
131 mp = (session_unlisten_msg_t *) app_evt->evt->data;
132 memset (mp, 0, sizeof (*mp));
Florin Corascc7c88e2020-09-15 15:56:51 -0700133 mp->client_index = wrk->api_client_handle;
Florin Coras458089b2019-08-21 16:20:44 -0700134 mp->wrk_index = wrk->vpp_wrk_index;
135 mp->handle = s->vpp_handle;
136 mp->context = wrk->wrk_index;
137 app_send_ctrl_evt_to_vpp (mq, app_evt);
138}
139
140static void
liuyacan534468e2021-05-09 03:50:40 +0000141vcl_send_session_shutdown (vcl_worker_t *wrk, vcl_session_t *s)
142{
143 app_session_evt_t _app_evt, *app_evt = &_app_evt;
144 session_shutdown_msg_t *mp;
145 svm_msg_q_t *mq;
146
147 /* Send to thread that owns the session */
148 mq = s->vpp_evt_q;
149 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_SHUTDOWN);
150 mp = (session_shutdown_msg_t *) app_evt->evt->data;
151 memset (mp, 0, sizeof (*mp));
152 mp->client_index = wrk->api_client_handle;
153 mp->handle = s->vpp_handle;
154 app_send_ctrl_evt_to_vpp (mq, app_evt);
155}
156
157static void
Florin Coras458089b2019-08-21 16:20:44 -0700158vcl_send_session_disconnect (vcl_worker_t * wrk, vcl_session_t * s)
159{
160 app_session_evt_t _app_evt, *app_evt = &_app_evt;
161 session_disconnect_msg_t *mp;
162 svm_msg_q_t *mq;
163
164 /* Send to thread that owns the session */
165 mq = s->vpp_evt_q;
166 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_DISCONNECT);
167 mp = (session_disconnect_msg_t *) app_evt->evt->data;
168 memset (mp, 0, sizeof (*mp));
Florin Corascc7c88e2020-09-15 15:56:51 -0700169 mp->client_index = wrk->api_client_handle;
Florin Coras458089b2019-08-21 16:20:44 -0700170 mp->handle = s->vpp_handle;
171 app_send_ctrl_evt_to_vpp (mq, app_evt);
172}
173
174static void
175vcl_send_app_detach (vcl_worker_t * wrk)
176{
177 app_session_evt_t _app_evt, *app_evt = &_app_evt;
178 session_app_detach_msg_t *mp;
179 svm_msg_q_t *mq;
180
181 mq = vcl_worker_ctrl_mq (wrk);
182 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_DETACH);
183 mp = (session_app_detach_msg_t *) app_evt->evt->data;
184 memset (mp, 0, sizeof (*mp));
Florin Corascc7c88e2020-09-15 15:56:51 -0700185 mp->client_index = wrk->api_client_handle;
Florin Coras458089b2019-08-21 16:20:44 -0700186 app_send_ctrl_evt_to_vpp (mq, app_evt);
187}
Florin Coras697faea2018-06-27 17:10:49 -0700188
Florin Coras54693d22018-07-17 10:46:29 -0700189static void
190vcl_send_session_accepted_reply (svm_msg_q_t * mq, u32 context,
191 session_handle_t handle, int retval)
192{
193 app_session_evt_t _app_evt, *app_evt = &_app_evt;
194 session_accepted_reply_msg_t *rmp;
195 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_ACCEPTED_REPLY);
196 rmp = (session_accepted_reply_msg_t *) app_evt->evt->data;
197 rmp->handle = handle;
198 rmp->context = context;
199 rmp->retval = retval;
200 app_send_ctrl_evt_to_vpp (mq, app_evt);
201}
202
Florin Coras99368312018-08-02 10:45:44 -0700203static void
Florin Coras52dd29f2020-11-18 19:02:17 -0800204vcl_send_session_disconnected_reply (vcl_worker_t * wrk, vcl_session_t * s,
205 int retval)
Florin Coras99368312018-08-02 10:45:44 -0700206{
207 app_session_evt_t _app_evt, *app_evt = &_app_evt;
208 session_disconnected_reply_msg_t *rmp;
Florin Coras52dd29f2020-11-18 19:02:17 -0800209 app_alloc_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt,
Florin Coras99368312018-08-02 10:45:44 -0700210 SESSION_CTRL_EVT_DISCONNECTED_REPLY);
211 rmp = (session_disconnected_reply_msg_t *) app_evt->evt->data;
Florin Coras52dd29f2020-11-18 19:02:17 -0800212 rmp->handle = s->vpp_handle;
213 rmp->context = wrk->api_client_handle;
Florin Coras99368312018-08-02 10:45:44 -0700214 rmp->retval = retval;
Florin Coras52dd29f2020-11-18 19:02:17 -0800215 app_send_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt);
Florin Coras99368312018-08-02 10:45:44 -0700216}
217
Florin Corasc9fbd662018-08-24 12:59:56 -0700218static void
Florin Coras52dd29f2020-11-18 19:02:17 -0800219vcl_send_session_reset_reply (vcl_worker_t * wrk, vcl_session_t * s,
220 int retval)
Florin Corasc9fbd662018-08-24 12:59:56 -0700221{
222 app_session_evt_t _app_evt, *app_evt = &_app_evt;
223 session_reset_reply_msg_t *rmp;
Florin Coras52dd29f2020-11-18 19:02:17 -0800224 app_alloc_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt,
225 SESSION_CTRL_EVT_RESET_REPLY);
Florin Corasc9fbd662018-08-24 12:59:56 -0700226 rmp = (session_reset_reply_msg_t *) app_evt->evt->data;
Florin Coras52dd29f2020-11-18 19:02:17 -0800227 rmp->handle = s->vpp_handle;
228 rmp->context = wrk->api_client_handle;
Florin Corasc9fbd662018-08-24 12:59:56 -0700229 rmp->retval = retval;
Florin Coras52dd29f2020-11-18 19:02:17 -0800230 app_send_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt);
Florin Corasc9fbd662018-08-24 12:59:56 -0700231}
232
Florin Coras30e79c22019-01-02 19:31:22 -0800233void
234vcl_send_session_worker_update (vcl_worker_t * wrk, vcl_session_t * s,
235 u32 wrk_index)
236{
237 app_session_evt_t _app_evt, *app_evt = &_app_evt;
238 session_worker_update_msg_t *mp;
Florin Coras30e79c22019-01-02 19:31:22 -0800239
Florin Coras52dd29f2020-11-18 19:02:17 -0800240 app_alloc_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt,
241 SESSION_CTRL_EVT_WORKER_UPDATE);
Florin Coras30e79c22019-01-02 19:31:22 -0800242 mp = (session_worker_update_msg_t *) app_evt->evt->data;
Florin Corascc7c88e2020-09-15 15:56:51 -0700243 mp->client_index = wrk->api_client_handle;
Florin Coras30e79c22019-01-02 19:31:22 -0800244 mp->handle = s->vpp_handle;
245 mp->req_wrk_index = wrk->vpp_wrk_index;
246 mp->wrk_index = wrk_index;
Florin Coras52dd29f2020-11-18 19:02:17 -0800247 app_send_ctrl_evt_to_vpp (s->vpp_evt_q, app_evt);
Florin Coras30e79c22019-01-02 19:31:22 -0800248}
249
hanlina3a48962020-07-13 11:09:15 +0800250int
Florin Coras40c07ce2020-07-16 20:46:17 -0700251vcl_send_worker_rpc (u32 dst_wrk_index, void *data, u32 data_len)
252{
253 app_session_evt_t _app_evt, *app_evt = &_app_evt;
254 session_app_wrk_rpc_msg_t *mp;
255 vcl_worker_t *dst_wrk, *wrk;
256 svm_msg_q_t *mq;
hanlina3a48962020-07-13 11:09:15 +0800257 int ret = -1;
Florin Coras40c07ce2020-07-16 20:46:17 -0700258
259 if (data_len > sizeof (mp->data))
260 goto done;
261
262 clib_spinlock_lock (&vcm->workers_lock);
263
264 dst_wrk = vcl_worker_get_if_valid (dst_wrk_index);
265 if (!dst_wrk)
266 goto done;
267
268 wrk = vcl_worker_get_current ();
269 mq = vcl_worker_ctrl_mq (wrk);
270 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_APP_WRK_RPC);
271 mp = (session_app_wrk_rpc_msg_t *) app_evt->evt->data;
Florin Corascc7c88e2020-09-15 15:56:51 -0700272 mp->client_index = wrk->api_client_handle;
Florin Coras40c07ce2020-07-16 20:46:17 -0700273 mp->wrk_index = dst_wrk->vpp_wrk_index;
274 clib_memcpy (mp->data, data, data_len);
275 app_send_ctrl_evt_to_vpp (mq, app_evt);
hanlina3a48962020-07-13 11:09:15 +0800276 ret = 0;
Florin Coras40c07ce2020-07-16 20:46:17 -0700277
278done:
279 clib_spinlock_unlock (&vcm->workers_lock);
hanlina3a48962020-07-13 11:09:15 +0800280 return ret;
Florin Coras40c07ce2020-07-16 20:46:17 -0700281}
282
Florin Coras04ae8272021-04-12 19:55:37 -0700283int
284vcl_session_transport_attr (vcl_worker_t *wrk, vcl_session_t *s, u8 is_get,
285 transport_endpt_attr_t *attr)
286{
287 app_session_evt_t _app_evt, *app_evt = &_app_evt;
288 session_transport_attr_msg_t *mp;
289 svm_msg_q_t *mq;
290 f64 timeout;
291
292 ASSERT (!wrk->session_attr_op);
Liangxing Wang22112772022-05-13 04:24:19 +0000293 mq = s->vpp_evt_q;
294 if (PREDICT_FALSE (!mq))
295 {
296 /* FIXME: attribute should be stored and sent once session is
297 * bound/connected to vpp */
298 return 0;
299 }
300
Florin Coras04ae8272021-04-12 19:55:37 -0700301 wrk->session_attr_op = 1;
302 wrk->session_attr_op_rv = -1;
303
Florin Coras04ae8272021-04-12 19:55:37 -0700304 app_alloc_ctrl_evt_to_vpp (mq, app_evt, SESSION_CTRL_EVT_TRANSPORT_ATTR);
305 mp = (session_transport_attr_msg_t *) app_evt->evt->data;
306 memset (mp, 0, sizeof (*mp));
307 mp->client_index = wrk->api_client_handle;
308 mp->handle = s->vpp_handle;
309 mp->is_get = is_get;
310 mp->attr = *attr;
311 app_send_ctrl_evt_to_vpp (mq, app_evt);
312
313 timeout = clib_time_now (&wrk->clib_time) + 1;
314
315 while (wrk->session_attr_op && clib_time_now (&wrk->clib_time) < timeout)
316 vcl_flush_mq_events ();
317
318 if (!wrk->session_attr_op_rv && is_get)
319 *attr = wrk->session_attr_rv;
320
321 wrk->session_attr_op = 0;
322
323 return wrk->session_attr_op_rv;
324}
325
Florin Coras54693d22018-07-17 10:46:29 -0700326static u32
Florin Coras00cca802019-06-06 09:38:44 -0700327vcl_session_accepted_handler (vcl_worker_t * wrk, session_accepted_msg_t * mp,
328 u32 ls_index)
Florin Coras54693d22018-07-17 10:46:29 -0700329{
330 vcl_session_t *session, *listen_session;
Florin Coras99368312018-08-02 10:45:44 -0700331 svm_msg_q_t *evt_q;
Florin Coras54693d22018-07-17 10:46:29 -0700332
Florin Coras134a9962018-08-28 11:32:04 -0700333 session = vcl_session_alloc (wrk);
Florin Coras99368312018-08-02 10:45:44 -0700334
Florin Coras00cca802019-06-06 09:38:44 -0700335 listen_session = vcl_session_get (wrk, ls_index);
336 if (listen_session->vpp_handle != mp->listener_handle)
Florin Coras54693d22018-07-17 10:46:29 -0700337 {
Florin Coras00cca802019-06-06 09:38:44 -0700338 VDBG (0, "ERROR: listener handle %lu does not match session %u",
339 mp->listener_handle, ls_index);
340 goto error;
341 }
342
Florin Corasf6e284b2021-07-21 18:17:20 -0700343 if (vcl_segment_attach_session (
344 mp->segment_handle, mp->server_rx_fifo, mp->server_tx_fifo,
345 mp->vpp_event_queue_address, mp->mq_index, 0, session))
Florin Coras00cca802019-06-06 09:38:44 -0700346 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800347 VDBG (0, "session %u [0x%llx]: failed to attach fifos",
348 session->session_index, mp->handle);
Florin Coras00cca802019-06-06 09:38:44 -0700349 goto error;
Florin Coras54693d22018-07-17 10:46:29 -0700350 }
351
Florin Coras54693d22018-07-17 10:46:29 -0700352 session->vpp_handle = mp->handle;
Florin Corasdfffdd72020-10-15 10:54:47 -0700353 session->session_state = VCL_STATE_READY;
qinyangaf9b7152023-06-27 01:11:53 -0700354 if (mp->rmt.is_ip4)
355 {
356 session->original_dst_ip4 = mp->original_dst_ip4;
357 session->original_dst_port = mp->original_dst_port;
358 }
Florin Coras09d18c22019-04-24 11:10:02 -0700359 session->transport.rmt_port = mp->rmt.port;
360 session->transport.is_ip4 = mp->rmt.is_ip4;
361 clib_memcpy_fast (&session->transport.rmt_ip, &mp->rmt.ip,
Dave Barach178cf492018-11-13 16:34:13 -0500362 sizeof (ip46_address_t));
Florin Coras54693d22018-07-17 10:46:29 -0700363
Florin Coras134a9962018-08-28 11:32:04 -0700364 vcl_session_table_add_vpp_handle (wrk, mp->handle, session->session_index);
Florin Coras67c90a32021-03-09 18:36:06 -0800365 session->transport.lcl_port = mp->lcl.port;
366 session->transport.lcl_ip = mp->lcl.ip;
Florin Coras460dce62018-07-27 05:45:06 -0700367 session->session_type = listen_session->session_type;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +0200368 session->is_dgram = vcl_proto_is_dgram (session->session_type);
Florin Coras89627e82023-04-27 13:38:35 -0700369 if (session->is_dgram)
370 session->flags |= (listen_session->flags & VCL_SESSION_F_CONNECTED);
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +0200371 session->listener_index = listen_session->session_index;
372 listen_session->n_accepted_sessions++;
Florin Coras54693d22018-07-17 10:46:29 -0700373
Florin Coras54693d22018-07-17 10:46:29 -0700374 vcl_evt (VCL_EVT_ACCEPT, session, listen_session, session_index);
375
Florin Coras00cca802019-06-06 09:38:44 -0700376 vcl_send_session_accepted_reply (session->vpp_evt_q, mp->context,
377 session->vpp_handle, 0);
378
Florin Coras134a9962018-08-28 11:32:04 -0700379 return session->session_index;
Florin Coras00cca802019-06-06 09:38:44 -0700380
381error:
Florin Corasb4624182020-12-11 13:58:12 -0800382 vcl_segment_attach_mq (vcl_vpp_worker_segment_handle (0),
383 mp->vpp_event_queue_address, mp->mq_index, &evt_q);
Florin Coras00cca802019-06-06 09:38:44 -0700384 vcl_send_session_accepted_reply (evt_q, mp->context, mp->handle,
385 VNET_API_ERROR_INVALID_ARGUMENT);
386 vcl_session_free (wrk, session);
387 return VCL_INVALID_SESSION_INDEX;
Florin Coras54693d22018-07-17 10:46:29 -0700388}
389
390static u32
Florin Coras134a9962018-08-28 11:32:04 -0700391vcl_session_connected_handler (vcl_worker_t * wrk,
392 session_connected_msg_t * mp)
Florin Coras54693d22018-07-17 10:46:29 -0700393{
Florin Coras99368312018-08-02 10:45:44 -0700394 vcl_session_t *session = 0;
Florin Coras52dd29f2020-11-18 19:02:17 -0800395 u32 session_index;
Florin Coras54693d22018-07-17 10:46:29 -0700396
397 session_index = mp->context;
Florin Coras134a9962018-08-28 11:32:04 -0700398 session = vcl_session_get (wrk, session_index);
Florin Corasef2c0f12021-11-10 22:44:52 -0800399 if (PREDICT_FALSE (!session))
Florin Coras070453d2018-08-24 17:04:27 -0700400 {
Florin Coras8d42c752021-11-29 23:26:19 -0800401 VERR ("vpp handle 0x%llx has no session index (%u)!", mp->handle,
402 session_index);
Florin Corasef2c0f12021-11-10 22:44:52 -0800403 /* Should not happen but if it does, force vpp session cleanup */
404 vcl_session_t tmp_session = {
405 .vpp_handle = mp->handle,
406 .vpp_evt_q = 0,
407 };
408 vcl_segment_attach_session (
409 mp->segment_handle, mp->server_rx_fifo, mp->server_tx_fifo,
410 mp->vpp_event_queue_address, mp->mq_index, 0, session);
411 if (tmp_session.vpp_evt_q)
412 vcl_send_session_disconnect (wrk, &tmp_session);
Florin Coras070453d2018-08-24 17:04:27 -0700413 return VCL_INVALID_SESSION_INDEX;
414 }
Florin Coras8d42c752021-11-29 23:26:19 -0800415
Florin Coras54693d22018-07-17 10:46:29 -0700416 if (mp->retval)
417 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800418 VDBG (0, "session %u: connect failed! %U", session_index,
Florin Coras8d42c752021-11-29 23:26:19 -0800419 format_session_error, mp->retval);
Florin Corasc127d5a2020-10-14 16:35:58 -0700420 session->session_state = VCL_STATE_DETACHED;
Florin Coras8d42c752021-11-29 23:26:19 -0800421 session->vpp_handle = VCL_INVALID_SESSION_HANDLE;
Radha krishna Saragadamdd92bde2022-07-18 19:41:05 +0530422 session->vpp_error = mp->retval;
Florin Coras070453d2018-08-24 17:04:27 -0700423 return session_index;
Florin Coras54693d22018-07-17 10:46:29 -0700424 }
425
Florin Corasdbc9c592019-09-25 16:37:43 -0700426 session->vpp_handle = mp->handle;
Florin Corasc547e912020-12-08 17:50:45 -0800427
Florin Coras8d42c752021-11-29 23:26:19 -0800428 /* Add to lookup table. Even if something fails, session cannot be
429 * cleaned up prior to notifying vpp and going through the cleanup
430 * "procedure" see @ref vcl_session_cleanup_handler */
431 vcl_session_table_add_vpp_handle (wrk, mp->handle, session_index);
432
Florin Corasf6e284b2021-07-21 18:17:20 -0700433 if (vcl_segment_attach_session (
434 mp->segment_handle, mp->server_rx_fifo, mp->server_tx_fifo,
435 mp->vpp_event_queue_address, mp->mq_index, 0, session))
Florin Corasd85de682018-11-29 17:02:29 -0800436 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800437 VDBG (0, "session %u [0x%llx]: failed to attach fifos",
438 session->session_index, session->vpp_handle);
Florin Coras8d42c752021-11-29 23:26:19 -0800439 session->session_state = VCL_STATE_UPDATED;
Florin Corasdbc9c592019-09-25 16:37:43 -0700440 vcl_send_session_disconnect (wrk, session);
441 return session_index;
Florin Corasd85de682018-11-29 17:02:29 -0800442 }
443
Florin Coras653e43f2019-03-04 10:56:23 -0800444 if (mp->ct_rx_fifo)
Florin Coras99368312018-08-02 10:45:44 -0700445 {
Florin Corasc547e912020-12-08 17:50:45 -0800446 if (vcl_segment_attach_session (mp->ct_segment_handle, mp->ct_rx_fifo,
Florin Corasf6e284b2021-07-21 18:17:20 -0700447 mp->ct_tx_fifo, (uword) ~0, ~0, 1,
448 session))
Florin Coras653e43f2019-03-04 10:56:23 -0800449 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800450 VDBG (0, "session %u [0x%llx]: failed to attach ct fifos",
451 session->session_index, session->vpp_handle);
Florin Coras8d42c752021-11-29 23:26:19 -0800452 session->session_state = VCL_STATE_UPDATED;
Florin Corasdbc9c592019-09-25 16:37:43 -0700453 vcl_send_session_disconnect (wrk, session);
454 return session_index;
Florin Coras653e43f2019-03-04 10:56:23 -0800455 }
Florin Coras99368312018-08-02 10:45:44 -0700456 }
Florin Coras54693d22018-07-17 10:46:29 -0700457
Florin Coras09d18c22019-04-24 11:10:02 -0700458 session->transport.is_ip4 = mp->lcl.is_ip4;
459 clib_memcpy_fast (&session->transport.lcl_ip, &mp->lcl.ip,
Dave Barach178cf492018-11-13 16:34:13 -0500460 sizeof (session->transport.lcl_ip));
Florin Coras09d18c22019-04-24 11:10:02 -0700461 session->transport.lcl_port = mp->lcl.port;
Florin Corasa5ea8212020-08-24 21:23:51 -0700462
463 /* Application closed session before connect reply */
Florin Corasac422d62020-10-19 20:51:36 -0700464 if (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK)
Florin Corasc127d5a2020-10-14 16:35:58 -0700465 && session->session_state == VCL_STATE_CLOSED)
Florin Corasa5ea8212020-08-24 21:23:51 -0700466 vcl_send_session_disconnect (wrk, session);
467 else
Florin Corasdfffdd72020-10-15 10:54:47 -0700468 session->session_state = VCL_STATE_READY;
Florin Coras54693d22018-07-17 10:46:29 -0700469
Florin Coras5ffb9642021-12-03 17:24:13 -0800470 VDBG (0, "session %u [0x%llx] connected local: %U:%u remote %U:%u",
471 session->session_index, session->vpp_handle, vcl_format_ip46_address,
472 &session->transport.lcl_ip,
473 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
474 clib_net_to_host_u16 (session->transport.lcl_port),
475 vcl_format_ip46_address, &session->transport.rmt_ip,
476 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
477 clib_net_to_host_u16 (session->transport.rmt_port));
Florin Coras070453d2018-08-24 17:04:27 -0700478
Florin Coras54693d22018-07-17 10:46:29 -0700479 return session_index;
480}
481
Florin Coras3c7d4f92018-12-14 11:28:43 -0800482static int
483vcl_flag_accepted_session (vcl_session_t * session, u64 handle, u32 flags)
484{
485 vcl_session_msg_t *accepted_msg;
486 int i;
487
488 for (i = 0; i < vec_len (session->accept_evts_fifo); i++)
489 {
490 accepted_msg = &session->accept_evts_fifo[i];
491 if (accepted_msg->accepted_msg.handle == handle)
492 {
Florin Corasb0f662f2018-12-27 14:51:46 -0800493 accepted_msg->flags |= flags;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800494 return 1;
495 }
496 }
497 return 0;
498}
499
Florin Corasc9fbd662018-08-24 12:59:56 -0700500static u32
Florin Coras134a9962018-08-28 11:32:04 -0700501vcl_session_reset_handler (vcl_worker_t * wrk,
502 session_reset_msg_t * reset_msg)
Florin Corasc9fbd662018-08-24 12:59:56 -0700503{
504 vcl_session_t *session;
505 u32 sid;
506
Florin Coras134a9962018-08-28 11:32:04 -0700507 sid = vcl_session_index_from_vpp_handle (wrk, reset_msg->handle);
508 session = vcl_session_get (wrk, sid);
Florin Corasc9fbd662018-08-24 12:59:56 -0700509 if (!session)
510 {
511 VDBG (0, "request to reset unknown handle 0x%llx", reset_msg->handle);
512 return VCL_INVALID_SESSION_INDEX;
513 }
Florin Coras3c7d4f92018-12-14 11:28:43 -0800514
515 /* Caught a reset before actually accepting the session */
wanghanlin696db202023-08-07 17:23:53 +0800516 if (session->session_state == VCL_STATE_LISTEN ||
517 session->session_state == VCL_STATE_LISTEN_NO_MQ)
Florin Coras3c7d4f92018-12-14 11:28:43 -0800518 {
Florin Coras3c7d4f92018-12-14 11:28:43 -0800519 if (!vcl_flag_accepted_session (session, reset_msg->handle,
520 VCL_ACCEPTED_F_RESET))
521 VDBG (0, "session was not accepted!");
522 return VCL_INVALID_SESSION_INDEX;
523 }
524
Florin Corasc127d5a2020-10-14 16:35:58 -0700525 if (session->session_state != VCL_STATE_CLOSED)
526 session->session_state = VCL_STATE_DISCONNECT;
Yacan Liuf5e0a172022-09-20 14:19:19 +0800527
528 session->flags |= (VCL_SESSION_F_RD_SHUTDOWN | VCL_SESSION_F_WR_SHUTDOWN);
Florin Coras5ffb9642021-12-03 17:24:13 -0800529 VDBG (0, "session %u [0x%llx]: reset", sid, reset_msg->handle);
Florin Corasc9fbd662018-08-24 12:59:56 -0700530 return sid;
531}
532
Florin Coras60116992018-08-27 09:52:18 -0700533static u32
Florin Coras134a9962018-08-28 11:32:04 -0700534vcl_session_bound_handler (vcl_worker_t * wrk, session_bound_msg_t * mp)
Florin Coras60116992018-08-27 09:52:18 -0700535{
536 vcl_session_t *session;
537 u32 sid = mp->context;
538
Florin Coras134a9962018-08-28 11:32:04 -0700539 session = vcl_session_get (wrk, sid);
Florin Coras60116992018-08-27 09:52:18 -0700540 if (mp->retval)
541 {
Florin Coras5e062572019-03-14 19:07:51 -0700542 VERR ("session %u [0x%llx]: bind failed: %U", sid, mp->handle,
Florin Coras00e01d32019-10-21 16:07:46 -0700543 format_session_error, mp->retval);
Florin Coras60116992018-08-27 09:52:18 -0700544 if (session)
545 {
Florin Corasc127d5a2020-10-14 16:35:58 -0700546 session->session_state = VCL_STATE_DETACHED;
Florin Coras60116992018-08-27 09:52:18 -0700547 session->vpp_handle = mp->handle;
Mohammed Hawarif4e3ee12023-10-30 10:52:19 +0100548 session->vpp_error = mp->retval;
Florin Coras60116992018-08-27 09:52:18 -0700549 return sid;
550 }
551 else
552 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800553 VDBG (0, "session %u [0x%llx]: Invalid session index!", sid,
554 mp->handle);
Florin Coras60116992018-08-27 09:52:18 -0700555 return VCL_INVALID_SESSION_INDEX;
556 }
557 }
558
559 session->vpp_handle = mp->handle;
560 session->transport.is_ip4 = mp->lcl_is_ip4;
Dave Barach178cf492018-11-13 16:34:13 -0500561 clib_memcpy_fast (&session->transport.lcl_ip, mp->lcl_ip,
562 sizeof (ip46_address_t));
Florin Coras60116992018-08-27 09:52:18 -0700563 session->transport.lcl_port = mp->lcl_port;
Florin Coras134a9962018-08-28 11:32:04 -0700564 vcl_session_table_add_listener (wrk, mp->handle, sid);
Florin Corasc127d5a2020-10-14 16:35:58 -0700565 session->session_state = VCL_STATE_LISTEN;
Florin Coras32881932023-02-06 13:30:13 -0800566 session->flags &= ~VCL_SESSION_F_PENDING_LISTEN;
Florin Coras5f45e012019-01-23 09:21:30 -0800567
Florin Coras6e3c1f82020-01-15 01:30:46 +0000568 if (vcl_session_is_cl (session))
Florin Coras60116992018-08-27 09:52:18 -0700569 {
Florin Corasc547e912020-12-08 17:50:45 -0800570 if (vcl_segment_attach_session (mp->segment_handle, mp->rx_fifo,
Florin Corasf6e284b2021-07-21 18:17:20 -0700571 mp->tx_fifo, mp->vpp_evt_q, mp->mq_index,
572 0, session))
Florin Corasc547e912020-12-08 17:50:45 -0800573 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800574 VDBG (0, "session %u [0x%llx]: failed to attach fifos",
575 session->session_index, session->vpp_handle);
Florin Corasc547e912020-12-08 17:50:45 -0800576 session->session_state = VCL_STATE_DETACHED;
577 return VCL_INVALID_SESSION_INDEX;
578 }
Florin Coras60116992018-08-27 09:52:18 -0700579 }
580
Florin Coras05ecfcc2018-12-12 18:19:39 -0800581 VDBG (0, "session %u [0x%llx]: listen succeeded!", sid, mp->handle);
Florin Coras60116992018-08-27 09:52:18 -0700582 return sid;
583}
584
Florin Corasdfae9f92019-02-20 19:48:31 -0800585static void
586vcl_session_unlisten_reply_handler (vcl_worker_t * wrk, void *data)
587{
588 session_unlisten_reply_msg_t *mp = (session_unlisten_reply_msg_t *) data;
589 vcl_session_t *s;
590
591 s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
Florin Coras0a1e1832020-03-29 18:54:04 +0000592 if (!s)
Florin Corasdfae9f92019-02-20 19:48:31 -0800593 {
594 VDBG (0, "Unlisten reply with wrong handle %llx", mp->handle);
595 return;
596 }
Florin Corasc127d5a2020-10-14 16:35:58 -0700597 if (s->session_state != VCL_STATE_DISCONNECT)
Florin Coras0a1e1832020-03-29 18:54:04 +0000598 {
599 /* Connected udp listener */
600 if (s->session_type == VPPCOM_PROTO_UDP
Florin Corasc127d5a2020-10-14 16:35:58 -0700601 && s->session_state == VCL_STATE_CLOSED)
Florin Coras0a1e1832020-03-29 18:54:04 +0000602 return;
603
604 VDBG (0, "Unlisten session in wrong state %llx", mp->handle);
605 return;
606 }
Florin Corasdfae9f92019-02-20 19:48:31 -0800607
608 if (mp->retval)
609 VDBG (0, "ERROR: session %u [0xllx]: unlisten failed: %U",
Florin Coras00e01d32019-10-21 16:07:46 -0700610 s->session_index, mp->handle, format_session_error, mp->retval);
Florin Corasdfae9f92019-02-20 19:48:31 -0800611
612 if (mp->context != wrk->wrk_index)
613 VDBG (0, "wrong context");
614
615 vcl_session_table_del_vpp_handle (wrk, mp->handle);
616 vcl_session_free (wrk, s);
617}
618
Florin Coras68b7e582020-01-21 18:33:23 -0800619static void
620vcl_session_migrated_handler (vcl_worker_t * wrk, void *data)
621{
622 session_migrated_msg_t *mp = (session_migrated_msg_t *) data;
623 vcl_session_t *s;
Florin Corasc547e912020-12-08 17:50:45 -0800624 u32 fs_index;
Florin Coras68b7e582020-01-21 18:33:23 -0800625
626 s = vcl_session_get_w_vpp_handle (wrk, mp->handle);
627 if (!s)
628 {
629 VDBG (0, "Migrated notification with wrong handle %llx", mp->handle);
630 return;
631 }
632
Florin Coras1bb74942021-02-10 15:26:37 -0800633 /* Only validate if a value is provided */
634 if (mp->segment_handle != SESSION_INVALID_HANDLE)
Florin Corasc547e912020-12-08 17:50:45 -0800635 {
Florin Coras1bb74942021-02-10 15:26:37 -0800636 fs_index = vcl_segment_table_lookup (mp->segment_handle);
637 if (fs_index == VCL_INVALID_SEGMENT_INDEX)
638 {
639 VDBG (0, "segment %lx for session %u is not mounted!",
640 mp->segment_handle, s->session_index);
641 s->session_state = VCL_STATE_DETACHED;
642 return;
643 }
Florin Corasc547e912020-12-08 17:50:45 -0800644 }
645
Florin Coras57660d92020-04-04 22:45:34 +0000646 s->vpp_handle = mp->new_handle;
Florin Corasb4624182020-12-11 13:58:12 -0800647
648 vcl_segment_attach_mq (vcl_vpp_worker_segment_handle (0), mp->vpp_evt_q,
649 mp->vpp_thread_index, &s->vpp_evt_q);
Florin Coras68b7e582020-01-21 18:33:23 -0800650
Florin Coras68b7e582020-01-21 18:33:23 -0800651 vcl_session_table_del_vpp_handle (wrk, mp->handle);
652 vcl_session_table_add_vpp_handle (wrk, mp->new_handle, s->session_index);
653
654 /* Generate new tx event if we have outstanding data */
655 if (svm_fifo_has_event (s->tx_fifo))
Florin Corasc547e912020-12-08 17:50:45 -0800656 app_send_io_evt_to_vpp (s->vpp_evt_q,
657 s->tx_fifo->shr->master_session_index,
Florin Coras68b7e582020-01-21 18:33:23 -0800658 SESSION_IO_EVT_TX, SVM_Q_WAIT);
659
Florin Coras57660d92020-04-04 22:45:34 +0000660 VDBG (0, "Migrated 0x%lx to thread %u 0x%lx", mp->handle,
Florin Coras52dd29f2020-11-18 19:02:17 -0800661 mp->vpp_thread_index, mp->new_handle);
Florin Coras68b7e582020-01-21 18:33:23 -0800662}
663
Florin Coras3c7d4f92018-12-14 11:28:43 -0800664static vcl_session_t *
665vcl_session_accepted (vcl_worker_t * wrk, session_accepted_msg_t * msg)
666{
667 vcl_session_msg_t *vcl_msg;
668 vcl_session_t *session;
669
670 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
671 if (PREDICT_FALSE (session != 0))
Florin Corasb0f662f2018-12-27 14:51:46 -0800672 VWRN ("session overlap handle %lu state %u!", msg->handle,
673 session->session_state);
Florin Coras3c7d4f92018-12-14 11:28:43 -0800674
675 session = vcl_session_table_lookup_listener (wrk, msg->listener_handle);
676 if (!session)
677 {
678 VERR ("couldn't find listen session: listener handle %llx",
679 msg->listener_handle);
680 return 0;
681 }
682
683 clib_fifo_add2 (session->accept_evts_fifo, vcl_msg);
Florin Corase88845e2020-02-13 20:04:28 +0000684 vcl_msg->flags = 0;
Florin Coras3c7d4f92018-12-14 11:28:43 -0800685 vcl_msg->accepted_msg = *msg;
686 /* Session handle points to listener until fully accepted by app */
687 vcl_session_table_add_vpp_handle (wrk, msg->handle, session->session_index);
688
689 return session;
690}
691
692static vcl_session_t *
693vcl_session_disconnected_handler (vcl_worker_t * wrk,
694 session_disconnected_msg_t * msg)
695{
696 vcl_session_t *session;
697
698 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
699 if (!session)
700 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800701 VWRN ("request to disconnect unknown handle 0x%llx", msg->handle);
Florin Coras3c7d4f92018-12-14 11:28:43 -0800702 return 0;
703 }
704
Florin Corasadcfb152020-02-12 08:50:29 +0000705 /* Late disconnect notification on a session that has been closed */
Florin Corasc127d5a2020-10-14 16:35:58 -0700706 if (session->session_state == VCL_STATE_CLOSED)
Florin Corasadcfb152020-02-12 08:50:29 +0000707 return 0;
708
Florin Coras3c7d4f92018-12-14 11:28:43 -0800709 /* Caught a disconnect before actually accepting the session */
wanghanlin696db202023-08-07 17:23:53 +0800710 if (session->session_state == VCL_STATE_LISTEN ||
711 session->session_state == VCL_STATE_LISTEN_NO_MQ)
Florin Coras3c7d4f92018-12-14 11:28:43 -0800712 {
Florin Coras3c7d4f92018-12-14 11:28:43 -0800713 if (!vcl_flag_accepted_session (session, msg->handle,
714 VCL_ACCEPTED_F_CLOSED))
715 VDBG (0, "session was not accepted!");
716 return 0;
717 }
718
Florin Corasadcfb152020-02-12 08:50:29 +0000719 /* If not already reset change state */
Florin Corasc127d5a2020-10-14 16:35:58 -0700720 if (session->session_state != VCL_STATE_DISCONNECT)
721 session->session_state = VCL_STATE_VPP_CLOSING;
Florin Corasadcfb152020-02-12 08:50:29 +0000722
Florin Coras3c7d4f92018-12-14 11:28:43 -0800723 return session;
724}
725
liuyacan534468e2021-05-09 03:50:40 +0000726int
liuyacan55c952e2021-06-13 14:54:55 +0800727vppcom_session_shutdown (uint32_t session_handle, int how)
liuyacan534468e2021-05-09 03:50:40 +0000728{
729 vcl_worker_t *wrk = vcl_worker_get_current ();
730 vcl_session_t *session;
731 vcl_session_state_t state;
732 u64 vpp_handle;
733
734 session = vcl_session_get_w_handle (wrk, session_handle);
735 if (PREDICT_FALSE (!session))
736 return VPPCOM_EBADFD;
737
738 vpp_handle = session->vpp_handle;
739 state = session->session_state;
740
741 VDBG (1, "session %u [0x%llx] state 0x%x (%s)", session->session_index,
Florin Coras60687192021-11-29 21:00:47 -0800742 vpp_handle, state, vcl_session_state_str (state));
liuyacan534468e2021-05-09 03:50:40 +0000743
744 if (PREDICT_FALSE (state == VCL_STATE_LISTEN))
745 {
746 VDBG (0, "ERROR: Cannot shutdown a listen socket!");
747 return VPPCOM_EBADFD;
748 }
749
liuyacan55c952e2021-06-13 14:54:55 +0800750 if (how == SHUT_RD || how == SHUT_RDWR)
751 {
752 session->flags |= VCL_SESSION_F_RD_SHUTDOWN;
753 if (how == SHUT_RD)
754 return VPPCOM_OK;
755 }
756 session->flags |= VCL_SESSION_F_WR_SHUTDOWN;
757
liuyacan534468e2021-05-09 03:50:40 +0000758 if (PREDICT_TRUE (state == VCL_STATE_READY))
759 {
760 VDBG (1, "session %u [0x%llx]: sending shutdown...",
761 session->session_index, vpp_handle);
762
763 vcl_send_session_shutdown (wrk, session);
liuyacan534468e2021-05-09 03:50:40 +0000764 }
765
766 return VPPCOM_OK;
767}
768
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700769static int
770vppcom_session_disconnect (u32 session_handle)
771{
772 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700773 vcl_session_t *session, *listen_session;
774 vcl_session_state_t state;
775 u64 vpp_handle;
776
777 session = vcl_session_get_w_handle (wrk, session_handle);
778 if (!session)
779 return VPPCOM_EBADFD;
780
781 vpp_handle = session->vpp_handle;
782 state = session->session_state;
783
Florin Coras5ffb9642021-12-03 17:24:13 -0800784 VDBG (1, "session %u [0x%llx]: disconnecting state (%s)",
785 session->session_index, vpp_handle, vcl_session_state_str (state));
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700786
787 if (PREDICT_FALSE (state == VCL_STATE_LISTEN))
788 {
789 VDBG (0, "ERROR: Cannot disconnect a listen socket!");
790 return VPPCOM_EBADFD;
791 }
792
793 if (state == VCL_STATE_VPP_CLOSING)
794 {
Florin Coras52dd29f2020-11-18 19:02:17 -0800795 vcl_send_session_disconnected_reply (wrk, session, 0);
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700796 VDBG (1, "session %u [0x%llx]: sending disconnect REPLY...",
797 session->session_index, vpp_handle);
798 }
799 else
800 {
801 /* Session doesn't have an event queue yet. Probably a non-blocking
802 * connect. Wait for the reply */
803 if (PREDICT_FALSE (!session->vpp_evt_q))
804 return VPPCOM_OK;
805
Florin Coras5ffb9642021-12-03 17:24:13 -0800806 VDBG (1, "session %u [0x%llx]: sending disconnect",
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700807 session->session_index, vpp_handle);
808 vcl_send_session_disconnect (wrk, session);
809 }
810
811 if (session->listener_index != VCL_INVALID_SESSION_INDEX)
812 {
813 listen_session = vcl_session_get (wrk, session->listener_index);
Florin Corasb52bd3a2022-06-28 20:01:20 -0700814 if (listen_session)
815 listen_session->n_accepted_sessions--;
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700816 }
817
818 return VPPCOM_OK;
819}
820
Florin Coras30e79c22019-01-02 19:31:22 -0800821static void
Florin Coras9ace36d2019-10-28 13:14:17 -0700822vcl_session_cleanup_handler (vcl_worker_t * wrk, void *data)
823{
824 session_cleanup_msg_t *msg;
825 vcl_session_t *session;
826
827 msg = (session_cleanup_msg_t *) data;
828 session = vcl_session_get_w_vpp_handle (wrk, msg->handle);
829 if (!session)
830 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800831 VWRN ("disconnect confirmed for unknown handle 0x%llx", msg->handle);
Florin Coras9ace36d2019-10-28 13:14:17 -0700832 return;
833 }
834
Florin Coras36d49392020-04-24 23:00:11 +0000835 if (msg->type == SESSION_CLEANUP_TRANSPORT)
836 {
837 /* Transport was cleaned up before we confirmed close. Probably the
838 * app is still waiting for some data that cannot be delivered.
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700839 * Confirm close to make sure everything is cleaned up.
840 * Move to undetermined state to ensure that the session is not
841 * removed before both vpp and the app cleanup.
842 * - If the app closes first, the session is moved to CLOSED state
843 * and the session cleanup notification from vpp removes the
844 * session.
845 * - If vpp cleans up the session first, the session is moved to
846 * DETACHED state lower and subsequently the close from the app
847 * frees the session
848 */
849 if (session->session_state == VCL_STATE_VPP_CLOSING)
Florin Coras0ff7eec2020-08-03 18:55:40 -0700850 {
Florin Coras2bd8b3a2020-10-25 20:28:23 -0700851 vppcom_session_disconnect (vcl_session_handle (session));
852 session->session_state = VCL_STATE_UPDATED;
853 }
854 else if (session->session_state == VCL_STATE_DISCONNECT)
855 {
Florin Coras52dd29f2020-11-18 19:02:17 -0800856 vcl_send_session_reset_reply (wrk, session, 0);
Florin Coras0ff7eec2020-08-03 18:55:40 -0700857 session->session_state = VCL_STATE_UPDATED;
858 }
Florin Coras36d49392020-04-24 23:00:11 +0000859 return;
860 }
861
Florin Corasf0fe1ea2021-12-03 17:03:37 -0800862 /* VPP will reuse the handle so clean it up now */
Florin Coras9ace36d2019-10-28 13:14:17 -0700863 vcl_session_table_del_vpp_handle (wrk, msg->handle);
Florin Corasf0fe1ea2021-12-03 17:03:37 -0800864
865 /* App did not close the connection yet so don't free it. */
Florin Corasc127d5a2020-10-14 16:35:58 -0700866 if (session->session_state != VCL_STATE_CLOSED)
Florin Corasadcfb152020-02-12 08:50:29 +0000867 {
Florin Coras5ffb9642021-12-03 17:24:13 -0800868 VDBG (0, "session %u: app did not close", session->session_index);
Florin Corasc127d5a2020-10-14 16:35:58 -0700869 session->session_state = VCL_STATE_DETACHED;
Florin Corasadcfb152020-02-12 08:50:29 +0000870 session->vpp_handle = VCL_INVALID_SESSION_HANDLE;
871 return;
872 }
Florin Corasf0fe1ea2021-12-03 17:03:37 -0800873
874 /* Session probably tracked with epoll, disconnect not yet handled and
875 * 1) both transport and session cleanup completed 2) app closed. Wait
876 * until message is drained to free the session.
877 * See @ref vcl_handle_mq_event */
878 if (session->flags & VCL_SESSION_F_PENDING_DISCONNECT)
879 {
880 session->flags |= VCL_SESSION_F_PENDING_FREE;
881 return;
882 }
883
Florin Coras9ace36d2019-10-28 13:14:17 -0700884 vcl_session_free (wrk, session);
885}
886
887static void
Florin Coras30e79c22019-01-02 19:31:22 -0800888vcl_session_req_worker_update_handler (vcl_worker_t * wrk, void *data)
889{
890 session_req_worker_update_msg_t *msg;
891 vcl_session_t *s;
892
893 msg = (session_req_worker_update_msg_t *) data;
894 s = vcl_session_get_w_vpp_handle (wrk, msg->session_handle);
895 if (!s)
896 return;
897
898 vec_add1 (wrk->pending_session_wrk_updates, s->session_index);
899}
900
901static void
902vcl_session_worker_update_reply_handler (vcl_worker_t * wrk, void *data)
903{
904 session_worker_update_reply_msg_t *msg;
905 vcl_session_t *s;
906
907 msg = (session_worker_update_reply_msg_t *) data;
908 s = vcl_session_get_w_vpp_handle (wrk, msg->handle);
909 if (!s)
910 {
911 VDBG (0, "unknown handle 0x%llx", msg->handle);
912 return;
913 }
Florin Coras30e79c22019-01-02 19:31:22 -0800914
Florin Coras5f45e012019-01-23 09:21:30 -0800915 if (s->rx_fifo)
916 {
Florin Corasc547e912020-12-08 17:50:45 -0800917 if (vcl_segment_attach_session (msg->segment_handle, msg->rx_fifo,
Florin Corasf6e284b2021-07-21 18:17:20 -0700918 msg->tx_fifo, (uword) ~0, ~0, 0, s))
Florin Corasc547e912020-12-08 17:50:45 -0800919 {
920 VDBG (0, "failed to attach fifos for %u", s->session_index);
921 return;
922 }
Florin Coras5f45e012019-01-23 09:21:30 -0800923 }
Florin Corasc127d5a2020-10-14 16:35:58 -0700924 s->session_state = VCL_STATE_UPDATED;
Florin Coras30e79c22019-01-02 19:31:22 -0800925
Florin Coras30e79c22019-01-02 19:31:22 -0800926 VDBG (0, "session %u[0x%llx] moved to worker %u", s->session_index,
927 s->vpp_handle, wrk->wrk_index);
928}
929
Florin Coras935ce752020-09-08 22:43:47 -0700930static int
931vcl_api_recv_fd (vcl_worker_t * wrk, int *fds, int n_fds)
932{
933
934 if (vcm->cfg.vpp_app_socket_api)
935 return vcl_sapi_recv_fds (wrk, fds, n_fds);
936
937 return vcl_bapi_recv_fds (wrk, fds, n_fds);
938}
939
Florin Corasc4c4cf52019-08-24 18:17:34 -0700940static void
941vcl_session_app_add_segment_handler (vcl_worker_t * wrk, void *data)
942{
943 ssvm_segment_type_t seg_type = SSVM_SEGMENT_SHM;
944 session_app_add_segment_msg_t *msg;
945 u64 segment_handle;
946 int fd = -1;
947
948 msg = (session_app_add_segment_msg_t *) data;
949
950 if (msg->fd_flags)
951 {
Florin Coras935ce752020-09-08 22:43:47 -0700952 vcl_api_recv_fd (wrk, &fd, 1);
Florin Corasc4c4cf52019-08-24 18:17:34 -0700953 seg_type = SSVM_SEGMENT_MEMFD;
954 }
955
956 segment_handle = msg->segment_handle;
957 if (segment_handle == VCL_INVALID_SEGMENT_HANDLE)
958 {
959 clib_warning ("invalid segment handle");
960 return;
961 }
962
963 if (vcl_segment_attach (segment_handle, (char *) msg->segment_name,
964 seg_type, fd))
965 {
966 VDBG (0, "vcl_segment_attach ('%s') failed", msg->segment_name);
967 return;
968 }
969
970 VDBG (1, "mapped new segment '%s' size %d", msg->segment_name,
971 msg->segment_size);
972}
973
974static void
975vcl_session_app_del_segment_handler (vcl_worker_t * wrk, void *data)
976{
977 session_app_del_segment_msg_t *msg = (session_app_del_segment_msg_t *) data;
978 vcl_segment_detach (msg->segment_handle);
Florin Coras1f40ab42023-06-13 17:18:56 -0700979 VDBG (1, "Unmapped segment: %lx", msg->segment_handle);
Florin Corasc4c4cf52019-08-24 18:17:34 -0700980}
981
Florin Coras40c07ce2020-07-16 20:46:17 -0700982static void
983vcl_worker_rpc_handler (vcl_worker_t * wrk, void *data)
984{
985 if (!vcm->wrk_rpc_fn)
986 return;
987
hanlina3a48962020-07-13 11:09:15 +0800988 (vcm->wrk_rpc_fn) (((session_app_wrk_rpc_msg_t *) data)->data);
Florin Coras40c07ce2020-07-16 20:46:17 -0700989}
990
Florin Coras04ae8272021-04-12 19:55:37 -0700991static void
992vcl_session_transport_attr_reply_handler (vcl_worker_t *wrk, void *data)
993{
994 session_transport_attr_reply_msg_t *mp;
995
996 if (!wrk->session_attr_op)
997 return;
998
999 mp = (session_transport_attr_reply_msg_t *) data;
1000
1001 wrk->session_attr_op_rv = mp->retval;
1002 wrk->session_attr_op = 0;
1003 wrk->session_attr_rv = mp->attr;
1004}
1005
Florin Coras86f04502018-09-12 16:08:01 -07001006static int
1007vcl_handle_mq_event (vcl_worker_t * wrk, session_event_t * e)
Florin Coras54693d22018-07-17 10:46:29 -07001008{
Florin Coras54693d22018-07-17 10:46:29 -07001009 session_disconnected_msg_t *disconnected_msg;
Florin Corasb242d312020-10-26 15:35:40 -07001010 session_connected_msg_t *connected_msg;
1011 session_reset_msg_t *reset_msg;
1012 session_event_t *ecpy;
Florin Corasc127d5a2020-10-14 16:35:58 -07001013 vcl_session_t *s;
Florin Corasb242d312020-10-26 15:35:40 -07001014 u32 sid;
Florin Coras54693d22018-07-17 10:46:29 -07001015
1016 switch (e->event_type)
1017 {
Florin Coras653e43f2019-03-04 10:56:23 -08001018 case SESSION_IO_EVT_RX:
1019 case SESSION_IO_EVT_TX:
Florin Corasc127d5a2020-10-14 16:35:58 -07001020 s = vcl_session_get (wrk, e->session_index);
Florin Coras1bc919e2020-10-18 20:17:49 -07001021 if (!s || !vcl_session_is_open (s))
Florin Coras653e43f2019-03-04 10:56:23 -08001022 break;
Florin Coras86f04502018-09-12 16:08:01 -07001023 vec_add1 (wrk->unhandled_evts_vector, *e);
Florin Coras54693d22018-07-17 10:46:29 -07001024 break;
Florin Corasb242d312020-10-26 15:35:40 -07001025 case SESSION_CTRL_EVT_BOUND:
1026 /* We can only wait for only one listen so not postponed */
1027 vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
1028 break;
Florin Coras54693d22018-07-17 10:46:29 -07001029 case SESSION_CTRL_EVT_ACCEPTED:
Florin Corasb242d312020-10-26 15:35:40 -07001030 s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
1031 if (vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK))
1032 {
1033 vec_add2 (wrk->unhandled_evts_vector, ecpy, 1);
1034 *ecpy = *e;
1035 ecpy->postponed = 1;
1036 ecpy->session_index = s->session_index;
1037 }
Florin Coras54693d22018-07-17 10:46:29 -07001038 break;
1039 case SESSION_CTRL_EVT_CONNECTED:
Florin Corasb242d312020-10-26 15:35:40 -07001040 connected_msg = (session_connected_msg_t *) e->data;
1041 sid = vcl_session_connected_handler (wrk, connected_msg);
1042 if (!(s = vcl_session_get (wrk, sid)))
1043 break;
1044 if (vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK))
1045 {
1046 vec_add2 (wrk->unhandled_evts_vector, ecpy, 1);
1047 *ecpy = *e;
1048 ecpy->postponed = 1;
1049 ecpy->session_index = s->session_index;
1050 }
Florin Coras54693d22018-07-17 10:46:29 -07001051 break;
1052 case SESSION_CTRL_EVT_DISCONNECTED:
1053 disconnected_msg = (session_disconnected_msg_t *) e->data;
Florin Corasb242d312020-10-26 15:35:40 -07001054 if (!(s = vcl_session_get_w_vpp_handle (wrk, disconnected_msg->handle)))
1055 break;
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001056 if (s->session_state == VCL_STATE_CLOSED)
1057 break;
wanghanlin4747b342023-08-08 10:40:04 +08001058 /* We do not postpone for blocking sessions or listen sessions because:
1059 * 1. Blocking sessions are not part of epoll instead they're used in a
1060 * synchronous manner, such as read/write and etc.
1061 * 2. Listen sessions that have not yet been accepted can't change to
1062 * VPP_CLOSING state instead can been marked as ACCEPTED_F_CLOSED.
1063 */
1064 if (vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK) &&
1065 !(s->session_state == VCL_STATE_LISTEN ||
1066 s->session_state == VCL_STATE_LISTEN_NO_MQ))
Florin Corasb242d312020-10-26 15:35:40 -07001067 {
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001068 s->session_state = VCL_STATE_VPP_CLOSING;
1069 s->flags |= VCL_SESSION_F_PENDING_DISCONNECT;
1070 vec_add2 (wrk->unhandled_evts_vector, ecpy, 1);
1071 *ecpy = *e;
1072 ecpy->postponed = 1;
1073 ecpy->session_index = s->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07001074 break;
1075 }
1076 if (!(s = vcl_session_disconnected_handler (wrk, disconnected_msg)))
Florin Coras3c7d4f92018-12-14 11:28:43 -08001077 break;
Florin Corasc127d5a2020-10-14 16:35:58 -07001078 VDBG (0, "disconnected session %u [0x%llx]", s->session_index,
1079 s->vpp_handle);
Florin Corasc9fbd662018-08-24 12:59:56 -07001080 break;
1081 case SESSION_CTRL_EVT_RESET:
Florin Corasb242d312020-10-26 15:35:40 -07001082 reset_msg = (session_reset_msg_t *) e->data;
1083 if (!(s = vcl_session_get_w_vpp_handle (wrk, reset_msg->handle)))
1084 break;
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001085 if (s->session_state == VCL_STATE_CLOSED)
1086 break;
wanghanlin4747b342023-08-08 10:40:04 +08001087 /* We do not postpone for blocking sessions or listen sessions because:
1088 * 1. Blocking sessions are not part of epoll instead they're used in a
1089 * synchronous manner, such as read/write and etc.
1090 * 2. Listen sessions that have not yet been accepted can't change to
1091 * DISCONNECT state instead can been marked as ACCEPTED_F_RESET.
1092 */
1093 if (vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK) &&
1094 !(s->session_state == VCL_STATE_LISTEN ||
1095 s->session_state == VCL_STATE_LISTEN_NO_MQ))
Florin Corasb242d312020-10-26 15:35:40 -07001096 {
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001097 s->flags |= VCL_SESSION_F_PENDING_DISCONNECT;
1098 s->session_state = VCL_STATE_DISCONNECT;
Yacan Liuf5e0a172022-09-20 14:19:19 +08001099 s->flags |= (VCL_SESSION_F_RD_SHUTDOWN | VCL_SESSION_F_WR_SHUTDOWN);
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001100 vec_add2 (wrk->unhandled_evts_vector, ecpy, 1);
1101 *ecpy = *e;
1102 ecpy->postponed = 1;
1103 ecpy->session_index = s->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07001104 break;
1105 }
Florin Coras134a9962018-08-28 11:32:04 -07001106 vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
Florin Coras60116992018-08-27 09:52:18 -07001107 break;
Florin Corasdfae9f92019-02-20 19:48:31 -08001108 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
1109 vcl_session_unlisten_reply_handler (wrk, e->data);
1110 break;
Florin Coras68b7e582020-01-21 18:33:23 -08001111 case SESSION_CTRL_EVT_MIGRATED:
1112 vcl_session_migrated_handler (wrk, e->data);
1113 break;
Florin Coras9ace36d2019-10-28 13:14:17 -07001114 case SESSION_CTRL_EVT_CLEANUP:
1115 vcl_session_cleanup_handler (wrk, e->data);
1116 break;
Florin Coras30e79c22019-01-02 19:31:22 -08001117 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
1118 vcl_session_req_worker_update_handler (wrk, e->data);
1119 break;
1120 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
1121 vcl_session_worker_update_reply_handler (wrk, e->data);
1122 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -07001123 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
1124 vcl_session_app_add_segment_handler (wrk, e->data);
1125 break;
1126 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
1127 vcl_session_app_del_segment_handler (wrk, e->data);
1128 break;
hanlina3a48962020-07-13 11:09:15 +08001129 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -07001130 vcl_worker_rpc_handler (wrk, e->data);
1131 break;
Florin Coras04ae8272021-04-12 19:55:37 -07001132 case SESSION_CTRL_EVT_TRANSPORT_ATTR_REPLY:
1133 vcl_session_transport_attr_reply_handler (wrk, e->data);
1134 break;
Florin Coras54693d22018-07-17 10:46:29 -07001135 default:
1136 clib_warning ("unhandled %u", e->event_type);
1137 }
1138 return VPPCOM_OK;
1139}
1140
Florin Coras30e79c22019-01-02 19:31:22 -08001141static int
Florin Coras697faea2018-06-27 17:10:49 -07001142vppcom_wait_for_session_state_change (u32 session_index,
Florin Coras288eaab2019-02-03 15:26:14 -08001143 vcl_session_state_t state,
Florin Coras697faea2018-06-27 17:10:49 -07001144 f64 wait_for_time)
1145{
Florin Coras134a9962018-08-28 11:32:04 -07001146 vcl_worker_t *wrk = vcl_worker_get_current ();
1147 f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
Florin Coras697faea2018-06-27 17:10:49 -07001148 vcl_session_t *volatile session;
Florin Coras54693d22018-07-17 10:46:29 -07001149 svm_msg_q_msg_t msg;
1150 session_event_t *e;
Dave Wallace543852a2017-08-03 02:11:34 -04001151
Florin Coras697faea2018-06-27 17:10:49 -07001152 do
Dave Wallace543852a2017-08-03 02:11:34 -04001153 {
Florin Coras134a9962018-08-28 11:32:04 -07001154 session = vcl_session_get (wrk, session_index);
Florin Coras070453d2018-08-24 17:04:27 -07001155 if (PREDICT_FALSE (!session))
Florin Coras697faea2018-06-27 17:10:49 -07001156 {
Florin Coras070453d2018-08-24 17:04:27 -07001157 return VPPCOM_EBADFD;
Florin Coras697faea2018-06-27 17:10:49 -07001158 }
Florin Corasdfffdd72020-10-15 10:54:47 -07001159 if (session->session_state == state)
Florin Coras697faea2018-06-27 17:10:49 -07001160 {
Florin Coras697faea2018-06-27 17:10:49 -07001161 return VPPCOM_OK;
1162 }
Florin Corasc127d5a2020-10-14 16:35:58 -07001163 if (session->session_state == VCL_STATE_DETACHED)
Florin Coras697faea2018-06-27 17:10:49 -07001164 {
Mohammed Hawarif4e3ee12023-10-30 10:52:19 +01001165 if (session->vpp_error == SESSION_E_ALREADY_LISTENING)
1166 return VPPCOM_EADDRINUSE;
1167 else
1168 return VPPCOM_ECONNREFUSED;
Florin Coras697faea2018-06-27 17:10:49 -07001169 }
Florin Coras54693d22018-07-17 10:46:29 -07001170
Florin Coras134a9962018-08-28 11:32:04 -07001171 if (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0))
Florin Corasdc2e2512018-12-03 17:47:26 -08001172 {
1173 usleep (100);
1174 continue;
1175 }
Florin Coras134a9962018-08-28 11:32:04 -07001176 e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
Florin Coras86f04502018-09-12 16:08:01 -07001177 vcl_handle_mq_event (wrk, e);
Florin Coras134a9962018-08-28 11:32:04 -07001178 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
Florin Corasdcf55ce2017-11-16 15:32:50 -08001179 }
Florin Coras134a9962018-08-28 11:32:04 -07001180 while (clib_time_now (&wrk->clib_time) < timeout);
Florin Corasdcf55ce2017-11-16 15:32:50 -08001181
Florin Coras05ecfcc2018-12-12 18:19:39 -08001182 VDBG (0, "timeout waiting for state 0x%x (%s)", state,
Florin Coras60687192021-11-29 21:00:47 -08001183 vcl_session_state_str (state));
Florin Coras697faea2018-06-27 17:10:49 -07001184 vcl_evt (VCL_EVT_SESSION_TIMEOUT, session, session_state);
Dave Wallace543852a2017-08-03 02:11:34 -04001185
Florin Coras697faea2018-06-27 17:10:49 -07001186 return VPPCOM_ETIMEDOUT;
Dave Wallace60caa062017-11-10 17:07:13 -05001187}
1188
Florin Coras30e79c22019-01-02 19:31:22 -08001189static void
1190vcl_handle_pending_wrk_updates (vcl_worker_t * wrk)
1191{
Florin Coras288eaab2019-02-03 15:26:14 -08001192 vcl_session_state_t state;
Florin Coras30e79c22019-01-02 19:31:22 -08001193 vcl_session_t *s;
1194 u32 *sip;
1195
1196 if (PREDICT_TRUE (vec_len (wrk->pending_session_wrk_updates) == 0))
1197 return;
1198
1199 vec_foreach (sip, wrk->pending_session_wrk_updates)
1200 {
1201 s = vcl_session_get (wrk, *sip);
1202 vcl_send_session_worker_update (wrk, s, wrk->wrk_index);
1203 state = s->session_state;
Florin Corasc127d5a2020-10-14 16:35:58 -07001204 vppcom_wait_for_session_state_change (s->session_index, VCL_STATE_UPDATED,
1205 5);
Florin Coras30e79c22019-01-02 19:31:22 -08001206 s->session_state = state;
1207 }
1208 vec_reset_length (wrk->pending_session_wrk_updates);
1209}
1210
Florin Corasf9240dc2019-01-15 08:03:17 -08001211void
Florin Coras5398dfb2021-01-25 20:31:27 -08001212vcl_worker_flush_mq_events (vcl_worker_t *wrk)
Florin Coras30e79c22019-01-02 19:31:22 -08001213{
Florin Coras30e79c22019-01-02 19:31:22 -08001214 svm_msg_q_msg_t *msg;
1215 session_event_t *e;
1216 svm_msg_q_t *mq;
1217 int i;
1218
1219 mq = wrk->app_event_queue;
Florin Corase003a1b2019-06-05 10:47:16 -07001220 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras30e79c22019-01-02 19:31:22 -08001221
1222 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
1223 {
1224 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
1225 e = svm_msg_q_msg_data (mq, msg);
1226 vcl_handle_mq_event (wrk, e);
1227 svm_msg_q_free_msg (mq, msg);
1228 }
1229 vec_reset_length (wrk->mq_msg_vector);
1230 vcl_handle_pending_wrk_updates (wrk);
1231}
1232
Florin Coras5398dfb2021-01-25 20:31:27 -08001233void
1234vcl_flush_mq_events (void)
1235{
1236 vcl_worker_flush_mq_events (vcl_worker_get_current ());
1237}
1238
Florin Coras697faea2018-06-27 17:10:49 -07001239static int
Florin Corasab2f6db2018-08-31 14:31:41 -07001240vppcom_session_unbind (u32 session_handle)
Dave Wallace543852a2017-08-03 02:11:34 -04001241{
Florin Coras134a9962018-08-28 11:32:04 -07001242 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Corasaaff5ee2019-07-08 13:00:00 -07001243 session_accepted_msg_t *accepted_msg;
Florin Coras7e12d942018-06-27 14:32:43 -07001244 vcl_session_t *session = 0;
Florin Corasaaff5ee2019-07-08 13:00:00 -07001245 vcl_session_msg_t *evt;
Dave Wallace543852a2017-08-03 02:11:34 -04001246
Florin Corasab2f6db2018-08-31 14:31:41 -07001247 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001248 if (!session)
1249 return VPPCOM_EBADFD;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001250
Florin Corasaaff5ee2019-07-08 13:00:00 -07001251 /* Flush pending accept events, if any */
1252 while (clib_fifo_elts (session->accept_evts_fifo))
1253 {
1254 clib_fifo_sub2 (session->accept_evts_fifo, evt);
1255 accepted_msg = &evt->accepted_msg;
1256 vcl_session_table_del_vpp_handle (wrk, accepted_msg->handle);
1257 vcl_send_session_accepted_reply (session->vpp_evt_q,
1258 accepted_msg->context,
wanghanlin785b6ea2020-12-10 19:12:22 +08001259 accepted_msg->handle, -1);
Florin Corasaaff5ee2019-07-08 13:00:00 -07001260 }
1261 clib_fifo_free (session->accept_evts_fifo);
1262
Florin Coras458089b2019-08-21 16:20:44 -07001263 vcl_send_session_unlisten (wrk, session);
Dave Wallace4878cbe2017-11-21 03:45:09 -05001264
Florin Coras32881932023-02-06 13:30:13 -08001265 VDBG (0, "session %u [0x%llx]: sending unbind!", session->session_index,
Florin Coras458089b2019-08-21 16:20:44 -07001266 session->vpp_handle);
Florin Coras0d427d82018-06-27 03:24:07 -07001267 vcl_evt (VCL_EVT_UNBIND, session);
Florin Coras458089b2019-08-21 16:20:44 -07001268
Florin Coras980ee742023-08-04 16:14:01 -07001269 session->vpp_handle = SESSION_INVALID_HANDLE;
Florin Corasc127d5a2020-10-14 16:35:58 -07001270 session->session_state = VCL_STATE_DISCONNECT;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001271
Florin Coras070453d2018-08-24 17:04:27 -07001272 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001273}
1274
Florin Coras940f78f2018-11-30 12:11:20 -08001275/**
1276 * Handle app exit
1277 *
1278 * Notify vpp of the disconnect and mark the worker as free. If we're the
1279 * last worker, do a full cleanup otherwise, since we're probably a forked
1280 * child, avoid syscalls as much as possible. We might've lost privileges.
1281 */
1282void
1283vppcom_app_exit (void)
1284{
1285 if (!pool_elts (vcm->workers))
1286 return;
Florin Coras01f3f892018-12-02 12:45:53 -08001287 vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
1288 vcl_set_worker_index (~0);
Florin Coras940f78f2018-11-30 12:11:20 -08001289 vcl_elog_stop (vcm);
Florin Coras940f78f2018-11-30 12:11:20 -08001290}
1291
Florin Coras935ce752020-09-08 22:43:47 -07001292static int
1293vcl_api_attach (void)
1294{
1295 if (vcm->cfg.vpp_app_socket_api)
1296 return vcl_sapi_attach ();
1297
1298 return vcl_bapi_attach ();
1299}
1300
Maros Ondrejicka0db15752022-10-12 22:58:01 +02001301int
1302vcl_is_first_reattach_to_execute ()
1303{
1304 if (vcm->reattach_count == 0)
1305 return 1;
1306
1307 return 0;
1308}
1309
1310void
1311vcl_set_reattach_counter ()
1312{
1313 ++vcm->reattach_count;
1314
1315 if (vcm->reattach_count == vec_len (vcm->workers))
1316 vcm->reattach_count = 0;
1317}
1318
1319/**
1320 * Reattach vcl to vpp after it has previously been disconnected.
1321 *
1322 * The logic should be:
1323 * - first worker to hit `vcl_api_retry_attach` should attach to vpp,
1324 * to reproduce the `vcl_api_attach` in `vppcom_app_create`.
1325 * - the rest of the workers should `reproduce vcl_worker_register_with_vpp`
1326 * from `vppcom_worker_register` since they were already allocated.
1327 */
1328
Florin Coras935ce752020-09-08 22:43:47 -07001329static void
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00001330vcl_api_retry_attach (vcl_worker_t *wrk)
1331{
1332 vcl_session_t *s;
1333
Maros Ondrejicka0db15752022-10-12 22:58:01 +02001334 clib_spinlock_lock (&vcm->workers_lock);
1335 if (vcl_is_first_reattach_to_execute ())
1336 {
1337 if (vcl_api_attach ())
1338 {
1339 clib_spinlock_unlock (&vcm->workers_lock);
1340 return;
1341 }
1342 vcl_set_reattach_counter ();
1343 clib_spinlock_unlock (&vcm->workers_lock);
1344 }
1345 else
1346 {
1347 vcl_set_reattach_counter ();
1348 clib_spinlock_unlock (&vcm->workers_lock);
1349 vcl_worker_register_with_vpp ();
1350 }
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00001351
1352 /* Treat listeners as configuration that needs to be re-added to vpp */
1353 pool_foreach (s, wrk->sessions)
1354 {
1355 if (s->flags & VCL_SESSION_F_IS_VEP)
1356 continue;
1357 if (s->session_state == VCL_STATE_LISTEN_NO_MQ)
1358 vppcom_session_listen (vcl_session_handle (s), 10);
1359 else
1360 VDBG (0, "internal error: unexpected state %d", s->session_state);
1361 }
1362}
1363
1364static void
1365vcl_api_handle_disconnect (vcl_worker_t *wrk)
1366{
1367 wrk->api_client_handle = ~0;
1368 vcl_worker_detach_sessions (wrk);
1369}
1370
1371static void
Florin Coras935ce752020-09-08 22:43:47 -07001372vcl_api_detach (vcl_worker_t * wrk)
1373{
Florin Corasd04ea442022-03-30 16:08:25 -07001374 if (wrk->api_client_handle == ~0)
1375 return;
1376
Florin Coras935ce752020-09-08 22:43:47 -07001377 vcl_send_app_detach (wrk);
1378
1379 if (vcm->cfg.vpp_app_socket_api)
1380 return vcl_sapi_detach (wrk);
1381
1382 return vcl_bapi_disconnect_from_vpp ();
1383}
1384
Dave Wallace543852a2017-08-03 02:11:34 -04001385/*
1386 * VPPCOM Public API functions
1387 */
1388int
Florin Coras66ec4672020-06-15 07:59:40 -07001389vppcom_app_create (const char *app_name)
Dave Wallace543852a2017-08-03 02:11:34 -04001390{
Dave Wallace543852a2017-08-03 02:11:34 -04001391 vppcom_cfg_t *vcl_cfg = &vcm->cfg;
Dave Wallace543852a2017-08-03 02:11:34 -04001392 int rv;
1393
Florin Coras47c40e22018-11-26 17:01:36 -08001394 if (vcm->is_init)
Dave Wallace543852a2017-08-03 02:11:34 -04001395 {
Florin Coras955bfbb2018-12-04 13:43:45 -08001396 VDBG (1, "already initialized");
1397 return VPPCOM_EEXIST;
Dave Wallace543852a2017-08-03 02:11:34 -04001398 }
1399
Florin Coras47c40e22018-11-26 17:01:36 -08001400 vcm->is_init = 1;
1401 vppcom_cfg (&vcm->cfg);
1402 vcl_cfg = &vcm->cfg;
1403
1404 vcm->main_cpu = pthread_self ();
1405 vcm->main_pid = getpid ();
1406 vcm->app_name = format (0, "%s", app_name);
Florin Coras41bc8612021-10-01 14:57:03 -07001407 fifo_segment_main_init (&vcm->segment_main, (uword) ~0,
1408 20 /* timeout in secs */);
Florin Coras47c40e22018-11-26 17:01:36 -08001409 pool_alloc (vcm->workers, vcl_cfg->max_workers);
1410 clib_spinlock_init (&vcm->workers_lock);
Florin Corasd85de682018-11-29 17:02:29 -08001411 clib_rwlock_init (&vcm->segment_table_lock);
Florin Coras940f78f2018-11-30 12:11:20 -08001412 atexit (vppcom_app_exit);
Florin Corasb88de902020-09-08 16:47:57 -07001413 vcl_elog_init (vcm);
Florin Coras47c40e22018-11-26 17:01:36 -08001414
1415 /* Allocate default worker */
1416 vcl_worker_alloc_and_init ();
1417
Florin Coras935ce752020-09-08 22:43:47 -07001418 if ((rv = vcl_api_attach ()))
Florin Corasd04ea442022-03-30 16:08:25 -07001419 {
1420 vppcom_app_destroy ();
1421 return rv;
1422 }
Florin Coras47c40e22018-11-26 17:01:36 -08001423
1424 VDBG (0, "app_name '%s', my_client_index %d (0x%x)", app_name,
Florin Corascc7c88e2020-09-15 15:56:51 -07001425 vcm->workers[0].api_client_handle, vcm->workers[0].api_client_handle);
Dave Wallace543852a2017-08-03 02:11:34 -04001426
1427 return VPPCOM_OK;
1428}
1429
1430void
1431vppcom_app_destroy (void)
1432{
Florin Corasdc0ded72020-04-30 02:59:55 +00001433 vcl_worker_t *wrk, *current_wrk;
Damjan Marion4537c302020-09-28 19:03:37 +02001434 void *heap;
Dave Wallace543852a2017-08-03 02:11:34 -04001435
Florin Coras940f78f2018-11-30 12:11:20 -08001436 if (!pool_elts (vcm->workers))
1437 return;
1438
Florin Coras0d427d82018-06-27 03:24:07 -07001439 vcl_evt (VCL_EVT_DETACH, vcm);
Keith Burns (alagalah)8aa9aaf2018-01-05 12:16:22 -08001440
Florin Corasdc0ded72020-04-30 02:59:55 +00001441 current_wrk = vcl_worker_get_current ();
Keith Burns (alagalah)8aa9aaf2018-01-05 12:16:22 -08001442
Damjan Marionb2c31b62020-12-13 21:47:40 +01001443 pool_foreach (wrk, vcm->workers) {
Florin Corasdc0ded72020-04-30 02:59:55 +00001444 if (current_wrk != wrk)
1445 vcl_worker_cleanup (wrk, 0 /* notify vpp */ );
Damjan Marionb2c31b62020-12-13 21:47:40 +01001446 }
Florin Corasce815de2020-04-16 18:47:27 +00001447
Florin Coras935ce752020-09-08 22:43:47 -07001448 vcl_api_detach (current_wrk);
Florin Corasdc0ded72020-04-30 02:59:55 +00001449 vcl_worker_cleanup (current_wrk, 0 /* notify vpp */ );
Florin Corasd04ea442022-03-30 16:08:25 -07001450 vcl_set_worker_index (~0);
Florin Corasdc0ded72020-04-30 02:59:55 +00001451
Florin Coras0d427d82018-06-27 03:24:07 -07001452 vcl_elog_stop (vcm);
Florin Corasce815de2020-04-16 18:47:27 +00001453
1454 /*
1455 * Free the heap and fix vcm
1456 */
1457 heap = clib_mem_get_heap ();
Damjan Marion4537c302020-09-28 19:03:37 +02001458 munmap (clib_mem_get_heap_base (heap), clib_mem_get_heap_size (heap));
Florin Corasce815de2020-04-16 18:47:27 +00001459
1460 vcm = &_vppcom_main;
Florin Coras69d97252020-05-11 17:19:31 +00001461 vcm->is_init = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001462}
1463
1464int
Dave Wallacec04cbf12018-02-07 18:14:02 -05001465vppcom_session_create (u8 proto, u8 is_nonblocking)
Dave Wallace543852a2017-08-03 02:11:34 -04001466{
Florin Coras134a9962018-08-28 11:32:04 -07001467 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001468 vcl_session_t *session;
Dave Wallace543852a2017-08-03 02:11:34 -04001469
Florin Coras134a9962018-08-28 11:32:04 -07001470 session = vcl_session_alloc (wrk);
Dave Wallace543852a2017-08-03 02:11:34 -04001471
Florin Coras7e12d942018-06-27 14:32:43 -07001472 session->session_type = proto;
Florin Corasc127d5a2020-10-14 16:35:58 -07001473 session->session_state = VCL_STATE_CLOSED;
Florin Coras980ee742023-08-04 16:14:01 -07001474 session->vpp_handle = SESSION_INVALID_HANDLE;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001475 session->is_dgram = vcl_proto_is_dgram (proto);
Radha krishna Saragadamdd92bde2022-07-18 19:41:05 +05301476 session->vpp_error = SESSION_E_NONE;
Dave Wallace543852a2017-08-03 02:11:34 -04001477
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001478 if (is_nonblocking)
Florin Corasac422d62020-10-19 20:51:36 -07001479 vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
Dave Wallace543852a2017-08-03 02:11:34 -04001480
Florin Coras7e12d942018-06-27 14:32:43 -07001481 vcl_evt (VCL_EVT_CREATE, session, session_type, session->session_state,
1482 is_nonblocking, session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001483
Florin Coras5e062572019-03-14 19:07:51 -07001484 VDBG (0, "created session %u", session->session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001485
Florin Coras134a9962018-08-28 11:32:04 -07001486 return vcl_session_handle (session);
Dave Wallace543852a2017-08-03 02:11:34 -04001487}
1488
Florin Corasfe286f72021-06-04 10:07:55 -07001489static void
Florin Corasfa3884f2021-06-22 20:04:46 -07001490vcl_epoll_lt_add (vcl_worker_t *wrk, vcl_session_t *s)
Florin Corasfe286f72021-06-04 10:07:55 -07001491{
Florin Corasfa3884f2021-06-22 20:04:46 -07001492 vcl_session_t *cur, *prev;
Florin Corasfe286f72021-06-04 10:07:55 -07001493
Florin Corasc2a14172023-02-28 21:13:50 -08001494 ASSERT (s->vep.lt_next == VCL_INVALID_SESSION_INDEX);
1495
Florin Corasfa3884f2021-06-22 20:04:46 -07001496 if (wrk->ep_lt_current == VCL_INVALID_SESSION_INDEX)
Florin Corasfe286f72021-06-04 10:07:55 -07001497 {
Florin Corasfa3884f2021-06-22 20:04:46 -07001498 wrk->ep_lt_current = s->session_index;
1499 s->vep.lt_next = s->session_index;
1500 s->vep.lt_prev = s->session_index;
1501 return;
Florin Corasfe286f72021-06-04 10:07:55 -07001502 }
Florin Corasfa3884f2021-06-22 20:04:46 -07001503
1504 cur = vcl_session_get (wrk, wrk->ep_lt_current);
1505 prev = vcl_session_get (wrk, cur->vep.lt_prev);
1506
1507 prev->vep.lt_next = s->session_index;
1508 s->vep.lt_prev = prev->session_index;
1509
1510 s->vep.lt_next = cur->session_index;
1511 cur->vep.lt_prev = s->session_index;
1512}
1513
1514static void
1515vcl_epoll_lt_del (vcl_worker_t *wrk, vcl_session_t *s)
1516{
1517 vcl_session_t *prev, *next;
1518
Florin Corasc2a14172023-02-28 21:13:50 -08001519 ASSERT (s->vep.lt_next != VCL_INVALID_SESSION_INDEX);
1520
Florin Corasfa3884f2021-06-22 20:04:46 -07001521 if (s->vep.lt_next == s->session_index)
1522 {
1523 wrk->ep_lt_current = VCL_INVALID_SESSION_INDEX;
1524 s->vep.lt_next = VCL_INVALID_SESSION_INDEX;
Florin Corasc2a14172023-02-28 21:13:50 -08001525 s->vep.lt_prev = VCL_INVALID_SESSION_INDEX;
Florin Corasfa3884f2021-06-22 20:04:46 -07001526 return;
1527 }
1528
1529 prev = vcl_session_get (wrk, s->vep.lt_prev);
1530 next = vcl_session_get (wrk, s->vep.lt_next);
1531
1532 prev->vep.lt_next = next->session_index;
1533 next->vep.lt_prev = prev->session_index;
1534
1535 if (s->session_index == wrk->ep_lt_current)
1536 wrk->ep_lt_current = s->vep.lt_next;
1537
1538 s->vep.lt_next = VCL_INVALID_SESSION_INDEX;
Florin Corasc2a14172023-02-28 21:13:50 -08001539 s->vep.lt_prev = VCL_INVALID_SESSION_INDEX;
Florin Corasfe286f72021-06-04 10:07:55 -07001540}
1541
Dave Wallace543852a2017-08-03 02:11:34 -04001542int
Florin Corasc127d5a2020-10-14 16:35:58 -07001543vcl_session_cleanup (vcl_worker_t * wrk, vcl_session_t * s,
Florin Corasf9240dc2019-01-15 08:03:17 -08001544 vcl_session_handle_t sh, u8 do_disconnect)
Dave Wallace543852a2017-08-03 02:11:34 -04001545{
Florin Coras134a9962018-08-28 11:32:04 -07001546 int rv = VPPCOM_OK;
Florin Coras47c40e22018-11-26 17:01:36 -08001547
Florin Coras6c3b2182020-10-19 18:36:48 -07001548 VDBG (1, "session %u [0x%llx] closing", s->session_index, s->vpp_handle);
Dave Wallace543852a2017-08-03 02:11:34 -04001549
Florin Coras6c3b2182020-10-19 18:36:48 -07001550 if (s->flags & VCL_SESSION_F_IS_VEP)
Dave Wallace543852a2017-08-03 02:11:34 -04001551 {
Florin Coras6c3b2182020-10-19 18:36:48 -07001552 u32 next_sh = s->vep.next_sh;
Florin Coras134a9962018-08-28 11:32:04 -07001553 while (next_sh != ~0)
Dave Wallace19481612017-09-15 18:47:44 -04001554 {
Florin Corasf9240dc2019-01-15 08:03:17 -08001555 rv = vppcom_epoll_ctl (sh, EPOLL_CTL_DEL, next_sh, 0);
Florin Coras0d427d82018-06-27 03:24:07 -07001556 if (PREDICT_FALSE (rv < 0))
Florin Coras5e062572019-03-14 19:07:51 -07001557 VDBG (0, "vpp handle 0x%llx, sh %u: EPOLL_CTL_DEL vep_idx %u"
Florin Coras6c3b2182020-10-19 18:36:48 -07001558 " failed! rv %d (%s)", s->vpp_handle, next_sh,
1559 s->vep.vep_sh, rv, vppcom_retval_str (rv));
Florin Corasc127d5a2020-10-14 16:35:58 -07001560 next_sh = s->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04001561 }
Florin Corase4306b62020-10-28 12:51:10 -07001562 goto free_session;
Dave Wallacef7f809c2017-10-03 01:48:42 -04001563 }
Florin Coras9ace36d2019-10-28 13:14:17 -07001564
Florin Coras6c3b2182020-10-19 18:36:48 -07001565 if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
Dave Wallacef7f809c2017-10-03 01:48:42 -04001566 {
Florin Coras6c3b2182020-10-19 18:36:48 -07001567 rv = vppcom_epoll_ctl (s->vep.vep_sh, EPOLL_CTL_DEL, sh, 0);
Florin Coras9ace36d2019-10-28 13:14:17 -07001568 if (rv < 0)
1569 VDBG (0, "session %u [0x%llx]: EPOLL_CTL_DEL vep_idx %u "
Florin Coras6c3b2182020-10-19 18:36:48 -07001570 "failed! rv %d (%s)", s->session_index, s->vpp_handle,
1571 s->vep.vep_sh, rv, vppcom_retval_str (rv));
Dave Wallace19481612017-09-15 18:47:44 -04001572 }
Dave Wallace4878cbe2017-11-21 03:45:09 -05001573
Florin Coras9ace36d2019-10-28 13:14:17 -07001574 if (!do_disconnect)
1575 {
1576 VDBG (1, "session %u [0x%llx] disconnect skipped",
Florin Coras6c3b2182020-10-19 18:36:48 -07001577 s->session_index, s->vpp_handle);
Florin Coras9ace36d2019-10-28 13:14:17 -07001578 goto cleanup;
1579 }
1580
Florin Coras6c3b2182020-10-19 18:36:48 -07001581 if (s->session_state == VCL_STATE_LISTEN)
Florin Coras9ace36d2019-10-28 13:14:17 -07001582 {
1583 rv = vppcom_session_unbind (sh);
1584 if (PREDICT_FALSE (rv < 0))
1585 VDBG (0, "session %u [0x%llx]: listener unbind failed! "
Florin Coras6c3b2182020-10-19 18:36:48 -07001586 "rv %d (%s)", s->session_index, s->vpp_handle, rv,
Florin Coras9ace36d2019-10-28 13:14:17 -07001587 vppcom_retval_str (rv));
1588 return rv;
1589 }
Florin Coras1bc919e2020-10-18 20:17:49 -07001590 else if (vcl_session_is_ready (s)
Florin Corasc127d5a2020-10-14 16:35:58 -07001591 || (vcl_session_is_connectable_listener (wrk, s)))
Florin Coras9ace36d2019-10-28 13:14:17 -07001592 {
1593 rv = vppcom_session_disconnect (sh);
1594 if (PREDICT_FALSE (rv < 0))
1595 VDBG (0, "ERROR: session %u [0x%llx]: disconnect failed!"
Florin Coras6c3b2182020-10-19 18:36:48 -07001596 " rv %d (%s)", s->session_index, s->vpp_handle,
Florin Coras9ace36d2019-10-28 13:14:17 -07001597 rv, vppcom_retval_str (rv));
1598 }
Florin Coras6c3b2182020-10-19 18:36:48 -07001599 else if (s->session_state == VCL_STATE_DISCONNECT)
Florin Coras9ace36d2019-10-28 13:14:17 -07001600 {
Florin Coras52dd29f2020-11-18 19:02:17 -08001601 vcl_send_session_reset_reply (wrk, s, 0);
Florin Coras9ace36d2019-10-28 13:14:17 -07001602 }
Florin Coras6c3b2182020-10-19 18:36:48 -07001603 else if (s->session_state == VCL_STATE_DETACHED)
Florin Corasadcfb152020-02-12 08:50:29 +00001604 {
Florin Corasc127d5a2020-10-14 16:35:58 -07001605 VDBG (0, "vpp freed session %d before close", s->session_index);
Florin Corasf0fe1ea2021-12-03 17:03:37 -08001606
1607 if (!(s->flags & VCL_SESSION_F_PENDING_DISCONNECT))
1608 goto free_session;
1609
1610 /* Disconnect/reset messages pending but vpp transport and session
1611 * cleanups already done. Free only after messages drained. */
1612 s->flags |= VCL_SESSION_F_PENDING_FREE;
Florin Corasadcfb152020-02-12 08:50:29 +00001613 }
Florin Coras9ace36d2019-10-28 13:14:17 -07001614
Florin Corasc127d5a2020-10-14 16:35:58 -07001615 s->session_state = VCL_STATE_CLOSED;
Florin Coras54140622020-02-04 19:04:34 +00001616
Florin Coras9ace36d2019-10-28 13:14:17 -07001617 /* Session is removed only after vpp confirms the disconnect */
1618 return rv;
Florin Coras0ef8ef22019-01-18 08:37:13 -08001619
Florin Corasf9240dc2019-01-15 08:03:17 -08001620cleanup:
Florin Coras6c3b2182020-10-19 18:36:48 -07001621 vcl_session_table_del_vpp_handle (wrk, s->vpp_handle);
Florin Corasadcfb152020-02-12 08:50:29 +00001622free_session:
Florin Corasc127d5a2020-10-14 16:35:58 -07001623 vcl_session_free (wrk, s);
1624 vcl_evt (VCL_EVT_CLOSE, s, rv);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08001625
Dave Wallace543852a2017-08-03 02:11:34 -04001626 return rv;
1627}
1628
1629int
Florin Corasf9240dc2019-01-15 08:03:17 -08001630vppcom_session_close (uint32_t session_handle)
1631{
1632 vcl_worker_t *wrk = vcl_worker_get_current ();
1633 vcl_session_t *session;
1634
1635 session = vcl_session_get_w_handle (wrk, session_handle);
1636 if (!session)
1637 return VPPCOM_EBADFD;
1638 return vcl_session_cleanup (wrk, session, session_handle,
1639 1 /* do_disconnect */ );
1640}
1641
1642int
Florin Coras134a9962018-08-28 11:32:04 -07001643vppcom_session_bind (uint32_t session_handle, vppcom_endpt_t * ep)
Dave Wallace543852a2017-08-03 02:11:34 -04001644{
Florin Coras134a9962018-08-28 11:32:04 -07001645 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001646 vcl_session_t *session = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04001647
1648 if (!ep || !ep->ip)
1649 return VPPCOM_EINVAL;
1650
Florin Coras134a9962018-08-28 11:32:04 -07001651 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001652 if (!session)
1653 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001654
Florin Coras6c3b2182020-10-19 18:36:48 -07001655 if (session->flags & VCL_SESSION_F_IS_VEP)
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001656 {
Florin Coras5e062572019-03-14 19:07:51 -07001657 VDBG (0, "ERROR: cannot bind to epoll session %u!",
1658 session->session_index);
Florin Coras070453d2018-08-24 17:04:27 -07001659 return VPPCOM_EBADFD;
Dave Wallace9c4b5b22017-10-18 21:15:48 -04001660 }
1661
Florin Coras7e12d942018-06-27 14:32:43 -07001662 session->transport.is_ip4 = ep->is_ip4;
Florin Coras54693d22018-07-17 10:46:29 -07001663 if (ep->is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05001664 clib_memcpy_fast (&session->transport.lcl_ip.ip4, ep->ip,
1665 sizeof (ip4_address_t));
Florin Coras54693d22018-07-17 10:46:29 -07001666 else
Dave Barach178cf492018-11-13 16:34:13 -05001667 clib_memcpy_fast (&session->transport.lcl_ip.ip6, ep->ip,
1668 sizeof (ip6_address_t));
Florin Coras7e12d942018-06-27 14:32:43 -07001669 session->transport.lcl_port = ep->port;
Stevenac1f96d2017-10-24 16:03:58 -07001670
Florin Coras60687192021-11-29 21:00:47 -08001671 VDBG (0,
1672 "session %u handle %u: binding to local %s address %U port %u, "
1673 "proto %s",
1674 session->session_index, session_handle,
1675 session->transport.is_ip4 ? "IPv4" : "IPv6", vcl_format_ip46_address,
1676 &session->transport.lcl_ip,
Florin Coras7e12d942018-06-27 14:32:43 -07001677 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
1678 clib_net_to_host_u16 (session->transport.lcl_port),
Ping Yu34a3a082018-11-30 19:16:17 -05001679 vppcom_proto_str (session->session_type));
Florin Coras0d427d82018-06-27 03:24:07 -07001680 vcl_evt (VCL_EVT_BIND, session);
Florin Coras460dce62018-07-27 05:45:06 -07001681
1682 if (session->session_type == VPPCOM_PROTO_UDP)
Mohammed Hawarif4e3ee12023-10-30 10:52:19 +01001683 return vppcom_session_listen (session_handle, 10);
Florin Coras460dce62018-07-27 05:45:06 -07001684
Florin Coras070453d2018-08-24 17:04:27 -07001685 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001686}
1687
1688int
Florin Coras134a9962018-08-28 11:32:04 -07001689vppcom_session_listen (uint32_t listen_sh, uint32_t q_len)
Dave Wallace543852a2017-08-03 02:11:34 -04001690{
Florin Coras134a9962018-08-28 11:32:04 -07001691 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001692 vcl_session_t *listen_session = 0;
Dave Wallaceee45d412017-11-24 21:44:06 -05001693 u64 listen_vpp_handle;
Florin Coras070453d2018-08-24 17:04:27 -07001694 int rv;
1695
Florin Coras134a9962018-08-28 11:32:04 -07001696 listen_session = vcl_session_get_w_handle (wrk, listen_sh);
Florin Coras6c3b2182020-10-19 18:36:48 -07001697 if (!listen_session || (listen_session->flags & VCL_SESSION_F_IS_VEP))
Florin Coras070453d2018-08-24 17:04:27 -07001698 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001699
Dave Wallaceee45d412017-11-24 21:44:06 -05001700 listen_vpp_handle = listen_session->vpp_handle;
Florin Corasc127d5a2020-10-14 16:35:58 -07001701 if (listen_session->session_state == VCL_STATE_LISTEN)
Dave Wallacee695cb42017-11-02 22:04:42 -04001702 {
Florin Coras05ecfcc2018-12-12 18:19:39 -08001703 VDBG (0, "session %u [0x%llx]: already in listen state!",
1704 listen_sh, listen_vpp_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001705 return VPPCOM_OK;
Dave Wallacee695cb42017-11-02 22:04:42 -04001706 }
1707
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001708 VDBG (0, "session %u: sending vpp listen request...", listen_sh);
Dave Wallace543852a2017-08-03 02:11:34 -04001709
Florin Coras070453d2018-08-24 17:04:27 -07001710 /*
1711 * Send listen request to vpp and wait for reply
1712 */
Florin Coras458089b2019-08-21 16:20:44 -07001713 vcl_send_session_listen (wrk, listen_session);
Florin Coras134a9962018-08-28 11:32:04 -07001714 rv = vppcom_wait_for_session_state_change (listen_session->session_index,
Florin Corasc127d5a2020-10-14 16:35:58 -07001715 VCL_STATE_LISTEN,
Florin Coras134a9962018-08-28 11:32:04 -07001716 vcm->cfg.session_timeout);
Dave Wallace543852a2017-08-03 02:11:34 -04001717
Florin Coras070453d2018-08-24 17:04:27 -07001718 if (PREDICT_FALSE (rv))
Dave Wallaceee45d412017-11-24 21:44:06 -05001719 {
Florin Coras134a9962018-08-28 11:32:04 -07001720 listen_session = vcl_session_get_w_handle (wrk, listen_sh);
Florin Coras05ecfcc2018-12-12 18:19:39 -08001721 VDBG (0, "session %u [0x%llx]: listen failed! returning %d (%s)",
1722 listen_sh, listen_session->vpp_handle, rv,
1723 vppcom_retval_str (rv));
Florin Coras070453d2018-08-24 17:04:27 -07001724 return rv;
Dave Wallaceee45d412017-11-24 21:44:06 -05001725 }
1726
Florin Coras070453d2018-08-24 17:04:27 -07001727 return VPPCOM_OK;
Keith Burns (alagalah)0d2b0d52018-03-06 15:55:22 -08001728}
1729
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001730int
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001731vppcom_unformat_proto (uint8_t * proto, char *proto_str)
1732{
1733 if (!strcmp (proto_str, "TCP"))
1734 *proto = VPPCOM_PROTO_TCP;
1735 else if (!strcmp (proto_str, "tcp"))
1736 *proto = VPPCOM_PROTO_TCP;
1737 else if (!strcmp (proto_str, "UDP"))
1738 *proto = VPPCOM_PROTO_UDP;
1739 else if (!strcmp (proto_str, "udp"))
1740 *proto = VPPCOM_PROTO_UDP;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001741 else if (!strcmp (proto_str, "TLS"))
1742 *proto = VPPCOM_PROTO_TLS;
1743 else if (!strcmp (proto_str, "tls"))
1744 *proto = VPPCOM_PROTO_TLS;
1745 else if (!strcmp (proto_str, "QUIC"))
1746 *proto = VPPCOM_PROTO_QUIC;
1747 else if (!strcmp (proto_str, "quic"))
1748 *proto = VPPCOM_PROTO_QUIC;
Florin Coras4b47ee22020-11-19 13:38:26 -08001749 else if (!strcmp (proto_str, "DTLS"))
1750 *proto = VPPCOM_PROTO_DTLS;
1751 else if (!strcmp (proto_str, "dtls"))
1752 *proto = VPPCOM_PROTO_DTLS;
Florin Coras6621abf2021-01-06 17:35:17 -08001753 else if (!strcmp (proto_str, "SRTP"))
1754 *proto = VPPCOM_PROTO_SRTP;
1755 else if (!strcmp (proto_str, "srtp"))
1756 *proto = VPPCOM_PROTO_SRTP;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001757 else
1758 return 1;
1759 return 0;
1760}
1761
1762int
Florin Coras32881932023-02-06 13:30:13 -08001763vppcom_session_accept (uint32_t ls_handle, vppcom_endpt_t *ep, uint32_t flags)
Dave Wallace543852a2017-08-03 02:11:34 -04001764{
Florin Coras32881932023-02-06 13:30:13 -08001765 u32 client_session_index = ~0, ls_index, accept_flags = 0;
Florin Coras134a9962018-08-28 11:32:04 -07001766 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras54693d22018-07-17 10:46:29 -07001767 session_accepted_msg_t accepted_msg;
Florin Coras32881932023-02-06 13:30:13 -08001768 vcl_session_t *ls, *client_session = 0;
Florin Coras54693d22018-07-17 10:46:29 -07001769 vcl_session_msg_t *evt;
Florin Coras54693d22018-07-17 10:46:29 -07001770 u8 is_nonblocking;
Dave Wallace543852a2017-08-03 02:11:34 -04001771
Florin Coras5398dfb2021-01-25 20:31:27 -08001772again:
1773
Florin Coras32881932023-02-06 13:30:13 -08001774 ls = vcl_session_get_w_handle (wrk, ls_handle);
1775 if (!ls)
Florin Coras070453d2018-08-24 17:04:27 -07001776 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001777
Florin Coras32881932023-02-06 13:30:13 -08001778 if ((ls->session_state != VCL_STATE_LISTEN) &&
1779 (ls->session_state != VCL_STATE_LISTEN_NO_MQ) &&
1780 (!vcl_session_is_connectable_listener (wrk, ls)))
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001781 {
Florin Coras32881932023-02-06 13:30:13 -08001782 VDBG (0, "ERROR: session [0x%llx]: not in listen state! state (%s)",
1783 ls->vpp_handle, vcl_session_state_str (ls->session_state));
1784 return VPPCOM_EBADFD;
1785 }
1786
1787 ls_index = ls->session_index;
1788
1789 if (clib_fifo_elts (ls->accept_evts_fifo))
1790 {
1791 clib_fifo_sub2 (ls->accept_evts_fifo, evt);
Florin Coras3c7d4f92018-12-14 11:28:43 -08001792 accept_flags = evt->flags;
Florin Coras54693d22018-07-17 10:46:29 -07001793 accepted_msg = evt->accepted_msg;
1794 goto handle;
1795 }
1796
Florin Coras32881932023-02-06 13:30:13 -08001797 is_nonblocking = vcl_session_has_attr (ls, VCL_SESS_ATTR_NONBLOCK);
Florin Coras54693d22018-07-17 10:46:29 -07001798 while (1)
1799 {
Carl Smith592a9092019-11-12 14:57:37 +13001800 if (svm_msg_q_is_empty (wrk->app_event_queue) && is_nonblocking)
1801 return VPPCOM_EAGAIN;
1802
Florin Coras5398dfb2021-01-25 20:31:27 -08001803 svm_msg_q_wait (wrk->app_event_queue, SVM_MQ_WAIT_EMPTY);
1804 vcl_worker_flush_mq_events (wrk);
1805 goto again;
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001806 }
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001807
Florin Coras54693d22018-07-17 10:46:29 -07001808handle:
Keith Burns (alagalah)00f44cc2018-03-07 09:26:38 -08001809
Florin Coras32881932023-02-06 13:30:13 -08001810 client_session_index =
1811 vcl_session_accepted_handler (wrk, &accepted_msg, ls_index);
Florin Coras00cca802019-06-06 09:38:44 -07001812 if (client_session_index == VCL_INVALID_SESSION_INDEX)
1813 return VPPCOM_ECONNABORTED;
1814
Florin Coras32881932023-02-06 13:30:13 -08001815 ls = vcl_session_get (wrk, ls_index);
Florin Coras134a9962018-08-28 11:32:04 -07001816 client_session = vcl_session_get (wrk, client_session_index);
Dave Wallace543852a2017-08-03 02:11:34 -04001817
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001818 if (flags & O_NONBLOCK)
Florin Corasac422d62020-10-19 20:51:36 -07001819 vcl_session_set_attr (client_session, VCL_SESS_ATTR_NONBLOCK);
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08001820
Florin Coras32881932023-02-06 13:30:13 -08001821 VDBG (1,
1822 "listener %u [0x%llx]: Got a connect request! session %u [0x%llx],"
1823 " flags %d, is_nonblocking %u",
1824 ls->session_index, ls->vpp_handle, client_session_index,
Florin Coras5e062572019-03-14 19:07:51 -07001825 client_session->vpp_handle, flags,
Florin Corasac422d62020-10-19 20:51:36 -07001826 vcl_session_has_attr (client_session, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace543852a2017-08-03 02:11:34 -04001827
Dave Wallace048b1d62018-01-03 22:24:41 -05001828 if (ep)
1829 {
Florin Coras7e12d942018-06-27 14:32:43 -07001830 ep->is_ip4 = client_session->transport.is_ip4;
1831 ep->port = client_session->transport.rmt_port;
1832 if (client_session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05001833 clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip4,
1834 sizeof (ip4_address_t));
Dave Wallace048b1d62018-01-03 22:24:41 -05001835 else
Dave Barach178cf492018-11-13 16:34:13 -05001836 clib_memcpy_fast (ep->ip, &client_session->transport.rmt_ip.ip6,
1837 sizeof (ip6_address_t));
Dave Wallace048b1d62018-01-03 22:24:41 -05001838 }
Dave Wallace60caa062017-11-10 17:07:13 -05001839
Florin Coras60687192021-11-29 21:00:47 -08001840 VDBG (0,
1841 "listener %u [0x%llx] accepted %u [0x%llx] peer: %U:%u "
1842 "local: %U:%u",
Florin Coras32881932023-02-06 13:30:13 -08001843 ls_handle, ls->vpp_handle, client_session_index,
1844 client_session->vpp_handle, vcl_format_ip46_address,
1845 &client_session->transport.rmt_ip,
Florin Coras54693d22018-07-17 10:46:29 -07001846 client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001847 clib_net_to_host_u16 (client_session->transport.rmt_port),
Florin Coras60687192021-11-29 21:00:47 -08001848 vcl_format_ip46_address, &client_session->transport.lcl_ip,
Florin Coras54693d22018-07-17 10:46:29 -07001849 client_session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001850 clib_net_to_host_u16 (client_session->transport.lcl_port));
Florin Coras32881932023-02-06 13:30:13 -08001851 vcl_evt (VCL_EVT_ACCEPT, client_session, ls, client_session_index);
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08001852
Florin Coras3c7d4f92018-12-14 11:28:43 -08001853 /*
1854 * Session might have been closed already
1855 */
1856 if (accept_flags)
1857 {
Florin Coras3c7d4f92018-12-14 11:28:43 -08001858 if (accept_flags & VCL_ACCEPTED_F_CLOSED)
Florin Corasc127d5a2020-10-14 16:35:58 -07001859 client_session->session_state = VCL_STATE_VPP_CLOSING;
Florin Coras3c7d4f92018-12-14 11:28:43 -08001860 else if (accept_flags & VCL_ACCEPTED_F_RESET)
Florin Corasc127d5a2020-10-14 16:35:58 -07001861 client_session->session_state = VCL_STATE_DISCONNECT;
Florin Coras3c7d4f92018-12-14 11:28:43 -08001862 }
Florin Corasab2f6db2018-08-31 14:31:41 -07001863 return vcl_session_handle (client_session);
Dave Wallace543852a2017-08-03 02:11:34 -04001864}
1865
1866int
Florin Coras134a9962018-08-28 11:32:04 -07001867vppcom_session_connect (uint32_t session_handle, vppcom_endpt_t * server_ep)
Dave Wallace543852a2017-08-03 02:11:34 -04001868{
Florin Coras134a9962018-08-28 11:32:04 -07001869 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07001870 vcl_session_t *session = 0;
Florin Coras134a9962018-08-28 11:32:04 -07001871 u32 session_index;
Florin Coras070453d2018-08-24 17:04:27 -07001872 int rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001873
Florin Coras134a9962018-08-28 11:32:04 -07001874 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07001875 if (!session)
1876 return VPPCOM_EBADFD;
Florin Coras134a9962018-08-28 11:32:04 -07001877 session_index = session->session_index;
Dave Wallace4878cbe2017-11-21 03:45:09 -05001878
Florin Coras6c3b2182020-10-19 18:36:48 -07001879 if (PREDICT_FALSE (session->flags & VCL_SESSION_F_IS_VEP))
Dave Wallace543852a2017-08-03 02:11:34 -04001880 {
Florin Coras5ffb9642021-12-03 17:24:13 -08001881 VWRN ("cannot connect epoll session %u!", session->session_index);
Florin Coras070453d2018-08-24 17:04:27 -07001882 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04001883 }
1884
Florin Corasc127d5a2020-10-14 16:35:58 -07001885 if (PREDICT_FALSE (vcl_session_is_ready (session)))
Dave Wallace543852a2017-08-03 02:11:34 -04001886 {
Florin Coras60687192021-11-29 21:00:47 -08001887 VDBG (0,
Florin Coras5ffb9642021-12-03 17:24:13 -08001888 "session %u [0x%llx]: already connected to %U:%d proto %s,"
1889 " state (%s)",
1890 session->session_index, session->vpp_handle,
Florin Coras60687192021-11-29 21:00:47 -08001891 vcl_format_ip46_address, &session->transport.rmt_ip,
1892 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001893 clib_net_to_host_u16 (session->transport.rmt_port),
Florin Coras5ffb9642021-12-03 17:24:13 -08001894 vppcom_proto_str (session->session_type),
Florin Coras60687192021-11-29 21:00:47 -08001895 vcl_session_state_str (session->session_state));
Florin Coras070453d2018-08-24 17:04:27 -07001896 return VPPCOM_OK;
Dave Wallace543852a2017-08-03 02:11:34 -04001897 }
1898
Florin Coras0a1e1832020-03-29 18:54:04 +00001899 /* Attempt to connect a connectionless listener */
Florin Corasc127d5a2020-10-14 16:35:58 -07001900 if (PREDICT_FALSE (session->session_state == VCL_STATE_LISTEN))
Florin Coras0a1e1832020-03-29 18:54:04 +00001901 {
1902 if (session->session_type != VPPCOM_PROTO_UDP)
1903 return VPPCOM_EINVAL;
1904 vcl_send_session_unlisten (wrk, session);
Florin Corasc127d5a2020-10-14 16:35:58 -07001905 session->session_state = VCL_STATE_CLOSED;
Florin Coras0a1e1832020-03-29 18:54:04 +00001906 }
1907
Florin Coras7e12d942018-06-27 14:32:43 -07001908 session->transport.is_ip4 = server_ep->is_ip4;
Florin Corasef7cbf62019-10-17 09:56:27 -07001909 vcl_ip_copy_from_ep (&session->transport.rmt_ip, server_ep);
Florin Coras7e12d942018-06-27 14:32:43 -07001910 session->transport.rmt_port = server_ep->port;
Nathan Skrzypczak8ac1d6d2019-07-17 11:02:20 +02001911 session->parent_handle = VCL_INVALID_SESSION_HANDLE;
Florin Coras0a1e1832020-03-29 18:54:04 +00001912 session->flags |= VCL_SESSION_F_CONNECTED;
Dave Wallace543852a2017-08-03 02:11:34 -04001913
Florin Coras5ffb9642021-12-03 17:24:13 -08001914 VDBG (0, "session %u: connecting to peer %U:%d proto %s",
1915 session->session_index, vcl_format_ip46_address,
Florin Coras60687192021-11-29 21:00:47 -08001916 &session->transport.rmt_ip,
1917 session->transport.is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
Florin Coras7e12d942018-06-27 14:32:43 -07001918 clib_net_to_host_u16 (session->transport.rmt_port),
Ping Yu34a3a082018-11-30 19:16:17 -05001919 vppcom_proto_str (session->session_type));
Dave Wallace543852a2017-08-03 02:11:34 -04001920
Florin Coras458089b2019-08-21 16:20:44 -07001921 vcl_send_session_connect (wrk, session);
Florin Coras57c88932019-08-29 12:03:17 -07001922
Florin Corasac422d62020-10-19 20:51:36 -07001923 if (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK))
Florin Corasa5ea8212020-08-24 21:23:51 -07001924 {
fanyfa49d59d2020-10-13 17:07:16 +08001925 /* State set to STATE_UPDATED to ensure the session is not assumed
Florin Corasdfffdd72020-10-15 10:54:47 -07001926 * to be ready and to also allow the app to close it prior to vpp's
fanyfa49d59d2020-10-13 17:07:16 +08001927 * connected reply. */
Florin Corasc127d5a2020-10-14 16:35:58 -07001928 session->session_state = VCL_STATE_UPDATED;
Florin Corasa5ea8212020-08-24 21:23:51 -07001929 return VPPCOM_EINPROGRESS;
1930 }
Florin Coras57c88932019-08-29 12:03:17 -07001931
1932 /*
1933 * Wait for reply from vpp if blocking
1934 */
Florin Corasdfffdd72020-10-15 10:54:47 -07001935 rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
Florin Coras070453d2018-08-24 17:04:27 -07001936 vcm->cfg.session_timeout);
Dave Wallace4878cbe2017-11-21 03:45:09 -05001937
Florin Coras134a9962018-08-28 11:32:04 -07001938 session = vcl_session_get (wrk, session_index);
Florin Coras5e062572019-03-14 19:07:51 -07001939 VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
1940 session->vpp_handle, rv ? "failed" : "succeeded");
Dave Wallaceee45d412017-11-24 21:44:06 -05001941
Dave Wallace4878cbe2017-11-21 03:45:09 -05001942 return rv;
Dave Wallace543852a2017-08-03 02:11:34 -04001943}
1944
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001945int
1946vppcom_session_stream_connect (uint32_t session_handle,
1947 uint32_t parent_session_handle)
1948{
1949 vcl_worker_t *wrk = vcl_worker_get_current ();
1950 vcl_session_t *session, *parent_session;
1951 u32 session_index, parent_session_index;
1952 int rv;
1953
1954 session = vcl_session_get_w_handle (wrk, session_handle);
1955 if (!session)
1956 return VPPCOM_EBADFD;
1957 parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
1958 if (!parent_session)
1959 return VPPCOM_EBADFD;
1960
1961 session_index = session->session_index;
1962 parent_session_index = parent_session->session_index;
Florin Coras6c3b2182020-10-19 18:36:48 -07001963 if (PREDICT_FALSE (session->flags & VCL_SESSION_F_IS_VEP))
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001964 {
1965 VDBG (0, "ERROR: cannot connect epoll session %u!",
1966 session->session_index);
1967 return VPPCOM_EBADFD;
1968 }
1969
Florin Corasc127d5a2020-10-14 16:35:58 -07001970 if (PREDICT_FALSE (vcl_session_is_ready (session)))
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001971 {
Florin Coras60687192021-11-29 21:00:47 -08001972 VDBG (0,
1973 "session handle %u [0x%llx]: session already "
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001974 "connected to session %u [0x%llx] proto %s, state 0x%x (%s)",
Florin Coras60687192021-11-29 21:00:47 -08001975 session_handle, session->vpp_handle, parent_session_handle,
1976 parent_session->vpp_handle,
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001977 vppcom_proto_str (session->session_type), session->session_state,
Florin Coras60687192021-11-29 21:00:47 -08001978 vcl_session_state_str (session->session_state));
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001979 return VPPCOM_OK;
1980 }
1981
1982 /* Connect to quic session specifics */
1983 session->transport.is_ip4 = parent_session->transport.is_ip4;
1984 session->transport.rmt_ip.ip4.as_u32 = (uint32_t) 1;
1985 session->transport.rmt_port = 0;
Nathan Skrzypczak8ac1d6d2019-07-17 11:02:20 +02001986 session->parent_handle = parent_session->vpp_handle;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001987
1988 VDBG (0, "session handle %u: connecting to session %u [0x%llx]",
1989 session_handle, parent_session_handle, parent_session->vpp_handle);
1990
1991 /*
1992 * Send connect request and wait for reply from vpp
1993 */
Florin Coras458089b2019-08-21 16:20:44 -07001994 vcl_send_session_connect (wrk, session);
Florin Corasdfffdd72020-10-15 10:54:47 -07001995 rv = vppcom_wait_for_session_state_change (session_index, VCL_STATE_READY,
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02001996 vcm->cfg.session_timeout);
1997
1998 session->listener_index = parent_session_index;
1999 parent_session = vcl_session_get_w_handle (wrk, parent_session_handle);
Florin Coras028eaf02019-07-19 12:15:52 -07002000 if (parent_session)
2001 parent_session->n_accepted_sessions++;
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02002002
2003 session = vcl_session_get (wrk, session_index);
2004 VDBG (0, "session %u [0x%llx]: connect %s!", session->session_index,
2005 session->vpp_handle, rv ? "failed" : "succeeded");
2006
2007 return rv;
2008}
2009
Steven58f464e2017-10-25 12:33:12 -07002010static inline int
Florin Coras134a9962018-08-28 11:32:04 -07002011vppcom_session_read_internal (uint32_t session_handle, void *buf, int n,
Steven58f464e2017-10-25 12:33:12 -07002012 u8 peek)
Dave Wallace543852a2017-08-03 02:11:34 -04002013{
Florin Coras134a9962018-08-28 11:32:04 -07002014 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras4a2c7942020-09-10 12:27:14 -07002015 int rv, n_read = 0, is_nonblocking;
Florin Coras460dce62018-07-27 05:45:06 -07002016 vcl_session_t *s = 0;
Dave Wallace543852a2017-08-03 02:11:34 -04002017 svm_fifo_t *rx_fifo;
Florin Coras54693d22018-07-17 10:46:29 -07002018 session_event_t *e;
2019 svm_msg_q_t *mq;
Florin Coras41c9e042018-09-11 00:10:41 -07002020 u8 is_ct;
Dave Wallace543852a2017-08-03 02:11:34 -04002021
Florin Coras070453d2018-08-24 17:04:27 -07002022 if (PREDICT_FALSE (!buf))
wanghanlin72228a22021-07-06 17:18:29 +08002023 return VPPCOM_EFAULT;
Dave Wallace543852a2017-08-03 02:11:34 -04002024
Florin Coras134a9962018-08-28 11:32:04 -07002025 s = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras6c3b2182020-10-19 18:36:48 -07002026 if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
Florin Coras070453d2018-08-24 17:04:27 -07002027 return VPPCOM_EBADFD;
Dave Wallace4878cbe2017-11-21 03:45:09 -05002028
Florin Coras6d0106e2019-01-29 20:11:58 -08002029 if (PREDICT_FALSE (!vcl_session_is_open (s)))
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002030 {
Florin Coras5e062572019-03-14 19:07:51 -07002031 VDBG (0, "session %u[0x%llx] is not open! state 0x%x (%s)",
Florin Coras6d0106e2019-01-29 20:11:58 -08002032 s->session_index, s->vpp_handle, s->session_state,
Florin Coras60687192021-11-29 21:00:47 -08002033 vcl_session_state_str (s->session_state));
Florin Coras6d0106e2019-01-29 20:11:58 -08002034 return vcl_session_closed_error (s);
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002035 }
2036
liuyacan55c952e2021-06-13 14:54:55 +08002037 if (PREDICT_FALSE (s->flags & VCL_SESSION_F_RD_SHUTDOWN))
2038 {
2039 /* Vpp would ack the incoming data and enqueue it for reading.
2040 * So even if SHUT_RD is set, we can still read() the data if
2041 * the session is ready.
2042 */
2043 if (!vcl_session_read_ready (s))
2044 {
2045 return 0;
2046 }
2047 }
2048
Florin Corasac422d62020-10-19 20:51:36 -07002049 is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
Florin Coras41c9e042018-09-11 00:10:41 -07002050 is_ct = vcl_session_is_ct (s);
Florin Coras653e43f2019-03-04 10:56:23 -08002051 mq = wrk->app_event_queue;
2052 rx_fifo = is_ct ? s->ct_rx_fifo : s->rx_fifo;
Florin Corasac422d62020-10-19 20:51:36 -07002053 s->flags &= ~VCL_SESSION_F_HAS_RX_EVT;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002054
Sirshak Das28aa5392019-02-05 01:33:33 -06002055 if (svm_fifo_is_empty_cons (rx_fifo))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002056 {
Florin Corasae036d32023-09-18 09:56:21 -07002057 if (is_ct)
2058 svm_fifo_unset_event (s->rx_fifo);
2059 svm_fifo_unset_event (rx_fifo);
Florin Coras54693d22018-07-17 10:46:29 -07002060 if (is_nonblocking)
Florin Corasc0737e92019-03-04 14:19:39 -08002061 {
Yu Pingb2955352019-12-04 06:49:04 +08002062 if (vcl_session_is_closing (s))
2063 return vcl_session_closing_error (s);
Florin Corasc0737e92019-03-04 14:19:39 -08002064 return VPPCOM_EWOULDBLOCK;
2065 }
Sirshak Das28aa5392019-02-05 01:33:33 -06002066 while (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras54693d22018-07-17 10:46:29 -07002067 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002068 if (vcl_session_is_closing (s))
2069 return vcl_session_closing_error (s);
2070
Florin Corasd6894562020-10-28 00:37:15 -07002071 if (is_ct)
2072 svm_fifo_unset_event (s->rx_fifo);
2073 svm_fifo_unset_event (rx_fifo);
Florin Coras99368312018-08-02 10:45:44 -07002074
Florin Coras5398dfb2021-01-25 20:31:27 -08002075 svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
2076 vcl_worker_flush_mq_events (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07002077 }
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002078 }
Florin Coras54693d22018-07-17 10:46:29 -07002079
Florin Coras4a2c7942020-09-10 12:27:14 -07002080read_again:
2081
Florin Coras460dce62018-07-27 05:45:06 -07002082 if (s->is_dgram)
Florin Coras4a2c7942020-09-10 12:27:14 -07002083 rv = app_recv_dgram_raw (rx_fifo, buf, n, &s->transport, 0, peek);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002084 else
Florin Coras4a2c7942020-09-10 12:27:14 -07002085 rv = app_recv_stream_raw (rx_fifo, buf, n, 0, peek);
2086
2087 ASSERT (rv >= 0);
Florin Coras23368312021-06-04 16:28:18 -07002088
2089 if (peek)
2090 return rv;
2091
Florin Coras4a2c7942020-09-10 12:27:14 -07002092 n_read += rv;
Florin Coras54693d22018-07-17 10:46:29 -07002093
Sirshak Das28aa5392019-02-05 01:33:33 -06002094 if (svm_fifo_is_empty_cons (rx_fifo))
Florin Corasc34118b2020-08-12 18:58:25 -07002095 {
Florin Corasd6894562020-10-28 00:37:15 -07002096 if (is_ct)
2097 svm_fifo_unset_event (s->rx_fifo);
2098 svm_fifo_unset_event (rx_fifo);
Florin Corasc34118b2020-08-12 18:58:25 -07002099 if (!svm_fifo_is_empty_cons (rx_fifo)
Florin Corasd6894562020-10-28 00:37:15 -07002100 && svm_fifo_set_event (rx_fifo) && is_nonblocking)
Florin Corasc34118b2020-08-12 18:58:25 -07002101 {
Florin Corasc34118b2020-08-12 18:58:25 -07002102 vec_add2 (wrk->unhandled_evts_vector, e, 1);
2103 e->event_type = SESSION_IO_EVT_RX;
2104 e->session_index = s->session_index;
2105 }
2106 }
Florin Coras54fe32c2020-11-25 20:26:32 -08002107 else if (PREDICT_FALSE (rv < n && !s->is_dgram))
Florin Coras4a2c7942020-09-10 12:27:14 -07002108 {
2109 /* More data enqueued while reading. Try to drain it
Florin Coras54fe32c2020-11-25 20:26:32 -08002110 * or fill the buffer. Avoid doing that for dgrams */
Florin Coras4a2c7942020-09-10 12:27:14 -07002111 buf += rv;
2112 n -= rv;
2113 goto read_again;
2114 }
Florin Coras41c9e042018-09-11 00:10:41 -07002115
Florin Coras17ec5772020-08-12 20:46:29 -07002116 if (PREDICT_FALSE (svm_fifo_needs_deq_ntf (rx_fifo, n_read)))
Florin Coras317b8e02019-04-17 09:57:46 -07002117 {
Florin Coras17ec5772020-08-12 20:46:29 -07002118 svm_fifo_clear_deq_ntf (rx_fifo);
Florin Corasc547e912020-12-08 17:50:45 -08002119 app_send_io_evt_to_vpp (s->vpp_evt_q,
2120 s->rx_fifo->shr->master_session_index,
Florin Coras317b8e02019-04-17 09:57:46 -07002121 SESSION_IO_EVT_RX, SVM_Q_WAIT);
Florin Coras317b8e02019-04-17 09:57:46 -07002122 }
2123
Florin Coras5e062572019-03-14 19:07:51 -07002124 VDBG (2, "session %u[0x%llx]: read %d bytes from (%p)", s->session_index,
2125 s->vpp_handle, n_read, rx_fifo);
Florin Coras2cba8532018-09-11 16:33:36 -07002126
Florin Coras54693d22018-07-17 10:46:29 -07002127 return n_read;
Dave Wallace543852a2017-08-03 02:11:34 -04002128}
2129
Steven58f464e2017-10-25 12:33:12 -07002130int
Florin Coras134a9962018-08-28 11:32:04 -07002131vppcom_session_read (uint32_t session_handle, void *buf, size_t n)
Steven58f464e2017-10-25 12:33:12 -07002132{
Florin Coras134a9962018-08-28 11:32:04 -07002133 return (vppcom_session_read_internal (session_handle, buf, n, 0));
Steven58f464e2017-10-25 12:33:12 -07002134}
2135
2136static int
Florin Coras134a9962018-08-28 11:32:04 -07002137vppcom_session_peek (uint32_t session_handle, void *buf, int n)
Steven58f464e2017-10-25 12:33:12 -07002138{
Florin Coras134a9962018-08-28 11:32:04 -07002139 return (vppcom_session_read_internal (session_handle, buf, n, 1));
Steven58f464e2017-10-25 12:33:12 -07002140}
2141
Florin Coras2cba8532018-09-11 16:33:36 -07002142int
2143vppcom_session_read_segments (uint32_t session_handle,
Florin Corasd68faf82020-09-25 15:18:13 -07002144 vppcom_data_segment_t * ds, uint32_t n_segments,
2145 uint32_t max_bytes)
Florin Coras2cba8532018-09-11 16:33:36 -07002146{
2147 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras6d0106e2019-01-29 20:11:58 -08002148 int n_read = 0, is_nonblocking;
Florin Coras2cba8532018-09-11 16:33:36 -07002149 vcl_session_t *s = 0;
2150 svm_fifo_t *rx_fifo;
Florin Coras2cba8532018-09-11 16:33:36 -07002151 svm_msg_q_t *mq;
2152 u8 is_ct;
2153
2154 s = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras6c3b2182020-10-19 18:36:48 -07002155 if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
Florin Coras2cba8532018-09-11 16:33:36 -07002156 return VPPCOM_EBADFD;
2157
Florin Coras6d0106e2019-01-29 20:11:58 -08002158 if (PREDICT_FALSE (!vcl_session_is_open (s)))
2159 return vcl_session_closed_error (s);
Florin Coras2cba8532018-09-11 16:33:36 -07002160
Florin Corasac422d62020-10-19 20:51:36 -07002161 is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
Florin Coras2cba8532018-09-11 16:33:36 -07002162 is_ct = vcl_session_is_ct (s);
Florin Corasac422d62020-10-19 20:51:36 -07002163 mq = wrk->app_event_queue;
Florin Coras62877022020-10-28 21:22:04 -07002164 rx_fifo = is_ct ? s->ct_rx_fifo : s->rx_fifo;
Florin Corasac422d62020-10-19 20:51:36 -07002165 s->flags &= ~VCL_SESSION_F_HAS_RX_EVT;
Florin Coras2cba8532018-09-11 16:33:36 -07002166
Sirshak Das28aa5392019-02-05 01:33:33 -06002167 if (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras2cba8532018-09-11 16:33:36 -07002168 {
Florin Corasae036d32023-09-18 09:56:21 -07002169 if (is_ct)
2170 svm_fifo_unset_event (s->rx_fifo);
2171 svm_fifo_unset_event (rx_fifo);
Florin Coras2cba8532018-09-11 16:33:36 -07002172 if (is_nonblocking)
2173 {
Florin Corasae036d32023-09-18 09:56:21 -07002174 if (vcl_session_is_closing (s))
2175 return vcl_session_closing_error (s);
Florin Corasaa27eb92018-10-13 12:20:01 -07002176 return VPPCOM_EWOULDBLOCK;
Florin Coras2cba8532018-09-11 16:33:36 -07002177 }
Sirshak Das28aa5392019-02-05 01:33:33 -06002178 while (svm_fifo_is_empty_cons (rx_fifo))
Florin Coras2cba8532018-09-11 16:33:36 -07002179 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002180 if (vcl_session_is_closing (s))
2181 return vcl_session_closing_error (s);
2182
Florin Coras62877022020-10-28 21:22:04 -07002183 if (is_ct)
2184 svm_fifo_unset_event (s->rx_fifo);
Florin Coras2cba8532018-09-11 16:33:36 -07002185 svm_fifo_unset_event (rx_fifo);
Florin Coras2cba8532018-09-11 16:33:36 -07002186
Florin Coras5398dfb2021-01-25 20:31:27 -08002187 svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
2188 vcl_worker_flush_mq_events (wrk);
Florin Coras2cba8532018-09-11 16:33:36 -07002189 }
2190 }
2191
Florin Corasd1cc38d2020-10-11 11:05:04 -07002192 n_read = svm_fifo_segments (rx_fifo, s->rx_bytes_pending,
Florin Corasb85de192022-01-18 20:51:08 -08002193 (svm_fifo_seg_t *) ds, &n_segments, max_bytes);
Florin Corasd1cc38d2020-10-11 11:05:04 -07002194 if (n_read < 0)
2195 return VPPCOM_EAGAIN;
2196
2197 if (svm_fifo_max_dequeue_cons (rx_fifo) == n_read)
2198 {
Florin Coras62877022020-10-28 21:22:04 -07002199 if (is_ct)
2200 svm_fifo_unset_event (s->rx_fifo);
2201 svm_fifo_unset_event (rx_fifo);
Florin Corasd1cc38d2020-10-11 11:05:04 -07002202 if (svm_fifo_max_dequeue_cons (rx_fifo) != n_read
Florin Coras62877022020-10-28 21:22:04 -07002203 && svm_fifo_set_event (rx_fifo)
Florin Corasac422d62020-10-19 20:51:36 -07002204 && vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK))
Florin Corasd1cc38d2020-10-11 11:05:04 -07002205 {
2206 session_event_t *e;
2207 vec_add2 (wrk->unhandled_evts_vector, e, 1);
2208 e->event_type = SESSION_IO_EVT_RX;
2209 e->session_index = s->session_index;
2210 }
2211 }
2212
2213 s->rx_bytes_pending += n_read;
Florin Coras2cba8532018-09-11 16:33:36 -07002214 return n_read;
2215}
2216
2217void
Florin Corasd68faf82020-09-25 15:18:13 -07002218vppcom_session_free_segments (uint32_t session_handle, uint32_t n_bytes)
Florin Coras2cba8532018-09-11 16:33:36 -07002219{
2220 vcl_worker_t *wrk = vcl_worker_get_current ();
2221 vcl_session_t *s;
Florin Coras62877022020-10-28 21:22:04 -07002222 u8 is_ct;
Florin Coras2cba8532018-09-11 16:33:36 -07002223
2224 s = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras6c3b2182020-10-19 18:36:48 -07002225 if (PREDICT_FALSE (!s || (s->flags & VCL_SESSION_F_IS_VEP)))
Florin Coras2cba8532018-09-11 16:33:36 -07002226 return;
2227
Florin Coras62877022020-10-28 21:22:04 -07002228 is_ct = vcl_session_is_ct (s);
2229 svm_fifo_dequeue_drop (is_ct ? s->ct_rx_fifo : s->rx_fifo, n_bytes);
Florin Corasd1cc38d2020-10-11 11:05:04 -07002230
Florin Coras8cc93212021-09-10 11:37:12 -07002231 ASSERT (s->rx_bytes_pending >= n_bytes);
Florin Corasd1cc38d2020-10-11 11:05:04 -07002232 s->rx_bytes_pending -= n_bytes;
Florin Coras2cba8532018-09-11 16:33:36 -07002233}
2234
Florin Coras7a2abce2020-04-05 19:25:44 +00002235always_inline u8
2236vcl_fifo_is_writeable (svm_fifo_t * f, u32 len, u8 is_dgram)
Dave Wallace543852a2017-08-03 02:11:34 -04002237{
Florin Coras7a2abce2020-04-05 19:25:44 +00002238 u32 max_enq = svm_fifo_max_enqueue_prod (f);
2239 if (is_dgram)
2240 return max_enq >= (sizeof (session_dgram_hdr_t) + len);
2241 else
2242 return max_enq > 0;
Florin Coras7a2abce2020-04-05 19:25:44 +00002243}
2244
2245always_inline int
Dou Chao243a0432022-11-29 19:41:34 +08002246vppcom_session_write_inline (vcl_worker_t *wrk, vcl_session_t *s, void *buf,
Florin Coraseff5f7a2023-02-07 17:36:17 -08002247 size_t n, u8 is_flush, u8 is_dgram)
Florin Coras7a2abce2020-04-05 19:25:44 +00002248{
Florin Coras6d0106e2019-01-29 20:11:58 -08002249 int n_write, is_nonblocking;
Florin Coras460dce62018-07-27 05:45:06 -07002250 session_evt_type_t et;
Florin Coras14ed6df2019-03-06 21:13:42 -08002251 svm_fifo_t *tx_fifo;
Florin Coras3c2fed52018-07-04 04:15:05 -07002252 svm_msg_q_t *mq;
Florin Coras0e88e852018-09-17 22:09:02 -07002253 u8 is_ct;
Dave Wallace543852a2017-08-03 02:11:34 -04002254
Florin Coras0b0d28e2021-06-04 17:31:53 -07002255 /* Accept zero length writes but just return */
2256 if (PREDICT_FALSE (!n))
2257 return VPPCOM_OK;
2258
2259 if (PREDICT_FALSE (!buf))
2260 return VPPCOM_EFAULT;
Dave Wallace543852a2017-08-03 02:11:34 -04002261
Florin Coras6c3b2182020-10-19 18:36:48 -07002262 if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
Dave Wallace543852a2017-08-03 02:11:34 -04002263 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002264 VDBG (0, "ERROR: session %u [0x%llx]: cannot write to an epoll"
2265 " session!", s->session_index, s->vpp_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002266 return VPPCOM_EBADFD;
Dave Wallace543852a2017-08-03 02:11:34 -04002267 }
2268
liuyacan55c952e2021-06-13 14:54:55 +08002269 if (PREDICT_FALSE (!vcl_session_is_open (s)))
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002270 {
Florin Coras6d0106e2019-01-29 20:11:58 -08002271 VDBG (1, "session %u [0x%llx]: is not open! state 0x%x (%s)",
2272 s->session_index, s->vpp_handle, s->session_state,
Florin Coras60687192021-11-29 21:00:47 -08002273 vcl_session_state_str (s->session_state));
Florin Coras6d0106e2019-01-29 20:11:58 -08002274 return vcl_session_closed_error (s);;
Dave Wallace9c4b5b22017-10-18 21:15:48 -04002275 }
2276
liuyacan55c952e2021-06-13 14:54:55 +08002277 if (PREDICT_FALSE (s->flags & VCL_SESSION_F_WR_SHUTDOWN))
2278 {
2279 VDBG (1, "session %u [0x%llx]: is shutdown! state 0x%x (%s)",
2280 s->session_index, s->vpp_handle, s->session_state,
Florin Coras60687192021-11-29 21:00:47 -08002281 vcl_session_state_str (s->session_state));
liuyacan55c952e2021-06-13 14:54:55 +08002282 return VPPCOM_EPIPE;
2283 }
2284
Florin Coras0e88e852018-09-17 22:09:02 -07002285 is_ct = vcl_session_is_ct (s);
Florin Coras653e43f2019-03-04 10:56:23 -08002286 tx_fifo = is_ct ? s->ct_tx_fifo : s->tx_fifo;
Florin Corasac422d62020-10-19 20:51:36 -07002287 is_nonblocking = vcl_session_has_attr (s, VCL_SESS_ATTR_NONBLOCK);
Sirshak Das28aa5392019-02-05 01:33:33 -06002288
Florin Coras653e43f2019-03-04 10:56:23 -08002289 mq = wrk->app_event_queue;
Florin Coras7a2abce2020-04-05 19:25:44 +00002290 if (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
Dave Wallace543852a2017-08-03 02:11:34 -04002291 {
Florin Coras54693d22018-07-17 10:46:29 -07002292 if (is_nonblocking)
2293 {
Florin Coras070453d2018-08-24 17:04:27 -07002294 return VPPCOM_EWOULDBLOCK;
Florin Coras54693d22018-07-17 10:46:29 -07002295 }
Florin Coras7a2abce2020-04-05 19:25:44 +00002296 while (!vcl_fifo_is_writeable (tx_fifo, n, is_dgram))
Florin Coras54693d22018-07-17 10:46:29 -07002297 {
Florin Coras2d379d82019-06-28 12:45:12 -07002298 svm_fifo_add_want_deq_ntf (tx_fifo, SVM_FIFO_WANT_DEQ_NOTIF);
Florin Coras6d0106e2019-01-29 20:11:58 -08002299 if (vcl_session_is_closing (s))
2300 return vcl_session_closing_error (s);
Florin Coras0e88e852018-09-17 22:09:02 -07002301
Florin Coras5398dfb2021-01-25 20:31:27 -08002302 svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
2303 vcl_worker_flush_mq_events (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07002304 }
Dave Wallace543852a2017-08-03 02:11:34 -04002305 }
Dave Wallace543852a2017-08-03 02:11:34 -04002306
Florin Coras653e43f2019-03-04 10:56:23 -08002307 et = SESSION_IO_EVT_TX;
2308 if (is_flush && !is_ct)
Florin Coras42ceddb2018-12-12 10:56:01 -08002309 et = SESSION_IO_EVT_TX_FLUSH;
2310
Florin Coras7a2abce2020-04-05 19:25:44 +00002311 if (is_dgram)
Florin Corasaeb7c1c2023-03-10 10:22:21 -08002312 {
2313 et = vcl_session_dgram_tx_evt (s, et);
2314 n_write =
2315 app_send_dgram_raw_gso (tx_fifo, &s->transport, s->vpp_evt_q, buf, n,
2316 s->gso_size, et, 0 /* do_evt */, SVM_Q_WAIT);
2317 }
Florin Coras460dce62018-07-27 05:45:06 -07002318 else
Florin Corasaeb7c1c2023-03-10 10:22:21 -08002319 {
2320 n_write = app_send_stream_raw (tx_fifo, s->vpp_evt_q, buf, n, et,
2321 0 /* do_evt */, SVM_Q_WAIT);
2322 }
Florin Coras653e43f2019-03-04 10:56:23 -08002323
Florin Coras14ed6df2019-03-06 21:13:42 -08002324 if (svm_fifo_set_event (s->tx_fifo))
Florin Corasc547e912020-12-08 17:50:45 -08002325 app_send_io_evt_to_vpp (
2326 s->vpp_evt_q, s->tx_fifo->shr->master_session_index, et, SVM_Q_WAIT);
Keith Burns (alagalah)410bcca2018-03-23 13:42:49 -07002327
Florin Coras56230092020-09-02 20:52:58 -07002328 /* The underlying fifo segment can run out of memory */
2329 if (PREDICT_FALSE (n_write < 0))
2330 return VPPCOM_EAGAIN;
Dave Wallace543852a2017-08-03 02:11:34 -04002331
Florin Coras6d0106e2019-01-29 20:11:58 -08002332 VDBG (2, "session %u [0x%llx]: wrote %d bytes", s->session_index,
2333 s->vpp_handle, n_write);
Florin Coras0e88e852018-09-17 22:09:02 -07002334
Florin Coras54693d22018-07-17 10:46:29 -07002335 return n_write;
Dave Wallace543852a2017-08-03 02:11:34 -04002336}
2337
Florin Coras42ceddb2018-12-12 10:56:01 -08002338int
2339vppcom_session_write (uint32_t session_handle, void *buf, size_t n)
2340{
Florin Coras7a2abce2020-04-05 19:25:44 +00002341 vcl_worker_t *wrk = vcl_worker_get_current ();
2342 vcl_session_t *s;
2343
2344 s = vcl_session_get_w_handle (wrk, session_handle);
2345 if (PREDICT_FALSE (!s))
2346 return VPPCOM_EBADFD;
2347
Florin Coraseff5f7a2023-02-07 17:36:17 -08002348 return vppcom_session_write_inline (wrk, s, buf, n, 0 /* is_flush */,
Dou Chao243a0432022-11-29 19:41:34 +08002349 s->is_dgram ? 1 : 0);
Florin Coras42ceddb2018-12-12 10:56:01 -08002350}
2351
Florin Corasb0f662f2018-12-27 14:51:46 -08002352int
2353vppcom_session_write_msg (uint32_t session_handle, void *buf, size_t n)
2354{
Florin Coras7a2abce2020-04-05 19:25:44 +00002355 vcl_worker_t *wrk = vcl_worker_get_current ();
2356 vcl_session_t *s;
2357
2358 s = vcl_session_get_w_handle (wrk, session_handle);
2359 if (PREDICT_FALSE (!s))
2360 return VPPCOM_EBADFD;
2361
Florin Coraseff5f7a2023-02-07 17:36:17 -08002362 return vppcom_session_write_inline (wrk, s, buf, n, 1 /* is_flush */,
Dou Chao243a0432022-11-29 19:41:34 +08002363 s->is_dgram ? 1 : 0);
Florin Corasb0f662f2018-12-27 14:51:46 -08002364}
2365
Florin Corasc0737e92019-03-04 14:19:39 -08002366#define vcl_fifo_rx_evt_valid_or_break(_s) \
Florin Corasbd52e462019-10-28 13:22:37 -07002367if (PREDICT_FALSE (!_s->rx_fifo)) \
2368 break; \
Florin Corasc0737e92019-03-04 14:19:39 -08002369if (PREDICT_FALSE (svm_fifo_is_empty (_s->rx_fifo))) \
2370 { \
2371 if (!vcl_session_is_ct (_s)) \
2372 { \
2373 svm_fifo_unset_event (_s->rx_fifo); \
2374 if (svm_fifo_is_empty (_s->rx_fifo)) \
2375 break; \
2376 } \
2377 else if (svm_fifo_is_empty (_s->ct_rx_fifo)) \
2378 { \
Florin Coras2f630182020-08-13 00:17:51 -07002379 svm_fifo_unset_event (_s->rx_fifo); /* rx evts on actual fifo*/ \
Florin Corasc0737e92019-03-04 14:19:39 -08002380 if (svm_fifo_is_empty (_s->ct_rx_fifo)) \
2381 break; \
2382 } \
2383 } \
Florin Coras6d4bb422018-09-04 22:07:27 -07002384
Florin Coras86f04502018-09-12 16:08:01 -07002385static void
2386vcl_select_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
2387 unsigned long n_bits, unsigned long *read_map,
2388 unsigned long *write_map,
2389 unsigned long *except_map, u32 * bits_set)
Florin Coras54693d22018-07-17 10:46:29 -07002390{
2391 session_disconnected_msg_t *disconnected_msg;
Florin Coras99368312018-08-02 10:45:44 -07002392 session_connected_msg_t *connected_msg;
Florin Corasb242d312020-10-26 15:35:40 -07002393 vcl_session_t *s;
Florin Coras86f04502018-09-12 16:08:01 -07002394 u32 sid;
2395
2396 switch (e->event_type)
2397 {
Florin Corasc0737e92019-03-04 14:19:39 -08002398 case SESSION_IO_EVT_RX:
2399 sid = e->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07002400 s = vcl_session_get (wrk, sid);
2401 if (!s || !vcl_session_is_open (s))
Florin Coras86f04502018-09-12 16:08:01 -07002402 break;
Florin Corasb242d312020-10-26 15:35:40 -07002403 vcl_fifo_rx_evt_valid_or_break (s);
Florin Coras86f04502018-09-12 16:08:01 -07002404 if (sid < n_bits && read_map)
2405 {
David Johnsond9818dd2018-12-14 14:53:41 -05002406 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002407 *bits_set += 1;
2408 }
2409 break;
Florin Corasfe97da32019-03-06 10:09:04 -08002410 case SESSION_IO_EVT_TX:
Florin Corasc0737e92019-03-04 14:19:39 -08002411 sid = e->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07002412 s = vcl_session_get (wrk, sid);
2413 if (!s || !vcl_session_is_open (s))
Florin Coras86f04502018-09-12 16:08:01 -07002414 break;
2415 if (sid < n_bits && write_map)
2416 {
David Johnsond9818dd2018-12-14 14:53:41 -05002417 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002418 *bits_set += 1;
2419 }
2420 break;
Florin Coras86f04502018-09-12 16:08:01 -07002421 case SESSION_CTRL_EVT_ACCEPTED:
Florin Corasb242d312020-10-26 15:35:40 -07002422 if (!e->postponed)
2423 s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
2424 else
2425 s = vcl_session_get (wrk, e->session_index);
2426 if (!s)
Florin Coras3c7d4f92018-12-14 11:28:43 -08002427 break;
Florin Corasb242d312020-10-26 15:35:40 -07002428 sid = s->session_index;
Florin Coras86f04502018-09-12 16:08:01 -07002429 if (sid < n_bits && read_map)
2430 {
David Johnsond9818dd2018-12-14 14:53:41 -05002431 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002432 *bits_set += 1;
2433 }
2434 break;
2435 case SESSION_CTRL_EVT_CONNECTED:
Florin Corasb242d312020-10-26 15:35:40 -07002436 if (!e->postponed)
2437 {
2438 connected_msg = (session_connected_msg_t *) e->data;
2439 sid = vcl_session_connected_handler (wrk, connected_msg);
2440 }
2441 else
2442 sid = e->session_index;
Florin Coras57c88932019-08-29 12:03:17 -07002443 if (sid == VCL_INVALID_SESSION_INDEX)
2444 break;
Florin Coras66c675e2023-03-09 16:43:02 -08002445 if (!(sid < n_bits && write_map))
2446 break;
2447 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2448 *bits_set += 1;
2449 s = vcl_session_get (wrk, sid);
Florin Coras66c675e2023-03-09 16:43:02 -08002450 /* We didn't have a fifo when the event was added */
Florin Coras607eb202023-05-29 16:19:38 -07002451 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Coras86f04502018-09-12 16:08:01 -07002452 break;
2453 case SESSION_CTRL_EVT_DISCONNECTED:
Florin Coras5e9ed0d2023-08-15 11:16:34 -07002454 if (!e->postponed)
2455 {
2456 disconnected_msg = (session_disconnected_msg_t *) e->data;
2457 s = vcl_session_disconnected_handler (wrk, disconnected_msg);
2458 if (!s)
2459 break;
2460 }
2461 else
2462 {
2463 s = vcl_session_get (wrk, e->session_index);
2464 s->flags &= ~VCL_SESSION_F_PENDING_DISCONNECT;
2465 }
2466 if (vcl_session_is_closed (s))
2467 {
2468 if (s && (s->flags & VCL_SESSION_F_PENDING_FREE))
2469 vcl_session_free (wrk, s);
2470 break;
2471 }
Florin Corasb242d312020-10-26 15:35:40 -07002472 sid = s->session_index;
Florin Coras86f04502018-09-12 16:08:01 -07002473 if (sid < n_bits && except_map)
2474 {
David Johnsond9818dd2018-12-14 14:53:41 -05002475 clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002476 *bits_set += 1;
2477 }
2478 break;
2479 case SESSION_CTRL_EVT_RESET:
Florin Coras5e9ed0d2023-08-15 11:16:34 -07002480 if (!e->postponed)
2481 {
2482 sid =
2483 vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
2484 s = vcl_session_get (wrk, sid);
2485 }
2486 else
2487 {
2488 sid = e->session_index;
2489 s = vcl_session_get (wrk, sid);
2490 s->flags &= ~VCL_SESSION_F_PENDING_DISCONNECT;
2491 }
2492 if (vcl_session_is_closed (s))
2493 {
2494 if (s && (s->flags & VCL_SESSION_F_PENDING_FREE))
2495 vcl_session_free (wrk, s);
2496 break;
2497 }
Florin Coras86f04502018-09-12 16:08:01 -07002498 if (sid < n_bits && except_map)
2499 {
David Johnsond9818dd2018-12-14 14:53:41 -05002500 clib_bitmap_set_no_check ((uword *) except_map, sid, 1);
Florin Coras86f04502018-09-12 16:08:01 -07002501 *bits_set += 1;
2502 }
2503 break;
Florin Corasdfae9f92019-02-20 19:48:31 -08002504 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
2505 vcl_session_unlisten_reply_handler (wrk, e->data);
2506 break;
Florin Coras68b7e582020-01-21 18:33:23 -08002507 case SESSION_CTRL_EVT_MIGRATED:
2508 vcl_session_migrated_handler (wrk, e->data);
2509 break;
Florin Coras9ace36d2019-10-28 13:14:17 -07002510 case SESSION_CTRL_EVT_CLEANUP:
2511 vcl_session_cleanup_handler (wrk, e->data);
2512 break;
Florin Coras30e79c22019-01-02 19:31:22 -08002513 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
2514 vcl_session_worker_update_reply_handler (wrk, e->data);
2515 break;
2516 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
2517 vcl_session_req_worker_update_handler (wrk, e->data);
2518 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -07002519 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
2520 vcl_session_app_add_segment_handler (wrk, e->data);
2521 break;
2522 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
2523 vcl_session_app_del_segment_handler (wrk, e->data);
2524 break;
hanlina3a48962020-07-13 11:09:15 +08002525 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -07002526 vcl_worker_rpc_handler (wrk, e->data);
2527 break;
Florin Coras86f04502018-09-12 16:08:01 -07002528 default:
2529 clib_warning ("unhandled: %u", e->event_type);
2530 break;
2531 }
2532}
2533
2534static int
2535vcl_select_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
2536 unsigned long n_bits, unsigned long *read_map,
2537 unsigned long *write_map, unsigned long *except_map,
2538 double time_to_wait, u32 * bits_set)
2539{
Florin Coras99368312018-08-02 10:45:44 -07002540 svm_msg_q_msg_t *msg;
Florin Coras54693d22018-07-17 10:46:29 -07002541 session_event_t *e;
Florin Coras86f04502018-09-12 16:08:01 -07002542 u32 i;
Florin Coras54693d22018-07-17 10:46:29 -07002543
Florin Coras54693d22018-07-17 10:46:29 -07002544 if (svm_msg_q_is_empty (mq))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002545 {
Florin Coras54693d22018-07-17 10:46:29 -07002546 if (*bits_set)
Florin Coras5398dfb2021-01-25 20:31:27 -08002547 return 0;
Dave Wallace4878cbe2017-11-21 03:45:09 -05002548
Florin Coras54693d22018-07-17 10:46:29 -07002549 if (!time_to_wait)
Florin Coras5398dfb2021-01-25 20:31:27 -08002550 return 0;
Florin Coras54693d22018-07-17 10:46:29 -07002551 else if (time_to_wait < 0)
Florin Coras5398dfb2021-01-25 20:31:27 -08002552 svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
Florin Coras54693d22018-07-17 10:46:29 -07002553 else
2554 {
2555 if (svm_msg_q_timedwait (mq, time_to_wait))
Florin Coras5398dfb2021-01-25 20:31:27 -08002556 return 0;
Dave Wallace4878cbe2017-11-21 03:45:09 -05002557 }
2558 }
Florin Corase003a1b2019-06-05 10:47:16 -07002559 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras54693d22018-07-17 10:46:29 -07002560
Florin Coras134a9962018-08-28 11:32:04 -07002561 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
Florin Coras54693d22018-07-17 10:46:29 -07002562 {
Florin Coras134a9962018-08-28 11:32:04 -07002563 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
Florin Coras99368312018-08-02 10:45:44 -07002564 e = svm_msg_q_msg_data (mq, msg);
Florin Coras86f04502018-09-12 16:08:01 -07002565 vcl_select_handle_mq_event (wrk, e, n_bits, read_map, write_map,
2566 except_map, bits_set);
Florin Coras99368312018-08-02 10:45:44 -07002567 svm_msg_q_free_msg (mq, msg);
Florin Coras54693d22018-07-17 10:46:29 -07002568 }
Florin Coras134a9962018-08-28 11:32:04 -07002569 vec_reset_length (wrk->mq_msg_vector);
Florin Coras30e79c22019-01-02 19:31:22 -08002570 vcl_handle_pending_wrk_updates (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07002571 return *bits_set;
Dave Wallace543852a2017-08-03 02:11:34 -04002572}
2573
Florin Coras99368312018-08-02 10:45:44 -07002574static int
Florin Coras294afe22019-01-07 17:49:17 -08002575vppcom_select_condvar (vcl_worker_t * wrk, int n_bits,
2576 vcl_si_set * read_map, vcl_si_set * write_map,
2577 vcl_si_set * except_map, double time_to_wait,
Florin Coras134a9962018-08-28 11:32:04 -07002578 u32 * bits_set)
Florin Coras99368312018-08-02 10:45:44 -07002579{
Florin Coras14ed6df2019-03-06 21:13:42 -08002580 double wait = 0, start = 0;
2581
2582 if (!*bits_set)
2583 {
2584 wait = time_to_wait;
2585 start = clib_time_now (&wrk->clib_time);
2586 }
2587
2588 do
2589 {
2590 vcl_select_handle_mq (wrk, wrk->app_event_queue, n_bits, read_map,
2591 write_map, except_map, wait, bits_set);
2592 if (*bits_set)
2593 return *bits_set;
2594 if (wait == -1)
2595 continue;
2596
2597 wait = wait - (clib_time_now (&wrk->clib_time) - start);
2598 }
2599 while (wait > 0);
2600
2601 return 0;
Florin Coras99368312018-08-02 10:45:44 -07002602}
2603
2604static int
Florin Coras294afe22019-01-07 17:49:17 -08002605vppcom_select_eventfd (vcl_worker_t * wrk, int n_bits,
2606 vcl_si_set * read_map, vcl_si_set * write_map,
2607 vcl_si_set * except_map, double time_to_wait,
Florin Coras134a9962018-08-28 11:32:04 -07002608 u32 * bits_set)
Florin Coras99368312018-08-02 10:45:44 -07002609{
2610 vcl_mq_evt_conn_t *mqc;
2611 int __clib_unused n_read;
2612 int n_mq_evts, i;
2613 u64 buf;
2614
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00002615 if (PREDICT_FALSE (wrk->api_client_handle == ~0))
2616 {
2617 vcl_api_retry_attach (wrk);
2618 return 0;
2619 }
2620
Florin Coras134a9962018-08-28 11:32:04 -07002621 vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
2622 n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
2623 vec_len (wrk->mq_events), time_to_wait);
Florin Coras99368312018-08-02 10:45:44 -07002624 for (i = 0; i < n_mq_evts; i++)
2625 {
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00002626 if (PREDICT_FALSE (wrk->mq_events[i].data.u32 == ~0))
2627 {
2628 vcl_api_handle_disconnect (wrk);
2629 continue;
2630 }
2631
Florin Coras134a9962018-08-28 11:32:04 -07002632 mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
Florin Coras99368312018-08-02 10:45:44 -07002633 n_read = read (mqc->mq_fd, &buf, sizeof (buf));
Florin Coras134a9962018-08-28 11:32:04 -07002634 vcl_select_handle_mq (wrk, mqc->mq, n_bits, read_map, write_map,
Florin Coras99368312018-08-02 10:45:44 -07002635 except_map, 0, bits_set);
2636 }
2637
2638 return (n_mq_evts > 0 ? (int) *bits_set : 0);
2639}
2640
Dave Wallace543852a2017-08-03 02:11:34 -04002641int
Florin Coras294afe22019-01-07 17:49:17 -08002642vppcom_select (int n_bits, vcl_si_set * read_map, vcl_si_set * write_map,
2643 vcl_si_set * except_map, double time_to_wait)
Dave Wallace543852a2017-08-03 02:11:34 -04002644{
Florin Coras54693d22018-07-17 10:46:29 -07002645 u32 sid, minbits = clib_max (n_bits, BITS (uword)), bits_set = 0;
Florin Coras134a9962018-08-28 11:32:04 -07002646 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras096a1d52021-01-27 15:33:51 -08002647 vcl_session_t *s = 0;
Florin Corasf49cf472020-04-19 23:12:08 +00002648 int i;
Dave Wallace543852a2017-08-03 02:11:34 -04002649
Dave Wallace7876d392017-10-19 03:53:57 -04002650 if (n_bits && read_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002651 {
Florin Coras134a9962018-08-28 11:32:04 -07002652 clib_bitmap_validate (wrk->rd_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002653 clib_memcpy_fast (wrk->rd_bitmap, read_map,
Florin Coras294afe22019-01-07 17:49:17 -08002654 vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
2655 memset (read_map, 0, vec_len (wrk->rd_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002656 }
Dave Wallace7876d392017-10-19 03:53:57 -04002657 if (n_bits && write_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002658 {
Florin Coras134a9962018-08-28 11:32:04 -07002659 clib_bitmap_validate (wrk->wr_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002660 clib_memcpy_fast (wrk->wr_bitmap, write_map,
Florin Coras294afe22019-01-07 17:49:17 -08002661 vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
2662 memset (write_map, 0, vec_len (wrk->wr_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002663 }
Dave Wallace7876d392017-10-19 03:53:57 -04002664 if (n_bits && except_map)
Dave Wallace543852a2017-08-03 02:11:34 -04002665 {
Florin Coras134a9962018-08-28 11:32:04 -07002666 clib_bitmap_validate (wrk->ex_bitmap, minbits);
Dave Barach178cf492018-11-13 16:34:13 -05002667 clib_memcpy_fast (wrk->ex_bitmap, except_map,
Florin Coras294afe22019-01-07 17:49:17 -08002668 vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
2669 memset (except_map, 0, vec_len (wrk->ex_bitmap) * sizeof (vcl_si_set));
Dave Wallace543852a2017-08-03 02:11:34 -04002670 }
2671
Florin Coras54693d22018-07-17 10:46:29 -07002672 if (!n_bits)
2673 return 0;
2674
2675 if (!write_map)
2676 goto check_rd;
2677
Florin Coras096a1d52021-01-27 15:33:51 -08002678 clib_bitmap_foreach (sid, wrk->wr_bitmap)
2679 {
2680 if (!(s = vcl_session_get (wrk, sid)))
2681 {
2682 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2683 bits_set++;
2684 continue;
2685 }
Florin Coras54693d22018-07-17 10:46:29 -07002686
Florin Coras096a1d52021-01-27 15:33:51 -08002687 if (vcl_session_write_ready (s))
2688 {
2689 clib_bitmap_set_no_check ((uword *) write_map, sid, 1);
2690 bits_set++;
2691 }
Florin Coras607eb202023-05-29 16:19:38 -07002692 else
Florin Coras096a1d52021-01-27 15:33:51 -08002693 {
Florin Coras607eb202023-05-29 16:19:38 -07002694 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF);
Florin Coras096a1d52021-01-27 15:33:51 -08002695 }
2696 }
Florin Coras54693d22018-07-17 10:46:29 -07002697
2698check_rd:
2699 if (!read_map)
2700 goto check_mq;
Florin Coras99368312018-08-02 10:45:44 -07002701
Florin Coras096a1d52021-01-27 15:33:51 -08002702 clib_bitmap_foreach (sid, wrk->rd_bitmap)
2703 {
2704 if (!(s = vcl_session_get (wrk, sid)))
2705 {
2706 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2707 bits_set++;
2708 continue;
2709 }
Florin Coras54693d22018-07-17 10:46:29 -07002710
Florin Coras096a1d52021-01-27 15:33:51 -08002711 if (vcl_session_read_ready (s))
2712 {
2713 clib_bitmap_set_no_check ((uword *) read_map, sid, 1);
2714 bits_set++;
2715 }
2716 }
Florin Coras54693d22018-07-17 10:46:29 -07002717
2718check_mq:
Dave Wallace543852a2017-08-03 02:11:34 -04002719
Florin Coras86f04502018-09-12 16:08:01 -07002720 for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
2721 {
2722 vcl_select_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i], n_bits,
2723 read_map, write_map, except_map, &bits_set);
2724 }
2725 vec_reset_length (wrk->unhandled_evts_vector);
2726
Florin Coras99368312018-08-02 10:45:44 -07002727 if (vcm->cfg.use_mq_eventfd)
Florin Coras134a9962018-08-28 11:32:04 -07002728 vppcom_select_eventfd (wrk, n_bits, read_map, write_map, except_map,
Florin Coras99368312018-08-02 10:45:44 -07002729 time_to_wait, &bits_set);
2730 else
Florin Coras134a9962018-08-28 11:32:04 -07002731 vppcom_select_condvar (wrk, n_bits, read_map, write_map, except_map,
Florin Coras99368312018-08-02 10:45:44 -07002732 time_to_wait, &bits_set);
Florin Coras54693d22018-07-17 10:46:29 -07002733
Dave Wallace543852a2017-08-03 02:11:34 -04002734 return (bits_set);
2735}
2736
Dave Wallacef7f809c2017-10-03 01:48:42 -04002737static inline void
Florin Coras7e5e62b2019-07-30 14:08:23 -07002738vep_verify_epoll_chain (vcl_worker_t * wrk, u32 vep_handle)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002739{
Dave Wallacef7f809c2017-10-03 01:48:42 -04002740 vppcom_epoll_t *vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002741 u32 sh = vep_handle;
Florin Coras6c3b2182020-10-19 18:36:48 -07002742 vcl_session_t *s;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002743
Florin Corasaeb7c1c2023-03-10 10:22:21 -08002744 if (VPPCOM_DEBUG <= 3)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002745 return;
2746
Florin Coras6c3b2182020-10-19 18:36:48 -07002747 s = vcl_session_get_w_handle (wrk, vep_handle);
2748 if (PREDICT_FALSE (!s))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002749 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002750 VDBG (0, "ERROR: Invalid vep_sh (%u)!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002751 goto done;
2752 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002753 if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP)))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002754 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002755 VDBG (0, "ERROR: vep_sh (%u) is not a vep!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002756 goto done;
2757 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002758 vep = &s->vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002759 VDBG (0, "vep_sh (%u): Dumping epoll chain\n"
Florin Coras5e062572019-03-14 19:07:51 -07002760 "{\n"
2761 " is_vep = %u\n"
2762 " is_vep_session = %u\n"
Florin Coras7e5e62b2019-07-30 14:08:23 -07002763 " next_sh = 0x%x (%u)\n"
Florin Coras6c3b2182020-10-19 18:36:48 -07002764 "}\n", vep_handle, s->flags & VCL_SESSION_F_IS_VEP,
2765 s->flags & VCL_SESSION_F_IS_VEP_SESSION, vep->next_sh, vep->next_sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002766
Florin Coras7e5e62b2019-07-30 14:08:23 -07002767 for (sh = vep->next_sh; sh != ~0; sh = vep->next_sh)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002768 {
Florin Coras6c3b2182020-10-19 18:36:48 -07002769 s = vcl_session_get_w_handle (wrk, sh);
2770 if (PREDICT_FALSE (!s))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002771 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002772 VDBG (0, "ERROR: Invalid sh (%u)!", sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002773 goto done;
2774 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002775 if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
Florin Coras5e062572019-03-14 19:07:51 -07002776 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002777 VDBG (0, "ERROR: sh (%u) is a vep!", vep_handle);
Florin Coras5e062572019-03-14 19:07:51 -07002778 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002779 else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002780 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002781 VDBG (0, "ERROR: sh (%u) is not a vep session handle!", sh);
Dave Wallace4878cbe2017-11-21 03:45:09 -05002782 goto done;
2783 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002784 vep = &s->vep;
Florin Coras7e5e62b2019-07-30 14:08:23 -07002785 if (PREDICT_FALSE (vep->vep_sh != vep_handle))
2786 VDBG (0, "ERROR: session (%u) vep_sh (%u) != vep_sh (%u)!",
Florin Coras6c3b2182020-10-19 18:36:48 -07002787 sh, s->vep.vep_sh, vep_handle);
2788 if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
Dave Wallace4878cbe2017-11-21 03:45:09 -05002789 {
Florin Coras7e5e62b2019-07-30 14:08:23 -07002790 VDBG (0, "vep_sh[%u]: sh 0x%x (%u)\n"
Florin Coras5e062572019-03-14 19:07:51 -07002791 "{\n"
Florin Coras7e5e62b2019-07-30 14:08:23 -07002792 " next_sh = 0x%x (%u)\n"
2793 " prev_sh = 0x%x (%u)\n"
2794 " vep_sh = 0x%x (%u)\n"
Florin Coras5e062572019-03-14 19:07:51 -07002795 " ev.events = 0x%x\n"
2796 " ev.data.u64 = 0x%llx\n"
2797 " et_mask = 0x%x\n"
2798 "}\n",
Florin Coras7e5e62b2019-07-30 14:08:23 -07002799 vep_handle, sh, sh, vep->next_sh, vep->next_sh, vep->prev_sh,
Florin Coras5e062572019-03-14 19:07:51 -07002800 vep->prev_sh, vep->vep_sh, vep->vep_sh, vep->ev.events,
2801 vep->ev.data.u64, vep->et_mask);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002802 }
2803 }
Dave Wallacef7f809c2017-10-03 01:48:42 -04002804
2805done:
Florin Coras7e5e62b2019-07-30 14:08:23 -07002806 VDBG (0, "vep_sh (%u): Dump complete!\n", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002807}
2808
2809int
2810vppcom_epoll_create (void)
2811{
Florin Coras134a9962018-08-28 11:32:04 -07002812 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07002813 vcl_session_t *vep_session;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002814
Florin Coras134a9962018-08-28 11:32:04 -07002815 vep_session = vcl_session_alloc (wrk);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002816
Florin Coras6c3b2182020-10-19 18:36:48 -07002817 vep_session->flags |= VCL_SESSION_F_IS_VEP;
Florin Coras134a9962018-08-28 11:32:04 -07002818 vep_session->vep.vep_sh = ~0;
2819 vep_session->vep.next_sh = ~0;
2820 vep_session->vep.prev_sh = ~0;
Florin Coras980ee742023-08-04 16:14:01 -07002821 vep_session->vpp_handle = SESSION_INVALID_HANDLE;
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08002822
Florin Corasa7a1a222018-12-30 17:11:31 -08002823 vcl_evt (VCL_EVT_EPOLL_CREATE, vep_session, vep_session->session_index);
2824 VDBG (0, "Created vep_idx %u", vep_session->session_index);
Keith Burns (alagalah)09b27842018-01-05 14:39:59 -08002825
Florin Corasab2f6db2018-08-31 14:31:41 -07002826 return vcl_session_handle (vep_session);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002827}
2828
Florin Corasb0116a12023-02-07 09:11:47 -08002829static void
2830vcl_epoll_ctl_add_unhandled_event (vcl_worker_t *wrk, vcl_session_t *s,
Florin Corase81f27f2024-02-12 22:33:41 -05002831 u32 is_epollet, session_evt_type_t evt)
Florin Corasb0116a12023-02-07 09:11:47 -08002832{
2833 if (!is_epollet)
2834 {
Florin Corasc2a14172023-02-28 21:13:50 -08002835 if (s->vep.lt_next == VCL_INVALID_SESSION_INDEX)
2836 vcl_epoll_lt_add (wrk, s);
Florin Corasb0116a12023-02-07 09:11:47 -08002837 return;
2838 }
2839
2840 session_event_t e = { 0 };
2841 e.session_index = s->session_index;
2842 e.event_type = evt;
2843 if (evt == SESSION_IO_EVT_RX)
2844 s->flags &= ~VCL_SESSION_F_HAS_RX_EVT;
2845 vec_add1 (wrk->unhandled_evts_vector, e);
2846}
2847
Dave Wallacef7f809c2017-10-03 01:48:42 -04002848int
Florin Coras134a9962018-08-28 11:32:04 -07002849vppcom_epoll_ctl (uint32_t vep_handle, int op, uint32_t session_handle,
Dave Wallacef7f809c2017-10-03 01:48:42 -04002850 struct epoll_event *event)
2851{
Florin Coras134a9962018-08-28 11:32:04 -07002852 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras87f76002021-06-28 19:13:29 -07002853 int rv = VPPCOM_OK, add_evt = 0;
Florin Coras7e12d942018-06-27 14:32:43 -07002854 vcl_session_t *vep_session;
Florin Coras17ec5772020-08-12 20:46:29 -07002855 vcl_session_t *s;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002856
Florin Coras134a9962018-08-28 11:32:04 -07002857 if (vep_handle == session_handle)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002858 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002859 VDBG (0, "vep_sh == session handle (%u)!", vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002860 return VPPCOM_EINVAL;
2861 }
2862
Florin Coras134a9962018-08-28 11:32:04 -07002863 vep_session = vcl_session_get_w_handle (wrk, vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002864 if (PREDICT_FALSE (!vep_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002865 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002866 VDBG (0, "Invalid vep_sh (%u)!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002867 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002868 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002869 if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002870 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002871 VDBG (0, "vep_sh (%u) is not a vep!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002872 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002873 }
2874
Florin Coras134a9962018-08-28 11:32:04 -07002875 ASSERT (vep_session->vep.vep_sh == ~0);
2876 ASSERT (vep_session->vep.prev_sh == ~0);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002877
Florin Coras17ec5772020-08-12 20:46:29 -07002878 s = vcl_session_get_w_handle (wrk, session_handle);
2879 if (PREDICT_FALSE (!s))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002880 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002881 VDBG (0, "Invalid session_handle (%u)!", session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002882 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002883 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002884 if (PREDICT_FALSE (s->flags & VCL_SESSION_F_IS_VEP))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002885 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002886 VDBG (0, "session_handle (%u) is a vep!", vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002887 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002888 }
2889
2890 switch (op)
2891 {
2892 case EPOLL_CTL_ADD:
2893 if (PREDICT_FALSE (!event))
2894 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002895 VDBG (0, "EPOLL_CTL_ADD: NULL pointer to epoll_event structure!");
Florin Coras070453d2018-08-24 17:04:27 -07002896 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002897 }
Florin Coras2645f682021-06-04 15:59:41 -07002898 if (s->flags & VCL_SESSION_F_IS_VEP_SESSION)
2899 {
2900 VDBG (0, "EPOLL_CTL_ADD: %u already epolled!", s->session_index);
2901 rv = VPPCOM_EEXIST;
2902 goto done;
2903 }
Florin Coras134a9962018-08-28 11:32:04 -07002904 if (vep_session->vep.next_sh != ~0)
Dave Wallacef7f809c2017-10-03 01:48:42 -04002905 {
Florin Coras7e12d942018-06-27 14:32:43 -07002906 vcl_session_t *next_session;
Florin Corasab2f6db2018-08-31 14:31:41 -07002907 next_session = vcl_session_get_w_handle (wrk,
2908 vep_session->vep.next_sh);
Florin Coras070453d2018-08-24 17:04:27 -07002909 if (PREDICT_FALSE (!next_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002910 {
Florin Coras5e062572019-03-14 19:07:51 -07002911 VDBG (0, "EPOLL_CTL_ADD: Invalid vep.next_sh (%u) on "
Florin Corasa7a1a222018-12-30 17:11:31 -08002912 "vep_idx (%u)!", vep_session->vep.next_sh, vep_handle);
Florin Coras070453d2018-08-24 17:04:27 -07002913 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002914 }
Florin Coras134a9962018-08-28 11:32:04 -07002915 ASSERT (next_session->vep.prev_sh == vep_handle);
2916 next_session->vep.prev_sh = session_handle;
Dave Wallacef7f809c2017-10-03 01:48:42 -04002917 }
Florin Coras17ec5772020-08-12 20:46:29 -07002918 s->vep.next_sh = vep_session->vep.next_sh;
2919 s->vep.prev_sh = vep_handle;
2920 s->vep.vep_sh = vep_handle;
2921 s->vep.et_mask = VEP_DEFAULT_ET_MASK;
Florin Corasfa3884f2021-06-22 20:04:46 -07002922 s->vep.lt_next = VCL_INVALID_SESSION_INDEX;
Florin Coras17ec5772020-08-12 20:46:29 -07002923 s->vep.ev = *event;
Florin Coras30ecfa82023-06-14 11:15:36 -07002924 s->vep.ev.events |= EPOLLHUP | EPOLLERR;
Florin Coras6c3b2182020-10-19 18:36:48 -07002925 s->flags &= ~VCL_SESSION_F_IS_VEP;
2926 s->flags |= VCL_SESSION_F_IS_VEP_SESSION;
Florin Coras134a9962018-08-28 11:32:04 -07002927 vep_session->vep.next_sh = session_handle;
Keith Burns (alagalah)3cf2d642018-02-23 10:17:01 -08002928
Florin Coras470d72f2023-06-01 21:50:03 -07002929 if ((event->events & EPOLLOUT))
hanlin475c9d72019-12-26 11:44:28 +08002930 {
Florin Coras470d72f2023-06-01 21:50:03 -07002931 int write_ready = vcl_session_write_ready (s);
2932
2933 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2934 if (write_ready > 0)
2935 {
2936 /* Generate EPOLLOUT if tx fifo not full */
2937 vcl_epoll_ctl_add_unhandled_event (
2938 wrk, s, event->events & EPOLLET, SESSION_IO_EVT_TX);
2939 add_evt = 1;
2940 }
2941 else
2942 {
2943 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF);
2944 }
hanlin475c9d72019-12-26 11:44:28 +08002945 }
Florin Coras6e3c1f82020-01-15 01:30:46 +00002946 /* Generate EPOLLIN if rx fifo has data */
Florin Coras17ec5772020-08-12 20:46:29 -07002947 if ((event->events & EPOLLIN) && (vcl_session_read_ready (s) > 0))
Florin Coras6e3c1f82020-01-15 01:30:46 +00002948 {
Florin Corasb0116a12023-02-07 09:11:47 -08002949 vcl_epoll_ctl_add_unhandled_event (wrk, s, event->events & EPOLLET,
2950 SESSION_IO_EVT_RX);
Florin Coras87f76002021-06-28 19:13:29 -07002951 add_evt = 1;
2952 }
2953 if (!add_evt && vcl_session_is_closing (s))
2954 {
2955 session_event_t e = { 0 };
2956 if (s->session_state == VCL_STATE_VPP_CLOSING)
2957 e.event_type = SESSION_CTRL_EVT_DISCONNECTED;
2958 else
2959 e.event_type = SESSION_CTRL_EVT_RESET;
2960 e.session_index = s->session_index;
2961 e.postponed = 1;
2962 vec_add1 (wrk->unhandled_evts_vector, e);
Florin Coras6e3c1f82020-01-15 01:30:46 +00002963 }
Florin Corasa7a1a222018-12-30 17:11:31 -08002964 VDBG (1, "EPOLL_CTL_ADD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
2965 vep_handle, session_handle, event->events, event->data.u64);
Florin Coras17ec5772020-08-12 20:46:29 -07002966 vcl_evt (VCL_EVT_EPOLL_CTLADD, s, event->events, event->data.u64);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002967 break;
2968
2969 case EPOLL_CTL_MOD:
2970 if (PREDICT_FALSE (!event))
2971 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002972 VDBG (0, "EPOLL_CTL_MOD: NULL pointer to epoll_event structure!");
Dave Wallacef7f809c2017-10-03 01:48:42 -04002973 rv = VPPCOM_EINVAL;
2974 goto done;
2975 }
Florin Coras6c3b2182020-10-19 18:36:48 -07002976 else if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
Dave Wallacef7f809c2017-10-03 01:48:42 -04002977 {
Florin Coras5e062572019-03-14 19:07:51 -07002978 VDBG (0, "sh %u EPOLL_CTL_MOD: not a vep session!", session_handle);
Florin Coras2645f682021-06-04 15:59:41 -07002979 rv = VPPCOM_ENOENT;
Dave Wallace4878cbe2017-11-21 03:45:09 -05002980 goto done;
2981 }
Florin Coras17ec5772020-08-12 20:46:29 -07002982 else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
Dave Wallace4878cbe2017-11-21 03:45:09 -05002983 {
Florin Corasa7a1a222018-12-30 17:11:31 -08002984 VDBG (0, "EPOLL_CTL_MOD: sh %u vep_sh (%u) != vep_sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07002985 session_handle, s->vep.vep_sh, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04002986 rv = VPPCOM_EINVAL;
2987 goto done;
2988 }
hanlin475c9d72019-12-26 11:44:28 +08002989
Florin Coras470d72f2023-06-01 21:50:03 -07002990 /* Generate EPOLLOUT if session write ready and event was not on */
2991 if ((event->events & EPOLLOUT) && !(s->vep.ev.events & EPOLLOUT))
2992 {
2993 /* Fifo size load acq synchronized with update store rel */
2994 int write_ready = vcl_session_write_ready (s);
2995
2996 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
2997 if (write_ready > 0)
2998 vcl_epoll_ctl_add_unhandled_event (wrk, s, event->events & EPOLLET,
2999 SESSION_IO_EVT_TX);
3000 else
3001 /* Request deq ntf in case dequeue happened while updating flag */
3002 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF);
3003 }
3004 else if (!(event->events & EPOLLOUT))
Florin Coras607eb202023-05-29 16:19:38 -07003005 vcl_session_del_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Corasbc4d5b02023-05-17 23:20:56 -07003006
Florin Coras2645f682021-06-04 15:59:41 -07003007 /* Generate EPOLLIN if session read ready and event was not on */
3008 if ((event->events & EPOLLIN) && !(s->vep.ev.events & EPOLLIN) &&
3009 (vcl_session_read_ready (s) > 0))
3010 {
Florin Corasb0116a12023-02-07 09:11:47 -08003011 vcl_epoll_ctl_add_unhandled_event (wrk, s, event->events & EPOLLET,
3012 SESSION_IO_EVT_RX);
Florin Coras2645f682021-06-04 15:59:41 -07003013 }
Florin Coras17ec5772020-08-12 20:46:29 -07003014 s->vep.et_mask = VEP_DEFAULT_ET_MASK;
3015 s->vep.ev = *event;
Florin Coras30ecfa82023-06-14 11:15:36 -07003016 s->vep.ev.events |= EPOLLHUP | EPOLLERR;
Florin Corasbc4d5b02023-05-17 23:20:56 -07003017
Florin Corasa7a1a222018-12-30 17:11:31 -08003018 VDBG (1, "EPOLL_CTL_MOD: vep_sh %u, sh %u, events 0x%x, data 0x%llx!",
3019 vep_handle, session_handle, event->events, event->data.u64);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003020 break;
3021
3022 case EPOLL_CTL_DEL:
Florin Coras6c3b2182020-10-19 18:36:48 -07003023 if (PREDICT_FALSE (!(s->flags & VCL_SESSION_F_IS_VEP_SESSION)))
Dave Wallacef7f809c2017-10-03 01:48:42 -04003024 {
Florin Corasa7a1a222018-12-30 17:11:31 -08003025 VDBG (0, "EPOLL_CTL_DEL: %u not a vep session!", session_handle);
Florin Coras2645f682021-06-04 15:59:41 -07003026 rv = VPPCOM_ENOENT;
Dave Wallace4878cbe2017-11-21 03:45:09 -05003027 goto done;
3028 }
Florin Coras17ec5772020-08-12 20:46:29 -07003029 else if (PREDICT_FALSE (s->vep.vep_sh != vep_handle))
Dave Wallace4878cbe2017-11-21 03:45:09 -05003030 {
Florin Corasa7a1a222018-12-30 17:11:31 -08003031 VDBG (0, "EPOLL_CTL_DEL: sh %u vep_sh (%u) != vep_sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07003032 session_handle, s->vep.vep_sh, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003033 rv = VPPCOM_EINVAL;
3034 goto done;
3035 }
3036
Florin Coras17ec5772020-08-12 20:46:29 -07003037 if (s->vep.prev_sh == vep_handle)
3038 vep_session->vep.next_sh = s->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003039 else
3040 {
Florin Coras7e12d942018-06-27 14:32:43 -07003041 vcl_session_t *prev_session;
Florin Coras17ec5772020-08-12 20:46:29 -07003042 prev_session = vcl_session_get_w_handle (wrk, s->vep.prev_sh);
Florin Coras070453d2018-08-24 17:04:27 -07003043 if (PREDICT_FALSE (!prev_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04003044 {
Florin Coras5e062572019-03-14 19:07:51 -07003045 VDBG (0, "EPOLL_CTL_DEL: Invalid prev_sh (%u) on sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07003046 s->vep.prev_sh, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07003047 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003048 }
Florin Coras134a9962018-08-28 11:32:04 -07003049 ASSERT (prev_session->vep.next_sh == session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07003050 prev_session->vep.next_sh = s->vep.next_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003051 }
Florin Coras17ec5772020-08-12 20:46:29 -07003052 if (s->vep.next_sh != ~0)
Dave Wallacef7f809c2017-10-03 01:48:42 -04003053 {
Florin Coras7e12d942018-06-27 14:32:43 -07003054 vcl_session_t *next_session;
Florin Coras17ec5772020-08-12 20:46:29 -07003055 next_session = vcl_session_get_w_handle (wrk, s->vep.next_sh);
Florin Coras070453d2018-08-24 17:04:27 -07003056 if (PREDICT_FALSE (!next_session))
Dave Wallacef7f809c2017-10-03 01:48:42 -04003057 {
Florin Coras5e062572019-03-14 19:07:51 -07003058 VDBG (0, "EPOLL_CTL_DEL: Invalid next_sh (%u) on sh (%u)!",
Florin Coras17ec5772020-08-12 20:46:29 -07003059 s->vep.next_sh, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07003060 return VPPCOM_EBADFD;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003061 }
Florin Coras134a9962018-08-28 11:32:04 -07003062 ASSERT (next_session->vep.prev_sh == session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07003063 next_session->vep.prev_sh = s->vep.prev_sh;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003064 }
3065
Florin Corasfa3884f2021-06-22 20:04:46 -07003066 if (s->vep.lt_next != VCL_INVALID_SESSION_INDEX)
3067 vcl_epoll_lt_del (wrk, s);
3068
Florin Coras17ec5772020-08-12 20:46:29 -07003069 memset (&s->vep, 0, sizeof (s->vep));
3070 s->vep.next_sh = ~0;
3071 s->vep.prev_sh = ~0;
3072 s->vep.vep_sh = ~0;
Florin Corasfa3884f2021-06-22 20:04:46 -07003073 s->vep.lt_next = VCL_INVALID_SESSION_INDEX;
Florin Coras6c3b2182020-10-19 18:36:48 -07003074 s->flags &= ~VCL_SESSION_F_IS_VEP_SESSION;
Florin Coras1bcad5c2019-01-09 20:04:38 -08003075
Florin Corasf1ddeeb2021-06-10 08:08:53 -07003076 if (vcl_session_is_open (s))
Florin Coras607eb202023-05-29 16:19:38 -07003077 vcl_session_del_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Coras1bcad5c2019-01-09 20:04:38 -08003078
Florin Coras5e062572019-03-14 19:07:51 -07003079 VDBG (1, "EPOLL_CTL_DEL: vep_idx %u, sh %u!", vep_handle,
Florin Corasa7a1a222018-12-30 17:11:31 -08003080 session_handle);
Florin Coras17ec5772020-08-12 20:46:29 -07003081 vcl_evt (VCL_EVT_EPOLL_CTLDEL, s, vep_sh);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003082 break;
3083
3084 default:
Florin Corasa7a1a222018-12-30 17:11:31 -08003085 VDBG (0, "Invalid operation (%d)!", op);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003086 rv = VPPCOM_EINVAL;
3087 }
3088
Florin Coras134a9962018-08-28 11:32:04 -07003089 vep_verify_epoll_chain (wrk, vep_handle);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003090
3091done:
Dave Wallacef7f809c2017-10-03 01:48:42 -04003092 return rv;
3093}
3094
Florin Coras30ecfa82023-06-14 11:15:36 -07003095always_inline u8
3096vcl_ep_session_needs_evt (vcl_session_t *s, u32 evt)
3097{
3098 /* No event if not epolled / events reset on hup or level-trigger on */
3099 return ((s->vep.ev.events & evt) &&
3100 s->vep.lt_next == VCL_INVALID_SESSION_INDEX);
3101}
3102
Florin Coras86f04502018-09-12 16:08:01 -07003103static inline void
3104vcl_epoll_wait_handle_mq_event (vcl_worker_t * wrk, session_event_t * e,
3105 struct epoll_event *events, u32 * num_ev)
Florin Coras54693d22018-07-17 10:46:29 -07003106{
3107 session_disconnected_msg_t *disconnected_msg;
3108 session_connected_msg_t *connected_msg;
Florin Coras99368312018-08-02 10:45:44 -07003109 u32 sid = ~0, session_events;
Florin Coras3c7d4f92018-12-14 11:28:43 -08003110 u64 session_evt_data = ~0;
Florin Corasb242d312020-10-26 15:35:40 -07003111 vcl_session_t *s;
Florin Coras86f04502018-09-12 16:08:01 -07003112 u8 add_event = 0;
3113
3114 switch (e->event_type)
3115 {
Florin Coras653e43f2019-03-04 10:56:23 -08003116 case SESSION_IO_EVT_RX:
Florin Corasc0737e92019-03-04 14:19:39 -08003117 sid = e->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07003118 s = vcl_session_get (wrk, sid);
3119 if (vcl_session_is_closed (s))
Florin Corasfa915f82018-12-26 16:29:06 -08003120 break;
Florin Corasb242d312020-10-26 15:35:40 -07003121 vcl_fifo_rx_evt_valid_or_break (s);
Florin Coras30ecfa82023-06-14 11:15:36 -07003122 if (!vcl_ep_session_needs_evt (s, EPOLLIN) ||
3123 (s->flags & VCL_SESSION_F_HAS_RX_EVT))
Florin Coras86f04502018-09-12 16:08:01 -07003124 break;
Florin Coras30ecfa82023-06-14 11:15:36 -07003125 session_events = s->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07003126 add_event = 1;
wanghanlin9e42cc22021-06-25 17:40:13 +08003127 events[*num_ev].events = EPOLLIN;
Florin Corasb242d312020-10-26 15:35:40 -07003128 session_evt_data = s->vep.ev.data.u64;
3129 s->flags |= VCL_SESSION_F_HAS_RX_EVT;
Florin Coras86f04502018-09-12 16:08:01 -07003130 break;
Florin Coras653e43f2019-03-04 10:56:23 -08003131 case SESSION_IO_EVT_TX:
Florin Corasc0737e92019-03-04 14:19:39 -08003132 sid = e->session_index;
Florin Corasb242d312020-10-26 15:35:40 -07003133 s = vcl_session_get (wrk, sid);
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003134 if (!s || !vcl_session_is_open (s))
Florin Corasfa915f82018-12-26 16:29:06 -08003135 break;
Florin Coras48178552023-05-17 22:59:40 -07003136 svm_fifo_reset_has_deq_ntf (vcl_session_is_ct (s) ? s->ct_tx_fifo :
3137 s->tx_fifo);
Florin Coras30ecfa82023-06-14 11:15:36 -07003138 if (!vcl_ep_session_needs_evt (s, EPOLLOUT))
Florin Coras86f04502018-09-12 16:08:01 -07003139 break;
Florin Coras30ecfa82023-06-14 11:15:36 -07003140 session_events = s->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07003141 add_event = 1;
wanghanlin9e42cc22021-06-25 17:40:13 +08003142 events[*num_ev].events = EPOLLOUT;
Florin Corasb242d312020-10-26 15:35:40 -07003143 session_evt_data = s->vep.ev.data.u64;
Florin Coras86f04502018-09-12 16:08:01 -07003144 break;
Florin Coras86f04502018-09-12 16:08:01 -07003145 case SESSION_CTRL_EVT_ACCEPTED:
Florin Corasb242d312020-10-26 15:35:40 -07003146 if (!e->postponed)
3147 s = vcl_session_accepted (wrk, (session_accepted_msg_t *) e->data);
3148 else
3149 s = vcl_session_get (wrk, e->session_index);
Florin Coras30ecfa82023-06-14 11:15:36 -07003150 if (!s || !vcl_ep_session_needs_evt (s, EPOLLIN))
Florin Coras3c7d4f92018-12-14 11:28:43 -08003151 break;
Florin Corasb242d312020-10-26 15:35:40 -07003152 sid = s->session_index;
Florin Coras30ecfa82023-06-14 11:15:36 -07003153 session_events = s->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07003154 add_event = 1;
wanghanlin9e42cc22021-06-25 17:40:13 +08003155 events[*num_ev].events = EPOLLIN;
Florin Corasb242d312020-10-26 15:35:40 -07003156 session_evt_data = s->vep.ev.data.u64;
Florin Coras86f04502018-09-12 16:08:01 -07003157 break;
3158 case SESSION_CTRL_EVT_CONNECTED:
Florin Corasb242d312020-10-26 15:35:40 -07003159 if (!e->postponed)
3160 {
3161 connected_msg = (session_connected_msg_t *) e->data;
3162 sid = vcl_session_connected_handler (wrk, connected_msg);
3163 }
3164 else
3165 sid = e->session_index;
3166 s = vcl_session_get (wrk, sid);
Florin Coras30ecfa82023-06-14 11:15:36 -07003167 if (vcl_session_is_closed (s) || !vcl_ep_session_needs_evt (s, EPOLLOUT))
Florin Coras72f77822019-01-22 19:05:52 -08003168 break;
Florin Coras1d84abc2023-01-13 09:44:14 -08003169 /* We didn't have a fifo when the event was added */
Florin Coras607eb202023-05-29 16:19:38 -07003170 vcl_session_add_want_deq_ntf (s, SVM_FIFO_WANT_DEQ_NOTIF_IF_FULL);
Florin Coras72f77822019-01-22 19:05:52 -08003171 add_event = 1;
Florin Coras30ecfa82023-06-14 11:15:36 -07003172 session_events = s->vep.ev.events;
3173 /* Generate EPOLLOUT because there's no connected event */
wanghanlin9e42cc22021-06-25 17:40:13 +08003174 events[*num_ev].events = EPOLLOUT;
Florin Corasb242d312020-10-26 15:35:40 -07003175 session_evt_data = s->vep.ev.data.u64;
3176 if (s->session_state == VCL_STATE_DETACHED)
Florin Coras30ecfa82023-06-14 11:15:36 -07003177 {
3178 events[*num_ev].events |= EPOLLHUP;
3179 s->vep.ev.events = 0;
3180 }
Florin Coras86f04502018-09-12 16:08:01 -07003181 break;
3182 case SESSION_CTRL_EVT_DISCONNECTED:
Florin Coras87f76002021-06-28 19:13:29 -07003183 if (!e->postponed)
3184 {
3185 disconnected_msg = (session_disconnected_msg_t *) e->data;
3186 s = vcl_session_disconnected_handler (wrk, disconnected_msg);
3187 }
3188 else
3189 {
3190 s = vcl_session_get (wrk, e->session_index);
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003191 s->flags &= ~VCL_SESSION_F_PENDING_DISCONNECT;
Florin Coras87f76002021-06-28 19:13:29 -07003192 }
Florin Coras30ecfa82023-06-14 11:15:36 -07003193 if (vcl_session_is_closed (s) || !vcl_ep_session_needs_evt (s, EPOLLHUP))
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003194 {
Florin Coras74254b52021-12-08 11:03:08 -08003195 if (s && (s->flags & VCL_SESSION_F_PENDING_FREE))
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003196 vcl_session_free (wrk, s);
3197 break;
3198 }
Florin Corasb242d312020-10-26 15:35:40 -07003199 sid = s->session_index;
3200 session_events = s->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07003201 add_event = 1;
liuyacanffd9ddb2021-11-01 10:22:09 +08003202 if (EPOLLRDHUP & session_events)
3203 {
3204 /* If app can distinguish between RDHUP and HUP,
3205 * we make finer control */
3206 events[*num_ev].events = EPOLLRDHUP;
3207 if (s->flags & VCL_SESSION_F_WR_SHUTDOWN)
3208 {
3209 events[*num_ev].events |= EPOLLHUP;
3210 }
3211 }
3212 else
3213 {
3214 events[*num_ev].events = EPOLLHUP;
3215 }
Florin Corasb242d312020-10-26 15:35:40 -07003216 session_evt_data = s->vep.ev.data.u64;
Florin Coras30ecfa82023-06-14 11:15:36 -07003217 s->vep.ev.events = 0;
Florin Coras86f04502018-09-12 16:08:01 -07003218 break;
Florin Coras01ee7a72023-03-01 00:49:25 -08003219 case SESSION_CTRL_EVT_BOUND:
3220 vcl_session_bound_handler (wrk, (session_bound_msg_t *) e->data);
3221 break;
Florin Coras86f04502018-09-12 16:08:01 -07003222 case SESSION_CTRL_EVT_RESET:
Florin Coras87f76002021-06-28 19:13:29 -07003223 if (!e->postponed)
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003224 {
3225 sid =
3226 vcl_session_reset_handler (wrk, (session_reset_msg_t *) e->data);
3227 s = vcl_session_get (wrk, sid);
3228 }
Florin Coras87f76002021-06-28 19:13:29 -07003229 else
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003230 {
3231 sid = e->session_index;
3232 s = vcl_session_get (wrk, sid);
3233 s->flags &= ~VCL_SESSION_F_PENDING_DISCONNECT;
3234 }
Florin Coras30ecfa82023-06-14 11:15:36 -07003235 if (vcl_session_is_closed (s) || !vcl_ep_session_needs_evt (s, EPOLLHUP))
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003236 {
Florin Coras74254b52021-12-08 11:03:08 -08003237 if (s && (s->flags & VCL_SESSION_F_PENDING_FREE))
Florin Corasf0fe1ea2021-12-03 17:03:37 -08003238 vcl_session_free (wrk, s);
3239 break;
3240 }
Florin Corasb242d312020-10-26 15:35:40 -07003241 session_events = s->vep.ev.events;
Florin Coras86f04502018-09-12 16:08:01 -07003242 add_event = 1;
Yacan Liuf5e0a172022-09-20 14:19:19 +08003243 events[*num_ev].events = EPOLLERR | EPOLLHUP;
3244 if ((EPOLLRDHUP & session_events) &&
3245 (s->flags & VCL_SESSION_F_RD_SHUTDOWN))
3246 {
Yacan Liuab157702022-09-26 16:41:32 +08003247 events[*num_ev].events |= EPOLLRDHUP;
Yacan Liuf5e0a172022-09-20 14:19:19 +08003248 }
3249 if ((EPOLLIN & session_events) && (s->flags & VCL_SESSION_F_RD_SHUTDOWN))
3250 {
3251 events[*num_ev].events |= EPOLLIN;
3252 }
Florin Corasb242d312020-10-26 15:35:40 -07003253 session_evt_data = s->vep.ev.data.u64;
Florin Coras30ecfa82023-06-14 11:15:36 -07003254 s->vep.ev.events = 0;
Florin Coras86f04502018-09-12 16:08:01 -07003255 break;
Florin Corasdfae9f92019-02-20 19:48:31 -08003256 case SESSION_CTRL_EVT_UNLISTEN_REPLY:
3257 vcl_session_unlisten_reply_handler (wrk, e->data);
3258 break;
Florin Coras68b7e582020-01-21 18:33:23 -08003259 case SESSION_CTRL_EVT_MIGRATED:
3260 vcl_session_migrated_handler (wrk, e->data);
3261 break;
Florin Coras9ace36d2019-10-28 13:14:17 -07003262 case SESSION_CTRL_EVT_CLEANUP:
3263 vcl_session_cleanup_handler (wrk, e->data);
3264 break;
Florin Coras30e79c22019-01-02 19:31:22 -08003265 case SESSION_CTRL_EVT_REQ_WORKER_UPDATE:
3266 vcl_session_req_worker_update_handler (wrk, e->data);
3267 break;
3268 case SESSION_CTRL_EVT_WORKER_UPDATE_REPLY:
3269 vcl_session_worker_update_reply_handler (wrk, e->data);
3270 break;
Florin Corasc4c4cf52019-08-24 18:17:34 -07003271 case SESSION_CTRL_EVT_APP_ADD_SEGMENT:
3272 vcl_session_app_add_segment_handler (wrk, e->data);
3273 break;
3274 case SESSION_CTRL_EVT_APP_DEL_SEGMENT:
3275 vcl_session_app_del_segment_handler (wrk, e->data);
3276 break;
hanlina3a48962020-07-13 11:09:15 +08003277 case SESSION_CTRL_EVT_APP_WRK_RPC:
Florin Coras40c07ce2020-07-16 20:46:17 -07003278 vcl_worker_rpc_handler (wrk, e->data);
3279 break;
Florin Coras86f04502018-09-12 16:08:01 -07003280 default:
3281 VDBG (0, "unhandled: %u", e->event_type);
3282 break;
3283 }
3284
3285 if (add_event)
3286 {
Florin Coras30ecfa82023-06-14 11:15:36 -07003287 ASSERT (s->flags & VCL_SESSION_F_IS_VEP_SESSION);
Florin Coras86f04502018-09-12 16:08:01 -07003288 events[*num_ev].data.u64 = session_evt_data;
3289 if (EPOLLONESHOT & session_events)
3290 {
Florin Corasb242d312020-10-26 15:35:40 -07003291 s = vcl_session_get (wrk, sid);
Florin Coras30ecfa82023-06-14 11:15:36 -07003292 if (!(events[*num_ev].events & EPOLLHUP))
3293 s->vep.ev.events = EPOLLHUP | EPOLLERR;
Florin Coras86f04502018-09-12 16:08:01 -07003294 }
Florin Corasfa3884f2021-06-22 20:04:46 -07003295 else if (!(EPOLLET & session_events))
Florin Corasfe286f72021-06-04 10:07:55 -07003296 {
Florin Corasfa3884f2021-06-22 20:04:46 -07003297 s = vcl_session_get (wrk, sid);
3298 if (s->vep.lt_next == VCL_INVALID_SESSION_INDEX)
3299 vcl_epoll_lt_add (wrk, s);
Florin Corasfe286f72021-06-04 10:07:55 -07003300 }
Florin Coras86f04502018-09-12 16:08:01 -07003301 *num_ev += 1;
3302 }
3303}
3304
3305static int
3306vcl_epoll_wait_handle_mq (vcl_worker_t * wrk, svm_msg_q_t * mq,
3307 struct epoll_event *events, u32 maxevents,
3308 double wait_for_time, u32 * num_ev)
3309{
Florin Coras99368312018-08-02 10:45:44 -07003310 svm_msg_q_msg_t *msg;
Florin Coras54693d22018-07-17 10:46:29 -07003311 session_event_t *e;
Florin Coras54693d22018-07-17 10:46:29 -07003312 int i;
3313
Florin Coras539663c2018-09-28 14:59:37 -07003314 if (vec_len (wrk->mq_msg_vector) && svm_msg_q_is_empty (mq))
3315 goto handle_dequeued;
3316
Florin Coras54693d22018-07-17 10:46:29 -07003317 if (svm_msg_q_is_empty (mq))
3318 {
3319 if (!wait_for_time)
Florin Coras5398dfb2021-01-25 20:31:27 -08003320 return 0;
Florin Coras54693d22018-07-17 10:46:29 -07003321 else if (wait_for_time < 0)
Florin Coras5398dfb2021-01-25 20:31:27 -08003322 svm_msg_q_wait (mq, SVM_MQ_WAIT_EMPTY);
Florin Coras54693d22018-07-17 10:46:29 -07003323 else
3324 {
Florin Coras60f1fc12018-08-16 14:57:31 -07003325 if (svm_msg_q_timedwait (mq, wait_for_time / 1e3))
Florin Coras5398dfb2021-01-25 20:31:27 -08003326 return 0;
Florin Coras54693d22018-07-17 10:46:29 -07003327 }
3328 }
Florin Corase003a1b2019-06-05 10:47:16 -07003329 ASSERT (maxevents > *num_ev);
hanlind0e646f2020-05-11 22:20:37 +08003330 vcl_mq_dequeue_batch (wrk, mq, ~0);
Florin Coras54693d22018-07-17 10:46:29 -07003331
Florin Coras539663c2018-09-28 14:59:37 -07003332handle_dequeued:
Florin Coras134a9962018-08-28 11:32:04 -07003333 for (i = 0; i < vec_len (wrk->mq_msg_vector); i++)
Florin Coras54693d22018-07-17 10:46:29 -07003334 {
Florin Coras134a9962018-08-28 11:32:04 -07003335 msg = vec_elt_at_index (wrk->mq_msg_vector, i);
Florin Coras99368312018-08-02 10:45:44 -07003336 e = svm_msg_q_msg_data (mq, msg);
hanlind0e646f2020-05-11 22:20:37 +08003337 if (*num_ev < maxevents)
3338 vcl_epoll_wait_handle_mq_event (wrk, e, events, num_ev);
3339 else
3340 vcl_handle_mq_event (wrk, e);
Florin Coras99368312018-08-02 10:45:44 -07003341 svm_msg_q_free_msg (mq, msg);
Florin Coras54693d22018-07-17 10:46:29 -07003342 }
Florin Corasaa27eb92018-10-13 12:20:01 -07003343 vec_reset_length (wrk->mq_msg_vector);
Florin Coras30e79c22019-01-02 19:31:22 -08003344 vcl_handle_pending_wrk_updates (wrk);
Florin Coras54693d22018-07-17 10:46:29 -07003345 return *num_ev;
3346}
3347
Florin Coras99368312018-08-02 10:45:44 -07003348static int
Florin Corasccdb8b82021-04-28 13:01:06 -07003349vppcom_epoll_wait_condvar (vcl_worker_t *wrk, struct epoll_event *events,
3350 int maxevents, u32 n_evts, double timeout_ms)
Florin Coras99368312018-08-02 10:45:44 -07003351{
Florin Corasccdb8b82021-04-28 13:01:06 -07003352 double end = -1;
Florin Coras14ed6df2019-03-06 21:13:42 -08003353
3354 if (!n_evts)
3355 {
Florin Corasccdb8b82021-04-28 13:01:06 -07003356 if (timeout_ms > 0)
3357 end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
Florin Coras14ed6df2019-03-06 21:13:42 -08003358 }
3359
3360 do
3361 {
3362 vcl_epoll_wait_handle_mq (wrk, wrk->app_event_queue, events, maxevents,
Florin Corasccdb8b82021-04-28 13:01:06 -07003363 timeout_ms, &n_evts);
3364 if (n_evts || !timeout_ms)
Florin Coras14ed6df2019-03-06 21:13:42 -08003365 return n_evts;
Florin Coras14ed6df2019-03-06 21:13:42 -08003366 }
Florin Corasccdb8b82021-04-28 13:01:06 -07003367 while (end == -1 || clib_time_now (&wrk->clib_time) < end);
Florin Coras14ed6df2019-03-06 21:13:42 -08003368
3369 return 0;
Florin Coras99368312018-08-02 10:45:44 -07003370}
3371
3372static int
Florin Corasccdb8b82021-04-28 13:01:06 -07003373vppcom_epoll_wait_eventfd (vcl_worker_t *wrk, struct epoll_event *events,
3374 int maxevents, u32 n_evts, double timeout_ms)
Florin Coras99368312018-08-02 10:45:44 -07003375{
Florin Coras99368312018-08-02 10:45:44 -07003376 int __clib_unused n_read;
Florin Corasb13ba132021-01-27 16:05:24 -08003377 vcl_mq_evt_conn_t *mqc;
Florin Coras99368312018-08-02 10:45:44 -07003378 int n_mq_evts, i;
Florin Corasccdb8b82021-04-28 13:01:06 -07003379 double end = -1;
Florin Coras99368312018-08-02 10:45:44 -07003380 u64 buf;
3381
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00003382 if (PREDICT_FALSE (wrk->api_client_handle == ~0))
3383 {
3384 vcl_api_retry_attach (wrk);
3385 return n_evts;
3386 }
3387
Florin Coras134a9962018-08-28 11:32:04 -07003388 vec_validate (wrk->mq_events, pool_elts (wrk->mq_evt_conns));
Florin Corasb13ba132021-01-27 16:05:24 -08003389 if (!n_evts)
Florin Coras99368312018-08-02 10:45:44 -07003390 {
Florin Corasccdb8b82021-04-28 13:01:06 -07003391 if (timeout_ms > 0)
3392 end = clib_time_now (&wrk->clib_time) + (timeout_ms / 1e3);
Florin Coras99368312018-08-02 10:45:44 -07003393 }
3394
Florin Corasb13ba132021-01-27 16:05:24 -08003395 do
3396 {
3397 n_mq_evts = epoll_wait (wrk->mqs_epfd, wrk->mq_events,
Florin Corasccdb8b82021-04-28 13:01:06 -07003398 vec_len (wrk->mq_events), timeout_ms);
Florin Corasb13ba132021-01-27 16:05:24 -08003399 if (n_mq_evts < 0)
3400 {
3401 VDBG (0, "epoll_wait error %u", errno);
3402 return n_evts;
3403 }
3404
3405 for (i = 0; i < n_mq_evts; i++)
3406 {
Filip Tehlar8ccc6b32022-02-02 17:38:20 +00003407 if (PREDICT_FALSE (wrk->mq_events[i].data.u32 == ~0))
3408 {
3409 /* api socket was closed */
3410 vcl_api_handle_disconnect (wrk);
3411 continue;
3412 }
3413
Florin Corasb13ba132021-01-27 16:05:24 -08003414 mqc = vcl_mq_evt_conn_get (wrk, wrk->mq_events[i].data.u32);
3415 n_read = read (mqc->mq_fd, &buf, sizeof (buf));
3416 vcl_epoll_wait_handle_mq (wrk, mqc->mq, events, maxevents, 0,
3417 &n_evts);
3418 }
3419
Florin Corasccdb8b82021-04-28 13:01:06 -07003420 if (n_evts || !timeout_ms)
Florin Corasb13ba132021-01-27 16:05:24 -08003421 return n_evts;
Florin Corasb13ba132021-01-27 16:05:24 -08003422 }
Florin Corasccdb8b82021-04-28 13:01:06 -07003423 while (end == -1 || clib_time_now (&wrk->clib_time) < end);
Florin Corasb13ba132021-01-27 16:05:24 -08003424
3425 return 0;
Florin Coras99368312018-08-02 10:45:44 -07003426}
3427
Florin Corasfe286f72021-06-04 10:07:55 -07003428static void
Florin Corasfe286f72021-06-04 10:07:55 -07003429vcl_epoll_wait_handle_lt (vcl_worker_t *wrk, struct epoll_event *events,
3430 int maxevents, u32 *n_evts)
3431{
Florin Corasdb3fbfc2023-05-22 09:19:57 -07003432 u32 add_event = 0, evt_flags = 0, next, *to_remove = 0, *si;
Florin Corasfe286f72021-06-04 10:07:55 -07003433 vcl_session_t *s;
3434 u64 evt_data;
Florin Corasfa3884f2021-06-22 20:04:46 -07003435 int rv;
Florin Corasfe286f72021-06-04 10:07:55 -07003436
Florin Corasfa3884f2021-06-22 20:04:46 -07003437 ASSERT (wrk->ep_lt_current != VCL_INVALID_SESSION_INDEX);
Florin Corasfe286f72021-06-04 10:07:55 -07003438 if (*n_evts >= maxevents)
Florin Corasfa3884f2021-06-22 20:04:46 -07003439 return;
Florin Corasfe286f72021-06-04 10:07:55 -07003440
Florin Corasfa3884f2021-06-22 20:04:46 -07003441 next = wrk->ep_lt_current;
3442 do
Florin Corasfe286f72021-06-04 10:07:55 -07003443 {
Florin Corasfa3884f2021-06-22 20:04:46 -07003444 s = vcl_session_get (wrk, next);
3445 next = s->vep.lt_next;
3446
Florin Coras30ecfa82023-06-14 11:15:36 -07003447 if (s->vep.ev.events == 0)
3448 {
3449 vec_add1 (to_remove, s->session_index);
3450 continue;
3451 }
Florin Corasfa3884f2021-06-22 20:04:46 -07003452 if ((s->vep.ev.events & EPOLLIN) && (rv = vcl_session_read_ready (s)))
Florin Corasfe286f72021-06-04 10:07:55 -07003453 {
3454 add_event = 1;
Florin Corasdb3fbfc2023-05-22 09:19:57 -07003455 evt_flags |= rv > 0 ? EPOLLIN : EPOLLHUP | EPOLLRDHUP;
Florin Corasfe286f72021-06-04 10:07:55 -07003456 evt_data = s->vep.ev.data.u64;
3457 }
Florin Corasfa3884f2021-06-22 20:04:46 -07003458 if ((s->vep.ev.events & EPOLLOUT) && (rv = vcl_session_write_ready (s)))
Florin Corasfe286f72021-06-04 10:07:55 -07003459 {
3460 add_event = 1;
Florin Corasdb3fbfc2023-05-22 09:19:57 -07003461 evt_flags |= rv > 0 ? EPOLLOUT : EPOLLHUP | EPOLLRDHUP;
Florin Corasfa3884f2021-06-22 20:04:46 -07003462 evt_data = s->vep.ev.data.u64;
3463 }
3464 if (!add_event && s->session_state > VCL_STATE_READY)
3465 {
3466 add_event = 1;
Florin Corasdb3fbfc2023-05-22 09:19:57 -07003467 evt_flags |= EPOLLHUP | EPOLLRDHUP;
Florin Corasfe286f72021-06-04 10:07:55 -07003468 evt_data = s->vep.ev.data.u64;
3469 }
3470 if (add_event)
3471 {
Florin Corasdb3fbfc2023-05-22 09:19:57 -07003472 events[*n_evts].events = evt_flags;
Florin Corasfe286f72021-06-04 10:07:55 -07003473 events[*n_evts].data.u64 = evt_data;
Florin Corasfa3884f2021-06-22 20:04:46 -07003474 if (EPOLLONESHOT & s->vep.ev.events)
Florin Coras30ecfa82023-06-14 11:15:36 -07003475 s->vep.ev.events = EPOLLHUP | EPOLLERR;
3476 if (evt_flags & EPOLLHUP)
Florin Corasfa3884f2021-06-22 20:04:46 -07003477 s->vep.ev.events = 0;
Florin Coras5cca6692023-06-20 08:47:37 -07003478 *n_evts += 1;
3479 add_event = 0;
3480 evt_flags = 0;
Florin Corasfe286f72021-06-04 10:07:55 -07003481 if (*n_evts == maxevents)
3482 {
Florin Corasfa3884f2021-06-22 20:04:46 -07003483 wrk->ep_lt_current = next;
Florin Corasfe286f72021-06-04 10:07:55 -07003484 break;
3485 }
3486 }
Florin Corasfa3884f2021-06-22 20:04:46 -07003487 else
3488 {
Florin Coras7ff72742023-05-16 13:05:28 -07003489 vec_add1 (to_remove, s->session_index);
Florin Corasfa3884f2021-06-22 20:04:46 -07003490 }
Florin Corasfe286f72021-06-04 10:07:55 -07003491 }
Florin Corasfa3884f2021-06-22 20:04:46 -07003492 while (next != wrk->ep_lt_current);
Florin Coras7ff72742023-05-16 13:05:28 -07003493
3494 vec_foreach (si, to_remove)
3495 {
3496 s = vcl_session_get (wrk, *si);
3497 vcl_epoll_lt_del (wrk, s);
3498 }
3499 vec_free (to_remove);
Florin Corasfe286f72021-06-04 10:07:55 -07003500}
3501
Dave Wallacef7f809c2017-10-03 01:48:42 -04003502int
Florin Coras134a9962018-08-28 11:32:04 -07003503vppcom_epoll_wait (uint32_t vep_handle, struct epoll_event *events,
Dave Wallacef7f809c2017-10-03 01:48:42 -04003504 int maxevents, double wait_for_time)
3505{
Florin Coras134a9962018-08-28 11:32:04 -07003506 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras7e12d942018-06-27 14:32:43 -07003507 vcl_session_t *vep_session;
Florin Corasfa3884f2021-06-22 20:04:46 -07003508 u32 n_evts = 0;
Florin Coras86f04502018-09-12 16:08:01 -07003509 int i;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003510
3511 if (PREDICT_FALSE (maxevents <= 0))
3512 {
Florin Coras5e062572019-03-14 19:07:51 -07003513 VDBG (0, "ERROR: Invalid maxevents (%d)!", maxevents);
Dave Wallacef7f809c2017-10-03 01:48:42 -04003514 return VPPCOM_EINVAL;
3515 }
Dave Wallacef7f809c2017-10-03 01:48:42 -04003516
Florin Coras134a9962018-08-28 11:32:04 -07003517 vep_session = vcl_session_get_w_handle (wrk, vep_handle);
Florin Coras14598772018-09-04 19:47:52 -07003518 if (!vep_session)
3519 return VPPCOM_EBADFD;
3520
Florin Coras6c3b2182020-10-19 18:36:48 -07003521 if (PREDICT_FALSE (!(vep_session->flags & VCL_SESSION_F_IS_VEP)))
Dave Wallacef7f809c2017-10-03 01:48:42 -04003522 {
Florin Coras5e062572019-03-14 19:07:51 -07003523 VDBG (0, "ERROR: vep_idx (%u) is not a vep!", vep_handle);
Florin Coras54693d22018-07-17 10:46:29 -07003524 return VPPCOM_EINVAL;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003525 }
Florin Coras54693d22018-07-17 10:46:29 -07003526
Florin Coras86f04502018-09-12 16:08:01 -07003527 if (vec_len (wrk->unhandled_evts_vector))
3528 {
3529 for (i = 0; i < vec_len (wrk->unhandled_evts_vector); i++)
3530 {
3531 vcl_epoll_wait_handle_mq_event (wrk, &wrk->unhandled_evts_vector[i],
3532 events, &n_evts);
3533 if (n_evts == maxevents)
3534 {
Florin Corase003a1b2019-06-05 10:47:16 -07003535 vec_delete (wrk->unhandled_evts_vector, i + 1, 0);
3536 return n_evts;
Florin Coras86f04502018-09-12 16:08:01 -07003537 }
3538 }
Florin Corase003a1b2019-06-05 10:47:16 -07003539 vec_reset_length (wrk->unhandled_evts_vector);
Florin Coras86f04502018-09-12 16:08:01 -07003540 }
liuyacancba87102021-09-10 15:14:05 +08003541
3542 if (PREDICT_FALSE (wrk->ep_lt_current != VCL_INVALID_SESSION_INDEX))
3543 vcl_epoll_wait_handle_lt (wrk, events, maxevents, &n_evts);
3544
wanghanlin8919fec2021-03-18 20:00:41 +08003545 /* Request to only drain unhandled */
3546 if ((int) wait_for_time == -2)
3547 return n_evts;
Florin Coras86f04502018-09-12 16:08:01 -07003548
Florin Coras86f04502018-09-12 16:08:01 -07003549
Florin Corasfe286f72021-06-04 10:07:55 -07003550 if (vcm->cfg.use_mq_eventfd)
3551 n_evts = vppcom_epoll_wait_eventfd (wrk, events, maxevents, n_evts,
3552 wait_for_time);
3553 else
3554 n_evts = vppcom_epoll_wait_condvar (wrk, events, maxevents, n_evts,
3555 wait_for_time);
3556
Florin Corasfe286f72021-06-04 10:07:55 -07003557 return n_evts;
Dave Wallacef7f809c2017-10-03 01:48:42 -04003558}
3559
Dave Wallace35830af2017-10-09 01:43:42 -04003560int
Florin Coras134a9962018-08-28 11:32:04 -07003561vppcom_session_attr (uint32_t session_handle, uint32_t op,
Dave Wallace35830af2017-10-09 01:43:42 -04003562 void *buffer, uint32_t * buflen)
3563{
Florin Coras134a9962018-08-28 11:32:04 -07003564 vcl_worker_t *wrk = vcl_worker_get_current ();
liuyacan55c952e2021-06-13 14:54:55 +08003565 u32 *flags = buffer;
Steven2199aab2017-10-15 20:18:47 -07003566 vppcom_endpt_t *ep = buffer;
Florin Coras04ae8272021-04-12 19:55:37 -07003567 transport_endpt_attr_t tea;
Florin Corasa5a9efd2021-01-05 17:03:29 -08003568 vcl_session_t *session;
3569 int rv = VPPCOM_OK;
Dave Wallace35830af2017-10-09 01:43:42 -04003570
Florin Coras134a9962018-08-28 11:32:04 -07003571 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras070453d2018-08-24 17:04:27 -07003572 if (!session)
3573 return VPPCOM_EBADFD;
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003574
Dave Wallace35830af2017-10-09 01:43:42 -04003575 switch (op)
3576 {
3577 case VPPCOM_ATTR_GET_NREAD:
Florin Coras0ef8ef22019-01-18 08:37:13 -08003578 rv = vcl_session_read_ready (session);
Florin Coras5e062572019-03-14 19:07:51 -07003579 VDBG (2, "VPPCOM_ATTR_GET_NREAD: sh %u, nread = %d", session_handle,
3580 rv);
Dave Wallace35830af2017-10-09 01:43:42 -04003581 break;
3582
Dave Wallace227867f2017-11-13 21:21:53 -05003583 case VPPCOM_ATTR_GET_NWRITE:
Florin Coras0ef8ef22019-01-18 08:37:13 -08003584 rv = vcl_session_write_ready (session);
Florin Coras5e062572019-03-14 19:07:51 -07003585 VDBG (2, "VPPCOM_ATTR_GET_NWRITE: sh %u, nwrite = %d", session_handle,
3586 rv);
Dave Wallace35830af2017-10-09 01:43:42 -04003587 break;
Florin Coras8ae63db2024-03-18 12:25:38 -07003588 case VPPCOM_ATTR_GET_NWRITEQ:
3589 if (PREDICT_FALSE (!buffer || !buflen || *buflen != sizeof (int)))
3590 {
3591 rv = VPPCOM_EINVAL;
3592 break;
3593 }
3594 if (!session->tx_fifo || session->session_state == VCL_STATE_DETACHED)
3595 {
3596 rv = VPPCOM_EINVAL;
3597 break;
3598 }
3599 *(int *) buffer = svm_fifo_max_dequeue (session->tx_fifo);
3600 break;
Dave Wallace35830af2017-10-09 01:43:42 -04003601 case VPPCOM_ATTR_GET_FLAGS:
Dave Wallace048b1d62018-01-03 22:24:41 -05003602 if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*flags))))
Dave Wallace35830af2017-10-09 01:43:42 -04003603 {
Simon Zhang53ec9672020-08-07 05:20:47 +08003604 *flags =
3605 O_RDWR |
Florin Corasac422d62020-10-19 20:51:36 -07003606 (vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK) ?
Simon Zhang53ec9672020-08-07 05:20:47 +08003607 O_NONBLOCK : 0);
Dave Wallace35830af2017-10-09 01:43:42 -04003608 *buflen = sizeof (*flags);
Florin Coras5e062572019-03-14 19:07:51 -07003609 VDBG (2, "VPPCOM_ATTR_GET_FLAGS: sh %u, flags = 0x%08x, "
3610 "is_nonblocking = %u", session_handle, *flags,
Florin Corasac422d62020-10-19 20:51:36 -07003611 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace35830af2017-10-09 01:43:42 -04003612 }
3613 else
3614 rv = VPPCOM_EINVAL;
3615 break;
3616
3617 case VPPCOM_ATTR_SET_FLAGS:
Dave Wallace048b1d62018-01-03 22:24:41 -05003618 if (PREDICT_TRUE (buffer && buflen && (*buflen == sizeof (*flags))))
Dave Wallace35830af2017-10-09 01:43:42 -04003619 {
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003620 if (*flags & O_NONBLOCK)
Florin Corasac422d62020-10-19 20:51:36 -07003621 vcl_session_set_attr (session, VCL_SESS_ATTR_NONBLOCK);
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003622 else
Florin Corasac422d62020-10-19 20:51:36 -07003623 vcl_session_clear_attr (session, VCL_SESS_ATTR_NONBLOCK);
Keith Burns (alagalah)9b793772018-02-16 08:20:56 -08003624
Florin Coras5e062572019-03-14 19:07:51 -07003625 VDBG (2, "VPPCOM_ATTR_SET_FLAGS: sh %u, flags = 0x%08x,"
3626 " is_nonblocking = %u", session_handle, *flags,
Florin Corasac422d62020-10-19 20:51:36 -07003627 vcl_session_has_attr (session, VCL_SESS_ATTR_NONBLOCK));
Dave Wallace35830af2017-10-09 01:43:42 -04003628 }
3629 else
3630 rv = VPPCOM_EINVAL;
3631 break;
3632
3633 case VPPCOM_ATTR_GET_PEER_ADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003634 if (PREDICT_TRUE (buffer && buflen &&
3635 (*buflen >= sizeof (*ep)) && ep->ip))
Dave Wallace35830af2017-10-09 01:43:42 -04003636 {
Florin Coras7e12d942018-06-27 14:32:43 -07003637 ep->is_ip4 = session->transport.is_ip4;
3638 ep->port = session->transport.rmt_port;
3639 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05003640 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
3641 sizeof (ip4_address_t));
Steven2199aab2017-10-15 20:18:47 -07003642 else
Dave Barach178cf492018-11-13 16:34:13 -05003643 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
3644 sizeof (ip6_address_t));
Steven2199aab2017-10-15 20:18:47 -07003645 *buflen = sizeof (*ep);
Florin Coras60687192021-11-29 21:00:47 -08003646 VDBG (1,
3647 "VPPCOM_ATTR_GET_PEER_ADDR: sh %u, is_ip4 = %u, "
3648 "addr = %U, port %u",
3649 session_handle, ep->is_ip4, vcl_format_ip46_address,
3650 &session->transport.rmt_ip,
Florin Coras0d427d82018-06-27 03:24:07 -07003651 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3652 clib_net_to_host_u16 (ep->port));
Dave Wallace35830af2017-10-09 01:43:42 -04003653 }
3654 else
3655 rv = VPPCOM_EINVAL;
3656 break;
3657
3658 case VPPCOM_ATTR_GET_LCL_ADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003659 if (PREDICT_TRUE (buffer && buflen &&
3660 (*buflen >= sizeof (*ep)) && ep->ip))
Dave Wallace35830af2017-10-09 01:43:42 -04003661 {
Florin Coras7e12d942018-06-27 14:32:43 -07003662 ep->is_ip4 = session->transport.is_ip4;
3663 ep->port = session->transport.lcl_port;
3664 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05003665 clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip4,
3666 sizeof (ip4_address_t));
Steven2199aab2017-10-15 20:18:47 -07003667 else
Dave Barach178cf492018-11-13 16:34:13 -05003668 clib_memcpy_fast (ep->ip, &session->transport.lcl_ip.ip6,
3669 sizeof (ip6_address_t));
Steven2199aab2017-10-15 20:18:47 -07003670 *buflen = sizeof (*ep);
Florin Coras60687192021-11-29 21:00:47 -08003671 VDBG (1,
3672 "VPPCOM_ATTR_GET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3673 " port %d",
3674 session_handle, ep->is_ip4, vcl_format_ip46_address,
Florin Coras7e12d942018-06-27 14:32:43 -07003675 &session->transport.lcl_ip,
Florin Coras0d427d82018-06-27 03:24:07 -07003676 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3677 clib_net_to_host_u16 (ep->port));
Dave Wallace35830af2017-10-09 01:43:42 -04003678 }
3679 else
3680 rv = VPPCOM_EINVAL;
3681 break;
Stevenb5a11602017-10-11 09:59:30 -07003682
qinyangaf9b7152023-06-27 01:11:53 -07003683 case VPPCOM_ATTR_GET_ORIGINAL_DST:
3684 if (!session->transport.is_ip4)
3685 {
3686 /* now original dst only support ipv4*/
3687 rv = VPPCOM_EAFNOSUPPORT;
3688 break;
3689 }
3690 if (PREDICT_TRUE (buffer && buflen && (*buflen >= sizeof (*ep)) &&
3691 ep->ip))
3692 {
3693 ep->is_ip4 = session->transport.is_ip4;
3694 ep->port = session->original_dst_port;
3695 clib_memcpy_fast (ep->ip, &session->original_dst_ip4,
3696 sizeof (ip4_address_t));
3697 *buflen = sizeof (*ep);
3698 VDBG (1,
3699 "VPPCOM_ATTR_GET_ORIGINAL_DST: sh %u, is_ip4 = %u, addr = %U"
3700 " port %d",
3701 session_handle, ep->is_ip4, vcl_format_ip4_address,
3702 (ip4_address_t *) (&session->original_dst_ip4),
3703 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3704 clib_net_to_host_u16 (ep->port));
3705 }
3706 else
3707 rv = VPPCOM_EINVAL;
3708 break;
3709
Florin Corasef7cbf62019-10-17 09:56:27 -07003710 case VPPCOM_ATTR_SET_LCL_ADDR:
3711 if (PREDICT_TRUE (buffer && buflen &&
3712 (*buflen >= sizeof (*ep)) && ep->ip))
3713 {
3714 session->transport.is_ip4 = ep->is_ip4;
3715 session->transport.lcl_port = ep->port;
3716 vcl_ip_copy_from_ep (&session->transport.lcl_ip, ep);
3717 *buflen = sizeof (*ep);
Florin Coras60687192021-11-29 21:00:47 -08003718 VDBG (1,
3719 "VPPCOM_ATTR_SET_LCL_ADDR: sh %u, is_ip4 = %u, addr = %U"
3720 " port %d",
3721 session_handle, ep->is_ip4, vcl_format_ip46_address,
Florin Corasef7cbf62019-10-17 09:56:27 -07003722 &session->transport.lcl_ip,
3723 ep->is_ip4 ? IP46_TYPE_IP4 : IP46_TYPE_IP6,
3724 clib_net_to_host_u16 (ep->port));
3725 }
3726 else
3727 rv = VPPCOM_EINVAL;
3728 break;
3729
Dave Wallace048b1d62018-01-03 22:24:41 -05003730 case VPPCOM_ATTR_GET_LIBC_EPFD:
3731 rv = session->libc_epfd;
Florin Coras5e062572019-03-14 19:07:51 -07003732 VDBG (2, "VPPCOM_ATTR_GET_LIBC_EPFD: libc_epfd %d", rv);
Dave Wallace048b1d62018-01-03 22:24:41 -05003733 break;
3734
3735 case VPPCOM_ATTR_SET_LIBC_EPFD:
3736 if (PREDICT_TRUE (buffer && buflen &&
3737 (*buflen == sizeof (session->libc_epfd))))
3738 {
3739 session->libc_epfd = *(int *) buffer;
3740 *buflen = sizeof (session->libc_epfd);
3741
Florin Coras5e062572019-03-14 19:07:51 -07003742 VDBG (2, "VPPCOM_ATTR_SET_LIBC_EPFD: libc_epfd %d, buflen %d",
3743 session->libc_epfd, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003744 }
3745 else
3746 rv = VPPCOM_EINVAL;
3747 break;
3748
3749 case VPPCOM_ATTR_GET_PROTOCOL:
3750 if (buffer && buflen && (*buflen >= sizeof (int)))
3751 {
Florin Coras7e12d942018-06-27 14:32:43 -07003752 *(int *) buffer = session->session_type;
Dave Wallace048b1d62018-01-03 22:24:41 -05003753 *buflen = sizeof (int);
3754
Florin Coras5e062572019-03-14 19:07:51 -07003755 VDBG (2, "VPPCOM_ATTR_GET_PROTOCOL: %d (%s), buflen %d",
3756 *(int *) buffer, *(int *) buffer ? "UDP" : "TCP", *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003757 }
3758 else
3759 rv = VPPCOM_EINVAL;
3760 break;
3761
3762 case VPPCOM_ATTR_GET_LISTEN:
3763 if (buffer && buflen && (*buflen >= sizeof (int)))
3764 {
Florin Corasac422d62020-10-19 20:51:36 -07003765 *(int *) buffer = vcl_session_has_attr (session,
3766 VCL_SESS_ATTR_LISTEN);
Dave Wallace048b1d62018-01-03 22:24:41 -05003767 *buflen = sizeof (int);
3768
Florin Coras5e062572019-03-14 19:07:51 -07003769 VDBG (2, "VPPCOM_ATTR_GET_LISTEN: %d, buflen %d", *(int *) buffer,
3770 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003771 }
3772 else
3773 rv = VPPCOM_EINVAL;
3774 break;
3775
3776 case VPPCOM_ATTR_GET_ERROR:
3777 if (buffer && buflen && (*buflen >= sizeof (int)))
3778 {
3779 *(int *) buffer = 0;
3780 *buflen = sizeof (int);
3781
Florin Coras5e062572019-03-14 19:07:51 -07003782 VDBG (2, "VPPCOM_ATTR_GET_ERROR: %d, buflen %d, #VPP-TBD#",
3783 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003784 }
3785 else
3786 rv = VPPCOM_EINVAL;
3787 break;
3788
3789 case VPPCOM_ATTR_GET_TX_FIFO_LEN:
3790 if (buffer && buflen && (*buflen >= sizeof (u32)))
3791 {
Dave Wallace048b1d62018-01-03 22:24:41 -05003792
3793 /* VPP-TBD */
3794 *(size_t *) buffer = (session->sndbuf_size ? session->sndbuf_size :
Florin Corasf22f4e52019-12-19 16:10:58 -08003795 session->tx_fifo ?
3796 svm_fifo_size (session->tx_fifo) :
Dave Wallace048b1d62018-01-03 22:24:41 -05003797 vcm->cfg.tx_fifo_size);
3798 *buflen = sizeof (u32);
3799
Florin Coras5e062572019-03-14 19:07:51 -07003800 VDBG (2, "VPPCOM_ATTR_GET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3801 " #VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer,
3802 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003803 }
3804 else
3805 rv = VPPCOM_EINVAL;
3806 break;
3807
Filip Tehlar2f09bfc2021-11-15 10:26:56 +00003808 case VPPCOM_ATTR_SET_DSCP:
3809 if (buffer && buflen && (*buflen >= sizeof (u8)))
3810 {
3811 session->dscp = *(u8 *) buffer;
3812
3813 VDBG (2, "VPPCOM_ATTR_SET_DSCP: %u (0x%x), buflen %d,",
3814 *(u8 *) buffer, *(u8 *) buffer, *buflen);
3815 }
3816 else
3817 rv = VPPCOM_EINVAL;
3818 break;
3819
Dave Wallace048b1d62018-01-03 22:24:41 -05003820 case VPPCOM_ATTR_SET_TX_FIFO_LEN:
3821 if (buffer && buflen && (*buflen == sizeof (u32)))
3822 {
3823 /* VPP-TBD */
3824 session->sndbuf_size = *(u32 *) buffer;
Florin Coras5e062572019-03-14 19:07:51 -07003825 VDBG (2, "VPPCOM_ATTR_SET_TX_FIFO_LEN: %u (0x%x), buflen %d,"
3826 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3827 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003828 }
3829 else
3830 rv = VPPCOM_EINVAL;
3831 break;
3832
3833 case VPPCOM_ATTR_GET_RX_FIFO_LEN:
3834 if (buffer && buflen && (*buflen >= sizeof (u32)))
3835 {
Dave Wallace048b1d62018-01-03 22:24:41 -05003836
3837 /* VPP-TBD */
3838 *(size_t *) buffer = (session->rcvbuf_size ? session->rcvbuf_size :
Florin Corasf22f4e52019-12-19 16:10:58 -08003839 session->rx_fifo ?
3840 svm_fifo_size (session->rx_fifo) :
Dave Wallace048b1d62018-01-03 22:24:41 -05003841 vcm->cfg.rx_fifo_size);
3842 *buflen = sizeof (u32);
3843
Florin Coras5e062572019-03-14 19:07:51 -07003844 VDBG (2, "VPPCOM_ATTR_GET_RX_FIFO_LEN: %u (0x%x), buflen %d, "
3845 "#VPP-TBD#", *(size_t *) buffer, *(size_t *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003846 }
3847 else
3848 rv = VPPCOM_EINVAL;
3849 break;
3850
3851 case VPPCOM_ATTR_SET_RX_FIFO_LEN:
3852 if (buffer && buflen && (*buflen == sizeof (u32)))
3853 {
3854 /* VPP-TBD */
3855 session->rcvbuf_size = *(u32 *) buffer;
Florin Coras5e062572019-03-14 19:07:51 -07003856 VDBG (2, "VPPCOM_ATTR_SET_RX_FIFO_LEN: %u (0x%x), buflen %d,"
3857 " #VPP-TBD#", session->sndbuf_size, session->sndbuf_size,
3858 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003859 }
3860 else
3861 rv = VPPCOM_EINVAL;
3862 break;
3863
3864 case VPPCOM_ATTR_GET_REUSEADDR:
3865 if (buffer && buflen && (*buflen >= sizeof (int)))
3866 {
3867 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07003868 *(int *) buffer = vcl_session_has_attr (session,
3869 VCL_SESS_ATTR_REUSEADDR);
Dave Wallace048b1d62018-01-03 22:24:41 -05003870 *buflen = sizeof (int);
3871
Florin Coras5e062572019-03-14 19:07:51 -07003872 VDBG (2, "VPPCOM_ATTR_GET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
3873 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003874 }
3875 else
3876 rv = VPPCOM_EINVAL;
3877 break;
3878
Stevenb5a11602017-10-11 09:59:30 -07003879 case VPPCOM_ATTR_SET_REUSEADDR:
Dave Wallace048b1d62018-01-03 22:24:41 -05003880 if (buffer && buflen && (*buflen == sizeof (int)) &&
Florin Corasac422d62020-10-19 20:51:36 -07003881 !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
Dave Wallace048b1d62018-01-03 22:24:41 -05003882 {
3883 /* VPP-TBD */
3884 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07003885 vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEADDR);
Dave Wallace048b1d62018-01-03 22:24:41 -05003886 else
Florin Corasac422d62020-10-19 20:51:36 -07003887 vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEADDR);
Dave Wallace048b1d62018-01-03 22:24:41 -05003888
Florin Coras5e062572019-03-14 19:07:51 -07003889 VDBG (2, "VPPCOM_ATTR_SET_REUSEADDR: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07003890 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEADDR),
Florin Coras5e062572019-03-14 19:07:51 -07003891 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003892 }
3893 else
3894 rv = VPPCOM_EINVAL;
3895 break;
3896
3897 case VPPCOM_ATTR_GET_REUSEPORT:
3898 if (buffer && buflen && (*buflen >= sizeof (int)))
3899 {
3900 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07003901 *(int *) buffer = vcl_session_has_attr (session,
3902 VCL_SESS_ATTR_REUSEPORT);
Dave Wallace048b1d62018-01-03 22:24:41 -05003903 *buflen = sizeof (int);
3904
Florin Coras5e062572019-03-14 19:07:51 -07003905 VDBG (2, "VPPCOM_ATTR_GET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
3906 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003907 }
3908 else
3909 rv = VPPCOM_EINVAL;
3910 break;
3911
3912 case VPPCOM_ATTR_SET_REUSEPORT:
3913 if (buffer && buflen && (*buflen == sizeof (int)) &&
Florin Corasac422d62020-10-19 20:51:36 -07003914 !vcl_session_has_attr (session, VCL_SESS_ATTR_LISTEN))
Dave Wallace048b1d62018-01-03 22:24:41 -05003915 {
3916 /* VPP-TBD */
3917 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07003918 vcl_session_set_attr (session, VCL_SESS_ATTR_REUSEPORT);
Dave Wallace048b1d62018-01-03 22:24:41 -05003919 else
Florin Corasac422d62020-10-19 20:51:36 -07003920 vcl_session_clear_attr (session, VCL_SESS_ATTR_REUSEPORT);
Dave Wallace048b1d62018-01-03 22:24:41 -05003921
Florin Coras5e062572019-03-14 19:07:51 -07003922 VDBG (2, "VPPCOM_ATTR_SET_REUSEPORT: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07003923 vcl_session_has_attr (session, VCL_SESS_ATTR_REUSEPORT),
Florin Coras5e062572019-03-14 19:07:51 -07003924 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003925 }
3926 else
3927 rv = VPPCOM_EINVAL;
3928 break;
3929
3930 case VPPCOM_ATTR_GET_BROADCAST:
3931 if (buffer && buflen && (*buflen >= sizeof (int)))
3932 {
3933 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07003934 *(int *) buffer = vcl_session_has_attr (session,
3935 VCL_SESS_ATTR_BROADCAST);
Dave Wallace048b1d62018-01-03 22:24:41 -05003936 *buflen = sizeof (int);
3937
Florin Coras5e062572019-03-14 19:07:51 -07003938 VDBG (2, "VPPCOM_ATTR_GET_BROADCAST: %d, buflen %d, #VPP-TBD#",
3939 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003940 }
3941 else
3942 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07003943 break;
3944
3945 case VPPCOM_ATTR_SET_BROADCAST:
Dave Wallace048b1d62018-01-03 22:24:41 -05003946 if (buffer && buflen && (*buflen == sizeof (int)))
3947 {
3948 /* VPP-TBD */
3949 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07003950 vcl_session_set_attr (session, VCL_SESS_ATTR_BROADCAST);
Dave Wallace048b1d62018-01-03 22:24:41 -05003951 else
Florin Corasac422d62020-10-19 20:51:36 -07003952 vcl_session_clear_attr (session, VCL_SESS_ATTR_BROADCAST);
Dave Wallace048b1d62018-01-03 22:24:41 -05003953
Florin Coras5e062572019-03-14 19:07:51 -07003954 VDBG (2, "VPPCOM_ATTR_SET_BROADCAST: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07003955 vcl_session_has_attr (session, VCL_SESS_ATTR_BROADCAST),
Florin Coras5e062572019-03-14 19:07:51 -07003956 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003957 }
3958 else
3959 rv = VPPCOM_EINVAL;
3960 break;
3961
3962 case VPPCOM_ATTR_GET_V6ONLY:
3963 if (buffer && buflen && (*buflen >= sizeof (int)))
3964 {
3965 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07003966 *(int *) buffer = vcl_session_has_attr (session,
3967 VCL_SESS_ATTR_V6ONLY);
Dave Wallace048b1d62018-01-03 22:24:41 -05003968 *buflen = sizeof (int);
3969
Florin Coras5e062572019-03-14 19:07:51 -07003970 VDBG (2, "VPPCOM_ATTR_GET_V6ONLY: %d, buflen %d, #VPP-TBD#",
3971 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003972 }
3973 else
3974 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07003975 break;
3976
3977 case VPPCOM_ATTR_SET_V6ONLY:
Dave Wallace048b1d62018-01-03 22:24:41 -05003978 if (buffer && buflen && (*buflen == sizeof (int)))
3979 {
3980 /* VPP-TBD */
3981 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07003982 vcl_session_set_attr (session, VCL_SESS_ATTR_V6ONLY);
Dave Wallace048b1d62018-01-03 22:24:41 -05003983 else
Florin Corasac422d62020-10-19 20:51:36 -07003984 vcl_session_clear_attr (session, VCL_SESS_ATTR_V6ONLY);
Dave Wallace048b1d62018-01-03 22:24:41 -05003985
Florin Coras5e062572019-03-14 19:07:51 -07003986 VDBG (2, "VPPCOM_ATTR_SET_V6ONLY: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07003987 vcl_session_has_attr (session, VCL_SESS_ATTR_V6ONLY),
Florin Coras5e062572019-03-14 19:07:51 -07003988 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05003989 }
3990 else
3991 rv = VPPCOM_EINVAL;
3992 break;
3993
3994 case VPPCOM_ATTR_GET_KEEPALIVE:
3995 if (buffer && buflen && (*buflen >= sizeof (int)))
3996 {
3997 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07003998 *(int *) buffer = vcl_session_has_attr (session,
3999 VCL_SESS_ATTR_KEEPALIVE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004000 *buflen = sizeof (int);
4001
Florin Coras5e062572019-03-14 19:07:51 -07004002 VDBG (2, "VPPCOM_ATTR_GET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
4003 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004004 }
4005 else
4006 rv = VPPCOM_EINVAL;
Stevenb5a11602017-10-11 09:59:30 -07004007 break;
Stevenbccd3392017-10-12 20:42:21 -07004008
4009 case VPPCOM_ATTR_SET_KEEPALIVE:
Dave Wallace048b1d62018-01-03 22:24:41 -05004010 if (buffer && buflen && (*buflen == sizeof (int)))
4011 {
4012 /* VPP-TBD */
4013 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07004014 vcl_session_set_attr (session, VCL_SESS_ATTR_KEEPALIVE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004015 else
Florin Corasac422d62020-10-19 20:51:36 -07004016 vcl_session_clear_attr (session, VCL_SESS_ATTR_KEEPALIVE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004017
Florin Coras5e062572019-03-14 19:07:51 -07004018 VDBG (2, "VPPCOM_ATTR_SET_KEEPALIVE: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07004019 vcl_session_has_attr (session, VCL_SESS_ATTR_KEEPALIVE),
Florin Coras5e062572019-03-14 19:07:51 -07004020 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004021 }
4022 else
4023 rv = VPPCOM_EINVAL;
4024 break;
4025
4026 case VPPCOM_ATTR_GET_TCP_NODELAY:
4027 if (buffer && buflen && (*buflen >= sizeof (int)))
4028 {
4029 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07004030 *(int *) buffer = vcl_session_has_attr (session,
4031 VCL_SESS_ATTR_TCP_NODELAY);
Dave Wallace048b1d62018-01-03 22:24:41 -05004032 *buflen = sizeof (int);
4033
Florin Coras5e062572019-03-14 19:07:51 -07004034 VDBG (2, "VPPCOM_ATTR_GET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
4035 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004036 }
4037 else
4038 rv = VPPCOM_EINVAL;
4039 break;
4040
4041 case VPPCOM_ATTR_SET_TCP_NODELAY:
4042 if (buffer && buflen && (*buflen == sizeof (int)))
4043 {
4044 /* VPP-TBD */
4045 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07004046 vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
Dave Wallace048b1d62018-01-03 22:24:41 -05004047 else
Florin Corasac422d62020-10-19 20:51:36 -07004048 vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_NODELAY);
Dave Wallace048b1d62018-01-03 22:24:41 -05004049
Florin Coras5e062572019-03-14 19:07:51 -07004050 VDBG (2, "VPPCOM_ATTR_SET_TCP_NODELAY: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07004051 vcl_session_has_attr (session, VCL_SESS_ATTR_TCP_NODELAY),
Florin Coras5e062572019-03-14 19:07:51 -07004052 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004053 }
4054 else
4055 rv = VPPCOM_EINVAL;
4056 break;
4057
4058 case VPPCOM_ATTR_GET_TCP_KEEPIDLE:
4059 if (buffer && buflen && (*buflen >= sizeof (int)))
4060 {
4061 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07004062 *(int *) buffer = vcl_session_has_attr (session,
4063 VCL_SESS_ATTR_TCP_KEEPIDLE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004064 *buflen = sizeof (int);
4065
Florin Coras5e062572019-03-14 19:07:51 -07004066 VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
4067 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004068 }
4069 else
4070 rv = VPPCOM_EINVAL;
Stevenbccd3392017-10-12 20:42:21 -07004071 break;
4072
4073 case VPPCOM_ATTR_SET_TCP_KEEPIDLE:
Dave Wallace048b1d62018-01-03 22:24:41 -05004074 if (buffer && buflen && (*buflen == sizeof (int)))
4075 {
4076 /* VPP-TBD */
4077 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07004078 vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004079 else
Florin Corasac422d62020-10-19 20:51:36 -07004080 vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPIDLE);
Dave Wallace048b1d62018-01-03 22:24:41 -05004081
Florin Coras5e062572019-03-14 19:07:51 -07004082 VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPIDLE: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07004083 vcl_session_has_attr (session,
4084 VCL_SESS_ATTR_TCP_KEEPIDLE), *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004085 }
4086 else
4087 rv = VPPCOM_EINVAL;
4088 break;
4089
4090 case VPPCOM_ATTR_GET_TCP_KEEPINTVL:
4091 if (buffer && buflen && (*buflen >= sizeof (int)))
4092 {
4093 /* VPP-TBD */
Florin Corasac422d62020-10-19 20:51:36 -07004094 *(int *) buffer = vcl_session_has_attr (session,
4095 VCL_SESS_ATTR_TCP_KEEPINTVL);
Dave Wallace048b1d62018-01-03 22:24:41 -05004096 *buflen = sizeof (int);
4097
Florin Coras5e062572019-03-14 19:07:51 -07004098 VDBG (2, "VPPCOM_ATTR_GET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
4099 *(int *) buffer, *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004100 }
4101 else
4102 rv = VPPCOM_EINVAL;
Stevenbccd3392017-10-12 20:42:21 -07004103 break;
4104
4105 case VPPCOM_ATTR_SET_TCP_KEEPINTVL:
Dave Wallace048b1d62018-01-03 22:24:41 -05004106 if (buffer && buflen && (*buflen == sizeof (int)))
4107 {
4108 /* VPP-TBD */
4109 if (*(int *) buffer)
Florin Corasac422d62020-10-19 20:51:36 -07004110 vcl_session_set_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
Dave Wallace048b1d62018-01-03 22:24:41 -05004111 else
Florin Corasac422d62020-10-19 20:51:36 -07004112 vcl_session_clear_attr (session, VCL_SESS_ATTR_TCP_KEEPINTVL);
Dave Wallace048b1d62018-01-03 22:24:41 -05004113
Florin Coras5e062572019-03-14 19:07:51 -07004114 VDBG (2, "VPPCOM_ATTR_SET_TCP_KEEPINTVL: %d, buflen %d, #VPP-TBD#",
Florin Corasac422d62020-10-19 20:51:36 -07004115 vcl_session_has_attr (session,
4116 VCL_SESS_ATTR_TCP_KEEPINTVL), *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004117 }
4118 else
4119 rv = VPPCOM_EINVAL;
4120 break;
4121
4122 case VPPCOM_ATTR_GET_TCP_USER_MSS:
Florin Coras04ae8272021-04-12 19:55:37 -07004123 if (!(buffer && buflen && (*buflen >= sizeof (u32))))
Dave Wallace048b1d62018-01-03 22:24:41 -05004124 {
Florin Coras04ae8272021-04-12 19:55:37 -07004125 rv = VPPCOM_EINVAL;
4126 break;
Dave Wallace048b1d62018-01-03 22:24:41 -05004127 }
Florin Coras04ae8272021-04-12 19:55:37 -07004128
4129 tea.type = TRANSPORT_ENDPT_ATTR_MSS;
4130 tea.mss = *(u32 *) buffer;
4131 if (vcl_session_transport_attr (wrk, session, 1 /* is_get */, &tea))
4132 rv = VPPCOM_ENOPROTOOPT;
4133
4134 if (!rv)
4135 {
4136 *(u32 *) buffer = tea.mss;
4137 *buflen = sizeof (int);
4138 }
4139
4140 VDBG (2, "VPPCOM_ATTR_GET_TCP_USER_MSS: %d, buflen %d", *(int *) buffer,
4141 *buflen);
Dave Wallace048b1d62018-01-03 22:24:41 -05004142 break;
Dave Wallace048b1d62018-01-03 22:24:41 -05004143 case VPPCOM_ATTR_SET_TCP_USER_MSS:
Florin Coras04ae8272021-04-12 19:55:37 -07004144 if (!(buffer && buflen && (*buflen == sizeof (u32))))
Dave Wallace048b1d62018-01-03 22:24:41 -05004145 {
Florin Coras04ae8272021-04-12 19:55:37 -07004146 rv = VPPCOM_EINVAL;
4147 break;
Dave Wallace048b1d62018-01-03 22:24:41 -05004148 }
Florin Coras04ae8272021-04-12 19:55:37 -07004149
4150 tea.type = TRANSPORT_ENDPT_ATTR_MSS;
4151 tea.mss = *(u32 *) buffer;
4152 if (vcl_session_transport_attr (wrk, session, 0 /* is_get */, &tea))
4153 rv = VPPCOM_ENOPROTOOPT;
4154
4155 VDBG (2, "VPPCOM_ATTR_SET_TCP_USER_MSS: %u, buflen %d", tea.mss,
4156 *buflen);
Stevenbccd3392017-10-12 20:42:21 -07004157 break;
Dave Wallacee22aa742017-10-20 12:30:38 -04004158
Florin Coras1e966172020-05-16 18:18:14 +00004159 case VPPCOM_ATTR_SET_CONNECTED:
4160 session->flags |= VCL_SESSION_F_CONNECTED;
4161 break;
4162
Florin Corasa5a9efd2021-01-05 17:03:29 -08004163 case VPPCOM_ATTR_SET_CKPAIR:
4164 if (!(buffer && buflen && (*buflen == sizeof (int))) ||
4165 !vcl_session_has_crypto (session))
4166 {
4167 rv = VPPCOM_EINVAL;
4168 break;
4169 }
Florin Corasa54b62d2021-04-21 09:05:56 -07004170 if (!session->ext_config)
4171 {
Florin Coras87f63892021-05-01 16:01:40 -07004172 vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_CRYPTO,
4173 sizeof (transport_endpt_ext_cfg_t));
Florin Corasa54b62d2021-04-21 09:05:56 -07004174 }
4175 else if (session->ext_config->type != TRANSPORT_ENDPT_EXT_CFG_CRYPTO)
4176 {
4177 rv = VPPCOM_EINVAL;
4178 break;
4179 }
4180
4181 session->ext_config->crypto.ckpair_index = *(uint32_t *) buffer;
Florin Corasa5a9efd2021-01-05 17:03:29 -08004182 break;
4183
Florin Coras6a6555a2021-01-28 11:39:27 -08004184 case VPPCOM_ATTR_SET_VRF:
4185 if (!(buffer && buflen && (*buflen == sizeof (u32))))
4186 {
4187 rv = VPPCOM_EINVAL;
4188 break;
4189 }
4190 session->vrf = *(u32 *) buffer;
4191 break;
4192
4193 case VPPCOM_ATTR_GET_VRF:
4194 if (!(buffer && buflen && (*buflen >= sizeof (u32))))
4195 {
4196 rv = VPPCOM_EINVAL;
4197 break;
4198 }
4199 *(u32 *) buffer = session->vrf;
4200 *buflen = sizeof (u32);
4201 break;
4202
wanghanlin0674f852021-02-22 10:38:36 +08004203 case VPPCOM_ATTR_GET_DOMAIN:
Florin Corasd77325c2021-02-23 12:03:03 -08004204 if (!(buffer && buflen && (*buflen >= sizeof (int))))
wanghanlin0674f852021-02-22 10:38:36 +08004205 {
Florin Corasd77325c2021-02-23 12:03:03 -08004206 rv = VPPCOM_EINVAL;
4207 break;
wanghanlin0674f852021-02-22 10:38:36 +08004208 }
Florin Corasd77325c2021-02-23 12:03:03 -08004209
4210 if (session->transport.is_ip4)
4211 *(int *) buffer = AF_INET;
wanghanlin0674f852021-02-22 10:38:36 +08004212 else
Florin Corasd77325c2021-02-23 12:03:03 -08004213 *(int *) buffer = AF_INET6;
4214 *buflen = sizeof (int);
4215
wanghanlin0674f852021-02-22 10:38:36 +08004216 VDBG (2, "VPPCOM_ATTR_GET_DOMAIN: %d, buflen %u", *(int *) buffer,
4217 *buflen);
4218 break;
4219
Florin Coras87f63892021-05-01 16:01:40 -07004220 case VPPCOM_ATTR_SET_ENDPT_EXT_CFG:
4221 if (!(buffer && buflen && (*buflen > 0)))
4222 {
4223 rv = VPPCOM_EINVAL;
4224 break;
4225 }
4226 if (session->ext_config)
4227 {
4228 rv = VPPCOM_EINVAL;
4229 break;
4230 }
4231 vcl_session_alloc_ext_cfg (session, TRANSPORT_ENDPT_EXT_CFG_NONE,
4232 *buflen + sizeof (u32));
4233 clib_memcpy (session->ext_config->data, buffer, *buflen);
4234 session->ext_config->len = *buflen;
4235 break;
Florin Coraseff5f7a2023-02-07 17:36:17 -08004236 case VPPCOM_ATTR_SET_IP_PKTINFO:
4237 if (buffer && buflen && (*buflen == sizeof (int)) &&
4238 !vcl_session_has_attr (session, VCL_SESS_ATTR_IP_PKTINFO))
4239 {
4240 if (*(int *) buffer)
4241 vcl_session_set_attr (session, VCL_SESS_ATTR_IP_PKTINFO);
4242 else
4243 vcl_session_clear_attr (session, VCL_SESS_ATTR_IP_PKTINFO);
4244
4245 VDBG (2, "VCL_SESS_ATTR_IP_PKTINFO: %d, buflen %d",
4246 vcl_session_has_attr (session, VCL_SESS_ATTR_IP_PKTINFO),
4247 *buflen);
4248 }
4249 else
4250 rv = VPPCOM_EINVAL;
4251 break;
4252
4253 case VPPCOM_ATTR_GET_IP_PKTINFO:
4254 if (buffer && buflen && (*buflen >= sizeof (int)))
4255 {
4256 *(int *) buffer =
4257 vcl_session_has_attr (session, VCL_SESS_ATTR_IP_PKTINFO);
4258 *buflen = sizeof (int);
4259
4260 VDBG (2, "VCL_SESS_ATTR_IP_PKTINFO: %d, buflen %d", *(int *) buffer,
4261 *buflen);
4262 }
4263 else
4264 rv = VPPCOM_EINVAL;
4265 break;
Florin Coras87f63892021-05-01 16:01:40 -07004266
Dave Wallacee22aa742017-10-20 12:30:38 -04004267 default:
4268 rv = VPPCOM_EINVAL;
4269 break;
Dave Wallace35830af2017-10-09 01:43:42 -04004270 }
4271
Dave Wallace35830af2017-10-09 01:43:42 -04004272 return rv;
4273}
4274
Stevenac1f96d2017-10-24 16:03:58 -07004275int
Florin Coras134a9962018-08-28 11:32:04 -07004276vppcom_session_recvfrom (uint32_t session_handle, void *buffer,
Stevenac1f96d2017-10-24 16:03:58 -07004277 uint32_t buflen, int flags, vppcom_endpt_t * ep)
4278{
Florin Coras134a9962018-08-28 11:32:04 -07004279 vcl_worker_t *wrk = vcl_worker_get_current ();
Florin Coras5da10c42020-04-01 04:31:21 +00004280 vcl_session_t *session;
Stevenac1f96d2017-10-24 16:03:58 -07004281 int rv = VPPCOM_OK;
Steven58f464e2017-10-25 12:33:12 -07004282
4283 if (flags == 0)
Florin Coras134a9962018-08-28 11:32:04 -07004284 rv = vppcom_session_read (session_handle, buffer, buflen);
Stevenac1f96d2017-10-24 16:03:58 -07004285 else if (flags & MSG_PEEK)
Florin Coras134a9962018-08-28 11:32:04 -07004286 rv = vppcom_session_peek (session_handle, buffer, buflen);
Stevenac1f96d2017-10-24 16:03:58 -07004287 else
4288 {
Florin Corasa7a1a222018-12-30 17:11:31 -08004289 VDBG (0, "Unsupport flags for recvfrom %d", flags);
Florin Coras460dce62018-07-27 05:45:06 -07004290 return VPPCOM_EAFNOSUPPORT;
Stevenac1f96d2017-10-24 16:03:58 -07004291 }
4292
Florin Coras0a1e1832020-03-29 18:54:04 +00004293 if (ep && rv > 0)
Florin Coras99368312018-08-02 10:45:44 -07004294 {
Florin Coras5da10c42020-04-01 04:31:21 +00004295 session = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras99368312018-08-02 10:45:44 -07004296 if (session->transport.is_ip4)
Dave Barach178cf492018-11-13 16:34:13 -05004297 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip4,
4298 sizeof (ip4_address_t));
Florin Coras99368312018-08-02 10:45:44 -07004299 else
Dave Barach178cf492018-11-13 16:34:13 -05004300 clib_memcpy_fast (ep->ip, &session->transport.rmt_ip.ip6,
4301 sizeof (ip6_address_t));
Florin Coras5da10c42020-04-01 04:31:21 +00004302 ep->is_ip4 = session->transport.is_ip4;
4303 ep->port = session->transport.rmt_port;
Florin Coras99368312018-08-02 10:45:44 -07004304 }
Florin Coras460dce62018-07-27 05:45:06 -07004305
Stevenac1f96d2017-10-24 16:03:58 -07004306 return rv;
4307}
4308
Florin Coraseff5f7a2023-02-07 17:36:17 -08004309static void
4310vcl_handle_ep_app_tlvs (vcl_session_t *s, vppcom_endpt_t *ep)
4311{
4312 vppcom_endpt_tlv_t *tlv = ep->app_tlvs;
4313
4314 do
4315 {
4316 switch (tlv->data_type)
4317 {
4318 case VCL_UDP_SEGMENT:
4319 s->gso_size = *(u16 *) tlv->data;
4320 break;
4321 case VCL_IP_PKTINFO:
4322 clib_memcpy_fast (&s->transport.lcl_ip, (ip4_address_t *) tlv->data,
4323 sizeof (ip4_address_t));
4324 break;
4325 default:
4326 VDBG (0, "Ignorning unsupported app tlv %u", tlv->data_type);
4327 break;
4328 }
4329 tlv = VCL_EP_NEXT_APP_TLV (ep, tlv);
4330 }
4331 while (tlv);
4332}
4333
Stevenac1f96d2017-10-24 16:03:58 -07004334int
Florin Coras134a9962018-08-28 11:32:04 -07004335vppcom_session_sendto (uint32_t session_handle, void *buffer,
Stevenac1f96d2017-10-24 16:03:58 -07004336 uint32_t buflen, int flags, vppcom_endpt_t * ep)
4337{
Florin Coras7a2abce2020-04-05 19:25:44 +00004338 vcl_worker_t *wrk = vcl_worker_get_current ();
4339 vcl_session_t *s;
4340
4341 s = vcl_session_get_w_handle (wrk, session_handle);
Florin Coras0b0d28e2021-06-04 17:31:53 -07004342 if (PREDICT_FALSE (!s))
Florin Coras7a2abce2020-04-05 19:25:44 +00004343 return VPPCOM_EBADFD;
4344
Dave Wallace617dffa2017-10-26 14:47:06 -04004345 if (ep)
4346 {
Florin Coras5824cc52020-10-20 18:44:41 -07004347 if (!vcl_session_is_cl (s))
Florin Coras5da10c42020-04-01 04:31:21 +00004348 return VPPCOM_EINVAL;
4349
liuyacanbc0c7542021-08-02 20:15:05 +08004350 s->transport.is_ip4 = ep->is_ip4;
4351 s->transport.rmt_port = ep->port;
4352 vcl_ip_copy_from_ep (&s->transport.rmt_ip, ep);
4353
Florin Coraseff5f7a2023-02-07 17:36:17 -08004354 if (ep->app_tlvs)
4355 vcl_handle_ep_app_tlvs (s, ep);
Dou Chao243a0432022-11-29 19:41:34 +08004356
Florin Coras5da10c42020-04-01 04:31:21 +00004357 /* Session not connected/bound in vpp. Create it by 'connecting' it */
Florin Corasc127d5a2020-10-14 16:35:58 -07004358 if (PREDICT_FALSE (s->session_state == VCL_STATE_CLOSED))
Florin Coras5da10c42020-04-01 04:31:21 +00004359 {
Florin Coras5824cc52020-10-20 18:44:41 -07004360 u32 session_index = s->session_index;
4361 f64 timeout = vcm->cfg.session_timeout;
4362 int rv;
4363
Florin Coras5da10c42020-04-01 04:31:21 +00004364 vcl_send_session_connect (wrk, s);
Florin Coras5824cc52020-10-20 18:44:41 -07004365 rv = vppcom_wait_for_session_state_change (session_index,
4366 VCL_STATE_READY,
4367 timeout);
4368 if (rv < 0)
4369 return rv;
4370 s = vcl_session_get (wrk, session_index);
Florin Coras5da10c42020-04-01 04:31:21 +00004371 }
Dave Wallace617dffa2017-10-26 14:47:06 -04004372 }
4373
4374 if (flags)
4375 {
4376 // TBD check the flags and do the right thing
Florin Coras5e062572019-03-14 19:07:51 -07004377 VDBG (2, "handling flags 0x%u (%d) not implemented yet.", flags, flags);
Dave Wallace617dffa2017-10-26 14:47:06 -04004378 }
4379
Florin Coraseff5f7a2023-02-07 17:36:17 -08004380 return (vppcom_session_write_inline (wrk, s, buffer, buflen, 1,
Florin Coras7a2abce2020-04-05 19:25:44 +00004381 s->is_dgram ? 1 : 0));
Stevenac1f96d2017-10-24 16:03:58 -07004382}
4383
Dave Wallace048b1d62018-01-03 22:24:41 -05004384int
4385vppcom_poll (vcl_poll_t * vp, uint32_t n_sids, double wait_for_time)
4386{
Florin Coras134a9962018-08-28 11:32:04 -07004387 vcl_worker_t *wrk = vcl_worker_get_current ();
4388 f64 timeout = clib_time_now (&wrk->clib_time) + wait_for_time;
Dave Wallace048b1d62018-01-03 22:24:41 -05004389 u32 i, keep_trying = 1;
Florin Coras6917b942018-11-13 22:44:54 -08004390 svm_msg_q_msg_t msg;
4391 session_event_t *e;
Dave Wallace048b1d62018-01-03 22:24:41 -05004392 int rv, num_ev = 0;
4393
Florin Coras5e062572019-03-14 19:07:51 -07004394 VDBG (3, "vp %p, nsids %u, wait_for_time %f", vp, n_sids, wait_for_time);
Dave Wallace048b1d62018-01-03 22:24:41 -05004395
4396 if (!vp)
4397 return VPPCOM_EFAULT;
4398
4399 do
4400 {
Florin Coras7e12d942018-06-27 14:32:43 -07004401 vcl_session_t *session;
Dave Wallace048b1d62018-01-03 22:24:41 -05004402
Florin Coras6917b942018-11-13 22:44:54 -08004403 /* Dequeue all events and drop all unhandled io events */
4404 while (svm_msg_q_sub (wrk->app_event_queue, &msg, SVM_Q_NOWAIT, 0) == 0)
4405 {
4406 e = svm_msg_q_msg_data (wrk->app_event_queue, &msg);
4407 vcl_handle_mq_event (wrk, e);
4408 svm_msg_q_free_msg (wrk->app_event_queue, &msg);
4409 }
4410 vec_reset_length (wrk->unhandled_evts_vector);
4411
Dave Wallace048b1d62018-01-03 22:24:41 -05004412 for (i = 0; i < n_sids; i++)
4413 {
Florin Coras7baeb712019-01-04 17:05:43 -08004414 session = vcl_session_get (wrk, vp[i].sh);
Florin Coras070453d2018-08-24 17:04:27 -07004415 if (!session)
Florin Coras6917b942018-11-13 22:44:54 -08004416 {
4417 vp[i].revents = POLLHUP;
4418 num_ev++;
4419 continue;
4420 }
Dave Wallace048b1d62018-01-03 22:24:41 -05004421
Florin Coras6917b942018-11-13 22:44:54 -08004422 vp[i].revents = 0;
Dave Wallace048b1d62018-01-03 22:24:41 -05004423
4424 if (POLLIN & vp[i].events)
4425 {
Florin Coras0ef8ef22019-01-18 08:37:13 -08004426 rv = vcl_session_read_ready (session);
Dave Wallace048b1d62018-01-03 22:24:41 -05004427 if (rv > 0)
4428 {
Florin Coras6917b942018-11-13 22:44:54 -08004429 vp[i].revents |= POLLIN;
Dave Wallace048b1d62018-01-03 22:24:41 -05004430 num_ev++;
4431 }
4432 else if (rv < 0)
4433 {
4434 switch (rv)
4435 {
4436 case VPPCOM_ECONNRESET:
Florin Coras6917b942018-11-13 22:44:54 -08004437 vp[i].revents = POLLHUP;
Dave Wallace048b1d62018-01-03 22:24:41 -05004438 break;
4439
4440 default:
Florin Coras6917b942018-11-13 22:44:54 -08004441 vp[i].revents = POLLERR;
Dave Wallace048b1d62018-01-03 22:24:41 -05004442 break;
4443 }
4444 num_ev++;
4445 }
4446 }
4447
4448 if (POLLOUT & vp[i].events)
4449 {
Florin Coras0ef8ef22019-01-18 08:37:13 -08004450 rv = vcl_session_write_ready (session);
Dave Wallace048b1d62018-01-03 22:24:41 -05004451 if (rv > 0)
4452 {
Florin Coras6917b942018-11-13 22:44:54 -08004453 vp[i].revents |= POLLOUT;
Dave Wallace048b1d62018-01-03 22:24:41 -05004454 num_ev++;
4455 }
4456 else if (rv < 0)
4457 {
4458 switch (rv)
4459 {
4460 case VPPCOM_ECONNRESET:
Florin Coras6917b942018-11-13 22:44:54 -08004461 vp[i].revents = POLLHUP;
Dave Wallace048b1d62018-01-03 22:24:41 -05004462 break;
4463
4464 default:
Florin Coras6917b942018-11-13 22:44:54 -08004465 vp[i].revents = POLLERR;
Dave Wallace048b1d62018-01-03 22:24:41 -05004466 break;
4467 }
4468 num_ev++;
4469 }
4470 }
4471
Dave Wallace7e607a72018-06-18 18:41:32 -04004472 if (0) // Note "done:" label used by VCL_SESSION_LOCK_AND_GET()
Dave Wallace048b1d62018-01-03 22:24:41 -05004473 {
Florin Coras6917b942018-11-13 22:44:54 -08004474 vp[i].revents = POLLNVAL;
Dave Wallace048b1d62018-01-03 22:24:41 -05004475 num_ev++;
4476 }
4477 }
4478 if (wait_for_time != -1)
Florin Coras134a9962018-08-28 11:32:04 -07004479 keep_trying = (clib_time_now (&wrk->clib_time) <= timeout) ? 1 : 0;
Dave Wallace048b1d62018-01-03 22:24:41 -05004480 }
4481 while ((num_ev == 0) && keep_trying);
4482
Dave Wallace048b1d62018-01-03 22:24:41 -05004483 return num_ev;
4484}
4485
Florin Coras99368312018-08-02 10:45:44 -07004486int
4487vppcom_mq_epoll_fd (void)
4488{
Florin Coras134a9962018-08-28 11:32:04 -07004489 vcl_worker_t *wrk = vcl_worker_get_current ();
4490 return wrk->mqs_epfd;
4491}
4492
4493int
Florin Coras30e79c22019-01-02 19:31:22 -08004494vppcom_session_index (vcl_session_handle_t session_handle)
Florin Coras134a9962018-08-28 11:32:04 -07004495{
4496 return session_handle & 0xFFFFFF;
4497}
4498
4499int
Florin Coras30e79c22019-01-02 19:31:22 -08004500vppcom_session_worker (vcl_session_handle_t session_handle)
4501{
4502 return session_handle >> 24;
4503}
4504
4505int
Florin Coras134a9962018-08-28 11:32:04 -07004506vppcom_worker_register (void)
4507{
Florin Coras47c40e22018-11-26 17:01:36 -08004508 if (!vcl_worker_alloc_and_init ())
4509 return VPPCOM_EEXIST;
4510
Florin Coras47c40e22018-11-26 17:01:36 -08004511 if (vcl_worker_register_with_vpp ())
4512 return VPPCOM_EEXIST;
4513
4514 return VPPCOM_OK;
Florin Coras99368312018-08-02 10:45:44 -07004515}
4516
Florin Coras369db832019-07-08 12:34:45 -07004517void
4518vppcom_worker_unregister (void)
4519{
4520 vcl_worker_cleanup (vcl_worker_get_current (), 1 /* notify vpp */ );
4521 vcl_set_worker_index (~0);
4522}
4523
Pivo6017ff02020-05-04 17:57:33 +02004524void
4525vppcom_worker_index_set (int index)
4526{
4527 vcl_set_worker_index (index);
4528}
4529
Florin Corasdfe4cf42018-11-28 22:13:45 -08004530int
4531vppcom_worker_index (void)
4532{
4533 return vcl_get_worker_index ();
4534}
4535
Florin Corase0982e52019-01-25 13:19:56 -08004536int
4537vppcom_worker_mqs_epfd (void)
4538{
4539 vcl_worker_t *wrk = vcl_worker_get_current ();
4540 if (!vcm->cfg.use_mq_eventfd)
4541 return -1;
4542 return wrk->mqs_epfd;
4543}
4544
Nathan Skrzypczak9fd99622019-05-16 14:38:44 +02004545int
4546vppcom_session_is_connectable_listener (uint32_t session_handle)
4547{
4548 vcl_session_t *session;
4549 vcl_worker_t *wrk = vcl_worker_get_current ();
4550 session = vcl_session_get_w_handle (wrk, session_handle);
4551 if (!session)
4552 return VPPCOM_EBADFD;
4553 return vcl_session_is_connectable_listener (wrk, session);
4554}
4555
4556int
4557vppcom_session_listener (uint32_t session_handle)
4558{
4559 vcl_worker_t *wrk = vcl_worker_get_current ();
4560 vcl_session_t *listen_session, *session;
4561 session = vcl_session_get_w_handle (wrk, session_handle);
4562 if (!session)
4563 return VPPCOM_EBADFD;
4564 if (session->listener_index == VCL_INVALID_SESSION_INDEX)
4565 return VPPCOM_EBADFD;
4566 listen_session = vcl_session_get_w_handle (wrk, session->listener_index);
4567 if (!listen_session)
4568 return VPPCOM_EBADFD;
4569 return vcl_session_handle (listen_session);
4570}
4571
4572int
4573vppcom_session_n_accepted (uint32_t session_handle)
4574{
4575 vcl_worker_t *wrk = vcl_worker_get_current ();
4576 vcl_session_t *session = vcl_session_get_w_handle (wrk, session_handle);
4577 if (!session)
4578 return VPPCOM_EBADFD;
4579 return session->n_accepted_sessions;
4580}
4581
Florin Coras66ec4672020-06-15 07:59:40 -07004582const char *
4583vppcom_proto_str (vppcom_proto_t proto)
4584{
4585 char const *proto_str;
4586
4587 switch (proto)
4588 {
4589 case VPPCOM_PROTO_TCP:
4590 proto_str = "TCP";
4591 break;
4592 case VPPCOM_PROTO_UDP:
4593 proto_str = "UDP";
4594 break;
4595 case VPPCOM_PROTO_TLS:
4596 proto_str = "TLS";
4597 break;
4598 case VPPCOM_PROTO_QUIC:
4599 proto_str = "QUIC";
4600 break;
Florin Coras4b47ee22020-11-19 13:38:26 -08004601 case VPPCOM_PROTO_DTLS:
4602 proto_str = "DTLS";
4603 break;
Florin Coras6621abf2021-01-06 17:35:17 -08004604 case VPPCOM_PROTO_SRTP:
4605 proto_str = "SRTP";
4606 break;
Florin Coras66ec4672020-06-15 07:59:40 -07004607 default:
4608 proto_str = "UNKNOWN";
4609 break;
4610 }
4611 return proto_str;
4612}
4613
4614const char *
4615vppcom_retval_str (int retval)
4616{
4617 char const *st;
4618
4619 switch (retval)
4620 {
4621 case VPPCOM_OK:
4622 st = "VPPCOM_OK";
4623 break;
4624
4625 case VPPCOM_EAGAIN:
4626 st = "VPPCOM_EAGAIN";
4627 break;
4628
4629 case VPPCOM_EFAULT:
4630 st = "VPPCOM_EFAULT";
4631 break;
4632
4633 case VPPCOM_ENOMEM:
4634 st = "VPPCOM_ENOMEM";
4635 break;
4636
4637 case VPPCOM_EINVAL:
4638 st = "VPPCOM_EINVAL";
4639 break;
4640
4641 case VPPCOM_EBADFD:
4642 st = "VPPCOM_EBADFD";
4643 break;
4644
4645 case VPPCOM_EAFNOSUPPORT:
4646 st = "VPPCOM_EAFNOSUPPORT";
4647 break;
4648
4649 case VPPCOM_ECONNABORTED:
4650 st = "VPPCOM_ECONNABORTED";
4651 break;
4652
4653 case VPPCOM_ECONNRESET:
4654 st = "VPPCOM_ECONNRESET";
4655 break;
4656
4657 case VPPCOM_ENOTCONN:
4658 st = "VPPCOM_ENOTCONN";
4659 break;
4660
4661 case VPPCOM_ECONNREFUSED:
4662 st = "VPPCOM_ECONNREFUSED";
4663 break;
4664
4665 case VPPCOM_ETIMEDOUT:
4666 st = "VPPCOM_ETIMEDOUT";
4667 break;
4668
Radha krishna Saragadamdd92bde2022-07-18 19:41:05 +05304669 case VPPCOM_EADDRINUSE:
4670 st = "VPPCOM_EADDRINUSE";
4671 break;
4672
Florin Coras66ec4672020-06-15 07:59:40 -07004673 default:
4674 st = "UNKNOWN_STATE";
4675 break;
4676 }
4677
4678 return st;
4679}
4680
Florin Corasa5a9efd2021-01-05 17:03:29 -08004681int
4682vppcom_add_cert_key_pair (vppcom_cert_key_pair_t *ckpair)
4683{
4684 if (vcm->cfg.vpp_app_socket_api)
Florin Corase191d762021-08-11 14:55:49 -07004685 return vcl_sapi_add_cert_key_pair (ckpair);
4686 else
4687 return vcl_bapi_add_cert_key_pair (ckpair);
Florin Corasa5a9efd2021-01-05 17:03:29 -08004688}
4689
4690int
4691vppcom_del_cert_key_pair (uint32_t ckpair_index)
4692{
4693 if (vcm->cfg.vpp_app_socket_api)
Florin Corase191d762021-08-11 14:55:49 -07004694 return vcl_sapi_del_cert_key_pair (ckpair_index);
4695 else
4696 return vcl_bapi_del_cert_key_pair (ckpair_index);
Florin Corasa5a9efd2021-01-05 17:03:29 -08004697}
4698
Radha krishna Saragadamdd92bde2022-07-18 19:41:05 +05304699int
4700vppcom_session_get_error (uint32_t session_handle)
4701{
4702 vcl_worker_t *wrk = vcl_worker_get_current ();
4703 vcl_session_t *session = 0;
4704
4705 session = vcl_session_get_w_handle (wrk, session_handle);
4706 if (!session)
4707 return VPPCOM_EBADFD;
4708
4709 if (PREDICT_FALSE (session->flags & VCL_SESSION_F_IS_VEP))
4710 {
4711 VWRN ("epoll session %u! will not have connect", session->session_index);
4712 return VPPCOM_EBADFD;
4713 }
4714
4715 if (session->vpp_error == SESSION_E_PORTINUSE)
4716 return VPPCOM_EADDRINUSE;
4717 else if (session->vpp_error == SESSION_E_REFUSED)
4718 return VPPCOM_ECONNREFUSED;
4719 else if (session->vpp_error != SESSION_E_NONE)
4720 return VPPCOM_EFAULT;
4721 else
4722 return VPPCOM_OK;
4723}
4724
Maros Ondrejicka6ff8e902022-10-06 18:17:05 +02004725int
4726vppcom_worker_is_detached (void)
4727{
4728 vcl_worker_t *wrk = vcl_worker_get_current ();
4729
4730 if (!vcm->cfg.use_mq_eventfd)
4731 return VPPCOM_ENOTSUP;
4732
4733 return wrk->api_client_handle == ~0;
4734}
4735
Dave Wallacee22aa742017-10-20 12:30:38 -04004736/*
4737 * fd.io coding-style-patch-verification: ON
4738 *
4739 * Local Variables:
4740 * eval: (c-set-style "gnu")
4741 * End:
4742 */