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

refactor: change session key from address list to six tuple

Browse Source
This commit is contained in:
luwenpeng
2023-12-14 15:23:13 +08:00
parent 5620ac211b
commit a232045764
14 changed files with 773 additions and 563 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
src/session/gtest_session.cpp
View File

@@ -2,6 +2,39 @@
#include "session_private.h"
#define SESSION_KEY_IPV4_TCP(name) \
struct session_key name; \
memset(&name, 0, sizeof(struct session_key)); \
(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).security_zone = 0;
#define SESSION_KEY_IPV6_UDP(name) \
struct session_key name; \
memset(&name, 0, sizeof(struct session_key)); \
(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).security_zone = 0;
#define SESSION_KEY_IPV6_TCP(name) \
struct session_key name; \
memset(&name, 0, sizeof(struct session_key)); \
(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).security_zone = 0;
void free_ex_data(struct session *sess, uint8_t idx, void *ex_ptr, void *arg)
{
if (ex_ptr)
@@ -78,6 +111,67 @@ TEST(SESSION, EV_QUEUE)
}
}
TEST(SESSION, SESSION_KEY)
{
char buf[128] = {0};
SESSION_KEY_IPV4_TCP(key1);
SESSION_KEY_IPV6_UDP(key2);
SESSION_KEY_IPV6_TCP(key3);
struct session_key reverse_key1;
struct session_key reverse_key2;
struct session_key reverse_key3;
struct session_key reverse_temp;
// tostring
memset(buf, 0, sizeof(buf));
session_key_tostring(&key1, buf, sizeof(buf));
EXPECT_STREQ(buf, "192.168.1.2:1234 -> 192.168.1.3:5678, proto: 6, zone: 0");
memset(buf, 0, sizeof(buf));
session_key_tostring(&key2, buf, sizeof(buf));
EXPECT_STREQ(buf, "2001:db8::ff00:42:8329:1234 -> 2001:db8::ff00:42:832a:5678, proto: 17, zone: 0");
memset(buf, 0, sizeof(buf));
session_key_tostring(&key3, buf, sizeof(buf));
EXPECT_STREQ(buf, "2001:db8::ff00:42:8329:1234 -> 2001:db8::ff00:42:832a:5678, proto: 6, zone: 0");
// reverse
session_key_reverse(&key1, &reverse_key1);
session_key_reverse(&key2, &reverse_key2);
session_key_reverse(&key3, &reverse_key3);
memset(buf, 0, sizeof(buf));
session_key_tostring(&reverse_key1, buf, sizeof(buf));
EXPECT_STREQ(buf, "192.168.1.3:5678 -> 192.168.1.2:1234, proto: 6, zone: 0");
memset(buf, 0, sizeof(buf));
session_key_tostring(&reverse_key2, buf, sizeof(buf));
EXPECT_STREQ(buf, "2001:db8::ff00:42:832a:5678 -> 2001:db8::ff00:42:8329:1234, proto: 17, zone: 0");
memset(buf, 0, sizeof(buf));
session_key_tostring(&reverse_key3, buf, sizeof(buf));
EXPECT_STREQ(buf, "2001:db8::ff00:42:832a:5678 -> 2001:db8::ff00:42:8329:1234, proto: 6, zone: 0");
// hash
EXPECT_TRUE(session_key_hash(&key1) == session_key_hash(&reverse_key1));
EXPECT_TRUE(session_key_hash(&key2) == session_key_hash(&reverse_key2));
EXPECT_TRUE(session_key_hash(&key3) == session_key_hash(&reverse_key3));
// cmp
EXPECT_TRUE(session_key_cmp(&key1, &key1) == 0);
EXPECT_TRUE(session_key_cmp(&key2, &key2) == 0);
EXPECT_TRUE(session_key_cmp(&key3, &key3) == 0);
EXPECT_TRUE(session_key_cmp(&key1, &reverse_key1) != 0);
EXPECT_TRUE(session_key_cmp(&key2, &reverse_key2) != 0);
EXPECT_TRUE(session_key_cmp(&key3, &reverse_key3) != 0);
session_key_reverse(&reverse_key1, &reverse_temp);
EXPECT_TRUE(session_key_cmp(&key1, &reverse_temp) == 0);
session_key_reverse(&reverse_key2, &reverse_temp);
EXPECT_TRUE(session_key_cmp(&key2, &reverse_temp) == 0);
session_key_reverse(&reverse_key3, &reverse_temp);
EXPECT_TRUE(session_key_cmp(&key3, &reverse_temp) == 0);
}
int main(int argc, char **argv)
{
::testing::InitGoogleTest(&argc, argv);