From 3f1a1f3fc3b83d893fa8176e2c51d653036148c3 Mon Sep 17 00:00:00 2001 From: luwenpeng Date: Wed, 24 Jan 2024 16:45:02 +0800 Subject: [PATCH] add udp init -> opening -> active -> closing -> closed test case --- src/session/test/CMakeLists.txt | 10 +- ...p_init_to_opening_to_active_to_closing.cpp | 102 ++++++++++++++++++ ... gtest_udp_init_to_opening_to_closing.cpp} | 10 +- 3 files changed, 116 insertions(+), 6 deletions(-) create mode 100644 src/session/test/gtest_udp_init_to_opening_to_active_to_closing.cpp rename src/session/test/{gtest_udp_init_to_opening.cpp => gtest_udp_init_to_opening_to_closing.cpp} (94%) diff --git a/src/session/test/CMakeLists.txt b/src/session/test/CMakeLists.txt index 8df64ab..66f739b 100644 --- a/src/session/test/CMakeLists.txt +++ b/src/session/test/CMakeLists.txt @@ -32,8 +32,11 @@ 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) -add_executable(gtest_udp_init_to_opening gtest_udp_init_to_opening.cpp) -target_link_libraries(gtest_udp_init_to_opening session_manager gtest) +add_executable(gtest_udp_init_to_opening_to_closing gtest_udp_init_to_opening_to_closing.cpp) +target_link_libraries(gtest_udp_init_to_opening_to_closing session_manager gtest) + +add_executable(gtest_udp_init_to_opening_to_active_to_closing gtest_udp_init_to_opening_to_active_to_closing.cpp) +target_link_libraries(gtest_udp_init_to_opening_to_active_to_closing session_manager gtest) include(GoogleTest) gtest_discover_tests(gtest_session) @@ -46,4 +49,5 @@ gtest_discover_tests(gtest_tcp_opening_to_active) gtest_discover_tests(gtest_tcp_active_to_closing) gtest_discover_tests(gtest_tcp_opening_to_closing) gtest_discover_tests(gtest_tcp_init_to_opening_to_active_to_closing_to_closed) -gtest_discover_tests(gtest_udp_init_to_opening) \ No newline at end of file +gtest_discover_tests(gtest_udp_init_to_opening_to_closing) +gtest_discover_tests(gtest_udp_init_to_opening_to_active_to_closing) \ No newline at end of file diff --git a/src/session/test/gtest_udp_init_to_opening_to_active_to_closing.cpp b/src/session/test/gtest_udp_init_to_opening_to_active_to_closing.cpp new file mode 100644 index 0000000..fa1d473 --- /dev/null +++ b/src/session/test/gtest_udp_init_to_opening_to_active_to_closing.cpp @@ -0,0 +1,102 @@ +// UDP state machine test: init -> opening -> active -> closing + +#include "test_utils.h" + +/****************************************************************************** + * case: UDP init -> opening (by C2S Packet) + * + * case UDP opening -> active (by S2C Packet) + * + * case UDP active -> closing (by timeout) + ******************************************************************************/ + +#if 1 +TEST(UDP_INIT_TO_OPENING_TO_ACTIVE_TO_CLOSING, TEST) +{ + char buffer[1024] = {0}; + struct packet pkt; + struct session *sess = NULL; + struct session_manager *mgr = NULL; + + timestamp_update(); + + mgr = session_manager_create(&config); + EXPECT_TRUE(mgr != NULL); + + // C2S REQ Packet + printf("=> packet parse: UDP C2S REQ packet\n"); + packet_parse(&pkt, (const char *)udp_pkt1_dns_req, sizeof(udp_pkt1_dns_req)); + printf("<= packet parse\n\n"); + sess = session_manager_update_session(mgr, &pkt); + EXPECT_TRUE(sess); + + EXPECT_TRUE(session_get_id(sess) == 0); + tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer)); + EXPECT_STREQ(buffer, "192.168.38.105:61099 -> 121.14.154.93:53, proto: 17, 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_UDP); + EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0); + EXPECT_TRUE(session_get_closing_reasion(sess) == 0); + EXPECT_TRUE(session_get_c2s_bytes(sess) == 74); + 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_udp_state(sess) == UDP_C2S_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, 0, 0, 0, 1, 0, 0); + + sleep(1); + timestamp_update(); + + // S2C RESP Packet + printf("=> packet parse: UDP S2C RESP packet\n"); + packet_parse(&pkt, (const char *)udp_pkt2_dns_resp, sizeof(udp_pkt2_dns_resp)); + printf("<= packet parse\n\n"); + sess = session_manager_update_session(mgr, &pkt); + EXPECT_TRUE(sess); + + EXPECT_TRUE(session_get_id(sess) == 0); + tuple6_tostring(session_get0_key(sess), buffer, sizeof(buffer)); + EXPECT_STREQ(buffer, "192.168.38.105:61099 -> 121.14.154.93:53, proto: 17, 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_UDP); + EXPECT_TRUE(session_get_dup_traffic_flag(sess) == 0); + EXPECT_TRUE(session_get_closing_reasion(sess) == 0); + EXPECT_TRUE(session_get_c2s_bytes(sess) == 74); + EXPECT_TRUE(session_get_s2c_bytes(sess) == 550); + 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_udp_state(sess) == (UDP_C2S_RECVED | UDP_S2C_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, 0, 0, 1, 0); + __session_manager_dispatch(mgr, CLOSING_BY_TIMEOUT); + __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(); +} diff --git a/src/session/test/gtest_udp_init_to_opening.cpp b/src/session/test/gtest_udp_init_to_opening_to_closing.cpp similarity index 94% rename from src/session/test/gtest_udp_init_to_opening.cpp rename to src/session/test/gtest_udp_init_to_opening_to_closing.cpp index a877e58..6b0fefe 100644 --- a/src/session/test/gtest_udp_init_to_opening.cpp +++ b/src/session/test/gtest_udp_init_to_opening_to_closing.cpp @@ -4,10 +4,12 @@ /****************************************************************************** * case: UDP init -> opening (by C2S Packet) - ******************************************************************************/ + * + * case: UDP opening -> closing (by timeout) + * ******************************************************************************/ #if 1 -TEST(UDP_INIT_TO_OPENING, BY_C2S) +TEST(UDP_INIT_TO_OPENING_TO_CLOSING, BY_C2S) { char buffer[1024] = {0}; struct packet pkt; @@ -59,10 +61,12 @@ TEST(UDP_INIT_TO_OPENING, BY_C2S) /****************************************************************************** * case: UDP init -> opening (by S2C Packet) + * + * case: UDP opening -> closing (by timeout) ******************************************************************************/ #if 1 -TEST(UDP_INIT_TO_OPENING, BY_S2C) +TEST(UDP_INIT_TO_OPENING_TO_CLOSING, BY_S2C) { char buffer[1024] = {0}; struct packet pkt;