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

add tcp init -> opening -> active -> closing -> closed test case

Browse Source
This commit is contained in:
luwenpeng
2024-01-24 11:19:57 +08:00
parent 73b2bce8cf
commit bbec791d6b
9 changed files with 444 additions and 1673 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/session/session.cpp
View File

@@ -549,7 +549,7 @@ const char *dup_traffic_flag_tostring(enum dup_traffic_flag flag)
void session_dump(struct session *sess) void session_dump(struct session *sess)
{ {
char buffer[128] = {0}; char buffer[1024] = {0};
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer)); tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
printf("session id : %" PRIu64 "\n", session_get_id(sess)); printf("session id : %" PRIu64 "\n", session_get_id(sess));

6
src/session/test/CMakeLists.txt
View File

@@ -29,6 +29,9 @@ target_link_libraries(gtest_tcp_active_to_closing session_manager gtest)
add_executable(gtest_tcp_opening_to_closing gtest_tcp_opening_to_closing.cpp) add_executable(gtest_tcp_opening_to_closing gtest_tcp_opening_to_closing.cpp)
target_link_libraries(gtest_tcp_opening_to_closing session_manager gtest) target_link_libraries(gtest_tcp_opening_to_closing session_manager gtest)
add_executable(gtest_tcp_init_to_opening_to_active_to_closing_to_closed gtest_tcp_init_to_opening_to_active_to_closing_to_closed.cpp)
target_link_libraries(gtest_tcp_init_to_opening_to_active_to_closing_to_closed session_manager gtest)
include(GoogleTest) include(GoogleTest)
gtest_discover_tests(gtest_session) gtest_discover_tests(gtest_session)
gtest_discover_tests(gtest_session_pool) gtest_discover_tests(gtest_session_pool)
@@ -38,4 +41,5 @@ gtest_discover_tests(gtest_session_queue)
gtest_discover_tests(gtest_tcp_init_to_opening) gtest_discover_tests(gtest_tcp_init_to_opening)
gtest_discover_tests(gtest_tcp_opening_to_active) gtest_discover_tests(gtest_tcp_opening_to_active)
gtest_discover_tests(gtest_tcp_active_to_closing) gtest_discover_tests(gtest_tcp_active_to_closing)
gtest_discover_tests(gtest_tcp_opening_to_closing) gtest_discover_tests(gtest_tcp_opening_to_closing)
gtest_discover_tests(gtest_tcp_init_to_opening_to_active_to_closing_to_closed)

1644
src/session/test/gtest_session_manager.cpp
View File

File diff suppressed because it is too large Load Diff

16
src/session/test/gtest_tcp_active_to_closing.cpp
View File

