blob: b13df21ca9e4e7cbe1bbc55184787a0b666c1301 [file] [log] [blame]
/*
* Copyright (c) 2017 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <vnet/session/segment_manager.h>
#include <vnet/session/session.h>
#include <vnet/session/application.h>
/**
* Counter used to build segment names
*/
u32 segment_name_counter = 0;
/**
* Pool of segment managers
*/
segment_manager_t *segment_managers = 0;
/**
* Process private segment index
*/
u32 private_segment_index = ~0;
/**
* Default fifo and segment size. TODO config.
*/
u32 default_fifo_size = 1 << 16;
u32 default_segment_size = 1 << 20;
void
segment_manager_get_segment_info (u32 index, u8 ** name, u32 * size)
{
svm_fifo_segment_private_t *s;
s = svm_fifo_get_segment (index);
*name = s->h->segment_name;
*size = s->ssvm.ssvm_size;
}
always_inline int
session_manager_add_segment_i (segment_manager_t * sm, u32 segment_size,
u8 * segment_name)
{
svm_fifo_segment_create_args_t _ca, *ca = &_ca;
int rv;
memset (ca, 0, sizeof (*ca));
ca->segment_name = (char *) segment_name;
ca->segment_size = segment_size;
rv = svm_fifo_segment_create (ca);
if (rv)
{
clib_warning ("svm_fifo_segment_create ('%s', %d) failed",
ca->segment_name, ca->segment_size);
vec_free (segment_name);
return VNET_API_ERROR_SVM_SEGMENT_CREATE_FAIL;
}
vec_add1 (sm->segment_indices, ca->new_segment_index);
return 0;
}
int
session_manager_add_segment (segment_manager_t * sm)
{
u8 *segment_name;
svm_fifo_segment_create_args_t _ca, *ca = &_ca;
u32 add_segment_size;
int rv;
memset (ca, 0, sizeof (*ca));
segment_name = format (0, "%d-%d%c", getpid (), segment_name_counter++, 0);
add_segment_size = sm->properties->add_segment_size ?
sm->properties->add_segment_size : default_segment_size;
rv = session_manager_add_segment_i (sm, add_segment_size, segment_name);
vec_free (segment_name);
return rv;
}
int
session_manager_add_first_segment (segment_manager_t * sm, u32 segment_size)
{
u8 *segment_name;
int rv;
segment_name = format (0, "%d-%d%c", getpid (), segment_name_counter++, 0);
rv = session_manager_add_segment_i (sm, segment_size, segment_name);
vec_free (segment_name);
return rv;
}
static void
segment_manager_alloc_process_private_segment ()
{
svm_fifo_segment_create_args_t _a, *a = &_a;
if (private_segment_index != ~0)
return;
memset (a, 0, sizeof (*a));
a->segment_name = "process-private-segment";
a->segment_size = ~0;
a->new_segment_index = ~0;
if (svm_fifo_segment_create_process_private (a))
clib_warning ("Failed to create process private segment");
private_segment_index = a->new_segment_index;
ASSERT (private_segment_index != ~0);
}
/**
* Initializes segment manager based on options provided.
* Returns error if svm segment allocation fails.
*/
int
segment_manager_init (segment_manager_t * sm,
segment_manager_properties_t * properties,
u32 first_seg_size)
{
int rv;
/* app allocates these */
sm->properties = properties;
first_seg_size = first_seg_size > 0 ? first_seg_size : default_segment_size;
if (sm->properties->use_private_segment == 0)
{
rv = session_manager_add_first_segment (sm, first_seg_size);
if (rv)
{
clib_warning ("Failed to allocate segment");
return rv;
}
}
else
{
if (private_segment_index == ~0)
segment_manager_alloc_process_private_segment ();
ASSERT (private_segment_index != ~0);
vec_add1 (sm->segment_indices, private_segment_index);
}
clib_spinlock_init (&sm->lockp);
return 0;
}
/**
* Removes segment manager.
*
* Since the fifos allocated in the segment keep backpointers to the sessions
* prior to removing the segment, we call session disconnect. This
* subsequently propages into transport.
*/
void
segment_manager_del (segment_manager_t * sm)
{
u32 *deleted_sessions = 0;
u32 *deleted_thread_indices = 0;
int i, j;
/* Across all fifo segments used by the server */
for (j = 0; j < vec_len (sm->segment_indices); j++)
{
svm_fifo_segment_private_t *fifo_segment;
svm_fifo_t **fifos;
/* Vector of fifos allocated in the segment */
fifo_segment = svm_fifo_get_segment (sm->segment_indices[j]);
fifos = svm_fifo_segment_get_fifos (fifo_segment);
/*
* Remove any residual sessions from the session lookup table
* Don't bother deleting the individual fifos, we're going to
* throw away the fifo segment in a minute.
*/
for (i = 0; i < vec_len (fifos); i++)
{
svm_fifo_t *fifo;
u32 session_index, thread_index;
stream_session_t *session;
fifo = fifos[i];
session_index = fifo->master_session_index;
thread_index = fifo->master_thread_index;
session = stream_session_get (session_index, thread_index);
/* Add to the deleted_sessions vector (once!) */
if (!session->is_deleted)
{
session->is_deleted = 1;
vec_add1 (deleted_sessions, session_index);
vec_add1 (deleted_thread_indices, thread_index);
}
}
for (i = 0; i < vec_len (deleted_sessions); i++)
{
stream_session_t *session;
session = stream_session_get (deleted_sessions[i],
deleted_thread_indices[i]);
/* Instead of directly removing the session call disconnect */
session_send_session_evt_to_thread (stream_session_handle (session),
FIFO_EVENT_DISCONNECT,
deleted_thread_indices[i]);
/*
stream_session_table_del (smm, session);
pool_put(smm->sessions[deleted_thread_indices[i]], session);
*/
}
vec_reset_length (deleted_sessions);
vec_reset_length (deleted_thread_indices);
/* Instead of removing the segment, test when removing the session if
* the segment can be removed
*/
/* svm_fifo_segment_delete (fifo_segment); */
}
clib_spinlock_free (&sm->lockp);
vec_free (deleted_sessions);
vec_free (deleted_thread_indices);
pool_put (segment_managers, sm);
}
static int
segment_manager_notify_app_seg_add (segment_manager_t * sm,
u32 fifo_segment_index)
{
application_t *app = application_get (sm->app_index);
u32 seg_size = 0;
u8 *seg_name;
/* Send an API message to the external app, to map new segment */
ASSERT (app->cb_fns.add_segment_callback);
segment_manager_get_segment_info (fifo_segment_index, &seg_name, &seg_size);
return app->cb_fns.add_segment_callback (app->api_client_index, seg_name,
seg_size);
}
int
segment_manager_alloc_session_fifos (segment_manager_t * sm,
svm_fifo_t ** server_rx_fifo,
svm_fifo_t ** server_tx_fifo,
u32 * fifo_segment_index)
{
svm_fifo_segment_private_t *fifo_segment;
u32 fifo_size, sm_index;
u8 added_a_segment = 0;
int i;
ASSERT (vec_len (sm->segment_indices));
/* Make sure we don't have multiple threads trying to allocate segments
* at the same time. */
clib_spinlock_lock (&sm->lockp);
/* Allocate svm fifos */
again:
for (i = 0; i < vec_len (sm->segment_indices); i++)
{
*fifo_segment_index = sm->segment_indices[i];
fifo_segment = svm_fifo_get_segment (*fifo_segment_index);
fifo_size = sm->properties->rx_fifo_size;
fifo_size = (fifo_size == 0) ? default_fifo_size : fifo_size;
*server_rx_fifo = svm_fifo_segment_alloc_fifo (fifo_segment, fifo_size);
fifo_size = sm->properties->tx_fifo_size;
fifo_size = (fifo_size == 0) ? default_fifo_size : fifo_size;
*server_tx_fifo = svm_fifo_segment_alloc_fifo (fifo_segment, fifo_size);
if (*server_rx_fifo == 0)
{
/* This would be very odd, but handle it... */
if (*server_tx_fifo != 0)
{
svm_fifo_segment_free_fifo (fifo_segment, *server_tx_fifo);
*server_tx_fifo = 0;
}
continue;
}
if (*server_tx_fifo == 0)
{
if (*server_rx_fifo != 0)
{
svm_fifo_segment_free_fifo (fifo_segment, *server_rx_fifo);
*server_rx_fifo = 0;
}
continue;
}
break;
}
/* See if we're supposed to create another segment */
if (*server_rx_fifo == 0)
{
if (sm->properties->add_segment)
{
if (added_a_segment)
{
clib_warning ("added a segment, still cant allocate a fifo");
return SESSION_ERROR_NEW_SEG_NO_SPACE;
}
if (session_manager_add_segment (sm))
{
return VNET_API_ERROR_URI_FIFO_CREATE_FAILED;
}
added_a_segment = 1;
goto again;
}
else
{
clib_warning ("No space to allocate fifos!");
return SESSION_ERROR_NO_SPACE;
}
}
/* Backpointers to segment manager */
sm_index = segment_manager_index (sm);
(*server_tx_fifo)->segment_manager = sm_index;
(*server_rx_fifo)->segment_manager = sm_index;
clib_spinlock_unlock (&sm->lockp);
if (added_a_segment)
return segment_manager_notify_app_seg_add (sm, *fifo_segment_index);
return 0;
}
void
segment_manager_dealloc_fifos (u32 svm_segment_index, svm_fifo_t * rx_fifo,
svm_fifo_t * tx_fifo)
{
segment_manager_t *sm;
svm_fifo_segment_private_t *fifo_segment;
sm = segment_manager_get_if_valid (rx_fifo->segment_manager);
/* It's possible to have no segment manager if the session was removed
* as result of a detach */
if (!sm)
return;
fifo_segment = svm_fifo_get_segment (svm_segment_index);
svm_fifo_segment_free_fifo (fifo_segment, rx_fifo);
svm_fifo_segment_free_fifo (fifo_segment, tx_fifo);
/* Remove segment only if it holds no fifos and not the first */
if (sm->segment_indices[0] != svm_segment_index
&& !svm_fifo_segment_has_fifos (fifo_segment))
{
svm_fifo_segment_delete (fifo_segment);
vec_del1 (sm->segment_indices, svm_segment_index);
}
}
/**
* Allocates shm queue in the first segment
*/
unix_shared_memory_queue_t *
segment_manager_alloc_queue (segment_manager_t * sm, u32 queue_size)
{
ssvm_shared_header_t *sh;
svm_fifo_segment_private_t *segment;
unix_shared_memory_queue_t *q;
void *oldheap;
ASSERT (sm->segment_indices != 0);
segment = svm_fifo_get_segment (sm->segment_indices[0]);
sh = segment->ssvm.sh;
oldheap = ssvm_push_heap (sh);
q =
unix_shared_memory_queue_init (queue_size, sizeof (session_fifo_event_t),
0 /* consumer pid */ , 0
/* signal when queue non-empty */ );
ssvm_pop_heap (oldheap);
return q;
}
/**
* Frees shm queue allocated in the first segment
*/
void
segment_manager_dealloc_queue (segment_manager_t * sm,
unix_shared_memory_queue_t * q)
{
ssvm_shared_header_t *sh;
svm_fifo_segment_private_t *segment;
void *oldheap;
ASSERT (sm->segment_indices != 0);
segment = svm_fifo_get_segment (sm->segment_indices[0]);
sh = segment->ssvm.sh;
oldheap = ssvm_push_heap (sh);
unix_shared_memory_queue_free (q);
ssvm_pop_heap (oldheap);
}
/*
* fd.io coding-style-patch-verification: ON
*
* Local Variables:
* eval: (c-set-style "gnu")
* End:
*/