2023-12-13 19:20:34 +08:00
|
|
|
#include <stdlib.h>
|
|
|
|
|
#include <assert.h>
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
#include "stellar.h"
|
2023-12-13 19:20:34 +08:00
|
|
|
#include "timestamp.h"
|
2024-01-17 11:47:55 +08:00
|
|
|
#include "session_private.h"
|
2023-12-13 19:20:34 +08:00
|
|
|
#include "session_pool.h"
|
|
|
|
|
#include "session_table.h"
|
|
|
|
|
#include "session_timer.h"
|
2023-12-19 10:47:26 +08:00
|
|
|
#include "session_queue.h"
|
2024-01-17 11:47:55 +08:00
|
|
|
#include "session_manager.h"
|
2024-02-21 11:49:20 +08:00
|
|
|
#include "tcp_utils.h"
|
|
|
|
|
#include "udp_utils.h"
|
2024-03-09 19:28:14 +08:00
|
|
|
#include "duplicated_packet_filter.h"
|
|
|
|
|
#include "evicted_session_filter.h"
|
2024-01-31 14:45:50 +08:00
|
|
|
#include "id_generator.h"
|
2023-12-13 19:20:34 +08:00
|
|
|
|
|
|
|
|
struct session_manager
|
|
|
|
|
{
|
|
|
|
|
struct session_pool *sess_pool;
|
2024-01-17 11:47:55 +08:00
|
|
|
struct session_table *tcp_sess_table;
|
|
|
|
|
struct session_table *udp_sess_table;
|
2023-12-13 19:20:34 +08:00
|
|
|
struct session_timer *sess_timer;
|
2024-01-17 11:47:55 +08:00
|
|
|
struct session_queue *sess_evicted_queue;
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
struct duplicated_packet_filter *tcp_dup_pkt_filter;
|
|
|
|
|
struct evicted_session_filter *udp_evc_sess_filter;
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
struct session_manager_options opts;
|
2023-12-13 19:20:34 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
/***************************************************************
|
|
|
|
|
* session manager status
|
|
|
|
|
***************************************************************/
|
2023-12-19 10:47:26 +08:00
|
|
|
|
|
|
|
|
// session number
|
2024-01-17 11:47:55 +08:00
|
|
|
uint64_t tcp_sess_num;
|
2023-12-19 10:47:26 +08:00
|
|
|
uint64_t tcp_opening_sess_num;
|
|
|
|
|
uint64_t tcp_active_sess_num;
|
2024-01-17 11:47:55 +08:00
|
|
|
uint64_t tcp_closing_sess_num;
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
uint64_t udp_sess_num;
|
2023-12-19 10:47:26 +08:00
|
|
|
uint64_t udp_opening_sess_num;
|
|
|
|
|
uint64_t udp_active_sess_num;
|
2024-01-17 11:47:55 +08:00
|
|
|
uint64_t udp_closing_sess_num;
|
|
|
|
|
|
2024-01-26 14:13:54 +08:00
|
|
|
uint64_t tcp_overload_evict_old_sess_num;
|
|
|
|
|
uint64_t tcp_overload_evict_new_sess_num;
|
|
|
|
|
uint64_t udp_overload_evict_old_sess_num;
|
|
|
|
|
uint64_t udp_overload_evict_new_sess_num;
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
// packet filter status
|
|
|
|
|
uint64_t npkts_miss_l4_proto; // fast forward
|
|
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
uint64_t npkts_hit_tcp_miss_sess; // fast forward
|
|
|
|
|
uint64_t npkts_hit_tcp_dupkt; // fast forward
|
|
|
|
|
uint64_t npkts_hit_tcp_discard; // drop
|
2024-01-17 11:47:55 +08:00
|
|
|
|
|
|
|
|
uint64_t npkts_hit_udp_evicted; // fast forward
|
2023-12-13 19:20:34 +08:00
|
|
|
};
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
enum packet_flag
|
|
|
|
|
{
|
|
|
|
|
NONE = 0,
|
|
|
|
|
TCP_SYN = 1 << 0,
|
|
|
|
|
TCP_SYN_ACK = 1 << 1,
|
|
|
|
|
TCP_FIN = 1 << 2,
|
|
|
|
|
TCP_RST = 1 << 3,
|
|
|
|
|
TCP_DATA = 1 << 4,
|
|
|
|
|
UDP_DATA = 1 << 5,
|
|
|
|
|
TIMEOUT = 1 << 6,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
struct session_finite_state_machine
|
|
|
|
|
{
|
|
|
|
|
int flag_mask;
|
|
|
|
|
int next_state;
|
|
|
|
|
} sess_fsm[SESSION_STATE_MAX][8];
|
|
|
|
|
|
|
|
|
|
void session_finite_state_machine_init()
|
|
|
|
|
{
|
|
|
|
|
// SESSION_STATE_INIT -> SESSION_STATE_OPENING
|
|
|
|
|
// SESSION_STATE_INIT -> SESSION_STATE_INIT
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_INIT][0].flag_mask = TCP_SYN | TCP_SYN_ACK | UDP_DATA;
|
|
|
|
|
sess_fsm[SESSION_STATE_INIT][0].next_state = SESSION_STATE_OPENING;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_INIT][1].flag_mask = NONE;
|
|
|
|
|
sess_fsm[SESSION_STATE_INIT][1].next_state = SESSION_STATE_INIT;
|
|
|
|
|
|
|
|
|
|
// SESSION_STATE_OPENING -> SESSION_STATE_ACTIVE
|
|
|
|
|
// SESSION_STATE_OPENING -> SESSION_STATE_CLOSING
|
|
|
|
|
// SESSION_STATE_OPENING -> SESSION_STATE_CLOSED
|
|
|
|
|
// SESSION_STATE_OPENING -> SESSION_STATE_OPENING
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][0].flag_mask = TCP_DATA | UDP_DATA;
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][0].next_state = SESSION_STATE_ACTIVE;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][1].flag_mask = TCP_FIN | TCP_RST;
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][1].next_state = SESSION_STATE_CLOSING;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][2].flag_mask = TIMEOUT;
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][2].next_state = SESSION_STATE_CLOSED;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][3].flag_mask = NONE;
|
|
|
|
|
sess_fsm[SESSION_STATE_OPENING][3].next_state = SESSION_STATE_OPENING;
|
|
|
|
|
|
|
|
|
|
// SESSION_STATE_ACTIVE -> SESSION_STATE_CLOSING
|
|
|
|
|
// SESSION_STATE_ACTIVE -> SESSION_STATE_ACTIVE
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_ACTIVE][0].flag_mask = TCP_FIN | TCP_RST | TIMEOUT;
|
|
|
|
|
sess_fsm[SESSION_STATE_ACTIVE][0].next_state = SESSION_STATE_CLOSING;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_ACTIVE][1].flag_mask = NONE;
|
|
|
|
|
sess_fsm[SESSION_STATE_ACTIVE][1].next_state = SESSION_STATE_ACTIVE;
|
|
|
|
|
|
|
|
|
|
// SESSION_STATE_CLOSING -> SESSION_STATE_CLOSED
|
|
|
|
|
// SESSION_STATE_CLOSING -> SESSION_STATE_CLOSING
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_CLOSING][0].flag_mask = TIMEOUT;
|
|
|
|
|
sess_fsm[SESSION_STATE_CLOSING][0].next_state = SESSION_STATE_CLOSED;
|
|
|
|
|
|
|
|
|
|
sess_fsm[SESSION_STATE_CLOSING][1].flag_mask = NONE;
|
|
|
|
|
sess_fsm[SESSION_STATE_CLOSING][1].next_state = SESSION_STATE_CLOSING;
|
|
|
|
|
|
|
|
|
|
// SESSION_STATE_CLOSED -> SESSION_STATE_INIT
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int session_finite_state_machine_run(enum session_state curr_state, int input_pkt_flag)
|
|
|
|
|
{
|
|
|
|
|
struct session_finite_state_machine *list = sess_fsm[curr_state];
|
|
|
|
|
for (int i = 0; i < 8; i++)
|
|
|
|
|
{
|
|
|
|
|
int mask = list[i].flag_mask;
|
|
|
|
|
if (mask & input_pkt_flag)
|
|
|
|
|
{
|
|
|
|
|
return list[i].next_state;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (mask == NONE)
|
|
|
|
|
{
|
|
|
|
|
return list[i].next_state;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
assert(0);
|
|
|
|
|
|
|
|
|
|
return curr_state;
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_init_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
static inline void tcp_handshake_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
static inline void tcp_data_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
static inline void tcp_half_closed_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
static inline void tcp_time_wait_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
static inline void udp_data_timeout_cb(struct session *sess, void *arg);
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
static inline int session_manager_check_options(struct session_manager_options *opts);
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline int session_manager_update_tcp_filter(struct session_manager *mgr, struct session *sess, const struct packet *pkt, enum session_dir curr_dir);
|
|
|
|
|
|
|
|
|
|
static inline enum session_dir judge_direction_by_tuple6(const struct tuple6 *key);
|
|
|
|
|
static inline enum session_dir judge_direction_by_session(const struct session *sess, const struct tuple6 *key);
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
static inline void session_update_tcp_state(struct session *sess, const struct layer *tcp_layer, enum session_dir curr_dir);
|
|
|
|
|
static inline void session_update_udp_state(struct session *sess, const struct layer *udp_layer, enum session_dir curr_dir);
|
2024-01-31 14:45:50 +08:00
|
|
|
|
|
|
|
|
static inline void session_manager_update_session_state(struct session_manager *mgr, struct session *sess, enum session_state state);
|
|
|
|
|
static inline void session_manager_update_session_timer(struct session_manager *mgr, struct session *sess, session_expire_cb expire_cb, uint64_t timeout_sec);
|
|
|
|
|
static inline void session_manager_update_session_base(struct session_manager *mgr, struct session *sess, const struct tuple6 *key, enum session_dir curr_dir);
|
|
|
|
|
static inline void session_manager_update_session_packet(struct session_manager *mgr, struct session *sess, const struct packet *pkt, enum session_dir curr_dir);
|
|
|
|
|
|
|
|
|
|
static inline void session_manager_update_udp_to_opening(struct session_manager *mgr, struct session *sess);
|
|
|
|
|
static inline void session_manager_update_udp_to_active(struct session_manager *mgr, struct session *sess);
|
|
|
|
|
static inline void session_manager_update_udp_to_closing(struct session_manager *mgr, struct session *sess);
|
|
|
|
|
static inline void session_manager_update_tcp_to_opening(struct session_manager *mgr, struct session *sess, int opening_by_syn);
|
|
|
|
|
static inline void session_manager_update_tcp_to_active(struct session_manager *mgr, struct session *sess);
|
|
|
|
|
static inline void session_manager_update_tcp_to_closing(struct session_manager *mgr, struct session *sess, int enable_time_wait);
|
|
|
|
|
|
|
|
|
|
static inline void session_manager_handle_tcp_on_opening(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state);
|
|
|
|
|
static inline void session_manager_handle_tcp_on_active(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state);
|
|
|
|
|
static inline void session_manager_handle_tcp_on_closing(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state);
|
|
|
|
|
|
|
|
|
|
static inline struct session *session_manager_new_tcp_session(struct session_manager *mgr, const struct packet *pkt, const struct tuple6 *key);
|
|
|
|
|
static inline struct session *session_manager_new_udp_session(struct session_manager *mgr, const struct packet *pkt, const struct tuple6 *key);
|
2024-03-08 18:10:38 +08:00
|
|
|
static inline int session_manager_update_tcp_session(struct session_manager *mgr, struct session *sess, const struct packet *pkt);
|
|
|
|
|
static inline int session_manager_update_udp_session(struct session_manager *mgr, struct session *sess, const struct packet *pkt);
|
2024-01-31 14:45:50 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
void session_manager_free_session(struct session_manager *mgr, struct session *sess);
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_evicte_session(struct session_manager *mgr, struct session *sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
|
2023-12-19 10:47:26 +08:00
|
|
|
/******************************************************************************
|
2024-01-23 14:30:46 +08:00
|
|
|
* Private API
|
2023-12-19 10:47:26 +08:00
|
|
|
******************************************************************************/
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_init_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run tcp_init_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
2024-03-08 13:33:49 +08:00
|
|
|
if (session_get_closing_reason(sess) == 0)
|
2024-01-23 18:23:56 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_TIMEOUT);
|
2024-01-23 18:23:56 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_handshake_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run tcp_handshake_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
2024-03-08 13:33:49 +08:00
|
|
|
if (session_get_closing_reason(sess) == 0)
|
2024-01-23 18:23:56 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_TIMEOUT);
|
2024-01-23 18:23:56 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_data_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run tcp_data_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
2024-03-08 13:33:49 +08:00
|
|
|
if (session_get_closing_reason(sess) == 0)
|
2024-01-23 18:23:56 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_TIMEOUT);
|
2024-01-23 18:23:56 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_half_closed_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run tcp_half_closed_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
2024-03-08 13:33:49 +08:00
|
|
|
if (session_get_closing_reason(sess) == 0)
|
2024-01-23 18:23:56 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_TIMEOUT);
|
2024-01-23 18:23:56 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void tcp_time_wait_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run tcp_time_wait_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
|
|
|
|
session_manager_free_session(mgr, sess);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void udp_data_timeout_cb(struct session *sess, void *arg)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("run udp_data_timeout_cb on session %lu", session_get_id(sess));
|
|
|
|
|
struct session_manager *mgr = (struct session_manager *)arg;
|
|
|
|
|
assert(mgr != NULL);
|
|
|
|
|
|
2024-03-08 13:33:49 +08:00
|
|
|
if (session_get_closing_reason(sess) == 0)
|
2024-01-23 18:23:56 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_TIMEOUT);
|
2024-01-23 18:23:56 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_udp_to_closing(mgr, sess);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
// return 0: success
|
2024-03-08 14:51:21 +08:00
|
|
|
// return -1: invalid opts
|
|
|
|
|
static inline int session_manager_check_options(struct session_manager_options *opts)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts == NULL)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
SESSION_LOG_ERROR("invalid opts");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// max session number
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->max_tcp_session_num < 2)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid max tcp session number");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->max_udp_session_num < 2)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid max udp session number");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-26 14:13:54 +08:00
|
|
|
// session overload
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_overload_evict_old_sess != 0 && opts->tcp_overload_evict_old_sess != 1)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp overload evict old session, support range: 0-1");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->udp_overload_evict_old_sess != 0 && opts->udp_overload_evict_old_sess != 1)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid udp overload evict old session, support range: 0-1");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
// TCP timeout opts
|
|
|
|
|
if (opts->tcp_timeout_init < 1 || opts->tcp_timeout_init > 60)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout init, support range: 1-60");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_timeout_handshake < 1 || opts->tcp_timeout_handshake > 60)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout handshake, support range: 1-60");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_timeout_data < 1 || opts->tcp_timeout_data > 15999999)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout data, support range: 1-15,999,999");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_timeout_half_closed < 1 || opts->tcp_timeout_half_closed > 604800)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout half closed, support range: 1-604,800");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_timeout_time_wait < 1 || opts->tcp_timeout_time_wait > 600)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout time wait, support range: 1-600");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-08 14:51:21 +08:00
|
|
|
if (opts->tcp_timeout_discard < 1 || opts->tcp_timeout_discard > 15999999)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid tcp timeout discard, support range: 1-15,999,999");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
// UDP timeout opts
|
|
|
|
|
if (opts->udp_timeout_data < 1 || opts->udp_timeout_data > 15999999)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_ERROR("invalid udp timeout data, support range: 1-15,999,999");
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
// duplicate packet filter opts
|
|
|
|
|
if (opts->duplicated_packet_filter_enable != 0 && opts->duplicated_packet_filter_enable != 1)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid duplicate packet filter enable, support range: 0-1");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->duplicated_packet_filter_enable)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->duplicated_packet_filter_capacity == 0)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid duplicate packet filter capacity");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->duplicated_packet_filter_timeout < 1 || opts->duplicated_packet_filter_timeout > 60)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid duplicate packet filter timeout, support range: 1-60");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->duplicated_packet_filter_error_rate < 0 || opts->duplicated_packet_filter_error_rate > 1)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid duplicate packet filter error rate, support range: 0-1");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
// eviction filter opts
|
|
|
|
|
if (opts->evicted_session_filter_enable != 0 && opts->evicted_session_filter_enable != 1)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid eviction filter enable, support range: 0-1");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->evicted_session_filter_enable)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->evicted_session_filter_capacity == 0)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid eviction filter capacity");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->evicted_session_filter_timeout < 1 || opts->evicted_session_filter_timeout > 60)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid eviction filter timeout, support range: 1-60");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
2024-03-09 19:28:14 +08:00
|
|
|
if (opts->evicted_session_filter_error_rate < 0 || opts->evicted_session_filter_error_rate > 1)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
SESSION_LOG_ERROR("invalid eviction filter error rate, support range: 0-1");
|
2024-01-17 11:47:55 +08:00
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// return 1: duplicate packet
|
|
|
|
|
// return 0: not duplicate packet
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline int session_manager_update_tcp_filter(struct session_manager *mgr, struct session *sess, const struct packet *pkt, enum session_dir curr_dir)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
2023-12-22 14:45:31 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
if (session_get_c2s_packets(sess) < 3)
|
|
|
|
|
{
|
|
|
|
|
goto dupkt_fitler;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
|
|
|
|
{
|
|
|
|
|
if (session_get_s2c_packets(sess) < 3)
|
|
|
|
|
{
|
|
|
|
|
goto dupkt_fitler;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (session_get_dup_traffic_flag(sess) == DUP_TRAFFIC_YES)
|
|
|
|
|
{
|
|
|
|
|
goto dupkt_fitler;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
dupkt_fitler:
|
2024-03-09 19:28:14 +08:00
|
|
|
if (duplicated_packet_filter_lookup(mgr->tcp_dup_pkt_filter, pkt, timestamp_get_sec()))
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
duplicated_packet_filter_add(mgr->tcp_dup_pkt_filter, pkt, timestamp_get_sec());
|
2024-01-17 11:47:55 +08:00
|
|
|
return 0;
|
2023-12-22 14:45:31 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline enum session_dir judge_direction_by_tuple6(const struct tuple6 *key)
|
2023-12-22 14:45:31 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
uint16_t src_port = ntohs(key->src_port);
|
|
|
|
|
uint16_t dst_port = ntohs(key->dst_port);
|
2023-12-13 19:20:34 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// big port is client
|
|
|
|
|
if (src_port > dst_port)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
return SESSION_DIR_C2S;
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
else if (src_port < dst_port)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
return SESSION_DIR_S2C;
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
else
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
// if port is equal, first packet is C2S
|
|
|
|
|
return SESSION_DIR_C2S;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline enum session_dir judge_direction_by_session(const struct session *sess, const struct tuple6 *key)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
if (tuple6_cmp(session_get0_key(sess), key) == 0)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
return session_get_key_dir(sess);
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
else
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
return (session_get_key_dir(sess) == SESSION_DIR_C2S ? SESSION_DIR_S2C : SESSION_DIR_C2S);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
static inline void session_update_tcp_state(struct session *sess, const struct layer *tcp_layer, enum session_dir curr_dir)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
const struct tcphdr *hdr = (const struct tcphdr *)tcp_layer->hdr_ptr;
|
2024-01-23 14:30:46 +08:00
|
|
|
uint64_t state = session_get_tcp_state(sess);
|
2024-02-21 11:49:20 +08:00
|
|
|
if (tcp_hdr_get_syn_flag(hdr))
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-02-21 11:49:20 +08:00
|
|
|
state |= (tcp_hdr_get_ack_flag(hdr) ? TCP_SYNACK_RECVED : TCP_SYN_RECVED);
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
else
|
|
|
|
|
{
|
2024-02-21 11:49:20 +08:00
|
|
|
if (tcp_hdr_get_ack_flag(hdr))
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
|
|
|
|
{
|
|
|
|
|
state |= TCP_C2S_ACK_RECVED;
|
|
|
|
|
}
|
|
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
|
|
|
|
{
|
|
|
|
|
state |= TCP_S2C_ACK_RECVED;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-02-21 11:49:20 +08:00
|
|
|
if (tcp_hdr_get_fin_flag(hdr))
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_C2S_FIN_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_S2C_FIN_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
2024-02-21 11:49:20 +08:00
|
|
|
if (tcp_hdr_get_rst_flag(hdr))
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_C2S_RST_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_S2C_RST_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
if (tcp_layer->pld_len > 0)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_C2S_DATA_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
state |= TCP_S2C_DATA_RECVED;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
session_set_tcp_state(sess, (enum tcp_state)state);
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
static inline void session_update_udp_state(struct session *sess, const struct layer *udp_layer, enum session_dir curr_dir)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
uint64_t state = session_get_udp_state(sess);
|
2024-01-17 11:47:55 +08:00
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
|
|
|
|
{
|
|
|
|
|
state |= UDP_C2S_RECVED;
|
|
|
|
|
}
|
|
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
|
|
|
|
{
|
|
|
|
|
state |= UDP_S2C_RECVED;
|
|
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
session_set_udp_state(sess, (enum udp_state)state);
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
/*
|
|
|
|
|
on opening update session
|
|
|
|
|
[*] session_init
|
|
|
|
|
[*] session_set_id
|
2024-01-23 14:30:46 +08:00
|
|
|
[*] session_set_key
|
|
|
|
|
[*] session_set_key_dir
|
2024-01-17 11:47:55 +08:00
|
|
|
[*] session_set_type
|
2024-03-08 14:20:36 +08:00
|
|
|
[*] session_set_new_time
|
2024-01-17 11:47:55 +08:00
|
|
|
[*] session_set_state
|
|
|
|
|
|
|
|
|
|
on packet update session
|
|
|
|
|
[*] session_inc_c2s_metrics
|
|
|
|
|
[*] session_inc_s2c_metrics
|
|
|
|
|
[*] session_set_c2s_1st_pkt
|
|
|
|
|
[*] session_set_s2c_1st_pkt
|
|
|
|
|
[*] session_set_c2s_1st_pkt_md
|
|
|
|
|
[*] session_set_s2c_1st_pkt_md
|
|
|
|
|
[*] session_set0_cur_pkt
|
|
|
|
|
[*] session_set_cur_dir
|
|
|
|
|
[*] session_set_last_time
|
|
|
|
|
session_set_state
|
|
|
|
|
session_set_dup_traffic_flag
|
|
|
|
|
|
|
|
|
|
on closing update session
|
|
|
|
|
[*] session_set_state
|
2024-03-08 13:33:49 +08:00
|
|
|
[*] session_set_closing_reason
|
2024-01-17 11:47:55 +08:00
|
|
|
*/
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_session_state(struct session_manager *mgr, struct session *sess, enum session_state state)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
// session state not change
|
|
|
|
|
if (session_get_state(sess) == state)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
return;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
enum session_type type = session_get_type(sess);
|
|
|
|
|
if (type == SESSION_TYPE_TCP)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
// handle old state
|
|
|
|
|
switch (session_get_state(sess))
|
|
|
|
|
{
|
|
|
|
|
case SESSION_STATE_OPENING:
|
|
|
|
|
mgr->tcp_opening_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
|
|
|
|
mgr->tcp_active_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSING:
|
|
|
|
|
mgr->tcp_closing_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
/* void */
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
// handle new state
|
|
|
|
|
switch (state)
|
|
|
|
|
{
|
|
|
|
|
case SESSION_STATE_OPENING:
|
|
|
|
|
mgr->tcp_opening_sess_num++;
|
|
|
|
|
mgr->tcp_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
|
|
|
|
mgr->tcp_active_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSING:
|
|
|
|
|
mgr->tcp_closing_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
mgr->tcp_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else if (type == SESSION_TYPE_UDP)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
// handle old state
|
|
|
|
|
switch (session_get_state(sess))
|
|
|
|
|
{
|
|
|
|
|
case SESSION_STATE_OPENING:
|
|
|
|
|
mgr->udp_opening_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
|
|
|
|
mgr->udp_active_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSING:
|
|
|
|
|
mgr->udp_closing_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
/* void */
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// handle new state
|
|
|
|
|
switch (state)
|
|
|
|
|
{
|
|
|
|
|
case SESSION_STATE_OPENING:
|
|
|
|
|
mgr->udp_opening_sess_num++;
|
|
|
|
|
mgr->udp_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
|
|
|
|
mgr->udp_active_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSING:
|
|
|
|
|
mgr->udp_closing_sess_num++;
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
mgr->udp_sess_num--;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
session_set_state(sess, state);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_session_timer(struct session_manager *mgr, struct session *sess, session_expire_cb expire_cb, uint64_t timeout_sec)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
session_timer_del_session(mgr->sess_timer, sess);
|
|
|
|
|
session_set_expirecb(sess, expire_cb, mgr, timestamp_get_sec() + timeout_sec);
|
|
|
|
|
session_timer_add_session(mgr->sess_timer, sess);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_session_base(struct session_manager *mgr, struct session *sess, const struct tuple6 *key, enum session_dir curr_dir)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
session_init(sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_set_key(sess, key);
|
|
|
|
|
session_set_key_dir(sess, curr_dir);
|
2024-01-17 11:47:55 +08:00
|
|
|
if (key->ip_proto == IPPROTO_UDP)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
session_set_type(sess, SESSION_TYPE_UDP);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
else if (key->ip_proto == IPPROTO_TCP)
|
|
|
|
|
{
|
|
|
|
|
session_set_type(sess, SESSION_TYPE_TCP);
|
|
|
|
|
}
|
2024-03-08 14:20:36 +08:00
|
|
|
session_set_new_time(sess, timestamp_get_sec());
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_session_packet(struct session_manager *mgr, struct session *sess, const struct packet *pkt, enum session_dir curr_dir)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-02-21 11:49:20 +08:00
|
|
|
uint64_t len = packet_get_len(pkt);
|
2023-12-19 10:47:26 +08:00
|
|
|
if (curr_dir == SESSION_DIR_C2S)
|
|
|
|
|
{
|
|
|
|
|
session_inc_c2s_metrics(sess, 1, len);
|
2024-01-23 14:30:46 +08:00
|
|
|
if (session_get0_c2s_1st_pkt(sess) == NULL)
|
|
|
|
|
{
|
|
|
|
|
session_set_c2s_1st_pkt(sess, pkt);
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
else if (curr_dir == SESSION_DIR_S2C)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
|
|
|
|
session_inc_s2c_metrics(sess, 1, len);
|
2024-01-23 14:30:46 +08:00
|
|
|
if (session_get0_s2c_1st_pkt(sess) == NULL)
|
|
|
|
|
{
|
|
|
|
|
session_set_s2c_1st_pkt(sess, pkt);
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
session_set0_cur_pkt(sess, pkt);
|
|
|
|
|
session_set_cur_dir(sess, curr_dir);
|
2024-01-25 16:08:55 +08:00
|
|
|
session_set_last_time(sess, timestamp_get_sec());
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_udp_to_opening(struct session_manager *mgr, struct session *sess)
|
2024-01-24 12:04:20 +08:00
|
|
|
{
|
|
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_OPENING);
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, udp_data_timeout_cb, mgr->opts.udp_timeout_data);
|
2024-01-24 12:04:20 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_udp_to_active(struct session_manager *mgr, struct session *sess)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_ACTIVE);
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, udp_data_timeout_cb, mgr->opts.udp_timeout_data);
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_udp_to_closing(struct session_manager *mgr, struct session *sess)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_CLOSING);
|
|
|
|
|
session_timer_del_session(mgr->sess_timer, sess);
|
2024-03-09 19:28:14 +08:00
|
|
|
evicted_session_filter_add(mgr->udp_evc_sess_filter, session_get0_key(sess), timestamp_get_sec());
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_tcp_to_opening(struct session_manager *mgr, struct session *sess, int opening_by_syn)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_OPENING);
|
|
|
|
|
if (opening_by_syn)
|
|
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_init_timeout_cb, mgr->opts.tcp_timeout_init);
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_handshake_timeout_cb, mgr->opts.tcp_timeout_handshake);
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_tcp_to_active(struct session_manager *mgr, struct session *sess)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_ACTIVE);
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_data_timeout_cb, mgr->opts.tcp_timeout_data);
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_update_tcp_to_closing(struct session_manager *mgr, struct session *sess, int enable_time_wait)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_CLOSING);
|
|
|
|
|
if (enable_time_wait)
|
|
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_time_wait_timeout_cb, mgr->opts.tcp_timeout_time_wait);
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
else
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
session_timer_del_session(mgr->sess_timer, sess);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// opening -> opening
|
|
|
|
|
// opening -> active
|
|
|
|
|
// opening -> closing
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_handle_tcp_on_opening(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
uint64_t tcp_mod_state = tcp_curr_state & (~tcp_old_state);
|
|
|
|
|
|
|
|
|
|
// opening -> closing
|
|
|
|
|
if ((tcp_curr_state & TCP_C2S_FIN_RECVED) && (tcp_curr_state & TCP_S2C_FIN_RECVED))
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
SESSION_LOG_DEBUG("TCP FIN-FIN received, session %lu opening -> closing", session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (tcp_mod_state & (TCP_C2S_RST_RECVED | TCP_S2C_RST_RECVED))
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP %s RST received, session %lu opening -> closing", (tcp_curr_state & TCP_C2S_RST_RECVED ? "C2S" : "S2C"), session_get_id(sess));
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, (tcp_mod_state & TCP_C2S_RST_RECVED) ? CLOSING_BY_CLIENT_RST : CLOSING_BY_SERVER_RST);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// opening -> active
|
|
|
|
|
if (tcp_mod_state & (TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED))
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP %s DATA received, session %lu opening -> active", (tcp_curr_state & TCP_C2S_DATA_RECVED ? "C2S" : "S2C"), session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_active(mgr, sess);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// opening -> opening
|
|
|
|
|
if (tcp_mod_state & (TCP_C2S_FIN_RECVED | TCP_S2C_FIN_RECVED))
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP %s FIN received, session %lu opening -> opening", (tcp_curr_state & TCP_C2S_FIN_RECVED ? "C2S" : "S2C"), session_get_id(sess));
|
|
|
|
|
// still opening, only update timeout
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, (tcp_mod_state & TCP_C2S_FIN_RECVED) ? CLOSING_BY_CLIENT_FIN : CLOSING_BY_SERVER_FIN);
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_half_closed_timeout_cb, mgr->opts.tcp_timeout_half_closed);
|
2024-01-23 14:30:46 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tcp_mod_state & (TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED))
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP %s ACK received, session %lu opening -> opening", (tcp_curr_state & TCP_C2S_ACK_RECVED ? "C2S" : "S2C"), session_get_id(sess));
|
|
|
|
|
// still opening, only update timeout
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_data_timeout_cb, mgr->opts.tcp_timeout_data);
|
2024-01-23 14:30:46 +08:00
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tcp_mod_state & TCP_SYNACK_RECVED)
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP SYNACK received, session %lu opening -> opening", session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_opening(mgr, sess, 0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tcp_mod_state == 0)
|
|
|
|
|
{
|
|
|
|
|
if (tcp_curr_state & TCP_SYN_RECVED)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
SESSION_LOG_DEBUG("TCP SYN retransmission received, session %lu opening -> opening", session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_opening(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (tcp_curr_state & TCP_SYNACK_RECVED)
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP SYNACK retransmission received, session %lu opening -> opening", session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_opening(mgr, sess, 0);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// active -> active
|
|
|
|
|
// active -> closing
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_handle_tcp_on_active(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
// active -> closing
|
2024-01-23 18:23:56 +08:00
|
|
|
if ((tcp_curr_state & TCP_C2S_FIN_RECVED) && (tcp_curr_state & TCP_S2C_FIN_RECVED))
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP FIN-FIN received, session %lu active -> closing", session_get_id(sess));
|
|
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (tcp_curr_state & (TCP_C2S_RST_RECVED | TCP_S2C_RST_RECVED))
|
|
|
|
|
{
|
|
|
|
|
SESSION_LOG_DEBUG("TCP %s RST received, session %lu active -> closing", (tcp_curr_state & TCP_C2S_RST_RECVED) ? "C2S" : "S2C", session_get_id(sess));
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, (tcp_curr_state & TCP_C2S_RST_RECVED) ? CLOSING_BY_CLIENT_RST : CLOSING_BY_SERVER_RST);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
return;
|
|
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// active -> active
|
|
|
|
|
if (tcp_curr_state & (TCP_C2S_FIN_RECVED | TCP_S2C_FIN_RECVED))
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
SESSION_LOG_DEBUG("TCP %s FIN received, session %lu active -> active", (tcp_curr_state & TCP_C2S_FIN_RECVED) ? "C2S" : "S2C", session_get_id(sess));
|
|
|
|
|
// still active
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, (tcp_curr_state & TCP_C2S_FIN_RECVED) ? CLOSING_BY_CLIENT_FIN : CLOSING_BY_SERVER_FIN);
|
2024-03-08 14:51:21 +08:00
|
|
|
session_manager_update_session_timer(mgr, sess, tcp_half_closed_timeout_cb, mgr->opts.tcp_timeout_half_closed);
|
2024-01-23 14:30:46 +08:00
|
|
|
return;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
|
|
|
|
|
// still active
|
|
|
|
|
session_manager_update_tcp_to_active(mgr, sess);
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// closing -> closing
|
|
|
|
|
// closing -> closed
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_handle_tcp_on_closing(struct session_manager *mgr, struct session *sess, enum tcp_state tcp_old_state, enum tcp_state tcp_curr_state)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
|
|
|
|
// still closing
|
|
|
|
|
session_manager_update_tcp_to_closing(mgr, sess, 1);
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline struct session *session_manager_new_tcp_session(struct session_manager *mgr, const struct packet *pkt, const struct tuple6 *key)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
const struct layer *tcp_layer = packet_get_innermost_layer(pkt, LAYER_TYPE_TCP);
|
2024-01-03 09:57:06 +08:00
|
|
|
if (tcp_layer == NULL)
|
|
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
mgr->npkts_miss_l4_proto++;
|
|
|
|
|
return NULL;
|
2024-01-03 09:57:06 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-01-03 09:57:06 +08:00
|
|
|
const struct tcphdr *hdr = (const struct tcphdr *)tcp_layer->hdr_ptr;
|
2024-02-21 11:49:20 +08:00
|
|
|
if (!tcp_hdr_get_syn_flag(hdr))
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
mgr->npkts_hit_tcp_miss_sess++;
|
2024-01-17 11:47:55 +08:00
|
|
|
return NULL;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
if (mgr->tcp_sess_num >= mgr->opts.max_tcp_session_num - RX_BURST_MAX - 1)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (mgr->opts.tcp_overload_evict_old_sess)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
struct session *evicted_sess = session_table_find_least_recently_unused_session(mgr->tcp_sess_table);
|
|
|
|
|
assert(evicted_sess);
|
|
|
|
|
session_manager_evicte_session(mgr, evicted_sess);
|
|
|
|
|
mgr->tcp_overload_evict_old_sess_num++;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (mgr->tcp_sess_num >= mgr->opts.max_tcp_session_num)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
mgr->tcp_overload_evict_new_sess_num++;
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
struct session *sess = session_pool_pop(mgr->sess_pool);
|
2024-01-17 11:47:55 +08:00
|
|
|
assert(sess);
|
|
|
|
|
|
2024-02-21 11:49:20 +08:00
|
|
|
enum session_dir curr_dir = tcp_hdr_get_ack_flag(hdr) ? SESSION_DIR_S2C : SESSION_DIR_C2S;
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_session_base(mgr, sess, key, curr_dir);
|
|
|
|
|
session_manager_update_session_packet(mgr, sess, pkt, curr_dir);
|
|
|
|
|
session_update_tcp_state(sess, tcp_layer, curr_dir);
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
char buffer[128] = {0};
|
2024-03-08 13:43:03 +08:00
|
|
|
tuple6_to_str(session_get0_key(sess), buffer, sizeof(buffer));
|
2024-01-31 14:45:50 +08:00
|
|
|
SESSION_LOG_DEBUG("session new: %lu %s", session_get_id(sess), buffer);
|
2024-01-23 14:30:46 +08:00
|
|
|
SESSION_LOG_DEBUG("TCP %s received, session %lu init -> opening", (curr_dir == SESSION_DIR_C2S ? "SYN" : "SYNACK"), session_get_id(sess));
|
2024-01-31 14:45:50 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_tcp_to_opening(mgr, sess, curr_dir == SESSION_DIR_C2S);
|
|
|
|
|
session_table_add_session(mgr->tcp_sess_table, key, sess);
|
2024-03-09 19:28:14 +08:00
|
|
|
duplicated_packet_filter_add(mgr->tcp_dup_pkt_filter, pkt, timestamp_get_sec());
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
return sess;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline struct session *session_manager_new_udp_session(struct session_manager *mgr, const struct packet *pkt, const struct tuple6 *key)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
if (evicted_session_filter_lookup(mgr->udp_evc_sess_filter, key, timestamp_get_sec()))
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
mgr->npkts_hit_udp_evicted++;
|
2024-01-17 11:47:55 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
if (mgr->udp_sess_num >= mgr->opts.max_udp_session_num - RX_BURST_MAX - 1)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (mgr->opts.udp_overload_evict_old_sess)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
struct session *evicted_sess = session_table_find_least_recently_unused_session(mgr->udp_sess_table);
|
|
|
|
|
assert(evicted_sess);
|
|
|
|
|
session_manager_evicte_session(mgr, evicted_sess);
|
|
|
|
|
mgr->udp_overload_evict_old_sess_num++;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (mgr->udp_sess_num >= mgr->opts.max_udp_session_num)
|
2024-01-26 14:13:54 +08:00
|
|
|
{
|
|
|
|
|
mgr->udp_overload_evict_new_sess_num++;
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
struct session *sess = session_pool_pop(mgr->sess_pool);
|
2024-01-23 14:30:46 +08:00
|
|
|
assert(sess);
|
|
|
|
|
|
|
|
|
|
enum session_dir curr_dir = judge_direction_by_tuple6(key);
|
|
|
|
|
session_manager_update_session_base(mgr, sess, key, curr_dir);
|
|
|
|
|
session_manager_update_session_packet(mgr, sess, pkt, curr_dir);
|
|
|
|
|
session_update_udp_state(sess, NULL, curr_dir);
|
2024-01-24 12:04:20 +08:00
|
|
|
session_manager_update_udp_to_opening(mgr, sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_table_add_session(mgr->udp_sess_table, key, sess);
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
char buffer[128] = {0};
|
2024-03-08 13:43:03 +08:00
|
|
|
tuple6_to_str(session_get0_key(sess), buffer, sizeof(buffer));
|
2024-01-31 14:45:50 +08:00
|
|
|
SESSION_LOG_DEBUG("session new: %lu %s", session_get_id(sess), buffer);
|
|
|
|
|
SESSION_LOG_DEBUG("UDP %s first packet received, session %lu init -> opening", (curr_dir == SESSION_DIR_C2S ? "C2S" : "S2C"), session_get_id(sess));
|
|
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
return sess;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
static inline int session_manager_update_tcp_session(struct session_manager *mgr, struct session *sess, const struct packet *pkt)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
struct tuple6 key;
|
|
|
|
|
memset(&key, 0, sizeof(struct tuple6));
|
|
|
|
|
if (packet_get_innermost_tuple6(pkt, &key) == -1)
|
|
|
|
|
{
|
|
|
|
|
mgr->npkts_miss_l4_proto++;
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
const struct layer *tcp_layer = packet_get_innermost_layer(pkt, LAYER_TYPE_TCP);
|
2024-01-17 11:47:55 +08:00
|
|
|
if (tcp_layer == NULL)
|
|
|
|
|
{
|
|
|
|
|
mgr->npkts_miss_l4_proto++;
|
2024-03-08 18:10:38 +08:00
|
|
|
return -1;
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
enum session_dir curr_dir = judge_direction_by_session(sess, &key);
|
2024-01-23 14:30:46 +08:00
|
|
|
if (session_manager_update_tcp_filter(mgr, sess, pkt, curr_dir))
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
mgr->npkts_hit_tcp_dupkt++;
|
|
|
|
|
session_set_dup_traffic_flag(sess, DUP_TRAFFIC_YES);
|
2024-03-08 18:10:38 +08:00
|
|
|
return -1;
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
enum session_state sess_state = session_get_state(sess);
|
|
|
|
|
enum tcp_state tcp_old_state = session_get_tcp_state(sess);
|
|
|
|
|
session_manager_update_session_packet(mgr, sess, pkt, curr_dir);
|
|
|
|
|
session_update_tcp_state(sess, tcp_layer, curr_dir);
|
|
|
|
|
enum tcp_state tcp_curr_state = session_get_tcp_state(sess);
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
switch (sess_state)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
case SESSION_STATE_INIT:
|
|
|
|
|
assert(0);
|
|
|
|
|
break;
|
2024-01-23 14:30:46 +08:00
|
|
|
case SESSION_STATE_OPENING:
|
|
|
|
|
session_manager_handle_tcp_on_opening(mgr, sess, tcp_old_state, tcp_curr_state);
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
|
|
|
|
session_manager_handle_tcp_on_active(mgr, sess, tcp_old_state, tcp_curr_state);
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSING:
|
|
|
|
|
session_manager_handle_tcp_on_closing(mgr, sess, tcp_old_state, tcp_curr_state);
|
|
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
assert(0);
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
break;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
return 0;
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
static inline int session_manager_update_udp_session(struct session_manager *mgr, struct session *sess, const struct packet *pkt)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
struct tuple6 key;
|
|
|
|
|
memset(&key, 0, sizeof(struct tuple6));
|
|
|
|
|
if (packet_get_innermost_tuple6(pkt, &key) == -1)
|
|
|
|
|
{
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
enum session_dir curr_dir = judge_direction_by_session(sess, &key);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_manager_update_session_packet(mgr, sess, pkt, curr_dir);
|
|
|
|
|
session_update_udp_state(sess, NULL, curr_dir);
|
|
|
|
|
enum session_state sess_state = session_get_state(sess);
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
switch (sess_state)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
case SESSION_STATE_INIT:
|
|
|
|
|
assert(0);
|
|
|
|
|
break;
|
2024-01-23 14:30:46 +08:00
|
|
|
case SESSION_STATE_OPENING:
|
2024-01-24 12:04:20 +08:00
|
|
|
session_manager_update_udp_to_active(mgr, sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_ACTIVE:
|
2024-01-24 12:04:20 +08:00
|
|
|
session_manager_update_udp_to_active(mgr, sess);
|
2024-03-08 18:10:38 +08:00
|
|
|
break;
|
2024-01-23 14:30:46 +08:00
|
|
|
case SESSION_STATE_CLOSING:
|
2024-01-24 12:04:20 +08:00
|
|
|
assert(0);
|
2024-01-23 14:30:46 +08:00
|
|
|
break;
|
|
|
|
|
case SESSION_STATE_CLOSED:
|
|
|
|
|
assert(0);
|
|
|
|
|
break;
|
2024-03-09 19:28:14 +08:00
|
|
|
default:
|
|
|
|
|
break;
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
return 0;
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-31 14:45:50 +08:00
|
|
|
static inline void session_manager_evicte_session(struct session_manager *mgr, struct session *sess)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-03-08 13:33:49 +08:00
|
|
|
session_set_closing_reason(sess, CLOSING_BY_EVICTED);
|
2024-01-25 18:35:31 +08:00
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_CLOSING);
|
2024-01-23 14:30:46 +08:00
|
|
|
session_queue_push(mgr->sess_evicted_queue, sess);
|
2024-01-25 18:35:31 +08:00
|
|
|
session_timer_del_session(mgr->sess_timer, sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
|
|
|
|
|
if (session_get_type(sess) == SESSION_TYPE_UDP)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-03-09 19:28:14 +08:00
|
|
|
evicted_session_filter_add(mgr->udp_evc_sess_filter, session_get0_key(sess), timestamp_get_sec());
|
2023-12-19 10:47:26 +08:00
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/******************************************************************************
|
2024-01-23 14:30:46 +08:00
|
|
|
* Public API
|
2023-12-13 19:20:34 +08:00
|
|
|
******************************************************************************/
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
struct session_manager *session_manager_new(struct session_manager_options *opts)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-03-08 14:51:21 +08:00
|
|
|
if (session_manager_check_options(opts))
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2023-12-13 19:20:34 +08:00
|
|
|
struct session_manager *mgr = (struct session_manager *)calloc(1, sizeof(struct session_manager));
|
|
|
|
|
if (mgr == NULL)
|
|
|
|
|
{
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
memcpy(&mgr->opts, opts, sizeof(struct session_manager_options));
|
2024-03-09 19:28:14 +08:00
|
|
|
struct duplicated_packet_filter_options dup_pkt_opts = {
|
|
|
|
|
.enable = mgr->opts.duplicated_packet_filter_enable,
|
|
|
|
|
.capacity = mgr->opts.duplicated_packet_filter_capacity,
|
|
|
|
|
.timeout_sec = mgr->opts.duplicated_packet_filter_timeout,
|
|
|
|
|
.error_rate = mgr->opts.duplicated_packet_filter_error_rate,
|
|
|
|
|
};
|
|
|
|
|
struct evicted_session_filter_options evc_sess_opts = {
|
|
|
|
|
.enable = mgr->opts.evicted_session_filter_enable,
|
|
|
|
|
.capacity = mgr->opts.evicted_session_filter_capacity,
|
|
|
|
|
.timeout_sec = mgr->opts.evicted_session_filter_timeout,
|
|
|
|
|
.error_rate = mgr->opts.evicted_session_filter_error_rate,
|
|
|
|
|
};
|
|
|
|
|
|
2024-03-08 14:51:21 +08:00
|
|
|
mgr->sess_pool = session_pool_new(mgr->opts.max_tcp_session_num + mgr->opts.max_udp_session_num);
|
2023-12-13 19:20:34 +08:00
|
|
|
if (mgr->sess_pool == NULL)
|
|
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:20:36 +08:00
|
|
|
mgr->tcp_sess_table = session_table_new();
|
2024-01-17 11:47:55 +08:00
|
|
|
if (mgr->tcp_sess_table == NULL)
|
|
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:20:36 +08:00
|
|
|
mgr->udp_sess_table = session_table_new();
|
2024-01-17 11:47:55 +08:00
|
|
|
if (mgr->udp_sess_table == NULL)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:20:36 +08:00
|
|
|
mgr->sess_timer = session_timer_new();
|
2023-12-13 19:20:34 +08:00
|
|
|
if (mgr->sess_timer == NULL)
|
|
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:20:36 +08:00
|
|
|
mgr->sess_evicted_queue = session_queue_new();
|
2024-01-17 11:47:55 +08:00
|
|
|
if (mgr->sess_evicted_queue == NULL)
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
mgr->tcp_dup_pkt_filter = duplicated_packet_filter_new(&dup_pkt_opts, timestamp_get_sec());
|
|
|
|
|
if (mgr->tcp_dup_pkt_filter == NULL)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
mgr->udp_evc_sess_filter = evicted_session_filter_new(&evc_sess_opts, timestamp_get_sec());
|
|
|
|
|
if (mgr->udp_evc_sess_filter == NULL)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
|
|
|
|
goto error;
|
|
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2023-12-13 19:20:34 +08:00
|
|
|
return mgr;
|
|
|
|
|
|
|
|
|
|
error:
|
2024-03-08 14:20:36 +08:00
|
|
|
session_manager_free(mgr);
|
2023-12-13 19:20:34 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 14:20:36 +08:00
|
|
|
void session_manager_free(struct session_manager *mgr)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-31 14:45:50 +08:00
|
|
|
struct session *sess;
|
2023-12-13 19:20:34 +08:00
|
|
|
if (mgr)
|
|
|
|
|
{
|
2024-01-31 14:45:50 +08:00
|
|
|
// move all evicted session to closed queue
|
2024-03-08 18:10:38 +08:00
|
|
|
while (mgr->sess_evicted_queue && (sess = session_manager_get_evicted_session(mgr)))
|
2024-01-31 14:45:50 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
session_manager_free_session(mgr, sess);
|
2024-01-31 14:45:50 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// free all udp session which is not in closed state
|
|
|
|
|
while (mgr->udp_sess_table && (sess = session_table_find_least_recently_unused_session(mgr->udp_sess_table)))
|
|
|
|
|
{
|
|
|
|
|
session_manager_free_session(mgr, sess);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// free all tcp session which is not in closed state
|
|
|
|
|
while (mgr->tcp_sess_table && (sess = session_table_find_least_recently_unused_session(mgr->tcp_sess_table)))
|
|
|
|
|
{
|
|
|
|
|
session_manager_free_session(mgr, sess);
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-09 19:28:14 +08:00
|
|
|
evicted_session_filter_free(mgr->udp_evc_sess_filter);
|
|
|
|
|
duplicated_packet_filter_free(mgr->tcp_dup_pkt_filter);
|
2024-03-08 14:20:36 +08:00
|
|
|
session_queue_free(mgr->sess_evicted_queue);
|
|
|
|
|
session_timer_free(mgr->sess_timer);
|
|
|
|
|
session_table_free(mgr->udp_sess_table);
|
|
|
|
|
session_table_free(mgr->tcp_sess_table);
|
|
|
|
|
session_pool_free(mgr->sess_pool);
|
2023-12-13 19:20:34 +08:00
|
|
|
free(mgr);
|
|
|
|
|
mgr = NULL;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-23 14:30:46 +08:00
|
|
|
// only use the packet six-tuple to find the session, not update it
|
2024-03-08 18:10:38 +08:00
|
|
|
struct session *session_manager_lookup_session(struct session_manager *mgr, const struct packet *pkt)
|
2023-12-22 18:18:21 +08:00
|
|
|
{
|
|
|
|
|
struct tuple6 key;
|
2024-01-17 11:47:55 +08:00
|
|
|
memset(&key, 0, sizeof(struct tuple6));
|
|
|
|
|
if (packet_get_innermost_tuple6(pkt, &key))
|
2023-12-22 18:18:21 +08:00
|
|
|
{
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
if (key.ip_proto == IPPROTO_UDP)
|
|
|
|
|
{
|
|
|
|
|
return session_table_find_session(mgr->udp_sess_table, &key);
|
|
|
|
|
}
|
|
|
|
|
else if (key.ip_proto == IPPROTO_TCP)
|
|
|
|
|
{
|
|
|
|
|
return session_table_find_session(mgr->tcp_sess_table, &key);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
2023-12-22 18:18:21 +08:00
|
|
|
}
|
|
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
/*
|
2024-03-08 18:10:38 +08:00
|
|
|
* return NULL following case:
|
|
|
|
|
* 1.Not TCP or UDP
|
|
|
|
|
* 2.UDP eviction packet
|
|
|
|
|
* 3.UDP overloading and config to bypass new session
|
|
|
|
|
* 4.TCP no SYN flag
|
|
|
|
|
* 5.UDP overloading and config to bypass new session
|
2024-01-17 11:47:55 +08:00
|
|
|
*/
|
2024-03-08 18:10:38 +08:00
|
|
|
struct session *session_manager_new_session(struct session_manager *mgr, const struct packet *pkt)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2023-12-19 10:47:26 +08:00
|
|
|
struct tuple6 key;
|
2024-01-17 11:47:55 +08:00
|
|
|
memset(&key, 0, sizeof(struct tuple6));
|
|
|
|
|
if (packet_get_innermost_tuple6(pkt, &key))
|
2023-12-19 10:47:26 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
mgr->npkts_miss_l4_proto++;
|
2023-12-19 10:47:26 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
2023-12-13 19:20:34 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
if (key.ip_proto == IPPROTO_UDP)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
return session_manager_new_udp_session(mgr, pkt, &key);
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
|
|
|
|
else if (key.ip_proto == IPPROTO_TCP)
|
|
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
return session_manager_new_tcp_session(mgr, pkt, &key);
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
return NULL;
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
void session_manager_free_session(struct session_manager *mgr, struct session *sess)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-01-23 14:30:46 +08:00
|
|
|
if (sess)
|
2023-12-22 18:18:21 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
SESSION_LOG_DEBUG("%s, session %lu closing -> closed", session_closing_reason_to_str(session_get_closing_reason(sess)), session_get_id(sess));
|
|
|
|
|
|
|
|
|
|
session_manager_update_session_state(mgr, sess, SESSION_STATE_CLOSED);
|
|
|
|
|
session_timer_del_session(mgr->sess_timer, sess);
|
2024-01-23 14:30:46 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
session_table_del_session(mgr->tcp_sess_table, session_get0_key(sess));
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
|
|
|
|
if (session_get_type(sess) == SESSION_TYPE_UDP)
|
|
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
session_table_del_session(mgr->udp_sess_table, session_get0_key(sess));
|
2024-01-23 14:30:46 +08:00
|
|
|
}
|
2024-03-08 18:10:38 +08:00
|
|
|
session_set0_cur_pkt(sess, NULL);
|
|
|
|
|
session_set_cur_dir(sess, SESSION_DIR_NONE);
|
|
|
|
|
session_free(sess);
|
2024-03-09 19:28:14 +08:00
|
|
|
session_pool_push(mgr->sess_pool, sess);
|
2024-03-08 18:10:38 +08:00
|
|
|
sess = NULL;
|
2023-12-22 18:18:21 +08:00
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
}
|
2023-12-22 18:18:21 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
/*
|
|
|
|
|
* return NULL following case:
|
|
|
|
|
* 1.Not TCP or UDP
|
|
|
|
|
* 2.TCP duplicate packet
|
|
|
|
|
*/
|
|
|
|
|
int session_manager_update_session(struct session_manager *mgr, struct session *sess, const struct packet *pkt)
|
2024-01-17 11:47:55 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
if (session_get_type(sess) == SESSION_TYPE_TCP)
|
|
|
|
|
{
|
|
|
|
|
return session_manager_update_tcp_session(mgr, sess, pkt);
|
|
|
|
|
}
|
|
|
|
|
else if (session_get_type(sess) == SESSION_TYPE_UDP)
|
|
|
|
|
{
|
|
|
|
|
return session_manager_update_udp_session(mgr, sess, pkt);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-01-17 11:47:55 +08:00
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
// return session need free by session_manager_free_session()
|
|
|
|
|
struct session *session_manager_get_expired_session(struct session_manager *mgr)
|
|
|
|
|
{
|
|
|
|
|
struct session *sess = session_timer_expire_session(mgr->sess_timer, timestamp_get_sec());
|
2024-01-17 11:47:55 +08:00
|
|
|
if (sess)
|
2023-12-13 19:20:34 +08:00
|
|
|
{
|
2024-03-08 18:10:38 +08:00
|
|
|
session_run_expirecb(sess);
|
|
|
|
|
if (session_get_state(sess) == SESSION_STATE_CLOSING)
|
|
|
|
|
{
|
|
|
|
|
return sess;
|
|
|
|
|
}
|
2023-12-22 18:18:21 +08:00
|
|
|
}
|
|
|
|
|
|
2024-03-08 18:10:38 +08:00
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// return session need free by session_manager_free_session()
|
|
|
|
|
struct session *session_manager_get_evicted_session(struct session_manager *mgr)
|
|
|
|
|
{
|
|
|
|
|
return session_queue_pop(mgr->sess_evicted_queue);
|
2023-12-22 18:18:21 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-01-17 11:47:55 +08:00
|
|
|
uint64_t session_manager_get_expire_interval(struct session_manager *mgr)
|
2023-12-22 18:18:21 +08:00
|
|
|
{
|
2024-01-17 11:47:55 +08:00
|
|
|
return session_timer_next_expire_interval(mgr->sess_timer);
|
2023-12-13 19:20:34 +08:00
|
|
|
}
|
2023-12-19 10:47:26 +08:00
|
|
|
|
2024-03-08 14:25:01 +08:00
|
|
|
void session_manager_print_stat(struct session_manager *mgr)
|
2024-01-23 14:30:46 +08:00
|
|
|
{
|
2024-01-26 14:13:54 +08:00
|
|
|
printf("session manager status:\n");
|
2024-01-23 14:30:46 +08:00
|
|
|
|
|
|
|
|
printf("tcp session number : %lu\n", mgr->tcp_sess_num);
|
|
|
|
|
printf(" opening number : %lu\n", mgr->tcp_opening_sess_num);
|
|
|
|
|
printf(" active number : %lu\n", mgr->tcp_active_sess_num);
|
|
|
|
|
printf(" closing number : %lu\n", mgr->tcp_closing_sess_num);
|
|
|
|
|
printf("udp session number : %lu\n", mgr->udp_sess_num);
|
|
|
|
|
printf(" opening number : %lu\n", mgr->udp_opening_sess_num);
|
|
|
|
|
printf(" active number : %lu\n", mgr->udp_active_sess_num);
|
|
|
|
|
printf(" closing number : %lu\n", mgr->udp_closing_sess_num);
|
2024-01-26 14:13:54 +08:00
|
|
|
printf("tcp overload evict : \n");
|
|
|
|
|
printf(" old session number: %lu\n", mgr->tcp_overload_evict_old_sess_num);
|
|
|
|
|
printf(" new session number: %lu\n", mgr->tcp_overload_evict_new_sess_num);
|
|
|
|
|
printf("udp overload evict : \n");
|
|
|
|
|
printf(" old session number: %lu\n", mgr->udp_overload_evict_old_sess_num);
|
|
|
|
|
printf(" new session number: %lu\n", mgr->udp_overload_evict_new_sess_num);
|
2024-01-23 14:30:46 +08:00
|
|
|
|
|
|
|
|
printf("packet status:\n");
|
|
|
|
|
printf(" miss l4 proto : %lu (bypass)\n", mgr->npkts_miss_l4_proto);
|
|
|
|
|
|
|
|
|
|
printf(" hit tcp miss sess : %lu (bypass)\n", mgr->npkts_hit_tcp_miss_sess);
|
|
|
|
|
printf(" hit tcp dupkt : %lu (bypass)\n", mgr->npkts_hit_tcp_dupkt);
|
|
|
|
|
printf(" hit tcp discard : %lu (drop)\n", mgr->npkts_hit_tcp_discard);
|
|
|
|
|
|
|
|
|
|
printf(" hit udp evicted : %lu (bypass)\n", mgr->npkts_hit_udp_evicted);
|
2024-03-08 14:25:01 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void session_manager_get_stat(struct session_manager *mgr, struct session_manager_stat *out)
|
|
|
|
|
{
|
|
|
|
|
out->tcp_sess_num = mgr->tcp_sess_num;
|
|
|
|
|
out->tcp_opening_sess_num = mgr->tcp_opening_sess_num;
|
|
|
|
|
out->tcp_active_sess_num = mgr->tcp_active_sess_num;
|
|
|
|
|
out->tcp_closing_sess_num = mgr->tcp_closing_sess_num;
|
|
|
|
|
|
|
|
|
|
out->udp_sess_num = mgr->udp_sess_num;
|
|
|
|
|
out->udp_opening_sess_num = mgr->udp_opening_sess_num;
|
|
|
|
|
out->udp_active_sess_num = mgr->udp_active_sess_num;
|
|
|
|
|
out->udp_closing_sess_num = mgr->udp_closing_sess_num;
|
|
|
|
|
|
|
|
|
|
out->tcp_overload_evict_old_sess_num = mgr->tcp_overload_evict_old_sess_num;
|
|
|
|
|
out->tcp_overload_evict_new_sess_num = mgr->tcp_overload_evict_new_sess_num;
|
|
|
|
|
out->udp_overload_evict_old_sess_num = mgr->udp_overload_evict_old_sess_num;
|
|
|
|
|
out->udp_overload_evict_new_sess_num = mgr->udp_overload_evict_new_sess_num;
|
|
|
|
|
}
|