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

Add test cases for session manager (session state machine)

Browse Source
This commit is contained in:
luwenpeng
2023-12-21 18:07:52 +08:00
parent 2c26879182
commit cd47b46e7a
4 changed files with 2260 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2234
src/session/gtest_session_manager.cpp
View File

File diff suppressed because it is too large Load Diff

27
src/session/session.cpp
View File

@@ -393,47 +393,42 @@ static void tcp_ex_data_tostring(uint64_t ex_data, char *buffer, size_t buffer_l
int nused = 0;
if (ex_data & TCP_SYN_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_SYN_RECVED ");
nused += 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 ");
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_SYNACK_RECVED ");
}
if (ex_data & TCP_C2S_PAYLOAD_RECVED)
{
snprintf(buffer + nused, buffer_len - nused, "TCP_C2S_PAYLOAD_RECVED ");
nused += 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 ");
nused += 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 ");
nused += 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 ");
nused += 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 ");
nused += 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 ");
nused += snprintf(buffer + nused, buffer_len - nused, "TCP_S2C_RST_RECVED ");
}
}
@@ -532,7 +527,7 @@ void session_dump(struct session *sess)
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 key : %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)));
@@ -544,8 +539,8 @@ void session_dump(struct session *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 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++)
{

29
src/session/session_manager.cpp
View File

@@ -296,11 +296,13 @@ static void update_tcp_ex_data(struct session *sess, const struct packet *pkt, e
{
if (curr_dir == SESSION_DIR_C2S)
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_C2S_RST_RECVED));
state |= TCP_C2S_RST_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
else
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_S2C_RST_RECVED));
state |= TCP_S2C_RST_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
}
@@ -308,11 +310,13 @@ static void update_tcp_ex_data(struct session *sess, const struct packet *pkt, e
{
if (curr_dir == SESSION_DIR_C2S)
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_C2S_FIN_RECVED));
state |= TCP_C2S_FIN_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
else
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state & TCP_S2C_FIN_RECVED));
state |= TCP_S2C_FIN_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
}
@@ -320,28 +324,27 @@ static void update_tcp_ex_data(struct session *sess, const struct packet *pkt, e
{
if (packet_has_tcp_flag_ack(pkt))
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_SYNACK_RECVED));
state |= TCP_SYNACK_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
else
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_SYN_RECVED));
state |= TCP_SYN_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
}
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));
state |= TCP_C2S_PAYLOAD_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
else
{
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state | TCP_S2C_PAYLOAD_RECVED));
state |= TCP_S2C_PAYLOAD_RECVED;
session_set_ex_data(sess, tcp_builtin_ex, (void *)(state));
}
}
}

13
src/session/session_private.h
View File

@@ -22,16 +22,15 @@ enum tcp_ex_data
// HANDSHAKE
TCP_SYN_RECVED = 1 << 0,
TCP_SYNACK_RECVED = 1 << 1,
TCP_ACK_RECVED = 1 << 2,
// ESTABLISHED
TCP_C2S_PAYLOAD_RECVED = 1 << 3,
TCP_S2C_PAYLOAD_RECVED = 1 << 4,
TCP_C2S_PAYLOAD_RECVED = 1 << 2,
TCP_S2C_PAYLOAD_RECVED = 1 << 3,
// FIN
TCP_C2S_FIN_RECVED = 1 << 5,
TCP_S2C_FIN_RECVED = 1 << 6,
TCP_C2S_FIN_RECVED = 1 << 4,
TCP_S2C_FIN_RECVED = 1 << 5,
// RST
TCP_C2S_RST_RECVED = 1 << 7,
TCP_S2C_RST_RECVED = 1 << 8,
TCP_C2S_RST_RECVED = 1 << 6,
TCP_S2C_RST_RECVED = 1 << 7,
};
enum udp_ex_data