2 * Copyright (c) 2017 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:
7 * http://www.apache.org/licenses/LICENSE-2.0
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.
16 #include <vnet/session/segment_manager.h>
17 #include <vnet/session/session.h>
18 #include <vnet/session/application.h>
21 * Counter used to build segment names
23 u32 segment_name_counter = 0;
26 * Pool of segment managers
28 segment_manager_t *segment_managers = 0;
31 * Process private segment index
33 u32 *private_segment_indices;
36 * Default fifo and segment size. TODO config.
38 u32 default_fifo_size = 1 << 16;
39 u32 default_segment_size = 1 << 20;
42 segment_manager_get_segment_info (u32 index, u8 ** name, u32 * size)
44 svm_fifo_segment_private_t *s;
45 s = svm_fifo_segment_get_segment (index);
46 *name = s->h->segment_name;
47 *size = s->ssvm.ssvm_size;
51 session_manager_add_segment_i (segment_manager_t * sm, u32 segment_size,
54 svm_fifo_segment_create_args_t _ca, *ca = &_ca;
57 memset (ca, 0, sizeof (*ca));
59 if (!sm->properties->use_private_segment)
61 ca->segment_name = (char *) segment_name;
62 ca->segment_size = segment_size;
63 ca->rx_fifo_size = sm->properties->rx_fifo_size;
64 ca->tx_fifo_size = sm->properties->tx_fifo_size;
65 ca->preallocated_fifo_pairs = sm->properties->preallocated_fifo_pairs;
67 rv = svm_fifo_segment_create (ca);
70 clib_warning ("svm_fifo_segment_create ('%s', %d) failed",
71 ca->segment_name, ca->segment_size);
72 return VNET_API_ERROR_SVM_SEGMENT_CREATE_FAIL;
77 u32 rx_fifo_size, tx_fifo_size, rx_rounded_data_size,
79 u32 approx_segment_count;
80 u64 approx_total_size;
82 ca->segment_name = "process-private-segment";
83 ca->segment_size = ~0;
84 ca->rx_fifo_size = sm->properties->rx_fifo_size;
85 ca->tx_fifo_size = sm->properties->tx_fifo_size;
86 ca->preallocated_fifo_pairs = sm->properties->preallocated_fifo_pairs;
87 ca->private_segment_count = sm->properties->private_segment_count;
88 ca->private_segment_size = sm->properties->private_segment_size;
90 /* Default to a small private segment */
91 if (ca->private_segment_size == 0)
92 ca->private_segment_size = 128 << 20;
94 /* Calculate space requirements */
95 rx_rounded_data_size = (1 << (max_log2 (ca->rx_fifo_size)));
96 tx_rounded_data_size = (1 << (max_log2 (ca->tx_fifo_size)));
98 rx_fifo_size = sizeof (svm_fifo_t) + rx_rounded_data_size;
99 tx_fifo_size = sizeof (svm_fifo_t) + tx_rounded_data_size;
101 approx_total_size = (u64) ca->preallocated_fifo_pairs
102 * (rx_fifo_size + tx_fifo_size);
103 approx_segment_count =
105 (ca->private_segment_size - 1)) / (u64) ca->private_segment_size;
107 /* The user asked us to figure it out... */
108 if (ca->private_segment_count == 0)
110 ca->private_segment_count = approx_segment_count;
112 /* Follow directions, but issue a warning */
113 else if (approx_segment_count != ca->private_segment_count)
116 ("Honoring segment count %u, but calculated count was %u",
117 ca->private_segment_count, approx_segment_count);
120 if (svm_fifo_segment_create_process_private (ca))
121 clib_warning ("Failed to create process private segment");
123 ASSERT (vec_len (ca->new_segment_indices));
125 vec_append (sm->segment_indices, ca->new_segment_indices);
126 vec_free (ca->new_segment_indices);
131 session_manager_add_segment (segment_manager_t * sm)
134 svm_fifo_segment_create_args_t _ca, *ca = &_ca;
135 u32 add_segment_size;
138 memset (ca, 0, sizeof (*ca));
139 segment_name = format (0, "%d-%d%c", getpid (), segment_name_counter++, 0);
140 add_segment_size = sm->properties->add_segment_size ?
141 sm->properties->add_segment_size : default_segment_size;
143 rv = session_manager_add_segment_i (sm, add_segment_size, segment_name);
144 vec_free (segment_name);
149 session_manager_add_first_segment (segment_manager_t * sm, u32 segment_size)
154 segment_name = format (0, "%d-%d%c", getpid (), segment_name_counter++, 0);
155 rv = session_manager_add_segment_i (sm, segment_size, segment_name);
156 vec_free (segment_name);
161 segment_manager_new ()
163 segment_manager_t *sm;
164 pool_get (segment_managers, sm);
165 memset (sm, 0, sizeof (*sm));
170 * Initializes segment manager based on options provided.
171 * Returns error if svm segment allocation fails.
174 segment_manager_init (segment_manager_t * sm,
175 segment_manager_properties_t * properties,
180 /* app allocates these */
181 sm->properties = properties;
183 first_seg_size = first_seg_size > 0 ? first_seg_size : default_segment_size;
185 rv = session_manager_add_first_segment (sm, first_seg_size);
188 clib_warning ("Failed to allocate segment");
192 clib_spinlock_init (&sm->lockp);
197 segment_manager_has_fifos (segment_manager_t * sm)
199 svm_fifo_segment_private_t *segment;
200 /* Weird, but handle it */
201 if (vec_len (sm->segment_indices) == 0)
203 if (vec_len (sm->segment_indices) == 1)
205 segment = svm_fifo_segment_get_segment (sm->segment_indices[0]);
206 if (svm_fifo_segment_num_fifos (segment) == 0)
211 svm_fifo_segment_private_t *segment;
213 for (i = 1; i < vec_len (sm->segment_indices); i++)
215 segment = svm_fifo_segment_get_segment (sm->segment_indices[i]);
216 if (!svm_fifo_segment_has_fifos (segment))
217 clib_warning ("segment has no fifos!");
224 segment_manager_del_segment (segment_manager_t * sm, u32 segment_index)
226 svm_fifo_segment_private_t *fifo_segment;
227 u32 svm_segment_index;
228 clib_spinlock_lock (&sm->lockp);
229 svm_segment_index = sm->segment_indices[segment_index];
230 fifo_segment = svm_fifo_segment_get_segment (svm_segment_index);
231 svm_fifo_segment_delete (fifo_segment);
232 vec_del1 (sm->segment_indices, segment_index);
233 clib_spinlock_unlock (&sm->lockp);
237 * Initiate disconnects for all sessions 'owned' by a segment manager
240 segment_manager_del_sessions (segment_manager_t * sm)
243 svm_fifo_segment_private_t *fifo_segment;
246 ASSERT (vec_len (sm->segment_indices));
248 /* Across all fifo segments used by the server */
249 for (j = 0; j < vec_len (sm->segment_indices); j++)
251 fifo_segment = svm_fifo_segment_get_segment (sm->segment_indices[j]);
252 fifo = svm_fifo_segment_get_fifo_list (fifo_segment);
255 * Remove any residual sessions from the session lookup table
256 * Don't bother deleting the individual fifos, we're going to
257 * throw away the fifo segment in a minute.
261 u32 session_index, thread_index;
262 stream_session_t *session;
264 session_index = fifo->master_session_index;
265 thread_index = fifo->master_thread_index;
266 session = stream_session_get (session_index, thread_index);
268 /* Instead of directly removing the session call disconnect */
269 if (session->session_state != SESSION_STATE_CLOSED)
271 session->session_state = SESSION_STATE_CLOSED;
272 session_send_session_evt_to_thread (stream_session_handle
274 FIFO_EVENT_DISCONNECT,
280 /* Instead of removing the segment, test when cleaning up disconnected
281 * sessions if the segment can be removed.
287 * Removes segment manager.
289 * Since the fifos allocated in the segment keep backpointers to the sessions
290 * prior to removing the segment, we call session disconnect. This
291 * subsequently propages into transport.
294 segment_manager_del (segment_manager_t * sm)
297 ASSERT (vec_len (sm->segment_indices) <= 1);
298 if (vec_len (sm->segment_indices))
300 /* The first segment in the first segment manager is not removed when
301 * all fifos are removed. It can only be removed when the manager is
302 * explicitly deleted/detached by the app. */
305 svm_fifo_segment_private_t *fifo_segment;
307 svm_fifo_segment_get_segment (sm->segment_indices[0]);
308 ASSERT (!svm_fifo_segment_has_fifos (fifo_segment));
310 segment_manager_del_segment (sm, 0);
312 clib_spinlock_free (&sm->lockp);
314 memset (sm, 0xfe, sizeof (*sm));
315 pool_put (segment_managers, sm);
319 segment_manager_init_del (segment_manager_t * sm)
321 if (segment_manager_has_fifos (sm))
322 segment_manager_del_sessions (sm);
325 ASSERT (!sm->first_is_protected
326 || sm->app_index == SEGMENT_MANAGER_INVALID_APP_INDEX);
327 segment_manager_del (sm);
332 segment_manager_alloc_session_fifos (segment_manager_t * sm,
333 svm_fifo_t ** server_rx_fifo,
334 svm_fifo_t ** server_tx_fifo,
335 u32 * fifo_segment_index)
337 svm_fifo_segment_private_t *fifo_segment;
338 u32 fifo_size, sm_index;
339 u8 added_a_segment = 0;
342 ASSERT (vec_len (sm->segment_indices));
344 /* Make sure we don't have multiple threads trying to allocate segments
345 * at the same time. */
346 clib_spinlock_lock (&sm->lockp);
348 /* Allocate svm fifos */
350 for (i = 0; i < vec_len (sm->segment_indices); i++)
352 *fifo_segment_index = sm->segment_indices[i];
353 fifo_segment = svm_fifo_segment_get_segment (*fifo_segment_index);
355 fifo_size = sm->properties->rx_fifo_size;
356 fifo_size = (fifo_size == 0) ? default_fifo_size : fifo_size;
358 svm_fifo_segment_alloc_fifo (fifo_segment, fifo_size,
359 FIFO_SEGMENT_RX_FREELIST);
361 fifo_size = sm->properties->tx_fifo_size;
362 fifo_size = (fifo_size == 0) ? default_fifo_size : fifo_size;
364 svm_fifo_segment_alloc_fifo (fifo_segment, fifo_size,
365 FIFO_SEGMENT_TX_FREELIST);
367 if (*server_rx_fifo == 0)
369 /* This would be very odd, but handle it... */
370 if (*server_tx_fifo != 0)
372 svm_fifo_segment_free_fifo (fifo_segment, *server_tx_fifo,
373 FIFO_SEGMENT_TX_FREELIST);
378 if (*server_tx_fifo == 0)
380 if (*server_rx_fifo != 0)
382 svm_fifo_segment_free_fifo (fifo_segment, *server_rx_fifo,
383 FIFO_SEGMENT_RX_FREELIST);
391 /* See if we're supposed to create another segment */
392 if (*server_rx_fifo == 0)
394 if (sm->properties->add_segment && !sm->properties->use_private_segment)
398 clib_warning ("added a segment, still can't allocate a fifo");
399 clib_spinlock_unlock (&sm->lockp);
400 return SESSION_ERROR_NEW_SEG_NO_SPACE;
403 if (session_manager_add_segment (sm))
405 clib_spinlock_unlock (&sm->lockp);
406 return VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
414 clib_warning ("No space to allocate fifos!");
415 clib_spinlock_unlock (&sm->lockp);
416 return SESSION_ERROR_NO_SPACE;
420 /* Backpointers to segment manager */
421 sm_index = segment_manager_index (sm);
422 (*server_tx_fifo)->segment_manager = sm_index;
423 (*server_rx_fifo)->segment_manager = sm_index;
425 clib_spinlock_unlock (&sm->lockp);
428 return application_add_segment_notify (sm->app_index,
429 *fifo_segment_index);
435 segment_manager_dealloc_fifos (u32 svm_segment_index, svm_fifo_t * rx_fifo,
436 svm_fifo_t * tx_fifo)
438 segment_manager_t *sm;
439 svm_fifo_segment_private_t *fifo_segment;
440 u32 i, segment_index = ~0;
443 sm = segment_manager_get_if_valid (rx_fifo->segment_manager);
445 /* It's possible to have no segment manager if the session was removed
446 * as result of a detach. */
450 fifo_segment = svm_fifo_segment_get_segment (svm_segment_index);
451 svm_fifo_segment_free_fifo (fifo_segment, rx_fifo,
452 FIFO_SEGMENT_RX_FREELIST);
453 svm_fifo_segment_free_fifo (fifo_segment, tx_fifo,
454 FIFO_SEGMENT_TX_FREELIST);
457 * Try to remove svm segment if it has no fifos. This can be done only if
458 * the segment is not the first in the segment manager or if it is first
459 * and it is not protected. Moreover, if the segment is first and the app
460 * has detached from the segment manager, remove the segment manager.
462 if (!svm_fifo_segment_has_fifos (fifo_segment))
464 is_first = sm->segment_indices[0] == svm_segment_index;
466 /* Remove segment if it holds no fifos or first but not protected */
467 if (!is_first || !sm->first_is_protected)
469 /* Find the segment manager segment index */
470 for (i = 0; i < vec_len (sm->segment_indices); i++)
471 if (sm->segment_indices[i] == svm_segment_index)
476 ASSERT (segment_index != (u32) ~ 0);
477 segment_manager_del_segment (sm, segment_index);
480 /* Remove segment manager if no sessions and detached from app */
481 if (sm->app_index == SEGMENT_MANAGER_INVALID_APP_INDEX && is_first)
482 segment_manager_del (sm);
487 * Allocates shm queue in the first segment
489 unix_shared_memory_queue_t *
490 segment_manager_alloc_queue (segment_manager_t * sm, u32 queue_size)
492 ssvm_shared_header_t *sh;
493 svm_fifo_segment_private_t *segment;
494 unix_shared_memory_queue_t *q;
497 ASSERT (sm->segment_indices != 0);
499 segment = svm_fifo_segment_get_segment (sm->segment_indices[0]);
500 sh = segment->ssvm.sh;
502 oldheap = ssvm_push_heap (sh);
503 q = unix_shared_memory_queue_init (queue_size,
504 sizeof (session_fifo_event_t),
505 0 /* consumer pid */ ,
506 0 /* signal when queue non-empty */ );
507 ssvm_pop_heap (oldheap);
512 * Frees shm queue allocated in the first segment
515 segment_manager_dealloc_queue (segment_manager_t * sm,
516 unix_shared_memory_queue_t * q)
518 ssvm_shared_header_t *sh;
519 svm_fifo_segment_private_t *segment;
522 ASSERT (sm->segment_indices != 0);
524 segment = svm_fifo_segment_get_segment (sm->segment_indices[0]);
525 sh = segment->ssvm.sh;
527 oldheap = ssvm_push_heap (sh);
528 unix_shared_memory_queue_free (q);
529 ssvm_pop_heap (oldheap);
532 static clib_error_t *
533 segment_manager_show_fn (vlib_main_t * vm, unformat_input_t * input,
534 vlib_cli_command_t * cmd)
536 svm_fifo_segment_private_t *segments, *seg;
537 segment_manager_t *sm;
538 u8 show_segments = 0, verbose = 0, *name;
544 mheap_t *heap_header;
546 while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT)
548 if (unformat (input, "segments"))
550 else if (unformat (input, "verbose"))
553 return clib_error_return (0, "unknown input `%U'",
554 format_unformat_error, input);
556 vlib_cli_output (vm, "%d segment managers allocated",
557 pool_elts (segment_managers));
558 if (verbose && pool_elts (segment_managers))
560 vlib_cli_output (vm, "%-10s%=15s%=12s", "Index", "App Index",
564 pool_foreach (sm, segment_managers, ({
565 vlib_cli_output (vm, "%-10d%=15d%=12d", segment_manager_index(sm),
566 sm->app_index, vec_len (sm->segment_indices));
573 segments = svm_fifo_segment_segments_pool ();
574 vlib_cli_output (vm, "%d svm fifo segments allocated",
575 pool_elts (segments));
576 vlib_cli_output (vm, "%-20s%=12s%=16s%=16s%=16s", "Name",
577 "HeapSize (M)", "ActiveFifos", "FreeFifos", "Address");
580 pool_foreach (seg, segments, ({
581 if (seg->h->flags & FIFO_SEGMENT_F_IS_PRIVATE)
583 address = pointer_to_uword (seg->ssvm.sh->heap);
584 if (seg->h->flags & FIFO_SEGMENT_F_IS_MAIN_HEAP)
585 name = format (0, "main heap");
587 name = format (0, "private heap");
588 heap_header = mheap_header (seg->ssvm.sh->heap);
589 size = heap_header->max_size;
593 address = seg->ssvm.sh->ssvm_va;
594 size = seg->ssvm.ssvm_size;
595 name = seg->ssvm.sh->name;
597 active_fifos = svm_fifo_segment_num_fifos (seg);
598 free_fifos = svm_fifo_segment_num_free_fifos (seg, ~0 /* size */);
599 vlib_cli_output (vm, "%-20v%=16llu%=16u%=16u%16llx",
600 name, size >> 20ULL, active_fifos, free_fifos,
603 vlib_cli_output (vm, "%U",
604 format_svm_fifo_segment, seg, verbose);
605 if (seg->h->flags & FIFO_SEGMENT_F_IS_PRIVATE)
615 VLIB_CLI_COMMAND (segment_manager_show_command, static) =
617 .path = "show segment-manager",
618 .short_help = "show segment-manager [segments][verbose]",
619 .function = segment_manager_show_fn,
624 * fd.io coding-style-patch-verification: ON
627 * eval: (c-set-style "gnu")