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

session table can search session by tuple6 or tuple4 or id

Browse Source
This commit is contained in:
luwenpeng
2024-04-24 11:39:15 +08:00
parent 74f0504d3d
commit 476c5bac56
5 changed files with 233 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

199
src/session/test/gtest_session_table.cpp
View File

@@ -3,38 +3,51 @@
#include "session_pool.h"
#include "session_table.h"
#define SESSION_KEY_IPV4_TCP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V4; \
(name).src_addr.v4.s_addr = inet_addr("192.168.1.2"); \
(name).dst_addr.v4.s_addr = inet_addr("192.168.1.3"); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_TCP; \
(name).domain = 0;
#define TUPLE6_SET_V4_TCP(t6) \
{ \
memset(&t6, 0, sizeof(struct tuple6)); \
(t6).ip_type = IP_TYPE_V4; \
(t6).src_addr.v4.s_addr = inet_addr("192.168.1.2"); \
(t6).dst_addr.v4.s_addr = inet_addr("192.168.1.3"); \
(t6).src_port = htons(1234); \
(t6).dst_port = htons(5678); \
(t6).ip_proto = IPPROTO_TCP; \
(t6).domain = 123; \
}
#define SESSION_KEY_IPV6_UDP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(name).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(name).dst_addr.v6); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_UDP; \
(name).domain = 0;
#define TUPLE6_SET_V6_UDP(t6) \
{ \
memset(&t6, 0, sizeof(struct tuple6)); \
(t6).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(t6).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(t6).dst_addr.v6); \
(t6).src_port = htons(2345); \
(t6).dst_port = htons(6789); \
(t6).ip_proto = IPPROTO_UDP; \
(t6).domain = 456; \
}
#define SESSION_KEY_IPV6_TCP(name) \
struct tuple6 name; \
memset(&name, 0, sizeof(struct tuple6)); \
(name).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(name).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(name).dst_addr.v6); \
(name).src_port = htons(1234); \
(name).dst_port = htons(5678); \
(name).ip_proto = IPPROTO_TCP; \
(name).domain = 0;
#define TUPLE6_SET_V6_TCP(t6) \
{ \
memset(&t6, 0, sizeof(struct tuple6)); \
(t6).ip_type = IP_TYPE_V6; \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:8329", &(t6).src_addr.v6); \
inet_pton(AF_INET6, "2001:db8:0:0:0:ff00:42:832a", &(t6).dst_addr.v6); \
(t6).src_port = htons(3456); \
(t6).dst_port = htons(7890); \
(t6).ip_proto = IPPROTO_TCP; \
(t6).domain = 789; \
}
#define TUPLE6_TO_TUPLE4(t6, t4) \
{ \
memset(&(t4), 0, sizeof(struct tuple4)); \
(t4).ip_type = (t6).ip_type; \
(t4).src_addr = (t6).src_addr; \
(t4).dst_addr = (t6).dst_addr; \
(t4).src_port = (t6).src_port; \
(t4).dst_port = (t6).dst_port; \
}
static void session_free_callback(struct session *sess, void *arg)
{
@@ -46,25 +59,47 @@ static void session_free_callback(struct session *sess, void *arg)
}
}
#if 1
TEST(SESSION_TABLE, OP_SESSION)
{
struct session *sess1 = NULL;
struct session *sess2 = NULL;
struct session *sess3 = NULL;
struct tuple4 sess1_tup4;
struct tuple4 sess2_tup4;
struct tuple4 sess3_tup4;
struct tuple4 sess1_rev_tup4;
struct tuple4 sess2_rev_tup4;
struct tuple4 sess3_rev_tup4;
struct tuple6 sess1_tup6;
struct tuple6 sess2_tup6;
struct tuple6 sess3_tup6;
struct tuple6 sess1_rev_tup6;
struct tuple6 sess2_rev_tup6;
struct tuple6 sess3_rev_tup6;
struct session_pool *sess_pool = NULL;
struct session_table *sess_table = NULL;
SESSION_KEY_IPV4_TCP(tuple_1);
SESSION_KEY_IPV6_UDP(tuple_2);
SESSION_KEY_IPV6_TCP(tuple_3);
TUPLE6_SET_V4_TCP(sess1_tup6);
TUPLE6_SET_V6_UDP(sess2_tup6);
TUPLE6_SET_V6_TCP(sess3_tup6);
struct tuple6 reversed_tuple_1;
struct tuple6 reversed_tuple_2;
struct tuple6 reversed_tuple_3;
TUPLE6_TO_TUPLE4(sess1_tup6, sess1_tup4);
TUPLE6_TO_TUPLE4(sess2_tup6, sess2_tup4);
TUPLE6_TO_TUPLE4(sess3_tup6, sess3_tup4);
tuple6_reverse(&tuple_1, &reversed_tuple_1);
tuple6_reverse(&tuple_2, &reversed_tuple_2);
tuple6_reverse(&tuple_3, &reversed_tuple_3);
tuple4_reverse(&sess1_tup4, &sess1_rev_tup4);
tuple4_reverse(&sess2_tup4, &sess2_rev_tup4);
tuple4_reverse(&sess3_tup4, &sess3_rev_tup4);
tuple6_reverse(&sess1_tup6, &sess1_rev_tup6);
tuple6_reverse(&sess2_tup6, &sess2_rev_tup6);
tuple6_reverse(&sess3_tup6, &sess3_rev_tup6);
// Create
sess_pool = session_pool_new(3);
@@ -77,66 +112,85 @@ TEST(SESSION_TABLE, OP_SESSION)
sess1 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess1 != NULL);
session_set_id(sess1, 1);
session_set_tuple(sess1, &tuple_1);
session_set_tuple(sess1, &sess1_tup6);
sess2 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess2 != NULL);
session_set_id(sess2, 2);
session_set_tuple(sess2, &tuple_2);
session_set_tuple(sess2, &sess2_tup6);
sess3 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess3 != NULL);
session_set_id(sess3, 3);
session_set_tuple(sess3, &tuple_3);
session_set_tuple(sess3, &sess3_tup6);
EXPECT_TRUE(session_table_add(sess_table, &tuple_1, sess1) == 0);
EXPECT_TRUE(session_table_add(sess_table, &sess1_tup6, sess1) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 1);
EXPECT_TRUE(session_table_add(sess_table, &tuple_2, sess2) == 0);
EXPECT_TRUE(session_table_add(sess_table, &sess2_tup6, sess2) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 2);
EXPECT_TRUE(session_table_add(sess_table, &tuple_3, sess3) == 0);
EXPECT_TRUE(session_table_add(sess_table, &sess3_tup6, sess3) == 0);
EXPECT_TRUE(session_table_get_count(sess_table) == 3);
// Search
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_1) == sess1);
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_2) == sess2);
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_3) == sess3);
EXPECT_TRUE(session_table_find_sessid(sess_table, 1) == sess1);
EXPECT_TRUE(session_table_find_sessid(sess_table, 2) == sess2);
EXPECT_TRUE(session_table_find_sessid(sess_table, 3) == sess3);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_1) == sess1);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_2) == sess2);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_3) == sess3);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess1_tup4) == sess1);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess2_tup4) == sess2);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess3_tup4) == sess3);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess1_rev_tup4) == sess1);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess2_rev_tup4) == sess2);
EXPECT_TRUE(session_table_find_tuple4(sess_table, &sess3_rev_tup4) == sess3);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess1_tup6) == sess1);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess2_tup6) == sess2);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess3_tup6) == sess3);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess1_rev_tup6) == sess1);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess2_rev_tup6) == sess2);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess3_rev_tup6) == sess3);
// Delete
session_table_del(sess_table, &tuple_1);
session_table_del(sess_table, &sess1_tup6);
EXPECT_TRUE(session_table_get_count(sess_table) == 2);
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_1) == NULL);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_1) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess1_tup6) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess1_rev_tup6) == NULL);
session_table_del(sess_table, &reversed_tuple_2);
session_table_del(sess_table, &sess2_rev_tup6);
EXPECT_TRUE(session_table_get_count(sess_table) == 1);
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_2) == NULL);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_2) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess2_tup6) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess2_rev_tup6) == NULL);
session_table_del(sess_table, &tuple_3);
session_table_del(sess_table, &sess3_tup6);
EXPECT_TRUE(session_table_get_count(sess_table) == 0);
EXPECT_TRUE(session_table_find_tuple(sess_table, &tuple_3) == NULL);
EXPECT_TRUE(session_table_find_tuple(sess_table, &reversed_tuple_3) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess3_tup6) == NULL);
EXPECT_TRUE(session_table_find_tuple6(sess_table, &sess3_rev_tup6) == NULL);
// Destroy
session_table_free(sess_table);
session_pool_free(sess_pool);
}
#endif
#if 1
TEST(SESSION_TABLE, FIND_OLDEST_NEWEST)
{
struct session *sess1 = NULL;
struct session *sess2 = NULL;
struct session *sess3 = NULL;
struct tuple6 sess1_tup6;
struct tuple6 sess2_tup6;
struct tuple6 sess3_tup6;
struct session_pool *sess_pool = NULL;
struct session_table *sess_table = NULL;
SESSION_KEY_IPV4_TCP(tuple_1);
SESSION_KEY_IPV6_UDP(tuple_2);
SESSION_KEY_IPV6_TCP(tuple_3);
TUPLE6_SET_V4_TCP(sess1_tup6);
TUPLE6_SET_V6_UDP(sess2_tup6);
TUPLE6_SET_V6_TCP(sess3_tup6);
// Create
sess_pool = session_pool_new(3);
@@ -152,39 +206,40 @@ TEST(SESSION_TABLE, FIND_OLDEST_NEWEST)
sess1 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess1 != NULL);
session_set_id(sess1, 1);
session_set_tuple(sess1, &tuple_1);
EXPECT_TRUE(session_table_add(sess_table, &tuple_1, sess1) == 0);
session_set_tuple(sess1, &sess1_tup6);
EXPECT_TRUE(session_table_add(sess_table, &sess1_tup6, sess1) == 0);
EXPECT_TRUE(session_table_find_lru(sess_table) == sess1);
sess2 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess2 != NULL);
session_set_id(sess2, 2);
session_set_tuple(sess2, &tuple_2);
EXPECT_TRUE(session_table_add(sess_table, &tuple_2, sess2) == 0);
session_set_tuple(sess2, &sess2_tup6);
EXPECT_TRUE(session_table_add(sess_table, &sess2_tup6, sess2) == 0);
EXPECT_TRUE(session_table_find_lru(sess_table) == sess1);
sess3 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess3 != NULL);
session_set_id(sess3, 3);
session_set_tuple(sess3, &tuple_3);
EXPECT_TRUE(session_table_add(sess_table, &tuple_3, sess3) == 0);
session_set_tuple(sess3, &sess3_tup6);
EXPECT_TRUE(session_table_add(sess_table, &sess3_tup6, sess3) == 0);
EXPECT_TRUE(session_table_find_lru(sess_table) == sess1);
// Delete Session
session_table_del(sess_table, &tuple_1);
session_table_del(sess_table, &sess1_tup6);
EXPECT_TRUE(session_table_find_lru(sess_table) == sess2);
session_table_del(sess_table, &tuple_2);
session_table_del(sess_table, &sess2_tup6);
EXPECT_TRUE(session_table_find_lru(sess_table) == sess3);
session_table_del(sess_table, &tuple_3);
session_table_del(sess_table, &sess3_tup6);
EXPECT_TRUE(session_table_find_lru(sess_table) == NULL);
// Destroy
session_table_free(sess_table);
session_pool_free(sess_pool);
}
#endif
int main(int argc, char **argv)
{