implementation session manager
This commit is contained in:
luwenpeng
@@ -6,110 +6,507 @@
|
||||
#include "session_pool.h"
|
||||
#include "session_table.h"
|
||||
#include "session_timer.h"
|
||||
#include "session_queue.h"
|
||||
#include "session_private.h"
|
||||
#include "packet_helpers.h"
|
||||
|
||||
struct session_manager
|
||||
{
|
||||
struct session_pool *sess_pool;
|
||||
struct session_table *sess_table;
|
||||
struct session_timer *sess_timer;
|
||||
struct session_queue *sess_queue; // ready session queue
|
||||
|
||||
session_event_cb event_cb;
|
||||
void *arg;
|
||||
|
||||
struct session *queue_head_ptr;
|
||||
struct session *queue_tail_ptr;
|
||||
// timeout config
|
||||
uint64_t packet_timeout_ms;
|
||||
uint64_t closing_timeout_ms;
|
||||
|
||||
// session number
|
||||
uint64_t tcp_opening_sess_num;
|
||||
uint64_t tcp_closing_sess_num;
|
||||
uint64_t tcp_active_sess_num;
|
||||
|
||||
uint64_t udp_opening_sess_num;
|
||||
uint64_t udp_closing_sess_num;
|
||||
uint64_t udp_active_sess_num;
|
||||
};
|
||||
|
||||
#define HANDSHAKE_TIMEOUT_MS (5 * 1000)
|
||||
#define DATA_TIMEOUT_MS (60 * 1000)
|
||||
|
||||
/******************************************************************************
|
||||
* private API
|
||||
* utils
|
||||
******************************************************************************/
|
||||
|
||||
static void session_manager_handle_new_session(struct session_manager *mgr, struct session_key *key, struct session *sess, const struct packet *pkt)
|
||||
// TODO
|
||||
static uint64_t alloc_session_id(void)
|
||||
{
|
||||
// TODO
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void session_manager_handle_old_session(struct session_manager *mgr, struct session_key *key, struct session *sess, const struct packet *pkt)
|
||||
// TODO
|
||||
struct metadata *packet_get0_metadata(const struct packet *pkt)
|
||||
{
|
||||
// TODO
|
||||
static struct metadata md = {0};
|
||||
return &md;
|
||||
};
|
||||
|
||||
/******************************************************************************
|
||||
* session manager counter
|
||||
******************************************************************************/
|
||||
|
||||
static void update_counter_on_opening(struct session_manager *mgr, struct session *sess)
|
||||
{
|
||||
if (session_get_state(sess) == SESSION_STATE_INIT)
|
||||
{
|
||||
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
||||
{
|
||||
mgr->tcp_opening_sess_num++;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->udp_opening_sess_num++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Enqueue ready session to queue tail
|
||||
static void session_manager_enqueue_ready_session(struct session_manager *mgr, struct session *sess)
|
||||
static void update_counter_on_active(struct session_manager *mgr, struct session *sess)
|
||||
{
|
||||
if (mgr == NULL || sess == NULL)
|
||||
if (session_get_state(sess) == SESSION_STATE_OPENING)
|
||||
{
|
||||
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
||||
{
|
||||
mgr->tcp_opening_sess_num--;
|
||||
mgr->tcp_active_sess_num++;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->udp_opening_sess_num--;
|
||||
mgr->udp_active_sess_num++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void update_counter_on_closing(struct session_manager *mgr, struct session *sess)
|
||||
{
|
||||
if (session_get_state(sess) == SESSION_STATE_OPENING)
|
||||
{
|
||||
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
||||
{
|
||||
mgr->tcp_opening_sess_num--;
|
||||
mgr->tcp_closing_sess_num++;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->udp_opening_sess_num--;
|
||||
mgr->udp_closing_sess_num++;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if (session_get_state(sess) == SESSION_STATE_ACTIVE)
|
||||
{
|
||||
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
||||
{
|
||||
mgr->tcp_active_sess_num--;
|
||||
mgr->tcp_closing_sess_num++;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->udp_active_sess_num--;
|
||||
mgr->udp_closing_sess_num++;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
static void update_counter_on_closed(struct session_manager *mgr, struct session *sess)
|
||||
{
|
||||
if (session_get_state(sess) == SESSION_STATE_CLOSING)
|
||||
{
|
||||
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
||||
{
|
||||
mgr->tcp_closing_sess_num--;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->udp_closing_sess_num--;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* judge session direction
|
||||
******************************************************************************/
|
||||
|
||||
static enum session_dir judge_direction_by_tuple6(const struct tuple6 *key)
|
||||
{
|
||||
// big port is client
|
||||
if (ntohs(key->src_port) > ntohs(key->dst_port))
|
||||
{
|
||||
return SESSION_DIR_C2S;
|
||||
}
|
||||
else
|
||||
{
|
||||
return SESSION_DIR_S2C;
|
||||
}
|
||||
}
|
||||
|
||||
static enum session_dir judge_direction_by_session(const struct session *sess, const struct tuple6 *key)
|
||||
{
|
||||
if (tuple6_cmp(session_get0_tuple6(sess), key) == 0)
|
||||
{
|
||||
return session_get_tuple6_dir(sess);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (session_get_tuple6_dir(sess) == SESSION_DIR_C2S)
|
||||
{
|
||||
return SESSION_DIR_S2C;
|
||||
}
|
||||
else
|
||||
{
|
||||
return SESSION_DIR_C2S;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* update session event and timer
|
||||
******************************************************************************/
|
||||
|
||||
void session_manager_trigger_session_event(struct session_manager *mgr, struct session *sess, uint32_t event)
|
||||
{
|
||||
session_push_event(sess, event);
|
||||
session_queue_push(mgr->sess_queue, sess);
|
||||
}
|
||||
|
||||
void session_manager_update_session_timer(struct session_manager *mgr, struct session *sess, session_expire_cb cb, uint64_t timeout_ms)
|
||||
{
|
||||
session_timer_del_session(mgr->sess_timer, sess);
|
||||
session_set_expirecb(sess, cb, mgr, timestamp_get_msec() + timeout_ms);
|
||||
session_timer_add_session(mgr->sess_timer, sess);
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* expire callback
|
||||
******************************************************************************/
|
||||
|
||||
static void closing_expire_callback(struct session *sess, void *arg)
|
||||
{
|
||||
SESSION_MANAGER_LOG_DEBUG("session %lu closing expire, free session", session_get_id(sess));
|
||||
struct session_manager *mgr = (struct session_manager *)arg;
|
||||
assert(mgr != NULL);
|
||||
|
||||
uint32_t event;
|
||||
while (session_pop_event(sess, &event))
|
||||
{
|
||||
}
|
||||
|
||||
update_counter_on_closed(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_CLOSED);
|
||||
session_set0_cur_pkt(sess, NULL);
|
||||
session_set_cur_dir(sess, SESSION_DIR_NONE);
|
||||
for (uint8_t i = 0; i < EX_DATA_MAX_COUNT; i++)
|
||||
{
|
||||
session_free_ex_data(sess, i);
|
||||
}
|
||||
session_table_delete_session(mgr->sess_table, session_get0_tuple6(sess));
|
||||
session_timer_del_session(mgr->sess_timer, sess);
|
||||
session_pool_free(mgr->sess_pool, sess);
|
||||
}
|
||||
|
||||
static void packet_expire_callback(struct session *sess, void *arg)
|
||||
{
|
||||
SESSION_MANAGER_LOG_DEBUG("session %lu packet expire, trigger closing event", session_get_id(sess));
|
||||
struct session_manager *mgr = (struct session_manager *)arg;
|
||||
assert(mgr != NULL);
|
||||
|
||||
update_counter_on_closing(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_CLOSING);
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_update_session_timer(mgr, sess, closing_expire_callback, mgr->closing_timeout_ms);
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* session ex data
|
||||
******************************************************************************/
|
||||
|
||||
static int tcp_need_closing(uint64_t state)
|
||||
{
|
||||
if ((state & TCP_C2S_FIN_RECVED) && (state & TCP_S2C_FIN_RECVED))
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (state & TCP_C2S_RST_RECVED)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (state & TCP_S2C_RST_RECVED)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int tcp_need_active(uint64_t state)
|
||||
{
|
||||
if ((state & TCP_C2S_PAYLOAD_RECVED) || (state & TCP_S2C_PAYLOAD_RECVED))
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void update_session_base(struct session *sess, const struct packet *pkt, enum session_dir curr_dir)
|
||||
{
|
||||
uint64_t len = packet_get_raw_len(pkt);
|
||||
struct metadata *md = packet_get0_metadata(pkt);
|
||||
if (curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_inc_c2s_metrics(sess, 1, len);
|
||||
if (session_get0_c2s_1st_md(sess) == NULL)
|
||||
{
|
||||
session_set_c2s_1st_md(sess, md);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
session_inc_s2c_metrics(sess, 1, len);
|
||||
if (session_get0_s2c_1st_md(sess) == NULL)
|
||||
{
|
||||
session_set_s2c_1st_md(sess, md);
|
||||
}
|
||||
}
|
||||
session_set_last_time(sess, timestamp_get_msec());
|
||||
session_set0_cur_pkt(sess, pkt);
|
||||
session_set_cur_dir(sess, curr_dir);
|
||||
}
|
||||
|
||||
static void update_tcp_ex_data(struct session *sess, const struct packet *pkt, enum session_dir curr_dir)
|
||||
{
|
||||
uint64_t state = (uint64_t)session_get0_ex_data(sess, tcp_builtin_ex);
|
||||
if (packet_has_tcp_flag_rst(pkt))
|
||||
{
|
||||
if (curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_C2S_RST_RECVED));
|
||||
}
|
||||
else
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_S2C_RST_RECVED));
|
||||
}
|
||||
}
|
||||
|
||||
if (packet_has_tcp_flag_fin(pkt))
|
||||
{
|
||||
if (curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_C2S_FIN_RECVED));
|
||||
}
|
||||
else
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_S2C_FIN_RECVED));
|
||||
}
|
||||
}
|
||||
|
||||
if (packet_has_tcp_flag_syn(pkt))
|
||||
{
|
||||
if (packet_has_tcp_flag_ack(pkt))
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_SYNACK_RECVED));
|
||||
}
|
||||
else
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_SYN_RECVED));
|
||||
}
|
||||
}
|
||||
|
||||
if (packet_has_tcp_flag_ack(pkt) && curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_ACK_RECVED));
|
||||
}
|
||||
|
||||
if (packet_get_tcp_pld_len(pkt) > 0)
|
||||
{
|
||||
if (curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_C2S_PAYLOAD_RECVED));
|
||||
}
|
||||
else
|
||||
{
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_S2C_PAYLOAD_RECVED));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void update_udp_ex_data(struct session *sess, const struct packet *pkt, enum session_dir curr_dir)
|
||||
{
|
||||
uint64_t state = (uint64_t)session_get0_ex_data(sess, udp_builtin_ex);
|
||||
if (curr_dir == SESSION_DIR_C2S)
|
||||
{
|
||||
session_set_ex_data(sess, udp_builtin_ex, (void *)(state | UDP_C2S_RECVED));
|
||||
}
|
||||
else
|
||||
{
|
||||
session_set_ex_data(sess, udp_builtin_ex, (void *)(state | UDP_S2C_RECVED));
|
||||
}
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* handle session
|
||||
******************************************************************************/
|
||||
|
||||
// return 0: success
|
||||
// return -1: tcp not syn packet, discard
|
||||
static int handle_tcp_new_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
if (!packet_has_tcp_flag_syn(pkt))
|
||||
{
|
||||
// not syn packet, discard
|
||||
return -1;
|
||||
}
|
||||
|
||||
enum session_dir curr_dir = SESSION_DIR_NONE;
|
||||
|
||||
session_init(sess);
|
||||
// syn packet
|
||||
if (!packet_has_tcp_flag_ack(pkt))
|
||||
{
|
||||
curr_dir = SESSION_DIR_C2S;
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)TCP_SYN_RECVED);
|
||||
}
|
||||
// syn ack packet
|
||||
else
|
||||
{
|
||||
curr_dir = SESSION_DIR_S2C;
|
||||
session_set_ex_data(sess, tcp_builtin_ex, (void *)TCP_SYNACK_RECVED);
|
||||
}
|
||||
|
||||
session_set_id(sess, alloc_session_id());
|
||||
session_set_tuple6(sess, key);
|
||||
session_set_tuple6_dir(sess, curr_dir);
|
||||
|
||||
session_set_type(sess, SESSION_TYPE_TCP);
|
||||
update_counter_on_opening(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_OPENING);
|
||||
|
||||
session_set_create_time(sess, timestamp_get_msec());
|
||||
update_session_base(sess, pkt, curr_dir);
|
||||
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_OPENING);
|
||||
session_manager_update_session_timer(mgr, sess, packet_expire_callback, mgr->packet_timeout_ms);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
// always return 0
|
||||
static int handle_udp_new_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
enum session_dir curr_dir = judge_direction_by_tuple6(key);
|
||||
|
||||
session_init(sess);
|
||||
update_udp_ex_data(sess, pkt, curr_dir);
|
||||
session_set_id(sess, alloc_session_id());
|
||||
session_set_tuple6(sess, key);
|
||||
session_set_tuple6_dir(sess, curr_dir);
|
||||
|
||||
/*
|
||||
* when a UDP Session is created, the Opening and active events are triggered,
|
||||
* (the plugin is called twice by the opening/active events in turn),
|
||||
* and the state of the UDP session is directly switched to the active state.
|
||||
*/
|
||||
session_set_type(sess, SESSION_TYPE_UDP);
|
||||
update_counter_on_opening(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_OPENING);
|
||||
update_counter_on_active(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_ACTIVE);
|
||||
|
||||
session_set_create_time(sess, timestamp_get_msec());
|
||||
update_session_base(sess, pkt, curr_dir);
|
||||
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_OPENING);
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_ACTIVE);
|
||||
session_manager_update_session_timer(mgr, sess, packet_expire_callback, mgr->packet_timeout_ms);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void handle_tcp_old_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
enum session_dir curr_dir = judge_direction_by_session(sess, key);
|
||||
update_tcp_ex_data(sess, pkt, curr_dir);
|
||||
update_session_base(sess, pkt, curr_dir);
|
||||
|
||||
uint64_t state = (uint64_t)session_get0_ex_data(sess, tcp_builtin_ex);
|
||||
if (tcp_need_closing(state))
|
||||
{
|
||||
update_counter_on_closing(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_CLOSING);
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_update_session_timer(mgr, sess, closing_expire_callback, mgr->closing_timeout_ms);
|
||||
return;
|
||||
}
|
||||
|
||||
if (tcp_need_active(state))
|
||||
{
|
||||
update_counter_on_active(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_ACTIVE);
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_ACTIVE);
|
||||
session_manager_update_session_timer(mgr, sess, packet_expire_callback, mgr->packet_timeout_ms);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
static void handle_udp_old_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
enum session_dir curr_dir = judge_direction_by_session(sess, key);
|
||||
update_udp_ex_data(sess, pkt, curr_dir);
|
||||
update_session_base(sess, pkt, curr_dir);
|
||||
|
||||
update_counter_on_active(mgr, sess);
|
||||
session_set_state(sess, SESSION_STATE_ACTIVE);
|
||||
session_manager_trigger_session_event(mgr, sess, SESSION_EVENT_ACTIVE);
|
||||
session_manager_update_session_timer(mgr, sess, packet_expire_callback, mgr->packet_timeout_ms);
|
||||
}
|
||||
|
||||
// return 0: success
|
||||
// return -1: tcp not syn packet, discard
|
||||
static int handle_new_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
if (packet_has_tcp(pkt))
|
||||
{
|
||||
return handle_tcp_new_session(mgr, key, sess, pkt);
|
||||
}
|
||||
else
|
||||
{
|
||||
return handle_udp_new_session(mgr, key, sess, pkt);
|
||||
}
|
||||
}
|
||||
|
||||
static void handle_old_session(struct session_manager *mgr, struct tuple6 *key, struct session *sess, const struct packet *pkt)
|
||||
{
|
||||
if (session_get_state(sess) == SESSION_STATE_CLOSING)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (mgr->queue_head_ptr == NULL)
|
||||
if (packet_has_tcp(pkt))
|
||||
{
|
||||
mgr->queue_head_ptr = sess;
|
||||
mgr->queue_tail_ptr = sess;
|
||||
sess->next_ready_ptr = NULL;
|
||||
handle_tcp_old_session(mgr, key, sess, pkt);
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->queue_tail_ptr->next_ready_ptr = sess;
|
||||
mgr->queue_tail_ptr = sess;
|
||||
sess->next_ready_ptr = NULL;
|
||||
handle_udp_old_session(mgr, key, sess, pkt);
|
||||
}
|
||||
}
|
||||
|
||||
// Dequeue ready session from queue head
|
||||
struct session *session_manager_ready_session_dequeue(struct session_manager *mgr)
|
||||
{
|
||||
if (mgr == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
struct session *sess = mgr->queue_head_ptr;
|
||||
if (sess == NULL)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (mgr->queue_head_ptr == mgr->queue_tail_ptr)
|
||||
{
|
||||
mgr->queue_head_ptr = NULL;
|
||||
mgr->queue_tail_ptr = NULL;
|
||||
}
|
||||
else
|
||||
{
|
||||
mgr->queue_head_ptr = sess->next_ready_ptr;
|
||||
}
|
||||
|
||||
sess->next_ready_ptr = NULL;
|
||||
return sess;
|
||||
}
|
||||
|
||||
// handshake expire set to discard
|
||||
static void handshake_expire_cb(struct session *sess, void *arg)
|
||||
{
|
||||
struct session_manager *mgr = (struct session_manager *)arg;
|
||||
assert(mgr != NULL);
|
||||
|
||||
session_set_state(sess, SESSION_STATE_DISCARD);
|
||||
session_push_event(sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_enqueue_ready_session(mgr, sess);
|
||||
}
|
||||
|
||||
// data expire set to closing
|
||||
static void data_expire_cb(struct session *sess, void *arg)
|
||||
{
|
||||
struct session_manager *mgr = (struct session_manager *)arg;
|
||||
assert(mgr != NULL);
|
||||
|
||||
session_set_state(sess, SESSION_STATE_CLOSING);
|
||||
session_push_event(sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_enqueue_ready_session(mgr, sess);
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* public API
|
||||
******************************************************************************/
|
||||
@@ -140,8 +537,25 @@ struct session_manager *session_manager_create(uint64_t max_session_num)
|
||||
goto error;
|
||||
}
|
||||
|
||||
mgr->queue_head_ptr = NULL;
|
||||
mgr->queue_tail_ptr = NULL;
|
||||
mgr->sess_queue = session_queue_create();
|
||||
if (mgr->sess_queue == NULL)
|
||||
{
|
||||
goto error;
|
||||
}
|
||||
|
||||
mgr->closing_timeout_ms = 2 * 1000;
|
||||
mgr->packet_timeout_ms = 5 * 1000;
|
||||
|
||||
mgr->tcp_opening_sess_num = 0;
|
||||
mgr->tcp_closing_sess_num = 0;
|
||||
mgr->tcp_active_sess_num = 0;
|
||||
|
||||
mgr->udp_opening_sess_num = 0;
|
||||
mgr->udp_closing_sess_num = 0;
|
||||
mgr->udp_active_sess_num = 0;
|
||||
|
||||
tcp_builtin_ex = session_get_ex_new_index("tcp_builtin_ex", NULL, NULL);
|
||||
udp_builtin_ex = session_get_ex_new_index("udp_builtin_ex", NULL, NULL);
|
||||
|
||||
return mgr;
|
||||
|
||||
@@ -154,6 +568,7 @@ void session_manager_destroy(struct session_manager *mgr)
|
||||
{
|
||||
if (mgr)
|
||||
{
|
||||
session_queue_destroy(mgr->sess_queue);
|
||||
session_timer_destroy(mgr->sess_timer);
|
||||
session_table_destroy(mgr->sess_table);
|
||||
session_pool_destroy(mgr->sess_pool);
|
||||
@@ -168,31 +583,60 @@ void session_manager_set_session_eventcb(struct session_manager *mgr, session_ev
|
||||
mgr->arg = arg;
|
||||
}
|
||||
|
||||
void session_manager_set_packet_timeout(struct session_manager *mgr, uint64_t timeout_ms)
|
||||
{
|
||||
mgr->packet_timeout_ms = timeout_ms;
|
||||
}
|
||||
|
||||
void session_manager_set_closing_timeout(struct session_manager *mgr, uint64_t timeout_ms)
|
||||
{
|
||||
mgr->closing_timeout_ms = timeout_ms;
|
||||
}
|
||||
|
||||
// return NULL: discard
|
||||
// * tuple6 not find
|
||||
// * tcp first packet not syn
|
||||
struct session *session_manager_find_session(struct session_manager *mgr, const struct packet *pkt)
|
||||
{
|
||||
struct session_key key;
|
||||
// TODO packet to key
|
||||
struct tuple6 key;
|
||||
if (packet_get_innermost_tuple6(pkt, &key) == -1)
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
struct session *sess = session_table_find_session(mgr->sess_table, &key);
|
||||
if (sess == NULL)
|
||||
{
|
||||
// if session pool is full, discard oldest session
|
||||
if (session_pool_get_count(mgr->sess_pool) == 1)
|
||||
{
|
||||
struct session *oldest_sess = session_table_find_oldest_session(mgr->sess_table);
|
||||
assert(oldest_sess == NULL);
|
||||
session_set_state(oldest_sess, SESSION_STATE_DISCARD);
|
||||
session_push_event(oldest_sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_enqueue_ready_session(mgr, oldest_sess);
|
||||
struct session *unused_sess = session_table_find_least_recently_unused_session(mgr->sess_table);
|
||||
assert(unused_sess);
|
||||
|
||||
update_counter_on_closing(mgr, unused_sess);
|
||||
session_set_state(unused_sess, SESSION_STATE_DISCARD);
|
||||
session_manager_trigger_session_event(mgr, unused_sess, SESSION_EVENT_CLOSING);
|
||||
session_manager_update_session_timer(mgr, unused_sess, closing_expire_callback, mgr->closing_timeout_ms);
|
||||
}
|
||||
|
||||
sess = session_pool_alloc(mgr->sess_pool);
|
||||
assert(sess != NULL);
|
||||
|
||||
session_manager_handle_new_session(mgr, &key, sess, pkt);
|
||||
// return 0: success
|
||||
// return -1: tcp not syn packet, discard
|
||||
if (handle_new_session(mgr, &key, sess, pkt) == 0)
|
||||
{
|
||||
session_table_add_session(mgr->sess_table, &key, sess);
|
||||
}
|
||||
else
|
||||
{
|
||||
session_pool_free(mgr->sess_pool, sess);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
session_manager_handle_old_session(mgr, &key, sess, pkt);
|
||||
handle_old_session(mgr, &key, sess, pkt);
|
||||
}
|
||||
|
||||
return sess;
|
||||
@@ -206,19 +650,24 @@ void session_manager_dispatch(struct session_manager *mgr)
|
||||
void *cb_arg = mgr->arg;
|
||||
session_event_cb event_cb = mgr->event_cb;
|
||||
|
||||
while (1)
|
||||
SESSION_MANAGER_LOG_DEBUG("current timestamp: %lu s", timestamp_get_sec());
|
||||
|
||||
// limit expire session number
|
||||
for (int i = 0; i < 100; i++)
|
||||
{
|
||||
sess = session_timer_expire_session(mgr->sess_timer, timestamp_get_msec());
|
||||
if (sess == NULL)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
session_run_expirecb(sess);
|
||||
}
|
||||
|
||||
while (1)
|
||||
{
|
||||
sess = session_manager_ready_session_dequeue(mgr);
|
||||
// get session from ready queue
|
||||
sess = session_queue_pop(mgr->sess_queue);
|
||||
if (sess == NULL)
|
||||
{
|
||||
break;
|
||||
@@ -226,15 +675,49 @@ void session_manager_dispatch(struct session_manager *mgr)
|
||||
|
||||
while (1)
|
||||
{
|
||||
// get event from session
|
||||
if (session_pop_event(sess, &event) == false)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
SESSION_MANAGER_LOG_DEBUG("handle \"%s\" event on session %lu", session_event_tostring((enum session_event)event), session_get_id(sess));
|
||||
if (event_cb)
|
||||
{
|
||||
event_cb(sess, event, cb_arg);
|
||||
}
|
||||
}
|
||||
session_set0_cur_pkt(sess, NULL);
|
||||
session_set_cur_dir(sess, SESSION_DIR_NONE);
|
||||
};
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_tcp_opening_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->tcp_opening_sess_num;
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_tcp_closing_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->tcp_closing_sess_num;
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_tcp_active_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->tcp_active_sess_num;
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_udp_opening_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->udp_opening_sess_num;
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_udp_closing_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->udp_closing_sess_num;
|
||||
}
|
||||
|
||||
uint64_t session_manager_get_udp_active_sess_num(struct session_manager *mgr)
|
||||
{
|
||||
return mgr->udp_active_sess_num;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user