src/vnet/session/session_node.c

   1 /*
   2  * Copyright (c) 2017-2019 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <math.h>
  17 #include <vlib/vlib.h>
  18 #include <vnet/vnet.h>
  19 #include <vppinfra/elog.h>
  20 #include <vnet/session/transport.h>
  21 #include <vnet/session/session.h>
  22 #include <vnet/session/application.h>
  23 #include <vnet/session/application_interface.h>
  24 #include <vnet/session/application_local.h>
  25 #include <vnet/session/session_debug.h>
  26 #include <svm/queue.h>
  27
  28 static void session_mq_accepted_reply_handler (void *data);
  29
  30 static void
  31 accepted_notify_cb (void *data, u32 data_len)
  32 {
  33   session_mq_accepted_reply_handler (data);
  34 }
  35
  36 static void
  37 session_mq_accepted_reply_handler (void *data)
  38 {
  39   session_accepted_reply_msg_t *mp = (session_accepted_reply_msg_t *) data;
  40   vnet_disconnect_args_t _a = { 0 }, *a = &_a;
  41   session_state_t old_state;
  42   app_worker_t *app_wrk;
  43   session_t *s;
  44
  45   /* Server isn't interested, kill the session */
  46   if (mp->retval)
  47     {
  48       a->app_index = mp->context;
  49       a->handle = mp->handle;
  50       vnet_disconnect_session (a);
  51       return;
  52     }
  53
  54   /* Mail this back from the main thread. We're not polling in main
  55    * thread so we're using other workers for notifications. */
  56   if (vlib_num_workers () && vlib_get_thread_index () != 0
  57       && session_thread_from_handle (mp->handle) == 0)
  58     {
  59       vl_api_rpc_call_main_thread (accepted_notify_cb, data,
  60                                    sizeof (session_accepted_reply_msg_t));
  61       return;
  62     }
  63
  64   s = session_get_from_handle_if_valid (mp->handle);
  65   if (!s)
  66     return;
  67
  68   app_wrk = app_worker_get (s->app_wrk_index);
  69   if (app_wrk->app_index != mp->context)
  70     {
  71       clib_warning ("app doesn't own session");
  72       return;
  73     }
  74
  75   if (!session_has_transport (s))
  76     {
  77       s->session_state = SESSION_STATE_READY;
  78       if (ct_session_connect_notify (s))
  79         return;
  80     }
  81   else
  82     {
  83       old_state = s->session_state;
  84       s->session_state = SESSION_STATE_READY;
  85
  86       if (!svm_fifo_is_empty_prod (s->rx_fifo))
  87         app_worker_lock_and_send_event (app_wrk, s, SESSION_IO_EVT_RX);
  88
  89       /* Closed while waiting for app to reply. Resend disconnect */
  90       if (old_state >= SESSION_STATE_TRANSPORT_CLOSING)
  91         {
  92           app_worker_close_notify (app_wrk, s);
  93           s->session_state = old_state;
  94           return;
  95         }
  96     }
  97 }
  98
  99 static void
 100 session_mq_reset_reply_handler (void *data)
 101 {
 102   vnet_disconnect_args_t _a = { 0 }, *a = &_a;
 103   session_reset_reply_msg_t *mp;
 104   app_worker_t *app_wrk;
 105   session_t *s;
 106   application_t *app;
 107   u32 index, thread_index;
 108
 109   mp = (session_reset_reply_msg_t *) data;
 110   app = application_lookup (mp->context);
 111   if (!app)
 112     return;
 113
 114   session_parse_handle (mp->handle, &index, &thread_index);
 115   s = session_get_if_valid (index, thread_index);
 116
 117   /* Session was already closed or already cleaned up */
 118   if (!s || s->session_state != SESSION_STATE_TRANSPORT_CLOSING)
 119     return;
 120
 121   app_wrk = app_worker_get (s->app_wrk_index);
 122   if (!app_wrk || app_wrk->app_index != app->app_index)
 123     {
 124       clib_warning ("App % does not own handle 0x%lx!", app->app_index,
 125                     mp->handle);
 126       return;
 127     }
 128
 129   /* Client objected to resetting the session, log and continue */
 130   if (mp->retval)
 131     {
 132       clib_warning ("client retval %d", mp->retval);
 133       return;
 134     }
 135
 136   /* This comes as a response to a reset, transport only waiting for
 137    * confirmation to remove connection state, no need to disconnect */
 138   a->handle = mp->handle;
 139   a->app_index = app->app_index;
 140   vnet_disconnect_session (a);
 141 }
 142
 143 static void
 144 session_mq_disconnected_handler (void *data)
 145 {
 146   session_disconnected_reply_msg_t *rmp;
 147   vnet_disconnect_args_t _a, *a = &_a;
 148   svm_msg_q_msg_t _msg, *msg = &_msg;
 149   session_disconnected_msg_t *mp;
 150   app_worker_t *app_wrk;
 151   session_event_t *evt;
 152   session_t *s;
 153   application_t *app;
 154   int rv = 0;
 155
 156   mp = (session_disconnected_msg_t *) data;
 157   if (!(s = session_get_from_handle_if_valid (mp->handle)))
 158     {
 159       clib_warning ("could not disconnect handle %llu", mp->handle);
 160       return;
 161     }
 162   app_wrk = app_worker_get (s->app_wrk_index);
 163   app = application_lookup (mp->client_index);
 164   if (!(app_wrk && app && app->app_index == app_wrk->app_index))
 165     {
 166       clib_warning ("could not disconnect session: %llu app: %u",
 167                     mp->handle, mp->client_index);
 168       return;
 169     }
 170
 171   a->handle = mp->handle;
 172   a->app_index = app_wrk->wrk_index;
 173   rv = vnet_disconnect_session (a);
 174
 175   svm_msg_q_lock_and_alloc_msg_w_ring (app_wrk->event_queue,
 176                                        SESSION_MQ_CTRL_EVT_RING,
 177                                        SVM_Q_WAIT, msg);
 178   svm_msg_q_unlock (app_wrk->event_queue);
 179   evt = svm_msg_q_msg_data (app_wrk->event_queue, msg);
 180   clib_memset (evt, 0, sizeof (*evt));
 181   evt->event_type = SESSION_CTRL_EVT_DISCONNECTED_REPLY;
 182   rmp = (session_disconnected_reply_msg_t *) evt->data;
 183   rmp->handle = mp->handle;
 184   rmp->context = mp->context;
 185   rmp->retval = rv;
 186   svm_msg_q_add (app_wrk->event_queue, msg, SVM_Q_WAIT);
 187 }
 188
 189 static void
 190 session_mq_disconnected_reply_handler (void *data)
 191 {
 192   session_disconnected_reply_msg_t *mp;
 193   vnet_disconnect_args_t _a, *a = &_a;
 194   application_t *app;
 195
 196   mp = (session_disconnected_reply_msg_t *) data;
 197
 198   /* Client objected to disconnecting the session, log and continue */
 199   if (mp->retval)
 200     {
 201       clib_warning ("client retval %d", mp->retval);
 202       return;
 203     }
 204
 205   /* Disconnect has been confirmed. Confirm close to transport */
 206   app = application_lookup (mp->context);
 207   if (app)
 208     {
 209       a->handle = mp->handle;
 210       a->app_index = app->app_index;
 211       vnet_disconnect_session (a);
 212     }
 213 }
 214
 215 static void
 216 session_mq_worker_update_handler (void *data)
 217 {
 218   session_worker_update_msg_t *mp = (session_worker_update_msg_t *) data;
 219   session_worker_update_reply_msg_t *rmp;
 220   svm_msg_q_msg_t _msg, *msg = &_msg;
 221   app_worker_t *app_wrk;
 222   u32 owner_app_wrk_map;
 223   session_event_t *evt;
 224   session_t *s;
 225   application_t *app;
 226
 227   app = application_lookup (mp->client_index);
 228   if (!app)
 229     return;
 230   if (!(s = session_get_from_handle_if_valid (mp->handle)))
 231     {
 232       clib_warning ("invalid handle %llu", mp->handle);
 233       return;
 234     }
 235   app_wrk = app_worker_get (s->app_wrk_index);
 236   if (app_wrk->app_index != app->app_index)
 237     {
 238       clib_warning ("app %u does not own session %llu", app->app_index,
 239                     mp->handle);
 240       return;
 241     }
 242   owner_app_wrk_map = app_wrk->wrk_map_index;
 243   app_wrk = application_get_worker (app, mp->wrk_index);
 244
 245   /* This needs to come from the new owner */
 246   if (mp->req_wrk_index == owner_app_wrk_map)
 247     {
 248       session_req_worker_update_msg_t *wump;
 249
 250       svm_msg_q_lock_and_alloc_msg_w_ring (app_wrk->event_queue,
 251                                            SESSION_MQ_CTRL_EVT_RING,
 252                                            SVM_Q_WAIT, msg);
 253       svm_msg_q_unlock (app_wrk->event_queue);
 254       evt = svm_msg_q_msg_data (app_wrk->event_queue, msg);
 255       clib_memset (evt, 0, sizeof (*evt));
 256       evt->event_type = SESSION_CTRL_EVT_REQ_WORKER_UPDATE;
 257       wump = (session_req_worker_update_msg_t *) evt->data;
 258       wump->session_handle = mp->handle;
 259       svm_msg_q_add (app_wrk->event_queue, msg, SVM_Q_WAIT);
 260       return;
 261     }
 262
 263   app_worker_own_session (app_wrk, s);
 264
 265   /*
 266    * Send reply
 267    */
 268   svm_msg_q_lock_and_alloc_msg_w_ring (app_wrk->event_queue,
 269                                        SESSION_MQ_CTRL_EVT_RING,
 270                                        SVM_Q_WAIT, msg);
 271   svm_msg_q_unlock (app_wrk->event_queue);
 272   evt = svm_msg_q_msg_data (app_wrk->event_queue, msg);
 273   clib_memset (evt, 0, sizeof (*evt));
 274   evt->event_type = SESSION_CTRL_EVT_WORKER_UPDATE_REPLY;
 275   rmp = (session_worker_update_reply_msg_t *) evt->data;
 276   rmp->handle = mp->handle;
 277   rmp->rx_fifo = pointer_to_uword (s->rx_fifo);
 278   rmp->tx_fifo = pointer_to_uword (s->tx_fifo);
 279   rmp->segment_handle = session_segment_handle (s);
 280   svm_msg_q_add (app_wrk->event_queue, msg, SVM_Q_WAIT);
 281
 282   /*
 283    * Retransmit messages that may have been lost
 284    */
 285   if (s->tx_fifo && !svm_fifo_is_empty (s->tx_fifo))
 286     session_send_io_evt_to_thread (s->tx_fifo, SESSION_IO_EVT_TX);
 287
 288   if (s->rx_fifo && !svm_fifo_is_empty (s->rx_fifo))
 289     app_worker_lock_and_send_event (app_wrk, s, SESSION_IO_EVT_RX);
 290
 291   if (s->session_state >= SESSION_STATE_TRANSPORT_CLOSING)
 292     app_worker_close_notify (app_wrk, s);
 293 }
 294
 295 vlib_node_registration_t session_queue_node;
 296
 297 typedef struct
 298 {
 299   u32 session_index;
 300   u32 server_thread_index;
 301 } session_queue_trace_t;
 302
 303 /* packet trace format function */
 304 static u8 *
 305 format_session_queue_trace (u8 * s, va_list * args)
 306 {
 307   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
 308   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
 309   session_queue_trace_t *t = va_arg (*args, session_queue_trace_t *);
 310
 311   s = format (s, "SESSION_QUEUE: session index %d, server thread index %d",
 312               t->session_index, t->server_thread_index);
 313   return s;
 314 }
 315
 316 #define foreach_session_queue_error             \
 317 _(TX, "Packets transmitted")                    \
 318 _(TIMER, "Timer events")                        \
 319 _(NO_BUFFER, "Out of buffers")
 320
 321 typedef enum
 322 {
 323 #define _(sym,str) SESSION_QUEUE_ERROR_##sym,
 324   foreach_session_queue_error
 325 #undef _
 326     SESSION_QUEUE_N_ERROR,
 327 } session_queue_error_t;
 328
 329 static char *session_queue_error_strings[] = {
 330 #define _(sym,string) string,
 331   foreach_session_queue_error
 332 #undef _
 333 };
 334
 335 enum
 336 {
 337   SESSION_TX_NO_BUFFERS = -2,
 338   SESSION_TX_NO_DATA,
 339   SESSION_TX_OK
 340 };
 341
 342 static void
 343 session_tx_trace_frame (vlib_main_t * vm, vlib_node_runtime_t * node,
 344                         u32 next_index, u32 * to_next, u16 n_segs,
 345                         session_t * s, u32 n_trace)
 346 {
 347   session_queue_trace_t *t;
 348   vlib_buffer_t *b;
 349   int i;
 350
 351   for (i = 0; i < clib_min (n_trace, n_segs); i++)
 352     {
 353       b = vlib_get_buffer (vm, to_next[i - n_segs]);
 354       vlib_trace_buffer (vm, node, next_index, b, 1 /* follow_chain */ );
 355       t = vlib_add_trace (vm, node, b, sizeof (*t));
 356       t->session_index = s->session_index;
 357       t->server_thread_index = s->thread_index;
 358     }
 359   vlib_set_trace_count (vm, node, n_trace - i);
 360 }
 361
 362 always_inline void
 363 session_tx_fifo_chain_tail (vlib_main_t * vm, session_tx_context_t * ctx,
 364                             vlib_buffer_t * b, u16 * n_bufs, u8 peek_data)
 365 {
 366   vlib_buffer_t *chain_b, *prev_b;
 367   u32 chain_bi0, to_deq, left_from_seg;
 368   session_worker_t *wrk;
 369   u16 len_to_deq, n_bytes_read;
 370   u8 *data, j;
 371
 372   wrk = session_main_get_worker (ctx->s->thread_index);
 373   b->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
 374   b->total_length_not_including_first_buffer = 0;
 375
 376   chain_b = b;
 377   left_from_seg = clib_min (ctx->snd_mss - b->current_length,
 378                             ctx->left_to_snd);
 379   to_deq = left_from_seg;
 380   for (j = 1; j < ctx->n_bufs_per_seg; j++)
 381     {
 382       prev_b = chain_b;
 383       len_to_deq = clib_min (to_deq, ctx->deq_per_buf);
 384
 385       *n_bufs -= 1;
 386       chain_bi0 = wrk->tx_buffers[*n_bufs];
 387       chain_b = vlib_get_buffer (vm, chain_bi0);
 388       chain_b->current_data = 0;
 389       data = vlib_buffer_get_current (chain_b);
 390       if (peek_data)
 391         {
 392           n_bytes_read = svm_fifo_peek (ctx->s->tx_fifo,
 393                                         ctx->tx_offset, len_to_deq, data);
 394           ctx->tx_offset += n_bytes_read;
 395         }
 396       else
 397         {
 398           if (ctx->transport_vft->transport_options.tx_type ==
 399               TRANSPORT_TX_DGRAM)
 400             {
 401               svm_fifo_t *f = ctx->s->tx_fifo;
 402               session_dgram_hdr_t *hdr = &ctx->hdr;
 403               u16 deq_now;
 404               deq_now = clib_min (hdr->data_length - hdr->data_offset,
 405                                   len_to_deq);
 406               n_bytes_read = svm_fifo_peek (f, hdr->data_offset, deq_now,
 407                                             data);
 408               ASSERT (n_bytes_read > 0);
 409
 410               hdr->data_offset += n_bytes_read;
 411               if (hdr->data_offset == hdr->data_length)
 412                 {
 413                   u32 offset = hdr->data_length + SESSION_CONN_HDR_LEN;
 414                   svm_fifo_dequeue_drop (f, offset);
 415                 }
 416             }
 417           else
 418             n_bytes_read = svm_fifo_dequeue (ctx->s->tx_fifo,
 419                                              len_to_deq, data);
 420         }
 421       ASSERT (n_bytes_read == len_to_deq);
 422       chain_b->current_length = n_bytes_read;
 423       b->total_length_not_including_first_buffer += chain_b->current_length;
 424
 425       /* update previous buffer */
 426       prev_b->next_buffer = chain_bi0;
 427       prev_b->flags |= VLIB_BUFFER_NEXT_PRESENT;
 428
 429       /* update current buffer */
 430       chain_b->next_buffer = 0;
 431
 432       to_deq -= n_bytes_read;
 433       if (to_deq == 0)
 434         break;
 435     }
 436   ASSERT (to_deq == 0
 437           && b->total_length_not_including_first_buffer == left_from_seg);
 438   ctx->left_to_snd -= left_from_seg;
 439 }
 440
 441 always_inline void
 442 session_tx_fill_buffer (vlib_main_t * vm, session_tx_context_t * ctx,
 443                         vlib_buffer_t * b, u16 * n_bufs, u8 peek_data)
 444 {
 445   u32 len_to_deq;
 446   u8 *data0;
 447   int n_bytes_read;
 448
 449   /*
 450    * Start with the first buffer in chain
 451    */
 452   b->error = 0;
 453   b->flags = VNET_BUFFER_F_LOCALLY_ORIGINATED;
 454   b->current_data = 0;
 455
 456   data0 = vlib_buffer_make_headroom (b, TRANSPORT_MAX_HDRS_LEN);
 457   len_to_deq = clib_min (ctx->left_to_snd, ctx->deq_per_first_buf);
 458
 459   if (peek_data)
 460     {
 461       n_bytes_read = svm_fifo_peek (ctx->s->tx_fifo, ctx->tx_offset,
 462                                     len_to_deq, data0);
 463       ASSERT (n_bytes_read > 0);
 464       /* Keep track of progress locally, transport is also supposed to
 465        * increment it independently when pushing the header */
 466       ctx->tx_offset += n_bytes_read;
 467     }
 468   else
 469     {
 470       if (ctx->transport_vft->transport_options.tx_type == TRANSPORT_TX_DGRAM)
 471         {
 472           session_dgram_hdr_t *hdr = &ctx->hdr;
 473           svm_fifo_t *f = ctx->s->tx_fifo;
 474           u16 deq_now;
 475           u32 offset;
 476
 477           ASSERT (hdr->data_length > hdr->data_offset);
 478           deq_now = clib_min (hdr->data_length - hdr->data_offset,
 479                               len_to_deq);
 480           offset = hdr->data_offset + SESSION_CONN_HDR_LEN;
 481           n_bytes_read = svm_fifo_peek (f, offset, deq_now, data0);
 482           ASSERT (n_bytes_read > 0);
 483
 484           if (ctx->s->session_state == SESSION_STATE_LISTENING)
 485             {
 486               ip_copy (&ctx->tc->rmt_ip, &hdr->rmt_ip, ctx->tc->is_ip4);
 487               ctx->tc->rmt_port = hdr->rmt_port;
 488             }
 489           hdr->data_offset += n_bytes_read;
 490           if (hdr->data_offset == hdr->data_length)
 491             {
 492               offset = hdr->data_length + SESSION_CONN_HDR_LEN;
 493               svm_fifo_dequeue_drop (f, offset);
 494             }
 495         }
 496       else
 497         {
 498           n_bytes_read = svm_fifo_dequeue (ctx->s->tx_fifo,
 499                                            len_to_deq, data0);
 500           ASSERT (n_bytes_read > 0);
 501         }
 502     }
 503   b->current_length = n_bytes_read;
 504   ctx->left_to_snd -= n_bytes_read;
 505
 506   /*
 507    * Fill in the remaining buffers in the chain, if any
 508    */
 509   if (PREDICT_FALSE (ctx->n_bufs_per_seg > 1 && ctx->left_to_snd))
 510     session_tx_fifo_chain_tail (vm, ctx, b, n_bufs, peek_data);
 511 }
 512
 513 always_inline u8
 514 session_tx_not_ready (session_t * s, u8 peek_data)
 515 {
 516   if (peek_data)
 517     {
 518       /* Can retransmit for closed sessions but can't send new data if
 519        * session is not ready or closed */
 520       if (s->session_state < SESSION_STATE_READY)
 521         return 1;
 522       if (s->session_state >= SESSION_STATE_TRANSPORT_CLOSED)
 523         return 2;
 524     }
 525   return 0;
 526 }
 527
 528 always_inline transport_connection_t *
 529 session_tx_get_transport (session_tx_context_t * ctx, u8 peek_data)
 530 {
 531   if (peek_data)
 532     {
 533       return ctx->transport_vft->get_connection (ctx->s->connection_index,
 534                                                  ctx->s->thread_index);
 535     }
 536   else
 537     {
 538       if (ctx->s->session_state == SESSION_STATE_LISTENING)
 539         return ctx->transport_vft->get_listener (ctx->s->connection_index);
 540       else
 541         {
 542           return ctx->transport_vft->get_connection (ctx->s->connection_index,
 543                                                      ctx->s->thread_index);
 544         }
 545     }
 546 }
 547
 548 always_inline void
 549 session_tx_set_dequeue_params (vlib_main_t * vm, session_tx_context_t * ctx,
 550                                u32 max_segs, u8 peek_data)
 551 {
 552   u32 n_bytes_per_buf, n_bytes_per_seg;
 553   ctx->max_dequeue = svm_fifo_max_dequeue_cons (ctx->s->tx_fifo);
 554   if (peek_data)
 555     {
 556       /* Offset in rx fifo from where to peek data */
 557       ctx->tx_offset = ctx->transport_vft->tx_fifo_offset (ctx->tc);
 558       if (PREDICT_FALSE (ctx->tx_offset >= ctx->max_dequeue))
 559         {
 560           ctx->max_len_to_snd = 0;
 561           return;
 562         }
 563       ctx->max_dequeue -= ctx->tx_offset;
 564     }
 565   else
 566     {
 567       if (ctx->transport_vft->transport_options.tx_type == TRANSPORT_TX_DGRAM)
 568         {
 569           if (ctx->max_dequeue <= sizeof (ctx->hdr))
 570             {
 571               ctx->max_len_to_snd = 0;
 572               return;
 573             }
 574           svm_fifo_peek (ctx->s->tx_fifo, 0, sizeof (ctx->hdr),
 575                          (u8 *) & ctx->hdr);
 576           ASSERT (ctx->hdr.data_length > ctx->hdr.data_offset);
 577           ctx->max_dequeue = ctx->hdr.data_length - ctx->hdr.data_offset;
 578         }
 579     }
 580   ASSERT (ctx->max_dequeue > 0);
 581
 582   /* Ensure we're not writing more than transport window allows */
 583   if (ctx->max_dequeue < ctx->snd_space)
 584     {
 585       /* Constrained by tx queue. Try to send only fully formed segments */
 586       ctx->max_len_to_snd =
 587         (ctx->max_dequeue > ctx->snd_mss) ?
 588         ctx->max_dequeue - ctx->max_dequeue % ctx->snd_mss : ctx->max_dequeue;
 589       /* TODO Nagle ? */
 590     }
 591   else
 592     {
 593       /* Expectation is that snd_space0 is already a multiple of snd_mss */
 594       ctx->max_len_to_snd = ctx->snd_space;
 595     }
 596
 597   /* Check if we're tx constrained by the node */
 598   ctx->n_segs_per_evt = ceil ((f64) ctx->max_len_to_snd / ctx->snd_mss);
 599   if (ctx->n_segs_per_evt > max_segs)
 600     {
 601       ctx->n_segs_per_evt = max_segs;
 602       ctx->max_len_to_snd = max_segs * ctx->snd_mss;
 603     }
 604
 605   n_bytes_per_buf = vlib_buffer_get_default_data_size (vm);
 606   ASSERT (n_bytes_per_buf > TRANSPORT_MAX_HDRS_LEN);
 607   n_bytes_per_seg = TRANSPORT_MAX_HDRS_LEN + ctx->snd_mss;
 608   ctx->n_bufs_per_seg = ceil ((f64) n_bytes_per_seg / n_bytes_per_buf);
 609   ctx->deq_per_buf = clib_min (ctx->snd_mss, n_bytes_per_buf);
 610   ctx->deq_per_first_buf = clib_min (ctx->snd_mss,
 611                                      n_bytes_per_buf -
 612                                      TRANSPORT_MAX_HDRS_LEN);
 613 }
 614
 615 always_inline int
 616 session_tx_fifo_read_and_snd_i (session_worker_t * wrk,
 617                                 vlib_node_runtime_t * node,
 618                                 session_evt_elt_t * elt,
 619                                 int *n_tx_packets, u8 peek_data)
 620 {
 621   u32 next_index, next0, next1, *to_next, n_left_to_next, max_burst;
 622   u32 n_trace, n_bufs_needed = 0, n_left, pbi;
 623   session_tx_context_t *ctx = &wrk->ctx;
 624   session_main_t *smm = &session_main;
 625   session_event_t *e = &elt->evt;
 626   vlib_main_t *vm = wrk->vm;
 627   transport_proto_t tp;
 628   vlib_buffer_t *pb;
 629   u16 n_bufs, rv;
 630
 631   if (PREDICT_FALSE ((rv = session_tx_not_ready (ctx->s, peek_data))))
 632     {
 633       if (rv < 2)
 634         session_evt_add_old (wrk, elt);
 635       return SESSION_TX_NO_DATA;
 636     }
 637
 638   next_index = smm->session_type_to_next[ctx->s->session_type];
 639   next0 = next1 = next_index;
 640   max_burst = VLIB_FRAME_SIZE - *n_tx_packets;
 641
 642   tp = session_get_transport_proto (ctx->s);
 643   ctx->transport_vft = transport_protocol_get_vft (tp);
 644   ctx->tc = session_tx_get_transport (ctx, peek_data);
 645   ctx->snd_mss = ctx->transport_vft->send_mss (ctx->tc);
 646
 647   if (PREDICT_FALSE (e->event_type == SESSION_IO_EVT_TX_FLUSH))
 648     {
 649       if (ctx->transport_vft->flush_data)
 650         ctx->transport_vft->flush_data (ctx->tc);
 651     }
 652
 653   if (ctx->s->flags & SESSION_F_CUSTOM_TX)
 654     {
 655       u32 n_custom_tx;
 656       ctx->s->flags &= ~SESSION_F_CUSTOM_TX;
 657       n_custom_tx = ctx->transport_vft->custom_tx (ctx->tc, max_burst);
 658       *n_tx_packets += n_custom_tx;
 659       max_burst -= n_custom_tx;
 660       if (!max_burst)
 661         {
 662           session_evt_add_old (wrk, elt);
 663           return SESSION_TX_OK;
 664         }
 665     }
 666
 667   ctx->snd_space = transport_connection_snd_space (ctx->tc,
 668                                                    vm->clib_time.
 669                                                    last_cpu_time,
 670                                                    ctx->snd_mss);
 671
 672   if (ctx->snd_space == 0 || ctx->snd_mss == 0)
 673     {
 674       session_evt_add_old (wrk, elt);
 675       return SESSION_TX_NO_DATA;
 676     }
 677
 678   /* Allow enqueuing of a new event */
 679   svm_fifo_unset_event (ctx->s->tx_fifo);
 680
 681   /* Check how much we can pull. */
 682   session_tx_set_dequeue_params (vm, ctx, max_burst, peek_data);
 683
 684   if (PREDICT_FALSE (!ctx->max_len_to_snd))
 685     return SESSION_TX_NO_DATA;
 686
 687   n_bufs_needed = ctx->n_segs_per_evt * ctx->n_bufs_per_seg;
 688   vec_validate_aligned (wrk->tx_buffers, n_bufs_needed - 1,
 689                         CLIB_CACHE_LINE_BYTES);
 690   n_bufs = vlib_buffer_alloc (vm, wrk->tx_buffers, n_bufs_needed);
 691   if (PREDICT_FALSE (n_bufs < n_bufs_needed))
 692     {
 693       if (n_bufs)
 694         vlib_buffer_free (vm, wrk->tx_buffers, n_bufs);
 695       session_evt_add_old (wrk, elt);
 696       vlib_node_increment_counter (wrk->vm, node->node_index,
 697                                    SESSION_QUEUE_ERROR_NO_BUFFER, 1);
 698       return SESSION_TX_NO_BUFFERS;
 699     }
 700
 701   /*
 702    * Write until we fill up a frame
 703    */
 704   vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
 705   if (PREDICT_FALSE (ctx->n_segs_per_evt > n_left_to_next))
 706     {
 707       ctx->n_segs_per_evt = n_left_to_next;
 708       ctx->max_len_to_snd = ctx->snd_mss * n_left_to_next;
 709     }
 710   ctx->left_to_snd = ctx->max_len_to_snd;
 711   n_left = ctx->n_segs_per_evt;
 712
 713   while (n_left >= 4)
 714     {
 715       vlib_buffer_t *b0, *b1;
 716       u32 bi0, bi1;
 717
 718       pbi = wrk->tx_buffers[n_bufs - 3];
 719       pb = vlib_get_buffer (vm, pbi);
 720       vlib_prefetch_buffer_header (pb, STORE);
 721       pbi = wrk->tx_buffers[n_bufs - 4];
 722       pb = vlib_get_buffer (vm, pbi);
 723       vlib_prefetch_buffer_header (pb, STORE);
 724
 725       to_next[0] = bi0 = wrk->tx_buffers[--n_bufs];
 726       to_next[1] = bi1 = wrk->tx_buffers[--n_bufs];
 727
 728       b0 = vlib_get_buffer (vm, bi0);
 729       b1 = vlib_get_buffer (vm, bi1);
 730
 731       session_tx_fill_buffer (vm, ctx, b0, &n_bufs, peek_data);
 732       session_tx_fill_buffer (vm, ctx, b1, &n_bufs, peek_data);
 733
 734       ctx->transport_vft->push_header (ctx->tc, b0);
 735       ctx->transport_vft->push_header (ctx->tc, b1);
 736
 737       to_next += 2;
 738       n_left_to_next -= 2;
 739       n_left -= 2;
 740
 741       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0);
 742       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b1);
 743
 744       vlib_validate_buffer_enqueue_x2 (vm, node, next_index, to_next,
 745                                        n_left_to_next, bi0, bi1, next0,
 746                                        next1);
 747     }
 748   while (n_left)
 749     {
 750       vlib_buffer_t *b0;
 751       u32 bi0;
 752
 753       if (n_left > 1)
 754         {
 755           pbi = wrk->tx_buffers[n_bufs - 2];
 756           pb = vlib_get_buffer (vm, pbi);
 757           vlib_prefetch_buffer_header (pb, STORE);
 758         }
 759
 760       to_next[0] = bi0 = wrk->tx_buffers[--n_bufs];
 761       b0 = vlib_get_buffer (vm, bi0);
 762       session_tx_fill_buffer (vm, ctx, b0, &n_bufs, peek_data);
 763
 764       /* Ask transport to push header after current_length and
 765        * total_length_not_including_first_buffer are updated */
 766       ctx->transport_vft->push_header (ctx->tc, b0);
 767
 768       to_next += 1;
 769       n_left_to_next -= 1;
 770       n_left -= 1;
 771
 772       VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0);
 773
 774       vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
 775                                        n_left_to_next, bi0, next0);
 776     }
 777
 778   if (PREDICT_FALSE ((n_trace = vlib_get_trace_count (vm, node)) > 0))
 779     session_tx_trace_frame (vm, node, next_index, to_next,
 780                             ctx->n_segs_per_evt, ctx->s, n_trace);
 781
 782   if (PREDICT_FALSE (n_bufs))
 783     vlib_buffer_free (vm, wrk->tx_buffers, n_bufs);
 784
 785   *n_tx_packets += ctx->n_segs_per_evt;
 786   transport_connection_update_tx_stats (ctx->tc, ctx->max_len_to_snd);
 787   vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 788
 789   SESSION_EVT (SESSION_EVT_DEQ, ctx->s, ctx->max_len_to_snd, ctx->max_dequeue,
 790                ctx->s->tx_fifo->has_event, wrk->last_vlib_time);
 791
 792   /* If we couldn't dequeue all bytes mark as partially read */
 793   ASSERT (ctx->left_to_snd == 0);
 794   if (ctx->max_len_to_snd < ctx->max_dequeue)
 795     if (svm_fifo_set_event (ctx->s->tx_fifo))
 796       session_evt_add_old (wrk, elt);
 797
 798   if (!peek_data
 799       && ctx->transport_vft->transport_options.tx_type == TRANSPORT_TX_DGRAM)
 800     {
 801       /* Fix dgram pre header */
 802       if (ctx->max_len_to_snd < ctx->max_dequeue)
 803         svm_fifo_overwrite_head (ctx->s->tx_fifo, (u8 *) & ctx->hdr,
 804                                  sizeof (session_dgram_pre_hdr_t));
 805       /* More data needs to be read */
 806       else if (svm_fifo_max_dequeue_cons (ctx->s->tx_fifo) > 0)
 807         if (svm_fifo_set_event (ctx->s->tx_fifo))
 808           session_evt_add_old (wrk, elt);
 809
 810       if (svm_fifo_needs_deq_ntf (ctx->s->tx_fifo, ctx->max_len_to_snd))
 811         session_dequeue_notify (ctx->s);
 812     }
 813   return SESSION_TX_OK;
 814 }
 815
 816 int
 817 session_tx_fifo_peek_and_snd (session_worker_t * wrk,
 818                               vlib_node_runtime_t * node,
 819                               session_evt_elt_t * e, int *n_tx_packets)
 820 {
 821   return session_tx_fifo_read_and_snd_i (wrk, node, e, n_tx_packets, 1);
 822 }
 823
 824 int
 825 session_tx_fifo_dequeue_and_snd (session_worker_t * wrk,
 826                                  vlib_node_runtime_t * node,
 827                                  session_evt_elt_t * e, int *n_tx_packets)
 828 {
 829   return session_tx_fifo_read_and_snd_i (wrk, node, e, n_tx_packets, 0);
 830 }
 831
 832 int
 833 session_tx_fifo_dequeue_internal (session_worker_t * wrk,
 834                                   vlib_node_runtime_t * node,
 835                                   session_evt_elt_t * e, int *n_tx_packets)
 836 {
 837   session_t *s = wrk->ctx.s;
 838
 839   if (PREDICT_FALSE (s->session_state >= SESSION_STATE_TRANSPORT_CLOSED))
 840     return 0;
 841   svm_fifo_unset_event (s->tx_fifo);
 842   return transport_custom_tx (session_get_transport_proto (s), s,
 843                               VLIB_FRAME_SIZE - *n_tx_packets);
 844 }
 845
 846 always_inline session_t *
 847 session_event_get_session (session_event_t * e, u8 thread_index)
 848 {
 849   return session_get_if_valid (e->session_index, thread_index);
 850 }
 851
 852 always_inline void
 853 session_event_dispatch (session_worker_t * wrk, vlib_node_runtime_t * node,
 854                         session_evt_elt_t * elt, u32 thread_index,
 855                         int *n_tx_packets)
 856 {
 857   session_main_t *smm = &session_main;
 858   app_worker_t *app_wrk;
 859   clib_llist_index_t ei;
 860   void (*fp) (void *);
 861   session_event_t *e;
 862   session_t *s;
 863
 864   ei = clib_llist_entry_index (wrk->event_elts, elt);
 865   e = &elt->evt;
 866
 867   switch (e->event_type)
 868     {
 869     case SESSION_IO_EVT_TX_FLUSH:
 870     case SESSION_IO_EVT_TX:
 871       s = session_event_get_session (e, thread_index);
 872       if (PREDICT_FALSE (!s))
 873         {
 874           clib_warning ("session %u was freed!", e->session_index);
 875           break;
 876         }
 877       CLIB_PREFETCH (s->tx_fifo, 2 * CLIB_CACHE_LINE_BYTES, LOAD);
 878       wrk->ctx.s = s;
 879       /* Spray packets in per session type frames, since they go to
 880        * different nodes */
 881       (smm->session_tx_fns[s->session_type]) (wrk, node, elt, n_tx_packets);
 882       break;
 883     case SESSION_IO_EVT_RX:
 884       s = session_event_get_session (e, thread_index);
 885       if (!s)
 886         break;
 887       transport_app_rx_evt (session_get_transport_proto (s),
 888                             s->connection_index, s->thread_index);
 889       break;
 890     case SESSION_CTRL_EVT_CLOSE:
 891       s = session_get_from_handle_if_valid (e->session_handle);
 892       if (PREDICT_FALSE (!s))
 893         break;
 894       session_transport_close (s);
 895       break;
 896     case SESSION_IO_EVT_BUILTIN_RX:
 897       s = session_event_get_session (e, thread_index);
 898       if (PREDICT_FALSE (!s || s->session_state >= SESSION_STATE_CLOSING))
 899         break;
 900       svm_fifo_unset_event (s->rx_fifo);
 901       app_wrk = app_worker_get (s->app_wrk_index);
 902       app_worker_builtin_rx (app_wrk, s);
 903       break;
 904     case SESSION_IO_EVT_BUILTIN_TX:
 905       s = session_get_from_handle_if_valid (e->session_handle);
 906       wrk->ctx.s = s;
 907       if (PREDICT_TRUE (s != 0))
 908         session_tx_fifo_dequeue_internal (wrk, node, elt, n_tx_packets);
 909       break;
 910     case SESSION_CTRL_EVT_RPC:
 911       fp = e->rpc_args.fp;
 912       (*fp) (e->rpc_args.arg);
 913       break;
 914     case SESSION_CTRL_EVT_DISCONNECTED:
 915       session_mq_disconnected_handler (e->data);
 916       break;
 917     case SESSION_CTRL_EVT_ACCEPTED_REPLY:
 918       session_mq_accepted_reply_handler (e->data);
 919       break;
 920     case SESSION_CTRL_EVT_CONNECTED_REPLY:
 921       break;
 922     case SESSION_CTRL_EVT_DISCONNECTED_REPLY:
 923       session_mq_disconnected_reply_handler (e->data);
 924       break;
 925     case SESSION_CTRL_EVT_RESET_REPLY:
 926       session_mq_reset_reply_handler (e->data);
 927       break;
 928     case SESSION_CTRL_EVT_WORKER_UPDATE:
 929       session_mq_worker_update_handler (e->data);
 930       break;
 931     default:
 932       clib_warning ("unhandled event type %d", e->event_type);
 933     }
 934
 935   /* Regrab elements in case pool moved */
 936   elt = pool_elt_at_index (wrk->event_elts, ei);
 937   if (!clib_llist_elt_is_linked (elt, evt_list))
 938     session_evt_elt_free (wrk, elt);
 939 }
 940
 941 static uword
 942 session_queue_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node,
 943                        vlib_frame_t * frame)
 944 {
 945   session_main_t *smm = vnet_get_session_main ();
 946   u32 thread_index = vm->thread_index, n_to_dequeue;
 947   session_worker_t *wrk = &smm->wrk[thread_index];
 948   session_evt_elt_t *elt, *ctrl_he, *new_he, *old_he;
 949   svm_msg_q_msg_t _msg, *msg = &_msg;
 950   clib_llist_index_t old_ti;
 951   int i, n_tx_packets = 0;
 952   session_event_t *evt;
 953   svm_msg_q_t *mq;
 954
 955   SESSION_EVT (SESSION_EVT_DISPATCH_START, wrk);
 956
 957   wrk->last_vlib_time = vlib_time_now (vm);
 958
 959   /*
 960    *  Update transport time
 961    */
 962   transport_update_time (wrk->last_vlib_time, thread_index);
 963
 964   /*
 965    *  Dequeue and handle new events
 966    */
 967
 968   /* Try to dequeue what is available. Don't wait for lock.
 969    * XXX: we may need priorities here */
 970   mq = wrk->vpp_event_queue;
 971   n_to_dequeue = svm_msg_q_size (mq);
 972   if (n_to_dequeue && svm_msg_q_try_lock (mq) == 0)
 973     {
 974       for (i = 0; i < n_to_dequeue; i++)
 975         {
 976           svm_msg_q_sub_w_lock (mq, msg);
 977           evt = svm_msg_q_msg_data (mq, msg);
 978           if (evt->event_type > SESSION_IO_EVT_BUILTIN_TX)
 979             elt = session_evt_alloc_ctrl (wrk);
 980           else
 981             elt = session_evt_alloc_new (wrk);
 982           /* Works because reply messages are smaller than a session evt.
 983            * If we ever need to support bigger messages this needs to be
 984            * fixed */
 985           clib_memcpy_fast (&elt->evt, evt, sizeof (elt->evt));
 986           svm_msg_q_free_msg (mq, msg);
 987         }
 988       svm_msg_q_unlock (mq);
 989     }
 990
 991   /*
 992    * Handle control events
 993    */
 994
 995   ctrl_he = pool_elt_at_index (wrk->event_elts, wrk->ctrl_head);
 996
 997   /* *INDENT-OFF* */
 998   clib_llist_foreach_safe (wrk->event_elts, evt_list, ctrl_he, elt, ({
 999     clib_llist_remove (wrk->event_elts, evt_list, elt);
1000     session_event_dispatch (wrk, node, elt, thread_index, &n_tx_packets);
1001   }));
1002   /* *INDENT-ON* */
1003
1004   /*
1005    * Handle the new io events.
1006    */
1007
1008   new_he = pool_elt_at_index (wrk->event_elts, wrk->new_head);
1009
1010   /* *INDENT-OFF* */
1011   clib_llist_foreach_safe (wrk->event_elts, evt_list, new_he, elt, ({
1012     session_evt_type_t et;
1013
1014     et = elt->evt.event_type;
1015     clib_llist_remove (wrk->event_elts, evt_list, elt);
1016
1017     /* Postpone tx events if we can't handle them this dispatch cycle */
1018     if (n_tx_packets >= VLIB_FRAME_SIZE
1019         && (et == SESSION_IO_EVT_TX || et == SESSION_IO_EVT_TX_FLUSH))
1020       {
1021         clib_llist_add (wrk->event_elts, evt_list, elt, new_he);
1022         continue;
1023       }
1024
1025     session_event_dispatch (wrk, node, elt, thread_index, &n_tx_packets);
1026   }));
1027   /* *INDENT-ON* */
1028
1029   /*
1030    * Handle the old io events
1031    */
1032
1033   old_he = pool_elt_at_index (wrk->event_elts, wrk->old_head);
1034   old_ti = clib_llist_prev_index (old_he, evt_list);
1035
1036   while (!clib_llist_is_empty (wrk->event_elts, evt_list, old_he))
1037     {
1038       clib_llist_index_t ei;
1039
1040       clib_llist_pop_first (wrk->event_elts, evt_list, elt, old_he);
1041       ei = clib_llist_entry_index (wrk->event_elts, elt);
1042       session_event_dispatch (wrk, node, elt, thread_index, &n_tx_packets);
1043
1044       old_he = pool_elt_at_index (wrk->event_elts, wrk->old_head);
1045       if (n_tx_packets >= VLIB_FRAME_SIZE || ei == old_ti)
1046         break;
1047     };
1048
1049   vlib_node_increment_counter (vm, session_queue_node.index,
1050                                SESSION_QUEUE_ERROR_TX, n_tx_packets);
1051
1052   SESSION_EVT (SESSION_EVT_DISPATCH_END, wrk, n_tx_packets);
1053
1054   return n_tx_packets;
1055 }
1056
1057 /* *INDENT-OFF* */
1058 VLIB_REGISTER_NODE (session_queue_node) =
1059 {
1060   .function = session_queue_node_fn,
1061   .name = "session-queue",
1062   .format_trace = format_session_queue_trace,
1063   .type = VLIB_NODE_TYPE_INPUT,
1064   .n_errors = ARRAY_LEN (session_queue_error_strings),
1065   .error_strings = session_queue_error_strings,
1066   .state = VLIB_NODE_STATE_DISABLED,
1067 };
1068 /* *INDENT-ON* */
1069
1070 void
1071 dump_thread_0_event_queue (void)
1072 {
1073   session_main_t *smm = vnet_get_session_main ();
1074   vlib_main_t *vm = &vlib_global_main;
1075   u32 my_thread_index = vm->thread_index;
1076   session_event_t _e, *e = &_e;
1077   svm_msg_q_ring_t *ring;
1078   session_t *s0;
1079   svm_msg_q_msg_t *msg;
1080   svm_msg_q_t *mq;
1081   int i, index;
1082
1083   mq = smm->wrk[my_thread_index].vpp_event_queue;
1084   index = mq->q->head;
1085
1086   for (i = 0; i < mq->q->cursize; i++)
1087     {
1088       msg = (svm_msg_q_msg_t *) (&mq->q->data[0] + mq->q->elsize * index);
1089       ring = svm_msg_q_ring (mq, msg->ring_index);
1090       clib_memcpy_fast (e, svm_msg_q_msg_data (mq, msg), ring->elsize);
1091
1092       switch (e->event_type)
1093         {
1094         case SESSION_IO_EVT_TX:
1095           s0 = session_event_get_session (e, my_thread_index);
1096           fformat (stdout, "[%04d] TX session %d\n", i, s0->session_index);
1097           break;
1098
1099         case SESSION_CTRL_EVT_CLOSE:
1100           s0 = session_get_from_handle (e->session_handle);
1101           fformat (stdout, "[%04d] disconnect session %d\n", i,
1102                    s0->session_index);
1103           break;
1104
1105         case SESSION_IO_EVT_BUILTIN_RX:
1106           s0 = session_event_get_session (e, my_thread_index);
1107           fformat (stdout, "[%04d] builtin_rx %d\n", i, s0->session_index);
1108           break;
1109
1110         case SESSION_CTRL_EVT_RPC:
1111           fformat (stdout, "[%04d] RPC call %llx with %llx\n",
1112                    i, (u64) (uword) (e->rpc_args.fp),
1113                    (u64) (uword) (e->rpc_args.arg));
1114           break;
1115
1116         default:
1117           fformat (stdout, "[%04d] unhandled event type %d\n",
1118                    i, e->event_type);
1119           break;
1120         }
1121
1122       index++;
1123
1124       if (index == mq->q->maxsize)
1125         index = 0;
1126     }
1127 }
1128
1129 static u8
1130 session_node_cmp_event (session_event_t * e, svm_fifo_t * f)
1131 {
1132   session_t *s;
1133   switch (e->event_type)
1134     {
1135     case SESSION_IO_EVT_RX:
1136     case SESSION_IO_EVT_TX:
1137     case SESSION_IO_EVT_BUILTIN_RX:
1138       if (e->session_index == f->master_session_index)
1139         return 1;
1140       break;
1141     case SESSION_CTRL_EVT_CLOSE:
1142       break;
1143     case SESSION_CTRL_EVT_RPC:
1144       s = session_get_from_handle (e->session_handle);
1145       if (!s)
1146         {
1147           clib_warning ("session has event but doesn't exist!");
1148           break;
1149         }
1150       if (s->rx_fifo == f || s->tx_fifo == f)
1151         return 1;
1152       break;
1153     default:
1154       break;
1155     }
1156   return 0;
1157 }
1158
1159 u8
1160 session_node_lookup_fifo_event (svm_fifo_t * f, session_event_t * e)
1161 {
1162   session_evt_elt_t *elt;
1163   session_worker_t *wrk;
1164   int i, index, found = 0;
1165   svm_msg_q_msg_t *msg;
1166   svm_msg_q_ring_t *ring;
1167   svm_msg_q_t *mq;
1168   u8 thread_index;
1169
1170   ASSERT (e);
1171   thread_index = f->master_thread_index;
1172   wrk = session_main_get_worker (thread_index);
1173
1174   /*
1175    * Search evt queue
1176    */
1177   mq = wrk->vpp_event_queue;
1178   index = mq->q->head;
1179   for (i = 0; i < mq->q->cursize; i++)
1180     {
1181       msg = (svm_msg_q_msg_t *) (&mq->q->data[0] + mq->q->elsize * index);
1182       ring = svm_msg_q_ring (mq, msg->ring_index);
1183       clib_memcpy_fast (e, svm_msg_q_msg_data (mq, msg), ring->elsize);
1184       found = session_node_cmp_event (e, f);
1185       if (found)
1186         return 1;
1187       if (++index == mq->q->maxsize)
1188         index = 0;
1189     }
1190   /*
1191    * Search pending events vector
1192    */
1193
1194   /* *INDENT-OFF* */
1195   clib_llist_foreach (wrk->event_elts, evt_list,
1196                       pool_elt_at_index (wrk->event_elts, wrk->old_head),
1197                       elt, ({
1198     found = session_node_cmp_event (&elt->evt, f);
1199     if (found)
1200       {
1201         clib_memcpy_fast (e, &elt->evt, sizeof (*e));
1202         break;
1203       }
1204
1205   }));
1206   /* *INDENT-ON* */
1207
1208   return found;
1209 }
1210
1211 static clib_error_t *
1212 session_queue_exit (vlib_main_t * vm)
1213 {
1214   if (vec_len (vlib_mains) < 2)
1215     return 0;
1216
1217   /*
1218    * Shut off (especially) worker-thread session nodes.
1219    * Otherwise, vpp can crash as the main thread unmaps the
1220    * API segment.
1221    */
1222   vlib_worker_thread_barrier_sync (vm);
1223   session_node_enable_disable (0 /* is_enable */ );
1224   vlib_worker_thread_barrier_release (vm);
1225   return 0;
1226 }
1227
1228 VLIB_MAIN_LOOP_EXIT_FUNCTION (session_queue_exit);
1229
1230 static uword
1231 session_queue_process (vlib_main_t * vm, vlib_node_runtime_t * rt,
1232                        vlib_frame_t * f)
1233 {
1234   f64 now, timeout = 1.0;
1235   uword *event_data = 0;
1236   uword event_type;
1237
1238   while (1)
1239     {
1240       vlib_process_wait_for_event_or_clock (vm, timeout);
1241       now = vlib_time_now (vm);
1242       event_type = vlib_process_get_events (vm, (uword **) & event_data);
1243
1244       switch (event_type)
1245         {
1246         case SESSION_Q_PROCESS_FLUSH_FRAMES:
1247           /* Flush the frames by updating all transports times */
1248           transport_update_time (now, 0);
1249           break;
1250         case SESSION_Q_PROCESS_STOP:
1251           timeout = 100000.0;
1252           break;
1253         case ~0:
1254           /* Timed out. Update time for all transports to trigger all
1255            * outstanding retransmits. */
1256           transport_update_time (now, 0);
1257           break;
1258         }
1259       vec_reset_length (event_data);
1260     }
1261   return 0;
1262 }
1263
1264 /* *INDENT-OFF* */
1265 VLIB_REGISTER_NODE (session_queue_process_node) =
1266 {
1267   .function = session_queue_process,
1268   .type = VLIB_NODE_TYPE_PROCESS,
1269   .name = "session-queue-process",
1270   .state = VLIB_NODE_STATE_DISABLED,
1271 };
1272 /* *INDENT-ON* */
1273
1274 static_always_inline uword
1275 session_queue_pre_input_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
1276                                 vlib_frame_t * frame)
1277 {
1278   session_main_t *sm = &session_main;
1279   if (!sm->wrk[0].vpp_event_queue)
1280     return 0;
1281   return session_queue_node_fn (vm, node, frame);
1282 }
1283
1284 /* *INDENT-OFF* */
1285 VLIB_REGISTER_NODE (session_queue_pre_input_node) =
1286 {
1287   .function = session_queue_pre_input_inline,
1288   .type = VLIB_NODE_TYPE_PRE_INPUT,
1289   .name = "session-queue-main",
1290   .state = VLIB_NODE_STATE_DISABLED,
1291 };
1292 /* *INDENT-ON* */
1293
1294 /*
1295  * fd.io coding-style-patch-verification: ON
1296  *
1297  * Local Variables:
1298  * eval: (c-set-style "gnu")
1299  * End:
1300  */