session tcp: handle rxt and acks as custom events
[vpp.git] / src / vnet / session / session.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  * @file
17  * @brief Session and session manager
18  */
19
20 #include <vnet/session/session.h>
21 #include <vnet/session/session_debug.h>
22 #include <vnet/session/application.h>
23 #include <vnet/dpo/load_balance.h>
24 #include <vnet/fib/ip4_fib.h>
25
26 session_main_t session_main;
27
28 static inline int
29 session_send_evt_to_thread (void *data, void *args, u32 thread_index,
30                             session_evt_type_t evt_type)
31 {
32   session_event_t *evt;
33   svm_msg_q_msg_t msg;
34   svm_msg_q_t *mq;
35
36   mq = session_main_get_vpp_event_queue (thread_index);
37   if (PREDICT_FALSE (svm_msg_q_lock (mq)))
38     return -1;
39   if (PREDICT_FALSE (svm_msg_q_ring_is_full (mq, SESSION_MQ_IO_EVT_RING)))
40     {
41       svm_msg_q_unlock (mq);
42       return -2;
43     }
44   msg = svm_msg_q_alloc_msg_w_ring (mq, SESSION_MQ_IO_EVT_RING);
45   if (PREDICT_FALSE (svm_msg_q_msg_is_invalid (&msg)))
46     {
47       svm_msg_q_unlock (mq);
48       return -2;
49     }
50   evt = (session_event_t *) svm_msg_q_msg_data (mq, &msg);
51   evt->event_type = evt_type;
52   switch (evt_type)
53     {
54     case SESSION_CTRL_EVT_RPC:
55       evt->rpc_args.fp = data;
56       evt->rpc_args.arg = args;
57       break;
58     case SESSION_IO_EVT_RX:
59     case SESSION_IO_EVT_TX:
60     case SESSION_IO_EVT_TX_FLUSH:
61     case SESSION_IO_EVT_BUILTIN_RX:
62       evt->session_index = *(u32 *) data;
63       break;
64     case SESSION_IO_EVT_BUILTIN_TX:
65     case SESSION_CTRL_EVT_CLOSE:
66       evt->session_handle = session_handle ((session_t *) data);
67       break;
68     default:
69       clib_warning ("evt unhandled!");
70       svm_msg_q_unlock (mq);
71       return -1;
72     }
73
74   svm_msg_q_add_and_unlock (mq, &msg);
75   return 0;
76 }
77
78 int
79 session_send_io_evt_to_thread (svm_fifo_t * f, session_evt_type_t evt_type)
80 {
81   return session_send_evt_to_thread (&f->master_session_index, 0,
82                                      f->master_thread_index, evt_type);
83 }
84
85 int
86 session_send_io_evt_to_thread_custom (void *data, u32 thread_index,
87                                       session_evt_type_t evt_type)
88 {
89   return session_send_evt_to_thread (data, 0, thread_index, evt_type);
90 }
91
92 int
93 session_send_ctrl_evt_to_thread (session_t * s, session_evt_type_t evt_type)
94 {
95   /* only event supported for now is disconnect */
96   ASSERT (evt_type == SESSION_CTRL_EVT_CLOSE);
97   return session_send_evt_to_thread (s, 0, s->thread_index,
98                                      SESSION_CTRL_EVT_CLOSE);
99 }
100
101 void
102 session_send_rpc_evt_to_thread_force (u32 thread_index, void *fp,
103                                       void *rpc_args)
104 {
105   session_send_evt_to_thread (fp, rpc_args, thread_index,
106                               SESSION_CTRL_EVT_RPC);
107 }
108
109 void
110 session_send_rpc_evt_to_thread (u32 thread_index, void *fp, void *rpc_args)
111 {
112   if (thread_index != vlib_get_thread_index ())
113     session_send_rpc_evt_to_thread_force (thread_index, fp, rpc_args);
114   else
115     {
116       void (*fnp) (void *) = fp;
117       fnp (rpc_args);
118     }
119 }
120
121 void
122 session_add_self_custom_tx_evt (transport_connection_t * tc, u8 has_prio)
123 {
124   session_t *s;
125
126   s = session_get (tc->s_index, tc->thread_index);
127   ASSERT (s->thread_index == vlib_get_thread_index ());
128   if (!(s->flags & SESSION_F_CUSTOM_TX))
129     {
130       s->flags |= SESSION_F_CUSTOM_TX;
131       if (svm_fifo_set_event (s->tx_fifo))
132         {
133           session_worker_t *wrk;
134           session_evt_elt_t *elt;
135           wrk = session_main_get_worker (tc->thread_index);
136           if (has_prio)
137             elt = session_evt_alloc_new (wrk);
138           else
139             elt = session_evt_alloc_old (wrk);
140           elt->evt.session_index = tc->s_index;
141           elt->evt.event_type = SESSION_IO_EVT_TX;
142         }
143     }
144 }
145
146 static void
147 session_program_transport_close (session_t * s)
148 {
149   u32 thread_index = vlib_get_thread_index ();
150   session_evt_elt_t *elt;
151   session_worker_t *wrk;
152
153   /* If we are in the handler thread, or being called with the worker barrier
154    * held, just append a new event to pending disconnects vector. */
155   if (vlib_thread_is_main_w_barrier () || thread_index == s->thread_index)
156     {
157       wrk = session_main_get_worker (s->thread_index);
158       elt = session_evt_elt_alloc (wrk);
159       clib_memset (&elt->evt, 0, sizeof (session_event_t));
160       elt->evt.session_handle = session_handle (s);
161       elt->evt.event_type = SESSION_CTRL_EVT_CLOSE;
162       session_evt_add_pending_disconnects (wrk, elt);
163     }
164   else
165     session_send_ctrl_evt_to_thread (s, SESSION_CTRL_EVT_CLOSE);
166 }
167
168 session_t *
169 session_alloc (u32 thread_index)
170 {
171   session_worker_t *wrk = &session_main.wrk[thread_index];
172   session_t *s;
173   u8 will_expand = 0;
174   pool_get_aligned_will_expand (wrk->sessions, will_expand,
175                                 CLIB_CACHE_LINE_BYTES);
176   /* If we have peekers, let them finish */
177   if (PREDICT_FALSE (will_expand && vlib_num_workers ()))
178     {
179       clib_rwlock_writer_lock (&wrk->peekers_rw_locks);
180       pool_get_aligned (wrk->sessions, s, CLIB_CACHE_LINE_BYTES);
181       clib_rwlock_writer_unlock (&wrk->peekers_rw_locks);
182     }
183   else
184     {
185       pool_get_aligned (wrk->sessions, s, CLIB_CACHE_LINE_BYTES);
186     }
187   clib_memset (s, 0, sizeof (*s));
188   s->session_index = s - wrk->sessions;
189   s->thread_index = thread_index;
190   s->app_index = APP_INVALID_INDEX;
191   return s;
192 }
193
194 void
195 session_free (session_t * s)
196 {
197   if (CLIB_DEBUG)
198     {
199       u8 thread_index = s->thread_index;
200       clib_memset (s, 0xFA, sizeof (*s));
201       pool_put (session_main.wrk[thread_index].sessions, s);
202       return;
203     }
204   SESSION_EVT (SESSION_EVT_FREE, s);
205   pool_put (session_main.wrk[s->thread_index].sessions, s);
206 }
207
208 static void
209 session_cleanup_notify (session_t * s, session_cleanup_ntf_t ntf)
210 {
211   app_worker_t *app_wrk;
212
213   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
214   if (!app_wrk)
215     return;
216   app_worker_cleanup_notify (app_wrk, s, ntf);
217 }
218
219 void
220 session_free_w_fifos (session_t * s)
221 {
222   session_cleanup_notify (s, SESSION_CLEANUP_SESSION);
223   segment_manager_dealloc_fifos (s->rx_fifo, s->tx_fifo);
224   session_free (s);
225 }
226
227 /**
228  * Cleans up session and lookup table.
229  *
230  * Transport connection must still be valid.
231  */
232 static void
233 session_delete (session_t * s)
234 {
235   int rv;
236
237   /* Delete from the main lookup table. */
238   if ((rv = session_lookup_del_session (s)))
239     clib_warning ("hash delete error, rv %d", rv);
240
241   session_free_w_fifos (s);
242 }
243
244 static session_t *
245 session_alloc_for_connection (transport_connection_t * tc)
246 {
247   session_t *s;
248   u32 thread_index = tc->thread_index;
249
250   ASSERT (thread_index == vlib_get_thread_index ()
251           || transport_protocol_is_cl (tc->proto));
252
253   s = session_alloc (thread_index);
254   s->session_type = session_type_from_proto_and_ip (tc->proto, tc->is_ip4);
255   s->session_state = SESSION_STATE_CLOSED;
256
257   /* Attach transport to session and vice versa */
258   s->connection_index = tc->c_index;
259   tc->s_index = s->session_index;
260   return s;
261 }
262
263 /**
264  * Discards bytes from buffer chain
265  *
266  * It discards n_bytes_to_drop starting at first buffer after chain_b
267  */
268 always_inline void
269 session_enqueue_discard_chain_bytes (vlib_main_t * vm, vlib_buffer_t * b,
270                                      vlib_buffer_t ** chain_b,
271                                      u32 n_bytes_to_drop)
272 {
273   vlib_buffer_t *next = *chain_b;
274   u32 to_drop = n_bytes_to_drop;
275   ASSERT (b->flags & VLIB_BUFFER_NEXT_PRESENT);
276   while (to_drop && (next->flags & VLIB_BUFFER_NEXT_PRESENT))
277     {
278       next = vlib_get_buffer (vm, next->next_buffer);
279       if (next->current_length > to_drop)
280         {
281           vlib_buffer_advance (next, to_drop);
282           to_drop = 0;
283         }
284       else
285         {
286           to_drop -= next->current_length;
287           next->current_length = 0;
288         }
289     }
290   *chain_b = next;
291
292   if (to_drop == 0)
293     b->total_length_not_including_first_buffer -= n_bytes_to_drop;
294 }
295
296 /**
297  * Enqueue buffer chain tail
298  */
299 always_inline int
300 session_enqueue_chain_tail (session_t * s, vlib_buffer_t * b,
301                             u32 offset, u8 is_in_order)
302 {
303   vlib_buffer_t *chain_b;
304   u32 chain_bi, len, diff;
305   vlib_main_t *vm = vlib_get_main ();
306   u8 *data;
307   u32 written = 0;
308   int rv = 0;
309
310   if (is_in_order && offset)
311     {
312       diff = offset - b->current_length;
313       if (diff > b->total_length_not_including_first_buffer)
314         return 0;
315       chain_b = b;
316       session_enqueue_discard_chain_bytes (vm, b, &chain_b, diff);
317       chain_bi = vlib_get_buffer_index (vm, chain_b);
318     }
319   else
320     chain_bi = b->next_buffer;
321
322   do
323     {
324       chain_b = vlib_get_buffer (vm, chain_bi);
325       data = vlib_buffer_get_current (chain_b);
326       len = chain_b->current_length;
327       if (!len)
328         continue;
329       if (is_in_order)
330         {
331           rv = svm_fifo_enqueue (s->rx_fifo, len, data);
332           if (rv == len)
333             {
334               written += rv;
335             }
336           else if (rv < len)
337             {
338               return (rv > 0) ? (written + rv) : written;
339             }
340           else if (rv > len)
341             {
342               written += rv;
343
344               /* written more than what was left in chain */
345               if (written > b->total_length_not_including_first_buffer)
346                 return written;
347
348               /* drop the bytes that have already been delivered */
349               session_enqueue_discard_chain_bytes (vm, b, &chain_b, rv - len);
350             }
351         }
352       else
353         {
354           rv = svm_fifo_enqueue_with_offset (s->rx_fifo, offset, len, data);
355           if (rv)
356             {
357               clib_warning ("failed to enqueue multi-buffer seg");
358               return -1;
359             }
360           offset += len;
361         }
362     }
363   while ((chain_bi = (chain_b->flags & VLIB_BUFFER_NEXT_PRESENT)
364           ? chain_b->next_buffer : 0));
365
366   if (is_in_order)
367     return written;
368
369   return 0;
370 }
371
372 /*
373  * Enqueue data for delivery to session peer. Does not notify peer of enqueue
374  * event but on request can queue notification events for later delivery by
375  * calling stream_server_flush_enqueue_events().
376  *
377  * @param tc Transport connection which is to be enqueued data
378  * @param b Buffer to be enqueued
379  * @param offset Offset at which to start enqueueing if out-of-order
380  * @param queue_event Flag to indicate if peer is to be notified or if event
381  *                    is to be queued. The former is useful when more data is
382  *                    enqueued and only one event is to be generated.
383  * @param is_in_order Flag to indicate if data is in order
384  * @return Number of bytes enqueued or a negative value if enqueueing failed.
385  */
386 int
387 session_enqueue_stream_connection (transport_connection_t * tc,
388                                    vlib_buffer_t * b, u32 offset,
389                                    u8 queue_event, u8 is_in_order)
390 {
391   session_t *s;
392   int enqueued = 0, rv, in_order_off;
393
394   s = session_get (tc->s_index, tc->thread_index);
395
396   if (is_in_order)
397     {
398       enqueued = svm_fifo_enqueue (s->rx_fifo,
399                                    b->current_length,
400                                    vlib_buffer_get_current (b));
401       if (PREDICT_FALSE ((b->flags & VLIB_BUFFER_NEXT_PRESENT)
402                          && enqueued >= 0))
403         {
404           in_order_off = enqueued > b->current_length ? enqueued : 0;
405           rv = session_enqueue_chain_tail (s, b, in_order_off, 1);
406           if (rv > 0)
407             enqueued += rv;
408         }
409     }
410   else
411     {
412       rv = svm_fifo_enqueue_with_offset (s->rx_fifo, offset,
413                                          b->current_length,
414                                          vlib_buffer_get_current (b));
415       if (PREDICT_FALSE ((b->flags & VLIB_BUFFER_NEXT_PRESENT) && !rv))
416         session_enqueue_chain_tail (s, b, offset + b->current_length, 0);
417       /* if something was enqueued, report even this as success for ooo
418        * segment handling */
419       return rv;
420     }
421
422   if (queue_event)
423     {
424       /* Queue RX event on this fifo. Eventually these will need to be flushed
425        * by calling stream_server_flush_enqueue_events () */
426       session_worker_t *wrk;
427
428       wrk = session_main_get_worker (s->thread_index);
429       if (!(s->flags & SESSION_F_RX_EVT))
430         {
431           s->flags |= SESSION_F_RX_EVT;
432           vec_add1 (wrk->session_to_enqueue[tc->proto], s->session_index);
433         }
434     }
435
436   return enqueued;
437 }
438
439 int
440 session_enqueue_dgram_connection (session_t * s,
441                                   session_dgram_hdr_t * hdr,
442                                   vlib_buffer_t * b, u8 proto, u8 queue_event)
443 {
444   int enqueued = 0, rv, in_order_off;
445
446   ASSERT (svm_fifo_max_enqueue_prod (s->rx_fifo)
447           >= b->current_length + sizeof (*hdr));
448
449   svm_fifo_enqueue (s->rx_fifo, sizeof (session_dgram_hdr_t), (u8 *) hdr);
450   enqueued = svm_fifo_enqueue (s->rx_fifo, b->current_length,
451                                vlib_buffer_get_current (b));
452   if (PREDICT_FALSE ((b->flags & VLIB_BUFFER_NEXT_PRESENT) && enqueued >= 0))
453     {
454       in_order_off = enqueued > b->current_length ? enqueued : 0;
455       rv = session_enqueue_chain_tail (s, b, in_order_off, 1);
456       if (rv > 0)
457         enqueued += rv;
458     }
459   if (queue_event)
460     {
461       /* Queue RX event on this fifo. Eventually these will need to be flushed
462        * by calling stream_server_flush_enqueue_events () */
463       session_worker_t *wrk;
464
465       wrk = session_main_get_worker (s->thread_index);
466       if (!(s->flags & SESSION_F_RX_EVT))
467         {
468           s->flags |= SESSION_F_RX_EVT;
469           vec_add1 (wrk->session_to_enqueue[proto], s->session_index);
470         }
471     }
472   return enqueued;
473 }
474
475 int
476 session_tx_fifo_peek_bytes (transport_connection_t * tc, u8 * buffer,
477                             u32 offset, u32 max_bytes)
478 {
479   session_t *s = session_get (tc->s_index, tc->thread_index);
480   return svm_fifo_peek (s->tx_fifo, offset, max_bytes, buffer);
481 }
482
483 u32
484 session_tx_fifo_dequeue_drop (transport_connection_t * tc, u32 max_bytes)
485 {
486   session_t *s = session_get (tc->s_index, tc->thread_index);
487   u32 rv;
488
489   rv = svm_fifo_dequeue_drop (s->tx_fifo, max_bytes);
490
491   if (svm_fifo_needs_deq_ntf (s->tx_fifo, max_bytes))
492     session_dequeue_notify (s);
493
494   return rv;
495 }
496
497 static inline int
498 session_notify_subscribers (u32 app_index, session_t * s,
499                             svm_fifo_t * f, session_evt_type_t evt_type)
500 {
501   app_worker_t *app_wrk;
502   application_t *app;
503   int i;
504
505   app = application_get (app_index);
506   if (!app)
507     return -1;
508
509   for (i = 0; i < f->n_subscribers; i++)
510     {
511       app_wrk = application_get_worker (app, f->subscribers[i]);
512       if (!app_wrk)
513         continue;
514       if (app_worker_lock_and_send_event (app_wrk, s, evt_type))
515         return -1;
516     }
517
518   return 0;
519 }
520
521 /**
522  * Notify session peer that new data has been enqueued.
523  *
524  * @param s     Stream session for which the event is to be generated.
525  * @param lock  Flag to indicate if call should lock message queue.
526  *
527  * @return 0 on success or negative number if failed to send notification.
528  */
529 static inline int
530 session_enqueue_notify_inline (session_t * s)
531 {
532   app_worker_t *app_wrk;
533   u32 session_index;
534   u8 n_subscribers;
535
536   session_index = s->session_index;
537   n_subscribers = svm_fifo_n_subscribers (s->rx_fifo);
538
539   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
540   if (PREDICT_FALSE (!app_wrk))
541     {
542       SESSION_DBG ("invalid s->app_index = %d", s->app_wrk_index);
543       return 0;
544     }
545
546   SESSION_EVT (SESSION_EVT_ENQ, s, svm_fifo_max_dequeue_prod (s->rx_fifo));
547
548   s->flags &= ~SESSION_F_RX_EVT;
549   if (PREDICT_FALSE (app_worker_lock_and_send_event (app_wrk, s,
550                                                      SESSION_IO_EVT_RX)))
551     return -1;
552
553   if (PREDICT_FALSE (n_subscribers))
554     {
555       s = session_get (session_index, vlib_get_thread_index ());
556       return session_notify_subscribers (app_wrk->app_index, s,
557                                          s->rx_fifo, SESSION_IO_EVT_RX);
558     }
559
560   return 0;
561 }
562
563 int
564 session_enqueue_notify (session_t * s)
565 {
566   return session_enqueue_notify_inline (s);
567 }
568
569 int
570 session_dequeue_notify (session_t * s)
571 {
572   app_worker_t *app_wrk;
573
574   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
575   if (PREDICT_FALSE (!app_wrk))
576     return -1;
577
578   if (PREDICT_FALSE (app_worker_lock_and_send_event (app_wrk, s,
579                                                      SESSION_IO_EVT_TX)))
580     return -1;
581
582   if (PREDICT_FALSE (s->tx_fifo->n_subscribers))
583     return session_notify_subscribers (app_wrk->app_index, s,
584                                        s->tx_fifo, SESSION_IO_EVT_TX);
585
586   svm_fifo_clear_deq_ntf (s->tx_fifo);
587
588   return 0;
589 }
590
591 /**
592  * Flushes queue of sessions that are to be notified of new data
593  * enqueued events.
594  *
595  * @param thread_index Thread index for which the flush is to be performed.
596  * @return 0 on success or a positive number indicating the number of
597  *         failures due to API queue being full.
598  */
599 int
600 session_main_flush_enqueue_events (u8 transport_proto, u32 thread_index)
601 {
602   session_worker_t *wrk = session_main_get_worker (thread_index);
603   session_t *s;
604   int i, errors = 0;
605   u32 *indices;
606
607   indices = wrk->session_to_enqueue[transport_proto];
608
609   for (i = 0; i < vec_len (indices); i++)
610     {
611       s = session_get_if_valid (indices[i], thread_index);
612       if (PREDICT_FALSE (!s))
613         {
614           errors++;
615           continue;
616         }
617
618       if (PREDICT_FALSE (session_enqueue_notify_inline (s)))
619         errors++;
620     }
621
622   vec_reset_length (indices);
623   wrk->session_to_enqueue[transport_proto] = indices;
624
625   return errors;
626 }
627
628 int
629 session_main_flush_all_enqueue_events (u8 transport_proto)
630 {
631   vlib_thread_main_t *vtm = vlib_get_thread_main ();
632   int i, errors = 0;
633   for (i = 0; i < 1 + vtm->n_threads; i++)
634     errors += session_main_flush_enqueue_events (transport_proto, i);
635   return errors;
636 }
637
638 static inline int
639 session_stream_connect_notify_inline (transport_connection_t * tc, u8 is_fail,
640                                       session_state_t opened_state)
641 {
642   u32 opaque = 0, new_ti, new_si;
643   app_worker_t *app_wrk;
644   session_t *s = 0;
645   u64 ho_handle;
646
647   /*
648    * Find connection handle and cleanup half-open table
649    */
650   ho_handle = session_lookup_half_open_handle (tc);
651   if (ho_handle == HALF_OPEN_LOOKUP_INVALID_VALUE)
652     {
653       SESSION_DBG ("half-open was removed!");
654       return -1;
655     }
656   session_lookup_del_half_open (tc);
657
658   /* Get the app's index from the handle we stored when opening connection
659    * and the opaque (api_context for external apps) from transport session
660    * index */
661   app_wrk = app_worker_get_if_valid (ho_handle >> 32);
662   if (!app_wrk)
663     return -1;
664
665   opaque = tc->s_index;
666
667   if (is_fail)
668     return app_worker_connect_notify (app_wrk, s, opaque);
669
670   s = session_alloc_for_connection (tc);
671   s->session_state = SESSION_STATE_CONNECTING;
672   s->app_wrk_index = app_wrk->wrk_index;
673   new_si = s->session_index;
674   new_ti = s->thread_index;
675
676   if (app_worker_init_connected (app_wrk, s))
677     {
678       session_free (s);
679       app_worker_connect_notify (app_wrk, 0, opaque);
680       return -1;
681     }
682
683   s = session_get (new_si, new_ti);
684   s->session_state = opened_state;
685   session_lookup_add_connection (tc, session_handle (s));
686
687   if (app_worker_connect_notify (app_wrk, s, opaque))
688     {
689       s = session_get (new_si, new_ti);
690       session_free_w_fifos (s);
691       return -1;
692     }
693
694   return 0;
695 }
696
697 int
698 session_stream_connect_notify (transport_connection_t * tc, u8 is_fail)
699 {
700   return session_stream_connect_notify_inline (tc, is_fail,
701                                                SESSION_STATE_READY);
702 }
703
704 int
705 session_ho_stream_connect_notify (transport_connection_t * tc, u8 is_fail)
706 {
707   return session_stream_connect_notify_inline (tc, is_fail,
708                                                SESSION_STATE_OPENED);
709 }
710
711 typedef struct _session_switch_pool_args
712 {
713   u32 session_index;
714   u32 thread_index;
715   u32 new_thread_index;
716   u32 new_session_index;
717 } session_switch_pool_args_t;
718
719 static void
720 session_switch_pool (void *cb_args)
721 {
722   session_switch_pool_args_t *args = (session_switch_pool_args_t *) cb_args;
723   session_t *s;
724   ASSERT (args->thread_index == vlib_get_thread_index ());
725   s = session_get (args->session_index, args->thread_index);
726   s->tx_fifo->master_session_index = args->new_session_index;
727   s->tx_fifo->master_thread_index = args->new_thread_index;
728   transport_cleanup (session_get_transport_proto (s), s->connection_index,
729                      s->thread_index);
730   session_free (s);
731   clib_mem_free (cb_args);
732 }
733
734 /**
735  * Move dgram session to the right thread
736  */
737 int
738 session_dgram_connect_notify (transport_connection_t * tc,
739                               u32 old_thread_index, session_t ** new_session)
740 {
741   session_t *new_s;
742   session_switch_pool_args_t *rpc_args;
743
744   /*
745    * Clone half-open session to the right thread.
746    */
747   new_s = session_clone_safe (tc->s_index, old_thread_index);
748   new_s->connection_index = tc->c_index;
749   new_s->rx_fifo->master_session_index = new_s->session_index;
750   new_s->rx_fifo->master_thread_index = new_s->thread_index;
751   new_s->session_state = SESSION_STATE_READY;
752   session_lookup_add_connection (tc, session_handle (new_s));
753
754   /*
755    * Ask thread owning the old session to clean it up and make us the tx
756    * fifo owner
757    */
758   rpc_args = clib_mem_alloc (sizeof (*rpc_args));
759   rpc_args->new_session_index = new_s->session_index;
760   rpc_args->new_thread_index = new_s->thread_index;
761   rpc_args->session_index = tc->s_index;
762   rpc_args->thread_index = old_thread_index;
763   session_send_rpc_evt_to_thread (rpc_args->thread_index, session_switch_pool,
764                                   rpc_args);
765
766   tc->s_index = new_s->session_index;
767   new_s->connection_index = tc->c_index;
768   *new_session = new_s;
769   return 0;
770 }
771
772 /**
773  * Notification from transport that connection is being closed.
774  *
775  * A disconnect is sent to application but state is not removed. Once
776  * disconnect is acknowledged by application, session disconnect is called.
777  * Ultimately this leads to close being called on transport (passive close).
778  */
779 void
780 session_transport_closing_notify (transport_connection_t * tc)
781 {
782   app_worker_t *app_wrk;
783   session_t *s;
784
785   s = session_get (tc->s_index, tc->thread_index);
786   if (s->session_state >= SESSION_STATE_TRANSPORT_CLOSING)
787     return;
788   s->session_state = SESSION_STATE_TRANSPORT_CLOSING;
789   app_wrk = app_worker_get (s->app_wrk_index);
790   app_worker_close_notify (app_wrk, s);
791 }
792
793 /**
794  * Notification from transport that connection is being deleted
795  *
796  * This removes the session if it is still valid. It should be called only on
797  * previously fully established sessions. For instance failed connects should
798  * call stream_session_connect_notify and indicate that the connect has
799  * failed.
800  */
801 void
802 session_transport_delete_notify (transport_connection_t * tc)
803 {
804   session_t *s;
805
806   /* App might've been removed already */
807   if (!(s = session_get_if_valid (tc->s_index, tc->thread_index)))
808     return;
809
810   switch (s->session_state)
811     {
812     case SESSION_STATE_CREATED:
813       /* Session was created but accept notification was not yet sent to the
814        * app. Cleanup everything. */
815       session_lookup_del_session (s);
816       session_free_w_fifos (s);
817       break;
818     case SESSION_STATE_ACCEPTING:
819     case SESSION_STATE_TRANSPORT_CLOSING:
820     case SESSION_STATE_CLOSING:
821     case SESSION_STATE_TRANSPORT_CLOSED:
822       /* If transport finishes or times out before we get a reply
823        * from the app, mark transport as closed and wait for reply
824        * before removing the session. Cleanup session table in advance
825        * because transport will soon be closed and closed sessions
826        * are assumed to have been removed from the lookup table */
827       session_lookup_del_session (s);
828       s->session_state = SESSION_STATE_TRANSPORT_DELETED;
829       session_cleanup_notify (s, SESSION_CLEANUP_TRANSPORT);
830       svm_fifo_dequeue_drop_all (s->tx_fifo);
831       break;
832     case SESSION_STATE_APP_CLOSED:
833       /* Cleanup lookup table as transport needs to still be valid.
834        * Program transport close to ensure that all session events
835        * have been cleaned up. Once transport close is called, the
836        * session is just removed because both transport and app have
837        * confirmed the close*/
838       session_lookup_del_session (s);
839       s->session_state = SESSION_STATE_CLOSED;
840       session_cleanup_notify (s, SESSION_CLEANUP_TRANSPORT);
841       svm_fifo_dequeue_drop_all (s->tx_fifo);
842       session_program_transport_close (s);
843       break;
844     case SESSION_STATE_TRANSPORT_DELETED:
845       break;
846     case SESSION_STATE_CLOSED:
847       session_cleanup_notify (s, SESSION_CLEANUP_TRANSPORT);
848       session_delete (s);
849       break;
850     default:
851       clib_warning ("session state %u", s->session_state);
852       session_cleanup_notify (s, SESSION_CLEANUP_TRANSPORT);
853       session_delete (s);
854       break;
855     }
856 }
857
858 /**
859  * Notification from transport that it is closed
860  *
861  * Should be called by transport, prior to calling delete notify, once it
862  * knows that no more data will be exchanged. This could serve as an
863  * early acknowledgment of an active close especially if transport delete
864  * can be delayed a long time, e.g., tcp time-wait.
865  */
866 void
867 session_transport_closed_notify (transport_connection_t * tc)
868 {
869   app_worker_t *app_wrk;
870   session_t *s;
871
872   if (!(s = session_get_if_valid (tc->s_index, tc->thread_index)))
873     return;
874
875   /* Transport thinks that app requested close but it actually didn't.
876    * Can happen for tcp if fin and rst are received in close succession. */
877   if (s->session_state == SESSION_STATE_READY)
878     {
879       session_transport_closing_notify (tc);
880       svm_fifo_dequeue_drop_all (s->tx_fifo);
881       s->session_state = SESSION_STATE_TRANSPORT_CLOSED;
882     }
883   /* If app close has not been received or has not yet resulted in
884    * a transport close, only mark the session transport as closed */
885   else if (s->session_state <= SESSION_STATE_CLOSING)
886     {
887       s->session_state = SESSION_STATE_TRANSPORT_CLOSED;
888     }
889   /* If app also closed, switch to closed */
890   else if (s->session_state == SESSION_STATE_APP_CLOSED)
891     s->session_state = SESSION_STATE_CLOSED;
892
893   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
894   if (app_wrk)
895     app_worker_transport_closed_notify (app_wrk, s);
896 }
897
898 /**
899  * Notify application that connection has been reset.
900  */
901 void
902 session_transport_reset_notify (transport_connection_t * tc)
903 {
904   app_worker_t *app_wrk;
905   session_t *s;
906
907   s = session_get (tc->s_index, tc->thread_index);
908   svm_fifo_dequeue_drop_all (s->tx_fifo);
909   if (s->session_state >= SESSION_STATE_TRANSPORT_CLOSING)
910     return;
911   s->session_state = SESSION_STATE_TRANSPORT_CLOSING;
912   app_wrk = app_worker_get (s->app_wrk_index);
913   app_worker_reset_notify (app_wrk, s);
914 }
915
916 int
917 session_stream_accept_notify (transport_connection_t * tc)
918 {
919   app_worker_t *app_wrk;
920   session_t *s;
921
922   s = session_get (tc->s_index, tc->thread_index);
923   app_wrk = app_worker_get_if_valid (s->app_wrk_index);
924   if (!app_wrk)
925     return -1;
926   s->session_state = SESSION_STATE_ACCEPTING;
927   return app_worker_accept_notify (app_wrk, s);
928 }
929
930 /**
931  * Accept a stream session. Optionally ping the server by callback.
932  */
933 int
934 session_stream_accept (transport_connection_t * tc, u32 listener_index,
935                        u32 thread_index, u8 notify)
936 {
937   session_t *s;
938   int rv;
939
940   s = session_alloc_for_connection (tc);
941   s->listener_handle = ((u64) thread_index << 32) | (u64) listener_index;
942   s->session_state = SESSION_STATE_CREATED;
943
944   if ((rv = app_worker_init_accepted (s)))
945     return rv;
946
947   session_lookup_add_connection (tc, session_handle (s));
948
949   /* Shoulder-tap the server */
950   if (notify)
951     {
952       app_worker_t *app_wrk = app_worker_get (s->app_wrk_index);
953       return app_worker_accept_notify (app_wrk, s);
954     }
955
956   return 0;
957 }
958
959 int
960 session_open_cl (u32 app_wrk_index, session_endpoint_t * rmt, u32 opaque)
961 {
962   transport_connection_t *tc;
963   transport_endpoint_cfg_t *tep;
964   app_worker_t *app_wrk;
965   session_handle_t sh;
966   session_t *s;
967   int rv;
968
969   tep = session_endpoint_to_transport_cfg (rmt);
970   rv = transport_connect (rmt->transport_proto, tep);
971   if (rv < 0)
972     {
973       SESSION_DBG ("Transport failed to open connection.");
974       return VNET_API_ERROR_SESSION_CONNECT;
975     }
976
977   tc = transport_get_half_open (rmt->transport_proto, (u32) rv);
978
979   /* For dgram type of service, allocate session and fifos now */
980   app_wrk = app_worker_get (app_wrk_index);
981   s = session_alloc_for_connection (tc);
982   s->app_wrk_index = app_wrk->wrk_index;
983   s->session_state = SESSION_STATE_OPENED;
984   if (app_worker_init_connected (app_wrk, s))
985     {
986       session_free (s);
987       return -1;
988     }
989
990   sh = session_handle (s);
991   session_lookup_add_connection (tc, sh);
992   return app_worker_connect_notify (app_wrk, s, opaque);
993 }
994
995 int
996 session_open_vc (u32 app_wrk_index, session_endpoint_t * rmt, u32 opaque)
997 {
998   transport_connection_t *tc;
999   transport_endpoint_cfg_t *tep;
1000   u64 handle;
1001   int rv;
1002
1003   tep = session_endpoint_to_transport_cfg (rmt);
1004   rv = transport_connect (rmt->transport_proto, tep);
1005   if (rv < 0)
1006     {
1007       SESSION_DBG ("Transport failed to open connection.");
1008       return VNET_API_ERROR_SESSION_CONNECT;
1009     }
1010
1011   tc = transport_get_half_open (rmt->transport_proto, (u32) rv);
1012
1013   /* If transport offers a stream service, only allocate session once the
1014    * connection has been established.
1015    * Add connection to half-open table and save app and tc index. The
1016    * latter is needed to help establish the connection while the former
1017    * is needed when the connect notify comes and we have to notify the
1018    * external app
1019    */
1020   handle = (((u64) app_wrk_index) << 32) | (u64) tc->c_index;
1021   session_lookup_add_half_open (tc, handle);
1022
1023   /* Store api_context (opaque) for when the reply comes. Not the nicest
1024    * thing but better than allocating a separate half-open pool.
1025    */
1026   tc->s_index = opaque;
1027   if (transport_half_open_has_fifos (rmt->transport_proto))
1028     return session_ho_stream_connect_notify (tc, 0 /* is_fail */ );
1029   return 0;
1030 }
1031
1032 int
1033 session_open_app (u32 app_wrk_index, session_endpoint_t * rmt, u32 opaque)
1034 {
1035   session_endpoint_cfg_t *sep = (session_endpoint_cfg_t *) rmt;
1036   transport_endpoint_cfg_t *tep_cfg = session_endpoint_to_transport_cfg (sep);
1037
1038   sep->app_wrk_index = app_wrk_index;
1039   sep->opaque = opaque;
1040
1041   return transport_connect (rmt->transport_proto, tep_cfg);
1042 }
1043
1044 typedef int (*session_open_service_fn) (u32, session_endpoint_t *, u32);
1045
1046 /* *INDENT-OFF* */
1047 static session_open_service_fn session_open_srv_fns[TRANSPORT_N_SERVICES] = {
1048   session_open_vc,
1049   session_open_cl,
1050   session_open_app,
1051 };
1052 /* *INDENT-ON* */
1053
1054 /**
1055  * Ask transport to open connection to remote transport endpoint.
1056  *
1057  * Stores handle for matching request with reply since the call can be
1058  * asynchronous. For instance, for TCP the 3-way handshake must complete
1059  * before reply comes. Session is only created once connection is established.
1060  *
1061  * @param app_index Index of the application requesting the connect
1062  * @param st Session type requested.
1063  * @param tep Remote transport endpoint
1064  * @param opaque Opaque data (typically, api_context) the application expects
1065  *               on open completion.
1066  */
1067 int
1068 session_open (u32 app_wrk_index, session_endpoint_t * rmt, u32 opaque)
1069 {
1070   transport_service_type_t tst;
1071   tst = transport_protocol_service_type (rmt->transport_proto);
1072   return session_open_srv_fns[tst] (app_wrk_index, rmt, opaque);
1073 }
1074
1075 /**
1076  * Ask transport to listen on session endpoint.
1077  *
1078  * @param s Session for which listen will be called. Note that unlike
1079  *          established sessions, listen sessions are not associated to a
1080  *          thread.
1081  * @param sep Local endpoint to be listened on.
1082  */
1083 int
1084 session_listen (session_t * ls, session_endpoint_cfg_t * sep)
1085 {
1086   transport_endpoint_t *tep;
1087   u32 tc_index, s_index;
1088
1089   /* Transport bind/listen */
1090   tep = session_endpoint_to_transport (sep);
1091   s_index = ls->session_index;
1092   tc_index = transport_start_listen (session_get_transport_proto (ls),
1093                                      s_index, tep);
1094
1095   if (tc_index == (u32) ~ 0)
1096     return -1;
1097
1098   /* Attach transport to session. Lookup tables are populated by the app
1099    * worker because local tables (for ct sessions) are not backed by a fib */
1100   ls = listen_session_get (s_index);
1101   ls->connection_index = tc_index;
1102
1103   return 0;
1104 }
1105
1106 /**
1107  * Ask transport to stop listening on local transport endpoint.
1108  *
1109  * @param s Session to stop listening on. It must be in state LISTENING.
1110  */
1111 int
1112 session_stop_listen (session_t * s)
1113 {
1114   transport_proto_t tp = session_get_transport_proto (s);
1115   transport_connection_t *tc;
1116
1117   if (s->session_state != SESSION_STATE_LISTENING)
1118     return -1;
1119
1120   tc = transport_get_listener (tp, s->connection_index);
1121   if (!tc)
1122     return VNET_API_ERROR_ADDRESS_NOT_IN_USE;
1123
1124   if (!(tc->flags & TRANSPORT_CONNECTION_F_NO_LOOKUP))
1125     session_lookup_del_connection (tc);
1126   transport_stop_listen (tp, s->connection_index);
1127   return 0;
1128 }
1129
1130 /**
1131  * Initialize session closing procedure.
1132  *
1133  * Request is always sent to session node to ensure that all outstanding
1134  * requests are served before transport is notified.
1135  */
1136 void
1137 session_close (session_t * s)
1138 {
1139   if (!s)
1140     return;
1141
1142   if (s->session_state >= SESSION_STATE_CLOSING)
1143     {
1144       /* Session will only be removed once both app and transport
1145        * acknowledge the close */
1146       if (s->session_state == SESSION_STATE_TRANSPORT_CLOSED
1147           || s->session_state == SESSION_STATE_TRANSPORT_DELETED)
1148         session_program_transport_close (s);
1149       return;
1150     }
1151
1152   s->session_state = SESSION_STATE_CLOSING;
1153   session_program_transport_close (s);
1154 }
1155
1156 /**
1157  * Notify transport the session can be disconnected. This should eventually
1158  * result in a delete notification that allows us to cleanup session state.
1159  * Called for both active/passive disconnects.
1160  *
1161  * Must be called from the session's thread.
1162  */
1163 void
1164 session_transport_close (session_t * s)
1165 {
1166   if (s->session_state >= SESSION_STATE_APP_CLOSED)
1167     {
1168       if (s->session_state == SESSION_STATE_TRANSPORT_CLOSED)
1169         s->session_state = SESSION_STATE_CLOSED;
1170       /* If transport is already deleted, just free the session */
1171       else if (s->session_state >= SESSION_STATE_TRANSPORT_DELETED)
1172         session_free_w_fifos (s);
1173       return;
1174     }
1175
1176   /* If the tx queue wasn't drained, the transport can continue to try
1177    * sending the outstanding data (in closed state it cannot). It MUST however
1178    * at one point, either after sending everything or after a timeout, call
1179    * delete notify. This will finally lead to the complete cleanup of the
1180    * session.
1181    */
1182   s->session_state = SESSION_STATE_APP_CLOSED;
1183
1184   transport_close (session_get_transport_proto (s), s->connection_index,
1185                    s->thread_index);
1186 }
1187
1188 /**
1189  * Cleanup transport and session state.
1190  *
1191  * Notify transport of the cleanup and free the session. This should
1192  * be called only if transport reported some error and is already
1193  * closed.
1194  */
1195 void
1196 session_transport_cleanup (session_t * s)
1197 {
1198   s->session_state = SESSION_STATE_CLOSED;
1199
1200   /* Delete from main lookup table before we axe the the transport */
1201   session_lookup_del_session (s);
1202   transport_cleanup (session_get_transport_proto (s), s->connection_index,
1203                      s->thread_index);
1204   /* Since we called cleanup, no delete notification will come. So, make
1205    * sure the session is properly freed. */
1206   session_free_w_fifos (s);
1207 }
1208
1209 /**
1210  * Allocate event queues in the shared-memory segment
1211  *
1212  * That can either be a newly created memfd segment, that will need to be
1213  * mapped by all stack users, or the binary api's svm region. The latter is
1214  * assumed to be already mapped. NOTE that this assumption DOES NOT hold if
1215  * api clients bootstrap shm api over sockets (i.e. use memfd segments) and
1216  * vpp uses api svm region for event queues.
1217  */
1218 void
1219 session_vpp_event_queues_allocate (session_main_t * smm)
1220 {
1221   u32 evt_q_length = 2048, evt_size = sizeof (session_event_t);
1222   ssvm_private_t *eqs = &smm->evt_qs_segment;
1223   api_main_t *am = &api_main;
1224   uword eqs_size = 64 << 20;
1225   pid_t vpp_pid = getpid ();
1226   void *oldheap;
1227   int i;
1228
1229   if (smm->configured_event_queue_length)
1230     evt_q_length = smm->configured_event_queue_length;
1231
1232   if (smm->evt_qs_use_memfd_seg)
1233     {
1234       if (smm->evt_qs_segment_size)
1235         eqs_size = smm->evt_qs_segment_size;
1236
1237       eqs->ssvm_size = eqs_size;
1238       eqs->i_am_master = 1;
1239       eqs->my_pid = vpp_pid;
1240       eqs->name = format (0, "%s%c", "evt-qs-segment", 0);
1241       eqs->requested_va = smm->session_baseva;
1242
1243       if (ssvm_master_init (eqs, SSVM_SEGMENT_MEMFD))
1244         {
1245           clib_warning ("failed to initialize queue segment");
1246           return;
1247         }
1248     }
1249
1250   if (smm->evt_qs_use_memfd_seg)
1251     oldheap = ssvm_push_heap (eqs->sh);
1252   else
1253     oldheap = svm_push_data_heap (am->vlib_rp);
1254
1255   for (i = 0; i < vec_len (smm->wrk); i++)
1256     {
1257       svm_msg_q_cfg_t _cfg, *cfg = &_cfg;
1258       svm_msg_q_ring_cfg_t rc[SESSION_MQ_N_RINGS] = {
1259         {evt_q_length, evt_size, 0}
1260         ,
1261         {evt_q_length >> 1, 256, 0}
1262       };
1263       cfg->consumer_pid = 0;
1264       cfg->n_rings = 2;
1265       cfg->q_nitems = evt_q_length;
1266       cfg->ring_cfgs = rc;
1267       smm->wrk[i].vpp_event_queue = svm_msg_q_alloc (cfg);
1268       if (smm->evt_qs_use_memfd_seg)
1269         {
1270           if (svm_msg_q_alloc_consumer_eventfd (smm->wrk[i].vpp_event_queue))
1271             clib_warning ("eventfd returned");
1272         }
1273     }
1274
1275   if (smm->evt_qs_use_memfd_seg)
1276     ssvm_pop_heap (oldheap);
1277   else
1278     svm_pop_heap (oldheap);
1279 }
1280
1281 ssvm_private_t *
1282 session_main_get_evt_q_segment (void)
1283 {
1284   session_main_t *smm = &session_main;
1285   if (smm->evt_qs_use_memfd_seg)
1286     return &smm->evt_qs_segment;
1287   return 0;
1288 }
1289
1290 u64
1291 session_segment_handle (session_t * s)
1292 {
1293   svm_fifo_t *f;
1294
1295   if (!s->rx_fifo)
1296     return SESSION_INVALID_HANDLE;
1297
1298   f = s->rx_fifo;
1299   return segment_manager_make_segment_handle (f->segment_manager,
1300                                               f->segment_index);
1301 }
1302
1303 /* *INDENT-OFF* */
1304 static session_fifo_rx_fn *session_tx_fns[TRANSPORT_TX_N_FNS] = {
1305     session_tx_fifo_peek_and_snd,
1306     session_tx_fifo_dequeue_and_snd,
1307     session_tx_fifo_dequeue_internal,
1308     session_tx_fifo_dequeue_and_snd
1309 };
1310 /* *INDENT-ON* */
1311
1312 /**
1313  * Initialize session layer for given transport proto and ip version
1314  *
1315  * Allocates per session type (transport proto + ip version) data structures
1316  * and adds arc from session queue node to session type output node.
1317  */
1318 void
1319 session_register_transport (transport_proto_t transport_proto,
1320                             const transport_proto_vft_t * vft, u8 is_ip4,
1321                             u32 output_node)
1322 {
1323   session_main_t *smm = &session_main;
1324   session_type_t session_type;
1325   u32 next_index = ~0;
1326
1327   session_type = session_type_from_proto_and_ip (transport_proto, is_ip4);
1328
1329   vec_validate (smm->session_type_to_next, session_type);
1330   vec_validate (smm->session_tx_fns, session_type);
1331
1332   /* *INDENT-OFF* */
1333   if (output_node != ~0)
1334     {
1335       foreach_vlib_main (({
1336           next_index = vlib_node_add_next (this_vlib_main,
1337                                            session_queue_node.index,
1338                                            output_node);
1339       }));
1340     }
1341   /* *INDENT-ON* */
1342
1343   smm->session_type_to_next[session_type] = next_index;
1344   smm->session_tx_fns[session_type] =
1345     session_tx_fns[vft->transport_options.tx_type];
1346 }
1347
1348 transport_connection_t *
1349 session_get_transport (session_t * s)
1350 {
1351   if (s->session_state != SESSION_STATE_LISTENING)
1352     return transport_get_connection (session_get_transport_proto (s),
1353                                      s->connection_index, s->thread_index);
1354   else
1355     return transport_get_listener (session_get_transport_proto (s),
1356                                    s->connection_index);
1357 }
1358
1359 void
1360 session_get_endpoint (session_t * s, transport_endpoint_t * tep, u8 is_lcl)
1361 {
1362   if (s->session_state != SESSION_STATE_LISTENING)
1363     return transport_get_endpoint (session_get_transport_proto (s),
1364                                    s->connection_index, s->thread_index, tep,
1365                                    is_lcl);
1366   else
1367     return transport_get_listener_endpoint (session_get_transport_proto (s),
1368                                             s->connection_index, tep, is_lcl);
1369 }
1370
1371 transport_connection_t *
1372 listen_session_get_transport (session_t * s)
1373 {
1374   return transport_get_listener (session_get_transport_proto (s),
1375                                  s->connection_index);
1376 }
1377
1378 void
1379 session_flush_frames_main_thread (vlib_main_t * vm)
1380 {
1381   ASSERT (vlib_get_thread_index () == 0);
1382   vlib_process_signal_event_mt (vm, session_queue_process_node.index,
1383                                 SESSION_Q_PROCESS_FLUSH_FRAMES, 0);
1384 }
1385
1386 static clib_error_t *
1387 session_manager_main_enable (vlib_main_t * vm)
1388 {
1389   segment_manager_main_init_args_t _sm_args = { 0 }, *sm_args = &_sm_args;
1390   session_main_t *smm = &session_main;
1391   vlib_thread_main_t *vtm = vlib_get_thread_main ();
1392   u32 num_threads, preallocated_sessions_per_worker;
1393   session_worker_t *wrk;
1394   int i;
1395
1396   num_threads = 1 /* main thread */  + vtm->n_threads;
1397
1398   if (num_threads < 1)
1399     return clib_error_return (0, "n_thread_stacks not set");
1400
1401   /* Allocate cache line aligned worker contexts */
1402   vec_validate_aligned (smm->wrk, num_threads - 1, CLIB_CACHE_LINE_BYTES);
1403
1404   for (i = 0; i < num_threads; i++)
1405     {
1406       wrk = &smm->wrk[i];
1407       wrk->new_head = clib_llist_make_head (wrk->event_elts, evt_list);
1408       wrk->old_head = clib_llist_make_head (wrk->event_elts, evt_list);
1409       wrk->postponed_head = clib_llist_make_head (wrk->event_elts, evt_list);
1410       wrk->disconnects_head = clib_llist_make_head (wrk->event_elts,
1411                                                     evt_list);
1412       wrk->vm = vlib_mains[i];
1413       wrk->last_vlib_time = vlib_time_now (vlib_mains[i]);
1414
1415       if (num_threads > 1)
1416         clib_rwlock_init (&smm->wrk[i].peekers_rw_locks);
1417     }
1418
1419   /* Allocate vpp event queues segment and queue */
1420   session_vpp_event_queues_allocate (smm);
1421
1422   /* Initialize fifo segment main baseva and timeout */
1423   sm_args->baseva = smm->session_baseva + smm->evt_qs_segment_size;
1424   sm_args->size = smm->session_va_space_size;
1425   segment_manager_main_init (sm_args);
1426
1427   /* Preallocate sessions */
1428   if (smm->preallocated_sessions)
1429     {
1430       if (num_threads == 1)
1431         {
1432           pool_init_fixed (smm->wrk[0].sessions, smm->preallocated_sessions);
1433         }
1434       else
1435         {
1436           int j;
1437           preallocated_sessions_per_worker =
1438             (1.1 * (f64) smm->preallocated_sessions /
1439              (f64) (num_threads - 1));
1440
1441           for (j = 1; j < num_threads; j++)
1442             {
1443               pool_init_fixed (smm->wrk[j].sessions,
1444                                preallocated_sessions_per_worker);
1445             }
1446         }
1447     }
1448
1449   session_lookup_init ();
1450   app_namespaces_init ();
1451   transport_init ();
1452
1453   smm->is_enabled = 1;
1454
1455   /* Enable transports */
1456   transport_enable_disable (vm, 1);
1457   transport_init_tx_pacers_period ();
1458   return 0;
1459 }
1460
1461 void
1462 session_node_enable_disable (u8 is_en)
1463 {
1464   u8 state = is_en ? VLIB_NODE_STATE_POLLING : VLIB_NODE_STATE_DISABLED;
1465   vlib_thread_main_t *vtm = vlib_get_thread_main ();
1466   u8 have_workers = vtm->n_threads != 0;
1467
1468   /* *INDENT-OFF* */
1469   foreach_vlib_main (({
1470     if (have_workers && ii == 0)
1471       {
1472         vlib_node_set_state (this_vlib_main, session_queue_process_node.index,
1473                              state);
1474         if (is_en)
1475           {
1476             vlib_node_t *n = vlib_get_node (this_vlib_main,
1477                                             session_queue_process_node.index);
1478             vlib_start_process (this_vlib_main, n->runtime_index);
1479           }
1480         else
1481           {
1482             vlib_process_signal_event_mt (this_vlib_main,
1483                                           session_queue_process_node.index,
1484                                           SESSION_Q_PROCESS_STOP, 0);
1485           }
1486
1487         continue;
1488       }
1489     vlib_node_set_state (this_vlib_main, session_queue_node.index,
1490                          state);
1491   }));
1492   /* *INDENT-ON* */
1493 }
1494
1495 clib_error_t *
1496 vnet_session_enable_disable (vlib_main_t * vm, u8 is_en)
1497 {
1498   clib_error_t *error = 0;
1499   if (is_en)
1500     {
1501       if (session_main.is_enabled)
1502         return 0;
1503
1504       error = session_manager_main_enable (vm);
1505       session_node_enable_disable (is_en);
1506     }
1507   else
1508     {
1509       session_main.is_enabled = 0;
1510       session_node_enable_disable (is_en);
1511     }
1512
1513   return error;
1514 }
1515
1516 clib_error_t *
1517 session_manager_main_init (vlib_main_t * vm)
1518 {
1519   session_main_t *smm = &session_main;
1520   smm->session_baseva = HIGH_SEGMENT_BASEVA;
1521 #if (HIGH_SEGMENT_BASEVA > (4ULL << 30))
1522   smm->session_va_space_size = 128ULL << 30;
1523   smm->evt_qs_segment_size = 64 << 20;
1524 #else
1525   smm->session_va_space_size = 128 << 20;
1526   smm->evt_qs_segment_size = 1 << 20;
1527 #endif
1528   smm->is_enabled = 0;
1529   return 0;
1530 }
1531
1532 VLIB_INIT_FUNCTION (session_manager_main_init);
1533
1534 static clib_error_t *
1535 session_config_fn (vlib_main_t * vm, unformat_input_t * input)
1536 {
1537   session_main_t *smm = &session_main;
1538   u32 nitems;
1539   uword tmp;
1540
1541   while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
1542     {
1543       if (unformat (input, "event-queue-length %d", &nitems))
1544         {
1545           if (nitems >= 2048)
1546             smm->configured_event_queue_length = nitems;
1547           else
1548             clib_warning ("event queue length %d too small, ignored", nitems);
1549         }
1550       else if (unformat (input, "preallocated-sessions %d",
1551                          &smm->preallocated_sessions))
1552         ;
1553       else if (unformat (input, "v4-session-table-buckets %d",
1554                          &smm->configured_v4_session_table_buckets))
1555         ;
1556       else if (unformat (input, "v4-halfopen-table-buckets %d",
1557                          &smm->configured_v4_halfopen_table_buckets))
1558         ;
1559       else if (unformat (input, "v6-session-table-buckets %d",
1560                          &smm->configured_v6_session_table_buckets))
1561         ;
1562       else if (unformat (input, "v6-halfopen-table-buckets %d",
1563                          &smm->configured_v6_halfopen_table_buckets))
1564         ;
1565       else if (unformat (input, "v4-session-table-memory %U",
1566                          unformat_memory_size, &tmp))
1567         {
1568           if (tmp >= 0x100000000)
1569             return clib_error_return (0, "memory size %llx (%lld) too large",
1570                                       tmp, tmp);
1571           smm->configured_v4_session_table_memory = tmp;
1572         }
1573       else if (unformat (input, "v4-halfopen-table-memory %U",
1574                          unformat_memory_size, &tmp))
1575         {
1576           if (tmp >= 0x100000000)
1577             return clib_error_return (0, "memory size %llx (%lld) too large",
1578                                       tmp, tmp);
1579           smm->configured_v4_halfopen_table_memory = tmp;
1580         }
1581       else if (unformat (input, "v6-session-table-memory %U",
1582                          unformat_memory_size, &tmp))
1583         {
1584           if (tmp >= 0x100000000)
1585             return clib_error_return (0, "memory size %llx (%lld) too large",
1586                                       tmp, tmp);
1587           smm->configured_v6_session_table_memory = tmp;
1588         }
1589       else if (unformat (input, "v6-halfopen-table-memory %U",
1590                          unformat_memory_size, &tmp))
1591         {
1592           if (tmp >= 0x100000000)
1593             return clib_error_return (0, "memory size %llx (%lld) too large",
1594                                       tmp, tmp);
1595           smm->configured_v6_halfopen_table_memory = tmp;
1596         }
1597       else if (unformat (input, "local-endpoints-table-memory %U",
1598                          unformat_memory_size, &tmp))
1599         {
1600           if (tmp >= 0x100000000)
1601             return clib_error_return (0, "memory size %llx (%lld) too large",
1602                                       tmp, tmp);
1603           smm->local_endpoints_table_memory = tmp;
1604         }
1605       else if (unformat (input, "local-endpoints-table-buckets %d",
1606                          &smm->local_endpoints_table_buckets))
1607         ;
1608       else if (unformat (input, "evt_qs_memfd_seg"))
1609         smm->evt_qs_use_memfd_seg = 1;
1610       else if (unformat (input, "evt_qs_seg_size %U", unformat_memory_size,
1611                          &smm->evt_qs_segment_size))
1612         ;
1613       else
1614         return clib_error_return (0, "unknown input `%U'",
1615                                   format_unformat_error, input);
1616     }
1617   return 0;
1618 }
1619
1620 VLIB_CONFIG_FUNCTION (session_config_fn, "session");
1621
1622 /*
1623  * fd.io coding-style-patch-verification: ON
1624  *
1625  * Local Variables:
1626  * eval: (c-set-style "gnu")
1627  * End:
1628  */