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

update session finite state machine & add tcp init -> opening test case

Browse Source
This commit is contained in:
luwenpeng
2024-01-23 14:30:46 +08:00
parent 4fbafab4e3
commit 1f668b1a01
16 changed files with 2049 additions and 1021 deletions
Download Patch File Download Diff File Expand all files Collapse all files

169
src/session/session.cpp
View File

@@ -1,6 +1,7 @@
#include <assert.h>
#include "session_private.h"
#include "packet_helpers.h"
#define EX_KEY_MAX_LEN 64
@@ -18,12 +19,6 @@ 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
uint8_t c2s_1st_md_ex = 0; // built-in ex_data index
uint8_t s2c_1st_md_ex = 0; // built-in ex_data index
uint8_t c2s_1st_pkt_ex = 0; // built-in ex_data index
uint8_t s2c_1st_pkt_ex = 0; // built-in ex_data index
/******************************************************************************
* session
@@ -46,22 +41,22 @@ uint64_t session_get_id(const struct session *sess)
}
// session tuple6
void session_set_tuple6(struct session *sess, const struct tuple6 *tuple)
void session_set_key(struct session *sess, const struct tuple6 *tuple)
{
memcpy(&sess->tuple, tuple, sizeof(struct tuple6));
}
const struct tuple6 *session_get0_tuple6(const struct session *sess)
const struct tuple6 *session_get0_key(const struct session *sess)
{
return &sess->tuple;
}
void session_set_tuple6_dir(struct session *sess, enum session_dir dir)
void session_set_key_dir(struct session *sess, enum session_dir dir)
{
sess->tuple_dir = dir;
}
enum session_dir session_get_tuple6_dir(const struct session *sess)
enum session_dir session_get_key_dir(const struct session *sess)
{
return sess->tuple_dir;
}
@@ -164,18 +159,50 @@ uint64_t session_get_last_time(const struct session *sess)
return sess->last_time;
}
// session tcp state
void session_set_tcp_state(struct session *sess, enum tcp_state state)
{
sess->tcp_state = state;
}
enum tcp_state session_get_tcp_state(const struct session *sess)
{
return sess->tcp_state;
}
// session udp state
void session_set_udp_state(struct session *sess, enum udp_state state)
{
sess->udp_state = state;
}
enum udp_state session_get_udp_state(const struct session *sess)
{
return sess->udp_state;
}
/******************************************************************************
* session packet
******************************************************************************/
void session_set_c2s_1st_pkt(struct session *sess, const struct packet *pkt)
{
sess->c2s_1st_pkt = packet_dup(pkt);
}
void session_set_s2c_1st_pkt(struct session *sess, const struct packet *pkt)
{
sess->s2c_1st_pkt = packet_dup(pkt);
}
const struct packet *session_get0_c2s_1st_pkt(const struct session *sess)
{
return (const struct packet *)session_get0_ex_data(sess, c2s_1st_pkt_ex);
return sess->c2s_1st_pkt;
}
const struct packet *session_get0_s2c_1st_pkt(const struct session *sess)
{
return (const struct packet *)session_get0_ex_data(sess, s2c_1st_pkt_ex);
return sess->s2c_1st_pkt;
}
const struct packet *session_get0_1st_pkt(const struct session *sess)
@@ -304,6 +331,29 @@ void session_free_ex_data(struct session *sess, uint8_t idx)
sess->ex_data[idx] = NULL;
}
void session_free(struct session *sess)
{
if (sess)
{
for (uint8_t i = 0; i < g_ex_manager.count; i++)
{
session_free_ex_data(sess, i);
}
if (sess->c2s_1st_pkt)
{
packet_free(sess->c2s_1st_pkt);
sess->c2s_1st_pkt = NULL;
}
if (sess->s2c_1st_pkt)
{
packet_free(sess->s2c_1st_pkt);
sess->s2c_1st_pkt = NULL;
}
}
}
/******************************************************************************
* session expire
******************************************************************************/
@@ -342,69 +392,79 @@ void session_run_expirecb(struct session *sess)
* session dump
******************************************************************************/
static void tcp_ex_data_tostring(uint64_t ex_data, char *buffer, size_t buffer_len)
static void tcp_state_tostring(enum tcp_state state, char *buffer, size_t buffer_len)
{
if (ex_data == 0)
if (state == 0)
{
return;
}
int nused = 0;
if (ex_data & TCP_SYN_RECVED)
if (state & TCP_SYN_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_SYN_RECVED ");
}
if (ex_data & TCP_SYNACK_RECVED)
if (state & TCP_SYNACK_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_SYNACK_RECVED ");
}
if (ex_data & TCP_C2S_DATA_RECVED)
if (state & TCP_C2S_ACK_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_ACK_RECVED ");
}
if (state & TCP_S2C_ACK_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_ACK_RECVED ");
}
if (state & TCP_C2S_DATA_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_DATA_RECVED ");
}
if (ex_data & TCP_S2C_DATA_RECVED)
if (state & TCP_S2C_DATA_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_DATA_RECVED ");
}
if (ex_data & TCP_C2S_FIN_RECVED)
if (state & TCP_C2S_FIN_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_FIN_RECVED ");
}
if (ex_data & TCP_S2C_FIN_RECVED)
if (state & TCP_S2C_FIN_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_FIN_RECVED ");
}
if (ex_data & TCP_C2S_RST_RECVED)
if (state & TCP_C2S_RST_RECVED)
{
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_RST_RECVED ");
}
if (ex_data & TCP_S2C_RST_RECVED)
if (state & TCP_S2C_RST_RECVED)
{
nused += 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)
static void udp_state_tostring(enum udp_state state, char *buffer, size_t buffer_len)
{
if (ex_data == 0)
if (state == 0)
{
return;
}
int nused = 0;
if (ex_data & UDP_C2S_RECVED)
if (state & UDP_C2S_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "UDP_C2S_RECVED ");
}
if (ex_data & UDP_S2C_RECVED)
if (state & UDP_S2C_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "UDP_S2C_RECVED ");
}
@@ -453,9 +513,9 @@ const char *session_type_tostring(enum session_type type)
switch (type)
{
case SESSION_TYPE_TCP:
return "tcp";
return "TCP";
case SESSION_TYPE_UDP:
return "udp";
return "UDP";
default:
return "unknown";
}
@@ -466,9 +526,22 @@ const char *session_dir_tostring(enum session_dir dir)
switch (dir)
{
case SESSION_DIR_C2S:
return "c2s";
return "C2S";
case SESSION_DIR_S2C:
return "s2c";
return "S2C";
default:
return "unknown";
}
}
const char *dup_traffic_flag_tostring(enum dup_traffic_flag flag)
{
switch (flag)
{
case DUP_TRAFFIC_YES:
return "YES";
case DUP_TRAFFIC_NO:
return "NO";
default:
return "unknown";
}
@@ -477,13 +550,14 @@ const char *session_dir_tostring(enum session_dir dir)
void session_dump(struct session *sess)
{
char buffer[128] = {0};
tuple6_tostring(session_get0_tuple6(sess), buffer, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
printf("session id : %" PRIu64 "\n", session_get_id(sess));
printf("session tuple6 key : %s\n", buffer);
printf("session tuple6 dir : %s\n", session_dir_tostring(session_get_tuple6_dir(sess)));
printf("session key : %s\n", buffer);
printf("session key dir : %s\n", session_dir_tostring(session_get_key_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 dup traffic flag : %s\n", dup_traffic_flag_tostring(session_get_dup_traffic_flag(sess)));
printf("session closing reasion : %s\n", session_closing_reasion_tostring(session_get_closing_reasion(sess)));
printf("session c2s packets : %" PRIu64 "\n", session_get_c2s_packets(sess));
printf("session c2s bytes : %" PRIu64 "\n", session_get_c2s_bytes(sess));
@@ -491,26 +565,23 @@ void session_dump(struct session *sess)
printf("session s2c bytes : %" PRIu64 "\n", session_get_s2c_bytes(sess));
printf("session create time : %" PRIu64 "\n", session_get_create_time(sess));
printf("session last time : %" PRIu64 "\n", session_get_last_time(sess));
if (session_get_type(sess) == SESSION_TYPE_TCP)
{
memset(buffer, 0, sizeof(buffer));
tcp_state_tostring(session_get_tcp_state(sess), buffer, sizeof(buffer));
printf("session tcp state : %s\n", buffer);
}
else if (session_get_type(sess) == SESSION_TYPE_UDP)
{
memset(buffer, 0, sizeof(buffer));
udp_state_tostring(session_get_udp_state(sess), buffer, sizeof(buffer));
printf("session udp state : %s\n", buffer);
}
printf("session current packet ptr : %p\n", (void *)session_get0_cur_pkt(sess));
printf("session current packet 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]);
}
printf(" ex_idx: %d, ex_key: %s, ex_data: %p\n", i, g_ex_manager.schemas[i].key, sess->ex_data[i]);
}
}