@@ -1,13 +1,13 @@
// TCP state machine test: active -> closing // TCP state machine test: active -> closing
#include "test_state_machine.h" #include "test_utils.h"
static void build_active_tcp_session(struct session_manager *mgr, struct session *sess) static void build_active_tcp_session(struct session_manager *mgr, struct session *sess)
{ {
char buffer[1024] = {0}; char buffer[1024] = {0};
struct packet pkt; struct packet pkt;
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -94,7 +94,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_FIN_FIN)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
// C2S FIN Packet // C2S FIN Packet
@@ -185,7 +185,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_C2S_RST)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
// C2S RST Packet // C2S RST Packet
@@ -249,7 +249,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_S2C_RST)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
// S2C RST Packet // S2C RST Packet
@@ -311,7 +311,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_DATA_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0); __session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
@@ -339,7 +339,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_C2S_HALF_CLOSED_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
// C2S FIN Packet // C2S FIN Packet
@@ -396,7 +396,7 @@ TEST(TCP_ACTIVE_TO_CLOSING, BY_S2C_HALF_CLOSED_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet & C2S DATA Packet // C2S SYN Packet & C2S DATA Packet
build_active_tcp_session(mgr, sess); build_active_tcp_session(mgr, sess);
// S2C FIN Packet // S2C FIN Packet

14
src/session/test/gtest_tcp_init_to_opening.cpp
View File

@@ -1,6 +1,6 @@
// TCP state machine test: init -> opening // TCP state machine test: init -> opening
#include "test_state_machine.h" #include "test_utils.h"
/****************************************************************************** /******************************************************************************
* case: TCP init -> opening (by SYN) * case: TCP init -> opening (by SYN)
@@ -19,7 +19,7 @@ TEST(TCP_INIT_TO_OPENING, BY_SYN)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -129,7 +129,7 @@ TEST(TCP_INIT_TO_OPENING, BY_SYN_SYNACK)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -219,7 +219,7 @@ TEST(TCP_INIT_TO_OPENING, BY_SYN_SYNACK_ACK)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -345,7 +345,7 @@ TEST(TCP_INIT_TO_OPENING, BY_SYN_RETRANSMISSION)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -380,7 +380,7 @@ TEST(TCP_INIT_TO_OPENING, BY_SYN_RETRANSMISSION)
sleep(1); sleep(1);
timestamp_update(); timestamp_update();
// SYN Packet retransmission // C2S SYN Packet retransmission
printf("=> packet parse: TCP C2S SYN retransmission packet\n"); printf("=> packet parse: TCP C2S SYN retransmission packet\n");
char tcp_pkt_c2s_syn_retransmission[1500] = {0}; char tcp_pkt_c2s_syn_retransmission[1500] = {0};
memcpy(tcp_pkt_c2s_syn_retransmission, tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); memcpy(tcp_pkt_c2s_syn_retransmission, tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
@@ -539,7 +539,7 @@ TEST(TCP_INIT_TO_OPENING, BY_C2S_ASMMETRIC)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");

411
src/session/test/gtest_tcp_init_to_opening_to_active_to_closing_to_closed.cpp Normal file
View File

@@ -0,0 +1,411 @@
// TCP state machine test: init -> opening -> active -> closing -> closed
#include "test_utils.h"
#if 1
TEST(TCP_INIT_TO_OPENING_TO_ACTIVE_TO_CLOSING_TO_CLOSED, TEST)
{
char buffer[1024];
struct packet pkt;
struct session *sess = NULL;
struct session_manager *mgr = NULL;
timestamp_update();
mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL);
// C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_OPENING);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 0);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 0);
EXPECT_TRUE(session_get_create_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == TCP_SYN_RECVED);
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) == NULL);
EXPECT_TRUE(session_get0_1st_pkt(sess) == session_get0_c2s_1st_pkt(sess));
__session_dispatch(sess);
__session_manager_count(mgr, 1, 0, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// S2C SYNACK Packet
printf("=> packet parse: TCP S2C SYNACK packet\n");
packet_parse(&pkt, (const char *)tcp_pkt2_s2c_syn_ack, sizeof(tcp_pkt2_s2c_syn_ack));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_OPENING);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_S2C);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 1, 0, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// C2S ACK Packet
printf("=> packet parse: TCP C2S ACK packet\n");
packet_parse(&pkt, (const char *)tcp_pkt3_c2s_ack, sizeof(tcp_pkt3_c2s_ack));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_OPENING);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 1, 0, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// C2S REQ Packet
printf("=> packet parse: TCP C2S REQ packet\n");
packet_parse(&pkt, (const char *)tcp_pkt4_c2s_http_req, sizeof(tcp_pkt4_c2s_http_req));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_C2S_DATA_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// S2C ACK Packet
printf("=> packet parse: TCP S2C ACK packet\n");
packet_parse(&pkt, (const char *)tcp_pkt5_s2c_ack, sizeof(tcp_pkt5_s2c_ack));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_S2C);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// S2C HTTP Resp Packet1
printf("=> packet parse: TCP S2C Resp packet1\n");
packet_parse(&pkt, (const char *)tcp_pkt6_s2c_http_resq_1, sizeof(tcp_pkt6_s2c_http_resq_1));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_S2C);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// S2C HTTP Resp Packet2
printf("=> packet parse: TCP S2C Resp packet2\n");
packet_parse(&pkt, (const char *)tcp_pkt7_s2c_http_resp_2, sizeof(tcp_pkt7_s2c_http_resp_2));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354 + 385);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_S2C);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// C2S ACK Packet
printf("=> packet parse: TCP C2S ACK packet\n");
packet_parse(&pkt, (const char *)tcp_pkt8_c2s_ack, sizeof(tcp_pkt8_c2s_ack));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == 0);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145 + 66);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354 + 385);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// C2S FIN Packet
printf("=> packet parse: TCP C2S FIN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt9_c2s_fin, sizeof(tcp_pkt9_c2s_fin));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_ACTIVE);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == CLOSING_BY_CLIENT_FIN);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145 + 66 + 66);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354 + 385);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED | TCP_C2S_FIN_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 1, 0, 0, 0, 0);
sleep(1);
timestamp_update();
// S2C FIN Packet
printf("=> packet parse: TCP S2C FIN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt10_s2c_fin, sizeof(tcp_pkt10_s2c_fin));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_CLOSING);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == CLOSING_BY_CLIENT_FIN);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145 + 66 + 66);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354 + 385 + 66);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_S2C);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED | TCP_C2S_FIN_RECVED | TCP_S2C_FIN_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 0, 1, 0, 0, 0);
sleep(1);
timestamp_update();
// C2S ACK Packet
printf("=> packet parse: TCP C2S ACK packet\n");
packet_parse(&pkt, (const char *)tcp_pkt11_c2s_ack, sizeof(tcp_pkt11_c2s_ack));
printf("<= packet parse\n\n");
sess = session_manager_update_session(mgr, &pkt);
EXPECT_TRUE(sess);
EXPECT_TRUE(session_get_id(sess) == 0);
memset(buffer, 0, sizeof(buffer));
tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer));
EXPECT_STREQ(buffer, "192.168.38.105:60111 -> 93.184.216.34:80, proto: 6, domain: 0");
EXPECT_TRUE(session_get_key_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_state(sess) == SESSION_STATE_CLOSING);
EXPECT_TRUE(session_get_type(sess) == SESSION_TYPE_TCP);
EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0);
EXPECT_TRUE(session_get_closing_reasion(sess) == CLOSING_BY_CLIENT_FIN);
EXPECT_TRUE(session_get_c2s_bytes(sess) == 78 + 66 + 145 + 66 + 66 + 66);
EXPECT_TRUE(session_get_s2c_bytes(sess) == 74 + 66 + 1354 + 385 + 66);
EXPECT_TRUE(session_get_c2s_packets(sess) == 1 + 1 + 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_s2c_packets(sess) == 1 + 1 + 1 + 1 + 1);
EXPECT_TRUE(session_get_create_time(sess) < timestamp_get_msec());
EXPECT_TRUE(session_get_last_time(sess) == timestamp_get_msec());
EXPECT_TRUE(session_get0_cur_pkt(sess) == &pkt);
EXPECT_TRUE(session_get_cur_dir(sess) == SESSION_DIR_C2S);
EXPECT_TRUE(session_get_tcp_state(sess) == (TCP_SYN_RECVED | TCP_SYNACK_RECVED | TCP_C2S_ACK_RECVED | TCP_S2C_ACK_RECVED | TCP_C2S_DATA_RECVED | TCP_S2C_DATA_RECVED | TCP_C2S_FIN_RECVED | TCP_S2C_FIN_RECVED));
EXPECT_TRUE(session_get0_c2s_1st_pkt(sess) != NULL);
EXPECT_TRUE(session_get0_s2c_1st_pkt(sess) != NULL);
__session_dispatch(sess);
__session_manager_count(mgr, 0, 0, 1, 0, 0, 0);
__session_manager_dispatch(mgr, CLOSING_BY_CLIENT_FIN);
__session_manager_count(mgr, 0, 0, 0, 0, 0, 0);
session_manager_destroy(mgr);
}
#endif
int main(int argc, char **argv)
{
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}

4
src/session/test/gtest_tcp_opening_to_active.cpp
View File

@@ -1,6 +1,6 @@
// TCP state machine test: opening -> active // TCP state machine test: opening -> active
#include "test_state_machine.h" #include "test_utils.h"
/****************************************************************************** /******************************************************************************
* case: TCP opening -> active (by C2S DATA) * case: TCP opening -> active (by C2S DATA)
@@ -19,7 +19,7 @@ TEST(TCP_OPENING_TO_ACTIVE, BY_SYN_C2S_DATA)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");

20
src/session/test/gtest_tcp_opening_to_closing.cpp
View File

@@ -1,13 +1,13 @@
// TCP state machine test: opening -> closing // TCP state machine test: opening -> closing
#include "test_state_machine.h" #include "test_utils.h"
static void build_opening_tcp_session(struct session_manager *mgr, struct session *sess) static void build_opening_tcp_session(struct session_manager *mgr, struct session *sess)
{ {
char buffer[1024] = {0}; char buffer[1024] = {0};
struct packet pkt; struct packet pkt;
// SYN Packet // C2S SYN Packet
printf("=> packet parse: TCP C2S SYN packet\n"); printf("=> packet parse: TCP C2S SYN packet\n");
packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn)); packet_parse(&pkt, (const char *)tcp_pkt1_c2s_syn, sizeof(tcp_pkt1_c2s_syn));
printf("<= packet parse\n\n"); printf("<= packet parse\n\n");
@@ -60,7 +60,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_FIN_FIN)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// C2S FIN Packet // C2S FIN Packet
@@ -151,7 +151,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_C2S_RST)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// C2S RST Packet // C2S RST Packet
@@ -215,7 +215,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_S2C_RST)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// S2C RST Packet // S2C RST Packet
@@ -277,7 +277,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_INIT_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
__session_manager_dispatch(mgr, CLOSING_BY_TIMEOUT); __session_manager_dispatch(mgr, CLOSING_BY_TIMEOUT);
@@ -304,7 +304,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_HANDSHAKE_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// S2C SYNACK Packet // S2C SYNACK Packet
@@ -362,7 +362,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_DATA_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// S2C SYNACK Packet // S2C SYNACK Packet
@@ -454,7 +454,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_C2S_HALF_CLOSED_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// C2S FIN Packet // C2S FIN Packet
@@ -511,7 +511,7 @@ TEST(TCP_OPENING_TO_CLOSING, BY_S2C_HALF_CLOSED_TIMEOUT)
mgr = session_manager_create(&config); mgr = session_manager_create(&config);
EXPECT_TRUE(mgr != NULL); EXPECT_TRUE(mgr != NULL);
// SYN Packet // C2S SYN Packet
build_opening_tcp_session(mgr, sess); build_opening_tcp_session(mgr, sess);
// S2C FIN Packet // S2C FIN Packet

0
src/session/test/test_state_machine.h → src/session/test/test_utils.h
View File