TSG-8364: 重构代码，只解QUIC链接的第一个数据包
This commit is contained in:
liuxueli
336
src/quic_entry.cpp
Normal file
336
src/quic_entry.cpp
Normal file
@@ -0,0 +1,336 @@
|
||||
/*
|
||||
* quic_entry.cpp
|
||||
*
|
||||
* Created on: 2021-11-09
|
||||
* Author: liuxueli
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <MESA/stream_inc/stream_base.h>
|
||||
#include <MESA/MESA_handle_logger.h>
|
||||
#include <MESA/MESA_prof_load.h>
|
||||
#include "quic.h"
|
||||
#include "quic_entry.h"
|
||||
#include "quic_process.h"
|
||||
#include "quic_deprotection.h"
|
||||
|
||||
struct quic_param g_quic_param;
|
||||
const char *g_quic_proto_conffile="./conf/quic/main.conf";
|
||||
const char *g_quic_regionname_conffile="./conf/quic/quic.conf";
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C"
|
||||
{
|
||||
#endif
|
||||
|
||||
#define GIT_VERSION_CATTER(v) __attribute__((__used__)) const char * GIT_VERSION_##v = NULL
|
||||
#define GIT_VERSION_EXPEND(v) GIT_VERSION_CATTER(v)
|
||||
|
||||
/* VERSION TAG */
|
||||
#ifdef GIT_VERSION
|
||||
GIT_VERSION_EXPEND(GIT_VERSION);
|
||||
#else
|
||||
static __attribute__((__used__)) const char * GIT_VERSION_UNKNOWN = NULL;
|
||||
#endif
|
||||
#undef GIT_VERSION_CATTER
|
||||
#undef GIT_VERSION_EXPEND
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
const char QUIC_VERSION_20200603=0;
|
||||
|
||||
static int parse_quic_port(char *port_list, unsigned short *quic_port, int quic_port_num)
|
||||
{
|
||||
int i=0,ret=0;
|
||||
int port_num=0;
|
||||
int range_len=0,used_len=0;
|
||||
char buf[256]={0};
|
||||
unsigned short s_port=0,e_port=0;
|
||||
char *begin=NULL,*end=NULL,*pchr=NULL;
|
||||
|
||||
if(port_list==NULL)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
begin=port_list;
|
||||
end=NULL;
|
||||
range_len=strlen(port_list);
|
||||
|
||||
while(range_len>used_len)
|
||||
{
|
||||
end=index(begin, ';');
|
||||
if(end==NULL)
|
||||
{
|
||||
end=begin+range_len-used_len;
|
||||
}
|
||||
|
||||
if(end==begin)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
memset(buf, 0, sizeof(buf));
|
||||
strncpy(buf, begin, end-begin);
|
||||
used_len+=end-begin+1;
|
||||
if(range_len>used_len)
|
||||
{
|
||||
begin=end+1;
|
||||
}
|
||||
|
||||
pchr=strchr(buf, '-');
|
||||
if(pchr == NULL)
|
||||
{
|
||||
s_port=(unsigned short)atoi(buf);
|
||||
e_port=s_port;
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
ret=sscanf(buf, "%hu-%hu", &s_port, &e_port);
|
||||
if(ret!=2)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
for(i=s_port; i<=e_port && port_num<quic_port_num; i++)
|
||||
{
|
||||
quic_port[port_num++]=i;
|
||||
}
|
||||
}
|
||||
|
||||
return port_num;
|
||||
}
|
||||
|
||||
|
||||
int quic_init_context(void **pme, int thread_seq)
|
||||
{
|
||||
struct quic_context *context=(struct quic_context *)dictator_malloc(thread_seq, sizeof(struct quic_context));
|
||||
memset(context, 0, sizeof(struct quic_context));
|
||||
|
||||
*pme=(void*)context;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void quic_free_client_hello(struct quic_client_hello *client_hello, int thread_seq)
|
||||
{
|
||||
if(client_hello==NULL)
|
||||
{
|
||||
return ;
|
||||
}
|
||||
|
||||
if(client_hello->sni)
|
||||
{
|
||||
dictator_free(thread_seq, client_hello->sni);
|
||||
client_hello->sni=NULL;
|
||||
}
|
||||
|
||||
if(client_hello->user_agent)
|
||||
{
|
||||
dictator_free(thread_seq, client_hello->user_agent);
|
||||
client_hello->user_agent=NULL;
|
||||
}
|
||||
|
||||
dictator_free(thread_seq, client_hello);
|
||||
client_hello=NULL;
|
||||
|
||||
return ;
|
||||
}
|
||||
|
||||
void quic_free_context(void** pme, int thread_seq)
|
||||
{
|
||||
if(NULL==*pme)
|
||||
{
|
||||
return ;
|
||||
}
|
||||
|
||||
struct quic_context *context = (struct quic_context *)*pme;
|
||||
quic_free_client_hello(context->quic_info.client_hello, thread_seq);
|
||||
|
||||
dictator_free(thread_seq, *pme);
|
||||
*pme=NULL;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
extern "C" unsigned char QUIC_ENTRY(struct streaminfo *pstream, void**pme, int thread_seq, void *a_packet)
|
||||
{
|
||||
unsigned char state=0;
|
||||
struct quic_context *context=(struct quic_context *)*pme;
|
||||
|
||||
if((g_quic_param.quic_interested_region_flag<QUIC_KEY) || (!is_quic_port(pstream)))
|
||||
{
|
||||
return APP_STATE_DROPME;
|
||||
}
|
||||
|
||||
if(*pme==NULL)
|
||||
{
|
||||
quic_init_context(pme, thread_seq);
|
||||
context=(struct quic_context *)*pme;
|
||||
}
|
||||
|
||||
switch(pstream->opstate)
|
||||
{
|
||||
case OP_STATE_PENDING:
|
||||
state=quic_analyze_entry(pstream, context, thread_seq, a_packet);
|
||||
break;
|
||||
case OP_STATE_DATA:
|
||||
state=quic_call_business_plug(pstream, context, (char *)pstream->pudpdetail->pdata, pstream->pudpdetail->datalen, QUIC_APPLICATION_DATA_MASK, a_packet);
|
||||
break;
|
||||
case OP_STATE_CLOSE:
|
||||
if(pstream->pudpdetail->pdata!=NULL && pstream->pudpdetail->datalen>0)
|
||||
{
|
||||
state=quic_call_business_plug(pstream, context, (char *)pstream->pudpdetail->pdata, pstream->pudpdetail->datalen, QUIC_APPLICATION_DATA_MASK, a_packet);
|
||||
}
|
||||
state=quic_call_business_plug(pstream, context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet);
|
||||
break;
|
||||
}
|
||||
|
||||
if(state&APP_STATE_DROPME || state&APP_STATE_DROPPKT || pstream->opstate==OP_STATE_CLOSE)
|
||||
{
|
||||
quic_free_context(pme, thread_seq);
|
||||
*pme=NULL;
|
||||
return state;
|
||||
}
|
||||
|
||||
return APP_STATE_GIVEME;
|
||||
}
|
||||
|
||||
extern "C" int QUIC_INIT(void)
|
||||
{
|
||||
int ret=0;
|
||||
FILE *fp=NULL;
|
||||
char buff[2048]={0};
|
||||
int region_id=0;
|
||||
char region_name[REGION_NAME_LEN]={0};
|
||||
|
||||
memset(&g_quic_param,0,sizeof(struct quic_param));
|
||||
|
||||
MESA_load_profile_int_def(g_quic_proto_conffile, "QUIC", "LOG_LEVEL", &g_quic_param.level, RLOG_LV_FATAL);
|
||||
MESA_load_profile_string_def(g_quic_proto_conffile, "QUIC", "LOG_PATH", g_quic_param.log_path, sizeof(g_quic_param.log_path), "./log/quic/quic");
|
||||
|
||||
MESA_load_profile_int_def(g_quic_proto_conffile, "QUIC", "DECRYPTED_SWITCH", &g_quic_param.decrypted_switch, 2);
|
||||
|
||||
MESA_load_profile_string_def(g_quic_proto_conffile, "QUIC", "QUIC_PORT_LIST", buff, sizeof(buff), "443;8443;");
|
||||
g_quic_param.quic_port_num=parse_quic_port(buff, g_quic_param.quic_port_list, SUPPORT_QUIC_PORT_NUM);
|
||||
|
||||
g_quic_param.logger=MESA_create_runtime_log_handle(g_quic_param.log_path, g_quic_param.level);
|
||||
if(g_quic_param.logger==NULL)
|
||||
{
|
||||
printf("MESA_create_runtime_log_handle failed, level: %d log_path: %s", g_quic_param.level, g_quic_param.log_path);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if(((fp = fopen(g_quic_regionname_conffile, "r"))!=NULL))
|
||||
{
|
||||
while(fgets(buff, sizeof(buff), fp))
|
||||
{
|
||||
ret = sscanf(buff, "%d\t%s", ®ion_id, region_name);
|
||||
if(2>ret)
|
||||
{
|
||||
fclose(fp);
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_READCONF", "Read error, Please check %s, region_line: %s", g_quic_regionname_conffile, buff);
|
||||
return -1;
|
||||
}
|
||||
if(region_id>MAX_REGION_NUM)
|
||||
{
|
||||
fclose(fp);
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_READCONF", "Read error, Please check %s, bigger than MAX_REGION_NUM, region_line: %s", g_quic_regionname_conffile, buff);
|
||||
return -1;
|
||||
}
|
||||
|
||||
strncpy(g_quic_param.quic_conf_regionname[region_id], region_name, strlen(region_name));
|
||||
g_quic_param.quic_region_cnt++;
|
||||
memset(region_name, 0, sizeof(region_name));
|
||||
}
|
||||
|
||||
fclose(fp);
|
||||
}
|
||||
else
|
||||
{
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_READCONF", "Open %s error, Please check %s", g_quic_regionname_conffile, g_quic_regionname_conffile);
|
||||
return -1;
|
||||
}
|
||||
|
||||
gcry_init();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
extern "C" void QUIC_GETPLUGID(unsigned short plugid)
|
||||
{
|
||||
g_quic_param.quic_plugid = plugid;
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_GETPLUGID", "quic_plugid: %d", plugid);
|
||||
}
|
||||
|
||||
extern "C" void QUIC_PROT_FUNSTAT(unsigned long long protflag)
|
||||
{
|
||||
if(0==protflag){
|
||||
return;
|
||||
}
|
||||
g_quic_param.quic_interested_region_flag=protflag;
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_PROT_FUNSTAT", "interested_region_flag: %llu", g_quic_param.quic_interested_region_flag);
|
||||
return;
|
||||
}
|
||||
|
||||
extern "C" unsigned long long quic_getRegionID(char *string, int str_len,const char g_string[MAX_REGION_NUM][REGION_NAME_LEN])
|
||||
{
|
||||
unsigned long long i=0;
|
||||
for(i=0;i<g_quic_param.quic_region_cnt;i++)
|
||||
{
|
||||
if(0==strcasecmp(g_string[i], string))
|
||||
{
|
||||
return i;
|
||||
}
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
extern "C" long long QUIC_FLAG_CHANGE(char* flag_str)
|
||||
{
|
||||
if(flag_str==NULL) return -1;
|
||||
long long protflag = 0;
|
||||
long long region_id = 0;
|
||||
char *start_token = flag_str;
|
||||
char *end_token = flag_str;
|
||||
char *end_pos = flag_str+strlen(flag_str);
|
||||
char region_name[REGION_NAME_LEN] = {0};
|
||||
|
||||
while (end_token < end_pos)
|
||||
{
|
||||
end_token = (char*)memchr(start_token, ',', end_pos-start_token);
|
||||
if(end_token!=NULL)
|
||||
{
|
||||
memcpy(region_name, start_token, end_token-start_token);
|
||||
start_token = end_token+1;
|
||||
end_token += 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
memcpy(region_name, start_token, end_pos-start_token);
|
||||
end_token = end_pos;
|
||||
}
|
||||
region_id = quic_getRegionID(region_name, strlen(region_name), g_quic_param.quic_conf_regionname);
|
||||
if(-1==region_id)
|
||||
{
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_FLAG_CHANGE", "Read %s error, flag_str: %d", region_name, flag_str);
|
||||
return -1;
|
||||
}
|
||||
protflag |= ((long long)1)<<region_id;
|
||||
memset(region_name, 0, REGION_NAME_LEN);
|
||||
}
|
||||
|
||||
|
||||
MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_FLAG_CHANGE", "protflag: %llu", protflag);
|
||||
return protflag;
|
||||
}
|
||||
|
||||
extern "C" void QUIC_DESTROY(void)
|
||||
{
|
||||
return ;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user