339 lines
9.0 KiB
C
339 lines
9.0 KiB
C
/*
|
|
* Copyright (c) 2017, Shane Bryldt
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* * Neither the name of the original author; nor the names of any contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
|
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "blade.h"
|
|
|
|
struct blade_sessionmgr_s {
|
|
blade_handle_t *handle;
|
|
|
|
blade_session_t *loopback;
|
|
blade_session_t *upstream;
|
|
ks_hash_t *sessions; // id, blade_session_t*
|
|
ks_hash_t *callbacks; // id, blade_session_callback_data_t*
|
|
};
|
|
|
|
struct blade_session_callback_data_s {
|
|
const char *id;
|
|
void *data;
|
|
blade_session_callback_t callback;
|
|
};
|
|
|
|
|
|
static void blade_sessionmgr_cleanup(void *ptr, void *arg, ks_pool_cleanup_action_t action, ks_pool_cleanup_type_t type)
|
|
{
|
|
//blade_sessionmgr_t *bsmgr = (blade_sessionmgr_t *)ptr;
|
|
|
|
//ks_assert(bsmgr);
|
|
|
|
switch (action) {
|
|
case KS_MPCL_ANNOUNCE:
|
|
break;
|
|
case KS_MPCL_TEARDOWN:
|
|
break;
|
|
case KS_MPCL_DESTROY:
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void blade_session_callback_data_cleanup(void *ptr, void *arg, ks_pool_cleanup_action_t action, ks_pool_cleanup_type_t type)
|
|
{
|
|
blade_session_callback_data_t *bscd = (blade_session_callback_data_t *)ptr;
|
|
|
|
ks_assert(bscd);
|
|
|
|
switch (action) {
|
|
case KS_MPCL_ANNOUNCE:
|
|
break;
|
|
case KS_MPCL_TEARDOWN:
|
|
ks_pool_free(&bscd->id);
|
|
break;
|
|
case KS_MPCL_DESTROY:
|
|
break;
|
|
}
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_create(blade_sessionmgr_t **bsmgrP, blade_handle_t *bh)
|
|
{
|
|
ks_pool_t *pool = NULL;
|
|
blade_sessionmgr_t *bsmgr = NULL;
|
|
|
|
ks_assert(bsmgrP);
|
|
|
|
ks_pool_open(&pool);
|
|
ks_assert(pool);
|
|
|
|
bsmgr = ks_pool_alloc(pool, sizeof(blade_sessionmgr_t));
|
|
bsmgr->handle = bh;
|
|
|
|
ks_hash_create(&bsmgr->sessions, KS_HASH_MODE_CASE_INSENSITIVE, KS_HASH_FLAG_RWLOCK | KS_HASH_FLAG_DUP_CHECK | KS_HASH_FLAG_FREE_KEY, pool);
|
|
ks_assert(bsmgr->sessions);
|
|
|
|
ks_hash_create(&bsmgr->callbacks, KS_HASH_MODE_CASE_INSENSITIVE, KS_HASH_FLAG_RWLOCK | KS_HASH_FLAG_DUP_CHECK | KS_HASH_FLAG_FREE_KEY | KS_HASH_FLAG_FREE_VALUE, pool);
|
|
ks_assert(bsmgr->callbacks);
|
|
|
|
ks_pool_set_cleanup(bsmgr, NULL, blade_sessionmgr_cleanup);
|
|
|
|
*bsmgrP = bsmgr;
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_destroy(blade_sessionmgr_t **bsmgrP)
|
|
{
|
|
blade_sessionmgr_t *bsmgr = NULL;
|
|
ks_pool_t *pool;
|
|
|
|
ks_assert(bsmgrP);
|
|
ks_assert(*bsmgrP);
|
|
|
|
bsmgr = *bsmgrP;
|
|
*bsmgrP = NULL;
|
|
|
|
pool = ks_pool_get(bsmgr);
|
|
|
|
ks_pool_close(&pool);
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(blade_handle_t *) blade_sessionmgr_handle_get(blade_sessionmgr_t *bsmgr)
|
|
{
|
|
ks_assert(bsmgr);
|
|
return bsmgr->handle;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_startup(blade_sessionmgr_t *bsmgr, config_setting_t *config)
|
|
{
|
|
ks_assert(bsmgr);
|
|
|
|
blade_session_create(&bsmgr->loopback, bsmgr->handle, BLADE_SESSION_FLAGS_LOOPBACK, NULL);
|
|
ks_assert(bsmgr->loopback);
|
|
|
|
ks_log(KS_LOG_DEBUG, "Session (%s) created\n", blade_session_id_get(bsmgr->loopback));
|
|
|
|
if (blade_session_startup(bsmgr->loopback) != KS_STATUS_SUCCESS) {
|
|
ks_log(KS_LOG_DEBUG, "Session (%s) startup failed\n", blade_session_id_get(bsmgr->loopback));
|
|
blade_session_destroy(&bsmgr->loopback);
|
|
return KS_STATUS_FAIL;
|
|
}
|
|
|
|
ks_log(KS_LOG_DEBUG, "Session (%s) started\n", blade_session_id_get(bsmgr->loopback));
|
|
|
|
blade_sessionmgr_session_add(bsmgr, bsmgr->loopback);
|
|
|
|
blade_session_state_set(bsmgr->loopback, BLADE_SESSION_STATE_STARTUP);
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_shutdown(blade_sessionmgr_t *bsmgr)
|
|
{
|
|
ks_hash_iterator_t *it = NULL;
|
|
|
|
ks_assert(bsmgr);
|
|
|
|
//if (bsmgr->loopback) {
|
|
// blade_session_hangup(bsmgr->loopback);
|
|
// ks_sleep_ms(100);
|
|
//}
|
|
|
|
ks_hash_read_lock(bsmgr->sessions);
|
|
for (it = ks_hash_first(bsmgr->sessions, KS_UNLOCKED); it; it = ks_hash_next(&it)) {
|
|
void *key = NULL;
|
|
blade_session_t *value = NULL;
|
|
|
|
ks_hash_this(it, (const void **)&key, NULL, (void **)&value);
|
|
|
|
blade_session_hangup(value);
|
|
}
|
|
ks_hash_read_unlock(bsmgr->sessions);
|
|
while (ks_hash_count(bsmgr->sessions) > 0) ks_sleep_ms(100);
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(blade_session_t *) blade_sessionmgr_loopback_lookup(blade_sessionmgr_t *bsmgr)
|
|
{
|
|
ks_assert(bsmgr);
|
|
|
|
blade_session_read_lock(bsmgr->loopback, KS_TRUE);
|
|
return bsmgr->loopback;
|
|
}
|
|
|
|
KS_DECLARE(blade_session_t *) blade_sessionmgr_upstream_lookup(blade_sessionmgr_t *bsmgr)
|
|
{
|
|
ks_assert(bsmgr);
|
|
|
|
if (bsmgr->upstream) blade_session_read_lock(bsmgr->upstream, KS_TRUE);
|
|
return bsmgr->upstream;
|
|
}
|
|
|
|
KS_DECLARE(blade_session_t *) blade_sessionmgr_session_lookup(blade_sessionmgr_t *bsmgr, const char *id)
|
|
{
|
|
blade_session_t *bs = NULL;
|
|
|
|
ks_assert(bsmgr);
|
|
ks_assert(id);
|
|
|
|
bs = ks_hash_search(bsmgr->sessions, (void *)id, KS_READLOCKED);
|
|
if (bs) blade_session_read_lock(bs, KS_TRUE);
|
|
ks_hash_read_unlock(bsmgr->sessions);
|
|
|
|
return bs;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_session_add(blade_sessionmgr_t *bsmgr, blade_session_t *bs)
|
|
{
|
|
char *key = NULL;
|
|
|
|
ks_assert(bsmgr);
|
|
ks_assert(bs);
|
|
|
|
key = ks_pstrdup(ks_pool_get(bsmgr), blade_session_id_get(bs));
|
|
ks_hash_insert(bsmgr->sessions, (void *)key, bs);
|
|
|
|
ks_log(KS_LOG_DEBUG, "Session Added: %s\n", key);
|
|
|
|
if (blade_session_upstream(bs)) bsmgr->upstream = bs;
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_session_remove(blade_sessionmgr_t *bsmgr, blade_session_t *bs)
|
|
{
|
|
const char *id = NULL;
|
|
blade_routemgr_t *routemgr = NULL;
|
|
|
|
ks_assert(bsmgr);
|
|
ks_assert(bs);
|
|
|
|
blade_session_write_lock(bs, KS_TRUE);
|
|
|
|
id = blade_session_id_get(bs);
|
|
ks_hash_remove(bsmgr->sessions, (void *)id);
|
|
|
|
ks_log(KS_LOG_DEBUG, "Session Removed: %s\n", id);
|
|
|
|
routemgr = blade_handle_routemgr_get(bsmgr->handle);
|
|
if (blade_session_upstream(bs)) {
|
|
bsmgr->upstream = NULL;
|
|
blade_routemgr_local_set(routemgr, NULL);
|
|
blade_routemgr_master_set(routemgr, NULL);
|
|
|
|
ks_hash_read_lock(bsmgr->sessions);
|
|
for (ks_hash_iterator_t *it = ks_hash_first(bsmgr->sessions, KS_UNLOCKED); it; it = ks_hash_next(&it)) {
|
|
void *key = NULL;
|
|
blade_session_t *value = NULL;
|
|
|
|
ks_hash_this(it, (const void **)&key, NULL, (void **)&value);
|
|
|
|
if (blade_session_loopback(value)) continue;
|
|
|
|
blade_session_hangup(value);
|
|
}
|
|
ks_hash_read_unlock(bsmgr->sessions);
|
|
}
|
|
|
|
blade_session_write_unlock(bs);
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_callback_add(blade_sessionmgr_t *bsmgr, void *data, blade_session_callback_t callback, const char **id)
|
|
{
|
|
ks_pool_t *pool = NULL;
|
|
blade_session_callback_data_t *bscd = NULL;
|
|
uuid_t uuid;
|
|
|
|
ks_assert(bsmgr);
|
|
ks_assert(callback);
|
|
ks_assert(id);
|
|
|
|
pool = ks_pool_get(bsmgr);
|
|
|
|
ks_uuid(&uuid);
|
|
|
|
bscd = ks_pool_alloc(pool, sizeof(blade_session_callback_data_t));
|
|
bscd->id = ks_uuid_str(pool, &uuid);
|
|
bscd->data = data;
|
|
bscd->callback = callback;
|
|
|
|
ks_pool_set_cleanup(bscd, NULL, blade_session_callback_data_cleanup);
|
|
|
|
ks_hash_insert(bsmgr->callbacks, (void *)ks_pstrdup(pool, bscd->id), bscd);
|
|
|
|
*id = bscd->id;
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(ks_status_t) blade_sessionmgr_callback_remove(blade_sessionmgr_t *bsmgr, const char *id)
|
|
{
|
|
ks_assert(bsmgr);
|
|
ks_assert(id);
|
|
|
|
ks_hash_remove(bsmgr->callbacks, (void *)id);
|
|
|
|
return KS_STATUS_SUCCESS;
|
|
}
|
|
|
|
KS_DECLARE(void) blade_sessionmgr_callback_execute(blade_sessionmgr_t *bsmgr, blade_session_t *bs, blade_session_state_condition_t condition)
|
|
{
|
|
ks_assert(bsmgr);
|
|
ks_assert(bs);
|
|
|
|
if (blade_session_state_get(bs) == BLADE_SESSION_STATE_NONE) return;
|
|
|
|
ks_hash_read_lock(bsmgr->callbacks);
|
|
for (ks_hash_iterator_t *it = ks_hash_first(bsmgr->callbacks, KS_UNLOCKED); it; it = ks_hash_next(&it)) {
|
|
void *key = NULL;
|
|
blade_session_callback_data_t *value = NULL;
|
|
|
|
ks_hash_this(it, (const void **)&key, NULL, (void **)&value);
|
|
|
|
value->callback(bs, condition, value->data);
|
|
}
|
|
ks_hash_read_unlock(bsmgr->callbacks);
|
|
}
|
|
|
|
/* For Emacs:
|
|
* Local Variables:
|
|
* mode:c
|
|
* indent-tabs-mode:t
|
|
* tab-width:4
|
|
* c-basic-offset:4
|
|
* End:
|
|
* For VIM:
|
|
* vim:set softtabstop=4 shiftwidth=4 tabstop=4 noet:
|
|
*/
|