gfwleak/stellar-stellar
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

implementation session manager

Browse Source
This commit is contained in:
luwenpeng
2023-12-19 10:47:26 +08:00
parent 2e56bd810c
commit 2c26879182
15 changed files with 2269 additions and 193 deletions
Download Patch File Download Diff File Expand all files Collapse all files

231
src/session/session.cpp
View File

@@ -18,6 +18,8 @@ struct ex_manager
};
static struct ex_manager g_ex_manager = {0};
uint8_t tcp_builtin_ex = 0; // built-in ex_data index
uint8_t udp_builtin_ex = 0; // built-in ex_data index
/******************************************************************************
* ev queue
@@ -81,7 +83,7 @@ void session_set_id(struct session *sess, uint64_t id)
sess->id = id;
}
uint64_t session_get_id(struct session *sess)
uint64_t session_get_id(const struct session *sess)
{
return sess->id;
}
@@ -92,18 +94,28 @@ void session_set_tuple6(struct session *sess, struct tuple6 *tuple)
memcpy(&sess->tuple, tuple, sizeof(struct tuple6));
}
struct tuple6 *session_get0_tuple6(struct session *sess)
const struct tuple6 *session_get0_tuple6(const struct session *sess)
{
return &sess->tuple;
}
void session_set_tuple6_dir(struct session *sess, enum session_dir dir)
{
sess->tuple_dir = dir;
}
enum session_dir session_get_tuple6_dir(const struct session *sess)
{
return sess->tuple_dir;
}
// session state
void session_set_state(struct session *sess, enum session_state state)
{
sess->state = state;
}
enum session_state session_get_state(struct session *sess)
enum session_state session_get_state(const struct session *sess)
{
return sess->state;
}
@@ -114,7 +126,7 @@ void session_set_type(struct session *sess, enum session_type type)
sess->type = type;
}
enum session_type session_get_type(struct session *sess)
enum session_type session_get_type(const struct session *sess)
{
return sess->type;
}
@@ -132,22 +144,22 @@ void session_inc_s2c_metrics(struct session *sess, uint64_t packets, uint64_t by
sess->s2c_packets += packets;
}
uint64_t session_get_c2s_bytes(struct session *sess)
uint64_t session_get_c2s_bytes(const struct session *sess)
{
return sess->c2s_bytes;
}
uint64_t session_get_s2c_bytes(struct session *sess)
uint64_t session_get_s2c_bytes(const struct session *sess)
{
return sess->s2c_bytes;
}
uint64_t session_get_c2s_packets(struct session *sess)
uint64_t session_get_c2s_packets(const struct session *sess)
{
return sess->c2s_packets;
}
uint64_t session_get_s2c_packets(struct session *sess)
uint64_t session_get_s2c_packets(const struct session *sess)
{
return sess->s2c_packets;
}
@@ -163,7 +175,7 @@ void session_set_s2c_1st_md(struct session *sess, struct metadata *md)
memcpy(&sess->s2c_1st_md, md, sizeof(struct metadata));
}
struct metadata *session_get0_c2s_1st_md(struct session *sess)
const struct metadata *session_get0_c2s_1st_md(const struct session *sess)
{
if (sess->c2s_1st_md.len == 0)
{
@@ -172,7 +184,7 @@ struct metadata *session_get0_c2s_1st_md(struct session *sess)
return &sess->c2s_1st_md;
}
struct metadata *session_get0_s2c_1st_md(struct session *sess)
const struct metadata *session_get0_s2c_1st_md(const struct session *sess)
{
if (sess->s2c_1st_md.len == 0)
{
@@ -192,12 +204,12 @@ void session_set_last_time(struct session *sess, uint64_t timestamp)
sess->last_time = timestamp;
}
uint64_t session_get_create_time(struct session *sess)
uint64_t session_get_create_time(const struct session *sess)
{
return sess->create_time;
}
uint64_t session_get_last_time(struct session *sess)
uint64_t session_get_last_time(const struct session *sess)
{
return sess->last_time;
}
@@ -212,7 +224,7 @@ void session_set0_cur_pkt(struct session *sess, const struct packet *pkt)
sess->cur_pkt = pkt;
}
const struct packet *session_get0_cur_pkt(struct session *sess)
const struct packet *session_get0_cur_pkt(const struct session *sess)
{
return sess->cur_pkt;
}
@@ -223,7 +235,7 @@ void session_set_cur_dir(struct session *sess, enum session_dir dir)
sess->cur_dir = dir;
}
enum session_dir session_get_cur_dir(struct session *sess)
enum session_dir session_get_cur_dir(const struct session *sess)
{
return sess->cur_dir;
}
@@ -301,7 +313,7 @@ void session_set_ex_data(struct session *sess, uint8_t idx, void *val)
sess->ex_data[idx] = val;
}
void *session_get0_ex_data(struct session *sess, uint8_t idx)
void *session_get0_ex_data(const struct session *sess, uint8_t idx)
{
if (idx >= g_ex_manager.count)
{
@@ -365,4 +377,193 @@ void session_run_expirecb(struct session *sess)
{
sess->expire_cb(sess, sess->expire_arg);
}
}
/******************************************************************************
* session dump
******************************************************************************/
static void tcp_ex_data_tostring(uint64_t ex_data, char *buffer, size_t buffer_len)
{
if (ex_data == 0)
{
return;
}
int nused = 0;
if (ex_data & TCP_SYN_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_SYN_RECVED ");
}
if (ex_data & TCP_SYNACK_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_SYNACK_RECVED ");
}
if (ex_data & TCP_ACK_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_ACK_RECVED ");
}
if (ex_data & TCP_C2S_PAYLOAD_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_PAYLOAD_RECVED ");
}
if (ex_data & TCP_S2C_PAYLOAD_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_PAYLOAD_RECVED ");
}
if (ex_data & TCP_C2S_FIN_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_FIN_RECVED ");
}
if (ex_data & TCP_S2C_FIN_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_FIN_RECVED ");
}
if (ex_data & TCP_C2S_RST_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_RST_RECVED ");
}
if (ex_data & TCP_S2C_RST_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_RST_RECVED ");
}
}
static void udp_ex_data_tostring(uint64_t ex_data, char *buffer, size_t buffer_len)
{
if (ex_data == 0)
{
return;
}
int nused = 0;
if (ex_data & UDP_C2S_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "UDP_C2S_RECVED ");
}
if (ex_data & UDP_S2C_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "UDP_S2C_RECVED ");
}
}
const char *session_event_tostring(enum session_event event)
{
switch (event)
{
case SESSION_EVENT_NONE:
return "none";
case SESSION_EVENT_OPENING:
return "opening";
case SESSION_EVENT_ACTIVE:
return "active";
case SESSION_EVENT_CLOSING:
return "closing";
default:
return "unknown";
}
}
const char *session_state_tostring(enum session_state state)
{
switch (state)
{
case SESSION_STATE_INIT:
return "init";
case SESSION_STATE_OPENING:
return "opening";
case SESSION_STATE_ACTIVE:
return "active";
case SESSION_STATE_DISCARD:
return "discard";
case SESSION_STATE_CLOSING:
return "closing";
case SESSION_STATE_CLOSED:
return "closed";
default:
return "unknown";
}
}
const char *session_type_tostring(enum session_type type)
{
switch (type)
{
case SESSION_TYPE_NONE:
return "none";
case SESSION_TYPE_TCP:
return "tcp";
case SESSION_TYPE_TCP_STREAM:
return "tcp_stream";
case SESSION_TYPE_UDP:
return "udp";
default:
return "unknown";
}
}
const char *session_dir_tostring(enum session_dir dir)
{
switch (dir)
{
case SESSION_DIR_NONE:
return "none";
case SESSION_DIR_C2S:
return "c2s";
case SESSION_DIR_S2C:
return "s2c";
default:
return "unknown";
}
}
void session_dump(struct session *sess)
{
char buffer[128] = {0};
tuple6_tostring(session_get0_tuple6(sess), buffer, sizeof(buffer));
printf("session id : %" PRIu64 "\n", session_get_id(sess));
printf("session tuple6 : %s\n", buffer);
printf("session tuple6 dir : %s\n", session_dir_tostring(session_get_tuple6_dir(sess)));
printf("session state : %s\n", session_state_tostring(session_get_state(sess)));
printf("session type : %s\n", session_type_tostring(session_get_type(sess)));
printf("session c2s packets : %" PRIu64 "\n", session_get_c2s_packets(sess));
printf("session c2s bytes : %" PRIu64 "\n", session_get_c2s_bytes(sess));
printf("session s2c packets : %" PRIu64 "\n", session_get_s2c_packets(sess));
printf("session s2c bytes : %" PRIu64 "\n", session_get_s2c_bytes(sess));
printf("session c2s 1st metadata : %p\n", (void *)session_get0_c2s_1st_md(sess));
printf("session s2c 1st metadata : %p\n", (void *)session_get0_s2c_1st_md(sess));
printf("session create time : %" PRIu64 "\n", session_get_create_time(sess));
printf("session last time : %" PRIu64 "\n", session_get_last_time(sess));
printf("session current packet : %p\n", (void *)session_get0_cur_pkt(sess));
printf("session current dir : %s\n", session_dir_tostring(session_get_cur_dir(sess)));
printf("session ex data: \n");
for (uint8_t i = 0; i < g_ex_manager.count; i++)
{
if (i == tcp_builtin_ex)
{
memset(buffer, 0, sizeof(buffer));
tcp_ex_data_tostring((uint64_t)sess->ex_data[i], buffer, sizeof(buffer));
printf(" ex_idx: %d, ex_key: %s, ex_val: %s\n", i, g_ex_manager.schemas[i].key, buffer);
}
else if (i == udp_builtin_ex)
{
memset(buffer, 0, sizeof(buffer));
udp_ex_data_tostring((uint64_t)sess->ex_data[i], buffer, sizeof(buffer));
printf(" ex_idx: %d, ex_key: %s, ex_val: %s\n", i, g_ex_manager.schemas[i].key, buffer);
}
else
{
printf(" ex_idx: %d, ex_key: %s, ex_val: %p\n", i, g_ex_manager.schemas[i].key, sess->ex_data[i]);
}
}
}