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

适配quic头文件变化，优化CPU

Browse Source
This commit is contained in:
liuxueli
2023-06-27 13:57:00 +08:00
parent ca8eff0a2e
commit 9f17fec917
4 changed files with 83 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/src/gtest_common.cpp
View File

@@ -296,9 +296,11 @@ int quic_version_int2string(unsigned int version, char *buff, int buff_len)
return 0;
}
unsigned int quic_protocol_identify(struct streaminfo *a_stream, void *a_packet, char *out_sni, int *out_sni_len, char *out_ua, int *out_ua_len)
struct quic_info *g_test_quic_info;
struct quic_info *quic_protocol_identify(const struct streaminfo *a_stream)
{
return 0;
return g_test_quic_info;
}
extern "C" unsigned char sip_identify_from_to(char* buff,UINT32 buflen,char** from,UINT32* from_len,char** to,UINT32* to_len)

59
test/src/gtest_master.cpp
View File

@@ -2,6 +2,8 @@
#include <string.h>
#include <unistd.h>
#include <MESA/quic.h>
#include "gtest_common.h"
#include "tsg_entry.h"
#include "tsg_sync_state.h"
@@ -1104,6 +1106,63 @@ TEST(TSGMaster, SessionApplicationFullPath)
EXPECT_STREQ("ssl.https.built_in_youtube", out_full_path);
}
extern struct quic_info *g_test_quic_info;
extern int session_l7_protocol_identify(const struct streaminfo * a_stream, struct session_runtime_process_context * srt_process_context, void * a_packet);
TEST(TSGMaster, SessionQuicIdentify)
{
struct streaminfo a_stream={0};
struct tcpdetail pdetail={0};
a_stream.type=STREAM_TYPE_UDP;
a_stream.ptcpdetail=&pdetail;
struct session_runtime_process_context srt_process_context={0};
int ret=session_l7_protocol_identify((const struct streaminfo *)&a_stream, &srt_process_context, NULL);
EXPECT_EQ(0, ret);
struct quic_info quic_info_1;
quic_info_1.client_hello=NULL;
quic_info_1.quic_version=0x00000001;
g_test_quic_info=&quic_info_1;
ret=session_l7_protocol_identify((const struct streaminfo *)&a_stream, &srt_process_context, NULL);
EXPECT_EQ(1, ret);
EXPECT_EQ(0x00000001, srt_process_context.quic_version);
g_test_quic_info=NULL;
struct quic_info quic_info_2;
struct quic_client_hello chello_2={(char *)"google.com", NULL};
quic_info_2.client_hello=&chello_2;
quic_info_2.quic_version=0x00000001;
g_test_quic_info=&quic_info_2;
ret=session_l7_protocol_identify((const struct streaminfo *)&a_stream, &srt_process_context, NULL);
EXPECT_EQ(1, ret);
EXPECT_EQ(0x00000001, srt_process_context.quic_version);
EXPECT_STREQ("google.com", srt_process_context.quic_sni);
dictator_free(0, (void *)srt_process_context.quic_sni);
srt_process_context.quic_sni=NULL;
EXPECT_EQ(nullptr, srt_process_context.quic_ua);
g_test_quic_info=NULL;
struct quic_info quic_info_3;
struct quic_client_hello chello_3={(char *)"google.com", (char *)"Android"};
quic_info_3.client_hello=&chello_3;
quic_info_3.quic_version=0x00000001;
g_test_quic_info=&quic_info_3;
ret=session_l7_protocol_identify((const struct streaminfo *)&a_stream, &srt_process_context, NULL);
EXPECT_EQ(1, ret);
EXPECT_EQ(0x00000001, srt_process_context.quic_version);
EXPECT_STREQ("google.com", srt_process_context.quic_sni);
dictator_free(0, (void *)srt_process_context.quic_sni);
srt_process_context.quic_sni=NULL;
EXPECT_STREQ("Android", srt_process_context.quic_ua);
dictator_free(0, (void *)srt_process_context.quic_ua);
srt_process_context.quic_ua=NULL;
g_test_quic_info=NULL;
}
int main(int argc, char *argv[])
{
TSG_MASTER_INIT();

86
test/src/gtest_protocol.cpp
View File

@@ -1,86 +0,0 @@
#include <MESA/stream.h>
extern "C" void *http_field_parser(const char* buf, unsigned int buflen, int http_dir)
{
return NULL;
}
extern "C" int http_get_filed_result(void *result, long long field_flag, char **field_value)
{
return 0;
}
extern "C" void http_free_filed_result(void *result)
{
}
extern "C" int ftp_control_identify(struct streaminfo *a_tcp)
{
return 0;
}
int g_gtest_ftp_data_flag=0;
void gtest_set_ftp_data_flag(void)
{
g_gtest_ftp_data_flag=1;
}
void gtest_clean_ftp_data_flag(void)
{
g_gtest_ftp_data_flag=0;
}
extern "C" int ftp_data_identify(struct streaminfo *a_tcp)
{
return g_gtest_ftp_data_flag;
}
extern "C" int mail_protocol_identify_by_first_payload(struct streaminfo *a_tcp, char *payload, int payload_len, int thread_seq)
{
return 0;
}
extern "C" char stratum_identify(struct streaminfo* pstream,void** pme,int thread_seq,const void* a_packet)
{
return 0;
}
extern "C" int rdp_protocol_identify(const char *payload, unsigned int payload_len, int thread_seq)
{
return 0;
}
extern "C" int rdpudp_protocol_identify(const char *payload, unsigned int payload_len, int thread_seq)
{
return 0;
}
int quic_version_int2string(unsigned int version, char *buff, int buff_len)
{
return 0;
}
unsigned int quic_protocol_identify(struct streaminfo *a_stream, void *a_packet, char *out_sni, int *out_sni_len, char *out_ua, int *out_ua_len)
{
return 0;
}
extern "C" unsigned char sip_identify_from_to(char* buff,UINT32 buflen,char** from,UINT32* from_len,char** to,UINT32* to_len)
{
return 0;
}
extern "C" int dtls_parse_sni(const char *udp_payload, int udp_payload_len, char *servername, int servername_len)
{
return 0;
}
extern "C" bool dtls_identifyStream(struct streaminfo *a_udp)
{
return 0;
}
extern "C" struct _ssl_ja3_info_t *ssl_get_ja3_fingerprint(struct streaminfo *stream, unsigned char *payload, int payload_len, int thread_seq)
{
return NULL;
}